first version of code optimizer

This commit is contained in:
Roberto Ierusalimschy
2000-02-22 11:31:43 -02:00
parent 39e1f079bd
commit 3bc925138e
5 changed files with 314 additions and 272 deletions

View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lopcodes.h,v 1.39 2000/02/11 16:52:54 roberto Exp roberto $ ** $Id: lopcodes.h,v 1.40 2000/02/14 16:51:08 roberto Exp roberto $
** Opcodes for Lua virtual machine ** Opcodes for Lua virtual machine
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -47,6 +47,11 @@
/*
** K = U argument used as index to `kstr'
** J = S argument used as jump offset (relative to pc of next instruction)
*/
typedef enum { typedef enum {
/* name parm before after side effect /* name parm before after side effect
-----------------------------------------------------------------------------*/ -----------------------------------------------------------------------------*/
@@ -60,24 +65,25 @@ PUSHNIL,/* U - nil_0-nil_u */
POP,/* U a_u-a_1 - */ POP,/* U a_u-a_1 - */
PUSHINT,/* S - (real)s */ PUSHINT,/* S - (real)s */
PUSHSTRING,/* U - KSTR[u] */ PUSHSTRING,/* K - KSTR[k] */
PUSHNUMBER,/* U - KNUM[u] */ PUSHNUM,/* U - KNUM[u] */
PUSHNEGNUM,/* U - -KNUM[u] */
PUSHUPVALUE,/* U - Closure[u] */ PUSHUPVALUE,/* U - Closure[u] */
PUSHLOCAL,/* U - LOC[u] */ PUSHLOCAL,/* U - LOC[u] */
GETGLOBAL,/* U - VAR[CNST[u]] */ GETGLOBAL,/* K - VAR[CNST[k]] */
GETTABLE,/* - i t t[i] */ GETTABLE,/* - i t t[i] */
GETDOTTED,/* U t t[CNST[u]] */ GETDOTTED,/* K t t[KSTR[k]] */
PUSHSELF,/* U t t t[CNST[u]] */ PUSHSELF,/* K t t t[KSTR[k]] */
CREATETABLE,/* U - newarray(size = u) */ CREATETABLE,/* U - newarray(size = u) */
SETLOCAL,/* U x - LOC[u]=x */ SETLOCAL,/* U x - LOC[u]=x */
SETGLOBAL,/* U x - VAR[CNST[u]]=x */ SETGLOBAL,/* K x - VAR[KSTR[k]]=x */
SETTABLEPOP,/* - v i t - t[i]=v */ SETTABLEPOP,/* - v i t - t[i]=v */
SETTABLE,/* U v a_u-a_1 i t a_u-a_1 i t t[i]=v */ SETTABLE,/* U v a_u-a_1 i t a_u-a_1 i t t[i]=v */
SETLIST,/* A B v_b-v_0 t t t[i+a*FPF]=v_i */ SETLIST,/* A B v_b-v_0 t t t[i+a*FPF]=v_i */
SETMAP,/* U v_u k_u - v_0 k_0 t t t[k_i]=v_i */ SETMAP,/* U v_u k_u - v_0 k_0 t t t[k_i]=v_i */
@@ -88,7 +94,9 @@ LTOP,/* - y x (x<y)? 1 : nil */
LEOP,/* - y x (x<y)? 1 : nil */ LEOP,/* - y x (x<y)? 1 : nil */
GTOP,/* - y x (x>y)? 1 : nil */ GTOP,/* - y x (x>y)? 1 : nil */
GEOP,/* - y x (x>=y)? 1 : nil */ GEOP,/* - y x (x>=y)? 1 : nil */
ADDOP,/* - y x x+y */ ADDOP,/* - y x x+y */
ADDI,/* S x x+s */
SUBOP,/* - y x x-y */ SUBOP,/* - y x x-y */
MULTOP,/* - y x x*y */ MULTOP,/* - y x x*y */
DIVOP,/* - y x x/y */ DIVOP,/* - y x x/y */
@@ -97,11 +105,11 @@ CONCOP,/* - y x x..y */
MINUSOP,/* - x -x */ MINUSOP,/* - x -x */
NOTOP,/* - x (x==nil)? 1 : nil */ NOTOP,/* - x (x==nil)? 1 : nil */
ONTJMP,/* S x (x!=nil)? x : - (x!=nil)? PC+=s */ ONTJMP,/* J x (x!=nil)? x : - (x!=nil)? PC+=s */
ONFJMP,/* S x (x==nil)? x : - (x==nil)? PC+=s */ ONFJMP,/* J x (x==nil)? x : - (x==nil)? PC+=s */
JMP,/* S - - PC+=s */ JMP,/* J - - PC+=s */
IFTJMP,/* S x - (x!=nil)? PC+=s */ IFTJMP,/* J x - (x!=nil)? PC+=s */
IFFJMP,/* S x - (x==nil)? PC+=s */ IFFJMP,/* J x - (x==nil)? PC+=s */
CLOSURE,/* A B v_b-v_1 closure(CNST[a], v_b-v_1) */ CLOSURE,/* A B v_b-v_1 closure(CNST[a], v_b-v_1) */
@@ -111,7 +119,7 @@ SETLINE/* U - - LINE=u */
#define RFIELDS_PER_FLUSH 32 /* records (SETMAP) */ #define RFIELDS_PER_FLUSH 32 /* records (SETMAP) */
#define LFIELDS_PER_FLUSH 64 /* FPF - lists (SETLIST) (<MAXARG_B) */ #define LFIELDS_PER_FLUSH 64 /* FPF - lists (SETLIST) (<=MAXARG_B) */
#endif #endif

399
lparser.c
View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lparser.c,v 1.58 2000/02/11 16:52:54 roberto Exp roberto $ ** $Id: lparser.c,v 1.59 2000/02/14 16:51:08 roberto Exp roberto $
** LL(1) Parser and code generator for Lua ** LL(1) Parser and code generator for Lua
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -8,6 +8,9 @@
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#define LUA_REENTRANT
#include "lcode.h"
#include "ldo.h" #include "ldo.h"
#include "lfunc.h" #include "lfunc.h"
#include "llex.h" #include "llex.h"
@@ -19,50 +22,6 @@
#include "lstring.h" #include "lstring.h"
/* maximum number of local variables */
#ifndef MAXLOCALS
#define MAXLOCALS 200 /* arbitrary limit (<MAXARG_B) */
#endif
/* maximum number of upvalues */
#ifndef MAXUPVALUES
#define MAXUPVALUES 32 /* arbitrary limit (<MAXARG_B) */
#endif
/* maximum number of variables in the left side of an assignment */
#ifndef MAXVARSLH
#define MAXVARSLH 100 /* arbitrary limit (<MAXARG_B) */
#endif
/* maximum number of parameters in a function */
#ifndef MAXPARAMS
#define MAXPARAMS 100 /* arbitrary limit (<MAXLOCALS) */
#endif
/*
** Variable descriptor:
** must include an `exp' option because LL(1) cannot distinguish
** between variables, upvalues and function calls on first sight.
*/
typedef enum {
VGLOBAL, /* info is constant index of global name */
VLOCAL, /* info is stack index */
VDOT, /* info is constant index of index name */
VINDEXED, /* no info (table and index are on the stack) */
VEXP /* info is pc index of a call (or 0 if exp is closed) */
} varkind;
typedef struct vardesc {
varkind k;
int info;
} vardesc;
/* /*
** Expression List descriptor: ** Expression List descriptor:
** tells number of expressions in the list, ** tells number of expressions in the list,
@@ -87,19 +46,6 @@ typedef struct constdesc {
} constdesc; } constdesc;
/* state needed to generate code for a given function */
typedef struct FuncState {
TProtoFunc *f; /* current function header */
struct FuncState *prev; /* enclosing function */
int pc; /* next position to code */
int stacksize; /* number of values on activation register */
int nlocalvar; /* number of active local variables */
int nupvalues; /* number of upvalues */
int nvars; /* number of entries in f->locvars (-1 if no debug information) */
int lastsetline; /* line where last SETLINE was issued */
vardesc upvalues[MAXUPVALUES]; /* upvalues */
TaggedString *localvar[MAXLOCALS]; /* store local variable names */
} FuncState;
/* /*
@@ -112,12 +58,59 @@ static void expr (LexState *ls, vardesc *v);
static void exp1 (LexState *ls); static void exp1 (LexState *ls);
static void next (LexState *ls) {
ls->token = luaX_lex(ls);
}
static void luaY_error (LexState *ls, const char *msg) { static void luaY_error (LexState *ls, const char *msg) {
luaX_error(ls, msg, ls->token); luaX_error(ls, msg, ls->token);
} }
static void error_expected (LexState *ls, int token) {
char buff[100], t[TOKEN_LEN];
luaX_token2str(token, t);
sprintf(buff, "`%.20s' expected", t);
luaY_error(ls, buff);
}
static void error_unexpected (LexState *ls) {
luaY_error(ls, "unexpected token");
}
static void error_unmatched (LexState *ls, int what, int who, int where) {
if (where == ls->linenumber)
error_expected(ls, what);
else {
char buff[100];
char t_what[TOKEN_LEN], t_who[TOKEN_LEN];
luaX_token2str(what, t_what);
luaX_token2str(who, t_who);
sprintf(buff, "`%.20s' expected (to close `%.20s' at line %d)",
t_what, t_who, where);
luaY_error(ls, buff);
}
}
static void check (LexState *ls, int c) {
if (ls->token != c)
error_expected(ls, c);
next(ls);
}
static int optional (LexState *ls, int c) {
if (ls->token == c) {
next(ls);
return 1;
}
else return 0;
}
static void checklimit (LexState *ls, int val, int limit, const char *msg) { static void checklimit (LexState *ls, int val, int limit, const char *msg) {
if (val > limit) { if (val > limit) {
char buff[100]; char buff[100];
@@ -127,25 +120,12 @@ static void checklimit (LexState *ls, int val, int limit, const char *msg) {
} }
static int code_instruction (LexState *ls, Instruction i) {
FuncState *fs = ls->fs;
luaM_growvector(ls->L, fs->f->code, fs->pc, 1, Instruction, codeEM, MAXARG_S);
fs->f->code[fs->pc] = i;
return fs->pc++;
}
static void fix_jump (LexState *ls, int pc, int dest) {
Instruction *jmp = &ls->fs->f->code[pc];
/* jump is relative to position following jump instruction */
*jmp = SETARG_S(*jmp, dest-(pc+1));
}
static void deltastack (LexState *ls, int delta) { static void deltastack (LexState *ls, int delta) {
FuncState *fs = ls->fs; FuncState *fs = ls->fs;
fs->stacksize += delta; fs->stacksize += delta;
if (delta > 0 && fs->stacksize > fs->f->maxstacksize) { if (delta > 0 && fs->stacksize > fs->f->maxstacksize) {
checklimit(ls, fs->stacksize, MAXSTACK, "temporaries or local variables");
fs->f->maxstacksize = fs->stacksize; fs->f->maxstacksize = fs->stacksize;
} }
} }
@@ -153,7 +133,7 @@ static void deltastack (LexState *ls, int delta) {
static int aux_code (LexState *ls, OpCode op, Instruction i, int delta) { static int aux_code (LexState *ls, OpCode op, Instruction i, int delta) {
deltastack(ls, delta); deltastack(ls, delta);
return code_instruction(ls, SET_OPCODE(i, op)); return luaK_code(ls, SET_OPCODE(i, op));
} }
@@ -182,6 +162,22 @@ static int code_AB (LexState *ls, OpCode op, int a, int b, int delta) {
} }
static void check_debugline (LexState *ls) {
if (ls->L->debug && ls->linenumber != ls->fs->lastsetline) {
code_U(ls, SETLINE, ls->linenumber, 0);
ls->fs->lastsetline = ls->linenumber;
}
}
static void check_match (LexState *ls, int what, int who, int where) {
if (ls->token != what)
error_unmatched(ls, what, who, where);
check_debugline(ls); /* to `mark' the `what' */
next(ls);
}
static void code_kstr (LexState *ls, int c) { static void code_kstr (LexState *ls, int c) {
code_U(ls, PUSHSTRING, c, 1); code_U(ls, PUSHSTRING, c, 1);
} }
@@ -228,10 +224,26 @@ static int real_constant (LexState *ls, real r) {
static void code_number (LexState *ls, real f) { static void code_number (LexState *ls, real f) {
if ((real)(-MAXARG_S) <= f && f <= (real)MAXARG_S && (int)f == f) if (f <= (real)MAXARG_S && (int)f == f)
code_S(ls, PUSHINT, (int)f, 1); /* f has a short integer value */ code_S(ls, PUSHINT, (int)f, 1); /* f has a short integer value */
else else
code_U(ls, PUSHNUMBER, real_constant(ls, f), 1); code_U(ls, PUSHNUM, real_constant(ls, f), 1);
}
static int checkname (LexState *ls) {
int sc;
if (ls->token != NAME)
luaY_error(ls, "<name> expected");
sc = string_constant(ls, ls->fs, ls->seminfo.ts);
next(ls);
return sc;
}
static TaggedString *str_checkname (LexState *ls) {
int i = checkname(ls); /* this call may realloc `f->consts' */
return ls->fs->f->kstr[i];
} }
@@ -327,15 +339,6 @@ static void pushupvalue (LexState *ls, TaggedString *n) {
} }
static void check_debugline (LexState *ls) {
if (ls->L->debug && ls->linenumber != ls->fs->lastsetline) {
code_U(ls, SETLINE, ls->linenumber, 0);
ls->fs->lastsetline = ls->linenumber;
}
}
static void adjuststack (LexState *ls, int n) { static void adjuststack (LexState *ls, int n) {
if (n > 0) if (n > 0)
code_U(ls, POP, n, -n); code_U(ls, POP, n, -n);
@@ -344,7 +347,7 @@ static void adjuststack (LexState *ls, int n) {
} }
static void close_exp (LexState *ls, int pc, int nresults) { static void close_call (LexState *ls, int pc, int nresults) {
if (pc > 0) { /* expression is an open function call? */ if (pc > 0) { /* expression is an open function call? */
Instruction *i = &ls->fs->f->code[pc]; Instruction *i = &ls->fs->f->code[pc];
*i = SETARG_B(*i, nresults); /* set nresults */ *i = SETARG_B(*i, nresults); /* set nresults */
@@ -364,10 +367,10 @@ static void adjust_mult_assign (LexState *ls, int nvars, listdesc *d) {
diff--; /* do not count function call itself */ diff--; /* do not count function call itself */
if (diff <= 0) { /* more variables than values? */ if (diff <= 0) { /* more variables than values? */
/* function call must provide extra values */ /* function call must provide extra values */
close_exp(ls, d->pc, -diff); close_call(ls, d->pc, -diff);
} }
else { /* more values than variables */ else { /* more values than variables */
close_exp(ls, d->pc, 0); /* call should provide no value */ close_call(ls, d->pc, 0); /* call should provide no value */
adjuststack(ls, diff); /* pop eventual extra values */ adjuststack(ls, diff); /* pop eventual extra values */
} }
} }
@@ -390,16 +393,21 @@ static void code_args (LexState *ls, int nparams, int dots) {
} }
static void unloaddot (LexState *ls, vardesc *v) { static int getvarname (LexState *ls, vardesc *var) {
/* dotted variables <a.x> must be stored as regular indexed vars <a["x"]> */ switch (var->k) {
if (v->k == VDOT) { case VGLOBAL:
code_kstr(ls, v->info); return var->info;
v->k = VINDEXED; case VLOCAL:
return string_constant(ls, ls->fs, ls->fs->localvar[var->info]);
break;
default:
error_unexpected(ls); /* there is no `var name' */
return 0; /* to avoid warnings */
} }
} }
static void lua_pushvar (LexState *ls, vardesc *var) { static void close_exp (LexState *ls, vardesc *var) {
switch (var->k) { switch (var->k) {
case VLOCAL: case VLOCAL:
code_U(ls, PUSHLOCAL, var->info, 1); code_U(ls, PUSHLOCAL, var->info, 1);
@@ -408,14 +416,11 @@ static void lua_pushvar (LexState *ls, vardesc *var) {
code_U(ls, GETGLOBAL, var->info, 1); code_U(ls, GETGLOBAL, var->info, 1);
assertglobal(ls, var->info); /* make sure that there is a global */ assertglobal(ls, var->info); /* make sure that there is a global */
break; break;
case VDOT:
code_U(ls, GETDOTTED, var->info, 0);
break;
case VINDEXED: case VINDEXED:
code_0(ls, GETTABLE, -1); code_0(ls, GETTABLE, -1);
break; break;
case VEXP: case VEXP:
close_exp(ls, var->info, 1); /* function must return 1 value */ close_call(ls, var->info, 1); /* call must return 1 value */
break; break;
} }
var->k = VEXP; var->k = VEXP;
@@ -445,7 +450,7 @@ static void func_onstack (LexState *ls, FuncState *func) {
TProtoFunc *f = ls->fs->f; TProtoFunc *f = ls->fs->f;
int i; int i;
for (i=0; i<func->nupvalues; i++) for (i=0; i<func->nupvalues; i++)
lua_pushvar(ls, &func->upvalues[i]); close_exp(ls, &func->upvalues[i]);
luaM_growvector(ls->L, f->kproto, f->nkproto, 1, TProtoFunc *, luaM_growvector(ls->L, f->kproto, f->nkproto, 1, TProtoFunc *,
constantEM, MAXARG_A); constantEM, MAXARG_A);
f->kproto[f->nkproto++] = func->f; f->kproto[f->nkproto++] = func->f;
@@ -466,6 +471,7 @@ static void init_state (LexState *ls, FuncState *fs, TaggedString *source) {
fs->f = f; fs->f = f;
f->source = source; f->source = source;
fs->pc = 0; fs->pc = 0;
fs->last_pc = -1; /* invalid index to signal no last instruction */
f->code = NULL; f->code = NULL;
f->maxstacksize = 0; f->maxstacksize = 0;
f->numparams = 0; /* default for main chunk */ f->numparams = 0; /* default for main chunk */
@@ -495,76 +501,6 @@ static void close_func (LexState *ls) {
} }
static void next (LexState *ls) {
ls->token = luaX_lex(ls);
}
static void error_expected (LexState *ls, int token) {
char buff[100], t[TOKEN_LEN];
luaX_token2str(token, t);
sprintf(buff, "`%.20s' expected", t);
luaY_error(ls, buff);
}
static void error_unexpected (LexState *ls) {
luaY_error(ls, "unexpected token");
}
static void error_unmatched (LexState *ls, int what, int who, int where) {
if (where == ls->linenumber)
error_expected(ls, what);
else {
char buff[100];
char t_what[TOKEN_LEN], t_who[TOKEN_LEN];
luaX_token2str(what, t_what);
luaX_token2str(who, t_who);
sprintf(buff, "`%.20s' expected (to close `%.20s' at line %d)",
t_what, t_who, where);
luaY_error(ls, buff);
}
}
static void check (LexState *ls, int c) {
if (ls->token != c)
error_expected(ls, c);
next(ls);
}
static void check_match (LexState *ls, int what, int who, int where) {
if (ls->token != what)
error_unmatched(ls, what, who, where);
check_debugline(ls); /* to `mark' the `what' */
next(ls);
}
static int checkname (LexState *ls) {
int sc;
if (ls->token != NAME)
luaY_error(ls, "<name> expected");
sc = string_constant(ls, ls->fs, ls->seminfo.ts);
next(ls);
return sc;
}
static TaggedString *str_checkname (LexState *ls) {
int i = checkname(ls); /* this call may realloc `f->consts' */
return ls->fs->f->kstr[i];
}
static int optional (LexState *ls, int c) {
if (ls->token == c) {
next(ls);
return 1;
}
else return 0;
}
TProtoFunc *luaY_parser (lua_State *L, ZIO *z) { TProtoFunc *luaY_parser (lua_State *L, ZIO *z) {
struct LexState lexstate; struct LexState lexstate;
struct FuncState funcstate; struct FuncState funcstate;
@@ -591,14 +527,14 @@ static void explist1 (LexState *ls, listdesc *d) {
d->n = 1; d->n = 1;
while (ls->token == ',') { while (ls->token == ',') {
d->n++; d->n++;
lua_pushvar(ls, &v); close_exp(ls, &v);
next(ls); next(ls);
expr(ls, &v); expr(ls, &v);
} }
if (v.k == VEXP) if (v.k == VEXP)
d->pc = v.info; d->pc = v.info;
else { else {
lua_pushvar(ls, &v); close_exp(ls, &v);
d->pc = 0; d->pc = 0;
} }
} }
@@ -628,7 +564,7 @@ static int funcparams (LexState *ls, int slf) {
next(ls); next(ls);
explist(ls, &e); explist(ls, &e);
check_match(ls, ')', '(', line); check_match(ls, ')', '(', line);
close_exp(ls, e.pc, MULT_RET); /* close 1 for old semantics */ close_call(ls, e.pc, MULT_RET); /* close 1 for old semantics */
break; break;
} }
@@ -655,14 +591,14 @@ static void var_or_func_tail (LexState *ls, vardesc *v) {
switch (ls->token) { switch (ls->token) {
case '.': /* var_or_func_tail -> '.' NAME */ case '.': /* var_or_func_tail -> '.' NAME */
next(ls); next(ls);
lua_pushvar(ls, v); /* `v' must be on stack */ close_exp(ls, v); /* `v' must be on stack */
v->k = VDOT; code_kstr(ls, checkname(ls));
v->info = checkname(ls); v->k = VINDEXED;
break; break;
case '[': /* var_or_func_tail -> '[' exp1 ']' */ case '[': /* var_or_func_tail -> '[' exp1 ']' */
next(ls); next(ls);
lua_pushvar(ls, v); /* `v' must be on stack */ close_exp(ls, v); /* `v' must be on stack */
exp1(ls); exp1(ls);
check(ls, ']'); check(ls, ']');
v->k = VINDEXED; v->k = VINDEXED;
@@ -672,7 +608,7 @@ static void var_or_func_tail (LexState *ls, vardesc *v) {
int name; int name;
next(ls); next(ls);
name = checkname(ls); name = checkname(ls);
lua_pushvar(ls, v); /* `v' must be on stack */ close_exp(ls, v); /* `v' must be on stack */
code_U(ls, PUSHSELF, name, 1); code_U(ls, PUSHSELF, name, 1);
v->k = VEXP; v->k = VEXP;
v->info = funcparams(ls, 1); v->info = funcparams(ls, 1);
@@ -680,7 +616,7 @@ static void var_or_func_tail (LexState *ls, vardesc *v) {
} }
case '(': case STRING: case '{': /* var_or_func_tail -> funcparams */ case '(': case STRING: case '{': /* var_or_func_tail -> funcparams */
lua_pushvar(ls, v); /* `v' must be on stack */ close_exp(ls, v); /* `v' must be on stack */
v->k = VEXP; v->k = VEXP;
v->info = funcparams(ls, 0); v->info = funcparams(ls, 0);
break; break;
@@ -711,6 +647,7 @@ static void var_or_func (LexState *ls, vardesc *v) {
** ======================================================================= ** =======================================================================
*/ */
static void recfield (LexState *ls) { static void recfield (LexState *ls) {
/* recfield -> (NAME | '['exp1']') = exp1 */ /* recfield -> (NAME | '['exp1']') = exp1 */
switch (ls->token) { switch (ls->token) {
@@ -788,23 +725,14 @@ static void constructor_part (LexState *ls, constdesc *cd) {
vardesc v; vardesc v;
expr(ls, &v); expr(ls, &v);
if (ls->token == '=') { if (ls->token == '=') {
switch (v.k) { code_kstr(ls, getvarname(ls, &v));
case VGLOBAL: next(ls); /* skip '=' */
code_kstr(ls, v.info); exp1(ls);
break;
case VLOCAL:
code_string(ls, ls->fs->localvar[v.info]);
break;
default:
error_unexpected(ls);
}
next(ls);
exp1(ls);
cd->n = recfields(ls); cd->n = recfields(ls);
cd->k = 1; /* record */ cd->k = 1; /* record */
} }
else { else {
lua_pushvar(ls, &v); close_exp(ls, &v);
cd->n = listfields(ls); cd->n = listfields(ls);
cd->k = 0; /* list */ cd->k = 0; /* list */
} }
@@ -923,18 +851,12 @@ static void pop_to (LexState *ls, stack_op *s, int prio) {
} }
} }
static void simpleexp (LexState *ls, vardesc *v, stack_op *s) { static void simpleexp (LexState *ls, vardesc *v) {
check_debugline(ls); check_debugline(ls);
switch (ls->token) { switch (ls->token) {
case NUMBER: { /* simpleexp -> NUMBER */ case NUMBER: { /* simpleexp -> NUMBER */
real r = ls->seminfo.r; real r = ls->seminfo.r;
next(ls); next(ls);
/* dirty trick: check whether it is a -NUMBER not followed by '^' */
/* (because the priority of '^' is higher than the priority of '-') */
if (s->top > 0 && s->ops[s->top-1] == INDMINUS && ls->token != '^') {
s->top--; /* remove '-' from stack */
r = -r;
}
code_number(ls, r); code_number(ls, r);
break; break;
} }
@@ -982,7 +904,7 @@ static void prefixexp (LexState *ls, vardesc *v, stack_op *s) {
push(ls, s, (ls->token==NOT)?INDNOT:INDMINUS); push(ls, s, (ls->token==NOT)?INDNOT:INDMINUS);
next(ls); next(ls);
} }
simpleexp(ls, v, s); simpleexp(ls, v);
} }
@@ -992,16 +914,16 @@ static void arith_exp (LexState *ls, vardesc *v) {
s.top = 0; s.top = 0;
prefixexp(ls, v, &s); prefixexp(ls, v, &s);
while ((op = binop(ls->token)) >= 0) { while ((op = binop(ls->token)) >= 0) {
lua_pushvar(ls, v); close_exp(ls, v);
/* '^' is right associative, so must 'simulate' a higher priority */ /* '^' is right associative, so must 'simulate' a higher priority */
pop_to(ls, &s, (op == POW)?priority[op]+1:priority[op]); pop_to(ls, &s, (op == POW)?priority[op]+1:priority[op]);
push(ls, &s, op); push(ls, &s, op);
next(ls); next(ls);
prefixexp(ls, v, &s); prefixexp(ls, v, &s);
lua_pushvar(ls, v); close_exp(ls, v);
} }
if (s.top > 0) { if (s.top > 0) {
lua_pushvar(ls, v); close_exp(ls, v);
pop_to(ls, &s, 0); pop_to(ls, &s, 0);
} }
} }
@@ -1010,7 +932,7 @@ static void arith_exp (LexState *ls, vardesc *v) {
static void exp1 (LexState *ls) { static void exp1 (LexState *ls) {
vardesc v; vardesc v;
expr(ls, &v); expr(ls, &v);
lua_pushvar(ls, &v); close_exp(ls, &v);
} }
@@ -1020,12 +942,12 @@ static void expr (LexState *ls, vardesc *v) {
while (ls->token == AND || ls->token == OR) { while (ls->token == AND || ls->token == OR) {
OpCode op = (ls->token == AND) ? ONFJMP : ONTJMP; OpCode op = (ls->token == AND) ? ONFJMP : ONTJMP;
int pc; int pc;
lua_pushvar(ls, v); close_exp(ls, v);
next(ls); next(ls);
pc = code_S(ls, op, 0, -1); pc = code_S(ls, op, 0, -1);
arith_exp(ls, v); arith_exp(ls, v);
lua_pushvar(ls, v); close_exp(ls, v);
fix_jump(ls, pc, ls->fs->pc); luaK_fixjump(ls, pc, ls->fs->pc);
} }
} }
@@ -1054,7 +976,6 @@ static void block (LexState *ls) {
static int assignment (LexState *ls, vardesc *v, int nvars) { static int assignment (LexState *ls, vardesc *v, int nvars) {
int left = 0; int left = 0;
checklimit(ls, nvars, MAXVARSLH, "variables in a multiple assignment"); checklimit(ls, nvars, MAXVARSLH, "variables in a multiple assignment");
unloaddot(ls, v);
if (ls->token == ',') { /* assignment -> ',' NAME assignment */ if (ls->token == ',') { /* assignment -> ',' NAME assignment */
vardesc nv; vardesc nv;
next(ls); next(ls);
@@ -1083,19 +1004,37 @@ static int assignment (LexState *ls, vardesc *v, int nvars) {
} }
/* maximum size of a while condition */
#ifndef MAX_WHILE_EXP
#define MAX_WHILE_EXP 200 /* arbitrary limit */
#endif
static void whilestat (LexState *ls, int line) { static void whilestat (LexState *ls, int line) {
/* whilestat -> WHILE exp1 DO block END */ /* whilestat -> WHILE exp1 DO block END */
Instruction buffer[MAX_WHILE_EXP];
FuncState *fs = ls->fs; FuncState *fs = ls->fs;
int while_init = fs->pc; int while_init = fs->pc;
int j1; int cond_size;
next(ls); int i;
exp1(ls); next(ls); /* skip WHILE */
j1 = code_U(ls, IFFJMP, 0, -1); /* jump to exit loop */ exp1(ls); /* read condition */
cond_size = fs->pc - while_init;
/* save condition (to move it to after body) */
if (cond_size > MAX_WHILE_EXP)
luaY_error(ls, "while condition too complex");
for (i=0; i<cond_size; i++) buffer[i] = fs->f->code[while_init+i];
/* go back to state prior condition */
fs->pc = while_init;
deltastack(ls, -1);
code_S(ls, JMP, 0, 0); /* initial jump to condition */
check(ls, DO); check(ls, DO);
block(ls); block(ls);
check_match(ls, END, WHILE, line); check_match(ls, END, WHILE, line);
fix_jump(ls, code_U(ls, JMP, 0, 0), while_init); /* jump to keep loop */ luaK_fixjump(ls, while_init, fs->pc);
fix_jump(ls, j1, fs->pc); /* copy condition to new position, and correct stack */
for (i=0; i<cond_size; i++) luaK_primitivecode(ls, buffer[i]);
deltastack(ls, 1);
luaK_fixjump(ls, code_S(ls, IFTJMP, 0, -1), while_init+1);
} }
@@ -1107,7 +1046,7 @@ static void repeatstat (LexState *ls, int line) {
block(ls); block(ls);
check_match(ls, UNTIL, REPEAT, line); check_match(ls, UNTIL, REPEAT, line);
exp1(ls); exp1(ls);
fix_jump(ls, code_U(ls, IFFJMP, 0, -1), repeat_init); luaK_fixjump(ls, code_S(ls, IFFJMP, 0, -1), repeat_init);
} }
@@ -1157,7 +1096,7 @@ static int funcname (LexState *ls, vardesc *v) {
if (ls->token == ':' || ls->token == '.') { if (ls->token == ':' || ls->token == '.') {
needself = (ls->token == ':'); needself = (ls->token == ':');
next(ls); next(ls);
lua_pushvar(ls, v); close_exp(ls, v);
code_kstr(ls, checkname(ls)); code_kstr(ls, checkname(ls));
v->k = VINDEXED; v->k = VINDEXED;
} }
@@ -1188,7 +1127,7 @@ static void namestat (LexState *ls) {
if (v.k == VEXP) { /* stat -> func */ if (v.k == VEXP) { /* stat -> func */
if (v.info == 0) /* is just an upper value? */ if (v.info == 0) /* is just an upper value? */
luaY_error(ls, "syntax error"); luaY_error(ls, "syntax error");
close_exp(ls, v.info, 0); close_call(ls, v.info, 0); /* call statement uses no results */
} }
else { /* stat -> ['%'] NAME assignment */ else { /* stat -> ['%'] NAME assignment */
int left = assignment(ls, &v, 1); int left = assignment(ls, &v, 1);
@@ -1200,14 +1139,16 @@ static void namestat (LexState *ls) {
static void ifpart (LexState *ls, int line) { static void ifpart (LexState *ls, int line) {
/* ifpart -> cond THEN block [ELSE block | ELSEIF ifpart] */ /* ifpart -> cond THEN block [ELSE block | ELSEIF ifpart] */
FuncState *fs = ls->fs; FuncState *fs = ls->fs;
int c; int c; /* address of the conditional jump */
int je; int je; /* address of the unconditional jump (to skip `else' part) */
int elseinit;
next(ls); /* skip IF or ELSEIF */ next(ls); /* skip IF or ELSEIF */
exp1(ls); /* cond */ exp1(ls); /* cond */
c = code_U(ls, IFFJMP, 0, -1); /* jump `then' if `cond' is false */ c = code_S(ls, IFFJMP, 0, -1); /* jump `then' if `cond' is false */
check(ls, THEN); check(ls, THEN);
block(ls); /* `then' part */ block(ls); /* `then' part */
je = code_U(ls, JMP, 0, 0); /* jump `else' part after `then' */ je = code_S(ls, JMP, 0, 0); /* jump `else' part after `then' */
elseinit = fs->pc;
if (ls->token == ELSEIF) if (ls->token == ELSEIF)
ifpart(ls, line); ifpart(ls, line);
else { else {
@@ -1215,13 +1156,14 @@ static void ifpart (LexState *ls, int line) {
block(ls); /* `else' part */ block(ls); /* `else' part */
check_match(ls, END, IF, line); check_match(ls, END, IF, line);
} }
if (fs->pc == je+1) { /* `else' part empty? */ if (fs->pc > elseinit) /* is there an `else' part? */
luaK_fixjump(ls, je, fs->pc); /* last jump jumps over it */
else {
fs->pc--; /* remove last jump */ fs->pc--; /* remove last jump */
je--; /* first jump will be smaller */ elseinit--; /* first jump will be smaller */
LUA_ASSERT(L, fs->pc == je, "jump out of place");
} }
else luaK_fixjump(ls, c, elseinit); /* fix first jump to `else' part */
fix_jump(ls, je, fs->pc); /* fix last jump */
fix_jump(ls, c, je+1); /* fix first jump to beginning of `else' part */
} }
@@ -1329,13 +1271,8 @@ static void ret (LexState *ls) {
check_debugline(ls); check_debugline(ls);
next(ls); next(ls);
explist(ls, &e); explist(ls, &e);
if (e.pc > 0) { /* expression is an open function call? */ close_call(ls, e.pc, MULT_RET);
Instruction *i = &ls->fs->f->code[e.pc]; code_U(ls, RETCODE, ls->fs->nlocalvar, 0);
*i = SET_OPCODE(*i, TAILCALL); /* instead of a conventional CALL */
*i = SETARG_B(*i, ls->fs->nlocalvar);
}
else
code_U(ls, RETCODE, ls->fs->nlocalvar, 0);
ls->fs->stacksize = ls->fs->nlocalvar; /* removes all temp values */ ls->fs->stacksize = ls->fs->nlocalvar; /* removes all temp values */
optional(ls, ';'); optional(ls, ';');
} }

View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lparser.h,v 1.4 1999/08/16 20:52:00 roberto Exp roberto $ ** $Id: lparser.h,v 1.5 1999/11/22 13:12:07 roberto Exp roberto $
** LL(1) Parser and code generator for Lua ** LL(1) Parser and code generator for Lua
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -8,9 +8,81 @@
#define lparser_h #define lparser_h
#include "lobject.h" #include "lobject.h"
#include "lopcodes.h"
#include "lzio.h" #include "lzio.h"
/* maximum number of local variables */
#ifndef MAXLOCALS
#define MAXLOCALS 200 /* arbitrary limit (<=MAXARG_B) */
#endif
/* maximum number of upvalues */
#ifndef MAXUPVALUES
#define MAXUPVALUES 32 /* arbitrary limit (<=MAXARG_B) */
#endif
/* maximum number of variables in the left side of an assignment */
#ifndef MAXVARSLH
#define MAXVARSLH 100 /* arbitrary limit (<=MAXARG_B) */
#endif
/* maximum number of parameters in a function */
#ifndef MAXPARAMS
#define MAXPARAMS 100 /* arbitrary limit (<=MAXLOCALS) */
#endif
/* maximum stack size in a function */
#ifndef MAXSTACK
#define MAXSTACK 256 /* arbitrary limit (<=MAXARG_A) */
#endif
#if MAXLOCALS>MAXARG_U || MAXUPVALUES>MAXARG_B || MAXVARSLH>MAXARG_B || \
MAXPARAMS>MAXLOCALS || MAXSTACK>MAXARG_A || LFIELDS_PER_FLUSH>MAXARG_B
#error invalid limits
#endif
/*
** Variable descriptor:
** must include an `exp' option because LL(1) cannot distinguish
** between variables, upvalues and function calls on first sight.
*/
typedef enum {
VGLOBAL, /* info is constant index of global name */
VLOCAL, /* info is stack index */
VINDEXED, /* no info (table and index are on the stack) */
VEXP /* info is pc index of a call (or 0 if exp is closed) */
} varkind;
typedef struct vardesc {
varkind k;
int info;
} vardesc;
/* state needed to generate code for a given function */
typedef struct FuncState {
TProtoFunc *f; /* current function header */
struct FuncState *prev; /* enclosing function */
int pc; /* next position to code */
int last_pc; /* last instruction coded (for optimizations) */
int stacksize; /* number of values on activation register */
int nlocalvar; /* number of active local variables */
int nupvalues; /* number of upvalues */
int nvars; /* number of entries in f->locvars (-1 if no debug information) */
int lastsetline; /* line where last SETLINE was issued */
vardesc upvalues[MAXUPVALUES]; /* upvalues */
TaggedString *localvar[MAXLOCALS]; /* store local variable names */
} FuncState;
TProtoFunc *luaY_parser (lua_State *L, ZIO *z); TProtoFunc *luaY_parser (lua_State *L, ZIO *z);

20
lvm.c
View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lvm.c,v 1.86 2000/02/11 16:52:54 roberto Exp roberto $ ** $Id: lvm.c,v 1.87 2000/02/14 16:51:08 roberto Exp roberto $
** Lua virtual machine ** Lua virtual machine
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -376,12 +376,18 @@ StkId luaV_execute (lua_State *L, const Closure *cl, const TProtoFunc *tf,
top++; top++;
break; break;
case PUSHNUMBER: case PUSHNUM:
ttype(top) = LUA_T_NUMBER; ttype(top) = LUA_T_NUMBER;
nvalue(top) = tf->knum[GETARG_U(i)]; nvalue(top) = tf->knum[GETARG_U(i)];
top++; top++;
break; break;
case PUSHNEGNUM:
ttype(top) = LUA_T_NUMBER;
nvalue(top) = -tf->knum[GETARG_U(i)];
top++;
break;
case PUSHUPVALUE: case PUSHUPVALUE:
*top++ = cl->consts[GETARG_U(i)+1]; *top++ = cl->consts[GETARG_U(i)+1];
break; break;
@@ -524,6 +530,16 @@ StkId luaV_execute (lua_State *L, const Closure *cl, const TProtoFunc *tf,
top--; top--;
break; break;
case ADDI:
if (tonumber(top-1)) {
ttype(top) = LUA_T_NUMBER;
nvalue(top) = (real)GETARG_S(i);
call_arith(L, top+1, IM_ADD);
}
else
nvalue(top-1) += (real)GETARG_S(i);
break;
case SUBOP: case SUBOP:
if (tonumber(top-1) || tonumber(top-2)) if (tonumber(top-1) || tonumber(top-2))
call_arith(L, top, IM_SUB); call_arith(L, top, IM_SUB);

View File

@@ -1,5 +1,5 @@
# #
## $Id: makefile,v 1.20 1999/08/17 20:21:52 roberto Exp roberto $ ## $Id: makefile,v 1.21 1999/10/04 17:51:04 roberto Exp roberto $
## Makefile ## Makefile
## See Copyright Notice in lua.h ## See Copyright Notice in lua.h
# #
@@ -20,14 +20,17 @@
# #
# define LUA_COMPAT_GC if you need garbage-collect tag methods for tables # define LUA_COMPAT_GC if you need garbage-collect tag methods for tables
# (only for compatibility with previous versions) # (only for compatibility with previous versions)
#
# define COMPAT_READPATTERN if you need read patterns
# (only for compatibility with previous versions)
CONFIG = -DPOPEN -D_POSIX_SOURCE CONFIG = -DPOPEN -D_POSIX_SOURCE
#CONFIG = -DOLD_ANSI -DDEBUG -DLUA_COMPAT_GC #CONFIG = -DOLD_ANSI -DDEBUG -DLUA_COMPAT_GC -DCOMPAT_READPATTERN
# Compilation parameters # Compilation parameters
CC = gcc CC = gcc
CWARNS = -Wall -Wmissing-prototypes -Wshadow -pedantic -Wpointer-arith -Wcast-align -Waggregate-return -Wwrite-strings -Wcast-qual -Wstrict-prototypes -Wmissing-declarations -Wnested-externs -Werror CWARNS = -Wall -W -Wmissing-prototypes -Wshadow -pedantic -Wpointer-arith -Wcast-align -Waggregate-return -Wwrite-strings -Wcast-qual -Wnested-externs
CFLAGS = $(CONFIG) $(CWARNS) -ansi -O2 CFLAGS = $(CONFIG) $(CWARNS) -ansi -O2
@@ -45,6 +48,7 @@ LUAOBJS = \
lauxlib.o \ lauxlib.o \
lbuffer.o \ lbuffer.o \
lbuiltin.o \ lbuiltin.o \
ldebug.o \
ldo.o \ ldo.o \
lfunc.o \ lfunc.o \
lgc.o \ lgc.o \
@@ -52,6 +56,7 @@ LUAOBJS = \
lmem.o \ lmem.o \
lobject.o \ lobject.o \
lparser.o \ lparser.o \
lcode.o \
lref.o \ lref.o \
lstate.o \ lstate.o \
lstring.o \ lstring.o \
@@ -59,7 +64,8 @@ LUAOBJS = \
ltm.o \ ltm.o \
lvm.o \ lvm.o \
lundump.o \ lundump.o \
lzio.o lzio.o \
ltests.o
LIBOBJS = \ LIBOBJS = \
liolib.o \ liolib.o \
@@ -86,9 +92,7 @@ liblua.so.1.0 : lua.o
clear : clear :
rcsclean rcsclean
rm -f *.o rm -f *.o *.a
rm -f
co $(CO_OPTIONS) lua.h lualib.h luadebug.h
%.h : RCS/%.h,v %.h : RCS/%.h,v
@@ -98,42 +102,47 @@ clear :
co $(CO_OPTIONS) $@ co $(CO_OPTIONS) $@
lapi.o: lapi.c lapi.h lua.h lobject.h lauxlib.h ldo.h lstate.h \ lapi.o: lapi.c lapi.h lobject.h lua.h lauxlib.h ldo.h lstate.h \
luadebug.h lfunc.h lgc.h lmem.h lref.h lstring.h ltable.h ltm.h lvm.h luadebug.h lfunc.h lgc.h lmem.h lref.h lstring.h ltable.h ltm.h lvm.h
lauxlib.o: lauxlib.c lauxlib.h lua.h luadebug.h lauxlib.o: lauxlib.c lauxlib.h lua.h luadebug.h
lbuffer.o: lbuffer.c lauxlib.h lua.h lmem.h lstate.h lobject.h \ lbuffer.o: lbuffer.c lauxlib.h lua.h lmem.h lstate.h lobject.h \
luadebug.h luadebug.h
lbuiltin.o: lbuiltin.c lapi.h lua.h lobject.h lauxlib.h lbuiltin.h \ lbuiltin.o: lbuiltin.c lapi.h lobject.h lua.h lauxlib.h lbuiltin.h \
ldo.h lstate.h luadebug.h lfunc.h lmem.h lstring.h ltable.h ltm.h \ ldo.h lstate.h luadebug.h lfunc.h lmem.h lstring.h ltable.h ltm.h \
lundump.h lzio.h lvm.h lundump.h lzio.h lvm.h
lcode.o: lcode.c lcode.h llex.h lobject.h lua.h lzio.h lmem.h \
lopcodes.h lparser.h
ldblib.o: ldblib.c lauxlib.h lua.h luadebug.h lualib.h ldblib.o: ldblib.c lauxlib.h lua.h luadebug.h lualib.h
ldo.o: ldo.c lauxlib.h lua.h ldo.h lobject.h lstate.h luadebug.h lgc.h \ ldebug.o: ldebug.c lapi.h lobject.h lua.h lauxlib.h ldebug.h \
lmem.h lparser.h lzio.h lstring.h ltm.h lundump.h lvm.h luadebug.h ldo.h lstate.h lfunc.h ltable.h ltm.h
ldo.o: ldo.c lauxlib.h lua.h ldebug.h lobject.h luadebug.h ldo.h \
lstate.h lgc.h lmem.h lparser.h lopcodes.h lzio.h lstring.h ltm.h \
lundump.h lvm.h
lfunc.o: lfunc.c lfunc.h lobject.h lua.h lmem.h lstate.h luadebug.h lfunc.o: lfunc.c lfunc.h lobject.h lua.h lmem.h lstate.h luadebug.h
lgc.o: lgc.c ldo.h lobject.h lua.h lstate.h luadebug.h lfunc.h lgc.h \ lgc.o: lgc.c ldo.h lobject.h lua.h lstate.h luadebug.h lfunc.h lgc.h \
lref.h lstring.h ltable.h ltm.h lmem.h lref.h lstring.h ltable.h ltm.h
linit.o: linit.c lua.h lualib.h linit.o: linit.c lua.h lualib.h
liolib.o: liolib.c lauxlib.h lua.h luadebug.h lualib.h liolib.o: liolib.c lauxlib.h lua.h luadebug.h lualib.h
llex.o: llex.c lauxlib.h lua.h llex.h lobject.h lzio.h lmem.h \ llex.o: llex.c lauxlib.h lua.h llex.h lobject.h lzio.h lmem.h \
lparser.h lstate.h luadebug.h lstring.h lparser.h lopcodes.h lstate.h luadebug.h lstring.h
lmathlib.o: lmathlib.c lauxlib.h lua.h lualib.h lmathlib.o: lmathlib.c lauxlib.h lua.h lualib.h
lmem.o: lmem.c lmem.h lstate.h lobject.h lua.h luadebug.h lmem.o: lmem.c lmem.h lua.h lobject.h lstate.h luadebug.h
lobject.o: lobject.c lobject.h lua.h lobject.o: lobject.c lobject.h lua.h
lparser.o: lparser.c ldo.h lobject.h lua.h lstate.h luadebug.h lfunc.h \ lparser.o: lparser.c lcode.h llex.h lobject.h lua.h lzio.h ldo.h \
llex.h lzio.h lmem.h lopcodes.h lparser.h lstring.h lstate.h luadebug.h lfunc.h lmem.h lopcodes.h lparser.h lstring.h
lref.o: lref.c lmem.h lref.h lobject.h lua.h lstate.h luadebug.h lref.o: lref.c lapi.h lobject.h lua.h lmem.h lref.h lstate.h \
lstate.o: lstate.c lbuiltin.h ldo.h lobject.h lua.h lstate.h \ luadebug.h
luadebug.h lgc.h llex.h lzio.h lmem.h lstring.h ltm.h lstate.o: lstate.c lauxlib.h lua.h lbuiltin.h ldo.h lobject.h lstate.h \
lstring.o: lstring.c lmem.h lobject.h lua.h lstate.h luadebug.h \ luadebug.h lgc.h llex.h lzio.h lmem.h lref.h lstring.h ltm.h
lstring.o: lstring.c lmem.h lua.h lobject.h lstate.h luadebug.h \
lstring.h lstring.h
lstrlib.o: lstrlib.c lauxlib.h lua.h lualib.h lstrlib.o: lstrlib.c lauxlib.h lua.h lualib.h
ltable.o: ltable.c lauxlib.h lua.h lmem.h lobject.h lstate.h \ ltable.o: ltable.c lauxlib.h lua.h lmem.h lobject.h lstate.h \
luadebug.h ltable.h luadebug.h ltable.h
ltm.o: ltm.c lauxlib.h lua.h lmem.h lobject.h lstate.h luadebug.h \ ltm.o: ltm.c lauxlib.h lua.h lmem.h lobject.h lstate.h luadebug.h \
ltm.h ltm.h
lua.o: lua.c lua.h luadebug.h lualib.h
lundump.o: lundump.c lauxlib.h lua.h lfunc.h lobject.h lmem.h \ lundump.o: lundump.c lauxlib.h lua.h lfunc.h lobject.h lmem.h \
lopcodes.h lstring.h lundump.h lzio.h lopcodes.h lstring.h lstate.h luadebug.h lundump.h lzio.h
lvm.o: lvm.c lauxlib.h lua.h ldo.h lobject.h lstate.h luadebug.h \ lvm.o: lvm.c lauxlib.h lua.h ldebug.h lobject.h luadebug.h ldo.h \
lfunc.h lgc.h lopcodes.h lstring.h ltable.h ltm.h lvm.h lstate.h lfunc.h lgc.h lopcodes.h lstring.h ltable.h ltm.h lvm.h
lzio.o: lzio.c lzio.h lzio.o: lzio.c lzio.h