http.c 22.7 KB
Newer Older
1
#include "http.h"
2
#include "pack.h"
3 4

int data_received;
J
Junio C Hamano 已提交
5
int active_requests;
6
int http_is_verbose;
7 8

#ifdef USE_CURL_MULTI
M
Mike Hommey 已提交
9 10
static int max_requests = -1;
static CURLM *curlm;
11 12
#endif
#ifndef NO_CURL_EASY_DUPHANDLE
M
Mike Hommey 已提交
13
static CURL *curl_default;
14 15 16
#endif
char curl_errorstr[CURL_ERROR_SIZE];

M
Mike Hommey 已提交
17
static int curl_ssl_verify = -1;
J
Junio C Hamano 已提交
18
static const char *ssl_cert;
19
#if LIBCURL_VERSION_NUM >= 0x070902
J
Junio C Hamano 已提交
20
static const char *ssl_key;
21 22
#endif
#if LIBCURL_VERSION_NUM >= 0x070908
J
Junio C Hamano 已提交
23
static const char *ssl_capath;
24
#endif
J
Junio C Hamano 已提交
25
static const char *ssl_cainfo;
M
Mike Hommey 已提交
26 27
static long curl_low_speed_limit = -1;
static long curl_low_speed_time = -1;
J
Junio C Hamano 已提交
28 29
static int curl_ftp_no_epsv;
static const char *curl_http_proxy;
J
Junio C Hamano 已提交
30
static char *user_name, *user_pass;
31

M
Mike Hommey 已提交
32
static struct curl_slist *pragma_header;
33

34 35
struct curl_slist *no_pragma_header;

J
Junio C Hamano 已提交
36
static struct active_request_slot *active_queue_head;
37

38
size_t fread_buffer(void *ptr, size_t eltsize, size_t nmemb, void *buffer_)
39 40
{
	size_t size = eltsize * nmemb;
41 42
	struct buffer *buffer = buffer_;

M
Mike Hommey 已提交
43 44 45
	if (size > buffer->buf.len - buffer->posn)
		size = buffer->buf.len - buffer->posn;
	memcpy(ptr, buffer->buf.buf + buffer->posn, size);
46
	buffer->posn += size;
M
Mike Hommey 已提交
47

48 49 50
	return size;
}

51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
#ifndef NO_CURL_IOCTL
curlioerr ioctl_buffer(CURL *handle, int cmd, void *clientp)
{
	struct buffer *buffer = clientp;

	switch (cmd) {
	case CURLIOCMD_NOP:
		return CURLIOE_OK;

	case CURLIOCMD_RESTARTREAD:
		buffer->posn = 0;
		return CURLIOE_OK;

	default:
		return CURLIOE_UNKNOWNCMD;
	}
}
#endif

70
size_t fwrite_buffer(const void *ptr, size_t eltsize, size_t nmemb, void *buffer_)
71 72
{
	size_t size = eltsize * nmemb;
73 74
	struct strbuf *buffer = buffer_;

M
Mike Hommey 已提交
75
	strbuf_add(buffer, ptr, size);
76 77 78 79
	data_received++;
	return size;
}

80
size_t fwrite_null(const void *ptr, size_t eltsize, size_t nmemb, void *strbuf)
81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
{
	data_received++;
	return eltsize * nmemb;
}

static void finish_active_slot(struct active_request_slot *slot);

#ifdef USE_CURL_MULTI
static void process_curl_messages(void)
{
	int num_messages;
	struct active_request_slot *slot;
	CURLMsg *curl_message = curl_multi_info_read(curlm, &num_messages);

	while (curl_message != NULL) {
		if (curl_message->msg == CURLMSG_DONE) {
			int curl_result = curl_message->data.result;
			slot = active_queue_head;
			while (slot != NULL &&
			       slot->curl != curl_message->easy_handle)
				slot = slot->next;
			if (slot != NULL) {
				curl_multi_remove_handle(curlm, slot->curl);
				slot->curl_result = curl_result;
				finish_active_slot(slot);
			} else {
				fprintf(stderr, "Received DONE message for unknown request!\n");
			}
		} else {
			fprintf(stderr, "Unknown CURL message received: %d\n",
				(int)curl_message->msg);
		}
		curl_message = curl_multi_info_read(curlm, &num_messages);
	}
}
#endif

118
static int http_options(const char *var, const char *value, void *cb)
119 120
{
	if (!strcmp("http.sslverify", var)) {
121
		curl_ssl_verify = git_config_bool(var, value);
122 123
		return 0;
	}
124 125
	if (!strcmp("http.sslcert", var))
		return git_config_string(&ssl_cert, var, value);
126
#if LIBCURL_VERSION_NUM >= 0x070902
127 128
	if (!strcmp("http.sslkey", var))
		return git_config_string(&ssl_key, var, value);
129 130
#endif
#if LIBCURL_VERSION_NUM >= 0x070908
131 132
	if (!strcmp("http.sslcapath", var))
		return git_config_string(&ssl_capath, var, value);
133
#endif
134 135
	if (!strcmp("http.sslcainfo", var))
		return git_config_string(&ssl_cainfo, var, value);
J
Junio C Hamano 已提交
136
#ifdef USE_CURL_MULTI
137
	if (!strcmp("http.maxrequests", var)) {
138
		max_requests = git_config_int(var, value);
139 140 141 142
		return 0;
	}
#endif
	if (!strcmp("http.lowspeedlimit", var)) {
143
		curl_low_speed_limit = (long)git_config_int(var, value);
144 145 146
		return 0;
	}
	if (!strcmp("http.lowspeedtime", var)) {
147
		curl_low_speed_time = (long)git_config_int(var, value);
148 149 150
		return 0;
	}

151 152 153 154
	if (!strcmp("http.noepsv", var)) {
		curl_ftp_no_epsv = git_config_bool(var, value);
		return 0;
	}
155 156
	if (!strcmp("http.proxy", var))
		return git_config_string(&curl_http_proxy, var, value);
157

158
	/* Fall back on the default ones */
159
	return git_default_config(var, value, cb);
160 161
}

J
Junio C Hamano 已提交
162 163
static void init_curl_http_auth(CURL *result)
{
164
	if (user_name) {
J
Junio C Hamano 已提交
165 166 167 168 169 170 171 172 173
		struct strbuf up = STRBUF_INIT;
		if (!user_pass)
			user_pass = xstrdup(getpass("Password: "));
		strbuf_addf(&up, "%s:%s", user_name, user_pass);
		curl_easy_setopt(result, CURLOPT_USERPWD,
				 strbuf_detach(&up, NULL));
	}
}

J
Junio C Hamano 已提交
174
static CURL *get_curl_handle(void)
175
{
J
Junio C Hamano 已提交
176
	CURL *result = curl_easy_init();
177

178 179 180 181 182 183 184 185 186 187
	if (!curl_ssl_verify) {
		curl_easy_setopt(result, CURLOPT_SSL_VERIFYPEER, 0);
		curl_easy_setopt(result, CURLOPT_SSL_VERIFYHOST, 0);
	} else {
		/* Verify authenticity of the peer's certificate */
		curl_easy_setopt(result, CURLOPT_SSL_VERIFYPEER, 1);
		/* The name in the cert must match whom we tried to connect */
		curl_easy_setopt(result, CURLOPT_SSL_VERIFYHOST, 2);
	}

188 189 190 191
#if LIBCURL_VERSION_NUM >= 0x070907
	curl_easy_setopt(result, CURLOPT_NETRC, CURL_NETRC_OPTIONAL);
#endif

J
Junio C Hamano 已提交
192 193
	init_curl_http_auth(result);

194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216
	if (ssl_cert != NULL)
		curl_easy_setopt(result, CURLOPT_SSLCERT, ssl_cert);
#if LIBCURL_VERSION_NUM >= 0x070902
	if (ssl_key != NULL)
		curl_easy_setopt(result, CURLOPT_SSLKEY, ssl_key);
#endif
#if LIBCURL_VERSION_NUM >= 0x070908
	if (ssl_capath != NULL)
		curl_easy_setopt(result, CURLOPT_CAPATH, ssl_capath);
#endif
	if (ssl_cainfo != NULL)
		curl_easy_setopt(result, CURLOPT_CAINFO, ssl_cainfo);
	curl_easy_setopt(result, CURLOPT_FAILONERROR, 1);

	if (curl_low_speed_limit > 0 && curl_low_speed_time > 0) {
		curl_easy_setopt(result, CURLOPT_LOW_SPEED_LIMIT,
				 curl_low_speed_limit);
		curl_easy_setopt(result, CURLOPT_LOW_SPEED_TIME,
				 curl_low_speed_time);
	}

	curl_easy_setopt(result, CURLOPT_FOLLOWLOCATION, 1);

217 218 219
	if (getenv("GIT_CURL_VERBOSE"))
		curl_easy_setopt(result, CURLOPT_VERBOSE, 1);

220 221
	curl_easy_setopt(result, CURLOPT_USERAGENT, GIT_USER_AGENT);

222 223 224
	if (curl_ftp_no_epsv)
		curl_easy_setopt(result, CURLOPT_FTP_USE_EPSV, 0);

225 226 227
	if (curl_http_proxy)
		curl_easy_setopt(result, CURLOPT_PROXY, curl_http_proxy);

228 229 230
	return result;
}

J
Junio C Hamano 已提交
231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270
static void http_auth_init(const char *url)
{
	char *at, *colon, *cp, *slash;
	int len;

	cp = strstr(url, "://");
	if (!cp)
		return;

	/*
	 * Ok, the URL looks like "proto://something".  Which one?
	 * "proto://<user>:<pass>@<host>/...",
	 * "proto://<user>@<host>/...", or just
	 * "proto://<host>/..."?
	 */
	cp += 3;
	at = strchr(cp, '@');
	colon = strchr(cp, ':');
	slash = strchrnul(cp, '/');
	if (!at || slash <= at)
		return; /* No credentials */
	if (!colon || at <= colon) {
		/* Only username */
		len = at - cp;
		user_name = xmalloc(len + 1);
		memcpy(user_name, cp, len);
		user_name[len] = '\0';
		user_pass = NULL;
	} else {
		len = colon - cp;
		user_name = xmalloc(len + 1);
		memcpy(user_name, cp, len);
		user_name[len] = '\0';
		len = at - (colon + 1);
		user_pass = xmalloc(len + 1);
		memcpy(user_pass, colon + 1, len);
		user_pass[len] = '\0';
	}
}

271 272 273 274 275 276 277
static void set_from_env(const char **var, const char *envname)
{
	const char *val = getenv(envname);
	if (val)
		*var = val;
}

278
void http_init(struct remote *remote)
279 280 281 282
{
	char *low_speed_limit;
	char *low_speed_time;

283 284
	http_is_verbose = 0;

285 286
	git_config(http_options, NULL);

287 288
	curl_global_init(CURL_GLOBAL_ALL);

289 290 291
	if (remote && remote->http_proxy)
		curl_http_proxy = xstrdup(remote->http_proxy);

292
	pragma_header = curl_slist_append(pragma_header, "Pragma: no-cache");
293
	no_pragma_header = curl_slist_append(no_pragma_header, "Pragma:");
294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311

#ifdef USE_CURL_MULTI
	{
		char *http_max_requests = getenv("GIT_HTTP_MAX_REQUESTS");
		if (http_max_requests != NULL)
			max_requests = atoi(http_max_requests);
	}

	curlm = curl_multi_init();
	if (curlm == NULL) {
		fprintf(stderr, "Error creating curl multi handle.\n");
		exit(1);
	}
#endif

	if (getenv("GIT_SSL_NO_VERIFY"))
		curl_ssl_verify = 0;

312
	set_from_env(&ssl_cert, "GIT_SSL_CERT");
313
#if LIBCURL_VERSION_NUM >= 0x070902
314
	set_from_env(&ssl_key, "GIT_SSL_KEY");
315 316
#endif
#if LIBCURL_VERSION_NUM >= 0x070908
317
	set_from_env(&ssl_capath, "GIT_SSL_CAPATH");
318
#endif
319
	set_from_env(&ssl_cainfo, "GIT_SSL_CAINFO");
320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335

	low_speed_limit = getenv("GIT_HTTP_LOW_SPEED_LIMIT");
	if (low_speed_limit != NULL)
		curl_low_speed_limit = strtol(low_speed_limit, NULL, 10);
	low_speed_time = getenv("GIT_HTTP_LOW_SPEED_TIME");
	if (low_speed_time != NULL)
		curl_low_speed_time = strtol(low_speed_time, NULL, 10);

	if (curl_ssl_verify == -1)
		curl_ssl_verify = 1;

#ifdef USE_CURL_MULTI
	if (max_requests < 1)
		max_requests = DEFAULT_MAX_REQUESTS;
#endif

336 337 338
	if (getenv("GIT_CURL_FTP_NO_EPSV"))
		curl_ftp_no_epsv = 1;

J
Junio C Hamano 已提交
339 340 341
	if (remote && remote->url && remote->url[0])
		http_auth_init(remote->url[0]);

342 343 344 345 346 347 348 349 350 351
#ifndef NO_CURL_EASY_DUPHANDLE
	curl_default = get_curl_handle();
#endif
}

void http_cleanup(void)
{
	struct active_request_slot *slot = active_queue_head;

	while (slot != NULL) {
352
		struct active_request_slot *next = slot->next;
353
		if (slot->curl != NULL) {
354
#ifdef USE_CURL_MULTI
355
			curl_multi_remove_handle(curlm, slot->curl);
356 357
#endif
			curl_easy_cleanup(slot->curl);
358
		}
359 360
		free(slot);
		slot = next;
361
	}
362
	active_queue_head = NULL;
363 364 365 366 367 368 369 370 371

#ifndef NO_CURL_EASY_DUPHANDLE
	curl_easy_cleanup(curl_default);
#endif

#ifdef USE_CURL_MULTI
	curl_multi_cleanup(curlm);
#endif
	curl_global_cleanup();
N
Nick Hengeveld 已提交
372 373

	curl_slist_free_all(pragma_header);
374
	pragma_header = NULL;
375

376 377 378
	curl_slist_free_all(no_pragma_header);
	no_pragma_header = NULL;

379
	if (curl_http_proxy) {
380
		free((void *)curl_http_proxy);
381 382
		curl_http_proxy = NULL;
	}
383 384 385 386 387 388 389 390 391 392 393 394 395
}

struct active_request_slot *get_active_slot(void)
{
	struct active_request_slot *slot = active_queue_head;
	struct active_request_slot *newslot;

#ifdef USE_CURL_MULTI
	int num_transfers;

	/* Wait for a slot to open up if the queue is full */
	while (active_requests >= max_requests) {
		curl_multi_perform(curlm, &num_transfers);
J
Junio C Hamano 已提交
396
		if (num_transfers < active_requests)
397 398 399 400
			process_curl_messages();
	}
#endif

J
Junio C Hamano 已提交
401
	while (slot != NULL && slot->in_use)
402
		slot = slot->next;
J
Junio C Hamano 已提交
403

404 405 406 407 408 409 410 411 412 413
	if (slot == NULL) {
		newslot = xmalloc(sizeof(*newslot));
		newslot->curl = NULL;
		newslot->in_use = 0;
		newslot->next = NULL;

		slot = active_queue_head;
		if (slot == NULL) {
			active_queue_head = newslot;
		} else {
J
Junio C Hamano 已提交
414
			while (slot->next != NULL)
415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431
				slot = slot->next;
			slot->next = newslot;
		}
		slot = newslot;
	}

	if (slot->curl == NULL) {
#ifdef NO_CURL_EASY_DUPHANDLE
		slot->curl = get_curl_handle();
#else
		slot->curl = curl_easy_duphandle(curl_default);
#endif
	}

	active_requests++;
	slot->in_use = 1;
	slot->local = NULL;
432
	slot->results = NULL;
N
Nick Hengeveld 已提交
433
	slot->finished = NULL;
434 435 436 437
	slot->callback_data = NULL;
	slot->callback_func = NULL;
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, pragma_header);
	curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, curl_errorstr);
438 439 440 441 442
	curl_easy_setopt(slot->curl, CURLOPT_CUSTOMREQUEST, NULL);
	curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, NULL);
	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, NULL);
	curl_easy_setopt(slot->curl, CURLOPT_UPLOAD, 0);
	curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1);
443 444 445 446 447 448 449 450

	return slot;
}

int start_active_slot(struct active_request_slot *slot)
{
#ifdef USE_CURL_MULTI
	CURLMcode curlm_result = curl_multi_add_handle(curlm, slot->curl);
451
	int num_transfers;
452 453 454 455 456 457 458

	if (curlm_result != CURLM_OK &&
	    curlm_result != CURLM_CALL_MULTI_PERFORM) {
		active_requests--;
		slot->in_use = 0;
		return 0;
	}
459 460 461 462 463 464

	/*
	 * We know there must be something to do, since we just added
	 * something.
	 */
	curl_multi_perform(curlm, &num_transfers);
465 466 467 468 469
#endif
	return 1;
}

#ifdef USE_CURL_MULTI
470 471 472 473 474 475
struct fill_chain {
	void *data;
	int (*fill)(void *);
	struct fill_chain *next;
};

J
Junio C Hamano 已提交
476
static struct fill_chain *fill_cfg;
477 478 479

void add_fill_function(void *data, int (*fill)(void *))
{
480
	struct fill_chain *new = xmalloc(sizeof(*new));
481 482 483 484 485 486 487 488 489
	struct fill_chain **linkp = &fill_cfg;
	new->data = data;
	new->fill = fill;
	new->next = NULL;
	while (*linkp)
		linkp = &(*linkp)->next;
	*linkp = new;
}

490 491 492 493
void fill_active_slots(void)
{
	struct active_request_slot *slot = active_queue_head;

494 495 496 497 498 499 500
	while (active_requests < max_requests) {
		struct fill_chain *fill;
		for (fill = fill_cfg; fill; fill = fill->next)
			if (fill->fill(fill->data))
				break;

		if (!fill)
501
			break;
502
	}
503 504 505 506 507 508 509 510 511 512

	while (slot != NULL) {
		if (!slot->in_use && slot->curl != NULL) {
			curl_easy_cleanup(slot->curl);
			slot->curl = NULL;
		}
		slot = slot->next;
	}
}

513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537
void step_active_slots(void)
{
	int num_transfers;
	CURLMcode curlm_result;

	do {
		curlm_result = curl_multi_perform(curlm, &num_transfers);
	} while (curlm_result == CURLM_CALL_MULTI_PERFORM);
	if (num_transfers < active_requests) {
		process_curl_messages();
		fill_active_slots();
	}
}
#endif

void run_active_slot(struct active_request_slot *slot)
{
#ifdef USE_CURL_MULTI
	long last_pos = 0;
	long current_pos;
	fd_set readfds;
	fd_set writefds;
	fd_set excfds;
	int max_fd;
	struct timeval select_timeout;
N
Nick Hengeveld 已提交
538
	int finished = 0;
539

N
Nick Hengeveld 已提交
540 541
	slot->finished = &finished;
	while (!finished) {
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570
		data_received = 0;
		step_active_slots();

		if (!data_received && slot->local != NULL) {
			current_pos = ftell(slot->local);
			if (current_pos > last_pos)
				data_received++;
			last_pos = current_pos;
		}

		if (slot->in_use && !data_received) {
			max_fd = 0;
			FD_ZERO(&readfds);
			FD_ZERO(&writefds);
			FD_ZERO(&excfds);
			select_timeout.tv_sec = 0;
			select_timeout.tv_usec = 50000;
			select(max_fd, &readfds, &writefds,
			       &excfds, &select_timeout);
		}
	}
#else
	while (slot->in_use) {
		slot->curl_result = curl_easy_perform(slot->curl);
		finish_active_slot(slot);
	}
#endif
}

571
static void closedown_active_slot(struct active_request_slot *slot)
572
{
M
Mike Hommey 已提交
573 574
	active_requests--;
	slot->in_use = 0;
575 576 577 578 579 580
}

void release_active_slot(struct active_request_slot *slot)
{
	closedown_active_slot(slot);
	if (slot->curl) {
N
Nick Hengeveld 已提交
581
#ifdef USE_CURL_MULTI
582
		curl_multi_remove_handle(curlm, slot->curl);
N
Nick Hengeveld 已提交
583
#endif
584 585 586
		curl_easy_cleanup(slot->curl);
		slot->curl = NULL;
	}
N
Nick Hengeveld 已提交
587
#ifdef USE_CURL_MULTI
588
	fill_active_slots();
N
Nick Hengeveld 已提交
589
#endif
590 591 592 593 594
}

static void finish_active_slot(struct active_request_slot *slot)
{
	closedown_active_slot(slot);
M
Mike Hommey 已提交
595
	curl_easy_getinfo(slot->curl, CURLINFO_HTTP_CODE, &slot->http_code);
596

N
Nick Hengeveld 已提交
597 598 599
	if (slot->finished != NULL)
		(*slot->finished) = 1;

600 601 602 603 604 605
	/* Store slot results so they can be read after the slot is reused */
	if (slot->results != NULL) {
		slot->results->curl_result = slot->curl_result;
		slot->results->http_code = slot->http_code;
	}

M
Mike Hommey 已提交
606
	/* Run callback if appropriate */
J
Junio C Hamano 已提交
607
	if (slot->callback_func != NULL)
M
Mike Hommey 已提交
608
		slot->callback_func(slot->callback_data);
609 610 611 612 613 614 615 616 617 618 619 620 621 622
}

void finish_all_active_slots(void)
{
	struct active_request_slot *slot = active_queue_head;

	while (slot != NULL)
		if (slot->in_use) {
			run_active_slot(slot);
			slot = active_queue_head;
		} else {
			slot = slot->next;
		}
}
623

624
/* Helpers for modifying and creating URLs */
625 626 627 628 629 630 631 632 633 634 635 636 637 638
static inline int needs_quote(int ch)
{
	if (((ch >= 'A') && (ch <= 'Z'))
			|| ((ch >= 'a') && (ch <= 'z'))
			|| ((ch >= '0') && (ch <= '9'))
			|| (ch == '/')
			|| (ch == '-')
			|| (ch == '.'))
		return 0;
	return 1;
}

static inline int hex(int v)
{
J
Junio C Hamano 已提交
639 640 641 642
	if (v < 10)
		return '0' + v;
	else
		return 'A' + v - 10;
643 644
}

645 646 647 648 649 650 651
static void end_url_with_slash(struct strbuf *buf, const char *url)
{
	strbuf_addstr(buf, url);
	if (buf->len && buf->buf[buf->len - 1] != '/')
		strbuf_addstr(buf, "/");
}

652 653
static char *quote_ref_url(const char *base, const char *ref)
{
654
	struct strbuf buf = STRBUF_INIT;
655
	const char *cp;
656
	int ch;
657

658
	end_url_with_slash(&buf, base);
659 660

	for (cp = ref; (ch = *cp) != 0; cp++)
661
		if (needs_quote(ch))
662
			strbuf_addf(&buf, "%%%02x", ch);
663
		else
664
			strbuf_addch(&buf, *cp);
665

666
	return strbuf_detach(&buf, NULL);
667 668
}

669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771 772
/* http_request() targets */
#define HTTP_REQUEST_STRBUF	0
#define HTTP_REQUEST_FILE	1

static int http_request(const char *url, void *result, int target, int options)
{
	struct active_request_slot *slot;
	struct slot_results results;
	struct curl_slist *headers = NULL;
	struct strbuf buf = STRBUF_INIT;
	int ret;

	slot = get_active_slot();
	slot->results = &results;
	curl_easy_setopt(slot->curl, CURLOPT_HTTPGET, 1);

	if (result == NULL) {
		curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 1);
	} else {
		curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0);
		curl_easy_setopt(slot->curl, CURLOPT_FILE, result);

		if (target == HTTP_REQUEST_FILE) {
			long posn = ftell(result);
			curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION,
					 fwrite);
			if (posn > 0) {
				strbuf_addf(&buf, "Range: bytes=%ld-", posn);
				headers = curl_slist_append(headers, buf.buf);
				strbuf_reset(&buf);
			}
			slot->local = result;
		} else
			curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION,
					 fwrite_buffer);
	}

	strbuf_addstr(&buf, "Pragma:");
	if (options & HTTP_NO_CACHE)
		strbuf_addstr(&buf, " no-cache");

	headers = curl_slist_append(headers, buf.buf);

	curl_easy_setopt(slot->curl, CURLOPT_URL, url);
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, headers);

	if (start_active_slot(slot)) {
		run_active_slot(slot);
		if (results.curl_result == CURLE_OK)
			ret = HTTP_OK;
		else if (missing_target(&results))
			ret = HTTP_MISSING_TARGET;
		else
			ret = HTTP_ERROR;
	} else {
		error("Unable to start HTTP request for %s", url);
		ret = HTTP_START_FAILED;
	}

	slot->local = NULL;
	curl_slist_free_all(headers);
	strbuf_release(&buf);

	return ret;
}

int http_get_strbuf(const char *url, struct strbuf *result, int options)
{
	return http_request(url, result, HTTP_REQUEST_STRBUF, options);
}

int http_get_file(const char *url, const char *filename, int options)
{
	int ret;
	struct strbuf tmpfile = STRBUF_INIT;
	FILE *result;

	strbuf_addf(&tmpfile, "%s.temp", filename);
	result = fopen(tmpfile.buf, "a");
	if (! result) {
		error("Unable to open local file %s", tmpfile.buf);
		ret = HTTP_ERROR;
		goto cleanup;
	}

	ret = http_request(url, result, HTTP_REQUEST_FILE, options);
	fclose(result);

	if ((ret == HTTP_OK) && move_temp_to_file(tmpfile.buf, filename))
		ret = HTTP_ERROR;
cleanup:
	strbuf_release(&tmpfile);
	return ret;
}

int http_error(const char *url, int ret)
{
	/* http_request has already handled HTTP_START_FAILED. */
	if (ret != HTTP_START_FAILED)
		error("%s while accessing %s\n", curl_errorstr, url);

	return ret;
}

773
int http_fetch_ref(const char *base, struct ref *ref)
774 775 776
{
	char *url;
	struct strbuf buffer = STRBUF_INIT;
777
	int ret = -1;
778

779
	url = quote_ref_url(base, ref->name);
780 781 782 783 784 785 786
	if (http_get_strbuf(url, &buffer, HTTP_NO_CACHE) == HTTP_OK) {
		strbuf_rtrim(&buffer);
		if (buffer.len == 40)
			ret = get_sha1_hex(buffer.buf, ref->old_sha1);
		else if (!prefixcmp(buffer.buf, "ref: ")) {
			ref->symref = xstrdup(buffer.buf + 5);
			ret = 0;
787 788 789 790 791 792 793
		}
	}

	strbuf_release(&buffer);
	free(url);
	return ret;
}
T
Tay Ray Chuan 已提交
794 795 796 797 798 799 800 801 802 803 804 805 806 807 808

/* Helpers for fetching packs */
static int fetch_pack_index(unsigned char *sha1, const char *base_url)
{
	int ret = 0;
	char *hex = xstrdup(sha1_to_hex(sha1));
	char *filename;
	char *url;
	struct strbuf buf = STRBUF_INIT;

	/* Don't use the index if the pack isn't there */
	end_url_with_slash(&buf, base_url);
	strbuf_addf(&buf, "objects/pack/pack-%s.pack", hex);
	url = strbuf_detach(&buf, 0);

809 810 811 812
	if (http_get_strbuf(url, NULL, 0)) {
		ret = error("Unable to verify pack %s is available",
			    hex);
		goto cleanup;
T
Tay Ray Chuan 已提交
813 814 815 816
	}

	if (has_pack_index(sha1)) {
		ret = 0;
817
		goto cleanup;
T
Tay Ray Chuan 已提交
818 819 820 821 822 823 824 825 826 827
	}

	if (http_is_verbose)
		fprintf(stderr, "Getting index for pack %s\n", hex);

	end_url_with_slash(&buf, base_url);
	strbuf_addf(&buf, "objects/pack/pack-%s.idx", hex);
	url = strbuf_detach(&buf, NULL);

	filename = sha1_pack_index_name(sha1);
828 829
	if (http_get_file(url, filename, 0) != HTTP_OK)
		ret = error("Unable to get pack index %s\n", url);
T
Tay Ray Chuan 已提交
830

831
cleanup:
T
Tay Ray Chuan 已提交
832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892
	free(hex);
	free(url);
	return ret;
}

static int fetch_and_setup_pack_index(struct packed_git **packs_head,
	unsigned char *sha1, const char *base_url)
{
	struct packed_git *new_pack;

	if (fetch_pack_index(sha1, base_url))
		return -1;

	new_pack = parse_pack_index(sha1);
	if (!new_pack)
		return -1; /* parse_pack_index() already issued error message */
	new_pack->next = *packs_head;
	*packs_head = new_pack;
	return 0;
}

int http_get_info_packs(const char *base_url, struct packed_git **packs_head)
{
	int ret = 0, i = 0;
	char *url, *data;
	struct strbuf buf = STRBUF_INIT;
	unsigned char sha1[20];

	end_url_with_slash(&buf, base_url);
	strbuf_addstr(&buf, "objects/info/packs");
	url = strbuf_detach(&buf, NULL);

	ret = http_get_strbuf(url, &buf, HTTP_NO_CACHE);
	if (ret != HTTP_OK)
		goto cleanup;

	data = buf.buf;
	while (i < buf.len) {
		switch (data[i]) {
		case 'P':
			i++;
			if (i + 52 <= buf.len &&
			    !prefixcmp(data + i, " pack-") &&
			    !prefixcmp(data + i + 46, ".pack\n")) {
				get_sha1_hex(data + i + 6, sha1);
				fetch_and_setup_pack_index(packs_head, sha1,
						      base_url);
				i += 51;
				break;
			}
		default:
			while (i < buf.len && data[i] != '\n')
				i++;
		}
		i++;
	}

cleanup:
	free(url);
	return ret;
}
893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997

void release_http_pack_request(struct http_pack_request *preq)
{
	if (preq->packfile != NULL) {
		fclose(preq->packfile);
		preq->packfile = NULL;
		preq->slot->local = NULL;
	}
	if (preq->range_header != NULL) {
		curl_slist_free_all(preq->range_header);
		preq->range_header = NULL;
	}
	preq->slot = NULL;
	free(preq->url);
}

int finish_http_pack_request(struct http_pack_request *preq)
{
	int ret;
	struct packed_git **lst;

	preq->target->pack_size = ftell(preq->packfile);

	if (preq->packfile != NULL) {
		fclose(preq->packfile);
		preq->packfile = NULL;
		preq->slot->local = NULL;
	}

	ret = move_temp_to_file(preq->tmpfile, preq->filename);
	if (ret)
		return ret;

	lst = preq->lst;
	while (*lst != preq->target)
		lst = &((*lst)->next);
	*lst = (*lst)->next;

	if (verify_pack(preq->target))
		return -1;
	install_packed_git(preq->target);

	return 0;
}

struct http_pack_request *new_http_pack_request(
	struct packed_git *target, const char *base_url)
{
	char *url;
	char *filename;
	long prev_posn = 0;
	char range[RANGE_HEADER_SIZE];
	struct strbuf buf = STRBUF_INIT;
	struct http_pack_request *preq;

	preq = xmalloc(sizeof(*preq));
	preq->target = target;
	preq->range_header = NULL;

	end_url_with_slash(&buf, base_url);
	strbuf_addf(&buf, "objects/pack/pack-%s.pack",
		sha1_to_hex(target->sha1));
	url = strbuf_detach(&buf, NULL);
	preq->url = xstrdup(url);

	filename = sha1_pack_name(target->sha1);
	snprintf(preq->filename, sizeof(preq->filename), "%s", filename);
	snprintf(preq->tmpfile, sizeof(preq->tmpfile), "%s.temp", filename);
	preq->packfile = fopen(preq->tmpfile, "a");
	if (!preq->packfile) {
		error("Unable to open local file %s for pack",
		      preq->tmpfile);
		goto abort;
	}

	preq->slot = get_active_slot();
	preq->slot->local = preq->packfile;
	curl_easy_setopt(preq->slot->curl, CURLOPT_FILE, preq->packfile);
	curl_easy_setopt(preq->slot->curl, CURLOPT_WRITEFUNCTION, fwrite);
	curl_easy_setopt(preq->slot->curl, CURLOPT_URL, url);
	curl_easy_setopt(preq->slot->curl, CURLOPT_HTTPHEADER,
		no_pragma_header);

	/*
	 * If there is data present from a previous transfer attempt,
	 * resume where it left off
	 */
	prev_posn = ftell(preq->packfile);
	if (prev_posn>0) {
		if (http_is_verbose)
			fprintf(stderr,
				"Resuming fetch of pack %s at byte %ld\n",
				sha1_to_hex(target->sha1), prev_posn);
		sprintf(range, "Range: bytes=%ld-", prev_posn);
		preq->range_header = curl_slist_append(NULL, range);
		curl_easy_setopt(preq->slot->curl, CURLOPT_HTTPHEADER,
			preq->range_header);
	}

	return preq;

abort:
	free(filename);
	return NULL;
}