]> git.ipfire.org Git - thirdparty/git.git/blob - merge-ort.c
merge-ort: handle book-keeping around two- and three-way content merge
[thirdparty/git.git] / merge-ort.c
1 /*
2 * "Ostensibly Recursive's Twin" merge strategy, or "ort" for short. Meant
3 * as a drop-in replacement for the "recursive" merge strategy, allowing one
4 * to replace
5 *
6 * git merge [-s recursive]
7 *
8 * with
9 *
10 * git merge -s ort
11 *
12 * Note: git's parser allows the space between '-s' and its argument to be
13 * missing. (Should I have backronymed "ham", "alsa", "kip", "nap, "alvo",
14 * "cale", "peedy", or "ins" instead of "ort"?)
15 */
16
17 #include "cache.h"
18 #include "merge-ort.h"
19
20 #include "blob.h"
21 #include "cache-tree.h"
22 #include "commit-reach.h"
23 #include "diff.h"
24 #include "diffcore.h"
25 #include "dir.h"
26 #include "object-store.h"
27 #include "strmap.h"
28 #include "tree.h"
29 #include "unpack-trees.h"
30 #include "xdiff-interface.h"
31
32 /*
33 * We have many arrays of size 3. Whenever we have such an array, the
34 * indices refer to one of the sides of the three-way merge. This is so
35 * pervasive that the constants 0, 1, and 2 are used in many places in the
36 * code (especially in arithmetic operations to find the other side's index
37 * or to compute a relevant mask), but sometimes these enum names are used
38 * to aid code clarity.
39 *
40 * See also 'filemask' and 'dirmask' in struct conflict_info; the "ith side"
41 * referred to there is one of these three sides.
42 */
43 enum merge_side {
44 MERGE_BASE = 0,
45 MERGE_SIDE1 = 1,
46 MERGE_SIDE2 = 2
47 };
48
49 struct merge_options_internal {
50 /*
51 * paths: primary data structure in all of merge ort.
52 *
53 * The keys of paths:
54 * * are full relative paths from the toplevel of the repository
55 * (e.g. "drivers/firmware/raspberrypi.c").
56 * * store all relevant paths in the repo, both directories and
57 * files (e.g. drivers, drivers/firmware would also be included)
58 * * these keys serve to intern all the path strings, which allows
59 * us to do pointer comparison on directory names instead of
60 * strcmp; we just have to be careful to use the interned strings.
61 * (Technically paths_to_free may track some strings that were
62 * removed from froms paths.)
63 *
64 * The values of paths:
65 * * either a pointer to a merged_info, or a conflict_info struct
66 * * merged_info contains all relevant information for a
67 * non-conflicted entry.
68 * * conflict_info contains a merged_info, plus any additional
69 * information about a conflict such as the higher orders stages
70 * involved and the names of the paths those came from (handy
71 * once renames get involved).
72 * * a path may start "conflicted" (i.e. point to a conflict_info)
73 * and then a later step (e.g. three-way content merge) determines
74 * it can be cleanly merged, at which point it'll be marked clean
75 * and the algorithm will ignore any data outside the contained
76 * merged_info for that entry
77 * * If an entry remains conflicted, the merged_info portion of a
78 * conflict_info will later be filled with whatever version of
79 * the file should be placed in the working directory (e.g. an
80 * as-merged-as-possible variation that contains conflict markers).
81 */
82 struct strmap paths;
83
84 /*
85 * conflicted: a subset of keys->values from "paths"
86 *
87 * conflicted is basically an optimization between process_entries()
88 * and record_conflicted_index_entries(); the latter could loop over
89 * ALL the entries in paths AGAIN and look for the ones that are
90 * still conflicted, but since process_entries() has to loop over
91 * all of them, it saves the ones it couldn't resolve in this strmap
92 * so that record_conflicted_index_entries() can iterate just the
93 * relevant entries.
94 */
95 struct strmap conflicted;
96
97 /*
98 * paths_to_free: additional list of strings to free
99 *
100 * If keys are removed from "paths", they are added to paths_to_free
101 * to ensure they are later freed. We avoid free'ing immediately since
102 * other places (e.g. conflict_info.pathnames[]) may still be
103 * referencing these paths.
104 */
105 struct string_list paths_to_free;
106
107 /*
108 * output: special messages and conflict notices for various paths
109 *
110 * This is a map of pathnames (a subset of the keys in "paths" above)
111 * to strbufs. It gathers various warning/conflict/notice messages
112 * for later processing.
113 */
114 struct strmap output;
115
116 /*
117 * current_dir_name: temporary var used in collect_merge_info_callback()
118 *
119 * Used to set merged_info.directory_name; see documentation for that
120 * variable and the requirements placed on that field.
121 */
122 const char *current_dir_name;
123
124 /* call_depth: recursion level counter for merging merge bases */
125 int call_depth;
126 };
127
128 struct version_info {
129 struct object_id oid;
130 unsigned short mode;
131 };
132
133 struct merged_info {
134 /* if is_null, ignore result. otherwise result has oid & mode */
135 struct version_info result;
136 unsigned is_null:1;
137
138 /*
139 * clean: whether the path in question is cleanly merged.
140 *
141 * see conflict_info.merged for more details.
142 */
143 unsigned clean:1;
144
145 /*
146 * basename_offset: offset of basename of path.
147 *
148 * perf optimization to avoid recomputing offset of final '/'
149 * character in pathname (0 if no '/' in pathname).
150 */
151 size_t basename_offset;
152
153 /*
154 * directory_name: containing directory name.
155 *
156 * Note that we assume directory_name is constructed such that
157 * strcmp(dir1_name, dir2_name) == 0 iff dir1_name == dir2_name,
158 * i.e. string equality is equivalent to pointer equality. For this
159 * to hold, we have to be careful setting directory_name.
160 */
161 const char *directory_name;
162 };
163
164 struct conflict_info {
165 /*
166 * merged: the version of the path that will be written to working tree
167 *
168 * WARNING: It is critical to check merged.clean and ensure it is 0
169 * before reading any conflict_info fields outside of merged.
170 * Allocated merge_info structs will always have clean set to 1.
171 * Allocated conflict_info structs will have merged.clean set to 0
172 * initially. The merged.clean field is how we know if it is safe
173 * to access other parts of conflict_info besides merged; if a
174 * conflict_info's merged.clean is changed to 1, the rest of the
175 * algorithm is not allowed to look at anything outside of the
176 * merged member anymore.
177 */
178 struct merged_info merged;
179
180 /* oids & modes from each of the three trees for this path */
181 struct version_info stages[3];
182
183 /* pathnames for each stage; may differ due to rename detection */
184 const char *pathnames[3];
185
186 /* Whether this path is/was involved in a directory/file conflict */
187 unsigned df_conflict:1;
188
189 /*
190 * Whether this path is/was involved in a non-content conflict other
191 * than a directory/file conflict (e.g. rename/rename, rename/delete,
192 * file location based on possible directory rename).
193 */
194 unsigned path_conflict:1;
195
196 /*
197 * For filemask and dirmask, the ith bit corresponds to whether the
198 * ith entry is a file (filemask) or a directory (dirmask). Thus,
199 * filemask & dirmask is always zero, and filemask | dirmask is at
200 * most 7 but can be less when a path does not appear as either a
201 * file or a directory on at least one side of history.
202 *
203 * Note that these masks are related to enum merge_side, as the ith
204 * entry corresponds to side i.
205 *
206 * These values come from a traverse_trees() call; more info may be
207 * found looking at tree-walk.h's struct traverse_info,
208 * particularly the documentation above the "fn" member (note that
209 * filemask = mask & ~dirmask from that documentation).
210 */
211 unsigned filemask:3;
212 unsigned dirmask:3;
213
214 /*
215 * Optimization to track which stages match, to avoid the need to
216 * recompute it in multiple steps. Either 0 or at least 2 bits are
217 * set; if at least 2 bits are set, their corresponding stages match.
218 */
219 unsigned match_mask:3;
220 };
221
222 /*** Function Grouping: various utility functions ***/
223
224 /*
225 * For the next three macros, see warning for conflict_info.merged.
226 *
227 * In each of the below, mi is a struct merged_info*, and ci was defined
228 * as a struct conflict_info* (but we need to verify ci isn't actually
229 * pointed at a struct merged_info*).
230 *
231 * INITIALIZE_CI: Assign ci to mi but only if it's safe; set to NULL otherwise.
232 * VERIFY_CI: Ensure that something we assigned to a conflict_info* is one.
233 * ASSIGN_AND_VERIFY_CI: Similar to VERIFY_CI but do assignment first.
234 */
235 #define INITIALIZE_CI(ci, mi) do { \
236 (ci) = (!(mi) || (mi)->clean) ? NULL : (struct conflict_info *)(mi); \
237 } while (0)
238 #define VERIFY_CI(ci) assert(ci && !ci->merged.clean);
239 #define ASSIGN_AND_VERIFY_CI(ci, mi) do { \
240 (ci) = (struct conflict_info *)(mi); \
241 assert((ci) && !(mi)->clean); \
242 } while (0)
243
244 static void free_strmap_strings(struct strmap *map)
245 {
246 struct hashmap_iter iter;
247 struct strmap_entry *entry;
248
249 strmap_for_each_entry(map, &iter, entry) {
250 free((char*)entry->key);
251 }
252 }
253
254 static void clear_internal_opts(struct merge_options_internal *opti,
255 int reinitialize)
256 {
257 assert(!reinitialize);
258
259 /*
260 * We marked opti->paths with strdup_strings = 0, so that we
261 * wouldn't have to make another copy of the fullpath created by
262 * make_traverse_path from setup_path_info(). But, now that we've
263 * used it and have no other references to these strings, it is time
264 * to deallocate them.
265 */
266 free_strmap_strings(&opti->paths);
267 strmap_clear(&opti->paths, 1);
268
269 /*
270 * All keys and values in opti->conflicted are a subset of those in
271 * opti->paths. We don't want to deallocate anything twice, so we
272 * don't free the keys and we pass 0 for free_values.
273 */
274 strmap_clear(&opti->conflicted, 0);
275
276 /*
277 * opti->paths_to_free is similar to opti->paths; we created it with
278 * strdup_strings = 0 to avoid making _another_ copy of the fullpath
279 * but now that we've used it and have no other references to these
280 * strings, it is time to deallocate them. We do so by temporarily
281 * setting strdup_strings to 1.
282 */
283 opti->paths_to_free.strdup_strings = 1;
284 string_list_clear(&opti->paths_to_free, 0);
285 opti->paths_to_free.strdup_strings = 0;
286
287 if (!reinitialize) {
288 struct hashmap_iter iter;
289 struct strmap_entry *e;
290
291 /* Release and free each strbuf found in output */
292 strmap_for_each_entry(&opti->output, &iter, e) {
293 struct strbuf *sb = e->value;
294 strbuf_release(sb);
295 /*
296 * While strictly speaking we don't need to free(sb)
297 * here because we could pass free_values=1 when
298 * calling strmap_clear() on opti->output, that would
299 * require strmap_clear to do another
300 * strmap_for_each_entry() loop, so we just free it
301 * while we're iterating anyway.
302 */
303 free(sb);
304 }
305 strmap_clear(&opti->output, 0);
306 }
307 }
308
309 static int err(struct merge_options *opt, const char *err, ...)
310 {
311 va_list params;
312 struct strbuf sb = STRBUF_INIT;
313
314 strbuf_addstr(&sb, "error: ");
315 va_start(params, err);
316 strbuf_vaddf(&sb, err, params);
317 va_end(params);
318
319 error("%s", sb.buf);
320 strbuf_release(&sb);
321
322 return -1;
323 }
324
325 __attribute__((format (printf, 4, 5)))
326 static void path_msg(struct merge_options *opt,
327 const char *path,
328 int omittable_hint, /* skippable under --remerge-diff */
329 const char *fmt, ...)
330 {
331 va_list ap;
332 struct strbuf *sb = strmap_get(&opt->priv->output, path);
333 if (!sb) {
334 sb = xmalloc(sizeof(*sb));
335 strbuf_init(sb, 0);
336 strmap_put(&opt->priv->output, path, sb);
337 }
338
339 va_start(ap, fmt);
340 strbuf_vaddf(sb, fmt, ap);
341 va_end(ap);
342
343 strbuf_addch(sb, '\n');
344 }
345
346 /* add a string to a strbuf, but converting "/" to "_" */
347 static void add_flattened_path(struct strbuf *out, const char *s)
348 {
349 size_t i = out->len;
350 strbuf_addstr(out, s);
351 for (; i < out->len; i++)
352 if (out->buf[i] == '/')
353 out->buf[i] = '_';
354 }
355
356 static char *unique_path(struct strmap *existing_paths,
357 const char *path,
358 const char *branch)
359 {
360 struct strbuf newpath = STRBUF_INIT;
361 int suffix = 0;
362 size_t base_len;
363
364 strbuf_addf(&newpath, "%s~", path);
365 add_flattened_path(&newpath, branch);
366
367 base_len = newpath.len;
368 while (strmap_contains(existing_paths, newpath.buf)) {
369 strbuf_setlen(&newpath, base_len);
370 strbuf_addf(&newpath, "_%d", suffix++);
371 }
372
373 return strbuf_detach(&newpath, NULL);
374 }
375
376 /*** Function Grouping: functions related to collect_merge_info() ***/
377
378 static void setup_path_info(struct merge_options *opt,
379 struct string_list_item *result,
380 const char *current_dir_name,
381 int current_dir_name_len,
382 char *fullpath, /* we'll take over ownership */
383 struct name_entry *names,
384 struct name_entry *merged_version,
385 unsigned is_null, /* boolean */
386 unsigned df_conflict, /* boolean */
387 unsigned filemask,
388 unsigned dirmask,
389 int resolved /* boolean */)
390 {
391 /* result->util is void*, so mi is a convenience typed variable */
392 struct merged_info *mi;
393
394 assert(!is_null || resolved);
395 assert(!df_conflict || !resolved); /* df_conflict implies !resolved */
396 assert(resolved == (merged_version != NULL));
397
398 mi = xcalloc(1, resolved ? sizeof(struct merged_info) :
399 sizeof(struct conflict_info));
400 mi->directory_name = current_dir_name;
401 mi->basename_offset = current_dir_name_len;
402 mi->clean = !!resolved;
403 if (resolved) {
404 mi->result.mode = merged_version->mode;
405 oidcpy(&mi->result.oid, &merged_version->oid);
406 mi->is_null = !!is_null;
407 } else {
408 int i;
409 struct conflict_info *ci;
410
411 ASSIGN_AND_VERIFY_CI(ci, mi);
412 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++) {
413 ci->pathnames[i] = fullpath;
414 ci->stages[i].mode = names[i].mode;
415 oidcpy(&ci->stages[i].oid, &names[i].oid);
416 }
417 ci->filemask = filemask;
418 ci->dirmask = dirmask;
419 ci->df_conflict = !!df_conflict;
420 if (dirmask)
421 /*
422 * Assume is_null for now, but if we have entries
423 * under the directory then when it is complete in
424 * write_completed_directory() it'll update this.
425 * Also, for D/F conflicts, we have to handle the
426 * directory first, then clear this bit and process
427 * the file to see how it is handled -- that occurs
428 * near the top of process_entry().
429 */
430 mi->is_null = 1;
431 }
432 strmap_put(&opt->priv->paths, fullpath, mi);
433 result->string = fullpath;
434 result->util = mi;
435 }
436
437 static int collect_merge_info_callback(int n,
438 unsigned long mask,
439 unsigned long dirmask,
440 struct name_entry *names,
441 struct traverse_info *info)
442 {
443 /*
444 * n is 3. Always.
445 * common ancestor (mbase) has mask 1, and stored in index 0 of names
446 * head of side 1 (side1) has mask 2, and stored in index 1 of names
447 * head of side 2 (side2) has mask 4, and stored in index 2 of names
448 */
449 struct merge_options *opt = info->data;
450 struct merge_options_internal *opti = opt->priv;
451 struct string_list_item pi; /* Path Info */
452 struct conflict_info *ci; /* typed alias to pi.util (which is void*) */
453 struct name_entry *p;
454 size_t len;
455 char *fullpath;
456 const char *dirname = opti->current_dir_name;
457 unsigned filemask = mask & ~dirmask;
458 unsigned match_mask = 0; /* will be updated below */
459 unsigned mbase_null = !(mask & 1);
460 unsigned side1_null = !(mask & 2);
461 unsigned side2_null = !(mask & 4);
462 unsigned side1_matches_mbase = (!side1_null && !mbase_null &&
463 names[0].mode == names[1].mode &&
464 oideq(&names[0].oid, &names[1].oid));
465 unsigned side2_matches_mbase = (!side2_null && !mbase_null &&
466 names[0].mode == names[2].mode &&
467 oideq(&names[0].oid, &names[2].oid));
468 unsigned sides_match = (!side1_null && !side2_null &&
469 names[1].mode == names[2].mode &&
470 oideq(&names[1].oid, &names[2].oid));
471
472 /*
473 * Note: When a path is a file on one side of history and a directory
474 * in another, we have a directory/file conflict. In such cases, if
475 * the conflict doesn't resolve from renames and deletions, then we
476 * always leave directories where they are and move files out of the
477 * way. Thus, while struct conflict_info has a df_conflict field to
478 * track such conflicts, we ignore that field for any directories at
479 * a path and only pay attention to it for files at the given path.
480 * The fact that we leave directories were they are also means that
481 * we do not need to worry about getting additional df_conflict
482 * information propagated from parent directories down to children
483 * (unlike, say traverse_trees_recursive() in unpack-trees.c, which
484 * sets a newinfo.df_conflicts field specifically to propagate it).
485 */
486 unsigned df_conflict = (filemask != 0) && (dirmask != 0);
487
488 /* n = 3 is a fundamental assumption. */
489 if (n != 3)
490 BUG("Called collect_merge_info_callback wrong");
491
492 /*
493 * A bunch of sanity checks verifying that traverse_trees() calls
494 * us the way I expect. Could just remove these at some point,
495 * though maybe they are helpful to future code readers.
496 */
497 assert(mbase_null == is_null_oid(&names[0].oid));
498 assert(side1_null == is_null_oid(&names[1].oid));
499 assert(side2_null == is_null_oid(&names[2].oid));
500 assert(!mbase_null || !side1_null || !side2_null);
501 assert(mask > 0 && mask < 8);
502
503 /* Determine match_mask */
504 if (side1_matches_mbase)
505 match_mask = (side2_matches_mbase ? 7 : 3);
506 else if (side2_matches_mbase)
507 match_mask = 5;
508 else if (sides_match)
509 match_mask = 6;
510
511 /*
512 * Get the name of the relevant filepath, which we'll pass to
513 * setup_path_info() for tracking.
514 */
515 p = names;
516 while (!p->mode)
517 p++;
518 len = traverse_path_len(info, p->pathlen);
519
520 /* +1 in both of the following lines to include the NUL byte */
521 fullpath = xmalloc(len + 1);
522 make_traverse_path(fullpath, len + 1, info, p->path, p->pathlen);
523
524 /*
525 * If mbase, side1, and side2 all match, we can resolve early. Even
526 * if these are trees, there will be no renames or anything
527 * underneath.
528 */
529 if (side1_matches_mbase && side2_matches_mbase) {
530 /* mbase, side1, & side2 all match; use mbase as resolution */
531 setup_path_info(opt, &pi, dirname, info->pathlen, fullpath,
532 names, names+0, mbase_null, 0,
533 filemask, dirmask, 1);
534 return mask;
535 }
536
537 /*
538 * Record information about the path so we can resolve later in
539 * process_entries.
540 */
541 setup_path_info(opt, &pi, dirname, info->pathlen, fullpath,
542 names, NULL, 0, df_conflict, filemask, dirmask, 0);
543
544 ci = pi.util;
545 VERIFY_CI(ci);
546 ci->match_mask = match_mask;
547
548 /* If dirmask, recurse into subdirectories */
549 if (dirmask) {
550 struct traverse_info newinfo;
551 struct tree_desc t[3];
552 void *buf[3] = {NULL, NULL, NULL};
553 const char *original_dir_name;
554 int i, ret;
555
556 ci->match_mask &= filemask;
557 newinfo = *info;
558 newinfo.prev = info;
559 newinfo.name = p->path;
560 newinfo.namelen = p->pathlen;
561 newinfo.pathlen = st_add3(newinfo.pathlen, p->pathlen, 1);
562 /*
563 * If this directory we are about to recurse into cared about
564 * its parent directory (the current directory) having a D/F
565 * conflict, then we'd propagate the masks in this way:
566 * newinfo.df_conflicts |= (mask & ~dirmask);
567 * But we don't worry about propagating D/F conflicts. (See
568 * comment near setting of local df_conflict variable near
569 * the beginning of this function).
570 */
571
572 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++) {
573 if (i == 1 && side1_matches_mbase)
574 t[1] = t[0];
575 else if (i == 2 && side2_matches_mbase)
576 t[2] = t[0];
577 else if (i == 2 && sides_match)
578 t[2] = t[1];
579 else {
580 const struct object_id *oid = NULL;
581 if (dirmask & 1)
582 oid = &names[i].oid;
583 buf[i] = fill_tree_descriptor(opt->repo,
584 t + i, oid);
585 }
586 dirmask >>= 1;
587 }
588
589 original_dir_name = opti->current_dir_name;
590 opti->current_dir_name = pi.string;
591 ret = traverse_trees(NULL, 3, t, &newinfo);
592 opti->current_dir_name = original_dir_name;
593
594 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++)
595 free(buf[i]);
596
597 if (ret < 0)
598 return -1;
599 }
600
601 return mask;
602 }
603
604 static int collect_merge_info(struct merge_options *opt,
605 struct tree *merge_base,
606 struct tree *side1,
607 struct tree *side2)
608 {
609 int ret;
610 struct tree_desc t[3];
611 struct traverse_info info;
612 const char *toplevel_dir_placeholder = "";
613
614 opt->priv->current_dir_name = toplevel_dir_placeholder;
615 setup_traverse_info(&info, toplevel_dir_placeholder);
616 info.fn = collect_merge_info_callback;
617 info.data = opt;
618 info.show_all_errors = 1;
619
620 parse_tree(merge_base);
621 parse_tree(side1);
622 parse_tree(side2);
623 init_tree_desc(t + 0, merge_base->buffer, merge_base->size);
624 init_tree_desc(t + 1, side1->buffer, side1->size);
625 init_tree_desc(t + 2, side2->buffer, side2->size);
626
627 ret = traverse_trees(NULL, 3, t, &info);
628
629 return ret;
630 }
631
632 /*** Function Grouping: functions related to threeway content merges ***/
633
634 static int handle_content_merge(struct merge_options *opt,
635 const char *path,
636 const struct version_info *o,
637 const struct version_info *a,
638 const struct version_info *b,
639 const char *pathnames[3],
640 const int extra_marker_size,
641 struct version_info *result)
642 {
643 int clean = 0;
644 /*
645 * TODO: Needs a two-way or three-way content merge, but we're
646 * just being lazy and copying the version from HEAD and
647 * leaving it as conflicted.
648 */
649 result->mode = a->mode;
650 oidcpy(&result->oid, &a->oid);
651 return clean;
652 }
653
654 /*** Function Grouping: functions related to detect_and_process_renames(), ***
655 *** which are split into directory and regular rename detection sections. ***/
656
657 /*** Function Grouping: functions related to directory rename detection ***/
658
659 /*** Function Grouping: functions related to regular rename detection ***/
660
661 static int detect_and_process_renames(struct merge_options *opt,
662 struct tree *merge_base,
663 struct tree *side1,
664 struct tree *side2)
665 {
666 int clean = 1;
667
668 /*
669 * Rename detection works by detecting file similarity. Here we use
670 * a really easy-to-implement scheme: files are similar IFF they have
671 * the same filename. Therefore, by this scheme, there are no renames.
672 *
673 * TODO: Actually implement a real rename detection scheme.
674 */
675 return clean;
676 }
677
678 /*** Function Grouping: functions related to process_entries() ***/
679
680 static int string_list_df_name_compare(const char *one, const char *two)
681 {
682 int onelen = strlen(one);
683 int twolen = strlen(two);
684 /*
685 * Here we only care that entries for D/F conflicts are
686 * adjacent, in particular with the file of the D/F conflict
687 * appearing before files below the corresponding directory.
688 * The order of the rest of the list is irrelevant for us.
689 *
690 * To achieve this, we sort with df_name_compare and provide
691 * the mode S_IFDIR so that D/F conflicts will sort correctly.
692 * We use the mode S_IFDIR for everything else for simplicity,
693 * since in other cases any changes in their order due to
694 * sorting cause no problems for us.
695 */
696 int cmp = df_name_compare(one, onelen, S_IFDIR,
697 two, twolen, S_IFDIR);
698 /*
699 * Now that 'foo' and 'foo/bar' compare equal, we have to make sure
700 * that 'foo' comes before 'foo/bar'.
701 */
702 if (cmp)
703 return cmp;
704 return onelen - twolen;
705 }
706
707 struct directory_versions {
708 /*
709 * versions: list of (basename -> version_info)
710 *
711 * The basenames are in reverse lexicographic order of full pathnames,
712 * as processed in process_entries(). This puts all entries within
713 * a directory together, and covers the directory itself after
714 * everything within it, allowing us to write subtrees before needing
715 * to record information for the tree itself.
716 */
717 struct string_list versions;
718
719 /*
720 * offsets: list of (full relative path directories -> integer offsets)
721 *
722 * Since versions contains basenames from files in multiple different
723 * directories, we need to know which entries in versions correspond
724 * to which directories. Values of e.g.
725 * "" 0
726 * src 2
727 * src/moduleA 5
728 * Would mean that entries 0-1 of versions are files in the toplevel
729 * directory, entries 2-4 are files under src/, and the remaining
730 * entries starting at index 5 are files under src/moduleA/.
731 */
732 struct string_list offsets;
733
734 /*
735 * last_directory: directory that previously processed file found in
736 *
737 * last_directory starts NULL, but records the directory in which the
738 * previous file was found within. As soon as
739 * directory(current_file) != last_directory
740 * then we need to start updating accounting in versions & offsets.
741 * Note that last_directory is always the last path in "offsets" (or
742 * NULL if "offsets" is empty) so this exists just for quick access.
743 */
744 const char *last_directory;
745
746 /* last_directory_len: cached computation of strlen(last_directory) */
747 unsigned last_directory_len;
748 };
749
750 static int tree_entry_order(const void *a_, const void *b_)
751 {
752 const struct string_list_item *a = a_;
753 const struct string_list_item *b = b_;
754
755 const struct merged_info *ami = a->util;
756 const struct merged_info *bmi = b->util;
757 return base_name_compare(a->string, strlen(a->string), ami->result.mode,
758 b->string, strlen(b->string), bmi->result.mode);
759 }
760
761 static void write_tree(struct object_id *result_oid,
762 struct string_list *versions,
763 unsigned int offset,
764 size_t hash_size)
765 {
766 size_t maxlen = 0, extra;
767 unsigned int nr = versions->nr - offset;
768 struct strbuf buf = STRBUF_INIT;
769 struct string_list relevant_entries = STRING_LIST_INIT_NODUP;
770 int i;
771
772 /*
773 * We want to sort the last (versions->nr-offset) entries in versions.
774 * Do so by abusing the string_list API a bit: make another string_list
775 * that contains just those entries and then sort them.
776 *
777 * We won't use relevant_entries again and will let it just pop off the
778 * stack, so there won't be allocation worries or anything.
779 */
780 relevant_entries.items = versions->items + offset;
781 relevant_entries.nr = versions->nr - offset;
782 QSORT(relevant_entries.items, relevant_entries.nr, tree_entry_order);
783
784 /* Pre-allocate some space in buf */
785 extra = hash_size + 8; /* 8: 6 for mode, 1 for space, 1 for NUL char */
786 for (i = 0; i < nr; i++) {
787 maxlen += strlen(versions->items[offset+i].string) + extra;
788 }
789 strbuf_grow(&buf, maxlen);
790
791 /* Write each entry out to buf */
792 for (i = 0; i < nr; i++) {
793 struct merged_info *mi = versions->items[offset+i].util;
794 struct version_info *ri = &mi->result;
795 strbuf_addf(&buf, "%o %s%c",
796 ri->mode,
797 versions->items[offset+i].string, '\0');
798 strbuf_add(&buf, ri->oid.hash, hash_size);
799 }
800
801 /* Write this object file out, and record in result_oid */
802 write_object_file(buf.buf, buf.len, tree_type, result_oid);
803 strbuf_release(&buf);
804 }
805
806 static void record_entry_for_tree(struct directory_versions *dir_metadata,
807 const char *path,
808 struct merged_info *mi)
809 {
810 const char *basename;
811
812 if (mi->is_null)
813 /* nothing to record */
814 return;
815
816 basename = path + mi->basename_offset;
817 assert(strchr(basename, '/') == NULL);
818 string_list_append(&dir_metadata->versions,
819 basename)->util = &mi->result;
820 }
821
822 static void write_completed_directory(struct merge_options *opt,
823 const char *new_directory_name,
824 struct directory_versions *info)
825 {
826 const char *prev_dir;
827 struct merged_info *dir_info = NULL;
828 unsigned int offset;
829
830 /*
831 * Some explanation of info->versions and info->offsets...
832 *
833 * process_entries() iterates over all relevant files AND
834 * directories in reverse lexicographic order, and calls this
835 * function. Thus, an example of the paths that process_entries()
836 * could operate on (along with the directories for those paths
837 * being shown) is:
838 *
839 * xtract.c ""
840 * tokens.txt ""
841 * src/moduleB/umm.c src/moduleB
842 * src/moduleB/stuff.h src/moduleB
843 * src/moduleB/baz.c src/moduleB
844 * src/moduleB src
845 * src/moduleA/foo.c src/moduleA
846 * src/moduleA/bar.c src/moduleA
847 * src/moduleA src
848 * src ""
849 * Makefile ""
850 *
851 * info->versions:
852 *
853 * always contains the unprocessed entries and their
854 * version_info information. For example, after the first five
855 * entries above, info->versions would be:
856 *
857 * xtract.c <xtract.c's version_info>
858 * token.txt <token.txt's version_info>
859 * umm.c <src/moduleB/umm.c's version_info>
860 * stuff.h <src/moduleB/stuff.h's version_info>
861 * baz.c <src/moduleB/baz.c's version_info>
862 *
863 * Once a subdirectory is completed we remove the entries in
864 * that subdirectory from info->versions, writing it as a tree
865 * (write_tree()). Thus, as soon as we get to src/moduleB,
866 * info->versions would be updated to
867 *
868 * xtract.c <xtract.c's version_info>
869 * token.txt <token.txt's version_info>
870 * moduleB <src/moduleB's version_info>
871 *
872 * info->offsets:
873 *
874 * helps us track which entries in info->versions correspond to
875 * which directories. When we are N directories deep (e.g. 4
876 * for src/modA/submod/subdir/), we have up to N+1 unprocessed
877 * directories (+1 because of toplevel dir). Corresponding to
878 * the info->versions example above, after processing five entries
879 * info->offsets will be:
880 *
881 * "" 0
882 * src/moduleB 2
883 *
884 * which is used to know that xtract.c & token.txt are from the
885 * toplevel dirctory, while umm.c & stuff.h & baz.c are from the
886 * src/moduleB directory. Again, following the example above,
887 * once we need to process src/moduleB, then info->offsets is
888 * updated to
889 *
890 * "" 0
891 * src 2
892 *
893 * which says that moduleB (and only moduleB so far) is in the
894 * src directory.
895 *
896 * One unique thing to note about info->offsets here is that
897 * "src" was not added to info->offsets until there was a path
898 * (a file OR directory) immediately below src/ that got
899 * processed.
900 *
901 * Since process_entry() just appends new entries to info->versions,
902 * write_completed_directory() only needs to do work if the next path
903 * is in a directory that is different than the last directory found
904 * in info->offsets.
905 */
906
907 /*
908 * If we are working with the same directory as the last entry, there
909 * is no work to do. (See comments above the directory_name member of
910 * struct merged_info for why we can use pointer comparison instead of
911 * strcmp here.)
912 */
913 if (new_directory_name == info->last_directory)
914 return;
915
916 /*
917 * If we are just starting (last_directory is NULL), or last_directory
918 * is a prefix of the current directory, then we can just update
919 * info->offsets to record the offset where we started this directory
920 * and update last_directory to have quick access to it.
921 */
922 if (info->last_directory == NULL ||
923 !strncmp(new_directory_name, info->last_directory,
924 info->last_directory_len)) {
925 uintptr_t offset = info->versions.nr;
926
927 info->last_directory = new_directory_name;
928 info->last_directory_len = strlen(info->last_directory);
929 /*
930 * Record the offset into info->versions where we will
931 * start recording basenames of paths found within
932 * new_directory_name.
933 */
934 string_list_append(&info->offsets,
935 info->last_directory)->util = (void*)offset;
936 return;
937 }
938
939 /*
940 * The next entry that will be processed will be within
941 * new_directory_name. Since at this point we know that
942 * new_directory_name is within a different directory than
943 * info->last_directory, we have all entries for info->last_directory
944 * in info->versions and we need to create a tree object for them.
945 */
946 dir_info = strmap_get(&opt->priv->paths, info->last_directory);
947 assert(dir_info);
948 offset = (uintptr_t)info->offsets.items[info->offsets.nr-1].util;
949 if (offset == info->versions.nr) {
950 /*
951 * Actually, we don't need to create a tree object in this
952 * case. Whenever all files within a directory disappear
953 * during the merge (e.g. unmodified on one side and
954 * deleted on the other, or files were renamed elsewhere),
955 * then we get here and the directory itself needs to be
956 * omitted from its parent tree as well.
957 */
958 dir_info->is_null = 1;
959 } else {
960 /*
961 * Write out the tree to the git object directory, and also
962 * record the mode and oid in dir_info->result.
963 */
964 dir_info->is_null = 0;
965 dir_info->result.mode = S_IFDIR;
966 write_tree(&dir_info->result.oid, &info->versions, offset,
967 opt->repo->hash_algo->rawsz);
968 }
969
970 /*
971 * We've now used several entries from info->versions and one entry
972 * from info->offsets, so we get rid of those values.
973 */
974 info->offsets.nr--;
975 info->versions.nr = offset;
976
977 /*
978 * Now we've taken care of the completed directory, but we need to
979 * prepare things since future entries will be in
980 * new_directory_name. (In particular, process_entry() will be
981 * appending new entries to info->versions.) So, we need to make
982 * sure new_directory_name is the last entry in info->offsets.
983 */
984 prev_dir = info->offsets.nr == 0 ? NULL :
985 info->offsets.items[info->offsets.nr-1].string;
986 if (new_directory_name != prev_dir) {
987 uintptr_t c = info->versions.nr;
988 string_list_append(&info->offsets,
989 new_directory_name)->util = (void*)c;
990 }
991
992 /* And, of course, we need to update last_directory to match. */
993 info->last_directory = new_directory_name;
994 info->last_directory_len = strlen(info->last_directory);
995 }
996
997 /* Per entry merge function */
998 static void process_entry(struct merge_options *opt,
999 const char *path,
1000 struct conflict_info *ci,
1001 struct directory_versions *dir_metadata)
1002 {
1003 int df_file_index = 0;
1004
1005 VERIFY_CI(ci);
1006 assert(ci->filemask >= 0 && ci->filemask <= 7);
1007 /* ci->match_mask == 7 was handled in collect_merge_info_callback() */
1008 assert(ci->match_mask == 0 || ci->match_mask == 3 ||
1009 ci->match_mask == 5 || ci->match_mask == 6);
1010
1011 if (ci->dirmask) {
1012 record_entry_for_tree(dir_metadata, path, &ci->merged);
1013 if (ci->filemask == 0)
1014 /* nothing else to handle */
1015 return;
1016 assert(ci->df_conflict);
1017 }
1018
1019 if (ci->df_conflict && ci->merged.result.mode == 0) {
1020 int i;
1021
1022 /*
1023 * directory no longer in the way, but we do have a file we
1024 * need to place here so we need to clean away the "directory
1025 * merges to nothing" result.
1026 */
1027 ci->df_conflict = 0;
1028 assert(ci->filemask != 0);
1029 ci->merged.clean = 0;
1030 ci->merged.is_null = 0;
1031 /* and we want to zero out any directory-related entries */
1032 ci->match_mask = (ci->match_mask & ~ci->dirmask);
1033 ci->dirmask = 0;
1034 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++) {
1035 if (ci->filemask & (1 << i))
1036 continue;
1037 ci->stages[i].mode = 0;
1038 oidcpy(&ci->stages[i].oid, &null_oid);
1039 }
1040 } else if (ci->df_conflict && ci->merged.result.mode != 0) {
1041 /*
1042 * This started out as a D/F conflict, and the entries in
1043 * the competing directory were not removed by the merge as
1044 * evidenced by write_completed_directory() writing a value
1045 * to ci->merged.result.mode.
1046 */
1047 struct conflict_info *new_ci;
1048 const char *branch;
1049 const char *old_path = path;
1050 int i;
1051
1052 assert(ci->merged.result.mode == S_IFDIR);
1053
1054 /*
1055 * If filemask is 1, we can just ignore the file as having
1056 * been deleted on both sides. We do not want to overwrite
1057 * ci->merged.result, since it stores the tree for all the
1058 * files under it.
1059 */
1060 if (ci->filemask == 1) {
1061 ci->filemask = 0;
1062 return;
1063 }
1064
1065 /*
1066 * This file still exists on at least one side, and we want
1067 * the directory to remain here, so we need to move this
1068 * path to some new location.
1069 */
1070 new_ci = xcalloc(1, sizeof(*new_ci));
1071 /* We don't really want new_ci->merged.result copied, but it'll
1072 * be overwritten below so it doesn't matter. We also don't
1073 * want any directory mode/oid values copied, but we'll zero
1074 * those out immediately. We do want the rest of ci copied.
1075 */
1076 memcpy(new_ci, ci, sizeof(*ci));
1077 new_ci->match_mask = (new_ci->match_mask & ~new_ci->dirmask);
1078 new_ci->dirmask = 0;
1079 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++) {
1080 if (new_ci->filemask & (1 << i))
1081 continue;
1082 /* zero out any entries related to directories */
1083 new_ci->stages[i].mode = 0;
1084 oidcpy(&new_ci->stages[i].oid, &null_oid);
1085 }
1086
1087 /*
1088 * Find out which side this file came from; note that we
1089 * cannot just use ci->filemask, because renames could cause
1090 * the filemask to go back to 7. So we use dirmask, then
1091 * pick the opposite side's index.
1092 */
1093 df_file_index = (ci->dirmask & (1 << 1)) ? 2 : 1;
1094 branch = (df_file_index == 1) ? opt->branch1 : opt->branch2;
1095 path = unique_path(&opt->priv->paths, path, branch);
1096 strmap_put(&opt->priv->paths, path, new_ci);
1097
1098 path_msg(opt, path, 0,
1099 _("CONFLICT (file/directory): directory in the way "
1100 "of %s from %s; moving it to %s instead."),
1101 old_path, branch, path);
1102
1103 /*
1104 * Zero out the filemask for the old ci. At this point, ci
1105 * was just an entry for a directory, so we don't need to
1106 * do anything more with it.
1107 */
1108 ci->filemask = 0;
1109
1110 /*
1111 * Now note that we're working on the new entry (path was
1112 * updated above.
1113 */
1114 ci = new_ci;
1115 }
1116
1117 /*
1118 * NOTE: Below there is a long switch-like if-elseif-elseif... block
1119 * which the code goes through even for the df_conflict cases
1120 * above.
1121 */
1122 if (ci->match_mask) {
1123 ci->merged.clean = 1;
1124 if (ci->match_mask == 6) {
1125 /* stages[1] == stages[2] */
1126 ci->merged.result.mode = ci->stages[1].mode;
1127 oidcpy(&ci->merged.result.oid, &ci->stages[1].oid);
1128 } else {
1129 /* determine the mask of the side that didn't match */
1130 unsigned int othermask = 7 & ~ci->match_mask;
1131 int side = (othermask == 4) ? 2 : 1;
1132
1133 ci->merged.result.mode = ci->stages[side].mode;
1134 ci->merged.is_null = !ci->merged.result.mode;
1135 oidcpy(&ci->merged.result.oid, &ci->stages[side].oid);
1136
1137 assert(othermask == 2 || othermask == 4);
1138 assert(ci->merged.is_null ==
1139 (ci->filemask == ci->match_mask));
1140 }
1141 } else if (ci->filemask >= 6 &&
1142 (S_IFMT & ci->stages[1].mode) !=
1143 (S_IFMT & ci->stages[2].mode)) {
1144 /*
1145 * Two different items from (file/submodule/symlink)
1146 */
1147 die("Not yet implemented.");
1148 } else if (ci->filemask >= 6) {
1149 /* Need a two-way or three-way content merge */
1150 struct version_info merged_file;
1151 unsigned clean_merge;
1152 struct version_info *o = &ci->stages[0];
1153 struct version_info *a = &ci->stages[1];
1154 struct version_info *b = &ci->stages[2];
1155
1156 clean_merge = handle_content_merge(opt, path, o, a, b,
1157 ci->pathnames,
1158 opt->priv->call_depth * 2,
1159 &merged_file);
1160 ci->merged.clean = clean_merge &&
1161 !ci->df_conflict && !ci->path_conflict;
1162 ci->merged.result.mode = merged_file.mode;
1163 ci->merged.is_null = (merged_file.mode == 0);
1164 oidcpy(&ci->merged.result.oid, &merged_file.oid);
1165 if (clean_merge && ci->df_conflict) {
1166 assert(df_file_index == 1 || df_file_index == 2);
1167 ci->filemask = 1 << df_file_index;
1168 ci->stages[df_file_index].mode = merged_file.mode;
1169 oidcpy(&ci->stages[df_file_index].oid, &merged_file.oid);
1170 }
1171 if (!clean_merge) {
1172 const char *reason = _("content");
1173 if (ci->filemask == 6)
1174 reason = _("add/add");
1175 if (S_ISGITLINK(merged_file.mode))
1176 reason = _("submodule");
1177 path_msg(opt, path, 0,
1178 _("CONFLICT (%s): Merge conflict in %s"),
1179 reason, path);
1180 }
1181 } else if (ci->filemask == 3 || ci->filemask == 5) {
1182 /* Modify/delete */
1183 const char *modify_branch, *delete_branch;
1184 int side = (ci->filemask == 5) ? 2 : 1;
1185 int index = opt->priv->call_depth ? 0 : side;
1186
1187 ci->merged.result.mode = ci->stages[index].mode;
1188 oidcpy(&ci->merged.result.oid, &ci->stages[index].oid);
1189 ci->merged.clean = 0;
1190
1191 modify_branch = (side == 1) ? opt->branch1 : opt->branch2;
1192 delete_branch = (side == 1) ? opt->branch2 : opt->branch1;
1193
1194 path_msg(opt, path, 0,
1195 _("CONFLICT (modify/delete): %s deleted in %s "
1196 "and modified in %s. Version %s of %s left "
1197 "in tree."),
1198 path, delete_branch, modify_branch,
1199 modify_branch, path);
1200 } else if (ci->filemask == 2 || ci->filemask == 4) {
1201 /* Added on one side */
1202 int side = (ci->filemask == 4) ? 2 : 1;
1203 ci->merged.result.mode = ci->stages[side].mode;
1204 oidcpy(&ci->merged.result.oid, &ci->stages[side].oid);
1205 ci->merged.clean = !ci->df_conflict;
1206 } else if (ci->filemask == 1) {
1207 /* Deleted on both sides */
1208 ci->merged.is_null = 1;
1209 ci->merged.result.mode = 0;
1210 oidcpy(&ci->merged.result.oid, &null_oid);
1211 ci->merged.clean = 1;
1212 }
1213
1214 /*
1215 * If still conflicted, record it separately. This allows us to later
1216 * iterate over just conflicted entries when updating the index instead
1217 * of iterating over all entries.
1218 */
1219 if (!ci->merged.clean)
1220 strmap_put(&opt->priv->conflicted, path, ci);
1221 record_entry_for_tree(dir_metadata, path, &ci->merged);
1222 }
1223
1224 static void process_entries(struct merge_options *opt,
1225 struct object_id *result_oid)
1226 {
1227 struct hashmap_iter iter;
1228 struct strmap_entry *e;
1229 struct string_list plist = STRING_LIST_INIT_NODUP;
1230 struct string_list_item *entry;
1231 struct directory_versions dir_metadata = { STRING_LIST_INIT_NODUP,
1232 STRING_LIST_INIT_NODUP,
1233 NULL, 0 };
1234
1235 if (strmap_empty(&opt->priv->paths)) {
1236 oidcpy(result_oid, opt->repo->hash_algo->empty_tree);
1237 return;
1238 }
1239
1240 /* Hack to pre-allocate plist to the desired size */
1241 ALLOC_GROW(plist.items, strmap_get_size(&opt->priv->paths), plist.alloc);
1242
1243 /* Put every entry from paths into plist, then sort */
1244 strmap_for_each_entry(&opt->priv->paths, &iter, e) {
1245 string_list_append(&plist, e->key)->util = e->value;
1246 }
1247 plist.cmp = string_list_df_name_compare;
1248 string_list_sort(&plist);
1249
1250 /*
1251 * Iterate over the items in reverse order, so we can handle paths
1252 * below a directory before needing to handle the directory itself.
1253 *
1254 * This allows us to write subtrees before we need to write trees,
1255 * and it also enables sane handling of directory/file conflicts
1256 * (because it allows us to know whether the directory is still in
1257 * the way when it is time to process the file at the same path).
1258 */
1259 for (entry = &plist.items[plist.nr-1]; entry >= plist.items; --entry) {
1260 char *path = entry->string;
1261 /*
1262 * NOTE: mi may actually be a pointer to a conflict_info, but
1263 * we have to check mi->clean first to see if it's safe to
1264 * reassign to such a pointer type.
1265 */
1266 struct merged_info *mi = entry->util;
1267
1268 write_completed_directory(opt, mi->directory_name,
1269 &dir_metadata);
1270 if (mi->clean)
1271 record_entry_for_tree(&dir_metadata, path, mi);
1272 else {
1273 struct conflict_info *ci = (struct conflict_info *)mi;
1274 process_entry(opt, path, ci, &dir_metadata);
1275 }
1276 }
1277
1278 if (dir_metadata.offsets.nr != 1 ||
1279 (uintptr_t)dir_metadata.offsets.items[0].util != 0) {
1280 printf("dir_metadata.offsets.nr = %d (should be 1)\n",
1281 dir_metadata.offsets.nr);
1282 printf("dir_metadata.offsets.items[0].util = %u (should be 0)\n",
1283 (unsigned)(uintptr_t)dir_metadata.offsets.items[0].util);
1284 fflush(stdout);
1285 BUG("dir_metadata accounting completely off; shouldn't happen");
1286 }
1287 write_tree(result_oid, &dir_metadata.versions, 0,
1288 opt->repo->hash_algo->rawsz);
1289 string_list_clear(&plist, 0);
1290 string_list_clear(&dir_metadata.versions, 0);
1291 string_list_clear(&dir_metadata.offsets, 0);
1292 }
1293
1294 /*** Function Grouping: functions related to merge_switch_to_result() ***/
1295
1296 static int checkout(struct merge_options *opt,
1297 struct tree *prev,
1298 struct tree *next)
1299 {
1300 /* Switch the index/working copy from old to new */
1301 int ret;
1302 struct tree_desc trees[2];
1303 struct unpack_trees_options unpack_opts;
1304
1305 memset(&unpack_opts, 0, sizeof(unpack_opts));
1306 unpack_opts.head_idx = -1;
1307 unpack_opts.src_index = opt->repo->index;
1308 unpack_opts.dst_index = opt->repo->index;
1309
1310 setup_unpack_trees_porcelain(&unpack_opts, "merge");
1311
1312 /*
1313 * NOTE: if this were just "git checkout" code, we would probably
1314 * read or refresh the cache and check for a conflicted index, but
1315 * builtin/merge.c or sequencer.c really needs to read the index
1316 * and check for conflicted entries before starting merging for a
1317 * good user experience (no sense waiting for merges/rebases before
1318 * erroring out), so there's no reason to duplicate that work here.
1319 */
1320
1321 /* 2-way merge to the new branch */
1322 unpack_opts.update = 1;
1323 unpack_opts.merge = 1;
1324 unpack_opts.quiet = 0; /* FIXME: sequencer might want quiet? */
1325 unpack_opts.verbose_update = (opt->verbosity > 2);
1326 unpack_opts.fn = twoway_merge;
1327 if (1/* FIXME: opts->overwrite_ignore*/) {
1328 unpack_opts.dir = xcalloc(1, sizeof(*unpack_opts.dir));
1329 unpack_opts.dir->flags |= DIR_SHOW_IGNORED;
1330 setup_standard_excludes(unpack_opts.dir);
1331 }
1332 parse_tree(prev);
1333 init_tree_desc(&trees[0], prev->buffer, prev->size);
1334 parse_tree(next);
1335 init_tree_desc(&trees[1], next->buffer, next->size);
1336
1337 ret = unpack_trees(2, trees, &unpack_opts);
1338 clear_unpack_trees_porcelain(&unpack_opts);
1339 dir_clear(unpack_opts.dir);
1340 FREE_AND_NULL(unpack_opts.dir);
1341 return ret;
1342 }
1343
1344 static int record_conflicted_index_entries(struct merge_options *opt,
1345 struct index_state *index,
1346 struct strmap *paths,
1347 struct strmap *conflicted)
1348 {
1349 struct hashmap_iter iter;
1350 struct strmap_entry *e;
1351 int errs = 0;
1352 int original_cache_nr;
1353
1354 if (strmap_empty(conflicted))
1355 return 0;
1356
1357 original_cache_nr = index->cache_nr;
1358
1359 /* Put every entry from paths into plist, then sort */
1360 strmap_for_each_entry(conflicted, &iter, e) {
1361 const char *path = e->key;
1362 struct conflict_info *ci = e->value;
1363 int pos;
1364 struct cache_entry *ce;
1365 int i;
1366
1367 VERIFY_CI(ci);
1368
1369 /*
1370 * The index will already have a stage=0 entry for this path,
1371 * because we created an as-merged-as-possible version of the
1372 * file and checkout() moved the working copy and index over
1373 * to that version.
1374 *
1375 * However, previous iterations through this loop will have
1376 * added unstaged entries to the end of the cache which
1377 * ignore the standard alphabetical ordering of cache
1378 * entries and break invariants needed for index_name_pos()
1379 * to work. However, we know the entry we want is before
1380 * those appended cache entries, so do a temporary swap on
1381 * cache_nr to only look through entries of interest.
1382 */
1383 SWAP(index->cache_nr, original_cache_nr);
1384 pos = index_name_pos(index, path, strlen(path));
1385 SWAP(index->cache_nr, original_cache_nr);
1386 if (pos < 0) {
1387 if (ci->filemask != 1)
1388 BUG("Conflicted %s but nothing in basic working tree or index; this shouldn't happen", path);
1389 cache_tree_invalidate_path(index, path);
1390 } else {
1391 ce = index->cache[pos];
1392
1393 /*
1394 * Clean paths with CE_SKIP_WORKTREE set will not be
1395 * written to the working tree by the unpack_trees()
1396 * call in checkout(). Our conflicted entries would
1397 * have appeared clean to that code since we ignored
1398 * the higher order stages. Thus, we need override
1399 * the CE_SKIP_WORKTREE bit and manually write those
1400 * files to the working disk here.
1401 *
1402 * TODO: Implement this CE_SKIP_WORKTREE fixup.
1403 */
1404
1405 /*
1406 * Mark this cache entry for removal and instead add
1407 * new stage>0 entries corresponding to the
1408 * conflicts. If there are many conflicted entries, we
1409 * want to avoid memmove'ing O(NM) entries by
1410 * inserting the new entries one at a time. So,
1411 * instead, we just add the new cache entries to the
1412 * end (ignoring normal index requirements on sort
1413 * order) and sort the index once we're all done.
1414 */
1415 ce->ce_flags |= CE_REMOVE;
1416 }
1417
1418 for (i = MERGE_BASE; i <= MERGE_SIDE2; i++) {
1419 struct version_info *vi;
1420 if (!(ci->filemask & (1ul << i)))
1421 continue;
1422 vi = &ci->stages[i];
1423 ce = make_cache_entry(index, vi->mode, &vi->oid,
1424 path, i+1, 0);
1425 add_index_entry(index, ce, ADD_CACHE_JUST_APPEND);
1426 }
1427 }
1428
1429 /*
1430 * Remove the unused cache entries (and invalidate the relevant
1431 * cache-trees), then sort the index entries to get the conflicted
1432 * entries we added to the end into their right locations.
1433 */
1434 remove_marked_cache_entries(index, 1);
1435 QSORT(index->cache, index->cache_nr, cmp_cache_name_compare);
1436
1437 return errs;
1438 }
1439
1440 void merge_switch_to_result(struct merge_options *opt,
1441 struct tree *head,
1442 struct merge_result *result,
1443 int update_worktree_and_index,
1444 int display_update_msgs)
1445 {
1446 assert(opt->priv == NULL);
1447 if (result->clean >= 0 && update_worktree_and_index) {
1448 struct merge_options_internal *opti = result->priv;
1449
1450 if (checkout(opt, head, result->tree)) {
1451 /* failure to function */
1452 result->clean = -1;
1453 return;
1454 }
1455
1456 if (record_conflicted_index_entries(opt, opt->repo->index,
1457 &opti->paths,
1458 &opti->conflicted)) {
1459 /* failure to function */
1460 result->clean = -1;
1461 return;
1462 }
1463 }
1464
1465 if (display_update_msgs) {
1466 struct merge_options_internal *opti = result->priv;
1467 struct hashmap_iter iter;
1468 struct strmap_entry *e;
1469 struct string_list olist = STRING_LIST_INIT_NODUP;
1470 int i;
1471
1472 /* Hack to pre-allocate olist to the desired size */
1473 ALLOC_GROW(olist.items, strmap_get_size(&opti->output),
1474 olist.alloc);
1475
1476 /* Put every entry from output into olist, then sort */
1477 strmap_for_each_entry(&opti->output, &iter, e) {
1478 string_list_append(&olist, e->key)->util = e->value;
1479 }
1480 string_list_sort(&olist);
1481
1482 /* Iterate over the items, printing them */
1483 for (i = 0; i < olist.nr; ++i) {
1484 struct strbuf *sb = olist.items[i].util;
1485
1486 printf("%s", sb->buf);
1487 }
1488 string_list_clear(&olist, 0);
1489 }
1490
1491 merge_finalize(opt, result);
1492 }
1493
1494 void merge_finalize(struct merge_options *opt,
1495 struct merge_result *result)
1496 {
1497 struct merge_options_internal *opti = result->priv;
1498
1499 assert(opt->priv == NULL);
1500
1501 clear_internal_opts(opti, 0);
1502 FREE_AND_NULL(opti);
1503 }
1504
1505 /*** Function Grouping: helper functions for merge_incore_*() ***/
1506
1507 static void merge_start(struct merge_options *opt, struct merge_result *result)
1508 {
1509 /* Sanity checks on opt */
1510 assert(opt->repo);
1511
1512 assert(opt->branch1 && opt->branch2);
1513
1514 assert(opt->detect_directory_renames >= MERGE_DIRECTORY_RENAMES_NONE &&
1515 opt->detect_directory_renames <= MERGE_DIRECTORY_RENAMES_TRUE);
1516 assert(opt->rename_limit >= -1);
1517 assert(opt->rename_score >= 0 && opt->rename_score <= MAX_SCORE);
1518 assert(opt->show_rename_progress >= 0 && opt->show_rename_progress <= 1);
1519
1520 assert(opt->xdl_opts >= 0);
1521 assert(opt->recursive_variant >= MERGE_VARIANT_NORMAL &&
1522 opt->recursive_variant <= MERGE_VARIANT_THEIRS);
1523
1524 /*
1525 * detect_renames, verbosity, buffer_output, and obuf are ignored
1526 * fields that were used by "recursive" rather than "ort" -- but
1527 * sanity check them anyway.
1528 */
1529 assert(opt->detect_renames >= -1 &&
1530 opt->detect_renames <= DIFF_DETECT_COPY);
1531 assert(opt->verbosity >= 0 && opt->verbosity <= 5);
1532 assert(opt->buffer_output <= 2);
1533 assert(opt->obuf.len == 0);
1534
1535 assert(opt->priv == NULL);
1536
1537 /* Default to histogram diff. Actually, just hardcode it...for now. */
1538 opt->xdl_opts = DIFF_WITH_ALG(opt, HISTOGRAM_DIFF);
1539
1540 /* Initialization of opt->priv, our internal merge data */
1541 opt->priv = xcalloc(1, sizeof(*opt->priv));
1542
1543 /*
1544 * Although we initialize opt->priv->paths with strdup_strings=0,
1545 * that's just to avoid making yet another copy of an allocated
1546 * string. Putting the entry into paths means we are taking
1547 * ownership, so we will later free it. paths_to_free is similar.
1548 *
1549 * In contrast, conflicted just has a subset of keys from paths, so
1550 * we don't want to free those (it'd be a duplicate free).
1551 */
1552 strmap_init_with_options(&opt->priv->paths, NULL, 0);
1553 strmap_init_with_options(&opt->priv->conflicted, NULL, 0);
1554 string_list_init(&opt->priv->paths_to_free, 0);
1555
1556 /*
1557 * keys & strbufs in output will sometimes need to outlive "paths",
1558 * so it will have a copy of relevant keys. It's probably a small
1559 * subset of the overall paths that have special output.
1560 */
1561 strmap_init(&opt->priv->output);
1562 }
1563
1564 /*** Function Grouping: merge_incore_*() and their internal variants ***/
1565
1566 /*
1567 * Originally from merge_trees_internal(); heavily adapted, though.
1568 */
1569 static void merge_ort_nonrecursive_internal(struct merge_options *opt,
1570 struct tree *merge_base,
1571 struct tree *side1,
1572 struct tree *side2,
1573 struct merge_result *result)
1574 {
1575 struct object_id working_tree_oid;
1576
1577 if (collect_merge_info(opt, merge_base, side1, side2) != 0) {
1578 /*
1579 * TRANSLATORS: The %s arguments are: 1) tree hash of a merge
1580 * base, and 2-3) the trees for the two trees we're merging.
1581 */
1582 err(opt, _("collecting merge info failed for trees %s, %s, %s"),
1583 oid_to_hex(&merge_base->object.oid),
1584 oid_to_hex(&side1->object.oid),
1585 oid_to_hex(&side2->object.oid));
1586 result->clean = -1;
1587 return;
1588 }
1589
1590 result->clean = detect_and_process_renames(opt, merge_base,
1591 side1, side2);
1592 process_entries(opt, &working_tree_oid);
1593
1594 /* Set return values */
1595 result->tree = parse_tree_indirect(&working_tree_oid);
1596 /* existence of conflicted entries implies unclean */
1597 result->clean &= strmap_empty(&opt->priv->conflicted);
1598 if (!opt->priv->call_depth) {
1599 result->priv = opt->priv;
1600 opt->priv = NULL;
1601 }
1602 }
1603
1604 void merge_incore_nonrecursive(struct merge_options *opt,
1605 struct tree *merge_base,
1606 struct tree *side1,
1607 struct tree *side2,
1608 struct merge_result *result)
1609 {
1610 assert(opt->ancestor != NULL);
1611 merge_start(opt, result);
1612 merge_ort_nonrecursive_internal(opt, merge_base, side1, side2, result);
1613 }
1614
1615 void merge_incore_recursive(struct merge_options *opt,
1616 struct commit_list *merge_bases,
1617 struct commit *side1,
1618 struct commit *side2,
1619 struct merge_result *result)
1620 {
1621 die("Not yet implemented");
1622 }