connected.c 3.04 KB
Newer Older
1 2 3 4
#include "cache.h"
#include "run-command.h"
#include "sigchain.h"
#include "connected.h"
5
#include "transport.h"
6
#include "packfile.h"
7 8 9 10

/*
 * If we feed all the commits we want to verify to this command
 *
11
 *  $ git rev-list --objects --stdin --not --all
12 13
 *
 * and if it does not error out, that means everything reachable from
14 15
 * these commits locally exists and is connected to our existing refs.
 * Note that this does _not_ validate the individual objects.
16 17 18
 *
 * Returns 0 if everything is connected, non-zero otherwise.
 */
19
int check_connected(oid_iterate_fn fn, void *cb_data,
20
		    struct check_connected_options *opt)
21
{
22
	struct child_process rev_list = CHILD_PROCESS_INIT;
23
	struct check_connected_options defaults = CHECK_CONNECTED_INIT;
24 25
	char commit[GIT_MAX_HEXSZ + 1];
	struct object_id oid;
26
	int err = 0;
27
	struct packed_git *new_pack = NULL;
28
	struct transport *transport;
29
	size_t base_len;
30

31 32 33 34
	if (!opt)
		opt = &defaults;
	transport = opt->transport;

35
	if (fn(cb_data, &oid)) {
36 37
		if (opt->err_fd)
			close(opt->err_fd);
38
		return err;
39
	}
40

41 42 43
	if (transport && transport->smart_options &&
	    transport->smart_options->self_contained_and_connected &&
	    transport->pack_lockfile &&
44
	    strip_suffix(transport->pack_lockfile, ".keep", &base_len)) {
45
		struct strbuf idx_file = STRBUF_INIT;
46
		strbuf_add(&idx_file, transport->pack_lockfile, base_len);
47 48 49 50 51
		strbuf_addstr(&idx_file, ".idx");
		new_pack = add_packed_git(idx_file.buf, idx_file.len, 1);
		strbuf_release(&idx_file);
	}

52
	if (opt->shallow_file) {
53
		argv_array_push(&rev_list.args, "--shallow-file");
54
		argv_array_push(&rev_list.args, opt->shallow_file);
55
	}
56 57 58 59 60 61
	argv_array_push(&rev_list.args,"rev-list");
	argv_array_push(&rev_list.args, "--objects");
	argv_array_push(&rev_list.args, "--stdin");
	argv_array_push(&rev_list.args, "--not");
	argv_array_push(&rev_list.args, "--all");
	argv_array_push(&rev_list.args, "--quiet");
62 63 64
	if (opt->progress)
		argv_array_pushf(&rev_list.args, "--progress=%s",
				 _("Checking connectivity"));
65 66

	rev_list.git_cmd = 1;
67
	rev_list.env = opt->env;
68 69
	rev_list.in = -1;
	rev_list.no_stdout = 1;
70 71 72 73 74
	if (opt->err_fd)
		rev_list.err = opt->err_fd;
	else
		rev_list.no_stderr = opt->quiet;

75 76 77 78 79
	if (start_command(&rev_list))
		return error(_("Could not run 'git rev-list'"));

	sigchain_push(SIGPIPE, SIG_IGN);

80
	commit[GIT_SHA1_HEXSZ] = '\n';
81
	do {
82 83 84 85 86 87 88 89
		/*
		 * If index-pack already checked that:
		 * - there are no dangling pointers in the new pack
		 * - the pack is self contained
		 * Then if the updated ref is in the new pack, then we
		 * are sure the ref is good and not sending it to
		 * rev-list for verification.
		 */
90
		if (new_pack && find_pack_entry_one(oid.hash, new_pack))
91 92
			continue;

93 94
		memcpy(commit, oid_to_hex(&oid), GIT_SHA1_HEXSZ);
		if (write_in_full(rev_list.in, commit, GIT_SHA1_HEXSZ + 1) < 0) {
95
			if (errno != EPIPE && errno != EINVAL)
Duy Nguyen's avatar
Duy Nguyen committed
96
				error_errno(_("failed write to rev-list"));
97 98 99
			err = -1;
			break;
		}
100
	} while (!fn(cb_data, &oid));
101

Duy Nguyen's avatar
Duy Nguyen committed
102 103
	if (close(rev_list.in))
		err = error_errno(_("failed to close rev-list's stdin"));
104 105 106 107

	sigchain_pop(SIGPIPE);
	return finish_command(&rev_list) || err;
}