Merge branch 'mp/for-each-ref-missing-name-or-email'
[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 #include "branch.h"
27
28 #define FORCED_UPDATES_DELAY_WARNING_IN_MS (10 * 1000)
29
30 static 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
38 enum {
39         TAGS_UNSET = 0,
40         TAGS_DEFAULT = 1,
41         TAGS_SET = 2
42 };
43
44 static int fetch_prune_config = -1; /* unspecified */
45 static int fetch_show_forced_updates = 1;
46 static uint64_t forced_updates_ms = 0;
47 static int prune = -1; /* unspecified */
48 #define PRUNE_BY_DEFAULT 0 /* do we prune by default? */
49
50 static int fetch_prune_tags_config = -1; /* unspecified */
51 static int prune_tags = -1; /* unspecified */
52 #define PRUNE_TAGS_BY_DEFAULT 0 /* do we prune tags by default? */
53
54 static int all, append, dry_run, force, keep, multiple, update_head_ok;
55 static int verbosity, deepen_relative, set_upstream;
56 static int progress = -1;
57 static int enable_auto_gc = 1;
58 static int tags = TAGS_DEFAULT, unshallow, update_shallow, deepen;
59 static int max_children = 1;
60 static enum transport_family family;
61 static const char *depth;
62 static const char *deepen_since;
63 static const char *upload_pack;
64 static struct string_list deepen_not = STRING_LIST_INIT_NODUP;
65 static struct strbuf default_rla = STRBUF_INIT;
66 static struct transport *gtransport;
67 static struct transport *gsecondary;
68 static const char *submodule_prefix = "";
69 static int recurse_submodules = RECURSE_SUBMODULES_DEFAULT;
70 static int recurse_submodules_default = RECURSE_SUBMODULES_ON_DEMAND;
71 static int shown_url = 0;
72 static struct refspec refmap = REFSPEC_INIT_FETCH;
73 static struct list_objects_filter_options filter_options;
74 static struct string_list server_options = STRING_LIST_INIT_DUP;
75 static struct string_list negotiation_tip = STRING_LIST_INIT_NODUP;
76
77 static 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
111 static 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
124 static 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
193 static void unlock_pack(void)
194 {
195         if (gtransport)
196                 transport_unlock_pack(gtransport);
197         if (gsecondary)
198                 transport_unlock_pack(gsecondary);
199 }
200
201 static void unlock_pack_on_signal(int signo)
202 {
203         unlock_pack();
204         sigchain_pop(signo);
205         raise(signo);
206 }
207
208 static 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
246 static 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
257 struct 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
264 static 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
275 static 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
289 static 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
299 static void refname_hash_init(struct hashmap *map)
300 {
301         hashmap_init(map, refname_hash_entry_cmp, NULL, 0);
302 }
303
304 static int refname_hash_exists(struct hashmap *map, const char *refname)
305 {
306         return !!hashmap_get_from_hash(map, strhash(refname), refname);
307 }
308
309 static void clear_item(struct refname_hash_entry *item)
310 {
311         item->ignore = 1;
312 }
313
314 static 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
409 static 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
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
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
1436 static 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
1444 struct remote_group_data {
1445         const char *name;
1446         struct string_list *list;
1447 };
1448
1449 static 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
1468 static 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
1484 static 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
1515 static 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  */
1549 static 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
1596 static 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
1667 int 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 }