shallow: migrate shallow information into the object parser
[git] / remote.c
1 #include "cache.h"
2 #include "config.h"
3 #include "remote.h"
4 #include "refs.h"
5 #include "object-store.h"
6 #include "commit.h"
7 #include "diff.h"
8 #include "revision.h"
9 #include "dir.h"
10 #include "tag.h"
11 #include "string-list.h"
12 #include "mergesort.h"
13 #include "argv-array.h"
14
15 enum map_direction { FROM_SRC, FROM_DST };
16
17 static struct refspec s_tag_refspec = {
18         0,
19         1,
20         0,
21         0,
22         "refs/tags/*",
23         "refs/tags/*"
24 };
25
26 /* See TAG_REFSPEC for the string version */
27 const struct refspec *tag_refspec = &s_tag_refspec;
28
29 struct counted_string {
30         size_t len;
31         const char *s;
32 };
33 struct rewrite {
34         const char *base;
35         size_t baselen;
36         struct counted_string *instead_of;
37         int instead_of_nr;
38         int instead_of_alloc;
39 };
40 struct rewrites {
41         struct rewrite **rewrite;
42         int rewrite_alloc;
43         int rewrite_nr;
44 };
45
46 static struct remote **remotes;
47 static int remotes_alloc;
48 static int remotes_nr;
49 static struct hashmap remotes_hash;
50
51 static struct branch **branches;
52 static int branches_alloc;
53 static int branches_nr;
54
55 static struct branch *current_branch;
56 static const char *pushremote_name;
57
58 static struct rewrites rewrites;
59 static struct rewrites rewrites_push;
60
61 static int valid_remote(const struct remote *remote)
62 {
63         return (!!remote->url) || (!!remote->foreign_vcs);
64 }
65
66 static const char *alias_url(const char *url, struct rewrites *r)
67 {
68         int i, j;
69         struct counted_string *longest;
70         int longest_i;
71
72         longest = NULL;
73         longest_i = -1;
74         for (i = 0; i < r->rewrite_nr; i++) {
75                 if (!r->rewrite[i])
76                         continue;
77                 for (j = 0; j < r->rewrite[i]->instead_of_nr; j++) {
78                         if (starts_with(url, r->rewrite[i]->instead_of[j].s) &&
79                             (!longest ||
80                              longest->len < r->rewrite[i]->instead_of[j].len)) {
81                                 longest = &(r->rewrite[i]->instead_of[j]);
82                                 longest_i = i;
83                         }
84                 }
85         }
86         if (!longest)
87                 return url;
88
89         return xstrfmt("%s%s", r->rewrite[longest_i]->base, url + longest->len);
90 }
91
92 static void add_push_refspec(struct remote *remote, const char *ref)
93 {
94         ALLOC_GROW(remote->push_refspec,
95                    remote->push_refspec_nr + 1,
96                    remote->push_refspec_alloc);
97         remote->push_refspec[remote->push_refspec_nr++] = ref;
98 }
99
100 static void add_fetch_refspec(struct remote *remote, const char *ref)
101 {
102         ALLOC_GROW(remote->fetch_refspec,
103                    remote->fetch_refspec_nr + 1,
104                    remote->fetch_refspec_alloc);
105         remote->fetch_refspec[remote->fetch_refspec_nr++] = ref;
106 }
107
108 void add_prune_tags_to_fetch_refspec(struct remote *remote)
109 {
110         int nr = remote->fetch_refspec_nr;
111         int bufsize = nr  + 1;
112         int size = sizeof(struct refspec);
113
114         remote->fetch = xrealloc(remote->fetch, size  * bufsize);
115         memcpy(&remote->fetch[nr], tag_refspec, size);
116         add_fetch_refspec(remote, xstrdup(TAG_REFSPEC));
117 }
118
119 static void add_url(struct remote *remote, const char *url)
120 {
121         ALLOC_GROW(remote->url, remote->url_nr + 1, remote->url_alloc);
122         remote->url[remote->url_nr++] = url;
123 }
124
125 static void add_pushurl(struct remote *remote, const char *pushurl)
126 {
127         ALLOC_GROW(remote->pushurl, remote->pushurl_nr + 1, remote->pushurl_alloc);
128         remote->pushurl[remote->pushurl_nr++] = pushurl;
129 }
130
131 static void add_pushurl_alias(struct remote *remote, const char *url)
132 {
133         const char *pushurl = alias_url(url, &rewrites_push);
134         if (pushurl != url)
135                 add_pushurl(remote, pushurl);
136 }
137
138 static void add_url_alias(struct remote *remote, const char *url)
139 {
140         add_url(remote, alias_url(url, &rewrites));
141         add_pushurl_alias(remote, url);
142 }
143
144 struct remotes_hash_key {
145         const char *str;
146         int len;
147 };
148
149 static int remotes_hash_cmp(const void *unused_cmp_data,
150                             const void *entry,
151                             const void *entry_or_key,
152                             const void *keydata)
153 {
154         const struct remote *a = entry;
155         const struct remote *b = entry_or_key;
156         const struct remotes_hash_key *key = keydata;
157
158         if (key)
159                 return strncmp(a->name, key->str, key->len) || a->name[key->len];
160         else
161                 return strcmp(a->name, b->name);
162 }
163
164 static inline void init_remotes_hash(void)
165 {
166         if (!remotes_hash.cmpfn)
167                 hashmap_init(&remotes_hash, remotes_hash_cmp, NULL, 0);
168 }
169
170 static struct remote *make_remote(const char *name, int len)
171 {
172         struct remote *ret, *replaced;
173         struct remotes_hash_key lookup;
174         struct hashmap_entry lookup_entry;
175
176         if (!len)
177                 len = strlen(name);
178
179         init_remotes_hash();
180         lookup.str = name;
181         lookup.len = len;
182         hashmap_entry_init(&lookup_entry, memhash(name, len));
183
184         if ((ret = hashmap_get(&remotes_hash, &lookup_entry, &lookup)) != NULL)
185                 return ret;
186
187         ret = xcalloc(1, sizeof(struct remote));
188         ret->prune = -1;  /* unspecified */
189         ret->prune_tags = -1;  /* unspecified */
190         ALLOC_GROW(remotes, remotes_nr + 1, remotes_alloc);
191         remotes[remotes_nr++] = ret;
192         ret->name = xstrndup(name, len);
193
194         hashmap_entry_init(ret, lookup_entry.hash);
195         replaced = hashmap_put(&remotes_hash, ret);
196         assert(replaced == NULL);  /* no previous entry overwritten */
197         return ret;
198 }
199
200 static void add_merge(struct branch *branch, const char *name)
201 {
202         ALLOC_GROW(branch->merge_name, branch->merge_nr + 1,
203                    branch->merge_alloc);
204         branch->merge_name[branch->merge_nr++] = name;
205 }
206
207 static struct branch *make_branch(const char *name, int len)
208 {
209         struct branch *ret;
210         int i;
211
212         for (i = 0; i < branches_nr; i++) {
213                 if (len ? (!strncmp(name, branches[i]->name, len) &&
214                            !branches[i]->name[len]) :
215                     !strcmp(name, branches[i]->name))
216                         return branches[i];
217         }
218
219         ALLOC_GROW(branches, branches_nr + 1, branches_alloc);
220         ret = xcalloc(1, sizeof(struct branch));
221         branches[branches_nr++] = ret;
222         if (len)
223                 ret->name = xstrndup(name, len);
224         else
225                 ret->name = xstrdup(name);
226         ret->refname = xstrfmt("refs/heads/%s", ret->name);
227
228         return ret;
229 }
230
231 static struct rewrite *make_rewrite(struct rewrites *r, const char *base, int len)
232 {
233         struct rewrite *ret;
234         int i;
235
236         for (i = 0; i < r->rewrite_nr; i++) {
237                 if (len
238                     ? (len == r->rewrite[i]->baselen &&
239                        !strncmp(base, r->rewrite[i]->base, len))
240                     : !strcmp(base, r->rewrite[i]->base))
241                         return r->rewrite[i];
242         }
243
244         ALLOC_GROW(r->rewrite, r->rewrite_nr + 1, r->rewrite_alloc);
245         ret = xcalloc(1, sizeof(struct rewrite));
246         r->rewrite[r->rewrite_nr++] = ret;
247         if (len) {
248                 ret->base = xstrndup(base, len);
249                 ret->baselen = len;
250         }
251         else {
252                 ret->base = xstrdup(base);
253                 ret->baselen = strlen(base);
254         }
255         return ret;
256 }
257
258 static void add_instead_of(struct rewrite *rewrite, const char *instead_of)
259 {
260         ALLOC_GROW(rewrite->instead_of, rewrite->instead_of_nr + 1, rewrite->instead_of_alloc);
261         rewrite->instead_of[rewrite->instead_of_nr].s = instead_of;
262         rewrite->instead_of[rewrite->instead_of_nr].len = strlen(instead_of);
263         rewrite->instead_of_nr++;
264 }
265
266 static const char *skip_spaces(const char *s)
267 {
268         while (isspace(*s))
269                 s++;
270         return s;
271 }
272
273 static void read_remotes_file(struct remote *remote)
274 {
275         struct strbuf buf = STRBUF_INIT;
276         FILE *f = fopen_or_warn(git_path("remotes/%s", remote->name), "r");
277
278         if (!f)
279                 return;
280         remote->configured_in_repo = 1;
281         remote->origin = REMOTE_REMOTES;
282         while (strbuf_getline(&buf, f) != EOF) {
283                 const char *v;
284
285                 strbuf_rtrim(&buf);
286
287                 if (skip_prefix(buf.buf, "URL:", &v))
288                         add_url_alias(remote, xstrdup(skip_spaces(v)));
289                 else if (skip_prefix(buf.buf, "Push:", &v))
290                         add_push_refspec(remote, xstrdup(skip_spaces(v)));
291                 else if (skip_prefix(buf.buf, "Pull:", &v))
292                         add_fetch_refspec(remote, xstrdup(skip_spaces(v)));
293         }
294         strbuf_release(&buf);
295         fclose(f);
296 }
297
298 static void read_branches_file(struct remote *remote)
299 {
300         char *frag;
301         struct strbuf buf = STRBUF_INIT;
302         FILE *f = fopen_or_warn(git_path("branches/%s", remote->name), "r");
303
304         if (!f)
305                 return;
306
307         strbuf_getline_lf(&buf, f);
308         fclose(f);
309         strbuf_trim(&buf);
310         if (!buf.len) {
311                 strbuf_release(&buf);
312                 return;
313         }
314
315         remote->configured_in_repo = 1;
316         remote->origin = REMOTE_BRANCHES;
317
318         /*
319          * The branches file would have URL and optionally
320          * #branch specified.  The "master" (or specified) branch is
321          * fetched and stored in the local branch matching the
322          * remote name.
323          */
324         frag = strchr(buf.buf, '#');
325         if (frag)
326                 *(frag++) = '\0';
327         else
328                 frag = "master";
329
330         add_url_alias(remote, strbuf_detach(&buf, NULL));
331         add_fetch_refspec(remote, xstrfmt("refs/heads/%s:refs/heads/%s",
332                                           frag, remote->name));
333
334         /*
335          * Cogito compatible push: push current HEAD to remote #branch
336          * (master if missing)
337          */
338         add_push_refspec(remote, xstrfmt("HEAD:refs/heads/%s", frag));
339         remote->fetch_tags = 1; /* always auto-follow */
340 }
341
342 static int handle_config(const char *key, const char *value, void *cb)
343 {
344         const char *name;
345         int namelen;
346         const char *subkey;
347         struct remote *remote;
348         struct branch *branch;
349         if (parse_config_key(key, "branch", &name, &namelen, &subkey) >= 0) {
350                 if (!name)
351                         return 0;
352                 branch = make_branch(name, namelen);
353                 if (!strcmp(subkey, "remote")) {
354                         return git_config_string(&branch->remote_name, key, value);
355                 } else if (!strcmp(subkey, "pushremote")) {
356                         return git_config_string(&branch->pushremote_name, key, value);
357                 } else if (!strcmp(subkey, "merge")) {
358                         if (!value)
359                                 return config_error_nonbool(key);
360                         add_merge(branch, xstrdup(value));
361                 }
362                 return 0;
363         }
364         if (parse_config_key(key, "url", &name, &namelen, &subkey) >= 0) {
365                 struct rewrite *rewrite;
366                 if (!name)
367                         return 0;
368                 if (!strcmp(subkey, "insteadof")) {
369                         rewrite = make_rewrite(&rewrites, name, namelen);
370                         if (!value)
371                                 return config_error_nonbool(key);
372                         add_instead_of(rewrite, xstrdup(value));
373                 } else if (!strcmp(subkey, "pushinsteadof")) {
374                         rewrite = make_rewrite(&rewrites_push, name, namelen);
375                         if (!value)
376                                 return config_error_nonbool(key);
377                         add_instead_of(rewrite, xstrdup(value));
378                 }
379         }
380
381         if (parse_config_key(key, "remote", &name, &namelen, &subkey) < 0)
382                 return 0;
383
384         /* Handle remote.* variables */
385         if (!name && !strcmp(subkey, "pushdefault"))
386                 return git_config_string(&pushremote_name, key, value);
387
388         if (!name)
389                 return 0;
390         /* Handle remote.<name>.* variables */
391         if (*name == '/') {
392                 warning("Config remote shorthand cannot begin with '/': %s",
393                         name);
394                 return 0;
395         }
396         remote = make_remote(name, namelen);
397         remote->origin = REMOTE_CONFIG;
398         if (current_config_scope() == CONFIG_SCOPE_REPO)
399                 remote->configured_in_repo = 1;
400         if (!strcmp(subkey, "mirror"))
401                 remote->mirror = git_config_bool(key, value);
402         else if (!strcmp(subkey, "skipdefaultupdate"))
403                 remote->skip_default_update = git_config_bool(key, value);
404         else if (!strcmp(subkey, "skipfetchall"))
405                 remote->skip_default_update = git_config_bool(key, value);
406         else if (!strcmp(subkey, "prune"))
407                 remote->prune = git_config_bool(key, value);
408         else if (!strcmp(subkey, "prunetags"))
409                 remote->prune_tags = git_config_bool(key, value);
410         else if (!strcmp(subkey, "url")) {
411                 const char *v;
412                 if (git_config_string(&v, key, value))
413                         return -1;
414                 add_url(remote, v);
415         } else if (!strcmp(subkey, "pushurl")) {
416                 const char *v;
417                 if (git_config_string(&v, key, value))
418                         return -1;
419                 add_pushurl(remote, v);
420         } else if (!strcmp(subkey, "push")) {
421                 const char *v;
422                 if (git_config_string(&v, key, value))
423                         return -1;
424                 add_push_refspec(remote, v);
425         } else if (!strcmp(subkey, "fetch")) {
426                 const char *v;
427                 if (git_config_string(&v, key, value))
428                         return -1;
429                 add_fetch_refspec(remote, v);
430         } else if (!strcmp(subkey, "receivepack")) {
431                 const char *v;
432                 if (git_config_string(&v, key, value))
433                         return -1;
434                 if (!remote->receivepack)
435                         remote->receivepack = v;
436                 else
437                         error("more than one receivepack given, using the first");
438         } else if (!strcmp(subkey, "uploadpack")) {
439                 const char *v;
440                 if (git_config_string(&v, key, value))
441                         return -1;
442                 if (!remote->uploadpack)
443                         remote->uploadpack = v;
444                 else
445                         error("more than one uploadpack given, using the first");
446         } else if (!strcmp(subkey, "tagopt")) {
447                 if (!strcmp(value, "--no-tags"))
448                         remote->fetch_tags = -1;
449                 else if (!strcmp(value, "--tags"))
450                         remote->fetch_tags = 2;
451         } else if (!strcmp(subkey, "proxy")) {
452                 return git_config_string((const char **)&remote->http_proxy,
453                                          key, value);
454         } else if (!strcmp(subkey, "proxyauthmethod")) {
455                 return git_config_string((const char **)&remote->http_proxy_authmethod,
456                                          key, value);
457         } else if (!strcmp(subkey, "vcs")) {
458                 return git_config_string(&remote->foreign_vcs, key, value);
459         }
460         return 0;
461 }
462
463 static void alias_all_urls(void)
464 {
465         int i, j;
466         for (i = 0; i < remotes_nr; i++) {
467                 int add_pushurl_aliases;
468                 if (!remotes[i])
469                         continue;
470                 for (j = 0; j < remotes[i]->pushurl_nr; j++) {
471                         remotes[i]->pushurl[j] = alias_url(remotes[i]->pushurl[j], &rewrites);
472                 }
473                 add_pushurl_aliases = remotes[i]->pushurl_nr == 0;
474                 for (j = 0; j < remotes[i]->url_nr; j++) {
475                         if (add_pushurl_aliases)
476                                 add_pushurl_alias(remotes[i], remotes[i]->url[j]);
477                         remotes[i]->url[j] = alias_url(remotes[i]->url[j], &rewrites);
478                 }
479         }
480 }
481
482 static void read_config(void)
483 {
484         static int loaded;
485         int flag;
486
487         if (loaded)
488                 return;
489         loaded = 1;
490
491         current_branch = NULL;
492         if (startup_info->have_repository) {
493                 const char *head_ref = resolve_ref_unsafe("HEAD", 0, NULL, &flag);
494                 if (head_ref && (flag & REF_ISSYMREF) &&
495                     skip_prefix(head_ref, "refs/heads/", &head_ref)) {
496                         current_branch = make_branch(head_ref, 0);
497                 }
498         }
499         git_config(handle_config, NULL);
500         alias_all_urls();
501 }
502
503 static struct refspec *parse_refspec_internal(int nr_refspec, const char **refspec, int fetch, int verify)
504 {
505         int i;
506         struct refspec *rs = xcalloc(nr_refspec, sizeof(*rs));
507
508         for (i = 0; i < nr_refspec; i++) {
509                 size_t llen;
510                 int is_glob;
511                 const char *lhs, *rhs;
512                 int flags;
513
514                 is_glob = 0;
515
516                 lhs = refspec[i];
517                 if (*lhs == '+') {
518                         rs[i].force = 1;
519                         lhs++;
520                 }
521
522                 rhs = strrchr(lhs, ':');
523
524                 /*
525                  * Before going on, special case ":" (or "+:") as a refspec
526                  * for pushing matching refs.
527                  */
528                 if (!fetch && rhs == lhs && rhs[1] == '\0') {
529                         rs[i].matching = 1;
530                         continue;
531                 }
532
533                 if (rhs) {
534                         size_t rlen = strlen(++rhs);
535                         is_glob = (1 <= rlen && strchr(rhs, '*'));
536                         rs[i].dst = xstrndup(rhs, rlen);
537                 }
538
539                 llen = (rhs ? (rhs - lhs - 1) : strlen(lhs));
540                 if (1 <= llen && memchr(lhs, '*', llen)) {
541                         if ((rhs && !is_glob) || (!rhs && fetch))
542                                 goto invalid;
543                         is_glob = 1;
544                 } else if (rhs && is_glob) {
545                         goto invalid;
546                 }
547
548                 rs[i].pattern = is_glob;
549                 rs[i].src = xstrndup(lhs, llen);
550                 flags = REFNAME_ALLOW_ONELEVEL | (is_glob ? REFNAME_REFSPEC_PATTERN : 0);
551
552                 if (fetch) {
553                         struct object_id unused;
554
555                         /* LHS */
556                         if (!*rs[i].src)
557                                 ; /* empty is ok; it means "HEAD" */
558                         else if (llen == GIT_SHA1_HEXSZ && !get_oid_hex(rs[i].src, &unused))
559                                 rs[i].exact_sha1 = 1; /* ok */
560                         else if (!check_refname_format(rs[i].src, flags))
561                                 ; /* valid looking ref is ok */
562                         else
563                                 goto invalid;
564                         /* RHS */
565                         if (!rs[i].dst)
566                                 ; /* missing is ok; it is the same as empty */
567                         else if (!*rs[i].dst)
568                                 ; /* empty is ok; it means "do not store" */
569                         else if (!check_refname_format(rs[i].dst, flags))
570                                 ; /* valid looking ref is ok */
571                         else
572                                 goto invalid;
573                 } else {
574                         /*
575                          * LHS
576                          * - empty is allowed; it means delete.
577                          * - when wildcarded, it must be a valid looking ref.
578                          * - otherwise, it must be an extended SHA-1, but
579                          *   there is no existing way to validate this.
580                          */
581                         if (!*rs[i].src)
582                                 ; /* empty is ok */
583                         else if (is_glob) {
584                                 if (check_refname_format(rs[i].src, flags))
585                                         goto invalid;
586                         }
587                         else
588                                 ; /* anything goes, for now */
589                         /*
590                          * RHS
591                          * - missing is allowed, but LHS then must be a
592                          *   valid looking ref.
593                          * - empty is not allowed.
594                          * - otherwise it must be a valid looking ref.
595                          */
596                         if (!rs[i].dst) {
597                                 if (check_refname_format(rs[i].src, flags))
598                                         goto invalid;
599                         } else if (!*rs[i].dst) {
600                                 goto invalid;
601                         } else {
602                                 if (check_refname_format(rs[i].dst, flags))
603                                         goto invalid;
604                         }
605                 }
606         }
607         return rs;
608
609  invalid:
610         if (verify) {
611                 /*
612                  * nr_refspec must be greater than zero and i must be valid
613                  * since it is only possible to reach this point from within
614                  * the for loop above.
615                  */
616                 free_refspec(i+1, rs);
617                 return NULL;
618         }
619         die("Invalid refspec '%s'", refspec[i]);
620 }
621
622 int valid_fetch_refspec(const char *fetch_refspec_str)
623 {
624         struct refspec *refspec;
625
626         refspec = parse_refspec_internal(1, &fetch_refspec_str, 1, 1);
627         free_refspec(1, refspec);
628         return !!refspec;
629 }
630
631 struct refspec *parse_fetch_refspec(int nr_refspec, const char **refspec)
632 {
633         return parse_refspec_internal(nr_refspec, refspec, 1, 0);
634 }
635
636 struct refspec *parse_push_refspec(int nr_refspec, const char **refspec)
637 {
638         return parse_refspec_internal(nr_refspec, refspec, 0, 0);
639 }
640
641 void free_refspec(int nr_refspec, struct refspec *refspec)
642 {
643         int i;
644
645         if (!refspec)
646                 return;
647
648         for (i = 0; i < nr_refspec; i++) {
649                 free(refspec[i].src);
650                 free(refspec[i].dst);
651         }
652         free(refspec);
653 }
654
655 static int valid_remote_nick(const char *name)
656 {
657         if (!name[0] || is_dot_or_dotdot(name))
658                 return 0;
659
660         /* remote nicknames cannot contain slashes */
661         while (*name)
662                 if (is_dir_sep(*name++))
663                         return 0;
664         return 1;
665 }
666
667 const char *remote_for_branch(struct branch *branch, int *explicit)
668 {
669         if (branch && branch->remote_name) {
670                 if (explicit)
671                         *explicit = 1;
672                 return branch->remote_name;
673         }
674         if (explicit)
675                 *explicit = 0;
676         return "origin";
677 }
678
679 const char *pushremote_for_branch(struct branch *branch, int *explicit)
680 {
681         if (branch && branch->pushremote_name) {
682                 if (explicit)
683                         *explicit = 1;
684                 return branch->pushremote_name;
685         }
686         if (pushremote_name) {
687                 if (explicit)
688                         *explicit = 1;
689                 return pushremote_name;
690         }
691         return remote_for_branch(branch, explicit);
692 }
693
694 const char *remote_ref_for_branch(struct branch *branch, int for_push,
695                                   int *explicit)
696 {
697         if (branch) {
698                 if (!for_push) {
699                         if (branch->merge_nr) {
700                                 if (explicit)
701                                         *explicit = 1;
702                                 return branch->merge_name[0];
703                         }
704                 } else {
705                         const char *dst, *remote_name =
706                                 pushremote_for_branch(branch, NULL);
707                         struct remote *remote = remote_get(remote_name);
708
709                         if (remote && remote->push_refspec_nr &&
710                             (dst = apply_refspecs(remote->push,
711                                                   remote->push_refspec_nr,
712                                                   branch->refname))) {
713                                 if (explicit)
714                                         *explicit = 1;
715                                 return dst;
716                         }
717                 }
718         }
719         if (explicit)
720                 *explicit = 0;
721         return "";
722 }
723
724 static struct remote *remote_get_1(const char *name,
725                                    const char *(*get_default)(struct branch *, int *))
726 {
727         struct remote *ret;
728         int name_given = 0;
729
730         read_config();
731
732         if (name)
733                 name_given = 1;
734         else
735                 name = get_default(current_branch, &name_given);
736
737         ret = make_remote(name, 0);
738         if (valid_remote_nick(name) && have_git_dir()) {
739                 if (!valid_remote(ret))
740                         read_remotes_file(ret);
741                 if (!valid_remote(ret))
742                         read_branches_file(ret);
743         }
744         if (name_given && !valid_remote(ret))
745                 add_url_alias(ret, name);
746         if (!valid_remote(ret))
747                 return NULL;
748         ret->fetch = parse_fetch_refspec(ret->fetch_refspec_nr, ret->fetch_refspec);
749         ret->push = parse_push_refspec(ret->push_refspec_nr, ret->push_refspec);
750         return ret;
751 }
752
753 struct remote *remote_get(const char *name)
754 {
755         return remote_get_1(name, remote_for_branch);
756 }
757
758 struct remote *pushremote_get(const char *name)
759 {
760         return remote_get_1(name, pushremote_for_branch);
761 }
762
763 int remote_is_configured(struct remote *remote, int in_repo)
764 {
765         if (!remote)
766                 return 0;
767         if (in_repo)
768                 return remote->configured_in_repo;
769         return !!remote->origin;
770 }
771
772 int for_each_remote(each_remote_fn fn, void *priv)
773 {
774         int i, result = 0;
775         read_config();
776         for (i = 0; i < remotes_nr && !result; i++) {
777                 struct remote *r = remotes[i];
778                 if (!r)
779                         continue;
780                 if (!r->fetch)
781                         r->fetch = parse_fetch_refspec(r->fetch_refspec_nr,
782                                                        r->fetch_refspec);
783                 if (!r->push)
784                         r->push = parse_push_refspec(r->push_refspec_nr,
785                                                      r->push_refspec);
786                 result = fn(r, priv);
787         }
788         return result;
789 }
790
791 static void handle_duplicate(struct ref *ref1, struct ref *ref2)
792 {
793         if (strcmp(ref1->name, ref2->name)) {
794                 if (ref1->fetch_head_status != FETCH_HEAD_IGNORE &&
795                     ref2->fetch_head_status != FETCH_HEAD_IGNORE) {
796                         die(_("Cannot fetch both %s and %s to %s"),
797                             ref1->name, ref2->name, ref2->peer_ref->name);
798                 } else if (ref1->fetch_head_status != FETCH_HEAD_IGNORE &&
799                            ref2->fetch_head_status == FETCH_HEAD_IGNORE) {
800                         warning(_("%s usually tracks %s, not %s"),
801                                 ref2->peer_ref->name, ref2->name, ref1->name);
802                 } else if (ref1->fetch_head_status == FETCH_HEAD_IGNORE &&
803                            ref2->fetch_head_status == FETCH_HEAD_IGNORE) {
804                         die(_("%s tracks both %s and %s"),
805                             ref2->peer_ref->name, ref1->name, ref2->name);
806                 } else {
807                         /*
808                          * This last possibility doesn't occur because
809                          * FETCH_HEAD_IGNORE entries always appear at
810                          * the end of the list.
811                          */
812                         die(_("Internal error"));
813                 }
814         }
815         free(ref2->peer_ref);
816         free(ref2);
817 }
818
819 struct ref *ref_remove_duplicates(struct ref *ref_map)
820 {
821         struct string_list refs = STRING_LIST_INIT_NODUP;
822         struct ref *retval = NULL;
823         struct ref **p = &retval;
824
825         while (ref_map) {
826                 struct ref *ref = ref_map;
827
828                 ref_map = ref_map->next;
829                 ref->next = NULL;
830
831                 if (!ref->peer_ref) {
832                         *p = ref;
833                         p = &ref->next;
834                 } else {
835                         struct string_list_item *item =
836                                 string_list_insert(&refs, ref->peer_ref->name);
837
838                         if (item->util) {
839                                 /* Entry already existed */
840                                 handle_duplicate((struct ref *)item->util, ref);
841                         } else {
842                                 *p = ref;
843                                 p = &ref->next;
844                                 item->util = ref;
845                         }
846                 }
847         }
848
849         string_list_clear(&refs, 0);
850         return retval;
851 }
852
853 int remote_has_url(struct remote *remote, const char *url)
854 {
855         int i;
856         for (i = 0; i < remote->url_nr; i++) {
857                 if (!strcmp(remote->url[i], url))
858                         return 1;
859         }
860         return 0;
861 }
862
863 static int match_name_with_pattern(const char *key, const char *name,
864                                    const char *value, char **result)
865 {
866         const char *kstar = strchr(key, '*');
867         size_t klen;
868         size_t ksuffixlen;
869         size_t namelen;
870         int ret;
871         if (!kstar)
872                 die("Key '%s' of pattern had no '*'", key);
873         klen = kstar - key;
874         ksuffixlen = strlen(kstar + 1);
875         namelen = strlen(name);
876         ret = !strncmp(name, key, klen) && namelen >= klen + ksuffixlen &&
877                 !memcmp(name + namelen - ksuffixlen, kstar + 1, ksuffixlen);
878         if (ret && value) {
879                 struct strbuf sb = STRBUF_INIT;
880                 const char *vstar = strchr(value, '*');
881                 if (!vstar)
882                         die("Value '%s' of pattern has no '*'", value);
883                 strbuf_add(&sb, value, vstar - value);
884                 strbuf_add(&sb, name + klen, namelen - klen - ksuffixlen);
885                 strbuf_addstr(&sb, vstar + 1);
886                 *result = strbuf_detach(&sb, NULL);
887         }
888         return ret;
889 }
890
891 static void query_refspecs_multiple(struct refspec *refs, int ref_count, struct refspec *query, struct string_list *results)
892 {
893         int i;
894         int find_src = !query->src;
895
896         if (find_src && !query->dst)
897                 error("query_refspecs_multiple: need either src or dst");
898
899         for (i = 0; i < ref_count; i++) {
900                 struct refspec *refspec = &refs[i];
901                 const char *key = find_src ? refspec->dst : refspec->src;
902                 const char *value = find_src ? refspec->src : refspec->dst;
903                 const char *needle = find_src ? query->dst : query->src;
904                 char **result = find_src ? &query->src : &query->dst;
905
906                 if (!refspec->dst)
907                         continue;
908                 if (refspec->pattern) {
909                         if (match_name_with_pattern(key, needle, value, result))
910                                 string_list_append_nodup(results, *result);
911                 } else if (!strcmp(needle, key)) {
912                         string_list_append(results, value);
913                 }
914         }
915 }
916
917 int query_refspecs(struct refspec *refs, int ref_count, struct refspec *query)
918 {
919         int i;
920         int find_src = !query->src;
921         const char *needle = find_src ? query->dst : query->src;
922         char **result = find_src ? &query->src : &query->dst;
923
924         if (find_src && !query->dst)
925                 return error("query_refspecs: need either src or dst");
926
927         for (i = 0; i < ref_count; i++) {
928                 struct refspec *refspec = &refs[i];
929                 const char *key = find_src ? refspec->dst : refspec->src;
930                 const char *value = find_src ? refspec->src : refspec->dst;
931
932                 if (!refspec->dst)
933                         continue;
934                 if (refspec->pattern) {
935                         if (match_name_with_pattern(key, needle, value, result)) {
936                                 query->force = refspec->force;
937                                 return 0;
938                         }
939                 } else if (!strcmp(needle, key)) {
940                         *result = xstrdup(value);
941                         query->force = refspec->force;
942                         return 0;
943                 }
944         }
945         return -1;
946 }
947
948 char *apply_refspecs(struct refspec *refspecs, int nr_refspec,
949                      const char *name)
950 {
951         struct refspec query;
952
953         memset(&query, 0, sizeof(struct refspec));
954         query.src = (char *)name;
955
956         if (query_refspecs(refspecs, nr_refspec, &query))
957                 return NULL;
958
959         return query.dst;
960 }
961
962 int remote_find_tracking(struct remote *remote, struct refspec *refspec)
963 {
964         return query_refspecs(remote->fetch, remote->fetch_refspec_nr, refspec);
965 }
966
967 static struct ref *alloc_ref_with_prefix(const char *prefix, size_t prefixlen,
968                 const char *name)
969 {
970         size_t len = strlen(name);
971         struct ref *ref = xcalloc(1, st_add4(sizeof(*ref), prefixlen, len, 1));
972         memcpy(ref->name, prefix, prefixlen);
973         memcpy(ref->name + prefixlen, name, len);
974         return ref;
975 }
976
977 struct ref *alloc_ref(const char *name)
978 {
979         return alloc_ref_with_prefix("", 0, name);
980 }
981
982 struct ref *copy_ref(const struct ref *ref)
983 {
984         struct ref *cpy;
985         size_t len;
986         if (!ref)
987                 return NULL;
988         len = st_add3(sizeof(struct ref), strlen(ref->name), 1);
989         cpy = xmalloc(len);
990         memcpy(cpy, ref, len);
991         cpy->next = NULL;
992         cpy->symref = xstrdup_or_null(ref->symref);
993         cpy->remote_status = xstrdup_or_null(ref->remote_status);
994         cpy->peer_ref = copy_ref(ref->peer_ref);
995         return cpy;
996 }
997
998 struct ref *copy_ref_list(const struct ref *ref)
999 {
1000         struct ref *ret = NULL;
1001         struct ref **tail = &ret;
1002         while (ref) {
1003                 *tail = copy_ref(ref);
1004                 ref = ref->next;
1005                 tail = &((*tail)->next);
1006         }
1007         return ret;
1008 }
1009
1010 static void free_ref(struct ref *ref)
1011 {
1012         if (!ref)
1013                 return;
1014         free_ref(ref->peer_ref);
1015         free(ref->remote_status);
1016         free(ref->symref);
1017         free(ref);
1018 }
1019
1020 void free_refs(struct ref *ref)
1021 {
1022         struct ref *next;
1023         while (ref) {
1024                 next = ref->next;
1025                 free_ref(ref);
1026                 ref = next;
1027         }
1028 }
1029
1030 int ref_compare_name(const void *va, const void *vb)
1031 {
1032         const struct ref *a = va, *b = vb;
1033         return strcmp(a->name, b->name);
1034 }
1035
1036 static void *ref_list_get_next(const void *a)
1037 {
1038         return ((const struct ref *)a)->next;
1039 }
1040
1041 static void ref_list_set_next(void *a, void *next)
1042 {
1043         ((struct ref *)a)->next = next;
1044 }
1045
1046 void sort_ref_list(struct ref **l, int (*cmp)(const void *, const void *))
1047 {
1048         *l = llist_mergesort(*l, ref_list_get_next, ref_list_set_next, cmp);
1049 }
1050
1051 int count_refspec_match(const char *pattern,
1052                         struct ref *refs,
1053                         struct ref **matched_ref)
1054 {
1055         int patlen = strlen(pattern);
1056         struct ref *matched_weak = NULL;
1057         struct ref *matched = NULL;
1058         int weak_match = 0;
1059         int match = 0;
1060
1061         for (weak_match = match = 0; refs; refs = refs->next) {
1062                 char *name = refs->name;
1063                 int namelen = strlen(name);
1064
1065                 if (!refname_match(pattern, name))
1066                         continue;
1067
1068                 /* A match is "weak" if it is with refs outside
1069                  * heads or tags, and did not specify the pattern
1070                  * in full (e.g. "refs/remotes/origin/master") or at
1071                  * least from the toplevel (e.g. "remotes/origin/master");
1072                  * otherwise "git push $URL master" would result in
1073                  * ambiguity between remotes/origin/master and heads/master
1074                  * at the remote site.
1075                  */
1076                 if (namelen != patlen &&
1077                     patlen != namelen - 5 &&
1078                     !starts_with(name, "refs/heads/") &&
1079                     !starts_with(name, "refs/tags/")) {
1080                         /* We want to catch the case where only weak
1081                          * matches are found and there are multiple
1082                          * matches, and where more than one strong
1083                          * matches are found, as ambiguous.  One
1084                          * strong match with zero or more weak matches
1085                          * are acceptable as a unique match.
1086                          */
1087                         matched_weak = refs;
1088                         weak_match++;
1089                 }
1090                 else {
1091                         matched = refs;
1092                         match++;
1093                 }
1094         }
1095         if (!matched) {
1096                 if (matched_ref)
1097                         *matched_ref = matched_weak;
1098                 return weak_match;
1099         }
1100         else {
1101                 if (matched_ref)
1102                         *matched_ref = matched;
1103                 return match;
1104         }
1105 }
1106
1107 static void tail_link_ref(struct ref *ref, struct ref ***tail)
1108 {
1109         **tail = ref;
1110         while (ref->next)
1111                 ref = ref->next;
1112         *tail = &ref->next;
1113 }
1114
1115 static struct ref *alloc_delete_ref(void)
1116 {
1117         struct ref *ref = alloc_ref("(delete)");
1118         oidclr(&ref->new_oid);
1119         return ref;
1120 }
1121
1122 static int try_explicit_object_name(const char *name,
1123                                     struct ref **match)
1124 {
1125         struct object_id oid;
1126
1127         if (!*name) {
1128                 if (match)
1129                         *match = alloc_delete_ref();
1130                 return 0;
1131         }
1132
1133         if (get_oid(name, &oid))
1134                 return -1;
1135
1136         if (match) {
1137                 *match = alloc_ref(name);
1138                 oidcpy(&(*match)->new_oid, &oid);
1139         }
1140         return 0;
1141 }
1142
1143 static struct ref *make_linked_ref(const char *name, struct ref ***tail)
1144 {
1145         struct ref *ret = alloc_ref(name);
1146         tail_link_ref(ret, tail);
1147         return ret;
1148 }
1149
1150 static char *guess_ref(const char *name, struct ref *peer)
1151 {
1152         struct strbuf buf = STRBUF_INIT;
1153
1154         const char *r = resolve_ref_unsafe(peer->name, RESOLVE_REF_READING,
1155                                            NULL, NULL);
1156         if (!r)
1157                 return NULL;
1158
1159         if (starts_with(r, "refs/heads/"))
1160                 strbuf_addstr(&buf, "refs/heads/");
1161         else if (starts_with(r, "refs/tags/"))
1162                 strbuf_addstr(&buf, "refs/tags/");
1163         else
1164                 return NULL;
1165
1166         strbuf_addstr(&buf, name);
1167         return strbuf_detach(&buf, NULL);
1168 }
1169
1170 static int match_explicit_lhs(struct ref *src,
1171                               struct refspec *rs,
1172                               struct ref **match,
1173                               int *allocated_match)
1174 {
1175         switch (count_refspec_match(rs->src, src, match)) {
1176         case 1:
1177                 if (allocated_match)
1178                         *allocated_match = 0;
1179                 return 0;
1180         case 0:
1181                 /* The source could be in the get_sha1() format
1182                  * not a reference name.  :refs/other is a
1183                  * way to delete 'other' ref at the remote end.
1184                  */
1185                 if (try_explicit_object_name(rs->src, match) < 0)
1186                         return error("src refspec %s does not match any.", rs->src);
1187                 if (allocated_match)
1188                         *allocated_match = 1;
1189                 return 0;
1190         default:
1191                 return error("src refspec %s matches more than one.", rs->src);
1192         }
1193 }
1194
1195 static int match_explicit(struct ref *src, struct ref *dst,
1196                           struct ref ***dst_tail,
1197                           struct refspec *rs)
1198 {
1199         struct ref *matched_src, *matched_dst;
1200         int allocated_src;
1201
1202         const char *dst_value = rs->dst;
1203         char *dst_guess;
1204
1205         if (rs->pattern || rs->matching)
1206                 return 0;
1207
1208         matched_src = matched_dst = NULL;
1209         if (match_explicit_lhs(src, rs, &matched_src, &allocated_src) < 0)
1210                 return -1;
1211
1212         if (!dst_value) {
1213                 int flag;
1214
1215                 dst_value = resolve_ref_unsafe(matched_src->name,
1216                                                RESOLVE_REF_READING,
1217                                                NULL, &flag);
1218                 if (!dst_value ||
1219                     ((flag & REF_ISSYMREF) &&
1220                      !starts_with(dst_value, "refs/heads/")))
1221                         die("%s cannot be resolved to branch.",
1222                             matched_src->name);
1223         }
1224
1225         switch (count_refspec_match(dst_value, dst, &matched_dst)) {
1226         case 1:
1227                 break;
1228         case 0:
1229                 if (starts_with(dst_value, "refs/"))
1230                         matched_dst = make_linked_ref(dst_value, dst_tail);
1231                 else if (is_null_oid(&matched_src->new_oid))
1232                         error("unable to delete '%s': remote ref does not exist",
1233                               dst_value);
1234                 else if ((dst_guess = guess_ref(dst_value, matched_src))) {
1235                         matched_dst = make_linked_ref(dst_guess, dst_tail);
1236                         free(dst_guess);
1237                 } else
1238                         error("unable to push to unqualified destination: %s\n"
1239                               "The destination refspec neither matches an "
1240                               "existing ref on the remote nor\n"
1241                               "begins with refs/, and we are unable to "
1242                               "guess a prefix based on the source ref.",
1243                               dst_value);
1244                 break;
1245         default:
1246                 matched_dst = NULL;
1247                 error("dst refspec %s matches more than one.",
1248                       dst_value);
1249                 break;
1250         }
1251         if (!matched_dst)
1252                 return -1;
1253         if (matched_dst->peer_ref)
1254                 return error("dst ref %s receives from more than one src.",
1255                       matched_dst->name);
1256         else {
1257                 matched_dst->peer_ref = allocated_src ?
1258                                         matched_src :
1259                                         copy_ref(matched_src);
1260                 matched_dst->force = rs->force;
1261         }
1262         return 0;
1263 }
1264
1265 static int match_explicit_refs(struct ref *src, struct ref *dst,
1266                                struct ref ***dst_tail, struct refspec *rs,
1267                                int rs_nr)
1268 {
1269         int i, errs;
1270         for (i = errs = 0; i < rs_nr; i++)
1271                 errs += match_explicit(src, dst, dst_tail, &rs[i]);
1272         return errs;
1273 }
1274
1275 static char *get_ref_match(const struct refspec *rs, int rs_nr, const struct ref *ref,
1276                 int send_mirror, int direction, const struct refspec **ret_pat)
1277 {
1278         const struct refspec *pat;
1279         char *name;
1280         int i;
1281         int matching_refs = -1;
1282         for (i = 0; i < rs_nr; i++) {
1283                 if (rs[i].matching &&
1284                     (matching_refs == -1 || rs[i].force)) {
1285                         matching_refs = i;
1286                         continue;
1287                 }
1288
1289                 if (rs[i].pattern) {
1290                         const char *dst_side = rs[i].dst ? rs[i].dst : rs[i].src;
1291                         int match;
1292                         if (direction == FROM_SRC)
1293                                 match = match_name_with_pattern(rs[i].src, ref->name, dst_side, &name);
1294                         else
1295                                 match = match_name_with_pattern(dst_side, ref->name, rs[i].src, &name);
1296                         if (match) {
1297                                 matching_refs = i;
1298                                 break;
1299                         }
1300                 }
1301         }
1302         if (matching_refs == -1)
1303                 return NULL;
1304
1305         pat = rs + matching_refs;
1306         if (pat->matching) {
1307                 /*
1308                  * "matching refs"; traditionally we pushed everything
1309                  * including refs outside refs/heads/ hierarchy, but
1310                  * that does not make much sense these days.
1311                  */
1312                 if (!send_mirror && !starts_with(ref->name, "refs/heads/"))
1313                         return NULL;
1314                 name = xstrdup(ref->name);
1315         }
1316         if (ret_pat)
1317                 *ret_pat = pat;
1318         return name;
1319 }
1320
1321 static struct ref **tail_ref(struct ref **head)
1322 {
1323         struct ref **tail = head;
1324         while (*tail)
1325                 tail = &((*tail)->next);
1326         return tail;
1327 }
1328
1329 struct tips {
1330         struct commit **tip;
1331         int nr, alloc;
1332 };
1333
1334 static void add_to_tips(struct tips *tips, const struct object_id *oid)
1335 {
1336         struct commit *commit;
1337
1338         if (is_null_oid(oid))
1339                 return;
1340         commit = lookup_commit_reference_gently(oid, 1);
1341         if (!commit || (commit->object.flags & TMP_MARK))
1342                 return;
1343         commit->object.flags |= TMP_MARK;
1344         ALLOC_GROW(tips->tip, tips->nr + 1, tips->alloc);
1345         tips->tip[tips->nr++] = commit;
1346 }
1347
1348 static void add_missing_tags(struct ref *src, struct ref **dst, struct ref ***dst_tail)
1349 {
1350         struct string_list dst_tag = STRING_LIST_INIT_NODUP;
1351         struct string_list src_tag = STRING_LIST_INIT_NODUP;
1352         struct string_list_item *item;
1353         struct ref *ref;
1354         struct tips sent_tips;
1355
1356         /*
1357          * Collect everything we know they would have at the end of
1358          * this push, and collect all tags they have.
1359          */
1360         memset(&sent_tips, 0, sizeof(sent_tips));
1361         for (ref = *dst; ref; ref = ref->next) {
1362                 if (ref->peer_ref &&
1363                     !is_null_oid(&ref->peer_ref->new_oid))
1364                         add_to_tips(&sent_tips, &ref->peer_ref->new_oid);
1365                 else
1366                         add_to_tips(&sent_tips, &ref->old_oid);
1367                 if (starts_with(ref->name, "refs/tags/"))
1368                         string_list_append(&dst_tag, ref->name);
1369         }
1370         clear_commit_marks_many(sent_tips.nr, sent_tips.tip, TMP_MARK);
1371
1372         string_list_sort(&dst_tag);
1373
1374         /* Collect tags they do not have. */
1375         for (ref = src; ref; ref = ref->next) {
1376                 if (!starts_with(ref->name, "refs/tags/"))
1377                         continue; /* not a tag */
1378                 if (string_list_has_string(&dst_tag, ref->name))
1379                         continue; /* they already have it */
1380                 if (oid_object_info(the_repository, &ref->new_oid, NULL) != OBJ_TAG)
1381                         continue; /* be conservative */
1382                 item = string_list_append(&src_tag, ref->name);
1383                 item->util = ref;
1384         }
1385         string_list_clear(&dst_tag, 0);
1386
1387         /*
1388          * At this point, src_tag lists tags that are missing from
1389          * dst, and sent_tips lists the tips we are pushing or those
1390          * that we know they already have. An element in the src_tag
1391          * that is an ancestor of any of the sent_tips needs to be
1392          * sent to the other side.
1393          */
1394         if (sent_tips.nr) {
1395                 for_each_string_list_item(item, &src_tag) {
1396                         struct ref *ref = item->util;
1397                         struct ref *dst_ref;
1398                         struct commit *commit;
1399
1400                         if (is_null_oid(&ref->new_oid))
1401                                 continue;
1402                         commit = lookup_commit_reference_gently(&ref->new_oid,
1403                                                                 1);
1404                         if (!commit)
1405                                 /* not pushing a commit, which is not an error */
1406                                 continue;
1407
1408                         /*
1409                          * Is this tag, which they do not have, reachable from
1410                          * any of the commits we are sending?
1411                          */
1412                         if (!in_merge_bases_many(commit, sent_tips.nr, sent_tips.tip))
1413                                 continue;
1414
1415                         /* Add it in */
1416                         dst_ref = make_linked_ref(ref->name, dst_tail);
1417                         oidcpy(&dst_ref->new_oid, &ref->new_oid);
1418                         dst_ref->peer_ref = copy_ref(ref);
1419                 }
1420         }
1421         string_list_clear(&src_tag, 0);
1422         free(sent_tips.tip);
1423 }
1424
1425 struct ref *find_ref_by_name(const struct ref *list, const char *name)
1426 {
1427         for ( ; list; list = list->next)
1428                 if (!strcmp(list->name, name))
1429                         return (struct ref *)list;
1430         return NULL;
1431 }
1432
1433 static void prepare_ref_index(struct string_list *ref_index, struct ref *ref)
1434 {
1435         for ( ; ref; ref = ref->next)
1436                 string_list_append_nodup(ref_index, ref->name)->util = ref;
1437
1438         string_list_sort(ref_index);
1439 }
1440
1441 /*
1442  * Given only the set of local refs, sanity-check the set of push
1443  * refspecs. We can't catch all errors that match_push_refs would,
1444  * but we can catch some errors early before even talking to the
1445  * remote side.
1446  */
1447 int check_push_refs(struct ref *src, int nr_refspec, const char **refspec_names)
1448 {
1449         struct refspec *refspec = parse_push_refspec(nr_refspec, refspec_names);
1450         int ret = 0;
1451         int i;
1452
1453         for (i = 0; i < nr_refspec; i++) {
1454                 struct refspec *rs = refspec + i;
1455
1456                 if (rs->pattern || rs->matching)
1457                         continue;
1458
1459                 ret |= match_explicit_lhs(src, rs, NULL, NULL);
1460         }
1461
1462         free_refspec(nr_refspec, refspec);
1463         return ret;
1464 }
1465
1466 /*
1467  * Given the set of refs the local repository has, the set of refs the
1468  * remote repository has, and the refspec used for push, determine
1469  * what remote refs we will update and with what value by setting
1470  * peer_ref (which object is being pushed) and force (if the push is
1471  * forced) in elements of "dst". The function may add new elements to
1472  * dst (e.g. pushing to a new branch, done in match_explicit_refs).
1473  */
1474 int match_push_refs(struct ref *src, struct ref **dst,
1475                     int nr_refspec, const char **refspec, int flags)
1476 {
1477         struct refspec *rs;
1478         int send_all = flags & MATCH_REFS_ALL;
1479         int send_mirror = flags & MATCH_REFS_MIRROR;
1480         int send_prune = flags & MATCH_REFS_PRUNE;
1481         int errs;
1482         static const char *default_refspec[] = { ":", NULL };
1483         struct ref *ref, **dst_tail = tail_ref(dst);
1484         struct string_list dst_ref_index = STRING_LIST_INIT_NODUP;
1485
1486         if (!nr_refspec) {
1487                 nr_refspec = 1;
1488                 refspec = default_refspec;
1489         }
1490         rs = parse_push_refspec(nr_refspec, (const char **) refspec);
1491         errs = match_explicit_refs(src, *dst, &dst_tail, rs, nr_refspec);
1492
1493         /* pick the remainder */
1494         for (ref = src; ref; ref = ref->next) {
1495                 struct string_list_item *dst_item;
1496                 struct ref *dst_peer;
1497                 const struct refspec *pat = NULL;
1498                 char *dst_name;
1499
1500                 dst_name = get_ref_match(rs, nr_refspec, ref, send_mirror, FROM_SRC, &pat);
1501                 if (!dst_name)
1502                         continue;
1503
1504                 if (!dst_ref_index.nr)
1505                         prepare_ref_index(&dst_ref_index, *dst);
1506
1507                 dst_item = string_list_lookup(&dst_ref_index, dst_name);
1508                 dst_peer = dst_item ? dst_item->util : NULL;
1509                 if (dst_peer) {
1510                         if (dst_peer->peer_ref)
1511                                 /* We're already sending something to this ref. */
1512                                 goto free_name;
1513                 } else {
1514                         if (pat->matching && !(send_all || send_mirror))
1515                                 /*
1516                                  * Remote doesn't have it, and we have no
1517                                  * explicit pattern, and we don't have
1518                                  * --all or --mirror.
1519                                  */
1520                                 goto free_name;
1521
1522                         /* Create a new one and link it */
1523                         dst_peer = make_linked_ref(dst_name, &dst_tail);
1524                         oidcpy(&dst_peer->new_oid, &ref->new_oid);
1525                         string_list_insert(&dst_ref_index,
1526                                 dst_peer->name)->util = dst_peer;
1527                 }
1528                 dst_peer->peer_ref = copy_ref(ref);
1529                 dst_peer->force = pat->force;
1530         free_name:
1531                 free(dst_name);
1532         }
1533
1534         string_list_clear(&dst_ref_index, 0);
1535
1536         if (flags & MATCH_REFS_FOLLOW_TAGS)
1537                 add_missing_tags(src, dst, &dst_tail);
1538
1539         if (send_prune) {
1540                 struct string_list src_ref_index = STRING_LIST_INIT_NODUP;
1541                 /* check for missing refs on the remote */
1542                 for (ref = *dst; ref; ref = ref->next) {
1543                         char *src_name;
1544
1545                         if (ref->peer_ref)
1546                                 /* We're already sending something to this ref. */
1547                                 continue;
1548
1549                         src_name = get_ref_match(rs, nr_refspec, ref, send_mirror, FROM_DST, NULL);
1550                         if (src_name) {
1551                                 if (!src_ref_index.nr)
1552                                         prepare_ref_index(&src_ref_index, src);
1553                                 if (!string_list_has_string(&src_ref_index,
1554                                             src_name))
1555                                         ref->peer_ref = alloc_delete_ref();
1556                                 free(src_name);
1557                         }
1558                 }
1559                 string_list_clear(&src_ref_index, 0);
1560         }
1561         if (errs)
1562                 return -1;
1563         return 0;
1564 }
1565
1566 void set_ref_status_for_push(struct ref *remote_refs, int send_mirror,
1567                              int force_update)
1568 {
1569         struct ref *ref;
1570
1571         for (ref = remote_refs; ref; ref = ref->next) {
1572                 int force_ref_update = ref->force || force_update;
1573                 int reject_reason = 0;
1574
1575                 if (ref->peer_ref)
1576                         oidcpy(&ref->new_oid, &ref->peer_ref->new_oid);
1577                 else if (!send_mirror)
1578                         continue;
1579
1580                 ref->deletion = is_null_oid(&ref->new_oid);
1581                 if (!ref->deletion &&
1582                         !oidcmp(&ref->old_oid, &ref->new_oid)) {
1583                         ref->status = REF_STATUS_UPTODATE;
1584                         continue;
1585                 }
1586
1587                 /*
1588                  * If the remote ref has moved and is now different
1589                  * from what we expect, reject any push.
1590                  *
1591                  * It also is an error if the user told us to check
1592                  * with the remote-tracking branch to find the value
1593                  * to expect, but we did not have such a tracking
1594                  * branch.
1595                  */
1596                 if (ref->expect_old_sha1) {
1597                         if (oidcmp(&ref->old_oid, &ref->old_oid_expect))
1598                                 reject_reason = REF_STATUS_REJECT_STALE;
1599                         else
1600                                 /* If the ref isn't stale then force the update. */
1601                                 force_ref_update = 1;
1602                 }
1603
1604                 /*
1605                  * If the update isn't already rejected then check
1606                  * the usual "must fast-forward" rules.
1607                  *
1608                  * Decide whether an individual refspec A:B can be
1609                  * pushed.  The push will succeed if any of the
1610                  * following are true:
1611                  *
1612                  * (1) the remote reference B does not exist
1613                  *
1614                  * (2) the remote reference B is being removed (i.e.,
1615                  *     pushing :B where no source is specified)
1616                  *
1617                  * (3) the destination is not under refs/tags/, and
1618                  *     if the old and new value is a commit, the new
1619                  *     is a descendant of the old.
1620                  *
1621                  * (4) it is forced using the +A:B notation, or by
1622                  *     passing the --force argument
1623                  */
1624
1625                 if (!reject_reason && !ref->deletion && !is_null_oid(&ref->old_oid)) {
1626                         if (starts_with(ref->name, "refs/tags/"))
1627                                 reject_reason = REF_STATUS_REJECT_ALREADY_EXISTS;
1628                         else if (!has_object_file(&ref->old_oid))
1629                                 reject_reason = REF_STATUS_REJECT_FETCH_FIRST;
1630                         else if (!lookup_commit_reference_gently(&ref->old_oid, 1) ||
1631                                  !lookup_commit_reference_gently(&ref->new_oid, 1))
1632                                 reject_reason = REF_STATUS_REJECT_NEEDS_FORCE;
1633                         else if (!ref_newer(&ref->new_oid, &ref->old_oid))
1634                                 reject_reason = REF_STATUS_REJECT_NONFASTFORWARD;
1635                 }
1636
1637                 /*
1638                  * "--force" will defeat any rejection implemented
1639                  * by the rules above.
1640                  */
1641                 if (!force_ref_update)
1642                         ref->status = reject_reason;
1643                 else if (reject_reason)
1644                         ref->forced_update = 1;
1645         }
1646 }
1647
1648 static void set_merge(struct branch *ret)
1649 {
1650         struct remote *remote;
1651         char *ref;
1652         struct object_id oid;
1653         int i;
1654
1655         if (!ret)
1656                 return; /* no branch */
1657         if (ret->merge)
1658                 return; /* already run */
1659         if (!ret->remote_name || !ret->merge_nr) {
1660                 /*
1661                  * no merge config; let's make sure we don't confuse callers
1662                  * with a non-zero merge_nr but a NULL merge
1663                  */
1664                 ret->merge_nr = 0;
1665                 return;
1666         }
1667
1668         remote = remote_get(ret->remote_name);
1669
1670         ret->merge = xcalloc(ret->merge_nr, sizeof(*ret->merge));
1671         for (i = 0; i < ret->merge_nr; i++) {
1672                 ret->merge[i] = xcalloc(1, sizeof(**ret->merge));
1673                 ret->merge[i]->src = xstrdup(ret->merge_name[i]);
1674                 if (!remote_find_tracking(remote, ret->merge[i]) ||
1675                     strcmp(ret->remote_name, "."))
1676                         continue;
1677                 if (dwim_ref(ret->merge_name[i], strlen(ret->merge_name[i]),
1678                              &oid, &ref) == 1)
1679                         ret->merge[i]->dst = ref;
1680                 else
1681                         ret->merge[i]->dst = xstrdup(ret->merge_name[i]);
1682         }
1683 }
1684
1685 struct branch *branch_get(const char *name)
1686 {
1687         struct branch *ret;
1688
1689         read_config();
1690         if (!name || !*name || !strcmp(name, "HEAD"))
1691                 ret = current_branch;
1692         else
1693                 ret = make_branch(name, 0);
1694         set_merge(ret);
1695         return ret;
1696 }
1697
1698 int branch_has_merge_config(struct branch *branch)
1699 {
1700         return branch && !!branch->merge;
1701 }
1702
1703 int branch_merge_matches(struct branch *branch,
1704                                  int i,
1705                                  const char *refname)
1706 {
1707         if (!branch || i < 0 || i >= branch->merge_nr)
1708                 return 0;
1709         return refname_match(branch->merge[i]->src, refname);
1710 }
1711
1712 __attribute__((format (printf,2,3)))
1713 static const char *error_buf(struct strbuf *err, const char *fmt, ...)
1714 {
1715         if (err) {
1716                 va_list ap;
1717                 va_start(ap, fmt);
1718                 strbuf_vaddf(err, fmt, ap);
1719                 va_end(ap);
1720         }
1721         return NULL;
1722 }
1723
1724 const char *branch_get_upstream(struct branch *branch, struct strbuf *err)
1725 {
1726         if (!branch)
1727                 return error_buf(err, _("HEAD does not point to a branch"));
1728
1729         if (!branch->merge || !branch->merge[0]) {
1730                 /*
1731                  * no merge config; is it because the user didn't define any,
1732                  * or because it is not a real branch, and get_branch
1733                  * auto-vivified it?
1734                  */
1735                 if (!ref_exists(branch->refname))
1736                         return error_buf(err, _("no such branch: '%s'"),
1737                                          branch->name);
1738                 return error_buf(err,
1739                                  _("no upstream configured for branch '%s'"),
1740                                  branch->name);
1741         }
1742
1743         if (!branch->merge[0]->dst)
1744                 return error_buf(err,
1745                                  _("upstream branch '%s' not stored as a remote-tracking branch"),
1746                                  branch->merge[0]->src);
1747
1748         return branch->merge[0]->dst;
1749 }
1750
1751 static const char *tracking_for_push_dest(struct remote *remote,
1752                                           const char *refname,
1753                                           struct strbuf *err)
1754 {
1755         char *ret;
1756
1757         ret = apply_refspecs(remote->fetch, remote->fetch_refspec_nr, refname);
1758         if (!ret)
1759                 return error_buf(err,
1760                                  _("push destination '%s' on remote '%s' has no local tracking branch"),
1761                                  refname, remote->name);
1762         return ret;
1763 }
1764
1765 static const char *branch_get_push_1(struct branch *branch, struct strbuf *err)
1766 {
1767         struct remote *remote;
1768
1769         remote = remote_get(pushremote_for_branch(branch, NULL));
1770         if (!remote)
1771                 return error_buf(err,
1772                                  _("branch '%s' has no remote for pushing"),
1773                                  branch->name);
1774
1775         if (remote->push_refspec_nr) {
1776                 char *dst;
1777                 const char *ret;
1778
1779                 dst = apply_refspecs(remote->push, remote->push_refspec_nr,
1780                                      branch->refname);
1781                 if (!dst)
1782                         return error_buf(err,
1783                                          _("push refspecs for '%s' do not include '%s'"),
1784                                          remote->name, branch->name);
1785
1786                 ret = tracking_for_push_dest(remote, dst, err);
1787                 free(dst);
1788                 return ret;
1789         }
1790
1791         if (remote->mirror)
1792                 return tracking_for_push_dest(remote, branch->refname, err);
1793
1794         switch (push_default) {
1795         case PUSH_DEFAULT_NOTHING:
1796                 return error_buf(err, _("push has no destination (push.default is 'nothing')"));
1797
1798         case PUSH_DEFAULT_MATCHING:
1799         case PUSH_DEFAULT_CURRENT:
1800                 return tracking_for_push_dest(remote, branch->refname, err);
1801
1802         case PUSH_DEFAULT_UPSTREAM:
1803                 return branch_get_upstream(branch, err);
1804
1805         case PUSH_DEFAULT_UNSPECIFIED:
1806         case PUSH_DEFAULT_SIMPLE:
1807                 {
1808                         const char *up, *cur;
1809
1810                         up = branch_get_upstream(branch, err);
1811                         if (!up)
1812                                 return NULL;
1813                         cur = tracking_for_push_dest(remote, branch->refname, err);
1814                         if (!cur)
1815                                 return NULL;
1816                         if (strcmp(cur, up))
1817                                 return error_buf(err,
1818                                                  _("cannot resolve 'simple' push to a single destination"));
1819                         return cur;
1820                 }
1821         }
1822
1823         die("BUG: unhandled push situation");
1824 }
1825
1826 const char *branch_get_push(struct branch *branch, struct strbuf *err)
1827 {
1828         if (!branch)
1829                 return error_buf(err, _("HEAD does not point to a branch"));
1830
1831         if (!branch->push_tracking_ref)
1832                 branch->push_tracking_ref = branch_get_push_1(branch, err);
1833         return branch->push_tracking_ref;
1834 }
1835
1836 static int ignore_symref_update(const char *refname)
1837 {
1838         int flag;
1839
1840         if (!resolve_ref_unsafe(refname, 0, NULL, &flag))
1841                 return 0; /* non-existing refs are OK */
1842         return (flag & REF_ISSYMREF);
1843 }
1844
1845 /*
1846  * Create and return a list of (struct ref) consisting of copies of
1847  * each remote_ref that matches refspec.  refspec must be a pattern.
1848  * Fill in the copies' peer_ref to describe the local tracking refs to
1849  * which they map.  Omit any references that would map to an existing
1850  * local symbolic ref.
1851  */
1852 static struct ref *get_expanded_map(const struct ref *remote_refs,
1853                                     const struct refspec *refspec)
1854 {
1855         const struct ref *ref;
1856         struct ref *ret = NULL;
1857         struct ref **tail = &ret;
1858
1859         for (ref = remote_refs; ref; ref = ref->next) {
1860                 char *expn_name = NULL;
1861
1862                 if (strchr(ref->name, '^'))
1863                         continue; /* a dereference item */
1864                 if (match_name_with_pattern(refspec->src, ref->name,
1865                                             refspec->dst, &expn_name) &&
1866                     !ignore_symref_update(expn_name)) {
1867                         struct ref *cpy = copy_ref(ref);
1868
1869                         cpy->peer_ref = alloc_ref(expn_name);
1870                         if (refspec->force)
1871                                 cpy->peer_ref->force = 1;
1872                         *tail = cpy;
1873                         tail = &cpy->next;
1874                 }
1875                 free(expn_name);
1876         }
1877
1878         return ret;
1879 }
1880
1881 static const struct ref *find_ref_by_name_abbrev(const struct ref *refs, const char *name)
1882 {
1883         const struct ref *ref;
1884         for (ref = refs; ref; ref = ref->next) {
1885                 if (refname_match(name, ref->name))
1886                         return ref;
1887         }
1888         return NULL;
1889 }
1890
1891 struct ref *get_remote_ref(const struct ref *remote_refs, const char *name)
1892 {
1893         const struct ref *ref = find_ref_by_name_abbrev(remote_refs, name);
1894
1895         if (!ref)
1896                 return NULL;
1897
1898         return copy_ref(ref);
1899 }
1900
1901 static struct ref *get_local_ref(const char *name)
1902 {
1903         if (!name || name[0] == '\0')
1904                 return NULL;
1905
1906         if (starts_with(name, "refs/"))
1907                 return alloc_ref(name);
1908
1909         if (starts_with(name, "heads/") ||
1910             starts_with(name, "tags/") ||
1911             starts_with(name, "remotes/"))
1912                 return alloc_ref_with_prefix("refs/", 5, name);
1913
1914         return alloc_ref_with_prefix("refs/heads/", 11, name);
1915 }
1916
1917 int get_fetch_map(const struct ref *remote_refs,
1918                   const struct refspec *refspec,
1919                   struct ref ***tail,
1920                   int missing_ok)
1921 {
1922         struct ref *ref_map, **rmp;
1923
1924         if (refspec->pattern) {
1925                 ref_map = get_expanded_map(remote_refs, refspec);
1926         } else {
1927                 const char *name = refspec->src[0] ? refspec->src : "HEAD";
1928
1929                 if (refspec->exact_sha1) {
1930                         ref_map = alloc_ref(name);
1931                         get_oid_hex(name, &ref_map->old_oid);
1932                 } else {
1933                         ref_map = get_remote_ref(remote_refs, name);
1934                 }
1935                 if (!missing_ok && !ref_map)
1936                         die("Couldn't find remote ref %s", name);
1937                 if (ref_map) {
1938                         ref_map->peer_ref = get_local_ref(refspec->dst);
1939                         if (ref_map->peer_ref && refspec->force)
1940                                 ref_map->peer_ref->force = 1;
1941                 }
1942         }
1943
1944         for (rmp = &ref_map; *rmp; ) {
1945                 if ((*rmp)->peer_ref) {
1946                         if (!starts_with((*rmp)->peer_ref->name, "refs/") ||
1947                             check_refname_format((*rmp)->peer_ref->name, 0)) {
1948                                 struct ref *ignore = *rmp;
1949                                 error("* Ignoring funny ref '%s' locally",
1950                                       (*rmp)->peer_ref->name);
1951                                 *rmp = (*rmp)->next;
1952                                 free(ignore->peer_ref);
1953                                 free(ignore);
1954                                 continue;
1955                         }
1956                 }
1957                 rmp = &((*rmp)->next);
1958         }
1959
1960         if (ref_map)
1961                 tail_link_ref(ref_map, tail);
1962
1963         return 0;
1964 }
1965
1966 int resolve_remote_symref(struct ref *ref, struct ref *list)
1967 {
1968         if (!ref->symref)
1969                 return 0;
1970         for (; list; list = list->next)
1971                 if (!strcmp(ref->symref, list->name)) {
1972                         oidcpy(&ref->old_oid, &list->old_oid);
1973                         return 0;
1974                 }
1975         return 1;
1976 }
1977
1978 static void unmark_and_free(struct commit_list *list, unsigned int mark)
1979 {
1980         while (list) {
1981                 struct commit *commit = pop_commit(&list);
1982                 commit->object.flags &= ~mark;
1983         }
1984 }
1985
1986 int ref_newer(const struct object_id *new_oid, const struct object_id *old_oid)
1987 {
1988         struct object *o;
1989         struct commit *old_commit, *new_commit;
1990         struct commit_list *list, *used;
1991         int found = 0;
1992
1993         /*
1994          * Both new_commit and old_commit must be commit-ish and new_commit is descendant of
1995          * old_commit.  Otherwise we require --force.
1996          */
1997         o = deref_tag(parse_object(old_oid), NULL, 0);
1998         if (!o || o->type != OBJ_COMMIT)
1999                 return 0;
2000         old_commit = (struct commit *) o;
2001
2002         o = deref_tag(parse_object(new_oid), NULL, 0);
2003         if (!o || o->type != OBJ_COMMIT)
2004                 return 0;
2005         new_commit = (struct commit *) o;
2006
2007         if (parse_commit(new_commit) < 0)
2008                 return 0;
2009
2010         used = list = NULL;
2011         commit_list_insert(new_commit, &list);
2012         while (list) {
2013                 new_commit = pop_most_recent_commit(&list, TMP_MARK);
2014                 commit_list_insert(new_commit, &used);
2015                 if (new_commit == old_commit) {
2016                         found = 1;
2017                         break;
2018                 }
2019         }
2020         unmark_and_free(list, TMP_MARK);
2021         unmark_and_free(used, TMP_MARK);
2022         return found;
2023 }
2024
2025 /*
2026  * Lookup the upstream branch for the given branch and if present, optionally
2027  * compute the commit ahead/behind values for the pair.
2028  *
2029  * If abf is AHEAD_BEHIND_FULL, compute the full ahead/behind and return the
2030  * counts in *num_ours and *num_theirs.  If abf is AHEAD_BEHIND_QUICK, skip
2031  * the (potentially expensive) a/b computation (*num_ours and *num_theirs are
2032  * set to zero).
2033  *
2034  * The name of the upstream branch (or NULL if no upstream is defined) is
2035  * returned via *upstream_name, if it is not itself NULL.
2036  *
2037  * Returns -1 if num_ours and num_theirs could not be filled in (e.g., no
2038  * upstream defined, or ref does not exist).  Returns 0 if the commits are
2039  * identical.  Returns 1 if commits are different.
2040  */
2041 int stat_tracking_info(struct branch *branch, int *num_ours, int *num_theirs,
2042                        const char **upstream_name, enum ahead_behind_flags abf)
2043 {
2044         struct object_id oid;
2045         struct commit *ours, *theirs;
2046         struct rev_info revs;
2047         const char *base;
2048         struct argv_array argv = ARGV_ARRAY_INIT;
2049
2050         /* Cannot stat unless we are marked to build on top of somebody else. */
2051         base = branch_get_upstream(branch, NULL);
2052         if (upstream_name)
2053                 *upstream_name = base;
2054         if (!base)
2055                 return -1;
2056
2057         /* Cannot stat if what we used to build on no longer exists */
2058         if (read_ref(base, &oid))
2059                 return -1;
2060         theirs = lookup_commit_reference(&oid);
2061         if (!theirs)
2062                 return -1;
2063
2064         if (read_ref(branch->refname, &oid))
2065                 return -1;
2066         ours = lookup_commit_reference(&oid);
2067         if (!ours)
2068                 return -1;
2069
2070         *num_theirs = *num_ours = 0;
2071
2072         /* are we the same? */
2073         if (theirs == ours)
2074                 return 0;
2075         if (abf == AHEAD_BEHIND_QUICK)
2076                 return 1;
2077         if (abf != AHEAD_BEHIND_FULL)
2078                 BUG("stat_tracking_info: invalid abf '%d'", abf);
2079
2080         /* Run "rev-list --left-right ours...theirs" internally... */
2081         argv_array_push(&argv, ""); /* ignored */
2082         argv_array_push(&argv, "--left-right");
2083         argv_array_pushf(&argv, "%s...%s",
2084                          oid_to_hex(&ours->object.oid),
2085                          oid_to_hex(&theirs->object.oid));
2086         argv_array_push(&argv, "--");
2087
2088         init_revisions(&revs, NULL);
2089         setup_revisions(argv.argc, argv.argv, &revs, NULL);
2090         if (prepare_revision_walk(&revs))
2091                 die("revision walk setup failed");
2092
2093         /* ... and count the commits on each side. */
2094         while (1) {
2095                 struct commit *c = get_revision(&revs);
2096                 if (!c)
2097                         break;
2098                 if (c->object.flags & SYMMETRIC_LEFT)
2099                         (*num_ours)++;
2100                 else
2101                         (*num_theirs)++;
2102         }
2103
2104         /* clear object flags smudged by the above traversal */
2105         clear_commit_marks(ours, ALL_REV_FLAGS);
2106         clear_commit_marks(theirs, ALL_REV_FLAGS);
2107
2108         argv_array_clear(&argv);
2109         return 1;
2110 }
2111
2112 /*
2113  * Return true when there is anything to report, otherwise false.
2114  */
2115 int format_tracking_info(struct branch *branch, struct strbuf *sb,
2116                          enum ahead_behind_flags abf)
2117 {
2118         int ours, theirs, sti;
2119         const char *full_base;
2120         char *base;
2121         int upstream_is_gone = 0;
2122
2123         sti = stat_tracking_info(branch, &ours, &theirs, &full_base, abf);
2124         if (sti < 0) {
2125                 if (!full_base)
2126                         return 0;
2127                 upstream_is_gone = 1;
2128         }
2129
2130         base = shorten_unambiguous_ref(full_base, 0);
2131         if (upstream_is_gone) {
2132                 strbuf_addf(sb,
2133                         _("Your branch is based on '%s', but the upstream is gone.\n"),
2134                         base);
2135                 if (advice_status_hints)
2136                         strbuf_addstr(sb,
2137                                 _("  (use \"git branch --unset-upstream\" to fixup)\n"));
2138         } else if (!sti) {
2139                 strbuf_addf(sb,
2140                         _("Your branch is up to date with '%s'.\n"),
2141                         base);
2142         } else if (abf == AHEAD_BEHIND_QUICK) {
2143                 strbuf_addf(sb,
2144                             _("Your branch and '%s' refer to different commits.\n"),
2145                             base);
2146                 if (advice_status_hints)
2147                         strbuf_addf(sb, _("  (use \"%s\" for details)\n"),
2148                                     "git status --ahead-behind");
2149         } else if (!theirs) {
2150                 strbuf_addf(sb,
2151                         Q_("Your branch is ahead of '%s' by %d commit.\n",
2152                            "Your branch is ahead of '%s' by %d commits.\n",
2153                            ours),
2154                         base, ours);
2155                 if (advice_status_hints)
2156                         strbuf_addstr(sb,
2157                                 _("  (use \"git push\" to publish your local commits)\n"));
2158         } else if (!ours) {
2159                 strbuf_addf(sb,
2160                         Q_("Your branch is behind '%s' by %d commit, "
2161                                "and can be fast-forwarded.\n",
2162                            "Your branch is behind '%s' by %d commits, "
2163                                "and can be fast-forwarded.\n",
2164                            theirs),
2165                         base, theirs);
2166                 if (advice_status_hints)
2167                         strbuf_addstr(sb,
2168                                 _("  (use \"git pull\" to update your local branch)\n"));
2169         } else {
2170                 strbuf_addf(sb,
2171                         Q_("Your branch and '%s' have diverged,\n"
2172                                "and have %d and %d different commit each, "
2173                                "respectively.\n",
2174                            "Your branch and '%s' have diverged,\n"
2175                                "and have %d and %d different commits each, "
2176                                "respectively.\n",
2177                            ours + theirs),
2178                         base, ours, theirs);
2179                 if (advice_status_hints)
2180                         strbuf_addstr(sb,
2181                                 _("  (use \"git pull\" to merge the remote branch into yours)\n"));
2182         }
2183         free(base);
2184         return 1;
2185 }
2186
2187 static int one_local_ref(const char *refname, const struct object_id *oid,
2188                          int flag, void *cb_data)
2189 {
2190         struct ref ***local_tail = cb_data;
2191         struct ref *ref;
2192
2193         /* we already know it starts with refs/ to get here */
2194         if (check_refname_format(refname + 5, 0))
2195                 return 0;
2196
2197         ref = alloc_ref(refname);
2198         oidcpy(&ref->new_oid, oid);
2199         **local_tail = ref;
2200         *local_tail = &ref->next;
2201         return 0;
2202 }
2203
2204 struct ref *get_local_heads(void)
2205 {
2206         struct ref *local_refs = NULL, **local_tail = &local_refs;
2207
2208         for_each_ref(one_local_ref, &local_tail);
2209         return local_refs;
2210 }
2211
2212 struct ref *guess_remote_head(const struct ref *head,
2213                               const struct ref *refs,
2214                               int all)
2215 {
2216         const struct ref *r;
2217         struct ref *list = NULL;
2218         struct ref **tail = &list;
2219
2220         if (!head)
2221                 return NULL;
2222
2223         /*
2224          * Some transports support directly peeking at
2225          * where HEAD points; if that is the case, then
2226          * we don't have to guess.
2227          */
2228         if (head->symref)
2229                 return copy_ref(find_ref_by_name(refs, head->symref));
2230
2231         /* If refs/heads/master could be right, it is. */
2232         if (!all) {
2233                 r = find_ref_by_name(refs, "refs/heads/master");
2234                 if (r && !oidcmp(&r->old_oid, &head->old_oid))
2235                         return copy_ref(r);
2236         }
2237
2238         /* Look for another ref that points there */
2239         for (r = refs; r; r = r->next) {
2240                 if (r != head &&
2241                     starts_with(r->name, "refs/heads/") &&
2242                     !oidcmp(&r->old_oid, &head->old_oid)) {
2243                         *tail = copy_ref(r);
2244                         tail = &((*tail)->next);
2245                         if (!all)
2246                                 break;
2247                 }
2248         }
2249
2250         return list;
2251 }
2252
2253 struct stale_heads_info {
2254         struct string_list *ref_names;
2255         struct ref **stale_refs_tail;
2256         struct refspec *refs;
2257         int ref_count;
2258 };
2259
2260 static int get_stale_heads_cb(const char *refname, const struct object_id *oid,
2261                               int flags, void *cb_data)
2262 {
2263         struct stale_heads_info *info = cb_data;
2264         struct string_list matches = STRING_LIST_INIT_DUP;
2265         struct refspec query;
2266         int i, stale = 1;
2267         memset(&query, 0, sizeof(struct refspec));
2268         query.dst = (char *)refname;
2269
2270         query_refspecs_multiple(info->refs, info->ref_count, &query, &matches);
2271         if (matches.nr == 0)
2272                 goto clean_exit; /* No matches */
2273
2274         /*
2275          * If we did find a suitable refspec and it's not a symref and
2276          * it's not in the list of refs that currently exist in that
2277          * remote, we consider it to be stale. In order to deal with
2278          * overlapping refspecs, we need to go over all of the
2279          * matching refs.
2280          */
2281         if (flags & REF_ISSYMREF)
2282                 goto clean_exit;
2283
2284         for (i = 0; stale && i < matches.nr; i++)
2285                 if (string_list_has_string(info->ref_names, matches.items[i].string))
2286                         stale = 0;
2287
2288         if (stale) {
2289                 struct ref *ref = make_linked_ref(refname, &info->stale_refs_tail);
2290                 oidcpy(&ref->new_oid, oid);
2291         }
2292
2293 clean_exit:
2294         string_list_clear(&matches, 0);
2295         return 0;
2296 }
2297
2298 struct ref *get_stale_heads(struct refspec *refs, int ref_count, struct ref *fetch_map)
2299 {
2300         struct ref *ref, *stale_refs = NULL;
2301         struct string_list ref_names = STRING_LIST_INIT_NODUP;
2302         struct stale_heads_info info;
2303
2304         info.ref_names = &ref_names;
2305         info.stale_refs_tail = &stale_refs;
2306         info.refs = refs;
2307         info.ref_count = ref_count;
2308         for (ref = fetch_map; ref; ref = ref->next)
2309                 string_list_append(&ref_names, ref->name);
2310         string_list_sort(&ref_names);
2311         for_each_ref(get_stale_heads_cb, &info);
2312         string_list_clear(&ref_names, 0);
2313         return stale_refs;
2314 }
2315
2316 /*
2317  * Compare-and-swap
2318  */
2319 static void clear_cas_option(struct push_cas_option *cas)
2320 {
2321         int i;
2322
2323         for (i = 0; i < cas->nr; i++)
2324                 free(cas->entry[i].refname);
2325         free(cas->entry);
2326         memset(cas, 0, sizeof(*cas));
2327 }
2328
2329 static struct push_cas *add_cas_entry(struct push_cas_option *cas,
2330                                       const char *refname,
2331                                       size_t refnamelen)
2332 {
2333         struct push_cas *entry;
2334         ALLOC_GROW(cas->entry, cas->nr + 1, cas->alloc);
2335         entry = &cas->entry[cas->nr++];
2336         memset(entry, 0, sizeof(*entry));
2337         entry->refname = xmemdupz(refname, refnamelen);
2338         return entry;
2339 }
2340
2341 static int parse_push_cas_option(struct push_cas_option *cas, const char *arg, int unset)
2342 {
2343         const char *colon;
2344         struct push_cas *entry;
2345
2346         if (unset) {
2347                 /* "--no-<option>" */
2348                 clear_cas_option(cas);
2349                 return 0;
2350         }
2351
2352         if (!arg) {
2353                 /* just "--<option>" */
2354                 cas->use_tracking_for_rest = 1;
2355                 return 0;
2356         }
2357
2358         /* "--<option>=refname" or "--<option>=refname:value" */
2359         colon = strchrnul(arg, ':');
2360         entry = add_cas_entry(cas, arg, colon - arg);
2361         if (!*colon)
2362                 entry->use_tracking = 1;
2363         else if (!colon[1])
2364                 oidclr(&entry->expect);
2365         else if (get_oid(colon + 1, &entry->expect))
2366                 return error("cannot parse expected object name '%s'", colon + 1);
2367         return 0;
2368 }
2369
2370 int parseopt_push_cas_option(const struct option *opt, const char *arg, int unset)
2371 {
2372         return parse_push_cas_option(opt->value, arg, unset);
2373 }
2374
2375 int is_empty_cas(const struct push_cas_option *cas)
2376 {
2377         return !cas->use_tracking_for_rest && !cas->nr;
2378 }
2379
2380 /*
2381  * Look at remote.fetch refspec and see if we have a remote
2382  * tracking branch for the refname there.  Fill its current
2383  * value in sha1[].
2384  * If we cannot do so, return negative to signal an error.
2385  */
2386 static int remote_tracking(struct remote *remote, const char *refname,
2387                            struct object_id *oid)
2388 {
2389         char *dst;
2390
2391         dst = apply_refspecs(remote->fetch, remote->fetch_refspec_nr, refname);
2392         if (!dst)
2393                 return -1; /* no tracking ref for refname at remote */
2394         if (read_ref(dst, oid))
2395                 return -1; /* we know what the tracking ref is but we cannot read it */
2396         return 0;
2397 }
2398
2399 static void apply_cas(struct push_cas_option *cas,
2400                       struct remote *remote,
2401                       struct ref *ref)
2402 {
2403         int i;
2404
2405         /* Find an explicit --<option>=<name>[:<value>] entry */
2406         for (i = 0; i < cas->nr; i++) {
2407                 struct push_cas *entry = &cas->entry[i];
2408                 if (!refname_match(entry->refname, ref->name))
2409                         continue;
2410                 ref->expect_old_sha1 = 1;
2411                 if (!entry->use_tracking)
2412                         oidcpy(&ref->old_oid_expect, &entry->expect);
2413                 else if (remote_tracking(remote, ref->name, &ref->old_oid_expect))
2414                         oidclr(&ref->old_oid_expect);
2415                 return;
2416         }
2417
2418         /* Are we using "--<option>" to cover all? */
2419         if (!cas->use_tracking_for_rest)
2420                 return;
2421
2422         ref->expect_old_sha1 = 1;
2423         if (remote_tracking(remote, ref->name, &ref->old_oid_expect))
2424                 oidclr(&ref->old_oid_expect);
2425 }
2426
2427 void apply_push_cas(struct push_cas_option *cas,
2428                     struct remote *remote,
2429                     struct ref *remote_refs)
2430 {
2431         struct ref *ref;
2432         for (ref = remote_refs; ref; ref = ref->next)
2433                 apply_cas(cas, remote, ref);
2434 }