lauxlib.h 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276
  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. #define luaL_newlibtable(L,l) \
  84. lua_createtable(L, 0, sizeof(l)/sizeof((l)[0]) - 1)
  85. #define luaL_newlib(L,l) \
  86. (luaL_checkversion(L), luaL_newlibtable(L,l), luaL_setfuncs(L,l,0))
  87. #define luaL_argcheck(L, cond,arg,extramsg) \
  88. ((void)((cond) || luaL_argerror(L, (arg), (extramsg))))
  89. #define luaL_argexpected(L,cond,arg,tname) \
  90. ((void)((cond) || luaL_typeerror(L, (arg), (tname))))
  91. #define luaL_checkstring(L,n) (luaL_checklstring(L, (n), NULL))
  92. #define luaL_optstring(L,n,d) (luaL_optlstring(L, (n), (d), NULL))
  93. #define luaL_typename(L,i) lua_typename(L, lua_type(L,(i)))
  94. #define luaL_dofile(L, fn) \
  95. (luaL_loadfile(L, fn) || lua_pcall(L, 0, LUA_MULTRET, 0))
  96. #define luaL_dostring(L, s) \
  97. (luaL_loadstring(L, s) || lua_pcall(L, 0, LUA_MULTRET, 0))
  98. #define luaL_getmetatable(L,n) (lua_getfield(L, LUA_REGISTRYINDEX, (n)))
  99. #define luaL_opt(L,f,n,d) (lua_isnoneornil(L,(n)) ? (d) : f(L,(n)))
  100. #define luaL_loadbuffer(L,s,sz,n) luaL_loadbufferx(L,s,sz,n,NULL)
  101. /* push the value used to represent failure/error */
  102. #define luaL_pushfail(L) lua_pushnil(L)
  103. /*
  104. ** {======================================================
  105. ** Generic Buffer manipulation
  106. ** =======================================================
  107. */
  108. struct luaL_Buffer {
  109. char *b; /* buffer address */
  110. size_t size; /* buffer size */
  111. size_t n; /* number of characters in buffer */
  112. lua_State *L;
  113. union {
  114. LUAI_MAXALIGN; /* ensure maximum alignment for buffer */
  115. char b[LUAL_BUFFERSIZE]; /* initial buffer */
  116. } init;
  117. };
  118. #define luaL_bufflen(bf) ((bf)->n)
  119. #define luaL_buffaddr(bf) ((bf)->b)
  120. #define luaL_addchar(B,c) \
  121. ((void)((B)->n < (B)->size || luaL_prepbuffsize((B), 1)), \
  122. ((B)->b[(B)->n++] = (c)))
  123. #define luaL_addsize(B,s) ((B)->n += (s))
  124. #define luaL_buffsub(B,s) ((B)->n -= (s))
  125. LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
  126. LUALIB_API char *(luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
  127. LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
  128. LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
  129. LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
  130. LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
  131. LUALIB_API void (luaL_pushresultsize) (luaL_Buffer *B, size_t sz);
  132. LUALIB_API char *(luaL_buffinitsize) (lua_State *L, luaL_Buffer *B, size_t sz);
  133. #define luaL_prepbuffer(B) luaL_prepbuffsize(B, LUAL_BUFFERSIZE)
  134. /* }====================================================== */
  135. /*
  136. ** {======================================================
  137. ** File handles for IO library
  138. ** =======================================================
  139. */
  140. /*
  141. ** A file handle is a userdata with metatable 'LUA_FILEHANDLE' and
  142. ** initial structure 'luaL_Stream' (it may contain other fields
  143. ** after that initial structure).
  144. */
  145. #define LUA_FILEHANDLE "FILE*"
  146. typedef struct luaL_Stream {
  147. FILE *f; /* stream (NULL for incompletely created streams) */
  148. lua_CFunction closef; /* to close stream (NULL for closed streams) */
  149. } luaL_Stream;
  150. /* }====================================================== */
  151. /*
  152. ** {==================================================================
  153. ** "Abstraction Layer" for basic report of messages and errors
  154. ** ===================================================================
  155. */
  156. /* print a string */
  157. #if !defined(lua_writestring)
  158. #define lua_writestring(s,l) fwrite((s), sizeof(char), (l), stdout)
  159. #endif
  160. /* print a newline and flush the output */
  161. #if !defined(lua_writeline)
  162. #define lua_writeline() (lua_writestring("\n", 1), fflush(stdout))
  163. #endif
  164. /* print an error message */
  165. #if !defined(lua_writestringerror)
  166. #define lua_writestringerror(s,p) \
  167. (fprintf(stderr, (s), (p)), fflush(stderr))
  168. #endif
  169. /* }================================================================== */
  170. /*
  171. ** {============================================================
  172. ** Compatibility with deprecated conversions
  173. ** =============================================================
  174. */
  175. #if defined(LUA_COMPAT_APIINTCASTS)
  176. #define luaL_checkunsigned(L,a) ((lua_Unsigned)luaL_checkinteger(L,a))
  177. #define luaL_optunsigned(L,a,d) \
  178. ((lua_Unsigned)luaL_optinteger(L,a,(lua_Integer)(d)))
  179. #define luaL_checkint(L,n) ((int)luaL_checkinteger(L, (n)))
  180. #define luaL_optint(L,n,d) ((int)luaL_optinteger(L, (n), (d)))
  181. #define luaL_checklong(L,n) ((long)luaL_checkinteger(L, (n)))
  182. #define luaL_optlong(L,n,d) ((long)luaL_optinteger(L, (n), (d)))
  183. #endif
  184. /* }============================================================ */
  185. #endif