xref: /freebsd/libexec/flua/linit_flua.c (revision 9c7db0931d486ce800aa71d9f41abfad9f7f691f)
1 /*
2 ** $Id: linit.c,v 1.39.1.1 2017/04/19 17:20:42 roberto Exp $
3 ** Initialization of libraries for lua.c and other clients
4 ** See Copyright Notice in lua.h
5 */
6 
7 
8 #define linit_c
9 #define LUA_LIB
10 
11 /*
12 ** If you embed Lua in your program and need to open the standard
13 ** libraries, call luaL_openlibs in your program. If you need a
14 ** different set of libraries, copy this file to your project and edit
15 ** it to suit your needs.
16 **
17 ** You can also *preload* libraries, so that a later 'require' can
18 ** open the library, which is already linked to the application.
19 ** For that, do the following code:
20 **
21 **  luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_PRELOAD_TABLE);
22 **  lua_pushcfunction(L, luaopen_modname);
23 **  lua_setfield(L, -2, modname);
24 **  lua_pop(L, 1);  // remove PRELOAD table
25 */
26 
27 #include "lprefix.h"
28 
29 #include <stddef.h>
30 #include <stdlib.h>
31 
32 #include "lua.h"
33 
34 #include "lualib.h"
35 #include "lauxlib.h"
36 #include "lposix.h"
37 
38 #include "bootstrap.h"
39 
40 /*
41 ** these libs are loaded by lua.c and are readily available to any Lua
42 ** program
43 */
44 static const luaL_Reg loadedlibs[] = {
45   {"_G", luaopen_base},
46   {LUA_LOADLIBNAME, luaopen_package},
47   {LUA_COLIBNAME, luaopen_coroutine},
48   {LUA_TABLIBNAME, luaopen_table},
49   {LUA_IOLIBNAME, luaopen_io},
50   {LUA_OSLIBNAME, luaopen_os},
51   {LUA_STRLIBNAME, luaopen_string},
52   {LUA_MATHLIBNAME, luaopen_math},
53   {LUA_UTF8LIBNAME, luaopen_utf8},
54   {LUA_DBLIBNAME, luaopen_debug},
55 #if defined(LUA_COMPAT_BITLIB)
56   {LUA_BITLIBNAME, luaopen_bit32},
57 #endif
58   /* FreeBSD Extensions */
59   {"posix", luaopen_posix},
60   {NULL, NULL}
61 };
62 
63 #ifdef BOOTSTRAPPING
flua_init_env(void)64 static void __attribute__((constructor)) flua_init_env(void) {
65   /*
66    * This happens in the middle of luaopen_package().  We could move it into
67    * flua_setup_mods(), but it seems better to avoid its timing being so
68    * important that it would break some of our bootstrap modules if someone
69    * were to reorder things.
70    */
71   if (getenv("LUA_PATH") == NULL)
72     setenv("LUA_PATH", BOOTSTRAP_FLUA_PATH, 1);
73 }
74 
flua_setup_mods(lua_State * L)75 static void flua_setup_mods (lua_State *L) {
76   const luaL_Reg **flib;
77 
78   SET_FOREACH(flib, FLUA_MODULE_SETNAME) {
79     luaL_requiref(L, (*flib)->name, (*flib)->func, 1);
80     lua_pop(L, 1);  /* remove lib */
81   }
82 };
83 #endif
84 
luaL_openlibs(lua_State * L)85 LUALIB_API void luaL_openlibs (lua_State *L) {
86   const luaL_Reg *lib;
87   /* "require" functions from 'loadedlibs' and set results to global table */
88   for (lib = loadedlibs; lib->func; lib++) {
89     luaL_requiref(L, lib->name, lib->func, 1);
90     lua_pop(L, 1);  /* remove lib */
91   }
92 #ifdef BOOTSTRAPPING
93   flua_setup_mods(L);
94 #endif
95 }
96