Git 2.29.3
[git] / transport.c
1 #include "cache.h"
2 #include "config.h"
3 #include "transport.h"
4 #include "run-command.h"
5 #include "pkt-line.h"
6 #include "fetch-pack.h"
7 #include "remote.h"
8 #include "connect.h"
9 #include "send-pack.h"
10 #include "walker.h"
11 #include "bundle.h"
12 #include "dir.h"
13 #include "refs.h"
14 #include "refspec.h"
15 #include "branch.h"
16 #include "url.h"
17 #include "submodule.h"
18 #include "string-list.h"
19 #include "oid-array.h"
20 #include "sigchain.h"
21 #include "transport-internal.h"
22 #include "protocol.h"
23 #include "object-store.h"
24 #include "color.h"
25
26 static int transport_use_color = -1;
27 static char transport_colors[][COLOR_MAXLEN] = {
28         GIT_COLOR_RESET,
29         GIT_COLOR_RED           /* REJECTED */
30 };
31
32 enum color_transport {
33         TRANSPORT_COLOR_RESET = 0,
34         TRANSPORT_COLOR_REJECTED = 1
35 };
36
37 static int transport_color_config(void)
38 {
39         const char *keys[] = {
40                 "color.transport.reset",
41                 "color.transport.rejected"
42         }, *key = "color.transport";
43         char *value;
44         int i;
45         static int initialized;
46
47         if (initialized)
48                 return 0;
49         initialized = 1;
50
51         if (!git_config_get_string(key, &value))
52                 transport_use_color = git_config_colorbool(key, value);
53
54         if (!want_color_stderr(transport_use_color))
55                 return 0;
56
57         for (i = 0; i < ARRAY_SIZE(keys); i++)
58                 if (!git_config_get_string(keys[i], &value)) {
59                         if (!value)
60                                 return config_error_nonbool(keys[i]);
61                         if (color_parse(value, transport_colors[i]) < 0)
62                                 return -1;
63                 }
64
65         return 0;
66 }
67
68 static const char *transport_get_color(enum color_transport ix)
69 {
70         if (want_color_stderr(transport_use_color))
71                 return transport_colors[ix];
72         return "";
73 }
74
75 static void set_upstreams(struct transport *transport, struct ref *refs,
76         int pretend)
77 {
78         struct ref *ref;
79         for (ref = refs; ref; ref = ref->next) {
80                 const char *localname;
81                 const char *tmp;
82                 const char *remotename;
83                 int flag = 0;
84                 /*
85                  * Check suitability for tracking. Must be successful /
86                  * already up-to-date ref create/modify (not delete).
87                  */
88                 if (ref->status != REF_STATUS_OK &&
89                         ref->status != REF_STATUS_UPTODATE)
90                         continue;
91                 if (!ref->peer_ref)
92                         continue;
93                 if (is_null_oid(&ref->new_oid))
94                         continue;
95
96                 /* Follow symbolic refs (mainly for HEAD). */
97                 localname = ref->peer_ref->name;
98                 remotename = ref->name;
99                 tmp = resolve_ref_unsafe(localname, RESOLVE_REF_READING,
100                                          NULL, &flag);
101                 if (tmp && flag & REF_ISSYMREF &&
102                         starts_with(tmp, "refs/heads/"))
103                         localname = tmp;
104
105                 /* Both source and destination must be local branches. */
106                 if (!localname || !starts_with(localname, "refs/heads/"))
107                         continue;
108                 if (!remotename || !starts_with(remotename, "refs/heads/"))
109                         continue;
110
111                 if (!pretend)
112                         install_branch_config(BRANCH_CONFIG_VERBOSE,
113                                 localname + 11, transport->remote->name,
114                                 remotename);
115                 else
116                         printf(_("Would set upstream of '%s' to '%s' of '%s'\n"),
117                                 localname + 11, remotename + 11,
118                                 transport->remote->name);
119         }
120 }
121
122 struct bundle_transport_data {
123         int fd;
124         struct bundle_header header;
125         unsigned get_refs_from_bundle_called : 1;
126 };
127
128 static struct ref *get_refs_from_bundle(struct transport *transport,
129                                         int for_push,
130                                         const struct strvec *ref_prefixes)
131 {
132         struct bundle_transport_data *data = transport->data;
133         struct ref *result = NULL;
134         int i;
135
136         if (for_push)
137                 return NULL;
138
139         data->get_refs_from_bundle_called = 1;
140
141         if (data->fd > 0)
142                 close(data->fd);
143         data->fd = read_bundle_header(transport->url, &data->header);
144         if (data->fd < 0)
145                 die(_("could not read bundle '%s'"), transport->url);
146
147         transport->hash_algo = data->header.hash_algo;
148
149         for (i = 0; i < data->header.references.nr; i++) {
150                 struct ref_list_entry *e = data->header.references.list + i;
151                 struct ref *ref = alloc_ref(e->name);
152                 oidcpy(&ref->old_oid, &e->oid);
153                 ref->next = result;
154                 result = ref;
155         }
156         return result;
157 }
158
159 static int fetch_refs_from_bundle(struct transport *transport,
160                                int nr_heads, struct ref **to_fetch)
161 {
162         struct bundle_transport_data *data = transport->data;
163         int ret;
164
165         if (!data->get_refs_from_bundle_called)
166                 get_refs_from_bundle(transport, 0, NULL);
167         ret = unbundle(the_repository, &data->header, data->fd,
168                            transport->progress ? BUNDLE_VERBOSE : 0);
169         transport->hash_algo = data->header.hash_algo;
170         return ret;
171 }
172
173 static int close_bundle(struct transport *transport)
174 {
175         struct bundle_transport_data *data = transport->data;
176         if (data->fd > 0)
177                 close(data->fd);
178         free(data);
179         return 0;
180 }
181
182 struct git_transport_data {
183         struct git_transport_options options;
184         struct child_process *conn;
185         int fd[2];
186         unsigned got_remote_heads : 1;
187         enum protocol_version version;
188         struct oid_array extra_have;
189         struct oid_array shallow;
190 };
191
192 static int set_git_option(struct git_transport_options *opts,
193                           const char *name, const char *value)
194 {
195         if (!strcmp(name, TRANS_OPT_UPLOADPACK)) {
196                 opts->uploadpack = value;
197                 return 0;
198         } else if (!strcmp(name, TRANS_OPT_RECEIVEPACK)) {
199                 opts->receivepack = value;
200                 return 0;
201         } else if (!strcmp(name, TRANS_OPT_THIN)) {
202                 opts->thin = !!value;
203                 return 0;
204         } else if (!strcmp(name, TRANS_OPT_FOLLOWTAGS)) {
205                 opts->followtags = !!value;
206                 return 0;
207         } else if (!strcmp(name, TRANS_OPT_KEEP)) {
208                 opts->keep = !!value;
209                 return 0;
210         } else if (!strcmp(name, TRANS_OPT_UPDATE_SHALLOW)) {
211                 opts->update_shallow = !!value;
212                 return 0;
213         } else if (!strcmp(name, TRANS_OPT_DEPTH)) {
214                 if (!value)
215                         opts->depth = 0;
216                 else {
217                         char *end;
218                         opts->depth = strtol(value, &end, 0);
219                         if (*end)
220                                 die(_("transport: invalid depth option '%s'"), value);
221                 }
222                 return 0;
223         } else if (!strcmp(name, TRANS_OPT_DEEPEN_SINCE)) {
224                 opts->deepen_since = value;
225                 return 0;
226         } else if (!strcmp(name, TRANS_OPT_DEEPEN_NOT)) {
227                 opts->deepen_not = (const struct string_list *)value;
228                 return 0;
229         } else if (!strcmp(name, TRANS_OPT_DEEPEN_RELATIVE)) {
230                 opts->deepen_relative = !!value;
231                 return 0;
232         } else if (!strcmp(name, TRANS_OPT_FROM_PROMISOR)) {
233                 opts->from_promisor = !!value;
234                 return 0;
235         } else if (!strcmp(name, TRANS_OPT_LIST_OBJECTS_FILTER)) {
236                 list_objects_filter_die_if_populated(&opts->filter_options);
237                 parse_list_objects_filter(&opts->filter_options, value);
238                 return 0;
239         }
240         return 1;
241 }
242
243 static int connect_setup(struct transport *transport, int for_push)
244 {
245         struct git_transport_data *data = transport->data;
246         int flags = transport->verbose > 0 ? CONNECT_VERBOSE : 0;
247
248         if (data->conn)
249                 return 0;
250
251         switch (transport->family) {
252         case TRANSPORT_FAMILY_ALL: break;
253         case TRANSPORT_FAMILY_IPV4: flags |= CONNECT_IPV4; break;
254         case TRANSPORT_FAMILY_IPV6: flags |= CONNECT_IPV6; break;
255         }
256
257         data->conn = git_connect(data->fd, transport->url,
258                                  for_push ? data->options.receivepack :
259                                  data->options.uploadpack,
260                                  flags);
261
262         return 0;
263 }
264
265 static void die_if_server_options(struct transport *transport)
266 {
267         if (!transport->server_options || !transport->server_options->nr)
268                 return;
269         advise(_("see protocol.version in 'git help config' for more details"));
270         die(_("server options require protocol version 2 or later"));
271 }
272
273 /*
274  * Obtains the protocol version from the transport and writes it to
275  * transport->data->version, first connecting if not already connected.
276  *
277  * If the protocol version is one that allows skipping the listing of remote
278  * refs, and must_list_refs is 0, the listing of remote refs is skipped and
279  * this function returns NULL. Otherwise, this function returns the list of
280  * remote refs.
281  */
282 static struct ref *handshake(struct transport *transport, int for_push,
283                              const struct strvec *ref_prefixes,
284                              int must_list_refs)
285 {
286         struct git_transport_data *data = transport->data;
287         struct ref *refs = NULL;
288         struct packet_reader reader;
289
290         connect_setup(transport, for_push);
291
292         packet_reader_init(&reader, data->fd[0], NULL, 0,
293                            PACKET_READ_CHOMP_NEWLINE |
294                            PACKET_READ_GENTLE_ON_EOF |
295                            PACKET_READ_DIE_ON_ERR_PACKET);
296
297         data->version = discover_version(&reader);
298         switch (data->version) {
299         case protocol_v2:
300                 if (must_list_refs)
301                         get_remote_refs(data->fd[1], &reader, &refs, for_push,
302                                         ref_prefixes,
303                                         transport->server_options,
304                                         transport->stateless_rpc);
305                 break;
306         case protocol_v1:
307         case protocol_v0:
308                 die_if_server_options(transport);
309                 get_remote_heads(&reader, &refs,
310                                  for_push ? REF_NORMAL : 0,
311                                  &data->extra_have,
312                                  &data->shallow);
313                 break;
314         case protocol_unknown_version:
315                 BUG("unknown protocol version");
316         }
317         data->got_remote_heads = 1;
318         transport->hash_algo = reader.hash_algo;
319
320         if (reader.line_peeked)
321                 BUG("buffer must be empty at the end of handshake()");
322
323         return refs;
324 }
325
326 static struct ref *get_refs_via_connect(struct transport *transport, int for_push,
327                                         const struct strvec *ref_prefixes)
328 {
329         return handshake(transport, for_push, ref_prefixes, 1);
330 }
331
332 static int fetch_refs_via_pack(struct transport *transport,
333                                int nr_heads, struct ref **to_fetch)
334 {
335         int ret = 0;
336         struct git_transport_data *data = transport->data;
337         struct ref *refs = NULL;
338         struct fetch_pack_args args;
339         struct ref *refs_tmp = NULL;
340
341         memset(&args, 0, sizeof(args));
342         args.uploadpack = data->options.uploadpack;
343         args.keep_pack = data->options.keep;
344         args.lock_pack = 1;
345         args.use_thin_pack = data->options.thin;
346         args.include_tag = data->options.followtags;
347         args.verbose = (transport->verbose > 1);
348         args.quiet = (transport->verbose < 0);
349         args.no_progress = !transport->progress;
350         args.depth = data->options.depth;
351         args.deepen_since = data->options.deepen_since;
352         args.deepen_not = data->options.deepen_not;
353         args.deepen_relative = data->options.deepen_relative;
354         args.check_self_contained_and_connected =
355                 data->options.check_self_contained_and_connected;
356         args.cloning = transport->cloning;
357         args.update_shallow = data->options.update_shallow;
358         args.from_promisor = data->options.from_promisor;
359         args.filter_options = data->options.filter_options;
360         args.stateless_rpc = transport->stateless_rpc;
361         args.server_options = transport->server_options;
362         args.negotiation_tips = data->options.negotiation_tips;
363
364         if (!data->got_remote_heads) {
365                 int i;
366                 int must_list_refs = 0;
367                 for (i = 0; i < nr_heads; i++) {
368                         if (!to_fetch[i]->exact_oid) {
369                                 must_list_refs = 1;
370                                 break;
371                         }
372                 }
373                 refs_tmp = handshake(transport, 0, NULL, must_list_refs);
374         }
375
376         if (data->version == protocol_unknown_version)
377                 BUG("unknown protocol version");
378         else if (data->version <= protocol_v1)
379                 die_if_server_options(transport);
380
381         refs = fetch_pack(&args, data->fd,
382                           refs_tmp ? refs_tmp : transport->remote_refs,
383                           to_fetch, nr_heads, &data->shallow,
384                           &transport->pack_lockfiles, data->version);
385
386         close(data->fd[0]);
387         close(data->fd[1]);
388         if (finish_connect(data->conn))
389                 ret = -1;
390         data->conn = NULL;
391         data->got_remote_heads = 0;
392         data->options.self_contained_and_connected =
393                 args.self_contained_and_connected;
394         data->options.connectivity_checked = args.connectivity_checked;
395
396         if (refs == NULL)
397                 ret = -1;
398         if (report_unmatched_refs(to_fetch, nr_heads))
399                 ret = -1;
400
401         free_refs(refs_tmp);
402         free_refs(refs);
403         return ret;
404 }
405
406 static int push_had_errors(struct ref *ref)
407 {
408         for (; ref; ref = ref->next) {
409                 switch (ref->status) {
410                 case REF_STATUS_NONE:
411                 case REF_STATUS_UPTODATE:
412                 case REF_STATUS_OK:
413                         break;
414                 default:
415                         return 1;
416                 }
417         }
418         return 0;
419 }
420
421 int transport_refs_pushed(struct ref *ref)
422 {
423         for (; ref; ref = ref->next) {
424                 switch(ref->status) {
425                 case REF_STATUS_NONE:
426                 case REF_STATUS_UPTODATE:
427                         break;
428                 default:
429                         return 1;
430                 }
431         }
432         return 0;
433 }
434
435 static void update_one_tracking_ref(struct remote *remote, char *refname,
436                                     struct object_id *new_oid, int deletion,
437                                     int verbose)
438 {
439         struct refspec_item rs;
440
441         memset(&rs, 0, sizeof(rs));
442         rs.src = refname;
443         rs.dst = NULL;
444
445         if (!remote_find_tracking(remote, &rs)) {
446                 if (verbose)
447                         fprintf(stderr, "updating local tracking ref '%s'\n", rs.dst);
448                 if (deletion)
449                         delete_ref(NULL, rs.dst, NULL, 0);
450                 else
451                         update_ref("update by push", rs.dst, new_oid,
452                                    NULL, 0, 0);
453                 free(rs.dst);
454         }
455 }
456
457 void transport_update_tracking_ref(struct remote *remote, struct ref *ref, int verbose)
458 {
459         char *refname;
460         struct object_id *new_oid;
461         struct ref_push_report *report;
462
463         if (ref->status != REF_STATUS_OK && ref->status != REF_STATUS_UPTODATE)
464                 return;
465
466         report = ref->report;
467         if (!report)
468                 update_one_tracking_ref(remote, ref->name, &ref->new_oid,
469                                         ref->deletion, verbose);
470         else
471                 for (; report; report = report->next) {
472                         refname = report->ref_name ? (char *)report->ref_name : ref->name;
473                         new_oid = report->new_oid ? report->new_oid : &ref->new_oid;
474                         update_one_tracking_ref(remote, refname, new_oid,
475                                                 is_null_oid(new_oid), verbose);
476                 }
477 }
478
479 static void print_ref_status(char flag, const char *summary,
480                              struct ref *to, struct ref *from, const char *msg,
481                              struct ref_push_report *report,
482                              int porcelain, int summary_width)
483 {
484         const char *to_name;
485
486         if (report && report->ref_name)
487                 to_name = report->ref_name;
488         else
489                 to_name = to->name;
490
491         if (porcelain) {
492                 if (from)
493                         fprintf(stdout, "%c\t%s:%s\t", flag, from->name, to_name);
494                 else
495                         fprintf(stdout, "%c\t:%s\t", flag, to_name);
496                 if (msg)
497                         fprintf(stdout, "%s (%s)\n", summary, msg);
498                 else
499                         fprintf(stdout, "%s\n", summary);
500         } else {
501                 const char *red = "", *reset = "";
502                 if (push_had_errors(to)) {
503                         red = transport_get_color(TRANSPORT_COLOR_REJECTED);
504                         reset = transport_get_color(TRANSPORT_COLOR_RESET);
505                 }
506                 fprintf(stderr, " %s%c %-*s%s ", red, flag, summary_width,
507                         summary, reset);
508                 if (from)
509                         fprintf(stderr, "%s -> %s",
510                                 prettify_refname(from->name),
511                                 prettify_refname(to_name));
512                 else
513                         fputs(prettify_refname(to_name), stderr);
514                 if (msg) {
515                         fputs(" (", stderr);
516                         fputs(msg, stderr);
517                         fputc(')', stderr);
518                 }
519                 fputc('\n', stderr);
520         }
521 }
522
523 static void print_ok_ref_status(struct ref *ref,
524                                 struct ref_push_report *report,
525                                 int porcelain, int summary_width)
526 {
527         struct object_id *old_oid;
528         struct object_id *new_oid;
529         const char *ref_name;
530         int forced_update;
531
532         if (report && report->old_oid)
533                 old_oid = report->old_oid;
534         else
535                 old_oid = &ref->old_oid;
536         if (report && report->new_oid)
537                 new_oid = report->new_oid;
538         else
539                 new_oid = &ref->new_oid;
540         if (report && report->forced_update)
541                 forced_update = report->forced_update;
542         else
543                 forced_update = ref->forced_update;
544         if (report && report->ref_name)
545                 ref_name = report->ref_name;
546         else
547                 ref_name = ref->name;
548
549         if (ref->deletion)
550                 print_ref_status('-', "[deleted]", ref, NULL, NULL,
551                                  report, porcelain, summary_width);
552         else if (is_null_oid(old_oid))
553                 print_ref_status('*',
554                                  (starts_with(ref_name, "refs/tags/")
555                                   ? "[new tag]"
556                                   : (starts_with(ref_name, "refs/heads/")
557                                      ? "[new branch]"
558                                      : "[new reference]")),
559                                  ref, ref->peer_ref, NULL,
560                                  report, porcelain, summary_width);
561         else {
562                 struct strbuf quickref = STRBUF_INIT;
563                 char type;
564                 const char *msg;
565
566                 strbuf_add_unique_abbrev(&quickref, old_oid,
567                                          DEFAULT_ABBREV);
568                 if (forced_update) {
569                         strbuf_addstr(&quickref, "...");
570                         type = '+';
571                         msg = "forced update";
572                 } else {
573                         strbuf_addstr(&quickref, "..");
574                         type = ' ';
575                         msg = NULL;
576                 }
577                 strbuf_add_unique_abbrev(&quickref, new_oid,
578                                          DEFAULT_ABBREV);
579
580                 print_ref_status(type, quickref.buf, ref, ref->peer_ref, msg,
581                                  report, porcelain, summary_width);
582                 strbuf_release(&quickref);
583         }
584 }
585
586 static int print_one_push_report(struct ref *ref, const char *dest, int count,
587                                  struct ref_push_report *report,
588                                  int porcelain, int summary_width)
589 {
590         if (!count) {
591                 char *url = transport_anonymize_url(dest);
592                 fprintf(porcelain ? stdout : stderr, "To %s\n", url);
593                 free(url);
594         }
595
596         switch(ref->status) {
597         case REF_STATUS_NONE:
598                 print_ref_status('X', "[no match]", ref, NULL, NULL,
599                                  report, porcelain, summary_width);
600                 break;
601         case REF_STATUS_REJECT_NODELETE:
602                 print_ref_status('!', "[rejected]", ref, NULL,
603                                  "remote does not support deleting refs",
604                                  report, porcelain, summary_width);
605                 break;
606         case REF_STATUS_UPTODATE:
607                 print_ref_status('=', "[up to date]", ref,
608                                  ref->peer_ref, NULL,
609                                  report, porcelain, summary_width);
610                 break;
611         case REF_STATUS_REJECT_NONFASTFORWARD:
612                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
613                                  "non-fast-forward",
614                                  report, porcelain, summary_width);
615                 break;
616         case REF_STATUS_REJECT_ALREADY_EXISTS:
617                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
618                                  "already exists",
619                                  report, porcelain, summary_width);
620                 break;
621         case REF_STATUS_REJECT_FETCH_FIRST:
622                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
623                                  "fetch first",
624                                  report, porcelain, summary_width);
625                 break;
626         case REF_STATUS_REJECT_NEEDS_FORCE:
627                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
628                                  "needs force",
629                                  report, porcelain, summary_width);
630                 break;
631         case REF_STATUS_REJECT_STALE:
632                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
633                                  "stale info",
634                                  report, porcelain, summary_width);
635                 break;
636         case REF_STATUS_REJECT_SHALLOW:
637                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
638                                  "new shallow roots not allowed",
639                                  report, porcelain, summary_width);
640                 break;
641         case REF_STATUS_REMOTE_REJECT:
642                 print_ref_status('!', "[remote rejected]", ref,
643                                  ref->deletion ? NULL : ref->peer_ref,
644                                  ref->remote_status,
645                                  report, porcelain, summary_width);
646                 break;
647         case REF_STATUS_EXPECTING_REPORT:
648                 print_ref_status('!', "[remote failure]", ref,
649                                  ref->deletion ? NULL : ref->peer_ref,
650                                  "remote failed to report status",
651                                  report, porcelain, summary_width);
652                 break;
653         case REF_STATUS_ATOMIC_PUSH_FAILED:
654                 print_ref_status('!', "[rejected]", ref, ref->peer_ref,
655                                  "atomic push failed",
656                                  report, porcelain, summary_width);
657                 break;
658         case REF_STATUS_OK:
659                 print_ok_ref_status(ref, report, porcelain, summary_width);
660                 break;
661         }
662
663         return 1;
664 }
665
666 static int print_one_push_status(struct ref *ref, const char *dest, int count,
667                                  int porcelain, int summary_width)
668 {
669         struct ref_push_report *report;
670         int n = 0;
671
672         if (!ref->report)
673                 return print_one_push_report(ref, dest, count,
674                                              NULL, porcelain, summary_width);
675
676         for (report = ref->report; report; report = report->next)
677                 print_one_push_report(ref, dest, count + n++,
678                                       report, porcelain, summary_width);
679         return n;
680 }
681
682 static int measure_abbrev(const struct object_id *oid, int sofar)
683 {
684         char hex[GIT_MAX_HEXSZ + 1];
685         int w = find_unique_abbrev_r(hex, oid, DEFAULT_ABBREV);
686
687         return (w < sofar) ? sofar : w;
688 }
689
690 int transport_summary_width(const struct ref *refs)
691 {
692         int maxw = -1;
693
694         for (; refs; refs = refs->next) {
695                 maxw = measure_abbrev(&refs->old_oid, maxw);
696                 maxw = measure_abbrev(&refs->new_oid, maxw);
697         }
698         if (maxw < 0)
699                 maxw = FALLBACK_DEFAULT_ABBREV;
700         return (2 * maxw + 3);
701 }
702
703 void transport_print_push_status(const char *dest, struct ref *refs,
704                                   int verbose, int porcelain, unsigned int *reject_reasons)
705 {
706         struct ref *ref;
707         int n = 0;
708         char *head;
709         int summary_width = transport_summary_width(refs);
710
711         if (transport_color_config() < 0)
712                 warning(_("could not parse transport.color.* config"));
713
714         head = resolve_refdup("HEAD", RESOLVE_REF_READING, NULL, NULL);
715
716         if (verbose) {
717                 for (ref = refs; ref; ref = ref->next)
718                         if (ref->status == REF_STATUS_UPTODATE)
719                                 n += print_one_push_status(ref, dest, n,
720                                                            porcelain, summary_width);
721         }
722
723         for (ref = refs; ref; ref = ref->next)
724                 if (ref->status == REF_STATUS_OK)
725                         n += print_one_push_status(ref, dest, n,
726                                                    porcelain, summary_width);
727
728         *reject_reasons = 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,
734                                                    porcelain, summary_width);
735                 if (ref->status == REF_STATUS_REJECT_NONFASTFORWARD) {
736                         if (head != NULL && !strcmp(head, ref->name))
737                                 *reject_reasons |= REJECT_NON_FF_HEAD;
738                         else
739                                 *reject_reasons |= REJECT_NON_FF_OTHER;
740                 } else if (ref->status == REF_STATUS_REJECT_ALREADY_EXISTS) {
741                         *reject_reasons |= REJECT_ALREADY_EXISTS;
742                 } else if (ref->status == REF_STATUS_REJECT_FETCH_FIRST) {
743                         *reject_reasons |= REJECT_FETCH_FIRST;
744                 } else if (ref->status == REF_STATUS_REJECT_NEEDS_FORCE) {
745                         *reject_reasons |= REJECT_NEEDS_FORCE;
746                 }
747         }
748         free(head);
749 }
750
751 static int git_transport_push(struct transport *transport, struct ref *remote_refs, int flags)
752 {
753         struct git_transport_data *data = transport->data;
754         struct send_pack_args args;
755         int ret = 0;
756
757         if (transport_color_config() < 0)
758                 return -1;
759
760         if (!data->got_remote_heads)
761                 get_refs_via_connect(transport, 1, NULL);
762
763         memset(&args, 0, sizeof(args));
764         args.send_mirror = !!(flags & TRANSPORT_PUSH_MIRROR);
765         args.force_update = !!(flags & TRANSPORT_PUSH_FORCE);
766         args.use_thin_pack = data->options.thin;
767         args.verbose = (transport->verbose > 0);
768         args.quiet = (transport->verbose < 0);
769         args.progress = transport->progress;
770         args.dry_run = !!(flags & TRANSPORT_PUSH_DRY_RUN);
771         args.porcelain = !!(flags & TRANSPORT_PUSH_PORCELAIN);
772         args.atomic = !!(flags & TRANSPORT_PUSH_ATOMIC);
773         args.push_options = transport->push_options;
774         args.url = transport->url;
775
776         if (flags & TRANSPORT_PUSH_CERT_ALWAYS)
777                 args.push_cert = SEND_PACK_PUSH_CERT_ALWAYS;
778         else if (flags & TRANSPORT_PUSH_CERT_IF_ASKED)
779                 args.push_cert = SEND_PACK_PUSH_CERT_IF_ASKED;
780         else
781                 args.push_cert = SEND_PACK_PUSH_CERT_NEVER;
782
783         switch (data->version) {
784         case protocol_v2:
785                 die(_("support for protocol v2 not implemented yet"));
786                 break;
787         case protocol_v1:
788         case protocol_v0:
789                 ret = send_pack(&args, data->fd, data->conn, remote_refs,
790                                 &data->extra_have);
791                 break;
792         case protocol_unknown_version:
793                 BUG("unknown protocol version");
794         }
795
796         close(data->fd[1]);
797         close(data->fd[0]);
798         /*
799          * Atomic push may abort the connection early and close the pipe,
800          * which may cause an error for `finish_connect()`. Ignore this error
801          * for atomic git-push.
802          */
803         if (ret || args.atomic)
804                 finish_connect(data->conn);
805         else
806                 ret = finish_connect(data->conn);
807         data->conn = NULL;
808         data->got_remote_heads = 0;
809
810         return ret;
811 }
812
813 static int connect_git(struct transport *transport, const char *name,
814                        const char *executable, int fd[2])
815 {
816         struct git_transport_data *data = transport->data;
817         data->conn = git_connect(data->fd, transport->url,
818                                  executable, 0);
819         fd[0] = data->fd[0];
820         fd[1] = data->fd[1];
821         return 0;
822 }
823
824 static int disconnect_git(struct transport *transport)
825 {
826         struct git_transport_data *data = transport->data;
827         if (data->conn) {
828                 if (data->got_remote_heads && !transport->stateless_rpc)
829                         packet_flush(data->fd[1]);
830                 close(data->fd[0]);
831                 close(data->fd[1]);
832                 finish_connect(data->conn);
833         }
834
835         free(data);
836         return 0;
837 }
838
839 static struct transport_vtable taken_over_vtable = {
840         NULL,
841         get_refs_via_connect,
842         fetch_refs_via_pack,
843         git_transport_push,
844         NULL,
845         disconnect_git
846 };
847
848 void transport_take_over(struct transport *transport,
849                          struct child_process *child)
850 {
851         struct git_transport_data *data;
852
853         if (!transport->smart_options)
854                 BUG("taking over transport requires non-NULL "
855                     "smart_options field.");
856
857         data = xcalloc(1, sizeof(*data));
858         data->options = *transport->smart_options;
859         data->conn = child;
860         data->fd[0] = data->conn->out;
861         data->fd[1] = data->conn->in;
862         data->got_remote_heads = 0;
863         transport->data = data;
864
865         transport->vtable = &taken_over_vtable;
866         transport->smart_options = &(data->options);
867
868         transport->cannot_reuse = 1;
869 }
870
871 static int is_file(const char *url)
872 {
873         struct stat buf;
874         if (stat(url, &buf))
875                 return 0;
876         return S_ISREG(buf.st_mode);
877 }
878
879 static int external_specification_len(const char *url)
880 {
881         return strchr(url, ':') - url;
882 }
883
884 static const struct string_list *protocol_whitelist(void)
885 {
886         static int enabled = -1;
887         static struct string_list allowed = STRING_LIST_INIT_DUP;
888
889         if (enabled < 0) {
890                 const char *v = getenv("GIT_ALLOW_PROTOCOL");
891                 if (v) {
892                         string_list_split(&allowed, v, ':', -1);
893                         string_list_sort(&allowed);
894                         enabled = 1;
895                 } else {
896                         enabled = 0;
897                 }
898         }
899
900         return enabled ? &allowed : NULL;
901 }
902
903 enum protocol_allow_config {
904         PROTOCOL_ALLOW_NEVER = 0,
905         PROTOCOL_ALLOW_USER_ONLY,
906         PROTOCOL_ALLOW_ALWAYS
907 };
908
909 static enum protocol_allow_config parse_protocol_config(const char *key,
910                                                         const char *value)
911 {
912         if (!strcasecmp(value, "always"))
913                 return PROTOCOL_ALLOW_ALWAYS;
914         else if (!strcasecmp(value, "never"))
915                 return PROTOCOL_ALLOW_NEVER;
916         else if (!strcasecmp(value, "user"))
917                 return PROTOCOL_ALLOW_USER_ONLY;
918
919         die(_("unknown value for config '%s': %s"), key, value);
920 }
921
922 static enum protocol_allow_config get_protocol_config(const char *type)
923 {
924         char *key = xstrfmt("protocol.%s.allow", type);
925         char *value;
926
927         /* first check the per-protocol config */
928         if (!git_config_get_string(key, &value)) {
929                 enum protocol_allow_config ret =
930                         parse_protocol_config(key, value);
931                 free(key);
932                 free(value);
933                 return ret;
934         }
935         free(key);
936
937         /* if defined, fallback to user-defined default for unknown protocols */
938         if (!git_config_get_string("protocol.allow", &value)) {
939                 enum protocol_allow_config ret =
940                         parse_protocol_config("protocol.allow", value);
941                 free(value);
942                 return ret;
943         }
944
945         /* fallback to built-in defaults */
946         /* known safe */
947         if (!strcmp(type, "http") ||
948             !strcmp(type, "https") ||
949             !strcmp(type, "git") ||
950             !strcmp(type, "ssh") ||
951             !strcmp(type, "file"))
952                 return PROTOCOL_ALLOW_ALWAYS;
953
954         /* known scary; err on the side of caution */
955         if (!strcmp(type, "ext"))
956                 return PROTOCOL_ALLOW_NEVER;
957
958         /* unknown; by default let them be used only directly by the user */
959         return PROTOCOL_ALLOW_USER_ONLY;
960 }
961
962 int is_transport_allowed(const char *type, int from_user)
963 {
964         const struct string_list *whitelist = protocol_whitelist();
965         if (whitelist)
966                 return string_list_has_string(whitelist, type);
967
968         switch (get_protocol_config(type)) {
969         case PROTOCOL_ALLOW_ALWAYS:
970                 return 1;
971         case PROTOCOL_ALLOW_NEVER:
972                 return 0;
973         case PROTOCOL_ALLOW_USER_ONLY:
974                 if (from_user < 0)
975                         from_user = git_env_bool("GIT_PROTOCOL_FROM_USER", 1);
976                 return from_user;
977         }
978
979         BUG("invalid protocol_allow_config type");
980 }
981
982 void transport_check_allowed(const char *type)
983 {
984         if (!is_transport_allowed(type, -1))
985                 die(_("transport '%s' not allowed"), type);
986 }
987
988 static struct transport_vtable bundle_vtable = {
989         NULL,
990         get_refs_from_bundle,
991         fetch_refs_from_bundle,
992         NULL,
993         NULL,
994         close_bundle
995 };
996
997 static struct transport_vtable builtin_smart_vtable = {
998         NULL,
999         get_refs_via_connect,
1000         fetch_refs_via_pack,
1001         git_transport_push,
1002         connect_git,
1003         disconnect_git
1004 };
1005
1006 struct transport *transport_get(struct remote *remote, const char *url)
1007 {
1008         const char *helper;
1009         struct transport *ret = xcalloc(1, sizeof(*ret));
1010
1011         ret->progress = isatty(2);
1012         string_list_init(&ret->pack_lockfiles, 1);
1013
1014         if (!remote)
1015                 BUG("No remote provided to transport_get()");
1016
1017         ret->got_remote_refs = 0;
1018         ret->remote = remote;
1019         helper = remote->foreign_vcs;
1020
1021         if (!url && remote->url)
1022                 url = remote->url[0];
1023         ret->url = url;
1024
1025         /* maybe it is a foreign URL? */
1026         if (url) {
1027                 const char *p = url;
1028
1029                 while (is_urlschemechar(p == url, *p))
1030                         p++;
1031                 if (starts_with(p, "::"))
1032                         helper = xstrndup(url, p - url);
1033         }
1034
1035         if (helper) {
1036                 transport_helper_init(ret, helper);
1037         } else if (starts_with(url, "rsync:")) {
1038                 die(_("git-over-rsync is no longer supported"));
1039         } else if (url_is_local_not_ssh(url) && is_file(url) && is_bundle(url, 1)) {
1040                 struct bundle_transport_data *data = xcalloc(1, sizeof(*data));
1041                 transport_check_allowed("file");
1042                 ret->data = data;
1043                 ret->vtable = &bundle_vtable;
1044                 ret->smart_options = NULL;
1045         } else if (!is_url(url)
1046                 || starts_with(url, "file://")
1047                 || starts_with(url, "git://")
1048                 || starts_with(url, "ssh://")
1049                 || starts_with(url, "git+ssh://") /* deprecated - do not use */
1050                 || starts_with(url, "ssh+git://") /* deprecated - do not use */
1051                 ) {
1052                 /*
1053                  * These are builtin smart transports; "allowed" transports
1054                  * will be checked individually in git_connect.
1055                  */
1056                 struct git_transport_data *data = xcalloc(1, sizeof(*data));
1057                 ret->data = data;
1058                 ret->vtable = &builtin_smart_vtable;
1059                 ret->smart_options = &(data->options);
1060
1061                 data->conn = NULL;
1062                 data->got_remote_heads = 0;
1063         } else {
1064                 /* Unknown protocol in URL. Pass to external handler. */
1065                 int len = external_specification_len(url);
1066                 char *handler = xmemdupz(url, len);
1067                 transport_helper_init(ret, handler);
1068         }
1069
1070         if (ret->smart_options) {
1071                 ret->smart_options->thin = 1;
1072                 ret->smart_options->uploadpack = "git-upload-pack";
1073                 if (remote->uploadpack)
1074                         ret->smart_options->uploadpack = remote->uploadpack;
1075                 ret->smart_options->receivepack = "git-receive-pack";
1076                 if (remote->receivepack)
1077                         ret->smart_options->receivepack = remote->receivepack;
1078         }
1079
1080         ret->hash_algo = &hash_algos[GIT_HASH_SHA1];
1081
1082         return ret;
1083 }
1084
1085 const struct git_hash_algo *transport_get_hash_algo(struct transport *transport)
1086 {
1087         return transport->hash_algo;
1088 }
1089
1090 int transport_set_option(struct transport *transport,
1091                          const char *name, const char *value)
1092 {
1093         int git_reports = 1, protocol_reports = 1;
1094
1095         if (transport->smart_options)
1096                 git_reports = set_git_option(transport->smart_options,
1097                                              name, value);
1098
1099         if (transport->vtable->set_option)
1100                 protocol_reports = transport->vtable->set_option(transport,
1101                                                                  name, value);
1102
1103         /* If either report is 0, report 0 (success). */
1104         if (!git_reports || !protocol_reports)
1105                 return 0;
1106         /* If either reports -1 (invalid value), report -1. */
1107         if ((git_reports == -1) || (protocol_reports == -1))
1108                 return -1;
1109         /* Otherwise if both report unknown, report unknown. */
1110         return 1;
1111 }
1112
1113 void transport_set_verbosity(struct transport *transport, int verbosity,
1114         int force_progress)
1115 {
1116         if (verbosity >= 1)
1117                 transport->verbose = verbosity <= 3 ? verbosity : 3;
1118         if (verbosity < 0)
1119                 transport->verbose = -1;
1120
1121         /**
1122          * Rules used to determine whether to report progress (processing aborts
1123          * when a rule is satisfied):
1124          *
1125          *   . Report progress, if force_progress is 1 (ie. --progress).
1126          *   . Don't report progress, if force_progress is 0 (ie. --no-progress).
1127          *   . Don't report progress, if verbosity < 0 (ie. -q/--quiet ).
1128          *   . Report progress if isatty(2) is 1.
1129          **/
1130         if (force_progress >= 0)
1131                 transport->progress = !!force_progress;
1132         else
1133                 transport->progress = verbosity >= 0 && isatty(2);
1134 }
1135
1136 static void die_with_unpushed_submodules(struct string_list *needs_pushing)
1137 {
1138         int i;
1139
1140         fprintf(stderr, _("The following submodule paths contain changes that can\n"
1141                         "not be found on any remote:\n"));
1142         for (i = 0; i < needs_pushing->nr; i++)
1143                 fprintf(stderr, "  %s\n", needs_pushing->items[i].string);
1144         fprintf(stderr, _("\nPlease try\n\n"
1145                           "     git push --recurse-submodules=on-demand\n\n"
1146                           "or cd to the path and use\n\n"
1147                           "     git push\n\n"
1148                           "to push them to a remote.\n\n"));
1149
1150         string_list_clear(needs_pushing, 0);
1151
1152         die(_("Aborting."));
1153 }
1154
1155 static int run_pre_push_hook(struct transport *transport,
1156                              struct ref *remote_refs)
1157 {
1158         int ret = 0, x;
1159         struct ref *r;
1160         struct child_process proc = CHILD_PROCESS_INIT;
1161         struct strbuf buf;
1162         const char *argv[4];
1163
1164         if (!(argv[0] = find_hook("pre-push")))
1165                 return 0;
1166
1167         argv[1] = transport->remote->name;
1168         argv[2] = transport->url;
1169         argv[3] = NULL;
1170
1171         proc.argv = argv;
1172         proc.in = -1;
1173         proc.trace2_hook_name = "pre-push";
1174
1175         if (start_command(&proc)) {
1176                 finish_command(&proc);
1177                 return -1;
1178         }
1179
1180         sigchain_push(SIGPIPE, SIG_IGN);
1181
1182         strbuf_init(&buf, 256);
1183
1184         for (r = remote_refs; r; r = r->next) {
1185                 if (!r->peer_ref) continue;
1186                 if (r->status == REF_STATUS_REJECT_NONFASTFORWARD) continue;
1187                 if (r->status == REF_STATUS_REJECT_STALE) continue;
1188                 if (r->status == REF_STATUS_UPTODATE) continue;
1189
1190                 strbuf_reset(&buf);
1191                 strbuf_addf( &buf, "%s %s %s %s\n",
1192                          r->peer_ref->name, oid_to_hex(&r->new_oid),
1193                          r->name, oid_to_hex(&r->old_oid));
1194
1195                 if (write_in_full(proc.in, buf.buf, buf.len) < 0) {
1196                         /* We do not mind if a hook does not read all refs. */
1197                         if (errno != EPIPE)
1198                                 ret = -1;
1199                         break;
1200                 }
1201         }
1202
1203         strbuf_release(&buf);
1204
1205         x = close(proc.in);
1206         if (!ret)
1207                 ret = x;
1208
1209         sigchain_pop(SIGPIPE);
1210
1211         x = finish_command(&proc);
1212         if (!ret)
1213                 ret = x;
1214
1215         return ret;
1216 }
1217
1218 int transport_push(struct repository *r,
1219                    struct transport *transport,
1220                    struct refspec *rs, int flags,
1221                    unsigned int *reject_reasons)
1222 {
1223         *reject_reasons = 0;
1224
1225         if (transport_color_config() < 0)
1226                 return -1;
1227
1228         if (transport->vtable->push_refs) {
1229                 struct ref *remote_refs;
1230                 struct ref *local_refs = get_local_heads();
1231                 int match_flags = MATCH_REFS_NONE;
1232                 int verbose = (transport->verbose > 0);
1233                 int quiet = (transport->verbose < 0);
1234                 int porcelain = flags & TRANSPORT_PUSH_PORCELAIN;
1235                 int pretend = flags & TRANSPORT_PUSH_DRY_RUN;
1236                 int push_ret, ret, err;
1237                 struct strvec ref_prefixes = STRVEC_INIT;
1238
1239                 if (check_push_refs(local_refs, rs) < 0)
1240                         return -1;
1241
1242                 refspec_ref_prefixes(rs, &ref_prefixes);
1243
1244                 trace2_region_enter("transport_push", "get_refs_list", r);
1245                 remote_refs = transport->vtable->get_refs_list(transport, 1,
1246                                                                &ref_prefixes);
1247                 trace2_region_leave("transport_push", "get_refs_list", r);
1248
1249                 strvec_clear(&ref_prefixes);
1250
1251                 if (flags & TRANSPORT_PUSH_ALL)
1252                         match_flags |= MATCH_REFS_ALL;
1253                 if (flags & TRANSPORT_PUSH_MIRROR)
1254                         match_flags |= MATCH_REFS_MIRROR;
1255                 if (flags & TRANSPORT_PUSH_PRUNE)
1256                         match_flags |= MATCH_REFS_PRUNE;
1257                 if (flags & TRANSPORT_PUSH_FOLLOW_TAGS)
1258                         match_flags |= MATCH_REFS_FOLLOW_TAGS;
1259
1260                 if (match_push_refs(local_refs, &remote_refs, rs, match_flags))
1261                         return -1;
1262
1263                 if (transport->smart_options &&
1264                     transport->smart_options->cas &&
1265                     !is_empty_cas(transport->smart_options->cas))
1266                         apply_push_cas(transport->smart_options->cas,
1267                                        transport->remote, remote_refs);
1268
1269                 set_ref_status_for_push(remote_refs,
1270                         flags & TRANSPORT_PUSH_MIRROR,
1271                         flags & TRANSPORT_PUSH_FORCE);
1272
1273                 if (!(flags & TRANSPORT_PUSH_NO_HOOK))
1274                         if (run_pre_push_hook(transport, remote_refs))
1275                                 return -1;
1276
1277                 if ((flags & (TRANSPORT_RECURSE_SUBMODULES_ON_DEMAND |
1278                               TRANSPORT_RECURSE_SUBMODULES_ONLY)) &&
1279                     !is_bare_repository()) {
1280                         struct ref *ref = remote_refs;
1281                         struct oid_array commits = OID_ARRAY_INIT;
1282
1283                         trace2_region_enter("transport_push", "push_submodules", r);
1284                         for (; ref; ref = ref->next)
1285                                 if (!is_null_oid(&ref->new_oid))
1286                                         oid_array_append(&commits,
1287                                                           &ref->new_oid);
1288
1289                         if (!push_unpushed_submodules(r,
1290                                                       &commits,
1291                                                       transport->remote,
1292                                                       rs,
1293                                                       transport->push_options,
1294                                                       pretend)) {
1295                                 oid_array_clear(&commits);
1296                                 trace2_region_leave("transport_push", "push_submodules", r);
1297                                 die(_("failed to push all needed submodules"));
1298                         }
1299                         oid_array_clear(&commits);
1300                         trace2_region_leave("transport_push", "push_submodules", r);
1301                 }
1302
1303                 if (((flags & TRANSPORT_RECURSE_SUBMODULES_CHECK) ||
1304                      ((flags & (TRANSPORT_RECURSE_SUBMODULES_ON_DEMAND |
1305                                 TRANSPORT_RECURSE_SUBMODULES_ONLY)) &&
1306                       !pretend)) && !is_bare_repository()) {
1307                         struct ref *ref = remote_refs;
1308                         struct string_list needs_pushing = STRING_LIST_INIT_DUP;
1309                         struct oid_array commits = OID_ARRAY_INIT;
1310
1311                         trace2_region_enter("transport_push", "check_submodules", r);
1312                         for (; ref; ref = ref->next)
1313                                 if (!is_null_oid(&ref->new_oid))
1314                                         oid_array_append(&commits,
1315                                                           &ref->new_oid);
1316
1317                         if (find_unpushed_submodules(r,
1318                                                      &commits,
1319                                                      transport->remote->name,
1320                                                      &needs_pushing)) {
1321                                 oid_array_clear(&commits);
1322                                 trace2_region_leave("transport_push", "check_submodules", r);
1323                                 die_with_unpushed_submodules(&needs_pushing);
1324                         }
1325                         string_list_clear(&needs_pushing, 0);
1326                         oid_array_clear(&commits);
1327                         trace2_region_leave("transport_push", "check_submodules", r);
1328                 }
1329
1330                 if (!(flags & TRANSPORT_RECURSE_SUBMODULES_ONLY)) {
1331                         trace2_region_enter("transport_push", "push_refs", r);
1332                         push_ret = transport->vtable->push_refs(transport, remote_refs, flags);
1333                         trace2_region_leave("transport_push", "push_refs", r);
1334                 } else
1335                         push_ret = 0;
1336                 err = push_had_errors(remote_refs);
1337                 ret = push_ret | err;
1338
1339                 if (!quiet || err)
1340                         transport_print_push_status(transport->url, remote_refs,
1341                                         verbose | porcelain, porcelain,
1342                                         reject_reasons);
1343
1344                 if (flags & TRANSPORT_PUSH_SET_UPSTREAM)
1345                         set_upstreams(transport, remote_refs, pretend);
1346
1347                 if (!(flags & (TRANSPORT_PUSH_DRY_RUN |
1348                                TRANSPORT_RECURSE_SUBMODULES_ONLY))) {
1349                         struct ref *ref;
1350                         for (ref = remote_refs; ref; ref = ref->next)
1351                                 transport_update_tracking_ref(transport->remote, ref, verbose);
1352                 }
1353
1354                 if (porcelain && !push_ret)
1355                         puts("Done");
1356                 else if (!quiet && !ret && !transport_refs_pushed(remote_refs))
1357                         fprintf(stderr, "Everything up-to-date\n");
1358
1359                 return ret;
1360         }
1361         return 1;
1362 }
1363
1364 const struct ref *transport_get_remote_refs(struct transport *transport,
1365                                             const struct strvec *ref_prefixes)
1366 {
1367         if (!transport->got_remote_refs) {
1368                 transport->remote_refs =
1369                         transport->vtable->get_refs_list(transport, 0,
1370                                                          ref_prefixes);
1371                 transport->got_remote_refs = 1;
1372         }
1373
1374         return transport->remote_refs;
1375 }
1376
1377 int transport_fetch_refs(struct transport *transport, struct ref *refs)
1378 {
1379         int rc;
1380         int nr_heads = 0, nr_alloc = 0, nr_refs = 0;
1381         struct ref **heads = NULL;
1382         struct ref *rm;
1383
1384         for (rm = refs; rm; rm = rm->next) {
1385                 nr_refs++;
1386                 if (rm->peer_ref &&
1387                     !is_null_oid(&rm->old_oid) &&
1388                     oideq(&rm->peer_ref->old_oid, &rm->old_oid))
1389                         continue;
1390                 ALLOC_GROW(heads, nr_heads + 1, nr_alloc);
1391                 heads[nr_heads++] = rm;
1392         }
1393
1394         if (!nr_heads) {
1395                 /*
1396                  * When deepening of a shallow repository is requested,
1397                  * then local and remote refs are likely to still be equal.
1398                  * Just feed them all to the fetch method in that case.
1399                  * This condition shouldn't be met in a non-deepening fetch
1400                  * (see builtin/fetch.c:quickfetch()).
1401                  */
1402                 ALLOC_ARRAY(heads, nr_refs);
1403                 for (rm = refs; rm; rm = rm->next)
1404                         heads[nr_heads++] = rm;
1405         }
1406
1407         rc = transport->vtable->fetch(transport, nr_heads, heads);
1408
1409         free(heads);
1410         return rc;
1411 }
1412
1413 void transport_unlock_pack(struct transport *transport)
1414 {
1415         int i;
1416
1417         for (i = 0; i < transport->pack_lockfiles.nr; i++)
1418                 unlink_or_warn(transport->pack_lockfiles.items[i].string);
1419         string_list_clear(&transport->pack_lockfiles, 0);
1420 }
1421
1422 int transport_connect(struct transport *transport, const char *name,
1423                       const char *exec, int fd[2])
1424 {
1425         if (transport->vtable->connect)
1426                 return transport->vtable->connect(transport, name, exec, fd);
1427         else
1428                 die(_("operation not supported by protocol"));
1429 }
1430
1431 int transport_disconnect(struct transport *transport)
1432 {
1433         int ret = 0;
1434         if (transport->vtable->disconnect)
1435                 ret = transport->vtable->disconnect(transport);
1436         free(transport);
1437         return ret;
1438 }
1439
1440 /*
1441  * Strip username (and password) from a URL and return
1442  * it in a newly allocated string.
1443  */
1444 char *transport_anonymize_url(const char *url)
1445 {
1446         char *scheme_prefix, *anon_part;
1447         size_t anon_len, prefix_len = 0;
1448
1449         anon_part = strchr(url, '@');
1450         if (url_is_local_not_ssh(url) || !anon_part)
1451                 goto literal_copy;
1452
1453         anon_len = strlen(++anon_part);
1454         scheme_prefix = strstr(url, "://");
1455         if (!scheme_prefix) {
1456                 if (!strchr(anon_part, ':'))
1457                         /* cannot be "me@there:/path/name" */
1458                         goto literal_copy;
1459         } else {
1460                 const char *cp;
1461                 /* make sure scheme is reasonable */
1462                 for (cp = url; cp < scheme_prefix; cp++) {
1463                         switch (*cp) {
1464                                 /* RFC 1738 2.1 */
1465                         case '+': case '.': case '-':
1466                                 break; /* ok */
1467                         default:
1468                                 if (isalnum(*cp))
1469                                         break;
1470                                 /* it isn't */
1471                                 goto literal_copy;
1472                         }
1473                 }
1474                 /* @ past the first slash does not count */
1475                 cp = strchr(scheme_prefix + 3, '/');
1476                 if (cp && cp < anon_part)
1477                         goto literal_copy;
1478                 prefix_len = scheme_prefix - url + 3;
1479         }
1480         return xstrfmt("%.*s%.*s", (int)prefix_len, url,
1481                        (int)anon_len, anon_part);
1482 literal_copy:
1483         return xstrdup(url);
1484 }