Merge branch 'nd/rewritten-ref-is-per-worktree'
[git] / merge-recursive.c
1 /*
2  * Recursive Merge algorithm stolen from git-merge-recursive.py by
3  * Fredrik Kuivinen.
4  * The thieves were Alex Riesen and Johannes Schindelin, in June/July 2006
5  */
6 #include "cache.h"
7 #include "config.h"
8 #include "advice.h"
9 #include "lockfile.h"
10 #include "cache-tree.h"
11 #include "object-store.h"
12 #include "repository.h"
13 #include "commit.h"
14 #include "blob.h"
15 #include "builtin.h"
16 #include "tree-walk.h"
17 #include "diff.h"
18 #include "diffcore.h"
19 #include "tag.h"
20 #include "alloc.h"
21 #include "unpack-trees.h"
22 #include "string-list.h"
23 #include "xdiff-interface.h"
24 #include "ll-merge.h"
25 #include "attr.h"
26 #include "merge-recursive.h"
27 #include "dir.h"
28 #include "submodule.h"
29 #include "revision.h"
30 #include "commit-reach.h"
31
32 struct path_hashmap_entry {
33         struct hashmap_entry e;
34         char path[FLEX_ARRAY];
35 };
36
37 static int path_hashmap_cmp(const void *cmp_data,
38                             const void *entry,
39                             const void *entry_or_key,
40                             const void *keydata)
41 {
42         const struct path_hashmap_entry *a = entry;
43         const struct path_hashmap_entry *b = entry_or_key;
44         const char *key = keydata;
45
46         if (ignore_case)
47                 return strcasecmp(a->path, key ? key : b->path);
48         else
49                 return strcmp(a->path, key ? key : b->path);
50 }
51
52 static unsigned int path_hash(const char *path)
53 {
54         return ignore_case ? strihash(path) : strhash(path);
55 }
56
57 static struct dir_rename_entry *dir_rename_find_entry(struct hashmap *hashmap,
58                                                       char *dir)
59 {
60         struct dir_rename_entry key;
61
62         if (dir == NULL)
63                 return NULL;
64         hashmap_entry_init(&key, strhash(dir));
65         key.dir = dir;
66         return hashmap_get(hashmap, &key, NULL);
67 }
68
69 static int dir_rename_cmp(const void *unused_cmp_data,
70                           const void *entry,
71                           const void *entry_or_key,
72                           const void *unused_keydata)
73 {
74         const struct dir_rename_entry *e1 = entry;
75         const struct dir_rename_entry *e2 = entry_or_key;
76
77         return strcmp(e1->dir, e2->dir);
78 }
79
80 static void dir_rename_init(struct hashmap *map)
81 {
82         hashmap_init(map, dir_rename_cmp, NULL, 0);
83 }
84
85 static void dir_rename_entry_init(struct dir_rename_entry *entry,
86                                   char *directory)
87 {
88         hashmap_entry_init(entry, strhash(directory));
89         entry->dir = directory;
90         entry->non_unique_new_dir = 0;
91         strbuf_init(&entry->new_dir, 0);
92         string_list_init(&entry->possible_new_dirs, 0);
93 }
94
95 static struct collision_entry *collision_find_entry(struct hashmap *hashmap,
96                                                     char *target_file)
97 {
98         struct collision_entry key;
99
100         hashmap_entry_init(&key, strhash(target_file));
101         key.target_file = target_file;
102         return hashmap_get(hashmap, &key, NULL);
103 }
104
105 static int collision_cmp(void *unused_cmp_data,
106                          const struct collision_entry *e1,
107                          const struct collision_entry *e2,
108                          const void *unused_keydata)
109 {
110         return strcmp(e1->target_file, e2->target_file);
111 }
112
113 static void collision_init(struct hashmap *map)
114 {
115         hashmap_init(map, (hashmap_cmp_fn) collision_cmp, NULL, 0);
116 }
117
118 static void flush_output(struct merge_options *o)
119 {
120         if (o->buffer_output < 2 && o->obuf.len) {
121                 fputs(o->obuf.buf, stdout);
122                 strbuf_reset(&o->obuf);
123         }
124 }
125
126 static int err(struct merge_options *o, const char *err, ...)
127 {
128         va_list params;
129
130         if (o->buffer_output < 2)
131                 flush_output(o);
132         else {
133                 strbuf_complete(&o->obuf, '\n');
134                 strbuf_addstr(&o->obuf, "error: ");
135         }
136         va_start(params, err);
137         strbuf_vaddf(&o->obuf, err, params);
138         va_end(params);
139         if (o->buffer_output > 1)
140                 strbuf_addch(&o->obuf, '\n');
141         else {
142                 error("%s", o->obuf.buf);
143                 strbuf_reset(&o->obuf);
144         }
145
146         return -1;
147 }
148
149 static struct tree *shift_tree_object(struct repository *repo,
150                                       struct tree *one, struct tree *two,
151                                       const char *subtree_shift)
152 {
153         struct object_id shifted;
154
155         if (!*subtree_shift) {
156                 shift_tree(&one->object.oid, &two->object.oid, &shifted, 0);
157         } else {
158                 shift_tree_by(&one->object.oid, &two->object.oid, &shifted,
159                               subtree_shift);
160         }
161         if (oideq(&two->object.oid, &shifted))
162                 return two;
163         return lookup_tree(repo, &shifted);
164 }
165
166 static struct commit *make_virtual_commit(struct repository *repo,
167                                           struct tree *tree,
168                                           const char *comment)
169 {
170         struct commit *commit = alloc_commit_node(repo);
171
172         set_merge_remote_desc(commit, comment, (struct object *)commit);
173         commit->maybe_tree = tree;
174         commit->object.parsed = 1;
175         return commit;
176 }
177
178 /*
179  * Since we use get_tree_entry(), which does not put the read object into
180  * the object pool, we cannot rely on a == b.
181  */
182 static int oid_eq(const struct object_id *a, const struct object_id *b)
183 {
184         if (!a && !b)
185                 return 2;
186         return a && b && oideq(a, b);
187 }
188
189 enum rename_type {
190         RENAME_NORMAL = 0,
191         RENAME_VIA_DIR,
192         RENAME_ADD,
193         RENAME_DELETE,
194         RENAME_ONE_FILE_TO_ONE,
195         RENAME_ONE_FILE_TO_TWO,
196         RENAME_TWO_FILES_TO_ONE
197 };
198
199 struct rename_conflict_info {
200         enum rename_type rename_type;
201         struct diff_filepair *pair1;
202         struct diff_filepair *pair2;
203         const char *branch1;
204         const char *branch2;
205         struct stage_data *dst_entry1;
206         struct stage_data *dst_entry2;
207         struct diff_filespec ren1_other;
208         struct diff_filespec ren2_other;
209 };
210
211 /*
212  * Since we want to write the index eventually, we cannot reuse the index
213  * for these (temporary) data.
214  */
215 struct stage_data {
216         struct {
217                 unsigned mode;
218                 struct object_id oid;
219         } stages[4];
220         struct rename_conflict_info *rename_conflict_info;
221         unsigned processed:1;
222 };
223
224 static inline void setup_rename_conflict_info(enum rename_type rename_type,
225                                               struct diff_filepair *pair1,
226                                               struct diff_filepair *pair2,
227                                               const char *branch1,
228                                               const char *branch2,
229                                               struct stage_data *dst_entry1,
230                                               struct stage_data *dst_entry2,
231                                               struct merge_options *o,
232                                               struct stage_data *src_entry1,
233                                               struct stage_data *src_entry2)
234 {
235         int ostage1 = 0, ostage2;
236         struct rename_conflict_info *ci;
237
238         /*
239          * When we have two renames involved, it's easiest to get the
240          * correct things into stage 2 and 3, and to make sure that the
241          * content merge puts HEAD before the other branch if we just
242          * ensure that branch1 == o->branch1.  So, simply flip arguments
243          * around if we don't have that.
244          */
245         if (dst_entry2 && branch1 != o->branch1) {
246                 setup_rename_conflict_info(rename_type,
247                                            pair2,      pair1,
248                                            branch2,    branch1,
249                                            dst_entry2, dst_entry1,
250                                            o,
251                                            src_entry2, src_entry1);
252                 return;
253         }
254
255         ci = xcalloc(1, sizeof(struct rename_conflict_info));
256         ci->rename_type = rename_type;
257         ci->pair1 = pair1;
258         ci->branch1 = branch1;
259         ci->branch2 = branch2;
260
261         ci->dst_entry1 = dst_entry1;
262         dst_entry1->rename_conflict_info = ci;
263         dst_entry1->processed = 0;
264
265         assert(!pair2 == !dst_entry2);
266         if (dst_entry2) {
267                 ci->dst_entry2 = dst_entry2;
268                 ci->pair2 = pair2;
269                 dst_entry2->rename_conflict_info = ci;
270         }
271
272         /*
273          * For each rename, there could have been
274          * modifications on the side of history where that
275          * file was not renamed.
276          */
277         if (rename_type == RENAME_ADD ||
278             rename_type == RENAME_TWO_FILES_TO_ONE) {
279                 ostage1 = o->branch1 == branch1 ? 3 : 2;
280
281                 ci->ren1_other.path = pair1->one->path;
282                 oidcpy(&ci->ren1_other.oid, &src_entry1->stages[ostage1].oid);
283                 ci->ren1_other.mode = src_entry1->stages[ostage1].mode;
284         }
285
286         if (rename_type == RENAME_TWO_FILES_TO_ONE) {
287                 ostage2 = ostage1 ^ 1;
288
289                 ci->ren2_other.path = pair2->one->path;
290                 oidcpy(&ci->ren2_other.oid, &src_entry2->stages[ostage2].oid);
291                 ci->ren2_other.mode = src_entry2->stages[ostage2].mode;
292         }
293 }
294
295 static int show(struct merge_options *o, int v)
296 {
297         return (!o->call_depth && o->verbosity >= v) || o->verbosity >= 5;
298 }
299
300 __attribute__((format (printf, 3, 4)))
301 static void output(struct merge_options *o, int v, const char *fmt, ...)
302 {
303         va_list ap;
304
305         if (!show(o, v))
306                 return;
307
308         strbuf_addchars(&o->obuf, ' ', o->call_depth * 2);
309
310         va_start(ap, fmt);
311         strbuf_vaddf(&o->obuf, fmt, ap);
312         va_end(ap);
313
314         strbuf_addch(&o->obuf, '\n');
315         if (!o->buffer_output)
316                 flush_output(o);
317 }
318
319 static void output_commit_title(struct merge_options *o, struct commit *commit)
320 {
321         struct merge_remote_desc *desc;
322
323         strbuf_addchars(&o->obuf, ' ', o->call_depth * 2);
324         desc = merge_remote_util(commit);
325         if (desc)
326                 strbuf_addf(&o->obuf, "virtual %s\n", desc->name);
327         else {
328                 strbuf_add_unique_abbrev(&o->obuf, &commit->object.oid,
329                                          DEFAULT_ABBREV);
330                 strbuf_addch(&o->obuf, ' ');
331                 if (parse_commit(commit) != 0)
332                         strbuf_addstr(&o->obuf, _("(bad commit)\n"));
333                 else {
334                         const char *title;
335                         const char *msg = get_commit_buffer(commit, NULL);
336                         int len = find_commit_subject(msg, &title);
337                         if (len)
338                                 strbuf_addf(&o->obuf, "%.*s\n", len, title);
339                         unuse_commit_buffer(commit, msg);
340                 }
341         }
342         flush_output(o);
343 }
344
345 static int add_cacheinfo(struct merge_options *o,
346                          unsigned int mode, const struct object_id *oid,
347                          const char *path, int stage, int refresh, int options)
348 {
349         struct index_state *istate = o->repo->index;
350         struct cache_entry *ce;
351         int ret;
352
353         ce = make_cache_entry(istate, mode, oid ? oid : &null_oid, path, stage, 0);
354         if (!ce)
355                 return err(o, _("add_cacheinfo failed for path '%s'; merge aborting."), path);
356
357         ret = add_index_entry(istate, ce, options);
358         if (refresh) {
359                 struct cache_entry *nce;
360
361                 nce = refresh_cache_entry(istate, ce,
362                                           CE_MATCH_REFRESH | CE_MATCH_IGNORE_MISSING);
363                 if (!nce)
364                         return err(o, _("add_cacheinfo failed to refresh for path '%s'; merge aborting."), path);
365                 if (nce != ce)
366                         ret = add_index_entry(istate, nce, options);
367         }
368         return ret;
369 }
370
371 static void init_tree_desc_from_tree(struct tree_desc *desc, struct tree *tree)
372 {
373         parse_tree(tree);
374         init_tree_desc(desc, tree->buffer, tree->size);
375 }
376
377 static int unpack_trees_start(struct merge_options *o,
378                               struct tree *common,
379                               struct tree *head,
380                               struct tree *merge)
381 {
382         int rc;
383         struct tree_desc t[3];
384         struct index_state tmp_index = { NULL };
385
386         memset(&o->unpack_opts, 0, sizeof(o->unpack_opts));
387         if (o->call_depth)
388                 o->unpack_opts.index_only = 1;
389         else
390                 o->unpack_opts.update = 1;
391         o->unpack_opts.merge = 1;
392         o->unpack_opts.head_idx = 2;
393         o->unpack_opts.fn = threeway_merge;
394         o->unpack_opts.src_index = o->repo->index;
395         o->unpack_opts.dst_index = &tmp_index;
396         o->unpack_opts.aggressive = !merge_detect_rename(o);
397         setup_unpack_trees_porcelain(&o->unpack_opts, "merge");
398
399         init_tree_desc_from_tree(t+0, common);
400         init_tree_desc_from_tree(t+1, head);
401         init_tree_desc_from_tree(t+2, merge);
402
403         rc = unpack_trees(3, t, &o->unpack_opts);
404         cache_tree_free(&o->repo->index->cache_tree);
405
406         /*
407          * Update o->repo->index to match the new results, AFTER saving a copy
408          * in o->orig_index.  Update src_index to point to the saved copy.
409          * (verify_uptodate() checks src_index, and the original index is
410          * the one that had the necessary modification timestamps.)
411          */
412         o->orig_index = *o->repo->index;
413         *o->repo->index = tmp_index;
414         o->unpack_opts.src_index = &o->orig_index;
415
416         return rc;
417 }
418
419 static void unpack_trees_finish(struct merge_options *o)
420 {
421         discard_index(&o->orig_index);
422         clear_unpack_trees_porcelain(&o->unpack_opts);
423 }
424
425 struct tree *write_tree_from_memory(struct merge_options *o)
426 {
427         struct tree *result = NULL;
428         struct index_state *istate = o->repo->index;
429
430         if (unmerged_index(istate)) {
431                 int i;
432                 fprintf(stderr, "BUG: There are unmerged index entries:\n");
433                 for (i = 0; i < istate->cache_nr; i++) {
434                         const struct cache_entry *ce = istate->cache[i];
435                         if (ce_stage(ce))
436                                 fprintf(stderr, "BUG: %d %.*s\n", ce_stage(ce),
437                                         (int)ce_namelen(ce), ce->name);
438                 }
439                 BUG("unmerged index entries in merge-recursive.c");
440         }
441
442         if (!istate->cache_tree)
443                 istate->cache_tree = cache_tree();
444
445         if (!cache_tree_fully_valid(istate->cache_tree) &&
446             cache_tree_update(istate, 0) < 0) {
447                 err(o, _("error building trees"));
448                 return NULL;
449         }
450
451         result = lookup_tree(o->repo, &istate->cache_tree->oid);
452
453         return result;
454 }
455
456 static int save_files_dirs(const struct object_id *oid,
457                            struct strbuf *base, const char *path,
458                            unsigned int mode, int stage, void *context)
459 {
460         struct path_hashmap_entry *entry;
461         int baselen = base->len;
462         struct merge_options *o = context;
463
464         strbuf_addstr(base, path);
465
466         FLEX_ALLOC_MEM(entry, path, base->buf, base->len);
467         hashmap_entry_init(entry, path_hash(entry->path));
468         hashmap_add(&o->current_file_dir_set, entry);
469
470         strbuf_setlen(base, baselen);
471         return (S_ISDIR(mode) ? READ_TREE_RECURSIVE : 0);
472 }
473
474 static void get_files_dirs(struct merge_options *o, struct tree *tree)
475 {
476         struct pathspec match_all;
477         memset(&match_all, 0, sizeof(match_all));
478         read_tree_recursive(the_repository, tree, "", 0, 0,
479                             &match_all, save_files_dirs, o);
480 }
481
482 static int get_tree_entry_if_blob(const struct object_id *tree,
483                                   const char *path,
484                                   struct object_id *hashy,
485                                   unsigned int *mode_o)
486 {
487         int ret;
488
489         ret = get_tree_entry(tree, path, hashy, mode_o);
490         if (S_ISDIR(*mode_o)) {
491                 oidcpy(hashy, &null_oid);
492                 *mode_o = 0;
493         }
494         return ret;
495 }
496
497 /*
498  * Returns an index_entry instance which doesn't have to correspond to
499  * a real cache entry in Git's index.
500  */
501 static struct stage_data *insert_stage_data(const char *path,
502                 struct tree *o, struct tree *a, struct tree *b,
503                 struct string_list *entries)
504 {
505         struct string_list_item *item;
506         struct stage_data *e = xcalloc(1, sizeof(struct stage_data));
507         get_tree_entry_if_blob(&o->object.oid, path,
508                                &e->stages[1].oid, &e->stages[1].mode);
509         get_tree_entry_if_blob(&a->object.oid, path,
510                                &e->stages[2].oid, &e->stages[2].mode);
511         get_tree_entry_if_blob(&b->object.oid, path,
512                                &e->stages[3].oid, &e->stages[3].mode);
513         item = string_list_insert(entries, path);
514         item->util = e;
515         return e;
516 }
517
518 /*
519  * Create a dictionary mapping file names to stage_data objects. The
520  * dictionary contains one entry for every path with a non-zero stage entry.
521  */
522 static struct string_list *get_unmerged(struct index_state *istate)
523 {
524         struct string_list *unmerged = xcalloc(1, sizeof(struct string_list));
525         int i;
526
527         unmerged->strdup_strings = 1;
528
529         for (i = 0; i < istate->cache_nr; i++) {
530                 struct string_list_item *item;
531                 struct stage_data *e;
532                 const struct cache_entry *ce = istate->cache[i];
533                 if (!ce_stage(ce))
534                         continue;
535
536                 item = string_list_lookup(unmerged, ce->name);
537                 if (!item) {
538                         item = string_list_insert(unmerged, ce->name);
539                         item->util = xcalloc(1, sizeof(struct stage_data));
540                 }
541                 e = item->util;
542                 e->stages[ce_stage(ce)].mode = ce->ce_mode;
543                 oidcpy(&e->stages[ce_stage(ce)].oid, &ce->oid);
544         }
545
546         return unmerged;
547 }
548
549 static int string_list_df_name_compare(const char *one, const char *two)
550 {
551         int onelen = strlen(one);
552         int twolen = strlen(two);
553         /*
554          * Here we only care that entries for D/F conflicts are
555          * adjacent, in particular with the file of the D/F conflict
556          * appearing before files below the corresponding directory.
557          * The order of the rest of the list is irrelevant for us.
558          *
559          * To achieve this, we sort with df_name_compare and provide
560          * the mode S_IFDIR so that D/F conflicts will sort correctly.
561          * We use the mode S_IFDIR for everything else for simplicity,
562          * since in other cases any changes in their order due to
563          * sorting cause no problems for us.
564          */
565         int cmp = df_name_compare(one, onelen, S_IFDIR,
566                                   two, twolen, S_IFDIR);
567         /*
568          * Now that 'foo' and 'foo/bar' compare equal, we have to make sure
569          * that 'foo' comes before 'foo/bar'.
570          */
571         if (cmp)
572                 return cmp;
573         return onelen - twolen;
574 }
575
576 static void record_df_conflict_files(struct merge_options *o,
577                                      struct string_list *entries)
578 {
579         /* If there is a D/F conflict and the file for such a conflict
580          * currently exists in the working tree, we want to allow it to be
581          * removed to make room for the corresponding directory if needed.
582          * The files underneath the directories of such D/F conflicts will
583          * be processed before the corresponding file involved in the D/F
584          * conflict.  If the D/F directory ends up being removed by the
585          * merge, then we won't have to touch the D/F file.  If the D/F
586          * directory needs to be written to the working copy, then the D/F
587          * file will simply be removed (in make_room_for_path()) to make
588          * room for the necessary paths.  Note that if both the directory
589          * and the file need to be present, then the D/F file will be
590          * reinstated with a new unique name at the time it is processed.
591          */
592         struct string_list df_sorted_entries = STRING_LIST_INIT_NODUP;
593         const char *last_file = NULL;
594         int last_len = 0;
595         int i;
596
597         /*
598          * If we're merging merge-bases, we don't want to bother with
599          * any working directory changes.
600          */
601         if (o->call_depth)
602                 return;
603
604         /* Ensure D/F conflicts are adjacent in the entries list. */
605         for (i = 0; i < entries->nr; i++) {
606                 struct string_list_item *next = &entries->items[i];
607                 string_list_append(&df_sorted_entries, next->string)->util =
608                                    next->util;
609         }
610         df_sorted_entries.cmp = string_list_df_name_compare;
611         string_list_sort(&df_sorted_entries);
612
613         string_list_clear(&o->df_conflict_file_set, 1);
614         for (i = 0; i < df_sorted_entries.nr; i++) {
615                 const char *path = df_sorted_entries.items[i].string;
616                 int len = strlen(path);
617                 struct stage_data *e = df_sorted_entries.items[i].util;
618
619                 /*
620                  * Check if last_file & path correspond to a D/F conflict;
621                  * i.e. whether path is last_file+'/'+<something>.
622                  * If so, record that it's okay to remove last_file to make
623                  * room for path and friends if needed.
624                  */
625                 if (last_file &&
626                     len > last_len &&
627                     memcmp(path, last_file, last_len) == 0 &&
628                     path[last_len] == '/') {
629                         string_list_insert(&o->df_conflict_file_set, last_file);
630                 }
631
632                 /*
633                  * Determine whether path could exist as a file in the
634                  * working directory as a possible D/F conflict.  This
635                  * will only occur when it exists in stage 2 as a
636                  * file.
637                  */
638                 if (S_ISREG(e->stages[2].mode) || S_ISLNK(e->stages[2].mode)) {
639                         last_file = path;
640                         last_len = len;
641                 } else {
642                         last_file = NULL;
643                 }
644         }
645         string_list_clear(&df_sorted_entries, 0);
646 }
647
648 struct rename {
649         struct diff_filepair *pair;
650         /*
651          * Purpose of src_entry and dst_entry:
652          *
653          * If 'before' is renamed to 'after' then src_entry will contain
654          * the versions of 'before' from the merge_base, HEAD, and MERGE in
655          * stages 1, 2, and 3; dst_entry will contain the respective
656          * versions of 'after' in corresponding locations.  Thus, we have a
657          * total of six modes and oids, though some will be null.  (Stage 0
658          * is ignored; we're interested in handling conflicts.)
659          *
660          * Since we don't turn on break-rewrites by default, neither
661          * src_entry nor dst_entry can have all three of their stages have
662          * non-null oids, meaning at most four of the six will be non-null.
663          * Also, since this is a rename, both src_entry and dst_entry will
664          * have at least one non-null oid, meaning at least two will be
665          * non-null.  Of the six oids, a typical rename will have three be
666          * non-null.  Only two implies a rename/delete, and four implies a
667          * rename/add.
668          */
669         struct stage_data *src_entry;
670         struct stage_data *dst_entry;
671         unsigned add_turned_into_rename:1;
672         unsigned processed:1;
673 };
674
675 static int update_stages(struct merge_options *opt, const char *path,
676                          const struct diff_filespec *o,
677                          const struct diff_filespec *a,
678                          const struct diff_filespec *b)
679 {
680
681         /*
682          * NOTE: It is usually a bad idea to call update_stages on a path
683          * before calling update_file on that same path, since it can
684          * sometimes lead to spurious "refusing to lose untracked file..."
685          * messages from update_file (via make_room_for path via
686          * would_lose_untracked).  Instead, reverse the order of the calls
687          * (executing update_file first and then update_stages).
688          */
689         int clear = 1;
690         int options = ADD_CACHE_OK_TO_ADD | ADD_CACHE_SKIP_DFCHECK;
691         if (clear)
692                 if (remove_file_from_index(opt->repo->index, path))
693                         return -1;
694         if (o)
695                 if (add_cacheinfo(opt, o->mode, &o->oid, path, 1, 0, options))
696                         return -1;
697         if (a)
698                 if (add_cacheinfo(opt, a->mode, &a->oid, path, 2, 0, options))
699                         return -1;
700         if (b)
701                 if (add_cacheinfo(opt, b->mode, &b->oid, path, 3, 0, options))
702                         return -1;
703         return 0;
704 }
705
706 static void update_entry(struct stage_data *entry,
707                          struct diff_filespec *o,
708                          struct diff_filespec *a,
709                          struct diff_filespec *b)
710 {
711         entry->processed = 0;
712         entry->stages[1].mode = o->mode;
713         entry->stages[2].mode = a->mode;
714         entry->stages[3].mode = b->mode;
715         oidcpy(&entry->stages[1].oid, &o->oid);
716         oidcpy(&entry->stages[2].oid, &a->oid);
717         oidcpy(&entry->stages[3].oid, &b->oid);
718 }
719
720 static int remove_file(struct merge_options *o, int clean,
721                        const char *path, int no_wd)
722 {
723         int update_cache = o->call_depth || clean;
724         int update_working_directory = !o->call_depth && !no_wd;
725
726         if (update_cache) {
727                 if (remove_file_from_index(o->repo->index, path))
728                         return -1;
729         }
730         if (update_working_directory) {
731                 if (ignore_case) {
732                         struct cache_entry *ce;
733                         ce = index_file_exists(o->repo->index, path, strlen(path),
734                                                ignore_case);
735                         if (ce && ce_stage(ce) == 0 && strcmp(path, ce->name))
736                                 return 0;
737                 }
738                 if (remove_path(path))
739                         return -1;
740         }
741         return 0;
742 }
743
744 /* add a string to a strbuf, but converting "/" to "_" */
745 static void add_flattened_path(struct strbuf *out, const char *s)
746 {
747         size_t i = out->len;
748         strbuf_addstr(out, s);
749         for (; i < out->len; i++)
750                 if (out->buf[i] == '/')
751                         out->buf[i] = '_';
752 }
753
754 static char *unique_path(struct merge_options *o, const char *path, const char *branch)
755 {
756         struct path_hashmap_entry *entry;
757         struct strbuf newpath = STRBUF_INIT;
758         int suffix = 0;
759         size_t base_len;
760
761         strbuf_addf(&newpath, "%s~", path);
762         add_flattened_path(&newpath, branch);
763
764         base_len = newpath.len;
765         while (hashmap_get_from_hash(&o->current_file_dir_set,
766                                      path_hash(newpath.buf), newpath.buf) ||
767                (!o->call_depth && file_exists(newpath.buf))) {
768                 strbuf_setlen(&newpath, base_len);
769                 strbuf_addf(&newpath, "_%d", suffix++);
770         }
771
772         FLEX_ALLOC_MEM(entry, path, newpath.buf, newpath.len);
773         hashmap_entry_init(entry, path_hash(entry->path));
774         hashmap_add(&o->current_file_dir_set, entry);
775         return strbuf_detach(&newpath, NULL);
776 }
777
778 /**
779  * Check whether a directory in the index is in the way of an incoming
780  * file.  Return 1 if so.  If check_working_copy is non-zero, also
781  * check the working directory.  If empty_ok is non-zero, also return
782  * 0 in the case where the working-tree dir exists but is empty.
783  */
784 static int dir_in_way(struct index_state *istate, const char *path,
785                       int check_working_copy, int empty_ok)
786 {
787         int pos;
788         struct strbuf dirpath = STRBUF_INIT;
789         struct stat st;
790
791         strbuf_addstr(&dirpath, path);
792         strbuf_addch(&dirpath, '/');
793
794         pos = index_name_pos(istate, dirpath.buf, dirpath.len);
795
796         if (pos < 0)
797                 pos = -1 - pos;
798         if (pos < istate->cache_nr &&
799             !strncmp(dirpath.buf, istate->cache[pos]->name, dirpath.len)) {
800                 strbuf_release(&dirpath);
801                 return 1;
802         }
803
804         strbuf_release(&dirpath);
805         return check_working_copy && !lstat(path, &st) && S_ISDIR(st.st_mode) &&
806                 !(empty_ok && is_empty_dir(path));
807 }
808
809 /*
810  * Returns whether path was tracked in the index before the merge started,
811  * and its oid and mode match the specified values
812  */
813 static int was_tracked_and_matches(struct merge_options *o, const char *path,
814                                    const struct object_id *oid, unsigned mode)
815 {
816         int pos = index_name_pos(&o->orig_index, path, strlen(path));
817         struct cache_entry *ce;
818
819         if (0 > pos)
820                 /* we were not tracking this path before the merge */
821                 return 0;
822
823         /* See if the file we were tracking before matches */
824         ce = o->orig_index.cache[pos];
825         return (oid_eq(&ce->oid, oid) && ce->ce_mode == mode);
826 }
827
828 /*
829  * Returns whether path was tracked in the index before the merge started
830  */
831 static int was_tracked(struct merge_options *o, const char *path)
832 {
833         int pos = index_name_pos(&o->orig_index, path, strlen(path));
834
835         if (0 <= pos)
836                 /* we were tracking this path before the merge */
837                 return 1;
838
839         return 0;
840 }
841
842 static int would_lose_untracked(struct merge_options *o, const char *path)
843 {
844         struct index_state *istate = o->repo->index;
845
846         /*
847          * This may look like it can be simplified to:
848          *   return !was_tracked(o, path) && file_exists(path)
849          * but it can't.  This function needs to know whether path was in
850          * the working tree due to EITHER having been tracked in the index
851          * before the merge OR having been put into the working copy and
852          * index by unpack_trees().  Due to that either-or requirement, we
853          * check the current index instead of the original one.
854          *
855          * Note that we do not need to worry about merge-recursive itself
856          * updating the index after unpack_trees() and before calling this
857          * function, because we strictly require all code paths in
858          * merge-recursive to update the working tree first and the index
859          * second.  Doing otherwise would break
860          * update_file()/would_lose_untracked(); see every comment in this
861          * file which mentions "update_stages".
862          */
863         int pos = index_name_pos(istate, path, strlen(path));
864
865         if (pos < 0)
866                 pos = -1 - pos;
867         while (pos < istate->cache_nr &&
868                !strcmp(path, istate->cache[pos]->name)) {
869                 /*
870                  * If stage #0, it is definitely tracked.
871                  * If it has stage #2 then it was tracked
872                  * before this merge started.  All other
873                  * cases the path was not tracked.
874                  */
875                 switch (ce_stage(istate->cache[pos])) {
876                 case 0:
877                 case 2:
878                         return 0;
879                 }
880                 pos++;
881         }
882         return file_exists(path);
883 }
884
885 static int was_dirty(struct merge_options *o, const char *path)
886 {
887         struct cache_entry *ce;
888         int dirty = 1;
889
890         if (o->call_depth || !was_tracked(o, path))
891                 return !dirty;
892
893         ce = index_file_exists(o->unpack_opts.src_index,
894                                path, strlen(path), ignore_case);
895         dirty = verify_uptodate(ce, &o->unpack_opts) != 0;
896         return dirty;
897 }
898
899 static int make_room_for_path(struct merge_options *o, const char *path)
900 {
901         int status, i;
902         const char *msg = _("failed to create path '%s'%s");
903
904         /* Unlink any D/F conflict files that are in the way */
905         for (i = 0; i < o->df_conflict_file_set.nr; i++) {
906                 const char *df_path = o->df_conflict_file_set.items[i].string;
907                 size_t pathlen = strlen(path);
908                 size_t df_pathlen = strlen(df_path);
909                 if (df_pathlen < pathlen &&
910                     path[df_pathlen] == '/' &&
911                     strncmp(path, df_path, df_pathlen) == 0) {
912                         output(o, 3,
913                                _("Removing %s to make room for subdirectory\n"),
914                                df_path);
915                         unlink(df_path);
916                         unsorted_string_list_delete_item(&o->df_conflict_file_set,
917                                                          i, 0);
918                         break;
919                 }
920         }
921
922         /* Make sure leading directories are created */
923         status = safe_create_leading_directories_const(path);
924         if (status) {
925                 if (status == SCLD_EXISTS)
926                         /* something else exists */
927                         return err(o, msg, path, _(": perhaps a D/F conflict?"));
928                 return err(o, msg, path, "");
929         }
930
931         /*
932          * Do not unlink a file in the work tree if we are not
933          * tracking it.
934          */
935         if (would_lose_untracked(o, path))
936                 return err(o, _("refusing to lose untracked file at '%s'"),
937                            path);
938
939         /* Successful unlink is good.. */
940         if (!unlink(path))
941                 return 0;
942         /* .. and so is no existing file */
943         if (errno == ENOENT)
944                 return 0;
945         /* .. but not some other error (who really cares what?) */
946         return err(o, msg, path, _(": perhaps a D/F conflict?"));
947 }
948
949 static int update_file_flags(struct merge_options *o,
950                              const struct object_id *oid,
951                              unsigned mode,
952                              const char *path,
953                              int update_cache,
954                              int update_wd)
955 {
956         int ret = 0;
957
958         if (o->call_depth)
959                 update_wd = 0;
960
961         if (update_wd) {
962                 enum object_type type;
963                 void *buf;
964                 unsigned long size;
965
966                 if (S_ISGITLINK(mode)) {
967                         /*
968                          * We may later decide to recursively descend into
969                          * the submodule directory and update its index
970                          * and/or work tree, but we do not do that now.
971                          */
972                         update_wd = 0;
973                         goto update_index;
974                 }
975
976                 buf = read_object_file(oid, &type, &size);
977                 if (!buf)
978                         return err(o, _("cannot read object %s '%s'"), oid_to_hex(oid), path);
979                 if (type != OBJ_BLOB) {
980                         ret = err(o, _("blob expected for %s '%s'"), oid_to_hex(oid), path);
981                         goto free_buf;
982                 }
983                 if (S_ISREG(mode)) {
984                         struct strbuf strbuf = STRBUF_INIT;
985                         if (convert_to_working_tree(o->repo->index, path, buf, size, &strbuf)) {
986                                 free(buf);
987                                 size = strbuf.len;
988                                 buf = strbuf_detach(&strbuf, NULL);
989                         }
990                 }
991
992                 if (make_room_for_path(o, path) < 0) {
993                         update_wd = 0;
994                         goto free_buf;
995                 }
996                 if (S_ISREG(mode) || (!has_symlinks && S_ISLNK(mode))) {
997                         int fd;
998                         if (mode & 0100)
999                                 mode = 0777;
1000                         else
1001                                 mode = 0666;
1002                         fd = open(path, O_WRONLY | O_TRUNC | O_CREAT, mode);
1003                         if (fd < 0) {
1004                                 ret = err(o, _("failed to open '%s': %s"),
1005                                           path, strerror(errno));
1006                                 goto free_buf;
1007                         }
1008                         write_in_full(fd, buf, size);
1009                         close(fd);
1010                 } else if (S_ISLNK(mode)) {
1011                         char *lnk = xmemdupz(buf, size);
1012                         safe_create_leading_directories_const(path);
1013                         unlink(path);
1014                         if (symlink(lnk, path))
1015                                 ret = err(o, _("failed to symlink '%s': %s"),
1016                                           path, strerror(errno));
1017                         free(lnk);
1018                 } else
1019                         ret = err(o,
1020                                   _("do not know what to do with %06o %s '%s'"),
1021                                   mode, oid_to_hex(oid), path);
1022         free_buf:
1023                 free(buf);
1024         }
1025 update_index:
1026         if (!ret && update_cache)
1027                 if (add_cacheinfo(o, mode, oid, path, 0, update_wd,
1028                                   ADD_CACHE_OK_TO_ADD))
1029                         return -1;
1030         return ret;
1031 }
1032
1033 static int update_file(struct merge_options *o,
1034                        int clean,
1035                        const struct object_id *oid,
1036                        unsigned mode,
1037                        const char *path)
1038 {
1039         return update_file_flags(o, oid, mode, path, o->call_depth || clean, !o->call_depth);
1040 }
1041
1042 /* Low level file merging, update and removal */
1043
1044 struct merge_file_info {
1045         struct object_id oid;
1046         unsigned mode;
1047         unsigned clean:1,
1048                  merge:1;
1049 };
1050
1051 static int merge_3way(struct merge_options *o,
1052                       mmbuffer_t *result_buf,
1053                       const struct diff_filespec *one,
1054                       const struct diff_filespec *a,
1055                       const struct diff_filespec *b,
1056                       const char *branch1,
1057                       const char *branch2,
1058                       const int extra_marker_size)
1059 {
1060         mmfile_t orig, src1, src2;
1061         struct ll_merge_options ll_opts = {0};
1062         char *base_name, *name1, *name2;
1063         int merge_status;
1064
1065         ll_opts.renormalize = o->renormalize;
1066         ll_opts.extra_marker_size = extra_marker_size;
1067         ll_opts.xdl_opts = o->xdl_opts;
1068
1069         if (o->call_depth) {
1070                 ll_opts.virtual_ancestor = 1;
1071                 ll_opts.variant = 0;
1072         } else {
1073                 switch (o->recursive_variant) {
1074                 case MERGE_RECURSIVE_OURS:
1075                         ll_opts.variant = XDL_MERGE_FAVOR_OURS;
1076                         break;
1077                 case MERGE_RECURSIVE_THEIRS:
1078                         ll_opts.variant = XDL_MERGE_FAVOR_THEIRS;
1079                         break;
1080                 default:
1081                         ll_opts.variant = 0;
1082                         break;
1083                 }
1084         }
1085
1086         if (strcmp(a->path, b->path) ||
1087             (o->ancestor != NULL && strcmp(a->path, one->path) != 0)) {
1088                 base_name = o->ancestor == NULL ? NULL :
1089                         mkpathdup("%s:%s", o->ancestor, one->path);
1090                 name1 = mkpathdup("%s:%s", branch1, a->path);
1091                 name2 = mkpathdup("%s:%s", branch2, b->path);
1092         } else {
1093                 base_name = o->ancestor == NULL ? NULL :
1094                         mkpathdup("%s", o->ancestor);
1095                 name1 = mkpathdup("%s", branch1);
1096                 name2 = mkpathdup("%s", branch2);
1097         }
1098
1099         read_mmblob(&orig, &one->oid);
1100         read_mmblob(&src1, &a->oid);
1101         read_mmblob(&src2, &b->oid);
1102
1103         merge_status = ll_merge(result_buf, a->path, &orig, base_name,
1104                                 &src1, name1, &src2, name2,
1105                                 o->repo->index, &ll_opts);
1106
1107         free(base_name);
1108         free(name1);
1109         free(name2);
1110         free(orig.ptr);
1111         free(src1.ptr);
1112         free(src2.ptr);
1113         return merge_status;
1114 }
1115
1116 static int find_first_merges(struct repository *repo,
1117                              struct object_array *result, const char *path,
1118                              struct commit *a, struct commit *b)
1119 {
1120         int i, j;
1121         struct object_array merges = OBJECT_ARRAY_INIT;
1122         struct commit *commit;
1123         int contains_another;
1124
1125         char merged_revision[42];
1126         const char *rev_args[] = { "rev-list", "--merges", "--ancestry-path",
1127                                    "--all", merged_revision, NULL };
1128         struct rev_info revs;
1129         struct setup_revision_opt rev_opts;
1130
1131         memset(result, 0, sizeof(struct object_array));
1132         memset(&rev_opts, 0, sizeof(rev_opts));
1133
1134         /* get all revisions that merge commit a */
1135         xsnprintf(merged_revision, sizeof(merged_revision), "^%s",
1136                   oid_to_hex(&a->object.oid));
1137         repo_init_revisions(repo, &revs, NULL);
1138         rev_opts.submodule = path;
1139         /* FIXME: can't handle linked worktrees in submodules yet */
1140         revs.single_worktree = path != NULL;
1141         setup_revisions(ARRAY_SIZE(rev_args)-1, rev_args, &revs, &rev_opts);
1142
1143         /* save all revisions from the above list that contain b */
1144         if (prepare_revision_walk(&revs))
1145                 die("revision walk setup failed");
1146         while ((commit = get_revision(&revs)) != NULL) {
1147                 struct object *o = &(commit->object);
1148                 if (in_merge_bases(b, commit))
1149                         add_object_array(o, NULL, &merges);
1150         }
1151         reset_revision_walk();
1152
1153         /* Now we've got all merges that contain a and b. Prune all
1154          * merges that contain another found merge and save them in
1155          * result.
1156          */
1157         for (i = 0; i < merges.nr; i++) {
1158                 struct commit *m1 = (struct commit *) merges.objects[i].item;
1159
1160                 contains_another = 0;
1161                 for (j = 0; j < merges.nr; j++) {
1162                         struct commit *m2 = (struct commit *) merges.objects[j].item;
1163                         if (i != j && in_merge_bases(m2, m1)) {
1164                                 contains_another = 1;
1165                                 break;
1166                         }
1167                 }
1168
1169                 if (!contains_another)
1170                         add_object_array(merges.objects[i].item, NULL, result);
1171         }
1172
1173         object_array_clear(&merges);
1174         return result->nr;
1175 }
1176
1177 static void print_commit(struct commit *commit)
1178 {
1179         struct strbuf sb = STRBUF_INIT;
1180         struct pretty_print_context ctx = {0};
1181         ctx.date_mode.type = DATE_NORMAL;
1182         format_commit_message(commit, " %h: %m %s", &sb, &ctx);
1183         fprintf(stderr, "%s\n", sb.buf);
1184         strbuf_release(&sb);
1185 }
1186
1187 static int merge_submodule(struct merge_options *o,
1188                            struct object_id *result, const char *path,
1189                            const struct object_id *base, const struct object_id *a,
1190                            const struct object_id *b)
1191 {
1192         struct commit *commit_base, *commit_a, *commit_b;
1193         int parent_count;
1194         struct object_array merges;
1195
1196         int i;
1197         int search = !o->call_depth;
1198
1199         /* store a in result in case we fail */
1200         oidcpy(result, a);
1201
1202         /* we can not handle deletion conflicts */
1203         if (is_null_oid(base))
1204                 return 0;
1205         if (is_null_oid(a))
1206                 return 0;
1207         if (is_null_oid(b))
1208                 return 0;
1209
1210         if (add_submodule_odb(path)) {
1211                 output(o, 1, _("Failed to merge submodule %s (not checked out)"), path);
1212                 return 0;
1213         }
1214
1215         if (!(commit_base = lookup_commit_reference(o->repo, base)) ||
1216             !(commit_a = lookup_commit_reference(o->repo, a)) ||
1217             !(commit_b = lookup_commit_reference(o->repo, b))) {
1218                 output(o, 1, _("Failed to merge submodule %s (commits not present)"), path);
1219                 return 0;
1220         }
1221
1222         /* check whether both changes are forward */
1223         if (!in_merge_bases(commit_base, commit_a) ||
1224             !in_merge_bases(commit_base, commit_b)) {
1225                 output(o, 1, _("Failed to merge submodule %s (commits don't follow merge-base)"), path);
1226                 return 0;
1227         }
1228
1229         /* Case #1: a is contained in b or vice versa */
1230         if (in_merge_bases(commit_a, commit_b)) {
1231                 oidcpy(result, b);
1232                 if (show(o, 3)) {
1233                         output(o, 3, _("Fast-forwarding submodule %s to the following commit:"), path);
1234                         output_commit_title(o, commit_b);
1235                 } else if (show(o, 2))
1236                         output(o, 2, _("Fast-forwarding submodule %s"), path);
1237                 else
1238                         ; /* no output */
1239
1240                 return 1;
1241         }
1242         if (in_merge_bases(commit_b, commit_a)) {
1243                 oidcpy(result, a);
1244                 if (show(o, 3)) {
1245                         output(o, 3, _("Fast-forwarding submodule %s to the following commit:"), path);
1246                         output_commit_title(o, commit_a);
1247                 } else if (show(o, 2))
1248                         output(o, 2, _("Fast-forwarding submodule %s"), path);
1249                 else
1250                         ; /* no output */
1251
1252                 return 1;
1253         }
1254
1255         /*
1256          * Case #2: There are one or more merges that contain a and b in
1257          * the submodule. If there is only one, then present it as a
1258          * suggestion to the user, but leave it marked unmerged so the
1259          * user needs to confirm the resolution.
1260          */
1261
1262         /* Skip the search if makes no sense to the calling context.  */
1263         if (!search)
1264                 return 0;
1265
1266         /* find commit which merges them */
1267         parent_count = find_first_merges(o->repo, &merges, path,
1268                                          commit_a, commit_b);
1269         switch (parent_count) {
1270         case 0:
1271                 output(o, 1, _("Failed to merge submodule %s (merge following commits not found)"), path);
1272                 break;
1273
1274         case 1:
1275                 output(o, 1, _("Failed to merge submodule %s (not fast-forward)"), path);
1276                 output(o, 2, _("Found a possible merge resolution for the submodule:\n"));
1277                 print_commit((struct commit *) merges.objects[0].item);
1278                 output(o, 2, _(
1279                        "If this is correct simply add it to the index "
1280                        "for example\n"
1281                        "by using:\n\n"
1282                        "  git update-index --cacheinfo 160000 %s \"%s\"\n\n"
1283                        "which will accept this suggestion.\n"),
1284                        oid_to_hex(&merges.objects[0].item->oid), path);
1285                 break;
1286
1287         default:
1288                 output(o, 1, _("Failed to merge submodule %s (multiple merges found)"), path);
1289                 for (i = 0; i < merges.nr; i++)
1290                         print_commit((struct commit *) merges.objects[i].item);
1291         }
1292
1293         object_array_clear(&merges);
1294         return 0;
1295 }
1296
1297 static int merge_mode_and_contents(struct merge_options *o,
1298                                    const struct diff_filespec *one,
1299                                    const struct diff_filespec *a,
1300                                    const struct diff_filespec *b,
1301                                    const char *filename,
1302                                    const char *branch1,
1303                                    const char *branch2,
1304                                    const int extra_marker_size,
1305                                    struct merge_file_info *result)
1306 {
1307         if (o->branch1 != branch1) {
1308                 /*
1309                  * It's weird getting a reverse merge with HEAD on the bottom
1310                  * side of the conflict markers and the other branch on the
1311                  * top.  Fix that.
1312                  */
1313                 return merge_mode_and_contents(o, one, b, a,
1314                                                filename,
1315                                                branch2, branch1,
1316                                                extra_marker_size, result);
1317         }
1318
1319         result->merge = 0;
1320         result->clean = 1;
1321
1322         if ((S_IFMT & a->mode) != (S_IFMT & b->mode)) {
1323                 result->clean = 0;
1324                 if (S_ISREG(a->mode)) {
1325                         result->mode = a->mode;
1326                         oidcpy(&result->oid, &a->oid);
1327                 } else {
1328                         result->mode = b->mode;
1329                         oidcpy(&result->oid, &b->oid);
1330                 }
1331         } else {
1332                 if (!oid_eq(&a->oid, &one->oid) && !oid_eq(&b->oid, &one->oid))
1333                         result->merge = 1;
1334
1335                 /*
1336                  * Merge modes
1337                  */
1338                 if (a->mode == b->mode || a->mode == one->mode)
1339                         result->mode = b->mode;
1340                 else {
1341                         result->mode = a->mode;
1342                         if (b->mode != one->mode) {
1343                                 result->clean = 0;
1344                                 result->merge = 1;
1345                         }
1346                 }
1347
1348                 if (oid_eq(&a->oid, &b->oid) || oid_eq(&a->oid, &one->oid))
1349                         oidcpy(&result->oid, &b->oid);
1350                 else if (oid_eq(&b->oid, &one->oid))
1351                         oidcpy(&result->oid, &a->oid);
1352                 else if (S_ISREG(a->mode)) {
1353                         mmbuffer_t result_buf;
1354                         int ret = 0, merge_status;
1355
1356                         merge_status = merge_3way(o, &result_buf, one, a, b,
1357                                                   branch1, branch2,
1358                                                   extra_marker_size);
1359
1360                         if ((merge_status < 0) || !result_buf.ptr)
1361                                 ret = err(o, _("Failed to execute internal merge"));
1362
1363                         if (!ret &&
1364                             write_object_file(result_buf.ptr, result_buf.size,
1365                                               blob_type, &result->oid))
1366                                 ret = err(o, _("Unable to add %s to database"),
1367                                           a->path);
1368
1369                         free(result_buf.ptr);
1370                         if (ret)
1371                                 return ret;
1372                         result->clean = (merge_status == 0);
1373                 } else if (S_ISGITLINK(a->mode)) {
1374                         result->clean = merge_submodule(o, &result->oid,
1375                                                         one->path,
1376                                                         &one->oid,
1377                                                         &a->oid,
1378                                                         &b->oid);
1379                 } else if (S_ISLNK(a->mode)) {
1380                         switch (o->recursive_variant) {
1381                         case MERGE_RECURSIVE_NORMAL:
1382                                 oidcpy(&result->oid, &a->oid);
1383                                 if (!oid_eq(&a->oid, &b->oid))
1384                                         result->clean = 0;
1385                                 break;
1386                         case MERGE_RECURSIVE_OURS:
1387                                 oidcpy(&result->oid, &a->oid);
1388                                 break;
1389                         case MERGE_RECURSIVE_THEIRS:
1390                                 oidcpy(&result->oid, &b->oid);
1391                                 break;
1392                         }
1393                 } else
1394                         BUG("unsupported object type in the tree");
1395         }
1396
1397         if (result->merge)
1398                 output(o, 2, _("Auto-merging %s"), filename);
1399
1400         return 0;
1401 }
1402
1403 static int handle_rename_via_dir(struct merge_options *o,
1404                                  struct diff_filepair *pair,
1405                                  const char *rename_branch)
1406 {
1407         /*
1408          * Handle file adds that need to be renamed due to directory rename
1409          * detection.  This differs from handle_rename_normal, because
1410          * there is no content merge to do; just move the file into the
1411          * desired final location.
1412          */
1413         const struct diff_filespec *dest = pair->two;
1414
1415         if (!o->call_depth && would_lose_untracked(o, dest->path)) {
1416                 char *alt_path = unique_path(o, dest->path, rename_branch);
1417
1418                 output(o, 1, _("Error: Refusing to lose untracked file at %s; "
1419                                "writing to %s instead."),
1420                        dest->path, alt_path);
1421                 /*
1422                  * Write the file in worktree at alt_path, but not in the
1423                  * index.  Instead, write to dest->path for the index but
1424                  * only at the higher appropriate stage.
1425                  */
1426                 if (update_file(o, 0, &dest->oid, dest->mode, alt_path))
1427                         return -1;
1428                 free(alt_path);
1429                 return update_stages(o, dest->path, NULL,
1430                                      rename_branch == o->branch1 ? dest : NULL,
1431                                      rename_branch == o->branch1 ? NULL : dest);
1432         }
1433
1434         /* Update dest->path both in index and in worktree */
1435         if (update_file(o, 1, &dest->oid, dest->mode, dest->path))
1436                 return -1;
1437         return 0;
1438 }
1439
1440 static int handle_change_delete(struct merge_options *o,
1441                                 const char *path, const char *old_path,
1442                                 const struct object_id *o_oid, int o_mode,
1443                                 const struct object_id *changed_oid,
1444                                 int changed_mode,
1445                                 const char *change_branch,
1446                                 const char *delete_branch,
1447                                 const char *change, const char *change_past)
1448 {
1449         char *alt_path = NULL;
1450         const char *update_path = path;
1451         int ret = 0;
1452
1453         if (dir_in_way(o->repo->index, path, !o->call_depth, 0) ||
1454             (!o->call_depth && would_lose_untracked(o, path))) {
1455                 update_path = alt_path = unique_path(o, path, change_branch);
1456         }
1457
1458         if (o->call_depth) {
1459                 /*
1460                  * We cannot arbitrarily accept either a_sha or b_sha as
1461                  * correct; since there is no true "middle point" between
1462                  * them, simply reuse the base version for virtual merge base.
1463                  */
1464                 ret = remove_file_from_index(o->repo->index, path);
1465                 if (!ret)
1466                         ret = update_file(o, 0, o_oid, o_mode, update_path);
1467         } else {
1468                 /*
1469                  * Despite the four nearly duplicate messages and argument
1470                  * lists below and the ugliness of the nested if-statements,
1471                  * having complete messages makes the job easier for
1472                  * translators.
1473                  *
1474                  * The slight variance among the cases is due to the fact
1475                  * that:
1476                  *   1) directory/file conflicts (in effect if
1477                  *      !alt_path) could cause us to need to write the
1478                  *      file to a different path.
1479                  *   2) renames (in effect if !old_path) could mean that
1480                  *      there are two names for the path that the user
1481                  *      may know the file by.
1482                  */
1483                 if (!alt_path) {
1484                         if (!old_path) {
1485                                 output(o, 1, _("CONFLICT (%s/delete): %s deleted in %s "
1486                                        "and %s in %s. Version %s of %s left in tree."),
1487                                        change, path, delete_branch, change_past,
1488                                        change_branch, change_branch, path);
1489                         } else {
1490                                 output(o, 1, _("CONFLICT (%s/delete): %s deleted in %s "
1491                                        "and %s to %s in %s. Version %s of %s left in tree."),
1492                                        change, old_path, delete_branch, change_past, path,
1493                                        change_branch, change_branch, path);
1494                         }
1495                 } else {
1496                         if (!old_path) {
1497                                 output(o, 1, _("CONFLICT (%s/delete): %s deleted in %s "
1498                                        "and %s in %s. Version %s of %s left in tree at %s."),
1499                                        change, path, delete_branch, change_past,
1500                                        change_branch, change_branch, path, alt_path);
1501                         } else {
1502                                 output(o, 1, _("CONFLICT (%s/delete): %s deleted in %s "
1503                                        "and %s to %s in %s. Version %s of %s left in tree at %s."),
1504                                        change, old_path, delete_branch, change_past, path,
1505                                        change_branch, change_branch, path, alt_path);
1506                         }
1507                 }
1508                 /*
1509                  * No need to call update_file() on path when change_branch ==
1510                  * o->branch1 && !alt_path, since that would needlessly touch
1511                  * path.  We could call update_file_flags() with update_cache=0
1512                  * and update_wd=0, but that's a no-op.
1513                  */
1514                 if (change_branch != o->branch1 || alt_path)
1515                         ret = update_file(o, 0, changed_oid, changed_mode, update_path);
1516         }
1517         free(alt_path);
1518
1519         return ret;
1520 }
1521
1522 static int handle_rename_delete(struct merge_options *o,
1523                                 struct diff_filepair *pair,
1524                                 const char *rename_branch,
1525                                 const char *delete_branch)
1526 {
1527         const struct diff_filespec *orig = pair->one;
1528         const struct diff_filespec *dest = pair->two;
1529
1530         if (handle_change_delete(o,
1531                                  o->call_depth ? orig->path : dest->path,
1532                                  o->call_depth ? NULL : orig->path,
1533                                  &orig->oid, orig->mode,
1534                                  &dest->oid, dest->mode,
1535                                  rename_branch, delete_branch,
1536                                  _("rename"), _("renamed")))
1537                 return -1;
1538
1539         if (o->call_depth)
1540                 return remove_file_from_index(o->repo->index, dest->path);
1541         else
1542                 return update_stages(o, dest->path, NULL,
1543                                      rename_branch == o->branch1 ? dest : NULL,
1544                                      rename_branch == o->branch1 ? NULL : dest);
1545 }
1546
1547 static struct diff_filespec *filespec_from_entry(struct diff_filespec *target,
1548                                                  struct stage_data *entry,
1549                                                  int stage)
1550 {
1551         struct object_id *oid = &entry->stages[stage].oid;
1552         unsigned mode = entry->stages[stage].mode;
1553         if (mode == 0 || is_null_oid(oid))
1554                 return NULL;
1555         oidcpy(&target->oid, oid);
1556         target->mode = mode;
1557         return target;
1558 }
1559
1560 static int handle_file_collision(struct merge_options *o,
1561                                  const char *collide_path,
1562                                  const char *prev_path1,
1563                                  const char *prev_path2,
1564                                  const char *branch1, const char *branch2,
1565                                  const struct object_id *a_oid,
1566                                  unsigned int a_mode,
1567                                  const struct object_id *b_oid,
1568                                  unsigned int b_mode)
1569 {
1570         struct merge_file_info mfi;
1571         struct diff_filespec null, a, b;
1572         char *alt_path = NULL;
1573         const char *update_path = collide_path;
1574
1575         /*
1576          * It's easiest to get the correct things into stage 2 and 3, and
1577          * to make sure that the content merge puts HEAD before the other
1578          * branch if we just ensure that branch1 == o->branch1.  So, simply
1579          * flip arguments around if we don't have that.
1580          */
1581         if (branch1 != o->branch1) {
1582                 return handle_file_collision(o, collide_path,
1583                                              prev_path2, prev_path1,
1584                                              branch2, branch1,
1585                                              b_oid, b_mode,
1586                                              a_oid, a_mode);
1587         }
1588
1589         /*
1590          * In the recursive case, we just opt to undo renames
1591          */
1592         if (o->call_depth && (prev_path1 || prev_path2)) {
1593                 /* Put first file (a_oid, a_mode) in its original spot */
1594                 if (prev_path1) {
1595                         if (update_file(o, 1, a_oid, a_mode, prev_path1))
1596                                 return -1;
1597                 } else {
1598                         if (update_file(o, 1, a_oid, a_mode, collide_path))
1599                                 return -1;
1600                 }
1601
1602                 /* Put second file (b_oid, b_mode) in its original spot */
1603                 if (prev_path2) {
1604                         if (update_file(o, 1, b_oid, b_mode, prev_path2))
1605                                 return -1;
1606                 } else {
1607                         if (update_file(o, 1, b_oid, b_mode, collide_path))
1608                                 return -1;
1609                 }
1610
1611                 /* Don't leave something at collision path if unrenaming both */
1612                 if (prev_path1 && prev_path2)
1613                         remove_file(o, 1, collide_path, 0);
1614
1615                 return 0;
1616         }
1617
1618         /* Remove rename sources if rename/add or rename/rename(2to1) */
1619         if (prev_path1)
1620                 remove_file(o, 1, prev_path1,
1621                             o->call_depth || would_lose_untracked(o, prev_path1));
1622         if (prev_path2)
1623                 remove_file(o, 1, prev_path2,
1624                             o->call_depth || would_lose_untracked(o, prev_path2));
1625
1626         /*
1627          * Remove the collision path, if it wouldn't cause dirty contents
1628          * or an untracked file to get lost.  We'll either overwrite with
1629          * merged contents, or just write out to differently named files.
1630          */
1631         if (was_dirty(o, collide_path)) {
1632                 output(o, 1, _("Refusing to lose dirty file at %s"),
1633                        collide_path);
1634                 update_path = alt_path = unique_path(o, collide_path, "merged");
1635         } else if (would_lose_untracked(o, collide_path)) {
1636                 /*
1637                  * Only way we get here is if both renames were from
1638                  * a directory rename AND user had an untracked file
1639                  * at the location where both files end up after the
1640                  * two directory renames.  See testcase 10d of t6043.
1641                  */
1642                 output(o, 1, _("Refusing to lose untracked file at "
1643                                "%s, even though it's in the way."),
1644                        collide_path);
1645                 update_path = alt_path = unique_path(o, collide_path, "merged");
1646         } else {
1647                 /*
1648                  * FIXME: It's possible that the two files are identical
1649                  * and that the current working copy happens to match, in
1650                  * which case we are unnecessarily touching the working
1651                  * tree file.  It's not a likely enough scenario that I
1652                  * want to code up the checks for it and a better fix is
1653                  * available if we restructure how unpack_trees() and
1654                  * merge-recursive interoperate anyway, so punting for
1655                  * now...
1656                  */
1657                 remove_file(o, 0, collide_path, 0);
1658         }
1659
1660         /* Store things in diff_filespecs for functions that need it */
1661         memset(&a, 0, sizeof(struct diff_filespec));
1662         memset(&b, 0, sizeof(struct diff_filespec));
1663         null.path = a.path = b.path = (char *)collide_path;
1664         oidcpy(&null.oid, &null_oid);
1665         null.mode = 0;
1666         oidcpy(&a.oid, a_oid);
1667         a.mode = a_mode;
1668         a.oid_valid = 1;
1669         oidcpy(&b.oid, b_oid);
1670         b.mode = b_mode;
1671         b.oid_valid = 1;
1672
1673         if (merge_mode_and_contents(o, &null, &a, &b, collide_path,
1674                                     branch1, branch2, o->call_depth * 2, &mfi))
1675                 return -1;
1676         mfi.clean &= !alt_path;
1677         if (update_file(o, mfi.clean, &mfi.oid, mfi.mode, update_path))
1678                 return -1;
1679         if (!mfi.clean && !o->call_depth &&
1680             update_stages(o, collide_path, NULL, &a, &b))
1681                 return -1;
1682         free(alt_path);
1683         /*
1684          * FIXME: If both a & b both started with conflicts (only possible
1685          * if they came from a rename/rename(2to1)), but had IDENTICAL
1686          * contents including those conflicts, then in the next line we claim
1687          * it was clean.  If someone cares about this case, we should have the
1688          * caller notify us if we started with conflicts.
1689          */
1690         return mfi.clean;
1691 }
1692
1693 static int handle_rename_add(struct merge_options *o,
1694                              struct rename_conflict_info *ci)
1695 {
1696         /* a was renamed to c, and a separate c was added. */
1697         struct diff_filespec *a = ci->pair1->one;
1698         struct diff_filespec *c = ci->pair1->two;
1699         char *path = c->path;
1700         char *prev_path_desc;
1701         struct merge_file_info mfi;
1702
1703         int other_stage = (ci->branch1 == o->branch1 ? 3 : 2);
1704
1705         output(o, 1, _("CONFLICT (rename/add): "
1706                "Rename %s->%s in %s.  Added %s in %s"),
1707                a->path, c->path, ci->branch1,
1708                c->path, ci->branch2);
1709
1710         prev_path_desc = xstrfmt("version of %s from %s", path, a->path);
1711         if (merge_mode_and_contents(o, a, c, &ci->ren1_other, prev_path_desc,
1712                                     o->branch1, o->branch2,
1713                                     1 + o->call_depth * 2, &mfi))
1714                 return -1;
1715         free(prev_path_desc);
1716
1717         return handle_file_collision(o,
1718                                      c->path, a->path, NULL,
1719                                      ci->branch1, ci->branch2,
1720                                      &mfi.oid, mfi.mode,
1721                                      &ci->dst_entry1->stages[other_stage].oid,
1722                                      ci->dst_entry1->stages[other_stage].mode);
1723 }
1724
1725 static char *find_path_for_conflict(struct merge_options *o,
1726                                     const char *path,
1727                                     const char *branch1,
1728                                     const char *branch2)
1729 {
1730         char *new_path = NULL;
1731         if (dir_in_way(o->repo->index, path, !o->call_depth, 0)) {
1732                 new_path = unique_path(o, path, branch1);
1733                 output(o, 1, _("%s is a directory in %s adding "
1734                                "as %s instead"),
1735                        path, branch2, new_path);
1736         } else if (would_lose_untracked(o, path)) {
1737                 new_path = unique_path(o, path, branch1);
1738                 output(o, 1, _("Refusing to lose untracked file"
1739                                " at %s; adding as %s instead"),
1740                        path, new_path);
1741         }
1742
1743         return new_path;
1744 }
1745
1746 static int handle_rename_rename_1to2(struct merge_options *o,
1747                                      struct rename_conflict_info *ci)
1748 {
1749         /* One file was renamed in both branches, but to different names. */
1750         struct merge_file_info mfi;
1751         struct diff_filespec other;
1752         struct diff_filespec *add;
1753         struct diff_filespec *one = ci->pair1->one;
1754         struct diff_filespec *a = ci->pair1->two;
1755         struct diff_filespec *b = ci->pair2->two;
1756         char *path_desc;
1757
1758         output(o, 1, _("CONFLICT (rename/rename): "
1759                "Rename \"%s\"->\"%s\" in branch \"%s\" "
1760                "rename \"%s\"->\"%s\" in \"%s\"%s"),
1761                one->path, a->path, ci->branch1,
1762                one->path, b->path, ci->branch2,
1763                o->call_depth ? _(" (left unresolved)") : "");
1764
1765         path_desc = xstrfmt("%s and %s, both renamed from %s",
1766                             a->path, b->path, one->path);
1767         if (merge_mode_and_contents(o, one, a, b, path_desc,
1768                                     ci->branch1, ci->branch2,
1769                                     o->call_depth * 2, &mfi))
1770                 return -1;
1771         free(path_desc);
1772
1773         if (o->call_depth) {
1774                 /*
1775                  * FIXME: For rename/add-source conflicts (if we could detect
1776                  * such), this is wrong.  We should instead find a unique
1777                  * pathname and then either rename the add-source file to that
1778                  * unique path, or use that unique path instead of src here.
1779                  */
1780                 if (update_file(o, 0, &mfi.oid, mfi.mode, one->path))
1781                         return -1;
1782
1783                 /*
1784                  * Above, we put the merged content at the merge-base's
1785                  * path.  Now we usually need to delete both a->path and
1786                  * b->path.  However, the rename on each side of the merge
1787                  * could also be involved in a rename/add conflict.  In
1788                  * such cases, we should keep the added file around,
1789                  * resolving the conflict at that path in its favor.
1790                  */
1791                 add = filespec_from_entry(&other, ci->dst_entry1, 2 ^ 1);
1792                 if (add) {
1793                         if (update_file(o, 0, &add->oid, add->mode, a->path))
1794                                 return -1;
1795                 }
1796                 else
1797                         remove_file_from_index(o->repo->index, a->path);
1798                 add = filespec_from_entry(&other, ci->dst_entry2, 3 ^ 1);
1799                 if (add) {
1800                         if (update_file(o, 0, &add->oid, add->mode, b->path))
1801                                 return -1;
1802                 }
1803                 else
1804                         remove_file_from_index(o->repo->index, b->path);
1805         } else {
1806                 /*
1807                  * For each destination path, we need to see if there is a
1808                  * rename/add collision.  If not, we can write the file out
1809                  * to the specified location.
1810                  */
1811                 add = filespec_from_entry(&other, ci->dst_entry1, 2 ^ 1);
1812                 if (add) {
1813                         if (handle_file_collision(o, a->path,
1814                                                   NULL, NULL,
1815                                                   ci->branch1, ci->branch2,
1816                                                   &mfi.oid, mfi.mode,
1817                                                   &add->oid, add->mode) < 0)
1818                                 return -1;
1819                 } else {
1820                         char *new_path = find_path_for_conflict(o, a->path,
1821                                                                 ci->branch1,
1822                                                                 ci->branch2);
1823                         if (update_file(o, 0, &mfi.oid, mfi.mode, new_path ? new_path : a->path))
1824                                 return -1;
1825                         free(new_path);
1826                         if (update_stages(o, a->path, NULL, a, NULL))
1827                                 return -1;
1828                 }
1829
1830                 add = filespec_from_entry(&other, ci->dst_entry2, 3 ^ 1);
1831                 if (add) {
1832                         if (handle_file_collision(o, b->path,
1833                                                   NULL, NULL,
1834                                                   ci->branch1, ci->branch2,
1835                                                   &add->oid, add->mode,
1836                                                   &mfi.oid, mfi.mode) < 0)
1837                                 return -1;
1838                 } else {
1839                         char *new_path = find_path_for_conflict(o, b->path,
1840                                                                 ci->branch2,
1841                                                                 ci->branch1);
1842                         if (update_file(o, 0, &mfi.oid, mfi.mode, new_path ? new_path : b->path))
1843                                 return -1;
1844                         free(new_path);
1845                         if (update_stages(o, b->path, NULL, NULL, b))
1846                                 return -1;
1847                 }
1848         }
1849
1850         return 0;
1851 }
1852
1853 static int handle_rename_rename_2to1(struct merge_options *o,
1854                                      struct rename_conflict_info *ci)
1855 {
1856         /* Two files, a & b, were renamed to the same thing, c. */
1857         struct diff_filespec *a = ci->pair1->one;
1858         struct diff_filespec *b = ci->pair2->one;
1859         struct diff_filespec *c1 = ci->pair1->two;
1860         struct diff_filespec *c2 = ci->pair2->two;
1861         char *path = c1->path; /* == c2->path */
1862         char *path_side_1_desc;
1863         char *path_side_2_desc;
1864         struct merge_file_info mfi_c1;
1865         struct merge_file_info mfi_c2;
1866
1867         output(o, 1, _("CONFLICT (rename/rename): "
1868                "Rename %s->%s in %s. "
1869                "Rename %s->%s in %s"),
1870                a->path, c1->path, ci->branch1,
1871                b->path, c2->path, ci->branch2);
1872
1873         path_side_1_desc = xstrfmt("version of %s from %s", path, a->path);
1874         path_side_2_desc = xstrfmt("version of %s from %s", path, b->path);
1875         if (merge_mode_and_contents(o, a, c1, &ci->ren1_other, path_side_1_desc,
1876                                     o->branch1, o->branch2,
1877                                     1 + o->call_depth * 2, &mfi_c1) ||
1878             merge_mode_and_contents(o, b, &ci->ren2_other, c2, path_side_2_desc,
1879                                     o->branch1, o->branch2,
1880                                     1 + o->call_depth * 2, &mfi_c2))
1881                 return -1;
1882         free(path_side_1_desc);
1883         free(path_side_2_desc);
1884
1885         return handle_file_collision(o, path, a->path, b->path,
1886                                      ci->branch1, ci->branch2,
1887                                      &mfi_c1.oid, mfi_c1.mode,
1888                                      &mfi_c2.oid, mfi_c2.mode);
1889 }
1890
1891 /*
1892  * Get the diff_filepairs changed between o_tree and tree.
1893  */
1894 static struct diff_queue_struct *get_diffpairs(struct merge_options *o,
1895                                                struct tree *o_tree,
1896                                                struct tree *tree)
1897 {
1898         struct diff_queue_struct *ret;
1899         struct diff_options opts;
1900
1901         repo_diff_setup(o->repo, &opts);
1902         opts.flags.recursive = 1;
1903         opts.flags.rename_empty = 0;
1904         opts.detect_rename = merge_detect_rename(o);
1905         /*
1906          * We do not have logic to handle the detection of copies.  In
1907          * fact, it may not even make sense to add such logic: would we
1908          * really want a change to a base file to be propagated through
1909          * multiple other files by a merge?
1910          */
1911         if (opts.detect_rename > DIFF_DETECT_RENAME)
1912                 opts.detect_rename = DIFF_DETECT_RENAME;
1913         opts.rename_limit = o->merge_rename_limit >= 0 ? o->merge_rename_limit :
1914                             o->diff_rename_limit >= 0 ? o->diff_rename_limit :
1915                             1000;
1916         opts.rename_score = o->rename_score;
1917         opts.show_rename_progress = o->show_rename_progress;
1918         opts.output_format = DIFF_FORMAT_NO_OUTPUT;
1919         diff_setup_done(&opts);
1920         diff_tree_oid(&o_tree->object.oid, &tree->object.oid, "", &opts);
1921         diffcore_std(&opts);
1922         if (opts.needed_rename_limit > o->needed_rename_limit)
1923                 o->needed_rename_limit = opts.needed_rename_limit;
1924
1925         ret = xmalloc(sizeof(*ret));
1926         *ret = diff_queued_diff;
1927
1928         opts.output_format = DIFF_FORMAT_NO_OUTPUT;
1929         diff_queued_diff.nr = 0;
1930         diff_queued_diff.queue = NULL;
1931         diff_flush(&opts);
1932         return ret;
1933 }
1934
1935 static int tree_has_path(struct tree *tree, const char *path)
1936 {
1937         struct object_id hashy;
1938         unsigned int mode_o;
1939
1940         return !get_tree_entry(&tree->object.oid, path,
1941                                &hashy, &mode_o);
1942 }
1943
1944 /*
1945  * Return a new string that replaces the beginning portion (which matches
1946  * entry->dir), with entry->new_dir.  In perl-speak:
1947  *   new_path_name = (old_path =~ s/entry->dir/entry->new_dir/);
1948  * NOTE:
1949  *   Caller must ensure that old_path starts with entry->dir + '/'.
1950  */
1951 static char *apply_dir_rename(struct dir_rename_entry *entry,
1952                               const char *old_path)
1953 {
1954         struct strbuf new_path = STRBUF_INIT;
1955         int oldlen, newlen;
1956
1957         if (entry->non_unique_new_dir)
1958                 return NULL;
1959
1960         oldlen = strlen(entry->dir);
1961         newlen = entry->new_dir.len + (strlen(old_path) - oldlen) + 1;
1962         strbuf_grow(&new_path, newlen);
1963         strbuf_addbuf(&new_path, &entry->new_dir);
1964         strbuf_addstr(&new_path, &old_path[oldlen]);
1965
1966         return strbuf_detach(&new_path, NULL);
1967 }
1968
1969 static void get_renamed_dir_portion(const char *old_path, const char *new_path,
1970                                     char **old_dir, char **new_dir)
1971 {
1972         char *end_of_old, *end_of_new;
1973         int old_len, new_len;
1974
1975         *old_dir = NULL;
1976         *new_dir = NULL;
1977
1978         /*
1979          * For
1980          *    "a/b/c/d/e/foo.c" -> "a/b/some/thing/else/e/foo.c"
1981          * the "e/foo.c" part is the same, we just want to know that
1982          *    "a/b/c/d" was renamed to "a/b/some/thing/else"
1983          * so, for this example, this function returns "a/b/c/d" in
1984          * *old_dir and "a/b/some/thing/else" in *new_dir.
1985          *
1986          * Also, if the basename of the file changed, we don't care.  We
1987          * want to know which portion of the directory, if any, changed.
1988          */
1989         end_of_old = strrchr(old_path, '/');
1990         end_of_new = strrchr(new_path, '/');
1991
1992         if (end_of_old == NULL || end_of_new == NULL)
1993                 return;
1994         while (*--end_of_new == *--end_of_old &&
1995                end_of_old != old_path &&
1996                end_of_new != new_path)
1997                 ; /* Do nothing; all in the while loop */
1998         /*
1999          * We've found the first non-matching character in the directory
2000          * paths.  That means the current directory we were comparing
2001          * represents the rename.  Move end_of_old and end_of_new back
2002          * to the full directory name.
2003          */
2004         if (*end_of_old == '/')
2005                 end_of_old++;
2006         if (*end_of_old != '/')
2007                 end_of_new++;
2008         end_of_old = strchr(end_of_old, '/');
2009         end_of_new = strchr(end_of_new, '/');
2010
2011         /*
2012          * It may have been the case that old_path and new_path were the same
2013          * directory all along.  Don't claim a rename if they're the same.
2014          */
2015         old_len = end_of_old - old_path;
2016         new_len = end_of_new - new_path;
2017
2018         if (old_len != new_len || strncmp(old_path, new_path, old_len)) {
2019                 *old_dir = xstrndup(old_path, old_len);
2020                 *new_dir = xstrndup(new_path, new_len);
2021         }
2022 }
2023
2024 static void remove_hashmap_entries(struct hashmap *dir_renames,
2025                                    struct string_list *items_to_remove)
2026 {
2027         int i;
2028         struct dir_rename_entry *entry;
2029
2030         for (i = 0; i < items_to_remove->nr; i++) {
2031                 entry = items_to_remove->items[i].util;
2032                 hashmap_remove(dir_renames, entry, NULL);
2033         }
2034         string_list_clear(items_to_remove, 0);
2035 }
2036
2037 /*
2038  * See if there is a directory rename for path, and if there are any file
2039  * level conflicts for the renamed location.  If there is a rename and
2040  * there are no conflicts, return the new name.  Otherwise, return NULL.
2041  */
2042 static char *handle_path_level_conflicts(struct merge_options *o,
2043                                          const char *path,
2044                                          struct dir_rename_entry *entry,
2045                                          struct hashmap *collisions,
2046                                          struct tree *tree)
2047 {
2048         char *new_path = NULL;
2049         struct collision_entry *collision_ent;
2050         int clean = 1;
2051         struct strbuf collision_paths = STRBUF_INIT;
2052
2053         /*
2054          * entry has the mapping of old directory name to new directory name
2055          * that we want to apply to path.
2056          */
2057         new_path = apply_dir_rename(entry, path);
2058
2059         if (!new_path) {
2060                 /* This should only happen when entry->non_unique_new_dir set */
2061                 if (!entry->non_unique_new_dir)
2062                         BUG("entry->non_unqiue_dir not set and !new_path");
2063                 output(o, 1, _("CONFLICT (directory rename split): "
2064                                "Unclear where to place %s because directory "
2065                                "%s was renamed to multiple other directories, "
2066                                "with no destination getting a majority of the "
2067                                "files."),
2068                        path, entry->dir);
2069                 clean = 0;
2070                 return NULL;
2071         }
2072
2073         /*
2074          * The caller needs to have ensured that it has pre-populated
2075          * collisions with all paths that map to new_path.  Do a quick check
2076          * to ensure that's the case.
2077          */
2078         collision_ent = collision_find_entry(collisions, new_path);
2079         if (collision_ent == NULL)
2080                 BUG("collision_ent is NULL");
2081
2082         /*
2083          * Check for one-sided add/add/.../add conflicts, i.e.
2084          * where implicit renames from the other side doing
2085          * directory rename(s) can affect this side of history
2086          * to put multiple paths into the same location.  Warn
2087          * and bail on directory renames for such paths.
2088          */
2089         if (collision_ent->reported_already) {
2090                 clean = 0;
2091         } else if (tree_has_path(tree, new_path)) {
2092                 collision_ent->reported_already = 1;
2093                 strbuf_add_separated_string_list(&collision_paths, ", ",
2094                                                  &collision_ent->source_files);
2095                 output(o, 1, _("CONFLICT (implicit dir rename): Existing "
2096                                "file/dir at %s in the way of implicit "
2097                                "directory rename(s) putting the following "
2098                                "path(s) there: %s."),
2099                        new_path, collision_paths.buf);
2100                 clean = 0;
2101         } else if (collision_ent->source_files.nr > 1) {
2102                 collision_ent->reported_already = 1;
2103                 strbuf_add_separated_string_list(&collision_paths, ", ",
2104                                                  &collision_ent->source_files);
2105                 output(o, 1, _("CONFLICT (implicit dir rename): Cannot map "
2106                                "more than one path to %s; implicit directory "
2107                                "renames tried to put these paths there: %s"),
2108                        new_path, collision_paths.buf);
2109                 clean = 0;
2110         }
2111
2112         /* Free memory we no longer need */
2113         strbuf_release(&collision_paths);
2114         if (!clean && new_path) {
2115                 free(new_path);
2116                 return NULL;
2117         }
2118
2119         return new_path;
2120 }
2121
2122 /*
2123  * There are a couple things we want to do at the directory level:
2124  *   1. Check for both sides renaming to the same thing, in order to avoid
2125  *      implicit renaming of files that should be left in place.  (See
2126  *      testcase 6b in t6043 for details.)
2127  *   2. Prune directory renames if there are still files left in the
2128  *      the original directory.  These represent a partial directory rename,
2129  *      i.e. a rename where only some of the files within the directory
2130  *      were renamed elsewhere.  (Technically, this could be done earlier
2131  *      in get_directory_renames(), except that would prevent us from
2132  *      doing the previous check and thus failing testcase 6b.)
2133  *   3. Check for rename/rename(1to2) conflicts (at the directory level).
2134  *      In the future, we could potentially record this info as well and
2135  *      omit reporting rename/rename(1to2) conflicts for each path within
2136  *      the affected directories, thus cleaning up the merge output.
2137  *   NOTE: We do NOT check for rename/rename(2to1) conflicts at the
2138  *         directory level, because merging directories is fine.  If it
2139  *         causes conflicts for files within those merged directories, then
2140  *         that should be detected at the individual path level.
2141  */
2142 static void handle_directory_level_conflicts(struct merge_options *o,
2143                                              struct hashmap *dir_re_head,
2144                                              struct tree *head,
2145                                              struct hashmap *dir_re_merge,
2146                                              struct tree *merge)
2147 {
2148         struct hashmap_iter iter;
2149         struct dir_rename_entry *head_ent;
2150         struct dir_rename_entry *merge_ent;
2151
2152         struct string_list remove_from_head = STRING_LIST_INIT_NODUP;
2153         struct string_list remove_from_merge = STRING_LIST_INIT_NODUP;
2154
2155         hashmap_iter_init(dir_re_head, &iter);
2156         while ((head_ent = hashmap_iter_next(&iter))) {
2157                 merge_ent = dir_rename_find_entry(dir_re_merge, head_ent->dir);
2158                 if (merge_ent &&
2159                     !head_ent->non_unique_new_dir &&
2160                     !merge_ent->non_unique_new_dir &&
2161                     !strbuf_cmp(&head_ent->new_dir, &merge_ent->new_dir)) {
2162                         /* 1. Renamed identically; remove it from both sides */
2163                         string_list_append(&remove_from_head,
2164                                            head_ent->dir)->util = head_ent;
2165                         strbuf_release(&head_ent->new_dir);
2166                         string_list_append(&remove_from_merge,
2167                                            merge_ent->dir)->util = merge_ent;
2168                         strbuf_release(&merge_ent->new_dir);
2169                 } else if (tree_has_path(head, head_ent->dir)) {
2170                         /* 2. This wasn't a directory rename after all */
2171                         string_list_append(&remove_from_head,
2172                                            head_ent->dir)->util = head_ent;
2173                         strbuf_release(&head_ent->new_dir);
2174                 }
2175         }
2176
2177         remove_hashmap_entries(dir_re_head, &remove_from_head);
2178         remove_hashmap_entries(dir_re_merge, &remove_from_merge);
2179
2180         hashmap_iter_init(dir_re_merge, &iter);
2181         while ((merge_ent = hashmap_iter_next(&iter))) {
2182                 head_ent = dir_rename_find_entry(dir_re_head, merge_ent->dir);
2183                 if (tree_has_path(merge, merge_ent->dir)) {
2184                         /* 2. This wasn't a directory rename after all */
2185                         string_list_append(&remove_from_merge,
2186                                            merge_ent->dir)->util = merge_ent;
2187                 } else if (head_ent &&
2188                            !head_ent->non_unique_new_dir &&
2189                            !merge_ent->non_unique_new_dir) {
2190                         /* 3. rename/rename(1to2) */
2191                         /*
2192                          * We can assume it's not rename/rename(1to1) because
2193                          * that was case (1), already checked above.  So we
2194                          * know that head_ent->new_dir and merge_ent->new_dir
2195                          * are different strings.
2196                          */
2197                         output(o, 1, _("CONFLICT (rename/rename): "
2198                                        "Rename directory %s->%s in %s. "
2199                                        "Rename directory %s->%s in %s"),
2200                                head_ent->dir, head_ent->new_dir.buf, o->branch1,
2201                                head_ent->dir, merge_ent->new_dir.buf, o->branch2);
2202                         string_list_append(&remove_from_head,
2203                                            head_ent->dir)->util = head_ent;
2204                         strbuf_release(&head_ent->new_dir);
2205                         string_list_append(&remove_from_merge,
2206                                            merge_ent->dir)->util = merge_ent;
2207                         strbuf_release(&merge_ent->new_dir);
2208                 }
2209         }
2210
2211         remove_hashmap_entries(dir_re_head, &remove_from_head);
2212         remove_hashmap_entries(dir_re_merge, &remove_from_merge);
2213 }
2214
2215 static struct hashmap *get_directory_renames(struct diff_queue_struct *pairs)
2216 {
2217         struct hashmap *dir_renames;
2218         struct hashmap_iter iter;
2219         struct dir_rename_entry *entry;
2220         int i;
2221
2222         /*
2223          * Typically, we think of a directory rename as all files from a
2224          * certain directory being moved to a target directory.  However,
2225          * what if someone first moved two files from the original
2226          * directory in one commit, and then renamed the directory
2227          * somewhere else in a later commit?  At merge time, we just know
2228          * that files from the original directory went to two different
2229          * places, and that the bulk of them ended up in the same place.
2230          * We want each directory rename to represent where the bulk of the
2231          * files from that directory end up; this function exists to find
2232          * where the bulk of the files went.
2233          *
2234          * The first loop below simply iterates through the list of file
2235          * renames, finding out how often each directory rename pair
2236          * possibility occurs.
2237          */
2238         dir_renames = xmalloc(sizeof(*dir_renames));
2239         dir_rename_init(dir_renames);
2240         for (i = 0; i < pairs->nr; ++i) {
2241                 struct string_list_item *item;
2242                 int *count;
2243                 struct diff_filepair *pair = pairs->queue[i];
2244                 char *old_dir, *new_dir;
2245
2246                 /* File not part of directory rename if it wasn't renamed */
2247                 if (pair->status != 'R')
2248                         continue;
2249
2250                 get_renamed_dir_portion(pair->one->path, pair->two->path,
2251                                         &old_dir,        &new_dir);
2252                 if (!old_dir)
2253                         /* Directory didn't change at all; ignore this one. */
2254                         continue;
2255
2256                 entry = dir_rename_find_entry(dir_renames, old_dir);
2257                 if (!entry) {
2258                         entry = xmalloc(sizeof(*entry));
2259                         dir_rename_entry_init(entry, old_dir);
2260                         hashmap_put(dir_renames, entry);
2261                 } else {
2262                         free(old_dir);
2263                 }
2264                 item = string_list_lookup(&entry->possible_new_dirs, new_dir);
2265                 if (!item) {
2266                         item = string_list_insert(&entry->possible_new_dirs,
2267                                                   new_dir);
2268                         item->util = xcalloc(1, sizeof(int));
2269                 } else {
2270                         free(new_dir);
2271                 }
2272                 count = item->util;
2273                 *count += 1;
2274         }
2275
2276         /*
2277          * For each directory with files moved out of it, we find out which
2278          * target directory received the most files so we can declare it to
2279          * be the "winning" target location for the directory rename.  This
2280          * winner gets recorded in new_dir.  If there is no winner
2281          * (multiple target directories received the same number of files),
2282          * we set non_unique_new_dir.  Once we've determined the winner (or
2283          * that there is no winner), we no longer need possible_new_dirs.
2284          */
2285         hashmap_iter_init(dir_renames, &iter);
2286         while ((entry = hashmap_iter_next(&iter))) {
2287                 int max = 0;
2288                 int bad_max = 0;
2289                 char *best = NULL;
2290
2291                 for (i = 0; i < entry->possible_new_dirs.nr; i++) {
2292                         int *count = entry->possible_new_dirs.items[i].util;
2293
2294                         if (*count == max)
2295                                 bad_max = max;
2296                         else if (*count > max) {
2297                                 max = *count;
2298                                 best = entry->possible_new_dirs.items[i].string;
2299                         }
2300                 }
2301                 if (bad_max == max)
2302                         entry->non_unique_new_dir = 1;
2303                 else {
2304                         assert(entry->new_dir.len == 0);
2305                         strbuf_addstr(&entry->new_dir, best);
2306                 }
2307                 /*
2308                  * The relevant directory sub-portion of the original full
2309                  * filepaths were xstrndup'ed before inserting into
2310                  * possible_new_dirs, and instead of manually iterating the
2311                  * list and free'ing each, just lie and tell
2312                  * possible_new_dirs that it did the strdup'ing so that it
2313                  * will free them for us.
2314                  */
2315                 entry->possible_new_dirs.strdup_strings = 1;
2316                 string_list_clear(&entry->possible_new_dirs, 1);
2317         }
2318
2319         return dir_renames;
2320 }
2321
2322 static struct dir_rename_entry *check_dir_renamed(const char *path,
2323                                                   struct hashmap *dir_renames)
2324 {
2325         char *temp = xstrdup(path);
2326         char *end;
2327         struct dir_rename_entry *entry = NULL;
2328
2329         while ((end = strrchr(temp, '/'))) {
2330                 *end = '\0';
2331                 entry = dir_rename_find_entry(dir_renames, temp);
2332                 if (entry)
2333                         break;
2334         }
2335         free(temp);
2336         return entry;
2337 }
2338
2339 static void compute_collisions(struct hashmap *collisions,
2340                                struct hashmap *dir_renames,
2341                                struct diff_queue_struct *pairs)
2342 {
2343         int i;
2344
2345         /*
2346          * Multiple files can be mapped to the same path due to directory
2347          * renames done by the other side of history.  Since that other
2348          * side of history could have merged multiple directories into one,
2349          * if our side of history added the same file basename to each of
2350          * those directories, then all N of them would get implicitly
2351          * renamed by the directory rename detection into the same path,
2352          * and we'd get an add/add/.../add conflict, and all those adds
2353          * from *this* side of history.  This is not representable in the
2354          * index, and users aren't going to easily be able to make sense of
2355          * it.  So we need to provide a good warning about what's
2356          * happening, and fall back to no-directory-rename detection
2357          * behavior for those paths.
2358          *
2359          * See testcases 9e and all of section 5 from t6043 for examples.
2360          */
2361         collision_init(collisions);
2362
2363         for (i = 0; i < pairs->nr; ++i) {
2364                 struct dir_rename_entry *dir_rename_ent;
2365                 struct collision_entry *collision_ent;
2366                 char *new_path;
2367                 struct diff_filepair *pair = pairs->queue[i];
2368
2369                 if (pair->status != 'A' && pair->status != 'R')
2370                         continue;
2371                 dir_rename_ent = check_dir_renamed(pair->two->path,
2372                                                    dir_renames);
2373                 if (!dir_rename_ent)
2374                         continue;
2375
2376                 new_path = apply_dir_rename(dir_rename_ent, pair->two->path);
2377                 if (!new_path)
2378                         /*
2379                          * dir_rename_ent->non_unique_new_path is true, which
2380                          * means there is no directory rename for us to use,
2381                          * which means it won't cause us any additional
2382                          * collisions.
2383                          */
2384                         continue;
2385                 collision_ent = collision_find_entry(collisions, new_path);
2386                 if (!collision_ent) {
2387                         collision_ent = xcalloc(1,
2388                                                 sizeof(struct collision_entry));
2389                         hashmap_entry_init(collision_ent, strhash(new_path));
2390                         hashmap_put(collisions, collision_ent);
2391                         collision_ent->target_file = new_path;
2392                 } else {
2393                         free(new_path);
2394                 }
2395                 string_list_insert(&collision_ent->source_files,
2396                                    pair->two->path);
2397         }
2398 }
2399
2400 static char *check_for_directory_rename(struct merge_options *o,
2401                                         const char *path,
2402                                         struct tree *tree,
2403                                         struct hashmap *dir_renames,
2404                                         struct hashmap *dir_rename_exclusions,
2405                                         struct hashmap *collisions,
2406                                         int *clean_merge)
2407 {
2408         char *new_path = NULL;
2409         struct dir_rename_entry *entry = check_dir_renamed(path, dir_renames);
2410         struct dir_rename_entry *oentry = NULL;
2411
2412         if (!entry)
2413                 return new_path;
2414
2415         /*
2416          * This next part is a little weird.  We do not want to do an
2417          * implicit rename into a directory we renamed on our side, because
2418          * that will result in a spurious rename/rename(1to2) conflict.  An
2419          * example:
2420          *   Base commit: dumbdir/afile, otherdir/bfile
2421          *   Side 1:      smrtdir/afile, otherdir/bfile
2422          *   Side 2:      dumbdir/afile, dumbdir/bfile
2423          * Here, while working on Side 1, we could notice that otherdir was
2424          * renamed/merged to dumbdir, and change the diff_filepair for
2425          * otherdir/bfile into a rename into dumbdir/bfile.  However, Side
2426          * 2 will notice the rename from dumbdir to smrtdir, and do the
2427          * transitive rename to move it from dumbdir/bfile to
2428          * smrtdir/bfile.  That gives us bfile in dumbdir vs being in
2429          * smrtdir, a rename/rename(1to2) conflict.  We really just want
2430          * the file to end up in smrtdir.  And the way to achieve that is
2431          * to not let Side1 do the rename to dumbdir, since we know that is
2432          * the source of one of our directory renames.
2433          *
2434          * That's why oentry and dir_rename_exclusions is here.
2435          *
2436          * As it turns out, this also prevents N-way transient rename
2437          * confusion; See testcases 9c and 9d of t6043.
2438          */
2439         oentry = dir_rename_find_entry(dir_rename_exclusions, entry->new_dir.buf);
2440         if (oentry) {
2441                 output(o, 1, _("WARNING: Avoiding applying %s -> %s rename "
2442                                "to %s, because %s itself was renamed."),
2443                        entry->dir, entry->new_dir.buf, path, entry->new_dir.buf);
2444         } else {
2445                 new_path = handle_path_level_conflicts(o, path, entry,
2446                                                        collisions, tree);
2447                 *clean_merge &= (new_path != NULL);
2448         }
2449
2450         return new_path;
2451 }
2452
2453 static void apply_directory_rename_modifications(struct merge_options *o,
2454                                                  struct diff_filepair *pair,
2455                                                  char *new_path,
2456                                                  struct rename *re,
2457                                                  struct tree *tree,
2458                                                  struct tree *o_tree,
2459                                                  struct tree *a_tree,
2460                                                  struct tree *b_tree,
2461                                                  struct string_list *entries)
2462 {
2463         struct string_list_item *item;
2464         int stage = (tree == a_tree ? 2 : 3);
2465         int update_wd;
2466
2467         /*
2468          * In all cases where we can do directory rename detection,
2469          * unpack_trees() will have read pair->two->path into the
2470          * index and the working copy.  We need to remove it so that
2471          * we can instead place it at new_path.  It is guaranteed to
2472          * not be untracked (unpack_trees() would have errored out
2473          * saying the file would have been overwritten), but it might
2474          * be dirty, though.
2475          */
2476         update_wd = !was_dirty(o, pair->two->path);
2477         if (!update_wd)
2478                 output(o, 1, _("Refusing to lose dirty file at %s"),
2479                        pair->two->path);
2480         remove_file(o, 1, pair->two->path, !update_wd);
2481
2482         /* Find or create a new re->dst_entry */
2483         item = string_list_lookup(entries, new_path);
2484         if (item) {
2485                 /*
2486                  * Since we're renaming on this side of history, and it's
2487                  * due to a directory rename on the other side of history
2488                  * (which we only allow when the directory in question no
2489                  * longer exists on the other side of history), the
2490                  * original entry for re->dst_entry is no longer
2491                  * necessary...
2492                  */
2493                 re->dst_entry->processed = 1;
2494
2495                 /*
2496                  * ...because we'll be using this new one.
2497                  */
2498                 re->dst_entry = item->util;
2499         } else {
2500                 /*
2501                  * re->dst_entry is for the before-dir-rename path, and we
2502                  * need it to hold information for the after-dir-rename
2503                  * path.  Before creating a new entry, we need to mark the
2504                  * old one as unnecessary (...unless it is shared by
2505                  * src_entry, i.e. this didn't use to be a rename, in which
2506                  * case we can just allow the normal processing to happen
2507                  * for it).
2508                  */
2509                 if (pair->status == 'R')
2510                         re->dst_entry->processed = 1;
2511
2512                 re->dst_entry = insert_stage_data(new_path,
2513                                                   o_tree, a_tree, b_tree,
2514                                                   entries);
2515                 item = string_list_insert(entries, new_path);
2516                 item->util = re->dst_entry;
2517         }
2518
2519         /*
2520          * Update the stage_data with the information about the path we are
2521          * moving into place.  That slot will be empty and available for us
2522          * to write to because of the collision checks in
2523          * handle_path_level_conflicts().  In other words,
2524          * re->dst_entry->stages[stage].oid will be the null_oid, so it's
2525          * open for us to write to.
2526          *
2527          * It may be tempting to actually update the index at this point as
2528          * well, using update_stages_for_stage_data(), but as per the big
2529          * "NOTE" in update_stages(), doing so will modify the current
2530          * in-memory index which will break calls to would_lose_untracked()
2531          * that we need to make.  Instead, we need to just make sure that
2532          * the various handle_rename_*() functions update the index
2533          * explicitly rather than relying on unpack_trees() to have done it.
2534          */
2535         get_tree_entry(&tree->object.oid,
2536                        pair->two->path,
2537                        &re->dst_entry->stages[stage].oid,
2538                        &re->dst_entry->stages[stage].mode);
2539
2540         /* Update pair status */
2541         if (pair->status == 'A') {
2542                 /*
2543                  * Recording rename information for this add makes it look
2544                  * like a rename/delete conflict.  Make sure we can
2545                  * correctly handle this as an add that was moved to a new
2546                  * directory instead of reporting a rename/delete conflict.
2547                  */
2548                 re->add_turned_into_rename = 1;
2549         }
2550         /*
2551          * We don't actually look at pair->status again, but it seems
2552          * pedagogically correct to adjust it.
2553          */
2554         pair->status = 'R';
2555
2556         /*
2557          * Finally, record the new location.
2558          */
2559         pair->two->path = new_path;
2560 }
2561
2562 /*
2563  * Get information of all renames which occurred in 'pairs', making use of
2564  * any implicit directory renames inferred from the other side of history.
2565  * We need the three trees in the merge ('o_tree', 'a_tree' and 'b_tree')
2566  * to be able to associate the correct cache entries with the rename
2567  * information; tree is always equal to either a_tree or b_tree.
2568  */
2569 static struct string_list *get_renames(struct merge_options *o,
2570                                        struct diff_queue_struct *pairs,
2571                                        struct hashmap *dir_renames,
2572                                        struct hashmap *dir_rename_exclusions,
2573                                        struct tree *tree,
2574                                        struct tree *o_tree,
2575                                        struct tree *a_tree,
2576                                        struct tree *b_tree,
2577                                        struct string_list *entries,
2578                                        int *clean_merge)
2579 {
2580         int i;
2581         struct hashmap collisions;
2582         struct hashmap_iter iter;
2583         struct collision_entry *e;
2584         struct string_list *renames;
2585
2586         compute_collisions(&collisions, dir_renames, pairs);
2587         renames = xcalloc(1, sizeof(struct string_list));
2588
2589         for (i = 0; i < pairs->nr; ++i) {
2590                 struct string_list_item *item;
2591                 struct rename *re;
2592                 struct diff_filepair *pair = pairs->queue[i];
2593                 char *new_path; /* non-NULL only with directory renames */
2594
2595                 if (pair->status != 'A' && pair->status != 'R') {
2596                         diff_free_filepair(pair);
2597                         continue;
2598                 }
2599                 new_path = check_for_directory_rename(o, pair->two->path, tree,
2600                                                       dir_renames,
2601                                                       dir_rename_exclusions,
2602                                                       &collisions,
2603                                                       clean_merge);
2604                 if (pair->status != 'R' && !new_path) {
2605                         diff_free_filepair(pair);
2606                         continue;
2607                 }
2608
2609                 re = xmalloc(sizeof(*re));
2610                 re->processed = 0;
2611                 re->add_turned_into_rename = 0;
2612                 re->pair = pair;
2613                 item = string_list_lookup(entries, re->pair->one->path);
2614                 if (!item)
2615                         re->src_entry = insert_stage_data(re->pair->one->path,
2616                                         o_tree, a_tree, b_tree, entries);
2617                 else
2618                         re->src_entry = item->util;
2619
2620                 item = string_list_lookup(entries, re->pair->two->path);
2621                 if (!item)
2622                         re->dst_entry = insert_stage_data(re->pair->two->path,
2623                                         o_tree, a_tree, b_tree, entries);
2624                 else
2625                         re->dst_entry = item->util;
2626                 item = string_list_insert(renames, pair->one->path);
2627                 item->util = re;
2628                 if (new_path)
2629                         apply_directory_rename_modifications(o, pair, new_path,
2630                                                              re, tree, o_tree,
2631                                                              a_tree, b_tree,
2632                                                              entries);
2633         }
2634
2635         hashmap_iter_init(&collisions, &iter);
2636         while ((e = hashmap_iter_next(&iter))) {
2637                 free(e->target_file);
2638                 string_list_clear(&e->source_files, 0);
2639         }
2640         hashmap_free(&collisions, 1);
2641         return renames;
2642 }
2643
2644 static int process_renames(struct merge_options *o,
2645                            struct string_list *a_renames,
2646                            struct string_list *b_renames)
2647 {
2648         int clean_merge = 1, i, j;
2649         struct string_list a_by_dst = STRING_LIST_INIT_NODUP;
2650         struct string_list b_by_dst = STRING_LIST_INIT_NODUP;
2651         const struct rename *sre;
2652
2653         for (i = 0; i < a_renames->nr; i++) {
2654                 sre = a_renames->items[i].util;
2655                 string_list_insert(&a_by_dst, sre->pair->two->path)->util
2656                         = (void *)sre;
2657         }
2658         for (i = 0; i < b_renames->nr; i++) {
2659                 sre = b_renames->items[i].util;
2660                 string_list_insert(&b_by_dst, sre->pair->two->path)->util
2661                         = (void *)sre;
2662         }
2663
2664         for (i = 0, j = 0; i < a_renames->nr || j < b_renames->nr;) {
2665                 struct string_list *renames1, *renames2Dst;
2666                 struct rename *ren1 = NULL, *ren2 = NULL;
2667                 const char *branch1, *branch2;
2668                 const char *ren1_src, *ren1_dst;
2669                 struct string_list_item *lookup;
2670
2671                 if (i >= a_renames->nr) {
2672                         ren2 = b_renames->items[j++].util;
2673                 } else if (j >= b_renames->nr) {
2674                         ren1 = a_renames->items[i++].util;
2675                 } else {
2676                         int compare = strcmp(a_renames->items[i].string,
2677                                              b_renames->items[j].string);
2678                         if (compare <= 0)
2679                                 ren1 = a_renames->items[i++].util;
2680                         if (compare >= 0)
2681                                 ren2 = b_renames->items[j++].util;
2682                 }
2683
2684                 /* TODO: refactor, so that 1/2 are not needed */
2685                 if (ren1) {
2686                         renames1 = a_renames;
2687                         renames2Dst = &b_by_dst;
2688                         branch1 = o->branch1;
2689                         branch2 = o->branch2;
2690                 } else {
2691                         renames1 = b_renames;
2692                         renames2Dst = &a_by_dst;
2693                         branch1 = o->branch2;
2694                         branch2 = o->branch1;
2695                         SWAP(ren2, ren1);
2696                 }
2697
2698                 if (ren1->processed)
2699                         continue;
2700                 ren1->processed = 1;
2701                 ren1->dst_entry->processed = 1;
2702                 /* BUG: We should only mark src_entry as processed if we
2703                  * are not dealing with a rename + add-source case.
2704                  */
2705                 ren1->src_entry->processed = 1;
2706
2707                 ren1_src = ren1->pair->one->path;
2708                 ren1_dst = ren1->pair->two->path;
2709
2710                 if (ren2) {
2711                         /* One file renamed on both sides */
2712                         const char *ren2_src = ren2->pair->one->path;
2713                         const char *ren2_dst = ren2->pair->two->path;
2714                         enum rename_type rename_type;
2715                         if (strcmp(ren1_src, ren2_src) != 0)
2716                                 BUG("ren1_src != ren2_src");
2717                         ren2->dst_entry->processed = 1;
2718                         ren2->processed = 1;
2719                         if (strcmp(ren1_dst, ren2_dst) != 0) {
2720                                 rename_type = RENAME_ONE_FILE_TO_TWO;
2721                                 clean_merge = 0;
2722                         } else {
2723                                 rename_type = RENAME_ONE_FILE_TO_ONE;
2724                                 /* BUG: We should only remove ren1_src in
2725                                  * the base stage (think of rename +
2726                                  * add-source cases).
2727                                  */
2728                                 remove_file(o, 1, ren1_src, 1);
2729                                 update_entry(ren1->dst_entry,
2730                                              ren1->pair->one,
2731                                              ren1->pair->two,
2732                                              ren2->pair->two);
2733                         }
2734                         setup_rename_conflict_info(rename_type,
2735                                                    ren1->pair,
2736                                                    ren2->pair,
2737                                                    branch1,
2738                                                    branch2,
2739                                                    ren1->dst_entry,
2740                                                    ren2->dst_entry,
2741                                                    o,
2742                                                    NULL,
2743                                                    NULL);
2744                 } else if ((lookup = string_list_lookup(renames2Dst, ren1_dst))) {
2745                         /* Two different files renamed to the same thing */
2746                         char *ren2_dst;
2747                         ren2 = lookup->util;
2748                         ren2_dst = ren2->pair->two->path;
2749                         if (strcmp(ren1_dst, ren2_dst) != 0)
2750                                 BUG("ren1_dst != ren2_dst");
2751
2752                         clean_merge = 0;
2753                         ren2->processed = 1;
2754                         /*
2755                          * BUG: We should only mark src_entry as processed
2756                          * if we are not dealing with a rename + add-source
2757                          * case.
2758                          */
2759                         ren2->src_entry->processed = 1;
2760
2761                         setup_rename_conflict_info(RENAME_TWO_FILES_TO_ONE,
2762                                                    ren1->pair,
2763                                                    ren2->pair,
2764                                                    branch1,
2765                                                    branch2,
2766                                                    ren1->dst_entry,
2767                                                    ren2->dst_entry,
2768                                                    o,
2769                                                    ren1->src_entry,
2770                                                    ren2->src_entry);
2771
2772                 } else {
2773                         /* Renamed in 1, maybe changed in 2 */
2774                         /* we only use sha1 and mode of these */
2775                         struct diff_filespec src_other, dst_other;
2776                         int try_merge;
2777
2778                         /*
2779                          * unpack_trees loads entries from common-commit
2780                          * into stage 1, from head-commit into stage 2, and
2781                          * from merge-commit into stage 3.  We keep track
2782                          * of which side corresponds to the rename.
2783                          */
2784                         int renamed_stage = a_renames == renames1 ? 2 : 3;
2785                         int other_stage =   a_renames == renames1 ? 3 : 2;
2786
2787                         /* BUG: We should only remove ren1_src in the base
2788                          * stage and in other_stage (think of rename +
2789                          * add-source case).
2790                          */
2791                         remove_file(o, 1, ren1_src,
2792                                     renamed_stage == 2 || !was_tracked(o, ren1_src));
2793
2794                         oidcpy(&src_other.oid,
2795                                &ren1->src_entry->stages[other_stage].oid);
2796                         src_other.mode = ren1->src_entry->stages[other_stage].mode;
2797                         oidcpy(&dst_other.oid,
2798                                &ren1->dst_entry->stages[other_stage].oid);
2799                         dst_other.mode = ren1->dst_entry->stages[other_stage].mode;
2800                         try_merge = 0;
2801
2802                         if (oid_eq(&src_other.oid, &null_oid) &&
2803                             ren1->add_turned_into_rename) {
2804                                 setup_rename_conflict_info(RENAME_VIA_DIR,
2805                                                            ren1->pair,
2806                                                            NULL,
2807                                                            branch1,
2808                                                            branch2,
2809                                                            ren1->dst_entry,
2810                                                            NULL,
2811                                                            o,
2812                                                            NULL,
2813                                                            NULL);
2814                         } else if (oid_eq(&src_other.oid, &null_oid)) {
2815                                 setup_rename_conflict_info(RENAME_DELETE,
2816                                                            ren1->pair,
2817                                                            NULL,
2818                                                            branch1,
2819                                                            branch2,
2820                                                            ren1->dst_entry,
2821                                                            NULL,
2822                                                            o,
2823                                                            NULL,
2824                                                            NULL);
2825                         } else if ((dst_other.mode == ren1->pair->two->mode) &&
2826                                    oid_eq(&dst_other.oid, &ren1->pair->two->oid)) {
2827                                 /*
2828                                  * Added file on the other side identical to
2829                                  * the file being renamed: clean merge.
2830                                  * Also, there is no need to overwrite the
2831                                  * file already in the working copy, so call
2832                                  * update_file_flags() instead of
2833                                  * update_file().
2834                                  */
2835                                 if (update_file_flags(o,
2836                                                       &ren1->pair->two->oid,
2837                                                       ren1->pair->two->mode,
2838                                                       ren1_dst,
2839                                                       1, /* update_cache */
2840                                                       0  /* update_wd    */))
2841                                         clean_merge = -1;
2842                         } else if (!oid_eq(&dst_other.oid, &null_oid)) {
2843                                 /*
2844                                  * Probably not a clean merge, but it's
2845                                  * premature to set clean_merge to 0 here,
2846                                  * because if the rename merges cleanly and
2847                                  * the merge exactly matches the newly added
2848                                  * file, then the merge will be clean.
2849                                  */
2850                                 setup_rename_conflict_info(RENAME_ADD,
2851                                                            ren1->pair,
2852                                                            NULL,
2853                                                            branch1,
2854                                                            branch2,
2855                                                            ren1->dst_entry,
2856                                                            NULL,
2857                                                            o,
2858                                                            ren1->src_entry,
2859                                                            NULL);
2860                         } else
2861                                 try_merge = 1;
2862
2863                         if (clean_merge < 0)
2864                                 goto cleanup_and_return;
2865                         if (try_merge) {
2866                                 struct diff_filespec *one, *a, *b;
2867                                 src_other.path = (char *)ren1_src;
2868
2869                                 one = ren1->pair->one;
2870                                 if (a_renames == renames1) {
2871                                         a = ren1->pair->two;
2872                                         b = &src_other;
2873                                 } else {
2874                                         b = ren1->pair->two;
2875                                         a = &src_other;
2876                                 }
2877                                 update_entry(ren1->dst_entry, one, a, b);
2878                                 setup_rename_conflict_info(RENAME_NORMAL,
2879                                                            ren1->pair,
2880                                                            NULL,
2881                                                            branch1,
2882                                                            NULL,
2883                                                            ren1->dst_entry,
2884                                                            NULL,
2885                                                            o,
2886                                                            NULL,
2887                                                            NULL);
2888                         }
2889                 }
2890         }
2891 cleanup_and_return:
2892         string_list_clear(&a_by_dst, 0);
2893         string_list_clear(&b_by_dst, 0);
2894
2895         return clean_merge;
2896 }
2897
2898 struct rename_info {
2899         struct string_list *head_renames;
2900         struct string_list *merge_renames;
2901 };
2902
2903 static void initial_cleanup_rename(struct diff_queue_struct *pairs,
2904                                    struct hashmap *dir_renames)
2905 {
2906         struct hashmap_iter iter;
2907         struct dir_rename_entry *e;
2908
2909         hashmap_iter_init(dir_renames, &iter);
2910         while ((e = hashmap_iter_next(&iter))) {
2911                 free(e->dir);
2912                 strbuf_release(&e->new_dir);
2913                 /* possible_new_dirs already cleared in get_directory_renames */
2914         }
2915         hashmap_free(dir_renames, 1);
2916         free(dir_renames);
2917
2918         free(pairs->queue);
2919         free(pairs);
2920 }
2921
2922 static int detect_and_process_renames(struct merge_options *o,
2923                                       struct tree *common,
2924                                       struct tree *head,
2925                                       struct tree *merge,
2926                                       struct string_list *entries,
2927                                       struct rename_info *ri)
2928 {
2929         struct diff_queue_struct *head_pairs, *merge_pairs;
2930         struct hashmap *dir_re_head, *dir_re_merge;
2931         int clean = 1;
2932
2933         ri->head_renames = NULL;
2934         ri->merge_renames = NULL;
2935
2936         if (!merge_detect_rename(o))
2937                 return 1;
2938
2939         head_pairs = get_diffpairs(o, common, head);
2940         merge_pairs = get_diffpairs(o, common, merge);
2941
2942         if (o->detect_directory_renames) {
2943                 dir_re_head = get_directory_renames(head_pairs);
2944                 dir_re_merge = get_directory_renames(merge_pairs);
2945
2946                 handle_directory_level_conflicts(o,
2947                                                  dir_re_head, head,
2948                                                  dir_re_merge, merge);
2949         } else {
2950                 dir_re_head  = xmalloc(sizeof(*dir_re_head));
2951                 dir_re_merge = xmalloc(sizeof(*dir_re_merge));
2952                 dir_rename_init(dir_re_head);
2953                 dir_rename_init(dir_re_merge);
2954         }
2955
2956         ri->head_renames  = get_renames(o, head_pairs,
2957                                         dir_re_merge, dir_re_head, head,
2958                                         common, head, merge, entries,
2959                                         &clean);
2960         if (clean < 0)
2961                 goto cleanup;
2962         ri->merge_renames = get_renames(o, merge_pairs,
2963                                         dir_re_head, dir_re_merge, merge,
2964                                         common, head, merge, entries,
2965                                         &clean);
2966         if (clean < 0)
2967                 goto cleanup;
2968         clean &= process_renames(o, ri->head_renames, ri->merge_renames);
2969
2970 cleanup:
2971         /*
2972          * Some cleanup is deferred until cleanup_renames() because the
2973          * data structures are still needed and referenced in
2974          * process_entry().  But there are a few things we can free now.
2975          */
2976         initial_cleanup_rename(head_pairs, dir_re_head);
2977         initial_cleanup_rename(merge_pairs, dir_re_merge);
2978
2979         return clean;
2980 }
2981
2982 static void final_cleanup_rename(struct string_list *rename)
2983 {
2984         const struct rename *re;
2985         int i;
2986
2987         if (rename == NULL)
2988                 return;
2989
2990         for (i = 0; i < rename->nr; i++) {
2991                 re = rename->items[i].util;
2992                 diff_free_filepair(re->pair);
2993         }
2994         string_list_clear(rename, 1);
2995         free(rename);
2996 }
2997
2998 static void final_cleanup_renames(struct rename_info *re_info)
2999 {
3000         final_cleanup_rename(re_info->head_renames);
3001         final_cleanup_rename(re_info->merge_renames);
3002 }
3003
3004 static struct object_id *stage_oid(const struct object_id *oid, unsigned mode)
3005 {
3006         return (is_null_oid(oid) || mode == 0) ? NULL: (struct object_id *)oid;
3007 }
3008
3009 static int read_oid_strbuf(struct merge_options *o,
3010                            const struct object_id *oid,
3011                            struct strbuf *dst)
3012 {
3013         void *buf;
3014         enum object_type type;
3015         unsigned long size;
3016         buf = read_object_file(oid, &type, &size);
3017         if (!buf)
3018                 return err(o, _("cannot read object %s"), oid_to_hex(oid));
3019         if (type != OBJ_BLOB) {
3020                 free(buf);
3021                 return err(o, _("object %s is not a blob"), oid_to_hex(oid));
3022         }
3023         strbuf_attach(dst, buf, size, size + 1);
3024         return 0;
3025 }
3026
3027 static int blob_unchanged(struct merge_options *opt,
3028                           const struct object_id *o_oid,
3029                           unsigned o_mode,
3030                           const struct object_id *a_oid,
3031                           unsigned a_mode,
3032                           int renormalize, const char *path)
3033 {
3034         struct strbuf o = STRBUF_INIT;
3035         struct strbuf a = STRBUF_INIT;
3036         int ret = 0; /* assume changed for safety */
3037
3038         if (a_mode != o_mode)
3039                 return 0;
3040         if (oid_eq(o_oid, a_oid))
3041                 return 1;
3042         if (!renormalize)
3043                 return 0;
3044
3045         assert(o_oid && a_oid);
3046         if (read_oid_strbuf(opt, o_oid, &o) || read_oid_strbuf(opt, a_oid, &a))
3047                 goto error_return;
3048         /*
3049          * Note: binary | is used so that both renormalizations are
3050          * performed.  Comparison can be skipped if both files are
3051          * unchanged since their sha1s have already been compared.
3052          */
3053         if (renormalize_buffer(opt->repo->index, path, o.buf, o.len, &o) |
3054             renormalize_buffer(opt->repo->index, path, a.buf, a.len, &a))
3055                 ret = (o.len == a.len && !memcmp(o.buf, a.buf, o.len));
3056
3057 error_return:
3058         strbuf_release(&o);
3059         strbuf_release(&a);
3060         return ret;
3061 }
3062
3063 static int handle_modify_delete(struct merge_options *o,
3064                                 const char *path,
3065                                 struct object_id *o_oid, int o_mode,
3066                                 struct object_id *a_oid, int a_mode,
3067                                 struct object_id *b_oid, int b_mode)
3068 {
3069         const char *modify_branch, *delete_branch;
3070         struct object_id *changed_oid;
3071         int changed_mode;
3072
3073         if (a_oid) {
3074                 modify_branch = o->branch1;
3075                 delete_branch = o->branch2;
3076                 changed_oid = a_oid;
3077                 changed_mode = a_mode;
3078         } else {
3079                 modify_branch = o->branch2;
3080                 delete_branch = o->branch1;
3081                 changed_oid = b_oid;
3082                 changed_mode = b_mode;
3083         }
3084
3085         return handle_change_delete(o,
3086                                     path, NULL,
3087                                     o_oid, o_mode,
3088                                     changed_oid, changed_mode,
3089                                     modify_branch, delete_branch,
3090                                     _("modify"), _("modified"));
3091 }
3092
3093 static int handle_content_merge(struct merge_options *o,
3094                                 const char *path,
3095                                 int is_dirty,
3096                                 struct object_id *o_oid, int o_mode,
3097                                 struct object_id *a_oid, int a_mode,
3098                                 struct object_id *b_oid, int b_mode,
3099                                 struct rename_conflict_info *rename_conflict_info)
3100 {
3101         const char *reason = _("content");
3102         const char *path1 = NULL, *path2 = NULL;
3103         struct merge_file_info mfi;
3104         struct diff_filespec one, a, b;
3105         unsigned df_conflict_remains = 0;
3106
3107         if (!o_oid) {
3108                 reason = _("add/add");
3109                 o_oid = (struct object_id *)&null_oid;
3110         }
3111         one.path = a.path = b.path = (char *)path;
3112         oidcpy(&one.oid, o_oid);
3113         one.mode = o_mode;
3114         oidcpy(&a.oid, a_oid);
3115         a.mode = a_mode;
3116         oidcpy(&b.oid, b_oid);
3117         b.mode = b_mode;
3118
3119         if (rename_conflict_info) {
3120                 struct diff_filepair *pair1 = rename_conflict_info->pair1;
3121
3122                 path1 = (o->branch1 == rename_conflict_info->branch1) ?
3123                         pair1->two->path : pair1->one->path;
3124                 /* If rename_conflict_info->pair2 != NULL, we are in
3125                  * RENAME_ONE_FILE_TO_ONE case.  Otherwise, we have a
3126                  * normal rename.
3127                  */
3128                 path2 = (rename_conflict_info->pair2 ||
3129                          o->branch2 == rename_conflict_info->branch1) ?
3130                         pair1->two->path : pair1->one->path;
3131                 one.path = pair1->one->path;
3132                 a.path = (char *)path1;
3133                 b.path = (char *)path2;
3134
3135                 if (dir_in_way(o->repo->index, path, !o->call_depth,
3136                                S_ISGITLINK(pair1->two->mode)))
3137                         df_conflict_remains = 1;
3138         }
3139         if (merge_mode_and_contents(o, &one, &a, &b, path,
3140                                     o->branch1, o->branch2,
3141                                     o->call_depth * 2, &mfi))
3142                 return -1;
3143
3144         /*
3145          * We can skip updating the working tree file iff:
3146          *   a) The merge is clean
3147          *   b) The merge matches what was in HEAD (content, mode, pathname)
3148          *   c) The target path is usable (i.e. not involved in D/F conflict)
3149          */
3150         if (mfi.clean &&
3151             was_tracked_and_matches(o, path, &mfi.oid, mfi.mode) &&
3152             !df_conflict_remains) {
3153                 int pos;
3154                 struct cache_entry *ce;
3155
3156                 output(o, 3, _("Skipped %s (merged same as existing)"), path);
3157                 if (add_cacheinfo(o, mfi.mode, &mfi.oid, path,
3158                                   0, (!o->call_depth && !is_dirty), 0))
3159                         return -1;
3160                 /*
3161                  * However, add_cacheinfo() will delete the old cache entry
3162                  * and add a new one.  We need to copy over any skip_worktree
3163                  * flag to avoid making the file appear as if it were
3164                  * deleted by the user.
3165                  */
3166                 pos = index_name_pos(&o->orig_index, path, strlen(path));
3167                 ce = o->orig_index.cache[pos];
3168                 if (ce_skip_worktree(ce)) {
3169                         pos = index_name_pos(o->repo->index, path, strlen(path));
3170                         ce = o->repo->index->cache[pos];
3171                         ce->ce_flags |= CE_SKIP_WORKTREE;
3172                 }
3173                 return mfi.clean;
3174         }
3175
3176         if (!mfi.clean) {
3177                 if (S_ISGITLINK(mfi.mode))
3178                         reason = _("submodule");
3179                 output(o, 1, _("CONFLICT (%s): Merge conflict in %s"),
3180                                 reason, path);
3181                 if (rename_conflict_info && !df_conflict_remains)
3182                         if (update_stages(o, path, &one, &a, &b))
3183                                 return -1;
3184         }
3185
3186         if (df_conflict_remains || is_dirty) {
3187                 char *new_path;
3188                 if (o->call_depth) {
3189                         remove_file_from_index(o->repo->index, path);
3190                 } else {
3191                         if (!mfi.clean) {
3192                                 if (update_stages(o, path, &one, &a, &b))
3193                                         return -1;
3194                         } else {
3195                                 int file_from_stage2 = was_tracked(o, path);
3196                                 struct diff_filespec merged;
3197                                 oidcpy(&merged.oid, &mfi.oid);
3198                                 merged.mode = mfi.mode;
3199
3200                                 if (update_stages(o, path, NULL,
3201                                                   file_from_stage2 ? &merged : NULL,
3202                                                   file_from_stage2 ? NULL : &merged))
3203                                         return -1;
3204                         }
3205
3206                 }
3207                 new_path = unique_path(o, path, rename_conflict_info->branch1);
3208                 if (is_dirty) {
3209                         output(o, 1, _("Refusing to lose dirty file at %s"),
3210                                path);
3211                 }
3212                 output(o, 1, _("Adding as %s instead"), new_path);
3213                 if (update_file(o, 0, &mfi.oid, mfi.mode, new_path)) {
3214                         free(new_path);
3215                         return -1;
3216                 }
3217                 free(new_path);
3218                 mfi.clean = 0;
3219         } else if (update_file(o, mfi.clean, &mfi.oid, mfi.mode, path))
3220                 return -1;
3221         return !is_dirty && mfi.clean;
3222 }
3223
3224 static int handle_rename_normal(struct merge_options *o,
3225                                 const char *path,
3226                                 struct object_id *o_oid, unsigned int o_mode,
3227                                 struct object_id *a_oid, unsigned int a_mode,
3228                                 struct object_id *b_oid, unsigned int b_mode,
3229                                 struct rename_conflict_info *ci)
3230 {
3231         /* Merge the content and write it out */
3232         return handle_content_merge(o, path, was_dirty(o, path),
3233                                     o_oid, o_mode, a_oid, a_mode, b_oid, b_mode,
3234                                     ci);
3235 }
3236
3237 /* Per entry merge function */
3238 static int process_entry(struct merge_options *o,
3239                          const char *path, struct stage_data *entry)
3240 {
3241         int clean_merge = 1;
3242         int normalize = o->renormalize;
3243         unsigned o_mode = entry->stages[1].mode;
3244         unsigned a_mode = entry->stages[2].mode;
3245         unsigned b_mode = entry->stages[3].mode;
3246         struct object_id *o_oid = stage_oid(&entry->stages[1].oid, o_mode);
3247         struct object_id *a_oid = stage_oid(&entry->stages[2].oid, a_mode);
3248         struct object_id *b_oid = stage_oid(&entry->stages[3].oid, b_mode);
3249
3250         entry->processed = 1;
3251         if (entry->rename_conflict_info) {
3252                 struct rename_conflict_info *conflict_info = entry->rename_conflict_info;
3253                 switch (conflict_info->rename_type) {
3254                 case RENAME_NORMAL:
3255                 case RENAME_ONE_FILE_TO_ONE:
3256                         clean_merge = handle_rename_normal(o,
3257                                                            path,
3258                                                            o_oid, o_mode,
3259                                                            a_oid, a_mode,
3260                                                            b_oid, b_mode,
3261                                                            conflict_info);
3262                         break;
3263                 case RENAME_VIA_DIR:
3264                         clean_merge = 1;
3265                         if (handle_rename_via_dir(o,
3266                                                   conflict_info->pair1,
3267                                                   conflict_info->branch1))
3268                                 clean_merge = -1;
3269                         break;
3270                 case RENAME_ADD:
3271                         /*
3272                          * Probably unclean merge, but if the renamed file
3273                          * merges cleanly and the result can then be
3274                          * two-way merged cleanly with the added file, I
3275                          * guess it's a clean merge?
3276                          */
3277                         clean_merge = handle_rename_add(o, conflict_info);
3278                         break;
3279                 case RENAME_DELETE:
3280                         clean_merge = 0;
3281                         if (handle_rename_delete(o,
3282                                                  conflict_info->pair1,
3283                                                  conflict_info->branch1,
3284                                                  conflict_info->branch2))
3285                                 clean_merge = -1;
3286                         break;
3287                 case RENAME_ONE_FILE_TO_TWO:
3288                         clean_merge = 0;
3289                         if (handle_rename_rename_1to2(o, conflict_info))
3290                                 clean_merge = -1;
3291                         break;
3292                 case RENAME_TWO_FILES_TO_ONE:
3293                         /*
3294                          * Probably unclean merge, but if the two renamed
3295                          * files merge cleanly and the two resulting files
3296                          * can then be two-way merged cleanly, I guess it's
3297                          * a clean merge?
3298                          */
3299                         clean_merge = handle_rename_rename_2to1(o,
3300                                                                 conflict_info);
3301                         break;
3302                 default:
3303                         entry->processed = 0;
3304                         break;
3305                 }
3306         } else if (o_oid && (!a_oid || !b_oid)) {
3307                 /* Case A: Deleted in one */
3308                 if ((!a_oid && !b_oid) ||
3309                     (!b_oid && blob_unchanged(o, o_oid, o_mode, a_oid, a_mode, normalize, path)) ||
3310                     (!a_oid && blob_unchanged(o, o_oid, o_mode, b_oid, b_mode, normalize, path))) {
3311                         /* Deleted in both or deleted in one and
3312                          * unchanged in the other */
3313                         if (a_oid)
3314                                 output(o, 2, _("Removing %s"), path);
3315                         /* do not touch working file if it did not exist */
3316                         remove_file(o, 1, path, !a_oid);
3317                 } else {
3318                         /* Modify/delete; deleted side may have put a directory in the way */
3319                         clean_merge = 0;
3320                         if (handle_modify_delete(o, path, o_oid, o_mode,
3321                                                  a_oid, a_mode, b_oid, b_mode))
3322                                 clean_merge = -1;
3323                 }
3324         } else if ((!o_oid && a_oid && !b_oid) ||
3325                    (!o_oid && !a_oid && b_oid)) {
3326                 /* Case B: Added in one. */
3327                 /* [nothing|directory] -> ([nothing|directory], file) */
3328
3329                 const char *add_branch;
3330                 const char *other_branch;
3331                 unsigned mode;
3332                 const struct object_id *oid;
3333                 const char *conf;
3334
3335                 if (a_oid) {
3336                         add_branch = o->branch1;
3337                         other_branch = o->branch2;
3338                         mode = a_mode;
3339                         oid = a_oid;
3340                         conf = _("file/directory");
3341                 } else {
3342                         add_branch = o->branch2;
3343                         other_branch = o->branch1;
3344                         mode = b_mode;
3345                         oid = b_oid;
3346                         conf = _("directory/file");
3347                 }
3348                 if (dir_in_way(o->repo->index, path,
3349                                !o->call_depth && !S_ISGITLINK(a_mode),
3350                                0)) {
3351                         char *new_path = unique_path(o, path, add_branch);
3352                         clean_merge = 0;
3353                         output(o, 1, _("CONFLICT (%s): There is a directory with name %s in %s. "
3354                                "Adding %s as %s"),
3355                                conf, path, other_branch, path, new_path);
3356                         if (update_file(o, 0, oid, mode, new_path))
3357                                 clean_merge = -1;
3358                         else if (o->call_depth)
3359                                 remove_file_from_index(o->repo->index, path);
3360                         free(new_path);
3361                 } else {
3362                         output(o, 2, _("Adding %s"), path);
3363                         /* do not overwrite file if already present */
3364                         if (update_file_flags(o, oid, mode, path, 1, !a_oid))
3365                                 clean_merge = -1;
3366                 }
3367         } else if (a_oid && b_oid) {
3368                 if (!o_oid) {
3369                         /* Case C: Added in both (check for same permissions) */
3370                         output(o, 1,
3371                                _("CONFLICT (add/add): Merge conflict in %s"),
3372                                path);
3373                         clean_merge = handle_file_collision(o,
3374                                                             path, NULL, NULL,
3375                                                             o->branch1,
3376                                                             o->branch2,
3377                                                             a_oid, a_mode,
3378                                                             b_oid, b_mode);
3379                 } else {
3380                         /* case D: Modified in both, but differently. */
3381                         int is_dirty = 0; /* unpack_trees would have bailed if dirty */
3382                         clean_merge = handle_content_merge(o, path,
3383                                                            is_dirty,
3384                                                            o_oid, o_mode,
3385                                                            a_oid, a_mode,
3386                                                            b_oid, b_mode,
3387                                                            NULL);
3388                 }
3389         } else if (!o_oid && !a_oid && !b_oid) {
3390                 /*
3391                  * this entry was deleted altogether. a_mode == 0 means
3392                  * we had that path and want to actively remove it.
3393                  */
3394                 remove_file(o, 1, path, !a_mode);
3395         } else
3396                 BUG("fatal merge failure, shouldn't happen.");
3397
3398         return clean_merge;
3399 }
3400
3401 int merge_trees(struct merge_options *o,
3402                 struct tree *head,
3403                 struct tree *merge,
3404                 struct tree *common,
3405                 struct tree **result)
3406 {
3407         struct index_state *istate = o->repo->index;
3408         int code, clean;
3409         struct strbuf sb = STRBUF_INIT;
3410
3411         if (!o->call_depth && repo_index_has_changes(o->repo, head, &sb)) {
3412                 err(o, _("Your local changes to the following files would be overwritten by merge:\n  %s"),
3413                     sb.buf);
3414                 return -1;
3415         }
3416
3417         if (o->subtree_shift) {
3418                 merge = shift_tree_object(o->repo, head, merge, o->subtree_shift);
3419                 common = shift_tree_object(o->repo, head, common, o->subtree_shift);
3420         }
3421
3422         if (oid_eq(&common->object.oid, &merge->object.oid)) {
3423                 output(o, 0, _("Already up to date!"));
3424                 *result = head;
3425                 return 1;
3426         }
3427
3428         code = unpack_trees_start(o, common, head, merge);
3429
3430         if (code != 0) {
3431                 if (show(o, 4) || o->call_depth)
3432                         err(o, _("merging of trees %s and %s failed"),
3433                             oid_to_hex(&head->object.oid),
3434                             oid_to_hex(&merge->object.oid));
3435                 unpack_trees_finish(o);
3436                 return -1;
3437         }
3438
3439         if (unmerged_index(istate)) {
3440                 struct string_list *entries;
3441                 struct rename_info re_info;
3442                 int i;
3443                 /*
3444                  * Only need the hashmap while processing entries, so
3445                  * initialize it here and free it when we are done running
3446                  * through the entries. Keeping it in the merge_options as
3447                  * opposed to decaring a local hashmap is for convenience
3448                  * so that we don't have to pass it to around.
3449                  */
3450                 hashmap_init(&o->current_file_dir_set, path_hashmap_cmp, NULL, 512);
3451                 get_files_dirs(o, head);
3452                 get_files_dirs(o, merge);
3453
3454                 entries = get_unmerged(o->repo->index);
3455                 clean = detect_and_process_renames(o, common, head, merge,
3456                                                    entries, &re_info);
3457                 record_df_conflict_files(o, entries);
3458                 if (clean < 0)
3459                         goto cleanup;
3460                 for (i = entries->nr-1; 0 <= i; i--) {
3461                         const char *path = entries->items[i].string;
3462                         struct stage_data *e = entries->items[i].util;
3463                         if (!e->processed) {
3464                                 int ret = process_entry(o, path, e);
3465                                 if (!ret)
3466                                         clean = 0;
3467                                 else if (ret < 0) {
3468                                         clean = ret;
3469                                         goto cleanup;
3470                                 }
3471                         }
3472                 }
3473                 for (i = 0; i < entries->nr; i++) {
3474                         struct stage_data *e = entries->items[i].util;
3475                         if (!e->processed)
3476                                 BUG("unprocessed path??? %s",
3477                                     entries->items[i].string);
3478                 }
3479
3480         cleanup:
3481                 final_cleanup_renames(&re_info);
3482
3483                 string_list_clear(entries, 1);
3484                 free(entries);
3485
3486                 hashmap_free(&o->current_file_dir_set, 1);
3487
3488                 if (clean < 0) {
3489                         unpack_trees_finish(o);
3490                         return clean;
3491                 }
3492         }
3493         else
3494                 clean = 1;
3495
3496         unpack_trees_finish(o);
3497
3498         if (o->call_depth && !(*result = write_tree_from_memory(o)))
3499                 return -1;
3500
3501         return clean;
3502 }
3503
3504 static struct commit_list *reverse_commit_list(struct commit_list *list)
3505 {
3506         struct commit_list *next = NULL, *current, *backup;
3507         for (current = list; current; current = backup) {
3508                 backup = current->next;
3509                 current->next = next;
3510                 next = current;
3511         }
3512         return next;
3513 }
3514
3515 /*
3516  * Merge the commits h1 and h2, return the resulting virtual
3517  * commit object and a flag indicating the cleanness of the merge.
3518  */
3519 int merge_recursive(struct merge_options *o,
3520                     struct commit *h1,
3521                     struct commit *h2,
3522                     struct commit_list *ca,
3523                     struct commit **result)
3524 {
3525         struct commit_list *iter;
3526         struct commit *merged_common_ancestors;
3527         struct tree *mrtree;
3528         int clean;
3529
3530         if (show(o, 4)) {
3531                 output(o, 4, _("Merging:"));
3532                 output_commit_title(o, h1);
3533                 output_commit_title(o, h2);
3534         }
3535
3536         if (!ca) {
3537                 ca = get_merge_bases(h1, h2);
3538                 ca = reverse_commit_list(ca);
3539         }
3540
3541         if (show(o, 5)) {
3542                 unsigned cnt = commit_list_count(ca);
3543
3544                 output(o, 5, Q_("found %u common ancestor:",
3545                                 "found %u common ancestors:", cnt), cnt);
3546                 for (iter = ca; iter; iter = iter->next)
3547                         output_commit_title(o, iter->item);
3548         }
3549
3550         merged_common_ancestors = pop_commit(&ca);
3551         if (merged_common_ancestors == NULL) {
3552                 /* if there is no common ancestor, use an empty tree */
3553                 struct tree *tree;
3554
3555                 tree = lookup_tree(o->repo, o->repo->hash_algo->empty_tree);
3556                 merged_common_ancestors = make_virtual_commit(o->repo, tree, "ancestor");
3557         }
3558
3559         for (iter = ca; iter; iter = iter->next) {
3560                 const char *saved_b1, *saved_b2;
3561                 o->call_depth++;
3562                 /*
3563                  * When the merge fails, the result contains files
3564                  * with conflict markers. The cleanness flag is
3565                  * ignored (unless indicating an error), it was never
3566                  * actually used, as result of merge_trees has always
3567                  * overwritten it: the committed "conflicts" were
3568                  * already resolved.
3569                  */
3570                 discard_index(o->repo->index);
3571                 saved_b1 = o->branch1;
3572                 saved_b2 = o->branch2;
3573                 o->branch1 = "Temporary merge branch 1";
3574                 o->branch2 = "Temporary merge branch 2";
3575                 if (merge_recursive(o, merged_common_ancestors, iter->item,
3576                                     NULL, &merged_common_ancestors) < 0)
3577                         return -1;
3578                 o->branch1 = saved_b1;
3579                 o->branch2 = saved_b2;
3580                 o->call_depth--;
3581
3582                 if (!merged_common_ancestors)
3583                         return err(o, _("merge returned no commit"));
3584         }
3585
3586         discard_index(o->repo->index);
3587         if (!o->call_depth)
3588                 repo_read_index(o->repo);
3589
3590         o->ancestor = "merged common ancestors";
3591         clean = merge_trees(o, get_commit_tree(h1), get_commit_tree(h2),
3592                             get_commit_tree(merged_common_ancestors),
3593                             &mrtree);
3594         if (clean < 0) {
3595                 flush_output(o);
3596                 return clean;
3597         }
3598
3599         if (o->call_depth) {
3600                 *result = make_virtual_commit(o->repo, mrtree, "merged tree");
3601                 commit_list_insert(h1, &(*result)->parents);
3602                 commit_list_insert(h2, &(*result)->parents->next);
3603         }
3604         flush_output(o);
3605         if (!o->call_depth && o->buffer_output < 2)
3606                 strbuf_release(&o->obuf);
3607         if (show(o, 2))
3608                 diff_warn_rename_limit("merge.renamelimit",
3609                                        o->needed_rename_limit, 0);
3610         return clean;
3611 }
3612
3613 static struct commit *get_ref(struct repository *repo, const struct object_id *oid,
3614                               const char *name)
3615 {
3616         struct object *object;
3617
3618         object = deref_tag(repo, parse_object(repo, oid),
3619                            name, strlen(name));
3620         if (!object)
3621                 return NULL;
3622         if (object->type == OBJ_TREE)
3623                 return make_virtual_commit(repo, (struct tree*)object, name);
3624         if (object->type != OBJ_COMMIT)
3625                 return NULL;
3626         if (parse_commit((struct commit *)object))
3627                 return NULL;
3628         return (struct commit *)object;
3629 }
3630
3631 int merge_recursive_generic(struct merge_options *o,
3632                             const struct object_id *head,
3633                             const struct object_id *merge,
3634                             int num_base_list,
3635                             const struct object_id **base_list,
3636                             struct commit **result)
3637 {
3638         int clean;
3639         struct lock_file lock = LOCK_INIT;
3640         struct commit *head_commit = get_ref(o->repo, head, o->branch1);
3641         struct commit *next_commit = get_ref(o->repo, merge, o->branch2);
3642         struct commit_list *ca = NULL;
3643
3644         if (base_list) {
3645                 int i;
3646                 for (i = 0; i < num_base_list; ++i) {
3647                         struct commit *base;
3648                         if (!(base = get_ref(o->repo, base_list[i], oid_to_hex(base_list[i]))))
3649                                 return err(o, _("Could not parse object '%s'"),
3650                                            oid_to_hex(base_list[i]));
3651                         commit_list_insert(base, &ca);
3652                 }
3653         }
3654
3655         repo_hold_locked_index(o->repo, &lock, LOCK_DIE_ON_ERROR);
3656         clean = merge_recursive(o, head_commit, next_commit, ca,
3657                                 result);
3658         if (clean < 0) {
3659                 rollback_lock_file(&lock);
3660                 return clean;
3661         }
3662
3663         if (write_locked_index(o->repo->index, &lock,
3664                                COMMIT_LOCK | SKIP_IF_UNCHANGED))
3665                 return err(o, _("Unable to write index."));
3666
3667         return clean ? 0 : 1;
3668 }
3669
3670 static void merge_recursive_config(struct merge_options *o)
3671 {
3672         char *value = NULL;
3673         git_config_get_int("merge.verbosity", &o->verbosity);
3674         git_config_get_int("diff.renamelimit", &o->diff_rename_limit);
3675         git_config_get_int("merge.renamelimit", &o->merge_rename_limit);
3676         if (!git_config_get_string("diff.renames", &value)) {
3677                 o->diff_detect_rename = git_config_rename("diff.renames", value);
3678                 free(value);
3679         }
3680         if (!git_config_get_string("merge.renames", &value)) {
3681                 o->merge_detect_rename = git_config_rename("merge.renames", value);
3682                 free(value);
3683         }
3684         git_config(git_xmerge_config, NULL);
3685 }
3686
3687 void init_merge_options(struct merge_options *o,
3688                         struct repository *repo)
3689 {
3690         const char *merge_verbosity;
3691         memset(o, 0, sizeof(struct merge_options));
3692         o->repo = repo;
3693         o->verbosity = 2;
3694         o->buffer_output = 1;
3695         o->diff_rename_limit = -1;
3696         o->merge_rename_limit = -1;
3697         o->renormalize = 0;
3698         o->diff_detect_rename = -1;
3699         o->merge_detect_rename = -1;
3700         o->detect_directory_renames = 1;
3701         merge_recursive_config(o);
3702         merge_verbosity = getenv("GIT_MERGE_VERBOSITY");
3703         if (merge_verbosity)
3704                 o->verbosity = strtol(merge_verbosity, NULL, 10);
3705         if (o->verbosity >= 5)
3706                 o->buffer_output = 0;
3707         strbuf_init(&o->obuf, 0);
3708         string_list_init(&o->df_conflict_file_set, 1);
3709 }
3710
3711 int parse_merge_opt(struct merge_options *o, const char *s)
3712 {
3713         const char *arg;
3714
3715         if (!s || !*s)
3716                 return -1;
3717         if (!strcmp(s, "ours"))
3718                 o->recursive_variant = MERGE_RECURSIVE_OURS;
3719         else if (!strcmp(s, "theirs"))
3720                 o->recursive_variant = MERGE_RECURSIVE_THEIRS;
3721         else if (!strcmp(s, "subtree"))
3722                 o->subtree_shift = "";
3723         else if (skip_prefix(s, "subtree=", &arg))
3724                 o->subtree_shift = arg;
3725         else if (!strcmp(s, "patience"))
3726                 o->xdl_opts = DIFF_WITH_ALG(o, PATIENCE_DIFF);
3727         else if (!strcmp(s, "histogram"))
3728                 o->xdl_opts = DIFF_WITH_ALG(o, HISTOGRAM_DIFF);
3729         else if (skip_prefix(s, "diff-algorithm=", &arg)) {
3730                 long value = parse_algorithm_value(arg);
3731                 if (value < 0)
3732                         return -1;
3733                 /* clear out previous settings */
3734                 DIFF_XDL_CLR(o, NEED_MINIMAL);
3735                 o->xdl_opts &= ~XDF_DIFF_ALGORITHM_MASK;
3736                 o->xdl_opts |= value;
3737         }
3738         else if (!strcmp(s, "ignore-space-change"))
3739                 DIFF_XDL_SET(o, IGNORE_WHITESPACE_CHANGE);
3740         else if (!strcmp(s, "ignore-all-space"))
3741                 DIFF_XDL_SET(o, IGNORE_WHITESPACE);
3742         else if (!strcmp(s, "ignore-space-at-eol"))
3743                 DIFF_XDL_SET(o, IGNORE_WHITESPACE_AT_EOL);
3744         else if (!strcmp(s, "ignore-cr-at-eol"))
3745                 DIFF_XDL_SET(o, IGNORE_CR_AT_EOL);
3746         else if (!strcmp(s, "renormalize"))
3747                 o->renormalize = 1;
3748         else if (!strcmp(s, "no-renormalize"))
3749                 o->renormalize = 0;
3750         else if (!strcmp(s, "no-renames"))
3751                 o->merge_detect_rename = 0;
3752         else if (!strcmp(s, "find-renames")) {
3753                 o->merge_detect_rename = 1;
3754                 o->rename_score = 0;
3755         }
3756         else if (skip_prefix(s, "find-renames=", &arg) ||
3757                  skip_prefix(s, "rename-threshold=", &arg)) {
3758                 if ((o->rename_score = parse_rename_score(&arg)) == -1 || *arg != 0)
3759                         return -1;
3760                 o->merge_detect_rename = 1;
3761         }
3762         /*
3763          * Please update $__git_merge_strategy_options in
3764          * git-completion.bash when you add new options
3765          */
3766         else
3767                 return -1;
3768         return 0;
3769 }