1#include "builtin.h"
2#include "exec_cmd.h"
3#include "cache.h"
4#include "quote.h"
5
6const char git_usage_string[] =
7 "git [--version] [--exec-path[=GIT_EXEC_PATH]] [-p|--paginate|--no-pager] [--bare] [--git-dir=GIT_DIR] [--work-tree=GIT_WORK_TREE] [--help] COMMAND [ARGS]";
8
9static int handle_options(const char*** argv, int* argc, int* envchanged)
10{
11 int handled = 0;
12
13 while (*argc > 0) {
14 const char *cmd = (*argv)[0];
15 if (cmd[0] != '-')
16 break;
17
18 /*
19 * For legacy reasons, the "version" and "help"
20 * commands can be written with "--" prepended
21 * to make them look like flags.
22 */
23 if (!strcmp(cmd, "--help") || !strcmp(cmd, "--version"))
24 break;
25
26 /*
27 * Check remaining flags.
28 */
29 if (!prefixcmp(cmd, "--exec-path")) {
30 cmd += 11;
31 if (*cmd == '=')
32 git_set_argv_exec_path(cmd + 1);
33 else {
34 puts(git_exec_path());
35 exit(0);
36 }
37 } else if (!strcmp(cmd, "-p") || !strcmp(cmd, "--paginate")) {
38 setup_pager();
39 } else if (!strcmp(cmd, "--no-pager")) {
40 setenv("GIT_PAGER", "cat", 1);
41 if (envchanged)
42 *envchanged = 1;
43 } else if (!strcmp(cmd, "--git-dir")) {
44 if (*argc < 2) {
45 fprintf(stderr, "No directory given for --git-dir.\n" );
46 usage(git_usage_string);
47 }
48 setenv(GIT_DIR_ENVIRONMENT, (*argv)[1], 1);
49 if (envchanged)
50 *envchanged = 1;
51 (*argv)++;
52 (*argc)--;
53 handled++;
54 } else if (!prefixcmp(cmd, "--git-dir=")) {
55 setenv(GIT_DIR_ENVIRONMENT, cmd + 10, 1);
56 if (envchanged)
57 *envchanged = 1;
58 } else if (!strcmp(cmd, "--work-tree")) {
59 if (*argc < 2) {
60 fprintf(stderr, "No directory given for --work-tree.\n" );
61 usage(git_usage_string);
62 }
63 setenv(GIT_WORK_TREE_ENVIRONMENT, (*argv)[1], 1);
64 if (envchanged)
65 *envchanged = 1;
66 (*argv)++;
67 (*argc)--;
68 } else if (!prefixcmp(cmd, "--work-tree=")) {
69 setenv(GIT_WORK_TREE_ENVIRONMENT, cmd + 12, 1);
70 if (envchanged)
71 *envchanged = 1;
72 } else if (!strcmp(cmd, "--bare")) {
73 static char git_dir[PATH_MAX+1];
74 is_bare_repository_cfg = 1;
75 setenv(GIT_DIR_ENVIRONMENT, getcwd(git_dir, sizeof(git_dir)), 0);
76 if (envchanged)
77 *envchanged = 1;
78 } else {
79 fprintf(stderr, "Unknown option: %s\n", cmd);
80 usage(git_usage_string);
81 }
82
83 (*argv)++;
84 (*argc)--;
85 handled++;
86 }
87 return handled;
88}
89
90static int split_cmdline(char *cmdline, const char ***argv)
91{
92 int src, dst, count = 0, size = 16;
93 char quoted = 0;
94
95 *argv = xmalloc(sizeof(char*) * size);
96
97 /* split alias_string */
98 (*argv)[count++] = cmdline;
99 for (src = dst = 0; cmdline[src];) {
100 char c = cmdline[src];
101 if (!quoted && isspace(c)) {
102 cmdline[dst++] = 0;
103 while (cmdline[++src]
104 && isspace(cmdline[src]))
105 ; /* skip */
106 if (count >= size) {
107 size += 16;
108 *argv = xrealloc(*argv, sizeof(char*) * size);
109 }
110 (*argv)[count++] = cmdline + dst;
111 } else if(!quoted && (c == '\'' || c == '"')) {
112 quoted = c;
113 src++;
114 } else if (c == quoted) {
115 quoted = 0;
116 src++;
117 } else {
118 if (c == '\\' && quoted != '\'') {
119 src++;
120 c = cmdline[src];
121 if (!c) {
122 free(*argv);
123 *argv = NULL;
124 return error("cmdline ends with \\");
125 }
126 }
127 cmdline[dst++] = c;
128 src++;
129 }
130 }
131
132 cmdline[dst] = 0;
133
134 if (quoted) {
135 free(*argv);
136 *argv = NULL;
137 return error("unclosed quote");
138 }
139
140 return count;
141}
142
143static int handle_alias(int *argcp, const char ***argv)
144{
145 int nongit = 0, envchanged = 0, ret = 0, saved_errno = errno;
146 const char *subdir;
147 int count, option_count;
148 const char** new_argv;
149 const char *alias_command;
150 char *alias_string;
151
152 subdir = setup_git_directory_gently(&nongit);
153
154 alias_command = (*argv)[0];
155 alias_string = alias_lookup(alias_command);
156 if (alias_string) {
157 if (alias_string[0] == '!') {
158 if (*argcp > 1) {
159 struct strbuf buf;
160
161 strbuf_init(&buf, PATH_MAX);
162 strbuf_addstr(&buf, alias_string);
163 sq_quote_argv(&buf, (*argv) + 1, PATH_MAX);
164 free(alias_string);
165 alias_string = buf.buf;
166 }
167 trace_printf("trace: alias to shell cmd: %s => %s\n",
168 alias_command, alias_string + 1);
169 ret = system(alias_string + 1);
170 if (ret >= 0 && WIFEXITED(ret) &&
171 WEXITSTATUS(ret) != 127)
172 exit(WEXITSTATUS(ret));
173 die("Failed to run '%s' when expanding alias '%s'\n",
174 alias_string + 1, alias_command);
175 }
176 count = split_cmdline(alias_string, &new_argv);
177 option_count = handle_options(&new_argv, &count, &envchanged);
178 if (envchanged)
179 die("alias '%s' changes environment variables\n"
180 "You can use '!git' in the alias to do this.",
181 alias_command);
182 memmove(new_argv - option_count, new_argv,
183 count * sizeof(char *));
184 new_argv -= option_count;
185
186 if (count < 1)
187 die("empty alias for %s", alias_command);
188
189 if (!strcmp(alias_command, new_argv[0]))
190 die("recursive alias: %s", alias_command);
191
192 trace_argv_printf(new_argv,
193 "trace: alias expansion: %s =>",
194 alias_command);
195
196 new_argv = xrealloc(new_argv, sizeof(char*) *
197 (count + *argcp + 1));
198 /* insert after command name */
199 memcpy(new_argv + count, *argv + 1, sizeof(char*) * *argcp);
200 new_argv[count+*argcp] = NULL;
201
202 *argv = new_argv;
203 *argcp += count - 1;
204
205 ret = 1;
206 }
207
208 if (subdir)
209 chdir(subdir);
210
211 errno = saved_errno;
212
213 return ret;
214}
215
216const char git_version_string[] = GIT_VERSION;
217
218#define RUN_SETUP (1<<0)
219#define USE_PAGER (1<<1)
220/*
221 * require working tree to be present -- anything uses this needs
222 * RUN_SETUP for reading from the configuration file.
223 */
224#define NEED_WORK_TREE (1<<2)
225
226struct cmd_struct {
227 const char *cmd;
228 int (*fn)(int, const char **, const char *);
229 int option;
230};
231
232static int run_command(struct cmd_struct *p, int argc, const char **argv)
233{
234 int status;
235 struct stat st;
236 const char *prefix;
237
238 prefix = NULL;
239 if (p->option & RUN_SETUP)
240 prefix = setup_git_directory();
241 if (p->option & USE_PAGER)
242 setup_pager();
243 if (p->option & NEED_WORK_TREE)
244 setup_work_tree();
245
246 trace_argv_printf(argv, "trace: built-in: git");
247
248 status = p->fn(argc, argv, prefix);
249 if (status)
250 return status & 0xff;
251
252 /* Somebody closed stdout? */
253 if (fstat(fileno(stdout), &st))
254 return 0;
255 /* Ignore write errors for pipes and sockets.. */
256 if (S_ISFIFO(st.st_mode) || S_ISSOCK(st.st_mode))
257 return 0;
258
259 /* Check for ENOSPC and EIO errors.. */
260 if (fflush(stdout))
261 die("write failure on standard output: %s", strerror(errno));
262 if (ferror(stdout))
263 die("unknown write failure on standard output");
264 if (fclose(stdout))
265 die("close failed on standard output: %s", strerror(errno));
266 return 0;
267}
268
269static void handle_internal_command(int argc, const char **argv)
270{
271 const char *cmd = argv[0];
272 static struct cmd_struct commands[] = {
273 { "add", cmd_add, RUN_SETUP | NEED_WORK_TREE },
274 { "annotate", cmd_annotate, RUN_SETUP },
275 { "apply", cmd_apply },
276 { "archive", cmd_archive },
277 { "blame", cmd_blame, RUN_SETUP },
278 { "branch", cmd_branch, RUN_SETUP },
279 { "bundle", cmd_bundle },
280 { "cat-file", cmd_cat_file, RUN_SETUP },
281 { "checkout", cmd_checkout, RUN_SETUP | NEED_WORK_TREE },
282 { "checkout-index", cmd_checkout_index,
283 RUN_SETUP | NEED_WORK_TREE},
284 { "check-ref-format", cmd_check_ref_format },
285 { "check-attr", cmd_check_attr, RUN_SETUP | NEED_WORK_TREE },
286 { "cherry", cmd_cherry, RUN_SETUP },
287 { "cherry-pick", cmd_cherry_pick, RUN_SETUP | NEED_WORK_TREE },
288 { "clean", cmd_clean, RUN_SETUP | NEED_WORK_TREE },
289 { "commit", cmd_commit, RUN_SETUP | NEED_WORK_TREE },
290 { "commit-tree", cmd_commit_tree, RUN_SETUP },
291 { "config", cmd_config },
292 { "count-objects", cmd_count_objects, RUN_SETUP },
293 { "describe", cmd_describe, RUN_SETUP },
294 { "diff", cmd_diff },
295 { "diff-files", cmd_diff_files },
296 { "diff-index", cmd_diff_index, RUN_SETUP },
297 { "diff-tree", cmd_diff_tree, RUN_SETUP },
298 { "fast-export", cmd_fast_export, RUN_SETUP },
299 { "fetch", cmd_fetch, RUN_SETUP },
300 { "fetch-pack", cmd_fetch_pack, RUN_SETUP },
301 { "fetch--tool", cmd_fetch__tool, RUN_SETUP },
302 { "fmt-merge-msg", cmd_fmt_merge_msg, RUN_SETUP },
303 { "for-each-ref", cmd_for_each_ref, RUN_SETUP },
304 { "format-patch", cmd_format_patch, RUN_SETUP },
305 { "fsck", cmd_fsck, RUN_SETUP },
306 { "fsck-objects", cmd_fsck, RUN_SETUP },
307 { "gc", cmd_gc, RUN_SETUP },
308 { "get-tar-commit-id", cmd_get_tar_commit_id },
309 { "grep", cmd_grep, RUN_SETUP | USE_PAGER },
310 { "help", cmd_help },
311#ifndef NO_CURL
312 { "http-fetch", cmd_http_fetch, RUN_SETUP },
313#endif
314 { "init", cmd_init_db },
315 { "init-db", cmd_init_db },
316 { "log", cmd_log, RUN_SETUP | USE_PAGER },
317 { "ls-files", cmd_ls_files, RUN_SETUP },
318 { "ls-tree", cmd_ls_tree, RUN_SETUP },
319 { "ls-remote", cmd_ls_remote },
320 { "mailinfo", cmd_mailinfo },
321 { "mailsplit", cmd_mailsplit },
322 { "merge-base", cmd_merge_base, RUN_SETUP },
323 { "merge-file", cmd_merge_file },
324 { "merge-ours", cmd_merge_ours, RUN_SETUP },
325 { "merge-recursive", cmd_merge_recursive, RUN_SETUP | NEED_WORK_TREE },
326 { "merge-subtree", cmd_merge_recursive, RUN_SETUP | NEED_WORK_TREE },
327 { "mv", cmd_mv, RUN_SETUP | NEED_WORK_TREE },
328 { "name-rev", cmd_name_rev, RUN_SETUP },
329 { "pack-objects", cmd_pack_objects, RUN_SETUP },
330 { "peek-remote", cmd_ls_remote },
331 { "pickaxe", cmd_blame, RUN_SETUP },
332 { "prune", cmd_prune, RUN_SETUP },
333 { "prune-packed", cmd_prune_packed, RUN_SETUP },
334 { "push", cmd_push, RUN_SETUP },
335 { "read-tree", cmd_read_tree, RUN_SETUP },
336 { "reflog", cmd_reflog, RUN_SETUP },
337 { "remote", cmd_remote, RUN_SETUP },
338 { "repo-config", cmd_config },
339 { "rerere", cmd_rerere, RUN_SETUP },
340 { "reset", cmd_reset, RUN_SETUP },
341 { "rev-list", cmd_rev_list, RUN_SETUP },
342 { "rev-parse", cmd_rev_parse },
343 { "revert", cmd_revert, RUN_SETUP | NEED_WORK_TREE },
344 { "rm", cmd_rm, RUN_SETUP },
345 { "send-pack", cmd_send_pack, RUN_SETUP },
346 { "shortlog", cmd_shortlog, RUN_SETUP | USE_PAGER },
347 { "show-branch", cmd_show_branch, RUN_SETUP },
348 { "show", cmd_show, RUN_SETUP | USE_PAGER },
349 { "status", cmd_status, RUN_SETUP | NEED_WORK_TREE },
350 { "stripspace", cmd_stripspace },
351 { "symbolic-ref", cmd_symbolic_ref, RUN_SETUP },
352 { "tag", cmd_tag, RUN_SETUP },
353 { "tar-tree", cmd_tar_tree },
354 { "unpack-objects", cmd_unpack_objects, RUN_SETUP },
355 { "update-index", cmd_update_index, RUN_SETUP },
356 { "update-ref", cmd_update_ref, RUN_SETUP },
357 { "upload-archive", cmd_upload_archive },
358 { "verify-tag", cmd_verify_tag, RUN_SETUP },
359 { "version", cmd_version },
360 { "whatchanged", cmd_whatchanged, RUN_SETUP | USE_PAGER },
361 { "write-tree", cmd_write_tree, RUN_SETUP },
362 { "verify-pack", cmd_verify_pack },
363 { "show-ref", cmd_show_ref, RUN_SETUP },
364 { "pack-refs", cmd_pack_refs, RUN_SETUP },
365 };
366 int i;
367
368 /* Turn "git cmd --help" into "git help cmd" */
369 if (argc > 1 && !strcmp(argv[1], "--help")) {
370 argv[1] = argv[0];
371 argv[0] = cmd = "help";
372 }
373
374 for (i = 0; i < ARRAY_SIZE(commands); i++) {
375 struct cmd_struct *p = commands+i;
376 if (strcmp(p->cmd, cmd))
377 continue;
378 exit(run_command(p, argc, argv));
379 }
380}
381
382int main(int argc, const char **argv)
383{
384 const char *cmd = argv[0] ? argv[0] : "git-help";
385 char *slash = strrchr(cmd, '/');
386 const char *cmd_path = NULL;
387 int done_alias = 0;
388
389 /*
390 * Take the basename of argv[0] as the command
391 * name, and the dirname as the default exec_path
392 * if we don't have anything better.
393 */
394 if (slash) {
395 *slash++ = 0;
396 cmd_path = cmd;
397 cmd = slash;
398 }
399
400 /*
401 * "git-xxxx" is the same as "git xxxx", but we obviously:
402 *
403 * - cannot take flags in between the "git" and the "xxxx".
404 * - cannot execute it externally (since it would just do
405 * the same thing over again)
406 *
407 * So we just directly call the internal command handler, and
408 * die if that one cannot handle it.
409 */
410 if (!prefixcmp(cmd, "git-")) {
411 cmd += 4;
412 argv[0] = cmd;
413 handle_internal_command(argc, argv);
414 die("cannot handle %s internally", cmd);
415 }
416
417 /* Look for flags.. */
418 argv++;
419 argc--;
420 handle_options(&argv, &argc, NULL);
421 if (argc > 0) {
422 if (!prefixcmp(argv[0], "--"))
423 argv[0] += 2;
424 } else {
425 /* The user didn't specify a command; give them help */
426 printf("usage: %s\n\n", git_usage_string);
427 list_common_cmds_help();
428 exit(1);
429 }
430 cmd = argv[0];
431
432 /*
433 * We use PATH to find git commands, but we prepend some higher
434 * precidence paths: the "--exec-path" option, the GIT_EXEC_PATH
435 * environment, and the $(gitexecdir) from the Makefile at build
436 * time.
437 */
438 setup_path(cmd_path);
439
440 while (1) {
441 /* See if it's an internal command */
442 handle_internal_command(argc, argv);
443
444 /* .. then try the external ones */
445 execv_git_cmd(argv);
446
447 /* It could be an alias -- this works around the insanity
448 * of overriding "git log" with "git show" by having
449 * alias.log = show
450 */
451 if (done_alias || !handle_alias(&argc, &argv))
452 break;
453 done_alias = 1;
454 }
455
456 if (errno == ENOENT) {
457 if (done_alias) {
458 fprintf(stderr, "Expansion of alias '%s' failed; "
459 "'%s' is not a git-command\n",
460 cmd, argv[0]);
461 exit(1);
462 }
463 help_unknown_cmd(cmd);
464 }
465
466 fprintf(stderr, "Failed to run command '%s': %s\n",
467 cmd, strerror(errno));
468
469 return 1;
470}