Merge branch 'ds/clarify-hashwrite'
[alt-git.git] / fetch-pack.c
blobfb04a76ca263042b145b40adf1577a81c6d7e12d
1 #include "cache.h"
2 #include "repository.h"
3 #include "config.h"
4 #include "lockfile.h"
5 #include "refs.h"
6 #include "pkt-line.h"
7 #include "commit.h"
8 #include "tag.h"
9 #include "exec-cmd.h"
10 #include "pack.h"
11 #include "sideband.h"
12 #include "fetch-pack.h"
13 #include "remote.h"
14 #include "run-command.h"
15 #include "connect.h"
16 #include "transport.h"
17 #include "version.h"
18 #include "oid-array.h"
19 #include "oidset.h"
20 #include "packfile.h"
21 #include "object-store.h"
22 #include "connected.h"
23 #include "fetch-negotiator.h"
24 #include "fsck.h"
25 #include "shallow.h"
27 static int transfer_unpack_limit = -1;
28 static int fetch_unpack_limit = -1;
29 static int unpack_limit = 100;
30 static int prefer_ofs_delta = 1;
31 static int no_done;
32 static int deepen_since_ok;
33 static int deepen_not_ok;
34 static int fetch_fsck_objects = -1;
35 static int transfer_fsck_objects = -1;
36 static int agent_supported;
37 static int server_supports_filtering;
38 static int advertise_sid;
39 static struct shallow_lock shallow_lock;
40 static const char *alternate_shallow_file;
41 static struct strbuf fsck_msg_types = STRBUF_INIT;
42 static struct string_list uri_protocols = STRING_LIST_INIT_DUP;
44 /* Remember to update object flag allocation in object.h */
45 #define COMPLETE (1U << 0)
46 #define ALTERNATE (1U << 1)
49 * After sending this many "have"s if we do not get any new ACK , we
50 * give up traversing our history.
52 #define MAX_IN_VAIN 256
54 static int multi_ack, use_sideband;
55 /* Allow specifying sha1 if it is a ref tip. */
56 #define ALLOW_TIP_SHA1 01
57 /* Allow request of a sha1 if it is reachable from a ref (possibly hidden ref). */
58 #define ALLOW_REACHABLE_SHA1 02
59 static unsigned int allow_unadvertised_object_request;
61 __attribute__((format (printf, 2, 3)))
62 static inline void print_verbose(const struct fetch_pack_args *args,
63 const char *fmt, ...)
65 va_list params;
67 if (!args->verbose)
68 return;
70 va_start(params, fmt);
71 vfprintf(stderr, fmt, params);
72 va_end(params);
73 fputc('\n', stderr);
76 struct alternate_object_cache {
77 struct object **items;
78 size_t nr, alloc;
81 static void cache_one_alternate(const struct object_id *oid,
82 void *vcache)
84 struct alternate_object_cache *cache = vcache;
85 struct object *obj = parse_object(the_repository, oid);
87 if (!obj || (obj->flags & ALTERNATE))
88 return;
90 obj->flags |= ALTERNATE;
91 ALLOC_GROW(cache->items, cache->nr + 1, cache->alloc);
92 cache->items[cache->nr++] = obj;
95 static void for_each_cached_alternate(struct fetch_negotiator *negotiator,
96 void (*cb)(struct fetch_negotiator *,
97 struct object *))
99 static int initialized;
100 static struct alternate_object_cache cache;
101 size_t i;
103 if (!initialized) {
104 for_each_alternate_ref(cache_one_alternate, &cache);
105 initialized = 1;
108 for (i = 0; i < cache.nr; i++)
109 cb(negotiator, cache.items[i]);
112 static struct commit *deref_without_lazy_fetch(const struct object_id *oid,
113 int mark_tags_complete)
115 enum object_type type;
116 struct object_info info = { .typep = &type };
118 while (1) {
119 if (oid_object_info_extended(the_repository, oid, &info,
120 OBJECT_INFO_SKIP_FETCH_OBJECT | OBJECT_INFO_QUICK))
121 return NULL;
122 if (type == OBJ_TAG) {
123 struct tag *tag = (struct tag *)
124 parse_object(the_repository, oid);
126 if (!tag->tagged)
127 return NULL;
128 if (mark_tags_complete)
129 tag->object.flags |= COMPLETE;
130 oid = &tag->tagged->oid;
131 } else {
132 break;
135 if (type == OBJ_COMMIT)
136 return (struct commit *) parse_object(the_repository, oid);
137 return NULL;
140 static int rev_list_insert_ref(struct fetch_negotiator *negotiator,
141 const struct object_id *oid)
143 struct commit *c = deref_without_lazy_fetch(oid, 0);
145 if (c)
146 negotiator->add_tip(negotiator, c);
147 return 0;
150 static int rev_list_insert_ref_oid(const char *refname, const struct object_id *oid,
151 int flag, void *cb_data)
153 return rev_list_insert_ref(cb_data, oid);
156 enum ack_type {
157 NAK = 0,
158 ACK,
159 ACK_continue,
160 ACK_common,
161 ACK_ready
164 static void consume_shallow_list(struct fetch_pack_args *args,
165 struct packet_reader *reader)
167 if (args->stateless_rpc && args->deepen) {
168 /* If we sent a depth we will get back "duplicate"
169 * shallow and unshallow commands every time there
170 * is a block of have lines exchanged.
172 while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
173 if (starts_with(reader->line, "shallow "))
174 continue;
175 if (starts_with(reader->line, "unshallow "))
176 continue;
177 die(_("git fetch-pack: expected shallow list"));
179 if (reader->status != PACKET_READ_FLUSH)
180 die(_("git fetch-pack: expected a flush packet after shallow list"));
184 static enum ack_type get_ack(struct packet_reader *reader,
185 struct object_id *result_oid)
187 int len;
188 const char *arg;
190 if (packet_reader_read(reader) != PACKET_READ_NORMAL)
191 die(_("git fetch-pack: expected ACK/NAK, got a flush packet"));
192 len = reader->pktlen;
194 if (!strcmp(reader->line, "NAK"))
195 return NAK;
196 if (skip_prefix(reader->line, "ACK ", &arg)) {
197 const char *p;
198 if (!parse_oid_hex(arg, result_oid, &p)) {
199 len -= p - reader->line;
200 if (len < 1)
201 return ACK;
202 if (strstr(p, "continue"))
203 return ACK_continue;
204 if (strstr(p, "common"))
205 return ACK_common;
206 if (strstr(p, "ready"))
207 return ACK_ready;
208 return ACK;
211 die(_("git fetch-pack: expected ACK/NAK, got '%s'"), reader->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 if (write_in_full(fd, buf->buf, buf->len) < 0)
222 die_errno(_("unable to write to remote"));
226 static void insert_one_alternate_object(struct fetch_negotiator *negotiator,
227 struct object *obj)
229 rev_list_insert_ref(negotiator, &obj->oid);
232 #define INITIAL_FLUSH 16
233 #define PIPESAFE_FLUSH 32
234 #define LARGE_FLUSH 16384
236 static int next_flush(int stateless_rpc, int count)
238 if (stateless_rpc) {
239 if (count < LARGE_FLUSH)
240 count <<= 1;
241 else
242 count = count * 11 / 10;
243 } else {
244 if (count < PIPESAFE_FLUSH)
245 count <<= 1;
246 else
247 count += PIPESAFE_FLUSH;
249 return count;
252 static void mark_tips(struct fetch_negotiator *negotiator,
253 const struct oid_array *negotiation_tips)
255 int i;
257 if (!negotiation_tips) {
258 for_each_rawref(rev_list_insert_ref_oid, negotiator);
259 return;
262 for (i = 0; i < negotiation_tips->nr; i++)
263 rev_list_insert_ref(negotiator, &negotiation_tips->oid[i]);
264 return;
267 static int find_common(struct fetch_negotiator *negotiator,
268 struct fetch_pack_args *args,
269 int fd[2], struct object_id *result_oid,
270 struct ref *refs)
272 int fetching;
273 int count = 0, flushes = 0, flush_at = INITIAL_FLUSH, retval;
274 const struct object_id *oid;
275 unsigned in_vain = 0;
276 int got_continue = 0;
277 int got_ready = 0;
278 struct strbuf req_buf = STRBUF_INIT;
279 size_t state_len = 0;
280 struct packet_reader reader;
282 if (args->stateless_rpc && multi_ack == 1)
283 die(_("--stateless-rpc requires multi_ack_detailed"));
285 packet_reader_init(&reader, fd[0], NULL, 0,
286 PACKET_READ_CHOMP_NEWLINE |
287 PACKET_READ_DIE_ON_ERR_PACKET);
289 mark_tips(negotiator, args->negotiation_tips);
290 for_each_cached_alternate(negotiator, insert_one_alternate_object);
292 fetching = 0;
293 for ( ; refs ; refs = refs->next) {
294 struct object_id *remote = &refs->old_oid;
295 const char *remote_hex;
296 struct object *o;
299 * If that object is complete (i.e. it is an ancestor of a
300 * local ref), we tell them we have it but do not have to
301 * tell them about its ancestors, which they already know
302 * about.
304 * We use lookup_object here because we are only
305 * interested in the case we *know* the object is
306 * reachable and we have already scanned it.
308 if (((o = lookup_object(the_repository, remote)) != NULL) &&
309 (o->flags & COMPLETE)) {
310 continue;
313 remote_hex = oid_to_hex(remote);
314 if (!fetching) {
315 struct strbuf c = STRBUF_INIT;
316 if (multi_ack == 2) strbuf_addstr(&c, " multi_ack_detailed");
317 if (multi_ack == 1) strbuf_addstr(&c, " multi_ack");
318 if (no_done) strbuf_addstr(&c, " no-done");
319 if (use_sideband == 2) strbuf_addstr(&c, " side-band-64k");
320 if (use_sideband == 1) strbuf_addstr(&c, " side-band");
321 if (args->deepen_relative) strbuf_addstr(&c, " deepen-relative");
322 if (args->use_thin_pack) strbuf_addstr(&c, " thin-pack");
323 if (args->no_progress) strbuf_addstr(&c, " no-progress");
324 if (args->include_tag) strbuf_addstr(&c, " include-tag");
325 if (prefer_ofs_delta) strbuf_addstr(&c, " ofs-delta");
326 if (deepen_since_ok) strbuf_addstr(&c, " deepen-since");
327 if (deepen_not_ok) strbuf_addstr(&c, " deepen-not");
328 if (agent_supported) strbuf_addf(&c, " agent=%s",
329 git_user_agent_sanitized());
330 if (advertise_sid)
331 strbuf_addf(&c, " session-id=%s", trace2_session_id());
332 if (args->filter_options.choice)
333 strbuf_addstr(&c, " filter");
334 packet_buf_write(&req_buf, "want %s%s\n", remote_hex, c.buf);
335 strbuf_release(&c);
336 } else
337 packet_buf_write(&req_buf, "want %s\n", remote_hex);
338 fetching++;
341 if (!fetching) {
342 strbuf_release(&req_buf);
343 packet_flush(fd[1]);
344 return 1;
347 if (is_repository_shallow(the_repository))
348 write_shallow_commits(&req_buf, 1, NULL);
349 if (args->depth > 0)
350 packet_buf_write(&req_buf, "deepen %d", args->depth);
351 if (args->deepen_since) {
352 timestamp_t max_age = approxidate(args->deepen_since);
353 packet_buf_write(&req_buf, "deepen-since %"PRItime, max_age);
355 if (args->deepen_not) {
356 int i;
357 for (i = 0; i < args->deepen_not->nr; i++) {
358 struct string_list_item *s = args->deepen_not->items + i;
359 packet_buf_write(&req_buf, "deepen-not %s", s->string);
362 if (server_supports_filtering && args->filter_options.choice) {
363 const char *spec =
364 expand_list_objects_filter_spec(&args->filter_options);
365 packet_buf_write(&req_buf, "filter %s", spec);
367 packet_buf_flush(&req_buf);
368 state_len = req_buf.len;
370 if (args->deepen) {
371 const char *arg;
372 struct object_id oid;
374 send_request(args, fd[1], &req_buf);
375 while (packet_reader_read(&reader) == PACKET_READ_NORMAL) {
376 if (skip_prefix(reader.line, "shallow ", &arg)) {
377 if (get_oid_hex(arg, &oid))
378 die(_("invalid shallow line: %s"), reader.line);
379 register_shallow(the_repository, &oid);
380 continue;
382 if (skip_prefix(reader.line, "unshallow ", &arg)) {
383 if (get_oid_hex(arg, &oid))
384 die(_("invalid unshallow line: %s"), reader.line);
385 if (!lookup_object(the_repository, &oid))
386 die(_("object not found: %s"), reader.line);
387 /* make sure that it is parsed as shallow */
388 if (!parse_object(the_repository, &oid))
389 die(_("error in object: %s"), reader.line);
390 if (unregister_shallow(&oid))
391 die(_("no shallow found: %s"), reader.line);
392 continue;
394 die(_("expected shallow/unshallow, got %s"), reader.line);
396 } else if (!args->stateless_rpc)
397 send_request(args, fd[1], &req_buf);
399 if (!args->stateless_rpc) {
400 /* If we aren't using the stateless-rpc interface
401 * we don't need to retain the headers.
403 strbuf_setlen(&req_buf, 0);
404 state_len = 0;
407 trace2_region_enter("fetch-pack", "negotiation_v0_v1", the_repository);
408 flushes = 0;
409 retval = -1;
410 while ((oid = negotiator->next(negotiator))) {
411 packet_buf_write(&req_buf, "have %s\n", oid_to_hex(oid));
412 print_verbose(args, "have %s", oid_to_hex(oid));
413 in_vain++;
414 if (flush_at <= ++count) {
415 int ack;
417 packet_buf_flush(&req_buf);
418 send_request(args, fd[1], &req_buf);
419 strbuf_setlen(&req_buf, state_len);
420 flushes++;
421 flush_at = next_flush(args->stateless_rpc, count);
424 * We keep one window "ahead" of the other side, and
425 * will wait for an ACK only on the next one
427 if (!args->stateless_rpc && count == INITIAL_FLUSH)
428 continue;
430 consume_shallow_list(args, &reader);
431 do {
432 ack = get_ack(&reader, result_oid);
433 if (ack)
434 print_verbose(args, _("got %s %d %s"), "ack",
435 ack, oid_to_hex(result_oid));
436 switch (ack) {
437 case ACK:
438 flushes = 0;
439 multi_ack = 0;
440 retval = 0;
441 goto done;
442 case ACK_common:
443 case ACK_ready:
444 case ACK_continue: {
445 struct commit *commit =
446 lookup_commit(the_repository,
447 result_oid);
448 int was_common;
450 if (!commit)
451 die(_("invalid commit %s"), oid_to_hex(result_oid));
452 was_common = negotiator->ack(negotiator, commit);
453 if (args->stateless_rpc
454 && ack == ACK_common
455 && !was_common) {
456 /* We need to replay the have for this object
457 * on the next RPC request so the peer knows
458 * it is in common with us.
460 const char *hex = oid_to_hex(result_oid);
461 packet_buf_write(&req_buf, "have %s\n", hex);
462 state_len = req_buf.len;
464 * Reset in_vain because an ack
465 * for this commit has not been
466 * seen.
468 in_vain = 0;
469 } else if (!args->stateless_rpc
470 || ack != ACK_common)
471 in_vain = 0;
472 retval = 0;
473 got_continue = 1;
474 if (ack == ACK_ready)
475 got_ready = 1;
476 break;
479 } while (ack);
480 flushes--;
481 if (got_continue && MAX_IN_VAIN < in_vain) {
482 print_verbose(args, _("giving up"));
483 break; /* give up */
485 if (got_ready)
486 break;
489 done:
490 trace2_region_leave("fetch-pack", "negotiation_v0_v1", the_repository);
491 if (!got_ready || !no_done) {
492 packet_buf_write(&req_buf, "done\n");
493 send_request(args, fd[1], &req_buf);
495 print_verbose(args, _("done"));
496 if (retval != 0) {
497 multi_ack = 0;
498 flushes++;
500 strbuf_release(&req_buf);
502 if (!got_ready || !no_done)
503 consume_shallow_list(args, &reader);
504 while (flushes || multi_ack) {
505 int ack = get_ack(&reader, result_oid);
506 if (ack) {
507 print_verbose(args, _("got %s (%d) %s"), "ack",
508 ack, oid_to_hex(result_oid));
509 if (ack == ACK)
510 return 0;
511 multi_ack = 1;
512 continue;
514 flushes--;
516 /* it is no error to fetch into a completely empty repo */
517 return count ? retval : 0;
520 static struct commit_list *complete;
522 static int mark_complete(const struct object_id *oid)
524 struct commit *commit = deref_without_lazy_fetch(oid, 1);
526 if (commit && !(commit->object.flags & COMPLETE)) {
527 commit->object.flags |= COMPLETE;
528 commit_list_insert(commit, &complete);
530 return 0;
533 static int mark_complete_oid(const char *refname, const struct object_id *oid,
534 int flag, void *cb_data)
536 return mark_complete(oid);
539 static void mark_recent_complete_commits(struct fetch_pack_args *args,
540 timestamp_t cutoff)
542 while (complete && cutoff <= complete->item->date) {
543 print_verbose(args, _("Marking %s as complete"),
544 oid_to_hex(&complete->item->object.oid));
545 pop_most_recent_commit(&complete, COMPLETE);
549 static void add_refs_to_oidset(struct oidset *oids, struct ref *refs)
551 for (; refs; refs = refs->next)
552 oidset_insert(oids, &refs->old_oid);
555 static int is_unmatched_ref(const struct ref *ref)
557 struct object_id oid;
558 const char *p;
559 return ref->match_status == REF_NOT_MATCHED &&
560 !parse_oid_hex(ref->name, &oid, &p) &&
561 *p == '\0' &&
562 oideq(&oid, &ref->old_oid);
565 static void filter_refs(struct fetch_pack_args *args,
566 struct ref **refs,
567 struct ref **sought, int nr_sought)
569 struct ref *newlist = NULL;
570 struct ref **newtail = &newlist;
571 struct ref *unmatched = NULL;
572 struct ref *ref, *next;
573 struct oidset tip_oids = OIDSET_INIT;
574 int i;
575 int strict = !(allow_unadvertised_object_request &
576 (ALLOW_TIP_SHA1 | ALLOW_REACHABLE_SHA1));
578 i = 0;
579 for (ref = *refs; ref; ref = next) {
580 int keep = 0;
581 next = ref->next;
583 if (starts_with(ref->name, "refs/") &&
584 check_refname_format(ref->name, 0)) {
586 * trash or a peeled value; do not even add it to
587 * unmatched list
589 free_one_ref(ref);
590 continue;
591 } else {
592 while (i < nr_sought) {
593 int cmp = strcmp(ref->name, sought[i]->name);
594 if (cmp < 0)
595 break; /* definitely do not have it */
596 else if (cmp == 0) {
597 keep = 1; /* definitely have it */
598 sought[i]->match_status = REF_MATCHED;
600 i++;
603 if (!keep && args->fetch_all &&
604 (!args->deepen || !starts_with(ref->name, "refs/tags/")))
605 keep = 1;
608 if (keep) {
609 *newtail = ref;
610 ref->next = NULL;
611 newtail = &ref->next;
612 } else {
613 ref->next = unmatched;
614 unmatched = ref;
618 if (strict) {
619 for (i = 0; i < nr_sought; i++) {
620 ref = sought[i];
621 if (!is_unmatched_ref(ref))
622 continue;
624 add_refs_to_oidset(&tip_oids, unmatched);
625 add_refs_to_oidset(&tip_oids, newlist);
626 break;
630 /* Append unmatched requests to the list */
631 for (i = 0; i < nr_sought; i++) {
632 ref = sought[i];
633 if (!is_unmatched_ref(ref))
634 continue;
636 if (!strict || oidset_contains(&tip_oids, &ref->old_oid)) {
637 ref->match_status = REF_MATCHED;
638 *newtail = copy_ref(ref);
639 newtail = &(*newtail)->next;
640 } else {
641 ref->match_status = REF_UNADVERTISED_NOT_ALLOWED;
645 oidset_clear(&tip_oids);
646 free_refs(unmatched);
648 *refs = newlist;
651 static void mark_alternate_complete(struct fetch_negotiator *unused,
652 struct object *obj)
654 mark_complete(&obj->oid);
657 struct loose_object_iter {
658 struct oidset *loose_object_set;
659 struct ref *refs;
663 * Mark recent commits available locally and reachable from a local ref as
664 * COMPLETE.
666 * The cutoff time for recency is determined by this heuristic: it is the
667 * earliest commit time of the objects in refs that are commits and that we know
668 * the commit time of.
670 static void mark_complete_and_common_ref(struct fetch_negotiator *negotiator,
671 struct fetch_pack_args *args,
672 struct ref **refs)
674 struct ref *ref;
675 int old_save_commit_buffer = save_commit_buffer;
676 timestamp_t cutoff = 0;
678 save_commit_buffer = 0;
680 trace2_region_enter("fetch-pack", "parse_remote_refs_and_find_cutoff", NULL);
681 for (ref = *refs; ref; ref = ref->next) {
682 struct object *o;
684 if (!has_object_file_with_flags(&ref->old_oid,
685 OBJECT_INFO_QUICK |
686 OBJECT_INFO_SKIP_FETCH_OBJECT))
687 continue;
688 o = parse_object(the_repository, &ref->old_oid);
689 if (!o)
690 continue;
693 * We already have it -- which may mean that we were
694 * in sync with the other side at some time after
695 * that (it is OK if we guess wrong here).
697 if (o->type == OBJ_COMMIT) {
698 struct commit *commit = (struct commit *)o;
699 if (!cutoff || cutoff < commit->date)
700 cutoff = commit->date;
703 trace2_region_leave("fetch-pack", "parse_remote_refs_and_find_cutoff", NULL);
706 * This block marks all local refs as COMPLETE, and then recursively marks all
707 * parents of those refs as COMPLETE.
709 trace2_region_enter("fetch-pack", "mark_complete_local_refs", NULL);
710 if (!args->deepen) {
711 for_each_rawref(mark_complete_oid, NULL);
712 for_each_cached_alternate(NULL, mark_alternate_complete);
713 commit_list_sort_by_date(&complete);
714 if (cutoff)
715 mark_recent_complete_commits(args, cutoff);
717 trace2_region_leave("fetch-pack", "mark_complete_local_refs", NULL);
720 * Mark all complete remote refs as common refs.
721 * Don't mark them common yet; the server has to be told so first.
723 trace2_region_enter("fetch-pack", "mark_common_remote_refs", NULL);
724 for (ref = *refs; ref; ref = ref->next) {
725 struct commit *c = deref_without_lazy_fetch(&ref->old_oid, 0);
727 if (!c || !(c->object.flags & COMPLETE))
728 continue;
730 negotiator->known_common(negotiator, c);
732 trace2_region_leave("fetch-pack", "mark_common_remote_refs", NULL);
734 save_commit_buffer = old_save_commit_buffer;
738 * Returns 1 if every object pointed to by the given remote refs is available
739 * locally and reachable from a local ref, and 0 otherwise.
741 static int everything_local(struct fetch_pack_args *args,
742 struct ref **refs)
744 struct ref *ref;
745 int retval;
747 for (retval = 1, ref = *refs; ref ; ref = ref->next) {
748 const struct object_id *remote = &ref->old_oid;
749 struct object *o;
751 o = lookup_object(the_repository, remote);
752 if (!o || !(o->flags & COMPLETE)) {
753 retval = 0;
754 print_verbose(args, "want %s (%s)", oid_to_hex(remote),
755 ref->name);
756 continue;
758 print_verbose(args, _("already have %s (%s)"), oid_to_hex(remote),
759 ref->name);
762 return retval;
765 static int sideband_demux(int in, int out, void *data)
767 int *xd = data;
768 int ret;
770 ret = recv_sideband("fetch-pack", xd[0], out);
771 close(out);
772 return ret;
775 static void create_promisor_file(const char *keep_name,
776 struct ref **sought, int nr_sought)
778 struct strbuf promisor_name = STRBUF_INIT;
779 int suffix_stripped;
781 strbuf_addstr(&promisor_name, keep_name);
782 suffix_stripped = strbuf_strip_suffix(&promisor_name, ".keep");
783 if (!suffix_stripped)
784 BUG("name of pack lockfile should end with .keep (was '%s')",
785 keep_name);
786 strbuf_addstr(&promisor_name, ".promisor");
788 write_promisor_file(promisor_name.buf, sought, nr_sought);
790 strbuf_release(&promisor_name);
793 static void parse_gitmodules_oids(int fd, struct oidset *gitmodules_oids)
795 int len = the_hash_algo->hexsz + 1; /* hash + NL */
797 do {
798 char hex_hash[GIT_MAX_HEXSZ + 1];
799 int read_len = read_in_full(fd, hex_hash, len);
800 struct object_id oid;
801 const char *end;
803 if (!read_len)
804 return;
805 if (read_len != len)
806 die("invalid length read %d", read_len);
807 if (parse_oid_hex(hex_hash, &oid, &end) || *end != '\n')
808 die("invalid hash");
809 oidset_insert(gitmodules_oids, &oid);
810 } while (1);
814 * If packfile URIs were provided, pass a non-NULL pointer to index_pack_args.
815 * The strings to pass as the --index-pack-arg arguments to http-fetch will be
816 * stored there. (It must be freed by the caller.)
818 static int get_pack(struct fetch_pack_args *args,
819 int xd[2], struct string_list *pack_lockfiles,
820 struct strvec *index_pack_args,
821 struct ref **sought, int nr_sought,
822 struct oidset *gitmodules_oids)
824 struct async demux;
825 int do_keep = args->keep_pack;
826 const char *cmd_name;
827 struct pack_header header;
828 int pass_header = 0;
829 struct child_process cmd = CHILD_PROCESS_INIT;
830 int fsck_objects = 0;
831 int ret;
833 memset(&demux, 0, sizeof(demux));
834 if (use_sideband) {
835 /* xd[] is talking with upload-pack; subprocess reads from
836 * xd[0], spits out band#2 to stderr, and feeds us band#1
837 * through demux->out.
839 demux.proc = sideband_demux;
840 demux.data = xd;
841 demux.out = -1;
842 demux.isolate_sigpipe = 1;
843 if (start_async(&demux))
844 die(_("fetch-pack: unable to fork off sideband demultiplexer"));
846 else
847 demux.out = xd[0];
849 if (!args->keep_pack && unpack_limit && !index_pack_args) {
851 if (read_pack_header(demux.out, &header))
852 die(_("protocol error: bad pack header"));
853 pass_header = 1;
854 if (ntohl(header.hdr_entries) < unpack_limit)
855 do_keep = 0;
856 else
857 do_keep = 1;
860 if (alternate_shallow_file) {
861 strvec_push(&cmd.args, "--shallow-file");
862 strvec_push(&cmd.args, alternate_shallow_file);
865 if (fetch_fsck_objects >= 0
866 ? fetch_fsck_objects
867 : transfer_fsck_objects >= 0
868 ? transfer_fsck_objects
869 : 0)
870 fsck_objects = 1;
872 if (do_keep || args->from_promisor || index_pack_args || fsck_objects) {
873 if (pack_lockfiles || fsck_objects)
874 cmd.out = -1;
875 cmd_name = "index-pack";
876 strvec_push(&cmd.args, cmd_name);
877 strvec_push(&cmd.args, "--stdin");
878 if (!args->quiet && !args->no_progress)
879 strvec_push(&cmd.args, "-v");
880 if (args->use_thin_pack)
881 strvec_push(&cmd.args, "--fix-thin");
882 if ((do_keep || index_pack_args) && (args->lock_pack || unpack_limit)) {
883 char hostname[HOST_NAME_MAX + 1];
884 if (xgethostname(hostname, sizeof(hostname)))
885 xsnprintf(hostname, sizeof(hostname), "localhost");
886 strvec_pushf(&cmd.args,
887 "--keep=fetch-pack %"PRIuMAX " on %s",
888 (uintmax_t)getpid(), hostname);
890 if (!index_pack_args && args->check_self_contained_and_connected)
891 strvec_push(&cmd.args, "--check-self-contained-and-connected");
892 else
894 * We cannot perform any connectivity checks because
895 * not all packs have been downloaded; let the caller
896 * have this responsibility.
898 args->check_self_contained_and_connected = 0;
900 if (args->from_promisor)
902 * create_promisor_file() may be called afterwards but
903 * we still need index-pack to know that this is a
904 * promisor pack. For example, if transfer.fsckobjects
905 * is true, index-pack needs to know that .gitmodules
906 * is a promisor object (so that it won't complain if
907 * it is missing).
909 strvec_push(&cmd.args, "--promisor");
911 else {
912 cmd_name = "unpack-objects";
913 strvec_push(&cmd.args, cmd_name);
914 if (args->quiet || args->no_progress)
915 strvec_push(&cmd.args, "-q");
916 args->check_self_contained_and_connected = 0;
919 if (pass_header)
920 strvec_pushf(&cmd.args, "--pack_header=%"PRIu32",%"PRIu32,
921 ntohl(header.hdr_version),
922 ntohl(header.hdr_entries));
923 if (fsck_objects) {
924 if (args->from_promisor || index_pack_args)
926 * We cannot use --strict in index-pack because it
927 * checks both broken objects and links, but we only
928 * want to check for broken objects.
930 strvec_push(&cmd.args, "--fsck-objects");
931 else
932 strvec_pushf(&cmd.args, "--strict%s",
933 fsck_msg_types.buf);
936 if (index_pack_args) {
937 int i;
939 for (i = 0; i < cmd.args.nr; i++)
940 strvec_push(index_pack_args, cmd.args.v[i]);
943 cmd.in = demux.out;
944 cmd.git_cmd = 1;
945 if (start_command(&cmd))
946 die(_("fetch-pack: unable to fork off %s"), cmd_name);
947 if (do_keep && (pack_lockfiles || fsck_objects)) {
948 int is_well_formed;
949 char *pack_lockfile = index_pack_lockfile(cmd.out, &is_well_formed);
951 if (!is_well_formed)
952 die(_("fetch-pack: invalid index-pack output"));
953 if (pack_lockfile)
954 string_list_append_nodup(pack_lockfiles, pack_lockfile);
955 parse_gitmodules_oids(cmd.out, gitmodules_oids);
956 close(cmd.out);
959 if (!use_sideband)
960 /* Closed by start_command() */
961 xd[0] = -1;
963 ret = finish_command(&cmd);
964 if (!ret || (args->check_self_contained_and_connected && ret == 1))
965 args->self_contained_and_connected =
966 args->check_self_contained_and_connected &&
967 ret == 0;
968 else
969 die(_("%s failed"), cmd_name);
970 if (use_sideband && finish_async(&demux))
971 die(_("error in sideband demultiplexer"));
974 * Now that index-pack has succeeded, write the promisor file using the
975 * obtained .keep filename if necessary
977 if (do_keep && pack_lockfiles && pack_lockfiles->nr && args->from_promisor)
978 create_promisor_file(pack_lockfiles->items[0].string, sought, nr_sought);
980 return 0;
983 static int cmp_ref_by_name(const void *a_, const void *b_)
985 const struct ref *a = *((const struct ref **)a_);
986 const struct ref *b = *((const struct ref **)b_);
987 return strcmp(a->name, b->name);
990 static void fsck_gitmodules_oids(struct oidset *gitmodules_oids)
992 struct oidset_iter iter;
993 const struct object_id *oid;
994 struct fsck_options fo = FSCK_OPTIONS_STRICT;
996 if (!oidset_size(gitmodules_oids))
997 return;
999 oidset_iter_init(gitmodules_oids, &iter);
1000 while ((oid = oidset_iter_next(&iter)))
1001 register_found_gitmodules(oid);
1002 if (fsck_finish(&fo))
1003 die("fsck failed");
1006 static struct ref *do_fetch_pack(struct fetch_pack_args *args,
1007 int fd[2],
1008 const struct ref *orig_ref,
1009 struct ref **sought, int nr_sought,
1010 struct shallow_info *si,
1011 struct string_list *pack_lockfiles)
1013 struct repository *r = the_repository;
1014 struct ref *ref = copy_ref_list(orig_ref);
1015 struct object_id oid;
1016 const char *agent_feature;
1017 int agent_len;
1018 struct fetch_negotiator negotiator_alloc;
1019 struct fetch_negotiator *negotiator;
1020 struct oidset gitmodules_oids = OIDSET_INIT;
1022 negotiator = &negotiator_alloc;
1023 fetch_negotiator_init(r, negotiator);
1025 sort_ref_list(&ref, ref_compare_name);
1026 QSORT(sought, nr_sought, cmp_ref_by_name);
1028 if ((agent_feature = server_feature_value("agent", &agent_len))) {
1029 agent_supported = 1;
1030 if (agent_len)
1031 print_verbose(args, _("Server version is %.*s"),
1032 agent_len, agent_feature);
1035 if (!server_supports("session-id"))
1036 advertise_sid = 0;
1038 if (server_supports("shallow"))
1039 print_verbose(args, _("Server supports %s"), "shallow");
1040 else if (args->depth > 0 || is_repository_shallow(r))
1041 die(_("Server does not support shallow clients"));
1042 if (args->depth > 0 || args->deepen_since || args->deepen_not)
1043 args->deepen = 1;
1044 if (server_supports("multi_ack_detailed")) {
1045 print_verbose(args, _("Server supports %s"), "multi_ack_detailed");
1046 multi_ack = 2;
1047 if (server_supports("no-done")) {
1048 print_verbose(args, _("Server supports %s"), "no-done");
1049 if (args->stateless_rpc)
1050 no_done = 1;
1053 else if (server_supports("multi_ack")) {
1054 print_verbose(args, _("Server supports %s"), "multi_ack");
1055 multi_ack = 1;
1057 if (server_supports("side-band-64k")) {
1058 print_verbose(args, _("Server supports %s"), "side-band-64k");
1059 use_sideband = 2;
1061 else if (server_supports("side-band")) {
1062 print_verbose(args, _("Server supports %s"), "side-band");
1063 use_sideband = 1;
1065 if (server_supports("allow-tip-sha1-in-want")) {
1066 print_verbose(args, _("Server supports %s"), "allow-tip-sha1-in-want");
1067 allow_unadvertised_object_request |= ALLOW_TIP_SHA1;
1069 if (server_supports("allow-reachable-sha1-in-want")) {
1070 print_verbose(args, _("Server supports %s"), "allow-reachable-sha1-in-want");
1071 allow_unadvertised_object_request |= ALLOW_REACHABLE_SHA1;
1073 if (server_supports("thin-pack"))
1074 print_verbose(args, _("Server supports %s"), "thin-pack");
1075 else
1076 args->use_thin_pack = 0;
1077 if (server_supports("no-progress"))
1078 print_verbose(args, _("Server supports %s"), "no-progress");
1079 else
1080 args->no_progress = 0;
1081 if (server_supports("include-tag"))
1082 print_verbose(args, _("Server supports %s"), "include-tag");
1083 else
1084 args->include_tag = 0;
1085 if (server_supports("ofs-delta"))
1086 print_verbose(args, _("Server supports %s"), "ofs-delta");
1087 else
1088 prefer_ofs_delta = 0;
1090 if (server_supports("filter")) {
1091 server_supports_filtering = 1;
1092 print_verbose(args, _("Server supports %s"), "filter");
1093 } else if (args->filter_options.choice) {
1094 warning("filtering not recognized by server, ignoring");
1097 if (server_supports("deepen-since")) {
1098 print_verbose(args, _("Server supports %s"), "deepen-since");
1099 deepen_since_ok = 1;
1100 } else if (args->deepen_since)
1101 die(_("Server does not support --shallow-since"));
1102 if (server_supports("deepen-not")) {
1103 print_verbose(args, _("Server supports %s"), "deepen-not");
1104 deepen_not_ok = 1;
1105 } else if (args->deepen_not)
1106 die(_("Server does not support --shallow-exclude"));
1107 if (server_supports("deepen-relative"))
1108 print_verbose(args, _("Server supports %s"), "deepen-relative");
1109 else if (args->deepen_relative)
1110 die(_("Server does not support --deepen"));
1111 if (!server_supports_hash(the_hash_algo->name, NULL))
1112 die(_("Server does not support this repository's object format"));
1114 mark_complete_and_common_ref(negotiator, args, &ref);
1115 filter_refs(args, &ref, sought, nr_sought);
1116 if (everything_local(args, &ref)) {
1117 packet_flush(fd[1]);
1118 goto all_done;
1120 if (find_common(negotiator, args, fd, &oid, ref) < 0)
1121 if (!args->keep_pack)
1122 /* When cloning, it is not unusual to have
1123 * no common commit.
1125 warning(_("no common commits"));
1127 if (args->stateless_rpc)
1128 packet_flush(fd[1]);
1129 if (args->deepen)
1130 setup_alternate_shallow(&shallow_lock, &alternate_shallow_file,
1131 NULL);
1132 else if (si->nr_ours || si->nr_theirs)
1133 alternate_shallow_file = setup_temporary_shallow(si->shallow);
1134 else
1135 alternate_shallow_file = NULL;
1136 if (get_pack(args, fd, pack_lockfiles, NULL, sought, nr_sought,
1137 &gitmodules_oids))
1138 die(_("git fetch-pack: fetch failed."));
1139 fsck_gitmodules_oids(&gitmodules_oids);
1141 all_done:
1142 if (negotiator)
1143 negotiator->release(negotiator);
1144 return ref;
1147 static void add_shallow_requests(struct strbuf *req_buf,
1148 const struct fetch_pack_args *args)
1150 if (is_repository_shallow(the_repository))
1151 write_shallow_commits(req_buf, 1, NULL);
1152 if (args->depth > 0)
1153 packet_buf_write(req_buf, "deepen %d", args->depth);
1154 if (args->deepen_since) {
1155 timestamp_t max_age = approxidate(args->deepen_since);
1156 packet_buf_write(req_buf, "deepen-since %"PRItime, max_age);
1158 if (args->deepen_not) {
1159 int i;
1160 for (i = 0; i < args->deepen_not->nr; i++) {
1161 struct string_list_item *s = args->deepen_not->items + i;
1162 packet_buf_write(req_buf, "deepen-not %s", s->string);
1165 if (args->deepen_relative)
1166 packet_buf_write(req_buf, "deepen-relative\n");
1169 static void add_wants(const struct ref *wants, struct strbuf *req_buf)
1171 int use_ref_in_want = server_supports_feature("fetch", "ref-in-want", 0);
1173 for ( ; wants ; wants = wants->next) {
1174 const struct object_id *remote = &wants->old_oid;
1175 struct object *o;
1178 * If that object is complete (i.e. it is an ancestor of a
1179 * local ref), we tell them we have it but do not have to
1180 * tell them about its ancestors, which they already know
1181 * about.
1183 * We use lookup_object here because we are only
1184 * interested in the case we *know* the object is
1185 * reachable and we have already scanned it.
1187 if (((o = lookup_object(the_repository, remote)) != NULL) &&
1188 (o->flags & COMPLETE)) {
1189 continue;
1192 if (!use_ref_in_want || wants->exact_oid)
1193 packet_buf_write(req_buf, "want %s\n", oid_to_hex(remote));
1194 else
1195 packet_buf_write(req_buf, "want-ref %s\n", wants->name);
1199 static void add_common(struct strbuf *req_buf, struct oidset *common)
1201 struct oidset_iter iter;
1202 const struct object_id *oid;
1203 oidset_iter_init(common, &iter);
1205 while ((oid = oidset_iter_next(&iter))) {
1206 packet_buf_write(req_buf, "have %s\n", oid_to_hex(oid));
1210 static int add_haves(struct fetch_negotiator *negotiator,
1211 int seen_ack,
1212 struct strbuf *req_buf,
1213 int *haves_to_send, int *in_vain)
1215 int ret = 0;
1216 int haves_added = 0;
1217 const struct object_id *oid;
1219 while ((oid = negotiator->next(negotiator))) {
1220 packet_buf_write(req_buf, "have %s\n", oid_to_hex(oid));
1221 if (++haves_added >= *haves_to_send)
1222 break;
1225 *in_vain += haves_added;
1226 if (!haves_added || (seen_ack && *in_vain >= MAX_IN_VAIN)) {
1227 /* Send Done */
1228 packet_buf_write(req_buf, "done\n");
1229 ret = 1;
1232 /* Increase haves to send on next round */
1233 *haves_to_send = next_flush(1, *haves_to_send);
1235 return ret;
1238 static int send_fetch_request(struct fetch_negotiator *negotiator, int fd_out,
1239 struct fetch_pack_args *args,
1240 const struct ref *wants, struct oidset *common,
1241 int *haves_to_send, int *in_vain,
1242 int sideband_all, int seen_ack)
1244 int ret = 0;
1245 const char *hash_name;
1246 struct strbuf req_buf = STRBUF_INIT;
1248 if (server_supports_v2("fetch", 1))
1249 packet_buf_write(&req_buf, "command=fetch");
1250 if (server_supports_v2("agent", 0))
1251 packet_buf_write(&req_buf, "agent=%s", git_user_agent_sanitized());
1252 if (advertise_sid && server_supports_v2("session-id", 0))
1253 packet_buf_write(&req_buf, "session-id=%s", trace2_session_id());
1254 if (args->server_options && args->server_options->nr &&
1255 server_supports_v2("server-option", 1)) {
1256 int i;
1257 for (i = 0; i < args->server_options->nr; i++)
1258 packet_buf_write(&req_buf, "server-option=%s",
1259 args->server_options->items[i].string);
1262 if (server_feature_v2("object-format", &hash_name)) {
1263 int hash_algo = hash_algo_by_name(hash_name);
1264 if (hash_algo_by_ptr(the_hash_algo) != hash_algo)
1265 die(_("mismatched algorithms: client %s; server %s"),
1266 the_hash_algo->name, hash_name);
1267 packet_write_fmt(fd_out, "object-format=%s", the_hash_algo->name);
1268 } else if (hash_algo_by_ptr(the_hash_algo) != GIT_HASH_SHA1) {
1269 die(_("the server does not support algorithm '%s'"),
1270 the_hash_algo->name);
1273 packet_buf_delim(&req_buf);
1274 if (args->use_thin_pack)
1275 packet_buf_write(&req_buf, "thin-pack");
1276 if (args->no_progress)
1277 packet_buf_write(&req_buf, "no-progress");
1278 if (args->include_tag)
1279 packet_buf_write(&req_buf, "include-tag");
1280 if (prefer_ofs_delta)
1281 packet_buf_write(&req_buf, "ofs-delta");
1282 if (sideband_all)
1283 packet_buf_write(&req_buf, "sideband-all");
1285 /* Add shallow-info and deepen request */
1286 if (server_supports_feature("fetch", "shallow", 0))
1287 add_shallow_requests(&req_buf, args);
1288 else if (is_repository_shallow(the_repository) || args->deepen)
1289 die(_("Server does not support shallow requests"));
1291 /* Add filter */
1292 if (server_supports_feature("fetch", "filter", 0) &&
1293 args->filter_options.choice) {
1294 const char *spec =
1295 expand_list_objects_filter_spec(&args->filter_options);
1296 print_verbose(args, _("Server supports filter"));
1297 packet_buf_write(&req_buf, "filter %s", spec);
1298 } else if (args->filter_options.choice) {
1299 warning("filtering not recognized by server, ignoring");
1302 if (server_supports_feature("fetch", "packfile-uris", 0)) {
1303 int i;
1304 struct strbuf to_send = STRBUF_INIT;
1306 for (i = 0; i < uri_protocols.nr; i++) {
1307 const char *s = uri_protocols.items[i].string;
1309 if (!strcmp(s, "https") || !strcmp(s, "http")) {
1310 if (to_send.len)
1311 strbuf_addch(&to_send, ',');
1312 strbuf_addstr(&to_send, s);
1315 if (to_send.len) {
1316 packet_buf_write(&req_buf, "packfile-uris %s",
1317 to_send.buf);
1318 strbuf_release(&to_send);
1322 /* add wants */
1323 add_wants(wants, &req_buf);
1325 /* Add all of the common commits we've found in previous rounds */
1326 add_common(&req_buf, common);
1328 /* Add initial haves */
1329 ret = add_haves(negotiator, seen_ack, &req_buf,
1330 haves_to_send, in_vain);
1332 /* Send request */
1333 packet_buf_flush(&req_buf);
1334 if (write_in_full(fd_out, req_buf.buf, req_buf.len) < 0)
1335 die_errno(_("unable to write request to remote"));
1337 strbuf_release(&req_buf);
1338 return ret;
1342 * Processes a section header in a server's response and checks if it matches
1343 * `section`. If the value of `peek` is 1, the header line will be peeked (and
1344 * not consumed); if 0, the line will be consumed and the function will die if
1345 * the section header doesn't match what was expected.
1347 static int process_section_header(struct packet_reader *reader,
1348 const char *section, int peek)
1350 int ret;
1352 if (packet_reader_peek(reader) != PACKET_READ_NORMAL)
1353 die(_("error reading section header '%s'"), section);
1355 ret = !strcmp(reader->line, section);
1357 if (!peek) {
1358 if (!ret)
1359 die(_("expected '%s', received '%s'"),
1360 section, reader->line);
1361 packet_reader_read(reader);
1364 return ret;
1367 enum common_found {
1369 * No commit was found to be possessed by both the client and the
1370 * server, and "ready" was not received.
1372 NO_COMMON_FOUND,
1375 * At least one commit was found to be possessed by both the client and
1376 * the server, and "ready" was not received.
1378 COMMON_FOUND,
1381 * "ready" was received, indicating that the server is ready to send
1382 * the packfile without any further negotiation.
1384 READY
1387 static enum common_found process_acks(struct fetch_negotiator *negotiator,
1388 struct packet_reader *reader,
1389 struct oidset *common)
1391 /* received */
1392 int received_ready = 0;
1393 int received_ack = 0;
1395 process_section_header(reader, "acknowledgments", 0);
1396 while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
1397 const char *arg;
1399 if (!strcmp(reader->line, "NAK"))
1400 continue;
1402 if (skip_prefix(reader->line, "ACK ", &arg)) {
1403 struct object_id oid;
1404 received_ack = 1;
1405 if (!get_oid_hex(arg, &oid)) {
1406 struct commit *commit;
1407 oidset_insert(common, &oid);
1408 commit = lookup_commit(the_repository, &oid);
1409 if (negotiator)
1410 negotiator->ack(negotiator, commit);
1412 continue;
1415 if (!strcmp(reader->line, "ready")) {
1416 received_ready = 1;
1417 continue;
1420 die(_("unexpected acknowledgment line: '%s'"), reader->line);
1423 if (reader->status != PACKET_READ_FLUSH &&
1424 reader->status != PACKET_READ_DELIM)
1425 die(_("error processing acks: %d"), reader->status);
1428 * If an "acknowledgments" section is sent, a packfile is sent if and
1429 * only if "ready" was sent in this section. The other sections
1430 * ("shallow-info" and "wanted-refs") are sent only if a packfile is
1431 * sent. Therefore, a DELIM is expected if "ready" is sent, and a FLUSH
1432 * otherwise.
1434 if (received_ready && reader->status != PACKET_READ_DELIM)
1435 die(_("expected packfile to be sent after 'ready'"));
1436 if (!received_ready && reader->status != PACKET_READ_FLUSH)
1437 die(_("expected no other sections to be sent after no 'ready'"));
1439 return received_ready ? READY :
1440 (received_ack ? COMMON_FOUND : NO_COMMON_FOUND);
1443 static void receive_shallow_info(struct fetch_pack_args *args,
1444 struct packet_reader *reader,
1445 struct oid_array *shallows,
1446 struct shallow_info *si)
1448 int unshallow_received = 0;
1450 process_section_header(reader, "shallow-info", 0);
1451 while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
1452 const char *arg;
1453 struct object_id oid;
1455 if (skip_prefix(reader->line, "shallow ", &arg)) {
1456 if (get_oid_hex(arg, &oid))
1457 die(_("invalid shallow line: %s"), reader->line);
1458 oid_array_append(shallows, &oid);
1459 continue;
1461 if (skip_prefix(reader->line, "unshallow ", &arg)) {
1462 if (get_oid_hex(arg, &oid))
1463 die(_("invalid unshallow line: %s"), reader->line);
1464 if (!lookup_object(the_repository, &oid))
1465 die(_("object not found: %s"), reader->line);
1466 /* make sure that it is parsed as shallow */
1467 if (!parse_object(the_repository, &oid))
1468 die(_("error in object: %s"), reader->line);
1469 if (unregister_shallow(&oid))
1470 die(_("no shallow found: %s"), reader->line);
1471 unshallow_received = 1;
1472 continue;
1474 die(_("expected shallow/unshallow, got %s"), reader->line);
1477 if (reader->status != PACKET_READ_FLUSH &&
1478 reader->status != PACKET_READ_DELIM)
1479 die(_("error processing shallow info: %d"), reader->status);
1481 if (args->deepen || unshallow_received) {
1483 * Treat these as shallow lines caused by our depth settings.
1484 * In v0, these lines cannot cause refs to be rejected; do the
1485 * same.
1487 int i;
1489 for (i = 0; i < shallows->nr; i++)
1490 register_shallow(the_repository, &shallows->oid[i]);
1491 setup_alternate_shallow(&shallow_lock, &alternate_shallow_file,
1492 NULL);
1493 args->deepen = 1;
1494 } else if (shallows->nr) {
1496 * Treat these as shallow lines caused by the remote being
1497 * shallow. In v0, remote refs that reach these objects are
1498 * rejected (unless --update-shallow is set); do the same.
1500 prepare_shallow_info(si, shallows);
1501 if (si->nr_ours || si->nr_theirs)
1502 alternate_shallow_file =
1503 setup_temporary_shallow(si->shallow);
1504 else
1505 alternate_shallow_file = NULL;
1506 } else {
1507 alternate_shallow_file = NULL;
1511 static int cmp_name_ref(const void *name, const void *ref)
1513 return strcmp(name, (*(struct ref **)ref)->name);
1516 static void receive_wanted_refs(struct packet_reader *reader,
1517 struct ref **sought, int nr_sought)
1519 process_section_header(reader, "wanted-refs", 0);
1520 while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
1521 struct object_id oid;
1522 const char *end;
1523 struct ref **found;
1525 if (parse_oid_hex(reader->line, &oid, &end) || *end++ != ' ')
1526 die(_("expected wanted-ref, got '%s'"), reader->line);
1528 found = bsearch(end, sought, nr_sought, sizeof(*sought),
1529 cmp_name_ref);
1530 if (!found)
1531 die(_("unexpected wanted-ref: '%s'"), reader->line);
1532 oidcpy(&(*found)->old_oid, &oid);
1535 if (reader->status != PACKET_READ_DELIM)
1536 die(_("error processing wanted refs: %d"), reader->status);
1539 static void receive_packfile_uris(struct packet_reader *reader,
1540 struct string_list *uris)
1542 process_section_header(reader, "packfile-uris", 0);
1543 while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
1544 if (reader->pktlen < the_hash_algo->hexsz ||
1545 reader->line[the_hash_algo->hexsz] != ' ')
1546 die("expected '<hash> <uri>', got: %s\n", reader->line);
1548 string_list_append(uris, reader->line);
1550 if (reader->status != PACKET_READ_DELIM)
1551 die("expected DELIM");
1554 enum fetch_state {
1555 FETCH_CHECK_LOCAL = 0,
1556 FETCH_SEND_REQUEST,
1557 FETCH_PROCESS_ACKS,
1558 FETCH_GET_PACK,
1559 FETCH_DONE,
1562 static void do_check_stateless_delimiter(const struct fetch_pack_args *args,
1563 struct packet_reader *reader)
1565 check_stateless_delimiter(args->stateless_rpc, reader,
1566 _("git fetch-pack: expected response end packet"));
1569 static struct ref *do_fetch_pack_v2(struct fetch_pack_args *args,
1570 int fd[2],
1571 const struct ref *orig_ref,
1572 struct ref **sought, int nr_sought,
1573 struct oid_array *shallows,
1574 struct shallow_info *si,
1575 struct string_list *pack_lockfiles)
1577 struct repository *r = the_repository;
1578 struct ref *ref = copy_ref_list(orig_ref);
1579 enum fetch_state state = FETCH_CHECK_LOCAL;
1580 struct oidset common = OIDSET_INIT;
1581 struct packet_reader reader;
1582 int in_vain = 0, negotiation_started = 0;
1583 int haves_to_send = INITIAL_FLUSH;
1584 struct fetch_negotiator negotiator_alloc;
1585 struct fetch_negotiator *negotiator;
1586 int seen_ack = 0;
1587 struct string_list packfile_uris = STRING_LIST_INIT_DUP;
1588 int i;
1589 struct strvec index_pack_args = STRVEC_INIT;
1590 struct oidset gitmodules_oids = OIDSET_INIT;
1592 negotiator = &negotiator_alloc;
1593 fetch_negotiator_init(r, negotiator);
1595 packet_reader_init(&reader, fd[0], NULL, 0,
1596 PACKET_READ_CHOMP_NEWLINE |
1597 PACKET_READ_DIE_ON_ERR_PACKET);
1598 if (git_env_bool("GIT_TEST_SIDEBAND_ALL", 1) &&
1599 server_supports_feature("fetch", "sideband-all", 0)) {
1600 reader.use_sideband = 1;
1601 reader.me = "fetch-pack";
1604 while (state != FETCH_DONE) {
1605 switch (state) {
1606 case FETCH_CHECK_LOCAL:
1607 sort_ref_list(&ref, ref_compare_name);
1608 QSORT(sought, nr_sought, cmp_ref_by_name);
1610 /* v2 supports these by default */
1611 allow_unadvertised_object_request |= ALLOW_REACHABLE_SHA1;
1612 use_sideband = 2;
1613 if (args->depth > 0 || args->deepen_since || args->deepen_not)
1614 args->deepen = 1;
1616 /* Filter 'ref' by 'sought' and those that aren't local */
1617 mark_complete_and_common_ref(negotiator, args, &ref);
1618 filter_refs(args, &ref, sought, nr_sought);
1619 if (everything_local(args, &ref))
1620 state = FETCH_DONE;
1621 else
1622 state = FETCH_SEND_REQUEST;
1624 mark_tips(negotiator, args->negotiation_tips);
1625 for_each_cached_alternate(negotiator,
1626 insert_one_alternate_object);
1627 break;
1628 case FETCH_SEND_REQUEST:
1629 if (!negotiation_started) {
1630 negotiation_started = 1;
1631 trace2_region_enter("fetch-pack",
1632 "negotiation_v2",
1633 the_repository);
1635 if (send_fetch_request(negotiator, fd[1], args, ref,
1636 &common,
1637 &haves_to_send, &in_vain,
1638 reader.use_sideband,
1639 seen_ack))
1640 state = FETCH_GET_PACK;
1641 else
1642 state = FETCH_PROCESS_ACKS;
1643 break;
1644 case FETCH_PROCESS_ACKS:
1645 /* Process ACKs/NAKs */
1646 switch (process_acks(negotiator, &reader, &common)) {
1647 case READY:
1649 * Don't check for response delimiter; get_pack() will
1650 * read the rest of this response.
1652 state = FETCH_GET_PACK;
1653 break;
1654 case COMMON_FOUND:
1655 in_vain = 0;
1656 seen_ack = 1;
1657 /* fallthrough */
1658 case NO_COMMON_FOUND:
1659 do_check_stateless_delimiter(args, &reader);
1660 state = FETCH_SEND_REQUEST;
1661 break;
1663 break;
1664 case FETCH_GET_PACK:
1665 trace2_region_leave("fetch-pack",
1666 "negotiation_v2",
1667 the_repository);
1668 /* Check for shallow-info section */
1669 if (process_section_header(&reader, "shallow-info", 1))
1670 receive_shallow_info(args, &reader, shallows, si);
1672 if (process_section_header(&reader, "wanted-refs", 1))
1673 receive_wanted_refs(&reader, sought, nr_sought);
1675 /* get the pack(s) */
1676 if (process_section_header(&reader, "packfile-uris", 1))
1677 receive_packfile_uris(&reader, &packfile_uris);
1678 process_section_header(&reader, "packfile", 0);
1679 if (get_pack(args, fd, pack_lockfiles,
1680 packfile_uris.nr ? &index_pack_args : NULL,
1681 sought, nr_sought, &gitmodules_oids))
1682 die(_("git fetch-pack: fetch failed."));
1683 do_check_stateless_delimiter(args, &reader);
1685 state = FETCH_DONE;
1686 break;
1687 case FETCH_DONE:
1688 continue;
1692 for (i = 0; i < packfile_uris.nr; i++) {
1693 int j;
1694 struct child_process cmd = CHILD_PROCESS_INIT;
1695 char packname[GIT_MAX_HEXSZ + 1];
1696 const char *uri = packfile_uris.items[i].string +
1697 the_hash_algo->hexsz + 1;
1699 strvec_push(&cmd.args, "http-fetch");
1700 strvec_pushf(&cmd.args, "--packfile=%.*s",
1701 (int) the_hash_algo->hexsz,
1702 packfile_uris.items[i].string);
1703 for (j = 0; j < index_pack_args.nr; j++)
1704 strvec_pushf(&cmd.args, "--index-pack-arg=%s",
1705 index_pack_args.v[j]);
1706 strvec_push(&cmd.args, uri);
1707 cmd.git_cmd = 1;
1708 cmd.no_stdin = 1;
1709 cmd.out = -1;
1710 if (start_command(&cmd))
1711 die("fetch-pack: unable to spawn http-fetch");
1713 if (read_in_full(cmd.out, packname, 5) < 0 ||
1714 memcmp(packname, "keep\t", 5))
1715 die("fetch-pack: expected keep then TAB at start of http-fetch output");
1717 if (read_in_full(cmd.out, packname,
1718 the_hash_algo->hexsz + 1) < 0 ||
1719 packname[the_hash_algo->hexsz] != '\n')
1720 die("fetch-pack: expected hash then LF at end of http-fetch output");
1722 packname[the_hash_algo->hexsz] = '\0';
1724 parse_gitmodules_oids(cmd.out, &gitmodules_oids);
1726 close(cmd.out);
1728 if (finish_command(&cmd))
1729 die("fetch-pack: unable to finish http-fetch");
1731 if (memcmp(packfile_uris.items[i].string, packname,
1732 the_hash_algo->hexsz))
1733 die("fetch-pack: pack downloaded from %s does not match expected hash %.*s",
1734 uri, (int) the_hash_algo->hexsz,
1735 packfile_uris.items[i].string);
1737 string_list_append_nodup(pack_lockfiles,
1738 xstrfmt("%s/pack/pack-%s.keep",
1739 get_object_directory(),
1740 packname));
1742 string_list_clear(&packfile_uris, 0);
1743 strvec_clear(&index_pack_args);
1745 fsck_gitmodules_oids(&gitmodules_oids);
1747 if (negotiator)
1748 negotiator->release(negotiator);
1750 oidset_clear(&common);
1751 return ref;
1754 static int fetch_pack_config_cb(const char *var, const char *value, void *cb)
1756 if (strcmp(var, "fetch.fsck.skiplist") == 0) {
1757 const char *path;
1759 if (git_config_pathname(&path, var, value))
1760 return 1;
1761 strbuf_addf(&fsck_msg_types, "%cskiplist=%s",
1762 fsck_msg_types.len ? ',' : '=', path);
1763 free((char *)path);
1764 return 0;
1767 if (skip_prefix(var, "fetch.fsck.", &var)) {
1768 if (is_valid_msg_type(var, value))
1769 strbuf_addf(&fsck_msg_types, "%c%s=%s",
1770 fsck_msg_types.len ? ',' : '=', var, value);
1771 else
1772 warning("Skipping unknown msg id '%s'", var);
1773 return 0;
1776 return git_default_config(var, value, cb);
1779 static void fetch_pack_config(void)
1781 git_config_get_int("fetch.unpacklimit", &fetch_unpack_limit);
1782 git_config_get_int("transfer.unpacklimit", &transfer_unpack_limit);
1783 git_config_get_bool("repack.usedeltabaseoffset", &prefer_ofs_delta);
1784 git_config_get_bool("fetch.fsckobjects", &fetch_fsck_objects);
1785 git_config_get_bool("transfer.fsckobjects", &transfer_fsck_objects);
1786 git_config_get_bool("transfer.advertisesid", &advertise_sid);
1787 if (!uri_protocols.nr) {
1788 char *str;
1790 if (!git_config_get_string("fetch.uriprotocols", &str) && str) {
1791 string_list_split(&uri_protocols, str, ',', -1);
1792 free(str);
1796 git_config(fetch_pack_config_cb, NULL);
1799 static void fetch_pack_setup(void)
1801 static int did_setup;
1802 if (did_setup)
1803 return;
1804 fetch_pack_config();
1805 if (0 <= transfer_unpack_limit)
1806 unpack_limit = transfer_unpack_limit;
1807 else if (0 <= fetch_unpack_limit)
1808 unpack_limit = fetch_unpack_limit;
1809 did_setup = 1;
1812 static int remove_duplicates_in_refs(struct ref **ref, int nr)
1814 struct string_list names = STRING_LIST_INIT_NODUP;
1815 int src, dst;
1817 for (src = dst = 0; src < nr; src++) {
1818 struct string_list_item *item;
1819 item = string_list_insert(&names, ref[src]->name);
1820 if (item->util)
1821 continue; /* already have it */
1822 item->util = ref[src];
1823 if (src != dst)
1824 ref[dst] = ref[src];
1825 dst++;
1827 for (src = dst; src < nr; src++)
1828 ref[src] = NULL;
1829 string_list_clear(&names, 0);
1830 return dst;
1833 static void update_shallow(struct fetch_pack_args *args,
1834 struct ref **sought, int nr_sought,
1835 struct shallow_info *si)
1837 struct oid_array ref = OID_ARRAY_INIT;
1838 int *status;
1839 int i;
1841 if (args->deepen && alternate_shallow_file) {
1842 if (*alternate_shallow_file == '\0') { /* --unshallow */
1843 unlink_or_warn(git_path_shallow(the_repository));
1844 rollback_shallow_file(the_repository, &shallow_lock);
1845 } else
1846 commit_shallow_file(the_repository, &shallow_lock);
1847 alternate_shallow_file = NULL;
1848 return;
1851 if (!si->shallow || !si->shallow->nr)
1852 return;
1854 if (args->cloning) {
1856 * remote is shallow, but this is a clone, there are
1857 * no objects in repo to worry about. Accept any
1858 * shallow points that exist in the pack (iow in repo
1859 * after get_pack() and reprepare_packed_git())
1861 struct oid_array extra = OID_ARRAY_INIT;
1862 struct object_id *oid = si->shallow->oid;
1863 for (i = 0; i < si->shallow->nr; i++)
1864 if (has_object_file(&oid[i]))
1865 oid_array_append(&extra, &oid[i]);
1866 if (extra.nr) {
1867 setup_alternate_shallow(&shallow_lock,
1868 &alternate_shallow_file,
1869 &extra);
1870 commit_shallow_file(the_repository, &shallow_lock);
1871 alternate_shallow_file = NULL;
1873 oid_array_clear(&extra);
1874 return;
1877 if (!si->nr_ours && !si->nr_theirs)
1878 return;
1880 remove_nonexistent_theirs_shallow(si);
1881 if (!si->nr_ours && !si->nr_theirs)
1882 return;
1883 for (i = 0; i < nr_sought; i++)
1884 oid_array_append(&ref, &sought[i]->old_oid);
1885 si->ref = &ref;
1887 if (args->update_shallow) {
1889 * remote is also shallow, .git/shallow may be updated
1890 * so all refs can be accepted. Make sure we only add
1891 * shallow roots that are actually reachable from new
1892 * refs.
1894 struct oid_array extra = OID_ARRAY_INIT;
1895 struct object_id *oid = si->shallow->oid;
1896 assign_shallow_commits_to_refs(si, NULL, NULL);
1897 if (!si->nr_ours && !si->nr_theirs) {
1898 oid_array_clear(&ref);
1899 return;
1901 for (i = 0; i < si->nr_ours; i++)
1902 oid_array_append(&extra, &oid[si->ours[i]]);
1903 for (i = 0; i < si->nr_theirs; i++)
1904 oid_array_append(&extra, &oid[si->theirs[i]]);
1905 setup_alternate_shallow(&shallow_lock,
1906 &alternate_shallow_file,
1907 &extra);
1908 commit_shallow_file(the_repository, &shallow_lock);
1909 oid_array_clear(&extra);
1910 oid_array_clear(&ref);
1911 alternate_shallow_file = NULL;
1912 return;
1916 * remote is also shallow, check what ref is safe to update
1917 * without updating .git/shallow
1919 CALLOC_ARRAY(status, nr_sought);
1920 assign_shallow_commits_to_refs(si, NULL, status);
1921 if (si->nr_ours || si->nr_theirs) {
1922 for (i = 0; i < nr_sought; i++)
1923 if (status[i])
1924 sought[i]->status = REF_STATUS_REJECT_SHALLOW;
1926 free(status);
1927 oid_array_clear(&ref);
1930 static int iterate_ref_map(void *cb_data, struct object_id *oid)
1932 struct ref **rm = cb_data;
1933 struct ref *ref = *rm;
1935 if (!ref)
1936 return -1; /* end of the list */
1937 *rm = ref->next;
1938 oidcpy(oid, &ref->old_oid);
1939 return 0;
1942 struct ref *fetch_pack(struct fetch_pack_args *args,
1943 int fd[],
1944 const struct ref *ref,
1945 struct ref **sought, int nr_sought,
1946 struct oid_array *shallow,
1947 struct string_list *pack_lockfiles,
1948 enum protocol_version version)
1950 struct ref *ref_cpy;
1951 struct shallow_info si;
1952 struct oid_array shallows_scratch = OID_ARRAY_INIT;
1954 fetch_pack_setup();
1955 if (nr_sought)
1956 nr_sought = remove_duplicates_in_refs(sought, nr_sought);
1958 if (version != protocol_v2 && !ref) {
1959 packet_flush(fd[1]);
1960 die(_("no matching remote head"));
1962 if (version == protocol_v2) {
1963 if (shallow->nr)
1964 BUG("Protocol V2 does not provide shallows at this point in the fetch");
1965 memset(&si, 0, sizeof(si));
1966 ref_cpy = do_fetch_pack_v2(args, fd, ref, sought, nr_sought,
1967 &shallows_scratch, &si,
1968 pack_lockfiles);
1969 } else {
1970 prepare_shallow_info(&si, shallow);
1971 ref_cpy = do_fetch_pack(args, fd, ref, sought, nr_sought,
1972 &si, pack_lockfiles);
1974 reprepare_packed_git(the_repository);
1976 if (!args->cloning && args->deepen) {
1977 struct check_connected_options opt = CHECK_CONNECTED_INIT;
1978 struct ref *iterator = ref_cpy;
1979 opt.shallow_file = alternate_shallow_file;
1980 if (args->deepen)
1981 opt.is_deepening_fetch = 1;
1982 if (check_connected(iterate_ref_map, &iterator, &opt)) {
1983 error(_("remote did not send all necessary objects"));
1984 free_refs(ref_cpy);
1985 ref_cpy = NULL;
1986 rollback_shallow_file(the_repository, &shallow_lock);
1987 goto cleanup;
1989 args->connectivity_checked = 1;
1992 update_shallow(args, sought, nr_sought, &si);
1993 cleanup:
1994 clear_shallow_info(&si);
1995 oid_array_clear(&shallows_scratch);
1996 return ref_cpy;
1999 int report_unmatched_refs(struct ref **sought, int nr_sought)
2001 int i, ret = 0;
2003 for (i = 0; i < nr_sought; i++) {
2004 if (!sought[i])
2005 continue;
2006 switch (sought[i]->match_status) {
2007 case REF_MATCHED:
2008 continue;
2009 case REF_NOT_MATCHED:
2010 error(_("no such remote ref %s"), sought[i]->name);
2011 break;
2012 case REF_UNADVERTISED_NOT_ALLOWED:
2013 error(_("Server does not allow request for unadvertised object %s"),
2014 sought[i]->name);
2015 break;
2017 ret = 1;
2019 return ret;