lauxlib.h 8.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270
  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. /*
  102. ** {======================================================
  103. ** Generic Buffer manipulation
  104. ** =======================================================
  105. */
  106. struct luaL_Buffer {
  107. char *b; /* buffer address */
  108. size_t size; /* buffer size */
  109. size_t n; /* number of characters in buffer */
  110. lua_State *L;
  111. union {
  112. LUAI_MAXALIGN; /* ensure maximum alignment for buffer */
  113. char b[LUAL_BUFFERSIZE]; /* initial buffer */
  114. } init;
  115. };
  116. #define luaL_bufflen(bf) ((bf)->n)
  117. #define luaL_buffaddr(bf) ((bf)->b)
  118. #define luaL_addchar(B,c) \
  119. ((void)((B)->n < (B)->size || luaL_prepbuffsize((B), 1)), \
  120. ((B)->b[(B)->n++] = (c)))
  121. #define luaL_addsize(B,s) ((B)->n += (s))
  122. LUALIB_API void (luaL_buffinit) (lua_State *L, luaL_Buffer *B);
  123. LUALIB_API char *(luaL_prepbuffsize) (luaL_Buffer *B, size_t sz);
  124. LUALIB_API void (luaL_addlstring) (luaL_Buffer *B, const char *s, size_t l);
  125. LUALIB_API void (luaL_addstring) (luaL_Buffer *B, const char *s);
  126. LUALIB_API void (luaL_addvalue) (luaL_Buffer *B);
  127. LUALIB_API void (luaL_pushresult) (luaL_Buffer *B);
  128. LUALIB_API void (luaL_pushresultsize) (luaL_Buffer *B, size_t sz);
  129. LUALIB_API char *(luaL_buffinitsize) (lua_State *L, luaL_Buffer *B, size_t sz);
  130. #define luaL_prepbuffer(B) luaL_prepbuffsize(B, LUAL_BUFFERSIZE)
  131. /* }====================================================== */
  132. /*
  133. ** {======================================================
  134. ** File handles for IO library
  135. ** =======================================================
  136. */
  137. /*
  138. ** A file handle is a userdata with metatable 'LUA_FILEHANDLE' and
  139. ** initial structure 'luaL_Stream' (it may contain other fields
  140. ** after that initial structure).
  141. */
  142. #define LUA_FILEHANDLE "FILE*"
  143. typedef struct luaL_Stream {
  144. FILE *f; /* stream (NULL for incompletely created streams) */
  145. lua_CFunction closef; /* to close stream (NULL for closed streams) */
  146. } luaL_Stream;
  147. /* }====================================================== */
  148. /*
  149. ** {==================================================================
  150. ** "Abstraction Layer" for basic report of messages and errors
  151. ** ===================================================================
  152. */
  153. /* print a string */
  154. #if !defined(lua_writestring)
  155. #define lua_writestring(s,l) fwrite((s), sizeof(char), (l), stdout)
  156. #endif
  157. /* print a newline and flush the output */
  158. #if !defined(lua_writeline)
  159. #define lua_writeline() (lua_writestring("\n", 1), fflush(stdout))
  160. #endif
  161. /* print an error message */
  162. #if !defined(lua_writestringerror)
  163. #define lua_writestringerror(s,p) \
  164. (fprintf(stderr, (s), (p)), fflush(stderr))
  165. #endif
  166. /* }================================================================== */
  167. /*
  168. ** {============================================================
  169. ** Compatibility with deprecated conversions
  170. ** =============================================================
  171. */
  172. #if defined(LUA_COMPAT_APIINTCASTS)
  173. #define luaL_checkunsigned(L,a) ((lua_Unsigned)luaL_checkinteger(L,a))
  174. #define luaL_optunsigned(L,a,d) \
  175. ((lua_Unsigned)luaL_optinteger(L,a,(lua_Integer)(d)))
  176. #define luaL_checkint(L,n) ((int)luaL_checkinteger(L, (n)))
  177. #define luaL_optint(L,n,d) ((int)luaL_optinteger(L, (n), (d)))
  178. #define luaL_checklong(L,n) ((long)luaL_checkinteger(L, (n)))
  179. #define luaL_optlong(L,n,d) ((long)luaL_optinteger(L, (n), (d)))
  180. #endif
  181. /* }============================================================ */
  182. #endif