diff options
Diffstat (limited to 'ldo.c')
-rw-r--r-- | ldo.c | 9 |
1 files changed, 6 insertions, 3 deletions
@@ -383,8 +383,10 @@ static void moveresults (lua_State *L, StkId res, int nres, int wanted) { | |||
383 | wanted = nres; /* we want all results */ | 383 | wanted = nres; /* we want all results */ |
384 | break; | 384 | break; |
385 | default: /* multiple results (or to-be-closed variables) */ | 385 | default: /* multiple results (or to-be-closed variables) */ |
386 | if (hastocloseCfunc(wanted)) { | 386 | if (hastocloseCfunc(wanted)) { /* to-be-closed variables? */ |
387 | luaF_close(L, res, LUA_OK); | 387 | ptrdiff_t savedres = savestack(L, res); |
388 | luaF_close(L, res, LUA_OK); /* may change the stack */ | ||
389 | res = restorestack(L, savedres); | ||
388 | wanted = codeNresults(wanted); /* correct value */ | 390 | wanted = codeNresults(wanted); /* correct value */ |
389 | if (wanted == LUA_MULTRET) | 391 | if (wanted == LUA_MULTRET) |
390 | wanted = nres; | 392 | wanted = nres; |
@@ -590,7 +592,8 @@ static int recover (lua_State *L, int status) { | |||
590 | if (ci == NULL) return 0; /* no recovery point */ | 592 | if (ci == NULL) return 0; /* no recovery point */ |
591 | /* "finish" luaD_pcall */ | 593 | /* "finish" luaD_pcall */ |
592 | oldtop = restorestack(L, ci->u2.funcidx); | 594 | oldtop = restorestack(L, ci->u2.funcidx); |
593 | luaF_close(L, oldtop, status); | 595 | luaF_close(L, oldtop, status); /* may change the stack */ |
596 | oldtop = restorestack(L, ci->u2.funcidx); | ||
594 | luaD_seterrorobj(L, status, oldtop); | 597 | luaD_seterrorobj(L, status, oldtop); |
595 | L->ci = ci; | 598 | L->ci = ci; |
596 | L->allowhook = getoah(ci->callstatus); /* restore original 'allowhook' */ | 599 | L->allowhook = getoah(ci->callstatus); /* restore original 'allowhook' */ |