]>
Commit | Line | Data |
---|---|---|
1 | #include "cache.h" | |
2 | #include "commit.h" | |
3 | #include "pack.h" | |
4 | #include "tag.h" | |
5 | #include "blob.h" | |
6 | #include "http.h" | |
7 | #include "refs.h" | |
8 | #include "diff.h" | |
9 | #include "revision.h" | |
10 | #include "exec_cmd.h" | |
11 | #include "remote.h" | |
12 | #include "list-objects.h" | |
13 | #include "sigchain.h" | |
14 | ||
15 | #include <expat.h> | |
16 | ||
17 | static const char http_push_usage[] = | |
18 | "git http-push [--all] [--dry-run] [--force] [--verbose] <remote> [<head>...]\n"; | |
19 | ||
20 | #ifndef XML_STATUS_OK | |
21 | enum XML_Status { | |
22 | XML_STATUS_OK = 1, | |
23 | XML_STATUS_ERROR = 0 | |
24 | }; | |
25 | #define XML_STATUS_OK 1 | |
26 | #define XML_STATUS_ERROR 0 | |
27 | #endif | |
28 | ||
29 | #define PREV_BUF_SIZE 4096 | |
30 | #define RANGE_HEADER_SIZE 30 | |
31 | ||
32 | /* DAV methods */ | |
33 | #define DAV_LOCK "LOCK" | |
34 | #define DAV_MKCOL "MKCOL" | |
35 | #define DAV_MOVE "MOVE" | |
36 | #define DAV_PROPFIND "PROPFIND" | |
37 | #define DAV_PUT "PUT" | |
38 | #define DAV_UNLOCK "UNLOCK" | |
39 | #define DAV_DELETE "DELETE" | |
40 | ||
41 | /* DAV lock flags */ | |
42 | #define DAV_PROP_LOCKWR (1u << 0) | |
43 | #define DAV_PROP_LOCKEX (1u << 1) | |
44 | #define DAV_LOCK_OK (1u << 2) | |
45 | ||
46 | /* DAV XML properties */ | |
47 | #define DAV_CTX_LOCKENTRY ".multistatus.response.propstat.prop.supportedlock.lockentry" | |
48 | #define DAV_CTX_LOCKTYPE_WRITE ".multistatus.response.propstat.prop.supportedlock.lockentry.locktype.write" | |
49 | #define DAV_CTX_LOCKTYPE_EXCLUSIVE ".multistatus.response.propstat.prop.supportedlock.lockentry.lockscope.exclusive" | |
50 | #define DAV_ACTIVELOCK_OWNER ".prop.lockdiscovery.activelock.owner.href" | |
51 | #define DAV_ACTIVELOCK_TIMEOUT ".prop.lockdiscovery.activelock.timeout" | |
52 | #define DAV_ACTIVELOCK_TOKEN ".prop.lockdiscovery.activelock.locktoken.href" | |
53 | #define DAV_PROPFIND_RESP ".multistatus.response" | |
54 | #define DAV_PROPFIND_NAME ".multistatus.response.href" | |
55 | #define DAV_PROPFIND_COLLECTION ".multistatus.response.propstat.prop.resourcetype.collection" | |
56 | ||
57 | /* DAV request body templates */ | |
58 | #define PROPFIND_SUPPORTEDLOCK_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:propfind xmlns:D=\"DAV:\">\n<D:prop xmlns:R=\"%s\">\n<D:supportedlock/>\n</D:prop>\n</D:propfind>" | |
59 | #define PROPFIND_ALL_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:propfind xmlns:D=\"DAV:\">\n<D:allprop/>\n</D:propfind>" | |
60 | #define LOCK_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:lockinfo xmlns:D=\"DAV:\">\n<D:lockscope><D:exclusive/></D:lockscope>\n<D:locktype><D:write/></D:locktype>\n<D:owner>\n<D:href>mailto:%s</D:href>\n</D:owner>\n</D:lockinfo>" | |
61 | ||
62 | #define LOCK_TIME 600 | |
63 | #define LOCK_REFRESH 30 | |
64 | ||
65 | /* bits #0-15 in revision.h */ | |
66 | ||
67 | #define LOCAL (1u<<16) | |
68 | #define REMOTE (1u<<17) | |
69 | #define FETCHING (1u<<18) | |
70 | #define PUSHING (1u<<19) | |
71 | ||
72 | /* We allow "recursive" symbolic refs. Only within reason, though */ | |
73 | #define MAXDEPTH 5 | |
74 | ||
75 | static int pushing; | |
76 | static int aborted; | |
77 | static signed char remote_dir_exists[256]; | |
78 | ||
79 | static struct curl_slist *no_pragma_header; | |
80 | ||
81 | static int push_verbosely; | |
82 | static int push_all = MATCH_REFS_NONE; | |
83 | static int force_all; | |
84 | static int dry_run; | |
85 | ||
86 | static struct object_list *objects; | |
87 | ||
88 | struct repo | |
89 | { | |
90 | char *url; | |
91 | char *path; | |
92 | int path_len; | |
93 | int has_info_refs; | |
94 | int can_update_info_refs; | |
95 | int has_info_packs; | |
96 | struct packed_git *packs; | |
97 | struct remote_lock *locks; | |
98 | }; | |
99 | ||
100 | static struct repo *repo; | |
101 | ||
102 | enum transfer_state { | |
103 | NEED_FETCH, | |
104 | RUN_FETCH_LOOSE, | |
105 | RUN_FETCH_PACKED, | |
106 | NEED_PUSH, | |
107 | RUN_MKCOL, | |
108 | RUN_PUT, | |
109 | RUN_MOVE, | |
110 | ABORTED, | |
111 | COMPLETE, | |
112 | }; | |
113 | ||
114 | struct transfer_request | |
115 | { | |
116 | struct object *obj; | |
117 | char *url; | |
118 | char *dest; | |
119 | struct remote_lock *lock; | |
120 | struct curl_slist *headers; | |
121 | struct buffer buffer; | |
122 | char filename[PATH_MAX]; | |
123 | char tmpfile[PATH_MAX]; | |
124 | int local_fileno; | |
125 | FILE *local_stream; | |
126 | enum transfer_state state; | |
127 | CURLcode curl_result; | |
128 | char errorstr[CURL_ERROR_SIZE]; | |
129 | long http_code; | |
130 | unsigned char real_sha1[20]; | |
131 | git_SHA_CTX c; | |
132 | z_stream stream; | |
133 | int zret; | |
134 | int rename; | |
135 | void *userData; | |
136 | struct active_request_slot *slot; | |
137 | struct transfer_request *next; | |
138 | }; | |
139 | ||
140 | static struct transfer_request *request_queue_head; | |
141 | ||
142 | struct xml_ctx | |
143 | { | |
144 | char *name; | |
145 | int len; | |
146 | char *cdata; | |
147 | void (*userFunc)(struct xml_ctx *ctx, int tag_closed); | |
148 | void *userData; | |
149 | }; | |
150 | ||
151 | struct remote_lock | |
152 | { | |
153 | char *url; | |
154 | char *owner; | |
155 | char *token; | |
156 | char tmpfile_suffix[41]; | |
157 | time_t start_time; | |
158 | long timeout; | |
159 | int refreshing; | |
160 | struct remote_lock *next; | |
161 | }; | |
162 | ||
163 | /* Flags that control remote_ls processing */ | |
164 | #define PROCESS_FILES (1u << 0) | |
165 | #define PROCESS_DIRS (1u << 1) | |
166 | #define RECURSIVE (1u << 2) | |
167 | ||
168 | /* Flags that remote_ls passes to callback functions */ | |
169 | #define IS_DIR (1u << 0) | |
170 | ||
171 | struct remote_ls_ctx | |
172 | { | |
173 | char *path; | |
174 | void (*userFunc)(struct remote_ls_ctx *ls); | |
175 | void *userData; | |
176 | int flags; | |
177 | char *dentry_name; | |
178 | int dentry_flags; | |
179 | struct remote_ls_ctx *parent; | |
180 | }; | |
181 | ||
182 | /* get_dav_token_headers options */ | |
183 | enum dav_header_flag { | |
184 | DAV_HEADER_IF = (1u << 0), | |
185 | DAV_HEADER_LOCK = (1u << 1), | |
186 | DAV_HEADER_TIMEOUT = (1u << 2) | |
187 | }; | |
188 | ||
189 | static char *xml_entities(char *s) | |
190 | { | |
191 | struct strbuf buf = STRBUF_INIT; | |
192 | while (*s) { | |
193 | size_t len = strcspn(s, "\"<>&"); | |
194 | strbuf_add(&buf, s, len); | |
195 | s += len; | |
196 | switch (*s) { | |
197 | case '"': | |
198 | strbuf_addstr(&buf, """); | |
199 | break; | |
200 | case '<': | |
201 | strbuf_addstr(&buf, "<"); | |
202 | break; | |
203 | case '>': | |
204 | strbuf_addstr(&buf, ">"); | |
205 | break; | |
206 | case '&': | |
207 | strbuf_addstr(&buf, "&"); | |
208 | break; | |
209 | } | |
210 | s++; | |
211 | } | |
212 | return strbuf_detach(&buf, NULL); | |
213 | } | |
214 | ||
215 | static struct curl_slist *get_dav_token_headers(struct remote_lock *lock, enum dav_header_flag options) | |
216 | { | |
217 | struct strbuf buf = STRBUF_INIT; | |
218 | struct curl_slist *dav_headers = NULL; | |
219 | ||
220 | if (options & DAV_HEADER_IF) { | |
221 | strbuf_addf(&buf, "If: (<%s>)", lock->token); | |
222 | dav_headers = curl_slist_append(dav_headers, buf.buf); | |
223 | strbuf_reset(&buf); | |
224 | } | |
225 | if (options & DAV_HEADER_LOCK) { | |
226 | strbuf_addf(&buf, "Lock-Token: <%s>", lock->token); | |
227 | dav_headers = curl_slist_append(dav_headers, buf.buf); | |
228 | strbuf_reset(&buf); | |
229 | } | |
230 | if (options & DAV_HEADER_TIMEOUT) { | |
231 | strbuf_addf(&buf, "Timeout: Second-%ld", lock->timeout); | |
232 | dav_headers = curl_slist_append(dav_headers, buf.buf); | |
233 | strbuf_reset(&buf); | |
234 | } | |
235 | strbuf_release(&buf); | |
236 | ||
237 | return dav_headers; | |
238 | } | |
239 | ||
240 | static void append_remote_object_url(struct strbuf *buf, const char *url, | |
241 | const char *hex, | |
242 | int only_two_digit_prefix) | |
243 | { | |
244 | strbuf_addf(buf, "%sobjects/%.*s/", url, 2, hex); | |
245 | if (!only_two_digit_prefix) | |
246 | strbuf_addf(buf, "%s", hex+2); | |
247 | } | |
248 | ||
249 | static void finish_request(struct transfer_request *request); | |
250 | static void release_request(struct transfer_request *request); | |
251 | ||
252 | static void process_response(void *callback_data) | |
253 | { | |
254 | struct transfer_request *request = | |
255 | (struct transfer_request *)callback_data; | |
256 | ||
257 | finish_request(request); | |
258 | } | |
259 | ||
260 | #ifdef USE_CURL_MULTI | |
261 | ||
262 | static char *get_remote_object_url(const char *url, const char *hex, | |
263 | int only_two_digit_prefix) | |
264 | { | |
265 | struct strbuf buf = STRBUF_INIT; | |
266 | append_remote_object_url(&buf, url, hex, only_two_digit_prefix); | |
267 | return strbuf_detach(&buf, NULL); | |
268 | } | |
269 | ||
270 | static size_t fwrite_sha1_file(void *ptr, size_t eltsize, size_t nmemb, | |
271 | void *data) | |
272 | { | |
273 | unsigned char expn[4096]; | |
274 | size_t size = eltsize * nmemb; | |
275 | int posn = 0; | |
276 | struct transfer_request *request = (struct transfer_request *)data; | |
277 | do { | |
278 | ssize_t retval = xwrite(request->local_fileno, | |
279 | (char *) ptr + posn, size - posn); | |
280 | if (retval < 0) | |
281 | return posn; | |
282 | posn += retval; | |
283 | } while (posn < size); | |
284 | ||
285 | request->stream.avail_in = size; | |
286 | request->stream.next_in = ptr; | |
287 | do { | |
288 | request->stream.next_out = expn; | |
289 | request->stream.avail_out = sizeof(expn); | |
290 | request->zret = git_inflate(&request->stream, Z_SYNC_FLUSH); | |
291 | git_SHA1_Update(&request->c, expn, | |
292 | sizeof(expn) - request->stream.avail_out); | |
293 | } while (request->stream.avail_in && request->zret == Z_OK); | |
294 | data_received++; | |
295 | return size; | |
296 | } | |
297 | ||
298 | static void start_fetch_loose(struct transfer_request *request) | |
299 | { | |
300 | char *hex = sha1_to_hex(request->obj->sha1); | |
301 | char *filename; | |
302 | char prevfile[PATH_MAX]; | |
303 | char *url; | |
304 | int prevlocal; | |
305 | unsigned char prev_buf[PREV_BUF_SIZE]; | |
306 | ssize_t prev_read = 0; | |
307 | long prev_posn = 0; | |
308 | char range[RANGE_HEADER_SIZE]; | |
309 | struct curl_slist *range_header = NULL; | |
310 | struct active_request_slot *slot; | |
311 | ||
312 | filename = sha1_file_name(request->obj->sha1); | |
313 | snprintf(request->filename, sizeof(request->filename), "%s", filename); | |
314 | snprintf(request->tmpfile, sizeof(request->tmpfile), | |
315 | "%s.temp", filename); | |
316 | ||
317 | snprintf(prevfile, sizeof(prevfile), "%s.prev", request->filename); | |
318 | unlink_or_warn(prevfile); | |
319 | rename(request->tmpfile, prevfile); | |
320 | unlink_or_warn(request->tmpfile); | |
321 | ||
322 | if (request->local_fileno != -1) | |
323 | error("fd leakage in start: %d", request->local_fileno); | |
324 | request->local_fileno = open(request->tmpfile, | |
325 | O_WRONLY | O_CREAT | O_EXCL, 0666); | |
326 | /* This could have failed due to the "lazy directory creation"; | |
327 | * try to mkdir the last path component. | |
328 | */ | |
329 | if (request->local_fileno < 0 && errno == ENOENT) { | |
330 | char *dir = strrchr(request->tmpfile, '/'); | |
331 | if (dir) { | |
332 | *dir = 0; | |
333 | mkdir(request->tmpfile, 0777); | |
334 | *dir = '/'; | |
335 | } | |
336 | request->local_fileno = open(request->tmpfile, | |
337 | O_WRONLY | O_CREAT | O_EXCL, 0666); | |
338 | } | |
339 | ||
340 | if (request->local_fileno < 0) { | |
341 | request->state = ABORTED; | |
342 | error("Couldn't create temporary file %s for %s: %s", | |
343 | request->tmpfile, request->filename, strerror(errno)); | |
344 | return; | |
345 | } | |
346 | ||
347 | memset(&request->stream, 0, sizeof(request->stream)); | |
348 | ||
349 | git_inflate_init(&request->stream); | |
350 | ||
351 | git_SHA1_Init(&request->c); | |
352 | ||
353 | url = get_remote_object_url(repo->url, hex, 0); | |
354 | request->url = xstrdup(url); | |
355 | ||
356 | /* If a previous temp file is present, process what was already | |
357 | fetched. */ | |
358 | prevlocal = open(prevfile, O_RDONLY); | |
359 | if (prevlocal != -1) { | |
360 | do { | |
361 | prev_read = xread(prevlocal, prev_buf, PREV_BUF_SIZE); | |
362 | if (prev_read>0) { | |
363 | if (fwrite_sha1_file(prev_buf, | |
364 | 1, | |
365 | prev_read, | |
366 | request) == prev_read) { | |
367 | prev_posn += prev_read; | |
368 | } else { | |
369 | prev_read = -1; | |
370 | } | |
371 | } | |
372 | } while (prev_read > 0); | |
373 | close(prevlocal); | |
374 | } | |
375 | unlink_or_warn(prevfile); | |
376 | ||
377 | /* Reset inflate/SHA1 if there was an error reading the previous temp | |
378 | file; also rewind to the beginning of the local file. */ | |
379 | if (prev_read == -1) { | |
380 | memset(&request->stream, 0, sizeof(request->stream)); | |
381 | git_inflate_init(&request->stream); | |
382 | git_SHA1_Init(&request->c); | |
383 | if (prev_posn>0) { | |
384 | prev_posn = 0; | |
385 | lseek(request->local_fileno, 0, SEEK_SET); | |
386 | ftruncate(request->local_fileno, 0); | |
387 | } | |
388 | } | |
389 | ||
390 | slot = get_active_slot(); | |
391 | slot->callback_func = process_response; | |
392 | slot->callback_data = request; | |
393 | request->slot = slot; | |
394 | ||
395 | curl_easy_setopt(slot->curl, CURLOPT_FILE, request); | |
396 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_sha1_file); | |
397 | curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, request->errorstr); | |
398 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
399 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header); | |
400 | ||
401 | /* If we have successfully processed data from a previous fetch | |
402 | attempt, only fetch the data we don't already have. */ | |
403 | if (prev_posn>0) { | |
404 | if (push_verbosely) | |
405 | fprintf(stderr, | |
406 | "Resuming fetch of object %s at byte %ld\n", | |
407 | hex, prev_posn); | |
408 | sprintf(range, "Range: bytes=%ld-", prev_posn); | |
409 | range_header = curl_slist_append(range_header, range); | |
410 | curl_easy_setopt(slot->curl, | |
411 | CURLOPT_HTTPHEADER, range_header); | |
412 | } | |
413 | ||
414 | /* Try to get the request started, abort the request on error */ | |
415 | request->state = RUN_FETCH_LOOSE; | |
416 | if (!start_active_slot(slot)) { | |
417 | fprintf(stderr, "Unable to start GET request\n"); | |
418 | repo->can_update_info_refs = 0; | |
419 | release_request(request); | |
420 | } | |
421 | } | |
422 | ||
423 | static void start_mkcol(struct transfer_request *request) | |
424 | { | |
425 | char *hex = sha1_to_hex(request->obj->sha1); | |
426 | struct active_request_slot *slot; | |
427 | ||
428 | request->url = get_remote_object_url(repo->url, hex, 1); | |
429 | ||
430 | slot = get_active_slot(); | |
431 | slot->callback_func = process_response; | |
432 | slot->callback_data = request; | |
433 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); /* undo PUT setup */ | |
434 | curl_easy_setopt(slot->curl, CURLOPT_URL, request->url); | |
435 | curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, request->errorstr); | |
436 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_MKCOL); | |
437 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
438 | ||
439 | if (start_active_slot(slot)) { | |
440 | request->slot = slot; | |
441 | request->state = RUN_MKCOL; | |
442 | } else { | |
443 | request->state = ABORTED; | |
444 | free(request->url); | |
445 | request->url = NULL; | |
446 | } | |
447 | } | |
448 | #endif | |
449 | ||
450 | static void start_fetch_packed(struct transfer_request *request) | |
451 | { | |
452 | char *url; | |
453 | struct packed_git *target; | |
454 | FILE *packfile; | |
455 | char *filename; | |
456 | long prev_posn = 0; | |
457 | char range[RANGE_HEADER_SIZE]; | |
458 | struct curl_slist *range_header = NULL; | |
459 | ||
460 | struct transfer_request *check_request = request_queue_head; | |
461 | struct active_request_slot *slot; | |
462 | ||
463 | target = find_sha1_pack(request->obj->sha1, repo->packs); | |
464 | if (!target) { | |
465 | fprintf(stderr, "Unable to fetch %s, will not be able to update server info refs\n", sha1_to_hex(request->obj->sha1)); | |
466 | repo->can_update_info_refs = 0; | |
467 | release_request(request); | |
468 | return; | |
469 | } | |
470 | ||
471 | fprintf(stderr, "Fetching pack %s\n", sha1_to_hex(target->sha1)); | |
472 | fprintf(stderr, " which contains %s\n", sha1_to_hex(request->obj->sha1)); | |
473 | ||
474 | filename = sha1_pack_name(target->sha1); | |
475 | snprintf(request->filename, sizeof(request->filename), "%s", filename); | |
476 | snprintf(request->tmpfile, sizeof(request->tmpfile), | |
477 | "%s.temp", filename); | |
478 | ||
479 | url = xmalloc(strlen(repo->url) + 64); | |
480 | sprintf(url, "%sobjects/pack/pack-%s.pack", | |
481 | repo->url, sha1_to_hex(target->sha1)); | |
482 | ||
483 | /* Make sure there isn't another open request for this pack */ | |
484 | while (check_request) { | |
485 | if (check_request->state == RUN_FETCH_PACKED && | |
486 | !strcmp(check_request->url, url)) { | |
487 | free(url); | |
488 | release_request(request); | |
489 | return; | |
490 | } | |
491 | check_request = check_request->next; | |
492 | } | |
493 | ||
494 | packfile = fopen(request->tmpfile, "a"); | |
495 | if (!packfile) { | |
496 | fprintf(stderr, "Unable to open local file %s for pack", | |
497 | request->tmpfile); | |
498 | repo->can_update_info_refs = 0; | |
499 | free(url); | |
500 | return; | |
501 | } | |
502 | ||
503 | slot = get_active_slot(); | |
504 | slot->callback_func = process_response; | |
505 | slot->callback_data = request; | |
506 | request->slot = slot; | |
507 | request->local_stream = packfile; | |
508 | request->userData = target; | |
509 | ||
510 | request->url = url; | |
511 | curl_easy_setopt(slot->curl, CURLOPT_FILE, packfile); | |
512 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite); | |
513 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
514 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header); | |
515 | slot->local = packfile; | |
516 | ||
517 | /* If there is data present from a previous transfer attempt, | |
518 | resume where it left off */ | |
519 | prev_posn = ftell(packfile); | |
520 | if (prev_posn>0) { | |
521 | if (push_verbosely) | |
522 | fprintf(stderr, | |
523 | "Resuming fetch of pack %s at byte %ld\n", | |
524 | sha1_to_hex(target->sha1), prev_posn); | |
525 | sprintf(range, "Range: bytes=%ld-", prev_posn); | |
526 | range_header = curl_slist_append(range_header, range); | |
527 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, range_header); | |
528 | } | |
529 | ||
530 | /* Try to get the request started, abort the request on error */ | |
531 | request->state = RUN_FETCH_PACKED; | |
532 | if (!start_active_slot(slot)) { | |
533 | fprintf(stderr, "Unable to start GET request\n"); | |
534 | repo->can_update_info_refs = 0; | |
535 | release_request(request); | |
536 | } | |
537 | } | |
538 | ||
539 | static void start_put(struct transfer_request *request) | |
540 | { | |
541 | char *hex = sha1_to_hex(request->obj->sha1); | |
542 | struct active_request_slot *slot; | |
543 | struct strbuf buf = STRBUF_INIT; | |
544 | enum object_type type; | |
545 | char hdr[50]; | |
546 | void *unpacked; | |
547 | unsigned long len; | |
548 | int hdrlen; | |
549 | ssize_t size; | |
550 | z_stream stream; | |
551 | ||
552 | unpacked = read_sha1_file(request->obj->sha1, &type, &len); | |
553 | hdrlen = sprintf(hdr, "%s %lu", typename(type), len) + 1; | |
554 | ||
555 | /* Set it up */ | |
556 | memset(&stream, 0, sizeof(stream)); | |
557 | deflateInit(&stream, zlib_compression_level); | |
558 | size = deflateBound(&stream, len + hdrlen); | |
559 | strbuf_init(&request->buffer.buf, size); | |
560 | request->buffer.posn = 0; | |
561 | ||
562 | /* Compress it */ | |
563 | stream.next_out = (unsigned char *)request->buffer.buf.buf; | |
564 | stream.avail_out = size; | |
565 | ||
566 | /* First header.. */ | |
567 | stream.next_in = (void *)hdr; | |
568 | stream.avail_in = hdrlen; | |
569 | while (deflate(&stream, 0) == Z_OK) | |
570 | /* nothing */; | |
571 | ||
572 | /* Then the data itself.. */ | |
573 | stream.next_in = unpacked; | |
574 | stream.avail_in = len; | |
575 | while (deflate(&stream, Z_FINISH) == Z_OK) | |
576 | /* nothing */; | |
577 | deflateEnd(&stream); | |
578 | free(unpacked); | |
579 | ||
580 | request->buffer.buf.len = stream.total_out; | |
581 | ||
582 | strbuf_addstr(&buf, "Destination: "); | |
583 | append_remote_object_url(&buf, repo->url, hex, 0); | |
584 | request->dest = strbuf_detach(&buf, NULL); | |
585 | ||
586 | append_remote_object_url(&buf, repo->url, hex, 0); | |
587 | strbuf_add(&buf, request->lock->tmpfile_suffix, 41); | |
588 | request->url = strbuf_detach(&buf, NULL); | |
589 | ||
590 | slot = get_active_slot(); | |
591 | slot->callback_func = process_response; | |
592 | slot->callback_data = request; | |
593 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &request->buffer); | |
594 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, request->buffer.buf.len); | |
595 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
596 | #ifndef NO_CURL_IOCTL | |
597 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
598 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &request->buffer); | |
599 | #endif | |
600 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
601 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PUT); | |
602 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
603 | curl_easy_setopt(slot->curl, CURLOPT_PUT, 1); | |
604 | curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0); | |
605 | curl_easy_setopt(slot->curl, CURLOPT_URL, request->url); | |
606 | ||
607 | if (start_active_slot(slot)) { | |
608 | request->slot = slot; | |
609 | request->state = RUN_PUT; | |
610 | } else { | |
611 | request->state = ABORTED; | |
612 | free(request->url); | |
613 | request->url = NULL; | |
614 | } | |
615 | } | |
616 | ||
617 | static void start_move(struct transfer_request *request) | |
618 | { | |
619 | struct active_request_slot *slot; | |
620 | struct curl_slist *dav_headers = NULL; | |
621 | ||
622 | slot = get_active_slot(); | |
623 | slot->callback_func = process_response; | |
624 | slot->callback_data = request; | |
625 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); /* undo PUT setup */ | |
626 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_MOVE); | |
627 | dav_headers = curl_slist_append(dav_headers, request->dest); | |
628 | dav_headers = curl_slist_append(dav_headers, "Overwrite: T"); | |
629 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
630 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
631 | curl_easy_setopt(slot->curl, CURLOPT_URL, request->url); | |
632 | ||
633 | if (start_active_slot(slot)) { | |
634 | request->slot = slot; | |
635 | request->state = RUN_MOVE; | |
636 | } else { | |
637 | request->state = ABORTED; | |
638 | free(request->url); | |
639 | request->url = NULL; | |
640 | } | |
641 | } | |
642 | ||
643 | static int refresh_lock(struct remote_lock *lock) | |
644 | { | |
645 | struct active_request_slot *slot; | |
646 | struct slot_results results; | |
647 | struct curl_slist *dav_headers; | |
648 | int rc = 0; | |
649 | ||
650 | lock->refreshing = 1; | |
651 | ||
652 | dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF | DAV_HEADER_TIMEOUT); | |
653 | ||
654 | slot = get_active_slot(); | |
655 | slot->results = &results; | |
656 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); | |
657 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
658 | curl_easy_setopt(slot->curl, CURLOPT_URL, lock->url); | |
659 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_LOCK); | |
660 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
661 | ||
662 | if (start_active_slot(slot)) { | |
663 | run_active_slot(slot); | |
664 | if (results.curl_result != CURLE_OK) { | |
665 | fprintf(stderr, "LOCK HTTP error %ld\n", | |
666 | results.http_code); | |
667 | } else { | |
668 | lock->start_time = time(NULL); | |
669 | rc = 1; | |
670 | } | |
671 | } | |
672 | ||
673 | lock->refreshing = 0; | |
674 | curl_slist_free_all(dav_headers); | |
675 | ||
676 | return rc; | |
677 | } | |
678 | ||
679 | static void check_locks(void) | |
680 | { | |
681 | struct remote_lock *lock = repo->locks; | |
682 | time_t current_time = time(NULL); | |
683 | int time_remaining; | |
684 | ||
685 | while (lock) { | |
686 | time_remaining = lock->start_time + lock->timeout - | |
687 | current_time; | |
688 | if (!lock->refreshing && time_remaining < LOCK_REFRESH) { | |
689 | if (!refresh_lock(lock)) { | |
690 | fprintf(stderr, | |
691 | "Unable to refresh lock for %s\n", | |
692 | lock->url); | |
693 | aborted = 1; | |
694 | return; | |
695 | } | |
696 | } | |
697 | lock = lock->next; | |
698 | } | |
699 | } | |
700 | ||
701 | static void release_request(struct transfer_request *request) | |
702 | { | |
703 | struct transfer_request *entry = request_queue_head; | |
704 | ||
705 | if (request == request_queue_head) { | |
706 | request_queue_head = request->next; | |
707 | } else { | |
708 | while (entry->next != NULL && entry->next != request) | |
709 | entry = entry->next; | |
710 | if (entry->next == request) | |
711 | entry->next = entry->next->next; | |
712 | } | |
713 | ||
714 | if (request->local_fileno != -1) | |
715 | close(request->local_fileno); | |
716 | if (request->local_stream) | |
717 | fclose(request->local_stream); | |
718 | free(request->url); | |
719 | free(request); | |
720 | } | |
721 | ||
722 | static void finish_request(struct transfer_request *request) | |
723 | { | |
724 | struct stat st; | |
725 | struct packed_git *target; | |
726 | struct packed_git **lst; | |
727 | ||
728 | request->curl_result = request->slot->curl_result; | |
729 | request->http_code = request->slot->http_code; | |
730 | request->slot = NULL; | |
731 | ||
732 | /* Keep locks active */ | |
733 | check_locks(); | |
734 | ||
735 | if (request->headers != NULL) | |
736 | curl_slist_free_all(request->headers); | |
737 | ||
738 | /* URL is reused for MOVE after PUT */ | |
739 | if (request->state != RUN_PUT) { | |
740 | free(request->url); | |
741 | request->url = NULL; | |
742 | } | |
743 | ||
744 | if (request->state == RUN_MKCOL) { | |
745 | if (request->curl_result == CURLE_OK || | |
746 | request->http_code == 405) { | |
747 | remote_dir_exists[request->obj->sha1[0]] = 1; | |
748 | start_put(request); | |
749 | } else { | |
750 | fprintf(stderr, "MKCOL %s failed, aborting (%d/%ld)\n", | |
751 | sha1_to_hex(request->obj->sha1), | |
752 | request->curl_result, request->http_code); | |
753 | request->state = ABORTED; | |
754 | aborted = 1; | |
755 | } | |
756 | } else if (request->state == RUN_PUT) { | |
757 | if (request->curl_result == CURLE_OK) { | |
758 | start_move(request); | |
759 | } else { | |
760 | fprintf(stderr, "PUT %s failed, aborting (%d/%ld)\n", | |
761 | sha1_to_hex(request->obj->sha1), | |
762 | request->curl_result, request->http_code); | |
763 | request->state = ABORTED; | |
764 | aborted = 1; | |
765 | } | |
766 | } else if (request->state == RUN_MOVE) { | |
767 | if (request->curl_result == CURLE_OK) { | |
768 | if (push_verbosely) | |
769 | fprintf(stderr, " sent %s\n", | |
770 | sha1_to_hex(request->obj->sha1)); | |
771 | request->obj->flags |= REMOTE; | |
772 | release_request(request); | |
773 | } else { | |
774 | fprintf(stderr, "MOVE %s failed, aborting (%d/%ld)\n", | |
775 | sha1_to_hex(request->obj->sha1), | |
776 | request->curl_result, request->http_code); | |
777 | request->state = ABORTED; | |
778 | aborted = 1; | |
779 | } | |
780 | } else if (request->state == RUN_FETCH_LOOSE) { | |
781 | close(request->local_fileno); request->local_fileno = -1; | |
782 | ||
783 | if (request->curl_result != CURLE_OK && | |
784 | request->http_code != 416) { | |
785 | if (stat(request->tmpfile, &st) == 0) { | |
786 | if (st.st_size == 0) | |
787 | unlink_or_warn(request->tmpfile); | |
788 | } | |
789 | } else { | |
790 | if (request->http_code == 416) | |
791 | warning("requested range invalid; we may already have all the data."); | |
792 | ||
793 | git_inflate_end(&request->stream); | |
794 | git_SHA1_Final(request->real_sha1, &request->c); | |
795 | if (request->zret != Z_STREAM_END) { | |
796 | unlink_or_warn(request->tmpfile); | |
797 | } else if (hashcmp(request->obj->sha1, request->real_sha1)) { | |
798 | unlink_or_warn(request->tmpfile); | |
799 | } else { | |
800 | request->rename = | |
801 | move_temp_to_file( | |
802 | request->tmpfile, | |
803 | request->filename); | |
804 | if (request->rename == 0) { | |
805 | request->obj->flags |= (LOCAL | REMOTE); | |
806 | } | |
807 | } | |
808 | } | |
809 | ||
810 | /* Try fetching packed if necessary */ | |
811 | if (request->obj->flags & LOCAL) | |
812 | release_request(request); | |
813 | else | |
814 | start_fetch_packed(request); | |
815 | ||
816 | } else if (request->state == RUN_FETCH_PACKED) { | |
817 | if (request->curl_result != CURLE_OK) { | |
818 | fprintf(stderr, "Unable to get pack file %s\n%s", | |
819 | request->url, curl_errorstr); | |
820 | repo->can_update_info_refs = 0; | |
821 | } else { | |
822 | off_t pack_size = ftell(request->local_stream); | |
823 | ||
824 | fclose(request->local_stream); | |
825 | request->local_stream = NULL; | |
826 | if (!move_temp_to_file(request->tmpfile, | |
827 | request->filename)) { | |
828 | target = (struct packed_git *)request->userData; | |
829 | target->pack_size = pack_size; | |
830 | lst = &repo->packs; | |
831 | while (*lst != target) | |
832 | lst = &((*lst)->next); | |
833 | *lst = (*lst)->next; | |
834 | ||
835 | if (!verify_pack(target)) | |
836 | install_packed_git(target); | |
837 | else | |
838 | repo->can_update_info_refs = 0; | |
839 | } | |
840 | } | |
841 | release_request(request); | |
842 | } | |
843 | } | |
844 | ||
845 | #ifdef USE_CURL_MULTI | |
846 | static int fill_active_slot(void *unused) | |
847 | { | |
848 | struct transfer_request *request; | |
849 | ||
850 | if (aborted) | |
851 | return 0; | |
852 | ||
853 | for (request = request_queue_head; request; request = request->next) { | |
854 | if (request->state == NEED_FETCH) { | |
855 | start_fetch_loose(request); | |
856 | return 1; | |
857 | } else if (pushing && request->state == NEED_PUSH) { | |
858 | if (remote_dir_exists[request->obj->sha1[0]] == 1) { | |
859 | start_put(request); | |
860 | } else { | |
861 | start_mkcol(request); | |
862 | } | |
863 | return 1; | |
864 | } | |
865 | } | |
866 | return 0; | |
867 | } | |
868 | #endif | |
869 | ||
870 | static void get_remote_object_list(unsigned char parent); | |
871 | ||
872 | static void add_fetch_request(struct object *obj) | |
873 | { | |
874 | struct transfer_request *request; | |
875 | ||
876 | check_locks(); | |
877 | ||
878 | /* | |
879 | * Don't fetch the object if it's known to exist locally | |
880 | * or is already in the request queue | |
881 | */ | |
882 | if (remote_dir_exists[obj->sha1[0]] == -1) | |
883 | get_remote_object_list(obj->sha1[0]); | |
884 | if (obj->flags & (LOCAL | FETCHING)) | |
885 | return; | |
886 | ||
887 | obj->flags |= FETCHING; | |
888 | request = xmalloc(sizeof(*request)); | |
889 | request->obj = obj; | |
890 | request->url = NULL; | |
891 | request->lock = NULL; | |
892 | request->headers = NULL; | |
893 | request->local_fileno = -1; | |
894 | request->local_stream = NULL; | |
895 | request->state = NEED_FETCH; | |
896 | request->next = request_queue_head; | |
897 | request_queue_head = request; | |
898 | ||
899 | #ifdef USE_CURL_MULTI | |
900 | fill_active_slots(); | |
901 | step_active_slots(); | |
902 | #endif | |
903 | } | |
904 | ||
905 | static int add_send_request(struct object *obj, struct remote_lock *lock) | |
906 | { | |
907 | struct transfer_request *request = request_queue_head; | |
908 | struct packed_git *target; | |
909 | ||
910 | /* Keep locks active */ | |
911 | check_locks(); | |
912 | ||
913 | /* | |
914 | * Don't push the object if it's known to exist on the remote | |
915 | * or is already in the request queue | |
916 | */ | |
917 | if (remote_dir_exists[obj->sha1[0]] == -1) | |
918 | get_remote_object_list(obj->sha1[0]); | |
919 | if (obj->flags & (REMOTE | PUSHING)) | |
920 | return 0; | |
921 | target = find_sha1_pack(obj->sha1, repo->packs); | |
922 | if (target) { | |
923 | obj->flags |= REMOTE; | |
924 | return 0; | |
925 | } | |
926 | ||
927 | obj->flags |= PUSHING; | |
928 | request = xmalloc(sizeof(*request)); | |
929 | request->obj = obj; | |
930 | request->url = NULL; | |
931 | request->lock = lock; | |
932 | request->headers = NULL; | |
933 | request->local_fileno = -1; | |
934 | request->local_stream = NULL; | |
935 | request->state = NEED_PUSH; | |
936 | request->next = request_queue_head; | |
937 | request_queue_head = request; | |
938 | ||
939 | #ifdef USE_CURL_MULTI | |
940 | fill_active_slots(); | |
941 | step_active_slots(); | |
942 | #endif | |
943 | ||
944 | return 1; | |
945 | } | |
946 | ||
947 | static int fetch_index(unsigned char *sha1) | |
948 | { | |
949 | char *hex = sha1_to_hex(sha1); | |
950 | char *filename; | |
951 | char *url; | |
952 | char tmpfile[PATH_MAX]; | |
953 | long prev_posn = 0; | |
954 | char range[RANGE_HEADER_SIZE]; | |
955 | struct curl_slist *range_header = NULL; | |
956 | ||
957 | FILE *indexfile; | |
958 | struct active_request_slot *slot; | |
959 | struct slot_results results; | |
960 | ||
961 | /* Don't use the index if the pack isn't there */ | |
962 | url = xmalloc(strlen(repo->url) + 64); | |
963 | sprintf(url, "%sobjects/pack/pack-%s.pack", repo->url, hex); | |
964 | slot = get_active_slot(); | |
965 | slot->results = &results; | |
966 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
967 | curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 1); | |
968 | if (start_active_slot(slot)) { | |
969 | run_active_slot(slot); | |
970 | if (results.curl_result != CURLE_OK) { | |
971 | free(url); | |
972 | return error("Unable to verify pack %s is available", | |
973 | hex); | |
974 | } | |
975 | } else { | |
976 | free(url); | |
977 | return error("Unable to start request"); | |
978 | } | |
979 | ||
980 | if (has_pack_index(sha1)) { | |
981 | free(url); | |
982 | return 0; | |
983 | } | |
984 | ||
985 | if (push_verbosely) | |
986 | fprintf(stderr, "Getting index for pack %s\n", hex); | |
987 | ||
988 | sprintf(url, "%sobjects/pack/pack-%s.idx", repo->url, hex); | |
989 | ||
990 | filename = sha1_pack_index_name(sha1); | |
991 | snprintf(tmpfile, sizeof(tmpfile), "%s.temp", filename); | |
992 | indexfile = fopen(tmpfile, "a"); | |
993 | if (!indexfile) { | |
994 | free(url); | |
995 | return error("Unable to open local file %s for pack index", | |
996 | tmpfile); | |
997 | } | |
998 | ||
999 | slot = get_active_slot(); | |
1000 | slot->results = &results; | |
1001 | curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0); | |
1002 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); | |
1003 | curl_easy_setopt(slot->curl, CURLOPT_FILE, indexfile); | |
1004 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite); | |
1005 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
1006 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, no_pragma_header); | |
1007 | slot->local = indexfile; | |
1008 | ||
1009 | /* If there is data present from a previous transfer attempt, | |
1010 | resume where it left off */ | |
1011 | prev_posn = ftell(indexfile); | |
1012 | if (prev_posn>0) { | |
1013 | if (push_verbosely) | |
1014 | fprintf(stderr, | |
1015 | "Resuming fetch of index for pack %s at byte %ld\n", | |
1016 | hex, prev_posn); | |
1017 | sprintf(range, "Range: bytes=%ld-", prev_posn); | |
1018 | range_header = curl_slist_append(range_header, range); | |
1019 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, range_header); | |
1020 | } | |
1021 | ||
1022 | if (start_active_slot(slot)) { | |
1023 | run_active_slot(slot); | |
1024 | if (results.curl_result != CURLE_OK) { | |
1025 | free(url); | |
1026 | fclose(indexfile); | |
1027 | return error("Unable to get pack index %s\n%s", url, | |
1028 | curl_errorstr); | |
1029 | } | |
1030 | } else { | |
1031 | free(url); | |
1032 | fclose(indexfile); | |
1033 | return error("Unable to start request"); | |
1034 | } | |
1035 | ||
1036 | free(url); | |
1037 | fclose(indexfile); | |
1038 | ||
1039 | return move_temp_to_file(tmpfile, filename); | |
1040 | } | |
1041 | ||
1042 | static int setup_index(unsigned char *sha1) | |
1043 | { | |
1044 | struct packed_git *new_pack; | |
1045 | ||
1046 | if (fetch_index(sha1)) | |
1047 | return -1; | |
1048 | ||
1049 | new_pack = parse_pack_index(sha1); | |
1050 | new_pack->next = repo->packs; | |
1051 | repo->packs = new_pack; | |
1052 | return 0; | |
1053 | } | |
1054 | ||
1055 | static int fetch_indices(void) | |
1056 | { | |
1057 | unsigned char sha1[20]; | |
1058 | char *url; | |
1059 | struct strbuf buffer = STRBUF_INIT; | |
1060 | char *data; | |
1061 | int i = 0; | |
1062 | ||
1063 | struct active_request_slot *slot; | |
1064 | struct slot_results results; | |
1065 | ||
1066 | if (push_verbosely) | |
1067 | fprintf(stderr, "Getting pack list\n"); | |
1068 | ||
1069 | url = xmalloc(strlen(repo->url) + 20); | |
1070 | sprintf(url, "%sobjects/info/packs", repo->url); | |
1071 | ||
1072 | slot = get_active_slot(); | |
1073 | slot->results = &results; | |
1074 | curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer); | |
1075 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer); | |
1076 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
1077 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL); | |
1078 | if (start_active_slot(slot)) { | |
1079 | run_active_slot(slot); | |
1080 | if (results.curl_result != CURLE_OK) { | |
1081 | strbuf_release(&buffer); | |
1082 | free(url); | |
1083 | if (results.http_code == 404) | |
1084 | return 0; | |
1085 | else | |
1086 | return error("%s", curl_errorstr); | |
1087 | } | |
1088 | } else { | |
1089 | strbuf_release(&buffer); | |
1090 | free(url); | |
1091 | return error("Unable to start request"); | |
1092 | } | |
1093 | free(url); | |
1094 | ||
1095 | data = buffer.buf; | |
1096 | while (i < buffer.len) { | |
1097 | switch (data[i]) { | |
1098 | case 'P': | |
1099 | i++; | |
1100 | if (i + 52 < buffer.len && | |
1101 | !prefixcmp(data + i, " pack-") && | |
1102 | !prefixcmp(data + i + 46, ".pack\n")) { | |
1103 | get_sha1_hex(data + i + 6, sha1); | |
1104 | setup_index(sha1); | |
1105 | i += 51; | |
1106 | break; | |
1107 | } | |
1108 | default: | |
1109 | while (data[i] != '\n') | |
1110 | i++; | |
1111 | } | |
1112 | i++; | |
1113 | } | |
1114 | ||
1115 | strbuf_release(&buffer); | |
1116 | return 0; | |
1117 | } | |
1118 | ||
1119 | static void one_remote_object(const char *hex) | |
1120 | { | |
1121 | unsigned char sha1[20]; | |
1122 | struct object *obj; | |
1123 | ||
1124 | if (get_sha1_hex(hex, sha1) != 0) | |
1125 | return; | |
1126 | ||
1127 | obj = lookup_object(sha1); | |
1128 | if (!obj) | |
1129 | obj = parse_object(sha1); | |
1130 | ||
1131 | /* Ignore remote objects that don't exist locally */ | |
1132 | if (!obj) | |
1133 | return; | |
1134 | ||
1135 | obj->flags |= REMOTE; | |
1136 | if (!object_list_contains(objects, obj)) | |
1137 | object_list_insert(obj, &objects); | |
1138 | } | |
1139 | ||
1140 | static void handle_lockprop_ctx(struct xml_ctx *ctx, int tag_closed) | |
1141 | { | |
1142 | int *lock_flags = (int *)ctx->userData; | |
1143 | ||
1144 | if (tag_closed) { | |
1145 | if (!strcmp(ctx->name, DAV_CTX_LOCKENTRY)) { | |
1146 | if ((*lock_flags & DAV_PROP_LOCKEX) && | |
1147 | (*lock_flags & DAV_PROP_LOCKWR)) { | |
1148 | *lock_flags |= DAV_LOCK_OK; | |
1149 | } | |
1150 | *lock_flags &= DAV_LOCK_OK; | |
1151 | } else if (!strcmp(ctx->name, DAV_CTX_LOCKTYPE_WRITE)) { | |
1152 | *lock_flags |= DAV_PROP_LOCKWR; | |
1153 | } else if (!strcmp(ctx->name, DAV_CTX_LOCKTYPE_EXCLUSIVE)) { | |
1154 | *lock_flags |= DAV_PROP_LOCKEX; | |
1155 | } | |
1156 | } | |
1157 | } | |
1158 | ||
1159 | static void handle_new_lock_ctx(struct xml_ctx *ctx, int tag_closed) | |
1160 | { | |
1161 | struct remote_lock *lock = (struct remote_lock *)ctx->userData; | |
1162 | git_SHA_CTX sha_ctx; | |
1163 | unsigned char lock_token_sha1[20]; | |
1164 | ||
1165 | if (tag_closed && ctx->cdata) { | |
1166 | if (!strcmp(ctx->name, DAV_ACTIVELOCK_OWNER)) { | |
1167 | lock->owner = xmalloc(strlen(ctx->cdata) + 1); | |
1168 | strcpy(lock->owner, ctx->cdata); | |
1169 | } else if (!strcmp(ctx->name, DAV_ACTIVELOCK_TIMEOUT)) { | |
1170 | if (!prefixcmp(ctx->cdata, "Second-")) | |
1171 | lock->timeout = | |
1172 | strtol(ctx->cdata + 7, NULL, 10); | |
1173 | } else if (!strcmp(ctx->name, DAV_ACTIVELOCK_TOKEN)) { | |
1174 | lock->token = xmalloc(strlen(ctx->cdata) + 1); | |
1175 | strcpy(lock->token, ctx->cdata); | |
1176 | ||
1177 | git_SHA1_Init(&sha_ctx); | |
1178 | git_SHA1_Update(&sha_ctx, lock->token, strlen(lock->token)); | |
1179 | git_SHA1_Final(lock_token_sha1, &sha_ctx); | |
1180 | ||
1181 | lock->tmpfile_suffix[0] = '_'; | |
1182 | memcpy(lock->tmpfile_suffix + 1, sha1_to_hex(lock_token_sha1), 40); | |
1183 | } | |
1184 | } | |
1185 | } | |
1186 | ||
1187 | static void one_remote_ref(char *refname); | |
1188 | ||
1189 | static void | |
1190 | xml_start_tag(void *userData, const char *name, const char **atts) | |
1191 | { | |
1192 | struct xml_ctx *ctx = (struct xml_ctx *)userData; | |
1193 | const char *c = strchr(name, ':'); | |
1194 | int new_len; | |
1195 | ||
1196 | if (c == NULL) | |
1197 | c = name; | |
1198 | else | |
1199 | c++; | |
1200 | ||
1201 | new_len = strlen(ctx->name) + strlen(c) + 2; | |
1202 | ||
1203 | if (new_len > ctx->len) { | |
1204 | ctx->name = xrealloc(ctx->name, new_len); | |
1205 | ctx->len = new_len; | |
1206 | } | |
1207 | strcat(ctx->name, "."); | |
1208 | strcat(ctx->name, c); | |
1209 | ||
1210 | free(ctx->cdata); | |
1211 | ctx->cdata = NULL; | |
1212 | ||
1213 | ctx->userFunc(ctx, 0); | |
1214 | } | |
1215 | ||
1216 | static void | |
1217 | xml_end_tag(void *userData, const char *name) | |
1218 | { | |
1219 | struct xml_ctx *ctx = (struct xml_ctx *)userData; | |
1220 | const char *c = strchr(name, ':'); | |
1221 | char *ep; | |
1222 | ||
1223 | ctx->userFunc(ctx, 1); | |
1224 | ||
1225 | if (c == NULL) | |
1226 | c = name; | |
1227 | else | |
1228 | c++; | |
1229 | ||
1230 | ep = ctx->name + strlen(ctx->name) - strlen(c) - 1; | |
1231 | *ep = 0; | |
1232 | } | |
1233 | ||
1234 | static void | |
1235 | xml_cdata(void *userData, const XML_Char *s, int len) | |
1236 | { | |
1237 | struct xml_ctx *ctx = (struct xml_ctx *)userData; | |
1238 | free(ctx->cdata); | |
1239 | ctx->cdata = xmemdupz(s, len); | |
1240 | } | |
1241 | ||
1242 | static struct remote_lock *lock_remote(const char *path, long timeout) | |
1243 | { | |
1244 | struct active_request_slot *slot; | |
1245 | struct slot_results results; | |
1246 | struct buffer out_buffer = { STRBUF_INIT, 0 }; | |
1247 | struct strbuf in_buffer = STRBUF_INIT; | |
1248 | char *url; | |
1249 | char *ep; | |
1250 | char timeout_header[25]; | |
1251 | struct remote_lock *lock = NULL; | |
1252 | struct curl_slist *dav_headers = NULL; | |
1253 | struct xml_ctx ctx; | |
1254 | char *escaped; | |
1255 | ||
1256 | url = xmalloc(strlen(repo->url) + strlen(path) + 1); | |
1257 | sprintf(url, "%s%s", repo->url, path); | |
1258 | ||
1259 | /* Make sure leading directories exist for the remote ref */ | |
1260 | ep = strchr(url + strlen(repo->url) + 1, '/'); | |
1261 | while (ep) { | |
1262 | char saved_character = ep[1]; | |
1263 | ep[1] = '\0'; | |
1264 | slot = get_active_slot(); | |
1265 | slot->results = &results; | |
1266 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); | |
1267 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
1268 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_MKCOL); | |
1269 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
1270 | if (start_active_slot(slot)) { | |
1271 | run_active_slot(slot); | |
1272 | if (results.curl_result != CURLE_OK && | |
1273 | results.http_code != 405) { | |
1274 | fprintf(stderr, | |
1275 | "Unable to create branch path %s\n", | |
1276 | url); | |
1277 | free(url); | |
1278 | return NULL; | |
1279 | } | |
1280 | } else { | |
1281 | fprintf(stderr, "Unable to start MKCOL request\n"); | |
1282 | free(url); | |
1283 | return NULL; | |
1284 | } | |
1285 | ep[1] = saved_character; | |
1286 | ep = strchr(ep + 1, '/'); | |
1287 | } | |
1288 | ||
1289 | escaped = xml_entities(git_default_email); | |
1290 | strbuf_addf(&out_buffer.buf, LOCK_REQUEST, escaped); | |
1291 | free(escaped); | |
1292 | ||
1293 | sprintf(timeout_header, "Timeout: Second-%ld", timeout); | |
1294 | dav_headers = curl_slist_append(dav_headers, timeout_header); | |
1295 | dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml"); | |
1296 | ||
1297 | slot = get_active_slot(); | |
1298 | slot->results = &results; | |
1299 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &out_buffer); | |
1300 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, out_buffer.buf.len); | |
1301 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
1302 | #ifndef NO_CURL_IOCTL | |
1303 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
1304 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &out_buffer); | |
1305 | #endif | |
1306 | curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer); | |
1307 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer); | |
1308 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
1309 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
1310 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_LOCK); | |
1311 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1312 | ||
1313 | lock = xcalloc(1, sizeof(*lock)); | |
1314 | lock->timeout = -1; | |
1315 | ||
1316 | if (start_active_slot(slot)) { | |
1317 | run_active_slot(slot); | |
1318 | if (results.curl_result == CURLE_OK) { | |
1319 | XML_Parser parser = XML_ParserCreate(NULL); | |
1320 | enum XML_Status result; | |
1321 | ctx.name = xcalloc(10, 1); | |
1322 | ctx.len = 0; | |
1323 | ctx.cdata = NULL; | |
1324 | ctx.userFunc = handle_new_lock_ctx; | |
1325 | ctx.userData = lock; | |
1326 | XML_SetUserData(parser, &ctx); | |
1327 | XML_SetElementHandler(parser, xml_start_tag, | |
1328 | xml_end_tag); | |
1329 | XML_SetCharacterDataHandler(parser, xml_cdata); | |
1330 | result = XML_Parse(parser, in_buffer.buf, | |
1331 | in_buffer.len, 1); | |
1332 | free(ctx.name); | |
1333 | if (result != XML_STATUS_OK) { | |
1334 | fprintf(stderr, "XML error: %s\n", | |
1335 | XML_ErrorString( | |
1336 | XML_GetErrorCode(parser))); | |
1337 | lock->timeout = -1; | |
1338 | } | |
1339 | XML_ParserFree(parser); | |
1340 | } | |
1341 | } else { | |
1342 | fprintf(stderr, "Unable to start LOCK request\n"); | |
1343 | } | |
1344 | ||
1345 | curl_slist_free_all(dav_headers); | |
1346 | strbuf_release(&out_buffer.buf); | |
1347 | strbuf_release(&in_buffer); | |
1348 | ||
1349 | if (lock->token == NULL || lock->timeout <= 0) { | |
1350 | free(lock->token); | |
1351 | free(lock->owner); | |
1352 | free(url); | |
1353 | free(lock); | |
1354 | lock = NULL; | |
1355 | } else { | |
1356 | lock->url = url; | |
1357 | lock->start_time = time(NULL); | |
1358 | lock->next = repo->locks; | |
1359 | repo->locks = lock; | |
1360 | } | |
1361 | ||
1362 | return lock; | |
1363 | } | |
1364 | ||
1365 | static int unlock_remote(struct remote_lock *lock) | |
1366 | { | |
1367 | struct active_request_slot *slot; | |
1368 | struct slot_results results; | |
1369 | struct remote_lock *prev = repo->locks; | |
1370 | struct curl_slist *dav_headers; | |
1371 | int rc = 0; | |
1372 | ||
1373 | dav_headers = get_dav_token_headers(lock, DAV_HEADER_LOCK); | |
1374 | ||
1375 | slot = get_active_slot(); | |
1376 | slot->results = &results; | |
1377 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
1378 | curl_easy_setopt(slot->curl, CURLOPT_URL, lock->url); | |
1379 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_UNLOCK); | |
1380 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1381 | ||
1382 | if (start_active_slot(slot)) { | |
1383 | run_active_slot(slot); | |
1384 | if (results.curl_result == CURLE_OK) | |
1385 | rc = 1; | |
1386 | else | |
1387 | fprintf(stderr, "UNLOCK HTTP error %ld\n", | |
1388 | results.http_code); | |
1389 | } else { | |
1390 | fprintf(stderr, "Unable to start UNLOCK request\n"); | |
1391 | } | |
1392 | ||
1393 | curl_slist_free_all(dav_headers); | |
1394 | ||
1395 | if (repo->locks == lock) { | |
1396 | repo->locks = lock->next; | |
1397 | } else { | |
1398 | while (prev && prev->next != lock) | |
1399 | prev = prev->next; | |
1400 | if (prev) | |
1401 | prev->next = prev->next->next; | |
1402 | } | |
1403 | ||
1404 | free(lock->owner); | |
1405 | free(lock->url); | |
1406 | free(lock->token); | |
1407 | free(lock); | |
1408 | ||
1409 | return rc; | |
1410 | } | |
1411 | ||
1412 | static void remove_locks(void) | |
1413 | { | |
1414 | struct remote_lock *lock = repo->locks; | |
1415 | ||
1416 | fprintf(stderr, "Removing remote locks...\n"); | |
1417 | while (lock) { | |
1418 | struct remote_lock *next = lock->next; | |
1419 | unlock_remote(lock); | |
1420 | lock = next; | |
1421 | } | |
1422 | } | |
1423 | ||
1424 | static void remove_locks_on_signal(int signo) | |
1425 | { | |
1426 | remove_locks(); | |
1427 | sigchain_pop(signo); | |
1428 | raise(signo); | |
1429 | } | |
1430 | ||
1431 | static void remote_ls(const char *path, int flags, | |
1432 | void (*userFunc)(struct remote_ls_ctx *ls), | |
1433 | void *userData); | |
1434 | ||
1435 | static void process_ls_object(struct remote_ls_ctx *ls) | |
1436 | { | |
1437 | unsigned int *parent = (unsigned int *)ls->userData; | |
1438 | char *path = ls->dentry_name; | |
1439 | char *obj_hex; | |
1440 | ||
1441 | if (!strcmp(ls->path, ls->dentry_name) && (ls->flags & IS_DIR)) { | |
1442 | remote_dir_exists[*parent] = 1; | |
1443 | return; | |
1444 | } | |
1445 | ||
1446 | if (strlen(path) != 49) | |
1447 | return; | |
1448 | path += 8; | |
1449 | obj_hex = xmalloc(strlen(path)); | |
1450 | /* NB: path is not null-terminated, can not use strlcpy here */ | |
1451 | memcpy(obj_hex, path, 2); | |
1452 | strcpy(obj_hex + 2, path + 3); | |
1453 | one_remote_object(obj_hex); | |
1454 | free(obj_hex); | |
1455 | } | |
1456 | ||
1457 | static void process_ls_ref(struct remote_ls_ctx *ls) | |
1458 | { | |
1459 | if (!strcmp(ls->path, ls->dentry_name) && (ls->dentry_flags & IS_DIR)) { | |
1460 | fprintf(stderr, " %s\n", ls->dentry_name); | |
1461 | return; | |
1462 | } | |
1463 | ||
1464 | if (!(ls->dentry_flags & IS_DIR)) | |
1465 | one_remote_ref(ls->dentry_name); | |
1466 | } | |
1467 | ||
1468 | static void handle_remote_ls_ctx(struct xml_ctx *ctx, int tag_closed) | |
1469 | { | |
1470 | struct remote_ls_ctx *ls = (struct remote_ls_ctx *)ctx->userData; | |
1471 | ||
1472 | if (tag_closed) { | |
1473 | if (!strcmp(ctx->name, DAV_PROPFIND_RESP) && ls->dentry_name) { | |
1474 | if (ls->dentry_flags & IS_DIR) { | |
1475 | if (ls->flags & PROCESS_DIRS) { | |
1476 | ls->userFunc(ls); | |
1477 | } | |
1478 | if (strcmp(ls->dentry_name, ls->path) && | |
1479 | ls->flags & RECURSIVE) { | |
1480 | remote_ls(ls->dentry_name, | |
1481 | ls->flags, | |
1482 | ls->userFunc, | |
1483 | ls->userData); | |
1484 | } | |
1485 | } else if (ls->flags & PROCESS_FILES) { | |
1486 | ls->userFunc(ls); | |
1487 | } | |
1488 | } else if (!strcmp(ctx->name, DAV_PROPFIND_NAME) && ctx->cdata) { | |
1489 | char *path = ctx->cdata; | |
1490 | if (*ctx->cdata == 'h') { | |
1491 | path = strstr(path, "//"); | |
1492 | if (path) { | |
1493 | path = strchr(path+2, '/'); | |
1494 | } | |
1495 | } | |
1496 | if (path) { | |
1497 | path += repo->path_len; | |
1498 | ls->dentry_name = xstrdup(path); | |
1499 | } | |
1500 | } else if (!strcmp(ctx->name, DAV_PROPFIND_COLLECTION)) { | |
1501 | ls->dentry_flags |= IS_DIR; | |
1502 | } | |
1503 | } else if (!strcmp(ctx->name, DAV_PROPFIND_RESP)) { | |
1504 | free(ls->dentry_name); | |
1505 | ls->dentry_name = NULL; | |
1506 | ls->dentry_flags = 0; | |
1507 | } | |
1508 | } | |
1509 | ||
1510 | /* | |
1511 | * NEEDSWORK: remote_ls() ignores info/refs on the remote side. But it | |
1512 | * should _only_ heed the information from that file, instead of trying to | |
1513 | * determine the refs from the remote file system (badly: it does not even | |
1514 | * know about packed-refs). | |
1515 | */ | |
1516 | static void remote_ls(const char *path, int flags, | |
1517 | void (*userFunc)(struct remote_ls_ctx *ls), | |
1518 | void *userData) | |
1519 | { | |
1520 | char *url = xmalloc(strlen(repo->url) + strlen(path) + 1); | |
1521 | struct active_request_slot *slot; | |
1522 | struct slot_results results; | |
1523 | struct strbuf in_buffer = STRBUF_INIT; | |
1524 | struct buffer out_buffer = { STRBUF_INIT, 0 }; | |
1525 | struct curl_slist *dav_headers = NULL; | |
1526 | struct xml_ctx ctx; | |
1527 | struct remote_ls_ctx ls; | |
1528 | ||
1529 | ls.flags = flags; | |
1530 | ls.path = xstrdup(path); | |
1531 | ls.dentry_name = NULL; | |
1532 | ls.dentry_flags = 0; | |
1533 | ls.userData = userData; | |
1534 | ls.userFunc = userFunc; | |
1535 | ||
1536 | sprintf(url, "%s%s", repo->url, path); | |
1537 | ||
1538 | strbuf_addf(&out_buffer.buf, PROPFIND_ALL_REQUEST); | |
1539 | ||
1540 | dav_headers = curl_slist_append(dav_headers, "Depth: 1"); | |
1541 | dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml"); | |
1542 | ||
1543 | slot = get_active_slot(); | |
1544 | slot->results = &results; | |
1545 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &out_buffer); | |
1546 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, out_buffer.buf.len); | |
1547 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
1548 | #ifndef NO_CURL_IOCTL | |
1549 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
1550 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &out_buffer); | |
1551 | #endif | |
1552 | curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer); | |
1553 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer); | |
1554 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
1555 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
1556 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PROPFIND); | |
1557 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1558 | ||
1559 | if (start_active_slot(slot)) { | |
1560 | run_active_slot(slot); | |
1561 | if (results.curl_result == CURLE_OK) { | |
1562 | XML_Parser parser = XML_ParserCreate(NULL); | |
1563 | enum XML_Status result; | |
1564 | ctx.name = xcalloc(10, 1); | |
1565 | ctx.len = 0; | |
1566 | ctx.cdata = NULL; | |
1567 | ctx.userFunc = handle_remote_ls_ctx; | |
1568 | ctx.userData = &ls; | |
1569 | XML_SetUserData(parser, &ctx); | |
1570 | XML_SetElementHandler(parser, xml_start_tag, | |
1571 | xml_end_tag); | |
1572 | XML_SetCharacterDataHandler(parser, xml_cdata); | |
1573 | result = XML_Parse(parser, in_buffer.buf, | |
1574 | in_buffer.len, 1); | |
1575 | free(ctx.name); | |
1576 | ||
1577 | if (result != XML_STATUS_OK) { | |
1578 | fprintf(stderr, "XML error: %s\n", | |
1579 | XML_ErrorString( | |
1580 | XML_GetErrorCode(parser))); | |
1581 | } | |
1582 | XML_ParserFree(parser); | |
1583 | } | |
1584 | } else { | |
1585 | fprintf(stderr, "Unable to start PROPFIND request\n"); | |
1586 | } | |
1587 | ||
1588 | free(ls.path); | |
1589 | free(url); | |
1590 | strbuf_release(&out_buffer.buf); | |
1591 | strbuf_release(&in_buffer); | |
1592 | curl_slist_free_all(dav_headers); | |
1593 | } | |
1594 | ||
1595 | static void get_remote_object_list(unsigned char parent) | |
1596 | { | |
1597 | char path[] = "objects/XX/"; | |
1598 | static const char hex[] = "0123456789abcdef"; | |
1599 | unsigned int val = parent; | |
1600 | ||
1601 | path[8] = hex[val >> 4]; | |
1602 | path[9] = hex[val & 0xf]; | |
1603 | remote_dir_exists[val] = 0; | |
1604 | remote_ls(path, (PROCESS_FILES | PROCESS_DIRS), | |
1605 | process_ls_object, &val); | |
1606 | } | |
1607 | ||
1608 | static int locking_available(void) | |
1609 | { | |
1610 | struct active_request_slot *slot; | |
1611 | struct slot_results results; | |
1612 | struct strbuf in_buffer = STRBUF_INIT; | |
1613 | struct buffer out_buffer = { STRBUF_INIT, 0 }; | |
1614 | struct curl_slist *dav_headers = NULL; | |
1615 | struct xml_ctx ctx; | |
1616 | int lock_flags = 0; | |
1617 | char *escaped; | |
1618 | ||
1619 | escaped = xml_entities(repo->url); | |
1620 | strbuf_addf(&out_buffer.buf, PROPFIND_SUPPORTEDLOCK_REQUEST, escaped); | |
1621 | free(escaped); | |
1622 | ||
1623 | dav_headers = curl_slist_append(dav_headers, "Depth: 0"); | |
1624 | dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml"); | |
1625 | ||
1626 | slot = get_active_slot(); | |
1627 | slot->results = &results; | |
1628 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &out_buffer); | |
1629 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, out_buffer.buf.len); | |
1630 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
1631 | #ifndef NO_CURL_IOCTL | |
1632 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
1633 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &out_buffer); | |
1634 | #endif | |
1635 | curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer); | |
1636 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer); | |
1637 | curl_easy_setopt(slot->curl, CURLOPT_URL, repo->url); | |
1638 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
1639 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PROPFIND); | |
1640 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1641 | ||
1642 | if (start_active_slot(slot)) { | |
1643 | run_active_slot(slot); | |
1644 | if (results.curl_result == CURLE_OK) { | |
1645 | XML_Parser parser = XML_ParserCreate(NULL); | |
1646 | enum XML_Status result; | |
1647 | ctx.name = xcalloc(10, 1); | |
1648 | ctx.len = 0; | |
1649 | ctx.cdata = NULL; | |
1650 | ctx.userFunc = handle_lockprop_ctx; | |
1651 | ctx.userData = &lock_flags; | |
1652 | XML_SetUserData(parser, &ctx); | |
1653 | XML_SetElementHandler(parser, xml_start_tag, | |
1654 | xml_end_tag); | |
1655 | result = XML_Parse(parser, in_buffer.buf, | |
1656 | in_buffer.len, 1); | |
1657 | free(ctx.name); | |
1658 | ||
1659 | if (result != XML_STATUS_OK) { | |
1660 | fprintf(stderr, "XML error: %s\n", | |
1661 | XML_ErrorString( | |
1662 | XML_GetErrorCode(parser))); | |
1663 | lock_flags = 0; | |
1664 | } | |
1665 | XML_ParserFree(parser); | |
1666 | if (!lock_flags) | |
1667 | error("no DAV locking support on %s", | |
1668 | repo->url); | |
1669 | ||
1670 | } else { | |
1671 | error("Cannot access URL %s, return code %d", | |
1672 | repo->url, results.curl_result); | |
1673 | lock_flags = 0; | |
1674 | } | |
1675 | } else { | |
1676 | error("Unable to start PROPFIND request on %s", repo->url); | |
1677 | } | |
1678 | ||
1679 | strbuf_release(&out_buffer.buf); | |
1680 | strbuf_release(&in_buffer); | |
1681 | curl_slist_free_all(dav_headers); | |
1682 | ||
1683 | return lock_flags; | |
1684 | } | |
1685 | ||
1686 | static struct object_list **add_one_object(struct object *obj, struct object_list **p) | |
1687 | { | |
1688 | struct object_list *entry = xmalloc(sizeof(struct object_list)); | |
1689 | entry->item = obj; | |
1690 | entry->next = *p; | |
1691 | *p = entry; | |
1692 | return &entry->next; | |
1693 | } | |
1694 | ||
1695 | static struct object_list **process_blob(struct blob *blob, | |
1696 | struct object_list **p, | |
1697 | struct name_path *path, | |
1698 | const char *name) | |
1699 | { | |
1700 | struct object *obj = &blob->object; | |
1701 | ||
1702 | obj->flags |= LOCAL; | |
1703 | ||
1704 | if (obj->flags & (UNINTERESTING | SEEN)) | |
1705 | return p; | |
1706 | ||
1707 | obj->flags |= SEEN; | |
1708 | return add_one_object(obj, p); | |
1709 | } | |
1710 | ||
1711 | static struct object_list **process_tree(struct tree *tree, | |
1712 | struct object_list **p, | |
1713 | struct name_path *path, | |
1714 | const char *name) | |
1715 | { | |
1716 | struct object *obj = &tree->object; | |
1717 | struct tree_desc desc; | |
1718 | struct name_entry entry; | |
1719 | struct name_path me; | |
1720 | ||
1721 | obj->flags |= LOCAL; | |
1722 | ||
1723 | if (obj->flags & (UNINTERESTING | SEEN)) | |
1724 | return p; | |
1725 | if (parse_tree(tree) < 0) | |
1726 | die("bad tree object %s", sha1_to_hex(obj->sha1)); | |
1727 | ||
1728 | obj->flags |= SEEN; | |
1729 | name = xstrdup(name); | |
1730 | p = add_one_object(obj, p); | |
1731 | me.up = path; | |
1732 | me.elem = name; | |
1733 | me.elem_len = strlen(name); | |
1734 | ||
1735 | init_tree_desc(&desc, tree->buffer, tree->size); | |
1736 | ||
1737 | while (tree_entry(&desc, &entry)) | |
1738 | switch (object_type(entry.mode)) { | |
1739 | case OBJ_TREE: | |
1740 | p = process_tree(lookup_tree(entry.sha1), p, &me, name); | |
1741 | break; | |
1742 | case OBJ_BLOB: | |
1743 | p = process_blob(lookup_blob(entry.sha1), p, &me, name); | |
1744 | break; | |
1745 | default: | |
1746 | /* Subproject commit - not in this repository */ | |
1747 | break; | |
1748 | } | |
1749 | ||
1750 | free(tree->buffer); | |
1751 | tree->buffer = NULL; | |
1752 | return p; | |
1753 | } | |
1754 | ||
1755 | static int get_delta(struct rev_info *revs, struct remote_lock *lock) | |
1756 | { | |
1757 | int i; | |
1758 | struct commit *commit; | |
1759 | struct object_list **p = &objects; | |
1760 | int count = 0; | |
1761 | ||
1762 | while ((commit = get_revision(revs)) != NULL) { | |
1763 | p = process_tree(commit->tree, p, NULL, ""); | |
1764 | commit->object.flags |= LOCAL; | |
1765 | if (!(commit->object.flags & UNINTERESTING)) | |
1766 | count += add_send_request(&commit->object, lock); | |
1767 | } | |
1768 | ||
1769 | for (i = 0; i < revs->pending.nr; i++) { | |
1770 | struct object_array_entry *entry = revs->pending.objects + i; | |
1771 | struct object *obj = entry->item; | |
1772 | const char *name = entry->name; | |
1773 | ||
1774 | if (obj->flags & (UNINTERESTING | SEEN)) | |
1775 | continue; | |
1776 | if (obj->type == OBJ_TAG) { | |
1777 | obj->flags |= SEEN; | |
1778 | p = add_one_object(obj, p); | |
1779 | continue; | |
1780 | } | |
1781 | if (obj->type == OBJ_TREE) { | |
1782 | p = process_tree((struct tree *)obj, p, NULL, name); | |
1783 | continue; | |
1784 | } | |
1785 | if (obj->type == OBJ_BLOB) { | |
1786 | p = process_blob((struct blob *)obj, p, NULL, name); | |
1787 | continue; | |
1788 | } | |
1789 | die("unknown pending object %s (%s)", sha1_to_hex(obj->sha1), name); | |
1790 | } | |
1791 | ||
1792 | while (objects) { | |
1793 | if (!(objects->item->flags & UNINTERESTING)) | |
1794 | count += add_send_request(objects->item, lock); | |
1795 | objects = objects->next; | |
1796 | } | |
1797 | ||
1798 | return count; | |
1799 | } | |
1800 | ||
1801 | static int update_remote(unsigned char *sha1, struct remote_lock *lock) | |
1802 | { | |
1803 | struct active_request_slot *slot; | |
1804 | struct slot_results results; | |
1805 | struct buffer out_buffer = { STRBUF_INIT, 0 }; | |
1806 | struct curl_slist *dav_headers; | |
1807 | ||
1808 | dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF); | |
1809 | ||
1810 | strbuf_addf(&out_buffer.buf, "%s\n", sha1_to_hex(sha1)); | |
1811 | ||
1812 | slot = get_active_slot(); | |
1813 | slot->results = &results; | |
1814 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &out_buffer); | |
1815 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, out_buffer.buf.len); | |
1816 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
1817 | #ifndef NO_CURL_IOCTL | |
1818 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
1819 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &out_buffer); | |
1820 | #endif | |
1821 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
1822 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PUT); | |
1823 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1824 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
1825 | curl_easy_setopt(slot->curl, CURLOPT_PUT, 1); | |
1826 | curl_easy_setopt(slot->curl, CURLOPT_URL, lock->url); | |
1827 | ||
1828 | if (start_active_slot(slot)) { | |
1829 | run_active_slot(slot); | |
1830 | strbuf_release(&out_buffer.buf); | |
1831 | if (results.curl_result != CURLE_OK) { | |
1832 | fprintf(stderr, | |
1833 | "PUT error: curl result=%d, HTTP code=%ld\n", | |
1834 | results.curl_result, results.http_code); | |
1835 | /* We should attempt recovery? */ | |
1836 | return 0; | |
1837 | } | |
1838 | } else { | |
1839 | strbuf_release(&out_buffer.buf); | |
1840 | fprintf(stderr, "Unable to start PUT request\n"); | |
1841 | return 0; | |
1842 | } | |
1843 | ||
1844 | return 1; | |
1845 | } | |
1846 | ||
1847 | static struct ref *remote_refs, **remote_tail; | |
1848 | ||
1849 | static void one_remote_ref(char *refname) | |
1850 | { | |
1851 | struct ref *ref; | |
1852 | struct object *obj; | |
1853 | ||
1854 | ref = alloc_ref(refname); | |
1855 | ||
1856 | if (http_fetch_ref(repo->url, ref) != 0) { | |
1857 | fprintf(stderr, | |
1858 | "Unable to fetch ref %s from %s\n", | |
1859 | refname, repo->url); | |
1860 | free(ref); | |
1861 | return; | |
1862 | } | |
1863 | ||
1864 | /* | |
1865 | * Fetch a copy of the object if it doesn't exist locally - it | |
1866 | * may be required for updating server info later. | |
1867 | */ | |
1868 | if (repo->can_update_info_refs && !has_sha1_file(ref->old_sha1)) { | |
1869 | obj = lookup_unknown_object(ref->old_sha1); | |
1870 | if (obj) { | |
1871 | fprintf(stderr, " fetch %s for %s\n", | |
1872 | sha1_to_hex(ref->old_sha1), refname); | |
1873 | add_fetch_request(obj); | |
1874 | } | |
1875 | } | |
1876 | ||
1877 | *remote_tail = ref; | |
1878 | remote_tail = &ref->next; | |
1879 | } | |
1880 | ||
1881 | static void get_dav_remote_heads(void) | |
1882 | { | |
1883 | remote_tail = &remote_refs; | |
1884 | remote_ls("refs/", (PROCESS_FILES | PROCESS_DIRS | RECURSIVE), process_ls_ref, NULL); | |
1885 | } | |
1886 | ||
1887 | static int is_zero_sha1(const unsigned char *sha1) | |
1888 | { | |
1889 | int i; | |
1890 | ||
1891 | for (i = 0; i < 20; i++) { | |
1892 | if (*sha1++) | |
1893 | return 0; | |
1894 | } | |
1895 | return 1; | |
1896 | } | |
1897 | ||
1898 | static void add_remote_info_ref(struct remote_ls_ctx *ls) | |
1899 | { | |
1900 | struct strbuf *buf = (struct strbuf *)ls->userData; | |
1901 | struct object *o; | |
1902 | int len; | |
1903 | char *ref_info; | |
1904 | struct ref *ref; | |
1905 | ||
1906 | ref = alloc_ref(ls->dentry_name); | |
1907 | ||
1908 | if (http_fetch_ref(repo->url, ref) != 0) { | |
1909 | fprintf(stderr, | |
1910 | "Unable to fetch ref %s from %s\n", | |
1911 | ls->dentry_name, repo->url); | |
1912 | aborted = 1; | |
1913 | free(ref); | |
1914 | return; | |
1915 | } | |
1916 | ||
1917 | o = parse_object(ref->old_sha1); | |
1918 | if (!o) { | |
1919 | fprintf(stderr, | |
1920 | "Unable to parse object %s for remote ref %s\n", | |
1921 | sha1_to_hex(ref->old_sha1), ls->dentry_name); | |
1922 | aborted = 1; | |
1923 | free(ref); | |
1924 | return; | |
1925 | } | |
1926 | ||
1927 | len = strlen(ls->dentry_name) + 42; | |
1928 | ref_info = xcalloc(len + 1, 1); | |
1929 | sprintf(ref_info, "%s %s\n", | |
1930 | sha1_to_hex(ref->old_sha1), ls->dentry_name); | |
1931 | fwrite_buffer(ref_info, 1, len, buf); | |
1932 | free(ref_info); | |
1933 | ||
1934 | if (o->type == OBJ_TAG) { | |
1935 | o = deref_tag(o, ls->dentry_name, 0); | |
1936 | if (o) { | |
1937 | len = strlen(ls->dentry_name) + 45; | |
1938 | ref_info = xcalloc(len + 1, 1); | |
1939 | sprintf(ref_info, "%s %s^{}\n", | |
1940 | sha1_to_hex(o->sha1), ls->dentry_name); | |
1941 | fwrite_buffer(ref_info, 1, len, buf); | |
1942 | free(ref_info); | |
1943 | } | |
1944 | } | |
1945 | free(ref); | |
1946 | } | |
1947 | ||
1948 | static void update_remote_info_refs(struct remote_lock *lock) | |
1949 | { | |
1950 | struct buffer buffer = { STRBUF_INIT, 0 }; | |
1951 | struct active_request_slot *slot; | |
1952 | struct slot_results results; | |
1953 | struct curl_slist *dav_headers; | |
1954 | ||
1955 | remote_ls("refs/", (PROCESS_FILES | RECURSIVE), | |
1956 | add_remote_info_ref, &buffer.buf); | |
1957 | if (!aborted) { | |
1958 | dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF); | |
1959 | ||
1960 | slot = get_active_slot(); | |
1961 | slot->results = &results; | |
1962 | curl_easy_setopt(slot->curl, CURLOPT_INFILE, &buffer); | |
1963 | curl_easy_setopt(slot->curl, CURLOPT_INFILESIZE, buffer.buf.len); | |
1964 | curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, fread_buffer); | |
1965 | #ifndef NO_CURL_IOCTL | |
1966 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer); | |
1967 | curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, &buffer); | |
1968 | #endif | |
1969 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
1970 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_PUT); | |
1971 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers); | |
1972 | curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 1); | |
1973 | curl_easy_setopt(slot->curl, CURLOPT_PUT, 1); | |
1974 | curl_easy_setopt(slot->curl, CURLOPT_URL, lock->url); | |
1975 | ||
1976 | if (start_active_slot(slot)) { | |
1977 | run_active_slot(slot); | |
1978 | if (results.curl_result != CURLE_OK) { | |
1979 | fprintf(stderr, | |
1980 | "PUT error: curl result=%d, HTTP code=%ld\n", | |
1981 | results.curl_result, results.http_code); | |
1982 | } | |
1983 | } | |
1984 | } | |
1985 | strbuf_release(&buffer.buf); | |
1986 | } | |
1987 | ||
1988 | static int remote_exists(const char *path) | |
1989 | { | |
1990 | char *url = xmalloc(strlen(repo->url) + strlen(path) + 1); | |
1991 | struct active_request_slot *slot; | |
1992 | struct slot_results results; | |
1993 | int ret = -1; | |
1994 | ||
1995 | sprintf(url, "%s%s", repo->url, path); | |
1996 | ||
1997 | slot = get_active_slot(); | |
1998 | slot->results = &results; | |
1999 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
2000 | curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 1); | |
2001 | ||
2002 | if (start_active_slot(slot)) { | |
2003 | run_active_slot(slot); | |
2004 | if (results.http_code == 404) | |
2005 | ret = 0; | |
2006 | else if (results.curl_result == CURLE_OK) | |
2007 | ret = 1; | |
2008 | else | |
2009 | fprintf(stderr, "HEAD HTTP error %ld\n", results.http_code); | |
2010 | } else { | |
2011 | fprintf(stderr, "Unable to start HEAD request\n"); | |
2012 | } | |
2013 | ||
2014 | free(url); | |
2015 | return ret; | |
2016 | } | |
2017 | ||
2018 | static void fetch_symref(const char *path, char **symref, unsigned char *sha1) | |
2019 | { | |
2020 | char *url; | |
2021 | struct strbuf buffer = STRBUF_INIT; | |
2022 | struct active_request_slot *slot; | |
2023 | struct slot_results results; | |
2024 | ||
2025 | url = xmalloc(strlen(repo->url) + strlen(path) + 1); | |
2026 | sprintf(url, "%s%s", repo->url, path); | |
2027 | ||
2028 | slot = get_active_slot(); | |
2029 | slot->results = &results; | |
2030 | curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer); | |
2031 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer); | |
2032 | curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL); | |
2033 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
2034 | if (start_active_slot(slot)) { | |
2035 | run_active_slot(slot); | |
2036 | if (results.curl_result != CURLE_OK) { | |
2037 | die("Couldn't get %s for remote symref\n%s", | |
2038 | url, curl_errorstr); | |
2039 | } | |
2040 | } else { | |
2041 | die("Unable to start remote symref request"); | |
2042 | } | |
2043 | free(url); | |
2044 | ||
2045 | free(*symref); | |
2046 | *symref = NULL; | |
2047 | hashclr(sha1); | |
2048 | ||
2049 | if (buffer.len == 0) | |
2050 | return; | |
2051 | ||
2052 | /* If it's a symref, set the refname; otherwise try for a sha1 */ | |
2053 | if (!prefixcmp((char *)buffer.buf, "ref: ")) { | |
2054 | *symref = xmemdupz((char *)buffer.buf + 5, buffer.len - 6); | |
2055 | } else { | |
2056 | get_sha1_hex(buffer.buf, sha1); | |
2057 | } | |
2058 | ||
2059 | strbuf_release(&buffer); | |
2060 | } | |
2061 | ||
2062 | static int verify_merge_base(unsigned char *head_sha1, unsigned char *branch_sha1) | |
2063 | { | |
2064 | struct commit *head = lookup_commit(head_sha1); | |
2065 | struct commit *branch = lookup_commit(branch_sha1); | |
2066 | struct commit_list *merge_bases = get_merge_bases(head, branch, 1); | |
2067 | ||
2068 | return (merge_bases && !merge_bases->next && merge_bases->item == branch); | |
2069 | } | |
2070 | ||
2071 | static int delete_remote_branch(char *pattern, int force) | |
2072 | { | |
2073 | struct ref *refs = remote_refs; | |
2074 | struct ref *remote_ref = NULL; | |
2075 | unsigned char head_sha1[20]; | |
2076 | char *symref = NULL; | |
2077 | int match; | |
2078 | int patlen = strlen(pattern); | |
2079 | int i; | |
2080 | struct active_request_slot *slot; | |
2081 | struct slot_results results; | |
2082 | char *url; | |
2083 | ||
2084 | /* Find the remote branch(es) matching the specified branch name */ | |
2085 | for (match = 0; refs; refs = refs->next) { | |
2086 | char *name = refs->name; | |
2087 | int namelen = strlen(name); | |
2088 | if (namelen < patlen || | |
2089 | memcmp(name + namelen - patlen, pattern, patlen)) | |
2090 | continue; | |
2091 | if (namelen != patlen && name[namelen - patlen - 1] != '/') | |
2092 | continue; | |
2093 | match++; | |
2094 | remote_ref = refs; | |
2095 | } | |
2096 | if (match == 0) | |
2097 | return error("No remote branch matches %s", pattern); | |
2098 | if (match != 1) | |
2099 | return error("More than one remote branch matches %s", | |
2100 | pattern); | |
2101 | ||
2102 | /* | |
2103 | * Remote HEAD must be a symref (not exactly foolproof; a remote | |
2104 | * symlink to a symref will look like a symref) | |
2105 | */ | |
2106 | fetch_symref("HEAD", &symref, head_sha1); | |
2107 | if (!symref) | |
2108 | return error("Remote HEAD is not a symref"); | |
2109 | ||
2110 | /* Remote branch must not be the remote HEAD */ | |
2111 | for (i=0; symref && i<MAXDEPTH; i++) { | |
2112 | if (!strcmp(remote_ref->name, symref)) | |
2113 | return error("Remote branch %s is the current HEAD", | |
2114 | remote_ref->name); | |
2115 | fetch_symref(symref, &symref, head_sha1); | |
2116 | } | |
2117 | ||
2118 | /* Run extra sanity checks if delete is not forced */ | |
2119 | if (!force) { | |
2120 | /* Remote HEAD must resolve to a known object */ | |
2121 | if (symref) | |
2122 | return error("Remote HEAD symrefs too deep"); | |
2123 | if (is_zero_sha1(head_sha1)) | |
2124 | return error("Unable to resolve remote HEAD"); | |
2125 | if (!has_sha1_file(head_sha1)) | |
2126 | return error("Remote HEAD resolves to object %s\nwhich does not exist locally, perhaps you need to fetch?", sha1_to_hex(head_sha1)); | |
2127 | ||
2128 | /* Remote branch must resolve to a known object */ | |
2129 | if (is_zero_sha1(remote_ref->old_sha1)) | |
2130 | return error("Unable to resolve remote branch %s", | |
2131 | remote_ref->name); | |
2132 | if (!has_sha1_file(remote_ref->old_sha1)) | |
2133 | return error("Remote branch %s resolves to object %s\nwhich does not exist locally, perhaps you need to fetch?", remote_ref->name, sha1_to_hex(remote_ref->old_sha1)); | |
2134 | ||
2135 | /* Remote branch must be an ancestor of remote HEAD */ | |
2136 | if (!verify_merge_base(head_sha1, remote_ref->old_sha1)) { | |
2137 | return error("The branch '%s' is not an ancestor " | |
2138 | "of your current HEAD.\n" | |
2139 | "If you are sure you want to delete it," | |
2140 | " run:\n\t'git http-push -D %s %s'", | |
2141 | remote_ref->name, repo->url, pattern); | |
2142 | } | |
2143 | } | |
2144 | ||
2145 | /* Send delete request */ | |
2146 | fprintf(stderr, "Removing remote branch '%s'\n", remote_ref->name); | |
2147 | if (dry_run) | |
2148 | return 0; | |
2149 | url = xmalloc(strlen(repo->url) + strlen(remote_ref->name) + 1); | |
2150 | sprintf(url, "%s%s", repo->url, remote_ref->name); | |
2151 | slot = get_active_slot(); | |
2152 | slot->results = &results; | |
2153 | curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1); | |
2154 | curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_null); | |
2155 | curl_easy_setopt(slot->curl, CURLOPT_URL, url); | |
2156 | curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, DAV_DELETE); | |
2157 | if (start_active_slot(slot)) { | |
2158 | run_active_slot(slot); | |
2159 | free(url); | |
2160 | if (results.curl_result != CURLE_OK) | |
2161 | return error("DELETE request failed (%d/%ld)\n", | |
2162 | results.curl_result, results.http_code); | |
2163 | } else { | |
2164 | free(url); | |
2165 | return error("Unable to start DELETE request"); | |
2166 | } | |
2167 | ||
2168 | return 0; | |
2169 | } | |
2170 | ||
2171 | int main(int argc, char **argv) | |
2172 | { | |
2173 | struct transfer_request *request; | |
2174 | struct transfer_request *next_request; | |
2175 | int nr_refspec = 0; | |
2176 | char **refspec = NULL; | |
2177 | struct remote_lock *ref_lock = NULL; | |
2178 | struct remote_lock *info_ref_lock = NULL; | |
2179 | struct rev_info revs; | |
2180 | int delete_branch = 0; | |
2181 | int force_delete = 0; | |
2182 | int objects_to_send; | |
2183 | int rc = 0; | |
2184 | int i; | |
2185 | int new_refs; | |
2186 | struct ref *ref, *local_refs; | |
2187 | struct remote *remote; | |
2188 | char *rewritten_url = NULL; | |
2189 | ||
2190 | git_extract_argv0_path(argv[0]); | |
2191 | ||
2192 | setup_git_directory(); | |
2193 | ||
2194 | repo = xcalloc(sizeof(*repo), 1); | |
2195 | ||
2196 | argv++; | |
2197 | for (i = 1; i < argc; i++, argv++) { | |
2198 | char *arg = *argv; | |
2199 | ||
2200 | if (*arg == '-') { | |
2201 | if (!strcmp(arg, "--all")) { | |
2202 | push_all = MATCH_REFS_ALL; | |
2203 | continue; | |
2204 | } | |
2205 | if (!strcmp(arg, "--force")) { | |
2206 | force_all = 1; | |
2207 | continue; | |
2208 | } | |
2209 | if (!strcmp(arg, "--dry-run")) { | |
2210 | dry_run = 1; | |
2211 | continue; | |
2212 | } | |
2213 | if (!strcmp(arg, "--verbose")) { | |
2214 | push_verbosely = 1; | |
2215 | continue; | |
2216 | } | |
2217 | if (!strcmp(arg, "-d")) { | |
2218 | delete_branch = 1; | |
2219 | continue; | |
2220 | } | |
2221 | if (!strcmp(arg, "-D")) { | |
2222 | delete_branch = 1; | |
2223 | force_delete = 1; | |
2224 | continue; | |
2225 | } | |
2226 | } | |
2227 | if (!repo->url) { | |
2228 | char *path = strstr(arg, "//"); | |
2229 | repo->url = arg; | |
2230 | repo->path_len = strlen(arg); | |
2231 | if (path) { | |
2232 | repo->path = strchr(path+2, '/'); | |
2233 | if (repo->path) | |
2234 | repo->path_len = strlen(repo->path); | |
2235 | } | |
2236 | continue; | |
2237 | } | |
2238 | refspec = argv; | |
2239 | nr_refspec = argc - i; | |
2240 | break; | |
2241 | } | |
2242 | ||
2243 | #ifndef USE_CURL_MULTI | |
2244 | die("git-push is not available for http/https repository when not compiled with USE_CURL_MULTI"); | |
2245 | #endif | |
2246 | ||
2247 | if (!repo->url) | |
2248 | usage(http_push_usage); | |
2249 | ||
2250 | if (delete_branch && nr_refspec != 1) | |
2251 | die("You must specify only one branch name when deleting a remote branch"); | |
2252 | ||
2253 | memset(remote_dir_exists, -1, 256); | |
2254 | ||
2255 | /* | |
2256 | * Create a minimum remote by hand to give to http_init(), | |
2257 | * primarily to allow it to look at the URL. | |
2258 | */ | |
2259 | remote = xcalloc(sizeof(*remote), 1); | |
2260 | ALLOC_GROW(remote->url, remote->url_nr + 1, remote->url_alloc); | |
2261 | remote->url[remote->url_nr++] = repo->url; | |
2262 | http_init(remote); | |
2263 | ||
2264 | no_pragma_header = curl_slist_append(no_pragma_header, "Pragma:"); | |
2265 | ||
2266 | if (repo->url && repo->url[strlen(repo->url)-1] != '/') { | |
2267 | rewritten_url = xmalloc(strlen(repo->url)+2); | |
2268 | strcpy(rewritten_url, repo->url); | |
2269 | strcat(rewritten_url, "/"); | |
2270 | repo->path = rewritten_url + (repo->path - repo->url); | |
2271 | repo->path_len++; | |
2272 | repo->url = rewritten_url; | |
2273 | } | |
2274 | ||
2275 | /* Verify DAV compliance/lock support */ | |
2276 | if (!locking_available()) { | |
2277 | rc = 1; | |
2278 | goto cleanup; | |
2279 | } | |
2280 | ||
2281 | sigchain_push_common(remove_locks_on_signal); | |
2282 | ||
2283 | /* Check whether the remote has server info files */ | |
2284 | repo->can_update_info_refs = 0; | |
2285 | repo->has_info_refs = remote_exists("info/refs"); | |
2286 | repo->has_info_packs = remote_exists("objects/info/packs"); | |
2287 | if (repo->has_info_refs) { | |
2288 | info_ref_lock = lock_remote("info/refs", LOCK_TIME); | |
2289 | if (info_ref_lock) | |
2290 | repo->can_update_info_refs = 1; | |
2291 | else { | |
2292 | error("cannot lock existing info/refs"); | |
2293 | rc = 1; | |
2294 | goto cleanup; | |
2295 | } | |
2296 | } | |
2297 | if (repo->has_info_packs) | |
2298 | fetch_indices(); | |
2299 | ||
2300 | /* Get a list of all local and remote heads to validate refspecs */ | |
2301 | local_refs = get_local_heads(); | |
2302 | fprintf(stderr, "Fetching remote heads...\n"); | |
2303 | get_dav_remote_heads(); | |
2304 | ||
2305 | /* Remove a remote branch if -d or -D was specified */ | |
2306 | if (delete_branch) { | |
2307 | if (delete_remote_branch(refspec[0], force_delete) == -1) | |
2308 | fprintf(stderr, "Unable to delete remote branch %s\n", | |
2309 | refspec[0]); | |
2310 | goto cleanup; | |
2311 | } | |
2312 | ||
2313 | /* match them up */ | |
2314 | if (!remote_tail) | |
2315 | remote_tail = &remote_refs; | |
2316 | if (match_refs(local_refs, remote_refs, &remote_tail, | |
2317 | nr_refspec, (const char **) refspec, push_all)) { | |
2318 | rc = -1; | |
2319 | goto cleanup; | |
2320 | } | |
2321 | if (!remote_refs) { | |
2322 | fprintf(stderr, "No refs in common and none specified; doing nothing.\n"); | |
2323 | rc = 0; | |
2324 | goto cleanup; | |
2325 | } | |
2326 | ||
2327 | new_refs = 0; | |
2328 | for (ref = remote_refs; ref; ref = ref->next) { | |
2329 | char old_hex[60], *new_hex; | |
2330 | const char *commit_argv[4]; | |
2331 | int commit_argc; | |
2332 | char *new_sha1_hex, *old_sha1_hex; | |
2333 | ||
2334 | if (!ref->peer_ref) | |
2335 | continue; | |
2336 | ||
2337 | if (is_zero_sha1(ref->peer_ref->new_sha1)) { | |
2338 | if (delete_remote_branch(ref->name, 1) == -1) { | |
2339 | error("Could not remove %s", ref->name); | |
2340 | rc = -4; | |
2341 | } | |
2342 | new_refs++; | |
2343 | continue; | |
2344 | } | |
2345 | ||
2346 | if (!hashcmp(ref->old_sha1, ref->peer_ref->new_sha1)) { | |
2347 | if (push_verbosely || 1) | |
2348 | fprintf(stderr, "'%s': up-to-date\n", ref->name); | |
2349 | continue; | |
2350 | } | |
2351 | ||
2352 | if (!force_all && | |
2353 | !is_zero_sha1(ref->old_sha1) && | |
2354 | !ref->force) { | |
2355 | if (!has_sha1_file(ref->old_sha1) || | |
2356 | !ref_newer(ref->peer_ref->new_sha1, | |
2357 | ref->old_sha1)) { | |
2358 | /* | |
2359 | * We do not have the remote ref, or | |
2360 | * we know that the remote ref is not | |
2361 | * an ancestor of what we are trying to | |
2362 | * push. Either way this can be losing | |
2363 | * commits at the remote end and likely | |
2364 | * we were not up to date to begin with. | |
2365 | */ | |
2366 | error("remote '%s' is not an ancestor of\n" | |
2367 | "local '%s'.\n" | |
2368 | "Maybe you are not up-to-date and " | |
2369 | "need to pull first?", | |
2370 | ref->name, | |
2371 | ref->peer_ref->name); | |
2372 | rc = -2; | |
2373 | continue; | |
2374 | } | |
2375 | } | |
2376 | hashcpy(ref->new_sha1, ref->peer_ref->new_sha1); | |
2377 | new_refs++; | |
2378 | strcpy(old_hex, sha1_to_hex(ref->old_sha1)); | |
2379 | new_hex = sha1_to_hex(ref->new_sha1); | |
2380 | ||
2381 | fprintf(stderr, "updating '%s'", ref->name); | |
2382 | if (strcmp(ref->name, ref->peer_ref->name)) | |
2383 | fprintf(stderr, " using '%s'", ref->peer_ref->name); | |
2384 | fprintf(stderr, "\n from %s\n to %s\n", old_hex, new_hex); | |
2385 | if (dry_run) | |
2386 | continue; | |
2387 | ||
2388 | /* Lock remote branch ref */ | |
2389 | ref_lock = lock_remote(ref->name, LOCK_TIME); | |
2390 | if (ref_lock == NULL) { | |
2391 | fprintf(stderr, "Unable to lock remote branch %s\n", | |
2392 | ref->name); | |
2393 | rc = 1; | |
2394 | continue; | |
2395 | } | |
2396 | ||
2397 | /* Set up revision info for this refspec */ | |
2398 | commit_argc = 3; | |
2399 | new_sha1_hex = xstrdup(sha1_to_hex(ref->new_sha1)); | |
2400 | old_sha1_hex = NULL; | |
2401 | commit_argv[1] = "--objects"; | |
2402 | commit_argv[2] = new_sha1_hex; | |
2403 | if (!push_all && !is_zero_sha1(ref->old_sha1)) { | |
2404 | old_sha1_hex = xmalloc(42); | |
2405 | sprintf(old_sha1_hex, "^%s", | |
2406 | sha1_to_hex(ref->old_sha1)); | |
2407 | commit_argv[3] = old_sha1_hex; | |
2408 | commit_argc++; | |
2409 | } | |
2410 | init_revisions(&revs, setup_git_directory()); | |
2411 | setup_revisions(commit_argc, commit_argv, &revs, NULL); | |
2412 | revs.edge_hint = 0; /* just in case */ | |
2413 | free(new_sha1_hex); | |
2414 | if (old_sha1_hex) { | |
2415 | free(old_sha1_hex); | |
2416 | commit_argv[1] = NULL; | |
2417 | } | |
2418 | ||
2419 | /* Generate a list of objects that need to be pushed */ | |
2420 | pushing = 0; | |
2421 | if (prepare_revision_walk(&revs)) | |
2422 | die("revision walk setup failed"); | |
2423 | mark_edges_uninteresting(revs.commits, &revs, NULL); | |
2424 | objects_to_send = get_delta(&revs, ref_lock); | |
2425 | finish_all_active_slots(); | |
2426 | ||
2427 | /* Push missing objects to remote, this would be a | |
2428 | convenient time to pack them first if appropriate. */ | |
2429 | pushing = 1; | |
2430 | if (objects_to_send) | |
2431 | fprintf(stderr, " sending %d objects\n", | |
2432 | objects_to_send); | |
2433 | #ifdef USE_CURL_MULTI | |
2434 | fill_active_slots(); | |
2435 | add_fill_function(NULL, fill_active_slot); | |
2436 | #endif | |
2437 | do { | |
2438 | finish_all_active_slots(); | |
2439 | #ifdef USE_CURL_MULTI | |
2440 | fill_active_slots(); | |
2441 | #endif | |
2442 | } while (request_queue_head && !aborted); | |
2443 | ||
2444 | /* Update the remote branch if all went well */ | |
2445 | if (aborted || !update_remote(ref->new_sha1, ref_lock)) | |
2446 | rc = 1; | |
2447 | ||
2448 | if (!rc) | |
2449 | fprintf(stderr, " done\n"); | |
2450 | unlock_remote(ref_lock); | |
2451 | check_locks(); | |
2452 | } | |
2453 | ||
2454 | /* Update remote server info if appropriate */ | |
2455 | if (repo->has_info_refs && new_refs) { | |
2456 | if (info_ref_lock && repo->can_update_info_refs) { | |
2457 | fprintf(stderr, "Updating remote server info\n"); | |
2458 | if (!dry_run) | |
2459 | update_remote_info_refs(info_ref_lock); | |
2460 | } else { | |
2461 | fprintf(stderr, "Unable to update server info\n"); | |
2462 | } | |
2463 | } | |
2464 | ||
2465 | cleanup: | |
2466 | free(rewritten_url); | |
2467 | if (info_ref_lock) | |
2468 | unlock_remote(info_ref_lock); | |
2469 | free(repo); | |
2470 | ||
2471 | curl_slist_free_all(no_pragma_header); | |
2472 | ||
2473 | http_cleanup(); | |
2474 | ||
2475 | request = request_queue_head; | |
2476 | while (request != NULL) { | |
2477 | next_request = request->next; | |
2478 | release_request(request); | |
2479 | request = next_request; | |
2480 | } | |
2481 | ||
2482 | return rc; | |
2483 | } |