lauxlib.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793
  1. /*
  2. ** $Id: lauxlib.c,v 1.201 2010/02/18 19:37:57 roberto Exp roberto $
  3. ** Auxiliary functions for building Lua libraries
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <errno.h>
  7. #include <stdarg.h>
  8. #include <stdio.h>
  9. #include <stdlib.h>
  10. #include <string.h>
  11. /* This file uses only the official API of Lua.
  12. ** Any function declared here could be written as an application function.
  13. */
  14. #define lauxlib_c
  15. #define LUA_LIB
  16. #include "lua.h"
  17. #include "lauxlib.h"
  18. /* convert a stack index to positive */
  19. #define abs_index(L, i) ((i) > 0 || (i) <= LUA_REGISTRYINDEX ? (i) : \
  20. lua_gettop(L) + (i) + 1)
  21. /*
  22. ** {======================================================
  23. ** Traceback
  24. ** =======================================================
  25. */
  26. #define LEVELS1 12 /* size of the first part of the stack */
  27. #define LEVELS2 10 /* size of the second part of the stack */
  28. /*
  29. ** search for 'objidx' in table at index -1.
  30. ** return 1 + string at top if find a good name.
  31. */
  32. static int findfield (lua_State *L, int objidx, int level) {
  33. int found = 0;
  34. if (level == 0 || !lua_istable(L, -1))
  35. return 0; /* not found */
  36. lua_pushnil(L); /* start 'next' loop */
  37. while (!found && lua_next(L, -2)) { /* for each pair in table */
  38. if (lua_type(L, -2) == LUA_TSTRING) { /* ignore non-string keys */
  39. if (lua_rawequal(L, objidx, -1)) { /* found object? */
  40. lua_pop(L, 1); /* remove value (but keep name) */
  41. return 1;
  42. }
  43. else if (findfield(L, objidx, level - 1)) { /* try recursively */
  44. lua_remove(L, -2); /* remove table (but keep name) */
  45. lua_pushliteral(L, ".");
  46. lua_insert(L, -2); /* place '.' between the two names */
  47. lua_concat(L, 3);
  48. return 1;
  49. }
  50. }
  51. lua_pop(L, 1); /* remove value */
  52. }
  53. return 0; /* not found */
  54. }
  55. static int pushglobalfuncname (lua_State *L, lua_Debug *ar) {
  56. int top = lua_gettop(L);
  57. lua_getinfo(L, "f", ar); /* push function */
  58. lua_pushglobaltable(L);
  59. if (findfield(L, top + 1, 2)) {
  60. lua_copy(L, -1, top + 1); /* move name to proper place */
  61. lua_pop(L, 2); /* remove pushed values */
  62. return 1;
  63. }
  64. else {
  65. lua_settop(L, top); /* remove function and global table */
  66. return 0;
  67. }
  68. }
  69. static void pushfuncname (lua_State *L, lua_Debug *ar) {
  70. if (*ar->namewhat != '\0') /* is there a name? */
  71. lua_pushfstring(L, "function " LUA_QS, ar->name);
  72. else if (*ar->what == 'm') /* main? */
  73. lua_pushfstring(L, "main chunk");
  74. else if (*ar->what == 'C' || *ar->what == 't') {
  75. if (pushglobalfuncname(L, ar)) {
  76. lua_pushfstring(L, "function " LUA_QS, lua_tostring(L, -1));
  77. lua_remove(L, -2); /* remove name */
  78. }
  79. else
  80. lua_pushliteral(L, "?");
  81. }
  82. else
  83. lua_pushfstring(L, "function <%s:%d>", ar->short_src, ar->linedefined);
  84. }
  85. static int countlevels (lua_State *L) {
  86. lua_Debug ar;
  87. int li = 1, le = 1;
  88. /* find an upper bound */
  89. while (lua_getstack(L, le, &ar)) { li = le; le *= 2; }
  90. /* do a binary search */
  91. while (li < le) {
  92. int m = (li + le)/2;
  93. if (lua_getstack(L, m, &ar)) li = m + 1;
  94. else le = m;
  95. }
  96. return le - 1;
  97. }
  98. LUALIB_API void luaL_traceback (lua_State *L, lua_State *L1,
  99. const char *msg, int level) {
  100. lua_Debug ar;
  101. int top = lua_gettop(L);
  102. int numlevels = countlevels(L1);
  103. int mark = (numlevels > LEVELS1 + LEVELS2) ? LEVELS1 : 0;
  104. if (msg) lua_pushfstring(L, "%s\n", msg);
  105. lua_pushliteral(L, "stack traceback:");
  106. while (lua_getstack(L1, level++, &ar)) {
  107. if (level == mark) { /* too many levels? */
  108. lua_pushliteral(L, "\n\t..."); /* add a '...' */
  109. level = numlevels - LEVELS2; /* and skip to last ones */
  110. }
  111. else {
  112. lua_getinfo(L1, "Slnt", &ar);
  113. lua_pushfstring(L, "\n\t%s:", ar.short_src);
  114. if (ar.currentline > 0)
  115. lua_pushfstring(L, "%d:", ar.currentline);
  116. lua_pushliteral(L, " in ");
  117. pushfuncname(L, &ar);
  118. if (ar.istailcall)
  119. lua_pushliteral(L, "\n\t(...tail calls...)");
  120. lua_concat(L, lua_gettop(L) - top);
  121. }
  122. }
  123. lua_concat(L, lua_gettop(L) - top);
  124. }
  125. /* }====================================================== */
  126. /*
  127. ** {======================================================
  128. ** Error-report functions
  129. ** =======================================================
  130. */
  131. LUALIB_API int luaL_argerror (lua_State *L, int narg, const char *extramsg) {
  132. lua_Debug ar;
  133. if (!lua_getstack(L, 0, &ar)) /* no stack frame? */
  134. return luaL_error(L, "bad argument #%d (%s)", narg, extramsg);
  135. lua_getinfo(L, "n", &ar);
  136. if (strcmp(ar.namewhat, "method") == 0) {
  137. narg--; /* do not count `self' */
  138. if (narg == 0) /* error is in the self argument itself? */
  139. return luaL_error(L, "calling " LUA_QS " on bad self", ar.name);
  140. }
  141. if (ar.name == NULL)
  142. ar.name = (pushglobalfuncname(L, &ar)) ? lua_tostring(L, -1) : "?";
  143. return luaL_error(L, "bad argument #%d to " LUA_QS " (%s)",
  144. narg, ar.name, extramsg);
  145. }
  146. LUALIB_API int luaL_typeerror (lua_State *L, int narg, const char *tname) {
  147. const char *msg = lua_pushfstring(L, "%s expected, got %s",
  148. tname, luaL_typename(L, narg));
  149. return luaL_argerror(L, narg, msg);
  150. }
  151. static void tag_error (lua_State *L, int narg, int tag) {
  152. luaL_typeerror(L, narg, lua_typename(L, tag));
  153. }
  154. LUALIB_API void luaL_where (lua_State *L, int level) {
  155. lua_Debug ar;
  156. if (lua_getstack(L, level, &ar)) { /* check function at level */
  157. lua_getinfo(L, "Sl", &ar); /* get info about it */
  158. if (ar.currentline > 0) { /* is there info? */
  159. lua_pushfstring(L, "%s:%d: ", ar.short_src, ar.currentline);
  160. return;
  161. }
  162. }
  163. lua_pushliteral(L, ""); /* else, no information available... */
  164. }
  165. LUALIB_API int luaL_error (lua_State *L, const char *fmt, ...) {
  166. va_list argp;
  167. va_start(argp, fmt);
  168. luaL_where(L, 1);
  169. lua_pushvfstring(L, fmt, argp);
  170. va_end(argp);
  171. lua_concat(L, 2);
  172. return lua_error(L);
  173. }
  174. /* }====================================================== */
  175. /*
  176. ** {======================================================
  177. ** Userdata's metatable manipulation
  178. ** =======================================================
  179. */
  180. LUALIB_API int luaL_newmetatable (lua_State *L, const char *tname) {
  181. lua_getfield(L, LUA_REGISTRYINDEX, tname); /* get registry.name */
  182. if (!lua_isnil(L, -1)) /* name already in use? */
  183. return 0; /* leave previous value on top, but return 0 */
  184. lua_pop(L, 1);
  185. lua_newtable(L); /* create metatable */
  186. lua_pushvalue(L, -1);
  187. lua_setfield(L, LUA_REGISTRYINDEX, tname); /* registry.name = metatable */
  188. return 1;
  189. }
  190. LUALIB_API void *luaL_testudata (lua_State *L, int ud, const char *tname) {
  191. void *p = lua_touserdata(L, ud);
  192. if (p != NULL) { /* value is a userdata? */
  193. if (lua_getmetatable(L, ud)) { /* does it have a metatable? */
  194. lua_getfield(L, LUA_REGISTRYINDEX, tname); /* get correct metatable */
  195. if (!lua_rawequal(L, -1, -2)) /* not the same? */
  196. p = NULL; /* value is a userdata with wrong metatable */
  197. lua_pop(L, 2); /* remove both metatables */
  198. return p;
  199. }
  200. }
  201. return NULL; /* value is not a userdata with a metatable */
  202. }
  203. LUALIB_API void *luaL_checkudata (lua_State *L, int ud, const char *tname) {
  204. void *p = luaL_testudata(L, ud, tname);
  205. if (p == NULL) luaL_typeerror(L, ud, tname);
  206. return p;
  207. }
  208. /* }====================================================== */
  209. /*
  210. ** {======================================================
  211. ** Argument check functions
  212. ** =======================================================
  213. */
  214. LUALIB_API int luaL_checkoption (lua_State *L, int narg, const char *def,
  215. const char *const lst[]) {
  216. const char *name = (def) ? luaL_optstring(L, narg, def) :
  217. luaL_checkstring(L, narg);
  218. int i;
  219. for (i=0; lst[i]; i++)
  220. if (strcmp(lst[i], name) == 0)
  221. return i;
  222. return luaL_argerror(L, narg,
  223. lua_pushfstring(L, "invalid option " LUA_QS, name));
  224. }
  225. LUALIB_API void luaL_checkstack (lua_State *L, int space, const char *msg) {
  226. if (!lua_checkstack(L, space)) {
  227. if (msg)
  228. luaL_error(L, "stack overflow (%s)", msg);
  229. else
  230. luaL_error(L, "stack overflow");
  231. }
  232. }
  233. LUALIB_API void luaL_checktype (lua_State *L, int narg, int t) {
  234. if (lua_type(L, narg) != t)
  235. tag_error(L, narg, t);
  236. }
  237. LUALIB_API void luaL_checkany (lua_State *L, int narg) {
  238. if (lua_type(L, narg) == LUA_TNONE)
  239. luaL_argerror(L, narg, "value expected");
  240. }
  241. LUALIB_API const char *luaL_checklstring (lua_State *L, int narg, size_t *len) {
  242. const char *s = lua_tolstring(L, narg, len);
  243. if (!s) tag_error(L, narg, LUA_TSTRING);
  244. return s;
  245. }
  246. LUALIB_API const char *luaL_optlstring (lua_State *L, int narg,
  247. const char *def, size_t *len) {
  248. if (lua_isnoneornil(L, narg)) {
  249. if (len)
  250. *len = (def ? strlen(def) : 0);
  251. return def;
  252. }
  253. else return luaL_checklstring(L, narg, len);
  254. }
  255. LUALIB_API lua_Number luaL_checknumber (lua_State *L, int narg) {
  256. lua_Number d = lua_tonumber(L, narg);
  257. if (d == 0 && !lua_isnumber(L, narg)) /* avoid extra test when d is not 0 */
  258. tag_error(L, narg, LUA_TNUMBER);
  259. return d;
  260. }
  261. LUALIB_API lua_Number luaL_optnumber (lua_State *L, int narg, lua_Number def) {
  262. return luaL_opt(L, luaL_checknumber, narg, def);
  263. }
  264. LUALIB_API lua_Integer luaL_checkinteger (lua_State *L, int narg) {
  265. lua_Integer d = lua_tointeger(L, narg);
  266. if (d == 0 && !lua_isnumber(L, narg)) /* avoid extra test when d is not 0 */
  267. tag_error(L, narg, LUA_TNUMBER);
  268. return d;
  269. }
  270. LUALIB_API lua_Integer luaL_optinteger (lua_State *L, int narg,
  271. lua_Integer def) {
  272. return luaL_opt(L, luaL_checkinteger, narg, def);
  273. }
  274. /* }====================================================== */
  275. /*
  276. ** {======================================================
  277. ** Generic Buffer manipulation
  278. ** =======================================================
  279. */
  280. #define bufflen(B) ((B)->p - (B)->buffer)
  281. #define bufffree(B) ((size_t)(LUAL_BUFFERSIZE - bufflen(B)))
  282. #define LIMIT (LUA_MINSTACK/2)
  283. static int emptybuffer (luaL_Buffer *B) {
  284. size_t l = bufflen(B);
  285. if (l == 0) return 0; /* put nothing on stack */
  286. else {
  287. lua_pushlstring(B->L, B->buffer, l);
  288. B->p = B->buffer;
  289. B->lvl++;
  290. return 1;
  291. }
  292. }
  293. static void adjuststack (luaL_Buffer *B) {
  294. if (B->lvl > 1) {
  295. lua_State *L = B->L;
  296. int toget = 1; /* number of levels to concat */
  297. size_t toplen = lua_rawlen(L, -1);
  298. do {
  299. size_t l = lua_rawlen(L, -(toget+1));
  300. if (B->lvl - toget + 1 >= LIMIT || toplen > l) {
  301. toplen += l;
  302. toget++;
  303. }
  304. else break;
  305. } while (toget < B->lvl);
  306. lua_concat(L, toget);
  307. B->lvl = B->lvl - toget + 1;
  308. }
  309. }
  310. LUALIB_API char *luaL_prepbuffer (luaL_Buffer *B) {
  311. if (emptybuffer(B))
  312. adjuststack(B);
  313. return B->buffer;
  314. }
  315. LUALIB_API void luaL_addlstring (luaL_Buffer *B, const char *s, size_t l) {
  316. while (l) {
  317. size_t space = bufffree(B);
  318. if (space == 0) {
  319. luaL_prepbuffer(B);
  320. lua_assert(bufffree(B) == LUAL_BUFFERSIZE);
  321. space = LUAL_BUFFERSIZE;
  322. }
  323. if (space > l) space = l;
  324. memcpy(B->p, s, space);
  325. B->p += space;
  326. s += space;
  327. l -= space;
  328. }
  329. }
  330. LUALIB_API void luaL_addstring (luaL_Buffer *B, const char *s) {
  331. luaL_addlstring(B, s, strlen(s));
  332. }
  333. LUALIB_API void luaL_pushresult (luaL_Buffer *B) {
  334. emptybuffer(B);
  335. lua_concat(B->L, B->lvl);
  336. B->lvl = 1;
  337. }
  338. LUALIB_API void luaL_addvalue (luaL_Buffer *B) {
  339. lua_State *L = B->L;
  340. size_t vl;
  341. const char *s = lua_tolstring(L, -1, &vl);
  342. if (vl <= bufffree(B)) { /* fit into buffer? */
  343. memcpy(B->p, s, vl); /* put it there */
  344. B->p += vl;
  345. lua_pop(L, 1); /* remove from stack */
  346. }
  347. else {
  348. if (emptybuffer(B))
  349. lua_insert(L, -2); /* put buffer before new value */
  350. B->lvl++; /* add new value into B stack */
  351. adjuststack(B);
  352. }
  353. }
  354. LUALIB_API void luaL_buffinit (lua_State *L, luaL_Buffer *B) {
  355. luaL_checkstack(L, LIMIT + LUA_MINSTACK, "no space for new buffer");
  356. B->L = L;
  357. B->p = B->buffer;
  358. B->lvl = 0;
  359. }
  360. /* }====================================================== */
  361. /*
  362. ** {======================================================
  363. ** Reference system
  364. ** =======================================================
  365. */
  366. /* number of prereserved references (for internal use) */
  367. #define FREELIST_REF (LUA_RIDX_LAST + 1) /* free list of references */
  368. LUALIB_API int luaL_ref (lua_State *L, int t) {
  369. int ref;
  370. t = abs_index(L, t);
  371. if (lua_isnil(L, -1)) {
  372. lua_pop(L, 1); /* remove from stack */
  373. return LUA_REFNIL; /* `nil' has a unique fixed reference */
  374. }
  375. lua_rawgeti(L, t, FREELIST_REF); /* get first free element */
  376. ref = (int)lua_tointeger(L, -1); /* ref = t[FREELIST_REF] */
  377. lua_pop(L, 1); /* remove it from stack */
  378. if (ref != 0) { /* any free element? */
  379. lua_rawgeti(L, t, ref); /* remove it from list */
  380. lua_rawseti(L, t, FREELIST_REF); /* (t[FREELIST_REF] = t[ref]) */
  381. }
  382. else { /* no free elements */
  383. ref = (int)lua_rawlen(L, t) + 1; /* get a new reference */
  384. if (ref == FREELIST_REF) { /* FREELIST_REF not initialized? */
  385. lua_pushinteger(L, 0);
  386. lua_rawseti(L, t, FREELIST_REF);
  387. ref = FREELIST_REF + 1;
  388. }
  389. }
  390. lua_rawseti(L, t, ref);
  391. return ref;
  392. }
  393. LUALIB_API void luaL_unref (lua_State *L, int t, int ref) {
  394. if (ref >= 0) {
  395. t = abs_index(L, t);
  396. lua_rawgeti(L, t, FREELIST_REF);
  397. lua_rawseti(L, t, ref); /* t[ref] = t[FREELIST_REF] */
  398. lua_pushinteger(L, ref);
  399. lua_rawseti(L, t, FREELIST_REF); /* t[FREELIST_REF] = ref */
  400. }
  401. }
  402. /* }====================================================== */
  403. /*
  404. ** {======================================================
  405. ** Load functions
  406. ** =======================================================
  407. */
  408. typedef struct LoadF {
  409. int first;
  410. FILE *f;
  411. char buff[LUAL_BUFFERSIZE];
  412. } LoadF;
  413. static const char *getF (lua_State *L, void *ud, size_t *size) {
  414. LoadF *lf = (LoadF *)ud;
  415. (void)L;
  416. if (lf->first != EOF) {
  417. *size = 1;
  418. lf->buff[0] = (char)lf->first;
  419. lf->first = EOF;
  420. }
  421. else {
  422. /* 'fread' can return > 0 *and* set the EOF flag. If next call to
  423. 'getF' called 'fread', it might still wait for user input.
  424. The next check avoids this problem. */
  425. if (feof(lf->f)) return NULL;
  426. *size = fread(lf->buff, 1, sizeof(lf->buff), lf->f);
  427. }
  428. return lf->buff;
  429. }
  430. static int errfile (lua_State *L, const char *what, int fnameindex) {
  431. const char *serr = strerror(errno);
  432. const char *filename = lua_tostring(L, fnameindex) + 1;
  433. lua_pushfstring(L, "cannot %s %s: %s", what, filename, serr);
  434. lua_remove(L, fnameindex);
  435. return LUA_ERRFILE;
  436. }
  437. LUALIB_API int luaL_loadfile (lua_State *L, const char *filename) {
  438. LoadF lf;
  439. int status, readstatus;
  440. int c;
  441. int fnameindex = lua_gettop(L) + 1; /* index of filename on the stack */
  442. if (filename == NULL) {
  443. lua_pushliteral(L, "=stdin");
  444. lf.f = stdin;
  445. }
  446. else {
  447. lua_pushfstring(L, "@%s", filename);
  448. lf.f = fopen(filename, "r");
  449. if (lf.f == NULL) return errfile(L, "open", fnameindex);
  450. }
  451. c = getc(lf.f);
  452. if (c == '#') { /* Unix exec. file? */
  453. while ((c = getc(lf.f)) != EOF && c != '\n') ; /* skip first line */
  454. }
  455. else if (c == LUA_SIGNATURE[0] && filename) { /* binary file? */
  456. lf.f = freopen(filename, "rb", lf.f); /* reopen in binary mode */
  457. if (lf.f == NULL) return errfile(L, "reopen", fnameindex);
  458. }
  459. lf.first = c; /* 'c' is the first character of the stream */
  460. status = lua_load(L, getF, &lf, lua_tostring(L, -1));
  461. readstatus = ferror(lf.f);
  462. if (filename) fclose(lf.f); /* close file (even in case of errors) */
  463. if (readstatus) {
  464. lua_settop(L, fnameindex); /* ignore results from `lua_load' */
  465. return errfile(L, "read", fnameindex);
  466. }
  467. lua_remove(L, fnameindex);
  468. return status;
  469. }
  470. typedef struct LoadS {
  471. const char *s;
  472. size_t size;
  473. } LoadS;
  474. static const char *getS (lua_State *L, void *ud, size_t *size) {
  475. LoadS *ls = (LoadS *)ud;
  476. (void)L;
  477. if (ls->size == 0) return NULL;
  478. *size = ls->size;
  479. ls->size = 0;
  480. return ls->s;
  481. }
  482. LUALIB_API int luaL_loadbuffer (lua_State *L, const char *buff, size_t size,
  483. const char *name) {
  484. LoadS ls;
  485. ls.s = buff;
  486. ls.size = size;
  487. return lua_load(L, getS, &ls, name);
  488. }
  489. LUALIB_API int luaL_loadstring (lua_State *L, const char *s) {
  490. return luaL_loadbuffer(L, s, strlen(s), s);
  491. }
  492. /* }====================================================== */
  493. LUALIB_API int luaL_getmetafield (lua_State *L, int obj, const char *event) {
  494. if (!lua_getmetatable(L, obj)) /* no metatable? */
  495. return 0;
  496. lua_pushstring(L, event);
  497. lua_rawget(L, -2);
  498. if (lua_isnil(L, -1)) {
  499. lua_pop(L, 2); /* remove metatable and metafield */
  500. return 0;
  501. }
  502. else {
  503. lua_remove(L, -2); /* remove only metatable */
  504. return 1;
  505. }
  506. }
  507. LUALIB_API int luaL_callmeta (lua_State *L, int obj, const char *event) {
  508. obj = abs_index(L, obj);
  509. if (!luaL_getmetafield(L, obj, event)) /* no metafield? */
  510. return 0;
  511. lua_pushvalue(L, obj);
  512. lua_call(L, 1, 1);
  513. return 1;
  514. }
  515. LUALIB_API int luaL_len (lua_State *L, int idx) {
  516. int l;
  517. lua_len(L, idx);
  518. l = lua_tointeger(L, -1);
  519. if (l == 0 && !lua_isnumber(L, -1))
  520. luaL_error(L, "object length is not a number");
  521. lua_pop(L, 1); /* remove object */
  522. return l;
  523. }
  524. LUALIB_API const char *luaL_tolstring (lua_State *L, int idx, size_t *len) {
  525. if (!luaL_callmeta(L, idx, "__tostring")) { /* no metafield? */
  526. switch (lua_type(L, idx)) {
  527. case LUA_TNUMBER:
  528. case LUA_TSTRING:
  529. lua_pushvalue(L, idx);
  530. break;
  531. case LUA_TBOOLEAN:
  532. lua_pushstring(L, (lua_toboolean(L, idx) ? "true" : "false"));
  533. break;
  534. case LUA_TNIL:
  535. lua_pushliteral(L, "nil");
  536. break;
  537. default:
  538. lua_pushfstring(L, "%s: %p", luaL_typename(L, idx),
  539. lua_topointer(L, idx));
  540. break;
  541. }
  542. }
  543. return lua_tolstring(L, -1, len);
  544. }
  545. static int libsize (const luaL_Reg *l) {
  546. int size = 0;
  547. for (; l->name; l++) size++;
  548. return size;
  549. }
  550. LUALIB_API void luaL_register (lua_State *L, const char *libname,
  551. const luaL_Reg *l) {
  552. luaL_checkversion(L);
  553. if (libname) {
  554. /* check whether lib already exists */
  555. luaL_findtable(L, LUA_REGISTRYINDEX, "_LOADED", 1);
  556. lua_getfield(L, -1, libname); /* get _LOADED[libname] */
  557. if (!lua_istable(L, -1)) { /* not found? */
  558. lua_pop(L, 1); /* remove previous result */
  559. /* try global variable (and create one if it does not exist) */
  560. lua_pushglobaltable(L);
  561. if (luaL_findtable(L, 0, libname, libsize(l)) != NULL)
  562. luaL_error(L, "name conflict for module " LUA_QS, libname);
  563. lua_pushvalue(L, -1);
  564. lua_setfield(L, -3, libname); /* _LOADED[libname] = new table */
  565. }
  566. lua_remove(L, -2); /* remove _LOADED table */
  567. }
  568. if (l == NULL) return; /* nothing to register? */
  569. for (; l->name; l++) { /* else fill the table with given functions */
  570. lua_pushcfunction(L, l->func);
  571. lua_setfield(L, -2, l->name);
  572. }
  573. }
  574. LUALIB_API const char *luaL_gsub (lua_State *L, const char *s, const char *p,
  575. const char *r) {
  576. const char *wild;
  577. size_t l = strlen(p);
  578. luaL_Buffer b;
  579. luaL_buffinit(L, &b);
  580. while ((wild = strstr(s, p)) != NULL) {
  581. luaL_addlstring(&b, s, wild - s); /* push prefix */
  582. luaL_addstring(&b, r); /* push replacement in place of pattern */
  583. s = wild + l; /* continue after `p' */
  584. }
  585. luaL_addstring(&b, s); /* push last suffix */
  586. luaL_pushresult(&b);
  587. return lua_tostring(L, -1);
  588. }
  589. LUALIB_API const char *luaL_findtable (lua_State *L, int idx,
  590. const char *fname, int szhint) {
  591. const char *e;
  592. if (idx) lua_pushvalue(L, idx);
  593. do {
  594. e = strchr(fname, '.');
  595. if (e == NULL) e = fname + strlen(fname);
  596. lua_pushlstring(L, fname, e - fname);
  597. lua_rawget(L, -2);
  598. if (lua_isnil(L, -1)) { /* no such field? */
  599. lua_pop(L, 1); /* remove this nil */
  600. lua_createtable(L, 0, (*e == '.' ? 1 : szhint)); /* new table for field */
  601. lua_pushlstring(L, fname, e - fname);
  602. lua_pushvalue(L, -2);
  603. lua_settable(L, -4); /* set new table into field */
  604. }
  605. else if (!lua_istable(L, -1)) { /* field has a non-table value? */
  606. lua_pop(L, 2); /* remove table and value */
  607. return fname; /* return problematic part of the name */
  608. }
  609. lua_remove(L, -2); /* remove previous table */
  610. fname = e + 1;
  611. } while (*e == '.');
  612. return NULL;
  613. }
  614. static void *l_alloc (void *ud, void *ptr, size_t osize, size_t nsize) {
  615. (void)ud;
  616. (void)osize;
  617. if (nsize == 0) {
  618. free(ptr);
  619. return NULL;
  620. }
  621. else
  622. return realloc(ptr, nsize);
  623. }
  624. static int panic (lua_State *L) {
  625. luai_writestringerror("PANIC: unprotected error in call to Lua API (%s)\n",
  626. lua_tostring(L, -1));
  627. return 0; /* return to Lua to abort */
  628. }
  629. LUALIB_API lua_State *luaL_newstate (void) {
  630. lua_State *L = lua_newstate(l_alloc, NULL);
  631. if (L) lua_atpanic(L, &panic);
  632. return L;
  633. }
  634. LUALIB_API void luaL_checkversion_ (lua_State *L, lua_Number ver) {
  635. const lua_Number *v = lua_version(L);
  636. if (v != lua_version(NULL))
  637. luaL_error(L, "multiple Lua VMs detected");
  638. else if (*v != ver)
  639. luaL_error(L, "version mismatch: app. needs %d, Lua core provides %f",
  640. ver, *v);
  641. }
  642. LUALIB_API int luaL_cpcall (lua_State *L, lua_CFunction f, int nargs,
  643. int nresults) {
  644. nargs++; /* to include function itself */
  645. lua_rawgeti(L, LUA_REGISTRYINDEX, LUA_RIDX_CPCALL);
  646. lua_insert(L, -nargs);
  647. lua_pushlightuserdata(L, &f);
  648. lua_insert(L, -nargs);
  649. return lua_pcall(L, nargs, nresults, 0);
  650. }