882a90be48
Lexes $NAME as TK_ENVVAR, compiles reads as __getenv("NAME") calls
and writes ($NAME = expr) as __setenv("NAME", expr) calls. Runtime
functions wrap getenv/setenv/unsetenv with automatic tostring coercion.
103 lines
2.3 KiB
C
103 lines
2.3 KiB
C
/*
|
|
** $Id: linit.c $
|
|
** Initialization of libraries for lua.c and other clients
|
|
** See Copyright Notice in lua.h
|
|
*/
|
|
|
|
|
|
#define linit_c
|
|
#define LUA_LIB
|
|
|
|
|
|
#include "lprefix.h"
|
|
|
|
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "lua.h"
|
|
|
|
#include "lualib.h"
|
|
#include "lauxlib.h"
|
|
#include "llimits.h"
|
|
#include "lcmd.h"
|
|
|
|
|
|
/*
|
|
** Standard Libraries. (Must be listed in the same ORDER of their
|
|
** respective constants LUA_<libname>K.)
|
|
*/
|
|
static const luaL_Reg stdlibs[] = {
|
|
{LUA_GNAME, luaopen_base},
|
|
{LUA_LOADLIBNAME, luaopen_package},
|
|
{LUA_COLIBNAME, luaopen_coroutine},
|
|
{LUA_DBLIBNAME, luaopen_debug},
|
|
{LUA_IOLIBNAME, luaopen_io},
|
|
{LUA_MATHLIBNAME, luaopen_math},
|
|
{LUA_OSLIBNAME, luaopen_os},
|
|
{LUA_STRLIBNAME, luaopen_string},
|
|
{LUA_TABLIBNAME, luaopen_table},
|
|
{LUA_UTF8LIBNAME, luaopen_utf8},
|
|
{NULL, NULL}
|
|
};
|
|
|
|
|
|
static int luaB_getenv (lua_State *L) {
|
|
const char *name = luaL_checkstring(L, 1);
|
|
const char *val = getenv(name);
|
|
if (val == NULL)
|
|
lua_pushnil(L);
|
|
else
|
|
lua_pushstring(L, val);
|
|
return 1;
|
|
}
|
|
|
|
|
|
static int luaB_setenv (lua_State *L) {
|
|
const char *name = luaL_checkstring(L, 1);
|
|
if (lua_isnoneornil(L, 2))
|
|
unsetenv(name);
|
|
else {
|
|
const char *val = luaL_tolstring(L, 2, NULL);
|
|
setenv(name, val, 1);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*
|
|
** Register shell built-in globals (called after standard libraries).
|
|
*/
|
|
static void opencommand (lua_State *L) {
|
|
lua_pushcfunction(L, luaB_command);
|
|
lua_setglobal(L, "__command");
|
|
lua_pushcfunction(L, luaB_getenv);
|
|
lua_setglobal(L, "__getenv");
|
|
lua_pushcfunction(L, luaB_setenv);
|
|
lua_setglobal(L, "__setenv");
|
|
}
|
|
|
|
|
|
/*
|
|
** require and preload selected standard libraries
|
|
*/
|
|
LUALIB_API void luaL_openselectedlibs (lua_State *L, int load, int preload) {
|
|
int mask;
|
|
const luaL_Reg *lib;
|
|
luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_PRELOAD_TABLE);
|
|
for (lib = stdlibs, mask = 1; lib->name != NULL; lib++, mask <<= 1) {
|
|
if (load & mask) { /* selected? */
|
|
luaL_requiref(L, lib->name, lib->func, 1); /* require library */
|
|
lua_pop(L, 1); /* remove result from the stack */
|
|
}
|
|
else if (preload & mask) { /* selected? */
|
|
lua_pushcfunction(L, lib->func);
|
|
lua_setfield(L, -2, lib->name); /* add library to PRELOAD table */
|
|
}
|
|
}
|
|
lua_assert((mask >> 1) == LUA_UTF8LIBK);
|
|
lua_pop(L, 1); /* remove PRELOAD table */
|
|
opencommand(L); /* register __command global */
|
|
}
|
|
|