remote-curl.c 35.2 KB
Newer Older
1
#include "cache.h"
2
#include "config.h"
3
#include "remote.h"
4
#include "connect.h"
5 6 7
#include "strbuf.h"
#include "walker.h"
#include "http.h"
8
#include "exec-cmd.h"
9
#include "run-command.h"
10
#include "pkt-line.h"
11
#include "string-list.h"
12
#include "sideband.h"
J
Junio C Hamano 已提交
13
#include "argv-array.h"
14
#include "credential.h"
15
#include "sha1-array.h"
16
#include "send-pack.h"
17
#include "protocol.h"
18
#include "quote.h"
19

20
static struct remote *remote;
J
Jeff King 已提交
21 22
/* always ends with a trailing slash */
static struct strbuf url = STRBUF_INIT;
23

24 25 26
struct options {
	int verbosity;
	unsigned long depth;
27
	char *deepen_since;
28
	struct string_list deepen_not;
29
	struct string_list push_options;
J
Jeff Hostetler 已提交
30
	char *filter;
31
	unsigned progress : 1,
32
		check_self_contained_and_connected : 1,
33 34
		cloning : 1,
		update_shallow : 1,
35
		followtags : 1,
36
		dry_run : 1,
37
		thin : 1,
38
		/* One of the SEND_PACK_PUSH_CERT_* constants. */
39
		push_cert : 2,
40 41 42
		deepen_relative : 1,
		from_promisor : 1,
		no_dependents : 1;
43 44
};
static struct options options;
45
static struct string_list cas_options = STRING_LIST_INIT_DUP;
46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63

static int set_option(const char *name, const char *value)
{
	if (!strcmp(name, "verbosity")) {
		char *end;
		int v = strtol(value, &end, 10);
		if (value == end || *end)
			return -1;
		options.verbosity = v;
		return 0;
	}
	else if (!strcmp(name, "progress")) {
		if (!strcmp(value, "true"))
			options.progress = 1;
		else if (!strcmp(value, "false"))
			options.progress = 0;
		else
			return -1;
64
		return 0;
65 66 67 68 69 70 71
	}
	else if (!strcmp(name, "depth")) {
		char *end;
		unsigned long v = strtoul(value, &end, 10);
		if (value == end || *end)
			return -1;
		options.depth = v;
72
		return 0;
73
	}
74 75 76 77
	else if (!strcmp(name, "deepen-since")) {
		options.deepen_since = xstrdup(value);
		return 0;
	}
78 79 80 81
	else if (!strcmp(name, "deepen-not")) {
		string_list_append(&options.deepen_not, value);
		return 0;
	}
82 83 84 85 86 87 88 89 90
	else if (!strcmp(name, "deepen-relative")) {
		if (!strcmp(value, "true"))
			options.deepen_relative = 1;
		else if (!strcmp(value, "false"))
			options.deepen_relative = 0;
		else
			return -1;
		return 0;
	}
91 92 93 94 95 96 97
	else if (!strcmp(name, "followtags")) {
		if (!strcmp(value, "true"))
			options.followtags = 1;
		else if (!strcmp(value, "false"))
			options.followtags = 0;
		else
			return -1;
98
		return 0;
99
	}
100 101 102 103 104 105 106 107 108
	else if (!strcmp(name, "dry-run")) {
		if (!strcmp(value, "true"))
			options.dry_run = 1;
		else if (!strcmp(value, "false"))
			options.dry_run = 0;
		else
			return -1;
		return 0;
	}
109 110 111 112 113 114 115 116 117
	else if (!strcmp(name, "check-connectivity")) {
		if (!strcmp(value, "true"))
			options.check_self_contained_and_connected = 1;
		else if (!strcmp(value, "false"))
			options.check_self_contained_and_connected = 0;
		else
			return -1;
		return 0;
	}
118 119 120 121 122 123
	else if (!strcmp(name, "cas")) {
		struct strbuf val = STRBUF_INIT;
		strbuf_addf(&val, "--" CAS_OPT_NAME "=%s", value);
		string_list_append(&cas_options, val.buf);
		strbuf_release(&val);
		return 0;
124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139
	} else if (!strcmp(name, "cloning")) {
		if (!strcmp(value, "true"))
			options.cloning = 1;
		else if (!strcmp(value, "false"))
			options.cloning = 0;
		else
			return -1;
		return 0;
	} else if (!strcmp(name, "update-shallow")) {
		if (!strcmp(value, "true"))
			options.update_shallow = 1;
		else if (!strcmp(value, "false"))
			options.update_shallow = 0;
		else
			return -1;
		return 0;
140 141
	} else if (!strcmp(name, "pushcert")) {
		if (!strcmp(value, "true"))
142
			options.push_cert = SEND_PACK_PUSH_CERT_ALWAYS;
143
		else if (!strcmp(value, "false"))
144 145 146
			options.push_cert = SEND_PACK_PUSH_CERT_NEVER;
		else if (!strcmp(value, "if-asked"))
			options.push_cert = SEND_PACK_PUSH_CERT_IF_ASKED;
147 148 149
		else
			return -1;
		return 0;
150
	} else if (!strcmp(name, "push-option")) {
151 152 153 154 155 156 157 158 159
		if (*value != '"')
			string_list_append(&options.push_options, value);
		else {
			struct strbuf unquoted = STRBUF_INIT;
			if (unquote_c_style(&unquoted, value, NULL) < 0)
				die("invalid quoting in push-option value");
			string_list_append_nodup(&options.push_options,
						 strbuf_detach(&unquoted, NULL));
		}
160
		return 0;
161 162 163 164 165 166 167 168 169 170 171 172 173

#if LIBCURL_VERSION_NUM >= 0x070a08
	} else if (!strcmp(name, "family")) {
		if (!strcmp(value, "ipv4"))
			git_curl_ipresolve = CURL_IPRESOLVE_V4;
		else if (!strcmp(value, "ipv6"))
			git_curl_ipresolve = CURL_IPRESOLVE_V6;
		else if (!strcmp(value, "all"))
			git_curl_ipresolve = CURL_IPRESOLVE_WHATEVER;
		else
			return -1;
		return 0;
#endif /* LIBCURL_VERSION_NUM >= 0x070a08 */
174 175 176 177 178 179
	} else if (!strcmp(name, "from-promisor")) {
		options.from_promisor = 1;
		return 0;
	} else if (!strcmp(name, "no-dependents")) {
		options.no_dependents = 1;
		return 0;
J
Jeff Hostetler 已提交
180
	} else if (!strcmp(name, "filter")) {
181
		options.filter = xstrdup(value);
J
Jeff Hostetler 已提交
182
		return 0;
183
	} else {
184 185 186 187
		return 1 /* unsupported */;
	}
}

188
struct discovery {
189
	char *service;
190 191 192
	char *buf_alloc;
	char *buf;
	size_t len;
193
	struct ref *refs;
194
	struct oid_array shallow;
195
	enum protocol_version version;
196 197 198 199
	unsigned proto_git : 1;
};
static struct discovery *last_discovery;

200 201 202
static struct ref *parse_git_refs(struct discovery *heads, int for_push)
{
	struct ref *list = NULL;
203 204 205 206 207 208
	struct packet_reader reader;

	packet_reader_init(&reader, -1, heads->buf, heads->len,
			   PACKET_READ_CHOMP_NEWLINE |
			   PACKET_READ_GENTLE_ON_EOF);

209 210
	heads->version = discover_version(&reader);
	switch (heads->version) {
211
	case protocol_v2:
212 213 214 215 216 217
		/*
		 * Do nothing.  This isn't a list of refs but rather a
		 * capability advertisement.  Client would have run
		 * 'stateless-connect' so we'll dump this capability listing
		 * and let them request the refs themselves.
		 */
218
		break;
219 220 221 222 223 224 225 226 227
	case protocol_v1:
	case protocol_v0:
		get_remote_heads(&reader, &list, for_push ? REF_NORMAL : 0,
				 NULL, &heads->shallow);
		break;
	case protocol_unknown_version:
		BUG("unknown protocol version");
	}

228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
	return list;
}

static struct ref *parse_info_refs(struct discovery *heads)
{
	char *data, *start, *mid;
	char *ref_name;
	int i = 0;

	struct ref *refs = NULL;
	struct ref *ref = NULL;
	struct ref *last_ref = NULL;

	data = heads->buf;
	start = NULL;
	mid = data;
	while (i < heads->len) {
		if (!start) {
			start = &data[i];
		}
		if (data[i] == '\t')
			mid = &data[i];
		if (data[i] == '\n') {
			if (mid - start != 40)
J
Jeff King 已提交
252 253
				die("%sinfo/refs not valid: is this a git repository?",
				    url.buf);
254 255
			data[i] = 0;
			ref_name = mid + 1;
256
			ref = alloc_ref(ref_name);
257
			get_oid_hex(start, &ref->old_oid);
258 259 260 261 262 263 264 265 266 267 268
			if (!refs)
				refs = ref;
			if (last_ref)
				last_ref->next = ref;
			last_ref = ref;
			start = NULL;
		}
		i++;
	}

	ref = alloc_ref("HEAD");
J
Jeff King 已提交
269
	if (!http_fetch_ref(url.buf, ref) &&
270 271 272 273 274 275 276 277 278 279
	    !resolve_remote_symref(ref, refs)) {
		ref->next = refs;
		refs = ref;
	} else {
		free(ref);
	}

	return refs;
}

280 281 282 283 284
static void free_discovery(struct discovery *d)
{
	if (d) {
		if (d == last_discovery)
			last_discovery = NULL;
285
		free(d->shallow.oid);
286
		free(d->buf_alloc);
287
		free_refs(d->refs);
288
		free(d->service);
289 290 291 292
		free(d);
	}
}

293 294
static int show_http_message(struct strbuf *type, struct strbuf *charset,
			     struct strbuf *msg)
295 296 297 298 299 300 301
{
	const char *p, *eol;

	/*
	 * We only show text/plain parts, as other types are likely
	 * to be ugly to look at on the user's terminal.
	 */
302
	if (strcmp(type->buf, "text/plain"))
303
		return -1;
304 305
	if (charset->len)
		strbuf_reencode(msg, charset->buf, get_log_output_encoding());
306 307 308 309 310 311 312 313 314 315 316 317 318 319

	strbuf_trim(msg);
	if (!msg->len)
		return -1;

	p = msg->buf;
	do {
		eol = strchrnul(p, '\n');
		fprintf(stderr, "remote: %.*s\n", (int)(eol - p), p);
		p = eol + 1;
	} while(*eol);
	return 0;
}

320 321 322 323 324 325 326 327 328 329 330 331 332
static int get_protocol_http_header(enum protocol_version version,
				    struct strbuf *header)
{
	if (version > 0) {
		strbuf_addf(header, GIT_PROTOCOL_HEADER ": version=%d",
			    version);

		return 1;
	}

	return 0;
}

333
static struct discovery *discover_refs(const char *service, int for_push)
334
{
335 336
	struct strbuf exp = STRBUF_INIT;
	struct strbuf type = STRBUF_INIT;
337
	struct strbuf charset = STRBUF_INIT;
338
	struct strbuf buffer = STRBUF_INIT;
J
Jeff King 已提交
339
	struct strbuf refs_url = STRBUF_INIT;
340
	struct strbuf effective_url = STRBUF_INIT;
341 342
	struct strbuf protocol_header = STRBUF_INIT;
	struct string_list extra_headers = STRING_LIST_INIT_DUP;
343
	struct discovery *last = last_discovery;
J
Jeff King 已提交
344
	int http_ret, maybe_smart = 0;
345
	struct http_get_options http_options;
346
	enum protocol_version version = get_protocol_version_config();
347

348 349 350
	if (last && !strcmp(service, last->service))
		return last;
	free_discovery(last);
351

J
Jeff King 已提交
352
	strbuf_addf(&refs_url, "%sinfo/refs", url.buf);
353
	if ((starts_with(url.buf, "http://") || starts_with(url.buf, "https://")) &&
354
	     git_env_bool("GIT_SMART_HTTP", 1)) {
J
Jeff King 已提交
355
		maybe_smart = 1;
J
Jeff King 已提交
356
		if (!strchr(url.buf, '?'))
J
Jeff King 已提交
357
			strbuf_addch(&refs_url, '?');
358
		else
J
Jeff King 已提交
359 360
			strbuf_addch(&refs_url, '&');
		strbuf_addf(&refs_url, "service=%s", service);
361
	}
362

363 364 365 366 367 368 369 370
	/*
	 * NEEDSWORK: If we are trying to use protocol v2 and we are planning
	 * to perform a push, then fallback to v0 since the client doesn't know
	 * how to push yet using v2.
	 */
	if (version == protocol_v2 && !strcmp("git-receive-pack", service))
		version = protocol_v0;

371
	/* Add the extra Git-Protocol header */
372
	if (get_protocol_http_header(version, &protocol_header))
373 374
		string_list_append(&extra_headers, protocol_header.buf);

375 376 377 378 379
	memset(&http_options, 0, sizeof(http_options));
	http_options.content_type = &type;
	http_options.charset = &charset;
	http_options.effective_url = &effective_url;
	http_options.base_url = &url;
380
	http_options.extra_headers = &extra_headers;
J
Jeff King 已提交
381
	http_options.initial_request = 1;
382 383
	http_options.no_cache = 1;
	http_options.keep_error = 1;
J
Jeff King 已提交
384

385
	http_ret = http_get_strbuf(refs_url.buf, &buffer, &http_options);
386 387 388 389
	switch (http_ret) {
	case HTTP_OK:
		break;
	case HTTP_MISSING_TARGET:
390
		show_http_message(&type, &charset, &buffer);
J
Jeff King 已提交
391
		die("repository '%s' not found", url.buf);
392
	case HTTP_NOAUTH:
393
		show_http_message(&type, &charset, &buffer);
J
Jeff King 已提交
394
		die("Authentication failed for '%s'", url.buf);
395
	default:
396
		show_http_message(&type, &charset, &buffer);
J
Jeff King 已提交
397
		die("unable to access '%s': %s", url.buf, curl_errorstr);
398 399
	}

J
Jeff King 已提交
400 401 402
	if (options.verbosity && !starts_with(refs_url.buf, url.buf))
		warning(_("redirecting to %s"), url.buf);

403
	last= xcalloc(1, sizeof(*last_discovery));
404
	last->service = xstrdup(service);
405 406 407
	last->buf_alloc = strbuf_detach(&buffer, &last->len);
	last->buf = last->buf_alloc;

408 409 410 411
	strbuf_addf(&exp, "application/x-%s-advertisement", service);
	if (maybe_smart &&
	    (5 <= last->len && last->buf[4] == '#') &&
	    !strbuf_cmp(&exp, &type)) {
412 413
		char *line;

414 415
		/*
		 * smart HTTP response; validate that the service
416 417
		 * pkt-line matches our request.
		 */
418
		line = packet_read_line_buf(&last->buf, &last->len, NULL);
419 420
		if (!line)
			die("invalid server response; expected service, got flush packet");
421

422
		strbuf_reset(&exp);
423
		strbuf_addf(&exp, "# service=%s", service);
424 425
		if (strcmp(line, exp.buf))
			die("invalid server response; got '%s'", line);
426 427 428 429 430 431
		strbuf_release(&exp);

		/* The header can include additional metadata lines, up
		 * until a packet flush marker.  Ignore these now, but
		 * in the future we might start to scan them.
		 */
432 433
		while (packet_read_line_buf(&last->buf, &last->len, NULL))
			;
434 435

		last->proto_git = 1;
436 437 438
	} else if (maybe_smart &&
		   last->len > 5 && starts_with(last->buf + 4, "version 2")) {
		last->proto_git = 1;
439 440
	}

441 442 443 444 445
	if (last->proto_git)
		last->refs = parse_git_refs(last, for_push);
	else
		last->refs = parse_info_refs(last);

J
Jeff King 已提交
446
	strbuf_release(&refs_url);
447 448
	strbuf_release(&exp);
	strbuf_release(&type);
449
	strbuf_release(&charset);
450
	strbuf_release(&effective_url);
451
	strbuf_release(&buffer);
452 453
	strbuf_release(&protocol_header);
	string_list_clear(&extra_headers, 0);
454 455 456 457 458 459 460 461 462
	last_discovery = last;
	return last;
}

static struct ref *get_refs(int for_push)
{
	struct discovery *heads;

	if (for_push)
463
		heads = discover_refs("git-receive-pack", for_push);
464
	else
465
		heads = discover_refs("git-upload-pack", for_push);
466

467
	return heads->refs;
468 469
}

470 471 472 473 474 475 476
static void output_refs(struct ref *refs)
{
	struct ref *posn;
	for (posn = refs; posn; posn = posn->next) {
		if (posn->symref)
			printf("@%s %s\n", posn->symref, posn->name);
		else
477
			printf("%s %s\n", oid_to_hex(&posn->old_oid), posn->name);
478 479 480 481 482
	}
	printf("\n");
	fflush(stdout);
}

483 484 485
struct rpc_state {
	const char *service_name;
	const char **argv;
486
	struct strbuf *stdin_preamble;
487 488 489
	char *service_url;
	char *hdr_content_type;
	char *hdr_accept;
490
	char *protocol_header;
491 492 493 494 495 496
	char *buf;
	size_t alloc;
	size_t len;
	size_t pos;
	int in;
	int out;
497
	int any_written;
498
	struct strbuf result;
S
Shawn O. Pearce 已提交
499
	unsigned gzip_request : 1;
500
	unsigned initial_buffer : 1;
501 502 503 504 505 506 507 508 509 510
};

static size_t rpc_out(void *ptr, size_t eltsize,
		size_t nmemb, void *buffer_)
{
	size_t max = eltsize * nmemb;
	struct rpc_state *rpc = buffer_;
	size_t avail = rpc->len - rpc->pos;

	if (!avail) {
511
		rpc->initial_buffer = 0;
512
		avail = packet_read(rpc->out, NULL, NULL, rpc->buf, rpc->alloc, 0);
513 514 515 516 517 518
		if (!avail)
			return 0;
		rpc->pos = 0;
		rpc->len = avail;
	}

T
Tay Ray Chuan 已提交
519
	if (max < avail)
520 521 522 523 524 525
		avail = max;
	memcpy(ptr, rpc->buf + rpc->pos, avail);
	rpc->pos += avail;
	return avail;
}

526
#ifndef NO_CURL_IOCTL
527
static curlioerr rpc_ioctl(CURL *handle, int cmd, void *clientp)
528 529 530 531 532 533 534 535 536 537 538 539
{
	struct rpc_state *rpc = clientp;

	switch (cmd) {
	case CURLIOCMD_NOP:
		return CURLIOE_OK;

	case CURLIOCMD_RESTARTREAD:
		if (rpc->initial_buffer) {
			rpc->pos = 0;
			return CURLIOE_OK;
		}
540
		error("unable to rewind rpc post data - try increasing http.postBuffer");
541 542 543 544 545 546 547 548
		return CURLIOE_FAILRESTART;

	default:
		return CURLIOE_UNKNOWNCMD;
	}
}
#endif

549
static size_t rpc_in(char *ptr, size_t eltsize,
550 551 552 553
		size_t nmemb, void *buffer_)
{
	size_t size = eltsize * nmemb;
	struct rpc_state *rpc = buffer_;
554 555
	if (size)
		rpc->any_written = 1;
556 557 558 559
	write_or_die(rpc->in, ptr, size);
	return size;
}

560 561
static int run_slot(struct active_request_slot *slot,
		    struct slot_results *results)
562
{
563
	int err;
564
	struct slot_results results_buf;
565

566 567 568
	if (!results)
		results = &results_buf;

J
Jeff King 已提交
569
	err = run_one_slot(slot, results);
570

571
	if (err != HTTP_OK && err != HTTP_REAUTH) {
572 573 574 575 576 577 578 579 580 581 582 583 584 585
		struct strbuf msg = STRBUF_INIT;
		if (results->http_code && results->http_code != 200)
			strbuf_addf(&msg, "HTTP %ld", results->http_code);
		if (results->curl_result != CURLE_OK) {
			if (msg.len)
				strbuf_addch(&msg, ' ');
			strbuf_addf(&msg, "curl %d", results->curl_result);
			if (curl_errorstr[0]) {
				strbuf_addch(&msg, ' ');
				strbuf_addstr(&msg, curl_errorstr);
			}
		}
		error("RPC failed; %s", msg.buf);
		strbuf_release(&msg);
586 587 588 589 590
	}

	return err;
}

591
static int probe_rpc(struct rpc_state *rpc, struct slot_results *results)
592 593
{
	struct active_request_slot *slot;
594
	struct curl_slist *headers = http_copy_default_headers();
595 596 597 598 599 600 601 602 603 604 605
	struct strbuf buf = STRBUF_INIT;
	int err;

	slot = get_active_slot();

	headers = curl_slist_append(headers, rpc->hdr_content_type);
	headers = curl_slist_append(headers, rpc->hdr_accept);

	curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0);
	curl_easy_setopt(slot->curl, CURLOPT_POST, 1);
	curl_easy_setopt(slot->curl, CURLOPT_URL, rpc->service_url);
606
	curl_easy_setopt(slot->curl, CURLOPT_ENCODING, NULL);
607 608 609 610 611 612
	curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDS, "0000");
	curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDSIZE, 4);
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, headers);
	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, fwrite_buffer);
	curl_easy_setopt(slot->curl, CURLOPT_FILE, &buf);

613
	err = run_slot(slot, results);
614 615 616 617 618 619

	curl_slist_free_all(headers);
	strbuf_release(&buf);
	return err;
}

620 621 622
static curl_off_t xcurl_off_t(size_t len) {
	uintmax_t size = len;
	if (size > maximum_signed_value_of_type(curl_off_t))
623
		die("cannot handle pushes this big");
624
	return (curl_off_t)size;
625 626
}

627 628 629
static int post_rpc(struct rpc_state *rpc)
{
	struct active_request_slot *slot;
630
	struct curl_slist *headers = http_copy_default_headers();
S
Shawn O. Pearce 已提交
631 632
	int use_gzip = rpc->gzip_request;
	char *gzip_body = NULL;
633
	size_t gzip_size = 0;
634
	int err, large_request = 0;
635
	int needs_100_continue = 0;
636 637 638 639 640 641 642 643 644 645 646 647

	/* Try to load the entire request, if we can fit it into the
	 * allocated buffer space we can use HTTP/1.0 and avoid the
	 * chunked encoding mess.
	 */
	while (1) {
		size_t left = rpc->alloc - rpc->len;
		char *buf = rpc->buf + rpc->len;
		int n;

		if (left < LARGE_PACKET_MAX) {
			large_request = 1;
S
Shawn O. Pearce 已提交
648
			use_gzip = 0;
649 650 651
			break;
		}

652
		n = packet_read(rpc->out, NULL, NULL, buf, left, 0);
653 654 655 656 657
		if (!n)
			break;
		rpc->len += n;
	}

658
	if (large_request) {
659 660
		struct slot_results results;

661
		do {
662
			err = probe_rpc(rpc, &results);
663 664
			if (err == HTTP_REAUTH)
				credential_fill(&http_auth);
665 666 667
		} while (err == HTTP_REAUTH);
		if (err != HTTP_OK)
			return -1;
668 669 670

		if (results.auth_avail & CURLAUTH_GSSNEGOTIATE)
			needs_100_continue = 1;
671 672
	}

673 674
	headers = curl_slist_append(headers, rpc->hdr_content_type);
	headers = curl_slist_append(headers, rpc->hdr_accept);
675 676
	headers = curl_slist_append(headers, needs_100_continue ?
		"Expect: 100-continue" : "Expect:");
677

678 679 680 681
	/* Add the extra Git-Protocol header */
	if (rpc->protocol_header)
		headers = curl_slist_append(headers, rpc->protocol_header);

682
retry:
683 684 685
	slot = get_active_slot();

	curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0);
686
	curl_easy_setopt(slot->curl, CURLOPT_POST, 1);
687
	curl_easy_setopt(slot->curl, CURLOPT_URL, rpc->service_url);
688
	curl_easy_setopt(slot->curl, CURLOPT_ENCODING, "");
689 690 691 692 693 694

	if (large_request) {
		/* The request body is large and the size cannot be predicted.
		 * We must use chunked encoding to send it.
		 */
		headers = curl_slist_append(headers, "Transfer-Encoding: chunked");
695
		rpc->initial_buffer = 1;
696 697
		curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, rpc_out);
		curl_easy_setopt(slot->curl, CURLOPT_INFILE, rpc);
698 699 700 701
#ifndef NO_CURL_IOCTL
		curl_easy_setopt(slot->curl, CURLOPT_IOCTLFUNCTION, rpc_ioctl);
		curl_easy_setopt(slot->curl, CURLOPT_IOCTLDATA, rpc);
#endif
702 703 704 705 706
		if (options.verbosity > 1) {
			fprintf(stderr, "POST %s (chunked)\n", rpc->service_name);
			fflush(stderr);
		}

707 708 709 710 711 712 713
	} else if (gzip_body) {
		/*
		 * If we are looping to retry authentication, then the previous
		 * run will have set up the headers and gzip buffer already,
		 * and we just need to send it.
		 */
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDS, gzip_body);
714
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDSIZE_LARGE, xcurl_off_t(gzip_size));
715

S
Shawn O. Pearce 已提交
716 717
	} else if (use_gzip && 1024 < rpc->len) {
		/* The client backend isn't giving us compressed data so
718
		 * we can try to deflate it ourselves, this may save on
S
Shawn O. Pearce 已提交
719 720
		 * the transfer time.
		 */
721
		git_zstream stream;
S
Shawn O. Pearce 已提交
722 723
		int ret;

724
		git_deflate_init_gzip(&stream, Z_BEST_COMPRESSION);
725 726
		gzip_size = git_deflate_bound(&stream, rpc->len);
		gzip_body = xmalloc(gzip_size);
S
Shawn O. Pearce 已提交
727 728 729 730

		stream.next_in = (unsigned char *)rpc->buf;
		stream.avail_in = rpc->len;
		stream.next_out = (unsigned char *)gzip_body;
731
		stream.avail_out = gzip_size;
S
Shawn O. Pearce 已提交
732

733
		ret = git_deflate(&stream, Z_FINISH);
S
Shawn O. Pearce 已提交
734 735 736
		if (ret != Z_STREAM_END)
			die("cannot deflate request; zlib deflate error %d", ret);

737
		ret = git_deflate_end_gently(&stream);
S
Shawn O. Pearce 已提交
738 739 740
		if (ret != Z_OK)
			die("cannot deflate request; zlib end error %d", ret);

741
		gzip_size = stream.total_out;
S
Shawn O. Pearce 已提交
742 743 744

		headers = curl_slist_append(headers, "Content-Encoding: gzip");
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDS, gzip_body);
745
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDSIZE_LARGE, xcurl_off_t(gzip_size));
S
Shawn O. Pearce 已提交
746 747 748 749

		if (options.verbosity > 1) {
			fprintf(stderr, "POST %s (gzip %lu to %lu bytes)\n",
				rpc->service_name,
750
				(unsigned long)rpc->len, (unsigned long)gzip_size);
S
Shawn O. Pearce 已提交
751 752
			fflush(stderr);
		}
753 754 755 756 757
	} else {
		/* We know the complete request size in advance, use the
		 * more normal Content-Length approach.
		 */
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDS, rpc->buf);
758
		curl_easy_setopt(slot->curl, CURLOPT_POSTFIELDSIZE_LARGE, xcurl_off_t(rpc->len));
759 760 761 762 763 764 765 766 767 768 769
		if (options.verbosity > 1) {
			fprintf(stderr, "POST %s (%lu bytes)\n",
				rpc->service_name, (unsigned long)rpc->len);
			fflush(stderr);
		}
	}

	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, headers);
	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, rpc_in);
	curl_easy_setopt(slot->curl, CURLOPT_FILE, rpc);

770 771

	rpc->any_written = 0;
772
	err = run_slot(slot, NULL);
773 774
	if (err == HTTP_REAUTH && !large_request) {
		credential_fill(&http_auth);
775
		goto retry;
776
	}
777 778
	if (err != HTTP_OK)
		err = -1;
779

780 781 782
	if (!rpc->any_written)
		err = -1;

783
	curl_slist_free_all(headers);
S
Shawn O. Pearce 已提交
784
	free(gzip_body);
785 786 787 788 789 790 791
	return err;
}

static int rpc_service(struct rpc_state *rpc, struct discovery *heads)
{
	const char *svc = rpc->service_name;
	struct strbuf buf = STRBUF_INIT;
792
	struct strbuf *preamble = rpc->stdin_preamble;
793
	struct child_process client = CHILD_PROCESS_INIT;
794 795 796 797 798 799 800 801
	int err = 0;

	client.in = -1;
	client.out = -1;
	client.git_cmd = 1;
	client.argv = rpc->argv;
	if (start_command(&client))
		exit(1);
802 803
	if (preamble)
		write_or_die(client.in, preamble->buf, preamble->len);
804 805 806 807 808 809 810 811 812
	if (heads)
		write_or_die(client.in, heads->buf, heads->len);

	rpc->alloc = http_post_buffer;
	rpc->buf = xmalloc(rpc->alloc);
	rpc->in = client.in;
	rpc->out = client.out;
	strbuf_init(&rpc->result, 0);

J
Jeff King 已提交
813
	strbuf_addf(&buf, "%s%s", url.buf, svc);
814 815 816 817 818
	rpc->service_url = strbuf_detach(&buf, NULL);

	strbuf_addf(&buf, "Content-Type: application/x-%s-request", svc);
	rpc->hdr_content_type = strbuf_detach(&buf, NULL);

819
	strbuf_addf(&buf, "Accept: application/x-%s-result", svc);
820 821
	rpc->hdr_accept = strbuf_detach(&buf, NULL);

822 823 824 825 826
	if (get_protocol_http_header(heads->version, &buf))
		rpc->protocol_header = strbuf_detach(&buf, NULL);
	else
		rpc->protocol_header = NULL;

827
	while (!err) {
828
		int n = packet_read(rpc->out, NULL, NULL, rpc->buf, rpc->alloc, 0);
829 830 831 832 833 834 835 836 837
		if (!n)
			break;
		rpc->pos = 0;
		rpc->len = n;
		err |= post_rpc(rpc);
	}

	close(client.in);
	client.in = -1;
838 839 840 841 842 843 844 845
	if (!err) {
		strbuf_read(&rpc->result, client.out, 0);
	} else {
		char buf[4096];
		for (;;)
			if (xread(client.out, buf, sizeof(buf)) <= 0)
				break;
	}
846 847

	close(client.out);
848 849 850 851 852 853
	client.out = -1;

	err |= finish_command(&client);
	free(rpc->service_url);
	free(rpc->hdr_content_type);
	free(rpc->hdr_accept);
854
	free(rpc->protocol_header);
855 856 857 858 859
	free(rpc->buf);
	strbuf_release(&buf);
	return err;
}

860 861
static int fetch_dumb(int nr_heads, struct ref **to_fetch)
{
862
	struct walker *walker;
J
Jeff King 已提交
863
	char **targets;
864 865
	int ret, i;

J
Jeff King 已提交
866
	ALLOC_ARRAY(targets, nr_heads);
867 868
	if (options.depth || options.deepen_since)
		die("dumb http transport does not support shallow capabilities");
869
	for (i = 0; i < nr_heads; i++)
870
		targets[i] = xstrdup(oid_to_hex(&to_fetch[i]->old_oid));
871

J
Jeff King 已提交
872
	walker = get_http_walker(url.buf);
873
	walker->get_verbosely = options.verbosity >= 3;
874 875
	walker->get_recover = 0;
	ret = walker_fetch(walker, nr_heads, targets, NULL, NULL);
876
	walker_free(walker);
877 878 879 880 881

	for (i = 0; i < nr_heads; i++)
		free(targets[i]);
	free(targets);

882
	return ret ? error("fetch failed.") : 0;
883 884
}

885 886 887 888
static int fetch_git(struct discovery *heads,
	int nr_heads, struct ref **to_fetch)
{
	struct rpc_state rpc;
889
	struct strbuf preamble = STRBUF_INIT;
890 891 892 893 894
	int i, err;
	struct argv_array args = ARGV_ARRAY_INIT;

	argv_array_pushl(&args, "fetch-pack", "--stateless-rpc",
			 "--stdin", "--lock-pack", NULL);
895
	if (options.followtags)
896
		argv_array_push(&args, "--include-tag");
897
	if (options.thin)
898 899 900
		argv_array_push(&args, "--thin");
	if (options.verbosity >= 3)
		argv_array_pushl(&args, "-v", "-v", NULL);
901
	if (options.check_self_contained_and_connected)
902
		argv_array_push(&args, "--check-self-contained-and-connected");
903
	if (options.cloning)
904
		argv_array_push(&args, "--cloning");
905
	if (options.update_shallow)
906
		argv_array_push(&args, "--update-shallow");
907
	if (!options.progress)
908 909 910
		argv_array_push(&args, "--no-progress");
	if (options.depth)
		argv_array_pushf(&args, "--depth=%lu", options.depth);
911 912
	if (options.deepen_since)
		argv_array_pushf(&args, "--shallow-since=%s", options.deepen_since);
913 914 915
	for (i = 0; i < options.deepen_not.nr; i++)
		argv_array_pushf(&args, "--shallow-exclude=%s",
				 options.deepen_not.items[i].string);
916 917
	if (options.deepen_relative && options.depth)
		argv_array_push(&args, "--deepen-relative");
918 919 920 921
	if (options.from_promisor)
		argv_array_push(&args, "--from-promisor");
	if (options.no_dependents)
		argv_array_push(&args, "--no-dependents");
J
Jeff Hostetler 已提交
922 923
	if (options.filter)
		argv_array_pushf(&args, "--filter=%s", options.filter);
924
	argv_array_push(&args, url.buf);
925

926 927
	for (i = 0; i < nr_heads; i++) {
		struct ref *ref = to_fetch[i];
928
		if (!*ref->name)
929
			die("cannot fetch by sha1 over smart http");
930
		packet_buf_write(&preamble, "%s %s\n",
931
				 oid_to_hex(&ref->old_oid), ref->name);
932
	}
933
	packet_buf_flush(&preamble);
934 935 936

	memset(&rpc, 0, sizeof(rpc));
	rpc.service_name = "git-upload-pack",
937
	rpc.argv = args.argv;
938
	rpc.stdin_preamble = &preamble;
S
Shawn O. Pearce 已提交
939
	rpc.gzip_request = 1;
940 941 942

	err = rpc_service(&rpc, heads);
	if (rpc.result.len)
J
Jeff King 已提交
943
		write_or_die(1, rpc.result.buf, rpc.result.len);
944
	strbuf_release(&rpc.result);
945
	strbuf_release(&preamble);
946
	argv_array_clear(&args);
947 948 949 950 951
	return err;
}

static int fetch(int nr_heads, struct ref **to_fetch)
{
952
	struct discovery *d = discover_refs("git-upload-pack", 0);
953 954 955 956 957 958
	if (d->proto_git)
		return fetch_git(d, nr_heads, to_fetch);
	else
		return fetch_dumb(nr_heads, to_fetch);
}

959 960 961 962 963 964 965 966
static void parse_fetch(struct strbuf *buf)
{
	struct ref **to_fetch = NULL;
	struct ref *list_head = NULL;
	struct ref **list = &list_head;
	int alloc_heads = 0, nr_heads = 0;

	do {
967 968 969
		const char *p;
		if (skip_prefix(buf->buf, "fetch ", &p)) {
			const char *name;
970
			struct ref *ref;
971
			struct object_id old_oid;
972

973
			if (get_oid_hex(p, &old_oid))
974
				die("protocol error: expected sha/ref, got %s'", p);
975 976 977
			if (p[GIT_SHA1_HEXSZ] == ' ')
				name = p + GIT_SHA1_HEXSZ + 1;
			else if (!p[GIT_SHA1_HEXSZ])
978 979 980 981 982
				name = "";
			else
				die("protocol error: expected sha/ref, got %s'", p);

			ref = alloc_ref(name);
983
			oidcpy(&ref->old_oid, &old_oid);
984 985 986 987 988 989 990 991 992 993 994

			*list = ref;
			list = &ref->next;

			ALLOC_GROW(to_fetch, nr_heads + 1, alloc_heads);
			to_fetch[nr_heads++] = ref;
		}
		else
			die("http transport does not support %s", buf->buf);

		strbuf_reset(buf);
995
		if (strbuf_getline_lf(buf, stdin) == EOF)
996 997 998 999 1000
			return;
		if (!*buf->buf)
			break;
	} while (1);

1001
	if (fetch(nr_heads, to_fetch))
1002 1003 1004 1005 1006 1007 1008 1009 1010
		exit(128); /* error already reported */
	free_refs(list_head);
	free(to_fetch);

	printf("\n");
	fflush(stdout);
	strbuf_reset(buf);
}

1011 1012
static int push_dav(int nr_spec, char **specs)
{
1013 1014
	struct child_process child = CHILD_PROCESS_INIT;
	size_t i;
1015

1016 1017 1018
	child.git_cmd = 1;
	argv_array_push(&child.args, "http-push");
	argv_array_push(&child.args, "--helper-status");
1019
	if (options.dry_run)
1020
		argv_array_push(&child.args, "--dry-run");
1021
	if (options.verbosity > 1)
1022 1023
		argv_array_push(&child.args, "--verbose");
	argv_array_push(&child.args, url.buf);
1024
	for (i = 0; i < nr_spec; i++)
1025
		argv_array_push(&child.args, specs[i]);
1026

1027 1028
	if (run_command(&child))
		die("git-http-push failed");
1029 1030 1031
	return 0;
}

1032 1033 1034
static int push_git(struct discovery *heads, int nr_spec, char **specs)
{
	struct rpc_state rpc;
J
Junio C Hamano 已提交
1035 1036
	int i, err;
	struct argv_array args;
1037
	struct string_list_item *cas_option;
J
Jeff King 已提交
1038
	struct strbuf preamble = STRBUF_INIT;
J
Junio C Hamano 已提交
1039 1040 1041 1042

	argv_array_init(&args);
	argv_array_pushl(&args, "send-pack", "--stateless-rpc", "--helper-status",
			 NULL);
1043 1044

	if (options.thin)
J
Junio C Hamano 已提交
1045
		argv_array_push(&args, "--thin");
1046
	if (options.dry_run)
J
Junio C Hamano 已提交
1047
		argv_array_push(&args, "--dry-run");
1048 1049 1050 1051
	if (options.push_cert == SEND_PACK_PUSH_CERT_ALWAYS)
		argv_array_push(&args, "--signed=yes");
	else if (options.push_cert == SEND_PACK_PUSH_CERT_IF_ASKED)
		argv_array_push(&args, "--signed=if-asked");
1052
	if (options.verbosity == 0)
J
Junio C Hamano 已提交
1053
		argv_array_push(&args, "--quiet");
1054
	else if (options.verbosity > 1)
J
Junio C Hamano 已提交
1055
		argv_array_push(&args, "--verbose");
1056 1057 1058
	for (i = 0; i < options.push_options.nr; i++)
		argv_array_pushf(&args, "--push-option=%s",
				 options.push_options.items[i].string);
J
Junio C Hamano 已提交
1059
	argv_array_push(&args, options.progress ? "--progress" : "--no-progress");
1060
	for_each_string_list_item(cas_option, &cas_options)
J
Junio C Hamano 已提交
1061
		argv_array_push(&args, cas_option->string);
J
Jeff King 已提交
1062
	argv_array_push(&args, url.buf);
J
Jeff King 已提交
1063 1064

	argv_array_push(&args, "--stdin");
1065
	for (i = 0; i < nr_spec; i++)
J
Jeff King 已提交
1066 1067
		packet_buf_write(&preamble, "%s\n", specs[i]);
	packet_buf_flush(&preamble);
1068 1069 1070

	memset(&rpc, 0, sizeof(rpc));
	rpc.service_name = "git-receive-pack",
J
Junio C Hamano 已提交
1071
	rpc.argv = args.argv;
J
Jeff King 已提交
1072
	rpc.stdin_preamble = &preamble;
1073 1074 1075

	err = rpc_service(&rpc, heads);
	if (rpc.result.len)
J
Jeff King 已提交
1076
		write_or_die(1, rpc.result.buf, rpc.result.len);
1077
	strbuf_release(&rpc.result);
J
Jeff King 已提交
1078
	strbuf_release(&preamble);
J
Junio C Hamano 已提交
1079
	argv_array_clear(&args);
1080 1081 1082 1083 1084
	return err;
}

static int push(int nr_spec, char **specs)
{
1085
	struct discovery *heads = discover_refs("git-receive-pack", 1);
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095
	int ret;

	if (heads->proto_git)
		ret = push_git(heads, nr_spec, specs);
	else
		ret = push_dav(nr_spec, specs);
	free_discovery(heads);
	return ret;
}

1096 1097 1098
static void parse_push(struct strbuf *buf)
{
	char **specs = NULL;
1099
	int alloc_spec = 0, nr_spec = 0, i, ret;
1100 1101

	do {
1102
		if (starts_with(buf->buf, "push ")) {
1103 1104 1105 1106 1107 1108 1109
			ALLOC_GROW(specs, nr_spec + 1, alloc_spec);
			specs[nr_spec++] = xstrdup(buf->buf + 5);
		}
		else
			die("http transport does not support %s", buf->buf);

		strbuf_reset(buf);
1110
		if (strbuf_getline_lf(buf, stdin) == EOF)
J
Jim Meyering 已提交
1111
			goto free_specs;
1112 1113 1114 1115
		if (!*buf->buf)
			break;
	} while (1);

1116
	ret = push(nr_spec, specs);
1117 1118
	printf("\n");
	fflush(stdout);
J
Jim Meyering 已提交
1119

1120 1121 1122
	if (ret)
		exit(128); /* error already reported */

J
Jim Meyering 已提交
1123 1124 1125 1126
 free_specs:
	for (i = 0; i < nr_spec; i++)
		free(specs[i]);
	free(specs);
1127 1128
}

1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
/*
 * Used to represent the state of a connection to an HTTP server when
 * communicating using git's wire-protocol version 2.
 */
struct proxy_state {
	char *service_name;
	char *service_url;
	struct curl_slist *headers;
	struct strbuf request_buffer;
	int in;
	int out;
	struct packet_reader reader;
	size_t pos;
	int seen_flush;
};

static void proxy_state_init(struct proxy_state *p, const char *service_name,
			     enum protocol_version version)
{
	struct strbuf buf = STRBUF_INIT;

	memset(p, 0, sizeof(*p));
	p->service_name = xstrdup(service_name);

	p->in = 0;
	p->out = 1;
	strbuf_init(&p->request_buffer, 0);

	strbuf_addf(&buf, "%s%s", url.buf, p->service_name);
	p->service_url = strbuf_detach(&buf, NULL);

	p->headers = http_copy_default_headers();

	strbuf_addf(&buf, "Content-Type: application/x-%s-request", p->service_name);
	p->headers = curl_slist_append(p->headers, buf.buf);
	strbuf_reset(&buf);

	strbuf_addf(&buf, "Accept: application/x-%s-result", p->service_name);
	p->headers = curl_slist_append(p->headers, buf.buf);
	strbuf_reset(&buf);

	p->headers = curl_slist_append(p->headers, "Transfer-Encoding: chunked");

	/* Add the Git-Protocol header */
	if (get_protocol_http_header(version, &buf))
		p->headers = curl_slist_append(p->headers, buf.buf);

	packet_reader_init(&p->reader, p->in, NULL, 0,
			   PACKET_READ_GENTLE_ON_EOF);

	strbuf_release(&buf);
}

static void proxy_state_clear(struct proxy_state *p)
{
	free(p->service_name);
	free(p->service_url);
	curl_slist_free_all(p->headers);
	strbuf_release(&p->request_buffer);
}

/*
 * CURLOPT_READFUNCTION callback function.
 * Attempts to copy over a single packet-line at a time into the
 * curl provided buffer.
 */
static size_t proxy_in(char *buffer, size_t eltsize,
		       size_t nmemb, void *userdata)
{
	size_t max;
	struct proxy_state *p = userdata;
	size_t avail = p->request_buffer.len - p->pos;


	if (eltsize != 1)
		BUG("curl read callback called with size = %"PRIuMAX" != 1",
		    (uintmax_t)eltsize);
	max = nmemb;

	if (!avail) {
		if (p->seen_flush) {
			p->seen_flush = 0;
			return 0;
		}

		strbuf_reset(&p->request_buffer);
		switch (packet_reader_read(&p->reader)) {
		case PACKET_READ_EOF:
			die("unexpected EOF when reading from parent process");
		case PACKET_READ_NORMAL:
			packet_buf_write_len(&p->request_buffer, p->reader.line,
					     p->reader.pktlen);
			break;
		case PACKET_READ_DELIM:
			packet_buf_delim(&p->request_buffer);
			break;
		case PACKET_READ_FLUSH:
			packet_buf_flush(&p->request_buffer);
			p->seen_flush = 1;
			break;
		}
		p->pos = 0;
		avail = p->request_buffer.len;
	}

	if (max < avail)
		avail = max;
	memcpy(buffer, p->request_buffer.buf + p->pos, avail);
	p->pos += avail;
	return avail;
}

static size_t proxy_out(char *buffer, size_t eltsize,
			size_t nmemb, void *userdata)
{
	size_t size;
	struct proxy_state *p = userdata;

	if (eltsize != 1)
		BUG("curl read callback called with size = %"PRIuMAX" != 1",
		    (uintmax_t)eltsize);
	size = nmemb;

	write_or_die(p->out, buffer, size);
	return size;
}

/* Issues a request to the HTTP server configured in `p` */
static int proxy_request(struct proxy_state *p)
{
	struct active_request_slot *slot;

	slot = get_active_slot();

1263
	curl_easy_setopt(slot->curl, CURLOPT_ENCODING, "");
1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325
	curl_easy_setopt(slot->curl, CURLOPT_NOBODY, 0);
	curl_easy_setopt(slot->curl, CURLOPT_POST, 1);
	curl_easy_setopt(slot->curl, CURLOPT_URL, p->service_url);
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, p->headers);

	/* Setup function to read request from client */
	curl_easy_setopt(slot->curl, CURLOPT_READFUNCTION, proxy_in);
	curl_easy_setopt(slot->curl, CURLOPT_READDATA, p);

	/* Setup function to write server response to client */
	curl_easy_setopt(slot->curl, CURLOPT_WRITEFUNCTION, proxy_out);
	curl_easy_setopt(slot->curl, CURLOPT_WRITEDATA, p);

	if (run_slot(slot, NULL) != HTTP_OK)
		return -1;

	return 0;
}

static int stateless_connect(const char *service_name)
{
	struct discovery *discover;
	struct proxy_state p;

	/*
	 * Run the info/refs request and see if the server supports protocol
	 * v2.  If and only if the server supports v2 can we successfully
	 * establish a stateless connection, otherwise we need to tell the
	 * client to fallback to using other transport helper functions to
	 * complete their request.
	 */
	discover = discover_refs(service_name, 0);
	if (discover->version != protocol_v2) {
		printf("fallback\n");
		fflush(stdout);
		return -1;
	} else {
		/* Stateless Connection established */
		printf("\n");
		fflush(stdout);
	}

	proxy_state_init(&p, service_name, discover->version);

	/*
	 * Dump the capability listing that we got from the server earlier
	 * during the info/refs request.
	 */
	write_or_die(p.out, discover->buf, discover->len);

	/* Peek the next packet line.  Until we see EOF keep sending POSTs */
	while (packet_reader_peek(&p.reader) != PACKET_READ_EOF) {
		if (proxy_request(&p)) {
			/* We would have an err here */
			break;
		}
	}

	proxy_state_clear(&p);
	return 0;
}

1326
int cmd_main(int argc, const char **argv)
1327 1328
{
	struct strbuf buf = STRBUF_INIT;
1329
	int nongit;
1330

1331
	setup_git_directory_gently(&nongit);
1332
	if (argc < 2) {
1333
		error("remote-curl: usage: git remote-curl <remote> [<url>]");
1334 1335 1336
		return 1;
	}

1337 1338
	options.verbosity = 1;
	options.progress = !!isatty(2);
1339
	options.thin = 1;
1340
	string_list_init(&options.deepen_not, 1);
1341
	string_list_init(&options.push_options, 1);
1342

1343 1344 1345
	remote = remote_get(argv[1]);

	if (argc > 2) {
J
Jeff King 已提交
1346
		end_url_with_slash(&url, argv[2]);
1347
	} else {
J
Jeff King 已提交
1348
		end_url_with_slash(&url, remote->url[0]);
1349 1350
	}

J
Jeff King 已提交
1351
	http_init(remote, url.buf, 0);
1352

1353
	do {
1354 1355
		const char *arg;

1356
		if (strbuf_getline_lf(&buf, stdin) == EOF) {
1357
			if (ferror(stdin))
1358
				error("remote-curl: error reading command stream from git");
1359 1360 1361
			return 1;
		}
		if (buf.len == 0)
1362
			break;
1363
		if (starts_with(buf.buf, "fetch ")) {
1364
			if (nongit)
1365
				die("remote-curl: fetch attempted without a local repo");
1366 1367
			parse_fetch(&buf);

1368
		} else if (!strcmp(buf.buf, "list") || starts_with(buf.buf, "list ")) {
1369 1370
			int for_push = !!strstr(buf.buf + 4, "for-push");
			output_refs(get_refs(for_push));
1371

1372
		} else if (starts_with(buf.buf, "push ")) {
1373 1374
			parse_push(&buf);

1375 1376
		} else if (skip_prefix(buf.buf, "option ", &arg)) {
			char *value = strchr(arg, ' ');
1377 1378 1379 1380 1381 1382 1383
			int result;

			if (value)
				*value++ = '\0';
			else
				value = "true";

1384
			result = set_option(arg, value);
1385 1386 1387 1388 1389 1390
			if (!result)
				printf("ok\n");
			else if (result < 0)
				printf("error invalid value\n");
			else
				printf("unsupported\n");
1391
			fflush(stdout);
1392

1393
		} else if (!strcmp(buf.buf, "capabilities")) {
1394
			printf("stateless-connect\n");
1395
			printf("fetch\n");
1396
			printf("option\n");
1397
			printf("push\n");
1398
			printf("check-connectivity\n");
1399 1400
			printf("\n");
			fflush(stdout);
1401 1402 1403
		} else if (skip_prefix(buf.buf, "stateless-connect ", &arg)) {
			if (!stateless_connect(arg))
				break;
1404
		} else {
1405
			error("remote-curl: unknown command '%s' from git", buf.buf);
1406 1407 1408 1409
			return 1;
		}
		strbuf_reset(&buf);
	} while (1);
1410 1411 1412

	http_cleanup();

1413 1414
	return 0;
}