OFFSETOF_VAR macro to simplify hashmap iterators
[git] / builtin / fetch.c
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
27 #define FORCED_UPDATES_DELAY_WARNING_IN_MS (10 * 1000)
28
29 static const char * const builtin_fetch_usage[] = {
30         N_("git fetch [<options>] [<repository> [<refspec>...]]"),
31         N_("git fetch [<options>] <group>"),
32         N_("git fetch --multiple [<options>] [(<repository> | <group>)...]"),
33         N_("git fetch --all [<options>]"),
34         NULL
35 };
36
37 enum {
38         TAGS_UNSET = 0,
39         TAGS_DEFAULT = 1,
40         TAGS_SET = 2
41 };
42
43 static int fetch_prune_config = -1; /* unspecified */
44 static int fetch_show_forced_updates = 1;
45 static uint64_t forced_updates_ms = 0;
46 static int prune = -1; /* unspecified */
47 #define PRUNE_BY_DEFAULT 0 /* do we prune by default? */
48
49 static int fetch_prune_tags_config = -1; /* unspecified */
50 static int prune_tags = -1; /* unspecified */
51 #define PRUNE_TAGS_BY_DEFAULT 0 /* do we prune tags by default? */
52
53 static int all, append, dry_run, force, keep, multiple, update_head_ok, verbosity, deepen_relative;
54 static int progress = -1;
55 static int enable_auto_gc = 1;
56 static int tags = TAGS_DEFAULT, unshallow, update_shallow, deepen;
57 static int max_children = 1;
58 static enum transport_family family;
59 static const char *depth;
60 static const char *deepen_since;
61 static const char *upload_pack;
62 static struct string_list deepen_not = STRING_LIST_INIT_NODUP;
63 static struct strbuf default_rla = STRBUF_INIT;
64 static struct transport *gtransport;
65 static struct transport *gsecondary;
66 static const char *submodule_prefix = "";
67 static int recurse_submodules = RECURSE_SUBMODULES_DEFAULT;
68 static int recurse_submodules_default = RECURSE_SUBMODULES_ON_DEMAND;
69 static int shown_url = 0;
70 static struct refspec refmap = REFSPEC_INIT_FETCH;
71 static struct list_objects_filter_options filter_options;
72 static struct string_list server_options = STRING_LIST_INIT_DUP;
73 static struct string_list negotiation_tip = STRING_LIST_INIT_NODUP;
74
75 static int git_fetch_config(const char *k, const char *v, void *cb)
76 {
77         if (!strcmp(k, "fetch.prune")) {
78                 fetch_prune_config = git_config_bool(k, v);
79                 return 0;
80         }
81
82         if (!strcmp(k, "fetch.prunetags")) {
83                 fetch_prune_tags_config = git_config_bool(k, v);
84                 return 0;
85         }
86
87         if (!strcmp(k, "fetch.showforcedupdates")) {
88                 fetch_show_forced_updates = git_config_bool(k, v);
89                 return 0;
90         }
91
92         if (!strcmp(k, "submodule.recurse")) {
93                 int r = git_config_bool(k, v) ?
94                         RECURSE_SUBMODULES_ON : RECURSE_SUBMODULES_OFF;
95                 recurse_submodules = r;
96         }
97
98         if (!strcmp(k, "submodule.fetchjobs")) {
99                 max_children = parse_submodule_fetchjobs(k, v);
100                 return 0;
101         } else if (!strcmp(k, "fetch.recursesubmodules")) {
102                 recurse_submodules = parse_fetch_recurse_submodules_arg(k, v);
103                 return 0;
104         }
105
106         return git_default_config(k, v, cb);
107 }
108
109 static int parse_refmap_arg(const struct option *opt, const char *arg, int unset)
110 {
111         BUG_ON_OPT_NEG(unset);
112
113         /*
114          * "git fetch --refmap='' origin foo"
115          * can be used to tell the command not to store anywhere
116          */
117         refspec_append(&refmap, arg);
118
119         return 0;
120 }
121
122 static struct option builtin_fetch_options[] = {
123         OPT__VERBOSITY(&verbosity),
124         OPT_BOOL(0, "all", &all,
125                  N_("fetch from all remotes")),
126         OPT_BOOL('a', "append", &append,
127                  N_("append to .git/FETCH_HEAD instead of overwriting")),
128         OPT_STRING(0, "upload-pack", &upload_pack, N_("path"),
129                    N_("path to upload pack on remote end")),
130         OPT__FORCE(&force, N_("force overwrite of local reference"), 0),
131         OPT_BOOL('m', "multiple", &multiple,
132                  N_("fetch from multiple remotes")),
133         OPT_SET_INT('t', "tags", &tags,
134                     N_("fetch all tags and associated objects"), TAGS_SET),
135         OPT_SET_INT('n', NULL, &tags,
136                     N_("do not fetch all tags (--no-tags)"), TAGS_UNSET),
137         OPT_INTEGER('j', "jobs", &max_children,
138                     N_("number of submodules fetched in parallel")),
139         OPT_BOOL('p', "prune", &prune,
140                  N_("prune remote-tracking branches no longer on remote")),
141         OPT_BOOL('P', "prune-tags", &prune_tags,
142                  N_("prune local tags no longer on remote and clobber changed tags")),
143         { OPTION_CALLBACK, 0, "recurse-submodules", &recurse_submodules, N_("on-demand"),
144                     N_("control recursive fetching of submodules"),
145                     PARSE_OPT_OPTARG, option_fetch_parse_recurse_submodules },
146         OPT_BOOL(0, "dry-run", &dry_run,
147                  N_("dry run")),
148         OPT_BOOL('k', "keep", &keep, N_("keep downloaded pack")),
149         OPT_BOOL('u', "update-head-ok", &update_head_ok,
150                     N_("allow updating of HEAD ref")),
151         OPT_BOOL(0, "progress", &progress, N_("force progress reporting")),
152         OPT_STRING(0, "depth", &depth, N_("depth"),
153                    N_("deepen history of shallow clone")),
154         OPT_STRING(0, "shallow-since", &deepen_since, N_("time"),
155                    N_("deepen history of shallow repository based on time")),
156         OPT_STRING_LIST(0, "shallow-exclude", &deepen_not, N_("revision"),
157                         N_("deepen history of shallow clone, excluding rev")),
158         OPT_INTEGER(0, "deepen", &deepen_relative,
159                     N_("deepen history of shallow clone")),
160         OPT_SET_INT_F(0, "unshallow", &unshallow,
161                       N_("convert to a complete repository"),
162                       1, PARSE_OPT_NONEG),
163         { OPTION_STRING, 0, "submodule-prefix", &submodule_prefix, N_("dir"),
164                    N_("prepend this to submodule path output"), PARSE_OPT_HIDDEN },
165         { OPTION_CALLBACK, 0, "recurse-submodules-default",
166                    &recurse_submodules_default, N_("on-demand"),
167                    N_("default for recursive fetching of submodules "
168                       "(lower priority than config files)"),
169                    PARSE_OPT_HIDDEN, option_fetch_parse_recurse_submodules },
170         OPT_BOOL(0, "update-shallow", &update_shallow,
171                  N_("accept refs that update .git/shallow")),
172         { OPTION_CALLBACK, 0, "refmap", NULL, N_("refmap"),
173           N_("specify fetch refmap"), PARSE_OPT_NONEG, parse_refmap_arg },
174         OPT_STRING_LIST('o', "server-option", &server_options, N_("server-specific"), N_("option to transmit")),
175         OPT_SET_INT('4', "ipv4", &family, N_("use IPv4 addresses only"),
176                         TRANSPORT_FAMILY_IPV4),
177         OPT_SET_INT('6', "ipv6", &family, N_("use IPv6 addresses only"),
178                         TRANSPORT_FAMILY_IPV6),
179         OPT_STRING_LIST(0, "negotiation-tip", &negotiation_tip, N_("revision"),
180                         N_("report that we have only objects reachable from this object")),
181         OPT_PARSE_LIST_OBJECTS_FILTER(&filter_options),
182         OPT_BOOL(0, "auto-gc", &enable_auto_gc,
183                  N_("run 'gc --auto' after fetching")),
184         OPT_BOOL(0, "show-forced-updates", &fetch_show_forced_updates,
185                  N_("check for forced-updates on all updated branches")),
186         OPT_END()
187 };
188
189 static void unlock_pack(void)
190 {
191         if (gtransport)
192                 transport_unlock_pack(gtransport);
193         if (gsecondary)
194                 transport_unlock_pack(gsecondary);
195 }
196
197 static void unlock_pack_on_signal(int signo)
198 {
199         unlock_pack();
200         sigchain_pop(signo);
201         raise(signo);
202 }
203
204 static void add_merge_config(struct ref **head,
205                            const struct ref *remote_refs,
206                            struct branch *branch,
207                            struct ref ***tail)
208 {
209         int i;
210
211         for (i = 0; i < branch->merge_nr; i++) {
212                 struct ref *rm, **old_tail = *tail;
213                 struct refspec_item refspec;
214
215                 for (rm = *head; rm; rm = rm->next) {
216                         if (branch_merge_matches(branch, i, rm->name)) {
217                                 rm->fetch_head_status = FETCH_HEAD_MERGE;
218                                 break;
219                         }
220                 }
221                 if (rm)
222                         continue;
223
224                 /*
225                  * Not fetched to a remote-tracking branch?  We need to fetch
226                  * it anyway to allow this branch's "branch.$name.merge"
227                  * to be honored by 'git pull', but we do not have to
228                  * fail if branch.$name.merge is misconfigured to point
229                  * at a nonexisting branch.  If we were indeed called by
230                  * 'git pull', it will notice the misconfiguration because
231                  * there is no entry in the resulting FETCH_HEAD marked
232                  * for merging.
233                  */
234                 memset(&refspec, 0, sizeof(refspec));
235                 refspec.src = branch->merge[i]->src;
236                 get_fetch_map(remote_refs, &refspec, tail, 1);
237                 for (rm = *old_tail; rm; rm = rm->next)
238                         rm->fetch_head_status = FETCH_HEAD_MERGE;
239         }
240 }
241
242 static int will_fetch(struct ref **head, const unsigned char *sha1)
243 {
244         struct ref *rm = *head;
245         while (rm) {
246                 if (hasheq(rm->old_oid.hash, sha1))
247                         return 1;
248                 rm = rm->next;
249         }
250         return 0;
251 }
252
253 struct refname_hash_entry {
254         struct hashmap_entry ent; /* must be the first member */
255         struct object_id oid;
256         int ignore;
257         char refname[FLEX_ARRAY];
258 };
259
260 static int refname_hash_entry_cmp(const void *hashmap_cmp_fn_data,
261                                   const struct hashmap_entry *eptr,
262                                   const struct hashmap_entry *entry_or_key,
263                                   const void *keydata)
264 {
265         const struct refname_hash_entry *e1, *e2;
266
267         e1 = container_of(eptr, const struct refname_hash_entry, ent);
268         e2 = container_of(entry_or_key, const struct refname_hash_entry, ent);
269         return strcmp(e1->refname, keydata ? keydata : e2->refname);
270 }
271
272 static struct refname_hash_entry *refname_hash_add(struct hashmap *map,
273                                                    const char *refname,
274                                                    const struct object_id *oid)
275 {
276         struct refname_hash_entry *ent;
277         size_t len = strlen(refname);
278
279         FLEX_ALLOC_MEM(ent, refname, refname, len);
280         hashmap_entry_init(&ent->ent, strhash(refname));
281         oidcpy(&ent->oid, oid);
282         hashmap_add(map, &ent->ent);
283         return ent;
284 }
285
286 static int add_one_refname(const char *refname,
287                            const struct object_id *oid,
288                            int flag, void *cbdata)
289 {
290         struct hashmap *refname_map = cbdata;
291
292         (void) refname_hash_add(refname_map, refname, oid);
293         return 0;
294 }
295
296 static void refname_hash_init(struct hashmap *map)
297 {
298         hashmap_init(map, refname_hash_entry_cmp, NULL, 0);
299 }
300
301 static int refname_hash_exists(struct hashmap *map, const char *refname)
302 {
303         return !!hashmap_get_from_hash(map, strhash(refname), refname);
304 }
305
306 static void clear_item(struct refname_hash_entry *item)
307 {
308         item->ignore = 1;
309 }
310
311 static void find_non_local_tags(const struct ref *refs,
312                                 struct ref **head,
313                                 struct ref ***tail)
314 {
315         struct hashmap existing_refs;
316         struct hashmap remote_refs;
317         struct string_list remote_refs_list = STRING_LIST_INIT_NODUP;
318         struct string_list_item *remote_ref_item;
319         const struct ref *ref;
320         struct refname_hash_entry *item = NULL;
321
322         refname_hash_init(&existing_refs);
323         refname_hash_init(&remote_refs);
324
325         for_each_ref(add_one_refname, &existing_refs);
326         for (ref = refs; ref; ref = ref->next) {
327                 if (!starts_with(ref->name, "refs/tags/"))
328                         continue;
329
330                 /*
331                  * The peeled ref always follows the matching base
332                  * ref, so if we see a peeled ref that we don't want
333                  * to fetch then we can mark the ref entry in the list
334                  * as one to ignore by setting util to NULL.
335                  */
336                 if (ends_with(ref->name, "^{}")) {
337                         if (item &&
338                             !has_object_file_with_flags(&ref->old_oid,
339                                                         OBJECT_INFO_QUICK) &&
340                             !will_fetch(head, ref->old_oid.hash) &&
341                             !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
342                             !will_fetch(head, item->oid.hash))
343                                 clear_item(item);
344                         item = NULL;
345                         continue;
346                 }
347
348                 /*
349                  * If item is non-NULL here, then we previously saw a
350                  * ref not followed by a peeled reference, so we need
351                  * to check if it is a lightweight tag that we want to
352                  * fetch.
353                  */
354                 if (item &&
355                     !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
356                     !will_fetch(head, item->oid.hash))
357                         clear_item(item);
358
359                 item = NULL;
360
361                 /* skip duplicates and refs that we already have */
362                 if (refname_hash_exists(&remote_refs, ref->name) ||
363                     refname_hash_exists(&existing_refs, ref->name))
364                         continue;
365
366                 item = refname_hash_add(&remote_refs, ref->name, &ref->old_oid);
367                 string_list_insert(&remote_refs_list, ref->name);
368         }
369         hashmap_free_entries(&existing_refs, struct refname_hash_entry, ent);
370
371         /*
372          * We may have a final lightweight tag that needs to be
373          * checked to see if it needs fetching.
374          */
375         if (item &&
376             !has_object_file_with_flags(&item->oid, OBJECT_INFO_QUICK) &&
377             !will_fetch(head, item->oid.hash))
378                 clear_item(item);
379
380         /*
381          * For all the tags in the remote_refs_list,
382          * add them to the list of refs to be fetched
383          */
384         for_each_string_list_item(remote_ref_item, &remote_refs_list) {
385                 const char *refname = remote_ref_item->string;
386                 struct ref *rm;
387                 unsigned int hash = strhash(refname);
388
389                 item = hashmap_get_entry_from_hash(&remote_refs, hash, refname,
390                                         struct refname_hash_entry, ent);
391                 if (!item)
392                         BUG("unseen remote ref?");
393
394                 /* Unless we have already decided to ignore this item... */
395                 if (item->ignore)
396                         continue;
397
398                 rm = alloc_ref(item->refname);
399                 rm->peer_ref = alloc_ref(item->refname);
400                 oidcpy(&rm->old_oid, &item->oid);
401                 **tail = rm;
402                 *tail = &rm->next;
403         }
404         hashmap_free_entries(&remote_refs, struct refname_hash_entry, ent);
405         string_list_clear(&remote_refs_list, 0);
406 }
407
408 static struct ref *get_ref_map(struct remote *remote,
409                                const struct ref *remote_refs,
410                                struct refspec *rs,
411                                int tags, int *autotags)
412 {
413         int i;
414         struct ref *rm;
415         struct ref *ref_map = NULL;
416         struct ref **tail = &ref_map;
417
418         /* opportunistically-updated references: */
419         struct ref *orefs = NULL, **oref_tail = &orefs;
420
421         struct hashmap existing_refs;
422
423         if (rs->nr) {
424                 struct refspec *fetch_refspec;
425
426                 for (i = 0; i < rs->nr; i++) {
427                         get_fetch_map(remote_refs, &rs->items[i], &tail, 0);
428                         if (rs->items[i].dst && rs->items[i].dst[0])
429                                 *autotags = 1;
430                 }
431                 /* Merge everything on the command line (but not --tags) */
432                 for (rm = ref_map; rm; rm = rm->next)
433                         rm->fetch_head_status = FETCH_HEAD_MERGE;
434
435                 /*
436                  * For any refs that we happen to be fetching via
437                  * command-line arguments, the destination ref might
438                  * have been missing or have been different than the
439                  * remote-tracking ref that would be derived from the
440                  * configured refspec.  In these cases, we want to
441                  * take the opportunity to update their configured
442                  * remote-tracking reference.  However, we do not want
443                  * to mention these entries in FETCH_HEAD at all, as
444                  * they would simply be duplicates of existing
445                  * entries, so we set them FETCH_HEAD_IGNORE below.
446                  *
447                  * We compute these entries now, based only on the
448                  * refspecs specified on the command line.  But we add
449                  * them to the list following the refspecs resulting
450                  * from the tags option so that one of the latter,
451                  * which has FETCH_HEAD_NOT_FOR_MERGE, is not removed
452                  * by ref_remove_duplicates() in favor of one of these
453                  * opportunistic entries with FETCH_HEAD_IGNORE.
454                  */
455                 if (refmap.nr)
456                         fetch_refspec = &refmap;
457                 else
458                         fetch_refspec = &remote->fetch;
459
460                 for (i = 0; i < fetch_refspec->nr; i++)
461                         get_fetch_map(ref_map, &fetch_refspec->items[i], &oref_tail, 1);
462         } else if (refmap.nr) {
463                 die("--refmap option is only meaningful with command-line refspec(s).");
464         } else {
465                 /* Use the defaults */
466                 struct branch *branch = branch_get(NULL);
467                 int has_merge = branch_has_merge_config(branch);
468                 if (remote &&
469                     (remote->fetch.nr ||
470                      /* Note: has_merge implies non-NULL branch->remote_name */
471                      (has_merge && !strcmp(branch->remote_name, remote->name)))) {
472                         for (i = 0; i < remote->fetch.nr; i++) {
473                                 get_fetch_map(remote_refs, &remote->fetch.items[i], &tail, 0);
474                                 if (remote->fetch.items[i].dst &&
475                                     remote->fetch.items[i].dst[0])
476                                         *autotags = 1;
477                                 if (!i && !has_merge && ref_map &&
478                                     !remote->fetch.items[0].pattern)
479                                         ref_map->fetch_head_status = FETCH_HEAD_MERGE;
480                         }
481                         /*
482                          * if the remote we're fetching from is the same
483                          * as given in branch.<name>.remote, we add the
484                          * ref given in branch.<name>.merge, too.
485                          *
486                          * Note: has_merge implies non-NULL branch->remote_name
487                          */
488                         if (has_merge &&
489                             !strcmp(branch->remote_name, remote->name))
490                                 add_merge_config(&ref_map, remote_refs, branch, &tail);
491                 } else {
492                         ref_map = get_remote_ref(remote_refs, "HEAD");
493                         if (!ref_map)
494                                 die(_("Couldn't find remote ref HEAD"));
495                         ref_map->fetch_head_status = FETCH_HEAD_MERGE;
496                         tail = &ref_map->next;
497                 }
498         }
499
500         if (tags == TAGS_SET)
501                 /* also fetch all tags */
502                 get_fetch_map(remote_refs, tag_refspec, &tail, 0);
503         else if (tags == TAGS_DEFAULT && *autotags)
504                 find_non_local_tags(remote_refs, &ref_map, &tail);
505
506         /* Now append any refs to be updated opportunistically: */
507         *tail = orefs;
508         for (rm = orefs; rm; rm = rm->next) {
509                 rm->fetch_head_status = FETCH_HEAD_IGNORE;
510                 tail = &rm->next;
511         }
512
513         ref_map = ref_remove_duplicates(ref_map);
514
515         refname_hash_init(&existing_refs);
516         for_each_ref(add_one_refname, &existing_refs);
517
518         for (rm = ref_map; rm; rm = rm->next) {
519                 if (rm->peer_ref) {
520                         const char *refname = rm->peer_ref->name;
521                         struct refname_hash_entry *peer_item;
522                         unsigned int hash = strhash(refname);
523
524                         peer_item = hashmap_get_entry_from_hash(&existing_refs,
525                                                 hash, refname,
526                                                 struct refname_hash_entry, ent);
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_entries(&existing_refs, struct refname_hash_entry, ent);
534
535         return ref_map;
536 }
537
538 #define STORE_REF_ERROR_OTHER 1
539 #define STORE_REF_ERROR_DF_CONFLICT 2
540
541 static 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;
575 fail:
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
584 static int refcol_width = 10;
585 static int compact_format;
586
587 static 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
623 static 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
647 static 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
653 static 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
680 static 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
702 static 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
718 static 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
843 static 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
857 static const char warn_show_forced_updates[] =
858 N_("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'.");
861 static const char warn_time_show_forced_updates[] =
862 N_("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
866 static 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  */
1040 static 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
1069 static 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 */
1085 static 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
1097 static 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
1152 static 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
1166 static 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
1177 static 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
1189 static 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
1198 static 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
1222 static 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
1262 static 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
1292 static 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         free_refs(ref_map);
1375
1376         /* if neither --no-tags nor --tags was specified, do automated tag
1377          * following ... */
1378         if (tags == TAGS_DEFAULT && autotags) {
1379                 struct ref **tail = &ref_map;
1380                 ref_map = NULL;
1381                 find_non_local_tags(remote_refs, &ref_map, &tail);
1382                 if (ref_map)
1383                         backfill_tags(transport, ref_map);
1384                 free_refs(ref_map);
1385         }
1386
1387  cleanup:
1388         return retcode;
1389 }
1390
1391 static int get_one_remote_for_fetch(struct remote *remote, void *priv)
1392 {
1393         struct string_list *list = priv;
1394         if (!remote->skip_default_update)
1395                 string_list_append(list, remote->name);
1396         return 0;
1397 }
1398
1399 struct remote_group_data {
1400         const char *name;
1401         struct string_list *list;
1402 };
1403
1404 static int get_remote_group(const char *key, const char *value, void *priv)
1405 {
1406         struct remote_group_data *g = priv;
1407
1408         if (skip_prefix(key, "remotes.", &key) && !strcmp(key, g->name)) {
1409                 /* split list by white space */
1410                 while (*value) {
1411                         size_t wordlen = strcspn(value, " \t\n");
1412
1413                         if (wordlen >= 1)
1414                                 string_list_append_nodup(g->list,
1415                                                    xstrndup(value, wordlen));
1416                         value += wordlen + (value[wordlen] != '\0');
1417                 }
1418         }
1419
1420         return 0;
1421 }
1422
1423 static int add_remote_or_group(const char *name, struct string_list *list)
1424 {
1425         int prev_nr = list->nr;
1426         struct remote_group_data g;
1427         g.name = name; g.list = list;
1428
1429         git_config(get_remote_group, &g);
1430         if (list->nr == prev_nr) {
1431                 struct remote *remote = remote_get(name);
1432                 if (!remote_is_configured(remote, 0))
1433                         return 0;
1434                 string_list_append(list, remote->name);
1435         }
1436         return 1;
1437 }
1438
1439 static void add_options_to_argv(struct argv_array *argv)
1440 {
1441         if (dry_run)
1442                 argv_array_push(argv, "--dry-run");
1443         if (prune != -1)
1444                 argv_array_push(argv, prune ? "--prune" : "--no-prune");
1445         if (prune_tags != -1)
1446                 argv_array_push(argv, prune_tags ? "--prune-tags" : "--no-prune-tags");
1447         if (update_head_ok)
1448                 argv_array_push(argv, "--update-head-ok");
1449         if (force)
1450                 argv_array_push(argv, "--force");
1451         if (keep)
1452                 argv_array_push(argv, "--keep");
1453         if (recurse_submodules == RECURSE_SUBMODULES_ON)
1454                 argv_array_push(argv, "--recurse-submodules");
1455         else if (recurse_submodules == RECURSE_SUBMODULES_ON_DEMAND)
1456                 argv_array_push(argv, "--recurse-submodules=on-demand");
1457         if (tags == TAGS_SET)
1458                 argv_array_push(argv, "--tags");
1459         else if (tags == TAGS_UNSET)
1460                 argv_array_push(argv, "--no-tags");
1461         if (verbosity >= 2)
1462                 argv_array_push(argv, "-v");
1463         if (verbosity >= 1)
1464                 argv_array_push(argv, "-v");
1465         else if (verbosity < 0)
1466                 argv_array_push(argv, "-q");
1467
1468 }
1469
1470 static int fetch_multiple(struct string_list *list)
1471 {
1472         int i, result = 0;
1473         struct argv_array argv = ARGV_ARRAY_INIT;
1474
1475         if (!append && !dry_run) {
1476                 int errcode = truncate_fetch_head();
1477                 if (errcode)
1478                         return errcode;
1479         }
1480
1481         argv_array_pushl(&argv, "fetch", "--append", "--no-auto-gc", NULL);
1482         add_options_to_argv(&argv);
1483
1484         for (i = 0; i < list->nr; i++) {
1485                 const char *name = list->items[i].string;
1486                 argv_array_push(&argv, name);
1487                 if (verbosity >= 0)
1488                         printf(_("Fetching %s\n"), name);
1489                 if (run_command_v_opt(argv.argv, RUN_GIT_CMD)) {
1490                         error(_("Could not fetch %s"), name);
1491                         result = 1;
1492                 }
1493                 argv_array_pop(&argv);
1494         }
1495
1496         argv_array_clear(&argv);
1497         return result;
1498 }
1499
1500 /*
1501  * Fetching from the promisor remote should use the given filter-spec
1502  * or inherit the default filter-spec from the config.
1503  */
1504 static inline void fetch_one_setup_partial(struct remote *remote)
1505 {
1506         /*
1507          * Explicit --no-filter argument overrides everything, regardless
1508          * of any prior partial clones and fetches.
1509          */
1510         if (filter_options.no_filter)
1511                 return;
1512
1513         /*
1514          * If no prior partial clone/fetch and the current fetch DID NOT
1515          * request a partial-fetch, do a normal fetch.
1516          */
1517         if (!repository_format_partial_clone && !filter_options.choice)
1518                 return;
1519
1520         /*
1521          * If this is the FIRST partial-fetch request, we enable partial
1522          * on this repo and remember the given filter-spec as the default
1523          * for subsequent fetches to this remote.
1524          */
1525         if (!repository_format_partial_clone && filter_options.choice) {
1526                 partial_clone_register(remote->name, &filter_options);
1527                 return;
1528         }
1529
1530         /*
1531          * We are currently limited to only ONE promisor remote and only
1532          * allow partial-fetches from the promisor remote.
1533          */
1534         if (strcmp(remote->name, repository_format_partial_clone)) {
1535                 if (filter_options.choice)
1536                         die(_("--filter can only be used with the remote "
1537                               "configured in extensions.partialClone"));
1538                 return;
1539         }
1540
1541         /*
1542          * Do a partial-fetch from the promisor remote using either the
1543          * explicitly given filter-spec or inherit the filter-spec from
1544          * the config.
1545          */
1546         if (!filter_options.choice)
1547                 partial_clone_get_default_filter_spec(&filter_options);
1548         return;
1549 }
1550
1551 static int fetch_one(struct remote *remote, int argc, const char **argv, int prune_tags_ok)
1552 {
1553         struct refspec rs = REFSPEC_INIT_FETCH;
1554         int i;
1555         int exit_code;
1556         int maybe_prune_tags;
1557         int remote_via_config = remote_is_configured(remote, 0);
1558
1559         if (!remote)
1560                 die(_("No remote repository specified.  Please, specify either a URL or a\n"
1561                     "remote name from which new revisions should be fetched."));
1562
1563         gtransport = prepare_transport(remote, 1);
1564
1565         if (prune < 0) {
1566                 /* no command line request */
1567                 if (0 <= remote->prune)
1568                         prune = remote->prune;
1569                 else if (0 <= fetch_prune_config)
1570                         prune = fetch_prune_config;
1571                 else
1572                         prune = PRUNE_BY_DEFAULT;
1573         }
1574
1575         if (prune_tags < 0) {
1576                 /* no command line request */
1577                 if (0 <= remote->prune_tags)
1578                         prune_tags = remote->prune_tags;
1579                 else if (0 <= fetch_prune_tags_config)
1580                         prune_tags = fetch_prune_tags_config;
1581                 else
1582                         prune_tags = PRUNE_TAGS_BY_DEFAULT;
1583         }
1584
1585         maybe_prune_tags = prune_tags_ok && prune_tags;
1586         if (maybe_prune_tags && remote_via_config)
1587                 refspec_append(&remote->fetch, TAG_REFSPEC);
1588
1589         if (maybe_prune_tags && (argc || !remote_via_config))
1590                 refspec_append(&rs, TAG_REFSPEC);
1591
1592         for (i = 0; i < argc; i++) {
1593                 if (!strcmp(argv[i], "tag")) {
1594                         char *tag;
1595                         i++;
1596                         if (i >= argc)
1597                                 die(_("You need to specify a tag name."));
1598
1599                         tag = xstrfmt("refs/tags/%s:refs/tags/%s",
1600                                       argv[i], argv[i]);
1601                         refspec_append(&rs, tag);
1602                         free(tag);
1603                 } else {
1604                         refspec_append(&rs, argv[i]);
1605                 }
1606         }
1607
1608         if (server_options.nr)
1609                 gtransport->server_options = &server_options;
1610
1611         sigchain_push_common(unlock_pack_on_signal);
1612         atexit(unlock_pack);
1613         sigchain_push(SIGPIPE, SIG_IGN);
1614         exit_code = do_fetch(gtransport, &rs);
1615         sigchain_pop(SIGPIPE);
1616         refspec_clear(&rs);
1617         transport_disconnect(gtransport);
1618         gtransport = NULL;
1619         return exit_code;
1620 }
1621
1622 int cmd_fetch(int argc, const char **argv, const char *prefix)
1623 {
1624         int i;
1625         struct string_list list = STRING_LIST_INIT_DUP;
1626         struct remote *remote = NULL;
1627         int result = 0;
1628         int prune_tags_ok = 1;
1629         struct argv_array argv_gc_auto = ARGV_ARRAY_INIT;
1630
1631         packet_trace_identity("fetch");
1632
1633         fetch_if_missing = 0;
1634
1635         /* Record the command line for the reflog */
1636         strbuf_addstr(&default_rla, "fetch");
1637         for (i = 1; i < argc; i++)
1638                 strbuf_addf(&default_rla, " %s", argv[i]);
1639
1640         fetch_config_from_gitmodules(&max_children, &recurse_submodules);
1641         git_config(git_fetch_config, NULL);
1642
1643         argc = parse_options(argc, argv, prefix,
1644                              builtin_fetch_options, builtin_fetch_usage, 0);
1645
1646         if (deepen_relative) {
1647                 if (deepen_relative < 0)
1648                         die(_("Negative depth in --deepen is not supported"));
1649                 if (depth)
1650                         die(_("--deepen and --depth are mutually exclusive"));
1651                 depth = xstrfmt("%d", deepen_relative);
1652         }
1653         if (unshallow) {
1654                 if (depth)
1655                         die(_("--depth and --unshallow cannot be used together"));
1656                 else if (!is_repository_shallow(the_repository))
1657                         die(_("--unshallow on a complete repository does not make sense"));
1658                 else
1659                         depth = xstrfmt("%d", INFINITE_DEPTH);
1660         }
1661
1662         /* no need to be strict, transport_set_option() will validate it again */
1663         if (depth && atoi(depth) < 1)
1664                 die(_("depth %s is not a positive number"), depth);
1665         if (depth || deepen_since || deepen_not.nr)
1666                 deepen = 1;
1667
1668         if (filter_options.choice && !repository_format_partial_clone)
1669                 die("--filter can only be used when extensions.partialClone is set");
1670
1671         if (all) {
1672                 if (argc == 1)
1673                         die(_("fetch --all does not take a repository argument"));
1674                 else if (argc > 1)
1675                         die(_("fetch --all does not make sense with refspecs"));
1676                 (void) for_each_remote(get_one_remote_for_fetch, &list);
1677         } else if (argc == 0) {
1678                 /* No arguments -- use default remote */
1679                 remote = remote_get(NULL);
1680         } else if (multiple) {
1681                 /* All arguments are assumed to be remotes or groups */
1682                 for (i = 0; i < argc; i++)
1683                         if (!add_remote_or_group(argv[i], &list))
1684                                 die(_("No such remote or remote group: %s"), argv[i]);
1685         } else {
1686                 /* Single remote or group */
1687                 (void) add_remote_or_group(argv[0], &list);
1688                 if (list.nr > 1) {
1689                         /* More than one remote */
1690                         if (argc > 1)
1691                                 die(_("Fetching a group and specifying refspecs does not make sense"));
1692                 } else {
1693                         /* Zero or one remotes */
1694                         remote = remote_get(argv[0]);
1695                         prune_tags_ok = (argc == 1);
1696                         argc--;
1697                         argv++;
1698                 }
1699         }
1700
1701         if (remote) {
1702                 if (filter_options.choice || repository_format_partial_clone)
1703                         fetch_one_setup_partial(remote);
1704                 result = fetch_one(remote, argc, argv, prune_tags_ok);
1705         } else {
1706                 if (filter_options.choice)
1707                         die(_("--filter can only be used with the remote "
1708                               "configured in extensions.partialclone"));
1709                 /* TODO should this also die if we have a previous partial-clone? */
1710                 result = fetch_multiple(&list);
1711         }
1712
1713         if (!result && (recurse_submodules != RECURSE_SUBMODULES_OFF)) {
1714                 struct argv_array options = ARGV_ARRAY_INIT;
1715
1716                 add_options_to_argv(&options);
1717                 result = fetch_populated_submodules(the_repository,
1718                                                     &options,
1719                                                     submodule_prefix,
1720                                                     recurse_submodules,
1721                                                     recurse_submodules_default,
1722                                                     verbosity < 0,
1723                                                     max_children);
1724                 argv_array_clear(&options);
1725         }
1726
1727         string_list_clear(&list, 0);
1728
1729         close_object_store(the_repository->objects);
1730
1731         if (enable_auto_gc) {
1732                 argv_array_pushl(&argv_gc_auto, "gc", "--auto", NULL);
1733                 if (verbosity < 0)
1734                         argv_array_push(&argv_gc_auto, "--quiet");
1735                 run_command_v_opt(argv_gc_auto.argv, RUN_GIT_CMD);
1736                 argv_array_clear(&argv_gc_auto);
1737         }
1738
1739         return result;
1740 }