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

#include <signal.h>
#include <sys/wait.h>
#include <sys/poll.h>
#include "cache.h"
#include "refs.h"
#include "pkt-line.h"
#include "sideband.h"
#include "tag.h"
#include "object.h"
#include "commit.h"
#include "exec_cmd.h"

static const char upload_pack_usage[] = "git-upload-pack [--strict] [--timeout=nn] <dir>";

#define THEY_HAVE (1U << 0)
#define OUR_REF (1U << 1)
#define WANTED (1U << 2)
static int multi_ack, nr_our_refs;
static int use_thin_pack, use_ofs_delta;
static struct object_array have_obj;
static struct object_array want_obj;
static unsigned int timeout;
/* 0 for no sideband,
 * otherwise maximum packet size (up to 65520 bytes).
 */
static int use_sideband;

static void reset_timeout(void)
{
	alarm(timeout);
}

static int strip(char *line, int len)
{
	if (len && line[len-1] == '\n')
		line[--len] = 0;
	return len;
}

static ssize_t send_client_data(int fd, const char *data, ssize_t sz)
{
	if (use_sideband)
		return send_sideband(1, fd, data, sz, use_sideband);
	if (fd == 3)
		/* emergency quit */
		fd = 2;
	if (fd == 2) {
		xwrite(fd, data, sz);
		return sz;
	}
	return safe_write(fd, data, sz);
}

static void create_pack_file(void)
{
	/* Pipes between rev-list to pack-objects, pack-objects to us
	 * and pack-objects error stream for progress bar.
	 */
	int lp_pipe[2], pu_pipe[2], pe_pipe[2];
	pid_t pid_rev_list, pid_pack_objects;
	int create_full_pack = (nr_our_refs == want_obj.nr && !have_obj.nr);
	char data[8193], progress[128];
	char abort_msg[] = "aborting due to possible repository "
		"corruption on the remote side.";
	int buffered = -1;

	if (pipe(lp_pipe) < 0)
		die("git-upload-pack: unable to create pipe");
	pid_rev_list = fork();
	if (pid_rev_list < 0)
		die("git-upload-pack: unable to fork git-rev-list");

	if (!pid_rev_list) {
		int i;
		int args;
		const char **argv;
		const char **p;
		char *buf;

		if (create_full_pack) {
			args = 10;
			use_thin_pack = 0; /* no point doing it */
		}
		else
			args = have_obj.nr + want_obj.nr + 5;
		p = xmalloc(args * sizeof(char *));
		argv = (const char **) p;
		buf = xmalloc(args * 45);

		dup2(lp_pipe[1], 1);
		close(0);
		close(lp_pipe[0]);
		close(lp_pipe[1]);
		*p++ = "rev-list";
		*p++ = use_thin_pack ? "--objects-edge" : "--objects";
		if (create_full_pack)
			*p++ = "--all";
		else {
			for (i = 0; i < want_obj.nr; i++) {
				struct object *o = want_obj.objects[i].item;
				*p++ = buf;
				memcpy(buf, sha1_to_hex(o->sha1), 41);
				buf += 41;
			}
		}
		if (!create_full_pack)
			for (i = 0; i < have_obj.nr; i++) {
				struct object *o = have_obj.objects[i].item;
				*p++ = buf;
				*buf++ = '^';
				memcpy(buf, sha1_to_hex(o->sha1), 41);
				buf += 41;
			}
		*p++ = NULL;
		execv_git_cmd(argv);
		die("git-upload-pack: unable to exec git-rev-list");
	}

	if (pipe(pu_pipe) < 0)
		die("git-upload-pack: unable to create pipe");
	if (pipe(pe_pipe) < 0)
		die("git-upload-pack: unable to create pipe");
	pid_pack_objects = fork();
	if (pid_pack_objects < 0) {
		/* daemon sets things up to ignore TERM */
		kill(pid_rev_list, SIGKILL);
		die("git-upload-pack: unable to fork git-pack-objects");
	}
	if (!pid_pack_objects) {
		dup2(lp_pipe[0], 0);
		dup2(pu_pipe[1], 1);
		dup2(pe_pipe[1], 2);

		close(lp_pipe[0]);
		close(lp_pipe[1]);
		close(pu_pipe[0]);
		close(pu_pipe[1]);
		close(pe_pipe[0]);
		close(pe_pipe[1]);
		execl_git_cmd("pack-objects", "--stdout", "--progress",
			      use_ofs_delta ? "--delta-base-offset" : NULL,
			      NULL);
		kill(pid_rev_list, SIGKILL);
		die("git-upload-pack: unable to exec git-pack-objects");
	}

	close(lp_pipe[0]);
	close(lp_pipe[1]);

	/* We read from pe_pipe[0] to capture stderr output for
	 * progress bar, and pu_pipe[0] to capture the pack data.
	 */
	close(pe_pipe[1]);
	close(pu_pipe[1]);

	while (1) {
		const char *who;
		struct pollfd pfd[2];
		pid_t pid;
		int status;
		ssize_t sz;
		int pe, pu, pollsize;

		reset_timeout();

		pollsize = 0;
		pe = pu = -1;

		if (0 <= pu_pipe[0]) {
			pfd[pollsize].fd = pu_pipe[0];
			pfd[pollsize].events = POLLIN;
			pu = pollsize;
			pollsize++;
		}
		if (0 <= pe_pipe[0]) {
			pfd[pollsize].fd = pe_pipe[0];
			pfd[pollsize].events = POLLIN;
			pe = pollsize;
			pollsize++;
		}

		if (pollsize) {
			if (poll(pfd, pollsize, -1) < 0) {
				if (errno != EINTR) {
					error("poll failed, resuming: %s",
					      strerror(errno));
					sleep(1);
				}
				continue;
			}
			if (0 <= pu && (pfd[pu].revents & (POLLIN|POLLHUP))) {
				/* Data ready; we keep the last byte
				 * to ourselves in case we detect
				 * broken rev-list, so that we can
				 * leave the stream corrupted.  This
				 * is unfortunate -- unpack-objects
				 * would happily accept a valid pack
				 * data with trailing garbage, so
				 * appending garbage after we pass all
				 * the pack data is not good enough to
				 * signal breakage to downstream.
				 */
				char *cp = data;
				ssize_t outsz = 0;
				if (0 <= buffered) {
					*cp++ = buffered;
					outsz++;
				}
				sz = read(pu_pipe[0], cp,
					  sizeof(data) - outsz);
				if (0 < sz)
						;
				else if (sz == 0) {
					close(pu_pipe[0]);
					pu_pipe[0] = -1;
				}
				else
					goto fail;
				sz += outsz;
				if (1 < sz) {
					buffered = data[sz-1] & 0xFF;
					sz--;
				}
				else
					buffered = -1;
				sz = send_client_data(1, data, sz);
				if (sz < 0)
					goto fail;
			}
			if (0 <= pe && (pfd[pe].revents & (POLLIN|POLLHUP))) {
				/* Status ready; we ship that in the side-band
				 * or dump to the standard error.
				 */
				sz = read(pe_pipe[0], progress,
					  sizeof(progress));
				if (0 < sz)
					send_client_data(2, progress, sz);
				else if (sz == 0) {
					close(pe_pipe[0]);
					pe_pipe[0] = -1;
				}
				else
					goto fail;
			}
		}

		/* See if the children are still there */
		if (pid_rev_list || pid_pack_objects) {
			pid = waitpid(-1, &status, WNOHANG);
			if (!pid)
				continue;
			who = ((pid == pid_rev_list) ? "git-rev-list" :
			       (pid == pid_pack_objects) ? "git-pack-objects" :
			       NULL);
			if (!who) {
				if (pid < 0) {
					error("git-upload-pack: %s",
					      strerror(errno));
					goto fail;
				}
				error("git-upload-pack: we weren't "
				      "waiting for %d", pid);
				continue;
			}
			if (!WIFEXITED(status) || WEXITSTATUS(status) > 0) {
				error("git-upload-pack: %s died with error.",
				      who);
				goto fail;
			}
			if (pid == pid_rev_list)
				pid_rev_list = 0;
			if (pid == pid_pack_objects)
				pid_pack_objects = 0;
			if (pid_rev_list || pid_pack_objects)
				continue;
		}

		/* both died happily */
		if (pollsize)
			continue;

		/* flush the data */
		if (0 <= buffered) {
			data[0] = buffered;
			sz = send_client_data(1, data, 1);
			if (sz < 0)
				goto fail;
			fprintf(stderr, "flushed.\n");
		}
		if (use_sideband)
			packet_flush(1);
		return;
	}
 fail:
	if (pid_pack_objects)
		kill(pid_pack_objects, SIGKILL);
	if (pid_rev_list)
		kill(pid_rev_list, SIGKILL);
	send_client_data(3, abort_msg, sizeof(abort_msg));
	die("git-upload-pack: %s", abort_msg);
}

static int got_sha1(char *hex, unsigned char *sha1)
{
	struct object *o;

	if (get_sha1_hex(hex, sha1))
		die("git-upload-pack: expected SHA1 object, got '%s'", hex);
	if (!has_sha1_file(sha1))
		return 0;

	o = lookup_object(sha1);
	if (!(o && o->parsed))
		o = parse_object(sha1);
	if (!o)
		die("oops (%s)", sha1_to_hex(sha1));
	if (o->type == OBJ_COMMIT) {
		struct commit_list *parents;
		if (o->flags & THEY_HAVE)
			return 0;
		o->flags |= THEY_HAVE;
		for (parents = ((struct commit*)o)->parents;
		     parents;
		     parents = parents->next)
			parents->item->object.flags |= THEY_HAVE;
	}
	add_object_array(o, NULL, &have_obj);
	return 1;
}

static int get_common_commits(void)
{
	static char line[1000];
	unsigned char sha1[20];
	char hex[41], last_hex[41];
	int len;

	track_object_refs = 0;
	save_commit_buffer = 0;

	for(;;) {
		len = packet_read_line(0, line, sizeof(line));
		reset_timeout();

		if (!len) {
			if (have_obj.nr == 0 || multi_ack)
				packet_write(1, "NAK\n");
			continue;
		}
		len = strip(line, len);
		if (!strncmp(line, "have ", 5)) {
			if (got_sha1(line+5, sha1)) {
				memcpy(hex, sha1_to_hex(sha1), 41);
				if (multi_ack) {
					const char *msg = "ACK %s continue\n";
					packet_write(1, msg, hex);
					memcpy(last_hex, hex, 41);
				}
				else if (have_obj.nr == 1)
					packet_write(1, "ACK %s\n", hex);
			}
			continue;
		}
		if (!strcmp(line, "done")) {
			if (have_obj.nr > 0) {
				if (multi_ack)
					packet_write(1, "ACK %s\n", last_hex);
				return 0;
			}
			packet_write(1, "NAK\n");
			return -1;
		}
		die("git-upload-pack: expected SHA1 list, got '%s'", line);
	}
}

static void receive_needs(void)
{
	static char line[1000];
	int len;

	for (;;) {
		struct object *o;
		unsigned char sha1_buf[20];
		len = packet_read_line(0, line, sizeof(line));
		reset_timeout();
		if (!len)
			return;

		if (strncmp("want ", line, 5) ||
		    get_sha1_hex(line+5, sha1_buf))
			die("git-upload-pack: protocol error, "
			    "expected to get sha, not '%s'", line);
		if (strstr(line+45, "multi_ack"))
			multi_ack = 1;
		if (strstr(line+45, "thin-pack"))
			use_thin_pack = 1;
		if (strstr(line+45, "ofs-delta"))
			use_ofs_delta = 1;
		if (strstr(line+45, "side-band-64k"))
			use_sideband = LARGE_PACKET_MAX;
		else if (strstr(line+45, "side-band"))
			use_sideband = DEFAULT_PACKET_MAX;

		/* We have sent all our refs already, and the other end
		 * should have chosen out of them; otherwise they are
		 * asking for nonsense.
		 *
		 * Hmph.  We may later want to allow "want" line that
		 * asks for something like "master~10" (symbolic)...
		 * would it make sense?  I don't know.
		 */
		o = lookup_object(sha1_buf);
		if (!o || !(o->flags & OUR_REF))
			die("git-upload-pack: not our ref %s", line+5);
		if (!(o->flags & WANTED)) {
			o->flags |= WANTED;
			add_object_array(o, NULL, &want_obj);
		}
	}
}

static int send_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
{
	static const char *capabilities = "multi_ack thin-pack side-band side-band-64k ofs-delta";
	struct object *o = parse_object(sha1);

	if (!o)
		die("git-upload-pack: cannot find object %s:", sha1_to_hex(sha1));

	if (capabilities)
		packet_write(1, "%s %s%c%s\n", sha1_to_hex(sha1), refname,
			0, capabilities);
	else
		packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname);
	capabilities = NULL;
	if (!(o->flags & OUR_REF)) {
		o->flags |= OUR_REF;
		nr_our_refs++;
	}
	if (o->type == OBJ_TAG) {
		o = deref_tag(o, refname, 0);
		packet_write(1, "%s %s^{}\n", sha1_to_hex(o->sha1), refname);
	}
	return 0;
}

static void upload_pack(void)
{
	reset_timeout();
	head_ref(send_ref, NULL);
	for_each_ref(send_ref, NULL);
	packet_flush(1);
	receive_needs();
	if (want_obj.nr) {
		get_common_commits();
		create_pack_file();
	}
}

int main(int argc, char **argv)
{
	char *dir;
	int i;
	int strict = 0;

	for (i = 1; i < argc; i++) {
		char *arg = argv[i];

		if (arg[0] != '-')
			break;
		if (!strcmp(arg, "--strict")) {
			strict = 1;
			continue;
		}
		if (!strncmp(arg, "--timeout=", 10)) {
			timeout = atoi(arg+10);
			continue;
		}
		if (!strcmp(arg, "--")) {
			i++;
			break;
		}
	}
	
	if (i != argc-1)
		usage(upload_pack_usage);
	dir = argv[i];

	if (!enter_repo(dir, strict))
		die("'%s': unable to chdir or not a git archive", dir);

	upload_pack();
	return 0;
}
Commit	Line	Data
e3a56298 JH	1	#include <signal.h>
	2	#include <sys/wait.h>
	3	#include <sys/poll.h>
def88e9a LT	4	#include "cache.h"
	5	#include "refs.h"
	6	#include "pkt-line.h"
958c24b1	7	#include "sideband.h"
f6b42a81 JH	8	#include "tag.h"
f6b42a81 JH	9	#include "object.h"
f0243f26	10	#include "commit.h"
77cb17e9	11	#include "exec_cmd.h"
def88e9a	12
960deccb	13	static const char upload_pack_usage[] = "git-upload-pack [--strict] [--timeout=nn] <dir>";
def88e9a	14
f0243f26	15	#define THEY_HAVE (1U << 0)
565ebbf7 JH	16	#define OUR_REF (1U << 1)
565ebbf7 JH	17	#define WANTED (1U << 2)
96f1e58f	18	static int multi_ack, nr_our_refs;
e4fe4b8e	19	static int use_thin_pack, use_ofs_delta;
b1e9fff7 JH	20	static struct object_array have_obj;
b1e9fff7 JH	21	static struct object_array want_obj;
96f1e58f	22	static unsigned int timeout;
d47f3db7 JH	23	/* 0 for no sideband,
	24	* otherwise maximum packet size (up to 65520 bytes).
	25	*/
96f1e58f	26	static int use_sideband;
960deccb PA	27
	28	static void reset_timeout(void)
	29	{
	30	alarm(timeout);
	31	}
fb9040cc	32
75bfc6c2 LT	33	static int strip(char *line, int len)
	34	{
	35	if (len && line[len-1] == '\n')
	36	line[--len] = 0;
	37	return len;
	38	}
	39
583b7ea3 JH	40	static ssize_t send_client_data(int fd, const char *data, ssize_t sz)
583b7ea3 JH	41	{
958c24b1	42	if (use_sideband)
d47f3db7	43	return send_sideband(1, fd, data, sz, use_sideband);
958c24b1 JH	44	if (fd == 3)
	45	/* emergency quit */
	46	fd = 2;
	47	if (fd == 2) {
	48	xwrite(fd, data, sz);
	49	return sz;
583b7ea3	50	}
958c24b1	51	return safe_write(fd, data, sz);
583b7ea3 JH	52	}
583b7ea3 JH	53
fb9040cc LT	54	static void create_pack_file(void)
fb9040cc LT	55	{
363b7817 JH	56	/* Pipes between rev-list to pack-objects, pack-objects to us
	57	* and pack-objects error stream for progress bar.
	58	*/
	59	int lp_pipe[2], pu_pipe[2], pe_pipe[2];
b1c71b72	60	pid_t pid_rev_list, pid_pack_objects;
b1e9fff7	61	int create_full_pack = (nr_our_refs == want_obj.nr && !have_obj.nr);
363b7817	62	char data[8193], progress[128];
583b7ea3 JH	63	char abort_msg[] = "aborting due to possible repository "
583b7ea3 JH	64	"corruption on the remote side.";
b1c71b72	65	int buffered = -1;
75bfc6c2	66
b1c71b72	67	if (pipe(lp_pipe) < 0)
75bfc6c2	68	die("git-upload-pack: unable to create pipe");
b1c71b72 JH	69	pid_rev_list = fork();
b1c71b72 JH	70	if (pid_rev_list < 0)
75bfc6c2 LT	71	die("git-upload-pack: unable to fork git-rev-list");
75bfc6c2 LT	72
b1c71b72	73	if (!pid_rev_list) {
75bfc6c2	74	int i;
e091eb93	75	int args;
9201c707	76	const char **argv;
554fe20d	77	const char **p;
e091eb93	78	char *buf;
e091eb93	79
b19696c2	80	if (create_full_pack) {
565ebbf7	81	args = 10;
b19696c2 JH	82	use_thin_pack = 0; /* no point doing it */
b19696c2 JH	83	}
e091eb93	84	else
b1e9fff7	85	args = have_obj.nr + want_obj.nr + 5;
9201c707 JH	86	p = xmalloc(args * sizeof(char *));
9201c707 JH	87	argv = (const char **) p;
e091eb93	88	buf = xmalloc(args * 45);
75bfc6c2	89
b1c71b72	90	dup2(lp_pipe[1], 1);
75bfc6c2	91	close(0);
b1c71b72 JH	92	close(lp_pipe[0]);
b1c71b72 JH	93	close(lp_pipe[1]);
77cb17e9	94	*p++ = "rev-list";
b19696c2	95	*p++ = use_thin_pack ? "--objects-edge" : "--objects";
b1e9fff7	96	if (create_full_pack)
e091eb93 JH	97	*p++ = "--all";
e091eb93 JH	98	else {
b1e9fff7 JH	99	for (i = 0; i < want_obj.nr; i++) {
b1e9fff7 JH	100	struct object *o = want_obj.objects[i].item;
e091eb93	101	*p++ = buf;
b1e9fff7	102	memcpy(buf, sha1_to_hex(o->sha1), 41);
e091eb93 JH	103	buf += 41;
e091eb93 JH	104	}
75bfc6c2	105	}
b5c367f7	106	if (!create_full_pack)
b1e9fff7 JH	107	for (i = 0; i < have_obj.nr; i++) {
b1e9fff7 JH	108	struct object *o = have_obj.objects[i].item;
b5c367f7 JS	109	*p++ = buf;
b5c367f7 JS	110	*buf++ = '^';
b1e9fff7	111	memcpy(buf, sha1_to_hex(o->sha1), 41);
b5c367f7 JS	112	buf += 41;
b5c367f7 JS	113	}
75bfc6c2	114	*p++ = NULL;
77cb17e9	115	execv_git_cmd(argv);
75bfc6c2 LT	116	die("git-upload-pack: unable to exec git-rev-list");
75bfc6c2 LT	117	}
b1c71b72 JH	118
	119	if (pipe(pu_pipe) < 0)
	120	die("git-upload-pack: unable to create pipe");
363b7817 JH	121	if (pipe(pe_pipe) < 0)
363b7817 JH	122	die("git-upload-pack: unable to create pipe");
b1c71b72 JH	123	pid_pack_objects = fork();
	124	if (pid_pack_objects < 0) {
	125	/* daemon sets things up to ignore TERM */
	126	kill(pid_rev_list, SIGKILL);
	127	die("git-upload-pack: unable to fork git-pack-objects");
	128	}
	129	if (!pid_pack_objects) {
	130	dup2(lp_pipe[0], 0);
	131	dup2(pu_pipe[1], 1);
363b7817	132	dup2(pe_pipe[1], 2);
b1c71b72 JH	133
	134	close(lp_pipe[0]);
	135	close(lp_pipe[1]);
	136	close(pu_pipe[0]);
	137	close(pu_pipe[1]);
363b7817 JH	138	close(pe_pipe[0]);
363b7817 JH	139	close(pe_pipe[1]);
e4fe4b8e NP	140	execl_git_cmd("pack-objects", "--stdout", "--progress",
	141	use_ofs_delta ? "--delta-base-offset" : NULL,
	142	NULL);
b1c71b72 JH	143	kill(pid_rev_list, SIGKILL);
	144	die("git-upload-pack: unable to exec git-pack-objects");
	145	}
	146
	147	close(lp_pipe[0]);
	148	close(lp_pipe[1]);
	149
363b7817 JH	150	/* We read from pe_pipe[0] to capture stderr output for
363b7817 JH	151	* progress bar, and pu_pipe[0] to capture the pack data.
b1c71b72	152	*/
363b7817	153	close(pe_pipe[1]);
b1c71b72 JH	154	close(pu_pipe[1]);
	155
	156	while (1) {
	157	const char *who;
	158	struct pollfd pfd[2];
	159	pid_t pid;
	160	int status;
	161	ssize_t sz;
363b7817	162	int pe, pu, pollsize;
b1c71b72	163
0d516ada ML	164	reset_timeout();
0d516ada ML	165
b1c71b72	166	pollsize = 0;
363b7817	167	pe = pu = -1;
b1c71b72 JH	168
	169	if (0 <= pu_pipe[0]) {
	170	pfd[pollsize].fd = pu_pipe[0];
	171	pfd[pollsize].events = POLLIN;
	172	pu = pollsize;
	173	pollsize++;
	174	}
363b7817 JH	175	if (0 <= pe_pipe[0]) {
	176	pfd[pollsize].fd = pe_pipe[0];
	177	pfd[pollsize].events = POLLIN;
	178	pe = pollsize;
	179	pollsize++;
	180	}
b1c71b72 JH	181
	182	if (pollsize) {
	183	if (poll(pfd, pollsize, -1) < 0) {
	184	if (errno != EINTR) {
	185	error("poll failed, resuming: %s",
	186	strerror(errno));
	187	sleep(1);
	188	}
	189	continue;
	190	}
	191	if (0 <= pu && (pfd[pu].revents & (POLLIN\|POLLHUP))) {
	192	/* Data ready; we keep the last byte
	193	* to ourselves in case we detect
	194	* broken rev-list, so that we can
	195	* leave the stream corrupted. This
	196	* is unfortunate -- unpack-objects
	197	* would happily accept a valid pack
	198	* data with trailing garbage, so
	199	* appending garbage after we pass all
	200	* the pack data is not good enough to
	201	* signal breakage to downstream.
	202	*/
	203	char *cp = data;
	204	ssize_t outsz = 0;
	205	if (0 <= buffered) {
	206	*cp++ = buffered;
	207	outsz++;
	208	}
	209	sz = read(pu_pipe[0], cp,
	210	sizeof(data) - outsz);
	211	if (0 < sz)
	212	;
	213	else if (sz == 0) {
	214	close(pu_pipe[0]);
	215	pu_pipe[0] = -1;
	216	}
	217	else
	218	goto fail;
	219	sz += outsz;
	220	if (1 < sz) {
	221	buffered = data[sz-1] & 0xFF;
	222	sz--;
	223	}
	224	else
	225	buffered = -1;
583b7ea3	226	sz = send_client_data(1, data, sz);
b1c71b72 JH	227	if (sz < 0)
	228	goto fail;
	229	}
363b7817	230	if (0 <= pe && (pfd[pe].revents & (POLLIN\|POLLHUP))) {
583b7ea3 JH	231	/* Status ready; we ship that in the side-band
583b7ea3 JH	232	* or dump to the standard error.
363b7817 JH	233	*/
	234	sz = read(pe_pipe[0], progress,
	235	sizeof(progress));
	236	if (0 < sz)
583b7ea3	237	send_client_data(2, progress, sz);
363b7817 JH	238	else if (sz == 0) {
	239	close(pe_pipe[0]);
	240	pe_pipe[0] = -1;
	241	}
	242	else
	243	goto fail;
	244	}
b1c71b72 JH	245	}
	246
	247	/* See if the children are still there */
	248	if (pid_rev_list \|\| pid_pack_objects) {
	249	pid = waitpid(-1, &status, WNOHANG);
	250	if (!pid)
	251	continue;
	252	who = ((pid == pid_rev_list) ? "git-rev-list" :
	253	(pid == pid_pack_objects) ? "git-pack-objects" :
	254	NULL);
	255	if (!who) {
	256	if (pid < 0) {
	257	error("git-upload-pack: %s",
	258	strerror(errno));
	259	goto fail;
	260	}
	261	error("git-upload-pack: we weren't "
	262	"waiting for %d", pid);
	263	continue;
	264	}
	265	if (!WIFEXITED(status) \|\| WEXITSTATUS(status) > 0) {
	266	error("git-upload-pack: %s died with error.",
	267	who);
	268	goto fail;
	269	}
	270	if (pid == pid_rev_list)
	271	pid_rev_list = 0;
	272	if (pid == pid_pack_objects)
	273	pid_pack_objects = 0;
	274	if (pid_rev_list \|\| pid_pack_objects)
	275	continue;
	276	}
	277
	278	/* both died happily */
	279	if (pollsize)
	280	continue;
	281
	282	/* flush the data */
	283	if (0 <= buffered) {
	284	data[0] = buffered;
583b7ea3	285	sz = send_client_data(1, data, 1);
b1c71b72 JH	286	if (sz < 0)
	287	goto fail;
	288	fprintf(stderr, "flushed.\n");
	289	}
958c24b1 JH	290	if (use_sideband)
958c24b1 JH	291	packet_flush(1);
b1c71b72 JH	292	return;
	293	}
	294	fail:
	295	if (pid_pack_objects)
	296	kill(pid_pack_objects, SIGKILL);
	297	if (pid_rev_list)
	298	kill(pid_rev_list, SIGKILL);
583b7ea3 JH	299	send_client_data(3, abort_msg, sizeof(abort_msg));
583b7ea3 JH	300	die("git-upload-pack: %s", abort_msg);
fb9040cc LT	301	}
fb9040cc LT	302
def88e9a LT	303	static int got_sha1(char hex, unsigned char sha1)
def88e9a LT	304	{
b1e9fff7 JH	305	struct object *o;
b1e9fff7 JH	306
def88e9a LT	307	if (get_sha1_hex(hex, sha1))
def88e9a LT	308	die("git-upload-pack: expected SHA1 object, got '%s'", hex);
fb9040cc LT	309	if (!has_sha1_file(sha1))
fb9040cc LT	310	return 0;
b1e9fff7 JH	311
	312	o = lookup_object(sha1);
	313	if (!(o && o->parsed))
	314	o = parse_object(sha1);
	315	if (!o)
	316	die("oops (%s)", sha1_to_hex(sha1));
182a8dab	317	if (o->type == OBJ_COMMIT) {
b1e9fff7 JH	318	struct commit_list *parents;
	319	if (o->flags & THEY_HAVE)
	320	return 0;
	321	o->flags \|= THEY_HAVE;
	322	for (parents = ((struct commit*)o)->parents;
	323	parents;
	324	parents = parents->next)
	325	parents->item->object.flags \|= THEY_HAVE;
fb9040cc	326	}
b1e9fff7	327	add_object_array(o, NULL, &have_obj);
fb9040cc	328	return 1;
def88e9a LT	329	}
	330
	331	static int get_common_commits(void)
	332	{
	333	static char line[1000];
c04c4e57 JH	334	unsigned char sha1[20];
c04c4e57 JH	335	char hex[41], last_hex[41];
def88e9a LT	336	int len;
def88e9a LT	337
f0243f26 JS	338	track_object_refs = 0;
	339	save_commit_buffer = 0;
	340
def88e9a LT	341	for(;;) {
def88e9a LT	342	len = packet_read_line(0, line, sizeof(line));
960deccb	343	reset_timeout();
def88e9a LT	344
def88e9a LT	345	if (!len) {
b1e9fff7	346	if (have_obj.nr == 0 \|\| multi_ack)
1bd8c8f0	347	packet_write(1, "NAK\n");
def88e9a LT	348	continue;
def88e9a LT	349	}
75bfc6c2	350	len = strip(line, len);
def88e9a	351	if (!strncmp(line, "have ", 5)) {
c04c4e57 JH	352	if (got_sha1(line+5, sha1)) {
	353	memcpy(hex, sha1_to_hex(sha1), 41);
	354	if (multi_ack) {
	355	const char *msg = "ACK %s continue\n";
	356	packet_write(1, msg, hex);
	357	memcpy(last_hex, hex, 41);
	358	}
	359	else if (have_obj.nr == 1)
	360	packet_write(1, "ACK %s\n", hex);
af2d3aa4	361	}
def88e9a LT	362	continue;
	363	}
	364	if (!strcmp(line, "done")) {
b1e9fff7	365	if (have_obj.nr > 0) {
1bd8c8f0	366	if (multi_ack)
c04c4e57	367	packet_write(1, "ACK %s\n", last_hex);
1bd8c8f0 JS	368	return 0;
1bd8c8f0 JS	369	}
def88e9a LT	370	packet_write(1, "NAK\n");
	371	return -1;
	372	}
	373	die("git-upload-pack: expected SHA1 list, got '%s'", line);
	374	}
def88e9a LT	375	}
def88e9a LT	376
b1e9fff7	377	static void receive_needs(void)
fb9040cc LT	378	{
fb9040cc LT	379	static char line[1000];
b1e9fff7	380	int len;
fb9040cc	381
fb9040cc	382	for (;;) {
565ebbf7	383	struct object *o;
6ece0d30	384	unsigned char sha1_buf[20];
fb9040cc	385	len = packet_read_line(0, line, sizeof(line));
960deccb	386	reset_timeout();
fb9040cc	387	if (!len)
b1e9fff7	388	return;
e091eb93	389
6ece0d30 JH	390	if (strncmp("want ", line, 5) \|\|
6ece0d30 JH	391	get_sha1_hex(line+5, sha1_buf))
e091eb93 JH	392	die("git-upload-pack: protocol error, "
e091eb93 JH	393	"expected to get sha, not '%s'", line);
1bd8c8f0 JS	394	if (strstr(line+45, "multi_ack"))
1bd8c8f0 JS	395	multi_ack = 1;
b19696c2 JH	396	if (strstr(line+45, "thin-pack"))
b19696c2 JH	397	use_thin_pack = 1;
e4fe4b8e NP	398	if (strstr(line+45, "ofs-delta"))
e4fe4b8e NP	399	use_ofs_delta = 1;
d47f3db7 JH	400	if (strstr(line+45, "side-band-64k"))
	401	use_sideband = LARGE_PACKET_MAX;
	402	else if (strstr(line+45, "side-band"))
	403	use_sideband = DEFAULT_PACKET_MAX;
565ebbf7 JH	404
	405	/* We have sent all our refs already, and the other end
	406	* should have chosen out of them; otherwise they are
	407	* asking for nonsense.
	408	*
	409	* Hmph. We may later want to allow "want" line that
	410	* asks for something like "master~10" (symbolic)...
	411	* would it make sense? I don't know.
	412	*/
	413	o = lookup_object(sha1_buf);
	414	if (!o \|\| !(o->flags & OUR_REF))
	415	die("git-upload-pack: not our ref %s", line+5);
	416	if (!(o->flags & WANTED)) {
	417	o->flags \|= WANTED;
b1e9fff7	418	add_object_array(o, NULL, &want_obj);
565ebbf7	419	}
fb9040cc LT	420	}
	421	}
	422
8da19775	423	static int send_ref(const char refname, const unsigned char sha1, int flag, void *cb_data)
def88e9a	424	{
e4fe4b8e	425	static const char *capabilities = "multi_ack thin-pack side-band side-band-64k ofs-delta";
f6b42a81 JH	426	struct object *o = parse_object(sha1);
f6b42a81 JH	427
b5b16990 CW	428	if (!o)
	429	die("git-upload-pack: cannot find object %s:", sha1_to_hex(sha1));
	430
1f5881bb JS	431	if (capabilities)
	432	packet_write(1, "%s %s%c%s\n", sha1_to_hex(sha1), refname,
	433	0, capabilities);
	434	else
	435	packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname);
	436	capabilities = NULL;
565ebbf7 JH	437	if (!(o->flags & OUR_REF)) {
	438	o->flags \|= OUR_REF;
	439	nr_our_refs++;
	440	}
1974632c	441	if (o->type == OBJ_TAG) {
9534f40b	442	o = deref_tag(o, refname, 0);
f6b42a81 JH	443	packet_write(1, "%s %s^{}\n", sha1_to_hex(o->sha1), refname);
f6b42a81 JH	444	}
def88e9a LT	445	return 0;
	446	}
	447
59076eba	448	static void upload_pack(void)
def88e9a	449	{
960deccb	450	reset_timeout();
cb5d709f JH	451	head_ref(send_ref, NULL);
cb5d709f JH	452	for_each_ref(send_ref, NULL);
def88e9a	453	packet_flush(1);
b1e9fff7	454	receive_needs();
59076eba DR	455	if (want_obj.nr) {
	456	get_common_commits();
	457	create_pack_file();
	458	}
def88e9a LT	459	}
	460
	461	int main(int argc, char **argv)
	462	{
8d630132	463	char *dir;
960deccb PA	464	int i;
	465	int strict = 0;
	466
	467	for (i = 1; i < argc; i++) {
	468	char *arg = argv[i];
	469
	470	if (arg[0] != '-')
	471	break;
	472	if (!strcmp(arg, "--strict")) {
	473	strict = 1;
	474	continue;
	475	}
	476	if (!strncmp(arg, "--timeout=", 10)) {
	477	timeout = atoi(arg+10);
	478	continue;
	479	}
	480	if (!strcmp(arg, "--")) {
	481	i++;
	482	break;
	483	}
	484	}
	485
	486	if (i != argc-1)
def88e9a	487	usage(upload_pack_usage);
960deccb	488	dir = argv[i];
113b9475	489
8d630132 AE	490	if (!enter_repo(dir, strict))
8d630132 AE	491	die("'%s': unable to chdir or not a git archive", dir);
960deccb	492
def88e9a LT	493	upload_pack();
	494	return 0;
	495	}