Lines Matching refs:fs
59 lua_assert(ls->fs != NULL || ls->t.token == TK_EOS); in anchor_token()
80 static l_noret errorlimit (FuncState *fs, int limit, const char *what) { in errorlimit() argument
81 lua_State *L = fs->ls->L; in errorlimit()
83 int line = fs->f->linedefined; in errorlimit()
89 luaX_syntaxerror(fs->ls, msg); in errorlimit()
93 static void checklimit (FuncState *fs, int v, int l, const char *what) { in checklimit() argument
94 if (v > l) errorlimit(fs, l, what); in checklimit()
153 init_exp(e, VK, luaK_stringK(ls->fs, s)); in codestring()
163 FuncState *fs = ls->fs; in registerlocalvar() local
164 Proto *f = fs->f; in registerlocalvar()
166 luaM_growvector(ls->L, f->locvars, fs->nlocvars, f->sizelocvars, in registerlocalvar()
169 f->locvars[fs->nlocvars].varname = varname; in registerlocalvar()
171 return fs->nlocvars++; in registerlocalvar()
176 FuncState *fs = ls->fs; in new_localvar() local
179 checklimit(fs, dyd->actvar.n + 1 - fs->firstlocal, in new_localvar()
195 static LocVar *getlocvar (FuncState *fs, int i) { in getlocvar() argument
196 int idx = fs->ls->dyd->actvar.arr[fs->firstlocal + i].idx; in getlocvar()
197 lua_assert(idx < fs->nlocvars); in getlocvar()
198 return &fs->f->locvars[idx]; in getlocvar()
203 FuncState *fs = ls->fs; in adjustlocalvars() local
204 fs->nactvar = cast_byte(fs->nactvar + nvars); in adjustlocalvars()
206 getlocvar(fs, fs->nactvar - nvars)->startpc = fs->pc; in adjustlocalvars()
211 static void removevars (FuncState *fs, int tolevel) { in removevars() argument
212 fs->ls->dyd->actvar.n -= (fs->nactvar - tolevel); in removevars()
213 while (fs->nactvar > tolevel) in removevars()
214 getlocvar(fs, --fs->nactvar)->endpc = fs->pc; in removevars()
218 static int searchupvalue (FuncState *fs, TString *name) { in searchupvalue() argument
220 Upvaldesc *up = fs->f->upvalues; in searchupvalue()
221 for (i = 0; i < fs->nups; i++) { in searchupvalue()
228 static int newupvalue (FuncState *fs, TString *name, expdesc *v) { in newupvalue() argument
229 Proto *f = fs->f; in newupvalue()
231 checklimit(fs, fs->nups + 1, MAXUPVAL, "upvalues"); in newupvalue()
232 luaM_growvector(fs->ls->L, f->upvalues, fs->nups, f->sizeupvalues, in newupvalue()
235 f->upvalues[fs->nups].instack = (v->k == VLOCAL); in newupvalue()
236 f->upvalues[fs->nups].idx = cast_byte(v->u.info); in newupvalue()
237 f->upvalues[fs->nups].name = name; in newupvalue()
238 luaC_objbarrier(fs->ls->L, f, name); in newupvalue()
239 return fs->nups++; in newupvalue()
243 static int searchvar (FuncState *fs, TString *n) { in searchvar() argument
245 for (i = cast_int(fs->nactvar) - 1; i >= 0; i--) { in searchvar()
246 if (luaS_eqstr(n, getlocvar(fs, i)->varname)) in searchvar()
257 static void markupval (FuncState *fs, int level) { in markupval() argument
258 BlockCnt *bl = fs->bl; in markupval()
268 static int singlevaraux (FuncState *fs, TString *n, expdesc *var, int base) { in singlevaraux() argument
269 if (fs == NULL) /* no more levels? */ in singlevaraux()
272 int v = searchvar(fs, n); /* look up locals at current level */ in singlevaraux()
276 markupval(fs, v); /* local will be used as an upval */ in singlevaraux()
280 int idx = searchupvalue(fs, n); /* try existing upvalues */ in singlevaraux()
282 if (singlevaraux(fs->prev, n, var, 0) == VVOID) /* try upper levels */ in singlevaraux()
285 idx = newupvalue(fs, n, var); /* will be a new upvalue */ in singlevaraux()
296 FuncState *fs = ls->fs; in singlevar() local
297 if (singlevaraux(fs, varname, var, 1) == VVOID) { /* global name? */ in singlevar()
299 singlevaraux(fs, ls->envn, var, 1); /* get environment variable */ in singlevar()
302 luaK_indexed(fs, var, &key); /* env[varname] */ in singlevar()
308 FuncState *fs = ls->fs; in adjust_assign() local
313 luaK_setreturns(fs, e, extra); /* last exp. provides the difference */ in adjust_assign()
314 if (extra > 1) luaK_reserveregs(fs, extra-1); in adjust_assign()
317 if (e->k != VVOID) luaK_exp2nextreg(fs, e); /* close last expression */ in adjust_assign()
319 int reg = fs->freereg; in adjust_assign()
320 luaK_reserveregs(fs, extra); in adjust_assign()
321 luaK_nil(fs, reg, extra); in adjust_assign()
330 checklimit(ls->fs, L->nCcalls, LUAI_MAXCCALLS, "C levels"); in enterlevel()
339 FuncState *fs = ls->fs; in closegoto() local
344 TString *vname = getlocvar(fs, gt->nactvar)->varname; in closegoto()
350 luaK_patchlist(fs, gt->pc, label->pc); in closegoto()
363 BlockCnt *bl = ls->fs->bl; in findlabel()
372 luaK_patchclose(ls->fs, gt->pc, lb->nactvar); in findlabel()
388 l->arr[n].nactvar = ls->fs->nactvar; in newlabelentry()
401 int i = ls->fs->bl->firstgoto; in findgotos()
417 static void movegotosout (FuncState *fs, BlockCnt *bl) { in movegotosout() argument
419 Labellist *gl = &fs->ls->dyd->gt; in movegotosout()
426 luaK_patchclose(fs, gt->pc, bl->nactvar); in movegotosout()
429 if (!findlabel(fs->ls, i)) in movegotosout()
435 static void enterblock (FuncState *fs, BlockCnt *bl, lu_byte isloop) { in enterblock() argument
437 bl->nactvar = fs->nactvar; in enterblock()
438 bl->firstlabel = fs->ls->dyd->label.n; in enterblock()
439 bl->firstgoto = fs->ls->dyd->gt.n; in enterblock()
441 bl->previous = fs->bl; in enterblock()
442 fs->bl = bl; in enterblock()
443 lua_assert(fs->freereg == fs->nactvar); in enterblock()
452 int l = newlabelentry(ls, &ls->dyd->label, n, 0, ls->fs->pc); in breaklabel()
469 static void leaveblock (FuncState *fs) { in leaveblock() argument
470 BlockCnt *bl = fs->bl; in leaveblock()
471 LexState *ls = fs->ls; in leaveblock()
474 int j = luaK_jump(fs); in leaveblock()
475 luaK_patchclose(fs, j, bl->nactvar); in leaveblock()
476 luaK_patchtohere(fs, j); in leaveblock()
480 fs->bl = bl->previous; in leaveblock()
481 removevars(fs, bl->nactvar); in leaveblock()
482 lua_assert(bl->nactvar == fs->nactvar); in leaveblock()
483 fs->freereg = fs->nactvar; /* free registers */ in leaveblock()
486 movegotosout(fs, bl); /* update pending gotos to outer block */ in leaveblock()
498 FuncState *fs = ls->fs; in addprototype() local
499 Proto *f = fs->f; /* prototype of current function */ in addprototype()
500 if (fs->np >= f->sizep) { in addprototype()
502 luaM_growvector(L, f->p, fs->np, f->sizep, Proto *, MAXARG_Bx, "functions"); in addprototype()
505 f->p[fs->np++] = clp = luaF_newproto(L); in addprototype()
518 FuncState *fs = ls->fs->prev; in codeclosure() local
519 init_exp(v, VRELOCABLE, luaK_codeABx(fs, OP_CLOSURE, 0, fs->np - 1)); in codeclosure()
520 luaK_exp2nextreg(fs, v); /* fix it at the last register */ in codeclosure()
524 static void open_func (LexState *ls, FuncState *fs, BlockCnt *bl) { in open_func() argument
527 fs->prev = ls->fs; /* linked list of funcstates */ in open_func()
528 fs->ls = ls; in open_func()
529 ls->fs = fs; in open_func()
530 fs->pc = 0; in open_func()
531 fs->lasttarget = 0; in open_func()
532 fs->jpc = NO_JUMP; in open_func()
533 fs->freereg = 0; in open_func()
534 fs->nk = 0; in open_func()
535 fs->np = 0; in open_func()
536 fs->nups = 0; in open_func()
537 fs->nlocvars = 0; in open_func()
538 fs->nactvar = 0; in open_func()
539 fs->firstlocal = ls->dyd->actvar.n; in open_func()
540 fs->bl = NULL; in open_func()
541 f = fs->f; in open_func()
544 fs->h = luaH_new(L); in open_func()
546 sethvalue2s(L, L->top, fs->h); in open_func()
548 enterblock(fs, bl, 0); in open_func()
554 FuncState *fs = ls->fs; in close_func() local
555 Proto *f = fs->f; in close_func()
556 luaK_ret(fs, 0, 0); /* final return */ in close_func()
557 leaveblock(fs); in close_func()
558 luaM_reallocvector(L, f->code, f->sizecode, fs->pc, Instruction); in close_func()
559 f->sizecode = fs->pc; in close_func()
560 luaM_reallocvector(L, f->lineinfo, f->sizelineinfo, fs->pc, int); in close_func()
561 f->sizelineinfo = fs->pc; in close_func()
562 luaM_reallocvector(L, f->k, f->sizek, fs->nk, TValue); in close_func()
563 f->sizek = fs->nk; in close_func()
564 luaM_reallocvector(L, f->p, f->sizep, fs->np, Proto *); in close_func()
565 f->sizep = fs->np; in close_func()
566 luaM_reallocvector(L, f->locvars, f->sizelocvars, fs->nlocvars, LocVar); in close_func()
567 f->sizelocvars = fs->nlocvars; in close_func()
568 luaM_reallocvector(L, f->upvalues, f->sizeupvalues, fs->nups, Upvaldesc); in close_func()
569 f->sizeupvalues = fs->nups; in close_func()
570 lua_assert(fs->bl == NULL); in close_func()
571 ls->fs = fs->prev; in close_func()
621 FuncState *fs = ls->fs; in fieldsel() local
623 luaK_exp2anyregup(fs, v); in fieldsel()
626 luaK_indexed(fs, v, &key); in fieldsel()
634 luaK_exp2val(ls->fs, v); in yindex()
657 FuncState *fs = ls->fs; in recfield() local
658 int reg = ls->fs->freereg; in recfield()
662 checklimit(fs, cc->nh, MAX_INT, "items in a constructor"); in recfield()
669 rkkey = luaK_exp2RK(fs, &key); in recfield()
671 luaK_codeABC(fs, OP_SETTABLE, cc->t->u.info, rkkey, luaK_exp2RK(fs, &val)); in recfield()
672 fs->freereg = reg; /* free registers */ in recfield()
676 static void closelistfield (FuncState *fs, struct ConsControl *cc) { in closelistfield() argument
678 luaK_exp2nextreg(fs, &cc->v); in closelistfield()
681 luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore); /* flush */ in closelistfield()
687 static void lastlistfield (FuncState *fs, struct ConsControl *cc) { in lastlistfield() argument
690 luaK_setmultret(fs, &cc->v); in lastlistfield()
691 luaK_setlist(fs, cc->t->u.info, cc->na, LUA_MULTRET); in lastlistfield()
696 luaK_exp2nextreg(fs, &cc->v); in lastlistfield()
697 luaK_setlist(fs, cc->t->u.info, cc->na, cc->tostore); in lastlistfield()
705 checklimit(ls->fs, cc->na, MAX_INT, "items in a constructor"); in listfield()
736 FuncState *fs = ls->fs; in constructor() local
738 int pc = luaK_codeABC(fs, OP_NEWTABLE, 0, 0, 0); in constructor()
744 luaK_exp2nextreg(ls->fs, t); /* fix it at stack top */ in constructor()
749 closelistfield(fs, &cc); in constructor()
753 lastlistfield(fs, &cc); in constructor()
754 SETARG_B(fs->f->code[pc], luaO_int2fb(cc.na)); /* set initial array size */ in constructor()
755 SETARG_C(fs->f->code[pc], luaO_int2fb(cc.nh)); /* set initial table size */ in constructor()
764 FuncState *fs = ls->fs; in parlist() local
765 Proto *f = fs->f; in parlist()
786 f->numparams = cast_byte(fs->nactvar); in parlist()
787 luaK_reserveregs(fs, fs->nactvar); /* reserve register for parameters */ in parlist()
818 luaK_exp2nextreg(ls->fs, v); in explist()
827 FuncState *fs = ls->fs; in funcargs() local
837 luaK_setmultret(fs, &args); in funcargs()
861 luaK_exp2nextreg(fs, &args); /* close last argument */ in funcargs()
862 nparams = fs->freereg - (base+1); in funcargs()
864 init_exp(f, VCALL, luaK_codeABC(fs, OP_CALL, base, nparams+1, 2)); in funcargs()
865 luaK_fixline(fs, line); in funcargs()
866 fs->freereg = base+1; /* call remove function and arguments and leaves in funcargs()
888 luaK_dischargevars(ls->fs, v); in primaryexp()
905 FuncState *fs = ls->fs; in suffixedexp() local
916 luaK_exp2anyregup(fs, v); in suffixedexp()
918 luaK_indexed(fs, v, &key); in suffixedexp()
925 luaK_self(fs, v, &key); in suffixedexp()
930 luaK_exp2nextreg(fs, v); in suffixedexp()
966 FuncState *fs = ls->fs; in simpleexp() local
967 check_condition(ls, fs->f->is_vararg, in simpleexp()
969 init_exp(v, VVARARG, luaK_codeABC(fs, OP_VARARG, 0, 1, 0)); in simpleexp()
1049 luaK_prefix(ls->fs, uop, v, line); in subexpr()
1059 luaK_infix(ls->fs, op, v); in subexpr()
1062 luaK_posfix(ls->fs, op, v, &v2, line); in subexpr()
1087 FuncState *fs = ls->fs; in block() local
1089 enterblock(fs, &bl, 0); in block()
1091 leaveblock(fs); in block()
1112 FuncState *fs = ls->fs; in check_conflict() local
1113 int extra = fs->freereg; /* eventual position to save local variable */ in check_conflict()
1133 luaK_codeABC(fs, op, extra, v->u.info, 0); in check_conflict()
1134 luaK_reserveregs(fs, 1); in check_conflict()
1148 checklimit(ls->fs, nvars + ls->L->nCcalls, LUAI_MAXCCALLS, in assignment()
1159 ls->fs->freereg -= nexps - nvars; /* remove extra values */ in assignment()
1162 luaK_setoneret(ls->fs, &e); /* close last expression */ in assignment()
1163 luaK_storevar(ls->fs, &lh->v, &e); in assignment()
1167 init_exp(&e, VNONRELOC, ls->fs->freereg-1); /* default assignment */ in assignment()
1168 luaK_storevar(ls->fs, &lh->v, &e); in assignment()
1177 luaK_goiftrue(ls->fs, &v); in cond()
1198 static void checkrepeated (FuncState *fs, Labellist *ll, TString *label) { in checkrepeated() argument
1200 for (i = fs->bl->firstlabel; i < ll->n; i++) { in checkrepeated()
1202 const char *msg = luaO_pushfstring(fs->ls->L, in checkrepeated()
1205 semerror(fs->ls, msg); in checkrepeated()
1220 FuncState *fs = ls->fs; in labelstat() local
1223 checkrepeated(fs, ll, label); /* check for repeated labels */ in labelstat()
1226 l = newlabelentry(ls, ll, label, line, fs->pc); in labelstat()
1230 ll->arr[l].nactvar = fs->bl->nactvar; in labelstat()
1238 FuncState *fs = ls->fs; in whilestat() local
1243 whileinit = luaK_getlabel(fs); in whilestat()
1245 enterblock(fs, &bl, 1); in whilestat()
1248 luaK_jumpto(fs, whileinit); in whilestat()
1250 leaveblock(fs); in whilestat()
1251 luaK_patchtohere(fs, condexit); /* false conditions finish the loop */ in whilestat()
1258 FuncState *fs = ls->fs; in repeatstat() local
1259 int repeat_init = luaK_getlabel(fs); in repeatstat()
1261 enterblock(fs, &bl1, 1); /* loop block */ in repeatstat()
1262 enterblock(fs, &bl2, 0); /* scope block */ in repeatstat()
1268 luaK_patchclose(fs, condexit, bl2.nactvar); in repeatstat()
1269 leaveblock(fs); /* finish scope */ in repeatstat()
1270 luaK_patchlist(fs, condexit, repeat_init); /* close the loop */ in repeatstat()
1271 leaveblock(fs); /* finish loop */ in repeatstat()
1279 luaK_exp2nextreg(ls->fs, &e); in exp1()
1289 FuncState *fs = ls->fs; in forbody() local
1293 prep = isnum ? luaK_codeAsBx(fs, OP_FORPREP, base, NO_JUMP) : luaK_jump(fs); in forbody()
1294 enterblock(fs, &bl, 0); /* scope for declared variables */ in forbody()
1296 luaK_reserveregs(fs, nvars); in forbody()
1298 leaveblock(fs); /* end of scope for declared variables */ in forbody()
1299 luaK_patchtohere(fs, prep); in forbody()
1301 endfor = luaK_codeAsBx(fs, OP_FORLOOP, base, NO_JUMP); in forbody()
1303 luaK_codeABC(fs, OP_TFORCALL, base, 0, nvars); in forbody()
1304 luaK_fixline(fs, line); in forbody()
1305 endfor = luaK_codeAsBx(fs, OP_TFORLOOP, base + 2, NO_JUMP); in forbody()
1307 luaK_patchlist(fs, endfor, prep + 1); in forbody()
1308 luaK_fixline(fs, line); in forbody()
1314 FuncState *fs = ls->fs; in fornum() local
1315 int base = fs->freereg; in fornum()
1327 luaK_codek(fs, fs->freereg, luaK_numberK(fs, 1)); in fornum()
1328 luaK_reserveregs(fs, 1); in fornum()
1336 FuncState *fs = ls->fs; in forlist() local
1340 int base = fs->freereg; in forlist()
1354 luaK_checkstack(fs, 3); /* extra space to call generator */ in forlist()
1361 FuncState *fs = ls->fs; in forstat() local
1364 enterblock(fs, &bl, 1); /* scope for loop and control variables */ in forstat()
1373 leaveblock(fs); /* loop scope (`break' jumps to this point) */ in forstat()
1381 FuncState *fs = ls->fs; in test_then_block() local
1388 luaK_goiffalse(ls->fs, &v); /* will jump to label if condition is true */ in test_then_block()
1389 enterblock(fs, &bl, 0); /* must enter block before 'goto' */ in test_then_block()
1393 leaveblock(fs); in test_then_block()
1397 jf = luaK_jump(fs); in test_then_block()
1400 luaK_goiftrue(ls->fs, &v); /* skip over block if condition is false */ in test_then_block()
1401 enterblock(fs, &bl, 0); in test_then_block()
1405 leaveblock(fs); in test_then_block()
1408 luaK_concat(fs, escapelist, luaK_jump(fs)); /* must jump over it */ in test_then_block()
1409 luaK_patchtohere(fs, jf); in test_then_block()
1415 FuncState *fs = ls->fs; in ifstat() local
1423 luaK_patchtohere(fs, escapelist); /* patch escape list to 'if' end */ in ifstat()
1429 FuncState *fs = ls->fs; in localfunc() local
1434 getlocvar(fs, b.u.info)->startpc = fs->pc; in localfunc()
1479 luaK_storevar(ls->fs, &v, &b); in funcstat()
1480 luaK_fixline(ls->fs, line); /* definition `happens' in the first line */ in funcstat()
1486 FuncState *fs = ls->fs; in exprstat() local
1495 SETARG_C(getcode(fs, &v.v), 1); /* call statement uses no results */ in exprstat()
1502 FuncState *fs = ls->fs; in retstat() local
1510 luaK_setmultret(fs, &e); in retstat()
1512 SET_OPCODE(getcode(fs,&e), OP_TAILCALL); in retstat()
1513 lua_assert(GETARG_A(getcode(fs,&e)) == fs->nactvar); in retstat()
1515 first = fs->nactvar; in retstat()
1520 first = luaK_exp2anyreg(fs, &e); in retstat()
1522 luaK_exp2nextreg(fs, &e); /* values must go to the `stack' */ in retstat()
1523 first = fs->nactvar; /* return all `active' values */ in retstat()
1524 lua_assert(nret == fs->freereg - first); in retstat()
1528 luaK_ret(fs, first, nret); in retstat()
1587 gotostat(ls, luaK_jump(ls->fs)); in statement()
1595 lua_assert(ls->fs->f->maxstacksize >= ls->fs->freereg && in statement()
1596 ls->fs->freereg >= ls->fs->nactvar); in statement()
1597 ls->fs->freereg = ls->fs->nactvar; /* free registers */ in statement()
1608 static void mainfunc (LexState *ls, FuncState *fs) { in mainfunc() argument
1611 open_func(ls, fs, &bl); in mainfunc()
1612 fs->f->is_vararg = 1; /* main function is always vararg */ in mainfunc()
1614 newupvalue(fs, ls->envn, &v); /* ...set environment upvalue */ in mainfunc()
1637 lua_assert(!funcstate.prev && funcstate.nups == 1 && !lexstate.fs); in luaY_parser()