]>
Commit | Line | Data |
---|---|---|
f7192598 | 1 | #include "cache.h" |
41cb7488 | 2 | #include "pkt-line.h" |
b10d0ec7 | 3 | #include "quote.h" |
f7192598 | 4 | #include <sys/wait.h> |
2386d658 LT |
5 | #include <sys/socket.h> |
6 | #include <netinet/in.h> | |
7 | #include <arpa/inet.h> | |
8 | #include <netdb.h> | |
f7192598 | 9 | |
d1c133f5 LT |
10 | /* |
11 | * Read all the refs from the other end | |
12 | */ | |
13 | struct ref **get_remote_heads(int in, struct ref **list, int nr_match, char **match) | |
14 | { | |
15 | *list = NULL; | |
16 | for (;;) { | |
17 | struct ref *ref; | |
18 | unsigned char old_sha1[20]; | |
19 | static char buffer[1000]; | |
20 | char *name; | |
21 | int len; | |
22 | ||
23 | len = packet_read_line(in, buffer, sizeof(buffer)); | |
24 | if (!len) | |
25 | break; | |
26 | if (buffer[len-1] == '\n') | |
27 | buffer[--len] = 0; | |
28 | ||
29 | if (len < 42 || get_sha1_hex(buffer, old_sha1) || buffer[40] != ' ') | |
30 | die("protocol error: expected sha/ref, got '%s'", buffer); | |
31 | name = buffer + 41; | |
32 | if (nr_match && !path_match(name, nr_match, match)) | |
33 | continue; | |
f88395ac | 34 | ref = xcalloc(1, sizeof(*ref) + len - 40); |
d1c133f5 | 35 | memcpy(ref->old_sha1, old_sha1, 20); |
d1c133f5 | 36 | memcpy(ref->name, buffer + 41, len - 40); |
d1c133f5 LT |
37 | *list = ref; |
38 | list = &ref->next; | |
39 | } | |
40 | return list; | |
41 | } | |
42 | ||
41cb7488 LT |
43 | int get_ack(int fd, unsigned char *result_sha1) |
44 | { | |
45 | static char line[1000]; | |
46 | int len = packet_read_line(fd, line, sizeof(line)); | |
47 | ||
48 | if (!len) | |
49 | die("git-fetch-pack: expected ACK/NAK, got EOF"); | |
50 | if (line[len-1] == '\n') | |
51 | line[--len] = 0; | |
52 | if (!strcmp(line, "NAK")) | |
53 | return 0; | |
54 | if (!strncmp(line, "ACK ", 3)) { | |
55 | if (!get_sha1_hex(line+4, result_sha1)) | |
56 | return 1; | |
57 | } | |
58 | die("git-fetch_pack: expected ACK/NAK, got '%s'", line); | |
59 | } | |
60 | ||
013e7c7f LT |
61 | int path_match(const char *path, int nr, char **match) |
62 | { | |
63 | int i; | |
64 | int pathlen = strlen(path); | |
65 | ||
66 | for (i = 0; i < nr; i++) { | |
67 | char *s = match[i]; | |
68 | int len = strlen(s); | |
69 | ||
70 | if (!len || len > pathlen) | |
71 | continue; | |
72 | if (memcmp(path + pathlen - len, s, len)) | |
73 | continue; | |
74 | if (pathlen > len && path[pathlen - len - 1] != '/') | |
75 | continue; | |
76 | *s = 0; | |
77 | return 1; | |
78 | } | |
79 | return 0; | |
80 | } | |
81 | ||
f88395ac JH |
82 | struct refspec { |
83 | char *src; | |
84 | char *dst; | |
85 | }; | |
86 | ||
87 | static struct refspec *parse_ref_spec(int nr_refspec, char **refspec) | |
88 | { | |
89 | int i; | |
90 | struct refspec *rs = xmalloc(sizeof(*rs) * (nr_refspec + 1)); | |
91 | for (i = 0; i < nr_refspec; i++) { | |
92 | char *sp, *dp, *ep; | |
93 | sp = refspec[i]; | |
94 | ep = strchr(sp, ':'); | |
95 | if (ep) { | |
96 | dp = ep + 1; | |
97 | *ep = 0; | |
98 | } | |
99 | else | |
100 | dp = sp; | |
101 | rs[i].src = sp; | |
102 | rs[i].dst = dp; | |
103 | } | |
104 | rs[nr_refspec].src = rs[nr_refspec].dst = NULL; | |
105 | return rs; | |
106 | } | |
107 | ||
108 | static int count_refspec_match(const char *pattern, | |
109 | struct ref *refs, | |
110 | struct ref **matched_ref) | |
111 | { | |
112 | int match; | |
113 | int patlen = strlen(pattern); | |
114 | ||
115 | for (match = 0; refs; refs = refs->next) { | |
116 | char *name = refs->name; | |
117 | int namelen = strlen(name); | |
118 | if (namelen < patlen || | |
119 | memcmp(name + namelen - patlen, pattern, patlen)) | |
120 | continue; | |
121 | if (namelen != patlen && name[namelen - patlen - 1] != '/') | |
122 | continue; | |
123 | match++; | |
124 | *matched_ref = refs; | |
125 | } | |
126 | return match; | |
127 | } | |
128 | ||
129 | static void link_dst_tail(struct ref *ref, struct ref ***tail) | |
130 | { | |
131 | **tail = ref; | |
132 | *tail = &ref->next; | |
133 | **tail = NULL; | |
134 | } | |
135 | ||
136 | static int match_explicit_refs(struct ref *src, struct ref *dst, | |
137 | struct ref ***dst_tail, struct refspec *rs) | |
138 | { | |
139 | int i, errs; | |
140 | for (i = errs = 0; rs[i].src; i++) { | |
141 | struct ref *matched_src, *matched_dst; | |
142 | ||
143 | matched_src = matched_dst = NULL; | |
144 | switch (count_refspec_match(rs[i].src, src, &matched_src)) { | |
145 | case 1: | |
146 | break; | |
147 | case 0: | |
148 | errs = 1; | |
149 | error("src refspec %s does not match any."); | |
150 | break; | |
151 | default: | |
152 | errs = 1; | |
153 | error("src refspec %s matches more than one.", | |
154 | rs[i].src); | |
155 | break; | |
156 | } | |
157 | switch (count_refspec_match(rs[i].dst, dst, &matched_dst)) { | |
158 | case 1: | |
159 | break; | |
160 | case 0: | |
161 | if (!memcmp(rs[i].dst, "refs/", 5)) { | |
162 | int len = strlen(rs[i].dst) + 1; | |
163 | matched_dst = xcalloc(1, sizeof(*dst) + len); | |
164 | memcpy(matched_dst->name, rs[i].dst, len); | |
165 | link_dst_tail(matched_dst, dst_tail); | |
166 | } | |
167 | else if (!strcmp(rs[i].src, rs[i].dst) && | |
168 | matched_src) { | |
169 | /* pushing "master:master" when | |
170 | * remote does not have master yet. | |
171 | */ | |
172 | int len = strlen(matched_src->name); | |
173 | matched_dst = xcalloc(1, sizeof(*dst) + len); | |
174 | memcpy(matched_dst->name, matched_src->name, | |
175 | len); | |
176 | link_dst_tail(matched_dst, dst_tail); | |
177 | } | |
178 | else { | |
179 | errs = 1; | |
180 | error("dst refspec %s does not match any " | |
181 | "existing ref on the remote and does " | |
182 | "not start with refs/.", rs[i].dst); | |
183 | } | |
184 | break; | |
185 | default: | |
186 | errs = 1; | |
187 | error("dst refspec %s matches more than one.", | |
188 | rs[i].dst); | |
189 | break; | |
190 | } | |
191 | if (errs) | |
192 | continue; | |
193 | if (matched_src->peer_ref) { | |
194 | errs = 1; | |
195 | error("src ref %s is sent to more than one dst.", | |
196 | matched_src->name); | |
197 | } | |
198 | else | |
199 | matched_src->peer_ref = matched_dst; | |
200 | if (matched_dst->peer_ref) { | |
201 | errs = 1; | |
202 | error("dst ref %s receives from more than one src.", | |
203 | matched_dst->name); | |
204 | } | |
205 | else | |
206 | matched_dst->peer_ref = matched_src; | |
207 | } | |
208 | return -errs; | |
209 | } | |
210 | ||
211 | static struct ref *find_ref_by_name(struct ref *list, const char *name) | |
212 | { | |
213 | for ( ; list; list = list->next) | |
214 | if (!strcmp(list->name, name)) | |
215 | return list; | |
216 | return NULL; | |
217 | } | |
218 | ||
219 | int match_refs(struct ref *src, struct ref *dst, struct ref ***dst_tail, | |
220 | int nr_refspec, char **refspec, int all) | |
221 | { | |
222 | struct refspec *rs = parse_ref_spec(nr_refspec, refspec); | |
223 | ||
224 | if (nr_refspec) | |
225 | return match_explicit_refs(src, dst, dst_tail, rs); | |
226 | ||
227 | /* pick the remainder */ | |
228 | for ( ; src; src = src->next) { | |
229 | struct ref *dst_peer; | |
230 | if (src->peer_ref) | |
231 | continue; | |
232 | dst_peer = find_ref_by_name(dst, src->name); | |
233 | if (dst_peer && dst_peer->peer_ref) | |
234 | continue; | |
235 | if (!dst_peer) { | |
236 | if (!all) | |
237 | continue; | |
238 | /* Create a new one and link it */ | |
239 | int len = strlen(src->name) + 1; | |
240 | dst_peer = xcalloc(1, sizeof(*dst_peer) + len); | |
241 | memcpy(dst_peer->name, src->name, len); | |
242 | memcpy(dst_peer->new_sha1, src->new_sha1, 20); | |
243 | link_dst_tail(dst_peer, dst_tail); | |
244 | } | |
245 | dst_peer->peer_ref = src; | |
246 | } | |
247 | return 0; | |
248 | } | |
249 | ||
2386d658 LT |
250 | enum protocol { |
251 | PROTO_LOCAL = 1, | |
252 | PROTO_SSH, | |
253 | PROTO_GIT, | |
254 | }; | |
255 | ||
256 | static enum protocol get_protocol(const char *name) | |
257 | { | |
258 | if (!strcmp(name, "ssh")) | |
259 | return PROTO_SSH; | |
260 | if (!strcmp(name, "git")) | |
261 | return PROTO_GIT; | |
262 | die("I don't handle protocol '%s'", name); | |
263 | } | |
264 | ||
5ba88448 YH |
265 | #define STR_(s) # s |
266 | #define STR(s) STR_(s) | |
2386d658 LT |
267 | |
268 | static int git_tcp_connect(int fd[2], const char *prog, char *host, char *path) | |
269 | { | |
5ba88448 YH |
270 | int sockfd = -1; |
271 | char *colon, *end; | |
272 | char *port = STR(DEFAULT_GIT_PORT); | |
273 | struct addrinfo hints, *ai0, *ai; | |
274 | int gai; | |
275 | ||
276 | if (host[0] == '[') { | |
277 | end = strchr(host + 1, ']'); | |
278 | if (end) { | |
279 | *end = 0; | |
280 | end++; | |
281 | host++; | |
282 | } else | |
283 | end = host; | |
284 | } else | |
285 | end = host; | |
286 | colon = strchr(end, ':'); | |
287 | ||
ce6f8e7e LT |
288 | if (colon) { |
289 | *colon = 0; | |
5ba88448 | 290 | port = colon + 1; |
ce6f8e7e | 291 | } |
5ba88448 YH |
292 | |
293 | memset(&hints, 0, sizeof(hints)); | |
294 | hints.ai_socktype = SOCK_STREAM; | |
295 | hints.ai_protocol = IPPROTO_TCP; | |
296 | ||
297 | gai = getaddrinfo(host, port, &hints, &ai); | |
298 | if (gai) | |
299 | die("Unable to look up %s (%s)", host, gai_strerror(gai)); | |
300 | ||
301 | for (ai0 = ai; ai; ai = ai->ai_next) { | |
302 | sockfd = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); | |
303 | if (sockfd < 0) | |
304 | continue; | |
305 | if (connect(sockfd, ai->ai_addr, ai->ai_addrlen) < 0) { | |
306 | close(sockfd); | |
307 | sockfd = -1; | |
308 | continue; | |
2386d658 | 309 | } |
5ba88448 | 310 | break; |
2386d658 LT |
311 | } |
312 | ||
5ba88448 | 313 | freeaddrinfo(ai0); |
2386d658 | 314 | |
2386d658 | 315 | if (sockfd < 0) |
5ba88448 YH |
316 | die("unable to connect a socket (%s)", strerror(errno)); |
317 | ||
2386d658 LT |
318 | fd[0] = sockfd; |
319 | fd[1] = sockfd; | |
320 | packet_write(sockfd, "%s %s\n", prog, path); | |
321 | return 0; | |
322 | } | |
323 | ||
f7192598 LT |
324 | /* |
325 | * Yeah, yeah, fixme. Need to pass in the heads etc. | |
326 | */ | |
327 | int git_connect(int fd[2], char *url, const char *prog) | |
328 | { | |
329 | char command[1024]; | |
2386d658 | 330 | char *host, *path; |
f7192598 LT |
331 | char *colon; |
332 | int pipefd[2][2]; | |
333 | pid_t pid; | |
2386d658 | 334 | enum protocol protocol; |
f7192598 | 335 | |
f7192598 LT |
336 | host = NULL; |
337 | path = url; | |
338 | colon = strchr(url, ':'); | |
2386d658 | 339 | protocol = PROTO_LOCAL; |
f7192598 LT |
340 | if (colon) { |
341 | *colon = 0; | |
342 | host = url; | |
343 | path = colon+1; | |
2386d658 LT |
344 | protocol = PROTO_SSH; |
345 | if (!memcmp(path, "//", 2)) { | |
346 | char *slash = strchr(path + 2, '/'); | |
347 | if (slash) { | |
348 | int nr = slash - path - 2; | |
349 | memmove(path, path+2, nr); | |
350 | path[nr] = 0; | |
351 | protocol = get_protocol(url); | |
352 | host = path; | |
353 | path = slash; | |
354 | } | |
355 | } | |
f7192598 | 356 | } |
2386d658 LT |
357 | |
358 | if (protocol == PROTO_GIT) | |
359 | return git_tcp_connect(fd, prog, host, path); | |
360 | ||
f7192598 LT |
361 | if (pipe(pipefd[0]) < 0 || pipe(pipefd[1]) < 0) |
362 | die("unable to create pipe pair for communication"); | |
363 | pid = fork(); | |
364 | if (!pid) { | |
b10d0ec7 JH |
365 | snprintf(command, sizeof(command), "%s %s", prog, |
366 | sq_quote(path)); | |
f7192598 LT |
367 | dup2(pipefd[1][0], 0); |
368 | dup2(pipefd[0][1], 1); | |
369 | close(pipefd[0][0]); | |
370 | close(pipefd[0][1]); | |
371 | close(pipefd[1][0]); | |
372 | close(pipefd[1][1]); | |
2386d658 | 373 | if (protocol == PROTO_SSH) |
f7192598 LT |
374 | execlp("ssh", "ssh", host, command, NULL); |
375 | else | |
376 | execlp("sh", "sh", "-c", command, NULL); | |
377 | die("exec failed"); | |
378 | } | |
379 | fd[0] = pipefd[0][0]; | |
380 | fd[1] = pipefd[1][1]; | |
381 | close(pipefd[0][1]); | |
382 | close(pipefd[1][0]); | |
383 | return pid; | |
384 | } | |
385 | ||
386 | int finish_connect(pid_t pid) | |
387 | { | |
388 | int ret; | |
389 | ||
390 | for (;;) { | |
391 | ret = waitpid(pid, NULL, 0); | |
392 | if (!ret) | |
393 | break; | |
394 | if (errno != EINTR) | |
395 | break; | |
396 | } | |
397 | return ret; | |
398 | } |