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

#include "builtin.h"
#include "config.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 oid_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.
	 */
	struct child_process po = CHILD_PROCESS_INIT;
	FILE *po_in;
	int i;
	int rc;

	argv_array_push(&po.args, "pack-objects");
	argv_array_push(&po.args, "--all-progress-implied");
	argv_array_push(&po.args, "--revs");
	argv_array_push(&po.args, "--stdout");
	if (args->use_thin_pack)
		argv_array_push(&po.args, "--thin");
	if (args->use_ofs_delta)
		argv_array_push(&po.args, "--delta-base-offset");
	if (args->quiet || !args->progress)
		argv_array_push(&po.args, "-q");
	if (args->progress)
		argv_array_push(&po.args, "--progress");
	if (is_repository_shallow())
		argv_array_push(&po.args, "--shallow");
	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->oid[i].hash, 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;
	}

	rc = finish_command(&po);
	if (rc) {
		/*
		 * For a normal non-zero exit, we assume pack-objects wrote
		 * something useful to stderr. For death by signal, though,
		 * we should mention it to the user. The exception is SIGPIPE
		 * (141), because that's a normal occurrence if the remote end
		 * hangs up (and we'll report that by trying to read the unpack
		 * status).
		 */
		if (rc > 128 && rc != 141)
			error("pack-objects died of signal %d", rc - 128);
		return -1;
	}
	return 0;
}

static int receive_unpack_status(int in)
{
	const char *line = packet_read_line(in, NULL);
	if (!line)
		return error(_("unexpected flush packet while reading remote unpack status"));
	if (!skip_prefix(line, "unpack ", &line))
		return error(_("unable to parse remote unpack status: %s"), line);
	if (strcmp(line, "ok"))
		return error(_("remote unpack failed: %s"), line);
	return 0;
}

static int receive_status(int in, struct ref *refs)
{
	struct ref *hint;
	int ret;

	hint = NULL;
	ret = receive_unpack_status(in);
	while (1) {
		char *refname;
		char *msg;
		char *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;
		}
		hint->remote_status = xstrdup_or_null(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 oid_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) {
				strbuf_release(&req_buf);
				strbuf_release(&cap_buf);
				return atomic_push_failure(args, remote_refs, ref);
			}
			/* else fallthrough */
		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 (use_push_options) {
		struct string_list_item *item;

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

	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;
		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);

			/*
			 * Do not even bother with the return value; we know we
			 * are failing, and just want the error() side effects.
			 */
			if (status_report)
				receive_unpack_status(in);

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