Clean up work-tree handling
[git/git.git] / git.c
CommitLineData
85023577 1#include "builtin.h"
77cb17e9 2#include "exec_cmd.h"
2b11e317 3#include "cache.h"
575ba9d6 4#include "quote.h"
8e49d503 5
822a7d50 6const char git_usage_string[] =
892c41b9 7 "git [--version] [--exec-path[=GIT_EXEC_PATH]] [-p|--paginate] [--bare] [--git-dir=GIT_DIR] [--work-tree=GIT_WORK_TREE] [--help] COMMAND [ARGS]";
822a7d50 8
8e49d503
AE
9static void prepend_to_path(const char *dir, int len)
10{
554fe20d
TH
11 const char *old_path = getenv("PATH");
12 char *path;
8e49d503
AE
13 int path_len = len;
14
15 if (!old_path)
7dbc2c04 16 old_path = "/usr/local/bin:/usr/bin:/bin";
8e49d503
AE
17
18 path_len = len + strlen(old_path) + 1;
19
2d7320d0 20 path = xmalloc(path_len + 1);
8e49d503
AE
21
22 memcpy(path, dir, len);
23 path[len] = ':';
24 memcpy(path + len + 1, old_path, path_len - len);
25
26 setenv("PATH", path, 1);
0caea90b
CC
27
28 free(path);
8e49d503
AE
29}
30
4394efec 31static int handle_options(const char*** argv, int* argc, int* envchanged)
4ab243a9
JS
32{
33 int handled = 0;
34
35 while (*argc > 0) {
36 const char *cmd = (*argv)[0];
37 if (cmd[0] != '-')
38 break;
39
6acbcb92
JS
40 /*
41 * For legacy reasons, the "version" and "help"
42 * commands can be written with "--" prepended
43 * to make them look like flags.
44 */
45 if (!strcmp(cmd, "--help") || !strcmp(cmd, "--version"))
46 break;
47
48 /*
49 * Check remaining flags.
50 */
cc44c765 51 if (!prefixcmp(cmd, "--exec-path")) {
6acbcb92
JS
52 cmd += 11;
53 if (*cmd == '=')
54 git_set_exec_path(cmd + 1);
55 else {
56 puts(git_exec_path());
57 exit(0);
58 }
59 } else if (!strcmp(cmd, "-p") || !strcmp(cmd, "--paginate")) {
4ab243a9 60 setup_pager();
6acbcb92 61 } else if (!strcmp(cmd, "--git-dir")) {
c321f00d
BG
62 if (*argc < 2) {
63 fprintf(stderr, "No directory given for --git-dir.\n" );
64 usage(git_usage_string);
65 }
45b09794 66 setenv(GIT_DIR_ENVIRONMENT, (*argv)[1], 1);
4394efec
ML
67 if (envchanged)
68 *envchanged = 1;
6acbcb92
JS
69 (*argv)++;
70 (*argc)--;
e4b02333 71 handled++;
cc44c765 72 } else if (!prefixcmp(cmd, "--git-dir=")) {
45b09794 73 setenv(GIT_DIR_ENVIRONMENT, cmd + 10, 1);
4394efec
ML
74 if (envchanged)
75 *envchanged = 1;
892c41b9
ML
76 } else if (!strcmp(cmd, "--work-tree")) {
77 if (*argc < 2) {
78 fprintf(stderr, "No directory given for --work-tree.\n" );
79 usage(git_usage_string);
80 }
81 setenv(GIT_WORK_TREE_ENVIRONMENT, (*argv)[1], 1);
4394efec
ML
82 if (envchanged)
83 *envchanged = 1;
892c41b9
ML
84 (*argv)++;
85 (*argc)--;
86 } else if (!prefixcmp(cmd, "--work-tree=")) {
87 setenv(GIT_WORK_TREE_ENVIRONMENT, cmd + 12, 1);
4394efec
ML
88 if (envchanged)
89 *envchanged = 1;
6acbcb92 90 } else if (!strcmp(cmd, "--bare")) {
ef5ddb2f 91 static char git_dir[PATH_MAX+1];
45b09794 92 setenv(GIT_DIR_ENVIRONMENT, getcwd(git_dir, sizeof(git_dir)), 1);
4394efec
ML
93 if (envchanged)
94 *envchanged = 1;
6acbcb92
JS
95 } else {
96 fprintf(stderr, "Unknown option: %s\n", cmd);
822a7d50 97 usage(git_usage_string);
6acbcb92 98 }
4ab243a9
JS
99
100 (*argv)++;
101 (*argc)--;
102 handled++;
103 }
104 return handled;
105}
106
2b11e317 107static const char *alias_command;
96f1e58f 108static char *alias_string;
2b11e317
JS
109
110static int git_alias_config(const char *var, const char *value)
111{
cc44c765 112 if (!prefixcmp(var, "alias.") && !strcmp(var + 6, alias_command)) {
9befac47 113 alias_string = xstrdup(value);
2b11e317
JS
114 }
115 return 0;
116}
117
118static int split_cmdline(char *cmdline, const char ***argv)
119{
120 int src, dst, count = 0, size = 16;
121 char quoted = 0;
122
3301521a 123 *argv = xmalloc(sizeof(char*) * size);
2b11e317
JS
124
125 /* split alias_string */
126 (*argv)[count++] = cmdline;
127 for (src = dst = 0; cmdline[src];) {
128 char c = cmdline[src];
129 if (!quoted && isspace(c)) {
130 cmdline[dst++] = 0;
131 while (cmdline[++src]
132 && isspace(cmdline[src]))
133 ; /* skip */
134 if (count >= size) {
135 size += 16;
83572c1a 136 *argv = xrealloc(*argv, sizeof(char*) * size);
2b11e317
JS
137 }
138 (*argv)[count++] = cmdline + dst;
139 } else if(!quoted && (c == '\'' || c == '"')) {
140 quoted = c;
141 src++;
142 } else if (c == quoted) {
143 quoted = 0;
144 src++;
145 } else {
146 if (c == '\\' && quoted != '\'') {
147 src++;
148 c = cmdline[src];
149 if (!c) {
150 free(*argv);
151 *argv = NULL;
152 return error("cmdline ends with \\");
153 }
154 }
155 cmdline[dst++] = c;
156 src++;
157 }
158 }
159
160 cmdline[dst] = 0;
161
162 if (quoted) {
163 free(*argv);
164 *argv = NULL;
165 return error("unclosed quote");
166 }
167
168 return count;
169}
170
171static int handle_alias(int *argcp, const char ***argv)
172{
4394efec 173 int nongit = 0, envchanged = 0, ret = 0, saved_errno = errno;
2b11e317 174 const char *subdir;
0347a8c7
ML
175 int count, option_count;
176 const char** new_argv;
2b11e317
JS
177
178 subdir = setup_git_directory_gently(&nongit);
0347a8c7
ML
179
180 alias_command = (*argv)[0];
181 git_config(git_alias_config);
182 if (alias_string) {
dfd42a3c 183 if (alias_string[0] == '!') {
a2f8028d
JS
184 if (*argcp > 1) {
185 int i, sz = PATH_MAX;
186 char *s = xmalloc(sz), *new_alias = s;
187
188 add_to_string(&s, &sz, alias_string, 0);
189 free(alias_string);
190 alias_string = new_alias;
191 for (i = 1; i < *argcp &&
192 !add_to_string(&s, &sz, " ", 0) &&
193 !add_to_string(&s, &sz, (*argv)[i], 1)
194 ; i++)
195 ; /* do nothing */
196 if (!sz)
197 die("Too many or long arguments");
198 }
dfd42a3c
TT
199 trace_printf("trace: alias to shell cmd: %s => %s\n",
200 alias_command, alias_string + 1);
201 ret = system(alias_string + 1);
202 if (ret >= 0 && WIFEXITED(ret) &&
203 WEXITSTATUS(ret) != 127)
204 exit(WEXITSTATUS(ret));
205 die("Failed to run '%s' when expanding alias '%s'\n",
206 alias_string + 1, alias_command);
207 }
0347a8c7 208 count = split_cmdline(alias_string, &new_argv);
4394efec
ML
209 option_count = handle_options(&new_argv, &count, &envchanged);
210 if (envchanged)
211 die("alias '%s' changes environment variables\n"
212 "You can use '!git' in the alias to do this.",
213 alias_command);
0347a8c7
ML
214 memmove(new_argv - option_count, new_argv,
215 count * sizeof(char *));
216 new_argv -= option_count;
217
218 if (count < 1)
219 die("empty alias for %s", alias_command);
220
221 if (!strcmp(alias_command, new_argv[0]))
222 die("recursive alias: %s", alias_command);
223
6ce4e61f
CC
224 trace_argv_printf(new_argv, count,
225 "trace: alias expansion: %s =>",
226 alias_command);
575ba9d6 227
83572c1a
JF
228 new_argv = xrealloc(new_argv, sizeof(char*) *
229 (count + *argcp + 1));
0347a8c7
ML
230 /* insert after command name */
231 memcpy(new_argv + count, *argv + 1, sizeof(char*) * *argcp);
232 new_argv[count+*argcp] = NULL;
2b11e317 233
0347a8c7
ML
234 *argv = new_argv;
235 *argcp += count - 1;
2b11e317 236
0347a8c7 237 ret = 1;
2b11e317
JS
238 }
239
240 if (subdir)
241 chdir(subdir);
242
47e5c0ca
JS
243 errno = saved_errno;
244
2b11e317
JS
245 return ret;
246}
247
70827b15 248const char git_version_string[] = GIT_VERSION;
5b84f4d8 249
efffea03
JH
250#define RUN_SETUP (1<<0)
251#define USE_PAGER (1<<1)
7eff28a9
SP
252/*
253 * require working tree to be present -- anything uses this needs
254 * RUN_SETUP for reading from the configuration file.
255 */
7ae3df8c 256#define NEED_WORK_TREE (1<<2)
a633fca0 257
47d0b4ff
LT
258struct cmd_struct {
259 const char *cmd;
260 int (*fn)(int, const char **, const char *);
261 int option;
262};
263
264static int run_command(struct cmd_struct *p, int argc, const char **argv)
265{
0f157315
LT
266 int status;
267 struct stat st;
47d0b4ff
LT
268 const char *prefix;
269
270 prefix = NULL;
271 if (p->option & RUN_SETUP)
272 prefix = setup_git_directory();
273 if (p->option & USE_PAGER)
274 setup_pager();
e90fdc39
JS
275 if (p->option & NEED_WORK_TREE) {
276 const char *work_tree = get_git_work_tree();
277 const char *git_dir = get_git_dir();
278 if (!is_absolute_path(git_dir))
279 set_git_dir(make_absolute_path(git_dir));
280 if (!work_tree || chdir(work_tree))
281 die("%s must be run in a work tree", p->cmd);
282 }
47d0b4ff
LT
283 trace_argv_printf(argv, argc, "trace: built-in: git");
284
0f157315
LT
285 status = p->fn(argc, argv, prefix);
286 if (status)
287 return status;
288
289 /* Somebody closed stdout? */
290 if (fstat(fileno(stdout), &st))
291 return 0;
292 /* Ignore write errors for pipes and sockets.. */
293 if (S_ISFIFO(st.st_mode) || S_ISSOCK(st.st_mode))
294 return 0;
295
296 /* Check for ENOSPC and EIO errors.. */
0227f988 297 if (fflush(stdout))
0f157315 298 die("write failure on standard output: %s", strerror(errno));
0227f988
LT
299 if (ferror(stdout))
300 die("unknown write failure on standard output");
301 if (fclose(stdout))
302 die("close failed on standard output: %s", strerror(errno));
0f157315 303 return 0;
47d0b4ff
LT
304}
305
306static void handle_internal_command(int argc, const char **argv)
231af832
LT
307{
308 const char *cmd = argv[0];
47d0b4ff 309 static struct cmd_struct commands[] = {
7ae3df8c 310 { "add", cmd_add, RUN_SETUP | NEED_WORK_TREE },
d97bc5de 311 { "annotate", cmd_annotate, RUN_SETUP },
f754fa9c 312 { "apply", cmd_apply },
265d5280 313 { "archive", cmd_archive },
4f0219a4 314 { "blame", cmd_blame, RUN_SETUP },
1da1b3a3 315 { "branch", cmd_branch, RUN_SETUP },
2e0afafe 316 { "bundle", cmd_bundle },
efffea03
JH
317 { "cat-file", cmd_cat_file, RUN_SETUP },
318 { "checkout-index", cmd_checkout_index, RUN_SETUP },
0864f264 319 { "check-ref-format", cmd_check_ref_format },
7ae3df8c 320 { "check-attr", cmd_check_attr, RUN_SETUP | NEED_WORK_TREE },
e827633a 321 { "cherry", cmd_cherry, RUN_SETUP },
7ae3df8c 322 { "cherry-pick", cmd_cherry_pick, RUN_SETUP | NEED_WORK_TREE },
efffea03 323 { "commit-tree", cmd_commit_tree, RUN_SETUP },
e0d10e1c 324 { "config", cmd_config },
8112894d 325 { "count-objects", cmd_count_objects, RUN_SETUP },
9a0eaf83 326 { "describe", cmd_describe, RUN_SETUP },
d516c2d1
JS
327 { "diff", cmd_diff, USE_PAGER },
328 { "diff-files", cmd_diff_files },
efffea03 329 { "diff-index", cmd_diff_index, RUN_SETUP },
efffea03 330 { "diff-tree", cmd_diff_tree, RUN_SETUP },
d4289fff 331 { "fetch--tool", cmd_fetch__tool, RUN_SETUP },
efffea03 332 { "fmt-merge-msg", cmd_fmt_merge_msg, RUN_SETUP },
9f613ddd 333 { "for-each-ref", cmd_for_each_ref, RUN_SETUP },
efffea03 334 { "format-patch", cmd_format_patch, RUN_SETUP },
b4dfefe0
MW
335 { "fsck", cmd_fsck, RUN_SETUP },
336 { "fsck-objects", cmd_fsck, RUN_SETUP },
6757ada4 337 { "gc", cmd_gc, RUN_SETUP },
f754fa9c 338 { "get-tar-commit-id", cmd_get_tar_commit_id },
34c6a82b 339 { "grep", cmd_grep, RUN_SETUP | USE_PAGER },
f754fa9c 340 { "help", cmd_help },
515377ea 341 { "init", cmd_init_db },
f754fa9c 342 { "init-db", cmd_init_db },
efffea03
JH
343 { "log", cmd_log, RUN_SETUP | USE_PAGER },
344 { "ls-files", cmd_ls_files, RUN_SETUP },
345 { "ls-tree", cmd_ls_tree, RUN_SETUP },
f754fa9c
JH
346 { "mailinfo", cmd_mailinfo },
347 { "mailsplit", cmd_mailsplit },
71dfbf22 348 { "merge-base", cmd_merge_base, RUN_SETUP },
ba1f5f35 349 { "merge-file", cmd_merge_file },
7ae3df8c 350 { "mv", cmd_mv, RUN_SETUP | NEED_WORK_TREE },
efffea03
JH
351 { "name-rev", cmd_name_rev, RUN_SETUP },
352 { "pack-objects", cmd_pack_objects, RUN_SETUP },
d97bc5de 353 { "pickaxe", cmd_blame, RUN_SETUP },
efffea03
JH
354 { "prune", cmd_prune, RUN_SETUP },
355 { "prune-packed", cmd_prune_packed, RUN_SETUP },
102cb085 356 { "push", cmd_push, RUN_SETUP },
efffea03 357 { "read-tree", cmd_read_tree, RUN_SETUP },
4264dc15 358 { "reflog", cmd_reflog, RUN_SETUP },
e0d10e1c 359 { "repo-config", cmd_config },
658f3650 360 { "rerere", cmd_rerere, RUN_SETUP },
efffea03
JH
361 { "rev-list", cmd_rev_list, RUN_SETUP },
362 { "rev-parse", cmd_rev_parse, RUN_SETUP },
7ae3df8c
ML
363 { "revert", cmd_revert, RUN_SETUP | NEED_WORK_TREE },
364 { "rm", cmd_rm, RUN_SETUP | NEED_WORK_TREE },
365 { "runstatus", cmd_runstatus, RUN_SETUP | NEED_WORK_TREE },
1d541c12 366 { "shortlog", cmd_shortlog, RUN_SETUP | USE_PAGER },
efffea03
JH
367 { "show-branch", cmd_show_branch, RUN_SETUP },
368 { "show", cmd_show, RUN_SETUP | USE_PAGER },
f754fa9c 369 { "stripspace", cmd_stripspace },
efffea03 370 { "symbolic-ref", cmd_symbolic_ref, RUN_SETUP },
9cb90b80 371 { "tar-tree", cmd_tar_tree },
efffea03
JH
372 { "unpack-objects", cmd_unpack_objects, RUN_SETUP },
373 { "update-index", cmd_update_index, RUN_SETUP },
374 { "update-ref", cmd_update_ref, RUN_SETUP },
39345a21 375 { "upload-archive", cmd_upload_archive },
f754fa9c 376 { "version", cmd_version },
efffea03
JH
377 { "whatchanged", cmd_whatchanged, RUN_SETUP | USE_PAGER },
378 { "write-tree", cmd_write_tree, RUN_SETUP },
2e3ed670 379 { "verify-pack", cmd_verify_pack },
358ddb62 380 { "show-ref", cmd_show_ref, RUN_SETUP },
e1e22e37 381 { "pack-refs", cmd_pack_refs, RUN_SETUP },
231af832
LT
382 };
383 int i;
384
1cd95087
LT
385 /* Turn "git cmd --help" into "git help cmd" */
386 if (argc > 1 && !strcmp(argv[1], "--help")) {
387 argv[1] = argv[0];
388 argv[0] = cmd = "help";
389 }
390
231af832
LT
391 for (i = 0; i < ARRAY_SIZE(commands); i++) {
392 struct cmd_struct *p = commands+i;
393 if (strcmp(p->cmd, cmd))
394 continue;
47d0b4ff 395 exit(run_command(p, argc, argv));
231af832
LT
396 }
397}
398
47d0b4ff 399int main(int argc, const char **argv)
8e49d503 400{
5b6df8e4 401 const char *cmd = argv[0] ? argv[0] : "git-help";
231af832 402 char *slash = strrchr(cmd, '/');
231af832 403 const char *exec_path = NULL;
a025463b 404 int done_alias = 0;
231af832
LT
405
406 /*
407 * Take the basename of argv[0] as the command
408 * name, and the dirname as the default exec_path
409 * if it's an absolute path and we don't have
410 * anything better.
411 */
412 if (slash) {
413 *slash++ = 0;
414 if (*cmd == '/')
415 exec_path = cmd;
416 cmd = slash;
417 }
8e49d503 418
231af832
LT
419 /*
420 * "git-xxxx" is the same as "git xxxx", but we obviously:
421 *
422 * - cannot take flags in between the "git" and the "xxxx".
423 * - cannot execute it externally (since it would just do
424 * the same thing over again)
425 *
426 * So we just directly call the internal command handler, and
427 * die if that one cannot handle it.
428 */
cc44c765 429 if (!prefixcmp(cmd, "git-")) {
231af832
LT
430 cmd += 4;
431 argv[0] = cmd;
47d0b4ff 432 handle_internal_command(argc, argv);
231af832
LT
433 die("cannot handle %s internally", cmd);
434 }
8e49d503 435
231af832 436 /* Look for flags.. */
6acbcb92
JS
437 argv++;
438 argc--;
4394efec 439 handle_options(&argv, &argc, NULL);
6acbcb92 440 if (argc > 0) {
cc44c765 441 if (!prefixcmp(argv[0], "--"))
6acbcb92
JS
442 argv[0] += 2;
443 } else {
444 /* Default command: "help" */
445 argv[0] = "help";
446 argc = 1;
97fc6c5f 447 }
6acbcb92 448 cmd = argv[0];
231af832
LT
449
450 /*
cf32190a
JH
451 * We execute external git command via execv_git_cmd(),
452 * which looks at "--exec-path" option, GIT_EXEC_PATH
453 * environment, and $(gitexecdir) in Makefile while built,
454 * in this order. For scripted commands, we prepend
455 * the value of the exec_path variable to the PATH.
231af832
LT
456 */
457 if (exec_path)
458 prepend_to_path(exec_path, strlen(exec_path));
77cb17e9
MO
459 exec_path = git_exec_path();
460 prepend_to_path(exec_path, strlen(exec_path));
8e49d503 461
a025463b
JH
462 while (1) {
463 /* See if it's an internal command */
47d0b4ff 464 handle_internal_command(argc, argv);
2b11e317 465
a025463b
JH
466 /* .. then try the external ones */
467 execv_git_cmd(argv);
231af832 468
a025463b
JH
469 /* It could be an alias -- this works around the insanity
470 * of overriding "git log" with "git show" by having
471 * alias.log = show
472 */
473 if (done_alias || !handle_alias(&argc, &argv))
474 break;
475 done_alias = 1;
476 }
10b15b86 477
f3dd015c
TT
478 if (errno == ENOENT) {
479 if (done_alias) {
480 fprintf(stderr, "Expansion of alias '%s' failed; "
481 "'%s' is not a git-command\n",
482 cmd, argv[0]);
483 exit(1);
484 }
822a7d50 485 help_unknown_cmd(cmd);
f3dd015c 486 }
10b15b86
AR
487
488 fprintf(stderr, "Failed to run command '%s': %s\n",
33ebb871 489 cmd, strerror(errno));
8e49d503
AE
490
491 return 1;
492}