]> git.ipfire.org Git - thirdparty/git.git/blame - unpack-trees.c
sparse-checkout: avoid using internal API of unpack-trees, take 2
[thirdparty/git.git] / unpack-trees.c
CommitLineData
16da134b 1#include "cache.h"
dbbcd44f 2#include "strvec.h"
33028713 3#include "repository.h"
b2141fc1 4#include "config.h"
f8a9d428 5#include "dir.h"
16da134b
JS
6#include "tree.h"
7#include "tree-walk.h"
076b0adc 8#include "cache-tree.h"
16da134b 9#include "unpack-trees.h"
96a02f8f 10#include "progress.h"
0cf73755 11#include "refs.h"
06f33c17 12#include "attr.h"
5fc2fc8f 13#include "split-index.h"
0f329b9a 14#include "sparse-index.h"
a7bc845a
SB
15#include "submodule.h"
16#include "submodule-config.h"
883e248b 17#include "fsmonitor.h"
cbd53a21 18#include "object-store.h"
b14ed5ad 19#include "promisor-remote.h"
d052cc03 20#include "entry.h"
04155bda 21#include "parallel-checkout.h"
16da134b 22
8ccba008
JH
23/*
24 * Error messages expected by scripts out of plumbing commands such as
25 * read-tree. Non-scripted Porcelain is not required to use these messages
26 * and in fact are encouraged to reword them to better suit their particular
23cbf11b 27 * situation better. See how "git checkout" and "git merge" replaces
dc1166e6 28 * them using setup_unpack_trees_porcelain(), for example.
8ccba008 29 */
6271d77c 30static const char *unpack_plumbing_errors[NB_UNPACK_TREES_WARNING_TYPES] = {
08353ebb 31 /* ERROR_WOULD_OVERWRITE */
8ccba008
JH
32 "Entry '%s' would be overwritten by merge. Cannot merge.",
33
08353ebb 34 /* ERROR_NOT_UPTODATE_FILE */
8ccba008
JH
35 "Entry '%s' not uptodate. Cannot merge.",
36
08353ebb 37 /* ERROR_NOT_UPTODATE_DIR */
8ccba008
JH
38 "Updating '%s' would lose untracked files in it",
39
b817e545
EN
40 /* ERROR_CWD_IN_THE_WAY */
41 "Refusing to remove '%s' since it is the current working directory.",
42
08402b04
MM
43 /* ERROR_WOULD_LOSE_UNTRACKED_OVERWRITTEN */
44 "Untracked working tree file '%s' would be overwritten by merge.",
8ccba008 45
08402b04
MM
46 /* ERROR_WOULD_LOSE_UNTRACKED_REMOVED */
47 "Untracked working tree file '%s' would be removed by merge.",
8ccba008 48
08353ebb 49 /* ERROR_BIND_OVERLAP */
8ccba008 50 "Entry '%s' overlaps with '%s'. Cannot bind.",
e800ec9d 51
cd002c15
EN
52 /* ERROR_WOULD_LOSE_SUBMODULE */
53 "Submodule '%s' cannot checkout new HEAD.",
e800ec9d 54
6271d77c
EN
55 /* NB_UNPACK_TREES_ERROR_TYPES; just a meta value */
56 "",
08402b04 57
1ac83f42 58 /* WARNING_SPARSE_NOT_UPTODATE_FILE */
22ab0b37 59 "Path '%s' not uptodate; will not remove from working tree.",
a7bc845a 60
ebb568b9
EN
61 /* WARNING_SPARSE_UNMERGED_FILE */
62 "Path '%s' unmerged; will not remove from working tree.",
63
1ac83f42 64 /* WARNING_SPARSE_ORPHANED_NOT_OVERWRITTEN */
22ab0b37 65 "Path '%s' already present; will not overwrite with sparse update.",
8ccba008
JH
66};
67
08353ebb
MM
68#define ERRORMSG(o,type) \
69 ( ((o) && (o)->msgs[(type)]) \
70 ? ((o)->msgs[(type)]) \
71 : (unpack_plumbing_errors[(type)]) )
8ccba008 72
4002ec3d 73static const char *super_prefixed(const char *path, const char *super_prefix)
3d415425
SB
74{
75 /*
76 * It is necessary and sufficient to have two static buffers
77 * here, as the return value of this function is fed to
78 * error() using the unpack_*_errors[] templates we see above.
79 */
80 static struct strbuf buf[2] = {STRBUF_INIT, STRBUF_INIT};
81 static int super_prefix_len = -1;
82 static unsigned idx = ARRAY_SIZE(buf) - 1;
83
84 if (super_prefix_len < 0) {
3d415425
SB
85 if (!super_prefix) {
86 super_prefix_len = 0;
87 } else {
88 int i;
89 for (i = 0; i < ARRAY_SIZE(buf); i++)
90 strbuf_addstr(&buf[i], super_prefix);
91 super_prefix_len = buf[0].len;
92 }
93 }
94
95 if (!super_prefix_len)
96 return path;
97
98 if (++idx >= ARRAY_SIZE(buf))
99 idx = 0;
100
101 strbuf_setlen(&buf[idx], super_prefix_len);
102 strbuf_addstr(&buf[idx], path);
103
104 return buf[idx].buf;
105}
106
e294030f
MM
107void setup_unpack_trees_porcelain(struct unpack_trees_options *opts,
108 const char *cmd)
dc1166e6 109{
7980872d 110 int i;
e294030f 111 const char **msgs = opts->msgs;
dc1166e6 112 const char *msg;
fa3f60b7 113
c972bf4c 114 strvec_init(&opts->msgs_to_free);
1c41d280 115
2e3926b9 116 if (!strcmp(cmd, "checkout"))
ed9bff08 117 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 118 ? _("Your local changes to the following files would be overwritten by checkout:\n%%s"
c2691e2a 119 "Please commit your changes or stash them before you switch branches.")
2e3926b9
VA
120 : _("Your local changes to the following files would be overwritten by checkout:\n%%s");
121 else if (!strcmp(cmd, "merge"))
ed9bff08 122 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 123 ? _("Your local changes to the following files would be overwritten by merge:\n%%s"
c2691e2a 124 "Please commit your changes or stash them before you merge.")
2e3926b9 125 : _("Your local changes to the following files would be overwritten by merge:\n%%s");
dc1166e6 126 else
ed9bff08 127 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 128 ? _("Your local changes to the following files would be overwritten by %s:\n%%s"
c2691e2a 129 "Please commit your changes or stash them before you %s.")
2e3926b9 130 : _("Your local changes to the following files would be overwritten by %s:\n%%s");
fa3f60b7 131 msgs[ERROR_WOULD_OVERWRITE] = msgs[ERROR_NOT_UPTODATE_FILE] =
c972bf4c 132 strvec_pushf(&opts->msgs_to_free, msg, cmd, cmd);
dc1166e6
MM
133
134 msgs[ERROR_NOT_UPTODATE_DIR] =
584f99c8 135 _("Updating the following directories would lose untracked files in them:\n%s");
dc1166e6 136
b817e545
EN
137 msgs[ERROR_CWD_IN_THE_WAY] =
138 _("Refusing to remove the current working directory:\n%s");
139
2e3926b9 140 if (!strcmp(cmd, "checkout"))
ed9bff08 141 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 142 ? _("The following untracked working tree files would be removed by checkout:\n%%s"
c2691e2a 143 "Please move or remove them before you switch branches.")
2e3926b9
VA
144 : _("The following untracked working tree files would be removed by checkout:\n%%s");
145 else if (!strcmp(cmd, "merge"))
ed9bff08 146 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 147 ? _("The following untracked working tree files would be removed by merge:\n%%s"
c2691e2a 148 "Please move or remove them before you merge.")
2e3926b9 149 : _("The following untracked working tree files would be removed by merge:\n%%s");
dc1166e6 150 else
ed9bff08 151 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 152 ? _("The following untracked working tree files would be removed by %s:\n%%s"
c2691e2a 153 "Please move or remove them before you %s.")
2e3926b9 154 : _("The following untracked working tree files would be removed by %s:\n%%s");
1c41d280 155 msgs[ERROR_WOULD_LOSE_UNTRACKED_REMOVED] =
c972bf4c 156 strvec_pushf(&opts->msgs_to_free, msg, cmd, cmd);
2e3926b9
VA
157
158 if (!strcmp(cmd, "checkout"))
ed9bff08 159 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 160 ? _("The following untracked working tree files would be overwritten by checkout:\n%%s"
c2691e2a 161 "Please move or remove them before you switch branches.")
2e3926b9
VA
162 : _("The following untracked working tree files would be overwritten by checkout:\n%%s");
163 else if (!strcmp(cmd, "merge"))
ed9bff08 164 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 165 ? _("The following untracked working tree files would be overwritten by merge:\n%%s"
c2691e2a 166 "Please move or remove them before you merge.")
2e3926b9
VA
167 : _("The following untracked working tree files would be overwritten by merge:\n%%s");
168 else
ed9bff08 169 msg = advice_enabled(ADVICE_COMMIT_BEFORE_MERGE)
2e3926b9 170 ? _("The following untracked working tree files would be overwritten by %s:\n%%s"
c2691e2a 171 "Please move or remove them before you %s.")
2e3926b9 172 : _("The following untracked working tree files would be overwritten by %s:\n%%s");
1c41d280 173 msgs[ERROR_WOULD_LOSE_UNTRACKED_OVERWRITTEN] =
c972bf4c 174 strvec_pushf(&opts->msgs_to_free, msg, cmd, cmd);
dc1166e6
MM
175
176 /*
177 * Special case: ERROR_BIND_OVERLAP refers to a pair of paths, we
178 * cannot easily display it as a list.
179 */
ed47fdf7 180 msgs[ERROR_BIND_OVERLAP] = _("Entry '%s' overlaps with '%s'. Cannot bind.");
dc1166e6 181
a7bc845a 182 msgs[ERROR_WOULD_LOSE_SUBMODULE] =
6362ed0b 183 _("Cannot update submodule:\n%s");
5e65ee35 184
1ac83f42 185 msgs[WARNING_SPARSE_NOT_UPTODATE_FILE] =
22ab0b37 186 _("The following paths are not up to date and were left despite sparse patterns:\n%s");
ebb568b9
EN
187 msgs[WARNING_SPARSE_UNMERGED_FILE] =
188 _("The following paths are unmerged and were left despite sparse patterns:\n%s");
1ac83f42 189 msgs[WARNING_SPARSE_ORPHANED_NOT_OVERWRITTEN] =
22ab0b37 190 _("The following paths were already present and thus not updated despite sparse patterns:\n%s");
5e65ee35
MM
191
192 opts->show_all_errors = 1;
7980872d
CB
193 /* rejected paths may not have a static buffer */
194 for (i = 0; i < ARRAY_SIZE(opts->unpack_rejects); i++)
195 opts->unpack_rejects[i].strdup_strings = 1;
dc1166e6
MM
196}
197
1c41d280
198void clear_unpack_trees_porcelain(struct unpack_trees_options *opts)
199{
c972bf4c 200 strvec_clear(&opts->msgs_to_free);
1c41d280
201 memset(opts->msgs, 0, sizeof(opts->msgs));
202}
203
46169180 204static int do_add_entry(struct unpack_trees_options *o, struct cache_entry *ce,
6ff264ee 205 unsigned int set, unsigned int clear)
b48d5a05 206{
419a597f 207 clear |= CE_HASHED;
34110cd4 208
700e66d6
NTND
209 if (set & CE_REMOVE)
210 set |= CE_WT_REMOVE;
211
6ff264ee 212 ce->ce_flags = (ce->ce_flags & ~clear) | set;
46169180
JK
213 return add_index_entry(&o->result, ce,
214 ADD_CACHE_OK_TO_ADD | ADD_CACHE_OK_TO_REPLACE);
6ff264ee
RS
215}
216
a33bd4d3
RS
217static void add_entry(struct unpack_trees_options *o,
218 const struct cache_entry *ce,
219 unsigned int set, unsigned int clear)
220{
8e72d675 221 do_add_entry(o, dup_cache_entry(ce, &o->result), set, clear);
b48d5a05
LT
222}
223
e6c111b4
MM
224/*
225 * add error messages on path <path>
226 * corresponding to the type <e> with the message <msg>
227 * indicating if it should be display in porcelain or not
228 */
229static int add_rejected_path(struct unpack_trees_options *o,
230 enum unpack_trees_error_types e,
231 const char *path)
232{
b165fac8 233 if (o->quiet)
191e9d2c
NTND
234 return -1;
235
e6c111b4 236 if (!o->show_all_errors)
4002ec3d
ÆAB
237 return error(ERRORMSG(o, e), super_prefixed(path,
238 o->super_prefix));
e6c111b4
MM
239
240 /*
241 * Otherwise, insert in a list for future display by
6271d77c 242 * display_(error|warning)_msgs()
e6c111b4 243 */
7980872d 244 string_list_append(&o->unpack_rejects[e], path);
e6c111b4
MM
245 return -1;
246}
247
e6c111b4
MM
248/*
249 * display all the error messages stored in a nice way
250 */
251static void display_error_msgs(struct unpack_trees_options *o)
252{
6271d77c
EN
253 int e;
254 unsigned error_displayed = 0;
e6c111b4 255 for (e = 0; e < NB_UNPACK_TREES_ERROR_TYPES; e++) {
7980872d 256 struct string_list *rejects = &o->unpack_rejects[e];
6271d77c 257
7980872d 258 if (rejects->nr > 0) {
6271d77c 259 int i;
e6c111b4 260 struct strbuf path = STRBUF_INIT;
6271d77c
EN
261
262 error_displayed = 1;
7980872d
CB
263 for (i = 0; i < rejects->nr; i++)
264 strbuf_addf(&path, "\t%s\n", rejects->items[i].string);
4002ec3d
ÆAB
265 error(ERRORMSG(o, e), super_prefixed(path.buf,
266 o->super_prefix));
e6c111b4 267 strbuf_release(&path);
e6c111b4 268 }
7980872d 269 string_list_clear(rejects, 0);
e6c111b4 270 }
6271d77c 271 if (error_displayed)
ed47fdf7 272 fprintf(stderr, _("Aborting\n"));
e6c111b4
MM
273}
274
6271d77c
EN
275/*
276 * display all the warning messages stored in a nice way
277 */
278static void display_warning_msgs(struct unpack_trees_options *o)
279{
280 int e;
281 unsigned warning_displayed = 0;
282 for (e = NB_UNPACK_TREES_ERROR_TYPES + 1;
283 e < NB_UNPACK_TREES_WARNING_TYPES; e++) {
284 struct string_list *rejects = &o->unpack_rejects[e];
285
286 if (rejects->nr > 0) {
287 int i;
288 struct strbuf path = STRBUF_INIT;
289
290 warning_displayed = 1;
291 for (i = 0; i < rejects->nr; i++)
292 strbuf_addf(&path, "\t%s\n", rejects->items[i].string);
4002ec3d
ÆAB
293 warning(ERRORMSG(o, e), super_prefixed(path.buf,
294 o->super_prefix));
6271d77c
EN
295 strbuf_release(&path);
296 }
297 string_list_clear(rejects, 0);
298 }
299 if (warning_displayed)
300 fprintf(stderr, _("After fixing the above paths, you may want to run `git sparse-checkout reapply`.\n"));
301}
a7bc845a
SB
302static int check_submodule_move_head(const struct cache_entry *ce,
303 const char *old_id,
304 const char *new_id,
305 struct unpack_trees_options *o)
306{
cd279e2e 307 unsigned flags = SUBMODULE_MOVE_HEAD_DRY_RUN;
a7bc845a 308 const struct submodule *sub = submodule_from_ce(ce);
7463e2ec 309
a7bc845a
SB
310 if (!sub)
311 return 0;
312
cd279e2e
SB
313 if (o->reset)
314 flags |= SUBMODULE_MOVE_HEAD_FORCE;
315
4002ec3d
ÆAB
316 if (submodule_move_head(ce->name, o->super_prefix, old_id, new_id,
317 flags))
191e9d2c 318 return add_rejected_path(o, ERROR_WOULD_LOSE_SUBMODULE, ce->name);
7463e2ec 319 return 0;
a7bc845a
SB
320}
321
33028713 322/*
15beaaa3 323 * Perform the loading of the repository's gitmodules file. This function is
33028713 324 * used by 'check_update()' to perform loading of the gitmodules file in two
15beaaa3 325 * different situations:
33028713
BW
326 * (1) before removing entries from the working tree if the gitmodules file has
327 * been marked for removal. This situation is specified by 'state' == NULL.
328 * (2) before checking out entries to the working tree if the gitmodules file
329 * has been marked for update. This situation is specified by 'state' != NULL.
330 */
331static void load_gitmodules_file(struct index_state *index,
332 struct checkout *state)
a7bc845a 333{
33028713
BW
334 int pos = index_name_pos(index, GITMODULES_FILE, strlen(GITMODULES_FILE));
335
336 if (pos >= 0) {
337 struct cache_entry *ce = index->cache[pos];
338 if (!state && ce->ce_flags & CE_WT_REMOVE) {
d7992421 339 repo_read_gitmodules(the_repository, 0);
33028713 340 } else if (state && (ce->ce_flags & CE_UPDATE)) {
f793b895 341 submodule_free(the_repository);
0f086e6d 342 checkout_entry(ce, state, NULL, NULL);
d7992421 343 repo_read_gitmodules(the_repository, 0);
a7bc845a
SB
344 }
345 }
346}
347
6c34239d
EN
348static struct progress *get_progress(struct unpack_trees_options *o,
349 struct index_state *index)
16da134b 350{
96a02f8f 351 unsigned cnt = 0, total = 0;
384f1a16
SB
352
353 if (!o->update || !o->verbose_update)
354 return NULL;
355
356 for (; cnt < index->cache_nr; cnt++) {
357 const struct cache_entry *ce = index->cache[cnt];
358 if (ce->ce_flags & (CE_UPDATE | CE_WT_REMOVE))
359 total++;
360 }
361
328c6cb8 362 return start_delayed_progress(_("Updating files"), total);
384f1a16
SB
363}
364
b878579a
NTND
365static void setup_collided_checkout_detection(struct checkout *state,
366 struct index_state *index)
367{
368 int i;
369
370 state->clone = 1;
371 for (i = 0; i < index->cache_nr; i++)
372 index->cache[i]->ce_flags &= ~CE_MATCHED;
373}
374
375static void report_collided_checkout(struct index_state *index)
376{
377 struct string_list list = STRING_LIST_INIT_NODUP;
378 int i;
379
380 for (i = 0; i < index->cache_nr; i++) {
381 struct cache_entry *ce = index->cache[i];
382
383 if (!(ce->ce_flags & CE_MATCHED))
384 continue;
385
386 string_list_append(&list, ce->name);
387 ce->ce_flags &= ~CE_MATCHED;
388 }
389
390 list.cmp = fspathcmp;
391 string_list_sort(&list);
392
393 if (list.nr) {
394 warning(_("the following paths have collided (e.g. case-sensitive paths\n"
395 "on a case-insensitive filesystem) and only one from the same\n"
396 "colliding group is in the working tree:\n"));
397
398 for (i = 0; i < list.nr; i++)
399 fprintf(stderr, " '%s'\n", list.items[i].string);
400 }
401
402 string_list_clear(&list, 0);
403}
404
b2896d27
JT
405static int must_checkout(const struct cache_entry *ce)
406{
407 return ce->ce_flags & CE_UPDATE;
408}
409
b0a5a12a
EN
410static int check_updates(struct unpack_trees_options *o,
411 struct index_state *index)
384f1a16
SB
412{
413 unsigned cnt = 0;
30ac275b 414 int errs = 0;
07f967ad 415 struct progress *progress;
30ac275b 416 struct checkout state = CHECKOUT_INIT;
7531e4b6 417 int i, pc_workers, pc_threshold;
16da134b 418
0d1ed596 419 trace_performance_enter();
4002ec3d 420 state.super_prefix = o->super_prefix;
30ac275b
SB
421 state.force = 1;
422 state.quiet = 1;
423 state.refresh_cache = 1;
424 state.istate = index;
13e7ed6a 425 clone_checkout_metadata(&state.meta, &o->meta, NULL);
16da134b 426
26f924d5
EN
427 if (!o->update || o->dry_run) {
428 remove_marked_cache_entries(index, 0);
429 trace_performance_leave("check_updates");
430 return 0;
431 }
432
b878579a
NTND
433 if (o->clone)
434 setup_collided_checkout_detection(&state, index);
435
6c34239d 436 progress = get_progress(o, index);
16da134b 437
22539ec3
MT
438 /* Start with clean cache to avoid using any possibly outdated info. */
439 invalidate_lstat_cache();
440
26f924d5 441 git_attr_set_direction(GIT_ATTR_CHECKOUT);
33028713 442
26f924d5 443 if (should_update_submodules())
33028713
BW
444 load_gitmodules_file(index, NULL);
445
34110cd4 446 for (i = 0; i < index->cache_nr; i++) {
9c5e6c80 447 const struct cache_entry *ce = index->cache[i];
16da134b 448
e663db2f
NTND
449 if (ce->ce_flags & CE_WT_REMOVE) {
450 display_progress(progress, ++cnt);
4002ec3d 451 unlink_entry(ce, o->super_prefix);
e663db2f 452 }
1fa6ead4 453 }
26f924d5 454
6fdc2057 455 remove_marked_cache_entries(index, 0);
78478927 456 remove_scheduled_dirs();
1fa6ead4 457
26f924d5 458 if (should_update_submodules())
33028713 459 load_gitmodules_file(index, &state);
a7bc845a 460
b2896d27 461 if (has_promisor_remote())
c0c578b3
JT
462 /*
463 * Prefetch the objects that are to be checked out in the loop
464 * below.
465 */
b2896d27 466 prefetch_cache_entries(index, must_checkout);
04155bda 467
7531e4b6
MT
468 get_parallel_checkout_configs(&pc_workers, &pc_threshold);
469
04155bda 470 enable_delayed_checkout(&state);
7531e4b6
MT
471 if (pc_workers > 1)
472 init_parallel_checkout();
1fa6ead4
LT
473 for (i = 0; i < index->cache_nr; i++) {
474 struct cache_entry *ce = index->cache[i];
475
b2896d27 476 if (must_checkout(ce)) {
1c4d6f46
MT
477 size_t last_pc_queue_size = pc_queue_size();
478
7d782416 479 if (ce->ce_flags & CE_WT_REMOVE)
033abf97 480 BUG("both update and delete flags are set on %s",
7d782416 481 ce->name);
7a51ed66 482 ce->ce_flags &= ~CE_UPDATE;
26f924d5 483 errs |= checkout_entry(ce, &state, NULL, NULL);
1c4d6f46
MT
484
485 if (last_pc_queue_size == pc_queue_size())
486 display_progress(progress, ++cnt);
16da134b
JS
487 }
488 }
7531e4b6 489 if (pc_workers > 1)
1c4d6f46
MT
490 errs |= run_parallel_checkout(&state, pc_workers, pc_threshold,
491 progress, &cnt);
4d4fcc54 492 stop_progress(&progress);
611c7785 493 errs |= finish_delayed_checkout(&state, o->verbose_update);
26f924d5 494 git_attr_set_direction(GIT_ATTR_CHECKIN);
b878579a
NTND
495
496 if (o->clone)
497 report_collided_checkout(index);
498
0d1ed596 499 trace_performance_leave("check_updates");
c4758d3c 500 return errs != 0;
16da134b
JS
501}
502
eb9ae4b5
RS
503static int verify_uptodate_sparse(const struct cache_entry *ce,
504 struct unpack_trees_options *o);
505static int verify_absent_sparse(const struct cache_entry *ce,
506 enum unpack_trees_error_types,
507 struct unpack_trees_options *o);
e800ec9d 508
a5c446f1
NTND
509static int apply_sparse_checkout(struct index_state *istate,
510 struct cache_entry *ce,
511 struct unpack_trees_options *o)
e800ec9d
NTND
512{
513 int was_skip_worktree = ce_skip_worktree(ce);
514
2431afbf 515 if (ce->ce_flags & CE_NEW_SKIP_WORKTREE)
e800ec9d
NTND
516 ce->ce_flags |= CE_SKIP_WORKTREE;
517 else
518 ce->ce_flags &= ~CE_SKIP_WORKTREE;
078a58e8
NTND
519 if (was_skip_worktree != ce_skip_worktree(ce)) {
520 ce->ce_flags |= CE_UPDATE_IN_BASE;
883e248b 521 mark_fsmonitor_invalid(istate, ce);
a5c446f1 522 istate->cache_changed |= CE_ENTRY_CHANGED;
078a58e8 523 }
e800ec9d
NTND
524
525 /*
eec3fc03
NTND
526 * if (!was_skip_worktree && !ce_skip_worktree()) {
527 * This is perfectly normal. Move on;
528 * }
e800ec9d 529 */
eec3fc03
NTND
530
531 /*
532 * Merge strategies may set CE_UPDATE|CE_REMOVE outside checkout
533 * area as a result of ce_skip_worktree() shortcuts in
700e66d6
NTND
534 * verify_absent() and verify_uptodate().
535 * Make sure they don't modify worktree if they are already
536 * outside checkout area
eec3fc03 537 */
700e66d6
NTND
538 if (was_skip_worktree && ce_skip_worktree(ce)) {
539 ce->ce_flags &= ~CE_UPDATE;
540
541 /*
542 * By default, when CE_REMOVE is on, CE_WT_REMOVE is also
543 * on to get that file removed from both index and worktree.
544 * If that file is already outside worktree area, don't
545 * bother remove it.
546 */
547 if (ce->ce_flags & CE_REMOVE)
548 ce->ce_flags &= ~CE_WT_REMOVE;
549 }
e800ec9d
NTND
550
551 if (!was_skip_worktree && ce_skip_worktree(ce)) {
552 /*
553 * If CE_UPDATE is set, verify_uptodate() must be called already
554 * also stat info may have lost after merged_entry() so calling
555 * verify_uptodate() again may fail
556 */
3cc7c504
EN
557 if (!(ce->ce_flags & CE_UPDATE) &&
558 verify_uptodate_sparse(ce, o)) {
559 ce->ce_flags &= ~CE_SKIP_WORKTREE;
e800ec9d 560 return -1;
3cc7c504 561 }
e800ec9d 562 ce->ce_flags |= CE_WT_REMOVE;
7d782416 563 ce->ce_flags &= ~CE_UPDATE;
e800ec9d
NTND
564 }
565 if (was_skip_worktree && !ce_skip_worktree(ce)) {
1ac83f42 566 if (verify_absent_sparse(ce, WARNING_SPARSE_ORPHANED_NOT_OVERWRITTEN, o))
e800ec9d
NTND
567 return -1;
568 ce->ce_flags |= CE_UPDATE;
569 }
570 return 0;
571}
572
ebb568b9
EN
573static int warn_conflicted_path(struct index_state *istate,
574 int i,
575 struct unpack_trees_options *o)
576{
577 char *conflicting_path = istate->cache[i]->name;
578 int count = 0;
579
580 add_rejected_path(o, WARNING_SPARSE_UNMERGED_FILE, conflicting_path);
581
0eeb3be4
DS
582 /* Find out how many higher stage entries are at same path */
583 while ((++count) + i < istate->cache_nr &&
584 !strcmp(conflicting_path, istate->cache[count + i]->name))
585 ; /* do nothing */
586
ebb568b9
EN
587 return count;
588}
589
5828e835
RS
590static inline int call_unpack_fn(const struct cache_entry * const *src,
591 struct unpack_trees_options *o)
01904572 592{
34110cd4
LT
593 int ret = o->fn(src, o);
594 if (ret > 0)
01904572 595 ret = 0;
01904572
LT
596 return ret;
597}
598
da165f47
JH
599static void mark_ce_used(struct cache_entry *ce, struct unpack_trees_options *o)
600{
601 ce->ce_flags |= CE_UNPACKED;
602
603 if (o->cache_bottom < o->src_index->cache_nr &&
604 o->src_index->cache[o->cache_bottom] == ce) {
605 int bottom = o->cache_bottom;
606 while (bottom < o->src_index->cache_nr &&
607 o->src_index->cache[bottom]->ce_flags & CE_UNPACKED)
608 bottom++;
609 o->cache_bottom = bottom;
610 }
611}
612
613static void mark_all_ce_unused(struct index_state *index)
614{
615 int i;
616 for (i = 0; i < index->cache_nr; i++)
2431afbf 617 index->cache[i]->ce_flags &= ~(CE_UNPACKED | CE_ADDED | CE_NEW_SKIP_WORKTREE);
da165f47
JH
618}
619
eb9ae4b5 620static int locate_in_src_index(const struct cache_entry *ce,
da165f47
JH
621 struct unpack_trees_options *o)
622{
623 struct index_state *index = o->src_index;
624 int len = ce_namelen(ce);
625 int pos = index_name_pos(index, ce->name, len);
626 if (pos < 0)
627 pos = -1 - pos;
628 return pos;
629}
630
631/*
632 * We call unpack_index_entry() with an unmerged cache entry
633 * only in diff-index, and it wants a single callback. Skip
634 * the other unmerged entry with the same name.
635 */
636static void mark_ce_used_same_name(struct cache_entry *ce,
637 struct unpack_trees_options *o)
638{
639 struct index_state *index = o->src_index;
640 int len = ce_namelen(ce);
641 int pos;
642
643 for (pos = locate_in_src_index(ce, o); pos < index->cache_nr; pos++) {
644 struct cache_entry *next = index->cache[pos];
645 if (len != ce_namelen(next) ||
646 memcmp(ce->name, next->name, len))
647 break;
648 mark_ce_used(next, o);
649 }
650}
651
99430aa1 652static struct cache_entry *next_cache_entry(struct unpack_trees_options *o)
da165f47
JH
653{
654 const struct index_state *index = o->src_index;
655 int pos = o->cache_bottom;
656
657 while (pos < index->cache_nr) {
658 struct cache_entry *ce = index->cache[pos];
99430aa1 659 if (!(ce->ce_flags & CE_UNPACKED))
da165f47
JH
660 return ce;
661 pos++;
662 }
663 return NULL;
664}
665
9c5e6c80 666static void add_same_unmerged(const struct cache_entry *ce,
da165f47
JH
667 struct unpack_trees_options *o)
668{
669 struct index_state *index = o->src_index;
670 int len = ce_namelen(ce);
671 int pos = index_name_pos(index, ce->name, len);
672
673 if (0 <= pos)
674 die("programming error in a caller of mark_ce_used_same_name");
675 for (pos = -pos - 1; pos < index->cache_nr; pos++) {
676 struct cache_entry *next = index->cache[pos];
677 if (len != ce_namelen(next) ||
678 memcmp(ce->name, next->name, len))
679 break;
680 add_entry(o, next, 0, 0);
681 mark_ce_used(next, o);
682 }
683}
684
685static int unpack_index_entry(struct cache_entry *ce,
686 struct unpack_trees_options *o)
01904572 687{
5828e835 688 const struct cache_entry *src[MAX_UNPACK_TREES + 1] = { NULL, };
da165f47 689 int ret;
34110cd4 690
66dbfd55
GV
691 src[0] = ce;
692
da165f47 693 mark_ce_used(ce, o);
01904572
LT
694 if (ce_stage(ce)) {
695 if (o->skip_unmerged) {
34110cd4
LT
696 add_entry(o, ce, 0, 0);
697 return 0;
01904572 698 }
01904572 699 }
da165f47
JH
700 ret = call_unpack_fn(src, o);
701 if (ce_stage(ce))
702 mark_ce_used_same_name(ce, o);
703 return ret;
01904572
LT
704}
705
90553847 706static int find_cache_pos(struct traverse_info *, const char *p, size_t len);
730f7284
JH
707
708static void restore_cache_bottom(struct traverse_info *info, int bottom)
709{
710 struct unpack_trees_options *o = info->data;
711
712 if (o->diff_index_cached)
713 return;
714 o->cache_bottom = bottom;
715}
716
717static int switch_cache_bottom(struct traverse_info *info)
718{
719 struct unpack_trees_options *o = info->data;
720 int ret, pos;
721
722 if (o->diff_index_cached)
723 return 0;
724 ret = o->cache_bottom;
90553847 725 pos = find_cache_pos(info->prev, info->name, info->namelen);
730f7284
JH
726
727 if (pos < -1)
728 o->cache_bottom = -2 - pos;
729 else if (pos < 0)
730 o->cache_bottom = o->src_index->cache_nr;
731 return ret;
01904572
LT
732}
733
d12a8cf0
JH
734static inline int are_same_oid(struct name_entry *name_j, struct name_entry *name_k)
735{
ea82b2a0 736 return !is_null_oid(&name_j->oid) && !is_null_oid(&name_k->oid) && oideq(&name_j->oid, &name_k->oid);
d12a8cf0
JH
737}
738
b4da3738
NTND
739static int all_trees_same_as_cache_tree(int n, unsigned long dirmask,
740 struct name_entry *names,
741 struct traverse_info *info)
742{
743 struct unpack_trees_options *o = info->data;
744 int i;
745
746 if (!o->merge || dirmask != ((1 << n) - 1))
747 return 0;
748
749 for (i = 1; i < n; i++)
750 if (!are_same_oid(names, names + i))
751 return 0;
752
753 return cache_tree_matches_traversal(o->src_index->cache_tree, names, info);
754}
755
756static int index_pos_by_traverse_info(struct name_entry *names,
757 struct traverse_info *info)
758{
759 struct unpack_trees_options *o = info->data;
c43ab062 760 struct strbuf name = STRBUF_INIT;
b4da3738
NTND
761 int pos;
762
c43ab062
JK
763 strbuf_make_traverse_path(&name, info, names->path, names->pathlen);
764 strbuf_addch(&name, '/');
765 pos = index_name_pos(o->src_index, name.buf, name.len);
13e13312
DS
766 if (pos >= 0) {
767 if (!o->src_index->sparse_index ||
768 !(o->src_index->cache[pos]->ce_flags & CE_SKIP_WORKTREE))
769 BUG("This is a directory and should not exist in index");
770 } else {
771 pos = -pos - 1;
772 }
573117df
ES
773 if (pos >= o->src_index->cache_nr ||
774 !starts_with(o->src_index->cache[pos]->name, name.buf) ||
c43ab062 775 (pos > 0 && starts_with(o->src_index->cache[pos-1]->name, name.buf)))
573117df
ES
776 BUG("pos %d doesn't point to the first entry of %s in index",
777 pos, name.buf);
c43ab062 778 strbuf_release(&name);
b4da3738
NTND
779 return pos;
780}
781
782/*
783 * Fast path if we detect that all trees are the same as cache-tree at this
5f4436a7
NTND
784 * path. We'll walk these trees in an iterative loop using cache-tree/index
785 * instead of ODB since we already know what these trees contain.
b4da3738
NTND
786 */
787static int traverse_by_cache_tree(int pos, int nr_entries, int nr_names,
b4da3738
NTND
788 struct traverse_info *info)
789{
790 struct cache_entry *src[MAX_UNPACK_TREES + 1] = { NULL, };
791 struct unpack_trees_options *o = info->data;
f1e11c65
NTND
792 struct cache_entry *tree_ce = NULL;
793 int ce_len = 0;
b4da3738
NTND
794 int i, d;
795
796 if (!o->merge)
797 BUG("We need cache-tree to do this optimization");
798
799 /*
bd6a3fd7 800 * Do what unpack_callback() and unpack_single_entry() normally
b4da3738
NTND
801 * do. But we walk all paths in an iterative loop instead.
802 *
803 * D/F conflicts and higher stage entries are not a concern
804 * because cache-tree would be invalidated and we would never
805 * get here in the first place.
806 */
807 for (i = 0; i < nr_entries; i++) {
f1e11c65 808 int new_ce_len, len, rc;
b4da3738
NTND
809
810 src[0] = o->src_index->cache[pos + i];
811
812 len = ce_namelen(src[0]);
f1e11c65
NTND
813 new_ce_len = cache_entry_size(len);
814
815 if (new_ce_len > ce_len) {
816 new_ce_len <<= 1;
817 tree_ce = xrealloc(tree_ce, new_ce_len);
818 memset(tree_ce, 0, new_ce_len);
819 ce_len = new_ce_len;
820
821 tree_ce->ce_flags = create_ce_flags(0);
822
823 for (d = 1; d <= nr_names; d++)
824 src[d] = tree_ce;
825 }
b4da3738
NTND
826
827 tree_ce->ce_mode = src[0]->ce_mode;
b4da3738
NTND
828 tree_ce->ce_namelen = len;
829 oidcpy(&tree_ce->oid, &src[0]->oid);
830 memcpy(tree_ce->name, src[0]->name, len + 1);
831
b4da3738 832 rc = call_unpack_fn((const struct cache_entry * const *)src, o);
f1e11c65
NTND
833 if (rc < 0) {
834 free(tree_ce);
b4da3738 835 return rc;
f1e11c65 836 }
b4da3738
NTND
837
838 mark_ce_used(src[0], o);
839 }
f1e11c65 840 free(tree_ce);
b4da3738
NTND
841 if (o->debug_unpack)
842 printf("Unpacked %d entries from %s to %s using cache-tree\n",
843 nr_entries,
844 o->src_index->cache[pos]->name,
845 o->src_index->cache[pos + nr_entries - 1]->name);
846 return 0;
847}
848
84563a62
JH
849static int traverse_trees_recursive(int n, unsigned long dirmask,
850 unsigned long df_conflicts,
851 struct name_entry *names,
852 struct traverse_info *info)
16da134b 853{
67022e02 854 struct unpack_trees_options *o = info->data;
730f7284 855 int i, ret, bottom;
d12a8cf0 856 int nr_buf = 0;
ca885a4f 857 struct tree_desc t[MAX_UNPACK_TREES];
1ce584b0 858 void *buf[MAX_UNPACK_TREES];
01904572
LT
859 struct traverse_info newinfo;
860 struct name_entry *p;
b4da3738
NTND
861 int nr_entries;
862
863 nr_entries = all_trees_same_as_cache_tree(n, dirmask, names, info);
864 if (nr_entries > 0) {
b4da3738
NTND
865 int pos = index_pos_by_traverse_info(names, info);
866
867 if (!o->merge || df_conflicts)
868 BUG("Wrong condition to get here buddy");
869
870 /*
871 * All entries up to 'pos' must have been processed
872 * (i.e. marked CE_UNPACKED) at this point. But to be safe,
873 * save and restore cache_bottom anyway to not miss
874 * unprocessed entries before 'pos'.
875 */
876 bottom = o->cache_bottom;
66429698 877 ret = traverse_by_cache_tree(pos, nr_entries, n, info);
b4da3738
NTND
878 o->cache_bottom = bottom;
879 return ret;
880 }
01904572
LT
881
882 p = names;
883 while (!p->mode)
884 p++;
885
886 newinfo = *info;
887 newinfo.prev = info;
40e37256 888 newinfo.pathspec = info->pathspec;
90553847
JK
889 newinfo.name = p->path;
890 newinfo.namelen = p->pathlen;
891 newinfo.mode = p->mode;
37806080 892 newinfo.pathlen = st_add3(newinfo.pathlen, tree_entry_len(p), 1);
603d2498 893 newinfo.df_conflicts |= df_conflicts;
01904572 894
d12a8cf0
JH
895 /*
896 * Fetch the tree from the ODB for each peer directory in the
897 * n commits.
898 *
899 * For 2- and 3-way traversals, we try to avoid hitting the
900 * ODB twice for the same OID. This should yield a nice speed
901 * up in checkouts and merges when the commits are similar.
902 *
903 * We don't bother doing the full O(n^2) search for larger n,
904 * because wider traversals don't happen that often and we
905 * avoid the search setup.
906 *
907 * When 2 peer OIDs are the same, we just copy the tree
908 * descriptor data. This implicitly borrows the buffer
909 * data from the earlier cell.
910 */
01904572 911 for (i = 0; i < n; i++, dirmask >>= 1) {
d12a8cf0
JH
912 if (i > 0 && are_same_oid(&names[i], &names[i - 1]))
913 t[i] = t[i - 1];
914 else if (i > 1 && are_same_oid(&names[i], &names[i - 2]))
915 t[i] = t[i - 2];
916 else {
5c377d3d 917 const struct object_id *oid = NULL;
d12a8cf0 918 if (dirmask & 1)
ea82b2a0 919 oid = &names[i].oid;
5e575807 920 buf[nr_buf++] = fill_tree_descriptor(the_repository, t + i, oid);
d12a8cf0 921 }
01904572 922 }
730f7284
JH
923
924 bottom = switch_cache_bottom(&newinfo);
67022e02 925 ret = traverse_trees(o->src_index, n, t, &newinfo);
730f7284 926 restore_cache_bottom(&newinfo, bottom);
1ce584b0 927
d12a8cf0 928 for (i = 0; i < nr_buf; i++)
1ce584b0
JN
929 free(buf[i]);
930
730f7284 931 return ret;
01904572
LT
932}
933
934/*
935 * Compare the traverse-path to the cache entry without actually
936 * having to generate the textual representation of the traverse
937 * path.
938 *
939 * NOTE! This *only* compares up to the size of the traverse path
940 * itself - the caller needs to do the final check for the cache
941 * entry having more data at the end!
942 */
90553847
JK
943static int do_compare_entry_piecewise(const struct cache_entry *ce,
944 const struct traverse_info *info,
945 const char *name, size_t namelen,
946 unsigned mode)
01904572 947{
90553847 948 int pathlen, ce_len;
01904572
LT
949 const char *ce_name;
950
951 if (info->prev) {
d9c2bd56 952 int cmp = do_compare_entry_piecewise(ce, info->prev,
90553847
JK
953 info->name, info->namelen,
954 info->mode);
01904572
LT
955 if (cmp)
956 return cmp;
957 }
958 pathlen = info->pathlen;
959 ce_len = ce_namelen(ce);
960
961 /* If ce_len < pathlen then we must have previously hit "name == directory" entry */
962 if (ce_len < pathlen)
963 return -1;
964
965 ce_len -= pathlen;
966 ce_name = ce->name + pathlen;
967
90553847 968 return df_name_compare(ce_name, ce_len, S_IFREG, name, namelen, mode);
01904572
LT
969}
970
d9c2bd56
DT
971static int do_compare_entry(const struct cache_entry *ce,
972 const struct traverse_info *info,
90553847
JK
973 const char *name, size_t namelen,
974 unsigned mode)
d9c2bd56 975{
90553847 976 int pathlen, ce_len;
d9c2bd56
DT
977 const char *ce_name;
978 int cmp;
cd807a5c 979 unsigned ce_mode;
d9c2bd56
DT
980
981 /*
982 * If we have not precomputed the traverse path, it is quicker
983 * to avoid doing so. But if we have precomputed it,
984 * it is quicker to use the precomputed version.
985 */
986 if (!info->traverse_path)
90553847 987 return do_compare_entry_piecewise(ce, info, name, namelen, mode);
d9c2bd56
DT
988
989 cmp = strncmp(ce->name, info->traverse_path, info->pathlen);
990 if (cmp)
991 return cmp;
992
993 pathlen = info->pathlen;
994 ce_len = ce_namelen(ce);
995
996 if (ce_len < pathlen)
997 return -1;
998
999 ce_len -= pathlen;
1000 ce_name = ce->name + pathlen;
1001
cd807a5c
DS
1002 ce_mode = S_ISSPARSEDIR(ce->ce_mode) ? S_IFDIR : S_IFREG;
1003 return df_name_compare(ce_name, ce_len, ce_mode, name, namelen, mode);
d9c2bd56
DT
1004}
1005
01904572
LT
1006static int compare_entry(const struct cache_entry *ce, const struct traverse_info *info, const struct name_entry *n)
1007{
90553847 1008 int cmp = do_compare_entry(ce, info, n->path, n->pathlen, n->mode);
01904572
LT
1009 if (cmp)
1010 return cmp;
1011
cd807a5c
DS
1012 /*
1013 * At this point, we know that we have a prefix match. If ce
1014 * is a sparse directory, then allow an exact match. This only
1015 * works when the input name is a directory, since ce->name
1016 * ends in a directory separator.
1017 */
1018 if (S_ISSPARSEDIR(ce->ce_mode) &&
1019 ce->ce_namelen == traverse_path_len(info, tree_entry_len(n)) + 1)
1020 return 0;
1021
01904572
LT
1022 /*
1023 * Even if the beginning compared identically, the ce should
1024 * compare as bigger than a directory leading up to it!
1025 */
b3b3cbcb 1026 return ce_namelen(ce) > traverse_path_len(info, tree_entry_len(n));
01904572
LT
1027}
1028
da165f47
JH
1029static int ce_in_traverse_path(const struct cache_entry *ce,
1030 const struct traverse_info *info)
1031{
1032 if (!info->prev)
1033 return 1;
90553847
JK
1034 if (do_compare_entry(ce, info->prev,
1035 info->name, info->namelen, info->mode))
da165f47
JH
1036 return 0;
1037 /*
1038 * If ce (blob) is the same name as the path (which is a tree
1039 * we will be descending into), it won't be inside it.
1040 */
1041 return (info->pathlen < ce_namelen(ce));
1042}
1043
a849735b
JM
1044static struct cache_entry *create_ce_entry(const struct traverse_info *info,
1045 const struct name_entry *n,
1046 int stage,
1047 struct index_state *istate,
523506df
DS
1048 int is_transient,
1049 int is_sparse_directory)
01904572 1050{
b3b3cbcb 1051 size_t len = traverse_path_len(info, tree_entry_len(n));
523506df 1052 size_t alloc_len = is_sparse_directory ? len + 1 : len;
a849735b
JM
1053 struct cache_entry *ce =
1054 is_transient ?
523506df
DS
1055 make_empty_transient_cache_entry(alloc_len, NULL) :
1056 make_empty_cache_entry(istate, alloc_len);
01904572
LT
1057
1058 ce->ce_mode = create_ce_mode(n->mode);
b60e188c
TG
1059 ce->ce_flags = create_ce_flags(stage);
1060 ce->ce_namelen = len;
ea82b2a0 1061 oidcpy(&ce->oid, &n->oid);
5aa02f98
JK
1062 /* len+1 because the cache_entry allocates space for NUL */
1063 make_traverse_path(ce->name, len + 1, info, n->path, n->pathlen);
01904572 1064
523506df
DS
1065 if (is_sparse_directory) {
1066 ce->name[len] = '/';
1067 ce->name[len + 1] = '\0';
1068 ce->ce_namelen++;
1069 ce->ce_flags |= CE_SKIP_WORKTREE;
1070 }
1071
01904572
LT
1072 return ce;
1073}
1074
b15207b8
VD
1075/*
1076 * Determine whether the path specified by 'p' should be unpacked as a new
1077 * sparse directory in a sparse index. A new sparse directory 'A/':
1078 * - must be outside the sparse cone.
1079 * - must not already be in the index (i.e., no index entry with name 'A/'
1080 * exists).
1081 * - must not have any child entries in the index (i.e., no index entry
1082 * 'A/<something>' exists).
1083 * If 'p' meets the above requirements, return 1; otherwise, return 0.
1084 */
1085static int entry_is_new_sparse_dir(const struct traverse_info *info,
1086 const struct name_entry *p)
1087{
1088 int res, pos;
1089 struct strbuf dirpath = STRBUF_INIT;
1090 struct unpack_trees_options *o = info->data;
1091
1092 if (!S_ISDIR(p->mode))
1093 return 0;
1094
1095 /*
1096 * If the path is inside the sparse cone, it can't be a sparse directory.
1097 */
1098 strbuf_add(&dirpath, info->traverse_path, info->pathlen);
1099 strbuf_add(&dirpath, p->path, p->pathlen);
1100 strbuf_addch(&dirpath, '/');
1101 if (path_in_cone_mode_sparse_checkout(dirpath.buf, o->src_index)) {
1102 res = 0;
1103 goto cleanup;
1104 }
1105
1106 pos = index_name_pos_sparse(o->src_index, dirpath.buf, dirpath.len);
1107 if (pos >= 0) {
1108 /* Path is already in the index, not a new sparse dir */
1109 res = 0;
1110 goto cleanup;
1111 }
1112
1113 /* Where would this sparse dir be inserted into the index? */
1114 pos = -pos - 1;
1115 if (pos >= o->src_index->cache_nr) {
1116 /*
1117 * Sparse dir would be inserted at the end of the index, so we
1118 * know it has no child entries.
1119 */
1120 res = 1;
1121 goto cleanup;
1122 }
1123
1124 /*
1125 * If the dir has child entries in the index, the first would be at the
1126 * position the sparse directory would be inserted. If the entry at this
1127 * position is inside the dir, not a new sparse dir.
1128 */
1129 res = strncmp(o->src_index->cache[pos]->name, dirpath.buf, dirpath.len);
1130
1131cleanup:
1132 strbuf_release(&dirpath);
1133 return res;
1134}
1135
b4da3738
NTND
1136/*
1137 * Note that traverse_by_cache_tree() duplicates some logic in this function
1138 * without actually calling it. If you change the logic here you may need to
1139 * check and change there as well.
1140 */
bd6a3fd7
DS
1141static int unpack_single_entry(int n, unsigned long mask,
1142 unsigned long dirmask,
1143 struct cache_entry **src,
1144 const struct name_entry *names,
b15207b8
VD
1145 const struct traverse_info *info,
1146 int *is_new_sparse_dir)
01904572
LT
1147{
1148 int i;
1149 struct unpack_trees_options *o = info->data;
603d2498 1150 unsigned long conflicts = info->df_conflicts | dirmask;
b15207b8 1151 const struct name_entry *p = names;
01904572 1152
b15207b8
VD
1153 *is_new_sparse_dir = 0;
1154 if (mask == dirmask && !src[0]) {
1155 /*
1156 * If we're not in a sparse index, we can't unpack a directory
1157 * without recursing into it, so we return.
1158 */
1159 if (!o->src_index->sparse_index)
1160 return 0;
1161
1162 /* Find first entry with a real name (we could use "mask" too) */
1163 while (!p->mode)
1164 p++;
1165
1166 /*
1167 * If the directory is completely missing from the index but
1168 * would otherwise be a sparse directory, we should unpack it.
1169 * If not, we'll return and continue recursively traversing the
1170 * tree.
1171 */
1172 *is_new_sparse_dir = entry_is_new_sparse_dir(info, p);
1173 if (!*is_new_sparse_dir)
1174 return 0;
1175 }
01904572 1176
523506df 1177 /*
b15207b8
VD
1178 * When we are unpacking a sparse directory, then this isn't necessarily
1179 * a directory-file conflict.
523506df 1180 */
b15207b8
VD
1181 if (mask == dirmask &&
1182 (*is_new_sparse_dir || (src[0] && S_ISSPARSEDIR(src[0]->ce_mode))))
523506df
DS
1183 conflicts = 0;
1184
01904572
LT
1185 /*
1186 * Ok, we've filled in up to any potential index entry in src[0],
1187 * now do the rest.
1188 */
1189 for (i = 0; i < n; i++) {
1190 int stage;
1191 unsigned int bit = 1ul << i;
1192 if (conflicts & bit) {
1193 src[i + o->merge] = o->df_conflict_entry;
1194 continue;
1195 }
1196 if (!(mask & bit))
1197 continue;
1198 if (!o->merge)
1199 stage = 0;
1200 else if (i + 1 < o->head_idx)
1201 stage = 1;
1202 else if (i + 1 > o->head_idx)
1203 stage = 3;
1204 else
1205 stage = 2;
a849735b
JM
1206
1207 /*
1208 * If the merge bit is set, then the cache entries are
1209 * discarded in the following block. In this case,
1210 * construct "transient" cache_entries, as they are
1211 * not stored in the index. otherwise construct the
1212 * cache entry from the index aware logic.
1213 */
523506df
DS
1214 src[i + o->merge] = create_ce_entry(info, names + i, stage,
1215 &o->result, o->merge,
1216 bit & dirmask);
01904572
LT
1217 }
1218
5d80ef5a
RS
1219 if (o->merge) {
1220 int rc = call_unpack_fn((const struct cache_entry * const *)src,
1221 o);
1222 for (i = 0; i < n; i++) {
1223 struct cache_entry *ce = src[i + o->merge];
1224 if (ce != o->df_conflict_entry)
a849735b 1225 discard_cache_entry(ce);
5d80ef5a
RS
1226 }
1227 return rc;
1228 }
01904572 1229
01904572 1230 for (i = 0; i < n; i++)
aab3b9a1 1231 if (src[i] && src[i] != o->df_conflict_entry)
46169180
JK
1232 if (do_add_entry(o, src[i], 0, 0))
1233 return -1;
1234
01904572
LT
1235 return 0;
1236}
1237
353c5eeb
JH
1238static int unpack_failed(struct unpack_trees_options *o, const char *message)
1239{
1240 discard_index(&o->result);
b165fac8 1241 if (!o->quiet && !o->exiting_early) {
353c5eeb
JH
1242 if (message)
1243 return error("%s", message);
1244 return -1;
1245 }
1246 return -1;
1247}
1248
730f7284
JH
1249/*
1250 * The tree traversal is looking at name p. If we have a matching entry,
1251 * return it. If name p is a directory in the index, do not return
1252 * anything, as we will want to match it when the traversal descends into
1253 * the directory.
1254 */
1255static int find_cache_pos(struct traverse_info *info,
90553847 1256 const char *p, size_t p_len)
730f7284
JH
1257{
1258 int pos;
1259 struct unpack_trees_options *o = info->data;
1260 struct index_state *index = o->src_index;
1261 int pfxlen = info->pathlen;
730f7284
JH
1262
1263 for (pos = o->cache_bottom; pos < index->cache_nr; pos++) {
9c5e6c80 1264 const struct cache_entry *ce = index->cache[pos];
730f7284
JH
1265 const char *ce_name, *ce_slash;
1266 int cmp, ce_len;
1267
e53e6b44
BD
1268 if (ce->ce_flags & CE_UNPACKED) {
1269 /*
1270 * cache_bottom entry is already unpacked, so
1271 * we can never match it; don't check it
1272 * again.
1273 */
1274 if (pos == o->cache_bottom)
1275 ++o->cache_bottom;
730f7284 1276 continue;
e53e6b44 1277 }
a6720955
DT
1278 if (!ce_in_traverse_path(ce, info)) {
1279 /*
1280 * Check if we can skip future cache checks
1281 * (because we're already past all possible
1282 * entries in the traverse path).
1283 */
1284 if (info->traverse_path) {
1285 if (strncmp(ce->name, info->traverse_path,
1286 info->pathlen) > 0)
1287 break;
1288 }
730f7284 1289 continue;
a6720955 1290 }
730f7284
JH
1291 ce_name = ce->name + pfxlen;
1292 ce_slash = strchr(ce_name, '/');
1293 if (ce_slash)
1294 ce_len = ce_slash - ce_name;
1295 else
1296 ce_len = ce_namelen(ce) - pfxlen;
90553847 1297 cmp = name_compare(p, p_len, ce_name, ce_len);
730f7284
JH
1298 /*
1299 * Exact match; if we have a directory we need to
1300 * delay returning it.
1301 */
1302 if (!cmp)
1303 return ce_slash ? -2 - pos : pos;
1304 if (0 < cmp)
1305 continue; /* keep looking */
1306 /*
1307 * ce_name sorts after p->path; could it be that we
1308 * have files under p->path directory in the index?
1309 * E.g. ce_name == "t-i", and p->path == "t"; we may
1310 * have "t/a" in the index.
1311 */
90553847 1312 if (p_len < ce_len && !memcmp(ce_name, p, p_len) &&
730f7284
JH
1313 ce_name[p_len] < '/')
1314 continue; /* keep looking */
1315 break;
1316 }
1317 return -1;
1318}
1319
523506df
DS
1320/*
1321 * Given a sparse directory entry 'ce', compare ce->name to
8c5de0d2
DS
1322 * info->traverse_path + p->path + '/' if info->traverse_path
1323 * is non-empty.
1324 *
523506df
DS
1325 * Compare ce->name to p->path + '/' otherwise. Note that
1326 * ce->name must end in a trailing '/' because it is a sparse
1327 * directory entry.
1328 */
1329static int sparse_dir_matches_path(const struct cache_entry *ce,
1330 struct traverse_info *info,
1331 const struct name_entry *p)
1332{
1333 assert(S_ISSPARSEDIR(ce->ce_mode));
1334 assert(ce->name[ce->ce_namelen - 1] == '/');
1335
8c5de0d2
DS
1336 if (info->pathlen)
1337 return ce->ce_namelen == info->pathlen + p->pathlen + 1 &&
1338 ce->name[info->pathlen - 1] == '/' &&
1339 !strncmp(ce->name, info->traverse_path, info->pathlen) &&
1340 !strncmp(ce->name + info->pathlen, p->path, p->pathlen);
523506df
DS
1341 return ce->ce_namelen == p->pathlen + 1 &&
1342 !strncmp(ce->name, p->path, p->pathlen);
1343}
1344
730f7284
JH
1345static struct cache_entry *find_cache_entry(struct traverse_info *info,
1346 const struct name_entry *p)
1347{
72d84ea3 1348 const char *path;
90553847 1349 int pos = find_cache_pos(info, p->path, p->pathlen);
730f7284
JH
1350 struct unpack_trees_options *o = info->data;
1351
1352 if (0 <= pos)
1353 return o->src_index->cache[pos];
523506df
DS
1354
1355 /*
1356 * Check for a sparse-directory entry named "path/".
1357 * Due to the input p->path not having a trailing
1358 * slash, the negative 'pos' value overshoots the
1359 * expected position, hence "-2" instead of "-1".
1360 */
1361 pos = -pos - 2;
1362
1363 if (pos < 0 || pos >= o->src_index->cache_nr)
730f7284 1364 return NULL;
523506df
DS
1365
1366 /*
1367 * Due to lexicographic sorting and sparse directory
1368 * entries ending with a trailing slash, our path as a
1369 * sparse directory (e.g "subdir/") and our path as a
1370 * file (e.g. "subdir") might be separated by other
1371 * paths (e.g. "subdir-").
1372 */
1373 while (pos >= 0) {
72d84ea3 1374 struct cache_entry *ce = o->src_index->cache[pos];
523506df 1375
72d84ea3
DS
1376 if (!skip_prefix(ce->name, info->traverse_path, &path) ||
1377 strncmp(path, p->path, p->pathlen) ||
1378 path[p->pathlen] != '/')
523506df
DS
1379 return NULL;
1380
1381 if (S_ISSPARSEDIR(ce->ce_mode) &&
1382 sparse_dir_matches_path(ce, info, p))
1383 return ce;
1384
1385 pos--;
1386 }
1387
1388 return NULL;
730f7284
JH
1389}
1390
ba655da5
JH
1391static void debug_path(struct traverse_info *info)
1392{
1393 if (info->prev) {
1394 debug_path(info->prev);
90553847 1395 if (*info->prev->name)
ba655da5
JH
1396 putchar('/');
1397 }
90553847 1398 printf("%s", info->name);
ba655da5
JH
1399}
1400
1401static void debug_name_entry(int i, struct name_entry *n)
1402{
1403 printf("ent#%d %06o %s\n", i,
1404 n->path ? n->mode : 0,
1405 n->path ? n->path : "(missing)");
1406}
1407
1408static void debug_unpack_callback(int n,
1409 unsigned long mask,
1410 unsigned long dirmask,
1411 struct name_entry *names,
1412 struct traverse_info *info)
1413{
1414 int i;
1415 printf("* unpack mask %lu, dirmask %lu, cnt %d ",
1416 mask, dirmask, n);
1417 debug_path(info);
1418 putchar('\n');
1419 for (i = 0; i < n; i++)
1420 debug_name_entry(i, names + i);
1421}
1422
523506df
DS
1423/*
1424 * Returns true if and only if the given cache_entry is a
1425 * sparse-directory entry that matches the given name_entry
1426 * from the tree walk at the given traverse_info.
1427 */
1428static int is_sparse_directory_entry(struct cache_entry *ce,
037f8ea6 1429 const struct name_entry *name,
523506df
DS
1430 struct traverse_info *info)
1431{
1432 if (!ce || !name || !S_ISSPARSEDIR(ce->ce_mode))
1433 return 0;
1434
1435 return sparse_dir_matches_path(ce, info, name);
1436}
1437
ab81047a
VD
1438static int unpack_sparse_callback(int n, unsigned long mask, unsigned long dirmask, struct name_entry *names, struct traverse_info *info)
1439{
1440 struct cache_entry *src[MAX_UNPACK_TREES + 1] = { NULL, };
1441 struct unpack_trees_options *o = info->data;
b15207b8 1442 int ret, is_new_sparse_dir;
ab81047a
VD
1443
1444 assert(o->merge);
1445
1446 /*
1447 * Unlike in 'unpack_callback', where src[0] is derived from the index when
1448 * merging, src[0] is a transient cache entry derived from the first tree
1449 * provided. Create the temporary entry as if it came from a non-sparse index.
1450 */
1451 if (!is_null_oid(&names[0].oid)) {
1452 src[0] = create_ce_entry(info, &names[0], 0,
1453 &o->result, 1,
1454 dirmask & (1ul << 0));
1455 src[0]->ce_flags |= (CE_SKIP_WORKTREE | CE_NEW_SKIP_WORKTREE);
1456 }
1457
1458 /*
1459 * 'unpack_single_entry' assumes that src[0] is derived directly from
1460 * the index, rather than from an entry in 'names'. This is *not* true when
1461 * merging a sparse directory, in which case names[0] is the "index" source
1462 * entry. To match the expectations of 'unpack_single_entry', shift past the
1463 * "index" tree (i.e., names[0]) and adjust 'names', 'n', 'mask', and
1464 * 'dirmask' accordingly.
1465 */
b15207b8 1466 ret = unpack_single_entry(n - 1, mask >> 1, dirmask >> 1, src, names + 1, info, &is_new_sparse_dir);
ab81047a
VD
1467
1468 if (src[0])
1469 discard_cache_entry(src[0]);
1470
1471 return ret >= 0 ? mask : -1;
1472}
1473
b4da3738
NTND
1474/*
1475 * Note that traverse_by_cache_tree() duplicates some logic in this function
1476 * without actually calling it. If you change the logic here you may need to
1477 * check and change there as well.
1478 */
01904572
LT
1479static int unpack_callback(int n, unsigned long mask, unsigned long dirmask, struct name_entry *names, struct traverse_info *info)
1480{
c7cddc1a 1481 struct cache_entry *src[MAX_UNPACK_TREES + 1] = { NULL, };
01904572 1482 struct unpack_trees_options *o = info->data;
01904572 1483 const struct name_entry *p = names;
b15207b8 1484 int is_new_sparse_dir;
01904572
LT
1485
1486 /* Find first entry with a real name (we could use "mask" too) */
1487 while (!p->mode)
1488 p++;
1489
ba655da5
JH
1490 if (o->debug_unpack)
1491 debug_unpack_callback(n, mask, dirmask, names, info);
1492
01904572
LT
1493 /* Are we supposed to look at the index too? */
1494 if (o->merge) {
da165f47 1495 while (1) {
da165f47 1496 int cmp;
730f7284
JH
1497 struct cache_entry *ce;
1498
1499 if (o->diff_index_cached)
99430aa1 1500 ce = next_cache_entry(o);
730f7284
JH
1501 else
1502 ce = find_cache_entry(info, p);
1503
da165f47
JH
1504 if (!ce)
1505 break;
1506 cmp = compare_entry(ce, info, p);
01904572
LT
1507 if (cmp < 0) {
1508 if (unpack_index_entry(ce, o) < 0)
353c5eeb 1509 return unpack_failed(o, NULL);
01904572
LT
1510 continue;
1511 }
1512 if (!cmp) {
1513 if (ce_stage(ce)) {
1514 /*
da165f47
JH
1515 * If we skip unmerged index
1516 * entries, we'll skip this
1517 * entry *and* the tree
1518 * entries associated with it!
01904572 1519 */
34110cd4 1520 if (o->skip_unmerged) {
da165f47 1521 add_same_unmerged(ce, o);
01904572 1522 return mask;
34110cd4 1523 }
01904572
LT
1524 }
1525 src[0] = ce;
01904572
LT
1526 }
1527 break;
1528 }
1529 }
1530
b15207b8 1531 if (unpack_single_entry(n, mask, dirmask, src, names, info, &is_new_sparse_dir))
01904572
LT
1532 return -1;
1533
97e5954b 1534 if (o->merge && src[0]) {
da165f47
JH
1535 if (ce_stage(src[0]))
1536 mark_ce_used_same_name(src[0], o);
1537 else
1538 mark_ce_used(src[0], o);
1539 }
1540
01904572
LT
1541 /* Now handle any directories.. */
1542 if (dirmask) {
b65982b6
JH
1543 /* special case: "diff-index --cached" looking at a tree */
1544 if (o->diff_index_cached &&
1545 n == 1 && dirmask == 1 && S_ISDIR(names->mode)) {
1546 int matches;
1547 matches = cache_tree_matches_traversal(o->src_index->cache_tree,
1548 names, info);
1549 /*
da165f47
JH
1550 * Everything under the name matches; skip the
1551 * entire hierarchy. diff_index_cached codepath
1552 * special cases D/F conflicts in such a way that
1553 * it does not do any look-ahead, so this is safe.
b65982b6
JH
1554 */
1555 if (matches) {
bfc763df
VD
1556 /*
1557 * Only increment the cache_bottom if the
1558 * directory isn't a sparse directory index
1559 * entry (if it is, it was already incremented)
1560 * in 'mark_ce_used()'
1561 */
1562 if (!src[0] || !S_ISSPARSEDIR(src[0]->ce_mode))
1563 o->cache_bottom += matches;
b65982b6
JH
1564 return mask;
1565 }
1566 }
1567
037f8ea6 1568 if (!is_sparse_directory_entry(src[0], p, info) &&
b15207b8 1569 !is_new_sparse_dir &&
523506df
DS
1570 traverse_trees_recursive(n, dirmask, mask & ~dirmask,
1571 names, info) < 0) {
542c264b 1572 return -1;
523506df
DS
1573 }
1574
01904572
LT
1575 return mask;
1576 }
1577
1578 return mask;
1579}
1580
27c82fb3
NTND
1581static int clear_ce_flags_1(struct index_state *istate,
1582 struct cache_entry **cache, int nr,
fc2b6214 1583 struct strbuf *prefix,
28911091 1584 int select_mask, int clear_mask,
468ce99b 1585 struct pattern_list *pl,
4dcd4def
DS
1586 enum pattern_match_result default_match,
1587 int progress_nr);
28911091 1588
9037026d 1589/* Whole directory matching */
27c82fb3
NTND
1590static int clear_ce_flags_dir(struct index_state *istate,
1591 struct cache_entry **cache, int nr,
fc2b6214 1592 struct strbuf *prefix,
9037026d
NTND
1593 char *basename,
1594 int select_mask, int clear_mask,
468ce99b 1595 struct pattern_list *pl,
4dcd4def
DS
1596 enum pattern_match_result default_match,
1597 int progress_nr)
9037026d 1598{
28911091 1599 struct cache_entry **cache_end;
9037026d 1600 int dtype = DT_DIR;
fc2b6214 1601 int rc;
eb42feca
DS
1602 enum pattern_match_result ret, orig_ret;
1603 orig_ret = path_matches_pattern_list(prefix->buf, prefix->len,
1604 basename, &dtype, pl, istate);
9037026d 1605
fc2b6214 1606 strbuf_addch(prefix, '/');
9037026d 1607
28911091 1608 /* If undecided, use matching result of parent dir in defval */
eb42feca 1609 if (orig_ret == UNDECIDED)
468ce99b 1610 ret = default_match;
eb42feca
DS
1611 else
1612 ret = orig_ret;
9037026d 1613
28911091
NTND
1614 for (cache_end = cache; cache_end != cache + nr; cache_end++) {
1615 struct cache_entry *ce = *cache_end;
fc2b6214 1616 if (strncmp(ce->name, prefix->buf, prefix->len))
28911091 1617 break;
9037026d
NTND
1618 }
1619
eb42feca
DS
1620 if (pl->use_cone_patterns && orig_ret == MATCHED_RECURSIVE) {
1621 struct cache_entry **ce = cache;
4c6c7971 1622 rc = cache_end - cache;
eb42feca
DS
1623
1624 while (ce < cache_end) {
1625 (*ce)->ce_flags &= ~clear_mask;
1626 ce++;
1627 }
1628 } else if (pl->use_cone_patterns && orig_ret == NOT_MATCHED) {
4c6c7971 1629 rc = cache_end - cache;
eb42feca
DS
1630 } else {
1631 rc = clear_ce_flags_1(istate, cache, cache_end - cache,
1632 prefix,
1633 select_mask, clear_mask,
4dcd4def
DS
1634 pl, ret,
1635 progress_nr);
eb42feca
DS
1636 }
1637
fc2b6214
AP
1638 strbuf_setlen(prefix, prefix->len - 1);
1639 return rc;
9037026d
NTND
1640}
1641
1642/*
1643 * Traverse the index, find every entry that matches according to
caa3d554 1644 * o->pl. Do "ce_flags &= ~clear_mask" on those entries. Return the
9037026d
NTND
1645 * number of traversed entries.
1646 *
1647 * If select_mask is non-zero, only entries whose ce_flags has on of
1648 * those bits enabled are traversed.
1649 *
1650 * cache : pointer to an index entry
1651 * prefix_len : an offset to its path
1652 *
1653 * The current path ("prefix") including the trailing '/' is
1654 * cache[0]->name[0..(prefix_len-1)]
1655 * Top level path has prefix_len zero.
1656 */
27c82fb3
NTND
1657static int clear_ce_flags_1(struct index_state *istate,
1658 struct cache_entry **cache, int nr,
fc2b6214 1659 struct strbuf *prefix,
9037026d 1660 int select_mask, int clear_mask,
468ce99b 1661 struct pattern_list *pl,
4dcd4def
DS
1662 enum pattern_match_result default_match,
1663 int progress_nr)
9037026d 1664{
d20bc01a 1665 struct cache_entry **cache_end = nr ? cache + nr : cache;
9037026d
NTND
1666
1667 /*
1668 * Process all entries that have the given prefix and meet
1669 * select_mask condition
1670 */
1671 while(cache != cache_end) {
1672 struct cache_entry *ce = *cache;
1673 const char *name, *slash;
468ce99b
DS
1674 int len, dtype;
1675 enum pattern_match_result ret;
9037026d 1676
4dcd4def
DS
1677 display_progress(istate->progress, progress_nr);
1678
9037026d
NTND
1679 if (select_mask && !(ce->ce_flags & select_mask)) {
1680 cache++;
4dcd4def 1681 progress_nr++;
9037026d
NTND
1682 continue;
1683 }
1684
fc2b6214 1685 if (prefix->len && strncmp(ce->name, prefix->buf, prefix->len))
9037026d
NTND
1686 break;
1687
fc2b6214 1688 name = ce->name + prefix->len;
9037026d
NTND
1689 slash = strchr(name, '/');
1690
1691 /* If it's a directory, try whole directory match first */
1692 if (slash) {
1693 int processed;
1694
1695 len = slash - name;
fc2b6214 1696 strbuf_add(prefix, name, len);
9037026d 1697
27c82fb3 1698 processed = clear_ce_flags_dir(istate, cache, cache_end - cache,
fc2b6214
AP
1699 prefix,
1700 prefix->buf + prefix->len - len,
9037026d 1701 select_mask, clear_mask,
4dcd4def
DS
1702 pl, default_match,
1703 progress_nr);
9037026d
NTND
1704
1705 /* clear_c_f_dir eats a whole dir already? */
1706 if (processed) {
1707 cache += processed;
4dcd4def 1708 progress_nr += processed;
fc2b6214 1709 strbuf_setlen(prefix, prefix->len - len);
9037026d
NTND
1710 continue;
1711 }
1712
fc2b6214 1713 strbuf_addch(prefix, '/');
4dcd4def
DS
1714 processed = clear_ce_flags_1(istate, cache, cache_end - cache,
1715 prefix,
1716 select_mask, clear_mask, pl,
1717 default_match, progress_nr);
1718
1719 cache += processed;
1720 progress_nr += processed;
1721
fc2b6214 1722 strbuf_setlen(prefix, prefix->len - len - 1);
9037026d
NTND
1723 continue;
1724 }
1725
1726 /* Non-directory */
1727 dtype = ce_to_dtype(ce);
468ce99b
DS
1728 ret = path_matches_pattern_list(ce->name,
1729 ce_namelen(ce),
1730 name, &dtype, pl, istate);
1731 if (ret == UNDECIDED)
1732 ret = default_match;
f998a3f1 1733 if (ret == MATCHED || ret == MATCHED_RECURSIVE)
9037026d
NTND
1734 ce->ce_flags &= ~clear_mask;
1735 cache++;
4dcd4def 1736 progress_nr++;
9037026d 1737 }
4dcd4def
DS
1738
1739 display_progress(istate->progress, progress_nr);
9037026d
NTND
1740 return nr - (cache_end - cache);
1741}
1742
27c82fb3
NTND
1743static int clear_ce_flags(struct index_state *istate,
1744 int select_mask, int clear_mask,
4dcd4def
DS
1745 struct pattern_list *pl,
1746 int show_progress)
9037026d 1747{
fc2b6214 1748 static struct strbuf prefix = STRBUF_INIT;
e6152e35
JH
1749 char label[100];
1750 int rval;
fc2b6214
AP
1751
1752 strbuf_reset(&prefix);
4dcd4def
DS
1753 if (show_progress)
1754 istate->progress = start_delayed_progress(
1755 _("Updating index flags"),
1756 istate->cache_nr);
fc2b6214 1757
e6152e35
JH
1758 xsnprintf(label, sizeof(label), "clear_ce_flags(0x%08lx,0x%08lx)",
1759 (unsigned long)select_mask, (unsigned long)clear_mask);
1760 trace2_region_enter("unpack_trees", label, the_repository);
1761 rval = clear_ce_flags_1(istate,
27c82fb3
NTND
1762 istate->cache,
1763 istate->cache_nr,
fc2b6214 1764 &prefix,
9037026d 1765 select_mask, clear_mask,
4dcd4def 1766 pl, 0, 0);
e6152e35
JH
1767 trace2_region_leave("unpack_trees", label, the_repository);
1768
4dcd4def 1769 stop_progress(&istate->progress);
e6152e35 1770 return rval;
9037026d
NTND
1771}
1772
2431afbf
NTND
1773/*
1774 * Set/Clear CE_NEW_SKIP_WORKTREE according to $GIT_DIR/info/sparse-checkout
1775 */
caa3d554 1776static void mark_new_skip_worktree(struct pattern_list *pl,
86016ec3 1777 struct index_state *istate,
4dcd4def
DS
1778 int select_flag, int skip_wt_flag,
1779 int show_progress)
2431afbf
NTND
1780{
1781 int i;
1782
9037026d
NTND
1783 /*
1784 * 1. Pretend the narrowest worktree: only unmerged entries
1785 * are checked out
1786 */
86016ec3
NTND
1787 for (i = 0; i < istate->cache_nr; i++) {
1788 struct cache_entry *ce = istate->cache[i];
2431afbf
NTND
1789
1790 if (select_flag && !(ce->ce_flags & select_flag))
1791 continue;
1792
b33fdfc3 1793 if (!ce_stage(ce) && !(ce->ce_flags & CE_CONFLICTED))
2431afbf
NTND
1794 ce->ce_flags |= skip_wt_flag;
1795 else
1796 ce->ce_flags &= ~skip_wt_flag;
1797 }
9037026d
NTND
1798
1799 /*
1800 * 2. Widen worktree according to sparse-checkout file.
1801 * Matched entries will have skip_wt_flag cleared (i.e. "in")
1802 */
4dcd4def 1803 clear_ce_flags(istate, select_flag, skip_wt_flag, pl, show_progress);
2431afbf
NTND
1804}
1805
30e89c12
EN
1806static void populate_from_existing_patterns(struct unpack_trees_options *o,
1807 struct pattern_list *pl)
1808{
dd23022a 1809 if (get_sparse_checkout_patterns(pl) < 0)
30e89c12
EN
1810 o->skip_sparse_checkout = 1;
1811 else
1812 o->pl = pl;
30e89c12
EN
1813}
1814
74970392
VD
1815static void update_sparsity_for_prefix(const char *prefix,
1816 struct index_state *istate)
1817{
1818 int prefix_len = strlen(prefix);
1819 struct strbuf ce_prefix = STRBUF_INIT;
1820
1821 if (!istate->sparse_index)
1822 return;
1823
1824 while (prefix_len > 0 && prefix[prefix_len - 1] == '/')
1825 prefix_len--;
1826
1827 if (prefix_len <= 0)
1828 BUG("Invalid prefix passed to update_sparsity_for_prefix");
1829
1830 strbuf_grow(&ce_prefix, prefix_len + 1);
1831 strbuf_add(&ce_prefix, prefix, prefix_len);
1832 strbuf_addch(&ce_prefix, '/');
1833
1834 /*
1835 * If the prefix points to a sparse directory or a path inside a sparse
1836 * directory, the index should be expanded. This is accomplished in one
1837 * of two ways:
1838 * - if the prefix is inside a sparse directory, it will be expanded by
1839 * the 'ensure_full_index(...)' call in 'index_name_pos(...)'.
1840 * - if the prefix matches an existing sparse directory entry,
1841 * 'index_name_pos(...)' will return its index position, triggering
1842 * the 'ensure_full_index(...)' below.
1843 */
1844 if (!path_in_cone_mode_sparse_checkout(ce_prefix.buf, istate) &&
1845 index_name_pos(istate, ce_prefix.buf, ce_prefix.len) >= 0)
1846 ensure_full_index(istate);
1847
1848 strbuf_release(&ce_prefix);
1849}
30e89c12 1850
eb9ae4b5
RS
1851static int verify_absent(const struct cache_entry *,
1852 enum unpack_trees_error_types,
1853 struct unpack_trees_options *);
2e2b887d
JH
1854/*
1855 * N-way merge "len" trees. Returns 0 on success, -1 on failure to manipulate the
1856 * resulting index, -2 on failure to reflect the changes to the work tree.
2431afbf
NTND
1857 *
1858 * CE_ADDED, CE_UNPACKED and CE_NEW_SKIP_WORKTREE are used internally
2e2b887d 1859 */
01904572
LT
1860int unpack_trees(unsigned len, struct tree_desc *t, struct unpack_trees_options *o)
1861{
6863df35 1862 struct repository *repo = the_repository;
99430aa1 1863 int i, ret;
0fb1eaa8 1864 static struct cache_entry *dfc;
caa3d554 1865 struct pattern_list pl;
fa0bde45 1866 int free_pattern_list = 0;
c42e0b64 1867 struct dir_struct dir = DIR_INIT;
16da134b 1868
480d3d6b
EN
1869 if (o->reset == UNPACK_RESET_INVALID)
1870 BUG("o->reset had a value of 1; should be UNPACK_TREES_*_UNTRACKED");
16da134b 1871
ca885a4f
JH
1872 if (len > MAX_UNPACK_TREES)
1873 die("unpack_trees takes at most %d trees", MAX_UNPACK_TREES);
c42e0b64
EN
1874 if (o->dir)
1875 BUG("o->dir is for internal use only");
5d4f4a59
EN
1876 if (o->pl)
1877 BUG("o->pl is for internal use only");
16da134b 1878
0d1ed596 1879 trace_performance_enter();
c338898a
DS
1880 trace2_region_enter("unpack_trees", "unpack_trees", the_repository);
1881
6863df35
DS
1882 prepare_repo_settings(repo);
1883 if (repo->settings.command_requires_full_index) {
1884 ensure_full_index(o->src_index);
29fefafc
ÆAB
1885 if (o->dst_index)
1886 ensure_full_index(o->dst_index);
6863df35
DS
1887 }
1888
480d3d6b
EN
1889 if (o->reset == UNPACK_RESET_OVERWRITE_UNTRACKED &&
1890 o->preserve_ignored)
1891 BUG("UNPACK_RESET_OVERWRITE_UNTRACKED incompatible with preserved ignored files");
1892
04988c8d 1893 if (!o->preserve_ignored) {
c42e0b64 1894 o->dir = &dir;
04988c8d
EN
1895 o->dir->flags |= DIR_SHOW_IGNORED;
1896 setup_standard_excludes(o->dir);
1897 }
1898
74970392
VD
1899 if (o->prefix)
1900 update_sparsity_for_prefix(o->prefix, o->src_index);
1901
08aefc9e
NTND
1902 if (!core_apply_sparse_checkout || !o->update)
1903 o->skip_sparse_checkout = 1;
5d4f4a59 1904 if (!o->skip_sparse_checkout) {
30e89c12 1905 memset(&pl, 0, sizeof(pl));
fa0bde45 1906 free_pattern_list = 1;
30e89c12 1907 populate_from_existing_patterns(o, &pl);
08aefc9e
NTND
1908 }
1909
6269f8ea 1910 index_state_init(&o->result, o->src_index->repo);
913e0e99 1911 o->result.initialized = 1;
da165f47
JH
1912 o->result.timestamp.sec = o->src_index->timestamp.sec;
1913 o->result.timestamp.nsec = o->src_index->timestamp.nsec;
9170c7ab 1914 o->result.version = o->src_index->version;
7db11830
EN
1915 if (!o->src_index->split_index) {
1916 o->result.split_index = NULL;
1917 } else if (o->src_index == o->dst_index) {
1918 /*
1919 * o->dst_index (and thus o->src_index) will be discarded
1920 * and overwritten with o->result at the end of this function,
1921 * so just use src_index's split_index to avoid having to
1922 * create a new one.
1923 */
1924 o->result.split_index = o->src_index->split_index;
5fc2fc8f 1925 o->result.split_index->refcount++;
7db11830
EN
1926 } else {
1927 o->result.split_index = init_split_index(&o->result);
1928 }
75691ea3 1929 oidcpy(&o->result.oid, &o->src_index->oid);
16da134b 1930 o->merge_size = len;
da165f47 1931 mark_all_ce_unused(o->src_index);
0fb1eaa8 1932
3dfd3059
JS
1933 o->result.fsmonitor_last_update =
1934 xstrdup_or_null(o->src_index->fsmonitor_last_update);
9968ed73 1935 o->result.fsmonitor_has_run_once = o->src_index->fsmonitor_has_run_once;
679f2f9f 1936
0f329b9a
VD
1937 if (!o->src_index->initialized &&
1938 !repo->settings.command_requires_full_index &&
1939 is_sparse_index_allowed(&o->result, 0))
1940 o->result.sparse_index = 1;
1941
2431afbf
NTND
1942 /*
1943 * Sparse checkout loop #1: set NEW_SKIP_WORKTREE on existing entries
1944 */
1945 if (!o->skip_sparse_checkout)
4dcd4def
DS
1946 mark_new_skip_worktree(o->pl, o->src_index, 0,
1947 CE_NEW_SKIP_WORKTREE, o->verbose_update);
2431afbf 1948
0fb1eaa8 1949 if (!dfc)
13494ed1 1950 dfc = xcalloc(1, cache_entry_size(0));
0fb1eaa8 1951 o->df_conflict_entry = dfc;
16da134b
JS
1952
1953 if (len) {
01904572
LT
1954 const char *prefix = o->prefix ? o->prefix : "";
1955 struct traverse_info info;
1956
1957 setup_traverse_info(&info, prefix);
1958 info.fn = unpack_callback;
1959 info.data = o;
e6c111b4 1960 info.show_all_errors = o->show_all_errors;
40e37256 1961 info.pathspec = o->pathspec;
01904572 1962
da165f47
JH
1963 if (o->prefix) {
1964 /*
1965 * Unpack existing index entries that sort before the
1966 * prefix the tree is spliced into. Note that o->merge
1967 * is always true in this case.
1968 */
1969 while (1) {
99430aa1 1970 struct cache_entry *ce = next_cache_entry(o);
da165f47
JH
1971 if (!ce)
1972 break;
1973 if (ce_in_traverse_path(ce, &info))
1974 break;
1975 if (unpack_index_entry(ce, o) < 0)
1976 goto return_failed;
1977 }
08aefc9e 1978 }
da165f47 1979
0d1ed596 1980 trace_performance_enter();
c338898a 1981 trace2_region_enter("unpack_trees", "traverse_trees", the_repository);
67022e02 1982 ret = traverse_trees(o->src_index, len, t, &info);
c338898a 1983 trace2_region_leave("unpack_trees", "traverse_trees", the_repository);
0d1ed596
NTND
1984 trace_performance_leave("traverse_trees");
1985 if (ret < 0)
da165f47 1986 goto return_failed;
16da134b
JS
1987 }
1988
01904572
LT
1989 /* Any left-over entries in the index? */
1990 if (o->merge) {
da165f47 1991 while (1) {
99430aa1 1992 struct cache_entry *ce = next_cache_entry(o);
da165f47
JH
1993 if (!ce)
1994 break;
01904572 1995 if (unpack_index_entry(ce, o) < 0)
da165f47 1996 goto return_failed;
17e46426 1997 }
17e46426 1998 }
da165f47 1999 mark_all_ce_unused(o->src_index);
16da134b 2000
08aefc9e
NTND
2001 if (o->trivial_merges_only && o->nontrivial_merge) {
2002 ret = unpack_failed(o, "Merge requires file-level merging");
2003 goto done;
2004 }
01904572 2005
e800ec9d 2006 if (!o->skip_sparse_checkout) {
2431afbf
NTND
2007 /*
2008 * Sparse checkout loop #2: set NEW_SKIP_WORKTREE on entries not in loop #1
031ba55b 2009 * If they will have NEW_SKIP_WORKTREE, also set CE_SKIP_WORKTREE
2431afbf
NTND
2010 * so apply_sparse_checkout() won't attempt to remove it from worktree
2011 */
4dcd4def
DS
2012 mark_new_skip_worktree(o->pl, &o->result,
2013 CE_ADDED, CE_SKIP_WORKTREE | CE_NEW_SKIP_WORKTREE,
2014 o->verbose_update);
2431afbf 2015
17d26a4d 2016 ret = 0;
2431afbf 2017 for (i = 0; i < o->result.cache_nr; i++) {
e800ec9d
NTND
2018 struct cache_entry *ce = o->result.cache[i];
2019
2431afbf
NTND
2020 /*
2021 * Entries marked with CE_ADDED in merged_entry() do not have
2022 * verify_absent() check (the check is effectively disabled
2023 * because CE_NEW_SKIP_WORKTREE is set unconditionally).
2024 *
2025 * Do the real check now because we have had
2026 * correct CE_NEW_SKIP_WORKTREE
2027 */
2028 if (ce->ce_flags & CE_ADDED &&
681c637b
EN
2029 verify_absent(ce, WARNING_SPARSE_ORPHANED_NOT_OVERWRITTEN, o))
2030 ret = 1;
2031
2032 if (apply_sparse_checkout(&o->result, ce, o))
2033 ret = 1;
9e1afb16 2034 }
681c637b
EN
2035 if (ret == 1) {
2036 /*
2037 * Inability to sparsify or de-sparsify individual
2038 * paths is not an error, but just a warning.
2039 */
2040 if (o->show_all_errors)
2041 display_warning_msgs(o);
2042 ret = 0;
2043 }
e800ec9d 2044 }
01904572 2045
b0a5a12a 2046 ret = check_updates(o, &o->result) ? (-2) : 0;
e28f7641 2047 if (o->dst_index) {
836ef2b6 2048 move_index_extensions(&o->result, o->src_index);
52fca218 2049 if (!ret) {
4592e608 2050 if (git_env_bool("GIT_TEST_CHECK_CACHE_TREE", 0))
c207e9e1 2051 cache_tree_verify(the_repository, &o->result);
68fcd48b
VD
2052 if (!o->skip_cache_tree_update &&
2053 !cache_tree_fully_valid(o->result.cache_tree))
52fca218
BD
2054 cache_tree_update(&o->result,
2055 WRITE_TREE_SILENT |
2056 WRITE_TREE_REPAIR);
2057 }
1956ecd0
BP
2058
2059 o->result.updated_workdir = 1;
e28f7641 2060 discard_index(o->dst_index);
34110cd4 2061 *o->dst_index = o->result;
a16cc8b2
JH
2062 } else {
2063 discard_index(&o->result);
e28f7641 2064 }
7db11830 2065 o->src_index = NULL;
08aefc9e
NTND
2066
2067done:
fa0bde45 2068 if (free_pattern_list)
e091228e 2069 clear_pattern_list(&pl);
04988c8d
EN
2070 if (o->dir) {
2071 dir_clear(o->dir);
c42e0b64 2072 o->dir = NULL;
04988c8d 2073 }
c338898a 2074 trace2_region_leave("unpack_trees", "unpack_trees", the_repository);
fa0bde45 2075 trace_performance_leave("unpack_trees");
2e2b887d 2076 return ret;
da165f47
JH
2077
2078return_failed:
e6c111b4
MM
2079 if (o->show_all_errors)
2080 display_error_msgs(o);
da165f47 2081 mark_all_ce_unused(o->src_index);
026680f8 2082 ret = unpack_failed(o, NULL);
b4194828
JH
2083 if (o->exiting_early)
2084 ret = 0;
026680f8 2085 goto done;
16da134b 2086}
076b0adc 2087
7af7a258
EN
2088/*
2089 * Update SKIP_WORKTREE bits according to sparsity patterns, and update
2090 * working directory to match.
2091 *
2092 * CE_NEW_SKIP_WORKTREE is used internally.
2093 */
1147c56f
EN
2094enum update_sparsity_result update_sparsity(struct unpack_trees_options *o,
2095 struct pattern_list *pl)
7af7a258
EN
2096{
2097 enum update_sparsity_result ret = UPDATE_SPARSITY_SUCCESS;
ace224ac 2098 int i;
7af7a258
EN
2099 unsigned old_show_all_errors;
2100 int free_pattern_list = 0;
2101
2102 old_show_all_errors = o->show_all_errors;
2103 o->show_all_errors = 1;
33b1b4c7 2104 index_state_init(&o->result, o->src_index->repo);
7af7a258
EN
2105
2106 /* Sanity checks */
2107 if (!o->update || o->index_only || o->skip_sparse_checkout)
2108 BUG("update_sparsity() is for reflecting sparsity patterns in working directory");
2109 if (o->src_index != o->dst_index || o->fn)
2110 BUG("update_sparsity() called wrong");
2111
2112 trace_performance_enter();
2113
2114 /* If we weren't given patterns, use the recorded ones */
1147c56f 2115 if (!pl) {
7af7a258 2116 free_pattern_list = 1;
1147c56f
EN
2117 pl = xcalloc(1, sizeof(*pl));
2118 populate_from_existing_patterns(o, pl);
7af7a258 2119 }
1147c56f 2120 o->pl = pl;
7af7a258 2121
598b1e7d
DS
2122 /* Expand sparse directories as needed */
2123 expand_index(o->src_index, o->pl);
2124
7af7a258
EN
2125 /* Set NEW_SKIP_WORKTREE on existing entries. */
2126 mark_all_ce_unused(o->src_index);
2127 mark_new_skip_worktree(o->pl, o->src_index, 0,
2128 CE_NEW_SKIP_WORKTREE, o->verbose_update);
2129
2130 /* Then loop over entries and update/remove as needed */
2131 ret = UPDATE_SPARSITY_SUCCESS;
7af7a258
EN
2132 for (i = 0; i < o->src_index->cache_nr; i++) {
2133 struct cache_entry *ce = o->src_index->cache[i];
2134
ebb568b9
EN
2135
2136 if (ce_stage(ce)) {
2137 /* -1 because for loop will increment by 1 */
2138 i += warn_conflicted_path(o->src_index, i, o) - 1;
2139 ret = UPDATE_SPARSITY_WARNINGS;
2140 continue;
2141 }
2142
7af7a258
EN
2143 if (apply_sparse_checkout(o->src_index, ce, o))
2144 ret = UPDATE_SPARSITY_WARNINGS;
7af7a258
EN
2145 }
2146
7af7a258
EN
2147 if (check_updates(o, o->src_index))
2148 ret = UPDATE_SPARSITY_WORKTREE_UPDATE_FAILURES;
2149
6271d77c 2150 display_warning_msgs(o);
7af7a258 2151 o->show_all_errors = old_show_all_errors;
1147c56f
EN
2152 if (free_pattern_list) {
2153 clear_pattern_list(pl);
2154 free(pl);
2155 o->pl = NULL;
2156 }
7af7a258
EN
2157 trace_performance_leave("update_sparsity");
2158 return ret;
2159}
2160
076b0adc
JS
2161/* Here come the merge functions */
2162
eb9ae4b5
RS
2163static int reject_merge(const struct cache_entry *ce,
2164 struct unpack_trees_options *o)
076b0adc 2165{
191e9d2c 2166 return add_rejected_path(o, ERROR_WOULD_OVERWRITE, ce->name);
076b0adc
JS
2167}
2168
eb9ae4b5 2169static int same(const struct cache_entry *a, const struct cache_entry *b)
076b0adc
JS
2170{
2171 if (!!a != !!b)
2172 return 0;
2173 if (!a && !b)
2174 return 1;
e11d7b59
JH
2175 if ((a->ce_flags | b->ce_flags) & CE_CONFLICTED)
2176 return 0;
076b0adc 2177 return a->ce_mode == b->ce_mode &&
4a7e27e9 2178 oideq(&a->oid, &b->oid);
076b0adc
JS
2179}
2180
2181
2182/*
2183 * When a CE gets turned into an unmerged entry, we
2184 * want it to be up-to-date
2185 */
eb9ae4b5
RS
2186static int verify_uptodate_1(const struct cache_entry *ce,
2187 struct unpack_trees_options *o,
2188 enum unpack_trees_error_types error_type)
076b0adc
JS
2189{
2190 struct stat st;
2191
d5b66299
NTND
2192 if (o->index_only)
2193 return 0;
2194
2195 /*
2196 * CE_VALID and CE_SKIP_WORKTREE cheat, we better check again
2197 * if this entry is truly up-to-date because this file may be
2198 * overwritten.
2199 */
2200 if ((ce->ce_flags & CE_VALID) || ce_skip_worktree(ce))
2201 ; /* keep checking */
2202 else if (o->reset || ce_uptodate(ce))
203a2fe1 2203 return 0;
076b0adc
JS
2204
2205 if (!lstat(ce->name, &st)) {
d5b66299
NTND
2206 int flags = CE_MATCH_IGNORE_VALID|CE_MATCH_IGNORE_SKIP_WORKTREE;
2207 unsigned changed = ie_match_stat(o->src_index, ce, &st, flags);
a7bc845a
SB
2208
2209 if (submodule_from_ce(ce)) {
2210 int r = check_submodule_move_head(ce,
2211 "HEAD", oid_to_hex(&ce->oid), o);
2212 if (r)
191e9d2c 2213 return add_rejected_path(o, error_type, ce->name);
a7bc845a
SB
2214 return 0;
2215 }
2216
076b0adc 2217 if (!changed)
203a2fe1 2218 return 0;
936492d3 2219 /*
a7bc845a
SB
2220 * Historic default policy was to allow submodule to be out
2221 * of sync wrt the superproject index. If the submodule was
2222 * not considered interesting above, we don't care here.
936492d3 2223 */
7a51ed66 2224 if (S_ISGITLINK(ce->ce_mode))
203a2fe1 2225 return 0;
a7bc845a 2226
076b0adc
JS
2227 errno = 0;
2228 }
076b0adc 2229 if (errno == ENOENT)
203a2fe1 2230 return 0;
191e9d2c 2231 return add_rejected_path(o, error_type, ce->name);
35a5aa79
NTND
2232}
2233
64b1abe9
EN
2234int verify_uptodate(const struct cache_entry *ce,
2235 struct unpack_trees_options *o)
35a5aa79 2236{
26b5d6b0
EN
2237 if (!o->skip_sparse_checkout &&
2238 (ce->ce_flags & CE_SKIP_WORKTREE) &&
2239 (ce->ce_flags & CE_NEW_SKIP_WORKTREE))
f1f523ea 2240 return 0;
08402b04 2241 return verify_uptodate_1(ce, o, ERROR_NOT_UPTODATE_FILE);
076b0adc
JS
2242}
2243
eb9ae4b5 2244static int verify_uptodate_sparse(const struct cache_entry *ce,
e800ec9d
NTND
2245 struct unpack_trees_options *o)
2246{
1ac83f42 2247 return verify_uptodate_1(ce, o, WARNING_SPARSE_NOT_UPTODATE_FILE);
076b0adc
JS
2248}
2249
383480ba
NTND
2250/*
2251 * TODO: We should actually invalidate o->result, not src_index [1].
2252 * But since cache tree and untracked cache both are not copied to
2253 * o->result until unpacking is complete, we invalidate them on
2254 * src_index instead with the assumption that they will be copied to
2255 * dst_index at the end.
2256 *
2257 * [1] src_index->cache_tree is also used in unpack_callback() so if
2258 * we invalidate o->result, we need to update it to use
2259 * o->result.cache_tree as well.
2260 */
eb9ae4b5
RS
2261static void invalidate_ce_path(const struct cache_entry *ce,
2262 struct unpack_trees_options *o)
076b0adc 2263{
e931371a
NTND
2264 if (!ce)
2265 return;
2266 cache_tree_invalidate_path(o->src_index, ce->name);
0cacebf0 2267 untracked_cache_invalidate_path(o->src_index, ce->name, 1);
076b0adc
JS
2268}
2269
0cf73755
SV
2270/*
2271 * Check that checking out ce->sha1 in subdir ce->name is not
2272 * going to overwrite any working files.
0cf73755 2273 */
d6b12300
SB
2274static int verify_clean_submodule(const char *old_sha1,
2275 const struct cache_entry *ce,
eb9ae4b5 2276 struct unpack_trees_options *o)
0cf73755 2277{
a7bc845a
SB
2278 if (!submodule_from_ce(ce))
2279 return 0;
2280
2281 return check_submodule_move_head(ce, old_sha1,
2282 oid_to_hex(&ce->oid), o);
0cf73755
SV
2283}
2284
eb9ae4b5 2285static int verify_clean_subdirectory(const struct cache_entry *ce,
eb9ae4b5 2286 struct unpack_trees_options *o)
c8193534
JH
2287{
2288 /*
0cf73755 2289 * we are about to extract "ce->name"; we would not want to lose
c8193534
JH
2290 * anything in the existing directory there.
2291 */
2292 int namelen;
7b9e3ce0 2293 int i;
c8193534
JH
2294 struct dir_struct d;
2295 char *pathbuf;
2296 int cnt = 0;
0cf73755 2297
d6b12300 2298 if (S_ISGITLINK(ce->ce_mode)) {
1053fe82 2299 struct object_id oid;
a98e6101 2300 int sub_head = resolve_gitlink_ref(ce->name, "HEAD", &oid);
d6b12300
SB
2301 /*
2302 * If we are not going to update the submodule, then
0cf73755
SV
2303 * we don't care.
2304 */
4a7e27e9 2305 if (!sub_head && oideq(&oid, &ce->oid))
0cf73755 2306 return 0;
1053fe82 2307 return verify_clean_submodule(sub_head ? NULL : oid_to_hex(&oid),
df351c6e 2308 ce, o);
0cf73755 2309 }
c8193534
JH
2310
2311 /*
2312 * First let's make sure we do not have a local modification
2313 * in that directory.
2314 */
68c4f6a5 2315 namelen = ce_namelen(ce);
da165f47
JH
2316 for (i = locate_in_src_index(ce, o);
2317 i < o->src_index->cache_nr;
2318 i++) {
837e5fe9
CB
2319 struct cache_entry *ce2 = o->src_index->cache[i];
2320 int len = ce_namelen(ce2);
c8193534 2321 if (len < namelen ||
837e5fe9
CB
2322 strncmp(ce->name, ce2->name, namelen) ||
2323 ce2->name[namelen] != '/')
c8193534
JH
2324 break;
2325 /*
da165f47
JH
2326 * ce2->name is an entry in the subdirectory to be
2327 * removed.
c8193534 2328 */
837e5fe9
CB
2329 if (!ce_stage(ce2)) {
2330 if (verify_uptodate(ce2, o))
203a2fe1 2331 return -1;
837e5fe9 2332 add_entry(o, ce2, CE_REMOVE, 0);
5697ca9a 2333 invalidate_ce_path(ce, o);
da165f47 2334 mark_ce_used(ce2, o);
c8193534
JH
2335 }
2336 cnt++;
2337 }
2338
b817e545 2339 /* Do not lose a locally present file that is not ignored. */
75faa45a 2340 pathbuf = xstrfmt("%.*s/", namelen, ce->name);
c8193534
JH
2341
2342 memset(&d, 0, sizeof(d));
2343 if (o->dir)
b413a827 2344 setup_standard_excludes(&d);
c7f3259d 2345 i = read_directory(&d, o->src_index, pathbuf, namelen+1, NULL);
e5a917fc
ÆAB
2346 dir_clear(&d);
2347 free(pathbuf);
c8193534 2348 if (i)
191e9d2c 2349 return add_rejected_path(o, ERROR_NOT_UPTODATE_DIR, ce->name);
b817e545
EN
2350
2351 /* Do not lose startup_info->original_cwd */
2352 if (startup_info->original_cwd &&
2353 !strcmp(startup_info->original_cwd, ce->name))
2354 return add_rejected_path(o, ERROR_CWD_IN_THE_WAY, ce->name);
2355
c8193534
JH
2356 return cnt;
2357}
2358
32260ad5
LT
2359/*
2360 * This gets called when there was no index entry for the tree entry 'dst',
2361 * but we found a file in the working tree that 'lstat()' said was fine,
2362 * and we're on a case-insensitive filesystem.
2363 *
2364 * See if we can find a case-insensitive match in the index that also
2365 * matches the stat information, and assume it's that other file!
2366 */
a9307f5a 2367static int icase_exists(struct unpack_trees_options *o, const char *name, int len, struct stat *st)
32260ad5 2368{
9c5e6c80 2369 const struct cache_entry *src;
32260ad5 2370
ebbd7439 2371 src = index_file_exists(o->src_index, name, len, 1);
56cac48c 2372 return src && !ie_match_stat(o->src_index, src, st, CE_MATCH_IGNORE_VALID|CE_MATCH_IGNORE_SKIP_WORKTREE);
32260ad5
LT
2373}
2374
1fdd51aa
EN
2375enum absent_checking_type {
2376 COMPLETELY_ABSENT,
2377 ABSENT_ANY_DIRECTORY
2378};
2379
a9307f5a 2380static int check_ok_to_remove(const char *name, int len, int dtype,
eb9ae4b5 2381 const struct cache_entry *ce, struct stat *st,
a9307f5a 2382 enum unpack_trees_error_types error_type,
1fdd51aa 2383 enum absent_checking_type absent_type,
a9307f5a
CB
2384 struct unpack_trees_options *o)
2385{
9c5e6c80 2386 const struct cache_entry *result;
a9307f5a
CB
2387
2388 /*
2389 * It may be that the 'lstat()' succeeded even though
2390 * target 'ce' was absent, because there is an old
2391 * entry that is different only in case..
2392 *
2393 * Ignore that lstat() if it matches.
2394 */
2395 if (ignore_case && icase_exists(o, name, len, st))
2396 return 0;
2397
589570db 2398 if (o->dir &&
c7f3259d 2399 is_excluded(o->dir, o->src_index, name, &dtype))
a9307f5a
CB
2400 /*
2401 * ce->name is explicitly excluded, so it is Ok to
2402 * overwrite it.
2403 */
2404 return 0;
2405 if (S_ISDIR(st->st_mode)) {
2406 /*
2407 * We are checking out path "foo" and
2408 * found "foo/." in the working tree.
2409 * This is tricky -- if we have modified
2410 * files that are in "foo/" we would lose
2411 * them.
2412 */
df351c6e 2413 if (verify_clean_subdirectory(ce, o) < 0)
a9307f5a
CB
2414 return -1;
2415 return 0;
2416 }
2417
1fdd51aa
EN
2418 /* If we only care about directories, then we can remove */
2419 if (absent_type == ABSENT_ANY_DIRECTORY)
2420 return 0;
2421
a9307f5a
CB
2422 /*
2423 * The previous round may already have decided to
2424 * delete this path, which is in a subdirectory that
2425 * is being replaced with a blob.
2426 */
ebbd7439 2427 result = index_file_exists(&o->result, name, len, 0);
a9307f5a
CB
2428 if (result) {
2429 if (result->ce_flags & CE_REMOVE)
2430 return 0;
2431 }
2432
191e9d2c 2433 return add_rejected_path(o, error_type, name);
a9307f5a
CB
2434}
2435
076b0adc
JS
2436/*
2437 * We do not want to remove or overwrite a working tree file that
f8a9d428 2438 * is not tracked, unless it is ignored.
076b0adc 2439 */
eb9ae4b5
RS
2440static int verify_absent_1(const struct cache_entry *ce,
2441 enum unpack_trees_error_types error_type,
1fdd51aa 2442 enum absent_checking_type absent_type,
eb9ae4b5 2443 struct unpack_trees_options *o)
076b0adc 2444{
f66caaf9 2445 int len;
076b0adc
JS
2446 struct stat st;
2447
0b0ee338 2448 if (o->index_only || !o->update)
203a2fe1 2449 return 0;
c8193534 2450
0b0ee338
EN
2451 if (o->reset == UNPACK_RESET_OVERWRITE_UNTRACKED) {
2452 /* Avoid nuking startup_info->original_cwd... */
2453 if (startup_info->original_cwd &&
2454 !strcmp(startup_info->original_cwd, ce->name))
2455 return add_rejected_path(o, ERROR_CWD_IN_THE_WAY,
2456 ce->name);
2457 /* ...but nuke anything else. */
203a2fe1 2458 return 0;
0b0ee338 2459 }
c8193534 2460
fab78a0c 2461 len = check_leading_path(ce->name, ce_namelen(ce), 0);
f66caaf9 2462 if (!len)
203a2fe1 2463 return 0;
f66caaf9 2464 else if (len > 0) {
f514ef97
JK
2465 char *path;
2466 int ret;
2467
2468 path = xmemdupz(ce->name, len);
a93e5301 2469 if (lstat(path, &st))
43c728e2 2470 ret = error_errno("cannot stat '%s'", path);
a7bc845a
SB
2471 else {
2472 if (submodule_from_ce(ce))
2473 ret = check_submodule_move_head(ce,
2474 oid_to_hex(&ce->oid),
2475 NULL, o);
2476 else
2477 ret = check_ok_to_remove(path, len, DT_UNKNOWN, NULL,
1fdd51aa
EN
2478 &st, error_type,
2479 absent_type, o);
a7bc845a 2480 }
f514ef97
JK
2481 free(path);
2482 return ret;
92fda79e
JN
2483 } else if (lstat(ce->name, &st)) {
2484 if (errno != ENOENT)
43c728e2 2485 return error_errno("cannot stat '%s'", ce->name);
92fda79e
JN
2486 return 0;
2487 } else {
a7bc845a
SB
2488 if (submodule_from_ce(ce))
2489 return check_submodule_move_head(ce, oid_to_hex(&ce->oid),
2490 NULL, o);
2491
a9307f5a 2492 return check_ok_to_remove(ce->name, ce_namelen(ce),
92fda79e 2493 ce_to_dtype(ce), ce, &st,
1fdd51aa 2494 error_type, absent_type, o);
92fda79e 2495 }
076b0adc 2496}
a9307f5a 2497
eb9ae4b5 2498static int verify_absent(const struct cache_entry *ce,
08402b04 2499 enum unpack_trees_error_types error_type,
35a5aa79
NTND
2500 struct unpack_trees_options *o)
2501{
2431afbf 2502 if (!o->skip_sparse_checkout && (ce->ce_flags & CE_NEW_SKIP_WORKTREE))
f1f523ea 2503 return 0;
1fdd51aa
EN
2504 return verify_absent_1(ce, error_type, COMPLETELY_ABSENT, o);
2505}
2506
2507static int verify_absent_if_directory(const struct cache_entry *ce,
2508 enum unpack_trees_error_types error_type,
2509 struct unpack_trees_options *o)
2510{
2511 if (!o->skip_sparse_checkout && (ce->ce_flags & CE_NEW_SKIP_WORKTREE))
2512 return 0;
2513 return verify_absent_1(ce, error_type, ABSENT_ANY_DIRECTORY, o);
35a5aa79 2514}
076b0adc 2515
eb9ae4b5
RS
2516static int verify_absent_sparse(const struct cache_entry *ce,
2517 enum unpack_trees_error_types error_type,
2518 struct unpack_trees_options *o)
e800ec9d 2519{
1fdd51aa 2520 return verify_absent_1(ce, error_type, COMPLETELY_ABSENT, o);
e800ec9d 2521}
076b0adc 2522
f2fa3542 2523static int merged_entry(const struct cache_entry *ce,
eb9ae4b5 2524 const struct cache_entry *old,
f2fa3542 2525 struct unpack_trees_options *o)
076b0adc 2526{
7f8ab8dc 2527 int update = CE_UPDATE;
8e72d675 2528 struct cache_entry *merge = dup_cache_entry(ce, &o->result);
7f8ab8dc 2529
e11d7b59 2530 if (!old) {
2431afbf
NTND
2531 /*
2532 * New index entries. In sparse checkout, the following
2533 * verify_absent() will be delayed until after
2534 * traverse_trees() finishes in unpack_trees(), then:
2535 *
2536 * - CE_NEW_SKIP_WORKTREE will be computed correctly
2537 * - verify_absent() be called again, this time with
2538 * correct CE_NEW_SKIP_WORKTREE
2539 *
2540 * verify_absent() call here does nothing in sparse
2541 * checkout (i.e. o->skip_sparse_checkout == 0)
2542 */
2543 update |= CE_ADDED;
2544 merge->ce_flags |= CE_NEW_SKIP_WORKTREE;
2545
f2fa3542
RS
2546 if (verify_absent(merge,
2547 ERROR_WOULD_LOSE_UNTRACKED_OVERWRITTEN, o)) {
a849735b 2548 discard_cache_entry(merge);
e11d7b59 2549 return -1;
f2fa3542 2550 }
e11d7b59 2551 invalidate_ce_path(merge, o);
a7bc845a 2552
e84704f1 2553 if (submodule_from_ce(ce) && file_exists(ce->name)) {
a7bc845a
SB
2554 int ret = check_submodule_move_head(ce, NULL,
2555 oid_to_hex(&ce->oid),
2556 o);
2557 if (ret)
2558 return ret;
2559 }
2560
e11d7b59 2561 } else if (!(old->ce_flags & CE_CONFLICTED)) {
076b0adc
JS
2562 /*
2563 * See if we can re-use the old CE directly?
2564 * That way we get the uptodate stat info.
2565 *
7f8ab8dc
LT
2566 * This also removes the UPDATE flag on a match; otherwise
2567 * we will end up overwriting local changes in the work tree.
076b0adc
JS
2568 */
2569 if (same(old, merge)) {
eb7a2f1d 2570 copy_cache_entry(merge, old);
7f8ab8dc 2571 update = 0;
076b0adc 2572 } else {
f2fa3542 2573 if (verify_uptodate(old, o)) {
a849735b 2574 discard_cache_entry(merge);
203a2fe1 2575 return -1;
f2fa3542 2576 }
2431afbf
NTND
2577 /* Migrate old flags over */
2578 update |= old->ce_flags & (CE_SKIP_WORKTREE | CE_NEW_SKIP_WORKTREE);
bc052d7f 2579 invalidate_ce_path(old, o);
076b0adc 2580 }
a7bc845a 2581
e84704f1 2582 if (submodule_from_ce(ce) && file_exists(ce->name)) {
a7bc845a
SB
2583 int ret = check_submodule_move_head(ce, oid_to_hex(&old->oid),
2584 oid_to_hex(&ce->oid),
2585 o);
2586 if (ret)
2587 return ret;
2588 }
e11d7b59
JH
2589 } else {
2590 /*
2591 * Previously unmerged entry left as an existence
2592 * marker by read_index_unmerged();
2593 */
1fdd51aa
EN
2594 if (verify_absent_if_directory(merge,
2595 ERROR_WOULD_LOSE_UNTRACKED_OVERWRITTEN, o)) {
2596 discard_cache_entry(merge);
2597 return -1;
2598 }
2599
e11d7b59 2600 invalidate_ce_path(old, o);
076b0adc
JS
2601 }
2602
cc756edd
JS
2603 if (do_add_entry(o, merge, update, CE_STAGEMASK) < 0)
2604 return -1;
076b0adc
JS
2605 return 1;
2606}
2607
ab81047a
VD
2608static int merged_sparse_dir(const struct cache_entry * const *src, int n,
2609 struct unpack_trees_options *o)
2610{
2611 struct tree_desc t[MAX_UNPACK_TREES + 1];
2612 void * tree_bufs[MAX_UNPACK_TREES + 1];
2613 struct traverse_info info;
2614 int i, ret;
2615
2616 /*
2617 * Create the tree traversal information for traversing into *only* the
2618 * sparse directory.
2619 */
2620 setup_traverse_info(&info, src[0]->name);
2621 info.fn = unpack_sparse_callback;
2622 info.data = o;
2623 info.show_all_errors = o->show_all_errors;
2624 info.pathspec = o->pathspec;
2625
2626 /* Get the tree descriptors of the sparse directory in each of the merging trees */
2627 for (i = 0; i < n; i++)
2628 tree_bufs[i] = fill_tree_descriptor(o->src_index->repo, &t[i],
2629 src[i] && !is_null_oid(&src[i]->oid) ? &src[i]->oid : NULL);
2630
2631 ret = traverse_trees(o->src_index, n, t, &info);
2632
2633 for (i = 0; i < n; i++)
2634 free(tree_bufs[i]);
2635
2636 return ret;
2637}
2638
eb9ae4b5
RS
2639static int deleted_entry(const struct cache_entry *ce,
2640 const struct cache_entry *old,
2641 struct unpack_trees_options *o)
076b0adc 2642{
34110cd4
LT
2643 /* Did it exist in the index? */
2644 if (!old) {
08402b04 2645 if (verify_absent(ce, ERROR_WOULD_LOSE_UNTRACKED_REMOVED, o))
203a2fe1 2646 return -1;
34110cd4 2647 return 0;
56d06fe4
EN
2648 } else if (verify_absent_if_directory(ce, ERROR_WOULD_LOSE_UNTRACKED_REMOVED, o)) {
2649 return -1;
34110cd4 2650 }
56d06fe4 2651
e11d7b59 2652 if (!(old->ce_flags & CE_CONFLICTED) && verify_uptodate(old, o))
34110cd4
LT
2653 return -1;
2654 add_entry(o, ce, CE_REMOVE, 0);
bc052d7f 2655 invalidate_ce_path(ce, o);
076b0adc
JS
2656 return 1;
2657}
2658
eb9ae4b5
RS
2659static int keep_entry(const struct cache_entry *ce,
2660 struct unpack_trees_options *o)
076b0adc 2661{
34110cd4 2662 add_entry(o, ce, 0, 0);
5697ca9a
NTND
2663 if (ce_stage(ce))
2664 invalidate_ce_path(ce, o);
076b0adc
JS
2665 return 1;
2666}
2667
2668#if DBRT_DEBUG
2669static void show_stage_entry(FILE *o,
2670 const char *label, const struct cache_entry *ce)
2671{
2672 if (!ce)
2673 fprintf(o, "%s (missing)\n", label);
2674 else
2675 fprintf(o, "%s%06o %s %d\t%s\n",
2676 label,
7a51ed66 2677 ce->ce_mode,
99d1a986 2678 oid_to_hex(&ce->oid),
076b0adc
JS
2679 ce_stage(ce),
2680 ce->name);
2681}
2682#endif
2683
5828e835
RS
2684int threeway_merge(const struct cache_entry * const *stages,
2685 struct unpack_trees_options *o)
076b0adc 2686{
eb9ae4b5
RS
2687 const struct cache_entry *index;
2688 const struct cache_entry *head;
2689 const struct cache_entry *remote = stages[o->head_idx + 1];
076b0adc
JS
2690 int count;
2691 int head_match = 0;
2692 int remote_match = 0;
076b0adc
JS
2693
2694 int df_conflict_head = 0;
2695 int df_conflict_remote = 0;
2696
2697 int any_anc_missing = 0;
2698 int no_anc_exists = 1;
2699 int i;
2700
2701 for (i = 1; i < o->head_idx; i++) {
4c4caafc 2702 if (!stages[i] || stages[i] == o->df_conflict_entry)
076b0adc 2703 any_anc_missing = 1;
4c4caafc 2704 else
076b0adc 2705 no_anc_exists = 0;
076b0adc
JS
2706 }
2707
2708 index = stages[0];
2709 head = stages[o->head_idx];
2710
2711 if (head == o->df_conflict_entry) {
2712 df_conflict_head = 1;
2713 head = NULL;
2714 }
2715
2716 if (remote == o->df_conflict_entry) {
2717 df_conflict_remote = 1;
2718 remote = NULL;
2719 }
2720
cee2d6ae
JH
2721 /*
2722 * First, if there's a #16 situation, note that to prevent #13
076b0adc
JS
2723 * and #14.
2724 */
2725 if (!same(remote, head)) {
2726 for (i = 1; i < o->head_idx; i++) {
2727 if (same(stages[i], head)) {
2728 head_match = i;
2729 }
2730 if (same(stages[i], remote)) {
2731 remote_match = i;
2732 }
2733 }
2734 }
2735
cee2d6ae
JH
2736 /*
2737 * We start with cases where the index is allowed to match
076b0adc
JS
2738 * something other than the head: #14(ALT) and #2ALT, where it
2739 * is permitted to match the result instead.
2740 */
2741 /* #14, #14ALT, #2ALT */
2742 if (remote && !df_conflict_head && head_match && !remote_match) {
f27c170f
VD
2743 if (index && !same(index, remote) && !same(index, head)) {
2744 if (S_ISSPARSEDIR(index->ce_mode))
2745 return merged_sparse_dir(stages, 4, o);
2746 else
2747 return reject_merge(index, o);
2748 }
076b0adc
JS
2749 return merged_entry(remote, index, o);
2750 }
2751 /*
2752 * If we have an entry in the index cache, then we want to
2753 * make sure that it matches head.
2754 */
f27c170f
VD
2755 if (index && !same(index, head)) {
2756 if (S_ISSPARSEDIR(index->ce_mode))
2757 return merged_sparse_dir(stages, 4, o);
2758 else
2759 return reject_merge(index, o);
2760 }
076b0adc
JS
2761
2762 if (head) {
2763 /* #5ALT, #15 */
2764 if (same(head, remote))
2765 return merged_entry(head, index, o);
2766 /* #13, #3ALT */
2767 if (!df_conflict_remote && remote_match && !head_match)
2768 return merged_entry(head, index, o);
2769 }
2770
2771 /* #1 */
34110cd4 2772 if (!head && !remote && any_anc_missing)
076b0adc
JS
2773 return 0;
2774
cee2d6ae
JH
2775 /*
2776 * Under the "aggressive" rule, we resolve mostly trivial
076b0adc
JS
2777 * cases that we historically had git-merge-one-file resolve.
2778 */
2779 if (o->aggressive) {
cee2d6ae
JH
2780 int head_deleted = !head;
2781 int remote_deleted = !remote;
eb9ae4b5 2782 const struct cache_entry *ce = NULL;
4c4caafc
JH
2783
2784 if (index)
0cf73755 2785 ce = index;
4c4caafc 2786 else if (head)
0cf73755 2787 ce = head;
4c4caafc 2788 else if (remote)
0cf73755 2789 ce = remote;
4c4caafc
JH
2790 else {
2791 for (i = 1; i < o->head_idx; i++) {
2792 if (stages[i] && stages[i] != o->df_conflict_entry) {
0cf73755 2793 ce = stages[i];
4c4caafc
JH
2794 break;
2795 }
2796 }
2797 }
2798
076b0adc
JS
2799 /*
2800 * Deleted in both.
2801 * Deleted in one and unchanged in the other.
2802 */
2803 if ((head_deleted && remote_deleted) ||
2804 (head_deleted && remote && remote_match) ||
2805 (remote_deleted && head && head_match)) {
2806 if (index)
2807 return deleted_entry(index, index, o);
34110cd4 2808 if (ce && !head_deleted) {
08402b04 2809 if (verify_absent(ce, ERROR_WOULD_LOSE_UNTRACKED_REMOVED, o))
203a2fe1
DB
2810 return -1;
2811 }
076b0adc
JS
2812 return 0;
2813 }
2814 /*
2815 * Added in both, identically.
2816 */
2817 if (no_anc_exists && head && remote && same(head, remote))
2818 return merged_entry(head, index, o);
2819
2820 }
2821
f27c170f 2822 /* Handle "no merge" cases (see t/t1000-read-tree-m-3way.sh) */
076b0adc 2823 if (index) {
f27c170f
VD
2824 /*
2825 * If we've reached the "no merge" cases and we're merging
2826 * a sparse directory, we may have an "edit/edit" conflict that
2827 * can be resolved by individually merging directory contents.
2828 */
2829 if (S_ISSPARSEDIR(index->ce_mode))
2830 return merged_sparse_dir(stages, 4, o);
2831
2832 /*
2833 * If we're not merging a sparse directory, ensure the index is
2834 * up-to-date to avoid files getting overwritten with conflict
2835 * resolution files
2836 */
203a2fe1
DB
2837 if (verify_uptodate(index, o))
2838 return -1;
076b0adc 2839 }
076b0adc
JS
2840
2841 o->nontrivial_merge = 1;
2842
ea4b52a8 2843 /* #2, #3, #4, #6, #7, #9, #10, #11. */
076b0adc
JS
2844 count = 0;
2845 if (!head_match || !remote_match) {
2846 for (i = 1; i < o->head_idx; i++) {
4c4caafc 2847 if (stages[i] && stages[i] != o->df_conflict_entry) {
7f7932ab 2848 keep_entry(stages[i], o);
076b0adc
JS
2849 count++;
2850 break;
2851 }
2852 }
2853 }
2854#if DBRT_DEBUG
2855 else {
2856 fprintf(stderr, "read-tree: warning #16 detected\n");
2857 show_stage_entry(stderr, "head ", stages[head_match]);
2858 show_stage_entry(stderr, "remote ", stages[remote_match]);
2859 }
2860#endif
7f7932ab
JH
2861 if (head) { count += keep_entry(head, o); }
2862 if (remote) { count += keep_entry(remote, o); }
076b0adc
JS
2863 return count;
2864}
2865
2866/*
2867 * Two-way merge.
2868 *
2869 * The rule is to "carry forward" what is in the index without losing
a75d7b54 2870 * information across a "fast-forward", favoring a successful merge
076b0adc
JS
2871 * over a merge failure when it makes sense. For details of the
2872 * "carry forward" rule, please see <Documentation/git-read-tree.txt>.
2873 *
2874 */
5828e835
RS
2875int twoway_merge(const struct cache_entry * const *src,
2876 struct unpack_trees_options *o)
076b0adc 2877{
eb9ae4b5
RS
2878 const struct cache_entry *current = src[0];
2879 const struct cache_entry *oldtree = src[1];
2880 const struct cache_entry *newtree = src[2];
076b0adc
JS
2881
2882 if (o->merge_size != 2)
2883 return error("Cannot do a twoway merge of %d trees",
2884 o->merge_size);
2885
b8ba1535
JH
2886 if (oldtree == o->df_conflict_entry)
2887 oldtree = NULL;
2888 if (newtree == o->df_conflict_entry)
2889 newtree = NULL;
2890
076b0adc 2891 if (current) {
b018ff60
JK
2892 if (current->ce_flags & CE_CONFLICTED) {
2893 if (same(oldtree, newtree) || o->reset) {
2894 if (!newtree)
2895 return deleted_entry(current, current, o);
2896 else
2897 return merged_entry(newtree, current, o);
2898 }
6a143aa2 2899 return reject_merge(current, o);
6c1db1b3 2900 } else if ((!oldtree && !newtree) || /* 4 and 5 */
b018ff60
JK
2901 (!oldtree && newtree &&
2902 same(current, newtree)) || /* 6 and 7 */
2903 (oldtree && newtree &&
2904 same(oldtree, newtree)) || /* 14 and 15 */
2905 (oldtree && newtree &&
2906 !same(oldtree, newtree) && /* 18 and 19 */
2907 same(current, newtree))) {
7f7932ab 2908 return keep_entry(current, o);
6c1db1b3 2909 } else if (oldtree && !newtree && same(current, oldtree)) {
076b0adc
JS
2910 /* 10 or 11 */
2911 return deleted_entry(oldtree, current, o);
6c1db1b3 2912 } else if (oldtree && newtree &&
076b0adc
JS
2913 same(current, oldtree) && !same(current, newtree)) {
2914 /* 20 or 21 */
2915 return merged_entry(newtree, current, o);
e05cdb17
DS
2916 } else if (current && !oldtree && newtree &&
2917 S_ISSPARSEDIR(current->ce_mode) != S_ISSPARSEDIR(newtree->ce_mode) &&
2918 ce_stage(current) == 0) {
2919 /*
2920 * This case is a directory/file conflict across the sparse-index
2921 * boundary. When we are changing from one path to another via
2922 * 'git checkout', then we want to replace one entry with another
2923 * via merged_entry(). If there are staged changes, then we should
2924 * reject the merge instead.
2925 */
2926 return merged_entry(newtree, current, o);
ab81047a
VD
2927 } else if (S_ISSPARSEDIR(current->ce_mode)) {
2928 /*
2929 * The sparse directories differ, but we don't know whether that's
2930 * because of two different files in the directory being modified
2931 * (can be trivially merged) or if there is a real file conflict.
2932 * Merge the sparse directory by OID to compare file-by-file.
2933 */
2934 return merged_sparse_dir(src, 3, o);
6c1db1b3 2935 } else
6a143aa2 2936 return reject_merge(current, o);
076b0adc 2937 }
55218834
JH
2938 else if (newtree) {
2939 if (oldtree && !o->initial_checkout) {
2940 /*
2941 * deletion of the path was staged;
2942 */
2943 if (same(oldtree, newtree))
2944 return 1;
2945 return reject_merge(oldtree, o);
2946 }
076b0adc 2947 return merged_entry(newtree, current, o);
55218834 2948 }
d699676d 2949 return deleted_entry(oldtree, current, o);
076b0adc
JS
2950}
2951
2952/*
2953 * Bind merge.
2954 *
2955 * Keep the index entries at stage0, collapse stage1 but make sure
2956 * stage0 does not have anything there.
2957 */
5828e835
RS
2958int bind_merge(const struct cache_entry * const *src,
2959 struct unpack_trees_options *o)
076b0adc 2960{
eb9ae4b5
RS
2961 const struct cache_entry *old = src[0];
2962 const struct cache_entry *a = src[1];
076b0adc
JS
2963
2964 if (o->merge_size != 1)
82247e9b 2965 return error("Cannot do a bind merge of %d trees",
076b0adc
JS
2966 o->merge_size);
2967 if (a && old)
b165fac8 2968 return o->quiet ? -1 :
3d415425 2969 error(ERRORMSG(o, ERROR_BIND_OVERLAP),
4002ec3d
ÆAB
2970 super_prefixed(a->name, o->super_prefix),
2971 super_prefixed(old->name, o->super_prefix));
076b0adc 2972 if (!a)
7f7932ab 2973 return keep_entry(old, o);
076b0adc
JS
2974 else
2975 return merged_entry(a, NULL, o);
2976}
2977
2978/*
2979 * One-way merge.
2980 *
2981 * The rule is:
2982 * - take the stat information from stage0, take the data from stage1
2983 */
5828e835
RS
2984int oneway_merge(const struct cache_entry * const *src,
2985 struct unpack_trees_options *o)
076b0adc 2986{
eb9ae4b5
RS
2987 const struct cache_entry *old = src[0];
2988 const struct cache_entry *a = src[1];
076b0adc
JS
2989
2990 if (o->merge_size != 1)
2991 return error("Cannot do a oneway merge of %d trees",
2992 o->merge_size);
2993
78d3b06e 2994 if (!a || a == o->df_conflict_entry)
076b0adc 2995 return deleted_entry(old, old, o);
34110cd4 2996
076b0adc 2997 if (old && same(old, a)) {
34110cd4 2998 int update = 0;
679f2f9f
US
2999 if (o->reset && o->update && !ce_uptodate(old) && !ce_skip_worktree(old) &&
3000 !(old->ce_flags & CE_FSMONITOR_VALID)) {
076b0adc
JS
3001 struct stat st;
3002 if (lstat(old->name, &st) ||
56cac48c 3003 ie_match_stat(o->src_index, old, &st, CE_MATCH_IGNORE_VALID|CE_MATCH_IGNORE_SKIP_WORKTREE))
34110cd4 3004 update |= CE_UPDATE;
076b0adc 3005 }
ad17312e
SB
3006 if (o->update && S_ISGITLINK(old->ce_mode) &&
3007 should_update_submodules() && !verify_uptodate(old, o))
3008 update |= CE_UPDATE;
ab5af825 3009 add_entry(o, old, update, CE_STAGEMASK);
34110cd4 3010 return 0;
076b0adc
JS
3011 }
3012 return merged_entry(a, old, o);
3013}
d3c7bf73
DL
3014
3015/*
3016 * Merge worktree and untracked entries in a stash entry.
3017 *
3018 * Ignore all index entries. Collapse remaining trees but make sure that they
3019 * don't have any conflicting files.
3020 */
3021int stash_worktree_untracked_merge(const struct cache_entry * const *src,
3022 struct unpack_trees_options *o)
3023{
3024 const struct cache_entry *worktree = src[1];
3025 const struct cache_entry *untracked = src[2];
3026
3027 if (o->merge_size != 2)
3028 BUG("invalid merge_size: %d", o->merge_size);
3029
3030 if (worktree && untracked)
3031 return error(_("worktree and untracked commit have duplicate entries: %s"),
4002ec3d 3032 super_prefixed(worktree->name, o->super_prefix));
d3c7bf73
DL
3033
3034 return merged_entry(worktree ? worktree : untracked, NULL, o);
3035}