fetch-pack: implement ref-in-want
[git/git.git] / connected.c
CommitLineData
f96400cb
JH
1#include "cache.h"
2#include "run-command.h"
3#include "sigchain.h"
4#include "connected.h"
c6807a40 5#include "transport.h"
0abe14f6 6#include "packfile.h"
f96400cb
JH
7
8/*
9 * If we feed all the commits we want to verify to this command
10 *
d21c463d 11 * $ git rev-list --objects --stdin --not --all
f96400cb
JH
12 *
13 * and if it does not error out, that means everything reachable from
d21c463d
JH
14 * these commits locally exists and is connected to our existing refs.
15 * Note that this does _not_ validate the individual objects.
f96400cb
JH
16 *
17 * Returns 0 if everything is connected, non-zero otherwise.
18 */
6ccac9ee 19int check_connected(oid_iterate_fn fn, void *cb_data,
7043c707 20 struct check_connected_options *opt)
f96400cb 21{
d3180279 22 struct child_process rev_list = CHILD_PROCESS_INIT;
7043c707 23 struct check_connected_options defaults = CHECK_CONNECTED_INIT;
6ccac9ee 24 char commit[GIT_MAX_HEXSZ + 1];
25 struct object_id oid;
3be89f9b 26 int err = 0;
c6807a40 27 struct packed_git *new_pack = NULL;
7043c707 28 struct transport *transport;
26936bfd 29 size_t base_len;
f96400cb 30
7043c707
JK
31 if (!opt)
32 opt = &defaults;
33 transport = opt->transport;
34
6ccac9ee 35 if (fn(cb_data, &oid)) {
e0331849
JK
36 if (opt->err_fd)
37 close(opt->err_fd);
f96400cb 38 return err;
e0331849 39 }
f96400cb 40
c6807a40
NTND
41 if (transport && transport->smart_options &&
42 transport->smart_options->self_contained_and_connected &&
43 transport->pack_lockfile &&
26936bfd 44 strip_suffix(transport->pack_lockfile, ".keep", &base_len)) {
c6807a40 45 struct strbuf idx_file = STRBUF_INIT;
26936bfd 46 strbuf_add(&idx_file, transport->pack_lockfile, base_len);
c6807a40
NTND
47 strbuf_addstr(&idx_file, ".idx");
48 new_pack = add_packed_git(idx_file.buf, idx_file.len, 1);
49 strbuf_release(&idx_file);
50 }
51
7043c707 52 if (opt->shallow_file) {
3be89f9b 53 argv_array_push(&rev_list.args, "--shallow-file");
7043c707 54 argv_array_push(&rev_list.args, opt->shallow_file);
614db3e2 55 }
3be89f9b
JK
56 argv_array_push(&rev_list.args,"rev-list");
57 argv_array_push(&rev_list.args, "--objects");
58 argv_array_push(&rev_list.args, "--stdin");
acb0c572
JH
59 if (repository_format_partial_clone)
60 argv_array_push(&rev_list.args, "--exclude-promisor-objects");
3be89f9b
JK
61 argv_array_push(&rev_list.args, "--not");
62 argv_array_push(&rev_list.args, "--all");
63 argv_array_push(&rev_list.args, "--quiet");
70d5e2d7
JK
64 if (opt->progress)
65 argv_array_pushf(&rev_list.args, "--progress=%s",
66 _("Checking connectivity"));
f96400cb 67
f96400cb 68 rev_list.git_cmd = 1;
526f108a 69 rev_list.env = opt->env;
f96400cb
JH
70 rev_list.in = -1;
71 rev_list.no_stdout = 1;
e0331849
JK
72 if (opt->err_fd)
73 rev_list.err = opt->err_fd;
74 else
75 rev_list.no_stderr = opt->quiet;
76
f96400cb
JH
77 if (start_command(&rev_list))
78 return error(_("Could not run 'git rev-list'"));
79
80 sigchain_push(SIGPIPE, SIG_IGN);
81
6ccac9ee 82 commit[GIT_SHA1_HEXSZ] = '\n';
f96400cb 83 do {
c6807a40
NTND
84 /*
85 * If index-pack already checked that:
86 * - there are no dangling pointers in the new pack
87 * - the pack is self contained
88 * Then if the updated ref is in the new pack, then we
89 * are sure the ref is good and not sending it to
90 * rev-list for verification.
91 */
6ccac9ee 92 if (new_pack && find_pack_entry_one(oid.hash, new_pack))
c6807a40
NTND
93 continue;
94
6ccac9ee 95 memcpy(commit, oid_to_hex(&oid), GIT_SHA1_HEXSZ);
96 if (write_in_full(rev_list.in, commit, GIT_SHA1_HEXSZ + 1) < 0) {
f96400cb 97 if (errno != EPIPE && errno != EINVAL)
5cc026e2 98 error_errno(_("failed write to rev-list"));
f96400cb
JH
99 err = -1;
100 break;
101 }
6ccac9ee 102 } while (!fn(cb_data, &oid));
f96400cb 103
5cc026e2
NTND
104 if (close(rev_list.in))
105 err = error_errno(_("failed to close rev-list's stdin"));
f96400cb
JH
106
107 sigchain_pop(SIGPIPE);
108 return finish_command(&rev_list) || err;
109}