transport.c

   1 #include "cache.h"
   2 #include "transport.h"
   3 #include "run-command.h"
   4 #ifndef NO_CURL
   5 #include "http.h"
   6 #endif
   7 #include "pkt-line.h"
   8 #include "fetch-pack.h"
   9 #include "walker.h"
  10 #include "bundle.h"
  11 #include "dir.h"
  12 #include "refs.h"
  13
  14 /* rsync support */
  15
  16 /*
  17  * We copy packed-refs and refs/ into a temporary file, then read the
  18  * loose refs recursively (sorting whenever possible), and then inserting
  19  * those packed refs that are not yet in the list (not validating, but
  20  * assuming that the file is sorted).
  21  *
  22  * Appears refactoring this from refs.c is too cumbersome.
  23  */
  24
  25 static int str_cmp(const void *a, const void *b)
  26 {
  27         const char *s1 = a;
  28         const char *s2 = b;
  29
  30         return strcmp(s1, s2);
  31 }
  32
  33 /* path->buf + name_offset is expected to point to "refs/" */
  34
  35 static int read_loose_refs(struct strbuf *path, int name_offset,
  36                 struct ref **tail)
  37 {
  38         DIR *dir = opendir(path->buf);
  39         struct dirent *de;
  40         struct {
  41                 char **entries;
  42                 int nr, alloc;
  43         } list;
  44         int i, pathlen;
  45
  46         if (!dir)
  47                 return -1;
  48
  49         memset (&list, 0, sizeof(list));
  50
  51         while ((de = readdir(dir))) {
  52                 if (de->d_name[0] == '.' && (de->d_name[1] == '\0' ||
  53                                 (de->d_name[1] == '.' &&
  54                                  de->d_name[2] == '\0')))
  55                         continue;
  56                 ALLOC_GROW(list.entries, list.nr + 1, list.alloc);
  57                 list.entries[list.nr++] = xstrdup(de->d_name);
  58         }
  59         closedir(dir);
  60
  61         /* sort the list */
  62
  63         qsort(list.entries, list.nr, sizeof(char *), str_cmp);
  64
  65         pathlen = path->len;
  66         strbuf_addch(path, '/');
  67
  68         for (i = 0; i < list.nr; i++, strbuf_setlen(path, pathlen + 1)) {
  69                 strbuf_addstr(path, list.entries[i]);
  70                 if (read_loose_refs(path, name_offset, tail)) {
  71                         int fd = open(path->buf, O_RDONLY);
  72                         char buffer[40];
  73                         struct ref *next;
  74
  75                         if (fd < 0)
  76                                 continue;
  77                         next = alloc_ref(path->len - name_offset + 1);
  78                         if (read_in_full(fd, buffer, 40) != 40 ||
  79                                         get_sha1_hex(buffer, next->old_sha1)) {
  80                                 close(fd);
  81                                 free(next);
  82                                 continue;
  83                         }
  84                         close(fd);
  85                         strcpy(next->name, path->buf + name_offset);
  86                         (*tail)->next = next;
  87                         *tail = next;
  88                 }
  89         }
  90         strbuf_setlen(path, pathlen);
  91
  92         for (i = 0; i < list.nr; i++)
  93                 free(list.entries[i]);
  94         free(list.entries);
  95
  96         return 0;
  97 }
  98
  99 /* insert the packed refs for which no loose refs were found */
 100
 101 static void insert_packed_refs(const char *packed_refs, struct ref **list)
 102 {
 103         FILE *f = fopen(packed_refs, "r");
 104         static char buffer[PATH_MAX];
 105
 106         if (!f)
 107                 return;
 108
 109         for (;;) {
 110                 int cmp, len;
 111
 112                 if (!fgets(buffer, sizeof(buffer), f)) {
 113                         fclose(f);
 114                         return;
 115                 }
 116
 117                 if (hexval(buffer[0]) > 0xf)
 118                         continue;
 119                 len = strlen(buffer);
 120                 if (buffer[len - 1] == '\n')
 121                         buffer[--len] = '\0';
 122                 if (len < 41)
 123                         continue;
 124                 while ((*list)->next &&
 125                                 (cmp = strcmp(buffer + 41,
 126                                       (*list)->next->name)) > 0)
 127                         list = &(*list)->next;
 128                 if (!(*list)->next || cmp < 0) {
 129                         struct ref *next = alloc_ref(len - 40);
 130                         buffer[40] = '\0';
 131                         if (get_sha1_hex(buffer, next->old_sha1)) {
 132                                 warning ("invalid SHA-1: %s", buffer);
 133                                 free(next);
 134                                 continue;
 135                         }
 136                         strcpy(next->name, buffer + 41);
 137                         next->next = (*list)->next;
 138                         (*list)->next = next;
 139                         list = &(*list)->next;
 140                 }
 141         }
 142 }
 143
 144 static struct ref *get_refs_via_rsync(const struct transport *transport)
 145 {
 146         struct strbuf buf = STRBUF_INIT, temp_dir = STRBUF_INIT;
 147         struct ref dummy, *tail = &dummy;
 148         struct child_process rsync;
 149         const char *args[5];
 150         int temp_dir_len;
 151
 152         /* copy the refs to the temporary directory */
 153
 154         strbuf_addstr(&temp_dir, git_path("rsync-refs-XXXXXX"));
 155         if (!mkdtemp(temp_dir.buf))
 156                 die ("Could not make temporary directory");
 157         temp_dir_len = temp_dir.len;
 158
 159         strbuf_addstr(&buf, transport->url);
 160         strbuf_addstr(&buf, "/refs");
 161
 162         memset(&rsync, 0, sizeof(rsync));
 163         rsync.argv = args;
 164         rsync.stdout_to_stderr = 1;
 165         args[0] = "rsync";
 166         args[1] = (transport->verbose > 0) ? "-rv" : "-r";
 167         args[2] = buf.buf;
 168         args[3] = temp_dir.buf;
 169         args[4] = NULL;
 170
 171         if (run_command(&rsync))
 172                 die ("Could not run rsync to get refs");
 173
 174         strbuf_reset(&buf);
 175         strbuf_addstr(&buf, transport->url);
 176         strbuf_addstr(&buf, "/packed-refs");
 177
 178         args[2] = buf.buf;
 179
 180         if (run_command(&rsync))
 181                 die ("Could not run rsync to get refs");
 182
 183         /* read the copied refs */
 184
 185         strbuf_addstr(&temp_dir, "/refs");
 186         read_loose_refs(&temp_dir, temp_dir_len + 1, &tail);
 187         strbuf_setlen(&temp_dir, temp_dir_len);
 188
 189         tail = &dummy;
 190         strbuf_addstr(&temp_dir, "/packed-refs");
 191         insert_packed_refs(temp_dir.buf, &tail);
 192         strbuf_setlen(&temp_dir, temp_dir_len);
 193
 194         if (remove_dir_recursively(&temp_dir, 0))
 195                 warning ("Error removing temporary directory %s.",
 196                                 temp_dir.buf);
 197
 198         strbuf_release(&buf);
 199         strbuf_release(&temp_dir);
 200
 201         return dummy.next;
 202 }
 203
 204 static int fetch_objs_via_rsync(struct transport *transport,
 205                                  int nr_objs, struct ref **to_fetch)
 206 {
 207         struct strbuf buf = STRBUF_INIT;
 208         struct child_process rsync;
 209         const char *args[8];
 210         int result;
 211
 212         strbuf_addstr(&buf, transport->url);
 213         strbuf_addstr(&buf, "/objects/");
 214
 215         memset(&rsync, 0, sizeof(rsync));
 216         rsync.argv = args;
 217         rsync.stdout_to_stderr = 1;
 218         args[0] = "rsync";
 219         args[1] = (transport->verbose > 0) ? "-rv" : "-r";
 220         args[2] = "--ignore-existing";
 221         args[3] = "--exclude";
 222         args[4] = "info";
 223         args[5] = buf.buf;
 224         args[6] = get_object_directory();
 225         args[7] = NULL;
 226
 227         /* NEEDSWORK: handle one level of alternates */
 228         result = run_command(&rsync);
 229
 230         strbuf_release(&buf);
 231
 232         return result;
 233 }
 234
 235 static int write_one_ref(const char *name, const unsigned char *sha1,
 236                 int flags, void *data)
 237 {
 238         struct strbuf *buf = data;
 239         int len = buf->len;
 240         FILE *f;
 241
 242         /* when called via for_each_ref(), flags is non-zero */
 243         if (flags && prefixcmp(name, "refs/heads/") &&
 244                         prefixcmp(name, "refs/tags/"))
 245                 return 0;
 246
 247         strbuf_addstr(buf, name);
 248         if (safe_create_leading_directories(buf->buf) ||
 249                         !(f = fopen(buf->buf, "w")) ||
 250                         fprintf(f, "%s\n", sha1_to_hex(sha1)) < 0 ||
 251                         fclose(f))
 252                 return error("problems writing temporary file %s", buf->buf);
 253         strbuf_setlen(buf, len);
 254         return 0;
 255 }
 256
 257 static int write_refs_to_temp_dir(struct strbuf *temp_dir,
 258                 int refspec_nr, const char **refspec)
 259 {
 260         int i;
 261
 262         for (i = 0; i < refspec_nr; i++) {
 263                 unsigned char sha1[20];
 264                 char *ref;
 265
 266                 if (dwim_ref(refspec[i], strlen(refspec[i]), sha1, &ref) != 1)
 267                         return error("Could not get ref %s", refspec[i]);
 268
 269                 if (write_one_ref(ref, sha1, 0, temp_dir)) {
 270                         free(ref);
 271                         return -1;
 272                 }
 273                 free(ref);
 274         }
 275         return 0;
 276 }
 277
 278 static int rsync_transport_push(struct transport *transport,
 279                 int refspec_nr, const char **refspec, int flags)
 280 {
 281         struct strbuf buf = STRBUF_INIT, temp_dir = STRBUF_INIT;
 282         int result = 0, i;
 283         struct child_process rsync;
 284         const char *args[10];
 285
 286         /* first push the objects */
 287
 288         strbuf_addstr(&buf, transport->url);
 289         strbuf_addch(&buf, '/');
 290
 291         memset(&rsync, 0, sizeof(rsync));
 292         rsync.argv = args;
 293         rsync.stdout_to_stderr = 1;
 294         i = 0;
 295         args[i++] = "rsync";
 296         args[i++] = "-a";
 297         if (flags & TRANSPORT_PUSH_DRY_RUN)
 298                 args[i++] = "--dry-run";
 299         if (transport->verbose > 0)
 300                 args[i++] = "-v";
 301         args[i++] = "--ignore-existing";
 302         args[i++] = "--exclude";
 303         args[i++] = "info";
 304         args[i++] = get_object_directory();
 305         args[i++] = buf.buf;
 306         args[i++] = NULL;
 307
 308         if (run_command(&rsync))
 309                 return error("Could not push objects to %s", transport->url);
 310
 311         /* copy the refs to the temporary directory; they could be packed. */
 312
 313         strbuf_addstr(&temp_dir, git_path("rsync-refs-XXXXXX"));
 314         if (!mkdtemp(temp_dir.buf))
 315                 die ("Could not make temporary directory");
 316         strbuf_addch(&temp_dir, '/');
 317
 318         if (flags & TRANSPORT_PUSH_ALL) {
 319                 if (for_each_ref(write_one_ref, &temp_dir))
 320                         return -1;
 321         } else if (write_refs_to_temp_dir(&temp_dir, refspec_nr, refspec))
 322                 return -1;
 323
 324         i = 2;
 325         if (flags & TRANSPORT_PUSH_DRY_RUN)
 326                 args[i++] = "--dry-run";
 327         if (!(flags & TRANSPORT_PUSH_FORCE))
 328                 args[i++] = "--ignore-existing";
 329         args[i++] = temp_dir.buf;
 330         args[i++] = transport->url;
 331         args[i++] = NULL;
 332         if (run_command(&rsync))
 333                 result = error("Could not push to %s", transport->url);
 334
 335         if (remove_dir_recursively(&temp_dir, 0))
 336                 warning ("Could not remove temporary directory %s.",
 337                                 temp_dir.buf);
 338
 339         strbuf_release(&buf);
 340         strbuf_release(&temp_dir);
 341
 342         return result;
 343 }
 344
 345 /* Generic functions for using commit walkers */
 346
 347 static int fetch_objs_via_walker(struct transport *transport,
 348                                  int nr_objs, struct ref **to_fetch)
 349 {
 350         char *dest = xstrdup(transport->url);
 351         struct walker *walker = transport->data;
 352         char **objs = xmalloc(nr_objs * sizeof(*objs));
 353         int i;
 354
 355         walker->get_all = 1;
 356         walker->get_tree = 1;
 357         walker->get_history = 1;
 358         walker->get_verbosely = transport->verbose >= 0;
 359         walker->get_recover = 0;
 360
 361         for (i = 0; i < nr_objs; i++)
 362                 objs[i] = xstrdup(sha1_to_hex(to_fetch[i]->old_sha1));
 363
 364         if (walker_fetch(walker, nr_objs, objs, NULL, NULL))
 365                 die("Fetch failed.");
 366
 367         for (i = 0; i < nr_objs; i++)
 368                 free(objs[i]);
 369         free(objs);
 370         free(dest);
 371         return 0;
 372 }
 373
 374 static int disconnect_walker(struct transport *transport)
 375 {
 376         struct walker *walker = transport->data;
 377         if (walker)
 378                 walker_free(walker);
 379         return 0;
 380 }
 381
 382 #ifndef NO_CURL
 383 static int curl_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
 384         const char **argv;
 385         int argc;
 386         int err;
 387
 388         argv = xmalloc((refspec_nr + 11) * sizeof(char *));
 389         argv[0] = "http-push";
 390         argc = 1;
 391         if (flags & TRANSPORT_PUSH_ALL)
 392                 argv[argc++] = "--all";
 393         if (flags & TRANSPORT_PUSH_FORCE)
 394                 argv[argc++] = "--force";
 395         if (flags & TRANSPORT_PUSH_DRY_RUN)
 396                 argv[argc++] = "--dry-run";
 397         argv[argc++] = transport->url;
 398         while (refspec_nr--)
 399                 argv[argc++] = *refspec++;
 400         argv[argc] = NULL;
 401         err = run_command_v_opt(argv, RUN_GIT_CMD);
 402         switch (err) {
 403         case -ERR_RUN_COMMAND_FORK:
 404                 error("unable to fork for %s", argv[0]);
 405         case -ERR_RUN_COMMAND_EXEC:
 406                 error("unable to exec %s", argv[0]);
 407                 break;
 408         case -ERR_RUN_COMMAND_WAITPID:
 409         case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
 410         case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
 411         case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
 412                 error("%s died with strange error", argv[0]);
 413         }
 414         return !!err;
 415 }
 416
 417 static int missing__target(int code, int result)
 418 {
 419         return  /* file:// URL -- do we ever use one??? */
 420                 (result == CURLE_FILE_COULDNT_READ_FILE) ||
 421                 /* http:// and https:// URL */
 422                 (code == 404 && result == CURLE_HTTP_RETURNED_ERROR) ||
 423                 /* ftp:// URL */
 424                 (code == 550 && result == CURLE_FTP_COULDNT_RETR_FILE)
 425                 ;
 426 }
 427
 428 #define missing_target(a) missing__target((a)->http_code, (a)->curl_result)
 429
 430 static struct ref *get_refs_via_curl(const struct transport *transport)
 431 {
 432         struct buffer buffer;
 433         char *data, *start, *mid;
 434         char *ref_name;
 435         char *refs_url;
 436         int i = 0;
 437
 438         struct active_request_slot *slot;
 439         struct slot_results results;
 440
 441         struct ref *refs = NULL;
 442         struct ref *ref = NULL;
 443         struct ref *last_ref = NULL;
 444
 445         data = xmalloc(4096);
 446         buffer.size = 4096;
 447         buffer.posn = 0;
 448         buffer.buffer = data;
 449
 450         refs_url = xmalloc(strlen(transport->url) + 11);
 451         sprintf(refs_url, "%s/info/refs", transport->url);
 452
 453         http_init();
 454
 455         slot = get_active_slot();
 456         slot->results = &results;
 457         curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
 458         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 459         curl_easy_setopt(slot->curl, CURLOPT_URL, refs_url);
 460         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
 461         if (start_active_slot(slot)) {
 462                 run_active_slot(slot);
 463                 if (results.curl_result != CURLE_OK) {
 464                         if (missing_target(&results)) {
 465                                 free(buffer.buffer);
 466                                 return NULL;
 467                         } else {
 468                                 free(buffer.buffer);
 469                                 error("%s", curl_errorstr);
 470                                 return NULL;
 471                         }
 472                 }
 473         } else {
 474                 free(buffer.buffer);
 475                 error("Unable to start request");
 476                 return NULL;
 477         }
 478
 479         http_cleanup();
 480
 481         data = buffer.buffer;
 482         start = NULL;
 483         mid = data;
 484         while (i < buffer.posn) {
 485                 if (!start)
 486                         start = &data[i];
 487                 if (data[i] == '\t')
 488                         mid = &data[i];
 489                 if (data[i] == '\n') {
 490                         data[i] = 0;
 491                         ref_name = mid + 1;
 492                         ref = xmalloc(sizeof(struct ref) +
 493                                       strlen(ref_name) + 1);
 494                         memset(ref, 0, sizeof(struct ref));
 495                         strcpy(ref->name, ref_name);
 496                         get_sha1_hex(start, ref->old_sha1);
 497                         if (!refs)
 498                                 refs = ref;
 499                         if (last_ref)
 500                                 last_ref->next = ref;
 501                         last_ref = ref;
 502                         start = NULL;
 503                 }
 504                 i++;
 505         }
 506
 507         free(buffer.buffer);
 508
 509         return refs;
 510 }
 511
 512 static int fetch_objs_via_curl(struct transport *transport,
 513                                  int nr_objs, struct ref **to_fetch)
 514 {
 515         if (!transport->data)
 516                 transport->data = get_http_walker(transport->url);
 517         return fetch_objs_via_walker(transport, nr_objs, to_fetch);
 518 }
 519
 520 #endif
 521
 522 struct bundle_transport_data {
 523         int fd;
 524         struct bundle_header header;
 525 };
 526
 527 static struct ref *get_refs_from_bundle(const struct transport *transport)
 528 {
 529         struct bundle_transport_data *data = transport->data;
 530         struct ref *result = NULL;
 531         int i;
 532
 533         if (data->fd > 0)
 534                 close(data->fd);
 535         data->fd = read_bundle_header(transport->url, &data->header);
 536         if (data->fd < 0)
 537                 die ("Could not read bundle '%s'.", transport->url);
 538         for (i = 0; i < data->header.references.nr; i++) {
 539                 struct ref_list_entry *e = data->header.references.list + i;
 540                 struct ref *ref = alloc_ref(strlen(e->name) + 1);
 541                 hashcpy(ref->old_sha1, e->sha1);
 542                 strcpy(ref->name, e->name);
 543                 ref->next = result;
 544                 result = ref;
 545         }
 546         return result;
 547 }
 548
 549 static int fetch_refs_from_bundle(struct transport *transport,
 550                                int nr_heads, struct ref **to_fetch)
 551 {
 552         struct bundle_transport_data *data = transport->data;
 553         return unbundle(&data->header, data->fd);
 554 }
 555
 556 static int close_bundle(struct transport *transport)
 557 {
 558         struct bundle_transport_data *data = transport->data;
 559         if (data->fd > 0)
 560                 close(data->fd);
 561         free(data);
 562         return 0;
 563 }
 564
 565 struct git_transport_data {
 566         unsigned thin : 1;
 567         unsigned keep : 1;
 568         int depth;
 569         const char *uploadpack;
 570         const char *receivepack;
 571 };
 572
 573 static int set_git_option(struct transport *connection,
 574                           const char *name, const char *value)
 575 {
 576         struct git_transport_data *data = connection->data;
 577         if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
 578                 data->uploadpack = value;
 579                 return 0;
 580         } else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
 581                 data->receivepack = value;
 582                 return 0;
 583         } else if (!strcmp(name, TRANS_OPT_THIN)) {
 584                 data->thin = !!value;
 585                 return 0;
 586         } else if (!strcmp(name, TRANS_OPT_KEEP)) {
 587                 data->keep = !!value;
 588                 return 0;
 589         } else if (!strcmp(name, TRANS_OPT_DEPTH)) {
 590                 if (!value)
 591                         data->depth = 0;
 592                 else
 593                         data->depth = atoi(value);
 594                 return 0;
 595         }
 596         return 1;
 597 }
 598
 599 static struct ref *get_refs_via_connect(const struct transport *transport)
 600 {
 601         struct git_transport_data *data = transport->data;
 602         struct ref *refs;
 603         int fd[2];
 604         pid_t pid;
 605         char *dest = xstrdup(transport->url);
 606
 607         pid = git_connect(fd, dest, data->uploadpack, 0);
 608
 609         if (pid < 0)
 610                 die("Failed to connect to \"%s\"", transport->url);
 611
 612         get_remote_heads(fd[0], &refs, 0, NULL, 0);
 613         packet_flush(fd[1]);
 614
 615         finish_connect(pid);
 616
 617         free(dest);
 618
 619         return refs;
 620 }
 621
 622 static int fetch_refs_via_pack(struct transport *transport,
 623                                int nr_heads, struct ref **to_fetch)
 624 {
 625         struct git_transport_data *data = transport->data;
 626         char **heads = xmalloc(nr_heads * sizeof(*heads));
 627         char **origh = xmalloc(nr_heads * sizeof(*origh));
 628         struct ref *refs;
 629         char *dest = xstrdup(transport->url);
 630         struct fetch_pack_args args;
 631         int i;
 632
 633         memset(&args, 0, sizeof(args));
 634         args.uploadpack = data->uploadpack;
 635         args.keep_pack = data->keep;
 636         args.lock_pack = 1;
 637         args.use_thin_pack = data->thin;
 638         args.verbose = transport->verbose > 0;
 639         args.depth = data->depth;
 640
 641         for (i = 0; i < nr_heads; i++)
 642                 origh[i] = heads[i] = xstrdup(to_fetch[i]->name);
 643         refs = fetch_pack(&args, dest, nr_heads, heads, &transport->pack_lockfile);
 644
 645         for (i = 0; i < nr_heads; i++)
 646                 free(origh[i]);
 647         free(origh);
 648         free(heads);
 649         free_refs(refs);
 650         free(dest);
 651         return 0;
 652 }
 653
 654 static int git_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
 655         struct git_transport_data *data = transport->data;
 656         const char **argv;
 657         char *rem;
 658         int argc;
 659         int err;
 660
 661         argv = xmalloc((refspec_nr + 11) * sizeof(char *));
 662         argv[0] = "send-pack";
 663         argc = 1;
 664         if (flags & TRANSPORT_PUSH_ALL)
 665                 argv[argc++] = "--all";
 666         if (flags & TRANSPORT_PUSH_FORCE)
 667                 argv[argc++] = "--force";
 668         if (flags & TRANSPORT_PUSH_DRY_RUN)
 669                 argv[argc++] = "--dry-run";
 670         if (data->receivepack) {
 671                 char *rp = xmalloc(strlen(data->receivepack) + 16);
 672                 sprintf(rp, "--receive-pack=%s", data->receivepack);
 673                 argv[argc++] = rp;
 674         }
 675         if (data->thin)
 676                 argv[argc++] = "--thin";
 677         rem = xmalloc(strlen(transport->remote->name) + 10);
 678         sprintf(rem, "--remote=%s", transport->remote->name);
 679         argv[argc++] = rem;
 680         argv[argc++] = transport->url;
 681         while (refspec_nr--)
 682                 argv[argc++] = *refspec++;
 683         argv[argc] = NULL;
 684         err = run_command_v_opt(argv, RUN_GIT_CMD);
 685         switch (err) {
 686         case -ERR_RUN_COMMAND_FORK:
 687                 error("unable to fork for %s", argv[0]);
 688         case -ERR_RUN_COMMAND_EXEC:
 689                 error("unable to exec %s", argv[0]);
 690                 break;
 691         case -ERR_RUN_COMMAND_WAITPID:
 692         case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
 693         case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
 694         case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
 695                 error("%s died with strange error", argv[0]);
 696         }
 697         return !!err;
 698 }
 699
 700 static int disconnect_git(struct transport *transport)
 701 {
 702         free(transport->data);
 703         return 0;
 704 }
 705
 706 static int is_local(const char *url)
 707 {
 708         const char *colon = strchr(url, ':');
 709         const char *slash = strchr(url, '/');
 710         return !colon || (slash && slash < colon);
 711 }
 712
 713 static int is_file(const char *url)
 714 {
 715         struct stat buf;
 716         if (stat(url, &buf))
 717                 return 0;
 718         return S_ISREG(buf.st_mode);
 719 }
 720
 721 struct transport *transport_get(struct remote *remote, const char *url)
 722 {
 723         struct transport *ret = xcalloc(1, sizeof(*ret));
 724
 725         ret->remote = remote;
 726         ret->url = url;
 727
 728         if (!prefixcmp(url, "rsync://")) {
 729                 ret->get_refs_list = get_refs_via_rsync;
 730                 ret->fetch = fetch_objs_via_rsync;
 731                 ret->push = rsync_transport_push;
 732
 733         } else if (!prefixcmp(url, "http://")
 734                 || !prefixcmp(url, "https://")
 735                 || !prefixcmp(url, "ftp://")) {
 736 #ifdef NO_CURL
 737                 error("git was compiled without libcurl support.");
 738 #else
 739                 ret->get_refs_list = get_refs_via_curl;
 740                 ret->fetch = fetch_objs_via_curl;
 741                 ret->push = curl_transport_push;
 742 #endif
 743                 ret->disconnect = disconnect_walker;
 744
 745         } else if (is_local(url) && is_file(url)) {
 746                 struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
 747                 ret->data = data;
 748                 ret->get_refs_list = get_refs_from_bundle;
 749                 ret->fetch = fetch_refs_from_bundle;
 750                 ret->disconnect = close_bundle;
 751
 752         } else {
 753                 struct git_transport_data *data = xcalloc(1, sizeof(*data));
 754                 ret->data = data;
 755                 ret->set_option = set_git_option;
 756                 ret->get_refs_list = get_refs_via_connect;
 757                 ret->fetch = fetch_refs_via_pack;
 758                 ret->push = git_transport_push;
 759                 ret->disconnect = disconnect_git;
 760
 761                 data->thin = 1;
 762                 data->uploadpack = "git-upload-pack";
 763                 if (remote && remote->uploadpack)
 764                         data->uploadpack = remote->uploadpack;
 765                 data->receivepack = "git-receive-pack";
 766                 if (remote && remote->receivepack)
 767                         data->receivepack = remote->receivepack;
 768         }
 769
 770         return ret;
 771 }
 772
 773 int transport_set_option(struct transport *transport,
 774                          const char *name, const char *value)
 775 {
 776         if (transport->set_option)
 777                 return transport->set_option(transport, name, value);
 778         return 1;
 779 }
 780
 781 int transport_push(struct transport *transport,
 782                    int refspec_nr, const char **refspec, int flags)
 783 {
 784         if (!transport->push)
 785                 return 1;
 786         return transport->push(transport, refspec_nr, refspec, flags);
 787 }
 788
 789 struct ref *transport_get_remote_refs(struct transport *transport)
 790 {
 791         if (!transport->remote_refs)
 792                 transport->remote_refs = transport->get_refs_list(transport);
 793         return transport->remote_refs;
 794 }
 795
 796 int transport_fetch_refs(struct transport *transport, struct ref *refs)
 797 {
 798         int rc;
 799         int nr_heads = 0, nr_alloc = 0;
 800         struct ref **heads = NULL;
 801         struct ref *rm;
 802
 803         for (rm = refs; rm; rm = rm->next) {
 804                 if (rm->peer_ref &&
 805                     !hashcmp(rm->peer_ref->old_sha1, rm->old_sha1))
 806                         continue;
 807                 ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
 808                 heads[nr_heads++] = rm;
 809         }
 810
 811         rc = transport->fetch(transport, nr_heads, heads);
 812         free(heads);
 813         return rc;
 814 }
 815
 816 void transport_unlock_pack(struct transport *transport)
 817 {
 818         if (transport->pack_lockfile) {
 819                 unlink(transport->pack_lockfile);
 820                 free(transport->pack_lockfile);
 821                 transport->pack_lockfile = NULL;
 822         }
 823 }
 824
 825 int transport_disconnect(struct transport *transport)
 826 {
 827         int ret = 0;
 828         if (transport->disconnect)
 829                 ret = transport->disconnect(transport);
 830         free(transport);
 831         return ret;
 832 }