builtin/fetch.c

   1 /*
   2  * "git fetch"
   3  */
   4 #include "cache.h"
   5 #include "refs.h"
   6 #include "commit.h"
   7 #include "builtin.h"
   8 #include "string-list.h"
   9 #include "remote.h"
  10 #include "transport.h"
  11 #include "run-command.h"
  12 #include "parse-options.h"
  13 #include "sigchain.h"
  14 #include "transport.h"
  15 #include "submodule.h"
  16
  17 static const char * const builtin_fetch_usage[] = {
  18         "git fetch [<options>] [<repository> [<refspec>...]]",
  19         "git fetch [<options>] <group>",
  20         "git fetch --multiple [<options>] [(<repository> | <group>)...]",
  21         "git fetch --all [<options>]",
  22         NULL
  23 };
  24
  25 enum {
  26         TAGS_UNSET = 0,
  27         TAGS_DEFAULT = 1,
  28         TAGS_SET = 2
  29 };
  30
  31 static int all, append, dry_run, force, keep, multiple, prune, update_head_ok, verbosity;
  32 static int progress, recurse_submodules = RECURSE_SUBMODULES_DEFAULT;
  33 static int tags = TAGS_DEFAULT;
  34 static const char *depth;
  35 static const char *upload_pack;
  36 static struct strbuf default_rla = STRBUF_INIT;
  37 static struct transport *transport;
  38 static const char *submodule_prefix = "";
  39 static const char *recurse_submodules_default;
  40
  41 static int option_parse_recurse_submodules(const struct option *opt,
  42                                    const char *arg, int unset)
  43 {
  44         if (unset) {
  45                 recurse_submodules = RECURSE_SUBMODULES_OFF;
  46         } else {
  47                 if (arg)
  48                         recurse_submodules = parse_fetch_recurse_submodules_arg(opt->long_name, arg);
  49                 else
  50                         recurse_submodules = RECURSE_SUBMODULES_ON;
  51         }
  52         return 0;
  53 }
  54
  55 static struct option builtin_fetch_options[] = {
  56         OPT__VERBOSITY(&verbosity),
  57         OPT_BOOLEAN(0, "all", &all,
  58                     "fetch from all remotes"),
  59         OPT_BOOLEAN('a', "append", &append,
  60                     "append to .git/FETCH_HEAD instead of overwriting"),
  61         OPT_STRING(0, "upload-pack", &upload_pack, "path",
  62                    "path to upload pack on remote end"),
  63         OPT__FORCE(&force, "force overwrite of local branch"),
  64         OPT_BOOLEAN('m', "multiple", &multiple,
  65                     "fetch from multiple remotes"),
  66         OPT_SET_INT('t', "tags", &tags,
  67                     "fetch all tags and associated objects", TAGS_SET),
  68         OPT_SET_INT('n', NULL, &tags,
  69                     "do not fetch all tags (--no-tags)", TAGS_UNSET),
  70         OPT_BOOLEAN('p', "prune", &prune,
  71                     "prune remote-tracking branches no longer on remote"),
  72         { OPTION_CALLBACK, 0, "recurse-submodules", NULL, "on-demand",
  73                     "control recursive fetching of submodules",
  74                     PARSE_OPT_OPTARG, option_parse_recurse_submodules },
  75         OPT_BOOLEAN(0, "dry-run", &dry_run,
  76                     "dry run"),
  77         OPT_BOOLEAN('k', "keep", &keep, "keep downloaded pack"),
  78         OPT_BOOLEAN('u', "update-head-ok", &update_head_ok,
  79                     "allow updating of HEAD ref"),
  80         OPT_BOOLEAN(0, "progress", &progress, "force progress reporting"),
  81         OPT_STRING(0, "depth", &depth, "depth",
  82                    "deepen history of shallow clone"),
  83         { OPTION_STRING, 0, "submodule-prefix", &submodule_prefix, "dir",
  84                    "prepend this to submodule path output", PARSE_OPT_HIDDEN },
  85         { OPTION_STRING, 0, "recurse-submodules-default",
  86                    &recurse_submodules_default, NULL,
  87                    "default mode for recursion", PARSE_OPT_HIDDEN },
  88         OPT_END()
  89 };
  90
  91 static void unlock_pack(void)
  92 {
  93         if (transport)
  94                 transport_unlock_pack(transport);
  95 }
  96
  97 static void unlock_pack_on_signal(int signo)
  98 {
  99         unlock_pack();
 100         sigchain_pop(signo);
 101         raise(signo);
 102 }
 103
 104 static void add_merge_config(struct ref **head,
 105                            const struct ref *remote_refs,
 106                            struct branch *branch,
 107                            struct ref ***tail)
 108 {
 109         int i;
 110
 111         for (i = 0; i < branch->merge_nr; i++) {
 112                 struct ref *rm, **old_tail = *tail;
 113                 struct refspec refspec;
 114
 115                 for (rm = *head; rm; rm = rm->next) {
 116                         if (branch_merge_matches(branch, i, rm->name)) {
 117                                 rm->merge = 1;
 118                                 break;
 119                         }
 120                 }
 121                 if (rm)
 122                         continue;
 123
 124                 /*
 125                  * Not fetched to a remote-tracking branch?  We need to fetch
 126                  * it anyway to allow this branch's "branch.$name.merge"
 127                  * to be honored by 'git pull', but we do not have to
 128                  * fail if branch.$name.merge is misconfigured to point
 129                  * at a nonexisting branch.  If we were indeed called by
 130                  * 'git pull', it will notice the misconfiguration because
 131                  * there is no entry in the resulting FETCH_HEAD marked
 132                  * for merging.
 133                  */
 134                 memset(&refspec, 0, sizeof(refspec));
 135                 refspec.src = branch->merge[i]->src;
 136                 get_fetch_map(remote_refs, &refspec, tail, 1);
 137                 for (rm = *old_tail; rm; rm = rm->next)
 138                         rm->merge = 1;
 139         }
 140 }
 141
 142 static void find_non_local_tags(struct transport *transport,
 143                         struct ref **head,
 144                         struct ref ***tail);
 145
 146 static struct ref *get_ref_map(struct transport *transport,
 147                                struct refspec *refs, int ref_count, int tags,
 148                                int *autotags)
 149 {
 150         int i;
 151         struct ref *rm;
 152         struct ref *ref_map = NULL;
 153         struct ref **tail = &ref_map;
 154
 155         const struct ref *remote_refs = transport_get_remote_refs(transport);
 156
 157         if (ref_count || tags == TAGS_SET) {
 158                 for (i = 0; i < ref_count; i++) {
 159                         get_fetch_map(remote_refs, &refs[i], &tail, 0);
 160                         if (refs[i].dst && refs[i].dst[0])
 161                                 *autotags = 1;
 162                 }
 163                 /* Merge everything on the command line, but not --tags */
 164                 for (rm = ref_map; rm; rm = rm->next)
 165                         rm->merge = 1;
 166                 if (tags == TAGS_SET)
 167                         get_fetch_map(remote_refs, tag_refspec, &tail, 0);
 168         } else {
 169                 /* Use the defaults */
 170                 struct remote *remote = transport->remote;
 171                 struct branch *branch = branch_get(NULL);
 172                 int has_merge = branch_has_merge_config(branch);
 173                 if (remote &&
 174                     (remote->fetch_refspec_nr ||
 175                      /* Note: has_merge implies non-NULL branch->remote_name */
 176                      (has_merge && !strcmp(branch->remote_name, remote->name)))) {
 177                         for (i = 0; i < remote->fetch_refspec_nr; i++) {
 178                                 get_fetch_map(remote_refs, &remote->fetch[i], &tail, 0);
 179                                 if (remote->fetch[i].dst &&
 180                                     remote->fetch[i].dst[0])
 181                                         *autotags = 1;
 182                                 if (!i && !has_merge && ref_map &&
 183                                     !remote->fetch[0].pattern)
 184                                         ref_map->merge = 1;
 185                         }
 186                         /*
 187                          * if the remote we're fetching from is the same
 188                          * as given in branch.<name>.remote, we add the
 189                          * ref given in branch.<name>.merge, too.
 190                          *
 191                          * Note: has_merge implies non-NULL branch->remote_name
 192                          */
 193                         if (has_merge &&
 194                             !strcmp(branch->remote_name, remote->name))
 195                                 add_merge_config(&ref_map, remote_refs, branch, &tail);
 196                 } else {
 197                         ref_map = get_remote_ref(remote_refs, "HEAD");
 198                         if (!ref_map)
 199                                 die(_("Couldn't find remote ref HEAD"));
 200                         ref_map->merge = 1;
 201                         tail = &ref_map->next;
 202                 }
 203         }
 204         if (tags == TAGS_DEFAULT && *autotags)
 205                 find_non_local_tags(transport, &ref_map, &tail);
 206         ref_remove_duplicates(ref_map);
 207
 208         return ref_map;
 209 }
 210
 211 #define STORE_REF_ERROR_OTHER 1
 212 #define STORE_REF_ERROR_DF_CONFLICT 2
 213
 214 static int s_update_ref(const char *action,
 215                         struct ref *ref,
 216                         int check_old)
 217 {
 218         char msg[1024];
 219         char *rla = getenv("GIT_REFLOG_ACTION");
 220         static struct ref_lock *lock;
 221
 222         if (dry_run)
 223                 return 0;
 224         if (!rla)
 225                 rla = default_rla.buf;
 226         snprintf(msg, sizeof(msg), "%s: %s", rla, action);
 227         lock = lock_any_ref_for_update(ref->name,
 228                                        check_old ? ref->old_sha1 : NULL, 0);
 229         if (!lock)
 230                 return errno == ENOTDIR ? STORE_REF_ERROR_DF_CONFLICT :
 231                                           STORE_REF_ERROR_OTHER;
 232         if (write_ref_sha1(lock, ref->new_sha1, msg) < 0)
 233                 return errno == ENOTDIR ? STORE_REF_ERROR_DF_CONFLICT :
 234                                           STORE_REF_ERROR_OTHER;
 235         return 0;
 236 }
 237
 238 #define REFCOL_WIDTH  10
 239
 240 static int update_local_ref(struct ref *ref,
 241                             const char *remote,
 242                             struct strbuf *display)
 243 {
 244         struct commit *current = NULL, *updated;
 245         enum object_type type;
 246         struct branch *current_branch = branch_get(NULL);
 247         const char *pretty_ref = prettify_refname(ref->name);
 248
 249         type = sha1_object_info(ref->new_sha1, NULL);
 250         if (type < 0)
 251                 die(_("object %s not found"), sha1_to_hex(ref->new_sha1));
 252
 253         if (!hashcmp(ref->old_sha1, ref->new_sha1)) {
 254                 if (verbosity > 0)
 255                         strbuf_addf(display, "= %-*s %-*s -> %s",
 256                                     TRANSPORT_SUMMARY_WIDTH,
 257                                     _("[up to date]"), REFCOL_WIDTH,
 258                                     remote, pretty_ref);
 259                 return 0;
 260         }
 261
 262         if (current_branch &&
 263             !strcmp(ref->name, current_branch->name) &&
 264             !(update_head_ok || is_bare_repository()) &&
 265             !is_null_sha1(ref->old_sha1)) {
 266                 /*
 267                  * If this is the head, and it's not okay to update
 268                  * the head, and the old value of the head isn't empty...
 269                  */
 270                 strbuf_addf(display,
 271                             _("! %-*s %-*s -> %s  (can't fetch in current branch)"),
 272                             TRANSPORT_SUMMARY_WIDTH, _("[rejected]"),
 273                             REFCOL_WIDTH, remote, pretty_ref);
 274                 return 1;
 275         }
 276
 277         if (!is_null_sha1(ref->old_sha1) &&
 278             !prefixcmp(ref->name, "refs/tags/")) {
 279                 int r;
 280                 r = s_update_ref("updating tag", ref, 0);
 281                 strbuf_addf(display, "%c %-*s %-*s -> %s%s",
 282                             r ? '!' : '-',
 283                             TRANSPORT_SUMMARY_WIDTH, _("[tag update]"),
 284                             REFCOL_WIDTH, remote, pretty_ref,
 285                             r ? _("  (unable to update local ref)") : "");
 286                 return r;
 287         }
 288
 289         current = lookup_commit_reference_gently(ref->old_sha1, 1);
 290         updated = lookup_commit_reference_gently(ref->new_sha1, 1);
 291         if (!current || !updated) {
 292                 const char *msg;
 293                 const char *what;
 294                 int r;
 295                 if (!strncmp(ref->name, "refs/tags/", 10)) {
 296                         msg = "storing tag";
 297                         what = _("[new tag]");
 298                 }
 299                 else {
 300                         msg = "storing head";
 301                         what = _("[new branch]");
 302                         if ((recurse_submodules != RECURSE_SUBMODULES_OFF) &&
 303                             (recurse_submodules != RECURSE_SUBMODULES_ON))
 304                                 check_for_new_submodule_commits(ref->new_sha1);
 305                 }
 306
 307                 r = s_update_ref(msg, ref, 0);
 308                 strbuf_addf(display, "%c %-*s %-*s -> %s%s",
 309                             r ? '!' : '*',
 310                             TRANSPORT_SUMMARY_WIDTH, what,
 311                             REFCOL_WIDTH, remote, pretty_ref,
 312                             r ? _("  (unable to update local ref)") : "");
 313                 return r;
 314         }
 315
 316         if (in_merge_bases(current, &updated, 1)) {
 317                 char quickref[83];
 318                 int r;
 319                 strcpy(quickref, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
 320                 strcat(quickref, "..");
 321                 strcat(quickref, find_unique_abbrev(ref->new_sha1, DEFAULT_ABBREV));
 322                 if ((recurse_submodules != RECURSE_SUBMODULES_OFF) &&
 323                     (recurse_submodules != RECURSE_SUBMODULES_ON))
 324                         check_for_new_submodule_commits(ref->new_sha1);
 325                 r = s_update_ref("fast-forward", ref, 1);
 326                 strbuf_addf(display, "%c %-*s %-*s -> %s%s",
 327                             r ? '!' : ' ',
 328                             TRANSPORT_SUMMARY_WIDTH, quickref,
 329                             REFCOL_WIDTH, remote, pretty_ref,
 330                             r ? _("  (unable to update local ref)") : "");
 331                 return r;
 332         } else if (force || ref->force) {
 333                 char quickref[84];
 334                 int r;
 335                 strcpy(quickref, find_unique_abbrev(current->object.sha1, DEFAULT_ABBREV));
 336                 strcat(quickref, "...");
 337                 strcat(quickref, find_unique_abbrev(ref->new_sha1, DEFAULT_ABBREV));
 338                 if ((recurse_submodules != RECURSE_SUBMODULES_OFF) &&
 339                     (recurse_submodules != RECURSE_SUBMODULES_ON))
 340                         check_for_new_submodule_commits(ref->new_sha1);
 341                 r = s_update_ref("forced-update", ref, 1);
 342                 strbuf_addf(display, "%c %-*s %-*s -> %s  (%s)",
 343                             r ? '!' : '+',
 344                             TRANSPORT_SUMMARY_WIDTH, quickref,
 345                             REFCOL_WIDTH, remote, pretty_ref,
 346                             r ? _("unable to update local ref") : _("forced update"));
 347                 return r;
 348         } else {
 349                 strbuf_addf(display, "! %-*s %-*s -> %s  %s",
 350                             TRANSPORT_SUMMARY_WIDTH, _("[rejected]"),
 351                             REFCOL_WIDTH, remote, pretty_ref,
 352                             _("(non-fast-forward)"));
 353                 return 1;
 354         }
 355 }
 356
 357 static int store_updated_refs(const char *raw_url, const char *remote_name,
 358                 struct ref *ref_map)
 359 {
 360         FILE *fp;
 361         struct commit *commit;
 362         int url_len, i, shown_url = 0, rc = 0;
 363         struct strbuf note = STRBUF_INIT;
 364         const char *what, *kind;
 365         struct ref *rm;
 366         char *url, *filename = dry_run ? "/dev/null" : git_path("FETCH_HEAD");
 367
 368         fp = fopen(filename, "a");
 369         if (!fp)
 370                 return error(_("cannot open %s: %s\n"), filename, strerror(errno));
 371
 372         if (raw_url)
 373                 url = transport_anonymize_url(raw_url);
 374         else
 375                 url = xstrdup("foreign");
 376         for (rm = ref_map; rm; rm = rm->next) {
 377                 struct ref *ref = NULL;
 378
 379                 if (rm->peer_ref) {
 380                         ref = xcalloc(1, sizeof(*ref) + strlen(rm->peer_ref->name) + 1);
 381                         strcpy(ref->name, rm->peer_ref->name);
 382                         hashcpy(ref->old_sha1, rm->peer_ref->old_sha1);
 383                         hashcpy(ref->new_sha1, rm->old_sha1);
 384                         ref->force = rm->peer_ref->force;
 385                 }
 386
 387                 commit = lookup_commit_reference_gently(rm->old_sha1, 1);
 388                 if (!commit)
 389                         rm->merge = 0;
 390
 391                 if (!strcmp(rm->name, "HEAD")) {
 392                         kind = "";
 393                         what = "";
 394                 }
 395                 else if (!prefixcmp(rm->name, "refs/heads/")) {
 396                         kind = "branch";
 397                         what = rm->name + 11;
 398                 }
 399                 else if (!prefixcmp(rm->name, "refs/tags/")) {
 400                         kind = "tag";
 401                         what = rm->name + 10;
 402                 }
 403                 else if (!prefixcmp(rm->name, "refs/remotes/")) {
 404                         kind = "remote-tracking branch";
 405                         what = rm->name + 13;
 406                 }
 407                 else {
 408                         kind = "";
 409                         what = rm->name;
 410                 }
 411
 412                 url_len = strlen(url);
 413                 for (i = url_len - 1; url[i] == '/' && 0 <= i; i--)
 414                         ;
 415                 url_len = i + 1;
 416                 if (4 < i && !strncmp(".git", url + i - 3, 4))
 417                         url_len = i - 3;
 418
 419                 strbuf_reset(&note);
 420                 if (*what) {
 421                         if (*kind)
 422                                 strbuf_addf(&note, "%s ", kind);
 423                         strbuf_addf(&note, "'%s' of ", what);
 424                 }
 425                 fprintf(fp, "%s\t%s\t%s",
 426                         sha1_to_hex(commit ? commit->object.sha1 :
 427                                     rm->old_sha1),
 428                         rm->merge ? "" : "not-for-merge",
 429                         note.buf);
 430                 for (i = 0; i < url_len; ++i)
 431                         if ('\n' == url[i])
 432                                 fputs("\\n", fp);
 433                         else
 434                                 fputc(url[i], fp);
 435                 fputc('\n', fp);
 436
 437                 strbuf_reset(&note);
 438                 if (ref) {
 439                         rc |= update_local_ref(ref, what, &note);
 440                         free(ref);
 441                 } else
 442                         strbuf_addf(&note, "* %-*s %-*s -> FETCH_HEAD",
 443                                     TRANSPORT_SUMMARY_WIDTH,
 444                                     *kind ? kind : "branch",
 445                                     REFCOL_WIDTH,
 446                                     *what ? what : "HEAD");
 447                 if (note.len) {
 448                         if (verbosity >= 0 && !shown_url) {
 449                                 fprintf(stderr, _("From %.*s\n"),
 450                                                 url_len, url);
 451                                 shown_url = 1;
 452                         }
 453                         if (verbosity >= 0)
 454                                 fprintf(stderr, " %s\n", note.buf);
 455                 }
 456         }
 457         free(url);
 458         fclose(fp);
 459         if (rc & STORE_REF_ERROR_DF_CONFLICT)
 460                 error(_("some local refs could not be updated; try running\n"
 461                       " 'git remote prune %s' to remove any old, conflicting "
 462                       "branches"), remote_name);
 463         strbuf_release(&note);
 464         return rc;
 465 }
 466
 467 /*
 468  * We would want to bypass the object transfer altogether if
 469  * everything we are going to fetch already exists and is connected
 470  * locally.
 471  *
 472  * The refs we are going to fetch are in ref_map.  If running
 473  *
 474  *  $ git rev-list --objects --stdin --not --all
 475  *
 476  * (feeding all the refs in ref_map on its standard input)
 477  * does not error out, that means everything reachable from the
 478  * refs we are going to fetch exists and is connected to some of
 479  * our existing refs.
 480  */
 481 static int quickfetch(struct ref *ref_map)
 482 {
 483         struct child_process revlist;
 484         struct ref *ref;
 485         int err;
 486         const char *argv[] = {"rev-list",
 487                 "--quiet", "--objects", "--stdin", "--not", "--all", NULL};
 488
 489         /*
 490          * If we are deepening a shallow clone we already have these
 491          * objects reachable.  Running rev-list here will return with
 492          * a good (0) exit status and we'll bypass the fetch that we
 493          * really need to perform.  Claiming failure now will ensure
 494          * we perform the network exchange to deepen our history.
 495          */
 496         if (depth)
 497                 return -1;
 498
 499         if (!ref_map)
 500                 return 0;
 501
 502         memset(&revlist, 0, sizeof(revlist));
 503         revlist.argv = argv;
 504         revlist.git_cmd = 1;
 505         revlist.no_stdout = 1;
 506         revlist.no_stderr = 1;
 507         revlist.in = -1;
 508
 509         err = start_command(&revlist);
 510         if (err) {
 511                 error(_("could not run rev-list"));
 512                 return err;
 513         }
 514
 515         /*
 516          * If rev-list --stdin encounters an unknown commit, it terminates,
 517          * which will cause SIGPIPE in the write loop below.
 518          */
 519         sigchain_push(SIGPIPE, SIG_IGN);
 520
 521         for (ref = ref_map; ref; ref = ref->next) {
 522                 if (write_in_full(revlist.in, sha1_to_hex(ref->old_sha1), 40) < 0 ||
 523                     write_str_in_full(revlist.in, "\n") < 0) {
 524                         if (errno != EPIPE && errno != EINVAL)
 525                                 error(_("failed write to rev-list: %s"), strerror(errno));
 526                         err = -1;
 527                         break;
 528                 }
 529         }
 530
 531         if (close(revlist.in)) {
 532                 error(_("failed to close rev-list's stdin: %s"), strerror(errno));
 533                 err = -1;
 534         }
 535
 536         sigchain_pop(SIGPIPE);
 537
 538         return finish_command(&revlist) || err;
 539 }
 540
 541 static int fetch_refs(struct transport *transport, struct ref *ref_map)
 542 {
 543         int ret = quickfetch(ref_map);
 544         if (ret)
 545                 ret = transport_fetch_refs(transport, ref_map);
 546         if (!ret)
 547                 ret |= store_updated_refs(transport->url,
 548                                 transport->remote->name,
 549                                 ref_map);
 550         transport_unlock_pack(transport);
 551         return ret;
 552 }
 553
 554 static int prune_refs(struct transport *transport, struct ref *ref_map)
 555 {
 556         int result = 0;
 557         struct ref *ref, *stale_refs = get_stale_heads(transport->remote, ref_map);
 558         const char *dangling_msg = dry_run
 559                 ? _("   (%s will become dangling)\n")
 560                 : _("   (%s has become dangling)\n");
 561
 562         for (ref = stale_refs; ref; ref = ref->next) {
 563                 if (!dry_run)
 564                         result |= delete_ref(ref->name, NULL, 0);
 565                 if (verbosity >= 0) {
 566                         fprintf(stderr, " x %-*s %-*s -> %s\n",
 567                                 TRANSPORT_SUMMARY_WIDTH, _("[deleted]"),
 568                                 REFCOL_WIDTH, _("(none)"), prettify_refname(ref->name));
 569                         warn_dangling_symref(stderr, dangling_msg, ref->name);
 570                 }
 571         }
 572         free_refs(stale_refs);
 573         return result;
 574 }
 575
 576 static int add_existing(const char *refname, const unsigned char *sha1,
 577                         int flag, void *cbdata)
 578 {
 579         struct string_list *list = (struct string_list *)cbdata;
 580         struct string_list_item *item = string_list_insert(list, refname);
 581         item->util = (void *)sha1;
 582         return 0;
 583 }
 584
 585 static int will_fetch(struct ref **head, const unsigned char *sha1)
 586 {
 587         struct ref *rm = *head;
 588         while (rm) {
 589                 if (!hashcmp(rm->old_sha1, sha1))
 590                         return 1;
 591                 rm = rm->next;
 592         }
 593         return 0;
 594 }
 595
 596 static void find_non_local_tags(struct transport *transport,
 597                         struct ref **head,
 598                         struct ref ***tail)
 599 {
 600         struct string_list existing_refs = STRING_LIST_INIT_NODUP;
 601         struct string_list remote_refs = STRING_LIST_INIT_NODUP;
 602         const struct ref *ref;
 603         struct string_list_item *item = NULL;
 604
 605         for_each_ref(add_existing, &existing_refs);
 606         for (ref = transport_get_remote_refs(transport); ref; ref = ref->next) {
 607                 if (prefixcmp(ref->name, "refs/tags"))
 608                         continue;
 609
 610                 /*
 611                  * The peeled ref always follows the matching base
 612                  * ref, so if we see a peeled ref that we don't want
 613                  * to fetch then we can mark the ref entry in the list
 614                  * as one to ignore by setting util to NULL.
 615                  */
 616                 if (!suffixcmp(ref->name, "^{}")) {
 617                         if (item && !has_sha1_file(ref->old_sha1) &&
 618                             !will_fetch(head, ref->old_sha1) &&
 619                             !has_sha1_file(item->util) &&
 620                             !will_fetch(head, item->util))
 621                                 item->util = NULL;
 622                         item = NULL;
 623                         continue;
 624                 }
 625
 626                 /*
 627                  * If item is non-NULL here, then we previously saw a
 628                  * ref not followed by a peeled reference, so we need
 629                  * to check if it is a lightweight tag that we want to
 630                  * fetch.
 631                  */
 632                 if (item && !has_sha1_file(item->util) &&
 633                     !will_fetch(head, item->util))
 634                         item->util = NULL;
 635
 636                 item = NULL;
 637
 638                 /* skip duplicates and refs that we already have */
 639                 if (string_list_has_string(&remote_refs, ref->name) ||
 640                     string_list_has_string(&existing_refs, ref->name))
 641                         continue;
 642
 643                 item = string_list_insert(&remote_refs, ref->name);
 644                 item->util = (void *)ref->old_sha1;
 645         }
 646         string_list_clear(&existing_refs, 0);
 647
 648         /*
 649          * We may have a final lightweight tag that needs to be
 650          * checked to see if it needs fetching.
 651          */
 652         if (item && !has_sha1_file(item->util) &&
 653             !will_fetch(head, item->util))
 654                 item->util = NULL;
 655
 656         /*
 657          * For all the tags in the remote_refs string list,
 658          * add them to the list of refs to be fetched
 659          */
 660         for_each_string_list_item(item, &remote_refs) {
 661                 /* Unless we have already decided to ignore this item... */
 662                 if (item->util)
 663                 {
 664                         struct ref *rm = alloc_ref(item->string);
 665                         rm->peer_ref = alloc_ref(item->string);
 666                         hashcpy(rm->old_sha1, item->util);
 667                         **tail = rm;
 668                         *tail = &rm->next;
 669                 }
 670         }
 671
 672         string_list_clear(&remote_refs, 0);
 673 }
 674
 675 static void check_not_current_branch(struct ref *ref_map)
 676 {
 677         struct branch *current_branch = branch_get(NULL);
 678
 679         if (is_bare_repository() || !current_branch)
 680                 return;
 681
 682         for (; ref_map; ref_map = ref_map->next)
 683                 if (ref_map->peer_ref && !strcmp(current_branch->refname,
 684                                         ref_map->peer_ref->name))
 685                         die(_("Refusing to fetch into current branch %s "
 686                             "of non-bare repository"), current_branch->refname);
 687 }
 688
 689 static int truncate_fetch_head(void)
 690 {
 691         char *filename = git_path("FETCH_HEAD");
 692         FILE *fp = fopen(filename, "w");
 693
 694         if (!fp)
 695                 return error(_("cannot open %s: %s\n"), filename, strerror(errno));
 696         fclose(fp);
 697         return 0;
 698 }
 699
 700 static int do_fetch(struct transport *transport,
 701                     struct refspec *refs, int ref_count)
 702 {
 703         struct string_list existing_refs = STRING_LIST_INIT_NODUP;
 704         struct string_list_item *peer_item = NULL;
 705         struct ref *ref_map;
 706         struct ref *rm;
 707         int autotags = (transport->remote->fetch_tags == 1);
 708
 709         for_each_ref(add_existing, &existing_refs);
 710
 711         if (tags == TAGS_DEFAULT) {
 712                 if (transport->remote->fetch_tags == 2)
 713                         tags = TAGS_SET;
 714                 if (transport->remote->fetch_tags == -1)
 715                         tags = TAGS_UNSET;
 716         }
 717
 718         if (!transport->get_refs_list || !transport->fetch)
 719                 die(_("Don't know how to fetch from %s"), transport->url);
 720
 721         /* if not appending, truncate FETCH_HEAD */
 722         if (!append && !dry_run) {
 723                 int errcode = truncate_fetch_head();
 724                 if (errcode)
 725                         return errcode;
 726         }
 727
 728         ref_map = get_ref_map(transport, refs, ref_count, tags, &autotags);
 729         if (!update_head_ok)
 730                 check_not_current_branch(ref_map);
 731
 732         for (rm = ref_map; rm; rm = rm->next) {
 733                 if (rm->peer_ref) {
 734                         peer_item = string_list_lookup(&existing_refs,
 735                                                        rm->peer_ref->name);
 736                         if (peer_item)
 737                                 hashcpy(rm->peer_ref->old_sha1,
 738                                         peer_item->util);
 739                 }
 740         }
 741
 742         if (tags == TAGS_DEFAULT && autotags)
 743                 transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, "1");
 744         if (fetch_refs(transport, ref_map)) {
 745                 free_refs(ref_map);
 746                 return 1;
 747         }
 748         if (prune)
 749                 prune_refs(transport, ref_map);
 750         free_refs(ref_map);
 751
 752         /* if neither --no-tags nor --tags was specified, do automated tag
 753          * following ... */
 754         if (tags == TAGS_DEFAULT && autotags) {
 755                 struct ref **tail = &ref_map;
 756                 ref_map = NULL;
 757                 find_non_local_tags(transport, &ref_map, &tail);
 758                 if (ref_map) {
 759                         transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, NULL);
 760                         transport_set_option(transport, TRANS_OPT_DEPTH, "0");
 761                         fetch_refs(transport, ref_map);
 762                 }
 763                 free_refs(ref_map);
 764         }
 765
 766         return 0;
 767 }
 768
 769 static void set_option(const char *name, const char *value)
 770 {
 771         int r = transport_set_option(transport, name, value);
 772         if (r < 0)
 773                 die(_("Option \"%s\" value \"%s\" is not valid for %s"),
 774                         name, value, transport->url);
 775         if (r > 0)
 776                 warning(_("Option \"%s\" is ignored for %s\n"),
 777                         name, transport->url);
 778 }
 779
 780 static int get_one_remote_for_fetch(struct remote *remote, void *priv)
 781 {
 782         struct string_list *list = priv;
 783         if (!remote->skip_default_update)
 784                 string_list_append(list, remote->name);
 785         return 0;
 786 }
 787
 788 struct remote_group_data {
 789         const char *name;
 790         struct string_list *list;
 791 };
 792
 793 static int get_remote_group(const char *key, const char *value, void *priv)
 794 {
 795         struct remote_group_data *g = priv;
 796
 797         if (!prefixcmp(key, "remotes.") &&
 798                         !strcmp(key + 8, g->name)) {
 799                 /* split list by white space */
 800                 int space = strcspn(value, " \t\n");
 801                 while (*value) {
 802                         if (space > 1) {
 803                                 string_list_append(g->list,
 804                                                    xstrndup(value, space));
 805                         }
 806                         value += space + (value[space] != '\0');
 807                         space = strcspn(value, " \t\n");
 808                 }
 809         }
 810
 811         return 0;
 812 }
 813
 814 static int add_remote_or_group(const char *name, struct string_list *list)
 815 {
 816         int prev_nr = list->nr;
 817         struct remote_group_data g;
 818         g.name = name; g.list = list;
 819
 820         git_config(get_remote_group, &g);
 821         if (list->nr == prev_nr) {
 822                 struct remote *remote;
 823                 if (!remote_is_configured(name))
 824                         return 0;
 825                 remote = remote_get(name);
 826                 string_list_append(list, remote->name);
 827         }
 828         return 1;
 829 }
 830
 831 static void add_options_to_argv(int *argc, const char **argv)
 832 {
 833         if (dry_run)
 834                 argv[(*argc)++] = "--dry-run";
 835         if (prune)
 836                 argv[(*argc)++] = "--prune";
 837         if (update_head_ok)
 838                 argv[(*argc)++] = "--update-head-ok";
 839         if (force)
 840                 argv[(*argc)++] = "--force";
 841         if (keep)
 842                 argv[(*argc)++] = "--keep";
 843         if (recurse_submodules == RECURSE_SUBMODULES_ON)
 844                 argv[(*argc)++] = "--recurse-submodules";
 845         else if (recurse_submodules == RECURSE_SUBMODULES_ON_DEMAND)
 846                 argv[(*argc)++] = "--recurse-submodules=on-demand";
 847         if (verbosity >= 2)
 848                 argv[(*argc)++] = "-v";
 849         if (verbosity >= 1)
 850                 argv[(*argc)++] = "-v";
 851         else if (verbosity < 0)
 852                 argv[(*argc)++] = "-q";
 853
 854 }
 855
 856 static int fetch_multiple(struct string_list *list)
 857 {
 858         int i, result = 0;
 859         const char *argv[12] = { "fetch", "--append" };
 860         int argc = 2;
 861
 862         add_options_to_argv(&argc, argv);
 863
 864         if (!append && !dry_run) {
 865                 int errcode = truncate_fetch_head();
 866                 if (errcode)
 867                         return errcode;
 868         }
 869
 870         for (i = 0; i < list->nr; i++) {
 871                 const char *name = list->items[i].string;
 872                 argv[argc] = name;
 873                 argv[argc + 1] = NULL;
 874                 if (verbosity >= 0)
 875                         printf(_("Fetching %s\n"), name);
 876                 if (run_command_v_opt(argv, RUN_GIT_CMD)) {
 877                         error(_("Could not fetch %s"), name);
 878                         result = 1;
 879                 }
 880         }
 881
 882         return result;
 883 }
 884
 885 static int fetch_one(struct remote *remote, int argc, const char **argv)
 886 {
 887         int i;
 888         static const char **refs = NULL;
 889         struct refspec *refspec;
 890         int ref_nr = 0;
 891         int exit_code;
 892
 893         if (!remote)
 894                 die(_("No remote repository specified.  Please, specify either a URL or a\n"
 895                     "remote name from which new revisions should be fetched."));
 896
 897         transport = transport_get(remote, NULL);
 898         transport_set_verbosity(transport, verbosity, progress);
 899         if (upload_pack)
 900                 set_option(TRANS_OPT_UPLOADPACK, upload_pack);
 901         if (keep)
 902                 set_option(TRANS_OPT_KEEP, "yes");
 903         if (depth)
 904                 set_option(TRANS_OPT_DEPTH, depth);
 905
 906         if (argc > 0) {
 907                 int j = 0;
 908                 refs = xcalloc(argc + 1, sizeof(const char *));
 909                 for (i = 0; i < argc; i++) {
 910                         if (!strcmp(argv[i], "tag")) {
 911                                 char *ref;
 912                                 i++;
 913                                 if (i >= argc)
 914                                         die(_("You need to specify a tag name."));
 915                                 ref = xmalloc(strlen(argv[i]) * 2 + 22);
 916                                 strcpy(ref, "refs/tags/");
 917                                 strcat(ref, argv[i]);
 918                                 strcat(ref, ":refs/tags/");
 919                                 strcat(ref, argv[i]);
 920                                 refs[j++] = ref;
 921                         } else
 922                                 refs[j++] = argv[i];
 923                 }
 924                 refs[j] = NULL;
 925                 ref_nr = j;
 926         }
 927
 928         sigchain_push_common(unlock_pack_on_signal);
 929         atexit(unlock_pack);
 930         refspec = parse_fetch_refspec(ref_nr, refs);
 931         exit_code = do_fetch(transport, refspec, ref_nr);
 932         free(refspec);
 933         transport_disconnect(transport);
 934         transport = NULL;
 935         return exit_code;
 936 }
 937
 938 int cmd_fetch(int argc, const char **argv, const char *prefix)
 939 {
 940         int i;
 941         struct string_list list = STRING_LIST_INIT_NODUP;
 942         struct remote *remote;
 943         int result = 0;
 944
 945         packet_trace_identity("fetch");
 946
 947         /* Record the command line for the reflog */
 948         strbuf_addstr(&default_rla, "fetch");
 949         for (i = 1; i < argc; i++)
 950                 strbuf_addf(&default_rla, " %s", argv[i]);
 951
 952         argc = parse_options(argc, argv, prefix,
 953                              builtin_fetch_options, builtin_fetch_usage, 0);
 954
 955         if (recurse_submodules != RECURSE_SUBMODULES_OFF) {
 956                 if (recurse_submodules_default) {
 957                         int arg = parse_fetch_recurse_submodules_arg("--recurse-submodules-default", recurse_submodules_default);
 958                         set_config_fetch_recurse_submodules(arg);
 959                 }
 960                 gitmodules_config();
 961                 git_config(submodule_config, NULL);
 962         }
 963
 964         if (all) {
 965                 if (argc == 1)
 966                         die(_("fetch --all does not take a repository argument"));
 967                 else if (argc > 1)
 968                         die(_("fetch --all does not make sense with refspecs"));
 969                 (void) for_each_remote(get_one_remote_for_fetch, &list);
 970                 result = fetch_multiple(&list);
 971         } else if (argc == 0) {
 972                 /* No arguments -- use default remote */
 973                 remote = remote_get(NULL);
 974                 result = fetch_one(remote, argc, argv);
 975         } else if (multiple) {
 976                 /* All arguments are assumed to be remotes or groups */
 977                 for (i = 0; i < argc; i++)
 978                         if (!add_remote_or_group(argv[i], &list))
 979                                 die(_("No such remote or remote group: %s"), argv[i]);
 980                 result = fetch_multiple(&list);
 981         } else {
 982                 /* Single remote or group */
 983                 (void) add_remote_or_group(argv[0], &list);
 984                 if (list.nr > 1) {
 985                         /* More than one remote */
 986                         if (argc > 1)
 987                                 die(_("Fetching a group and specifying refspecs does not make sense"));
 988                         result = fetch_multiple(&list);
 989                 } else {
 990                         /* Zero or one remotes */
 991                         remote = remote_get(argv[0]);
 992                         result = fetch_one(remote, argc-1, argv+1);
 993                 }
 994         }
 995
 996         if (!result && (recurse_submodules != RECURSE_SUBMODULES_OFF)) {
 997                 const char *options[10];
 998                 int num_options = 0;
 999                 add_options_to_argv(&num_options, options);
1000                 result = fetch_populated_submodules(num_options, options,
1001                                                     submodule_prefix,
1002                                                     recurse_submodules,
1003                                                     verbosity < 0);
1004         }
1005
1006         /* All names were strdup()ed or strndup()ed */
1007         list.strdup_strings = 1;
1008         string_list_clear(&list, 0);
1009
1010         return result;
1011 }