lauxlib.h 8.9 KB

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