Merge branch 'mm/readme-markdown' into maint
[git] / builtin / receive-pack.c
1 #include "builtin.h"
2 #include "lockfile.h"
3 #include "pack.h"
4 #include "refs.h"
5 #include "pkt-line.h"
6 #include "sideband.h"
7 #include "run-command.h"
8 #include "exec_cmd.h"
9 #include "commit.h"
10 #include "object.h"
11 #include "remote.h"
12 #include "connect.h"
13 #include "transport.h"
14 #include "string-list.h"
15 #include "sha1-array.h"
16 #include "connected.h"
17 #include "argv-array.h"
18 #include "version.h"
19 #include "tag.h"
20 #include "gpg-interface.h"
21 #include "sigchain.h"
22 #include "fsck.h"
23
24 static const char receive_pack_usage[] = "git receive-pack <git-dir>";
25
26 enum deny_action {
27         DENY_UNCONFIGURED,
28         DENY_IGNORE,
29         DENY_WARN,
30         DENY_REFUSE,
31         DENY_UPDATE_INSTEAD
32 };
33
34 static int deny_deletes;
35 static int deny_non_fast_forwards;
36 static enum deny_action deny_current_branch = DENY_UNCONFIGURED;
37 static enum deny_action deny_delete_current = DENY_UNCONFIGURED;
38 static int receive_fsck_objects = -1;
39 static int transfer_fsck_objects = -1;
40 static struct strbuf fsck_msg_types = STRBUF_INIT;
41 static int receive_unpack_limit = -1;
42 static int transfer_unpack_limit = -1;
43 static int advertise_atomic_push = 1;
44 static int unpack_limit = 100;
45 static int report_status;
46 static int use_sideband;
47 static int use_atomic;
48 static int quiet;
49 static int prefer_ofs_delta = 1;
50 static int auto_update_server_info;
51 static int auto_gc = 1;
52 static int fix_thin = 1;
53 static int stateless_rpc;
54 static const char *service_dir;
55 static const char *head_name;
56 static void *head_name_to_free;
57 static int sent_capabilities;
58 static int shallow_update;
59 static const char *alt_shallow_file;
60 static struct strbuf push_cert = STRBUF_INIT;
61 static unsigned char push_cert_sha1[20];
62 static struct signature_check sigcheck;
63 static const char *push_cert_nonce;
64 static const char *cert_nonce_seed;
65
66 static const char *NONCE_UNSOLICITED = "UNSOLICITED";
67 static const char *NONCE_BAD = "BAD";
68 static const char *NONCE_MISSING = "MISSING";
69 static const char *NONCE_OK = "OK";
70 static const char *NONCE_SLOP = "SLOP";
71 static const char *nonce_status;
72 static long nonce_stamp_slop;
73 static unsigned long nonce_stamp_slop_limit;
74 static struct ref_transaction *transaction;
75
76 static enum deny_action parse_deny_action(const char *var, const char *value)
77 {
78         if (value) {
79                 if (!strcasecmp(value, "ignore"))
80                         return DENY_IGNORE;
81                 if (!strcasecmp(value, "warn"))
82                         return DENY_WARN;
83                 if (!strcasecmp(value, "refuse"))
84                         return DENY_REFUSE;
85                 if (!strcasecmp(value, "updateinstead"))
86                         return DENY_UPDATE_INSTEAD;
87         }
88         if (git_config_bool(var, value))
89                 return DENY_REFUSE;
90         return DENY_IGNORE;
91 }
92
93 static int receive_pack_config(const char *var, const char *value, void *cb)
94 {
95         int status = parse_hide_refs_config(var, value, "receive");
96
97         if (status)
98                 return status;
99
100         if (strcmp(var, "receive.denydeletes") == 0) {
101                 deny_deletes = git_config_bool(var, value);
102                 return 0;
103         }
104
105         if (strcmp(var, "receive.denynonfastforwards") == 0) {
106                 deny_non_fast_forwards = git_config_bool(var, value);
107                 return 0;
108         }
109
110         if (strcmp(var, "receive.unpacklimit") == 0) {
111                 receive_unpack_limit = git_config_int(var, value);
112                 return 0;
113         }
114
115         if (strcmp(var, "transfer.unpacklimit") == 0) {
116                 transfer_unpack_limit = git_config_int(var, value);
117                 return 0;
118         }
119
120         if (strcmp(var, "receive.fsck.skiplist") == 0) {
121                 const char *path;
122
123                 if (git_config_pathname(&path, var, value))
124                         return 1;
125                 strbuf_addf(&fsck_msg_types, "%cskiplist=%s",
126                         fsck_msg_types.len ? ',' : '=', path);
127                 free((char *)path);
128                 return 0;
129         }
130
131         if (skip_prefix(var, "receive.fsck.", &var)) {
132                 if (is_valid_msg_type(var, value))
133                         strbuf_addf(&fsck_msg_types, "%c%s=%s",
134                                 fsck_msg_types.len ? ',' : '=', var, value);
135                 else
136                         warning("Skipping unknown msg id '%s'", var);
137                 return 0;
138         }
139
140         if (strcmp(var, "receive.fsckobjects") == 0) {
141                 receive_fsck_objects = git_config_bool(var, value);
142                 return 0;
143         }
144
145         if (strcmp(var, "transfer.fsckobjects") == 0) {
146                 transfer_fsck_objects = git_config_bool(var, value);
147                 return 0;
148         }
149
150         if (!strcmp(var, "receive.denycurrentbranch")) {
151                 deny_current_branch = parse_deny_action(var, value);
152                 return 0;
153         }
154
155         if (strcmp(var, "receive.denydeletecurrent") == 0) {
156                 deny_delete_current = parse_deny_action(var, value);
157                 return 0;
158         }
159
160         if (strcmp(var, "repack.usedeltabaseoffset") == 0) {
161                 prefer_ofs_delta = git_config_bool(var, value);
162                 return 0;
163         }
164
165         if (strcmp(var, "receive.updateserverinfo") == 0) {
166                 auto_update_server_info = git_config_bool(var, value);
167                 return 0;
168         }
169
170         if (strcmp(var, "receive.autogc") == 0) {
171                 auto_gc = git_config_bool(var, value);
172                 return 0;
173         }
174
175         if (strcmp(var, "receive.shallowupdate") == 0) {
176                 shallow_update = git_config_bool(var, value);
177                 return 0;
178         }
179
180         if (strcmp(var, "receive.certnonceseed") == 0)
181                 return git_config_string(&cert_nonce_seed, var, value);
182
183         if (strcmp(var, "receive.certnonceslop") == 0) {
184                 nonce_stamp_slop_limit = git_config_ulong(var, value);
185                 return 0;
186         }
187
188         if (strcmp(var, "receive.advertiseatomic") == 0) {
189                 advertise_atomic_push = git_config_bool(var, value);
190                 return 0;
191         }
192
193         return git_default_config(var, value, cb);
194 }
195
196 static void show_ref(const char *path, const unsigned char *sha1)
197 {
198         if (sent_capabilities) {
199                 packet_write(1, "%s %s\n", sha1_to_hex(sha1), path);
200         } else {
201                 struct strbuf cap = STRBUF_INIT;
202
203                 strbuf_addstr(&cap,
204                               "report-status delete-refs side-band-64k quiet");
205                 if (advertise_atomic_push)
206                         strbuf_addstr(&cap, " atomic");
207                 if (prefer_ofs_delta)
208                         strbuf_addstr(&cap, " ofs-delta");
209                 if (push_cert_nonce)
210                         strbuf_addf(&cap, " push-cert=%s", push_cert_nonce);
211                 strbuf_addf(&cap, " agent=%s", git_user_agent_sanitized());
212                 packet_write(1, "%s %s%c%s\n",
213                              sha1_to_hex(sha1), path, 0, cap.buf);
214                 strbuf_release(&cap);
215                 sent_capabilities = 1;
216         }
217 }
218
219 static int show_ref_cb(const char *path_full, const struct object_id *oid,
220                        int flag, void *unused)
221 {
222         const char *path = strip_namespace(path_full);
223
224         if (ref_is_hidden(path, path_full))
225                 return 0;
226
227         /*
228          * Advertise refs outside our current namespace as ".have"
229          * refs, so that the client can use them to minimize data
230          * transfer but will otherwise ignore them. This happens to
231          * cover ".have" that are thrown in by add_one_alternate_ref()
232          * to mark histories that are complete in our alternates as
233          * well.
234          */
235         if (!path)
236                 path = ".have";
237         show_ref(path, oid->hash);
238         return 0;
239 }
240
241 static void show_one_alternate_sha1(const unsigned char sha1[20], void *unused)
242 {
243         show_ref(".have", sha1);
244 }
245
246 static void collect_one_alternate_ref(const struct ref *ref, void *data)
247 {
248         struct sha1_array *sa = data;
249         sha1_array_append(sa, ref->old_oid.hash);
250 }
251
252 static void write_head_info(void)
253 {
254         struct sha1_array sa = SHA1_ARRAY_INIT;
255
256         for_each_alternate_ref(collect_one_alternate_ref, &sa);
257         sha1_array_for_each_unique(&sa, show_one_alternate_sha1, NULL);
258         sha1_array_clear(&sa);
259         for_each_ref(show_ref_cb, NULL);
260         if (!sent_capabilities)
261                 show_ref("capabilities^{}", null_sha1);
262
263         advertise_shallow_grafts(1);
264
265         /* EOF */
266         packet_flush(1);
267 }
268
269 struct command {
270         struct command *next;
271         const char *error_string;
272         unsigned int skip_update:1,
273                      did_not_exist:1;
274         int index;
275         unsigned char old_sha1[20];
276         unsigned char new_sha1[20];
277         char ref_name[FLEX_ARRAY]; /* more */
278 };
279
280 static void rp_error(const char *err, ...) __attribute__((format (printf, 1, 2)));
281 static void rp_warning(const char *err, ...) __attribute__((format (printf, 1, 2)));
282
283 static void report_message(const char *prefix, const char *err, va_list params)
284 {
285         int sz;
286         char msg[4096];
287
288         sz = xsnprintf(msg, sizeof(msg), "%s", prefix);
289         sz += vsnprintf(msg + sz, sizeof(msg) - sz, err, params);
290         if (sz > (sizeof(msg) - 1))
291                 sz = sizeof(msg) - 1;
292         msg[sz++] = '\n';
293
294         if (use_sideband)
295                 send_sideband(1, 2, msg, sz, use_sideband);
296         else
297                 xwrite(2, msg, sz);
298 }
299
300 static void rp_warning(const char *err, ...)
301 {
302         va_list params;
303         va_start(params, err);
304         report_message("warning: ", err, params);
305         va_end(params);
306 }
307
308 static void rp_error(const char *err, ...)
309 {
310         va_list params;
311         va_start(params, err);
312         report_message("error: ", err, params);
313         va_end(params);
314 }
315
316 static int copy_to_sideband(int in, int out, void *arg)
317 {
318         char data[128];
319         while (1) {
320                 ssize_t sz = xread(in, data, sizeof(data));
321                 if (sz <= 0)
322                         break;
323                 send_sideband(1, 2, data, sz, use_sideband);
324         }
325         close(in);
326         return 0;
327 }
328
329 #define HMAC_BLOCK_SIZE 64
330
331 static void hmac_sha1(unsigned char *out,
332                       const char *key_in, size_t key_len,
333                       const char *text, size_t text_len)
334 {
335         unsigned char key[HMAC_BLOCK_SIZE];
336         unsigned char k_ipad[HMAC_BLOCK_SIZE];
337         unsigned char k_opad[HMAC_BLOCK_SIZE];
338         int i;
339         git_SHA_CTX ctx;
340
341         /* RFC 2104 2. (1) */
342         memset(key, '\0', HMAC_BLOCK_SIZE);
343         if (HMAC_BLOCK_SIZE < key_len) {
344                 git_SHA1_Init(&ctx);
345                 git_SHA1_Update(&ctx, key_in, key_len);
346                 git_SHA1_Final(key, &ctx);
347         } else {
348                 memcpy(key, key_in, key_len);
349         }
350
351         /* RFC 2104 2. (2) & (5) */
352         for (i = 0; i < sizeof(key); i++) {
353                 k_ipad[i] = key[i] ^ 0x36;
354                 k_opad[i] = key[i] ^ 0x5c;
355         }
356
357         /* RFC 2104 2. (3) & (4) */
358         git_SHA1_Init(&ctx);
359         git_SHA1_Update(&ctx, k_ipad, sizeof(k_ipad));
360         git_SHA1_Update(&ctx, text, text_len);
361         git_SHA1_Final(out, &ctx);
362
363         /* RFC 2104 2. (6) & (7) */
364         git_SHA1_Init(&ctx);
365         git_SHA1_Update(&ctx, k_opad, sizeof(k_opad));
366         git_SHA1_Update(&ctx, out, 20);
367         git_SHA1_Final(out, &ctx);
368 }
369
370 static char *prepare_push_cert_nonce(const char *path, unsigned long stamp)
371 {
372         struct strbuf buf = STRBUF_INIT;
373         unsigned char sha1[20];
374
375         strbuf_addf(&buf, "%s:%lu", path, stamp);
376         hmac_sha1(sha1, buf.buf, buf.len, cert_nonce_seed, strlen(cert_nonce_seed));;
377         strbuf_release(&buf);
378
379         /* RFC 2104 5. HMAC-SHA1-80 */
380         strbuf_addf(&buf, "%lu-%.*s", stamp, 20, sha1_to_hex(sha1));
381         return strbuf_detach(&buf, NULL);
382 }
383
384 /*
385  * NEEDSWORK: reuse find_commit_header() from jk/commit-author-parsing
386  * after dropping "_commit" from its name and possibly moving it out
387  * of commit.c
388  */
389 static char *find_header(const char *msg, size_t len, const char *key)
390 {
391         int key_len = strlen(key);
392         const char *line = msg;
393
394         while (line && line < msg + len) {
395                 const char *eol = strchrnul(line, '\n');
396
397                 if ((msg + len <= eol) || line == eol)
398                         return NULL;
399                 if (line + key_len < eol &&
400                     !memcmp(line, key, key_len) && line[key_len] == ' ') {
401                         int offset = key_len + 1;
402                         return xmemdupz(line + offset, (eol - line) - offset);
403                 }
404                 line = *eol ? eol + 1 : NULL;
405         }
406         return NULL;
407 }
408
409 static const char *check_nonce(const char *buf, size_t len)
410 {
411         char *nonce = find_header(buf, len, "nonce");
412         unsigned long stamp, ostamp;
413         char *bohmac, *expect = NULL;
414         const char *retval = NONCE_BAD;
415
416         if (!nonce) {
417                 retval = NONCE_MISSING;
418                 goto leave;
419         } else if (!push_cert_nonce) {
420                 retval = NONCE_UNSOLICITED;
421                 goto leave;
422         } else if (!strcmp(push_cert_nonce, nonce)) {
423                 retval = NONCE_OK;
424                 goto leave;
425         }
426
427         if (!stateless_rpc) {
428                 /* returned nonce MUST match what we gave out earlier */
429                 retval = NONCE_BAD;
430                 goto leave;
431         }
432
433         /*
434          * In stateless mode, we may be receiving a nonce issued by
435          * another instance of the server that serving the same
436          * repository, and the timestamps may not match, but the
437          * nonce-seed and dir should match, so we can recompute and
438          * report the time slop.
439          *
440          * In addition, when a nonce issued by another instance has
441          * timestamp within receive.certnonceslop seconds, we pretend
442          * as if we issued that nonce when reporting to the hook.
443          */
444
445         /* nonce is concat(<seconds-since-epoch>, "-", <hmac>) */
446         if (*nonce <= '0' || '9' < *nonce) {
447                 retval = NONCE_BAD;
448                 goto leave;
449         }
450         stamp = strtoul(nonce, &bohmac, 10);
451         if (bohmac == nonce || bohmac[0] != '-') {
452                 retval = NONCE_BAD;
453                 goto leave;
454         }
455
456         expect = prepare_push_cert_nonce(service_dir, stamp);
457         if (strcmp(expect, nonce)) {
458                 /* Not what we would have signed earlier */
459                 retval = NONCE_BAD;
460                 goto leave;
461         }
462
463         /*
464          * By how many seconds is this nonce stale?  Negative value
465          * would mean it was issued by another server with its clock
466          * skewed in the future.
467          */
468         ostamp = strtoul(push_cert_nonce, NULL, 10);
469         nonce_stamp_slop = (long)ostamp - (long)stamp;
470
471         if (nonce_stamp_slop_limit &&
472             labs(nonce_stamp_slop) <= nonce_stamp_slop_limit) {
473                 /*
474                  * Pretend as if the received nonce (which passes the
475                  * HMAC check, so it is not a forged by third-party)
476                  * is what we issued.
477                  */
478                 free((void *)push_cert_nonce);
479                 push_cert_nonce = xstrdup(nonce);
480                 retval = NONCE_OK;
481         } else {
482                 retval = NONCE_SLOP;
483         }
484
485 leave:
486         free(nonce);
487         free(expect);
488         return retval;
489 }
490
491 static void prepare_push_cert_sha1(struct child_process *proc)
492 {
493         static int already_done;
494
495         if (!push_cert.len)
496                 return;
497
498         if (!already_done) {
499                 struct strbuf gpg_output = STRBUF_INIT;
500                 struct strbuf gpg_status = STRBUF_INIT;
501                 int bogs /* beginning_of_gpg_sig */;
502
503                 already_done = 1;
504                 if (write_sha1_file(push_cert.buf, push_cert.len, "blob", push_cert_sha1))
505                         hashclr(push_cert_sha1);
506
507                 memset(&sigcheck, '\0', sizeof(sigcheck));
508                 sigcheck.result = 'N';
509
510                 bogs = parse_signature(push_cert.buf, push_cert.len);
511                 if (verify_signed_buffer(push_cert.buf, bogs,
512                                          push_cert.buf + bogs, push_cert.len - bogs,
513                                          &gpg_output, &gpg_status) < 0) {
514                         ; /* error running gpg */
515                 } else {
516                         sigcheck.payload = push_cert.buf;
517                         sigcheck.gpg_output = gpg_output.buf;
518                         sigcheck.gpg_status = gpg_status.buf;
519                         parse_gpg_output(&sigcheck);
520                 }
521
522                 strbuf_release(&gpg_output);
523                 strbuf_release(&gpg_status);
524                 nonce_status = check_nonce(push_cert.buf, bogs);
525         }
526         if (!is_null_sha1(push_cert_sha1)) {
527                 argv_array_pushf(&proc->env_array, "GIT_PUSH_CERT=%s",
528                                  sha1_to_hex(push_cert_sha1));
529                 argv_array_pushf(&proc->env_array, "GIT_PUSH_CERT_SIGNER=%s",
530                                  sigcheck.signer ? sigcheck.signer : "");
531                 argv_array_pushf(&proc->env_array, "GIT_PUSH_CERT_KEY=%s",
532                                  sigcheck.key ? sigcheck.key : "");
533                 argv_array_pushf(&proc->env_array, "GIT_PUSH_CERT_STATUS=%c",
534                                  sigcheck.result);
535                 if (push_cert_nonce) {
536                         argv_array_pushf(&proc->env_array,
537                                          "GIT_PUSH_CERT_NONCE=%s",
538                                          push_cert_nonce);
539                         argv_array_pushf(&proc->env_array,
540                                          "GIT_PUSH_CERT_NONCE_STATUS=%s",
541                                          nonce_status);
542                         if (nonce_status == NONCE_SLOP)
543                                 argv_array_pushf(&proc->env_array,
544                                                  "GIT_PUSH_CERT_NONCE_SLOP=%ld",
545                                                  nonce_stamp_slop);
546                 }
547         }
548 }
549
550 typedef int (*feed_fn)(void *, const char **, size_t *);
551 static int run_and_feed_hook(const char *hook_name, feed_fn feed, void *feed_state)
552 {
553         struct child_process proc = CHILD_PROCESS_INIT;
554         struct async muxer;
555         const char *argv[2];
556         int code;
557
558         argv[0] = find_hook(hook_name);
559         if (!argv[0])
560                 return 0;
561
562         argv[1] = NULL;
563
564         proc.argv = argv;
565         proc.in = -1;
566         proc.stdout_to_stderr = 1;
567
568         if (use_sideband) {
569                 memset(&muxer, 0, sizeof(muxer));
570                 muxer.proc = copy_to_sideband;
571                 muxer.in = -1;
572                 code = start_async(&muxer);
573                 if (code)
574                         return code;
575                 proc.err = muxer.in;
576         }
577
578         prepare_push_cert_sha1(&proc);
579
580         code = start_command(&proc);
581         if (code) {
582                 if (use_sideband)
583                         finish_async(&muxer);
584                 return code;
585         }
586
587         sigchain_push(SIGPIPE, SIG_IGN);
588
589         while (1) {
590                 const char *buf;
591                 size_t n;
592                 if (feed(feed_state, &buf, &n))
593                         break;
594                 if (write_in_full(proc.in, buf, n) != n)
595                         break;
596         }
597         close(proc.in);
598         if (use_sideband)
599                 finish_async(&muxer);
600
601         sigchain_pop(SIGPIPE);
602
603         return finish_command(&proc);
604 }
605
606 struct receive_hook_feed_state {
607         struct command *cmd;
608         int skip_broken;
609         struct strbuf buf;
610 };
611
612 static int feed_receive_hook(void *state_, const char **bufp, size_t *sizep)
613 {
614         struct receive_hook_feed_state *state = state_;
615         struct command *cmd = state->cmd;
616
617         while (cmd &&
618                state->skip_broken && (cmd->error_string || cmd->did_not_exist))
619                 cmd = cmd->next;
620         if (!cmd)
621                 return -1; /* EOF */
622         strbuf_reset(&state->buf);
623         strbuf_addf(&state->buf, "%s %s %s\n",
624                     sha1_to_hex(cmd->old_sha1), sha1_to_hex(cmd->new_sha1),
625                     cmd->ref_name);
626         state->cmd = cmd->next;
627         if (bufp) {
628                 *bufp = state->buf.buf;
629                 *sizep = state->buf.len;
630         }
631         return 0;
632 }
633
634 static int run_receive_hook(struct command *commands, const char *hook_name,
635                             int skip_broken)
636 {
637         struct receive_hook_feed_state state;
638         int status;
639
640         strbuf_init(&state.buf, 0);
641         state.cmd = commands;
642         state.skip_broken = skip_broken;
643         if (feed_receive_hook(&state, NULL, NULL))
644                 return 0;
645         state.cmd = commands;
646         status = run_and_feed_hook(hook_name, feed_receive_hook, &state);
647         strbuf_release(&state.buf);
648         return status;
649 }
650
651 static int run_update_hook(struct command *cmd)
652 {
653         const char *argv[5];
654         struct child_process proc = CHILD_PROCESS_INIT;
655         int code;
656
657         argv[0] = find_hook("update");
658         if (!argv[0])
659                 return 0;
660
661         argv[1] = cmd->ref_name;
662         argv[2] = sha1_to_hex(cmd->old_sha1);
663         argv[3] = sha1_to_hex(cmd->new_sha1);
664         argv[4] = NULL;
665
666         proc.no_stdin = 1;
667         proc.stdout_to_stderr = 1;
668         proc.err = use_sideband ? -1 : 0;
669         proc.argv = argv;
670
671         code = start_command(&proc);
672         if (code)
673                 return code;
674         if (use_sideband)
675                 copy_to_sideband(proc.err, -1, NULL);
676         return finish_command(&proc);
677 }
678
679 static int is_ref_checked_out(const char *ref)
680 {
681         if (is_bare_repository())
682                 return 0;
683
684         if (!head_name)
685                 return 0;
686         return !strcmp(head_name, ref);
687 }
688
689 static char *refuse_unconfigured_deny_msg[] = {
690         "By default, updating the current branch in a non-bare repository",
691         "is denied, because it will make the index and work tree inconsistent",
692         "with what you pushed, and will require 'git reset --hard' to match",
693         "the work tree to HEAD.",
694         "",
695         "You can set 'receive.denyCurrentBranch' configuration variable to",
696         "'ignore' or 'warn' in the remote repository to allow pushing into",
697         "its current branch; however, this is not recommended unless you",
698         "arranged to update its work tree to match what you pushed in some",
699         "other way.",
700         "",
701         "To squelch this message and still keep the default behaviour, set",
702         "'receive.denyCurrentBranch' configuration variable to 'refuse'."
703 };
704
705 static void refuse_unconfigured_deny(void)
706 {
707         int i;
708         for (i = 0; i < ARRAY_SIZE(refuse_unconfigured_deny_msg); i++)
709                 rp_error("%s", refuse_unconfigured_deny_msg[i]);
710 }
711
712 static char *refuse_unconfigured_deny_delete_current_msg[] = {
713         "By default, deleting the current branch is denied, because the next",
714         "'git clone' won't result in any file checked out, causing confusion.",
715         "",
716         "You can set 'receive.denyDeleteCurrent' configuration variable to",
717         "'warn' or 'ignore' in the remote repository to allow deleting the",
718         "current branch, with or without a warning message.",
719         "",
720         "To squelch this message, you can set it to 'refuse'."
721 };
722
723 static void refuse_unconfigured_deny_delete_current(void)
724 {
725         int i;
726         for (i = 0;
727              i < ARRAY_SIZE(refuse_unconfigured_deny_delete_current_msg);
728              i++)
729                 rp_error("%s", refuse_unconfigured_deny_delete_current_msg[i]);
730 }
731
732 static int command_singleton_iterator(void *cb_data, unsigned char sha1[20]);
733 static int update_shallow_ref(struct command *cmd, struct shallow_info *si)
734 {
735         static struct lock_file shallow_lock;
736         struct sha1_array extra = SHA1_ARRAY_INIT;
737         const char *alt_file;
738         uint32_t mask = 1 << (cmd->index % 32);
739         int i;
740
741         trace_printf_key(&trace_shallow,
742                          "shallow: update_shallow_ref %s\n", cmd->ref_name);
743         for (i = 0; i < si->shallow->nr; i++)
744                 if (si->used_shallow[i] &&
745                     (si->used_shallow[i][cmd->index / 32] & mask) &&
746                     !delayed_reachability_test(si, i))
747                         sha1_array_append(&extra, si->shallow->sha1[i]);
748
749         setup_alternate_shallow(&shallow_lock, &alt_file, &extra);
750         if (check_shallow_connected(command_singleton_iterator,
751                                     0, cmd, alt_file)) {
752                 rollback_lock_file(&shallow_lock);
753                 sha1_array_clear(&extra);
754                 return -1;
755         }
756
757         commit_lock_file(&shallow_lock);
758
759         /*
760          * Make sure setup_alternate_shallow() for the next ref does
761          * not lose these new roots..
762          */
763         for (i = 0; i < extra.nr; i++)
764                 register_shallow(extra.sha1[i]);
765
766         si->shallow_ref[cmd->index] = 0;
767         sha1_array_clear(&extra);
768         return 0;
769 }
770
771 /*
772  * NEEDSWORK: we should consolidate various implementions of "are we
773  * on an unborn branch?" test into one, and make the unified one more
774  * robust. !get_sha1() based check used here and elsewhere would not
775  * allow us to tell an unborn branch from corrupt ref, for example.
776  * For the purpose of fixing "deploy-to-update does not work when
777  * pushing into an empty repository" issue, this should suffice for
778  * now.
779  */
780 static int head_has_history(void)
781 {
782         unsigned char sha1[20];
783
784         return !get_sha1("HEAD", sha1);
785 }
786
787 static const char *push_to_deploy(unsigned char *sha1,
788                                   struct argv_array *env,
789                                   const char *work_tree)
790 {
791         const char *update_refresh[] = {
792                 "update-index", "-q", "--ignore-submodules", "--refresh", NULL
793         };
794         const char *diff_files[] = {
795                 "diff-files", "--quiet", "--ignore-submodules", "--", NULL
796         };
797         const char *diff_index[] = {
798                 "diff-index", "--quiet", "--cached", "--ignore-submodules",
799                 NULL, "--", NULL
800         };
801         const char *read_tree[] = {
802                 "read-tree", "-u", "-m", NULL, NULL
803         };
804         struct child_process child = CHILD_PROCESS_INIT;
805
806         child.argv = update_refresh;
807         child.env = env->argv;
808         child.dir = work_tree;
809         child.no_stdin = 1;
810         child.stdout_to_stderr = 1;
811         child.git_cmd = 1;
812         if (run_command(&child))
813                 return "Up-to-date check failed";
814
815         /* run_command() does not clean up completely; reinitialize */
816         child_process_init(&child);
817         child.argv = diff_files;
818         child.env = env->argv;
819         child.dir = work_tree;
820         child.no_stdin = 1;
821         child.stdout_to_stderr = 1;
822         child.git_cmd = 1;
823         if (run_command(&child))
824                 return "Working directory has unstaged changes";
825
826         /* diff-index with either HEAD or an empty tree */
827         diff_index[4] = head_has_history() ? "HEAD" : EMPTY_TREE_SHA1_HEX;
828
829         child_process_init(&child);
830         child.argv = diff_index;
831         child.env = env->argv;
832         child.no_stdin = 1;
833         child.no_stdout = 1;
834         child.stdout_to_stderr = 0;
835         child.git_cmd = 1;
836         if (run_command(&child))
837                 return "Working directory has staged changes";
838
839         read_tree[3] = sha1_to_hex(sha1);
840         child_process_init(&child);
841         child.argv = read_tree;
842         child.env = env->argv;
843         child.dir = work_tree;
844         child.no_stdin = 1;
845         child.no_stdout = 1;
846         child.stdout_to_stderr = 0;
847         child.git_cmd = 1;
848         if (run_command(&child))
849                 return "Could not update working tree to new HEAD";
850
851         return NULL;
852 }
853
854 static const char *push_to_checkout_hook = "push-to-checkout";
855
856 static const char *push_to_checkout(unsigned char *sha1,
857                                     struct argv_array *env,
858                                     const char *work_tree)
859 {
860         argv_array_pushf(env, "GIT_WORK_TREE=%s", absolute_path(work_tree));
861         if (run_hook_le(env->argv, push_to_checkout_hook,
862                         sha1_to_hex(sha1), NULL))
863                 return "push-to-checkout hook declined";
864         else
865                 return NULL;
866 }
867
868 static const char *update_worktree(unsigned char *sha1)
869 {
870         const char *retval;
871         const char *work_tree = git_work_tree_cfg ? git_work_tree_cfg : "..";
872         struct argv_array env = ARGV_ARRAY_INIT;
873
874         if (is_bare_repository())
875                 return "denyCurrentBranch = updateInstead needs a worktree";
876
877         argv_array_pushf(&env, "GIT_DIR=%s", absolute_path(get_git_dir()));
878
879         if (!find_hook(push_to_checkout_hook))
880                 retval = push_to_deploy(sha1, &env, work_tree);
881         else
882                 retval = push_to_checkout(sha1, &env, work_tree);
883
884         argv_array_clear(&env);
885         return retval;
886 }
887
888 static const char *update(struct command *cmd, struct shallow_info *si)
889 {
890         const char *name = cmd->ref_name;
891         struct strbuf namespaced_name_buf = STRBUF_INIT;
892         const char *namespaced_name, *ret;
893         unsigned char *old_sha1 = cmd->old_sha1;
894         unsigned char *new_sha1 = cmd->new_sha1;
895
896         /* only refs/... are allowed */
897         if (!starts_with(name, "refs/") || check_refname_format(name + 5, 0)) {
898                 rp_error("refusing to create funny ref '%s' remotely", name);
899                 return "funny refname";
900         }
901
902         strbuf_addf(&namespaced_name_buf, "%s%s", get_git_namespace(), name);
903         namespaced_name = strbuf_detach(&namespaced_name_buf, NULL);
904
905         if (is_ref_checked_out(namespaced_name)) {
906                 switch (deny_current_branch) {
907                 case DENY_IGNORE:
908                         break;
909                 case DENY_WARN:
910                         rp_warning("updating the current branch");
911                         break;
912                 case DENY_REFUSE:
913                 case DENY_UNCONFIGURED:
914                         rp_error("refusing to update checked out branch: %s", name);
915                         if (deny_current_branch == DENY_UNCONFIGURED)
916                                 refuse_unconfigured_deny();
917                         return "branch is currently checked out";
918                 case DENY_UPDATE_INSTEAD:
919                         ret = update_worktree(new_sha1);
920                         if (ret)
921                                 return ret;
922                         break;
923                 }
924         }
925
926         if (!is_null_sha1(new_sha1) && !has_sha1_file(new_sha1)) {
927                 error("unpack should have generated %s, "
928                       "but I can't find it!", sha1_to_hex(new_sha1));
929                 return "bad pack";
930         }
931
932         if (!is_null_sha1(old_sha1) && is_null_sha1(new_sha1)) {
933                 if (deny_deletes && starts_with(name, "refs/heads/")) {
934                         rp_error("denying ref deletion for %s", name);
935                         return "deletion prohibited";
936                 }
937
938                 if (head_name && !strcmp(namespaced_name, head_name)) {
939                         switch (deny_delete_current) {
940                         case DENY_IGNORE:
941                                 break;
942                         case DENY_WARN:
943                                 rp_warning("deleting the current branch");
944                                 break;
945                         case DENY_REFUSE:
946                         case DENY_UNCONFIGURED:
947                         case DENY_UPDATE_INSTEAD:
948                                 if (deny_delete_current == DENY_UNCONFIGURED)
949                                         refuse_unconfigured_deny_delete_current();
950                                 rp_error("refusing to delete the current branch: %s", name);
951                                 return "deletion of the current branch prohibited";
952                         default:
953                                 return "Invalid denyDeleteCurrent setting";
954                         }
955                 }
956         }
957
958         if (deny_non_fast_forwards && !is_null_sha1(new_sha1) &&
959             !is_null_sha1(old_sha1) &&
960             starts_with(name, "refs/heads/")) {
961                 struct object *old_object, *new_object;
962                 struct commit *old_commit, *new_commit;
963
964                 old_object = parse_object(old_sha1);
965                 new_object = parse_object(new_sha1);
966
967                 if (!old_object || !new_object ||
968                     old_object->type != OBJ_COMMIT ||
969                     new_object->type != OBJ_COMMIT) {
970                         error("bad sha1 objects for %s", name);
971                         return "bad ref";
972                 }
973                 old_commit = (struct commit *)old_object;
974                 new_commit = (struct commit *)new_object;
975                 if (!in_merge_bases(old_commit, new_commit)) {
976                         rp_error("denying non-fast-forward %s"
977                                  " (you should pull first)", name);
978                         return "non-fast-forward";
979                 }
980         }
981         if (run_update_hook(cmd)) {
982                 rp_error("hook declined to update %s", name);
983                 return "hook declined";
984         }
985
986         if (is_null_sha1(new_sha1)) {
987                 struct strbuf err = STRBUF_INIT;
988                 if (!parse_object(old_sha1)) {
989                         old_sha1 = NULL;
990                         if (ref_exists(name)) {
991                                 rp_warning("Allowing deletion of corrupt ref.");
992                         } else {
993                                 rp_warning("Deleting a non-existent ref.");
994                                 cmd->did_not_exist = 1;
995                         }
996                 }
997                 if (ref_transaction_delete(transaction,
998                                            namespaced_name,
999                                            old_sha1,
1000                                            0, "push", &err)) {
1001                         rp_error("%s", err.buf);
1002                         strbuf_release(&err);
1003                         return "failed to delete";
1004                 }
1005                 strbuf_release(&err);
1006                 return NULL; /* good */
1007         }
1008         else {
1009                 struct strbuf err = STRBUF_INIT;
1010                 if (shallow_update && si->shallow_ref[cmd->index] &&
1011                     update_shallow_ref(cmd, si))
1012                         return "shallow error";
1013
1014                 if (ref_transaction_update(transaction,
1015                                            namespaced_name,
1016                                            new_sha1, old_sha1,
1017                                            0, "push",
1018                                            &err)) {
1019                         rp_error("%s", err.buf);
1020                         strbuf_release(&err);
1021
1022                         return "failed to update ref";
1023                 }
1024                 strbuf_release(&err);
1025
1026                 return NULL; /* good */
1027         }
1028 }
1029
1030 static void run_update_post_hook(struct command *commands)
1031 {
1032         struct command *cmd;
1033         int argc;
1034         struct child_process proc = CHILD_PROCESS_INIT;
1035         const char *hook;
1036
1037         hook = find_hook("post-update");
1038         for (argc = 0, cmd = commands; cmd; cmd = cmd->next) {
1039                 if (cmd->error_string || cmd->did_not_exist)
1040                         continue;
1041                 argc++;
1042         }
1043         if (!argc || !hook)
1044                 return;
1045
1046         argv_array_push(&proc.args, hook);
1047         for (cmd = commands; cmd; cmd = cmd->next) {
1048                 if (cmd->error_string || cmd->did_not_exist)
1049                         continue;
1050                 argv_array_push(&proc.args, cmd->ref_name);
1051         }
1052
1053         proc.no_stdin = 1;
1054         proc.stdout_to_stderr = 1;
1055         proc.err = use_sideband ? -1 : 0;
1056
1057         if (!start_command(&proc)) {
1058                 if (use_sideband)
1059                         copy_to_sideband(proc.err, -1, NULL);
1060                 finish_command(&proc);
1061         }
1062 }
1063
1064 static void check_aliased_update(struct command *cmd, struct string_list *list)
1065 {
1066         struct strbuf buf = STRBUF_INIT;
1067         const char *dst_name;
1068         struct string_list_item *item;
1069         struct command *dst_cmd;
1070         unsigned char sha1[GIT_SHA1_RAWSZ];
1071         char cmd_oldh[GIT_SHA1_HEXSZ + 1],
1072              cmd_newh[GIT_SHA1_HEXSZ + 1],
1073              dst_oldh[GIT_SHA1_HEXSZ + 1],
1074              dst_newh[GIT_SHA1_HEXSZ + 1];
1075         int flag;
1076
1077         strbuf_addf(&buf, "%s%s", get_git_namespace(), cmd->ref_name);
1078         dst_name = resolve_ref_unsafe(buf.buf, 0, sha1, &flag);
1079         strbuf_release(&buf);
1080
1081         if (!(flag & REF_ISSYMREF))
1082                 return;
1083
1084         dst_name = strip_namespace(dst_name);
1085         if (!dst_name) {
1086                 rp_error("refusing update to broken symref '%s'", cmd->ref_name);
1087                 cmd->skip_update = 1;
1088                 cmd->error_string = "broken symref";
1089                 return;
1090         }
1091
1092         if ((item = string_list_lookup(list, dst_name)) == NULL)
1093                 return;
1094
1095         cmd->skip_update = 1;
1096
1097         dst_cmd = (struct command *) item->util;
1098
1099         if (!hashcmp(cmd->old_sha1, dst_cmd->old_sha1) &&
1100             !hashcmp(cmd->new_sha1, dst_cmd->new_sha1))
1101                 return;
1102
1103         dst_cmd->skip_update = 1;
1104
1105         find_unique_abbrev_r(cmd_oldh, cmd->old_sha1, DEFAULT_ABBREV);
1106         find_unique_abbrev_r(cmd_newh, cmd->new_sha1, DEFAULT_ABBREV);
1107         find_unique_abbrev_r(dst_oldh, dst_cmd->old_sha1, DEFAULT_ABBREV);
1108         find_unique_abbrev_r(dst_newh, dst_cmd->new_sha1, DEFAULT_ABBREV);
1109         rp_error("refusing inconsistent update between symref '%s' (%s..%s) and"
1110                  " its target '%s' (%s..%s)",
1111                  cmd->ref_name, cmd_oldh, cmd_newh,
1112                  dst_cmd->ref_name, dst_oldh, dst_newh);
1113
1114         cmd->error_string = dst_cmd->error_string =
1115                 "inconsistent aliased update";
1116 }
1117
1118 static void check_aliased_updates(struct command *commands)
1119 {
1120         struct command *cmd;
1121         struct string_list ref_list = STRING_LIST_INIT_NODUP;
1122
1123         for (cmd = commands; cmd; cmd = cmd->next) {
1124                 struct string_list_item *item =
1125                         string_list_append(&ref_list, cmd->ref_name);
1126                 item->util = (void *)cmd;
1127         }
1128         string_list_sort(&ref_list);
1129
1130         for (cmd = commands; cmd; cmd = cmd->next) {
1131                 if (!cmd->error_string)
1132                         check_aliased_update(cmd, &ref_list);
1133         }
1134
1135         string_list_clear(&ref_list, 0);
1136 }
1137
1138 static int command_singleton_iterator(void *cb_data, unsigned char sha1[20])
1139 {
1140         struct command **cmd_list = cb_data;
1141         struct command *cmd = *cmd_list;
1142
1143         if (!cmd || is_null_sha1(cmd->new_sha1))
1144                 return -1; /* end of list */
1145         *cmd_list = NULL; /* this returns only one */
1146         hashcpy(sha1, cmd->new_sha1);
1147         return 0;
1148 }
1149
1150 static void set_connectivity_errors(struct command *commands,
1151                                     struct shallow_info *si)
1152 {
1153         struct command *cmd;
1154
1155         for (cmd = commands; cmd; cmd = cmd->next) {
1156                 struct command *singleton = cmd;
1157                 if (shallow_update && si->shallow_ref[cmd->index])
1158                         /* to be checked in update_shallow_ref() */
1159                         continue;
1160                 if (!check_everything_connected(command_singleton_iterator,
1161                                                 0, &singleton))
1162                         continue;
1163                 cmd->error_string = "missing necessary objects";
1164         }
1165 }
1166
1167 struct iterate_data {
1168         struct command *cmds;
1169         struct shallow_info *si;
1170 };
1171
1172 static int iterate_receive_command_list(void *cb_data, unsigned char sha1[20])
1173 {
1174         struct iterate_data *data = cb_data;
1175         struct command **cmd_list = &data->cmds;
1176         struct command *cmd = *cmd_list;
1177
1178         for (; cmd; cmd = cmd->next) {
1179                 if (shallow_update && data->si->shallow_ref[cmd->index])
1180                         /* to be checked in update_shallow_ref() */
1181                         continue;
1182                 if (!is_null_sha1(cmd->new_sha1) && !cmd->skip_update) {
1183                         hashcpy(sha1, cmd->new_sha1);
1184                         *cmd_list = cmd->next;
1185                         return 0;
1186                 }
1187         }
1188         *cmd_list = NULL;
1189         return -1; /* end of list */
1190 }
1191
1192 static void reject_updates_to_hidden(struct command *commands)
1193 {
1194         struct strbuf refname_full = STRBUF_INIT;
1195         size_t prefix_len;
1196         struct command *cmd;
1197
1198         strbuf_addstr(&refname_full, get_git_namespace());
1199         prefix_len = refname_full.len;
1200
1201         for (cmd = commands; cmd; cmd = cmd->next) {
1202                 if (cmd->error_string)
1203                         continue;
1204
1205                 strbuf_setlen(&refname_full, prefix_len);
1206                 strbuf_addstr(&refname_full, cmd->ref_name);
1207
1208                 if (!ref_is_hidden(cmd->ref_name, refname_full.buf))
1209                         continue;
1210                 if (is_null_sha1(cmd->new_sha1))
1211                         cmd->error_string = "deny deleting a hidden ref";
1212                 else
1213                         cmd->error_string = "deny updating a hidden ref";
1214         }
1215
1216         strbuf_release(&refname_full);
1217 }
1218
1219 static int should_process_cmd(struct command *cmd)
1220 {
1221         return !cmd->error_string && !cmd->skip_update;
1222 }
1223
1224 static void warn_if_skipped_connectivity_check(struct command *commands,
1225                                                struct shallow_info *si)
1226 {
1227         struct command *cmd;
1228         int checked_connectivity = 1;
1229
1230         for (cmd = commands; cmd; cmd = cmd->next) {
1231                 if (should_process_cmd(cmd) && si->shallow_ref[cmd->index]) {
1232                         error("BUG: connectivity check has not been run on ref %s",
1233                               cmd->ref_name);
1234                         checked_connectivity = 0;
1235                 }
1236         }
1237         if (!checked_connectivity)
1238                 die("BUG: connectivity check skipped???");
1239 }
1240
1241 static void execute_commands_non_atomic(struct command *commands,
1242                                         struct shallow_info *si)
1243 {
1244         struct command *cmd;
1245         struct strbuf err = STRBUF_INIT;
1246
1247         for (cmd = commands; cmd; cmd = cmd->next) {
1248                 if (!should_process_cmd(cmd))
1249                         continue;
1250
1251                 transaction = ref_transaction_begin(&err);
1252                 if (!transaction) {
1253                         rp_error("%s", err.buf);
1254                         strbuf_reset(&err);
1255                         cmd->error_string = "transaction failed to start";
1256                         continue;
1257                 }
1258
1259                 cmd->error_string = update(cmd, si);
1260
1261                 if (!cmd->error_string
1262                     && ref_transaction_commit(transaction, &err)) {
1263                         rp_error("%s", err.buf);
1264                         strbuf_reset(&err);
1265                         cmd->error_string = "failed to update ref";
1266                 }
1267                 ref_transaction_free(transaction);
1268         }
1269         strbuf_release(&err);
1270 }
1271
1272 static void execute_commands_atomic(struct command *commands,
1273                                         struct shallow_info *si)
1274 {
1275         struct command *cmd;
1276         struct strbuf err = STRBUF_INIT;
1277         const char *reported_error = "atomic push failure";
1278
1279         transaction = ref_transaction_begin(&err);
1280         if (!transaction) {
1281                 rp_error("%s", err.buf);
1282                 strbuf_reset(&err);
1283                 reported_error = "transaction failed to start";
1284                 goto failure;
1285         }
1286
1287         for (cmd = commands; cmd; cmd = cmd->next) {
1288                 if (!should_process_cmd(cmd))
1289                         continue;
1290
1291                 cmd->error_string = update(cmd, si);
1292
1293                 if (cmd->error_string)
1294                         goto failure;
1295         }
1296
1297         if (ref_transaction_commit(transaction, &err)) {
1298                 rp_error("%s", err.buf);
1299                 reported_error = "atomic transaction failed";
1300                 goto failure;
1301         }
1302         goto cleanup;
1303
1304 failure:
1305         for (cmd = commands; cmd; cmd = cmd->next)
1306                 if (!cmd->error_string)
1307                         cmd->error_string = reported_error;
1308
1309 cleanup:
1310         ref_transaction_free(transaction);
1311         strbuf_release(&err);
1312 }
1313
1314 static void execute_commands(struct command *commands,
1315                              const char *unpacker_error,
1316                              struct shallow_info *si)
1317 {
1318         struct command *cmd;
1319         unsigned char sha1[20];
1320         struct iterate_data data;
1321
1322         if (unpacker_error) {
1323                 for (cmd = commands; cmd; cmd = cmd->next)
1324                         cmd->error_string = "unpacker error";
1325                 return;
1326         }
1327
1328         data.cmds = commands;
1329         data.si = si;
1330         if (check_everything_connected(iterate_receive_command_list, 0, &data))
1331                 set_connectivity_errors(commands, si);
1332
1333         reject_updates_to_hidden(commands);
1334
1335         if (run_receive_hook(commands, "pre-receive", 0)) {
1336                 for (cmd = commands; cmd; cmd = cmd->next) {
1337                         if (!cmd->error_string)
1338                                 cmd->error_string = "pre-receive hook declined";
1339                 }
1340                 return;
1341         }
1342
1343         check_aliased_updates(commands);
1344
1345         free(head_name_to_free);
1346         head_name = head_name_to_free = resolve_refdup("HEAD", 0, sha1, NULL);
1347
1348         if (use_atomic)
1349                 execute_commands_atomic(commands, si);
1350         else
1351                 execute_commands_non_atomic(commands, si);
1352
1353         if (shallow_update)
1354                 warn_if_skipped_connectivity_check(commands, si);
1355 }
1356
1357 static struct command **queue_command(struct command **tail,
1358                                       const char *line,
1359                                       int linelen)
1360 {
1361         unsigned char old_sha1[20], new_sha1[20];
1362         struct command *cmd;
1363         const char *refname;
1364         int reflen;
1365
1366         if (linelen < 83 ||
1367             line[40] != ' ' ||
1368             line[81] != ' ' ||
1369             get_sha1_hex(line, old_sha1) ||
1370             get_sha1_hex(line + 41, new_sha1))
1371                 die("protocol error: expected old/new/ref, got '%s'", line);
1372
1373         refname = line + 82;
1374         reflen = linelen - 82;
1375         cmd = xcalloc(1, st_add3(sizeof(struct command), reflen, 1));
1376         hashcpy(cmd->old_sha1, old_sha1);
1377         hashcpy(cmd->new_sha1, new_sha1);
1378         memcpy(cmd->ref_name, refname, reflen);
1379         cmd->ref_name[reflen] = '\0';
1380         *tail = cmd;
1381         return &cmd->next;
1382 }
1383
1384 static void queue_commands_from_cert(struct command **tail,
1385                                      struct strbuf *push_cert)
1386 {
1387         const char *boc, *eoc;
1388
1389         if (*tail)
1390                 die("protocol error: got both push certificate and unsigned commands");
1391
1392         boc = strstr(push_cert->buf, "\n\n");
1393         if (!boc)
1394                 die("malformed push certificate %.*s", 100, push_cert->buf);
1395         else
1396                 boc += 2;
1397         eoc = push_cert->buf + parse_signature(push_cert->buf, push_cert->len);
1398
1399         while (boc < eoc) {
1400                 const char *eol = memchr(boc, '\n', eoc - boc);
1401                 tail = queue_command(tail, boc, eol ? eol - boc : eoc - eol);
1402                 boc = eol ? eol + 1 : eoc;
1403         }
1404 }
1405
1406 static struct command *read_head_info(struct sha1_array *shallow)
1407 {
1408         struct command *commands = NULL;
1409         struct command **p = &commands;
1410         for (;;) {
1411                 char *line;
1412                 int len, linelen;
1413
1414                 line = packet_read_line(0, &len);
1415                 if (!line)
1416                         break;
1417
1418                 if (len == 48 && starts_with(line, "shallow ")) {
1419                         unsigned char sha1[20];
1420                         if (get_sha1_hex(line + 8, sha1))
1421                                 die("protocol error: expected shallow sha, got '%s'",
1422                                     line + 8);
1423                         sha1_array_append(shallow, sha1);
1424                         continue;
1425                 }
1426
1427                 linelen = strlen(line);
1428                 if (linelen < len) {
1429                         const char *feature_list = line + linelen + 1;
1430                         if (parse_feature_request(feature_list, "report-status"))
1431                                 report_status = 1;
1432                         if (parse_feature_request(feature_list, "side-band-64k"))
1433                                 use_sideband = LARGE_PACKET_MAX;
1434                         if (parse_feature_request(feature_list, "quiet"))
1435                                 quiet = 1;
1436                         if (advertise_atomic_push
1437                             && parse_feature_request(feature_list, "atomic"))
1438                                 use_atomic = 1;
1439                 }
1440
1441                 if (!strcmp(line, "push-cert")) {
1442                         int true_flush = 0;
1443                         char certbuf[1024];
1444
1445                         for (;;) {
1446                                 len = packet_read(0, NULL, NULL,
1447                                                   certbuf, sizeof(certbuf), 0);
1448                                 if (!len) {
1449                                         true_flush = 1;
1450                                         break;
1451                                 }
1452                                 if (!strcmp(certbuf, "push-cert-end\n"))
1453                                         break; /* end of cert */
1454                                 strbuf_addstr(&push_cert, certbuf);
1455                         }
1456
1457                         if (true_flush)
1458                                 break;
1459                         continue;
1460                 }
1461
1462                 p = queue_command(p, line, linelen);
1463         }
1464
1465         if (push_cert.len)
1466                 queue_commands_from_cert(p, &push_cert);
1467
1468         return commands;
1469 }
1470
1471 static const char *parse_pack_header(struct pack_header *hdr)
1472 {
1473         switch (read_pack_header(0, hdr)) {
1474         case PH_ERROR_EOF:
1475                 return "eof before pack header was fully read";
1476
1477         case PH_ERROR_PACK_SIGNATURE:
1478                 return "protocol error (pack signature mismatch detected)";
1479
1480         case PH_ERROR_PROTOCOL:
1481                 return "protocol error (pack version unsupported)";
1482
1483         default:
1484                 return "unknown error in parse_pack_header";
1485
1486         case 0:
1487                 return NULL;
1488         }
1489 }
1490
1491 static const char *pack_lockfile;
1492
1493 static const char *unpack(int err_fd, struct shallow_info *si)
1494 {
1495         struct pack_header hdr;
1496         const char *hdr_err;
1497         int status;
1498         char hdr_arg[38];
1499         struct child_process child = CHILD_PROCESS_INIT;
1500         int fsck_objects = (receive_fsck_objects >= 0
1501                             ? receive_fsck_objects
1502                             : transfer_fsck_objects >= 0
1503                             ? transfer_fsck_objects
1504                             : 0);
1505
1506         hdr_err = parse_pack_header(&hdr);
1507         if (hdr_err) {
1508                 if (err_fd > 0)
1509                         close(err_fd);
1510                 return hdr_err;
1511         }
1512         snprintf(hdr_arg, sizeof(hdr_arg),
1513                         "--pack_header=%"PRIu32",%"PRIu32,
1514                         ntohl(hdr.hdr_version), ntohl(hdr.hdr_entries));
1515
1516         if (si->nr_ours || si->nr_theirs) {
1517                 alt_shallow_file = setup_temporary_shallow(si->shallow);
1518                 argv_array_push(&child.args, "--shallow-file");
1519                 argv_array_push(&child.args, alt_shallow_file);
1520         }
1521
1522         if (ntohl(hdr.hdr_entries) < unpack_limit) {
1523                 argv_array_pushl(&child.args, "unpack-objects", hdr_arg, NULL);
1524                 if (quiet)
1525                         argv_array_push(&child.args, "-q");
1526                 if (fsck_objects)
1527                         argv_array_pushf(&child.args, "--strict%s",
1528                                 fsck_msg_types.buf);
1529                 child.no_stdout = 1;
1530                 child.err = err_fd;
1531                 child.git_cmd = 1;
1532                 status = run_command(&child);
1533                 if (status)
1534                         return "unpack-objects abnormal exit";
1535         } else {
1536                 char hostname[256];
1537
1538                 argv_array_pushl(&child.args, "index-pack",
1539                                  "--stdin", hdr_arg, NULL);
1540
1541                 if (gethostname(hostname, sizeof(hostname)))
1542                         xsnprintf(hostname, sizeof(hostname), "localhost");
1543                 argv_array_pushf(&child.args,
1544                                  "--keep=receive-pack %"PRIuMAX" on %s",
1545                                  (uintmax_t)getpid(),
1546                                  hostname);
1547
1548                 if (fsck_objects)
1549                         argv_array_pushf(&child.args, "--strict%s",
1550                                 fsck_msg_types.buf);
1551                 if (fix_thin)
1552                         argv_array_push(&child.args, "--fix-thin");
1553                 child.out = -1;
1554                 child.err = err_fd;
1555                 child.git_cmd = 1;
1556                 status = start_command(&child);
1557                 if (status)
1558                         return "index-pack fork failed";
1559                 pack_lockfile = index_pack_lockfile(child.out);
1560                 close(child.out);
1561                 status = finish_command(&child);
1562                 if (status)
1563                         return "index-pack abnormal exit";
1564                 reprepare_packed_git();
1565         }
1566         return NULL;
1567 }
1568
1569 static const char *unpack_with_sideband(struct shallow_info *si)
1570 {
1571         struct async muxer;
1572         const char *ret;
1573
1574         if (!use_sideband)
1575                 return unpack(0, si);
1576
1577         memset(&muxer, 0, sizeof(muxer));
1578         muxer.proc = copy_to_sideband;
1579         muxer.in = -1;
1580         if (start_async(&muxer))
1581                 return NULL;
1582
1583         ret = unpack(muxer.in, si);
1584
1585         finish_async(&muxer);
1586         return ret;
1587 }
1588
1589 static void prepare_shallow_update(struct command *commands,
1590                                    struct shallow_info *si)
1591 {
1592         int i, j, k, bitmap_size = (si->ref->nr + 31) / 32;
1593
1594         ALLOC_ARRAY(si->used_shallow, si->shallow->nr);
1595         assign_shallow_commits_to_refs(si, si->used_shallow, NULL);
1596
1597         si->need_reachability_test =
1598                 xcalloc(si->shallow->nr, sizeof(*si->need_reachability_test));
1599         si->reachable =
1600                 xcalloc(si->shallow->nr, sizeof(*si->reachable));
1601         si->shallow_ref = xcalloc(si->ref->nr, sizeof(*si->shallow_ref));
1602
1603         for (i = 0; i < si->nr_ours; i++)
1604                 si->need_reachability_test[si->ours[i]] = 1;
1605
1606         for (i = 0; i < si->shallow->nr; i++) {
1607                 if (!si->used_shallow[i])
1608                         continue;
1609                 for (j = 0; j < bitmap_size; j++) {
1610                         if (!si->used_shallow[i][j])
1611                                 continue;
1612                         si->need_reachability_test[i]++;
1613                         for (k = 0; k < 32; k++)
1614                                 if (si->used_shallow[i][j] & (1U << k))
1615                                         si->shallow_ref[j * 32 + k]++;
1616                 }
1617
1618                 /*
1619                  * true for those associated with some refs and belong
1620                  * in "ours" list aka "step 7 not done yet"
1621                  */
1622                 si->need_reachability_test[i] =
1623                         si->need_reachability_test[i] > 1;
1624         }
1625
1626         /*
1627          * keep hooks happy by forcing a temporary shallow file via
1628          * env variable because we can't add --shallow-file to every
1629          * command. check_everything_connected() will be done with
1630          * true .git/shallow though.
1631          */
1632         setenv(GIT_SHALLOW_FILE_ENVIRONMENT, alt_shallow_file, 1);
1633 }
1634
1635 static void update_shallow_info(struct command *commands,
1636                                 struct shallow_info *si,
1637                                 struct sha1_array *ref)
1638 {
1639         struct command *cmd;
1640         int *ref_status;
1641         remove_nonexistent_theirs_shallow(si);
1642         if (!si->nr_ours && !si->nr_theirs) {
1643                 shallow_update = 0;
1644                 return;
1645         }
1646
1647         for (cmd = commands; cmd; cmd = cmd->next) {
1648                 if (is_null_sha1(cmd->new_sha1))
1649                         continue;
1650                 sha1_array_append(ref, cmd->new_sha1);
1651                 cmd->index = ref->nr - 1;
1652         }
1653         si->ref = ref;
1654
1655         if (shallow_update) {
1656                 prepare_shallow_update(commands, si);
1657                 return;
1658         }
1659
1660         ALLOC_ARRAY(ref_status, ref->nr);
1661         assign_shallow_commits_to_refs(si, NULL, ref_status);
1662         for (cmd = commands; cmd; cmd = cmd->next) {
1663                 if (is_null_sha1(cmd->new_sha1))
1664                         continue;
1665                 if (ref_status[cmd->index]) {
1666                         cmd->error_string = "shallow update not allowed";
1667                         cmd->skip_update = 1;
1668                 }
1669         }
1670         free(ref_status);
1671 }
1672
1673 static void report(struct command *commands, const char *unpack_status)
1674 {
1675         struct command *cmd;
1676         struct strbuf buf = STRBUF_INIT;
1677
1678         packet_buf_write(&buf, "unpack %s\n",
1679                          unpack_status ? unpack_status : "ok");
1680         for (cmd = commands; cmd; cmd = cmd->next) {
1681                 if (!cmd->error_string)
1682                         packet_buf_write(&buf, "ok %s\n",
1683                                          cmd->ref_name);
1684                 else
1685                         packet_buf_write(&buf, "ng %s %s\n",
1686                                          cmd->ref_name, cmd->error_string);
1687         }
1688         packet_buf_flush(&buf);
1689
1690         if (use_sideband)
1691                 send_sideband(1, 1, buf.buf, buf.len, use_sideband);
1692         else
1693                 write_or_die(1, buf.buf, buf.len);
1694         strbuf_release(&buf);
1695 }
1696
1697 static int delete_only(struct command *commands)
1698 {
1699         struct command *cmd;
1700         for (cmd = commands; cmd; cmd = cmd->next) {
1701                 if (!is_null_sha1(cmd->new_sha1))
1702                         return 0;
1703         }
1704         return 1;
1705 }
1706
1707 int cmd_receive_pack(int argc, const char **argv, const char *prefix)
1708 {
1709         int advertise_refs = 0;
1710         int i;
1711         struct command *commands;
1712         struct sha1_array shallow = SHA1_ARRAY_INIT;
1713         struct sha1_array ref = SHA1_ARRAY_INIT;
1714         struct shallow_info si;
1715
1716         packet_trace_identity("receive-pack");
1717
1718         argv++;
1719         for (i = 1; i < argc; i++) {
1720                 const char *arg = *argv++;
1721
1722                 if (*arg == '-') {
1723                         if (!strcmp(arg, "--quiet")) {
1724                                 quiet = 1;
1725                                 continue;
1726                         }
1727
1728                         if (!strcmp(arg, "--advertise-refs")) {
1729                                 advertise_refs = 1;
1730                                 continue;
1731                         }
1732                         if (!strcmp(arg, "--stateless-rpc")) {
1733                                 stateless_rpc = 1;
1734                                 continue;
1735                         }
1736                         if (!strcmp(arg, "--reject-thin-pack-for-testing")) {
1737                                 fix_thin = 0;
1738                                 continue;
1739                         }
1740
1741                         usage(receive_pack_usage);
1742                 }
1743                 if (service_dir)
1744                         usage(receive_pack_usage);
1745                 service_dir = arg;
1746         }
1747         if (!service_dir)
1748                 usage(receive_pack_usage);
1749
1750         setup_path();
1751
1752         if (!enter_repo(service_dir, 0))
1753                 die("'%s' does not appear to be a git repository", service_dir);
1754
1755         git_config(receive_pack_config, NULL);
1756         if (cert_nonce_seed)
1757                 push_cert_nonce = prepare_push_cert_nonce(service_dir, time(NULL));
1758
1759         if (0 <= transfer_unpack_limit)
1760                 unpack_limit = transfer_unpack_limit;
1761         else if (0 <= receive_unpack_limit)
1762                 unpack_limit = receive_unpack_limit;
1763
1764         if (advertise_refs || !stateless_rpc) {
1765                 write_head_info();
1766         }
1767         if (advertise_refs)
1768                 return 0;
1769
1770         if ((commands = read_head_info(&shallow)) != NULL) {
1771                 const char *unpack_status = NULL;
1772
1773                 prepare_shallow_info(&si, &shallow);
1774                 if (!si.nr_ours && !si.nr_theirs)
1775                         shallow_update = 0;
1776                 if (!delete_only(commands)) {
1777                         unpack_status = unpack_with_sideband(&si);
1778                         update_shallow_info(commands, &si, &ref);
1779                 }
1780                 execute_commands(commands, unpack_status, &si);
1781                 if (pack_lockfile)
1782                         unlink_or_warn(pack_lockfile);
1783                 if (report_status)
1784                         report(commands, unpack_status);
1785                 run_receive_hook(commands, "post-receive", 1);
1786                 run_update_post_hook(commands);
1787                 if (auto_gc) {
1788                         const char *argv_gc_auto[] = {
1789                                 "gc", "--auto", "--quiet", NULL,
1790                         };
1791                         int opt = RUN_GIT_CMD | RUN_COMMAND_STDOUT_TO_STDERR;
1792                         close_all_packs();
1793                         run_command_v_opt(argv_gc_auto, opt);
1794                 }
1795                 if (auto_update_server_info)
1796                         update_server_info(0);
1797                 clear_shallow_info(&si);
1798         }
1799         if (use_sideband)
1800                 packet_flush(1);
1801         sha1_array_clear(&shallow);
1802         sha1_array_clear(&ref);
1803         free((void *)push_cert_nonce);
1804         return 0;
1805 }