Home | History | Annotate | Line # | Download | only in src
      1 /*	$NetBSD: lauxlib.h,v 1.9 2023/04/16 20:46:17 nikita Exp $	*/
      2 
      3 /*
      4 ** Id: lauxlib.h
      5 ** Auxiliary functions for building Lua libraries
      6 ** See Copyright Notice in lua.h
      7 */
      8 
      9 
     10 #ifndef lauxlib_h
     11 #define lauxlib_h
     12 
     13 
     14 #ifndef _KERNEL
     15 #include <stddef.h>
     16 #include <stdio.h>
     17 #endif /* _KERNEL */
     18 
     19 #include "luaconf.h"
     20 #include "lua.h"
     21 
     22 
     23 /* global table */
     24 #define LUA_GNAME	"_G"
     25 
     26 
     27 typedef struct luaL_Buffer luaL_Buffer;
     28 
     29 
     30 /* extra error code for 'luaL_loadfilex' */
     31 #define LUA_ERRFILE     (LUA_ERRERR+1)
     32 
     33 
     34 /* key, in the registry, for table of loaded modules */
     35 #define LUA_LOADED_TABLE	"_LOADED"
     36 
     37 
     38 /* key, in the registry, for table of preloaded loaders */
     39 #define LUA_PRELOAD_TABLE	"_PRELOAD"
     40 
     41 
     42 typedef struct luaL_Reg {
     43   const char *name;
     44   lua_CFunction func;
     45 } luaL_Reg;
     46 
     47 
     48 #define LUAL_NUMSIZES	(sizeof(lua_Integer)*16 + sizeof(lua_Number))
     49 
     50 LUALIB_API void (luaL_checkversion_) (lua_State *L, lua_Number ver, size_t sz);
     51 #define luaL_checkversion(L)  \
     52 	  luaL_checkversion_(L, LUA_VERSION_NUM, LUAL_NUMSIZES)
     53 
     54 LUALIB_API int (luaL_getmetafield) (lua_State *L, int obj, const char *e);
     55 LUALIB_API int (luaL_callmeta) (lua_State *L, int obj, const char *e);
     56 LUALIB_API const char *(luaL_tolstring) (lua_State *L, int idx, size_t *len);
     57 LUALIB_API int (luaL_argerror) (lua_State *L, int arg, const char *extramsg);
     58 LUALIB_API int (luaL_typeerror) (lua_State *L, int arg, const char *tname);
     59 LUALIB_API const char *(luaL_checklstring) (lua_State *L, int arg,
     60                                                           size_t *l);
     61 LUALIB_API const char *(luaL_optlstring) (lua_State *L, int arg,
     62                                           const char *def, size_t *l);
     63 LUALIB_API lua_Number (luaL_checknumber) (lua_State *L, int arg);
     64 LUALIB_API lua_Number (luaL_optnumber) (lua_State *L, int arg, lua_Number def);
     65 
     66 #ifndef _KERNEL
     67 LUALIB_API lua_Integer (luaL_checkinteger) (lua_State *L, int arg);
     68 LUALIB_API lua_Integer (luaL_optinteger) (lua_State *L, int arg,
     69                                           lua_Integer def);
     70 #else /* _KERNEL */
     71 #define luaL_checkinteger		luaL_checknumber
     72 #define luaL_optinteger(L,a,d)	luaL_optnumber(L, (a), (lua_Number)(d))
     73 #endif /* _KERNEL */
     74 
     75 LUALIB_API void (luaL_checkstack) (lua_State *L, int sz, const char *msg);
     76 LUALIB_API void (luaL_checktype) (lua_State *L, int arg, int t);
     77 LUALIB_API void (luaL_checkany) (lua_State *L, int arg);
     78 
     79 LUALIB_API int   (luaL_newmetatable) (lua_State *L, const char *tname);
     80 LUALIB_API void  (luaL_setmetatable) (lua_State *L, const char *tname);
     81 LUALIB_API void *(luaL_testudata) (lua_State *L, int ud, const char *tname);
     82 LUALIB_API void *(luaL_checkudata) (lua_State *L, int ud, const char *tname);
     83 
     84 LUALIB_API void (luaL_where) (lua_State *L, int lvl);
     85 LUALIB_API int (luaL_error) (lua_State *L, const char *fmt, ...);
     86 
     87 LUALIB_API int (luaL_checkoption) (lua_State *L, int arg, const char *def,
     88                                    const char *const lst[]);
     89 
     90 #ifndef _KERNEL
     91 LUALIB_API int (luaL_fileresult) (lua_State *L, int stat, const char *fname);
     92 LUALIB_API int (luaL_execresult) (lua_State *L, int stat);
     93 #endif /* _KERNEL */
     94 
     95 
     96 /* predefined references */
     97 #define LUA_NOREF       (-2)
     98 #define LUA_REFNIL      (-1)
     99 
    100 LUALIB_API int (luaL_ref) (lua_State *L, int t);
    101 LUALIB_API void (luaL_unref) (lua_State *L, int t, int ref);
    102 
    103 #ifndef _KERNEL
    104 LUALIB_API int (luaL_loadfilex) (lua_State *L, const char *filename,
    105                                                const char *mode);
    106 
    107 #define luaL_loadfile(L,f)	luaL_loadfilex(L,f,NULL)
    108 #endif /* _KERNEL */
    109 
    110 LUALIB_API int (luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz,
    111                                    const char *name, const char *mode);
    112 LUALIB_API int (luaL_loadstring) (lua_State *L, const char *s);
    113 
    114 LUALIB_API lua_State *(luaL_newstate) (void);
    115 
    116 LUALIB_API lua_Integer (luaL_len) (lua_State *L, int idx);
    117 
    118 LUALIB_API void (luaL_addgsub) (luaL_Buffer *b, const char *s,
    119                                      const char *p, const char *r);
    120 LUALIB_API const char *(luaL_gsub) (lua_State *L, const char *s,
    121                                     const char *p, const char *r);
    122 
    123 LUALIB_API void (luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
    124 
    125 LUALIB_API int (luaL_getsubtable) (lua_State *L, int idx, const char *fname);
    126 
    127 LUALIB_API void (luaL_traceback) (lua_State *L, lua_State *L1,
    128                                   const char *msg, int level);
    129 
    130 LUALIB_API void (luaL_requiref) (lua_State *L, const char *modname,
    131                                  lua_CFunction openf, int glb);
    132 
    133 /*
    134 ** ===============================================================
    135 ** some useful macros
    136 ** ===============================================================
    137 */
    138 
    139 
    140 #define luaL_newlibtable(L,l)	\
    141   lua_createtable(L, 0, sizeof(l)/sizeof((l)[0]) - 1)
    142 
    143 #define luaL_newlib(L,l)  \
    144   (luaL_checkversion(L), luaL_newlibtable(L,l), luaL_setfuncs(L,l,0))
    145 
    146 #define luaL_argcheck(L, cond,arg,extramsg)	\
    147 	((void)(luai_likely(cond) || luaL_argerror(L, (arg), (extramsg))))
    148 
    149 #define luaL_argexpected(L,cond,arg,tname)	\
    150 	((void)(luai_likely(cond) || luaL_typeerror(L, (arg), (tname))))
    151 
    152 #define luaL_checkstring(L,n)	(luaL_checklstring(L, (n), NULL))
    153 #define luaL_optstring(L,n,d)	(luaL_optlstring(L, (n), (d), NULL))
    154 
    155 #define luaL_typename(L,i)	lua_typename(L, lua_type(L,(i)))
    156 
    157 #ifndef _KERNEL
    158 #define luaL_dofile(L, fn) \
    159 	(luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
    160 #endif /* _KERNEL */
    161 
    162 #define luaL_dostring(L, s) \
    163 	(luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
    164 
    165 #define luaL_getmetatable(L,n)	(lua_getfield(L, LUA_REGISTRYINDEX, (n)))
    166 
    167 #define luaL_opt(L,f,n,d)	(lua_isnoneornil(L,(n)) ? (d) : f(L,(n)))
    168 
    169 #define luaL_loadbuffer(L,s,sz,n)	luaL_loadbufferx(L,s,sz,n,NULL)
    170 
    171 
    172 /*
    173 ** Perform arithmetic operations on lua_Integer values with wrap-around
    174 ** semantics, as the Lua core does.
    175 */
    176 #define luaL_intop(op,v1,v2)  \
    177 	((lua_Integer)((lua_Unsigned)(v1) op (lua_Unsigned)(v2)))
    178 
    179 
    180 /* push the value used to represent failure/error */
    181 #define luaL_pushfail(L)	lua_pushnil(L)
    182 
    183 
    184 /*
    185 ** Internal assertions for in-house debugging
    186 */
    187 #if !defined(lua_assert)
    188 
    189 #if defined LUAI_ASSERT
    190   #include <assert.h>
    191   #define lua_assert(c)		assert(c)
    192 #else
    193   #define lua_assert(c)		((void)0)
    194 #endif
    195 
    196 #endif
    197 
    198 
    199 
    200 /*
    201 ** {======================================================
    202 ** Generic Buffer manipulation
    203 ** =======================================================
    204 */
    205 
    206 struct luaL_Buffer {
    207   char *b;  /* buffer address */
    208   size_t size;  /* buffer size */
    209   size_t n;  /* number of characters in buffer */
    210   lua_State *L;
    211   union {
    212     LUAI_MAXALIGN;  /* ensure maximum alignment for buffer */
    213     char b[LUAL_BUFFERSIZE];  /* initial buffer */
    214   } init;
    215 };
    216 
    217 
    218 #define luaL_bufflen(bf)	((bf)->n)
    219 #define luaL_buffaddr(bf)	((bf)->b)
    220 
    221 
    222 #define luaL_addchar(B,c) \
    223   ((void)((B)->n < (B)->size || luaL_prepbuffsize((B), 1)), \
    224    ((B)->b[(B)->n++] = (c)))
    225 
    226 #define luaL_addsize(B,s)	((B)->n += (s))
    227 
    228 #define luaL_buffsub(B,s)	((B)->n -= (s))
    229 
    230 LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
    231 LUALIB_API char *(luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
    232 LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
    233 LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
    234 LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
    235 LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
    236 LUALIB_API void (luaL_pushresultsize) (luaL_Buffer *B, size_t sz);
    237 LUALIB_API char *(luaL_buffinitsize) (lua_State *L, luaL_Buffer *B, size_t sz);
    238 
    239 #define luaL_prepbuffer(B)	luaL_prepbuffsize(B, LUAL_BUFFERSIZE)
    240 
    241 /* }====================================================== */
    242 
    243 
    244 
    245 #ifndef _KERNEL
    246 /*
    247 ** {======================================================
    248 ** File handles for IO library
    249 ** =======================================================
    250 */
    251 
    252 /*
    253 ** A file handle is a userdata with metatable 'LUA_FILEHANDLE' and
    254 ** initial structure 'luaL_Stream' (it may contain other fields
    255 ** after that initial structure).
    256 */
    257 
    258 #define LUA_FILEHANDLE          "FILE*"
    259 
    260 
    261 typedef struct luaL_Stream {
    262   FILE *f;  /* stream (NULL for incompletely created streams) */
    263   lua_CFunction closef;  /* to close stream (NULL for closed streams) */
    264 } luaL_Stream;
    265 
    266 /* }====================================================== */
    267 #endif /* _KERNEL */
    268 
    269 
    270 #ifndef _KERNEL
    271 /*
    272 ** {==================================================================
    273 ** "Abstraction Layer" for basic report of messages and errors
    274 ** ===================================================================
    275 */
    276 
    277 /* print a string */
    278 #if !defined(lua_writestring)
    279 #define lua_writestring(s,l)   fwrite((s), sizeof(char), (l), stdout)
    280 #endif
    281 
    282 /* print a newline and flush the output */
    283 #if !defined(lua_writeline)
    284 #define lua_writeline()        (lua_writestring("\n", 1), fflush(stdout))
    285 #endif
    286 
    287 /* print an error message */
    288 #if !defined(lua_writestringerror)
    289 #define lua_writestringerror(s,p) \
    290         (fprintf(stderr, (s), (p)), fflush(stderr))
    291 #endif
    292 
    293 /* }================================================================== */
    294 #endif /* _KERNEL */
    295 
    296 
    297 /*
    298 ** {============================================================
    299 ** Compatibility with deprecated conversions
    300 ** =============================================================
    301 */
    302 #if defined(LUA_COMPAT_APIINTCASTS)
    303 
    304 #define luaL_checkunsigned(L,a)	((lua_Unsigned)luaL_checkinteger(L,a))
    305 #define luaL_optunsigned(L,a,d)	\
    306 	((lua_Unsigned)luaL_optinteger(L,a,(lua_Integer)(d)))
    307 
    308 #define luaL_checkint(L,n)	((int)luaL_checkinteger(L, (n)))
    309 #define luaL_optint(L,n,d)	((int)luaL_optinteger(L, (n), (d)))
    310 
    311 #define luaL_checklong(L,n)	((long)luaL_checkinteger(L, (n)))
    312 #define luaL_optlong(L,n,d)	((long)luaL_optinteger(L, (n), (d)))
    313 
    314 #endif
    315 /* }============================================================ */
    316 
    317 
    318 
    319 #endif
    320 
    321 
    322