http-push.c 49.7 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
	struct strbuf buf = STRBUF_INIT;
214
	struct curl_slist *dav_headers = http_copy_default_headers();
215

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
	}
}

static void start_move(struct transfer_request *request)
{
	struct active_request_slot *slot;
420
	struct curl_slist *dav_headers = http_copy_default_headers();
421 422

	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 = http_copy_default_headers();
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 };
1129
	struct curl_slist *dav_headers = http_copy_default_headers();
N
Nick Hengeveld 已提交
1130
	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 = http_copy_default_headers();
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
static struct object_list **process_blob(struct blob *blob,
J
Jeff King 已提交
1280
					 struct object_list **p)
1281
{
N
Nick Hengeveld 已提交
1282
	struct object *obj = &blob->object;
1283

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

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

	obj->flags |= SEEN;
1290
	return add_one_object(obj, p);
N
Nick Hengeveld 已提交
1291 1292 1293
}

static struct object_list **process_tree(struct tree *tree,
J
Jeff King 已提交
1294
					 struct object_list **p)
N
Nick Hengeveld 已提交
1295 1296
{
	struct object *obj = &tree->object;
1297
	struct tree_desc desc;
1298
	struct name_entry entry;
N
Nick Hengeveld 已提交
1299 1300 1301 1302 1303 1304

	obj->flags |= LOCAL;

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

	obj->flags |= SEEN;
1308
	p = add_one_object(obj, p);
1309

1310
	init_tree_desc(&desc, tree->buffer, tree->size);
1311

1312 1313 1314
	while (tree_entry(&desc, &entry))
		switch (object_type(entry.mode)) {
		case OBJ_TREE:
1315
			p = process_tree(lookup_tree(entry.oid->hash), p);
1316 1317
			break;
		case OBJ_BLOB:
1318
			p = process_blob(lookup_blob(entry.oid->hash), p);
1319 1320 1321 1322 1323 1324
			break;
		default:
			/* Subproject commit - not in this repository */
			break;
		}

1325
	free_tree_buffer(tree);
N
Nick Hengeveld 已提交
1326
	return p;
1327 1328
}

N
Nick Hengeveld 已提交
1329
static int get_delta(struct rev_info *revs, struct remote_lock *lock)
1330
{
1331
	int i;
1332
	struct commit *commit;
1333
	struct object_list **p = &objects;
N
Nick Hengeveld 已提交
1334
	int count = 0;
1335

N
Nick Hengeveld 已提交
1336
	while ((commit = get_revision(revs)) != NULL) {
J
Jeff King 已提交
1337
		p = process_tree(commit->tree, p);
N
Nick Hengeveld 已提交
1338 1339
		commit->object.flags |= LOCAL;
		if (!(commit->object.flags & UNINTERESTING))
N
Nick Hengeveld 已提交
1340
			count += add_send_request(&commit->object, lock);
N
Nick Hengeveld 已提交
1341
	}
1342

1343 1344 1345 1346
	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;
1347

N
Nick Hengeveld 已提交
1348 1349
		if (obj->flags & (UNINTERESTING | SEEN))
			continue;
1350
		if (obj->type == OBJ_TAG) {
N
Nick Hengeveld 已提交
1351
			obj->flags |= SEEN;
1352
			p = add_one_object(obj, p);
N
Nick Hengeveld 已提交
1353
			continue;
1354
		}
1355
		if (obj->type == OBJ_TREE) {
J
Jeff King 已提交
1356
			p = process_tree((struct tree *)obj, p);
N
Nick Hengeveld 已提交
1357
			continue;
1358
		}
1359
		if (obj->type == OBJ_BLOB) {
J
Jeff King 已提交
1360
			p = process_blob((struct blob *)obj, p);
N
Nick Hengeveld 已提交
1361
			continue;
1362
		}
1363
		die("unknown pending object %s (%s)", oid_to_hex(&obj->oid), name);
N
Nick Hengeveld 已提交
1364 1365 1366 1367
	}

	while (objects) {
		if (!(objects->item->flags & UNINTERESTING))
N
Nick Hengeveld 已提交
1368
			count += add_send_request(objects->item, lock);
N
Nick Hengeveld 已提交
1369
		objects = objects->next;
1370
	}
N
Nick Hengeveld 已提交
1371 1372

	return count;
1373 1374
}

N
Nick Hengeveld 已提交
1375
static int update_remote(unsigned char *sha1, struct remote_lock *lock)
1376 1377
{
	struct active_request_slot *slot;
N
Nick Hengeveld 已提交
1378
	struct slot_results results;
M
Mike Hommey 已提交
1379
	struct buffer out_buffer = { STRBUF_INIT, 0 };
1380
	struct curl_slist *dav_headers;
1381

1382
	dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF);
1383

M
Mike Hommey 已提交
1384
	strbuf_addf(&out_buffer.buf, "%s\n", sha1_to_hex(sha1));
1385 1386

	slot = get_active_slot();
N
Nick Hengeveld 已提交
1387
	slot->results = &results;
1388 1389
	curl_setup_http(slot->curl, lock->url, DAV_PUT,
			&out_buffer, fwrite_null);
1390 1391 1392 1393
	curl_easy_setopt(slot->curl, CURLOPT_HTTPHEADER, dav_headers);

	if (start_active_slot(slot)) {
		run_active_slot(slot);
M
Mike Hommey 已提交
1394
		strbuf_release(&out_buffer.buf);
N
Nick Hengeveld 已提交
1395
		if (results.curl_result != CURLE_OK) {
1396 1397
			fprintf(stderr,
				"PUT error: curl result=%d, HTTP code=%ld\n",
N
Nick Hengeveld 已提交
1398
				results.curl_result, results.http_code);
1399 1400 1401 1402
			/* We should attempt recovery? */
			return 0;
		}
	} else {
M
Mike Hommey 已提交
1403
		strbuf_release(&out_buffer.buf);
1404 1405 1406 1407 1408 1409 1410
		fprintf(stderr, "Unable to start PUT request\n");
		return 0;
	}

	return 1;
}

1411
static struct ref *remote_refs;
N
Nick Hengeveld 已提交
1412

1413
static void one_remote_ref(const char *refname)
N
Nick Hengeveld 已提交
1414 1415
{
	struct ref *ref;
1416
	struct object *obj;
N
Nick Hengeveld 已提交
1417

1418
	ref = alloc_ref(refname);
1419

1420
	if (http_fetch_ref(repo->url, ref) != 0) {
N
Nick Hengeveld 已提交
1421 1422
		fprintf(stderr,
			"Unable to fetch ref %s from %s\n",
1423
			refname, repo->url);
1424
		free(ref);
N
Nick Hengeveld 已提交
1425 1426 1427
		return;
	}

1428 1429 1430 1431
	/*
	 * Fetch a copy of the object if it doesn't exist locally - it
	 * may be required for updating server info later.
	 */
1432 1433
	if (repo->can_update_info_refs && !has_object_file(&ref->old_oid)) {
		obj = lookup_unknown_object(ref->old_oid.hash);
1434 1435
		if (obj) {
			fprintf(stderr,	"  fetch %s for %s\n",
1436
				oid_to_hex(&ref->old_oid), refname);
1437 1438 1439 1440
			add_fetch_request(obj);
		}
	}

1441 1442
	ref->next = remote_refs;
	remote_refs = ref;
N
Nick Hengeveld 已提交
1443 1444 1445 1446
}

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

1450 1451
static void add_remote_info_ref(struct remote_ls_ctx *ls)
{
M
Mike Hommey 已提交
1452
	struct strbuf *buf = (struct strbuf *)ls->userData;
1453
	struct object *o;
1454 1455
	struct ref *ref;

1456
	ref = alloc_ref(ls->dentry_name);
1457

1458
	if (http_fetch_ref(repo->url, ref) != 0) {
1459 1460
		fprintf(stderr,
			"Unable to fetch ref %s from %s\n",
1461
			ls->dentry_name, repo->url);
1462
		aborted = 1;
1463
		free(ref);
1464 1465 1466
		return;
	}

1467
	o = parse_object(ref->old_oid.hash);
1468 1469 1470
	if (!o) {
		fprintf(stderr,
			"Unable to parse object %s for remote ref %s\n",
1471
			oid_to_hex(&ref->old_oid), ls->dentry_name);
1472
		aborted = 1;
1473
		free(ref);
1474 1475 1476
		return;
	}

1477
	strbuf_addf(buf, "%s\t%s\n",
1478
		    oid_to_hex(&ref->old_oid), ls->dentry_name);
1479

1480
	if (o->type == OBJ_TAG) {
1481
		o = deref_tag(o, ls->dentry_name, 0);
1482 1483
		if (o)
			strbuf_addf(buf, "%s\t%s^{}\n",
1484
				    oid_to_hex(&o->oid), ls->dentry_name);
1485
	}
1486
	free(ref);
1487 1488 1489 1490
}

static void update_remote_info_refs(struct remote_lock *lock)
{
M
Mike Hommey 已提交
1491
	struct buffer buffer = { STRBUF_INIT, 0 };
1492 1493
	struct active_request_slot *slot;
	struct slot_results results;
1494
	struct curl_slist *dav_headers;
1495 1496

	remote_ls("refs/", (PROCESS_FILES | RECURSIVE),
M
Mike Hommey 已提交
1497
		  add_remote_info_ref, &buffer.buf);
1498
	if (!aborted) {
1499
		dav_headers = get_dav_token_headers(lock, DAV_HEADER_IF);
1500 1501 1502

		slot = get_active_slot();
		slot->results = &results;
1503 1504
		curl_setup_http(slot->curl, lock->url, DAV_PUT,
				&buffer, fwrite_null);
1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515
		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 已提交
1516
	strbuf_release(&buffer.buf);
1517 1518 1519 1520
}

static int remote_exists(const char *path)
{
1521
	char *url = xstrfmt("%s%s", repo->url, path);
1522
	int ret;
1523 1524


J
Jeff King 已提交
1525
	switch (http_get_strbuf(url, NULL, NULL)) {
1526 1527 1528 1529 1530 1531 1532
	case HTTP_OK:
		ret = 1;
		break;
	case HTTP_MISSING_TARGET:
		ret = 0;
		break;
	case HTTP_ERROR:
J
Jeff King 已提交
1533
		error("unable to access '%s': %s", url, curl_errorstr);
1534 1535
	default:
		ret = -1;
1536
	}
1537 1538
	free(url);
	return ret;
1539 1540
}

T
Timo Hirvonen 已提交
1541
static void fetch_symref(const char *path, char **symref, unsigned char *sha1)
1542
{
1543
	char *url = xstrfmt("%s%s", repo->url, path);
M
Mike Hommey 已提交
1544
	struct strbuf buffer = STRBUF_INIT;
1545
	const char *name;
1546

J
Jeff King 已提交
1547
	if (http_get_strbuf(url, &buffer, NULL) != HTTP_OK)
1548 1549
		die("Couldn't get %s for remote symref\n%s", url,
		    curl_errorstr);
1550 1551
	free(url);

1552
	free(*symref);
1553
	*symref = NULL;
1554
	hashclr(sha1);
1555

M
Mike Hommey 已提交
1556
	if (buffer.len == 0)
1557 1558
		return;

1559 1560 1561
	/* Cut off trailing newline. */
	strbuf_rtrim(&buffer);

1562
	/* If it's a symref, set the refname; otherwise try for a sha1 */
1563 1564
	if (skip_prefix(buffer.buf, "ref: ", &name)) {
		*symref = xmemdupz(name, buffer.len - (name - buffer.buf));
1565
	} else {
M
Mike Hommey 已提交
1566
		get_sha1_hex(buffer.buf, sha1);
1567 1568
	}

M
Mike Hommey 已提交
1569
	strbuf_release(&buffer);
1570 1571
}

1572
static int verify_merge_base(unsigned char *head_sha1, struct ref *remote)
1573
{
1574
	struct commit *head = lookup_commit_or_die(head_sha1, "HEAD");
1575
	struct commit *branch = lookup_commit_or_die(remote->old_oid.hash, remote->name);
1576

1577
	return in_merge_bases(branch, head);
1578 1579
}

1580
static int delete_remote_branch(const char *pattern, int force)
1581 1582 1583 1584 1585 1586 1587 1588 1589 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
{
	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 */
1620
	for (i = 0; symref && i < MAXDEPTH; i++) {
1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631
		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");
1632
		if (is_null_sha1(head_sha1))
1633 1634 1635 1636 1637
			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 */
1638
		if (is_null_oid(&remote_ref->old_oid))
1639 1640
			return error("Unable to resolve remote branch %s",
				     remote_ref->name);
1641 1642
		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));
1643 1644

		/* Remote branch must be an ancestor of remote HEAD */
1645
		if (!verify_merge_base(head_sha1, remote_ref)) {
1646 1647 1648 1649
			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'",
1650
				     remote_ref->name, repo->url, pattern);
1651 1652 1653 1654 1655
		}
	}

	/* Send delete request */
	fprintf(stderr, "Removing remote branch '%s'\n", remote_ref->name);
1656 1657
	if (dry_run)
		return 0;
1658
	url = xstrfmt("%s%s", repo->url, remote_ref->name);
1659 1660
	slot = get_active_slot();
	slot->results = &results;
1661
	curl_setup_http_get(slot->curl, url, DAV_DELETE);
1662 1663 1664 1665
	if (start_active_slot(slot)) {
		run_active_slot(slot);
		free(url);
		if (results.curl_result != CURLE_OK)
1666
			return error("DELETE request failed (%d/%ld)",
1667 1668 1669 1670 1671 1672 1673 1674 1675
				     results.curl_result, results.http_code);
	} else {
		free(url);
		return error("Unable to start DELETE request");
	}

	return 0;
}

1676
static void run_request_queue(void)
1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694
{
#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
}

1695
int cmd_main(int argc, const char **argv)
1696 1697 1698 1699
{
	struct transfer_request *request;
	struct transfer_request *next_request;
	int nr_refspec = 0;
1700
	const char **refspec = NULL;
1701
	struct remote_lock *ref_lock = NULL;
1702
	struct remote_lock *info_ref_lock = NULL;
N
Nick Hengeveld 已提交
1703
	struct rev_info revs;
1704 1705
	int delete_branch = 0;
	int force_delete = 0;
N
Nick Hengeveld 已提交
1706
	int objects_to_send;
1707 1708
	int rc = 0;
	int i;
1709
	int new_refs;
1710
	struct ref *ref, *local_refs;
1711

1712
	repo = xcalloc(1, sizeof(*repo));
1713 1714 1715

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

		if (*arg == '-') {
N
Nick Hengeveld 已提交
1719
			if (!strcmp(arg, "--all")) {
A
Andy Whitcroft 已提交
1720
				push_all = MATCH_REFS_ALL;
1721 1722 1723 1724 1725 1726
				continue;
			}
			if (!strcmp(arg, "--force")) {
				force_all = 1;
				continue;
			}
1727 1728 1729 1730
			if (!strcmp(arg, "--dry-run")) {
				dry_run = 1;
				continue;
			}
1731 1732 1733 1734
			if (!strcmp(arg, "--helper-status")) {
				helper_status = 1;
				continue;
			}
1735 1736
			if (!strcmp(arg, "--verbose")) {
				push_verbosely = 1;
1737
				http_is_verbose = 1;
1738 1739
				continue;
			}
1740 1741 1742 1743 1744 1745 1746 1747 1748
			if (!strcmp(arg, "-d")) {
				delete_branch = 1;
				continue;
			}
			if (!strcmp(arg, "-D")) {
				delete_branch = 1;
				force_delete = 1;
				continue;
			}
1749 1750
			if (!strcmp(arg, "-h"))
				usage(http_push_usage);
1751
		}
1752
		if (!repo->url) {
N
Nick Hengeveld 已提交
1753
			char *path = strstr(arg, "//");
1754 1755
			str_end_url_with_slash(arg, &repo->url);
			repo->path_len = strlen(repo->url);
N
Nick Hengeveld 已提交
1756
			if (path) {
1757 1758 1759
				repo->path = strchr(path+2, '/');
				if (repo->path)
					repo->path_len = strlen(repo->path);
N
Nick Hengeveld 已提交
1760
			}
1761 1762 1763 1764 1765 1766 1767
			continue;
		}
		refspec = argv;
		nr_refspec = argc - i;
		break;
	}

1768 1769 1770 1771
#ifndef USE_CURL_MULTI
	die("git-push is not available for http/https repository when not compiled with USE_CURL_MULTI");
#endif

1772
	if (!repo->url)
N
Nick Hengeveld 已提交
1773 1774
		usage(http_push_usage);

1775 1776 1777
	if (delete_branch && nr_refspec != 1)
		die("You must specify only one branch name when deleting a remote branch");

1778 1779
	setup_git_directory();

N
Nick Hengeveld 已提交
1780
	memset(remote_dir_exists, -1, 256);
1781

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

1784
#ifdef USE_CURL_MULTI
1785
	is_running_queue = 0;
1786
#endif
1787

1788
	/* Verify DAV compliance/lock support */
N
Nick Hengeveld 已提交
1789
	if (!locking_available()) {
1790 1791 1792 1793
		rc = 1;
		goto cleanup;
	}

1794
	sigchain_push_common(remove_locks_on_signal);
1795

1796
	/* Check whether the remote has server info files */
1797 1798 1799 1800
	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) {
1801 1802
		info_ref_lock = lock_remote("info/refs", LOCK_TIME);
		if (info_ref_lock)
1803
			repo->can_update_info_refs = 1;
1804
		else {
1805
			error("cannot lock existing info/refs");
1806 1807 1808
			rc = 1;
			goto cleanup;
		}
1809
	}
1810
	if (repo->has_info_packs)
1811 1812
		fetch_indices();

N
Nick Hengeveld 已提交
1813
	/* Get a list of all local and remote heads to validate refspecs */
1814
	local_refs = get_local_heads();
N
Nick Hengeveld 已提交
1815 1816
	fprintf(stderr, "Fetching remote heads...\n");
	get_dav_remote_heads();
1817
	run_request_queue();
N
Nick Hengeveld 已提交
1818

1819 1820
	/* Remove a remote branch if -d or -D was specified */
	if (delete_branch) {
1821
		if (delete_remote_branch(refspec[0], force_delete) == -1) {
1822 1823
			fprintf(stderr, "Unable to delete remote branch %s\n",
				refspec[0]);
1824 1825 1826
			if (helper_status)
				printf("error %s cannot remove\n", refspec[0]);
		}
1827 1828 1829
		goto cleanup;
	}

N
Nick Hengeveld 已提交
1830
	/* match them up */
1831 1832
	if (match_push_refs(local_refs, &remote_refs,
			    nr_refspec, (const char **) refspec, push_all)) {
1833 1834 1835
		rc = -1;
		goto cleanup;
	}
N
Nick Hengeveld 已提交
1836 1837
	if (!remote_refs) {
		fprintf(stderr, "No refs in common and none specified; doing nothing.\n");
1838 1839
		if (helper_status)
			printf("error null no match\n");
1840 1841
		rc = 0;
		goto cleanup;
N
Nick Hengeveld 已提交
1842 1843
	}

1844
	new_refs = 0;
N
Nick Hengeveld 已提交
1845
	for (ref = remote_refs; ref; ref = ref->next) {
1846
		struct argv_array commit_argv = ARGV_ARRAY_INIT;
1847

N
Nick Hengeveld 已提交
1848 1849
		if (!ref->peer_ref)
			continue;
1850

1851
		if (is_null_oid(&ref->peer_ref->new_oid)) {
1852 1853
			if (delete_remote_branch(ref->name, 1) == -1) {
				error("Could not remove %s", ref->name);
1854 1855
				if (helper_status)
					printf("error %s cannot remove\n", ref->name);
1856 1857
				rc = -4;
			}
1858 1859
			else if (helper_status)
				printf("ok %s\n", ref->name);
1860 1861 1862 1863
			new_refs++;
			continue;
		}

1864
		if (!oidcmp(&ref->old_oid, &ref->peer_ref->new_oid)) {
1865
			if (push_verbosely)
N
Nick Hengeveld 已提交
1866
				fprintf(stderr, "'%s': up-to-date\n", ref->name);
1867 1868
			if (helper_status)
				printf("ok %s up to date\n", ref->name);
N
Nick Hengeveld 已提交
1869 1870 1871 1872
			continue;
		}

		if (!force_all &&
1873
		    !is_null_oid(&ref->old_oid) &&
N
Nick Hengeveld 已提交
1874
		    !ref->force) {
1875
			if (!has_object_file(&ref->old_oid) ||
1876 1877
			    !ref_newer(&ref->peer_ref->new_oid,
				       &ref->old_oid)) {
1878 1879
				/*
				 * We do not have the remote ref, or
N
Nick Hengeveld 已提交
1880 1881 1882 1883 1884 1885
				 * 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.
				 */
1886 1887 1888
				error("remote '%s' is not an ancestor of\n"
				      "local '%s'.\n"
				      "Maybe you are not up-to-date and "
N
Nick Hengeveld 已提交
1889 1890 1891
				      "need to pull first?",
				      ref->name,
				      ref->peer_ref->name);
1892 1893
				if (helper_status)
					printf("error %s non-fast forward\n", ref->name);
N
Nick Hengeveld 已提交
1894
				rc = -2;
N
Nick Hengeveld 已提交
1895 1896
				continue;
			}
1897
		}
1898
		oidcpy(&ref->new_oid, &ref->peer_ref->new_oid);
N
Nick Hengeveld 已提交
1899 1900 1901 1902 1903
		new_refs++;

		fprintf(stderr, "updating '%s'", ref->name);
		if (strcmp(ref->name, ref->peer_ref->name))
			fprintf(stderr, " using '%s'", ref->peer_ref->name);
1904
		fprintf(stderr, "\n  from %s\n  to   %s\n",
1905
			oid_to_hex(&ref->old_oid), oid_to_hex(&ref->new_oid));
1906 1907 1908
		if (dry_run) {
			if (helper_status)
				printf("ok %s\n", ref->name);
1909
			continue;
1910
		}
1911 1912

		/* Lock remote branch ref */
N
Nick Hengeveld 已提交
1913 1914
		ref_lock = lock_remote(ref->name, LOCK_TIME);
		if (ref_lock == NULL) {
1915
			fprintf(stderr, "Unable to lock remote branch %s\n",
N
Nick Hengeveld 已提交
1916
				ref->name);
1917 1918
			if (helper_status)
				printf("error %s lock error\n", ref->name);
1919 1920 1921 1922
			rc = 1;
			continue;
		}

N
Nick Hengeveld 已提交
1923
		/* Set up revision info for this refspec */
1924 1925
		argv_array_push(&commit_argv, ""); /* ignored */
		argv_array_push(&commit_argv, "--objects");
1926 1927
		argv_array_push(&commit_argv, oid_to_hex(&ref->new_oid));
		if (!push_all && !is_null_oid(&ref->old_oid))
1928
			argv_array_pushf(&commit_argv, "^%s",
1929
					 oid_to_hex(&ref->old_oid));
1930
		init_revisions(&revs, setup_git_directory());
1931
		setup_revisions(commit_argv.argc, commit_argv.argv, &revs, NULL);
1932
		revs.edge_hint = 0; /* just in case */
1933

N
Nick Hengeveld 已提交
1934
		/* Generate a list of objects that need to be pushed */
1935
		pushing = 0;
1936 1937
		if (prepare_revision_walk(&revs))
			die("revision walk setup failed");
1938
		mark_edges_uninteresting(&revs, NULL);
N
Nick Hengeveld 已提交
1939
		objects_to_send = get_delta(&revs, ref_lock);
1940
		finish_all_active_slots();
1941 1942 1943 1944

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

		run_request_queue();
1950 1951

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

N
Nick Hengeveld 已提交
1955 1956
		if (!rc)
			fprintf(stderr, "    done\n");
1957 1958
		if (helper_status)
			printf("%s %s\n", !rc ? "ok" : "error", ref->name);
N
Nick Hengeveld 已提交
1959
		unlock_remote(ref_lock);
1960
		check_locks();
1961
		argv_array_clear(&commit_argv);
1962 1963
	}

1964
	/* Update remote server info if appropriate */
1965 1966
	if (repo->has_info_refs && new_refs) {
		if (info_ref_lock && repo->can_update_info_refs) {
1967
			fprintf(stderr, "Updating remote server info\n");
1968 1969
			if (!dry_run)
				update_remote_info_refs(info_ref_lock);
1970 1971 1972 1973 1974
		} else {
			fprintf(stderr, "Unable to update server info\n");
		}
	}

1975
 cleanup:
1976 1977
	if (info_ref_lock)
		unlock_remote(info_ref_lock);
1978
	free(repo);
1979

1980
	http_cleanup();
1981 1982 1983 1984 1985 1986 1987 1988 1989 1990

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

	return rc;
}