Added test for variadic arguments
[xonotic/gmqcc.git] / main.c
1 /*
2  * Copyright (C) 2012
3  *     Dale Weiler
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a copy of
6  * this software and associated documentation files (the "Software"), to deal in
7  * the Software without restriction, including without limitation the rights to
8  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
9  * of the Software, and to permit persons to whom the Software is furnished to do
10  * so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in all
13  * copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21  * SOFTWARE.
22  */
23 #include "gmqcc.h"
24 #include "lexer.h"
25
26 uint32_t    opts_flags[1 + (COUNT_FLAGS / 32)];
27 uint32_t    opts_warn [1 + (COUNT_WARNINGS / 32)];
28
29 uint32_t    opts_O        = 1;
30 const char *opts_output   = "progs.dat";
31 int         opts_standard = COMPILER_GMQCC;
32 bool        opts_debug    = false;
33 bool        opts_memchk   = false;
34 bool        opts_dump     = false;
35 bool        opts_werror   = false;
36 bool        opts_forcecrc = false;
37 bool        opts_pp_only  = false;
38 size_t      opts_max_array_size = 1024;
39
40 uint16_t    opts_forced_crc;
41
42 static bool opts_output_wasset = false;
43
44 /* set by the standard */
45 const oper_info *operators      = NULL;
46 size_t           operator_count = 0;
47
48 typedef struct { char *filename; int type; } argitem;
49 static argitem *items = NULL;
50
51 #define TYPE_QC  0
52 #define TYPE_ASM 1
53 #define TYPE_SRC 2
54
55 static const char *app_name;
56
57 static int usage() {
58     con_out("usage: %s [options] [files...]", app_name);
59     con_out("options:\n"
60            "  -h, --help             show this help message\n"
61            "  -debug                 turns on compiler debug messages\n"
62            "  -memchk                turns on compiler memory leak check\n");
63     con_out("  -o, --output=file      output file, defaults to progs.dat\n"
64            "  -a filename            add an asm file to be assembled\n"
65            "  -s filename            add a progs.src file to be used\n");
66     con_out("  -E                     stop after preprocessing\n");
67     con_out("  -f<flag>               enable a flag\n"
68            "  -fno-<flag>            disable a flag\n"
69            "  -std standard          select one of the following standards\n"
70            "       -std=qcc          original QuakeC\n"
71            "       -std=fteqcc       fteqcc QuakeC\n"
72            "       -std=gmqcc        this compiler (default)\n");
73     con_out("  -W<warning>            enable a warning\n"
74            "  -Wno-<warning>         disable a warning\n"
75            "  -Wall                  enable all warnings\n"
76            "  -Werror                treat warnings as errors\n");
77     con_out("  -force-crc=num         force a specific checksum into the header\n");
78     con_out("\n");
79     con_out("flags:\n"
80            "  -fadjust-vector-fields\n"
81            "            when assigning a vector field, its _y and _z fields also get assigned\n"
82            );
83     return -1;
84 }
85
86 static bool options_setflag_all(const char *name, bool on, uint32_t *flags, const opts_flag_def *list, size_t listsize) {
87     size_t i;
88
89     for (i = 0; i < listsize; ++i) {
90         if (!strcmp(name, list[i].name)) {
91             longbit lb = list[i].bit;
92 #if 0
93             if (on)
94                 flags[lb.idx] |= (1<<(lb.bit));
95             else
96                 flags[lb.idx] &= ~(1<<(lb.bit));
97 #else
98             if (on)
99                 flags[0] |= (1<<lb);
100             else
101                 flags[0] &= ~(1<<(lb));
102 #endif
103             return true;
104         }
105     }
106     return false;
107 }
108 static bool options_setflag(const char *name, bool on) {
109     return options_setflag_all(name, on, opts_flags, opts_flag_list, COUNT_FLAGS);
110 }
111 static bool options_setwarn(const char *name, bool on) {
112     return options_setflag_all(name, on, opts_warn, opts_warn_list, COUNT_WARNINGS);
113 }
114
115 static bool options_witharg(int *argc_, char ***argv_, char **out) {
116     int  argc   = *argc_;
117     char **argv = *argv_;
118
119     if (argv[0][2]) {
120         *out = argv[0]+2;
121         return true;
122     }
123     /* eat up the next */
124     if (argc < 2) /* no parameter was provided */
125         return false;
126
127     *out = argv[1];
128     --*argc_;
129     ++*argv_;
130     return true;
131 }
132
133 static bool options_long_witharg_all(const char *optname, int *argc_, char ***argv_, char **out, int ds, bool split) {
134     int  argc   = *argc_;
135     char **argv = *argv_;
136
137     size_t len = strlen(optname);
138
139     if (strncmp(argv[0]+ds, optname, len))
140         return false;
141
142     /* it's --optname, check how the parameter is supplied */
143     if (argv[0][ds+len] == '=') {
144         /* using --opt=param */
145         *out = argv[0]+ds+len+1;
146         return true;
147     }
148
149     if (!split || argc < ds) /* no parameter was provided, or only single-arg form accepted */
150         return false;
151
152     /* using --opt param */
153     *out = argv[1];
154     --*argc_;
155     ++*argv_;
156     return true;
157 }
158 static bool options_long_witharg(const char *optname, int *argc_, char ***argv_, char **out) {
159     return options_long_witharg_all(optname, argc_, argv_, out, 2, true);
160 }
161 static bool options_long_gcc(const char *optname, int *argc_, char ***argv_, char **out) {
162     return options_long_witharg_all(optname, argc_, argv_, out, 1, false);
163 }
164
165 static void options_set(uint32_t *flags, size_t idx, bool on)
166 {
167     longbit lb = LONGBIT(idx);
168 #if 0
169     if (on)
170         flags[lb.idx] |= (1<<(lb.bit));
171     else
172         flags[lb.idx] &= ~(1<<(lb.bit));
173 #else
174     if (on)
175         flags[0] |= (1<<(lb));
176     else
177         flags[0] &= ~(1<<(lb));
178 #endif
179 }
180
181 static bool options_parse(int argc, char **argv) {
182     bool argend = false;
183     size_t itr;
184     char  buffer[1024];
185     char *redirout = (char*)stdout;
186     char *redirerr = (char*)stderr;
187
188     while (!argend && argc > 1) {
189         char *argarg;
190         argitem item;
191
192         ++argv;
193         --argc;
194
195         if (argv[0][0] == '-') {
196     /* All gcc-type long options */
197             if (options_long_gcc("std", &argc, &argv, &argarg)) {
198                 if      (!strcmp(argarg, "gmqcc") || !strcmp(argarg, "default")) {
199                     options_set(opts_flags, ADJUST_VECTOR_FIELDS, true);
200                     opts_standard = COMPILER_GMQCC;
201                 } else if (!strcmp(argarg, "qcc")) {
202                     options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
203                     opts_standard = COMPILER_QCC;
204                 } else if (!strcmp(argarg, "fte") || !strcmp(argarg, "fteqcc")) {
205                     options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
206                     opts_standard = COMPILER_FTEQCC;
207                 } else if (!strcmp(argarg, "qccx")) {
208                     options_set(opts_flags, ADJUST_VECTOR_FIELDS, false);
209                     opts_standard = COMPILER_QCCX;
210                 } else {
211                     con_out("Unknown standard: %s\n", argarg);
212                     return false;
213                 }
214                 continue;
215             }
216             if (options_long_gcc("force-crc", &argc, &argv, &argarg)) {
217                 opts_forcecrc = true;
218                 opts_forced_crc = strtol(argarg, NULL, 0);
219                 continue;
220             }
221             if (options_long_gcc("redirout", &argc, &argv, &redirout)) {
222                 continue;
223             }
224             if (options_long_gcc("redirerr", &argc, &argv, &redirerr)) {
225                 continue;
226             }
227             
228             con_change(redirout, redirerr);
229
230             if (!strcmp(argv[0]+1, "debug")) {
231                 opts_debug = true;
232                 continue;
233             }
234             if (!strcmp(argv[0]+1, "dump")) {
235                 opts_dump = true;
236                 continue;
237             }
238             if (!strcmp(argv[0]+1, "memchk")) {
239                 opts_memchk = true;
240                 continue;
241             }
242             if (!strcmp(argv[0]+1, "nocolor")) {
243                 con_color(0);
244                 continue;
245             }
246
247             switch (argv[0][1]) {
248                 /* -h, show usage but exit with 0 */
249                 case 'h':
250                     usage();
251                     exit(0);
252                     break;
253
254                 case 'E':
255                     opts_pp_only = true;
256                     break;
257
258                 /* handle all -fflags */
259                 case 'f':
260                     util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
261                     if (!strcmp(argv[0]+2, "HELP")) {
262                         con_out("Possible flags:\n");
263                         for (itr = 0; itr < COUNT_FLAGS; ++itr) {
264                             util_strtononcmd(opts_flag_list[itr].name, buffer, sizeof(buffer));
265                             con_out(" -f%s\n", buffer);
266                         }
267                         exit(0);
268                     }
269                     else if (!strncmp(argv[0]+2, "NO_", 3)) {
270                         if (!options_setflag(argv[0]+5, false)) {
271                             con_out("unknown flag: %s\n", argv[0]+2);
272                             return false;
273                         }
274                     }
275                     else if (!options_setflag(argv[0]+2, true)) {
276                         con_out("unknown flag: %s\n", argv[0]+2);
277                         return false;
278                     }
279                     break;
280                 case 'W':
281                     util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
282                     if (!strcmp(argv[0]+2, "HELP")) {
283                         con_out("Possible warnings:\n");
284                         for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
285                             util_strtononcmd(opts_warn_list[itr].name, buffer, sizeof(buffer));
286                             con_out(" -W%s\n", buffer);
287                         }
288                         exit(0);
289                     }
290                     else if (!strcmp(argv[0]+2, "NO_ERROR")) {
291                         opts_werror = false;
292                         break;
293                     }
294                     else if (!strcmp(argv[0]+2, "ERROR")) {
295                         opts_werror = true;
296                         break;
297                     }
298                     else if (!strcmp(argv[0]+2, "NONE")) {
299                         for (itr = 0; itr < sizeof(opts_warn)/sizeof(opts_warn[0]); ++itr)
300                             opts_warn[itr] = 0;
301                         break;
302                     }
303                     else if (!strcmp(argv[0]+2, "ALL")) {
304                         for (itr = 0; itr < sizeof(opts_warn)/sizeof(opts_warn[0]); ++itr)
305                             opts_warn[itr] = 0xFFFFFFFFL;
306                         break;
307                     }
308                     if (!strncmp(argv[0]+2, "NO_", 3)) {
309                         if (!options_setwarn(argv[0]+5, false)) {
310                             con_out("unknown warning: %s\n", argv[0]+2);
311                             return false;
312                         }
313                     }
314                     else if (!options_setwarn(argv[0]+2, true)) {
315                         con_out("unknown warning: %s\n", argv[0]+2);
316                         return false;
317                     }
318                     break;
319
320                 case 'O':
321                     if (!options_witharg(&argc, &argv, &argarg)) {
322                         con_out("option -O requires a numerical argument\n");
323                         return false;
324                     }
325                     opts_O = atoi(argarg);
326                     break;
327
328                 case 'o':
329                     if (!options_witharg(&argc, &argv, &argarg)) {
330                         con_out("option -o requires an argument: the output file name\n");
331                         return false;
332                     }
333                     opts_output = argarg;
334                     opts_output_wasset = true;
335                     break;
336
337                 case 'a':
338                 case 's':
339                     item.type = argv[0][1] == 'a' ? TYPE_ASM : TYPE_SRC;
340                     if (!options_witharg(&argc, &argv, &argarg)) {
341                         con_out("option -a requires a filename %s\n",
342                                 (argv[0][1] == 'a' ? "containing QC-asm" : "containing a progs.src formatted list"));
343                         return false;
344                     }
345                     item.filename = argarg;
346                     vec_push(items, item);
347                     break;
348
349                 case '-':
350                     if (!argv[0][2]) {
351                         /* anything following -- is considered a non-option argument */
352                         argend = true;
353                         break;
354                     }
355             /* All long options without arguments */
356                     else if (!strcmp(argv[0]+2, "help")) {
357                         usage();
358                         exit(0);
359                     }
360                     else {
361             /* All long options with arguments */
362                         if (options_long_witharg("output", &argc, &argv, &argarg)) {
363                             opts_output = argarg;
364                             opts_output_wasset = true;
365                         } else {
366                             con_out("Unknown parameter: %s\n", argv[0]);
367                             return false;
368                         }
369                     }
370                     break;
371
372                 default:
373                     con_out("Unknown parameter: %s\n", argv[0]);
374                     return false;
375             }
376         }
377         else
378         {
379             /* it's a QC filename */
380             argitem item;
381             item.filename = argv[0];
382             item.type     = TYPE_QC;
383             vec_push(items, item);
384         }
385     }
386     return true;
387 }
388
389 /* returns the line number, or -1 on error */
390 static bool progs_nextline(char **out, size_t *alen,FILE *src) {
391     int    len;
392     char  *line;
393     char  *start;
394     char  *end;
395
396     line = *out;
397     len = util_getline(&line, alen, src);
398     if (len == -1)
399         return false;
400
401     /* start at first non-blank */
402     for (start = line; isspace(*start); ++start) {}
403     /* end at the first non-blank */
404     for (end = start;  *end && !isspace(*end);  ++end)   {}
405
406     *out = line;
407     /* move the actual filename to the beginning */
408     while (start != end) {
409         *line++ = *start++;
410     }
411     *line = 0;
412     return true;
413 }
414
415 int main(int argc, char **argv) {
416     size_t itr;
417     int retval = 0;
418     bool opts_output_free = false;
419
420     app_name = argv[0];
421     con_init();
422
423     /* default options / warn flags */
424     options_set(opts_warn, WARN_UNKNOWN_CONTROL_SEQUENCE, true);
425     options_set(opts_warn, WARN_EXTENSIONS, true);
426     options_set(opts_warn, WARN_FIELD_REDECLARED, true);
427     options_set(opts_warn, WARN_TOO_FEW_PARAMETERS, true);
428     options_set(opts_warn, WARN_MISSING_RETURN_VALUES, true);
429     options_set(opts_warn, WARN_USED_UNINITIALIZED, true);
430     options_set(opts_warn, WARN_LOCAL_CONSTANTS, true);
431     options_set(opts_warn, WARN_VOID_VARIABLES, true);
432     options_set(opts_warn, WARN_IMPLICIT_FUNCTION_POINTER, true);
433     options_set(opts_warn, WARN_VARIADIC_FUNCTION, true);
434     options_set(opts_warn, WARN_FRAME_MACROS, true);
435     options_set(opts_warn, WARN_UNUSED_VARIABLE, true);
436     options_set(opts_warn, WARN_EFFECTLESS_STATEMENT, true);
437     options_set(opts_warn, WARN_END_SYS_FIELDS, true);
438     options_set(opts_warn, WARN_ASSIGN_FUNCTION_TYPES, true);
439
440     options_set(opts_flags, ADJUST_VECTOR_FIELDS, true);
441
442     if (!options_parse(argc, argv)) {
443         return usage();
444     }
445
446     /* the standard decides which set of operators to use */
447     if (opts_standard == COMPILER_GMQCC) {
448         operators = c_operators;
449         operator_count = c_operator_count;
450     } else {
451         operators = qcc_operators;
452         operator_count = qcc_operator_count;
453     }
454
455     if (opts_dump) {
456         for (itr = 0; itr < COUNT_FLAGS; ++itr) {
457             con_out("Flag %s = %i\n", opts_flag_list[itr].name, OPTS_FLAG(itr));
458         }
459         for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
460             con_out("Warning %s = %i\n", opts_warn_list[itr].name, OPTS_WARN(itr));
461         }
462         con_out("output = %s\n", opts_output);
463         con_out("optimization level = %i\n", (int)opts_O);
464         con_out("standard = %i\n", opts_standard);
465     }
466
467     if (!parser_init()) {
468         con_out("failed to initialize parser\n");
469         retval = 1;
470         goto cleanup;
471     }
472
473     util_debug("COM", "starting ...\n");
474
475     if (vec_size(items)) {
476         con_out("Mode: manual\n");
477         con_out("There are %lu items to compile:\n", (unsigned long)vec_size(items));
478         for (itr = 0; itr < vec_size(items); ++itr) {
479             con_out("  item: %s (%s)\n",
480                    items[itr].filename,
481                    ( (items[itr].type == TYPE_QC ? "qc" :
482                      (items[itr].type == TYPE_ASM ? "asm" :
483                      (items[itr].type == TYPE_SRC ? "progs.src" :
484                      ("unknown"))))));
485
486             if (!parser_compile_file(items[itr].filename))
487             {
488                     retval = 1;
489                     goto cleanup;
490             }
491         }
492
493         if (!parser_finish(opts_output)) {
494             retval = 1;
495             goto cleanup;
496         }
497
498     } else {
499         FILE *src;
500         char *line;
501         size_t linelen = 0;
502
503         con_out("Mode: progs.src\n");
504         src = util_fopen("progs.src", "rb");
505         if (!src) {
506             con_out("failed to open `progs.src` for reading\n");
507             retval = 1;
508             goto cleanup;
509         }
510
511         line = NULL;
512         if (!progs_nextline(&line, &linelen, src) || !line[0]) {
513             con_out("illformatted progs.src file: expected output filename in first line\n");
514             retval = 1;
515             goto srcdone;
516         }
517
518         if (!opts_output_wasset) {
519             opts_output = util_strdup(line);
520             opts_output_free = true;
521         }
522
523         while (progs_nextline(&line, &linelen, src)) {
524             if (!line[0] || (line[0] == '/' && line[1] == '/'))
525                 continue;
526             con_out("  src: %s\n", line);
527             if (!parser_compile_file(line)) {
528                 retval = 1;
529                 goto srcdone;
530             }
531         }
532
533         parser_finish(opts_output);
534
535 srcdone:
536         fclose(src);
537         mem_d(line);
538     }
539
540     /* stuff */
541
542 cleanup:
543     util_debug("COM", "cleaning ...\n");
544     con_close();
545     vec_free(items);
546
547     parser_cleanup();
548     if (opts_output_free)
549         mem_d((char*)opts_output);
550
551     lex_cleanup();
552     util_meminfo();
553     return retval;
554 }