[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)
static int multi_ack = 0, nr_our_refs = 0;
static int use_thin_pack = 0;
static struct object_array have_obj;
static struct object_array want_obj;
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 == 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", 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)
{
	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, "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;
			add_object_array(o, NULL, &want_obj);
		}
	}
}

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);
	receive_needs();
	if (!want_obj.nr)
		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)
b1e9fff7	17	static int multi_ack = 0, nr_our_refs = 0;
b19696c2	18	static int use_thin_pack = 0;
b1e9fff7 JH	19	static struct object_array have_obj;
b1e9fff7 JH	20	static struct object_array want_obj;
960deccb	21	static unsigned int timeout = 0;
583b7ea3	22	static int use_sideband = 0;
960deccb PA	23
	24	static void reset_timeout(void)
	25	{
	26	alarm(timeout);
	27	}
fb9040cc	28
75bfc6c2 LT	29	static int strip(char *line, int len)
	30	{
	31	if (len && line[len-1] == '\n')
	32	line[--len] = 0;
	33	return len;
	34	}
	35
583b7ea3 JH	36	#define PACKET_MAX 1000
	37	static ssize_t send_client_data(int fd, const char *data, ssize_t sz)
	38	{
	39	ssize_t ssz;
	40	const char *p;
	41
	42	if (!data) {
	43	if (!use_sideband)
	44	return 0;
	45	packet_flush(1);
	46	}
	47
	48	if (!use_sideband) {
	49	if (fd == 3)
	50	/* emergency quit */
	51	fd = 2;
5f490ce0 ML	52	if (fd == 2) {
	53	xwrite(fd, data, sz);
	54	return sz;
	55	}
583b7ea3 JH	56	return safe_write(fd, data, sz);
	57	}
	58	p = data;
	59	ssz = sz;
	60	while (sz) {
	61	unsigned n;
	62	char hdr[5];
	63
	64	n = sz;
	65	if (PACKET_MAX - 5 < n)
	66	n = PACKET_MAX - 5;
	67	sprintf(hdr, "%04x", n + 5);
	68	hdr[4] = fd;
	69	safe_write(1, hdr, 5);
	70	safe_write(1, p, n);
	71	p += n;
	72	sz -= n;
	73	}
	74	return ssz;
	75	}
	76
fb9040cc LT	77	static void create_pack_file(void)
fb9040cc LT	78	{
363b7817 JH	79	/* Pipes between rev-list to pack-objects, pack-objects to us
	80	* and pack-objects error stream for progress bar.
	81	*/
	82	int lp_pipe[2], pu_pipe[2], pe_pipe[2];
b1c71b72	83	pid_t pid_rev_list, pid_pack_objects;
b1e9fff7	84	int create_full_pack = (nr_our_refs == want_obj.nr && !have_obj.nr);
363b7817	85	char data[8193], progress[128];
583b7ea3 JH	86	char abort_msg[] = "aborting due to possible repository "
583b7ea3 JH	87	"corruption on the remote side.";
b1c71b72	88	int buffered = -1;
75bfc6c2	89
b1c71b72	90	if (pipe(lp_pipe) < 0)
75bfc6c2	91	die("git-upload-pack: unable to create pipe");
b1c71b72 JH	92	pid_rev_list = fork();
b1c71b72 JH	93	if (pid_rev_list < 0)
75bfc6c2 LT	94	die("git-upload-pack: unable to fork git-rev-list");
75bfc6c2 LT	95
b1c71b72	96	if (!pid_rev_list) {
75bfc6c2	97	int i;
e091eb93	98	int args;
9201c707	99	const char **argv;
554fe20d	100	const char **p;
e091eb93	101	char *buf;
e091eb93	102
b19696c2	103	if (create_full_pack) {
565ebbf7	104	args = 10;
b19696c2 JH	105	use_thin_pack = 0; /* no point doing it */
b19696c2 JH	106	}
e091eb93	107	else
b1e9fff7	108	args = have_obj.nr + want_obj.nr + 5;
9201c707 JH	109	p = xmalloc(args * sizeof(char *));
9201c707 JH	110	argv = (const char **) p;
e091eb93	111	buf = xmalloc(args * 45);
75bfc6c2	112
b1c71b72	113	dup2(lp_pipe[1], 1);
75bfc6c2	114	close(0);
b1c71b72 JH	115	close(lp_pipe[0]);
b1c71b72 JH	116	close(lp_pipe[1]);
77cb17e9	117	*p++ = "rev-list";
b19696c2	118	*p++ = use_thin_pack ? "--objects-edge" : "--objects";
b1e9fff7	119	if (create_full_pack)
e091eb93 JH	120	*p++ = "--all";
e091eb93 JH	121	else {
b1e9fff7 JH	122	for (i = 0; i < want_obj.nr; i++) {
b1e9fff7 JH	123	struct object *o = want_obj.objects[i].item;
e091eb93	124	*p++ = buf;
b1e9fff7	125	memcpy(buf, sha1_to_hex(o->sha1), 41);
e091eb93 JH	126	buf += 41;
e091eb93 JH	127	}
75bfc6c2	128	}
b5c367f7	129	if (!create_full_pack)
b1e9fff7 JH	130	for (i = 0; i < have_obj.nr; i++) {
b1e9fff7 JH	131	struct object *o = have_obj.objects[i].item;
b5c367f7 JS	132	*p++ = buf;
b5c367f7 JS	133	*buf++ = '^';
b1e9fff7	134	memcpy(buf, sha1_to_hex(o->sha1), 41);
b5c367f7 JS	135	buf += 41;
b5c367f7 JS	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)
def88e9a LT	324	{
b1e9fff7 JH	325	struct object *o;
b1e9fff7 JH	326
def88e9a LT	327	if (get_sha1_hex(hex, sha1))
def88e9a LT	328	die("git-upload-pack: expected SHA1 object, got '%s'", hex);
fb9040cc LT	329	if (!has_sha1_file(sha1))
fb9040cc LT	330	return 0;
b1e9fff7 JH	331
	332	o = lookup_object(sha1);
	333	if (!(o && o->parsed))
	334	o = parse_object(sha1);
	335	if (!o)
	336	die("oops (%s)", sha1_to_hex(sha1));
182a8dab	337	if (o->type == OBJ_COMMIT) {
b1e9fff7 JH	338	struct commit_list *parents;
	339	if (o->flags & THEY_HAVE)
	340	return 0;
	341	o->flags \|= THEY_HAVE;
	342	for (parents = ((struct commit*)o)->parents;
	343	parents;
	344	parents = parents->next)
	345	parents->item->object.flags \|= THEY_HAVE;
fb9040cc	346	}
b1e9fff7	347	add_object_array(o, NULL, &have_obj);
fb9040cc	348	return 1;
def88e9a LT	349	}
	350
	351	static int get_common_commits(void)
	352	{
	353	static char line[1000];
c04c4e57 JH	354	unsigned char sha1[20];
c04c4e57 JH	355	char hex[41], last_hex[41];
def88e9a LT	356	int len;
def88e9a LT	357
f0243f26 JS	358	track_object_refs = 0;
	359	save_commit_buffer = 0;
	360
def88e9a LT	361	for(;;) {
def88e9a LT	362	len = packet_read_line(0, line, sizeof(line));
960deccb	363	reset_timeout();
def88e9a LT	364
def88e9a LT	365	if (!len) {
b1e9fff7	366	if (have_obj.nr == 0 \|\| multi_ack)
1bd8c8f0	367	packet_write(1, "NAK\n");
def88e9a LT	368	continue;
def88e9a LT	369	}
75bfc6c2	370	len = strip(line, len);
def88e9a	371	if (!strncmp(line, "have ", 5)) {
c04c4e57 JH	372	if (got_sha1(line+5, sha1)) {
	373	memcpy(hex, sha1_to_hex(sha1), 41);
	374	if (multi_ack) {
	375	const char *msg = "ACK %s continue\n";
	376	packet_write(1, msg, hex);
	377	memcpy(last_hex, hex, 41);
	378	}
	379	else if (have_obj.nr == 1)
	380	packet_write(1, "ACK %s\n", hex);
af2d3aa4	381	}
def88e9a LT	382	continue;
	383	}
	384	if (!strcmp(line, "done")) {
b1e9fff7	385	if (have_obj.nr > 0) {
1bd8c8f0	386	if (multi_ack)
c04c4e57	387	packet_write(1, "ACK %s\n", last_hex);
1bd8c8f0 JS	388	return 0;
1bd8c8f0 JS	389	}
def88e9a LT	390	packet_write(1, "NAK\n");
	391	return -1;
	392	}
	393	die("git-upload-pack: expected SHA1 list, got '%s'", line);
	394	}
def88e9a LT	395	}
def88e9a LT	396
b1e9fff7	397	static void receive_needs(void)
fb9040cc LT	398	{
fb9040cc LT	399	static char line[1000];
b1e9fff7	400	int len;
fb9040cc	401
fb9040cc	402	for (;;) {
565ebbf7	403	struct object *o;
6ece0d30	404	unsigned char sha1_buf[20];
fb9040cc	405	len = packet_read_line(0, line, sizeof(line));
960deccb	406	reset_timeout();
fb9040cc	407	if (!len)
b1e9fff7	408	return;
e091eb93	409
6ece0d30 JH	410	if (strncmp("want ", line, 5) \|\|
6ece0d30 JH	411	get_sha1_hex(line+5, sha1_buf))
e091eb93 JH	412	die("git-upload-pack: protocol error, "
e091eb93 JH	413	"expected to get sha, not '%s'", line);
1bd8c8f0 JS	414	if (strstr(line+45, "multi_ack"))
1bd8c8f0 JS	415	multi_ack = 1;
b19696c2 JH	416	if (strstr(line+45, "thin-pack"))
b19696c2 JH	417	use_thin_pack = 1;
583b7ea3 JH	418	if (strstr(line+45, "side-band"))
583b7ea3 JH	419	use_sideband = 1;
565ebbf7 JH	420
	421	/* We have sent all our refs already, and the other end
	422	* should have chosen out of them; otherwise they are
	423	* asking for nonsense.
	424	*
	425	* Hmph. We may later want to allow "want" line that
	426	* asks for something like "master~10" (symbolic)...
	427	* would it make sense? I don't know.
	428	*/
	429	o = lookup_object(sha1_buf);
	430	if (!o \|\| !(o->flags & OUR_REF))
	431	die("git-upload-pack: not our ref %s", line+5);
	432	if (!(o->flags & WANTED)) {
	433	o->flags \|= WANTED;
b1e9fff7	434	add_object_array(o, NULL, &want_obj);
565ebbf7	435	}
fb9040cc LT	436	}
	437	}
	438
def88e9a LT	439	static int send_ref(const char refname, const unsigned char sha1)
def88e9a LT	440	{
554fe20d	441	static const char *capabilities = "multi_ack thin-pack side-band";
f6b42a81 JH	442	struct object *o = parse_object(sha1);
f6b42a81 JH	443
b5b16990 CW	444	if (!o)
	445	die("git-upload-pack: cannot find object %s:", sha1_to_hex(sha1));
	446
1f5881bb JS	447	if (capabilities)
	448	packet_write(1, "%s %s%c%s\n", sha1_to_hex(sha1), refname,
	449	0, capabilities);
	450	else
	451	packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname);
	452	capabilities = NULL;
565ebbf7 JH	453	if (!(o->flags & OUR_REF)) {
	454	o->flags \|= OUR_REF;
	455	nr_our_refs++;
	456	}
1974632c	457	if (o->type == OBJ_TAG) {
9534f40b	458	o = deref_tag(o, refname, 0);
f6b42a81 JH	459	packet_write(1, "%s %s^{}\n", sha1_to_hex(o->sha1), refname);
f6b42a81 JH	460	}
def88e9a LT	461	return 0;
	462	}
	463
	464	static int upload_pack(void)
	465	{
960deccb	466	reset_timeout();
723c31fe	467	head_ref(send_ref);
def88e9a LT	468	for_each_ref(send_ref);
def88e9a LT	469	packet_flush(1);
b1e9fff7 JH	470	receive_needs();
b1e9fff7 JH	471	if (!want_obj.nr)
fb9040cc	472	return 0;
def88e9a	473	get_common_commits();
fb9040cc	474	create_pack_file();
def88e9a LT	475	return 0;
	476	}
	477
	478	int main(int argc, char **argv)
	479	{
8d630132	480	char *dir;
960deccb PA	481	int i;
	482	int strict = 0;
	483
	484	for (i = 1; i < argc; i++) {
	485	char *arg = argv[i];
	486
	487	if (arg[0] != '-')
	488	break;
	489	if (!strcmp(arg, "--strict")) {
	490	strict = 1;
	491	continue;
	492	}
	493	if (!strncmp(arg, "--timeout=", 10)) {
	494	timeout = atoi(arg+10);
	495	continue;
	496	}
	497	if (!strcmp(arg, "--")) {
	498	i++;
	499	break;
	500	}
	501	}
	502
	503	if (i != argc-1)
def88e9a	504	usage(upload_pack_usage);
960deccb	505	dir = argv[i];
113b9475	506
8d630132 AE	507	if (!enter_repo(dir, strict))
8d630132 AE	508	die("'%s': unable to chdir or not a git archive", dir);
960deccb	509
def88e9a LT	510	upload_pack();
	511	return 0;
	512	}