connect.c

   1 #include "git-compat-util.h"
   2 #include "cache.h"
   3 #include "config.h"
   4 #include "pkt-line.h"
   5 #include "quote.h"
   6 #include "refs.h"
   7 #include "run-command.h"
   8 #include "remote.h"
   9 #include "connect.h"
  10 #include "url.h"
  11 #include "string-list.h"
  12 #include "oid-array.h"
  13 #include "transport.h"
  14 #include "strbuf.h"
  15 #include "version.h"
  16 #include "protocol.h"
  17 #include "alias.h"
  18
  19 static char *server_capabilities_v1;
  20 static struct strvec server_capabilities_v2 = STRVEC_INIT;
  21 static const char *next_server_feature_value(const char *feature, int *len, int *offset);
  22
  23 static int check_ref(const char *name, unsigned int flags)
  24 {
  25         if (!flags)
  26                 return 1;
  27
  28         if (!skip_prefix(name, "refs/", &name))
  29                 return 0;
  30
  31         /* REF_NORMAL means that we don't want the magic fake tag refs */
  32         if ((flags & REF_NORMAL) && check_refname_format(name, 0))
  33                 return 0;
  34
  35         /* REF_HEADS means that we want regular branch heads */
  36         if ((flags & REF_HEADS) && starts_with(name, "heads/"))
  37                 return 1;
  38
  39         /* REF_TAGS means that we want tags */
  40         if ((flags & REF_TAGS) && starts_with(name, "tags/"))
  41                 return 1;
  42
  43         /* All type bits clear means that we are ok with anything */
  44         return !(flags & ~REF_NORMAL);
  45 }
  46
  47 int check_ref_type(const struct ref *ref, int flags)
  48 {
  49         return check_ref(ref->name, flags);
  50 }
  51
  52 static NORETURN void die_initial_contact(int unexpected)
  53 {
  54         /*
  55          * A hang-up after seeing some response from the other end
  56          * means that it is unexpected, as we know the other end is
  57          * willing to talk to us.  A hang-up before seeing any
  58          * response does not necessarily mean an ACL problem, though.
  59          */
  60         if (unexpected)
  61                 die(_("the remote end hung up upon initial contact"));
  62         else
  63                 die(_("Could not read from remote repository.\n\n"
  64                       "Please make sure you have the correct access rights\n"
  65                       "and the repository exists."));
  66 }
  67
  68 /* Checks if the server supports the capability 'c' */
  69 int server_supports_v2(const char *c, int die_on_error)
  70 {
  71         int i;
  72
  73         for (i = 0; i < server_capabilities_v2.nr; i++) {
  74                 const char *out;
  75                 if (skip_prefix(server_capabilities_v2.v[i], c, &out) &&
  76                     (!*out || *out == '='))
  77                         return 1;
  78         }
  79
  80         if (die_on_error)
  81                 die(_("server doesn't support '%s'"), c);
  82
  83         return 0;
  84 }
  85
  86 int server_feature_v2(const char *c, const char **v)
  87 {
  88         int i;
  89
  90         for (i = 0; i < server_capabilities_v2.nr; i++) {
  91                 const char *out;
  92                 if (skip_prefix(server_capabilities_v2.v[i], c, &out) &&
  93                     (*out == '=')) {
  94                         *v = out + 1;
  95                         return 1;
  96                 }
  97         }
  98         return 0;
  99 }
 100
 101 int server_supports_feature(const char *c, const char *feature,
 102                             int die_on_error)
 103 {
 104         int i;
 105
 106         for (i = 0; i < server_capabilities_v2.nr; i++) {
 107                 const char *out;
 108                 if (skip_prefix(server_capabilities_v2.v[i], c, &out) &&
 109                     (!*out || *(out++) == '=')) {
 110                         if (parse_feature_request(out, feature))
 111                                 return 1;
 112                         else
 113                                 break;
 114                 }
 115         }
 116
 117         if (die_on_error)
 118                 die(_("server doesn't support feature '%s'"), feature);
 119
 120         return 0;
 121 }
 122
 123 static void process_capabilities_v2(struct packet_reader *reader)
 124 {
 125         while (packet_reader_read(reader) == PACKET_READ_NORMAL)
 126                 strvec_push(&server_capabilities_v2, reader->line);
 127
 128         if (reader->status != PACKET_READ_FLUSH)
 129                 die(_("expected flush after capabilities"));
 130 }
 131
 132 enum protocol_version discover_version(struct packet_reader *reader)
 133 {
 134         enum protocol_version version = protocol_unknown_version;
 135
 136         /*
 137          * Peek the first line of the server's response to
 138          * determine the protocol version the server is speaking.
 139          */
 140         switch (packet_reader_peek(reader)) {
 141         case PACKET_READ_EOF:
 142                 die_initial_contact(0);
 143         case PACKET_READ_FLUSH:
 144         case PACKET_READ_DELIM:
 145         case PACKET_READ_RESPONSE_END:
 146                 version = protocol_v0;
 147                 break;
 148         case PACKET_READ_NORMAL:
 149                 version = determine_protocol_version_client(reader->line);
 150                 break;
 151         }
 152
 153         switch (version) {
 154         case protocol_v2:
 155                 process_capabilities_v2(reader);
 156                 break;
 157         case protocol_v1:
 158                 /* Read the peeked version line */
 159                 packet_reader_read(reader);
 160                 break;
 161         case protocol_v0:
 162                 break;
 163         case protocol_unknown_version:
 164                 BUG("unknown protocol version");
 165         }
 166
 167         trace2_data_intmax("transfer", NULL, "negotiated-version", version);
 168
 169         return version;
 170 }
 171
 172 static void parse_one_symref_info(struct string_list *symref, const char *val, int len)
 173 {
 174         char *sym, *target;
 175         struct string_list_item *item;
 176
 177         if (!len)
 178                 return; /* just "symref" */
 179         /* e.g. "symref=HEAD:refs/heads/master" */
 180         sym = xmemdupz(val, len);
 181         target = strchr(sym, ':');
 182         if (!target)
 183                 /* just "symref=something" */
 184                 goto reject;
 185         *(target++) = '\0';
 186         if (check_refname_format(sym, REFNAME_ALLOW_ONELEVEL) ||
 187             check_refname_format(target, REFNAME_ALLOW_ONELEVEL))
 188                 /* "symref=bogus:pair */
 189                 goto reject;
 190         item = string_list_append_nodup(symref, sym);
 191         item->util = target;
 192         return;
 193 reject:
 194         free(sym);
 195         return;
 196 }
 197
 198 static void annotate_refs_with_symref_info(struct ref *ref)
 199 {
 200         struct string_list symref = STRING_LIST_INIT_DUP;
 201         int offset = 0;
 202
 203         while (1) {
 204                 int len;
 205                 const char *val;
 206
 207                 val = next_server_feature_value("symref", &len, &offset);
 208                 if (!val)
 209                         break;
 210                 parse_one_symref_info(&symref, val, len);
 211         }
 212         string_list_sort(&symref);
 213
 214         for (; ref; ref = ref->next) {
 215                 struct string_list_item *item;
 216                 item = string_list_lookup(&symref, ref->name);
 217                 if (!item)
 218                         continue;
 219                 ref->symref = xstrdup((char *)item->util);
 220         }
 221         string_list_clear(&symref, 0);
 222 }
 223
 224 static void process_capabilities(struct packet_reader *reader, int *linelen)
 225 {
 226         const char *feat_val;
 227         int feat_len;
 228         const char *line = reader->line;
 229         int nul_location = strlen(line);
 230         if (nul_location == *linelen)
 231                 return;
 232         server_capabilities_v1 = xstrdup(line + nul_location + 1);
 233         *linelen = nul_location;
 234
 235         feat_val = server_feature_value("object-format", &feat_len);
 236         if (feat_val) {
 237                 char *hash_name = xstrndup(feat_val, feat_len);
 238                 int hash_algo = hash_algo_by_name(hash_name);
 239                 if (hash_algo != GIT_HASH_UNKNOWN)
 240                         reader->hash_algo = &hash_algos[hash_algo];
 241                 free(hash_name);
 242         } else {
 243                 reader->hash_algo = &hash_algos[GIT_HASH_SHA1];
 244         }
 245 }
 246
 247 static int process_dummy_ref(const struct packet_reader *reader)
 248 {
 249         const char *line = reader->line;
 250         struct object_id oid;
 251         const char *name;
 252
 253         if (parse_oid_hex_algop(line, &oid, &name, reader->hash_algo))
 254                 return 0;
 255         if (*name != ' ')
 256                 return 0;
 257         name++;
 258
 259         return oideq(null_oid(), &oid) && !strcmp(name, "capabilities^{}");
 260 }
 261
 262 static void check_no_capabilities(const char *line, int len)
 263 {
 264         if (strlen(line) != len)
 265                 warning(_("ignoring capabilities after first line '%s'"),
 266                         line + strlen(line));
 267 }
 268
 269 static int process_ref(const struct packet_reader *reader, int len,
 270                        struct ref ***list, unsigned int flags,
 271                        struct oid_array *extra_have)
 272 {
 273         const char *line = reader->line;
 274         struct object_id old_oid;
 275         const char *name;
 276
 277         if (parse_oid_hex_algop(line, &old_oid, &name, reader->hash_algo))
 278                 return 0;
 279         if (*name != ' ')
 280                 return 0;
 281         name++;
 282
 283         if (extra_have && !strcmp(name, ".have")) {
 284                 oid_array_append(extra_have, &old_oid);
 285         } else if (!strcmp(name, "capabilities^{}")) {
 286                 die(_("protocol error: unexpected capabilities^{}"));
 287         } else if (check_ref(name, flags)) {
 288                 struct ref *ref = alloc_ref(name);
 289                 oidcpy(&ref->old_oid, &old_oid);
 290                 **list = ref;
 291                 *list = &ref->next;
 292         }
 293         check_no_capabilities(line, len);
 294         return 1;
 295 }
 296
 297 static int process_shallow(const struct packet_reader *reader, int len,
 298                            struct oid_array *shallow_points)
 299 {
 300         const char *line = reader->line;
 301         const char *arg;
 302         struct object_id old_oid;
 303
 304         if (!skip_prefix(line, "shallow ", &arg))
 305                 return 0;
 306
 307         if (get_oid_hex_algop(arg, &old_oid, reader->hash_algo))
 308                 die(_("protocol error: expected shallow sha-1, got '%s'"), arg);
 309         if (!shallow_points)
 310                 die(_("repository on the other end cannot be shallow"));
 311         oid_array_append(shallow_points, &old_oid);
 312         check_no_capabilities(line, len);
 313         return 1;
 314 }
 315
 316 enum get_remote_heads_state {
 317         EXPECTING_FIRST_REF = 0,
 318         EXPECTING_REF,
 319         EXPECTING_SHALLOW,
 320         EXPECTING_DONE,
 321 };
 322
 323 /*
 324  * Read all the refs from the other end
 325  */
 326 struct ref **get_remote_heads(struct packet_reader *reader,
 327                               struct ref **list, unsigned int flags,
 328                               struct oid_array *extra_have,
 329                               struct oid_array *shallow_points)
 330 {
 331         struct ref **orig_list = list;
 332         int len = 0;
 333         enum get_remote_heads_state state = EXPECTING_FIRST_REF;
 334
 335         *list = NULL;
 336
 337         while (state != EXPECTING_DONE) {
 338                 switch (packet_reader_read(reader)) {
 339                 case PACKET_READ_EOF:
 340                         die_initial_contact(1);
 341                 case PACKET_READ_NORMAL:
 342                         len = reader->pktlen;
 343                         break;
 344                 case PACKET_READ_FLUSH:
 345                         state = EXPECTING_DONE;
 346                         break;
 347                 case PACKET_READ_DELIM:
 348                 case PACKET_READ_RESPONSE_END:
 349                         die(_("invalid packet"));
 350                 }
 351
 352                 switch (state) {
 353                 case EXPECTING_FIRST_REF:
 354                         process_capabilities(reader, &len);
 355                         if (process_dummy_ref(reader)) {
 356                                 state = EXPECTING_SHALLOW;
 357                                 break;
 358                         }
 359                         state = EXPECTING_REF;
 360                         /* fallthrough */
 361                 case EXPECTING_REF:
 362                         if (process_ref(reader, len, &list, flags, extra_have))
 363                                 break;
 364                         state = EXPECTING_SHALLOW;
 365                         /* fallthrough */
 366                 case EXPECTING_SHALLOW:
 367                         if (process_shallow(reader, len, shallow_points))
 368                                 break;
 369                         die(_("protocol error: unexpected '%s'"), reader->line);
 370                 case EXPECTING_DONE:
 371                         break;
 372                 }
 373         }
 374
 375         annotate_refs_with_symref_info(*orig_list);
 376
 377         return list;
 378 }
 379
 380 /* Returns 1 when a valid ref has been added to `list`, 0 otherwise */
 381 static int process_ref_v2(struct packet_reader *reader, struct ref ***list,
 382                           char **unborn_head_target)
 383 {
 384         int ret = 1;
 385         int i = 0;
 386         struct object_id old_oid;
 387         struct ref *ref;
 388         struct string_list line_sections = STRING_LIST_INIT_DUP;
 389         const char *end;
 390         const char *line = reader->line;
 391
 392         /*
 393          * Ref lines have a number of fields which are space deliminated.  The
 394          * first field is the OID of the ref.  The second field is the ref
 395          * name.  Subsequent fields (symref-target and peeled) are optional and
 396          * don't have a particular order.
 397          */
 398         if (string_list_split(&line_sections, line, ' ', -1) < 2) {
 399                 ret = 0;
 400                 goto out;
 401         }
 402
 403         if (!strcmp("unborn", line_sections.items[i].string)) {
 404                 i++;
 405                 if (unborn_head_target &&
 406                     !strcmp("HEAD", line_sections.items[i++].string)) {
 407                         /*
 408                          * Look for the symref target (if any). If found,
 409                          * return it to the caller.
 410                          */
 411                         for (; i < line_sections.nr; i++) {
 412                                 const char *arg = line_sections.items[i].string;
 413
 414                                 if (skip_prefix(arg, "symref-target:", &arg)) {
 415                                         *unborn_head_target = xstrdup(arg);
 416                                         break;
 417                                 }
 418                         }
 419                 }
 420                 goto out;
 421         }
 422         if (parse_oid_hex_algop(line_sections.items[i++].string, &old_oid, &end, reader->hash_algo) ||
 423             *end) {
 424                 ret = 0;
 425                 goto out;
 426         }
 427
 428         ref = alloc_ref(line_sections.items[i++].string);
 429
 430         memcpy(ref->old_oid.hash, old_oid.hash, reader->hash_algo->rawsz);
 431         **list = ref;
 432         *list = &ref->next;
 433
 434         for (; i < line_sections.nr; i++) {
 435                 const char *arg = line_sections.items[i].string;
 436                 if (skip_prefix(arg, "symref-target:", &arg))
 437                         ref->symref = xstrdup(arg);
 438
 439                 if (skip_prefix(arg, "peeled:", &arg)) {
 440                         struct object_id peeled_oid;
 441                         char *peeled_name;
 442                         struct ref *peeled;
 443                         if (parse_oid_hex_algop(arg, &peeled_oid, &end,
 444                                                 reader->hash_algo) || *end) {
 445                                 ret = 0;
 446                                 goto out;
 447                         }
 448
 449                         peeled_name = xstrfmt("%s^{}", ref->name);
 450                         peeled = alloc_ref(peeled_name);
 451
 452                         memcpy(peeled->old_oid.hash, peeled_oid.hash,
 453                                reader->hash_algo->rawsz);
 454                         **list = peeled;
 455                         *list = &peeled->next;
 456
 457                         free(peeled_name);
 458                 }
 459         }
 460
 461 out:
 462         string_list_clear(&line_sections, 0);
 463         return ret;
 464 }
 465
 466 void check_stateless_delimiter(int stateless_rpc,
 467                               struct packet_reader *reader,
 468                               const char *error)
 469 {
 470         if (!stateless_rpc)
 471                 return; /* not in stateless mode, no delimiter expected */
 472         if (packet_reader_read(reader) != PACKET_READ_RESPONSE_END)
 473                 die("%s", error);
 474 }
 475
 476 struct ref **get_remote_refs(int fd_out, struct packet_reader *reader,
 477                              struct ref **list, int for_push,
 478                              struct transport_ls_refs_options *transport_options,
 479                              const struct string_list *server_options,
 480                              int stateless_rpc)
 481 {
 482         int i;
 483         const char *hash_name;
 484         struct strvec *ref_prefixes = transport_options ?
 485                 &transport_options->ref_prefixes : NULL;
 486         char **unborn_head_target = transport_options ?
 487                 &transport_options->unborn_head_target : NULL;
 488         *list = NULL;
 489
 490         if (server_supports_v2("ls-refs", 1))
 491                 packet_write_fmt(fd_out, "command=ls-refs\n");
 492
 493         if (server_supports_v2("agent", 0))
 494                 packet_write_fmt(fd_out, "agent=%s", git_user_agent_sanitized());
 495
 496         if (server_feature_v2("object-format", &hash_name)) {
 497                 int hash_algo = hash_algo_by_name(hash_name);
 498                 if (hash_algo == GIT_HASH_UNKNOWN)
 499                         die(_("unknown object format '%s' specified by server"), hash_name);
 500                 reader->hash_algo = &hash_algos[hash_algo];
 501                 packet_write_fmt(fd_out, "object-format=%s", reader->hash_algo->name);
 502         } else {
 503                 reader->hash_algo = &hash_algos[GIT_HASH_SHA1];
 504         }
 505
 506         if (server_options && server_options->nr &&
 507             server_supports_v2("server-option", 1))
 508                 for (i = 0; i < server_options->nr; i++)
 509                         packet_write_fmt(fd_out, "server-option=%s",
 510                                          server_options->items[i].string);
 511
 512         packet_delim(fd_out);
 513         /* When pushing we don't want to request the peeled tags */
 514         if (!for_push)
 515                 packet_write_fmt(fd_out, "peel\n");
 516         packet_write_fmt(fd_out, "symrefs\n");
 517         if (server_supports_feature("ls-refs", "unborn", 0))
 518                 packet_write_fmt(fd_out, "unborn\n");
 519         for (i = 0; ref_prefixes && i < ref_prefixes->nr; i++) {
 520                 packet_write_fmt(fd_out, "ref-prefix %s\n",
 521                                  ref_prefixes->v[i]);
 522         }
 523         packet_flush(fd_out);
 524
 525         /* Process response from server */
 526         while (packet_reader_read(reader) == PACKET_READ_NORMAL) {
 527                 if (!process_ref_v2(reader, &list, unborn_head_target))
 528                         die(_("invalid ls-refs response: %s"), reader->line);
 529         }
 530
 531         if (reader->status != PACKET_READ_FLUSH)
 532                 die(_("expected flush after ref listing"));
 533
 534         check_stateless_delimiter(stateless_rpc, reader,
 535                                   _("expected response end packet after ref listing"));
 536
 537         return list;
 538 }
 539
 540 const char *parse_feature_value(const char *feature_list, const char *feature, int *lenp, int *offset)
 541 {
 542         int len;
 543
 544         if (!feature_list)
 545                 return NULL;
 546
 547         len = strlen(feature);
 548         if (offset)
 549                 feature_list += *offset;
 550         while (*feature_list) {
 551                 const char *found = strstr(feature_list, feature);
 552                 if (!found)
 553                         return NULL;
 554                 if (feature_list == found || isspace(found[-1])) {
 555                         const char *value = found + len;
 556                         /* feature with no value (e.g., "thin-pack") */
 557                         if (!*value || isspace(*value)) {
 558                                 if (lenp)
 559                                         *lenp = 0;
 560                                 if (offset)
 561                                         *offset = found + len - feature_list;
 562                                 return value;
 563                         }
 564                         /* feature with a value (e.g., "agent=git/1.2.3") */
 565                         else if (*value == '=') {
 566                                 int end;
 567
 568                                 value++;
 569                                 end = strcspn(value, " \t\n");
 570                                 if (lenp)
 571                                         *lenp = end;
 572                                 if (offset)
 573                                         *offset = value + end - feature_list;
 574                                 return value;
 575                         }
 576                         /*
 577                          * otherwise we matched a substring of another feature;
 578                          * keep looking
 579                          */
 580                 }
 581                 feature_list = found + 1;
 582         }
 583         return NULL;
 584 }
 585
 586 int server_supports_hash(const char *desired, int *feature_supported)
 587 {
 588         int offset = 0;
 589         int len;
 590         const char *hash;
 591
 592         hash = next_server_feature_value("object-format", &len, &offset);
 593         if (feature_supported)
 594                 *feature_supported = !!hash;
 595         if (!hash) {
 596                 hash = hash_algos[GIT_HASH_SHA1].name;
 597                 len = strlen(hash);
 598         }
 599         while (hash) {
 600                 if (!xstrncmpz(desired, hash, len))
 601                         return 1;
 602
 603                 hash = next_server_feature_value("object-format", &len, &offset);
 604         }
 605         return 0;
 606 }
 607
 608 int parse_feature_request(const char *feature_list, const char *feature)
 609 {
 610         return !!parse_feature_value(feature_list, feature, NULL, NULL);
 611 }
 612
 613 static const char *next_server_feature_value(const char *feature, int *len, int *offset)
 614 {
 615         return parse_feature_value(server_capabilities_v1, feature, len, offset);
 616 }
 617
 618 const char *server_feature_value(const char *feature, int *len)
 619 {
 620         return parse_feature_value(server_capabilities_v1, feature, len, NULL);
 621 }
 622
 623 int server_supports(const char *feature)
 624 {
 625         return !!server_feature_value(feature, NULL);
 626 }
 627
 628 enum protocol {
 629         PROTO_LOCAL = 1,
 630         PROTO_FILE,
 631         PROTO_SSH,
 632         PROTO_GIT
 633 };
 634
 635 int url_is_local_not_ssh(const char *url)
 636 {
 637         const char *colon = strchr(url, ':');
 638         const char *slash = strchr(url, '/');
 639         return !colon || (slash && slash < colon) ||
 640                 (has_dos_drive_prefix(url) && is_valid_path(url));
 641 }
 642
 643 static const char *prot_name(enum protocol protocol)
 644 {
 645         switch (protocol) {
 646                 case PROTO_LOCAL:
 647                 case PROTO_FILE:
 648                         return "file";
 649                 case PROTO_SSH:
 650                         return "ssh";
 651                 case PROTO_GIT:
 652                         return "git";
 653                 default:
 654                         return "unknown protocol";
 655         }
 656 }
 657
 658 static enum protocol get_protocol(const char *name)
 659 {
 660         if (!strcmp(name, "ssh"))
 661                 return PROTO_SSH;
 662         if (!strcmp(name, "git"))
 663                 return PROTO_GIT;
 664         if (!strcmp(name, "git+ssh")) /* deprecated - do not use */
 665                 return PROTO_SSH;
 666         if (!strcmp(name, "ssh+git")) /* deprecated - do not use */
 667                 return PROTO_SSH;
 668         if (!strcmp(name, "file"))
 669                 return PROTO_FILE;
 670         die(_("protocol '%s' is not supported"), name);
 671 }
 672
 673 static char *host_end(char **hoststart, int removebrackets)
 674 {
 675         char *host = *hoststart;
 676         char *end;
 677         char *start = strstr(host, "@[");
 678         if (start)
 679                 start++; /* Jump over '@' */
 680         else
 681                 start = host;
 682         if (start[0] == '[') {
 683                 end = strchr(start + 1, ']');
 684                 if (end) {
 685                         if (removebrackets) {
 686                                 *end = 0;
 687                                 memmove(start, start + 1, end - start);
 688                                 end++;
 689                         }
 690                 } else
 691                         end = host;
 692         } else
 693                 end = host;
 694         return end;
 695 }
 696
 697 #define STR_(s) # s
 698 #define STR(s)  STR_(s)
 699
 700 static void get_host_and_port(char **host, const char **port)
 701 {
 702         char *colon, *end;
 703         end = host_end(host, 1);
 704         colon = strchr(end, ':');
 705         if (colon) {
 706                 long portnr = strtol(colon + 1, &end, 10);
 707                 if (end != colon + 1 && *end == '\0' && 0 <= portnr && portnr < 65536) {
 708                         *colon = 0;
 709                         *port = colon + 1;
 710                 } else if (!colon[1]) {
 711                         *colon = 0;
 712                 }
 713         }
 714 }
 715
 716 static void enable_keepalive(int sockfd)
 717 {
 718         int ka = 1;
 719
 720         if (setsockopt(sockfd, SOL_SOCKET, SO_KEEPALIVE, &ka, sizeof(ka)) < 0)
 721                 error_errno(_("unable to set SO_KEEPALIVE on socket"));
 722 }
 723
 724 #ifndef NO_IPV6
 725
 726 static const char *ai_name(const struct addrinfo *ai)
 727 {
 728         static char addr[NI_MAXHOST];
 729         if (getnameinfo(ai->ai_addr, ai->ai_addrlen, addr, sizeof(addr), NULL, 0,
 730                         NI_NUMERICHOST) != 0)
 731                 xsnprintf(addr, sizeof(addr), "(unknown)");
 732
 733         return addr;
 734 }
 735
 736 /*
 737  * Returns a connected socket() fd, or else die()s.
 738  */
 739 static int git_tcp_connect_sock(char *host, int flags)
 740 {
 741         struct strbuf error_message = STRBUF_INIT;
 742         int sockfd = -1;
 743         const char *port = STR(DEFAULT_GIT_PORT);
 744         struct addrinfo hints, *ai0, *ai;
 745         int gai;
 746         int cnt = 0;
 747
 748         get_host_and_port(&host, &port);
 749         if (!*port)
 750                 port = "<none>";
 751
 752         memset(&hints, 0, sizeof(hints));
 753         if (flags & CONNECT_IPV4)
 754                 hints.ai_family = AF_INET;
 755         else if (flags & CONNECT_IPV6)
 756                 hints.ai_family = AF_INET6;
 757         hints.ai_socktype = SOCK_STREAM;
 758         hints.ai_protocol = IPPROTO_TCP;
 759
 760         if (flags & CONNECT_VERBOSE)
 761                 fprintf(stderr, _("Looking up %s ... "), host);
 762
 763         gai = getaddrinfo(host, port, &hints, &ai);
 764         if (gai)
 765                 die(_("unable to look up %s (port %s) (%s)"), host, port, gai_strerror(gai));
 766
 767         if (flags & CONNECT_VERBOSE)
 768                 /* TRANSLATORS: this is the end of "Looking up %s ... " */
 769                 fprintf(stderr, _("done.\nConnecting to %s (port %s) ... "), host, port);
 770
 771         for (ai0 = ai; ai; ai = ai->ai_next, cnt++) {
 772                 sockfd = socket(ai->ai_family,
 773                                 ai->ai_socktype, ai->ai_protocol);
 774                 if ((sockfd < 0) ||
 775                     (connect(sockfd, ai->ai_addr, ai->ai_addrlen) < 0)) {
 776                         strbuf_addf(&error_message, "%s[%d: %s]: errno=%s\n",
 777                                     host, cnt, ai_name(ai), strerror(errno));
 778                         if (0 <= sockfd)
 779                                 close(sockfd);
 780                         sockfd = -1;
 781                         continue;
 782                 }
 783                 if (flags & CONNECT_VERBOSE)
 784                         fprintf(stderr, "%s ", ai_name(ai));
 785                 break;
 786         }
 787
 788         freeaddrinfo(ai0);
 789
 790         if (sockfd < 0)
 791                 die(_("unable to connect to %s:\n%s"), host, error_message.buf);
 792
 793         enable_keepalive(sockfd);
 794
 795         if (flags & CONNECT_VERBOSE)
 796                 /* TRANSLATORS: this is the end of "Connecting to %s (port %s) ... " */
 797                 fprintf_ln(stderr, _("done."));
 798
 799         strbuf_release(&error_message);
 800
 801         return sockfd;
 802 }
 803
 804 #else /* NO_IPV6 */
 805
 806 /*
 807  * Returns a connected socket() fd, or else die()s.
 808  */
 809 static int git_tcp_connect_sock(char *host, int flags)
 810 {
 811         struct strbuf error_message = STRBUF_INIT;
 812         int sockfd = -1;
 813         const char *port = STR(DEFAULT_GIT_PORT);
 814         char *ep;
 815         struct hostent *he;
 816         struct sockaddr_in sa;
 817         char **ap;
 818         unsigned int nport;
 819         int cnt;
 820
 821         get_host_and_port(&host, &port);
 822
 823         if (flags & CONNECT_VERBOSE)
 824                 fprintf(stderr, _("Looking up %s ... "), host);
 825
 826         he = gethostbyname(host);
 827         if (!he)
 828                 die(_("unable to look up %s (%s)"), host, hstrerror(h_errno));
 829         nport = strtoul(port, &ep, 10);
 830         if ( ep == port || *ep ) {
 831                 /* Not numeric */
 832                 struct servent *se = getservbyname(port,"tcp");
 833                 if ( !se )
 834                         die(_("unknown port %s"), port);
 835                 nport = se->s_port;
 836         }
 837
 838         if (flags & CONNECT_VERBOSE)
 839                 /* TRANSLATORS: this is the end of "Looking up %s ... " */
 840                 fprintf(stderr, _("done.\nConnecting to %s (port %s) ... "), host, port);
 841
 842         for (cnt = 0, ap = he->h_addr_list; *ap; ap++, cnt++) {
 843                 memset(&sa, 0, sizeof sa);
 844                 sa.sin_family = he->h_addrtype;
 845                 sa.sin_port = htons(nport);
 846                 memcpy(&sa.sin_addr, *ap, he->h_length);
 847
 848                 sockfd = socket(he->h_addrtype, SOCK_STREAM, 0);
 849                 if ((sockfd < 0) ||
 850                     connect(sockfd, (struct sockaddr *)&sa, sizeof sa) < 0) {
 851                         strbuf_addf(&error_message, "%s[%d: %s]: errno=%s\n",
 852                                 host,
 853                                 cnt,
 854                                 inet_ntoa(*(struct in_addr *)&sa.sin_addr),
 855                                 strerror(errno));
 856                         if (0 <= sockfd)
 857                                 close(sockfd);
 858                         sockfd = -1;
 859                         continue;
 860                 }
 861                 if (flags & CONNECT_VERBOSE)
 862                         fprintf(stderr, "%s ",
 863                                 inet_ntoa(*(struct in_addr *)&sa.sin_addr));
 864                 break;
 865         }
 866
 867         if (sockfd < 0)
 868                 die(_("unable to connect to %s:\n%s"), host, error_message.buf);
 869
 870         enable_keepalive(sockfd);
 871
 872         if (flags & CONNECT_VERBOSE)
 873                 /* TRANSLATORS: this is the end of "Connecting to %s (port %s) ... " */
 874                 fprintf_ln(stderr, _("done."));
 875
 876         return sockfd;
 877 }
 878
 879 #endif /* NO_IPV6 */
 880
 881
 882 /*
 883  * Dummy child_process returned by git_connect() if the transport protocol
 884  * does not need fork(2).
 885  */
 886 static struct child_process no_fork = CHILD_PROCESS_INIT;
 887
 888 int git_connection_is_socket(struct child_process *conn)
 889 {
 890         return conn == &no_fork;
 891 }
 892
 893 static struct child_process *git_tcp_connect(int fd[2], char *host, int flags)
 894 {
 895         int sockfd = git_tcp_connect_sock(host, flags);
 896
 897         fd[0] = sockfd;
 898         fd[1] = dup(sockfd);
 899
 900         return &no_fork;
 901 }
 902
 903
 904 static char *git_proxy_command;
 905
 906 static int git_proxy_command_options(const char *var, const char *value,
 907                 void *cb)
 908 {
 909         if (!strcmp(var, "core.gitproxy")) {
 910                 const char *for_pos;
 911                 int matchlen = -1;
 912                 int hostlen;
 913                 const char *rhost_name = cb;
 914                 int rhost_len = strlen(rhost_name);
 915
 916                 if (git_proxy_command)
 917                         return 0;
 918                 if (!value)
 919                         return config_error_nonbool(var);
 920                 /* [core]
 921                  * ;# matches www.kernel.org as well
 922                  * gitproxy = netcatter-1 for kernel.org
 923                  * gitproxy = netcatter-2 for sample.xz
 924                  * gitproxy = netcatter-default
 925                  */
 926                 for_pos = strstr(value, " for ");
 927                 if (!for_pos)
 928                         /* matches everybody */
 929                         matchlen = strlen(value);
 930                 else {
 931                         hostlen = strlen(for_pos + 5);
 932                         if (rhost_len < hostlen)
 933                                 matchlen = -1;
 934                         else if (!strncmp(for_pos + 5,
 935                                           rhost_name + rhost_len - hostlen,
 936                                           hostlen) &&
 937                                  ((rhost_len == hostlen) ||
 938                                   rhost_name[rhost_len - hostlen -1] == '.'))
 939                                 matchlen = for_pos - value;
 940                         else
 941                                 matchlen = -1;
 942                 }
 943                 if (0 <= matchlen) {
 944                         /* core.gitproxy = none for kernel.org */
 945                         if (matchlen == 4 &&
 946                             !memcmp(value, "none", 4))
 947                                 matchlen = 0;
 948                         git_proxy_command = xmemdupz(value, matchlen);
 949                 }
 950                 return 0;
 951         }
 952
 953         return git_default_config(var, value, cb);
 954 }
 955
 956 static int git_use_proxy(const char *host)
 957 {
 958         git_proxy_command = getenv("GIT_PROXY_COMMAND");
 959         git_config(git_proxy_command_options, (void*)host);
 960         return (git_proxy_command && *git_proxy_command);
 961 }
 962
 963 static struct child_process *git_proxy_connect(int fd[2], char *host)
 964 {
 965         const char *port = STR(DEFAULT_GIT_PORT);
 966         struct child_process *proxy;
 967
 968         get_host_and_port(&host, &port);
 969
 970         if (looks_like_command_line_option(host))
 971                 die(_("strange hostname '%s' blocked"), host);
 972         if (looks_like_command_line_option(port))
 973                 die(_("strange port '%s' blocked"), port);
 974
 975         proxy = xmalloc(sizeof(*proxy));
 976         child_process_init(proxy);
 977         strvec_push(&proxy->args, git_proxy_command);
 978         strvec_push(&proxy->args, host);
 979         strvec_push(&proxy->args, port);
 980         proxy->in = -1;
 981         proxy->out = -1;
 982         if (start_command(proxy))
 983                 die(_("cannot start proxy %s"), git_proxy_command);
 984         fd[0] = proxy->out; /* read from proxy stdout */
 985         fd[1] = proxy->in;  /* write to proxy stdin */
 986         return proxy;
 987 }
 988
 989 static char *get_port(char *host)
 990 {
 991         char *end;
 992         char *p = strchr(host, ':');
 993
 994         if (p) {
 995                 long port = strtol(p + 1, &end, 10);
 996                 if (end != p + 1 && *end == '\0' && 0 <= port && port < 65536) {
 997                         *p = '\0';
 998                         return p+1;
 999                 }
1000         }
1001
1002         return NULL;
1003 }
1004
1005 /*
1006  * Extract protocol and relevant parts from the specified connection URL.
1007  * The caller must free() the returned strings.
1008  */
1009 static enum protocol parse_connect_url(const char *url_orig, char **ret_host,
1010                                        char **ret_path)
1011 {
1012         char *url;
1013         char *host, *path;
1014         char *end;
1015         int separator = '/';
1016         enum protocol protocol = PROTO_LOCAL;
1017
1018         if (is_url(url_orig))
1019                 url = url_decode(url_orig);
1020         else
1021                 url = xstrdup(url_orig);
1022
1023         host = strstr(url, "://");
1024         if (host) {
1025                 *host = '\0';
1026                 protocol = get_protocol(url);
1027                 host += 3;
1028         } else {
1029                 host = url;
1030                 if (!url_is_local_not_ssh(url)) {
1031                         protocol = PROTO_SSH;
1032                         separator = ':';
1033                 }
1034         }
1035
1036         /*
1037          * Don't do destructive transforms as protocol code does
1038          * '[]' unwrapping in get_host_and_port()
1039          */
1040         end = host_end(&host, 0);
1041
1042         if (protocol == PROTO_LOCAL)
1043                 path = end;
1044         else if (protocol == PROTO_FILE && *host != '/' &&
1045                  !has_dos_drive_prefix(host) &&
1046                  offset_1st_component(host - 2) > 1)
1047                 path = host - 2; /* include the leading "//" */
1048         else if (protocol == PROTO_FILE && has_dos_drive_prefix(end))
1049                 path = end; /* "file://$(pwd)" may be "file://C:/projects/repo" */
1050         else
1051                 path = strchr(end, separator);
1052
1053         if (!path || !*path)
1054                 die(_("no path specified; see 'git help pull' for valid url syntax"));
1055
1056         /*
1057          * null-terminate hostname and point path to ~ for URL's like this:
1058          *    ssh://host.xz/~user/repo
1059          */
1060
1061         end = path; /* Need to \0 terminate host here */
1062         if (separator == ':')
1063                 path++; /* path starts after ':' */
1064         if (protocol == PROTO_GIT || protocol == PROTO_SSH) {
1065                 if (path[1] == '~')
1066                         path++;
1067         }
1068
1069         path = xstrdup(path);
1070         *end = '\0';
1071
1072         *ret_host = xstrdup(host);
1073         *ret_path = path;
1074         free(url);
1075         return protocol;
1076 }
1077
1078 static const char *get_ssh_command(void)
1079 {
1080         const char *ssh;
1081
1082         if ((ssh = getenv("GIT_SSH_COMMAND")))
1083                 return ssh;
1084
1085         if (!git_config_get_string_tmp("core.sshcommand", &ssh))
1086                 return ssh;
1087
1088         return NULL;
1089 }
1090
1091 enum ssh_variant {
1092         VARIANT_AUTO,
1093         VARIANT_SIMPLE,
1094         VARIANT_SSH,
1095         VARIANT_PLINK,
1096         VARIANT_PUTTY,
1097         VARIANT_TORTOISEPLINK,
1098 };
1099
1100 static void override_ssh_variant(enum ssh_variant *ssh_variant)
1101 {
1102         const char *variant = getenv("GIT_SSH_VARIANT");
1103
1104         if (!variant && git_config_get_string_tmp("ssh.variant", &variant))
1105                 return;
1106
1107         if (!strcmp(variant, "auto"))
1108                 *ssh_variant = VARIANT_AUTO;
1109         else if (!strcmp(variant, "plink"))
1110                 *ssh_variant = VARIANT_PLINK;
1111         else if (!strcmp(variant, "putty"))
1112                 *ssh_variant = VARIANT_PUTTY;
1113         else if (!strcmp(variant, "tortoiseplink"))
1114                 *ssh_variant = VARIANT_TORTOISEPLINK;
1115         else if (!strcmp(variant, "simple"))
1116                 *ssh_variant = VARIANT_SIMPLE;
1117         else
1118                 *ssh_variant = VARIANT_SSH;
1119 }
1120
1121 static enum ssh_variant determine_ssh_variant(const char *ssh_command,
1122                                               int is_cmdline)
1123 {
1124         enum ssh_variant ssh_variant = VARIANT_AUTO;
1125         const char *variant;
1126         char *p = NULL;
1127
1128         override_ssh_variant(&ssh_variant);
1129
1130         if (ssh_variant != VARIANT_AUTO)
1131                 return ssh_variant;
1132
1133         if (!is_cmdline) {
1134                 p = xstrdup(ssh_command);
1135                 variant = basename(p);
1136         } else {
1137                 const char **ssh_argv;
1138
1139                 p = xstrdup(ssh_command);
1140                 if (split_cmdline(p, &ssh_argv) > 0) {
1141                         variant = basename((char *)ssh_argv[0]);
1142                         /*
1143                          * At this point, variant points into the buffer
1144                          * referenced by p, hence we do not need ssh_argv
1145                          * any longer.
1146                          */
1147                         free(ssh_argv);
1148                 } else {
1149                         free(p);
1150                         return ssh_variant;
1151                 }
1152         }
1153
1154         if (!strcasecmp(variant, "ssh") ||
1155             !strcasecmp(variant, "ssh.exe"))
1156                 ssh_variant = VARIANT_SSH;
1157         else if (!strcasecmp(variant, "plink") ||
1158                  !strcasecmp(variant, "plink.exe"))
1159                 ssh_variant = VARIANT_PLINK;
1160         else if (!strcasecmp(variant, "tortoiseplink") ||
1161                  !strcasecmp(variant, "tortoiseplink.exe"))
1162                 ssh_variant = VARIANT_TORTOISEPLINK;
1163
1164         free(p);
1165         return ssh_variant;
1166 }
1167
1168 /*
1169  * Open a connection using Git's native protocol.
1170  *
1171  * The caller is responsible for freeing hostandport, but this function may
1172  * modify it (for example, to truncate it to remove the port part).
1173  */
1174 static struct child_process *git_connect_git(int fd[2], char *hostandport,
1175                                              const char *path, const char *prog,
1176                                              enum protocol_version version,
1177                                              int flags)
1178 {
1179         struct child_process *conn;
1180         struct strbuf request = STRBUF_INIT;
1181         /*
1182          * Set up virtual host information based on where we will
1183          * connect, unless the user has overridden us in
1184          * the environment.
1185          */
1186         char *target_host = getenv("GIT_OVERRIDE_VIRTUAL_HOST");
1187         if (target_host)
1188                 target_host = xstrdup(target_host);
1189         else
1190                 target_host = xstrdup(hostandport);
1191
1192         transport_check_allowed("git");
1193         if (strchr(target_host, '\n') || strchr(path, '\n'))
1194                 die(_("newline is forbidden in git:// hosts and repo paths"));
1195
1196         /*
1197          * These underlying connection commands die() if they
1198          * cannot connect.
1199          */
1200         if (git_use_proxy(hostandport))
1201                 conn = git_proxy_connect(fd, hostandport);
1202         else
1203                 conn = git_tcp_connect(fd, hostandport, flags);
1204         /*
1205          * Separate original protocol components prog and path
1206          * from extended host header with a NUL byte.
1207          *
1208          * Note: Do not add any other headers here!  Doing so
1209          * will cause older git-daemon servers to crash.
1210          */
1211         strbuf_addf(&request,
1212                     "%s %s%chost=%s%c",
1213                     prog, path, 0,
1214                     target_host, 0);
1215
1216         /* If using a new version put that stuff here after a second null byte */
1217         if (version > 0) {
1218                 strbuf_addch(&request, '\0');
1219                 strbuf_addf(&request, "version=%d%c",
1220                             version, '\0');
1221         }
1222
1223         packet_write(fd[1], request.buf, request.len);
1224
1225         free(target_host);
1226         strbuf_release(&request);
1227         return conn;
1228 }
1229
1230 /*
1231  * Append the appropriate environment variables to `env` and options to
1232  * `args` for running ssh in Git's SSH-tunneled transport.
1233  */
1234 static void push_ssh_options(struct strvec *args, struct strvec *env,
1235                              enum ssh_variant variant, const char *port,
1236                              enum protocol_version version, int flags)
1237 {
1238         if (variant == VARIANT_SSH &&
1239             version > 0) {
1240                 strvec_push(args, "-o");
1241                 strvec_push(args, "SendEnv=" GIT_PROTOCOL_ENVIRONMENT);
1242                 strvec_pushf(env, GIT_PROTOCOL_ENVIRONMENT "=version=%d",
1243                              version);
1244         }
1245
1246         if (flags & CONNECT_IPV4) {
1247                 switch (variant) {
1248                 case VARIANT_AUTO:
1249                         BUG("VARIANT_AUTO passed to push_ssh_options");
1250                 case VARIANT_SIMPLE:
1251                         die(_("ssh variant 'simple' does not support -4"));
1252                 case VARIANT_SSH:
1253                 case VARIANT_PLINK:
1254                 case VARIANT_PUTTY:
1255                 case VARIANT_TORTOISEPLINK:
1256                         strvec_push(args, "-4");
1257                 }
1258         } else if (flags & CONNECT_IPV6) {
1259                 switch (variant) {
1260                 case VARIANT_AUTO:
1261                         BUG("VARIANT_AUTO passed to push_ssh_options");
1262                 case VARIANT_SIMPLE:
1263                         die(_("ssh variant 'simple' does not support -6"));
1264                 case VARIANT_SSH:
1265                 case VARIANT_PLINK:
1266                 case VARIANT_PUTTY:
1267                 case VARIANT_TORTOISEPLINK:
1268                         strvec_push(args, "-6");
1269                 }
1270         }
1271
1272         if (variant == VARIANT_TORTOISEPLINK)
1273                 strvec_push(args, "-batch");
1274
1275         if (port) {
1276                 switch (variant) {
1277                 case VARIANT_AUTO:
1278                         BUG("VARIANT_AUTO passed to push_ssh_options");
1279                 case VARIANT_SIMPLE:
1280                         die(_("ssh variant 'simple' does not support setting port"));
1281                 case VARIANT_SSH:
1282                         strvec_push(args, "-p");
1283                         break;
1284                 case VARIANT_PLINK:
1285                 case VARIANT_PUTTY:
1286                 case VARIANT_TORTOISEPLINK:
1287                         strvec_push(args, "-P");
1288                 }
1289
1290                 strvec_push(args, port);
1291         }
1292 }
1293
1294 /* Prepare a child_process for use by Git's SSH-tunneled transport. */
1295 static void fill_ssh_args(struct child_process *conn, const char *ssh_host,
1296                           const char *port, enum protocol_version version,
1297                           int flags)
1298 {
1299         const char *ssh;
1300         enum ssh_variant variant;
1301
1302         if (looks_like_command_line_option(ssh_host))
1303                 die(_("strange hostname '%s' blocked"), ssh_host);
1304
1305         ssh = get_ssh_command();
1306         if (ssh) {
1307                 variant = determine_ssh_variant(ssh, 1);
1308         } else {
1309                 /*
1310                  * GIT_SSH is the no-shell version of
1311                  * GIT_SSH_COMMAND (and must remain so for
1312                  * historical compatibility).
1313                  */
1314                 conn->use_shell = 0;
1315
1316                 ssh = getenv("GIT_SSH");
1317                 if (!ssh)
1318                         ssh = "ssh";
1319                 variant = determine_ssh_variant(ssh, 0);
1320         }
1321
1322         if (variant == VARIANT_AUTO) {
1323                 struct child_process detect = CHILD_PROCESS_INIT;
1324
1325                 detect.use_shell = conn->use_shell;
1326                 detect.no_stdin = detect.no_stdout = detect.no_stderr = 1;
1327
1328                 strvec_push(&detect.args, ssh);
1329                 strvec_push(&detect.args, "-G");
1330                 push_ssh_options(&detect.args, &detect.env_array,
1331                                  VARIANT_SSH, port, version, flags);
1332                 strvec_push(&detect.args, ssh_host);
1333
1334                 variant = run_command(&detect) ? VARIANT_SIMPLE : VARIANT_SSH;
1335         }
1336
1337         strvec_push(&conn->args, ssh);
1338         push_ssh_options(&conn->args, &conn->env_array, variant, port, version, flags);
1339         strvec_push(&conn->args, ssh_host);
1340 }
1341
1342 /*
1343  * This returns the dummy child_process `no_fork` if the transport protocol
1344  * does not need fork(2), or a struct child_process object if it does.  Once
1345  * done, finish the connection with finish_connect() with the value returned
1346  * from this function (it is safe to call finish_connect() with NULL to
1347  * support the former case).
1348  *
1349  * If it returns, the connect is successful; it just dies on errors (this
1350  * will hopefully be changed in a libification effort, to return NULL when
1351  * the connection failed).
1352  */
1353 struct child_process *git_connect(int fd[2], const char *url,
1354                                   const char *prog, int flags)
1355 {
1356         char *hostandport, *path;
1357         struct child_process *conn;
1358         enum protocol protocol;
1359         enum protocol_version version = get_protocol_version_config();
1360
1361         /*
1362          * NEEDSWORK: If we are trying to use protocol v2 and we are planning
1363          * to perform a push, then fallback to v0 since the client doesn't know
1364          * how to push yet using v2.
1365          */
1366         if (version == protocol_v2 && !strcmp("git-receive-pack", prog))
1367                 version = protocol_v0;
1368
1369         /* Without this we cannot rely on waitpid() to tell
1370          * what happened to our children.
1371          */
1372         signal(SIGCHLD, SIG_DFL);
1373
1374         protocol = parse_connect_url(url, &hostandport, &path);
1375         if ((flags & CONNECT_DIAG_URL) && (protocol != PROTO_SSH)) {
1376                 printf("Diag: url=%s\n", url ? url : "NULL");
1377                 printf("Diag: protocol=%s\n", prot_name(protocol));
1378                 printf("Diag: hostandport=%s\n", hostandport ? hostandport : "NULL");
1379                 printf("Diag: path=%s\n", path ? path : "NULL");
1380                 conn = NULL;
1381         } else if (protocol == PROTO_GIT) {
1382                 conn = git_connect_git(fd, hostandport, path, prog, version, flags);
1383                 conn->trace2_child_class = "transport/git";
1384         } else {
1385                 struct strbuf cmd = STRBUF_INIT;
1386                 const char *const *var;
1387
1388                 conn = xmalloc(sizeof(*conn));
1389                 child_process_init(conn);
1390
1391                 if (looks_like_command_line_option(path))
1392                         die(_("strange pathname '%s' blocked"), path);
1393
1394                 strbuf_addstr(&cmd, prog);
1395                 strbuf_addch(&cmd, ' ');
1396                 sq_quote_buf(&cmd, path);
1397
1398                 /* remove repo-local variables from the environment */
1399                 for (var = local_repo_env; *var; var++)
1400                         strvec_push(&conn->env_array, *var);
1401
1402                 conn->use_shell = 1;
1403                 conn->in = conn->out = -1;
1404                 if (protocol == PROTO_SSH) {
1405                         char *ssh_host = hostandport;
1406                         const char *port = NULL;
1407                         transport_check_allowed("ssh");
1408                         get_host_and_port(&ssh_host, &port);
1409
1410                         if (!port)
1411                                 port = get_port(ssh_host);
1412
1413                         if (flags & CONNECT_DIAG_URL) {
1414                                 printf("Diag: url=%s\n", url ? url : "NULL");
1415                                 printf("Diag: protocol=%s\n", prot_name(protocol));
1416                                 printf("Diag: userandhost=%s\n", ssh_host ? ssh_host : "NULL");
1417                                 printf("Diag: port=%s\n", port ? port : "NONE");
1418                                 printf("Diag: path=%s\n", path ? path : "NULL");
1419
1420                                 free(hostandport);
1421                                 free(path);
1422                                 free(conn);
1423                                 strbuf_release(&cmd);
1424                                 return NULL;
1425                         }
1426                         conn->trace2_child_class = "transport/ssh";
1427                         fill_ssh_args(conn, ssh_host, port, version, flags);
1428                 } else {
1429                         transport_check_allowed("file");
1430                         conn->trace2_child_class = "transport/file";
1431                         if (version > 0) {
1432                                 strvec_pushf(&conn->env_array,
1433                                              GIT_PROTOCOL_ENVIRONMENT "=version=%d",
1434                                              version);
1435                         }
1436                 }
1437                 strvec_push(&conn->args, cmd.buf);
1438
1439                 if (start_command(conn))
1440                         die(_("unable to fork"));
1441
1442                 fd[0] = conn->out; /* read from child's stdout */
1443                 fd[1] = conn->in;  /* write to child's stdin */
1444                 strbuf_release(&cmd);
1445         }
1446         free(hostandport);
1447         free(path);
1448         return conn;
1449 }
1450
1451 int finish_connect(struct child_process *conn)
1452 {
1453         int code;
1454         if (!conn || git_connection_is_socket(conn))
1455                 return 0;
1456
1457         code = finish_command(conn);
1458         free(conn);
1459         return code;
1460 }