transport.c

   1 #include "cache.h"
   2 #include "config.h"
   3 #include "transport.h"
   4 #include "run-command.h"
   5 #include "pkt-line.h"
   6 #include "fetch-pack.h"
   7 #include "remote.h"
   8 #include "connect.h"
   9 #include "send-pack.h"
  10 #include "walker.h"
  11 #include "bundle.h"
  12 #include "dir.h"
  13 #include "refs.h"
  14 #include "branch.h"
  15 #include "url.h"
  16 #include "submodule.h"
  17 #include "string-list.h"
  18 #include "sha1-array.h"
  19 #include "sigchain.h"
  20 #include "transport-internal.h"
  21
  22 static void set_upstreams(struct transport *transport, struct ref *refs,
  23         int pretend)
  24 {
  25         struct ref *ref;
  26         for (ref = refs; ref; ref = ref->next) {
  27                 const char *localname;
  28                 const char *tmp;
  29                 const char *remotename;
  30                 int flag = 0;
  31                 /*
  32                  * Check suitability for tracking. Must be successful /
  33                  * already up-to-date ref create/modify (not delete).
  34                  */
  35                 if (ref->status != REF_STATUS_OK &&
  36                         ref->status != REF_STATUS_UPTODATE)
  37                         continue;
  38                 if (!ref->peer_ref)
  39                         continue;
  40                 if (is_null_oid(&ref->new_oid))
  41                         continue;
  42
  43                 /* Follow symbolic refs (mainly for HEAD). */
  44                 localname = ref->peer_ref->name;
  45                 remotename = ref->name;
  46                 tmp = resolve_ref_unsafe(localname, RESOLVE_REF_READING,
  47                                          NULL, &flag);
  48                 if (tmp && flag & REF_ISSYMREF &&
  49                         starts_with(tmp, "refs/heads/"))
  50                         localname = tmp;
  51
  52                 /* Both source and destination must be local branches. */
  53                 if (!localname || !starts_with(localname, "refs/heads/"))
  54                         continue;
  55                 if (!remotename || !starts_with(remotename, "refs/heads/"))
  56                         continue;
  57
  58                 if (!pretend)
  59                         install_branch_config(BRANCH_CONFIG_VERBOSE,
  60                                 localname + 11, transport->remote->name,
  61                                 remotename);
  62                 else
  63                         printf(_("Would set upstream of '%s' to '%s' of '%s'\n"),
  64                                 localname + 11, remotename + 11,
  65                                 transport->remote->name);
  66         }
  67 }
  68
  69 struct bundle_transport_data {
  70         int fd;
  71         struct bundle_header header;
  72 };
  73
  74 static struct ref *get_refs_from_bundle(struct transport *transport, int for_push)
  75 {
  76         struct bundle_transport_data *data = transport->data;
  77         struct ref *result = NULL;
  78         int i;
  79
  80         if (for_push)
  81                 return NULL;
  82
  83         if (data->fd > 0)
  84                 close(data->fd);
  85         data->fd = read_bundle_header(transport->url, &data->header);
  86         if (data->fd < 0)
  87                 die ("Could not read bundle '%s'.", transport->url);
  88         for (i = 0; i < data->header.references.nr; i++) {
  89                 struct ref_list_entry *e = data->header.references.list + i;
  90                 struct ref *ref = alloc_ref(e->name);
  91                 oidcpy(&ref->old_oid, &e->oid);
  92                 ref->next = result;
  93                 result = ref;
  94         }
  95         return result;
  96 }
  97
  98 static int fetch_refs_from_bundle(struct transport *transport,
  99                                int nr_heads, struct ref **to_fetch)
 100 {
 101         struct bundle_transport_data *data = transport->data;
 102         return unbundle(&data->header, data->fd,
 103                         transport->progress ? BUNDLE_VERBOSE : 0);
 104 }
 105
 106 static int close_bundle(struct transport *transport)
 107 {
 108         struct bundle_transport_data *data = transport->data;
 109         if (data->fd > 0)
 110                 close(data->fd);
 111         free(data);
 112         return 0;
 113 }
 114
 115 struct git_transport_data {
 116         struct git_transport_options options;
 117         struct child_process *conn;
 118         int fd[2];
 119         unsigned got_remote_heads : 1;
 120         struct oid_array extra_have;
 121         struct oid_array shallow;
 122 };
 123
 124 static int set_git_option(struct git_transport_options *opts,
 125                           const char *name, const char *value)
 126 {
 127         if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
 128                 opts->uploadpack = value;
 129                 return 0;
 130         } else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
 131                 opts->receivepack = value;
 132                 return 0;
 133         } else if (!strcmp(name, TRANS_OPT_THIN)) {
 134                 opts->thin = !!value;
 135                 return 0;
 136         } else if (!strcmp(name, TRANS_OPT_FOLLOWTAGS)) {
 137                 opts->followtags = !!value;
 138                 return 0;
 139         } else if (!strcmp(name, TRANS_OPT_KEEP)) {
 140                 opts->keep = !!value;
 141                 return 0;
 142         } else if (!strcmp(name, TRANS_OPT_UPDATE_SHALLOW)) {
 143                 opts->update_shallow = !!value;
 144                 return 0;
 145         } else if (!strcmp(name, TRANS_OPT_DEPTH)) {
 146                 if (!value)
 147                         opts->depth = 0;
 148                 else {
 149                         char *end;
 150                         opts->depth = strtol(value, &end, 0);
 151                         if (*end)
 152                                 die(_("transport: invalid depth option '%s'"), value);
 153                 }
 154                 return 0;
 155         } else if (!strcmp(name, TRANS_OPT_DEEPEN_SINCE)) {
 156                 opts->deepen_since = value;
 157                 return 0;
 158         } else if (!strcmp(name, TRANS_OPT_DEEPEN_NOT)) {
 159                 opts->deepen_not = (const struct string_list *)value;
 160                 return 0;
 161         } else if (!strcmp(name, TRANS_OPT_DEEPEN_RELATIVE)) {
 162                 opts->deepen_relative = !!value;
 163                 return 0;
 164         } else if (!strcmp(name, TRANS_OPT_FROM_PROMISOR)) {
 165                 opts->from_promisor = !!value;
 166                 return 0;
 167         } else if (!strcmp(name, TRANS_OPT_NO_DEPENDENTS)) {
 168                 opts->no_dependents = !!value;
 169                 return 0;
 170         } else if (!strcmp(name, TRANS_OPT_LIST_OBJECTS_FILTER)) {
 171                 parse_list_objects_filter(&opts->filter_options, value);
 172                 return 0;
 173         }
 174         return 1;
 175 }
 176
 177 static int connect_setup(struct transport *transport, int for_push)
 178 {
 179         struct git_transport_data *data = transport->data;
 180         int flags = transport->verbose > 0 ? CONNECT_VERBOSE : 0;
 181
 182         if (data->conn)
 183                 return 0;
 184
 185         switch (transport->family) {
 186         case TRANSPORT_FAMILY_ALL: break;
 187         case TRANSPORT_FAMILY_IPV4: flags |= CONNECT_IPV4; break;
 188         case TRANSPORT_FAMILY_IPV6: flags |= CONNECT_IPV6; break;
 189         }
 190
 191         data->conn = git_connect(data->fd, transport->url,
 192                                  for_push ? data->options.receivepack :
 193                                  data->options.uploadpack,
 194                                  flags);
 195
 196         return 0;
 197 }
 198
 199 static struct ref *get_refs_via_connect(struct transport *transport, int for_push)
 200 {
 201         struct git_transport_data *data = transport->data;
 202         struct ref *refs;
 203
 204         connect_setup(transport, for_push);
 205         get_remote_heads(data->fd[0], NULL, 0, &refs,
 206                          for_push ? REF_NORMAL : 0,
 207                          &data->extra_have,
 208                          &data->shallow);
 209         data->got_remote_heads = 1;
 210
 211         return refs;
 212 }
 213
 214 static int fetch_refs_via_pack(struct transport *transport,
 215                                int nr_heads, struct ref **to_fetch)
 216 {
 217         int ret = 0;
 218         struct git_transport_data *data = transport->data;
 219         struct ref *refs;
 220         char *dest = xstrdup(transport->url);
 221         struct fetch_pack_args args;
 222         struct ref *refs_tmp = NULL;
 223
 224         memset(&args, 0, sizeof(args));
 225         args.uploadpack = data->options.uploadpack;
 226         args.keep_pack = data->options.keep;
 227         args.lock_pack = 1;
 228         args.use_thin_pack = data->options.thin;
 229         args.include_tag = data->options.followtags;
 230         args.verbose = (transport->verbose > 1);
 231         args.quiet = (transport->verbose < 0);
 232         args.no_progress = !transport->progress;
 233         args.depth = data->options.depth;
 234         args.deepen_since = data->options.deepen_since;
 235         args.deepen_not = data->options.deepen_not;
 236         args.deepen_relative = data->options.deepen_relative;
 237         args.check_self_contained_and_connected =
 238                 data->options.check_self_contained_and_connected;
 239         args.cloning = transport->cloning;
 240         args.update_shallow = data->options.update_shallow;
 241         args.from_promisor = data->options.from_promisor;
 242         args.no_dependents = data->options.no_dependents;
 243         args.filter_options = data->options.filter_options;
 244
 245         if (!data->got_remote_heads) {
 246                 connect_setup(transport, 0);
 247                 get_remote_heads(data->fd[0], NULL, 0, &refs_tmp, 0,
 248                                  NULL, &data->shallow);
 249                 data->got_remote_heads = 1;
 250         }
 251
 252         refs = fetch_pack(&args, data->fd, data->conn,
 253                           refs_tmp ? refs_tmp : transport->remote_refs,
 254                           dest, to_fetch, nr_heads, &data->shallow,
 255                           &transport->pack_lockfile);
 256         close(data->fd[0]);
 257         close(data->fd[1]);
 258         if (finish_connect(data->conn))
 259                 ret = -1;
 260         data->conn = NULL;
 261         data->got_remote_heads = 0;
 262         data->options.self_contained_and_connected =
 263                 args.self_contained_and_connected;
 264
 265         if (refs == NULL)
 266                 ret = -1;
 267         if (report_unmatched_refs(to_fetch, nr_heads))
 268                 ret = -1;
 269
 270         free_refs(refs_tmp);
 271         free_refs(refs);
 272         free(dest);
 273         return ret;
 274 }
 275
 276 static int push_had_errors(struct ref *ref)
 277 {
 278         for (; ref; ref = ref->next) {
 279                 switch (ref->status) {
 280                 case REF_STATUS_NONE:
 281                 case REF_STATUS_UPTODATE:
 282                 case REF_STATUS_OK:
 283                         break;
 284                 default:
 285                         return 1;
 286                 }
 287         }
 288         return 0;
 289 }
 290
 291 int transport_refs_pushed(struct ref *ref)
 292 {
 293         for (; ref; ref = ref->next) {
 294                 switch(ref->status) {
 295                 case REF_STATUS_NONE:
 296                 case REF_STATUS_UPTODATE:
 297                         break;
 298                 default:
 299                         return 1;
 300                 }
 301         }
 302         return 0;
 303 }
 304
 305 void transport_update_tracking_ref(struct remote *remote, struct ref *ref, int verbose)
 306 {
 307         struct refspec rs;
 308
 309         if (ref->status != REF_STATUS_OK && ref->status != REF_STATUS_UPTODATE)
 310                 return;
 311
 312         rs.src = ref->name;
 313         rs.dst = NULL;
 314
 315         if (!remote_find_tracking(remote, &rs)) {
 316                 if (verbose)
 317                         fprintf(stderr, "updating local tracking ref '%s'\n", rs.dst);
 318                 if (ref->deletion) {
 319                         delete_ref(NULL, rs.dst, NULL, 0);
 320                 } else
 321                         update_ref("update by push", rs.dst, &ref->new_oid,
 322                                    NULL, 0, 0);
 323                 free(rs.dst);
 324         }
 325 }
 326
 327 static void print_ref_status(char flag, const char *summary,
 328                              struct ref *to, struct ref *from, const char *msg,
 329                              int porcelain, int summary_width)
 330 {
 331         if (porcelain) {
 332                 if (from)
 333                         fprintf(stdout, "%c\t%s:%s\t", flag, from->name, to->name);
 334                 else
 335                         fprintf(stdout, "%c\t:%s\t", flag, to->name);
 336                 if (msg)
 337                         fprintf(stdout, "%s (%s)\n", summary, msg);
 338                 else
 339                         fprintf(stdout, "%s\n", summary);
 340         } else {
 341                 fprintf(stderr, " %c %-*s ", flag, summary_width, summary);
 342                 if (from)
 343                         fprintf(stderr, "%s -> %s", prettify_refname(from->name), prettify_refname(to->name));
 344                 else
 345                         fputs(prettify_refname(to->name), stderr);
 346                 if (msg) {
 347                         fputs(" (", stderr);
 348                         fputs(msg, stderr);
 349                         fputc(')', stderr);
 350                 }
 351                 fputc('\n', stderr);
 352         }
 353 }
 354
 355 static void print_ok_ref_status(struct ref *ref, int porcelain, int summary_width)
 356 {
 357         if (ref->deletion)
 358                 print_ref_status('-', "[deleted]", ref, NULL, NULL,
 359                                  porcelain, summary_width);
 360         else if (is_null_oid(&ref->old_oid))
 361                 print_ref_status('*',
 362                         (starts_with(ref->name, "refs/tags/") ? "[new tag]" :
 363                         "[new branch]"),
 364                         ref, ref->peer_ref, NULL, porcelain, summary_width);
 365         else {
 366                 struct strbuf quickref = STRBUF_INIT;
 367                 char type;
 368                 const char *msg;
 369
 370                 strbuf_add_unique_abbrev(&quickref, &ref->old_oid,
 371                                          DEFAULT_ABBREV);
 372                 if (ref->forced_update) {
 373                         strbuf_addstr(&quickref, "...");
 374                         type = '+';
 375                         msg = "forced update";
 376                 } else {
 377                         strbuf_addstr(&quickref, "..");
 378                         type = ' ';
 379                         msg = NULL;
 380                 }
 381                 strbuf_add_unique_abbrev(&quickref, &ref->new_oid,
 382                                          DEFAULT_ABBREV);
 383
 384                 print_ref_status(type, quickref.buf, ref, ref->peer_ref, msg,
 385                                  porcelain, summary_width);
 386                 strbuf_release(&quickref);
 387         }
 388 }
 389
 390 static int print_one_push_status(struct ref *ref, const char *dest, int count,
 391                                  int porcelain, int summary_width)
 392 {
 393         if (!count) {
 394                 char *url = transport_anonymize_url(dest);
 395                 fprintf(porcelain ? stdout : stderr, "To %s\n", url);
 396                 free(url);
 397         }
 398
 399         switch(ref->status) {
 400         case REF_STATUS_NONE:
 401                 print_ref_status('X', "[no match]", ref, NULL, NULL,
 402                                  porcelain, summary_width);
 403                 break;
 404         case REF_STATUS_REJECT_NODELETE:
 405                 print_ref_status('!', "[rejected]", ref, NULL,
 406                                  "remote does not support deleting refs",
 407                                  porcelain, summary_width);
 408                 break;
 409         case REF_STATUS_UPTODATE:
 410                 print_ref_status('=', "[up to date]", ref,
 411                                  ref->peer_ref, NULL, porcelain, summary_width);
 412                 break;
 413         case REF_STATUS_REJECT_NONFASTFORWARD:
 414                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 415                                  "non-fast-forward", porcelain, summary_width);
 416                 break;
 417         case REF_STATUS_REJECT_ALREADY_EXISTS:
 418                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 419                                  "already exists", porcelain, summary_width);
 420                 break;
 421         case REF_STATUS_REJECT_FETCH_FIRST:
 422                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 423                                  "fetch first", porcelain, summary_width);
 424                 break;
 425         case REF_STATUS_REJECT_NEEDS_FORCE:
 426                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 427                                  "needs force", porcelain, summary_width);
 428                 break;
 429         case REF_STATUS_REJECT_STALE:
 430                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 431                                  "stale info", porcelain, summary_width);
 432                 break;
 433         case REF_STATUS_REJECT_SHALLOW:
 434                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 435                                  "new shallow roots not allowed",
 436                                  porcelain, summary_width);
 437                 break;
 438         case REF_STATUS_REMOTE_REJECT:
 439                 print_ref_status('!', "[remote rejected]", ref,
 440                                  ref->deletion ? NULL : ref->peer_ref,
 441                                  ref->remote_status, porcelain, summary_width);
 442                 break;
 443         case REF_STATUS_EXPECTING_REPORT:
 444                 print_ref_status('!', "[remote failure]", ref,
 445                                  ref->deletion ? NULL : ref->peer_ref,
 446                                  "remote failed to report status",
 447                                  porcelain, summary_width);
 448                 break;
 449         case REF_STATUS_ATOMIC_PUSH_FAILED:
 450                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
 451                                  "atomic push failed", porcelain, summary_width);
 452                 break;
 453         case REF_STATUS_OK:
 454                 print_ok_ref_status(ref, porcelain, summary_width);
 455                 break;
 456         }
 457
 458         return 1;
 459 }
 460
 461 static int measure_abbrev(const struct object_id *oid, int sofar)
 462 {
 463         char hex[GIT_MAX_HEXSZ + 1];
 464         int w = find_unique_abbrev_r(hex, oid, DEFAULT_ABBREV);
 465
 466         return (w < sofar) ? sofar : w;
 467 }
 468
 469 int transport_summary_width(const struct ref *refs)
 470 {
 471         int maxw = -1;
 472
 473         for (; refs; refs = refs->next) {
 474                 maxw = measure_abbrev(&refs->old_oid, maxw);
 475                 maxw = measure_abbrev(&refs->new_oid, maxw);
 476         }
 477         if (maxw < 0)
 478                 maxw = FALLBACK_DEFAULT_ABBREV;
 479         return (2 * maxw + 3);
 480 }
 481
 482 void transport_print_push_status(const char *dest, struct ref *refs,
 483                                   int verbose, int porcelain, unsigned int *reject_reasons)
 484 {
 485         struct ref *ref;
 486         int n = 0;
 487         char *head;
 488         int summary_width = transport_summary_width(refs);
 489
 490         head = resolve_refdup("HEAD", RESOLVE_REF_READING, NULL, NULL);
 491
 492         if (verbose) {
 493                 for (ref = refs; ref; ref = ref->next)
 494                         if (ref->status == REF_STATUS_UPTODATE)
 495                                 n += print_one_push_status(ref, dest, n,
 496                                                            porcelain, summary_width);
 497         }
 498
 499         for (ref = refs; ref; ref = ref->next)
 500                 if (ref->status == REF_STATUS_OK)
 501                         n += print_one_push_status(ref, dest, n,
 502                                                    porcelain, summary_width);
 503
 504         *reject_reasons = 0;
 505         for (ref = refs; ref; ref = ref->next) {
 506                 if (ref->status != REF_STATUS_NONE &&
 507                     ref->status != REF_STATUS_UPTODATE &&
 508                     ref->status != REF_STATUS_OK)
 509                         n += print_one_push_status(ref, dest, n,
 510                                                    porcelain, summary_width);
 511                 if (ref->status == REF_STATUS_REJECT_NONFASTFORWARD) {
 512                         if (head != NULL && !strcmp(head, ref->name))
 513                                 *reject_reasons |= REJECT_NON_FF_HEAD;
 514                         else
 515                                 *reject_reasons |= REJECT_NON_FF_OTHER;
 516                 } else if (ref->status == REF_STATUS_REJECT_ALREADY_EXISTS) {
 517                         *reject_reasons |= REJECT_ALREADY_EXISTS;
 518                 } else if (ref->status == REF_STATUS_REJECT_FETCH_FIRST) {
 519                         *reject_reasons |= REJECT_FETCH_FIRST;
 520                 } else if (ref->status == REF_STATUS_REJECT_NEEDS_FORCE) {
 521                         *reject_reasons |= REJECT_NEEDS_FORCE;
 522                 }
 523         }
 524         free(head);
 525 }
 526
 527 void transport_verify_remote_names(int nr_heads, const char **heads)
 528 {
 529         int i;
 530
 531         for (i = 0; i < nr_heads; i++) {
 532                 const char *local = heads[i];
 533                 const char *remote = strrchr(heads[i], ':');
 534
 535                 if (*local == '+')
 536                         local++;
 537
 538                 /* A matching refspec is okay.  */
 539                 if (remote == local && remote[1] == '\0')
 540                         continue;
 541
 542                 remote = remote ? (remote + 1) : local;
 543                 if (check_refname_format(remote,
 544                                 REFNAME_ALLOW_ONELEVEL|REFNAME_REFSPEC_PATTERN))
 545                         die("remote part of refspec is not a valid name in %s",
 546                                 heads[i]);
 547         }
 548 }
 549
 550 static int git_transport_push(struct transport *transport, struct ref *remote_refs, int flags)
 551 {
 552         struct git_transport_data *data = transport->data;
 553         struct send_pack_args args;
 554         int ret;
 555
 556         if (!data->got_remote_heads) {
 557                 struct ref *tmp_refs;
 558                 connect_setup(transport, 1);
 559
 560                 get_remote_heads(data->fd[0], NULL, 0, &tmp_refs, REF_NORMAL,
 561                                  NULL, &data->shallow);
 562                 data->got_remote_heads = 1;
 563         }
 564
 565         memset(&args, 0, sizeof(args));
 566         args.send_mirror = !!(flags & TRANSPORT_PUSH_MIRROR);
 567         args.force_update = !!(flags & TRANSPORT_PUSH_FORCE);
 568         args.use_thin_pack = data->options.thin;
 569         args.verbose = (transport->verbose > 0);
 570         args.quiet = (transport->verbose < 0);
 571         args.progress = transport->progress;
 572         args.dry_run = !!(flags & TRANSPORT_PUSH_DRY_RUN);
 573         args.porcelain = !!(flags & TRANSPORT_PUSH_PORCELAIN);
 574         args.atomic = !!(flags & TRANSPORT_PUSH_ATOMIC);
 575         args.push_options = transport->push_options;
 576         args.url = transport->url;
 577
 578         if (flags & TRANSPORT_PUSH_CERT_ALWAYS)
 579                 args.push_cert = SEND_PACK_PUSH_CERT_ALWAYS;
 580         else if (flags & TRANSPORT_PUSH_CERT_IF_ASKED)
 581                 args.push_cert = SEND_PACK_PUSH_CERT_IF_ASKED;
 582         else
 583                 args.push_cert = SEND_PACK_PUSH_CERT_NEVER;
 584
 585         ret = send_pack(&args, data->fd, data->conn, remote_refs,
 586                         &data->extra_have);
 587
 588         close(data->fd[1]);
 589         close(data->fd[0]);
 590         ret |= finish_connect(data->conn);
 591         data->conn = NULL;
 592         data->got_remote_heads = 0;
 593
 594         return ret;
 595 }
 596
 597 static int connect_git(struct transport *transport, const char *name,
 598                        const char *executable, int fd[2])
 599 {
 600         struct git_transport_data *data = transport->data;
 601         data->conn = git_connect(data->fd, transport->url,
 602                                  executable, 0);
 603         fd[0] = data->fd[0];
 604         fd[1] = data->fd[1];
 605         return 0;
 606 }
 607
 608 static int disconnect_git(struct transport *transport)
 609 {
 610         struct git_transport_data *data = transport->data;
 611         if (data->conn) {
 612                 if (data->got_remote_heads)
 613                         packet_flush(data->fd[1]);
 614                 close(data->fd[0]);
 615                 close(data->fd[1]);
 616                 finish_connect(data->conn);
 617         }
 618
 619         free(data);
 620         return 0;
 621 }
 622
 623 static struct transport_vtable taken_over_vtable = {
 624         NULL,
 625         get_refs_via_connect,
 626         fetch_refs_via_pack,
 627         git_transport_push,
 628         NULL,
 629         disconnect_git
 630 };
 631
 632 void transport_take_over(struct transport *transport,
 633                          struct child_process *child)
 634 {
 635         struct git_transport_data *data;
 636
 637         if (!transport->smart_options)
 638                 die("BUG: taking over transport requires non-NULL "
 639                     "smart_options field.");
 640
 641         data = xcalloc(1, sizeof(*data));
 642         data->options = *transport->smart_options;
 643         data->conn = child;
 644         data->fd[0] = data->conn->out;
 645         data->fd[1] = data->conn->in;
 646         data->got_remote_heads = 0;
 647         transport->data = data;
 648
 649         transport->vtable = &taken_over_vtable;
 650         transport->smart_options = &(data->options);
 651
 652         transport->cannot_reuse = 1;
 653 }
 654
 655 static int is_file(const char *url)
 656 {
 657         struct stat buf;
 658         if (stat(url, &buf))
 659                 return 0;
 660         return S_ISREG(buf.st_mode);
 661 }
 662
 663 static int external_specification_len(const char *url)
 664 {
 665         return strchr(url, ':') - url;
 666 }
 667
 668 static const struct string_list *protocol_whitelist(void)
 669 {
 670         static int enabled = -1;
 671         static struct string_list allowed = STRING_LIST_INIT_DUP;
 672
 673         if (enabled < 0) {
 674                 const char *v = getenv("GIT_ALLOW_PROTOCOL");
 675                 if (v) {
 676                         string_list_split(&allowed, v, ':', -1);
 677                         string_list_sort(&allowed);
 678                         enabled = 1;
 679                 } else {
 680                         enabled = 0;
 681                 }
 682         }
 683
 684         return enabled ? &allowed : NULL;
 685 }
 686
 687 enum protocol_allow_config {
 688         PROTOCOL_ALLOW_NEVER = 0,
 689         PROTOCOL_ALLOW_USER_ONLY,
 690         PROTOCOL_ALLOW_ALWAYS
 691 };
 692
 693 static enum protocol_allow_config parse_protocol_config(const char *key,
 694                                                         const char *value)
 695 {
 696         if (!strcasecmp(value, "always"))
 697                 return PROTOCOL_ALLOW_ALWAYS;
 698         else if (!strcasecmp(value, "never"))
 699                 return PROTOCOL_ALLOW_NEVER;
 700         else if (!strcasecmp(value, "user"))
 701                 return PROTOCOL_ALLOW_USER_ONLY;
 702
 703         die("unknown value for config '%s': %s", key, value);
 704 }
 705
 706 static enum protocol_allow_config get_protocol_config(const char *type)
 707 {
 708         char *key = xstrfmt("protocol.%s.allow", type);
 709         char *value;
 710
 711         /* first check the per-protocol config */
 712         if (!git_config_get_string(key, &value)) {
 713                 enum protocol_allow_config ret =
 714                         parse_protocol_config(key, value);
 715                 free(key);
 716                 free(value);
 717                 return ret;
 718         }
 719         free(key);
 720
 721         /* if defined, fallback to user-defined default for unknown protocols */
 722         if (!git_config_get_string("protocol.allow", &value)) {
 723                 enum protocol_allow_config ret =
 724                         parse_protocol_config("protocol.allow", value);
 725                 free(value);
 726                 return ret;
 727         }
 728
 729         /* fallback to built-in defaults */
 730         /* known safe */
 731         if (!strcmp(type, "http") ||
 732             !strcmp(type, "https") ||
 733             !strcmp(type, "git") ||
 734             !strcmp(type, "ssh") ||
 735             !strcmp(type, "file"))
 736                 return PROTOCOL_ALLOW_ALWAYS;
 737
 738         /* known scary; err on the side of caution */
 739         if (!strcmp(type, "ext"))
 740                 return PROTOCOL_ALLOW_NEVER;
 741
 742         /* unknown; by default let them be used only directly by the user */
 743         return PROTOCOL_ALLOW_USER_ONLY;
 744 }
 745
 746 int is_transport_allowed(const char *type, int from_user)
 747 {
 748         const struct string_list *whitelist = protocol_whitelist();
 749         if (whitelist)
 750                 return string_list_has_string(whitelist, type);
 751
 752         switch (get_protocol_config(type)) {
 753         case PROTOCOL_ALLOW_ALWAYS:
 754                 return 1;
 755         case PROTOCOL_ALLOW_NEVER:
 756                 return 0;
 757         case PROTOCOL_ALLOW_USER_ONLY:
 758                 if (from_user < 0)
 759                         from_user = git_env_bool("GIT_PROTOCOL_FROM_USER", 1);
 760                 return from_user;
 761         }
 762
 763         die("BUG: invalid protocol_allow_config type");
 764 }
 765
 766 void transport_check_allowed(const char *type)
 767 {
 768         if (!is_transport_allowed(type, -1))
 769                 die("transport '%s' not allowed", type);
 770 }
 771
 772 static struct transport_vtable bundle_vtable = {
 773         NULL,
 774         get_refs_from_bundle,
 775         fetch_refs_from_bundle,
 776         NULL,
 777         NULL,
 778         close_bundle
 779 };
 780
 781 static struct transport_vtable builtin_smart_vtable = {
 782         NULL,
 783         get_refs_via_connect,
 784         fetch_refs_via_pack,
 785         git_transport_push,
 786         connect_git,
 787         disconnect_git
 788 };
 789
 790 struct transport *transport_get(struct remote *remote, const char *url)
 791 {
 792         const char *helper;
 793         struct transport *ret = xcalloc(1, sizeof(*ret));
 794
 795         ret->progress = isatty(2);
 796
 797         if (!remote)
 798                 die("No remote provided to transport_get()");
 799
 800         ret->got_remote_refs = 0;
 801         ret->remote = remote;
 802         helper = remote->foreign_vcs;
 803
 804         if (!url && remote->url)
 805                 url = remote->url[0];
 806         ret->url = url;
 807
 808         /* maybe it is a foreign URL? */
 809         if (url) {
 810                 const char *p = url;
 811
 812                 while (is_urlschemechar(p == url, *p))
 813                         p++;
 814                 if (starts_with(p, "::"))
 815                         helper = xstrndup(url, p - url);
 816         }
 817
 818         if (helper) {
 819                 transport_helper_init(ret, helper);
 820         } else if (starts_with(url, "rsync:")) {
 821                 die("git-over-rsync is no longer supported");
 822         } else if (url_is_local_not_ssh(url) && is_file(url) && is_bundle(url, 1)) {
 823                 struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
 824                 transport_check_allowed("file");
 825                 ret->data = data;
 826                 ret->vtable = &bundle_vtable;
 827                 ret->smart_options = NULL;
 828         } else if (!is_url(url)
 829                 || starts_with(url, "file://")
 830                 || starts_with(url, "git://")
 831                 || starts_with(url, "ssh://")
 832                 || starts_with(url, "git+ssh://") /* deprecated - do not use */
 833                 || starts_with(url, "ssh+git://") /* deprecated - do not use */
 834                 ) {
 835                 /*
 836                  * These are builtin smart transports; "allowed" transports
 837                  * will be checked individually in git_connect.
 838                  */
 839                 struct git_transport_data *data = xcalloc(1, sizeof(*data));
 840                 ret->data = data;
 841                 ret->vtable = &builtin_smart_vtable;
 842                 ret->smart_options = &(data->options);
 843
 844                 data->conn = NULL;
 845                 data->got_remote_heads = 0;
 846         } else {
 847                 /* Unknown protocol in URL. Pass to external handler. */
 848                 int len = external_specification_len(url);
 849                 char *handler = xmemdupz(url, len);
 850                 transport_helper_init(ret, handler);
 851         }
 852
 853         if (ret->smart_options) {
 854                 ret->smart_options->thin = 1;
 855                 ret->smart_options->uploadpack = "git-upload-pack";
 856                 if (remote->uploadpack)
 857                         ret->smart_options->uploadpack = remote->uploadpack;
 858                 ret->smart_options->receivepack = "git-receive-pack";
 859                 if (remote->receivepack)
 860                         ret->smart_options->receivepack = remote->receivepack;
 861         }
 862
 863         return ret;
 864 }
 865
 866 int transport_set_option(struct transport *transport,
 867                          const char *name, const char *value)
 868 {
 869         int git_reports = 1, protocol_reports = 1;
 870
 871         if (transport->smart_options)
 872                 git_reports = set_git_option(transport->smart_options,
 873                                              name, value);
 874
 875         if (transport->vtable->set_option)
 876                 protocol_reports = transport->vtable->set_option(transport,
 877                                                                  name, value);
 878
 879         /* If either report is 0, report 0 (success). */
 880         if (!git_reports || !protocol_reports)
 881                 return 0;
 882         /* If either reports -1 (invalid value), report -1. */
 883         if ((git_reports == -1) || (protocol_reports == -1))
 884                 return -1;
 885         /* Otherwise if both report unknown, report unknown. */
 886         return 1;
 887 }
 888
 889 void transport_set_verbosity(struct transport *transport, int verbosity,
 890         int force_progress)
 891 {
 892         if (verbosity >= 1)
 893                 transport->verbose = verbosity <= 3 ? verbosity : 3;
 894         if (verbosity < 0)
 895                 transport->verbose = -1;
 896
 897         /**
 898          * Rules used to determine whether to report progress (processing aborts
 899          * when a rule is satisfied):
 900          *
 901          *   . Report progress, if force_progress is 1 (ie. --progress).
 902          *   . Don't report progress, if force_progress is 0 (ie. --no-progress).
 903          *   . Don't report progress, if verbosity < 0 (ie. -q/--quiet ).
 904          *   . Report progress if isatty(2) is 1.
 905          **/
 906         if (force_progress >= 0)
 907                 transport->progress = !!force_progress;
 908         else
 909                 transport->progress = verbosity >= 0 && isatty(2);
 910 }
 911
 912 static void die_with_unpushed_submodules(struct string_list *needs_pushing)
 913 {
 914         int i;
 915
 916         fprintf(stderr, _("The following submodule paths contain changes that can\n"
 917                         "not be found on any remote:\n"));
 918         for (i = 0; i < needs_pushing->nr; i++)
 919                 fprintf(stderr, "  %s\n", needs_pushing->items[i].string);
 920         fprintf(stderr, _("\nPlease try\n\n"
 921                           "     git push --recurse-submodules=on-demand\n\n"
 922                           "or cd to the path and use\n\n"
 923                           "     git push\n\n"
 924                           "to push them to a remote.\n\n"));
 925
 926         string_list_clear(needs_pushing, 0);
 927
 928         die(_("Aborting."));
 929 }
 930
 931 static int run_pre_push_hook(struct transport *transport,
 932                              struct ref *remote_refs)
 933 {
 934         int ret = 0, x;
 935         struct ref *r;
 936         struct child_process proc = CHILD_PROCESS_INIT;
 937         struct strbuf buf;
 938         const char *argv[4];
 939
 940         if (!(argv[0] = find_hook("pre-push")))
 941                 return 0;
 942
 943         argv[1] = transport->remote->name;
 944         argv[2] = transport->url;
 945         argv[3] = NULL;
 946
 947         proc.argv = argv;
 948         proc.in = -1;
 949
 950         if (start_command(&proc)) {
 951                 finish_command(&proc);
 952                 return -1;
 953         }
 954
 955         sigchain_push(SIGPIPE, SIG_IGN);
 956
 957         strbuf_init(&buf, 256);
 958
 959         for (r = remote_refs; r; r = r->next) {
 960                 if (!r->peer_ref) continue;
 961                 if (r->status == REF_STATUS_REJECT_NONFASTFORWARD) continue;
 962                 if (r->status == REF_STATUS_REJECT_STALE) continue;
 963                 if (r->status == REF_STATUS_UPTODATE) continue;
 964
 965                 strbuf_reset(&buf);
 966                 strbuf_addf( &buf, "%s %s %s %s\n",
 967                          r->peer_ref->name, oid_to_hex(&r->new_oid),
 968                          r->name, oid_to_hex(&r->old_oid));
 969
 970                 if (write_in_full(proc.in, buf.buf, buf.len) < 0) {
 971                         /* We do not mind if a hook does not read all refs. */
 972                         if (errno != EPIPE)
 973                                 ret = -1;
 974                         break;
 975                 }
 976         }
 977
 978         strbuf_release(&buf);
 979
 980         x = close(proc.in);
 981         if (!ret)
 982                 ret = x;
 983
 984         sigchain_pop(SIGPIPE);
 985
 986         x = finish_command(&proc);
 987         if (!ret)
 988                 ret = x;
 989
 990         return ret;
 991 }
 992
 993 int transport_push(struct transport *transport,
 994                    int refspec_nr, const char **refspec, int flags,
 995                    unsigned int *reject_reasons)
 996 {
 997         *reject_reasons = 0;
 998         transport_verify_remote_names(refspec_nr, refspec);
 999
1000         if (transport->vtable->push_refs) {
1001                 struct ref *remote_refs;
1002                 struct ref *local_refs = get_local_heads();
1003                 int match_flags = MATCH_REFS_NONE;
1004                 int verbose = (transport->verbose > 0);
1005                 int quiet = (transport->verbose < 0);
1006                 int porcelain = flags & TRANSPORT_PUSH_PORCELAIN;
1007                 int pretend = flags & TRANSPORT_PUSH_DRY_RUN;
1008                 int push_ret, ret, err;
1009
1010                 if (check_push_refs(local_refs, refspec_nr, refspec) < 0)
1011                         return -1;
1012
1013                 remote_refs = transport->vtable->get_refs_list(transport, 1);
1014
1015                 if (flags & TRANSPORT_PUSH_ALL)
1016                         match_flags |= MATCH_REFS_ALL;
1017                 if (flags & TRANSPORT_PUSH_MIRROR)
1018                         match_flags |= MATCH_REFS_MIRROR;
1019                 if (flags & TRANSPORT_PUSH_PRUNE)
1020                         match_flags |= MATCH_REFS_PRUNE;
1021                 if (flags & TRANSPORT_PUSH_FOLLOW_TAGS)
1022                         match_flags |= MATCH_REFS_FOLLOW_TAGS;
1023
1024                 if (match_push_refs(local_refs, &remote_refs,
1025                                     refspec_nr, refspec, match_flags)) {
1026                         return -1;
1027                 }
1028
1029                 if (transport->smart_options &&
1030                     transport->smart_options->cas &&
1031                     !is_empty_cas(transport->smart_options->cas))
1032                         apply_push_cas(transport->smart_options->cas,
1033                                        transport->remote, remote_refs);
1034
1035                 set_ref_status_for_push(remote_refs,
1036                         flags & TRANSPORT_PUSH_MIRROR,
1037                         flags & TRANSPORT_PUSH_FORCE);
1038
1039                 if (!(flags & TRANSPORT_PUSH_NO_HOOK))
1040                         if (run_pre_push_hook(transport, remote_refs))
1041                                 return -1;
1042
1043                 if ((flags & (TRANSPORT_RECURSE_SUBMODULES_ON_DEMAND |
1044                               TRANSPORT_RECURSE_SUBMODULES_ONLY)) &&
1045                     !is_bare_repository()) {
1046                         struct ref *ref = remote_refs;
1047                         struct oid_array commits = OID_ARRAY_INIT;
1048
1049                         for (; ref; ref = ref->next)
1050                                 if (!is_null_oid(&ref->new_oid))
1051                                         oid_array_append(&commits,
1052                                                           &ref->new_oid);
1053
1054                         if (!push_unpushed_submodules(&commits,
1055                                                       transport->remote,
1056                                                       refspec, refspec_nr,
1057                                                       transport->push_options,
1058                                                       pretend)) {
1059                                 oid_array_clear(&commits);
1060                                 die("Failed to push all needed submodules!");
1061                         }
1062                         oid_array_clear(&commits);
1063                 }
1064
1065                 if (((flags & TRANSPORT_RECURSE_SUBMODULES_CHECK) ||
1066                      ((flags & (TRANSPORT_RECURSE_SUBMODULES_ON_DEMAND |
1067                                 TRANSPORT_RECURSE_SUBMODULES_ONLY)) &&
1068                       !pretend)) && !is_bare_repository()) {
1069                         struct ref *ref = remote_refs;
1070                         struct string_list needs_pushing = STRING_LIST_INIT_DUP;
1071                         struct oid_array commits = OID_ARRAY_INIT;
1072
1073                         for (; ref; ref = ref->next)
1074                                 if (!is_null_oid(&ref->new_oid))
1075                                         oid_array_append(&commits,
1076                                                           &ref->new_oid);
1077
1078                         if (find_unpushed_submodules(&commits, transport->remote->name,
1079                                                 &needs_pushing)) {
1080                                 oid_array_clear(&commits);
1081                                 die_with_unpushed_submodules(&needs_pushing);
1082                         }
1083                         string_list_clear(&needs_pushing, 0);
1084                         oid_array_clear(&commits);
1085                 }
1086
1087                 if (!(flags & TRANSPORT_RECURSE_SUBMODULES_ONLY))
1088                         push_ret = transport->vtable->push_refs(transport, remote_refs, flags);
1089                 else
1090                         push_ret = 0;
1091                 err = push_had_errors(remote_refs);
1092                 ret = push_ret | err;
1093
1094                 if (!quiet || err)
1095                         transport_print_push_status(transport->url, remote_refs,
1096                                         verbose | porcelain, porcelain,
1097                                         reject_reasons);
1098
1099                 if (flags & TRANSPORT_PUSH_SET_UPSTREAM)
1100                         set_upstreams(transport, remote_refs, pretend);
1101
1102                 if (!(flags & (TRANSPORT_PUSH_DRY_RUN |
1103                                TRANSPORT_RECURSE_SUBMODULES_ONLY))) {
1104                         struct ref *ref;
1105                         for (ref = remote_refs; ref; ref = ref->next)
1106                                 transport_update_tracking_ref(transport->remote, ref, verbose);
1107                 }
1108
1109                 if (porcelain && !push_ret)
1110                         puts("Done");
1111                 else if (!quiet && !ret && !transport_refs_pushed(remote_refs))
1112                         fprintf(stderr, "Everything up-to-date\n");
1113
1114                 return ret;
1115         }
1116         return 1;
1117 }
1118
1119 const struct ref *transport_get_remote_refs(struct transport *transport)
1120 {
1121         if (!transport->got_remote_refs) {
1122                 transport->remote_refs = transport->vtable->get_refs_list(transport, 0);
1123                 transport->got_remote_refs = 1;
1124         }
1125
1126         return transport->remote_refs;
1127 }
1128
1129 int transport_fetch_refs(struct transport *transport, struct ref *refs)
1130 {
1131         int rc;
1132         int nr_heads = 0, nr_alloc = 0, nr_refs = 0;
1133         struct ref **heads = NULL;
1134         struct ref *rm;
1135
1136         for (rm = refs; rm; rm = rm->next) {
1137                 nr_refs++;
1138                 if (rm->peer_ref &&
1139                     !is_null_oid(&rm->old_oid) &&
1140                     !oidcmp(&rm->peer_ref->old_oid, &rm->old_oid))
1141                         continue;
1142                 ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
1143                 heads[nr_heads++] = rm;
1144         }
1145
1146         if (!nr_heads) {
1147                 /*
1148                  * When deepening of a shallow repository is requested,
1149                  * then local and remote refs are likely to still be equal.
1150                  * Just feed them all to the fetch method in that case.
1151                  * This condition shouldn't be met in a non-deepening fetch
1152                  * (see builtin/fetch.c:quickfetch()).
1153                  */
1154                 ALLOC_ARRAY(heads, nr_refs);
1155                 for (rm = refs; rm; rm = rm->next)
1156                         heads[nr_heads++] = rm;
1157         }
1158
1159         rc = transport->vtable->fetch(transport, nr_heads, heads);
1160
1161         free(heads);
1162         return rc;
1163 }
1164
1165 void transport_unlock_pack(struct transport *transport)
1166 {
1167         if (transport->pack_lockfile) {
1168                 unlink_or_warn(transport->pack_lockfile);
1169                 FREE_AND_NULL(transport->pack_lockfile);
1170         }
1171 }
1172
1173 int transport_connect(struct transport *transport, const char *name,
1174                       const char *exec, int fd[2])
1175 {
1176         if (transport->vtable->connect)
1177                 return transport->vtable->connect(transport, name, exec, fd);
1178         else
1179                 die("Operation not supported by protocol");
1180 }
1181
1182 int transport_disconnect(struct transport *transport)
1183 {
1184         int ret = 0;
1185         if (transport->vtable->disconnect)
1186                 ret = transport->vtable->disconnect(transport);
1187         free(transport);
1188         return ret;
1189 }
1190
1191 /*
1192  * Strip username (and password) from a URL and return
1193  * it in a newly allocated string.
1194  */
1195 char *transport_anonymize_url(const char *url)
1196 {
1197         char *scheme_prefix, *anon_part;
1198         size_t anon_len, prefix_len = 0;
1199
1200         anon_part = strchr(url, '@');
1201         if (url_is_local_not_ssh(url) || !anon_part)
1202                 goto literal_copy;
1203
1204         anon_len = strlen(++anon_part);
1205         scheme_prefix = strstr(url, "://");
1206         if (!scheme_prefix) {
1207                 if (!strchr(anon_part, ':'))
1208                         /* cannot be "me@there:/path/name" */
1209                         goto literal_copy;
1210         } else {
1211                 const char *cp;
1212                 /* make sure scheme is reasonable */
1213                 for (cp = url; cp < scheme_prefix; cp++) {
1214                         switch (*cp) {
1215                                 /* RFC 1738 2.1 */
1216                         case '+': case '.': case '-':
1217                                 break; /* ok */
1218                         default:
1219                                 if (isalnum(*cp))
1220                                         break;
1221                                 /* it isn't */
1222                                 goto literal_copy;
1223                         }
1224                 }
1225                 /* @ past the first slash does not count */
1226                 cp = strchr(scheme_prefix + 3, '/');
1227                 if (cp && cp < anon_part)
1228                         goto literal_copy;
1229                 prefix_len = scheme_prefix - url + 3;
1230         }
1231         return xstrfmt("%.*s%.*s", (int)prefix_len, url,
1232                        (int)anon_len, anon_part);
1233 literal_copy:
1234         return xstrdup(url);
1235 }
1236
1237 static void read_alternate_refs(const char *path,
1238                                 alternate_ref_fn *cb,
1239                                 void *data)
1240 {
1241         struct child_process cmd = CHILD_PROCESS_INIT;
1242         struct strbuf line = STRBUF_INIT;
1243         FILE *fh;
1244
1245         cmd.git_cmd = 1;
1246         argv_array_pushf(&cmd.args, "--git-dir=%s", path);
1247         argv_array_push(&cmd.args, "for-each-ref");
1248         argv_array_push(&cmd.args, "--format=%(objectname) %(refname)");
1249         cmd.env = local_repo_env;
1250         cmd.out = -1;
1251
1252         if (start_command(&cmd))
1253                 return;
1254
1255         fh = xfdopen(cmd.out, "r");
1256         while (strbuf_getline_lf(&line, fh) != EOF) {
1257                 struct object_id oid;
1258
1259                 if (get_oid_hex(line.buf, &oid) ||
1260                     line.buf[GIT_SHA1_HEXSZ] != ' ') {
1261                         warning("invalid line while parsing alternate refs: %s",
1262                                 line.buf);
1263                         break;
1264                 }
1265
1266                 cb(line.buf + GIT_SHA1_HEXSZ + 1, &oid, data);
1267         }
1268
1269         fclose(fh);
1270         finish_command(&cmd);
1271 }
1272
1273 struct alternate_refs_data {
1274         alternate_ref_fn *fn;
1275         void *data;
1276 };
1277
1278 static int refs_from_alternate_cb(struct alternate_object_database *e,
1279                                   void *data)
1280 {
1281         struct strbuf path = STRBUF_INIT;
1282         size_t base_len;
1283         struct alternate_refs_data *cb = data;
1284
1285         if (!strbuf_realpath(&path, e->path, 0))
1286                 goto out;
1287         if (!strbuf_strip_suffix(&path, "/objects"))
1288                 goto out;
1289         base_len = path.len;
1290
1291         /* Is this a git repository with refs? */
1292         strbuf_addstr(&path, "/refs");
1293         if (!is_directory(path.buf))
1294                 goto out;
1295         strbuf_setlen(&path, base_len);
1296
1297         read_alternate_refs(path.buf, cb->fn, cb->data);
1298
1299 out:
1300         strbuf_release(&path);
1301         return 0;
1302 }
1303
1304 void for_each_alternate_ref(alternate_ref_fn fn, void *data)
1305 {
1306         struct alternate_refs_data cb;
1307         cb.fn = fn;
1308         cb.data = data;
1309         foreach_alt_odb(refs_from_alternate_cb, &cb);
1310 }