2 * Copyright (C) 2012, 2013
6 * Permission is hereby granted, free of charge, to any person obtaining a copy of
7 * this software and associated documentation files (the "Software"), to deal in
8 * the Software without restriction, including without limitation the rights to
9 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
10 * of the Software, and to permit persons to whom the Software is furnished to do
11 * so, subject to the following conditions:
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
37 /* a copy from the lexer */
51 /* yes we need an extra flag since `#define FOO x` is not the same as `#define FOO() x` */
64 ppcondition *conditions;
75 * Implement the predef subsystem now. We can do this safely with the
76 * help of lexer contexts.
78 static uint32_t ftepp_predef_countval = 0;
79 static uint32_t ftepp_predef_randval = 0;
82 char *ftepp_predef_date(lex_file *context) {
85 char *value = mem_a(82);
86 /* 82 is enough for strftime but we also have " " in our string */
92 itime = localtime(&rtime);
94 strftime(value, 82, "\"%b %d %Y\"", itime);
100 char *ftepp_predef_time(lex_file *context) {
103 char *value = mem_a(82);
104 /* 82 is enough for strftime but we also have " " in our string */
110 itime = localtime(&rtime);
112 strftime(value, 82, "\"%X\"", itime);
118 char *ftepp_predef_line(lex_file *context) {
120 util_asprintf(&value, "%d", (int)context->line);
124 char *ftepp_predef_file(lex_file *context) {
125 size_t length = strlen(context->name) + 3; /* two quotes and a terminator */
126 char *value = (char*)mem_a(length);
127 memset (value, 0, length);
128 sprintf(value, "\"%s\"", context->name);
132 /* __COUNTER_LAST__ */
133 char *ftepp_predef_counterlast(lex_file *context) {
135 util_asprintf(&value, "%u", ftepp_predef_countval);
141 char *ftepp_predef_counter(lex_file *context) {
143 ftepp_predef_countval ++;
144 util_asprintf(&value, "%u", ftepp_predef_countval);
150 char *ftepp_predef_random(lex_file *context) {
152 ftepp_predef_randval = (util_rand() % 0xFF) + 1;
153 util_asprintf(&value, "%u", ftepp_predef_randval);
158 /* __RANDOM_LAST__ */
159 char *ftepp_predef_randomlast(lex_file *context) {
161 util_asprintf(&value, "%u", ftepp_predef_randval);
167 const ftepp_predef_t ftepp_predefs[FTEPP_PREDEF_COUNT] = {
168 { "__LINE__", &ftepp_predef_line },
169 { "__FILE__", &ftepp_predef_file },
170 { "__COUNTER__", &ftepp_predef_counter },
171 { "__COUNTER_LAST__", &ftepp_predef_counterlast },
172 { "__RANDOM__", &ftepp_predef_random },
173 { "__RANDOM_LAST__", &ftepp_predef_randomlast },
174 { "__DATE__", &ftepp_predef_date },
175 { "__TIME__", &ftepp_predef_time }
178 #define ftepp_tokval(f) ((f)->lex->tok.value)
179 #define ftepp_ctx(f) ((f)->lex->tok.ctx)
181 static void ftepp_errorat(ftepp_t *ftepp, lex_ctx ctx, const char *fmt, ...)
188 con_cvprintmsg((void*)&ctx, LVL_ERROR, "error", fmt, ap);
192 static void ftepp_error(ftepp_t *ftepp, const char *fmt, ...)
199 con_cvprintmsg((void*)&ftepp->lex->tok.ctx, LVL_ERROR, "error", fmt, ap);
203 static bool GMQCC_WARN ftepp_warn(ftepp_t *ftepp, int warntype, const char *fmt, ...)
209 r = vcompile_warning(ftepp->lex->tok.ctx, warntype, fmt, ap);
214 static pptoken *pptoken_make(ftepp_t *ftepp)
216 pptoken *token = (pptoken*)mem_a(sizeof(pptoken));
217 token->token = ftepp->token;
219 if (token->token == TOKEN_WHITE)
220 token->value = util_strdup(" ");
223 token->value = util_strdup(ftepp_tokval(ftepp));
225 memcpy(&token->constval, &ftepp->lex->tok.constval, sizeof(token->constval));
229 static void pptoken_delete(pptoken *self)
235 static ppmacro *ppmacro_new(lex_ctx ctx, const char *name)
237 ppmacro *macro = (ppmacro*)mem_a(sizeof(ppmacro));
240 memset(macro, 0, sizeof(*macro));
241 macro->name = util_strdup(name);
245 static void ppmacro_delete(ppmacro *self)
248 for (i = 0; i < vec_size(self->params); ++i)
249 mem_d(self->params[i]);
250 vec_free(self->params);
251 for (i = 0; i < vec_size(self->output); ++i)
252 pptoken_delete(self->output[i]);
253 vec_free(self->output);
258 static ftepp_t* ftepp_new()
262 ftepp = (ftepp_t*)mem_a(sizeof(*ftepp));
263 memset(ftepp, 0, sizeof(*ftepp));
265 ftepp->output_on = true;
270 static void ftepp_flush_do(ftepp_t *self)
272 vec_free(self->output_string);
275 static void ftepp_delete(ftepp_t *self)
278 ftepp_flush_do(self);
280 mem_d(self->itemname);
281 if (self->includename)
282 vec_free(self->includename);
283 for (i = 0; i < vec_size(self->macros); ++i)
284 ppmacro_delete(self->macros[i]);
285 vec_free(self->macros);
286 vec_free(self->conditions);
288 lex_close(self->lex);
292 static void ftepp_out(ftepp_t *ftepp, const char *str, bool ignore_cond)
294 if (ignore_cond || ftepp->output_on)
299 data = vec_add(ftepp->output_string, len);
300 memcpy(data, str, len);
304 static void ftepp_update_output_condition(ftepp_t *ftepp)
307 ftepp->output_on = true;
308 for (i = 0; i < vec_size(ftepp->conditions); ++i)
309 ftepp->output_on = ftepp->output_on && ftepp->conditions[i].on;
312 static ppmacro* ftepp_macro_find(ftepp_t *ftepp, const char *name)
315 for (i = 0; i < vec_size(ftepp->macros); ++i) {
316 if (!strcmp(name, ftepp->macros[i]->name))
317 return ftepp->macros[i];
322 static void ftepp_macro_delete(ftepp_t *ftepp, const char *name)
325 for (i = 0; i < vec_size(ftepp->macros); ++i) {
326 if (!strcmp(name, ftepp->macros[i]->name)) {
327 vec_remove(ftepp->macros, i, 1);
333 static GMQCC_INLINE int ftepp_next(ftepp_t *ftepp)
335 return (ftepp->token = lex_do(ftepp->lex));
338 /* Important: this does not skip newlines! */
339 static bool ftepp_skipspace(ftepp_t *ftepp)
341 if (ftepp->token != TOKEN_WHITE)
343 while (ftepp_next(ftepp) == TOKEN_WHITE) {}
344 if (ftepp->token >= TOKEN_EOF) {
345 ftepp_error(ftepp, "unexpected end of preprocessor directive");
351 /* this one skips EOLs as well */
352 static bool ftepp_skipallwhite(ftepp_t *ftepp)
354 if (ftepp->token != TOKEN_WHITE && ftepp->token != TOKEN_EOL)
358 } while (ftepp->token == TOKEN_WHITE || ftepp->token == TOKEN_EOL);
359 if (ftepp->token >= TOKEN_EOF) {
360 ftepp_error(ftepp, "unexpected end of preprocessor directive");
367 * The huge macro parsing code...
369 static bool ftepp_define_params(ftepp_t *ftepp, ppmacro *macro)
373 if (!ftepp_skipspace(ftepp))
375 if (ftepp->token == ')')
377 switch (ftepp->token) {
381 vec_push(macro->params, util_strdup(ftepp_tokval(ftepp)));
384 macro->variadic = true;
387 ftepp_error(ftepp, "unexpected token in parameter list");
391 if (!ftepp_skipspace(ftepp))
393 if (macro->variadic && ftepp->token != ')') {
394 ftepp_error(ftepp, "cannot have parameters after the variadic parameters");
397 } while (ftepp->token == ',');
398 if (ftepp->token != ')') {
399 ftepp_error(ftepp, "expected closing paren after macro parameter list");
403 /* skipspace happens in ftepp_define */
407 static bool ftepp_define_body(ftepp_t *ftepp, ppmacro *macro)
410 while (ftepp->token != TOKEN_EOL && ftepp->token < TOKEN_EOF) {
412 if (macro->variadic && !strcmp(ftepp_tokval(ftepp), "__VA_ARGS__")) {
413 /* remember the token */
414 if (ftepp_next(ftepp) == '#' &&
415 ftepp_next(ftepp) == '#' &&
416 ftepp_next(ftepp) == '[')
418 if (ftepp_next(ftepp) != TOKEN_INTCONST) {
419 ftepp_error(ftepp, "expected index for __VA_ARGS__ subscript");
423 index = atoi(ftepp_tokval(ftepp));
425 if (ftepp_next(ftepp) != ']') {
426 ftepp_error(ftepp, "expected `]` in __VA_ARGS__ subscript");
431 * mark it as an array to be handled later as such and not
432 * as traditional __VA_ARGS__
434 ftepp->token = TOKEN_VA_ARGS_ARRAY;
435 ptok = pptoken_make(ftepp);
436 ptok->constval.i = index;
437 vec_push(macro->output, ptok);
440 int old = ftepp->token;
441 ftepp->token = TOKEN_VA_ARGS;
442 ptok = pptoken_make(ftepp);
443 vec_push(macro->output, ptok);
449 ptok = pptoken_make(ftepp);
450 vec_push(macro->output, ptok);
454 /* recursive expansion can cause EOFs here */
455 if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
456 ftepp_error(ftepp, "unexpected junk after macro or unexpected end of file");
462 static bool ftepp_define(ftepp_t *ftepp)
465 size_t l = ftepp_ctx(ftepp).line;
467 (void)ftepp_next(ftepp);
468 if (!ftepp_skipspace(ftepp))
471 switch (ftepp->token) {
475 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
476 if (macro && ftepp->output_on) {
477 if (ftepp_warn(ftepp, WARN_PREPROCESSOR, "redefining `%s`", ftepp_tokval(ftepp)))
479 ftepp_macro_delete(ftepp, ftepp_tokval(ftepp));
481 macro = ppmacro_new(ftepp_ctx(ftepp), ftepp_tokval(ftepp));
484 ftepp_error(ftepp, "expected macro name");
488 (void)ftepp_next(ftepp);
490 if (ftepp->token == '(') {
491 macro->has_params = true;
492 if (!ftepp_define_params(ftepp, macro))
496 if (!ftepp_skipspace(ftepp))
499 if (!ftepp_define_body(ftepp, macro))
502 if (ftepp->output_on)
503 vec_push(ftepp->macros, macro);
505 ppmacro_delete(macro);
508 for (; l < ftepp_ctx(ftepp).line; ++l)
509 ftepp_out(ftepp, "\n", true);
514 * When a macro is used we have to handle parameters as well
515 * as special-concatenation via ## or stringification via #
517 * Note: parenthesis can nest, so FOO((a),b) is valid, but only
518 * this kind of parens. Curly braces or [] don't count towards the
525 static void macroparam_clean(macroparam *self)
528 for (i = 0; i < vec_size(self->tokens); ++i)
529 pptoken_delete(self->tokens[i]);
530 vec_free(self->tokens);
533 /* need to leave the last token up */
534 static bool ftepp_macro_call_params(ftepp_t *ftepp, macroparam **out_params)
536 macroparam *params = NULL;
542 if (!ftepp_skipallwhite(ftepp))
544 while (ftepp->token != ')') {
546 if (!ftepp_skipallwhite(ftepp))
548 while (parens || ftepp->token != ',') {
549 if (ftepp->token == '(')
551 else if (ftepp->token == ')') {
556 ptok = pptoken_make(ftepp);
557 vec_push(mp.tokens, ptok);
558 if (ftepp_next(ftepp) >= TOKEN_EOF) {
559 ftepp_error(ftepp, "unexpected EOF in macro call");
563 vec_push(params, mp);
565 if (ftepp->token == ')')
567 if (ftepp->token != ',') {
568 ftepp_error(ftepp, "expected closing paren or comma in macro call");
571 if (ftepp_next(ftepp) >= TOKEN_EOF) {
572 ftepp_error(ftepp, "unexpected EOF in macro call");
576 /* need to leave that up
577 if (ftepp_next(ftepp) >= TOKEN_EOF) {
578 ftepp_error(ftepp, "unexpected EOF in macro call");
582 *out_params = params;
587 macroparam_clean(&mp);
588 for (i = 0; i < vec_size(params); ++i)
589 macroparam_clean(¶ms[i]);
594 static bool macro_params_find(ppmacro *macro, const char *name, size_t *idx)
597 for (i = 0; i < vec_size(macro->params); ++i) {
598 if (!strcmp(macro->params[i], name)) {
606 static void ftepp_stringify_token(ftepp_t *ftepp, pptoken *token)
611 switch (token->token) {
612 case TOKEN_STRINGCONST:
615 /* in preprocessor mode strings already are string,
616 * so we don't get actual newline bytes here.
617 * Still need to escape backslashes and quotes.
620 case '\\': ftepp_out(ftepp, "\\\\", false); break;
621 case '"': ftepp_out(ftepp, "\\\"", false); break;
624 ftepp_out(ftepp, chs, false);
631 ftepp_out(ftepp, " ", false);
634 ftepp_out(ftepp, "\\n", false);
637 ftepp_out(ftepp, token->value, false);
642 static void ftepp_stringify(ftepp_t *ftepp, macroparam *param)
645 ftepp_out(ftepp, "\"", false);
646 for (i = 0; i < vec_size(param->tokens); ++i)
647 ftepp_stringify_token(ftepp, param->tokens[i]);
648 ftepp_out(ftepp, "\"", false);
651 static void ftepp_recursion_header(ftepp_t *ftepp)
653 ftepp_out(ftepp, "\n#pragma push(line)\n", false);
656 static void ftepp_recursion_footer(ftepp_t *ftepp)
658 ftepp_out(ftepp, "\n#pragma pop(line)\n", false);
661 static void ftepp_param_out(ftepp_t *ftepp, macroparam *param)
665 for (i = 0; i < vec_size(param->tokens); ++i) {
666 out = param->tokens[i];
667 if (out->token == TOKEN_EOL)
668 ftepp_out(ftepp, "\n", false);
670 ftepp_out(ftepp, out->value, false);
674 static bool ftepp_preprocess(ftepp_t *ftepp);
675 static bool ftepp_macro_expand(ftepp_t *ftepp, ppmacro *macro, macroparam *params, bool resetline)
677 char *old_string = ftepp->output_string;
679 lex_file *old_lexer = ftepp->lex;
680 size_t vararg_start = vec_size(macro->params);
692 if (vararg_start < vec_size(params))
693 varargs = vec_size(params) - vararg_start;
698 if (!vec_size(macro->output))
701 ftepp->output_string = NULL;
702 for (o = 0; o < vec_size(macro->output); ++o) {
703 pptoken *out = macro->output[o];
704 switch (out->token) {
706 if (!macro->variadic) {
707 ftepp_error(ftepp, "internal preprocessor error: TOKEN_VA_ARGS in non-variadic macro");
708 vec_free(old_string);
715 ftepp_param_out(ftepp, ¶ms[pi + vararg_start]);
716 for (++pi; pi < varargs; ++pi) {
717 ftepp_out(ftepp, ", ", false);
718 ftepp_param_out(ftepp, ¶ms[pi + vararg_start]);
722 case TOKEN_VA_ARGS_ARRAY:
723 if (out->constval.i >= varargs) {
724 ftepp_error(ftepp, "subscript of `[%u]` is out of bounds for `__VA_ARGS__`", out->constval.i);
725 vec_free(old_string);
729 ftepp_param_out(ftepp, ¶ms[out->constval.i + vararg_start]);
735 if (!macro_params_find(macro, out->value, &pi)) {
736 ftepp_out(ftepp, out->value, false);
739 ftepp_param_out(ftepp, ¶ms[pi]);
742 if (o + 1 < vec_size(macro->output)) {
743 nextok = macro->output[o+1]->token;
745 /* raw concatenation */
749 if ( (nextok == TOKEN_IDENT ||
750 nextok == TOKEN_KEYWORD ||
751 nextok == TOKEN_TYPENAME) &&
752 macro_params_find(macro, macro->output[o+1]->value, &pi))
755 ftepp_stringify(ftepp, ¶ms[pi]);
759 ftepp_out(ftepp, "#", false);
762 ftepp_out(ftepp, "\n", false);
765 ftepp_out(ftepp, out->value, false);
769 vec_push(ftepp->output_string, 0);
770 /* Now run the preprocessor recursively on this string buffer */
772 printf("__________\n%s\n=========\n", ftepp->output_string);
774 inlex = lex_open_string(ftepp->output_string, vec_size(ftepp->output_string)-1, ftepp->lex->name);
776 ftepp_error(ftepp, "internal error: failed to instantiate lexer");
781 inlex->line = ftepp->lex->line;
782 inlex->sline = ftepp->lex->sline;
785 old_inmacro = ftepp->in_macro;
786 ftepp->in_macro = true;
787 ftepp->output_string = NULL;
788 if (!ftepp_preprocess(ftepp)) {
789 ftepp->in_macro = old_inmacro;
790 vec_free(ftepp->lex->open_string);
791 vec_free(ftepp->output_string);
792 lex_close(ftepp->lex);
796 ftepp->in_macro = old_inmacro;
797 vec_free(ftepp->lex->open_string);
798 lex_close(ftepp->lex);
800 inner_string = ftepp->output_string;
801 ftepp->output_string = old_string;
803 has_newlines = (strchr(inner_string, '\n') != NULL);
805 if (has_newlines && !old_inmacro)
806 ftepp_recursion_header(ftepp);
808 vec_append(ftepp->output_string, vec_size(inner_string), inner_string);
809 vec_free(inner_string);
811 if (has_newlines && !old_inmacro)
812 ftepp_recursion_footer(ftepp);
814 if (resetline && !ftepp->in_macro) {
816 sprintf(lineno, "\n#pragma line(%lu)\n", (unsigned long)(old_lexer->sline));
817 ftepp_out(ftepp, lineno, false);
820 old_string = ftepp->output_string;
822 ftepp->lex = old_lexer;
823 ftepp->output_string = old_string;
827 static bool ftepp_macro_call(ftepp_t *ftepp, ppmacro *macro)
830 macroparam *params = NULL;
834 if (!macro->has_params) {
835 if (!ftepp_macro_expand(ftepp, macro, NULL, false))
842 if (!ftepp_skipallwhite(ftepp))
845 if (ftepp->token != '(') {
846 ftepp_error(ftepp, "expected macro parameters in parenthesis");
851 paramline = ftepp->lex->sline;
852 if (!ftepp_macro_call_params(ftepp, ¶ms))
855 if ( vec_size(params) < vec_size(macro->params) ||
856 (vec_size(params) > vec_size(macro->params) && !macro->variadic) )
858 ftepp_error(ftepp, "macro %s expects%s %u paramteters, %u provided", macro->name,
859 (macro->variadic ? " at least" : ""),
860 (unsigned int)vec_size(macro->params),
861 (unsigned int)vec_size(params));
866 if (!ftepp_macro_expand(ftepp, macro, params, (paramline != ftepp->lex->sline)))
871 for (o = 0; o < vec_size(params); ++o)
872 macroparam_clean(¶ms[o]);
878 * #if - the FTEQCC way:
879 * defined(FOO) => true if FOO was #defined regardless of parameters or contents
880 * <numbers> => True if the number is not 0
881 * !<factor> => True if the factor yields false
882 * !!<factor> => ERROR on 2 or more unary nots
883 * <macro> => becomes the macro's FIRST token regardless of parameters
884 * <e> && <e> => True if both expressions are true
885 * <e> || <e> => True if either expression is true
887 * <ident> => False (remember for macros the <macro> rule applies instead)
888 * Unary + and - are weird and wrong in fteqcc so we don't allow them
889 * parenthesis in expressions are allowed
890 * parameter lists on macros are errors
891 * No mathematical calculations are executed
893 static bool ftepp_if_expr(ftepp_t *ftepp, bool *out, double *value_out);
894 static bool ftepp_if_op(ftepp_t *ftepp)
896 ftepp->lex->flags.noops = false;
898 if (!ftepp_skipspace(ftepp))
900 ftepp->lex->flags.noops = true;
903 static bool ftepp_if_value(ftepp_t *ftepp, bool *out, double *value_out)
909 if (!ftepp_skipspace(ftepp))
912 while (ftepp->token == '!') {
915 if (!ftepp_skipspace(ftepp))
919 if (ftepp->token == TOKEN_OPERATOR && !strcmp(ftepp_tokval(ftepp), "-"))
923 if (!ftepp_skipspace(ftepp))
927 switch (ftepp->token) {
931 if (!strcmp(ftepp_tokval(ftepp), "defined")) {
933 if (!ftepp_skipspace(ftepp))
935 if (ftepp->token != '(') {
936 ftepp_error(ftepp, "`defined` keyword in #if requires a macro name in parenthesis");
940 if (!ftepp_skipspace(ftepp))
942 if (ftepp->token != TOKEN_IDENT &&
943 ftepp->token != TOKEN_TYPENAME &&
944 ftepp->token != TOKEN_KEYWORD)
946 ftepp_error(ftepp, "defined() used on an unexpected token type");
949 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
952 if (!ftepp_skipspace(ftepp))
954 if (ftepp->token != ')') {
955 ftepp_error(ftepp, "expected closing paren");
961 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
962 if (!macro || !vec_size(macro->output)) {
966 /* This does not expand recursively! */
967 switch (macro->output[0]->token) {
969 *value_out = macro->output[0]->constval.i;
970 *out = !!(macro->output[0]->constval.i);
972 case TOKEN_FLOATCONST:
973 *value_out = macro->output[0]->constval.f;
974 *out = !!(macro->output[0]->constval.f);
982 case TOKEN_STRINGCONST:
987 *value_out = ftepp->lex->tok.constval.i;
988 *out = !!(ftepp->lex->tok.constval.i);
990 case TOKEN_FLOATCONST:
991 *value_out = ftepp->lex->tok.constval.f;
992 *out = !!(ftepp->lex->tok.constval.f);
997 if (!ftepp_if_expr(ftepp, out, value_out))
999 if (ftepp->token != ')') {
1000 ftepp_error(ftepp, "expected closing paren in #if expression");
1006 ftepp_error(ftepp, "junk in #if: `%s` ...", ftepp_tokval(ftepp));
1008 ftepp_error(ftepp, "internal: token %i\n", ftepp->token);
1012 *value_out = -*value_out;
1015 *value_out = (*out ? 1 : 0);
1021 static bool ftepp_if_nextvalue(ftepp_t *ftepp, bool *out, double *value_out)
1023 if (!ftepp_next(ftepp))
1025 return ftepp_if_value(ftepp, out, value_out);
1029 static bool ftepp_if_expr(ftepp_t *ftepp, bool *out, double *value_out)
1031 if (!ftepp_if_value(ftepp, out, value_out))
1034 if (!ftepp_if_op(ftepp))
1037 if (ftepp->token == ')' || ftepp->token != TOKEN_OPERATOR)
1040 /* FTEQCC is all right-associative and no precedence here */
1041 if (!strcmp(ftepp_tokval(ftepp), "&&") ||
1042 !strcmp(ftepp_tokval(ftepp), "||"))
1045 char opc = ftepp_tokval(ftepp)[0];
1049 if (!ftepp_next(ftepp))
1051 if (!ftepp_if_expr(ftepp, &next, &nextvalue))
1055 *out = *out && next;
1057 *out = *out || next;
1059 *value_out = (*out ? 1 : 0);
1062 else if (!strcmp(ftepp_tokval(ftepp), "==") ||
1063 !strcmp(ftepp_tokval(ftepp), "!=") ||
1064 !strcmp(ftepp_tokval(ftepp), ">=") ||
1065 !strcmp(ftepp_tokval(ftepp), "<=") ||
1066 !strcmp(ftepp_tokval(ftepp), ">") ||
1067 !strcmp(ftepp_tokval(ftepp), "<"))
1070 const char opc0 = ftepp_tokval(ftepp)[0];
1071 const char opc1 = ftepp_tokval(ftepp)[1];
1074 if (!ftepp_next(ftepp))
1076 if (!ftepp_if_expr(ftepp, &next, &other))
1080 *out = (*value_out == other);
1081 else if (opc0 == '!')
1082 *out = (*value_out != other);
1083 else if (opc0 == '>') {
1084 if (opc1 == '=') *out = (*value_out >= other);
1085 else *out = (*value_out > other);
1087 else if (opc0 == '<') {
1088 if (opc1 == '=') *out = (*value_out <= other);
1089 else *out = (*value_out < other);
1091 *value_out = (*out ? 1 : 0);
1096 ftepp_error(ftepp, "junk after #if");
1101 static bool ftepp_if(ftepp_t *ftepp, ppcondition *cond)
1103 bool result = false;
1106 memset(cond, 0, sizeof(*cond));
1107 (void)ftepp_next(ftepp);
1109 if (!ftepp_skipspace(ftepp))
1111 if (ftepp->token == TOKEN_EOL) {
1112 ftepp_error(ftepp, "expected expression for #if-directive");
1116 if (!ftepp_if_expr(ftepp, &result, &dummy))
1124 * ifdef is rather simple
1126 static bool ftepp_ifdef(ftepp_t *ftepp, ppcondition *cond)
1129 memset(cond, 0, sizeof(*cond));
1130 (void)ftepp_next(ftepp);
1131 if (!ftepp_skipspace(ftepp))
1134 switch (ftepp->token) {
1136 case TOKEN_TYPENAME:
1138 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1141 ftepp_error(ftepp, "expected macro name");
1145 (void)ftepp_next(ftepp);
1146 if (!ftepp_skipspace(ftepp))
1148 /* relaxing this condition
1149 if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
1150 ftepp_error(ftepp, "stray tokens after #ifdef");
1159 * undef is also simple
1161 static bool ftepp_undef(ftepp_t *ftepp)
1163 (void)ftepp_next(ftepp);
1164 if (!ftepp_skipspace(ftepp))
1167 if (ftepp->output_on) {
1168 switch (ftepp->token) {
1170 case TOKEN_TYPENAME:
1172 ftepp_macro_delete(ftepp, ftepp_tokval(ftepp));
1175 ftepp_error(ftepp, "expected macro name");
1180 (void)ftepp_next(ftepp);
1181 if (!ftepp_skipspace(ftepp))
1183 /* relaxing this condition
1184 if (ftepp->token != TOKEN_EOL && ftepp->token != TOKEN_EOF) {
1185 ftepp_error(ftepp, "stray tokens after #ifdef");
1192 /* Special unescape-string function which skips a leading quote
1193 * and stops at a quote, not just at \0
1195 static void unescape(const char *str, char *out) {
1197 while (*str && *str != '"') {
1201 case '\\': *out++ = *str; break;
1202 case '"': *out++ = *str; break;
1203 case 'a': *out++ = '\a'; break;
1204 case 'b': *out++ = '\b'; break;
1205 case 'r': *out++ = '\r'; break;
1206 case 'n': *out++ = '\n'; break;
1207 case 't': *out++ = '\t'; break;
1208 case 'f': *out++ = '\f'; break;
1209 case 'v': *out++ = '\v'; break;
1224 static char *ftepp_include_find_path(const char *file, const char *pathfile)
1227 char *filename = NULL;
1228 const char *last_slash;
1234 last_slash = strrchr(pathfile, '/');
1237 len = last_slash - pathfile;
1238 memcpy(vec_add(filename, len), pathfile, len);
1239 vec_push(filename, '/');
1243 memcpy(vec_add(filename, len+1), file, len);
1244 vec_last(filename) = 0;
1246 fp = file_open(filename, "rb");
1255 static char *ftepp_include_find(ftepp_t *ftepp, const char *file)
1257 char *filename = NULL;
1259 filename = ftepp_include_find_path(file, ftepp->includename);
1261 filename = ftepp_include_find_path(file, ftepp->itemname);
1265 static bool ftepp_directive_warning(ftepp_t *ftepp) {
1266 char *message = NULL;
1268 if (!ftepp_skipspace(ftepp))
1271 /* handle the odd non string constant case so it works like C */
1272 if (ftepp->token != TOKEN_STRINGCONST) {
1274 vec_upload(message, "#warning", 8);
1276 while (ftepp->token != TOKEN_EOL) {
1277 vec_upload(message, ftepp_tokval(ftepp), strlen(ftepp_tokval(ftepp)));
1280 vec_push(message, '\0');
1281 store = ftepp_warn(ftepp, WARN_CPP, message);
1286 unescape (ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1287 return ftepp_warn(ftepp, WARN_CPP, "#warning %s", ftepp_tokval(ftepp));
1290 static void ftepp_directive_error(ftepp_t *ftepp) {
1291 char *message = NULL;
1293 if (!ftepp_skipspace(ftepp))
1296 /* handle the odd non string constant case so it works like C */
1297 if (ftepp->token != TOKEN_STRINGCONST) {
1298 vec_upload(message, "#error", 6);
1300 while (ftepp->token != TOKEN_EOL) {
1301 vec_upload(message, ftepp_tokval(ftepp), strlen(ftepp_tokval(ftepp)));
1304 vec_push(message, '\0');
1305 ftepp_error(ftepp, message);
1310 unescape (ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1311 ftepp_error(ftepp, "#error %s", ftepp_tokval(ftepp));
1316 * FIXME: do we need/want a -I option?
1317 * FIXME: what about when dealing with files in subdirectories coming from a progs.src?
1319 static bool ftepp_include(ftepp_t *ftepp)
1321 lex_file *old_lexer = ftepp->lex;
1326 char *old_includename;
1328 (void)ftepp_next(ftepp);
1329 if (!ftepp_skipspace(ftepp))
1332 if (ftepp->token != TOKEN_STRINGCONST) {
1333 ftepp_error(ftepp, "expected filename to include");
1337 ctx = ftepp_ctx(ftepp);
1339 unescape(ftepp_tokval(ftepp), ftepp_tokval(ftepp));
1341 ftepp_out(ftepp, "\n#pragma file(", false);
1342 ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1343 ftepp_out(ftepp, ")\n#pragma line(1)\n", false);
1345 filename = ftepp_include_find(ftepp, ftepp_tokval(ftepp));
1347 ftepp_error(ftepp, "failed to open include file `%s`", ftepp_tokval(ftepp));
1350 inlex = lex_open(filename);
1352 ftepp_error(ftepp, "open failed on include file `%s`", filename);
1357 old_includename = ftepp->includename;
1358 ftepp->includename = filename;
1359 if (!ftepp_preprocess(ftepp)) {
1360 vec_free(ftepp->includename);
1361 ftepp->includename = old_includename;
1362 lex_close(ftepp->lex);
1363 ftepp->lex = old_lexer;
1366 vec_free(ftepp->includename);
1367 ftepp->includename = old_includename;
1368 lex_close(ftepp->lex);
1369 ftepp->lex = old_lexer;
1371 ftepp_out(ftepp, "\n#pragma file(", false);
1372 ftepp_out(ftepp, ctx.file, false);
1373 snprintf(lineno, sizeof(lineno), ")\n#pragma line(%lu)\n", (unsigned long)(ctx.line+1));
1374 ftepp_out(ftepp, lineno, false);
1377 (void)ftepp_next(ftepp);
1378 if (!ftepp_skipspace(ftepp))
1380 if (ftepp->token != TOKEN_EOL) {
1381 ftepp_error(ftepp, "stray tokens after #include");
1384 (void)ftepp_next(ftepp);
1389 /* Basic structure handlers */
1390 static bool ftepp_else_allowed(ftepp_t *ftepp)
1392 if (!vec_size(ftepp->conditions)) {
1393 ftepp_error(ftepp, "#else without #if");
1396 if (vec_last(ftepp->conditions).had_else) {
1397 ftepp_error(ftepp, "multiple #else for a single #if");
1403 static bool ftepp_hash(ftepp_t *ftepp)
1408 lex_ctx ctx = ftepp_ctx(ftepp);
1410 if (!ftepp_skipspace(ftepp))
1413 switch (ftepp->token) {
1416 case TOKEN_TYPENAME:
1417 if (!strcmp(ftepp_tokval(ftepp), "define")) {
1418 return ftepp_define(ftepp);
1420 else if (!strcmp(ftepp_tokval(ftepp), "undef")) {
1421 return ftepp_undef(ftepp);
1423 else if (!strcmp(ftepp_tokval(ftepp), "ifdef")) {
1424 if (!ftepp_ifdef(ftepp, &cond))
1426 cond.was_on = cond.on;
1427 vec_push(ftepp->conditions, cond);
1428 ftepp->output_on = ftepp->output_on && cond.on;
1431 else if (!strcmp(ftepp_tokval(ftepp), "ifndef")) {
1432 if (!ftepp_ifdef(ftepp, &cond))
1435 cond.was_on = cond.on;
1436 vec_push(ftepp->conditions, cond);
1437 ftepp->output_on = ftepp->output_on && cond.on;
1440 else if (!strcmp(ftepp_tokval(ftepp), "elifdef")) {
1441 if (!ftepp_else_allowed(ftepp))
1443 if (!ftepp_ifdef(ftepp, &cond))
1445 pc = &vec_last(ftepp->conditions);
1446 pc->on = !pc->was_on && cond.on;
1447 pc->was_on = pc->was_on || pc->on;
1448 ftepp_update_output_condition(ftepp);
1451 else if (!strcmp(ftepp_tokval(ftepp), "elifndef")) {
1452 if (!ftepp_else_allowed(ftepp))
1454 if (!ftepp_ifdef(ftepp, &cond))
1457 pc = &vec_last(ftepp->conditions);
1458 pc->on = !pc->was_on && cond.on;
1459 pc->was_on = pc->was_on || pc->on;
1460 ftepp_update_output_condition(ftepp);
1463 else if (!strcmp(ftepp_tokval(ftepp), "elif")) {
1464 if (!ftepp_else_allowed(ftepp))
1466 if (!ftepp_if(ftepp, &cond))
1468 pc = &vec_last(ftepp->conditions);
1469 pc->on = !pc->was_on && cond.on;
1470 pc->was_on = pc->was_on || pc->on;
1471 ftepp_update_output_condition(ftepp);
1474 else if (!strcmp(ftepp_tokval(ftepp), "if")) {
1475 if (!ftepp_if(ftepp, &cond))
1477 cond.was_on = cond.on;
1478 vec_push(ftepp->conditions, cond);
1479 ftepp->output_on = ftepp->output_on && cond.on;
1482 else if (!strcmp(ftepp_tokval(ftepp), "else")) {
1483 if (!ftepp_else_allowed(ftepp))
1485 pc = &vec_last(ftepp->conditions);
1486 pc->on = !pc->was_on;
1487 pc->had_else = true;
1489 ftepp_update_output_condition(ftepp);
1492 else if (!strcmp(ftepp_tokval(ftepp), "endif")) {
1493 if (!vec_size(ftepp->conditions)) {
1494 ftepp_error(ftepp, "#endif without #if");
1497 vec_pop(ftepp->conditions);
1499 ftepp_update_output_condition(ftepp);
1502 else if (!strcmp(ftepp_tokval(ftepp), "include")) {
1503 return ftepp_include(ftepp);
1505 else if (!strcmp(ftepp_tokval(ftepp), "pragma")) {
1506 ftepp_out(ftepp, "#", false);
1509 else if (!strcmp(ftepp_tokval(ftepp), "warning")) {
1510 ftepp_directive_warning(ftepp);
1513 else if (!strcmp(ftepp_tokval(ftepp), "error")) {
1514 ftepp_directive_error(ftepp);
1518 if (ftepp->output_on) {
1519 ftepp_error(ftepp, "unrecognized preprocessor directive: `%s`", ftepp_tokval(ftepp));
1526 /* break; never reached */
1528 ftepp_error(ftepp, "unexpected preprocessor token: `%s`", ftepp_tokval(ftepp));
1531 ftepp_errorat(ftepp, ctx, "empty preprocessor directive");
1534 ftepp_error(ftepp, "missing newline at end of file", ftepp_tokval(ftepp));
1537 /* Builtins! Don't forget the builtins! */
1538 case TOKEN_INTCONST:
1539 case TOKEN_FLOATCONST:
1540 ftepp_out(ftepp, "#", false);
1543 if (!ftepp_skipspace(ftepp))
1548 static bool ftepp_preprocess(ftepp_t *ftepp)
1551 bool newline = true;
1554 char *expand = NULL;
1557 ftepp->lex->flags.preprocessing = true;
1558 ftepp->lex->flags.mergelines = false;
1559 ftepp->lex->flags.noops = true;
1564 if (ftepp->token >= TOKEN_EOF)
1570 switch (ftepp->token) {
1573 case TOKEN_TYPENAME:
1574 /* is it a predef? */
1575 if (OPTS_FLAG(FTEPP_PREDEFS)) {
1576 for (i = 0; i < sizeof(ftepp_predefs) / sizeof (*ftepp_predefs); i++) {
1577 if (!strcmp(ftepp_predefs[i].name, ftepp_tokval(ftepp))) {
1578 expand = ftepp_predefs[i].func(ftepp->lex);
1579 ftepp_out(ftepp, expand, false);
1580 ftepp_next(ftepp); /* skip */
1582 mem_d(expand); /* free memory */
1588 if (ftepp->output_on)
1589 macro = ftepp_macro_find(ftepp, ftepp_tokval(ftepp));
1594 ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1598 if (!ftepp_macro_call(ftepp, macro))
1599 ftepp->token = TOKEN_ERROR;
1603 ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1607 ftepp->lex->flags.mergelines = true;
1608 if (ftepp_next(ftepp) >= TOKEN_EOF) {
1609 ftepp_error(ftepp, "error in preprocessor directive");
1610 ftepp->token = TOKEN_ERROR;
1613 if (!ftepp_hash(ftepp))
1614 ftepp->token = TOKEN_ERROR;
1615 ftepp->lex->flags.mergelines = false;
1619 ftepp_out(ftepp, "\n", true);
1623 /* same as default but don't set newline=false */
1624 ftepp_out(ftepp, ftepp_tokval(ftepp), true);
1629 ftepp_out(ftepp, ftepp_tokval(ftepp), false);
1633 } while (!ftepp->errors && ftepp->token < TOKEN_EOF);
1635 /* force a 0 at the end but don't count it as added to the output */
1636 vec_push(ftepp->output_string, 0);
1637 vec_shrinkby(ftepp->output_string, 1);
1639 return (ftepp->token == TOKEN_EOF);
1642 /* Like in parser.c - files keep the previous state so we have one global
1643 * preprocessor. Except here we will want to warn about dangling #ifs.
1645 static ftepp_t *ftepp;
1647 static bool ftepp_preprocess_done()
1650 if (vec_size(ftepp->conditions)) {
1651 if (ftepp_warn(ftepp, WARN_MULTIFILE_IF, "#if spanning multiple files, is this intended?"))
1654 lex_close(ftepp->lex);
1656 if (ftepp->itemname) {
1657 mem_d(ftepp->itemname);
1658 ftepp->itemname = NULL;
1663 bool ftepp_preprocess_file(const char *filename)
1665 ftepp->lex = lex_open(filename);
1666 ftepp->itemname = util_strdup(filename);
1668 con_out("failed to open file \"%s\"\n", filename);
1671 if (!ftepp_preprocess(ftepp))
1673 return ftepp_preprocess_done();
1676 bool ftepp_preprocess_string(const char *name, const char *str)
1678 ftepp->lex = lex_open_string(str, strlen(str), name);
1679 ftepp->itemname = util_strdup(name);
1681 con_out("failed to create lexer for string \"%s\"\n", name);
1684 if (!ftepp_preprocess(ftepp))
1686 return ftepp_preprocess_done();
1690 void ftepp_add_macro(const char *name, const char *value) {
1691 char *create = NULL;
1693 /* use saner path for empty macros */
1695 ftepp_add_define("__builtin__", name);
1699 vec_upload(create, "#define ", 8);
1700 vec_upload(create, name, strlen(name));
1701 vec_push (create, ' ');
1702 vec_upload(create, value, strlen(value));
1703 vec_push (create, 0);
1705 ftepp_preprocess_string("__builtin__", create);
1714 ftepp = ftepp_new();
1718 memset(minor, 0, sizeof(minor));
1719 memset(major, 0, sizeof(major));
1721 /* set the right macro based on the selected standard */
1722 ftepp_add_define(NULL, "GMQCC");
1723 if (opts.standard == COMPILER_FTEQCC) {
1724 ftepp_add_define(NULL, "__STD_FTEQCC__");
1733 } else if (opts.standard == COMPILER_GMQCC) {
1734 ftepp_add_define(NULL, "__STD_GMQCC__");
1735 sprintf(major, "\"%d\"", GMQCC_VERSION_MAJOR);
1736 sprintf(minor, "\"%d\"", GMQCC_VERSION_MINOR);
1737 } else if (opts.standard == COMPILER_QCCX) {
1738 ftepp_add_define(NULL, "__STD_QCCX__");
1739 sprintf(major, "\"%d\"", GMQCC_VERSION_MAJOR);
1740 sprintf(minor, "\"%d\"", GMQCC_VERSION_MINOR);
1741 } else if (opts.standard == COMPILER_QCC) {
1742 ftepp_add_define(NULL, "__STD_QCC__");
1753 ftepp_add_macro("__STD_VERSION_MINOR__", minor);
1754 ftepp_add_macro("__STD_VERSION_MAJOR__", major);
1759 void ftepp_add_define(const char *source, const char *name)
1762 lex_ctx ctx = { "__builtin__", 0 };
1764 macro = ppmacro_new(ctx, name);
1765 vec_push(ftepp->macros, macro);
1768 const char *ftepp_get()
1770 return ftepp->output_string;
1775 ftepp_flush_do(ftepp);
1782 ftepp_delete(ftepp);