Swift Unsafe API内存分配如何优雅失败而非直接崩溃？

Swift的UnsafeMutableRawPointer.allocate()在内存不足时会直接触发崩溃，而C的malloc在分配失败时会返回NULL（对应Swift中的nil），可以借此实现优雅降级。以下是具体的处理方案：

方案一：直接调用C标准库的malloc/free #

Swift可以直接调用C的内存分配函数，这是最直接的解决方案。示例代码如下：

let targetSize = 10_000_000_000
// 尝试分配目标大小内存
if let rawPtr = malloc(targetSize) {
    let ptr = rawPtr.assumingMemoryBound(to: UInt8.self)
    defer { free(ptr) } // 确保内存最终释放
    // 分配成功，执行内存操作逻辑
} else {
    // 内存不足，执行降级处理，比如申请更小的内存
    print("内存分配失败，尝试申请更小容量")
    let fallbackSize = 1_000_000_000
    if let fallbackRawPtr = malloc(fallbackSize) {
        let fallbackPtr = fallbackRawPtr.assumingMemoryBound(to: UInt8.self)
        defer { free(fallbackPtr) }
        // 使用降级后的内存执行逻辑
    }
}

方案二：封装优雅分配的扩展方法 #

如果希望保持Swift Unsafe API的调用风格，可以封装一个扩展方法，内部基于malloc实现：

extension UnsafeMutableRawPointer {
    static func allocateGracefully(byteCount: Int, alignment: Int) -> UnsafeMutableRawPointer? {
        guard alignment > 0, byteCount >= 0 else { return nil }
        // 调用malloc并转换为UnsafeMutableRawPointer
        return malloc(byteCount)?.assumingMemoryBound(to: UInt8.self).rawPointer
    }
}

// 使用示例
if let p = UnsafeMutableRawPointer.allocateGracefully(
    byteCount: 10_000_000_000,
    alignment: MemoryLayout<UInt8>.alignment
) {
    defer { p.deallocate() }
    // 处理内存操作
} else {
    // 分配失败，执行错误处理或降级逻辑
}

无需单独编写C分配器 #

不需要额外编写C手动内存分配器，Swift本身支持直接调用C标准库的malloc/free，完全可以满足优雅处理内存分配失败的需求。Swift原生Unsafe API的allocate()设计为分配失败时触发致命错误，没有内置失败反馈机制，因此借助C的内存分配函数是最优选择。

内容的提问来源于stack exchange，提问作者Céline