http.c

   1 #include "http.h"
   2
   3 int data_received;
   4 int active_requests = 0;
   5
   6 #ifdef USE_CURL_MULTI
   7 static int max_requests = -1;
   8 static CURLM *curlm;
   9 #endif
  10 #ifndef NO_CURL_EASY_DUPHANDLE
  11 static CURL *curl_default;
  12 #endif
  13 char curl_errorstr[CURL_ERROR_SIZE];
  14
  15 static int curl_ssl_verify = -1;
  16 static char *ssl_cert = NULL;
  17 #if LIBCURL_VERSION_NUM >= 0x070902
  18 static char *ssl_key = NULL;
  19 #endif
  20 #if LIBCURL_VERSION_NUM >= 0x070908
  21 static char *ssl_capath = NULL;
  22 #endif
  23 static char *ssl_cainfo = NULL;
  24 static long curl_low_speed_limit = -1;
  25 static long curl_low_speed_time = -1;
  26 static int curl_ftp_no_epsv = 0;
  27 static char *curl_http_proxy = NULL;
  28
  29 static struct curl_slist *pragma_header;
  30
  31 static struct active_request_slot *active_queue_head = NULL;
  32
  33 size_t fread_buffer(void *ptr, size_t eltsize, size_t nmemb,
  34                            struct buffer *buffer)
  35 {
  36         size_t size = eltsize * nmemb;
  37         if (size > buffer->buf.len - buffer->posn)
  38                 size = buffer->buf.len - buffer->posn;
  39         memcpy(ptr, buffer->buf.buf + buffer->posn, size);
  40         buffer->posn += size;
  41
  42         return size;
  43 }
  44
  45 size_t fwrite_buffer(const void *ptr, size_t eltsize,
  46                             size_t nmemb, struct strbuf *buffer)
  47 {
  48         size_t size = eltsize * nmemb;
  49         strbuf_add(buffer, ptr, size);
  50         data_received++;
  51         return size;
  52 }
  53
  54 size_t fwrite_null(const void *ptr, size_t eltsize,
  55                           size_t nmemb, struct strbuf *buffer)
  56 {
  57         data_received++;
  58         return eltsize * nmemb;
  59 }
  60
  61 static void finish_active_slot(struct active_request_slot *slot);
  62
  63 #ifdef USE_CURL_MULTI
  64 static void process_curl_messages(void)
  65 {
  66         int num_messages;
  67         struct active_request_slot *slot;
  68         CURLMsg *curl_message = curl_multi_info_read(curlm, &num_messages);
  69
  70         while (curl_message != NULL) {
  71                 if (curl_message->msg == CURLMSG_DONE) {
  72                         int curl_result = curl_message->data.result;
  73                         slot = active_queue_head;
  74                         while (slot != NULL &&
  75                                slot->curl != curl_message->easy_handle)
  76                                 slot = slot->next;
  77                         if (slot != NULL) {
  78                                 curl_multi_remove_handle(curlm, slot->curl);
  79                                 slot->curl_result = curl_result;
  80                                 finish_active_slot(slot);
  81                         } else {
  82                                 fprintf(stderr, "Received DONE message for unknown request!\n");
  83                         }
  84                 } else {
  85                         fprintf(stderr, "Unknown CURL message received: %d\n",
  86                                 (int)curl_message->msg);
  87                 }
  88                 curl_message = curl_multi_info_read(curlm, &num_messages);
  89         }
  90 }
  91 #endif
  92
  93 static int http_options(const char *var, const char *value)
  94 {
  95         if (!strcmp("http.sslverify", var)) {
  96                 if (curl_ssl_verify == -1) {
  97                         curl_ssl_verify = git_config_bool(var, value);
  98                 }
  99                 return 0;
 100         }
 101
 102         if (!strcmp("http.sslcert", var)) {
 103                 if (ssl_cert == NULL) {
 104                         if (!value)
 105                                 return config_error_nonbool(var);
 106                         ssl_cert = xstrdup(value);
 107                 }
 108                 return 0;
 109         }
 110 #if LIBCURL_VERSION_NUM >= 0x070902
 111         if (!strcmp("http.sslkey", var)) {
 112                 if (ssl_key == NULL) {
 113                         if (!value)
 114                                 return config_error_nonbool(var);
 115                         ssl_key = xstrdup(value);
 116                 }
 117                 return 0;
 118         }
 119 #endif
 120 #if LIBCURL_VERSION_NUM >= 0x070908
 121         if (!strcmp("http.sslcapath", var)) {
 122                 if (ssl_capath == NULL) {
 123                         if (!value)
 124                                 return config_error_nonbool(var);
 125                         ssl_capath = xstrdup(value);
 126                 }
 127                 return 0;
 128         }
 129 #endif
 130         if (!strcmp("http.sslcainfo", var)) {
 131                 if (ssl_cainfo == NULL) {
 132                         if (!value)
 133                                 return config_error_nonbool(var);
 134                         ssl_cainfo = xstrdup(value);
 135                 }
 136                 return 0;
 137         }
 138
 139 #ifdef USE_CURL_MULTI
 140         if (!strcmp("http.maxrequests", var)) {
 141                 if (max_requests == -1)
 142                         max_requests = git_config_int(var, value);
 143                 return 0;
 144         }
 145 #endif
 146
 147         if (!strcmp("http.lowspeedlimit", var)) {
 148                 if (curl_low_speed_limit == -1)
 149                         curl_low_speed_limit = (long)git_config_int(var, value);
 150                 return 0;
 151         }
 152         if (!strcmp("http.lowspeedtime", var)) {
 153                 if (curl_low_speed_time == -1)
 154                         curl_low_speed_time = (long)git_config_int(var, value);
 155                 return 0;
 156         }
 157
 158         if (!strcmp("http.noepsv", var)) {
 159                 curl_ftp_no_epsv = git_config_bool(var, value);
 160                 return 0;
 161         }
 162         if (!strcmp("http.proxy", var)) {
 163                 if (curl_http_proxy == NULL) {
 164                         if (!value)
 165                                 return config_error_nonbool(var);
 166                         curl_http_proxy = xstrdup(value);
 167                 }
 168                 return 0;
 169         }
 170
 171         /* Fall back on the default ones */
 172         return git_default_config(var, value);
 173 }
 174
 175 static CURL* get_curl_handle(void)
 176 {
 177         CURL* result = curl_easy_init();
 178
 179         curl_easy_setopt(result, CURLOPT_SSL_VERIFYPEER, curl_ssl_verify);
 180 #if LIBCURL_VERSION_NUM >= 0x070907
 181         curl_easy_setopt(result, CURLOPT_NETRC, CURL_NETRC_OPTIONAL);
 182 #endif
 183
 184         if (ssl_cert != NULL)
 185                 curl_easy_setopt(result, CURLOPT_SSLCERT, ssl_cert);
 186 #if LIBCURL_VERSION_NUM >= 0x070902
 187         if (ssl_key != NULL)
 188                 curl_easy_setopt(result, CURLOPT_SSLKEY, ssl_key);
 189 #endif
 190 #if LIBCURL_VERSION_NUM >= 0x070908
 191         if (ssl_capath != NULL)
 192                 curl_easy_setopt(result, CURLOPT_CAPATH, ssl_capath);
 193 #endif
 194         if (ssl_cainfo != NULL)
 195                 curl_easy_setopt(result, CURLOPT_CAINFO, ssl_cainfo);
 196         curl_easy_setopt(result, CURLOPT_FAILONERROR, 1);
 197
 198         if (curl_low_speed_limit > 0 && curl_low_speed_time > 0) {
 199                 curl_easy_setopt(result, CURLOPT_LOW_SPEED_LIMIT,
 200                                  curl_low_speed_limit);
 201                 curl_easy_setopt(result, CURLOPT_LOW_SPEED_TIME,
 202                                  curl_low_speed_time);
 203         }
 204
 205         curl_easy_setopt(result, CURLOPT_FOLLOWLOCATION, 1);
 206
 207         if (getenv("GIT_CURL_VERBOSE"))
 208                 curl_easy_setopt(result, CURLOPT_VERBOSE, 1);
 209
 210         curl_easy_setopt(result, CURLOPT_USERAGENT, GIT_USER_AGENT);
 211
 212         if (curl_ftp_no_epsv)
 213                 curl_easy_setopt(result, CURLOPT_FTP_USE_EPSV, 0);
 214
 215         if (curl_http_proxy)
 216                 curl_easy_setopt(result, CURLOPT_PROXY, curl_http_proxy);
 217
 218         return result;
 219 }
 220
 221 void http_init(struct remote *remote)
 222 {
 223         char *low_speed_limit;
 224         char *low_speed_time;
 225
 226         curl_global_init(CURL_GLOBAL_ALL);
 227
 228         if (remote && remote->http_proxy)
 229                 curl_http_proxy = xstrdup(remote->http_proxy);
 230
 231         pragma_header = curl_slist_append(pragma_header, "Pragma: no-cache");
 232
 233 #ifdef USE_CURL_MULTI
 234         {
 235                 char *http_max_requests = getenv("GIT_HTTP_MAX_REQUESTS");
 236                 if (http_max_requests != NULL)
 237                         max_requests = atoi(http_max_requests);
 238         }
 239
 240         curlm = curl_multi_init();
 241         if (curlm == NULL) {
 242                 fprintf(stderr, "Error creating curl multi handle.\n");
 243                 exit(1);
 244         }
 245 #endif
 246
 247         if (getenv("GIT_SSL_NO_VERIFY"))
 248                 curl_ssl_verify = 0;
 249
 250         ssl_cert = getenv("GIT_SSL_CERT");
 251 #if LIBCURL_VERSION_NUM >= 0x070902
 252         ssl_key = getenv("GIT_SSL_KEY");
 253 #endif
 254 #if LIBCURL_VERSION_NUM >= 0x070908
 255         ssl_capath = getenv("GIT_SSL_CAPATH");
 256 #endif
 257         ssl_cainfo = getenv("GIT_SSL_CAINFO");
 258
 259         low_speed_limit = getenv("GIT_HTTP_LOW_SPEED_LIMIT");
 260         if (low_speed_limit != NULL)
 261                 curl_low_speed_limit = strtol(low_speed_limit, NULL, 10);
 262         low_speed_time = getenv("GIT_HTTP_LOW_SPEED_TIME");
 263         if (low_speed_time != NULL)
 264                 curl_low_speed_time = strtol(low_speed_time, NULL, 10);
 265
 266         git_config(http_options);
 267
 268         if (curl_ssl_verify == -1)
 269                 curl_ssl_verify = 1;
 270
 271 #ifdef USE_CURL_MULTI
 272         if (max_requests < 1)
 273                 max_requests = DEFAULT_MAX_REQUESTS;
 274 #endif
 275
 276         if (getenv("GIT_CURL_FTP_NO_EPSV"))
 277                 curl_ftp_no_epsv = 1;
 278
 279 #ifndef NO_CURL_EASY_DUPHANDLE
 280         curl_default = get_curl_handle();
 281 #endif
 282 }
 283
 284 void http_cleanup(void)
 285 {
 286         struct active_request_slot *slot = active_queue_head;
 287 #ifdef USE_CURL_MULTI
 288         char *wait_url;
 289 #endif
 290
 291         while (slot != NULL) {
 292                 struct active_request_slot *next = slot->next;
 293 #ifdef USE_CURL_MULTI
 294                 if (slot->in_use) {
 295                         curl_easy_getinfo(slot->curl,
 296                                           CURLINFO_EFFECTIVE_URL,
 297                                           &wait_url);
 298                         fprintf(stderr, "Waiting for %s\n", wait_url);
 299                         run_active_slot(slot);
 300                 }
 301 #endif
 302                 if (slot->curl != NULL)
 303                         curl_easy_cleanup(slot->curl);
 304                 free(slot);
 305                 slot = next;
 306         }
 307         active_queue_head = NULL;
 308
 309 #ifndef NO_CURL_EASY_DUPHANDLE
 310         curl_easy_cleanup(curl_default);
 311 #endif
 312
 313 #ifdef USE_CURL_MULTI
 314         curl_multi_cleanup(curlm);
 315 #endif
 316         curl_global_cleanup();
 317
 318         curl_slist_free_all(pragma_header);
 319         pragma_header = NULL;
 320
 321         if (curl_http_proxy) {
 322                 free(curl_http_proxy);
 323                 curl_http_proxy = NULL;
 324         }
 325 }
 326
 327 struct active_request_slot *get_active_slot(void)
 328 {
 329         struct active_request_slot *slot = active_queue_head;
 330         struct active_request_slot *newslot;
 331
 332 #ifdef USE_CURL_MULTI
 333         int num_transfers;
 334
 335         /* Wait for a slot to open up if the queue is full */
 336         while (active_requests >= max_requests) {
 337                 curl_multi_perform(curlm, &num_transfers);
 338                 if (num_transfers < active_requests) {
 339                         process_curl_messages();
 340                 }
 341         }
 342 #endif
 343
 344         while (slot != NULL && slot->in_use) {
 345                 slot = slot->next;
 346         }
 347         if (slot == NULL) {
 348                 newslot = xmalloc(sizeof(*newslot));
 349                 newslot->curl = NULL;
 350                 newslot->in_use = 0;
 351                 newslot->next = NULL;
 352
 353                 slot = active_queue_head;
 354                 if (slot == NULL) {
 355                         active_queue_head = newslot;
 356                 } else {
 357                         while (slot->next != NULL) {
 358                                 slot = slot->next;
 359                         }
 360                         slot->next = newslot;
 361                 }
 362                 slot = newslot;
 363         }
 364
 365         if (slot->curl == NULL) {
 366 #ifdef NO_CURL_EASY_DUPHANDLE
 367                 slot->curl = get_curl_handle();
 368 #else
 369                 slot->curl = curl_easy_duphandle(curl_default);
 370 #endif
 371         }
 372
 373         active_requests++;
 374         slot->in_use = 1;
 375         slot->local = NULL;
 376         slot->results = NULL;
 377         slot->finished = NULL;
 378         slot->callback_data = NULL;
 379         slot->callback_func = NULL;
 380         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, pragma_header);
 381         curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, curl_errorstr);
 382         curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, NULL);
 383         curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, NULL);
 384         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, NULL);
 385         curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 0);
 386         curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1);
 387
 388         return slot;
 389 }
 390
 391 int start_active_slot(struct active_request_slot *slot)
 392 {
 393 #ifdef USE_CURL_MULTI
 394         CURLMcode curlm_result = curl_multi_add_handle(curlm, slot->curl);
 395         int num_transfers;
 396
 397         if (curlm_result != CURLM_OK &&
 398             curlm_result != CURLM_CALL_MULTI_PERFORM) {
 399                 active_requests--;
 400                 slot->in_use = 0;
 401                 return 0;
 402         }
 403
 404         /*
 405          * We know there must be something to do, since we just added
 406          * something.
 407          */
 408         curl_multi_perform(curlm, &num_transfers);
 409 #endif
 410         return 1;
 411 }
 412
 413 #ifdef USE_CURL_MULTI
 414 struct fill_chain {
 415         void *data;
 416         int (*fill)(void *);
 417         struct fill_chain *next;
 418 };
 419
 420 static struct fill_chain *fill_cfg = NULL;
 421
 422 void add_fill_function(void *data, int (*fill)(void *))
 423 {
 424         struct fill_chain *new = malloc(sizeof(*new));
 425         struct fill_chain **linkp = &fill_cfg;
 426         new->data = data;
 427         new->fill = fill;
 428         new->next = NULL;
 429         while (*linkp)
 430                 linkp = &(*linkp)->next;
 431         *linkp = new;
 432 }
 433
 434 void fill_active_slots(void)
 435 {
 436         struct active_request_slot *slot = active_queue_head;
 437
 438         while (active_requests < max_requests) {
 439                 struct fill_chain *fill;
 440                 for (fill = fill_cfg; fill; fill = fill->next)
 441                         if (fill->fill(fill->data))
 442                                 break;
 443
 444                 if (!fill)
 445                         break;
 446         }
 447
 448         while (slot != NULL) {
 449                 if (!slot->in_use && slot->curl != NULL) {
 450                         curl_easy_cleanup(slot->curl);
 451                         slot->curl = NULL;
 452                 }
 453                 slot = slot->next;
 454         }
 455 }
 456
 457 void step_active_slots(void)
 458 {
 459         int num_transfers;
 460         CURLMcode curlm_result;
 461
 462         do {
 463                 curlm_result = curl_multi_perform(curlm, &num_transfers);
 464         } while (curlm_result == CURLM_CALL_MULTI_PERFORM);
 465         if (num_transfers < active_requests) {
 466                 process_curl_messages();
 467                 fill_active_slots();
 468         }
 469 }
 470 #endif
 471
 472 void run_active_slot(struct active_request_slot *slot)
 473 {
 474 #ifdef USE_CURL_MULTI
 475         long last_pos = 0;
 476         long current_pos;
 477         fd_set readfds;
 478         fd_set writefds;
 479         fd_set excfds;
 480         int max_fd;
 481         struct timeval select_timeout;
 482         int finished = 0;
 483
 484         slot->finished = &finished;
 485         while (!finished) {
 486                 data_received = 0;
 487                 step_active_slots();
 488
 489                 if (!data_received && slot->local != NULL) {
 490                         current_pos = ftell(slot->local);
 491                         if (current_pos > last_pos)
 492                                 data_received++;
 493                         last_pos = current_pos;
 494                 }
 495
 496                 if (slot->in_use && !data_received) {
 497                         max_fd = 0;
 498                         FD_ZERO(&readfds);
 499                         FD_ZERO(&writefds);
 500                         FD_ZERO(&excfds);
 501                         select_timeout.tv_sec = 0;
 502                         select_timeout.tv_usec = 50000;
 503                         select(max_fd, &readfds, &writefds,
 504                                &excfds, &select_timeout);
 505                 }
 506         }
 507 #else
 508         while (slot->in_use) {
 509                 slot->curl_result = curl_easy_perform(slot->curl);
 510                 finish_active_slot(slot);
 511         }
 512 #endif
 513 }
 514
 515 static void closedown_active_slot(struct active_request_slot *slot)
 516 {
 517         active_requests--;
 518         slot->in_use = 0;
 519 }
 520
 521 void release_active_slot(struct active_request_slot *slot)
 522 {
 523         closedown_active_slot(slot);
 524         if (slot->curl) {
 525 #ifdef USE_CURL_MULTI
 526                 curl_multi_remove_handle(curlm, slot->curl);
 527 #endif
 528                 curl_easy_cleanup(slot->curl);
 529                 slot->curl = NULL;
 530         }
 531 #ifdef USE_CURL_MULTI
 532         fill_active_slots();
 533 #endif
 534 }
 535
 536 static void finish_active_slot(struct active_request_slot *slot)
 537 {
 538         closedown_active_slot(slot);
 539         curl_easy_getinfo(slot->curl, CURLINFO_HTTP_CODE, &slot->http_code);
 540
 541         if (slot->finished != NULL)
 542                 (*slot->finished) = 1;
 543
 544         /* Store slot results so they can be read after the slot is reused */
 545         if (slot->results != NULL) {
 546                 slot->results->curl_result = slot->curl_result;
 547                 slot->results->http_code = slot->http_code;
 548         }
 549
 550         /* Run callback if appropriate */
 551         if (slot->callback_func != NULL) {
 552                 slot->callback_func(slot->callback_data);
 553         }
 554 }
 555
 556 void finish_all_active_slots(void)
 557 {
 558         struct active_request_slot *slot = active_queue_head;
 559
 560         while (slot != NULL)
 561                 if (slot->in_use) {
 562                         run_active_slot(slot);
 563                         slot = active_queue_head;
 564                 } else {
 565                         slot = slot->next;
 566                 }
 567 }
 568
 569 static inline int needs_quote(int ch)
 570 {
 571         if (((ch >= 'A') && (ch <= 'Z'))
 572                         || ((ch >= 'a') && (ch <= 'z'))
 573                         || ((ch >= '0') && (ch <= '9'))
 574                         || (ch == '/')
 575                         || (ch == '-')
 576                         || (ch == '.'))
 577                 return 0;
 578         return 1;
 579 }
 580
 581 static inline int hex(int v)
 582 {
 583         if (v < 10) return '0' + v;
 584         else return 'A' + v - 10;
 585 }
 586
 587 static char *quote_ref_url(const char *base, const char *ref)
 588 {
 589         const char *cp;
 590         char *dp, *qref;
 591         int len, baselen, ch;
 592
 593         baselen = strlen(base);
 594         len = baselen + 7; /* "/refs/" + NUL */
 595         for (cp = ref; (ch = *cp) != 0; cp++, len++)
 596                 if (needs_quote(ch))
 597                         len += 2; /* extra two hex plus replacement % */
 598         qref = xmalloc(len);
 599         memcpy(qref, base, baselen);
 600         memcpy(qref + baselen, "/refs/", 6);
 601         for (cp = ref, dp = qref + baselen + 6; (ch = *cp) != 0; cp++) {
 602                 if (needs_quote(ch)) {
 603                         *dp++ = '%';
 604                         *dp++ = hex((ch >> 4) & 0xF);
 605                         *dp++ = hex(ch & 0xF);
 606                 }
 607                 else
 608                         *dp++ = ch;
 609         }
 610         *dp = 0;
 611
 612         return qref;
 613 }
 614
 615 int http_fetch_ref(const char *base, const char *ref, unsigned char *sha1)
 616 {
 617         char *url;
 618         struct strbuf buffer = STRBUF_INIT;
 619         struct active_request_slot *slot;
 620         struct slot_results results;
 621         int ret;
 622
 623         url = quote_ref_url(base, ref);
 624         slot = get_active_slot();
 625         slot->results = &results;
 626         curl_easy_setopt(slot->curl, CURLOPT_FILE, &buffer);
 627         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
 628         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, NULL);
 629         curl_easy_setopt(slot->curl, CURLOPT_URL, url);
 630         if (start_active_slot(slot)) {
 631                 run_active_slot(slot);
 632                 if (results.curl_result == CURLE_OK) {
 633                         strbuf_rtrim(&buffer);
 634                         if (buffer.len == 40)
 635                                 ret = get_sha1_hex(buffer.buf, sha1);
 636                         else
 637                                 ret = 1;
 638                 } else {
 639                         ret = error("Couldn't get %s for %s\n%s",
 640                                     url, ref, curl_errorstr);
 641                 }
 642         } else {
 643                 ret = error("Unable to start request");
 644         }
 645
 646         strbuf_release(&buffer);
 647         free(url);
 648         return ret;
 649 }