]>
Commit | Line | Data |
---|---|---|
1 | /* | |
2 | ** $Id: lapi.c,v 2.55.1.5 2008/07/04 18:41:18 roberto Exp $ | |
3 | ** Lua API | |
4 | ** See Copyright Notice in lua.h | |
5 | */ | |
6 | ||
7 | ||
8 | #include <assert.h> | |
9 | #include <math.h> | |
10 | #include <stdarg.h> | |
11 | #include <string.h> | |
12 | ||
13 | #define lapi_c | |
14 | #define LUA_CORE | |
15 | ||
16 | #include "lua.h" | |
17 | ||
18 | #include "lapi.h" | |
19 | #include "ldebug.h" | |
20 | #include "ldo.h" | |
21 | #include "lfunc.h" | |
22 | #include "lgc.h" | |
23 | #include "lmem.h" | |
24 | #include "lobject.h" | |
25 | #include "lstate.h" | |
26 | #include "lstring.h" | |
27 | #include "ltable.h" | |
28 | #include "ltm.h" | |
29 | #include "lundump.h" | |
30 | #include "lvm.h" | |
31 | ||
32 | ||
33 | ||
34 | const char lua_ident[] = | |
35 | "$Lua: " LUA_RELEASE " " LUA_COPYRIGHT " $\n" | |
36 | "$Authors: " LUA_AUTHORS " $\n" | |
37 | "$URL: www.lua.org $\n"; | |
38 | ||
39 | ||
40 | ||
41 | #define api_checknelems(L, n) api_check(L, (n) <= (L->top - L->base)) | |
42 | ||
43 | #define api_checkvalidindex(L, i) api_check(L, (i) != luaO_nilobject) | |
44 | ||
45 | #define api_incr_top(L) {api_check(L, L->top < L->ci->top); L->top++;} | |
46 | ||
47 | ||
48 | ||
49 | static TValue *index2adr (lua_State *L, int idx) { | |
50 | if (idx > 0) { | |
51 | TValue *o = L->base + (idx - 1); | |
52 | api_check(L, idx <= L->ci->top - L->base); | |
53 | if (o >= L->top) return cast(TValue *, luaO_nilobject); | |
54 | else return o; | |
55 | } | |
56 | else if (idx > LUA_REGISTRYINDEX) { | |
57 | api_check(L, idx != 0 && -idx <= L->top - L->base); | |
58 | return L->top + idx; | |
59 | } | |
60 | else switch (idx) { /* pseudo-indices */ | |
61 | case LUA_REGISTRYINDEX: return registry(L); | |
62 | case LUA_ENVIRONINDEX: { | |
63 | Closure *func = curr_func(L); | |
64 | sethvalue(L, &L->env, func->c.env); | |
65 | return &L->env; | |
66 | } | |
67 | case LUA_GLOBALSINDEX: return gt(L); | |
68 | default: { | |
69 | Closure *func = curr_func(L); | |
70 | idx = LUA_GLOBALSINDEX - idx; | |
71 | return (idx <= func->c.nupvalues) | |
72 | ? &func->c.upvalue[idx-1] | |
73 | : cast(TValue *, luaO_nilobject); | |
74 | } | |
75 | } | |
76 | } | |
77 | ||
78 | ||
79 | static Table *getcurrenv (lua_State *L) { | |
80 | if (L->ci == L->base_ci) /* no enclosing function? */ | |
81 | return hvalue(gt(L)); /* use global table as environment */ | |
82 | else { | |
83 | Closure *func = curr_func(L); | |
84 | return func->c.env; | |
85 | } | |
86 | } | |
87 | ||
88 | ||
89 | void luaA_pushobject (lua_State *L, const TValue *o) { | |
90 | setobj2s(L, L->top, o); | |
91 | api_incr_top(L); | |
92 | } | |
93 | ||
94 | ||
95 | LUA_API int lua_checkstack (lua_State *L, int size) { | |
96 | int res = 1; | |
97 | lua_lock(L); | |
98 | if (size > LUAI_MAXCSTACK || (L->top - L->base + size) > LUAI_MAXCSTACK) | |
99 | res = 0; /* stack overflow */ | |
100 | else if (size > 0) { | |
101 | luaD_checkstack(L, size); | |
102 | if (L->ci->top < L->top + size) | |
103 | L->ci->top = L->top + size; | |
104 | } | |
105 | lua_unlock(L); | |
106 | return res; | |
107 | } | |
108 | ||
109 | ||
110 | LUA_API void lua_xmove (lua_State *from, lua_State *to, int n) { | |
111 | int i; | |
112 | if (from == to) return; | |
113 | lua_lock(to); | |
114 | api_checknelems(from, n); | |
115 | api_check(from, G(from) == G(to)); | |
116 | api_check(from, to->ci->top - to->top >= n); | |
117 | from->top -= n; | |
118 | for (i = 0; i < n; i++) { | |
119 | setobj2s(to, to->top++, from->top + i); | |
120 | } | |
121 | lua_unlock(to); | |
122 | } | |
123 | ||
124 | ||
125 | LUA_API void lua_setlevel (lua_State *from, lua_State *to) { | |
126 | to->nCcalls = from->nCcalls; | |
127 | } | |
128 | ||
129 | ||
130 | LUA_API lua_CFunction lua_atpanic (lua_State *L, lua_CFunction panicf) { | |
131 | lua_CFunction old; | |
132 | lua_lock(L); | |
133 | old = G(L)->panic; | |
134 | G(L)->panic = panicf; | |
135 | lua_unlock(L); | |
136 | return old; | |
137 | } | |
138 | ||
139 | ||
140 | LUA_API lua_State *lua_newthread (lua_State *L) { | |
141 | lua_State *L1; | |
142 | lua_lock(L); | |
143 | luaC_checkGC(L); | |
144 | L1 = luaE_newthread(L); | |
145 | setthvalue(L, L->top, L1); | |
146 | api_incr_top(L); | |
147 | lua_unlock(L); | |
148 | luai_userstatethread(L, L1); | |
149 | return L1; | |
150 | } | |
151 | ||
152 | ||
153 | ||
154 | /* | |
155 | ** basic stack manipulation | |
156 | */ | |
157 | ||
158 | ||
159 | LUA_API int lua_gettop (lua_State *L) { | |
160 | return cast_int(L->top - L->base); | |
161 | } | |
162 | ||
163 | ||
164 | LUA_API void lua_settop (lua_State *L, int idx) { | |
165 | lua_lock(L); | |
166 | if (idx >= 0) { | |
167 | api_check(L, idx <= L->stack_last - L->base); | |
168 | while (L->top < L->base + idx) | |
169 | setnilvalue(L->top++); | |
170 | L->top = L->base + idx; | |
171 | } | |
172 | else { | |
173 | api_check(L, -(idx+1) <= (L->top - L->base)); | |
174 | L->top += idx+1; /* `subtract' index (index is negative) */ | |
175 | } | |
176 | lua_unlock(L); | |
177 | } | |
178 | ||
179 | ||
180 | LUA_API void lua_remove (lua_State *L, int idx) { | |
181 | StkId p; | |
182 | lua_lock(L); | |
183 | p = index2adr(L, idx); | |
184 | api_checkvalidindex(L, p); | |
185 | while (++p < L->top) setobjs2s(L, p-1, p); | |
186 | L->top--; | |
187 | lua_unlock(L); | |
188 | } | |
189 | ||
190 | ||
191 | LUA_API void lua_insert (lua_State *L, int idx) { | |
192 | StkId p; | |
193 | StkId q; | |
194 | lua_lock(L); | |
195 | p = index2adr(L, idx); | |
196 | api_checkvalidindex(L, p); | |
197 | for (q = L->top; q>p; q--) setobjs2s(L, q, q-1); | |
198 | setobjs2s(L, p, L->top); | |
199 | lua_unlock(L); | |
200 | } | |
201 | ||
202 | ||
203 | LUA_API void lua_replace (lua_State *L, int idx) { | |
204 | StkId o; | |
205 | lua_lock(L); | |
206 | /* explicit test for incompatible code */ | |
207 | if (idx == LUA_ENVIRONINDEX && L->ci == L->base_ci) | |
208 | luaG_runerror(L, "no calling environment"); | |
209 | api_checknelems(L, 1); | |
210 | o = index2adr(L, idx); | |
211 | api_checkvalidindex(L, o); | |
212 | if (idx == LUA_ENVIRONINDEX) { | |
213 | Closure *func = curr_func(L); | |
214 | api_check(L, ttistable(L->top - 1)); | |
215 | func->c.env = hvalue(L->top - 1); | |
216 | luaC_barrier(L, func, L->top - 1); | |
217 | } | |
218 | else { | |
219 | setobj(L, o, L->top - 1); | |
220 | if (idx < LUA_GLOBALSINDEX) /* function upvalue? */ | |
221 | luaC_barrier(L, curr_func(L), L->top - 1); | |
222 | } | |
223 | L->top--; | |
224 | lua_unlock(L); | |
225 | } | |
226 | ||
227 | ||
228 | LUA_API void lua_pushvalue (lua_State *L, int idx) { | |
229 | lua_lock(L); | |
230 | setobj2s(L, L->top, index2adr(L, idx)); | |
231 | api_incr_top(L); | |
232 | lua_unlock(L); | |
233 | } | |
234 | ||
235 | ||
236 | ||
237 | /* | |
238 | ** access functions (stack -> C) | |
239 | */ | |
240 | ||
241 | ||
242 | LUA_API int lua_type (lua_State *L, int idx) { | |
243 | StkId o = index2adr(L, idx); | |
244 | return (o == luaO_nilobject) ? LUA_TNONE : ttype(o); | |
245 | } | |
246 | ||
247 | ||
248 | LUA_API const char *lua_typename (lua_State *L, int t) { | |
249 | UNUSED(L); | |
250 | return (t == LUA_TNONE) ? "no value" : luaT_typenames[t]; | |
251 | } | |
252 | ||
253 | ||
254 | LUA_API int lua_iscfunction (lua_State *L, int idx) { | |
255 | StkId o = index2adr(L, idx); | |
256 | return iscfunction(o); | |
257 | } | |
258 | ||
259 | ||
260 | LUA_API int lua_isnumber (lua_State *L, int idx) { | |
261 | TValue n; | |
262 | const TValue *o = index2adr(L, idx); | |
263 | return tonumber(o, &n); | |
264 | } | |
265 | ||
266 | ||
267 | LUA_API int lua_isstring (lua_State *L, int idx) { | |
268 | int t = lua_type(L, idx); | |
269 | return (t == LUA_TSTRING || t == LUA_TNUMBER); | |
270 | } | |
271 | ||
272 | ||
273 | LUA_API int lua_isuserdata (lua_State *L, int idx) { | |
274 | const TValue *o = index2adr(L, idx); | |
275 | return (ttisuserdata(o) || ttislightuserdata(o)); | |
276 | } | |
277 | ||
278 | ||
279 | LUA_API int lua_rawequal (lua_State *L, int index1, int index2) { | |
280 | StkId o1 = index2adr(L, index1); | |
281 | StkId o2 = index2adr(L, index2); | |
282 | return (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0 | |
283 | : luaO_rawequalObj(o1, o2); | |
284 | } | |
285 | ||
286 | ||
287 | LUA_API int lua_equal (lua_State *L, int index1, int index2) { | |
288 | StkId o1, o2; | |
289 | int i; | |
290 | lua_lock(L); /* may call tag method */ | |
291 | o1 = index2adr(L, index1); | |
292 | o2 = index2adr(L, index2); | |
293 | i = (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0 : equalobj(L, o1, o2); | |
294 | lua_unlock(L); | |
295 | return i; | |
296 | } | |
297 | ||
298 | ||
299 | LUA_API int lua_lessthan (lua_State *L, int index1, int index2) { | |
300 | StkId o1, o2; | |
301 | int i; | |
302 | lua_lock(L); /* may call tag method */ | |
303 | o1 = index2adr(L, index1); | |
304 | o2 = index2adr(L, index2); | |
305 | i = (o1 == luaO_nilobject || o2 == luaO_nilobject) ? 0 | |
306 | : luaV_lessthan(L, o1, o2); | |
307 | lua_unlock(L); | |
308 | return i; | |
309 | } | |
310 | ||
311 | ||
312 | ||
313 | LUA_API lua_Number lua_tonumber (lua_State *L, int idx) { | |
314 | TValue n; | |
315 | const TValue *o = index2adr(L, idx); | |
316 | if (tonumber(o, &n)) | |
317 | return nvalue(o); | |
318 | else | |
319 | return 0; | |
320 | } | |
321 | ||
322 | ||
323 | LUA_API lua_Integer lua_tointeger (lua_State *L, int idx) { | |
324 | TValue n; | |
325 | const TValue *o = index2adr(L, idx); | |
326 | if (tonumber(o, &n)) { | |
327 | lua_Integer res; | |
328 | lua_Number num = nvalue(o); | |
329 | lua_number2integer(res, num); | |
330 | return res; | |
331 | } | |
332 | else | |
333 | return 0; | |
334 | } | |
335 | ||
336 | ||
337 | LUA_API int lua_toboolean (lua_State *L, int idx) { | |
338 | const TValue *o = index2adr(L, idx); | |
339 | return !l_isfalse(o); | |
340 | } | |
341 | ||
342 | ||
343 | LUA_API const char *lua_tolstring (lua_State *L, int idx, size_t *len) { | |
344 | StkId o = index2adr(L, idx); | |
345 | if (!ttisstring(o)) { | |
346 | lua_lock(L); /* `luaV_tostring' may create a new string */ | |
347 | if (!luaV_tostring(L, o)) { /* conversion failed? */ | |
348 | if (len != NULL) *len = 0; | |
349 | lua_unlock(L); | |
350 | return NULL; | |
351 | } | |
352 | luaC_checkGC(L); | |
353 | o = index2adr(L, idx); /* previous call may reallocate the stack */ | |
354 | lua_unlock(L); | |
355 | } | |
356 | if (len != NULL) *len = tsvalue(o)->len; | |
357 | return svalue(o); | |
358 | } | |
359 | ||
360 | ||
361 | LUA_API size_t lua_objlen (lua_State *L, int idx) { | |
362 | StkId o = index2adr(L, idx); | |
363 | switch (ttype(o)) { | |
364 | case LUA_TSTRING: return tsvalue(o)->len; | |
365 | case LUA_TUSERDATA: return uvalue(o)->len; | |
366 | case LUA_TTABLE: return luaH_getn(hvalue(o)); | |
367 | case LUA_TNUMBER: { | |
368 | size_t l; | |
369 | lua_lock(L); /* `luaV_tostring' may create a new string */ | |
370 | l = (luaV_tostring(L, o) ? tsvalue(o)->len : 0); | |
371 | lua_unlock(L); | |
372 | return l; | |
373 | } | |
374 | default: return 0; | |
375 | } | |
376 | } | |
377 | ||
378 | ||
379 | LUA_API lua_CFunction lua_tocfunction (lua_State *L, int idx) { | |
380 | StkId o = index2adr(L, idx); | |
381 | return (!iscfunction(o)) ? NULL : clvalue(o)->c.f; | |
382 | } | |
383 | ||
384 | ||
385 | LUA_API void *lua_touserdata (lua_State *L, int idx) { | |
386 | StkId o = index2adr(L, idx); | |
387 | switch (ttype(o)) { | |
388 | case LUA_TUSERDATA: return (rawuvalue(o) + 1); | |
389 | case LUA_TLIGHTUSERDATA: return pvalue(o); | |
390 | default: return NULL; | |
391 | } | |
392 | } | |
393 | ||
394 | ||
395 | LUA_API lua_State *lua_tothread (lua_State *L, int idx) { | |
396 | StkId o = index2adr(L, idx); | |
397 | return (!ttisthread(o)) ? NULL : thvalue(o); | |
398 | } | |
399 | ||
400 | ||
401 | LUA_API const void *lua_topointer (lua_State *L, int idx) { | |
402 | StkId o = index2adr(L, idx); | |
403 | switch (ttype(o)) { | |
404 | case LUA_TTABLE: return hvalue(o); | |
405 | case LUA_TFUNCTION: return clvalue(o); | |
406 | case LUA_TTHREAD: return thvalue(o); | |
407 | case LUA_TUSERDATA: | |
408 | case LUA_TLIGHTUSERDATA: | |
409 | return lua_touserdata(L, idx); | |
410 | default: return NULL; | |
411 | } | |
412 | } | |
413 | ||
414 | ||
415 | ||
416 | /* | |
417 | ** push functions (C -> stack) | |
418 | */ | |
419 | ||
420 | ||
421 | LUA_API void lua_pushnil (lua_State *L) { | |
422 | lua_lock(L); | |
423 | setnilvalue(L->top); | |
424 | api_incr_top(L); | |
425 | lua_unlock(L); | |
426 | } | |
427 | ||
428 | ||
429 | LUA_API void lua_pushnumber (lua_State *L, lua_Number n) { | |
430 | lua_lock(L); | |
431 | setnvalue(L->top, n); | |
432 | api_incr_top(L); | |
433 | lua_unlock(L); | |
434 | } | |
435 | ||
436 | ||
437 | LUA_API void lua_pushinteger (lua_State *L, lua_Integer n) { | |
438 | lua_lock(L); | |
439 | setnvalue(L->top, cast_num(n)); | |
440 | api_incr_top(L); | |
441 | lua_unlock(L); | |
442 | } | |
443 | ||
444 | ||
445 | LUA_API void lua_pushlstring (lua_State *L, const char *s, size_t len) { | |
446 | lua_lock(L); | |
447 | luaC_checkGC(L); | |
448 | setsvalue2s(L, L->top, luaS_newlstr(L, s, len)); | |
449 | api_incr_top(L); | |
450 | lua_unlock(L); | |
451 | } | |
452 | ||
453 | ||
454 | LUA_API void lua_pushstring (lua_State *L, const char *s) { | |
455 | if (s == NULL) | |
456 | lua_pushnil(L); | |
457 | else | |
458 | lua_pushlstring(L, s, strlen(s)); | |
459 | } | |
460 | ||
461 | ||
462 | LUA_API const char *lua_pushvfstring (lua_State *L, const char *fmt, | |
463 | va_list argp) { | |
464 | const char *ret; | |
465 | lua_lock(L); | |
466 | luaC_checkGC(L); | |
467 | ret = luaO_pushvfstring(L, fmt, argp); | |
468 | lua_unlock(L); | |
469 | return ret; | |
470 | } | |
471 | ||
472 | ||
473 | LUA_API const char *lua_pushfstring (lua_State *L, const char *fmt, ...) { | |
474 | const char *ret; | |
475 | va_list argp; | |
476 | lua_lock(L); | |
477 | luaC_checkGC(L); | |
478 | va_start(argp, fmt); | |
479 | ret = luaO_pushvfstring(L, fmt, argp); | |
480 | va_end(argp); | |
481 | lua_unlock(L); | |
482 | return ret; | |
483 | } | |
484 | ||
485 | ||
486 | LUA_API void lua_pushcclosure (lua_State *L, lua_CFunction fn, int n) { | |
487 | Closure *cl; | |
488 | lua_lock(L); | |
489 | luaC_checkGC(L); | |
490 | api_checknelems(L, n); | |
491 | cl = luaF_newCclosure(L, n, getcurrenv(L)); | |
492 | cl->c.f = fn; | |
493 | L->top -= n; | |
494 | while (n--) | |
495 | setobj2n(L, &cl->c.upvalue[n], L->top+n); | |
496 | setclvalue(L, L->top, cl); | |
497 | lua_assert(iswhite(obj2gco(cl))); | |
498 | api_incr_top(L); | |
499 | lua_unlock(L); | |
500 | } | |
501 | ||
502 | ||
503 | LUA_API void lua_pushboolean (lua_State *L, int b) { | |
504 | lua_lock(L); | |
505 | setbvalue(L->top, (b != 0)); /* ensure that true is 1 */ | |
506 | api_incr_top(L); | |
507 | lua_unlock(L); | |
508 | } | |
509 | ||
510 | ||
511 | LUA_API void lua_pushlightuserdata (lua_State *L, void *p) { | |
512 | lua_lock(L); | |
513 | setpvalue(L->top, p); | |
514 | api_incr_top(L); | |
515 | lua_unlock(L); | |
516 | } | |
517 | ||
518 | ||
519 | LUA_API int lua_pushthread (lua_State *L) { | |
520 | lua_lock(L); | |
521 | setthvalue(L, L->top, L); | |
522 | api_incr_top(L); | |
523 | lua_unlock(L); | |
524 | return (G(L)->mainthread == L); | |
525 | } | |
526 | ||
527 | ||
528 | ||
529 | /* | |
530 | ** get functions (Lua -> stack) | |
531 | */ | |
532 | ||
533 | ||
534 | LUA_API void lua_gettable (lua_State *L, int idx) { | |
535 | StkId t; | |
536 | lua_lock(L); | |
537 | t = index2adr(L, idx); | |
538 | api_checkvalidindex(L, t); | |
539 | luaV_gettable(L, t, L->top - 1, L->top - 1); | |
540 | lua_unlock(L); | |
541 | } | |
542 | ||
543 | ||
544 | LUA_API void lua_getfield (lua_State *L, int idx, const char *k) { | |
545 | StkId t; | |
546 | TValue key; | |
547 | lua_lock(L); | |
548 | t = index2adr(L, idx); | |
549 | api_checkvalidindex(L, t); | |
550 | setsvalue(L, &key, luaS_new(L, k)); | |
551 | luaV_gettable(L, t, &key, L->top); | |
552 | api_incr_top(L); | |
553 | lua_unlock(L); | |
554 | } | |
555 | ||
556 | ||
557 | LUA_API void lua_rawget (lua_State *L, int idx) { | |
558 | StkId t; | |
559 | lua_lock(L); | |
560 | t = index2adr(L, idx); | |
561 | api_check(L, ttistable(t)); | |
562 | setobj2s(L, L->top - 1, luaH_get(hvalue(t), L->top - 1)); | |
563 | lua_unlock(L); | |
564 | } | |
565 | ||
566 | ||
567 | LUA_API void lua_rawgeti (lua_State *L, int idx, int n) { | |
568 | StkId o; | |
569 | lua_lock(L); | |
570 | o = index2adr(L, idx); | |
571 | api_check(L, ttistable(o)); | |
572 | setobj2s(L, L->top, luaH_getnum(hvalue(o), n)); | |
573 | api_incr_top(L); | |
574 | lua_unlock(L); | |
575 | } | |
576 | ||
577 | ||
578 | LUA_API void lua_createtable (lua_State *L, int narray, int nrec) { | |
579 | lua_lock(L); | |
580 | luaC_checkGC(L); | |
581 | sethvalue(L, L->top, luaH_new(L, narray, nrec)); | |
582 | api_incr_top(L); | |
583 | lua_unlock(L); | |
584 | } | |
585 | ||
586 | ||
587 | LUA_API int lua_getmetatable (lua_State *L, int objindex) { | |
588 | const TValue *obj; | |
589 | Table *mt = NULL; | |
590 | int res; | |
591 | lua_lock(L); | |
592 | obj = index2adr(L, objindex); | |
593 | switch (ttype(obj)) { | |
594 | case LUA_TTABLE: | |
595 | mt = hvalue(obj)->metatable; | |
596 | break; | |
597 | case LUA_TUSERDATA: | |
598 | mt = uvalue(obj)->metatable; | |
599 | break; | |
600 | default: | |
601 | mt = G(L)->mt[ttype(obj)]; | |
602 | break; | |
603 | } | |
604 | if (mt == NULL) | |
605 | res = 0; | |
606 | else { | |
607 | sethvalue(L, L->top, mt); | |
608 | api_incr_top(L); | |
609 | res = 1; | |
610 | } | |
611 | lua_unlock(L); | |
612 | return res; | |
613 | } | |
614 | ||
615 | ||
616 | LUA_API void lua_getfenv (lua_State *L, int idx) { | |
617 | StkId o; | |
618 | lua_lock(L); | |
619 | o = index2adr(L, idx); | |
620 | api_checkvalidindex(L, o); | |
621 | switch (ttype(o)) { | |
622 | case LUA_TFUNCTION: | |
623 | sethvalue(L, L->top, clvalue(o)->c.env); | |
624 | break; | |
625 | case LUA_TUSERDATA: | |
626 | sethvalue(L, L->top, uvalue(o)->env); | |
627 | break; | |
628 | case LUA_TTHREAD: | |
629 | setobj2s(L, L->top, gt(thvalue(o))); | |
630 | break; | |
631 | default: | |
632 | setnilvalue(L->top); | |
633 | break; | |
634 | } | |
635 | api_incr_top(L); | |
636 | lua_unlock(L); | |
637 | } | |
638 | ||
639 | ||
640 | /* | |
641 | ** set functions (stack -> Lua) | |
642 | */ | |
643 | ||
644 | ||
645 | LUA_API void lua_settable (lua_State *L, int idx) { | |
646 | StkId t; | |
647 | lua_lock(L); | |
648 | api_checknelems(L, 2); | |
649 | t = index2adr(L, idx); | |
650 | api_checkvalidindex(L, t); | |
651 | luaV_settable(L, t, L->top - 2, L->top - 1); | |
652 | L->top -= 2; /* pop index and value */ | |
653 | lua_unlock(L); | |
654 | } | |
655 | ||
656 | ||
657 | LUA_API void lua_setfield (lua_State *L, int idx, const char *k) { | |
658 | StkId t; | |
659 | TValue key; | |
660 | lua_lock(L); | |
661 | api_checknelems(L, 1); | |
662 | t = index2adr(L, idx); | |
663 | api_checkvalidindex(L, t); | |
664 | setsvalue(L, &key, luaS_new(L, k)); | |
665 | luaV_settable(L, t, &key, L->top - 1); | |
666 | L->top--; /* pop value */ | |
667 | lua_unlock(L); | |
668 | } | |
669 | ||
670 | ||
671 | LUA_API void lua_rawset (lua_State *L, int idx) { | |
672 | StkId t; | |
673 | lua_lock(L); | |
674 | api_checknelems(L, 2); | |
675 | t = index2adr(L, idx); | |
676 | api_check(L, ttistable(t)); | |
677 | setobj2t(L, luaH_set(L, hvalue(t), L->top-2), L->top-1); | |
678 | luaC_barriert(L, hvalue(t), L->top-1); | |
679 | L->top -= 2; | |
680 | lua_unlock(L); | |
681 | } | |
682 | ||
683 | ||
684 | LUA_API void lua_rawseti (lua_State *L, int idx, int n) { | |
685 | StkId o; | |
686 | lua_lock(L); | |
687 | api_checknelems(L, 1); | |
688 | o = index2adr(L, idx); | |
689 | api_check(L, ttistable(o)); | |
690 | setobj2t(L, luaH_setnum(L, hvalue(o), n), L->top-1); | |
691 | luaC_barriert(L, hvalue(o), L->top-1); | |
692 | L->top--; | |
693 | lua_unlock(L); | |
694 | } | |
695 | ||
696 | ||
697 | LUA_API int lua_setmetatable (lua_State *L, int objindex) { | |
698 | TValue *obj; | |
699 | Table *mt; | |
700 | lua_lock(L); | |
701 | api_checknelems(L, 1); | |
702 | obj = index2adr(L, objindex); | |
703 | api_checkvalidindex(L, obj); | |
704 | if (ttisnil(L->top - 1)) | |
705 | mt = NULL; | |
706 | else { | |
707 | api_check(L, ttistable(L->top - 1)); | |
708 | mt = hvalue(L->top - 1); | |
709 | } | |
710 | switch (ttype(obj)) { | |
711 | case LUA_TTABLE: { | |
712 | hvalue(obj)->metatable = mt; | |
713 | if (mt) | |
714 | luaC_objbarriert(L, hvalue(obj), mt); | |
715 | break; | |
716 | } | |
717 | case LUA_TUSERDATA: { | |
718 | uvalue(obj)->metatable = mt; | |
719 | if (mt) | |
720 | luaC_objbarrier(L, rawuvalue(obj), mt); | |
721 | break; | |
722 | } | |
723 | default: { | |
724 | G(L)->mt[ttype(obj)] = mt; | |
725 | break; | |
726 | } | |
727 | } | |
728 | L->top--; | |
729 | lua_unlock(L); | |
730 | return 1; | |
731 | } | |
732 | ||
733 | ||
734 | LUA_API int lua_setfenv (lua_State *L, int idx) { | |
735 | StkId o; | |
736 | int res = 1; | |
737 | lua_lock(L); | |
738 | api_checknelems(L, 1); | |
739 | o = index2adr(L, idx); | |
740 | api_checkvalidindex(L, o); | |
741 | api_check(L, ttistable(L->top - 1)); | |
742 | switch (ttype(o)) { | |
743 | case LUA_TFUNCTION: | |
744 | clvalue(o)->c.env = hvalue(L->top - 1); | |
745 | break; | |
746 | case LUA_TUSERDATA: | |
747 | uvalue(o)->env = hvalue(L->top - 1); | |
748 | break; | |
749 | case LUA_TTHREAD: | |
750 | sethvalue(L, gt(thvalue(o)), hvalue(L->top - 1)); | |
751 | break; | |
752 | default: | |
753 | res = 0; | |
754 | break; | |
755 | } | |
756 | if (res) luaC_objbarrier(L, gcvalue(o), hvalue(L->top - 1)); | |
757 | L->top--; | |
758 | lua_unlock(L); | |
759 | return res; | |
760 | } | |
761 | ||
762 | ||
763 | /* | |
764 | ** `load' and `call' functions (run Lua code) | |
765 | */ | |
766 | ||
767 | ||
768 | #define adjustresults(L,nres) \ | |
769 | { if (nres == LUA_MULTRET && L->top >= L->ci->top) L->ci->top = L->top; } | |
770 | ||
771 | ||
772 | #define checkresults(L,na,nr) \ | |
773 | api_check(L, (nr) == LUA_MULTRET || (L->ci->top - L->top >= (nr) - (na))) | |
774 | ||
775 | ||
776 | LUA_API void lua_call (lua_State *L, int nargs, int nresults) { | |
777 | StkId func; | |
778 | lua_lock(L); | |
779 | api_checknelems(L, nargs+1); | |
780 | checkresults(L, nargs, nresults); | |
781 | func = L->top - (nargs+1); | |
782 | luaD_call(L, func, nresults); | |
783 | adjustresults(L, nresults); | |
784 | lua_unlock(L); | |
785 | } | |
786 | ||
787 | ||
788 | ||
789 | /* | |
790 | ** Execute a protected call. | |
791 | */ | |
792 | struct CallS { /* data to `f_call' */ | |
793 | StkId func; | |
794 | int nresults; | |
795 | }; | |
796 | ||
797 | ||
798 | static void f_call (lua_State *L, void *ud) { | |
799 | struct CallS *c = cast(struct CallS *, ud); | |
800 | luaD_call(L, c->func, c->nresults); | |
801 | } | |
802 | ||
803 | ||
804 | ||
805 | LUA_API int lua_pcall (lua_State *L, int nargs, int nresults, int errfunc) { | |
806 | struct CallS c; | |
807 | int status; | |
808 | ptrdiff_t func; | |
809 | lua_lock(L); | |
810 | api_checknelems(L, nargs+1); | |
811 | checkresults(L, nargs, nresults); | |
812 | if (errfunc == 0) | |
813 | func = 0; | |
814 | else { | |
815 | StkId o = index2adr(L, errfunc); | |
816 | api_checkvalidindex(L, o); | |
817 | func = savestack(L, o); | |
818 | } | |
819 | c.func = L->top - (nargs+1); /* function to be called */ | |
820 | c.nresults = nresults; | |
821 | status = luaD_pcall(L, f_call, &c, savestack(L, c.func), func); | |
822 | adjustresults(L, nresults); | |
823 | lua_unlock(L); | |
824 | return status; | |
825 | } | |
826 | ||
827 | ||
828 | /* | |
829 | ** Execute a protected C call. | |
830 | */ | |
831 | struct CCallS { /* data to `f_Ccall' */ | |
832 | lua_CFunction func; | |
833 | void *ud; | |
834 | }; | |
835 | ||
836 | ||
837 | static void f_Ccall (lua_State *L, void *ud) { | |
838 | struct CCallS *c = cast(struct CCallS *, ud); | |
839 | Closure *cl; | |
840 | cl = luaF_newCclosure(L, 0, getcurrenv(L)); | |
841 | cl->c.f = c->func; | |
842 | setclvalue(L, L->top, cl); /* push function */ | |
843 | api_incr_top(L); | |
844 | setpvalue(L->top, c->ud); /* push only argument */ | |
845 | api_incr_top(L); | |
846 | luaD_call(L, L->top - 2, 0); | |
847 | } | |
848 | ||
849 | ||
850 | LUA_API int lua_cpcall (lua_State *L, lua_CFunction func, void *ud) { | |
851 | struct CCallS c; | |
852 | int status; | |
853 | lua_lock(L); | |
854 | c.func = func; | |
855 | c.ud = ud; | |
856 | status = luaD_pcall(L, f_Ccall, &c, savestack(L, L->top), 0); | |
857 | lua_unlock(L); | |
858 | return status; | |
859 | } | |
860 | ||
861 | ||
862 | LUA_API int lua_load (lua_State *L, lua_Reader reader, void *data, | |
863 | const char *chunkname) { | |
864 | ZIO z; | |
865 | int status; | |
866 | lua_lock(L); | |
867 | if (!chunkname) chunkname = "?"; | |
868 | luaZ_init(L, &z, reader, data); | |
869 | status = luaD_protectedparser(L, &z, chunkname); | |
870 | lua_unlock(L); | |
871 | return status; | |
872 | } | |
873 | ||
874 | ||
875 | LUA_API int lua_dump (lua_State *L, lua_Writer writer, void *data) { | |
876 | int status; | |
877 | TValue *o; | |
878 | lua_lock(L); | |
879 | api_checknelems(L, 1); | |
880 | o = L->top - 1; | |
881 | if (isLfunction(o)) | |
882 | status = luaU_dump(L, clvalue(o)->l.p, writer, data, 0); | |
883 | else | |
884 | status = 1; | |
885 | lua_unlock(L); | |
886 | return status; | |
887 | } | |
888 | ||
889 | ||
890 | LUA_API int lua_status (lua_State *L) { | |
891 | return L->status; | |
892 | } | |
893 | ||
894 | ||
895 | /* | |
896 | ** Garbage-collection function | |
897 | */ | |
898 | ||
899 | LUA_API int lua_gc (lua_State *L, int what, int data) { | |
900 | int res = 0; | |
901 | global_State *g; | |
902 | lua_lock(L); | |
903 | g = G(L); | |
904 | switch (what) { | |
905 | case LUA_GCSTOP: { | |
906 | g->GCthreshold = MAX_LUMEM; | |
907 | break; | |
908 | } | |
909 | case LUA_GCRESTART: { | |
910 | g->GCthreshold = g->totalbytes; | |
911 | break; | |
912 | } | |
913 | case LUA_GCCOLLECT: { | |
914 | luaC_fullgc(L); | |
915 | break; | |
916 | } | |
917 | case LUA_GCCOUNT: { | |
918 | /* GC values are expressed in Kbytes: #bytes/2^10 */ | |
919 | res = cast_int(g->totalbytes >> 10); | |
920 | break; | |
921 | } | |
922 | case LUA_GCCOUNTB: { | |
923 | res = cast_int(g->totalbytes & 0x3ff); | |
924 | break; | |
925 | } | |
926 | case LUA_GCSTEP: { | |
927 | lu_mem a = (cast(lu_mem, data) << 10); | |
928 | if (a <= g->totalbytes) | |
929 | g->GCthreshold = g->totalbytes - a; | |
930 | else | |
931 | g->GCthreshold = 0; | |
932 | while (g->GCthreshold <= g->totalbytes) { | |
933 | luaC_step(L); | |
934 | if (g->gcstate == GCSpause) { /* end of cycle? */ | |
935 | res = 1; /* signal it */ | |
936 | break; | |
937 | } | |
938 | } | |
939 | break; | |
940 | } | |
941 | case LUA_GCSETPAUSE: { | |
942 | res = g->gcpause; | |
943 | g->gcpause = data; | |
944 | break; | |
945 | } | |
946 | case LUA_GCSETSTEPMUL: { | |
947 | res = g->gcstepmul; | |
948 | g->gcstepmul = data; | |
949 | break; | |
950 | } | |
951 | default: res = -1; /* invalid option */ | |
952 | } | |
953 | lua_unlock(L); | |
954 | return res; | |
955 | } | |
956 | ||
957 | ||
958 | ||
959 | /* | |
960 | ** miscellaneous functions | |
961 | */ | |
962 | ||
963 | ||
964 | LUA_API int lua_error (lua_State *L) { | |
965 | lua_lock(L); | |
966 | api_checknelems(L, 1); | |
967 | luaG_errormsg(L); | |
968 | lua_unlock(L); | |
969 | return 0; /* to avoid warnings */ | |
970 | } | |
971 | ||
972 | ||
973 | LUA_API int lua_next (lua_State *L, int idx) { | |
974 | StkId t; | |
975 | int more; | |
976 | lua_lock(L); | |
977 | t = index2adr(L, idx); | |
978 | api_check(L, ttistable(t)); | |
979 | more = luaH_next(L, hvalue(t), L->top - 1); | |
980 | if (more) { | |
981 | api_incr_top(L); | |
982 | } | |
983 | else /* no more elements */ | |
984 | L->top -= 1; /* remove key */ | |
985 | lua_unlock(L); | |
986 | return more; | |
987 | } | |
988 | ||
989 | ||
990 | LUA_API void lua_concat (lua_State *L, int n) { | |
991 | lua_lock(L); | |
992 | api_checknelems(L, n); | |
993 | if (n >= 2) { | |
994 | luaC_checkGC(L); | |
995 | luaV_concat(L, n, cast_int(L->top - L->base) - 1); | |
996 | L->top -= (n-1); | |
997 | } | |
998 | else if (n == 0) { /* push empty string */ | |
999 | setsvalue2s(L, L->top, luaS_newlstr(L, "", 0)); | |
1000 | api_incr_top(L); | |
1001 | } | |
1002 | /* else n == 1; nothing to do */ | |
1003 | lua_unlock(L); | |
1004 | } | |
1005 | ||
1006 | ||
1007 | LUA_API lua_Alloc lua_getallocf (lua_State *L, void **ud) { | |
1008 | lua_Alloc f; | |
1009 | lua_lock(L); | |
1010 | if (ud) *ud = G(L)->ud; | |
1011 | f = G(L)->frealloc; | |
1012 | lua_unlock(L); | |
1013 | return f; | |
1014 | } | |
1015 | ||
1016 | ||
1017 | LUA_API void lua_setallocf (lua_State *L, lua_Alloc f, void *ud) { | |
1018 | lua_lock(L); | |
1019 | G(L)->ud = ud; | |
1020 | G(L)->frealloc = f; | |
1021 | lua_unlock(L); | |
1022 | } | |
1023 | ||
1024 | ||
1025 | LUA_API void *lua_newuserdata (lua_State *L, size_t size) { | |
1026 | Udata *u; | |
1027 | lua_lock(L); | |
1028 | luaC_checkGC(L); | |
1029 | u = luaS_newudata(L, size, getcurrenv(L)); | |
1030 | setuvalue(L, L->top, u); | |
1031 | api_incr_top(L); | |
1032 | lua_unlock(L); | |
1033 | return u + 1; | |
1034 | } | |
1035 | ||
1036 | ||
1037 | ||
1038 | ||
1039 | static const char *aux_upvalue (StkId fi, int n, TValue **val) { | |
1040 | Closure *f; | |
1041 | if (!ttisfunction(fi)) return NULL; | |
1042 | f = clvalue(fi); | |
1043 | if (f->c.isC) { | |
1044 | if (!(1 <= n && n <= f->c.nupvalues)) return NULL; | |
1045 | *val = &f->c.upvalue[n-1]; | |
1046 | return ""; | |
1047 | } | |
1048 | else { | |
1049 | Proto *p = f->l.p; | |
1050 | if (!(1 <= n && n <= p->sizeupvalues)) return NULL; | |
1051 | *val = f->l.upvals[n-1]->v; | |
1052 | return getstr(p->upvalues[n-1]); | |
1053 | } | |
1054 | } | |
1055 | ||
1056 | ||
1057 | LUA_API const char *lua_getupvalue (lua_State *L, int funcindex, int n) { | |
1058 | const char *name; | |
1059 | TValue *val; | |
1060 | lua_lock(L); | |
1061 | name = aux_upvalue(index2adr(L, funcindex), n, &val); | |
1062 | if (name) { | |
1063 | setobj2s(L, L->top, val); | |
1064 | api_incr_top(L); | |
1065 | } | |
1066 | lua_unlock(L); | |
1067 | return name; | |
1068 | } | |
1069 | ||
1070 | ||
1071 | LUA_API const char *lua_setupvalue (lua_State *L, int funcindex, int n) { | |
1072 | const char *name; | |
1073 | TValue *val; | |
1074 | StkId fi; | |
1075 | lua_lock(L); | |
1076 | fi = index2adr(L, funcindex); | |
1077 | api_checknelems(L, 1); | |
1078 | name = aux_upvalue(fi, n, &val); | |
1079 | if (name) { | |
1080 | L->top--; | |
1081 | setobj(L, val, L->top); | |
1082 | luaC_barrier(L, clvalue(fi), L->top); | |
1083 | } | |
1084 | lua_unlock(L); | |
1085 | return name; | |
1086 | } | |
1087 |