[thirdparty/git.git] / send-pack.c

#include "builtin.h"
#include "commit.h"
#include "refs.h"
#include "pkt-line.h"
#include "sideband.h"
#include "run-command.h"
#include "remote.h"
#include "connect.h"
#include "send-pack.h"
#include "quote.h"
#include "transport.h"
#include "version.h"
#include "sha1-array.h"
#include "gpg-interface.h"
#include "cache.h"

int option_parse_push_signed(const struct option *opt,
			     const char *arg, int unset)
{
	if (unset) {
		*(int *)(opt->value) = SEND_PACK_PUSH_CERT_NEVER;
		return 0;
	}
	switch (git_parse_maybe_bool(arg)) {
	case 1:
		*(int *)(opt->value) = SEND_PACK_PUSH_CERT_ALWAYS;
		return 0;
	case 0:
		*(int *)(opt->value) = SEND_PACK_PUSH_CERT_NEVER;
		return 0;
	}
	if (!strcasecmp("if-asked", arg)) {
		*(int *)(opt->value) = SEND_PACK_PUSH_CERT_IF_ASKED;
		return 0;
	}
	die("bad %s argument: %s", opt->long_name, arg);
}

static void feed_object(const unsigned char *sha1, FILE *fh, int negative)
{
	if (negative && !has_sha1_file(sha1))
		return;

	if (negative)
		putc('^', fh);
	fputs(sha1_to_hex(sha1), fh);
	putc('\n', fh);
}

/*
 * Make a pack stream and spit it out into file descriptor fd
 */
static int pack_objects(int fd, struct ref *refs, struct sha1_array *extra, struct send_pack_args *args)
{
	/*
	 * The child becomes pack-objects --revs; we feed
	 * the revision parameters to it via its stdin and
	 * let its stdout go back to the other end.
	 */
	const char *argv[] = {
		"pack-objects",
		"--all-progress-implied",
		"--revs",
		"--stdout",
		NULL,
		NULL,
		NULL,
		NULL,
		NULL,
		NULL,
	};
	struct child_process po = CHILD_PROCESS_INIT;
	FILE *po_in;
	int i;

	i = 4;
	if (args->use_thin_pack)
		argv[i++] = "--thin";
	if (args->use_ofs_delta)
		argv[i++] = "--delta-base-offset";
	if (args->quiet || !args->progress)
		argv[i++] = "-q";
	if (args->progress)
		argv[i++] = "--progress";
	if (is_repository_shallow())
		argv[i++] = "--shallow";
	po.argv = argv;
	po.in = -1;
	po.out = args->stateless_rpc ? -1 : fd;
	po.git_cmd = 1;
	if (start_command(&po))
		die_errno("git pack-objects failed");

	/*
	 * We feed the pack-objects we just spawned with revision
	 * parameters by writing to the pipe.
	 */
	po_in = xfdopen(po.in, "w");
	for (i = 0; i < extra->nr; i++)
		feed_object(extra->sha1[i], po_in, 1);

	while (refs) {
		if (!is_null_oid(&refs->old_oid))
			feed_object(refs->old_oid.hash, po_in, 1);
		if (!is_null_oid(&refs->new_oid))
			feed_object(refs->new_oid.hash, po_in, 0);
		refs = refs->next;
	}

	fflush(po_in);
	if (ferror(po_in))
		die_errno("error writing to pack-objects");
	fclose(po_in);

	if (args->stateless_rpc) {
		char *buf = xmalloc(LARGE_PACKET_MAX);
		while (1) {
			ssize_t n = xread(po.out, buf, LARGE_PACKET_MAX);
			if (n <= 0)
				break;
			send_sideband(fd, -1, buf, n, LARGE_PACKET_MAX);
		}
		free(buf);
		close(po.out);
		po.out = -1;
	}

	if (finish_command(&po))
		return -1;
	return 0;
}

static int receive_status(int in, struct ref *refs)
{
	struct ref *hint;
	int ret = 0;
	char *line = packet_read_line(in, NULL);
	if (!starts_with(line, "unpack "))
		return error("did not receive remote status");
	if (strcmp(line, "unpack ok")) {
		error("unpack failed: %s", line + 7);
		ret = -1;
	}
	hint = NULL;
	while (1) {
		char *refname;
		char *msg;
		line = packet_read_line(in, NULL);
		if (!line)
			break;
		if (!starts_with(line, "ok ") && !starts_with(line, "ng ")) {
			error("invalid ref status from remote: %s", line);
			ret = -1;
			break;
		}

		refname = line + 3;
		msg = strchr(refname, ' ');
		if (msg)
			*msg++ = '\0';

		/* first try searching at our hint, falling back to all refs */
		if (hint)
			hint = find_ref_by_name(hint, refname);
		if (!hint)
			hint = find_ref_by_name(refs, refname);
		if (!hint) {
			warning("remote reported status on unknown ref: %s",
					refname);
			continue;
		}
		if (hint->status != REF_STATUS_EXPECTING_REPORT) {
			warning("remote reported status on unexpected ref: %s",
					refname);
			continue;
		}

		if (line[0] == 'o' && line[1] == 'k')
			hint->status = REF_STATUS_OK;
		else {
			hint->status = REF_STATUS_REMOTE_REJECT;
			ret = -1;
		}
		if (msg)
			hint->remote_status = xstrdup(msg);
		/* start our next search from the next ref */
		hint = hint->next;
	}
	return ret;
}

static int sideband_demux(int in, int out, void *data)
{
	int *fd = data, ret;
#ifdef NO_PTHREADS
	close(fd[1]);
#endif
	ret = recv_sideband("send-pack", fd[0], out);
	close(out);
	return ret;
}

static int advertise_shallow_grafts_cb(const struct commit_graft *graft, void *cb)
{
	struct strbuf *sb = cb;
	if (graft->nr_parent == -1)
		packet_buf_write(sb, "shallow %s\n", oid_to_hex(&graft->oid));
	return 0;
}

static void advertise_shallow_grafts_buf(struct strbuf *sb)
{
	if (!is_repository_shallow())
		return;
	for_each_commit_graft(advertise_shallow_grafts_cb, sb);
}

#define CHECK_REF_NO_PUSH -1
#define CHECK_REF_STATUS_REJECTED -2
#define CHECK_REF_UPTODATE -3
static int check_to_send_update(const struct ref *ref, const struct send_pack_args *args)
{
	if (!ref->peer_ref && !args->send_mirror)
		return CHECK_REF_NO_PUSH;

	/* Check for statuses set by set_ref_status_for_push() */
	switch (ref->status) {
	case REF_STATUS_REJECT_NONFASTFORWARD:
	case REF_STATUS_REJECT_ALREADY_EXISTS:
	case REF_STATUS_REJECT_FETCH_FIRST:
	case REF_STATUS_REJECT_NEEDS_FORCE:
	case REF_STATUS_REJECT_STALE:
	case REF_STATUS_REJECT_NODELETE:
		return CHECK_REF_STATUS_REJECTED;
	case REF_STATUS_UPTODATE:
		return CHECK_REF_UPTODATE;
	default:
		return 0;
	}
}

/*
 * the beginning of the next line, or the end of buffer.
 *
 * NEEDSWORK: perhaps move this to git-compat-util.h or somewhere and
 * convert many similar uses found by "git grep -A4 memchr".
 */
static const char *next_line(const char *line, size_t len)
{
	const char *nl = memchr(line, '\n', len);
	if (!nl)
		return line + len; /* incomplete line */
	return nl + 1;
}

static int generate_push_cert(struct strbuf *req_buf,
			      const struct ref *remote_refs,
			      struct send_pack_args *args,
			      const char *cap_string,
			      const char *push_cert_nonce)
{
	const struct ref *ref;
	struct string_list_item *item;
	char *signing_key = xstrdup(get_signing_key());
	const char *cp, *np;
	struct strbuf cert = STRBUF_INIT;
	int update_seen = 0;

	strbuf_addstr(&cert, "certificate version 0.1\n");
	strbuf_addf(&cert, "pusher %s ", signing_key);
	datestamp(&cert);
	strbuf_addch(&cert, '\n');
	if (args->url && *args->url) {
		char *anon_url = transport_anonymize_url(args->url);
		strbuf_addf(&cert, "pushee %s\n", anon_url);
		free(anon_url);
	}
	if (push_cert_nonce[0])
		strbuf_addf(&cert, "nonce %s\n", push_cert_nonce);
	if (args->push_options)
		for_each_string_list_item(item, args->push_options)
			strbuf_addf(&cert, "push-option %s\n", item->string);
	strbuf_addstr(&cert, "\n");

	for (ref = remote_refs; ref; ref = ref->next) {
		if (check_to_send_update(ref, args) < 0)
			continue;
		update_seen = 1;
		strbuf_addf(&cert, "%s %s %s\n",
			    oid_to_hex(&ref->old_oid),
			    oid_to_hex(&ref->new_oid),
			    ref->name);
	}
	if (!update_seen)
		goto free_return;

	if (sign_buffer(&cert, &cert, signing_key))
		die(_("failed to sign the push certificate"));

	packet_buf_write(req_buf, "push-cert%c%s", 0, cap_string);
	for (cp = cert.buf; cp < cert.buf + cert.len; cp = np) {
		np = next_line(cp, cert.buf + cert.len - cp);
		packet_buf_write(req_buf,
				 "%.*s", (int)(np - cp), cp);
	}
	packet_buf_write(req_buf, "push-cert-end\n");

free_return:
	free(signing_key);
	strbuf_release(&cert);
	return update_seen;
}


static int atomic_push_failure(struct send_pack_args *args,
			       struct ref *remote_refs,
			       struct ref *failing_ref)
{
	struct ref *ref;
	/* Mark other refs as failed */
	for (ref = remote_refs; ref; ref = ref->next) {
		if (!ref->peer_ref && !args->send_mirror)
			continue;

		switch (ref->status) {
		case REF_STATUS_EXPECTING_REPORT:
			ref->status = REF_STATUS_ATOMIC_PUSH_FAILED;
			continue;
		default:
			break; /* do nothing */
		}
	}
	return error("atomic push failed for ref %s. status: %d\n",
		     failing_ref->name, failing_ref->status);
}

#define NONCE_LEN_LIMIT 256

static void reject_invalid_nonce(const char *nonce, int len)
{
	int i = 0;

	if (NONCE_LEN_LIMIT <= len)
		die("the receiving end asked to sign an invalid nonce <%.*s>",
		    len, nonce);

	for (i = 0; i < len; i++) {
		int ch = nonce[i] & 0xFF;
		if (isalnum(ch) ||
		    ch == '-' || ch == '.' ||
		    ch == '/' || ch == '+' ||
		    ch == '=' || ch == '_')
			continue;
		die("the receiving end asked to sign an invalid nonce <%.*s>",
		    len, nonce);
	}
}

int send_pack(struct send_pack_args *args,
	      int fd[], struct child_process *conn,
	      struct ref *remote_refs,
	      struct sha1_array *extra_have)
{
	int in = fd[0];
	int out = fd[1];
	struct strbuf req_buf = STRBUF_INIT;
	struct strbuf cap_buf = STRBUF_INIT;
	struct ref *ref;
	int need_pack_data = 0;
	int allow_deleting_refs = 0;
	int status_report = 0;
	int use_sideband = 0;
	int quiet_supported = 0;
	int agent_supported = 0;
	int use_atomic = 0;
	int atomic_supported = 0;
	int use_push_options = 0;
	int push_options_supported = 0;
	unsigned cmds_sent = 0;
	int ret;
	struct async demux;
	const char *push_cert_nonce = NULL;

	/* Does the other end support the reporting? */
	if (server_supports("report-status"))
		status_report = 1;
	if (server_supports("delete-refs"))
		allow_deleting_refs = 1;
	if (server_supports("ofs-delta"))
		args->use_ofs_delta = 1;
	if (server_supports("side-band-64k"))
		use_sideband = 1;
	if (server_supports("quiet"))
		quiet_supported = 1;
	if (server_supports("agent"))
		agent_supported = 1;
	if (server_supports("no-thin"))
		args->use_thin_pack = 0;
	if (server_supports("atomic"))
		atomic_supported = 1;
	if (server_supports("push-options"))
		push_options_supported = 1;

	if (args->push_cert != SEND_PACK_PUSH_CERT_NEVER) {
		int len;
		push_cert_nonce = server_feature_value("push-cert", &len);
		if (push_cert_nonce) {
			reject_invalid_nonce(push_cert_nonce, len);
			push_cert_nonce = xmemdupz(push_cert_nonce, len);
		} else if (args->push_cert == SEND_PACK_PUSH_CERT_ALWAYS) {
			die(_("the receiving end does not support --signed push"));
		} else if (args->push_cert == SEND_PACK_PUSH_CERT_IF_ASKED) {
			warning(_("not sending a push certificate since the"
				  " receiving end does not support --signed"
				  " push"));
		}
	}

	if (!remote_refs) {
		fprintf(stderr, "No refs in common and none specified; doing nothing.\n"
			"Perhaps you should specify a branch such as 'master'.\n");
		return 0;
	}
	if (args->atomic && !atomic_supported)
		die(_("the receiving end does not support --atomic push"));

	use_atomic = atomic_supported && args->atomic;

	if (args->push_options && !push_options_supported)
		die(_("the receiving end does not support push options"));

	use_push_options = push_options_supported && args->push_options;

	if (status_report)
		strbuf_addstr(&cap_buf, " report-status");
	if (use_sideband)
		strbuf_addstr(&cap_buf, " side-band-64k");
	if (quiet_supported && (args->quiet || !args->progress))
		strbuf_addstr(&cap_buf, " quiet");
	if (use_atomic)
		strbuf_addstr(&cap_buf, " atomic");
	if (use_push_options)
		strbuf_addstr(&cap_buf, " push-options");
	if (agent_supported)
		strbuf_addf(&cap_buf, " agent=%s", git_user_agent_sanitized());

	/*
	 * NEEDSWORK: why does delete-refs have to be so specific to
	 * send-pack machinery that set_ref_status_for_push() cannot
	 * set this bit for us???
	 */
	for (ref = remote_refs; ref; ref = ref->next)
		if (ref->deletion && !allow_deleting_refs)
			ref->status = REF_STATUS_REJECT_NODELETE;

	if (!args->dry_run)
		advertise_shallow_grafts_buf(&req_buf);

	if (!args->dry_run && push_cert_nonce)
		cmds_sent = generate_push_cert(&req_buf, remote_refs, args,
					       cap_buf.buf, push_cert_nonce);

	/*
	 * Clear the status for each ref and see if we need to send
	 * the pack data.
	 */
	for (ref = remote_refs; ref; ref = ref->next) {
		switch (check_to_send_update(ref, args)) {
		case 0: /* no error */
			break;
		case CHECK_REF_STATUS_REJECTED:
			/*
			 * When we know the server would reject a ref update if
			 * we were to send it and we're trying to send the refs
			 * atomically, abort the whole operation.
			 */
			if (use_atomic)
				return atomic_push_failure(args, remote_refs, ref);
			/* Fallthrough for non atomic case. */
		default:
			continue;
		}
		if (!ref->deletion)
			need_pack_data = 1;

		if (args->dry_run || !status_report)
			ref->status = REF_STATUS_OK;
		else
			ref->status = REF_STATUS_EXPECTING_REPORT;
	}

	/*
	 * Finally, tell the other end!
	 */
	for (ref = remote_refs; ref; ref = ref->next) {
		char *old_hex, *new_hex;

		if (args->dry_run || push_cert_nonce)
			continue;

		if (check_to_send_update(ref, args) < 0)
			continue;

		old_hex = oid_to_hex(&ref->old_oid);
		new_hex = oid_to_hex(&ref->new_oid);
		if (!cmds_sent) {
			packet_buf_write(&req_buf,
					 "%s %s %s%c%s",
					 old_hex, new_hex, ref->name, 0,
					 cap_buf.buf);
			cmds_sent = 1;
		} else {
			packet_buf_write(&req_buf, "%s %s %s",
					 old_hex, new_hex, ref->name);
		}
	}

	if (args->stateless_rpc) {
		if (!args->dry_run && (cmds_sent || is_repository_shallow())) {
			packet_buf_flush(&req_buf);
			send_sideband(out, -1, req_buf.buf, req_buf.len, LARGE_PACKET_MAX);
		}
	} else {
		write_or_die(out, req_buf.buf, req_buf.len);
		packet_flush(out);
	}
	strbuf_release(&req_buf);
	strbuf_release(&cap_buf);

	if (use_push_options) {
		struct string_list_item *item;
		struct strbuf sb = STRBUF_INIT;

		for_each_string_list_item(item, args->push_options)
			packet_buf_write(&sb, "%s", item->string);

		write_or_die(out, sb.buf, sb.len);
		packet_flush(out);
		strbuf_release(&sb);
	}

	if (use_sideband && cmds_sent) {
		memset(&demux, 0, sizeof(demux));
		demux.proc = sideband_demux;
		demux.data = fd;
		demux.out = -1;
		demux.isolate_sigpipe = 1;
		if (start_async(&demux))
			die("send-pack: unable to fork off sideband demultiplexer");
		in = demux.out;
	}

	if (need_pack_data && cmds_sent) {
		if (pack_objects(out, remote_refs, extra_have, args) < 0) {
			for (ref = remote_refs; ref; ref = ref->next)
				ref->status = REF_STATUS_NONE;
			if (args->stateless_rpc)
				close(out);
			if (git_connection_is_socket(conn))
				shutdown(fd[0], SHUT_WR);
			if (use_sideband) {
				close(demux.out);
				finish_async(&demux);
			}
			fd[1] = -1;
			return -1;
		}
		if (!args->stateless_rpc)
			/* Closed by pack_objects() via start_command() */
			fd[1] = -1;
	}
	if (args->stateless_rpc && cmds_sent)
		packet_flush(out);

	if (status_report && cmds_sent)
		ret = receive_status(in, remote_refs);
	else
		ret = 0;
	if (args->stateless_rpc)
		packet_flush(out);

	if (use_sideband && cmds_sent) {
		close(demux.out);
		if (finish_async(&demux)) {
			error("error in sideband demultiplexer");
			ret = -1;
		}
	}

	if (ret < 0)
		return ret;

	if (args->porcelain)
		return 0;

	for (ref = remote_refs; ref; ref = ref->next) {
		switch (ref->status) {
		case REF_STATUS_NONE:
		case REF_STATUS_UPTODATE:
		case REF_STATUS_OK:
			break;
		default:
			return -1;
		}
	}
	return 0;
}
Commit	Line	Data
f5d942e1 NTND	1	#include "builtin.h"
	2	#include "commit.h"
	3	#include "refs.h"
	4	#include "pkt-line.h"
	5	#include "sideband.h"
	6	#include "run-command.h"
	7	#include "remote.h"
47a59185	8	#include "connect.h"
f5d942e1 NTND	9	#include "send-pack.h"
	10	#include "quote.h"
	11	#include "transport.h"
	12	#include "version.h"
13eb4626	13	#include "sha1-array.h"
a85b377d	14	#include "gpg-interface.h"
30261094 DB	15	#include "cache.h"
	16
	17	int option_parse_push_signed(const struct option *opt,
	18	const char *arg, int unset)
	19	{
	20	if (unset) {
	21	(int )(opt->value) = SEND_PACK_PUSH_CERT_NEVER;
	22	return 0;
	23	}
	24	switch (git_parse_maybe_bool(arg)) {
	25	case 1:
	26	(int )(opt->value) = SEND_PACK_PUSH_CERT_ALWAYS;
	27	return 0;
	28	case 0:
	29	(int )(opt->value) = SEND_PACK_PUSH_CERT_NEVER;
	30	return 0;
	31	}
	32	if (!strcasecmp("if-asked", arg)) {
	33	(int )(opt->value) = SEND_PACK_PUSH_CERT_IF_ASKED;
	34	return 0;
	35	}
	36	die("bad %s argument: %s", opt->long_name, arg);
	37	}
f5d942e1	38
f0bca72d	39	static void feed_object(const unsigned char sha1, FILE fh, int negative)
f5d942e1	40	{
f5d942e1	41	if (negative && !has_sha1_file(sha1))
f0bca72d	42	return;
f5d942e1	43
f5d942e1	44	if (negative)
f0bca72d JK	45	putc('^', fh);
	46	fputs(sha1_to_hex(sha1), fh);
	47	putc('\n', fh);
f5d942e1 NTND	48	}
	49
	50	/*
	51	* Make a pack stream and spit it out into file descriptor fd
	52	*/
13eb4626	53	static int pack_objects(int fd, struct ref refs, struct sha1_array extra, struct send_pack_args *args)
f5d942e1 NTND	54	{
	55	/*
	56	* The child becomes pack-objects --revs; we feed
	57	* the revision parameters to it via its stdin and
	58	* let its stdout go back to the other end.
	59	*/
	60	const char *argv[] = {
	61	"pack-objects",
	62	"--all-progress-implied",
	63	"--revs",
	64	"--stdout",
	65	NULL,
	66	NULL,
	67	NULL,
	68	NULL,
	69	NULL,
2dacf26d	70	NULL,
f5d942e1	71	};
d3180279	72	struct child_process po = CHILD_PROCESS_INIT;
f0bca72d	73	FILE *po_in;
f5d942e1 NTND	74	int i;
	75
	76	i = 4;
	77	if (args->use_thin_pack)
	78	argv[i++] = "--thin";
	79	if (args->use_ofs_delta)
	80	argv[i++] = "--delta-base-offset";
	81	if (args->quiet \|\| !args->progress)
	82	argv[i++] = "-q";
	83	if (args->progress)
	84	argv[i++] = "--progress";
2dacf26d	85	if (is_repository_shallow())
2dacf26d	86	argv[i++] = "--shallow";
f5d942e1 NTND	87	po.argv = argv;
	88	po.in = -1;
	89	po.out = args->stateless_rpc ? -1 : fd;
	90	po.git_cmd = 1;
	91	if (start_command(&po))
	92	die_errno("git pack-objects failed");
	93
	94	/*
	95	* We feed the pack-objects we just spawned with revision
	96	* parameters by writing to the pipe.
	97	*/
f0bca72d	98	po_in = xfdopen(po.in, "w");
f5d942e1	99	for (i = 0; i < extra->nr; i++)
f0bca72d	100	feed_object(extra->sha1[i], po_in, 1);
f5d942e1 NTND	101
f5d942e1 NTND	102	while (refs) {
f0bca72d JK	103	if (!is_null_oid(&refs->old_oid))
	104	feed_object(refs->old_oid.hash, po_in, 1);
	105	if (!is_null_oid(&refs->new_oid))
	106	feed_object(refs->new_oid.hash, po_in, 0);
f5d942e1 NTND	107	refs = refs->next;
	108	}
	109
f0bca72d JK	110	fflush(po_in);
	111	if (ferror(po_in))
	112	die_errno("error writing to pack-objects");
	113	fclose(po_in);
f5d942e1 NTND	114
	115	if (args->stateless_rpc) {
	116	char *buf = xmalloc(LARGE_PACKET_MAX);
	117	while (1) {
	118	ssize_t n = xread(po.out, buf, LARGE_PACKET_MAX);
	119	if (n <= 0)
	120	break;
	121	send_sideband(fd, -1, buf, n, LARGE_PACKET_MAX);
	122	}
	123	free(buf);
	124	close(po.out);
	125	po.out = -1;
	126	}
	127
	128	if (finish_command(&po))
	129	return -1;
	130	return 0;
	131	}
	132
	133	static int receive_status(int in, struct ref *refs)
	134	{
	135	struct ref *hint;
f5d942e1	136	int ret = 0;
74543a04	137	char *line = packet_read_line(in, NULL);
59556548	138	if (!starts_with(line, "unpack "))
f5d942e1	139	return error("did not receive remote status");
819b929d	140	if (strcmp(line, "unpack ok")) {
f5d942e1 NTND	141	error("unpack failed: %s", line + 7);
	142	ret = -1;
	143	}
	144	hint = NULL;
	145	while (1) {
	146	char *refname;
	147	char *msg;
74543a04 JK	148	line = packet_read_line(in, NULL);
74543a04 JK	149	if (!line)
f5d942e1	150	break;
59556548	151	if (!starts_with(line, "ok ") && !starts_with(line, "ng ")) {
8f9e3e49	152	error("invalid ref status from remote: %s", line);
f5d942e1 NTND	153	ret = -1;
	154	break;
	155	}
	156
f5d942e1 NTND	157	refname = line + 3;
	158	msg = strchr(refname, ' ');
	159	if (msg)
	160	*msg++ = '\0';
	161
	162	/* first try searching at our hint, falling back to all refs */
	163	if (hint)
	164	hint = find_ref_by_name(hint, refname);
	165	if (!hint)
	166	hint = find_ref_by_name(refs, refname);
	167	if (!hint) {
	168	warning("remote reported status on unknown ref: %s",
	169	refname);
	170	continue;
	171	}
	172	if (hint->status != REF_STATUS_EXPECTING_REPORT) {
	173	warning("remote reported status on unexpected ref: %s",
	174	refname);
	175	continue;
	176	}
	177
	178	if (line[0] == 'o' && line[1] == 'k')
	179	hint->status = REF_STATUS_OK;
	180	else {
	181	hint->status = REF_STATUS_REMOTE_REJECT;
	182	ret = -1;
	183	}
	184	if (msg)
	185	hint->remote_status = xstrdup(msg);
	186	/* start our next search from the next ref */
	187	hint = hint->next;
	188	}
	189	return ret;
	190	}
	191
	192	static int sideband_demux(int in, int out, void *data)
	193	{
	194	int *fd = data, ret;
	195	#ifdef NO_PTHREADS
	196	close(fd[1]);
	197	#endif
	198	ret = recv_sideband("send-pack", fd[0], out);
	199	close(out);
	200	return ret;
	201	}
	202
f2c681cf NTND	203	static int advertise_shallow_grafts_cb(const struct commit_graft graft, void cb)
	204	{
	205	struct strbuf *sb = cb;
	206	if (graft->nr_parent == -1)
7683e2e6	207	packet_buf_write(sb, "shallow %s\n", oid_to_hex(&graft->oid));
f2c681cf NTND	208	return 0;
	209	}
	210
16a2743c	211	static void advertise_shallow_grafts_buf(struct strbuf *sb)
f2c681cf NTND	212	{
	213	if (!is_repository_shallow())
	214	return;
	215	for_each_commit_graft(advertise_shallow_grafts_cb, sb);
	216	}
	217
7582e939 SB	218	#define CHECK_REF_NO_PUSH -1
	219	#define CHECK_REF_STATUS_REJECTED -2
	220	#define CHECK_REF_UPTODATE -3
	221	static int check_to_send_update(const struct ref ref, const struct send_pack_args args)
e40671a3 JH	222	{
e40671a3 JH	223	if (!ref->peer_ref && !args->send_mirror)
7582e939	224	return CHECK_REF_NO_PUSH;
e40671a3 JH	225
	226	/* Check for statuses set by set_ref_status_for_push() */
	227	switch (ref->status) {
	228	case REF_STATUS_REJECT_NONFASTFORWARD:
	229	case REF_STATUS_REJECT_ALREADY_EXISTS:
	230	case REF_STATUS_REJECT_FETCH_FIRST:
	231	case REF_STATUS_REJECT_NEEDS_FORCE:
	232	case REF_STATUS_REJECT_STALE:
	233	case REF_STATUS_REJECT_NODELETE:
7582e939	234	return CHECK_REF_STATUS_REJECTED;
e40671a3	235	case REF_STATUS_UPTODATE:
7582e939	236	return CHECK_REF_UPTODATE;
e40671a3	237	default:
7582e939	238	return 0;
e40671a3 JH	239	}
	240	}
	241
a85b377d JH	242	/*
	243	* the beginning of the next line, or the end of buffer.
	244	*
	245	* NEEDSWORK: perhaps move this to git-compat-util.h or somewhere and
	246	* convert many similar uses found by "git grep -A4 memchr".
	247	*/
	248	static const char next_line(const char line, size_t len)
	249	{
	250	const char *nl = memchr(line, '\n', len);
	251	if (!nl)
	252	return line + len; /* incomplete line */
	253	return nl + 1;
	254	}
	255
20a7558f JH	256	static int generate_push_cert(struct strbuf *req_buf,
	257	const struct ref *remote_refs,
	258	struct send_pack_args *args,
b89363e4 JH	259	const char *cap_string,
b89363e4 JH	260	const char *push_cert_nonce)
a85b377d JH	261	{
a85b377d JH	262	const struct ref *ref;
f6a4e61f	263	struct string_list_item *item;
a85b377d JH	264	char *signing_key = xstrdup(get_signing_key());
	265	const char cp, np;
	266	struct strbuf cert = STRBUF_INIT;
	267	int update_seen = 0;
	268
02962d36	269	strbuf_addstr(&cert, "certificate version 0.1\n");
fb06b528 JH	270	strbuf_addf(&cert, "pusher %s ", signing_key);
	271	datestamp(&cert);
	272	strbuf_addch(&cert, '\n');
9be89160 JH	273	if (args->url && *args->url) {
	274	char *anon_url = transport_anonymize_url(args->url);
	275	strbuf_addf(&cert, "pushee %s\n", anon_url);
	276	free(anon_url);
	277	}
b89363e4 JH	278	if (push_cert_nonce[0])
b89363e4 JH	279	strbuf_addf(&cert, "nonce %s\n", push_cert_nonce);
f6a4e61f SB	280	if (args->push_options)
	281	for_each_string_list_item(item, args->push_options)
	282	strbuf_addf(&cert, "push-option %s\n", item->string);
a85b377d JH	283	strbuf_addstr(&cert, "\n");
	284
	285	for (ref = remote_refs; ref; ref = ref->next) {
7582e939	286	if (check_to_send_update(ref, args) < 0)
a85b377d JH	287	continue;
	288	update_seen = 1;
	289	strbuf_addf(&cert, "%s %s %s\n",
f4e54d02	290	oid_to_hex(&ref->old_oid),
f4e54d02	291	oid_to_hex(&ref->new_oid),
a85b377d JH	292	ref->name);
	293	}
	294	if (!update_seen)
	295	goto free_return;
	296
	297	if (sign_buffer(&cert, &cert, signing_key))
	298	die(_("failed to sign the push certificate"));
	299
20a7558f	300	packet_buf_write(req_buf, "push-cert%c%s", 0, cap_string);
a85b377d JH	301	for (cp = cert.buf; cp < cert.buf + cert.len; cp = np) {
	302	np = next_line(cp, cert.buf + cert.len - cp);
	303	packet_buf_write(req_buf,
	304	"%.*s", (int)(np - cp), cp);
	305	}
	306	packet_buf_write(req_buf, "push-cert-end\n");
	307
	308	free_return:
	309	free(signing_key);
	310	strbuf_release(&cert);
20a7558f	311	return update_seen;
a85b377d JH	312	}
a85b377d JH	313
4ff17f10 RS	314
	315	static int atomic_push_failure(struct send_pack_args *args,
	316	struct ref *remote_refs,
	317	struct ref *failing_ref)
	318	{
	319	struct ref *ref;
	320	/* Mark other refs as failed */
	321	for (ref = remote_refs; ref; ref = ref->next) {
	322	if (!ref->peer_ref && !args->send_mirror)
	323	continue;
	324
	325	switch (ref->status) {
	326	case REF_STATUS_EXPECTING_REPORT:
	327	ref->status = REF_STATUS_ATOMIC_PUSH_FAILED;
	328	continue;
	329	default:
	330	break; /* do nothing */
	331	}
	332	}
	333	return error("atomic push failed for ref %s. status: %d\n",
	334	failing_ref->name, failing_ref->status);
	335	}
	336
afcb6ee3 JH	337	#define NONCE_LEN_LIMIT 256
	338
	339	static void reject_invalid_nonce(const char *nonce, int len)
	340	{
	341	int i = 0;
	342
	343	if (NONCE_LEN_LIMIT <= len)
	344	die("the receiving end asked to sign an invalid nonce <%.*s>",
	345	len, nonce);
	346
	347	for (i = 0; i < len; i++) {
	348	int ch = nonce[i] & 0xFF;
	349	if (isalnum(ch) \|\|
	350	ch == '-' \|\| ch == '.' \|\|
	351	ch == '/' \|\| ch == '+' \|\|
	352	ch == '=' \|\| ch == '_')
	353	continue;
	354	die("the receiving end asked to sign an invalid nonce <%.*s>",
	355	len, nonce);
	356	}
	357	}
	358
f5d942e1 NTND	359	int send_pack(struct send_pack_args *args,
	360	int fd[], struct child_process *conn,
	361	struct ref *remote_refs,
13eb4626	362	struct sha1_array *extra_have)
f5d942e1 NTND	363	{
	364	int in = fd[0];
	365	int out = fd[1];
	366	struct strbuf req_buf = STRBUF_INIT;
887f3533	367	struct strbuf cap_buf = STRBUF_INIT;
f5d942e1	368	struct ref *ref;
ab2b0c90	369	int need_pack_data = 0;
f5d942e1 NTND	370	int allow_deleting_refs = 0;
	371	int status_report = 0;
	372	int use_sideband = 0;
	373	int quiet_supported = 0;
	374	int agent_supported = 0;
4ff17f10 RS	375	int use_atomic = 0;
4ff17f10 RS	376	int atomic_supported = 0;
f6a4e61f SB	377	int use_push_options = 0;
f6a4e61f SB	378	int push_options_supported = 0;
f5d942e1 NTND	379	unsigned cmds_sent = 0;
	380	int ret;
	381	struct async demux;
b89363e4	382	const char *push_cert_nonce = NULL;
f5d942e1 NTND	383
	384	/* Does the other end support the reporting? */
	385	if (server_supports("report-status"))
	386	status_report = 1;
	387	if (server_supports("delete-refs"))
	388	allow_deleting_refs = 1;
	389	if (server_supports("ofs-delta"))
	390	args->use_ofs_delta = 1;
	391	if (server_supports("side-band-64k"))
	392	use_sideband = 1;
	393	if (server_supports("quiet"))
	394	quiet_supported = 1;
	395	if (server_supports("agent"))
	396	agent_supported = 1;
1ba98a79 CMN	397	if (server_supports("no-thin"))
1ba98a79 CMN	398	args->use_thin_pack = 0;
4ff17f10 RS	399	if (server_supports("atomic"))
4ff17f10 RS	400	atomic_supported = 1;
f6a4e61f SB	401	if (server_supports("push-options"))
f6a4e61f SB	402	push_options_supported = 1;
b89363e4	403
30261094 DB	404	if (args->push_cert != SEND_PACK_PUSH_CERT_NEVER) {
30261094 DB	405	int len;
b89363e4	406	push_cert_nonce = server_feature_value("push-cert", &len);
30261094 DB	407	if (push_cert_nonce) {
	408	reject_invalid_nonce(push_cert_nonce, len);
	409	push_cert_nonce = xmemdupz(push_cert_nonce, len);
	410	} else if (args->push_cert == SEND_PACK_PUSH_CERT_ALWAYS) {
b89363e4	411	die(_("the receiving end does not support --signed push"));
30261094 DB	412	} else if (args->push_cert == SEND_PACK_PUSH_CERT_IF_ASKED) {
	413	warning(_("not sending a push certificate since the"
	414	" receiving end does not support --signed"
	415	" push"));
	416	}
b89363e4	417	}
f5d942e1 NTND	418
	419	if (!remote_refs) {
	420	fprintf(stderr, "No refs in common and none specified; doing nothing.\n"
	421	"Perhaps you should specify a branch such as 'master'.\n");
	422	return 0;
	423	}
4ff17f10	424	if (args->atomic && !atomic_supported)
c8b8f22a	425	die(_("the receiving end does not support --atomic push"));
4ff17f10 RS	426
4ff17f10 RS	427	use_atomic = atomic_supported && args->atomic;
f5d942e1	428
f6a4e61f SB	429	if (args->push_options && !push_options_supported)
	430	die(_("the receiving end does not support push options"));
	431
	432	use_push_options = push_options_supported && args->push_options;
	433
887f3533 JH	434	if (status_report)
	435	strbuf_addstr(&cap_buf, " report-status");
	436	if (use_sideband)
	437	strbuf_addstr(&cap_buf, " side-band-64k");
	438	if (quiet_supported && (args->quiet \|\| !args->progress))
	439	strbuf_addstr(&cap_buf, " quiet");
4ff17f10 RS	440	if (use_atomic)
4ff17f10 RS	441	strbuf_addstr(&cap_buf, " atomic");
f6a4e61f SB	442	if (use_push_options)
f6a4e61f SB	443	strbuf_addstr(&cap_buf, " push-options");
887f3533 JH	444	if (agent_supported)
	445	strbuf_addf(&cap_buf, " agent=%s", git_user_agent_sanitized());
	446
621b0599 JH	447	/*
	448	* NEEDSWORK: why does delete-refs have to be so specific to
	449	* send-pack machinery that set_ref_status_for_push() cannot
	450	* set this bit for us???
	451	*/
	452	for (ref = remote_refs; ref; ref = ref->next)
	453	if (ref->deletion && !allow_deleting_refs)
	454	ref->status = REF_STATUS_REJECT_NODELETE;
	455
5dbd7676	456	if (!args->dry_run)
f2c681cf	457	advertise_shallow_grafts_buf(&req_buf);
5dbd7676	458
30261094	459	if (!args->dry_run && push_cert_nonce)
20a7558f	460	cmds_sent = generate_push_cert(&req_buf, remote_refs, args,
b89363e4	461	cap_buf.buf, push_cert_nonce);
a85b377d	462
f5d942e1	463	/*
b783aa71 JH	464	* Clear the status for each ref and see if we need to send
b783aa71 JH	465	* the pack data.
f5d942e1	466	*/
f5d942e1	467	for (ref = remote_refs; ref; ref = ref->next) {
4ff17f10 RS	468	switch (check_to_send_update(ref, args)) {
	469	case 0: /* no error */
	470	break;
	471	case CHECK_REF_STATUS_REJECTED:
	472	/*
	473	* When we know the server would reject a ref update if
	474	* we were to send it and we're trying to send the refs
	475	* atomically, abort the whole operation.
	476	*/
	477	if (use_atomic)
	478	return atomic_push_failure(args, remote_refs, ref);
	479	/* Fallthrough for non atomic case. */
	480	default:
f5d942e1	481	continue;
4ff17f10	482	}
f5d942e1	483	if (!ref->deletion)
ab2b0c90	484	need_pack_data = 1;
f5d942e1	485
b783aa71	486	if (args->dry_run \|\| !status_report)
f5d942e1	487	ref->status = REF_STATUS_OK;
b783aa71 JH	488	else
	489	ref->status = REF_STATUS_EXPECTING_REPORT;
	490	}
	491
	492	/*
	493	* Finally, tell the other end!
	494	*/
	495	for (ref = remote_refs; ref; ref = ref->next) {
	496	char old_hex, new_hex;
	497
30261094	498	if (args->dry_run \|\| push_cert_nonce)
f5d942e1	499	continue;
f5d942e1	500
7582e939	501	if (check_to_send_update(ref, args) < 0)
b783aa71	502	continue;
f5d942e1	503
f4e54d02	504	old_hex = oid_to_hex(&ref->old_oid);
f4e54d02	505	new_hex = oid_to_hex(&ref->new_oid);
c67072b9	506	if (!cmds_sent) {
b783aa71 JH	507	packet_buf_write(&req_buf,
	508	"%s %s %s%c%s",
	509	old_hex, new_hex, ref->name, 0,
	510	cap_buf.buf);
c67072b9	511	cmds_sent = 1;
f5d942e1	512	} else {
b783aa71 JH	513	packet_buf_write(&req_buf, "%s %s %s",
b783aa71 JH	514	old_hex, new_hex, ref->name);
f5d942e1 NTND	515	}
	516	}
	517
	518	if (args->stateless_rpc) {
f2c681cf	519	if (!args->dry_run && (cmds_sent \|\| is_repository_shallow())) {
f5d942e1 NTND	520	packet_buf_flush(&req_buf);
	521	send_sideband(out, -1, req_buf.buf, req_buf.len, LARGE_PACKET_MAX);
	522	}
	523	} else {
cdf4fb8e	524	write_or_die(out, req_buf.buf, req_buf.len);
f5d942e1 NTND	525	packet_flush(out);
	526	}
	527	strbuf_release(&req_buf);
887f3533	528	strbuf_release(&cap_buf);
f5d942e1	529
f6a4e61f SB	530	if (use_push_options) {
	531	struct string_list_item *item;
	532	struct strbuf sb = STRBUF_INIT;
	533
	534	for_each_string_list_item(item, args->push_options)
	535	packet_buf_write(&sb, "%s", item->string);
	536
	537	write_or_die(out, sb.buf, sb.len);
	538	packet_flush(out);
	539	strbuf_release(&sb);
	540	}
	541
f5d942e1 NTND	542	if (use_sideband && cmds_sent) {
	543	memset(&demux, 0, sizeof(demux));
	544	demux.proc = sideband_demux;
	545	demux.data = fd;
	546	demux.out = -1;
3e8b06d0	547	demux.isolate_sigpipe = 1;
f5d942e1 NTND	548	if (start_async(&demux))
	549	die("send-pack: unable to fork off sideband demultiplexer");
	550	in = demux.out;
	551	}
	552
ab2b0c90	553	if (need_pack_data && cmds_sent) {
f5d942e1 NTND	554	if (pack_objects(out, remote_refs, extra_have, args) < 0) {
	555	for (ref = remote_refs; ref; ref = ref->next)
	556	ref->status = REF_STATUS_NONE;
	557	if (args->stateless_rpc)
	558	close(out);
	559	if (git_connection_is_socket(conn))
	560	shutdown(fd[0], SHUT_WR);
739cf491 JK	561	if (use_sideband) {
739cf491 JK	562	close(demux.out);
f5d942e1	563	finish_async(&demux);
739cf491	564	}
37cb1dd6	565	fd[1] = -1;
f5d942e1 NTND	566	return -1;
f5d942e1 NTND	567	}
37cb1dd6 JL	568	if (!args->stateless_rpc)
	569	/* Closed by pack_objects() via start_command() */
	570	fd[1] = -1;
f5d942e1 NTND	571	}
	572	if (args->stateless_rpc && cmds_sent)
	573	packet_flush(out);
	574
	575	if (status_report && cmds_sent)
	576	ret = receive_status(in, remote_refs);
	577	else
	578	ret = 0;
	579	if (args->stateless_rpc)
	580	packet_flush(out);
	581
	582	if (use_sideband && cmds_sent) {
739cf491	583	close(demux.out);
f5d942e1 NTND	584	if (finish_async(&demux)) {
	585	error("error in sideband demultiplexer");
	586	ret = -1;
	587	}
f5d942e1 NTND	588	}
	589
	590	if (ret < 0)
	591	return ret;
	592
	593	if (args->porcelain)
	594	return 0;
	595
	596	for (ref = remote_refs; ref; ref = ref->next) {
	597	switch (ref->status) {
	598	case REF_STATUS_NONE:
	599	case REF_STATUS_UPTODATE:
	600	case REF_STATUS_OK:
	601	break;
	602	default:
	603	return -1;
	604	}
	605	}
	606	return 0;
	607	}