q98/LuaJIT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add LUA_INIT

Browse Source
This commit is contained in:
fesily 2023-08-25 09:29:28 +08:00
parent e1136896aa
commit 24e8adf315
2 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CMakeLists.txt
View File

@ -358,6 +358,7 @@ ELSE()
PREFIX "lib" IMPORT_PREFIX "lib" OUTPUT_NAME "luajit") PREFIX "lib" IMPORT_PREFIX "lib" OUTPUT_NAME "luajit")
ENDIF() ENDIF()
target_compile_definitions(luajit-5.1 PRIVATE DO_LUA_INIT)
target_link_libraries (luajit-5.1 ${LIBS} ) target_link_libraries (luajit-5.1 ${LIBS} )
IF(WIN32) IF(WIN32)

18
src/lib_init.c
View File

@ -15,6 +15,8 @@
#include "lj_arch.h" #include "lj_arch.h"
#include <stdlib.h>
static const luaL_Reg lj_lib_load[] = { static const luaL_Reg lj_lib_load[] = {
{ "", luaopen_base }, { "", luaopen_base },
{ LUA_LOADLIBNAME, luaopen_package }, { LUA_LOADLIBNAME, luaopen_package },
@ -77,6 +79,22 @@ LUALIB_API void luaL_openlibs(lua_State *L)
if (luaL_loadstring(L, dump_fix) == 0) if (luaL_loadstring(L, dump_fix) == 0)
lua_pcall(L, 0, 0, 0); lua_pcall(L, 0, 0, 0);
#endif #endif
#ifdef DO_LUA_INIT
void handle_luainit(lua_State *L);
handle_luainit(L);
#endif
#endif #endif
} }
static void handle_luainit(lua_State *L)
{
const char *init = getenv(LUA_INIT);
if (init == NULL)
return LUA_OK;
else if (init[0] == '@')
return luaL_dofile(L, init+1);
else
return luaL_dostring(L, init);
}