Andrew's git / gitweb.git / http-fetch.c
?
http-fetch.con commit git-am: Don't accept an mbox on stdin of we already have a .dotest directory (cbd64af)
   1#include "cache.h"
   2#include "commit.h"
   3#include "pack.h"
   4#include "fetch.h"
   5#include "http.h"
   6
   7#ifndef NO_EXPAT
   8#include <expat.h>
   9
  10/* Definitions for DAV requests */
  11#define DAV_PROPFIND "PROPFIND"
  12#define DAV_PROPFIND_RESP ".multistatus.response"
  13#define DAV_PROPFIND_NAME ".multistatus.response.href"
  14#define DAV_PROPFIND_COLLECTION ".multistatus.response.propstat.prop.resourcetype.collection"
  15#define PROPFIND_ALL_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:propfind xmlns:D=\"DAV:\">\n<D:allprop/>\n</D:propfind>"
  16
  17/* Definitions for processing XML DAV responses */
  18#ifndef XML_STATUS_OK
  19enum XML_Status {
  20  XML_STATUS_OK = 1,
  21  XML_STATUS_ERROR = 0
  22};
  23#define XML_STATUS_OK    1
  24#define XML_STATUS_ERROR 0
  25#endif
  26
  27/* Flags that control remote_ls processing */
  28#define PROCESS_FILES (1u << 0)
  29#define PROCESS_DIRS  (1u << 1)
  30#define RECURSIVE     (1u << 2)
  31
  32/* Flags that remote_ls passes to callback functions */
  33#define IS_DIR (1u << 0)
  34#endif
  35
  36#define PREV_BUF_SIZE 4096
  37#define RANGE_HEADER_SIZE 30
  38
  39static int got_alternates = -1;
  40static int corrupt_object_found = 0;
  41
  42static struct curl_slist *no_pragma_header;
  43
  44struct alt_base
  45{
  46        char *base;
  47        int path_len;
  48        int got_indices;
  49        struct packed_git *packs;
  50        struct alt_base *next;
  51};
  52
  53static struct alt_base *alt = NULL;
  54
  55enum object_request_state {
  56        WAITING,
  57        ABORTED,
  58        ACTIVE,
  59        COMPLETE,
  60};
  61
  62struct object_request
  63{
  64        unsigned char sha1[20];
  65        struct alt_base *repo;
  66        char *url;
  67        char filename[PATH_MAX];
  68        char tmpfile[PATH_MAX];
  69        int local;
  70        enum object_request_state state;
  71        CURLcode curl_result;
  72        char errorstr[CURL_ERROR_SIZE];
  73        long http_code;
  74        unsigned char real_sha1[20];
  75        SHA_CTX c;
  76        z_stream stream;
  77        int zret;
  78        int rename;
  79        struct active_request_slot *slot;
  80        struct object_request *next;
  81};
  82
  83struct alternates_request {
  84        char *base;
  85        char *url;
  86        struct buffer *buffer;
  87        struct active_request_slot *slot;
  88        int http_specific;
  89};
  90
  91#ifndef NO_EXPAT
  92struct xml_ctx
  93{
  94        char *name;
  95        int len;
  96        char *cdata;
  97        void (*userFunc)(struct xml_ctx *ctx, int tag_closed);
  98        void *userData;
  99};
 100
 101struct remote_ls_ctx
 102{
 103        struct alt_base *repo;
 104        char *path;
 105        void (*userFunc)(struct remote_ls_ctx *ls);
 106        void *userData;
 107        int flags;
 108        char *dentry_name;
 109        int dentry_flags;
 110        int rc;
 111        struct remote_ls_ctx *parent;
 112};
 113#endif
 114
 115static struct object_request *object_queue_head = NULL;
 116
 117static size_t fwrite_sha1_file(void *ptr, size_t eltsize, size_t nmemb,
 118                               void *data)
 119{
 120        unsigned char expn[4096];
 121        size_t size = eltsize * nmemb;
 122        int posn = 0;
 123        struct object_request *obj_req = (struct object_request *)data;
 124        do {
 125                ssize_t retval = write(obj_req->local,
 126                                       (char *) ptr + posn, size - posn);
 127                if (retval < 0)
 128                        return posn;
 129                posn += retval;
 130        } while (posn < size);
 131
 132        obj_req->stream.avail_in = size;
 133        obj_req->stream.next_in = ptr;
 134        do {
 135                obj_req->stream.next_out = expn;
 136                obj_req->stream.avail_out = sizeof(expn);
 137                obj_req->zret = inflate(&obj_req->stream, Z_SYNC_FLUSH);
 138                SHA1_Update(&obj_req->c, expn,
 139                            sizeof(expn) - obj_req->stream.avail_out);
 140        } while (obj_req->stream.avail_in && obj_req->zret == Z_OK);
 141        data_received++;
 142        return size;
 143}
 144
 145static void fetch_alternates(char *base);
 146
 147static void process_object_response(void *callback_data);
 148
 149static void start_object_request(struct object_request *obj_req)
 150{
 151        char *hex = sha1_to_hex(obj_req->sha1);
 152        char prevfile[PATH_MAX];
 153        char *url;
 154        char *posn;
 155        int prevlocal;
 156        unsigned char prev_buf[PREV_BUF_SIZE];
 157        ssize_t prev_read = 0;
 158        long prev_posn = 0;
 159        char range[RANGE_HEADER_SIZE];
 160        struct curl_slist *range_header = NULL;
 161        struct active_request_slot *slot;
 162
 163        snprintf(prevfile, sizeof(prevfile), "%s.prev", obj_req->filename);
 164        unlink(prevfile);
 165        rename(obj_req->tmpfile, prevfile);
 166        unlink(obj_req->tmpfile);
 167
 168        if (obj_req->local != -1)
 169                error("fd leakage in start: %d", obj_req->local);
 170        obj_req->local = open(obj_req->tmpfile,
 171                              O_WRONLY | O_CREAT | O_EXCL, 0666);
 172        /* This could have failed due to the "lazy directory creation";
 173         * try to mkdir the last path component.
 174         */
 175        if (obj_req->local < 0 && errno == ENOENT) {
 176                char *dir = strrchr(obj_req->tmpfile, '/');
 177                if (dir) {
 178                        *dir = 0;
 179                        mkdir(obj_req->tmpfile, 0777);
 180                        *dir = '/';
 181                }
 182                obj_req->local = open(obj_req->tmpfile,
 183                                      O_WRONLY | O_CREAT | O_EXCL, 0666);
 184        }
 185
 186        if (obj_req->local < 0) {
 187                obj_req->state = ABORTED;
 188                error("Couldn't create temporary file %s for %s: %s",
 189                      obj_req->tmpfile, obj_req->filename, strerror(errno));
 190                return;
 191        }
 192
 193        memset(&obj_req->stream, 0, sizeof(obj_req->stream));
 194
 195        inflateInit(&obj_req->stream);
 196
 197        SHA1_Init(&obj_req->c);
 198
 199        url = xmalloc(strlen(obj_req->repo->base) + 50);
 200        obj_req->url = xmalloc(strlen(obj_req->repo->base) + 50);
 201        strcpy(url, obj_req->repo->base);
 202        posn = url + strlen(obj_req->repo->base);
 203        strcpy(posn, "objects/");
 204        posn += 8;
 205        memcpy(posn, hex, 2);
 206        posn += 2;
 207        *(posn++) = '/';
 208        strcpy(posn, hex + 2);
 209        strcpy(obj_req->url, url);
 210
 211        /* If a previous temp file is present, process what was already
 212           fetched. */
 213        prevlocal = open(prevfile, O_RDONLY);
 214        if (prevlocal != -1) {
 215                do {
 216                        prev_read = read(prevlocal, prev_buf, PREV_BUF_SIZE);
 217                        if (prev_read>0) {
 218                                if (fwrite_sha1_file(prev_buf,
 219                                                     1,
 220                                                     prev_read,
 221                                                     obj_req) == prev_read) {
 222                                        prev_posn += prev_read;
 223                                } else {
 224                                        prev_read = -1;
 225                                }
 226                        }
 227                } while (prev_read > 0);
 228                close(prevlocal);
 229        }
 230        unlink(prevfile);
 231
 232        /* Reset inflate/SHA1 if there was an error reading the previous temp
 233           file; also rewind to the beginning of the local file. */
 234        if (prev_read == -1) {
 235                memset(&obj_req->stream, 0, sizeof(obj_req->stream));
 236                inflateInit(&obj_req->stream);
 237                SHA1_Init(&obj_req->c);
 238                if (prev_posn>0) {
 239                        prev_posn = 0;
 240                        lseek(obj_req->local, SEEK_SET, 0);
 241                        ftruncate(obj_req->local, 0);
 242                }
 243        }
 244
 245        slot = get_active_slot();
 246        slot->callback_func = process_object_response;
 247        slot->callback_data = obj_req;
 248        obj_req->slot = slot;
 249
 250        curl_easy_setopt(slot->curl, CURLOPT_FILE, obj_req);
 251        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_sha1_file);
 252        curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, obj_req->errorstr);
 253        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 254        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header);
 255
 256        /* If we have successfully processed data from a previous fetch
 257           attempt, only fetch the data we don't already have. */
 258        if (prev_posn>0) {
 259                if (get_verbosely)
 260                        fprintf(stderr,
 261                                "Resuming fetch of object %s at byte %ld\n",
 262                                hex, prev_posn);
 263                sprintf(range, "Range: bytes=%ld-", prev_posn);
 264                range_header = curl_slist_append(range_header, range);
 265                curl_easy_setopt(slot->curl,
 266                                 CURLOPT_HTTPHEADER, range_header);
 267        }
 268
 269        /* Try to get the request started, abort the request on error */
 270        obj_req->state = ACTIVE;
 271        if (!start_active_slot(slot)) {
 272                obj_req->state = ABORTED;
 273                obj_req->slot = NULL;
 274                close(obj_req->local); obj_req->local = -1;
 275                free(obj_req->url);
 276                return;
 277        }
 278}
 279
 280static void finish_object_request(struct object_request *obj_req)
 281{
 282        struct stat st;
 283
 284        fchmod(obj_req->local, 0444);
 285        close(obj_req->local); obj_req->local = -1;
 286
 287        if (obj_req->http_code == 416) {
 288                fprintf(stderr, "Warning: requested range invalid; we may already have all the data.\n");
 289        } else if (obj_req->curl_result != CURLE_OK) {
 290                if (stat(obj_req->tmpfile, &st) == 0)
 291                        if (st.st_size == 0)
 292                                unlink(obj_req->tmpfile);
 293                return;
 294        }
 295
 296        inflateEnd(&obj_req->stream);
 297        SHA1_Final(obj_req->real_sha1, &obj_req->c);
 298        if (obj_req->zret != Z_STREAM_END) {
 299                unlink(obj_req->tmpfile);
 300                return;
 301        }
 302        if (memcmp(obj_req->sha1, obj_req->real_sha1, 20)) {
 303                unlink(obj_req->tmpfile);
 304                return;
 305        }
 306        obj_req->rename =
 307                move_temp_to_file(obj_req->tmpfile, obj_req->filename);
 308
 309        if (obj_req->rename == 0)
 310                pull_say("got %s\n", sha1_to_hex(obj_req->sha1));
 311}
 312
 313static void process_object_response(void *callback_data)
 314{
 315        struct object_request *obj_req =
 316                (struct object_request *)callback_data;
 317
 318        obj_req->curl_result = obj_req->slot->curl_result;
 319        obj_req->http_code = obj_req->slot->http_code;
 320        obj_req->slot = NULL;
 321        obj_req->state = COMPLETE;
 322
 323        /* Use alternates if necessary */
 324        if (obj_req->http_code == 404 ||
 325            obj_req->curl_result == CURLE_FILE_COULDNT_READ_FILE) {
 326                fetch_alternates(alt->base);
 327                if (obj_req->repo->next != NULL) {
 328                        obj_req->repo =
 329                                obj_req->repo->next;
 330                        close(obj_req->local);
 331                        obj_req->local = -1;
 332                        start_object_request(obj_req);
 333                        return;
 334                }
 335        }
 336
 337        finish_object_request(obj_req);
 338}
 339
 340static void release_object_request(struct object_request *obj_req)
 341{
 342        struct object_request *entry = object_queue_head;
 343
 344        if (obj_req->local != -1)
 345                error("fd leakage in release: %d", obj_req->local);
 346        if (obj_req == object_queue_head) {
 347                object_queue_head = obj_req->next;
 348        } else {
 349                while (entry->next != NULL && entry->next != obj_req)
 350                        entry = entry->next;
 351                if (entry->next == obj_req)
 352                        entry->next = entry->next->next;
 353        }
 354
 355        free(obj_req->url);
 356        free(obj_req);
 357}
 358
 359#ifdef USE_CURL_MULTI
 360void fill_active_slots(void)
 361{
 362        struct object_request *obj_req = object_queue_head;
 363        struct active_request_slot *slot = active_queue_head;
 364        int num_transfers;
 365
 366        while (active_requests < max_requests && obj_req != NULL) {
 367                if (obj_req->state == WAITING) {
 368                        if (has_sha1_file(obj_req->sha1))
 369                                obj_req->state = COMPLETE;
 370                        else
 371                                start_object_request(obj_req);
 372                        curl_multi_perform(curlm, &num_transfers);
 373                }
 374                obj_req = obj_req->next;
 375        }
 376
 377        while (slot != NULL) {
 378                if (!slot->in_use && slot->curl != NULL) {
 379                        curl_easy_cleanup(slot->curl);
 380                        slot->curl = NULL;
 381                }
 382                slot = slot->next;
 383        }
 384}
 385#endif
 386
 387void prefetch(unsigned char *sha1)
 388{
 389        struct object_request *newreq;
 390        struct object_request *tail;
 391        char *filename = sha1_file_name(sha1);
 392
 393        newreq = xmalloc(sizeof(*newreq));
 394        memcpy(newreq->sha1, sha1, 20);
 395        newreq->repo = alt;
 396        newreq->url = NULL;
 397        newreq->local = -1;
 398        newreq->state = WAITING;
 399        snprintf(newreq->filename, sizeof(newreq->filename), "%s", filename);
 400        snprintf(newreq->tmpfile, sizeof(newreq->tmpfile),
 401                 "%s.temp", filename);
 402        newreq->slot = NULL;
 403        newreq->next = NULL;
 404
 405        if (object_queue_head == NULL) {
 406                object_queue_head = newreq;
 407        } else {
 408                tail = object_queue_head;
 409                while (tail->next != NULL) {
 410                        tail = tail->next;
 411                }
 412                tail->next = newreq;
 413        }
 414
 415#ifdef USE_CURL_MULTI
 416        fill_active_slots();
 417        step_active_slots();
 418#endif
 419}
 420
 421static int fetch_index(struct alt_base *repo, unsigned char *sha1)
 422{
 423        char *hex = sha1_to_hex(sha1);
 424        char *filename;
 425        char *url;
 426        char tmpfile[PATH_MAX];
 427        long prev_posn = 0;
 428        char range[RANGE_HEADER_SIZE];
 429        struct curl_slist *range_header = NULL;
 430
 431        FILE *indexfile;
 432        struct active_request_slot *slot;
 433        struct slot_results results;
 434
 435        if (has_pack_index(sha1))
 436                return 0;
 437
 438        if (get_verbosely)
 439                fprintf(stderr, "Getting index for pack %s\n", hex);
 440
 441        url = xmalloc(strlen(repo->base) + 64);
 442        sprintf(url, "%s/objects/pack/pack-%s.idx", repo->base, hex);
 443
 444        filename = sha1_pack_index_name(sha1);
 445        snprintf(tmpfile, sizeof(tmpfile), "%s.temp", filename);
 446        indexfile = fopen(tmpfile, "a");
 447        if (!indexfile)
 448                return error("Unable to open local file %s for pack index",
 449                             filename);
 450
 451        slot = get_active_slot();
 452        slot->results = &results;
 453        curl_easy_setopt(slot->curl, CURLOPT_FILE, indexfile);
 454        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite);
 455        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 456        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header);
 457        slot->local = indexfile;
 458
 459        /* If there is data present from a previous transfer attempt,
 460           resume where it left off */
 461        prev_posn = ftell(indexfile);
 462        if (prev_posn>0) {
 463                if (get_verbosely)
 464                        fprintf(stderr,
 465                                "Resuming fetch of index for pack %s at byte %ld\n",
 466                                hex, prev_posn);
 467                sprintf(range, "Range: bytes=%ld-", prev_posn);
 468                range_header = curl_slist_append(range_header, range);
 469                curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, range_header);
 470        }
 471
 472        if (start_active_slot(slot)) {
 473                run_active_slot(slot);
 474                if (results.curl_result != CURLE_OK) {
 475                        fclose(indexfile);
 476                        return error("Unable to get pack index %s\n%s", url,
 477                                     curl_errorstr);
 478                }
 479        } else {
 480                fclose(indexfile);
 481                return error("Unable to start request");
 482        }
 483
 484        fclose(indexfile);
 485
 486        return move_temp_to_file(tmpfile, filename);
 487}
 488
 489static int setup_index(struct alt_base *repo, unsigned char *sha1)
 490{
 491        struct packed_git *new_pack;
 492        if (has_pack_file(sha1))
 493                return 0; /* don't list this as something we can get */
 494
 495        if (fetch_index(repo, sha1))
 496                return -1;
 497
 498        new_pack = parse_pack_index(sha1);
 499        new_pack->next = repo->packs;
 500        repo->packs = new_pack;
 501        return 0;
 502}
 503
 504static void process_alternates_response(void *callback_data)
 505{
 506        struct alternates_request *alt_req =
 507                (struct alternates_request *)callback_data;
 508        struct active_request_slot *slot = alt_req->slot;
 509        struct alt_base *tail = alt;
 510        char *base = alt_req->base;
 511        static const char null_byte = '\0';
 512        char *data;
 513        int i = 0;
 514
 515        if (alt_req->http_specific) {
 516                if (slot->curl_result != CURLE_OK ||
 517                    !alt_req->buffer->posn) {
 518
 519                        /* Try reusing the slot to get non-http alternates */
 520                        alt_req->http_specific = 0;
 521                        sprintf(alt_req->url, "%s/objects/info/alternates",
 522                                base);
 523                        curl_easy_setopt(slot->curl, CURLOPT_URL,
 524                                         alt_req->url);
 525                        active_requests++;
 526                        slot->in_use = 1;
 527                        if (slot->finished != NULL)
 528                                (*slot->finished) = 0;
 529                        if (!start_active_slot(slot)) {
 530                                got_alternates = -1;
 531                                slot->in_use = 0;
 532                                if (slot->finished != NULL)
 533                                        (*slot->finished) = 1;
 534                        }
 535                        return;
 536                }
 537        } else if (slot->curl_result != CURLE_OK) {
 538                if (slot->http_code != 404 &&
 539                    slot->curl_result != CURLE_FILE_COULDNT_READ_FILE) {
 540                        got_alternates = -1;
 541                        return;
 542                }
 543        }
 544
 545        fwrite_buffer(&null_byte, 1, 1, alt_req->buffer);
 546        alt_req->buffer->posn--;
 547        data = alt_req->buffer->buffer;
 548
 549        while (i < alt_req->buffer->posn) {
 550                int posn = i;
 551                while (posn < alt_req->buffer->posn && data[posn] != '\n')
 552                        posn++;
 553                if (data[posn] == '\n') {
 554                        int okay = 0;
 555                        int serverlen = 0;
 556                        struct alt_base *newalt;
 557                        char *target = NULL;
 558                        char *path;
 559                        if (data[i] == '/') {
 560                                serverlen = strchr(base + 8, '/') - base;
 561                                okay = 1;
 562                        } else if (!memcmp(data + i, "../", 3)) {
 563                                i += 3;
 564                                serverlen = strlen(base);
 565                                while (i + 2 < posn &&
 566                                       !memcmp(data + i, "../", 3)) {
 567                                        do {
 568                                                serverlen--;
 569                                        } while (serverlen &&
 570                                                 base[serverlen - 1] != '/');
 571                                        i += 3;
 572                                }
 573                                /* If the server got removed, give up. */
 574                                okay = strchr(base, ':') - base + 3 <
 575                                        serverlen;
 576                        } else if (alt_req->http_specific) {
 577                                char *colon = strchr(data + i, ':');
 578                                char *slash = strchr(data + i, '/');
 579                                if (colon && slash && colon < data + posn &&
 580                                    slash < data + posn && colon < slash) {
 581                                        okay = 1;
 582                                }
 583                        }
 584                        /* skip 'objects' at end */
 585                        if (okay) {
 586                                target = xmalloc(serverlen + posn - i - 6);
 587                                strlcpy(target, base, serverlen);
 588                                strlcpy(target + serverlen, data + i, posn - i - 6);
 589                                if (get_verbosely)
 590                                        fprintf(stderr,
 591                                                "Also look at %s\n", target);
 592                                newalt = xmalloc(sizeof(*newalt));
 593                                newalt->next = NULL;
 594                                newalt->base = target;
 595                                newalt->got_indices = 0;
 596                                newalt->packs = NULL;
 597                                path = strstr(target, "//");
 598                                if (path) {
 599                                        path = strchr(path+2, '/');
 600                                        if (path)
 601                                                newalt->path_len = strlen(path);
 602                                }
 603
 604                                while (tail->next != NULL)
 605                                        tail = tail->next;
 606                                tail->next = newalt;
 607                        }
 608                }
 609                i = posn + 1;
 610        }
 611
 612        got_alternates = 1;
 613}
 614
 615static void fetch_alternates(char *base)
 616{
 617        struct buffer buffer;
 618        char *url;
 619        char *data;
 620        struct active_request_slot *slot;
 621        struct alternates_request alt_req;
 622
 623        /* If another request has already started fetching alternates,
 624           wait for them to arrive and return to processing this request's
 625           curl message */
 626#ifdef USE_CURL_MULTI
 627        while (got_alternates == 0) {
 628                step_active_slots();
 629        }
 630#endif
 631
 632        /* Nothing to do if they've already been fetched */
 633        if (got_alternates == 1)
 634                return;
 635
 636        /* Start the fetch */
 637        got_alternates = 0;
 638
 639        data = xmalloc(4096);
 640        buffer.size = 4096;
 641        buffer.posn = 0;
 642        buffer.buffer = data;
 643
 644        if (get_verbosely)
 645                fprintf(stderr, "Getting alternates list for %s\n", base);
 646
 647        url = xmalloc(strlen(base) + 31);
 648        sprintf(url, "%s/objects/info/http-alternates", base);
 649
 650        /* Use a callback to process the result, since another request
 651           may fail and need to have alternates loaded before continuing */
 652        slot = get_active_slot();
 653        slot->callback_func = process_alternates_response;
 654        slot->callback_data = &alt_req;
 655
 656        curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
 657        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 658        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 659
 660        alt_req.base = base;
 661        alt_req.url = url;
 662        alt_req.buffer = &buffer;
 663        alt_req.http_specific = 1;
 664        alt_req.slot = slot;
 665
 666        if (start_active_slot(slot))
 667                run_active_slot(slot);
 668        else
 669                got_alternates = -1;
 670
 671        free(data);
 672        free(url);
 673}
 674
 675#ifndef NO_EXPAT
 676static void
 677xml_start_tag(void *userData, const char *name, const char **atts)
 678{
 679        struct xml_ctx *ctx = (struct xml_ctx *)userData;
 680        const char *c = strchr(name, ':');
 681        int new_len;
 682
 683        if (c == NULL)
 684                c = name;
 685        else
 686                c++;
 687
 688        new_len = strlen(ctx->name) + strlen(c) + 2;
 689
 690        if (new_len > ctx->len) {
 691                ctx->name = xrealloc(ctx->name, new_len);
 692                ctx->len = new_len;
 693        }
 694        strcat(ctx->name, ".");
 695        strcat(ctx->name, c);
 696
 697        if (ctx->cdata) {
 698                free(ctx->cdata);
 699                ctx->cdata = NULL;
 700        }
 701
 702        ctx->userFunc(ctx, 0);
 703}
 704
 705static void
 706xml_end_tag(void *userData, const char *name)
 707{
 708        struct xml_ctx *ctx = (struct xml_ctx *)userData;
 709        const char *c = strchr(name, ':');
 710        char *ep;
 711
 712        ctx->userFunc(ctx, 1);
 713
 714        if (c == NULL)
 715                c = name;
 716        else
 717                c++;
 718
 719        ep = ctx->name + strlen(ctx->name) - strlen(c) - 1;
 720        *ep = 0;
 721}
 722
 723static void
 724xml_cdata(void *userData, const XML_Char *s, int len)
 725{
 726        struct xml_ctx *ctx = (struct xml_ctx *)userData;
 727        if (ctx->cdata)
 728                free(ctx->cdata);
 729        ctx->cdata = xmalloc(len + 1);
 730        strlcpy(ctx->cdata, s, len + 1);
 731}
 732
 733static int remote_ls(struct alt_base *repo, const char *path, int flags,
 734                     void (*userFunc)(struct remote_ls_ctx *ls),
 735                     void *userData);
 736
 737static void handle_remote_ls_ctx(struct xml_ctx *ctx, int tag_closed)
 738{
 739        struct remote_ls_ctx *ls = (struct remote_ls_ctx *)ctx->userData;
 740
 741        if (tag_closed) {
 742                if (!strcmp(ctx->name, DAV_PROPFIND_RESP) && ls->dentry_name) {
 743                        if (ls->dentry_flags & IS_DIR) {
 744                                if (ls->flags & PROCESS_DIRS) {
 745                                        ls->userFunc(ls);
 746                                }
 747                                if (strcmp(ls->dentry_name, ls->path) &&
 748                                    ls->flags & RECURSIVE) {
 749                                        ls->rc = remote_ls(ls->repo,
 750                                                           ls->dentry_name,
 751                                                           ls->flags,
 752                                                           ls->userFunc,
 753                                                           ls->userData);
 754                                }
 755                        } else if (ls->flags & PROCESS_FILES) {
 756                                ls->userFunc(ls);
 757                        }
 758                } else if (!strcmp(ctx->name, DAV_PROPFIND_NAME) && ctx->cdata) {
 759                        ls->dentry_name = xmalloc(strlen(ctx->cdata) -
 760                                                  ls->repo->path_len + 1);
 761                        strcpy(ls->dentry_name, ctx->cdata + ls->repo->path_len);
 762                } else if (!strcmp(ctx->name, DAV_PROPFIND_COLLECTION)) {
 763                        ls->dentry_flags |= IS_DIR;
 764                }
 765        } else if (!strcmp(ctx->name, DAV_PROPFIND_RESP)) {
 766                if (ls->dentry_name) {
 767                        free(ls->dentry_name);
 768                }
 769                ls->dentry_name = NULL;
 770                ls->dentry_flags = 0;
 771        }
 772}
 773
 774static int remote_ls(struct alt_base *repo, const char *path, int flags,
 775                     void (*userFunc)(struct remote_ls_ctx *ls),
 776                     void *userData)
 777{
 778        char *url = xmalloc(strlen(repo->base) + strlen(path) + 1);
 779        struct active_request_slot *slot;
 780        struct slot_results results;
 781        struct buffer in_buffer;
 782        struct buffer out_buffer;
 783        char *in_data;
 784        char *out_data;
 785        XML_Parser parser = XML_ParserCreate(NULL);
 786        enum XML_Status result;
 787        struct curl_slist *dav_headers = NULL;
 788        struct xml_ctx ctx;
 789        struct remote_ls_ctx ls;
 790
 791        ls.flags = flags;
 792        ls.repo = repo;
 793        ls.path = strdup(path);
 794        ls.dentry_name = NULL;
 795        ls.dentry_flags = 0;
 796        ls.userData = userData;
 797        ls.userFunc = userFunc;
 798        ls.rc = 0;
 799
 800        sprintf(url, "%s%s", repo->base, path);
 801
 802        out_buffer.size = strlen(PROPFIND_ALL_REQUEST);
 803        out_data = xmalloc(out_buffer.size + 1);
 804        snprintf(out_data, out_buffer.size + 1, PROPFIND_ALL_REQUEST);
 805        out_buffer.posn = 0;
 806        out_buffer.buffer = out_data;
 807
 808        in_buffer.size = 4096;
 809        in_data = xmalloc(in_buffer.size);
 810        in_buffer.posn = 0;
 811        in_buffer.buffer = in_data;
 812
 813        dav_headers = curl_slist_append(dav_headers, "Depth: 1");
 814        dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml");
 815
 816        slot = get_active_slot();
 817        slot->results = &results;
 818        curl_easy_setopt(slot->curl, CURLOPT_INFILE, &out_buffer);
 819        curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, out_buffer.size);
 820        curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer);
 821        curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer);
 822        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 823        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 824        curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1);
 825        curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PROPFIND);
 826        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);
 827
 828        if (start_active_slot(slot)) {
 829                run_active_slot(slot);
 830                if (results.curl_result == CURLE_OK) {
 831                        ctx.name = xcalloc(10, 1);
 832                        ctx.len = 0;
 833                        ctx.cdata = NULL;
 834                        ctx.userFunc = handle_remote_ls_ctx;
 835                        ctx.userData = &ls;
 836                        XML_SetUserData(parser, &ctx);
 837                        XML_SetElementHandler(parser, xml_start_tag,
 838                                              xml_end_tag);
 839                        XML_SetCharacterDataHandler(parser, xml_cdata);
 840                        result = XML_Parse(parser, in_buffer.buffer,
 841                                           in_buffer.posn, 1);
 842                        free(ctx.name);
 843
 844                        if (result != XML_STATUS_OK) {
 845                                ls.rc = error("XML error: %s",
 846                                              XML_ErrorString(
 847                                                      XML_GetErrorCode(parser)));
 848                        }
 849                } else {
 850                        ls.rc = -1;
 851                }
 852        } else {
 853                ls.rc = error("Unable to start PROPFIND request");
 854        }
 855
 856        free(ls.path);
 857        free(url);
 858        free(out_data);
 859        free(in_buffer.buffer);
 860        curl_slist_free_all(dav_headers);
 861
 862        return ls.rc;
 863}
 864
 865static void process_ls_pack(struct remote_ls_ctx *ls)
 866{
 867        unsigned char sha1[20];
 868
 869        if (strlen(ls->dentry_name) == 63 &&
 870            !strncmp(ls->dentry_name, "objects/pack/pack-", 18) &&
 871            !strncmp(ls->dentry_name+58, ".pack", 5)) {
 872                get_sha1_hex(ls->dentry_name + 18, sha1);
 873                setup_index(ls->repo, sha1);
 874        }
 875}
 876#endif
 877
 878static int fetch_indices(struct alt_base *repo)
 879{
 880        unsigned char sha1[20];
 881        char *url;
 882        struct buffer buffer;
 883        char *data;
 884        int i = 0;
 885
 886        struct active_request_slot *slot;
 887        struct slot_results results;
 888
 889        if (repo->got_indices)
 890                return 0;
 891
 892        data = xmalloc(4096);
 893        buffer.size = 4096;
 894        buffer.posn = 0;
 895        buffer.buffer = data;
 896
 897        if (get_verbosely)
 898                fprintf(stderr, "Getting pack list for %s\n", repo->base);
 899
 900#ifndef NO_EXPAT
 901        if (remote_ls(repo, "objects/pack/", PROCESS_FILES,
 902                      process_ls_pack, NULL) == 0)
 903                return 0;
 904#endif
 905
 906        url = xmalloc(strlen(repo->base) + 21);
 907        sprintf(url, "%s/objects/info/packs", repo->base);
 908
 909        slot = get_active_slot();
 910        slot->results = &results;
 911        curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
 912        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 913        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 914        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
 915        if (start_active_slot(slot)) {
 916                run_active_slot(slot);
 917                if (results.curl_result != CURLE_OK) {
 918                        if (results.http_code == 404 ||
 919                            results.curl_result == CURLE_FILE_COULDNT_READ_FILE) {
 920                                repo->got_indices = 1;
 921                                free(buffer.buffer);
 922                                return 0;
 923                        } else {
 924                                repo->got_indices = 0;
 925                                free(buffer.buffer);
 926                                return error("%s", curl_errorstr);
 927                        }
 928                }
 929        } else {
 930                repo->got_indices = 0;
 931                free(buffer.buffer);
 932                return error("Unable to start request");
 933        }
 934
 935        data = buffer.buffer;
 936        while (i < buffer.posn) {
 937                switch (data[i]) {
 938                case 'P':
 939                        i++;
 940                        if (i + 52 <= buffer.posn &&
 941                            !strncmp(data + i, " pack-", 6) &&
 942                            !strncmp(data + i + 46, ".pack\n", 6)) {
 943                                get_sha1_hex(data + i + 6, sha1);
 944                                setup_index(repo, sha1);
 945                                i += 51;
 946                                break;
 947                        }
 948                default:
 949                        while (i < buffer.posn && data[i] != '\n')
 950                                i++;
 951                }
 952                i++;
 953        }
 954
 955        free(buffer.buffer);
 956        repo->got_indices = 1;
 957        return 0;
 958}
 959
 960static int fetch_pack(struct alt_base *repo, unsigned char *sha1)
 961{
 962        char *url;
 963        struct packed_git *target;
 964        struct packed_git **lst;
 965        FILE *packfile;
 966        char *filename;
 967        char tmpfile[PATH_MAX];
 968        int ret;
 969        long prev_posn = 0;
 970        char range[RANGE_HEADER_SIZE];
 971        struct curl_slist *range_header = NULL;
 972
 973        struct active_request_slot *slot;
 974        struct slot_results results;
 975
 976        if (fetch_indices(repo))
 977                return -1;
 978        target = find_sha1_pack(sha1, repo->packs);
 979        if (!target)
 980                return -1;
 981
 982        if (get_verbosely) {
 983                fprintf(stderr, "Getting pack %s\n",
 984                        sha1_to_hex(target->sha1));
 985                fprintf(stderr, " which contains %s\n",
 986                        sha1_to_hex(sha1));
 987        }
 988
 989        url = xmalloc(strlen(repo->base) + 65);
 990        sprintf(url, "%s/objects/pack/pack-%s.pack",
 991                repo->base, sha1_to_hex(target->sha1));
 992
 993        filename = sha1_pack_name(target->sha1);
 994        snprintf(tmpfile, sizeof(tmpfile), "%s.temp", filename);
 995        packfile = fopen(tmpfile, "a");
 996        if (!packfile)
 997                return error("Unable to open local file %s for pack",
 998                             filename);
 999
1000        slot = get_active_slot();
1001        slot->results = &results;
1002        curl_easy_setopt(slot->curl, CURLOPT_FILE, packfile);
1003        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite);
1004        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
1005        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header);
1006        slot->local = packfile;
1007
1008        /* If there is data present from a previous transfer attempt,
1009           resume where it left off */
1010        prev_posn = ftell(packfile);
1011        if (prev_posn>0) {
1012                if (get_verbosely)
1013                        fprintf(stderr,
1014                                "Resuming fetch of pack %s at byte %ld\n",
1015                                sha1_to_hex(target->sha1), prev_posn);
1016                sprintf(range, "Range: bytes=%ld-", prev_posn);
1017                range_header = curl_slist_append(range_header, range);
1018                curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, range_header);
1019        }
1020
1021        if (start_active_slot(slot)) {
1022                run_active_slot(slot);
1023                if (results.curl_result != CURLE_OK) {
1024                        fclose(packfile);
1025                        return error("Unable to get pack file %s\n%s", url,
1026                                     curl_errorstr);
1027                }
1028        } else {
1029                fclose(packfile);
1030                return error("Unable to start request");
1031        }
1032
1033        fclose(packfile);
1034
1035        ret = move_temp_to_file(tmpfile, filename);
1036        if (ret)
1037                return ret;
1038
1039        lst = &repo->packs;
1040        while (*lst != target)
1041                lst = &((*lst)->next);
1042        *lst = (*lst)->next;
1043
1044        if (verify_pack(target, 0))
1045                return -1;
1046        install_packed_git(target);
1047
1048        return 0;
1049}
1050
1051static void abort_object_request(struct object_request *obj_req)
1052{
1053        if (obj_req->local >= 0) {
1054                close(obj_req->local);
1055                obj_req->local = -1;
1056        }
1057        unlink(obj_req->tmpfile);
1058        if (obj_req->slot) {
1059                release_active_slot(obj_req->slot);
1060                obj_req->slot = NULL;
1061        }
1062        release_object_request(obj_req);
1063}
1064
1065static int fetch_object(struct alt_base *repo, unsigned char *sha1)
1066{
1067        char *hex = sha1_to_hex(sha1);
1068        int ret = 0;
1069        struct object_request *obj_req = object_queue_head;
1070
1071        while (obj_req != NULL && memcmp(obj_req->sha1, sha1, 20))
1072                obj_req = obj_req->next;
1073        if (obj_req == NULL)
1074                return error("Couldn't find request for %s in the queue", hex);
1075
1076        if (has_sha1_file(obj_req->sha1)) {
1077                abort_object_request(obj_req);
1078                return 0;
1079        }
1080
1081#ifdef USE_CURL_MULTI
1082        while (obj_req->state == WAITING) {
1083                step_active_slots();
1084        }
1085#else
1086        start_object_request(obj_req);
1087#endif
1088
1089        while (obj_req->state == ACTIVE) {
1090                run_active_slot(obj_req->slot);
1091        }
1092        if (obj_req->local != -1) {
1093                close(obj_req->local); obj_req->local = -1;
1094        }
1095
1096        if (obj_req->state == ABORTED) {
1097                ret = error("Request for %s aborted", hex);
1098        } else if (obj_req->curl_result != CURLE_OK &&
1099                   obj_req->http_code != 416) {
1100                if (obj_req->http_code == 404 ||
1101                    obj_req->curl_result == CURLE_FILE_COULDNT_READ_FILE)
1102                        ret = -1; /* Be silent, it is probably in a pack. */
1103                else
1104                        ret = error("%s (curl_result = %d, http_code = %ld, sha1 = %s)",
1105                                    obj_req->errorstr, obj_req->curl_result,
1106                                    obj_req->http_code, hex);
1107        } else if (obj_req->zret != Z_STREAM_END) {
1108                corrupt_object_found++;
1109                ret = error("File %s (%s) corrupt", hex, obj_req->url);
1110        } else if (memcmp(obj_req->sha1, obj_req->real_sha1, 20)) {
1111                ret = error("File %s has bad hash", hex);
1112        } else if (obj_req->rename < 0) {
1113                ret = error("unable to write sha1 filename %s",
1114                            obj_req->filename);
1115        }
1116
1117        release_object_request(obj_req);
1118        return ret;
1119}
1120
1121int fetch(unsigned char *sha1)
1122{
1123        struct alt_base *altbase = alt;
1124
1125        if (!fetch_object(altbase, sha1))
1126                return 0;
1127        while (altbase) {
1128                if (!fetch_pack(altbase, sha1))
1129                        return 0;
1130                fetch_alternates(alt->base);
1131                altbase = altbase->next;
1132        }
1133        return error("Unable to find %s under %s", sha1_to_hex(sha1),
1134                     alt->base);
1135}
1136
1137static inline int needs_quote(int ch)
1138{
1139        if (((ch >= 'A') && (ch <= 'Z'))
1140                        || ((ch >= 'a') && (ch <= 'z'))
1141                        || ((ch >= '0') && (ch <= '9'))
1142                        || (ch == '/')
1143                        || (ch == '-')
1144                        || (ch == '.'))
1145                return 0;
1146        return 1;
1147}
1148
1149static inline int hex(int v)
1150{
1151        if (v < 10) return '0' + v;
1152        else return 'A' + v - 10;
1153}
1154
1155static char *quote_ref_url(const char *base, const char *ref)
1156{
1157        const char *cp;
1158        char *dp, *qref;
1159        int len, baselen, ch;
1160
1161        baselen = strlen(base);
1162        len = baselen + 6; /* "refs/" + NUL */
1163        for (cp = ref; (ch = *cp) != 0; cp++, len++)
1164                if (needs_quote(ch))
1165                        len += 2; /* extra two hex plus replacement % */
1166        qref = xmalloc(len);
1167        memcpy(qref, base, baselen);
1168        memcpy(qref + baselen, "refs/", 5);
1169        for (cp = ref, dp = qref + baselen + 5; (ch = *cp) != 0; cp++) {
1170                if (needs_quote(ch)) {
1171                        *dp++ = '%';
1172                        *dp++ = hex((ch >> 4) & 0xF);
1173                        *dp++ = hex(ch & 0xF);
1174                }
1175                else
1176                        *dp++ = ch;
1177        }
1178        *dp = 0;
1179
1180        return qref;
1181}
1182
1183int fetch_ref(char *ref, unsigned char *sha1)
1184{
1185        char *url;
1186        char hex[42];
1187        struct buffer buffer;
1188        char *base = alt->base;
1189        struct active_request_slot *slot;
1190        struct slot_results results;
1191        buffer.size = 41;
1192        buffer.posn = 0;
1193        buffer.buffer = hex;
1194        hex[41] = '\0';
1195
1196        url = quote_ref_url(base, ref);
1197        slot = get_active_slot();
1198        slot->results = &results;
1199        curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
1200        curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
1201        curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
1202        curl_easy_setopt(slot->curl, CURLOPT_URL, url);
1203        if (start_active_slot(slot)) {
1204                run_active_slot(slot);
1205                if (results.curl_result != CURLE_OK)
1206                        return error("Couldn't get %s for %s\n%s",
1207                                     url, ref, curl_errorstr);
1208        } else {
1209                return error("Unable to start request");
1210        }
1211
1212        hex[40] = '\0';
1213        get_sha1_hex(hex, sha1);
1214        return 0;
1215}
1216
1217int main(int argc, char **argv)
1218{
1219        char *commit_id;
1220        char *url;
1221        char *path;
1222        int arg = 1;
1223        int rc = 0;
1224
1225        setup_git_directory();
1226        git_config(git_default_config);
1227
1228        while (arg < argc && argv[arg][0] == '-') {
1229                if (argv[arg][1] == 't') {
1230                        get_tree = 1;
1231                } else if (argv[arg][1] == 'c') {
1232                        get_history = 1;
1233                } else if (argv[arg][1] == 'a') {
1234                        get_all = 1;
1235                        get_tree = 1;
1236                        get_history = 1;
1237                } else if (argv[arg][1] == 'v') {
1238                        get_verbosely = 1;
1239                } else if (argv[arg][1] == 'w') {
1240                        write_ref = argv[arg + 1];
1241                        arg++;
1242                } else if (!strcmp(argv[arg], "--recover")) {
1243                        get_recover = 1;
1244                }
1245                arg++;
1246        }
1247        if (argc < arg + 2) {
1248                usage("git-http-fetch [-c] [-t] [-a] [-d] [-v] [--recover] [-w ref] commit-id url");
1249                return 1;
1250        }
1251        commit_id = argv[arg];
1252        url = argv[arg + 1];
1253        write_ref_log_details = url;
1254
1255        http_init();
1256
1257        no_pragma_header = curl_slist_append(no_pragma_header, "Pragma:");
1258
1259        alt = xmalloc(sizeof(*alt));
1260        alt->base = url;
1261        alt->got_indices = 0;
1262        alt->packs = NULL;
1263        alt->next = NULL;
1264        path = strstr(url, "//");
1265        if (path) {
1266                path = strchr(path+2, '/');
1267                if (path)
1268                        alt->path_len = strlen(path);
1269        }
1270
1271        if (pull(commit_id))
1272                rc = 1;
1273
1274        http_cleanup();
1275
1276        curl_slist_free_all(no_pragma_header);
1277
1278        if (corrupt_object_found) {
1279                fprintf(stderr,
1280"Some loose object were found to be corrupt, but they might be just\n"
1281"a false '404 Not Found' error message sent with incorrect HTTP\n"
1282"status code.  Suggest running git fsck-objects.\n");
1283        }
1284        return rc;
1285}