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[8];
 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         args[0] = "rsync";
 295         args[1] = (transport->verbose > 0) ? "-av" : "-a";
 296         args[2] = "--ignore-existing";
 297         args[3] = "--exclude";
 298         args[4] = "info";
 299         args[5] = get_object_directory();;
 300         args[6] = buf.buf;
 301         args[7] = NULL;
 302
 303         if (run_command(&rsync))
 304                 return error("Could not push objects to %s", transport->url);
 305
 306         /* copy the refs to the temporary directory; they could be packed. */
 307
 308         strbuf_addstr(&temp_dir, git_path("rsync-refs-XXXXXX"));
 309         if (!mkdtemp(temp_dir.buf))
 310                 die ("Could not make temporary directory");
 311         strbuf_addch(&temp_dir, '/');
 312
 313         if (flags & TRANSPORT_PUSH_ALL) {
 314                 if (for_each_ref(write_one_ref, &temp_dir))
 315                         return -1;
 316         } else if (write_refs_to_temp_dir(&temp_dir, refspec_nr, refspec))
 317                 return -1;
 318
 319         i = (flags & TRANSPORT_PUSH_FORCE) ? 2 : 3;
 320         args[i++] = temp_dir.buf;
 321         args[i++] = transport->url;
 322         args[i++] = NULL;
 323         if (run_command(&rsync))
 324                 result = error("Could not push to %s", transport->url);
 325
 326         if (remove_dir_recursively(&temp_dir, 0))
 327                 warning ("Could not remove temporary directory %s.",
 328                                 temp_dir.buf);
 329
 330         strbuf_release(&buf);
 331         strbuf_release(&temp_dir);
 332
 333         return result;
 334 }
 335
 336 /* Generic functions for using commit walkers */
 337
 338 static int fetch_objs_via_walker(struct transport *transport,
 339                                  int nr_objs, struct ref **to_fetch)
 340 {
 341         char *dest = xstrdup(transport->url);
 342         struct walker *walker = transport->data;
 343         char **objs = xmalloc(nr_objs * sizeof(*objs));
 344         int i;
 345
 346         walker->get_all = 1;
 347         walker->get_tree = 1;
 348         walker->get_history = 1;
 349         walker->get_verbosely = transport->verbose >= 0;
 350         walker->get_recover = 0;
 351
 352         for (i = 0; i < nr_objs; i++)
 353                 objs[i] = xstrdup(sha1_to_hex(to_fetch[i]->old_sha1));
 354
 355         if (walker_fetch(walker, nr_objs, objs, NULL, NULL))
 356                 die("Fetch failed.");
 357
 358         for (i = 0; i < nr_objs; i++)
 359                 free(objs[i]);
 360         free(objs);
 361         free(dest);
 362         return 0;
 363 }
 364
 365 static int disconnect_walker(struct transport *transport)
 366 {
 367         struct walker *walker = transport->data;
 368         if (walker)
 369                 walker_free(walker);
 370         return 0;
 371 }
 372
 373 #ifndef NO_CURL
 374 static int curl_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
 375         const char **argv;
 376         int argc;
 377         int err;
 378
 379         argv = xmalloc((refspec_nr + 11) * sizeof(char *));
 380         argv[0] = "http-push";
 381         argc = 1;
 382         if (flags & TRANSPORT_PUSH_ALL)
 383                 argv[argc++] = "--all";
 384         if (flags & TRANSPORT_PUSH_FORCE)
 385                 argv[argc++] = "--force";
 386         argv[argc++] = transport->url;
 387         while (refspec_nr--)
 388                 argv[argc++] = *refspec++;
 389         argv[argc] = NULL;
 390         err = run_command_v_opt(argv, RUN_GIT_CMD);
 391         switch (err) {
 392         case -ERR_RUN_COMMAND_FORK:
 393                 error("unable to fork for %s", argv[0]);
 394         case -ERR_RUN_COMMAND_EXEC:
 395                 error("unable to exec %s", argv[0]);
 396                 break;
 397         case -ERR_RUN_COMMAND_WAITPID:
 398         case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
 399         case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
 400         case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
 401                 error("%s died with strange error", argv[0]);
 402         }
 403         return !!err;
 404 }
 405
 406 static int missing__target(int code, int result)
 407 {
 408         return  /* file:// URL -- do we ever use one??? */
 409                 (result == CURLE_FILE_COULDNT_READ_FILE) ||
 410                 /* http:// and https:// URL */
 411                 (code == 404 && result == CURLE_HTTP_RETURNED_ERROR) ||
 412                 /* ftp:// URL */
 413                 (code == 550 && result == CURLE_FTP_COULDNT_RETR_FILE)
 414                 ;
 415 }
 416
 417 #define missing_target(a) missing__target((a)->http_code, (a)->curl_result)
 418
 419 static struct ref *get_refs_via_curl(const struct transport *transport)
 420 {
 421         struct buffer buffer;
 422         char *data, *start, *mid;
 423         char *ref_name;
 424         char *refs_url;
 425         int i = 0;
 426
 427         struct active_request_slot *slot;
 428         struct slot_results results;
 429
 430         struct ref *refs = NULL;
 431         struct ref *ref = NULL;
 432         struct ref *last_ref = NULL;
 433
 434         data = xmalloc(4096);
 435         buffer.size = 4096;
 436         buffer.posn = 0;
 437         buffer.buffer = data;
 438
 439         refs_url = xmalloc(strlen(transport->url) + 11);
 440         sprintf(refs_url, "%s/info/refs", transport->url);
 441
 442         http_init();
 443
 444         slot = get_active_slot();
 445         slot->results = &results;
 446         curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
 447         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 448         curl_easy_setopt(slot->curl, CURLOPT_URL, refs_url);
 449         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
 450         if (start_active_slot(slot)) {
 451                 run_active_slot(slot);
 452                 if (results.curl_result != CURLE_OK) {
 453                         if (missing_target(&results)) {
 454                                 free(buffer.buffer);
 455                                 return NULL;
 456                         } else {
 457                                 free(buffer.buffer);
 458                                 error("%s", curl_errorstr);
 459                                 return NULL;
 460                         }
 461                 }
 462         } else {
 463                 free(buffer.buffer);
 464                 error("Unable to start request");
 465                 return NULL;
 466         }
 467
 468         http_cleanup();
 469
 470         data = buffer.buffer;
 471         start = NULL;
 472         mid = data;
 473         while (i < buffer.posn) {
 474                 if (!start)
 475                         start = &data[i];
 476                 if (data[i] == '\t')
 477                         mid = &data[i];
 478                 if (data[i] == '\n') {
 479                         data[i] = 0;
 480                         ref_name = mid + 1;
 481                         ref = xmalloc(sizeof(struct ref) +
 482                                       strlen(ref_name) + 1);
 483                         memset(ref, 0, sizeof(struct ref));
 484                         strcpy(ref->name, ref_name);
 485                         get_sha1_hex(start, ref->old_sha1);
 486                         if (!refs)
 487                                 refs = ref;
 488                         if (last_ref)
 489                                 last_ref->next = ref;
 490                         last_ref = ref;
 491                         start = NULL;
 492                 }
 493                 i++;
 494         }
 495
 496         free(buffer.buffer);
 497
 498         return refs;
 499 }
 500
 501 static int fetch_objs_via_curl(struct transport *transport,
 502                                  int nr_objs, struct ref **to_fetch)
 503 {
 504         if (!transport->data)
 505                 transport->data = get_http_walker(transport->url);
 506         return fetch_objs_via_walker(transport, nr_objs, to_fetch);
 507 }
 508
 509 #endif
 510
 511 struct bundle_transport_data {
 512         int fd;
 513         struct bundle_header header;
 514 };
 515
 516 static struct ref *get_refs_from_bundle(const struct transport *transport)
 517 {
 518         struct bundle_transport_data *data = transport->data;
 519         struct ref *result = NULL;
 520         int i;
 521
 522         if (data->fd > 0)
 523                 close(data->fd);
 524         data->fd = read_bundle_header(transport->url, &data->header);
 525         if (data->fd < 0)
 526                 die ("Could not read bundle '%s'.", transport->url);
 527         for (i = 0; i < data->header.references.nr; i++) {
 528                 struct ref_list_entry *e = data->header.references.list + i;
 529                 struct ref *ref = alloc_ref(strlen(e->name) + 1);
 530                 hashcpy(ref->old_sha1, e->sha1);
 531                 strcpy(ref->name, e->name);
 532                 ref->next = result;
 533                 result = ref;
 534         }
 535         return result;
 536 }
 537
 538 static int fetch_refs_from_bundle(struct transport *transport,
 539                                int nr_heads, struct ref **to_fetch)
 540 {
 541         struct bundle_transport_data *data = transport->data;
 542         return unbundle(&data->header, data->fd);
 543 }
 544
 545 static int close_bundle(struct transport *transport)
 546 {
 547         struct bundle_transport_data *data = transport->data;
 548         if (data->fd > 0)
 549                 close(data->fd);
 550         free(data);
 551         return 0;
 552 }
 553
 554 struct git_transport_data {
 555         unsigned thin : 1;
 556         unsigned keep : 1;
 557         int depth;
 558         const char *uploadpack;
 559         const char *receivepack;
 560 };
 561
 562 static int set_git_option(struct transport *connection,
 563                           const char *name, const char *value)
 564 {
 565         struct git_transport_data *data = connection->data;
 566         if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
 567                 data->uploadpack = value;
 568                 return 0;
 569         } else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
 570                 data->receivepack = value;
 571                 return 0;
 572         } else if (!strcmp(name, TRANS_OPT_THIN)) {
 573                 data->thin = !!value;
 574                 return 0;
 575         } else if (!strcmp(name, TRANS_OPT_KEEP)) {
 576                 data->keep = !!value;
 577                 return 0;
 578         } else if (!strcmp(name, TRANS_OPT_DEPTH)) {
 579                 if (!value)
 580                         data->depth = 0;
 581                 else
 582                         data->depth = atoi(value);
 583                 return 0;
 584         }
 585         return 1;
 586 }
 587
 588 static struct ref *get_refs_via_connect(const struct transport *transport)
 589 {
 590         struct git_transport_data *data = transport->data;
 591         struct ref *refs;
 592         int fd[2];
 593         pid_t pid;
 594         char *dest = xstrdup(transport->url);
 595
 596         pid = git_connect(fd, dest, data->uploadpack, 0);
 597
 598         if (pid < 0)
 599                 die("Failed to connect to \"%s\"", transport->url);
 600
 601         get_remote_heads(fd[0], &refs, 0, NULL, 0);
 602         packet_flush(fd[1]);
 603
 604         finish_connect(pid);
 605
 606         free(dest);
 607
 608         return refs;
 609 }
 610
 611 static int fetch_refs_via_pack(struct transport *transport,
 612                                int nr_heads, struct ref **to_fetch)
 613 {
 614         struct git_transport_data *data = transport->data;
 615         char **heads = xmalloc(nr_heads * sizeof(*heads));
 616         char **origh = xmalloc(nr_heads * sizeof(*origh));
 617         struct ref *refs;
 618         char *dest = xstrdup(transport->url);
 619         struct fetch_pack_args args;
 620         int i;
 621
 622         memset(&args, 0, sizeof(args));
 623         args.uploadpack = data->uploadpack;
 624         args.keep_pack = data->keep;
 625         args.lock_pack = 1;
 626         args.use_thin_pack = data->thin;
 627         args.verbose = transport->verbose > 0;
 628         args.depth = data->depth;
 629
 630         for (i = 0; i < nr_heads; i++)
 631                 origh[i] = heads[i] = xstrdup(to_fetch[i]->name);
 632         refs = fetch_pack(&args, dest, nr_heads, heads, &transport->pack_lockfile);
 633
 634         for (i = 0; i < nr_heads; i++)
 635                 free(origh[i]);
 636         free(origh);
 637         free(heads);
 638         free_refs(refs);
 639         free(dest);
 640         return 0;
 641 }
 642
 643 static int git_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
 644         struct git_transport_data *data = transport->data;
 645         const char **argv;
 646         char *rem;
 647         int argc;
 648         int err;
 649
 650         argv = xmalloc((refspec_nr + 11) * sizeof(char *));
 651         argv[0] = "send-pack";
 652         argc = 1;
 653         if (flags & TRANSPORT_PUSH_ALL)
 654                 argv[argc++] = "--all";
 655         if (flags & TRANSPORT_PUSH_FORCE)
 656                 argv[argc++] = "--force";
 657         if (flags & TRANSPORT_PUSH_DRY_RUN)
 658                 argv[argc++] = "--dry-run";
 659         if (data->receivepack) {
 660                 char *rp = xmalloc(strlen(data->receivepack) + 16);
 661                 sprintf(rp, "--receive-pack=%s", data->receivepack);
 662                 argv[argc++] = rp;
 663         }
 664         if (data->thin)
 665                 argv[argc++] = "--thin";
 666         rem = xmalloc(strlen(transport->remote->name) + 10);
 667         sprintf(rem, "--remote=%s", transport->remote->name);
 668         argv[argc++] = rem;
 669         argv[argc++] = transport->url;
 670         while (refspec_nr--)
 671                 argv[argc++] = *refspec++;
 672         argv[argc] = NULL;
 673         err = run_command_v_opt(argv, RUN_GIT_CMD);
 674         switch (err) {
 675         case -ERR_RUN_COMMAND_FORK:
 676                 error("unable to fork for %s", argv[0]);
 677         case -ERR_RUN_COMMAND_EXEC:
 678                 error("unable to exec %s", argv[0]);
 679                 break;
 680         case -ERR_RUN_COMMAND_WAITPID:
 681         case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
 682         case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
 683         case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
 684                 error("%s died with strange error", argv[0]);
 685         }
 686         return !!err;
 687 }
 688
 689 static int disconnect_git(struct transport *transport)
 690 {
 691         free(transport->data);
 692         return 0;
 693 }
 694
 695 static int is_local(const char *url)
 696 {
 697         const char *colon = strchr(url, ':');
 698         const char *slash = strchr(url, '/');
 699         return !colon || (slash && slash < colon);
 700 }
 701
 702 static int is_file(const char *url)
 703 {
 704         struct stat buf;
 705         if (stat(url, &buf))
 706                 return 0;
 707         return S_ISREG(buf.st_mode);
 708 }
 709
 710 struct transport *transport_get(struct remote *remote, const char *url)
 711 {
 712         struct transport *ret = xcalloc(1, sizeof(*ret));
 713
 714         ret->remote = remote;
 715         ret->url = url;
 716
 717         if (!prefixcmp(url, "rsync://")) {
 718                 ret->get_refs_list = get_refs_via_rsync;
 719                 ret->fetch = fetch_objs_via_rsync;
 720                 ret->push = rsync_transport_push;
 721
 722         } else if (!prefixcmp(url, "http://")
 723                 || !prefixcmp(url, "https://")
 724                 || !prefixcmp(url, "ftp://")) {
 725 #ifdef NO_CURL
 726                 error("git was compiled without libcurl support.");
 727 #else
 728                 ret->get_refs_list = get_refs_via_curl;
 729                 ret->fetch = fetch_objs_via_curl;
 730                 ret->push = curl_transport_push;
 731 #endif
 732                 ret->disconnect = disconnect_walker;
 733
 734         } else if (is_local(url) && is_file(url)) {
 735                 struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
 736                 ret->data = data;
 737                 ret->get_refs_list = get_refs_from_bundle;
 738                 ret->fetch = fetch_refs_from_bundle;
 739                 ret->disconnect = close_bundle;
 740
 741         } else {
 742                 struct git_transport_data *data = xcalloc(1, sizeof(*data));
 743                 ret->data = data;
 744                 ret->set_option = set_git_option;
 745                 ret->get_refs_list = get_refs_via_connect;
 746                 ret->fetch = fetch_refs_via_pack;
 747                 ret->push = git_transport_push;
 748                 ret->disconnect = disconnect_git;
 749
 750                 data->thin = 1;
 751                 data->uploadpack = "git-upload-pack";
 752                 if (remote && remote->uploadpack)
 753                         data->uploadpack = remote->uploadpack;
 754                 data->receivepack = "git-receive-pack";
 755                 if (remote && remote->receivepack)
 756                         data->receivepack = remote->receivepack;
 757         }
 758
 759         return ret;
 760 }
 761
 762 int transport_set_option(struct transport *transport,
 763                          const char *name, const char *value)
 764 {
 765         if (transport->set_option)
 766                 return transport->set_option(transport, name, value);
 767         return 1;
 768 }
 769
 770 int transport_push(struct transport *transport,
 771                    int refspec_nr, const char **refspec, int flags)
 772 {
 773         if (!transport->push)
 774                 return 1;
 775         return transport->push(transport, refspec_nr, refspec, flags);
 776 }
 777
 778 struct ref *transport_get_remote_refs(struct transport *transport)
 779 {
 780         if (!transport->remote_refs)
 781                 transport->remote_refs = transport->get_refs_list(transport);
 782         return transport->remote_refs;
 783 }
 784
 785 int transport_fetch_refs(struct transport *transport, struct ref *refs)
 786 {
 787         int rc;
 788         int nr_heads = 0, nr_alloc = 0;
 789         struct ref **heads = NULL;
 790         struct ref *rm;
 791
 792         for (rm = refs; rm; rm = rm->next) {
 793                 if (rm->peer_ref &&
 794                     !hashcmp(rm->peer_ref->old_sha1, rm->old_sha1))
 795                         continue;
 796                 ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
 797                 heads[nr_heads++] = rm;
 798         }
 799
 800         rc = transport->fetch(transport, nr_heads, heads);
 801         free(heads);
 802         return rc;
 803 }
 804
 805 void transport_unlock_pack(struct transport *transport)
 806 {
 807         if (transport->pack_lockfile) {
 808                 unlink(transport->pack_lockfile);
 809                 free(transport->pack_lockfile);
 810                 transport->pack_lockfile = NULL;
 811         }
 812 }
 813
 814 int transport_disconnect(struct transport *transport)
 815 {
 816         int ret = 0;
 817         if (transport->disconnect)
 818                 ret = transport->disconnect(transport);
 819         free(transport);
 820         return ret;
 821 }