使用Haskell inline-c读取二维数组出现段错误该如何解决

问题根因 #

段错误的核心原因是C侧返回的指针类型和Haskell侧声明的类型不匹配：

• C代码中的double (*ptd)[2]是指向长度为2的double数组的指针，对应内存结构是连续存储的4个double值，ptd本身就是这段连续内存的首地址，没有额外的指针层级
• 你在Haskell侧将其声明为Ptr (Ptr CDouble)，这要求指向的内存中存储的是指针值，调用peekElemOff e 0时，程序会把内存中第一个double值1.78对应的二进制数据当成内存地址去访问，必然触发段错误。

解决方案 #

方案1：修改Haskell侧类型匹配现有C代码（推荐，无额外内存开销） #

无需修改C代码，直接调整Haskell侧的类型声明即可，两种可选写法：

写法1：直接按扁平数组处理

因为二维数组的内存是连续存储的，直接把返回值当成一维double数组的指针，按索引计算即可：

-- 修改返回类型为单级double指针
cArray :: IO (Ptr CDouble)
cArray = do
  x <- [C.block| double (*)[2] {
  double (*ptd)[2] = malloc(sizeof(double[2][2]));
  ptd[0][0] = 1.78;
  ptd[0][1] = 1.68;
  ptd[1][0] = 1.58;
  ptd[1][1] = 1.48;
  printf("the firstelement is: %f\n", ptd[1][1]);
  return ptd;
}|]
  return x

调用侧写法：

e <- cArray :: IO (Ptr CDouble)
-- 取第i行第j列的元素，索引计算公式为 行号 * 列数 + 列号
e2 <- peekElemOff e (1*2 + 0) -- 对应C侧的ptd[1][0]
print e2 -- 输出1.58

写法2：用数组指针类型明确声明

如果你想要更严格的类型检查，可以声明固定长度数组类型：

import Foreign.Array (Array)

cArray :: IO (Ptr (Array 2 CDouble))
cArray = -- 其余C代码不变

调用时按需转换为Ptr CDouble按偏移访问即可。

方案2：修改C代码返回指针数组适配现有Haskell逻辑 #

如果你不想修改Haskell侧的现有调用逻辑，可以把C代码调整为返回真正的二级指针结构：

double** cArray() {
  // 先分配存储2个行指针的内存空间
  double** ptd = malloc(sizeof(double*) * 2);
  // 分别为每一行分配存储空间
  ptd[0] = malloc(sizeof(double) * 2);
  ptd[1] = malloc(sizeof(double) * 2);
  ptd[0][0] = 1.78;
  ptd[0][1] = 1.68;
  ptd[1][0] = 1.58;
  ptd[1][1] = 1.48;
  printf("the firstelement is: %f\n", ptd[1][1]);
  return ptd;
}

注意使用该方案时，释放内存需要先释放每一行的指针，再释放顶层指针，避免内存泄漏。

注意事项 #

两种方案中malloc申请的内存都需要手动调用free释放，避免出现内存泄漏问题。

内容的提问来源于stack exchange，提问作者McBear Holden