http-push.c 49.9 KB
Newer Older
1 2 3 4
#include "cache.h"
#include "commit.h"
#include "tag.h"
#include "blob.h"
5
#include "http.h"
N
Nick Hengeveld 已提交
6
#include "refs.h"
7
#include "diff.h"
N
Nick Hengeveld 已提交
8
#include "revision.h"
9
#include "exec_cmd.h"
10
#include "remote.h"
11
#include "list-objects.h"
12
#include "sigchain.h"
13
#include "argv-array.h"
14

M
Matt Kraai 已提交
15 16 17
#ifdef EXPAT_NEEDS_XMLPARSE_H
#include <xmlparse.h>
#else
18
#include <expat.h>
M
Matt Kraai 已提交
19
#endif
20 21

static const char http_push_usage[] =
S
Stephan Beyer 已提交
22
"git http-push [--all] [--dry-run] [--force] [--verbose] <remote> [<head>...]\n";
23

24 25 26 27 28 29 30 31 32
#ifndef XML_STATUS_OK
enum XML_Status {
  XML_STATUS_OK = 1,
  XML_STATUS_ERROR = 0
};
#define XML_STATUS_OK    1
#define XML_STATUS_ERROR 0
#endif

33
#define PREV_BUF_SIZE 4096
34

N
Nick Hengeveld 已提交
35
/* DAV methods */
36 37 38 39 40 41
#define DAV_LOCK "LOCK"
#define DAV_MKCOL "MKCOL"
#define DAV_MOVE "MOVE"
#define DAV_PROPFIND "PROPFIND"
#define DAV_PUT "PUT"
#define DAV_UNLOCK "UNLOCK"
42
#define DAV_DELETE "DELETE"
N
Nick Hengeveld 已提交
43 44 45 46 47 48 49 50 51 52 53 54 55

/* DAV lock flags */
#define DAV_PROP_LOCKWR (1u << 0)
#define DAV_PROP_LOCKEX (1u << 1)
#define DAV_LOCK_OK (1u << 2)

/* DAV XML properties */
#define DAV_CTX_LOCKENTRY ".multistatus.response.propstat.prop.supportedlock.lockentry"
#define DAV_CTX_LOCKTYPE_WRITE ".multistatus.response.propstat.prop.supportedlock.lockentry.locktype.write"
#define DAV_CTX_LOCKTYPE_EXCLUSIVE ".multistatus.response.propstat.prop.supportedlock.lockentry.lockscope.exclusive"
#define DAV_ACTIVELOCK_OWNER ".prop.lockdiscovery.activelock.owner.href"
#define DAV_ACTIVELOCK_TIMEOUT ".prop.lockdiscovery.activelock.timeout"
#define DAV_ACTIVELOCK_TOKEN ".prop.lockdiscovery.activelock.locktoken.href"
N
Nick Hengeveld 已提交
56 57 58
#define DAV_PROPFIND_RESP ".multistatus.response"
#define DAV_PROPFIND_NAME ".multistatus.response.href"
#define DAV_PROPFIND_COLLECTION ".multistatus.response.propstat.prop.resourcetype.collection"
N
Nick Hengeveld 已提交
59 60

/* DAV request body templates */
N
Nick Hengeveld 已提交
61 62
#define PROPFIND_SUPPORTEDLOCK_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:propfind xmlns:D=\"DAV:\">\n<D:prop xmlns:R=\"%s\">\n<D:supportedlock/>\n</D:prop>\n</D:propfind>"
#define PROPFIND_ALL_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:propfind xmlns:D=\"DAV:\">\n<D:allprop/>\n</D:propfind>"
63 64
#define LOCK_REQUEST "<?xml version=\"1.0\" encoding=\"utf-8\" ?>\n<D:lockinfo xmlns:D=\"DAV:\">\n<D:lockscope><D:exclusive/></D:lockscope>\n<D:locktype><D:write/></D:locktype>\n<D:owner>\n<D:href>mailto:%s</D:href>\n</D:owner>\n</D:lockinfo>"

65 66 67
#define LOCK_TIME 600
#define LOCK_REFRESH 30

68
/* Remember to update object flag allocation in object.h */
69 70 71 72
#define LOCAL    (1u<<16)
#define REMOTE   (1u<<17)
#define FETCHING (1u<<18)
#define PUSHING  (1u<<19)
N
Nick Hengeveld 已提交
73

74 75 76
/* We allow "recursive" symbolic refs. Only within reason, though */
#define MAXDEPTH 5

77 78
static int pushing;
static int aborted;
79
static signed char remote_dir_exists[256];
80

81
static int push_verbosely;
A
Andy Whitcroft 已提交
82
static int push_all = MATCH_REFS_NONE;
83
static int force_all;
84
static int dry_run;
85
static int helper_status;
86

87
static struct object_list *objects;
N
Nick Hengeveld 已提交
88

89
struct repo {
90
	char *url;
91
	char *path;
N
Nick Hengeveld 已提交
92
	int path_len;
93 94 95
	int has_info_refs;
	int can_update_info_refs;
	int has_info_packs;
96
	struct packed_git *packs;
97
	struct remote_lock *locks;
98 99
};

100
static struct repo *repo;
101 102

enum transfer_state {
103 104 105
	NEED_FETCH,
	RUN_FETCH_LOOSE,
	RUN_FETCH_PACKED,
106 107 108 109 110
	NEED_PUSH,
	RUN_MKCOL,
	RUN_PUT,
	RUN_MOVE,
	ABORTED,
111
	COMPLETE
112 113
};

114
struct transfer_request {
N
Nick Hengeveld 已提交
115
	struct object *obj;
116 117
	char *url;
	char *dest;
N
Nick Hengeveld 已提交
118
	struct remote_lock *lock;
119 120 121 122 123 124
	struct curl_slist *headers;
	struct buffer buffer;
	enum transfer_state state;
	CURLcode curl_result;
	char errorstr[CURL_ERROR_SIZE];
	long http_code;
125
	void *userData;
126 127 128 129
	struct active_request_slot *slot;
	struct transfer_request *next;
};

130
static struct transfer_request *request_queue_head;
131

132
struct xml_ctx {
N
Nick Hengeveld 已提交
133 134 135 136 137 138 139
	char *name;
	int len;
	char *cdata;
	void (*userFunc)(struct xml_ctx *ctx, int tag_closed);
	void *userData;
};

140
struct remote_lock {
141
	char *url;
N
Nick Hengeveld 已提交
142
	char *owner;
143
	char *token;
144
	char tmpfile_suffix[41];
N
Nick Hengeveld 已提交
145 146
	time_t start_time;
	long timeout;
147
	int refreshing;
N
Nick Hengeveld 已提交
148 149 150
	struct remote_lock *next;
};

151 152 153 154 155 156 157 158
/* Flags that control remote_ls processing */
#define PROCESS_FILES (1u << 0)
#define PROCESS_DIRS  (1u << 1)
#define RECURSIVE     (1u << 2)

/* Flags that remote_ls passes to callback functions */
#define IS_DIR (1u << 0)

159
struct remote_ls_ctx {
160 161 162 163 164 165 166
	char *path;
	void (*userFunc)(struct remote_ls_ctx *ls);
	void *userData;
	int flags;
	char *dentry_name;
	int dentry_flags;
	struct remote_ls_ctx *parent;
N
Nick Hengeveld 已提交
167 168
};

169 170 171 172 173 174 175
/* get_dav_token_headers options */
enum dav_header_flag {
	DAV_HEADER_IF = (1u << 0),
	DAV_HEADER_LOCK = (1u << 1),
	DAV_HEADER_TIMEOUT = (1u << 2)
};

176
static char *xml_entities(const char *s)
177 178
{
	struct strbuf buf = STRBUF_INIT;
179
	strbuf_addstr_xml_quoted(&buf, s);
180 181 182
	return strbuf_detach(&buf, NULL);
}

183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
static void curl_setup_http_get(CURL *curl, const char *url,
		const char *custom_req)
{
	curl_easy_setopt(curl, CURLOPT_HTTPGET, 1);
	curl_easy_setopt(curl, CURLOPT_URL, url);
	curl_easy_setopt(curl, CURLOPT_CUSTOMREQUEST, custom_req);
	curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, fwrite_null);
}

static void curl_setup_http(CURL *curl, const char *url,
		const char *custom_req, struct buffer *buffer,
		curl_write_callback write_fn)
{
	curl_easy_setopt(curl, CURLOPT_PUT, 1);
	curl_easy_setopt(curl, CURLOPT_URL, url);
	curl_easy_setopt(curl, CURLOPT_INFILE, buffer);
	curl_easy_setopt(curl, CURLOPT_INFILESIZE, buffer->buf.len);
	curl_easy_setopt(curl, CURLOPT_READFUNCTION, fread_buffer);
#ifndef NO_CURL_IOCTL
	curl_easy_setopt(curl, CURLOPT_IOCTLFUNCTION, ioctl_buffer);
203
	curl_easy_setopt(curl, CURLOPT_IOCTLDATA, buffer);
204 205 206 207 208 209 210
#endif
	curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_fn);
	curl_easy_setopt(curl, CURLOPT_NOBODY, 0);
	curl_easy_setopt(curl, CURLOPT_CUSTOMREQUEST, custom_req);
	curl_easy_setopt(curl, CURLOPT_UPLOAD, 1);
}

J
Junio C Hamano 已提交
211 212
static struct curl_slist *get_dav_token_headers(struct remote_lock *lock, enum dav_header_flag options)
{
213 214 215
	struct strbuf buf = STRBUF_INIT;
	struct curl_slist *dav_headers = NULL;

J
Junio C Hamano 已提交
216
	if (options & DAV_HEADER_IF) {
217 218 219 220
		strbuf_addf(&buf, "If: (<%s>)", lock->token);
		dav_headers = curl_slist_append(dav_headers, buf.buf);
		strbuf_reset(&buf);
	}
J
Junio C Hamano 已提交
221
	if (options & DAV_HEADER_LOCK) {
222 223 224 225
		strbuf_addf(&buf, "Lock-Token: <%s>", lock->token);
		dav_headers = curl_slist_append(dav_headers, buf.buf);
		strbuf_reset(&buf);
	}
J
Junio C Hamano 已提交
226
	if (options & DAV_HEADER_TIMEOUT) {
227 228 229 230 231 232 233 234 235
		strbuf_addf(&buf, "Timeout: Second-%ld", lock->timeout);
		dav_headers = curl_slist_append(dav_headers, buf.buf);
		strbuf_reset(&buf);
	}
	strbuf_release(&buf);

	return dav_headers;
}

236
static void finish_request(struct transfer_request *request);
237
static void release_request(struct transfer_request *request);
238

239
static void process_response(void *callback_data)
240
{
241 242
	struct transfer_request *request =
		(struct transfer_request *)callback_data;
243

244
	finish_request(request);
245 246
}

J
Junio C Hamano 已提交
247
#ifdef USE_CURL_MULTI
248

249 250 251
static void start_fetch_loose(struct transfer_request *request)
{
	struct active_request_slot *slot;
252
	struct http_object_request *obj_req;
253

B
brian m. carlson 已提交
254
	obj_req = new_http_object_request(repo->url, request->obj->oid.hash);
255
	if (obj_req == NULL) {
256 257 258 259
		request->state = ABORTED;
		return;
	}

260
	slot = obj_req->slot;
261 262 263
	slot->callback_func = process_response;
	slot->callback_data = request;
	request->slot = slot;
264
	request->userData = obj_req;
265 266 267 268 269

	/* Try to get the request started, abort the request on error */
	request->state = RUN_FETCH_LOOSE;
	if (!start_active_slot(slot)) {
		fprintf(stderr, "Unable to start GET request\n");
270
		repo->can_update_info_refs = 0;
271
		release_http_object_request(obj_req);
272 273 274 275
		release_request(request);
	}
}

J
Junio C Hamano 已提交
276 277
static void start_mkcol(struct transfer_request *request)
{
278
	char *hex = oid_to_hex(&request->obj->oid);
J
Junio C Hamano 已提交
279 280
	struct active_request_slot *slot;

281
	request->url = get_remote_object_url(repo->url, hex, 1);
J
Junio C Hamano 已提交
282 283 284 285

	slot = get_active_slot();
	slot->callback_func = process_response;
	slot->callback_data = request;
286
	curl_setup_http_get(slot->curl, request->url, DAV_MKCOL);
J
Junio C Hamano 已提交
287 288 289 290 291 292 293 294 295 296 297 298 299
	curl_easy_setopt(slot->curl, CURLOPT_ERRORBUFFER, request->errorstr);

	if (start_active_slot(slot)) {
		request->slot = slot;
		request->state = RUN_MKCOL;
	} else {
		request->state = ABORTED;
		free(request->url);
		request->url = NULL;
	}
}
#endif

300 301 302 303 304
static void start_fetch_packed(struct transfer_request *request)
{
	struct packed_git *target;

	struct transfer_request *check_request = request_queue_head;
305
	struct http_pack_request *preq;
306

B
brian m. carlson 已提交
307
	target = find_sha1_pack(request->obj->oid.hash, repo->packs);
308
	if (!target) {
309
		fprintf(stderr, "Unable to fetch %s, will not be able to update server info refs\n", oid_to_hex(&request->obj->oid));
310
		repo->can_update_info_refs = 0;
311 312 313 314 315
		release_request(request);
		return;
	}

	fprintf(stderr,	"Fetching pack %s\n", sha1_to_hex(target->sha1));
316
	fprintf(stderr, " which contains %s\n", oid_to_hex(&request->obj->oid));
317

318 319 320 321 322 323
	preq = new_http_pack_request(target, repo->url);
	if (preq == NULL) {
		repo->can_update_info_refs = 0;
		return;
	}
	preq->lst = &repo->packs;
324 325 326 327

	/* Make sure there isn't another open request for this pack */
	while (check_request) {
		if (check_request->state == RUN_FETCH_PACKED &&
328 329
		    !strcmp(check_request->url, preq->url)) {
			release_http_pack_request(preq);
330 331 332 333 334 335
			release_request(request);
			return;
		}
		check_request = check_request->next;
	}

336 337 338 339
	preq->slot->callback_func = process_response;
	preq->slot->callback_data = request;
	request->slot = preq->slot;
	request->userData = preq;
340 341 342

	/* Try to get the request started, abort the request on error */
	request->state = RUN_FETCH_PACKED;
343
	if (!start_active_slot(preq->slot)) {
344
		fprintf(stderr, "Unable to start GET request\n");
345
		release_http_pack_request(preq);
346
		repo->can_update_info_refs = 0;
347 348 349 350
		release_request(request);
	}
}

351 352
static void start_put(struct transfer_request *request)
{
353
	char *hex = oid_to_hex(&request->obj->oid);
354
	struct active_request_slot *slot;
355
	struct strbuf buf = STRBUF_INIT;
356
	enum object_type type;
357 358 359 360 361
	char hdr[50];
	void *unpacked;
	unsigned long len;
	int hdrlen;
	ssize_t size;
362
	git_zstream stream;
363

B
brian m. carlson 已提交
364
	unpacked = read_sha1_file(request->obj->oid.hash, &type, &len);
365
	hdrlen = xsnprintf(hdr, sizeof(hdr), "%s %lu", typename(type), len) + 1;
366 367

	/* Set it up */
368
	git_deflate_init(&stream, zlib_compression_level);
J
Junio C Hamano 已提交
369
	size = git_deflate_bound(&stream, len + hdrlen);
M
Mike Hommey 已提交
370 371
	strbuf_init(&request->buffer.buf, size);
	request->buffer.posn = 0;
372 373

	/* Compress it */
M
Mike Hommey 已提交
374
	stream.next_out = (unsigned char *)request->buffer.buf.buf;
375 376 377 378 379
	stream.avail_out = size;

	/* First header.. */
	stream.next_in = (void *)hdr;
	stream.avail_in = hdrlen;
380 381
	while (git_deflate(&stream, 0) == Z_OK)
		; /* nothing */
382 383 384 385

	/* Then the data itself.. */
	stream.next_in = unpacked;
	stream.avail_in = len;
386 387 388
	while (git_deflate(&stream, Z_FINISH) == Z_OK)
		; /* nothing */
	git_deflate_end(&stream);
389 390
	free(unpacked);

M
Mike Hommey 已提交
391
	request->buffer.buf.len = stream.total_out;
392

393
	strbuf_addstr(&buf, "Destination: ");
394
	append_remote_object_url(&buf, repo->url, hex, 0);
395 396
	request->dest = strbuf_detach(&buf, NULL);

397
	append_remote_object_url(&buf, repo->url, hex, 0);
398
	strbuf_add(&buf, request->lock->tmpfile_suffix, 41);
399
	request->url = strbuf_detach(&buf, NULL);
400 401

	slot = get_active_slot();
402 403
	slot->callback_func = process_response;
	slot->callback_data = request;
404 405
	curl_setup_http(slot->curl, request->url, DAV_PUT,
			&request->buffer, fwrite_null);
406 407 408 409 410 411 412

	if (start_active_slot(slot)) {
		request->slot = slot;
		request->state = RUN_PUT;
	} else {
		request->state = ABORTED;
		free(request->url);
N
Nick Hengeveld 已提交
413
		request->url = NULL;
414 415 416 417 418 419 420 421 422
	}
}

static void start_move(struct transfer_request *request)
{
	struct active_request_slot *slot;
	struct curl_slist *dav_headers = NULL;

	slot = get_active_slot();
423 424
	slot->callback_func = process_response;
	slot->callback_data = request;
425
	curl_setup_http_get(slot->curl, request->url, DAV_MOVE);
426 427 428 429 430 431 432 433 434 435
	dav_headers = curl_slist_append(dav_headers, request->dest);
	dav_headers = curl_slist_append(dav_headers, "Overwrite: T");
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);

	if (start_active_slot(slot)) {
		request->slot = slot;
		request->state = RUN_MOVE;
	} else {
		request->state = ABORTED;
		free(request->url);
N
Nick Hengeveld 已提交
436
		request->url = NULL;
437 438 439
	}
}

440
static int refresh_lock(struct remote_lock *lock)
441 442
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
443
	struct slot_results results;
444
	struct curl_slist *dav_headers;
445
	int rc = 0;
446

447
	lock->refreshing = 1;
448

449
	dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF | DAV_HEADER_TIMEOUT);
450

451 452
	slot = get_active_slot();
	slot->results = &results;
453
	curl_setup_http_get(slot->curl, lock->url, DAV_LOCK);
454
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);
455

456 457 458 459 460 461 462 463 464 465
	if (start_active_slot(slot)) {
		run_active_slot(slot);
		if (results.curl_result != CURLE_OK) {
			fprintf(stderr, "LOCK HTTP error %ld\n",
				results.http_code);
		} else {
			lock->start_time = time(NULL);
			rc = 1;
		}
	}
N
Nick Hengeveld 已提交
466

467 468
	lock->refreshing = 0;
	curl_slist_free_all(dav_headers);
N
Nick Hengeveld 已提交
469

470 471 472
	return rc;
}

473
static void check_locks(void)
474
{
475
	struct remote_lock *lock = repo->locks;
476 477 478 479 480 481 482 483 484 485 486 487 488
	time_t current_time = time(NULL);
	int time_remaining;

	while (lock) {
		time_remaining = lock->start_time + lock->timeout -
			current_time;
		if (!lock->refreshing && time_remaining < LOCK_REFRESH) {
			if (!refresh_lock(lock)) {
				fprintf(stderr,
					"Unable to refresh lock for %s\n",
					lock->url);
				aborted = 1;
				return;
N
Nick Hengeveld 已提交
489
			}
490
		}
491
		lock = lock->next;
492
	}
N
Nick Hengeveld 已提交
493
}
494

N
Nick Hengeveld 已提交
495 496 497 498 499 500 501 502 503 504 505 506 507
static void release_request(struct transfer_request *request)
{
	struct transfer_request *entry = request_queue_head;

	if (request == request_queue_head) {
		request_queue_head = request->next;
	} else {
		while (entry->next != NULL && entry->next != request)
			entry = entry->next;
		if (entry->next == request)
			entry->next = entry->next->next;
	}

508
	free(request->url);
N
Nick Hengeveld 已提交
509
	free(request);
510 511
}

512 513
static void finish_request(struct transfer_request *request)
{
514
	struct http_pack_request *preq;
515
	struct http_object_request *obj_req;
516 517

	request->curl_result = request->slot->curl_result;
518 519
	request->http_code = request->slot->http_code;
	request->slot = NULL;
520

N
Nick Hengeveld 已提交
521
	/* Keep locks active */
522
	check_locks();
523

524 525
	if (request->headers != NULL)
		curl_slist_free_all(request->headers);
N
Nick Hengeveld 已提交
526 527 528 529 530

	/* URL is reused for MOVE after PUT */
	if (request->state != RUN_PUT) {
		free(request->url);
		request->url = NULL;
N
Nick Hengeveld 已提交
531
	}
N
Nick Hengeveld 已提交
532

N
Nick Hengeveld 已提交
533
	if (request->state == RUN_MKCOL) {
534 535
		if (request->curl_result == CURLE_OK ||
		    request->http_code == 405) {
B
brian m. carlson 已提交
536
			remote_dir_exists[request->obj->oid.hash[0]] = 1;
537 538 539
			start_put(request);
		} else {
			fprintf(stderr, "MKCOL %s failed, aborting (%d/%ld)\n",
540
				oid_to_hex(&request->obj->oid),
541 542 543 544 545 546 547 548 549
				request->curl_result, request->http_code);
			request->state = ABORTED;
			aborted = 1;
		}
	} else if (request->state == RUN_PUT) {
		if (request->curl_result == CURLE_OK) {
			start_move(request);
		} else {
			fprintf(stderr,	"PUT %s failed, aborting (%d/%ld)\n",
550
				oid_to_hex(&request->obj->oid),
551 552 553 554 555 556
				request->curl_result, request->http_code);
			request->state = ABORTED;
			aborted = 1;
		}
	} else if (request->state == RUN_MOVE) {
		if (request->curl_result == CURLE_OK) {
N
Nick Hengeveld 已提交
557 558
			if (push_verbosely)
				fprintf(stderr, "    sent %s\n",
559
					oid_to_hex(&request->obj->oid));
N
Nick Hengeveld 已提交
560 561
			request->obj->flags |= REMOTE;
			release_request(request);
562 563
		} else {
			fprintf(stderr, "MOVE %s failed, aborting (%d/%ld)\n",
564
				oid_to_hex(&request->obj->oid),
565 566 567 568
				request->curl_result, request->http_code);
			request->state = ABORTED;
			aborted = 1;
		}
569
	} else if (request->state == RUN_FETCH_LOOSE) {
570
		obj_req = (struct http_object_request *)request->userData;
571

572 573 574
		if (finish_http_object_request(obj_req) == 0)
			if (obj_req->rename == 0)
				request->obj->flags |= (LOCAL | REMOTE);
575 576

		/* Try fetching packed if necessary */
577 578
		if (request->obj->flags & LOCAL) {
			release_http_object_request(obj_req);
579
			release_request(request);
580
		} else
581 582 583
			start_fetch_packed(request);

	} else if (request->state == RUN_FETCH_PACKED) {
584
		int fail = 1;
585 586 587 588
		if (request->curl_result != CURLE_OK) {
			fprintf(stderr, "Unable to get pack file %s\n%s",
				request->url, curl_errorstr);
		} else {
589 590 591
			preq = (struct http_pack_request *)request->userData;

			if (preq) {
592
				if (finish_http_pack_request(preq) == 0)
593 594
					fail = 0;
				release_http_pack_request(preq);
595 596
			}
		}
597 598
		if (fail)
			repo->can_update_info_refs = 0;
599
		release_request(request);
600 601 602
	}
}

N
Nick Hengeveld 已提交
603
#ifdef USE_CURL_MULTI
604
static int is_running_queue;
605
static int fill_active_slot(void *unused)
606
{
607
	struct transfer_request *request;
608

609
	if (aborted || !is_running_queue)
610
		return 0;
611

612
	for (request = request_queue_head; request; request = request->next) {
613 614
		if (request->state == NEED_FETCH) {
			start_fetch_loose(request);
615
			return 1;
616
		} else if (pushing && request->state == NEED_PUSH) {
B
brian m. carlson 已提交
617
			if (remote_dir_exists[request->obj->oid.hash[0]] == 1) {
618
				start_put(request);
N
Nick Hengeveld 已提交
619
			} else {
620
				start_mkcol(request);
N
Nick Hengeveld 已提交
621
			}
622
			return 1;
623
		}
N
Nick Hengeveld 已提交
624
	}
625
	return 0;
626
}
N
Nick Hengeveld 已提交
627
#endif
628

N
Nick Hengeveld 已提交
629 630
static void get_remote_object_list(unsigned char parent);

631 632 633 634 635 636 637 638 639 640
static void add_fetch_request(struct object *obj)
{
	struct transfer_request *request;

	check_locks();

	/*
	 * Don't fetch the object if it's known to exist locally
	 * or is already in the request queue
	 */
B
brian m. carlson 已提交
641 642
	if (remote_dir_exists[obj->oid.hash[0]] == -1)
		get_remote_object_list(obj->oid.hash[0]);
643 644 645 646 647 648 649 650 651 652 653 654 655
	if (obj->flags & (LOCAL | FETCHING))
		return;

	obj->flags |= FETCHING;
	request = xmalloc(sizeof(*request));
	request->obj = obj;
	request->url = NULL;
	request->lock = NULL;
	request->headers = NULL;
	request->state = NEED_FETCH;
	request->next = request_queue_head;
	request_queue_head = request;

N
Nick Hengeveld 已提交
656
#ifdef USE_CURL_MULTI
657 658
	fill_active_slots();
	step_active_slots();
N
Nick Hengeveld 已提交
659
#endif
660 661
}

N
Nick Hengeveld 已提交
662
static int add_send_request(struct object *obj, struct remote_lock *lock)
663
{
664
	struct transfer_request *request;
665 666
	struct packed_git *target;

667 668 669
	/* Keep locks active */
	check_locks();

N
Nick Hengeveld 已提交
670 671 672 673
	/*
	 * Don't push the object if it's known to exist on the remote
	 * or is already in the request queue
	 */
B
brian m. carlson 已提交
674 675
	if (remote_dir_exists[obj->oid.hash[0]] == -1)
		get_remote_object_list(obj->oid.hash[0]);
N
Nick Hengeveld 已提交
676
	if (obj->flags & (REMOTE | PUSHING))
N
Nick Hengeveld 已提交
677
		return 0;
B
brian m. carlson 已提交
678
	target = find_sha1_pack(obj->oid.hash, repo->packs);
N
Nick Hengeveld 已提交
679 680
	if (target) {
		obj->flags |= REMOTE;
N
Nick Hengeveld 已提交
681
		return 0;
N
Nick Hengeveld 已提交
682
	}
683

N
Nick Hengeveld 已提交
684
	obj->flags |= PUSHING;
685
	request = xmalloc(sizeof(*request));
N
Nick Hengeveld 已提交
686
	request->obj = obj;
687
	request->url = NULL;
N
Nick Hengeveld 已提交
688
	request->lock = lock;
689
	request->headers = NULL;
N
Nick Hengeveld 已提交
690
	request->state = NEED_PUSH;
691 692
	request->next = request_queue_head;
	request_queue_head = request;
693

N
Nick Hengeveld 已提交
694
#ifdef USE_CURL_MULTI
695 696
	fill_active_slots();
	step_active_slots();
N
Nick Hengeveld 已提交
697
#endif
N
Nick Hengeveld 已提交
698 699

	return 1;
700 701
}

702
static int fetch_indices(void)
703
{
T
Tay Ray Chuan 已提交
704
	int ret;
705 706 707

	if (push_verbosely)
		fprintf(stderr, "Getting pack list\n");
N
Nick Hengeveld 已提交
708

T
Tay Ray Chuan 已提交
709 710 711 712 713 714 715
	switch (http_get_info_packs(repo->url, &repo->packs)) {
	case HTTP_OK:
	case HTTP_MISSING_TARGET:
		ret = 0;
		break;
	default:
		ret = -1;
716 717
	}

T
Tay Ray Chuan 已提交
718
	return ret;
719 720
}

721
static void one_remote_object(const unsigned char *sha1)
N
Nick Hengeveld 已提交
722 723 724 725 726 727 728 729 730 731 732 733 734
{
	struct object *obj;

	obj = lookup_object(sha1);
	if (!obj)
		obj = parse_object(sha1);

	/* Ignore remote objects that don't exist locally */
	if (!obj)
		return;

	obj->flags |= REMOTE;
	if (!object_list_contains(objects, obj))
735
		object_list_insert(obj, &objects);
N
Nick Hengeveld 已提交
736 737
}

N
Nick Hengeveld 已提交
738
static void handle_lockprop_ctx(struct xml_ctx *ctx, int tag_closed)
N
Nick Hengeveld 已提交
739
{
N
Nick Hengeveld 已提交
740 741 742 743 744 745 746 747 748 749 750 751 752 753 754
	int *lock_flags = (int *)ctx->userData;

	if (tag_closed) {
		if (!strcmp(ctx->name, DAV_CTX_LOCKENTRY)) {
			if ((*lock_flags & DAV_PROP_LOCKEX) &&
			    (*lock_flags & DAV_PROP_LOCKWR)) {
				*lock_flags |= DAV_LOCK_OK;
			}
			*lock_flags &= DAV_LOCK_OK;
		} else if (!strcmp(ctx->name, DAV_CTX_LOCKTYPE_WRITE)) {
			*lock_flags |= DAV_PROP_LOCKWR;
		} else if (!strcmp(ctx->name, DAV_CTX_LOCKTYPE_EXCLUSIVE)) {
			*lock_flags |= DAV_PROP_LOCKEX;
		}
	}
N
Nick Hengeveld 已提交
755 756
}

N
Nick Hengeveld 已提交
757
static void handle_new_lock_ctx(struct xml_ctx *ctx, int tag_closed)
N
Nick Hengeveld 已提交
758
{
N
Nick Hengeveld 已提交
759
	struct remote_lock *lock = (struct remote_lock *)ctx->userData;
760 761
	git_SHA_CTX sha_ctx;
	unsigned char lock_token_sha1[20];
N
Nick Hengeveld 已提交
762 763 764

	if (tag_closed && ctx->cdata) {
		if (!strcmp(ctx->name, DAV_ACTIVELOCK_OWNER)) {
765
			lock->owner = xstrdup(ctx->cdata);
N
Nick Hengeveld 已提交
766
		} else if (!strcmp(ctx->name, DAV_ACTIVELOCK_TIMEOUT)) {
767 768 769
			const char *arg;
			if (skip_prefix(ctx->cdata, "Second-", &arg))
				lock->timeout = strtol(arg, NULL, 10);
N
Nick Hengeveld 已提交
770
		} else if (!strcmp(ctx->name, DAV_ACTIVELOCK_TOKEN)) {
771
			lock->token = xstrdup(ctx->cdata);
772 773 774 775 776 777 778

			git_SHA1_Init(&sha_ctx);
			git_SHA1_Update(&sha_ctx, lock->token, strlen(lock->token));
			git_SHA1_Final(lock_token_sha1, &sha_ctx);

			lock->tmpfile_suffix[0] = '_';
			memcpy(lock->tmpfile_suffix + 1, sha1_to_hex(lock_token_sha1), 40);
N
Nick Hengeveld 已提交
779
		}
N
Nick Hengeveld 已提交
780 781 782
	}
}

783
static void one_remote_ref(const char *refname);
N
Nick Hengeveld 已提交
784

N
Nick Hengeveld 已提交
785
static void
N
Nick Hengeveld 已提交
786
xml_start_tag(void *userData, const char *name, const char **atts)
N
Nick Hengeveld 已提交
787
{
N
Nick Hengeveld 已提交
788
	struct xml_ctx *ctx = (struct xml_ctx *)userData;
D
Dennis Stosberg 已提交
789
	const char *c = strchr(name, ':');
790
	int old_namelen, new_len;
N
Nick Hengeveld 已提交
791 792 793 794 795 796

	if (c == NULL)
		c = name;
	else
		c++;

797 798
	old_namelen = strlen(ctx->name);
	new_len = old_namelen + strlen(c) + 2;
N
Nick Hengeveld 已提交
799 800 801 802

	if (new_len > ctx->len) {
		ctx->name = xrealloc(ctx->name, new_len);
		ctx->len = new_len;
N
Nick Hengeveld 已提交
803
	}
804
	xsnprintf(ctx->name + old_namelen, ctx->len - old_namelen, ".%s", c);
N
Nick Hengeveld 已提交
805

J
Junio C Hamano 已提交
806 807
	free(ctx->cdata);
	ctx->cdata = NULL;
N
Nick Hengeveld 已提交
808 809

	ctx->userFunc(ctx, 0);
N
Nick Hengeveld 已提交
810 811
}

812
static void
N
Nick Hengeveld 已提交
813
xml_end_tag(void *userData, const char *name)
814
{
N
Nick Hengeveld 已提交
815
	struct xml_ctx *ctx = (struct xml_ctx *)userData;
D
Dennis Stosberg 已提交
816
	const char *c = strchr(name, ':');
N
Nick Hengeveld 已提交
817
	char *ep;
818

N
Nick Hengeveld 已提交
819 820 821 822 823 824 825 826 827
	ctx->userFunc(ctx, 1);

	if (c == NULL)
		c = name;
	else
		c++;

	ep = ctx->name + strlen(ctx->name) - strlen(c) - 1;
	*ep = 0;
828 829 830
}

static void
N
Nick Hengeveld 已提交
831
xml_cdata(void *userData, const XML_Char *s, int len)
832
{
N
Nick Hengeveld 已提交
833
	struct xml_ctx *ctx = (struct xml_ctx *)userData;
J
Junio C Hamano 已提交
834
	free(ctx->cdata);
P
Pierre Habouzit 已提交
835
	ctx->cdata = xmemdupz(s, len);
836 837
}

T
Timo Hirvonen 已提交
838
static struct remote_lock *lock_remote(const char *path, long timeout)
839 840
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
841
	struct slot_results results;
M
Mike Hommey 已提交
842 843
	struct buffer out_buffer = { STRBUF_INIT, 0 };
	struct strbuf in_buffer = STRBUF_INIT;
844
	char *url;
845
	char *ep;
846
	char timeout_header[25];
847
	struct remote_lock *lock = NULL;
848
	struct curl_slist *dav_headers = NULL;
N
Nick Hengeveld 已提交
849
	struct xml_ctx ctx;
850
	char *escaped;
851

852
	url = xstrfmt("%s%s", repo->url, path);
N
Nick Hengeveld 已提交
853

854
	/* Make sure leading directories exist for the remote ref */
855
	ep = strchr(url + strlen(repo->url) + 1, '/');
856
	while (ep) {
857 858
		char saved_character = ep[1];
		ep[1] = '\0';
859
		slot = get_active_slot();
N
Nick Hengeveld 已提交
860
		slot->results = &results;
861
		curl_setup_http_get(slot->curl, url, DAV_MKCOL);
862 863
		if (start_active_slot(slot)) {
			run_active_slot(slot);
N
Nick Hengeveld 已提交
864 865
			if (results.curl_result != CURLE_OK &&
			    results.http_code != 405) {
866 867 868 869 870 871 872
				fprintf(stderr,
					"Unable to create branch path %s\n",
					url);
				free(url);
				return NULL;
			}
		} else {
N
Nick Hengeveld 已提交
873
			fprintf(stderr, "Unable to start MKCOL request\n");
874 875 876
			free(url);
			return NULL;
		}
877
		ep[1] = saved_character;
878 879 880
		ep = strchr(ep + 1, '/');
	}

881
	escaped = xml_entities(ident_default_email());
882 883
	strbuf_addf(&out_buffer.buf, LOCK_REQUEST, escaped);
	free(escaped);
N
Nick Hengeveld 已提交
884

885
	xsnprintf(timeout_header, sizeof(timeout_header), "Timeout: Second-%ld", timeout);
886 887 888 889
	dav_headers = curl_slist_append(dav_headers, timeout_header);
	dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml");

	slot = get_active_slot();
N
Nick Hengeveld 已提交
890
	slot->results = &results;
891
	curl_setup_http(slot->curl, url, DAV_LOCK, &out_buffer, fwrite_buffer);
892
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);
893
	curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer);
894

N
Nick Hengeveld 已提交
895 896
	lock = xcalloc(1, sizeof(*lock));
	lock->timeout = -1;
N
Nick Hengeveld 已提交
897

898 899
	if (start_active_slot(slot)) {
		run_active_slot(slot);
N
Nick Hengeveld 已提交
900
		if (results.curl_result == CURLE_OK) {
M
Mike Hommey 已提交
901 902
			XML_Parser parser = XML_ParserCreate(NULL);
			enum XML_Status result;
N
Nick Hengeveld 已提交
903 904 905 906
			ctx.name = xcalloc(10, 1);
			ctx.len = 0;
			ctx.cdata = NULL;
			ctx.userFunc = handle_new_lock_ctx;
N
Nick Hengeveld 已提交
907
			ctx.userData = lock;
N
Nick Hengeveld 已提交
908 909 910 911
			XML_SetUserData(parser, &ctx);
			XML_SetElementHandler(parser, xml_start_tag,
					      xml_end_tag);
			XML_SetCharacterDataHandler(parser, xml_cdata);
M
Mike Hommey 已提交
912 913
			result = XML_Parse(parser, in_buffer.buf,
					   in_buffer.len, 1);
N
Nick Hengeveld 已提交
914 915 916 917 918
			free(ctx.name);
			if (result != XML_STATUS_OK) {
				fprintf(stderr, "XML error: %s\n",
					XML_ErrorString(
						XML_GetErrorCode(parser)));
N
Nick Hengeveld 已提交
919
				lock->timeout = -1;
N
Nick Hengeveld 已提交
920
			}
M
Mike Hommey 已提交
921
			XML_ParserFree(parser);
922 923
		}
	} else {
N
Nick Hengeveld 已提交
924
		fprintf(stderr, "Unable to start LOCK request\n");
925 926
	}

N
Nick Hengeveld 已提交
927
	curl_slist_free_all(dav_headers);
M
Mike Hommey 已提交
928 929
	strbuf_release(&out_buffer.buf);
	strbuf_release(&in_buffer);
N
Nick Hengeveld 已提交
930

N
Nick Hengeveld 已提交
931
	if (lock->token == NULL || lock->timeout <= 0) {
932 933
		free(lock->token);
		free(lock->owner);
934
		free(url);
N
Nick Hengeveld 已提交
935 936
		free(lock);
		lock = NULL;
N
Nick Hengeveld 已提交
937
	} else {
N
Nick Hengeveld 已提交
938 939
		lock->url = url;
		lock->start_time = time(NULL);
940 941
		lock->next = repo->locks;
		repo->locks = lock;
N
Nick Hengeveld 已提交
942 943
	}

N
Nick Hengeveld 已提交
944
	return lock;
945 946
}

N
Nick Hengeveld 已提交
947
static int unlock_remote(struct remote_lock *lock)
948 949
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
950
	struct slot_results results;
951
	struct remote_lock *prev = repo->locks;
952
	struct curl_slist *dav_headers;
953 954
	int rc = 0;

955
	dav_headers = get_dav_token_headers(lock, DAV_HEADER_LOCK);
956 957

	slot = get_active_slot();
N
Nick Hengeveld 已提交
958
	slot->results = &results;
959
	curl_setup_http_get(slot->curl, lock->url, DAV_UNLOCK);
960 961 962 963
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);

	if (start_active_slot(slot)) {
		run_active_slot(slot);
N
Nick Hengeveld 已提交
964
		if (results.curl_result == CURLE_OK)
965 966
			rc = 1;
		else
967
			fprintf(stderr, "UNLOCK HTTP error %ld\n",
N
Nick Hengeveld 已提交
968
				results.http_code);
969
	} else {
970
		fprintf(stderr, "Unable to start UNLOCK request\n");
971 972 973
	}

	curl_slist_free_all(dav_headers);
974

975 976
	if (repo->locks == lock) {
		repo->locks = lock->next;
977 978 979 980 981 982 983
	} else {
		while (prev && prev->next != lock)
			prev = prev->next;
		if (prev)
			prev->next = prev->next->next;
	}

984
	free(lock->owner);
985 986 987
	free(lock->url);
	free(lock->token);
	free(lock);
988 989 990 991

	return rc;
}

992 993
static void remove_locks(void)
{
994
	struct remote_lock *lock = repo->locks;
995 996 997

	fprintf(stderr, "Removing remote locks...\n");
	while (lock) {
998
		struct remote_lock *next = lock->next;
999
		unlock_remote(lock);
1000
		lock = next;
1001 1002 1003 1004 1005 1006
	}
}

static void remove_locks_on_signal(int signo)
{
	remove_locks();
1007
	sigchain_pop(signo);
1008 1009 1010
	raise(signo);
}

1011 1012 1013
static void remote_ls(const char *path, int flags,
		      void (*userFunc)(struct remote_ls_ctx *ls),
		      void *userData);
N
Nick Hengeveld 已提交
1014

1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
/* extract hex from sharded "xx/x{40}" filename */
static int get_sha1_hex_from_objpath(const char *path, unsigned char *sha1)
{
	char hex[40];

	if (strlen(path) != 41)
		return -1;

	memcpy(hex, path, 2);
	path += 2;
	path++; /* skip '/' */
	memcpy(hex, path, 38);

	return get_sha1_hex(hex, sha1);
}

1031 1032 1033
static void process_ls_object(struct remote_ls_ctx *ls)
{
	unsigned int *parent = (unsigned int *)ls->userData;
1034 1035
	const char *path = ls->dentry_name;
	unsigned char sha1[20];
N
Nick Hengeveld 已提交
1036

1037 1038 1039 1040
	if (!strcmp(ls->path, ls->dentry_name) && (ls->flags & IS_DIR)) {
		remote_dir_exists[*parent] = 1;
		return;
	}
N
Nick Hengeveld 已提交
1041

1042 1043
	if (!skip_prefix(path, "objects/", &path) ||
	    get_sha1_hex_from_objpath(path, sha1))
1044
		return;
1045 1046

	one_remote_object(sha1);
1047
}
N
Nick Hengeveld 已提交
1048

1049 1050 1051 1052 1053 1054
static void process_ls_ref(struct remote_ls_ctx *ls)
{
	if (!strcmp(ls->path, ls->dentry_name) && (ls->dentry_flags & IS_DIR)) {
		fprintf(stderr, "  %s\n", ls->dentry_name);
		return;
	}
N
Nick Hengeveld 已提交
1055

1056 1057 1058
	if (!(ls->dentry_flags & IS_DIR))
		one_remote_ref(ls->dentry_name);
}
N
Nick Hengeveld 已提交
1059

1060 1061 1062
static void handle_remote_ls_ctx(struct xml_ctx *ctx, int tag_closed)
{
	struct remote_ls_ctx *ls = (struct remote_ls_ctx *)ctx->userData;
N
Nick Hengeveld 已提交
1063

1064 1065 1066
	if (tag_closed) {
		if (!strcmp(ctx->name, DAV_PROPFIND_RESP) && ls->dentry_name) {
			if (ls->dentry_flags & IS_DIR) {
1067 1068 1069 1070

				/* ensure collection names end with slash */
				str_end_url_with_slash(ls->dentry_name, &ls->dentry_name);

1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082
				if (ls->flags & PROCESS_DIRS) {
					ls->userFunc(ls);
				}
				if (strcmp(ls->dentry_name, ls->path) &&
				    ls->flags & RECURSIVE) {
					remote_ls(ls->dentry_name,
						  ls->flags,
						  ls->userFunc,
						  ls->userData);
				}
			} else if (ls->flags & PROCESS_FILES) {
				ls->userFunc(ls);
N
Nick Hengeveld 已提交
1083
			}
1084
		} else if (!strcmp(ctx->name, DAV_PROPFIND_NAME) && ctx->cdata) {
1085 1086 1087 1088 1089 1090 1091 1092
			char *path = ctx->cdata;
			if (*ctx->cdata == 'h') {
				path = strstr(path, "//");
				if (path) {
					path = strchr(path+2, '/');
				}
			}
			if (path) {
1093 1094 1095 1096
				const char *url = repo->url;
				if (repo->path)
					url = repo->path;
				if (strncmp(path, url, repo->path_len))
1097
					error("Parsed path '%s' does not match url: '%s'",
1098 1099 1100 1101 1102
					      path, url);
				else {
					path += repo->path_len;
					ls->dentry_name = xstrdup(path);
				}
1103
			}
1104 1105
		} else if (!strcmp(ctx->name, DAV_PROPFIND_COLLECTION)) {
			ls->dentry_flags |= IS_DIR;
N
Nick Hengeveld 已提交
1106
		}
1107
	} else if (!strcmp(ctx->name, DAV_PROPFIND_RESP)) {
J
Junio C Hamano 已提交
1108
		free(ls->dentry_name);
1109 1110
		ls->dentry_name = NULL;
		ls->dentry_flags = 0;
N
Nick Hengeveld 已提交
1111 1112 1113
	}
}

1114 1115 1116 1117 1118 1119
/*
 * NEEDSWORK: remote_ls() ignores info/refs on the remote side.  But it
 * should _only_ heed the information from that file, instead of trying to
 * determine the refs from the remote file system (badly: it does not even
 * know about packed-refs).
 */
1120 1121 1122
static void remote_ls(const char *path, int flags,
		      void (*userFunc)(struct remote_ls_ctx *ls),
		      void *userData)
N
Nick Hengeveld 已提交
1123
{
1124
	char *url = xstrfmt("%s%s", repo->url, path);
N
Nick Hengeveld 已提交
1125
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
1126
	struct slot_results results;
M
Mike Hommey 已提交
1127 1128
	struct strbuf in_buffer = STRBUF_INIT;
	struct buffer out_buffer = { STRBUF_INIT, 0 };
N
Nick Hengeveld 已提交
1129 1130
	struct curl_slist *dav_headers = NULL;
	struct xml_ctx ctx;
1131 1132 1133
	struct remote_ls_ctx ls;

	ls.flags = flags;
1134
	ls.path = xstrdup(path);
1135 1136 1137 1138
	ls.dentry_name = NULL;
	ls.dentry_flags = 0;
	ls.userData = userData;
	ls.userFunc = userFunc;
N
Nick Hengeveld 已提交
1139

M
Mike Hommey 已提交
1140
	strbuf_addf(&out_buffer.buf, PROPFIND_ALL_REQUEST);
N
Nick Hengeveld 已提交
1141 1142 1143 1144 1145

	dav_headers = curl_slist_append(dav_headers, "Depth: 1");
	dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml");

	slot = get_active_slot();
N
Nick Hengeveld 已提交
1146
	slot->results = &results;
1147 1148
	curl_setup_http(slot->curl, url, DAV_PROPFIND,
			&out_buffer, fwrite_buffer);
N
Nick Hengeveld 已提交
1149
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);
1150
	curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer);
N
Nick Hengeveld 已提交
1151 1152 1153

	if (start_active_slot(slot)) {
		run_active_slot(slot);
N
Nick Hengeveld 已提交
1154
		if (results.curl_result == CURLE_OK) {
M
Mike Hommey 已提交
1155 1156
			XML_Parser parser = XML_ParserCreate(NULL);
			enum XML_Status result;
N
Nick Hengeveld 已提交
1157 1158 1159
			ctx.name = xcalloc(10, 1);
			ctx.len = 0;
			ctx.cdata = NULL;
1160 1161
			ctx.userFunc = handle_remote_ls_ctx;
			ctx.userData = &ls;
N
Nick Hengeveld 已提交
1162 1163 1164 1165
			XML_SetUserData(parser, &ctx);
			XML_SetElementHandler(parser, xml_start_tag,
					      xml_end_tag);
			XML_SetCharacterDataHandler(parser, xml_cdata);
M
Mike Hommey 已提交
1166 1167
			result = XML_Parse(parser, in_buffer.buf,
					   in_buffer.len, 1);
N
Nick Hengeveld 已提交
1168 1169 1170 1171 1172 1173 1174
			free(ctx.name);

			if (result != XML_STATUS_OK) {
				fprintf(stderr, "XML error: %s\n",
					XML_ErrorString(
						XML_GetErrorCode(parser)));
			}
M
Mike Hommey 已提交
1175
			XML_ParserFree(parser);
N
Nick Hengeveld 已提交
1176 1177
		}
	} else {
1178
		fprintf(stderr, "Unable to start PROPFIND request\n");
N
Nick Hengeveld 已提交
1179 1180
	}

1181
	free(ls.path);
N
Nick Hengeveld 已提交
1182
	free(url);
M
Mike Hommey 已提交
1183 1184
	strbuf_release(&out_buffer.buf);
	strbuf_release(&in_buffer);
N
Nick Hengeveld 已提交
1185 1186 1187
	curl_slist_free_all(dav_headers);
}

1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200
static void get_remote_object_list(unsigned char parent)
{
	char path[] = "objects/XX/";
	static const char hex[] = "0123456789abcdef";
	unsigned int val = parent;

	path[8] = hex[val >> 4];
	path[9] = hex[val & 0xf];
	remote_dir_exists[val] = 0;
	remote_ls(path, (PROCESS_FILES | PROCESS_DIRS),
		  process_ls_object, &val);
}

N
Nick Hengeveld 已提交
1201
static int locking_available(void)
1202 1203
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
1204
	struct slot_results results;
M
Mike Hommey 已提交
1205 1206
	struct strbuf in_buffer = STRBUF_INIT;
	struct buffer out_buffer = { STRBUF_INIT, 0 };
1207
	struct curl_slist *dav_headers = NULL;
N
Nick Hengeveld 已提交
1208 1209
	struct xml_ctx ctx;
	int lock_flags = 0;
1210
	char *escaped;
1211

1212 1213 1214
	escaped = xml_entities(repo->url);
	strbuf_addf(&out_buffer.buf, PROPFIND_SUPPORTEDLOCK_REQUEST, escaped);
	free(escaped);
1215 1216 1217

	dav_headers = curl_slist_append(dav_headers, "Depth: 0");
	dav_headers = curl_slist_append(dav_headers, "Content-Type: text/xml");
N
Nick Hengeveld 已提交
1218

1219
	slot = get_active_slot();
N
Nick Hengeveld 已提交
1220
	slot->results = &results;
1221 1222
	curl_setup_http(slot->curl, repo->url, DAV_PROPFIND,
			&out_buffer, fwrite_buffer);
1223
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);
1224
	curl_easy_setopt(slot->curl, CURLOPT_FILE, &in_buffer);
1225 1226 1227

	if (start_active_slot(slot)) {
		run_active_slot(slot);
N
Nick Hengeveld 已提交
1228
		if (results.curl_result == CURLE_OK) {
M
Mike Hommey 已提交
1229 1230
			XML_Parser parser = XML_ParserCreate(NULL);
			enum XML_Status result;
N
Nick Hengeveld 已提交
1231 1232 1233 1234 1235 1236 1237 1238
			ctx.name = xcalloc(10, 1);
			ctx.len = 0;
			ctx.cdata = NULL;
			ctx.userFunc = handle_lockprop_ctx;
			ctx.userData = &lock_flags;
			XML_SetUserData(parser, &ctx);
			XML_SetElementHandler(parser, xml_start_tag,
					      xml_end_tag);
M
Mike Hommey 已提交
1239 1240
			result = XML_Parse(parser, in_buffer.buf,
					   in_buffer.len, 1);
N
Nick Hengeveld 已提交
1241 1242 1243 1244 1245 1246 1247 1248
			free(ctx.name);

			if (result != XML_STATUS_OK) {
				fprintf(stderr, "XML error: %s\n",
					XML_ErrorString(
						XML_GetErrorCode(parser)));
				lock_flags = 0;
			}
M
Mike Hommey 已提交
1249
			XML_ParserFree(parser);
1250
			if (!lock_flags)
1251
				error("no DAV locking support on %s",
1252
				      repo->url);
1253 1254 1255

		} else {
			error("Cannot access URL %s, return code %d",
1256
			      repo->url, results.curl_result);
1257
			lock_flags = 0;
1258 1259
		}
	} else {
1260
		error("Unable to start PROPFIND request on %s", repo->url);
1261 1262
	}

M
Mike Hommey 已提交
1263 1264
	strbuf_release(&out_buffer.buf);
	strbuf_release(&in_buffer);
N
Nick Hengeveld 已提交
1265 1266 1267
	curl_slist_free_all(dav_headers);

	return lock_flags;
1268 1269
}

P
Pierre Habouzit 已提交
1270
static struct object_list **add_one_object(struct object *obj, struct object_list **p)
1271 1272 1273 1274 1275 1276 1277 1278
{
	struct object_list *entry = xmalloc(sizeof(struct object_list));
	entry->item = obj;
	entry->next = *p;
	*p = entry;
	return &entry->next;
}

N
Nick Hengeveld 已提交
1279 1280 1281 1282
static struct object_list **process_blob(struct blob *blob,
					 struct object_list **p,
					 struct name_path *path,
					 const char *name)
1283
{
N
Nick Hengeveld 已提交
1284
	struct object *obj = &blob->object;
1285

N
Nick Hengeveld 已提交
1286 1287 1288 1289 1290 1291
	obj->flags |= LOCAL;

	if (obj->flags & (UNINTERESTING | SEEN))
		return p;

	obj->flags |= SEEN;
1292
	return add_one_object(obj, p);
N
Nick Hengeveld 已提交
1293 1294 1295 1296 1297 1298 1299 1300
}

static struct object_list **process_tree(struct tree *tree,
					 struct object_list **p,
					 struct name_path *path,
					 const char *name)
{
	struct object *obj = &tree->object;
1301
	struct tree_desc desc;
1302
	struct name_entry entry;
N
Nick Hengeveld 已提交
1303 1304 1305 1306 1307 1308 1309
	struct name_path me;

	obj->flags |= LOCAL;

	if (obj->flags & (UNINTERESTING | SEEN))
		return p;
	if (parse_tree(tree) < 0)
1310
		die("bad tree object %s", oid_to_hex(&obj->oid));
N
Nick Hengeveld 已提交
1311 1312

	obj->flags |= SEEN;
1313
	name = xstrdup(name);
1314
	p = add_one_object(obj, p);
N
Nick Hengeveld 已提交
1315 1316 1317
	me.up = path;
	me.elem = name;
	me.elem_len = strlen(name);
1318

1319
	init_tree_desc(&desc, tree->buffer, tree->size);
1320

1321 1322 1323
	while (tree_entry(&desc, &entry))
		switch (object_type(entry.mode)) {
		case OBJ_TREE:
1324
			p = process_tree(lookup_tree(entry.sha1), p, &me, name);
1325 1326
			break;
		case OBJ_BLOB:
1327
			p = process_blob(lookup_blob(entry.sha1), p, &me, name);
1328 1329 1330 1331 1332 1333
			break;
		default:
			/* Subproject commit - not in this repository */
			break;
		}

1334
	free_tree_buffer(tree);
N
Nick Hengeveld 已提交
1335
	return p;
1336 1337
}

N
Nick Hengeveld 已提交
1338
static int get_delta(struct rev_info *revs, struct remote_lock *lock)
1339
{
1340
	int i;
1341
	struct commit *commit;
1342
	struct object_list **p = &objects;
N
Nick Hengeveld 已提交
1343
	int count = 0;
1344

N
Nick Hengeveld 已提交
1345 1346 1347 1348
	while ((commit = get_revision(revs)) != NULL) {
		p = process_tree(commit->tree, p, NULL, "");
		commit->object.flags |= LOCAL;
		if (!(commit->object.flags & UNINTERESTING))
N
Nick Hengeveld 已提交
1349
			count += add_send_request(&commit->object, lock);
N
Nick Hengeveld 已提交
1350
	}
1351

1352 1353 1354 1355
	for (i = 0; i < revs->pending.nr; i++) {
		struct object_array_entry *entry = revs->pending.objects + i;
		struct object *obj = entry->item;
		const char *name = entry->name;
1356

N
Nick Hengeveld 已提交
1357 1358
		if (obj->flags & (UNINTERESTING | SEEN))
			continue;
1359
		if (obj->type == OBJ_TAG) {
N
Nick Hengeveld 已提交
1360
			obj->flags |= SEEN;
1361
			p = add_one_object(obj, p);
N
Nick Hengeveld 已提交
1362
			continue;
1363
		}
1364
		if (obj->type == OBJ_TREE) {
N
Nick Hengeveld 已提交
1365 1366
			p = process_tree((struct tree *)obj, p, NULL, name);
			continue;
1367
		}
1368
		if (obj->type == OBJ_BLOB) {
N
Nick Hengeveld 已提交
1369 1370
			p = process_blob((struct blob *)obj, p, NULL, name);
			continue;
1371
		}
1372
		die("unknown pending object %s (%s)", oid_to_hex(&obj->oid), name);
N
Nick Hengeveld 已提交
1373 1374 1375 1376
	}

	while (objects) {
		if (!(objects->item->flags & UNINTERESTING))
N
Nick Hengeveld 已提交
1377
			count += add_send_request(objects->item, lock);
N
Nick Hengeveld 已提交
1378
		objects = objects->next;
1379
	}
N
Nick Hengeveld 已提交
1380 1381

	return count;
1382 1383
}

N
Nick Hengeveld 已提交
1384
static int update_remote(unsigned char *sha1, struct remote_lock *lock)
1385 1386
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
1387
	struct slot_results results;
M
Mike Hommey 已提交
1388
	struct buffer out_buffer = { STRBUF_INIT, 0 };
1389
	struct curl_slist *dav_headers;
1390

1391
	dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF);
1392

M
Mike Hommey 已提交
1393
	strbuf_addf(&out_buffer.buf, "%s\n", sha1_to_hex(sha1));
1394 1395

	slot = get_active_slot();
N
Nick Hengeveld 已提交
1396
	slot->results = &results;
1397 1398
	curl_setup_http(slot->curl, lock->url, DAV_PUT,
			&out_buffer, fwrite_null);
1399 1400 1401 1402
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);

	if (start_active_slot(slot)) {
		run_active_slot(slot);
M
Mike Hommey 已提交
1403
		strbuf_release(&out_buffer.buf);
N
Nick Hengeveld 已提交
1404
		if (results.curl_result != CURLE_OK) {
1405 1406
			fprintf(stderr,
				"PUT error: curl result=%d, HTTP code=%ld\n",
N
Nick Hengeveld 已提交
1407
				results.curl_result, results.http_code);
1408 1409 1410 1411
			/* We should attempt recovery? */
			return 0;
		}
	} else {
M
Mike Hommey 已提交
1412
		strbuf_release(&out_buffer.buf);
1413 1414 1415 1416 1417 1418 1419
		fprintf(stderr, "Unable to start PUT request\n");
		return 0;
	}

	return 1;
}

1420
static struct ref *remote_refs;
N
Nick Hengeveld 已提交
1421

1422
static void one_remote_ref(const char *refname)
N
Nick Hengeveld 已提交
1423 1424
{
	struct ref *ref;
1425
	struct object *obj;
N
Nick Hengeveld 已提交
1426

1427
	ref = alloc_ref(refname);
1428

1429
	if (http_fetch_ref(repo->url, ref) != 0) {
N
Nick Hengeveld 已提交
1430 1431
		fprintf(stderr,
			"Unable to fetch ref %s from %s\n",
1432
			refname, repo->url);
1433
		free(ref);
N
Nick Hengeveld 已提交
1434 1435 1436
		return;
	}

1437 1438 1439 1440
	/*
	 * Fetch a copy of the object if it doesn't exist locally - it
	 * may be required for updating server info later.
	 */
1441 1442
	if (repo->can_update_info_refs && !has_object_file(&ref->old_oid)) {
		obj = lookup_unknown_object(ref->old_oid.hash);
1443 1444
		if (obj) {
			fprintf(stderr,	"  fetch %s for %s\n",
1445
				oid_to_hex(&ref->old_oid), refname);
1446 1447 1448 1449
			add_fetch_request(obj);
		}
	}

1450 1451
	ref->next = remote_refs;
	remote_refs = ref;
N
Nick Hengeveld 已提交
1452 1453 1454 1455
}

static void get_dav_remote_heads(void)
{
1456
	remote_ls("refs/", (PROCESS_FILES | PROCESS_DIRS | RECURSIVE), process_ls_ref, NULL);
N
Nick Hengeveld 已提交
1457 1458
}

1459 1460
static void add_remote_info_ref(struct remote_ls_ctx *ls)
{
M
Mike Hommey 已提交
1461
	struct strbuf *buf = (struct strbuf *)ls->userData;
1462
	struct object *o;
1463 1464
	struct ref *ref;

1465
	ref = alloc_ref(ls->dentry_name);
1466

1467
	if (http_fetch_ref(repo->url, ref) != 0) {
1468 1469
		fprintf(stderr,
			"Unable to fetch ref %s from %s\n",
1470
			ls->dentry_name, repo->url);
1471
		aborted = 1;
1472
		free(ref);
1473 1474 1475
		return;
	}

1476
	o = parse_object(ref->old_oid.hash);
1477 1478 1479
	if (!o) {
		fprintf(stderr,
			"Unable to parse object %s for remote ref %s\n",
1480
			oid_to_hex(&ref->old_oid), ls->dentry_name);
1481
		aborted = 1;
1482
		free(ref);
1483 1484 1485
		return;
	}

1486
	strbuf_addf(buf, "%s\t%s\n",
1487
		    oid_to_hex(&ref->old_oid), ls->dentry_name);
1488

1489
	if (o->type == OBJ_TAG) {
1490
		o = deref_tag(o, ls->dentry_name, 0);
1491 1492
		if (o)
			strbuf_addf(buf, "%s\t%s^{}\n",
1493
				    oid_to_hex(&o->oid), ls->dentry_name);
1494
	}
1495
	free(ref);
1496 1497 1498 1499
}

static void update_remote_info_refs(struct remote_lock *lock)
{
M
Mike Hommey 已提交
1500
	struct buffer buffer = { STRBUF_INIT, 0 };
1501 1502
	struct active_request_slot *slot;
	struct slot_results results;
1503
	struct curl_slist *dav_headers;
1504 1505

	remote_ls("refs/", (PROCESS_FILES | RECURSIVE),
M
Mike Hommey 已提交
1506
		  add_remote_info_ref, &buffer.buf);
1507
	if (!aborted) {
1508
		dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF);
1509 1510 1511

		slot = get_active_slot();
		slot->results = &results;
1512 1513
		curl_setup_http(slot->curl, lock->url, DAV_PUT,
				&buffer, fwrite_null);
1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524
		curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);

		if (start_active_slot(slot)) {
			run_active_slot(slot);
			if (results.curl_result != CURLE_OK) {
				fprintf(stderr,
					"PUT error: curl result=%d, HTTP code=%ld\n",
					results.curl_result, results.http_code);
			}
		}
	}
M
Mike Hommey 已提交
1525
	strbuf_release(&buffer.buf);
1526 1527 1528 1529
}

static int remote_exists(const char *path)
{
1530
	char *url = xstrfmt("%s%s", repo->url, path);
1531
	int ret;
1532 1533


J
Jeff King 已提交
1534
	switch (http_get_strbuf(url, NULL, NULL)) {
1535 1536 1537 1538 1539 1540 1541
	case HTTP_OK:
		ret = 1;
		break;
	case HTTP_MISSING_TARGET:
		ret = 0;
		break;
	case HTTP_ERROR:
J
Jeff King 已提交
1542
		error("unable to access '%s': %s", url, curl_errorstr);
1543 1544
	default:
		ret = -1;
1545
	}
1546 1547
	free(url);
	return ret;
1548 1549
}

T
Timo Hirvonen 已提交
1550
static void fetch_symref(const char *path, char **symref, unsigned char *sha1)
1551
{
1552
	char *url = xstrfmt("%s%s", repo->url, path);
M
Mike Hommey 已提交
1553
	struct strbuf buffer = STRBUF_INIT;
1554
	const char *name;
1555

J
Jeff King 已提交
1556
	if (http_get_strbuf(url, &buffer, NULL) != HTTP_OK)
1557 1558
		die("Couldn't get %s for remote symref\n%s", url,
		    curl_errorstr);
1559 1560
	free(url);

1561
	free(*symref);
1562
	*symref = NULL;
1563
	hashclr(sha1);
1564

M
Mike Hommey 已提交
1565
	if (buffer.len == 0)
1566 1567
		return;

1568 1569 1570
	/* Cut off trailing newline. */
	strbuf_rtrim(&buffer);

1571
	/* If it's a symref, set the refname; otherwise try for a sha1 */
1572 1573
	if (skip_prefix(buffer.buf, "ref: ", &name)) {
		*symref = xmemdupz(name, buffer.len - (name - buffer.buf));
1574
	} else {
M
Mike Hommey 已提交
1575
		get_sha1_hex(buffer.buf, sha1);
1576 1577
	}

M
Mike Hommey 已提交
1578
	strbuf_release(&buffer);
1579 1580
}

1581
static int verify_merge_base(unsigned char *head_sha1, struct ref *remote)
1582
{
1583
	struct commit *head = lookup_commit_or_die(head_sha1, "HEAD");
1584
	struct commit *branch = lookup_commit_or_die(remote->old_oid.hash, remote->name);
1585

1586
	return in_merge_bases(branch, head);
1587 1588
}

1589
static int delete_remote_branch(const char *pattern, int force)
1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628
{
	struct ref *refs = remote_refs;
	struct ref *remote_ref = NULL;
	unsigned char head_sha1[20];
	char *symref = NULL;
	int match;
	int patlen = strlen(pattern);
	int i;
	struct active_request_slot *slot;
	struct slot_results results;
	char *url;

	/* Find the remote branch(es) matching the specified branch name */
	for (match = 0; refs; refs = refs->next) {
		char *name = refs->name;
		int namelen = strlen(name);
		if (namelen < patlen ||
		    memcmp(name + namelen - patlen, pattern, patlen))
			continue;
		if (namelen != patlen && name[namelen - patlen - 1] != '/')
			continue;
		match++;
		remote_ref = refs;
	}
	if (match == 0)
		return error("No remote branch matches %s", pattern);
	if (match != 1)
		return error("More than one remote branch matches %s",
			     pattern);

	/*
	 * Remote HEAD must be a symref (not exactly foolproof; a remote
	 * symlink to a symref will look like a symref)
	 */
	fetch_symref("HEAD", &symref, head_sha1);
	if (!symref)
		return error("Remote HEAD is not a symref");

	/* Remote branch must not be the remote HEAD */
1629
	for (i = 0; symref && i < MAXDEPTH; i++) {
1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640
		if (!strcmp(remote_ref->name, symref))
			return error("Remote branch %s is the current HEAD",
				     remote_ref->name);
		fetch_symref(symref, &symref, head_sha1);
	}

	/* Run extra sanity checks if delete is not forced */
	if (!force) {
		/* Remote HEAD must resolve to a known object */
		if (symref)
			return error("Remote HEAD symrefs too deep");
1641
		if (is_null_sha1(head_sha1))
1642 1643 1644 1645 1646
			return error("Unable to resolve remote HEAD");
		if (!has_sha1_file(head_sha1))
			return error("Remote HEAD resolves to object %s\nwhich does not exist locally, perhaps you need to fetch?", sha1_to_hex(head_sha1));

		/* Remote branch must resolve to a known object */
1647
		if (is_null_oid(&remote_ref->old_oid))
1648 1649
			return error("Unable to resolve remote branch %s",
				     remote_ref->name);
1650 1651
		if (!has_object_file(&remote_ref->old_oid))
			return error("Remote branch %s resolves to object %s\nwhich does not exist locally, perhaps you need to fetch?", remote_ref->name, oid_to_hex(&remote_ref->old_oid));
1652 1653

		/* Remote branch must be an ancestor of remote HEAD */
1654
		if (!verify_merge_base(head_sha1, remote_ref)) {
1655 1656 1657 1658
			return error("The branch '%s' is not an ancestor "
				     "of your current HEAD.\n"
				     "If you are sure you want to delete it,"
				     " run:\n\t'git http-push -D %s %s'",
1659
				     remote_ref->name, repo->url, pattern);
1660 1661 1662 1663 1664
		}
	}

	/* Send delete request */
	fprintf(stderr, "Removing remote branch '%s'\n", remote_ref->name);
1665 1666
	if (dry_run)
		return 0;
1667
	url = xstrfmt("%s%s", repo->url, remote_ref->name);
1668 1669
	slot = get_active_slot();
	slot->results = &results;
1670
	curl_setup_http_get(slot->curl, url, DAV_DELETE);
1671 1672 1673 1674
	if (start_active_slot(slot)) {
		run_active_slot(slot);
		free(url);
		if (results.curl_result != CURLE_OK)
1675
			return error("DELETE request failed (%d/%ld)",
1676 1677 1678 1679 1680 1681 1682 1683 1684
				     results.curl_result, results.http_code);
	} else {
		free(url);
		return error("Unable to start DELETE request");
	}

	return 0;
}

1685
static void run_request_queue(void)
1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703
{
#ifdef USE_CURL_MULTI
	is_running_queue = 1;
	fill_active_slots();
	add_fill_function(NULL, fill_active_slot);
#endif
	do {
		finish_all_active_slots();
#ifdef USE_CURL_MULTI
		fill_active_slots();
#endif
	} while (request_queue_head && !aborted);

#ifdef USE_CURL_MULTI
	is_running_queue = 0;
#endif
}

1704 1705 1706 1707 1708 1709
int main(int argc, char **argv)
{
	struct transfer_request *request;
	struct transfer_request *next_request;
	int nr_refspec = 0;
	char **refspec = NULL;
1710
	struct remote_lock *ref_lock = NULL;
1711
	struct remote_lock *info_ref_lock = NULL;
N
Nick Hengeveld 已提交
1712
	struct rev_info revs;
1713 1714
	int delete_branch = 0;
	int force_delete = 0;
N
Nick Hengeveld 已提交
1715
	int objects_to_send;
1716 1717
	int rc = 0;
	int i;
1718
	int new_refs;
1719
	struct ref *ref, *local_refs;
1720

1721 1722
	git_setup_gettext();

1723 1724
	git_extract_argv0_path(argv[0]);

1725
	repo = xcalloc(1, sizeof(*repo));
1726 1727 1728 1729 1730 1731

	argv++;
	for (i = 1; i < argc; i++, argv++) {
		char *arg = *argv;

		if (*arg == '-') {
N
Nick Hengeveld 已提交
1732
			if (!strcmp(arg, "--all")) {
A
Andy Whitcroft 已提交
1733
				push_all = MATCH_REFS_ALL;
1734 1735 1736 1737 1738 1739
				continue;
			}
			if (!strcmp(arg, "--force")) {
				force_all = 1;
				continue;
			}
1740 1741 1742 1743
			if (!strcmp(arg, "--dry-run")) {
				dry_run = 1;
				continue;
			}
1744 1745 1746 1747
			if (!strcmp(arg, "--helper-status")) {
				helper_status = 1;
				continue;
			}
1748 1749
			if (!strcmp(arg, "--verbose")) {
				push_verbosely = 1;
1750
				http_is_verbose = 1;
1751 1752
				continue;
			}
1753 1754 1755 1756 1757 1758 1759 1760 1761
			if (!strcmp(arg, "-d")) {
				delete_branch = 1;
				continue;
			}
			if (!strcmp(arg, "-D")) {
				delete_branch = 1;
				force_delete = 1;
				continue;
			}
1762 1763
			if (!strcmp(arg, "-h"))
				usage(http_push_usage);
1764
		}
1765
		if (!repo->url) {
N
Nick Hengeveld 已提交
1766
			char *path = strstr(arg, "//");
1767 1768
			str_end_url_with_slash(arg, &repo->url);
			repo->path_len = strlen(repo->url);
N
Nick Hengeveld 已提交
1769
			if (path) {
1770 1771 1772
				repo->path = strchr(path+2, '/');
				if (repo->path)
					repo->path_len = strlen(repo->path);
N
Nick Hengeveld 已提交
1773
			}
1774 1775 1776 1777 1778 1779 1780
			continue;
		}
		refspec = argv;
		nr_refspec = argc - i;
		break;
	}

1781 1782 1783 1784
#ifndef USE_CURL_MULTI
	die("git-push is not available for http/https repository when not compiled with USE_CURL_MULTI");
#endif

1785
	if (!repo->url)
N
Nick Hengeveld 已提交
1786 1787
		usage(http_push_usage);

1788 1789 1790
	if (delete_branch && nr_refspec != 1)
		die("You must specify only one branch name when deleting a remote branch");

1791 1792
	setup_git_directory();

N
Nick Hengeveld 已提交
1793
	memset(remote_dir_exists, -1, 256);
1794

J
Jeff King 已提交
1795
	http_init(NULL, repo->url, 1);
1796

1797
#ifdef USE_CURL_MULTI
1798
	is_running_queue = 0;
1799
#endif
1800

1801
	/* Verify DAV compliance/lock support */
N
Nick Hengeveld 已提交
1802
	if (!locking_available()) {
1803 1804 1805 1806
		rc = 1;
		goto cleanup;
	}

1807
	sigchain_push_common(remove_locks_on_signal);
1808

1809
	/* Check whether the remote has server info files */
1810 1811 1812 1813
	repo->can_update_info_refs = 0;
	repo->has_info_refs = remote_exists("info/refs");
	repo->has_info_packs = remote_exists("objects/info/packs");
	if (repo->has_info_refs) {
1814 1815
		info_ref_lock = lock_remote("info/refs", LOCK_TIME);
		if (info_ref_lock)
1816
			repo->can_update_info_refs = 1;
1817
		else {
1818
			error("cannot lock existing info/refs");
1819 1820 1821
			rc = 1;
			goto cleanup;
		}
1822
	}
1823
	if (repo->has_info_packs)
1824 1825
		fetch_indices();

N
Nick Hengeveld 已提交
1826
	/* Get a list of all local and remote heads to validate refspecs */
1827
	local_refs = get_local_heads();
N
Nick Hengeveld 已提交
1828 1829
	fprintf(stderr, "Fetching remote heads...\n");
	get_dav_remote_heads();
1830
	run_request_queue();
N
Nick Hengeveld 已提交
1831

1832 1833
	/* Remove a remote branch if -d or -D was specified */
	if (delete_branch) {
1834
		if (delete_remote_branch(refspec[0], force_delete) == -1) {
1835 1836
			fprintf(stderr, "Unable to delete remote branch %s\n",
				refspec[0]);
1837 1838 1839
			if (helper_status)
				printf("error %s cannot remove\n", refspec[0]);
		}
1840 1841 1842
		goto cleanup;
	}

N
Nick Hengeveld 已提交
1843
	/* match them up */
1844 1845
	if (match_push_refs(local_refs, &remote_refs,
			    nr_refspec, (const char **) refspec, push_all)) {
1846 1847 1848
		rc = -1;
		goto cleanup;
	}
N
Nick Hengeveld 已提交
1849 1850
	if (!remote_refs) {
		fprintf(stderr, "No refs in common and none specified; doing nothing.\n");
1851 1852
		if (helper_status)
			printf("error null no match\n");
1853 1854
		rc = 0;
		goto cleanup;
N
Nick Hengeveld 已提交
1855 1856
	}

1857
	new_refs = 0;
N
Nick Hengeveld 已提交
1858
	for (ref = remote_refs; ref; ref = ref->next) {
1859
		struct argv_array commit_argv = ARGV_ARRAY_INIT;
1860

N
Nick Hengeveld 已提交
1861 1862
		if (!ref->peer_ref)
			continue;
1863

1864
		if (is_null_oid(&ref->peer_ref->new_oid)) {
1865 1866
			if (delete_remote_branch(ref->name, 1) == -1) {
				error("Could not remove %s", ref->name);
1867 1868
				if (helper_status)
					printf("error %s cannot remove\n", ref->name);
1869 1870
				rc = -4;
			}
1871 1872
			else if (helper_status)
				printf("ok %s\n", ref->name);
1873 1874 1875 1876
			new_refs++;
			continue;
		}

1877
		if (!oidcmp(&ref->old_oid, &ref->peer_ref->new_oid)) {
1878
			if (push_verbosely)
N
Nick Hengeveld 已提交
1879
				fprintf(stderr, "'%s': up-to-date\n", ref->name);
1880 1881
			if (helper_status)
				printf("ok %s up to date\n", ref->name);
N
Nick Hengeveld 已提交
1882 1883 1884 1885
			continue;
		}

		if (!force_all &&
1886
		    !is_null_oid(&ref->old_oid) &&
N
Nick Hengeveld 已提交
1887
		    !ref->force) {
1888
			if (!has_object_file(&ref->old_oid) ||
1889 1890
			    !ref_newer(&ref->peer_ref->new_oid,
				       &ref->old_oid)) {
1891 1892
				/*
				 * We do not have the remote ref, or
N
Nick Hengeveld 已提交
1893 1894 1895 1896 1897 1898
				 * we know that the remote ref is not
				 * an ancestor of what we are trying to
				 * push.  Either way this can be losing
				 * commits at the remote end and likely
				 * we were not up to date to begin with.
				 */
1899 1900 1901
				error("remote '%s' is not an ancestor of\n"
				      "local '%s'.\n"
				      "Maybe you are not up-to-date and "
N
Nick Hengeveld 已提交
1902 1903 1904
				      "need to pull first?",
				      ref->name,
				      ref->peer_ref->name);
1905 1906
				if (helper_status)
					printf("error %s non-fast forward\n", ref->name);
N
Nick Hengeveld 已提交
1907
				rc = -2;
N
Nick Hengeveld 已提交
1908 1909
				continue;
			}
1910
		}
1911
		oidcpy(&ref->new_oid, &ref->peer_ref->new_oid);
N
Nick Hengeveld 已提交
1912 1913 1914 1915 1916
		new_refs++;

		fprintf(stderr, "updating '%s'", ref->name);
		if (strcmp(ref->name, ref->peer_ref->name))
			fprintf(stderr, " using '%s'", ref->peer_ref->name);
1917
		fprintf(stderr, "\n  from %s\n  to   %s\n",
1918
			oid_to_hex(&ref->old_oid), oid_to_hex(&ref->new_oid));
1919 1920 1921
		if (dry_run) {
			if (helper_status)
				printf("ok %s\n", ref->name);
1922
			continue;
1923
		}
1924 1925

		/* Lock remote branch ref */
N
Nick Hengeveld 已提交
1926 1927
		ref_lock = lock_remote(ref->name, LOCK_TIME);
		if (ref_lock == NULL) {
1928
			fprintf(stderr, "Unable to lock remote branch %s\n",
N
Nick Hengeveld 已提交
1929
				ref->name);
1930 1931
			if (helper_status)
				printf("error %s lock error\n", ref->name);
1932 1933 1934 1935
			rc = 1;
			continue;
		}

N
Nick Hengeveld 已提交
1936
		/* Set up revision info for this refspec */
1937 1938
		argv_array_push(&commit_argv, ""); /* ignored */
		argv_array_push(&commit_argv, "--objects");
1939 1940
		argv_array_push(&commit_argv, oid_to_hex(&ref->new_oid));
		if (!push_all && !is_null_oid(&ref->old_oid))
1941
			argv_array_pushf(&commit_argv, "^%s",
1942
					 oid_to_hex(&ref->old_oid));
1943
		init_revisions(&revs, setup_git_directory());
1944
		setup_revisions(commit_argv.argc, commit_argv.argv, &revs, NULL);
1945
		revs.edge_hint = 0; /* just in case */
1946

N
Nick Hengeveld 已提交
1947
		/* Generate a list of objects that need to be pushed */
1948
		pushing = 0;
1949 1950
		if (prepare_revision_walk(&revs))
			die("revision walk setup failed");
1951
		mark_edges_uninteresting(&revs, NULL);
N
Nick Hengeveld 已提交
1952
		objects_to_send = get_delta(&revs, ref_lock);
1953
		finish_all_active_slots();
1954 1955 1956 1957

		/* Push missing objects to remote, this would be a
		   convenient time to pack them first if appropriate. */
		pushing = 1;
N
Nick Hengeveld 已提交
1958 1959 1960
		if (objects_to_send)
			fprintf(stderr, "    sending %d objects\n",
				objects_to_send);
1961 1962

		run_request_queue();
1963 1964

		/* Update the remote branch if all went well */
1965
		if (aborted || !update_remote(ref->new_oid.hash, ref_lock))
N
Nick Hengeveld 已提交
1966
			rc = 1;
1967

N
Nick Hengeveld 已提交
1968 1969
		if (!rc)
			fprintf(stderr, "    done\n");
1970 1971
		if (helper_status)
			printf("%s %s\n", !rc ? "ok" : "error", ref->name);
N
Nick Hengeveld 已提交
1972
		unlock_remote(ref_lock);
1973
		check_locks();
1974
		argv_array_clear(&commit_argv);
1975 1976
	}

1977
	/* Update remote server info if appropriate */
1978 1979
	if (repo->has_info_refs && new_refs) {
		if (info_ref_lock && repo->can_update_info_refs) {
1980
			fprintf(stderr, "Updating remote server info\n");
1981 1982
			if (!dry_run)
				update_remote_info_refs(info_ref_lock);
1983 1984 1985 1986 1987
		} else {
			fprintf(stderr, "Unable to update server info\n");
		}
	}

1988
 cleanup:
1989 1990
	if (info_ref_lock)
		unlock_remote(info_ref_lock);
1991
	free(repo);
1992

1993
	http_cleanup();
1994 1995 1996 1997 1998 1999 2000 2001 2002 2003

	request = request_queue_head;
	while (request != NULL) {
		next_request = request->next;
		release_request(request);
		request = next_request;
	}

	return rc;
}