Migrate linit

Author: khvzak

build/build.rs

@@ -5,17 +5,16 @@ fn main() {
         .define("LUA_USE_POSIX", None)
         // Lua core
         .file("build/lapi.c")
+        .file("build/lauxlib.c")
         .file("build/lcode.c")
         .file("build/ldump.c")
         .file("build/lgc.c")
-        .file("build/linit.c")
         .file("build/llex.c")
         .file("build/loadlib.c")
         .file("build/lparser.c")
         .file("build/lundump.c")
         .file("build/lvm.c")
         // Non core libs
-        .file("build/lauxlib.c")
         .file("build/lbaselib.c")
         .file("build/lbitlib.c")
         .file("build/lcorolib.c")

build/linit.c

@@ -2,6 +2,8 @@
 ** Lua API
 */
 
+use libc::c_int;
+
 use crate::lstate::lua_State;
 use crate::types::{lua_Number, LUA_MULTRET};
 
@@ -23,6 +25,11 @@ pub(crate) unsafe fn api_checknelems(L: *mut lua_State, n: i32) {
     );
 }
 
+pub unsafe fn lua_pop(L: *mut lua_State, n: c_int) {
+    lua_settop(L, -n - 1)
+}
+
 extern "C" {
     pub fn lua_version(L: *mut lua_State) -> *const lua_Number;
+    pub fn lua_settop(L: *mut lua_State, idx: c_int);
 }

src/lapi.rs

@@ -0,0 +1,23 @@
+/*
+** Auxiliary functions for building Lua libraries
+*/
+
+use libc::{c_char, c_int};
+
+use crate::{lstate::lua_State, types::lua_CFunction};
+
+#[derive(Copy, Clone)]
+#[repr(C)]
+pub struct luaL_Reg {
+    pub name: *const c_char,
+    pub func: lua_CFunction,
+}
+
+extern "C" {
+    pub fn luaL_requiref(
+        L: *mut lua_State,
+        modname: *const c_char,
+        openf: lua_CFunction,
+        glb: c_int,
+    );
+}

src/lauxlib.rs

@@ -15,10 +15,13 @@ pub(crate) mod lstate;
 pub(crate) mod lobject;
 
 pub(crate) mod lapi;
+pub(crate) mod lauxlib;
 pub(crate) mod ldebug;
 pub(crate) mod ldo;
 pub(crate) mod lfunc;
 pub(crate) mod lgc;
+pub(crate) mod libs;
+pub(crate) mod linit;
 pub(crate) mod llex;
 pub(crate) mod llimits;
 pub(crate) mod lmem;

src/lib.rs

@@ -0,0 +1,16 @@
+use libc::c_int;
+
+use crate::lstate::lua_State;
+
+extern "C" {
+    pub fn luaopen_base(L: *mut lua_State) -> c_int;
+    pub fn luaopen_package(L: *mut lua_State) -> c_int;
+    pub fn luaopen_coroutine(L: *mut lua_State) -> c_int;
+    pub fn luaopen_table(L: *mut lua_State) -> c_int;
+    pub fn luaopen_io(L: *mut lua_State) -> c_int;
+    pub fn luaopen_os(L: *mut lua_State) -> c_int;
+    pub fn luaopen_string(L: *mut lua_State) -> c_int;
+    pub fn luaopen_math(L: *mut lua_State) -> c_int;
+    pub fn luaopen_utf8(L: *mut lua_State) -> c_int;
+    pub fn luaopen_debug(L: *mut lua_State) -> c_int;
+}

src/libs/mod.rs

@@ -0,0 +1,83 @@
+/*
+** Initialization of libraries
+*/
+
+use crate::lapi::lua_pop;
+use crate::lauxlib::{luaL_Reg, luaL_requiref};
+use crate::libs::{
+    luaopen_base, luaopen_coroutine, luaopen_debug, luaopen_io, luaopen_math, luaopen_os,
+    luaopen_package, luaopen_string, luaopen_table, luaopen_utf8,
+};
+use crate::lstate::lua_State;
+
+/*
+** If you embed Lua in your program and need to open the standard
+** libraries, call luaL_openlibs in your program. If you need a
+** different set of libraries, copy this file to your project and edit
+** it to suit your needs.
+**
+** You can also *preload* libraries, so that a later 'require' can
+** open the library, which is already linked to the application.
+** For that, do the following code:
+**
+**  luaL_getsubtable(L, LUA_REGISTRYINDEX, LUA_PRELOAD_TABLE);
+**  lua_pushcfunction(L, luaopen_modname);
+**  lua_setfield(L, -2, modname);
+**  lua_pop(L, 1);  // remove PRELOAD table
+*/
+
+/*
+** these libs are loaded by lua.c and are readily available to any Lua
+** program
+*/
+const loadedlibs: [luaL_Reg; 10] = [
+    luaL_Reg {
+        name: cstr!("_G"),
+        func: Some(luaopen_base),
+    },
+    luaL_Reg {
+        name: cstr!("package"), // LUA_LOADLIBNAME
+        func: Some(luaopen_package),
+    },
+    luaL_Reg {
+        name: cstr!("coroutine"), // LUA_COLIBNAME
+        func: Some(luaopen_coroutine),
+    },
+    luaL_Reg {
+        name: cstr!("table"), // LUA_TABLIBNAME
+        func: Some(luaopen_table),
+    },
+    luaL_Reg {
+        name: cstr!("io"), // LUA_IOLIBNAME
+        func: Some(luaopen_io),
+    },
+    luaL_Reg {
+        name: cstr!("os"), // LUA_OSLIBNAME
+        func: Some(luaopen_os),
+    },
+    luaL_Reg {
+        name: cstr!("string"), // LUA_STRLIBNAME
+        func: Some(luaopen_string),
+    },
+    luaL_Reg {
+        name: cstr!("math"), // LUA_MATHLIBNAME
+        func: Some(luaopen_math),
+    },
+    luaL_Reg {
+        name: cstr!("utf8"), // LUA_UTF8LIBNAME
+        func: Some(luaopen_utf8),
+    },
+    luaL_Reg {
+        name: cstr!("debug"), // LUA_DBLIBNAME
+        func: Some(luaopen_debug),
+    },
+];
+
+#[no_mangle]
+pub unsafe extern "C" fn luaL_openlibs(L: *mut lua_State) {
+    /* "require" functions from 'loadedlibs' and set results to global table */
+    for lib in &loadedlibs {
+        luaL_requiref(L, lib.name, lib.func, 1);
+        lua_pop(L, 1);
+    }
+}