git.oblomov.eu Git - git/blob - transport.c

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