mlua_sys/lua51/
lauxlib.rs

1//! Contains definitions from `lauxlib.h`.
2
3use std::os::raw::{c_char, c_int, c_void};
4use std::ptr;
5
6use super::lua::{self, lua_CFunction, lua_Integer, lua_Number, lua_State};
7
8// Extra error code for 'luaL_load'
9pub const LUA_ERRFILE: c_int = lua::LUA_ERRERR + 1;
10
11// Key, in the registry, for table of loaded modules
12pub const LUA_LOADED_TABLE: *const c_char = cstr!("_LOADED");
13
14#[repr(C)]
15pub struct luaL_Reg {
16    pub name: *const c_char,
17    pub func: lua_CFunction,
18}
19
20#[cfg_attr(all(windows, raw_dylib), link(name = "lua51", kind = "raw-dylib"))]
21unsafe extern "C-unwind" {
22    pub fn luaL_register(L: *mut lua_State, libname: *const c_char, l: *const luaL_Reg);
23    #[link_name = "luaL_getmetafield"]
24    pub fn luaL_getmetafield_(L: *mut lua_State, obj: c_int, e: *const c_char) -> c_int;
25    pub fn luaL_callmeta(L: *mut lua_State, obj: c_int, e: *const c_char) -> c_int;
26    pub fn luaL_typerror(L: *mut lua_State, narg: c_int, tname: *const c_char) -> c_int;
27    pub fn luaL_argerror(L: *mut lua_State, narg: c_int, extramsg: *const c_char) -> c_int;
28    pub fn luaL_checklstring(L: *mut lua_State, narg: c_int, l: *mut usize) -> *const c_char;
29    pub fn luaL_optlstring(
30        L: *mut lua_State,
31        narg: c_int,
32        def: *const c_char,
33        l: *mut usize,
34    ) -> *const c_char;
35    pub fn luaL_checknumber(L: *mut lua_State, narg: c_int) -> lua_Number;
36    pub fn luaL_optnumber(L: *mut lua_State, narg: c_int, def: lua_Number) -> lua_Number;
37    pub fn luaL_checkinteger(L: *mut lua_State, narg: c_int) -> lua_Integer;
38    pub fn luaL_optinteger(L: *mut lua_State, narg: c_int, def: lua_Integer) -> lua_Integer;
39    #[link_name = "luaL_checkstack"]
40    pub fn luaL_checkstack_(L: *mut lua_State, sz: c_int, msg: *const c_char);
41    pub fn luaL_checktype(L: *mut lua_State, narg: c_int, t: c_int);
42    pub fn luaL_checkany(L: *mut lua_State, narg: c_int);
43
44    #[link_name = "luaL_newmetatable"]
45    pub fn luaL_newmetatable_(L: *mut lua_State, tname: *const c_char) -> c_int;
46    pub fn luaL_checkudata(L: *mut lua_State, ud: c_int, tname: *const c_char) -> *mut c_void;
47
48    pub fn luaL_where(L: *mut lua_State, lvl: c_int);
49    pub fn luaL_error(L: *mut lua_State, fmt: *const c_char, ...) -> c_int;
50
51    pub fn luaL_checkoption(
52        L: *mut lua_State,
53        narg: c_int,
54        def: *const c_char,
55        lst: *const *const c_char,
56    ) -> c_int;
57}
58
59// Pre-defined references
60pub const LUA_NOREF: c_int = -2;
61pub const LUA_REFNIL: c_int = -1;
62
63#[cfg_attr(all(windows, raw_dylib), link(name = "lua51", kind = "raw-dylib"))]
64unsafe extern "C-unwind" {
65    pub fn luaL_ref(L: *mut lua_State, t: c_int) -> c_int;
66    pub fn luaL_unref(L: *mut lua_State, t: c_int, r#ref: c_int);
67
68    pub fn luaL_loadfile(L: *mut lua_State, filename: *const c_char) -> c_int;
69    pub fn luaL_loadbuffer(L: *mut lua_State, buff: *const c_char, sz: usize, name: *const c_char) -> c_int;
70    pub fn luaL_loadstring(L: *mut lua_State, s: *const c_char) -> c_int;
71
72    pub fn luaL_newstate() -> *mut lua_State;
73
74    pub fn luaL_gsub(
75        L: *mut lua_State,
76        s: *const c_char,
77        p: *const c_char,
78        r: *const c_char,
79    ) -> *const c_char;
80
81    pub fn luaL_findtable(
82        L: *mut lua_State,
83        idx: c_int,
84        fname: *const c_char,
85        szhint: c_int,
86    ) -> *const c_char;
87}
88
89//
90// Some useful macros (implemented as Rust functions)
91//
92
93#[inline(always)]
94pub unsafe fn luaL_argcheck(L: *mut lua_State, cond: c_int, narg: c_int, extramsg: *const c_char) {
95    if cond == 0 {
96        luaL_argerror(L, narg, extramsg);
97    }
98}
99
100#[inline(always)]
101pub unsafe fn luaL_checkstring(L: *mut lua_State, n: c_int) -> *const c_char {
102    luaL_checklstring(L, n, ptr::null_mut())
103}
104
105#[inline(always)]
106pub unsafe fn luaL_optstring(L: *mut lua_State, n: c_int, d: *const c_char) -> *const c_char {
107    luaL_optlstring(L, n, d, ptr::null_mut())
108}
109
110#[inline(always)]
111pub unsafe fn luaL_typename(L: *mut lua_State, i: c_int) -> *const c_char {
112    lua::lua_typename(L, lua::lua_type(L, i))
113}
114
115pub unsafe fn luaL_dofile(L: *mut lua_State, filename: *const c_char) -> c_int {
116    let status = luaL_loadfile(L, filename);
117    if status == 0 {
118        lua::lua_pcall(L, 0, lua::LUA_MULTRET, 0)
119    } else {
120        status
121    }
122}
123
124#[inline(always)]
125pub unsafe fn luaL_dostring(L: *mut lua_State, s: *const c_char) -> c_int {
126    let status = luaL_loadstring(L, s);
127    if status == 0 {
128        lua::lua_pcall(L, 0, lua::LUA_MULTRET, 0)
129    } else {
130        status
131    }
132}
133
134#[inline(always)]
135pub unsafe fn luaL_getmetatable(L: *mut lua_State, n: *const c_char) {
136    lua::lua_getfield_(L, lua::LUA_REGISTRYINDEX, n);
137}
138
139#[inline(always)]
140pub unsafe fn luaL_opt<T>(
141    L: *mut lua_State,
142    f: unsafe extern "C-unwind" fn(*mut lua_State, c_int) -> T,
143    n: c_int,
144    d: T,
145) -> T {
146    if lua::lua_isnoneornil(L, n) != 0 {
147        d
148    } else {
149        f(L, n)
150    }
151}
152
153//
154// Generic Buffer Manipulation
155//
156
157#[cfg(target_arch = "wasm32")]
158const BUFSIZ: usize = 1024; // WASI libc's BUFSIZ is 1024
159#[cfg(not(target_arch = "wasm32"))]
160const BUFSIZ: usize = libc::BUFSIZ as usize;
161
162// The buffer size used by the lauxlib buffer system.
163// The "16384" workaround is taken from the LuaJIT source code.
164pub const LUAL_BUFFERSIZE: usize = if BUFSIZ > 16384 { 8192 } else { BUFSIZ };
165
166#[repr(C)]
167pub struct luaL_Buffer {
168    pub p: *mut c_char, // current position in buffer
169    pub lvl: c_int,     // number of strings in the stack
170    pub L: *mut lua_State,
171    pub buffer: [c_char; LUAL_BUFFERSIZE],
172}
173
174#[cfg_attr(all(windows, raw_dylib), link(name = "lua51", kind = "raw-dylib"))]
175unsafe extern "C-unwind" {
176    pub fn luaL_buffinit(L: *mut lua_State, B: *mut luaL_Buffer);
177    pub fn luaL_prepbuffer(B: *mut luaL_Buffer) -> *mut c_char;
178    pub fn luaL_addlstring(B: *mut luaL_Buffer, s: *const c_char, l: usize);
179    pub fn luaL_addstring(B: *mut luaL_Buffer, s: *const c_char);
180    pub fn luaL_addvalue(B: *mut luaL_Buffer);
181    pub fn luaL_pushresult(B: *mut luaL_Buffer);
182}
183
184#[inline(always)]
185pub unsafe fn luaL_addchar(B: *mut luaL_Buffer, c: c_char) {
186    let buffer_end = (*B).buffer.as_mut_ptr().add(LUAL_BUFFERSIZE);
187    if (*B).p >= buffer_end {
188        luaL_prepbuffer(B);
189    }
190    *(*B).p = c;
191    (*B).p = (*B).p.add(1);
192}
193
194#[inline(always)]
195pub unsafe fn luaL_addsize(B: *mut luaL_Buffer, n: usize) {
196    (*B).p = (*B).p.add(n);
197}
mlua_sys/lua51/lauxlib.rs

mlua_sys/lua51/
lauxlib.rs
