Swift调用DriverKit传递含数组结构体的问题排查

问题背景 #

Swift端定义了包含UInt8数组的结构体CmdStruct，在调用DriverKit的IOConnectCallStructMethod传递参数时，Xcode抛出警告，且DriverKit端无法正确获取Swift端传入的参数值。

Swift端代码 #

struct CmdStruct {
    var cmd:[UInt8]
    init() {
        cmd = [UInt8](repeating: 0, count:16)
    }
}

func sendCmd() {
    var ret:kern_return_t = kIOReturnSuccess
    var input = CmdStruct()
    var output:[UInt8] = [UInt8](repeating:0, count: 16)
    var inputSize = MemoryLayout<CmdStruct>.size  // 输出值为8
    var outputSize = output.count // 输出值为16

    input.cmd[0] = 0x12
    input.cmd[1] = 0x34

    ret = IOConnectCallStructMethod(Connection, selector, &input, inputSize, &output, &outputSize)
}

Xcode警告：Forming 'UnsafeRawPointer' to a variable of type 'CmdStruct'; this is likely incorrect because 'CmdStruct' may contain an object reference.

DriverKit端C代码 #

struct CmdStruct
{
    uint8_t cmd[16];
};

kern_return_t myTest::ExternalMethod(uint64_t selector, IOUserClientMethodArguments* arguments, const IOUserClientMethodDispatch* dispatch, OSObject* target, void* reference) 
{
    int i = 0;
    CmdStruct* input = nullptr;
    kern_return_t ret = kIOReturnSuccess;

    if (arguments == nullptr) {
        ret = KIOReturnBadArgument;
        return ret;
    }

    if (arguments->structureInput != nullptr) {
        input = (CmdStruct*)arguments->structureInput->getBytesNoCopy();
    }

    // 此处打印input->cmd[0]和input->cmd[1]，结果并非预期的0x12和0x34
}

问题与解决方案 #

1. 如何修复Swift中调用IOConnectCallStructMethod时input参数的警告？ #

警告核心原因是原Swift结构体中的[UInt8]是动态数组类型，本质是带引用计数的对象，而非C语言式的连续内存块，直接取指针传递的是对象元数据而非数组实际内容。

修复方案：修改Swift结构体，使其内存布局与C端完全一致，用固定大小数组替代动态数组：

// Swift 5.1+支持的固定大小数组写法，确保内存连续
@frozen
@_fixedLayout
struct CmdStruct {
    var cmd: [UInt8] = Array(repeating: 0, count: 16)
}

调用时，通过安全方式获取指针，避免直接取对象地址：

withUnsafePointer(to: &input) { ptr in
    ret = IOConnectCallStructMethod(Connection, selector, UnsafeRawPointer(ptr), inputSize, &output, &outputSize)
}

2. 使用MemoryLayout.size得到的inputSize为8，但结构体包含16字节数组，如何在Swift中获取正确的结构体大小？ #

原Swift结构体的[UInt8]是动态数组，64位系统中仅占用8字节（存储指针和长度），而非16字节的数组内容。

修复后，使用固定大小数组的结构体，MemoryLayout<CmdStruct>.size会返回正确的16字节。也可以直接指定或通过对齐步长确保准确性：

// 直接指定数组总大小
let inputSize = MemoryLayout<UInt8>.size * 16
// 或用修复后结构体的内存步长（包含对齐的总大小）
let inputSize = MemoryLayout<CmdStruct>.stride

3. C端使用(CmdStruct*)arguments->structureInput->getBytesNoCopy()转换数据为C结构体是否正确？ #

写法本身无问题，但需补充验证逻辑避免错误：

• 必须确保输入数据长度与C端CmdStruct大小一致，否则会出现内存越界或数据截断：

if (arguments->structureInput != nullptr && arguments->structureInput->getLength() == sizeof(CmdStruct)) {
    input = (CmdStruct*)arguments->structureInput->getBytesNoCopy();
} else {
    ret = KIOReturnBadArgument;
    return ret;
}

• getBytesNoCopy()返回的是原始内存指针，只要Swift端传递的内存布局与C端完全匹配，就能正确解析数据。

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