builtin-merge-recursive.c 44.2 KB
Newer Older
1 2 3 4 5 6 7 8 9
/*
 * Recursive Merge algorithm stolen from git-merge-recursive.py by
 * Fredrik Kuivinen.
 * The thieves were Alex Riesen and Johannes Schindelin, in June/July 2006
 */
#include "cache.h"
#include "cache-tree.h"
#include "commit.h"
#include "blob.h"
D
Daniel Barkalow 已提交
10
#include "builtin.h"
11 12 13 14 15
#include "tree-walk.h"
#include "diff.h"
#include "diffcore.h"
#include "run-command.h"
#include "tag.h"
16
#include "unpack-trees.h"
17
#include "path-list.h"
J
Junio C Hamano 已提交
18
#include "xdiff-interface.h"
19
#include "interpolate.h"
20
#include "attr.h"
D
Daniel Barkalow 已提交
21
#include "merge-recursive.h"
22

J
Junio C Hamano 已提交
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38
static int subtree_merge;

static struct tree *shift_tree_object(struct tree *one, struct tree *two)
{
	unsigned char shifted[20];

	/*
	 * NEEDSWORK: this limits the recursion depth to hardcoded
	 * value '2' to avoid excessive overhead.
	 */
	shift_tree(one->object.sha1, two->object.sha1, shifted, 2);
	if (!hashcmp(two->object.sha1, shifted))
		return two;
	return lookup_tree(shifted);
}

39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59
/*
 * A virtual commit has
 * - (const char *)commit->util set to the name, and
 * - *(int *)commit->object.sha1 set to the virtual id.
 */

static unsigned commit_list_count(const struct commit_list *l)
{
	unsigned c = 0;
	for (; l; l = l->next )
		c++;
	return c;
}

static struct commit *make_virtual_commit(struct tree *tree, const char *comment)
{
	struct commit *commit = xcalloc(1, sizeof(struct commit));
	static unsigned virtual_id = 1;
	commit->tree = tree;
	commit->util = (void*)comment;
	*(int*)commit->object.sha1 = virtual_id++;
60 61
	/* avoid warnings */
	commit->object.parsed = 1;
62 63 64 65
	return commit;
}

/*
66 67
 * Since we use get_tree_entry(), which does not put the read object into
 * the object pool, we cannot rely on a == b.
68 69 70
 */
static int sha_eq(const unsigned char *a, const unsigned char *b)
{
71
	if (!a && !b)
72
		return 2;
J
Junio C Hamano 已提交
73
	return a && b && hashcmp(a, b) == 0;
74 75 76
}

/*
77 78
 * Since we want to write the index eventually, we cannot reuse the index
 * for these (temporary) data.
79 80 81 82 83 84 85 86 87 88 89
 */
struct stage_data
{
	struct
	{
		unsigned mode;
		unsigned char sha[20];
	} stages[4];
	unsigned processed:1;
};

90 91
static struct path_list current_file_set = {NULL, 0, 0, 1};
static struct path_list current_directory_set = {NULL, 0, 0, 1};
92

93
static int call_depth = 0;
94
static int verbosity = 2;
95
static int rename_limit = -1;
96
static int buffer_output = 1;
97
static struct strbuf obuf = STRBUF_INIT;
98

99
static int show(int v)
100 101 102 103
{
	return (!call_depth && verbosity >= v) || verbosity >= 5;
}

104
static void flush_output(void)
105
{
106 107 108
	if (obuf.len) {
		fputs(obuf.buf, stdout);
		strbuf_reset(&obuf);
109
	}
110 111
}

112
static void output(int v, const char *fmt, ...)
113
{
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137
	int len;
	va_list ap;

	if (!show(v))
		return;

	strbuf_grow(&obuf, call_depth * 2 + 2);
	memset(obuf.buf + obuf.len, ' ', call_depth * 2);
	strbuf_setlen(&obuf, obuf.len + call_depth * 2);

	va_start(ap, fmt);
	len = vsnprintf(obuf.buf + obuf.len, strbuf_avail(&obuf), fmt, ap);
	va_end(ap);

	if (len < 0)
		len = 0;
	if (len >= strbuf_avail(&obuf)) {
		strbuf_grow(&obuf, len + 2);
		va_start(ap, fmt);
		len = vsnprintf(obuf.buf + obuf.len, strbuf_avail(&obuf), fmt, ap);
		va_end(ap);
		if (len >= strbuf_avail(&obuf)) {
			die("this should not happen, your snprintf is broken");
		}
138
	}
139 140 141 142
	strbuf_setlen(&obuf, obuf.len + len);
	strbuf_add(&obuf, "\n", 1);
	if (!buffer_output)
		flush_output();
143 144
}

145 146 147
static void output_commit_title(struct commit *commit)
{
	int i;
148
	flush_output();
149
	for (i = call_depth; i--;)
150 151 152 153
		fputs("  ", stdout);
	if (commit->util)
		printf("virtual %s\n", (char *)commit->util);
	else {
154
		printf("%s ", find_unique_abbrev(commit->object.sha1, DEFAULT_ABBREV));
155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171
		if (parse_commit(commit) != 0)
			printf("(bad commit)\n");
		else {
			const char *s;
			int len;
			for (s = commit->buffer; *s; s++)
				if (*s == '\n' && s[1] == '\n') {
					s += 2;
					break;
				}
			for (len = 0; s[len] && '\n' != s[len]; len++)
				; /* do nothing */
			printf("%.*s\n", len, s);
		}
	}
}

172 173 174 175 176 177
static int add_cacheinfo(unsigned int mode, const unsigned char *sha1,
		const char *path, int stage, int refresh, int options)
{
	struct cache_entry *ce;
	ce = make_cache_entry(mode, sha1 ? sha1 : null_sha1, path, stage, refresh);
	if (!ce)
178
		return error("addinfo_cache failed for path '%s'", path);
179 180 181 182 183 184 185 186 187 188 189 190 191 192
	return add_cache_entry(ce, options);
}

/*
 * This is a global variable which is used in a number of places but
 * only written to in the 'merge' function.
 *
 * index_only == 1    => Don't leave any non-stage 0 entries in the cache and
 *                       don't update the working directory.
 *               0    => Leave unmerged entries in the cache and update
 *                       the working directory.
 */
static int index_only = 0;

193 194 195 196 197 198
static void init_tree_desc_from_tree(struct tree_desc *desc, struct tree *tree)
{
	parse_tree(tree);
	init_tree_desc(desc, tree->buffer, tree->size);
}

199
static int git_merge_trees(int index_only,
200 201 202 203
			   struct tree *common,
			   struct tree *head,
			   struct tree *merge)
{
204
	int rc;
205
	struct tree_desc t[3];
206 207 208 209 210 211 212 213 214 215
	struct unpack_trees_options opts;

	memset(&opts, 0, sizeof(opts));
	if (index_only)
		opts.index_only = 1;
	else
		opts.update = 1;
	opts.merge = 1;
	opts.head_idx = 2;
	opts.fn = threeway_merge;
216
	opts.index = &the_index;
217

218 219 220
	init_tree_desc_from_tree(t+0, common);
	init_tree_desc_from_tree(t+1, head);
	init_tree_desc_from_tree(t+2, merge);
221

222
	rc = unpack_trees(3, t, &opts);
223 224
	cache_tree_free(&active_cache_tree);
	return rc;
225 226
}

D
Daniel Barkalow 已提交
227
struct tree *write_tree_from_memory(void)
228
{
229 230
	struct tree *result = NULL;

231
	if (unmerged_cache()) {
232 233 234 235 236 237 238
		int i;
		output(0, "There are unmerged index entries:");
		for (i = 0; i < active_nr; i++) {
			struct cache_entry *ce = active_cache[i];
			if (ce_stage(ce))
				output(0, "%d %.*s", ce_stage(ce), ce_namelen(ce), ce->name);
		}
239
		return NULL;
240
	}
241 242 243 244 245

	if (!active_cache_tree)
		active_cache_tree = cache_tree();

	if (!cache_tree_fully_valid(active_cache_tree) &&
246 247
	    cache_tree_update(active_cache_tree,
			      active_cache, active_nr, 0, 0) < 0)
248 249 250 251 252
		die("error building trees");

	result = lookup_tree(active_cache_tree->sha1);

	return result;
253 254 255 256 257 258 259
}

static int save_files_dirs(const unsigned char *sha1,
		const char *base, int baselen, const char *path,
		unsigned int mode, int stage)
{
	int len = strlen(path);
J
Jonas Fonseca 已提交
260
	char *newpath = xmalloc(baselen + len + 1);
261 262 263 264 265
	memcpy(newpath, base, baselen);
	memcpy(newpath + baselen, path, len);
	newpath[baselen + len] = '\0';

	if (S_ISDIR(mode))
266
		path_list_insert(newpath, &current_directory_set);
267
	else
268
		path_list_insert(newpath, &current_file_set);
269 270 271 272 273
	free(newpath);

	return READ_TREE_RECURSIVE;
}

274
static int get_files_dirs(struct tree *tree)
275 276
{
	int n;
277
	if (read_tree_recursive(tree, "", 0, 0, NULL, save_files_dirs) != 0)
278
		return 0;
279
	n = current_file_set.nr + current_directory_set.nr;
280 281 282 283
	return n;
}

/*
284
 * Returns an index_entry instance which doesn't have to correspond to
285 286
 * a real cache entry in Git's index.
 */
287 288 289
static struct stage_data *insert_stage_data(const char *path,
		struct tree *o, struct tree *a, struct tree *b,
		struct path_list *entries)
290
{
291
	struct path_list_item *item;
292 293 294 295 296 297 298
	struct stage_data *e = xcalloc(1, sizeof(struct stage_data));
	get_tree_entry(o->object.sha1, path,
			e->stages[1].sha, &e->stages[1].mode);
	get_tree_entry(a->object.sha1, path,
			e->stages[2].sha, &e->stages[2].mode);
	get_tree_entry(b->object.sha1, path,
			e->stages[3].sha, &e->stages[3].mode);
299 300
	item = path_list_insert(path, entries);
	item->util = e;
301 302 303 304
	return e;
}

/*
305
 * Create a dictionary mapping file names to stage_data objects. The
306 307
 * dictionary contains one entry for every path with a non-zero stage entry.
 */
308
static struct path_list *get_unmerged(void)
309 310 311 312 313
{
	struct path_list *unmerged = xcalloc(1, sizeof(struct path_list));
	int i;

	unmerged->strdup_paths = 1;
314

315
	for (i = 0; i < active_nr; i++) {
316 317
		struct path_list_item *item;
		struct stage_data *e;
318 319 320 321
		struct cache_entry *ce = active_cache[i];
		if (!ce_stage(ce))
			continue;

322 323 324 325 326 327
		item = path_list_lookup(ce->name, unmerged);
		if (!item) {
			item = path_list_insert(ce->name, unmerged);
			item->util = xcalloc(1, sizeof(struct stage_data));
		}
		e = item->util;
328
		e->stages[ce_stage(ce)].mode = ce->ce_mode;
329
		hashcpy(e->stages[ce_stage(ce)].sha, ce->sha1);
330 331 332 333 334 335 336 337 338 339 340 341 342 343
	}

	return unmerged;
}

struct rename
{
	struct diff_filepair *pair;
	struct stage_data *src_entry;
	struct stage_data *dst_entry;
	unsigned processed:1;
};

/*
P
Pavel Roskin 已提交
344
 * Get information of all renames which occurred between 'o_tree' and
345 346 347
 * 'tree'. We need the three trees in the merge ('o_tree', 'a_tree' and
 * 'b_tree') to be able to associate the correct cache entries with
 * the rename information. 'tree' is always equal to either a_tree or b_tree.
348 349
 */
static struct path_list *get_renames(struct tree *tree,
350 351 352
					struct tree *o_tree,
					struct tree *a_tree,
					struct tree *b_tree,
353 354
					struct path_list *entries)
{
355 356 357 358 359
	int i;
	struct path_list *renames;
	struct diff_options opts;

	renames = xcalloc(1, sizeof(struct path_list));
360
	diff_setup(&opts);
361
	DIFF_OPT_SET(&opts, RECURSIVE);
362
	opts.detect_rename = DIFF_DETECT_RENAME;
363
	opts.rename_limit = rename_limit;
364 365 366
	opts.output_format = DIFF_FORMAT_NO_OUTPUT;
	if (diff_setup_done(&opts) < 0)
		die("diff setup failed");
367
	diff_tree_sha1(o_tree->object.sha1, tree->object.sha1, "", &opts);
368 369
	diffcore_std(&opts);
	for (i = 0; i < diff_queued_diff.nr; ++i) {
370
		struct path_list_item *item;
371 372 373 374 375 376 377 378 379
		struct rename *re;
		struct diff_filepair *pair = diff_queued_diff.queue[i];
		if (pair->status != 'R') {
			diff_free_filepair(pair);
			continue;
		}
		re = xmalloc(sizeof(*re));
		re->processed = 0;
		re->pair = pair;
380 381 382
		item = path_list_lookup(re->pair->one->path, entries);
		if (!item)
			re->src_entry = insert_stage_data(re->pair->one->path,
383
					o_tree, a_tree, b_tree, entries);
384 385 386 387 388 389
		else
			re->src_entry = item->util;

		item = path_list_lookup(re->pair->two->path, entries);
		if (!item)
			re->dst_entry = insert_stage_data(re->pair->two->path,
390
					o_tree, a_tree, b_tree, entries);
391 392 393
		else
			re->dst_entry = item->util;
		item = path_list_insert(pair->one->path, renames);
394 395 396 397 398 399 400 401
		item->util = re;
	}
	opts.output_format = DIFF_FORMAT_NO_OUTPUT;
	diff_queued_diff.nr = 0;
	diff_flush(&opts);
	return renames;
}

402 403 404
static int update_stages(const char *path, struct diff_filespec *o,
			 struct diff_filespec *a, struct diff_filespec *b,
			 int clear)
405 406
{
	int options = ADD_CACHE_OK_TO_ADD | ADD_CACHE_OK_TO_REPLACE;
407 408
	if (clear)
		if (remove_file_from_cache(path))
409
			return -1;
410 411
	if (o)
		if (add_cacheinfo(o->mode, o->sha1, path, 1, 0, options))
412
			return -1;
413 414
	if (a)
		if (add_cacheinfo(a->mode, a->sha1, path, 2, 0, options))
415
			return -1;
416 417
	if (b)
		if (add_cacheinfo(b->mode, b->sha1, path, 3, 0, options))
418 419 420 421 422 423
			return -1;
	return 0;
}

static int remove_path(const char *name)
{
P
Pierre Habouzit 已提交
424
	int ret;
425
	char *slash, *dirs;
426 427

	ret = unlink(name);
428
	if (ret)
429
		return ret;
P
Pierre Habouzit 已提交
430
	dirs = xstrdup(name);
431
	while ((slash = strrchr(name, '/'))) {
432
		*slash = '\0';
433
		if (rmdir(name) != 0)
434 435 436 437 438 439
			break;
	}
	free(dirs);
	return ret;
}

440
static int remove_file(int clean, const char *path, int no_wd)
441
{
442
	int update_cache = index_only || clean;
443
	int update_working_directory = !index_only && !no_wd;
444

445
	if (update_cache) {
446 447 448
		if (remove_file_from_cache(path))
			return -1;
	}
449
	if (update_working_directory) {
450
		unlink(path);
451
		if (errno != ENOENT || errno != EISDIR)
452 453 454 455 456 457 458 459 460
			return -1;
		remove_path(path);
	}
	return 0;
}

static char *unique_path(const char *path, const char *branch)
{
	char *newpath = xmalloc(strlen(path) + 1 + strlen(branch) + 8 + 1);
461 462
	int suffix = 0;
	struct stat st;
463
	char *p = newpath + strlen(path);
464
	strcpy(newpath, path);
465
	*(p++) = '~';
466
	strcpy(p, branch);
467 468
	for (; *p; ++p)
		if ('/' == *p)
469
			*p = '_';
470 471
	while (path_list_has_path(&current_file_set, newpath) ||
	       path_list_has_path(&current_directory_set, newpath) ||
472
	       lstat(newpath, &st) == 0)
473
		sprintf(p, "_%d", suffix++);
474

475
	path_list_insert(newpath, &current_file_set);
476 477 478
	return newpath;
}

479
static int mkdir_p(const char *path, unsigned long mode)
480
{
481 482
	/* path points to cache entries, so xstrdup before messing with it */
	char *buf = xstrdup(path);
483
	int result = safe_create_leading_directories(buf);
484
	free(buf);
485
	return result;
486 487 488 489 490
}

static void flush_buffer(int fd, const char *buf, unsigned long size)
{
	while (size > 0) {
491
		long ret = write_in_full(fd, buf, size);
492 493 494 495 496 497 498 499 500 501 502 503 504
		if (ret < 0) {
			/* Ignore epipe */
			if (errno == EPIPE)
				break;
			die("merge-recursive: %s", strerror(errno));
		} else if (!ret) {
			die("merge-recursive: disk full?");
		}
		size -= ret;
		buf += ret;
	}
}

505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529
static int make_room_for_path(const char *path)
{
	int status;
	const char *msg = "failed to create path '%s'%s";

	status = mkdir_p(path, 0777);
	if (status) {
		if (status == -3) {
			/* something else exists */
			error(msg, path, ": perhaps a D/F conflict?");
			return -1;
		}
		die(msg, path, "");
	}

	/* Successful unlink is good.. */
	if (!unlink(path))
		return 0;
	/* .. and so is no existing file */
	if (errno == ENOENT)
		return 0;
	/* .. but not some other error (who really cares what?) */
	return error(msg, path, ": perhaps a D/F conflict?");
}

530 531 532 533 534
static void update_file_flags(const unsigned char *sha,
			      unsigned mode,
			      const char *path,
			      int update_cache,
			      int update_wd)
535
{
536 537
	if (index_only)
		update_wd = 0;
538

539
	if (update_wd) {
540
		enum object_type type;
541 542 543
		void *buf;
		unsigned long size;

544 545 546 547
		if (S_ISGITLINK(mode))
			die("cannot read object %s '%s': It is a submodule!",
			    sha1_to_hex(sha), path);

548
		buf = read_sha1_file(sha, &type, &size);
549 550
		if (!buf)
			die("cannot read object %s '%s'", sha1_to_hex(sha), path);
551
		if (type != OBJ_BLOB)
552 553
			die("blob expected for %s '%s'", sha1_to_hex(sha), path);

554 555 556 557
		if (make_room_for_path(path) < 0) {
			update_wd = 0;
			goto update_index;
		}
558
		if (S_ISREG(mode) || (!has_symlinks && S_ISLNK(mode))) {
559 560
			int fd;
			if (mode & 0100)
561 562 563
				mode = 0777;
			else
				mode = 0666;
564 565
			fd = open(path, O_WRONLY | O_TRUNC | O_CREAT, mode);
			if (fd < 0)
566 567 568
				die("failed to open %s: %s", path, strerror(errno));
			flush_buffer(fd, buf, size);
			close(fd);
569
		} else if (S_ISLNK(mode)) {
P
Pierre Habouzit 已提交
570
			char *lnk = xmemdupz(buf, size);
571
			mkdir_p(path, 0777);
572
			unlink(path);
573
			symlink(lnk, path);
574
			free(lnk);
575 576 577 578
		} else
			die("do not know what to do with %06o %s '%s'",
			    mode, sha1_to_hex(sha), path);
	}
579
 update_index:
580 581
	if (update_cache)
		add_cacheinfo(mode, sha, path, 0, update_wd, ADD_CACHE_OK_TO_ADD);
582 583
}

584 585 586 587
static void update_file(int clean,
			const unsigned char *sha,
			unsigned mode,
			const char *path)
588 589 590 591 592 593 594 595 596 597 598 599 600 601
{
	update_file_flags(sha, mode, path, index_only || clean, !index_only);
}

/* Low level file merging, update and removal */

struct merge_file_info
{
	unsigned char sha[20];
	unsigned mode;
	unsigned clean:1,
		 merge:1;
};

J
Junio C Hamano 已提交
602
static void fill_mm(const unsigned char *sha1, mmfile_t *mm)
603 604
{
	unsigned long size;
605
	enum object_type type;
606

607 608 609 610 611
	if (!hashcmp(sha1, null_sha1)) {
		mm->ptr = xstrdup("");
		mm->size = 0;
		return;
	}
612

613 614
	mm->ptr = read_sha1_file(sha1, &type, &size);
	if (!mm->ptr || type != OBJ_BLOB)
615
		die("unable to read blob object %s", sha1_to_hex(sha1));
J
Junio C Hamano 已提交
616
	mm->size = size;
617 618
}

619 620 621 622 623 624 625
/*
 * Customizable low-level merge drivers support.
 */

struct ll_merge_driver;
typedef int (*ll_merge_fn)(const struct ll_merge_driver *,
			   const char *path,
626
			   mmfile_t *orig,
627 628 629 630
			   mmfile_t *src1, const char *name1,
			   mmfile_t *src2, const char *name2,
			   mmbuffer_t *result);

631 632 633 634
struct ll_merge_driver {
	const char *name;
	const char *description;
	ll_merge_fn fn;
635
	const char *recursive;
636 637 638 639 640 641 642
	struct ll_merge_driver *next;
	char *cmdline;
};

/*
 * Built-in low-levels
 */
643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662
static int ll_binary_merge(const struct ll_merge_driver *drv_unused,
			   const char *path_unused,
			   mmfile_t *orig,
			   mmfile_t *src1, const char *name1,
			   mmfile_t *src2, const char *name2,
			   mmbuffer_t *result)
{
	/*
	 * The tentative merge result is "ours" for the final round,
	 * or common ancestor for an internal merge.  Still return
	 * "conflicted merge" status.
	 */
	mmfile_t *stolen = index_only ? orig : src1;

	result->ptr = stolen->ptr;
	result->size = stolen->size;
	stolen->ptr = NULL;
	return 1;
}

663 664
static int ll_xdl_merge(const struct ll_merge_driver *drv_unused,
			const char *path_unused,
665
			mmfile_t *orig,
666 667 668 669 670 671
			mmfile_t *src1, const char *name1,
			mmfile_t *src2, const char *name2,
			mmbuffer_t *result)
{
	xpparam_t xpp;

672
	if (buffer_is_binary(orig->ptr, orig->size) ||
673 674 675
	    buffer_is_binary(src1->ptr, src1->size) ||
	    buffer_is_binary(src2->ptr, src2->size)) {
		warning("Cannot merge binary files: %s vs. %s\n",
676
			name1, name2);
677 678 679 680 681
		return ll_binary_merge(drv_unused, path_unused,
				       orig, src1, name1,
				       src2, name2,
				       result);
	}
682

683 684 685 686 687 688 689 690
	memset(&xpp, 0, sizeof(xpp));
	return xdl_merge(orig,
			 src1, name1,
			 src2, name2,
			 &xpp, XDL_MERGE_ZEALOUS,
			 result);
}

691 692
static int ll_union_merge(const struct ll_merge_driver *drv_unused,
			  const char *path_unused,
693
			  mmfile_t *orig,
694 695 696 697 698 699 700 701
			  mmfile_t *src1, const char *name1,
			  mmfile_t *src2, const char *name2,
			  mmbuffer_t *result)
{
	char *src, *dst;
	long size;
	const int marker_size = 7;

702 703
	int status = ll_xdl_merge(drv_unused, path_unused,
				  orig, src1, NULL, src2, NULL, result);
704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733
	if (status <= 0)
		return status;
	size = result->size;
	src = dst = result->ptr;
	while (size) {
		char ch;
		if ((marker_size < size) &&
		    (*src == '<' || *src == '=' || *src == '>')) {
			int i;
			ch = *src;
			for (i = 0; i < marker_size; i++)
				if (src[i] != ch)
					goto not_a_marker;
			if (src[marker_size] != '\n')
				goto not_a_marker;
			src += marker_size + 1;
			size -= marker_size + 1;
			continue;
		}
	not_a_marker:
		do {
			ch = *src++;
			*dst++ = ch;
			size--;
		} while (ch != '\n' && size);
	}
	result->size = dst - result->ptr;
	return 0;
}

734 735 736 737 738 739 740
#define LL_BINARY_MERGE 0
#define LL_TEXT_MERGE 1
#define LL_UNION_MERGE 2
static struct ll_merge_driver ll_merge_drv[] = {
	{ "binary", "built-in binary merge", ll_binary_merge },
	{ "text", "built-in 3-way text merge", ll_xdl_merge },
	{ "union", "built-in union merge", ll_union_merge },
741 742
};

743 744 745 746 747
static void create_temp(mmfile_t *src, char *path)
{
	int fd;

	strcpy(path, ".merge_file_XXXXXX");
748
	fd = xmkstemp(path);
749 750 751 752 753
	if (write_in_full(fd, src->ptr, src->size) != src->size)
		die("unable to write temp-file");
	close(fd);
}

754 755 756 757 758
/*
 * User defined low-level merge driver support.
 */
static int ll_ext_merge(const struct ll_merge_driver *fn,
			const char *path,
759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
			mmfile_t *orig,
			mmfile_t *src1, const char *name1,
			mmfile_t *src2, const char *name2,
			mmbuffer_t *result)
{
	char temp[3][50];
	char cmdbuf[2048];
	struct interp table[] = {
		{ "%O" },
		{ "%A" },
		{ "%B" },
	};
	struct child_process child;
	const char *args[20];
	int status, fd, i;
	struct stat st;

J
Junio C Hamano 已提交
776 777 778
	if (fn->cmdline == NULL)
		die("custom merge driver %s lacks command line.", fn->name);

779 780 781 782 783 784 785 786 787 788
	result->ptr = NULL;
	result->size = 0;
	create_temp(orig, temp[0]);
	create_temp(src1, temp[1]);
	create_temp(src2, temp[2]);

	interp_set_entry(table, 0, temp[0]);
	interp_set_entry(table, 1, temp[1]);
	interp_set_entry(table, 2, temp[2]);

789 790 791 792
	output(1, "merging %s using %s", path,
	       fn->description ? fn->description : fn->name);

	interpolate(cmdbuf, sizeof(cmdbuf), fn->cmdline, table, 3);
793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828

	memset(&child, 0, sizeof(child));
	child.argv = args;
	args[0] = "sh";
	args[1] = "-c";
	args[2] = cmdbuf;
	args[3] = NULL;

	status = run_command(&child);
	if (status < -ERR_RUN_COMMAND_FORK)
		; /* failure in run-command */
	else
		status = -status;
	fd = open(temp[1], O_RDONLY);
	if (fd < 0)
		goto bad;
	if (fstat(fd, &st))
		goto close_bad;
	result->size = st.st_size;
	result->ptr = xmalloc(result->size + 1);
	if (read_in_full(fd, result->ptr, result->size) != result->size) {
		free(result->ptr);
		result->ptr = NULL;
		result->size = 0;
	}
 close_bad:
	close(fd);
 bad:
	for (i = 0; i < 3; i++)
		unlink(temp[i]);
	return status;
}

/*
 * merge.default and merge.driver configuration items
 */
829
static struct ll_merge_driver *ll_user_merge, **ll_user_merge_tail;
830
static const char *default_ll_merge;
831 832 833

static int read_merge_config(const char *var, const char *value)
{
834 835 836
	struct ll_merge_driver *fn;
	const char *ep, *name;
	int namelen;
837

838
	if (!strcmp(var, "merge.default")) {
839 840 841
		if (!value)
			return config_error_nonbool(var);
		default_ll_merge = strdup(value);
842 843 844
		return 0;
	}

845 846 847 848 849
	/*
	 * We are not interested in anything but "merge.<name>.variable";
	 * especially, we do not want to look at variables such as
	 * "merge.summary", "merge.tool", and "merge.verbosity".
	 */
J
Junio C Hamano 已提交
850
	if (prefixcmp(var, "merge.") || (ep = strrchr(var, '.')) == var + 5)
851
		return 0;
852

853
	/*
854 855
	 * Find existing one as we might be processing merge.<name>.var2
	 * after seeing merge.<name>.var1.
856
	 */
857 858 859 860 861 862 863
	name = var + 6;
	namelen = ep - name;
	for (fn = ll_user_merge; fn; fn = fn->next)
		if (!strncmp(fn->name, name, namelen) && !fn->name[namelen])
			break;
	if (!fn) {
		fn = xcalloc(1, sizeof(struct ll_merge_driver));
P
Pierre Habouzit 已提交
864
		fn->name = xmemdupz(name, namelen);
865 866
		fn->fn = ll_ext_merge;
		*ll_user_merge_tail = fn;
867
		ll_user_merge_tail = &(fn->next);
868 869 870 871 872 873
	}

	ep++;

	if (!strcmp("name", ep)) {
		if (!value)
874
			return config_error_nonbool(var);
875 876 877 878 879 880
		fn->description = strdup(value);
		return 0;
	}

	if (!strcmp("driver", ep)) {
		if (!value)
881
			return config_error_nonbool(var);
882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901
		/*
		 * merge.<name>.driver specifies the command line:
		 *
		 *	command-line
		 *
		 * The command-line will be interpolated with the following
		 * tokens and is given to the shell:
		 *
		 *    %O - temporary file name for the merge base.
		 *    %A - temporary file name for our version.
		 *    %B - temporary file name for the other branches' version.
		 *
		 * The external merge driver should write the results in the
		 * file named by %A, and signal that it has done with zero exit
		 * status.
		 */
		fn->cmdline = strdup(value);
		return 0;
	}

902 903
	if (!strcmp("recursive", ep)) {
		if (!value)
904
			return config_error_nonbool(var);
905 906 907 908
		fn->recursive = strdup(value);
		return 0;
	}

909 910 911 912
	return 0;
}

static void initialize_ll_merge(void)
913
{
914
	if (ll_user_merge_tail)
915
		return;
916
	ll_user_merge_tail = &ll_user_merge;
917 918 919
	git_config(read_merge_config);
}

920
static const struct ll_merge_driver *find_ll_merge_driver(const char *merge_attr)
921
{
922
	struct ll_merge_driver *fn;
923 924 925
	const char *name;
	int i;

926 927 928
	initialize_ll_merge();

	if (ATTR_TRUE(merge_attr))
929
		return &ll_merge_drv[LL_TEXT_MERGE];
930
	else if (ATTR_FALSE(merge_attr))
931
		return &ll_merge_drv[LL_BINARY_MERGE];
932 933
	else if (ATTR_UNSET(merge_attr)) {
		if (!default_ll_merge)
934
			return &ll_merge_drv[LL_TEXT_MERGE];
935 936 937
		else
			name = default_ll_merge;
	}
938 939 940
	else
		name = merge_attr;

941 942 943
	for (fn = ll_user_merge; fn; fn = fn->next)
		if (!strcmp(fn->name, name))
			return fn;
944

945 946 947
	for (i = 0; i < ARRAY_SIZE(ll_merge_drv); i++)
		if (!strcmp(ll_merge_drv[i].name, name))
			return &ll_merge_drv[i];
948 949

	/* default to the 3-way */
950
	return &ll_merge_drv[LL_TEXT_MERGE];
951 952
}

953
static const char *git_path_check_merge(const char *path)
954 955 956 957 958 959 960
{
	static struct git_attr_check attr_merge_check;

	if (!attr_merge_check.attr)
		attr_merge_check.attr = git_attr("merge", 5);

	if (git_checkattr(path, 1, &attr_merge_check))
961
		return NULL;
962 963 964
	return attr_merge_check.value;
}

965 966 967 968 969 970 971 972 973 974
static int ll_merge(mmbuffer_t *result_buf,
		    struct diff_filespec *o,
		    struct diff_filespec *a,
		    struct diff_filespec *b,
		    const char *branch1,
		    const char *branch2)
{
	mmfile_t orig, src1, src2;
	char *name1, *name2;
	int merge_status;
975
	const char *ll_driver_name;
976
	const struct ll_merge_driver *driver;
977 978 979 980 981 982 983 984

	name1 = xstrdup(mkpath("%s:%s", branch1, a->path));
	name2 = xstrdup(mkpath("%s:%s", branch2, b->path));

	fill_mm(o->sha1, &orig);
	fill_mm(a->sha1, &src1);
	fill_mm(b->sha1, &src2);

985 986
	ll_driver_name = git_path_check_merge(a->path);
	driver = find_ll_merge_driver(ll_driver_name);
987

988 989
	if (index_only && driver->recursive)
		driver = find_ll_merge_driver(driver->recursive);
990 991 992
	merge_status = driver->fn(driver, a->path,
				  &orig, &src1, name1, &src2, name2,
				  result_buf);
993

994 995 996 997 998 999 1000 1001
	free(name1);
	free(name2);
	free(orig.ptr);
	free(src1.ptr);
	free(src2.ptr);
	return merge_status;
}

1002 1003
static struct merge_file_info merge_file(struct diff_filespec *o,
		struct diff_filespec *a, struct diff_filespec *b,
1004
		const char *branch1, const char *branch2)
1005 1006 1007 1008 1009
{
	struct merge_file_info result;
	result.merge = 0;
	result.clean = 1;

1010
	if ((S_IFMT & a->mode) != (S_IFMT & b->mode)) {
1011
		result.clean = 0;
1012 1013
		if (S_ISREG(a->mode)) {
			result.mode = a->mode;
1014
			hashcpy(result.sha, a->sha1);
1015
		} else {
1016
			result.mode = b->mode;
1017
			hashcpy(result.sha, b->sha1);
1018 1019
		}
	} else {
1020
		if (!sha_eq(a->sha1, o->sha1) && !sha_eq(b->sha1, o->sha1))
1021 1022
			result.merge = 1;

1023
		result.mode = a->mode == o->mode ? b->mode: a->mode;
1024

1025
		if (sha_eq(a->sha1, o->sha1))
1026
			hashcpy(result.sha, b->sha1);
1027
		else if (sha_eq(b->sha1, o->sha1))
1028
			hashcpy(result.sha, a->sha1);
1029
		else if (S_ISREG(a->mode)) {
J
Junio C Hamano 已提交
1030 1031 1032
			mmbuffer_t result_buf;
			int merge_status;

1033 1034
			merge_status = ll_merge(&result_buf, o, a, b,
						branch1, branch2);
J
Junio C Hamano 已提交
1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045

			if ((merge_status < 0) || !result_buf.ptr)
				die("Failed to execute internal merge");

			if (write_sha1_file(result_buf.ptr, result_buf.size,
					    blob_type, result.sha))
				die("Unable to add %s to database",
				    a->path);

			free(result_buf.ptr);
			result.clean = (merge_status == 0);
1046 1047 1048 1049
		} else if (S_ISGITLINK(a->mode)) {
			result.clean = 0;
			hashcpy(result.sha, a->sha1);
		} else if (S_ISLNK(a->mode)) {
1050
			hashcpy(result.sha, a->sha1);
1051

1052
			if (!sha_eq(a->sha1, b->sha1))
1053
				result.clean = 0;
1054 1055
		} else {
			die("unsupported object type in the tree");
1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070
		}
	}

	return result;
}

static void conflict_rename_rename(struct rename *ren1,
				   const char *branch1,
				   struct rename *ren2,
				   const char *branch2)
{
	char *del[2];
	int delp = 0;
	const char *ren1_dst = ren1->pair->two->path;
	const char *ren2_dst = ren2->pair->two->path;
1071 1072 1073 1074
	const char *dst_name1 = ren1_dst;
	const char *dst_name2 = ren2_dst;
	if (path_list_has_path(&current_directory_set, ren1_dst)) {
		dst_name1 = del[delp++] = unique_path(ren1_dst, branch1);
1075
		output(1, "%s is a directory in %s added as %s instead",
1076
		       ren1_dst, branch2, dst_name1);
1077
		remove_file(0, ren1_dst, 0);
1078
	}
1079 1080
	if (path_list_has_path(&current_directory_set, ren2_dst)) {
		dst_name2 = del[delp++] = unique_path(ren2_dst, branch2);
1081
		output(1, "%s is a directory in %s added as %s instead",
1082
		       ren2_dst, branch1, dst_name2);
1083
		remove_file(0, ren2_dst, 0);
1084
	}
1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097
	if (index_only) {
		remove_file_from_cache(dst_name1);
		remove_file_from_cache(dst_name2);
		/*
		 * Uncomment to leave the conflicting names in the resulting tree
		 *
		 * update_file(0, ren1->pair->two->sha1, ren1->pair->two->mode, dst_name1);
		 * update_file(0, ren2->pair->two->sha1, ren2->pair->two->mode, dst_name2);
		 */
	} else {
		update_stages(dst_name1, NULL, ren1->pair->two, NULL, 1);
		update_stages(dst_name2, NULL, NULL, ren2->pair->two, 1);
	}
1098
	while (delp--)
1099 1100 1101 1102 1103 1104
		free(del[delp]);
}

static void conflict_rename_dir(struct rename *ren1,
				const char *branch1)
{
1105
	char *new_path = unique_path(ren1->pair->two->path, branch1);
1106
	output(1, "Renamed %s to %s instead", ren1->pair->one->path, new_path);
1107
	remove_file(0, ren1->pair->two->path, 0);
1108 1109
	update_file(0, ren1->pair->two->sha1, ren1->pair->two->mode, new_path);
	free(new_path);
1110 1111 1112 1113 1114 1115 1116
}

static void conflict_rename_rename_2(struct rename *ren1,
				     const char *branch1,
				     struct rename *ren2,
				     const char *branch2)
{
1117 1118
	char *new_path1 = unique_path(ren1->pair->two->path, branch1);
	char *new_path2 = unique_path(ren2->pair->two->path, branch2);
1119
	output(1, "Renamed %s to %s and %s to %s instead",
1120 1121
	       ren1->pair->one->path, new_path1,
	       ren2->pair->one->path, new_path2);
1122
	remove_file(0, ren1->pair->two->path, 0);
1123 1124 1125 1126
	update_file(0, ren1->pair->two->sha1, ren1->pair->two->mode, new_path1);
	update_file(0, ren2->pair->two->sha1, ren2->pair->two->mode, new_path2);
	free(new_path2);
	free(new_path1);
1127 1128
}

1129 1130 1131 1132
static int process_renames(struct path_list *a_renames,
			   struct path_list *b_renames,
			   const char *a_branch,
			   const char *b_branch)
1133
{
1134 1135
	int clean_merge = 1, i, j;
	struct path_list a_by_dst = {NULL, 0, 0, 0}, b_by_dst = {NULL, 0, 0, 0};
1136 1137
	const struct rename *sre;

1138 1139 1140
	for (i = 0; i < a_renames->nr; i++) {
		sre = a_renames->items[i].util;
		path_list_insert(sre->pair->two->path, &a_by_dst)->util
1141 1142
			= sre->dst_entry;
	}
1143 1144 1145
	for (i = 0; i < b_renames->nr; i++) {
		sre = b_renames->items[i].util;
		path_list_insert(sre->pair->two->path, &b_by_dst)->util
1146 1147 1148
			= sre->dst_entry;
	}

1149
	for (i = 0, j = 0; i < a_renames->nr || j < b_renames->nr;) {
1150 1151
		int compare;
		char *src;
1152
		struct path_list *renames1, *renames2, *renames2Dst;
1153
		struct rename *ren1 = NULL, *ren2 = NULL;
1154
		const char *branch1, *branch2;
1155 1156
		const char *ren1_src, *ren1_dst;

1157
		if (i >= a_renames->nr) {
1158
			compare = 1;
1159 1160
			ren2 = b_renames->items[j++].util;
		} else if (j >= b_renames->nr) {
1161
			compare = -1;
1162
			ren1 = a_renames->items[i++].util;
1163
		} else {
1164 1165
			compare = strcmp(a_renames->items[i].path,
					b_renames->items[j].path);
1166 1167 1168 1169
			if (compare <= 0)
				ren1 = a_renames->items[i++].util;
			if (compare >= 0)
				ren2 = b_renames->items[j++].util;
1170 1171
		}

1172
		/* TODO: refactor, so that 1/2 are not needed */
1173
		if (ren1) {
1174 1175 1176 1177 1178
			renames1 = a_renames;
			renames2 = b_renames;
			renames2Dst = &b_by_dst;
			branch1 = a_branch;
			branch2 = b_branch;
1179
		} else {
1180
			struct rename *tmp;
1181 1182 1183 1184 1185
			renames1 = b_renames;
			renames2 = a_renames;
			renames2Dst = &a_by_dst;
			branch1 = b_branch;
			branch2 = a_branch;
1186
			tmp = ren2;
1187 1188 1189
			ren2 = ren1;
			ren1 = tmp;
		}
1190
		src = ren1->pair->one->path;
1191 1192 1193 1194

		ren1->dst_entry->processed = 1;
		ren1->src_entry->processed = 1;

1195
		if (ren1->processed)
1196 1197 1198
			continue;
		ren1->processed = 1;

1199 1200
		ren1_src = ren1->pair->one->path;
		ren1_dst = ren1->pair->two->path;
1201

1202
		if (ren2) {
1203 1204 1205 1206 1207 1208 1209 1210
			const char *ren2_src = ren2->pair->one->path;
			const char *ren2_dst = ren2->pair->two->path;
			/* Renamed in 1 and renamed in 2 */
			if (strcmp(ren1_src, ren2_src) != 0)
				die("ren1.src != ren2.src");
			ren2->dst_entry->processed = 1;
			ren2->processed = 1;
			if (strcmp(ren1_dst, ren2_dst) != 0) {
1211
				clean_merge = 0;
1212
				output(1, "CONFLICT (rename/rename): "
1213 1214
				       "Rename \"%s\"->\"%s\" in branch \"%s\" "
				       "rename \"%s\"->\"%s\" in \"%s\"%s",
1215
				       src, ren1_dst, branch1,
1216 1217 1218 1219 1220 1221 1222
				       src, ren2_dst, branch2,
				       index_only ? " (left unresolved)": "");
				if (index_only) {
					remove_file_from_cache(src);
					update_file(0, ren1->pair->one->sha1,
						    ren1->pair->one->mode, src);
				}
1223
				conflict_rename_rename(ren1, branch1, ren2, branch2);
1224 1225
			} else {
				struct merge_file_info mfi;
1226
				remove_file(1, ren1_src, 1);
1227 1228 1229
				mfi = merge_file(ren1->pair->one,
						 ren1->pair->two,
						 ren2->pair->two,
1230 1231
						 branch1,
						 branch2);
1232
				if (mfi.merge || !mfi.clean)
1233
					output(1, "Renamed %s->%s", src, ren1_dst);
1234

1235
				if (mfi.merge)
1236
					output(2, "Auto-merged %s", ren1_dst);
1237

1238
				if (!mfi.clean) {
1239
					output(1, "CONFLICT (content): merge conflict in %s",
1240
					       ren1_dst);
1241
					clean_merge = 0;
1242

1243
					if (!index_only)
1244
						update_stages(ren1_dst,
1245 1246 1247
							      ren1->pair->one,
							      ren1->pair->two,
							      ren2->pair->two,
1248 1249 1250 1251 1252 1253
							      1 /* clear */);
				}
				update_file(mfi.clean, mfi.sha, mfi.mode, ren1_dst);
			}
		} else {
			/* Renamed in 1, maybe changed in 2 */
1254 1255 1256
			struct path_list_item *item;
			/* we only use sha1 and mode of these */
			struct diff_filespec src_other, dst_other;
1257
			int try_merge, stage = a_renames == renames1 ? 3: 2;
1258

1259
			remove_file(1, ren1_src, index_only || stage == 3);
1260

J
Junio C Hamano 已提交
1261
			hashcpy(src_other.sha1, ren1->src_entry->stages[stage].sha);
1262
			src_other.mode = ren1->src_entry->stages[stage].mode;
J
Junio C Hamano 已提交
1263
			hashcpy(dst_other.sha1, ren1->dst_entry->stages[stage].sha);
1264
			dst_other.mode = ren1->dst_entry->stages[stage].mode;
1265

1266
			try_merge = 0;
1267

1268 1269
			if (path_list_has_path(&current_directory_set, ren1_dst)) {
				clean_merge = 0;
1270
				output(1, "CONFLICT (rename/directory): Renamed %s->%s in %s "
1271
				       " directory %s added in %s",
1272 1273 1274
				       ren1_src, ren1_dst, branch1,
				       ren1_dst, branch2);
				conflict_rename_dir(ren1, branch1);
1275
			} else if (sha_eq(src_other.sha1, null_sha1)) {
1276
				clean_merge = 0;
1277
				output(1, "CONFLICT (rename/delete): Renamed %s->%s in %s "
1278
				       "and deleted in %s",
1279 1280
				       ren1_src, ren1_dst, branch1,
				       branch2);
1281
				update_file(0, ren1->pair->two->sha1, ren1->pair->two->mode, ren1_dst);
1282
			} else if (!sha_eq(dst_other.sha1, null_sha1)) {
1283 1284 1285
				const char *new_path;
				clean_merge = 0;
				try_merge = 1;
1286
				output(1, "CONFLICT (rename/add): Renamed %s->%s in %s. "
1287
				       "%s added in %s",
1288 1289 1290
				       ren1_src, ren1_dst, branch1,
				       ren1_dst, branch2);
				new_path = unique_path(ren1_dst, branch2);
1291
				output(1, "Added as %s instead", new_path);
1292
				update_file(0, dst_other.sha1, dst_other.mode, new_path);
1293 1294
			} else if ((item = path_list_lookup(ren1_dst, renames2Dst))) {
				ren2 = item->util;
1295
				clean_merge = 0;
1296
				ren2->processed = 1;
1297 1298
				output(1, "CONFLICT (rename/rename): Renamed %s->%s in %s. "
				       "Renamed %s->%s in %s",
1299 1300 1301
				       ren1_src, ren1_dst, branch1,
				       ren2->pair->one->path, ren2->pair->two->path, branch2);
				conflict_rename_rename_2(ren1, branch1, ren2, branch2);
1302
			} else
1303
				try_merge = 1;
1304

1305
			if (try_merge) {
1306
				struct diff_filespec *o, *a, *b;
1307
				struct merge_file_info mfi;
1308 1309 1310
				src_other.path = (char *)ren1_src;

				o = ren1->pair->one;
1311
				if (a_renames == renames1) {
1312 1313 1314 1315 1316 1317 1318
					a = ren1->pair->two;
					b = &src_other;
				} else {
					b = ren1->pair->two;
					a = &src_other;
				}
				mfi = merge_file(o, a, b,
1319
						a_branch, b_branch);
1320

1321
				if (mfi.clean &&
1322 1323
				    sha_eq(mfi.sha, ren1->pair->two->sha1) &&
				    mfi.mode == ren1->pair->two->mode)
1324 1325 1326 1327 1328
					/*
					 * This messaged is part of
					 * t6022 test. If you change
					 * it update the test too.
					 */
1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344
					output(3, "Skipped %s (merged same as existing)", ren1_dst);
				else {
					if (mfi.merge || !mfi.clean)
						output(1, "Renamed %s => %s", ren1_src, ren1_dst);
					if (mfi.merge)
						output(2, "Auto-merged %s", ren1_dst);
					if (!mfi.clean) {
						output(1, "CONFLICT (rename/modify): Merge conflict in %s",
						       ren1_dst);
						clean_merge = 0;

						if (!index_only)
							update_stages(ren1_dst,
								      o, a, b, 1);
					}
					update_file(mfi.clean, mfi.sha, mfi.mode, ren1_dst);
1345 1346 1347 1348
				}
			}
		}
	}
1349 1350
	path_list_clear(&a_by_dst, 0);
	path_list_clear(&b_by_dst, 0);
1351

1352
	return clean_merge;
1353 1354
}

1355
static unsigned char *stage_sha(const unsigned char *sha, unsigned mode)
1356
{
1357
	return (is_null_sha1(sha) || mode == 0) ? NULL: (unsigned char *)sha;
1358 1359 1360 1361
}

/* Per entry merge function */
static int process_entry(const char *path, struct stage_data *entry,
1362 1363
			 const char *branch1,
			 const char *branch2)
1364 1365 1366 1367 1368
{
	/*
	printf("processing entry, clean cache: %s\n", index_only ? "yes": "no");
	print_index_entry("\tpath: ", entry);
	*/
1369 1370 1371 1372
	int clean_merge = 1;
	unsigned o_mode = entry->stages[1].mode;
	unsigned a_mode = entry->stages[2].mode;
	unsigned b_mode = entry->stages[3].mode;
1373 1374 1375
	unsigned char *o_sha = stage_sha(entry->stages[1].sha, o_mode);
	unsigned char *a_sha = stage_sha(entry->stages[2].sha, a_mode);
	unsigned char *b_sha = stage_sha(entry->stages[3].sha, b_mode);
1376 1377

	if (o_sha && (!a_sha || !b_sha)) {
1378
		/* Case A: Deleted in one */
1379 1380 1381
		if ((!a_sha && !b_sha) ||
		    (sha_eq(a_sha, o_sha) && !b_sha) ||
		    (!a_sha && sha_eq(b_sha, o_sha))) {
1382 1383
			/* Deleted in both or deleted in one and
			 * unchanged in the other */
1384
			if (a_sha)
1385
				output(2, "Removed %s", path);
1386 1387
			/* do not touch working file if it did not exist */
			remove_file(1, path, !a_sha);
1388 1389
		} else {
			/* Deleted in one and changed in the other */
1390 1391
			clean_merge = 0;
			if (!a_sha) {
1392
				output(1, "CONFLICT (delete/modify): %s deleted in %s "
1393
				       "and modified in %s. Version %s of %s left in tree.",
1394 1395
				       path, branch1,
				       branch2, branch2, path);
1396
				update_file(0, b_sha, b_mode, path);
1397
			} else {
1398
				output(1, "CONFLICT (delete/modify): %s deleted in %s "
1399
				       "and modified in %s. Version %s of %s left in tree.",
1400 1401
				       path, branch2,
				       branch1, branch1, path);
1402
				update_file(0, a_sha, a_mode, path);
1403 1404 1405
			}
		}

1406 1407
	} else if ((!o_sha && a_sha && !b_sha) ||
		   (!o_sha && !a_sha && b_sha)) {
1408
		/* Case B: Added in one. */
1409 1410
		const char *add_branch;
		const char *other_branch;
1411 1412 1413 1414
		unsigned mode;
		const unsigned char *sha;
		const char *conf;

1415
		if (a_sha) {
1416 1417
			add_branch = branch1;
			other_branch = branch2;
1418 1419
			mode = a_mode;
			sha = a_sha;
1420 1421
			conf = "file/directory";
		} else {
1422 1423
			add_branch = branch2;
			other_branch = branch1;
1424 1425
			mode = b_mode;
			sha = b_sha;
1426 1427
			conf = "directory/file";
		}
1428 1429 1430
		if (path_list_has_path(&current_directory_set, path)) {
			const char *new_path = unique_path(path, add_branch);
			clean_merge = 0;
1431
			output(1, "CONFLICT (%s): There is a directory with name %s in %s. "
1432
			       "Added %s as %s",
1433
			       conf, path, other_branch, path, new_path);
1434
			remove_file(0, path, 0);
1435
			update_file(0, sha, mode, new_path);
1436
		} else {
1437
			output(2, "Added %s", path);
1438 1439
			update_file(1, sha, mode, path);
		}
1440 1441
	} else if (a_sha && b_sha) {
		/* Case C: Added in both (check for same permissions) and */
1442
		/* case D: Modified in both, but differently. */
1443
		const char *reason = "content";
1444
		struct merge_file_info mfi;
1445 1446
		struct diff_filespec o, a, b;

1447 1448 1449 1450
		if (!o_sha) {
			reason = "add/add";
			o_sha = (unsigned char *)null_sha1;
		}
1451
		output(2, "Auto-merged %s", path);
1452
		o.path = a.path = b.path = (char *)path;
1453
		hashcpy(o.sha1, o_sha);
1454
		o.mode = o_mode;
1455
		hashcpy(a.sha1, a_sha);
1456
		a.mode = a_mode;
1457
		hashcpy(b.sha1, b_sha);
1458
		b.mode = b_mode;
1459 1460

		mfi = merge_file(&o, &a, &b,
1461
				 branch1, branch2);
1462

1463
		clean_merge = mfi.clean;
1464
		if (mfi.clean)
1465
			update_file(1, mfi.sha, mfi.mode, path);
1466 1467 1468
		else if (S_ISGITLINK(mfi.mode))
			output(1, "CONFLICT (submodule): Merge conflict in %s "
			       "- needs %s", path, sha1_to_hex(b.sha1));
1469
		else {
1470
			output(1, "CONFLICT (%s): Merge conflict in %s",
1471
					reason, path);
1472

1473
			if (index_only)
1474 1475 1476
				update_file(0, mfi.sha, mfi.mode, path);
			else
				update_file_flags(mfi.sha, mfi.mode, path,
1477
					      0 /* update_cache */, 1 /* update_working_directory */);
1478
		}
1479 1480 1481 1482 1483 1484
	} else if (!o_sha && !a_sha && !b_sha) {
		/*
		 * this entry was deleted altogether. a_mode == 0 means
		 * we had that path and want to actively remove it.
		 */
		remove_file(1, path, !a_mode);
1485 1486 1487
	} else
		die("Fatal merge failure, shouldn't happen.");

1488
	return clean_merge;
1489 1490
}

D
Daniel Barkalow 已提交
1491 1492 1493 1494 1495 1496
int merge_trees(struct tree *head,
		struct tree *merge,
		struct tree *common,
		const char *branch1,
		const char *branch2,
		struct tree **result)
1497
{
1498
	int code, clean;
J
Junio C Hamano 已提交
1499 1500 1501 1502 1503 1504

	if (subtree_merge) {
		merge = shift_tree_object(head, merge);
		common = shift_tree_object(head, common);
	}

1505
	if (sha_eq(common->object.sha1, merge->object.sha1)) {
1506
		output(0, "Already uptodate!");
1507 1508
		*result = head;
		return 1;
1509 1510
	}

1511
	code = git_merge_trees(index_only, common, head, merge);
1512

1513
	if (code != 0)
1514 1515 1516 1517
		die("merging of trees %s and %s failed",
		    sha1_to_hex(head->object.sha1),
		    sha1_to_hex(merge->object.sha1));

1518
	if (unmerged_cache()) {
1519 1520
		struct path_list *entries, *re_head, *re_merge;
		int i;
1521 1522
		path_list_clear(&current_file_set, 1);
		path_list_clear(&current_directory_set, 1);
1523 1524
		get_files_dirs(head);
		get_files_dirs(merge);
1525

1526
		entries = get_unmerged();
1527 1528
		re_head  = get_renames(head, common, head, merge, entries);
		re_merge = get_renames(merge, common, head, merge, entries);
1529
		clean = process_renames(re_head, re_merge,
1530
				branch1, branch2);
1531
		for (i = 0; i < entries->nr; i++) {
1532 1533
			const char *path = entries->items[i].path;
			struct stage_data *e = entries->items[i].util;
1534 1535
			if (!e->processed
				&& !process_entry(path, e, branch1, branch2))
1536
				clean = 0;
1537 1538
		}

1539 1540 1541
		path_list_clear(re_merge, 0);
		path_list_clear(re_head, 0);
		path_list_clear(entries, 1);
1542 1543

	}
1544 1545 1546
	else
		clean = 1;

1547
	if (index_only)
D
Daniel Barkalow 已提交
1548
		*result = write_tree_from_memory();
1549

1550
	return clean;
1551 1552
}

1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563
static struct commit_list *reverse_commit_list(struct commit_list *list)
{
	struct commit_list *next = NULL, *current, *backup;
	for (current = list; current; current = backup) {
		backup = current->next;
		current->next = next;
		next = current;
	}
	return next;
}

1564 1565
/*
 * Merge the commits h1 and h2, return the resulting virtual
P
Pavel Roskin 已提交
1566
 * commit object and a flag indicating the cleanness of the merge.
1567
 */
D
Daniel Barkalow 已提交
1568 1569 1570 1571 1572 1573
int merge_recursive(struct commit *h1,
		    struct commit *h2,
		    const char *branch1,
		    const char *branch2,
		    struct commit_list *ca,
		    struct commit **result)
1574
{
1575
	struct commit_list *iter;
1576
	struct commit *merged_common_ancestors;
1577
	struct tree *mrtree = mrtree;
1578
	int clean;
1579

1580 1581 1582 1583 1584
	if (show(4)) {
		output(4, "Merging:");
		output_commit_title(h1);
		output_commit_title(h2);
	}
1585

1586 1587 1588 1589
	if (!ca) {
		ca = get_merge_bases(h1, h2, 1);
		ca = reverse_commit_list(ca);
	}
1590

1591 1592 1593 1594 1595
	if (show(5)) {
		output(5, "found %u common ancestor(s):", commit_list_count(ca));
		for (iter = ca; iter; iter = iter->next)
			output_commit_title(iter->item);
	}
1596

1597
	merged_common_ancestors = pop_commit(&ca);
1598 1599 1600 1601 1602 1603
	if (merged_common_ancestors == NULL) {
		/* if there is no common ancestor, make an empty tree */
		struct tree *tree = xcalloc(1, sizeof(struct tree));

		tree->object.parsed = 1;
		tree->object.type = OBJ_TREE;
1604
		pretend_sha1_file(NULL, 0, OBJ_TREE, tree->object.sha1);
1605 1606
		merged_common_ancestors = make_virtual_commit(tree, "ancestor");
	}
1607 1608

	for (iter = ca; iter; iter = iter->next) {
1609
		call_depth++;
1610 1611 1612
		/*
		 * When the merge fails, the result contains files
		 * with conflict markers. The cleanness flag is
P
Pavel Roskin 已提交
1613 1614
		 * ignored, it was never actually used, as result of
		 * merge_trees has always overwritten it: the committed
1615 1616
		 * "conflicts" were already resolved.
		 */
1617
		discard_cache();
D
Daniel Barkalow 已提交
1618 1619 1620 1621 1622
		merge_recursive(merged_common_ancestors, iter->item,
				"Temporary merge branch 1",
				"Temporary merge branch 2",
				NULL,
				&merged_common_ancestors);
1623
		call_depth--;
1624

1625
		if (!merged_common_ancestors)
1626 1627 1628
			die("merge returned no commit");
	}

1629
	discard_cache();
1630
	if (!call_depth) {
1631
		read_cache();
1632
		index_only = 0;
1633
	} else
1634 1635
		index_only = 1;

1636
	clean = merge_trees(h1->tree, h2->tree, merged_common_ancestors->tree,
1637
			    branch1, branch2, &mrtree);
1638

1639
	if (index_only) {
1640 1641 1642
		*result = make_virtual_commit(mrtree, "merged tree");
		commit_list_insert(h1, &(*result)->parents);
		commit_list_insert(h2, &(*result)->parents->next);
1643
	}
1644
	flush_output();
1645
	return clean;
1646 1647
}

1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659
static const char *better_branch_name(const char *branch)
{
	static char githead_env[8 + 40 + 1];
	char *name;

	if (strlen(branch) != 40)
		return branch;
	sprintf(githead_env, "GITHEAD_%s", branch);
	name = getenv(githead_env);
	return name ? name : branch;
}

1660 1661 1662 1663 1664 1665 1666 1667
static struct commit *get_ref(const char *ref)
{
	unsigned char sha1[20];
	struct object *object;

	if (get_sha1(ref, sha1))
		die("Could not resolve ref '%s'", ref);
	object = deref_tag(parse_object(sha1), ref, strlen(ref));
1668 1669
	if (!object)
		return NULL;
1670 1671 1672
	if (object->type == OBJ_TREE)
		return make_virtual_commit((struct tree*)object,
			better_branch_name(ref));
1673
	if (object->type != OBJ_COMMIT)
1674 1675 1676 1677 1678 1679
		return NULL;
	if (parse_commit((struct commit *)object))
		die("Could not parse commit '%s'", sha1_to_hex(object->sha1));
	return (struct commit *)object;
}

1680 1681 1682 1683 1684 1685
static int merge_config(const char *var, const char *value)
{
	if (!strcasecmp(var, "merge.verbosity")) {
		verbosity = git_config_int(var, value);
		return 0;
	}
1686 1687 1688 1689
	if (!strcasecmp(var, "diff.renamelimit")) {
		rename_limit = git_config_int(var, value);
		return 0;
	}
1690 1691 1692
	return git_default_config(var, value);
}

D
Daniel Barkalow 已提交
1693
int cmd_merge_recursive(int argc, const char **argv, const char *prefix)
1694
{
1695
	static const char *bases[20];
1696
	static unsigned bases_count = 0;
1697 1698 1699
	int i, clean;
	const char *branch1, *branch2;
	struct commit *result, *h1, *h2;
1700 1701 1702
	struct commit_list *ca = NULL;
	struct lock_file *lock = xcalloc(1, sizeof(struct lock_file));
	int index_fd;
1703

J
Junio C Hamano 已提交
1704 1705 1706 1707 1708 1709 1710
	if (argv[0]) {
		int namelen = strlen(argv[0]);
		if (8 < namelen &&
		    !strcmp(argv[0] + namelen - 8, "-subtree"))
			subtree_merge = 1;
	}

1711 1712 1713
	git_config(merge_config);
	if (getenv("GIT_MERGE_VERBOSITY"))
		verbosity = strtol(getenv("GIT_MERGE_VERBOSITY"), NULL, 10);
1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725

	if (argc < 4)
		die("Usage: %s <base>... -- <head> <remote> ...\n", argv[0]);

	for (i = 1; i < argc; ++i) {
		if (!strcmp(argv[i], "--"))
			break;
		if (bases_count < sizeof(bases)/sizeof(*bases))
			bases[bases_count++] = argv[i];
	}
	if (argc - i != 3) /* "--" "<head>" "<remote>" */
		die("Not handling anything other than two heads merge.");
1726
	if (verbosity >= 5)
1727
		buffer_output = 0;
1728 1729 1730 1731

	branch1 = argv[++i];
	branch2 = argv[++i];

1732 1733
	h1 = get_ref(branch1);
	h2 = get_ref(branch2);
1734

1735 1736
	branch1 = better_branch_name(branch1);
	branch2 = better_branch_name(branch2);
1737

1738 1739
	if (show(3))
		printf("Merging %s with %s\n", branch1, branch2);
1740

1741
	index_fd = hold_locked_index(lock, 1);
1742

1743 1744 1745 1746
	for (i = 0; i < bases_count; i++) {
		struct commit *ancestor = get_ref(bases[i]);
		ca = commit_list_insert(ancestor, &ca);
	}
D
Daniel Barkalow 已提交
1747
	clean = merge_recursive(h1, h2, branch1, branch2, ca, &result);
1748 1749 1750

	if (active_cache_changed &&
	    (write_cache(index_fd, active_cache, active_nr) ||
B
Brandon Casey 已提交
1751
	     commit_locked_index(lock)))
1752
			die ("unable to write %s", get_index_file());
1753

1754
	return clean ? 0: 1;
1755
}