ref: 299dbda156c8a8cb05775d9260e78c056a507ccf
dir: /jscompile.c/
#include "js.h" #include "jsparse.h" #include "jscompile.h" #include "jsstate.h" #define cexp js_cexp /* collision with math.h */ #define JF js_State *J, js_Function *F JS_NORETURN int jsC_error(js_State *J, js_Ast *node, const char *fmt, ...); static void cfunbody(JF, js_Ast *name, js_Ast *params, js_Ast *body); static void cexp(JF, js_Ast *exp); static void cstmlist(JF, js_Ast *list); static js_Function *newfun(js_State *J, js_Ast *name, js_Ast *params, js_Ast *body) { js_Function *F = malloc(sizeof *F); memset(F, 0, sizeof *F); F->filename = js_intern(J, J->filename); F->line = name ? name->line : params ? params->line : body->line; F->next = J->fun; J->fun = F; cfunbody(J, F, name, params, body); return F; } static void freefun(js_State *J, js_Function *F) { free(F->funtab); free(F->numtab); free(F->strtab); free(F->code); free(F); } /* Emit opcodes, constants and jumps */ static void emit(JF, int value) { if (F->codelen >= F->codecap) { F->codecap = F->codecap ? F->codecap * 2 : 64; F->code = realloc(F->code, F->codecap * sizeof *F->code); } F->code[F->codelen++] = value; } static int addfunction(JF, js_Function *value) { if (F->funlen >= F->funcap) { F->funcap = F->funcap ? F->funcap * 2 : 16; F->funtab = realloc(F->funtab, F->funcap * sizeof *F->funtab); } F->funtab[F->funlen] = value; return F->funlen++; } static int addnumber(JF, double value) { int i; for (i = 0; i < F->numlen; ++i) if (F->numtab[i] == value) return i; if (F->numlen >= F->numcap) { F->numcap = F->numcap ? F->numcap * 2 : 16; F->numtab = realloc(F->numtab, F->numcap * sizeof *F->numtab); } F->numtab[F->numlen] = value; return F->numlen++; } static int addstring(JF, const char *value) { int i; for (i = 0; i < F->strlen; ++i) if (!strcmp(F->strtab[i], value)) return i; if (F->strlen >= F->strcap) { F->strcap = F->strcap ? F->strcap * 2 : 16; F->strtab = realloc(F->strtab, F->strcap * sizeof *F->strtab); } F->strtab[F->strlen] = value; return F->strlen++; } static void emitfunction(JF, js_Function *fun) { emit(J, F, OP_CLOSURE); emit(J, F, addfunction(J, F, fun)); } static void emitnumber(JF, double num) { if (num == 0) emit(J, F, OP_NUMBER_0); else if (num == 1) emit(J, F, OP_NUMBER_1); else if (num == (short)num) { emit(J, F, OP_NUMBER_X); emit(J, F, (short)num); } else { emit(J, F, OP_NUMBER); emit(J, F, addnumber(J, F, num)); } } static void emitstring(JF, int opcode, const char *str) { emit(J, F, opcode); emit(J, F, addstring(J, F, str)); } static int here(JF) { return F->codelen; } static int jump(JF, int opcode) { int inst = F->codelen + 1; emit(J, F, opcode); emit(J, F, 0); return inst; } static void jumpto(JF, int opcode, int dest) { emit(J, F, opcode); emit(J, F, dest); } static void label(JF, int inst) { F->code[inst] = F->codelen; } /* Expressions */ static void cunary(JF, js_Ast *exp, int opcode) { cexp(J, F, exp->a); emit(J, F, opcode); } static void cbinary(JF, js_Ast *exp, int opcode) { cexp(J, F, exp->a); cexp(J, F, exp->b); emit(J, F, opcode); } static void carray(JF, js_Ast *list) { int i = 0; while (list) { if (list->a->type != EXP_UNDEF) { emit(J, F, OP_DUP); emit(J, F, OP_NUMBER_X); emit(J, F, i++); cexp(J, F, list->a); emit(J, F, OP_SETPROP); emit(J, F, OP_POP); } else { ++i; } list = list->b; } } static void cobject(JF, js_Ast *list) { while (list) { js_Ast *kv = list->a; if (kv->type == EXP_PROP_VAL) { js_Ast *prop = kv->a; emit(J, F, OP_DUP); if (prop->type == AST_IDENTIFIER || prop->type == AST_STRING) emitstring(J, F, OP_STRING, prop->string); else if (prop->type == AST_NUMBER) emitnumber(J, F, prop->number); else jsC_error(J, list, "illegal property name in object initializer"); cexp(J, F, kv->b); emit(J, F, OP_SETPROP); emit(J, F, OP_POP); } // TODO: set, get list = list->b; } } static int cargs(JF, js_Ast *list) { int n = 0; while (list) { cexp(J, F, list->a); list = list->b; ++n; } return n; } static void cassign(JF, js_Ast *lhs, js_Ast *rhs) { switch (lhs->type) { case AST_IDENTIFIER: cexp(J, F, rhs); emitstring(J, F, OP_SETVAR, lhs->string); break; case EXP_INDEX: cexp(J, F, lhs->a); cexp(J, F, lhs->b); cexp(J, F, rhs); emit(J, F, OP_SETPROP); break; case EXP_MEMBER: cexp(J, F, lhs->a); emitstring(J, F, OP_STRING, lhs->b->string); cexp(J, F, rhs); emit(J, F, OP_SETPROP); break; default: jsC_error(J, lhs, "invalid l-value in assignment"); break; } } static void cassignloop(JF, js_Ast *lhs) { switch (lhs->type) { case AST_IDENTIFIER: emitstring(J, F, OP_SETVAR, lhs->string); break; case EXP_INDEX: cexp(J, F, lhs->a); cexp(J, F, lhs->b); emit(J, F, OP_ROT3); emit(J, F, OP_SETPROP); break; case EXP_MEMBER: cexp(J, F, lhs->a); emitstring(J, F, OP_STRING, lhs->b->string); emit(J, F, OP_ROT3); emit(J, F, OP_SETPROP); break; default: jsC_error(J, lhs, "invalid l-value in assignment"); break; } } static void cassignop1(JF, js_Ast *lhs, int dup) { switch (lhs->type) { case AST_IDENTIFIER: emitstring(J, F, OP_GETVAR, lhs->string); if (dup) emit(J, F, OP_DUP); break; case EXP_INDEX: cexp(J, F, lhs->a); cexp(J, F, lhs->b); emit(J, F, OP_DUP2); emit(J, F, OP_GETPROP); if (dup) emit(J, F, OP_DUP1ROT4); break; case EXP_MEMBER: cexp(J, F, lhs->a); emitstring(J, F, OP_STRING, lhs->b->string); emit(J, F, OP_DUP2); emit(J, F, OP_GETPROP); if (dup) emit(J, F, OP_DUP1ROT4); break; default: jsC_error(J, lhs, "invalid l-value in assignment"); break; } } static void cassignop2(JF, js_Ast *lhs) { switch (lhs->type) { case AST_IDENTIFIER: emitstring(J, F, OP_SETVAR, lhs->string); break; case EXP_INDEX: case EXP_MEMBER: case EXP_CALL: emit(J, F, OP_SETPROP); break; default: jsC_error(J, lhs, "invalid l-value in assignment"); break; } } static void cassignop(JF, js_Ast *lhs, js_Ast *rhs, int opcode) { cassignop1(J, F, lhs, 0); cexp(J, F, rhs); emit(J, F, opcode); cassignop2(J, F, lhs); } static void cdelete(JF, js_Ast *exp) { switch (exp->type) { case AST_IDENTIFIER: emitstring(J, F, OP_DELVAR, exp->string); break; case EXP_INDEX: cexp(J, F, exp->a); cexp(J, F, exp->b); emit(J, F, OP_DELPROP); break; case EXP_MEMBER: cexp(J, F, exp->a); emitstring(J, F, OP_STRING, exp->b->string); emit(J, F, OP_DELPROP); break; default: jsC_error(J, exp, "invalid l-value in delete expression"); break; } } static void cvarinit(JF, js_Ast *list) { while (list) { js_Ast *var = list->a; if (var->b) { cexp(J, F, var->b); emitstring(J, F, OP_SETVAR, var->a->string); emit(J, F, OP_POP); } list = list->b; } } static void ccall(JF, js_Ast *fun, js_Ast *args) { int n; switch (fun->type) { case EXP_INDEX: cexp(J, F, fun->a); emit(J, F, OP_DUP); cexp(J, F, fun->b); emit(J, F, OP_GETPROP); emit(J, F, OP_ROT2); break; case EXP_MEMBER: cexp(J, F, fun->a); emit(J, F, OP_DUP); emitstring(J, F, OP_STRING, fun->b->string); emit(J, F, OP_GETPROP); emit(J, F, OP_ROT2); break; default: cexp(J, F, fun); emit(J, F, OP_GLOBAL); break; } n = cargs(J, F, args); emit(J, F, OP_CALL); emit(J, F, n); } static void cexp(JF, js_Ast *exp) { int then, end; int n; switch (exp->type) { case AST_STRING: emitstring(J, F, OP_STRING, exp->string); break; case AST_NUMBER: emitnumber(J, F, exp->number); break; case EXP_UNDEF: emit(J, F, OP_UNDEF); break; case EXP_NULL: emit(J, F, OP_NULL); break; case EXP_TRUE: emit(J, F, OP_TRUE); break; case EXP_FALSE: emit(J, F, OP_FALSE); break; case EXP_THIS: emit(J, F, OP_THIS); break; case EXP_OBJECT: emit(J, F, OP_NEWOBJECT); cobject(J, F, exp->a); break; case EXP_ARRAY: emit(J, F, OP_NEWARRAY); carray(J, F, exp->a); break; case EXP_FUN: emitfunction(J, F, newfun(J, exp->a, exp->b, exp->c)); break; case AST_IDENTIFIER: emitstring(J, F, OP_GETVAR, exp->string); break; case EXP_INDEX: cexp(J, F, exp->a); cexp(J, F, exp->b); emit(J, F, OP_GETPROP); break; case EXP_MEMBER: cexp(J, F, exp->a); emitstring(J, F, OP_STRING, exp->b->string); emit(J, F, OP_GETPROP); break; case EXP_CALL: ccall(J, F, exp->a, exp->b); break; case EXP_NEW: cexp(J, F, exp->a); n = cargs(J, F, exp->b); emit(J, F, OP_NEW); emit(J, F, n); break; case EXP_DELETE: cdelete(J, F, exp->a); break; case EXP_PREINC: cassignop1(J, F, exp->a, 0); emit(J, F, OP_NUMBER_1); emit(J, F, OP_ADD); cassignop2(J, F, exp->a); break; case EXP_PREDEC: cassignop1(J, F, exp->a, 0); emit(J, F, OP_NUMBER_1); emit(J, F, OP_SUB); cassignop2(J, F, exp->a); break; case EXP_POSTINC: cassignop1(J, F, exp->a, 1); emit(J, F, OP_NUMBER_1); emit(J, F, OP_ADD); cassignop2(J, F, exp->a); emit(J, F, OP_POP); break; case EXP_POSTDEC: cassignop1(J, F, exp->a, 1); emit(J, F, OP_NUMBER_1); emit(J, F, OP_SUB); cassignop2(J, F, exp->a); emit(J, F, OP_POP); break; case EXP_VOID: cexp(J, F, exp->a); emit(J, F, OP_POP); emit(J, F, OP_UNDEF); break; case EXP_TYPEOF: cunary(J, F, exp, OP_TYPEOF); break; case EXP_POS: cunary(J, F, exp, OP_POS); break; case EXP_NEG: cunary(J, F, exp, OP_NEG); break; case EXP_BITNOT: cunary(J, F, exp, OP_BITNOT); break; case EXP_LOGNOT: cunary(J, F, exp, OP_LOGNOT); break; case EXP_BITOR: cbinary(J, F, exp, OP_BITOR); break; case EXP_BITXOR: cbinary(J, F, exp, OP_BITXOR); break; case EXP_BITAND: cbinary(J, F, exp, OP_BITAND); break; case EXP_EQ: cbinary(J, F, exp, OP_EQ); break; case EXP_NE: cbinary(J, F, exp, OP_NE); break; case EXP_STRICTEQ: cbinary(J, F, exp, OP_STRICTEQ); break; case EXP_STRICTNE: cbinary(J, F, exp, OP_STRICTNE); break; case EXP_LT: cbinary(J, F, exp, OP_LT); break; case EXP_GT: cbinary(J, F, exp, OP_GT); break; case EXP_LE: cbinary(J, F, exp, OP_LE); break; case EXP_GE: cbinary(J, F, exp, OP_GE); break; case EXP_INSTANCEOF: cbinary(J, F, exp, OP_INSTANCEOF); break; case EXP_IN: cbinary(J, F, exp, OP_IN); break; case EXP_SHL: cbinary(J, F, exp, OP_SHL); break; case EXP_SHR: cbinary(J, F, exp, OP_SHR); break; case EXP_USHR: cbinary(J, F, exp, OP_USHR); break; case EXP_ADD: cbinary(J, F, exp, OP_ADD); break; case EXP_SUB: cbinary(J, F, exp, OP_SUB); break; case EXP_MUL: cbinary(J, F, exp, OP_MUL); break; case EXP_DIV: cbinary(J, F, exp, OP_DIV); break; case EXP_MOD: cbinary(J, F, exp, OP_MOD); break; case EXP_ASS: cassign(J, F, exp->a, exp->b); break; case EXP_ASS_MUL: cassignop(J, F, exp->a, exp->b, OP_MUL); break; case EXP_ASS_DIV: cassignop(J, F, exp->a, exp->b, OP_DIV); break; case EXP_ASS_MOD: cassignop(J, F, exp->a, exp->b, OP_MOD); break; case EXP_ASS_ADD: cassignop(J, F, exp->a, exp->b, OP_ADD); break; case EXP_ASS_SUB: cassignop(J, F, exp->a, exp->b, OP_SUB); break; case EXP_ASS_SHL: cassignop(J, F, exp->a, exp->b, OP_SHL); break; case EXP_ASS_SHR: cassignop(J, F, exp->a, exp->b, OP_SHR); break; case EXP_ASS_USHR: cassignop(J, F, exp->a, exp->b, OP_USHR); break; case EXP_ASS_BITAND: cassignop(J, F, exp->a, exp->b, OP_BITAND); break; case EXP_ASS_BITXOR: cassignop(J, F, exp->a, exp->b, OP_BITXOR); break; case EXP_ASS_BITOR: cassignop(J, F, exp->a, exp->b, OP_BITOR); break; case EXP_COMMA: cexp(J, F, exp->a); emit(J, F, OP_POP); cexp(J, F, exp->b); break; case EXP_LOGOR: cexp(J, F, exp->a); emit(J, F, OP_DUP); end = jump(J, F, OP_JTRUE); emit(J, F, OP_POP); cexp(J, F, exp->b); label(J, F, end); break; case EXP_LOGAND: cexp(J, F, exp->a); emit(J, F, OP_DUP); end = jump(J, F, OP_JFALSE); emit(J, F, OP_POP); cexp(J, F, exp->b); label(J, F, end); break; case EXP_COND: cexp(J, F, exp->a); then = jump(J, F, OP_JTRUE); cexp(J, F, exp->c); end = jump(J, F, OP_JUMP); label(J, F, then); cexp(J, F, exp->b); label(J, F, end); break; default: jsC_error(J, exp, "unknown expression"); } } /* Statements */ static void cstm(JF, js_Ast *stm) { int loop, then, end; switch (stm->type) { case AST_FUNDEC: break; case STM_BLOCK: cstmlist(J, F, stm->a); break; case STM_NOP: break; case STM_VAR: cvarinit(J, F, stm->a); break; case STM_IF: if (stm->c) { cexp(J, F, stm->a); then = jump(J, F, OP_JTRUE); cstm(J, F, stm->c); end = jump(J, F, OP_JUMP); label(J, F, then); cstm(J, F, stm->b); label(J, F, end); } else { cexp(J, F, stm->a); end = jump(J, F, OP_JFALSE); cstm(J, F, stm->b); label(J, F, end); } break; case STM_DO: loop = here(J, F); cstm(J, F, stm->a); cexp(J, F, stm->b); jumpto(J, F, OP_JTRUE, loop); break; case STM_WHILE: loop = here(J, F); cexp(J, F, stm->a); end = jump(J, F, OP_JFALSE); cstm(J, F, stm->b); jumpto(J, F, OP_JUMP, loop); label(J, F, end); break; case STM_FOR: case STM_FOR_VAR: if (stm->type == STM_FOR_VAR) { cvarinit(J, F, stm->a); } else { cexp(J, F, stm->a); emit(J, F, OP_POP); } loop = here(J, F); cexp(J, F, stm->b); end = jump(J, F, OP_JFALSE); cstm(J, F, stm->d); cexp(J, F, stm->c); emit(J, F, OP_POP); jumpto(J, F, OP_JUMP, loop); label(J, F, end); break; case STM_FOR_IN: case STM_FOR_IN_VAR: cexp(J, F, stm->b); emit(J, F, OP_UNDEF); loop = here(J, F); emit(J, F, OP_NEXTPROP); end = jump(J, F, OP_JFALSE); if (stm->type == STM_FOR_IN_VAR) { if (stm->a->b) jsC_error(J, stm->a->b, "more than one loop variable in for-in statement"); emitstring(J, F, OP_SETVAR, stm->a->a->a->string); /* stm(list(var-init(ident))) */ } else { cassignloop(J, F, stm->a); } cstm(J, F, stm->c); jumpto(J, F, OP_JUMP, loop); label(J, F, end); break; // label // break // continue case STM_RETURN: if (stm->a) cexp(J, F, stm->a); else emit(J, F, OP_UNDEF); emit(J, F, OP_RETURN); break; case STM_WITH: cexp(J, F, stm->a); emit(J, F, OP_WITH); cstm(J, F, stm->b); emit(J, F, OP_ENDWITH); break; // switch case STM_THROW: cexp(J, F, stm->a); emit(J, F, OP_THROW); break; // try case STM_DEBUGGER: emit(J, F, OP_DEBUGGER); break; default: cexp(J, F, stm); emit(J, F, OP_POP); break; } } static void cstmlist(JF, js_Ast *list) { while (list) { cstm(J, F, list->a); list = list->b; } } /* Declarations and programs */ static int listlength(js_Ast *list) { int n = 0; while (list) ++n, list = list->b; return n; } static void cparams(JF, js_Ast *list) { F->numparams = listlength(list); F->params = malloc(F->numparams * sizeof *F->params); int i = 0; while (list) { F->params[i++] = list->a->string; list = list->b; } } static void cfundecs(JF, js_Ast *list) { while (list) { js_Ast *stm = list->a; if (stm->type == AST_FUNDEC) { emitfunction(J, F, newfun(J, stm->a, stm->b, stm->c)); emitstring(J, F, OP_FUNDEC, stm->a->string); } list = list->b; } } static void cvardecs(JF, js_Ast *node) { if (node->type == EXP_VAR) { emitstring(J, F, OP_VARDEC, node->a->string); } else if (node->type != EXP_FUN && node->type != AST_FUNDEC) { if (node->a) cvardecs(J, F, node->a); if (node->b) cvardecs(J, F, node->b); if (node->c) cvardecs(J, F, node->c); if (node->d) cvardecs(J, F, node->d); } } static void cfunbody(JF, js_Ast *name, js_Ast *params, js_Ast *body) { if (name) { F->name = name->string; emitfunction(J, F, F); emitstring(J, F, OP_FUNDEC, name->string); } else { F->name = "<anonymous>"; } cparams(J, F, params); if (body) { cfundecs(J, F, body); cvardecs(J, F, body); cstmlist(J, F, body); } emit(J, F, OP_UNDEF); emit(J, F, OP_RETURN); } int jsC_error(js_State *J, js_Ast *node, const char *fmt, ...) { va_list ap; fprintf(stderr, "%s:%d: error: ", J->filename, node->line); va_start(ap, fmt); vfprintf(stderr, fmt, ap); va_end(ap); fprintf(stderr, "\n"); longjmp(J->jb, 1); } static void jsC_freecompile(js_State *J) { js_Function *F = J->fun; while (F) { js_Function *next = F->next; freefun(J, F); F = next; } J->fun = NULL; } js_Function *jsC_compile(js_State *J, js_Ast *prog) { js_Function *F; if (setjmp(J->jb)) { jsC_freecompile(J); return NULL; } F = newfun(J, NULL, NULL, prog); J->fun = NULL; return F; }