Refactor git_tcp_connect() functions a little.
Add client side sending of "\0host=%s\0" extended arg for git native protocol, backwards compatibly. Signed-off-by: Jon Loeliger <jdl@jdl.com> Signed-off-by: Junio C Hamano <junkio@cox.net>
This commit is contained in:
parent
9941afc051
commit
5ad312bede
62
connect.c
62
connect.c
@ -322,7 +322,10 @@ static enum protocol get_protocol(const char *name)
|
||||
|
||||
#ifndef NO_IPV6
|
||||
|
||||
static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
/*
|
||||
* Returns a connected socket() fd, or else die()s.
|
||||
*/
|
||||
static int git_tcp_connect_sock(char *host)
|
||||
{
|
||||
int sockfd = -1;
|
||||
char *colon, *end;
|
||||
@ -356,7 +359,8 @@ static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
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);
|
||||
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) {
|
||||
@ -372,15 +376,15 @@ static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
if (sockfd < 0)
|
||||
die("unable to connect a socket (%s)", strerror(errno));
|
||||
|
||||
fd[0] = sockfd;
|
||||
fd[1] = sockfd;
|
||||
packet_write(sockfd, "%s %s\n", prog, path);
|
||||
return 0;
|
||||
return sockfd;
|
||||
}
|
||||
|
||||
#else /* NO_IPV6 */
|
||||
|
||||
static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
/*
|
||||
* Returns a connected socket() fd, or else die()s.
|
||||
*/
|
||||
static int git_tcp_connect_sock(char *host)
|
||||
{
|
||||
int sockfd = -1;
|
||||
char *colon, *end;
|
||||
@ -407,7 +411,6 @@ static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
port = colon + 1;
|
||||
}
|
||||
|
||||
|
||||
he = gethostbyname(host);
|
||||
if (!he)
|
||||
die("Unable to look up %s (%s)", host, hstrerror(h_errno));
|
||||
@ -441,14 +444,22 @@ static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
if (sockfd < 0)
|
||||
die("unable to connect a socket (%s)", strerror(errno));
|
||||
|
||||
fd[0] = sockfd;
|
||||
fd[1] = sockfd;
|
||||
packet_write(sockfd, "%s %s\n", prog, path);
|
||||
return 0;
|
||||
return sockfd;
|
||||
}
|
||||
|
||||
#endif /* NO_IPV6 */
|
||||
|
||||
|
||||
static void git_tcp_connect(int fd[2],
|
||||
const char *prog, char *host, char *path)
|
||||
{
|
||||
int sockfd = git_tcp_connect_sock(host);
|
||||
|
||||
fd[0] = sockfd;
|
||||
fd[1] = sockfd;
|
||||
}
|
||||
|
||||
|
||||
static char *git_proxy_command = NULL;
|
||||
static const char *rhost_name = NULL;
|
||||
static int rhost_len;
|
||||
@ -510,7 +521,8 @@ static int git_use_proxy(const char *host)
|
||||
return (git_proxy_command && *git_proxy_command);
|
||||
}
|
||||
|
||||
static int git_proxy_connect(int fd[2], const char *prog, char *host, char *path)
|
||||
static void git_proxy_connect(int fd[2],
|
||||
const char *prog, char *host, char *path)
|
||||
{
|
||||
char *port = STR(DEFAULT_GIT_PORT);
|
||||
char *colon, *end;
|
||||
@ -547,12 +559,12 @@ static int git_proxy_connect(int fd[2], const char *prog, char *host, char *path
|
||||
execlp(git_proxy_command, git_proxy_command, host, port, NULL);
|
||||
die("exec failed");
|
||||
}
|
||||
if (pid < 0)
|
||||
die("fork failed");
|
||||
fd[0] = pipefd[0][0];
|
||||
fd[1] = pipefd[1][1];
|
||||
close(pipefd[0][1]);
|
||||
close(pipefd[1][0]);
|
||||
packet_write(fd[1], "%s %s\n", prog, path);
|
||||
return pid;
|
||||
}
|
||||
|
||||
/*
|
||||
@ -620,14 +632,26 @@ int git_connect(int fd[2], char *url, const char *prog)
|
||||
}
|
||||
|
||||
if (protocol == PROTO_GIT) {
|
||||
int ret;
|
||||
/* These underlying connection commands die() if they
|
||||
* cannot connect.
|
||||
*/
|
||||
char *target_host = strdup(host);
|
||||
if (git_use_proxy(host))
|
||||
ret = git_proxy_connect(fd, prog, host, path);
|
||||
git_proxy_connect(fd, prog, host, path);
|
||||
else
|
||||
ret = git_tcp_connect(fd, prog, host, path);
|
||||
git_tcp_connect(fd, prog, host, path);
|
||||
/*
|
||||
* 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);
|
||||
if (free_path)
|
||||
free(path);
|
||||
return ret;
|
||||
return 0;
|
||||
}
|
||||
|
||||
if (pipe(pipefd[0]) < 0 || pipe(pipefd[1]) < 0)
|
||||
|
9
daemon.c
9
daemon.c
@ -267,12 +267,17 @@ static int upload(char *dir)
|
||||
static int execute(void)
|
||||
{
|
||||
static char line[1000];
|
||||
int len;
|
||||
int pktlen, len;
|
||||
|
||||
alarm(init_timeout ? init_timeout : timeout);
|
||||
len = packet_read_line(0, line, sizeof(line));
|
||||
pktlen = packet_read_line(0, line, sizeof(line));
|
||||
alarm(0);
|
||||
|
||||
len = strlen(line);
|
||||
if (pktlen != len)
|
||||
loginfo("Extended attributes (%d bytes) exist <%.*s>",
|
||||
(int) pktlen - len,
|
||||
(int) pktlen - len, line + len + 1);
|
||||
if (len && line[len-1] == '\n')
|
||||
line[--len] = 0;
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user