aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--driver.c7
1 files changed, 3 insertions, 4 deletions
diff --git a/driver.c b/driver.c
index fee1954..0ff8451 100644
--- a/driver.c
+++ b/driver.c
@@ -328,12 +328,11 @@ main(int argc, char *argv[])
arrayaddptr(&phases[LINK].cmd, arg);
} else if (strcmp(arg, "-emit-qbe") == 0) {
last = COMPILE;
- } else if (strcmp(arg, "-include") == 0) {
- --argc, arg = *++argv;
- if (!arg)
+ } else if (strcmp(arg, "-include") == 0 || strcmp(arg, "-idirafter") == 0) {
+ if (!--argc)
usage();
- arrayaddptr(&phases[PREPROCESS].cmd, "-include");
arrayaddptr(&phases[PREPROCESS].cmd, arg);
+ arrayaddptr(&phases[PREPROCESS].cmd, *++argv);
} else if (strcmp(arg, "-pipe") == 0) {
/* ignore */
} else {