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