[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"
#include "url.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 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)

static void get_host_and_port(char **host, const char **port)
{
	char *colon, *end;

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

#ifndef NO_IPV6

static const char *ai_name(const struct addrinfo *ai)
{
	static char addr[NI_MAXHOST];
	if (getnameinfo(ai->ai_addr, ai->ai_addrlen, addr, sizeof(addr), NULL, 0,
			NI_NUMERICHOST) != 0)
		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;
	const char *port = STR(DEFAULT_GIT_PORT);
	struct addrinfo hints, *ai0, *ai;
	int gai;
	int cnt = 0;

	get_host_and_port(&host, &port);
	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;
	const char *port = STR(DEFAULT_GIT_PORT);
	char *ep;
	struct hostent *he;
	struct sockaddr_in sa;
	char **ap;
	unsigned int nport;
	int cnt;

	get_host_and_port(&host, &port);

	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);
	const char *argv[4];
	struct child_process proxy;

	get_host_and_port(&host, &port);

	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

static 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;
	char *host, *path;
	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);

	if (is_url(url_orig))
		url = url_decode(url_orig);
	else
		url = xstrdup(url_orig);

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

	/*
	 * Don't do destructive transforms with git:// as that
	 * protocol code does '[]' unwrapping of its own.
	 */
	if (host[0] == '[') {
		end = strchr(host + 1, ']');
		if (end) {
			if (protocol != PROTO_GIT) {
				*end = 0;
				host++;
			}
			end++;
		} 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 host header with a NUL byte.
		 *
		 * Note: Do not add any other headers here!  Doing so
		 * will cause older git-daemon servers to crash.
		 */
		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(7, sizeof(*arg));
	if (protocol == PROTO_SSH) {
		const char *ssh = getenv("GIT_SSH");
		int putty = ssh && strcasestr(ssh, "plink");
		if (!ssh) ssh = "ssh";

		*arg++ = ssh;
		if (putty && !strcasestr(ssh, "tortoiseplink"))
			*arg++ = "-batch";
		if (port) {
			/* P is for PuTTY, p is for OpenSSH */
			*arg++ = putty ? "-P" : "-p";
			*arg++ = port;
		}
		*arg++ = host;
	}
	else {
		/* remove repo-local variables from the environment */
		conn->env = local_repo_env;
		conn->use_shell = 1;
	}
	*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;
}

char *git_getpass(const char *prompt)
{
	const char *askpass;
	struct child_process pass;
	const char *args[3];
	static struct strbuf buffer = STRBUF_INIT;

	askpass = getenv("GIT_ASKPASS");
	if (!askpass)
		askpass = askpass_program;
	if (!askpass)
		askpass = getenv("SSH_ASKPASS");
	if (!askpass || !(*askpass)) {
		char *result = getpass(prompt);
		if (!result)
			die_errno("Could not read password");
		return result;
	}

	args[0] = askpass;
	args[1]	= prompt;
	args[2] = NULL;

	memset(&pass, 0, sizeof(pass));
	pass.argv = args;
	pass.out = -1;

	if (start_command(&pass))
		exit(1);

	strbuf_reset(&buffer);
	if (strbuf_read(&buffer, pass.out, 20) < 0)
		die("failed to read password from %s\n", askpass);

	close(pass.out);

	if (finish_command(&pass))
		exit(1);

	strbuf_setlen(&buffer, strcspn(buffer.buf, "\r\n"));

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