Start of "git-send-pack", the local part of sending off a pack
[git/git.git] / send-pack.c
CommitLineData
61221472
LT
1#include "cache.h"
2
3static const char send_pack_usage[] = "git-send-pack [--exec=other] destination [heads]*";
4
5static const char *exec = "git-receive-pack";
6
7static int send_pack(int in, int out)
8{
9 for (;;) {
10 static char buffer[1000];
11 int ret = read(in, buffer, sizeof(buffer));
12 if (ret > 0) {
13 write(1, buffer, ret);
14 continue;
15 }
16 break;
17 }
18 close(out);
19 return 0;
20}
21
22/*
23 * First, make it shell-safe. We do this by just disallowing any
24 * special characters. Somebody who cares can do escaping and let
25 * through the rest. But since we're doing to feed this to ssh as
26 * a command line, we're going to be pretty damn anal for now.
27 */
28static char *shell_safe(char *url)
29{
30 char *n = url;
31 unsigned char c;
32 static const char flags[256] = {
33 ['0'...'9'] = 1,
34 ['a'...'z'] = 1,
35 ['A'...'Z'] = 1,
36 ['.'] = 1, ['/'] = 1,
37 ['-'] = 1, ['+'] = 1,
38 [':'] = 1
39 };
40
41 while ((c = *n++) != 0) {
42 if (flags[c] != 1)
43 die("I don't like '%c'. Sue me.", c);
44 }
45 return url;
46}
47
48/*
49 * Yeah, yeah, fixme. Need to pass in the heads etc.
50 */
51static int setup_connection(int fd[2], char *url, char **heads)
52{
53 char command[1024];
54 const char *host, *path;
55 char *colon;
56 int pipefd[2][2];
57
58 url = shell_safe(url);
59 host = NULL;
60 path = url;
61 colon = strchr(url, ':');
62 if (colon) {
63 *colon = 0;
64 host = url;
65 path = colon+1;
66 }
67 snprintf(command, sizeof(command), "%s %s", exec, path);
68 if (pipe(pipefd[0]) < 0 || pipe(pipefd[1]) < 0)
69 die("unable to create pipe pair for communication");
70 if (!fork()) {
71 dup2(pipefd[1][0], 0);
72 dup2(pipefd[0][1], 1);
73 close(pipefd[0][0]);
74 close(pipefd[0][1]);
75 close(pipefd[1][0]);
76 close(pipefd[1][1]);
77 if (host)
78 execlp("ssh", "ssh", host, command, NULL);
79 else
80 execlp(host, command, NULL);
81 die("exec failed");
82 }
83 fd[0] = pipefd[0][0];
84 fd[1] = pipefd[1][1];
85 close(pipefd[0][1]);
86 close(pipefd[1][0]);
87 return 0;
88}
89
90int main(int argc, char **argv)
91{
92 int i, nr_heads = 0;
93 char *dest = NULL;
94 char **heads = NULL;
95 int fd[2];
96
97 argv++;
98 for (i = 1; i < argc; i++) {
99 char *arg = *argv++;
100
101 if (*arg == '-') {
102 if (!strncmp(arg, "--exec=", 7)) {
103 exec = arg + 7;
104 continue;
105 }
106 usage(send_pack_usage);
107 }
108 dest = arg;
109 heads = argv;
110 nr_heads = argc - i -1;
111 break;
112 }
113 if (!dest)
114 usage(send_pack_usage);
115 if (setup_connection(fd, dest, heads))
116 return 1;
117 return send_pack(fd[0], fd[1]);
118}