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)
 300 {
 301     int len = 2;
 302     putchar('"');
 303     while (*str) {
 304         switch (*str) {
 305             case '\a': len += 2; putchar('\\'); putchar('a'); break;
 306             case '\b': len += 2; putchar('\\'); putchar('b'); break;
 307             case '\r': len += 2; putchar('\\'); putchar('r'); break;
 308             case '\n': len += 2; putchar('\\'); putchar('n'); break;
 309             case '\t': len += 2; putchar('\\'); putchar('t'); break;
 310             case '\f': len += 2; putchar('\\'); putchar('f'); break;
 311             case '\v': len += 2; putchar('\\'); putchar('v'); break;
 312             case '\\': len += 2; putchar('\\'); putchar('\\'); break;
 313             case '"':  len += 2; putchar('\\'); putchar('"'); break;
 314             default:
 315                 ++len;
 316                 putchar(*str);
 317                 break;
 318         }
 319         ++str;
 320     }
 321     putchar('"');
 322     return len;
 323 }
 324
 325 static void trace_print_global(qc_program *prog, unsigned int glob, int vtype)
 326 {
 327     static char spaces[40+1] = "                                        ";
 328     prog_section_def *def;
 329     qcany    *value;
 330     int       len;
 331
 332     if (!glob) {
 333         len = printf("<null>,");
 334         goto done;
 335     }
 336
 337     def = prog_getdef(prog, glob);
 338     value = (qcany*)(&prog->globals[glob]);
 339
 340     if (def) {
 341         const char *name = prog_getstring(prog, def->name);
 342         if (name[0] == '#')
 343             len = printf("$");
 344         else
 345             len = printf("%s ", name);
 346         vtype = def->type;
 347     }
 348     else
 349         len = printf("[@%u] ", glob);
 350
 351     switch (vtype) {
 352         case TYPE_VOID:
 353         case TYPE_ENTITY:
 354         case TYPE_FIELD:
 355         case TYPE_FUNCTION:
 356         case TYPE_POINTER:
 357             len += printf("(%i),", value->_int);
 358             break;
 359         case TYPE_VECTOR:
 360             len += printf("'%g %g %g',", value->vector[0],
 361                                          value->vector[1],
 362                                          value->vector[2]);
 363             break;
 364         case TYPE_STRING:
 365             len += print_escaped_string(prog_getstring(prog, value->string));
 366             len += printf(",");
 367             /* len += printf("\"%s\",", prog_getstring(prog, value->string)); */
 368             break;
 369         case TYPE_FLOAT:
 370         default:
 371             len += printf("%g,", value->_float);
 372             break;
 373     }
 374 done:
 375     if (len < sizeof(spaces)-1) {
 376         spaces[sizeof(spaces)-1-len] = 0;
 377         printf(spaces);
 378         spaces[sizeof(spaces)-1-len] = ' ';
 379     }
 380 }
 381
 382 static void prog_print_statement(qc_program *prog, prog_section_statement *st)
 383 {
 384     if (st->opcode >= (sizeof(asm_instr)/sizeof(asm_instr[0]))) {
 385         printf("<illegal instruction %d>\n", st->opcode);
 386         return;
 387     }
 388     printf(" <> %-12s", asm_instr[st->opcode].m);
 389     if (st->opcode >= INSTR_IF &&
 390         st->opcode <= INSTR_IFNOT)
 391     {
 392         trace_print_global(prog, st->o1.u1, TYPE_FLOAT);
 393         printf("%d\n", st->o2.s1);
 394     }
 395     else if (st->opcode >= INSTR_CALL0 &&
 396              st->opcode <= INSTR_CALL8)
 397     {
 398         printf("\n");
 399     }
 400     else if (st->opcode == INSTR_GOTO)
 401     {
 402         printf("%i\n", st->o1.s1);
 403     }
 404     else
 405     {
 406         int t[3] = { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT };
 407         switch (st->opcode)
 408         {
 409             case INSTR_MUL_FV:
 410                 t[1] = t[2] = TYPE_VECTOR;
 411                 break;
 412             case INSTR_MUL_VF:
 413                 t[0] = t[2] = TYPE_VECTOR;
 414                 break;
 415             case INSTR_MUL_V:
 416                 t[0] = t[1] = TYPE_VECTOR;
 417                 break;
 418             case INSTR_ADD_V:
 419             case INSTR_SUB_V:
 420             case INSTR_EQ_V:
 421             case INSTR_NE_V:
 422                 t[0] = t[1] = t[2] = TYPE_VECTOR;
 423                 break;
 424             case INSTR_EQ_S:
 425             case INSTR_NE_S:
 426                 t[0] = t[1] = TYPE_STRING;
 427                 break;
 428             case INSTR_STORE_F:
 429             case INSTR_STOREP_F:
 430                 t[2] = -1;
 431                 break;
 432             case INSTR_STORE_V:
 433                 t[0] = t[1] = TYPE_VECTOR; t[2] = -1;
 434                 break;
 435             case INSTR_STORE_S:
 436                 t[0] = t[1] = TYPE_STRING; t[2] = -1;
 437                 break;
 438             case INSTR_STORE_ENT:
 439                 t[0] = t[1] = TYPE_ENTITY; t[2] = -1;
 440                 break;
 441             case INSTR_STORE_FLD:
 442                 t[0] = t[1] = TYPE_FIELD; t[2] = -1;
 443                 break;
 444             case INSTR_STORE_FNC:
 445                 t[0] = t[1] = TYPE_FUNCTION; t[2] = -1;
 446                 break;
 447             case INSTR_STOREP_V:
 448                 t[0] = TYPE_VECTOR; t[1] = TYPE_ENTITY; t[2] = -1;
 449                 break;
 450             case INSTR_STOREP_S:
 451                 t[0] = TYPE_STRING; t[1] = TYPE_ENTITY; t[2] = -1;
 452                 break;
 453             case INSTR_STOREP_ENT:
 454                 t[0] = TYPE_ENTITY; t[1] = TYPE_ENTITY; t[2] = -1;
 455                 break;
 456             case INSTR_STOREP_FLD:
 457                 t[0] = TYPE_FIELD; t[1] = TYPE_ENTITY; t[2] = -1;
 458                 break;
 459             case INSTR_STOREP_FNC:
 460                 t[0] = TYPE_FUNCTION; t[1] = TYPE_ENTITY; t[2] = -1;
 461                 break;
 462         }
 463         if (t[0] >= 0) trace_print_global(prog, st->o1.u1, t[0]);
 464         else           printf("(none),          ");
 465         if (t[1] >= 0) trace_print_global(prog, st->o2.u1, t[1]);
 466         else           printf("(none),          ");
 467         if (t[2] >= 0) trace_print_global(prog, st->o3.u1, t[2]);
 468         else           printf("(none)");
 469         printf("\n");
 470     }
 471     fflush(stdout);
 472 }
 473
 474 static qcint prog_enterfunction(qc_program *prog, prog_section_function *func)
 475 {
 476     qc_exec_stack st;
 477     size_t p, parampos;
 478
 479     /* back up locals */
 480     st.localsp  = prog->localstack_count;
 481     st.stmt     = prog->statement;
 482     st.function = func;
 483
 484 #ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
 485     if (prog->stack_count)
 486     {
 487         prog_section_function *cur;
 488         cur = prog->stack[prog->stack_count-1].function;
 489         if (cur)
 490         {
 491             qcint *globals = prog->globals + cur->firstlocal;
 492             if (!qc_program_localstack_append(prog, globals, cur->locals))
 493             {
 494                 printf("out of memory\n");
 495                 exit(1);
 496             }
 497         }
 498     }
 499 #else
 500     {
 501         qcint *globals = prog->globals + func->firstlocal;
 502         if (!qc_program_localstack_append(prog, globals, func->locals))
 503         {
 504             printf("out of memory\n");
 505             exit(1);
 506         }
 507     }
 508 #endif
 509
 510     /* copy parameters */
 511     parampos = func->firstlocal;
 512     for (p = 0; p < func->nargs; ++p)
 513     {
 514         size_t s;
 515         for (s = 0; s < func->argsize[p]; ++s) {
 516             prog->globals[parampos] = prog->globals[OFS_PARM0 + 3*p + s];
 517             ++parampos;
 518         }
 519     }
 520
 521     if (!qc_program_stack_add(prog, st)) {
 522         printf("out of memory\n");
 523         exit(1);
 524     }
 525
 526     return func->entry;
 527 }
 528
 529 static qcint prog_leavefunction(qc_program *prog)
 530 {
 531     prog_section_function *prev = NULL;
 532     size_t oldsp;
 533
 534     qc_exec_stack st = prog->stack[prog->stack_count-1];
 535
 536 #ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
 537     if (prog->stack_count > 1) {
 538         prev  = prog->stack[prog->stack_count-2].function;
 539         oldsp = prog->stack[prog->stack_count-2].localsp;
 540     }
 541 #else
 542     prev  = prog->stack[prog->stack_count-1].function;
 543     oldsp = prog->stack[prog->stack_count-1].localsp;
 544 #endif
 545     if (prev) {
 546         qcint *globals = prog->globals + prev->firstlocal;
 547         memcpy(globals, prog->localstack + oldsp, prev->locals);
 548         if (!qc_program_localstack_resize(prog, oldsp)) {
 549             printf("out of memory\n");
 550             exit(1);
 551         }
 552     }
 553
 554     if (!qc_program_stack_remove(prog, prog->stack_count-1)) {
 555         printf("out of memory\n");
 556         exit(1);
 557     }
 558
 559     return st.stmt - 1; /* offset the ++st */
 560 }
 561
 562 bool prog_exec(qc_program *prog, prog_section_function *func, size_t flags, long maxjumps)
 563 {
 564     long jumpcount = 0;
 565     prog_section_statement *st;
 566
 567     prog->vmerror = 0;
 568
 569     st = prog->code + prog_enterfunction(prog, func);
 570     --st;
 571     switch (flags)
 572     {
 573         default:
 574         case 0:
 575         {
 576 #define QCVM_PROFILE 0
 577 #define QCVM_TRACE   0
 578 #           include "execloop.h"
 579             break;
 580         }
 581         case (VMXF_TRACE):
 582         {
 583 #define QCVM_PROFILE 0
 584 #define QCVM_TRACE   1
 585 #           include "execloop.h"
 586             break;
 587         }
 588         case (VMXF_PROFILE):
 589         {
 590 #define QCVM_PROFILE 1
 591 #define QCVM_TRACE   0
 592 #           include "execloop.h"
 593             break;
 594         }
 595         case (VMXF_TRACE|VMXF_PROFILE):
 596         {
 597 #define QCVM_PROFILE 1
 598 #define QCVM_TRACE   1
 599 #           include "execloop.h"
 600             break;
 601         }
 602     };
 603
 604 cleanup:
 605     prog->localstack_count = 0;
 606     prog->stack_count = 0;
 607     if (prog->vmerror)
 608         return false;
 609     return true;
 610 }
 611
 612 /***********************************************************************
 613  * main for when building the standalone executor
 614  */
 615
 616 #if defined(QCVM_EXECUTOR)
 617 #include <math.h>
 618
 619 const char *type_name[TYPE_COUNT] = {
 620     "void",
 621     "string",
 622     "float",
 623     "vector",
 624     "entity",
 625     "field",
 626     "function",
 627     "pointer",
 628 #if 0
 629     "integer",
 630 #endif
 631     "variant"
 632 };
 633
 634 bool        opts_debug    = false;
 635 bool        opts_memchk   = false;
 636
 637 #define CheckArgs(num) do {                                                    \
 638     if (prog->argc != (num)) {                                                 \
 639         prog->vmerror++;                                                       \
 640         printf("ERROR: invalid number of arguments for %s: %i, expected %i\n", \
 641         __FUNCTION__, prog->argc, (num));                                      \
 642         return -1;                                                             \
 643     }                                                                          \
 644 } while (0)
 645
 646 #define GetGlobal(idx) ((qcany*)(prog->globals + (idx)))
 647 #define GetArg(num) GetGlobal(OFS_PARM0 + 3*(num))
 648 #define Return(any) *(GetGlobal(OFS_RETURN)) = (any)
 649
 650 static int qc_print(qc_program *prog)
 651 {
 652     size_t i;
 653     for (i = 0; i < prog->argc; ++i) {
 654         qcany *str = (qcany*)(prog->globals + OFS_PARM0 + 3*i);
 655         printf("%s", prog_getstring(prog, str->string));
 656     }
 657     return 0;
 658 }
 659
 660 static int qc_error(qc_program *prog)
 661 {
 662     printf("*** VM raised an error:\n");
 663     qc_print(prog);
 664     prog->vmerror++;
 665     return -1;
 666 }
 667
 668 static int qc_ftos(qc_program *prog)
 669 {
 670     char buffer[512];
 671     qcany *num;
 672     qcany str;
 673     CheckArgs(1);
 674     num = GetArg(0);
 675     snprintf(buffer, sizeof(buffer), "%g", num->_float);
 676     str.string = prog_tempstring(prog, buffer);
 677     Return(str);
 678     return 0;
 679 }
 680
 681 static int qc_vtos(qc_program *prog)
 682 {
 683     char buffer[512];
 684     qcany *num;
 685     qcany str;
 686     CheckArgs(1);
 687     num = GetArg(0);
 688     snprintf(buffer, sizeof(buffer), "`%g %g %g`", num->vector[0], num->vector[1], num->vector[2]);
 689     str.string = prog_tempstring(prog, buffer);
 690     Return(str);
 691     return 0;
 692 }
 693
 694 static int qc_spawn(qc_program *prog)
 695 {
 696     qcany ent;
 697     CheckArgs(0);
 698     ent.edict = prog_spawn_entity(prog);
 699     Return(ent);
 700     return (ent.edict ? 0 : -1);
 701 }
 702
 703 static int qc_kill(qc_program *prog)
 704 {
 705     qcany *ent;
 706     CheckArgs(1);
 707     ent = GetArg(0);
 708     prog_free_entity(prog, ent->edict);
 709     return 0;
 710 }
 711
 712 static int qc_vlen(qc_program *prog)
 713 {
 714     qcany *vec, len;
 715     CheckArgs(1);
 716     vec = GetArg(0);
 717     len._float = sqrt(vec->vector[0] * vec->vector[0] +
 718                       vec->vector[1] * vec->vector[1] +
 719                       vec->vector[2] * vec->vector[2]);
 720     Return(len);
 721     return 0;
 722 }
 723
 724 static prog_builtin qc_builtins[] = {
 725     NULL,
 726     &qc_print, /*   1   */
 727     &qc_ftos,  /*   2   */
 728     &qc_spawn, /*   3   */
 729     &qc_kill,  /*   4   */
 730     &qc_vtos,  /*   5   */
 731     &qc_error, /*   6   */
 732     &qc_vlen   /*   7   */
 733 };
 734 static size_t qc_builtins_count = sizeof(qc_builtins) / sizeof(qc_builtins[0]);
 735
 736 static const char *arg0 = NULL;
 737
 738 void usage()
 739 {
 740     printf("usage: [-debug] %s file\n", arg0);
 741     exit(1);
 742 }
 743
 744 int main(int argc, char **argv)
 745 {
 746     size_t      i;
 747     qcint       fnmain = -1;
 748     qc_program *prog;
 749     size_t      xflags = VMXF_DEFAULT;
 750     bool        opts_printfields = false;
 751     bool        opts_printdefs   = false;
 752     bool        opts_info  = false;
 753
 754     arg0 = argv[0];
 755
 756     if (argc < 2)
 757         usage();
 758
 759     while (argc > 2) {
 760         if (!strcmp(argv[1], "-trace")) {
 761             --argc;
 762             ++argv;
 763             xflags |= VMXF_TRACE;
 764         }
 765         else if (!strcmp(argv[1], "-profile")) {
 766             --argc;
 767             ++argv;
 768             xflags |= VMXF_PROFILE;
 769         }
 770         else if (!strcmp(argv[1], "-info")) {
 771             --argc;
 772             ++argv;
 773             opts_info = true;
 774         }
 775         else if (!strcmp(argv[1], "-printdefs")) {
 776             --argc;
 777             ++argv;
 778             opts_printdefs = true;
 779         }
 780         else if (!strcmp(argv[1], "-printfields")) {
 781             --argc;
 782             ++argv;
 783             opts_printfields = true;
 784         }
 785         else
 786             usage();
 787     }
 788
 789
 790     prog = prog_load(argv[1]);
 791     if (!prog) {
 792         printf("failed to load program '%s'\n", argv[1]);
 793         exit(1);
 794     }
 795
 796     prog->builtins       = qc_builtins;
 797     prog->builtins_count = qc_builtins_count;
 798     prog->builtins_alloc = 0;
 799
 800     printf("Program's system-checksum = 0x%04x\n", (int)prog->crc16);
 801
 802     for (i = 1; i < prog->functions_count; ++i) {
 803         const char *name = prog_getstring(prog, prog->functions[i].name);
 804         /* printf("Found function: %s\n", name); */
 805         if (!strcmp(name, "main"))
 806             fnmain = (qcint)i;
 807     }
 808     printf("Entity field space: %i\n", (int)prog->entityfields);
 809     if (opts_info) {
 810         prog_delete(prog);
 811         return 0;
 812     }
 813     if (opts_printdefs) {
 814         for (i = 0; i < prog->defs_count; ++i) {
 815             printf("Global: %8s %-16s at %u\n",
 816                    type_name[prog->defs[i].type],
 817                    prog_getstring(prog, prog->defs[i].name),
 818                    (unsigned int)prog->defs[i].offset);
 819         }
 820     }
 821     else if (opts_printfields) {
 822         for (i = 0; i < prog->fields_count; ++i) {
 823             printf("Field: %8s %-16s at %u\n",
 824                    type_name[prog->fields[i].type],
 825                    prog_getstring(prog, prog->fields[i].name),
 826                    (unsigned int)prog->fields[i].offset);
 827         }
 828     }
 829     else
 830     {
 831         if (fnmain > 0)
 832         {
 833             prog_exec(prog, &prog->functions[fnmain], xflags, VM_JUMPS_DEFAULT);
 834         }
 835         else
 836             printf("No main function found\n");
 837     }
 838
 839     prog_delete(prog);
 840     return 0;
 841 }
 842 #endif