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

		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 JH	184
b1c71b72 JH	185	pollsize = 0;
363b7817	186	pe = pu = -1;
b1c71b72 JH	187
	188	if (0 <= pu_pipe[0]) {
	189	pfd[pollsize].fd = pu_pipe[0];
	190	pfd[pollsize].events = POLLIN;
	191	pu = pollsize;
	192	pollsize++;
	193	}
363b7817 JH	194	if (0 <= pe_pipe[0]) {
	195	pfd[pollsize].fd = pe_pipe[0];
	196	pfd[pollsize].events = POLLIN;
	197	pe = pollsize;
	198	pollsize++;
	199	}
b1c71b72 JH	200
	201	if (pollsize) {
	202	if (poll(pfd, pollsize, -1) < 0) {
	203	if (errno != EINTR) {
	204	error("poll failed, resuming: %s",
	205	strerror(errno));
	206	sleep(1);
	207	}
	208	continue;
	209	}
	210	if (0 <= pu && (pfd[pu].revents & (POLLIN\|POLLHUP))) {
	211	/* Data ready; we keep the last byte
	212	* to ourselves in case we detect
	213	* broken rev-list, so that we can
	214	* leave the stream corrupted. This
	215	* is unfortunate -- unpack-objects
	216	* would happily accept a valid pack
	217	* data with trailing garbage, so
	218	* appending garbage after we pass all
	219	* the pack data is not good enough to
	220	* signal breakage to downstream.
	221	*/
	222	char *cp = data;
	223	ssize_t outsz = 0;
	224	if (0 <= buffered) {
	225	*cp++ = buffered;
	226	outsz++;
	227	}
	228	sz = read(pu_pipe[0], cp,
	229	sizeof(data) - outsz);
	230	if (0 < sz)
	231	;
	232	else if (sz == 0) {
	233	close(pu_pipe[0]);
	234	pu_pipe[0] = -1;
	235	}
	236	else
	237	goto fail;
	238	sz += outsz;
	239	if (1 < sz) {
	240	buffered = data[sz-1] & 0xFF;
	241	sz--;
	242	}
	243	else
	244	buffered = -1;
583b7ea3	245	sz = send_client_data(1, data, sz);
b1c71b72 JH	246	if (sz < 0)
	247	goto fail;
	248	}
363b7817	249	if (0 <= pe && (pfd[pe].revents & (POLLIN\|POLLHUP))) {
583b7ea3 JH	250	/* Status ready; we ship that in the side-band
583b7ea3 JH	251	* or dump to the standard error.
363b7817 JH	252	*/
	253	sz = read(pe_pipe[0], progress,
	254	sizeof(progress));
	255	if (0 < sz)
583b7ea3	256	send_client_data(2, progress, sz);
363b7817 JH	257	else if (sz == 0) {
	258	close(pe_pipe[0]);
	259	pe_pipe[0] = -1;
	260	}
	261	else
	262	goto fail;
	263	}
b1c71b72 JH	264	}
	265
	266	/* See if the children are still there */
	267	if (pid_rev_list \|\| pid_pack_objects) {
	268	pid = waitpid(-1, &status, WNOHANG);
	269	if (!pid)
	270	continue;
	271	who = ((pid == pid_rev_list) ? "git-rev-list" :
	272	(pid == pid_pack_objects) ? "git-pack-objects" :
	273	NULL);
	274	if (!who) {
	275	if (pid < 0) {
	276	error("git-upload-pack: %s",
	277	strerror(errno));
	278	goto fail;
	279	}
	280	error("git-upload-pack: we weren't "
	281	"waiting for %d", pid);
	282	continue;
	283	}
	284	if (!WIFEXITED(status) \|\| WEXITSTATUS(status) > 0) {
	285	error("git-upload-pack: %s died with error.",
	286	who);
	287	goto fail;
	288	}
	289	if (pid == pid_rev_list)
	290	pid_rev_list = 0;
	291	if (pid == pid_pack_objects)
	292	pid_pack_objects = 0;
	293	if (pid_rev_list \|\| pid_pack_objects)
	294	continue;
	295	}
	296
	297	/* both died happily */
	298	if (pollsize)
	299	continue;
	300
	301	/* flush the data */
	302	if (0 <= buffered) {
	303	data[0] = buffered;
583b7ea3	304	sz = send_client_data(1, data, 1);
b1c71b72 JH	305	if (sz < 0)
	306	goto fail;
	307	fprintf(stderr, "flushed.\n");
	308	}
583b7ea3	309	send_client_data(1, NULL, 0);
b1c71b72 JH	310	return;
	311	}
	312	fail:
	313	if (pid_pack_objects)
	314	kill(pid_pack_objects, SIGKILL);
	315	if (pid_rev_list)
	316	kill(pid_rev_list, SIGKILL);
583b7ea3 JH	317	send_client_data(3, abort_msg, sizeof(abort_msg));
583b7ea3 JH	318	die("git-upload-pack: %s", abort_msg);
fb9040cc LT	319	}
fb9040cc LT	320
def88e9a LT	321	static int got_sha1(char hex, unsigned char sha1)
	322	{
	323	if (get_sha1_hex(hex, sha1))
	324	die("git-upload-pack: expected SHA1 object, got '%s'", hex);
fb9040cc LT	325	if (!has_sha1_file(sha1))
fb9040cc LT	326	return 0;
f0243f26 JS	327	if (nr_has < MAX_HAS) {
	328	struct object *o = lookup_object(sha1);
	329	if (!(o && o->parsed))
	330	o = parse_object(sha1);
	331	if (!o)
	332	die("oops (%s)", sha1_to_hex(sha1));
1974632c	333	if (o->type == OBJ_COMMIT) {
f0243f26 JS	334	struct commit_list *parents;
	335	if (o->flags & THEY_HAVE)
	336	return 0;
	337	o->flags \|= THEY_HAVE;
	338	for (parents = ((struct commit*)o)->parents;
	339	parents;
	340	parents = parents->next)
	341	parents->item->object.flags \|= THEY_HAVE;
	342	}
	343	memcpy(has_sha1[nr_has++], sha1, 20);
fb9040cc LT	344	}
fb9040cc LT	345	return 1;
def88e9a LT	346	}
	347
	348	static int get_common_commits(void)
	349	{
	350	static char line[1000];
1bd8c8f0	351	unsigned char sha1[20], last_sha1[20];
def88e9a LT	352	int len;
def88e9a LT	353
f0243f26 JS	354	track_object_refs = 0;
	355	save_commit_buffer = 0;
	356
def88e9a LT	357	for(;;) {
def88e9a LT	358	len = packet_read_line(0, line, sizeof(line));
960deccb	359	reset_timeout();
def88e9a LT	360
def88e9a LT	361	if (!len) {
1bd8c8f0 JS	362	if (nr_has == 0 \|\| multi_ack)
1bd8c8f0 JS	363	packet_write(1, "NAK\n");
def88e9a LT	364	continue;
def88e9a LT	365	}
75bfc6c2	366	len = strip(line, len);
def88e9a	367	if (!strncmp(line, "have ", 5)) {
1bd8c8f0 JS	368	if (got_sha1(line+5, sha1) &&
	369	(multi_ack \|\| nr_has == 1)) {
	370	if (nr_has >= MAX_HAS)
	371	multi_ack = 0;
	372	packet_write(1, "ACK %s%s\n",
	373	sha1_to_hex(sha1),
	374	multi_ack ? " continue" : "");
	375	if (multi_ack)
	376	memcpy(last_sha1, sha1, 20);
af2d3aa4	377	}
def88e9a LT	378	continue;
	379	}
	380	if (!strcmp(line, "done")) {
1bd8c8f0 JS	381	if (nr_has > 0) {
	382	if (multi_ack)
	383	packet_write(1, "ACK %s\n",
	384	sha1_to_hex(last_sha1));
	385	return 0;
	386	}
def88e9a LT	387	packet_write(1, "NAK\n");
	388	return -1;
	389	}
	390	die("git-upload-pack: expected SHA1 list, got '%s'", line);
	391	}
def88e9a LT	392	}
def88e9a LT	393
fb9040cc LT	394	static int receive_needs(void)
	395	{
	396	static char line[1000];
	397	int len, needs;
	398
	399	needs = 0;
	400	for (;;) {
565ebbf7	401	struct object *o;
e091eb93	402	unsigned char dummy[20], *sha1_buf;
fb9040cc	403	len = packet_read_line(0, line, sizeof(line));
960deccb	404	reset_timeout();
fb9040cc LT	405	if (!len)
	406	return needs;
	407
e091eb93 JH	408	sha1_buf = dummy;
	409	if (needs == MAX_NEEDS) {
	410	fprintf(stderr,
	411	"warning: supporting only a max of %d requests. "
	412	"sending everything instead.\n",
	413	MAX_NEEDS);
	414	}
	415	else if (needs < MAX_NEEDS)
	416	sha1_buf = needs_sha1[needs];
	417
	418	if (strncmp("want ", line, 5) \|\| get_sha1_hex(line+5, sha1_buf))
	419	die("git-upload-pack: protocol error, "
	420	"expected to get sha, not '%s'", line);
1bd8c8f0 JS	421	if (strstr(line+45, "multi_ack"))
1bd8c8f0 JS	422	multi_ack = 1;
b19696c2 JH	423	if (strstr(line+45, "thin-pack"))
b19696c2 JH	424	use_thin_pack = 1;
583b7ea3 JH	425	if (strstr(line+45, "side-band"))
583b7ea3 JH	426	use_sideband = 1;
565ebbf7 JH	427
	428	/* We have sent all our refs already, and the other end
	429	* should have chosen out of them; otherwise they are
	430	* asking for nonsense.
	431	*
	432	* Hmph. We may later want to allow "want" line that
	433	* asks for something like "master~10" (symbolic)...
	434	* would it make sense? I don't know.
	435	*/
	436	o = lookup_object(sha1_buf);
	437	if (!o \|\| !(o->flags & OUR_REF))
	438	die("git-upload-pack: not our ref %s", line+5);
	439	if (!(o->flags & WANTED)) {
	440	o->flags \|= WANTED;
	441	needs++;
	442	}
fb9040cc LT	443	}
	444	}
	445
def88e9a LT	446	static int send_ref(const char refname, const unsigned char sha1)
def88e9a LT	447	{
554fe20d	448	static const char *capabilities = "multi_ack thin-pack side-band";
f6b42a81 JH	449	struct object *o = parse_object(sha1);
f6b42a81 JH	450
b5b16990 CW	451	if (!o)
	452	die("git-upload-pack: cannot find object %s:", sha1_to_hex(sha1));
	453
1f5881bb JS	454	if (capabilities)
	455	packet_write(1, "%s %s%c%s\n", sha1_to_hex(sha1), refname,
	456	0, capabilities);
	457	else
	458	packet_write(1, "%s %s\n", sha1_to_hex(sha1), refname);
	459	capabilities = NULL;
565ebbf7 JH	460	if (!(o->flags & OUR_REF)) {
	461	o->flags \|= OUR_REF;
	462	nr_our_refs++;
	463	}
1974632c	464	if (o->type == OBJ_TAG) {
9534f40b	465	o = deref_tag(o, refname, 0);
f6b42a81 JH	466	packet_write(1, "%s %s^{}\n", sha1_to_hex(o->sha1), refname);
f6b42a81 JH	467	}
def88e9a LT	468	return 0;
	469	}
	470
	471	static int upload_pack(void)
	472	{
960deccb	473	reset_timeout();
723c31fe	474	head_ref(send_ref);
def88e9a LT	475	for_each_ref(send_ref);
def88e9a LT	476	packet_flush(1);
fb9040cc LT	477	nr_needs = receive_needs();
	478	if (!nr_needs)
	479	return 0;
def88e9a	480	get_common_commits();
fb9040cc	481	create_pack_file();
def88e9a LT	482	return 0;
	483	}
	484
	485	int main(int argc, char **argv)
	486	{
8d630132	487	char *dir;
960deccb PA	488	int i;
	489	int strict = 0;
	490
	491	for (i = 1; i < argc; i++) {
	492	char *arg = argv[i];
	493
	494	if (arg[0] != '-')
	495	break;
	496	if (!strcmp(arg, "--strict")) {
	497	strict = 1;
	498	continue;
	499	}
	500	if (!strncmp(arg, "--timeout=", 10)) {
	501	timeout = atoi(arg+10);
	502	continue;
	503	}
	504	if (!strcmp(arg, "--")) {
	505	i++;
	506	break;
	507	}
	508	}
	509
	510	if (i != argc-1)
def88e9a	511	usage(upload_pack_usage);
960deccb	512	dir = argv[i];
113b9475	513
8d630132 AE	514	if (!enter_repo(dir, strict))
8d630132 AE	515	die("'%s': unable to chdir or not a git archive", dir);
960deccb	516
def88e9a LT	517	upload_pack();
	518	return 0;
	519	}