main.c

   1 /*
   2  * Copyright (C) 2012, 2013
   3  *     Dale Weiler
   4  *     Wolfgang Bumiller
   5  *
   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:
  12  *
  13  * The above copyright notice and this permission notice shall be included in all
  14  * copies or substantial portions of the Software.
  15  *
  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
  22  * SOFTWARE.
  23  */
  24 #include <time.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28
  29 #include "gmqcc.h"
  30 #include "lexer.h"
  31
  32 /* TODO: cleanup this whole file .. it's a fuckign mess */
  33
  34 /* set by the standard */
  35 const oper_info *operators      = NULL;
  36 size_t           operator_count = 0;
  37 static bool      opts_output_wasset = false;
  38
  39 typedef struct { char *filename; int   type;  } argitem;
  40 typedef struct { char *name;     char *value; } ppitem;
  41 static argitem *items = NULL;
  42 static ppitem  *ppems = NULL;
  43
  44 #define TYPE_QC  0
  45 #define TYPE_ASM 1
  46 #define TYPE_SRC 2
  47
  48 static const char *app_name;
  49
  50 static void version(void) {
  51     con_out("GMQCC %d.%d.%d Built %s %s\n" GMQCC_DEV_VERSION_STRING,
  52         GMQCC_VERSION_MAJOR,
  53         GMQCC_VERSION_MINOR,
  54         GMQCC_VERSION_PATCH,
  55         __DATE__,
  56         __TIME__
  57     );
  58 }
  59
  60 static int usage(void) {
  61     con_out("usage: %s [options] [files...]", app_name);
  62     con_out("options:\n"
  63             "  -h, --help             show this help message\n"
  64             "  -debug                 turns on compiler debug messages\n"
  65             "  -memchk                turns on compiler memory leak check\n");
  66     con_out("  -o, --output=file      output file, defaults to progs.dat\n"
  67             "  -s filename            add a progs.src file to be used\n");
  68     con_out("  -E                     stop after preprocessing\n");
  69     con_out("  -q, --quiet            be less verbose\n");
  70     con_out("  -config file           use the specified ini file\n");
  71     con_out("  -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("  -f<flag>               enable a flag\n"
  76             "  -fno-<flag>            disable a flag\n"
  77             "  -fhelp                 list possible flags\n");
  78     con_out("  -W<warning>            enable a warning\n"
  79             "  -Wno-<warning>         disable a warning\n"
  80             "  -Wall                  enable all warnings\n");
  81     con_out("  -Werror                treat warnings as errors\n"
  82             "  -Werror-<warning>      treat a warning as error\n"
  83             "  -Wno-error-<warning>   opposite of the above\n");
  84     con_out("  -Whelp                 list possible warnings\n");
  85     con_out("  -O<number>             optimization level\n"
  86             "  -O<name>               enable specific optimization\n"
  87             "  -Ono-<name>            disable specific optimization\n"
  88             "  -Ohelp                 list optimizations\n");
  89     con_out("  -force-crc=num         force a specific checksum into the header\n");
  90     return -1;
  91 }
  92
  93 /* command line parsing */
  94 static bool options_witharg(int *argc_, char ***argv_, char **out) {
  95     int  argc   = *argc_;
  96     char **argv = *argv_;
  97
  98     if (argv[0][2]) {
  99         *out = argv[0]+2;
 100         return true;
 101     }
 102     /* eat up the next */
 103     if (argc < 2) /* no parameter was provided */
 104         return false;
 105
 106     *out = argv[1];
 107     --*argc_;
 108     ++*argv_;
 109     return true;
 110 }
 111
 112 static bool options_long_witharg_all(const char *optname, int *argc_, char ***argv_, char **out, int ds, bool split) {
 113     int  argc   = *argc_;
 114     char **argv = *argv_;
 115
 116     size_t len = strlen(optname);
 117
 118     if (strncmp(argv[0]+ds, optname, len))
 119         return false;
 120
 121     /* it's --optname, check how the parameter is supplied */
 122     if (argv[0][ds+len] == '=') {
 123         /* using --opt=param */
 124         *out = argv[0]+ds+len+1;
 125         return true;
 126     }
 127
 128     if (!split || argc < ds) /* no parameter was provided, or only single-arg form accepted */
 129         return false;
 130
 131     /* using --opt param */
 132     *out = argv[1];
 133     --*argc_;
 134     ++*argv_;
 135     return true;
 136 }
 137 static bool options_long_witharg(const char *optname, int *argc_, char ***argv_, char **out) {
 138     return options_long_witharg_all(optname, argc_, argv_, out, 2, true);
 139 }
 140 static bool options_long_gcc(const char *optname, int *argc_, char ***argv_, char **out) {
 141     return options_long_witharg_all(optname, argc_, argv_, out, 1, false);
 142 }
 143
 144 static bool options_parse(int argc, char **argv) {
 145     bool argend = false;
 146     size_t itr;
 147     char  buffer[1024];
 148     char *redirout    = NULL;
 149     char *redirerr    = NULL;
 150     char *config      = NULL;
 151     char *memdumpcols = NULL;
 152
 153     while (!argend && argc > 1) {
 154         char *argarg;
 155         argitem item;
 156         ppitem  macro;
 157
 158         ++argv;
 159         --argc;
 160
 161         if (argv[0][0] == '-') {
 162             /* All gcc-type long options */
 163             if (options_long_gcc("std", &argc, &argv, &argarg)) {
 164                 if (!strcmp(argarg, "gmqcc") || !strcmp(argarg, "default")) {
 165
 166                     opts_set(opts.flags, ADJUST_VECTOR_FIELDS,          true);
 167                     opts_set(opts.flags, CORRECT_LOGIC,                 true);
 168                     opts_set(opts.flags, FALSE_EMPTY_STRINGS,           false);
 169                     opts_set(opts.flags, TRUE_EMPTY_STRINGS,            true);
 170                     opts_set(opts.flags, LOOP_LABELS,                   true);
 171                     opts_set(opts.flags, TRANSLATABLE_STRINGS,          true);
 172                     opts_set(opts.flags, INITIALIZED_NONCONSTANTS,      true);
 173                     opts_set(opts.werror, WARN_INVALID_PARAMETER_COUNT, true);
 174                     opts_set(opts.werror, WARN_MISSING_RETURN_VALUES,   true);
 175                     opts_set(opts.flags,  EXPRESSIONS_FOR_BUILTINS,     true);
 176                     opts_set(opts.warn,   WARN_BREAKDEF,                true);
 177
 178
 179                     OPTS_OPTION_U32(OPTION_STANDARD) = COMPILER_GMQCC;
 180
 181                 } else if (!strcmp(argarg, "qcc")) {
 182
 183                     opts_set(opts.flags, ADJUST_VECTOR_FIELDS,  false);
 184                     opts_set(opts.flags, ASSIGN_FUNCTION_TYPES, true);
 185
 186                     OPTS_OPTION_U32(OPTION_STANDARD) = COMPILER_QCC;
 187
 188                 } else if (!strcmp(argarg, "fte") || !strcmp(argarg, "fteqcc")) {
 189
 190                     opts_set(opts.flags, FTEPP,                    true);
 191                     opts_set(opts.flags, TRANSLATABLE_STRINGS,     true);
 192                     opts_set(opts.flags, ADJUST_VECTOR_FIELDS,     false);
 193                     opts_set(opts.flags, ASSIGN_FUNCTION_TYPES,    true);
 194                     opts_set(opts.flags, CORRECT_TERNARY,          false);
 195                     opts_set(opts.warn, WARN_TERNARY_PRECEDENCE,   true);
 196                     opts_set(opts.warn, WARN_BREAKDEF,             true);
 197
 198                     OPTS_OPTION_U32(OPTION_STANDARD) = COMPILER_FTEQCC;
 199
 200                 } else if (!strcmp(argarg, "qccx")) {
 201
 202                     opts_set(opts.flags, ADJUST_VECTOR_FIELDS,  false);
 203                     OPTS_OPTION_U32(OPTION_STANDARD) = COMPILER_QCCX;
 204
 205                 } else {
 206                     con_out("Unknown standard: %s\n", argarg);
 207                     return false;
 208                 }
 209                 continue;
 210             }
 211             if (options_long_gcc("force-crc", &argc, &argv, &argarg)) {
 212
 213                 OPTS_OPTION_BOOL(OPTION_FORCECRC)   = true;
 214                 OPTS_OPTION_U16 (OPTION_FORCED_CRC) = strtol(argarg, NULL, 0);
 215                 continue;
 216             }
 217             if (options_long_gcc("redirout", &argc, &argv, &redirout)) {
 218                 con_change(redirout, redirerr);
 219                 continue;
 220             }
 221             if (options_long_gcc("redirerr", &argc, &argv, &redirerr)) {
 222                 con_change(redirout, redirerr);
 223                 continue;
 224             }
 225             if (options_long_gcc("config", &argc, &argv, &argarg)) {
 226                 config = argarg;
 227                 continue;
 228             }
 229             if (options_long_gcc("memdumpcols", &argc, &argv, &memdumpcols)) {
 230                 OPTS_OPTION_U16(OPTION_MEMDUMPCOLS) = (uint16_t)strtol(memdumpcols, NULL, 10);
 231                 continue;
 232             }
 233
 234             /* show defaults (like pathscale) */
 235             if (!strcmp(argv[0]+1, "show-defaults")) {
 236                 for (itr = 0; itr < COUNT_FLAGS; ++itr) {
 237                     if (!OPTS_FLAG(itr))
 238                         continue;
 239
 240                     memset(buffer, 0, sizeof(buffer));
 241                     util_strtononcmd(opts_flag_list[itr].name, buffer, strlen(opts_flag_list[itr].name) + 1);
 242
 243                     con_out("-f%s ", buffer);
 244                 }
 245                 for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
 246                     if (!OPTS_WARN(itr))
 247                         continue;
 248
 249                     memset(buffer, 0, sizeof(buffer));
 250                     util_strtononcmd(opts_warn_list[itr].name, buffer, strlen(opts_warn_list[itr].name) + 1);
 251                     con_out("-W%s ", buffer);
 252                 }
 253                 con_out("\n");
 254                 exit(0);
 255             }
 256
 257             if (!strcmp(argv[0]+1, "debug")) {
 258                 OPTS_OPTION_BOOL(OPTION_DEBUG) = true;
 259                 continue;
 260             }
 261             if (!strcmp(argv[0]+1, "dump")) {
 262                 OPTS_OPTION_BOOL(OPTION_DUMP)  = true;
 263                 continue;
 264             }
 265             if (!strcmp(argv[0]+1, "dumpfin")) {
 266                 OPTS_OPTION_BOOL(OPTION_DUMPFIN) = true;
 267                 continue;
 268             }
 269             if (!strcmp(argv[0]+1, "memchk")) {
 270                 OPTS_OPTION_BOOL(OPTION_MEMCHK) = true;
 271                 continue;
 272             }
 273             if (!strcmp(argv[0]+1, "nocolor")) {
 274                 con_color(0);
 275                 continue;
 276             }
 277
 278             switch (argv[0][1]) {
 279                 /* -h, show usage but exit with 0 */
 280                 case 'h':
 281                     usage();
 282                     exit(0);
 283                     /* break; never reached because of exit(0) */
 284
 285                 case 'v':
 286                     version();
 287                     exit(0);
 288
 289                 case 'E':
 290                     OPTS_OPTION_BOOL(OPTION_PP_ONLY) = true;
 291                     opts_set(opts.flags, FTEPP_PREDEFS, true); /* predefs on for -E */
 292                     break;
 293
 294                 /* debug turns on -flno */
 295                 case 'g':
 296                     opts_setflag("LNO", true);
 297                     OPTS_OPTION_BOOL(OPTION_G) = true;
 298                     break;
 299
 300                 case 'q':
 301                     OPTS_OPTION_BOOL(OPTION_QUIET) = true;
 302                     break;
 303
 304                 case 'D':
 305                     if (!strlen(argv[0]+2)) {
 306                         con_err("expected name after -D\n");
 307                         exit(0);
 308                     }
 309
 310                     if (!(argarg = strchr(argv[0] + 2, '='))) {
 311                         macro.name  = util_strdup(argv[0]+2);
 312                         macro.value = NULL;
 313                     } else {
 314                         *argarg='\0'; /* terminate for name */
 315                         macro.name  = util_strdup(argv[0]+2);
 316                         macro.value = util_strdup(argarg+1);
 317                     }
 318                     vec_push(ppems, macro);
 319                     break;
 320
 321                 /* handle all -fflags */
 322                 case 'f':
 323                     util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
 324                     if (!strcmp(argv[0]+2, "HELP") || *(argv[0]+2) == '?') {
 325                         con_out("Possible flags:\n\n");
 326                         for (itr = 0; itr < COUNT_FLAGS; ++itr) {
 327                             util_strtononcmd(opts_flag_list[itr].name, buffer, sizeof(buffer));
 328                             con_out(" -f%s\n", buffer);
 329                         }
 330                         exit(0);
 331                     }
 332                     else if (!strncmp(argv[0]+2, "NO_", 3)) {
 333                         if (!opts_setflag(argv[0]+5, false)) {
 334                             con_out("unknown flag: %s\n", argv[0]+2);
 335                             return false;
 336                         }
 337                     }
 338                     else if (!opts_setflag(argv[0]+2, true)) {
 339                         con_out("unknown flag: %s\n", argv[0]+2);
 340                         return false;
 341                     }
 342                     break;
 343                 case 'W':
 344                     util_strtocmd(argv[0]+2, argv[0]+2, strlen(argv[0]+2)+1);
 345                     if (!strcmp(argv[0]+2, "HELP") || *(argv[0]+2) == '?') {
 346                         con_out("Possible warnings:\n");
 347                         for (itr = 0; itr < COUNT_WARNINGS; ++itr) {
 348                             util_strtononcmd(opts_warn_list[itr].name, buffer, sizeof(buffer));
 349                             con_out(" -W%s\n", buffer);
 350                             if (itr == WARN_DEBUG)
 351                                 con_out("   Warnings included by -Wall:\n");
 352                         }
 353                         exit(0);
 354                     }
 355                     else if (!strcmp(argv[0]+2, "NO_ERROR") ||
 356                              !strcmp(argv[0]+2, "NO_ERROR_ALL"))
 357                     {
 358                         for (itr = 0; itr < sizeof(opts.werror)/sizeof(opts.werror[0]); ++itr)
 359                             opts.werror[itr] = 0;
 360                         break;
 361                     }
 362                     else if (!strcmp(argv[0]+2, "ERROR") ||
 363                              !strcmp(argv[0]+2, "ERROR_ALL"))
 364                     {
 365                         opts_backup_non_Werror_all();
 366                         for (itr = 0; itr < sizeof(opts.werror)/sizeof(opts.werror[0]); ++itr)
 367                             opts.werror[itr] = 0xFFFFFFFFL;
 368                         opts_restore_non_Werror_all();
 369                         break;
 370                     }
 371                     else if (!strcmp(argv[0]+2, "NONE")) {
 372                         for (itr = 0; itr < sizeof(opts.warn)/sizeof(opts.warn[0]); ++itr)
 373                             opts.warn[itr] = 0;
 374                         break;
 375                     }
 376                     else if (!strcmp(argv[0]+2, "ALL")) {
 377                         opts_backup_non_Wall();
 378                         for (itr = 0; itr < sizeof(opts.warn)/sizeof(opts.warn[0]); ++itr)
 379                             opts.warn[itr] = 0xFFFFFFFFL;
 380                         opts_restore_non_Wall();
 381                         break;
 382                     }
 383                     else if (!strncmp(argv[0]+2, "ERROR_", 6)) {
 384                         if (!opts_setwerror(argv[0]+8, true)) {
 385                             con_out("unknown warning: %s\n", argv[0]+2);
 386                             return false;
 387                         }
 388                     }
 389                     else if (!strncmp(argv[0]+2, "NO_ERROR_", 9)) {
 390                         if (!opts_setwerror(argv[0]+11, false)) {
 391                             con_out("unknown warning: %s\n", argv[0]+2);
 392                             return false;
 393                         }
 394                     }
 395                     else if (!strncmp(argv[0]+2, "NO_", 3)) {
 396                         if (!opts_setwarn(argv[0]+5, false)) {
 397                             con_out("unknown warning: %s\n", argv[0]+2);
 398                             return false;
 399                         }
 400                     }
 401                     else if (!opts_setwarn(argv[0]+2, true)) {
 402                         con_out("unknown warning: %s\n", argv[0]+2);
 403                         return false;
 404                     }
 405                     break;
 406
 407                 case 'O':
 408                     if (!options_witharg(&argc, &argv, &argarg)) {
 409                         con_out("option -O requires a numerical argument, or optimization name with an optional 'no-' prefix\n");
 410                         return false;
 411                     }
 412                     if (util_isdigit(argarg[0])) {
 413                         uint32_t val = (uint32_t)strtol(argarg, NULL, 10);
 414                         OPTS_OPTION_U32(OPTION_O) = val;
 415                         opts_setoptimlevel(val);
 416                     } else {
 417                         util_strtocmd(argarg, argarg, strlen(argarg)+1);
 418                         if (!strcmp(argarg, "HELP")) {
 419                             con_out("Possible optimizations:\n");
 420                             for (itr = 0; itr < COUNT_OPTIMIZATIONS; ++itr) {
 421                                 util_strtononcmd(opts_opt_list[itr].name, buffer, sizeof(buffer));
 422                                 con_out(" -O%-20s (-O%u)\n", buffer, opts_opt_oflag[itr]);
 423                             }
 424                             exit(0);
 425                         }
 426                         else if (!strcmp(argarg, "ALL"))
 427                             opts_setoptimlevel(OPTS_OPTION_U32(OPTION_O) = 9999);
 428                         else if (!strncmp(argarg, "NO_", 3)) {
 429                             if (!opts_setoptim(argarg+3, false)) {
 430                                 con_out("unknown optimization: %s\n", argarg+3);
 431                                 return false;
 432                             }
 433                         }
 434                         else {
 435                             if (!opts_setoptim(argarg, true)) {
 436                                 con_out("unknown optimization: %s\n", argarg);
 437                                 return false;
 438                             }
 439                         }
 440                     }
 441                     break;
 442
 443                 case 'o':
 444                     if (!options_witharg(&argc, &argv, &argarg)) {
 445                         con_out("option -o requires an argument: the output file name\n");
 446                         return false;
 447                     }
 448                     OPTS_OPTION_STR(OPTION_OUTPUT) = argarg;
 449                     opts_output_wasset = true;
 450                     break;
 451
 452                 case 'a':
 453                 case 's':
 454                     item.type = argv[0][1] == 'a' ? TYPE_ASM : TYPE_SRC;
 455                     if (!options_witharg(&argc, &argv, &argarg)) {
 456                         con_out("option -a requires a filename %s\n",
 457                                 (argv[0][1] == 'a' ? "containing QC-asm" : "containing a progs.src formatted list"));
 458                         return false;
 459                     }
 460                     item.filename = argarg;
 461                     vec_push(items, item);
 462                     break;
 463
 464                 case '-':
 465                     if (!argv[0][2]) {
 466                         /* anything following -- is considered a non-option argument */
 467                         argend = true;
 468                         break;
 469                     }
 470             /* All long options without arguments */
 471                     else if (!strcmp(argv[0]+2, "help")) {
 472                         usage();
 473                         exit(0);
 474                     }
 475                     else if (!strcmp(argv[0]+2, "version")) {
 476                         version();
 477                         exit(0);
 478                     }
 479                     else if (!strcmp(argv[0]+2, "quiet")) {
 480                         OPTS_OPTION_BOOL(OPTION_QUIET) = true;
 481                         break;
 482                     }
 483                     else if (!strcmp(argv[0]+2, "correct")) {
 484                         OPTS_OPTION_BOOL(OPTION_CORRECTION) = true;
 485                         break;
 486                     }
 487                     else if (!strcmp(argv[0]+2, "no-correct")) {
 488                         OPTS_OPTION_BOOL(OPTION_CORRECTION) = false;
 489                         break;
 490                     }
 491                     else if (!strcmp(argv[0]+2, "add-info")) {
 492                         OPTS_OPTION_BOOL(OPTION_ADD_INFO) = true;
 493                         break;
 494                     }
 495                     else {
 496             /* All long options with arguments */
 497                         if (options_long_witharg("output", &argc, &argv, &argarg)) {
 498                             OPTS_OPTION_STR(OPTION_OUTPUT) = argarg;
 499                             opts_output_wasset = true;
 500                         } else {
 501                             con_out("Unknown parameter: %s\n", argv[0]);
 502                             return false;
 503                         }
 504                     }
 505                     break;
 506
 507                 default:
 508                     con_out("Unknown parameter: %s\n", argv[0]);
 509                     return false;
 510             }
 511         }
 512         else
 513         {
 514             /* it's a QC filename */
 515             item.filename = argv[0];
 516             item.type     = TYPE_QC;
 517             vec_push(items, item);
 518         }
 519     }
 520     opts_ini_init(config);
 521     return true;
 522 }
 523
 524 /* returns the line number, or -1 on error */
 525 static bool progs_nextline(char **out, size_t *alen,FILE *src) {
 526     int    len;
 527     char  *line;
 528     char  *start;
 529     char  *end;
 530
 531     line = *out;
 532     len  = fs_file_getline(&line, alen, src);
 533     if (len == -1)
 534         return false;
 535
 536     /* start at first non-blank */
 537     for (start = line; util_isspace(*start); ++start) {}
 538     /* end at the first non-blank */
 539     for (end = start; *end && !util_isspace(*end);  ++end)   {}
 540
 541     *out = line;
 542     /* move the actual filename to the beginning */
 543     while (start != end) {
 544         *line++ = *start++;
 545     }
 546     *line = 0;
 547     return true;
 548 }
 549
 550 int main(int argc, char **argv) {
 551     size_t          itr;
 552     int             retval           = 0;
 553     bool            opts_output_free = false;
 554     bool            operators_free   = false;
 555     bool            progs_src        = false;
 556     FILE            *outfile         = NULL;
 557     struct parser_s *parser          = NULL;
 558     struct ftepp_s  *ftepp           = NULL;
 559
 560     app_name = argv[0];
 561     con_init ();
 562     opts_init("progs.dat", COMPILER_QCC, (1024 << 3));
 563
 564     util_seed(time(0));
 565
 566     if (!options_parse(argc, argv)) {
 567         return usage();
 568     }
 569
 570     if (OPTS_FLAG(TRUE_EMPTY_STRINGS) && OPTS_FLAG(FALSE_EMPTY_STRINGS)) {
 571         con_err("-ftrue-empty-strings and -ffalse-empty-strings are mutually exclusive");
 572         exit(EXIT_FAILURE);
 573     }
 574
 575     /* the standard decides which set of operators to use */
 576     if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_GMQCC) {
 577         operators      = c_operators;
 578         operator_count = c_operator_count;
 579     } else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC) {
 580         operators      = fte_operators;
 581         operator_count = fte_operator_count;
 582     } else {
 583         operators      = qcc_operators;
 584         operator_count = qcc_operator_count;
 585     }
 586
 587     if (operators == fte_operators) {
 588         /* fix ternary? */
 589         if (OPTS_FLAG(CORRECT_TERNARY)) {
 590             oper_info *newops;
 591             if (operators[operator_count-2].id != opid1(',') ||
 592                 operators[operator_count-1].id != opid2(':','?'))
 593             {
 594                 con_err("internal error: operator precedence table wasn't updated correctly!\n");
 595                 exit(EXIT_FAILURE);
 596             }
 597             operators_free = true;
 598             newops = (oper_info*)mem_a(sizeof(operators[0]) * operator_count);
 599             memcpy(newops, operators, sizeof(operators[0]) * operator_count);
 600             memcpy(&newops[operator_count-2], &operators[operator_count-1], sizeof(newops[0]));
 601             memcpy(&newops[operator_count-1], &operators[operator_count-2], sizeof(newops[0]));
 602             newops[operator_count-2].prec = newops[operator_count-1].prec+1;
 603             operators = newops;
 604         }
 605     }
 606
 607     if (OPTS_OPTION_BOOL(OPTION_DUMP)) {
 608         for (itr = 0; itr < COUNT_FLAGS; ++itr)
 609             con_out("Flag %s = %i\n",    opts_flag_list[itr].name, OPTS_FLAG(itr));
 610         for (itr = 0; itr < COUNT_WARNINGS; ++itr)
 611             con_out("Warning %s = %i\n", opts_warn_list[itr].name, OPTS_WARN(itr));
 612
 613         con_out("output             = %s\n", OPTS_OPTION_STR(OPTION_OUTPUT));
 614         con_out("optimization level = %u\n", OPTS_OPTION_U32(OPTION_O));
 615         con_out("standard           = %u\n", OPTS_OPTION_U32(OPTION_STANDARD));
 616     }
 617
 618     if (OPTS_OPTION_BOOL(OPTION_PP_ONLY)) {
 619         if (opts_output_wasset) {
 620             outfile = fs_file_open(OPTS_OPTION_STR(OPTION_OUTPUT), "wb");
 621             if (!outfile) {
 622                 con_err("failed to open `%s` for writing\n", OPTS_OPTION_STR(OPTION_OUTPUT));
 623                 retval = 1;
 624                 goto cleanup;
 625             }
 626         }
 627         else {
 628             outfile = con_default_out();
 629         }
 630     }
 631
 632     if (!OPTS_OPTION_BOOL(OPTION_PP_ONLY)) {
 633         if (!(parser = parser_create())) {
 634             con_err("failed to initialize parser\n");
 635             retval = 1;
 636             goto cleanup;
 637         }
 638     }
 639
 640     if (OPTS_OPTION_BOOL(OPTION_PP_ONLY) || OPTS_FLAG(FTEPP)) {
 641         if (!(ftepp = ftepp_create())) {
 642             con_err("failed to initialize parser\n");
 643             retval = 1;
 644             goto cleanup;
 645         }
 646     }
 647
 648     util_debug("COM", "starting ...\n");
 649
 650     /* add macros */
 651     if (OPTS_OPTION_BOOL(OPTION_PP_ONLY) || OPTS_FLAG(FTEPP)) {
 652         for (itr = 0; itr < vec_size(ppems); itr++) {
 653             ftepp_add_macro(ftepp, ppems[itr].name, ppems[itr].value);
 654             mem_d(ppems[itr].name);
 655
 656             /* can be null */
 657             if (ppems[itr].value)
 658                 mem_d(ppems[itr].value);
 659         }
 660     }
 661
 662     if (!vec_size(items)) {
 663         FILE  *src;
 664         char  *line    = NULL;
 665         size_t linelen = 0;
 666         bool   hasline = false;
 667
 668         progs_src = true;
 669
 670         src = fs_file_open("progs.src", "rb");
 671         if (!src) {
 672             con_err("failed to open `progs.src` for reading\n");
 673             retval = 1;
 674             goto cleanup;
 675         }
 676
 677         while (progs_nextline(&line, &linelen, src)) {
 678             argitem item;
 679
 680             if (!line[0] || (line[0] == '/' && line[1] == '/'))
 681                 continue;
 682
 683             if (hasline) {
 684                 item.filename = util_strdup(line);
 685                 item.type     = TYPE_QC;
 686                 vec_push(items, item);
 687             } else if (!opts_output_wasset) {
 688                 OPTS_OPTION_STR(OPTION_OUTPUT) = util_strdup(line);
 689                 opts_output_free               = true;
 690                 hasline                        = true;
 691             }
 692         }
 693
 694         fs_file_close(src);
 695         mem_d(line);
 696     }
 697
 698     if (vec_size(items)) {
 699         if (!OPTS_OPTION_BOOL(OPTION_QUIET) &&
 700             !OPTS_OPTION_BOOL(OPTION_PP_ONLY))
 701         {
 702             con_out("Mode: %s\n", (progs_src ? "progs.src" : "manual"));
 703             con_out("There are %lu items to compile:\n", (unsigned long)vec_size(items));
 704         }
 705
 706         for (itr = 0; itr < vec_size(items); ++itr) {
 707             if (!OPTS_OPTION_BOOL(OPTION_QUIET) &&
 708                 !OPTS_OPTION_BOOL(OPTION_PP_ONLY))
 709             {
 710                 con_out("  item: %s (%s)\n",
 711                        items[itr].filename,
 712                        ( (items[itr].type == TYPE_QC ? "qc" :
 713                          (items[itr].type == TYPE_ASM ? "asm" :
 714                          (items[itr].type == TYPE_SRC ? "progs.src" :
 715                          ("unknown"))))));
 716             }
 717
 718             if (OPTS_OPTION_BOOL(OPTION_PP_ONLY)) {
 719                 const char *out;
 720                 if (!ftepp_preprocess_file(ftepp, items[itr].filename)) {
 721                     retval = 1;
 722                     goto cleanup;
 723                 }
 724                 out = ftepp_get(ftepp);
 725                 if (out)
 726                     fs_file_printf(outfile, "%s", out);
 727                 ftepp_flush(ftepp);
 728             }
 729             else {
 730                 if (OPTS_FLAG(FTEPP)) {
 731                     const char *data;
 732                     if (!ftepp_preprocess_file(ftepp, items[itr].filename)) {
 733                         retval = 1;
 734                         goto cleanup;
 735                     }
 736                     data = ftepp_get(ftepp);
 737                     if (vec_size(data)) {
 738                         if (!parser_compile_string(parser, items[itr].filename, data, vec_size(data))) {
 739                             retval = 1;
 740                             goto cleanup;
 741                         }
 742                     }
 743                     ftepp_flush(ftepp);
 744                 }
 745                 else {
 746                     if (!parser_compile_file(parser, items[itr].filename)) {
 747                         retval = 1;
 748                         goto cleanup;
 749                     }
 750                 }
 751             }
 752
 753             if (progs_src) {
 754                 mem_d(items[itr].filename);
 755                 items[itr].filename = NULL;
 756             }
 757         }
 758
 759         ftepp_finish(ftepp);
 760         ftepp = NULL;
 761         if (!OPTS_OPTION_BOOL(OPTION_PP_ONLY)) {
 762             if (!parser_finish(parser, OPTS_OPTION_STR(OPTION_OUTPUT))) {
 763                 retval = 1;
 764                 goto cleanup;
 765             }
 766         }
 767     }
 768
 769 cleanup:
 770     util_debug("COM", "cleaning ...\n");
 771     if (ftepp)
 772         ftepp_finish(ftepp);
 773     con_close();
 774     vec_free(items);
 775     vec_free(ppems);
 776
 777     if (!OPTS_OPTION_BOOL(OPTION_PP_ONLY))
 778         if(parser) parser_cleanup(parser);
 779     if (opts_output_free)
 780         mem_d(OPTS_OPTION_STR(OPTION_OUTPUT));
 781     if (operators_free)
 782         mem_d((void*)operators);
 783
 784     lex_cleanup();
 785     stat_info();
 786
 787     return retval;
 788 }