parseopt: wrap rev-parse --parseopt usage for eval consumption
[alt-git.git] / transport.c
blob8ce39364a1e58b338bc45a4eb524b637ce3a8881
1 #include "cache.h"
2 #include "transport.h"
3 #include "run-command.h"
4 #include "pkt-line.h"
5 #include "fetch-pack.h"
6 #include "send-pack.h"
7 #include "walker.h"
8 #include "bundle.h"
9 #include "dir.h"
10 #include "refs.h"
11 #include "branch.h"
13 /* rsync support */
16 * We copy packed-refs and refs/ into a temporary file, then read the
17 * loose refs recursively (sorting whenever possible), and then inserting
18 * those packed refs that are not yet in the list (not validating, but
19 * assuming that the file is sorted).
21 * Appears refactoring this from refs.c is too cumbersome.
24 static int str_cmp(const void *a, const void *b)
26 const char *s1 = a;
27 const char *s2 = b;
29 return strcmp(s1, s2);
32 /* path->buf + name_offset is expected to point to "refs/" */
34 static int read_loose_refs(struct strbuf *path, int name_offset,
35 struct ref **tail)
37 DIR *dir = opendir(path->buf);
38 struct dirent *de;
39 struct {
40 char **entries;
41 int nr, alloc;
42 } list;
43 int i, pathlen;
45 if (!dir)
46 return -1;
48 memset (&list, 0, sizeof(list));
50 while ((de = readdir(dir))) {
51 if (is_dot_or_dotdot(de->d_name))
52 continue;
53 ALLOC_GROW(list.entries, list.nr + 1, list.alloc);
54 list.entries[list.nr++] = xstrdup(de->d_name);
56 closedir(dir);
58 /* sort the list */
60 qsort(list.entries, list.nr, sizeof(char *), str_cmp);
62 pathlen = path->len;
63 strbuf_addch(path, '/');
65 for (i = 0; i < list.nr; i++, strbuf_setlen(path, pathlen + 1)) {
66 strbuf_addstr(path, list.entries[i]);
67 if (read_loose_refs(path, name_offset, tail)) {
68 int fd = open(path->buf, O_RDONLY);
69 char buffer[40];
70 struct ref *next;
72 if (fd < 0)
73 continue;
74 next = alloc_ref(path->buf + name_offset);
75 if (read_in_full(fd, buffer, 40) != 40 ||
76 get_sha1_hex(buffer, next->old_sha1)) {
77 close(fd);
78 free(next);
79 continue;
81 close(fd);
82 (*tail)->next = next;
83 *tail = next;
86 strbuf_setlen(path, pathlen);
88 for (i = 0; i < list.nr; i++)
89 free(list.entries[i]);
90 free(list.entries);
92 return 0;
95 /* insert the packed refs for which no loose refs were found */
97 static void insert_packed_refs(const char *packed_refs, struct ref **list)
99 FILE *f = fopen(packed_refs, "r");
100 static char buffer[PATH_MAX];
102 if (!f)
103 return;
105 for (;;) {
106 int cmp = cmp, len;
108 if (!fgets(buffer, sizeof(buffer), f)) {
109 fclose(f);
110 return;
113 if (hexval(buffer[0]) > 0xf)
114 continue;
115 len = strlen(buffer);
116 if (len && buffer[len - 1] == '\n')
117 buffer[--len] = '\0';
118 if (len < 41)
119 continue;
120 while ((*list)->next &&
121 (cmp = strcmp(buffer + 41,
122 (*list)->next->name)) > 0)
123 list = &(*list)->next;
124 if (!(*list)->next || cmp < 0) {
125 struct ref *next = alloc_ref(buffer + 41);
126 buffer[40] = '\0';
127 if (get_sha1_hex(buffer, next->old_sha1)) {
128 warning ("invalid SHA-1: %s", buffer);
129 free(next);
130 continue;
132 next->next = (*list)->next;
133 (*list)->next = next;
134 list = &(*list)->next;
139 static void set_upstreams(struct transport *transport, struct ref *refs,
140 int pretend)
142 struct ref *ref;
143 for (ref = refs; ref; ref = ref->next) {
144 const char *localname;
145 const char *tmp;
146 const char *remotename;
147 unsigned char sha[20];
148 int flag = 0;
150 * Check suitability for tracking. Must be successful /
151 * already up-to-date ref create/modify (not delete).
153 if (ref->status != REF_STATUS_OK &&
154 ref->status != REF_STATUS_UPTODATE)
155 continue;
156 if (!ref->peer_ref)
157 continue;
158 if (!ref->new_sha1 || is_null_sha1(ref->new_sha1))
159 continue;
161 /* Follow symbolic refs (mainly for HEAD). */
162 localname = ref->peer_ref->name;
163 remotename = ref->name;
164 tmp = resolve_ref(localname, sha, 1, &flag);
165 if (tmp && flag & REF_ISSYMREF &&
166 !prefixcmp(tmp, "refs/heads/"))
167 localname = tmp;
169 /* Both source and destination must be local branches. */
170 if (!localname || prefixcmp(localname, "refs/heads/"))
171 continue;
172 if (!remotename || prefixcmp(remotename, "refs/heads/"))
173 continue;
175 if (!pretend)
176 install_branch_config(BRANCH_CONFIG_VERBOSE,
177 localname + 11, transport->remote->name,
178 remotename);
179 else
180 printf("Would set upstream of '%s' to '%s' of '%s'\n",
181 localname + 11, remotename + 11,
182 transport->remote->name);
186 static const char *rsync_url(const char *url)
188 return prefixcmp(url, "rsync://") ? skip_prefix(url, "rsync:") : url;
191 static struct ref *get_refs_via_rsync(struct transport *transport, int for_push)
193 struct strbuf buf = STRBUF_INIT, temp_dir = STRBUF_INIT;
194 struct ref dummy = {0}, *tail = &dummy;
195 struct child_process rsync;
196 const char *args[5];
197 int temp_dir_len;
199 if (for_push)
200 return NULL;
202 /* copy the refs to the temporary directory */
204 strbuf_addstr(&temp_dir, git_path("rsync-refs-XXXXXX"));
205 if (!mkdtemp(temp_dir.buf))
206 die_errno ("Could not make temporary directory");
207 temp_dir_len = temp_dir.len;
209 strbuf_addstr(&buf, rsync_url(transport->url));
210 strbuf_addstr(&buf, "/refs");
212 memset(&rsync, 0, sizeof(rsync));
213 rsync.argv = args;
214 rsync.stdout_to_stderr = 1;
215 args[0] = "rsync";
216 args[1] = (transport->verbose > 0) ? "-rv" : "-r";
217 args[2] = buf.buf;
218 args[3] = temp_dir.buf;
219 args[4] = NULL;
221 if (run_command(&rsync))
222 die ("Could not run rsync to get refs");
224 strbuf_reset(&buf);
225 strbuf_addstr(&buf, rsync_url(transport->url));
226 strbuf_addstr(&buf, "/packed-refs");
228 args[2] = buf.buf;
230 if (run_command(&rsync))
231 die ("Could not run rsync to get refs");
233 /* read the copied refs */
235 strbuf_addstr(&temp_dir, "/refs");
236 read_loose_refs(&temp_dir, temp_dir_len + 1, &tail);
237 strbuf_setlen(&temp_dir, temp_dir_len);
239 tail = &dummy;
240 strbuf_addstr(&temp_dir, "/packed-refs");
241 insert_packed_refs(temp_dir.buf, &tail);
242 strbuf_setlen(&temp_dir, temp_dir_len);
244 if (remove_dir_recursively(&temp_dir, 0))
245 warning ("Error removing temporary directory %s.",
246 temp_dir.buf);
248 strbuf_release(&buf);
249 strbuf_release(&temp_dir);
251 return dummy.next;
254 static int fetch_objs_via_rsync(struct transport *transport,
255 int nr_objs, struct ref **to_fetch)
257 struct strbuf buf = STRBUF_INIT;
258 struct child_process rsync;
259 const char *args[8];
260 int result;
262 strbuf_addstr(&buf, rsync_url(transport->url));
263 strbuf_addstr(&buf, "/objects/");
265 memset(&rsync, 0, sizeof(rsync));
266 rsync.argv = args;
267 rsync.stdout_to_stderr = 1;
268 args[0] = "rsync";
269 args[1] = (transport->verbose > 0) ? "-rv" : "-r";
270 args[2] = "--ignore-existing";
271 args[3] = "--exclude";
272 args[4] = "info";
273 args[5] = buf.buf;
274 args[6] = get_object_directory();
275 args[7] = NULL;
277 /* NEEDSWORK: handle one level of alternates */
278 result = run_command(&rsync);
280 strbuf_release(&buf);
282 return result;
285 static int write_one_ref(const char *name, const unsigned char *sha1,
286 int flags, void *data)
288 struct strbuf *buf = data;
289 int len = buf->len;
290 FILE *f;
292 /* when called via for_each_ref(), flags is non-zero */
293 if (flags && prefixcmp(name, "refs/heads/") &&
294 prefixcmp(name, "refs/tags/"))
295 return 0;
297 strbuf_addstr(buf, name);
298 if (safe_create_leading_directories(buf->buf) ||
299 !(f = fopen(buf->buf, "w")) ||
300 fprintf(f, "%s\n", sha1_to_hex(sha1)) < 0 ||
301 fclose(f))
302 return error("problems writing temporary file %s", buf->buf);
303 strbuf_setlen(buf, len);
304 return 0;
307 static int write_refs_to_temp_dir(struct strbuf *temp_dir,
308 int refspec_nr, const char **refspec)
310 int i;
312 for (i = 0; i < refspec_nr; i++) {
313 unsigned char sha1[20];
314 char *ref;
316 if (dwim_ref(refspec[i], strlen(refspec[i]), sha1, &ref) != 1)
317 return error("Could not get ref %s", refspec[i]);
319 if (write_one_ref(ref, sha1, 0, temp_dir)) {
320 free(ref);
321 return -1;
323 free(ref);
325 return 0;
328 static int rsync_transport_push(struct transport *transport,
329 int refspec_nr, const char **refspec, int flags)
331 struct strbuf buf = STRBUF_INIT, temp_dir = STRBUF_INIT;
332 int result = 0, i;
333 struct child_process rsync;
334 const char *args[10];
336 if (flags & TRANSPORT_PUSH_MIRROR)
337 return error("rsync transport does not support mirror mode");
339 /* first push the objects */
341 strbuf_addstr(&buf, rsync_url(transport->url));
342 strbuf_addch(&buf, '/');
344 memset(&rsync, 0, sizeof(rsync));
345 rsync.argv = args;
346 rsync.stdout_to_stderr = 1;
347 i = 0;
348 args[i++] = "rsync";
349 args[i++] = "-a";
350 if (flags & TRANSPORT_PUSH_DRY_RUN)
351 args[i++] = "--dry-run";
352 if (transport->verbose > 0)
353 args[i++] = "-v";
354 args[i++] = "--ignore-existing";
355 args[i++] = "--exclude";
356 args[i++] = "info";
357 args[i++] = get_object_directory();
358 args[i++] = buf.buf;
359 args[i++] = NULL;
361 if (run_command(&rsync))
362 return error("Could not push objects to %s",
363 rsync_url(transport->url));
365 /* copy the refs to the temporary directory; they could be packed. */
367 strbuf_addstr(&temp_dir, git_path("rsync-refs-XXXXXX"));
368 if (!mkdtemp(temp_dir.buf))
369 die_errno ("Could not make temporary directory");
370 strbuf_addch(&temp_dir, '/');
372 if (flags & TRANSPORT_PUSH_ALL) {
373 if (for_each_ref(write_one_ref, &temp_dir))
374 return -1;
375 } else if (write_refs_to_temp_dir(&temp_dir, refspec_nr, refspec))
376 return -1;
378 i = 2;
379 if (flags & TRANSPORT_PUSH_DRY_RUN)
380 args[i++] = "--dry-run";
381 if (!(flags & TRANSPORT_PUSH_FORCE))
382 args[i++] = "--ignore-existing";
383 args[i++] = temp_dir.buf;
384 args[i++] = rsync_url(transport->url);
385 args[i++] = NULL;
386 if (run_command(&rsync))
387 result = error("Could not push to %s",
388 rsync_url(transport->url));
390 if (remove_dir_recursively(&temp_dir, 0))
391 warning ("Could not remove temporary directory %s.",
392 temp_dir.buf);
394 strbuf_release(&buf);
395 strbuf_release(&temp_dir);
397 return result;
400 struct bundle_transport_data {
401 int fd;
402 struct bundle_header header;
405 static struct ref *get_refs_from_bundle(struct transport *transport, int for_push)
407 struct bundle_transport_data *data = transport->data;
408 struct ref *result = NULL;
409 int i;
411 if (for_push)
412 return NULL;
414 if (data->fd > 0)
415 close(data->fd);
416 data->fd = read_bundle_header(transport->url, &data->header);
417 if (data->fd < 0)
418 die ("Could not read bundle '%s'.", transport->url);
419 for (i = 0; i < data->header.references.nr; i++) {
420 struct ref_list_entry *e = data->header.references.list + i;
421 struct ref *ref = alloc_ref(e->name);
422 hashcpy(ref->old_sha1, e->sha1);
423 ref->next = result;
424 result = ref;
426 return result;
429 static int fetch_refs_from_bundle(struct transport *transport,
430 int nr_heads, struct ref **to_fetch)
432 struct bundle_transport_data *data = transport->data;
433 return unbundle(&data->header, data->fd);
436 static int close_bundle(struct transport *transport)
438 struct bundle_transport_data *data = transport->data;
439 if (data->fd > 0)
440 close(data->fd);
441 free(data);
442 return 0;
445 struct git_transport_data {
446 struct git_transport_options options;
447 struct child_process *conn;
448 int fd[2];
449 unsigned got_remote_heads : 1;
450 struct extra_have_objects extra_have;
453 static int set_git_option(struct git_transport_options *opts,
454 const char *name, const char *value)
456 if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
457 opts->uploadpack = value;
458 return 0;
459 } else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
460 opts->receivepack = value;
461 return 0;
462 } else if (!strcmp(name, TRANS_OPT_THIN)) {
463 opts->thin = !!value;
464 return 0;
465 } else if (!strcmp(name, TRANS_OPT_FOLLOWTAGS)) {
466 opts->followtags = !!value;
467 return 0;
468 } else if (!strcmp(name, TRANS_OPT_KEEP)) {
469 opts->keep = !!value;
470 return 0;
471 } else if (!strcmp(name, TRANS_OPT_DEPTH)) {
472 if (!value)
473 opts->depth = 0;
474 else
475 opts->depth = atoi(value);
476 return 0;
478 return 1;
481 static int connect_setup(struct transport *transport, int for_push, int verbose)
483 struct git_transport_data *data = transport->data;
485 if (data->conn)
486 return 0;
488 data->conn = git_connect(data->fd, transport->url,
489 for_push ? data->options.receivepack :
490 data->options.uploadpack,
491 verbose ? CONNECT_VERBOSE : 0);
493 return 0;
496 static struct ref *get_refs_via_connect(struct transport *transport, int for_push)
498 struct git_transport_data *data = transport->data;
499 struct ref *refs;
501 connect_setup(transport, for_push, 0);
502 get_remote_heads(data->fd[0], &refs, 0, NULL,
503 for_push ? REF_NORMAL : 0, &data->extra_have);
504 data->got_remote_heads = 1;
506 return refs;
509 static int fetch_refs_via_pack(struct transport *transport,
510 int nr_heads, struct ref **to_fetch)
512 struct git_transport_data *data = transport->data;
513 char **heads = xmalloc(nr_heads * sizeof(*heads));
514 char **origh = xmalloc(nr_heads * sizeof(*origh));
515 const struct ref *refs;
516 char *dest = xstrdup(transport->url);
517 struct fetch_pack_args args;
518 int i;
519 struct ref *refs_tmp = NULL;
521 memset(&args, 0, sizeof(args));
522 args.uploadpack = data->options.uploadpack;
523 args.keep_pack = data->options.keep;
524 args.lock_pack = 1;
525 args.use_thin_pack = data->options.thin;
526 args.include_tag = data->options.followtags;
527 args.verbose = (transport->verbose > 0);
528 args.quiet = (transport->verbose < 0);
529 args.no_progress = !transport->progress;
530 args.depth = data->options.depth;
532 for (i = 0; i < nr_heads; i++)
533 origh[i] = heads[i] = xstrdup(to_fetch[i]->name);
535 if (!data->got_remote_heads) {
536 connect_setup(transport, 0, 0);
537 get_remote_heads(data->fd[0], &refs_tmp, 0, NULL, 0, NULL);
538 data->got_remote_heads = 1;
541 refs = fetch_pack(&args, data->fd, data->conn,
542 refs_tmp ? refs_tmp : transport->remote_refs,
543 dest, nr_heads, heads, &transport->pack_lockfile);
544 close(data->fd[0]);
545 close(data->fd[1]);
546 if (finish_connect(data->conn))
547 refs = NULL;
548 data->conn = NULL;
549 data->got_remote_heads = 0;
551 free_refs(refs_tmp);
553 for (i = 0; i < nr_heads; i++)
554 free(origh[i]);
555 free(origh);
556 free(heads);
557 free(dest);
558 return (refs ? 0 : -1);
561 static int push_had_errors(struct ref *ref)
563 for (; ref; ref = ref->next) {
564 switch (ref->status) {
565 case REF_STATUS_NONE:
566 case REF_STATUS_UPTODATE:
567 case REF_STATUS_OK:
568 break;
569 default:
570 return 1;
573 return 0;
576 int transport_refs_pushed(struct ref *ref)
578 for (; ref; ref = ref->next) {
579 switch(ref->status) {
580 case REF_STATUS_NONE:
581 case REF_STATUS_UPTODATE:
582 break;
583 default:
584 return 1;
587 return 0;
590 void transport_update_tracking_ref(struct remote *remote, struct ref *ref, int verbose)
592 struct refspec rs;
594 if (ref->status != REF_STATUS_OK && ref->status != REF_STATUS_UPTODATE)
595 return;
597 rs.src = ref->name;
598 rs.dst = NULL;
600 if (!remote_find_tracking(remote, &rs)) {
601 if (verbose)
602 fprintf(stderr, "updating local tracking ref '%s'\n", rs.dst);
603 if (ref->deletion) {
604 delete_ref(rs.dst, NULL, 0);
605 } else
606 update_ref("update by push", rs.dst,
607 ref->new_sha1, NULL, 0, 0);
608 free(rs.dst);
612 static void print_ref_status(char flag, const char *summary, struct ref *to, struct ref *from, const char *msg, int porcelain)
614 if (porcelain) {
615 if (from)
616 fprintf(stdout, "%c\t%s:%s\t", flag, from->name, to->name);
617 else
618 fprintf(stdout, "%c\t:%s\t", flag, to->name);
619 if (msg)
620 fprintf(stdout, "%s (%s)\n", summary, msg);
621 else
622 fprintf(stdout, "%s\n", summary);
623 } else {
624 fprintf(stderr, " %c %-*s ", flag, TRANSPORT_SUMMARY_WIDTH, summary);
625 if (from)
626 fprintf(stderr, "%s -> %s", prettify_refname(from->name), prettify_refname(to->name));
627 else
628 fputs(prettify_refname(to->name), stderr);
629 if (msg) {
630 fputs(" (", stderr);
631 fputs(msg, stderr);
632 fputc(')', stderr);
634 fputc('\n', stderr);
638 static const char *status_abbrev(unsigned char sha1[20])
640 return find_unique_abbrev(sha1, DEFAULT_ABBREV);
643 static void print_ok_ref_status(struct ref *ref, int porcelain)
645 if (ref->deletion)
646 print_ref_status('-', "[deleted]", ref, NULL, NULL, porcelain);
647 else if (is_null_sha1(ref->old_sha1))
648 print_ref_status('*',
649 (!prefixcmp(ref->name, "refs/tags/") ? "[new tag]" :
650 "[new branch]"),
651 ref, ref->peer_ref, NULL, porcelain);
652 else {
653 char quickref[84];
654 char type;
655 const char *msg;
657 strcpy(quickref, status_abbrev(ref->old_sha1));
658 if (ref->nonfastforward) {
659 strcat(quickref, "...");
660 type = '+';
661 msg = "forced update";
662 } else {
663 strcat(quickref, "..");
664 type = ' ';
665 msg = NULL;
667 strcat(quickref, status_abbrev(ref->new_sha1));
669 print_ref_status(type, quickref, ref, ref->peer_ref, msg, porcelain);
673 static int print_one_push_status(struct ref *ref, const char *dest, int count, int porcelain)
675 if (!count)
676 fprintf(porcelain ? stdout : stderr, "To %s\n", dest);
678 switch(ref->status) {
679 case REF_STATUS_NONE:
680 print_ref_status('X', "[no match]", ref, NULL, NULL, porcelain);
681 break;
682 case REF_STATUS_REJECT_NODELETE:
683 print_ref_status('!', "[rejected]", ref, NULL,
684 "remote does not support deleting refs", porcelain);
685 break;
686 case REF_STATUS_UPTODATE:
687 print_ref_status('=', "[up to date]", ref,
688 ref->peer_ref, NULL, porcelain);
689 break;
690 case REF_STATUS_REJECT_NONFASTFORWARD:
691 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
692 "non-fast-forward", porcelain);
693 break;
694 case REF_STATUS_REMOTE_REJECT:
695 print_ref_status('!', "[remote rejected]", ref,
696 ref->deletion ? NULL : ref->peer_ref,
697 ref->remote_status, porcelain);
698 break;
699 case REF_STATUS_EXPECTING_REPORT:
700 print_ref_status('!', "[remote failure]", ref,
701 ref->deletion ? NULL : ref->peer_ref,
702 "remote failed to report status", porcelain);
703 break;
704 case REF_STATUS_OK:
705 print_ok_ref_status(ref, porcelain);
706 break;
709 return 1;
712 void transport_print_push_status(const char *dest, struct ref *refs,
713 int verbose, int porcelain, int *nonfastforward)
715 struct ref *ref;
716 int n = 0;
718 if (verbose) {
719 for (ref = refs; ref; ref = ref->next)
720 if (ref->status == REF_STATUS_UPTODATE)
721 n += print_one_push_status(ref, dest, n, porcelain);
724 for (ref = refs; ref; ref = ref->next)
725 if (ref->status == REF_STATUS_OK)
726 n += print_one_push_status(ref, dest, n, porcelain);
728 *nonfastforward = 0;
729 for (ref = refs; ref; ref = ref->next) {
730 if (ref->status != REF_STATUS_NONE &&
731 ref->status != REF_STATUS_UPTODATE &&
732 ref->status != REF_STATUS_OK)
733 n += print_one_push_status(ref, dest, n, porcelain);
734 if (ref->status == REF_STATUS_REJECT_NONFASTFORWARD)
735 *nonfastforward = 1;
739 void transport_verify_remote_names(int nr_heads, const char **heads)
741 int i;
743 for (i = 0; i < nr_heads; i++) {
744 const char *local = heads[i];
745 const char *remote = strrchr(heads[i], ':');
747 if (*local == '+')
748 local++;
750 /* A matching refspec is okay. */
751 if (remote == local && remote[1] == '\0')
752 continue;
754 remote = remote ? (remote + 1) : local;
755 switch (check_ref_format(remote)) {
756 case 0: /* ok */
757 case CHECK_REF_FORMAT_ONELEVEL:
758 /* ok but a single level -- that is fine for
759 * a match pattern.
761 case CHECK_REF_FORMAT_WILDCARD:
762 /* ok but ends with a pattern-match character */
763 continue;
765 die("remote part of refspec is not a valid name in %s",
766 heads[i]);
770 static int git_transport_push(struct transport *transport, struct ref *remote_refs, int flags)
772 struct git_transport_data *data = transport->data;
773 struct send_pack_args args;
774 int ret;
776 if (!data->got_remote_heads) {
777 struct ref *tmp_refs;
778 connect_setup(transport, 1, 0);
780 get_remote_heads(data->fd[0], &tmp_refs, 0, NULL, REF_NORMAL,
781 NULL);
782 data->got_remote_heads = 1;
785 memset(&args, 0, sizeof(args));
786 args.send_mirror = !!(flags & TRANSPORT_PUSH_MIRROR);
787 args.force_update = !!(flags & TRANSPORT_PUSH_FORCE);
788 args.use_thin_pack = data->options.thin;
789 args.verbose = (transport->verbose > 0);
790 args.quiet = (transport->verbose < 0);
791 args.dry_run = !!(flags & TRANSPORT_PUSH_DRY_RUN);
792 args.porcelain = !!(flags & TRANSPORT_PUSH_PORCELAIN);
794 ret = send_pack(&args, data->fd, data->conn, remote_refs,
795 &data->extra_have);
797 close(data->fd[1]);
798 close(data->fd[0]);
799 ret |= finish_connect(data->conn);
800 data->conn = NULL;
801 data->got_remote_heads = 0;
803 return ret;
806 static int connect_git(struct transport *transport, const char *name,
807 const char *executable, int fd[2])
809 struct git_transport_data *data = transport->data;
810 data->conn = git_connect(data->fd, transport->url,
811 executable, 0);
812 fd[0] = data->fd[0];
813 fd[1] = data->fd[1];
814 return 0;
817 static int disconnect_git(struct transport *transport)
819 struct git_transport_data *data = transport->data;
820 if (data->conn) {
821 if (data->got_remote_heads)
822 packet_flush(data->fd[1]);
823 close(data->fd[0]);
824 close(data->fd[1]);
825 finish_connect(data->conn);
828 free(data);
829 return 0;
832 void transport_take_over(struct transport *transport,
833 struct child_process *child)
835 struct git_transport_data *data;
837 if (!transport->smart_options)
838 die("Bug detected: Taking over transport requires non-NULL "
839 "smart_options field.");
841 data = xcalloc(1, sizeof(*data));
842 data->options = *transport->smart_options;
843 data->conn = child;
844 data->fd[0] = data->conn->out;
845 data->fd[1] = data->conn->in;
846 data->got_remote_heads = 0;
847 transport->data = data;
849 transport->set_option = NULL;
850 transport->get_refs_list = get_refs_via_connect;
851 transport->fetch = fetch_refs_via_pack;
852 transport->push = NULL;
853 transport->push_refs = git_transport_push;
854 transport->disconnect = disconnect_git;
855 transport->smart_options = &(data->options);
858 static int is_local(const char *url)
860 const char *colon = strchr(url, ':');
861 const char *slash = strchr(url, '/');
862 return !colon || (slash && slash < colon) ||
863 has_dos_drive_prefix(url);
866 static int is_file(const char *url)
868 struct stat buf;
869 if (stat(url, &buf))
870 return 0;
871 return S_ISREG(buf.st_mode);
874 static int isurlschemechar(int first_flag, int ch)
877 * The set of valid URL schemes, as per STD66 (RFC3986) is
878 * '[A-Za-z][A-Za-z0-9+.-]*'. But use sightly looser check
879 * of '[A-Za-z0-9][A-Za-z0-9+.-]*' because earlier version
880 * of check used '[A-Za-z0-9]+' so not to break any remote
881 * helpers.
883 int alphanumeric, special;
884 alphanumeric = ch > 0 && isalnum(ch);
885 special = ch == '+' || ch == '-' || ch == '.';
886 return alphanumeric || (!first_flag && special);
889 static int is_url(const char *url)
891 const char *url2, *first_slash;
893 if (!url)
894 return 0;
895 url2 = url;
896 first_slash = strchr(url, '/');
898 /* Input with no slash at all or slash first can't be URL. */
899 if (!first_slash || first_slash == url)
900 return 0;
901 /* Character before must be : and next must be /. */
902 if (first_slash[-1] != ':' || first_slash[1] != '/')
903 return 0;
904 /* There must be something before the :// */
905 if (first_slash == url + 1)
906 return 0;
908 * Check all characters up to first slash - 1. Only alphanum
909 * is allowed.
911 url2 = url;
912 while (url2 < first_slash - 1) {
913 if (!isurlschemechar(url2 == url, (unsigned char)*url2))
914 return 0;
915 url2++;
918 /* Valid enough. */
919 return 1;
922 static int external_specification_len(const char *url)
924 return strchr(url, ':') - url;
927 struct transport *transport_get(struct remote *remote, const char *url)
929 const char *helper;
930 struct transport *ret = xcalloc(1, sizeof(*ret));
932 ret->progress = isatty(2);
934 if (!remote)
935 die("No remote provided to transport_get()");
937 ret->got_remote_refs = 0;
938 ret->remote = remote;
939 helper = remote->foreign_vcs;
941 if (!url && remote->url)
942 url = remote->url[0];
943 ret->url = url;
945 /* maybe it is a foreign URL? */
946 if (url) {
947 const char *p = url;
949 while (isurlschemechar(p == url, *p))
950 p++;
951 if (!prefixcmp(p, "::"))
952 helper = xstrndup(url, p - url);
955 if (helper) {
956 transport_helper_init(ret, helper);
957 } else if (!prefixcmp(url, "rsync:")) {
958 ret->get_refs_list = get_refs_via_rsync;
959 ret->fetch = fetch_objs_via_rsync;
960 ret->push = rsync_transport_push;
961 ret->smart_options = NULL;
962 } else if (is_local(url) && is_file(url)) {
963 struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
964 ret->data = data;
965 ret->get_refs_list = get_refs_from_bundle;
966 ret->fetch = fetch_refs_from_bundle;
967 ret->disconnect = close_bundle;
968 ret->smart_options = NULL;
969 } else if (!is_url(url)
970 || !prefixcmp(url, "file://")
971 || !prefixcmp(url, "git://")
972 || !prefixcmp(url, "ssh://")
973 || !prefixcmp(url, "git+ssh://")
974 || !prefixcmp(url, "ssh+git://")) {
975 /* These are builtin smart transports. */
976 struct git_transport_data *data = xcalloc(1, sizeof(*data));
977 ret->data = data;
978 ret->set_option = NULL;
979 ret->get_refs_list = get_refs_via_connect;
980 ret->fetch = fetch_refs_via_pack;
981 ret->push_refs = git_transport_push;
982 ret->connect = connect_git;
983 ret->disconnect = disconnect_git;
984 ret->smart_options = &(data->options);
986 data->conn = NULL;
987 data->got_remote_heads = 0;
988 } else {
989 /* Unknown protocol in URL. Pass to external handler. */
990 int len = external_specification_len(url);
991 char *handler = xmalloc(len + 1);
992 handler[len] = 0;
993 strncpy(handler, url, len);
994 transport_helper_init(ret, handler);
997 if (ret->smart_options) {
998 ret->smart_options->thin = 1;
999 ret->smart_options->uploadpack = "git-upload-pack";
1000 if (remote->uploadpack)
1001 ret->smart_options->uploadpack = remote->uploadpack;
1002 ret->smart_options->receivepack = "git-receive-pack";
1003 if (remote->receivepack)
1004 ret->smart_options->receivepack = remote->receivepack;
1007 return ret;
1010 int transport_set_option(struct transport *transport,
1011 const char *name, const char *value)
1013 int git_reports = 1, protocol_reports = 1;
1015 if (transport->smart_options)
1016 git_reports = set_git_option(transport->smart_options,
1017 name, value);
1019 if (transport->set_option)
1020 protocol_reports = transport->set_option(transport, name,
1021 value);
1023 /* If either report is 0, report 0 (success). */
1024 if (!git_reports || !protocol_reports)
1025 return 0;
1026 /* If either reports -1 (invalid value), report -1. */
1027 if ((git_reports == -1) || (protocol_reports == -1))
1028 return -1;
1029 /* Otherwise if both report unknown, report unknown. */
1030 return 1;
1033 void transport_set_verbosity(struct transport *transport, int verbosity,
1034 int force_progress)
1036 if (verbosity >= 2)
1037 transport->verbose = verbosity <= 3 ? verbosity : 3;
1038 if (verbosity < 0)
1039 transport->verbose = -1;
1042 * Rules used to determine whether to report progress (processing aborts
1043 * when a rule is satisfied):
1045 * 1. Report progress, if force_progress is 1 (ie. --progress).
1046 * 2. Don't report progress, if verbosity < 0 (ie. -q/--quiet ).
1047 * 3. Report progress if isatty(2) is 1.
1049 transport->progress = force_progress || (verbosity >= 0 && isatty(2));
1052 int transport_push(struct transport *transport,
1053 int refspec_nr, const char **refspec, int flags,
1054 int *nonfastforward)
1056 *nonfastforward = 0;
1057 transport_verify_remote_names(refspec_nr, refspec);
1059 if (transport->push) {
1060 /* Maybe FIXME. But no important transport uses this case. */
1061 if (flags & TRANSPORT_PUSH_SET_UPSTREAM)
1062 die("This transport does not support using --set-upstream");
1064 return transport->push(transport, refspec_nr, refspec, flags);
1065 } else if (transport->push_refs) {
1066 struct ref *remote_refs =
1067 transport->get_refs_list(transport, 1);
1068 struct ref *local_refs = get_local_heads();
1069 int match_flags = MATCH_REFS_NONE;
1070 int verbose = (transport->verbose > 0);
1071 int quiet = (transport->verbose < 0);
1072 int porcelain = flags & TRANSPORT_PUSH_PORCELAIN;
1073 int pretend = flags & TRANSPORT_PUSH_DRY_RUN;
1074 int push_ret, ret, err;
1076 if (flags & TRANSPORT_PUSH_ALL)
1077 match_flags |= MATCH_REFS_ALL;
1078 if (flags & TRANSPORT_PUSH_MIRROR)
1079 match_flags |= MATCH_REFS_MIRROR;
1081 if (match_refs(local_refs, &remote_refs,
1082 refspec_nr, refspec, match_flags)) {
1083 return -1;
1086 set_ref_status_for_push(remote_refs,
1087 flags & TRANSPORT_PUSH_MIRROR,
1088 flags & TRANSPORT_PUSH_FORCE);
1090 push_ret = transport->push_refs(transport, remote_refs, flags);
1091 err = push_had_errors(remote_refs);
1092 ret = push_ret | err;
1094 if (!quiet || err)
1095 transport_print_push_status(transport->url, remote_refs,
1096 verbose | porcelain, porcelain,
1097 nonfastforward);
1099 if (flags & TRANSPORT_PUSH_SET_UPSTREAM)
1100 set_upstreams(transport, remote_refs, pretend);
1102 if (!(flags & TRANSPORT_PUSH_DRY_RUN)) {
1103 struct ref *ref;
1104 for (ref = remote_refs; ref; ref = ref->next)
1105 transport_update_tracking_ref(transport->remote, ref, verbose);
1108 if (porcelain && !push_ret)
1109 puts("Done");
1110 else if (!quiet && !ret && !transport_refs_pushed(remote_refs))
1111 fprintf(stderr, "Everything up-to-date\n");
1113 return ret;
1115 return 1;
1118 const struct ref *transport_get_remote_refs(struct transport *transport)
1120 if (!transport->got_remote_refs) {
1121 transport->remote_refs = transport->get_refs_list(transport, 0);
1122 transport->got_remote_refs = 1;
1125 return transport->remote_refs;
1128 int transport_fetch_refs(struct transport *transport, struct ref *refs)
1130 int rc;
1131 int nr_heads = 0, nr_alloc = 0, nr_refs = 0;
1132 struct ref **heads = NULL;
1133 struct ref *rm;
1135 for (rm = refs; rm; rm = rm->next) {
1136 nr_refs++;
1137 if (rm->peer_ref &&
1138 !is_null_sha1(rm->old_sha1) &&
1139 !hashcmp(rm->peer_ref->old_sha1, rm->old_sha1))
1140 continue;
1141 ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
1142 heads[nr_heads++] = rm;
1145 if (!nr_heads) {
1147 * When deepening of a shallow repository is requested,
1148 * then local and remote refs are likely to still be equal.
1149 * Just feed them all to the fetch method in that case.
1150 * This condition shouldn't be met in a non-deepening fetch
1151 * (see builtin-fetch.c:quickfetch()).
1153 heads = xmalloc(nr_refs * sizeof(*heads));
1154 for (rm = refs; rm; rm = rm->next)
1155 heads[nr_heads++] = rm;
1158 rc = transport->fetch(transport, nr_heads, heads);
1160 free(heads);
1161 return rc;
1164 void transport_unlock_pack(struct transport *transport)
1166 if (transport->pack_lockfile) {
1167 unlink_or_warn(transport->pack_lockfile);
1168 free(transport->pack_lockfile);
1169 transport->pack_lockfile = NULL;
1173 int transport_connect(struct transport *transport, const char *name,
1174 const char *exec, int fd[2])
1176 if (transport->connect)
1177 return transport->connect(transport, name, exec, fd);
1178 else
1179 die("Operation not supported by protocol");
1182 int transport_disconnect(struct transport *transport)
1184 int ret = 0;
1185 if (transport->disconnect)
1186 ret = transport->disconnect(transport);
1187 free(transport);
1188 return ret;
1192 * Strip username (and password) from an url and return
1193 * it in a newly allocated string.
1195 char *transport_anonymize_url(const char *url)
1197 char *anon_url, *scheme_prefix, *anon_part;
1198 size_t anon_len, prefix_len = 0;
1200 anon_part = strchr(url, '@');
1201 if (is_local(url) || !anon_part)
1202 goto literal_copy;
1204 anon_len = strlen(++anon_part);
1205 scheme_prefix = strstr(url, "://");
1206 if (!scheme_prefix) {
1207 if (!strchr(anon_part, ':'))
1208 /* cannot be "me@there:/path/name" */
1209 goto literal_copy;
1210 } else {
1211 const char *cp;
1212 /* make sure scheme is reasonable */
1213 for (cp = url; cp < scheme_prefix; cp++) {
1214 switch (*cp) {
1215 /* RFC 1738 2.1 */
1216 case '+': case '.': case '-':
1217 break; /* ok */
1218 default:
1219 if (isalnum(*cp))
1220 break;
1221 /* it isn't */
1222 goto literal_copy;
1225 /* @ past the first slash does not count */
1226 cp = strchr(scheme_prefix + 3, '/');
1227 if (cp && cp < anon_part)
1228 goto literal_copy;
1229 prefix_len = scheme_prefix - url + 3;
1231 anon_url = xcalloc(1, 1 + prefix_len + anon_len);
1232 memcpy(anon_url, url, prefix_len);
1233 memcpy(anon_url + prefix_len, anon_part, anon_len);
1234 return anon_url;
1235 literal_copy:
1236 return xstrdup(url);