lauxlib.h 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301
  1. /*
  2. ** $Id: lauxlib.h $
  3. ** Auxiliary functions for building Lua libraries
  4. ** See Copyright Notice in lua.h
  5. */
  6. #ifndef lauxlib_h
  7. #define lauxlib_h
  8. #include <stddef.h>
  9. #include <stdio.h>
  10. #include "luaconf.h"
  11. #include "lua.h"
  12. /* global table */
  13. #define LUA_GNAME "_G"
  14. typedef struct luaL_Buffer luaL_Buffer;
  15. /* extra error code for 'luaL_loadfilex' */
  16. #define LUA_ERRFILE (LUA_ERRERR+1)
  17. /* key, in the registry, for table of loaded modules */
  18. #define LUA_LOADED_TABLE "_LOADED"
  19. /* key, in the registry, for table of preloaded loaders */
  20. #define LUA_PRELOAD_TABLE "_PRELOAD"
  21. typedef struct luaL_Reg {
  22. const char *name;
  23. lua_CFunction func;
  24. } luaL_Reg;
  25. #define LUAL_NUMSIZES (sizeof(lua_Integer)*16 + sizeof(lua_Number))
  26. LUALIB_API void (luaL_checkversion_) (lua_State *L, lua_Number ver, size_t sz);
  27. #define luaL_checkversion(L) \
  28. luaL_checkversion_(L, LUA_VERSION_NUM, LUAL_NUMSIZES)
  29. LUALIB_API int (luaL_getmetafield) (lua_State *L, int obj, const char *e);
  30. LUALIB_API int (luaL_callmeta) (lua_State *L, int obj, const char *e);
  31. LUALIB_API const char *(luaL_tolstring) (lua_State *L, int idx, size_t *len);
  32. LUALIB_API int (luaL_argerror) (lua_State *L, int arg, const char *extramsg);
  33. LUALIB_API int (luaL_typeerror) (lua_State *L, int arg, const char *tname);
  34. LUALIB_API const char *(luaL_checklstring) (lua_State *L, int arg,
  35. size_t *l);
  36. LUALIB_API const char *(luaL_optlstring) (lua_State *L, int arg,
  37. const char *def, size_t *l);
  38. LUALIB_API lua_Number (luaL_checknumber) (lua_State *L, int arg);
  39. LUALIB_API lua_Number (luaL_optnumber) (lua_State *L, int arg, lua_Number def);
  40. LUALIB_API lua_Integer (luaL_checkinteger) (lua_State *L, int arg);
  41. LUALIB_API lua_Integer (luaL_optinteger) (lua_State *L, int arg,
  42. lua_Integer def);
  43. LUALIB_API void (luaL_checkstack) (lua_State *L, int sz, const char *msg);
  44. LUALIB_API void (luaL_checktype) (lua_State *L, int arg, int t);
  45. LUALIB_API void (luaL_checkany) (lua_State *L, int arg);
  46. LUALIB_API int (luaL_newmetatable) (lua_State *L, const char *tname);
  47. LUALIB_API void (luaL_setmetatable) (lua_State *L, const char *tname);
  48. LUALIB_API void *(luaL_testudata) (lua_State *L, int ud, const char *tname);
  49. LUALIB_API void *(luaL_checkudata) (lua_State *L, int ud, const char *tname);
  50. LUALIB_API void (luaL_where) (lua_State *L, int lvl);
  51. LUALIB_API int (luaL_error) (lua_State *L, const char *fmt, ...);
  52. LUALIB_API int (luaL_checkoption) (lua_State *L, int arg, const char *def,
  53. const char *const lst[]);
  54. LUALIB_API int (luaL_fileresult) (lua_State *L, int stat, const char *fname);
  55. LUALIB_API int (luaL_execresult) (lua_State *L, int stat);
  56. /* predefined references */
  57. #define LUA_NOREF (-2)
  58. #define LUA_REFNIL (-1)
  59. LUALIB_API int (luaL_ref) (lua_State *L, int t);
  60. LUALIB_API void (luaL_unref) (lua_State *L, int t, int ref);
  61. LUALIB_API int (luaL_loadfilex) (lua_State *L, const char *filename,
  62. const char *mode);
  63. #define luaL_loadfile(L,f) luaL_loadfilex(L,f,NULL)
  64. LUALIB_API int (luaL_loadbufferx) (lua_State *L, const char *buff, size_t sz,
  65. const char *name, const char *mode);
  66. LUALIB_API int (luaL_loadstring) (lua_State *L, const char *s);
  67. LUALIB_API lua_State *(luaL_newstate) (void);
  68. LUALIB_API lua_Integer (luaL_len) (lua_State *L, int idx);
  69. LUALIB_API void (luaL_addgsub) (luaL_Buffer *b, const char *s,
  70. const char *p, const char *r);
  71. LUALIB_API const char *(luaL_gsub) (lua_State *L, const char *s,
  72. const char *p, const char *r);
  73. LUALIB_API void (luaL_setfuncs) (lua_State *L, const luaL_Reg *l, int nup);
  74. LUALIB_API int (luaL_getsubtable) (lua_State *L, int idx, const char *fname);
  75. LUALIB_API void (luaL_traceback) (lua_State *L, lua_State *L1,
  76. const char *msg, int level);
  77. LUALIB_API void (luaL_requiref) (lua_State *L, const char *modname,
  78. lua_CFunction openf, int glb);
  79. /*
  80. ** ===============================================================
  81. ** some useful macros
  82. ** ===============================================================
  83. */
  84. #define luaL_newlibtable(L,l) \
  85. lua_createtable(L, 0, sizeof(l)/sizeof((l)[0]) - 1)
  86. #define luaL_newlib(L,l) \
  87. (luaL_checkversion(L), luaL_newlibtable(L,l), luaL_setfuncs(L,l,0))
  88. #define luaL_argcheck(L, cond,arg,extramsg) \
  89. ((void)(luai_likely(cond) || luaL_argerror(L, (arg), (extramsg))))
  90. #define luaL_argexpected(L,cond,arg,tname) \
  91. ((void)(luai_likely(cond) || luaL_typeerror(L, (arg), (tname))))
  92. #define luaL_checkstring(L,n) (luaL_checklstring(L, (n), NULL))
  93. #define luaL_optstring(L,n,d) (luaL_optlstring(L, (n), (d), NULL))
  94. #define luaL_typename(L,i) lua_typename(L, lua_type(L,(i)))
  95. #define luaL_dofile(L, fn) \
  96. (luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
  97. #define luaL_dostring(L, s) \
  98. (luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
  99. #define luaL_getmetatable(L,n) (lua_getfield(L, LUA_REGISTRYINDEX, (n)))
  100. #define luaL_opt(L,f,n,d) (lua_isnoneornil(L,(n)) ? (d) : f(L,(n)))
  101. #define luaL_loadbuffer(L,s,sz,n) luaL_loadbufferx(L,s,sz,n,NULL)
  102. /*
  103. ** Perform arithmetic operations on lua_Integer values with wrap-around
  104. ** semantics, as the Lua core does.
  105. */
  106. #define luaL_intop(op,v1,v2) \
  107. ((lua_Integer)((lua_Unsigned)(v1) op (lua_Unsigned)(v2)))
  108. /* push the value used to represent failure/error */
  109. #define luaL_pushfail(L) lua_pushnil(L)
  110. /*
  111. ** Internal assertions for in-house debugging
  112. */
  113. #if !defined(lua_assert)
  114. #if defined LUAI_ASSERT
  115. #include <assert.h>
  116. #define lua_assert(c) assert(c)
  117. #else
  118. #define lua_assert(c) ((void)0)
  119. #endif
  120. #endif
  121. /*
  122. ** {======================================================
  123. ** Generic Buffer manipulation
  124. ** =======================================================
  125. */
  126. struct luaL_Buffer {
  127. char *b; /* buffer address */
  128. size_t size; /* buffer size */
  129. size_t n; /* number of characters in buffer */
  130. lua_State *L;
  131. union {
  132. LUAI_MAXALIGN; /* ensure maximum alignment for buffer */
  133. char b[LUAL_BUFFERSIZE]; /* initial buffer */
  134. } init;
  135. };
  136. #define luaL_bufflen(bf) ((bf)->n)
  137. #define luaL_buffaddr(bf) ((bf)->b)
  138. #define luaL_addchar(B,c) \
  139. ((void)((B)->n < (B)->size || luaL_prepbuffsize((B), 1)), \
  140. ((B)->b[(B)->n++] = (c)))
  141. #define luaL_addsize(B,s) ((B)->n += (s))
  142. #define luaL_buffsub(B,s) ((B)->n -= (s))
  143. LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
  144. LUALIB_API char *(luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
  145. LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
  146. LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
  147. LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
  148. LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
  149. LUALIB_API void (luaL_pushresultsize) (luaL_Buffer *B, size_t sz);
  150. LUALIB_API char *(luaL_buffinitsize) (lua_State *L, luaL_Buffer *B, size_t sz);
  151. #define luaL_prepbuffer(B) luaL_prepbuffsize(B, LUAL_BUFFERSIZE)
  152. /* }====================================================== */
  153. /*
  154. ** {======================================================
  155. ** File handles for IO library
  156. ** =======================================================
  157. */
  158. /*
  159. ** A file handle is a userdata with metatable 'LUA_FILEHANDLE' and
  160. ** initial structure 'luaL_Stream' (it may contain other fields
  161. ** after that initial structure).
  162. */
  163. #define LUA_FILEHANDLE "FILE*"
  164. typedef struct luaL_Stream {
  165. FILE *f; /* stream (NULL for incompletely created streams) */
  166. lua_CFunction closef; /* to close stream (NULL for closed streams) */
  167. } luaL_Stream;
  168. /* }====================================================== */
  169. /*
  170. ** {==================================================================
  171. ** "Abstraction Layer" for basic report of messages and errors
  172. ** ===================================================================
  173. */
  174. /* print a string */
  175. #if !defined(lua_writestring)
  176. #define lua_writestring(s,l) fwrite((s), sizeof(char), (l), stdout)
  177. #endif
  178. /* print a newline and flush the output */
  179. #if !defined(lua_writeline)
  180. #define lua_writeline() (lua_writestring("\n", 1), fflush(stdout))
  181. #endif
  182. /* print an error message */
  183. #if !defined(lua_writestringerror)
  184. #define lua_writestringerror(s,p) \
  185. (fprintf(stderr, (s), (p)), fflush(stderr))
  186. #endif
  187. /* }================================================================== */
  188. /*
  189. ** {============================================================
  190. ** Compatibility with deprecated conversions
  191. ** =============================================================
  192. */
  193. #if defined(LUA_COMPAT_APIINTCASTS)
  194. #define luaL_checkunsigned(L,a) ((lua_Unsigned)luaL_checkinteger(L,a))
  195. #define luaL_optunsigned(L,a,d) \
  196. ((lua_Unsigned)luaL_optinteger(L,a,(lua_Integer)(d)))
  197. #define luaL_checkint(L,n) ((int)luaL_checkinteger(L, (n)))
  198. #define luaL_optint(L,n,d) ((int)luaL_optinteger(L, (n), (d)))
  199. #define luaL_checklong(L,n) ((long)luaL_checkinteger(L, (n)))
  200. #define luaL_optlong(L,n,d) ((long)luaL_optinteger(L, (n), (d)))
  201. #endif
  202. /* }============================================================ */
  203. #endif