[thirdparty/git.git] / transport.c

#include "cache.h"
#include "transport.h"
#include "run-command.h"
#include "http.h"
#include "pkt-line.h"
#include "fetch-pack.h"
#include "walker.h"
#include "bundle.h"

/* Generic functions for using commit walkers */

static int fetch_objs_via_walker(struct transport *transport,
				 int nr_objs, struct ref **to_fetch)
{
	char *dest = xstrdup(transport->url);
	struct walker *walker = transport->data;
	char **objs = xmalloc(nr_objs * sizeof(*objs));
	int i;

	walker->get_all = 1;
	walker->get_tree = 1;
	walker->get_history = 1;
	walker->get_verbosely = transport->verbose;
	walker->get_recover = 0;

	for (i = 0; i < nr_objs; i++)
		objs[i] = xstrdup(sha1_to_hex(to_fetch[i]->old_sha1));

	if (walker_fetch(walker, nr_objs, objs, NULL, NULL))
		die("Fetch failed.");

	for (i = 0; i < nr_objs; i++)
		free(objs[i]);
	free(objs);
	free(dest);
	return 0;
}

static int disconnect_walker(struct transport *transport)
{
	struct walker *walker = transport->data;
	if (walker)
		walker_free(walker);
	return 0;
}

static const struct transport_ops rsync_transport;

static int curl_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
	const char **argv;
	int argc;
	int err;

	argv = xmalloc((refspec_nr + 11) * sizeof(char *));
	argv[0] = "http-push";
	argc = 1;
	if (flags & TRANSPORT_PUSH_ALL)
		argv[argc++] = "--all";
	if (flags & TRANSPORT_PUSH_FORCE)
		argv[argc++] = "--force";
	argv[argc++] = transport->url;
	while (refspec_nr--)
		argv[argc++] = *refspec++;
	argv[argc] = NULL;
	err = run_command_v_opt(argv, RUN_GIT_CMD);
	switch (err) {
	case -ERR_RUN_COMMAND_FORK:
		error("unable to fork for %s", argv[0]);
	case -ERR_RUN_COMMAND_EXEC:
		error("unable to exec %s", argv[0]);
		break;
	case -ERR_RUN_COMMAND_WAITPID:
	case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
	case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
	case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
		error("%s died with strange error", argv[0]);
	}
	return !!err;
}

#ifndef NO_CURL
static int missing__target(int code, int result)
{
	return	/* file:// URL -- do we ever use one??? */
		(result == CURLE_FILE_COULDNT_READ_FILE) ||
		/* http:// and https:// URL */
		(code == 404 && result == CURLE_HTTP_RETURNED_ERROR) ||
		/* ftp:// URL */
		(code == 550 && result == CURLE_FTP_COULDNT_RETR_FILE)
		;
}

#define missing_target(a) missing__target((a)->http_code, (a)->curl_result)

static struct ref *get_refs_via_curl(const struct transport *transport)
{
	struct buffer buffer;
	char *data, *start, *mid;
	char *ref_name;
	char *refs_url;
	int i = 0;

	struct active_request_slot *slot;
	struct slot_results results;

	struct ref *refs = NULL;
	struct ref *ref = NULL;
	struct ref *last_ref = NULL;

	data = xmalloc(4096);
	buffer.size = 4096;
	buffer.posn = 0;
	buffer.buffer = data;

	refs_url = xmalloc(strlen(transport->url) + 11);
	sprintf(refs_url, "%s/info/refs", transport->url);

	http_init();

	slot = get_active_slot();
	slot->results = &results;
	curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
	curl_easy_setopt(slot->curl, CURLOPT_URL, refs_url);
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
	if (start_active_slot(slot)) {
		run_active_slot(slot);
		if (results.curl_result != CURLE_OK) {
			if (missing_target(&results)) {
				free(buffer.buffer);
				return NULL;
			} else {
				free(buffer.buffer);
				error("%s", curl_errorstr);
				return NULL;
			}
		}
	} else {
		free(buffer.buffer);
		error("Unable to start request");
		return NULL;
	}

	http_cleanup();

	data = buffer.buffer;
	start = NULL;
	mid = data;
	while (i < buffer.posn) {
		if (!start)
			start = &data[i];
		if (data[i] == '\t')
			mid = &data[i];
		if (data[i] == '\n') {
			data[i] = 0;
			ref_name = mid + 1;
			ref = xmalloc(sizeof(struct ref) +
				      strlen(ref_name) + 1);
			memset(ref, 0, sizeof(struct ref));
			strcpy(ref->name, ref_name);
			get_sha1_hex(start, ref->old_sha1);
			if (!refs)
				refs = ref;
			if (last_ref)
				last_ref->next = ref;
			last_ref = ref;
			start = NULL;
		}
		i++;
	}

	free(buffer.buffer);

	return refs;
}

static int fetch_objs_via_curl(struct transport *transport,
				 int nr_objs, struct ref **to_fetch)
{
	if (!transport->data)
		transport->data = get_http_walker(transport->url);
	return fetch_objs_via_walker(transport, nr_objs, to_fetch);
}

#else

static struct ref *get_refs_via_curl(const struct transport *transport)
{
	die("Cannot fetch from '%s' without curl ...", transport->url);
	return NULL;
}

static int fetch_objs_via_curl(struct transport *transport,
				 int nr_objs, struct ref **to_fetch)
{
	die("Cannot fetch from '%s' without curl ...", transport->url);
	return -1;
}

#endif

static const struct transport_ops curl_transport = {
	/* set_option */	NULL,
	/* get_refs_list */	get_refs_via_curl,
	/* fetch */		fetch_objs_via_curl,
	/* push */		curl_transport_push,
	/* disconnect */	disconnect_walker
};

struct bundle_transport_data {
	int fd;
	struct bundle_header header;
};

static struct ref *get_refs_from_bundle(const struct transport *transport)
{
	struct bundle_transport_data *data = transport->data;
	struct ref *result = NULL;
	int i;

	if (data->fd > 0)
		close(data->fd);
	data->fd = read_bundle_header(transport->url, &data->header);
	if (data->fd < 0)
		die ("Could not read bundle '%s'.", transport->url);
	for (i = 0; i < data->header.references.nr; i++) {
		struct ref_list_entry *e = data->header.references.list + i;
		struct ref *ref = alloc_ref(strlen(e->name));
		hashcpy(ref->old_sha1, e->sha1);
		strcpy(ref->name, e->name);
		ref->next = result;
		result = ref;
	}
	return result;
}

static int fetch_refs_from_bundle(struct transport *transport,
			       int nr_heads, struct ref **to_fetch)
{
	struct bundle_transport_data *data = transport->data;
	return unbundle(&data->header, data->fd);
}

static int close_bundle(struct transport *transport)
{
	struct bundle_transport_data *data = transport->data;
	if (data->fd > 0)
		close(data->fd);
	return 0;
}

static const struct transport_ops bundle_transport = {
	/* set_option */	NULL,
	/* get_refs_list */	get_refs_from_bundle,
	/* fetch */		fetch_refs_from_bundle,
	/* push */		NULL,
	/* disconnect */	close_bundle
};

struct git_transport_data {
	unsigned thin : 1;
	unsigned keep : 1;

	int unpacklimit;

	int depth;

	const char *uploadpack;
	const char *receivepack;
};

static int set_git_option(struct transport *connection,
			  const char *name, const char *value)
{
	struct git_transport_data *data = connection->data;
	if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
		data->uploadpack = value;
		return 0;
	} else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
		data->receivepack = value;
		return 0;
	} else if (!strcmp(name, TRANS_OPT_THIN)) {
		data->thin = !!value;
		return 0;
	} else if (!strcmp(name, TRANS_OPT_KEEP)) {
		data->keep = !!value;
		return 0;
	} else if (!strcmp(name, TRANS_OPT_UNPACKLIMIT)) {
		data->unpacklimit = atoi(value);
		return 0;
	} else if (!strcmp(name, TRANS_OPT_DEPTH)) {
		if (!value)
			data->depth = 0;
		else
			data->depth = atoi(value);
		return 0;
	}
	return 1;
}

static struct ref *get_refs_via_connect(const struct transport *transport)
{
	struct git_transport_data *data = transport->data;
	struct ref *refs;
	int fd[2];
	pid_t pid;
	char *dest = xstrdup(transport->url);

	pid = git_connect(fd, dest, data->uploadpack, 0);

	if (pid < 0)
		die("Failed to connect to \"%s\"", transport->url);

	get_remote_heads(fd[0], &refs, 0, NULL, 0);
	packet_flush(fd[1]);

	finish_connect(pid);

	free(dest);

	return refs;
}

static int fetch_refs_via_pack(struct transport *transport,
			       int nr_heads, struct ref **to_fetch)
{
	struct git_transport_data *data = transport->data;
	char **heads = xmalloc(nr_heads * sizeof(*heads));
	char **origh = xmalloc(nr_heads * sizeof(*origh));
	struct ref *refs;
	char *dest = xstrdup(transport->url);
	struct fetch_pack_args args;
	int i;

	args.uploadpack = data->uploadpack;
	args.quiet = 0;
	args.keep_pack = data->keep;
	args.unpacklimit = data->unpacklimit;
	args.use_thin_pack = data->thin;
	args.fetch_all = 0;
	args.verbose = transport->verbose;
	args.depth = data->depth;
	args.no_progress = 0;

	setup_fetch_pack(&args);

	for (i = 0; i < nr_heads; i++)
		origh[i] = heads[i] = xstrdup(to_fetch[i]->name);
	refs = fetch_pack(dest, nr_heads, heads, &transport->pack_lockfile);

	for (i = 0; i < nr_heads; i++)
		free(origh[i]);
	free(origh);
	free(heads);
	free_refs(refs);
	free(dest);
	return 0;
}

static int git_transport_push(struct transport *transport, int refspec_nr, const char **refspec, int flags) {
	struct git_transport_data *data = transport->data;
	const char **argv;
	char *rem;
	int argc;
	int err;

	argv = xmalloc((refspec_nr + 11) * sizeof(char *));
	argv[0] = "send-pack";
	argc = 1;
	if (flags & TRANSPORT_PUSH_ALL)
		argv[argc++] = "--all";
	if (flags & TRANSPORT_PUSH_FORCE)
		argv[argc++] = "--force";
	if (data->receivepack) {
		char *rp = xmalloc(strlen(data->receivepack) + 16);
		sprintf(rp, "--receive-pack=%s", data->receivepack);
		argv[argc++] = rp;
	}
	if (data->thin)
		argv[argc++] = "--thin";
	rem = xmalloc(strlen(transport->remote->name) + 10);
	sprintf(rem, "--remote=%s", transport->remote->name);
	argv[argc++] = rem;
	argv[argc++] = transport->url;
	while (refspec_nr--)
		argv[argc++] = *refspec++;
	argv[argc] = NULL;
	err = run_command_v_opt(argv, RUN_GIT_CMD);
	switch (err) {
	case -ERR_RUN_COMMAND_FORK:
		error("unable to fork for %s", argv[0]);
	case -ERR_RUN_COMMAND_EXEC:
		error("unable to exec %s", argv[0]);
		break;
	case -ERR_RUN_COMMAND_WAITPID:
	case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
	case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
	case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
		error("%s died with strange error", argv[0]);
	}
	return !!err;
}

static const struct transport_ops git_transport = {
	/* set_option */	set_git_option,
	/* get_refs_list */	get_refs_via_connect,
	/* fetch */		fetch_refs_via_pack,
	/* push */		git_transport_push
};

static int is_local(const char *url)
{
	const char *colon = strchr(url, ':');
	const char *slash = strchr(url, '/');
	return !colon || (slash && slash < colon);
}

static int is_file(const char *url)
{
	struct stat buf;
	if (stat(url, &buf))
		return 0;
	return S_ISREG(buf.st_mode);
}

struct transport *transport_get(struct remote *remote, const char *url)
{
	struct transport *ret = xcalloc(1, sizeof(*ret));

	ret->remote = remote;
	ret->url = url;

	if (!prefixcmp(url, "rsync://")) {
		ret->ops = &rsync_transport;
	} else if (!prefixcmp(url, "http://")
	        || !prefixcmp(url, "https://")
	        || !prefixcmp(url, "ftp://")) {
		ret->ops = &curl_transport;
	} else if (is_local(url) && is_file(url)) {
		struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
		ret->data = data;
		ret->ops = &bundle_transport;
	} else {
		struct git_transport_data *data = xcalloc(1, sizeof(*data));
		ret->data = data;
		data->thin = 1;
		data->uploadpack = "git-upload-pack";
		if (remote && remote->uploadpack)
			data->uploadpack = remote->uploadpack;
		data->receivepack = "git-receive-pack";
		if (remote && remote->receivepack)
			data->receivepack = remote->receivepack;
		data->unpacklimit = -1;
		ret->ops = &git_transport;
	}

	return ret;
}

int transport_set_option(struct transport *transport,
			 const char *name, const char *value)
{
	if (transport->ops->set_option)
		return transport->ops->set_option(transport, name, value);
	return 1;
}

int transport_push(struct transport *transport,
		   int refspec_nr, const char **refspec, int flags)
{
	if (!transport->ops->push)
		return 1;
	return transport->ops->push(transport, refspec_nr, refspec, flags);
}

struct ref *transport_get_remote_refs(struct transport *transport)
{
	if (!transport->remote_refs)
		transport->remote_refs =
			transport->ops->get_refs_list(transport);
	return transport->remote_refs;
}

int transport_fetch_refs(struct transport *transport, struct ref *refs)
{
	int rc;
	int nr_heads = 0, nr_alloc = 0;
	struct ref **heads = NULL;
	struct ref *rm;

	for (rm = refs; rm; rm = rm->next) {
		if (rm->peer_ref &&
		    !hashcmp(rm->peer_ref->old_sha1, rm->old_sha1))
			continue;
		ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
		heads[nr_heads++] = rm;
	}

	rc = transport->ops->fetch(transport, nr_heads, heads);
	free(heads);
	return rc;
}

void transport_unlock_pack(struct transport *transport)
{
	if (transport->pack_lockfile) {
		unlink(transport->pack_lockfile);
		free(transport->pack_lockfile);
		transport->pack_lockfile = NULL;
	}
}

int transport_disconnect(struct transport *transport)
{
	int ret = 0;
	if (transport->ops->disconnect)
		ret = transport->ops->disconnect(transport);
	free(transport);
	return ret;
}
Commit	Line	Data
9b288516 DB	1	#include "cache.h"
	2	#include "transport.h"
	3	#include "run-command.h"
c29727d5 DB	4	#include "http.h"
	5	#include "pkt-line.h"
	6	#include "fetch-pack.h"
	7	#include "walker.h"
c7a8a162	8	#include "bundle.h"
c29727d5 DB	9
	10	/* Generic functions for using commit walkers */
	11
1788c39c	12	static int fetch_objs_via_walker(struct transport *transport,
425b1393	13	int nr_objs, struct ref **to_fetch)
c29727d5 DB	14	{
	15	char *dest = xstrdup(transport->url);
	16	struct walker *walker = transport->data;
425b1393 SP	17	char *objs = xmalloc(nr_objs sizeof(*objs));
425b1393 SP	18	int i;
c29727d5 DB	19
	20	walker->get_all = 1;
	21	walker->get_tree = 1;
	22	walker->get_history = 1;
	23	walker->get_verbosely = transport->verbose;
	24	walker->get_recover = 0;
	25
425b1393 SP	26	for (i = 0; i < nr_objs; i++)
	27	objs[i] = xstrdup(sha1_to_hex(to_fetch[i]->old_sha1));
	28
b6abb48a	29	if (walker_fetch(walker, nr_objs, objs, NULL, NULL))
c29727d5 DB	30	die("Fetch failed.");
c29727d5 DB	31
425b1393 SP	32	for (i = 0; i < nr_objs; i++)
	33	free(objs[i]);
	34	free(objs);
c29727d5 DB	35	free(dest);
	36	return 0;
	37	}
	38
	39	static int disconnect_walker(struct transport *transport)
	40	{
	41	struct walker *walker = transport->data;
	42	if (walker)
	43	walker_free(walker);
	44	return 0;
	45	}
9b288516 DB	46
	47	static const struct transport_ops rsync_transport;
	48
	49	static int curl_transport_push(struct transport transport, int refspec_nr, const char *refspec, int flags) {
	50	const char **argv;
	51	int argc;
	52	int err;
	53
	54	argv = xmalloc((refspec_nr + 11) * sizeof(char *));
	55	argv[0] = "http-push";
	56	argc = 1;
	57	if (flags & TRANSPORT_PUSH_ALL)
	58	argv[argc++] = "--all";
	59	if (flags & TRANSPORT_PUSH_FORCE)
	60	argv[argc++] = "--force";
	61	argv[argc++] = transport->url;
	62	while (refspec_nr--)
	63	argv[argc++] = *refspec++;
	64	argv[argc] = NULL;
	65	err = run_command_v_opt(argv, RUN_GIT_CMD);
	66	switch (err) {
	67	case -ERR_RUN_COMMAND_FORK:
	68	error("unable to fork for %s", argv[0]);
	69	case -ERR_RUN_COMMAND_EXEC:
	70	error("unable to exec %s", argv[0]);
	71	break;
	72	case -ERR_RUN_COMMAND_WAITPID:
	73	case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
	74	case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
	75	case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
	76	error("%s died with strange error", argv[0]);
	77	}
	78	return !!err;
	79	}
	80
c29727d5 DB	81	#ifndef NO_CURL
	82	static int missing__target(int code, int result)
	83	{
	84	return /* file:// URL -- do we ever use one??? */
	85	(result == CURLE_FILE_COULDNT_READ_FILE) \|\|
	86	/* http:// and https:// URL */
	87	(code == 404 && result == CURLE_HTTP_RETURNED_ERROR) \|\|
	88	/* ftp:// URL */
	89	(code == 550 && result == CURLE_FTP_COULDNT_RETR_FILE)
	90	;
	91	}
	92
	93	#define missing_target(a) missing__target((a)->http_code, (a)->curl_result)
	94
	95	static struct ref get_refs_via_curl(const struct transport transport)
	96	{
	97	struct buffer buffer;
	98	char data, start, *mid;
	99	char *ref_name;
	100	char *refs_url;
	101	int i = 0;
	102
	103	struct active_request_slot *slot;
	104	struct slot_results results;
	105
	106	struct ref *refs = NULL;
	107	struct ref *ref = NULL;
	108	struct ref *last_ref = NULL;
	109
	110	data = xmalloc(4096);
	111	buffer.size = 4096;
	112	buffer.posn = 0;
	113	buffer.buffer = data;
	114
	115	refs_url = xmalloc(strlen(transport->url) + 11);
	116	sprintf(refs_url, "%s/info/refs", transport->url);
	117
	118	http_init();
	119
	120	slot = get_active_slot();
	121	slot->results = &results;
	122	curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
	123	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
	124	curl_easy_setopt(slot->curl, CURLOPT_URL, refs_url);
	125	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
	126	if (start_active_slot(slot)) {
	127	run_active_slot(slot);
	128	if (results.curl_result != CURLE_OK) {
	129	if (missing_target(&results)) {
	130	free(buffer.buffer);
	131	return NULL;
	132	} else {
	133	free(buffer.buffer);
	134	error("%s", curl_errorstr);
	135	return NULL;
	136	}
	137	}
	138	} else {
	139	free(buffer.buffer);
	140	error("Unable to start request");
	141	return NULL;
	142	}
	143
	144	http_cleanup();
145
146	data = buffer.buffer;
147	start = NULL;
148	mid = data;
149	while (i < buffer.posn) {
150	if (!start)
151	start = &data[i];
152	if (data[i] == '\t')
153	mid = &data[i];
154	if (data[i] == '\n') {
155	data[i] = 0;
156	ref_name = mid + 1;
157	ref = xmalloc(sizeof(struct ref) +
158	strlen(ref_name) + 1);
159	memset(ref, 0, sizeof(struct ref));
160	strcpy(ref->name, ref_name);
161	get_sha1_hex(start, ref->old_sha1);
162	if (!refs)
163	refs = ref;
164	if (last_ref)
165	last_ref->next = ref;
166	last_ref = ref;
167	start = NULL;
168	}
169	i++;
170	}
171
172	free(buffer.buffer);
173
174	return refs;
175	}
176
e5f4e214 SP	177	static int fetch_objs_via_curl(struct transport *transport,
	178	int nr_objs, struct ref **to_fetch)
	179	{
	180	if (!transport->data)
	181	transport->data = get_http_walker(transport->url);
	182	return fetch_objs_via_walker(transport, nr_objs, to_fetch);
	183	}
	184
c29727d5 DB	185	#else
	186
	187	static struct ref get_refs_via_curl(const struct transport transport)
	188	{
	189	die("Cannot fetch from '%s' without curl ...", transport->url);
	190	return NULL;
	191	}
	192
e5f4e214 SP	193	static int fetch_objs_via_curl(struct transport *transport,
	194	int nr_objs, struct ref **to_fetch)
	195	{
	196	die("Cannot fetch from '%s' without curl ...", transport->url);
	197	return -1;
	198	}
	199
c29727d5 DB	200	#endif
c29727d5 DB	201
9b288516 DB	202	static const struct transport_ops curl_transport = {
9b288516 DB	203	/* set_option */ NULL,
c29727d5	204	/* get_refs_list */ get_refs_via_curl,
e5f4e214	205	/* fetch */ fetch_objs_via_curl,
c29727d5 DB	206	/* push */ curl_transport_push,
c29727d5 DB	207	/* disconnect */ disconnect_walker
9b288516 DB	208	};
9b288516 DB	209
c7a8a162 JS	210	struct bundle_transport_data {
	211	int fd;
	212	struct bundle_header header;
	213	};
	214
	215	static struct ref get_refs_from_bundle(const struct transport transport)
	216	{
	217	struct bundle_transport_data *data = transport->data;
	218	struct ref *result = NULL;
	219	int i;
	220
	221	if (data->fd > 0)
	222	close(data->fd);
	223	data->fd = read_bundle_header(transport->url, &data->header);
	224	if (data->fd < 0)
	225	die ("Could not read bundle '%s'.", transport->url);
	226	for (i = 0; i < data->header.references.nr; i++) {
	227	struct ref_list_entry *e = data->header.references.list + i;
	228	struct ref *ref = alloc_ref(strlen(e->name));
	229	hashcpy(ref->old_sha1, e->sha1);
	230	strcpy(ref->name, e->name);
	231	ref->next = result;
	232	result = ref;
	233	}
	234	return result;
	235	}
	236
1788c39c	237	static int fetch_refs_from_bundle(struct transport *transport,
425b1393	238	int nr_heads, struct ref **to_fetch)
c7a8a162 JS	239	{
	240	struct bundle_transport_data *data = transport->data;
	241	return unbundle(&data->header, data->fd);
	242	}
	243
	244	static int close_bundle(struct transport *transport)
	245	{
	246	struct bundle_transport_data *data = transport->data;
	247	if (data->fd > 0)
	248	close(data->fd);
	249	return 0;
	250	}
	251
9b288516	252	static const struct transport_ops bundle_transport = {
c7a8a162 JS	253	/* set_option */ NULL,
c7a8a162 JS	254	/* get_refs_list */ get_refs_from_bundle,
425b1393	255	/* fetch */ fetch_refs_from_bundle,
c7a8a162 JS	256	/* push */ NULL,
c7a8a162 JS	257	/* disconnect */ close_bundle
9b288516 DB	258	};
	259
	260	struct git_transport_data {
	261	unsigned thin : 1;
c29727d5 DB	262	unsigned keep : 1;
	263
	264	int unpacklimit;
9b288516	265
c29727d5 DB	266	int depth;
	267
	268	const char *uploadpack;
9b288516 DB	269	const char *receivepack;
	270	};
	271
	272	static int set_git_option(struct transport *connection,
	273	const char name, const char value)
	274	{
	275	struct git_transport_data *data = connection->data;
c29727d5 DB	276	if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
	277	data->uploadpack = value;
	278	return 0;
	279	} else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
9b288516 DB	280	data->receivepack = value;
	281	return 0;
	282	} else if (!strcmp(name, TRANS_OPT_THIN)) {
	283	data->thin = !!value;
	284	return 0;
c29727d5 DB	285	} else if (!strcmp(name, TRANS_OPT_KEEP)) {
	286	data->keep = !!value;
	287	return 0;
	288	} else if (!strcmp(name, TRANS_OPT_UNPACKLIMIT)) {
	289	data->unpacklimit = atoi(value);
	290	return 0;
	291	} else if (!strcmp(name, TRANS_OPT_DEPTH)) {
	292	if (!value)
	293	data->depth = 0;
	294	else
	295	data->depth = atoi(value);
	296	return 0;
9b288516 DB	297	}
	298	return 1;
	299	}
	300
c29727d5 DB	301	static struct ref get_refs_via_connect(const struct transport transport)
	302	{
	303	struct git_transport_data *data = transport->data;
	304	struct ref *refs;
	305	int fd[2];
	306	pid_t pid;
	307	char *dest = xstrdup(transport->url);
	308
	309	pid = git_connect(fd, dest, data->uploadpack, 0);
	310
	311	if (pid < 0)
	312	die("Failed to connect to \"%s\"", transport->url);
	313
	314	get_remote_heads(fd[0], &refs, 0, NULL, 0);
	315	packet_flush(fd[1]);
	316
	317	finish_connect(pid);
	318
	319	free(dest);
	320
	321	return refs;
	322	}
	323
1788c39c	324	static int fetch_refs_via_pack(struct transport *transport,
425b1393	325	int nr_heads, struct ref **to_fetch)
c29727d5 DB	326	{
c29727d5 DB	327	struct git_transport_data *data = transport->data;
425b1393	328	char *heads = xmalloc(nr_heads sizeof(*heads));
e4cd6c7a	329	char *origh = xmalloc(nr_heads sizeof(*origh));
c29727d5 DB	330	struct ref *refs;
	331	char *dest = xstrdup(transport->url);
	332	struct fetch_pack_args args;
425b1393	333	int i;
c29727d5 DB	334
	335	args.uploadpack = data->uploadpack;
	336	args.quiet = 0;
	337	args.keep_pack = data->keep;
	338	args.unpacklimit = data->unpacklimit;
	339	args.use_thin_pack = data->thin;
	340	args.fetch_all = 0;
	341	args.verbose = transport->verbose;
	342	args.depth = data->depth;
	343	args.no_progress = 0;
	344
	345	setup_fetch_pack(&args);
	346
425b1393	347	for (i = 0; i < nr_heads; i++)
e4cd6c7a	348	origh[i] = heads[i] = xstrdup(to_fetch[i]->name);
1788c39c	349	refs = fetch_pack(dest, nr_heads, heads, &transport->pack_lockfile);
c29727d5	350
425b1393	351	for (i = 0; i < nr_heads; i++)
e4cd6c7a SP	352	free(origh[i]);
	353	free(origh);
	354	free(heads);
c29727d5	355	free_refs(refs);
c29727d5 DB	356	free(dest);
	357	return 0;
	358	}
	359
9b288516 DB	360	static int git_transport_push(struct transport transport, int refspec_nr, const char *refspec, int flags) {
	361	struct git_transport_data *data = transport->data;
	362	const char **argv;
	363	char *rem;
	364	int argc;
	365	int err;
	366
	367	argv = xmalloc((refspec_nr + 11) * sizeof(char *));
	368	argv[0] = "send-pack";
	369	argc = 1;
	370	if (flags & TRANSPORT_PUSH_ALL)
	371	argv[argc++] = "--all";
	372	if (flags & TRANSPORT_PUSH_FORCE)
	373	argv[argc++] = "--force";
	374	if (data->receivepack) {
	375	char *rp = xmalloc(strlen(data->receivepack) + 16);
	376	sprintf(rp, "--receive-pack=%s", data->receivepack);
	377	argv[argc++] = rp;
	378	}
	379	if (data->thin)
	380	argv[argc++] = "--thin";
	381	rem = xmalloc(strlen(transport->remote->name) + 10);
	382	sprintf(rem, "--remote=%s", transport->remote->name);
	383	argv[argc++] = rem;
	384	argv[argc++] = transport->url;
	385	while (refspec_nr--)
	386	argv[argc++] = *refspec++;
	387	argv[argc] = NULL;
	388	err = run_command_v_opt(argv, RUN_GIT_CMD);
	389	switch (err) {
	390	case -ERR_RUN_COMMAND_FORK:
	391	error("unable to fork for %s", argv[0]);
	392	case -ERR_RUN_COMMAND_EXEC:
	393	error("unable to exec %s", argv[0]);
	394	break;
	395	case -ERR_RUN_COMMAND_WAITPID:
	396	case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
	397	case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
	398	case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
	399	error("%s died with strange error", argv[0]);
	400	}
	401	return !!err;
	402	}
	403
	404	static const struct transport_ops git_transport = {
	405	/* set_option */ set_git_option,
c29727d5	406	/* get_refs_list */ get_refs_via_connect,
425b1393	407	/* fetch */ fetch_refs_via_pack,
9b288516 DB	408	/* push */ git_transport_push
	409	};
	410
	411	static int is_local(const char *url)
	412	{
	413	const char *colon = strchr(url, ':');
	414	const char *slash = strchr(url, '/');
	415	return !colon \|\| (slash && slash < colon);
	416	}
	417
	418	static int is_file(const char *url)
	419	{
	420	struct stat buf;
	421	if (stat(url, &buf))
	422	return 0;
	423	return S_ISREG(buf.st_mode);
	424	}
	425
e5f4e214	426	struct transport transport_get(struct remote remote, const char *url)
9b288516	427	{
8eb554ae SP	428	struct transport ret = xcalloc(1, sizeof(ret));
	429
	430	ret->remote = remote;
	431	ret->url = url;
8eb554ae	432
9b288516	433	if (!prefixcmp(url, "rsync://")) {
9b288516	434	ret->ops = &rsync_transport;
8eb554ae SP	435	} else if (!prefixcmp(url, "http://")
	436	\|\| !prefixcmp(url, "https://")
	437	\|\| !prefixcmp(url, "ftp://")) {
9b288516	438	ret->ops = &curl_transport;
9b288516	439	} else if (is_local(url) && is_file(url)) {
c7a8a162	440	struct bundle_transport_data data = xcalloc(1, sizeof(data));
c7a8a162	441	ret->data = data;
9b288516 DB	442	ret->ops = &bundle_transport;
	443	} else {
	444	struct git_transport_data data = xcalloc(1, sizeof(data));
9b288516 DB	445	ret->data = data;
9b288516 DB	446	data->thin = 1;
c29727d5 DB	447	data->uploadpack = "git-upload-pack";
	448	if (remote && remote->uploadpack)
	449	data->uploadpack = remote->uploadpack;
9b288516 DB	450	data->receivepack = "git-receive-pack";
	451	if (remote && remote->receivepack)
	452	data->receivepack = remote->receivepack;
c29727d5	453	data->unpacklimit = -1;
9b288516 DB	454	ret->ops = &git_transport;
9b288516 DB	455	}
8eb554ae	456
9b288516 DB	457	return ret;
	458	}
	459
	460	int transport_set_option(struct transport *transport,
	461	const char name, const char value)
	462	{
9b288516	463	if (transport->ops->set_option)
ab865e6e SP	464	return transport->ops->set_option(transport, name, value);
ab865e6e SP	465	return 1;
9b288516 DB	466	}
	467
	468	int transport_push(struct transport *transport,
	469	int refspec_nr, const char **refspec, int flags)
	470	{
	471	if (!transport->ops->push)
	472	return 1;
	473	return transport->ops->push(transport, refspec_nr, refspec, flags);
	474	}
	475
c29727d5 DB	476	struct ref transport_get_remote_refs(struct transport transport)
	477	{
	478	if (!transport->remote_refs)
	479	transport->remote_refs =
	480	transport->ops->get_refs_list(transport);
	481	return transport->remote_refs;
	482	}
	483
c29727d5 DB	484	int transport_fetch_refs(struct transport transport, struct ref refs)
c29727d5 DB	485	{
425b1393	486	int rc;
7a2bff45	487	int nr_heads = 0, nr_alloc = 0;
425b1393	488	struct ref **heads = NULL;
c29727d5	489	struct ref *rm;
c29727d5 DB	490
	491	for (rm = refs; rm; rm = rm->next) {
	492	if (rm->peer_ref &&
	493	!hashcmp(rm->peer_ref->old_sha1, rm->old_sha1))
	494	continue;
7a2bff45	495	ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
425b1393	496	heads[nr_heads++] = rm;
c29727d5 DB	497	}
c29727d5 DB	498
425b1393	499	rc = transport->ops->fetch(transport, nr_heads, heads);
c29727d5	500	free(heads);
425b1393	501	return rc;
c29727d5 DB	502	}
c29727d5 DB	503
1788c39c SP	504	void transport_unlock_pack(struct transport *transport)
	505	{
	506	if (transport->pack_lockfile) {
	507	unlink(transport->pack_lockfile);
	508	free(transport->pack_lockfile);
	509	transport->pack_lockfile = NULL;
	510	}
	511	}
	512
9b288516 DB	513	int transport_disconnect(struct transport *transport)
	514	{
	515	int ret = 0;
	516	if (transport->ops->disconnect)
	517	ret = transport->ops->disconnect(transport);
	518	free(transport);
	519	return ret;
	520	}