[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 > 4 && !prefixcmp(buffer, "ERR "))
			die("remote error: %s", buffer + 4);

		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(buffer + 41);
		hashcpy(ref->old_sha1, old_sha1);
		*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", 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 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;
		const char *rhost_name = cb;
		int rhost_len = strlen(rhost_name);

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