%{ char *rcs_luastx = "$Id: lua.stx,v 3.5 1994/11/13 14:54:18 roberto Exp roberto $"; #include #include #include #include "opcode.h" #include "hash.h" #include "inout.h" #include "tree.h" #include "table.h" #include "lua.h" #ifndef LISTING #define LISTING 0 #endif #ifndef CODE_BLOCK #define CODE_BLOCK 256 #endif static Long maxcode; static Long maxmain; static Long maxcurr ; static Byte *funcCode = NULL; static Byte **initcode; static Byte *basepc; static Long maincode; static Long pc; #define MAXVAR 32 static long varbuffer[MAXVAR]; /* variables in an assignment list; it's long to store negative Word values */ static int nvarbuffer=0; /* number of variables at a list */ static Word localvar[STACKGAP]; /* store local variable names */ static int nlocalvar=0; /* number of local variables */ #define MAXFIELDS FIELDS_PER_FLUSH*2 static Word fields[MAXFIELDS]; /* fieldnames to be flushed */ static int nfields=0; /* Internal functions */ static void code_byte (Byte c) { if (pc>maxcurr-2) /* 1 byte free to code HALT of main code */ { maxcurr *= 2; basepc = (Byte *)realloc(basepc, maxcurr*sizeof(Byte)); if (basepc == NULL) lua_error ("not enough memory"); } basepc[pc++] = c; } static void code_word (Word n) { CodeWord code; code.w = n; code_byte(code.m.c1); code_byte(code.m.c2); } static void code_float (float n) { CodeFloat code; code.f = n; code_byte(code.m.c1); code_byte(code.m.c2); code_byte(code.m.c3); code_byte(code.m.c4); } static void code_code (Byte *b) { CodeCode code; code.b = b; code_byte(code.m.c1); code_byte(code.m.c2); code_byte(code.m.c3); code_byte(code.m.c4); } static void code_word_at (Byte *p, Word n) { CodeWord code; code.w = n; *p++ = code.m.c1; *p++ = code.m.c2; } static void push_field (Word name) { if (nfields < STACKGAP-1) fields[nfields++] = name; else lua_error ("too many fields in a constructor"); } static void flush_record (int n) { int i; if (n == 0) return; code_byte(STORERECORD); code_byte(n); for (i=0; i NUMBER %token STRING %token NAME %token DEBUG %type PrepJump %type expr, exprlist, exprlist1, varlist1, funcParams, funcvalue %type fieldlist, localdeclist, decinit %type ffieldlist1 %type lfieldlist1 %type var, singlevar %type body %left AND OR %left EQ NE '>' '<' LE GE %left CONC %left '+' '-' %left '*' '/' %left UNARY NOT %right '^' %% /* beginning of rules section */ functionlist : /* empty */ | functionlist { pc=maincode; basepc=*initcode; maxcurr=maxmain; nlocalvar=0; } stat sc { maincode=pc; *initcode=basepc; maxmain=maxcurr; } | functionlist function | functionlist method | functionlist setdebug ; function : FUNCTION NAME { init_function($2); } body { Word func = luaI_findsymbol($2); s_tag(func) = LUA_T_FUNCTION; s_bvalue(func) = $4; #if LISTING PrintCode(funcCode,funcCode+pc); #endif } ; method : FUNCTION NAME ':' NAME { init_function($4); localvar[nlocalvar]=luaI_findsymbolbyname("self"); add_nlocalvar(1); } body { #if LISTING PrintCode(funcCode,funcCode+pc); #endif /* assign function to table field */ pc=maincode; basepc=*initcode; maxcurr=maxmain; nlocalvar=0; lua_pushvar(luaI_findsymbol($2)+1); code_byte(PUSHSTRING); code_word(luaI_findconstant($4)); code_byte(PUSHFUNCTION); code_code($6); code_byte(STOREINDEXED0); maincode=pc; *initcode=basepc; maxmain=maxcurr; } ; body : '(' parlist ')' block END { codereturn(); $$ = calloc (pc, sizeof(Byte)); if ($$ == NULL) lua_error("not enough memory"); memcpy ($$, basepc, pc*sizeof(Byte)); funcCode = basepc; maxcode=maxcurr; } ; statlist : /* empty */ | statlist stat sc ; stat : { codedebugline(); } stat1 ; sc : /* empty */ | ';' ; cond : { codedebugline(); } expr1 ; stat1 : IF cond THEN PrepJump block PrepJump elsepart END { { Long elseinit = $6+sizeof(Word)+1; if (pc - elseinit == 0) /* no else */ { pc -= sizeof(Word)+1; elseinit = pc; } else { basepc[$6] = JMP; code_word_at(basepc+$6+1, pc - elseinit); } basepc[$4] = IFFJMP; code_word_at(basepc+$4+1,elseinit-($4+sizeof(Word)+1)); } } | WHILE {$$=pc;} cond DO PrepJump block PrepJump END { basepc[$5] = IFFJMP; code_word_at(basepc+$5+1, pc - ($5 + sizeof(Word)+1)); basepc[$7] = UPJMP; code_word_at(basepc+$7+1, pc - ($2)); } | REPEAT {$$=pc;} block UNTIL expr1 PrepJump { basepc[$6] = IFFUPJMP; code_word_at(basepc+$6+1, pc - ($2)); } | varlist1 '=' exprlist1 { { int i; adjust_mult_assign(nvarbuffer, $3, $1 * 2 + nvarbuffer); for (i=nvarbuffer-1; i>=0; i--) lua_codestore (i); if ($1 > 1 || ($1 == 1 && varbuffer[0] != 0)) lua_codeadjust (0); } } | functioncall { code_byte(0); } | LOCAL localdeclist decinit { add_nlocalvar($2); adjust_mult_assign($2, $3, 0); } ; elsepart : /* empty */ | ELSE block | ELSEIF expr1 THEN PrepJump block PrepJump elsepart { { Long elseinit = $6+sizeof(Word)+1; if (pc - elseinit == 0) /* no else */ { pc -= sizeof(Word)+1; elseinit = pc; } else { basepc[$6] = JMP; code_word_at(basepc+$6+1, pc - elseinit); } basepc[$4] = IFFJMP; code_word_at(basepc+$4+1, elseinit - ($4 + sizeof(Word)+1)); } } ; block : {$$ = nlocalvar;} statlist ret { if (nlocalvar != $1) { nlocalvar = $1; lua_codeadjust (0); } } ; ret : /* empty */ | { codedebugline(); } RETURN exprlist sc { if ($3 < 0) code_byte(MULT_RET); codereturn(); } ; PrepJump : /* empty */ { $$ = pc; code_byte(0); /* open space */ code_word (0); } expr1 : expr { if ($1 == 0) code_byte(1); } ; expr : '(' expr ')' { $$ = $2; } | expr1 EQ expr1 { code_byte(EQOP); $$ = 1; } | expr1 '<' expr1 { code_byte(LTOP); $$ = 1; } | expr1 '>' expr1 { code_byte(GTOP); $$ = 1; } | expr1 NE expr1 { code_byte(EQOP); code_byte(NOTOP); $$ = 1; } | expr1 LE expr1 { code_byte(LEOP); $$ = 1; } | expr1 GE expr1 { code_byte(GEOP); $$ = 1; } | expr1 '+' expr1 { code_byte(ADDOP); $$ = 1; } | expr1 '-' expr1 { code_byte(SUBOP); $$ = 1; } | expr1 '*' expr1 { code_byte(MULTOP); $$ = 1; } | expr1 '/' expr1 { code_byte(DIVOP); $$ = 1; } | expr1 '^' expr1 { code_byte(POWOP); $$ = 1; } | expr1 CONC expr1 { code_byte(CONCOP); $$ = 1; } | '+' expr1 %prec UNARY { $$ = 1; } | '-' expr1 %prec UNARY { code_byte(MINUSOP); $$ = 1;} | table { $$ = 1; } | varexp { $$ = 1;} | NUMBER { code_number($1); $$ = 1; } | STRING { code_byte(PUSHSTRING); code_word($1); $$ = 1; } | NIL {code_byte(PUSHNIL); $$ = 1; } | functioncall { $$ = 0; } | NOT expr1 { code_byte(NOTOP); $$ = 1;} | expr1 AND PrepJump {code_byte(POP); } expr1 { basepc[$3] = ONFJMP; code_word_at(basepc+$3+1, pc - ($3 + sizeof(Word)+1)); $$ = 1; } | expr1 OR PrepJump {code_byte(POP); } expr1 { basepc[$3] = ONTJMP; code_word_at(basepc+$3+1, pc - ($3 + sizeof(Word)+1)); $$ = 1; } ; table : { code_byte(CREATEARRAY); $$ = pc; code_word(0); } '{' fieldlist '}' { code_word_at(basepc+$1, $3); } ; functioncall : funcvalue funcParams { code_byte(CALLFUNC); code_byte($1+$2); } ; funcvalue : varexp { $$ = 0; } | varexp ':' NAME { code_byte(PUSHSTRING); code_word(luaI_findconstant($3)); code_byte(PUSHSELF); $$ = 1; } ; funcParams : '(' exprlist ')' { if ($2<0) { code_byte(1); $$ = -$2; } else $$ = $2; } | table { $$ = 1; } ; exprlist : /* empty */ { $$ = 0; } | exprlist1 { $$ = $1; } ; exprlist1 : expr { if ($1 == 0) $$ = -1; else $$ = 1; } | exprlist1 ',' { if ($1 < 0) code_byte(1); } expr { int r = $1 < 0 ? -$1 : $1; $$ = ($4 == 0) ? -(r+1) : r+1; } ; parlist : /* empty */ { lua_codeadjust(0); } | parlist1 { lua_codeadjust(0); } ; parlist1 : NAME { localvar[nlocalvar]=luaI_findsymbol($1); add_nlocalvar(1); } | parlist1 ',' NAME { localvar[nlocalvar]=luaI_findsymbol($3); add_nlocalvar(1); } ; fieldlist : /* empty */ { $$ = 0; } | lfieldlist1 lastcomma { $$ = $1; flush_list($1/FIELDS_PER_FLUSH, $1%FIELDS_PER_FLUSH); } | ffieldlist1 lastcomma { $$ = $1; flush_record($1%FIELDS_PER_FLUSH); } | lfieldlist1 ';' { flush_list($1/FIELDS_PER_FLUSH, $1%FIELDS_PER_FLUSH); } ffieldlist1 lastcomma { $$ = $1+$4; flush_record($4%FIELDS_PER_FLUSH); } ; lastcomma : /* empty */ | ',' ; ffieldlist1 : ffield {$$=1;} | ffieldlist1 ',' ffield { $$=$1+1; if ($$%FIELDS_PER_FLUSH == 0) flush_record(FIELDS_PER_FLUSH); } ; ffield : NAME '=' expr1 { push_field(luaI_findconstant($1)); } ; lfieldlist1 : expr1 {$$=1;} | lfieldlist1 ',' expr1 { $$=$1+1; if ($$%FIELDS_PER_FLUSH == 0) flush_list($$/FIELDS_PER_FLUSH - 1, FIELDS_PER_FLUSH); } ; varlist1 : var { nvarbuffer = 0; varbuffer[nvarbuffer] = $1; incr_nvarbuffer(); $$ = ($1 == 0) ? 1 : 0; } | varlist1 ',' var { varbuffer[nvarbuffer] = $3; incr_nvarbuffer(); $$ = ($3 == 0) ? $1 + 1 : $1; } ; var : singlevar { $$ = $1; } | varexp '[' expr1 ']' { $$ = 0; /* indexed variable */ } | varexp '.' NAME { code_byte(PUSHSTRING); code_word(luaI_findconstant($3)); $$ = 0; /* indexed variable */ } ; singlevar : NAME { Word s = luaI_findsymbol($1); int local = lua_localname (s); if (local == -1) /* global var */ $$ = s + 1; /* return positive value */ else $$ = -(local+1); /* return negative value */ } ; varexp : var { lua_pushvar($1); } ; localdeclist : NAME {localvar[nlocalvar]=luaI_findsymbol($1); $$ = 1;} | localdeclist ',' NAME { localvar[nlocalvar+$1]=luaI_findsymbol($3); $$ = $1+1; } ; decinit : /* empty */ { $$ = 0; } | '=' exprlist1 { $$ = $2; } ; setdebug : DEBUG {lua_debug = $1;} %% /* ** Search a local name and if find return its index. If do not find return -1 */ static int lua_localname (Word n) { int i; for (i=nlocalvar-1; i >= 0; i--) if (n == localvar[i]) return i; /* local var */ return -1; /* global var */ } /* ** Push a variable given a number. If number is positive, push global variable ** indexed by (number -1). If negative, push local indexed by ABS(number)-1. ** Otherwise, if zero, push indexed variable (record). */ static void lua_pushvar (long number) { if (number > 0) /* global var */ { code_byte(PUSHGLOBAL); code_word(number-1); } else if (number < 0) /* local var */ { number = (-number) - 1; if (number < 10) code_byte(PUSHLOCAL0 + number); else { code_byte(PUSHLOCAL); code_byte(number); } } else { code_byte(PUSHINDEXED); } } static void lua_codeadjust (int n) { if (n+nlocalvar == 0) code_byte(ADJUST0); else { code_byte(ADJUST); code_byte(n+nlocalvar); } } static void init_function (TreeNode *func) { if (funcCode == NULL) /* first function */ { funcCode = (Byte *) calloc(CODE_BLOCK, sizeof(Byte)); if (funcCode == NULL) lua_error("not enough memory"); maxcode = CODE_BLOCK; } pc=0; basepc=funcCode; maxcurr=maxcode; nlocalvar=0; if (lua_debug) { code_byte(SETFUNCTION); code_code((Byte *)strdup(lua_file[lua_nfile-1])); code_word(luaI_findconstant(func)); } } static void codereturn (void) { if (lua_debug) code_byte(RESET); if (nlocalvar == 0) code_byte(RETCODE0); else { code_byte(RETCODE); code_byte(nlocalvar); } } static void codedebugline (void) { if (lua_debug) { code_byte(SETLINE); code_word(lua_linenumber); } } static void adjust_mult_assign (int vars, int exps, int temps) { if (exps < 0) { int r = vars - (-exps-1); if (r >= 0) code_byte(r); else { code_byte(0); lua_codeadjust(temps); } } else if (vars != exps) lua_codeadjust(temps); } static void lua_codestore (int i) { if (varbuffer[i] > 0) /* global var */ { code_byte(STOREGLOBAL); code_word(varbuffer[i]-1); } else if (varbuffer[i] < 0) /* local var */ { int number = (-varbuffer[i]) - 1; if (number < 10) code_byte(STORELOCAL0 + number); else { code_byte(STORELOCAL); code_byte(number); } } else /* indexed var */ { int j; int upper=0; /* number of indexed variables upper */ int param; /* number of itens until indexed expression */ for (j=i+1; j