[thirdparty/git.git] / connect.c

#include "git-compat-util.h"
#include "cache.h"
#include "pkt-line.h"
#include "quote.h"
#include "refs.h"
#include "run-command.h"
#include "remote.h"

static char *server_capabilities;

static int check_ref(const char *name, int len, unsigned int flags)
{
	if (!flags)
		return 1;

	if (len < 5 || memcmp(name, "refs/", 5))
		return 0;

	/* Skip the "refs/" part */
	name += 5;
	len -= 5;

	/* REF_NORMAL means that we don't want the magic fake tag refs */
	if ((flags & REF_NORMAL) && check_ref_format(name) < 0)
		return 0;

	/* REF_HEADS means that we want regular branch heads */
	if ((flags & REF_HEADS) && !memcmp(name, "heads/", 6))
		return 1;

	/* REF_TAGS means that we want tags */
	if ((flags & REF_TAGS) && !memcmp(name, "tags/", 5))
		return 1;

	/* All type bits clear means that we are ok with anything */
	return !(flags & ~REF_NORMAL);
}

int check_ref_type(const struct ref *ref, int flags)
{
	return check_ref(ref->name, strlen(ref->name), flags);
}

static void add_extra_have(struct extra_have_objects *extra, unsigned char *sha1)
{
	ALLOC_GROW(extra->array, extra->nr + 1, extra->alloc);
	hashcpy(&(extra->array[extra->nr][0]), sha1);
	extra->nr++;
}

/*
 * Read all the refs from the other end
 */
struct ref **get_remote_heads(int in, struct ref **list,
			      int nr_match, char **match,
			      unsigned int flags,
			      struct extra_have_objects *extra_have)
{
	*list = NULL;
	for (;;) {
		struct ref *ref;
		unsigned char old_sha1[20];
		static char buffer[1000];
		char *name;
		int len, name_len;

		len = packet_read_line(in, buffer, sizeof(buffer));
		if (!len)
			break;
		if (buffer[len-1] == '\n')
			buffer[--len] = 0;

		if (len < 42 || get_sha1_hex(buffer, old_sha1) || buffer[40] != ' ')
			die("protocol error: expected sha/ref, got '%s'", buffer);
		name = buffer + 41;

		name_len = strlen(name);
		if (len != name_len + 41) {
			free(server_capabilities);
			server_capabilities = xstrdup(name + name_len + 1);
		}

		if (extra_have &&
		    name_len == 5 && !memcmp(".have", name, 5)) {
			add_extra_have(extra_have, old_sha1);
			continue;
		}

		if (!check_ref(name, name_len, flags))
			continue;
		if (nr_match && !path_match(name, nr_match, match))
			continue;
		ref = alloc_ref(name_len + 1);
		hashcpy(ref->old_sha1, old_sha1);
		memcpy(ref->name, buffer + 41, name_len + 1);
		*list = ref;
		list = &ref->next;
	}
	return list;
}

int server_supports(const char *feature)
{
	return server_capabilities &&
		strstr(server_capabilities, feature) != NULL;
}

int get_ack(int fd, unsigned char *result_sha1)
{
	static char line[1000];
	int len = packet_read_line(fd, line, sizeof(line));

	if (!len)
		die("git fetch-pack: expected ACK/NAK, got EOF");
	if (line[len-1] == '\n')
		line[--len] = 0;
	if (!strcmp(line, "NAK"))
		return 0;
	if (!prefixcmp(line, "ACK ")) {
		if (!get_sha1_hex(line+4, result_sha1)) {
			if (strstr(line+45, "continue"))
				return 2;
			return 1;
		}
	}
	die("git fetch_pack: expected ACK/NAK, got '%s'", line);
}

int path_match(const char *path, int nr, char **match)
{
	int i;
	int pathlen = strlen(path);

	for (i = 0; i < nr; i++) {
		char *s = match[i];
		int len = strlen(s);

		if (!len || len > pathlen)
			continue;
		if (memcmp(path + pathlen - len, s, len))
			continue;
		if (pathlen > len && path[pathlen - len - 1] != '/')
			continue;
		*s = 0;
		return (i + 1);
	}
	return 0;
}

enum protocol {
	PROTO_LOCAL = 1,
	PROTO_SSH,
	PROTO_GIT,
};

static enum protocol get_protocol(const char *name)
{
	if (!strcmp(name, "ssh"))
		return PROTO_SSH;
	if (!strcmp(name, "git"))
		return PROTO_GIT;
	if (!strcmp(name, "git+ssh"))
		return PROTO_SSH;
	if (!strcmp(name, "ssh+git"))
		return PROTO_SSH;
	if (!strcmp(name, "file"))
		return PROTO_LOCAL;
	die("I don't handle protocol '%s'", name);
}

#define STR_(s)	# s
#define STR(s)	STR_(s)

#ifndef NO_IPV6

static const char *ai_name(const struct addrinfo *ai)
{
	static char addr[INET_ADDRSTRLEN];
	if ( AF_INET == ai->ai_family ) {
		struct sockaddr_in *in;
		in = (struct sockaddr_in *)ai->ai_addr;
		inet_ntop(ai->ai_family, &in->sin_addr, addr, sizeof(addr));
	} else if ( AF_INET6 == ai->ai_family ) {
		struct sockaddr_in6 *in;
		in = (struct sockaddr_in6 *)ai->ai_addr;
		inet_ntop(ai->ai_family, &in->sin6_addr, addr, sizeof(addr));
	} else {
		strcpy(addr, "(unknown)");
	}
	return addr;
}

/*
 * Returns a connected socket() fd, or else die()s.
 */
static int git_tcp_connect_sock(char *host, int flags)
{
	int sockfd = -1, saved_errno = 0;
	char *colon, *end;
	const char *port = STR(DEFAULT_GIT_PORT);
	struct addrinfo hints, *ai0, *ai;
	int gai;
	int cnt = 0;

	if (host[0] == '[') {
		end = strchr(host + 1, ']');
		if (end) {
			*end = 0;
			end++;
			host++;
		} else
			end = host;
	} else
		end = host;
	colon = strchr(end, ':');

	if (colon) {
		*colon = 0;
		port = colon + 1;
		if (!*port)
			port = "<none>";
	}

	memset(&hints, 0, sizeof(hints));
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_protocol = IPPROTO_TCP;

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "Looking up %s ... ", host);

	gai = getaddrinfo(host, port, &hints, &ai);
	if (gai)
		die("Unable to look up %s (port %s) (%s)", host, port, gai_strerror(gai));

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "done.\nConnecting to %s (port %s) ... ", host, port);

	for (ai0 = ai; ai; ai = ai->ai_next) {
		sockfd = socket(ai->ai_family,
				ai->ai_socktype, ai->ai_protocol);
		if (sockfd < 0) {
			saved_errno = errno;
			continue;
		}
		if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) < 0) {
			saved_errno = errno;
			fprintf(stderr, "%s[%d: %s]: errno=%s\n",
				host,
				cnt,
				ai_name(ai),
				strerror(saved_errno));
			close(sockfd);
			sockfd = -1;
			continue;
		}
		if (flags & CONNECT_VERBOSE)
			fprintf(stderr, "%s ", ai_name(ai));
		break;
	}

	freeaddrinfo(ai0);

	if (sockfd < 0)
		die("unable to connect a socket (%s)", strerror(saved_errno));

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "done.\n");

	return sockfd;
}

#else /* NO_IPV6 */

/*
 * Returns a connected socket() fd, or else die()s.
 */
static int git_tcp_connect_sock(char *host, int flags)
{
	int sockfd = -1, saved_errno = 0;
	char *colon, *end;
	char *port = STR(DEFAULT_GIT_PORT), *ep;
	struct hostent *he;
	struct sockaddr_in sa;
	char **ap;
	unsigned int nport;
	int cnt;

	if (host[0] == '[') {
		end = strchr(host + 1, ']');
		if (end) {
			*end = 0;
			end++;
			host++;
		} else
			end = host;
	} else
		end = host;
	colon = strchr(end, ':');

	if (colon) {
		*colon = 0;
		port = colon + 1;
	}

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "Looking up %s ... ", host);

	he = gethostbyname(host);
	if (!he)
		die("Unable to look up %s (%s)", host, hstrerror(h_errno));
	nport = strtoul(port, &ep, 10);
	if ( ep == port || *ep ) {
		/* Not numeric */
		struct servent *se = getservbyname(port,"tcp");
		if ( !se )
			die("Unknown port %s\n", port);
		nport = se->s_port;
	}

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "done.\nConnecting to %s (port %s) ... ", host, port);

	for (cnt = 0, ap = he->h_addr_list; *ap; ap++, cnt++) {
		sockfd = socket(he->h_addrtype, SOCK_STREAM, 0);
		if (sockfd < 0) {
			saved_errno = errno;
			continue;
		}

		memset(&sa, 0, sizeof sa);
		sa.sin_family = he->h_addrtype;
		sa.sin_port = htons(nport);
		memcpy(&sa.sin_addr, *ap, he->h_length);

		if (connect(sockfd, (struct sockaddr *)&sa, sizeof sa) < 0) {
			saved_errno = errno;
			fprintf(stderr, "%s[%d: %s]: errno=%s\n",
				host,
				cnt,
				inet_ntoa(*(struct in_addr *)&sa.sin_addr),
				strerror(saved_errno));
			close(sockfd);
			sockfd = -1;
			continue;
		}
		if (flags & CONNECT_VERBOSE)
			fprintf(stderr, "%s ",
				inet_ntoa(*(struct in_addr *)&sa.sin_addr));
		break;
	}

	if (sockfd < 0)
		die("unable to connect a socket (%s)", strerror(saved_errno));

	if (flags & CONNECT_VERBOSE)
		fprintf(stderr, "done.\n");

	return sockfd;
}

#endif /* NO_IPV6 */


static void git_tcp_connect(int fd[2], char *host, int flags)
{
	int sockfd = git_tcp_connect_sock(host, flags);

	fd[0] = sockfd;
	fd[1] = dup(sockfd);
}


static char *git_proxy_command;
static const char *rhost_name;
static int rhost_len;

static int git_proxy_command_options(const char *var, const char *value,
		void *cb)
{
	if (!strcmp(var, "core.gitproxy")) {
		const char *for_pos;
		int matchlen = -1;
		int hostlen;

		if (git_proxy_command)
			return 0;
		if (!value)
			return config_error_nonbool(var);
		/* [core]
		 * ;# matches www.kernel.org as well
		 * gitproxy = netcatter-1 for kernel.org
		 * gitproxy = netcatter-2 for sample.xz
		 * gitproxy = netcatter-default
		 */
		for_pos = strstr(value, " for ");
		if (!for_pos)
			/* matches everybody */
			matchlen = strlen(value);
		else {
			hostlen = strlen(for_pos + 5);
			if (rhost_len < hostlen)
				matchlen = -1;
			else if (!strncmp(for_pos + 5,
					  rhost_name + rhost_len - hostlen,
					  hostlen) &&
				 ((rhost_len == hostlen) ||
				  rhost_name[rhost_len - hostlen -1] == '.'))
				matchlen = for_pos - value;
			else
				matchlen = -1;
		}
		if (0 <= matchlen) {
			/* core.gitproxy = none for kernel.org */
			if (matchlen == 4 &&
			    !memcmp(value, "none", 4))
				matchlen = 0;
			git_proxy_command = xmemdupz(value, matchlen);
		}
		return 0;
	}

	return git_default_config(var, value, cb);
}

static int git_use_proxy(const char *host)
{
	rhost_name = host;
	rhost_len = strlen(host);
	git_proxy_command = getenv("GIT_PROXY_COMMAND");
	git_config(git_proxy_command_options, NULL);
	rhost_name = NULL;
	return (git_proxy_command && *git_proxy_command);
}

static void git_proxy_connect(int fd[2], char *host)
{
	const char *port = STR(DEFAULT_GIT_PORT);
	char *colon, *end;
	const char *argv[4];
	struct child_process proxy;

	if (host[0] == '[') {
		end = strchr(host + 1, ']');
		if (end) {
			*end = 0;
			end++;
			host++;
		} else
			end = host;
	} else
		end = host;
	colon = strchr(end, ':');

	if (colon) {
		*colon = 0;
		port = colon + 1;
	}

	argv[0] = git_proxy_command;
	argv[1] = host;
	argv[2] = port;
	argv[3] = NULL;
	memset(&proxy, 0, sizeof(proxy));
	proxy.argv = argv;
	proxy.in = -1;
	proxy.out = -1;
	if (start_command(&proxy))
		die("cannot start proxy %s", argv[0]);
	fd[0] = proxy.out; /* read from proxy stdout */
	fd[1] = proxy.in;  /* write to proxy stdin */
}

#define MAX_CMD_LEN 1024

char *get_port(char *host)
{
	char *end;
	char *p = strchr(host, ':');

	if (p) {
		strtol(p+1, &end, 10);
		if (*end == '\0') {
			*p = '\0';
			return p+1;
		}
	}

	return NULL;
}

static struct child_process no_fork;

/*
 * This returns a dummy child_process if the transport protocol does not
 * need fork(2), or a struct child_process object if it does.  Once done,
 * finish the connection with finish_connect() with the value returned from
 * this function (it is safe to call finish_connect() with NULL to support
 * the former case).
 *
 * If it returns, the connect is successful; it just dies on errors (this
 * will hopefully be changed in a libification effort, to return NULL when
 * the connection failed).
 */
struct child_process *git_connect(int fd[2], const char *url_orig,
				  const char *prog, int flags)
{
	char *url = xstrdup(url_orig);
	char *host, *path = url;
	char *end;
	int c;
	struct child_process *conn;
	enum protocol protocol = PROTO_LOCAL;
	int free_path = 0;
	char *port = NULL;
	const char **arg;
	struct strbuf cmd;

	/* Without this we cannot rely on waitpid() to tell
	 * what happened to our children.
	 */
	signal(SIGCHLD, SIG_DFL);

	host = strstr(url, "://");
	if(host) {
		*host = '\0';
		protocol = get_protocol(url);
		host += 3;
		c = '/';
	} else {
		host = url;
		c = ':';
	}

	if (host[0] == '[') {
		end = strchr(host + 1, ']');
		if (end) {
			*end = 0;
			end++;
			host++;
		} else
			end = host;
	} else
		end = host;

	path = strchr(end, c);
	if (path && !has_dos_drive_prefix(end)) {
		if (c == ':') {
			protocol = PROTO_SSH;
			*path++ = '\0';
		}
	} else
		path = end;

	if (!path || !*path)
		die("No path specified. See 'man git-pull' for valid url syntax");

	/*
	 * null-terminate hostname and point path to ~ for URL's like this:
	 *    ssh://host.xz/~user/repo
	 */
	if (protocol != PROTO_LOCAL && host != url) {
		char *ptr = path;
		if (path[1] == '~')
			path++;
		else {
			path = xstrdup(ptr);
			free_path = 1;
		}

		*ptr = '\0';
	}

	/*
	 * Add support for ssh port: ssh://host.xy:<port>/...
	 */
	if (protocol == PROTO_SSH && host != url)
		port = get_port(host);

	if (protocol == PROTO_GIT) {
		/* These underlying connection commands die() if they
		 * cannot connect.
		 */
		char *target_host = xstrdup(host);
		if (git_use_proxy(host))
			git_proxy_connect(fd, host);
		else
			git_tcp_connect(fd, host, flags);
		/*
		 * Separate original protocol components prog and path
		 * from extended components with a NUL byte.
		 */
		packet_write(fd[1],
			     "%s %s%chost=%s%c",
			     prog, path, 0,
			     target_host, 0);
		free(target_host);
		free(url);
		if (free_path)
			free(path);
		return &no_fork;
	}

	conn = xcalloc(1, sizeof(*conn));

	strbuf_init(&cmd, MAX_CMD_LEN);
	strbuf_addstr(&cmd, prog);
	strbuf_addch(&cmd, ' ');
	sq_quote_buf(&cmd, path);
	if (cmd.len >= MAX_CMD_LEN)
		die("command line too long");

	conn->in = conn->out = -1;
	conn->argv = arg = xcalloc(6, sizeof(*arg));
	if (protocol == PROTO_SSH) {
		const char *ssh = getenv("GIT_SSH");
		if (!ssh) ssh = "ssh";

		*arg++ = ssh;
		if (port) {
			*arg++ = "-p";
			*arg++ = port;
		}
		*arg++ = host;
	}
	else {
		/* remove these from the environment */
		const char *env[] = {
			ALTERNATE_DB_ENVIRONMENT,
			DB_ENVIRONMENT,
			GIT_DIR_ENVIRONMENT,
			GIT_WORK_TREE_ENVIRONMENT,
			GRAFT_ENVIRONMENT,
			INDEX_ENVIRONMENT,
			NULL
		};
		conn->env = env;
		*arg++ = "sh";
		*arg++ = "-c";
	}
	*arg++ = cmd.buf;
	*arg = NULL;

	if (start_command(conn))
		die("unable to fork");

	fd[0] = conn->out; /* read from child's stdout */
	fd[1] = conn->in;  /* write to child's stdin */
	strbuf_release(&cmd);
	free(url);
	if (free_path)
		free(path);
	return conn;
}

int finish_connect(struct child_process *conn)
{
	int code;
	if (!conn || conn == &no_fork)
		return 0;

	code = finish_command(conn);
	free(conn->argv);
	free(conn);
	return code;
}
Commit	Line	Data
731043fd	1	#include "git-compat-util.h"
f7192598	2	#include "cache.h"
41cb7488	3	#include "pkt-line.h"
b10d0ec7	4	#include "quote.h"
6abf5c0c	5	#include "refs.h"
15a1c012	6	#include "run-command.h"
6b62816c	7	#include "remote.h"
f7192598	8
96f1e58f	9	static char *server_capabilities;
211b5f9e	10
2718ff09 LT	11	static int check_ref(const char *name, int len, unsigned int flags)
	12	{
	13	if (!flags)
	14	return 1;
	15
c41e20b3	16	if (len < 5 \|\| memcmp(name, "refs/", 5))
2718ff09 LT	17	return 0;
	18
	19	/* Skip the "refs/" part */
	20	name += 5;
	21	len -= 5;
	22
	23	/* REF_NORMAL means that we don't want the magic fake tag refs */
	24	if ((flags & REF_NORMAL) && check_ref_format(name) < 0)
	25	return 0;
	26
	27	/* REF_HEADS means that we want regular branch heads */
	28	if ((flags & REF_HEADS) && !memcmp(name, "heads/", 6))
	29	return 1;
	30
	31	/* REF_TAGS means that we want tags */
	32	if ((flags & REF_TAGS) && !memcmp(name, "tags/", 5))
	33	return 1;
	34
	35	/* All type bits clear means that we are ok with anything */
	36	return !(flags & ~REF_NORMAL);
	37	}
	38
4577370e DB	39	int check_ref_type(const struct ref *ref, int flags)
	40	{
	41	return check_ref(ref->name, strlen(ref->name), flags);
	42	}
	43
40c155ff JH	44	static void add_extra_have(struct extra_have_objects extra, unsigned char sha1)
	45	{
	46	ALLOC_GROW(extra->array, extra->nr + 1, extra->alloc);
	47	hashcpy(&(extra->array[extra->nr][0]), sha1);
	48	extra->nr++;
	49	}
	50
d1c133f5 LT	51	/*
	52	* Read all the refs from the other end
	53	*/
1a7141ff	54	struct ref get_remote_heads(int in, struct ref list,
2718ff09	55	int nr_match, char **match,
40c155ff JH	56	unsigned int flags,
40c155ff JH	57	struct extra_have_objects *extra_have)
d1c133f5 LT	58	{
	59	*list = NULL;
	60	for (;;) {
	61	struct ref *ref;
	62	unsigned char old_sha1[20];
	63	static char buffer[1000];
	64	char *name;
211b5f9e	65	int len, name_len;
d1c133f5 LT	66
	67	len = packet_read_line(in, buffer, sizeof(buffer));
	68	if (!len)
	69	break;
	70	if (buffer[len-1] == '\n')
	71	buffer[--len] = 0;
	72
	73	if (len < 42 \|\| get_sha1_hex(buffer, old_sha1) \|\| buffer[40] != ' ')
	74	die("protocol error: expected sha/ref, got '%s'", buffer);
	75	name = buffer + 41;
1a7141ff	76
211b5f9e JS	77	name_len = strlen(name);
211b5f9e JS	78	if (len != name_len + 41) {
8e0f7003	79	free(server_capabilities);
9befac47	80	server_capabilities = xstrdup(name + name_len + 1);
211b5f9e JS	81	}
211b5f9e JS	82
40c155ff JH	83	if (extra_have &&
	84	name_len == 5 && !memcmp(".have", name, 5)) {
	85	add_extra_have(extra_have, old_sha1);
	86	continue;
	87	}
	88
2718ff09	89	if (!check_ref(name, name_len, flags))
cfee10a7	90	continue;
d1c133f5 LT	91	if (nr_match && !path_match(name, nr_match, match))
d1c133f5 LT	92	continue;
90446a00	93	ref = alloc_ref(name_len + 1);
e702496e	94	hashcpy(ref->old_sha1, old_sha1);
90446a00	95	memcpy(ref->name, buffer + 41, name_len + 1);
d1c133f5 LT	96	*list = ref;
	97	list = &ref->next;
	98	}
	99	return list;
	100	}
	101
211b5f9e JS	102	int server_supports(const char *feature)
211b5f9e JS	103	{
1f5881bb JS	104	return server_capabilities &&
1f5881bb JS	105	strstr(server_capabilities, feature) != NULL;
211b5f9e JS	106	}
211b5f9e JS	107
41cb7488 LT	108	int get_ack(int fd, unsigned char *result_sha1)
	109	{
	110	static char line[1000];
	111	int len = packet_read_line(fd, line, sizeof(line));
	112
	113	if (!len)
7e44c935	114	die("git fetch-pack: expected ACK/NAK, got EOF");
41cb7488 LT	115	if (line[len-1] == '\n')
	116	line[--len] = 0;
	117	if (!strcmp(line, "NAK"))
	118	return 0;
cc44c765	119	if (!prefixcmp(line, "ACK ")) {
c4c86f07 JS	120	if (!get_sha1_hex(line+4, result_sha1)) {
	121	if (strstr(line+45, "continue"))
	122	return 2;
41cb7488	123	return 1;
c4c86f07	124	}
41cb7488	125	}
7e44c935	126	die("git fetch_pack: expected ACK/NAK, got '%s'", line);
41cb7488 LT	127	}
41cb7488 LT	128
013e7c7f LT	129	int path_match(const char path, int nr, char *match)
	130	{
	131	int i;
	132	int pathlen = strlen(path);
	133
	134	for (i = 0; i < nr; i++) {
	135	char *s = match[i];
	136	int len = strlen(s);
	137
	138	if (!len \|\| len > pathlen)
	139	continue;
	140	if (memcmp(path + pathlen - len, s, len))
	141	continue;
	142	if (pathlen > len && path[pathlen - len - 1] != '/')
	143	continue;
	144	*s = 0;
9546010b	145	return (i + 1);
013e7c7f LT	146	}
	147	return 0;
	148	}
	149
2386d658 LT	150	enum protocol {
	151	PROTO_LOCAL = 1,
	152	PROTO_SSH,
	153	PROTO_GIT,
	154	};
	155
	156	static enum protocol get_protocol(const char *name)
	157	{
	158	if (!strcmp(name, "ssh"))
	159	return PROTO_SSH;
	160	if (!strcmp(name, "git"))
	161	return PROTO_GIT;
c05186cc LT	162	if (!strcmp(name, "git+ssh"))
	163	return PROTO_SSH;
	164	if (!strcmp(name, "ssh+git"))
	165	return PROTO_SSH;
72a4f4b6 LT	166	if (!strcmp(name, "file"))
72a4f4b6 LT	167	return PROTO_LOCAL;
2386d658 LT	168	die("I don't handle protocol '%s'", name);
	169	}
	170
5ba88448 YH	171	#define STR_(s) # s
5ba88448 YH	172	#define STR(s) STR_(s)
2386d658	173
49744d63	174	#ifndef NO_IPV6
4c505f71	175
ba505322 AR	176	static const char ai_name(const struct addrinfo ai)
	177	{
	178	static char addr[INET_ADDRSTRLEN];
	179	if ( AF_INET == ai->ai_family ) {
	180	struct sockaddr_in *in;
	181	in = (struct sockaddr_in *)ai->ai_addr;
	182	inet_ntop(ai->ai_family, &in->sin_addr, addr, sizeof(addr));
	183	} else if ( AF_INET6 == ai->ai_family ) {
	184	struct sockaddr_in6 *in;
	185	in = (struct sockaddr_in6 *)ai->ai_addr;
	186	inet_ntop(ai->ai_family, &in->sin6_addr, addr, sizeof(addr));
	187	} else {
	188	strcpy(addr, "(unknown)");
	189	}
	190	return addr;
	191	}
	192
5ad312be JL	193	/*
	194	* Returns a connected socket() fd, or else die()s.
	195	*/
7841ce79	196	static int git_tcp_connect_sock(char *host, int flags)
2386d658	197	{
ac3bc6c1	198	int sockfd = -1, saved_errno = 0;
5ba88448	199	char colon, end;
554fe20d	200	const char *port = STR(DEFAULT_GIT_PORT);
5ba88448 YH	201	struct addrinfo hints, ai0, ai;
5ba88448 YH	202	int gai;
ba505322	203	int cnt = 0;
5ba88448 YH	204
	205	if (host[0] == '[') {
	206	end = strchr(host + 1, ']');
	207	if (end) {
	208	*end = 0;
	209	end++;
	210	host++;
	211	} else
	212	end = host;
	213	} else
	214	end = host;
	215	colon = strchr(end, ':');
	216
ce6f8e7e LT	217	if (colon) {
ce6f8e7e LT	218	*colon = 0;
5ba88448	219	port = colon + 1;
608d48b2 LT	220	if (!*port)
608d48b2 LT	221	port = "<none>";
ce6f8e7e	222	}
5ba88448 YH	223
	224	memset(&hints, 0, sizeof(hints));
	225	hints.ai_socktype = SOCK_STREAM;
	226	hints.ai_protocol = IPPROTO_TCP;
	227
7841ce79 MT	228	if (flags & CONNECT_VERBOSE)
	229	fprintf(stderr, "Looking up %s ... ", host);
	230
5ba88448 YH	231	gai = getaddrinfo(host, port, &hints, &ai);
5ba88448 YH	232	if (gai)
608d48b2	233	die("Unable to look up %s (port %s) (%s)", host, port, gai_strerror(gai));
5ba88448	234
7841ce79 MT	235	if (flags & CONNECT_VERBOSE)
	236	fprintf(stderr, "done.\nConnecting to %s (port %s) ... ", host, port);
	237
5ba88448	238	for (ai0 = ai; ai; ai = ai->ai_next) {
5ad312be JL	239	sockfd = socket(ai->ai_family,
5ad312be JL	240	ai->ai_socktype, ai->ai_protocol);
ac3bc6c1 PB	241	if (sockfd < 0) {
ac3bc6c1 PB	242	saved_errno = errno;
5ba88448	243	continue;
ac3bc6c1	244	}
5ba88448	245	if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) < 0) {
ac3bc6c1	246	saved_errno = errno;
7cbf2f24	247	fprintf(stderr, "%s[%d: %s]: errno=%s\n",
ba505322 AR	248	host,
	249	cnt,
	250	ai_name(ai),
ba505322	251	strerror(saved_errno));
5ba88448 YH	252	close(sockfd);
	253	sockfd = -1;
	254	continue;
2386d658	255	}
ba505322 AR	256	if (flags & CONNECT_VERBOSE)
ba505322 AR	257	fprintf(stderr, "%s ", ai_name(ai));
5ba88448	258	break;
2386d658 LT	259	}
2386d658 LT	260
5ba88448	261	freeaddrinfo(ai0);
2386d658	262
2386d658	263	if (sockfd < 0)
ac3bc6c1	264	die("unable to connect a socket (%s)", strerror(saved_errno));
5ba88448	265
7841ce79 MT	266	if (flags & CONNECT_VERBOSE)
	267	fprintf(stderr, "done.\n");
	268
5ad312be	269	return sockfd;
2386d658 LT	270	}
2386d658 LT	271
49744d63	272	#else /* NO_IPV6 */
4c505f71	273
5ad312be JL	274	/*
	275	* Returns a connected socket() fd, or else die()s.
	276	*/
7841ce79	277	static int git_tcp_connect_sock(char *host, int flags)
4c505f71	278	{
ac3bc6c1	279	int sockfd = -1, saved_errno = 0;
4c505f71 PA	280	char colon, end;
	281	char port = STR(DEFAULT_GIT_PORT), ep;
	282	struct hostent *he;
	283	struct sockaddr_in sa;
	284	char **ap;
	285	unsigned int nport;
ba505322	286	int cnt;
4c505f71 PA	287
	288	if (host[0] == '[') {
	289	end = strchr(host + 1, ']');
	290	if (end) {
	291	*end = 0;
	292	end++;
	293	host++;
	294	} else
	295	end = host;
	296	} else
	297	end = host;
	298	colon = strchr(end, ':');
	299
	300	if (colon) {
	301	*colon = 0;
	302	port = colon + 1;
	303	}
	304
7841ce79 MT	305	if (flags & CONNECT_VERBOSE)
	306	fprintf(stderr, "Looking up %s ... ", host);
	307
4c505f71 PA	308	he = gethostbyname(host);
	309	if (!he)
	310	die("Unable to look up %s (%s)", host, hstrerror(h_errno));
	311	nport = strtoul(port, &ep, 10);
	312	if ( ep == port \|\| *ep ) {
	313	/* Not numeric */
	314	struct servent *se = getservbyname(port,"tcp");
	315	if ( !se )
	316	die("Unknown port %s\n", port);
	317	nport = se->s_port;
	318	}
	319
7841ce79 MT	320	if (flags & CONNECT_VERBOSE)
	321	fprintf(stderr, "done.\nConnecting to %s (port %s) ... ", host, port);
	322
ba505322	323	for (cnt = 0, ap = he->h_addr_list; *ap; ap++, cnt++) {
4c505f71	324	sockfd = socket(he->h_addrtype, SOCK_STREAM, 0);
ac3bc6c1 PB	325	if (sockfd < 0) {
ac3bc6c1 PB	326	saved_errno = errno;
4c505f71	327	continue;
ac3bc6c1	328	}
4c505f71 PA	329
	330	memset(&sa, 0, sizeof sa);
	331	sa.sin_family = he->h_addrtype;
6573faff	332	sa.sin_port = htons(nport);
c6164218	333	memcpy(&sa.sin_addr, *ap, he->h_length);
4c505f71 PA	334
4c505f71 PA	335	if (connect(sockfd, (struct sockaddr *)&sa, sizeof sa) < 0) {
ac3bc6c1	336	saved_errno = errno;
7cbf2f24	337	fprintf(stderr, "%s[%d: %s]: errno=%s\n",
ba505322 AR	338	host,
	339	cnt,
	340	inet_ntoa((struct in_addr )&sa.sin_addr),
ba505322	341	strerror(saved_errno));
4c505f71 PA	342	close(sockfd);
	343	sockfd = -1;
	344	continue;
	345	}
ba505322 AR	346	if (flags & CONNECT_VERBOSE)
	347	fprintf(stderr, "%s ",
	348	inet_ntoa((struct in_addr )&sa.sin_addr));
4c505f71 PA	349	break;
	350	}
	351
	352	if (sockfd < 0)
ac3bc6c1	353	die("unable to connect a socket (%s)", strerror(saved_errno));
4c505f71	354
7841ce79 MT	355	if (flags & CONNECT_VERBOSE)
	356	fprintf(stderr, "done.\n");
	357
5ad312be JL	358	return sockfd;
	359	}
	360
	361	#endif /* NO_IPV6 */
	362
	363
7841ce79	364	static void git_tcp_connect(int fd[2], char *host, int flags)
5ad312be	365	{
7841ce79	366	int sockfd = git_tcp_connect_sock(host, flags);
5ad312be	367
4c505f71	368	fd[0] = sockfd;
ec587fde	369	fd[1] = dup(sockfd);
4c505f71 PA	370	}
4c505f71 PA	371
4c505f71	372
96f1e58f DR	373	static char *git_proxy_command;
96f1e58f DR	374	static const char *rhost_name;
e814bc4d	375	static int rhost_len;
f8014776	376
ef90d6d4 JS	377	static int git_proxy_command_options(const char var, const char value,
ef90d6d4 JS	378	void *cb)
f8014776	379	{
e814bc4d	380	if (!strcmp(var, "core.gitproxy")) {
c3df8568 YH	381	const char *for_pos;
	382	int matchlen = -1;
	383	int hostlen;
	384
e814bc4d	385	if (git_proxy_command)
f8014776	386	return 0;
c64b9ad0 JH	387	if (!value)
c64b9ad0 JH	388	return config_error_nonbool(var);
e814bc4d JH	389	/* [core]
	390	* ;# matches www.kernel.org as well
	391	* gitproxy = netcatter-1 for kernel.org
	392	* gitproxy = netcatter-2 for sample.xz
	393	* gitproxy = netcatter-default
	394	*/
c3df8568	395	for_pos = strstr(value, " for ");
e814bc4d JH	396	if (!for_pos)
	397	/* matches everybody */
	398	matchlen = strlen(value);
	399	else {
	400	hostlen = strlen(for_pos + 5);
	401	if (rhost_len < hostlen)
	402	matchlen = -1;
	403	else if (!strncmp(for_pos + 5,
	404	rhost_name + rhost_len - hostlen,
	405	hostlen) &&
	406	((rhost_len == hostlen) \|\|
	407	rhost_name[rhost_len - hostlen -1] == '.'))
	408	matchlen = for_pos - value;
	409	else
	410	matchlen = -1;
	411	}
	412	if (0 <= matchlen) {
	413	/* core.gitproxy = none for kernel.org */
a6080a0a	414	if (matchlen == 4 &&
e814bc4d JH	415	!memcmp(value, "none", 4))
e814bc4d JH	416	matchlen = 0;
182af834	417	git_proxy_command = xmemdupz(value, matchlen);
f8014776	418	}
e814bc4d	419	return 0;
f8014776 PC	420	}
f8014776 PC	421
ef90d6d4	422	return git_default_config(var, value, cb);
f8014776 PC	423	}
f8014776 PC	424
e814bc4d	425	static int git_use_proxy(const char *host)
f8014776	426	{
e814bc4d JH	427	rhost_name = host;
e814bc4d JH	428	rhost_len = strlen(host);
f8014776	429	git_proxy_command = getenv("GIT_PROXY_COMMAND");
ef90d6d4	430	git_config(git_proxy_command_options, NULL);
e814bc4d JH	431	rhost_name = NULL;
e814bc4d JH	432	return (git_proxy_command && *git_proxy_command);
f8014776 PC	433	}
f8014776 PC	434
c78963d2	435	static void git_proxy_connect(int fd[2], char *host)
f8014776	436	{
554fe20d	437	const char *port = STR(DEFAULT_GIT_PORT);
f8014776	438	char colon, end;
15a1c012 SP	439	const char *argv[4];
15a1c012 SP	440	struct child_process proxy;
f8014776 PC	441
	442	if (host[0] == '[') {
	443	end = strchr(host + 1, ']');
	444	if (end) {
	445	*end = 0;
	446	end++;
	447	host++;
	448	} else
	449	end = host;
	450	} else
	451	end = host;
	452	colon = strchr(end, ':');
	453
	454	if (colon) {
	455	*colon = 0;
	456	port = colon + 1;
	457	}
	458
15a1c012 SP	459	argv[0] = git_proxy_command;
	460	argv[1] = host;
	461	argv[2] = port;
	462	argv[3] = NULL;
	463	memset(&proxy, 0, sizeof(proxy));
	464	proxy.argv = argv;
	465	proxy.in = -1;
	466	proxy.out = -1;
	467	if (start_command(&proxy))
	468	die("cannot start proxy %s", argv[0]);
	469	fd[0] = proxy.out; /* read from proxy stdout */
	470	fd[1] = proxy.in; /* write to proxy stdin */
f8014776 PC	471	}
f8014776 PC	472
0f503d77 CC	473	#define MAX_CMD_LEN 1024
0f503d77 CC	474
2e776665 LT	475	char get_port(char host)
	476	{
	477	char *end;
	478	char *p = strchr(host, ':');
	479
	480	if (p) {
	481	strtol(p+1, &end, 10);
	482	if (*end == '\0') {
	483	*p = '\0';
	484	return p+1;
	485	}
	486	}
	487
	488	return NULL;
	489	}
	490
fb32c917 JS	491	static struct child_process no_fork;
fb32c917 JS	492
f7192598	493	/*
fb32c917 JS	494	* This returns a dummy child_process if the transport protocol does not
	495	* need fork(2), or a struct child_process object if it does. Once done,
	496	* finish the connection with finish_connect() with the value returned from
	497	* this function (it is safe to call finish_connect() with NULL to support
	498	* the former case).
f42a5c4e	499	*
fb32c917 JS	500	* If it returns, the connect is successful; it just dies on errors (this
	501	* will hopefully be changed in a libification effort, to return NULL when
	502	* the connection failed).
f7192598	503	*/
4577370e	504	struct child_process git_connect(int fd[2], const char url_orig,
98158e9c	505	const char *prog, int flags)
f7192598	506	{
4577370e	507	char *url = xstrdup(url_orig);
faea9ccb	508	char host, path = url;
356bece0 YH	509	char *end;
356bece0 YH	510	int c;
98158e9c	511	struct child_process *conn;
faea9ccb	512	enum protocol protocol = PROTO_LOCAL;
da2a95b2	513	int free_path = 0;
2e776665	514	char *port = NULL;
f364cb88 JS	515	const char **arg;
f364cb88 JS	516	struct strbuf cmd;
faea9ccb	517
f0b7367c JH	518	/* Without this we cannot rely on waitpid() to tell
	519	* what happened to our children.
	520	*/
	521	signal(SIGCHLD, SIG_DFL);
	522
faea9ccb AE	523	host = strstr(url, "://");
	524	if(host) {
	525	*host = '\0';
	526	protocol = get_protocol(url);
	527	host += 3;
356bece0 YH	528	c = '/';
356bece0 YH	529	} else {
f7192598	530	host = url;
356bece0 YH	531	c = ':';
	532	}
	533
	534	if (host[0] == '[') {
	535	end = strchr(host + 1, ']');
	536	if (end) {
	537	*end = 0;
	538	end++;
	539	host++;
	540	} else
	541	end = host;
	542	} else
	543	end = host;
	544
	545	path = strchr(end, c);
be501813	546	if (path && !has_dos_drive_prefix(end)) {
72a4f4b6	547	if (c == ':') {
faea9ccb	548	protocol = PROTO_SSH;
356bece0	549	*path++ = '\0';
72a4f4b6 LT	550	}
	551	} else
	552	path = end;
2386d658	553
faea9ccb AE	554	if (!path \|\| !*path)
	555	die("No path specified. See 'man git-pull' for valid url syntax");
	556
	557	/*
	558	* null-terminate hostname and point path to ~ for URL's like this:
	559	* ssh://host.xz/~user/repo
	560	*/
	561	if (protocol != PROTO_LOCAL && host != url) {
	562	char *ptr = path;
	563	if (path[1] == '~')
	564	path++;
da2a95b2	565	else {
9befac47	566	path = xstrdup(ptr);
da2a95b2 SH	567	free_path = 1;
da2a95b2 SH	568	}
faea9ccb AE	569
	570	*ptr = '\0';
	571	}
	572
2e776665 LT	573	/*
	574	* Add support for ssh port: ssh://host.xy:<port>/...
	575	*/
	576	if (protocol == PROTO_SSH && host != url)
	577	port = get_port(host);
	578
f8014776	579	if (protocol == PROTO_GIT) {
5ad312be JL	580	/* These underlying connection commands die() if they
	581	* cannot connect.
	582	*/
9befac47	583	char *target_host = xstrdup(host);
e814bc4d	584	if (git_use_proxy(host))
c78963d2	585	git_proxy_connect(fd, host);
da2a95b2	586	else
7841ce79	587	git_tcp_connect(fd, host, flags);
5ad312be JL	588	/*
	589	* Separate original protocol components prog and path
	590	* from extended components with a NUL byte.
	591	*/
	592	packet_write(fd[1],
	593	"%s %s%chost=%s%c",
	594	prog, path, 0,
	595	target_host, 0);
	596	free(target_host);
4577370e	597	free(url);
da2a95b2 SH	598	if (free_path)
da2a95b2 SH	599	free(path);
fb32c917	600	return &no_fork;
f8014776	601	}
2386d658	602
98158e9c	603	conn = xcalloc(1, sizeof(*conn));
2e776665	604
f364cb88 JS	605	strbuf_init(&cmd, MAX_CMD_LEN);
	606	strbuf_addstr(&cmd, prog);
	607	strbuf_addch(&cmd, ' ');
	608	sq_quote_buf(&cmd, path);
	609	if (cmd.len >= MAX_CMD_LEN)
	610	die("command line too long");
	611
	612	conn->in = conn->out = -1;
	613	conn->argv = arg = xcalloc(6, sizeof(*arg));
	614	if (protocol == PROTO_SSH) {
	615	const char *ssh = getenv("GIT_SSH");
	616	if (!ssh) ssh = "ssh";
	617
	618	*arg++ = ssh;
	619	if (port) {
	620	*arg++ = "-p";
	621	*arg++ = port;
4852f723	622	}
f364cb88 JS	623	*arg++ = host;
	624	}
	625	else {
	626	/* remove these from the environment */
	627	const char *env[] = {
	628	ALTERNATE_DB_ENVIRONMENT,
	629	DB_ENVIRONMENT,
	630	GIT_DIR_ENVIRONMENT,
	631	GIT_WORK_TREE_ENVIRONMENT,
	632	GRAFT_ENVIRONMENT,
	633	INDEX_ENVIRONMENT,
	634	NULL
	635	};
	636	conn->env = env;
	637	*arg++ = "sh";
	638	*arg++ = "-c";
016fb48b	639	}
f364cb88 JS	640	*arg++ = cmd.buf;
	641	*arg = NULL;
	642
	643	if (start_command(conn))
	644	die("unable to fork");
	645
	646	fd[0] = conn->out; /* read from child's stdout */
	647	fd[1] = conn->in; /* write to child's stdin */
	648	strbuf_release(&cmd);
4577370e	649	free(url);
da2a95b2 SH	650	if (free_path)
da2a95b2 SH	651	free(path);
98158e9c	652	return conn;
f7192598 LT	653	}
f7192598 LT	654
98158e9c	655	int finish_connect(struct child_process *conn)
f7192598	656	{
f364cb88	657	int code;
fb32c917	658	if (!conn \|\| conn == &no_fork)
f42a5c4e FBH	659	return 0;
f42a5c4e FBH	660
f364cb88 JS	661	code = finish_command(conn);
f364cb88 JS	662	free(conn->argv);
98158e9c	663	free(conn);
f364cb88	664	return code;
f7192598	665	}