builtin / fetch.con commit Merge branch 'rs/sort-oid-array-thread-safe' (2743b61)
   1/*
   2 * "git fetch"
   3 */
   4#include "cache.h"
   5#include "config.h"
   6#include "repository.h"
   7#include "refs.h"
   8#include "refspec.h"
   9#include "object-store.h"
  10#include "commit.h"
  11#include "builtin.h"
  12#include "string-list.h"
  13#include "remote.h"
  14#include "transport.h"
  15#include "run-command.h"
  16#include "parse-options.h"
  17#include "sigchain.h"
  18#include "submodule-config.h"
  19#include "submodule.h"
  20#include "connected.h"
  21#include "argv-array.h"
  22#include "utf8.h"
  23#include "packfile.h"
  24#include "list-objects-filter-options.h"
  25#include "commit-reach.h"
  26#include "branch.h"
  27
  28#define FORCED_UPDATES_DELAY_WARNING_IN_MS (10 * 1000)
  29
  30static const char * const builtin_fetch_usage[] = {
  31        N_("git fetch [<options>] [<repository> [<refspec>...]]"),
  32        N_("git fetch [<options>] <group>"),
  33        N_("git fetch --multiple [<options>] [(<repository> | <group>)...]"),
  34        N_("git fetch --all [<options>]"),
  35        NULL
  36};
  37
  38enum {
  39        TAGS_UNSET = 0,
  40        TAGS_DEFAULT = 1,
  41        TAGS_SET = 2
  42};
  43
  44static int fetch_prune_config = -1; /* unspecified */
  45static int fetch_show_forced_updates = 1;
  46static uint64_t forced_updates_ms = 0;
  47static int prune = -1; /* unspecified */
  48#define PRUNE_BY_DEFAULT 0 /* do we prune by default? */
  49
  50static int fetch_prune_tags_config = -1; /* unspecified */
  51static int prune_tags = -1; /* unspecified */
  52#define PRUNE_TAGS_BY_DEFAULT 0 /* do we prune tags by default? */
  53
  54static int all, append, dry_run, force, keep, multiple, update_head_ok;
  55static int verbosity, deepen_relative, set_upstream;
  56static int progress = -1;
  57static int enable_auto_gc = 1;
  58static int tags = TAGS_DEFAULT, unshallow, update_shallow, deepen;
  59static int max_children = 1;
  60static enum transport_family family;
  61static const char *depth;
  62static const char *deepen_since;
  63static const char *upload_pack;
  64static struct string_list deepen_not = STRING_LIST_INIT_NODUP;
  65static struct strbuf default_rla = STRBUF_INIT;
  66static struct transport *gtransport;
  67static struct transport *gsecondary;
  68static const char *submodule_prefix = "";
  69static int recurse_submodules = RECURSE_SUBMODULES_DEFAULT;
  70static int recurse_submodules_default = RECURSE_SUBMODULES_ON_DEMAND;
  71static int shown_url = 0;
  72static struct refspec refmap = REFSPEC_INIT_FETCH;
  73static struct list_objects_filter_options filter_options;
  74static struct string_list server_options = STRING_LIST_INIT_DUP;
  75static struct string_list negotiation_tip = STRING_LIST_INIT_NODUP;
  76
  77static int git_fetch_config(const char *k, const char *v, void *cb)
  78{
  79        if (!strcmp(k, "fetch.prune")) {
  80                fetch_prune_config = git_config_bool(k, v);
  81                return 0;
  82        }
  83
  84        if (!strcmp(k, "fetch.prunetags")) {
  85                fetch_prune_tags_config = git_config_bool(k, v);
  86                return 0;
  87        }
  88
  89        if (!strcmp(k, "fetch.showforcedupdates")) {
  90                fetch_show_forced_updates = git_config_bool(k, v);
  91                return 0;
  92        }
  93
  94        if (!strcmp(k, "submodule.recurse")) {
  95                int r = git_config_bool(k, v) ?
  96                        RECURSE_SUBMODULES_ON : RECURSE_SUBMODULES_OFF;
  97                recurse_submodules = r;
  98        }
  99
 100        if (!strcmp(k, "submodule.fetchjobs")) {
 101                max_children = parse_submodule_fetchjobs(k, v);
 102                return 0;
 103        } else if (!strcmp(k, "fetch.recursesubmodules")) {
 104                recurse_submodules = parse_fetch_recurse_submodules_arg(k, v);
 105                return 0;
 106        }
 107
 108        return git_default_config(k, v, cb);
 109}
 110
 111static int parse_refmap_arg(const struct option *opt, const char *arg, int unset)
 112{
 113        BUG_ON_OPT_NEG(unset);
 114
 115        /*
 116         * "git fetch --refmap='' origin foo"
 117         * can be used to tell the command not to store anywhere
 118         */
 119        refspec_append(&refmap, arg);
 120
 121        return 0;
 122}
 123
 124static struct option builtin_fetch_options[] = {
 125        OPT__VERBOSITY(&verbosity),
 126        OPT_BOOL(0, "all", &all,
 127                 N_("fetch from all remotes")),
 128        OPT_BOOL(0, "set-upstream", &set_upstream,
 129                 N_("set upstream for git pull/fetch")),
 130        OPT_BOOL('a', "append", &append,
 131                 N_("append to .git/FETCH_HEAD instead of overwriting")),
 132        OPT_STRING(0, "upload-pack", &upload_pack, N_("path"),
 133                   N_("path to upload pack on remote end")),
 134        OPT__FORCE(&force, N_("force overwrite of local reference"), 0),
 135        OPT_BOOL('m', "multiple", &multiple,
 136                 N_("fetch from multiple remotes")),
 137        OPT_SET_INT('t', "tags", &tags,
 138                    N_("fetch all tags and associated objects"), TAGS_SET),
 139        OPT_SET_INT('n', NULL, &tags,
 140                    N_("do not fetch all tags (--no-tags)"), TAGS_UNSET),
 141        OPT_INTEGER('j', "jobs", &max_children,
 142                    N_("number of submodules fetched in parallel")),
 143        OPT_BOOL('p', "prune", &prune,
 144                 N_("prune remote-tracking branches no longer on remote")),
 145        OPT_BOOL('P', "prune-tags", &prune_tags,
 146                 N_("prune local tags no longer on remote and clobber changed tags")),
 147        { OPTION_CALLBACK, 0, "recurse-submodules", &recurse_submodules, N_("on-demand"),
 148                    N_("control recursive fetching of submodules"),
 149                    PARSE_OPT_OPTARG, option_fetch_parse_recurse_submodules },
 150        OPT_BOOL(0, "dry-run", &dry_run,
 151                 N_("dry run")),
 152        OPT_BOOL('k', "keep", &keep, N_("keep downloaded pack")),
 153        OPT_BOOL('u', "update-head-ok", &update_head_ok,
 154                    N_("allow updating of HEAD ref")),
 155        OPT_BOOL(0, "progress", &progress, N_("force progress reporting")),
 156        OPT_STRING(0, "depth", &depth, N_("depth"),
 157                   N_("deepen history of shallow clone")),
 158        OPT_STRING(0, "shallow-since", &deepen_since, N_("time"),
 159                   N_("deepen history of shallow repository based on time")),
 160        OPT_STRING_LIST(0, "shallow-exclude", &deepen_not, N_("revision"),
 161                        N_("deepen history of shallow clone, excluding rev")),
 162        OPT_INTEGER(0, "deepen", &deepen_relative,
 163                    N_("deepen history of shallow clone")),
 164        OPT_SET_INT_F(0, "unshallow", &unshallow,
 165                      N_("convert to a complete repository"),
 166                      1, PARSE_OPT_NONEG),
 167        { OPTION_STRING, 0, "submodule-prefix", &submodule_prefix, N_("dir"),
 168                   N_("prepend this to submodule path output"), PARSE_OPT_HIDDEN },
 169        { OPTION_CALLBACK, 0, "recurse-submodules-default",
 170                   &recurse_submodules_default, N_("on-demand"),
 171                   N_("default for recursive fetching of submodules "
 172                      "(lower priority than config files)"),
 173                   PARSE_OPT_HIDDEN, option_fetch_parse_recurse_submodules },
 174        OPT_BOOL(0, "update-shallow", &update_shallow,
 175                 N_("accept refs that update .git/shallow")),
 176        { OPTION_CALLBACK, 0, "refmap", NULL, N_("refmap"),
 177          N_("specify fetch refmap"), PARSE_OPT_NONEG, parse_refmap_arg },
 178        OPT_STRING_LIST('o', "server-option", &server_options, N_("server-specific"), N_("option to transmit")),
 179        OPT_SET_INT('4', "ipv4", &family, N_("use IPv4 addresses only"),
 180                        TRANSPORT_FAMILY_IPV4),
 181        OPT_SET_INT('6', "ipv6", &family, N_("use IPv6 addresses only"),
 182                        TRANSPORT_FAMILY_IPV6),
 183        OPT_STRING_LIST(0, "negotiation-tip", &negotiation_tip, N_("revision"),
 184                        N_("report that we have only objects reachable from this object")),
 185        OPT_PARSE_LIST_OBJECTS_FILTER(&filter_options),
 186        OPT_BOOL(0, "auto-gc", &enable_auto_gc,
 187                 N_("run 'gc --auto' after fetching")),
 188        OPT_BOOL(0, "show-forced-updates", &fetch_show_forced_updates,
 189                 N_("check for forced-updates on all updated branches")),
 190        OPT_END()
 191};
 192
 193static void unlock_pack(void)
 194{
 195        if (gtransport)
 196                transport_unlock_pack(gtransport);
 197        if (gsecondary)
 198                transport_unlock_pack(gsecondary);
 199}
 200
 201static void unlock_pack_on_signal(int signo)
 202{
 203        unlock_pack();
 204        sigchain_pop(signo);
 205        raise(signo);
 206}
 207
 208static void add_merge_config(struct ref **head,
 209                           const struct ref *remote_refs,
 210                           struct branch *branch,
 211                           struct ref ***tail)
 212{
 213        int i;
 214
 215        for (i = 0; i < branch->merge_nr; i++) {
 216                struct ref *rm, **old_tail = *tail;
 217                struct refspec_item refspec;
 218
 219                for (rm = *head; rm; rm = rm->next) {
 220                        if (branch_merge_matches(branch, i, rm->name)) {
 221                                rm->fetch_head_status = FETCH_HEAD_MERGE;
 222                                break;
 223                        }
 224                }
 225                if (rm)
 226                        continue;
 227
 228                /*
 229                 * Not fetched to a remote-tracking branch?  We need to fetch
 230                 * it anyway to allow this branch's "branch.$name.merge"
 231                 * to be honored by 'git pull', but we do not have to
 232                 * fail if branch.$name.merge is misconfigured to point
 233                 * at a nonexisting branch.  If we were indeed called by
 234                 * 'git pull', it will notice the misconfiguration because
 235                 * there is no entry in the resulting FETCH_HEAD marked
 236                 * for merging.
 237                 */
 238                memset(&refspec, 0, sizeof(refspec));
 239                refspec.src = branch->merge[i]->src;
 240                get_fetch_map(remote_refs, &refspec, tail, 1);
 241                for (rm = *old_tail; rm; rm = rm->next)
 242                        rm->fetch_head_status = FETCH_HEAD_MERGE;
 243        }
 244}
 245
 246static int will_fetch(struct ref **head, const unsigned char *sha1)
 247{
 248        struct ref *rm = *head;
 249        while (rm) {
 250                if (hasheq(rm->old_oid.hash, sha1))
 251                        return 1;
 252                rm = rm->next;
 253        }
 254        return 0;
 255}
 256
 257struct refname_hash_entry {
 258        struct hashmap_entry ent; /* must be the first member */
 259        struct object_id oid;
 260        int ignore;
 261        char refname[FLEX_ARRAY];
 262};
 263
 264static int refname_hash_entry_cmp(const void *hashmap_cmp_fn_data,
 265                                  const void *e1_,
 266                                  const void *e2_,
 267                                  const void *keydata)
 268{
 269        const struct refname_hash_entry *e1 = e1_;
 270        const struct refname_hash_entry *e2 = e2_;
 271
 272        return strcmp(e1->refname, keydata ? keydata : e2->refname);
 273}
 274
 275static struct refname_hash_entry *refname_hash_add(struct hashmap *map,
 276                                                   const char *refname,
 277                                                   const struct object_id *oid)
 278{
 279        struct refname_hash_entry *ent;
 280        size_t len = strlen(refname);
 281
 282        FLEX_ALLOC_MEM(ent, refname, refname, len);
 283        hashmap_entry_init(ent, strhash(refname));
 284        oidcpy(&ent->oid, oid);
 285        hashmap_add(map, ent);
 286        return ent;
 287}
 288
 289static int add_one_refname(const char *refname,
 290                           const struct object_id *oid,
 291                           int flag, void *cbdata)
 292{
 293        struct hashmap *refname_map = cbdata;
 294
 295        (void) refname_hash_add(refname_map, refname, oid);
 296        return 0;
 297}
 298
 299static void refname_hash_init(struct hashmap *map)
 300{
 301        hashmap_init(map, refname_hash_entry_cmp, NULL, 0);
 302}
 303
 304static int refname_hash_exists(struct hashmap *map, const char *refname)
 305{
 306        return !!hashmap_get_from_hash(map, strhash(refname), refname);
 307}
 308
 309static void clear_item(struct refname_hash_entry *item)
 310{
 311        item->ignore = 1;
 312}
 313
 314static void find_non_local_tags(const struct ref *refs,
 315                                struct ref **head,
 316                                struct ref ***tail)
 317{
 318        struct hashmap existing_refs;
 319        struct hashmap remote_refs;
 320        struct string_list remote_refs_list = STRING_LIST_INIT_NODUP;
 321        struct string_list_item *remote_ref_item;
 322        const struct ref *ref;
 323        struct refname_hash_entry *item = NULL;
 324
 325        refname_hash_init(&existing_refs);
 326        refname_hash_init(&remote_refs);
 327
 328        for_each_ref(add_one_refname, &existing_refs);
 329        for (ref = refs; ref; ref = ref->next) {
 330                if (!starts_with(ref->name, "refs/tags/"))
 331                        continue;
 332
 333                /*
 334                 * The peeled ref always follows the matching base
 335                 * ref, so if we see a peeled ref that we don't want
 336                 * to fetch then we can mark the ref entry in the list
 337                 * as one to ignore by setting util to NULL.
 338                 */
 339                if (ends_with(ref->name, "^{}")) {
 340                        if (item &&
 341                            !has_object_file_with_flags(&ref->old_oid,
 342                                                        OBJECT_INFO_QUICK) &&
 343                            !will_fetch(head, ref->old_oid.hash) &&
 344                            !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
 345                            !will_fetch(head, item->oid.hash))
 346                                clear_item(item);
 347                        item = NULL;
 348                        continue;
 349                }
 350
 351                /*
 352                 * If item is non-NULL here, then we previously saw a
 353                 * ref not followed by a peeled reference, so we need
 354                 * to check if it is a lightweight tag that we want to
 355                 * fetch.
 356                 */
 357                if (item &&
 358                    !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
 359                    !will_fetch(head, item->oid.hash))
 360                        clear_item(item);
 361
 362                item = NULL;
 363
 364                /* skip duplicates and refs that we already have */
 365                if (refname_hash_exists(&remote_refs, ref->name) ||
 366                    refname_hash_exists(&existing_refs, ref->name))
 367                        continue;
 368
 369                item = refname_hash_add(&remote_refs, ref->name, &ref->old_oid);
 370                string_list_insert(&remote_refs_list, ref->name);
 371        }
 372        hashmap_free(&existing_refs, 1);
 373
 374        /*
 375         * We may have a final lightweight tag that needs to be
 376         * checked to see if it needs fetching.
 377         */
 378        if (item &&
 379            !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
 380            !will_fetch(head, item->oid.hash))
 381                clear_item(item);
 382
 383        /*
 384         * For all the tags in the remote_refs_list,
 385         * add them to the list of refs to be fetched
 386         */
 387        for_each_string_list_item(remote_ref_item, &remote_refs_list) {
 388                const char *refname = remote_ref_item->string;
 389                struct ref *rm;
 390
 391                item = hashmap_get_from_hash(&remote_refs, strhash(refname), refname);
 392                if (!item)
 393                        BUG("unseen remote ref?");
 394
 395                /* Unless we have already decided to ignore this item... */
 396                if (item->ignore)
 397                        continue;
 398
 399                rm = alloc_ref(item->refname);
 400                rm->peer_ref = alloc_ref(item->refname);
 401                oidcpy(&rm->old_oid, &item->oid);
 402                **tail = rm;
 403                *tail = &rm->next;
 404        }
 405        hashmap_free(&remote_refs, 1);
 406        string_list_clear(&remote_refs_list, 0);
 407}
 408
 409static struct ref *get_ref_map(struct remote *remote,
 410                               const struct ref *remote_refs,
 411                               struct refspec *rs,
 412                               int tags, int *autotags)
 413{
 414        int i;
 415        struct ref *rm;
 416        struct ref *ref_map = NULL;
 417        struct ref **tail = &ref_map;
 418
 419        /* opportunistically-updated references: */
 420        struct ref *orefs = NULL, **oref_tail = &orefs;
 421
 422        struct hashmap existing_refs;
 423
 424        if (rs->nr) {
 425                struct refspec *fetch_refspec;
 426
 427                for (i = 0; i < rs->nr; i++) {
 428                        get_fetch_map(remote_refs, &rs->items[i], &tail, 0);
 429                        if (rs->items[i].dst && rs->items[i].dst[0])
 430                                *autotags = 1;
 431                }
 432                /* Merge everything on the command line (but not --tags) */
 433                for (rm = ref_map; rm; rm = rm->next)
 434                        rm->fetch_head_status = FETCH_HEAD_MERGE;
 435
 436                /*
 437                 * For any refs that we happen to be fetching via
 438                 * command-line arguments, the destination ref might
 439                 * have been missing or have been different than the
 440                 * remote-tracking ref that would be derived from the
 441                 * configured refspec.  In these cases, we want to
 442                 * take the opportunity to update their configured
 443                 * remote-tracking reference.  However, we do not want
 444                 * to mention these entries in FETCH_HEAD at all, as
 445                 * they would simply be duplicates of existing
 446                 * entries, so we set them FETCH_HEAD_IGNORE below.
 447                 *
 448                 * We compute these entries now, based only on the
 449                 * refspecs specified on the command line.  But we add
 450                 * them to the list following the refspecs resulting
 451                 * from the tags option so that one of the latter,
 452                 * which has FETCH_HEAD_NOT_FOR_MERGE, is not removed
 453                 * by ref_remove_duplicates() in favor of one of these
 454                 * opportunistic entries with FETCH_HEAD_IGNORE.
 455                 */
 456                if (refmap.nr)
 457                        fetch_refspec = &refmap;
 458                else
 459                        fetch_refspec = &remote->fetch;
 460
 461                for (i = 0; i < fetch_refspec->nr; i++)
 462                        get_fetch_map(ref_map, &fetch_refspec->items[i], &oref_tail, 1);
 463        } else if (refmap.nr) {
 464                die("--refmap option is only meaningful with command-line refspec(s).");
 465        } else {
 466                /* Use the defaults */
 467                struct branch *branch = branch_get(NULL);
 468                int has_merge = branch_has_merge_config(branch);
 469                if (remote &&
 470                    (remote->fetch.nr ||
 471                     /* Note: has_merge implies non-NULL branch->remote_name */
 472                     (has_merge && !strcmp(branch->remote_name, remote->name)))) {
 473                        for (i = 0; i < remote->fetch.nr; i++) {
 474                                get_fetch_map(remote_refs, &remote->fetch.items[i], &tail, 0);
 475                                if (remote->fetch.items[i].dst &&
 476                                    remote->fetch.items[i].dst[0])
 477                                        *autotags = 1;
 478                                if (!i && !has_merge && ref_map &&
 479                                    !remote->fetch.items[0].pattern)
 480                                        ref_map->fetch_head_status = FETCH_HEAD_MERGE;
 481                        }
 482                        /*
 483                         * if the remote we're fetching from is the same
 484                         * as given in branch.<name>.remote, we add the
 485                         * ref given in branch.<name>.merge, too.
 486                         *
 487                         * Note: has_merge implies non-NULL branch->remote_name
 488                         */
 489                        if (has_merge &&
 490                            !strcmp(branch->remote_name, remote->name))
 491                                add_merge_config(&ref_map, remote_refs, branch, &tail);
 492                } else {
 493                        ref_map = get_remote_ref(remote_refs, "HEAD");
 494                        if (!ref_map)
 495                                die(_("Couldn't find remote ref HEAD"));
 496                        ref_map->fetch_head_status = FETCH_HEAD_MERGE;
 497                        tail = &ref_map->next;
 498                }
 499        }
 500
 501        if (tags == TAGS_SET)
 502                /* also fetch all tags */
 503                get_fetch_map(remote_refs, tag_refspec, &tail, 0);
 504        else if (tags == TAGS_DEFAULT && *autotags)
 505                find_non_local_tags(remote_refs, &ref_map, &tail);
 506
 507        /* Now append any refs to be updated opportunistically: */
 508        *tail = orefs;
 509        for (rm = orefs; rm; rm = rm->next) {
 510                rm->fetch_head_status = FETCH_HEAD_IGNORE;
 511                tail = &rm->next;
 512        }
 513
 514        ref_map = ref_remove_duplicates(ref_map);
 515
 516        refname_hash_init(&existing_refs);
 517        for_each_ref(add_one_refname, &existing_refs);
 518
 519        for (rm = ref_map; rm; rm = rm->next) {
 520                if (rm->peer_ref) {
 521                        const char *refname = rm->peer_ref->name;
 522                        struct refname_hash_entry *peer_item;
 523
 524                        peer_item = hashmap_get_from_hash(&existing_refs,
 525                                                          strhash(refname),
 526                                                          refname);
 527                        if (peer_item) {
 528                                struct object_id *old_oid = &peer_item->oid;
 529                                oidcpy(&rm->peer_ref->old_oid, old_oid);
 530                        }
 531                }
 532        }
 533        hashmap_free(&existing_refs, 1);
 534
 535        return ref_map;
 536}
 537
 538#define STORE_REF_ERROR_OTHER 1
 539#define STORE_REF_ERROR_DF_CONFLICT 2
 540
 541static int s_update_ref(const char *action,
 542                        struct ref *ref,
 543                        int check_old)
 544{
 545        char *msg;
 546        char *rla = getenv("GIT_REFLOG_ACTION");
 547        struct ref_transaction *transaction;
 548        struct strbuf err = STRBUF_INIT;
 549        int ret, df_conflict = 0;
 550
 551        if (dry_run)
 552                return 0;
 553        if (!rla)
 554                rla = default_rla.buf;
 555        msg = xstrfmt("%s: %s", rla, action);
 556
 557        transaction = ref_transaction_begin(&err);
 558        if (!transaction ||
 559            ref_transaction_update(transaction, ref->name,
 560                                   &ref->new_oid,
 561                                   check_old ? &ref->old_oid : NULL,
 562                                   0, msg, &err))
 563                goto fail;
 564
 565        ret = ref_transaction_commit(transaction, &err);
 566        if (ret) {
 567                df_conflict = (ret == TRANSACTION_NAME_CONFLICT);
 568                goto fail;
 569        }
 570
 571        ref_transaction_free(transaction);
 572        strbuf_release(&err);
 573        free(msg);
 574        return 0;
 575fail:
 576        ref_transaction_free(transaction);
 577        error("%s", err.buf);
 578        strbuf_release(&err);
 579        free(msg);
 580        return df_conflict ? STORE_REF_ERROR_DF_CONFLICT
 581                           : STORE_REF_ERROR_OTHER;
 582}
 583
 584static int refcol_width = 10;
 585static int compact_format;
 586
 587static void adjust_refcol_width(const struct ref *ref)
 588{
 589        int max, rlen, llen, len;
 590
 591        /* uptodate lines are only shown on high verbosity level */
 592        if (!verbosity && oideq(&ref->peer_ref->old_oid, &ref->old_oid))
 593                return;
 594
 595        max    = term_columns();
 596        rlen   = utf8_strwidth(prettify_refname(ref->name));
 597
 598        llen   = utf8_strwidth(prettify_refname(ref->peer_ref->name));
 599
 600        /*
 601         * rough estimation to see if the output line is too long and
 602         * should not be counted (we can't do precise calculation
 603         * anyway because we don't know if the error explanation part
 604         * will be printed in update_local_ref)
 605         */
 606        if (compact_format) {
 607                llen = 0;
 608                max = max * 2 / 3;
 609        }
 610        len = 21 /* flag and summary */ + rlen + 4 /* -> */ + llen;
 611        if (len >= max)
 612                return;
 613
 614        /*
 615         * Not precise calculation for compact mode because '*' can
 616         * appear on the left hand side of '->' and shrink the column
 617         * back.
 618         */
 619        if (refcol_width < rlen)
 620                refcol_width = rlen;
 621}
 622
 623static void prepare_format_display(struct ref *ref_map)
 624{
 625        struct ref *rm;
 626        const char *format = "full";
 627
 628        git_config_get_string_const("fetch.output", &format);
 629        if (!strcasecmp(format, "full"))
 630                compact_format = 0;
 631        else if (!strcasecmp(format, "compact"))
 632                compact_format = 1;
 633        else
 634                die(_("configuration fetch.output contains invalid value %s"),
 635                    format);
 636
 637        for (rm = ref_map; rm; rm = rm->next) {
 638                if (rm->status == REF_STATUS_REJECT_SHALLOW ||
 639                    !rm->peer_ref ||
 640                    !strcmp(rm->name, "HEAD"))
 641                        continue;
 642
 643                adjust_refcol_width(rm);
 644        }
 645}
 646
 647static void print_remote_to_local(struct strbuf *display,
 648                                  const char *remote, const char *local)
 649{
 650        strbuf_addf(display, "%-*s -> %s", refcol_width, remote, local);
 651}
 652
 653static int find_and_replace(struct strbuf *haystack,
 654                            const char *needle,
 655                            const char *placeholder)
 656{
 657        const char *p = NULL;
 658        int plen, nlen;
 659
 660        nlen = strlen(needle);
 661        if (ends_with(haystack->buf, needle))
 662                p = haystack->buf + haystack->len - nlen;
 663        else
 664                p = strstr(haystack->buf, needle);
 665        if (!p)
 666                return 0;
 667
 668        if (p > haystack->buf && p[-1] != '/')
 669                return 0;
 670
 671        plen = strlen(p);
 672        if (plen > nlen && p[nlen] != '/')
 673                return 0;
 674
 675        strbuf_splice(haystack, p - haystack->buf, nlen,
 676                      placeholder, strlen(placeholder));
 677        return 1;
 678}
 679
 680static void print_compact(struct strbuf *display,
 681                          const char *remote, const char *local)
 682{
 683        struct strbuf r = STRBUF_INIT;
 684        struct strbuf l = STRBUF_INIT;
 685
 686        if (!strcmp(remote, local)) {
 687                strbuf_addf(display, "%-*s -> *", refcol_width, remote);
 688                return;
 689        }
 690
 691        strbuf_addstr(&r, remote);
 692        strbuf_addstr(&l, local);
 693
 694        if (!find_and_replace(&r, local, "*"))
 695                find_and_replace(&l, remote, "*");
 696        print_remote_to_local(display, r.buf, l.buf);
 697
 698        strbuf_release(&r);
 699        strbuf_release(&l);
 700}
 701
 702static void format_display(struct strbuf *display, char code,
 703                           const char *summary, const char *error,
 704                           const char *remote, const char *local,
 705                           int summary_width)
 706{
 707        int width = (summary_width + strlen(summary) - gettext_width(summary));
 708
 709        strbuf_addf(display, "%c %-*s ", code, width, summary);
 710        if (!compact_format)
 711                print_remote_to_local(display, remote, local);
 712        else
 713                print_compact(display, remote, local);
 714        if (error)
 715                strbuf_addf(display, "  (%s)", error);
 716}
 717
 718static int update_local_ref(struct ref *ref,
 719                            const char *remote,
 720                            const struct ref *remote_ref,
 721                            struct strbuf *display,
 722                            int summary_width)
 723{
 724        struct commit *current = NULL, *updated;
 725        enum object_type type;
 726        struct branch *current_branch = branch_get(NULL);
 727        const char *pretty_ref = prettify_refname(ref->name);
 728        int fast_forward = 0;
 729
 730        type = oid_object_info(the_repository, &ref->new_oid, NULL);
 731        if (type < 0)
 732                die(_("object %s not found"), oid_to_hex(&ref->new_oid));
 733
 734        if (oideq(&ref->old_oid, &ref->new_oid)) {
 735                if (verbosity > 0)
 736                        format_display(display, '=', _("[up to date]"), NULL,
 737                                       remote, pretty_ref, summary_width);
 738                return 0;
 739        }
 740
 741        if (current_branch &&
 742            !strcmp(ref->name, current_branch->name) &&
 743            !(update_head_ok || is_bare_repository()) &&
 744            !is_null_oid(&ref->old_oid)) {
 745                /*
 746                 * If this is the head, and it's not okay to update
 747                 * the head, and the old value of the head isn't empty...
 748                 */
 749                format_display(display, '!', _("[rejected]"),
 750                               _("can't fetch in current branch"),
 751                               remote, pretty_ref, summary_width);
 752                return 1;
 753        }
 754
 755        if (!is_null_oid(&ref->old_oid) &&
 756            starts_with(ref->name, "refs/tags/")) {
 757                if (force || ref->force) {
 758                        int r;
 759                        r = s_update_ref("updating tag", ref, 0);
 760                        format_display(display, r ? '!' : 't', _("[tag update]"),
 761                                       r ? _("unable to update local ref") : NULL,
 762                                       remote, pretty_ref, summary_width);
 763                        return r;
 764                } else {
 765                        format_display(display, '!', _("[rejected]"), _("would clobber existing tag"),
 766                                       remote, pretty_ref, summary_width);
 767                        return 1;
 768                }
 769        }
 770
 771        current = lookup_commit_reference_gently(the_repository,
 772                                                 &ref->old_oid, 1);
 773        updated = lookup_commit_reference_gently(the_repository,
 774                                                 &ref->new_oid, 1);
 775        if (!current || !updated) {
 776                const char *msg;
 777                const char *what;
 778                int r;
 779                /*
 780                 * Nicely describe the new ref we're fetching.
 781                 * Base this on the remote's ref name, as it's
 782                 * more likely to follow a standard layout.
 783                 */
 784                const char *name = remote_ref ? remote_ref->name : "";
 785                if (starts_with(name, "refs/tags/")) {
 786                        msg = "storing tag";
 787                        what = _("[new tag]");
 788                } else if (starts_with(name, "refs/heads/")) {
 789                        msg = "storing head";
 790                        what = _("[new branch]");
 791                } else {
 792                        msg = "storing ref";
 793                        what = _("[new ref]");
 794                }
 795
 796                r = s_update_ref(msg, ref, 0);
 797                format_display(display, r ? '!' : '*', what,
 798                               r ? _("unable to update local ref") : NULL,
 799                               remote, pretty_ref, summary_width);
 800                return r;
 801        }
 802
 803        if (fetch_show_forced_updates) {
 804                uint64_t t_before = getnanotime();
 805                fast_forward = in_merge_bases(current, updated);
 806                forced_updates_ms += (getnanotime() - t_before) / 1000000;
 807        } else {
 808                fast_forward = 1;
 809        }
 810
 811        if (fast_forward) {
 812                struct strbuf quickref = STRBUF_INIT;
 813                int r;
 814
 815                strbuf_add_unique_abbrev(&quickref, &current->object.oid, DEFAULT_ABBREV);
 816                strbuf_addstr(&quickref, "..");
 817                strbuf_add_unique_abbrev(&quickref, &ref->new_oid, DEFAULT_ABBREV);
 818                r = s_update_ref("fast-forward", ref, 1);
 819                format_display(display, r ? '!' : ' ', quickref.buf,
 820                               r ? _("unable to update local ref") : NULL,
 821                               remote, pretty_ref, summary_width);
 822                strbuf_release(&quickref);
 823                return r;
 824        } else if (force || ref->force) {
 825                struct strbuf quickref = STRBUF_INIT;
 826                int r;
 827                strbuf_add_unique_abbrev(&quickref, &current->object.oid, DEFAULT_ABBREV);
 828                strbuf_addstr(&quickref, "...");
 829                strbuf_add_unique_abbrev(&quickref, &ref->new_oid, DEFAULT_ABBREV);
 830                r = s_update_ref("forced-update", ref, 1);
 831                format_display(display, r ? '!' : '+', quickref.buf,
 832                               r ? _("unable to update local ref") : _("forced update"),
 833                               remote, pretty_ref, summary_width);
 834                strbuf_release(&quickref);
 835                return r;
 836        } else {
 837                format_display(display, '!', _("[rejected]"), _("non-fast-forward"),
 838                               remote, pretty_ref, summary_width);
 839                return 1;
 840        }
 841}
 842
 843static int iterate_ref_map(void *cb_data, struct object_id *oid)
 844{
 845        struct ref **rm = cb_data;
 846        struct ref *ref = *rm;
 847
 848        while (ref && ref->status == REF_STATUS_REJECT_SHALLOW)
 849                ref = ref->next;
 850        if (!ref)
 851                return -1; /* end of the list */
 852        *rm = ref->next;
 853        oidcpy(oid, &ref->old_oid);
 854        return 0;
 855}
 856
 857static const char warn_show_forced_updates[] =
 858N_("Fetch normally indicates which branches had a forced update,\n"
 859   "but that check has been disabled. To re-enable, use '--show-forced-updates'\n"
 860   "flag or run 'git config fetch.showForcedUpdates true'.");
 861static const char warn_time_show_forced_updates[] =
 862N_("It took %.2f seconds to check forced updates. You can use\n"
 863   "'--no-show-forced-updates' or run 'git config fetch.showForcedUpdates false'\n"
 864   " to avoid this check.\n");
 865
 866static int store_updated_refs(const char *raw_url, const char *remote_name,
 867                              int connectivity_checked, struct ref *ref_map)
 868{
 869        FILE *fp;
 870        struct commit *commit;
 871        int url_len, i, rc = 0;
 872        struct strbuf note = STRBUF_INIT;
 873        const char *what, *kind;
 874        struct ref *rm;
 875        char *url;
 876        const char *filename = dry_run ? "/dev/null" : git_path_fetch_head(the_repository);
 877        int want_status;
 878        int summary_width = transport_summary_width(ref_map);
 879
 880        fp = fopen(filename, "a");
 881        if (!fp)
 882                return error_errno(_("cannot open %s"), filename);
 883
 884        if (raw_url)
 885                url = transport_anonymize_url(raw_url);
 886        else
 887                url = xstrdup("foreign");
 888
 889        if (!connectivity_checked) {
 890                rm = ref_map;
 891                if (check_connected(iterate_ref_map, &rm, NULL)) {
 892                        rc = error(_("%s did not send all necessary objects\n"), url);
 893                        goto abort;
 894                }
 895        }
 896
 897        prepare_format_display(ref_map);
 898
 899        /*
 900         * We do a pass for each fetch_head_status type in their enum order, so
 901         * merged entries are written before not-for-merge. That lets readers
 902         * use FETCH_HEAD as a refname to refer to the ref to be merged.
 903         */
 904        for (want_status = FETCH_HEAD_MERGE;
 905             want_status <= FETCH_HEAD_IGNORE;
 906             want_status++) {
 907                for (rm = ref_map; rm; rm = rm->next) {
 908                        struct ref *ref = NULL;
 909                        const char *merge_status_marker = "";
 910
 911                        if (rm->status == REF_STATUS_REJECT_SHALLOW) {
 912                                if (want_status == FETCH_HEAD_MERGE)
 913                                        warning(_("reject %s because shallow roots are not allowed to be updated"),
 914                                                rm->peer_ref ? rm->peer_ref->name : rm->name);
 915                                continue;
 916                        }
 917
 918                        commit = lookup_commit_reference_gently(the_repository,
 919                                                                &rm->old_oid,
 920                                                                1);
 921                        if (!commit)
 922                                rm->fetch_head_status = FETCH_HEAD_NOT_FOR_MERGE;
 923
 924                        if (rm->fetch_head_status != want_status)
 925                                continue;
 926
 927                        if (rm->peer_ref) {
 928                                ref = alloc_ref(rm->peer_ref->name);
 929                                oidcpy(&ref->old_oid, &rm->peer_ref->old_oid);
 930                                oidcpy(&ref->new_oid, &rm->old_oid);
 931                                ref->force = rm->peer_ref->force;
 932                        }
 933
 934                        if (recurse_submodules != RECURSE_SUBMODULES_OFF)
 935                                check_for_new_submodule_commits(&rm->old_oid);
 936
 937                        if (!strcmp(rm->name, "HEAD")) {
 938                                kind = "";
 939                                what = "";
 940                        }
 941                        else if (starts_with(rm->name, "refs/heads/")) {
 942                                kind = "branch";
 943                                what = rm->name + 11;
 944                        }
 945                        else if (starts_with(rm->name, "refs/tags/")) {
 946                                kind = "tag";
 947                                what = rm->name + 10;
 948                        }
 949                        else if (starts_with(rm->name, "refs/remotes/")) {
 950                                kind = "remote-tracking branch";
 951                                what = rm->name + 13;
 952                        }
 953                        else {
 954                                kind = "";
 955                                what = rm->name;
 956                        }
 957
 958                        url_len = strlen(url);
 959                        for (i = url_len - 1; url[i] == '/' && 0 <= i; i--)
 960                                ;
 961                        url_len = i + 1;
 962                        if (4 < i && !strncmp(".git", url + i - 3, 4))
 963                                url_len = i - 3;
 964
 965                        strbuf_reset(&note);
 966                        if (*what) {
 967                                if (*kind)
 968                                        strbuf_addf(&note, "%s ", kind);
 969                                strbuf_addf(&note, "'%s' of ", what);
 970                        }
 971                        switch (rm->fetch_head_status) {
 972                        case FETCH_HEAD_NOT_FOR_MERGE:
 973                                merge_status_marker = "not-for-merge";
 974                                /* fall-through */
 975                        case FETCH_HEAD_MERGE:
 976                                fprintf(fp, "%s\t%s\t%s",
 977                                        oid_to_hex(&rm->old_oid),
 978                                        merge_status_marker,
 979                                        note.buf);
 980                                for (i = 0; i < url_len; ++i)
 981                                        if ('\n' == url[i])
 982                                                fputs("\\n", fp);
 983                                        else
 984                                                fputc(url[i], fp);
 985                                fputc('\n', fp);
 986                                break;
 987                        default:
 988                                /* do not write anything to FETCH_HEAD */
 989                                break;
 990                        }
 991
 992                        strbuf_reset(&note);
 993                        if (ref) {
 994                                rc |= update_local_ref(ref, what, rm, &note,
 995                                                       summary_width);
 996                                free(ref);
 997                        } else
 998                                format_display(&note, '*',
 999                                               *kind ? kind : "branch", NULL,
1000                                               *what ? what : "HEAD",
1001                                               "FETCH_HEAD", summary_width);
1002                        if (note.len) {
1003                                if (verbosity >= 0 && !shown_url) {
1004                                        fprintf(stderr, _("From %.*s\n"),
1005                                                        url_len, url);
1006                                        shown_url = 1;
1007                                }
1008                                if (verbosity >= 0)
1009                                        fprintf(stderr, " %s\n", note.buf);
1010                        }
1011                }
1012        }
1013
1014        if (rc & STORE_REF_ERROR_DF_CONFLICT)
1015                error(_("some local refs could not be updated; try running\n"
1016                      " 'git remote prune %s' to remove any old, conflicting "
1017                      "branches"), remote_name);
1018
1019        if (advice_fetch_show_forced_updates) {
1020                if (!fetch_show_forced_updates) {
1021                        warning(_(warn_show_forced_updates));
1022                } else if (forced_updates_ms > FORCED_UPDATES_DELAY_WARNING_IN_MS) {
1023                        warning(_(warn_time_show_forced_updates),
1024                                forced_updates_ms / 1000.0);
1025                }
1026        }
1027
1028 abort:
1029        strbuf_release(&note);
1030        free(url);
1031        fclose(fp);
1032        return rc;
1033}
1034
1035/*
1036 * We would want to bypass the object transfer altogether if
1037 * everything we are going to fetch already exists and is connected
1038 * locally.
1039 */
1040static int check_exist_and_connected(struct ref *ref_map)
1041{
1042        struct ref *rm = ref_map;
1043        struct check_connected_options opt = CHECK_CONNECTED_INIT;
1044        struct ref *r;
1045
1046        /*
1047         * If we are deepening a shallow clone we already have these
1048         * objects reachable.  Running rev-list here will return with
1049         * a good (0) exit status and we'll bypass the fetch that we
1050         * really need to perform.  Claiming failure now will ensure
1051         * we perform the network exchange to deepen our history.
1052         */
1053        if (deepen)
1054                return -1;
1055
1056        /*
1057         * check_connected() allows objects to merely be promised, but
1058         * we need all direct targets to exist.
1059         */
1060        for (r = rm; r; r = r->next) {
1061                if (!has_object_file(&r->old_oid))
1062                        return -1;
1063        }
1064
1065        opt.quiet = 1;
1066        return check_connected(iterate_ref_map, &rm, &opt);
1067}
1068
1069static int fetch_refs(struct transport *transport, struct ref *ref_map)
1070{
1071        int ret = check_exist_and_connected(ref_map);
1072        if (ret)
1073                ret = transport_fetch_refs(transport, ref_map);
1074        if (!ret)
1075                /*
1076                 * Keep the new pack's ".keep" file around to allow the caller
1077                 * time to update refs to reference the new objects.
1078                 */
1079                return 0;
1080        transport_unlock_pack(transport);
1081        return ret;
1082}
1083
1084/* Update local refs based on the ref values fetched from a remote */
1085static int consume_refs(struct transport *transport, struct ref *ref_map)
1086{
1087        int connectivity_checked = transport->smart_options
1088                ? transport->smart_options->connectivity_checked : 0;
1089        int ret = store_updated_refs(transport->url,
1090                                     transport->remote->name,
1091                                     connectivity_checked,
1092                                     ref_map);
1093        transport_unlock_pack(transport);
1094        return ret;
1095}
1096
1097static int prune_refs(struct refspec *rs, struct ref *ref_map,
1098                      const char *raw_url)
1099{
1100        int url_len, i, result = 0;
1101        struct ref *ref, *stale_refs = get_stale_heads(rs, ref_map);
1102        char *url;
1103        int summary_width = transport_summary_width(stale_refs);
1104        const char *dangling_msg = dry_run
1105                ? _("   (%s will become dangling)")
1106                : _("   (%s has become dangling)");
1107
1108        if (raw_url)
1109                url = transport_anonymize_url(raw_url);
1110        else
1111                url = xstrdup("foreign");
1112
1113        url_len = strlen(url);
1114        for (i = url_len - 1; url[i] == '/' && 0 <= i; i--)
1115                ;
1116
1117        url_len = i + 1;
1118        if (4 < i && !strncmp(".git", url + i - 3, 4))
1119                url_len = i - 3;
1120
1121        if (!dry_run) {
1122                struct string_list refnames = STRING_LIST_INIT_NODUP;
1123
1124                for (ref = stale_refs; ref; ref = ref->next)
1125                        string_list_append(&refnames, ref->name);
1126
1127                result = delete_refs("fetch: prune", &refnames, 0);
1128                string_list_clear(&refnames, 0);
1129        }
1130
1131        if (verbosity >= 0) {
1132                for (ref = stale_refs; ref; ref = ref->next) {
1133                        struct strbuf sb = STRBUF_INIT;
1134                        if (!shown_url) {
1135                                fprintf(stderr, _("From %.*s\n"), url_len, url);
1136                                shown_url = 1;
1137                        }
1138                        format_display(&sb, '-', _("[deleted]"), NULL,
1139                                       _("(none)"), prettify_refname(ref->name),
1140                                       summary_width);
1141                        fprintf(stderr, " %s\n",sb.buf);
1142                        strbuf_release(&sb);
1143                        warn_dangling_symref(stderr, dangling_msg, ref->name);
1144                }
1145        }
1146
1147        free(url);
1148        free_refs(stale_refs);
1149        return result;
1150}
1151
1152static void check_not_current_branch(struct ref *ref_map)
1153{
1154        struct branch *current_branch = branch_get(NULL);
1155
1156        if (is_bare_repository() || !current_branch)
1157                return;
1158
1159        for (; ref_map; ref_map = ref_map->next)
1160                if (ref_map->peer_ref && !strcmp(current_branch->refname,
1161                                        ref_map->peer_ref->name))
1162                        die(_("Refusing to fetch into current branch %s "
1163                            "of non-bare repository"), current_branch->refname);
1164}
1165
1166static int truncate_fetch_head(void)
1167{
1168        const char *filename = git_path_fetch_head(the_repository);
1169        FILE *fp = fopen_for_writing(filename);
1170
1171        if (!fp)
1172                return error_errno(_("cannot open %s"), filename);
1173        fclose(fp);
1174        return 0;
1175}
1176
1177static void set_option(struct transport *transport, const char *name, const char *value)
1178{
1179        int r = transport_set_option(transport, name, value);
1180        if (r < 0)
1181                die(_("Option \"%s\" value \"%s\" is not valid for %s"),
1182                    name, value, transport->url);
1183        if (r > 0)
1184                warning(_("Option \"%s\" is ignored for %s\n"),
1185                        name, transport->url);
1186}
1187
1188
1189static int add_oid(const char *refname, const struct object_id *oid, int flags,
1190                   void *cb_data)
1191{
1192        struct oid_array *oids = cb_data;
1193
1194        oid_array_append(oids, oid);
1195        return 0;
1196}
1197
1198static void add_negotiation_tips(struct git_transport_options *smart_options)
1199{
1200        struct oid_array *oids = xcalloc(1, sizeof(*oids));
1201        int i;
1202
1203        for (i = 0; i < negotiation_tip.nr; i++) {
1204                const char *s = negotiation_tip.items[i].string;
1205                int old_nr;
1206                if (!has_glob_specials(s)) {
1207                        struct object_id oid;
1208                        if (get_oid(s, &oid))
1209                                die("%s is not a valid object", s);
1210                        oid_array_append(oids, &oid);
1211                        continue;
1212                }
1213                old_nr = oids->nr;
1214                for_each_glob_ref(add_oid, s, oids);
1215                if (old_nr == oids->nr)
1216                        warning("Ignoring --negotiation-tip=%s because it does not match any refs",
1217                                s);
1218        }
1219        smart_options->negotiation_tips = oids;
1220}
1221
1222static struct transport *prepare_transport(struct remote *remote, int deepen)
1223{
1224        struct transport *transport;
1225
1226        transport = transport_get(remote, NULL);
1227        transport_set_verbosity(transport, verbosity, progress);
1228        transport->family = family;
1229        if (upload_pack)
1230                set_option(transport, TRANS_OPT_UPLOADPACK, upload_pack);
1231        if (keep)
1232                set_option(transport, TRANS_OPT_KEEP, "yes");
1233        if (depth)
1234                set_option(transport, TRANS_OPT_DEPTH, depth);
1235        if (deepen && deepen_since)
1236                set_option(transport, TRANS_OPT_DEEPEN_SINCE, deepen_since);
1237        if (deepen && deepen_not.nr)
1238                set_option(transport, TRANS_OPT_DEEPEN_NOT,
1239                           (const char *)&deepen_not);
1240        if (deepen_relative)
1241                set_option(transport, TRANS_OPT_DEEPEN_RELATIVE, "yes");
1242        if (update_shallow)
1243                set_option(transport, TRANS_OPT_UPDATE_SHALLOW, "yes");
1244        if (filter_options.choice) {
1245                struct strbuf expanded_filter_spec = STRBUF_INIT;
1246                expand_list_objects_filter_spec(&filter_options,
1247                                                &expanded_filter_spec);
1248                set_option(transport, TRANS_OPT_LIST_OBJECTS_FILTER,
1249                           expanded_filter_spec.buf);
1250                set_option(transport, TRANS_OPT_FROM_PROMISOR, "1");
1251                strbuf_release(&expanded_filter_spec);
1252        }
1253        if (negotiation_tip.nr) {
1254                if (transport->smart_options)
1255                        add_negotiation_tips(transport->smart_options);
1256                else
1257                        warning("Ignoring --negotiation-tip because the protocol does not support it.");
1258        }
1259        return transport;
1260}
1261
1262static void backfill_tags(struct transport *transport, struct ref *ref_map)
1263{
1264        int cannot_reuse;
1265
1266        /*
1267         * Once we have set TRANS_OPT_DEEPEN_SINCE, we can't unset it
1268         * when remote helper is used (setting it to an empty string
1269         * is not unsetting). We could extend the remote helper
1270         * protocol for that, but for now, just force a new connection
1271         * without deepen-since. Similar story for deepen-not.
1272         */
1273        cannot_reuse = transport->cannot_reuse ||
1274                deepen_since || deepen_not.nr;
1275        if (cannot_reuse) {
1276                gsecondary = prepare_transport(transport->remote, 0);
1277                transport = gsecondary;
1278        }
1279
1280        transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, NULL);
1281        transport_set_option(transport, TRANS_OPT_DEPTH, "0");
1282        transport_set_option(transport, TRANS_OPT_DEEPEN_RELATIVE, NULL);
1283        if (!fetch_refs(transport, ref_map))
1284                consume_refs(transport, ref_map);
1285
1286        if (gsecondary) {
1287                transport_disconnect(gsecondary);
1288                gsecondary = NULL;
1289        }
1290}
1291
1292static int do_fetch(struct transport *transport,
1293                    struct refspec *rs)
1294{
1295        struct ref *ref_map;
1296        int autotags = (transport->remote->fetch_tags == 1);
1297        int retcode = 0;
1298        const struct ref *remote_refs;
1299        struct argv_array ref_prefixes = ARGV_ARRAY_INIT;
1300        int must_list_refs = 1;
1301
1302        if (tags == TAGS_DEFAULT) {
1303                if (transport->remote->fetch_tags == 2)
1304                        tags = TAGS_SET;
1305                if (transport->remote->fetch_tags == -1)
1306                        tags = TAGS_UNSET;
1307        }
1308
1309        /* if not appending, truncate FETCH_HEAD */
1310        if (!append && !dry_run) {
1311                retcode = truncate_fetch_head();
1312                if (retcode)
1313                        goto cleanup;
1314        }
1315
1316        if (rs->nr) {
1317                int i;
1318
1319                refspec_ref_prefixes(rs, &ref_prefixes);
1320
1321                /*
1322                 * We can avoid listing refs if all of them are exact
1323                 * OIDs
1324                 */
1325                must_list_refs = 0;
1326                for (i = 0; i < rs->nr; i++) {
1327                        if (!rs->items[i].exact_sha1) {
1328                                must_list_refs = 1;
1329                                break;
1330                        }
1331                }
1332        } else if (transport->remote && transport->remote->fetch.nr)
1333                refspec_ref_prefixes(&transport->remote->fetch, &ref_prefixes);
1334
1335        if (tags == TAGS_SET || tags == TAGS_DEFAULT) {
1336                must_list_refs = 1;
1337                if (ref_prefixes.argc)
1338                        argv_array_push(&ref_prefixes, "refs/tags/");
1339        }
1340
1341        if (must_list_refs)
1342                remote_refs = transport_get_remote_refs(transport, &ref_prefixes);
1343        else
1344                remote_refs = NULL;
1345
1346        argv_array_clear(&ref_prefixes);
1347
1348        ref_map = get_ref_map(transport->remote, remote_refs, rs,
1349                              tags, &autotags);
1350        if (!update_head_ok)
1351                check_not_current_branch(ref_map);
1352
1353        if (tags == TAGS_DEFAULT && autotags)
1354                transport_set_option(transport, TRANS_OPT_FOLLOWTAGS, "1");
1355        if (prune) {
1356                /*
1357                 * We only prune based on refspecs specified
1358                 * explicitly (via command line or configuration); we
1359                 * don't care whether --tags was specified.
1360                 */
1361                if (rs->nr) {
1362                        prune_refs(rs, ref_map, transport->url);
1363                } else {
1364                        prune_refs(&transport->remote->fetch,
1365                                   ref_map,
1366                                   transport->url);
1367                }
1368        }
1369        if (fetch_refs(transport, ref_map) || consume_refs(transport, ref_map)) {
1370                free_refs(ref_map);
1371                retcode = 1;
1372                goto cleanup;
1373        }
1374
1375        if (set_upstream) {
1376                struct branch *branch = branch_get("HEAD");
1377                struct ref *rm;
1378                struct ref *source_ref = NULL;
1379
1380                /*
1381                 * We're setting the upstream configuration for the
1382                 * current branch. The relevent upstream is the
1383                 * fetched branch that is meant to be merged with the
1384                 * current one, i.e. the one fetched to FETCH_HEAD.
1385                 *
1386                 * When there are several such branches, consider the
1387                 * request ambiguous and err on the safe side by doing
1388                 * nothing and just emit a warning.
1389                 */
1390                for (rm = ref_map; rm; rm = rm->next) {
1391                        if (!rm->peer_ref) {
1392                                if (source_ref) {
1393                                        warning(_("multiple branch detected, incompatible with --set-upstream"));
1394                                        goto skip;
1395                                } else {
1396                                        source_ref = rm;
1397                                }
1398                        }
1399                }
1400                if (source_ref) {
1401                        if (!strcmp(source_ref->name, "HEAD") ||
1402                            starts_with(source_ref->name, "refs/heads/"))
1403                                install_branch_config(0,
1404                                                      branch->name,
1405                                                      transport->remote->name,
1406                                                      source_ref->name);
1407                        else if (starts_with(source_ref->name, "refs/remotes/"))
1408                                warning(_("not setting upstream for a remote remote-tracking branch"));
1409                        else if (starts_with(source_ref->name, "refs/tags/"))
1410                                warning(_("not setting upstream for a remote tag"));
1411                        else
1412                                warning(_("unknown branch type"));
1413                } else {
1414                        warning(_("no source branch found.\n"
1415                                "you need to specify exactly one branch with the --set-upstream option."));
1416                }
1417        }
1418 skip:
1419        free_refs(ref_map);
1420
1421        /* if neither --no-tags nor --tags was specified, do automated tag
1422         * following ... */
1423        if (tags == TAGS_DEFAULT && autotags) {
1424                struct ref **tail = &ref_map;
1425                ref_map = NULL;
1426                find_non_local_tags(remote_refs, &ref_map, &tail);
1427                if (ref_map)
1428                        backfill_tags(transport, ref_map);
1429                free_refs(ref_map);
1430        }
1431
1432 cleanup:
1433        return retcode;
1434}
1435
1436static int get_one_remote_for_fetch(struct remote *remote, void *priv)
1437{
1438        struct string_list *list = priv;
1439        if (!remote->skip_default_update)
1440                string_list_append(list, remote->name);
1441        return 0;
1442}
1443
1444struct remote_group_data {
1445        const char *name;
1446        struct string_list *list;
1447};
1448
1449static int get_remote_group(const char *key, const char *value, void *priv)
1450{
1451        struct remote_group_data *g = priv;
1452
1453        if (skip_prefix(key, "remotes.", &key) && !strcmp(key, g->name)) {
1454                /* split list by white space */
1455                while (*value) {
1456                        size_t wordlen = strcspn(value, " \t\n");
1457
1458                        if (wordlen >= 1)
1459                                string_list_append_nodup(g->list,
1460                                                   xstrndup(value, wordlen));
1461                        value += wordlen + (value[wordlen] != '\0');
1462                }
1463        }
1464
1465        return 0;
1466}
1467
1468static int add_remote_or_group(const char *name, struct string_list *list)
1469{
1470        int prev_nr = list->nr;
1471        struct remote_group_data g;
1472        g.name = name; g.list = list;
1473
1474        git_config(get_remote_group, &g);
1475        if (list->nr == prev_nr) {
1476                struct remote *remote = remote_get(name);
1477                if (!remote_is_configured(remote, 0))
1478                        return 0;
1479                string_list_append(list, remote->name);
1480        }
1481        return 1;
1482}
1483
1484static void add_options_to_argv(struct argv_array *argv)
1485{
1486        if (dry_run)
1487                argv_array_push(argv, "--dry-run");
1488        if (prune != -1)
1489                argv_array_push(argv, prune ? "--prune" : "--no-prune");
1490        if (prune_tags != -1)
1491                argv_array_push(argv, prune_tags ? "--prune-tags" : "--no-prune-tags");
1492        if (update_head_ok)
1493                argv_array_push(argv, "--update-head-ok");
1494        if (force)
1495                argv_array_push(argv, "--force");
1496        if (keep)
1497                argv_array_push(argv, "--keep");
1498        if (recurse_submodules == RECURSE_SUBMODULES_ON)
1499                argv_array_push(argv, "--recurse-submodules");
1500        else if (recurse_submodules == RECURSE_SUBMODULES_ON_DEMAND)
1501                argv_array_push(argv, "--recurse-submodules=on-demand");
1502        if (tags == TAGS_SET)
1503                argv_array_push(argv, "--tags");
1504        else if (tags == TAGS_UNSET)
1505                argv_array_push(argv, "--no-tags");
1506        if (verbosity >= 2)
1507                argv_array_push(argv, "-v");
1508        if (verbosity >= 1)
1509                argv_array_push(argv, "-v");
1510        else if (verbosity < 0)
1511                argv_array_push(argv, "-q");
1512
1513}
1514
1515static int fetch_multiple(struct string_list *list)
1516{
1517        int i, result = 0;
1518        struct argv_array argv = ARGV_ARRAY_INIT;
1519
1520        if (!append && !dry_run) {
1521                int errcode = truncate_fetch_head();
1522                if (errcode)
1523                        return errcode;
1524        }
1525
1526        argv_array_pushl(&argv, "fetch", "--append", "--no-auto-gc", NULL);
1527        add_options_to_argv(&argv);
1528
1529        for (i = 0; i < list->nr; i++) {
1530                const char *name = list->items[i].string;
1531                argv_array_push(&argv, name);
1532                if (verbosity >= 0)
1533                        printf(_("Fetching %s\n"), name);
1534                if (run_command_v_opt(argv.argv, RUN_GIT_CMD)) {
1535                        error(_("Could not fetch %s"), name);
1536                        result = 1;
1537                }
1538                argv_array_pop(&argv);
1539        }
1540
1541        argv_array_clear(&argv);
1542        return result;
1543}
1544
1545/*
1546 * Fetching from the promisor remote should use the given filter-spec
1547 * or inherit the default filter-spec from the config.
1548 */
1549static inline void fetch_one_setup_partial(struct remote *remote)
1550{
1551        /*
1552         * Explicit --no-filter argument overrides everything, regardless
1553         * of any prior partial clones and fetches.
1554         */
1555        if (filter_options.no_filter)
1556                return;
1557
1558        /*
1559         * If no prior partial clone/fetch and the current fetch DID NOT
1560         * request a partial-fetch, do a normal fetch.
1561         */
1562        if (!repository_format_partial_clone && !filter_options.choice)
1563                return;
1564
1565        /*
1566         * If this is the FIRST partial-fetch request, we enable partial
1567         * on this repo and remember the given filter-spec as the default
1568         * for subsequent fetches to this remote.
1569         */
1570        if (!repository_format_partial_clone && filter_options.choice) {
1571                partial_clone_register(remote->name, &filter_options);
1572                return;
1573        }
1574
1575        /*
1576         * We are currently limited to only ONE promisor remote and only
1577         * allow partial-fetches from the promisor remote.
1578         */
1579        if (strcmp(remote->name, repository_format_partial_clone)) {
1580                if (filter_options.choice)
1581                        die(_("--filter can only be used with the remote "
1582                              "configured in extensions.partialClone"));
1583                return;
1584        }
1585
1586        /*
1587         * Do a partial-fetch from the promisor remote using either the
1588         * explicitly given filter-spec or inherit the filter-spec from
1589         * the config.
1590         */
1591        if (!filter_options.choice)
1592                partial_clone_get_default_filter_spec(&filter_options);
1593        return;
1594}
1595
1596static int fetch_one(struct remote *remote, int argc, const char **argv, int prune_tags_ok)
1597{
1598        struct refspec rs = REFSPEC_INIT_FETCH;
1599        int i;
1600        int exit_code;
1601        int maybe_prune_tags;
1602        int remote_via_config = remote_is_configured(remote, 0);
1603
1604        if (!remote)
1605                die(_("No remote repository specified.  Please, specify either a URL or a\n"
1606                    "remote name from which new revisions should be fetched."));
1607
1608        gtransport = prepare_transport(remote, 1);
1609
1610        if (prune < 0) {
1611                /* no command line request */
1612                if (0 <= remote->prune)
1613                        prune = remote->prune;
1614                else if (0 <= fetch_prune_config)
1615                        prune = fetch_prune_config;
1616                else
1617                        prune = PRUNE_BY_DEFAULT;
1618        }
1619
1620        if (prune_tags < 0) {
1621                /* no command line request */
1622                if (0 <= remote->prune_tags)
1623                        prune_tags = remote->prune_tags;
1624                else if (0 <= fetch_prune_tags_config)
1625                        prune_tags = fetch_prune_tags_config;
1626                else
1627                        prune_tags = PRUNE_TAGS_BY_DEFAULT;
1628        }
1629
1630        maybe_prune_tags = prune_tags_ok && prune_tags;
1631        if (maybe_prune_tags && remote_via_config)
1632                refspec_append(&remote->fetch, TAG_REFSPEC);
1633
1634        if (maybe_prune_tags && (argc || !remote_via_config))
1635                refspec_append(&rs, TAG_REFSPEC);
1636
1637        for (i = 0; i < argc; i++) {
1638                if (!strcmp(argv[i], "tag")) {
1639                        char *tag;
1640                        i++;
1641                        if (i >= argc)
1642                                die(_("You need to specify a tag name."));
1643
1644                        tag = xstrfmt("refs/tags/%s:refs/tags/%s",
1645                                      argv[i], argv[i]);
1646                        refspec_append(&rs, tag);
1647                        free(tag);
1648                } else {
1649                        refspec_append(&rs, argv[i]);
1650                }
1651        }
1652
1653        if (server_options.nr)
1654                gtransport->server_options = &server_options;
1655
1656        sigchain_push_common(unlock_pack_on_signal);
1657        atexit(unlock_pack);
1658        sigchain_push(SIGPIPE, SIG_IGN);
1659        exit_code = do_fetch(gtransport, &rs);
1660        sigchain_pop(SIGPIPE);
1661        refspec_clear(&rs);
1662        transport_disconnect(gtransport);
1663        gtransport = NULL;
1664        return exit_code;
1665}
1666
1667int cmd_fetch(int argc, const char **argv, const char *prefix)
1668{
1669        int i;
1670        struct string_list list = STRING_LIST_INIT_DUP;
1671        struct remote *remote = NULL;
1672        int result = 0;
1673        int prune_tags_ok = 1;
1674        struct argv_array argv_gc_auto = ARGV_ARRAY_INIT;
1675
1676        packet_trace_identity("fetch");
1677
1678        fetch_if_missing = 0;
1679
1680        /* Record the command line for the reflog */
1681        strbuf_addstr(&default_rla, "fetch");
1682        for (i = 1; i < argc; i++)
1683                strbuf_addf(&default_rla, " %s", argv[i]);
1684
1685        fetch_config_from_gitmodules(&max_children, &recurse_submodules);
1686        git_config(git_fetch_config, NULL);
1687
1688        argc = parse_options(argc, argv, prefix,
1689                             builtin_fetch_options, builtin_fetch_usage, 0);
1690
1691        if (deepen_relative) {
1692                if (deepen_relative < 0)
1693                        die(_("Negative depth in --deepen is not supported"));
1694                if (depth)
1695                        die(_("--deepen and --depth are mutually exclusive"));
1696                depth = xstrfmt("%d", deepen_relative);
1697        }
1698        if (unshallow) {
1699                if (depth)
1700                        die(_("--depth and --unshallow cannot be used together"));
1701                else if (!is_repository_shallow(the_repository))
1702                        die(_("--unshallow on a complete repository does not make sense"));
1703                else
1704                        depth = xstrfmt("%d", INFINITE_DEPTH);
1705        }
1706
1707        /* no need to be strict, transport_set_option() will validate it again */
1708        if (depth && atoi(depth) < 1)
1709                die(_("depth %s is not a positive number"), depth);
1710        if (depth || deepen_since || deepen_not.nr)
1711                deepen = 1;
1712
1713        if (filter_options.choice && !repository_format_partial_clone)
1714                die("--filter can only be used when extensions.partialClone is set");
1715
1716        if (all) {
1717                if (argc == 1)
1718                        die(_("fetch --all does not take a repository argument"));
1719                else if (argc > 1)
1720                        die(_("fetch --all does not make sense with refspecs"));
1721                (void) for_each_remote(get_one_remote_for_fetch, &list);
1722        } else if (argc == 0) {
1723                /* No arguments -- use default remote */
1724                remote = remote_get(NULL);
1725        } else if (multiple) {
1726                /* All arguments are assumed to be remotes or groups */
1727                for (i = 0; i < argc; i++)
1728                        if (!add_remote_or_group(argv[i], &list))
1729                                die(_("No such remote or remote group: %s"), argv[i]);
1730        } else {
1731                /* Single remote or group */
1732                (void) add_remote_or_group(argv[0], &list);
1733                if (list.nr > 1) {
1734                        /* More than one remote */
1735                        if (argc > 1)
1736                                die(_("Fetching a group and specifying refspecs does not make sense"));
1737                } else {
1738                        /* Zero or one remotes */
1739                        remote = remote_get(argv[0]);
1740                        prune_tags_ok = (argc == 1);
1741                        argc--;
1742                        argv++;
1743                }
1744        }
1745
1746        if (remote) {
1747                if (filter_options.choice || repository_format_partial_clone)
1748                        fetch_one_setup_partial(remote);
1749                result = fetch_one(remote, argc, argv, prune_tags_ok);
1750        } else {
1751                if (filter_options.choice)
1752                        die(_("--filter can only be used with the remote "
1753                              "configured in extensions.partialclone"));
1754                /* TODO should this also die if we have a previous partial-clone? */
1755                result = fetch_multiple(&list);
1756        }
1757
1758        if (!result && (recurse_submodules != RECURSE_SUBMODULES_OFF)) {
1759                struct argv_array options = ARGV_ARRAY_INIT;
1760
1761                add_options_to_argv(&options);
1762                result = fetch_populated_submodules(the_repository,
1763                                                    &options,
1764                                                    submodule_prefix,
1765                                                    recurse_submodules,
1766                                                    recurse_submodules_default,
1767                                                    verbosity < 0,
1768                                                    max_children);
1769                argv_array_clear(&options);
1770        }
1771
1772        string_list_clear(&list, 0);
1773
1774        close_object_store(the_repository->objects);
1775
1776        if (enable_auto_gc) {
1777                argv_array_pushl(&argv_gc_auto, "gc", "--auto", NULL);
1778                if (verbosity < 0)
1779                        argv_array_push(&argv_gc_auto, "--quiet");
1780                run_command_v_opt(argv_gc_auto.argv, RUN_GIT_CMD);
1781                argv_array_clear(&argv_gc_auto);
1782        }
1783
1784        return result;
1785}