coroutine.running, upvalue need to be attached to their thread

4 files changed, 9 insertions, 8 deletions

diff --git a/src/lfunc.js b/src/lfunc.js
index 7057f2d..3b3c8ce 100644
--- a/src/lfunc.js
+++ b/src/lfunc.js
@@ -24,7 +24,8 @@ class Proto {
 
 class UpVal {
 
-    constructor() {
+    constructor(L) {
+        this.L = L; // Keep track of the thread it comes from
         this.v = null; /* if null, upval is closed, value is in u.value */
         this.u = {
             open: { /* (when open) */
@@ -41,7 +42,7 @@ class UpVal {
 
     setval(L, ra) {
         if (this.v !== null) {
-            L.stack[this.v] = L.stack[ra];
+            this.L.stack[this.v] = L.stack[ra];
             this.v = ra;
         } else this.u.value = L.stack[ra];
     }
@@ -64,7 +65,7 @@ const findupval = function(L, level) {
         pp = p.u.open.next;
     }
 
-    let uv = new UpVal();
+    let uv = new UpVal(L);
     uv.refcount = 0;
     uv.u.open.next = pp;
     uv.u.open.touched = true;
@@ -92,7 +93,7 @@ const luaF_close = function(L, level) {
 
 const luaF_initupvals = function(L, cl) {
     for (let i = 0; i < cl.nupvalues; i++) {
-        let uv = new UpVal();
+        let uv = new UpVal(L);
         uv.refcount = 1;
         uv.u.value = null;
         uv.v = uv.u.value;
diff --git a/src/lobject.js b/src/lobject.js
index 4e4b701..086af00 100644
--- a/src/lobject.js
+++ b/src/lobject.js
@@ -163,13 +163,13 @@ class Table extends TValue {
 
 class LClosure extends TValue {
 
-    constructor(n) {
+    constructor(L, n) {
         super(CT.LUA_TLCL, null);
 
         this.p = null;
         this.nupvalues = n;
 
-        let _ENV = new UpVal();
+        let _ENV = new UpVal(L);
         _ENV.refcount = 0;
         _ENV.v = null;
         _ENV.u.open.next = null;
diff --git a/src/lundump.js b/src/lundump.js
index ab1dd6f..b79e6ea 100644
--- a/src/lundump.js
+++ b/src/lundump.js
@@ -293,7 +293,7 @@ class BytecodeParser {
     luaU_undump(L) {
         this.checkHeader();
 
-        let cl = new LClosure(this.readByte());
+        let cl = new LClosure(L, this.readByte());
 
         L.stack[L.top] = cl;
         L.top++;
diff --git a/src/lvm.js b/src/lvm.js
index 5615acd..95e8cc4 100644
--- a/src/lvm.js
+++ b/src/lvm.js
@@ -659,7 +659,7 @@ const luaV_execute = function(L) {
                 let p = cl.p.p[i.Bx];
                 let nup = p.upvalues.length;
                 let uv = p.upvalues;
-                let ncl = new LClosure(nup);
+                let ncl = new LClosure(L, nup);
                 ncl.p = p;
 
                 L.stack[ra] = ncl;