]> git.xonotic.org Git - xonotic/gmqcc.git/blob - exec.c
vtos should use the same quotes as we use in qc source
[xonotic/gmqcc.git] / exec.c
1 #include <errno.h>
2 #include <stdio.h>
3 #include <string.h>
4 #include <stdarg.h>
5
6 #include "gmqcc.h"
7
8 MEM_VEC_FUNCTIONS(qc_program,   prog_section_statement, code)
9 MEM_VEC_FUNCTIONS(qc_program,   prog_section_def,       defs)
10 MEM_VEC_FUNCTIONS(qc_program,   prog_section_def,       fields)
11 MEM_VEC_FUNCTIONS(qc_program,   prog_section_function,  functions)
12 MEM_VEC_FUNCTIONS(qc_program,   char,                   strings)
13 MEM_VEC_FUN_APPEND(qc_program,  char,                   strings)
14 MEM_VEC_FUN_RESIZE(qc_program,  char,                   strings)
15 MEM_VEC_FUNCTIONS(qc_program,   qcint,                  globals)
16 MEM_VEC_FUNCTIONS(qc_program,   qcint,                  entitydata)
17 MEM_VEC_FUNCTIONS(qc_program,   bool,                   entitypool)
18
19 MEM_VEC_FUNCTIONS(qc_program,   qcint,         localstack)
20 MEM_VEC_FUN_APPEND(qc_program,  qcint,         localstack)
21 MEM_VEC_FUN_RESIZE(qc_program,  qcint,         localstack)
22 MEM_VEC_FUNCTIONS(qc_program,   qc_exec_stack, stack)
23
24 MEM_VEC_FUNCTIONS(qc_program,   size_t, profile)
25 MEM_VEC_FUN_RESIZE(qc_program,  size_t, profile)
26
27 MEM_VEC_FUNCTIONS(qc_program,   prog_builtin, builtins)
28
29 static void loaderror(const char *fmt, ...)
30 {
31     int     err = errno;
32     va_list ap;
33     va_start(ap, fmt);
34     vprintf(fmt, ap);
35     va_end(ap);
36     printf(": %s\n", strerror(err));
37 }
38
39 static void qcvmerror(qc_program *prog, const char *fmt, ...)
40 {
41     va_list ap;
42
43     prog->vmerror++;
44
45     va_start(ap, fmt);
46     vprintf(fmt, ap);
47     va_end(ap);
48     putchar('\n');
49 }
50
51 qc_program* prog_load(const char *filename)
52 {
53     qc_program *prog;
54     prog_header header;
55     size_t      i;
56     FILE *file;
57
58     file = util_fopen(filename, "rb");
59     if (!file)
60         return NULL;
61
62     if (fread(&header, sizeof(header), 1, file) != 1) {
63         loaderror("failed to read header from '%s'", filename);
64         fclose(file);
65         return NULL;
66     }
67
68     if (header.version != 6) {
69         loaderror("header says this is a version %i progs, we need version 6\n", header.version);
70         fclose(file);
71         return NULL;
72     }
73
74     prog = (qc_program*)mem_a(sizeof(qc_program));
75     if (!prog) {
76         fclose(file);
77         printf("failed to allocate program data\n");
78         return NULL;
79     }
80     memset(prog, 0, sizeof(*prog));
81
82     prog->entityfields = header.entfield;
83     prog->crc16 = header.crc16;
84
85     prog->filename = util_strdup(filename);
86     if (!prog->filename) {
87         loaderror("failed to store program name");
88         goto error;
89     }
90
91 #define read_data(hdrvar, progvar, type)                                         \
92     if (fseek(file, header.hdrvar.offset, SEEK_SET) != 0) {                      \
93         loaderror("seek failed");                                                \
94         goto error;                                                              \
95     }                                                                            \
96     prog->progvar##_alloc = header.hdrvar.length;                                \
97     prog->progvar##_count = header.hdrvar.length;                                \
98     prog->progvar = (type*)mem_a(header.hdrvar.length * sizeof(*prog->progvar)); \
99     if (!prog->progvar)                                                          \
100         goto error;                                                              \
101     if (fread(prog->progvar, sizeof(*prog->progvar), header.hdrvar.length, file) \
102         != header.hdrvar.length) {                                               \
103         loaderror("read failed");                                                \
104         goto error;                                                              \
105     }
106 #define read_data1(x, y) read_data(x, x, y)
107
108     read_data (statements, code, prog_section_statement);
109     read_data1(defs,             prog_section_def);
110     read_data1(fields,           prog_section_def);
111     read_data1(functions,        prog_section_function);
112     read_data1(strings,          char);
113     read_data1(globals,          qcint);
114
115     fclose(file);
116
117     /* profile counters */
118     if (!qc_program_profile_resize(prog, prog->code_count))
119         goto error;
120
121     /* Add tempstring area */
122     prog->tempstring_start = prog->strings_count;
123     prog->tempstring_at    = prog->strings_count;
124     if (!qc_program_strings_resize(prog, prog->strings_count + 16*1024))
125         goto error;
126
127     /* spawn the world entity */
128     if (!qc_program_entitypool_add(prog, true)) {
129         loaderror("failed to allocate world entity\n");
130         goto error;
131     }
132     for (i = 0; i < prog->entityfields; ++i) {
133         if (!qc_program_entitydata_add(prog, 0)) {
134             loaderror("failed to allocate world data\n");
135             goto error;
136         }
137     }
138     prog->entities = 1;
139
140     return prog;
141
142 error:
143     if (prog->filename)   mem_d(prog->filename);
144     if (prog->code)       mem_d(prog->code);
145     if (prog->defs)       mem_d(prog->defs);
146     if (prog->fields)     mem_d(prog->fields);
147     if (prog->functions)  mem_d(prog->functions);
148     if (prog->strings)    mem_d(prog->strings);
149     if (prog->globals)    mem_d(prog->globals);
150     if (prog->entitydata) mem_d(prog->entitydata);
151     if (prog->entitypool) mem_d(prog->entitypool);
152     mem_d(prog);
153     return NULL;
154 }
155
156 void prog_delete(qc_program *prog)
157 {
158     if (prog->filename) mem_d(prog->filename);
159     MEM_VECTOR_CLEAR(prog, code);
160     MEM_VECTOR_CLEAR(prog, defs);
161     MEM_VECTOR_CLEAR(prog, fields);
162     MEM_VECTOR_CLEAR(prog, functions);
163     MEM_VECTOR_CLEAR(prog, strings);
164     MEM_VECTOR_CLEAR(prog, globals);
165     MEM_VECTOR_CLEAR(prog, entitydata);
166     MEM_VECTOR_CLEAR(prog, entitypool);
167     MEM_VECTOR_CLEAR(prog, localstack);
168     MEM_VECTOR_CLEAR(prog, stack);
169     MEM_VECTOR_CLEAR(prog, profile);
170
171     if (prog->builtins_alloc) {
172         MEM_VECTOR_CLEAR(prog, builtins);
173     }
174     /* otherwise the builtins were statically allocated */
175     mem_d(prog);
176 }
177
178 /***********************************************************************
179  * VM code
180  */
181
182 char* prog_getstring(qc_program *prog, qcint str)
183 {
184     if (str < 0 || str >= prog->strings_count)
185         return "<<<invalid string>>>";
186     return prog->strings + str;
187 }
188
189 prog_section_def* prog_entfield(qc_program *prog, qcint off)
190 {
191     size_t i;
192     for (i = 0; i < prog->fields_count; ++i) {
193         if (prog->fields[i].offset == off)
194             return (prog->fields + i);
195     }
196     return NULL;
197 }
198
199 prog_section_def* prog_getdef(qc_program *prog, qcint off)
200 {
201     size_t i;
202     for (i = 0; i < prog->defs_count; ++i) {
203         if (prog->defs[i].offset == off)
204             return (prog->defs + i);
205     }
206     return NULL;
207 }
208
209 qcany* prog_getedict(qc_program *prog, qcint e)
210 {
211     if (e >= prog->entitypool_count) {
212         prog->vmerror++;
213         printf("Accessing out of bounds edict %i\n", (int)e);
214         e = 0;
215     }
216     return (qcany*)(prog->entitydata + (prog->entityfields * e));
217 }
218
219 qcint prog_spawn_entity(qc_program *prog)
220 {
221     char  *data;
222     size_t i;
223     qcint  e;
224     for (e = 0; e < (qcint)prog->entitypool_count; ++e) {
225         if (!prog->entitypool[e]) {
226             data = (char*)(prog->entitydata + (prog->entityfields * e));
227             memset(data, 0, prog->entityfields * sizeof(qcint));
228             return e;
229         }
230     }
231     if (!qc_program_entitypool_add(prog, true)) {
232         prog->vmerror++;
233         printf("Failed to allocate entity\n");
234         return 0;
235     }
236     prog->entities++;
237     for (i = 0; i < prog->entityfields; ++i) {
238         if (!qc_program_entitydata_add(prog, 0)) {
239             printf("Failed to allocate entity\n");
240             return 0;
241         }
242     }
243     data = (char*)(prog->entitydata + (prog->entityfields * e));
244     memset(data, 0, prog->entityfields * sizeof(qcint));
245     return e;
246 }
247
248 void prog_free_entity(qc_program *prog, qcint e)
249 {
250     if (!e) {
251         prog->vmerror++;
252         printf("Trying to free world entity\n");
253         return;
254     }
255     if (e >= prog->entitypool_count) {
256         prog->vmerror++;
257         printf("Trying to free out of bounds entity\n");
258         return;
259     }
260     if (!prog->entitypool[e]) {
261         prog->vmerror++;
262         printf("Double free on entity\n");
263         return;
264     }
265     prog->entitypool[e] = false;
266 }
267
268 qcint prog_tempstring(qc_program *prog, const char *_str)
269 {
270     /* we don't access it, but the macro-generated functions don't use
271      * const
272      */
273     char *str = (char*)_str;
274
275     size_t len = strlen(str);
276     size_t at = prog->tempstring_at;
277
278     /* when we reach the end we start over */
279     if (at + len >= prog->strings_count)
280         at = prog->tempstring_start;
281
282     /* when it doesn't fit, reallocate */
283     if (at + len >= prog->strings_count)
284     {
285         prog->strings_count = at;
286         if (!qc_program_strings_append(prog, str, len+1)) {
287             prog->vmerror = VMERR_TEMPSTRING_ALLOC;
288             return 0;
289         }
290         return at;
291     }
292
293     /* when it fits, just copy */
294     memcpy(prog->strings + at, str, len+1);
295     prog->tempstring_at += len+1;
296     return at;
297 }
298
299 static int print_escaped_string(const char *str, size_t maxlen)
300 {
301     int len = 2;
302     putchar('"');
303     --maxlen; /* because we're lazy and have escape sequences */
304     while (*str) {
305         if (len >= maxlen) {
306             putchar('.');
307             putchar('.');
308             putchar('.');
309             len += 3;
310             break;
311         }
312         switch (*str) {
313             case '\a': len += 2; putchar('\\'); putchar('a'); break;
314             case '\b': len += 2; putchar('\\'); putchar('b'); break;
315             case '\r': len += 2; putchar('\\'); putchar('r'); break;
316             case '\n': len += 2; putchar('\\'); putchar('n'); break;
317             case '\t': len += 2; putchar('\\'); putchar('t'); break;
318             case '\f': len += 2; putchar('\\'); putchar('f'); break;
319             case '\v': len += 2; putchar('\\'); putchar('v'); break;
320             case '\\': len += 2; putchar('\\'); putchar('\\'); break;
321             case '"':  len += 2; putchar('\\'); putchar('"'); break;
322             default:
323                 ++len;
324                 putchar(*str);
325                 break;
326         }
327         ++str;
328     }
329     putchar('"');
330     return len;
331 }
332
333 static void trace_print_global(qc_program *prog, unsigned int glob, int vtype)
334 {
335     static char spaces[28+1] = "                            ";
336     prog_section_def *def;
337     qcany    *value;
338     int       len;
339
340     if (!glob) {
341         len = printf("<null>,");
342         goto done;
343     }
344
345     def = prog_getdef(prog, glob);
346     value = (qcany*)(&prog->globals[glob]);
347
348     if (def) {
349         const char *name = prog_getstring(prog, def->name);
350         if (name[0] == '#')
351             len = printf("$");
352         else
353             len = printf("%s ", name);
354         vtype = def->type;
355     }
356     else
357         len = printf("[@%u] ", glob);
358
359     switch (vtype) {
360         case TYPE_VOID:
361         case TYPE_ENTITY:
362         case TYPE_FIELD:
363         case TYPE_FUNCTION:
364         case TYPE_POINTER:
365             len += printf("(%i),", value->_int);
366             break;
367         case TYPE_VECTOR:
368             len += printf("'%g %g %g',", value->vector[0],
369                                          value->vector[1],
370                                          value->vector[2]);
371             break;
372         case TYPE_STRING:
373             len += print_escaped_string(prog_getstring(prog, value->string), sizeof(spaces)-len-5);
374             len += printf(",");
375             /* len += printf("\"%s\",", prog_getstring(prog, value->string)); */
376             break;
377         case TYPE_FLOAT:
378         default:
379             len += printf("%g,", value->_float);
380             break;
381     }
382 done:
383     if (len < sizeof(spaces)-1) {
384         spaces[sizeof(spaces)-1-len] = 0;
385         printf(spaces);
386         spaces[sizeof(spaces)-1-len] = ' ';
387     }
388 }
389
390 static void prog_print_statement(qc_program *prog, prog_section_statement *st)
391 {
392     if (st->opcode >= (sizeof(asm_instr)/sizeof(asm_instr[0]))) {
393         printf("<illegal instruction %d>\n", st->opcode);
394         return;
395     }
396     printf(" <> %-12s", asm_instr[st->opcode].m);
397     if (st->opcode >= INSTR_IF &&
398         st->opcode <= INSTR_IFNOT)
399     {
400         trace_print_global(prog, st->o1.u1, TYPE_FLOAT);
401         printf("%d\n", st->o2.s1);
402     }
403     else if (st->opcode >= INSTR_CALL0 &&
404              st->opcode <= INSTR_CALL8)
405     {
406         printf("\n");
407     }
408     else if (st->opcode == INSTR_GOTO)
409     {
410         printf("%i\n", st->o1.s1);
411     }
412     else
413     {
414         int t[3] = { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT };
415         switch (st->opcode)
416         {
417             case INSTR_MUL_FV:
418                 t[1] = t[2] = TYPE_VECTOR;
419                 break;
420             case INSTR_MUL_VF:
421                 t[0] = t[2] = TYPE_VECTOR;
422                 break;
423             case INSTR_MUL_V:
424                 t[0] = t[1] = TYPE_VECTOR;
425                 break;
426             case INSTR_ADD_V:
427             case INSTR_SUB_V:
428             case INSTR_EQ_V:
429             case INSTR_NE_V:
430                 t[0] = t[1] = t[2] = TYPE_VECTOR;
431                 break;
432             case INSTR_EQ_S:
433             case INSTR_NE_S:
434                 t[0] = t[1] = TYPE_STRING;
435                 break;
436             case INSTR_STORE_F:
437             case INSTR_STOREP_F:
438                 t[2] = -1;
439                 break;
440             case INSTR_STORE_V:
441                 t[0] = t[1] = TYPE_VECTOR; t[2] = -1;
442                 break;
443             case INSTR_STORE_S:
444                 t[0] = t[1] = TYPE_STRING; t[2] = -1;
445                 break;
446             case INSTR_STORE_ENT:
447                 t[0] = t[1] = TYPE_ENTITY; t[2] = -1;
448                 break;
449             case INSTR_STORE_FLD:
450                 t[0] = t[1] = TYPE_FIELD; t[2] = -1;
451                 break;
452             case INSTR_STORE_FNC:
453                 t[0] = t[1] = TYPE_FUNCTION; t[2] = -1;
454                 break;
455             case INSTR_STOREP_V:
456                 t[0] = TYPE_VECTOR; t[1] = TYPE_ENTITY; t[2] = -1;
457                 break;
458             case INSTR_STOREP_S:
459                 t[0] = TYPE_STRING; t[1] = TYPE_ENTITY; t[2] = -1;
460                 break;
461             case INSTR_STOREP_ENT:
462                 t[0] = TYPE_ENTITY; t[1] = TYPE_ENTITY; t[2] = -1;
463                 break;
464             case INSTR_STOREP_FLD:
465                 t[0] = TYPE_FIELD; t[1] = TYPE_ENTITY; t[2] = -1;
466                 break;
467             case INSTR_STOREP_FNC:
468                 t[0] = TYPE_FUNCTION; t[1] = TYPE_ENTITY; t[2] = -1;
469                 break;
470         }
471         if (t[0] >= 0) trace_print_global(prog, st->o1.u1, t[0]);
472         else           printf("(none),          ");
473         if (t[1] >= 0) trace_print_global(prog, st->o2.u1, t[1]);
474         else           printf("(none),          ");
475         if (t[2] >= 0) trace_print_global(prog, st->o3.u1, t[2]);
476         else           printf("(none)");
477         printf("\n");
478     }
479     fflush(stdout);
480 }
481
482 static qcint prog_enterfunction(qc_program *prog, prog_section_function *func)
483 {
484     qc_exec_stack st;
485     size_t p, parampos;
486
487     /* back up locals */
488     st.localsp  = prog->localstack_count;
489     st.stmt     = prog->statement;
490     st.function = func;
491
492 #ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
493     if (prog->stack_count)
494     {
495         prog_section_function *cur;
496         cur = prog->stack[prog->stack_count-1].function;
497         if (cur)
498         {
499             qcint *globals = prog->globals + cur->firstlocal;
500             if (!qc_program_localstack_append(prog, globals, cur->locals))
501             {
502                 printf("out of memory\n");
503                 exit(1);
504             }
505         }
506     }
507 #else
508     {
509         qcint *globals = prog->globals + func->firstlocal;
510         if (!qc_program_localstack_append(prog, globals, func->locals))
511         {
512             printf("out of memory\n");
513             exit(1);
514         }
515     }
516 #endif
517
518     /* copy parameters */
519     parampos = func->firstlocal;
520     for (p = 0; p < func->nargs; ++p)
521     {
522         size_t s;
523         for (s = 0; s < func->argsize[p]; ++s) {
524             prog->globals[parampos] = prog->globals[OFS_PARM0 + 3*p + s];
525             ++parampos;
526         }
527     }
528
529     if (!qc_program_stack_add(prog, st)) {
530         printf("out of memory\n");
531         exit(1);
532     }
533
534     return func->entry;
535 }
536
537 static qcint prog_leavefunction(qc_program *prog)
538 {
539     prog_section_function *prev = NULL;
540     size_t oldsp;
541
542     qc_exec_stack st = prog->stack[prog->stack_count-1];
543
544 #ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
545     if (prog->stack_count > 1) {
546         prev  = prog->stack[prog->stack_count-2].function;
547         oldsp = prog->stack[prog->stack_count-2].localsp;
548     }
549 #else
550     prev  = prog->stack[prog->stack_count-1].function;
551     oldsp = prog->stack[prog->stack_count-1].localsp;
552 #endif
553     if (prev) {
554         qcint *globals = prog->globals + prev->firstlocal;
555         memcpy(globals, prog->localstack + oldsp, prev->locals);
556         if (!qc_program_localstack_resize(prog, oldsp)) {
557             printf("out of memory\n");
558             exit(1);
559         }
560     }
561
562     if (!qc_program_stack_remove(prog, prog->stack_count-1)) {
563         printf("out of memory\n");
564         exit(1);
565     }
566
567     return st.stmt - 1; /* offset the ++st */
568 }
569
570 bool prog_exec(qc_program *prog, prog_section_function *func, size_t flags, long maxjumps)
571 {
572     long jumpcount = 0;
573     size_t oldxflags = prog->xflags;
574     prog_section_statement *st;
575
576     prog->vmerror = 0;
577     prog->xflags = flags;
578
579     st = prog->code + prog_enterfunction(prog, func);
580     --st;
581     switch (flags)
582     {
583         default:
584         case 0:
585         {
586 #define QCVM_PROFILE 0
587 #define QCVM_TRACE   0
588 #           include "execloop.h"
589             break;
590         }
591         case (VMXF_TRACE):
592         {
593 #define QCVM_PROFILE 0
594 #define QCVM_TRACE   1
595 #           include "execloop.h"
596             break;
597         }
598         case (VMXF_PROFILE):
599         {
600 #define QCVM_PROFILE 1
601 #define QCVM_TRACE   0
602 #           include "execloop.h"
603             break;
604         }
605         case (VMXF_TRACE|VMXF_PROFILE):
606         {
607 #define QCVM_PROFILE 1
608 #define QCVM_TRACE   1
609 #           include "execloop.h"
610             break;
611         }
612     };
613
614 cleanup:
615     prog->xflags = oldxflags;
616     prog->localstack_count = 0;
617     prog->stack_count = 0;
618     if (prog->vmerror)
619         return false;
620     return true;
621 }
622
623 /***********************************************************************
624  * main for when building the standalone executor
625  */
626
627 #if defined(QCVM_EXECUTOR)
628 #include <math.h>
629
630 const char *type_name[TYPE_COUNT] = {
631     "void",
632     "string",
633     "float",
634     "vector",
635     "entity",
636     "field",
637     "function",
638     "pointer",
639 #if 0
640     "integer",
641 #endif
642     "variant"
643 };
644
645 bool        opts_debug    = false;
646 bool        opts_memchk   = false;
647
648 #define CheckArgs(num) do {                                                    \
649     if (prog->argc != (num)) {                                                 \
650         prog->vmerror++;                                                       \
651         printf("ERROR: invalid number of arguments for %s: %i, expected %i\n", \
652         __FUNCTION__, prog->argc, (num));                                      \
653         return -1;                                                             \
654     }                                                                          \
655 } while (0)
656
657 #define GetGlobal(idx) ((qcany*)(prog->globals + (idx)))
658 #define GetArg(num) GetGlobal(OFS_PARM0 + 3*(num))
659 #define Return(any) *(GetGlobal(OFS_RETURN)) = (any)
660
661 static int qc_print(qc_program *prog)
662 {
663     size_t i;
664     const char *laststr = NULL;
665     for (i = 0; i < prog->argc; ++i) {
666         qcany *str = (qcany*)(prog->globals + OFS_PARM0 + 3*i);
667         printf("%s", (laststr = prog_getstring(prog, str->string)));
668     }
669     if (laststr && (prog->xflags & VMXF_TRACE)) {
670         size_t len = strlen(laststr);
671         if (!len || laststr[len-1] != '\n')
672             printf("\n");
673     }
674     return 0;
675 }
676
677 static int qc_error(qc_program *prog)
678 {
679     printf("*** VM raised an error:\n");
680     qc_print(prog);
681     prog->vmerror++;
682     return -1;
683 }
684
685 static int qc_ftos(qc_program *prog)
686 {
687     char buffer[512];
688     qcany *num;
689     qcany str;
690     CheckArgs(1);
691     num = GetArg(0);
692     snprintf(buffer, sizeof(buffer), "%g", num->_float);
693     str.string = prog_tempstring(prog, buffer);
694     Return(str);
695     return 0;
696 }
697
698 static int qc_vtos(qc_program *prog)
699 {
700     char buffer[512];
701     qcany *num;
702     qcany str;
703     CheckArgs(1);
704     num = GetArg(0);
705     snprintf(buffer, sizeof(buffer), "'%g %g %g'", num->vector[0], num->vector[1], num->vector[2]);
706     str.string = prog_tempstring(prog, buffer);
707     Return(str);
708     return 0;
709 }
710
711 static int qc_spawn(qc_program *prog)
712 {
713     qcany ent;
714     CheckArgs(0);
715     ent.edict = prog_spawn_entity(prog);
716     Return(ent);
717     return (ent.edict ? 0 : -1);
718 }
719
720 static int qc_kill(qc_program *prog)
721 {
722     qcany *ent;
723     CheckArgs(1);
724     ent = GetArg(0);
725     prog_free_entity(prog, ent->edict);
726     return 0;
727 }
728
729 static int qc_vlen(qc_program *prog)
730 {
731     qcany *vec, len;
732     CheckArgs(1);
733     vec = GetArg(0);
734     len._float = sqrt(vec->vector[0] * vec->vector[0] + 
735                       vec->vector[1] * vec->vector[1] +
736                       vec->vector[2] * vec->vector[2]);
737     Return(len);
738     return 0;
739 }
740
741 static prog_builtin qc_builtins[] = {
742     NULL,
743     &qc_print, /*   1   */
744     &qc_ftos,  /*   2   */
745     &qc_spawn, /*   3   */
746     &qc_kill,  /*   4   */
747     &qc_vtos,  /*   5   */
748     &qc_error, /*   6   */
749     &qc_vlen   /*   7   */
750 };
751 static size_t qc_builtins_count = sizeof(qc_builtins) / sizeof(qc_builtins[0]);
752
753 static const char *arg0 = NULL;
754
755 void usage()
756 {
757     printf("usage: [-debug] %s file\n", arg0);
758     exit(1);
759 }
760
761 int main(int argc, char **argv)
762 {
763     size_t      i;
764     qcint       fnmain = -1;
765     qc_program *prog;
766     size_t      xflags = VMXF_DEFAULT;
767     bool        opts_printfields = false;
768     bool        opts_printdefs   = false;
769     bool        opts_info  = false;
770
771     arg0 = argv[0];
772
773     if (argc < 2)
774         usage();
775
776     while (argc > 2) {
777         if (!strcmp(argv[1], "-trace")) {
778             --argc;
779             ++argv;
780             xflags |= VMXF_TRACE;
781         }
782         else if (!strcmp(argv[1], "-profile")) {
783             --argc;
784             ++argv;
785             xflags |= VMXF_PROFILE;
786         }
787         else if (!strcmp(argv[1], "-info")) {
788             --argc;
789             ++argv;
790             opts_info = true;
791         }
792         else if (!strcmp(argv[1], "-printdefs")) {
793             --argc;
794             ++argv;
795             opts_printdefs = true;
796         }
797         else if (!strcmp(argv[1], "-printfields")) {
798             --argc;
799             ++argv;
800             opts_printfields = true;
801         }
802         else
803             usage();
804     }
805
806
807     prog = prog_load(argv[1]);
808     if (!prog) {
809         printf("failed to load program '%s'\n", argv[1]);
810         exit(1);
811     }
812
813     prog->builtins       = qc_builtins;
814     prog->builtins_count = qc_builtins_count;
815     prog->builtins_alloc = 0;
816
817     printf("Program's system-checksum = 0x%04x\n", (int)prog->crc16);
818
819     for (i = 1; i < prog->functions_count; ++i) {
820         const char *name = prog_getstring(prog, prog->functions[i].name);
821         /* printf("Found function: %s\n", name); */
822         if (!strcmp(name, "main"))
823             fnmain = (qcint)i;
824     }
825     printf("Entity field space: %i\n", (int)prog->entityfields);
826     if (opts_info) {
827         prog_delete(prog);
828         return 0;
829     }
830     if (opts_printdefs) {
831         for (i = 0; i < prog->defs_count; ++i) {
832             printf("Global: %8s %-16s at %u\n",
833                    type_name[prog->defs[i].type],
834                    prog_getstring(prog, prog->defs[i].name),
835                    (unsigned int)prog->defs[i].offset);
836         }
837     }
838     else if (opts_printfields) {
839         for (i = 0; i < prog->fields_count; ++i) {
840             printf("Field: %8s %-16s at %u\n",
841                    type_name[prog->fields[i].type],
842                    prog_getstring(prog, prog->fields[i].name),
843                    (unsigned int)prog->fields[i].offset);
844         }
845     }
846     else
847     {
848         if (fnmain > 0)
849         {
850             prog_exec(prog, &prog->functions[fnmain], xflags, VM_JUMPS_DEFAULT);
851         }
852         else
853             printf("No main function found\n");
854     }
855
856     prog_delete(prog);
857     return 0;
858 }
859 #endif