t5537: fix incorrect expectation in test case 10
[alt-git.git] / fetch-pack.c
blob9bc29cfce78719fbd773e5fed8a678485f8e8e0a
1 #include "cache.h"
2 #include "refs.h"
3 #include "pkt-line.h"
4 #include "commit.h"
5 #include "tag.h"
6 #include "exec_cmd.h"
7 #include "pack.h"
8 #include "sideband.h"
9 #include "fetch-pack.h"
10 #include "remote.h"
11 #include "run-command.h"
12 #include "connect.h"
13 #include "transport.h"
14 #include "version.h"
15 #include "prio-queue.h"
16 #include "sha1-array.h"
18 static int transfer_unpack_limit = -1;
19 static int fetch_unpack_limit = -1;
20 static int unpack_limit = 100;
21 static int prefer_ofs_delta = 1;
22 static int no_done;
23 static int fetch_fsck_objects = -1;
24 static int transfer_fsck_objects = -1;
25 static int agent_supported;
26 static struct lock_file shallow_lock;
27 static const char *alternate_shallow_file;
29 #define COMPLETE (1U << 0)
30 #define COMMON (1U << 1)
31 #define COMMON_REF (1U << 2)
32 #define SEEN (1U << 3)
33 #define POPPED (1U << 4)
35 static int marked;
38 * After sending this many "have"s if we do not get any new ACK , we
39 * give up traversing our history.
41 #define MAX_IN_VAIN 256
43 static struct prio_queue rev_list = { compare_commits_by_commit_date };
44 static int non_common_revs, multi_ack, use_sideband, allow_tip_sha1_in_want;
46 static void rev_list_push(struct commit *commit, int mark)
48 if (!(commit->object.flags & mark)) {
49 commit->object.flags |= mark;
51 if (!(commit->object.parsed))
52 if (parse_commit(commit))
53 return;
55 prio_queue_put(&rev_list, commit);
57 if (!(commit->object.flags & COMMON))
58 non_common_revs++;
62 static int rev_list_insert_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
64 struct object *o = deref_tag(parse_object(sha1), refname, 0);
66 if (o && o->type == OBJ_COMMIT)
67 rev_list_push((struct commit *)o, SEEN);
69 return 0;
72 static int clear_marks(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
74 struct object *o = deref_tag(parse_object(sha1), refname, 0);
76 if (o && o->type == OBJ_COMMIT)
77 clear_commit_marks((struct commit *)o,
78 COMMON | COMMON_REF | SEEN | POPPED);
79 return 0;
83 This function marks a rev and its ancestors as common.
84 In some cases, it is desirable to mark only the ancestors (for example
85 when only the server does not yet know that they are common).
88 static void mark_common(struct commit *commit,
89 int ancestors_only, int dont_parse)
91 if (commit != NULL && !(commit->object.flags & COMMON)) {
92 struct object *o = (struct object *)commit;
94 if (!ancestors_only)
95 o->flags |= COMMON;
97 if (!(o->flags & SEEN))
98 rev_list_push(commit, SEEN);
99 else {
100 struct commit_list *parents;
102 if (!ancestors_only && !(o->flags & POPPED))
103 non_common_revs--;
104 if (!o->parsed && !dont_parse)
105 if (parse_commit(commit))
106 return;
108 for (parents = commit->parents;
109 parents;
110 parents = parents->next)
111 mark_common(parents->item, 0, dont_parse);
117 Get the next rev to send, ignoring the common.
120 static const unsigned char *get_rev(void)
122 struct commit *commit = NULL;
124 while (commit == NULL) {
125 unsigned int mark;
126 struct commit_list *parents;
128 if (rev_list.nr == 0 || non_common_revs == 0)
129 return NULL;
131 commit = prio_queue_get(&rev_list);
132 if (!commit->object.parsed)
133 parse_commit(commit);
134 parents = commit->parents;
136 commit->object.flags |= POPPED;
137 if (!(commit->object.flags & COMMON))
138 non_common_revs--;
140 if (commit->object.flags & COMMON) {
141 /* do not send "have", and ignore ancestors */
142 commit = NULL;
143 mark = COMMON | SEEN;
144 } else if (commit->object.flags & COMMON_REF)
145 /* send "have", and ignore ancestors */
146 mark = COMMON | SEEN;
147 else
148 /* send "have", also for its ancestors */
149 mark = SEEN;
151 while (parents) {
152 if (!(parents->item->object.flags & SEEN))
153 rev_list_push(parents->item, mark);
154 if (mark & COMMON)
155 mark_common(parents->item, 1, 0);
156 parents = parents->next;
160 return commit->object.sha1;
163 enum ack_type {
164 NAK = 0,
165 ACK,
166 ACK_continue,
167 ACK_common,
168 ACK_ready
171 static void consume_shallow_list(struct fetch_pack_args *args, int fd)
173 if (args->stateless_rpc && args->depth > 0) {
174 /* If we sent a depth we will get back "duplicate"
175 * shallow and unshallow commands every time there
176 * is a block of have lines exchanged.
178 char *line;
179 while ((line = packet_read_line(fd, NULL))) {
180 if (!prefixcmp(line, "shallow "))
181 continue;
182 if (!prefixcmp(line, "unshallow "))
183 continue;
184 die("git fetch-pack: expected shallow list");
189 static enum ack_type get_ack(int fd, unsigned char *result_sha1)
191 int len;
192 char *line = packet_read_line(fd, &len);
194 if (!len)
195 die("git fetch-pack: expected ACK/NAK, got EOF");
196 if (!strcmp(line, "NAK"))
197 return NAK;
198 if (!prefixcmp(line, "ACK ")) {
199 if (!get_sha1_hex(line+4, result_sha1)) {
200 if (len < 45)
201 return ACK;
202 if (strstr(line+45, "continue"))
203 return ACK_continue;
204 if (strstr(line+45, "common"))
205 return ACK_common;
206 if (strstr(line+45, "ready"))
207 return ACK_ready;
208 return ACK;
211 die("git fetch_pack: expected ACK/NAK, got '%s'", line);
214 static void send_request(struct fetch_pack_args *args,
215 int fd, struct strbuf *buf)
217 if (args->stateless_rpc) {
218 send_sideband(fd, -1, buf->buf, buf->len, LARGE_PACKET_MAX);
219 packet_flush(fd);
220 } else
221 write_or_die(fd, buf->buf, buf->len);
224 static void insert_one_alternate_ref(const struct ref *ref, void *unused)
226 rev_list_insert_ref(NULL, ref->old_sha1, 0, NULL);
229 #define INITIAL_FLUSH 16
230 #define PIPESAFE_FLUSH 32
231 #define LARGE_FLUSH 1024
233 static int next_flush(struct fetch_pack_args *args, int count)
235 int flush_limit = args->stateless_rpc ? LARGE_FLUSH : PIPESAFE_FLUSH;
237 if (count < flush_limit)
238 count <<= 1;
239 else
240 count += flush_limit;
241 return count;
244 static int find_common(struct fetch_pack_args *args,
245 int fd[2], unsigned char *result_sha1,
246 struct ref *refs)
248 int fetching;
249 int count = 0, flushes = 0, flush_at = INITIAL_FLUSH, retval;
250 const unsigned char *sha1;
251 unsigned in_vain = 0;
252 int got_continue = 0;
253 int got_ready = 0;
254 struct strbuf req_buf = STRBUF_INIT;
255 size_t state_len = 0;
257 if (args->stateless_rpc && multi_ack == 1)
258 die("--stateless-rpc requires multi_ack_detailed");
259 if (marked)
260 for_each_ref(clear_marks, NULL);
261 marked = 1;
263 for_each_ref(rev_list_insert_ref, NULL);
264 for_each_alternate_ref(insert_one_alternate_ref, NULL);
266 fetching = 0;
267 for ( ; refs ; refs = refs->next) {
268 unsigned char *remote = refs->old_sha1;
269 const char *remote_hex;
270 struct object *o;
273 * If that object is complete (i.e. it is an ancestor of a
274 * local ref), we tell them we have it but do not have to
275 * tell them about its ancestors, which they already know
276 * about.
278 * We use lookup_object here because we are only
279 * interested in the case we *know* the object is
280 * reachable and we have already scanned it.
282 if (((o = lookup_object(remote)) != NULL) &&
283 (o->flags & COMPLETE)) {
284 continue;
287 remote_hex = sha1_to_hex(remote);
288 if (!fetching) {
289 struct strbuf c = STRBUF_INIT;
290 if (multi_ack == 2) strbuf_addstr(&c, " multi_ack_detailed");
291 if (multi_ack == 1) strbuf_addstr(&c, " multi_ack");
292 if (no_done) strbuf_addstr(&c, " no-done");
293 if (use_sideband == 2) strbuf_addstr(&c, " side-band-64k");
294 if (use_sideband == 1) strbuf_addstr(&c, " side-band");
295 if (args->use_thin_pack) strbuf_addstr(&c, " thin-pack");
296 if (args->no_progress) strbuf_addstr(&c, " no-progress");
297 if (args->include_tag) strbuf_addstr(&c, " include-tag");
298 if (prefer_ofs_delta) strbuf_addstr(&c, " ofs-delta");
299 if (agent_supported) strbuf_addf(&c, " agent=%s",
300 git_user_agent_sanitized());
301 packet_buf_write(&req_buf, "want %s%s\n", remote_hex, c.buf);
302 strbuf_release(&c);
303 } else
304 packet_buf_write(&req_buf, "want %s\n", remote_hex);
305 fetching++;
308 if (!fetching) {
309 strbuf_release(&req_buf);
310 packet_flush(fd[1]);
311 return 1;
314 if (is_repository_shallow())
315 write_shallow_commits(&req_buf, 1, NULL);
316 if (args->depth > 0)
317 packet_buf_write(&req_buf, "deepen %d", args->depth);
318 packet_buf_flush(&req_buf);
319 state_len = req_buf.len;
321 if (args->depth > 0) {
322 char *line;
323 unsigned char sha1[20];
325 send_request(args, fd[1], &req_buf);
326 while ((line = packet_read_line(fd[0], NULL))) {
327 if (!prefixcmp(line, "shallow ")) {
328 if (get_sha1_hex(line + 8, sha1))
329 die("invalid shallow line: %s", line);
330 register_shallow(sha1);
331 continue;
333 if (!prefixcmp(line, "unshallow ")) {
334 if (get_sha1_hex(line + 10, sha1))
335 die("invalid unshallow line: %s", line);
336 if (!lookup_object(sha1))
337 die("object not found: %s", line);
338 /* make sure that it is parsed as shallow */
339 if (!parse_object(sha1))
340 die("error in object: %s", line);
341 if (unregister_shallow(sha1))
342 die("no shallow found: %s", line);
343 continue;
345 die("expected shallow/unshallow, got %s", line);
347 } else if (!args->stateless_rpc)
348 send_request(args, fd[1], &req_buf);
350 if (!args->stateless_rpc) {
351 /* If we aren't using the stateless-rpc interface
352 * we don't need to retain the headers.
354 strbuf_setlen(&req_buf, 0);
355 state_len = 0;
358 flushes = 0;
359 retval = -1;
360 while ((sha1 = get_rev())) {
361 packet_buf_write(&req_buf, "have %s\n", sha1_to_hex(sha1));
362 if (args->verbose)
363 fprintf(stderr, "have %s\n", sha1_to_hex(sha1));
364 in_vain++;
365 if (flush_at <= ++count) {
366 int ack;
368 packet_buf_flush(&req_buf);
369 send_request(args, fd[1], &req_buf);
370 strbuf_setlen(&req_buf, state_len);
371 flushes++;
372 flush_at = next_flush(args, count);
375 * We keep one window "ahead" of the other side, and
376 * will wait for an ACK only on the next one
378 if (!args->stateless_rpc && count == INITIAL_FLUSH)
379 continue;
381 consume_shallow_list(args, fd[0]);
382 do {
383 ack = get_ack(fd[0], result_sha1);
384 if (args->verbose && ack)
385 fprintf(stderr, "got ack %d %s\n", ack,
386 sha1_to_hex(result_sha1));
387 switch (ack) {
388 case ACK:
389 flushes = 0;
390 multi_ack = 0;
391 retval = 0;
392 goto done;
393 case ACK_common:
394 case ACK_ready:
395 case ACK_continue: {
396 struct commit *commit =
397 lookup_commit(result_sha1);
398 if (!commit)
399 die("invalid commit %s", sha1_to_hex(result_sha1));
400 if (args->stateless_rpc
401 && ack == ACK_common
402 && !(commit->object.flags & COMMON)) {
403 /* We need to replay the have for this object
404 * on the next RPC request so the peer knows
405 * it is in common with us.
407 const char *hex = sha1_to_hex(result_sha1);
408 packet_buf_write(&req_buf, "have %s\n", hex);
409 state_len = req_buf.len;
411 mark_common(commit, 0, 1);
412 retval = 0;
413 in_vain = 0;
414 got_continue = 1;
415 if (ack == ACK_ready) {
416 clear_prio_queue(&rev_list);
417 got_ready = 1;
419 break;
422 } while (ack);
423 flushes--;
424 if (got_continue && MAX_IN_VAIN < in_vain) {
425 if (args->verbose)
426 fprintf(stderr, "giving up\n");
427 break; /* give up */
431 done:
432 if (!got_ready || !no_done) {
433 packet_buf_write(&req_buf, "done\n");
434 send_request(args, fd[1], &req_buf);
436 if (args->verbose)
437 fprintf(stderr, "done\n");
438 if (retval != 0) {
439 multi_ack = 0;
440 flushes++;
442 strbuf_release(&req_buf);
444 consume_shallow_list(args, fd[0]);
445 while (flushes || multi_ack) {
446 int ack = get_ack(fd[0], result_sha1);
447 if (ack) {
448 if (args->verbose)
449 fprintf(stderr, "got ack (%d) %s\n", ack,
450 sha1_to_hex(result_sha1));
451 if (ack == ACK)
452 return 0;
453 multi_ack = 1;
454 continue;
456 flushes--;
458 /* it is no error to fetch into a completely empty repo */
459 return count ? retval : 0;
462 static struct commit_list *complete;
464 static int mark_complete(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
466 struct object *o = parse_object(sha1);
468 while (o && o->type == OBJ_TAG) {
469 struct tag *t = (struct tag *) o;
470 if (!t->tagged)
471 break; /* broken repository */
472 o->flags |= COMPLETE;
473 o = parse_object(t->tagged->sha1);
475 if (o && o->type == OBJ_COMMIT) {
476 struct commit *commit = (struct commit *)o;
477 if (!(commit->object.flags & COMPLETE)) {
478 commit->object.flags |= COMPLETE;
479 commit_list_insert(commit, &complete);
482 return 0;
485 static void mark_recent_complete_commits(struct fetch_pack_args *args,
486 unsigned long cutoff)
488 while (complete && cutoff <= complete->item->date) {
489 if (args->verbose)
490 fprintf(stderr, "Marking %s as complete\n",
491 sha1_to_hex(complete->item->object.sha1));
492 pop_most_recent_commit(&complete, COMPLETE);
496 static void filter_refs(struct fetch_pack_args *args,
497 struct ref **refs,
498 struct ref **sought, int nr_sought)
500 struct ref *newlist = NULL;
501 struct ref **newtail = &newlist;
502 struct ref *ref, *next;
503 int i;
505 i = 0;
506 for (ref = *refs; ref; ref = next) {
507 int keep = 0;
508 next = ref->next;
510 if (!memcmp(ref->name, "refs/", 5) &&
511 check_refname_format(ref->name + 5, 0))
512 ; /* trash */
513 else {
514 while (i < nr_sought) {
515 int cmp = strcmp(ref->name, sought[i]->name);
516 if (cmp < 0)
517 break; /* definitely do not have it */
518 else if (cmp == 0) {
519 keep = 1; /* definitely have it */
520 sought[i]->matched = 1;
522 i++;
526 if (!keep && args->fetch_all &&
527 (!args->depth || prefixcmp(ref->name, "refs/tags/")))
528 keep = 1;
530 if (keep) {
531 *newtail = ref;
532 ref->next = NULL;
533 newtail = &ref->next;
534 } else {
535 free(ref);
539 /* Append unmatched requests to the list */
540 if (allow_tip_sha1_in_want) {
541 for (i = 0; i < nr_sought; i++) {
542 ref = sought[i];
543 if (ref->matched)
544 continue;
545 if (get_sha1_hex(ref->name, ref->old_sha1))
546 continue;
548 ref->matched = 1;
549 *newtail = ref;
550 ref->next = NULL;
551 newtail = &ref->next;
554 *refs = newlist;
557 static void mark_alternate_complete(const struct ref *ref, void *unused)
559 mark_complete(NULL, ref->old_sha1, 0, NULL);
562 static int everything_local(struct fetch_pack_args *args,
563 struct ref **refs,
564 struct ref **sought, int nr_sought)
566 struct ref *ref;
567 int retval;
568 unsigned long cutoff = 0;
570 save_commit_buffer = 0;
572 for (ref = *refs; ref; ref = ref->next) {
573 struct object *o;
575 if (!has_sha1_file(ref->old_sha1))
576 continue;
578 o = parse_object(ref->old_sha1);
579 if (!o)
580 continue;
582 /* We already have it -- which may mean that we were
583 * in sync with the other side at some time after
584 * that (it is OK if we guess wrong here).
586 if (o->type == OBJ_COMMIT) {
587 struct commit *commit = (struct commit *)o;
588 if (!cutoff || cutoff < commit->date)
589 cutoff = commit->date;
593 if (!args->depth) {
594 for_each_ref(mark_complete, NULL);
595 for_each_alternate_ref(mark_alternate_complete, NULL);
596 commit_list_sort_by_date(&complete);
597 if (cutoff)
598 mark_recent_complete_commits(args, cutoff);
602 * Mark all complete remote refs as common refs.
603 * Don't mark them common yet; the server has to be told so first.
605 for (ref = *refs; ref; ref = ref->next) {
606 struct object *o = deref_tag(lookup_object(ref->old_sha1),
607 NULL, 0);
609 if (!o || o->type != OBJ_COMMIT || !(o->flags & COMPLETE))
610 continue;
612 if (!(o->flags & SEEN)) {
613 rev_list_push((struct commit *)o, COMMON_REF | SEEN);
615 mark_common((struct commit *)o, 1, 1);
619 filter_refs(args, refs, sought, nr_sought);
621 for (retval = 1, ref = *refs; ref ; ref = ref->next) {
622 const unsigned char *remote = ref->old_sha1;
623 unsigned char local[20];
624 struct object *o;
626 o = lookup_object(remote);
627 if (!o || !(o->flags & COMPLETE)) {
628 retval = 0;
629 if (!args->verbose)
630 continue;
631 fprintf(stderr,
632 "want %s (%s)\n", sha1_to_hex(remote),
633 ref->name);
634 continue;
637 hashcpy(ref->new_sha1, local);
638 if (!args->verbose)
639 continue;
640 fprintf(stderr,
641 "already have %s (%s)\n", sha1_to_hex(remote),
642 ref->name);
644 return retval;
647 static int sideband_demux(int in, int out, void *data)
649 int *xd = data;
651 int ret = recv_sideband("fetch-pack", xd[0], out);
652 close(out);
653 return ret;
656 static int get_pack(struct fetch_pack_args *args,
657 int xd[2], char **pack_lockfile)
659 struct async demux;
660 const char *argv[22];
661 char keep_arg[256];
662 char hdr_arg[256];
663 const char **av, *cmd_name;
664 int do_keep = args->keep_pack;
665 struct child_process cmd;
666 int ret;
668 memset(&demux, 0, sizeof(demux));
669 if (use_sideband) {
670 /* xd[] is talking with upload-pack; subprocess reads from
671 * xd[0], spits out band#2 to stderr, and feeds us band#1
672 * through demux->out.
674 demux.proc = sideband_demux;
675 demux.data = xd;
676 demux.out = -1;
677 if (start_async(&demux))
678 die("fetch-pack: unable to fork off sideband"
679 " demultiplexer");
681 else
682 demux.out = xd[0];
684 memset(&cmd, 0, sizeof(cmd));
685 cmd.argv = argv;
686 av = argv;
687 *hdr_arg = 0;
688 if (!args->keep_pack && unpack_limit) {
689 struct pack_header header;
691 if (read_pack_header(demux.out, &header))
692 die("protocol error: bad pack header");
693 snprintf(hdr_arg, sizeof(hdr_arg),
694 "--pack_header=%"PRIu32",%"PRIu32,
695 ntohl(header.hdr_version), ntohl(header.hdr_entries));
696 if (ntohl(header.hdr_entries) < unpack_limit)
697 do_keep = 0;
698 else
699 do_keep = 1;
702 if (alternate_shallow_file) {
703 *av++ = "--shallow-file";
704 *av++ = alternate_shallow_file;
707 if (do_keep) {
708 if (pack_lockfile)
709 cmd.out = -1;
710 *av++ = cmd_name = "index-pack";
711 *av++ = "--stdin";
712 if (!args->quiet && !args->no_progress)
713 *av++ = "-v";
714 if (args->use_thin_pack)
715 *av++ = "--fix-thin";
716 if (args->lock_pack || unpack_limit) {
717 int s = sprintf(keep_arg,
718 "--keep=fetch-pack %"PRIuMAX " on ", (uintmax_t) getpid());
719 if (gethostname(keep_arg + s, sizeof(keep_arg) - s))
720 strcpy(keep_arg + s, "localhost");
721 *av++ = keep_arg;
723 if (args->check_self_contained_and_connected)
724 *av++ = "--check-self-contained-and-connected";
726 else {
727 *av++ = cmd_name = "unpack-objects";
728 if (args->quiet || args->no_progress)
729 *av++ = "-q";
730 args->check_self_contained_and_connected = 0;
732 if (*hdr_arg)
733 *av++ = hdr_arg;
734 if (fetch_fsck_objects >= 0
735 ? fetch_fsck_objects
736 : transfer_fsck_objects >= 0
737 ? transfer_fsck_objects
738 : 0)
739 *av++ = "--strict";
740 *av++ = NULL;
742 cmd.in = demux.out;
743 cmd.git_cmd = 1;
744 if (start_command(&cmd))
745 die("fetch-pack: unable to fork off %s", cmd_name);
746 if (do_keep && pack_lockfile) {
747 *pack_lockfile = index_pack_lockfile(cmd.out);
748 close(cmd.out);
751 if (!use_sideband)
752 /* Closed by start_command() */
753 xd[0] = -1;
755 ret = finish_command(&cmd);
756 if (!ret || (args->check_self_contained_and_connected && ret == 1))
757 args->self_contained_and_connected =
758 args->check_self_contained_and_connected &&
759 ret == 0;
760 else
761 die("%s failed", cmd_name);
762 if (use_sideband && finish_async(&demux))
763 die("error in sideband demultiplexer");
764 return 0;
767 static int cmp_ref_by_name(const void *a_, const void *b_)
769 const struct ref *a = *((const struct ref **)a_);
770 const struct ref *b = *((const struct ref **)b_);
771 return strcmp(a->name, b->name);
774 static struct ref *do_fetch_pack(struct fetch_pack_args *args,
775 int fd[2],
776 const struct ref *orig_ref,
777 struct ref **sought, int nr_sought,
778 struct shallow_info *si,
779 char **pack_lockfile)
781 struct ref *ref = copy_ref_list(orig_ref);
782 unsigned char sha1[20];
783 const char *agent_feature;
784 int agent_len;
786 sort_ref_list(&ref, ref_compare_name);
787 qsort(sought, nr_sought, sizeof(*sought), cmp_ref_by_name);
789 if (is_repository_shallow() && !server_supports("shallow"))
790 die("Server does not support shallow clients");
791 if (server_supports("multi_ack_detailed")) {
792 if (args->verbose)
793 fprintf(stderr, "Server supports multi_ack_detailed\n");
794 multi_ack = 2;
795 if (server_supports("no-done")) {
796 if (args->verbose)
797 fprintf(stderr, "Server supports no-done\n");
798 if (args->stateless_rpc)
799 no_done = 1;
802 else if (server_supports("multi_ack")) {
803 if (args->verbose)
804 fprintf(stderr, "Server supports multi_ack\n");
805 multi_ack = 1;
807 if (server_supports("side-band-64k")) {
808 if (args->verbose)
809 fprintf(stderr, "Server supports side-band-64k\n");
810 use_sideband = 2;
812 else if (server_supports("side-band")) {
813 if (args->verbose)
814 fprintf(stderr, "Server supports side-band\n");
815 use_sideband = 1;
817 if (server_supports("allow-tip-sha1-in-want")) {
818 if (args->verbose)
819 fprintf(stderr, "Server supports allow-tip-sha1-in-want\n");
820 allow_tip_sha1_in_want = 1;
822 if (!server_supports("thin-pack"))
823 args->use_thin_pack = 0;
824 if (!server_supports("no-progress"))
825 args->no_progress = 0;
826 if (!server_supports("include-tag"))
827 args->include_tag = 0;
828 if (server_supports("ofs-delta")) {
829 if (args->verbose)
830 fprintf(stderr, "Server supports ofs-delta\n");
831 } else
832 prefer_ofs_delta = 0;
834 if ((agent_feature = server_feature_value("agent", &agent_len))) {
835 agent_supported = 1;
836 if (args->verbose && agent_len)
837 fprintf(stderr, "Server version is %.*s\n",
838 agent_len, agent_feature);
841 if (everything_local(args, &ref, sought, nr_sought)) {
842 packet_flush(fd[1]);
843 goto all_done;
845 if (find_common(args, fd, sha1, ref) < 0)
846 if (!args->keep_pack)
847 /* When cloning, it is not unusual to have
848 * no common commit.
850 warning("no common commits");
852 if (args->stateless_rpc)
853 packet_flush(fd[1]);
854 if (args->depth > 0)
855 setup_alternate_shallow(&shallow_lock, &alternate_shallow_file,
856 NULL);
857 else if (si->nr_ours || si->nr_theirs)
858 alternate_shallow_file = setup_temporary_shallow(si->shallow);
859 else
860 alternate_shallow_file = NULL;
861 if (get_pack(args, fd, pack_lockfile))
862 die("git fetch-pack: fetch failed.");
864 all_done:
865 return ref;
868 static int fetch_pack_config(const char *var, const char *value, void *cb)
870 if (strcmp(var, "fetch.unpacklimit") == 0) {
871 fetch_unpack_limit = git_config_int(var, value);
872 return 0;
875 if (strcmp(var, "transfer.unpacklimit") == 0) {
876 transfer_unpack_limit = git_config_int(var, value);
877 return 0;
880 if (strcmp(var, "repack.usedeltabaseoffset") == 0) {
881 prefer_ofs_delta = git_config_bool(var, value);
882 return 0;
885 if (!strcmp(var, "fetch.fsckobjects")) {
886 fetch_fsck_objects = git_config_bool(var, value);
887 return 0;
890 if (!strcmp(var, "transfer.fsckobjects")) {
891 transfer_fsck_objects = git_config_bool(var, value);
892 return 0;
895 return git_default_config(var, value, cb);
898 static void fetch_pack_setup(void)
900 static int did_setup;
901 if (did_setup)
902 return;
903 git_config(fetch_pack_config, NULL);
904 if (0 <= transfer_unpack_limit)
905 unpack_limit = transfer_unpack_limit;
906 else if (0 <= fetch_unpack_limit)
907 unpack_limit = fetch_unpack_limit;
908 did_setup = 1;
911 static int remove_duplicates_in_refs(struct ref **ref, int nr)
913 struct string_list names = STRING_LIST_INIT_NODUP;
914 int src, dst;
916 for (src = dst = 0; src < nr; src++) {
917 struct string_list_item *item;
918 item = string_list_insert(&names, ref[src]->name);
919 if (item->util)
920 continue; /* already have it */
921 item->util = ref[src];
922 if (src != dst)
923 ref[dst] = ref[src];
924 dst++;
926 for (src = dst; src < nr; src++)
927 ref[src] = NULL;
928 string_list_clear(&names, 0);
929 return dst;
932 static void update_shallow(struct fetch_pack_args *args,
933 struct ref **sought, int nr_sought,
934 struct shallow_info *si)
936 struct sha1_array ref = SHA1_ARRAY_INIT;
937 int *status;
938 int i;
940 if (args->depth > 0 && alternate_shallow_file) {
941 if (*alternate_shallow_file == '\0') { /* --unshallow */
942 unlink_or_warn(git_path("shallow"));
943 rollback_lock_file(&shallow_lock);
944 } else
945 commit_lock_file(&shallow_lock);
946 return;
949 if (!si->shallow || !si->shallow->nr)
950 return;
952 if (alternate_shallow_file) {
954 * The temporary shallow file is only useful for
955 * index-pack and unpack-objects because it may
956 * contain more roots than we want. Delete it.
958 if (*alternate_shallow_file)
959 unlink(alternate_shallow_file);
960 free((char *)alternate_shallow_file);
963 if (args->cloning) {
965 * remote is shallow, but this is a clone, there are
966 * no objects in repo to worry about. Accept any
967 * shallow points that exist in the pack (iow in repo
968 * after get_pack() and reprepare_packed_git())
970 struct sha1_array extra = SHA1_ARRAY_INIT;
971 unsigned char (*sha1)[20] = si->shallow->sha1;
972 for (i = 0; i < si->shallow->nr; i++)
973 if (has_sha1_file(sha1[i]))
974 sha1_array_append(&extra, sha1[i]);
975 if (extra.nr) {
976 setup_alternate_shallow(&shallow_lock,
977 &alternate_shallow_file,
978 &extra);
979 commit_lock_file(&shallow_lock);
981 sha1_array_clear(&extra);
982 return;
985 if (!si->nr_ours && !si->nr_theirs)
986 return;
988 remove_nonexistent_theirs_shallow(si);
989 if (!si->nr_ours && !si->nr_theirs)
990 return;
991 for (i = 0; i < nr_sought; i++)
992 sha1_array_append(&ref, sought[i]->old_sha1);
993 si->ref = &ref;
995 if (args->update_shallow) {
997 * remote is also shallow, .git/shallow may be updated
998 * so all refs can be accepted. Make sure we only add
999 * shallow roots that are actually reachable from new
1000 * refs.
1002 struct sha1_array extra = SHA1_ARRAY_INIT;
1003 unsigned char (*sha1)[20] = si->shallow->sha1;
1004 assign_shallow_commits_to_refs(si, NULL, NULL);
1005 if (!si->nr_ours && !si->nr_theirs) {
1006 sha1_array_clear(&ref);
1007 return;
1009 for (i = 0; i < si->nr_ours; i++)
1010 sha1_array_append(&extra, sha1[si->ours[i]]);
1011 for (i = 0; i < si->nr_theirs; i++)
1012 sha1_array_append(&extra, sha1[si->theirs[i]]);
1013 setup_alternate_shallow(&shallow_lock,
1014 &alternate_shallow_file,
1015 &extra);
1016 commit_lock_file(&shallow_lock);
1017 sha1_array_clear(&extra);
1018 sha1_array_clear(&ref);
1019 return;
1023 * remote is also shallow, check what ref is safe to update
1024 * without updating .git/shallow
1026 status = xcalloc(nr_sought, sizeof(*status));
1027 assign_shallow_commits_to_refs(si, NULL, status);
1028 if (si->nr_ours || si->nr_theirs) {
1029 for (i = 0; i < nr_sought; i++)
1030 if (status[i])
1031 sought[i]->status = REF_STATUS_REJECT_SHALLOW;
1033 free(status);
1034 sha1_array_clear(&ref);
1037 struct ref *fetch_pack(struct fetch_pack_args *args,
1038 int fd[], struct child_process *conn,
1039 const struct ref *ref,
1040 const char *dest,
1041 struct ref **sought, int nr_sought,
1042 struct sha1_array *shallow,
1043 char **pack_lockfile)
1045 struct ref *ref_cpy;
1046 struct shallow_info si;
1048 fetch_pack_setup();
1049 if (nr_sought)
1050 nr_sought = remove_duplicates_in_refs(sought, nr_sought);
1052 if (!ref) {
1053 packet_flush(fd[1]);
1054 die("no matching remote head");
1056 prepare_shallow_info(&si, shallow);
1057 ref_cpy = do_fetch_pack(args, fd, ref, sought, nr_sought,
1058 &si, pack_lockfile);
1059 reprepare_packed_git();
1060 update_shallow(args, sought, nr_sought, &si);
1061 clear_shallow_info(&si);
1062 return ref_cpy;