Export global only if LUA_COMPAT_MODULE defined.

1 files changed, 8 insertions, 3 deletions

diff --git a/src/unix.c b/src/unix.c
index 73e7b69..5710c27 100644
--- a/src/unix.c
+++ b/src/unix.c
@@ -69,6 +69,8 @@ static t_opt optset[] = {
 };
 
 /* our socket creation function */
+/* this is an ad-hoc module that returns a single function 
+ * as such, do not include other functions in this array. */
 static luaL_Reg func[] = {
     {"unix", global_create},
     {NULL,          NULL}
@@ -87,11 +89,14 @@ int luaopen_socket_unix(lua_State *L) {
     auxiliar_add2group(L, "unix{master}", "unix{any}");
     auxiliar_add2group(L, "unix{client}", "unix{any}");
     auxiliar_add2group(L, "unix{server}", "unix{any}");
-    /* make sure the function ends up in the package table */
+#if LUA_VERSION_NUM > 501 && !defined(LUA_COMPAT_MODULE)
+    lua_pushcfunction(L, global_create);
+#else
+    /* set function into socket namespace */
     luaL_openlib(L, "socket", func, 0);
+    lua_pushcfunction(L, global_create);
+#endif
     /* return the function instead of the 'socket' table */
-    lua_pushstring(L, "unix");
-    lua_gettable(L, -2);
     return 1;
 }