]> de.git.xonotic.org Git - xonotic/gmqcc.git/blob - gmqcc.h
type_eq_instr[], type_ne_instr[]
[xonotic/gmqcc.git] / gmqcc.h
1 /*
2  * Copyright (C) 2012
3  *     Dale Weiler, Wolfgang Bumiller
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a copy of
6  * this software and associated documentation files (the "Software"), to deal in
7  * the Software without restriction, including without limitation the rights to
8  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
9  * of the Software, and to permit persons to whom the Software is furnished to do
10  * so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice shall be included in all
13  * copies or substantial portions of the Software.
14  *
15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21  * SOFTWARE.
22  */
23 #ifndef GMQCC_HDR
24 #define GMQCC_HDR
25 #include <limits.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdio.h>
29 #include <ctype.h>
30
31 #define GMQCC_VERSION_MAJOR 0
32 #define GMQCC_VERSION_MINOR 1
33 #define GMQCC_VERSION_PATCH 0
34 #define GMQCC_VERSION_BUILD(J,N,P) (((J)<<16)|((N)<<8)|(P))
35 #define GMQCC_VERSION \
36     GMQCC_VERSION_BUILD(GMQCC_VERSION_MAJOR, GMQCC_VERSION_MINOR, GMQCC_VERSION_PATCH)
37
38 /*
39  * We cannoy rely on C99 at all, since compilers like MSVC
40  * simply don't support it.  We define our own boolean type
41  * as a result (since we cannot include <stdbool.h>). For
42  * compilers that are in 1999 mode (C99 compliant) we can use
43  * the language keyword _Bool which can allow for better code
44  * on GCC and GCC-like compilers, opposed to `int`.
45  */
46 #ifndef __cplusplus
47 #   ifdef  false
48 #       undef  false
49 #   endif /* !false */
50 #   ifdef  true
51 #       undef true
52 #   endif /* !true  */
53 #   define false (0)
54 #   define true  (1)
55 #   ifdef __STDC_VERSION__
56 #       if __STDC_VERSION__ < 199901L && __GNUC__ < 3
57             typedef int  bool;
58 #       else
59             typedef _Bool bool;
60 #       endif
61 #   else
62         typedef int bool;
63 #   endif /* !__STDC_VERSION__ */
64 #endif    /* !__cplusplus      */
65
66 /*
67  * Of some functions which are generated we want to make sure
68  * that the result isn't ignored. To find such function calls,
69  * we use this macro.
70  */
71 #if defined(__GNUC__) || defined(__CLANG__)
72 #   define GMQCC_WARN __attribute__((warn_unused_result))
73 #else
74 #   define GMQCC_WARN
75 #endif
76 /*
77  * This is a hack to silent clang regarding empty
78  * body if statements.
79  */
80 #define GMQCC_SUPPRESS_EMPTY_BODY do { } while (0)
81
82 /*
83  * Inline is not supported in < C90, however some compilers
84  * like gcc and clang might have an inline attribute we can
85  * use if present.
86  */
87 #ifdef __STDC_VERSION__
88 #    if __STDC_VERSION__ < 199901L
89 #       if defined(__GNUC__) || defined (__CLANG__)
90 #           if __GNUC__ < 2
91 #               define GMQCC_INLINE
92 #           else
93 #               define GMQCC_INLINE __attribute__ ((always_inline))
94 #           endif
95 #       else
96 #           define GMQCC_INLINE
97 #       endif
98 #    else
99 #       define GMQCC_INLINE inline
100 #    endif
101 #else
102 #    define GMQCC_INLINE
103 #endif /* !__STDC_VERSION__ */
104
105 /*
106  * noreturn is present in GCC and clang
107  * it's required for _ast_node_destory otherwise -Wmissing-noreturn
108  * in clang complains about there being no return since abort() is
109  * called.
110  */
111 #if (defined(__GNUC__) && __GNUC__ >= 2) || defined(__CLANG__)
112 #    define GMQCC_NORETURN __attribute__ ((noreturn))
113 #else
114 #    define GMQCC_NORETURN
115 #endif
116
117 /*
118  * stdint.h and inttypes.h -less subset
119  * for systems that don't have it, which we must
120  * assume is all systems. (int8_t not required)
121  */
122 #if   CHAR_MIN  == -128
123     typedef unsigned char  uint8_t; /* same as below */
124 #elif SCHAR_MIN == -128
125     typedef unsigned char  uint8_t; /* same as above */
126 #endif
127 #if   SHRT_MAX  == 0x7FFF
128     typedef short          int16_t;
129     typedef unsigned short uint16_t;
130 #elif INT_MAX   == 0x7FFF
131     typedef int            int16_t;
132     typedef unsigned int   uint16_t;
133 #endif
134 #if   INT_MAX   == 0x7FFFFFFF
135     typedef int            int32_t;
136     typedef unsigned int   uint32_t;
137     typedef long           int64_t;
138     typedef unsigned long  uint64_t;
139 #elif LONG_MAX  == 0x7FFFFFFF
140     typedef long           int32_t;
141     typedef unsigned long  uint32_t;
142
143     /*
144      * It's nearly impossible to figure out a 64bit type at
145      * this point without making assumptions about the build
146      * enviroment.  So if clang or gcc is detected use some
147      * compiler builtins to create a 64 signed and unsigned
148      * type.
149      */
150 #   if defined(__GNUC__) || defined (__CLANG__)
151         typedef int          int64_t  __attribute__((__mode__(__DI__)));
152         typedef unsigned int uint64_t __attribute__((__mode__(__DI__)));
153 #   else
154         /*
155          * Incoorectly size the types so static assertions below will
156          * fail.  There is no valid way to get a 64bit type at this point
157          * without making assumptions of too many things.
158          */
159         typedef struct { char _fail : 0; } int64_t;
160         typedef struct { char _fail : 0; } uint64_t;
161 #   endif
162 #endif
163 #ifdef _LP64 /* long pointer == 64 */
164     typedef unsigned long  uintptr_t;
165     typedef long           intptr_t;
166 #else
167     typedef unsigned int   uintptr_t;
168     typedef int            intptr_t;
169 #endif
170 /* Ensure type sizes are correct: */
171 typedef char uint8_size_is_correct  [sizeof(uint8_t)  == 1?1:-1];
172 typedef char uint16_size_is_correct [sizeof(uint16_t) == 2?1:-1];
173 typedef char uint32_size_is_correct [sizeof(uint32_t) == 4?1:-1];
174 typedef char uint64_size_is_correct [sizeof(uint64_t) == 8?1:-1];
175 typedef char int16_size_if_correct  [sizeof(int16_t)  == 2?1:-1];
176 typedef char int32_size_is_correct  [sizeof(int32_t)  == 4?1:-1];
177 typedef char int64_size_is_correct  [sizeof(int64_t)  == 8?1:-1];
178 /* intptr_t / uintptr_t correct size check */
179 typedef char uintptr_size_is_correct[sizeof(intptr_t) == sizeof(int*)?1:-1];
180 typedef char intptr_size_is_correct [sizeof(uintptr_t)== sizeof(int*)?1:-1];
181
182 /*===================================================================*/
183 /*============================ lex.c ================================*/
184 /*===================================================================*/
185 typedef struct lex_file_t {
186     FILE *file;        /* file handler */
187     char *name;        /* name of file */
188     char  peek  [5];
189     char  lastok[8192];
190
191     int   last;    /* last token                   */
192     int   current; /* current token                */
193     int   length;  /* bytes left to parse          */
194     int   size;    /* never changes (size of file) */
195     int   line;    /* what line are we on?         */
196 } lex_file;
197
198 /*
199  * It's important that this table never exceed 32 keywords, the ascii
200  * table starts at 33 (and we don't want conflicts)
201  */
202 enum {
203     TOKEN_DO       ,
204     TOKEN_ELSE     ,
205     TOKEN_IF       ,
206     TOKEN_WHILE    ,
207     TOKEN_BREAK    ,
208     TOKEN_CONTINUE ,
209     TOKEN_RETURN   ,
210     TOKEN_GOTO     ,
211     TOKEN_FOR      ,   /* extension */
212     TOKEN_TYPEDEF  ,   /* extension */
213
214     /* ensure the token types are out of the  */
215     /* bounds of anyothers that may conflict. */
216     TOKEN_FLOAT    = 110,
217     TOKEN_VECTOR        ,
218     TOKEN_STRING        ,
219     TOKEN_ENTITY        ,
220     TOKEN_VOID
221 };
222
223 /*
224  * Lexer state constants, these are numbers for where exactly in
225  * the lexing the lexer is at. Or where it decided to stop if a lexer
226  * error occurs.  These numbers must be > where the ascii-table ends
227  * and > the last type token which is TOKEN_VOID
228  */
229 enum {
230     LEX_COMMENT = 1128,
231     LEX_CHRLIT        ,
232     LEX_STRLIT        ,
233     LEX_IDENT
234 };
235
236 int       lex_token  (lex_file *);
237 void      lex_reset  (lex_file *);
238 void      lex_close  (lex_file *);
239 void      lex_parse  (lex_file *);
240 lex_file *lex_include(lex_file *, const char *);
241 void      lex_init   (const char *, lex_file **);
242
243 /*===================================================================*/
244 /*========================== error.c ================================*/
245 /*===================================================================*/
246 #define ERROR_LEX      (SHRT_MAX+0)
247 #define ERROR_PARSE    (SHRT_MAX+1)
248 #define ERROR_INTERNAL (SHRT_MAX+2)
249 #define ERROR_COMPILER (SHRT_MAX+3)
250 #define ERROR_PREPRO   (SHRT_MAX+4)
251 int error(lex_file *, int, const char *, ...);
252
253 /*===================================================================*/
254 /*========================== parse.c ================================*/
255 /*===================================================================*/
256 int parse_gen(lex_file *);
257
258 /*===================================================================*/
259 /*========================== typedef.c ==============================*/
260 /*===================================================================*/
261 typedef struct typedef_node_t {
262     char      *name;
263 } typedef_node;
264
265 void          typedef_init();
266 void          typedef_clear();
267 typedef_node *typedef_find(const char *);
268 int           typedef_add (lex_file *file, const char *, const char *);
269
270
271 /*===================================================================*/
272 /*=========================== util.c ================================*/
273 /*===================================================================*/
274 void *util_memory_a      (unsigned int, unsigned int, const char *);
275 void  util_memory_d      (void       *, unsigned int, const char *);
276 void  util_meminfo       ();
277
278 bool  util_strupper      (const char *);
279 bool  util_strdigit      (const char *);
280 bool  util_strncmpexact  (const char *, const char *, size_t);
281 char *util_strdup        (const char *);
282 char *util_strrq         (const char *);
283 char *util_strrnl        (const char *);
284 char *util_strsws        (const char *);
285 char *util_strchp        (const char *, const char *);
286 void  util_debug         (const char *, const char *, ...);
287 int   util_getline       (char **, size_t *, FILE *);
288 void  util_endianswap    (void *,  int, int);
289
290 size_t util_strtocmd    (const char *, char *, size_t);
291 size_t util_strtononcmd (const char *, char *, size_t);
292
293 uint32_t util_crc32(const char *, int, register const short);
294
295 #ifdef NOTRACK
296 #    define mem_a(x) malloc(x)
297 #    define mem_d(x) free  (x)
298 #else
299 #    define mem_a(x) util_memory_a((x), __LINE__, __FILE__)
300 #    define mem_d(x) util_memory_d((x), __LINE__, __FILE__)
301 #endif
302
303 /*
304  * TODO: make these safer to use.  Currently this only works on
305  * x86 and x86_64, some systems will likely not like this. Such
306  * as BE systems.
307  */
308 #define FLT2INT(Y) *((int32_t*)&(Y))
309 #define INT2FLT(Y) *((float  *)&(Y))
310
311 /* Builds vector type (usefull for inside structures) */
312 #define VECTOR_SNAP(X,Y) X ## Y
313 #define VECTOR_FILL(X,Y) VECTOR_SNAP(X,Y)
314 #define VECTOR_TYPE(T,N)                                        \
315     T*     N##_data      = NULL;                                \
316     long   N##_elements  = 0;                                   \
317     long   N##_allocated = 0
318 /* Builds vector add */
319 #define VECTOR_CORE(T,N)                                        \
320     int    N##_add(T element) {                                 \
321         void *temp = NULL;                                      \
322         if (N##_elements == N##_allocated) {                    \
323             if (N##_allocated == 0) {                           \
324                 N##_allocated = 12;                             \
325             } else {                                            \
326                 N##_allocated *= 2;                             \
327             }                                                   \
328             if  (!(temp = mem_a(N##_allocated * sizeof(T)))) {  \
329                 mem_d(temp);                                    \
330                 return -1;                                      \
331             }                                                   \
332             memcpy(temp, N##_data, (N##_elements * sizeof(T))); \
333             mem_d(N##_data);                                    \
334             N##_data = (T*)temp;                                \
335         }                                                       \
336         N##_data[N##_elements] = element;                       \
337         return   N##_elements++;                                \
338     }                                                           \
339     int N##_put(T* elements, size_t len) {                      \
340         len     --;                                             \
341         elements--;                                             \
342         while (N##_add(*++elements) != -1 && len--);            \
343         return N##_elements;                                    \
344     }                                                           \
345     typedef char VECTOR_FILL(extra_semicolon_##N,__COUNTER__)
346 #define VECTOR_PROT(T,N)                                        \
347     extern T*     N##_data     ;                                \
348     extern long   N##_elements ;                                \
349     extern long   N##_allocated;                                \
350     int           N##_add(T);                                   \
351     int           N##_put(T *, size_t)
352 #define VECTOR_MAKE(T,N) \
353     VECTOR_TYPE(T,N);    \
354     VECTOR_CORE(T,N)
355
356 /*===================================================================*/
357 /*=========================== code.c ================================*/
358 /*===================================================================*/
359
360 /* Note: if you change the order, fix type_sizeof in ir.c */
361 enum {
362     TYPE_VOID     ,
363     TYPE_STRING   ,
364     TYPE_FLOAT    ,
365     TYPE_VECTOR   ,
366     TYPE_ENTITY   ,
367     TYPE_FIELD    ,
368     TYPE_FUNCTION ,
369     TYPE_POINTER  ,
370     /* TYPE_INTEGER  , */
371     TYPE_VARIANT  ,
372
373     TYPE_COUNT
374 };
375
376 extern size_t type_sizeof[TYPE_COUNT];
377 extern uint16_t type_store_instr[TYPE_COUNT];
378 /* could use type_store_instr + INSTR_STOREP_F - INSTR_STORE_F
379  * but this breaks when TYPE_INTEGER is added, since with the enhanced
380  * instruction set, the old ones are left untouched, thus the _I instructions
381  * are at a seperate place.
382  */
383 extern uint16_t type_storep_instr[TYPE_COUNT];
384 /* other useful lists */
385 extern uint16_t type_eq_instr[TYPE_COUNT];
386 extern uint16_t type_ne_instr[TYPE_COUNT];
387
388 typedef struct {
389     uint32_t offset;      /* Offset in file of where data begins  */
390     uint32_t length;      /* Length of section (how many of)      */
391 } prog_section;
392
393 typedef struct {
394     uint32_t     version;      /* Program version (6)     */
395     uint16_t     crc16;        /* What is this?           */
396     uint16_t     skip;         /* see propsal.txt         */
397
398     prog_section statements;   /* prog_section_statement  */
399     prog_section defs;         /* prog_section_def        */
400     prog_section fields;       /* prog_section_field      */
401     prog_section functions;    /* prog_section_function   */
402     prog_section strings;      /* What is this?           */
403     prog_section globals;      /* What is this?           */
404     uint32_t     entfield;     /* Number of entity fields */
405 } prog_header;
406
407 /*
408  * Each paramater incerements by 3 since vector types hold
409  * 3 components (x,y,z).
410  */
411 #define OFS_NULL      0
412 #define OFS_RETURN    1
413 #define OFS_PARM0     (OFS_RETURN+3)
414 #define OFS_PARM1     (OFS_PARM0 +3)
415 #define OFS_PARM2     (OFS_PARM1 +3)
416 #define OFS_PARM3     (OFS_PARM2 +3)
417 #define OFS_PARM4     (OFS_PARM3 +3)
418 #define OFS_PARM5     (OFS_PARM4 +3)
419 #define OFS_PARM6     (OFS_PARM5 +3)
420 #define OFS_PARM7     (OFS_PARM6 +3)
421
422 typedef struct {
423     uint16_t opcode;
424
425     /* operand 1 */
426     union {
427         int16_t  s1; /* signed   */
428         uint16_t u1; /* unsigned */
429     } o1;
430     /* operand 2 */
431     union {
432         int16_t  s1; /* signed   */
433         uint16_t u1; /* unsigned */
434     } o2;
435     /* operand 3 */
436     union {
437         int16_t  s1; /* signed   */
438         uint16_t u1; /* unsigned */
439     } o3;
440
441     /*
442      * This is the same as the structure in darkplaces
443      * {
444      *     unsigned short op;
445      *     short          a,b,c;
446      * }
447      * But this one is more sane to work with, and the
448      * type sizes are guranteed.
449      */
450 } prog_section_statement;
451
452 typedef struct {
453     /* The types:
454      * 0 = ev_void
455      * 1 = ev_string
456      * 2 = ev_float
457      * 3 = ev_vector
458      * 4 = ev_entity
459      * 5 = ev_field
460      * 6 = ev_function
461      * 7 = ev_pointer -- engine only
462      * 8 = ev_bad     -- engine only
463      */
464     uint16_t type;
465     uint16_t offset;
466     uint32_t name;
467 } prog_section_both;
468 typedef prog_section_both prog_section_def;
469 typedef prog_section_both prog_section_field;
470
471 typedef struct {
472     int32_t   entry;      /* in statement table for instructions  */
473     uint32_t  firstlocal; /* First local in local table           */
474     uint32_t  locals;     /* Total ints of params + locals        */
475     uint32_t  profile;    /* Always zero (engine uses this)       */
476     uint32_t  name;       /* name of function in string table     */
477     uint32_t  file;       /* file of the source file              */
478     uint32_t  nargs;      /* number of arguments                  */
479     uint8_t   argsize[8]; /* size of arguments (keep 8 always?)   */
480 } prog_section_function;
481
482 /*
483  * Instructions
484  * These are the external instructions supported by the interperter
485  * this is what things compile to (from the C code).
486  */
487 enum {
488     INSTR_DONE,
489     INSTR_MUL_F,
490     INSTR_MUL_V,
491     INSTR_MUL_FV,
492     INSTR_MUL_VF,
493     INSTR_DIV_F,
494     INSTR_ADD_F,
495     INSTR_ADD_V,
496     INSTR_SUB_F,
497     INSTR_SUB_V,
498     INSTR_EQ_F,
499     INSTR_EQ_V,
500     INSTR_EQ_S,
501     INSTR_EQ_E,
502     INSTR_EQ_FNC,
503     INSTR_NE_F,
504     INSTR_NE_V,
505     INSTR_NE_S,
506     INSTR_NE_E,
507     INSTR_NE_FNC,
508     INSTR_LE,
509     INSTR_GE,
510     INSTR_LT,
511     INSTR_GT,
512     INSTR_LOAD_F,
513     INSTR_LOAD_V,
514     INSTR_LOAD_S,
515     INSTR_LOAD_ENT,
516     INSTR_LOAD_FLD,
517     INSTR_LOAD_FNC,
518     INSTR_ADDRESS,
519     INSTR_STORE_F,
520     INSTR_STORE_V,
521     INSTR_STORE_S,
522     INSTR_STORE_ENT,
523     INSTR_STORE_FLD,
524     INSTR_STORE_FNC,
525     INSTR_STOREP_F,
526     INSTR_STOREP_V,
527     INSTR_STOREP_S,
528     INSTR_STOREP_ENT,
529     INSTR_STOREP_FLD,
530     INSTR_STOREP_FNC,
531     INSTR_RETURN,
532     INSTR_NOT_F,
533     INSTR_NOT_V,
534     INSTR_NOT_S,
535     INSTR_NOT_ENT,
536     INSTR_NOT_FNC,
537     INSTR_IF,
538     INSTR_IFNOT,
539     INSTR_CALL0,
540     INSTR_CALL1,
541     INSTR_CALL2,
542     INSTR_CALL3,
543     INSTR_CALL4,
544     INSTR_CALL5,
545     INSTR_CALL6,
546     INSTR_CALL7,
547     INSTR_CALL8,
548     INSTR_STATE,
549     INSTR_GOTO,
550     INSTR_AND,
551     INSTR_OR,
552     INSTR_BITAND,
553     INSTR_BITOR,
554
555     /*
556      * Virtual instructions used by the assembler
557      * keep at the end but before virtual instructions
558      * for the IR below.
559      */
560     AINSTR_END,
561
562     /*
563      * Virtual instructions used by the IR
564      * Keep at the end!
565      */
566     VINSTR_PHI,
567     VINSTR_JUMP,
568     VINSTR_COND
569 };
570
571 /*
572  * The symbols below are created by the following
573  * expanded macros:
574  *
575  * VECTOR_MAKE(prog_section_statement, code_statements);
576  * VECTOR_MAKE(prog_section_def,       code_defs      );
577  * VECTOR_MAKE(prog_section_field,     code_fields    );
578  * VECTOR_MAKE(prog_section_function,  code_functions );
579  * VECTOR_MAKE(int,                    code_globals   );
580  * VECTOR_MAKE(char,                   code_chars     );
581  */
582 VECTOR_PROT(prog_section_statement, code_statements);
583 VECTOR_PROT(prog_section_statement, code_statements);
584 VECTOR_PROT(prog_section_def,       code_defs      );
585 VECTOR_PROT(prog_section_field,     code_fields    );
586 VECTOR_PROT(prog_section_function,  code_functions );
587 VECTOR_PROT(int,                    code_globals   );
588 VECTOR_PROT(char,                   code_chars     );
589
590 typedef float   qcfloat;
591 typedef int32_t qcint;
592
593 /*
594  * code_write -- writes out the compiled file
595  * code_init  -- prepares the code file
596  */
597 bool     code_write       (const char *filename);
598 void     code_init        ();
599 uint32_t code_genstring   (const char *string);
600 uint32_t code_cachedstring(const char *string);
601 qcint    code_alloc_field (size_t qcsize);
602
603 /*===================================================================*/
604 /*========================= assembler.c =============================*/
605 /*===================================================================*/
606 static const struct {
607     const char  *m; /* menomic     */
608     const size_t o; /* operands    */
609     const size_t l; /* menomic len */
610 } asm_instr[] = {
611     { "DONE"      , 1, 4 },
612     { "MUL_F"     , 3, 5 },
613     { "MUL_V"     , 3, 5 },
614     { "MUL_FV"    , 3, 6 },
615     { "MUL_VF"    , 3, 6 },
616     { "DIV"       , 0, 3 },
617     { "ADD_F"     , 3, 5 },
618     { "ADD_V"     , 3, 5 },
619     { "SUB_F"     , 3, 5 },
620     { "DUB_V"     , 3, 5 },
621     { "EQ_F"      , 0, 4 },
622     { "EQ_V"      , 0, 4 },
623     { "EQ_S"      , 0, 4 },
624     { "EQ_E"      , 0, 4 },
625     { "EQ_FNC"    , 0, 6 },
626     { "NE_F"      , 0, 4 },
627     { "NE_V"      , 0, 4 },
628     { "NE_S"      , 0, 4 },
629     { "NE_E"      , 0, 4 },
630     { "NE_FNC"    , 0, 6 },
631     { "LE"        , 0, 2 },
632     { "GE"        , 0, 2 },
633     { "LT"        , 0, 2 },
634     { "GT"        , 0, 2 },
635     { "FIELD_F"   , 0, 7 },
636     { "FIELD_V"   , 0, 7 },
637     { "FIELD_S"   , 0, 7 },
638     { "FIELD_ENT" , 0, 9 },
639     { "FIELD_FLD" , 0, 9 },
640     { "FIELD_FNC" , 0, 9 },
641     { "ADDRESS"   , 0, 7 },
642     { "STORE_F"   , 0, 7 },
643     { "STORE_V"   , 0, 7 },
644     { "STORE_S"   , 0, 7 },
645     { "STORE_ENT" , 0, 9 },
646     { "STORE_FLD" , 0, 9 },
647     { "STORE_FNC" , 0, 9 },
648     { "STOREP_F"  , 0, 8 },
649     { "STOREP_V"  , 0, 8 },
650     { "STOREP_S"  , 0, 8 },
651     { "STOREP_ENT", 0, 10},
652     { "STOREP_FLD", 0, 10},
653     { "STOREP_FNC", 0, 10},
654     { "RETURN"    , 0, 6 },
655     { "NOT_F"     , 0, 5 },
656     { "NOT_V"     , 0, 5 },
657     { "NOT_S"     , 0, 5 },
658     { "NOT_ENT"   , 0, 7 },
659     { "NOT_FNC"   , 0, 7 },
660     { "IF"        , 0, 2 },
661     { "IFNOT"     , 0, 5 },
662     { "CALL0"     , 1, 5 },
663     { "CALL1"     , 2, 5 },
664     { "CALL2"     , 3, 5 },
665     { "CALL3"     , 4, 5 },
666     { "CALL4"     , 5, 5 },
667     { "CALL5"     , 6, 5 },
668     { "CALL6"     , 7, 5 },
669     { "CALL7"     , 8, 5 },
670     { "CALL8"     , 9, 5 },
671     { "STATE"     , 0, 5 },
672     { "GOTO"      , 0, 4 },
673     { "AND"       , 0, 3 },
674     { "OR"        , 0, 2 },
675     { "BITAND"    , 0, 6 },
676     { "BITOR"     , 0, 5 },
677     { "END"       , 0, 3 } /* virtual assembler instruction */
678 };
679
680 void asm_init (const char *, FILE **);
681 void asm_close(FILE *);
682 void asm_parse(FILE *);
683 /*===================================================================*/
684 /*============================= ast.c ===============================*/
685 /*===================================================================*/
686 #define MEM_VECTOR_PROTO(Towner, Tmem, mem)                   \
687     bool GMQCC_WARN Towner##_##mem##_add(Towner*, Tmem);      \
688     bool GMQCC_WARN Towner##_##mem##_remove(Towner*, size_t)
689
690 #define MEM_VECTOR_PROTO_ALL(Towner, Tmem, mem)                    \
691     MEM_VECTOR_PROTO(Towner, Tmem, mem);                           \
692     bool GMQCC_WARN Towner##_##mem##_find(Towner*, Tmem, size_t*); \
693     void Towner##_##mem##_clear(Towner*)
694
695 #define MEM_VECTOR_MAKE(Twhat, name) \
696     Twhat  *name;                    \
697     size_t name##_count;             \
698     size_t name##_alloc
699
700 #define MEM_VEC_FUN_ADD(Tself, Twhat, mem)                           \
701 bool GMQCC_WARN Tself##_##mem##_add(Tself *self, Twhat f)            \
702 {                                                                    \
703     Twhat *reall;                                                    \
704     if (self->mem##_count == self->mem##_alloc) {                    \
705         if (!self->mem##_alloc) {                                    \
706             self->mem##_alloc = 16;                                  \
707         } else {                                                     \
708             self->mem##_alloc *= 2;                                  \
709         }                                                            \
710         reall = (Twhat*)mem_a(sizeof(Twhat) * self->mem##_alloc);    \
711         if (!reall) {                                                \
712             return false;                                            \
713         }                                                            \
714         memcpy(reall, self->mem, sizeof(Twhat) * self->mem##_count); \
715         mem_d(self->mem);                                            \
716         self->mem = reall;                                           \
717     }                                                                \
718     self->mem[self->mem##_count++] = f;                              \
719     return true;                                                     \
720 }
721
722 #define MEM_VEC_FUN_REMOVE(Tself, Twhat, mem)                        \
723 bool GMQCC_WARN Tself##_##mem##_remove(Tself *self, size_t idx)      \
724 {                                                                    \
725     size_t i;                                                        \
726     Twhat *reall;                                                    \
727     if (idx >= self->mem##_count) {                                  \
728         return true; /* huh... */                                    \
729     }                                                                \
730     for (i = idx; i < self->mem##_count-1; ++i) {                    \
731         self->mem[i] = self->mem[i+1];                               \
732     }                                                                \
733     self->mem##_count--;                                             \
734     if (self->mem##_count < self->mem##_count/2) {                   \
735         self->mem##_alloc /= 2;                                      \
736         reall = (Twhat*)mem_a(sizeof(Twhat) * self->mem##_count);    \
737         if (!reall) {                                                \
738             return false;                                            \
739         }                                                            \
740         memcpy(reall, self->mem, sizeof(Twhat) * self->mem##_count); \
741         mem_d(self->mem);                                            \
742         self->mem = reall;                                           \
743     }                                                                \
744     return true;                                                     \
745 }
746
747 #define MEM_VEC_FUN_FIND(Tself, Twhat, mem)                     \
748 bool GMQCC_WARN Tself##_##mem##_find(Tself *self, Twhat obj, size_t *idx) \
749 {                                                               \
750     size_t i;                                                   \
751     for (i = 0; i < self->mem##_count; ++i) {                   \
752         if (self->mem[i] == obj) {                              \
753             if (idx) {                                          \
754                 *idx = i;                                       \
755             }                                                   \
756             return true;                                        \
757         }                                                       \
758     }                                                           \
759     return false;                                               \
760 }
761
762 #define MEM_VEC_FUN_APPEND(Tself, Twhat, mem)                        \
763 bool GMQCC_WARN Tself##_##mem##_append(Tself *s, Twhat *p, size_t c) \
764 {                                                                    \
765     Twhat *reall;                                                    \
766     if (s->mem##_count+c >= s->mem##_alloc) {                        \
767         if (!s->mem##_alloc) {                                       \
768             s->mem##_alloc = c < 16 ? 16 : c;                        \
769         } else {                                                     \
770             s->mem##_alloc *= 2;                                     \
771             if (s->mem##_count+c >= s->mem##_alloc) {                \
772                 s->mem##_alloc = s->mem##_count+c;                   \
773             }                                                        \
774         }                                                            \
775         reall = (Twhat*)mem_a(sizeof(Twhat) * s->mem##_alloc);       \
776         if (!reall) {                                                \
777             return false;                                            \
778         }                                                            \
779         memcpy(reall, s->mem, sizeof(Twhat) * s->mem##_count);       \
780         mem_d(s->mem);                                               \
781         s->mem = reall;                                              \
782     }                                                                \
783     memcpy(&s->mem[s->mem##_count], p, c*sizeof(*p));                \
784     s->mem##_count += c;                                             \
785     return true;                                                     \
786 }
787
788 #define MEM_VEC_FUN_RESIZE(Tself, Twhat, mem)                    \
789 bool GMQCC_WARN Tself##_##mem##_resize(Tself *s, size_t c)       \
790 {                                                                \
791     Twhat *reall;                                                \
792     if (c > s->mem##_alloc) {                                    \
793         reall = (Twhat*)mem_a(sizeof(Twhat) * c);                \
794         if (!reall) { return false; }                            \
795         memcpy(reall, s->mem, sizeof(Twhat) * s->mem##_count);   \
796         s->mem##_alloc = c;                                      \
797         mem_d(s->mem);                                           \
798         s->mem = reall;                                          \
799         return true;                                             \
800     }                                                            \
801     s->mem##_count = c;                                          \
802     if (c < (s->mem##_alloc / 2)) {                              \
803         reall = (Twhat*)mem_a(sizeof(Twhat) * c);                \
804         if (!reall) { return false; }                            \
805         memcpy(reall, s->mem, sizeof(Twhat) * c);                \
806         mem_d(s->mem);                                           \
807         s->mem = reall;                                          \
808     }                                                            \
809     return true;                                                 \
810 }
811
812 #define MEM_VEC_FUN_CLEAR(Tself, mem)   \
813 void Tself##_##mem##_clear(Tself *self) \
814 {                                       \
815     if (!self->mem)                     \
816         return;                         \
817     mem_d((void*) self->mem);           \
818     self->mem = NULL;                   \
819     self->mem##_count = 0;              \
820     self->mem##_alloc = 0;              \
821 }
822
823 #define MEM_VECTOR_CLEAR(owner, mem)  \
824     if ((owner)->mem)                 \
825         mem_d((void*)((owner)->mem)); \
826     (owner)->mem = NULL;              \
827     (owner)->mem##_count = 0;         \
828     (owner)->mem##_alloc = 0
829
830 #define MEM_VECTOR_INIT(owner, mem) \
831 {                                   \
832     (owner)->mem = NULL;            \
833     (owner)->mem##_count = 0;       \
834     (owner)->mem##_alloc = 0;       \
835 }
836
837 #define MEM_VEC_FUNCTIONS(Tself, Twhat, mem) \
838 MEM_VEC_FUN_REMOVE(Tself, Twhat, mem)        \
839 MEM_VEC_FUN_ADD(Tself, Twhat, mem)
840
841 #define MEM_VEC_FUNCTIONS_ALL(Tself, Twhat, mem) \
842 MEM_VEC_FUNCTIONS(Tself, Twhat, mem)             \
843 MEM_VEC_FUN_CLEAR(Tself, mem)                    \
844 MEM_VEC_FUN_FIND(Tself, Twhat, mem)
845
846 enum store_types {
847     store_global,
848     store_local,  /* local, assignable for now, should get promoted later */
849     store_param,  /* parameters, they are locals with a fixed position */
850     store_value,  /* unassignable */
851     store_return  /* unassignable, at OFS_RETURN */
852 };
853
854 typedef struct {
855     float x, y, z;
856 } vector;
857
858 /*
859  * A shallow copy of a lex_file to remember where which ast node
860  * came from.
861  */
862 typedef struct {
863     const char *file;
864     size_t      line;
865 } lex_ctx;
866
867 /*===================================================================*/
868 /*============================= exec.c ==============================*/
869 /*===================================================================*/
870
871 /* darkplaces has (or will have) a 64 bit prog loader
872  * where the 32 bit qc program is autoconverted on load.
873  * Since we may want to support that as well, let's redefine
874  * float and int here.
875  */
876 typedef union {
877     qcint   _int;
878     qcint    string;
879     qcint    function;
880     qcint    edict;
881     qcfloat _float;
882     qcfloat vector[3];
883     qcint   ivector[3];
884 } qcany;
885
886 typedef char qcfloat_size_is_correct [sizeof(qcfloat) == 4 ?1:-1];
887 typedef char qcint_size_is_correct   [sizeof(qcint)   == 4 ?1:-1];
888
889 enum {
890     VMERR_OK,
891     VMERR_TEMPSTRING_ALLOC,
892
893     VMERR_END
894 };
895
896 #define VM_JUMPS_DEFAULT 1000000
897
898 /* execute-flags */
899 #define VMXF_DEFAULT 0x0000     /* default flags - nothing */
900 #define VMXF_TRACE   0x0001     /* trace: print statements before executing */
901 #define VMXF_PROFILE 0x0002     /* profile: increment the profile counters */
902
903 struct qc_program_s;
904
905 typedef int (*prog_builtin)(struct qc_program_s *prog);
906
907 typedef struct {
908     qcint                  stmt;
909     size_t                 localsp;
910     prog_section_function *function;
911 } qc_exec_stack;
912
913 typedef struct qc_program_s {
914     char           *filename;
915
916     MEM_VECTOR_MAKE(prog_section_statement, code);
917     MEM_VECTOR_MAKE(prog_section_def,       defs);
918     MEM_VECTOR_MAKE(prog_section_def,       fields);
919     MEM_VECTOR_MAKE(prog_section_function,  functions);
920     MEM_VECTOR_MAKE(char,                   strings);
921     MEM_VECTOR_MAKE(qcint,                  globals);
922     MEM_VECTOR_MAKE(qcint,                  entitydata);
923     MEM_VECTOR_MAKE(bool,                   entitypool);
924
925     size_t tempstring_start;
926     size_t tempstring_at;
927
928     qcint  vmerror;
929
930     MEM_VECTOR_MAKE(size_t, profile);
931
932     MEM_VECTOR_MAKE(prog_builtin, builtins);
933
934     /* size_t ip; */
935     qcint  entities;
936     size_t entityfields;
937     bool   allowworldwrites;
938
939     MEM_VECTOR_MAKE(qcint,         localstack);
940     MEM_VECTOR_MAKE(qc_exec_stack, stack);
941     size_t statement;
942
943     int    argc; /* current arg count for debugging */
944 } qc_program;
945
946 qc_program* prog_load(const char *filename);
947 void        prog_delete(qc_program *prog);
948
949 bool prog_exec(qc_program *prog, prog_section_function *func, size_t flags, long maxjumps);
950
951 char*             prog_getstring (qc_program *prog, qcint str);
952 prog_section_def* prog_entfield  (qc_program *prog, qcint off);
953 prog_section_def* prog_getdef    (qc_program *prog, qcint off);
954 qcany*            prog_getedict  (qc_program *prog, qcint e);
955 qcint             prog_tempstring(qc_program *prog, const char *_str);
956
957 /*===================================================================*/
958 /*======================= main.c commandline ========================*/
959 /*===================================================================*/
960
961 #if 0
962 /* Helpers to allow for a whole lot of flags. Otherwise we'd limit
963  * to 32 or 64 -f options...
964  */
965 typedef struct {
966     size_t  idx; /* index into an array of 32 bit words */
967     uint8_t bit; /* index _into_ the 32 bit word, thus just uint8 */
968 } longbit;
969 #define LONGBIT(bit) { ((bit)/32), ((bit)%32) }
970 #else
971 typedef uint32_t longbit;
972 #define LONGBIT(bit) (bit)
973 #endif
974
975 /* Used to store the list of flags with names */
976 typedef struct {
977     const char *name;
978     longbit    bit;
979 } opts_flag_def;
980
981 /*===================================================================*/
982 /* list of -f flags, like -fdarkplaces-string-table-bug */
983 enum {
984 # define GMQCC_DEFINE_FLAG(X) X,
985 #  include "flags.def"
986 # undef GMQCC_DEFINE_FLAG
987     COUNT_FLAGS
988 };
989 static const opts_flag_def opts_flag_list[] = {
990 # define GMQCC_DEFINE_FLAG(X) { #X, LONGBIT(X) },
991 #  include "flags.def"
992 # undef GMQCC_DEFINE_FLAG
993     { NULL, LONGBIT(0) }
994 };
995
996 enum {
997 # define GMQCC_DEFINE_FLAG(X) WARN_##X,
998 #  include "warns.def"
999 # undef GMQCC_DEFINE_FLAG
1000     COUNT_WARNINGS
1001 };
1002 static const opts_flag_def opts_warn_list[] = {
1003 # define GMQCC_DEFINE_FLAG(X) { #X, LONGBIT(WARN_##X) },
1004 #  include "warns.def"
1005 # undef GMQCC_DEFINE_FLAG
1006     { NULL, LONGBIT(0) }
1007 };
1008
1009 /* other options: */
1010 enum {
1011     COMPILER_QCC,     /* circa  QuakeC */
1012     COMPILER_FTEQCC,  /* fteqcc QuakeC */
1013     COMPILER_QCCX,    /* qccx   QuakeC */
1014     COMPILER_GMQCC    /* this   QuakeC */
1015 };
1016 extern uint32_t    opts_O;      /* -Ox */
1017 extern const char *opts_output; /* -o file */
1018 extern int         opts_standard;
1019 extern bool        opts_debug;
1020 extern bool        opts_memchk;
1021
1022 /*===================================================================*/
1023 #define OPTS_FLAG(i) (!! (opts_flags[(i)/32] & (1<< ((i)%32))))
1024 extern uint32_t opts_flags[1 + (COUNT_FLAGS / 32)];
1025 #define OPTS_WARN(i) (!! (opts_warn[(i)/32] & (1<< ((i)%32))))
1026 extern uint32_t opts_warn[1 + (COUNT_WARNINGS / 32)];
1027
1028 #endif