Added "splint" as a target to simplify running splint on all sources.
[nasm.git] / preproc.c
blobdbcd88d6668ecc1279650816af74a123f6a7b6fd
1 /* -*- mode: c; c-file-style: "bsd" -*- */
2 /* preproc.c macro preprocessor for the Netwide Assembler
4 * The Netwide Assembler is copyright (C) 1996 Simon Tatham and
5 * Julian Hall. All rights reserved. The software is
6 * redistributable under the licence given in the file "Licence"
7 * distributed in the NASM archive.
9 * initial version 18/iii/97 by Simon Tatham
12 /* Typical flow of text through preproc
14 * pp_getline gets tokenised lines, either
16 * from a macro expansion
18 * or
19 * {
20 * read_line gets raw text from stdmacpos, or predef, or current input file
21 * tokenise converts to tokens
22 * }
24 * expand_mmac_params is used to expand %1 etc., unless a macro is being
25 * defined or a false conditional is being processed
26 * (%0, %1, %+1, %-1, %%foo
28 * do_directive checks for directives
30 * expand_smacro is used to expand single line macros
32 * expand_mmacro is used to expand multi-line macros
34 * detoken is used to convert the line back to text
37 #include <stdio.h>
38 #include <stdarg.h>
39 #include <stdlib.h>
40 #include <stddef.h>
41 #include <string.h>
42 #include <ctype.h>
43 #include <limits.h>
45 #include "nasm.h"
46 #include "nasmlib.h"
48 typedef struct SMacro SMacro;
49 typedef struct MMacro MMacro;
50 typedef struct Context Context;
51 typedef struct Token Token;
52 typedef struct Blocks Blocks;
53 typedef struct Line Line;
54 typedef struct Include Include;
55 typedef struct Cond Cond;
56 typedef struct IncPath IncPath;
59 * Store the definition of a single-line macro.
61 struct SMacro
63 SMacro *next;
64 char *name;
65 int casesense;
66 int nparam;
67 int in_progress;
68 Token *expansion;
72 * Store the definition of a multi-line macro. This is also used to
73 * store the interiors of `%rep...%endrep' blocks, which are
74 * effectively self-re-invoking multi-line macros which simply
75 * don't have a name or bother to appear in the hash tables. %rep
76 * blocks are signified by having a NULL `name' field.
78 * In a MMacro describing a `%rep' block, the `in_progress' field
79 * isn't merely boolean, but gives the number of repeats left to
80 * run.
82 * The `next' field is used for storing MMacros in hash tables; the
83 * `next_active' field is for stacking them on istk entries.
85 * When a MMacro is being expanded, `params', `iline', `nparam',
86 * `paramlen', `rotate' and `unique' are local to the invocation.
88 struct MMacro
90 MMacro *next;
91 char *name;
92 int casesense;
93 int nparam_min, nparam_max;
94 int plus; /* is the last parameter greedy? */
95 int nolist; /* is this macro listing-inhibited? */
96 int in_progress;
97 Token *dlist; /* All defaults as one list */
98 Token **defaults; /* Parameter default pointers */
99 int ndefs; /* number of default parameters */
100 Line *expansion;
102 MMacro *next_active;
103 MMacro *rep_nest; /* used for nesting %rep */
104 Token **params; /* actual parameters */
105 Token *iline; /* invocation line */
106 int nparam, rotate, *paramlen;
107 unsigned long unique;
108 int lineno; /* Current line number on expansion */
112 * The context stack is composed of a linked list of these.
114 struct Context
116 Context *next;
117 SMacro *localmac;
118 char *name;
119 unsigned long number;
123 * This is the internal form which we break input lines up into.
124 * Typically stored in linked lists.
126 * Note that `type' serves a double meaning: TOK_SMAC_PARAM is not
127 * necessarily used as-is, but is intended to denote the number of
128 * the substituted parameter. So in the definition
130 * %define a(x,y) ( (x) & ~(y) )
132 * the token representing `x' will have its type changed to
133 * TOK_SMAC_PARAM, but the one representing `y' will be
134 * TOK_SMAC_PARAM+1.
136 * TOK_INTERNAL_STRING is a dirty hack: it's a single string token
137 * which doesn't need quotes around it. Used in the pre-include
138 * mechanism as an alternative to trying to find a sensible type of
139 * quote to use on the filename we were passed.
141 struct Token
143 Token *next;
144 char *text;
145 SMacro *mac; /* associated macro for TOK_SMAC_END */
146 int type;
148 enum
150 TOK_WHITESPACE = 1, TOK_COMMENT, TOK_ID, TOK_PREPROC_ID, TOK_STRING,
151 TOK_NUMBER, TOK_SMAC_END, TOK_OTHER, TOK_SMAC_PARAM,
152 TOK_INTERNAL_STRING
156 * Multi-line macro definitions are stored as a linked list of
157 * these, which is essentially a container to allow several linked
158 * lists of Tokens.
160 * Note that in this module, linked lists are treated as stacks
161 * wherever possible. For this reason, Lines are _pushed_ on to the
162 * `expansion' field in MMacro structures, so that the linked list,
163 * if walked, would give the macro lines in reverse order; this
164 * means that we can walk the list when expanding a macro, and thus
165 * push the lines on to the `expansion' field in _istk_ in reverse
166 * order (so that when popped back off they are in the right
167 * order). It may seem cockeyed, and it relies on my design having
168 * an even number of steps in, but it works...
170 * Some of these structures, rather than being actual lines, are
171 * markers delimiting the end of the expansion of a given macro.
172 * This is for use in the cycle-tracking and %rep-handling code.
173 * Such structures have `finishes' non-NULL, and `first' NULL. All
174 * others have `finishes' NULL, but `first' may still be NULL if
175 * the line is blank.
177 struct Line
179 Line *next;
180 MMacro *finishes;
181 Token *first;
185 * To handle an arbitrary level of file inclusion, we maintain a
186 * stack (ie linked list) of these things.
188 struct Include
190 Include *next;
191 FILE *fp;
192 Cond *conds;
193 Line *expansion;
194 char *fname;
195 int lineno, lineinc;
196 MMacro *mstk; /* stack of active macros/reps */
200 * Include search path. This is simply a list of strings which get
201 * prepended, in turn, to the name of an include file, in an
202 * attempt to find the file if it's not in the current directory.
204 struct IncPath
206 IncPath *next;
207 char *path;
211 * Conditional assembly: we maintain a separate stack of these for
212 * each level of file inclusion. (The only reason we keep the
213 * stacks separate is to ensure that a stray `%endif' in a file
214 * included from within the true branch of a `%if' won't terminate
215 * it and cause confusion: instead, rightly, it'll cause an error.)
217 struct Cond
219 Cond *next;
220 int state;
222 enum
225 * These states are for use just after %if or %elif: IF_TRUE
226 * means the condition has evaluated to truth so we are
227 * currently emitting, whereas IF_FALSE means we are not
228 * currently emitting but will start doing so if a %else comes
229 * up. In these states, all directives are admissible: %elif,
230 * %else and %endif. (And of course %if.)
232 COND_IF_TRUE, COND_IF_FALSE,
234 * These states come up after a %else: ELSE_TRUE means we're
235 * emitting, and ELSE_FALSE means we're not. In ELSE_* states,
236 * any %elif or %else will cause an error.
238 COND_ELSE_TRUE, COND_ELSE_FALSE,
240 * This state means that we're not emitting now, and also that
241 * nothing until %endif will be emitted at all. It's for use in
242 * two circumstances: (i) when we've had our moment of emission
243 * and have now started seeing %elifs, and (ii) when the
244 * condition construct in question is contained within a
245 * non-emitting branch of a larger condition construct.
247 COND_NEVER
249 #define emitting(x) ( (x) == COND_IF_TRUE || (x) == COND_ELSE_TRUE )
252 * These defines are used as the possible return values for do_directive
254 #define NO_DIRECTIVE_FOUND 0
255 #define DIRECTIVE_FOUND 1
258 * Condition codes. Note that we use c_ prefix not C_ because C_ is
259 * used in nasm.h for the "real" condition codes. At _this_ level,
260 * we treat CXZ and ECXZ as condition codes, albeit non-invertible
261 * ones, so we need a different enum...
263 static const char *conditions[] = {
264 "a", "ae", "b", "be", "c", "cxz", "e", "ecxz", "g", "ge", "l", "le",
265 "na", "nae", "nb", "nbe", "nc", "ne", "ng", "nge", "nl", "nle", "no",
266 "np", "ns", "nz", "o", "p", "pe", "po", "s", "z"
268 enum
270 c_A, c_AE, c_B, c_BE, c_C, c_CXZ, c_E, c_ECXZ, c_G, c_GE, c_L, c_LE,
271 c_NA, c_NAE, c_NB, c_NBE, c_NC, c_NE, c_NG, c_NGE, c_NL, c_NLE, c_NO,
272 c_NP, c_NS, c_NZ, c_O, c_P, c_PE, c_PO, c_S, c_Z
274 static int inverse_ccs[] = {
275 c_NA, c_NAE, c_NB, c_NBE, c_NC, -1, c_NE, -1, c_NG, c_NGE, c_NL, c_NLE,
276 c_A, c_AE, c_B, c_BE, c_C, c_E, c_G, c_GE, c_L, c_LE, c_O, c_P, c_S,
277 c_Z, c_NO, c_NP, c_PO, c_PE, c_NS, c_NZ
281 * Directive names.
283 static const char *directives[] = {
284 "%arg",
285 "%assign", "%clear", "%define", "%elif", "%elifctx", "%elifdef",
286 "%elifid", "%elifidn", "%elifidni", "%elifmacro", "%elifnctx", "%elifndef",
287 "%elifnid", "%elifnidn", "%elifnidni", "%elifnmacro", "%elifnnum", "%elifnstr",
288 "%elifnum", "%elifstr", "%else", "%endif", "%endm", "%endmacro",
289 "%endrep", "%error", "%exitrep", "%iassign", "%idefine", "%if",
290 "%ifctx", "%ifdef", "%ifid", "%ifidn", "%ifidni", "%ifmacro", "%ifnctx",
291 "%ifndef", "%ifnid", "%ifnidn", "%ifnidni", "%ifnmacro", "%ifnnum",
292 "%ifnstr", "%ifnum", "%ifstr", "%imacro", "%include",
293 "%ixdefine", "%line",
294 "%local",
295 "%macro", "%pop", "%push", "%rep", "%repl", "%rotate",
296 "%stacksize",
297 "%strlen", "%substr", "%undef", "%xdefine"
299 enum
301 PP_ARG,
302 PP_ASSIGN, PP_CLEAR, PP_DEFINE, PP_ELIF, PP_ELIFCTX, PP_ELIFDEF,
303 PP_ELIFID, PP_ELIFIDN, PP_ELIFIDNI, PP_ELIFMACRO, PP_ELIFNCTX, PP_ELIFNDEF,
304 PP_ELIFNID, PP_ELIFNIDN, PP_ELIFNIDNI, PP_ELIFNMACRO, PP_ELIFNNUM, PP_ELIFNSTR,
305 PP_ELIFNUM, PP_ELIFSTR, PP_ELSE, PP_ENDIF, PP_ENDM, PP_ENDMACRO,
306 PP_ENDREP, PP_ERROR, PP_EXITREP, PP_IASSIGN, PP_IDEFINE, PP_IF,
307 PP_IFCTX, PP_IFDEF, PP_IFID, PP_IFIDN, PP_IFIDNI, PP_IFMACRO, PP_IFNCTX,
308 PP_IFNDEF, PP_IFNID, PP_IFNIDN, PP_IFNIDNI, PP_IFNMACRO, PP_IFNNUM,
309 PP_IFNSTR, PP_IFNUM, PP_IFSTR, PP_IMACRO, PP_INCLUDE,
310 PP_IXDEFINE, PP_LINE,
311 PP_LOCAL,
312 PP_MACRO, PP_POP, PP_PUSH, PP_REP, PP_REPL, PP_ROTATE,
313 PP_STACKSIZE,
314 PP_STRLEN, PP_SUBSTR, PP_UNDEF, PP_XDEFINE
317 /* If this is a an IF, ELIF, ELSE or ENDIF keyword */
318 static int is_condition(int arg)
320 return ((arg >= PP_ELIF) && (arg <= PP_ENDIF)) ||
321 ((arg >= PP_IF) && (arg <= PP_IFSTR));
324 /* For TASM compatibility we need to be able to recognise TASM compatible
325 * conditional compilation directives. Using the NASM pre-processor does
326 * not work, so we look for them specifically from the following list and
327 * then jam in the equivalent NASM directive into the input stream.
330 #ifndef MAX
331 # define MAX(a,b) ( ((a) > (b)) ? (a) : (b))
332 #endif
334 enum
336 TM_ARG, TM_ELIF, TM_ELSE, TM_ENDIF, TM_IF, TM_IFDEF, TM_IFDIFI,
337 TM_IFNDEF, TM_INCLUDE, TM_LOCAL
340 static const char *tasm_directives[] = {
341 "arg", "elif", "else", "endif", "if", "ifdef", "ifdifi",
342 "ifndef", "include", "local"
345 static int StackSize = 4;
346 static char *StackPointer = "ebp";
347 static int ArgOffset = 8;
348 static int LocalOffset = 4;
351 static Context *cstk;
352 static Include *istk;
353 static IncPath *ipath = NULL;
355 static efunc _error; /* Pointer to client-provided error reporting function */
356 static evalfunc evaluate;
358 static int pass; /* HACK: pass 0 = generate dependencies only */
360 static unsigned long unique; /* unique identifier numbers */
362 static Line *predef = NULL;
364 static ListGen *list;
367 * The number of hash values we use for the macro lookup tables.
368 * FIXME: We should *really* be able to configure this at run time,
369 * or even have the hash table automatically expanding when necessary.
371 #define NHASH 31
374 * The current set of multi-line macros we have defined.
376 static MMacro *mmacros[NHASH];
379 * The current set of single-line macros we have defined.
381 static SMacro *smacros[NHASH];
384 * The multi-line macro we are currently defining, or the %rep
385 * block we are currently reading, if any.
387 static MMacro *defining;
390 * The number of macro parameters to allocate space for at a time.
392 #define PARAM_DELTA 16
395 * The standard macro set: defined as `static char *stdmac[]'. Also
396 * gives our position in the macro set, when we're processing it.
398 #include "macros.c"
399 static const char **stdmacpos;
402 * The extra standard macros that come from the object format, if
403 * any.
405 static const char **extrastdmac = NULL;
406 int any_extrastdmac;
409 * Tokens are allocated in blocks to improve speed
411 #define TOKEN_BLOCKSIZE 4096
412 static Token *freeTokens = NULL;
413 struct Blocks {
414 Blocks *next;
415 void *chunk;
418 static Blocks blocks = { NULL, NULL };
421 * Forward declarations.
423 static Token *expand_mmac_params(Token * tline);
424 static Token *expand_smacro(Token * tline);
425 static Token *expand_id(Token * tline);
426 static Context *get_ctx(char *name, int all_contexts);
427 static void make_tok_num(Token * tok, long val);
428 static void error(int severity, const char *fmt, ...);
429 static void *new_Block(size_t size);
430 static void delete_Blocks(void);
431 static Token *new_Token(Token * next, int type, char *text, int txtlen);
432 static Token *delete_Token(Token * t);
435 * Macros for safe checking of token pointers, avoid *(NULL)
437 #define tok_type_(x,t) ((x) && (x)->type == (t))
438 #define skip_white_(x) if (tok_type_((x), TOK_WHITESPACE)) (x)=(x)->next
439 #define tok_is_(x,v) (tok_type_((x), TOK_OTHER) && !strcmp((x)->text,(v)))
440 #define tok_isnt_(x,v) ((x) && ((x)->type!=TOK_OTHER || strcmp((x)->text,(v))))
442 /* Handle TASM specific directives, which do not contain a % in
443 * front of them. We do it here because I could not find any other
444 * place to do it for the moment, and it is a hack (ideally it would
445 * be nice to be able to use the NASM pre-processor to do it).
447 static char *
448 check_tasm_directive(char *line)
450 int i, j, k, m, len;
451 char *p = line, *oldline, oldchar;
453 /* Skip whitespace */
454 while (isspace(*p) && *p != 0)
455 p++;
457 /* Binary search for the directive name */
458 i = -1;
459 j = elements(tasm_directives);
460 len = 0;
461 while (!isspace(p[len]) && p[len] != 0)
462 len++;
463 if (len)
465 oldchar = p[len];
466 p[len] = 0;
467 while (j - i > 1)
469 k = (j + i) / 2;
470 m = nasm_stricmp(p, tasm_directives[k]);
471 if (m == 0)
473 /* We have found a directive, so jam a % in front of it
474 * so that NASM will then recognise it as one if it's own.
476 p[len] = oldchar;
477 len = strlen(p);
478 oldline = line;
479 line = nasm_malloc(len + 2);
480 line[0] = '%';
481 if (k == TM_IFDIFI)
483 /* NASM does not recognise IFDIFI, so we convert it to
484 * %ifdef BOGUS. This is not used in NASM comaptible
485 * code, but does need to parse for the TASM macro
486 * package.
488 strcpy(line + 1, "ifdef BOGUS");
490 else
492 memcpy(line + 1, p, len + 1);
494 nasm_free(oldline);
495 return line;
497 else if (m < 0)
499 j = k;
501 else
502 i = k;
504 p[len] = oldchar;
506 return line;
510 * The pre-preprocessing stage... This function translates line
511 * number indications as they emerge from GNU cpp (`# lineno "file"
512 * flags') into NASM preprocessor line number indications (`%line
513 * lineno file').
515 static char *
516 prepreproc(char *line)
518 int lineno, fnlen;
519 char *fname, *oldline;
521 if (line[0] == '#' && line[1] == ' ')
523 oldline = line;
524 fname = oldline + 2;
525 lineno = atoi(fname);
526 fname += strspn(fname, "0123456789 ");
527 if (*fname == '"')
528 fname++;
529 fnlen = strcspn(fname, "\"");
530 line = nasm_malloc(20 + fnlen);
531 snprintf(line, 20+fnlen,"%%line %d %.*s", lineno, fnlen, fname);
532 nasm_free(oldline);
534 if (tasm_compatible_mode)
535 return check_tasm_directive(line);
536 return line;
540 * The hash function for macro lookups. Note that due to some
541 * macros having case-insensitive names, the hash function must be
542 * invariant under case changes. We implement this by applying a
543 * perfectly normal hash function to the uppercase of the string.
545 static int
546 hash(char *s)
548 unsigned int h = 0;
549 int i = 0;
551 * Powers of three, mod 31.
553 static const int multipliers[] = {
554 1, 3, 9, 27, 19, 26, 16, 17, 20, 29, 25, 13, 8, 24, 10,
555 30, 28, 22, 4, 12, 5, 15, 14, 11, 2, 6, 18, 23, 7, 21
559 while (*s)
561 h += multipliers[i] * (unsigned char) (toupper(*s));
562 s++;
563 if (++i >= elements(multipliers))
564 i = 0;
566 h %= NHASH;
567 return h;
571 * Free a linked list of tokens.
573 static void
574 free_tlist(Token * list)
576 while (list)
578 list = delete_Token(list);
583 * Free a linked list of lines.
585 static void
586 free_llist(Line * list)
588 Line *l;
589 while (list)
591 l = list;
592 list = list->next;
593 free_tlist(l->first);
594 nasm_free(l);
599 * Free an MMacro
601 static void
602 free_mmacro(MMacro * m)
604 nasm_free(m->name);
605 free_tlist(m->dlist);
606 nasm_free(m->defaults);
607 free_llist(m->expansion);
608 nasm_free(m);
612 * Pop the context stack.
614 static void
615 ctx_pop(void)
617 Context *c = cstk;
618 SMacro *smac, *s;
620 cstk = cstk->next;
621 smac = c->localmac;
622 while (smac)
624 s = smac;
625 smac = smac->next;
626 nasm_free(s->name);
627 free_tlist(s->expansion);
628 nasm_free(s);
630 nasm_free(c->name);
631 nasm_free(c);
634 #define BUF_DELTA 512
636 * Read a line from the top file in istk, handling multiple CR/LFs
637 * at the end of the line read, and handling spurious ^Zs. Will
638 * return lines from the standard macro set if this has not already
639 * been done.
641 static char *
642 read_line(void)
644 char *buffer, *p, *q;
645 int bufsize, continued_count;
647 if (stdmacpos)
649 if (*stdmacpos)
651 char *ret = nasm_strdup(*stdmacpos++);
652 if (!*stdmacpos && any_extrastdmac)
654 stdmacpos = extrastdmac;
655 any_extrastdmac = FALSE;
656 return ret;
659 * Nasty hack: here we push the contents of `predef' on
660 * to the top-level expansion stack, since this is the
661 * most convenient way to implement the pre-include and
662 * pre-define features.
664 if (!*stdmacpos)
666 Line *pd, *l;
667 Token *head, **tail, *t;
669 for (pd = predef; pd; pd = pd->next)
671 head = NULL;
672 tail = &head;
673 for (t = pd->first; t; t = t->next)
675 *tail = new_Token(NULL, t->type, t->text, 0);
676 tail = &(*tail)->next;
678 l = nasm_malloc(sizeof(Line));
679 l->next = istk->expansion;
680 l->first = head;
681 l->finishes = FALSE;
682 istk->expansion = l;
685 return ret;
687 else
689 stdmacpos = NULL;
693 bufsize = BUF_DELTA;
694 buffer = nasm_malloc(BUF_DELTA);
695 p = buffer;
696 continued_count = 0;
697 while (1)
699 q = fgets(p, bufsize - (p - buffer), istk->fp);
700 if (!q)
701 break;
702 p += strlen(p);
703 if (p > buffer && p[-1] == '\n')
705 /* Convert backslash-CRLF line continuation sequences into
706 nothing at all (for DOS and Windows) */
707 if (((p - 2) > buffer) && (p[-3] == '\\') && (p[-2] == '\r')) {
708 p -= 3;
709 *p = 0;
710 continued_count++;
712 /* Also convert backslash-LF line continuation sequences into
713 nothing at all (for Unix) */
714 else if (((p - 1) > buffer) && (p[-2] == '\\')) {
715 p -= 2;
716 *p = 0;
717 continued_count++;
719 else {
720 break;
723 if (p - buffer > bufsize - 10)
725 long offset = p - buffer;
726 bufsize += BUF_DELTA;
727 buffer = nasm_realloc(buffer, bufsize);
728 p = buffer + offset; /* prevent stale-pointer problems */
732 if (!q && p == buffer)
734 nasm_free(buffer);
735 return NULL;
738 src_set_linnum(src_get_linnum() + istk->lineinc + (continued_count * istk->lineinc));
741 * Play safe: remove CRs as well as LFs, if any of either are
742 * present at the end of the line.
744 while (--p >= buffer && (*p == '\n' || *p == '\r'))
745 *p = '\0';
748 * Handle spurious ^Z, which may be inserted into source files
749 * by some file transfer utilities.
751 buffer[strcspn(buffer, "\032")] = '\0';
753 list->line(LIST_READ, buffer);
755 return buffer;
759 * Tokenise a line of text. This is a very simple process since we
760 * don't need to parse the value out of e.g. numeric tokens: we
761 * simply split one string into many.
763 static Token *
764 tokenise(char *line)
766 char *p = line;
767 int type;
768 Token *list = NULL;
769 Token *t, **tail = &list;
771 while (*line)
773 p = line;
774 if (*p == '%')
776 p++;
777 if ( isdigit(*p) ||
778 ((*p == '-' || *p == '+') && isdigit(p[1])) ||
779 ((*p == '+') && (isspace(p[1]) || !p[1])))
783 p++;
785 while (isdigit(*p));
786 type = TOK_PREPROC_ID;
788 else if (*p == '{')
790 p++;
791 while (*p && *p != '}')
793 p[-1] = *p;
794 p++;
796 p[-1] = '\0';
797 if (*p)
798 p++;
799 type = TOK_PREPROC_ID;
801 else if (isidchar(*p) ||
802 ((*p == '!' || *p == '%' || *p == '$') &&
803 isidchar(p[1])))
807 p++;
809 while (isidchar(*p));
810 type = TOK_PREPROC_ID;
812 else
814 type = TOK_OTHER;
815 if (*p == '%')
816 p++;
819 else if (isidstart(*p) || (*p == '$' && isidstart(p[1])))
821 type = TOK_ID;
822 p++;
823 while (*p && isidchar(*p))
824 p++;
826 else if (*p == '\'' || *p == '"')
829 * A string token.
831 char c = *p;
832 p++;
833 type = TOK_STRING;
834 while (*p && *p != c)
835 p++;
837 if (*p)
839 p++;
841 else
843 error(ERR_WARNING, "unterminated string");
844 /* Handling unterminated strings by UNV */
845 /* type = -1; */
848 else if (isnumstart(*p))
851 * A number token.
853 type = TOK_NUMBER;
854 p++;
855 while (*p && isnumchar(*p))
856 p++;
858 else if (isspace(*p))
860 type = TOK_WHITESPACE;
861 p++;
862 while (*p && isspace(*p))
863 p++;
865 * Whitespace just before end-of-line is discarded by
866 * pretending it's a comment; whitespace just before a
867 * comment gets lumped into the comment.
869 if (!*p || *p == ';')
871 type = TOK_COMMENT;
872 while (*p)
873 p++;
876 else if (*p == ';')
878 type = TOK_COMMENT;
879 while (*p)
880 p++;
882 else
885 * Anything else is an operator of some kind. We check
886 * for all the double-character operators (>>, <<, //,
887 * %%, <=, >=, ==, !=, <>, &&, ||, ^^), but anything
888 * else is a single-character operator.
890 type = TOK_OTHER;
891 if ((p[0] == '>' && p[1] == '>') ||
892 (p[0] == '<' && p[1] == '<') ||
893 (p[0] == '/' && p[1] == '/') ||
894 (p[0] == '<' && p[1] == '=') ||
895 (p[0] == '>' && p[1] == '=') ||
896 (p[0] == '=' && p[1] == '=') ||
897 (p[0] == '!' && p[1] == '=') ||
898 (p[0] == '<' && p[1] == '>') ||
899 (p[0] == '&' && p[1] == '&') ||
900 (p[0] == '|' && p[1] == '|') ||
901 (p[0] == '^' && p[1] == '^'))
903 p++;
905 p++;
908 /* Handling unterminated string by UNV */
909 /*if (type == -1)
911 *tail = t = new_Token(NULL, TOK_STRING, line, p-line+1);
912 t->text[p-line] = *line;
913 tail = &t->next;
915 else*/
916 if (type != TOK_COMMENT)
918 *tail = t = new_Token(NULL, type, line, p - line);
919 tail = &t->next;
921 line = p;
923 return list;
927 * this function allocates a new managed block of memory and
928 * returns a pointer to the block. The managed blocks are
929 * deleted only all at once by the delete_Blocks function.
931 static void *
932 new_Block(size_t size)
934 Blocks *b = &blocks;
936 /* first, get to the end of the linked list */
937 while (b->next)
938 b = b->next;
939 /* now allocate the requested chunk */
940 b->chunk = nasm_malloc(size);
942 /* now allocate a new block for the next request */
943 b->next = nasm_malloc(sizeof(Blocks));
944 /* and initialize the contents of the new block */
945 b->next->next = NULL;
946 b->next->chunk = NULL;
947 return b->chunk;
951 * this function deletes all managed blocks of memory
953 static void
954 delete_Blocks(void)
956 Blocks *a,*b = &blocks;
959 * keep in mind that the first block, pointed to by blocks
960 * is a static and not dynamically allocated, so we don't
961 * free it.
963 while (b)
965 if (b->chunk)
966 nasm_free(b->chunk);
967 a = b;
968 b = b->next;
969 if (a != &blocks)
970 nasm_free(a);
975 * this function creates a new Token and passes a pointer to it
976 * back to the caller. It sets the type and text elements, and
977 * also the mac and next elements to NULL.
979 static Token *
980 new_Token(Token * next, int type, char *text, int txtlen)
982 Token *t;
983 int i;
985 if (freeTokens == NULL)
987 freeTokens = (Token *)new_Block(TOKEN_BLOCKSIZE * sizeof(Token));
988 for (i = 0; i < TOKEN_BLOCKSIZE - 1; i++)
989 freeTokens[i].next = &freeTokens[i + 1];
990 freeTokens[i].next = NULL;
992 t = freeTokens;
993 freeTokens = t->next;
994 t->next = next;
995 t->mac = NULL;
996 t->type = type;
997 if (type == TOK_WHITESPACE || text == NULL)
999 t->text = NULL;
1001 else
1003 if (txtlen == 0)
1004 txtlen = strlen(text);
1005 t->text = nasm_malloc(1 + txtlen);
1006 strncpy(t->text, text, txtlen);
1007 t->text[txtlen] = '\0';
1009 return t;
1012 static Token *
1013 delete_Token(Token * t)
1015 Token *next = t->next;
1016 nasm_free(t->text);
1017 t->next = freeTokens;
1018 freeTokens = t;
1019 return next;
1023 * Convert a line of tokens back into text.
1024 * If expand_locals is not zero, identifiers of the form "%$*xxx"
1025 * will be transformed into ..@ctxnum.xxx
1027 static char *
1028 detoken(Token * tlist, int expand_locals)
1030 Token *t;
1031 int len;
1032 char *line, *p;
1034 len = 0;
1035 for (t = tlist; t; t = t->next)
1037 if (t->type == TOK_PREPROC_ID && t->text[1] == '!')
1039 char *p = getenv(t->text + 2);
1040 nasm_free(t->text);
1041 if (p)
1042 t->text = nasm_strdup(p);
1043 else
1044 t->text = NULL;
1046 /* Expand local macros here and not during preprocessing */
1047 if (expand_locals &&
1048 t->type == TOK_PREPROC_ID && t->text &&
1049 t->text[0] == '%' && t->text[1] == '$')
1051 Context *ctx = get_ctx(t->text, FALSE);
1052 if (ctx)
1054 char buffer[40];
1055 char *p, *q = t->text + 2;
1057 q += strspn(q, "$");
1058 snprintf(buffer, sizeof(buffer), "..@%lu.", ctx->number);
1059 p = nasm_strcat(buffer, q);
1060 nasm_free(t->text);
1061 t->text = p;
1064 if (t->type == TOK_WHITESPACE)
1066 len++;
1068 else if (t->text)
1070 len += strlen(t->text);
1073 p = line = nasm_malloc(len + 1);
1074 for (t = tlist; t; t = t->next)
1076 if (t->type == TOK_WHITESPACE)
1078 *p = ' ';
1079 p++;
1080 *p = '\0';
1082 else if (t->text)
1084 strcpy(p, t->text);
1085 p += strlen(p);
1088 *p = '\0';
1089 return line;
1093 * A scanner, suitable for use by the expression evaluator, which
1094 * operates on a line of Tokens. Expects a pointer to a pointer to
1095 * the first token in the line to be passed in as its private_data
1096 * field.
1098 static int
1099 ppscan(void *private_data, struct tokenval *tokval)
1101 Token **tlineptr = private_data;
1102 Token *tline;
1106 tline = *tlineptr;
1107 *tlineptr = tline ? tline->next : NULL;
1109 while (tline && (tline->type == TOK_WHITESPACE ||
1110 tline->type == TOK_COMMENT));
1112 if (!tline)
1113 return tokval->t_type = TOKEN_EOS;
1115 if (tline->text[0] == '$' && !tline->text[1])
1116 return tokval->t_type = TOKEN_HERE;
1117 if (tline->text[0] == '$' && tline->text[1] == '$' && !tline->text[2])
1118 return tokval->t_type = TOKEN_BASE;
1120 if (tline->type == TOK_ID)
1122 tokval->t_charptr = tline->text;
1123 if (tline->text[0] == '$')
1125 tokval->t_charptr++;
1126 return tokval->t_type = TOKEN_ID;
1130 * This is the only special case we actually need to worry
1131 * about in this restricted context.
1133 if (!nasm_stricmp(tline->text, "seg"))
1134 return tokval->t_type = TOKEN_SEG;
1136 return tokval->t_type = TOKEN_ID;
1139 if (tline->type == TOK_NUMBER)
1141 int rn_error;
1143 tokval->t_integer = readnum(tline->text, &rn_error);
1144 if (rn_error)
1145 return tokval->t_type = TOKEN_ERRNUM;
1146 tokval->t_charptr = NULL;
1147 return tokval->t_type = TOKEN_NUM;
1150 if (tline->type == TOK_STRING)
1152 int rn_warn;
1153 char q, *r;
1154 int l;
1156 r = tline->text;
1157 q = *r++;
1158 l = strlen(r);
1160 if (l == 0 || r[l - 1] != q)
1161 return tokval->t_type = TOKEN_ERRNUM;
1162 tokval->t_integer = readstrnum(r, l - 1, &rn_warn);
1163 if (rn_warn)
1164 error(ERR_WARNING | ERR_PASS1, "character constant too long");
1165 tokval->t_charptr = NULL;
1166 return tokval->t_type = TOKEN_NUM;
1169 if (tline->type == TOK_OTHER)
1171 if (!strcmp(tline->text, "<<"))
1172 return tokval->t_type = TOKEN_SHL;
1173 if (!strcmp(tline->text, ">>"))
1174 return tokval->t_type = TOKEN_SHR;
1175 if (!strcmp(tline->text, "//"))
1176 return tokval->t_type = TOKEN_SDIV;
1177 if (!strcmp(tline->text, "%%"))
1178 return tokval->t_type = TOKEN_SMOD;
1179 if (!strcmp(tline->text, "=="))
1180 return tokval->t_type = TOKEN_EQ;
1181 if (!strcmp(tline->text, "<>"))
1182 return tokval->t_type = TOKEN_NE;
1183 if (!strcmp(tline->text, "!="))
1184 return tokval->t_type = TOKEN_NE;
1185 if (!strcmp(tline->text, "<="))
1186 return tokval->t_type = TOKEN_LE;
1187 if (!strcmp(tline->text, ">="))
1188 return tokval->t_type = TOKEN_GE;
1189 if (!strcmp(tline->text, "&&"))
1190 return tokval->t_type = TOKEN_DBL_AND;
1191 if (!strcmp(tline->text, "^^"))
1192 return tokval->t_type = TOKEN_DBL_XOR;
1193 if (!strcmp(tline->text, "||"))
1194 return tokval->t_type = TOKEN_DBL_OR;
1198 * We have no other options: just return the first character of
1199 * the token text.
1201 return tokval->t_type = tline->text[0];
1205 * Compare a string to the name of an existing macro; this is a
1206 * simple wrapper which calls either strcmp or nasm_stricmp
1207 * depending on the value of the `casesense' parameter.
1209 static int
1210 mstrcmp(char *p, char *q, int casesense)
1212 return casesense ? strcmp(p, q) : nasm_stricmp(p, q);
1216 * Return the Context structure associated with a %$ token. Return
1217 * NULL, having _already_ reported an error condition, if the
1218 * context stack isn't deep enough for the supplied number of $
1219 * signs.
1220 * If all_contexts == TRUE, contexts that enclose current are
1221 * also scanned for such smacro, until it is found; if not -
1222 * only the context that directly results from the number of $'s
1223 * in variable's name.
1225 static Context *
1226 get_ctx(char *name, int all_contexts)
1228 Context *ctx;
1229 SMacro *m;
1230 int i;
1232 if (!name || name[0] != '%' || name[1] != '$')
1233 return NULL;
1235 if (!cstk)
1237 error(ERR_NONFATAL, "`%s': context stack is empty", name);
1238 return NULL;
1241 for (i = strspn(name + 2, "$"), ctx = cstk; (i > 0) && ctx; i--)
1243 ctx = ctx->next;
1244 /* i--; Lino - 02/25/02 */
1246 if (!ctx)
1248 error(ERR_NONFATAL, "`%s': context stack is only"
1249 " %d level%s deep", name, i - 1, (i == 2 ? "" : "s"));
1250 return NULL;
1252 if (!all_contexts)
1253 return ctx;
1257 /* Search for this smacro in found context */
1258 m = ctx->localmac;
1259 while (m)
1261 if (!mstrcmp(m->name, name, m->casesense))
1262 return ctx;
1263 m = m->next;
1265 ctx = ctx->next;
1267 while (ctx);
1268 return NULL;
1272 * Open an include file. This routine must always return a valid
1273 * file pointer if it returns - it's responsible for throwing an
1274 * ERR_FATAL and bombing out completely if not. It should also try
1275 * the include path one by one until it finds the file or reaches
1276 * the end of the path.
1278 static FILE *
1279 inc_fopen(char *file)
1281 FILE *fp;
1282 char *prefix = "", *combine;
1283 IncPath *ip = ipath;
1284 static int namelen = 0;
1285 int len = strlen(file);
1287 while (1)
1289 combine = nasm_malloc(strlen(prefix) + len + 1);
1290 strcpy(combine, prefix);
1291 strcat(combine, file);
1292 fp = fopen(combine, "r");
1293 if (pass == 0 && fp)
1295 namelen += strlen(combine) + 1;
1296 if (namelen > 62)
1298 printf(" \\\n ");
1299 namelen = 2;
1301 printf(" %s", combine);
1303 nasm_free(combine);
1304 if (fp)
1305 return fp;
1306 if (!ip)
1307 break;
1308 prefix = ip->path;
1309 ip = ip->next;
1312 error(ERR_FATAL, "unable to open include file `%s'", file);
1313 return NULL; /* never reached - placate compilers */
1317 * Determine if we should warn on defining a single-line macro of
1318 * name `name', with `nparam' parameters. If nparam is 0 or -1, will
1319 * return TRUE if _any_ single-line macro of that name is defined.
1320 * Otherwise, will return TRUE if a single-line macro with either
1321 * `nparam' or no parameters is defined.
1323 * If a macro with precisely the right number of parameters is
1324 * defined, or nparam is -1, the address of the definition structure
1325 * will be returned in `defn'; otherwise NULL will be returned. If `defn'
1326 * is NULL, no action will be taken regarding its contents, and no
1327 * error will occur.
1329 * Note that this is also called with nparam zero to resolve
1330 * `ifdef'.
1332 * If you already know which context macro belongs to, you can pass
1333 * the context pointer as first parameter; if you won't but name begins
1334 * with %$ the context will be automatically computed. If all_contexts
1335 * is true, macro will be searched in outer contexts as well.
1337 static int
1338 smacro_defined(Context * ctx, char *name, int nparam, SMacro ** defn,
1339 int nocase)
1341 SMacro *m;
1343 if (ctx)
1344 m = ctx->localmac;
1345 else if (name[0] == '%' && name[1] == '$')
1347 if (cstk)
1348 ctx = get_ctx(name, FALSE);
1349 if (!ctx)
1350 return FALSE; /* got to return _something_ */
1351 m = ctx->localmac;
1353 else
1354 m = smacros[hash(name)];
1356 while (m)
1358 if (!mstrcmp(m->name, name, m->casesense && nocase) &&
1359 (nparam <= 0 || m->nparam == 0 || nparam == m->nparam))
1361 if (defn)
1363 if (nparam == m->nparam || nparam == -1)
1364 *defn = m;
1365 else
1366 *defn = NULL;
1368 return TRUE;
1370 m = m->next;
1373 return FALSE;
1377 * Count and mark off the parameters in a multi-line macro call.
1378 * This is called both from within the multi-line macro expansion
1379 * code, and also to mark off the default parameters when provided
1380 * in a %macro definition line.
1382 static void
1383 count_mmac_params(Token * t, int *nparam, Token *** params)
1385 int paramsize, brace;
1387 *nparam = paramsize = 0;
1388 *params = NULL;
1389 while (t)
1391 if (*nparam >= paramsize)
1393 paramsize += PARAM_DELTA;
1394 *params = nasm_realloc(*params, sizeof(**params) * paramsize);
1396 skip_white_(t);
1397 brace = FALSE;
1398 if (tok_is_(t, "{"))
1399 brace = TRUE;
1400 (*params)[(*nparam)++] = t;
1401 while (tok_isnt_(t, brace ? "}" : ","))
1402 t = t->next;
1403 if (t)
1404 { /* got a comma/brace */
1405 t = t->next;
1406 if (brace)
1409 * Now we've found the closing brace, look further
1410 * for the comma.
1412 skip_white_(t);
1413 if (tok_isnt_(t, ","))
1415 error(ERR_NONFATAL,
1416 "braces do not enclose all of macro parameter");
1417 while (tok_isnt_(t, ","))
1418 t = t->next;
1420 if (t)
1421 t = t->next; /* eat the comma */
1428 * Determine whether one of the various `if' conditions is true or
1429 * not.
1431 * We must free the tline we get passed.
1433 static int
1434 if_condition(Token * tline, int i)
1436 int j, casesense;
1437 Token *t, *tt, **tptr, *origline;
1438 struct tokenval tokval;
1439 expr *evalresult;
1441 origline = tline;
1443 switch (i)
1445 case PP_IFCTX:
1446 case PP_ELIFCTX:
1447 case PP_IFNCTX:
1448 case PP_ELIFNCTX:
1449 j = FALSE; /* have we matched yet? */
1450 while (cstk && tline)
1452 skip_white_(tline);
1453 if (!tline || tline->type != TOK_ID)
1455 error(ERR_NONFATAL,
1456 "`%s' expects context identifiers",
1457 directives[i]);
1458 free_tlist(origline);
1459 return -1;
1461 if (!nasm_stricmp(tline->text, cstk->name))
1462 j = TRUE;
1463 tline = tline->next;
1465 if (i == PP_IFNCTX || i == PP_ELIFNCTX)
1466 j = !j;
1467 free_tlist(origline);
1468 return j;
1470 case PP_IFDEF:
1471 case PP_ELIFDEF:
1472 case PP_IFNDEF:
1473 case PP_ELIFNDEF:
1474 j = FALSE; /* have we matched yet? */
1475 while (tline)
1477 skip_white_(tline);
1478 if (!tline || (tline->type != TOK_ID &&
1479 (tline->type != TOK_PREPROC_ID ||
1480 tline->text[1] != '$')))
1482 error(ERR_NONFATAL,
1483 "`%s' expects macro identifiers",
1484 directives[i]);
1485 free_tlist(origline);
1486 return -1;
1488 if (smacro_defined(NULL, tline->text, 0, NULL, 1))
1489 j = TRUE;
1490 tline = tline->next;
1492 if (i == PP_IFNDEF || i == PP_ELIFNDEF)
1493 j = !j;
1494 free_tlist(origline);
1495 return j;
1497 case PP_IFIDN:
1498 case PP_ELIFIDN:
1499 case PP_IFNIDN:
1500 case PP_ELIFNIDN:
1501 case PP_IFIDNI:
1502 case PP_ELIFIDNI:
1503 case PP_IFNIDNI:
1504 case PP_ELIFNIDNI:
1505 tline = expand_smacro(tline);
1506 t = tt = tline;
1507 while (tok_isnt_(tt, ","))
1508 tt = tt->next;
1509 if (!tt)
1511 error(ERR_NONFATAL,
1512 "`%s' expects two comma-separated arguments",
1513 directives[i]);
1514 free_tlist(tline);
1515 return -1;
1517 tt = tt->next;
1518 casesense = (i == PP_IFIDN || i == PP_ELIFIDN ||
1519 i == PP_IFNIDN || i == PP_ELIFNIDN);
1520 j = TRUE; /* assume equality unless proved not */
1521 while ((t->type != TOK_OTHER || strcmp(t->text, ",")) && tt)
1523 if (tt->type == TOK_OTHER && !strcmp(tt->text, ","))
1525 error(ERR_NONFATAL, "`%s': more than one comma on line",
1526 directives[i]);
1527 free_tlist(tline);
1528 return -1;
1530 if (t->type == TOK_WHITESPACE)
1532 t = t->next;
1533 continue;
1535 if (tt->type == TOK_WHITESPACE)
1537 tt = tt->next;
1538 continue;
1540 if (tt->type != t->type)
1542 j = FALSE; /* found mismatching tokens */
1543 break;
1545 /* Unify surrounding quotes for strings */
1546 if (t->type == TOK_STRING)
1548 tt->text[0] = t->text[0];
1549 tt->text[strlen(tt->text) - 1] = t->text[0];
1551 if (mstrcmp(tt->text, t->text, casesense) != 0)
1553 j = FALSE; /* found mismatching tokens */
1554 break;
1557 t = t->next;
1558 tt = tt->next;
1560 if ((t->type != TOK_OTHER || strcmp(t->text, ",")) || tt)
1561 j = FALSE; /* trailing gunk on one end or other */
1562 if (i == PP_IFNIDN || i == PP_ELIFNIDN ||
1563 i == PP_IFNIDNI || i == PP_ELIFNIDNI)
1564 j = !j;
1565 free_tlist(tline);
1566 return j;
1568 case PP_IFMACRO:
1569 case PP_ELIFMACRO:
1570 case PP_IFNMACRO:
1571 case PP_ELIFNMACRO:
1573 int found = 0;
1574 MMacro searching, *mmac;
1576 tline = tline->next;
1577 skip_white_(tline);
1578 tline = expand_id(tline);
1579 if (!tok_type_(tline, TOK_ID))
1581 error(ERR_NONFATAL,
1582 "`%s' expects a macro name",
1583 directives[i]);
1584 return -1;
1586 searching.name = nasm_strdup(tline->text);
1587 searching.casesense = (i == PP_MACRO);
1588 searching.plus = FALSE;
1589 searching.nolist = FALSE;
1590 searching.in_progress = FALSE;
1591 searching.rep_nest = NULL;
1592 searching.nparam_min = 0;
1593 searching.nparam_max = INT_MAX;
1594 tline = expand_smacro(tline->next);
1595 skip_white_(tline);
1596 if (!tline)
1598 } else if (!tok_type_(tline, TOK_NUMBER))
1600 error(ERR_NONFATAL,
1601 "`%s' expects a parameter count or nothing",
1602 directives[i]);
1604 else
1606 searching.nparam_min = searching.nparam_max =
1607 readnum(tline->text, &j);
1608 if (j)
1609 error(ERR_NONFATAL,
1610 "unable to parse parameter count `%s'",
1611 tline->text);
1613 if (tline && tok_is_(tline->next, "-"))
1615 tline = tline->next->next;
1616 if (tok_is_(tline, "*"))
1617 searching.nparam_max = INT_MAX;
1618 else if (!tok_type_(tline, TOK_NUMBER))
1619 error(ERR_NONFATAL,
1620 "`%s' expects a parameter count after `-'",
1621 directives[i]);
1622 else
1624 searching.nparam_max = readnum(tline->text, &j);
1625 if (j)
1626 error(ERR_NONFATAL,
1627 "unable to parse parameter count `%s'",
1628 tline->text);
1629 if (searching.nparam_min > searching.nparam_max)
1630 error(ERR_NONFATAL,
1631 "minimum parameter count exceeds maximum");
1634 if (tline && tok_is_(tline->next, "+"))
1636 tline = tline->next;
1637 searching.plus = TRUE;
1639 mmac = mmacros[hash(searching.name)];
1640 while (mmac)
1642 if (!strcmp(mmac->name, searching.name) &&
1643 (mmac->nparam_min <= searching.nparam_max
1644 || searching.plus)
1645 && (searching.nparam_min <= mmac->nparam_max
1646 || mmac->plus))
1648 found = TRUE;
1649 break;
1651 mmac = mmac->next;
1653 nasm_free(searching.name);
1654 free_tlist(origline);
1655 if (i == PP_IFNMACRO || i == PP_ELIFNMACRO)
1656 found = !found;
1657 return found;
1660 case PP_IFID:
1661 case PP_ELIFID:
1662 case PP_IFNID:
1663 case PP_ELIFNID:
1664 case PP_IFNUM:
1665 case PP_ELIFNUM:
1666 case PP_IFNNUM:
1667 case PP_ELIFNNUM:
1668 case PP_IFSTR:
1669 case PP_ELIFSTR:
1670 case PP_IFNSTR:
1671 case PP_ELIFNSTR:
1672 tline = expand_smacro(tline);
1673 t = tline;
1674 while (tok_type_(t, TOK_WHITESPACE))
1675 t = t->next;
1676 j = FALSE; /* placate optimiser */
1677 if (t)
1678 switch (i)
1680 case PP_IFID:
1681 case PP_ELIFID:
1682 case PP_IFNID:
1683 case PP_ELIFNID:
1684 j = (t->type == TOK_ID);
1685 break;
1686 case PP_IFNUM:
1687 case PP_ELIFNUM:
1688 case PP_IFNNUM:
1689 case PP_ELIFNNUM:
1690 j = (t->type == TOK_NUMBER);
1691 break;
1692 case PP_IFSTR:
1693 case PP_ELIFSTR:
1694 case PP_IFNSTR:
1695 case PP_ELIFNSTR:
1696 j = (t->type == TOK_STRING);
1697 break;
1699 if (i == PP_IFNID || i == PP_ELIFNID ||
1700 i == PP_IFNNUM || i == PP_ELIFNNUM ||
1701 i == PP_IFNSTR || i == PP_ELIFNSTR)
1702 j = !j;
1703 free_tlist(tline);
1704 return j;
1706 case PP_IF:
1707 case PP_ELIF:
1708 t = tline = expand_smacro(tline);
1709 tptr = &t;
1710 tokval.t_type = TOKEN_INVALID;
1711 evalresult = evaluate(ppscan, tptr, &tokval,
1712 NULL, pass | CRITICAL, error, NULL);
1713 free_tlist(tline);
1714 if (!evalresult)
1715 return -1;
1716 if (tokval.t_type)
1717 error(ERR_WARNING,
1718 "trailing garbage after expression ignored");
1719 if (!is_simple(evalresult))
1721 error(ERR_NONFATAL,
1722 "non-constant value given to `%s'", directives[i]);
1723 return -1;
1725 return reloc_value(evalresult) != 0;
1727 default:
1728 error(ERR_FATAL,
1729 "preprocessor directive `%s' not yet implemented",
1730 directives[i]);
1731 free_tlist(origline);
1732 return -1; /* yeah, right */
1737 * Expand macros in a string. Used in %error and %include directives.
1738 * First tokenise the string, apply "expand_smacro" and then de-tokenise back.
1739 * The returned variable should ALWAYS be freed after usage.
1741 void
1742 expand_macros_in_string(char **p)
1744 Token *line = tokenise(*p);
1745 line = expand_smacro(line);
1746 *p = detoken(line, FALSE);
1750 * find and process preprocessor directive in passed line
1751 * Find out if a line contains a preprocessor directive, and deal
1752 * with it if so.
1754 * If a directive _is_ found, it is the responsibility of this routine
1755 * (and not the caller) to free_tlist() the line.
1757 * @param tline a pointer to the current tokeninzed line linked list
1758 * @return DIRECTIVE_FOUND or NO_DIRECTIVE_FOUND
1761 static int
1762 do_directive(Token * tline)
1764 int i, j, k, m, nparam, nolist;
1765 int offset;
1766 char *p, *mname;
1767 Include *inc;
1768 Context *ctx;
1769 Cond *cond;
1770 SMacro *smac, **smhead;
1771 MMacro *mmac;
1772 Token *t, *tt, *param_start, *macro_start, *last, **tptr, *origline;
1773 Line *l;
1774 struct tokenval tokval;
1775 expr *evalresult;
1776 MMacro *tmp_defining; /* Used when manipulating rep_nest */
1778 origline = tline;
1780 skip_white_(tline);
1781 if (!tok_type_(tline, TOK_PREPROC_ID) ||
1782 (tline->text[1] == '%' || tline->text[1] == '$'
1783 || tline->text[1] == '!'))
1784 return NO_DIRECTIVE_FOUND;
1786 i = -1;
1787 j = elements(directives);
1788 while (j - i > 1)
1790 k = (j + i) / 2;
1791 m = nasm_stricmp(tline->text, directives[k]);
1792 if (m == 0) {
1793 if (tasm_compatible_mode) {
1794 i = k;
1795 j = -2;
1796 } else if (k != PP_ARG && k != PP_LOCAL && k != PP_STACKSIZE) {
1797 i = k;
1798 j = -2;
1800 break;
1802 else if (m < 0) {
1803 j = k;
1805 else
1806 i = k;
1810 * If we're in a non-emitting branch of a condition construct,
1811 * or walking to the end of an already terminated %rep block,
1812 * we should ignore all directives except for condition
1813 * directives.
1815 if (((istk->conds && !emitting(istk->conds->state)) ||
1816 (istk->mstk && !istk->mstk->in_progress)) &&
1817 !is_condition(i))
1819 return NO_DIRECTIVE_FOUND;
1823 * If we're defining a macro or reading a %rep block, we should
1824 * ignore all directives except for %macro/%imacro (which
1825 * generate an error), %endm/%endmacro, and (only if we're in a
1826 * %rep block) %endrep. If we're in a %rep block, another %rep
1827 * causes an error, so should be let through.
1829 if (defining && i != PP_MACRO && i != PP_IMACRO &&
1830 i != PP_ENDMACRO && i != PP_ENDM &&
1831 (defining->name || (i != PP_ENDREP && i != PP_REP)))
1833 return NO_DIRECTIVE_FOUND;
1836 if (j != -2)
1838 error(ERR_NONFATAL, "unknown preprocessor directive `%s'",
1839 tline->text);
1840 return NO_DIRECTIVE_FOUND; /* didn't get it */
1843 switch (i)
1845 case PP_STACKSIZE:
1846 /* Directive to tell NASM what the default stack size is. The
1847 * default is for a 16-bit stack, and this can be overriden with
1848 * %stacksize large.
1849 * the following form:
1851 * ARG arg1:WORD, arg2:DWORD, arg4:QWORD
1853 tline = tline->next;
1854 if (tline && tline->type == TOK_WHITESPACE)
1855 tline = tline->next;
1856 if (!tline || tline->type != TOK_ID)
1858 error(ERR_NONFATAL, "`%%stacksize' missing size parameter");
1859 free_tlist(origline);
1860 return DIRECTIVE_FOUND;
1862 if (nasm_stricmp(tline->text, "flat") == 0)
1864 /* All subsequent ARG directives are for a 32-bit stack */
1865 StackSize = 4;
1866 StackPointer = "ebp";
1867 ArgOffset = 8;
1868 LocalOffset = 4;
1870 else if (nasm_stricmp(tline->text, "large") == 0)
1872 /* All subsequent ARG directives are for a 16-bit stack,
1873 * far function call.
1875 StackSize = 2;
1876 StackPointer = "bp";
1877 ArgOffset = 4;
1878 LocalOffset = 2;
1880 else if (nasm_stricmp(tline->text, "small") == 0)
1882 /* All subsequent ARG directives are for a 16-bit stack,
1883 * far function call. We don't support near functions.
1885 StackSize = 2;
1886 StackPointer = "bp";
1887 ArgOffset = 6;
1888 LocalOffset = 2;
1890 else
1892 error(ERR_NONFATAL, "`%%stacksize' invalid size type");
1893 free_tlist(origline);
1894 return DIRECTIVE_FOUND;
1896 free_tlist(origline);
1897 return DIRECTIVE_FOUND;
1899 case PP_ARG:
1900 /* TASM like ARG directive to define arguments to functions, in
1901 * the following form:
1903 * ARG arg1:WORD, arg2:DWORD, arg4:QWORD
1905 offset = ArgOffset;
1908 char *arg, directive[256];
1909 int size = StackSize;
1911 /* Find the argument name */
1912 tline = tline->next;
1913 if (tline && tline->type == TOK_WHITESPACE)
1914 tline = tline->next;
1915 if (!tline || tline->type != TOK_ID)
1917 error(ERR_NONFATAL, "`%%arg' missing argument parameter");
1918 free_tlist(origline);
1919 return DIRECTIVE_FOUND;
1921 arg = tline->text;
1923 /* Find the argument size type */
1924 tline = tline->next;
1925 if (!tline || tline->type != TOK_OTHER
1926 || tline->text[0] != ':')
1928 error(ERR_NONFATAL,
1929 "Syntax error processing `%%arg' directive");
1930 free_tlist(origline);
1931 return DIRECTIVE_FOUND;
1933 tline = tline->next;
1934 if (!tline || tline->type != TOK_ID)
1936 error(ERR_NONFATAL,
1937 "`%%arg' missing size type parameter");
1938 free_tlist(origline);
1939 return DIRECTIVE_FOUND;
1942 /* Allow macro expansion of type parameter */
1943 tt = tokenise(tline->text);
1944 tt = expand_smacro(tt);
1945 if (nasm_stricmp(tt->text, "byte") == 0)
1947 size = MAX(StackSize, 1);
1949 else if (nasm_stricmp(tt->text, "word") == 0)
1951 size = MAX(StackSize, 2);
1953 else if (nasm_stricmp(tt->text, "dword") == 0)
1955 size = MAX(StackSize, 4);
1957 else if (nasm_stricmp(tt->text, "qword") == 0)
1959 size = MAX(StackSize, 8);
1961 else if (nasm_stricmp(tt->text, "tword") == 0)
1963 size = MAX(StackSize, 10);
1965 else
1967 error(ERR_NONFATAL,
1968 "Invalid size type for `%%arg' missing directive");
1969 free_tlist(tt);
1970 free_tlist(origline);
1971 return DIRECTIVE_FOUND;
1973 free_tlist(tt);
1975 /* Now define the macro for the argument */
1976 snprintf(directive, sizeof(directive), "%%define %s (%s+%d)", arg, StackPointer,
1977 offset);
1978 do_directive(tokenise(directive));
1979 offset += size;
1981 /* Move to the next argument in the list */
1982 tline = tline->next;
1983 if (tline && tline->type == TOK_WHITESPACE)
1984 tline = tline->next;
1986 while (tline && tline->type == TOK_OTHER
1987 && tline->text[0] == ',');
1988 free_tlist(origline);
1989 return DIRECTIVE_FOUND;
1991 case PP_LOCAL:
1992 /* TASM like LOCAL directive to define local variables for a
1993 * function, in the following form:
1995 * LOCAL local1:WORD, local2:DWORD, local4:QWORD = LocalSize
1997 * The '= LocalSize' at the end is ignored by NASM, but is
1998 * required by TASM to define the local parameter size (and used
1999 * by the TASM macro package).
2001 offset = LocalOffset;
2004 char *local, directive[256];
2005 int size = StackSize;
2007 /* Find the argument name */
2008 tline = tline->next;
2009 if (tline && tline->type == TOK_WHITESPACE)
2010 tline = tline->next;
2011 if (!tline || tline->type != TOK_ID)
2013 error(ERR_NONFATAL,
2014 "`%%local' missing argument parameter");
2015 free_tlist(origline);
2016 return DIRECTIVE_FOUND;
2018 local = tline->text;
2020 /* Find the argument size type */
2021 tline = tline->next;
2022 if (!tline || tline->type != TOK_OTHER
2023 || tline->text[0] != ':')
2025 error(ERR_NONFATAL,
2026 "Syntax error processing `%%local' directive");
2027 free_tlist(origline);
2028 return DIRECTIVE_FOUND;
2030 tline = tline->next;
2031 if (!tline || tline->type != TOK_ID)
2033 error(ERR_NONFATAL,
2034 "`%%local' missing size type parameter");
2035 free_tlist(origline);
2036 return DIRECTIVE_FOUND;
2039 /* Allow macro expansion of type parameter */
2040 tt = tokenise(tline->text);
2041 tt = expand_smacro(tt);
2042 if (nasm_stricmp(tt->text, "byte") == 0)
2044 size = MAX(StackSize, 1);
2046 else if (nasm_stricmp(tt->text, "word") == 0)
2048 size = MAX(StackSize, 2);
2050 else if (nasm_stricmp(tt->text, "dword") == 0)
2052 size = MAX(StackSize, 4);
2054 else if (nasm_stricmp(tt->text, "qword") == 0)
2056 size = MAX(StackSize, 8);
2058 else if (nasm_stricmp(tt->text, "tword") == 0)
2060 size = MAX(StackSize, 10);
2062 else
2064 error(ERR_NONFATAL,
2065 "Invalid size type for `%%local' missing directive");
2066 free_tlist(tt);
2067 free_tlist(origline);
2068 return DIRECTIVE_FOUND;
2070 free_tlist(tt);
2072 /* Now define the macro for the argument */
2073 snprintf(directive, sizeof(directive), "%%define %s (%s-%d)", local, StackPointer,
2074 offset);
2075 do_directive(tokenise(directive));
2076 offset += size;
2078 /* Now define the assign to setup the enter_c macro correctly */
2079 snprintf(directive, sizeof(directive), "%%assign %%$localsize %%$localsize+%d",
2080 size);
2081 do_directive(tokenise(directive));
2083 /* Move to the next argument in the list */
2084 tline = tline->next;
2085 if (tline && tline->type == TOK_WHITESPACE)
2086 tline = tline->next;
2088 while (tline && tline->type == TOK_OTHER
2089 && tline->text[0] == ',');
2090 free_tlist(origline);
2091 return DIRECTIVE_FOUND;
2093 case PP_CLEAR:
2094 if (tline->next)
2095 error(ERR_WARNING,
2096 "trailing garbage after `%%clear' ignored");
2097 for (j = 0; j < NHASH; j++)
2099 while (mmacros[j])
2101 MMacro *m = mmacros[j];
2102 mmacros[j] = m->next;
2103 free_mmacro(m);
2105 while (smacros[j])
2107 SMacro *s = smacros[j];
2108 smacros[j] = smacros[j]->next;
2109 nasm_free(s->name);
2110 free_tlist(s->expansion);
2111 nasm_free(s);
2114 free_tlist(origline);
2115 return DIRECTIVE_FOUND;
2117 case PP_INCLUDE:
2118 tline = tline->next;
2119 skip_white_(tline);
2120 if (!tline || (tline->type != TOK_STRING &&
2121 tline->type != TOK_INTERNAL_STRING))
2123 error(ERR_NONFATAL, "`%%include' expects a file name");
2124 free_tlist(origline);
2125 return DIRECTIVE_FOUND; /* but we did _something_ */
2127 if (tline->next)
2128 error(ERR_WARNING,
2129 "trailing garbage after `%%include' ignored");
2130 if (tline->type != TOK_INTERNAL_STRING)
2132 p = tline->text + 1; /* point past the quote to the name */
2133 p[strlen(p) - 1] = '\0'; /* remove the trailing quote */
2135 else
2136 p = tline->text; /* internal_string is easier */
2137 expand_macros_in_string(&p);
2138 inc = nasm_malloc(sizeof(Include));
2139 inc->next = istk;
2140 inc->conds = NULL;
2141 inc->fp = inc_fopen(p);
2142 inc->fname = src_set_fname(p);
2143 inc->lineno = src_set_linnum(0);
2144 inc->lineinc = 1;
2145 inc->expansion = NULL;
2146 inc->mstk = NULL;
2147 istk = inc;
2148 list->uplevel(LIST_INCLUDE);
2149 free_tlist(origline);
2150 return DIRECTIVE_FOUND;
2152 case PP_PUSH:
2153 tline = tline->next;
2154 skip_white_(tline);
2155 tline = expand_id(tline);
2156 if (!tok_type_(tline, TOK_ID))
2158 error(ERR_NONFATAL, "`%%push' expects a context identifier");
2159 free_tlist(origline);
2160 return DIRECTIVE_FOUND; /* but we did _something_ */
2162 if (tline->next)
2163 error(ERR_WARNING, "trailing garbage after `%%push' ignored");
2164 ctx = nasm_malloc(sizeof(Context));
2165 ctx->next = cstk;
2166 ctx->localmac = NULL;
2167 ctx->name = nasm_strdup(tline->text);
2168 ctx->number = unique++;
2169 cstk = ctx;
2170 free_tlist(origline);
2171 break;
2173 case PP_REPL:
2174 tline = tline->next;
2175 skip_white_(tline);
2176 tline = expand_id(tline);
2177 if (!tok_type_(tline, TOK_ID))
2179 error(ERR_NONFATAL, "`%%repl' expects a context identifier");
2180 free_tlist(origline);
2181 return DIRECTIVE_FOUND; /* but we did _something_ */
2183 if (tline->next)
2184 error(ERR_WARNING, "trailing garbage after `%%repl' ignored");
2185 if (!cstk)
2186 error(ERR_NONFATAL, "`%%repl': context stack is empty");
2187 else
2189 nasm_free(cstk->name);
2190 cstk->name = nasm_strdup(tline->text);
2192 free_tlist(origline);
2193 break;
2195 case PP_POP:
2196 if (tline->next)
2197 error(ERR_WARNING, "trailing garbage after `%%pop' ignored");
2198 if (!cstk)
2199 error(ERR_NONFATAL,
2200 "`%%pop': context stack is already empty");
2201 else
2202 ctx_pop();
2203 free_tlist(origline);
2204 break;
2206 case PP_ERROR:
2207 tline->next = expand_smacro(tline->next);
2208 tline = tline->next;
2209 skip_white_(tline);
2210 if (tok_type_(tline, TOK_STRING))
2212 p = tline->text + 1; /* point past the quote to the name */
2213 p[strlen(p) - 1] = '\0'; /* remove the trailing quote */
2214 expand_macros_in_string(&p);
2215 error(ERR_NONFATAL, "%s", p);
2216 nasm_free(p);
2218 else
2220 p = detoken(tline, FALSE);
2221 error(ERR_WARNING, "%s", p);
2222 nasm_free(p);
2224 free_tlist(origline);
2225 break;
2227 case PP_IF:
2228 case PP_IFCTX:
2229 case PP_IFDEF:
2230 case PP_IFID:
2231 case PP_IFIDN:
2232 case PP_IFIDNI:
2233 case PP_IFMACRO:
2234 case PP_IFNCTX:
2235 case PP_IFNDEF:
2236 case PP_IFNID:
2237 case PP_IFNIDN:
2238 case PP_IFNIDNI:
2239 case PP_IFNMACRO:
2240 case PP_IFNNUM:
2241 case PP_IFNSTR:
2242 case PP_IFNUM:
2243 case PP_IFSTR:
2244 if (istk->conds && !emitting(istk->conds->state))
2245 j = COND_NEVER;
2246 else
2248 j = if_condition(tline->next, i);
2249 tline->next = NULL; /* it got freed */
2250 free_tlist(origline);
2251 j = j < 0 ? COND_NEVER : j ? COND_IF_TRUE : COND_IF_FALSE;
2253 cond = nasm_malloc(sizeof(Cond));
2254 cond->next = istk->conds;
2255 cond->state = j;
2256 istk->conds = cond;
2257 return DIRECTIVE_FOUND;
2259 case PP_ELIF:
2260 case PP_ELIFCTX:
2261 case PP_ELIFDEF:
2262 case PP_ELIFID:
2263 case PP_ELIFIDN:
2264 case PP_ELIFIDNI:
2265 case PP_ELIFMACRO:
2266 case PP_ELIFNCTX:
2267 case PP_ELIFNDEF:
2268 case PP_ELIFNID:
2269 case PP_ELIFNIDN:
2270 case PP_ELIFNIDNI:
2271 case PP_ELIFNMACRO:
2272 case PP_ELIFNNUM:
2273 case PP_ELIFNSTR:
2274 case PP_ELIFNUM:
2275 case PP_ELIFSTR:
2276 if (!istk->conds)
2277 error(ERR_FATAL, "`%s': no matching `%%if'", directives[i]);
2278 if (emitting(istk->conds->state)
2279 || istk->conds->state == COND_NEVER)
2280 istk->conds->state = COND_NEVER;
2281 else
2284 * IMPORTANT: In the case of %if, we will already have
2285 * called expand_mmac_params(); however, if we're
2286 * processing an %elif we must have been in a
2287 * non-emitting mode, which would have inhibited
2288 * the normal invocation of expand_mmac_params(). Therefore,
2289 * we have to do it explicitly here.
2291 j = if_condition(expand_mmac_params(tline->next), i);
2292 tline->next = NULL; /* it got freed */
2293 free_tlist(origline);
2294 istk->conds->state =
2295 j < 0 ? COND_NEVER : j ? COND_IF_TRUE : COND_IF_FALSE;
2297 return DIRECTIVE_FOUND;
2299 case PP_ELSE:
2300 if (tline->next)
2301 error(ERR_WARNING, "trailing garbage after `%%else' ignored");
2302 if (!istk->conds)
2303 error(ERR_FATAL, "`%%else': no matching `%%if'");
2304 if (emitting(istk->conds->state)
2305 || istk->conds->state == COND_NEVER)
2306 istk->conds->state = COND_ELSE_FALSE;
2307 else
2308 istk->conds->state = COND_ELSE_TRUE;
2309 free_tlist(origline);
2310 return DIRECTIVE_FOUND;
2312 case PP_ENDIF:
2313 if (tline->next)
2314 error(ERR_WARNING,
2315 "trailing garbage after `%%endif' ignored");
2316 if (!istk->conds)
2317 error(ERR_FATAL, "`%%endif': no matching `%%if'");
2318 cond = istk->conds;
2319 istk->conds = cond->next;
2320 nasm_free(cond);
2321 free_tlist(origline);
2322 return DIRECTIVE_FOUND;
2324 case PP_MACRO:
2325 case PP_IMACRO:
2326 if (defining)
2327 error(ERR_FATAL,
2328 "`%%%smacro': already defining a macro",
2329 (i == PP_IMACRO ? "i" : ""));
2330 tline = tline->next;
2331 skip_white_(tline);
2332 tline = expand_id(tline);
2333 if (!tok_type_(tline, TOK_ID))
2335 error(ERR_NONFATAL,
2336 "`%%%smacro' expects a macro name",
2337 (i == PP_IMACRO ? "i" : ""));
2338 return DIRECTIVE_FOUND;
2340 defining = nasm_malloc(sizeof(MMacro));
2341 defining->name = nasm_strdup(tline->text);
2342 defining->casesense = (i == PP_MACRO);
2343 defining->plus = FALSE;
2344 defining->nolist = FALSE;
2345 defining->in_progress = FALSE;
2346 defining->rep_nest = NULL;
2347 tline = expand_smacro(tline->next);
2348 skip_white_(tline);
2349 if (!tok_type_(tline, TOK_NUMBER))
2351 error(ERR_NONFATAL,
2352 "`%%%smacro' expects a parameter count",
2353 (i == PP_IMACRO ? "i" : ""));
2354 defining->nparam_min = defining->nparam_max = 0;
2356 else
2358 defining->nparam_min = defining->nparam_max =
2359 readnum(tline->text, &j);
2360 if (j)
2361 error(ERR_NONFATAL,
2362 "unable to parse parameter count `%s'",
2363 tline->text);
2365 if (tline && tok_is_(tline->next, "-"))
2367 tline = tline->next->next;
2368 if (tok_is_(tline, "*"))
2369 defining->nparam_max = INT_MAX;
2370 else if (!tok_type_(tline, TOK_NUMBER))
2371 error(ERR_NONFATAL,
2372 "`%%%smacro' expects a parameter count after `-'",
2373 (i == PP_IMACRO ? "i" : ""));
2374 else
2376 defining->nparam_max = readnum(tline->text, &j);
2377 if (j)
2378 error(ERR_NONFATAL,
2379 "unable to parse parameter count `%s'",
2380 tline->text);
2381 if (defining->nparam_min > defining->nparam_max)
2382 error(ERR_NONFATAL,
2383 "minimum parameter count exceeds maximum");
2386 if (tline && tok_is_(tline->next, "+"))
2388 tline = tline->next;
2389 defining->plus = TRUE;
2391 if (tline && tok_type_(tline->next, TOK_ID) &&
2392 !nasm_stricmp(tline->next->text, ".nolist"))
2394 tline = tline->next;
2395 defining->nolist = TRUE;
2397 mmac = mmacros[hash(defining->name)];
2398 while (mmac)
2400 if (!strcmp(mmac->name, defining->name) &&
2401 (mmac->nparam_min <= defining->nparam_max
2402 || defining->plus)
2403 && (defining->nparam_min <= mmac->nparam_max
2404 || mmac->plus))
2406 error(ERR_WARNING,
2407 "redefining multi-line macro `%s'",
2408 defining->name);
2409 break;
2411 mmac = mmac->next;
2414 * Handle default parameters.
2416 if (tline && tline->next)
2418 defining->dlist = tline->next;
2419 tline->next = NULL;
2420 count_mmac_params(defining->dlist, &defining->ndefs,
2421 &defining->defaults);
2423 else
2425 defining->dlist = NULL;
2426 defining->defaults = NULL;
2428 defining->expansion = NULL;
2429 free_tlist(origline);
2430 return DIRECTIVE_FOUND;
2432 case PP_ENDM:
2433 case PP_ENDMACRO:
2434 if (!defining)
2436 error(ERR_NONFATAL, "`%s': not defining a macro",
2437 tline->text);
2438 return DIRECTIVE_FOUND;
2440 k = hash(defining->name);
2441 defining->next = mmacros[k];
2442 mmacros[k] = defining;
2443 defining = NULL;
2444 free_tlist(origline);
2445 return DIRECTIVE_FOUND;
2447 case PP_ROTATE:
2448 if (tline->next && tline->next->type == TOK_WHITESPACE)
2449 tline = tline->next;
2450 if (tline->next == NULL)
2452 free_tlist(origline);
2453 error(ERR_NONFATAL, "`%%rotate' missing rotate count");
2454 return DIRECTIVE_FOUND;
2456 t = expand_smacro(tline->next);
2457 tline->next = NULL;
2458 free_tlist(origline);
2459 tline = t;
2460 tptr = &t;
2461 tokval.t_type = TOKEN_INVALID;
2462 evalresult =
2463 evaluate(ppscan, tptr, &tokval, NULL, pass, error, NULL);
2464 free_tlist(tline);
2465 if (!evalresult)
2466 return DIRECTIVE_FOUND;
2467 if (tokval.t_type)
2468 error(ERR_WARNING,
2469 "trailing garbage after expression ignored");
2470 if (!is_simple(evalresult))
2472 error(ERR_NONFATAL, "non-constant value given to `%%rotate'");
2473 return DIRECTIVE_FOUND;
2475 mmac = istk->mstk;
2476 while (mmac && !mmac->name) /* avoid mistaking %reps for macros */
2477 mmac = mmac->next_active;
2478 if (!mmac)
2480 error(ERR_NONFATAL,
2481 "`%%rotate' invoked outside a macro call");
2483 else if (mmac->nparam == 0)
2485 error(ERR_NONFATAL,
2486 "`%%rotate' invoked within macro without parameters");
2488 else
2490 mmac->rotate = mmac->rotate + reloc_value(evalresult);
2492 if (mmac->rotate < 0)
2493 mmac->rotate =
2494 mmac->nparam - (-mmac->rotate) % mmac->nparam;
2495 mmac->rotate %= mmac->nparam;
2497 return DIRECTIVE_FOUND;
2499 case PP_REP:
2500 nolist = FALSE;
2501 do {
2502 tline = tline->next;
2503 } while (tok_type_(tline, TOK_WHITESPACE));
2505 if (tok_type_(tline, TOK_ID) &&
2506 nasm_stricmp(tline->text, ".nolist") == 0)
2508 nolist = TRUE;
2509 do {
2510 tline = tline->next;
2511 } while (tok_type_(tline, TOK_WHITESPACE));
2514 if (tline)
2516 t = expand_smacro(tline);
2517 tptr = &t;
2518 tokval.t_type = TOKEN_INVALID;
2519 evalresult =
2520 evaluate(ppscan, tptr, &tokval, NULL, pass, error, NULL);
2521 if (!evalresult)
2523 free_tlist(origline);
2524 return DIRECTIVE_FOUND;
2526 if (tokval.t_type)
2527 error(ERR_WARNING,
2528 "trailing garbage after expression ignored");
2529 if (!is_simple(evalresult))
2531 error(ERR_NONFATAL, "non-constant value given to `%%rep'");
2532 return DIRECTIVE_FOUND;
2534 i = (int)reloc_value(evalresult) + 1;
2536 else
2538 error(ERR_NONFATAL, "`%%rep' expects a repeat count");
2539 i = 0;
2541 free_tlist(origline);
2543 tmp_defining = defining;
2544 defining = nasm_malloc(sizeof(MMacro));
2545 defining->name = NULL; /* flags this macro as a %rep block */
2546 defining->casesense = 0;
2547 defining->plus = FALSE;
2548 defining->nolist = nolist;
2549 defining->in_progress = i;
2550 defining->nparam_min = defining->nparam_max = 0;
2551 defining->defaults = NULL;
2552 defining->dlist = NULL;
2553 defining->expansion = NULL;
2554 defining->next_active = istk->mstk;
2555 defining->rep_nest = tmp_defining;
2556 return DIRECTIVE_FOUND;
2558 case PP_ENDREP:
2559 if (!defining || defining->name)
2561 error(ERR_NONFATAL, "`%%endrep': no matching `%%rep'");
2562 return DIRECTIVE_FOUND;
2566 * Now we have a "macro" defined - although it has no name
2567 * and we won't be entering it in the hash tables - we must
2568 * push a macro-end marker for it on to istk->expansion.
2569 * After that, it will take care of propagating itself (a
2570 * macro-end marker line for a macro which is really a %rep
2571 * block will cause the macro to be re-expanded, complete
2572 * with another macro-end marker to ensure the process
2573 * continues) until the whole expansion is forcibly removed
2574 * from istk->expansion by a %exitrep.
2576 l = nasm_malloc(sizeof(Line));
2577 l->next = istk->expansion;
2578 l->finishes = defining;
2579 l->first = NULL;
2580 istk->expansion = l;
2582 istk->mstk = defining;
2584 list->uplevel(defining->nolist ? LIST_MACRO_NOLIST : LIST_MACRO);
2585 tmp_defining = defining;
2586 defining = defining->rep_nest;
2587 free_tlist(origline);
2588 return DIRECTIVE_FOUND;
2590 case PP_EXITREP:
2592 * We must search along istk->expansion until we hit a
2593 * macro-end marker for a macro with no name. Then we set
2594 * its `in_progress' flag to 0.
2596 for (l = istk->expansion; l; l = l->next)
2597 if (l->finishes && !l->finishes->name)
2598 break;
2600 if (l)
2601 l->finishes->in_progress = 0;
2602 else
2603 error(ERR_NONFATAL, "`%%exitrep' not within `%%rep' block");
2604 free_tlist(origline);
2605 return DIRECTIVE_FOUND;
2607 case PP_XDEFINE:
2608 case PP_IXDEFINE:
2609 case PP_DEFINE:
2610 case PP_IDEFINE:
2611 tline = tline->next;
2612 skip_white_(tline);
2613 tline = expand_id(tline);
2614 if (!tline || (tline->type != TOK_ID &&
2615 (tline->type != TOK_PREPROC_ID ||
2616 tline->text[1] != '$')))
2618 error(ERR_NONFATAL,
2619 "`%%%s%sdefine' expects a macro identifier",
2620 ((i == PP_IDEFINE || i == PP_IXDEFINE) ? "i" : ""),
2621 ((i == PP_XDEFINE || i == PP_IXDEFINE) ? "x" : ""));
2622 free_tlist(origline);
2623 return DIRECTIVE_FOUND;
2626 ctx = get_ctx(tline->text, FALSE);
2627 if (!ctx)
2628 smhead = &smacros[hash(tline->text)];
2629 else
2630 smhead = &ctx->localmac;
2631 mname = tline->text;
2632 last = tline;
2633 param_start = tline = tline->next;
2634 nparam = 0;
2636 /* Expand the macro definition now for %xdefine and %ixdefine */
2637 if ((i == PP_XDEFINE) || (i == PP_IXDEFINE))
2638 tline = expand_smacro(tline);
2640 if (tok_is_(tline, "("))
2643 * This macro has parameters.
2646 tline = tline->next;
2647 while (1)
2649 skip_white_(tline);
2650 if (!tline)
2652 error(ERR_NONFATAL, "parameter identifier expected");
2653 free_tlist(origline);
2654 return DIRECTIVE_FOUND;
2656 if (tline->type != TOK_ID)
2658 error(ERR_NONFATAL,
2659 "`%s': parameter identifier expected",
2660 tline->text);
2661 free_tlist(origline);
2662 return DIRECTIVE_FOUND;
2664 tline->type = TOK_SMAC_PARAM + nparam++;
2665 tline = tline->next;
2666 skip_white_(tline);
2667 if (tok_is_(tline, ","))
2669 tline = tline->next;
2670 continue;
2672 if (!tok_is_(tline, ")"))
2674 error(ERR_NONFATAL,
2675 "`)' expected to terminate macro template");
2676 free_tlist(origline);
2677 return DIRECTIVE_FOUND;
2679 break;
2681 last = tline;
2682 tline = tline->next;
2684 if (tok_type_(tline, TOK_WHITESPACE))
2685 last = tline, tline = tline->next;
2686 macro_start = NULL;
2687 last->next = NULL;
2688 t = tline;
2689 while (t)
2691 if (t->type == TOK_ID)
2693 for (tt = param_start; tt; tt = tt->next)
2694 if (tt->type >= TOK_SMAC_PARAM &&
2695 !strcmp(tt->text, t->text))
2696 t->type = tt->type;
2698 tt = t->next;
2699 t->next = macro_start;
2700 macro_start = t;
2701 t = tt;
2704 * Good. We now have a macro name, a parameter count, and a
2705 * token list (in reverse order) for an expansion. We ought
2706 * to be OK just to create an SMacro, store it, and let
2707 * free_tlist have the rest of the line (which we have
2708 * carefully re-terminated after chopping off the expansion
2709 * from the end).
2711 if (smacro_defined(ctx, mname, nparam, &smac, i == PP_DEFINE))
2713 if (!smac)
2715 error(ERR_WARNING,
2716 "single-line macro `%s' defined both with and"
2717 " without parameters", mname);
2718 free_tlist(origline);
2719 free_tlist(macro_start);
2720 return DIRECTIVE_FOUND;
2722 else
2725 * We're redefining, so we have to take over an
2726 * existing SMacro structure. This means freeing
2727 * what was already in it.
2729 nasm_free(smac->name);
2730 free_tlist(smac->expansion);
2733 else
2735 smac = nasm_malloc(sizeof(SMacro));
2736 smac->next = *smhead;
2737 *smhead = smac;
2739 smac->name = nasm_strdup(mname);
2740 smac->casesense = ((i == PP_DEFINE) || (i == PP_XDEFINE));
2741 smac->nparam = nparam;
2742 smac->expansion = macro_start;
2743 smac->in_progress = FALSE;
2744 free_tlist(origline);
2745 return DIRECTIVE_FOUND;
2747 case PP_UNDEF:
2748 tline = tline->next;
2749 skip_white_(tline);
2750 tline = expand_id(tline);
2751 if (!tline || (tline->type != TOK_ID &&
2752 (tline->type != TOK_PREPROC_ID ||
2753 tline->text[1] != '$')))
2755 error(ERR_NONFATAL, "`%%undef' expects a macro identifier");
2756 free_tlist(origline);
2757 return DIRECTIVE_FOUND;
2759 if (tline->next)
2761 error(ERR_WARNING,
2762 "trailing garbage after macro name ignored");
2765 /* Find the context that symbol belongs to */
2766 ctx = get_ctx(tline->text, FALSE);
2767 if (!ctx)
2768 smhead = &smacros[hash(tline->text)];
2769 else
2770 smhead = &ctx->localmac;
2772 mname = tline->text;
2773 last = tline;
2774 last->next = NULL;
2777 * We now have a macro name... go hunt for it.
2779 while (smacro_defined(ctx, mname, -1, &smac, 1))
2781 /* Defined, so we need to find its predecessor and nuke it */
2782 SMacro **s;
2783 for (s = smhead; *s && *s != smac; s = &(*s)->next);
2784 if (*s)
2786 *s = smac->next;
2787 nasm_free(smac->name);
2788 free_tlist(smac->expansion);
2789 nasm_free(smac);
2792 free_tlist(origline);
2793 return DIRECTIVE_FOUND;
2795 case PP_STRLEN:
2796 tline = tline->next;
2797 skip_white_(tline);
2798 tline = expand_id(tline);
2799 if (!tline || (tline->type != TOK_ID &&
2800 (tline->type != TOK_PREPROC_ID ||
2801 tline->text[1] != '$')))
2803 error(ERR_NONFATAL,
2804 "`%%strlen' expects a macro identifier as first parameter");
2805 free_tlist(origline);
2806 return DIRECTIVE_FOUND;
2808 ctx = get_ctx(tline->text, FALSE);
2809 if (!ctx)
2810 smhead = &smacros[hash(tline->text)];
2811 else
2812 smhead = &ctx->localmac;
2813 mname = tline->text;
2814 last = tline;
2815 tline = expand_smacro(tline->next);
2816 last->next = NULL;
2818 t = tline;
2819 while (tok_type_(t, TOK_WHITESPACE))
2820 t = t->next;
2821 /* t should now point to the string */
2822 if (t->type != TOK_STRING)
2824 error(ERR_NONFATAL,
2825 "`%%strlen` requires string as second parameter");
2826 free_tlist(tline);
2827 free_tlist(origline);
2828 return DIRECTIVE_FOUND;
2831 macro_start = nasm_malloc(sizeof(*macro_start));
2832 macro_start->next = NULL;
2833 make_tok_num(macro_start, strlen(t->text) - 2);
2834 macro_start->mac = NULL;
2837 * We now have a macro name, an implicit parameter count of
2838 * zero, and a numeric token to use as an expansion. Create
2839 * and store an SMacro.
2841 if (smacro_defined(ctx, mname, 0, &smac, i == PP_STRLEN))
2843 if (!smac)
2844 error(ERR_WARNING,
2845 "single-line macro `%s' defined both with and"
2846 " without parameters", mname);
2847 else
2850 * We're redefining, so we have to take over an
2851 * existing SMacro structure. This means freeing
2852 * what was already in it.
2854 nasm_free(smac->name);
2855 free_tlist(smac->expansion);
2858 else
2860 smac = nasm_malloc(sizeof(SMacro));
2861 smac->next = *smhead;
2862 *smhead = smac;
2864 smac->name = nasm_strdup(mname);
2865 smac->casesense = (i == PP_STRLEN);
2866 smac->nparam = 0;
2867 smac->expansion = macro_start;
2868 smac->in_progress = FALSE;
2869 free_tlist(tline);
2870 free_tlist(origline);
2871 return DIRECTIVE_FOUND;
2873 case PP_SUBSTR:
2874 tline = tline->next;
2875 skip_white_(tline);
2876 tline = expand_id(tline);
2877 if (!tline || (tline->type != TOK_ID &&
2878 (tline->type != TOK_PREPROC_ID ||
2879 tline->text[1] != '$')))
2881 error(ERR_NONFATAL,
2882 "`%%substr' expects a macro identifier as first parameter");
2883 free_tlist(origline);
2884 return DIRECTIVE_FOUND;
2886 ctx = get_ctx(tline->text, FALSE);
2887 if (!ctx)
2888 smhead = &smacros[hash(tline->text)];
2889 else
2890 smhead = &ctx->localmac;
2891 mname = tline->text;
2892 last = tline;
2893 tline = expand_smacro(tline->next);
2894 last->next = NULL;
2896 t = tline->next;
2897 while (tok_type_(t, TOK_WHITESPACE))
2898 t = t->next;
2900 /* t should now point to the string */
2901 if (t->type != TOK_STRING)
2903 error(ERR_NONFATAL,
2904 "`%%substr` requires string as second parameter");
2905 free_tlist(tline);
2906 free_tlist(origline);
2907 return DIRECTIVE_FOUND;
2910 tt = t->next;
2911 tptr = &tt;
2912 tokval.t_type = TOKEN_INVALID;
2913 evalresult =
2914 evaluate(ppscan, tptr, &tokval, NULL, pass, error, NULL);
2915 if (!evalresult)
2917 free_tlist(tline);
2918 free_tlist(origline);
2919 return DIRECTIVE_FOUND;
2921 if (!is_simple(evalresult))
2923 error(ERR_NONFATAL, "non-constant value given to `%%substr`");
2924 free_tlist(tline);
2925 free_tlist(origline);
2926 return DIRECTIVE_FOUND;
2929 macro_start = nasm_malloc(sizeof(*macro_start));
2930 macro_start->next = NULL;
2931 macro_start->text = nasm_strdup("'''");
2932 if (evalresult->value > 0
2933 && evalresult->value < strlen(t->text) - 1)
2935 macro_start->text[1] = t->text[evalresult->value];
2937 else
2939 macro_start->text[2] = '\0';
2941 macro_start->type = TOK_STRING;
2942 macro_start->mac = NULL;
2945 * We now have a macro name, an implicit parameter count of
2946 * zero, and a numeric token to use as an expansion. Create
2947 * and store an SMacro.
2949 if (smacro_defined(ctx, mname, 0, &smac, i == PP_SUBSTR))
2951 if (!smac)
2952 error(ERR_WARNING,
2953 "single-line macro `%s' defined both with and"
2954 " without parameters", mname);
2955 else
2958 * We're redefining, so we have to take over an
2959 * existing SMacro structure. This means freeing
2960 * what was already in it.
2962 nasm_free(smac->name);
2963 free_tlist(smac->expansion);
2966 else
2968 smac = nasm_malloc(sizeof(SMacro));
2969 smac->next = *smhead;
2970 *smhead = smac;
2972 smac->name = nasm_strdup(mname);
2973 smac->casesense = (i == PP_SUBSTR);
2974 smac->nparam = 0;
2975 smac->expansion = macro_start;
2976 smac->in_progress = FALSE;
2977 free_tlist(tline);
2978 free_tlist(origline);
2979 return DIRECTIVE_FOUND;
2982 case PP_ASSIGN:
2983 case PP_IASSIGN:
2984 tline = tline->next;
2985 skip_white_(tline);
2986 tline = expand_id(tline);
2987 if (!tline || (tline->type != TOK_ID &&
2988 (tline->type != TOK_PREPROC_ID ||
2989 tline->text[1] != '$')))
2991 error(ERR_NONFATAL,
2992 "`%%%sassign' expects a macro identifier",
2993 (i == PP_IASSIGN ? "i" : ""));
2994 free_tlist(origline);
2995 return DIRECTIVE_FOUND;
2997 ctx = get_ctx(tline->text, FALSE);
2998 if (!ctx)
2999 smhead = &smacros[hash(tline->text)];
3000 else
3001 smhead = &ctx->localmac;
3002 mname = tline->text;
3003 last = tline;
3004 tline = expand_smacro(tline->next);
3005 last->next = NULL;
3007 t = tline;
3008 tptr = &t;
3009 tokval.t_type = TOKEN_INVALID;
3010 evalresult =
3011 evaluate(ppscan, tptr, &tokval, NULL, pass, error, NULL);
3012 free_tlist(tline);
3013 if (!evalresult)
3015 free_tlist(origline);
3016 return DIRECTIVE_FOUND;
3019 if (tokval.t_type)
3020 error(ERR_WARNING,
3021 "trailing garbage after expression ignored");
3023 if (!is_simple(evalresult))
3025 error(ERR_NONFATAL,
3026 "non-constant value given to `%%%sassign'",
3027 (i == PP_IASSIGN ? "i" : ""));
3028 free_tlist(origline);
3029 return DIRECTIVE_FOUND;
3032 macro_start = nasm_malloc(sizeof(*macro_start));
3033 macro_start->next = NULL;
3034 make_tok_num(macro_start, reloc_value(evalresult));
3035 macro_start->mac = NULL;
3038 * We now have a macro name, an implicit parameter count of
3039 * zero, and a numeric token to use as an expansion. Create
3040 * and store an SMacro.
3042 if (smacro_defined(ctx, mname, 0, &smac, i == PP_ASSIGN))
3044 if (!smac)
3045 error(ERR_WARNING,
3046 "single-line macro `%s' defined both with and"
3047 " without parameters", mname);
3048 else
3051 * We're redefining, so we have to take over an
3052 * existing SMacro structure. This means freeing
3053 * what was already in it.
3055 nasm_free(smac->name);
3056 free_tlist(smac->expansion);
3059 else
3061 smac = nasm_malloc(sizeof(SMacro));
3062 smac->next = *smhead;
3063 *smhead = smac;
3065 smac->name = nasm_strdup(mname);
3066 smac->casesense = (i == PP_ASSIGN);
3067 smac->nparam = 0;
3068 smac->expansion = macro_start;
3069 smac->in_progress = FALSE;
3070 free_tlist(origline);
3071 return DIRECTIVE_FOUND;
3073 case PP_LINE:
3075 * Syntax is `%line nnn[+mmm] [filename]'
3077 tline = tline->next;
3078 skip_white_(tline);
3079 if (!tok_type_(tline, TOK_NUMBER))
3081 error(ERR_NONFATAL, "`%%line' expects line number");
3082 free_tlist(origline);
3083 return DIRECTIVE_FOUND;
3085 k = readnum(tline->text, &j);
3086 m = 1;
3087 tline = tline->next;
3088 if (tok_is_(tline, "+"))
3090 tline = tline->next;
3091 if (!tok_type_(tline, TOK_NUMBER))
3093 error(ERR_NONFATAL, "`%%line' expects line increment");
3094 free_tlist(origline);
3095 return DIRECTIVE_FOUND;
3097 m = readnum(tline->text, &j);
3098 tline = tline->next;
3100 skip_white_(tline);
3101 src_set_linnum(k);
3102 istk->lineinc = m;
3103 if (tline)
3105 nasm_free(src_set_fname(detoken(tline, FALSE)));
3107 free_tlist(origline);
3108 return DIRECTIVE_FOUND;
3110 default:
3111 error(ERR_FATAL,
3112 "preprocessor directive `%s' not yet implemented",
3113 directives[i]);
3114 break;
3116 return DIRECTIVE_FOUND;
3120 * Ensure that a macro parameter contains a condition code and
3121 * nothing else. Return the condition code index if so, or -1
3122 * otherwise.
3124 static int
3125 find_cc(Token * t)
3127 Token *tt;
3128 int i, j, k, m;
3130 skip_white_(t);
3131 if (t->type != TOK_ID)
3132 return -1;
3133 tt = t->next;
3134 skip_white_(tt);
3135 if (tt && (tt->type != TOK_OTHER || strcmp(tt->text, ",")))
3136 return -1;
3138 i = -1;
3139 j = elements(conditions);
3140 while (j - i > 1)
3142 k = (j + i) / 2;
3143 m = nasm_stricmp(t->text, conditions[k]);
3144 if (m == 0)
3146 i = k;
3147 j = -2;
3148 break;
3150 else if (m < 0)
3152 j = k;
3154 else
3155 i = k;
3157 if (j != -2)
3158 return -1;
3159 return i;
3163 * Expand MMacro-local things: parameter references (%0, %n, %+n,
3164 * %-n) and MMacro-local identifiers (%%foo).
3166 static Token *
3167 expand_mmac_params(Token * tline)
3169 Token *t, *tt, **tail, *thead;
3171 tail = &thead;
3172 thead = NULL;
3174 while (tline)
3176 if (tline->type == TOK_PREPROC_ID &&
3177 (((tline->text[1] == '+' || tline->text[1] == '-')
3178 && tline->text[2]) || tline->text[1] == '%'
3179 || (tline->text[1] >= '0' && tline->text[1] <= '9')))
3181 char *text = NULL;
3182 int type = 0, cc; /* type = 0 to placate optimisers */
3183 char tmpbuf[30];
3184 int n, i;
3185 MMacro *mac;
3187 t = tline;
3188 tline = tline->next;
3190 mac = istk->mstk;
3191 while (mac && !mac->name) /* avoid mistaking %reps for macros */
3192 mac = mac->next_active;
3193 if (!mac)
3194 error(ERR_NONFATAL, "`%s': not in a macro call", t->text);
3195 else
3196 switch (t->text[1])
3199 * We have to make a substitution of one of the
3200 * forms %1, %-1, %+1, %%foo, %0.
3202 case '0':
3203 type = TOK_NUMBER;
3204 snprintf(tmpbuf, sizeof(tmpbuf), "%d", mac->nparam);
3205 text = nasm_strdup(tmpbuf);
3206 break;
3207 case '%':
3208 type = TOK_ID;
3209 snprintf(tmpbuf, sizeof(tmpbuf), "..@%lu.", mac->unique);
3210 text = nasm_strcat(tmpbuf, t->text + 2);
3211 break;
3212 case '-':
3213 n = atoi(t->text + 2) - 1;
3214 if (n >= mac->nparam)
3215 tt = NULL;
3216 else
3218 if (mac->nparam > 1)
3219 n = (n + mac->rotate) % mac->nparam;
3220 tt = mac->params[n];
3222 cc = find_cc(tt);
3223 if (cc == -1)
3225 error(ERR_NONFATAL,
3226 "macro parameter %d is not a condition code",
3227 n + 1);
3228 text = NULL;
3230 else
3232 type = TOK_ID;
3233 if (inverse_ccs[cc] == -1)
3235 error(ERR_NONFATAL,
3236 "condition code `%s' is not invertible",
3237 conditions[cc]);
3238 text = NULL;
3240 else
3241 text =
3242 nasm_strdup(conditions[inverse_ccs
3243 [cc]]);
3245 break;
3246 case '+':
3247 n = atoi(t->text + 2) - 1;
3248 if (n >= mac->nparam)
3249 tt = NULL;
3250 else
3252 if (mac->nparam > 1)
3253 n = (n + mac->rotate) % mac->nparam;
3254 tt = mac->params[n];
3256 cc = find_cc(tt);
3257 if (cc == -1)
3259 error(ERR_NONFATAL,
3260 "macro parameter %d is not a condition code",
3261 n + 1);
3262 text = NULL;
3264 else
3266 type = TOK_ID;
3267 text = nasm_strdup(conditions[cc]);
3269 break;
3270 default:
3271 n = atoi(t->text + 1) - 1;
3272 if (n >= mac->nparam)
3273 tt = NULL;
3274 else
3276 if (mac->nparam > 1)
3277 n = (n + mac->rotate) % mac->nparam;
3278 tt = mac->params[n];
3280 if (tt)
3282 for (i = 0; i < mac->paramlen[n]; i++)
3284 *tail =
3285 new_Token(NULL, tt->type, tt->text,
3287 tail = &(*tail)->next;
3288 tt = tt->next;
3291 text = NULL; /* we've done it here */
3292 break;
3294 if (!text)
3296 delete_Token(t);
3298 else
3300 *tail = t;
3301 tail = &t->next;
3302 t->type = type;
3303 nasm_free(t->text);
3304 t->text = text;
3305 t->mac = NULL;
3307 continue;
3309 else
3311 t = *tail = tline;
3312 tline = tline->next;
3313 t->mac = NULL;
3314 tail = &t->next;
3317 *tail = NULL;
3318 t = thead;
3319 for (; t && (tt = t->next) != NULL; t = t->next)
3320 switch (t->type)
3322 case TOK_WHITESPACE:
3323 if (tt->type == TOK_WHITESPACE)
3325 t->next = delete_Token(tt);
3327 break;
3328 case TOK_ID:
3329 if (tt->type == TOK_ID || tt->type == TOK_NUMBER)
3331 char *tmp = nasm_strcat(t->text, tt->text);
3332 nasm_free(t->text);
3333 t->text = tmp;
3334 t->next = delete_Token(tt);
3336 break;
3337 case TOK_NUMBER:
3338 if (tt->type == TOK_NUMBER)
3340 char *tmp = nasm_strcat(t->text, tt->text);
3341 nasm_free(t->text);
3342 t->text = tmp;
3343 t->next = delete_Token(tt);
3345 break;
3348 return thead;
3352 * Expand all single-line macro calls made in the given line.
3353 * Return the expanded version of the line. The original is deemed
3354 * to be destroyed in the process. (In reality we'll just move
3355 * Tokens from input to output a lot of the time, rather than
3356 * actually bothering to destroy and replicate.)
3358 static Token *
3359 expand_smacro(Token * tline)
3361 Token *t, *tt, *mstart, **tail, *thead;
3362 SMacro *head = NULL, *m;
3363 Token **params;
3364 int *paramsize;
3365 int nparam, sparam, brackets, rescan;
3366 Token *org_tline = tline;
3367 Context *ctx;
3368 char *mname;
3371 * Trick: we should avoid changing the start token pointer since it can
3372 * be contained in "next" field of other token. Because of this
3373 * we allocate a copy of first token and work with it; at the end of
3374 * routine we copy it back
3376 if (org_tline)
3378 tline =
3379 new_Token(org_tline->next, org_tline->type, org_tline->text,
3381 tline->mac = org_tline->mac;
3382 nasm_free(org_tline->text);
3383 org_tline->text = NULL;
3386 again:
3387 tail = &thead;
3388 thead = NULL;
3390 while (tline)
3391 { /* main token loop */
3392 if ((mname = tline->text))
3394 /* if this token is a local macro, look in local context */
3395 if (tline->type == TOK_ID || tline->type == TOK_PREPROC_ID)
3396 ctx = get_ctx(mname, TRUE);
3397 else
3398 ctx = NULL;
3399 if (!ctx)
3400 head = smacros[hash(mname)];
3401 else
3402 head = ctx->localmac;
3404 * We've hit an identifier. As in is_mmacro below, we first
3405 * check whether the identifier is a single-line macro at
3406 * all, then think about checking for parameters if
3407 * necessary.
3409 for (m = head; m; m = m->next)
3410 if (!mstrcmp(m->name, mname, m->casesense))
3411 break;
3412 if (m)
3414 mstart = tline;
3415 params = NULL;
3416 paramsize = NULL;
3417 if (m->nparam == 0)
3420 * Simple case: the macro is parameterless. Discard the
3421 * one token that the macro call took, and push the
3422 * expansion back on the to-do stack.
3424 if (!m->expansion)
3426 if (!strcmp("__FILE__", m->name))
3428 long num = 0;
3429 src_get(&num, &(tline->text));
3430 nasm_quote(&(tline->text));
3431 tline->type = TOK_STRING;
3432 continue;
3434 if (!strcmp("__LINE__", m->name))
3436 nasm_free(tline->text);
3437 make_tok_num(tline, src_get_linnum());
3438 continue;
3440 tline = delete_Token(tline);
3441 continue;
3444 else
3447 * Complicated case: at least one macro with this name
3448 * exists and takes parameters. We must find the
3449 * parameters in the call, count them, find the SMacro
3450 * that corresponds to that form of the macro call, and
3451 * substitute for the parameters when we expand. What a
3452 * pain.
3454 /*tline = tline->next;
3455 skip_white_(tline);*/
3456 do {
3457 t = tline->next;
3458 while (tok_type_(t, TOK_SMAC_END))
3460 t->mac->in_progress = FALSE;
3461 t->text = NULL;
3462 t = tline->next = delete_Token(t);
3464 tline = t;
3465 } while (tok_type_(tline, TOK_WHITESPACE));
3466 if (!tok_is_(tline, "("))
3469 * This macro wasn't called with parameters: ignore
3470 * the call. (Behaviour borrowed from gnu cpp.)
3472 tline = mstart;
3473 m = NULL;
3475 else
3477 int paren = 0;
3478 int white = 0;
3479 brackets = 0;
3480 nparam = 0;
3481 sparam = PARAM_DELTA;
3482 params = nasm_malloc(sparam * sizeof(Token *));
3483 params[0] = tline->next;
3484 paramsize = nasm_malloc(sparam * sizeof(int));
3485 paramsize[0] = 0;
3486 while (TRUE)
3487 { /* parameter loop */
3489 * For some unusual expansions
3490 * which concatenates function call
3492 t = tline->next;
3493 while (tok_type_(t, TOK_SMAC_END))
3495 t->mac->in_progress = FALSE;
3496 t->text = NULL;
3497 t = tline->next = delete_Token(t);
3499 tline = t;
3501 if (!tline)
3503 error(ERR_NONFATAL,
3504 "macro call expects terminating `)'");
3505 break;
3507 if (tline->type == TOK_WHITESPACE
3508 && brackets <= 0)
3510 if (paramsize[nparam])
3511 white++;
3512 else
3513 params[nparam] = tline->next;
3514 continue; /* parameter loop */
3516 if (tline->type == TOK_OTHER
3517 && tline->text[1] == 0)
3519 char ch = tline->text[0];
3520 if (ch == ',' && !paren && brackets <= 0)
3522 if (++nparam >= sparam)
3524 sparam += PARAM_DELTA;
3525 params = nasm_realloc(params,
3526 sparam * sizeof(Token *));
3527 paramsize = nasm_realloc(paramsize,
3528 sparam * sizeof(int));
3530 params[nparam] = tline->next;
3531 paramsize[nparam] = 0;
3532 white = 0;
3533 continue; /* parameter loop */
3535 if (ch == '{' &&
3536 (brackets > 0 || (brackets == 0 &&
3537 !paramsize[nparam])))
3539 if (!(brackets++))
3541 params[nparam] = tline->next;
3542 continue; /* parameter loop */
3545 if (ch == '}' && brackets > 0)
3546 if (--brackets == 0)
3548 brackets = -1;
3549 continue; /* parameter loop */
3551 if (ch == '(' && !brackets)
3552 paren++;
3553 if (ch == ')' && brackets <= 0)
3554 if (--paren < 0)
3555 break;
3557 if (brackets < 0)
3559 brackets = 0;
3560 error(ERR_NONFATAL, "braces do not "
3561 "enclose all of macro parameter");
3563 paramsize[nparam] += white + 1;
3564 white = 0;
3565 } /* parameter loop */
3566 nparam++;
3567 while (m && (m->nparam != nparam ||
3568 mstrcmp(m->name, mname,
3569 m->casesense)))
3570 m = m->next;
3571 if (!m)
3572 error(ERR_WARNING | ERR_WARN_MNP,
3573 "macro `%s' exists, "
3574 "but not taking %d parameters",
3575 mstart->text, nparam);
3578 if (m && m->in_progress)
3579 m = NULL;
3580 if (!m) /* in progess or didn't find '(' or wrong nparam */
3583 * Design question: should we handle !tline, which
3584 * indicates missing ')' here, or expand those
3585 * macros anyway, which requires the (t) test a few
3586 * lines down?
3588 nasm_free(params);
3589 nasm_free(paramsize);
3590 tline = mstart;
3592 else
3595 * Expand the macro: we are placed on the last token of the
3596 * call, so that we can easily split the call from the
3597 * following tokens. We also start by pushing an SMAC_END
3598 * token for the cycle removal.
3600 t = tline;
3601 if (t)
3603 tline = t->next;
3604 t->next = NULL;
3606 tt = new_Token(tline, TOK_SMAC_END, NULL, 0);
3607 tt->mac = m;
3608 m->in_progress = TRUE;
3609 tline = tt;
3610 for (t = m->expansion; t; t = t->next)
3612 if (t->type >= TOK_SMAC_PARAM)
3614 Token *pcopy = tline, **ptail = &pcopy;
3615 Token *ttt, *pt;
3616 int i;
3618 ttt = params[t->type - TOK_SMAC_PARAM];
3619 for (i = paramsize[t->type - TOK_SMAC_PARAM];
3620 --i >= 0;)
3622 pt = *ptail =
3623 new_Token(tline, ttt->type, ttt->text,
3625 ptail = &pt->next;
3626 ttt = ttt->next;
3628 tline = pcopy;
3630 else
3632 tt = new_Token(tline, t->type, t->text, 0);
3633 tline = tt;
3638 * Having done that, get rid of the macro call, and clean
3639 * up the parameters.
3641 nasm_free(params);
3642 nasm_free(paramsize);
3643 free_tlist(mstart);
3644 continue; /* main token loop */
3649 if (tline->type == TOK_SMAC_END)
3651 tline->mac->in_progress = FALSE;
3652 tline = delete_Token(tline);
3654 else
3656 t = *tail = tline;
3657 tline = tline->next;
3658 t->mac = NULL;
3659 t->next = NULL;
3660 tail = &t->next;
3665 * Now scan the entire line and look for successive TOK_IDs that resulted
3666 * after expansion (they can't be produced by tokenise()). The successive
3667 * TOK_IDs should be concatenated.
3668 * Also we look for %+ tokens and concatenate the tokens before and after
3669 * them (without white spaces in between).
3671 t = thead;
3672 rescan = 0;
3673 while (t)
3675 while (t && t->type != TOK_ID && t->type != TOK_PREPROC_ID)
3676 t = t->next;
3677 if (!t || !t->next)
3678 break;
3679 if (t->next->type == TOK_ID ||
3680 t->next->type == TOK_PREPROC_ID ||
3681 t->next->type == TOK_NUMBER)
3683 char *p = nasm_strcat(t->text, t->next->text);
3684 nasm_free(t->text);
3685 t->next = delete_Token(t->next);
3686 t->text = p;
3687 rescan = 1;
3689 else if (t->next->type == TOK_WHITESPACE && t->next->next &&
3690 t->next->next->type == TOK_PREPROC_ID &&
3691 strcmp(t->next->next->text, "%+") == 0)
3693 /* free the next whitespace, the %+ token and next whitespace */
3694 int i;
3695 for (i = 1; i <= 3; i++)
3697 if (!t->next || (i != 2 && t->next->type != TOK_WHITESPACE))
3698 break;
3699 t->next = delete_Token(t->next);
3700 } /* endfor */
3702 else
3703 t = t->next;
3705 /* If we concatenaded something, re-scan the line for macros */
3706 if (rescan)
3708 tline = thead;
3709 goto again;
3712 if (org_tline)
3714 if (thead)
3716 *org_tline = *thead;
3717 /* since we just gave text to org_line, don't free it */
3718 thead->text = NULL;
3719 delete_Token(thead);
3721 else
3723 /* the expression expanded to empty line;
3724 we can't return NULL for some reasons
3725 we just set the line to a single WHITESPACE token. */
3726 memset(org_tline, 0, sizeof(*org_tline));
3727 org_tline->text = NULL;
3728 org_tline->type = TOK_WHITESPACE;
3730 thead = org_tline;
3733 return thead;
3737 * Similar to expand_smacro but used exclusively with macro identifiers
3738 * right before they are fetched in. The reason is that there can be
3739 * identifiers consisting of several subparts. We consider that if there
3740 * are more than one element forming the name, user wants a expansion,
3741 * otherwise it will be left as-is. Example:
3743 * %define %$abc cde
3745 * the identifier %$abc will be left as-is so that the handler for %define
3746 * will suck it and define the corresponding value. Other case:
3748 * %define _%$abc cde
3750 * In this case user wants name to be expanded *before* %define starts
3751 * working, so we'll expand %$abc into something (if it has a value;
3752 * otherwise it will be left as-is) then concatenate all successive
3753 * PP_IDs into one.
3755 static Token *
3756 expand_id(Token * tline)
3758 Token *cur, *oldnext = NULL;
3760 if (!tline || !tline->next)
3761 return tline;
3763 cur = tline;
3764 while (cur->next &&
3765 (cur->next->type == TOK_ID ||
3766 cur->next->type == TOK_PREPROC_ID || cur->next->type == TOK_NUMBER))
3767 cur = cur->next;
3769 /* If identifier consists of just one token, don't expand */
3770 if (cur == tline)
3771 return tline;
3773 if (cur)
3775 oldnext = cur->next; /* Detach the tail past identifier */
3776 cur->next = NULL; /* so that expand_smacro stops here */
3779 tline = expand_smacro(tline);
3781 if (cur)
3783 /* expand_smacro possibly changhed tline; re-scan for EOL */
3784 cur = tline;
3785 while (cur && cur->next)
3786 cur = cur->next;
3787 if (cur)
3788 cur->next = oldnext;
3791 return tline;
3795 * Determine whether the given line constitutes a multi-line macro
3796 * call, and return the MMacro structure called if so. Doesn't have
3797 * to check for an initial label - that's taken care of in
3798 * expand_mmacro - but must check numbers of parameters. Guaranteed
3799 * to be called with tline->type == TOK_ID, so the putative macro
3800 * name is easy to find.
3802 static MMacro *
3803 is_mmacro(Token * tline, Token *** params_array)
3805 MMacro *head, *m;
3806 Token **params;
3807 int nparam;
3809 head = mmacros[hash(tline->text)];
3812 * Efficiency: first we see if any macro exists with the given
3813 * name. If not, we can return NULL immediately. _Then_ we
3814 * count the parameters, and then we look further along the
3815 * list if necessary to find the proper MMacro.
3817 for (m = head; m; m = m->next)
3818 if (!mstrcmp(m->name, tline->text, m->casesense))
3819 break;
3820 if (!m)
3821 return NULL;
3824 * OK, we have a potential macro. Count and demarcate the
3825 * parameters.
3827 count_mmac_params(tline->next, &nparam, &params);
3830 * So we know how many parameters we've got. Find the MMacro
3831 * structure that handles this number.
3833 while (m)
3835 if (m->nparam_min <= nparam && (m->plus || nparam <= m->nparam_max))
3838 * This one is right. Just check if cycle removal
3839 * prohibits us using it before we actually celebrate...
3841 if (m->in_progress)
3843 #if 0
3844 error(ERR_NONFATAL,
3845 "self-reference in multi-line macro `%s'", m->name);
3846 #endif
3847 nasm_free(params);
3848 return NULL;
3851 * It's right, and we can use it. Add its default
3852 * parameters to the end of our list if necessary.
3854 if (m->defaults && nparam < m->nparam_min + m->ndefs)
3856 params =
3857 nasm_realloc(params,
3858 ((m->nparam_min + m->ndefs + 1) * sizeof(*params)));
3859 while (nparam < m->nparam_min + m->ndefs)
3861 params[nparam] = m->defaults[nparam - m->nparam_min];
3862 nparam++;
3866 * If we've gone over the maximum parameter count (and
3867 * we're in Plus mode), ignore parameters beyond
3868 * nparam_max.
3870 if (m->plus && nparam > m->nparam_max)
3871 nparam = m->nparam_max;
3873 * Then terminate the parameter list, and leave.
3875 if (!params)
3876 { /* need this special case */
3877 params = nasm_malloc(sizeof(*params));
3878 nparam = 0;
3880 params[nparam] = NULL;
3881 *params_array = params;
3882 return m;
3885 * This one wasn't right: look for the next one with the
3886 * same name.
3888 for (m = m->next; m; m = m->next)
3889 if (!mstrcmp(m->name, tline->text, m->casesense))
3890 break;
3894 * After all that, we didn't find one with the right number of
3895 * parameters. Issue a warning, and fail to expand the macro.
3897 error(ERR_WARNING | ERR_WARN_MNP,
3898 "macro `%s' exists, but not taking %d parameters",
3899 tline->text, nparam);
3900 nasm_free(params);
3901 return NULL;
3905 * Expand the multi-line macro call made by the given line, if
3906 * there is one to be expanded. If there is, push the expansion on
3907 * istk->expansion and return 1. Otherwise return 0.
3909 static int
3910 expand_mmacro(Token * tline)
3912 Token *startline = tline;
3913 Token *label = NULL;
3914 int dont_prepend = 0;
3915 Token **params, *t, *tt;
3916 MMacro *m;
3917 Line *l, *ll;
3918 int i, nparam, *paramlen;
3920 t = tline;
3921 skip_white_(t);
3922 /* if (!tok_type_(t, TOK_ID)) Lino 02/25/02 */
3923 if (!tok_type_(t, TOK_ID) && !tok_type_(t, TOK_PREPROC_ID))
3924 return 0;
3925 m = is_mmacro(t, &params);
3926 if (!m)
3928 Token *last;
3930 * We have an id which isn't a macro call. We'll assume
3931 * it might be a label; we'll also check to see if a
3932 * colon follows it. Then, if there's another id after
3933 * that lot, we'll check it again for macro-hood.
3935 label = last = t;
3936 t = t->next;
3937 if (tok_type_(t, TOK_WHITESPACE))
3938 last = t, t = t->next;
3939 if (tok_is_(t, ":"))
3941 dont_prepend = 1;
3942 last = t, t = t->next;
3943 if (tok_type_(t, TOK_WHITESPACE))
3944 last = t, t = t->next;
3946 if (!tok_type_(t, TOK_ID) || (m = is_mmacro(t, &params)) == NULL)
3947 return 0;
3948 last->next = NULL;
3949 tline = t;
3953 * Fix up the parameters: this involves stripping leading and
3954 * trailing whitespace, then stripping braces if they are
3955 * present.
3957 for (nparam = 0; params[nparam]; nparam++)
3959 paramlen = nparam ? nasm_malloc(nparam * sizeof(*paramlen)) : NULL;
3961 for (i = 0; params[i]; i++)
3963 int brace = FALSE;
3964 int comma = (!m->plus || i < nparam - 1);
3966 t = params[i];
3967 skip_white_(t);
3968 if (tok_is_(t, "{"))
3969 t = t->next, brace = TRUE, comma = FALSE;
3970 params[i] = t;
3971 paramlen[i] = 0;
3972 while (t)
3974 if (comma && t->type == TOK_OTHER && !strcmp(t->text, ","))
3975 break; /* ... because we have hit a comma */
3976 if (comma && t->type == TOK_WHITESPACE && tok_is_(t->next, ","))
3977 break; /* ... or a space then a comma */
3978 if (brace && t->type == TOK_OTHER && !strcmp(t->text, "}"))
3979 break; /* ... or a brace */
3980 t = t->next;
3981 paramlen[i]++;
3986 * OK, we have a MMacro structure together with a set of
3987 * parameters. We must now go through the expansion and push
3988 * copies of each Line on to istk->expansion. Substitution of
3989 * parameter tokens and macro-local tokens doesn't get done
3990 * until the single-line macro substitution process; this is
3991 * because delaying them allows us to change the semantics
3992 * later through %rotate.
3994 * First, push an end marker on to istk->expansion, mark this
3995 * macro as in progress, and set up its invocation-specific
3996 * variables.
3998 ll = nasm_malloc(sizeof(Line));
3999 ll->next = istk->expansion;
4000 ll->finishes = m;
4001 ll->first = NULL;
4002 istk->expansion = ll;
4004 m->in_progress = TRUE;
4005 m->params = params;
4006 m->iline = tline;
4007 m->nparam = nparam;
4008 m->rotate = 0;
4009 m->paramlen = paramlen;
4010 m->unique = unique++;
4011 m->lineno = 0;
4013 m->next_active = istk->mstk;
4014 istk->mstk = m;
4016 for (l = m->expansion; l; l = l->next)
4018 Token **tail;
4020 ll = nasm_malloc(sizeof(Line));
4021 ll->finishes = NULL;
4022 ll->next = istk->expansion;
4023 istk->expansion = ll;
4024 tail = &ll->first;
4026 for (t = l->first; t; t = t->next)
4028 Token *x = t;
4029 if (t->type == TOK_PREPROC_ID &&
4030 t->text[1] == '0' && t->text[2] == '0')
4032 dont_prepend = -1;
4033 x = label;
4034 if (!x)
4035 continue;
4037 tt = *tail = new_Token(NULL, x->type, x->text, 0);
4038 tail = &tt->next;
4040 *tail = NULL;
4044 * If we had a label, push it on as the first line of
4045 * the macro expansion.
4047 if (label)
4049 if (dont_prepend < 0)
4050 free_tlist(startline);
4051 else
4053 ll = nasm_malloc(sizeof(Line));
4054 ll->finishes = NULL;
4055 ll->next = istk->expansion;
4056 istk->expansion = ll;
4057 ll->first = startline;
4058 if (!dont_prepend)
4060 while (label->next)
4061 label = label->next;
4062 label->next = tt = new_Token(NULL, TOK_OTHER, ":", 0);
4067 list->uplevel(m->nolist ? LIST_MACRO_NOLIST : LIST_MACRO);
4069 return 1;
4073 * Since preprocessor always operate only on the line that didn't
4074 * arrived yet, we should always use ERR_OFFBY1. Also since user
4075 * won't want to see same error twice (preprocessing is done once
4076 * per pass) we will want to show errors only during pass one.
4078 static void
4079 error(int severity, const char *fmt, ...)
4081 va_list arg;
4082 char buff[1024];
4084 /* If we're in a dead branch of IF or something like it, ignore the error */
4085 if (istk && istk->conds && !emitting(istk->conds->state))
4086 return;
4088 va_start(arg, fmt);
4089 vsnprintf(buff, sizeof(buff), fmt, arg);
4090 va_end(arg);
4092 if (istk && istk->mstk && istk->mstk->name)
4093 _error(severity | ERR_PASS1, "(%s:%d) %s", istk->mstk->name,
4094 istk->mstk->lineno, buff);
4095 else
4096 _error(severity | ERR_PASS1, "%s", buff);
4099 static void
4100 pp_reset(char *file, int apass, efunc errfunc, evalfunc eval,
4101 ListGen * listgen)
4103 int h;
4105 _error = errfunc;
4106 cstk = NULL;
4107 istk = nasm_malloc(sizeof(Include));
4108 istk->next = NULL;
4109 istk->conds = NULL;
4110 istk->expansion = NULL;
4111 istk->mstk = NULL;
4112 istk->fp = fopen(file, "r");
4113 istk->fname = NULL;
4114 src_set_fname(nasm_strdup(file));
4115 src_set_linnum(0);
4116 istk->lineinc = 1;
4117 if (!istk->fp)
4118 error(ERR_FATAL | ERR_NOFILE, "unable to open input file `%s'", file);
4119 defining = NULL;
4120 for (h = 0; h < NHASH; h++)
4122 mmacros[h] = NULL;
4123 smacros[h] = NULL;
4125 unique = 0;
4126 if (tasm_compatible_mode) {
4127 stdmacpos = stdmac;
4128 } else {
4129 stdmacpos = &stdmac[TASM_MACRO_COUNT];
4131 any_extrastdmac = (extrastdmac != NULL);
4132 list = listgen;
4133 evaluate = eval;
4134 pass = apass;
4137 static char *
4138 pp_getline(void)
4140 char *line;
4141 Token *tline;
4143 while (1)
4146 * Fetch a tokenised line, either from the macro-expansion
4147 * buffer or from the input file.
4149 tline = NULL;
4150 while (istk->expansion && istk->expansion->finishes)
4152 Line *l = istk->expansion;
4153 if (!l->finishes->name && l->finishes->in_progress > 1)
4155 Line *ll;
4158 * This is a macro-end marker for a macro with no
4159 * name, which means it's not really a macro at all
4160 * but a %rep block, and the `in_progress' field is
4161 * more than 1, meaning that we still need to
4162 * repeat. (1 means the natural last repetition; 0
4163 * means termination by %exitrep.) We have
4164 * therefore expanded up to the %endrep, and must
4165 * push the whole block on to the expansion buffer
4166 * again. We don't bother to remove the macro-end
4167 * marker: we'd only have to generate another one
4168 * if we did.
4170 l->finishes->in_progress--;
4171 for (l = l->finishes->expansion; l; l = l->next)
4173 Token *t, *tt, **tail;
4175 ll = nasm_malloc(sizeof(Line));
4176 ll->next = istk->expansion;
4177 ll->finishes = NULL;
4178 ll->first = NULL;
4179 tail = &ll->first;
4181 for (t = l->first; t; t = t->next)
4183 if (t->text || t->type == TOK_WHITESPACE)
4185 tt = *tail = new_Token(NULL, t->type, t->text, 0);
4186 tail = &tt->next;
4190 istk->expansion = ll;
4193 else
4196 * Check whether a `%rep' was started and not ended
4197 * within this macro expansion. This can happen and
4198 * should be detected. It's a fatal error because
4199 * I'm too confused to work out how to recover
4200 * sensibly from it.
4202 if (defining)
4204 if (defining->name)
4205 error(ERR_PANIC, "defining with name in expansion");
4206 else if (istk->mstk->name)
4207 error(ERR_FATAL, "`%%rep' without `%%endrep' within"
4208 " expansion of macro `%s'", istk->mstk->name);
4212 * FIXME: investigate the relationship at this point between
4213 * istk->mstk and l->finishes
4216 MMacro *m = istk->mstk;
4217 istk->mstk = m->next_active;
4218 if (m->name)
4221 * This was a real macro call, not a %rep, and
4222 * therefore the parameter information needs to
4223 * be freed.
4225 nasm_free(m->params);
4226 free_tlist(m->iline);
4227 nasm_free(m->paramlen);
4228 l->finishes->in_progress = FALSE;
4230 else
4231 free_mmacro(m);
4233 istk->expansion = l->next;
4234 nasm_free(l);
4235 list->downlevel(LIST_MACRO);
4238 while (1)
4239 { /* until we get a line we can use */
4241 if (istk->expansion)
4242 { /* from a macro expansion */
4243 char *p;
4244 Line *l = istk->expansion;
4245 if (istk->mstk)
4246 istk->mstk->lineno++;
4247 tline = l->first;
4248 istk->expansion = l->next;
4249 nasm_free(l);
4250 p = detoken(tline, FALSE);
4251 list->line(LIST_MACRO, p);
4252 nasm_free(p);
4253 break;
4255 line = read_line();
4256 if (line)
4257 { /* from the current input file */
4258 line = prepreproc(line);
4259 tline = tokenise(line);
4260 nasm_free(line);
4261 break;
4264 * The current file has ended; work down the istk
4267 Include *i = istk;
4268 fclose(i->fp);
4269 if (i->conds)
4270 error(ERR_FATAL, "expected `%%endif' before end of file");
4271 /* only set line and file name if there's a next node */
4272 if (i->next)
4274 src_set_linnum(i->lineno);
4275 nasm_free(src_set_fname(i->fname));
4277 istk = i->next;
4278 list->downlevel(LIST_INCLUDE);
4279 nasm_free(i);
4280 if (!istk)
4281 return NULL;
4286 * We must expand MMacro parameters and MMacro-local labels
4287 * _before_ we plunge into directive processing, to cope
4288 * with things like `%define something %1' such as STRUC
4289 * uses. Unless we're _defining_ a MMacro, in which case
4290 * those tokens should be left alone to go into the
4291 * definition; and unless we're in a non-emitting
4292 * condition, in which case we don't want to meddle with
4293 * anything.
4295 if (!defining && !(istk->conds && !emitting(istk->conds->state)))
4296 tline = expand_mmac_params(tline);
4299 * Check the line to see if it's a preprocessor directive.
4301 if (do_directive(tline) == DIRECTIVE_FOUND)
4303 continue;
4305 else if (defining)
4308 * We're defining a multi-line macro. We emit nothing
4309 * at all, and just
4310 * shove the tokenised line on to the macro definition.
4312 Line *l = nasm_malloc(sizeof(Line));
4313 l->next = defining->expansion;
4314 l->first = tline;
4315 l->finishes = FALSE;
4316 defining->expansion = l;
4317 continue;
4319 else if (istk->conds && !emitting(istk->conds->state))
4322 * We're in a non-emitting branch of a condition block.
4323 * Emit nothing at all, not even a blank line: when we
4324 * emerge from the condition we'll give a line-number
4325 * directive so we keep our place correctly.
4327 free_tlist(tline);
4328 continue;
4330 else if (istk->mstk && !istk->mstk->in_progress)
4333 * We're in a %rep block which has been terminated, so
4334 * we're walking through to the %endrep without
4335 * emitting anything. Emit nothing at all, not even a
4336 * blank line: when we emerge from the %rep block we'll
4337 * give a line-number directive so we keep our place
4338 * correctly.
4340 free_tlist(tline);
4341 continue;
4343 else
4345 tline = expand_smacro(tline);
4346 if (!expand_mmacro(tline))
4349 * De-tokenise the line again, and emit it.
4351 line = detoken(tline, TRUE);
4352 free_tlist(tline);
4353 break;
4355 else
4357 continue; /* expand_mmacro calls free_tlist */
4362 return line;
4365 static void
4366 pp_cleanup(int pass)
4368 int h;
4370 if (defining)
4372 error(ERR_NONFATAL, "end of file while still defining macro `%s'",
4373 defining->name);
4374 free_mmacro(defining);
4376 while (cstk)
4377 ctx_pop();
4378 for (h = 0; h < NHASH; h++)
4380 while (mmacros[h])
4382 MMacro *m = mmacros[h];
4383 mmacros[h] = mmacros[h]->next;
4384 free_mmacro(m);
4386 while (smacros[h])
4388 SMacro *s = smacros[h];
4389 smacros[h] = smacros[h]->next;
4390 nasm_free(s->name);
4391 free_tlist(s->expansion);
4392 nasm_free(s);
4395 while (istk)
4397 Include *i = istk;
4398 istk = istk->next;
4399 fclose(i->fp);
4400 nasm_free(i->fname);
4401 nasm_free(i);
4403 while (cstk)
4404 ctx_pop();
4405 if (pass == 0)
4407 free_llist(predef);
4408 delete_Blocks();
4412 void
4413 pp_include_path(char *path)
4415 IncPath *i;
4416 /* by alexfru: order of path inclusion fixed (was reverse order) */
4417 i = nasm_malloc(sizeof(IncPath));
4418 i->path = nasm_strdup(path);
4419 i->next = NULL;
4421 if (ipath != NULL)
4423 IncPath *j = ipath;
4424 while (j->next != NULL)
4425 j = j->next;
4426 j->next = i;
4428 else
4430 ipath = i;
4435 * added by alexfru:
4437 * This function is used to "export" the include paths, e.g.
4438 * the paths specified in the '-I' command switch.
4439 * The need for such exporting is due to the 'incbin' directive,
4440 * which includes raw binary files (unlike '%include', which
4441 * includes text source files). It would be real nice to be
4442 * able to specify paths to search for incbin'ned files also.
4443 * So, this is a simple workaround.
4445 * The function use is simple:
4447 * The 1st call (with NULL argument) returns a pointer to the 1st path
4448 * (char** type) or NULL if none include paths available.
4450 * All subsequent calls take as argument the value returned by this
4451 * function last. The return value is either the next path
4452 * (char** type) or NULL if the end of the paths list is reached.
4454 * It is maybe not the best way to do things, but I didn't want
4455 * to export too much, just one or two functions and no types or
4456 * variables exported.
4458 * Can't say I like the current situation with e.g. this path list either,
4459 * it seems to be never deallocated after creation...
4461 char**
4462 pp_get_include_path_ptr (char **pPrevPath)
4464 /* This macro returns offset of a member of a structure */
4465 #define GetMemberOffset(StructType,MemberName)\
4466 ((size_t)&((StructType*)0)->MemberName)
4467 IncPath *i;
4469 if (pPrevPath == NULL)
4471 if(ipath != NULL)
4472 return &ipath->path;
4473 else
4474 return NULL;
4476 i = (IncPath*) ((char*)pPrevPath - GetMemberOffset(IncPath,path));
4477 i = i->next;
4478 if (i != NULL)
4479 return &i->path;
4480 else
4481 return NULL;
4482 #undef GetMemberOffset
4485 void
4486 pp_pre_include(char *fname)
4488 Token *inc, *space, *name;
4489 Line *l;
4491 name = new_Token(NULL, TOK_INTERNAL_STRING, fname, 0);
4492 space = new_Token(name, TOK_WHITESPACE, NULL, 0);
4493 inc = new_Token(space, TOK_PREPROC_ID, "%include", 0);
4495 l = nasm_malloc(sizeof(Line));
4496 l->next = predef;
4497 l->first = inc;
4498 l->finishes = FALSE;
4499 predef = l;
4502 void
4503 pp_pre_define(char *definition)
4505 Token *def, *space;
4506 Line *l;
4507 char *equals;
4509 equals = strchr(definition, '=');
4510 space = new_Token(NULL, TOK_WHITESPACE, NULL, 0);
4511 def = new_Token(space, TOK_PREPROC_ID, "%define", 0);
4512 if (equals)
4513 *equals = ' ';
4514 space->next = tokenise(definition);
4515 if (equals)
4516 *equals = '=';
4518 l = nasm_malloc(sizeof(Line));
4519 l->next = predef;
4520 l->first = def;
4521 l->finishes = FALSE;
4522 predef = l;
4525 void
4526 pp_pre_undefine(char *definition)
4528 Token *def, *space;
4529 Line *l;
4531 space = new_Token(NULL, TOK_WHITESPACE, NULL, 0);
4532 def = new_Token(space, TOK_PREPROC_ID, "%undef", 0);
4533 space->next = tokenise(definition);
4535 l = nasm_malloc(sizeof(Line));
4536 l->next = predef;
4537 l->first = def;
4538 l->finishes = FALSE;
4539 predef = l;
4542 void
4543 pp_extra_stdmac(const char **macros)
4545 extrastdmac = macros;
4548 static void
4549 make_tok_num(Token * tok, long val)
4551 char numbuf[20];
4552 snprintf(numbuf, sizeof(numbuf), "%ld", val);
4553 tok->text = nasm_strdup(numbuf);
4554 tok->type = TOK_NUMBER;
4557 Preproc nasmpp = {
4558 pp_reset,
4559 pp_getline,
4560 pp_cleanup