// This file is part of the Luau programming language and is licensed under MIT License; see LICENSE.txt for details // This code is based on Lua 5.x implementation licensed under MIT License; see lua_LICENSE.txt for details #include "lbuiltins.h" #include "lstate.h" #include "lstring.h" #include "ltable.h" #include "lgc.h" #include "lnumutils.h" #include "ldo.h" #include "lbuffer.h" #include "lvector.h" #include #include #ifdef _MSC_VER #include #endif #ifdef LUAU_TARGET_SSE41 #include #ifndef _MSC_VER #include // on MSVC this comes from intrin.h #endif #endif // luauF functions implement FASTCALL instruction that performs a direct execution of some builtin functions from the VM // The rule of thumb is that FASTCALL functions can not call user code, yield, fail, or reallocate stack. // If types of the arguments mismatch, luauF_* needs to return -1 and the execution will fall back to the usual call path // If luauF_* succeeds, it needs to return *all* requested arguments, filling results with nil as appropriate. // On input, nparams refers to the actual number of arguments (0+), whereas nresults contains LUA_MULTRET for arbitrary returns or 0+ for a // fixed-length return // Because of this, and the fact that "extra" returned values will be ignored, implementations below typically check that nresults is <= expected // number, which covers the LUA_MULTRET case. static int luauF_assert(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults == 0 && !l_isfalse(arg0)) { return 0; } return -1; } static int luauF_abs(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, fabs(a1)); return 1; } return -1; } static int luauF_acos(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, acos(a1)); return 1; } return -1; } static int luauF_asin(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, asin(a1)); return 1; } return -1; } static int luauF_atan2(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); setnvalue(res, atan2(a1, a2)); return 1; } return -1; } static int luauF_atan(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, atan(a1)); return 1; } return -1; } LUAU_FASTMATH_BEGIN static int luauF_ceil(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, ceil(a1)); return 1; } return -1; } LUAU_FASTMATH_END static int luauF_cosh(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, cosh(a1)); return 1; } return -1; } static int luauF_cos(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, cos(a1)); return 1; } return -1; } static int luauF_deg(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); const double rpd = (3.14159265358979323846 / 180.0); setnvalue(res, a1 / rpd); return 1; } return -1; } static int luauF_exp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, exp(a1)); return 1; } return -1; } LUAU_FASTMATH_BEGIN static int luauF_floor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, floor(a1)); return 1; } return -1; } LUAU_FASTMATH_END static int luauF_fmod(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); setnvalue(res, fmod(a1, a2)); return 1; } return -1; } static int luauF_frexp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 2 && ttisnumber(arg0)) { double a1 = nvalue(arg0); int e; double f = frexp(a1, &e); setnvalue(res, f); setnvalue(res + 1, double(e)); return 2; } return -1; } static int luauF_ldexp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); setnvalue(res, ldexp(a1, int(a2))); return 1; } return -1; } static int luauF_log10(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, log10(a1)); return 1; } return -1; } static int luauF_log(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); if (nparams == 1) { setnvalue(res, log(a1)); return 1; } else if (ttisnumber(args)) { double a2 = nvalue(args); if (a2 == 2.0) { setnvalue(res, log2(a1)); return 1; } else if (a2 == 10.0) { setnvalue(res, log10(a1)); return 1; } else { setnvalue(res, log(a1) / log(a2)); return 1; } } } return -1; } static int luauF_max(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); double r = (a2 > a1) ? a2 : a1; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); r = (a > r) ? a : r; } setnvalue(res, r); return 1; } return -1; } static int luauF_min(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); double r = (a2 < a1) ? a2 : a1; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); r = (a < r) ? a : r; } setnvalue(res, r); return 1; } return -1; } static int luauF_modf(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 2 && ttisnumber(arg0)) { double a1 = nvalue(arg0); double ip; double fp = modf(a1, &ip); setnvalue(res, ip); setnvalue(res + 1, fp); return 2; } return -1; } static int luauF_pow(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); setnvalue(res, pow(a1, a2)); return 1; } return -1; } static int luauF_rad(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); const double rpd = (3.14159265358979323846 / 180.0); setnvalue(res, a1 * rpd); return 1; } return -1; } static int luauF_sinh(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, sinh(a1)); return 1; } return -1; } static int luauF_sin(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, sin(a1)); return 1; } return -1; } LUAU_FASTMATH_BEGIN static int luauF_sqrt(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, sqrt(a1)); return 1; } return -1; } LUAU_FASTMATH_END static int luauF_tanh(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, tanh(a1)); return 1; } return -1; } static int luauF_tan(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, tan(a1)); return 1; } return -1; } static int luauF_arshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u; luai_num2unsigned(u, a1); int s = int(a2); // note: we only specialize fast-path that doesn't require further conditionals (negative shifts and shifts greater or equal to bit width can // be handled generically) if (unsigned(s) < 32) { // note: technically right shift of negative values is UB, but this behavior is getting defined in C++20 and all compilers do the right // (shift) thing. uint32_t r = int32_t(u) >> s; setnvalue(res, double(r)); return 1; } } return -1; } static int luauF_band(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u1, u2; luai_num2unsigned(u1, a1); luai_num2unsigned(u2, a2); uint32_t r = u1 & u2; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); unsigned u; luai_num2unsigned(u, a); r &= u; } setnvalue(res, double(r)); return 1; } return -1; } static int luauF_bnot(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); unsigned u; luai_num2unsigned(u, a1); uint32_t r = ~u; setnvalue(res, double(r)); return 1; } return -1; } static int luauF_bor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u1, u2; luai_num2unsigned(u1, a1); luai_num2unsigned(u2, a2); uint32_t r = u1 | u2; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); unsigned u; luai_num2unsigned(u, a); r |= u; } setnvalue(res, double(r)); return 1; } return -1; } static int luauF_bxor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u1, u2; luai_num2unsigned(u1, a1); luai_num2unsigned(u2, a2); uint32_t r = u1 ^ u2; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); unsigned u; luai_num2unsigned(u, a); r ^= u; } setnvalue(res, double(r)); return 1; } return -1; } static int luauF_btest(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u1, u2; luai_num2unsigned(u1, a1); luai_num2unsigned(u2, a2); uint32_t r = u1 & u2; for (int i = 3; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; double a = nvalue(args + (i - 2)); unsigned u; luai_num2unsigned(u, a); r &= u; } setbvalue(res, r != 0); return 1; } return -1; } static int luauF_extract(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned n; luai_num2unsigned(n, a1); int f = int(a2); if (nparams == 2) { if (unsigned(f) < 32) { uint32_t m = 1; uint32_t r = (n >> f) & m; setnvalue(res, double(r)); return 1; } } else if (ttisnumber(args + 1)) { double a3 = nvalue(args + 1); int w = int(a3); if (f >= 0 && w > 0 && f + w <= 32) { uint32_t m = ~(0xfffffffeu << (w - 1)); uint32_t r = (n >> f) & m; setnvalue(res, double(r)); return 1; } } } return -1; } static int luauF_lrotate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u; luai_num2unsigned(u, a1); int s = int(a2); // MSVC doesn't recognize the rotate form that is UB-safe #ifdef _MSC_VER uint32_t r = _rotl(u, s); #else uint32_t r = (u << (s & 31)) | (u >> ((32 - s) & 31)); #endif setnvalue(res, double(r)); return 1; } return -1; } static int luauF_lshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u; luai_num2unsigned(u, a1); int s = int(a2); // note: we only specialize fast-path that doesn't require further conditionals (negative shifts and shifts greater or equal to bit width can // be handled generically) if (unsigned(s) < 32) { uint32_t r = u << s; setnvalue(res, double(r)); return 1; } } return -1; } static int luauF_replace(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { double a1 = nvalue(arg0); double a2 = nvalue(args); double a3 = nvalue(args + 1); unsigned n, v; luai_num2unsigned(n, a1); luai_num2unsigned(v, a2); int f = int(a3); if (nparams == 3) { if (unsigned(f) < 32) { uint32_t m = 1; uint32_t r = (n & ~(m << f)) | ((v & m) << f); setnvalue(res, double(r)); return 1; } } else if (ttisnumber(args + 2)) { double a4 = nvalue(args + 2); int w = int(a4); if (f >= 0 && w > 0 && f + w <= 32) { uint32_t m = ~(0xfffffffeu << (w - 1)); uint32_t r = (n & ~(m << f)) | ((v & m) << f); setnvalue(res, double(r)); return 1; } } } return -1; } static int luauF_rrotate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u; luai_num2unsigned(u, a1); int s = int(a2); // MSVC doesn't recognize the rotate form that is UB-safe #ifdef _MSC_VER uint32_t r = _rotr(u, s); #else uint32_t r = (u >> (s & 31)) | (u << ((32 - s) & 31)); #endif setnvalue(res, double(r)); return 1; } return -1; } static int luauF_rshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned u; luai_num2unsigned(u, a1); int s = int(a2); // note: we only specialize fast-path that doesn't require further conditionals (negative shifts and shifts greater or equal to bit width can // be handled generically) if (unsigned(s) < 32) { uint32_t r = u >> s; setnvalue(res, double(r)); return 1; } } return -1; } static int luauF_type(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1) { int tt = ttype(arg0); TString* ttname = L->global->ttname[tt]; setsvalue(L, res, ttname); return 1; } return -1; } static int luauF_byte(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && ttisstring(arg0) && ttisnumber(args)) { TString* ts = tsvalue(arg0); int i = int(nvalue(args)); int j = (nparams >= 3) ? (ttisnumber(args + 1) ? int(nvalue(args + 1)) : 0) : i; if (i >= 1 && j >= i && j <= int(ts->len)) { int c = j - i + 1; const char* s = getstr(ts); // for vararg returns, we only support a single result // this is because this frees us from concerns about stack space if (c == (nresults < 0 ? 1 : nresults)) { for (int k = 0; k < c; ++k) { setnvalue(res + k, uint8_t(s[i + k - 1])); } return c; } } } return -1; } static int luauF_char(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { char buffer[8]; if (nparams < int(sizeof(buffer)) && nresults <= 1) { if (luaC_needsGC(L)) return -1; // we can't call luaC_checkGC so fall back to C implementation if (nparams >= 1) { if (!ttisnumber(arg0)) return -1; int ch = int(nvalue(arg0)); if ((unsigned char)(ch) != ch) return -1; buffer[0] = ch; } for (int i = 2; i <= nparams; ++i) { if (!ttisnumber(args + (i - 2))) return -1; int ch = int(nvalue(args + (i - 2))); if ((unsigned char)(ch) != ch) return -1; buffer[i - 1] = ch; } buffer[nparams] = 0; setsvalue(L, res, luaS_newlstr(L, buffer, nparams)); return 1; } return -1; } static int luauF_len(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisstring(arg0)) { TString* ts = tsvalue(arg0); setnvalue(res, int(ts->len)); return 1; } return -1; } static int luauF_typeof(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1) { const TString* ttname = luaT_objtypenamestr(L, arg0); setsvalue(L, res, ttname); return 1; } return -1; } static int luauF_sub(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisstring(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { TString* ts = tsvalue(arg0); int i = int(nvalue(args)); int j = int(nvalue(args + 1)); if (luaC_needsGC(L)) return -1; // we can't call luaC_checkGC so fall back to C implementation if (i >= 1 && j >= i && unsigned(j - 1) < unsigned(ts->len)) { setsvalue(L, res, luaS_newlstr(L, getstr(ts) + (i - 1), j - i + 1)); return 1; } } return -1; } static int luauF_clamp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { double v = nvalue(arg0); double min = nvalue(args); double max = nvalue(args + 1); if (min <= max) { double r = v < min ? min : v; r = r > max ? max : r; setnvalue(res, r); return 1; } } return -1; } static int luauF_sign(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double v = nvalue(arg0); setnvalue(res, v > 0.0 ? 1.0 : v < 0.0 ? -1.0 : 0.0); return 1; } return -1; } LUAU_FASTMATH_BEGIN static int luauF_round(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double v = nvalue(arg0); setnvalue(res, round(v)); return 1; } return -1; } LUAU_FASTMATH_END static int luauF_rawequal(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1) { setbvalue(res, luaO_rawequalObj(arg0, args)); return 1; } return -1; } static int luauF_rawget(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttistable(arg0)) { setobj2s(L, res, luaH_get(hvalue(arg0), args)); return 1; } return -1; } static int luauF_rawset(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttistable(arg0)) { const TValue* key = args; if (ttisnil(key)) return -1; else if (ttisnumber(key) && luai_numisnan(nvalue(key))) return -1; else if (ttisvector(key) && luai_vecisnan(vvalue(key))) return -1; LuaTable* t = hvalue(arg0); if (t->readonly) return -1; setobj2s(L, res, arg0); setobj2t(L, luaH_set(L, t, args), args + 1); luaC_barriert(L, t, args + 1); return 1; } return -1; } static int luauF_tinsert(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams == 2 && nresults <= 0 && ttistable(arg0)) { LuaTable* t = hvalue(arg0); if (t->readonly) return -1; int pos = luaH_getn(t) + 1; setobj2t(L, luaH_setnum(L, t, pos), args); luaC_barriert(L, t, args); return 0; } return -1; } static int luauF_tunpack(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults < 0 && ttistable(arg0)) { LuaTable* t = hvalue(arg0); int n = -1; if (nparams == 1) n = luaH_getn(t); else if (nparams == 3 && ttisnumber(args) && ttisnumber(args + 1) && nvalue(args) == 1.0) n = int(nvalue(args + 1)); if (n >= 0 && n <= t->sizearray && cast_int(L->stack_last - res) >= n && n + nparams <= LUAI_MAXCSTACK) { TValue* array = t->array; for (int i = 0; i < n; ++i) setobj2s(L, res + i, array + i); expandstacklimit(L, res + n); return n; } } return -1; } static int luauF_vector(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args)) { LUA_VECTOR_TYPE x = LUA_VECTOR_TYPE(nvalue(arg0)); LUA_VECTOR_TYPE y = LUA_VECTOR_TYPE(nvalue(args)); LUA_VECTOR_TYPE z = LUA_VECTOR_TYPE(0.0); if (nparams >= 3) { if (!ttisnumber(args + 1)) return -1; z = LUA_VECTOR_TYPE(nvalue(args + 1)); } #if LUA_VECTOR_SIZE == 4 LUA_VECTOR_TYPE w = LUA_VECTOR_TYPE(0.0); if (nparams >= 4) { if (!ttisnumber(args + 2)) return -1; w = LUA_VECTOR_TYPE(nvalue(args + 2)); } setvvalue(L, res, x, y, z, w); #else setvvalue(L, res, x, y, z, 0.0); #endif return 1; } return -1; } static int luauF_countlz(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); unsigned n; luai_num2unsigned(n, a1); #ifdef _MSC_VER unsigned long rl; int r = _BitScanReverse(&rl, n) ? 31 - int(rl) : 32; #else int r = n == 0 ? 32 : __builtin_clz(n); #endif setnvalue(res, double(r)); return 1; } return -1; } static int luauF_countrz(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); unsigned n; luai_num2unsigned(n, a1); #ifdef _MSC_VER unsigned long rl; int r = _BitScanForward(&rl, n) ? int(rl) : 32; #else int r = n == 0 ? 32 : __builtin_ctz(n); #endif setnvalue(res, double(r)); return 1; } return -1; } static int luauF_select(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams == 1 && nresults == 1) { int n = cast_int(L->base - L->ci->func) - L->ci->p->numparams - 1; if (ttisnumber(arg0)) { int i = int(nvalue(arg0)); // i >= 1 && i <= n if (unsigned(i - 1) < unsigned(n)) { setobj2s(L, res, L->base - n + (i - 1)); return 1; } // note: for now we don't handle negative case (wrap around) and defer to fallback } else if (ttisstring(arg0) && *svalue(arg0) == '#') { setnvalue(res, double(n)); return 1; } } return -1; } static int luauF_rawlen(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1) { if (ttistable(arg0)) { LuaTable* h = hvalue(arg0); setnvalue(res, double(luaH_getn(h))); return 1; } else if (ttisstring(arg0)) { TString* ts = tsvalue(arg0); setnvalue(res, double(ts->len)); return 1; } } return -1; } static int luauF_extractk(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { // args is known to contain a number constant with packed in-range f/w if (nparams >= 2 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); double a2 = nvalue(args); unsigned n; luai_num2unsigned(n, a1); int fw = int(a2); int f = fw & 31; int w1 = fw >> 5; uint32_t m = ~(0xfffffffeu << w1); uint32_t r = (n >> f) & m; setnvalue(res, double(r)); return 1; } return -1; } static int luauF_getmetatable(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1) { LuaTable* mt = NULL; if (ttistable(arg0)) mt = hvalue(arg0)->metatable; else if (ttisuserdata(arg0)) mt = uvalue(arg0)->metatable; else mt = L->global->mt[ttype(arg0)]; const TValue* mtv = mt ? luaH_getstr(mt, L->global->tmname[TM_METATABLE]) : luaO_nilobject; if (!ttisnil(mtv)) { setobj2s(L, res, mtv); return 1; } if (mt) { sethvalue(L, res, mt); return 1; } else { setnilvalue(res); return 1; } } return -1; } static int luauF_setmetatable(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { // note: setmetatable(_, nil) is rare so we use fallback for it to optimize the fast path if (nparams >= 2 && nresults <= 1 && ttistable(arg0) && ttistable(args)) { LuaTable* t = hvalue(arg0); if (t->readonly || t->metatable != NULL) return -1; // note: overwriting non-null metatable is very rare but it requires __metatable check LuaTable* mt = hvalue(args); t->metatable = mt; luaC_objbarrier(L, t, mt); sethvalue(L, res, t); return 1; } return -1; } static int luauF_tonumber(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams == 1 && nresults <= 1) { double num; if (ttisnumber(arg0)) { setnvalue(res, nvalue(arg0)); return 1; } else if (ttisstring(arg0) && luaO_str2d(svalue(arg0), &num)) { setnvalue(res, num); return 1; } else { setnilvalue(res); return 1; } } return -1; } static int luauF_tostring(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1) { switch (ttype(arg0)) { case LUA_TNIL: { TString* s = L->global->ttname[LUA_TNIL]; setsvalue(L, res, s); return 1; } case LUA_TBOOLEAN: { TString* s = bvalue(arg0) ? luaS_newliteral(L, "true") : luaS_newliteral(L, "false"); setsvalue(L, res, s); return 1; } case LUA_TNUMBER: { if (luaC_needsGC(L)) return -1; // we can't call luaC_checkGC so fall back to C implementation char s[LUAI_MAXNUM2STR]; char* e = luai_num2str(s, nvalue(arg0)); setsvalue(L, res, luaS_newlstr(L, s, e - s)); return 1; } case LUA_TSTRING: { setsvalue(L, res, tsvalue(arg0)); return 1; } case LUA_TINTEGER: { if (luaC_needsGC(L)) return -1; // we can't call luaC_checkGC so fall back to C implementation char s[LUAI_MAXINT2STR]; char* e = luai_int2str(s, lvalue(arg0)); setsvalue(L, res, luaS_newlstr(L, s, e - s)); return 1; } } // fall back to generic C implementation } return -1; } static int luauF_byteswap(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); unsigned n; luai_num2unsigned(n, a1); n = (n << 24) | ((n << 8) & 0xff0000) | ((n >> 8) & 0xff00) | (n >> 24); setnvalue(res, double(n)); return 1; } return -1; } // because offset is limited to an integer, a single 64bit comparison can be used and will not overflow #define checkoutofbounds(offset, len, accessize) (uint64_t(unsigned(offset)) + (accessize - 1) >= uint64_t(len)) template static int luauF_readinteger(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 2 && nresults <= 1 && ttisbuffer(arg0) && ttisnumber(args)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(T))) return -1; T val; memcpy(&val, (char*)bufvalue(arg0)->data + unsigned(offset), sizeof(T)); setnvalue(res, double(val)); return 1; } #endif return -1; } template static int luauF_writeinteger(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 3 && nresults <= 0 && ttisbuffer(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(T))) return -1; unsigned value; double incoming = nvalue(args + 1); luai_num2unsigned(value, incoming); T val = T(value); memcpy((char*)bufvalue(arg0)->data + unsigned(offset), &val, sizeof(T)); return 0; } #endif return -1; } template static int luauF_readfp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 2 && nresults <= 1 && ttisbuffer(arg0) && ttisnumber(args)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(T))) return -1; T val; #ifdef _MSC_VER // avoid memcpy path on MSVC because it results in integer stack copy + floating-point ops on stack val = *(T*)((char*)bufvalue(arg0)->data + unsigned(offset)); #else memcpy(&val, (char*)bufvalue(arg0)->data + unsigned(offset), sizeof(T)); #endif setnvalue(res, double(val)); return 1; } #endif return -1; } template static int luauF_writefp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 3 && nresults <= 0 && ttisbuffer(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(T))) return -1; T val = T(nvalue(args + 1)); #ifdef _MSC_VER // avoid memcpy path on MSVC because it results in integer stack copy + floating-point ops on stack *(T*)((char*)bufvalue(arg0)->data + unsigned(offset)) = val; #else memcpy((char*)bufvalue(arg0)->data + unsigned(offset), &val, sizeof(T)); #endif return 0; } #endif return -1; } static int luauF_vectormagnitude(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 setnvalue(res, luai_sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2] + v[3] * v[3])); #else setnvalue(res, luai_sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2])); #endif return 1; } return -1; } static int luauF_vectornormalize(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 LUA_VECTOR_TYPE invSqrt = LUA_VECTOR_TYPE(1.0) / luai_sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2] + v[3] * v[3]); setvvalue(L, res, v[0] * invSqrt, v[1] * invSqrt, v[2] * invSqrt, v[3] * invSqrt); #else LUA_VECTOR_TYPE invSqrt = LUA_VECTOR_TYPE(1.0) / luai_sqrt(v[0] * v[0] + v[1] * v[1] + v[2] * v[2]); setvvalue(L, res, v[0] * invSqrt, v[1] * invSqrt, v[2] * invSqrt, 0.0); #endif return 1; } return -1; } static int luauF_vectorcross(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisvector(arg0) && ttisvector(args)) { const LUA_VECTOR_TYPE* a = vvalue(arg0); const LUA_VECTOR_TYPE* b = vvalue(args); // same for 3- and 4- wide vectors setvvalue(L, res, a[1] * b[2] - a[2] * b[1], a[2] * b[0] - a[0] * b[2], a[0] * b[1] - a[1] * b[0], 0.0); return 1; } return -1; } static int luauF_vectordot(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisvector(arg0) && ttisvector(args)) { const LUA_VECTOR_TYPE* a = vvalue(arg0); const LUA_VECTOR_TYPE* b = vvalue(args); #if LUA_VECTOR_SIZE == 4 setnvalue(res, a[0] * b[0] + a[1] * b[1] + a[2] * b[2] + a[3] * b[3]); #else setnvalue(res, a[0] * b[0] + a[1] * b[1] + a[2] * b[2]); #endif return 1; } return -1; } static int luauF_vectorfloor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 setvvalue(L, res, luai_floor(v[0]), luai_floor(v[1]), luai_floor(v[2]), luai_floor(v[3])); #else setvvalue(L, res, luai_floor(v[0]), luai_floor(v[1]), luai_floor(v[2]), 0.0); #endif return 1; } return -1; } static int luauF_vectorceil(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 setvvalue(L, res, luai_ceil(v[0]), luai_ceil(v[1]), luai_ceil(v[2]), luai_ceil(v[3])); #else setvvalue(L, res, luai_ceil(v[0]), luai_ceil(v[1]), luai_ceil(v[2]), 0.0); #endif return 1; } return -1; } static int luauF_vectorabs(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 setvvalue(L, res, luai_fabs(v[0]), luai_fabs(v[1]), luai_fabs(v[2]), luai_fabs(v[3])); #else setvvalue(L, res, luai_fabs(v[0]), luai_fabs(v[1]), luai_fabs(v[2]), 0.0); #endif return 1; } return -1; } static int luauF_vectorsign(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisvector(arg0)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); #if LUA_VECTOR_SIZE == 4 setvvalue(L, res, luai_sign(v[0]), luai_sign(v[1]), luai_sign(v[2]), luai_sign(v[3])); #else setvvalue(L, res, luai_sign(v[0]), luai_sign(v[1]), luai_sign(v[2]), 0.0); #endif return 1; } return -1; } static int luauF_vectorclamp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisvector(arg0) && ttisvector(args) && ttisvector(args + 1)) { const LUA_VECTOR_TYPE* v = vvalue(arg0); const LUA_VECTOR_TYPE* min = vvalue(args); const LUA_VECTOR_TYPE* max = vvalue(args + 1); if (min[0] <= max[0] && min[1] <= max[1] && min[2] <= max[2]) { #if LUA_VECTOR_SIZE == 4 setvvalue( L, res, luai_clamp(v[0], min[0], max[0]), luai_clamp(v[1], min[1], max[1]), luai_clamp(v[2], min[2], max[2]), luai_clamp(v[3], min[3], max[3]) ); #else setvvalue(L, res, luai_clamp(v[0], min[0], max[0]), luai_clamp(v[1], min[1], max[1]), luai_clamp(v[2], min[2], max[2]), 0.0); #endif return 1; } } return -1; } static int luauF_vectormin(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisvector(arg0) && ttisvector(args)) { const LUA_VECTOR_TYPE* a = vvalue(arg0); const LUA_VECTOR_TYPE* b = vvalue(args); LUA_VECTOR_TYPE result[4]; result[0] = (b[0] < a[0]) ? b[0] : a[0]; result[1] = (b[1] < a[1]) ? b[1] : a[1]; result[2] = (b[2] < a[2]) ? b[2] : a[2]; #if LUA_VECTOR_SIZE == 4 result[3] = (b[3] < a[3]) ? b[3] : a[3]; #else result[3] = 0.0; #endif for (int i = 3; i <= nparams; ++i) { if (!ttisvector(args + (i - 2))) return -1; const LUA_VECTOR_TYPE* c = vvalue(args + (i - 2)); result[0] = (c[0] < result[0]) ? c[0] : result[0]; result[1] = (c[1] < result[1]) ? c[1] : result[1]; result[2] = (c[2] < result[2]) ? c[2] : result[2]; #if LUA_VECTOR_SIZE == 4 result[3] = (c[3] < result[3]) ? c[3] : result[3]; #endif } setvvalue(L, res, result[0], result[1], result[2], result[3]); return 1; } return -1; } static int luauF_vectormax(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisvector(arg0) && ttisvector(args)) { const LUA_VECTOR_TYPE* a = vvalue(arg0); const LUA_VECTOR_TYPE* b = vvalue(args); LUA_VECTOR_TYPE result[4]; result[0] = (b[0] > a[0]) ? b[0] : a[0]; result[1] = (b[1] > a[1]) ? b[1] : a[1]; result[2] = (b[2] > a[2]) ? b[2] : a[2]; #if LUA_VECTOR_SIZE == 4 result[3] = (b[3] > a[3]) ? b[3] : a[3]; #else result[3] = 0.0; #endif for (int i = 3; i <= nparams; ++i) { if (!ttisvector(args + (i - 2))) return -1; const LUA_VECTOR_TYPE* c = vvalue(args + (i - 2)); result[0] = (c[0] > result[0]) ? c[0] : result[0]; result[1] = (c[1] > result[1]) ? c[1] : result[1]; result[2] = (c[2] > result[2]) ? c[2] : result[2]; #if LUA_VECTOR_SIZE == 4 result[3] = (c[3] > result[3]) ? c[3] : result[3]; #endif } setvvalue(L, res, result[0], result[1], result[2], result[3]); return 1; } return -1; } static int luauF_vectorlerp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisvector(arg0) && ttisvector(args) && ttisnumber(args + 1)) { const LUA_VECTOR_TYPE* a = vvalue(arg0); const LUA_VECTOR_TYPE* b = vvalue(args); const LUA_VECTOR_TYPE t = LUA_VECTOR_TYPE(nvalue(args + 1)); #if LUA_VECTOR_SIZE == 4 setvvalue(L, res, luai_lerp(a[0], b[0], t), luai_lerp(a[1], b[1], t), luai_lerp(a[2], b[2], t), luai_lerp(a[3], b[3], t)); #else setvvalue(L, res, luai_lerp(a[0], b[0], t), luai_lerp(a[1], b[1], t), luai_lerp(a[2], b[2], t), 0.0); #endif return 1; } return -1; } static int luauF_lerp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisnumber(arg0) && ttisnumber(args) && ttisnumber(args + 1)) { double a = nvalue(arg0); double b = nvalue(args); double t = nvalue(args + 1); double r = (t == 1.0) ? b : a + (b - a) * t; setnvalue(res, r); return 1; } return -1; } static int luauF_isnan(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double x = nvalue(arg0); setbvalue(res, isnan(x)); return 1; } return -1; } static int luauF_isinf(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double x = nvalue(arg0); setbvalue(res, isinf(x)); return 1; } return -1; } static int luauF_isfinite(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double x = nvalue(arg0); setbvalue(res, isfinite(x)); return 1; } return -1; } static int luauF_integertonumber(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { setnvalue(res, cast_num(lvalue(arg0))); return 1; } return -1; } static int luauF_integeradd(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); setlvalue(res, (int64_t)((uint64_t)a1 + (uint64_t)a2)); return 1; } return -1; } static int luauF_integersub(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); setlvalue(res, (int64_t)((uint64_t)a1 - (uint64_t)a2)); return 1; } return -1; } static int luauF_integerneg(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { setlvalue(res, (int64_t)(~(uint64_t)lvalue(arg0) + 1)); return 1; } return -1; } static int luauF_integerdiv(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); if ((b == 0) || ((a == LLONG_MIN) && (b == -1))) return -1; setlvalue(res, a / b); return 1; } return -1; } static int luauF_integerudiv(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); if (b == 0) return -1; setlvalue(res, a / b); return 1; } return -1; } static int luauF_integerband(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t r = (uint64_t)lvalue(arg0); for (int i = 2; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; r &= (uint64_t)lvalue(args + (i - 2)); } setlvalue(res, r); return 1; } return -1; } static int luauF_integerbor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t r = (uint64_t)lvalue(arg0); for (int i = 2; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; r |= (uint64_t)lvalue(args + (i - 2)); } setlvalue(res, r); return 1; } return -1; } static int luauF_integerbxor(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t r = (uint64_t)lvalue(arg0); for (int i = 2; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; r ^= (uint64_t)lvalue(args + (i - 2)); } setlvalue(res, r); return 1; } return -1; } static int luauF_integerbnot(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { setlvalue(res, ~(uint64_t)lvalue(arg0)); return 1; } return -1; } static int luauF_integerbswap(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t a = (uint64_t)lvalue(arg0); setlvalue( res, (a >> 56) | ((a & 0xFF000000000000) >> 40) | ((a & 0xFF0000000000) >> 24) | ((a & 0xFF00000000) >> 8) | ((a & 0xFF000000) << 8) | ((a & 0xFF0000) << 24) | ((a & 0xFF00) << 40) | ((a & 0xFF) << 56) ); return 1; } return -1; } static int luauF_integerbtest(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t r = (uint64_t)lvalue(arg0); for (int i = 2; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; r &= (uint64_t)lvalue(args + (i - 2)); } setbvalue(res, (r != 0)); return 1; } return -1; } static int luauF_integerlt(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); setbvalue(res, a < b); return 1; } return -1; } static int luauF_integerle(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); setbvalue(res, a <= b); return 1; } return -1; } static int luauF_integergt(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); setbvalue(res, a > b); return 1; } return -1; } static int luauF_integerge(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); setbvalue(res, a >= b); return 1; } return -1; } static int luauF_integerult(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); setbvalue(res, a < b); return 1; } return -1; } static int luauF_integerule(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); setbvalue(res, a <= b); return 1; } return -1; } static int luauF_integerugt(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); setbvalue(res, a > b); return 1; } return -1; } static int luauF_integeruge(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); setbvalue(res, a >= b); return 1; } return -1; } static int luauF_integerurem(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t a = (uint64_t)lvalue(arg0); uint64_t b = (uint64_t)lvalue(args); if (b == 0) return -1; setlvalue(res, a % b); return 1; } return -1; } static int luauF_integerrem(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a = lvalue(arg0); int64_t b = lvalue(args); if (b == 0) return -1; setlvalue(res, ((a == LLONG_MIN) && (b == -1)) ? 0 : (a % b)); return 1; } return -1; } static int luauF_integercountlz(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t n = (uint64_t)lvalue(arg0); int result; #ifdef _MSC_VER #ifdef _WIN64 unsigned long rl; result = _BitScanReverse64(&rl, n) ? 63 - int(rl) : 64; #else unsigned long rl; if (_BitScanReverse(&rl, uint32_t(n >> 32))) result = 31 - int(rl); else result = _BitScanReverse(&rl, uint32_t(n)) ? 63 - int(rl) : 64; #endif #else result = (n == 0) ? 64 : __builtin_clzll(n); #endif setlvalue(res, result); return 1; } return -1; } static int luauF_integercountrz(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisinteger(arg0)) { uint64_t n = (uint64_t)lvalue(arg0); int result; #ifdef _MSC_VER #ifdef _WIN64 unsigned long rl; result = _BitScanForward64(&rl, n) ? int(rl) : 64; #else unsigned long rl; if (_BitScanForward(&rl, uint32_t(n))) result = int(rl); else result = _BitScanForward(&rl, uint32_t(n >> 32)) ? int(rl) + 32 : 64; #endif #else result = (n == 0) ? 64 : __builtin_ctzll(n); #endif setlvalue(res, result); return 1; } return -1; } static int luauF_integerextract(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if ((nparams >= 3) && !ttisinteger(args + 1)) return -1; if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t n = lvalue(arg0); int64_t f = lvalue(args); int64_t w = (nparams >= 3) ? lvalue(args + 1) : 1; if ((f < 0) || (f > 63) || (w < 1) || (w > 64) || ((f + w) > 64)) return -1; setlvalue(res, (((uint64_t)n) >> f) & ((0xFFFFFFFFFFFFFFFFULL) >> (64 - w))); return 1; } return -1; } static int luauF_integerclamp(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 3 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args) && ttisinteger(args + 1)) { int64_t a = lvalue(arg0); int64_t rmin = lvalue(args); int64_t rmax = lvalue(args + 1); if (rmin > rmax) return -1; setlvalue(res, (a < rmin) ? rmin : ((a > rmax) ? rmax : a)); return 1; } return -1; } static int luauF_integerlrotate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t n = (uint64_t)lvalue(arg0); unsigned s = (unsigned)((uint64_t)lvalue(args) % 64); setlvalue(res, s != 0 ? (n << s) | (n >> (64 - s)) : n); return 1; } return -1; } static int luauF_integerrrotate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t n = (uint64_t)lvalue(arg0); unsigned s = (unsigned)((uint64_t)lvalue(args) % 64); setlvalue(res, s != 0 ? (n >> s) | (n << (64 - s)) : n); return 1; } return -1; } static int luauF_integerlshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t n = (uint64_t)lvalue(arg0); int64_t i = lvalue(args); setlvalue(res, ((i >= -63) && (i <= 63)) ? ((i < 0) ? (n >> (-i)) : (n << i)) : 0); return 1; } return -1; } static int luauF_integerarshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t n = lvalue(arg0); int64_t i = lvalue(args); if ((i >= -63) && (i <= 63)) { setlvalue(res, (i < 0) ? (int64_t)((uint64_t)n << (-i)) : (n >> i)); } else if (i < -63) { setlvalue(res, 0); } else { setlvalue(res, (n < 0) ? -1 : 0); } return 1; } return -1; } static int luauF_integerrshift(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { uint64_t n = (uint64_t)lvalue(arg0); int64_t i = lvalue(args); setlvalue(res, ((i >= -63) && (i <= 63)) ? ((i < 0) ? (n << (-i)) : (n >> i)) : 0); return 1; } return -1; } static int luauF_integermin(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); int64_t r = (a2 < a1) ? a2 : a1; for (int i = 3; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; int64_t a = lvalue(args + (i - 2)); r = (a < r) ? a : r; } setlvalue(res, r); return 1; } return -1; } static int luauF_integermax(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); int64_t r = (a2 < a1) ? a1 : a2; for (int i = 3; i <= nparams; ++i) { if (!ttisinteger(args + (i - 2))) return -1; int64_t a = lvalue(args + (i - 2)); r = (a > r) ? a : r; } setlvalue(res, r); return 1; } return -1; } static int luauF_integermul(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); setlvalue(res, (int64_t)((uint64_t)a1 * (uint64_t)a2)); return 1; } return -1; } static int luauF_integermod(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); if (a2 == 0) return -1; if ((a1 == LLONG_MIN) && (a2 == -1)) { setlvalue(res, 0); return 1; } int64_t remainder = a1 % a2; if (remainder && ((a1 < 0) != (a2 < 0))) remainder += a2; setlvalue(res, remainder); return 1; } return -1; } static int luauF_integeridiv(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 2 && nresults <= 1 && ttisinteger(arg0) && ttisinteger(args)) { int64_t a1 = lvalue(arg0); int64_t a2 = lvalue(args); if (a2 == 0) return -1; if ((a1 == LLONG_MIN) && (a2 == -1)) return -1; int64_t result = a1 / a2; // Floored division rounds toward -inf: adjust the truncated quotient down by 1 // when the operands have opposite signs and the division is inexact. if (((a1 ^ a2) < 0) && (a1 % a2)) result -= 1; setlvalue(res, result); return 1; } return -1; } static int luauF_integercreate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); if (a1 >= -9223372036854775808.0 && a1 < 9223372036854775808.0) { int64_t x = (int64_t)a1; if ((double)x == a1) { setlvalue(res, x); return 1; } } setnilvalue(res); return 1; } return -1; } static int luauF_bufferreadlong(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 2 && nresults <= 1 && ttisbuffer(arg0) && ttisnumber(args)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(int64_t))) return -1; int64_t val; memcpy(&val, (char*)bufvalue(arg0)->data + unsigned(offset), sizeof(int64_t)); setlvalue(res, val); return 1; } #endif return -1; } static int luauF_bufferwritelong(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { #if !defined(LUAU_BIG_ENDIAN) if (nparams >= 3 && nresults <= 0 && ttisbuffer(arg0) && ttisnumber(args) && ttisinteger(args + 1)) { int offset; luai_num2int(offset, nvalue(args)); if (checkoutofbounds(offset, bufvalue(arg0)->len, sizeof(int64_t))) return -1; int64_t val = lvalue(args + 1); memcpy((char*)bufvalue(arg0)->data + unsigned(offset), &val, sizeof(int64_t)); return 0; } #endif return -1; } static int luauF_missing(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { return -1; } #ifdef LUAU_TARGET_SSE41 template LUAU_TARGET_SSE41 inline double roundsd_sse41(double v) { __m128d av = _mm_set_sd(v); __m128d rv = _mm_round_sd(av, av, Rounding | _MM_FROUND_NO_EXC); return _mm_cvtsd_f64(rv); } LUAU_TARGET_SSE41 static int luauF_floor_sse41(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, roundsd_sse41<_MM_FROUND_TO_NEG_INF>(a1)); return 1; } return -1; } LUAU_TARGET_SSE41 static int luauF_ceil_sse41(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); setnvalue(res, roundsd_sse41<_MM_FROUND_TO_POS_INF>(a1)); return 1; } return -1; } LUAU_TARGET_SSE41 static int luauF_round_sse41(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double a1 = nvalue(arg0); // roundsd only supports bankers rounding natively, so we need to emulate rounding by using truncation // offset is prevfloat(0.5), which is important so that we round prevfloat(0.5) to 0. const double offset = 0.49999999999999994; __m128d va1 = _mm_set_sd(a1); __m128d sign = _mm_and_pd(va1, _mm_set_sd(-0.0)); __m128d off = _mm_or_pd(_mm_set_sd(offset), sign); __m128d sum = _mm_add_sd(va1, off); __m128d result = _mm_round_sd(sum, sum, _MM_FROUND_TO_ZERO | _MM_FROUND_NO_EXC); setnvalue(res, _mm_cvtsd_f64(result)); return 1; } return -1; } static bool luau_hassse41() { int cpuinfo[4] = {}; #ifdef _MSC_VER __cpuid(cpuinfo, 1); #else __cpuid(1, cpuinfo[0], cpuinfo[1], cpuinfo[2], cpuinfo[3]); #endif // We require SSE4.1 support for ROUNDSD // https://en.wikipedia.org/wiki/CPUID#EAX=1:_Processor_Info_and_Feature_Bits return (cpuinfo[2] & (1 << 19)) != 0; } #endif static int luauF_tablecreate(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 1 && ttisnumber(arg0)) { double n = nvalue(arg0); int size; luai_num2int(size, n); if (size < 0 || luaC_needsGC(L)) return -1; LuaTable* t = luaH_new(L, size, 0); if (nparams >= 2) { TValue* v = args; for (int i = 0; i < size; ++i) setobj2t(L, &t->array[i], v); } sethvalue(L, res, t); return 1; } return -1; } static int luauF_tableclear(lua_State* L, StkId res, TValue* arg0, int nresults, StkId args, int nparams) { if (nparams >= 1 && nresults <= 0 && ttistable(arg0)) { LuaTable* t = hvalue(arg0); if (t->readonly) return -1; luaH_clear(t); return 0; } return -1; } const luau_FastFunction luauF_table[256] = { NULL, luauF_assert, luauF_abs, luauF_acos, luauF_asin, luauF_atan2, luauF_atan, #ifdef LUAU_TARGET_SSE41 luau_hassse41() ? luauF_ceil_sse41 : luauF_ceil, #else luauF_ceil, #endif luauF_cosh, luauF_cos, luauF_deg, luauF_exp, #ifdef LUAU_TARGET_SSE41 luau_hassse41() ? luauF_floor_sse41 : luauF_floor, #else luauF_floor, #endif luauF_fmod, luauF_frexp, luauF_ldexp, luauF_log10, luauF_log, luauF_max, luauF_min, luauF_modf, luauF_pow, luauF_rad, luauF_sinh, luauF_sin, luauF_sqrt, luauF_tanh, luauF_tan, luauF_arshift, luauF_band, luauF_bnot, luauF_bor, luauF_bxor, luauF_btest, luauF_extract, luauF_lrotate, luauF_lshift, luauF_replace, luauF_rrotate, luauF_rshift, luauF_type, luauF_byte, luauF_char, luauF_len, luauF_typeof, luauF_sub, luauF_clamp, luauF_sign, #ifdef LUAU_TARGET_SSE41 luau_hassse41() ? luauF_round_sse41 : luauF_round, #else luauF_round, #endif luauF_rawset, luauF_rawget, luauF_rawequal, luauF_tinsert, luauF_tunpack, luauF_vector, luauF_countlz, luauF_countrz, luauF_select, luauF_rawlen, luauF_extractk, luauF_getmetatable, luauF_setmetatable, luauF_tonumber, luauF_tostring, luauF_byteswap, luauF_readinteger, luauF_readinteger, luauF_writeinteger, luauF_readinteger, luauF_readinteger, luauF_writeinteger, luauF_readinteger, luauF_readinteger, luauF_writeinteger, luauF_readfp, luauF_writefp, luauF_readfp, luauF_writefp, luauF_vectormagnitude, luauF_vectornormalize, luauF_vectorcross, luauF_vectordot, luauF_vectorfloor, luauF_vectorceil, luauF_vectorabs, luauF_vectorsign, luauF_vectorclamp, luauF_vectormin, luauF_vectormax, luauF_lerp, luauF_vectorlerp, luauF_isnan, luauF_isinf, luauF_isfinite, luauF_integercreate, luauF_integertonumber, luauF_integerneg, luauF_integeradd, luauF_integersub, luauF_integermul, luauF_integerdiv, luauF_integermin, luauF_integermax, luauF_integerrem, luauF_integeridiv, luauF_integerudiv, luauF_integerurem, luauF_integermod, luauF_integerclamp, luauF_integerband, luauF_integerbor, luauF_integerbnot, luauF_integerbxor, luauF_integerlt, luauF_integerle, luauF_integerult, luauF_integerule, luauF_integergt, luauF_integerge, luauF_integerugt, luauF_integeruge, luauF_integerlshift, luauF_integerrshift, luauF_integerarshift, luauF_integerlrotate, luauF_integerrrotate, luauF_integerextract, luauF_integerbtest, luauF_integercountrz, luauF_integercountlz, luauF_integerbswap, luauF_bufferreadlong, luauF_bufferwritelong, luauF_tablecreate, luauF_tableclear, // When adding builtins, add them above this line; what follows is 64 "dummy" entries with luauF_missing fallback. // This is important so that older versions of the runtime that don't support newer builtins automatically fall back via luauF_missing. // Given the builtin addition velocity this should always provide a larger compatibility window than bytecode versions suggest. #define MISSING8 luauF_missing, luauF_missing, luauF_missing, luauF_missing, luauF_missing, luauF_missing, luauF_missing, luauF_missing MISSING8, MISSING8, MISSING8, MISSING8, MISSING8, MISSING8, MISSING8, MISSING8, #undef MISSING8 }; static int luauPF_pcall(lua_State* L, StkId ra, int nparams) { if (nparams < 1 || !ttisfunction(ra + 1) || L->global->builtinPcall == nullptr) return -1; setclvalue(L, ra, L->global->builtinPcall); return 0; } static int luauPF_xpcall(lua_State* L, StkId ra, int nparams) { if (nparams < 2 || !ttisfunction(ra + 1) || !ttisfunction(ra + 2) || L->global->builtinXpcall == nullptr) return -1; setclvalue(L, ra, L->global->builtinXpcall); // swap 'f' and 'errf' so that we get 'errf, f, arguments' prepared for calling 'f' TValue tmp; setobj(L, &tmp, ra + 1); setobj2s(L, ra + 1, ra + 2); setobj2s(L, ra + 2, &tmp); return 1; } const luau_ProtectedFastFunction luauPF_table[2] = {luauPF_pcall, luauPF_xpcall};