在 Lua 绑定 C 代码中正确处理 malloc 调用

我正在编写一个小型的Lua绑定库。目前我提供了一个非常简单的结构来保存键值对,其中我的“value”是一个“void指针”。

struct kvPair {
       enum EVENT_TYPE type;
       char *key;
       void *value;
   };

typedef struct kvPair KeyValuePair;

KeyValuePair kvPairWithNumber(char *key, float number)
{
    KeyValuePair kv;
    float *aux = malloc(sizeof(float)); // <-- help
    *aux = number;
    kv.key = key;
    kv.type = NUMBER;
    kv.value = aux;
    return kv;
}

我想知道对于这种用例来说,应该如何处理malloc好。我不希望这样做:

if (aux == NULL) {
    exit(0);
}

也许我可以预先分配一个内存块,然后用自己简单的实现(具有简单堆栈的自定义内存分配器?)覆盖malloc调用?

希望能收到反馈和示例,简单性是首选。

点赞
用户856565
用户856565

你绝对不想退出。没有脚本或扩展应该使整个解释器崩溃。

基本上有两种方法来表达错误。第一种方法是使用 lua_error 或者 luaL_error,它们由 C API 提供。第二种方法更像是一种政策: 返回 nil 和一个错误消息 (通常与 assert 结合使用)。

你正在用 C 编写扩展,因此你基本上只能使用 C 风格的错误处理,通过返回码来实现。你可能需要改变你的函数签名以支持这一点。

int kvPairWithNumber(KeyValuePair * kv, char *key, float number)
{
    float *aux = malloc(sizeof(float));
    if (aux == NULL) {
        return 0;
        // 注意: 你可以在这里放置 lua_error
    }
    *aux = number;
    kv->key = key;
    kv->type = NUMBER;
    kv->value = aux;
    return 1;
}

然后你可以在 Lua 函数中处理错误:

int my_lua_function(lua_State * L)
{
    KeyValuePair kv = {0};

    // 第一种方法:
    if (!kvPairWithNumber(&kv, "asdf", 1.0)) {
        luaL_error(L, "could not create pair!");
    }

    // 第二种方法:
    if (!kvPairWithNumber(&kv, "asdf", 1.0)) {
        lua_pushnil(L);
        lua_pushstring(L, "could not create a pair!");
        return 2;
    }
}

使用你的扩展的 Lua 代码如下:

-- 第一种方法:
local result = xpcall(my_lua_function, error_handler)

-- 注意: 如果你需要传递参数,你将需要一个闭包
local result = xpcall(function() my_lua_function(x, y, z) end, error_handler)

-- 第二种方法:
local result = assert(my_lua_function())
2013-01-14 04:23:55