+ size_t i;
+ for (i = 0; i < vec_size(prog->defs); ++i) {
+ if (prog->defs[i].offset == off)
+ return (prog->defs + i);
+ }
+ return NULL;
+}
+
+qcany* prog_getedict(qc_program *prog, qcint e) {
+ if (e >= (qcint)vec_size(prog->entitypool)) {
+ prog->vmerror++;
+ fprintf(stderr, "Accessing out of bounds edict %i\n", (int)e);
+ e = 0;
+ }
+ return (qcany*)(prog->entitydata + (prog->entityfields * e));
+}
+
+qcint prog_spawn_entity(qc_program *prog) {
+ char *data;
+ qcint e;
+ for (e = 0; e < (qcint)vec_size(prog->entitypool); ++e) {
+ if (!prog->entitypool[e]) {
+ data = (char*)(prog->entitydata + (prog->entityfields * e));
+ memset(data, 0, prog->entityfields * sizeof(qcint));
+ return e;
+ }
+ }
+ vec_push(prog->entitypool, true);
+ prog->entities++;
+ data = (char*)vec_add(prog->entitydata, prog->entityfields);
+ memset(data, 0, prog->entityfields * sizeof(qcint));
+ return e;
+}
+
+void prog_free_entity(qc_program *prog, qcint e) {
+ if (!e) {
+ prog->vmerror++;
+ fprintf(stderr, "Trying to free world entity\n");
+ return;
+ }
+ if (e >= (qcint)vec_size(prog->entitypool)) {
+ prog->vmerror++;
+ fprintf(stderr, "Trying to free out of bounds entity\n");
+ return;
+ }
+ if (!prog->entitypool[e]) {
+ prog->vmerror++;
+ fprintf(stderr, "Double free on entity\n");
+ return;
+ }
+ prog->entitypool[e] = false;
+}
+
+qcint prog_tempstring(qc_program *prog, const char *str) {
+ size_t len = strlen(str);
+ size_t at = prog->tempstring_at;
+
+ /* when we reach the end we start over */
+ if (at + len >= vec_size(prog->strings))
+ at = prog->tempstring_start;
+
+ /* when it doesn't fit, reallocate */
+ if (at + len >= vec_size(prog->strings))
+ {
+ (void)vec_add(prog->strings, len+1);
+ memcpy(prog->strings + at, str, len+1);
+ return at;
+ }
+
+ /* when it fits, just copy */
+ memcpy(prog->strings + at, str, len+1);
+ prog->tempstring_at += len+1;
+ return at;
+}
+
+static size_t print_escaped_string(const char *str, size_t maxlen) {
+ size_t len = 2;
+ putchar('"');
+ --maxlen; /* because we're lazy and have escape sequences */
+ while (*str) {
+ if (len >= maxlen) {
+ putchar('.');
+ putchar('.');
+ putchar('.');
+ len += 3;
+ break;
+ }
+ switch (*str) {
+ case '\a': len += 2; putchar('\\'); putchar('a'); break;
+ case '\b': len += 2; putchar('\\'); putchar('b'); break;
+ case '\r': len += 2; putchar('\\'); putchar('r'); break;
+ case '\n': len += 2; putchar('\\'); putchar('n'); break;
+ case '\t': len += 2; putchar('\\'); putchar('t'); break;
+ case '\f': len += 2; putchar('\\'); putchar('f'); break;
+ case '\v': len += 2; putchar('\\'); putchar('v'); break;
+ case '\\': len += 2; putchar('\\'); putchar('\\'); break;
+ case '"': len += 2; putchar('\\'); putchar('"'); break;
+ default:
+ ++len;
+ putchar(*str);
+ break;
+ }
+ ++str;
+ }
+ putchar('"');
+ return len;
+}
+
+static void trace_print_global(qc_program *prog, unsigned int glob, int vtype) {
+ static char spaces[28+1] = " ";
+ prog_section_def *def;
+ qcany *value;
+ int len;
+
+ if (!glob) {
+ if ((len = printf("<null>,")) == -1)
+ len = 0;
+
+ goto done;
+ }
+
+ def = prog_getdef(prog, glob);
+ value = (qcany*)(&prog->globals[glob]);
+
+ len = printf("[@%u] ", glob);
+ if (def) {
+ const char *name = prog_getstring(prog, def->name);
+ if (name[0] == '#')
+ len += printf("$");
+ else
+ len += printf("%s ", name);
+ vtype = def->type & DEF_TYPEMASK;
+ }
+
+ switch (vtype) {
+ case TYPE_VOID:
+ case TYPE_ENTITY:
+ case TYPE_FIELD:
+ case TYPE_FUNCTION:
+ case TYPE_POINTER:
+ len += printf("(%i),", value->_int);
+ break;
+ case TYPE_VECTOR:
+ len += printf("'%g %g %g',", value->vector[0],
+ value->vector[1],
+ value->vector[2]);
+ break;
+ case TYPE_STRING:
+ if (value->string)
+ len += print_escaped_string(prog_getstring(prog, value->string), sizeof(spaces)-len-5);
+ else
+ len += printf("(null)");
+ len += printf(",");
+ /* len += printf("\"%s\",", prog_getstring(prog, value->string)); */
+ break;
+ case TYPE_FLOAT:
+ default:
+ len += printf("%g,", value->_float);
+ break;
+ }
+done:
+ if (len < (int)sizeof(spaces)-1) {
+ spaces[sizeof(spaces)-1-len] = 0;
+ fs_file_puts(stdout, spaces);
+ spaces[sizeof(spaces)-1-len] = ' ';
+ }
+}
+
+static void prog_print_statement(qc_program *prog, prog_section_statement *st) {
+ if (st->opcode >= (sizeof(asm_instr)/sizeof(asm_instr[0]))) {
+ printf("<illegal instruction %d>\n", st->opcode);
+ return;
+ }
+ if ((prog->xflags & VMXF_TRACE) && vec_size(prog->function_stack)) {
+ size_t i;
+ for (i = 0; i < vec_size(prog->function_stack); ++i)
+ printf("->");
+ printf("%s:", vec_last(prog->function_stack));
+ }
+ printf(" <> %-12s", asm_instr[st->opcode].m);
+ if (st->opcode >= INSTR_IF &&
+ st->opcode <= INSTR_IFNOT)
+ {
+ trace_print_global(prog, st->o1.u1, TYPE_FLOAT);
+ printf("%d\n", st->o2.s1);
+ }
+ else if (st->opcode >= INSTR_CALL0 &&
+ st->opcode <= INSTR_CALL8)
+ {
+ trace_print_global(prog, st->o1.u1, TYPE_FUNCTION);
+ printf("\n");
+ }
+ else if (st->opcode == INSTR_GOTO)
+ {
+ printf("%i\n", st->o1.s1);
+ }
+ else
+ {
+ int t[3] = { TYPE_FLOAT, TYPE_FLOAT, TYPE_FLOAT };
+ switch (st->opcode)
+ {
+ case INSTR_MUL_FV:
+ t[1] = t[2] = TYPE_VECTOR;
+ break;
+ case INSTR_MUL_VF:
+ t[0] = t[2] = TYPE_VECTOR;
+ break;
+ case INSTR_MUL_V:
+ t[0] = t[1] = TYPE_VECTOR;
+ break;
+ case INSTR_ADD_V:
+ case INSTR_SUB_V:
+ case INSTR_EQ_V:
+ case INSTR_NE_V:
+ t[0] = t[1] = t[2] = TYPE_VECTOR;
+ break;
+ case INSTR_EQ_S:
+ case INSTR_NE_S:
+ t[0] = t[1] = TYPE_STRING;
+ break;
+ case INSTR_STORE_F:
+ case INSTR_STOREP_F:
+ t[2] = -1;
+ break;
+ case INSTR_STORE_V:
+ t[0] = t[1] = TYPE_VECTOR; t[2] = -1;
+ break;
+ case INSTR_STORE_S:
+ t[0] = t[1] = TYPE_STRING; t[2] = -1;
+ break;
+ case INSTR_STORE_ENT:
+ t[0] = t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ case INSTR_STORE_FLD:
+ t[0] = t[1] = TYPE_FIELD; t[2] = -1;
+ break;
+ case INSTR_STORE_FNC:
+ t[0] = t[1] = TYPE_FUNCTION; t[2] = -1;
+ break;
+ case INSTR_STOREP_V:
+ t[0] = TYPE_VECTOR; t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ case INSTR_STOREP_S:
+ t[0] = TYPE_STRING; t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ case INSTR_STOREP_ENT:
+ t[0] = TYPE_ENTITY; t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ case INSTR_STOREP_FLD:
+ t[0] = TYPE_FIELD; t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ case INSTR_STOREP_FNC:
+ t[0] = TYPE_FUNCTION; t[1] = TYPE_ENTITY; t[2] = -1;
+ break;
+ }
+ if (t[0] >= 0) trace_print_global(prog, st->o1.u1, t[0]);
+ else printf("(none), ");
+ if (t[1] >= 0) trace_print_global(prog, st->o2.u1, t[1]);
+ else printf("(none), ");
+ if (t[2] >= 0) trace_print_global(prog, st->o3.u1, t[2]);
+ else printf("(none)");
+ printf("\n");
+ }
+}
+
+static qcint prog_enterfunction(qc_program *prog, prog_section_function *func) {
+ qc_exec_stack st;
+ size_t parampos;
+ int32_t p;
+
+ /* back up locals */
+ st.localsp = vec_size(prog->localstack);
+ st.stmt = prog->statement;
+ st.function = func;
+
+ if (prog->xflags & VMXF_TRACE) {
+ const char *str = prog_getstring(prog, func->name);
+ vec_push(prog->function_stack, str);
+ }
+
+#ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
+ if (vec_size(prog->stack))
+ {
+ prog_section_function *cur;
+ cur = prog->stack[vec_size(prog->stack)-1].function;
+ if (cur)
+ {
+ qcint *globals = prog->globals + cur->firstlocal;
+ vec_append(prog->localstack, cur->locals, globals);
+ }
+ }
+#else
+ {
+ qcint *globals = prog->globals + func->firstlocal;
+ vec_append(prog->localstack, func->locals, globals);
+ }
+#endif
+
+ /* copy parameters */
+ parampos = func->firstlocal;
+ for (p = 0; p < func->nargs; ++p)
+ {
+ size_t s;
+ for (s = 0; s < func->argsize[p]; ++s) {
+ prog->globals[parampos] = prog->globals[OFS_PARM0 + 3*p + s];
+ ++parampos;
+ }
+ }
+
+ vec_push(prog->stack, st);
+
+ return func->entry;
+}
+
+static qcint prog_leavefunction(qc_program *prog) {
+ prog_section_function *prev = NULL;
+ size_t oldsp;
+
+ qc_exec_stack st = vec_last(prog->stack);
+
+ if (prog->xflags & VMXF_TRACE) {
+ if (vec_size(prog->function_stack))
+ vec_pop(prog->function_stack);
+ }
+
+#ifdef QCVM_BACKUP_STRATEGY_CALLER_VARS
+ if (vec_size(prog->stack) > 1) {
+ prev = prog->stack[vec_size(prog->stack)-2].function;
+ oldsp = prog->stack[vec_size(prog->stack)-2].localsp;
+ }
+#else
+ prev = prog->stack[vec_size(prog->stack)-1].function;
+ oldsp = prog->stack[vec_size(prog->stack)-1].localsp;
+#endif
+ if (prev) {
+ qcint *globals = prog->globals + prev->firstlocal;
+ memcpy(globals, prog->localstack + oldsp, prev->locals * sizeof(prog->localstack[0]));
+ /* vec_remove(prog->localstack, oldsp, vec_size(prog->localstack)-oldsp); */
+ vec_shrinkto(prog->localstack, oldsp);
+ }
+
+ vec_pop(prog->stack);
+
+ return st.stmt - 1; /* offset the ++st */
+}
+
+bool prog_exec(qc_program *prog, prog_section_function *func, size_t flags, long maxjumps) {
+ long jumpcount = 0;
+ size_t oldxflags = prog->xflags;
+ prog_section_statement *st;
+
+ prog->vmerror = 0;
+ prog->xflags = flags;
+
+ st = prog->code + prog_enterfunction(prog, func);
+ --st;
+ switch (flags)
+ {
+ default:
+ case 0:
+ {
+#define QCVM_LOOP 1
+#define QCVM_PROFILE 0
+#define QCVM_TRACE 0
+# include __FILE__
+ }
+ case (VMXF_TRACE):
+ {
+#define QCVM_PROFILE 0
+#define QCVM_TRACE 1
+# include __FILE__
+ }
+ case (VMXF_PROFILE):
+ {
+#define QCVM_PROFILE 1
+#define QCVM_TRACE 0
+# include __FILE__
+ }
+ case (VMXF_TRACE|VMXF_PROFILE):
+ {
+#define QCVM_PROFILE 1
+#define QCVM_TRACE 1
+# include __FILE__
+ }
+ };
+
+cleanup:
+ prog->xflags = oldxflags;
+ vec_free(prog->localstack);
+ vec_free(prog->stack);
+ if (prog->vmerror)
+ return false;
+ return true;
+}
+
+/***********************************************************************
+ * main for when building the standalone executor
+ */
+
+#if defined(QCVM_EXECUTOR)
+#include <math.h>
+
+opts_cmd_t opts;
+
+const char *type_name[TYPE_COUNT] = {
+ "void",
+ "string",
+ "float",
+ "vector",
+ "entity",
+ "field",
+ "function",
+ "pointer",
+ "integer",
+
+ "variant",
+
+ "struct",
+ "union",
+ "array",
+
+ "nil",
+ "noexpr"
+};
+
+typedef struct {
+ int vtype;
+ const char *value;
+} qcvm_parameter;
+
+static qcvm_parameter *main_params = NULL;
+
+#define CheckArgs(num) do { \
+ if (prog->argc != (num)) { \
+ prog->vmerror++; \
+ fprintf(stderr, "ERROR: invalid number of arguments for %s: %i, expected %i\n", \
+ __FUNCTION__, prog->argc, (num)); \
+ return -1; \
+ } \
+} while (0)
+
+#define GetGlobal(idx) ((qcany*)(prog->globals + (idx)))
+#define GetArg(num) GetGlobal(OFS_PARM0 + 3*(num))
+#define Return(any) *(GetGlobal(OFS_RETURN)) = (any)
+
+static int qc_print(qc_program *prog) {
+ size_t i;
+ const char *laststr = NULL;
+ for (i = 0; i < (size_t)prog->argc; ++i) {
+ qcany *str = (qcany*)(prog->globals + OFS_PARM0 + 3*i);
+ laststr = prog_getstring(prog, str->string);
+ printf("%s", laststr);
+ }
+ if (laststr && (prog->xflags & VMXF_TRACE)) {
+ size_t len = strlen(laststr);
+ if (!len || laststr[len-1] != '\n')
+ printf("\n");
+ }
+ return 0;
+}
+
+static int qc_error(qc_program *prog) {
+ fprintf(stderr, "*** VM raised an error:\n");
+ qc_print(prog);
+ prog->vmerror++;
+ return -1;
+}
+
+static int qc_ftos(qc_program *prog) {
+ char buffer[512];
+ qcany *num;
+ qcany str;
+ CheckArgs(1);
+ num = GetArg(0);
+ util_snprintf(buffer, sizeof(buffer), "%g", num->_float);
+ str.string = prog_tempstring(prog, buffer);
+ Return(str);
+ return 0;
+}
+
+static int qc_stof(qc_program *prog) {
+ qcany *str;
+ qcany num;
+ CheckArgs(1);
+ str = GetArg(0);
+ num._float = (float)strtod(prog_getstring(prog, str->string), NULL);
+ Return(num);
+ return 0;
+}
+
+static int qc_vtos(qc_program *prog) {
+ char buffer[512];
+ qcany *num;
+ qcany str;
+ CheckArgs(1);
+ num = GetArg(0);
+ util_snprintf(buffer, sizeof(buffer), "'%g %g %g'", num->vector[0], num->vector[1], num->vector[2]);
+ str.string = prog_tempstring(prog, buffer);
+ Return(str);
+ return 0;
+}
+
+static int qc_etos(qc_program *prog) {
+ char buffer[512];
+ qcany *num;
+ qcany str;
+ CheckArgs(1);
+ num = GetArg(0);
+ util_snprintf(buffer, sizeof(buffer), "%i", num->_int);
+ str.string = prog_tempstring(prog, buffer);
+ Return(str);
+ return 0;
+}
+
+static int qc_spawn(qc_program *prog) {
+ qcany ent;
+ CheckArgs(0);
+ ent.edict = prog_spawn_entity(prog);
+ Return(ent);
+ return (ent.edict ? 0 : -1);
+}
+
+static int qc_kill(qc_program *prog) {
+ qcany *ent;
+ CheckArgs(1);
+ ent = GetArg(0);
+ prog_free_entity(prog, ent->edict);
+ return 0;
+}
+
+static int qc_sqrt(qc_program *prog) {
+ qcany *num, out;
+ CheckArgs(1);
+ num = GetArg(0);
+ out._float = sqrt(num->_float);
+ Return(out);
+ return 0;
+}
+
+static int qc_vlen(qc_program *prog) {
+ qcany *vec, len;
+ CheckArgs(1);
+ vec = GetArg(0);
+ len._float = sqrt(vec->vector[0] * vec->vector[0] +
+ vec->vector[1] * vec->vector[1] +
+ vec->vector[2] * vec->vector[2]);
+ Return(len);
+ return 0;
+}
+
+static int qc_normalize(qc_program *prog) {
+ double len;
+ qcany *vec;
+ qcany out;
+ CheckArgs(1);
+ vec = GetArg(0);
+ len = sqrt(vec->vector[0] * vec->vector[0] +
+ vec->vector[1] * vec->vector[1] +
+ vec->vector[2] * vec->vector[2]);
+ if (len)
+ len = 1.0 / len;
+ else
+ len = 0;
+ out.vector[0] = len * vec->vector[0];
+ out.vector[1] = len * vec->vector[1];
+ out.vector[2] = len * vec->vector[2];
+ Return(out);
+ return 0;
+}
+
+static int qc_strcat(qc_program *prog) {
+ char *buffer;
+ size_t len1, len2;
+ qcany *str1, *str2;
+ qcany out;
+
+ const char *cstr1;
+ const char *cstr2;
+
+ CheckArgs(2);
+ str1 = GetArg(0);
+ str2 = GetArg(1);
+ cstr1 = prog_getstring(prog, str1->string);
+ cstr2 = prog_getstring(prog, str2->string);
+ len1 = strlen(cstr1);
+ len2 = strlen(cstr2);
+ buffer = (char*)mem_a(len1 + len2 + 1);
+ memcpy(buffer, cstr1, len1);
+ memcpy(buffer+len1, cstr2, len2+1);
+ out.string = prog_tempstring(prog, buffer);
+ mem_d(buffer);
+ Return(out);
+ return 0;
+}
+
+static int qc_strcmp(qc_program *prog) {
+ qcany *str1, *str2;
+ qcany out;
+
+ const char *cstr1;
+ const char *cstr2;
+
+ if (prog->argc != 2 && prog->argc != 3) {
+ fprintf(stderr, "ERROR: invalid number of arguments for strcmp/strncmp: %i, expected 2 or 3\n",
+ prog->argc);
+ return -1;
+ }
+
+ str1 = GetArg(0);
+ str2 = GetArg(1);
+ cstr1 = prog_getstring(prog, str1->string);
+ cstr2 = prog_getstring(prog, str2->string);
+ if (prog->argc == 3)
+ out._float = strncmp(cstr1, cstr2, GetArg(2)->_float);
+ else
+ out._float = strcmp(cstr1, cstr2);
+ Return(out);
+ return 0;
+}
+
+static int qc_floor(qc_program *prog) {
+ qcany *num, out;
+ CheckArgs(1);
+ num = GetArg(0);
+ out._float = floor(num->_float);
+ Return(out);
+ return 0;
+}
+
+static prog_builtin qc_builtins[] = {
+ NULL,
+ &qc_print, /* 1 */
+ &qc_ftos, /* 2 */
+ &qc_spawn, /* 3 */
+ &qc_kill, /* 4 */
+ &qc_vtos, /* 5 */
+ &qc_error, /* 6 */
+ &qc_vlen, /* 7 */
+ &qc_etos, /* 8 */
+ &qc_stof, /* 9 */
+ &qc_strcat, /* 10 */
+ &qc_strcmp, /* 11 */
+ &qc_normalize, /* 12 */
+ &qc_sqrt, /* 13 */
+ &qc_floor /* 14 */
+};
+static size_t qc_builtins_count = sizeof(qc_builtins) / sizeof(qc_builtins[0]);
+
+static const char *arg0 = NULL;
+
+static void version(void) {
+ printf("GMQCC-QCVM %d.%d.%d Built %s %s\n",
+ GMQCC_VERSION_MAJOR,
+ GMQCC_VERSION_MINOR,
+ GMQCC_VERSION_PATCH,
+ __DATE__,
+ __TIME__
+ );
+}
+
+static void usage(void) {
+ printf("usage: %s [options] [parameters] file\n", arg0);
+ printf("options:\n");
+ printf(" -h, --help print this message\n"
+ " -trace trace the execution\n"
+ " -profile perform profiling during execution\n"
+ " -info print information from the prog's header\n"
+ " -disasm disassemble and exit\n"
+ " -disasm-func func disassemble and exit\n"
+ " -printdefs list the defs section\n"
+ " -printfields list the field section\n"
+ " -printfuns list functions information\n"
+ " -v be verbose\n"
+ " -vv be even more verbose\n");
+ printf("parameters:\n");
+ printf(" -vector <V> pass a vector parameter to main()\n"
+ " -float <f> pass a float parameter to main()\n"
+ " -string <s> pass a string parameter to main() \n");
+}
+
+static void prog_main_setparams(qc_program *prog) {
+ size_t i;
+ qcany *arg;
+
+ for (i = 0; i < vec_size(main_params); ++i) {
+ arg = GetGlobal(OFS_PARM0 + 3*i);
+ arg->vector[0] = 0;
+ arg->vector[1] = 0;
+ arg->vector[2] = 0;
+ switch (main_params[i].vtype) {
+ case TYPE_VECTOR:
+#ifdef _MSC_VER
+ (void)sscanf_s(main_params[i].value, " %f %f %f ",
+ &arg->vector[0],
+ &arg->vector[1],
+ &arg->vector[2]);
+#else
+ (void)sscanf(main_params[i].value, " %f %f %f ",
+ &arg->vector[0],
+ &arg->vector[1],
+ &arg->vector[2]);
+#endif
+ break;
+ case TYPE_FLOAT:
+ arg->_float = atof(main_params[i].value);
+ break;
+ case TYPE_STRING:
+ arg->string = prog_tempstring(prog, main_params[i].value);
+ break;
+ default:
+ fprintf(stderr, "error: unhandled parameter type: %i\n", main_params[i].vtype);
+ break;
+ }
+ }
+}
+
+void escapestring(char* dest, const char* src) {
+ char c;
+ while ((c = *(src++))) {
+ switch(c) {
+ case '\t':
+ *(dest++) = '\\', *(dest++) = 't';
+ break;
+ case '\n':
+ *(dest++) = '\\', *(dest++) = 'n';
+ break;
+ case '\r':
+ *(dest++) = '\\', *(dest++) = 'r';
+ break;
+ case '\\':
+ *(dest++) = '\\', *(dest++) = '\\';
+ break;
+ case '\"':
+ *(dest++) = '\\', *(dest++) = '\"';
+ break;
+ default:
+ *(dest++) = c;
+ }
+ }
+ *dest = '\0';
+}
+
+void prog_disasm_function(qc_program *prog, size_t id);
+
+int main(int argc, char **argv) {
+ size_t i;
+ qcint fnmain = -1;