Ошибка сегментации при использовании lua coroutine api и lua_close - PullRequest
Новая
       30

Ошибка сегментации при использовании lua coroutine api и lua_close

0 голосов
/ 29 сентября 2018

Я работаю над проектом прерывания сценария lua, я хочу использовать std :: Stack и сопрограмму lua для сохранения контекста.Но когда я установил размер стека более 38, он случайно вылетает в lua_resume и lua_close.test.lua: 

local stacksize = 40 --When changing stacksize less than 30, it runs fine.
function heavy_function(i)
    print("heavy_function start",i)
    if i < stacksize then 
        coroutine.yield(i+1)
    end
    print("heavy_function end",i)
end

main.cpp: 

#ifdef __cplusplus
extern "C" {
#endif
#include "lua.h"
#include "lualib.h"
#include "lauxlib.h"

#ifdef __cplusplus
}
#endif

#include <iostream>
#include <unistd.h>
#include <ctime>
#include <stdio.h>
#include <string>
#include <stack>
using namespace std;

int main()
{
    lua_State* L = luaL_newstate();
    luaL_openlibs(L);

    int ret = luaL_dofile(L, "test.lua");
    if (ret != 0)
    {
        // Error loading script. Return.
        printf("luaL_dofile error \n");
        return -1;
    }

    // Add a count hook that will trigger after "count" number instructions
    //lua_sethook(L, LUAHook, LUA_MASKLINE, 0);

    stack<lua_State *> Lstack;
    Lstack.push(lua_newthread(L));
    int init = 1;
    do{
        lua_getglobal(Lstack.top(), "heavy_function");
        lua_pushinteger(Lstack.top(),init);
        ret = lua_resume(Lstack.top(),L,1);
        if(ret == LUA_YIELD)
        {
            init = luaL_checkinteger(Lstack.top(),-1);
            Lstack.push(lua_newthread(L));
        }
        else if(ret == 0)
        {
            //lua_close(Lstack.top());
            lua_gc(L,LUA_GCCOLLECT,0);
            cout<<"Memory Usage:"<<lua_gc(L,LUA_GCCOUNT,0)<<endl;
            Lstack.pop();
        }
        else{
            cout<<"error"<<endl;
            return -1;
        }
    }while(Lstack.size()>0);
    //printf("lua script interrupted \n");
    lua_close(L);
    return 0;
}

Параметр компилятора: 

g++ -g main.cpp -o test -llua -ldl

Я подозреваю, что допустил ошибку при вызове lua_newthread.SoЯ сделал проверку стека перед вызовом lua_newstate, и он стал нормальным. 

if(ret == LUA_YIELD)
{
    init = luaL_checkinteger(Lstack.top(),-1);
    Lstack.push(lua_newthread(L));
    cout<<"lua_checkstack(L,10) = "<<lua_checkstack(L,1)<<endl;//Add a line in line 47
}

Хотел узнать, сделал ли я ошибку в этом и как я могу исправить это?

1 Ответ

0 голосов
/ 29 сентября 2018

Вы переполняете стек Lua, непрерывно генерируя новые потоки Lua и оставляя его объекты Lua в стеке.

lua_newstack() не только возвращает указатель на структуру lua_State, но также оставляет значение типаLUA_TTHREAD в стеке в вашем L состоянии.Вы должны либо соответствующим образом приспособить стек Lua, либо управлять возвращенными потоками Lua каким-либо другим способом.

Быстрое и грязное "исправление" будет означать вызов lua_checkstack(L, 10); прямо перед вашей Lstack.push(lua_newthread(L)); строкой.Это позволяет вашему коду работать как есть, но стек будет постоянно расти.Вместо этого вы должны взять новый объект потока со стека и поместить его в некоторую таблицу Lua, пока не придет время для его удаления.

...