issue #198 initialize the _ENV upvalue

dibyendumajumdar · Oct 5, 2020 · 523f07b · 523f07b
1 parent 71bb74f
commit 523f07b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/ravi_complib.c b/src/ravi_complib.c
@@ -247,6 +247,14 @@ static int load_and_compile(lua_State* L) {
   if (rc == 0) {
     lua_assert(cl->nupvalues == cl->p->sizeupvalues);
     luaF_initupvals(L, cl);
+    if (cl->nupvalues >= 1) {  /* does it have an upvalue? */
+      /* get global table from registry */
+      Table *reg = hvalue(&G(L)->l_registry);
+      const TValue *gt = luaH_getint(reg, LUA_RIDX_GLOBALS);
+      /* set global table as 1st upvalue of 'f' (may be LUA_ENV) */
+      setobj(L, cl->upvals[0]->v, gt);
+      luaC_upvalbarrier(L, cl->upvals[0], gt);
+    }
     return 1;
   }
   else {