[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"

static int feed_object(const unsigned char *sha1, int fd, int negative)
{
	char buf[42];

	if (negative && !has_sha1_file(sha1))
		return 1;

	memcpy(buf + negative, sha1_to_hex(sha1), 40);
	if (negative)
		buf[0] = '^';
	buf[40 + negative] = '\n';
	return write_or_whine(fd, buf, 41 + negative, "send-pack: send refs");
}

/*
 * 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;
	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.
	 */
	for (i = 0; i < extra->nr; i++)
		if (!feed_object(extra->sha1[i], po.in, 1))
			break;

	while (refs) {
		if (!is_null_sha1(refs->old_sha1) &&
		    !feed_object(refs->old_sha1, po.in, 1))
			break;
		if (!is_null_sha1(refs->new_sha1) &&
		    !feed_object(refs->new_sha1, po.in, 0))
			break;
		refs = refs->next;
	}

	close(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;
	char *signing_key = xstrdup(get_signing_key());
	const char *cp, *np;
	struct strbuf cert = STRBUF_INIT;
	int update_seen = 0;

	strbuf_addf(&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);
	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",
			    sha1_to_hex(ref->old_sha1),
			    sha1_to_hex(ref->new_sha1),
			    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;
	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 (args->push_cert) {
		int len;

		push_cert_nonce = server_feature_value("push-cert", &len);
		if (!push_cert_nonce)
			die(_("the receiving end does not support --signed push"));
		reject_invalid_nonce(push_cert_nonce, len);
		push_cert_nonce = xmemdupz(push_cert_nonce, len);
	}

	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 (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 (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 && args->push_cert)
		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 || args->push_cert)
			continue;

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

		old_hex = sha1_to_hex(ref->old_sha1);
		new_hex = sha1_to_hex(ref->new_sha1);
		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_sideband && cmds_sent) {
		memset(&demux, 0, sizeof(demux));
		demux.proc = sideband_demux;
		demux.data = fd;
		demux.out = -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)
				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) {
		if (finish_async(&demux)) {
			error("error in sideband demultiplexer");
			ret = -1;
		}
		close(demux.out);
	}

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