14 static const char fetch_pack_usage[] =
15 "git-fetch-pack [--all] [-q] [-v] [-k] [--thin] [--exec=upload-pack] [--depth=<n>] [host:]directory <refs>...";
16 static const char *exec = "git-upload-pack";
18 #define COMPLETE (1U << 0)
19 #define COMMON (1U << 1)
20 #define COMMON_REF (1U << 2)
21 #define SEEN (1U << 3)
22 #define POPPED (1U << 4)
25 * After sending this many "have"s if we do not get any new ACK , we
26 * give up traversing our history.
28 #define MAX_IN_VAIN 256
30 static struct commit_list *rev_list;
31 static int non_common_revs, multi_ack, use_thin_pack, use_sideband;
33 static void rev_list_push(struct commit *commit, int mark)
35 if (!(commit->object.flags & mark)) {
36 commit->object.flags |= mark;
38 if (!(commit->object.parsed))
41 insert_by_date(commit, &rev_list);
43 if (!(commit->object.flags & COMMON))
48 static int rev_list_insert_ref(const char *path, const unsigned char *sha1, int flag, void *cb_data)
50 struct object *o = deref_tag(parse_object(sha1), path, 0);
52 if (o && o->type == OBJ_COMMIT)
53 rev_list_push((struct commit *)o, SEEN);
59 This function marks a rev and its ancestors as common.
60 In some cases, it is desirable to mark only the ancestors (for example
61 when only the server does not yet know that they are common).
64 static void mark_common(struct commit *commit,
65 int ancestors_only, int dont_parse)
67 if (commit != NULL && !(commit->object.flags & COMMON)) {
68 struct object *o = (struct object *)commit;
73 if (!(o->flags & SEEN))
74 rev_list_push(commit, SEEN);
76 struct commit_list *parents;
78 if (!ancestors_only && !(o->flags & POPPED))
80 if (!o->parsed && !dont_parse)
83 for (parents = commit->parents;
85 parents = parents->next)
86 mark_common(parents->item, 0, dont_parse);
92 Get the next rev to send, ignoring the common.
95 static const unsigned char* get_rev(void)
97 struct commit *commit = NULL;
99 while (commit == NULL) {
101 struct commit_list* parents;
103 if (rev_list == NULL || non_common_revs == 0)
106 commit = rev_list->item;
107 if (!(commit->object.parsed))
108 parse_commit(commit);
109 commit->object.flags |= POPPED;
110 if (!(commit->object.flags & COMMON))
113 parents = commit->parents;
115 if (commit->object.flags & COMMON) {
116 /* do not send "have", and ignore ancestors */
118 mark = COMMON | SEEN;
119 } else if (commit->object.flags & COMMON_REF)
120 /* send "have", and ignore ancestors */
121 mark = COMMON | SEEN;
123 /* send "have", also for its ancestors */
127 if (!(parents->item->object.flags & SEEN))
128 rev_list_push(parents->item, mark);
130 mark_common(parents->item, 1, 0);
131 parents = parents->next;
134 rev_list = rev_list->next;
137 return commit->object.sha1;
140 static int find_common(int fd[2], unsigned char *result_sha1,
144 int count = 0, flushes = 0, retval;
145 const unsigned char *sha1;
146 unsigned in_vain = 0;
147 int got_continue = 0;
149 for_each_ref(rev_list_insert_ref, NULL);
152 for ( ; refs ; refs = refs->next) {
153 unsigned char *remote = refs->old_sha1;
157 * If that object is complete (i.e. it is an ancestor of a
158 * local ref), we tell them we have it but do not have to
159 * tell them about its ancestors, which they already know
162 * We use lookup_object here because we are only
163 * interested in the case we *know* the object is
164 * reachable and we have already scanned it.
166 if (((o = lookup_object(remote)) != NULL) &&
167 (o->flags & COMPLETE)) {
172 packet_write(fd[1], "want %s%s%s%s%s%s\n",
174 (multi_ack ? " multi_ack" : ""),
175 (use_sideband == 2 ? " side-band-64k" : ""),
176 (use_sideband == 1 ? " side-band" : ""),
177 (use_thin_pack ? " thin-pack" : ""),
180 packet_write(fd[1], "want %s\n", sha1_to_hex(remote));
183 if (is_repository_shallow())
184 write_shallow_commits(fd[1], 1);
186 packet_write(fd[1], "deepen %d", depth);
193 unsigned char sha1[20];
196 while ((len = packet_read_line(fd[0], line, sizeof(line)))) {
197 if (!strncmp("shallow ", line, 8)) {
198 if (get_sha1_hex(line + 8, sha1))
199 die("invalid shallow line: %s", line);
200 register_shallow(sha1);
203 if (!strncmp("unshallow ", line, 10)) {
204 if (get_sha1_hex(line + 10, sha1))
205 die("invalid unshallow line: %s", line);
206 if (!lookup_object(sha1))
207 die("object not found: %s", line);
208 /* make sure that it is parsed as shallow */
210 if (unregister_shallow(sha1))
211 die("no shallow found: %s", line);
214 die("expected shallow/unshallow, got %s", line);
220 while ((sha1 = get_rev())) {
221 packet_write(fd[1], "have %s\n", sha1_to_hex(sha1));
223 fprintf(stderr, "have %s\n", sha1_to_hex(sha1));
225 if (!(31 & ++count)) {
232 * We keep one window "ahead" of the other side, and
233 * will wait for an ACK only on the next one
239 ack = get_ack(fd[0], result_sha1);
241 fprintf(stderr, "got ack %d %s\n", ack,
242 sha1_to_hex(result_sha1));
248 } else if (ack == 2) {
249 struct commit *commit =
250 lookup_commit(result_sha1);
251 mark_common(commit, 0, 1);
258 if (got_continue && MAX_IN_VAIN < in_vain) {
260 fprintf(stderr, "giving up\n");
266 packet_write(fd[1], "done\n");
268 fprintf(stderr, "done\n");
273 while (flushes || multi_ack) {
274 int ack = get_ack(fd[0], result_sha1);
277 fprintf(stderr, "got ack (%d) %s\n", ack,
278 sha1_to_hex(result_sha1));
289 static struct commit_list *complete;
291 static int mark_complete(const char *path, const unsigned char *sha1, int flag, void *cb_data)
293 struct object *o = parse_object(sha1);
295 while (o && o->type == OBJ_TAG) {
296 struct tag *t = (struct tag *) o;
298 break; /* broken repository */
299 o->flags |= COMPLETE;
300 o = parse_object(t->tagged->sha1);
302 if (o && o->type == OBJ_COMMIT) {
303 struct commit *commit = (struct commit *)o;
304 commit->object.flags |= COMPLETE;
305 insert_by_date(commit, &complete);
310 static void mark_recent_complete_commits(unsigned long cutoff)
312 while (complete && cutoff <= complete->item->date) {
314 fprintf(stderr, "Marking %s as complete\n",
315 sha1_to_hex(complete->item->object.sha1));
316 pop_most_recent_commit(&complete, COMPLETE);
320 static void filter_refs(struct ref **refs, int nr_match, char **match)
322 struct ref **return_refs;
323 struct ref *newlist = NULL;
324 struct ref **newtail = &newlist;
325 struct ref *ref, *next;
326 struct ref *fastarray[32];
328 if (nr_match && !fetch_all) {
329 if (ARRAY_SIZE(fastarray) < nr_match)
330 return_refs = xcalloc(nr_match, sizeof(struct ref *));
332 return_refs = fastarray;
333 memset(return_refs, 0, sizeof(struct ref *) * nr_match);
339 for (ref = *refs; ref; ref = next) {
341 if (!memcmp(ref->name, "refs/", 5) &&
342 check_ref_format(ref->name + 5))
344 else if (fetch_all &&
345 (!depth || strncmp(ref->name, "refs/tags/", 10) )) {
348 newtail = &ref->next;
352 int order = path_match(ref->name, nr_match, match);
354 return_refs[order-1] = ref;
355 continue; /* we will link it later */
363 for (i = 0; i < nr_match; i++) {
364 ref = return_refs[i];
368 newtail = &ref->next;
371 if (return_refs != fastarray)
377 static int everything_local(struct ref **refs, int nr_match, char **match)
381 unsigned long cutoff = 0;
383 track_object_refs = 0;
384 save_commit_buffer = 0;
386 for (ref = *refs; ref; ref = ref->next) {
389 o = parse_object(ref->old_sha1);
393 /* We already have it -- which may mean that we were
394 * in sync with the other side at some time after
395 * that (it is OK if we guess wrong here).
397 if (o->type == OBJ_COMMIT) {
398 struct commit *commit = (struct commit *)o;
399 if (!cutoff || cutoff < commit->date)
400 cutoff = commit->date;
405 for_each_ref(mark_complete, NULL);
407 mark_recent_complete_commits(cutoff);
411 * Mark all complete remote refs as common refs.
412 * Don't mark them common yet; the server has to be told so first.
414 for (ref = *refs; ref; ref = ref->next) {
415 struct object *o = deref_tag(lookup_object(ref->old_sha1),
418 if (!o || o->type != OBJ_COMMIT || !(o->flags & COMPLETE))
421 if (!(o->flags & SEEN)) {
422 rev_list_push((struct commit *)o, COMMON_REF | SEEN);
424 mark_common((struct commit *)o, 1, 1);
428 filter_refs(refs, nr_match, match);
430 for (retval = 1, ref = *refs; ref ; ref = ref->next) {
431 const unsigned char *remote = ref->old_sha1;
432 unsigned char local[20];
435 o = lookup_object(remote);
436 if (!o || !(o->flags & COMPLETE)) {
441 "want %s (%s)\n", sha1_to_hex(remote),
446 hashcpy(ref->new_sha1, local);
450 "already have %s (%s)\n", sha1_to_hex(remote),
456 static pid_t setup_sideband(int fd[2], int xd[2])
465 /* xd[] is talking with upload-pack; subprocess reads from
466 * xd[0], spits out band#2 to stderr, and feeds us band#1
470 die("fetch-pack: unable to set up pipe");
473 die("fetch-pack: unable to fork off sideband demultiplexer");
479 if (recv_sideband("fetch-pack", xd[0], fd[1], 2))
489 static int get_pack(int xd[2], const char **argv)
495 side_pid = setup_sideband(fd, xd);
498 die("fetch-pack: unable to fork off %s", argv[0]);
504 die("%s exec failed", argv[0]);
508 while (waitpid(pid, &status, 0) < 0) {
510 die("waiting for %s: %s", argv[0], strerror(errno));
512 if (WIFEXITED(status)) {
513 int code = WEXITSTATUS(status);
515 die("%s died with error code %d", argv[0], code);
518 if (WIFSIGNALED(status)) {
519 int sig = WTERMSIG(status);
520 die("%s died of signal %d", argv[0], sig);
522 die("%s died of unnatural causes %d", argv[0], status);
525 static int explode_rx_pack(int xd[2])
527 const char *argv[3] = { "unpack-objects", quiet ? "-q" : NULL, NULL };
528 return get_pack(xd, argv);
531 static int keep_rx_pack(int xd[2])
537 argv[n++] = "index-pack";
538 argv[n++] = "--stdin";
542 argv[n++] = "--fix-thin";
544 int s = sprintf(keep_arg, "--keep=fetch-pack %i on ", getpid());
545 if (gethostname(keep_arg + s, sizeof(keep_arg) - s))
546 strcpy(keep_arg + s, "localhost");
547 argv[n++] = keep_arg;
550 return get_pack(xd, argv);
553 static int fetch_pack(int fd[2], int nr_match, char **match)
556 unsigned char sha1[20];
559 get_remote_heads(fd[0], &ref, 0, NULL, 0);
560 if (is_repository_shallow() && !server_supports("shallow"))
561 die("Server does not support shallow clients");
562 if (server_supports("multi_ack")) {
564 fprintf(stderr, "Server supports multi_ack\n");
567 if (server_supports("side-band-64k")) {
569 fprintf(stderr, "Server supports side-band-64k\n");
572 else if (server_supports("side-band")) {
574 fprintf(stderr, "Server supports side-band\n");
579 die("no matching remote head");
581 if (everything_local(&ref, nr_match, match)) {
585 if (find_common(fd, sha1, ref) < 0)
587 /* When cloning, it is not unusual to have
590 fprintf(stderr, "warning: no common commits\n");
592 status = (keep_pack) ? keep_rx_pack(fd) : explode_rx_pack(fd);
594 die("git-fetch-pack: fetch failed.");
599 sha1_to_hex(ref->old_sha1), ref->name);
605 static int remove_duplicates(int nr_heads, char **heads)
609 for (src = dst = 0; src < nr_heads; src++) {
610 /* If heads[src] is different from any of
611 * heads[0..dst], push it in.
614 for (i = 0; i < dst; i++) {
615 if (!strcmp(heads[i], heads[src]))
621 heads[dst] = heads[src];
628 static struct lock_file lock;
630 int main(int argc, char **argv)
632 int i, ret, nr_heads;
633 char *dest = NULL, **heads;
638 setup_git_directory();
642 for (i = 1; i < argc; i++) {
646 if (!strncmp("--exec=", arg, 7)) {
650 if (!strcmp("--quiet", arg) || !strcmp("-q", arg)) {
654 if (!strcmp("--keep", arg) || !strcmp("-k", arg)) {
658 if (!strcmp("--thin", arg)) {
662 if (!strcmp("--all", arg)) {
666 if (!strcmp("-v", arg)) {
670 if (!strncmp("--depth=", arg, 8)) {
671 depth = strtol(arg + 8, NULL, 0);
672 if (stat(git_path("shallow"), &st))
676 usage(fetch_pack_usage);
679 heads = argv + i + 1;
680 nr_heads = argc - i - 1;
684 usage(fetch_pack_usage);
685 pid = git_connect(fd, dest, exec);
688 if (heads && nr_heads)
689 nr_heads = remove_duplicates(nr_heads, heads);
690 ret = fetch_pack(fd, nr_heads, heads);
693 ret |= finish_connect(pid);
695 if (!ret && nr_heads) {
696 /* If the heads to pull were given, we should have
697 * consumed all of them by matching the remote.
698 * Otherwise, 'git-fetch remote no-such-ref' would
699 * silently succeed without issuing an error.
701 for (i = 0; i < nr_heads; i++)
702 if (heads[i] && heads[i][0]) {
703 error("no such remote ref %s", heads[i]);
708 if (!ret && depth > 0) {
709 struct cache_time mtime;
710 char *shallow = git_path("shallow");
713 mtime.sec = st.st_mtime;
715 mtime.usec = st.st_mtim.usec;
717 if (stat(shallow, &st)) {
719 die("shallow file was removed during fetch");
720 } else if (st.st_mtime != mtime.sec
722 || st.st_mtim.usec != mtime.usec
725 die("shallow file was changed during fetch");
727 fd = hold_lock_file_for_update(&lock, shallow, 1);
728 if (!write_shallow_commits(fd, 0)) {
730 rollback_lock_file(&lock);
733 commit_lock_file(&lock);