upload-pack: move shallow_nr to upload_pack_data
[git] / upload-pack.c
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 struct object_array extra_edge_obj;
55
56 /*
57  * Please annotate, and if possible group together, fields used only
58  * for protocol v0 or only for protocol v2.
59  */
60 struct upload_pack_data {
61         struct string_list symref;                              /* v0 only */
62         struct object_array want_obj;
63         struct object_array have_obj;
64         struct oid_array haves;                                 /* v2 only */
65         struct string_list wanted_refs;                         /* v2 only */
66
67         struct object_array shallows;
68         struct string_list deepen_not;
69         int depth;
70         timestamp_t deepen_since;
71         int deepen_rev_list;
72         int deepen_relative;
73         int keepalive;
74         int shallow_nr;
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 (pack_data->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 (pack_data->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 (pack_data->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 upload_pack_data *data,
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(&data->writer, "shallow %s",
710                                             oid_to_hex(&object->oid));
711                         register_shallow(the_repository, &object->oid);
712                         data->shallow_nr++;
713                 }
714                 result = result->next;
715         }
716 }
717
718 static void send_unshallow(struct upload_pack_data *data)
719 {
720         int i;
721
722         for (i = 0; i < data->shallows.nr; i++) {
723                 struct object *object = data->shallows.objects[i].item;
724                 if (object->flags & NOT_SHALLOW) {
725                         struct commit_list *parents;
726                         packet_writer_write(&data->writer, "unshallow %s",
727                                             oid_to_hex(&object->oid));
728                         object->flags &= ~CLIENT_SHALLOW;
729                         /*
730                          * We want to _register_ "object" as shallow, but we
731                          * also need to traverse object's parents to deepen a
732                          * shallow clone. Unregister it for now so we can
733                          * parse and add the parents to the want list, then
734                          * re-register it.
735                          */
736                         unregister_shallow(&object->oid);
737                         object->parsed = 0;
738                         parse_commit_or_die((struct commit *)object);
739                         parents = ((struct commit *)object)->parents;
740                         while (parents) {
741                                 add_object_array(&parents->item->object,
742                                                  NULL, &data->want_obj);
743                                 parents = parents->next;
744                         }
745                         add_object_array(object, NULL, &extra_edge_obj);
746                 }
747                 /* make sure commit traversal conforms to client */
748                 register_shallow(the_repository, &object->oid);
749         }
750 }
751
752 static int check_ref(const char *refname_full, const struct object_id *oid,
753                      int flag, void *cb_data);
754 static void deepen(struct upload_pack_data *data, int depth)
755 {
756         if (depth == INFINITE_DEPTH && !is_repository_shallow(the_repository)) {
757                 int i;
758
759                 for (i = 0; i < data->shallows.nr; i++) {
760                         struct object *object = data->shallows.objects[i].item;
761                         object->flags |= NOT_SHALLOW;
762                 }
763         } else if (data->deepen_relative) {
764                 struct object_array reachable_shallows = OBJECT_ARRAY_INIT;
765                 struct commit_list *result;
766
767                 /*
768                  * Checking for reachable shallows requires that our refs be
769                  * marked with OUR_REF.
770                  */
771                 head_ref_namespaced(check_ref, NULL);
772                 for_each_namespaced_ref(check_ref, NULL);
773
774                 get_reachable_list(&data->shallows, &reachable_shallows);
775                 result = get_shallow_commits(&reachable_shallows,
776                                              depth + 1,
777                                              SHALLOW, NOT_SHALLOW);
778                 send_shallow(data, result);
779                 free_commit_list(result);
780                 object_array_clear(&reachable_shallows);
781         } else {
782                 struct commit_list *result;
783
784                 result = get_shallow_commits(&data->want_obj, depth,
785                                              SHALLOW, NOT_SHALLOW);
786                 send_shallow(data, result);
787                 free_commit_list(result);
788         }
789
790         send_unshallow(data);
791 }
792
793 static void deepen_by_rev_list(struct upload_pack_data *data,
794                                int ac,
795                                const char **av)
796 {
797         struct commit_list *result;
798
799         disable_commit_graph(the_repository);
800         result = get_shallow_commits_by_rev_list(ac, av, SHALLOW, NOT_SHALLOW);
801         send_shallow(data, result);
802         free_commit_list(result);
803         send_unshallow(data);
804 }
805
806 /* Returns 1 if a shallow list is sent or 0 otherwise */
807 static int send_shallow_list(struct upload_pack_data *data)
808 {
809         int ret = 0;
810
811         if (data->depth > 0 && data->deepen_rev_list)
812                 die("git upload-pack: deepen and deepen-since (or deepen-not) cannot be used together");
813         if (data->depth > 0) {
814                 deepen(data, data->depth);
815                 ret = 1;
816         } else if (data->deepen_rev_list) {
817                 struct argv_array av = ARGV_ARRAY_INIT;
818                 int i;
819
820                 argv_array_push(&av, "rev-list");
821                 if (data->deepen_since)
822                         argv_array_pushf(&av, "--max-age=%"PRItime, data->deepen_since);
823                 if (data->deepen_not.nr) {
824                         argv_array_push(&av, "--not");
825                         for (i = 0; i < data->deepen_not.nr; i++) {
826                                 struct string_list_item *s = data->deepen_not.items + i;
827                                 argv_array_push(&av, s->string);
828                         }
829                         argv_array_push(&av, "--not");
830                 }
831                 for (i = 0; i < data->want_obj.nr; i++) {
832                         struct object *o = data->want_obj.objects[i].item;
833                         argv_array_push(&av, oid_to_hex(&o->oid));
834                 }
835                 deepen_by_rev_list(data, av.argc, av.argv);
836                 argv_array_clear(&av);
837                 ret = 1;
838         } else {
839                 if (data->shallows.nr > 0) {
840                         int i;
841                         for (i = 0; i < data->shallows.nr; i++)
842                                 register_shallow(the_repository,
843                                                  &data->shallows.objects[i].item->oid);
844                 }
845         }
846
847         data->shallow_nr += data->shallows.nr;
848         return ret;
849 }
850
851 static int process_shallow(const char *line, struct object_array *shallows)
852 {
853         const char *arg;
854         if (skip_prefix(line, "shallow ", &arg)) {
855                 struct object_id oid;
856                 struct object *object;
857                 if (get_oid_hex(arg, &oid))
858                         die("invalid shallow line: %s", line);
859                 object = parse_object(the_repository, &oid);
860                 if (!object)
861                         return 1;
862                 if (object->type != OBJ_COMMIT)
863                         die("invalid shallow object %s", oid_to_hex(&oid));
864                 if (!(object->flags & CLIENT_SHALLOW)) {
865                         object->flags |= CLIENT_SHALLOW;
866                         add_object_array(object, NULL, shallows);
867                 }
868                 return 1;
869         }
870
871         return 0;
872 }
873
874 static int process_deepen(const char *line, int *depth)
875 {
876         const char *arg;
877         if (skip_prefix(line, "deepen ", &arg)) {
878                 char *end = NULL;
879                 *depth = (int)strtol(arg, &end, 0);
880                 if (!end || *end || *depth <= 0)
881                         die("Invalid deepen: %s", line);
882                 return 1;
883         }
884
885         return 0;
886 }
887
888 static int process_deepen_since(const char *line, timestamp_t *deepen_since, int *deepen_rev_list)
889 {
890         const char *arg;
891         if (skip_prefix(line, "deepen-since ", &arg)) {
892                 char *end = NULL;
893                 *deepen_since = parse_timestamp(arg, &end, 0);
894                 if (!end || *end || !deepen_since ||
895                     /* revisions.c's max_age -1 is special */
896                     *deepen_since == -1)
897                         die("Invalid deepen-since: %s", line);
898                 *deepen_rev_list = 1;
899                 return 1;
900         }
901         return 0;
902 }
903
904 static int process_deepen_not(const char *line, struct string_list *deepen_not, int *deepen_rev_list)
905 {
906         const char *arg;
907         if (skip_prefix(line, "deepen-not ", &arg)) {
908                 char *ref = NULL;
909                 struct object_id oid;
910                 if (expand_ref(the_repository, arg, strlen(arg), &oid, &ref) != 1)
911                         die("git upload-pack: ambiguous deepen-not: %s", line);
912                 string_list_append(deepen_not, ref);
913                 free(ref);
914                 *deepen_rev_list = 1;
915                 return 1;
916         }
917         return 0;
918 }
919
920 static void receive_needs(struct upload_pack_data *data,
921                           struct packet_reader *reader)
922 {
923         int has_non_tip = 0;
924
925         data->shallow_nr = 0;
926         for (;;) {
927                 struct object *o;
928                 const char *features;
929                 struct object_id oid_buf;
930                 const char *arg;
931
932                 reset_timeout(data->timeout);
933                 if (packet_reader_read(reader) != PACKET_READ_NORMAL)
934                         break;
935
936                 if (process_shallow(reader->line, &data->shallows))
937                         continue;
938                 if (process_deepen(reader->line, &data->depth))
939                         continue;
940                 if (process_deepen_since(reader->line, &data->deepen_since, &data->deepen_rev_list))
941                         continue;
942                 if (process_deepen_not(reader->line, &data->deepen_not, &data->deepen_rev_list))
943                         continue;
944
945                 if (skip_prefix(reader->line, "filter ", &arg)) {
946                         if (!data->filter_capability_requested)
947                                 die("git upload-pack: filtering capability not negotiated");
948                         list_objects_filter_die_if_populated(&data->filter_options);
949                         parse_list_objects_filter(&data->filter_options, arg);
950                         continue;
951                 }
952
953                 if (!skip_prefix(reader->line, "want ", &arg) ||
954                     parse_oid_hex(arg, &oid_buf, &features))
955                         die("git upload-pack: protocol error, "
956                             "expected to get object ID, not '%s'", reader->line);
957
958                 if (parse_feature_request(features, "deepen-relative"))
959                         data->deepen_relative = 1;
960                 if (parse_feature_request(features, "multi_ack_detailed"))
961                         data->multi_ack = MULTI_ACK_DETAILED;
962                 else if (parse_feature_request(features, "multi_ack"))
963                         data->multi_ack = MULTI_ACK;
964                 if (parse_feature_request(features, "no-done"))
965                         data->no_done = 1;
966                 if (parse_feature_request(features, "thin-pack"))
967                         data->use_thin_pack = 1;
968                 if (parse_feature_request(features, "ofs-delta"))
969                         data->use_ofs_delta = 1;
970                 if (parse_feature_request(features, "side-band-64k"))
971                         data->use_sideband = LARGE_PACKET_MAX;
972                 else if (parse_feature_request(features, "side-band"))
973                         data->use_sideband = DEFAULT_PACKET_MAX;
974                 if (parse_feature_request(features, "no-progress"))
975                         data->no_progress = 1;
976                 if (parse_feature_request(features, "include-tag"))
977                         data->use_include_tag = 1;
978                 if (data->allow_filter &&
979                     parse_feature_request(features, "filter"))
980                         data->filter_capability_requested = 1;
981
982                 o = parse_object(the_repository, &oid_buf);
983                 if (!o) {
984                         packet_writer_error(&data->writer,
985                                             "upload-pack: not our ref %s",
986                                             oid_to_hex(&oid_buf));
987                         die("git upload-pack: not our ref %s",
988                             oid_to_hex(&oid_buf));
989                 }
990                 if (!(o->flags & WANTED)) {
991                         o->flags |= WANTED;
992                         if (!((allow_unadvertised_object_request & ALLOW_ANY_SHA1) == ALLOW_ANY_SHA1
993                               || is_our_ref(o)))
994                                 has_non_tip = 1;
995                         add_object_array(o, NULL, &data->want_obj);
996                 }
997         }
998
999         /*
1000          * We have sent all our refs already, and the other end
1001          * should have chosen out of them. When we are operating
1002          * in the stateless RPC mode, however, their choice may
1003          * have been based on the set of older refs advertised
1004          * by another process that handled the initial request.
1005          */
1006         if (has_non_tip)
1007                 check_non_tip(data);
1008
1009         if (!data->use_sideband && data->daemon_mode)
1010                 data->no_progress = 1;
1011
1012         if (data->depth == 0 && !data->deepen_rev_list && data->shallows.nr == 0)
1013                 return;
1014
1015         if (send_shallow_list(data))
1016                 packet_flush(1);
1017 }
1018
1019 /* return non-zero if the ref is hidden, otherwise 0 */
1020 static int mark_our_ref(const char *refname, const char *refname_full,
1021                         const struct object_id *oid)
1022 {
1023         struct object *o = lookup_unknown_object(oid);
1024
1025         if (ref_is_hidden(refname, refname_full)) {
1026                 o->flags |= HIDDEN_REF;
1027                 return 1;
1028         }
1029         o->flags |= OUR_REF;
1030         return 0;
1031 }
1032
1033 static int check_ref(const char *refname_full, const struct object_id *oid,
1034                      int flag, void *cb_data)
1035 {
1036         const char *refname = strip_namespace(refname_full);
1037
1038         mark_our_ref(refname, refname_full, oid);
1039         return 0;
1040 }
1041
1042 static void format_symref_info(struct strbuf *buf, struct string_list *symref)
1043 {
1044         struct string_list_item *item;
1045
1046         if (!symref->nr)
1047                 return;
1048         for_each_string_list_item(item, symref)
1049                 strbuf_addf(buf, " symref=%s:%s", item->string, (char *)item->util);
1050 }
1051
1052 static int send_ref(const char *refname, const struct object_id *oid,
1053                     int flag, void *cb_data)
1054 {
1055         static const char *capabilities = "multi_ack thin-pack side-band"
1056                 " side-band-64k ofs-delta shallow deepen-since deepen-not"
1057                 " deepen-relative no-progress include-tag multi_ack_detailed";
1058         const char *refname_nons = strip_namespace(refname);
1059         struct object_id peeled;
1060         struct upload_pack_data *data = cb_data;
1061
1062         if (mark_our_ref(refname_nons, refname, oid))
1063                 return 0;
1064
1065         if (capabilities) {
1066                 struct strbuf symref_info = STRBUF_INIT;
1067
1068                 format_symref_info(&symref_info, &data->symref);
1069                 packet_write_fmt(1, "%s %s%c%s%s%s%s%s%s agent=%s\n",
1070                              oid_to_hex(oid), refname_nons,
1071                              0, capabilities,
1072                              (allow_unadvertised_object_request & ALLOW_TIP_SHA1) ?
1073                                      " allow-tip-sha1-in-want" : "",
1074                              (allow_unadvertised_object_request & ALLOW_REACHABLE_SHA1) ?
1075                                      " allow-reachable-sha1-in-want" : "",
1076                              data->stateless_rpc ? " no-done" : "",
1077                              symref_info.buf,
1078                              data->allow_filter ? " filter" : "",
1079                              git_user_agent_sanitized());
1080                 strbuf_release(&symref_info);
1081         } else {
1082                 packet_write_fmt(1, "%s %s\n", oid_to_hex(oid), refname_nons);
1083         }
1084         capabilities = NULL;
1085         if (!peel_ref(refname, &peeled))
1086                 packet_write_fmt(1, "%s %s^{}\n", oid_to_hex(&peeled), refname_nons);
1087         return 0;
1088 }
1089
1090 static int find_symref(const char *refname, const struct object_id *oid,
1091                        int flag, void *cb_data)
1092 {
1093         const char *symref_target;
1094         struct string_list_item *item;
1095
1096         if ((flag & REF_ISSYMREF) == 0)
1097                 return 0;
1098         symref_target = resolve_ref_unsafe(refname, 0, NULL, &flag);
1099         if (!symref_target || (flag & REF_ISSYMREF) == 0)
1100                 die("'%s' is a symref but it is not?", refname);
1101         item = string_list_append(cb_data, strip_namespace(refname));
1102         item->util = xstrdup(strip_namespace(symref_target));
1103         return 0;
1104 }
1105
1106 static int upload_pack_config(const char *var, const char *value, void *cb_data)
1107 {
1108         struct upload_pack_data *data = cb_data;
1109
1110         if (!strcmp("uploadpack.allowtipsha1inwant", var)) {
1111                 if (git_config_bool(var, value))
1112                         allow_unadvertised_object_request |= ALLOW_TIP_SHA1;
1113                 else
1114                         allow_unadvertised_object_request &= ~ALLOW_TIP_SHA1;
1115         } else if (!strcmp("uploadpack.allowreachablesha1inwant", var)) {
1116                 if (git_config_bool(var, value))
1117                         allow_unadvertised_object_request |= ALLOW_REACHABLE_SHA1;
1118                 else
1119                         allow_unadvertised_object_request &= ~ALLOW_REACHABLE_SHA1;
1120         } else if (!strcmp("uploadpack.allowanysha1inwant", var)) {
1121                 if (git_config_bool(var, value))
1122                         allow_unadvertised_object_request |= ALLOW_ANY_SHA1;
1123                 else
1124                         allow_unadvertised_object_request &= ~ALLOW_ANY_SHA1;
1125         } else if (!strcmp("uploadpack.keepalive", var)) {
1126                 data->keepalive = git_config_int(var, value);
1127                 if (!data->keepalive)
1128                         data->keepalive = -1;
1129         } else if (!strcmp("uploadpack.allowfilter", var)) {
1130                 data->allow_filter = git_config_bool(var, value);
1131         } else if (!strcmp("uploadpack.allowrefinwant", var)) {
1132                 data->allow_ref_in_want = git_config_bool(var, value);
1133         } else if (!strcmp("uploadpack.allowsidebandall", var)) {
1134                 data->allow_sideband_all = git_config_bool(var, value);
1135         } else if (!strcmp("core.precomposeunicode", var)) {
1136                 precomposed_unicode = git_config_bool(var, value);
1137         }
1138
1139         if (current_config_scope() != CONFIG_SCOPE_LOCAL &&
1140         current_config_scope() != CONFIG_SCOPE_WORKTREE) {
1141                 if (!strcmp("uploadpack.packobjectshook", var))
1142                         return git_config_string(&data->pack_objects_hook, var, value);
1143         }
1144
1145         return parse_hide_refs_config(var, value, "uploadpack");
1146 }
1147
1148 void upload_pack(struct upload_pack_options *options)
1149 {
1150         struct packet_reader reader;
1151         struct upload_pack_data data;
1152
1153         upload_pack_data_init(&data);
1154
1155         git_config(upload_pack_config, &data);
1156
1157         data.stateless_rpc = options->stateless_rpc;
1158         data.daemon_mode = options->daemon_mode;
1159         data.timeout = options->timeout;
1160
1161         head_ref_namespaced(find_symref, &data.symref);
1162
1163         if (options->advertise_refs || !data.stateless_rpc) {
1164                 reset_timeout(data.timeout);
1165                 head_ref_namespaced(send_ref, &data);
1166                 for_each_namespaced_ref(send_ref, &data);
1167                 advertise_shallow_grafts(1);
1168                 packet_flush(1);
1169         } else {
1170                 head_ref_namespaced(check_ref, NULL);
1171                 for_each_namespaced_ref(check_ref, NULL);
1172         }
1173
1174         if (!options->advertise_refs) {
1175                 packet_reader_init(&reader, 0, NULL, 0,
1176                                    PACKET_READ_CHOMP_NEWLINE |
1177                                    PACKET_READ_DIE_ON_ERR_PACKET);
1178
1179                 receive_needs(&data, &reader);
1180                 if (data.want_obj.nr) {
1181                         get_common_commits(&data, &reader);
1182                         create_pack_file(&data);
1183                 }
1184         }
1185
1186         upload_pack_data_clear(&data);
1187 }
1188
1189 static int parse_want(struct packet_writer *writer, const char *line,
1190                       struct object_array *want_obj)
1191 {
1192         const char *arg;
1193         if (skip_prefix(line, "want ", &arg)) {
1194                 struct object_id oid;
1195                 struct object *o;
1196
1197                 if (get_oid_hex(arg, &oid))
1198                         die("git upload-pack: protocol error, "
1199                             "expected to get oid, not '%s'", line);
1200
1201                 o = parse_object(the_repository, &oid);
1202                 if (!o) {
1203                         packet_writer_error(writer,
1204                                             "upload-pack: not our ref %s",
1205                                             oid_to_hex(&oid));
1206                         die("git upload-pack: not our ref %s",
1207                             oid_to_hex(&oid));
1208                 }
1209
1210                 if (!(o->flags & WANTED)) {
1211                         o->flags |= WANTED;
1212                         add_object_array(o, NULL, want_obj);
1213                 }
1214
1215                 return 1;
1216         }
1217
1218         return 0;
1219 }
1220
1221 static int parse_want_ref(struct packet_writer *writer, const char *line,
1222                           struct string_list *wanted_refs,
1223                           struct object_array *want_obj)
1224 {
1225         const char *arg;
1226         if (skip_prefix(line, "want-ref ", &arg)) {
1227                 struct object_id oid;
1228                 struct string_list_item *item;
1229                 struct object *o;
1230
1231                 if (read_ref(arg, &oid)) {
1232                         packet_writer_error(writer, "unknown ref %s", arg);
1233                         die("unknown ref %s", arg);
1234                 }
1235
1236                 item = string_list_append(wanted_refs, arg);
1237                 item->util = oiddup(&oid);
1238
1239                 o = parse_object_or_die(&oid, arg);
1240                 if (!(o->flags & WANTED)) {
1241                         o->flags |= WANTED;
1242                         add_object_array(o, NULL, want_obj);
1243                 }
1244
1245                 return 1;
1246         }
1247
1248         return 0;
1249 }
1250
1251 static int parse_have(const char *line, struct oid_array *haves)
1252 {
1253         const char *arg;
1254         if (skip_prefix(line, "have ", &arg)) {
1255                 struct object_id oid;
1256
1257                 if (get_oid_hex(arg, &oid))
1258                         die("git upload-pack: expected SHA1 object, got '%s'", arg);
1259                 oid_array_append(haves, &oid);
1260                 return 1;
1261         }
1262
1263         return 0;
1264 }
1265
1266 static void process_args(struct packet_reader *request,
1267                          struct upload_pack_data *data)
1268 {
1269         while (packet_reader_read(request) == PACKET_READ_NORMAL) {
1270                 const char *arg = request->line;
1271                 const char *p;
1272
1273                 /* process want */
1274                 if (parse_want(&data->writer, arg, &data->want_obj))
1275                         continue;
1276                 if (data->allow_ref_in_want &&
1277                     parse_want_ref(&data->writer, arg, &data->wanted_refs,
1278                                    &data->want_obj))
1279                         continue;
1280                 /* process have line */
1281                 if (parse_have(arg, &data->haves))
1282                         continue;
1283
1284                 /* process args like thin-pack */
1285                 if (!strcmp(arg, "thin-pack")) {
1286                         data->use_thin_pack = 1;
1287                         continue;
1288                 }
1289                 if (!strcmp(arg, "ofs-delta")) {
1290                         data->use_ofs_delta = 1;
1291                         continue;
1292                 }
1293                 if (!strcmp(arg, "no-progress")) {
1294                         data->no_progress = 1;
1295                         continue;
1296                 }
1297                 if (!strcmp(arg, "include-tag")) {
1298                         data->use_include_tag = 1;
1299                         continue;
1300                 }
1301                 if (!strcmp(arg, "done")) {
1302                         data->done = 1;
1303                         continue;
1304                 }
1305
1306                 /* Shallow related arguments */
1307                 if (process_shallow(arg, &data->shallows))
1308                         continue;
1309                 if (process_deepen(arg, &data->depth))
1310                         continue;
1311                 if (process_deepen_since(arg, &data->deepen_since,
1312                                          &data->deepen_rev_list))
1313                         continue;
1314                 if (process_deepen_not(arg, &data->deepen_not,
1315                                        &data->deepen_rev_list))
1316                         continue;
1317                 if (!strcmp(arg, "deepen-relative")) {
1318                         data->deepen_relative = 1;
1319                         continue;
1320                 }
1321
1322                 if (data->allow_filter && skip_prefix(arg, "filter ", &p)) {
1323                         list_objects_filter_die_if_populated(&data->filter_options);
1324                         parse_list_objects_filter(&data->filter_options, p);
1325                         continue;
1326                 }
1327
1328                 if ((git_env_bool("GIT_TEST_SIDEBAND_ALL", 0) ||
1329                      data->allow_sideband_all) &&
1330                     !strcmp(arg, "sideband-all")) {
1331                         data->writer.use_sideband = 1;
1332                         continue;
1333                 }
1334
1335                 /* ignore unknown lines maybe? */
1336                 die("unexpected line: '%s'", arg);
1337         }
1338
1339         if (request->status != PACKET_READ_FLUSH)
1340                 die(_("expected flush after fetch arguments"));
1341 }
1342
1343 static int process_haves(struct oid_array *haves, struct oid_array *common,
1344                          struct object_array *have_obj)
1345 {
1346         int i;
1347
1348         /* Process haves */
1349         for (i = 0; i < haves->nr; i++) {
1350                 const struct object_id *oid = &haves->oid[i];
1351                 struct object *o;
1352                 int we_knew_they_have = 0;
1353
1354                 if (!has_object_file(oid))
1355                         continue;
1356
1357                 oid_array_append(common, oid);
1358
1359                 o = parse_object(the_repository, oid);
1360                 if (!o)
1361                         die("oops (%s)", oid_to_hex(oid));
1362                 if (o->type == OBJ_COMMIT) {
1363                         struct commit_list *parents;
1364                         struct commit *commit = (struct commit *)o;
1365                         if (o->flags & THEY_HAVE)
1366                                 we_knew_they_have = 1;
1367                         else
1368                                 o->flags |= THEY_HAVE;
1369                         if (!oldest_have || (commit->date < oldest_have))
1370                                 oldest_have = commit->date;
1371                         for (parents = commit->parents;
1372                              parents;
1373                              parents = parents->next)
1374                                 parents->item->object.flags |= THEY_HAVE;
1375                 }
1376                 if (!we_knew_they_have)
1377                         add_object_array(o, NULL, have_obj);
1378         }
1379
1380         return 0;
1381 }
1382
1383 static int send_acks(struct packet_writer *writer, struct oid_array *acks,
1384                      const struct object_array *have_obj,
1385                      struct object_array *want_obj)
1386 {
1387         int i;
1388
1389         packet_writer_write(writer, "acknowledgments\n");
1390
1391         /* Send Acks */
1392         if (!acks->nr)
1393                 packet_writer_write(writer, "NAK\n");
1394
1395         for (i = 0; i < acks->nr; i++) {
1396                 packet_writer_write(writer, "ACK %s\n",
1397                                     oid_to_hex(&acks->oid[i]));
1398         }
1399
1400         if (ok_to_give_up(have_obj, want_obj)) {
1401                 /* Send Ready */
1402                 packet_writer_write(writer, "ready\n");
1403                 return 1;
1404         }
1405
1406         return 0;
1407 }
1408
1409 static int process_haves_and_send_acks(struct upload_pack_data *data)
1410 {
1411         struct oid_array common = OID_ARRAY_INIT;
1412         int ret = 0;
1413
1414         process_haves(&data->haves, &common, &data->have_obj);
1415         if (data->done) {
1416                 ret = 1;
1417         } else if (send_acks(&data->writer, &common,
1418                              &data->have_obj, &data->want_obj)) {
1419                 packet_writer_delim(&data->writer);
1420                 ret = 1;
1421         } else {
1422                 /* Add Flush */
1423                 packet_writer_flush(&data->writer);
1424                 ret = 0;
1425         }
1426
1427         oid_array_clear(&data->haves);
1428         oid_array_clear(&common);
1429         return ret;
1430 }
1431
1432 static void send_wanted_ref_info(struct upload_pack_data *data)
1433 {
1434         const struct string_list_item *item;
1435
1436         if (!data->wanted_refs.nr)
1437                 return;
1438
1439         packet_writer_write(&data->writer, "wanted-refs\n");
1440
1441         for_each_string_list_item(item, &data->wanted_refs) {
1442                 packet_writer_write(&data->writer, "%s %s\n",
1443                                     oid_to_hex(item->util),
1444                                     item->string);
1445         }
1446
1447         packet_writer_delim(&data->writer);
1448 }
1449
1450 static void send_shallow_info(struct upload_pack_data *data)
1451 {
1452         /* No shallow info needs to be sent */
1453         if (!data->depth && !data->deepen_rev_list && !data->shallows.nr &&
1454             !is_repository_shallow(the_repository))
1455                 return;
1456
1457         packet_writer_write(&data->writer, "shallow-info\n");
1458
1459         if (!send_shallow_list(data) &&
1460             is_repository_shallow(the_repository))
1461                 deepen(data, INFINITE_DEPTH);
1462
1463         packet_delim(1);
1464 }
1465
1466 enum fetch_state {
1467         FETCH_PROCESS_ARGS = 0,
1468         FETCH_SEND_ACKS,
1469         FETCH_SEND_PACK,
1470         FETCH_DONE,
1471 };
1472
1473 int upload_pack_v2(struct repository *r, struct argv_array *keys,
1474                    struct packet_reader *request)
1475 {
1476         enum fetch_state state = FETCH_PROCESS_ARGS;
1477         struct upload_pack_data data;
1478
1479         clear_object_flags(ALL_FLAGS);
1480
1481         upload_pack_data_init(&data);
1482         data.use_sideband = LARGE_PACKET_MAX;
1483
1484         git_config(upload_pack_config, &data);
1485
1486         while (state != FETCH_DONE) {
1487                 switch (state) {
1488                 case FETCH_PROCESS_ARGS:
1489                         process_args(request, &data);
1490
1491                         if (!data.want_obj.nr) {
1492                                 /*
1493                                  * Request didn't contain any 'want' lines,
1494                                  * guess they didn't want anything.
1495                                  */
1496                                 state = FETCH_DONE;
1497                         } else if (data.haves.nr) {
1498                                 /*
1499                                  * Request had 'have' lines, so lets ACK them.
1500                                  */
1501                                 state = FETCH_SEND_ACKS;
1502                         } else {
1503                                 /*
1504                                  * Request had 'want's but no 'have's so we can
1505                                  * immedietly go to construct and send a pack.
1506                                  */
1507                                 state = FETCH_SEND_PACK;
1508                         }
1509                         break;
1510                 case FETCH_SEND_ACKS:
1511                         if (process_haves_and_send_acks(&data))
1512                                 state = FETCH_SEND_PACK;
1513                         else
1514                                 state = FETCH_DONE;
1515                         break;
1516                 case FETCH_SEND_PACK:
1517                         send_wanted_ref_info(&data);
1518                         send_shallow_info(&data);
1519
1520                         packet_writer_write(&data.writer, "packfile\n");
1521                         create_pack_file(&data);
1522                         state = FETCH_DONE;
1523                         break;
1524                 case FETCH_DONE:
1525                         continue;
1526                 }
1527         }
1528
1529         upload_pack_data_clear(&data);
1530         return 0;
1531 }
1532
1533 int upload_pack_advertise(struct repository *r,
1534                           struct strbuf *value)
1535 {
1536         if (value) {
1537                 int allow_filter_value;
1538                 int allow_ref_in_want;
1539                 int allow_sideband_all_value;
1540
1541                 strbuf_addstr(value, "shallow");
1542
1543                 if (!repo_config_get_bool(the_repository,
1544                                          "uploadpack.allowfilter",
1545                                          &allow_filter_value) &&
1546                     allow_filter_value)
1547                         strbuf_addstr(value, " filter");
1548
1549                 if (!repo_config_get_bool(the_repository,
1550                                          "uploadpack.allowrefinwant",
1551                                          &allow_ref_in_want) &&
1552                     allow_ref_in_want)
1553                         strbuf_addstr(value, " ref-in-want");
1554
1555                 if (git_env_bool("GIT_TEST_SIDEBAND_ALL", 0) ||
1556                     (!repo_config_get_bool(the_repository,
1557                                            "uploadpack.allowsidebandall",
1558                                            &allow_sideband_all_value) &&
1559                      allow_sideband_all_value))
1560                         strbuf_addstr(value, " sideband-all");
1561         }
1562
1563         return 1;
1564 }