mirror of
https://github.com/d0k3/GodMode9.git
synced 2025-06-25 21:22:47 +00:00
* Test implementation of lua * Trust that lua knows what its doing with this Silence warnings * actually update top screen when Thingy is called, disable unnecessary ShowPrompt calls * readme * change init for a simple test, print error on top screen too * Readme * change init for a simple test, print error on top screen too * enable more lua libs, edit init.lua with string examples * one more readme edit before bed * Readme * change init for a simple test, print error on top screen too * enable more lua libs, edit init.lua with string examples * make lua a proper file type, add test UI library with two functions, remove luacmd command * remove old attempts at editing lauxlib and liolib * README * README * FS lib, new UI stuff * consistency with "type* ptr" maybe * add custom package searcher, reset package.path * new functions for UI including basic print output buffer, add "Lua scripts..." option to home/power menu * build vram0.tar including subdirs of data * move default path to GM9LUA_DEFAULT_PATH * FS_FileGetData * testfgd, add GM9VERSION global, update README, fix indentation * FS_FileGetData will return a nil instead if it fails * it's actually luaL_pushfail * os * use luaL_tolstring instead of lua_tostring * fix test/remove debugging showprompt * os.clock float attempt * fix print for real * fix swapped offset and size for FileGetData * finish OS stuff * fix os.clock * shorten table in/out * remove .vscode dir * enum test * support building without lua * NO_LUA hides menu options (except when you directly select a lua file) * update UI lib to better match the ideas on #1 * dockermake * add DrawPNG function * whoops its ShowPNG * minor fixes, add DrawPNG * fix AskPrompt, add all showprompts mentioned in #1 * add newly added functions to readme * try to keep separate code and data * update lua to 5.4.7 * remove test libraries now that i want to attempt to implement in a real api * add nix flake for building * add various lua functions, some of them taken from the old attempt but with new names * add dev shell to flake.nix * remove test lua scripts in root * add test lua scripts in data/luascripts * add a whole bunch more lua functions and stuff * add more test lua scripts * add more functions, add preload script, add test io compatibility module * add more functions and test scripts * more functions and stuff * more functions and stuff * more functions and stuff, plus a wip ctrcheck reimpl * yet more functions and stuff * even more functions and stuff * command comparison table.ods * update command comparison table.ods * update command comparison table.ods * Add files via upload * update command comparison table.ods * update ui.show_text to use DrawStringCenter, update ctrcheck rewrite * Split up the ARM9 code (.text, .vectors) and data (.rodata, .data, .bss) sections into their own ELFs. This allows us to use more ARM9 WRAM while leaving the 128k BootROM mirror intact. * use the makefile definition * add title module, move around some functions, update command comparison table * add readme for lua * remove liolib.c and loslib.c * more functions and things, use CheckWritePermissionsLuaError in place of more manual checks, update command comparison table * add missing constant * set CURRDIR to nil instead of "(null)" if not found * remove gm9enum (unused since the restart) * add ui.check_key * split fs module to lua overlay and _fs internal module, and add a check for fs.write_file in lua * add fs.ask_select_file and fs.ask_select_dir * add fs.key_dump, replace overwrite_all and append_all with overwrite and append * add fs.cart_dump and sys.emu_base * add ctrtool, update flake.lock * add io append mode * make sure io.open with write mode starts with an empty file, add os.remove and os.rename aliases * properly implement os.remove compatibility * add fs.verify_with_sha_file, fix PathIsDirectory by using stat instead of opendir * add util.running_as_module (untested) * move scripts over to https://github.com/ihaveamac/GM9-lua-script-experiments * remove ods and dockermake.sh * remove data/scripts * add lua autorun (untested) * fix syntax error * add sys.check_embedded_backup * remove accidental symlink * add sys.check_raw_rtc * fix ui.show_file_text_viewer not freeing memory or reporting an error if OOM happens * add todo notes for ui * work-in-progress lua doc * formatting fix * up heading level for all sections * Revert "up heading level for all sections" This reverts commit 6ef14b619536b4253e341ba40b4dea728358979d. * separators * fix name and error for fs.move * do explicit permission checks in fs.move * fix error string for fs.copy * fix function name for fs.dir_info * fix error string for fs.find and fs.find_not * fix function name for fs.img_umount * partial fs doc * finished fs doc, string fixes for fs module * document fs.cart_dump encrypted opt, remove stat from fs.verify_with_sha_file * title doc * sys doc, error string updates * util doc * add json.lua to lua-doc * add fs.find_all * add 3dstool to flake * make fs.find_all recursive actually recursive, document fs.find_all * add "for" to comparison table * change fs.find to return nil if no path was found, instead of raising an error * change ui.echo to automatically word wrap (untested) * Revert "change ui.echo to automatically word wrap (untested)" This reverts commit 2524e7707708e9818162c31f9f004b6301a3061b. * switch devkitNix to upstream * flake.lock: Update Flake lock file updates: • Updated input 'devkitNix': 'github:ihaveamac/devkitNix/883d173b94e3da8dc4cc0860cdda8c36b738817c' (2024-12-05) → 'github:bandithedoge/devkitNix/95fd44f4ac7cecf24edf22daa899a516df73c6b7' (2025-01-11) • Updated input 'devkitNix/nixpkgs': 'github:NixOS/nixpkgs/566e53c2ad750c84f6d31f9ccb9d00f823165550' (2024-12-03) → 'github:NixOS/nixpkgs/4bc9c909d9ac828a039f288cf872d16d38185db8' (2025-01-08) • Updated input 'hax-nur': 'github:ihaveamac/nur-packages/c570b3830f7dd4d655afb109300529c896cd8855' (2024-12-05) → 'github:ihaveamac/nur-packages/cd49afba206c2eb10a349d92470fdf2cc942ae23' (2025-01-11) • Updated input 'hax-nur/nixpkgs': 'github:NixOS/nixpkgs/2c15aa59df0017ca140d9ba302412298ab4bf22a' (2024-12-02) → 'github:NixOS/nixpkgs/4bc9c909d9ac828a039f288cf872d16d38185db8' (2025-01-08) • Updated input 'nixpkgs': 'github:NixOS/nixpkgs/566e53c2ad750c84f6d31f9ccb9d00f823165550' (2024-12-03) → 'github:NixOS/nixpkgs/32af3611f6f05655ca166a0b1f47b57c762b5192' (2025-01-09) * flake.lock: Update Flake lock file updates: • Updated input 'devkitNix': 'github:bandithedoge/devkitNix/95fd44f4ac7cecf24edf22daa899a516df73c6b7' (2025-01-11) → 'github:bandithedoge/devkitNix/a344b0200a044f2d2ff99685f13ff7c53106428e' (2025-02-06) • Updated input 'devkitNix/nixpkgs': 'github:NixOS/nixpkgs/4bc9c909d9ac828a039f288cf872d16d38185db8' (2025-01-08) → 'github:NixOS/nixpkgs/5b2753b0356d1c951d7a3ef1d086ba5a71fff43c' (2025-02-05) • Updated input 'hax-nur': 'github:ihaveamac/nur-packages/cd49afba206c2eb10a349d92470fdf2cc942ae23' (2025-01-11) → 'github:ihaveamac/nur-packages/2ce890cab4e948109ad1ad82ba18e69240a0d352' (2025-02-06) • Updated input 'hax-nur/nixpkgs': 'github:NixOS/nixpkgs/4bc9c909d9ac828a039f288cf872d16d38185db8' (2025-01-08) → 'github:NixOS/nixpkgs/8532db2a88ba56de9188af72134d93e39fd825f3' (2025-02-02) • Added input 'hax-nur/treefmt-nix': 'github:numtide/treefmt-nix/bebf27d00f7d10ba75332a0541ac43676985dea3' (2025-01-28) • Added input 'hax-nur/treefmt-nix/nixpkgs': follows 'hax-nur/nixpkgs' • Updated input 'nixpkgs': 'github:NixOS/nixpkgs/32af3611f6f05655ca166a0b1f47b57c762b5192' (2025-01-09) → 'github:NixOS/nixpkgs/5b2753b0356d1c951d7a3ef1d086ba5a71fff43c' (2025-02-05) * make devkitNix and hax-nur inputs follow nixpkgs * Also dump section headers on .dis file * prepare for upstream merge * Restore original README. * Remove flake, was only used for my own testing * fix accidental removal of LIBS * copy lua-doc.md into release archive * README: update to mention Lua in place of GM9Script, add credits * lua-doc: fix typo * add sample HelloScript * lua-doc: remove wip notice, since all gm9script features are replicated * remove accidental inclusion of language.inl * Fix mixture of tabs and spaces * remove accidental nix leftover * re-add @ for add2tar command --------- Co-authored-by: luigoalma <luigoalma@hotmail.com> Co-authored-by: Gruetzig <florianavilov@gmail.com> Co-authored-by: Florian <88926852+Gruetzig@users.noreply.github.com> Co-authored-by: Wolfvak <soherrera1@hotmail.com>
759 lines
22 KiB
C
759 lines
22 KiB
C
/*
|
|
** $Id: loadlib.c $
|
|
** Dynamic library loader for Lua
|
|
** See Copyright Notice in lua.h
|
|
**
|
|
** This module contains an implementation of loadlib for Unix systems
|
|
** that have dlfcn, an implementation for Windows, and a stub for other
|
|
** systems.
|
|
*/
|
|
|
|
#define loadlib_c
|
|
#define LUA_LIB
|
|
|
|
#include "lprefix.h"
|
|
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "lua.h"
|
|
|
|
#include "lauxlib.h"
|
|
#include "lualib.h"
|
|
|
|
|
|
/*
|
|
** LUA_CSUBSEP is the character that replaces dots in submodule names
|
|
** when searching for a C loader.
|
|
** LUA_LSUBSEP is the character that replaces dots in submodule names
|
|
** when searching for a Lua loader.
|
|
*/
|
|
#if !defined(LUA_CSUBSEP)
|
|
#define LUA_CSUBSEP LUA_DIRSEP
|
|
#endif
|
|
|
|
#if !defined(LUA_LSUBSEP)
|
|
#define LUA_LSUBSEP LUA_DIRSEP
|
|
#endif
|
|
|
|
|
|
/* prefix for open functions in C libraries */
|
|
#define LUA_POF "luaopen_"
|
|
|
|
/* separator for open functions in C libraries */
|
|
#define LUA_OFSEP "_"
|
|
|
|
|
|
/*
|
|
** key for table in the registry that keeps handles
|
|
** for all loaded C libraries
|
|
*/
|
|
static const char *const CLIBS = "_CLIBS";
|
|
|
|
#define LIB_FAIL "open"
|
|
|
|
|
|
#define setprogdir(L) ((void)0)
|
|
|
|
|
|
/*
|
|
** Special type equivalent to '(void*)' for functions in gcc
|
|
** (to suppress warnings when converting function pointers)
|
|
*/
|
|
typedef void (*voidf)(void);
|
|
|
|
|
|
/*
|
|
** system-dependent functions
|
|
*/
|
|
|
|
/*
|
|
** unload library 'lib'
|
|
*/
|
|
static void lsys_unloadlib (void *lib);
|
|
|
|
/*
|
|
** load C library in file 'path'. If 'seeglb', load with all names in
|
|
** the library global.
|
|
** Returns the library; in case of error, returns NULL plus an
|
|
** error string in the stack.
|
|
*/
|
|
static void *lsys_load (lua_State *L, const char *path, int seeglb);
|
|
|
|
/*
|
|
** Try to find a function named 'sym' in library 'lib'.
|
|
** Returns the function; in case of error, returns NULL plus an
|
|
** error string in the stack.
|
|
*/
|
|
static lua_CFunction lsys_sym (lua_State *L, void *lib, const char *sym);
|
|
|
|
|
|
|
|
|
|
#if defined(LUA_USE_DLOPEN) /* { */
|
|
/*
|
|
** {========================================================================
|
|
** This is an implementation of loadlib based on the dlfcn interface.
|
|
** The dlfcn interface is available in Linux, SunOS, Solaris, IRIX, FreeBSD,
|
|
** NetBSD, AIX 4.2, HPUX 11, and probably most other Unix flavors, at least
|
|
** as an emulation layer on top of native functions.
|
|
** =========================================================================
|
|
*/
|
|
|
|
#include <dlfcn.h>
|
|
|
|
/*
|
|
** Macro to convert pointer-to-void* to pointer-to-function. This cast
|
|
** is undefined according to ISO C, but POSIX assumes that it works.
|
|
** (The '__extension__' in gnu compilers is only to avoid warnings.)
|
|
*/
|
|
#if defined(__GNUC__)
|
|
#define cast_func(p) (__extension__ (lua_CFunction)(p))
|
|
#else
|
|
#define cast_func(p) ((lua_CFunction)(p))
|
|
#endif
|
|
|
|
|
|
static void lsys_unloadlib (void *lib) {
|
|
dlclose(lib);
|
|
}
|
|
|
|
|
|
static void *lsys_load (lua_State *L, const char *path, int seeglb) {
|
|
void *lib = dlopen(path, RTLD_NOW | (seeglb ? RTLD_GLOBAL : RTLD_LOCAL));
|
|
if (l_unlikely(lib == NULL))
|
|
lua_pushstring(L, dlerror());
|
|
return lib;
|
|
}
|
|
|
|
|
|
static lua_CFunction lsys_sym (lua_State *L, void *lib, const char *sym) {
|
|
lua_CFunction f = cast_func(dlsym(lib, sym));
|
|
if (l_unlikely(f == NULL))
|
|
lua_pushstring(L, dlerror());
|
|
return f;
|
|
}
|
|
|
|
/* }====================================================== */
|
|
|
|
|
|
|
|
#elif defined(LUA_DL_DLL) /* }{ */
|
|
/*
|
|
** {======================================================================
|
|
** This is an implementation of loadlib for Windows using native functions.
|
|
** =======================================================================
|
|
*/
|
|
|
|
#include <windows.h>
|
|
|
|
|
|
/*
|
|
** optional flags for LoadLibraryEx
|
|
*/
|
|
#if !defined(LUA_LLE_FLAGS)
|
|
#define LUA_LLE_FLAGS 0
|
|
#endif
|
|
|
|
|
|
#undef setprogdir
|
|
|
|
|
|
/*
|
|
** Replace in the path (on the top of the stack) any occurrence
|
|
** of LUA_EXEC_DIR with the executable's path.
|
|
*/
|
|
static void setprogdir (lua_State *L) {
|
|
char buff[MAX_PATH + 1];
|
|
char *lb;
|
|
DWORD nsize = sizeof(buff)/sizeof(char);
|
|
DWORD n = GetModuleFileNameA(NULL, buff, nsize); /* get exec. name */
|
|
if (n == 0 || n == nsize || (lb = strrchr(buff, '\\')) == NULL)
|
|
luaL_error(L, "unable to get ModuleFileName");
|
|
else {
|
|
*lb = '\0'; /* cut name on the last '\\' to get the path */
|
|
luaL_gsub(L, lua_tostring(L, -1), LUA_EXEC_DIR, buff);
|
|
lua_remove(L, -2); /* remove original string */
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
static void pusherror (lua_State *L) {
|
|
int error = GetLastError();
|
|
char buffer[128];
|
|
if (FormatMessageA(FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL, error, 0, buffer, sizeof(buffer)/sizeof(char), NULL))
|
|
lua_pushstring(L, buffer);
|
|
else
|
|
lua_pushfstring(L, "system error %d\n", error);
|
|
}
|
|
|
|
static void lsys_unloadlib (void *lib) {
|
|
FreeLibrary((HMODULE)lib);
|
|
}
|
|
|
|
|
|
static void *lsys_load (lua_State *L, const char *path, int seeglb) {
|
|
HMODULE lib = LoadLibraryExA(path, NULL, LUA_LLE_FLAGS);
|
|
(void)(seeglb); /* not used: symbols are 'global' by default */
|
|
if (lib == NULL) pusherror(L);
|
|
return lib;
|
|
}
|
|
|
|
|
|
static lua_CFunction lsys_sym (lua_State *L, void *lib, const char *sym) {
|
|
lua_CFunction f = (lua_CFunction)(voidf)GetProcAddress((HMODULE)lib, sym);
|
|
if (f == NULL) pusherror(L);
|
|
return f;
|
|
}
|
|
|
|
/* }====================================================== */
|
|
|
|
|
|
#else /* }{ */
|
|
/*
|
|
** {======================================================
|
|
** Fallback for other systems
|
|
** =======================================================
|
|
*/
|
|
|
|
#undef LIB_FAIL
|
|
#define LIB_FAIL "absent"
|
|
|
|
|
|
#define DLMSG "dynamic libraries not enabled; check your Lua installation"
|
|
|
|
|
|
static void lsys_unloadlib (void *lib) {
|
|
(void)(lib); /* not used */
|
|
}
|
|
|
|
|
|
static void *lsys_load (lua_State *L, const char *path, int seeglb) {
|
|
(void)(path); (void)(seeglb); /* not used */
|
|
lua_pushliteral(L, DLMSG);
|
|
return NULL;
|
|
}
|
|
|
|
|
|
static lua_CFunction lsys_sym (lua_State *L, void *lib, const char *sym) {
|
|
(void)(lib); (void)(sym); /* not used */
|
|
lua_pushliteral(L, DLMSG);
|
|
return NULL;
|
|
}
|
|
|
|
/* }====================================================== */
|
|
#endif /* } */
|
|
|
|
|
|
/*
|
|
** {==================================================================
|
|
** Set Paths
|
|
** ===================================================================
|
|
*/
|
|
|
|
/*
|
|
** LUA_PATH_VAR and LUA_CPATH_VAR are the names of the environment
|
|
** variables that Lua check to set its paths.
|
|
*/
|
|
#if !defined(LUA_PATH_VAR)
|
|
#define LUA_PATH_VAR "LUA_PATH"
|
|
#endif
|
|
|
|
#if !defined(LUA_CPATH_VAR)
|
|
#define LUA_CPATH_VAR "LUA_CPATH"
|
|
#endif
|
|
|
|
|
|
|
|
/*
|
|
** return registry.LUA_NOENV as a boolean
|
|
*/
|
|
static int noenv (lua_State *L) {
|
|
int b;
|
|
lua_getfield(L, LUA_REGISTRYINDEX, "LUA_NOENV");
|
|
b = lua_toboolean(L, -1);
|
|
lua_pop(L, 1); /* remove value */
|
|
return b;
|
|
}
|
|
|
|
|
|
/*
|
|
** Set a path
|
|
*/
|
|
static void setpath (lua_State *L, const char *fieldname,
|
|
const char *envname,
|
|
const char *dft) {
|
|
const char *dftmark;
|
|
const char *nver = lua_pushfstring(L, "%s%s", envname, LUA_VERSUFFIX);
|
|
const char *path = getenv(nver); /* try versioned name */
|
|
if (path == NULL) /* no versioned environment variable? */
|
|
path = getenv(envname); /* try unversioned name */
|
|
if (path == NULL || noenv(L)) /* no environment variable? */
|
|
lua_pushstring(L, dft); /* use default */
|
|
else if ((dftmark = strstr(path, LUA_PATH_SEP LUA_PATH_SEP)) == NULL)
|
|
lua_pushstring(L, path); /* nothing to change */
|
|
else { /* path contains a ";;": insert default path in its place */
|
|
size_t len = strlen(path);
|
|
luaL_Buffer b;
|
|
luaL_buffinit(L, &b);
|
|
if (path < dftmark) { /* is there a prefix before ';;'? */
|
|
luaL_addlstring(&b, path, dftmark - path); /* add it */
|
|
luaL_addchar(&b, *LUA_PATH_SEP);
|
|
}
|
|
luaL_addstring(&b, dft); /* add default */
|
|
if (dftmark < path + len - 2) { /* is there a suffix after ';;'? */
|
|
luaL_addchar(&b, *LUA_PATH_SEP);
|
|
luaL_addlstring(&b, dftmark + 2, (path + len - 2) - dftmark);
|
|
}
|
|
luaL_pushresult(&b);
|
|
}
|
|
setprogdir(L);
|
|
lua_setfield(L, -3, fieldname); /* package[fieldname] = path value */
|
|
lua_pop(L, 1); /* pop versioned variable name ('nver') */
|
|
}
|
|
|
|
/* }================================================================== */
|
|
|
|
|
|
/*
|
|
** return registry.CLIBS[path]
|
|
*/
|
|
static void *checkclib (lua_State *L, const char *path) {
|
|
void *plib;
|
|
lua_getfield(L, LUA_REGISTRYINDEX, CLIBS);
|
|
lua_getfield(L, -1, path);
|
|
plib = lua_touserdata(L, -1); /* plib = CLIBS[path] */
|
|
lua_pop(L, 2); /* pop CLIBS table and 'plib' */
|
|
return plib;
|
|
}
|
|
|
|
|
|
/*
|
|
** registry.CLIBS[path] = plib -- for queries
|
|
** registry.CLIBS[#CLIBS + 1] = plib -- also keep a list of all libraries
|
|
*/
|
|
static void addtoclib (lua_State *L, const char *path, void *plib) {
|
|
lua_getfield(L, LUA_REGISTRYINDEX, CLIBS);
|
|
lua_pushlightuserdata(L, plib);
|
|
lua_pushvalue(L, -1);
|
|
lua_setfield(L, -3, path); /* CLIBS[path] = plib */
|
|
lua_rawseti(L, -2, luaL_len(L, -2) + 1); /* CLIBS[#CLIBS + 1] = plib */
|
|
lua_pop(L, 1); /* pop CLIBS table */
|
|
}
|
|
|
|
|
|
/*
|
|
** __gc tag method for CLIBS table: calls 'lsys_unloadlib' for all lib
|
|
** handles in list CLIBS
|
|
*/
|
|
static int gctm (lua_State *L) {
|
|
lua_Integer n = luaL_len(L, 1);
|
|
for (; n >= 1; n--) { /* for each handle, in reverse order */
|
|
lua_rawgeti(L, 1, n); /* get handle CLIBS[n] */
|
|
lsys_unloadlib(lua_touserdata(L, -1));
|
|
lua_pop(L, 1); /* pop handle */
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
|
|
/* error codes for 'lookforfunc' */
|
|
#define ERRLIB 1
|
|
#define ERRFUNC 2
|
|
|
|
/*
|
|
** Look for a C function named 'sym' in a dynamically loaded library
|
|
** 'path'.
|
|
** First, check whether the library is already loaded; if not, try
|
|
** to load it.
|
|
** Then, if 'sym' is '*', return true (as library has been loaded).
|
|
** Otherwise, look for symbol 'sym' in the library and push a
|
|
** C function with that symbol.
|
|
** Return 0 and 'true' or a function in the stack; in case of
|
|
** errors, return an error code and an error message in the stack.
|
|
*/
|
|
static int lookforfunc (lua_State *L, const char *path, const char *sym) {
|
|
void *reg = checkclib(L, path); /* check loaded C libraries */
|
|
if (reg == NULL) { /* must load library? */
|
|
reg = lsys_load(L, path, *sym == '*'); /* global symbols if 'sym'=='*' */
|
|
if (reg == NULL) return ERRLIB; /* unable to load library */
|
|
addtoclib(L, path, reg);
|
|
}
|
|
if (*sym == '*') { /* loading only library (no function)? */
|
|
lua_pushboolean(L, 1); /* return 'true' */
|
|
return 0; /* no errors */
|
|
}
|
|
else {
|
|
lua_CFunction f = lsys_sym(L, reg, sym);
|
|
if (f == NULL)
|
|
return ERRFUNC; /* unable to find function */
|
|
lua_pushcfunction(L, f); /* else create new function */
|
|
return 0; /* no errors */
|
|
}
|
|
}
|
|
|
|
|
|
static int ll_loadlib (lua_State *L) {
|
|
const char *path = luaL_checkstring(L, 1);
|
|
const char *init = luaL_checkstring(L, 2);
|
|
int stat = lookforfunc(L, path, init);
|
|
if (l_likely(stat == 0)) /* no errors? */
|
|
return 1; /* return the loaded function */
|
|
else { /* error; error message is on stack top */
|
|
luaL_pushfail(L);
|
|
lua_insert(L, -2);
|
|
lua_pushstring(L, (stat == ERRLIB) ? LIB_FAIL : "init");
|
|
return 3; /* return fail, error message, and where */
|
|
}
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
** {======================================================
|
|
** 'require' function
|
|
** =======================================================
|
|
*/
|
|
|
|
|
|
static int readable (const char *filename) {
|
|
FILE *f = fopen(filename, "r"); /* try to open file */
|
|
if (f == NULL) return 0; /* open failed */
|
|
fclose(f);
|
|
return 1;
|
|
}
|
|
|
|
|
|
/*
|
|
** Get the next name in '*path' = 'name1;name2;name3;...', changing
|
|
** the ending ';' to '\0' to create a zero-terminated string. Return
|
|
** NULL when list ends.
|
|
*/
|
|
static const char *getnextfilename (char **path, char *end) {
|
|
char *sep;
|
|
char *name = *path;
|
|
if (name == end)
|
|
return NULL; /* no more names */
|
|
else if (*name == '\0') { /* from previous iteration? */
|
|
*name = *LUA_PATH_SEP; /* restore separator */
|
|
name++; /* skip it */
|
|
}
|
|
sep = strchr(name, *LUA_PATH_SEP); /* find next separator */
|
|
if (sep == NULL) /* separator not found? */
|
|
sep = end; /* name goes until the end */
|
|
*sep = '\0'; /* finish file name */
|
|
*path = sep; /* will start next search from here */
|
|
return name;
|
|
}
|
|
|
|
|
|
/*
|
|
** Given a path such as ";blabla.so;blublu.so", pushes the string
|
|
**
|
|
** no file 'blabla.so'
|
|
** no file 'blublu.so'
|
|
*/
|
|
static void pusherrornotfound (lua_State *L, const char *path) {
|
|
luaL_Buffer b;
|
|
luaL_buffinit(L, &b);
|
|
luaL_addstring(&b, "no file '");
|
|
luaL_addgsub(&b, path, LUA_PATH_SEP, "'\n\tno file '");
|
|
luaL_addstring(&b, "'");
|
|
luaL_pushresult(&b);
|
|
}
|
|
|
|
|
|
static const char *searchpath (lua_State *L, const char *name,
|
|
const char *path,
|
|
const char *sep,
|
|
const char *dirsep) {
|
|
luaL_Buffer buff;
|
|
char *pathname; /* path with name inserted */
|
|
char *endpathname; /* its end */
|
|
const char *filename;
|
|
/* separator is non-empty and appears in 'name'? */
|
|
if (*sep != '\0' && strchr(name, *sep) != NULL)
|
|
name = luaL_gsub(L, name, sep, dirsep); /* replace it by 'dirsep' */
|
|
luaL_buffinit(L, &buff);
|
|
/* add path to the buffer, replacing marks ('?') with the file name */
|
|
luaL_addgsub(&buff, path, LUA_PATH_MARK, name);
|
|
luaL_addchar(&buff, '\0');
|
|
pathname = luaL_buffaddr(&buff); /* writable list of file names */
|
|
endpathname = pathname + luaL_bufflen(&buff) - 1;
|
|
while ((filename = getnextfilename(&pathname, endpathname)) != NULL) {
|
|
if (readable(filename)) /* does file exist and is readable? */
|
|
return lua_pushstring(L, filename); /* save and return name */
|
|
}
|
|
luaL_pushresult(&buff); /* push path to create error message */
|
|
pusherrornotfound(L, lua_tostring(L, -1)); /* create error message */
|
|
return NULL; /* not found */
|
|
}
|
|
|
|
|
|
static int ll_searchpath (lua_State *L) {
|
|
const char *f = searchpath(L, luaL_checkstring(L, 1),
|
|
luaL_checkstring(L, 2),
|
|
luaL_optstring(L, 3, "."),
|
|
luaL_optstring(L, 4, LUA_DIRSEP));
|
|
if (f != NULL) return 1;
|
|
else { /* error message is on top of the stack */
|
|
luaL_pushfail(L);
|
|
lua_insert(L, -2);
|
|
return 2; /* return fail + error message */
|
|
}
|
|
}
|
|
|
|
|
|
static const char *findfile (lua_State *L, const char *name,
|
|
const char *pname,
|
|
const char *dirsep) {
|
|
const char *path;
|
|
lua_getfield(L, lua_upvalueindex(1), pname);
|
|
path = lua_tostring(L, -1);
|
|
if (l_unlikely(path == NULL))
|
|
luaL_error(L, "'package.%s' must be a string", pname);
|
|
return searchpath(L, name, path, ".", dirsep);
|
|
}
|
|
|
|
|
|
static int checkload (lua_State *L, int stat, const char *filename) {
|
|
if (l_likely(stat)) { /* module loaded successfully? */
|
|
lua_pushstring(L, filename); /* will be 2nd argument to module */
|
|
return 2; /* return open function and file name */
|
|
}
|
|
else
|
|
return luaL_error(L, "error loading module '%s' from file '%s':\n\t%s",
|
|
lua_tostring(L, 1), filename, lua_tostring(L, -1));
|
|
}
|
|
|
|
|
|
static int searcher_Lua (lua_State *L) {
|
|
const char *filename;
|
|
const char *name = luaL_checkstring(L, 1);
|
|
filename = findfile(L, name, "path", LUA_LSUBSEP);
|
|
if (filename == NULL) return 1; /* module not found in this path */
|
|
return checkload(L, (luaL_loadfile(L, filename) == LUA_OK), filename);
|
|
}
|
|
|
|
|
|
/*
|
|
** Try to find a load function for module 'modname' at file 'filename'.
|
|
** First, change '.' to '_' in 'modname'; then, if 'modname' has
|
|
** the form X-Y (that is, it has an "ignore mark"), build a function
|
|
** name "luaopen_X" and look for it. (For compatibility, if that
|
|
** fails, it also tries "luaopen_Y".) If there is no ignore mark,
|
|
** look for a function named "luaopen_modname".
|
|
*/
|
|
static int loadfunc (lua_State *L, const char *filename, const char *modname) {
|
|
const char *openfunc;
|
|
const char *mark;
|
|
modname = luaL_gsub(L, modname, ".", LUA_OFSEP);
|
|
mark = strchr(modname, *LUA_IGMARK);
|
|
if (mark) {
|
|
int stat;
|
|
openfunc = lua_pushlstring(L, modname, mark - modname);
|
|
openfunc = lua_pushfstring(L, LUA_POF"%s", openfunc);
|
|
stat = lookforfunc(L, filename, openfunc);
|
|
if (stat != ERRFUNC) return stat;
|
|
modname = mark + 1; /* else go ahead and try old-style name */
|
|
}
|
|
openfunc = lua_pushfstring(L, LUA_POF"%s", modname);
|
|
return lookforfunc(L, filename, openfunc);
|
|
}
|
|
|
|
|
|
static int searcher_C (lua_State *L) {
|
|
const char *name = luaL_checkstring(L, 1);
|
|
const char *filename = findfile(L, name, "cpath", LUA_CSUBSEP);
|
|
if (filename == NULL) return 1; /* module not found in this path */
|
|
return checkload(L, (loadfunc(L, filename, name) == 0), filename);
|
|
}
|
|
|
|
|
|
static int searcher_Croot (lua_State *L) {
|
|
const char *filename;
|
|
const char *name = luaL_checkstring(L, 1);
|
|
const char *p = strchr(name, '.');
|
|
int stat;
|
|
if (p == NULL) return 0; /* is root */
|
|
lua_pushlstring(L, name, p - name);
|
|
filename = findfile(L, lua_tostring(L, -1), "cpath", LUA_CSUBSEP);
|
|
if (filename == NULL) return 1; /* root not found */
|
|
if ((stat = loadfunc(L, filename, name)) != 0) {
|
|
if (stat != ERRFUNC)
|
|
return checkload(L, 0, filename); /* real error */
|
|
else { /* open function not found */
|
|
lua_pushfstring(L, "no module '%s' in file '%s'", name, filename);
|
|
return 1;
|
|
}
|
|
}
|
|
lua_pushstring(L, filename); /* will be 2nd argument to module */
|
|
return 2;
|
|
}
|
|
|
|
|
|
static int searcher_preload (lua_State *L) {
|
|
const char *name = luaL_checkstring(L, 1);
|
|
lua_getfield(L, LUA_REGISTRYINDEX, LUA_PRELOAD_TABLE);
|
|
if (lua_getfield(L, -1, name) == LUA_TNIL) { /* not found? */
|
|
lua_pushfstring(L, "no field package.preload['%s']", name);
|
|
return 1;
|
|
}
|
|
else {
|
|
lua_pushliteral(L, ":preload:");
|
|
return 2;
|
|
}
|
|
}
|
|
|
|
|
|
static void findloader (lua_State *L, const char *name) {
|
|
int i;
|
|
luaL_Buffer msg; /* to build error message */
|
|
/* push 'package.searchers' to index 3 in the stack */
|
|
if (l_unlikely(lua_getfield(L, lua_upvalueindex(1), "searchers")
|
|
!= LUA_TTABLE))
|
|
luaL_error(L, "'package.searchers' must be a table");
|
|
luaL_buffinit(L, &msg);
|
|
/* iterate over available searchers to find a loader */
|
|
for (i = 1; ; i++) {
|
|
luaL_addstring(&msg, "\n\t"); /* error-message prefix */
|
|
if (l_unlikely(lua_rawgeti(L, 3, i) == LUA_TNIL)) { /* no more searchers? */
|
|
lua_pop(L, 1); /* remove nil */
|
|
luaL_buffsub(&msg, 2); /* remove prefix */
|
|
luaL_pushresult(&msg); /* create error message */
|
|
luaL_error(L, "module '%s' not found:%s", name, lua_tostring(L, -1));
|
|
}
|
|
lua_pushstring(L, name);
|
|
lua_call(L, 1, 2); /* call it */
|
|
if (lua_isfunction(L, -2)) /* did it find a loader? */
|
|
return; /* module loader found */
|
|
else if (lua_isstring(L, -2)) { /* searcher returned error message? */
|
|
lua_pop(L, 1); /* remove extra return */
|
|
luaL_addvalue(&msg); /* concatenate error message */
|
|
}
|
|
else { /* no error message */
|
|
lua_pop(L, 2); /* remove both returns */
|
|
luaL_buffsub(&msg, 2); /* remove prefix */
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
static int ll_require (lua_State *L) {
|
|
const char *name = luaL_checkstring(L, 1);
|
|
lua_settop(L, 1); /* LOADED table will be at index 2 */
|
|
lua_getfield(L, LUA_REGISTRYINDEX, LUA_LOADED_TABLE);
|
|
lua_getfield(L, 2, name); /* LOADED[name] */
|
|
if (lua_toboolean(L, -1)) /* is it there? */
|
|
return 1; /* package is already loaded */
|
|
/* else must load package */
|
|
lua_pop(L, 1); /* remove 'getfield' result */
|
|
findloader(L, name);
|
|
lua_rotate(L, -2, 1); /* function <-> loader data */
|
|
lua_pushvalue(L, 1); /* name is 1st argument to module loader */
|
|
lua_pushvalue(L, -3); /* loader data is 2nd argument */
|
|
/* stack: ...; loader data; loader function; mod. name; loader data */
|
|
lua_call(L, 2, 1); /* run loader to load module */
|
|
/* stack: ...; loader data; result from loader */
|
|
if (!lua_isnil(L, -1)) /* non-nil return? */
|
|
lua_setfield(L, 2, name); /* LOADED[name] = returned value */
|
|
else
|
|
lua_pop(L, 1); /* pop nil */
|
|
if (lua_getfield(L, 2, name) == LUA_TNIL) { /* module set no value? */
|
|
lua_pushboolean(L, 1); /* use true as result */
|
|
lua_copy(L, -1, -2); /* replace loader result */
|
|
lua_setfield(L, 2, name); /* LOADED[name] = true */
|
|
}
|
|
lua_rotate(L, -2, 1); /* loader data <-> module result */
|
|
return 2; /* return module result and loader data */
|
|
}
|
|
|
|
/* }====================================================== */
|
|
|
|
|
|
|
|
|
|
static const luaL_Reg pk_funcs[] = {
|
|
{"loadlib", ll_loadlib},
|
|
{"searchpath", ll_searchpath},
|
|
/* placeholders */
|
|
{"preload", NULL},
|
|
{"cpath", NULL},
|
|
{"path", NULL},
|
|
{"searchers", NULL},
|
|
{"loaded", NULL},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
|
|
static const luaL_Reg ll_funcs[] = {
|
|
{"require", ll_require},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
|
|
static void createsearcherstable (lua_State *L) {
|
|
static const lua_CFunction searchers[] = {
|
|
searcher_preload,
|
|
searcher_Lua,
|
|
searcher_C,
|
|
searcher_Croot,
|
|
NULL
|
|
};
|
|
int i;
|
|
/* create 'searchers' table */
|
|
lua_createtable(L, sizeof(searchers)/sizeof(searchers[0]) - 1, 0);
|
|
/* fill it with predefined searchers */
|
|
for (i=0; searchers[i] != NULL; i++) {
|
|
lua_pushvalue(L, -2); /* set 'package' as upvalue for all searchers */
|
|
lua_pushcclosure(L, searchers[i], 1);
|
|
lua_rawseti(L, -2, i+1);
|
|
}
|
|
lua_setfield(L, -2, "searchers"); /* put it in field 'searchers' */
|
|
}
|
|
|
|
|
|
/*
|
|
** create table CLIBS to keep track of loaded C libraries,
|
|
** setting a finalizer to close all libraries when closing state.
|
|
*/
|
|
static void createclibstable (lua_State *L) {
|
|
luaL_getsubtable(L, LUA_REGISTRYINDEX, CLIBS); /* create CLIBS table */
|
|
lua_createtable(L, 0, 1); /* create metatable for CLIBS */
|
|
lua_pushcfunction(L, gctm);
|
|
lua_setfield(L, -2, "__gc"); /* set finalizer for CLIBS table */
|
|
lua_setmetatable(L, -2);
|
|
}
|
|
|
|
|
|
LUAMOD_API int luaopen_package (lua_State *L) {
|
|
createclibstable(L);
|
|
luaL_newlib(L, pk_funcs); /* create 'package' table */
|
|
createsearcherstable(L);
|
|
/* set paths */
|
|
setpath(L, "path", LUA_PATH_VAR, LUA_PATH_DEFAULT);
|
|
setpath(L, "cpath", LUA_CPATH_VAR, LUA_CPATH_DEFAULT);
|
|
/* store config information */
|
|
lua_pushliteral(L, LUA_DIRSEP "\n" LUA_PATH_SEP "\n" LUA_PATH_MARK "\n"
|
|
LUA_EXEC_DIR "\n" LUA_IGMARK "\n");
|
|
lua_setfield(L, -2, "config");
|
|
/* set field 'loaded' */
|
|
luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_LOADED_TABLE);
|
|
lua_setfield(L, -2, "loaded");
|
|
/* set field 'preload' */
|
|
luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_PRELOAD_TABLE);
|
|
lua_setfield(L, -2, "preload");
|
|
lua_pushglobaltable(L);
|
|
lua_pushvalue(L, -2); /* set 'package' as upvalue for next lib */
|
|
luaL_setfuncs(L, ll_funcs, 1); /* open lib into global table */
|
|
lua_pop(L, 1); /* pop global table */
|
|
return 1; /* return 'package' table */
|
|
}
|
|
|