From ce9d484281f9771db617d71502097fc0d48317d0 Mon Sep 17 00:00:00 2001
From: batrick <batrick@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Sat, 9 Aug 2008 23:12:52 +0000
Subject: [PATCH] Made nsock functions get the proper environment. (bad use of
 setfenv earlier). nsock:connect function gets the global environment.

---
 nse_nsock.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/nse_nsock.cc b/nse_nsock.cc
index 087e04586..3da87524c 100644
--- a/nse_nsock.cc
+++ b/nse_nsock.cc
@@ -324,7 +324,7 @@ int luaopen_nsock (lua_State *L)
    * where the #defines are above.
    */
   lua_createtable(L, 5, 0);
-  lua_setfenv(L, 1);
+  lua_replace(L, LUA_ENVIRONINDEX);
 
   lua_createtable(L, 0, 10); // THREAD_PROXY
   lua_createtable(L, 0, 1); // metatable
@@ -357,6 +357,8 @@ int luaopen_nsock (lua_State *L)
   lua_pushcclosure(L, l_nsock_connect, 0);
   lua_pushcclosure(L, socket_lock, 0);
   lua_call(L, 3, 1); // leave connect function on stack...
+  lua_pushvalue(L, LUA_GLOBALSINDEX);
+  lua_setfenv(L, -2); // set the connect function's environment to _G
 
   /* Create the nsock metatable for sockets */
   luaL_newmetatable(L, "nsock");