rusefi
/
lua
mirror of https://github.com/rusefi/lua.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

new loader for "all-in-one" C packages

This commit is contained in:
Roberto Ierusalimschy 2005-08-01 01:21:17 -03:00
parent 1a343814d8
commit e82bdb24f3
1 changed files with 89 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

154
loadlib.c
View File

@ -1,5 +1,5 @@
/* /*
** $Id: loadlib.c,v 1.33 2005/07/12 21:17:46 roberto Exp roberto $ ** $Id: loadlib.c,v 1.34 2005/07/13 19:02:42 roberto Exp roberto $
** Dynamic library loader for Lua ** Dynamic library loader for Lua
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
** **
@ -38,6 +38,10 @@
#define LIB_FAIL "open" #define LIB_FAIL "open"
/* error codes for ll_loadfunc */
#define ERRLIB 1
#define ERRFUNC 2
#define setprogdir(L) ((void)0) #define setprogdir(L) ((void)0)
@ -294,43 +298,45 @@ static int gctm (lua_State *L) {
static int ll_loadfunc (lua_State *L, const char *path, const char *sym) { static int ll_loadfunc (lua_State *L, const char *path, const char *sym) {
const char *reason;
void **reg = ll_register(L, path); void **reg = ll_register(L, path);
if (*reg == NULL) *reg = ll_load(L, path); if (*reg == NULL) *reg = ll_load(L, path);
if (*reg == NULL) if (*reg == NULL)
reason = LIB_FAIL; return ERRLIB; /* unable to load library */
else { else {
lua_CFunction f = ll_sym(L, *reg, sym); lua_CFunction f = ll_sym(L, *reg, sym);
if (f) { if (f == NULL)
lua_pushcfunction(L, f); return ERRFUNC; /* unable to find function */
return 1; /* return function */ lua_pushcfunction(L, f);
} return 0; /* return function */
reason = "init";
} }
lua_pushnil(L);
lua_insert(L, -2);
lua_pushstring(L, reason);
return 3; /* return nil, ll_error, reason */
} }
static int ll_loadlib (lua_State *L) { static int ll_loadlib (lua_State *L) {
const char *path = luaL_checkstring(L, 1); const char *path = luaL_checkstring(L, 1);
const char *init = luaL_checkstring(L, 2); const char *init = luaL_checkstring(L, 2);
return ll_loadfunc(L, path, init); int stat = ll_loadfunc(L, path, init);
if (stat == 0) /* no errors? */
return 1; /* return the loaded function */
else { /* error; error message is on stack top */
lua_pushnil(L);
lua_insert(L, -2);
lua_pushstring(L, (stat == ERRLIB) ? LIB_FAIL : "init");
return 3; /* return nil, error message, and where */
}
} }
/* /*
** {====================================================== ** {======================================================
** `require' and `module' functions ** 'require' function
** ======================================================= ** =======================================================
*/ */
static int readable (const char *fname) { static int readable (const char *filename) {
FILE *f = fopen(fname, "r"); /* try to open file */ FILE *f = fopen(filename, "r"); /* try to open file */
if (f == NULL) return 0; /* open failed */ if (f == NULL) return 0; /* open failed */
fclose(f); fclose(f);
return 1; return 1;
@ -348,53 +354,85 @@ static const char *pushnexttemplate (lua_State *L, const char *path) {
} }
static const char *findfile (lua_State *L, const char *pname) { static const char *findfile (lua_State *L, const char *name,
const char *pname) {
const char *path; const char *path;
const char *name = luaL_checkstring(L, 1);
name = luaL_gsub(L, name, ".", LUA_DIRSEP); name = luaL_gsub(L, name, ".", LUA_DIRSEP);
lua_getfield(L, LUA_ENVIRONINDEX, pname); lua_getfield(L, LUA_ENVIRONINDEX, pname);
path = lua_tostring(L, -1); path = lua_tostring(L, -1);
if (path == NULL) if (path == NULL)
luaL_error(L, LUA_QL("package.%s") " must be a string", pname); luaL_error(L, LUA_QL("package.%s") " must be a string", pname);
while ((path = pushnexttemplate(L, path)) != NULL) { while ((path = pushnexttemplate(L, path)) != NULL) {
const char *fname; const char *filename;
fname = luaL_gsub(L, lua_tostring(L, -1), LUA_PATH_MARK, name); filename = luaL_gsub(L, lua_tostring(L, -1), LUA_PATH_MARK, name);
if (readable(fname)) /* does file exist and is readable? */ if (readable(filename)) /* does file exist and is readable? */
return fname; /* return that file name */ return filename; /* return that file name */
lua_pop(L, 2); /* remove path template and file name */ lua_pop(L, 2); /* remove path template and file name */
} }
return NULL; /* not found */ return NULL; /* not found */
} }
static void loaderror (lua_State *L, const char *msg) { static void loaderror (lua_State *L) {
luaL_error(L, "error loading package " LUA_QS " (%s)", luaL_error(L, "error loading package " LUA_QS " (%s)",
lua_tostring(L, 1), msg); lua_tostring(L, 1), lua_tostring(L, -1));
} }
static int loader_Lua (lua_State *L) { static int loader_Lua (lua_State *L) {
const char *fname; const char *filename;
fname = findfile(L, "path"); const char *name = luaL_checkstring(L, 1);
if (fname == NULL) return 0; /* library not found in this path */ filename = findfile(L, name, "path");
if (luaL_loadfile(L, fname) != 0) if (filename == NULL) return 0; /* library not found in this path */
loaderror(L, lua_tostring(L, -1)); if (luaL_loadfile(L, filename) != 0)
loaderror(L);
return 1; /* library loaded successfully */ return 1; /* library loaded successfully */
} }
static const char *mkfuncname (lua_State *L, const char *modname) {
const char *funcname;
const char *mark = strchr(modname, *LUA_IGMARK);
if (mark) modname = mark + 1;
funcname = luaL_gsub(L, modname, ".", LUA_OFSEP);
funcname = lua_pushfstring(L, POF"%s", funcname);
lua_remove(L, -2); /* remove 'gsub' result */
return funcname;
}
static int loader_C (lua_State *L) { static int loader_C (lua_State *L) {
const char *funcname; const char *funcname;
const char *fname = findfile(L, "cpath"); const char *name = luaL_checkstring(L, 1);
if (fname == NULL) return 0; /* library not found in this path */ const char *filename = findfile(L, name, "cpath");
funcname = luaL_gsub(L, lua_tostring(L, 1), ".", LUA_OFSEP); if (filename == NULL) return 0; /* library not found in this path */
funcname = lua_pushfstring(L, POF"%s", funcname); funcname = mkfuncname(L, name);
if (ll_loadfunc(L, fname, funcname) != 1) if (ll_loadfunc(L, filename, funcname) != 0)
loaderror(L, lua_tostring(L, -2)); loaderror(L);
return 1; /* library loaded successfully */ return 1; /* library loaded successfully */
} }
static int loader_Croot (lua_State *L) {
const char *funcname;
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");
if (filename == NULL) return 0; /* root not found */
funcname = mkfuncname(L, name);
if ((stat = ll_loadfunc(L, filename, funcname)) != 0) {
if (stat == ERRFUNC) return 0; /* function not found */
else
loaderror(L); /* real error */
}
return 1;
}
static int loader_preload (lua_State *L) { static int loader_preload (lua_State *L) {
lua_getfield(L, LUA_ENVIRONINDEX, "preload"); lua_getfield(L, LUA_ENVIRONINDEX, "preload");
if (!lua_istable(L, -1)) if (!lua_istable(L, -1))
@ -440,37 +478,23 @@ static int require_aux (lua_State *L, const char *name) {
} }
static void require_check (lua_State *L, const char *name) {
if (!require_aux(L, name)) { /* error? */
/* build and show error message */
const char *msg;
lua_settop(L, 1);
lua_getfield(L, LUA_ENVIRONINDEX, "path");
lua_getfield(L, LUA_ENVIRONINDEX, "cpath");
msg = lua_pushfstring(L,
"package " LUA_QS " not found in following paths:\n"
" Lua path: %s\n"
" C path: %s\n", name,
lua_tostring(L, -2), lua_tostring(L, -1));
msg = luaL_gsub(L, msg, LUA_PATHSEP, "\n ");
luaL_error(L, msg);
}
}
static int ll_require (lua_State *L) { static int ll_require (lua_State *L) {
const char *name = luaL_checkstring(L, 1); const char *name = luaL_checkstring(L, 1);
const char *pt; if (!require_aux(L, name)) /* error? */
/* load all parent modules */ luaL_error(L, "package " LUA_QS " not found", name);
for (pt = name; (pt = strchr(pt, '.')) != NULL; pt++) {
lua_settop(L, 1);
lua_pushlstring(L, name, pt - name);
require_aux(L, lua_tostring(L, -1));
}
require_check(L, name); /* load module itself */
return 1; return 1;
} }
/* }====================================================== */
/*
** {======================================================
** 'module' function
** =======================================================
*/
static void setfenv (lua_State *L) { static void setfenv (lua_State *L) {
lua_Debug ar; lua_Debug ar;
@ -524,14 +548,14 @@ static int ll_module (lua_State *L) {
return 0; return 0;
} }
/* }====================================================== */ /* }====================================================== */
/* auxiliary mark (for internal use) */ /* auxiliary mark (for internal use) */
#define AUXMARK "\1" #define AUXMARK "\1"
static void setpath (lua_State *L, const char *fname, const char *envname, static void setpath (lua_State *L, const char *fieldname, const char *envname,
const char *def) { const char *def) {
const char *path = getenv(envname); const char *path = getenv(envname);
if (path == NULL) /* no environment variable? */ if (path == NULL) /* no environment variable? */
@ -544,7 +568,7 @@ static void setpath (lua_State *L, const char *fname, const char *envname,
lua_remove(L, -2); lua_remove(L, -2);
} }
setprogdir(L); setprogdir(L);
lua_setfield(L, -2, fname); lua_setfield(L, -2, fieldname);
} }
@ -556,7 +580,7 @@ static const luaL_reg ll_funcs[] = {
static const lua_CFunction loaders[] = static const lua_CFunction loaders[] =
{loader_preload, loader_Lua, loader_C, NULL}; {loader_preload, loader_Lua, loader_C, loader_Croot, NULL};
LUALIB_API int luaopen_loadlib (lua_State *L) { LUALIB_API int luaopen_loadlib (lua_State *L) {