2 * GIT - The information manager from hell
4 * Copyright (C) Linus Torvalds, 2005
11 static int unpack_tree(unsigned char *sha1
)
17 buffer
= read_object_with_reference(sha1
, "tree", &size
, NULL
);
20 ret
= read_tree(buffer
, size
, stage
);
25 static int path_matches(struct cache_entry
*a
, struct cache_entry
*b
)
27 int len
= ce_namelen(a
);
28 return ce_namelen(b
) == len
&&
29 !memcmp(a
->name
, b
->name
, len
);
32 static int same(struct cache_entry
*a
, struct cache_entry
*b
)
34 return a
->ce_mode
== b
->ce_mode
&&
35 !memcmp(a
->sha1
, b
->sha1
, 20);
40 * This removes all trivial merges that don't change the tree
41 * and collapses them to state 0.
43 static struct cache_entry
*merge_entries(struct cache_entry
*a
,
44 struct cache_entry
*b
,
45 struct cache_entry
*c
)
48 * Ok, all three entries describe the same
49 * filename, but maybe the contents or file
52 * The trivial cases end up being the ones where two
53 * out of three files are the same:
54 * - both destinations the same, trivially take either
55 * - one of the destination versions hasn't changed,
58 * The "all entries exactly the same" case falls out as
59 * a special case of any of the "two same" cases.
61 * Here "a" is "original", and "b" and "c" are the two
62 * trees we are merging.
76 * When a CE gets turned into an unmerged entry, we
77 * want it to be up-to-date
79 static void verify_uptodate(struct cache_entry
*ce
)
83 if (!lstat(ce
->name
, &st
)) {
84 unsigned changed
= ce_match_stat(ce
, &st
);
91 die("Entry '%s' not uptodate. Cannot merge.", ce
->name
);
95 * If the old tree contained a CE that isn't even in the
96 * result, that's always a problem, regardless of whether
97 * it's up-to-date or not (ie it can be a file that we
98 * have updated but not committed yet).
100 static void reject_merge(struct cache_entry
*ce
)
102 die("Entry '%s' would be overwritten by merge. Cannot merge.", ce
->name
);
105 static int merged_entry(struct cache_entry
*merge
, struct cache_entry
*old
, struct cache_entry
**dst
)
107 merge
->ce_flags
|= htons(CE_UPDATE
);
110 * See if we can re-use the old CE directly?
111 * That way we get the uptodate stat info.
113 * This also removes the UPDATE flag on
116 if (same(old
, merge
)) {
119 verify_uptodate(old
);
122 merge
->ce_flags
&= ~htons(CE_STAGEMASK
);
127 static int threeway_merge(struct cache_entry
*stages
[4], struct cache_entry
**dst
)
129 struct cache_entry
*old
= stages
[0];
130 struct cache_entry
*a
= stages
[1], *b
= stages
[2], *c
= stages
[3];
131 struct cache_entry
*merge
;
135 * If we have an entry in the index cache ("old"), then we want
136 * to make sure that it matches any entries in stage 2 ("first
140 if (!b
|| !same(old
, b
))
143 merge
= merge_entries(a
, b
, c
);
145 return merged_entry(merge
, old
, dst
);
147 verify_uptodate(old
);
149 if (a
) { *dst
++ = a
; count
++; }
150 if (b
) { *dst
++ = b
; count
++; }
151 if (c
) { *dst
++ = c
; count
++; }
159 * - every current entry has to match the old tree
160 * - if the current entry matches the new tree, we leave it
161 * as-is. Otherwise we require that it be up-to-date.
163 static int twoway_merge(struct cache_entry
**src
, struct cache_entry
**dst
)
165 struct cache_entry
*old
= src
[0];
166 struct cache_entry
*a
= src
[1], *b
= src
[2];
172 if (!a
|| !same(old
, a
))
176 return merged_entry(b
, old
, dst
);
178 verify_uptodate(old
);
186 * - take the stat information from stage0, take the data from stage1
188 static int oneway_merge(struct cache_entry
**src
, struct cache_entry
**dst
)
190 struct cache_entry
*old
= src
[0];
191 struct cache_entry
*a
= src
[1];
193 if (src
[2] || src
[3])
198 if (old
&& same(old
, a
))
200 a
->ce_flags
&= ~htons(CE_STAGEMASK
);
205 static void check_updates(struct cache_entry
**src
, int nr
)
207 static struct checkout state
= {
213 unsigned short mask
= htons(CE_UPDATE
);
215 struct cache_entry
*ce
= *src
++;
216 if (ce
->ce_flags
& mask
) {
217 ce
->ce_flags
&= ~mask
;
219 checkout_entry(ce
, &state
);
224 static void merge_cache(struct cache_entry
**src
, int nr
, int (*fn
)(struct cache_entry
**, struct cache_entry
**))
226 struct cache_entry
**dst
= src
;
230 struct cache_entry
*name
, *ce
, *stages
[4] = { NULL
, };
234 int stage
= ce_stage(ce
);
240 if (!path_matches(ce
, name
))
244 entries
= fn(stages
, dst
);
248 active_nr
+= entries
;
250 check_updates(active_cache
, active_nr
);
253 static char *read_tree_usage
= "git-read-tree (<sha> | -m <sha1> [<sha2> [<sha3>]])";
255 static struct cache_file cache_file
;
257 int main(int argc
, char **argv
)
260 unsigned char sha1
[20];
262 newfd
= hold_index_file_for_update(&cache_file
, get_index_file());
264 die("unable to create new cachefile");
267 for (i
= 1; i
< argc
; i
++) {
268 const char *arg
= argv
[i
];
270 /* "-u" means "update", meaning that a merge will update the working directory */
271 if (!strcmp(arg
, "-u")) {
276 /* "-m" stands for "merge", meaning we start in stage 1 */
277 if (!strcmp(arg
, "-m")) {
280 die("-m needs to come first");
282 for (i
= 0; i
< active_nr
; i
++) {
283 if (ce_stage(active_cache
[i
]))
284 die("you need to resolve your current index first");
290 if (get_sha1(arg
, sha1
) < 0)
291 usage(read_tree_usage
);
293 usage(read_tree_usage
);
294 if (unpack_tree(sha1
) < 0)
295 die("failed to unpack tree object %s", arg
);
300 case 4: /* Three-way merge */
301 merge_cache(active_cache
, active_nr
, threeway_merge
);
303 case 3: /* Update from one tree to another */
304 merge_cache(active_cache
, active_nr
, twoway_merge
);
305 check_updates(active_cache
, active_nr
);
307 case 2: /* Just read a tree, merge with old cache contents */
308 merge_cache(active_cache
, active_nr
, oneway_merge
);
311 die("just how do you expect me to merge %d trees?", stage
-1);
314 if (write_cache(newfd
, active_cache
, active_nr
) ||
315 commit_index_file(&cache_file
))
316 die("unable to write new index file");