1*dfc11533SChris Williamson /*
2*dfc11533SChris Williamson ** $Id: lfunc.c,v 2.30.1.1 2013/04/12 18:48:47 roberto Exp $
3*dfc11533SChris Williamson ** Auxiliary functions to manipulate prototypes and closures
4*dfc11533SChris Williamson ** See Copyright Notice in lua.h
5*dfc11533SChris Williamson */
6*dfc11533SChris Williamson
7*dfc11533SChris Williamson
8*dfc11533SChris Williamson #include <sys/zfs_context.h>
9*dfc11533SChris Williamson
10*dfc11533SChris Williamson #define lfunc_c
11*dfc11533SChris Williamson #define LUA_CORE
12*dfc11533SChris Williamson
13*dfc11533SChris Williamson #include "lua.h"
14*dfc11533SChris Williamson
15*dfc11533SChris Williamson #include "lfunc.h"
16*dfc11533SChris Williamson #include "lgc.h"
17*dfc11533SChris Williamson #include "lmem.h"
18*dfc11533SChris Williamson #include "lobject.h"
19*dfc11533SChris Williamson #include "lstate.h"
20*dfc11533SChris Williamson
21*dfc11533SChris Williamson
22*dfc11533SChris Williamson
luaF_newCclosure(lua_State * L,int n)23*dfc11533SChris Williamson Closure *luaF_newCclosure (lua_State *L, int n) {
24*dfc11533SChris Williamson Closure *c = &luaC_newobj(L, LUA_TCCL, sizeCclosure(n), NULL, 0)->cl;
25*dfc11533SChris Williamson c->c.nupvalues = cast_byte(n);
26*dfc11533SChris Williamson return c;
27*dfc11533SChris Williamson }
28*dfc11533SChris Williamson
29*dfc11533SChris Williamson
luaF_newLclosure(lua_State * L,int n)30*dfc11533SChris Williamson Closure *luaF_newLclosure (lua_State *L, int n) {
31*dfc11533SChris Williamson Closure *c = &luaC_newobj(L, LUA_TLCL, sizeLclosure(n), NULL, 0)->cl;
32*dfc11533SChris Williamson c->l.p = NULL;
33*dfc11533SChris Williamson c->l.nupvalues = cast_byte(n);
34*dfc11533SChris Williamson while (n--) c->l.upvals[n] = NULL;
35*dfc11533SChris Williamson return c;
36*dfc11533SChris Williamson }
37*dfc11533SChris Williamson
38*dfc11533SChris Williamson
luaF_newupval(lua_State * L)39*dfc11533SChris Williamson UpVal *luaF_newupval (lua_State *L) {
40*dfc11533SChris Williamson UpVal *uv = &luaC_newobj(L, LUA_TUPVAL, sizeof(UpVal), NULL, 0)->uv;
41*dfc11533SChris Williamson uv->v = &uv->u.value;
42*dfc11533SChris Williamson setnilvalue(uv->v);
43*dfc11533SChris Williamson return uv;
44*dfc11533SChris Williamson }
45*dfc11533SChris Williamson
46*dfc11533SChris Williamson
luaF_findupval(lua_State * L,StkId level)47*dfc11533SChris Williamson UpVal *luaF_findupval (lua_State *L, StkId level) {
48*dfc11533SChris Williamson global_State *g = G(L);
49*dfc11533SChris Williamson GCObject **pp = &L->openupval;
50*dfc11533SChris Williamson UpVal *p;
51*dfc11533SChris Williamson UpVal *uv;
52*dfc11533SChris Williamson while (*pp != NULL && (p = gco2uv(*pp))->v >= level) {
53*dfc11533SChris Williamson GCObject *o = obj2gco(p);
54*dfc11533SChris Williamson lua_assert(p->v != &p->u.value);
55*dfc11533SChris Williamson lua_assert(!isold(o) || isold(obj2gco(L)));
56*dfc11533SChris Williamson if (p->v == level) { /* found a corresponding upvalue? */
57*dfc11533SChris Williamson if (isdead(g, o)) /* is it dead? */
58*dfc11533SChris Williamson changewhite(o); /* resurrect it */
59*dfc11533SChris Williamson return p;
60*dfc11533SChris Williamson }
61*dfc11533SChris Williamson pp = &p->next;
62*dfc11533SChris Williamson }
63*dfc11533SChris Williamson /* not found: create a new one */
64*dfc11533SChris Williamson uv = &luaC_newobj(L, LUA_TUPVAL, sizeof(UpVal), pp, 0)->uv;
65*dfc11533SChris Williamson uv->v = level; /* current value lives in the stack */
66*dfc11533SChris Williamson uv->u.l.prev = &g->uvhead; /* double link it in `uvhead' list */
67*dfc11533SChris Williamson uv->u.l.next = g->uvhead.u.l.next;
68*dfc11533SChris Williamson uv->u.l.next->u.l.prev = uv;
69*dfc11533SChris Williamson g->uvhead.u.l.next = uv;
70*dfc11533SChris Williamson lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
71*dfc11533SChris Williamson return uv;
72*dfc11533SChris Williamson }
73*dfc11533SChris Williamson
74*dfc11533SChris Williamson
unlinkupval(UpVal * uv)75*dfc11533SChris Williamson static void unlinkupval (UpVal *uv) {
76*dfc11533SChris Williamson lua_assert(uv->u.l.next->u.l.prev == uv && uv->u.l.prev->u.l.next == uv);
77*dfc11533SChris Williamson uv->u.l.next->u.l.prev = uv->u.l.prev; /* remove from `uvhead' list */
78*dfc11533SChris Williamson uv->u.l.prev->u.l.next = uv->u.l.next;
79*dfc11533SChris Williamson }
80*dfc11533SChris Williamson
81*dfc11533SChris Williamson
luaF_freeupval(lua_State * L,UpVal * uv)82*dfc11533SChris Williamson void luaF_freeupval (lua_State *L, UpVal *uv) {
83*dfc11533SChris Williamson if (uv->v != &uv->u.value) /* is it open? */
84*dfc11533SChris Williamson unlinkupval(uv); /* remove from open list */
85*dfc11533SChris Williamson luaM_free(L, uv); /* free upvalue */
86*dfc11533SChris Williamson }
87*dfc11533SChris Williamson
88*dfc11533SChris Williamson
luaF_close(lua_State * L,StkId level)89*dfc11533SChris Williamson void luaF_close (lua_State *L, StkId level) {
90*dfc11533SChris Williamson UpVal *uv;
91*dfc11533SChris Williamson global_State *g = G(L);
92*dfc11533SChris Williamson while (L->openupval != NULL && (uv = gco2uv(L->openupval))->v >= level) {
93*dfc11533SChris Williamson GCObject *o = obj2gco(uv);
94*dfc11533SChris Williamson lua_assert(!isblack(o) && uv->v != &uv->u.value);
95*dfc11533SChris Williamson L->openupval = uv->next; /* remove from `open' list */
96*dfc11533SChris Williamson if (isdead(g, o))
97*dfc11533SChris Williamson luaF_freeupval(L, uv); /* free upvalue */
98*dfc11533SChris Williamson else {
99*dfc11533SChris Williamson unlinkupval(uv); /* remove upvalue from 'uvhead' list */
100*dfc11533SChris Williamson setobj(L, &uv->u.value, uv->v); /* move value to upvalue slot */
101*dfc11533SChris Williamson uv->v = &uv->u.value; /* now current value lives here */
102*dfc11533SChris Williamson gch(o)->next = g->allgc; /* link upvalue into 'allgc' list */
103*dfc11533SChris Williamson g->allgc = o;
104*dfc11533SChris Williamson luaC_checkupvalcolor(g, uv);
105*dfc11533SChris Williamson }
106*dfc11533SChris Williamson }
107*dfc11533SChris Williamson }
108*dfc11533SChris Williamson
109*dfc11533SChris Williamson
luaF_newproto(lua_State * L)110*dfc11533SChris Williamson Proto *luaF_newproto (lua_State *L) {
111*dfc11533SChris Williamson Proto *f = &luaC_newobj(L, LUA_TPROTO, sizeof(Proto), NULL, 0)->p;
112*dfc11533SChris Williamson f->k = NULL;
113*dfc11533SChris Williamson f->sizek = 0;
114*dfc11533SChris Williamson f->p = NULL;
115*dfc11533SChris Williamson f->sizep = 0;
116*dfc11533SChris Williamson f->code = NULL;
117*dfc11533SChris Williamson f->cache = NULL;
118*dfc11533SChris Williamson f->sizecode = 0;
119*dfc11533SChris Williamson f->lineinfo = NULL;
120*dfc11533SChris Williamson f->sizelineinfo = 0;
121*dfc11533SChris Williamson f->upvalues = NULL;
122*dfc11533SChris Williamson f->sizeupvalues = 0;
123*dfc11533SChris Williamson f->numparams = 0;
124*dfc11533SChris Williamson f->is_vararg = 0;
125*dfc11533SChris Williamson f->maxstacksize = 0;
126*dfc11533SChris Williamson f->locvars = NULL;
127*dfc11533SChris Williamson f->sizelocvars = 0;
128*dfc11533SChris Williamson f->linedefined = 0;
129*dfc11533SChris Williamson f->lastlinedefined = 0;
130*dfc11533SChris Williamson f->source = NULL;
131*dfc11533SChris Williamson return f;
132*dfc11533SChris Williamson }
133*dfc11533SChris Williamson
134*dfc11533SChris Williamson
luaF_freeproto(lua_State * L,Proto * f)135*dfc11533SChris Williamson void luaF_freeproto (lua_State *L, Proto *f) {
136*dfc11533SChris Williamson luaM_freearray(L, f->code, f->sizecode);
137*dfc11533SChris Williamson luaM_freearray(L, f->p, f->sizep);
138*dfc11533SChris Williamson luaM_freearray(L, f->k, f->sizek);
139*dfc11533SChris Williamson luaM_freearray(L, f->lineinfo, f->sizelineinfo);
140*dfc11533SChris Williamson luaM_freearray(L, f->locvars, f->sizelocvars);
141*dfc11533SChris Williamson luaM_freearray(L, f->upvalues, f->sizeupvalues);
142*dfc11533SChris Williamson luaM_free(L, f);
143*dfc11533SChris Williamson }
144*dfc11533SChris Williamson
145*dfc11533SChris Williamson
146*dfc11533SChris Williamson /*
147*dfc11533SChris Williamson ** Look for n-th local variable at line `line' in function `func'.
148*dfc11533SChris Williamson ** Returns NULL if not found.
149*dfc11533SChris Williamson */
luaF_getlocalname(const Proto * f,int local_number,int pc)150*dfc11533SChris Williamson const char *luaF_getlocalname (const Proto *f, int local_number, int pc) {
151*dfc11533SChris Williamson int i;
152*dfc11533SChris Williamson for (i = 0; i<f->sizelocvars && f->locvars[i].startpc <= pc; i++) {
153*dfc11533SChris Williamson if (pc < f->locvars[i].endpc) { /* is variable active? */
154*dfc11533SChris Williamson local_number--;
155*dfc11533SChris Williamson if (local_number == 0)
156*dfc11533SChris Williamson return getstr(f->locvars[i].varname);
157*dfc11533SChris Williamson }
158*dfc11533SChris Williamson }
159*dfc11533SChris Williamson return NULL; /* not found */
160*dfc11533SChris Williamson }
161*dfc11533SChris Williamson
162