C语言 在Elixir NIF中正确处理资源

Question

我试图为简单的线性代数插入一个NIF。这是我的矩阵内部结构：

typedef struct la_matrix {
uint rows, columns;
double **data;
} la_matrix;

这是它的"构造函数"：

la_result
la_matrix_constructor(la_matrix **res,
const uint rows,
const uint columns)
{
if (rows == 0 || columns == 0)
return dimensional_problems;
// allocate memory for meta-structure
*res = malloc(sizeof(la_matrix));
if (*res == NULL)
return null_ptr;
// allocater memory for array of pointers to rows
(*res)->data = malloc(rows * sizeof(double*));
if ((*res)->data == NULL) {
free(*res);
return null_ptr;
}
//allocate memory for each row
uint i = 0;
bool failed = false;
for (; i < rows; i++) {
(*res)->data[i] = malloc(columns * sizeof(double));
if ((*res)->data[i] == NULL) {
failed = true;
break;
}
}
if (failed) {
// one step back, since i-th row wasn't allocated
i -= 1;
for(; i < ~((uint) 0); i--)
free((*res)->data[i]);
free((*res)->data);
free(*res);
return null_ptr;
}
(*res)->rows    = rows;
(*res)->columns = columns;

return ok;
}

然后我有两个用于 NIF 的包装器 - 一个用于构造函数：

static ERL_NIF_TERM
nif_matrix_constructor(ErlNifEnv *env,
int argc,
const ERL_NIF_TERM *argv)
{
uint rows, columns;
enif_get_uint(env, argv[0], &rows);
enif_get_uint(env, argv[1], &columns);

la_matrix **mat_res = enif_alloc_resource(LA_MATRIX_TYPE, sizeof(la_matrix *));

la_matrix *mat_ptr;
la_result result = la_matrix_constructor(&mat_ptr, rows, columns);
if (result != ok)
return enif_make_atom(env, "err");
memcpy((void *) mat_res, (void *) &mat_ptr, sizeof(la_matrix *));
ERL_NIF_TERM term = enif_make_resource(env, mat_res);
enif_release_resource(mat_res);

return term;
}

还有一个用于测试构造函数是否正常工作：

static ERL_NIF_TERM
nif_matrix_rows(ErlNifEnv *env,
int argc,
const ERL_NIF_TERM *argv)
{
la_matrix *mat_ptr;
if(!enif_get_resource(env, argv[0], LA_MATRIX_TYPE, (void **) &mat_ptr))
return enif_make_atom(env, "err");
return enif_make_uint(env, mat_ptr->rows);
}

构造函数包装器似乎工作得很好(我已经使用printf对其进行了测试)，但nif_matrix_rows返回奇怪的结果，例如

iex(1)> mat = LinearAlgebra.matrix(2,3)
""
iex(2)> LinearAlgebra.rows(mat)
1677732752

并且直接将LinearAlgebra.matrix(2,3)传递给LinearAlgebra.rows两次会导致段错误：

iex(3)> LinearAlgebra.rows(LinearAlgebra.matrix(2,3))
1543520864
iex(4)> LinearAlgebra.rows(LinearAlgebra.matrix(2,3))
zsh:  (core dumped)  iex -S mix

(请注意"相同"矩阵的不同结果)。

我正在遵循 Andrea Leopardi 的教程，进行了微小的(我不确定是否是这样)更改以对抗gcc警告。最重要的是，恕我直言，是这部分

la_matrix *mat_ptr;
if(!enif_get_resource(env, argv[0], LA_MATRIX_TYPE, (void **) &mat_ptr))
return enif_make_atom(env, "err");

而安德里亚·莱奥帕尔迪使用

db_conn_t **conn_res;
enif_get_resource(env, argv[0], DB_RES_TYPE, (void *) conn_res);
db_conn_t *conn = *conn_res;

但这对我来说看起来是无效的，因为 AFAIR，(void *) conn_res假设conn_res是发起的。

这是我使用 Andrea 的方式时发生的错误：

src/nif.c: In function ‘nif_matrix_rows’:
src/nif.c:72:3: warning: ‘mat_res’ is used uninitialized in this function [-Wuninitialized]
enif_get_resource(env, argv[0], LA_MATRIX_TYPE, (void *) mat_res);

从iex调用LinearAlgebra.rows会导致段错误。

有人能告诉我在 NIF 中处理结构的正确方法吗？

附言对不起 C 代码，我从来没有写过比一堆 helloworld 更多的东西。

Answer 1

问题确实出在nif_matrix_rows：使用我的代码Elixir传递一个指向结构(la_matrix **)的指针，我认为这将是一个正确的指针。

因此，快速修复是

static ERL_NIF_TERM
nif_matrix_rows(ErlNifEnv *env,
int argc,
const ERL_NIF_TERM *argv)
{
la_matrix const **mat_res;
if(!enif_get_resource(env, argv[0], LA_MATRIX_TYPE,(void **) &mat_res))
return enif_make_atom(env, "err");
la_matrix const *mat_ptr = *mat_res;
return enif_make_uint(env, mat_ptr->rows);
}

但是，我会等待一段时间以获得更优雅的解决方案，到目前为止不会接受这个答案。