ftepp.c

   1 /*
   2  * Copyright (C) 2012, 2013, 2014
   3  *     Wolfgang Bumiller
   4  *     Dale Weiler
   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 <string.h>
  25 #include <stdlib.h>
  26 #include <sys/stat.h>
  27
  28 #include "gmqcc.h"
  29 #include "lexer.h"
  30
  31 #define HT_MACROS 1024
  32
  33 typedef struct {
  34     bool on;
  35     bool was_on;
  36     bool had_else;
  37 } ppcondition;
  38
  39 typedef struct {
  40     int   token;
  41     char *value;
  42     /* a copy from the lexer */
  43     union {
  44         vec3_t v;
  45         int    i;
  46         double f;
  47         int    t; /* type */
  48     } constval;
  49 } pptoken;
  50
  51 typedef struct {
  52     lex_ctx_t ctx;
  53
  54     char   *name;
  55     char  **params;
  56     /* yes we need an extra flag since `#define FOO x` is not the same as `#define FOO() x` */
  57     bool    has_params;
  58     bool    variadic;
  59
  60     pptoken **output;
  61 } ppmacro;
  62
  63 typedef struct ftepp_s {
  64     lex_file    *lex;
  65     int          token;
  66     unsigned int errors;
  67
  68     bool         output_on;
  69     ppcondition *conditions;
  70     /*ppmacro    **macros;*/
  71     ht           macros;  /* hashtable<string, ppmacro*> */
  72     char        *output_string;
  73
  74     char        *itemname;
  75     char        *includename;
  76     bool         in_macro;
  77
  78     uint32_t predef_countval;
  79     uint32_t predef_randval;
  80 } ftepp_t;
  81
  82 /* __DATE__ */
  83 static char *ftepp_predef_date(ftepp_t *context) {
  84     const struct tm *itime = NULL;
  85     char            *value = (char*)mem_a(82);
  86     time_t           rtime;
  87
  88     (void)context;
  89
  90     time (&rtime);
  91     itime = util_localtime(&rtime);
  92     strftime(value, 82, "\"%b %d %Y\"", itime);
  93
  94     return value;
  95 }
  96
  97 /* __TIME__ */
  98 static char *ftepp_predef_time(ftepp_t *context) {
  99     const struct tm *itime = NULL;
 100     char            *value = (char*)mem_a(82);
 101     time_t           rtime;
 102
 103     (void)context;
 104
 105     time (&rtime);
 106     itime = util_localtime(&rtime);
 107     strftime(value, 82, "\"%X\"", itime);
 108
 109     return value;
 110 }
 111
 112 /* __LINE__ */
 113 static char *ftepp_predef_line(ftepp_t *context) {
 114     char *value;
 115
 116     util_asprintf(&value, "%d", (int)context->lex->line);
 117     return value;
 118 }
 119 /* __FILE__ */
 120 static char *ftepp_predef_file(ftepp_t *context) {
 121     size_t length = strlen(context->lex->name) + 3; /* two quotes and a terminator */
 122     char  *value  = (char*)mem_a(length);
 123
 124     util_snprintf(value, length, "\"%s\"", context->lex->name);
 125     return value;
 126 }
 127 /* __COUNTER_LAST__ */
 128 static char *ftepp_predef_counterlast(ftepp_t *context) {
 129     char *value;
 130     util_asprintf(&value, "%u", context->predef_countval);
 131     return value;
 132 }
 133 /* __COUNTER__ */
 134 static char *ftepp_predef_counter(ftepp_t *context) {
 135     char *value;
 136
 137     context->predef_countval ++;
 138     util_asprintf(&value, "%u", context->predef_countval);
 139
 140     return value;
 141 }
 142 /* __RANDOM__ */
 143 static char *ftepp_predef_random(ftepp_t *context) {
 144     char *value;
 145
 146     context->predef_randval = (util_rand() % 0xFF) + 1;
 147     util_asprintf(&value, "%u", context->predef_randval);
 148     return value;
 149 }
 150 /* __RANDOM_LAST__ */
 151 static char *ftepp_predef_randomlast(ftepp_t *context) {
 152     char *value;
 153
 154     util_asprintf(&value, "%u", context->predef_randval);
 155     return value;
 156 }
 157 /* __TIMESTAMP__ */
 158 static char *ftepp_predef_timestamp(ftepp_t *context) {
 159     struct stat finfo;
 160     const char *find;
 161     char       *value;
 162     size_t      size;
 163
 164     if (stat(context->lex->name, &finfo))
 165         return util_strdup("\"<failed to determine timestamp>\"");
 166
 167     find = util_ctime(&finfo.st_mtime);
 168     value = (char*)mem_a(strlen(find) + 1);
 169     memcpy(&value[1], find, (size = strlen(find)) - 1);
 170
 171     value[0]    = '"';
 172     value[size] = '"';
 173
 174     return value;
 175 }
 176
 177 typedef struct {
 178     const char   *name;
 179     char       *(*func)(ftepp_t *);
 180 } ftepp_predef_t;
 181
 182 static const ftepp_predef_t ftepp_predefs[] = {
 183     { "__LINE__",         &ftepp_predef_line        },
 184     { "__FILE__",         &ftepp_predef_file        },
 185     { "__COUNTER__",      &ftepp_predef_counter     },
 186     { "__COUNTER_LAST__", &ftepp_predef_counterlast },
 187     { "__RANDOM__",       &ftepp_predef_random      },
 188     { "__RANDOM_LAST__",  &ftepp_predef_randomlast  },
 189     { "__DATE__",         &ftepp_predef_date        },
 190     { "__TIME__",         &ftepp_predef_time        },
 191     { "__TIME_STAMP__",   &ftepp_predef_timestamp   }
 192 };
 193
 194 static GMQCC_INLINE size_t ftepp_predef_index(const char *name) {
 195     /* no hashtable here, we simply check for one to exist the naive way */
 196     size_t i;
 197     for(i = 1; i < GMQCC_ARRAY_COUNT(ftepp_predefs) + 1; i++)
 198         if (!strcmp(ftepp_predefs[i-1].name, name))
 199             return i;
 200     return 0;
 201 }
 202
 203 bool ftepp_predef_exists(const char *name);
 204 bool ftepp_predef_exists(const char *name) {
 205     return ftepp_predef_index(name) != 0;
 206 }
 207
 208 /* singleton because we're allowed */
 209 static GMQCC_INLINE char *(*ftepp_predef(const char *name))(ftepp_t *context) {
 210     size_t i = ftepp_predef_index(name);
 211     return (i != 0) ? ftepp_predefs[i-1].func : NULL;
 212 }
 213
 214 #define ftepp_tokval(f) ((f)->lex->tok.value)
 215 #define ftepp_ctx(f)    ((f)->lex->tok.ctx)
 216
 217 static void ftepp_errorat(ftepp_t *ftepp, lex_ctx_t ctx, const char *fmt, ...)
 218 {
 219     va_list ap;
 220
 221     ftepp->errors++;
 222
 223     va_start(ap, fmt);
 224     con_cvprintmsg(ctx, LVL_ERROR, "error", fmt, ap);
 225     va_end(ap);
 226 }
 227
 228 static void ftepp_error(ftepp_t *ftepp, const char *fmt, ...)
 229 {
 230     va_list ap;
 231
 232     ftepp->errors++;
 233
 234     va_start(ap, fmt);
 235     con_cvprintmsg(ftepp->lex->tok.ctx, LVL_ERROR, "error", fmt, ap);
 236     va_end(ap);
 237 }
 238
 239 static bool GMQCC_WARN ftepp_warn(ftepp_t *ftepp, int warntype, const char *fmt, ...)
 240 {
 241     bool    r;
 242     va_list ap;
 243
 244     va_start(ap, fmt);
 245     r = vcompile_warning(ftepp->lex->tok.ctx, warntype, fmt, ap);
 246     va_end(ap);
 247     return r;
 248 }
 249
 250 static pptoken *pptoken_make(ftepp_t *ftepp)
 251 {
 252     pptoken *token = (pptoken*)mem_a(sizeof(pptoken));
 253     token->token = ftepp->token;
 254 #if 0
 255     if (token->token == TOKEN_WHITE)
 256         token->value = util_strdup(" ");
 257     else
 258 #else
 259         token->value = util_strdup(ftepp_tokval(ftepp));
 260 #endif
 261     memcpy(&token->constval, &ftepp->lex->tok.constval, sizeof(token->constval));
 262     return token;
 263 }
 264
 265 static GMQCC_INLINE void pptoken_delete(pptoken *self)
 266 {
 267     mem_d(self->value);
 268     mem_d(self);
 269 }
 270
 271 static ppmacro *ppmacro_new(lex_ctx_t ctx, const char *name)
 272 {
 273     ppmacro *macro = (ppmacro*)mem_a(sizeof(ppmacro));
 274
 275     (void)ctx;
 276     memset(macro, 0, sizeof(*macro));
 277     macro->name = util_strdup(name);
 278     return macro;
 279 }
 280
 281 static void ppmacro_delete(ppmacro *self)
 282 {
 283     size_t i;
 284     for (i = 0; i < vec_size(self->params); ++i)
 285         mem_d(self->params[i]);
 286     vec_free(self->params);
 287     for (i = 0; i < vec_size(self->output); ++i)
 288         pptoken_delete(self->output[i]);
 289     vec_free(self->output);
 290     mem_d(self->name);
 291     mem_d(self);
 292 }
 293
 294 static ftepp_t* ftepp_new(void)
 295 {
 296     ftepp_t *ftepp;
 297
 298     ftepp = (ftepp_t*)mem_a(sizeof(*ftepp));
 299     memset(ftepp, 0, sizeof(*ftepp));
 300
 301     ftepp->macros          = util_htnew(HT_MACROS);
 302     ftepp->output_on       = true;
 303     ftepp->predef_countval = 0;
 304     ftepp->predef_randval  = 0;
 305
 306     return ftepp;
 307 }
 308
 309 static GMQCC_INLINE void ftepp_flush_do(ftepp_t *self)
 310 {
 311     vec_free(self->output_string);
 312 }
 313
 314 static void ftepp_delete(ftepp_t *self)
 315 {
 316     ftepp_flush_do(self);
 317     if (self->itemname)
 318         mem_d(self->itemname);
 319     if (self->includename)
 320         vec_free(self->includename);
 321
 322     util_htrem(self->macros, (void (*)(void*))&ppmacro_delete);
 323
 324     vec_free(self->conditions);
 325     if (self->lex)
 326         lex_close(self->lex);
 327     mem_d(self);
 328 }
 329
 330 static void ftepp_out(ftepp_t *ftepp, const char *str, bool ignore_cond)
 331 {
 332     if (ignore_cond || ftepp->output_on)
 333     {
 334         size_t len;
 335         char  *data;
 336         len = strlen(str);
 337         data = vec_add(ftepp->output_string, len);
 338         memcpy(data, str, len);
 339     }
 340 }
 341
 342 static GMQCC_INLINE void ftepp_update_output_condition(ftepp_t *ftepp)
 343 {
 344     size_t i;
 345     ftepp->output_on = true;
 346     for (i = 0; i < vec_size(ftepp->conditions); ++i)
 347         ftepp->output_on = ftepp->output_on && ftepp->conditions[i].on;
 348 }
 349
 350 static GMQCC_INLINE ppmacro* ftepp_macro_find(ftepp_t *ftepp, const char *name)
 351 {
 352     return (ppmacro*)util_htget(ftepp->macros, name);
 353 }
 354
 355 static GMQCC_INLINE void ftepp_macro_delete(ftepp_t *ftepp, const char *name)
 356 {
 357     util_htrm(ftepp->macros, name, (void (*)(void*))&ppmacro_delete);
 358 }
 359
 360 static GMQCC_INLINE int ftepp_next(ftepp_t *ftepp)
 361 {
 362     return (ftepp->token = lex_do(ftepp->lex));
 363 }
 364
 365 /* Important: this does not skip newlines! */
 366 static bool ftepp_skipspace(ftepp_t *ftepp)
 367 {
 368     if (ftepp->token != TOKEN_WHITE)
 369         return true;
 370     while (ftepp_next(ftepp) == TOKEN_WHITE) {}
 371     if (ftepp->token >= TOKEN_EOF) {
 372         ftepp_error(ftepp, "unexpected end of preprocessor directive");
 373         return false;
 374     }
 375     return true;
 376 }
 377
 378 /* this one skips EOLs as well */
 379 static bool ftepp_skipallwhite(ftepp_t *ftepp)
 380 {
 381     if (ftepp->token != TOKEN_WHITE && ftepp->token != TOKEN_EOL)
 382         return true;
 383     do {
 384         ftepp_next(ftepp);
 385     } while (ftepp->token == TOKEN_WHITE || ftepp->token == TOKEN_EOL);
 386     if (ftepp->token >= TOKEN_EOF) {
 387         ftepp_error(ftepp, "unexpected end of preprocessor directive");
 388         return false;
 389     }
 390     return true;
 391 }
 392
 393 /**
 394  * The huge macro parsing code...
 395  */
 396 static bool ftepp_define_params(ftepp_t *ftepp, ppmacro *macro)
 397 {
 398     do {
 399         ftepp_next(ftepp);
 400         if (!ftepp_skipspace(ftepp))
 401             return false;
 402         if (ftepp->token == ')')
 403             break;
 404         switch (ftepp->token) {
 405             case TOKEN_IDENT:
 406             case TOKEN_TYPENAME:
 407             case TOKEN_KEYWORD:
 408                 vec_push(macro->params, util_strdup(ftepp_tokval(ftepp)));
 409                 break;
 410             case TOKEN_DOTS:
 411                 macro->variadic = true;
 412                 break;
 413             default:
 414                 ftepp_error(ftepp, "unexpected token in parameter list");
 415                 return false;
 416         }
 417         ftepp_next(ftepp);
 418         if (!ftepp_skipspace(ftepp))
 419             return false;
 420         if (macro->variadic && ftepp->token != ')') {
 421             ftepp_error(ftepp, "cannot have parameters after the variadic parameters");
 422             return false;
 423         }
 424     } while (ftepp->token == ',');
 425
 426     if (ftepp->token != ')') {
 427         ftepp_error(ftepp, "expected closing paren after macro parameter list");
 428         return false;
 429     }
 430     ftepp_next(ftepp);
 431     /* skipspace happens in ftepp_define */
 432     return true;
 433 }
 434
 435 static bool ftepp_define_body(ftepp_t *ftepp, ppmacro *macro)
 436 {
 437     pptoken *ptok;
 438     while (ftepp->token != TOKEN_EOL && ftepp->token < TOKEN_EOF) {
 439         bool   subscript = false;
 440         size_t index     = 0;
 441         if (macro->variadic && !strcmp(ftepp_tokval(ftepp), "__VA_ARGS__")) {
 442             subscript = !!(ftepp_next(ftepp) == '#');
 443
 444             if (subscript && ftepp_next(ftepp) != '#') {
 445                 ftepp_error(ftepp, "expected `##` in __VA_ARGS__ for subscripting");
 446                 return false;
 447             } else if (subscript) {
 448                 if (ftepp_next(ftepp) == '[') {
 449                     if (ftepp_next(ftepp) != TOKEN_INTCONST) {
 450                         ftepp_error(ftepp, "expected index for __VA_ARGS__ subscript");
 451                         return false;
 452                     }
 453
 454                     index = (int)strtol(ftepp_tokval(ftepp), NULL, 10);
 455
 456                     if (ftepp_next(ftepp) != ']') {
 457                         ftepp_error(ftepp, "expected `]` in __VA_ARGS__ subscript");
 458                         return false;
 459                     }
 460
 461                     /*
 462                      * mark it as an array to be handled later as such and not
 463                      * as traditional __VA_ARGS__
 464                      */
 465                     ftepp->token = TOKEN_VA_ARGS_ARRAY;
 466                     ptok = pptoken_make(ftepp);
 467                     ptok->constval.i = index;
 468                     vec_push(macro->output, ptok);
 469                     ftepp_next(ftepp);
 470                 } else {
 471                     ftepp_error(ftepp, "expected `[` for subscripting of __VA_ARGS__");
 472                     return false;
 473                 }
 474             } else {
 475                 int old = ftepp->token;
 476                 ftepp->token = TOKEN_VA_ARGS;
 477                 ptok = pptoken_make(ftepp);
 478                 vec_push(macro->output, ptok);
 479                 ftepp->token = old;
 480             }
 481         }
 482         else if (macro->variadic && !strcmp(ftepp_tokval(ftepp), "__VA_COUNT__")) {
 483             ftepp->token = TOKEN_VA_COUNT;
 484             ptok         = pptoken_make(ftepp);
 485             vec_push(macro->output, ptok);
 486             ftepp_next(ftepp);
 487         } else {
 488             ptok = pptoken_make(ftepp);
 489             vec_push(macro->output, ptok);
 490             ftepp_next(ftepp);
 491         }
 492     }
 493     /* recursive expansion can cause EOFs here */
 494     if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
 495         ftepp_error(ftepp, "unexpected junk after macro or unexpected end of file");
 496         return false;
 497     }
 498     return true;
 499 }
 500
 501 static const char *ftepp_math_constants[][2] = {
 502     { "M_E",        "2.7182818284590452354"  }, /* e          */
 503     { "M_LOG2E",    "1.4426950408889634074"  }, /* log_2 e    */
 504     { "M_LOG10E",   "0.43429448190325182765" }, /* log_10 e   */
 505     { "M_LN2",      "0.69314718055994530942" }, /* log_e 2    */
 506     { "M_LN10",     "2.30258509299404568402" }, /* log_e 10   */
 507     { "M_PI",       "3.14159265358979323846" }, /* pi         */
 508     { "M_PI_2",     "1.57079632679489661923" }, /* pi/2       */
 509     { "M_PI_4",     "0.78539816339744830962" }, /* pi/4       */
 510     { "M_1_PI",     "0.31830988618379067154" }, /* 1/pi       */
 511     { "M_2_PI",     "0.63661977236758134308" }, /* 2/pi       */
 512     { "M_2_SQRTPI", "1.12837916709551257390" }, /* 2/sqrt(pi) */
 513     { "M_SQRT2",    "1.41421356237309504880" }, /* sqrt(2)    */
 514     { "M_SQRT1_2",  "0.70710678118654752440" }, /* 1/sqrt(2)  */
 515     { "M_TAU",      "6.28318530717958647692" }  /* pi*2       */
 516 };
 517
 518 static bool ftepp_define(ftepp_t *ftepp)
 519 {
 520     ppmacro *macro = NULL;
 521     size_t l = ftepp_ctx(ftepp).line;
 522     size_t i;
 523     bool   mathconstant = false;
 524
 525     (void)ftepp_next(ftepp);
 526     if (!ftepp_skipspace(ftepp))
 527         return false;
 528
 529     switch (ftepp->token) {
 530         case TOKEN_IDENT:
 531         case TOKEN_TYPENAME:
 532         case TOKEN_KEYWORD:
 533             if (OPTS_FLAG(FTEPP_MATHDEFS)) {
 534                 for (i = 0; i < GMQCC_ARRAY_COUNT(ftepp_math_constants); i++) {
 535                     if (!strcmp(ftepp_math_constants[i][0], ftepp_tokval(ftepp))) {
 536                         mathconstant = true;
 537                         break;
 538                     }
 539                 }
 540             }
 541
 542             macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
 543
 544             if (OPTS_FLAG(FTEPP_MATHDEFS)) {
 545                 /* user defined ones take precedence */
 546                 if (macro && mathconstant) {
 547                     ftepp_macro_delete(ftepp, ftepp_tokval(ftepp));
 548                     macro = NULL;
 549                 }
 550             }
 551
 552             if (macro && ftepp->output_on) {
 553                 if (ftepp_warn(ftepp, WARN_CPP, "redefining `%s`", ftepp_tokval(ftepp)))
 554                     return false;
 555                 ftepp_macro_delete(ftepp, ftepp_tokval(ftepp));
 556             }
 557             macro = ppmacro_new(ftepp_ctx(ftepp), ftepp_tokval(ftepp));
 558             break;
 559         default:
 560             ftepp_error(ftepp, "expected macro name");
 561             return false;
 562     }
 563
 564     (void)ftepp_next(ftepp);
 565
 566     if (ftepp->token == '(') {
 567         macro->has_params = true;
 568         if (!ftepp_define_params(ftepp, macro)) {
 569             ppmacro_delete(macro);
 570             return false;
 571         }
 572     }
 573
 574     if (!ftepp_skipspace(ftepp)) {
 575         ppmacro_delete(macro);
 576         return false;
 577     }
 578
 579     if (!ftepp_define_body(ftepp, macro)) {
 580         ppmacro_delete(macro);
 581         return false;
 582     }
 583
 584     if (ftepp->output_on)
 585         util_htset(ftepp->macros, macro->name, (void*)macro);
 586     else {
 587         ppmacro_delete(macro);
 588     }
 589
 590     for (; l < ftepp_ctx(ftepp).line; ++l)
 591         ftepp_out(ftepp, "\n", true);
 592     return true;
 593 }
 594
 595 /**
 596  * When a macro is used we have to handle parameters as well
 597  * as special-concatenation via ## or stringification via #
 598  *
 599  * Note: parenthesis can nest, so FOO((a),b) is valid, but only
 600  * this kind of parens. Curly braces or [] don't count towards the
 601  * paren-level.
 602  */
 603 typedef struct {
 604     pptoken **tokens;
 605 } macroparam;
 606
 607 static void macroparam_clean(macroparam *self)
 608 {
 609     size_t i;
 610     for (i = 0; i < vec_size(self->tokens); ++i)
 611         pptoken_delete(self->tokens[i]);
 612     vec_free(self->tokens);
 613 }
 614
 615 /* need to leave the last token up */
 616 static bool ftepp_macro_call_params(ftepp_t *ftepp, macroparam **out_params)
 617 {
 618     macroparam *params = NULL;
 619     pptoken    *ptok;
 620     macroparam  mp;
 621     size_t      parens = 0;
 622     size_t      i;
 623
 624     if (!ftepp_skipallwhite(ftepp))
 625         return false;
 626     while (ftepp->token != ')') {
 627         mp.tokens = NULL;
 628         if (!ftepp_skipallwhite(ftepp))
 629             return false;
 630         while (parens || ftepp->token != ',') {
 631             if (ftepp->token == '(')
 632                 ++parens;
 633             else if (ftepp->token == ')') {
 634                 if (!parens)
 635                     break;
 636                 --parens;
 637             }
 638             ptok = pptoken_make(ftepp);
 639             vec_push(mp.tokens, ptok);
 640             if (ftepp_next(ftepp) >= TOKEN_EOF) {
 641                 ftepp_error(ftepp, "unexpected end of file in macro call");
 642                 goto on_error;
 643             }
 644         }
 645         vec_push(params, mp);
 646         mp.tokens = NULL;
 647         if (ftepp->token == ')')
 648             break;
 649         if (ftepp->token != ',') {
 650             ftepp_error(ftepp, "expected closing paren or comma in macro call");
 651             goto on_error;
 652         }
 653         if (ftepp_next(ftepp) >= TOKEN_EOF) {
 654             ftepp_error(ftepp, "unexpected end of file in macro call");
 655             goto on_error;
 656         }
 657     }
 658     *out_params = params;
 659     return true;
 660
 661 on_error:
 662     if (mp.tokens)
 663         macroparam_clean(&mp);
 664     for (i = 0; i < vec_size(params); ++i)
 665         macroparam_clean(&params[i]);
 666     vec_free(params);
 667     return false;
 668 }
 669
 670 static bool macro_params_find(ppmacro *macro, const char *name, size_t *idx)
 671 {
 672     size_t i;
 673     for (i = 0; i < vec_size(macro->params); ++i) {
 674         if (!strcmp(macro->params[i], name)) {
 675             *idx = i;
 676             return true;
 677         }
 678     }
 679     return false;
 680 }
 681
 682 static void ftepp_stringify_token(ftepp_t *ftepp, pptoken *token)
 683 {
 684     char        chs[2];
 685     const char *ch;
 686     chs[1] = 0;
 687     switch (token->token) {
 688         case TOKEN_STRINGCONST:
 689             ch = token->value;
 690             while (*ch) {
 691                 /* in preprocessor mode strings already are string,
 692                  * so we don't get actual newline bytes here.
 693                  * Still need to escape backslashes and quotes.
 694                  */
 695                 switch (*ch) {
 696                     case '\\': ftepp_out(ftepp, "\\\\", false); break;
 697                     case '"':  ftepp_out(ftepp, "\\\"", false); break;
 698                     default:
 699                         chs[0] = *ch;
 700                         ftepp_out(ftepp, chs, false);
 701                         break;
 702                 }
 703                 ++ch;
 704             }
 705             break;
 706         case TOKEN_WHITE:
 707             ftepp_out(ftepp, " ", false);
 708             break;
 709         case TOKEN_EOL:
 710             ftepp_out(ftepp, "\\n", false);
 711             break;
 712         default:
 713             ftepp_out(ftepp, token->value, false);
 714             break;
 715     }
 716 }
 717
 718 static void ftepp_stringify(ftepp_t *ftepp, macroparam *param)
 719 {
 720     size_t i;
 721     ftepp_out(ftepp, "\"", false);
 722     for (i = 0; i < vec_size(param->tokens); ++i)
 723         ftepp_stringify_token(ftepp, param->tokens[i]);
 724     ftepp_out(ftepp, "\"", false);
 725 }
 726
 727 static void ftepp_recursion_header(ftepp_t *ftepp)
 728 {
 729     ftepp_out(ftepp, "\n#pragma push(line)\n", false);
 730 }
 731
 732 static void ftepp_recursion_footer(ftepp_t *ftepp)
 733 {
 734     ftepp_out(ftepp, "\n#pragma pop(line)\n", false);
 735 }
 736
 737 static void ftepp_param_out(ftepp_t *ftepp, macroparam *param)
 738 {
 739     size_t   i;
 740     pptoken *out;
 741     for (i = 0; i < vec_size(param->tokens); ++i) {
 742         out = param->tokens[i];
 743         if (out->token == TOKEN_EOL)
 744             ftepp_out(ftepp, "\n", false);
 745         else
 746             ftepp_out(ftepp, out->value, false);
 747     }
 748 }
 749
 750 static bool ftepp_preprocess(ftepp_t *ftepp);
 751 static bool ftepp_macro_expand(ftepp_t *ftepp, ppmacro *macro, macroparam *params, bool resetline)
 752 {
 753     char     *buffer       = NULL;
 754     char     *old_string   = ftepp->output_string;
 755     char     *inner_string;
 756     lex_file *old_lexer    = ftepp->lex;
 757     size_t    vararg_start = vec_size(macro->params);
 758     bool      retval       = true;
 759     bool      has_newlines;
 760     size_t    varargs;
 761
 762     size_t    o, pi;
 763     lex_file *inlex;
 764
 765     bool      old_inmacro;
 766     bool      strip = false;
 767
 768     int nextok;
 769
 770     if (vararg_start < vec_size(params))
 771         varargs = vec_size(params) - vararg_start;
 772     else
 773         varargs = 0;
 774
 775     /* really ... */
 776     if (!vec_size(macro->output))
 777         return true;
 778
 779     ftepp->output_string = NULL;
 780     for (o = 0; o < vec_size(macro->output); ++o) {
 781         pptoken *out = macro->output[o];
 782         switch (out->token) {
 783             case TOKEN_VA_ARGS:
 784                 if (!macro->variadic) {
 785                     ftepp_error(ftepp, "internal preprocessor error: TOKEN_VA_ARGS in non-variadic macro");
 786                     vec_free(old_string);
 787                     return false;
 788                 }
 789                 if (!varargs)
 790                     break;
 791
 792                 pi = 0;
 793                 ftepp_param_out(ftepp, &params[pi + vararg_start]);
 794                 for (++pi; pi < varargs; ++pi) {
 795                     ftepp_out(ftepp, ", ", false);
 796                     ftepp_param_out(ftepp, &params[pi + vararg_start]);
 797                 }
 798                 break;
 799
 800             case TOKEN_VA_ARGS_ARRAY:
 801                 if ((size_t)out->constval.i >= varargs) {
 802                     ftepp_error(ftepp, "subscript of `[%u]` is out of bounds for `__VA_ARGS__`", out->constval.i);
 803                     vec_free(old_string);
 804                     return false;
 805                 }
 806
 807                 ftepp_param_out(ftepp, &params[out->constval.i + vararg_start]);
 808                 break;
 809
 810             case TOKEN_VA_COUNT:
 811                 util_asprintf(&buffer, "%d", varargs);
 812                 ftepp_out(ftepp, buffer, false);
 813                 mem_d(buffer);
 814                 break;
 815
 816             case TOKEN_IDENT:
 817             case TOKEN_TYPENAME:
 818             case TOKEN_KEYWORD:
 819                 if (!macro_params_find(macro, out->value, &pi)) {
 820                     ftepp_out(ftepp, out->value, false);
 821                     break;
 822                 } else
 823                     ftepp_param_out(ftepp, &params[pi]);
 824                 break;
 825             case '#':
 826                 if (o + 1 < vec_size(macro->output)) {
 827                     nextok = macro->output[o+1]->token;
 828                     if (nextok == '#') {
 829                         /* raw concatenation */
 830                         ++o;
 831                         strip = true;
 832                         break;
 833                     }
 834                     if ( (nextok == TOKEN_IDENT    ||
 835                           nextok == TOKEN_KEYWORD  ||
 836                           nextok == TOKEN_TYPENAME) &&
 837                         macro_params_find(macro, macro->output[o+1]->value, &pi))
 838                     {
 839                         ++o;
 840
 841                         ftepp_stringify(ftepp, &params[pi]);
 842                         break;
 843                     }
 844                 }
 845                 ftepp_out(ftepp, "#", false);
 846                 break;
 847             case TOKEN_EOL:
 848                 ftepp_out(ftepp, "\n", false);
 849                 break;
 850             default:
 851                 buffer = out->value;
 852                 if (vec_size(macro->output) > o + 1 && macro->output[o+1]->token == '#')
 853                     buffer++;
 854                 if (strip) {
 855                     while (*buffer == ' ' || *buffer == '\t') buffer++;
 856                     strip = false;
 857                 }
 858                 ftepp_out(ftepp, buffer, false);
 859                 break;
 860         }
 861     }
 862     vec_push(ftepp->output_string, 0);
 863     /* Now run the preprocessor recursively on this string buffer */
 864     /*
 865     printf("__________\n%s\n=========\n", ftepp->output_string);
 866     */
 867     inlex = lex_open_string(ftepp->output_string, vec_size(ftepp->output_string)-1, ftepp->lex->name);
 868     if (!inlex) {
 869         ftepp_error(ftepp, "internal error: failed to instantiate lexer");
 870         retval = false;
 871         goto cleanup;
 872     }
 873
 874     inlex->line  = ftepp->lex->line;
 875     inlex->sline = ftepp->lex->sline;
 876     ftepp->lex   = inlex;
 877
 878     old_inmacro     = ftepp->in_macro;
 879     ftepp->in_macro = true;
 880     ftepp->output_string = NULL;
 881     if (!ftepp_preprocess(ftepp)) {
 882         ftepp->in_macro = old_inmacro;
 883         vec_free(ftepp->lex->open_string);
 884         vec_free(ftepp->output_string);
 885         lex_close(ftepp->lex);
 886         retval = false;
 887         goto cleanup;
 888     }
 889     ftepp->in_macro = old_inmacro;
 890     vec_free(ftepp->lex->open_string);
 891     lex_close(ftepp->lex);
 892
 893     inner_string = ftepp->output_string;
 894     ftepp->output_string = old_string;
 895
 896     has_newlines = (strchr(inner_string, '\n') != NULL);
 897
 898     if (has_newlines && !old_inmacro)
 899         ftepp_recursion_header(ftepp);
 900
 901     vec_append(ftepp->output_string, vec_size(inner_string), inner_string);
 902     vec_free(inner_string);
 903
 904     if (has_newlines && !old_inmacro)
 905         ftepp_recursion_footer(ftepp);
 906
 907     if (resetline && !ftepp->in_macro) {
 908         char lineno[128];
 909         util_snprintf(lineno, 128, "\n#pragma line(%lu)\n", (unsigned long)(old_lexer->sline));
 910         ftepp_out(ftepp, lineno, false);
 911     }
 912
 913     old_string = ftepp->output_string;
 914 cleanup:
 915     ftepp->lex           = old_lexer;
 916     ftepp->output_string = old_string;
 917     return retval;
 918 }
 919
 920 static bool ftepp_macro_call(ftepp_t *ftepp, ppmacro *macro)
 921 {
 922     size_t     o;
 923     macroparam *params = NULL;
 924     bool        retval = true;
 925     size_t      paramline;
 926
 927     if (!macro->has_params) {
 928         if (!ftepp_macro_expand(ftepp, macro, NULL, false))
 929             return false;
 930         ftepp_next(ftepp);
 931         return true;
 932     }
 933     ftepp_next(ftepp);
 934
 935     if (!ftepp_skipallwhite(ftepp))
 936         return false;
 937
 938     if (ftepp->token != '(') {
 939         ftepp_error(ftepp, "expected macro parameters in parenthesis");
 940         return false;
 941     }
 942
 943     ftepp_next(ftepp);
 944     paramline = ftepp->lex->sline;
 945     if (!ftepp_macro_call_params(ftepp, &params))
 946         return false;
 947
 948     if ( vec_size(params) < vec_size(macro->params) ||
 949         (vec_size(params) > vec_size(macro->params) && !macro->variadic) )
 950     {
 951         ftepp_error(ftepp, "macro %s expects%s %u paramteters, %u provided", macro->name,
 952                     (macro->variadic ? " at least" : ""),
 953                     (unsigned int)vec_size(macro->params),
 954                     (unsigned int)vec_size(params));
 955         retval = false;
 956         goto cleanup;
 957     }
 958
 959     if (!ftepp_macro_expand(ftepp, macro, params, (paramline != ftepp->lex->sline)))
 960         retval = false;
 961     ftepp_next(ftepp);
 962
 963 cleanup:
 964     for (o = 0; o < vec_size(params); ++o)
 965         macroparam_clean(&params[o]);
 966     vec_free(params);
 967     return retval;
 968 }
 969
 970 /**
 971  * #if - the FTEQCC way:
 972  *    defined(FOO) => true if FOO was #defined regardless of parameters or contents
 973  *    <numbers>    => True if the number is not 0
 974  *    !<factor>    => True if the factor yields false
 975  *    !!<factor>   => ERROR on 2 or more unary nots
 976  *    <macro>      => becomes the macro's FIRST token regardless of parameters
 977  *    <e> && <e>   => True if both expressions are true
 978  *    <e> || <e>   => True if either expression is true
 979  *    <string>     => False
 980  *    <ident>      => False (remember for macros the <macro> rule applies instead)
 981  * Unary + and - are weird and wrong in fteqcc so we don't allow them
 982  * parenthesis in expressions are allowed
 983  * parameter lists on macros are errors
 984  * No mathematical calculations are executed
 985  */
 986 static bool ftepp_if_expr(ftepp_t *ftepp, bool *out, double *value_out);
 987 static bool ftepp_if_op(ftepp_t *ftepp)
 988 {
 989     ftepp->lex->flags.noops = false;
 990     ftepp_next(ftepp);
 991     if (!ftepp_skipspace(ftepp))
 992         return false;
 993     ftepp->lex->flags.noops = true;
 994     return true;
 995 }
 996 static bool ftepp_if_value(ftepp_t *ftepp, bool *out, double *value_out)
 997 {
 998     ppmacro *macro;
 999     bool     wasnot = false;
1000     bool     wasneg = false;
1001
1002     if (!ftepp_skipspace(ftepp))
1003         return false;
1004
1005     while (ftepp->token == '!') {
1006         wasnot = true;
1007         ftepp_next(ftepp);
1008         if (!ftepp_skipspace(ftepp))
1009             return false;
1010     }
1011
1012     if (ftepp->token == TOKEN_OPERATOR && !strcmp(ftepp_tokval(ftepp), "-"))
1013     {
1014         wasneg = true;
1015         ftepp_next(ftepp);
1016         if (!ftepp_skipspace(ftepp))
1017             return false;
1018     }
1019
1020     switch (ftepp->token) {
1021         case TOKEN_IDENT:
1022         case TOKEN_TYPENAME:
1023         case TOKEN_KEYWORD:
1024             if (!strcmp(ftepp_tokval(ftepp), "defined")) {
1025                 ftepp_next(ftepp);
1026                 if (!ftepp_skipspace(ftepp))
1027                     return false;
1028                 if (ftepp->token != '(') {
1029                     ftepp_error(ftepp, "`defined` keyword in #if requires a macro name in parenthesis");
1030                     return false;
1031                 }
1032                 ftepp_next(ftepp);
1033                 if (!ftepp_skipspace(ftepp))
1034                     return false;
1035                 if (ftepp->token != TOKEN_IDENT &&
1036                     ftepp->token != TOKEN_TYPENAME &&
1037                     ftepp->token != TOKEN_KEYWORD)
1038                 {
1039                     ftepp_error(ftepp, "defined() used on an unexpected token type");
1040                     return false;
1041                 }
1042                 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1043                 *out = !!macro;
1044                 ftepp_next(ftepp);
1045                 if (!ftepp_skipspace(ftepp))
1046                     return false;
1047                 if (ftepp->token != ')') {
1048                     ftepp_error(ftepp, "expected closing paren");
1049                     return false;
1050                 }
1051                 break;
1052             }
1053
1054             macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1055             if (!macro || !vec_size(macro->output)) {
1056                 *out = false;
1057                 *value_out = 0;
1058             } else {
1059                 /* This does not expand recursively! */
1060                 switch (macro->output[0]->token) {
1061                     case TOKEN_INTCONST:
1062                         *value_out = macro->output[0]->constval.i;
1063                         *out = !!(macro->output[0]->constval.i);
1064                         break;
1065                     case TOKEN_FLOATCONST:
1066                         *value_out = macro->output[0]->constval.f;
1067                         *out = !!(macro->output[0]->constval.f);
1068                         break;
1069                     default:
1070                         *out = false;
1071                         break;
1072                 }
1073             }
1074             break;
1075         case TOKEN_STRINGCONST:
1076             *value_out = 0;
1077             *out = false;
1078             break;
1079         case TOKEN_INTCONST:
1080             *value_out = ftepp->lex->tok.constval.i;
1081             *out = !!(ftepp->lex->tok.constval.i);
1082             break;
1083         case TOKEN_FLOATCONST:
1084             *value_out = ftepp->lex->tok.constval.f;
1085             *out = !!(ftepp->lex->tok.constval.f);
1086             break;
1087
1088         case '(':
1089             ftepp_next(ftepp);
1090             if (!ftepp_if_expr(ftepp, out, value_out))
1091                 return false;
1092             if (ftepp->token != ')') {
1093                 ftepp_error(ftepp, "expected closing paren in #if expression");
1094                 return false;
1095             }
1096             break;
1097
1098         default:
1099             ftepp_error(ftepp, "junk in #if: `%s` ...", ftepp_tokval(ftepp));
1100             if (OPTS_OPTION_BOOL(OPTION_DEBUG))
1101                 ftepp_error(ftepp, "internal: token %i\n", ftepp->token);
1102             return false;
1103     }
1104     if (wasneg)
1105         *value_out = -*value_out;
1106     if (wasnot) {
1107         *out = !*out;
1108         *value_out = (*out ? 1 : 0);
1109     }
1110     return true;
1111 }
1112
1113 /*
1114 static bool ftepp_if_nextvalue(ftepp_t *ftepp, bool *out, double *value_out)
1115 {
1116     if (!ftepp_next(ftepp))
1117         return false;
1118     return ftepp_if_value(ftepp, out, value_out);
1119 }
1120 */
1121
1122 static bool ftepp_if_expr(ftepp_t *ftepp, bool *out, double *value_out)
1123 {
1124     if (!ftepp_if_value(ftepp, out, value_out))
1125         return false;
1126
1127     if (!ftepp_if_op(ftepp))
1128         return false;
1129
1130     if (ftepp->token == ')' || ftepp->token != TOKEN_OPERATOR)
1131         return true;
1132
1133     /* FTEQCC is all right-associative and no precedence here */
1134     if (!strcmp(ftepp_tokval(ftepp), "&&") ||
1135         !strcmp(ftepp_tokval(ftepp), "||"))
1136     {
1137         bool next = false;
1138         char opc  = ftepp_tokval(ftepp)[0];
1139         double nextvalue;
1140
1141         (void)nextvalue;
1142         if (!ftepp_next(ftepp))
1143             return false;
1144         if (!ftepp_if_expr(ftepp, &next, &nextvalue))
1145             return false;
1146
1147         if (opc == '&')
1148             *out = *out && next;
1149         else
1150             *out = *out || next;
1151
1152         *value_out = (*out ? 1 : 0);
1153         return true;
1154     }
1155     else if (!strcmp(ftepp_tokval(ftepp), "==") ||
1156              !strcmp(ftepp_tokval(ftepp), "!=") ||
1157              !strcmp(ftepp_tokval(ftepp), ">=") ||
1158              !strcmp(ftepp_tokval(ftepp), "<=") ||
1159              !strcmp(ftepp_tokval(ftepp), ">") ||
1160              !strcmp(ftepp_tokval(ftepp), "<"))
1161     {
1162         bool next = false;
1163         const char opc0 = ftepp_tokval(ftepp)[0];
1164         const char opc1 = ftepp_tokval(ftepp)[1];
1165         double other;
1166
1167         if (!ftepp_next(ftepp))
1168             return false;
1169         if (!ftepp_if_expr(ftepp, &next, &other))
1170             return false;
1171
1172         if (opc0 == '=')
1173             *out = (*value_out == other);
1174         else if (opc0 == '!')
1175             *out = (*value_out != other);
1176         else if (opc0 == '>') {
1177             if (opc1 == '=') *out = (*value_out >= other);
1178             else             *out = (*value_out > other);
1179         }
1180         else if (opc0 == '<') {
1181             if (opc1 == '=') *out = (*value_out <= other);
1182             else             *out = (*value_out < other);
1183         }
1184         *value_out = (*out ? 1 : 0);
1185
1186         return true;
1187     }
1188     else {
1189         ftepp_error(ftepp, "junk after #if");
1190         return false;
1191     }
1192 }
1193
1194 static bool ftepp_if(ftepp_t *ftepp, ppcondition *cond)
1195 {
1196     bool result = false;
1197     double dummy = 0;
1198
1199     memset(cond, 0, sizeof(*cond));
1200     (void)ftepp_next(ftepp);
1201
1202     if (!ftepp_skipspace(ftepp))
1203         return false;
1204     if (ftepp->token == TOKEN_EOL) {
1205         ftepp_error(ftepp, "expected expression for #if-directive");
1206         return false;
1207     }
1208
1209     if (!ftepp_if_expr(ftepp, &result, &dummy))
1210         return false;
1211
1212     cond->on = result;
1213     return true;
1214 }
1215
1216 /**
1217  * ifdef is rather simple
1218  */
1219 static bool ftepp_ifdef(ftepp_t *ftepp, ppcondition *cond)
1220 {
1221     ppmacro *macro;
1222     memset(cond, 0, sizeof(*cond));
1223     (void)ftepp_next(ftepp);
1224     if (!ftepp_skipspace(ftepp))
1225         return false;
1226
1227     switch (ftepp->token) {
1228         case TOKEN_IDENT:
1229         case TOKEN_TYPENAME:
1230         case TOKEN_KEYWORD:
1231             macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1232             break;
1233         default:
1234             ftepp_error(ftepp, "expected macro name");
1235             return false;
1236     }
1237
1238     (void)ftepp_next(ftepp);
1239     if (!ftepp_skipspace(ftepp))
1240         return false;
1241     /* relaxing this condition
1242     if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
1243         ftepp_error(ftepp, "stray tokens after #ifdef");
1244         return false;
1245     }
1246     */
1247     cond->on = !!macro;
1248     return true;
1249 }
1250
1251 /**
1252  * undef is also simple
1253  */
1254 static bool ftepp_undef(ftepp_t *ftepp)
1255 {
1256     (void)ftepp_next(ftepp);
1257     if (!ftepp_skipspace(ftepp))
1258         return false;
1259
1260     if (ftepp->output_on) {
1261         switch (ftepp->token) {
1262             case TOKEN_IDENT:
1263             case TOKEN_TYPENAME:
1264             case TOKEN_KEYWORD:
1265                 ftepp_macro_delete(ftepp, ftepp_tokval(ftepp));
1266                 break;
1267             default:
1268                 ftepp_error(ftepp, "expected macro name");
1269                 return false;
1270         }
1271     }
1272
1273     (void)ftepp_next(ftepp);
1274     if (!ftepp_skipspace(ftepp))
1275         return false;
1276     /* relaxing this condition
1277     if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
1278         ftepp_error(ftepp, "stray tokens after #ifdef");
1279         return false;
1280     }
1281     */
1282     return true;
1283 }
1284
1285 /* Special unescape-string function which skips a leading quote
1286  * and stops at a quote, not just at \0
1287  */
1288 static void unescape(const char *str, char *out) {
1289     ++str;
1290     while (*str && *str != '"') {
1291         if (*str == '\\') {
1292             ++str;
1293             switch (*str) {
1294                 case '\\': *out++ = *str; break;
1295                 case '"':  *out++ = *str; break;
1296                 case 'a':  *out++ = '\a'; break;
1297                 case 'b':  *out++ = '\b'; break;
1298                 case 'r':  *out++ = '\r'; break;
1299                 case 'n':  *out++ = '\n'; break;
1300                 case 't':  *out++ = '\t'; break;
1301                 case 'f':  *out++ = '\f'; break;
1302                 case 'v':  *out++ = '\v'; break;
1303                 default:
1304                     *out++ = '\\';
1305                     *out++ = *str;
1306                     break;
1307             }
1308             ++str;
1309             continue;
1310         }
1311
1312         *out++ = *str++;
1313     }
1314     *out = 0;
1315 }
1316
1317 static char *ftepp_include_find_path(const char *file, const char *pathfile)
1318 {
1319     fs_file_t  *fp;
1320     char       *filename = NULL;
1321     const char *last_slash;
1322     size_t      len;
1323
1324     if (!pathfile)
1325         return NULL;
1326
1327     last_slash = strrchr(pathfile, '/');
1328
1329     if (last_slash) {
1330         len = last_slash - pathfile;
1331         memcpy(vec_add(filename, len), pathfile, len);
1332         vec_push(filename, '/');
1333     }
1334
1335     len = strlen(file);
1336     memcpy(vec_add(filename, len+1), file, len);
1337     vec_last(filename) = 0;
1338
1339     fp = fs_file_open(filename, "rb");
1340     if (fp) {
1341         fs_file_close(fp);
1342         return filename;
1343     }
1344     vec_free(filename);
1345     return NULL;
1346 }
1347
1348 static char *ftepp_include_find(ftepp_t *ftepp, const char *file)
1349 {
1350     char *filename = NULL;
1351
1352     filename = ftepp_include_find_path(file, ftepp->includename);
1353     if (!filename)
1354         filename = ftepp_include_find_path(file, ftepp->itemname);
1355     return filename;
1356 }
1357
1358 static bool ftepp_directive_warning(ftepp_t *ftepp) {
1359     char *message = NULL;
1360
1361     if (!ftepp_skipspace(ftepp))
1362         return false;
1363
1364     /* handle the odd non string constant case so it works like C */
1365     if (ftepp->token != TOKEN_STRINGCONST) {
1366         bool  store   = false;
1367         vec_append(message, 8, "#warning");
1368         ftepp_next(ftepp);
1369         while (ftepp->token != TOKEN_EOL) {
1370             vec_append(message, strlen(ftepp_tokval(ftepp)), ftepp_tokval(ftepp));
1371             ftepp_next(ftepp);
1372         }
1373         vec_push(message, '\0');
1374         if (ftepp->output_on)
1375             store = ftepp_warn(ftepp, WARN_CPP, message);
1376         else
1377             store = false;
1378         vec_free(message);
1379         return store;
1380     }
1381
1382     if (!ftepp->output_on)
1383         return false;
1384
1385     unescape  (ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1386     return ftepp_warn(ftepp, WARN_CPP, "#warning %s", ftepp_tokval(ftepp));
1387 }
1388
1389 static void ftepp_directive_error(ftepp_t *ftepp) {
1390     char *message = NULL;
1391
1392     if (!ftepp_skipspace(ftepp))
1393         return;
1394
1395     /* handle the odd non string constant case so it works like C */
1396     if (ftepp->token != TOKEN_STRINGCONST) {
1397         vec_append(message, 6, "#error");
1398         ftepp_next(ftepp);
1399         while (ftepp->token != TOKEN_EOL) {
1400             vec_append(message, strlen(ftepp_tokval(ftepp)), ftepp_tokval(ftepp));
1401             ftepp_next(ftepp);
1402         }
1403         vec_push(message, '\0');
1404         if (ftepp->output_on)
1405             ftepp_error(ftepp, message);
1406         vec_free(message);
1407         return;
1408     }
1409
1410     if (!ftepp->output_on)
1411         return;
1412
1413     unescape  (ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1414     ftepp_error(ftepp, "#error %s", ftepp_tokval(ftepp));
1415 }
1416
1417 static void ftepp_directive_message(ftepp_t *ftepp) {
1418     char *message = NULL;
1419
1420     if (!ftepp_skipspace(ftepp))
1421         return;
1422
1423     /* handle the odd non string constant case so it works like C */
1424     if (ftepp->token != TOKEN_STRINGCONST) {
1425         vec_append(message, 8, "#message");
1426         ftepp_next(ftepp);
1427         while (ftepp->token != TOKEN_EOL) {
1428             vec_append(message, strlen(ftepp_tokval(ftepp)), ftepp_tokval(ftepp));
1429             ftepp_next(ftepp);
1430         }
1431         vec_push(message, '\0');
1432         if (ftepp->output_on)
1433             con_cprintmsg(ftepp->lex->tok.ctx, LVL_MSG, "message", message);
1434         vec_free(message);
1435         return;
1436     }
1437
1438     if (!ftepp->output_on)
1439         return;
1440
1441     unescape     (ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1442     con_cprintmsg(ftepp->lex->tok.ctx, LVL_MSG, "message",  ftepp_tokval(ftepp));
1443 }
1444
1445 /**
1446  * Include a file.
1447  * FIXME: do we need/want a -I option?
1448  * FIXME: what about when dealing with files in subdirectories coming from a progs.src?
1449  */
1450 static bool ftepp_include(ftepp_t *ftepp)
1451 {
1452     lex_file *old_lexer = ftepp->lex;
1453     lex_file *inlex;
1454     lex_ctx_t ctx;
1455     char     lineno[128];
1456     char     *filename;
1457     char     *old_includename;
1458
1459     (void)ftepp_next(ftepp);
1460     if (!ftepp_skipspace(ftepp))
1461         return false;
1462
1463     if (ftepp->token != TOKEN_STRINGCONST) {
1464         ftepp_error(ftepp, "expected filename to include");
1465         return false;
1466     }
1467
1468     if (!ftepp->output_on) {
1469         ftepp_next(ftepp);
1470         return true;
1471     }
1472
1473     ctx = ftepp_ctx(ftepp);
1474
1475     unescape(ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1476
1477     ftepp_out(ftepp, "\n#pragma file(", false);
1478     ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1479     ftepp_out(ftepp, ")\n#pragma line(1)\n", false);
1480
1481     filename = ftepp_include_find(ftepp, ftepp_tokval(ftepp));
1482     if (!filename) {
1483         ftepp_error(ftepp, "failed to open include file `%s`", ftepp_tokval(ftepp));
1484         return false;
1485     }
1486     inlex = lex_open(filename);
1487     if (!inlex) {
1488         ftepp_error(ftepp, "open failed on include file `%s`", filename);
1489         vec_free(filename);
1490         return false;
1491     }
1492     ftepp->lex = inlex;
1493     old_includename = ftepp->includename;
1494     ftepp->includename = filename;
1495     if (!ftepp_preprocess(ftepp)) {
1496         vec_free(ftepp->includename);
1497         ftepp->includename = old_includename;
1498         lex_close(ftepp->lex);
1499         ftepp->lex = old_lexer;
1500         return false;
1501     }
1502     vec_free(ftepp->includename);
1503     ftepp->includename = old_includename;
1504     lex_close(ftepp->lex);
1505     ftepp->lex = old_lexer;
1506
1507     ftepp_out(ftepp, "\n#pragma file(", false);
1508     ftepp_out(ftepp, ctx.file, false);
1509     util_snprintf(lineno, sizeof(lineno), ")\n#pragma line(%lu)\n", (unsigned long)(ctx.line+1));
1510     ftepp_out(ftepp, lineno, false);
1511
1512     /* skip the line */
1513     (void)ftepp_next(ftepp);
1514     if (!ftepp_skipspace(ftepp))
1515         return false;
1516     if (ftepp->token != TOKEN_EOL) {
1517         ftepp_error(ftepp, "stray tokens after #include");
1518         return false;
1519     }
1520     (void)ftepp_next(ftepp);
1521
1522     return true;
1523 }
1524
1525 /* Basic structure handlers */
1526 static bool ftepp_else_allowed(ftepp_t *ftepp)
1527 {
1528     if (!vec_size(ftepp->conditions)) {
1529         ftepp_error(ftepp, "#else without #if");
1530         return false;
1531     }
1532     if (vec_last(ftepp->conditions).had_else) {
1533         ftepp_error(ftepp, "multiple #else for a single #if");
1534         return false;
1535     }
1536     return true;
1537 }
1538
1539 static GMQCC_INLINE void ftepp_inmacro(ftepp_t *ftepp, const char *hash) {
1540     if (ftepp->in_macro)
1541         (void)!ftepp_warn(ftepp, WARN_DIRECTIVE_INMACRO, "`#%s` directive in macro", hash);
1542 }
1543
1544 static bool ftepp_hash(ftepp_t *ftepp)
1545 {
1546     ppcondition cond;
1547     ppcondition *pc;
1548
1549     lex_ctx_t ctx = ftepp_ctx(ftepp);
1550
1551     if (!ftepp_skipspace(ftepp))
1552         return false;
1553
1554     switch (ftepp->token) {
1555         case TOKEN_KEYWORD:
1556         case TOKEN_IDENT:
1557         case TOKEN_TYPENAME:
1558             if (!strcmp(ftepp_tokval(ftepp), "define")) {
1559                 ftepp_inmacro(ftepp, "define");
1560                 return ftepp_define(ftepp);
1561             }
1562             else if (!strcmp(ftepp_tokval(ftepp), "undef")) {
1563                 ftepp_inmacro(ftepp, "undef");
1564                 return ftepp_undef(ftepp);
1565             }
1566             else if (!strcmp(ftepp_tokval(ftepp), "ifdef")) {
1567                 ftepp_inmacro(ftepp, "ifdef");
1568                 if (!ftepp_ifdef(ftepp, &cond))
1569                     return false;
1570                 cond.was_on = cond.on;
1571                 vec_push(ftepp->conditions, cond);
1572                 ftepp->output_on = ftepp->output_on && cond.on;
1573                 break;
1574             }
1575             else if (!strcmp(ftepp_tokval(ftepp), "ifndef")) {
1576                 ftepp_inmacro(ftepp, "ifndef");
1577                 if (!ftepp_ifdef(ftepp, &cond))
1578                     return false;
1579                 cond.on = !cond.on;
1580                 cond.was_on = cond.on;
1581                 vec_push(ftepp->conditions, cond);
1582                 ftepp->output_on = ftepp->output_on && cond.on;
1583                 break;
1584             }
1585             else if (!strcmp(ftepp_tokval(ftepp), "elifdef")) {
1586                 ftepp_inmacro(ftepp, "elifdef");
1587                 if (!ftepp_else_allowed(ftepp))
1588                     return false;
1589                 if (!ftepp_ifdef(ftepp, &cond))
1590                     return false;
1591                 pc = &vec_last(ftepp->conditions);
1592                 pc->on     = !pc->was_on && cond.on;
1593                 pc->was_on = pc->was_on || pc->on;
1594                 ftepp_update_output_condition(ftepp);
1595                 break;
1596             }
1597             else if (!strcmp(ftepp_tokval(ftepp), "elifndef")) {
1598                 ftepp_inmacro(ftepp, "elifndef");
1599                 if (!ftepp_else_allowed(ftepp))
1600                     return false;
1601                 if (!ftepp_ifdef(ftepp, &cond))
1602                     return false;
1603                 cond.on = !cond.on;
1604                 pc = &vec_last(ftepp->conditions);
1605                 pc->on     = !pc->was_on && cond.on;
1606                 pc->was_on = pc->was_on || pc->on;
1607                 ftepp_update_output_condition(ftepp);
1608                 break;
1609             }
1610             else if (!strcmp(ftepp_tokval(ftepp), "elif")) {
1611                 ftepp_inmacro(ftepp, "elif");
1612                 if (!ftepp_else_allowed(ftepp))
1613                     return false;
1614                 if (!ftepp_if(ftepp, &cond))
1615                     return false;
1616                 pc = &vec_last(ftepp->conditions);
1617                 pc->on     = !pc->was_on && cond.on;
1618                 pc->was_on = pc->was_on  || pc->on;
1619                 ftepp_update_output_condition(ftepp);
1620                 break;
1621             }
1622             else if (!strcmp(ftepp_tokval(ftepp), "if")) {
1623                 ftepp_inmacro(ftepp, "if");
1624                 if (!ftepp_if(ftepp, &cond))
1625                     return false;
1626                 cond.was_on = cond.on;
1627                 vec_push(ftepp->conditions, cond);
1628                 ftepp->output_on = ftepp->output_on && cond.on;
1629                 break;
1630             }
1631             else if (!strcmp(ftepp_tokval(ftepp), "else")) {
1632                 ftepp_inmacro(ftepp, "else");
1633                 if (!ftepp_else_allowed(ftepp))
1634                     return false;
1635                 pc = &vec_last(ftepp->conditions);
1636                 pc->on = !pc->was_on;
1637                 pc->had_else = true;
1638                 ftepp_next(ftepp);
1639                 ftepp_update_output_condition(ftepp);
1640                 break;
1641             }
1642             else if (!strcmp(ftepp_tokval(ftepp), "endif")) {
1643                 ftepp_inmacro(ftepp, "endif");
1644                 if (!vec_size(ftepp->conditions)) {
1645                     ftepp_error(ftepp, "#endif without #if");
1646                     return false;
1647                 }
1648                 vec_pop(ftepp->conditions);
1649                 ftepp_next(ftepp);
1650                 ftepp_update_output_condition(ftepp);
1651                 break;
1652             }
1653             else if (!strcmp(ftepp_tokval(ftepp), "include")) {
1654                 ftepp_inmacro(ftepp, "include");
1655                 return ftepp_include(ftepp);
1656             }
1657             else if (!strcmp(ftepp_tokval(ftepp), "pragma")) {
1658                 ftepp_out(ftepp, "#", false);
1659                 break;
1660             }
1661             else if (!strcmp(ftepp_tokval(ftepp), "warning")) {
1662                 ftepp_directive_warning(ftepp);
1663                 break;
1664             }
1665             else if (!strcmp(ftepp_tokval(ftepp), "error")) {
1666                 ftepp_directive_error(ftepp);
1667                 break;
1668             }
1669             else if (!strcmp(ftepp_tokval(ftepp), "message")) {
1670                 ftepp_directive_message(ftepp);
1671                 break;
1672             }
1673             else {
1674                 if (ftepp->output_on) {
1675                     ftepp_error(ftepp, "unrecognized preprocessor directive: `%s`", ftepp_tokval(ftepp));
1676                     return false;
1677                 } else {
1678                     ftepp_next(ftepp);
1679                     break;
1680                 }
1681             }
1682             /* break; never reached */
1683         default:
1684             ftepp_error(ftepp, "unexpected preprocessor token: `%s`", ftepp_tokval(ftepp));
1685             return false;
1686         case TOKEN_EOL:
1687             ftepp_errorat(ftepp, ctx, "empty preprocessor directive");
1688             return false;
1689         case TOKEN_EOF:
1690             ftepp_error(ftepp, "missing newline at end of file", ftepp_tokval(ftepp));
1691             return false;
1692
1693         /* Builtins! Don't forget the builtins! */
1694         case TOKEN_INTCONST:
1695         case TOKEN_FLOATCONST:
1696             ftepp_out(ftepp, "#", false);
1697             return true;
1698     }
1699     if (!ftepp_skipspace(ftepp))
1700         return false;
1701     return true;
1702 }
1703
1704 static bool ftepp_preprocess(ftepp_t *ftepp)
1705 {
1706     ppmacro *macro;
1707     bool     newline = true;
1708
1709     /* predef stuff */
1710     char    *expand  = NULL;
1711
1712     ftepp->lex->flags.preprocessing = true;
1713     ftepp->lex->flags.mergelines    = false;
1714     ftepp->lex->flags.noops         = true;
1715
1716     ftepp_next(ftepp);
1717     do
1718     {
1719         if (ftepp->token >= TOKEN_EOF)
1720             break;
1721 #if 0
1722         newline = true;
1723 #endif
1724
1725         switch (ftepp->token) {
1726             case TOKEN_KEYWORD:
1727             case TOKEN_IDENT:
1728             case TOKEN_TYPENAME:
1729                 /* is it a predef? */
1730                 if (OPTS_FLAG(FTEPP_PREDEFS)) {
1731                     char *(*predef)(ftepp_t*) = ftepp_predef(ftepp_tokval(ftepp));
1732                     if (predef) {
1733                         expand = predef(ftepp);
1734                         ftepp_out (ftepp, expand, false);
1735                         ftepp_next(ftepp);
1736
1737                         mem_d(expand);
1738                         break;
1739                     }
1740                 }
1741
1742                 if (ftepp->output_on)
1743                     macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1744                 else
1745                     macro = NULL;
1746
1747                 if (!macro) {
1748                     ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1749                     ftepp_next(ftepp);
1750                     break;
1751                 }
1752                 if (!ftepp_macro_call(ftepp, macro))
1753                     ftepp->token = TOKEN_ERROR;
1754                 break;
1755             case '#':
1756                 if (!newline) {
1757                     ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1758                     ftepp_next(ftepp);
1759                     break;
1760                 }
1761                 ftepp->lex->flags.mergelines = true;
1762                 if (ftepp_next(ftepp) >= TOKEN_EOF) {
1763                     ftepp_error(ftepp, "error in preprocessor directive");
1764                     ftepp->token = TOKEN_ERROR;
1765                     break;
1766                 }
1767                 if (!ftepp_hash(ftepp))
1768                     ftepp->token = TOKEN_ERROR;
1769                 ftepp->lex->flags.mergelines = false;
1770                 break;
1771             case TOKEN_EOL:
1772                 newline = true;
1773                 ftepp_out(ftepp, "\n", true);
1774                 ftepp_next(ftepp);
1775                 break;
1776             case TOKEN_WHITE:
1777                 /* same as default but don't set newline=false */
1778                 ftepp_out(ftepp, ftepp_tokval(ftepp), true);
1779                 ftepp_next(ftepp);
1780                 break;
1781             default:
1782                 newline = false;
1783                 ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1784                 ftepp_next(ftepp);
1785                 break;
1786         }
1787     } while (!ftepp->errors && ftepp->token < TOKEN_EOF);
1788
1789     /* force a 0 at the end but don't count it as added to the output */
1790     vec_push(ftepp->output_string, 0);
1791     vec_shrinkby(ftepp->output_string, 1);
1792
1793     return (ftepp->token == TOKEN_EOF);
1794 }
1795
1796 /* Like in parser.c - files keep the previous state so we have one global
1797  * preprocessor. Except here we will want to warn about dangling #ifs.
1798  */
1799 static bool ftepp_preprocess_done(ftepp_t *ftepp)
1800 {
1801     bool retval = true;
1802     if (vec_size(ftepp->conditions)) {
1803         if (ftepp_warn(ftepp, WARN_MULTIFILE_IF, "#if spanning multiple files, is this intended?"))
1804             retval = false;
1805     }
1806     lex_close(ftepp->lex);
1807     ftepp->lex = NULL;
1808     if (ftepp->itemname) {
1809         mem_d(ftepp->itemname);
1810         ftepp->itemname = NULL;
1811     }
1812     return retval;
1813 }
1814
1815 bool ftepp_preprocess_file(ftepp_t *ftepp, const char *filename)
1816 {
1817     ftepp->lex = lex_open(filename);
1818     ftepp->itemname = util_strdup(filename);
1819     if (!ftepp->lex) {
1820         con_out("failed to open file \"%s\"\n", filename);
1821         return false;
1822     }
1823     if (!ftepp_preprocess(ftepp))
1824         return false;
1825     return ftepp_preprocess_done(ftepp);
1826 }
1827
1828 bool ftepp_preprocess_string(ftepp_t *ftepp, const char *name, const char *str)
1829 {
1830     ftepp->lex = lex_open_string(str, strlen(str), name);
1831     ftepp->itemname = util_strdup(name);
1832     if (!ftepp->lex) {
1833         con_out("failed to create lexer for string \"%s\"\n", name);
1834         return false;
1835     }
1836     if (!ftepp_preprocess(ftepp))
1837         return false;
1838     return ftepp_preprocess_done(ftepp);
1839 }
1840
1841
1842 void ftepp_add_macro(ftepp_t *ftepp, const char *name, const char *value) {
1843     char *create = NULL;
1844
1845     /* use saner path for empty macros */
1846     if (!value) {
1847         ftepp_add_define(ftepp, "__builtin__", name);
1848         return;
1849     }
1850
1851     vec_append(create, 8,           "#define ");
1852     vec_append(create, strlen(name), name);
1853     vec_push  (create, ' ');
1854     vec_append(create, strlen(value), value);
1855     vec_push  (create, 0);
1856
1857     ftepp_preprocess_string(ftepp, "__builtin__", create);
1858     vec_free  (create);
1859 }
1860
1861 ftepp_t *ftepp_create()
1862 {
1863     ftepp_t *ftepp;
1864     char minor[32];
1865     char major[32];
1866     size_t i;
1867
1868     ftepp = ftepp_new();
1869     if (!ftepp)
1870         return NULL;
1871
1872     memset(minor, 0, sizeof(minor));
1873     memset(major, 0, sizeof(major));
1874
1875     /* set the right macro based on the selected standard */
1876     ftepp_add_define(ftepp, NULL, "GMQCC");
1877     if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_FTEQCC) {
1878         ftepp_add_define(ftepp, NULL, "__STD_FTEQCC__");
1879         /* 1.00 */
1880         major[0] = '"';
1881         major[1] = '1';
1882         major[2] = '"';
1883
1884         minor[0] = '"';
1885         minor[1] = '0';
1886         minor[2] = '"';
1887     } else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_GMQCC) {
1888         ftepp_add_define(ftepp, NULL, "__STD_GMQCC__");
1889         util_snprintf(major, 32, "\"%d\"", GMQCC_VERSION_MAJOR);
1890         util_snprintf(minor, 32, "\"%d\"", GMQCC_VERSION_MINOR);
1891     } else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCCX) {
1892         ftepp_add_define(ftepp, NULL, "__STD_QCCX__");
1893         util_snprintf(major, 32, "\"%d\"", GMQCC_VERSION_MAJOR);
1894         util_snprintf(minor, 32, "\"%d\"", GMQCC_VERSION_MINOR);
1895     } else if (OPTS_OPTION_U32(OPTION_STANDARD) == COMPILER_QCC) {
1896         ftepp_add_define(ftepp, NULL, "__STD_QCC__");
1897         /* 1.0 */
1898         major[0] = '"';
1899         major[1] = '1';
1900         major[2] = '"';
1901
1902         minor[0] = '"';
1903         minor[1] = '0';
1904         minor[2] = '"';
1905     }
1906
1907     ftepp_add_macro(ftepp, "__STD_VERSION_MINOR__", minor);
1908     ftepp_add_macro(ftepp, "__STD_VERSION_MAJOR__", major);
1909
1910     /*
1911      * We're going to just make __NULL__ nil, which works for 60% of the
1912      * cases of __NULL_ for fteqcc.
1913      */
1914     ftepp_add_macro(ftepp, "__NULL__", "nil");
1915
1916     /* add all the math constants if they can be */
1917     if (OPTS_FLAG(FTEPP_MATHDEFS)) {
1918         for (i = 0; i < GMQCC_ARRAY_COUNT(ftepp_math_constants); i++)
1919             if (!ftepp_macro_find(ftepp, ftepp_math_constants[i][0]))
1920                 ftepp_add_macro(ftepp, ftepp_math_constants[i][0], ftepp_math_constants[i][1]);
1921     }
1922
1923     return ftepp;
1924 }
1925
1926 void ftepp_add_define(ftepp_t *ftepp, const char *source, const char *name)
1927 {
1928     ppmacro *macro;
1929     lex_ctx_t ctx = { "__builtin__", 0, 0 };
1930     ctx.file = source;
1931     macro = ppmacro_new(ctx, name);
1932     /*vec_push(ftepp->macros, macro);*/
1933     util_htset(ftepp->macros, name, macro);
1934 }
1935
1936 const char *ftepp_get(ftepp_t *ftepp)
1937 {
1938     return ftepp->output_string;
1939 }
1940
1941 void ftepp_flush(ftepp_t *ftepp)
1942 {
1943     ftepp_flush_do(ftepp);
1944 }
1945
1946 void ftepp_finish(ftepp_t *ftepp)
1947 {
1948     if (!ftepp)
1949         return;
1950     ftepp_delete(ftepp);
1951 }