Question

我有一些代码应该在Erlang中管理具有O（1）访问和读取时间的3维数组。因此，我使用的是Erlang NIF。除了release（）函数之外，一切正常。调用它时我总是遇到分段错误，我不明白为什么。

这是我的代码：

#include "erl_nif.h"

static ErlNifResourceType *DATA_RESOURCE;

typedef struct
{
    int size;
    ERL_NIF_TERM *** array;
    ERL_NIF_TERM defaultValue;
} DATA;

static ERL_NIF_TERM new3DimArray(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
    DATA *data = (DATA *)enif_alloc_resource(DATA_RESOURCE, sizeof(DATA));
    int size;
    enif_get_int(env, argv[0], &size);
    if(argc > 1)
    {
        data->defaultValue = argv[1];
    }else{
        data->defaultValue = NULL;
    }
    data->size = size;
    data->array =  (ERL_NIF_TERM ***)enif_alloc(sizeof(ERL_NIF_TERM **) * size);
    int x = 0;
    while(x < size)
    {
        data->array[x] = (ERL_NIF_TERM **)enif_alloc(sizeof(ERL_NIF_TERM *) * size);
        int y = 0;
        while(y < size)
        {
            data->array[x][y] =  (ERL_NIF_TERM *)enif_alloc(sizeof(ERL_NIF_TERM) * size);
            y++;
        }
        x++;
    }
    return enif_make_resource(env, data);
}

static ERL_NIF_TERM get_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
    DATA *data;
    enif_get_resource(env, argv[0], DATA_RESOURCE, &data);
    int x;
    int y;
    int z;
    enif_get_int(env, argv[1], &x);
    enif_get_int(env, argv[2], &y);
    enif_get_int(env, argv[3], &z);
    ERL_NIF_TERM res = data->array[x][y][z];
    if(res == NULL && data->defaultValue != NULL)
    {
        res = data->defaultValue;
    }
    return res;
}

static void set_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
    DATA *data;
    enif_get_resource(env, argv[0], DATA_RESOURCE, &data);
    int x;
    int y;
    int z;
    enif_get_int(env, argv[1], &x);
    enif_get_int(env, argv[2], &y);
    enif_get_int(env, argv[3], &z);
    ERL_NIF_TERM value = argv[4];
    data->array[x][y][z] = value;
}

static void release(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
    DATA *data;
    enif_get_resource(env, argv[0], DATA_RESOURCE, &data);
    int x = 0;
    while(x < data->size)
    {
        int y = 0;
        while(y < data->size)
        {
            enif_free(data->array[x][y]);
            y++;
        }
        enif_free(data->array[x]);
        x++;
    }
    enif_free(data->array);
    enif_release_resource(data);
}

static void cleanup(ErlNifEnv *env, void *obj){}

static int load(ErlNifEnv *env, void **priv_data, ERL_NIF_TERM load_info){
    DATA_RESOURCE = enif_open_resource_type(env, "mutArray", "DATA_RESOURCE", &cleanup, ERL_NIF_RT_CREATE, 0);
    return 0;
}

static ErlNifFunc nif_funcs[] = {
    {"new_3_dim_array", 1, new3DimArray},
    {"new_3_dim_array", 2, new3DimArray},
    {"get", 4, get_nif},
    {"set", 5, set_nif},
    {"release", 1, release}
};

ERL_NIF_INIT(mutArray, nif_funcs, load, NULL, NULL, NULL);

这是我的Erlang代码（为了让arity更清晰）：

module(mutArray).

%% ====================================================================
%% API functions
%% ====================================================================
-export([init/0, new_3_dim_array/1, new_3_dim_array/2, get/4, set/5, release/1]).

init() ->
    erlang:load_nif("./mutArray", 0).

new_3_dim_array(_Size) ->
    "NIF not loaded yet.".

new_3_dim_array(_Size, _DefaultValue) ->
    "NIF not loaded yet.".

get(_Array, _X, _Y, _Z) ->
    "NIF not loaded yet.".

set(_Array, _X, _Y, _Z, _Value) ->
    "NIF not loaded yet.".

release(_Array) ->
    "NIF not loaded yet.".

是的，这是我的测试代码：

mutArray:init(),
A = mutArray:new_3_dim_array(100),
mutArray:release(A).

编辑：好吧它变得越来越奇怪......经过一些测试我得知** exception error: []如果enif_free(data->array);是函数的最后一次调用。在其他每个位置，我仍然会遇到分段错误，即使enif_free(data->array);之后只有println（）。经过一些调试后，我还发现调用了enif_free(data->array);之前的每一行。所以异常似乎发生在enif_free(data->array)。有人知道这意味着什么吗？

EDIT2：简单地离开enif_free(data->array);也不会有任何帮助。我也遇到了分段错误。

Answer 1

通过解决几个问题，我能够正确运行代码。

首先，您的代码假定可以通过将ERL_NIF_TERM与NULL进行比较来检查enif_make_int(env, 0)的有效性，这是不正确的。您可以通过将所有数组元素初始化为0（通过调用ERL_NIF_TERM来设置每个元素）或使用每个结构包含unsigned char和{{1的结构数组来解决此问题。标志以指示该术语是否有效。如果您选择后一种方法，您可以简单地将memset结构值设置为0，如果调用者通过mutArray:get/4请求未初始化的元素，则只返回enif_make_badarg(env)以指示它们将错误的参数传递给电话。

其次，当set_nif和release函数需要返回void时，会声明它们返回ERL_NIF_TERM。要解决此问题，您可以更正其返回类型，然后从argv[4]返回set_nif，从enif_make_int(env, 0)返回release。

最后，enif_open_resource_type来电的第二个参数必须是NULL，而不是"mutArray"所传递的void page_kernel_only(int16_t page){ if(mode != KERNEL) { return; } page = page << 5; page = page >> 5; int16_t help = 8192; help = help & page_table[page]; if(help == 0) { page_table[page] += 8192; }值，正如the erl_nif man page所示。

调用enif_free（）时出现分段错误

1 个答案: