| diff --git a/connect.c b/connect.c |
| --- a/connect.c |
| +++ b/connect.c |
| @@ -96,42 +96,57 @@ static enum protocol get_protocol(const |
| die("I don't handle protocol '%s'", name); |
| } |
| |
| -static void lookup_host(const char *host, struct sockaddr *in) |
| -{ |
| - struct addrinfo *res; |
| - int ret; |
| - |
| - ret = getaddrinfo(host, NULL, NULL, &res); |
| - if (ret) |
| - die("Unable to look up %s (%s)", host, gai_strerror(ret)); |
| - *in = *res->ai_addr; |
| - freeaddrinfo(res); |
| -} |
| +#define STR_(s) # s |
| +#define STR(s) STR_(s) |
| |
| static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path) |
| { |
| - struct sockaddr addr; |
| - int port = DEFAULT_GIT_PORT, sockfd; |
| - char *colon; |
| - |
| - colon = strchr(host, ':'); |
| - if (colon) { |
| - char *end; |
| - unsigned long n = strtoul(colon+1, &end, 0); |
| - if (colon[1] && !*end) { |
| - *colon = 0; |
| - port = n; |
| + int sockfd = -1; |
| + char *colon, *end; |
| + char *port = STR(DEFAULT_GIT_PORT); |
| + struct addrinfo hints, *ai0, *ai; |
| + int gai; |
| + |
| + if (host[0] == '[') { |
| + end = strchr(host + 1, ']'); |
| + if (end) { |
| + *end = 0; |
| + end++; |
| + host++; |
| + } else |
| + end = host; |
| + } else |
| + end = host; |
| + colon = strchr(end, ':'); |
| + |
| + if (colon) |
| + port = colon + 1; |
| + |
| + memset(&hints, 0, sizeof(hints)); |
| + hints.ai_socktype = SOCK_STREAM; |
| + hints.ai_protocol = IPPROTO_TCP; |
| + |
| + gai = getaddrinfo(host, port, &hints, &ai); |
| + if (gai) |
| + die("Unable to look up %s (%s)", host, gai_strerror(gai)); |
| + |
| + for (ai0 = ai; ai; ai = ai->ai_next) { |
| + sockfd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); |
| + if (sockfd < 0) |
| + continue; |
| + if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) < 0) { |
| + close(sockfd); |
| + sockfd = -1; |
| + continue; |
| } |
| + break; |
| } |
| |
| - lookup_host(host, &addr); |
| - ((struct sockaddr_in *)&addr)->sin_port = htons(port); |
| + freeaddrinfo(ai0); |
| |
| - sockfd = socket(PF_INET, SOCK_STREAM, IPPROTO_IP); |
| if (sockfd < 0) |
| die("unable to create socket (%s)", strerror(errno)); |
| - if (connect(sockfd, (void *)&addr, sizeof(addr)) < 0) |
| - die("unable to connect (%s)", strerror(errno)); |
| + |
| fd[0] = sockfd; |
| fd[1] = sockfd; |
| packet_write(sockfd, "%s %s\n", prog, path); |
| |
| -- |
| YOSHIFUJI Hideaki @ USAGI Project <yoshfuji@linux-ipv6.org> |
| GPG-FP : 9022 65EB 1ECF 3AD1 0BDF 80D8 4807 F894 E062 0EEA |
| |