6 * Permission is hereby granted, free of charge, to any person obtaining a copy of
7 * this software and associated documentation files (the "Software"), to deal in
8 * the Software without restriction, including without limitation the rights to
9 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
10 * of the Software, and to permit persons to whom the Software is furnished to do
11 * so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
27 uint32_t opts_flags[1 + (COUNT_FLAGS / 32)];
28 uint32_t opts_warn [1 + (COUNT_WARNINGS / 32)];
31 const char *opts_output = "progs.dat";
32 int opts_standard = COMPILER_GMQCC;
33 bool opts_debug = false;
34 bool opts_memchk = false;
35 bool opts_dumpfin = false;
36 bool opts_dump = false;
37 bool opts_werror = false;
38 bool opts_forcecrc = false;
39 bool opts_pp_only = false;
40 size_t opts_max_array_size = 1024;
42 uint16_t opts_forced_crc;
44 static bool opts_output_wasset = false;
46 /* set by the standard */
47 const oper_info *operators = NULL;
48 size_t operator_count = 0;
50 typedef struct { char *filename; int type; } argitem;
51 static argitem *items = NULL;
57 static const char *app_name;
60 con_out("usage: %s [options] [files...]", app_name);
62 " -h, --help show this help message\n"
63 " -debug turns on compiler debug messages\n"
64 " -memchk turns on compiler memory leak check\n");
65 con_out(" -o, --output=file output file, defaults to progs.dat\n"
66 " -a filename add an asm file to be assembled\n"
67 " -s filename add a progs.src file to be used\n");
68 con_out(" -E stop after preprocessing\n");
69 con_out(" -f<flag> enable a flag\n"
70 " -fno-<flag> disable a flag\n"
71 " -std standard select one of the following standards\n"
72 " -std=qcc original QuakeC\n"
73 " -std=fteqcc fteqcc QuakeC\n"
74 " -std=gmqcc this compiler (default)\n");
75 con_out(" -W<warning> enable a warning\n"
76 " -Wno-<warning> disable a warning\n"
77 " -Wall enable all warnings\n"
78 " -Werror treat warnings as errors\n");
79 con_out(" -force-crc=num force a specific checksum into the header\n");
82 " -fadjust-vector-fields\n"
83 " when assigning a vector field, its _y and _z fields also get assigned\n"
88 static bool options_setflag_all(const char *name, bool on, uint32_t *flags, const opts_flag_def *list, size_t listsize) {
91 for (i = 0; i < listsize; ++i) {
92 if (!strcmp(name, list[i].name)) {
93 longbit lb = list[i].bit;
96 flags[lb.idx] |= (1<<(lb.bit));
98 flags[lb.idx] &= ~(1<<(lb.bit));
103 flags[0] &= ~(1<<(lb));
110 static bool options_setflag(const char *name, bool on) {
111 return options_setflag_all(name, on, opts_flags, opts_flag_list, COUNT_FLAGS);
113 static bool options_setwarn(const char *name, bool on) {
114 return options_setflag_all(name, on, opts_warn, opts_warn_list, COUNT_WARNINGS);
117 static bool options_witharg(int *argc_, char ***argv_, char **out) {
119 char **argv = *argv_;
125 /* eat up the next */
126 if (argc < 2) /* no parameter was provided */
135 static bool options_long_witharg_all(const char *optname, int *argc_, char ***argv_, char **out, int ds, bool split) {
137 char **argv = *argv_;
139 size_t len = strlen(optname);
141 if (strncmp(argv[0]+ds, optname, len))
144 /* it's --optname, check how the parameter is supplied */
145 if (argv[0][ds+len] == '=') {
146 /* using --opt=param */
147 *out = argv[0]+ds+len+1;
151 if (!split || argc < ds) /* no parameter was provided, or only single-arg form accepted */
154 /* using --opt param */
160 static bool options_long_witharg(const char *optname, int *argc_, char ***argv_, char **out) {
161 return options_long_witharg_all(optname, argc_, argv_, out, 2, true);
163 static bool options_long_gcc(const char *optname, int *argc_, char ***argv_, char **out) {
164 return options_long_witharg_all(optname, argc_, argv_, out, 1, false);
167 static void options_set(uint32_t *flags, size_t idx, bool on)
169 longbit lb = LONGBIT(idx);
172 flags[lb.idx] |= (1<<(lb.bit));
174 flags[lb.idx] &= ~(1<<(lb.bit));
177 flags[0] |= (1<<(lb));
179 flags[0] &= ~(1<<(lb));
183 static bool options_parse(int argc, char **argv) {
187 char *redirout = (char*)stdout;
188 char *redirerr = (char*)stderr;
190 while (!argend && argc > 1) {
197 if (argv[0][0] == '-') {
198 /* All gcc-type long options */
199 if (options_long_gcc("std", &argc, &argv, &argarg)) {
200 if (!strcmp(argarg, "gmqcc") || !strcmp(argarg, "default")) {
201 options_set(opts_flags, ADJUST_VECTOR_FIELDS, true);
202 opts_standard = COMPILER_GMQCC;
203 } else if (!strcmp(argarg, "qcc")) {
204 options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
205 opts_standard = COMPILER_QCC;
206 } else if (!strcmp(argarg, "fte") || !strcmp(argarg, "fteqcc")) {
207 options_set(opts_flags, FTEPP, true);
208 options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
209 opts_standard = COMPILER_FTEQCC;
210 } else if (!strcmp(argarg, "qccx")) {
211 options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
212 opts_standard = COMPILER_QCCX;
214 con_out("Unknown standard: %s\n", argarg);
219 if (options_long_gcc("force-crc", &argc, &argv, &argarg)) {
220 opts_forcecrc = true;
221 opts_forced_crc = strtol(argarg, NULL, 0);
224 if (options_long_gcc("redirout", &argc, &argv, &redirout)) {
225 con_change(redirout, redirerr);
228 if (options_long_gcc("redirerr", &argc, &argv, &redirerr)) {
229 con_change(redirout, redirerr);
233 /* show defaults (like pathscale) */
234 if (!strcmp(argv[0]+1, "show-defaults")) {
237 for (itr = 0; itr < COUNT_FLAGS; ++itr) {
241 memset(buffer, 0, sizeof(buffer));
242 util_strtononcmd(opts_flag_list[itr].name, buffer, strlen(opts_flag_list[itr].name) + 1);
244 con_out("-f%s ", buffer);
246 for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
250 memset(buffer, 0, sizeof(buffer));
251 util_strtononcmd(opts_warn_list[itr].name, buffer, strlen(opts_warn_list[itr].name) + 1);
252 con_out("-W%s ", buffer);
258 if (!strcmp(argv[0]+1, "debug")) {
262 if (!strcmp(argv[0]+1, "dump")) {
266 if (!strcmp(argv[0]+1, "dumpfin")) {
270 if (!strcmp(argv[0]+1, "memchk")) {
274 if (!strcmp(argv[0]+1, "nocolor")) {
279 switch (argv[0][1]) {
280 /* -h, show usage but exit with 0 */
284 /* break; never reached because of exit(0) */
290 /* handle all -fflags */
292 util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
293 if (!strcmp(argv[0]+2, "HELP") || *(argv[0]+2) == '?') {
294 con_out("Possible flags:\n");
295 for (itr = 0; itr < COUNT_FLAGS; ++itr) {
296 util_strtononcmd(opts_flag_list[itr].name, buffer, sizeof(buffer));
297 con_out(" -f%s\n", buffer);
301 else if (!strncmp(argv[0]+2, "NO_", 3)) {
302 if (!options_setflag(argv[0]+5, false)) {
303 con_out("unknown flag: %s\n", argv[0]+2);
307 else if (!options_setflag(argv[0]+2, true)) {
308 con_out("unknown flag: %s\n", argv[0]+2);
313 util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
314 if (!strcmp(argv[0]+2, "HELP") || *(argv[0]+2) == '?') {
315 con_out("Possible warnings:\n");
316 for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
317 util_strtononcmd(opts_warn_list[itr].name, buffer, sizeof(buffer));
318 con_out(" -W%s\n", buffer);
322 else if (!strcmp(argv[0]+2, "NO_ERROR")) {
326 else if (!strcmp(argv[0]+2, "ERROR")) {
330 else if (!strcmp(argv[0]+2, "NONE")) {
331 for (itr = 0; itr < sizeof(opts_warn)/sizeof(opts_warn[0]); ++itr)
335 else if (!strcmp(argv[0]+2, "ALL")) {
336 for (itr = 0; itr < sizeof(opts_warn)/sizeof(opts_warn[0]); ++itr)
337 opts_warn[itr] = 0xFFFFFFFFL;
340 if (!strncmp(argv[0]+2, "NO_", 3)) {
341 if (!options_setwarn(argv[0]+5, false)) {
342 con_out("unknown warning: %s\n", argv[0]+2);
346 else if (!options_setwarn(argv[0]+2, true)) {
347 con_out("unknown warning: %s\n", argv[0]+2);
353 if (!options_witharg(&argc, &argv, &argarg)) {
354 con_out("option -O requires a numerical argument\n");
357 opts_O = atoi(argarg);
361 if (!options_witharg(&argc, &argv, &argarg)) {
362 con_out("option -o requires an argument: the output file name\n");
365 opts_output = argarg;
366 opts_output_wasset = true;
371 item.type = argv[0][1] == 'a' ? TYPE_ASM : TYPE_SRC;
372 if (!options_witharg(&argc, &argv, &argarg)) {
373 con_out("option -a requires a filename %s\n",
374 (argv[0][1] == 'a' ? "containing QC-asm" : "containing a progs.src formatted list"));
377 item.filename = argarg;
378 vec_push(items, item);
383 /* anything following -- is considered a non-option argument */
387 /* All long options without arguments */
388 else if (!strcmp(argv[0]+2, "help")) {
393 /* All long options with arguments */
394 if (options_long_witharg("output", &argc, &argv, &argarg)) {
395 opts_output = argarg;
396 opts_output_wasset = true;
398 con_out("Unknown parameter: %s\n", argv[0]);
405 con_out("Unknown parameter: %s\n", argv[0]);
411 /* it's a QC filename */
412 item.filename = argv[0];
414 vec_push(items, item);
420 /* returns the line number, or -1 on error */
421 static bool progs_nextline(char **out, size_t *alen,FILE *src) {
428 len = util_getline(&line, alen, src);
432 /* start at first non-blank */
433 for (start = line; isspace(*start); ++start) {}
434 /* end at the first non-blank */
435 for (end = start; *end && !isspace(*end); ++end) {}
438 /* move the actual filename to the beginning */
439 while (start != end) {
446 int main(int argc, char **argv) {
449 bool opts_output_free = false;
450 bool progs_src = false;
451 FILE *outfile = NULL;
456 /* default options / warn flags */
457 options_set(opts_warn, WARN_UNKNOWN_CONTROL_SEQUENCE, true);
458 options_set(opts_warn, WARN_EXTENSIONS, true);
459 options_set(opts_warn, WARN_FIELD_REDECLARED, true);
460 options_set(opts_warn, WARN_TOO_FEW_PARAMETERS, true);
461 options_set(opts_warn, WARN_MISSING_RETURN_VALUES, true);
462 options_set(opts_warn, WARN_USED_UNINITIALIZED, true);
463 options_set(opts_warn, WARN_LOCAL_CONSTANTS, true);
464 options_set(opts_warn, WARN_VOID_VARIABLES, true);
465 options_set(opts_warn, WARN_IMPLICIT_FUNCTION_POINTER, true);
466 options_set(opts_warn, WARN_VARIADIC_FUNCTION, true);
467 options_set(opts_warn, WARN_FRAME_MACROS, true);
468 options_set(opts_warn, WARN_UNUSED_VARIABLE, true);
469 options_set(opts_warn, WARN_EFFECTLESS_STATEMENT, true);
470 options_set(opts_warn, WARN_END_SYS_FIELDS, true);
471 options_set(opts_warn, WARN_ASSIGN_FUNCTION_TYPES, true);
472 options_set(opts_warn, WARN_PREPROCESSOR, true);
473 options_set(opts_warn, WARN_MULTIFILE_IF, true);
474 options_set(opts_warn, WARN_DOUBLE_DECLARATION, true);
476 options_set(opts_flags, ADJUST_VECTOR_FIELDS, true);
477 options_set(opts_flags, FTEPP, false);
479 if (!options_parse(argc, argv)) {
483 /* the standard decides which set of operators to use */
484 if (opts_standard == COMPILER_GMQCC) {
485 operators = c_operators;
486 operator_count = c_operator_count;
487 } else if (opts_standard == COMPILER_FTEQCC) {
488 operators = fte_operators;
489 operator_count = fte_operator_count;
491 operators = qcc_operators;
492 operator_count = qcc_operator_count;
496 for (itr = 0; itr < COUNT_FLAGS; ++itr) {
497 con_out("Flag %s = %i\n", opts_flag_list[itr].name, OPTS_FLAG(itr));
499 for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
500 con_out("Warning %s = %i\n", opts_warn_list[itr].name, OPTS_WARN(itr));
502 con_out("output = %s\n", opts_output);
503 con_out("optimization level = %i\n", (int)opts_O);
504 con_out("standard = %i\n", opts_standard);
508 if (opts_output_wasset) {
509 outfile = util_fopen(opts_output, "wb");
511 con_err("failed to open `%s` for writing\n", opts_output);
521 if (!parser_init()) {
522 con_err("failed to initialize parser\n");
527 if (opts_pp_only || OPTS_FLAG(FTEPP)) {
529 con_err("failed to initialize parser\n");
535 util_debug("COM", "starting ...\n");
537 if (!vec_size(items)) {
544 src = util_fopen("progs.src", "rb");
546 con_err("failed to open `progs.src` for reading\n");
552 if (!progs_nextline(&line, &linelen, src) || !line[0]) {
553 con_err("illformatted progs.src file: expected output filename in first line\n");
558 if (!opts_output_wasset) {
559 opts_output = util_strdup(line);
560 opts_output_free = true;
563 while (progs_nextline(&line, &linelen, src)) {
565 if (!line[0] || (line[0] == '/' && line[1] == '/'))
567 item.filename = util_strdup(line);
569 vec_push(items, item);
580 if (vec_size(items)) {
582 con_out("Mode: %s\n", (progs_src ? "progs.src" : "manual"));
583 con_out("There are %lu items to compile:\n", (unsigned long)vec_size(items));
585 for (itr = 0; itr < vec_size(items); ++itr) {
587 con_out(" item: %s (%s)\n",
589 ( (items[itr].type == TYPE_QC ? "qc" :
590 (items[itr].type == TYPE_ASM ? "asm" :
591 (items[itr].type == TYPE_SRC ? "progs.src" :
597 if (!ftepp_preprocess_file(items[itr].filename)) {
603 fprintf(outfile, "%s", out);
607 if (OPTS_FLAG(FTEPP)) {
609 if (!ftepp_preprocess_file(items[itr].filename)) {
614 if (!parser_compile_string_len(items[itr].filename, data, vec_size(data)-1)) {
621 if (!parser_compile_file(items[itr].filename)) {
629 mem_d(items[itr].filename);
630 items[itr].filename = NULL;
636 if (!parser_finish(opts_output)) {
646 util_debug("COM", "cleaning ...\n");
653 if (opts_output_free)
654 mem_d((char*)opts_output);