blob: cf3019e05bdf45e1fa16a483621a6773976f98c5 [file] [log] [blame]
Stephen Boydc2e86ad2011-03-22 00:51:05 -07001#include "builtin.h"
Linus Torvaldsdef88e92005-07-04 13:26:53 -07002#include "pkt-line.h"
Daniel Barkalow2d4177c2007-09-10 23:03:00 -04003#include "fetch-pack.h"
Junio C Hamano47a59182013-07-08 13:56:53 -07004#include "remote.h"
5#include "connect.h"
Nguyễn Thái Ngọc Duy16094882013-12-05 20:02:50 +07006#include "sha1-array.h"
Shawn O. Pearcefa740522007-09-19 00:49:35 -04007
Junio C Hamano33b83032005-08-12 02:08:29 -07008static const char fetch_pack_usage[] =
Alex Henrie9c9b4f22015-01-13 00:44:47 -07009"git fetch-pack [--all] [--stdin] [--quiet | -q] [--keep | -k] [--thin] "
Ivan Todoroski078b8952012-04-02 17:13:48 +020010"[--include-tag] [--upload-pack=<git-upload-pack>] [--depth=<n>] "
Torsten Bögershausen5610b7c2013-11-28 20:49:17 +010011"[--no-progress] [--diag-url] [-v] [<host>:]<directory> [<refs>...]";
Linus Torvaldsdef88e92005-07-04 13:26:53 -070012
Junio C Hamanof2db8542013-01-29 14:02:15 -080013static void add_sought_entry_mem(struct ref ***sought, int *nr, int *alloc,
14 const char *name, int namelen)
15{
16 struct ref *ref = xcalloc(1, sizeof(*ref) + namelen + 1);
brian m. carlson854ecb92015-11-10 02:22:21 +000017 struct object_id oid;
18 const int chunksz = GIT_SHA1_HEXSZ + 1;
Nguyễn Thái Ngọc Duy58f2ed02013-12-05 20:02:49 +070019
brian m. carlson854ecb92015-11-10 02:22:21 +000020 if (namelen > chunksz && name[chunksz - 1] == ' ' &&
21 !get_oid_hex(name, &oid)) {
22 oidcpy(&ref->old_oid, &oid);
23 name += chunksz;
24 namelen -= chunksz;
Nguyễn Thái Ngọc Duy58f2ed02013-12-05 20:02:49 +070025 }
Junio C Hamanof2db8542013-01-29 14:02:15 -080026
27 memcpy(ref->name, name, namelen);
28 ref->name[namelen] = '\0';
29 (*nr)++;
30 ALLOC_GROW(*sought, *nr, *alloc);
31 (*sought)[*nr - 1] = ref;
32}
33
34static void add_sought_entry(struct ref ***sought, int *nr, int *alloc,
35 const char *string)
36{
37 add_sought_entry_mem(sought, nr, alloc, string, strlen(string));
38}
39
Daniel Barkalow2d4177c2007-09-10 23:03:00 -040040int cmd_fetch_pack(int argc, const char **argv, const char *prefix)
Linus Torvaldsdef88e92005-07-04 13:26:53 -070041{
Michael Haggerty57e6fc692012-05-21 09:59:59 +020042 int i, ret;
Daniel Barkalowba227852008-02-04 13:26:23 -050043 struct ref *ref = NULL;
Michael Haggerty9d19c6e2012-05-21 09:59:56 +020044 const char *dest = NULL;
Junio C Hamanof2db8542013-01-29 14:02:15 -080045 struct ref **sought = NULL;
46 int nr_sought = 0, alloc_sought = 0;
Daniel Barkalowba227852008-02-04 13:26:23 -050047 int fd[2];
Shawn O. Pearce249b2002009-10-30 17:47:42 -070048 char *pack_lockfile = NULL;
49 char **pack_lockfile_ptr = NULL;
Daniel Barkalowba227852008-02-04 13:26:23 -050050 struct child_process *conn;
Nguyễn Thái Ngọc Duyf8eb3032012-10-26 22:53:54 +070051 struct fetch_pack_args args;
Nguyễn Thái Ngọc Duy16094882013-12-05 20:02:50 +070052 struct sha1_array shallow = SHA1_ARRAY_INIT;
Linus Torvaldsdef88e92005-07-04 13:26:53 -070053
Jeff Kingbbc30f92011-02-24 09:30:19 -050054 packet_trace_identity("fetch-pack");
55
Nguyễn Thái Ngọc Duyf8eb3032012-10-26 22:53:54 +070056 memset(&args, 0, sizeof(args));
57 args.uploadpack = "git-upload-pack";
58
Michael Haggertyff22ff92012-05-21 09:59:58 +020059 for (i = 1; i < argc && *argv[i] == '-'; i++) {
Daniel Barkalow2d4177c2007-09-10 23:03:00 -040060 const char *arg = argv[i];
Linus Torvaldsdef88e92005-07-04 13:26:53 -070061
Christian Couder59556542013-11-30 21:55:40 +010062 if (starts_with(arg, "--upload-pack=")) {
Michael Haggertyff22ff92012-05-21 09:59:58 +020063 args.uploadpack = arg + 14;
64 continue;
Linus Torvaldsdef88e92005-07-04 13:26:53 -070065 }
Christian Couder59556542013-11-30 21:55:40 +010066 if (starts_with(arg, "--exec=")) {
Michael Haggertyff22ff92012-05-21 09:59:58 +020067 args.uploadpack = arg + 7;
68 continue;
69 }
70 if (!strcmp("--quiet", arg) || !strcmp("-q", arg)) {
71 args.quiet = 1;
72 continue;
73 }
74 if (!strcmp("--keep", arg) || !strcmp("-k", arg)) {
75 args.lock_pack = args.keep_pack;
76 args.keep_pack = 1;
77 continue;
78 }
79 if (!strcmp("--thin", arg)) {
80 args.use_thin_pack = 1;
81 continue;
82 }
83 if (!strcmp("--include-tag", arg)) {
84 args.include_tag = 1;
85 continue;
86 }
87 if (!strcmp("--all", arg)) {
88 args.fetch_all = 1;
89 continue;
90 }
91 if (!strcmp("--stdin", arg)) {
92 args.stdin_refs = 1;
93 continue;
94 }
Torsten Bögershausen5610b7c2013-11-28 20:49:17 +010095 if (!strcmp("--diag-url", arg)) {
96 args.diag_url = 1;
97 continue;
98 }
Michael Haggertyff22ff92012-05-21 09:59:58 +020099 if (!strcmp("-v", arg)) {
100 args.verbose = 1;
101 continue;
102 }
Christian Couder59556542013-11-30 21:55:40 +0100103 if (starts_with(arg, "--depth=")) {
Michael Haggertyff22ff92012-05-21 09:59:58 +0200104 args.depth = strtol(arg + 8, NULL, 0);
105 continue;
106 }
107 if (!strcmp("--no-progress", arg)) {
108 args.no_progress = 1;
109 continue;
110 }
111 if (!strcmp("--stateless-rpc", arg)) {
112 args.stateless_rpc = 1;
113 continue;
114 }
115 if (!strcmp("--lock-pack", arg)) {
116 args.lock_pack = 1;
117 pack_lockfile_ptr = &pack_lockfile;
118 continue;
119 }
Nguyễn Thái Ngọc Duy9ba38042013-07-21 15:18:05 +0700120 if (!strcmp("--check-self-contained-and-connected", arg)) {
121 args.check_self_contained_and_connected = 1;
122 continue;
123 }
Nguyễn Thái Ngọc Duy16094882013-12-05 20:02:50 +0700124 if (!strcmp("--cloning", arg)) {
125 args.cloning = 1;
126 continue;
127 }
128 if (!strcmp("--update-shallow", arg)) {
129 args.update_shallow = 1;
130 continue;
131 }
Michael Haggertyff22ff92012-05-21 09:59:58 +0200132 usage(fetch_pack_usage);
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700133 }
Michael Haggerty4cc00fc2012-05-21 09:59:57 +0200134
135 if (i < argc)
136 dest = argv[i++];
137 else
Linus Torvaldsdef88e92005-07-04 13:26:53 -0700138 usage(fetch_pack_usage);
Daniel Barkalow2d4177c2007-09-10 23:03:00 -0400139
Michael Haggerty57e6fc692012-05-21 09:59:59 +0200140 /*
141 * Copy refs from cmdline to growable list, then append any
142 * refs from the standard input:
143 */
Michael Haggerty57e6fc692012-05-21 09:59:59 +0200144 for (; i < argc; i++)
Junio C Hamanof2db8542013-01-29 14:02:15 -0800145 add_sought_entry(&sought, &nr_sought, &alloc_sought, argv[i]);
Ivan Todoroski078b8952012-04-02 17:13:48 +0200146 if (args.stdin_refs) {
Ivan Todoroski078b8952012-04-02 17:13:48 +0200147 if (args.stateless_rpc) {
148 /* in stateless RPC mode we use pkt-line to read
149 * from stdin, until we get a flush packet
150 */
Ivan Todoroski078b8952012-04-02 17:13:48 +0200151 for (;;) {
Jeff King74543a02013-02-20 15:02:57 -0500152 char *line = packet_read_line(0, NULL);
153 if (!line)
Ivan Todoroski078b8952012-04-02 17:13:48 +0200154 break;
Junio C Hamanoe013bda2013-04-01 08:59:37 -0700155 add_sought_entry(&sought, &nr_sought, &alloc_sought, line);
Ivan Todoroski078b8952012-04-02 17:13:48 +0200156 }
157 }
158 else {
159 /* read from stdin one ref per line, until EOF */
160 struct strbuf line = STRBUF_INIT;
Michael Haggerty8bee93d2012-09-09 08:19:40 +0200161 while (strbuf_getline(&line, stdin, '\n') != EOF)
Junio C Hamanof2db8542013-01-29 14:02:15 -0800162 add_sought_entry(&sought, &nr_sought, &alloc_sought, line.buf);
Ivan Todoroski078b8952012-04-02 17:13:48 +0200163 strbuf_release(&line);
164 }
165 }
166
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700167 if (args.stateless_rpc) {
168 conn = NULL;
169 fd[0] = 0;
170 fd[1] = 1;
Daniel Barkalowba227852008-02-04 13:26:23 -0500171 } else {
Torsten Bögershausen5610b7c2013-11-28 20:49:17 +0100172 int flags = args.verbose ? CONNECT_VERBOSE : 0;
173 if (args.diag_url)
174 flags |= CONNECT_DIAG_URL;
Michael Haggerty9d19c6e2012-05-21 09:59:56 +0200175 conn = git_connect(fd, dest, args.uploadpack,
Torsten Bögershausen5610b7c2013-11-28 20:49:17 +0100176 flags);
177 if (!conn)
178 return args.diag_url ? 0 : 1;
Daniel Barkalowba227852008-02-04 13:26:23 -0500179 }
Nguyễn Thái Ngọc Duy16094882013-12-05 20:02:50 +0700180 get_remote_heads(fd[0], NULL, 0, &ref, 0, NULL, &shallow);
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700181
Nguyễn Thái Ngọc Duy16094882013-12-05 20:02:50 +0700182 ref = fetch_pack(&args, fd, conn, ref, dest, sought, nr_sought,
183 &shallow, pack_lockfile_ptr);
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700184 if (pack_lockfile) {
185 printf("lock %s\n", pack_lockfile);
186 fflush(stdout);
187 }
Nguyễn Thái Ngọc Duy9ba38042013-07-21 15:18:05 +0700188 if (args.check_self_contained_and_connected &&
189 args.self_contained_and_connected) {
190 printf("connectivity-ok\n");
191 fflush(stdout);
192 }
Shawn O. Pearce249b2002009-10-30 17:47:42 -0700193 close(fd[0]);
194 close(fd[1]);
195 if (finish_connect(conn))
Michael Haggerty7418f1a2012-09-09 08:19:46 +0200196 return 1;
Junio C Hamano9e5d2b42005-11-06 00:09:59 -0800197
Junio C Hamanof2db8542013-01-29 14:02:15 -0800198 ret = !ref;
Michael Haggertyb2856682012-09-09 08:19:48 +0200199
200 /*
201 * If the heads to pull were given, we should have consumed
202 * all of them by matching the remote. Otherwise, 'git fetch
203 * remote no-such-ref' would silently succeed without issuing
204 * an error.
205 */
Junio C Hamanof2db8542013-01-29 14:02:15 -0800206 for (i = 0; i < nr_sought; i++) {
207 if (!sought[i] || sought[i]->matched)
208 continue;
209 error("no such remote ref %s", sought[i]->name);
210 ret = 1;
211 }
212
Daniel Barkalowba227852008-02-04 13:26:23 -0500213 while (ref) {
214 printf("%s %s\n",
brian m. carlsonf4e54d02015-11-10 02:22:20 +0000215 oid_to_hex(&ref->old_oid), ref->name);
Daniel Barkalowba227852008-02-04 13:26:23 -0500216 ref = ref->next;
217 }
Junio C Hamano9e5d2b42005-11-06 00:09:59 -0800218
Daniel Barkalowba227852008-02-04 13:26:23 -0500219 return ret;
220}