Revert overzealous upvalue check on on_state_create

2 files changed, 6 insertions, 16 deletions

diff --git a/docs/index.html b/docs/index.html
index ca91273..6e2b4ed 100644
--- a/docs/index.html
+++ b/docs/index.html
@@ -358,6 +358,7 @@
                         <td>
                                 If provided, will be called in every created Lua state right after initializing the base libraries, with a single string argument, either <tt>"lane"</tt> or <tt>"keeper"</tt>.<br />
                                 If it is a C function, a C closure will be reconstructed in the created state from the C pointer. Lanes will raise an error if the function has upvalues.<br />
+                                If it is a Lua function, it will be transfered normally before the call.<br />
                                 Keeper states will call it as well, but only if it is a C function (<a href="#keepers">Keeper states</a> are not able to execute any user Lua code).<br />
                                 Typical usage is twofold:
                                 <ul>
@@ -365,7 +366,6 @@
                                         <li>Load some additional C functions in the global space (of course only a C function will be able to do this).</li>
                                 </ul>
                                 That way, all changes in the state can be properly taken into account when building the function lookup database. Default is <tt>nil</tt>.<br />
-                                If <tt>on_state_create()</tt> is a Lua function, it will be transfered normally before the call.<br />
                         </td>
                 </tr>
 
diff --git a/src/universe.cpp b/src/universe.cpp
index 08fdf40..0dc5646 100644
--- a/src/universe.cpp
+++ b/src/universe.cpp
@@ -303,24 +303,14 @@ void Universe::initializeOnStateCreate(lua_State* const L_)
     STACK_CHECK_START_REL(L_, 0);                                                                  // L_: settings
     if (luaG_getfield(L_, -1, kOnStateCreate) != LuaType::NIL) {                                   // L_: settings on_state_create|nil
         LUA_ASSERT(L_, luaG_type(L_, -1) == LuaType::FUNCTION); // ensured by lanes.lua parameter validation
-        // make sure the function doesn't have upvalues other than _G
-        int _uvi{ 1 };
-        for (
-            char const* _upname{ lua_getupvalue(L_, -1, _uvi) };
-            _upname;
-            _upname = lua_getupvalue(L_, -1, ++_uvi)                                               // L_: settings on_state_create upvalue
-        ) {
-            // starting with Lua 5.2, functions have _ENV as their first upvalue. This is ok, it is mapped correctly
-            luaG_pushglobaltable(L_);                                                              // L_: settings on_state_create upvalue _G
-            if (!lua_rawequal(L_, -1, -2)) {
-                raise_luaL_error(L_, "%s with upvalues are forbidden", kOnStateCreate.data());
-            }
-            lua_pop(L_, 2);                                                                        // L_: settings on_state_create
-        }
-        STACK_CHECK(L_, 1); // make sure no garbage remains on the stack after upvalue check       // L_: settings on_state_create
         // store C function pointer in an internal variable
         lua_CFunction const _func{ lua_tocfunction(L_, -1) };                                      // L_: settings on_state_create
         if (_func) {
+            // make sure the function doesn't have upvalues
+            char const* _upname{ lua_getupvalue(L_, -1, 1) };                                      // L_: settings on_state_create upval?
+            if (_upname != nullptr) { // should be "" for C functions with upvalues if any
+                raise_luaL_error(L_, "%s shouldn't have upvalues", kOnStateCreate.data());
+            }
             onStateCreateFunc.emplace<lua_CFunction>(_func);
             // remove this C function from the config table so that it doesn't cause problems
             // when we transfer the config table in newly created Lua states