]> de.git.xonotic.org Git - xonotic/gmqcc.git/blob - gmqcc.h
Use memcmp with strlen for correct_exists (it's a hell of a lot faster)
[xonotic/gmqcc.git] / gmqcc.h
1 /*
2  * Copyright (C) 2012, 2013
3  *     Dale Weiler
4  *     Wolfgang Bumiller
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy of
7  * this software and associated documentation files (the "Software"), to deal in
8  * the Software without restriction, including without limitation the rights to
9  * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
10  * of the Software, and to permit persons to whom the Software is furnished to do
11  * so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #ifndef GMQCC_HDR
25 #define GMQCC_HDR
26 #include <limits.h>
27 #include <stdlib.h>
28 #include <string.h>
29 #include <stdarg.h>
30 #include <stdio.h>
31 #include <ctype.h>
32
33 /*
34  * Disable some over protective warnings in visual studio because fixing them is a waste
35  * of my time.
36  */
37 #ifdef _MSC_VER
38 #   pragma warning(disable : 4244 ) /* conversion from 'int' to 'float', possible loss of data */
39 #   pragma warning(disable : 4018 ) /* signed/unsigned mismatch                                */
40 #endif
41
42 #define GMQCC_VERSION_MAJOR 0
43 #define GMQCC_VERSION_MINOR 3
44 #define GMQCC_VERSION_PATCH 0
45 #define GMQCC_VERSION_BUILD(J,N,P) (((J)<<16)|((N)<<8)|(P))
46 #define GMQCC_VERSION \
47     GMQCC_VERSION_BUILD(GMQCC_VERSION_MAJOR, GMQCC_VERSION_MINOR, GMQCC_VERSION_PATCH)
48
49 #ifndef GMQCC_GITINFO
50 # define GMQCC_GITINFO "(no git info)"
51 #endif
52
53 /*
54  * We cannot rely on C99 at all, since compilers like MSVC
55  * simply don't support it.  We define our own boolean type
56  * as a result (since we cannot include <stdbool.h>). For
57  * compilers that are in 1999 mode (C99 compliant) we can use
58  * the language keyword _Bool which can allow for better code
59  * on GCC and GCC-like compilers, opposed to `int`.
60  */
61 #ifndef __cplusplus
62 #   ifdef  false
63 #       undef  false
64 #   endif /* !false */
65 #   ifdef  true
66 #       undef true
67 #   endif /* !true  */
68 #   define false (0)
69 #   define true  (1)
70 #   ifdef __STDC_VERSION__
71 #       if __STDC_VERSION__ < 199901L && __GNUC__ < 3
72             typedef int  bool;
73 #       else
74             typedef _Bool bool;
75 #       endif
76 #   else
77         typedef int bool;
78 #   endif /* !__STDC_VERSION__ */
79 #endif    /* !__cplusplus      */
80
81 /*
82  * Of some functions which are generated we want to make sure
83  * that the result isn't ignored. To find such function calls,
84  * we use this macro.
85  */
86 #if defined(__GNUC__) || defined(__CLANG__)
87 #   define GMQCC_WARN __attribute__((warn_unused_result))
88 #else
89 #   define GMQCC_WARN
90 #endif
91 /*
92  * This is a hack to silent clang regarding empty
93  * body if statements.
94  */
95 #define GMQCC_SUPPRESS_EMPTY_BODY do { } while (0)
96
97 /*
98  * Inline is not supported in < C90, however some compilers
99  * like gcc and clang might have an inline attribute we can
100  * use if present.
101  */
102 #ifdef __STDC_VERSION__
103 #    if __STDC_VERSION__ < 199901L
104 #       if defined(__GNUC__) || defined (__CLANG__)
105 #           if __GNUC__ < 2
106 #               define GMQCC_INLINE
107 #           else
108 #               define GMQCC_INLINE __attribute__ ((always_inline))
109 #           endif
110 #       else
111 #           define GMQCC_INLINE
112 #       endif
113 #    else
114 #       define GMQCC_INLINE inline
115 #    endif
116 /*
117  * Visual studio has __forcinline we can use.  So lets use that
118  * I suspect it also has just __inline of some sort, but our use
119  * of inline is correct (not guessed), WE WANT IT TO BE INLINE
120  */
121 #elif defined(_MSC_VER)
122 #    define GMQCC_INLINE __forceinline
123 #else
124 #    define GMQCC_INLINE
125 #endif /* !__STDC_VERSION__ */
126
127 /*
128  * noreturn is present in GCC and clang
129  * it's required for _ast_node_destory otherwise -Wmissing-noreturn
130  * in clang complains about there being no return since abort() is
131  * called.
132  */
133 #if (defined(__GNUC__) && __GNUC__ >= 2) || defined(__CLANG__)
134 #    define GMQCC_NORETURN __attribute__ ((noreturn))
135 #else
136 #    define GMQCC_NORETURN
137 #endif
138
139 #ifndef _MSC_VER
140 #   include <stdint.h>
141 #else
142     typedef unsigned __int8  uint8_t;
143     typedef unsigned __int16 uint16_t;
144     typedef unsigned __int32 uint32_t;
145     typedef unsigned __int64 uint64_t;
146
147     typedef __int16          int16_t;
148     typedef __int32          int32_t;
149     typedef __int64          int64_t;
150 #endif
151
152 /* 
153  *windows makes these prefixed because they're C99
154  * TODO: utility versions that are type-safe and not
155  * just plain textual subsitution.
156  */
157 #ifdef _MSC_VER
158 #    define snprintf(X, Y, Z, ...) _snprintf(X, Y, Z, __VA_ARGS__)
159     /* strtof doesn't exist -> strtod does though :) */
160 #    define strtof(X, Y)          (float)(strtod(X, Y))
161 #endif
162
163 /*
164  * Very roboust way at determining endianess at compile time: this handles
165  * almost every possible situation.  Otherwise a runtime check has to be
166  * performed.
167  */
168 #define GMQCC_BYTE_ORDER_LITTLE 1234
169 #define GMQCC_BYTE_ORDER_BIG    4321
170
171 #if defined (__GNUC__) || defined (__GNU_LIBRARY__)
172 #   if defined (__FreeBSD__) || defined (__OpenBSD__)
173 #       include <sys/endian.h>
174 #   elif defined (BSD) && (BSD >= 199103) || defined (__DJGPP__) || defined (__CYGWIN32__)
175 #       include <machine/endian.h>
176 #   elif defined (__APPLE__)
177 #       if defined (__BIG_ENDIAN__) && !defined(BIG_ENDIAN)
178 #           define BIG_ENDIAN
179 #       elif defined (__LITTLE_ENDIAN__) && !defined (LITTLE_ENDIAN)
180 #           define LITTLE_ENDIAN
181 #       endif
182 #   elif !defined (__MINGW32__)
183 #       include <endian.h>
184 #       if !defined (__BEOS__)
185 #           include <byteswap.h>
186 #       endif
187 #   endif
188 #endif
189 #if !defined(PLATFORM_BYTE_ORDER)
190 #   if defined (LITTLE_ENDIAN) || defined (BIG_ENDIAN)
191 #       if defined (LITTLE_ENDIAN) && !defined(BIG_ENDIAN)
192 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
193 #       elif !defined (LITTLE_ENDIAN) && defined (BIG_ENDIAN)
194 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
195 #       elif defined (BYTE_ORDER) && (BYTE_ORDER == LITTLE_ENDIAN)
196 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
197 #       elif defined (BYTE_ORDER) && (BYTE_ORDER == BIG_ENDIAN)
198 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
199 #       endif
200 #   elif defined (_LITTLE_ENDIAN) || defined (_BIG_ENDIAN)
201 #       if defined (_LITTLE_ENDIAN) && !defined(_BIG_ENDIAN)
202 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
203 #       elif !defined (_LITTLE_ENDIAN) && defined (_BIG_ENDIAN)
204 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
205 #       elif defined (_BYTE_ORDER) && (_BYTE_ORDER == _LITTLE_ENDIAN)
206 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
207 #       elif defined (_BYTE_ORDER) && (_BYTE_ORDER == _BIG_ENDIAN)
208 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
209 #       endif
210 #   elif defined (__LITTLE_ENDIAN__) || defined (__BIG_ENDIAN__)
211 #       if defined (__LITTLE_ENDIAN__) && !defined (__BIG_ENDIAN__)
212 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
213 #       elif !defined (__LITTLE_ENDIAN__) && defined (__BIG_ENDIAN__)
214 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
215 #       elif defined (__BYTE_ORDER__) && (__BYTE_ORDER__ == __LITTLE_ENDIAN__)
216 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
217 #       elif defined (__BYTE_ORDER__) && (__BYTE_ORDER__ == __BIG_ENDIAN__)
218 #           define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
219 #       endif
220 #   endif
221 #endif
222 #if !defined (PLATFORM_BYTE_ORDER)
223 #   if   defined (__alpha__) || defined (__alpha)    || defined (i386)       || \
224          defined (__i386__)  || defined (_M_I86)     || defined (_M_IX86)    || \
225          defined (__OS2__)   || defined (sun386)     || defined (__TURBOC__) || \
226          defined (vax)       || defined (vms)        || defined (VMS)        || \
227          defined (__VMS)     || defined (__x86_64__) || defined (_M_IA64)    || \
228          defined (_M_X64)    || defined (__i386)     || defined (__x86_64)
229 #       define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_LITTLE
230 #   elif defined (AMIGA)     || defined (applec)     || defined (__AS400__)  || \
231          defined (_CRAY)     || defined (__hppa)     || defined (__hp9000)   || \
232          defined (ibm370)    || defined (mc68000)    || defined (m68k)       || \
233          defined (__MRC__)   || defined (__MVS__)    || defined (__MWERKS__) || \
234          defined (sparc)     || defined (__sparc)    || defined (SYMANTEC_C) || \
235          defined (__TANDEM)  || defined (THINK_C)    || defined (__VMCMS__)  || \
236          defined (__PPC__)   || defined (__PPC)      || defined (PPC)
237 #       define PLATFORM_BYTE_ORDER GMQCC_BYTE_ORDER_BIG
238 #   else
239 #       define PLATFORM_BYTE_ORDER -1
240 #   endif
241 #endif
242
243
244
245 /*===================================================================*/
246 /*=========================== util.c ================================*/
247 /*===================================================================*/
248 void *util_memory_a      (size_t, /*****/ unsigned int, const char *);
249 void *util_memory_r      (void *, size_t, unsigned int, const char *);
250 void  util_memory_d      (void *);
251 void  util_meminfo       ();
252
253 bool  util_filexists     (const char *);
254 bool  util_strupper      (const char *);
255 bool  util_strdigit      (const char *);
256 char *util_strdup        (const char *);
257 void  util_debug         (const char *, const char *, ...);
258 void  util_endianswap    (void *,  size_t, unsigned int);
259
260 size_t util_strtocmd    (const char *, char *, size_t);
261 size_t util_strtononcmd (const char *, char *, size_t);
262
263 uint16_t util_crc16(uint16_t crc, const char *data, size_t len);
264
265 void     util_seed(uint32_t);
266 uint32_t util_rand();
267
268 int util_vasprintf(char **ret, const char *fmt, va_list);
269 int util_asprintf (char **ret, const char *fmt, ...);
270
271
272 #ifdef NOTRACK
273 #    define mem_a(x)    malloc (x)
274 #    define mem_d(x)    free   ((void*)x)
275 #    define mem_r(x, n) realloc((void*)x, n)
276 #else
277 #    define mem_a(x)    util_memory_a((x), __LINE__, __FILE__)
278 #    define mem_d(x)    util_memory_d((void*)(x))
279 #    define mem_r(x, n) util_memory_r((void*)(x), (n), __LINE__, __FILE__)
280 #endif
281
282 /*
283  * A flexible vector implementation: all vector pointers contain some
284  * data about themselfs exactly - sizeof(vector_t) behind the pointer
285  * this data is represented in the structure below.  Doing this allows
286  * us to use the array [] to access individual elements from the vector
287  * opposed to using set/get methods.
288  */     
289 typedef struct {
290     size_t  allocated;
291     size_t  used;
292
293     /* can be extended now! whoot */
294 } vector_t;
295
296 /* hidden interface */
297 void _util_vec_grow(void **a, size_t i, size_t s);
298 #define GMQCC_VEC_WILLGROW(X,Y) ( \
299     ((!(X) || vec_meta(X)->used + Y >= vec_meta(X)->allocated)) ? \
300         (void)_util_vec_grow(((void**)&(X)), (Y), sizeof(*(X))) : \
301         (void)0                                                   \
302 )
303
304 /* exposed interface */
305 #define vec_meta(A)       (((vector_t*)(A)) - 1)
306 #define vec_free(A)       ((A) ? (mem_d((void*)vec_meta(A)), (A) = NULL) : 0)
307 #define vec_push(A,V)     (GMQCC_VEC_WILLGROW((A),1), (A)[vec_meta(A)->used++] = (V))
308 #define vec_size(A)       ((A) ? vec_meta(A)->used : 0)
309 #define vec_add(A,N)      (GMQCC_VEC_WILLGROW((A),(N)), vec_meta(A)->used += (N), &(A)[vec_meta(A)->used-(N)])
310 #define vec_last(A)       ((A)[vec_meta(A)->used - 1])
311 #define vec_pop(A)        (vec_meta(A)->used -= 1)
312 #define vec_shrinkto(A,N) (vec_meta(A)->used  = (N))
313 #define vec_shrinkby(A,N) (vec_meta(A)->used -= (N))
314 #define vec_append(A,N,S) memcpy(vec_add((A), (N)), (S), (N) * sizeof(*(S)))
315 #define vec_upload(X,Y,S) memcpy(vec_add((X), (S) * sizeof(*(Y))), (Y), (S) * sizeof(*(Y)))
316 #define vec_remove(A,I,N) memmove((A)+(I),(A)+((I)+(N)),sizeof(*(A))*(vec_meta(A)->used-(I)-(N))),vec_meta(A)->used-=(N)
317
318 typedef struct trie_s {
319     struct trie_s *entries;
320     unsigned char  ch;
321     void          *value;
322 } correct_trie_t;
323
324 correct_trie_t* correct_trie_new();
325
326 typedef struct hash_table_t {
327     size_t                size;
328     struct hash_node_t **table;
329 } hash_table_t, *ht;
330
331 typedef struct hash_set_t {
332     size_t  bits;
333     size_t  mask;
334     size_t  capacity;
335     size_t *items;
336     size_t  total;
337 } hash_set_t, *hs;
338
339 /*
340  * hashtable implementation:
341  *
342  * Note:
343  *      This was designed for pointers:  you manage the life of the object yourself
344  *      if you do use this for non-pointers please be warned that the object may not
345  *      be valid if the duration of it exceeds (i.e on stack).  So you need to allocate
346  *      yourself, or put those in global scope to ensure duration is for the whole
347  *      runtime.
348  *
349  * util_htnew(size)                             -- to make a new hashtable
350  * util_htset(table, key, value, sizeof(value)) -- to set something in the table
351  * util_htget(table, key)                       -- to get something from the table
352  * util_htdel(table)                            -- to delete the table
353  *
354  * example of use:
355  *
356  * ht    foo  = util_htnew(1024);
357  * int   data = 100;
358  * char *test = "hello world\n";
359  * util_htset(foo, "foo", (void*)&data);
360  * util_gtset(foo, "bar", (void*)test);
361  *
362  * printf("foo: %d, bar %s",
363  *     *((int *)util_htget(foo, "foo")),
364  *      ((char*)util_htget(foo, "bar"))
365  * );
366  *
367  * util_htdel(foo);
368  */
369 hash_table_t *util_htnew (size_t size);
370 void          util_htset (hash_table_t *ht, const char *key, void *value);
371 void          util_htdel (hash_table_t *ht);
372 size_t        util_hthash(hash_table_t *ht, const char *key);
373 void          util_htseth(hash_table_t *ht, const char *key, size_t hash, void *value);
374
375 void         *util_htget (hash_table_t *ht, const char *key);
376 void         *util_htgeth(hash_table_t *ht, const char *key, size_t hash);
377
378 /*
379  * hashset implementation:
380  *      This was designed for pointers:  you manage the life of the object yourself
381  *      if you do use this for non-pointers please be warned that the object may not
382  *      be valid if the duration of it exceeds (i.e on stack).  So you need to allocate
383  *      yourself, or put those in global scope to ensure duration is for the whole
384  *      runtime.
385  *
386  * util_hsnew()                             -- to make a new hashset
387  * util_hsadd(set, key)                     -- to add something in the set
388  * util_hshas(set, key)                     -- to check if something is in the set
389  * util_hsrem(set, key)                     -- to remove something in the set
390  * util_hsdel(set)                          -- to delete the set
391  *
392  * example of use:
393  * 
394  * hs    foo = util_hsnew();
395  * char *bar = "hello blub\n";
396  * char *baz = "hello dale\n";
397  *
398  * util_hsadd(foo, bar);
399  * util_hsadd(foo, baz);
400  * util_hsrem(foo, baz);
401  *
402  * printf("bar %d | baz %d\n",
403  *     util_hshas(foo, bar),
404  *     util_hshad(foo, baz)
405  * );
406  *
407  * util_hsdel(foo);  
408  */
409
410 hash_set_t *util_hsnew(void);
411 int         util_hsadd(hash_set_t *, void *);
412 int         util_hshas(hash_set_t *, void *);
413 int         util_hsrem(hash_set_t *, void *);
414 void        util_hsdel(hash_set_t *);
415  
416 /*===================================================================*/
417 /*============================ file.c ===============================*/
418 /*===================================================================*/
419 GMQCC_INLINE void    file_close  (FILE *);
420 GMQCC_INLINE int     file_error  (FILE *);
421 GMQCC_INLINE int     file_getc   (FILE *);
422 GMQCC_INLINE int     file_printf (FILE *, const char *, ...);
423 GMQCC_INLINE int     file_puts   (FILE *, const char *);
424 GMQCC_INLINE int     file_putc   (FILE *, int);
425 GMQCC_INLINE int     file_seek   (FILE *, long int, int);
426
427 GMQCC_INLINE size_t  file_read   (void *,        size_t, size_t, FILE *);
428 GMQCC_INLINE size_t  file_write  (const void *,  size_t, size_t, FILE *);
429
430 GMQCC_INLINE FILE   *file_open   (const char *, const char *);
431 /*NOINLINE*/ int     file_getline(char  **, size_t *, FILE *);
432
433
434 /*===================================================================*/
435 /*=========================== correct.c =============================*/
436 /*===================================================================*/
437 void  correct_del(correct_trie_t*, size_t **);
438 void  correct_add(correct_trie_t*, size_t ***, const char *);
439 char *correct_str(correct_trie_t*, /********/  const char *);
440
441 /*===================================================================*/
442 /*=========================== code.c ================================*/
443 /*===================================================================*/
444
445 /* TODO: cleanup */
446 /* Note: if you change the order, fix type_sizeof in ir.c */
447 enum {
448     TYPE_VOID     ,
449     TYPE_STRING   ,
450     TYPE_FLOAT    ,
451     TYPE_VECTOR   ,
452     TYPE_ENTITY   ,
453     TYPE_FIELD    ,
454     TYPE_FUNCTION ,
455     TYPE_POINTER  ,
456     TYPE_INTEGER  ,
457     TYPE_VARIANT  ,
458     TYPE_STRUCT   ,
459     TYPE_UNION    ,
460     TYPE_ARRAY    ,
461
462     TYPE_NIL      , /* it's its own type / untyped */
463     TYPE_NOEXPR   , /* simply invalid in expressions */
464
465     TYPE_COUNT
466 };
467
468 /* const/var qualifiers */
469 #define CV_NONE   0
470 #define CV_CONST  1
471 #define CV_VAR   -1
472 #define CV_WRONG  0x8000 /* magic number to help parsing */
473
474 extern const char *type_name        [TYPE_COUNT];
475 extern uint16_t    type_store_instr [TYPE_COUNT];
476 extern uint16_t    field_store_instr[TYPE_COUNT];
477
478 /*
479  * could use type_store_instr + INSTR_STOREP_F - INSTR_STORE_F
480  * but this breaks when TYPE_INTEGER is added, since with the enhanced
481  * instruction set, the old ones are left untouched, thus the _I instructions
482  * are at a seperate place.
483  */
484 extern uint16_t type_storep_instr[TYPE_COUNT];
485 extern uint16_t type_eq_instr    [TYPE_COUNT];
486 extern uint16_t type_ne_instr    [TYPE_COUNT];
487 extern uint16_t type_not_instr   [TYPE_COUNT];
488
489 typedef struct {
490     uint32_t offset;      /* Offset in file of where data begins  */
491     uint32_t length;      /* Length of section (how many of)      */
492 } prog_section;
493
494 typedef struct {
495     uint32_t     version;      /* Program version (6)     */
496     uint16_t     crc16;
497     uint16_t     skip;
498
499     prog_section statements;   /* prog_section_statement  */
500     prog_section defs;         /* prog_section_def        */
501     prog_section fields;       /* prog_section_field      */
502     prog_section functions;    /* prog_section_function   */
503     prog_section strings;
504     prog_section globals;
505     uint32_t     entfield;     /* Number of entity fields */
506 } prog_header;
507
508 /*
509  * Each paramater incerements by 3 since vector types hold
510  * 3 components (x,y,z).
511  */
512 #define OFS_NULL      0
513 #define OFS_RETURN    1
514 #define OFS_PARM0     (OFS_RETURN+3)
515 #define OFS_PARM1     (OFS_PARM0 +3)
516 #define OFS_PARM2     (OFS_PARM1 +3)
517 #define OFS_PARM3     (OFS_PARM2 +3)
518 #define OFS_PARM4     (OFS_PARM3 +3)
519 #define OFS_PARM5     (OFS_PARM4 +3)
520 #define OFS_PARM6     (OFS_PARM5 +3)
521 #define OFS_PARM7     (OFS_PARM6 +3)
522
523 typedef struct {
524     uint16_t opcode;
525
526     /* operand 1 */
527     union {
528         int16_t  s1; /* signed   */
529         uint16_t u1; /* unsigned */
530     } o1;
531     /* operand 2 */
532     union {
533         int16_t  s1; /* signed   */
534         uint16_t u1; /* unsigned */
535     } o2;
536     /* operand 3 */
537     union {
538         int16_t  s1; /* signed   */
539         uint16_t u1; /* unsigned */
540     } o3;
541
542     /*
543      * This is the same as the structure in darkplaces
544      * {
545      *     unsigned short op;
546      *     short          a,b,c;
547      * }
548      * But this one is more sane to work with, and the
549      * type sizes are guranteed.
550      */
551 } prog_section_statement;
552
553 typedef struct {
554     /*
555      * The types:
556      * 0 = ev_void
557      * 1 = ev_string
558      * 2 = ev_float
559      * 3 = ev_vector
560      * 4 = ev_entity
561      * 5 = ev_field
562      * 6 = ev_function
563      * 7 = ev_pointer -- engine only
564      * 8 = ev_bad     -- engine only
565      */
566     uint16_t type;
567     uint16_t offset;
568     uint32_t name;
569 } prog_section_both;
570
571 typedef prog_section_both prog_section_def;
572 typedef prog_section_both prog_section_field;
573
574 /* this is ORed to the type */
575 #define DEF_SAVEGLOBAL (1<<15)
576 #define DEF_TYPEMASK   ((1<<15)-1)
577
578 typedef struct {
579     int32_t   entry;      /* in statement table for instructions  */
580     uint32_t  firstlocal; /* First local in local table           */
581     uint32_t  locals;     /* Total ints of params + locals        */
582     uint32_t  profile;    /* Always zero (engine uses this)       */
583     uint32_t  name;       /* name of function in string table     */
584     uint32_t  file;       /* file of the source file              */
585     int32_t   nargs;      /* number of arguments                  */
586     uint8_t   argsize[8]; /* size of arguments (keep 8 always?)   */
587 } prog_section_function;
588
589 /*
590  * Instructions
591  * These are the external instructions supported by the interperter
592  * this is what things compile to (from the C code).
593  */
594 enum {
595     INSTR_DONE,
596     INSTR_MUL_F,
597     INSTR_MUL_V,
598     INSTR_MUL_FV, /* NOTE: the float operands must NOT be at the same locations: A != C */
599     INSTR_MUL_VF, /* and here: B != C */
600     INSTR_DIV_F,
601     INSTR_ADD_F,
602     INSTR_ADD_V,
603     INSTR_SUB_F,
604     INSTR_SUB_V,
605     INSTR_EQ_F,
606     INSTR_EQ_V,
607     INSTR_EQ_S,
608     INSTR_EQ_E,
609     INSTR_EQ_FNC,
610     INSTR_NE_F,
611     INSTR_NE_V,
612     INSTR_NE_S,
613     INSTR_NE_E,
614     INSTR_NE_FNC,
615     INSTR_LE,
616     INSTR_GE,
617     INSTR_LT,
618     INSTR_GT,
619     INSTR_LOAD_F,
620     INSTR_LOAD_V,
621     INSTR_LOAD_S,
622     INSTR_LOAD_ENT,
623     INSTR_LOAD_FLD,
624     INSTR_LOAD_FNC,
625     INSTR_ADDRESS,
626     INSTR_STORE_F,
627     INSTR_STORE_V,
628     INSTR_STORE_S,
629     INSTR_STORE_ENT,
630     INSTR_STORE_FLD,
631     INSTR_STORE_FNC,
632     INSTR_STOREP_F,
633     INSTR_STOREP_V,
634     INSTR_STOREP_S,
635     INSTR_STOREP_ENT,
636     INSTR_STOREP_FLD,
637     INSTR_STOREP_FNC,
638     INSTR_RETURN,
639     INSTR_NOT_F,
640     INSTR_NOT_V,
641     INSTR_NOT_S,
642     INSTR_NOT_ENT,
643     INSTR_NOT_FNC,
644     INSTR_IF,
645     INSTR_IFNOT,
646     INSTR_CALL0,
647     INSTR_CALL1,
648     INSTR_CALL2,
649     INSTR_CALL3,
650     INSTR_CALL4,
651     INSTR_CALL5,
652     INSTR_CALL6,
653     INSTR_CALL7,
654     INSTR_CALL8,
655     INSTR_STATE,
656     INSTR_GOTO,
657     INSTR_AND,
658     INSTR_OR,
659     INSTR_BITAND,
660     INSTR_BITOR,
661
662     /*
663      * Virtual instructions used by the assembler
664      * keep at the end but before virtual instructions
665      * for the IR below.
666      */
667     AINSTR_END,
668
669     /*
670      * Virtual instructions used by the IR
671      * Keep at the end!
672      */
673     VINSTR_PHI,
674     VINSTR_JUMP,
675     VINSTR_COND,
676     /* A never returning CALL.
677      * Creating this causes IR blocks to be marked as 'final'.
678      * No-Return-Call
679      */
680     VINSTR_NRCALL
681 };
682
683 /* TODO: cleanup this mess */
684 extern prog_section_statement *code_statements;
685 extern int                    *code_linenums;
686 extern prog_section_def       *code_defs;
687 extern prog_section_field     *code_fields;
688 extern prog_section_function  *code_functions;
689 extern int                    *code_globals;
690 extern char                   *code_chars;
691 extern uint16_t code_crc;
692
693 /* uhh? */
694 typedef float   qcfloat;
695 typedef int32_t qcint;
696
697 /*
698  * code_write -- writes out the compiled file
699  * code_init  -- prepares the code file
700  */
701 bool     code_write       (const char *filename, const char *lno);
702 void     code_init        ();
703 uint32_t code_genstring   (const char *string);
704 qcint    code_alloc_field (size_t qcsize);
705
706 /* this function is used to keep statements and linenumbers together */
707 void     code_push_statement(prog_section_statement *stmt, int linenum);
708 void     code_pop_statement();
709
710 /*
711  * A shallow copy of a lex_file to remember where which ast node
712  * came from.
713  */
714 typedef struct {
715     const char *file;
716     size_t      line;
717 } lex_ctx;
718
719 /*===================================================================*/
720 /*============================ con.c ================================*/
721 /*===================================================================*/
722 enum {
723     CON_BLACK   = 30,
724     CON_RED,
725     CON_GREEN,
726     CON_BROWN,
727     CON_BLUE,
728     CON_MAGENTA,
729     CON_CYAN ,
730     CON_WHITE
731 };
732
733 /* message level */
734 enum {
735     LVL_MSG,
736     LVL_WARNING,
737     LVL_ERROR
738 };
739
740 FILE *con_default_out();
741 FILE *con_default_err();
742
743 void con_vprintmsg (int level, const char *name, size_t line, const char *msgtype, const char *msg, va_list ap);
744 void con_printmsg  (int level, const char *name, size_t line, const char *msgtype, const char *msg, ...);
745 void con_cvprintmsg(void *ctx, int lvl, const char *msgtype, const char *msg, va_list ap);
746 void con_cprintmsg (void *ctx, int lvl, const char *msgtype, const char *msg, ...);
747
748 void con_close ();
749 void con_init  ();
750 void con_reset ();
751 void con_color (int);
752 int  con_change(const char *, const char *);
753 int  con_verr  (const char *, va_list);
754 int  con_vout  (const char *, va_list);
755 int  con_err   (const char *, ...);
756 int  con_out   (const char *, ...);
757
758 /* error/warning interface */
759 extern size_t compile_errors;
760 extern size_t compile_Werrors;
761 extern size_t compile_warnings;
762
763 void /********/ compile_error   (lex_ctx ctx, /*LVL_ERROR*/ const char *msg, ...);
764 void /********/ vcompile_error  (lex_ctx ctx, /*LVL_ERROR*/ const char *msg, va_list ap);
765 bool GMQCC_WARN compile_warning (lex_ctx ctx, int warntype, const char *fmt, ...);
766 bool GMQCC_WARN vcompile_warning(lex_ctx ctx, int warntype, const char *fmt, va_list ap);
767 void            compile_show_werrors();
768
769 /*===================================================================*/
770 /*========================= assembler.c =============================*/
771 /*===================================================================*/
772 /* TODO: remove this ... */
773 static const struct {
774     const char  *m; /* menomic     */
775     const size_t o; /* operands    */
776     const size_t l; /* menomic len */
777 } asm_instr[] = {
778     { "DONE"      , 1, 4 },
779     { "MUL_F"     , 3, 5 },
780     { "MUL_V"     , 3, 5 },
781     { "MUL_FV"    , 3, 6 },
782     { "MUL_VF"    , 3, 6 },
783     { "DIV"       , 0, 3 },
784     { "ADD_F"     , 3, 5 },
785     { "ADD_V"     , 3, 5 },
786     { "SUB_F"     , 3, 5 },
787     { "SUB_V"     , 3, 5 },
788     { "EQ_F"      , 0, 4 },
789     { "EQ_V"      , 0, 4 },
790     { "EQ_S"      , 0, 4 },
791     { "EQ_E"      , 0, 4 },
792     { "EQ_FNC"    , 0, 6 },
793     { "NE_F"      , 0, 4 },
794     { "NE_V"      , 0, 4 },
795     { "NE_S"      , 0, 4 },
796     { "NE_E"      , 0, 4 },
797     { "NE_FNC"    , 0, 6 },
798     { "LE"        , 0, 2 },
799     { "GE"        , 0, 2 },
800     { "LT"        , 0, 2 },
801     { "GT"        , 0, 2 },
802     { "FIELD_F"   , 0, 7 },
803     { "FIELD_V"   , 0, 7 },
804     { "FIELD_S"   , 0, 7 },
805     { "FIELD_ENT" , 0, 9 },
806     { "FIELD_FLD" , 0, 9 },
807     { "FIELD_FNC" , 0, 9 },
808     { "ADDRESS"   , 0, 7 },
809     { "STORE_F"   , 0, 7 },
810     { "STORE_V"   , 0, 7 },
811     { "STORE_S"   , 0, 7 },
812     { "STORE_ENT" , 0, 9 },
813     { "STORE_FLD" , 0, 9 },
814     { "STORE_FNC" , 0, 9 },
815     { "STOREP_F"  , 0, 8 },
816     { "STOREP_V"  , 0, 8 },
817     { "STOREP_S"  , 0, 8 },
818     { "STOREP_ENT", 0, 10},
819     { "STOREP_FLD", 0, 10},
820     { "STOREP_FNC", 0, 10},
821     { "RETURN"    , 0, 6 },
822     { "NOT_F"     , 0, 5 },
823     { "NOT_V"     , 0, 5 },
824     { "NOT_S"     , 0, 5 },
825     { "NOT_ENT"   , 0, 7 },
826     { "NOT_FNC"   , 0, 7 },
827     { "IF"        , 0, 2 },
828     { "IFNOT"     , 0, 5 },
829     { "CALL0"     , 1, 5 },
830     { "CALL1"     , 2, 5 },
831     { "CALL2"     , 3, 5 },
832     { "CALL3"     , 4, 5 },
833     { "CALL4"     , 5, 5 },
834     { "CALL5"     , 6, 5 },
835     { "CALL6"     , 7, 5 },
836     { "CALL7"     , 8, 5 },
837     { "CALL8"     , 9, 5 },
838     { "STATE"     , 0, 5 },
839     { "GOTO"      , 0, 4 },
840     { "AND"       , 0, 3 },
841     { "OR"        , 0, 2 },
842     { "BITAND"    , 0, 6 },
843     { "BITOR"     , 0, 5 },
844
845     { "END"       , 0, 3 } /* virtual assembler instruction */
846 };
847 /*===================================================================*/
848 /*============================= ir.c ================================*/
849 /*===================================================================*/
850
851 enum store_types {
852     store_global,
853     store_local,  /* local, assignable for now, should get promoted later */
854     store_param,  /* parameters, they are locals with a fixed position */
855     store_value,  /* unassignable */
856     store_return  /* unassignable, at OFS_RETURN */
857 };
858
859 typedef struct {
860     qcfloat x, y, z;
861 } vector;
862
863 vector  vec3_add  (vector, vector);
864 vector  vec3_sub  (vector, vector);
865 qcfloat vec3_mulvv(vector, vector);
866 vector  vec3_mulvf(vector, float);
867
868 /*===================================================================*/
869 /*============================= exec.c ==============================*/
870 /*===================================================================*/
871
872 /* TODO: cleanup */
873 /*
874  * Darkplaces has (or will have) a 64 bit prog loader
875  * where the 32 bit qc program is autoconverted on load.
876  * Since we may want to support that as well, let's redefine
877  * float and int here.
878  */
879 typedef union {
880     qcint   _int;
881     qcint    string;
882     qcint    function;
883     qcint    edict;
884     qcfloat _float;
885     qcfloat vector[3];
886     qcint   ivector[3];
887 } qcany;
888
889 typedef char qcfloat_size_is_correct [sizeof(qcfloat) == 4 ?1:-1];
890 typedef char qcint_size_is_correct   [sizeof(qcint)   == 4 ?1:-1];
891
892 enum {
893     VMERR_OK,
894     VMERR_TEMPSTRING_ALLOC,
895
896     VMERR_END
897 };
898
899 #define VM_JUMPS_DEFAULT 1000000
900
901 /* execute-flags */
902 #define VMXF_DEFAULT 0x0000     /* default flags - nothing */
903 #define VMXF_TRACE   0x0001     /* trace: print statements before executing */
904 #define VMXF_PROFILE 0x0002     /* profile: increment the profile counters */
905
906 struct qc_program_s;
907
908 typedef int (*prog_builtin)(struct qc_program_s *prog);
909
910 typedef struct {
911     qcint                  stmt;
912     size_t                 localsp;
913     prog_section_function *function;
914 } qc_exec_stack;
915
916 typedef struct qc_program_s {
917     char           *filename;
918
919     prog_section_statement *code;
920     prog_section_def       *defs;
921     prog_section_def       *fields;
922     prog_section_function  *functions;
923     char                   *strings;
924     qcint                  *globals;
925     qcint                  *entitydata;
926     bool                   *entitypool;
927
928     const char*            *function_stack;
929
930     uint16_t crc16;
931
932     size_t tempstring_start;
933     size_t tempstring_at;
934
935     qcint  vmerror;
936
937     size_t *profile;
938
939     prog_builtin *builtins;
940     size_t        builtins_count;
941
942     /* size_t ip; */
943     qcint  entities;
944     size_t entityfields;
945     bool   allowworldwrites;
946
947     qcint         *localstack;
948     qc_exec_stack *stack;
949     size_t statement;
950
951     size_t xflags;
952
953     int    argc; /* current arg count for debugging */
954 } qc_program;
955
956 qc_program* prog_load(const char *filename);
957 void        prog_delete(qc_program *prog);
958
959 bool prog_exec(qc_program *prog, prog_section_function *func, size_t flags, long maxjumps);
960
961 char*             prog_getstring (qc_program *prog, qcint str);
962 prog_section_def* prog_entfield  (qc_program *prog, qcint off);
963 prog_section_def* prog_getdef    (qc_program *prog, qcint off);
964 qcany*            prog_getedict  (qc_program *prog, qcint e);
965 qcint             prog_tempstring(qc_program *prog, const char *_str);
966
967
968 /*===================================================================*/
969 /*===================== parser.c commandline ========================*/
970 /*===================================================================*/
971
972 bool parser_init          ();
973 bool parser_compile_file  (const char *);
974 bool parser_compile_string(const char *, const char *, size_t);
975 bool parser_finish        (const char *);
976 void parser_cleanup       ();
977
978 /*===================================================================*/
979 /*====================== ftepp.c commandline ========================*/
980 /*===================================================================*/
981 struct lex_file_s;
982 typedef struct {
983     const char  *name;
984     char      *(*func)(struct lex_file_s *);
985 } ftepp_predef_t;
986
987 /*
988  * line, file, counter, counter_last, random, random_last, date, time
989  * increment when items are added
990  */
991 #define FTEPP_PREDEF_COUNT 8
992
993 bool        ftepp_init             ();
994 bool        ftepp_preprocess_file  (const char *filename);
995 bool        ftepp_preprocess_string(const char *name, const char *str);
996 void        ftepp_finish           ();
997 const char *ftepp_get              ();
998 void        ftepp_flush            ();
999 void        ftepp_add_define       (const char *source, const char *name);
1000 void        ftepp_add_macro        (const char *name,   const char *value);
1001
1002 extern const ftepp_predef_t ftepp_predefs[FTEPP_PREDEF_COUNT];
1003
1004 /*===================================================================*/
1005 /*======================= main.c commandline ========================*/
1006 /*===================================================================*/
1007
1008 #if 1
1009 /* Helpers to allow for a whole lot of flags. Otherwise we'd limit
1010  * to 32 or 64 -f options...
1011  */
1012 typedef struct {
1013     size_t  idx; /* index into an array of 32 bit words */
1014     uint8_t bit; /* bit index for the 8 bit group idx points to */
1015 } longbit;
1016 #define LONGBIT(bit) { ((bit)/32), ((bit)%32) }
1017 #define LONGBIT_SET(B, I) ((B).idx = (I)/32, (B).bit = ((I)%32))
1018 #else
1019 typedef uint32_t longbit;
1020 #define LONGBIT(bit) (bit)
1021 #define LONGBIT_SET(B, I) ((B) = (I))
1022 #endif
1023
1024 /*===================================================================*/
1025 /*=========================== utf8lib.c =============================*/
1026 /*===================================================================*/
1027 typedef uint32_t uchar_t;
1028
1029 bool    u8_analyze (const char *_s, size_t *_start, size_t *_len, uchar_t *_ch, size_t _maxlen);
1030 size_t  u8_strlen  (const char*);
1031 size_t  u8_strnlen (const char*, size_t);
1032 uchar_t u8_getchar (const char*, const char**);
1033 uchar_t u8_getnchar(const char*, const char**, size_t);
1034 int     u8_fromchar(uchar_t w,   char *to,     size_t maxlen);
1035
1036 /*===================================================================*/
1037 /*============================= opts.c ==============================*/
1038 /*===================================================================*/
1039 typedef struct {
1040     const char *name;
1041     longbit     bit;
1042 } opts_flag_def;
1043
1044 bool opts_setflag  (const char *, bool);
1045 bool opts_setwarn  (const char *, bool);
1046 bool opts_setwerror(const char *, bool);
1047 bool opts_setoptim (const char *, bool);
1048
1049 void opts_init         (const char *, int, size_t);
1050 void opts_set          (uint32_t   *, size_t, bool);
1051 void opts_setoptimlevel(unsigned int);
1052 void opts_ini_init     (const char *);
1053
1054 enum {
1055 # define GMQCC_TYPE_FLAGS
1056 # define GMQCC_DEFINE_FLAG(X) X,
1057 #  include "opts.def"
1058     COUNT_FLAGS
1059 };
1060 static const opts_flag_def opts_flag_list[] = {
1061 # define GMQCC_TYPE_FLAGS
1062 # define GMQCC_DEFINE_FLAG(X) { #X, LONGBIT(X) },
1063 #  include "opts.def"
1064     { NULL, LONGBIT(0) }
1065 };
1066
1067 enum {
1068 # define GMQCC_TYPE_WARNS
1069 # define GMQCC_DEFINE_FLAG(X) WARN_##X,
1070 #  include "opts.def"
1071     COUNT_WARNINGS
1072 };
1073 static const opts_flag_def opts_warn_list[] = {
1074 # define GMQCC_TYPE_WARNS
1075 # define GMQCC_DEFINE_FLAG(X) { #X, LONGBIT(WARN_##X) },
1076 #  include "opts.def"
1077     { NULL, LONGBIT(0) }
1078 };
1079
1080 enum {
1081 # define GMQCC_TYPE_OPTIMIZATIONS
1082 # define GMQCC_DEFINE_FLAG(NAME, MIN_O) OPTIM_##NAME,
1083 #  include "opts.def"
1084     COUNT_OPTIMIZATIONS
1085 };
1086 static const opts_flag_def opts_opt_list[] = {
1087 # define GMQCC_TYPE_OPTIMIZATIONS
1088 # define GMQCC_DEFINE_FLAG(NAME, MIN_O) { #NAME, LONGBIT(OPTIM_##NAME) },
1089 #  include "opts.def"
1090     { NULL, LONGBIT(0) }
1091 };
1092 static const unsigned int opts_opt_oflag[] = {
1093 # define GMQCC_TYPE_OPTIMIZATIONS
1094 # define GMQCC_DEFINE_FLAG(NAME, MIN_O) MIN_O,
1095 #  include "opts.def"
1096     0
1097 };
1098 extern unsigned int opts_optimizationcount[COUNT_OPTIMIZATIONS];
1099
1100 /* other options: */
1101 typedef enum {
1102     COMPILER_QCC,     /* circa  QuakeC */
1103     COMPILER_FTEQCC,  /* fteqcc QuakeC */
1104     COMPILER_QCCX,    /* qccx   QuakeC */
1105     COMPILER_GMQCC    /* this   QuakeC */
1106 } opts_std_t;
1107
1108 /* TODO: cleanup this */
1109 typedef struct {
1110     uint32_t    O;              /* -Ox           */
1111     const char *output;         /* -o file       */
1112     bool        quiet;          /* -q --quiet    */
1113     bool        g;              /* -g            */
1114     opts_std_t  standard;       /* -std=         */
1115     bool        debug;          /* -debug        */
1116     bool        memchk;         /* -memchk       */
1117     bool        dumpfin;        /* -dumpfin      */
1118     bool        dump;           /* -dump         */
1119     bool        forcecrc;       /* --force-crc=  */
1120     uint16_t    forced_crc;     /* --force-crc=  */
1121     bool        pp_only;        /* -E            */
1122     size_t      max_array_size; /* --max-array=  */
1123
1124     uint32_t flags       [1 + (COUNT_FLAGS         / 32)];
1125     uint32_t warn        [1 + (COUNT_WARNINGS      / 32)];
1126     uint32_t werror      [1 + (COUNT_WARNINGS      / 32)];
1127     uint32_t optimization[1 + (COUNT_OPTIMIZATIONS / 32)];
1128 } opts_cmd_t;
1129
1130 extern opts_cmd_t opts;
1131
1132 #define OPTS_FLAG(i)         (!! (opts.flags       [(i)/32] & (1<< ((i)%32))))
1133 #define OPTS_WARN(i)         (!! (opts.warn        [(i)/32] & (1<< ((i)%32))))
1134 #define OPTS_WERROR(i)       (!! (opts.werror      [(i)/32] & (1<< ((i)%32))))
1135 #define OPTS_OPTIMIZATION(i) (!! (opts.optimization[(i)/32] & (1<< ((i)%32))))
1136
1137 #endif