[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 "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)
#define MAX_HAS 256
#define MAX_NEEDS 256
static int nr_has = 0, nr_needs = 0, multi_ack = 0, nr_our_refs = 0;
static int use_thin_pack = 0;
static unsigned char has_sha1[MAX_HAS][20];
static unsigned char needs_sha1[MAX_NEEDS][20];
static unsigned int timeout = 0;
static int use_sideband = 0;

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

#define PACKET_MAX 1000
static ssize_t send_client_data(int fd, const char *data, ssize_t sz)
{
	ssize_t ssz;
	const char *p;

	if (!data) {
		if (!use_sideband)
			return 0;
		packet_flush(1);
	}

	if (!use_sideband) {
		if (fd == 3)
			/* emergency quit */
			fd = 2;
		if (fd == 2) {
			xwrite(fd, data, sz);
			return sz;
		}
		return safe_write(fd, data, sz);
	}
	p = data;
	ssz = sz;
	while (sz) {
		unsigned n;
		char hdr[5];

		n = sz;
		if (PACKET_MAX - 5 < n)
			n = PACKET_MAX - 5;
		sprintf(hdr, "%04x", n + 5);
		hdr[4] = fd;
		safe_write(1, hdr, 5);
		safe_write(1, p, n);
		p += n;
		sz -= n;
	}
	return ssz;
}

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 == nr_needs && !nr_has);
	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 = nr_has + nr_needs + 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 || MAX_NEEDS <= nr_needs)
			*p++ = "--all";
		else {
			for (i = 0; i < nr_needs; i++) {
				*p++ = buf;
				memcpy(buf, sha1_to_hex(needs_sha1[i]), 41);
				buf += 41;
			}
		}
		if (!create_full_pack)
			for (i = 0; i < nr_has; i++) {
				*p++ = buf;
				*buf++ = '^';
				memcpy(buf, sha1_to_hex(has_sha1[i]), 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", 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");
		}
		send_client_data(1, NULL, 0);
		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)
{
	if (get_sha1_hex(hex, sha1))
		die("git-upload-pack: expected SHA1 object, got '%s'", hex);
	if (!has_sha1_file(sha1))
		return 0;
	if (nr_has < MAX_HAS) {
		struct object *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;
		}
		memcpy(has_sha1[nr_has++], sha1, 20);
	}
	return 1;
}

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

	track_object_refs = 0;
	save_commit_buffer = 0;

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

		if (!len) {
			if (nr_has == 0 || multi_ack)
				packet_write(1, "NAK\n");
			continue;
		}
		len = strip(line, len);
		if (!strncmp(line, "have ", 5)) {
			if (got_sha1(line+5, sha1) &&
					(multi_ack || nr_has == 1)) {
				if (nr_has >= MAX_HAS)
					multi_ack = 0;
				packet_write(1, "ACK %s%s\n",
					sha1_to_hex(sha1),
					multi_ack ?  " continue" : "");
				if (multi_ack)
					memcpy(last_sha1, sha1, 20);
			}
			continue;
		}
		if (!strcmp(line, "done")) {
			if (nr_has > 0) {
				if (multi_ack)
					packet_write(1, "ACK %s\n",
							sha1_to_hex(last_sha1));
				return 0;
			}
			packet_write(1, "NAK\n");
			return -1;
		}
		die("git-upload-pack: expected SHA1 list, got '%s'", line);
	}
}

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

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

		sha1_buf = dummy;
		if (needs == MAX_NEEDS) {
			fprintf(stderr,
				"warning: supporting only a max of %d requests. "
				"sending everything instead.\n",
				MAX_NEEDS);
		}
		else if (needs < MAX_NEEDS)
			sha1_buf = needs_sha1[needs];

		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, "side-band"))
			use_sideband = 1;

		/* 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;
			needs++;
		}
	}
}

static int send_ref(const char *refname, const unsigned char *sha1)
{
	static const char *capabilities = "multi_ack thin-pack side-band";
	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 int upload_pack(void)
{
	reset_timeout();
	head_ref(send_ref);
	for_each_ref(send_ref);
	packet_flush(1);
	nr_needs = receive_needs();
	if (!nr_needs)
		return 0;
	get_common_commits();
	create_pack_file();
	return 0;
}

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