1 files changed, 318 insertions, 0 deletions

diff --git a/tests/testmod.c b/tests/testmod.c
new file mode 100644
index 0000000..868136b
--- /dev/null
+++ b/tests/testmod.c
@@ -0,0 +1,318 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <lua.h>
+#include <lauxlib.h>
+#include "compat-5.3.h"
+
+
+static int test_isinteger (lua_State *L) {
+  lua_pushboolean(L, lua_isinteger(L, 1));
+  return 1;
+}
+
+
+static int test_rotate (lua_State *L) {
+  int r = luaL_checkint(L, 1);
+  int n = lua_gettop(L)-1;
+  luaL_argcheck(L, (r < 0 ? -r : r) <= n, 1, "not enough arguments");
+  lua_rotate(L, 2, r);
+  return n;
+}
+
+
+static int test_str2num (lua_State *L) {
+  const char *s = luaL_checkstring(L, 1);
+  size_t len = lua_stringtonumber(L, s);
+  if (len == 0)
+    lua_pushnumber(L, 0);
+  lua_pushinteger(L, (lua_Integer)len);
+  return 2;
+}
+
+
+static int my_mod (lua_State *L ) {
+  lua_newtable(L);
+  lua_pushboolean(L, 1);
+  lua_setfield(L, -2, "boolean");
+  return 1;
+}
+
+static int test_requiref (lua_State *L) {
+  lua_getfield(L, LUA_REGISTRYINDEX, "_LOADED");
+  lua_newtable(L);
+  lua_pushboolean(L, 0);
+  lua_setfield(L, -2, "boolean");
+  lua_setfield(L, -2, "requiref3");
+  lua_pop(L, 1);
+  luaL_requiref(L, "requiref1", my_mod, 0);
+  luaL_requiref(L, "requiref2", my_mod, 1);
+  luaL_requiref(L, "requiref3", my_mod, 1);
+  return 3;
+}
+
+static int test_getseti (lua_State *L) {
+  lua_Integer k = luaL_checkinteger(L, 2);
+  lua_Integer n = 0;
+  if (lua_geti(L, 1, k) == LUA_TNUMBER) {
+    n = lua_tointeger(L, -1);
+  } else {
+    lua_pop(L, 1);
+    lua_pushinteger(L, n);
+  }
+  lua_pushinteger(L, n+1);
+  lua_seti(L, 1, k);
+  return 1;
+}
+
+
+/* additional tests for Lua5.1 */
+#define NUP 3
+
+static int test_newproxy (lua_State *L) {
+  lua_settop(L, 0);
+  lua_newuserdata(L, 0);
+  lua_newtable(L);
+  lua_pushvalue(L, -1);
+  lua_pushboolean(L, 1);
+  lua_setfield(L, -2, "__gc");
+  lua_setmetatable(L, -3);
+  return 2;
+}
+
+static int test_absindex (lua_State *L) {
+  int i = 1;
+  for (i = 1; i <= NUP; ++i)
+    lua_pushvalue(L, lua_absindex(L, lua_upvalueindex(i)));
+  lua_pushvalue(L, lua_absindex(L, LUA_REGISTRYINDEX));
+  lua_pushstring(L, lua_typename(L, lua_type(L, lua_absindex(L, -1))));
+  lua_replace(L, lua_absindex(L, -2));
+  lua_pushvalue(L, lua_absindex(L, -2));
+  lua_pushvalue(L, lua_absindex(L, -4));
+  lua_pushvalue(L, lua_absindex(L, -6));
+  i += 3;
+  lua_pushvalue(L, lua_absindex(L, 1));
+  lua_pushvalue(L, lua_absindex(L, 2));
+  lua_pushvalue(L, lua_absindex(L, 3));
+  i += 3;
+  return i;
+}
+
+static int test_arith (lua_State *L) {
+  lua_settop(L, 2);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPADD);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPSUB);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPMUL);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPDIV);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPMOD);
+  lua_pushvalue(L, 1);
+  lua_pushvalue(L, 2);
+  lua_arith(L, LUA_OPPOW);
+  lua_pushvalue(L, 1);
+  lua_arith(L, LUA_OPUNM);
+  return lua_gettop(L)-2;
+}
+
+static int test_compare (lua_State *L) {
+  luaL_checknumber(L, 1);
+  luaL_checknumber(L, 2);
+  lua_settop(L, 2);
+  lua_pushboolean(L, lua_compare(L, 1, 2, LUA_OPEQ));
+  lua_pushboolean(L, lua_compare(L, 1, 2, LUA_OPLT));
+  lua_pushboolean(L, lua_compare(L, 1, 2, LUA_OPLE));
+  return 3;
+}
+
+static int test_globals (lua_State *L) {
+  lua_pushglobaltable(L);
+  return 1;
+}
+
+static int test_tonumber (lua_State *L) {
+  int isnum = 0;
+  lua_Number n = lua_tonumberx(L, 1, &isnum);
+  if (!isnum)
+    lua_pushnil(L);
+  else
+    lua_pushnumber(L, n);
+  return 1;
+}
+
+static int test_tointeger (lua_State *L) {
+  int isnum = 0;
+  lua_Integer n = lua_tointegerx(L, 1, &isnum);
+  if (!isnum)
+    lua_pushnil(L);
+  else
+    lua_pushinteger(L, n);
+  return 1;
+}
+
+static int test_len (lua_State *L) {
+  luaL_checkany(L, 1);
+  lua_len(L, 1);
+  lua_pushinteger(L, luaL_len(L, 1));
+  return 2;
+}
+
+static int test_copy (lua_State *L) {
+  int args = lua_gettop(L);
+  if (args >= 2) {
+    int i = 0;
+    for (i = args-1; i > 0; --i)
+      lua_copy(L, args, i);
+  }
+  return args;
+}
+
+/* need an address */
+static char const dummy = 0;
+
+static int test_rawxetp (lua_State *L) {
+  if (lua_gettop(L) > 0)
+    lua_pushvalue(L, 1);
+  else
+    lua_pushliteral(L, "hello again");
+  lua_rawsetp(L, LUA_REGISTRYINDEX, &dummy);
+  lua_settop(L, 0);
+  lua_rawgetp(L, LUA_REGISTRYINDEX, &dummy);
+  return 1;
+}
+
+static int test_udata (lua_State *L) {
+  const char *tname = luaL_optstring(L, 1, "utype1");
+  void *u1 = lua_newuserdata(L, 1);
+  int u1pos = lua_gettop(L);
+  void *u2 = lua_newuserdata(L, 1);
+  int u2pos = lua_gettop(L);
+  luaL_newmetatable(L, "utype1");
+  luaL_newmetatable(L, "utype2");
+  lua_pop(L, 2);
+  luaL_setmetatable(L, "utype2");
+  lua_pushvalue(L, u1pos);
+  luaL_setmetatable(L, "utype1");
+  lua_pop(L, 1);
+  (void)u1;
+  (void)u2;
+  lua_pushlightuserdata(L, luaL_testudata(L, u1pos, tname));
+  lua_pushlightuserdata(L, luaL_testudata(L, u2pos, tname));
+  luaL_getmetatable(L, "utype1");
+  lua_getfield(L, -1, "__name");
+  lua_replace(L, -2);
+  return 3;
+}
+
+static int test_subtable (lua_State *L) {
+  luaL_checktype(L, 1, LUA_TTABLE);
+  lua_settop(L, 1);
+  if (luaL_getsubtable(L, 1, "xxx")) {
+    lua_pushliteral(L, "oldtable");
+  } else {
+    lua_pushliteral(L, "newtable");
+  }
+  return 2;
+}
+
+static int test_uservalue (lua_State *L) {
+  void *udata = lua_newuserdata(L, 1);
+  int ui = lua_gettop(L);
+  lua_newtable(L);
+  lua_setuservalue(L, ui);
+  lua_getuservalue(L, ui);
+  (void)udata;
+  return 1;
+}
+
+static int test_upvalues (lua_State *L) {
+  int i = 1;
+  for (i = 1; i <= NUP; ++i)
+    lua_pushvalue(L, lua_upvalueindex(i));
+  return NUP;
+}
+
+static int test_tolstring (lua_State *L) {
+  size_t len = 0;
+  luaL_tolstring(L, 1, &len);
+  lua_pushinteger(L, (int)len);
+  return 2;
+}
+
+static int test_pushstring (lua_State *L) {
+  lua_pushstring(L, lua_pushliteral(L, "abc"));
+  lua_pushstring(L, lua_pushlstring(L, "abc", 2));
+  lua_pushstring(L, lua_pushlstring(L, NULL, 0));
+  lua_pushstring(L, lua_pushstring(L, "abc"));
+  lua_pushboolean(L, NULL == lua_pushstring(L, NULL));
+  return 10;
+}
+
+static int test_buffer (lua_State *L) {
+  luaL_Buffer b;
+  char *p = luaL_buffinitsize(L, &b, LUAL_BUFFERSIZE+1);
+  p[0] = 'a';
+  p[1] = 'b';
+  luaL_addsize(&b, 2);
+  luaL_addstring(&b, "c");
+  lua_pushliteral(L, "d");
+  luaL_addvalue(&b);
+  luaL_addchar(&b, 'e');
+  luaL_pushresult(&b);
+  return 1;
+}
+
+static int test_exec (lua_State *L) {
+  const char *cmd = luaL_checkstring(L, 1);
+  return luaL_execresult(L, system(cmd));
+}
+
+
+static const luaL_Reg funcs[] = {
+  { "isinteger", test_isinteger },
+  { "rotate", test_rotate },
+  { "strtonum", test_str2num },
+  { "requiref", test_requiref },
+  { "getseti", test_getseti },
+  { "newproxy", test_newproxy },
+  { "arith", test_arith },
+  { "compare", test_compare },
+  { "tonumber", test_tonumber },
+  { "tointeger", test_tointeger },
+  { "len", test_len },
+  { "copy", test_copy },
+  { "rawxetp", test_rawxetp },
+  { "subtable", test_subtable },
+  { "udata", test_udata },
+  { "uservalue", test_uservalue },
+  { "globals", test_globals },
+  { "tolstring", test_tolstring },
+  { "pushstring", test_pushstring },
+  { "buffer", test_buffer },
+  { "exec", test_exec },
+  { NULL, NULL }
+};
+
+static const luaL_Reg more_funcs[] = {
+  { "getupvalues", test_upvalues },
+  { "absindex", test_absindex },
+  { NULL, NULL }
+};
+
+
+int luaopen_testmod (lua_State *L) {
+  int i = 1;
+  luaL_newlib(L, funcs);
+  for (i = 1; i <= NUP; ++i)
+    lua_pushnumber(L, i);
+  luaL_setfuncs(L, more_funcs, NUP);
+  return 1;
+}
+