将 Objective-C malloc 转换为 Swift

Question

我正在开发一个用 Objective-C 编写的项目，需要更新到 Swift。我们使用 C 文件来传输数据。

这是我在 Objective-C 中给出的代码：

- (NSData *)prepareEndPacket {
    UInt8 *buff_data;
    buff_data = (uint8_t *)malloc(sizeof(uint8_t)*(PACKET_SIZE+5));
    // Call to C File
    PrepareEndPacket(buff_data);
    NSData *data_first = [NSData dataWithBytes:buff_data length:sizeof(uint8_t)*(PACKET_SIZE+5)];
    return data_first;
}

在 C .h 文件中，我有这个供参考：

#define PACKET_SIZE     ((uint32_t)128)

我似乎找不到将其转换为 Swift 的好方法。任何帮助，将不胜感激。

Answer 1

malloc 和 free 实际上在 Swift 中运行良好；但是，UnsafeMutablePointer API 更“原生”。我可能会使用Data 的bytesNoCopy 以获得更好的性能。如果你愿意，你可以使用Data(bytes:count:)，但这会复制数据（然后你需要确保在复制后释放指针，否则你会泄漏内存，这实际上是一个问题上面的 Objective-C 代码，因为它无法free 缓冲区）。

所以，类似：

func prepareEndPacket() -> Data {
    let count = PACKET_SIZE + 5
    let buf = UnsafeMutablePointer<UInt8>.allocate(capacity: count)

    PrepareEndPacket(buf)

    return Data(bytesNoCopy: buf, count: count, deallocator: .custom { ptr, _ in
        ptr.deallocate()
    })
}

通过使用bytesNoCopy，返回的Data 对象基本上是原始指针的包装，当Data 对象被销毁时，释放器将释放原始指针。

或者，您可以从头开始创建Data 对象并获取指向其内容的指针以传递给PrepareEndPacket()：

func prepareEndPacket() -> Data {
    var data = Data(count: PACKET_SIZE + 5)

    data.withUnsafeMutableBytes { (ptr: UnsafeMutablePointer<UInt8>) in
        PrepareEndPacket(ptr)
    }

    return data
}

这稍微效率较低，因为Data(count:) 初始化程序会将所有Data 的字节初始化为零（类似于使用calloc 而不是malloc），但在在很多情况下，这可能并不足以产生影响。