?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
http: refactor finish_http_pack_request()
[git.git] / upload-pack.c
blobbc7e3ca19dd4deea074b0680083f4fde1c5b1910
1 #include "cache.h"
2 #include "config.h"
3 #include "refs.h"
4 #include "pkt-line.h"
5 #include "sideband.h"
6 #include "repository.h"
7 #include "object-store.h"
8 #include "tag.h"
9 #include "object.h"
10 #include "commit.h"
11 #include "diff.h"
12 #include "revision.h"
13 #include "list-objects.h"
14 #include "list-objects-filter.h"
15 #include "list-objects-filter-options.h"
16 #include "run-command.h"
17 #include "connect.h"
18 #include "sigchain.h"
19 #include "version.h"
20 #include "string-list.h"
21 #include "argv-array.h"
22 #include "prio-queue.h"
23 #include "protocol.h"
24 #include "quote.h"
25 #include "upload-pack.h"
26 #include "serve.h"
27 #include "commit-graph.h"
28 #include "commit-reach.h"
29 #include "shallow.h"
30
31 /* Remember to update object flag allocation in object.h */
32 #define THEY_HAVE (1u << 11)
33 #define OUR_REF (1u << 12)
34 #define WANTED (1u << 13)
35 #define COMMON_KNOWN (1u << 14)
36
37 #define SHALLOW (1u << 16)
38 #define NOT_SHALLOW (1u << 17)
39 #define CLIENT_SHALLOW (1u << 18)
40 #define HIDDEN_REF (1u << 19)
41
42 #define ALL_FLAGS (THEY_HAVE | OUR_REF | WANTED | COMMON_KNOWN | SHALLOW | \
43 NOT_SHALLOW | CLIENT_SHALLOW | HIDDEN_REF)
44
45 static timestamp_t oldest_have;
46
47 /* Allow specifying sha1 if it is a ref tip. */
48 #define ALLOW_TIP_SHA1 01
49 /* Allow request of a sha1 if it is reachable from a ref (possibly hidden ref). */
50 #define ALLOW_REACHABLE_SHA1 02
51 /* Allow request of any sha1. Implies ALLOW_TIP_SHA1 and ALLOW_REACHABLE_SHA1. */
52 #define ALLOW_ANY_SHA1 07
53 static unsigned int allow_unadvertised_object_request;
54 static int shallow_nr;
55 static struct object_array extra_edge_obj;
56
57 /*
58 * Please annotate, and if possible group together, fields used only
59 * for protocol v0 or only for protocol v2.
60 */
61 struct upload_pack_data {
62 struct string_list symref; /* v0 only */
63 struct object_array want_obj;
64 struct object_array have_obj;
65 struct oid_array haves; /* v2 only */
66 struct string_list wanted_refs; /* v2 only */
67
68 struct object_array shallows;
69 struct string_list deepen_not;
70 int depth;
71 timestamp_t deepen_since;
72 int deepen_rev_list;
73 int deepen_relative;
74 int keepalive;
75
76 unsigned int timeout; /* v0 only */
77 enum {
78 NO_MULTI_ACK = 0,
79 MULTI_ACK = 1,
80 MULTI_ACK_DETAILED = 2
81 } multi_ack; /* v0 only */
82
83 /* 0 for no sideband, otherwise DEFAULT_PACKET_MAX or LARGE_PACKET_MAX */
84 int use_sideband;
85
86 struct list_objects_filter_options filter_options;
87
88 struct packet_writer writer;
89
90 const char *pack_objects_hook;
91
92 unsigned stateless_rpc : 1; /* v0 only */
93 unsigned no_done : 1; /* v0 only */
94 unsigned daemon_mode : 1; /* v0 only */
95 unsigned filter_capability_requested : 1; /* v0 only */
96
97 unsigned use_thin_pack : 1;
98 unsigned use_ofs_delta : 1;
99 unsigned no_progress : 1;
100 unsigned use_include_tag : 1;
101 unsigned allow_filter : 1;
102
103 unsigned done : 1; /* v2 only */
104 unsigned allow_ref_in_want : 1; /* v2 only */
105 unsigned allow_sideband_all : 1; /* v2 only */
106 };
107
108 static void upload_pack_data_init(struct upload_pack_data *data)
109 {
110 struct string_list symref = STRING_LIST_INIT_DUP;
111 struct string_list wanted_refs = STRING_LIST_INIT_DUP;
112 struct object_array want_obj = OBJECT_ARRAY_INIT;
113 struct object_array have_obj = OBJECT_ARRAY_INIT;
114 struct oid_array haves = OID_ARRAY_INIT;
115 struct object_array shallows = OBJECT_ARRAY_INIT;
116 struct string_list deepen_not = STRING_LIST_INIT_DUP;
117
118 memset(data, 0, sizeof(*data));
119 data->symref = symref;
120 data->wanted_refs = wanted_refs;
121 data->want_obj = want_obj;
122 data->have_obj = have_obj;
123 data->haves = haves;
124 data->shallows = shallows;
125 data->deepen_not = deepen_not;
126 packet_writer_init(&data->writer, 1);
127
128 data->keepalive = 5;
129 }
130
131 static void upload_pack_data_clear(struct upload_pack_data *data)
132 {
133 string_list_clear(&data->symref, 1);
134 string_list_clear(&data->wanted_refs, 1);
135 object_array_clear(&data->want_obj);
136 object_array_clear(&data->have_obj);
137 oid_array_clear(&data->haves);
138 object_array_clear(&data->shallows);
139 string_list_clear(&data->deepen_not, 0);
140 list_objects_filter_release(&data->filter_options);
141
142 free((char *)data->pack_objects_hook);
143 }
144
145 static void reset_timeout(unsigned int timeout)
146 {
147 alarm(timeout);
148 }
149
150 static void send_client_data(int fd, const char *data, ssize_t sz,
151 int use_sideband)
152 {
153 if (use_sideband) {
154 send_sideband(1, fd, data, sz, use_sideband);
155 return;
156 }
157 if (fd == 3)
158 /* emergency quit */
159 fd = 2;
160 if (fd == 2) {
161 /* XXX: are we happy to lose stuff here? */
162 xwrite(fd, data, sz);
163 return;
164 }
165 write_or_die(fd, data, sz);
166 }
167
168 static int write_one_shallow(const struct commit_graft *graft, void *cb_data)
169 {
170 FILE *fp = cb_data;
171 if (graft->nr_parent == -1)
172 fprintf(fp, "--shallow %s\n", oid_to_hex(&graft->oid));
173 return 0;
174 }
175
176 static void create_pack_file(struct upload_pack_data *pack_data)
177 {
178 struct child_process pack_objects = CHILD_PROCESS_INIT;
179 char data[8193], progress[128];
180 char abort_msg[] = "aborting due to possible repository "
181 "corruption on the remote side.";
182 int buffered = -1;
183 ssize_t sz;
184 int i;
185 FILE *pipe_fd;
186
187 if (!pack_data->pack_objects_hook)
188 pack_objects.git_cmd = 1;
189 else {
190 argv_array_push(&pack_objects.args, pack_data->pack_objects_hook);
191 argv_array_push(&pack_objects.args, "git");
192 pack_objects.use_shell = 1;
193 }
194
195 if (shallow_nr) {
196 argv_array_push(&pack_objects.args, "--shallow-file");
197 argv_array_push(&pack_objects.args, "");
198 }
199 argv_array_push(&pack_objects.args, "pack-objects");
200 argv_array_push(&pack_objects.args, "--revs");
201 if (pack_data->use_thin_pack)
202 argv_array_push(&pack_objects.args, "--thin");
203
204 argv_array_push(&pack_objects.args, "--stdout");
205 if (shallow_nr)
206 argv_array_push(&pack_objects.args, "--shallow");
207 if (!pack_data->no_progress)
208 argv_array_push(&pack_objects.args, "--progress");
209 if (pack_data->use_ofs_delta)
210 argv_array_push(&pack_objects.args, "--delta-base-offset");
211 if (pack_data->use_include_tag)
212 argv_array_push(&pack_objects.args, "--include-tag");
213 if (pack_data->filter_options.choice) {
214 const char *spec =
215 expand_list_objects_filter_spec(&pack_data->filter_options);
216 if (pack_objects.use_shell) {
217 struct strbuf buf = STRBUF_INIT;
218 sq_quote_buf(&buf, spec);
219 argv_array_pushf(&pack_objects.args, "--filter=%s", buf.buf);
220 strbuf_release(&buf);
221 } else {
222 argv_array_pushf(&pack_objects.args, "--filter=%s",
223 spec);
224 }
225 }
226
227 pack_objects.in = -1;
228 pack_objects.out = -1;
229 pack_objects.err = -1;
230
231 if (start_command(&pack_objects))
232 die("git upload-pack: unable to fork git-pack-objects");
233
234 pipe_fd = xfdopen(pack_objects.in, "w");
235
236 if (shallow_nr)
237 for_each_commit_graft(write_one_shallow, pipe_fd);
238
239 for (i = 0; i < pack_data->want_obj.nr; i++)
240 fprintf(pipe_fd, "%s\n",
241 oid_to_hex(&pack_data->want_obj.objects[i].item->oid));
242 fprintf(pipe_fd, "--not\n");
243 for (i = 0; i < pack_data->have_obj.nr; i++)
244 fprintf(pipe_fd, "%s\n",
245 oid_to_hex(&pack_data->have_obj.objects[i].item->oid));
246 for (i = 0; i < extra_edge_obj.nr; i++)
247 fprintf(pipe_fd, "%s\n",
248 oid_to_hex(&extra_edge_obj.objects[i].item->oid));
249 fprintf(pipe_fd, "\n");
250 fflush(pipe_fd);
251 fclose(pipe_fd);
252
253 /* We read from pack_objects.err to capture stderr output for
254 * progress bar, and pack_objects.out to capture the pack data.
255 */
256
257 while (1) {
258 struct pollfd pfd[2];
259 int pe, pu, pollsize, polltimeout;
260 int ret;
261
262 reset_timeout(pack_data->timeout);
263
264 pollsize = 0;
265 pe = pu = -1;
266
267 if (0 <= pack_objects.out) {
268 pfd[pollsize].fd = pack_objects.out;
269 pfd[pollsize].events = POLLIN;
270 pu = pollsize;
271 pollsize++;
272 }
273 if (0 <= pack_objects.err) {
274 pfd[pollsize].fd = pack_objects.err;
275 pfd[pollsize].events = POLLIN;
276 pe = pollsize;
277 pollsize++;
278 }
279
280 if (!pollsize)
281 break;
282
283 polltimeout = pack_data->keepalive < 0
284 ? -1
285 : 1000 * pack_data->keepalive;
286
287 ret = poll(pfd, pollsize, polltimeout);
288
289 if (ret < 0) {
290 if (errno != EINTR) {
291 error_errno("poll failed, resuming");
292 sleep(1);
293 }
294 continue;
295 }
296 if (0 <= pe && (pfd[pe].revents & (POLLIN|POLLHUP))) {
297 /* Status ready; we ship that in the side-band
298 * or dump to the standard error.
299 */
300 sz = xread(pack_objects.err, progress,
301 sizeof(progress));
302 if (0 < sz)
303 send_client_data(2, progress, sz,
304 pack_data->use_sideband);
305 else if (sz == 0) {
306 close(pack_objects.err);
307 pack_objects.err = -1;
308 }
309 else
310 goto fail;
311 /* give priority to status messages */
312 continue;
313 }
314 if (0 <= pu && (pfd[pu].revents & (POLLIN|POLLHUP))) {
315 /* Data ready; we keep the last byte to ourselves
316 * in case we detect broken rev-list, so that we
317 * can leave the stream corrupted. This is
318 * unfortunate -- unpack-objects would happily
319 * accept a valid packdata with trailing garbage,
320 * so appending garbage after we pass all the
321 * pack data is not good enough to signal
322 * breakage to downstream.
323 */
324 char *cp = data;
325 ssize_t outsz = 0;
326 if (0 <= buffered) {
327 *cp++ = buffered;
328 outsz++;
329 }
330 sz = xread(pack_objects.out, cp,
331 sizeof(data) - outsz);
332 if (0 < sz)
333 ;
334 else if (sz == 0) {
335 close(pack_objects.out);
336 pack_objects.out = -1;
337 }
338 else
339 goto fail;
340 sz += outsz;
341 if (1 < sz) {
342 buffered = data[sz-1] & 0xFF;
343 sz--;
344 }
345 else
346 buffered = -1;
347 send_client_data(1, data, sz,
348 pack_data->use_sideband);
349 }
350
351 /*
352 * We hit the keepalive timeout without saying anything; send
353 * an empty message on the data sideband just to let the other
354 * side know we're still working on it, but don't have any data
355 * yet.
356 *
357 * If we don't have a sideband channel, there's no room in the
358 * protocol to say anything, so those clients are just out of
359 * luck.
360 */
361 if (!ret && pack_data->use_sideband) {
362 static const char buf[] = "0005\1";
363 write_or_die(1, buf, 5);
364 }
365 }
366
367 if (finish_command(&pack_objects)) {
368 error("git upload-pack: git-pack-objects died with error.");
369 goto fail;
370 }
371
372 /* flush the data */
373 if (0 <= buffered) {
374 data[0] = buffered;
375 send_client_data(1, data, 1,
376 pack_data->use_sideband);
377 fprintf(stderr, "flushed.\n");
378 }
379 if (pack_data->use_sideband)
380 packet_flush(1);
381 return;
382
383 fail:
384 send_client_data(3, abort_msg, sizeof(abort_msg),
385 pack_data->use_sideband);
386 die("git upload-pack: %s", abort_msg);
387 }
388
389 static int got_oid(const char *hex, struct object_id *oid,
390 struct object_array *have_obj)
391 {
392 struct object *o;
393 int we_knew_they_have = 0;
394
395 if (get_oid_hex(hex, oid))
396 die("git upload-pack: expected SHA1 object, got '%s'", hex);
397 if (!has_object_file(oid))
398 return -1;
399
400 o = parse_object(the_repository, oid);
401 if (!o)
402 die("oops (%s)", oid_to_hex(oid));
403 if (o->type == OBJ_COMMIT) {
404 struct commit_list *parents;
405 struct commit *commit = (struct commit *)o;
406 if (o->flags & THEY_HAVE)
407 we_knew_they_have = 1;
408 else
409 o->flags |= THEY_HAVE;
410 if (!oldest_have || (commit->date < oldest_have))
411 oldest_have = commit->date;
412 for (parents = commit->parents;
413 parents;
414 parents = parents->next)
415 parents->item->object.flags |= THEY_HAVE;
416 }
417 if (!we_knew_they_have) {
418 add_object_array(o, NULL, have_obj);
419 return 1;
420 }
421 return 0;
422 }
423
424 static int ok_to_give_up(const struct object_array *have_obj,
425 struct object_array *want_obj)
426 {
427 uint32_t min_generation = GENERATION_NUMBER_ZERO;
428
429 if (!have_obj->nr)
430 return 0;
431
432 return can_all_from_reach_with_flag(want_obj, THEY_HAVE,
433 COMMON_KNOWN, oldest_have,
434 min_generation);
435 }
436
437 static int get_common_commits(struct upload_pack_data *data,
438 struct packet_reader *reader)
439 {
440 struct object_id oid;
441 char last_hex[GIT_MAX_HEXSZ + 1];
442 int got_common = 0;
443 int got_other = 0;
444 int sent_ready = 0;
445
446 save_commit_buffer = 0;
447
448 for (;;) {
449 const char *arg;
450
451 reset_timeout(data->timeout);
452
453 if (packet_reader_read(reader) != PACKET_READ_NORMAL) {
454 if (data->multi_ack == MULTI_ACK_DETAILED
455 && got_common
456 && !got_other
457 && ok_to_give_up(&data->have_obj, &data->want_obj)) {
458 sent_ready = 1;
459 packet_write_fmt(1, "ACK %s ready\n", last_hex);
460 }
461 if (data->have_obj.nr == 0 || data->multi_ack)
462 packet_write_fmt(1, "NAK\n");
463
464 if (data->no_done && sent_ready) {
465 packet_write_fmt(1, "ACK %s\n", last_hex);
466 return 0;
467 }
468 if (data->stateless_rpc)
469 exit(0);
470 got_common = 0;
471 got_other = 0;
472 continue;
473 }
474 if (skip_prefix(reader->line, "have ", &arg)) {
475 switch (got_oid(arg, &oid, &data->have_obj)) {
476 case -1: /* they have what we do not */
477 got_other = 1;
478 if (data->multi_ack
479 && ok_to_give_up(&data->have_obj, &data->want_obj)) {
480 const char *hex = oid_to_hex(&oid);
481 if (data->multi_ack == MULTI_ACK_DETAILED) {
482 sent_ready = 1;
483 packet_write_fmt(1, "ACK %s ready\n", hex);
484 } else
485 packet_write_fmt(1, "ACK %s continue\n", hex);
486 }
487 break;
488 default:
489 got_common = 1;
490 oid_to_hex_r(last_hex, &oid);
491 if (data->multi_ack == MULTI_ACK_DETAILED)
492 packet_write_fmt(1, "ACK %s common\n", last_hex);
493 else if (data->multi_ack)
494 packet_write_fmt(1, "ACK %s continue\n", last_hex);
495 else if (data->have_obj.nr == 1)
496 packet_write_fmt(1, "ACK %s\n", last_hex);
497 break;
498 }
499 continue;
500 }
501 if (!strcmp(reader->line, "done")) {
502 if (data->have_obj.nr > 0) {
503 if (data->multi_ack)
504 packet_write_fmt(1, "ACK %s\n", last_hex);
505 return 0;
506 }
507 packet_write_fmt(1, "NAK\n");
508 return -1;
509 }
510 die("git upload-pack: expected SHA1 list, got '%s'", reader->line);
511 }
512 }
513
514 static int is_our_ref(struct object *o)
515 {
516 int allow_hidden_ref = (allow_unadvertised_object_request &
517 (ALLOW_TIP_SHA1 | ALLOW_REACHABLE_SHA1));
518 return o->flags & ((allow_hidden_ref ? HIDDEN_REF : 0) | OUR_REF);
519 }
520
521 /*
522 * on successful case, it's up to the caller to close cmd->out
523 */
524 static int do_reachable_revlist(struct child_process *cmd,
525 struct object_array *src,
526 struct object_array *reachable)
527 {
528 static const char *argv[] = {
529 "rev-list", "--stdin", NULL,
530 };
531 struct object *o;
532 char namebuf[GIT_MAX_HEXSZ + 2]; /* ^ + hash + LF */
533 int i;
534 const unsigned hexsz = the_hash_algo->hexsz;
535
536 cmd->argv = argv;
537 cmd->git_cmd = 1;
538 cmd->no_stderr = 1;
539 cmd->in = -1;
540 cmd->out = -1;
541
542 /*
543 * If the next rev-list --stdin encounters an unknown commit,
544 * it terminates, which will cause SIGPIPE in the write loop
545 * below.
546 */
547 sigchain_push(SIGPIPE, SIG_IGN);
548
549 if (start_command(cmd))
550 goto error;
551
552 namebuf[0] = '^';
553 namebuf[hexsz + 1] = '\n';
554 for (i = get_max_object_index(); 0 < i; ) {
555 o = get_indexed_object(--i);
556 if (!o)
557 continue;
558 if (reachable && o->type == OBJ_COMMIT)
559 o->flags &= ~TMP_MARK;
560 if (!is_our_ref(o))
561 continue;
562 memcpy(namebuf + 1, oid_to_hex(&o->oid), hexsz);
563 if (write_in_full(cmd->in, namebuf, hexsz + 2) < 0)
564 goto error;
565 }
566 namebuf[hexsz] = '\n';
567 for (i = 0; i < src->nr; i++) {
568 o = src->objects[i].item;
569 if (is_our_ref(o)) {
570 if (reachable)
571 add_object_array(o, NULL, reachable);
572 continue;
573 }
574 if (reachable && o->type == OBJ_COMMIT)
575 o->flags |= TMP_MARK;
576 memcpy(namebuf, oid_to_hex(&o->oid), hexsz);
577 if (write_in_full(cmd->in, namebuf, hexsz + 1) < 0)
578 goto error;
579 }
580 close(cmd->in);
581 cmd->in = -1;
582 sigchain_pop(SIGPIPE);
583
584 return 0;
585
586 error:
587 sigchain_pop(SIGPIPE);
588
589 if (cmd->in >= 0)
590 close(cmd->in);
591 if (cmd->out >= 0)
592 close(cmd->out);
593 return -1;
594 }
595
596 static int get_reachable_list(struct object_array *src,
597 struct object_array *reachable)
598 {
599 struct child_process cmd = CHILD_PROCESS_INIT;
600 int i;
601 struct object *o;
602 char namebuf[GIT_MAX_HEXSZ + 2]; /* ^ + hash + LF */
603 const unsigned hexsz = the_hash_algo->hexsz;
604
605 if (do_reachable_revlist(&cmd, src, reachable) < 0)
606 return -1;
607
608 while ((i = read_in_full(cmd.out, namebuf, hexsz + 1)) == hexsz + 1) {
609 struct object_id oid;
610 const char *p;
611
612 if (parse_oid_hex(namebuf, &oid, &p) || *p != '\n')
613 break;
614
615 o = lookup_object(the_repository, &oid);
616 if (o && o->type == OBJ_COMMIT) {
617 o->flags &= ~TMP_MARK;
618 }
619 }
620 for (i = get_max_object_index(); 0 < i; i--) {
621 o = get_indexed_object(i - 1);
622 if (o && o->type == OBJ_COMMIT &&
623 (o->flags & TMP_MARK)) {
624 add_object_array(o, NULL, reachable);
625 o->flags &= ~TMP_MARK;
626 }
627 }
628 close(cmd.out);
629
630 if (finish_command(&cmd))
631 return -1;
632
633 return 0;
634 }
635
636 static int has_unreachable(struct object_array *src)
637 {
638 struct child_process cmd = CHILD_PROCESS_INIT;
639 char buf[1];
640 int i;
641
642 if (do_reachable_revlist(&cmd, src, NULL) < 0)
643 return 1;
644
645 /*
646 * The commits out of the rev-list are not ancestors of
647 * our ref.
648 */
649 i = read_in_full(cmd.out, buf, 1);
650 if (i)
651 goto error;
652 close(cmd.out);
653 cmd.out = -1;
654
655 /*
656 * rev-list may have died by encountering a bad commit
657 * in the history, in which case we do want to bail out
658 * even when it showed no commit.
659 */
660 if (finish_command(&cmd))
661 goto error;
662
663 /* All the non-tip ones are ancestors of what we advertised */
664 return 0;
665
666 error:
667 sigchain_pop(SIGPIPE);
668 if (cmd.out >= 0)
669 close(cmd.out);
670 return 1;
671 }
672
673 static void check_non_tip(struct upload_pack_data *data)
674 {
675 int i;
676
677 /*
678 * In the normal in-process case without
679 * uploadpack.allowReachableSHA1InWant,
680 * non-tip requests can never happen.
681 */
682 if (!data->stateless_rpc
683 && !(allow_unadvertised_object_request & ALLOW_REACHABLE_SHA1))
684 goto error;
685 if (!has_unreachable(&data->want_obj))
686 /* All the non-tip ones are ancestors of what we advertised */
687 return;
688
689 error:
690 /* Pick one of them (we know there at least is one) */
691 for (i = 0; i < data->want_obj.nr; i++) {
692 struct object *o = data->want_obj.objects[i].item;
693 if (!is_our_ref(o)) {
694 packet_writer_error(&data->writer,
695 "upload-pack: not our ref %s",
696 oid_to_hex(&o->oid));
697 die("git upload-pack: not our ref %s",
698 oid_to_hex(&o->oid));
699 }
700 }
701 }
702
703 static void send_shallow(struct packet_writer *writer,
704 struct commit_list *result)
705 {
706 while (result) {
707 struct object *object = &result->item->object;
708 if (!(object->flags & (CLIENT_SHALLOW|NOT_SHALLOW))) {
709 packet_writer_write(writer, "shallow %s",
710 oid_to_hex(&object->oid));
711 register_shallow(the_repository, &object->oid);
712 shallow_nr++;
713 }
714 result = result->next;
715 }
716 }
717
718 static void send_unshallow(struct packet_writer *writer,
719 const struct object_array *shallows,
720 struct object_array *want_obj)
721 {
722 int i;
723
724 for (i = 0; i < shallows->nr; i++) {
725 struct object *object = shallows->objects[i].item;
726 if (object->flags & NOT_SHALLOW) {
727 struct commit_list *parents;
728 packet_writer_write(writer, "unshallow %s",
729 oid_to_hex(&object->oid));
730 object->flags &= ~CLIENT_SHALLOW;
731 /*
732 * We want to _register_ "object" as shallow, but we
733 * also need to traverse object's parents to deepen a
734 * shallow clone. Unregister it for now so we can
735 * parse and add the parents to the want list, then
736 * re-register it.
737 */
738 unregister_shallow(&object->oid);
739 object->parsed = 0;
740 parse_commit_or_die((struct commit *)object);
741 parents = ((struct commit *)object)->parents;
742 while (parents) {
743 add_object_array(&parents->item->object,
744 NULL, want_obj);
745 parents = parents->next;
746 }
747 add_object_array(object, NULL, &extra_edge_obj);
748 }
749 /* make sure commit traversal conforms to client */
750 register_shallow(the_repository, &object->oid);
751 }
752 }
753
754 static int check_ref(const char *refname_full, const struct object_id *oid,
755 int flag, void *cb_data);
756 static void deepen(struct packet_writer *writer, int depth, int deepen_relative,
757 struct object_array *shallows, struct object_array *want_obj)
758 {
759 if (depth == INFINITE_DEPTH && !is_repository_shallow(the_repository)) {
760 int i;
761
762 for (i = 0; i < shallows->nr; i++) {
763 struct object *object = shallows->objects[i].item;
764 object->flags |= NOT_SHALLOW;
765 }
766 } else if (deepen_relative) {
767 struct object_array reachable_shallows = OBJECT_ARRAY_INIT;
768 struct commit_list *result;
769
770 /*
771 * Checking for reachable shallows requires that our refs be
772 * marked with OUR_REF.
773 */
774 head_ref_namespaced(check_ref, NULL);
775 for_each_namespaced_ref(check_ref, NULL);
776
777 get_reachable_list(shallows, &reachable_shallows);
778 result = get_shallow_commits(&reachable_shallows,
779 depth + 1,
780 SHALLOW, NOT_SHALLOW);
781 send_shallow(writer, result);
782 free_commit_list(result);
783 object_array_clear(&reachable_shallows);
784 } else {
785 struct commit_list *result;
786
787 result = get_shallow_commits(want_obj, depth,
788 SHALLOW, NOT_SHALLOW);
789 send_shallow(writer, result);
790 free_commit_list(result);
791 }
792
793 send_unshallow(writer, shallows, want_obj);
794 }
795
796 static void deepen_by_rev_list(struct packet_writer *writer, int ac,
797 const char **av,
798 struct object_array *shallows,
799 struct object_array *want_obj)
800 {
801 struct commit_list *result;
802
803 disable_commit_graph(the_repository);
804 result = get_shallow_commits_by_rev_list(ac, av, SHALLOW, NOT_SHALLOW);
805 send_shallow(writer, result);
806 free_commit_list(result);
807 send_unshallow(writer, shallows, want_obj);
808 }
809
810 /* Returns 1 if a shallow list is sent or 0 otherwise */
811 static int send_shallow_list(struct packet_writer *writer,
812 int depth, int deepen_rev_list,
813 timestamp_t deepen_since,
814 struct string_list *deepen_not,
815 int deepen_relative,
816 struct object_array *shallows,
817 struct object_array *want_obj)
818 {
819 int ret = 0;
820
821 if (depth > 0 && deepen_rev_list)
822 die("git upload-pack: deepen and deepen-since (or deepen-not) cannot be used together");
823 if (depth > 0) {
824 deepen(writer, depth, deepen_relative, shallows, want_obj);
825 ret = 1;
826 } else if (deepen_rev_list) {
827 struct argv_array av = ARGV_ARRAY_INIT;
828 int i;
829
830 argv_array_push(&av, "rev-list");
831 if (deepen_since)
832 argv_array_pushf(&av, "--max-age=%"PRItime, deepen_since);
833 if (deepen_not->nr) {
834 argv_array_push(&av, "--not");
835 for (i = 0; i < deepen_not->nr; i++) {
836 struct string_list_item *s = deepen_not->items + i;
837 argv_array_push(&av, s->string);
838 }
839 argv_array_push(&av, "--not");
840 }
841 for (i = 0; i < want_obj->nr; i++) {
842 struct object *o = want_obj->objects[i].item;
843 argv_array_push(&av, oid_to_hex(&o->oid));
844 }
845 deepen_by_rev_list(writer, av.argc, av.argv, shallows, want_obj);
846 argv_array_clear(&av);
847 ret = 1;
848 } else {
849 if (shallows->nr > 0) {
850 int i;
851 for (i = 0; i < shallows->nr; i++)
852 register_shallow(the_repository,
853 &shallows->objects[i].item->oid);
854 }
855 }
856
857 shallow_nr += shallows->nr;
858 return ret;
859 }
860
861 static int process_shallow(const char *line, struct object_array *shallows)
862 {
863 const char *arg;
864 if (skip_prefix(line, "shallow ", &arg)) {
865 struct object_id oid;
866 struct object *object;
867 if (get_oid_hex(arg, &oid))
868 die("invalid shallow line: %s", line);
869 object = parse_object(the_repository, &oid);
870 if (!object)
871 return 1;
872 if (object->type != OBJ_COMMIT)
873 die("invalid shallow object %s", oid_to_hex(&oid));
874 if (!(object->flags & CLIENT_SHALLOW)) {
875 object->flags |= CLIENT_SHALLOW;
876 add_object_array(object, NULL, shallows);
877 }
878 return 1;
879 }
880
881 return 0;
882 }
883
884 static int process_deepen(const char *line, int *depth)
885 {
886 const char *arg;
887 if (skip_prefix(line, "deepen ", &arg)) {
888 char *end = NULL;
889 *depth = (int)strtol(arg, &end, 0);
890 if (!end || *end || *depth <= 0)
891 die("Invalid deepen: %s", line);
892 return 1;
893 }
894
895 return 0;
896 }
897
898 static int process_deepen_since(const char *line, timestamp_t *deepen_since, int *deepen_rev_list)
899 {
900 const char *arg;
901 if (skip_prefix(line, "deepen-since ", &arg)) {
902 char *end = NULL;
903 *deepen_since = parse_timestamp(arg, &end, 0);
904 if (!end || *end || !deepen_since ||
905 /* revisions.c's max_age -1 is special */
906 *deepen_since == -1)
907 die("Invalid deepen-since: %s", line);
908 *deepen_rev_list = 1;
909 return 1;
910 }
911 return 0;
912 }
913
914 static int process_deepen_not(const char *line, struct string_list *deepen_not, int *deepen_rev_list)
915 {
916 const char *arg;
917 if (skip_prefix(line, "deepen-not ", &arg)) {
918 char *ref = NULL;
919 struct object_id oid;
920 if (expand_ref(the_repository, arg, strlen(arg), &oid, &ref) != 1)
921 die("git upload-pack: ambiguous deepen-not: %s", line);
922 string_list_append(deepen_not, ref);
923 free(ref);
924 *deepen_rev_list = 1;
925 return 1;
926 }
927 return 0;
928 }
929
930 static void receive_needs(struct upload_pack_data *data,
931 struct packet_reader *reader)
932 {
933 int has_non_tip = 0;
934
935 shallow_nr = 0;
936 for (;;) {
937 struct object *o;
938 const char *features;
939 struct object_id oid_buf;
940 const char *arg;
941
942 reset_timeout(data->timeout);
943 if (packet_reader_read(reader) != PACKET_READ_NORMAL)
944 break;
945
946 if (process_shallow(reader->line, &data->shallows))
947 continue;
948 if (process_deepen(reader->line, &data->depth))
949 continue;
950 if (process_deepen_since(reader->line, &data->deepen_since, &data->deepen_rev_list))
951 continue;
952 if (process_deepen_not(reader->line, &data->deepen_not, &data->deepen_rev_list))
953 continue;
954
955 if (skip_prefix(reader->line, "filter ", &arg)) {
956 if (!data->filter_capability_requested)
957 die("git upload-pack: filtering capability not negotiated");
958 list_objects_filter_die_if_populated(&data->filter_options);
959 parse_list_objects_filter(&data->filter_options, arg);
960 continue;
961 }
962
963 if (!skip_prefix(reader->line, "want ", &arg) ||
964 parse_oid_hex(arg, &oid_buf, &features))
965 die("git upload-pack: protocol error, "
966 "expected to get object ID, not '%s'", reader->line);
967
968 if (parse_feature_request(features, "deepen-relative"))
969 data->deepen_relative = 1;
970 if (parse_feature_request(features, "multi_ack_detailed"))
971 data->multi_ack = MULTI_ACK_DETAILED;
972 else if (parse_feature_request(features, "multi_ack"))
973 data->multi_ack = MULTI_ACK;
974 if (parse_feature_request(features, "no-done"))
975 data->no_done = 1;
976 if (parse_feature_request(features, "thin-pack"))
977 data->use_thin_pack = 1;
978 if (parse_feature_request(features, "ofs-delta"))
979 data->use_ofs_delta = 1;
980 if (parse_feature_request(features, "side-band-64k"))
981 data->use_sideband = LARGE_PACKET_MAX;
982 else if (parse_feature_request(features, "side-band"))
983 data->use_sideband = DEFAULT_PACKET_MAX;
984 if (parse_feature_request(features, "no-progress"))
985 data->no_progress = 1;
986 if (parse_feature_request(features, "include-tag"))
987 data->use_include_tag = 1;
988 if (data->allow_filter &&
989 parse_feature_request(features, "filter"))
990 data->filter_capability_requested = 1;
991
992 o = parse_object(the_repository, &oid_buf);
993 if (!o) {
994 packet_writer_error(&data->writer,
995 "upload-pack: not our ref %s",
996 oid_to_hex(&oid_buf));
997 die("git upload-pack: not our ref %s",
998 oid_to_hex(&oid_buf));
999 }
1000 if (!(o->flags & WANTED)) {
1001 o->flags |= WANTED;
1002 if (!((allow_unadvertised_object_request & ALLOW_ANY_SHA1) == ALLOW_ANY_SHA1
1003 || is_our_ref(o)))
1004 has_non_tip = 1;
1005 add_object_array(o, NULL, &data->want_obj);
1006 }
1007 }
1008
1009 /*
1010 * We have sent all our refs already, and the other end
1011 * should have chosen out of them. When we are operating
1012 * in the stateless RPC mode, however, their choice may
1013 * have been based on the set of older refs advertised
1014 * by another process that handled the initial request.
1015 */
1016 if (has_non_tip)
1017 check_non_tip(data);
1018
1019 if (!data->use_sideband && data->daemon_mode)
1020 data->no_progress = 1;
1021
1022 if (data->depth == 0 && !data->deepen_rev_list && data->shallows.nr == 0)
1023 return;
1024
1025 if (send_shallow_list(&data->writer,
1026 data->depth,
1027 data->deepen_rev_list,
1028 data->deepen_since,
1029 &data->deepen_not,
1030 data->deepen_relative,
1031 &data->shallows,
1032 &data->want_obj))
1033 packet_flush(1);
1034 }
1035
1036 /* return non-zero if the ref is hidden, otherwise 0 */
1037 static int mark_our_ref(const char *refname, const char *refname_full,
1038 const struct object_id *oid)
1039 {
1040 struct object *o = lookup_unknown_object(oid);
1041
1042 if (ref_is_hidden(refname, refname_full)) {
1043 o->flags |= HIDDEN_REF;
1044 return 1;
1045 }
1046 o->flags |= OUR_REF;
1047 return 0;
1048 }
1049
1050 static int check_ref(const char *refname_full, const struct object_id *oid,
1051 int flag, void *cb_data)
1052 {
1053 const char *refname = strip_namespace(refname_full);
1054
1055 mark_our_ref(refname, refname_full, oid);
1056 return 0;
1057 }
1058
1059 static void format_symref_info(struct strbuf *buf, struct string_list *symref)
1060 {
1061 struct string_list_item *item;
1062
1063 if (!symref->nr)
1064 return;
1065 for_each_string_list_item(item, symref)
1066 strbuf_addf(buf, " symref=%s:%s", item->string, (char *)item->util);
1067 }
1068
1069 static int send_ref(const char *refname, const struct object_id *oid,
1070 int flag, void *cb_data)
1071 {
1072 static const char *capabilities = "multi_ack thin-pack side-band"
1073 " side-band-64k ofs-delta shallow deepen-since deepen-not"
1074 " deepen-relative no-progress include-tag multi_ack_detailed";
1075 const char *refname_nons = strip_namespace(refname);
1076 struct object_id peeled;
1077 struct upload_pack_data *data = cb_data;
1078
1079 if (mark_our_ref(refname_nons, refname, oid))
1080 return 0;
1081
1082 if (capabilities) {
1083 struct strbuf symref_info = STRBUF_INIT;
1084
1085 format_symref_info(&symref_info, &data->symref);
1086 packet_write_fmt(1, "%s %s%c%s%s%s%s%s%s agent=%s\n",
1087 oid_to_hex(oid), refname_nons,
1088 0, capabilities,
1089 (allow_unadvertised_object_request & ALLOW_TIP_SHA1) ?
1090 " allow-tip-sha1-in-want" : "",
1091 (allow_unadvertised_object_request & ALLOW_REACHABLE_SHA1) ?
1092 " allow-reachable-sha1-in-want" : "",
1093 data->stateless_rpc ? " no-done" : "",
1094 symref_info.buf,
1095 data->allow_filter ? " filter" : "",
1096 git_user_agent_sanitized());
1097 strbuf_release(&symref_info);
1098 } else {
1099 packet_write_fmt(1, "%s %s\n", oid_to_hex(oid), refname_nons);
1100 }
1101 capabilities = NULL;
1102 if (!peel_ref(refname, &peeled))
1103 packet_write_fmt(1, "%s %s^{}\n", oid_to_hex(&peeled), refname_nons);
1104 return 0;
1105 }
1106
1107 static int find_symref(const char *refname, const struct object_id *oid,
1108 int flag, void *cb_data)
1109 {
1110 const char *symref_target;
1111 struct string_list_item *item;
1112
1113 if ((flag & REF_ISSYMREF) == 0)
1114 return 0;
1115 symref_target = resolve_ref_unsafe(refname, 0, NULL, &flag);
1116 if (!symref_target || (flag & REF_ISSYMREF) == 0)
1117 die("'%s' is a symref but it is not?", refname);
1118 item = string_list_append(cb_data, strip_namespace(refname));
1119 item->util = xstrdup(strip_namespace(symref_target));
1120 return 0;
1121 }
1122
1123 static int upload_pack_config(const char *var, const char *value, void *cb_data)
1124 {
1125 struct upload_pack_data *data = cb_data;
1126
1127 if (!strcmp("uploadpack.allowtipsha1inwant", var)) {
1128 if (git_config_bool(var, value))
1129 allow_unadvertised_object_request |= ALLOW_TIP_SHA1;
1130 else
1131 allow_unadvertised_object_request &= ~ALLOW_TIP_SHA1;
1132 } else if (!strcmp("uploadpack.allowreachablesha1inwant", var)) {
1133 if (git_config_bool(var, value))
1134 allow_unadvertised_object_request |= ALLOW_REACHABLE_SHA1;
1135 else
1136 allow_unadvertised_object_request &= ~ALLOW_REACHABLE_SHA1;
1137 } else if (!strcmp("uploadpack.allowanysha1inwant", var)) {
1138 if (git_config_bool(var, value))
1139 allow_unadvertised_object_request |= ALLOW_ANY_SHA1;
1140 else
1141 allow_unadvertised_object_request &= ~ALLOW_ANY_SHA1;
1142 } else if (!strcmp("uploadpack.keepalive", var)) {
1143 data->keepalive = git_config_int(var, value);
1144 if (!data->keepalive)
1145 data->keepalive = -1;
1146 } else if (!strcmp("uploadpack.allowfilter", var)) {
1147 data->allow_filter = git_config_bool(var, value);
1148 } else if (!strcmp("uploadpack.allowrefinwant", var)) {
1149 data->allow_ref_in_want = git_config_bool(var, value);
1150 } else if (!strcmp("uploadpack.allowsidebandall", var)) {
1151 data->allow_sideband_all = git_config_bool(var, value);
1152 } else if (!strcmp("core.precomposeunicode", var)) {
1153 precomposed_unicode = git_config_bool(var, value);
1154 }
1155
1156 if (current_config_scope() != CONFIG_SCOPE_LOCAL &&
1157 current_config_scope() != CONFIG_SCOPE_WORKTREE) {
1158 if (!strcmp("uploadpack.packobjectshook", var))
1159 return git_config_string(&data->pack_objects_hook, var, value);
1160 }
1161
1162 return parse_hide_refs_config(var, value, "uploadpack");
1163 }
1164
1165 void upload_pack(struct upload_pack_options *options)
1166 {
1167 struct packet_reader reader;
1168 struct upload_pack_data data;
1169
1170 upload_pack_data_init(&data);
1171
1172 git_config(upload_pack_config, &data);
1173
1174 data.stateless_rpc = options->stateless_rpc;
1175 data.daemon_mode = options->daemon_mode;
1176 data.timeout = options->timeout;
1177
1178 head_ref_namespaced(find_symref, &data.symref);
1179
1180 if (options->advertise_refs || !data.stateless_rpc) {
1181 reset_timeout(data.timeout);
1182 head_ref_namespaced(send_ref, &data);
1183 for_each_namespaced_ref(send_ref, &data);
1184 advertise_shallow_grafts(1);
1185 packet_flush(1);
1186 } else {
1187 head_ref_namespaced(check_ref, NULL);
1188 for_each_namespaced_ref(check_ref, NULL);
1189 }
1190
1191 if (!options->advertise_refs) {
1192 packet_reader_init(&reader, 0, NULL, 0,
1193 PACKET_READ_CHOMP_NEWLINE |
1194 PACKET_READ_DIE_ON_ERR_PACKET);
1195
1196 receive_needs(&data, &reader);
1197 if (data.want_obj.nr) {
1198 get_common_commits(&data, &reader);
1199 create_pack_file(&data);
1200 }
1201 }
1202
1203 upload_pack_data_clear(&data);
1204 }
1205
1206 static int parse_want(struct packet_writer *writer, const char *line,
1207 struct object_array *want_obj)
1208 {
1209 const char *arg;
1210 if (skip_prefix(line, "want ", &arg)) {
1211 struct object_id oid;
1212 struct object *o;
1213
1214 if (get_oid_hex(arg, &oid))
1215 die("git upload-pack: protocol error, "
1216 "expected to get oid, not '%s'", line);
1217
1218 o = parse_object(the_repository, &oid);
1219 if (!o) {
1220 packet_writer_error(writer,
1221 "upload-pack: not our ref %s",
1222 oid_to_hex(&oid));
1223 die("git upload-pack: not our ref %s",
1224 oid_to_hex(&oid));
1225 }
1226
1227 if (!(o->flags & WANTED)) {
1228 o->flags |= WANTED;
1229 add_object_array(o, NULL, want_obj);
1230 }
1231
1232 return 1;
1233 }
1234
1235 return 0;
1236 }
1237
1238 static int parse_want_ref(struct packet_writer *writer, const char *line,
1239 struct string_list *wanted_refs,
1240 struct object_array *want_obj)
1241 {
1242 const char *arg;
1243 if (skip_prefix(line, "want-ref ", &arg)) {
1244 struct object_id oid;
1245 struct string_list_item *item;
1246 struct object *o;
1247
1248 if (read_ref(arg, &oid)) {
1249 packet_writer_error(writer, "unknown ref %s", arg);
1250 die("unknown ref %s", arg);
1251 }
1252
1253 item = string_list_append(wanted_refs, arg);
1254 item->util = oiddup(&oid);
1255
1256 o = parse_object_or_die(&oid, arg);
1257 if (!(o->flags & WANTED)) {
1258 o->flags |= WANTED;
1259 add_object_array(o, NULL, want_obj);
1260 }
1261
1262 return 1;
1263 }
1264
1265 return 0;
1266 }
1267
1268 static int parse_have(const char *line, struct oid_array *haves)
1269 {
1270 const char *arg;
1271 if (skip_prefix(line, "have ", &arg)) {
1272 struct object_id oid;
1273
1274 if (get_oid_hex(arg, &oid))
1275 die("git upload-pack: expected SHA1 object, got '%s'", arg);
1276 oid_array_append(haves, &oid);
1277 return 1;
1278 }
1279
1280 return 0;
1281 }
1282
1283 static void process_args(struct packet_reader *request,
1284 struct upload_pack_data *data)
1285 {
1286 while (packet_reader_read(request) == PACKET_READ_NORMAL) {
1287 const char *arg = request->line;
1288 const char *p;
1289
1290 /* process want */
1291 if (parse_want(&data->writer, arg, &data->want_obj))
1292 continue;
1293 if (data->allow_ref_in_want &&
1294 parse_want_ref(&data->writer, arg, &data->wanted_refs,
1295 &data->want_obj))
1296 continue;
1297 /* process have line */
1298 if (parse_have(arg, &data->haves))
1299 continue;
1300
1301 /* process args like thin-pack */
1302 if (!strcmp(arg, "thin-pack")) {
1303 data->use_thin_pack = 1;
1304 continue;
1305 }
1306 if (!strcmp(arg, "ofs-delta")) {
1307 data->use_ofs_delta = 1;
1308 continue;
1309 }
1310 if (!strcmp(arg, "no-progress")) {
1311 data->no_progress = 1;
1312 continue;
1313 }
1314 if (!strcmp(arg, "include-tag")) {
1315 data->use_include_tag = 1;
1316 continue;
1317 }
1318 if (!strcmp(arg, "done")) {
1319 data->done = 1;
1320 continue;
1321 }
1322
1323 /* Shallow related arguments */
1324 if (process_shallow(arg, &data->shallows))
1325 continue;
1326 if (process_deepen(arg, &data->depth))
1327 continue;
1328 if (process_deepen_since(arg, &data->deepen_since,
1329 &data->deepen_rev_list))
1330 continue;
1331 if (process_deepen_not(arg, &data->deepen_not,
1332 &data->deepen_rev_list))
1333 continue;
1334 if (!strcmp(arg, "deepen-relative")) {
1335 data->deepen_relative = 1;
1336 continue;
1337 }
1338
1339 if (data->allow_filter && skip_prefix(arg, "filter ", &p)) {
1340 list_objects_filter_die_if_populated(&data->filter_options);
1341 parse_list_objects_filter(&data->filter_options, p);
1342 continue;
1343 }
1344
1345 if ((git_env_bool("GIT_TEST_SIDEBAND_ALL", 0) ||
1346 data->allow_sideband_all) &&
1347 !strcmp(arg, "sideband-all")) {
1348 data->writer.use_sideband = 1;
1349 continue;
1350 }
1351
1352 /* ignore unknown lines maybe? */
1353 die("unexpected line: '%s'", arg);
1354 }
1355
1356 if (request->status != PACKET_READ_FLUSH)
1357 die(_("expected flush after fetch arguments"));
1358 }
1359
1360 static int process_haves(struct oid_array *haves, struct oid_array *common,
1361 struct object_array *have_obj)
1362 {
1363 int i;
1364
1365 /* Process haves */
1366 for (i = 0; i < haves->nr; i++) {
1367 const struct object_id *oid = &haves->oid[i];
1368 struct object *o;
1369 int we_knew_they_have = 0;
1370
1371 if (!has_object_file(oid))
1372 continue;
1373
1374 oid_array_append(common, oid);
1375
1376 o = parse_object(the_repository, oid);
1377 if (!o)
1378 die("oops (%s)", oid_to_hex(oid));
1379 if (o->type == OBJ_COMMIT) {
1380 struct commit_list *parents;
1381 struct commit *commit = (struct commit *)o;
1382 if (o->flags & THEY_HAVE)
1383 we_knew_they_have = 1;
1384 else
1385 o->flags |= THEY_HAVE;
1386 if (!oldest_have || (commit->date < oldest_have))
1387 oldest_have = commit->date;
1388 for (parents = commit->parents;
1389 parents;
1390 parents = parents->next)
1391 parents->item->object.flags |= THEY_HAVE;
1392 }
1393 if (!we_knew_they_have)
1394 add_object_array(o, NULL, have_obj);
1395 }
1396
1397 return 0;
1398 }
1399
1400 static int send_acks(struct packet_writer *writer, struct oid_array *acks,
1401 const struct object_array *have_obj,
1402 struct object_array *want_obj)
1403 {
1404 int i;
1405
1406 packet_writer_write(writer, "acknowledgments\n");
1407
1408 /* Send Acks */
1409 if (!acks->nr)
1410 packet_writer_write(writer, "NAK\n");
1411
1412 for (i = 0; i < acks->nr; i++) {
1413 packet_writer_write(writer, "ACK %s\n",
1414 oid_to_hex(&acks->oid[i]));
1415 }
1416
1417 if (ok_to_give_up(have_obj, want_obj)) {
1418 /* Send Ready */
1419 packet_writer_write(writer, "ready\n");
1420 return 1;
1421 }
1422
1423 return 0;
1424 }
1425
1426 static int process_haves_and_send_acks(struct upload_pack_data *data)
1427 {
1428 struct oid_array common = OID_ARRAY_INIT;
1429 int ret = 0;
1430
1431 process_haves(&data->haves, &common, &data->have_obj);
1432 if (data->done) {
1433 ret = 1;
1434 } else if (send_acks(&data->writer, &common,
1435 &data->have_obj, &data->want_obj)) {
1436 packet_writer_delim(&data->writer);
1437 ret = 1;
1438 } else {
1439 /* Add Flush */
1440 packet_writer_flush(&data->writer);
1441 ret = 0;
1442 }
1443
1444 oid_array_clear(&data->haves);
1445 oid_array_clear(&common);
1446 return ret;
1447 }
1448
1449 static void send_wanted_ref_info(struct upload_pack_data *data)
1450 {
1451 const struct string_list_item *item;
1452
1453 if (!data->wanted_refs.nr)
1454 return;
1455
1456 packet_writer_write(&data->writer, "wanted-refs\n");
1457
1458 for_each_string_list_item(item, &data->wanted_refs) {
1459 packet_writer_write(&data->writer, "%s %s\n",
1460 oid_to_hex(item->util),
1461 item->string);
1462 }
1463
1464 packet_writer_delim(&data->writer);
1465 }
1466
1467 static void send_shallow_info(struct upload_pack_data *data)
1468 {
1469 /* No shallow info needs to be sent */
1470 if (!data->depth && !data->deepen_rev_list && !data->shallows.nr &&
1471 !is_repository_shallow(the_repository))
1472 return;
1473
1474 packet_writer_write(&data->writer, "shallow-info\n");
1475
1476 if (!send_shallow_list(&data->writer, data->depth,
1477 data->deepen_rev_list,
1478 data->deepen_since, &data->deepen_not,
1479 data->deepen_relative,
1480 &data->shallows, &data->want_obj) &&
1481 is_repository_shallow(the_repository))
1482 deepen(&data->writer, INFINITE_DEPTH, data->deepen_relative,
1483 &data->shallows, &data->want_obj);
1484
1485 packet_delim(1);
1486 }
1487
1488 enum fetch_state {
1489 FETCH_PROCESS_ARGS = 0,
1490 FETCH_SEND_ACKS,
1491 FETCH_SEND_PACK,
1492 FETCH_DONE,
1493 };
1494
1495 int upload_pack_v2(struct repository *r, struct argv_array *keys,
1496 struct packet_reader *request)
1497 {
1498 enum fetch_state state = FETCH_PROCESS_ARGS;
1499 struct upload_pack_data data;
1500
1501 clear_object_flags(ALL_FLAGS);
1502
1503 upload_pack_data_init(&data);
1504 data.use_sideband = LARGE_PACKET_MAX;
1505
1506 git_config(upload_pack_config, &data);
1507
1508 while (state != FETCH_DONE) {
1509 switch (state) {
1510 case FETCH_PROCESS_ARGS:
1511 process_args(request, &data);
1512
1513 if (!data.want_obj.nr) {
1514 /*
1515 * Request didn't contain any 'want' lines,
1516 * guess they didn't want anything.
1517 */
1518 state = FETCH_DONE;
1519 } else if (data.haves.nr) {
1520 /*
1521 * Request had 'have' lines, so lets ACK them.
1522 */
1523 state = FETCH_SEND_ACKS;
1524 } else {
1525 /*
1526 * Request had 'want's but no 'have's so we can
1527 * immedietly go to construct and send a pack.
1528 */
1529 state = FETCH_SEND_PACK;
1530 }
1531 break;
1532 case FETCH_SEND_ACKS:
1533 if (process_haves_and_send_acks(&data))
1534 state = FETCH_SEND_PACK;
1535 else
1536 state = FETCH_DONE;
1537 break;
1538 case FETCH_SEND_PACK:
1539 send_wanted_ref_info(&data);
1540 send_shallow_info(&data);
1541
1542 packet_writer_write(&data.writer, "packfile\n");
1543 create_pack_file(&data);
1544 state = FETCH_DONE;
1545 break;
1546 case FETCH_DONE:
1547 continue;
1548 }
1549 }
1550
1551 upload_pack_data_clear(&data);
1552 return 0;
1553 }
1554
1555 int upload_pack_advertise(struct repository *r,
1556 struct strbuf *value)
1557 {
1558 if (value) {
1559 int allow_filter_value;
1560 int allow_ref_in_want;
1561 int allow_sideband_all_value;
1562
1563 strbuf_addstr(value, "shallow");
1564
1565 if (!repo_config_get_bool(the_repository,
1566 "uploadpack.allowfilter",
1567 &allow_filter_value) &&
1568 allow_filter_value)
1569 strbuf_addstr(value, " filter");
1570
1571 if (!repo_config_get_bool(the_repository,
1572 "uploadpack.allowrefinwant",
1573 &allow_ref_in_want) &&
1574 allow_ref_in_want)
1575 strbuf_addstr(value, " ref-in-want");
1576
1577 if (git_env_bool("GIT_TEST_SIDEBAND_ALL", 0) ||
1578 (!repo_config_get_bool(the_repository,
1579 "uploadpack.allowsidebandall",
1580 &allow_sideband_all_value) &&
1581 allow_sideband_all_value))
1582 strbuf_addstr(value, " sideband-all");
1583 }
1584
1585 return 1;
1586 }
The Best VCS Around