Code

http.c: Use timeout suggested by curl instead of fixed 50ms timeout
[git.git] / http.c
1 #include "http.h"
2 #include "pack.h"
3 #include "sideband.h"
4 #include "run-command.h"
5 #include "url.h"
7 int data_received;
8 int active_requests;
9 int http_is_verbose;
10 size_t http_post_buffer = 16 * LARGE_PACKET_MAX;
12 #if LIBCURL_VERSION_NUM >= 0x070a06
13 #define LIBCURL_CAN_HANDLE_AUTH_ANY
14 #endif
16 static int min_curl_sessions = 1;
17 static int curl_session_count;
18 #ifdef USE_CURL_MULTI
19 static int max_requests = -1;
20 static CURLM *curlm;
21 #endif
22 #ifndef NO_CURL_EASY_DUPHANDLE
23 static CURL *curl_default;
24 #endif
26 #define PREV_BUF_SIZE 4096
27 #define RANGE_HEADER_SIZE 30
29 char curl_errorstr[CURL_ERROR_SIZE];
31 static int curl_ssl_verify = -1;
32 static const char *ssl_cert;
33 #if LIBCURL_VERSION_NUM >= 0x070903
34 static const char *ssl_key;
35 #endif
36 #if LIBCURL_VERSION_NUM >= 0x070908
37 static const char *ssl_capath;
38 #endif
39 static const char *ssl_cainfo;
40 static long curl_low_speed_limit = -1;
41 static long curl_low_speed_time = -1;
42 static int curl_ftp_no_epsv;
43 static const char *curl_http_proxy;
44 static char *user_name, *user_pass;
45 static const char *user_agent;
47 #if LIBCURL_VERSION_NUM >= 0x071700
48 /* Use CURLOPT_KEYPASSWD as is */
49 #elif LIBCURL_VERSION_NUM >= 0x070903
50 #define CURLOPT_KEYPASSWD CURLOPT_SSLKEYPASSWD
51 #else
52 #define CURLOPT_KEYPASSWD CURLOPT_SSLCERTPASSWD
53 #endif
55 static char *ssl_cert_password;
56 static int ssl_cert_password_required;
58 static struct curl_slist *pragma_header;
59 static struct curl_slist *no_pragma_header;
61 static struct active_request_slot *active_queue_head;
63 size_t fread_buffer(char *ptr, size_t eltsize, size_t nmemb, void *buffer_)
64 {
65         size_t size = eltsize * nmemb;
66         struct buffer *buffer = buffer_;
68         if (size > buffer->buf.len - buffer->posn)
69                 size = buffer->buf.len - buffer->posn;
70         memcpy(ptr, buffer->buf.buf + buffer->posn, size);
71         buffer->posn += size;
73         return size;
74 }
76 #ifndef NO_CURL_IOCTL
77 curlioerr ioctl_buffer(CURL *handle, int cmd, void *clientp)
78 {
79         struct buffer *buffer = clientp;
81         switch (cmd) {
82         case CURLIOCMD_NOP:
83                 return CURLIOE_OK;
85         case CURLIOCMD_RESTARTREAD:
86                 buffer->posn = 0;
87                 return CURLIOE_OK;
89         default:
90                 return CURLIOE_UNKNOWNCMD;
91         }
92 }
93 #endif
95 size_t fwrite_buffer(char *ptr, size_t eltsize, size_t nmemb, void *buffer_)
96 {
97         size_t size = eltsize * nmemb;
98         struct strbuf *buffer = buffer_;
100         strbuf_add(buffer, ptr, size);
101         data_received++;
102         return size;
105 size_t fwrite_null(char *ptr, size_t eltsize, size_t nmemb, void *strbuf)
107         data_received++;
108         return eltsize * nmemb;
111 #ifdef USE_CURL_MULTI
112 static void process_curl_messages(void)
114         int num_messages;
115         struct active_request_slot *slot;
116         CURLMsg *curl_message = curl_multi_info_read(curlm, &num_messages);
118         while (curl_message != NULL) {
119                 if (curl_message->msg == CURLMSG_DONE) {
120                         int curl_result = curl_message->data.result;
121                         slot = active_queue_head;
122                         while (slot != NULL &&
123                                slot->curl != curl_message->easy_handle)
124                                 slot = slot->next;
125                         if (slot != NULL) {
126                                 curl_multi_remove_handle(curlm, slot->curl);
127                                 slot->curl_result = curl_result;
128                                 finish_active_slot(slot);
129                         } else {
130                                 fprintf(stderr, "Received DONE message for unknown request!\n");
131                         }
132                 } else {
133                         fprintf(stderr, "Unknown CURL message received: %d\n",
134                                 (int)curl_message->msg);
135                 }
136                 curl_message = curl_multi_info_read(curlm, &num_messages);
137         }
139 #endif
141 static int http_options(const char *var, const char *value, void *cb)
143         if (!strcmp("http.sslverify", var)) {
144                 curl_ssl_verify = git_config_bool(var, value);
145                 return 0;
146         }
147         if (!strcmp("http.sslcert", var))
148                 return git_config_string(&ssl_cert, var, value);
149 #if LIBCURL_VERSION_NUM >= 0x070903
150         if (!strcmp("http.sslkey", var))
151                 return git_config_string(&ssl_key, var, value);
152 #endif
153 #if LIBCURL_VERSION_NUM >= 0x070908
154         if (!strcmp("http.sslcapath", var))
155                 return git_config_string(&ssl_capath, var, value);
156 #endif
157         if (!strcmp("http.sslcainfo", var))
158                 return git_config_string(&ssl_cainfo, var, value);
159         if (!strcmp("http.sslcertpasswordprotected", var)) {
160                 if (git_config_bool(var, value))
161                         ssl_cert_password_required = 1;
162                 return 0;
163         }
164         if (!strcmp("http.minsessions", var)) {
165                 min_curl_sessions = git_config_int(var, value);
166 #ifndef USE_CURL_MULTI
167                 if (min_curl_sessions > 1)
168                         min_curl_sessions = 1;
169 #endif
170                 return 0;
171         }
172 #ifdef USE_CURL_MULTI
173         if (!strcmp("http.maxrequests", var)) {
174                 max_requests = git_config_int(var, value);
175                 return 0;
176         }
177 #endif
178         if (!strcmp("http.lowspeedlimit", var)) {
179                 curl_low_speed_limit = (long)git_config_int(var, value);
180                 return 0;
181         }
182         if (!strcmp("http.lowspeedtime", var)) {
183                 curl_low_speed_time = (long)git_config_int(var, value);
184                 return 0;
185         }
187         if (!strcmp("http.noepsv", var)) {
188                 curl_ftp_no_epsv = git_config_bool(var, value);
189                 return 0;
190         }
191         if (!strcmp("http.proxy", var))
192                 return git_config_string(&curl_http_proxy, var, value);
194         if (!strcmp("http.postbuffer", var)) {
195                 http_post_buffer = git_config_int(var, value);
196                 if (http_post_buffer < LARGE_PACKET_MAX)
197                         http_post_buffer = LARGE_PACKET_MAX;
198                 return 0;
199         }
201         if (!strcmp("http.useragent", var))
202                 return git_config_string(&user_agent, var, value);
204         /* Fall back on the default ones */
205         return git_default_config(var, value, cb);
208 static void init_curl_http_auth(CURL *result)
210         if (user_name) {
211                 struct strbuf up = STRBUF_INIT;
212                 if (!user_pass)
213                         user_pass = xstrdup(git_getpass("Password: "));
214                 strbuf_addf(&up, "%s:%s", user_name, user_pass);
215                 curl_easy_setopt(result, CURLOPT_USERPWD,
216                                  strbuf_detach(&up, NULL));
217         }
220 static int has_cert_password(void)
222         if (ssl_cert_password != NULL)
223                 return 1;
224         if (ssl_cert == NULL || ssl_cert_password_required != 1)
225                 return 0;
226         /* Only prompt the user once. */
227         ssl_cert_password_required = -1;
228         ssl_cert_password = git_getpass("Certificate Password: ");
229         if (ssl_cert_password != NULL) {
230                 ssl_cert_password = xstrdup(ssl_cert_password);
231                 return 1;
232         } else
233                 return 0;
236 static CURL *get_curl_handle(void)
238         CURL *result = curl_easy_init();
240         if (!curl_ssl_verify) {
241                 curl_easy_setopt(result, CURLOPT_SSL_VERIFYPEER, 0);
242                 curl_easy_setopt(result, CURLOPT_SSL_VERIFYHOST, 0);
243         } else {
244                 /* Verify authenticity of the peer's certificate */
245                 curl_easy_setopt(result, CURLOPT_SSL_VERIFYPEER, 1);
246                 /* The name in the cert must match whom we tried to connect */
247                 curl_easy_setopt(result, CURLOPT_SSL_VERIFYHOST, 2);
248         }
250 #if LIBCURL_VERSION_NUM >= 0x070907
251         curl_easy_setopt(result, CURLOPT_NETRC, CURL_NETRC_OPTIONAL);
252 #endif
253 #ifdef LIBCURL_CAN_HANDLE_AUTH_ANY
254         curl_easy_setopt(result, CURLOPT_HTTPAUTH, CURLAUTH_ANY);
255 #endif
257         init_curl_http_auth(result);
259         if (ssl_cert != NULL)
260                 curl_easy_setopt(result, CURLOPT_SSLCERT, ssl_cert);
261         if (has_cert_password())
262                 curl_easy_setopt(result, CURLOPT_KEYPASSWD, ssl_cert_password);
263 #if LIBCURL_VERSION_NUM >= 0x070903
264         if (ssl_key != NULL)
265                 curl_easy_setopt(result, CURLOPT_SSLKEY, ssl_key);
266 #endif
267 #if LIBCURL_VERSION_NUM >= 0x070908
268         if (ssl_capath != NULL)
269                 curl_easy_setopt(result, CURLOPT_CAPATH, ssl_capath);
270 #endif
271         if (ssl_cainfo != NULL)
272                 curl_easy_setopt(result, CURLOPT_CAINFO, ssl_cainfo);
273         curl_easy_setopt(result, CURLOPT_FAILONERROR, 1);
275         if (curl_low_speed_limit > 0 && curl_low_speed_time > 0) {
276                 curl_easy_setopt(result, CURLOPT_LOW_SPEED_LIMIT,
277                                  curl_low_speed_limit);
278                 curl_easy_setopt(result, CURLOPT_LOW_SPEED_TIME,
279                                  curl_low_speed_time);
280         }
282         curl_easy_setopt(result, CURLOPT_FOLLOWLOCATION, 1);
283 #if LIBCURL_VERSION_NUM >= 0x071301
284         curl_easy_setopt(result, CURLOPT_POSTREDIR, CURL_REDIR_POST_ALL);
285 #elif LIBCURL_VERSION_NUM >= 0x071101
286         curl_easy_setopt(result, CURLOPT_POST301, 1);
287 #endif
289         if (getenv("GIT_CURL_VERBOSE"))
290                 curl_easy_setopt(result, CURLOPT_VERBOSE, 1);
292         curl_easy_setopt(result, CURLOPT_USERAGENT,
293                 user_agent ? user_agent : GIT_HTTP_USER_AGENT);
295         if (curl_ftp_no_epsv)
296                 curl_easy_setopt(result, CURLOPT_FTP_USE_EPSV, 0);
298         if (curl_http_proxy)
299                 curl_easy_setopt(result, CURLOPT_PROXY, curl_http_proxy);
301         return result;
304 static void http_auth_init(const char *url)
306         char *at, *colon, *cp, *slash, *decoded;
307         int len;
309         cp = strstr(url, "://");
310         if (!cp)
311                 return;
313         /*
314          * Ok, the URL looks like "proto://something".  Which one?
315          * "proto://<user>:<pass>@<host>/...",
316          * "proto://<user>@<host>/...", or just
317          * "proto://<host>/..."?
318          */
319         cp += 3;
320         at = strchr(cp, '@');
321         colon = strchr(cp, ':');
322         slash = strchrnul(cp, '/');
323         if (!at || slash <= at)
324                 return; /* No credentials */
325         if (!colon || at <= colon) {
326                 /* Only username */
327                 len = at - cp;
328                 user_name = xmalloc(len + 1);
329                 memcpy(user_name, cp, len);
330                 user_name[len] = '\0';
331                 decoded = url_decode(user_name);
332                 free(user_name);
333                 user_name = decoded;
334                 user_pass = NULL;
335         } else {
336                 len = colon - cp;
337                 user_name = xmalloc(len + 1);
338                 memcpy(user_name, cp, len);
339                 user_name[len] = '\0';
340                 decoded = url_decode(user_name);
341                 free(user_name);
342                 user_name = decoded;
343                 len = at - (colon + 1);
344                 user_pass = xmalloc(len + 1);
345                 memcpy(user_pass, colon + 1, len);
346                 user_pass[len] = '\0';
347                 decoded = url_decode(user_pass);
348                 free(user_pass);
349                 user_pass = decoded;
350         }
353 static void set_from_env(const char **var, const char *envname)
355         const char *val = getenv(envname);
356         if (val)
357                 *var = val;
360 void http_init(struct remote *remote)
362         char *low_speed_limit;
363         char *low_speed_time;
365         http_is_verbose = 0;
367         git_config(http_options, NULL);
369         curl_global_init(CURL_GLOBAL_ALL);
371         if (remote && remote->http_proxy)
372                 curl_http_proxy = xstrdup(remote->http_proxy);
374         pragma_header = curl_slist_append(pragma_header, "Pragma: no-cache");
375         no_pragma_header = curl_slist_append(no_pragma_header, "Pragma:");
377 #ifdef USE_CURL_MULTI
378         {
379                 char *http_max_requests = getenv("GIT_HTTP_MAX_REQUESTS");
380                 if (http_max_requests != NULL)
381                         max_requests = atoi(http_max_requests);
382         }
384         curlm = curl_multi_init();
385         if (curlm == NULL) {
386                 fprintf(stderr, "Error creating curl multi handle.\n");
387                 exit(1);
388         }
389 #endif
391         if (getenv("GIT_SSL_NO_VERIFY"))
392                 curl_ssl_verify = 0;
394         set_from_env(&ssl_cert, "GIT_SSL_CERT");
395 #if LIBCURL_VERSION_NUM >= 0x070903
396         set_from_env(&ssl_key, "GIT_SSL_KEY");
397 #endif
398 #if LIBCURL_VERSION_NUM >= 0x070908
399         set_from_env(&ssl_capath, "GIT_SSL_CAPATH");
400 #endif
401         set_from_env(&ssl_cainfo, "GIT_SSL_CAINFO");
403         set_from_env(&user_agent, "GIT_HTTP_USER_AGENT");
405         low_speed_limit = getenv("GIT_HTTP_LOW_SPEED_LIMIT");
406         if (low_speed_limit != NULL)
407                 curl_low_speed_limit = strtol(low_speed_limit, NULL, 10);
408         low_speed_time = getenv("GIT_HTTP_LOW_SPEED_TIME");
409         if (low_speed_time != NULL)
410                 curl_low_speed_time = strtol(low_speed_time, NULL, 10);
412         if (curl_ssl_verify == -1)
413                 curl_ssl_verify = 1;
415         curl_session_count = 0;
416 #ifdef USE_CURL_MULTI
417         if (max_requests < 1)
418                 max_requests = DEFAULT_MAX_REQUESTS;
419 #endif
421         if (getenv("GIT_CURL_FTP_NO_EPSV"))
422                 curl_ftp_no_epsv = 1;
424         if (remote && remote->url && remote->url[0]) {
425                 http_auth_init(remote->url[0]);
426                 if (!ssl_cert_password_required &&
427                     getenv("GIT_SSL_CERT_PASSWORD_PROTECTED") &&
428                     !prefixcmp(remote->url[0], "https://"))
429                         ssl_cert_password_required = 1;
430         }
432 #ifndef NO_CURL_EASY_DUPHANDLE
433         curl_default = get_curl_handle();
434 #endif
437 void http_cleanup(void)
439         struct active_request_slot *slot = active_queue_head;
441         while (slot != NULL) {
442                 struct active_request_slot *next = slot->next;
443                 if (slot->curl != NULL) {
444 #ifdef USE_CURL_MULTI
445                         curl_multi_remove_handle(curlm, slot->curl);
446 #endif
447                         curl_easy_cleanup(slot->curl);
448                 }
449                 free(slot);
450                 slot = next;
451         }
452         active_queue_head = NULL;
454 #ifndef NO_CURL_EASY_DUPHANDLE
455         curl_easy_cleanup(curl_default);
456 #endif
458 #ifdef USE_CURL_MULTI
459         curl_multi_cleanup(curlm);
460 #endif
461         curl_global_cleanup();
463         curl_slist_free_all(pragma_header);
464         pragma_header = NULL;
466         curl_slist_free_all(no_pragma_header);
467         no_pragma_header = NULL;
469         if (curl_http_proxy) {
470                 free((void *)curl_http_proxy);
471                 curl_http_proxy = NULL;
472         }
474         if (ssl_cert_password != NULL) {
475                 memset(ssl_cert_password, 0, strlen(ssl_cert_password));
476                 free(ssl_cert_password);
477                 ssl_cert_password = NULL;
478         }
479         ssl_cert_password_required = 0;
482 struct active_request_slot *get_active_slot(void)
484         struct active_request_slot *slot = active_queue_head;
485         struct active_request_slot *newslot;
487 #ifdef USE_CURL_MULTI
488         int num_transfers;
490         /* Wait for a slot to open up if the queue is full */
491         while (active_requests >= max_requests) {
492                 curl_multi_perform(curlm, &num_transfers);
493                 if (num_transfers < active_requests)
494                         process_curl_messages();
495         }
496 #endif
498         while (slot != NULL && slot->in_use)
499                 slot = slot->next;
501         if (slot == NULL) {
502                 newslot = xmalloc(sizeof(*newslot));
503                 newslot->curl = NULL;
504                 newslot->in_use = 0;
505                 newslot->next = NULL;
507                 slot = active_queue_head;
508                 if (slot == NULL) {
509                         active_queue_head = newslot;
510                 } else {
511                         while (slot->next != NULL)
512                                 slot = slot->next;
513                         slot->next = newslot;
514                 }
515                 slot = newslot;
516         }
518         if (slot->curl == NULL) {
519 #ifdef NO_CURL_EASY_DUPHANDLE
520                 slot->curl = get_curl_handle();
521 #else
522                 slot->curl = curl_easy_duphandle(curl_default);
523 #endif
524                 curl_session_count++;
525         }
527         active_requests++;
528         slot->in_use = 1;
529         slot->local = NULL;
530         slot->results = NULL;
531         slot->finished = NULL;
532         slot->callback_data = NULL;
533         slot->callback_func = NULL;
534         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, pragma_header);
535         curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, curl_errorstr);
536         curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, NULL);
537         curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, NULL);
538         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, NULL);
539         curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDS, NULL);
540         curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 0);
541         curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1);
543         return slot;
546 int start_active_slot(struct active_request_slot *slot)
548 #ifdef USE_CURL_MULTI
549         CURLMcode curlm_result = curl_multi_add_handle(curlm, slot->curl);
550         int num_transfers;
552         if (curlm_result != CURLM_OK &&
553             curlm_result != CURLM_CALL_MULTI_PERFORM) {
554                 active_requests--;
555                 slot->in_use = 0;
556                 return 0;
557         }
559         /*
560          * We know there must be something to do, since we just added
561          * something.
562          */
563         curl_multi_perform(curlm, &num_transfers);
564 #endif
565         return 1;
568 #ifdef USE_CURL_MULTI
569 struct fill_chain {
570         void *data;
571         int (*fill)(void *);
572         struct fill_chain *next;
573 };
575 static struct fill_chain *fill_cfg;
577 void add_fill_function(void *data, int (*fill)(void *))
579         struct fill_chain *new = xmalloc(sizeof(*new));
580         struct fill_chain **linkp = &fill_cfg;
581         new->data = data;
582         new->fill = fill;
583         new->next = NULL;
584         while (*linkp)
585                 linkp = &(*linkp)->next;
586         *linkp = new;
589 void fill_active_slots(void)
591         struct active_request_slot *slot = active_queue_head;
593         while (active_requests < max_requests) {
594                 struct fill_chain *fill;
595                 for (fill = fill_cfg; fill; fill = fill->next)
596                         if (fill->fill(fill->data))
597                                 break;
599                 if (!fill)
600                         break;
601         }
603         while (slot != NULL) {
604                 if (!slot->in_use && slot->curl != NULL
605                         && curl_session_count > min_curl_sessions) {
606                         curl_easy_cleanup(slot->curl);
607                         slot->curl = NULL;
608                         curl_session_count--;
609                 }
610                 slot = slot->next;
611         }
614 void step_active_slots(void)
616         int num_transfers;
617         CURLMcode curlm_result;
619         do {
620                 curlm_result = curl_multi_perform(curlm, &num_transfers);
621         } while (curlm_result == CURLM_CALL_MULTI_PERFORM);
622         if (num_transfers < active_requests) {
623                 process_curl_messages();
624                 fill_active_slots();
625         }
627 #endif
629 void run_active_slot(struct active_request_slot *slot)
631 #ifdef USE_CURL_MULTI
632         long last_pos = 0;
633         long current_pos;
634         fd_set readfds;
635         fd_set writefds;
636         fd_set excfds;
637         int max_fd;
638         struct timeval select_timeout;
639         int finished = 0;
641         slot->finished = &finished;
642         while (!finished) {
643                 data_received = 0;
644                 step_active_slots();
646                 if (!data_received && slot->local != NULL) {
647                         current_pos = ftell(slot->local);
648                         if (current_pos > last_pos)
649                                 data_received++;
650                         last_pos = current_pos;
651                 }
653                 if (slot->in_use && !data_received) {
654 #if LIBCURL_VERSION_NUM >= 0x070f04
655                         long curl_timeout;
656                         curl_multi_timeout(curlm, &curl_timeout);
657                         if (curl_timeout == 0) {
658                                 continue;
659                         } else if (curl_timeout == -1) {
660                                 select_timeout.tv_sec  = 0;
661                                 select_timeout.tv_usec = 50000;
662                         } else {
663                                 select_timeout.tv_sec  =  curl_timeout / 1000;
664                                 select_timeout.tv_usec = (curl_timeout % 1000) * 1000;
665                         }
666 #else
667                         select_timeout.tv_sec  = 0;
668                         select_timeout.tv_usec = 50000;
669 #endif
671                         max_fd = -1;
672                         FD_ZERO(&readfds);
673                         FD_ZERO(&writefds);
674                         FD_ZERO(&excfds);
675                         curl_multi_fdset(curlm, &readfds, &writefds, &excfds, &max_fd);
677                         select(max_fd+1, &readfds, &writefds, &excfds, &select_timeout);
678                 }
679         }
680 #else
681         while (slot->in_use) {
682                 slot->curl_result = curl_easy_perform(slot->curl);
683                 finish_active_slot(slot);
684         }
685 #endif
688 static void closedown_active_slot(struct active_request_slot *slot)
690         active_requests--;
691         slot->in_use = 0;
694 static void release_active_slot(struct active_request_slot *slot)
696         closedown_active_slot(slot);
697         if (slot->curl && curl_session_count > min_curl_sessions) {
698 #ifdef USE_CURL_MULTI
699                 curl_multi_remove_handle(curlm, slot->curl);
700 #endif
701                 curl_easy_cleanup(slot->curl);
702                 slot->curl = NULL;
703                 curl_session_count--;
704         }
705 #ifdef USE_CURL_MULTI
706         fill_active_slots();
707 #endif
710 void finish_active_slot(struct active_request_slot *slot)
712         closedown_active_slot(slot);
713         curl_easy_getinfo(slot->curl, CURLINFO_HTTP_CODE, &slot->http_code);
715         if (slot->finished != NULL)
716                 (*slot->finished) = 1;
718         /* Store slot results so they can be read after the slot is reused */
719         if (slot->results != NULL) {
720                 slot->results->curl_result = slot->curl_result;
721                 slot->results->http_code = slot->http_code;
722         }
724         /* Run callback if appropriate */
725         if (slot->callback_func != NULL)
726                 slot->callback_func(slot->callback_data);
729 void finish_all_active_slots(void)
731         struct active_request_slot *slot = active_queue_head;
733         while (slot != NULL)
734                 if (slot->in_use) {
735                         run_active_slot(slot);
736                         slot = active_queue_head;
737                 } else {
738                         slot = slot->next;
739                 }
742 /* Helpers for modifying and creating URLs */
743 static inline int needs_quote(int ch)
745         if (((ch >= 'A') && (ch <= 'Z'))
746                         || ((ch >= 'a') && (ch <= 'z'))
747                         || ((ch >= '0') && (ch <= '9'))
748                         || (ch == '/')
749                         || (ch == '-')
750                         || (ch == '.'))
751                 return 0;
752         return 1;
755 static inline int hex(int v)
757         if (v < 10)
758                 return '0' + v;
759         else
760                 return 'A' + v - 10;
763 static char *quote_ref_url(const char *base, const char *ref)
765         struct strbuf buf = STRBUF_INIT;
766         const char *cp;
767         int ch;
769         end_url_with_slash(&buf, base);
771         for (cp = ref; (ch = *cp) != 0; cp++)
772                 if (needs_quote(ch))
773                         strbuf_addf(&buf, "%%%02x", ch);
774                 else
775                         strbuf_addch(&buf, *cp);
777         return strbuf_detach(&buf, NULL);
780 void append_remote_object_url(struct strbuf *buf, const char *url,
781                               const char *hex,
782                               int only_two_digit_prefix)
784         end_url_with_slash(buf, url);
786         strbuf_addf(buf, "objects/%.*s/", 2, hex);
787         if (!only_two_digit_prefix)
788                 strbuf_addf(buf, "%s", hex+2);
791 char *get_remote_object_url(const char *url, const char *hex,
792                             int only_two_digit_prefix)
794         struct strbuf buf = STRBUF_INIT;
795         append_remote_object_url(&buf, url, hex, only_two_digit_prefix);
796         return strbuf_detach(&buf, NULL);
799 /* http_request() targets */
800 #define HTTP_REQUEST_STRBUF     0
801 #define HTTP_REQUEST_FILE       1
803 static int http_request(const char *url, void *result, int target, int options)
805         struct active_request_slot *slot;
806         struct slot_results results;
807         struct curl_slist *headers = NULL;
808         struct strbuf buf = STRBUF_INIT;
809         int ret;
811         slot = get_active_slot();
812         slot->results = &results;
813         curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1);
815         if (result == NULL) {
816                 curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 1);
817         } else {
818                 curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0);
819                 curl_easy_setopt(slot->curl, CURLOPT_FILE, result);
821                 if (target == HTTP_REQUEST_FILE) {
822                         long posn = ftell(result);
823                         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION,
824                                          fwrite);
825                         if (posn > 0) {
826                                 strbuf_addf(&buf, "Range: bytes=%ld-", posn);
827                                 headers = curl_slist_append(headers, buf.buf);
828                                 strbuf_reset(&buf);
829                         }
830                         slot->local = result;
831                 } else
832                         curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION,
833                                          fwrite_buffer);
834         }
836         strbuf_addstr(&buf, "Pragma:");
837         if (options & HTTP_NO_CACHE)
838                 strbuf_addstr(&buf, " no-cache");
840         headers = curl_slist_append(headers, buf.buf);
842         curl_easy_setopt(slot->curl, CURLOPT_URL, url);
843         curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, headers);
845         if (start_active_slot(slot)) {
846                 run_active_slot(slot);
847                 if (results.curl_result == CURLE_OK)
848                         ret = HTTP_OK;
849                 else if (missing_target(&results))
850                         ret = HTTP_MISSING_TARGET;
851                 else if (results.http_code == 401) {
852                         if (user_name) {
853                                 ret = HTTP_NOAUTH;
854                         } else {
855                                 /*
856                                  * git_getpass is needed here because its very likely stdin/stdout are
857                                  * pipes to our parent process.  So we instead need to use /dev/tty,
858                                  * but that is non-portable.  Using git_getpass() can at least be stubbed
859                                  * on other platforms with a different implementation if/when necessary.
860                                  */
861                                 user_name = xstrdup(git_getpass("Username: "));
862                                 init_curl_http_auth(slot->curl);
863                                 ret = HTTP_REAUTH;
864                         }
865                 } else
866                         ret = HTTP_ERROR;
867         } else {
868                 error("Unable to start HTTP request for %s", url);
869                 ret = HTTP_START_FAILED;
870         }
872         slot->local = NULL;
873         curl_slist_free_all(headers);
874         strbuf_release(&buf);
876         return ret;
879 int http_get_strbuf(const char *url, struct strbuf *result, int options)
881         int http_ret = http_request(url, result, HTTP_REQUEST_STRBUF, options);
882         if (http_ret == HTTP_REAUTH) {
883                 http_ret = http_request(url, result, HTTP_REQUEST_STRBUF, options);
884         }
885         return http_ret;
888 /*
889  * Downloads an url and stores the result in the given file.
890  *
891  * If a previous interrupted download is detected (i.e. a previous temporary
892  * file is still around) the download is resumed.
893  */
894 static int http_get_file(const char *url, const char *filename, int options)
896         int ret;
897         struct strbuf tmpfile = STRBUF_INIT;
898         FILE *result;
900         strbuf_addf(&tmpfile, "%s.temp", filename);
901         result = fopen(tmpfile.buf, "a");
902         if (! result) {
903                 error("Unable to open local file %s", tmpfile.buf);
904                 ret = HTTP_ERROR;
905                 goto cleanup;
906         }
908         ret = http_request(url, result, HTTP_REQUEST_FILE, options);
909         fclose(result);
911         if ((ret == HTTP_OK) && move_temp_to_file(tmpfile.buf, filename))
912                 ret = HTTP_ERROR;
913 cleanup:
914         strbuf_release(&tmpfile);
915         return ret;
918 int http_error(const char *url, int ret)
920         /* http_request has already handled HTTP_START_FAILED. */
921         if (ret != HTTP_START_FAILED)
922                 error("%s while accessing %s\n", curl_errorstr, url);
924         return ret;
927 int http_fetch_ref(const char *base, struct ref *ref)
929         char *url;
930         struct strbuf buffer = STRBUF_INIT;
931         int ret = -1;
933         url = quote_ref_url(base, ref->name);
934         if (http_get_strbuf(url, &buffer, HTTP_NO_CACHE) == HTTP_OK) {
935                 strbuf_rtrim(&buffer);
936                 if (buffer.len == 40)
937                         ret = get_sha1_hex(buffer.buf, ref->old_sha1);
938                 else if (!prefixcmp(buffer.buf, "ref: ")) {
939                         ref->symref = xstrdup(buffer.buf + 5);
940                         ret = 0;
941                 }
942         }
944         strbuf_release(&buffer);
945         free(url);
946         return ret;
949 /* Helpers for fetching packs */
950 static char *fetch_pack_index(unsigned char *sha1, const char *base_url)
952         char *url, *tmp;
953         struct strbuf buf = STRBUF_INIT;
955         if (http_is_verbose)
956                 fprintf(stderr, "Getting index for pack %s\n", sha1_to_hex(sha1));
958         end_url_with_slash(&buf, base_url);
959         strbuf_addf(&buf, "objects/pack/pack-%s.idx", sha1_to_hex(sha1));
960         url = strbuf_detach(&buf, NULL);
962         strbuf_addf(&buf, "%s.temp", sha1_pack_index_name(sha1));
963         tmp = strbuf_detach(&buf, NULL);
965         if (http_get_file(url, tmp, 0) != HTTP_OK) {
966                 error("Unable to get pack index %s\n", url);
967                 free(tmp);
968                 tmp = NULL;
969         }
971         free(url);
972         return tmp;
975 static int fetch_and_setup_pack_index(struct packed_git **packs_head,
976         unsigned char *sha1, const char *base_url)
978         struct packed_git *new_pack;
979         char *tmp_idx = NULL;
980         int ret;
982         if (has_pack_index(sha1)) {
983                 new_pack = parse_pack_index(sha1, NULL);
984                 if (!new_pack)
985                         return -1; /* parse_pack_index() already issued error message */
986                 goto add_pack;
987         }
989         tmp_idx = fetch_pack_index(sha1, base_url);
990         if (!tmp_idx)
991                 return -1;
993         new_pack = parse_pack_index(sha1, tmp_idx);
994         if (!new_pack) {
995                 unlink(tmp_idx);
996                 free(tmp_idx);
998                 return -1; /* parse_pack_index() already issued error message */
999         }
1001         ret = verify_pack_index(new_pack);
1002         if (!ret) {
1003                 close_pack_index(new_pack);
1004                 ret = move_temp_to_file(tmp_idx, sha1_pack_index_name(sha1));
1005         }
1006         free(tmp_idx);
1007         if (ret)
1008                 return -1;
1010 add_pack:
1011         new_pack->next = *packs_head;
1012         *packs_head = new_pack;
1013         return 0;
1016 int http_get_info_packs(const char *base_url, struct packed_git **packs_head)
1018         int ret = 0, i = 0;
1019         char *url, *data;
1020         struct strbuf buf = STRBUF_INIT;
1021         unsigned char sha1[20];
1023         end_url_with_slash(&buf, base_url);
1024         strbuf_addstr(&buf, "objects/info/packs");
1025         url = strbuf_detach(&buf, NULL);
1027         ret = http_get_strbuf(url, &buf, HTTP_NO_CACHE);
1028         if (ret != HTTP_OK)
1029                 goto cleanup;
1031         data = buf.buf;
1032         while (i < buf.len) {
1033                 switch (data[i]) {
1034                 case 'P':
1035                         i++;
1036                         if (i + 52 <= buf.len &&
1037                             !prefixcmp(data + i, " pack-") &&
1038                             !prefixcmp(data + i + 46, ".pack\n")) {
1039                                 get_sha1_hex(data + i + 6, sha1);
1040                                 fetch_and_setup_pack_index(packs_head, sha1,
1041                                                       base_url);
1042                                 i += 51;
1043                                 break;
1044                         }
1045                 default:
1046                         while (i < buf.len && data[i] != '\n')
1047                                 i++;
1048                 }
1049                 i++;
1050         }
1052 cleanup:
1053         free(url);
1054         return ret;
1057 void release_http_pack_request(struct http_pack_request *preq)
1059         if (preq->packfile != NULL) {
1060                 fclose(preq->packfile);
1061                 preq->packfile = NULL;
1062                 preq->slot->local = NULL;
1063         }
1064         if (preq->range_header != NULL) {
1065                 curl_slist_free_all(preq->range_header);
1066                 preq->range_header = NULL;
1067         }
1068         preq->slot = NULL;
1069         free(preq->url);
1072 int finish_http_pack_request(struct http_pack_request *preq)
1074         struct packed_git **lst;
1075         struct packed_git *p = preq->target;
1076         char *tmp_idx;
1077         struct child_process ip;
1078         const char *ip_argv[8];
1080         close_pack_index(p);
1082         fclose(preq->packfile);
1083         preq->packfile = NULL;
1084         preq->slot->local = NULL;
1086         lst = preq->lst;
1087         while (*lst != p)
1088                 lst = &((*lst)->next);
1089         *lst = (*lst)->next;
1091         tmp_idx = xstrdup(preq->tmpfile);
1092         strcpy(tmp_idx + strlen(tmp_idx) - strlen(".pack.temp"),
1093                ".idx.temp");
1095         ip_argv[0] = "index-pack";
1096         ip_argv[1] = "-o";
1097         ip_argv[2] = tmp_idx;
1098         ip_argv[3] = preq->tmpfile;
1099         ip_argv[4] = NULL;
1101         memset(&ip, 0, sizeof(ip));
1102         ip.argv = ip_argv;
1103         ip.git_cmd = 1;
1104         ip.no_stdin = 1;
1105         ip.no_stdout = 1;
1107         if (run_command(&ip)) {
1108                 unlink(preq->tmpfile);
1109                 unlink(tmp_idx);
1110                 free(tmp_idx);
1111                 return -1;
1112         }
1114         unlink(sha1_pack_index_name(p->sha1));
1116         if (move_temp_to_file(preq->tmpfile, sha1_pack_name(p->sha1))
1117          || move_temp_to_file(tmp_idx, sha1_pack_index_name(p->sha1))) {
1118                 free(tmp_idx);
1119                 return -1;
1120         }
1122         install_packed_git(p);
1123         free(tmp_idx);
1124         return 0;
1127 struct http_pack_request *new_http_pack_request(
1128         struct packed_git *target, const char *base_url)
1130         long prev_posn = 0;
1131         char range[RANGE_HEADER_SIZE];
1132         struct strbuf buf = STRBUF_INIT;
1133         struct http_pack_request *preq;
1135         preq = xmalloc(sizeof(*preq));
1136         preq->target = target;
1137         preq->range_header = NULL;
1139         end_url_with_slash(&buf, base_url);
1140         strbuf_addf(&buf, "objects/pack/pack-%s.pack",
1141                 sha1_to_hex(target->sha1));
1142         preq->url = strbuf_detach(&buf, NULL);
1144         snprintf(preq->tmpfile, sizeof(preq->tmpfile), "%s.temp",
1145                 sha1_pack_name(target->sha1));
1146         preq->packfile = fopen(preq->tmpfile, "a");
1147         if (!preq->packfile) {
1148                 error("Unable to open local file %s for pack",
1149                       preq->tmpfile);
1150                 goto abort;
1151         }
1153         preq->slot = get_active_slot();
1154         preq->slot->local = preq->packfile;
1155         curl_easy_setopt(preq->slot->curl, CURLOPT_FILE, preq->packfile);
1156         curl_easy_setopt(preq->slot->curl, CURLOPT_WRITEFUNCTION, fwrite);
1157         curl_easy_setopt(preq->slot->curl, CURLOPT_URL, preq->url);
1158         curl_easy_setopt(preq->slot->curl, CURLOPT_HTTPHEADER,
1159                 no_pragma_header);
1161         /*
1162          * If there is data present from a previous transfer attempt,
1163          * resume where it left off
1164          */
1165         prev_posn = ftell(preq->packfile);
1166         if (prev_posn>0) {
1167                 if (http_is_verbose)
1168                         fprintf(stderr,
1169                                 "Resuming fetch of pack %s at byte %ld\n",
1170                                 sha1_to_hex(target->sha1), prev_posn);
1171                 sprintf(range, "Range: bytes=%ld-", prev_posn);
1172                 preq->range_header = curl_slist_append(NULL, range);
1173                 curl_easy_setopt(preq->slot->curl, CURLOPT_HTTPHEADER,
1174                         preq->range_header);
1175         }
1177         return preq;
1179 abort:
1180         free(preq->url);
1181         free(preq);
1182         return NULL;
1185 /* Helpers for fetching objects (loose) */
1186 static size_t fwrite_sha1_file(char *ptr, size_t eltsize, size_t nmemb,
1187                                void *data)
1189         unsigned char expn[4096];
1190         size_t size = eltsize * nmemb;
1191         int posn = 0;
1192         struct http_object_request *freq =
1193                 (struct http_object_request *)data;
1194         do {
1195                 ssize_t retval = xwrite(freq->localfile,
1196                                         (char *) ptr + posn, size - posn);
1197                 if (retval < 0)
1198                         return posn;
1199                 posn += retval;
1200         } while (posn < size);
1202         freq->stream.avail_in = size;
1203         freq->stream.next_in = (void *)ptr;
1204         do {
1205                 freq->stream.next_out = expn;
1206                 freq->stream.avail_out = sizeof(expn);
1207                 freq->zret = git_inflate(&freq->stream, Z_SYNC_FLUSH);
1208                 git_SHA1_Update(&freq->c, expn,
1209                                 sizeof(expn) - freq->stream.avail_out);
1210         } while (freq->stream.avail_in && freq->zret == Z_OK);
1211         data_received++;
1212         return size;
1215 struct http_object_request *new_http_object_request(const char *base_url,
1216         unsigned char *sha1)
1218         char *hex = sha1_to_hex(sha1);
1219         char *filename;
1220         char prevfile[PATH_MAX];
1221         int prevlocal;
1222         char prev_buf[PREV_BUF_SIZE];
1223         ssize_t prev_read = 0;
1224         long prev_posn = 0;
1225         char range[RANGE_HEADER_SIZE];
1226         struct curl_slist *range_header = NULL;
1227         struct http_object_request *freq;
1229         freq = xmalloc(sizeof(*freq));
1230         hashcpy(freq->sha1, sha1);
1231         freq->localfile = -1;
1233         filename = sha1_file_name(sha1);
1234         snprintf(freq->tmpfile, sizeof(freq->tmpfile),
1235                  "%s.temp", filename);
1237         snprintf(prevfile, sizeof(prevfile), "%s.prev", filename);
1238         unlink_or_warn(prevfile);
1239         rename(freq->tmpfile, prevfile);
1240         unlink_or_warn(freq->tmpfile);
1242         if (freq->localfile != -1)
1243                 error("fd leakage in start: %d", freq->localfile);
1244         freq->localfile = open(freq->tmpfile,
1245                                O_WRONLY | O_CREAT | O_EXCL, 0666);
1246         /*
1247          * This could have failed due to the "lazy directory creation";
1248          * try to mkdir the last path component.
1249          */
1250         if (freq->localfile < 0 && errno == ENOENT) {
1251                 char *dir = strrchr(freq->tmpfile, '/');
1252                 if (dir) {
1253                         *dir = 0;
1254                         mkdir(freq->tmpfile, 0777);
1255                         *dir = '/';
1256                 }
1257                 freq->localfile = open(freq->tmpfile,
1258                                        O_WRONLY | O_CREAT | O_EXCL, 0666);
1259         }
1261         if (freq->localfile < 0) {
1262                 error("Couldn't create temporary file %s: %s",
1263                       freq->tmpfile, strerror(errno));
1264                 goto abort;
1265         }
1267         memset(&freq->stream, 0, sizeof(freq->stream));
1269         git_inflate_init(&freq->stream);
1271         git_SHA1_Init(&freq->c);
1273         freq->url = get_remote_object_url(base_url, hex, 0);
1275         /*
1276          * If a previous temp file is present, process what was already
1277          * fetched.
1278          */
1279         prevlocal = open(prevfile, O_RDONLY);
1280         if (prevlocal != -1) {
1281                 do {
1282                         prev_read = xread(prevlocal, prev_buf, PREV_BUF_SIZE);
1283                         if (prev_read>0) {
1284                                 if (fwrite_sha1_file(prev_buf,
1285                                                      1,
1286                                                      prev_read,
1287                                                      freq) == prev_read) {
1288                                         prev_posn += prev_read;
1289                                 } else {
1290                                         prev_read = -1;
1291                                 }
1292                         }
1293                 } while (prev_read > 0);
1294                 close(prevlocal);
1295         }
1296         unlink_or_warn(prevfile);
1298         /*
1299          * Reset inflate/SHA1 if there was an error reading the previous temp
1300          * file; also rewind to the beginning of the local file.
1301          */
1302         if (prev_read == -1) {
1303                 memset(&freq->stream, 0, sizeof(freq->stream));
1304                 git_inflate_init(&freq->stream);
1305                 git_SHA1_Init(&freq->c);
1306                 if (prev_posn>0) {
1307                         prev_posn = 0;
1308                         lseek(freq->localfile, 0, SEEK_SET);
1309                         if (ftruncate(freq->localfile, 0) < 0) {
1310                                 error("Couldn't truncate temporary file %s: %s",
1311                                           freq->tmpfile, strerror(errno));
1312                                 goto abort;
1313                         }
1314                 }
1315         }
1317         freq->slot = get_active_slot();
1319         curl_easy_setopt(freq->slot->curl, CURLOPT_FILE, freq);
1320         curl_easy_setopt(freq->slot->curl, CURLOPT_WRITEFUNCTION, fwrite_sha1_file);
1321         curl_easy_setopt(freq->slot->curl, CURLOPT_ERRORBUFFER, freq->errorstr);
1322         curl_easy_setopt(freq->slot->curl, CURLOPT_URL, freq->url);
1323         curl_easy_setopt(freq->slot->curl, CURLOPT_HTTPHEADER, no_pragma_header);
1325         /*
1326          * If we have successfully processed data from a previous fetch
1327          * attempt, only fetch the data we don't already have.
1328          */
1329         if (prev_posn>0) {
1330                 if (http_is_verbose)
1331                         fprintf(stderr,
1332                                 "Resuming fetch of object %s at byte %ld\n",
1333                                 hex, prev_posn);
1334                 sprintf(range, "Range: bytes=%ld-", prev_posn);
1335                 range_header = curl_slist_append(range_header, range);
1336                 curl_easy_setopt(freq->slot->curl,
1337                                  CURLOPT_HTTPHEADER, range_header);
1338         }
1340         return freq;
1342 abort:
1343         free(filename);
1344         free(freq->url);
1345         free(freq);
1346         return NULL;
1349 void process_http_object_request(struct http_object_request *freq)
1351         if (freq->slot == NULL)
1352                 return;
1353         freq->curl_result = freq->slot->curl_result;
1354         freq->http_code = freq->slot->http_code;
1355         freq->slot = NULL;
1358 int finish_http_object_request(struct http_object_request *freq)
1360         struct stat st;
1362         close(freq->localfile);
1363         freq->localfile = -1;
1365         process_http_object_request(freq);
1367         if (freq->http_code == 416) {
1368                 warning("requested range invalid; we may already have all the data.");
1369         } else if (freq->curl_result != CURLE_OK) {
1370                 if (stat(freq->tmpfile, &st) == 0)
1371                         if (st.st_size == 0)
1372                                 unlink_or_warn(freq->tmpfile);
1373                 return -1;
1374         }
1376         git_inflate_end(&freq->stream);
1377         git_SHA1_Final(freq->real_sha1, &freq->c);
1378         if (freq->zret != Z_STREAM_END) {
1379                 unlink_or_warn(freq->tmpfile);
1380                 return -1;
1381         }
1382         if (hashcmp(freq->sha1, freq->real_sha1)) {
1383                 unlink_or_warn(freq->tmpfile);
1384                 return -1;
1385         }
1386         freq->rename =
1387                 move_temp_to_file(freq->tmpfile, sha1_file_name(freq->sha1));
1389         return freq->rename;
1392 void abort_http_object_request(struct http_object_request *freq)
1394         unlink_or_warn(freq->tmpfile);
1396         release_http_object_request(freq);
1399 void release_http_object_request(struct http_object_request *freq)
1401         if (freq->localfile != -1) {
1402                 close(freq->localfile);
1403                 freq->localfile = -1;
1404         }
1405         if (freq->url != NULL) {
1406                 free(freq->url);
1407                 freq->url = NULL;
1408         }
1409         if (freq->slot != NULL) {
1410                 freq->slot->callback_func = NULL;
1411                 freq->slot->callback_data = NULL;
1412                 release_active_slot(freq->slot);
1413                 freq->slot = NULL;
1414         }