builtin-prune: memory diet.
[git] / builtin-push.c
1 /*
2  * "git push"
3  */
4 #include "cache.h"
5 #include "refs.h"
6 #include "run-command.h"
7 #include "builtin.h"
8
9 #define MAX_URI (16)
10
11 static const char push_usage[] = "git-push [--all] [--tags] [-f | --force] <repository> [<refspec>...]";
12
13 static int all, tags, force, thin = 1, verbose;
14 static const char *execute;
15
16 #define BUF_SIZE (2084)
17 static char buffer[BUF_SIZE];
18
19 static const char **refspec;
20 static int refspec_nr;
21
22 static void add_refspec(const char *ref)
23 {
24         int nr = refspec_nr + 1;
25         refspec = xrealloc(refspec, nr * sizeof(char *));
26         refspec[nr-1] = ref;
27         refspec_nr = nr;
28 }
29
30 static int expand_one_ref(const char *ref, const unsigned char *sha1, int flag, void *cb_data)
31 {
32         /* Ignore the "refs/" at the beginning of the refname */
33         ref += 5;
34
35         if (!strncmp(ref, "tags/", 5))
36                 add_refspec(xstrdup(ref));
37         return 0;
38 }
39
40 static void expand_refspecs(void)
41 {
42         if (all) {
43                 if (refspec_nr)
44                         die("cannot mix '--all' and a refspec");
45
46                 /*
47                  * No need to expand "--all" - we'll just use
48                  * the "--all" flag to send-pack
49                  */
50                 return;
51         }
52         if (!tags)
53                 return;
54         for_each_ref(expand_one_ref, NULL);
55 }
56
57 static void set_refspecs(const char **refs, int nr)
58 {
59         if (nr) {
60                 int pass;
61                 for (pass = 0; pass < 2; pass++) {
62                         /* pass 0 counts and allocates, pass 1 fills */
63                         int i, cnt;
64                         for (i = cnt = 0; i < nr; i++) {
65                                 if (!strcmp("tag", refs[i])) {
66                                         int len;
67                                         char *tag;
68                                         if (nr <= ++i)
69                                                 die("tag <tag> shorthand without <tag>");
70                                         if (pass) {
71                                                 len = strlen(refs[i]) + 11;
72                                                 tag = xmalloc(len);
73                                                 strcpy(tag, "refs/tags/");
74                                                 strcat(tag, refs[i]);
75                                                 refspec[cnt] = tag;
76                                         }
77                                         cnt++;
78                                         continue;
79                                 }
80                                 if (pass)
81                                         refspec[cnt] = refs[i];
82                                 cnt++;
83                         }
84                         if (!pass) {
85                                 size_t bytes = cnt * sizeof(char *);
86                                 refspec_nr = cnt;
87                                 refspec = xrealloc(refspec, bytes);
88                         }
89                 }
90         }
91         expand_refspecs();
92 }
93
94 static int get_remotes_uri(const char *repo, const char *uri[MAX_URI])
95 {
96         int n = 0;
97         FILE *f = fopen(git_path("remotes/%s", repo), "r");
98         int has_explicit_refspec = refspec_nr || all || tags;
99
100         if (!f)
101                 return -1;
102         while (fgets(buffer, BUF_SIZE, f)) {
103                 int is_refspec;
104                 char *s, *p;
105
106                 if (!strncmp("URL:", buffer, 4)) {
107                         is_refspec = 0;
108                         s = buffer + 4;
109                 } else if (!strncmp("Push:", buffer, 5)) {
110                         is_refspec = 1;
111                         s = buffer + 5;
112                 } else
113                         continue;
114
115                 /* Remove whitespace at the head.. */
116                 while (isspace(*s))
117                         s++;
118                 if (!*s)
119                         continue;
120
121                 /* ..and at the end */
122                 p = s + strlen(s);
123                 while (isspace(p[-1]))
124                         *--p = 0;
125
126                 if (!is_refspec) {
127                         if (n < MAX_URI)
128                                 uri[n++] = xstrdup(s);
129                         else
130                                 error("more than %d URL's specified, ignoring the rest", MAX_URI);
131                 }
132                 else if (is_refspec && !has_explicit_refspec)
133                         add_refspec(xstrdup(s));
134         }
135         fclose(f);
136         if (!n)
137                 die("remote '%s' has no URL", repo);
138         return n;
139 }
140
141 static const char **config_uri;
142 static const char *config_repo;
143 static int config_repo_len;
144 static int config_current_uri;
145 static int config_get_refspecs;
146
147 static int get_remote_config(const char* key, const char* value)
148 {
149         if (!strncmp(key, "remote.", 7) &&
150             !strncmp(key + 7, config_repo, config_repo_len)) {
151                 if (!strcmp(key + 7 + config_repo_len, ".url")) {
152                         if (config_current_uri < MAX_URI)
153                                 config_uri[config_current_uri++] = xstrdup(value);
154                         else
155                                 error("more than %d URL's specified, ignoring the rest", MAX_URI);
156                 }
157                 else if (config_get_refspecs &&
158                          !strcmp(key + 7 + config_repo_len, ".push"))
159                         add_refspec(xstrdup(value));
160         }
161         return 0;
162 }
163
164 static int get_config_remotes_uri(const char *repo, const char *uri[MAX_URI])
165 {
166         config_repo_len = strlen(repo);
167         config_repo = repo;
168         config_current_uri = 0;
169         config_uri = uri;
170         config_get_refspecs = !(refspec_nr || all || tags);
171
172         git_config(get_remote_config);
173         return config_current_uri;
174 }
175
176 static int get_branches_uri(const char *repo, const char *uri[MAX_URI])
177 {
178         const char *slash = strchr(repo, '/');
179         int n = slash ? slash - repo : 1000;
180         FILE *f = fopen(git_path("branches/%.*s", n, repo), "r");
181         char *s, *p;
182         int len;
183
184         if (!f)
185                 return 0;
186         s = fgets(buffer, BUF_SIZE, f);
187         fclose(f);
188         if (!s)
189                 return 0;
190         while (isspace(*s))
191                 s++;
192         if (!*s)
193                 return 0;
194         p = s + strlen(s);
195         while (isspace(p[-1]))
196                 *--p = 0;
197         len = p - s;
198         if (slash)
199                 len += strlen(slash);
200         p = xmalloc(len + 1);
201         strcpy(p, s);
202         if (slash)
203                 strcat(p, slash);
204         uri[0] = p;
205         return 1;
206 }
207
208 /*
209  * Read remotes and branches file, fill the push target URI
210  * list.  If there is no command line refspecs, read Push: lines
211  * to set up the *refspec list as well.
212  * return the number of push target URIs
213  */
214 static int read_config(const char *repo, const char *uri[MAX_URI])
215 {
216         int n;
217
218         if (*repo != '/') {
219                 n = get_remotes_uri(repo, uri);
220                 if (n > 0)
221                         return n;
222
223                 n = get_config_remotes_uri(repo, uri);
224                 if (n > 0)
225                         return n;
226
227                 n = get_branches_uri(repo, uri);
228                 if (n > 0)
229                         return n;
230         }
231
232         uri[0] = repo;
233         return 1;
234 }
235
236 static int do_push(const char *repo)
237 {
238         const char *uri[MAX_URI];
239         int i, n;
240         int common_argc;
241         const char **argv;
242         int argc;
243
244         n = read_config(repo, uri);
245         if (n <= 0)
246                 die("bad repository '%s'", repo);
247
248         argv = xmalloc((refspec_nr + 10) * sizeof(char *));
249         argv[0] = "dummy-send-pack";
250         argc = 1;
251         if (all)
252                 argv[argc++] = "--all";
253         if (force)
254                 argv[argc++] = "--force";
255         if (execute)
256                 argv[argc++] = execute;
257         common_argc = argc;
258
259         for (i = 0; i < n; i++) {
260                 int err;
261                 int dest_argc = common_argc;
262                 int dest_refspec_nr = refspec_nr;
263                 const char **dest_refspec = refspec;
264                 const char *dest = uri[i];
265                 const char *sender = "git-send-pack";
266                 if (!strncmp(dest, "http://", 7) ||
267                     !strncmp(dest, "https://", 8))
268                         sender = "git-http-push";
269                 else if (thin)
270                         argv[dest_argc++] = "--thin";
271                 argv[0] = sender;
272                 argv[dest_argc++] = dest;
273                 while (dest_refspec_nr--)
274                         argv[dest_argc++] = *dest_refspec++;
275                 argv[dest_argc] = NULL;
276                 if (verbose)
277                         fprintf(stderr, "Pushing to %s\n", dest);
278                 err = run_command_v(argv);
279                 if (!err)
280                         continue;
281                 switch (err) {
282                 case -ERR_RUN_COMMAND_FORK:
283                         die("unable to fork for %s", sender);
284                 case -ERR_RUN_COMMAND_EXEC:
285                         die("unable to exec %s", sender);
286                 case -ERR_RUN_COMMAND_WAITPID:
287                 case -ERR_RUN_COMMAND_WAITPID_WRONG_PID:
288                 case -ERR_RUN_COMMAND_WAITPID_SIGNAL:
289                 case -ERR_RUN_COMMAND_WAITPID_NOEXIT:
290                         die("%s died with strange error", sender);
291                 default:
292                         return -err;
293                 }
294         }
295         return 0;
296 }
297
298 int cmd_push(int argc, const char **argv, const char *prefix)
299 {
300         int i;
301         const char *repo = "origin";    /* default repository */
302
303         for (i = 1; i < argc; i++) {
304                 const char *arg = argv[i];
305
306                 if (arg[0] != '-') {
307                         repo = arg;
308                         i++;
309                         break;
310                 }
311                 if (!strcmp(arg, "-v")) {
312                         verbose=1;
313                         continue;
314                 }
315                 if (!strncmp(arg, "--repo=", 7)) {
316                         repo = arg+7;
317                         continue;
318                 }
319                 if (!strcmp(arg, "--all")) {
320                         all = 1;
321                         continue;
322                 }
323                 if (!strcmp(arg, "--tags")) {
324                         tags = 1;
325                         continue;
326                 }
327                 if (!strcmp(arg, "--force") || !strcmp(arg, "-f")) {
328                         force = 1;
329                         continue;
330                 }
331                 if (!strcmp(arg, "--thin")) {
332                         thin = 1;
333                         continue;
334                 }
335                 if (!strcmp(arg, "--no-thin")) {
336                         thin = 0;
337                         continue;
338                 }
339                 if (!strncmp(arg, "--exec=", 7)) {
340                         execute = arg;
341                         continue;
342                 }
343                 usage(push_usage);
344         }
345         set_refspecs(argv + i, argc - i);
346         return do_push(repo);
347 }