2 ** Base and coroutine library.
3 ** Copyright (C) 2005-2023 Mike Pall. See Copyright Notice in luajit.h
5 ** Major portions taken verbatim or adapted from the Lua interpreter.
6 ** Copyright (C) 1994-2011 Lua.org, PUC-Rio. See Copyright Notice in lua.h
34 #include "lj_dispatch.h"
36 #include "lj_strscan.h"
37 #include "lj_strfmt.h"
40 /* -- Base library: checks ------------------------------------------------ */
42 #define LJLIB_MODULE_base
44 LJLIB_ASM(assert) LJLIB_REC(.)
46 lj_lib_checkany(L
, 1);
47 if (L
->top
== L
->base
+1)
48 lj_err_caller(L
, LJ_ERR_ASSERT
);
49 else if (tvisstr(L
->base
+1) || tvisnumber(L
->base
+1))
50 lj_err_callermsg(L
, strdata(lj_lib_checkstr(L
, 2)));
53 return FFH_UNREACHABLE
;
59 LJLIB_PUSH(top
-1) /* boolean */
60 LJLIB_PUSH("userdata")
65 LJLIB_PUSH("function")
69 LJLIB_PUSH(top
-9) /* userdata */
71 LJLIB_ASM_(type
) LJLIB_REC(.)
72 /* Recycle the lj_lib_checkany(L, 1) from assert. */
74 /* -- Base library: iterators --------------------------------------------- */
76 /* This solves a circular dependency problem -- change FF_next_N as needed. */
77 LJ_STATIC_ASSERT((int)FF_next
== FF_next_N
);
79 LJLIB_ASM(next
) LJLIB_REC(.)
81 lj_lib_checktab(L
, 1);
82 lj_err_msg(L
, LJ_ERR_NEXTIDX
);
83 return FFH_UNREACHABLE
;
86 #if LJ_52 || LJ_HASFFI
87 static int ffh_pairs(lua_State
*L
, MMS mm
)
89 TValue
*o
= lj_lib_checkany(L
, 1);
90 cTValue
*mo
= lj_meta_lookup(L
, o
, mm
);
91 if ((LJ_52
|| tviscdata(o
)) && !tvisnil(mo
)) {
92 L
->top
= o
+1; /* Only keep one argument. */
93 copyTV(L
, L
->base
-1-LJ_FR2
, mo
); /* Replace callable. */
96 if (!tvistab(o
)) lj_err_argt(L
, 1, LUA_TTABLE
);
97 if (LJ_FR2
) { copyTV(L
, o
-1, o
); o
--; }
98 setfuncV(L
, o
-1, funcV(lj_lib_upvalue(L
, 1)));
99 if (mm
== MM_pairs
) setnilV(o
+1); else setintV(o
+1, 0);
104 #define ffh_pairs(L, mm) (lj_lib_checktab(L, 1), FFH_UNREACHABLE)
108 LJLIB_ASM(pairs
) LJLIB_REC(xpairs
0)
110 return ffh_pairs(L
, MM_pairs
);
113 LJLIB_NOREGUV
LJLIB_ASM(ipairs_aux
) LJLIB_REC(.)
115 lj_lib_checktab(L
, 1);
116 lj_lib_checkint(L
, 2);
117 return FFH_UNREACHABLE
;
121 LJLIB_ASM(ipairs
) LJLIB_REC(xpairs
1)
123 return ffh_pairs(L
, MM_ipairs
);
126 /* -- Base library: getters and setters ----------------------------------- */
128 LJLIB_ASM_(getmetatable
) LJLIB_REC(.)
129 /* Recycle the lj_lib_checkany(L, 1) from assert. */
131 LJLIB_ASM(setmetatable
) LJLIB_REC(.)
133 GCtab
*t
= lj_lib_checktab(L
, 1);
134 GCtab
*mt
= lj_lib_checktabornil(L
, 2);
135 if (!tvisnil(lj_meta_lookup(L
, L
->base
, MM_metatable
)))
136 lj_err_caller(L
, LJ_ERR_PROTMT
);
137 setgcref(t
->metatable
, obj2gco(mt
));
138 if (mt
) { lj_gc_objbarriert(L
, t
, mt
); }
139 settabV(L
, L
->base
-1-LJ_FR2
, t
);
143 LJLIB_CF(getfenv
) LJLIB_REC(.)
146 cTValue
*o
= L
->base
;
147 if (!(o
< L
->top
&& tvisfunc(o
))) {
148 int level
= lj_lib_optint(L
, 1, 1);
149 o
= lj_debug_frame(L
, level
, &level
);
151 lj_err_arg(L
, 1, LJ_ERR_INVLVL
);
155 settabV(L
, L
->top
++, isluafunc(fn
) ? tabref(fn
->l
.env
) : tabref(L
->env
));
162 GCtab
*t
= lj_lib_checktab(L
, 2);
163 cTValue
*o
= L
->base
;
164 if (!(o
< L
->top
&& tvisfunc(o
))) {
165 int level
= lj_lib_checkint(L
, 1);
167 /* NOBARRIER: A thread (i.e. L) is never black. */
168 setgcref(L
->env
, obj2gco(t
));
171 o
= lj_debug_frame(L
, level
, &level
);
173 lj_err_arg(L
, 1, LJ_ERR_INVLVL
);
178 lj_err_caller(L
, LJ_ERR_SETFENV
);
179 setgcref(fn
->l
.env
, obj2gco(t
));
180 lj_gc_objbarrier(L
, obj2gco(fn
), t
);
181 setfuncV(L
, L
->top
++, fn
);
185 LJLIB_ASM(rawget
) LJLIB_REC(.)
187 lj_lib_checktab(L
, 1);
188 lj_lib_checkany(L
, 2);
189 return FFH_UNREACHABLE
;
192 LJLIB_CF(rawset
) LJLIB_REC(.)
194 lj_lib_checktab(L
, 1);
195 lj_lib_checkany(L
, 2);
196 L
->top
= 1+lj_lib_checkany(L
, 3);
201 LJLIB_CF(rawequal
) LJLIB_REC(.)
203 cTValue
*o1
= lj_lib_checkany(L
, 1);
204 cTValue
*o2
= lj_lib_checkany(L
, 2);
205 setboolV(L
->top
-1, lj_obj_equal(o1
, o2
));
210 LJLIB_CF(rawlen
) LJLIB_REC(.)
212 cTValue
*o
= L
->base
;
214 if (L
->top
> o
&& tvisstr(o
))
215 len
= (int32_t)strV(o
)->len
;
217 len
= (int32_t)lj_tab_len(lj_lib_checktab(L
, 1));
218 setintV(L
->top
-1, len
);
225 GCtab
*t
= lj_lib_checktab(L
, 1);
226 int32_t n
, i
= lj_lib_optint(L
, 2, 1);
227 int32_t e
= (L
->base
+3-1 < L
->top
&& !tvisnil(L
->base
+3-1)) ?
228 lj_lib_checkint(L
, 3) : (int32_t)lj_tab_len(t
);
231 nu
= (uint32_t)e
- (uint32_t)i
;
233 if (nu
>= LUAI_MAXCSTACK
|| !lua_checkstack(L
, n
))
234 lj_err_caller(L
, LJ_ERR_UNPACK
);
236 cTValue
*tv
= lj_tab_getint(t
, i
);
238 copyTV(L
, L
->top
++, tv
);
246 LJLIB_CF(select
) LJLIB_REC(.)
248 int32_t n
= (int32_t)(L
->top
- L
->base
);
249 if (n
>= 1 && tvisstr(L
->base
) && *strVdata(L
->base
) == '#') {
250 setintV(L
->top
-1, n
-1);
253 int32_t i
= lj_lib_checkint(L
, 1);
254 if (i
< 0) i
= n
+ i
; else if (i
> n
) i
= n
;
256 lj_err_arg(L
, 1, LJ_ERR_IDXRNG
);
261 /* -- Base library: conversions ------------------------------------------- */
263 LJLIB_ASM(tonumber
) LJLIB_REC(.)
265 int32_t base
= lj_lib_optint(L
, 2, 10);
267 TValue
*o
= lj_lib_checkany(L
, 1);
268 if (lj_strscan_numberobj(o
)) {
269 copyTV(L
, L
->base
-1-LJ_FR2
, o
);
274 CTState
*cts
= ctype_cts(L
);
275 CType
*ct
= lj_ctype_rawref(cts
, cdataV(o
)->ctypeid
);
276 if (ctype_isenum(ct
->info
)) ct
= ctype_child(cts
, ct
);
277 if (ctype_isnum(ct
->info
) || ctype_iscomplex(ct
->info
)) {
278 if (LJ_DUALNUM
&& ctype_isinteger_or_bool(ct
->info
) &&
279 ct
->size
<= 4 && !(ct
->size
== 4 && (ct
->info
& CTF_UNSIGNED
))) {
281 lj_cconv_ct_tv(cts
, ctype_get(cts
, CTID_INT32
), (uint8_t *)&i
, o
, 0);
282 setintV(L
->base
-1-LJ_FR2
, i
);
285 lj_cconv_ct_tv(cts
, ctype_get(cts
, CTID_DOUBLE
),
286 (uint8_t *)&(L
->base
-1-LJ_FR2
)->n
, o
, 0);
292 const char *p
= strdata(lj_lib_checkstr(L
, 1));
294 unsigned int neg
= 0;
296 if (base
< 2 || base
> 36)
297 lj_err_arg(L
, 2, LJ_ERR_BASERNG
);
298 while (lj_char_isspace((unsigned char)(*p
))) p
++;
299 if (*p
== '-') { p
++; neg
= 1; } else if (*p
== '+') { p
++; }
300 if (lj_char_isalnum((unsigned char)(*p
))) {
301 ul
= strtoul(p
, &ep
, base
);
303 while (lj_char_isspace((unsigned char)(*ep
))) ep
++;
305 if (LJ_DUALNUM
&& LJ_LIKELY(ul
< 0x80000000u
+neg
)) {
306 if (neg
) ul
= ~ul
+1u;
307 setintV(L
->base
-1-LJ_FR2
, (int32_t)ul
);
309 lua_Number n
= (lua_Number
)ul
;
311 setnumV(L
->base
-1-LJ_FR2
, n
);
318 setnilV(L
->base
-1-LJ_FR2
);
322 LJLIB_ASM(tostring
) LJLIB_REC(.)
324 TValue
*o
= lj_lib_checkany(L
, 1);
326 L
->top
= o
+1; /* Only keep one argument. */
327 if (!tvisnil(mo
= lj_meta_lookup(L
, o
, MM_tostring
))) {
328 copyTV(L
, L
->base
-1-LJ_FR2
, mo
); /* Replace callable. */
332 setstrV(L
, L
->base
-1-LJ_FR2
, lj_strfmt_obj(L
, L
->base
));
336 /* -- Base library: throw and catch errors -------------------------------- */
340 int32_t level
= lj_lib_optint(L
, 2, 1);
342 if (lua_isstring(L
, 1) && level
> 0) {
343 luaL_where(L
, level
);
350 LJLIB_ASM(pcall
) LJLIB_REC(.)
352 lj_lib_checkany(L
, 1);
353 lj_lib_checkfunc(L
, 2); /* For xpcall only. */
354 return FFH_UNREACHABLE
;
356 LJLIB_ASM_(xpcall
) LJLIB_REC(.)
358 /* -- Base library: load Lua code ----------------------------------------- */
360 static int load_aux(lua_State
*L
, int status
, int envarg
)
362 if (status
== LUA_OK
) {
363 if (tvistab(L
->base
+envarg
-1)) {
364 GCfunc
*fn
= funcV(L
->top
-1);
365 GCtab
*t
= tabV(L
->base
+envarg
-1);
366 setgcref(fn
->c
.env
, obj2gco(t
));
367 lj_gc_objbarrier(L
, fn
, t
);
378 GCstr
*fname
= lj_lib_optstr(L
, 1);
379 GCstr
*mode
= lj_lib_optstr(L
, 2);
381 lua_settop(L
, 3); /* Ensure env arg exists. */
382 status
= luaL_loadfilex(L
, fname
? strdata(fname
) : NULL
,
383 mode
? strdata(mode
) : NULL
);
384 return load_aux(L
, status
, 3);
387 static const char *reader_func(lua_State
*L
, void *ud
, size_t *size
)
390 luaL_checkstack(L
, 2, "too many nested functions");
391 copyTV(L
, L
->top
++, L
->base
);
392 lua_call(L
, 0, 1); /* Call user-supplied function. */
394 if (tvisnil(L
->top
)) {
397 } else if (tvisstr(L
->top
) || tvisnumber(L
->top
)) {
398 copyTV(L
, L
->base
+4, L
->top
); /* Anchor string in reserved stack slot. */
399 return lua_tolstring(L
, 5, size
);
401 lj_err_caller(L
, LJ_ERR_RDRSTR
);
408 GCstr
*name
= lj_lib_optstr(L
, 2);
409 GCstr
*mode
= lj_lib_optstr(L
, 3);
411 if (L
->base
< L
->top
&&
412 (tvisstr(L
->base
) || tvisnumber(L
->base
) || tvisbuf(L
->base
))) {
415 if (tvisbuf(L
->base
)) {
416 SBufExt
*sbx
= bufV(L
->base
);
419 if (!name
) name
= &G(L
)->strempty
; /* Buffers are not NUL-terminated. */
421 GCstr
*str
= lj_lib_checkstr(L
, 1);
425 lua_settop(L
, 4); /* Ensure env arg exists. */
426 status
= luaL_loadbufferx(L
, s
, len
, name
? strdata(name
) : s
,
427 mode
? strdata(mode
) : NULL
);
429 lj_lib_checkfunc(L
, 1);
430 lua_settop(L
, 5); /* Reserve a slot for the string from the reader. */
431 status
= lua_loadx(L
, reader_func
, NULL
, name
? strdata(name
) : "=(load)",
432 mode
? strdata(mode
) : NULL
);
434 return load_aux(L
, status
, 4);
439 return lj_cf_load(L
);
444 GCstr
*fname
= lj_lib_optstr(L
, 1);
447 if (luaL_loadfile(L
, fname
? strdata(fname
) : NULL
) != LUA_OK
)
449 lua_call(L
, 0, LUA_MULTRET
);
450 return (int)(L
->top
- L
->base
) - 1;
453 /* -- Base library: GC control -------------------------------------------- */
457 setintV(L
->top
++, (int32_t)(G(L
)->gc
.total
>> 10));
461 LJLIB_CF(collectgarbage
)
463 int opt
= lj_lib_checkopt(L
, 1, LUA_GCCOLLECT
, /* ORDER LUA_GC* */
464 "\4stop\7restart\7collect\5count\1\377\4step\10setpause\12setstepmul\1\377\11isrunning");
465 int32_t data
= lj_lib_optint(L
, 2, 0);
466 if (opt
== LUA_GCCOUNT
) {
467 setnumV(L
->top
, (lua_Number
)G(L
)->gc
.total
/1024.0);
469 int res
= lua_gc(L
, opt
, data
);
470 if (opt
== LUA_GCSTEP
|| opt
== LUA_GCISRUNNING
)
471 setboolV(L
->top
, res
);
473 setintV(L
->top
, res
);
479 /* -- Base library: miscellaneous functions ------------------------------- */
481 LJLIB_PUSH(top
-2) /* Upvalue holds weak table. */
485 lua_newuserdata(L
, 0);
486 if (lua_toboolean(L
, 1) == 0) { /* newproxy(): without metatable. */
488 } else if (lua_isboolean(L
, 1)) { /* newproxy(true): with metatable. */
490 lua_pushvalue(L
, -1);
491 lua_pushboolean(L
, 1);
492 lua_rawset(L
, lua_upvalueindex(1)); /* Remember mt in weak table. */
493 } else { /* newproxy(proxy): inherit metatable. */
495 if (lua_getmetatable(L
, 1)) {
496 lua_rawget(L
, lua_upvalueindex(1));
497 validproxy
= lua_toboolean(L
, -1);
501 lj_err_arg(L
, 1, LJ_ERR_NOPROXY
);
502 lua_getmetatable(L
, 1);
504 lua_setmetatable(L
, 2);
508 LJLIB_PUSH("tostring")
511 ptrdiff_t i
, nargs
= L
->top
- L
->base
;
512 cTValue
*tv
= lj_tab_getstr(tabref(L
->env
), strV(lj_lib_upvalue(L
, 1)));
514 if (tv
&& !tvisnil(tv
)) {
515 copyTV(L
, L
->top
++, tv
);
517 setstrV(L
, L
->top
++, strV(lj_lib_upvalue(L
, 1)));
518 lua_gettable(L
, LUA_GLOBALSINDEX
);
521 shortcut
= (tvisfunc(tv
) && funcV(tv
)->c
.ffid
== FF_tostring
) &&
522 !gcrefu(basemt_it(G(L
), LJ_TNUMX
));
523 for (i
= 0; i
< nargs
; i
++) {
524 cTValue
*o
= &L
->base
[i
];
528 if (shortcut
&& (str
= lj_strfmt_wstrnum(L
, o
, &len
)) != NULL
) {
531 copyTV(L
, L
->top
+1, o
);
532 copyTV(L
, L
->top
, L
->top
-1);
535 str
= lua_tolstring(L
, -1, &size
);
537 lj_err_caller(L
, LJ_ERR_PRTOSTR
);
542 fwrite(str
, 1, size
, stdout
);
551 #include "lj_libdef.h"
553 /* -- Coroutine library --------------------------------------------------- */
555 #define LJLIB_MODULE_coroutine
557 LJLIB_CF(coroutine_status
)
561 if (!(L
->top
> L
->base
&& tvisthread(L
->base
)))
562 lj_err_arg(L
, 1, LJ_ERR_NOCORO
);
563 co
= threadV(L
->base
);
564 if (co
== L
) s
= "running";
565 else if (co
->status
== LUA_YIELD
) s
= "suspended";
566 else if (co
->status
!= LUA_OK
) s
= "dead";
567 else if (co
->base
> tvref(co
->stack
)+1+LJ_FR2
) s
= "normal";
568 else if (co
->top
== co
->base
) s
= "dead";
569 else s
= "suspended";
570 lua_pushstring(L
, s
);
574 LJLIB_CF(coroutine_running
)
577 int ismain
= lua_pushthread(L
);
578 setboolV(L
->top
++, ismain
);
581 if (lua_pushthread(L
))
587 LJLIB_CF(coroutine_isyieldable
)
589 setboolV(L
->top
++, cframe_canyield(L
->cframe
));
593 LJLIB_CF(coroutine_create
)
596 if (!(L
->base
< L
->top
&& tvisfunc(L
->base
)))
597 lj_err_argt(L
, 1, LUA_TFUNCTION
);
598 L1
= lua_newthread(L
);
599 setfuncV(L
, L1
->top
++, funcV(L
->base
));
603 LJLIB_ASM(coroutine_yield
)
605 lj_err_caller(L
, LJ_ERR_CYIELD
);
606 return FFH_UNREACHABLE
;
609 static int ffh_resume(lua_State
*L
, lua_State
*co
, int wrap
)
611 if (co
->cframe
!= NULL
|| co
->status
> LUA_YIELD
||
612 (co
->status
== LUA_OK
&& co
->top
== co
->base
)) {
613 ErrMsg em
= co
->cframe
? LJ_ERR_CORUN
: LJ_ERR_CODEAD
;
614 if (wrap
) lj_err_caller(L
, em
);
615 setboolV(L
->base
-1-LJ_FR2
, 0);
616 setstrV(L
, L
->base
-LJ_FR2
, lj_err_str(L
, em
));
619 lj_state_growstack(co
, (MSize
)(L
->top
- L
->base
));
623 LJLIB_ASM(coroutine_resume
)
625 if (!(L
->top
> L
->base
&& tvisthread(L
->base
)))
626 lj_err_arg(L
, 1, LJ_ERR_NOCORO
);
627 return ffh_resume(L
, threadV(L
->base
), 0);
630 LJLIB_NOREG
LJLIB_ASM(coroutine_wrap_aux
)
632 return ffh_resume(L
, threadV(lj_lib_upvalue(L
, 1)), 1);
635 /* Inline declarations. */
636 LJ_ASMF
void lj_ff_coroutine_wrap_aux(void);
637 #if !(LJ_TARGET_MIPS && defined(ljamalg_c))
638 LJ_FUNCA_NORET
void LJ_FASTCALL
lj_ffh_coroutine_wrap_err(lua_State
*L
,
642 /* Error handler, called from assembler VM. */
643 void LJ_FASTCALL
lj_ffh_coroutine_wrap_err(lua_State
*L
, lua_State
*co
)
645 co
->top
--; copyTV(L
, L
->top
, co
->top
); L
->top
++;
646 if (tvisstr(L
->top
-1))
647 lj_err_callermsg(L
, strVdata(L
->top
-1));
652 /* Forward declaration. */
653 static void setpc_wrap_aux(lua_State
*L
, GCfunc
*fn
);
655 LJLIB_CF(coroutine_wrap
)
658 lj_cf_coroutine_create(L
);
659 fn
= lj_lib_pushcc(L
, lj_ffh_coroutine_wrap_aux
, FF_coroutine_wrap_aux
, 1);
660 setpc_wrap_aux(L
, fn
);
664 #include "lj_libdef.h"
666 /* Fix the PC of wrap_aux. Really ugly workaround. */
667 static void setpc_wrap_aux(lua_State
*L
, GCfunc
*fn
)
669 setmref(fn
->c
.pc
, &L2GG(L
)->bcff
[lj_lib_init_coroutine
[1]+2]);
672 /* ------------------------------------------------------------------------ */
674 static void newproxy_weaktable(lua_State
*L
)
676 /* NOBARRIER: The table is new (marked white). */
677 GCtab
*t
= lj_tab_new(L
, 0, 1);
678 settabV(L
, L
->top
++, t
);
679 setgcref(t
->metatable
, obj2gco(t
));
680 setstrV(L
, lj_tab_setstr(L
, t
, lj_str_newlit(L
, "__mode")),
681 lj_str_newlit(L
, "kv"));
682 t
->nomm
= (uint8_t)(~(1u<<MM_mode
));
685 LUALIB_API
int luaopen_base(lua_State
*L
)
687 /* NOBARRIER: Table and value are the same. */
688 GCtab
*env
= tabref(L
->env
);
689 settabV(L
, lj_tab_setstr(L
, env
, lj_str_newlit(L
, "_G")), env
);
690 lua_pushliteral(L
, LUA_VERSION
); /* top-3. */
691 newproxy_weaktable(L
); /* top-2. */
692 LJ_LIB_REG(L
, "_G", base
);
693 LJ_LIB_REG(L
, LUA_COLIBNAME
, coroutine
);