blob: 7124c4b49cfba7985c5ba2046296f006de04e3bb [file] [log] [blame]
Stephen Boydc2e86ad2011-03-22 00:51:05 -07001#include "builtin.h"
Linus Torvaldsfb9040c2005-07-04 15:29:17 -07002#include "refs.h"
Linus Torvaldsdef88e92005-07-04 13:26:53 -07003#include "pkt-line.h"
Junio C Hamano49bb8052005-10-19 14:27:02 -07004#include "commit.h"
5#include "tag.h"
Nicolas Pitreda093d32006-11-01 17:06:23 -05006#include "exec_cmd.h"
Junio C Hamano9e10fd12007-01-22 22:37:33 -08007#include "pack.h"
Nicolas Pitreda093d32006-11-01 17:06:23 -05008#include "sideband.h"
Daniel Barkalow2d4177c2007-09-10 23:03:00 -04009#include "fetch-pack.h"
Daniel Barkalowba227852008-02-04 13:26:23 -050010#include "remote.h"
Johannes Sixt477822c2007-10-19 21:47:57 +020011#include "run-command.h"
Junio C Hamanoe52d7192011-03-11 11:53:52 -080012#include "transport.h"
Linus Torvaldsdef88e92005-07-04 13:26:53 -070013
Junio C Hamanoe28714c2007-01-24 17:02:15 -080014static int transfer_unpack_limit = -1;
15static int fetch_unpack_limit = -1;
Junio C Hamanoaf7cf262007-01-24 16:47:24 -080016static int unpack_limit = 100;
Nicolas Pitref04833e2009-05-01 20:18:02 -040017static int prefer_ofs_delta = 1;
Junio C Hamanoca0c9762011-10-05 12:36:20 -070018static int no_done;
Junio C Hamanodab76d32011-09-04 12:37:45 -070019static int fetch_fsck_objects = -1;
20static int transfer_fsck_objects = -1;
Daniel Barkalow09149c72007-10-29 22:35:08 -040021static struct fetch_pack_args args = {
22 /* .uploadpack = */ "git-upload-pack",
23};
Shawn O. Pearcefa740522007-09-19 00:49:35 -040024
Junio C Hamano33b83032005-08-12 02:08:29 -070025static const char fetch_pack_usage[] =
Stephan Beyer1b1dd232008-07-13 15:36:15 +020026"git fetch-pack [--all] [--quiet|-q] [--keep|-k] [--thin] [--include-tag] [--upload-pack=<git-upload-pack>] [--depth=<n>] [--no-progress] [-v] [<host>:]<directory> [<refs>...]";
Linus Torvaldsdef88e92005-07-04 13:26:53 -070027
Johannes Schindelin0a8944d2005-10-19 16:14:34 -070028#define COMPLETE (1U << 0)
Johannes Schindelin23d61f82005-10-28 04:46:27 +020029#define COMMON (1U << 1)
30#define COMMON_REF (1U << 2)
31#define SEEN (1U << 3)
32#define POPPED (1U << 4)
33
Daniel Barkalow420e9af2008-03-17 22:15:02 -040034static int marked;
35
Junio C Hamanof061e5f2006-05-24 21:48:34 -070036/*
37 * After sending this many "have"s if we do not get any new ACK , we
38 * give up traversing our history.
39 */
40#define MAX_IN_VAIN 256
41
David Rientjes96f1e582006-08-15 10:23:48 -070042static struct commit_list *rev_list;
Nicolas Pitre3891f392007-11-07 17:20:22 -050043static int non_common_revs, multi_ack, use_sideband;
Johannes Schindelin23d61f82005-10-28 04:46:27 +020044
45static void rev_list_push(struct commit *commit, int mark)
46{
47 if (!(commit->object.flags & mark)) {
48 commit->object.flags |= mark;
49
50 if (!(commit->object.parsed))
Martin Koeglerf3ec5492008-03-03 07:31:23 +010051 if (parse_commit(commit))
52 return;
Johannes Schindelin23d61f82005-10-28 04:46:27 +020053
Thiago Farina47e44ed2010-11-26 23:58:14 -020054 commit_list_insert_by_date(commit, &rev_list);
Johannes Schindelin23d61f82005-10-28 04:46:27 +020055
56 if (!(commit->object.flags & COMMON))
57 non_common_revs++;
58 }
59}
60
Michael Haggerty65385ef2012-02-11 07:20:57 +010061static int rev_list_insert_ref(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
Johannes Schindelin23d61f82005-10-28 04:46:27 +020062{
Michael Haggerty65385ef2012-02-11 07:20:57 +010063 struct object *o = deref_tag(parse_object(sha1), refname, 0);
Johannes Schindelin23d61f82005-10-28 04:46:27 +020064
Linus Torvalds19746322006-07-11 20:45:31 -070065 if (o && o->type == OBJ_COMMIT)
Johannes Schindelin23d61f82005-10-28 04:46:27 +020066 rev_list_push((struct commit *)o, SEEN);
67
68 return 0;
69}
70
Michael Haggerty65385ef2012-02-11 07:20:57 +010071static int clear_marks(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
Daniel Barkalow420e9af2008-03-17 22:15:02 -040072{
Michael Haggerty65385ef2012-02-11 07:20:57 +010073 struct object *o = deref_tag(parse_object(sha1), refname, 0);
Daniel Barkalow420e9af2008-03-17 22:15:02 -040074
75 if (o && o->type == OBJ_COMMIT)
76 clear_commit_marks((struct commit *)o,
77 COMMON | COMMON_REF | SEEN | POPPED);
78 return 0;
79}
80
Johannes Schindelin23d61f82005-10-28 04:46:27 +020081/*
82 This function marks a rev and its ancestors as common.
83 In some cases, it is desirable to mark only the ancestors (for example
84 when only the server does not yet know that they are common).
85*/
86
87static void mark_common(struct commit *commit,
88 int ancestors_only, int dont_parse)
89{
90 if (commit != NULL && !(commit->object.flags & COMMON)) {
91 struct object *o = (struct object *)commit;
92
93 if (!ancestors_only)
94 o->flags |= COMMON;
95
96 if (!(o->flags & SEEN))
97 rev_list_push(commit, SEEN);
98 else {
99 struct commit_list *parents;
100
101 if (!ancestors_only && !(o->flags & POPPED))
102 non_common_revs--;
103 if (!o->parsed && !dont_parse)
Martin Koeglerf3ec5492008-03-03 07:31:23 +0100104 if (parse_commit(commit))
105 return;
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200106
107 for (parents = commit->parents;
108 parents;
109 parents = parents->next)
110 mark_common(parents->item, 0, dont_parse);
111 }
112 }
113}
114
115/*
116 Get the next rev to send, ignoring the common.
117*/
118
Felipe Contreras4b25d092009-05-01 12:06:36 +0300119static const unsigned char *get_rev(void)
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200120{
121 struct commit *commit = NULL;
122
123 while (commit == NULL) {
124 unsigned int mark;
Linus Torvalds72269ad2008-04-28 16:27:49 -0700125 struct commit_list *parents;
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200126
127 if (rev_list == NULL || non_common_revs == 0)
128 return NULL;
129
130 commit = rev_list->item;
Junio C Hamano2d8bed92008-04-30 11:42:05 -0700131 if (!commit->object.parsed)
Linus Torvalds72269ad2008-04-28 16:27:49 -0700132 parse_commit(commit);
133 parents = commit->parents;
Martin Koeglerf3ec5492008-03-03 07:31:23 +0100134
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200135 commit->object.flags |= POPPED;
136 if (!(commit->object.flags & COMMON))
137 non_common_revs--;
Junio C Hamanoa6080a02007-06-07 00:04:01 -0700138
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200139 if (commit->object.flags & COMMON) {
140 /* do not send "have", and ignore ancestors */
141 commit = NULL;
142 mark = COMMON | SEEN;
143 } else if (commit->object.flags & COMMON_REF)
144 /* send "have", and ignore ancestors */
145 mark = COMMON | SEEN;
146 else
147 /* send "have", also for its ancestors */
148 mark = SEEN;
149
150 while (parents) {
151 if (!(parents->item->object.flags & SEEN))
152 rev_list_push(parents->item, mark);
153 if (mark & COMMON)
154 mark_common(parents->item, 1, 0);
155 parents = parents->next;
156 }
157
158 rev_list = rev_list->next;
159 }
160
161 return commit->object.sha1;
162}
Johannes Schindelin0a8944d2005-10-19 16:14:34 -0700163
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700164enum ack_type {
165 NAK = 0,
166 ACK,
167 ACK_continue,
168 ACK_common,
169 ACK_ready
170};
171
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700172static void consume_shallow_list(int fd)
173{
174 if (args.stateless_rpc && args.depth > 0) {
175 /* If we sent a depth we will get back "duplicate"
176 * shallow and unshallow commands every time there
177 * is a block of have lines exchanged.
178 */
179 char line[1000];
180 while (packet_read_line(fd, line, sizeof(line))) {
181 if (!prefixcmp(line, "shallow "))
182 continue;
183 if (!prefixcmp(line, "unshallow "))
184 continue;
185 die("git fetch-pack: expected shallow list");
186 }
187 }
188}
189
Nguyễn Thái Ngọc Duy294e15f2011-08-18 19:29:36 +0700190struct write_shallow_data {
191 struct strbuf *out;
192 int use_pack_protocol;
193 int count;
194};
195
196static int write_one_shallow(const struct commit_graft *graft, void *cb_data)
197{
198 struct write_shallow_data *data = cb_data;
199 const char *hex = sha1_to_hex(graft->sha1);
200 data->count++;
201 if (data->use_pack_protocol)
202 packet_buf_write(data->out, "shallow %s", hex);
203 else {
204 strbuf_addstr(data->out, hex);
205 strbuf_addch(data->out, '\n');
206 }
207 return 0;
208}
209
210static int write_shallow_commits(struct strbuf *out, int use_pack_protocol)
211{
212 struct write_shallow_data data;
213 data.out = out;
214 data.use_pack_protocol = use_pack_protocol;
215 data.count = 0;
216 for_each_commit_graft(write_one_shallow, &data);
217 return data.count;
218}
219
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700220static enum ack_type get_ack(int fd, unsigned char *result_sha1)
Shawn O. Pearce28754ab2009-10-30 17:47:24 -0700221{
222 static char line[1000];
223 int len = packet_read_line(fd, line, sizeof(line));
224
225 if (!len)
226 die("git fetch-pack: expected ACK/NAK, got EOF");
227 if (line[len-1] == '\n')
228 line[--len] = 0;
229 if (!strcmp(line, "NAK"))
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700230 return NAK;
Shawn O. Pearce28754ab2009-10-30 17:47:24 -0700231 if (!prefixcmp(line, "ACK ")) {
232 if (!get_sha1_hex(line+4, result_sha1)) {
233 if (strstr(line+45, "continue"))
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700234 return ACK_continue;
235 if (strstr(line+45, "common"))
236 return ACK_common;
237 if (strstr(line+45, "ready"))
238 return ACK_ready;
239 return ACK;
Shawn O. Pearce28754ab2009-10-30 17:47:24 -0700240 }
241 }
242 die("git fetch_pack: expected ACK/NAK, got '%s'", line);
243}
244
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700245static void send_request(int fd, struct strbuf *buf)
246{
247 if (args.stateless_rpc) {
248 send_sideband(fd, -1, buf->buf, buf->len, LARGE_PACKET_MAX);
249 packet_flush(fd);
250 } else
251 safe_write(fd, buf->buf, buf->len);
252}
253
Junio C Hamanoe52d7192011-03-11 11:53:52 -0800254static void insert_one_alternate_ref(const struct ref *ref, void *unused)
255{
256 rev_list_insert_ref(NULL, ref->old_sha1, 0, NULL);
257}
258
Junio C Hamano066bf4c2011-03-20 21:52:45 -0700259#define INITIAL_FLUSH 16
Junio C Hamano44d8dc52011-03-29 10:06:19 -0700260#define PIPESAFE_FLUSH 32
Junio C Hamano6afca452011-03-20 21:52:44 -0700261#define LARGE_FLUSH 1024
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700262
263static int next_flush(int count)
264{
Junio C Hamano44d8dc52011-03-29 10:06:19 -0700265 int flush_limit = args.stateless_rpc ? LARGE_FLUSH : PIPESAFE_FLUSH;
266
267 if (count < flush_limit)
Junio C Hamano6afca452011-03-20 21:52:44 -0700268 count <<= 1;
269 else
Junio C Hamano44d8dc52011-03-29 10:06:19 -0700270 count += flush_limit;
Junio C Hamano6afca452011-03-20 21:52:44 -0700271 return count;
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700272}
273
Junio C Hamano33b83032005-08-12 02:08:29 -0700274static int find_common(int fd[2], unsigned char *result_sha1,
275 struct ref *refs)
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700276{
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700277 int fetching;
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700278 int count = 0, flushes = 0, flush_at = INITIAL_FLUSH, retval;
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200279 const unsigned char *sha1;
Junio C Hamanof061e5f2006-05-24 21:48:34 -0700280 unsigned in_vain = 0;
281 int got_continue = 0;
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700282 int got_ready = 0;
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700283 struct strbuf req_buf = STRBUF_INIT;
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700284 size_t state_len = 0;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700285
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700286 if (args.stateless_rpc && multi_ack == 1)
287 die("--stateless-rpc requires multi_ack_detailed");
Daniel Barkalow420e9af2008-03-17 22:15:02 -0400288 if (marked)
289 for_each_ref(clear_marks, NULL);
290 marked = 1;
291
Junio C Hamanocb5d7092006-09-20 21:47:42 -0700292 for_each_ref(rev_list_insert_ref, NULL);
Michael Haggertyc41a8022012-02-11 07:20:58 +0100293 for_each_alternate_ref(insert_one_alternate_ref, NULL);
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200294
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700295 fetching = 0;
296 for ( ; refs ; refs = refs->next) {
Junio C Hamano33b83032005-08-12 02:08:29 -0700297 unsigned char *remote = refs->old_sha1;
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700298 const char *remote_hex;
Junio C Hamano4dab94d2005-10-19 18:28:17 -0700299 struct object *o;
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700300
Johannes Schindelin0a8944d2005-10-19 16:14:34 -0700301 /*
Junio C Hamano4dab94d2005-10-19 18:28:17 -0700302 * If that object is complete (i.e. it is an ancestor of a
303 * local ref), we tell them we have it but do not have to
304 * tell them about its ancestors, which they already know
305 * about.
Junio C Hamanof1f0a2b2005-10-19 21:55:49 -0700306 *
307 * We use lookup_object here because we are only
308 * interested in the case we *know* the object is
309 * reachable and we have already scanned it.
Junio C Hamano4dab94d2005-10-19 18:28:17 -0700310 */
Junio C Hamanof1f0a2b2005-10-19 21:55:49 -0700311 if (((o = lookup_object(remote)) != NULL) &&
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200312 (o->flags & COMPLETE)) {
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700313 continue;
Johannes Schindelin0a8944d2005-10-19 16:14:34 -0700314 }
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200315
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700316 remote_hex = sha1_to_hex(remote);
317 if (!fetching) {
318 struct strbuf c = STRBUF_INIT;
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700319 if (multi_ack == 2) strbuf_addstr(&c, " multi_ack_detailed");
320 if (multi_ack == 1) strbuf_addstr(&c, " multi_ack");
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700321 if (no_done) strbuf_addstr(&c, " no-done");
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700322 if (use_sideband == 2) strbuf_addstr(&c, " side-band-64k");
323 if (use_sideband == 1) strbuf_addstr(&c, " side-band");
324 if (args.use_thin_pack) strbuf_addstr(&c, " thin-pack");
325 if (args.no_progress) strbuf_addstr(&c, " no-progress");
326 if (args.include_tag) strbuf_addstr(&c, " include-tag");
327 if (prefer_ofs_delta) strbuf_addstr(&c, " ofs-delta");
328 packet_buf_write(&req_buf, "want %s%s\n", remote_hex, c.buf);
329 strbuf_release(&c);
330 } else
331 packet_buf_write(&req_buf, "want %s\n", remote_hex);
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700332 fetching++;
Junio C Hamano33b83032005-08-12 02:08:29 -0700333 }
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700334
335 if (!fetching) {
336 strbuf_release(&req_buf);
337 packet_flush(fd[1]);
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700338 return 1;
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700339 }
340
341 if (is_repository_shallow())
342 write_shallow_commits(&req_buf, 1);
343 if (args.depth > 0)
344 packet_buf_write(&req_buf, "deepen %d", args.depth);
345 packet_buf_flush(&req_buf);
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700346 state_len = req_buf.len;
Johannes Schindelin0a8944d2005-10-19 16:14:34 -0700347
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400348 if (args.depth > 0) {
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100349 char line[1024];
350 unsigned char sha1[20];
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100351
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700352 send_request(fd[1], &req_buf);
Benjamin Kramereb3a9dd2009-03-07 21:02:10 +0100353 while (packet_read_line(fd[0], line, sizeof(line))) {
Junio C Hamano599065a2007-02-20 01:54:00 -0800354 if (!prefixcmp(line, "shallow ")) {
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100355 if (get_sha1_hex(line + 8, sha1))
356 die("invalid shallow line: %s", line);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100357 register_shallow(sha1);
Junio C Hamanocf01bd52006-11-13 22:04:56 -0800358 continue;
359 }
Junio C Hamano599065a2007-02-20 01:54:00 -0800360 if (!prefixcmp(line, "unshallow ")) {
Johannes Schindelinf53514b2006-10-30 20:09:53 +0100361 if (get_sha1_hex(line + 10, sha1))
362 die("invalid unshallow line: %s", line);
363 if (!lookup_object(sha1))
364 die("object not found: %s", line);
365 /* make sure that it is parsed as shallow */
Martin Koeglerf3ec5492008-03-03 07:31:23 +0100366 if (!parse_object(sha1))
367 die("error in object: %s", line);
Johannes Schindelinf53514b2006-10-30 20:09:53 +0100368 if (unregister_shallow(sha1))
369 die("no shallow found: %s", line);
Junio C Hamanocf01bd52006-11-13 22:04:56 -0800370 continue;
371 }
372 die("expected shallow/unshallow, got %s", line);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100373 }
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700374 } else if (!args.stateless_rpc)
375 send_request(fd[1], &req_buf);
376
377 if (!args.stateless_rpc) {
378 /* If we aren't using the stateless-rpc interface
379 * we don't need to retain the headers.
380 */
381 strbuf_setlen(&req_buf, 0);
382 state_len = 0;
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100383 }
384
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200385 flushes = 0;
Linus Torvalds75bfc6c2005-07-04 16:35:13 -0700386 retval = -1;
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200387 while ((sha1 = get_rev())) {
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700388 packet_buf_write(&req_buf, "have %s\n", sha1_to_hex(sha1));
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400389 if (args.verbose)
Junio C Hamano33b83032005-08-12 02:08:29 -0700390 fprintf(stderr, "have %s\n", sha1_to_hex(sha1));
Junio C Hamanof061e5f2006-05-24 21:48:34 -0700391 in_vain++;
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700392 if (flush_at <= ++count) {
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200393 int ack;
394
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700395 packet_buf_flush(&req_buf);
396 send_request(fd[1], &req_buf);
397 strbuf_setlen(&req_buf, state_len);
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700398 flushes++;
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700399 flush_at = next_flush(count);
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700400
401 /*
402 * We keep one window "ahead" of the other side, and
403 * will wait for an ACK only on the next one
404 */
Junio C Hamanoc12f5912011-03-20 21:52:40 -0700405 if (!args.stateless_rpc && count == INITIAL_FLUSH)
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700406 continue;
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200407
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700408 consume_shallow_list(fd[0]);
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200409 do {
410 ack = get_ack(fd[0], result_sha1);
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400411 if (args.verbose && ack)
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200412 fprintf(stderr, "got ack %d %s\n", ack,
413 sha1_to_hex(result_sha1));
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700414 switch (ack) {
415 case ACK:
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200416 flushes = 0;
417 multi_ack = 0;
418 retval = 0;
419 goto done;
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700420 case ACK_common:
421 case ACK_ready:
422 case ACK_continue: {
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200423 struct commit *commit =
424 lookup_commit(result_sha1);
Nguyễn Thái Ngọc Duyec099542011-08-18 20:36:03 +0700425 if (!commit)
426 die("invalid commit %s", sha1_to_hex(result_sha1));
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700427 if (args.stateless_rpc
428 && ack == ACK_common
429 && !(commit->object.flags & COMMON)) {
430 /* We need to replay the have for this object
431 * on the next RPC request so the peer knows
432 * it is in common with us.
433 */
434 const char *hex = sha1_to_hex(result_sha1);
435 packet_buf_write(&req_buf, "have %s\n", hex);
436 state_len = req_buf.len;
437 }
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200438 mark_common(commit, 0, 1);
439 retval = 0;
Junio C Hamanof061e5f2006-05-24 21:48:34 -0700440 in_vain = 0;
441 got_continue = 1;
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700442 if (ack == ACK_ready) {
Shawn O. Pearcef2cba922011-03-14 16:48:38 -0700443 rev_list = NULL;
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700444 got_ready = 1;
445 }
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700446 break;
447 }
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200448 }
449 } while (ack);
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700450 flushes--;
Junio C Hamanof061e5f2006-05-24 21:48:34 -0700451 if (got_continue && MAX_IN_VAIN < in_vain) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400452 if (args.verbose)
Junio C Hamanof061e5f2006-05-24 21:48:34 -0700453 fprintf(stderr, "giving up\n");
454 break; /* give up */
455 }
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700456 }
457 }
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200458done:
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700459 if (!got_ready || !no_done) {
460 packet_buf_write(&req_buf, "done\n");
461 send_request(fd[1], &req_buf);
462 }
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400463 if (args.verbose)
Junio C Hamano33b83032005-08-12 02:08:29 -0700464 fprintf(stderr, "done\n");
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200465 if (retval != 0) {
466 multi_ack = 0;
Johannes Schindelin23d61f82005-10-28 04:46:27 +0200467 flushes++;
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200468 }
Shawn O. Pearceedace6f2009-10-30 17:47:23 -0700469 strbuf_release(&req_buf);
470
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700471 consume_shallow_list(fd[0]);
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200472 while (flushes || multi_ack) {
473 int ack = get_ack(fd[0], result_sha1);
474 if (ack) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400475 if (args.verbose)
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200476 fprintf(stderr, "got ack (%d) %s\n", ack,
477 sha1_to_hex(result_sha1));
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700478 if (ack == ACK)
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200479 return 0;
480 multi_ack = 1;
481 continue;
Junio C Hamano33b83032005-08-12 02:08:29 -0700482 }
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200483 flushes--;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700484 }
Johannes Schindelin8cb560f2008-07-02 18:06:56 +0100485 /* it is no error to fetch into a completely empty repo */
486 return count ? retval : 0;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700487}
488
David Rientjes96f1e582006-08-15 10:23:48 -0700489static struct commit_list *complete;
Junio C Hamano49bb8052005-10-19 14:27:02 -0700490
Michael Haggerty65385ef2012-02-11 07:20:57 +0100491static int mark_complete(const char *refname, const unsigned char *sha1, int flag, void *cb_data)
Junio C Hamano49bb8052005-10-19 14:27:02 -0700492{
493 struct object *o = parse_object(sha1);
494
Linus Torvalds19746322006-07-11 20:45:31 -0700495 while (o && o->type == OBJ_TAG) {
Junio C Hamanof1f0a2b2005-10-19 21:55:49 -0700496 struct tag *t = (struct tag *) o;
497 if (!t->tagged)
498 break; /* broken repository */
Junio C Hamano49bb8052005-10-19 14:27:02 -0700499 o->flags |= COMPLETE;
Junio C Hamanof1f0a2b2005-10-19 21:55:49 -0700500 o = parse_object(t->tagged->sha1);
Junio C Hamano49bb8052005-10-19 14:27:02 -0700501 }
Linus Torvalds19746322006-07-11 20:45:31 -0700502 if (o && o->type == OBJ_COMMIT) {
Junio C Hamano49bb8052005-10-19 14:27:02 -0700503 struct commit *commit = (struct commit *)o;
Jeff Kingea5f2202011-05-19 16:48:51 -0400504 if (!(commit->object.flags & COMPLETE)) {
505 commit->object.flags |= COMPLETE;
506 commit_list_insert_by_date(commit, &complete);
507 }
Junio C Hamano49bb8052005-10-19 14:27:02 -0700508 }
509 return 0;
510}
511
512static void mark_recent_complete_commits(unsigned long cutoff)
513{
514 while (complete && cutoff <= complete->item->date) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400515 if (args.verbose)
Junio C Hamano49bb8052005-10-19 14:27:02 -0700516 fprintf(stderr, "Marking %s as complete\n",
517 sha1_to_hex(complete->item->object.sha1));
518 pop_most_recent_commit(&complete, COMPLETE);
519 }
520}
521
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200522static void filter_refs(struct ref **refs, int nr_match, char **match)
523{
Junio C Hamano95460102006-05-11 15:28:44 -0700524 struct ref **return_refs;
525 struct ref *newlist = NULL;
526 struct ref **newtail = &newlist;
527 struct ref *ref, *next;
528 struct ref *fastarray[32];
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200529
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400530 if (nr_match && !args.fetch_all) {
Junio C Hamano95460102006-05-11 15:28:44 -0700531 if (ARRAY_SIZE(fastarray) < nr_match)
532 return_refs = xcalloc(nr_match, sizeof(struct ref *));
533 else {
534 return_refs = fastarray;
535 memset(return_refs, 0, sizeof(struct ref *) * nr_match);
536 }
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200537 }
Junio C Hamano95460102006-05-11 15:28:44 -0700538 else
539 return_refs = NULL;
540
541 for (ref = *refs; ref; ref = next) {
542 next = ref->next;
543 if (!memcmp(ref->name, "refs/", 5) &&
Michael Haggerty8d9c5012011-09-15 23:10:25 +0200544 check_refname_format(ref->name + 5, 0))
Junio C Hamano95460102006-05-11 15:28:44 -0700545 ; /* trash */
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400546 else if (args.fetch_all &&
547 (!args.depth || prefixcmp(ref->name, "refs/tags/") )) {
Junio C Hamano95460102006-05-11 15:28:44 -0700548 *newtail = ref;
549 ref->next = NULL;
550 newtail = &ref->next;
551 continue;
552 }
553 else {
Jeff King1e7ba0f2011-12-12 19:48:08 -0500554 int i;
555 for (i = 0; i < nr_match; i++) {
556 if (!strcmp(ref->name, match[i])) {
557 match[i][0] = '\0';
558 return_refs[i] = ref;
559 break;
560 }
Junio C Hamano95460102006-05-11 15:28:44 -0700561 }
Jeff King1e7ba0f2011-12-12 19:48:08 -0500562 if (i < nr_match)
563 continue; /* we will link it later */
Junio C Hamano95460102006-05-11 15:28:44 -0700564 }
565 free(ref);
566 }
567
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400568 if (!args.fetch_all) {
Junio C Hamano95460102006-05-11 15:28:44 -0700569 int i;
570 for (i = 0; i < nr_match; i++) {
571 ref = return_refs[i];
572 if (ref) {
573 *newtail = ref;
574 ref->next = NULL;
575 newtail = &ref->next;
576 }
577 }
578 if (return_refs != fastarray)
579 free(return_refs);
580 }
581 *refs = newlist;
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200582}
583
Michael Haggertyf2576592012-02-11 07:20:59 +0100584static void mark_alternate_complete(const struct ref *ref, void *unused)
585{
586 mark_complete(NULL, ref->old_sha1, 0, NULL);
587}
588
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200589static int everything_local(struct ref **refs, int nr_match, char **match)
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700590{
Junio C Hamano49bb8052005-10-19 14:27:02 -0700591 struct ref *ref;
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700592 int retval;
Junio C Hamano49bb8052005-10-19 14:27:02 -0700593 unsigned long cutoff = 0;
594
Junio C Hamano49bb8052005-10-19 14:27:02 -0700595 save_commit_buffer = 0;
596
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200597 for (ref = *refs; ref; ref = ref->next) {
Junio C Hamano49bb8052005-10-19 14:27:02 -0700598 struct object *o;
599
600 o = parse_object(ref->old_sha1);
601 if (!o)
602 continue;
603
604 /* We already have it -- which may mean that we were
605 * in sync with the other side at some time after
606 * that (it is OK if we guess wrong here).
607 */
Linus Torvalds19746322006-07-11 20:45:31 -0700608 if (o->type == OBJ_COMMIT) {
Junio C Hamano49bb8052005-10-19 14:27:02 -0700609 struct commit *commit = (struct commit *)o;
610 if (!cutoff || cutoff < commit->date)
611 cutoff = commit->date;
612 }
613 }
614
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400615 if (!args.depth) {
Johannes Schindelinf53514b2006-10-30 20:09:53 +0100616 for_each_ref(mark_complete, NULL);
Michael Haggertyf2576592012-02-11 07:20:59 +0100617 for_each_alternate_ref(mark_alternate_complete, NULL);
Johannes Schindelinf53514b2006-10-30 20:09:53 +0100618 if (cutoff)
619 mark_recent_complete_commits(cutoff);
620 }
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700621
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200622 /*
623 * Mark all complete remote refs as common refs.
624 * Don't mark them common yet; the server has to be told so first.
625 */
626 for (ref = *refs; ref; ref = ref->next) {
Junio C Hamano9534f402005-11-02 15:19:13 -0800627 struct object *o = deref_tag(lookup_object(ref->old_sha1),
628 NULL, 0);
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200629
Linus Torvalds19746322006-07-11 20:45:31 -0700630 if (!o || o->type != OBJ_COMMIT || !(o->flags & COMPLETE))
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200631 continue;
632
633 if (!(o->flags & SEEN)) {
634 rev_list_push((struct commit *)o, COMMON_REF | SEEN);
635
636 mark_common((struct commit *)o, 1, 1);
637 }
638 }
639
640 filter_refs(refs, nr_match, match);
641
642 for (retval = 1, ref = *refs; ref ; ref = ref->next) {
643 const unsigned char *remote = ref->old_sha1;
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700644 unsigned char local[20];
Junio C Hamano49bb8052005-10-19 14:27:02 -0700645 struct object *o;
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700646
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200647 o = lookup_object(remote);
Junio C Hamano49bb8052005-10-19 14:27:02 -0700648 if (!o || !(o->flags & COMPLETE)) {
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700649 retval = 0;
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400650 if (!args.verbose)
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700651 continue;
652 fprintf(stderr,
653 "want %s (%s)\n", sha1_to_hex(remote),
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200654 ref->name);
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700655 continue;
656 }
657
Shawn Pearcee7024962006-08-23 02:49:00 -0400658 hashcpy(ref->new_sha1, local);
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400659 if (!args.verbose)
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700660 continue;
661 fprintf(stderr,
662 "already have %s (%s)\n", sha1_to_hex(remote),
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200663 ref->name);
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700664 }
665 return retval;
666}
667
Erik Faye-Lundae6a5602010-02-05 12:57:38 -0800668static int sideband_demux(int in, int out, void *data)
Nicolas Pitreda093d32006-11-01 17:06:23 -0500669{
Johannes Sixt088fab52007-10-19 21:48:01 +0200670 int *xd = data;
Nicolas Pitreda093d32006-11-01 17:06:23 -0500671
Erik Faye-Lundae6a5602010-02-05 12:57:38 -0800672 int ret = recv_sideband("fetch-pack", xd[0], out);
673 close(out);
Johannes Sixt3ef67cf2009-06-08 10:51:22 +0200674 return ret;
Johannes Sixt088fab52007-10-19 21:48:01 +0200675}
676
Shawn O. Pearce1788c392007-09-14 03:31:23 -0400677static int get_pack(int xd[2], char **pack_lockfile)
Nicolas Pitreda093d32006-11-01 17:06:23 -0500678{
Johannes Sixt088fab52007-10-19 21:48:01 +0200679 struct async demux;
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800680 const char *argv[20];
681 char keep_arg[256];
682 char hdr_arg[256];
683 const char **av;
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400684 int do_keep = args.keep_pack;
Johannes Sixt477822c2007-10-19 21:47:57 +0200685 struct child_process cmd;
Nicolas Pitreda093d32006-11-01 17:06:23 -0500686
Johannes Sixt1f759ee2007-11-17 23:09:28 +0100687 memset(&demux, 0, sizeof(demux));
688 if (use_sideband) {
689 /* xd[] is talking with upload-pack; subprocess reads from
690 * xd[0], spits out band#2 to stderr, and feeds us band#1
691 * through demux->out.
692 */
693 demux.proc = sideband_demux;
694 demux.data = xd;
Erik Faye-Lundae6a5602010-02-05 12:57:38 -0800695 demux.out = -1;
Johannes Sixt1f759ee2007-11-17 23:09:28 +0100696 if (start_async(&demux))
697 die("fetch-pack: unable to fork off sideband"
698 " demultiplexer");
699 }
700 else
701 demux.out = xd[0];
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800702
Johannes Sixt477822c2007-10-19 21:47:57 +0200703 memset(&cmd, 0, sizeof(cmd));
704 cmd.argv = argv;
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800705 av = argv;
706 *hdr_arg = 0;
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400707 if (!args.keep_pack && unpack_limit) {
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800708 struct pack_header header;
709
Johannes Sixt1f759ee2007-11-17 23:09:28 +0100710 if (read_pack_header(demux.out, &header))
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800711 die("protocol error: bad pack header");
Ramsay Jones6e1c2342008-07-03 16:52:09 +0100712 snprintf(hdr_arg, sizeof(hdr_arg),
713 "--pack_header=%"PRIu32",%"PRIu32,
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800714 ntohl(header.hdr_version), ntohl(header.hdr_entries));
Junio C Hamanoaf7cf262007-01-24 16:47:24 -0800715 if (ntohl(header.hdr_entries) < unpack_limit)
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800716 do_keep = 0;
717 else
718 do_keep = 1;
719 }
720
721 if (do_keep) {
Johannes Sixt477822c2007-10-19 21:47:57 +0200722 if (pack_lockfile)
723 cmd.out = -1;
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800724 *av++ = "index-pack";
725 *av++ = "--stdin";
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400726 if (!args.quiet && !args.no_progress)
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800727 *av++ = "-v";
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400728 if (args.use_thin_pack)
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800729 *av++ = "--fix-thin";
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400730 if (args.lock_pack || unpack_limit) {
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800731 int s = sprintf(keep_arg,
David Soria Parra85e72832008-08-31 14:09:39 +0200732 "--keep=fetch-pack %"PRIuMAX " on ", (uintmax_t) getpid());
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800733 if (gethostname(keep_arg + s, sizeof(keep_arg) - s))
734 strcpy(keep_arg + s, "localhost");
735 *av++ = keep_arg;
736 }
737 }
738 else {
739 *av++ = "unpack-objects";
Clemens Buchacher01fdc212012-02-13 21:17:15 +0100740 if (args.quiet || args.no_progress)
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800741 *av++ = "-q";
742 }
743 if (*hdr_arg)
744 *av++ = hdr_arg;
Junio C Hamanodab76d32011-09-04 12:37:45 -0700745 if (fetch_fsck_objects >= 0
746 ? fetch_fsck_objects
747 : transfer_fsck_objects >= 0
748 ? transfer_fsck_objects
749 : 0)
Junio C Hamano5e838ea2011-09-04 12:26:14 -0700750 *av++ = "--strict";
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800751 *av++ = NULL;
752
Johannes Sixt1f759ee2007-11-17 23:09:28 +0100753 cmd.in = demux.out;
Johannes Sixt477822c2007-10-19 21:47:57 +0200754 cmd.git_cmd = 1;
755 if (start_command(&cmd))
Nicolas Pitreda093d32006-11-01 17:06:23 -0500756 die("fetch-pack: unable to fork off %s", argv[0]);
Johannes Sixte72ae282008-02-16 18:36:38 +0100757 if (do_keep && pack_lockfile) {
Johannes Sixt477822c2007-10-19 21:47:57 +0200758 *pack_lockfile = index_pack_lockfile(cmd.out);
Johannes Sixte72ae282008-02-16 18:36:38 +0100759 close(cmd.out);
760 }
Johannes Sixt477822c2007-10-19 21:47:57 +0200761
762 if (finish_command(&cmd))
763 die("%s failed", argv[0]);
Johannes Sixt088fab52007-10-19 21:48:01 +0200764 if (use_sideband && finish_async(&demux))
765 die("error in sideband demultiplexer");
Johannes Sixt477822c2007-10-19 21:47:57 +0200766 return 0;
Nicolas Pitreda093d32006-11-01 17:06:23 -0500767}
768
Shawn O. Pearce1788c392007-09-14 03:31:23 -0400769static struct ref *do_fetch_pack(int fd[2],
Daniel Barkalowba227852008-02-04 13:26:23 -0500770 const struct ref *orig_ref,
Shawn O. Pearce1788c392007-09-14 03:31:23 -0400771 int nr_match,
772 char **match,
773 char **pack_lockfile)
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700774{
Daniel Barkalowba227852008-02-04 13:26:23 -0500775 struct ref *ref = copy_ref_list(orig_ref);
Linus Torvaldsd1c133f2005-07-16 13:55:50 -0700776 unsigned char sha1[20];
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700777
Johannes Schindelined09aef2006-10-30 20:09:06 +0100778 if (is_repository_shallow() && !server_supports("shallow"))
779 die("Server does not support shallow clients");
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700780 if (server_supports("multi_ack_detailed")) {
781 if (args.verbose)
782 fprintf(stderr, "Server supports multi_ack_detailed\n");
783 multi_ack = 2;
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700784 if (server_supports("no-done")) {
785 if (args.verbose)
786 fprintf(stderr, "Server supports no-done\n");
Junio C Hamano8e9182e2011-03-29 10:16:29 -0700787 if (args.stateless_rpc)
788 no_done = 1;
Junio C Hamano4e10cf92011-03-29 12:29:10 -0700789 }
Shawn O. Pearce78affc42009-10-30 17:47:25 -0700790 }
791 else if (server_supports("multi_ack")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400792 if (args.verbose)
Johannes Schindelinc4c86f02005-10-28 04:50:26 +0200793 fprintf(stderr, "Server supports multi_ack\n");
794 multi_ack = 1;
795 }
Junio C Hamanod47f3db2006-09-10 16:27:08 -0700796 if (server_supports("side-band-64k")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400797 if (args.verbose)
Junio C Hamanod47f3db2006-09-10 16:27:08 -0700798 fprintf(stderr, "Server supports side-band-64k\n");
799 use_sideband = 2;
800 }
801 else if (server_supports("side-band")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400802 if (args.verbose)
Junio C Hamano583b7ea2006-06-21 00:30:21 -0700803 fprintf(stderr, "Server supports side-band\n");
804 use_sideband = 1;
805 }
Nicolas Pitref04833e2009-05-01 20:18:02 -0400806 if (server_supports("ofs-delta")) {
807 if (args.verbose)
808 fprintf(stderr, "Server supports ofs-delta\n");
809 } else
810 prefer_ofs_delta = 0;
Johannes Schindelin1baaae52005-10-28 04:47:07 +0200811 if (everything_local(&ref, nr_match, match)) {
Linus Torvalds2759cbc2005-10-18 11:35:17 -0700812 packet_flush(fd[1]);
813 goto all_done;
814 }
Junio C Hamano33b83032005-08-12 02:08:29 -0700815 if (find_common(fd, sha1, ref) < 0)
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400816 if (!args.keep_pack)
Junio C Hamanodfeff662006-03-20 00:21:10 -0800817 /* When cloning, it is not unusual to have
818 * no common commit.
819 */
Miklos Vajna78509d22009-03-24 02:09:12 +0100820 warning("no common commits");
Junio C Hamanoad897212005-12-14 22:17:38 -0800821
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700822 if (args.stateless_rpc)
823 packet_flush(fd[1]);
Shawn O. Pearce1788c392007-09-14 03:31:23 -0400824 if (get_pack(fd, pack_lockfile))
Junio C Hamano7e44c932008-08-31 09:39:19 -0700825 die("git fetch-pack: fetch failed.");
Junio C Hamanoad897212005-12-14 22:17:38 -0800826
827 all_done:
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400828 return ref;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700829}
830
Junio C Hamano310b86d2006-11-25 01:33:06 -0800831static int remove_duplicates(int nr_heads, char **heads)
832{
833 int src, dst;
834
835 for (src = dst = 0; src < nr_heads; src++) {
836 /* If heads[src] is different from any of
837 * heads[0..dst], push it in.
838 */
839 int i;
840 for (i = 0; i < dst; i++) {
841 if (!strcmp(heads[i], heads[src]))
842 break;
843 }
844 if (i < dst)
845 continue;
846 if (src != dst)
847 heads[dst] = heads[src];
848 dst++;
849 }
Junio C Hamano310b86d2006-11-25 01:33:06 -0800850 return dst;
851}
852
Johannes Schindelinef90d6d2008-05-14 18:46:53 +0100853static int fetch_pack_config(const char *var, const char *value, void *cb)
Junio C Hamanoaf7cf262007-01-24 16:47:24 -0800854{
855 if (strcmp(var, "fetch.unpacklimit") == 0) {
Junio C Hamanoe28714c2007-01-24 17:02:15 -0800856 fetch_unpack_limit = git_config_int(var, value);
857 return 0;
858 }
859
860 if (strcmp(var, "transfer.unpacklimit") == 0) {
861 transfer_unpack_limit = git_config_int(var, value);
Junio C Hamanoaf7cf262007-01-24 16:47:24 -0800862 return 0;
863 }
864
Nicolas Pitref04833e2009-05-01 20:18:02 -0400865 if (strcmp(var, "repack.usedeltabaseoffset") == 0) {
866 prefer_ofs_delta = git_config_bool(var, value);
867 return 0;
868 }
869
Junio C Hamano5e838ea2011-09-04 12:26:14 -0700870 if (!strcmp(var, "fetch.fsckobjects")) {
871 fetch_fsck_objects = git_config_bool(var, value);
872 return 0;
873 }
874
Junio C Hamanodab76d32011-09-04 12:37:45 -0700875 if (!strcmp(var, "transfer.fsckobjects")) {
876 transfer_fsck_objects = git_config_bool(var, value);
877 return 0;
878 }
879
Johannes Schindelinef90d6d2008-05-14 18:46:53 +0100880 return git_default_config(var, value, cb);
Junio C Hamanoaf7cf262007-01-24 16:47:24 -0800881}
882
Junio C Hamano54b9e022007-01-02 11:22:08 -0800883static struct lock_file lock;
884
Shawn O. Pearce50ab5fd2007-09-19 00:49:39 -0400885static void fetch_pack_setup(void)
886{
887 static int did_setup;
888 if (did_setup)
889 return;
Johannes Schindelinef90d6d2008-05-14 18:46:53 +0100890 git_config(fetch_pack_config, NULL);
Shawn O. Pearce50ab5fd2007-09-19 00:49:39 -0400891 if (0 <= transfer_unpack_limit)
892 unpack_limit = transfer_unpack_limit;
893 else if (0 <= fetch_unpack_limit)
894 unpack_limit = fetch_unpack_limit;
895 did_setup = 1;
896}
897
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400898int cmd_fetch_pack(int argc, const char **argv, const char *prefix)
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700899{
900 int i, ret, nr_heads;
Daniel Barkalowba227852008-02-04 13:26:23 -0500901 struct ref *ref = NULL;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700902 char *dest = NULL, **heads;
Daniel Barkalowba227852008-02-04 13:26:23 -0500903 int fd[2];
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700904 char *pack_lockfile = NULL;
905 char **pack_lockfile_ptr = NULL;
Daniel Barkalowba227852008-02-04 13:26:23 -0500906 struct child_process *conn;
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700907
Jeff Kingbbc30f92011-02-24 09:30:19 -0500908 packet_trace_identity("fetch-pack");
909
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700910 nr_heads = 0;
911 heads = NULL;
912 for (i = 1; i < argc; i++) {
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400913 const char *arg = argv[i];
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700914
915 if (*arg == '-') {
Junio C Hamano599065a2007-02-20 01:54:00 -0800916 if (!prefixcmp(arg, "--upload-pack=")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400917 args.uploadpack = arg + 14;
Uwe Kleine-König27dca072007-01-23 09:20:17 +0100918 continue;
919 }
Junio C Hamano599065a2007-02-20 01:54:00 -0800920 if (!prefixcmp(arg, "--exec=")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400921 args.uploadpack = arg + 7;
Junio C Hamano8b3d9dc2005-07-14 00:08:37 -0700922 continue;
923 }
Junio C Hamano2247efb2005-12-18 01:55:29 -0800924 if (!strcmp("--quiet", arg) || !strcmp("-q", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400925 args.quiet = 1;
Junio C Hamano33b83032005-08-12 02:08:29 -0700926 continue;
927 }
Junio C Hamano2247efb2005-12-18 01:55:29 -0800928 if (!strcmp("--keep", arg) || !strcmp("-k", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400929 args.lock_pack = args.keep_pack;
930 args.keep_pack = 1;
Junio C Hamano9e10fd12007-01-22 22:37:33 -0800931 continue;
932 }
Junio C Hamanob19696c2006-02-20 00:38:39 -0800933 if (!strcmp("--thin", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400934 args.use_thin_pack = 1;
Junio C Hamanob19696c2006-02-20 00:38:39 -0800935 continue;
936 }
Shawn O. Pearce348e3902008-03-03 22:27:33 -0500937 if (!strcmp("--include-tag", arg)) {
938 args.include_tag = 1;
939 continue;
940 }
Junio C Hamanodfeff662006-03-20 00:21:10 -0800941 if (!strcmp("--all", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400942 args.fetch_all = 1;
Junio C Hamanodfeff662006-03-20 00:21:10 -0800943 continue;
944 }
Junio C Hamano33b83032005-08-12 02:08:29 -0700945 if (!strcmp("-v", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400946 args.verbose = 1;
Junio C Hamano33b83032005-08-12 02:08:29 -0700947 continue;
948 }
Junio C Hamano599065a2007-02-20 01:54:00 -0800949 if (!prefixcmp(arg, "--depth=")) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400950 args.depth = strtol(arg + 8, NULL, 0);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +0100951 continue;
952 }
Johannes Schindelin83a5ad62007-02-20 03:01:44 +0100953 if (!strcmp("--no-progress", arg)) {
Shawn O. Pearcefa740522007-09-19 00:49:35 -0400954 args.no_progress = 1;
Johannes Schindelin83a5ad62007-02-20 03:01:44 +0100955 continue;
956 }
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700957 if (!strcmp("--stateless-rpc", arg)) {
958 args.stateless_rpc = 1;
959 continue;
960 }
961 if (!strcmp("--lock-pack", arg)) {
962 args.lock_pack = 1;
963 pack_lockfile_ptr = &pack_lockfile;
964 continue;
965 }
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700966 usage(fetch_pack_usage);
967 }
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400968 dest = (char *)arg;
969 heads = (char **)(argv + i + 1);
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700970 nr_heads = argc - i - 1;
971 break;
972 }
973 if (!dest)
974 usage(fetch_pack_usage);
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400975
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700976 if (args.stateless_rpc) {
977 conn = NULL;
978 fd[0] = 0;
979 fd[1] = 1;
Daniel Barkalowba227852008-02-04 13:26:23 -0500980 } else {
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700981 conn = git_connect(fd, (char *)dest, args.uploadpack,
982 args.verbose ? CONNECT_VERBOSE : 0);
Daniel Barkalowba227852008-02-04 13:26:23 -0500983 }
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700984
Jeff Kingafe7c5f2011-12-12 19:41:37 -0500985 get_remote_heads(fd[0], &ref, 0, NULL);
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700986
987 ref = fetch_pack(&args, fd, conn, ref, dest,
988 nr_heads, heads, pack_lockfile_ptr);
989 if (pack_lockfile) {
990 printf("lock %s\n", pack_lockfile);
991 fflush(stdout);
992 }
993 close(fd[0]);
994 close(fd[1]);
995 if (finish_connect(conn))
996 ref = NULL;
Daniel Barkalowba227852008-02-04 13:26:23 -0500997 ret = !ref;
Junio C Hamano9e5d2b42005-11-06 00:09:59 -0800998
999 if (!ret && nr_heads) {
1000 /* If the heads to pull were given, we should have
1001 * consumed all of them by matching the remote.
Heikki Orsila05207a22008-09-09 13:28:30 +03001002 * Otherwise, 'git fetch remote no-such-ref' would
Junio C Hamano9e5d2b42005-11-06 00:09:59 -08001003 * silently succeed without issuing an error.
1004 */
1005 for (i = 0; i < nr_heads; i++)
1006 if (heads[i] && heads[i][0]) {
1007 error("no such remote ref %s", heads[i]);
1008 ret = 1;
1009 }
1010 }
Daniel Barkalowba227852008-02-04 13:26:23 -05001011 while (ref) {
1012 printf("%s %s\n",
1013 sha1_to_hex(ref->old_sha1), ref->name);
1014 ref = ref->next;
1015 }
Junio C Hamano9e5d2b42005-11-06 00:09:59 -08001016
Daniel Barkalowba227852008-02-04 13:26:23 -05001017 return ret;
1018}
1019
1020struct ref *fetch_pack(struct fetch_pack_args *my_args,
1021 int fd[], struct child_process *conn,
1022 const struct ref *ref,
1023 const char *dest,
1024 int nr_heads,
1025 char **heads,
1026 char **pack_lockfile)
1027{
1028 struct stat st;
1029 struct ref *ref_cpy;
1030
1031 fetch_pack_setup();
Thomas Rastd551bba2008-12-06 21:50:09 +01001032 if (&args != my_args)
1033 memcpy(&args, my_args, sizeof(args));
Daniel Barkalowba227852008-02-04 13:26:23 -05001034 if (args.depth > 0) {
1035 if (stat(git_path("shallow"), &st))
1036 st.st_mtime = 0;
1037 }
1038
1039 if (heads && nr_heads)
1040 nr_heads = remove_duplicates(nr_heads, heads);
1041 if (!ref) {
1042 packet_flush(fd[1]);
1043 die("no matching remote head");
1044 }
1045 ref_cpy = do_fetch_pack(fd, ref, nr_heads, heads, pack_lockfile);
1046
1047 if (args.depth > 0) {
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001048 struct cache_time mtime;
Shawn O. Pearceedace6f2009-10-30 17:47:23 -07001049 struct strbuf sb = STRBUF_INIT;
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001050 char *shallow = git_path("shallow");
1051 int fd;
1052
1053 mtime.sec = st.st_mtime;
Kjetil Barvikc06ff492009-03-04 18:47:40 +01001054 mtime.nsec = ST_MTIME_NSEC(st);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001055 if (stat(shallow, &st)) {
1056 if (mtime.sec)
1057 die("shallow file was removed during fetch");
1058 } else if (st.st_mtime != mtime.sec
1059#ifdef USE_NSEC
Kjetil Barvik5bcf1092009-03-15 12:38:55 +01001060 || ST_MTIME_NSEC(st) != mtime.nsec
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001061#endif
1062 )
1063 die("shallow file was changed during fetch");
1064
Junio C Hamanoacd3b9e2008-10-17 15:44:39 -07001065 fd = hold_lock_file_for_update(&lock, shallow,
1066 LOCK_DIE_ON_ERROR);
Shawn O. Pearceedace6f2009-10-30 17:47:23 -07001067 if (!write_shallow_commits(&sb, 0)
1068 || write_in_full(fd, sb.buf, sb.len) != sb.len) {
Alex Riesen691f1a22009-04-29 23:22:56 +02001069 unlink_or_warn(shallow);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001070 rollback_lock_file(&lock);
1071 } else {
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001072 commit_lock_file(&lock);
1073 }
Shawn O. Pearceedace6f2009-10-30 17:47:23 -07001074 strbuf_release(&sb);
Johannes Schindelin016e6cc2006-10-30 20:09:29 +01001075 }
1076
Johan Herland48ec3e52008-06-15 16:04:20 +02001077 reprepare_packed_git();
Daniel Barkalowba227852008-02-04 13:26:23 -05001078 return ref_cpy;
Linus Torvaldsdef88e92005-07-04 13:26:53 -07001079}