1 files changed, 44 insertions, 0 deletions

diff --git a/src/compat.c b/src/compat.c
new file mode 100644
index 0000000..1a66f38
--- /dev/null
+++ b/src/compat.c
@@ -0,0 +1,44 @@
+/*
+ * ###############################################################################################
+ * ######################################### Lua 5.1/5.2 #########################################
+ * ###############################################################################################
+ */
+#include "compat.h"
+
+/*
+** Copied from Lua 5.2 loadlib.c
+*/
+#if LUA_VERSION_NUM == 501
+static int luaL_getsubtable (lua_State *L, int idx, const char *fname)
+{
+        lua_getfield(L, idx, fname);
+        if (lua_istable(L, -1))
+                return 1;  /* table already there */
+        else
+        {
+                lua_pop(L, 1);  /* remove previous result */
+                idx = lua_absindex(L, idx);
+                lua_newtable(L);
+                lua_pushvalue(L, -1);  /* copy to be left at top */
+                lua_setfield(L, idx, fname);  /* assign new table to field */
+                return 0;  /* false, because did not find table there */
+        }
+}
+
+void luaL_requiref (lua_State *L, const char *modname, lua_CFunction openf, int glb)
+{
+        lua_pushcfunction(L, openf);
+        lua_pushstring(L, modname);  /* argument to open function */
+        lua_call(L, 1, 1);  /* open module */
+        luaL_getsubtable(L, LUA_REGISTRYINDEX, "_LOADED");
+        lua_pushvalue(L, -2);  /* make copy of module (call result) */
+        lua_setfield(L, -2, modname);  /* _LOADED[modname] = module */
+        lua_pop(L, 1);  /* remove _LOADED table */
+        if (glb)
+        {
+                lua_pushvalue(L, -1);  /* copy of 'mod' */
+                lua_setglobal(L, modname);  /* _G[modname] = module */
+        }
+}
+#endif // LUA_VERSION_NUM
+