Haskell中返回allocaArray分配的指针是否会引发未定义行为？

首先要给你点个赞——能注意到allocaArray的作用域问题，这是非常专业的代码审查意识。虽然目前程序没崩溃，但这个写法存在严重的隐患，原因如下：

为什么返回allocaArray的指针是危险的？ #

allocaArray（包括alloca）是在栈上分配内存的函数，它的内存生命周期严格局限于传入的lambda作用域内。当lambda执行完毕后，当前栈帧会被销毁，这块内存会被标记为可复用状态。此时返回的resArray指针指向的是已经被“回收”的栈内存，后续任何对这个指针的访问都属于未定义行为：

• 可能暂时能读到正确数据（只是因为栈内存还没被新的栈帧覆盖）
• 也可能突然出现数据错乱、程序崩溃
• 极端情况下，还可能导致安全漏洞（比如敏感数据被泄露或篡改）

你提到的C语言类比非常准确——这和C中返回栈上变量的指针是完全一样的错误，都属于未定义行为范畴。

正确的写法应该怎么做？ #

针对Vulkan这种需要获取数组的场景，我们应该使用堆内存分配来替代栈分配，比如用Foreign.Marshal.Array中的mallocArray函数。堆内存的生命周期由开发者手动控制，不会随栈帧销毁而释放，调用者在使用完数组后需要用free来释放内存。

修改后的代码示例：

-- 类型声明仅作参考，实际代码中未包含
retrieveVulkanArray :: Storable a => (Ptr Word32 -> Ptr a -> IO b) -> IO (Ptr a, Int)
retrieveVulkanArray f = alloca $ \arrCount -> do
    f arrCount vkNullPtr
    arrCount' <- fromIntegral <$> peek arrCount
    -- 用mallocArray在堆上分配内存
    resArray <- mallocArray arrCount'
    f arrCount resArray
    pure (resArray, arrCount')

调用这个函数的代码需要记得在使用完数组后调用free resArray来释放内存，避免内存泄漏。

为什么现在没崩溃？ #

这只是“运气好”而已——栈内存被释放后并不会立即被覆盖，只要后续没有新的函数调用覆盖这块栈空间，指针暂时还能读到旧数据。但这种状态完全不可靠，一旦程序执行到其他函数（比如后续调用Vulkan的其他API、或者触发了函数调用栈的变化），这块内存就会被新的数据覆盖，届时程序就会出现各种诡异的问题。

内容的提问来源于stack exchange，提问作者Pavel Beliy