使用 SetupDiXxx 结构的堆栈损坏

Question

我在用于获取设备信息的结构方面遇到了一些问题。据我了解，正确设置 cbSize 有点棘手，因此 API 正在将数据写入超出预期的位置（导致堆栈损坏）。到目前为止，我有以下代码：

GUID guid;
HidD_GetHidGuid(&guid);

HDEVINFO info;
info = SetupDiGetClassDevs(&guid, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);

SP_DEVINFO_DATA DeviceInfoData;

memset(&DeviceInfoData, 0, sizeof(SP_DEVINFO_DATA));
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);


int deviceIndex = 0;
while (SetupDiEnumDeviceInfo(info, deviceIndex++, &DeviceInfoData))
{
    SP_INTERFACE_DEVICE_DATA data;
    data.cbSize = sizeof(SP_INTERFACE_DEVICE_DATA);

    int interfaceIndex = 0;


    while (SetupDiEnumDeviceInterfaces(info, &DeviceInfoData, &guid, interfaceIndex++, &data))
    {

        //https://msdn.microsoft.com/en-us/library/windows/hardware/ff551120%28v=vs.85%29.aspx
        //Get the required buffer size. Call SetupDiGetDeviceInterfaceDetail with a NULLDeviceInterfaceDetailData pointer, 
        //a DeviceInterfaceDetailDataSize of zero, and a valid RequiredSize variable. In response to such a call, this function
        //returns the required buffer size at RequiredSize and fails with GetLastError returning ERROR_INSUFFICIENT_BUFFER.

        SP_DEVICE_INTERFACE_DETAIL_DATA interfaceData;
        interfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);

        DWORD bufferSize = 0;
        SetupDiGetDeviceInterfaceDetail(info, &data, NULL, 0, &bufferSize, nullptr);

        if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
        {
            //Call the function again
            SetupDiGetDeviceInterfaceDetail(info, &data, &interfaceData, bufferSize, NULL, &DeviceInfoData);

            DWORD error = GetLastError();
            if (error != ERROR_SUCCESS)
            {
                printf("Could not obtain device interface details. Error: %d \n", error);
            }
        }
    }

我得到的错误是：

    Run-Time Check Failure #2 - Stack around the variable 'DeviceInfoData' was corrupted.

虽然我已经看到 SP_INTERFACE_DEVICE_DATA 和 SP_DEVICE_INTERFACE_DETAIL_DATA 导致同样的错误

非常感谢任何帮助！

Answer 1

您的interfaceData 缓冲区似乎太小了。

再次检查 the documentation 的 DeviceInterfaceDetailData 参数到 SetupDiGetDeviceInterfaceDetail。

Answer 2

如果你想获得更多关于驱动程序开发的信息，我推荐这本书USB Complete。我根据他们的解释解决了这个问题。问题如下：

首先，获取缓冲区大小：

SetupDiGetDeviceInterfaceDetail(info, &data, NULL, 0, &bufferSize, nullptr);

然后，根据返回的大小使用 malloc 手动分配 PSP_DEVICE_INTERFACE_DETAIL_DATA 结构：

PSP_DEVICE_INTERFACE_DETAIL_DATA interfaceData;
interfaceData->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);

注意PSP_DEVICE_INTERFACE_DETAIL_DATA 前面的P。这是此 API 的 Microsoft 语义。它代表指针；查看文档时很容易错过的东西（如果你也错过了->）

SetupDiGetDeviceInterfaceDetail 函数返回整个结构的大小，因此您需要将其分配给该大小。我已经看到尝试增加大小直到错误消失的示例。这种方法是错误的……有很多原因。从SetupDiGetDeviceInterfaceDetail 获取大小，然后根据该大小分配整个PSP_DEVICE_INTERFACE_DETAIL_DATA 内存块。不要忘记将 cbSize 设置为结构体的大小SP_DEVICE_INTERFACE_DETAIL_DATA

再次注意命名约定中的P，因为很容易误认sizeof(PSP_DEVICE_INTERFACE_DETAIL_DATA)。