Quellcode

#include <iostream>

#include <windows.h>
#include <initguid.h>
#include <devpkey.h>
#include <setupapi.h>

DEFINE_GUID(GUID_CLS_WINUSB, 0x88BAE032, 0x5A81, 0x49F0, 0xBC, 0x3D, 0xA4, 0xFF, 0x13, 0x82, 0x16, 0xD6);

int main(int argc, char **argv) {
    HDEVINFO deviceInfoSet = SetupDiGetClassDevsW(&GUID_CLS_WINUSB, L"USB", NULL, DIGCF_PRESENT);
    if (deviceInfoSet == INVALID_HANDLE_VALUE) {
        std::cout << "Failed enumarating devices";
        return 1;
    }

    int deviceInfoIndex = 0;
    SP_DEVINFO_DATA spDevinfoData = {0};
    spDevinfoData.cbSize = sizeof(spDevinfoData);

    std::cout << "Devices: " << std::endl;

    while (SetupDiEnumDeviceInfo(deviceInfoSet, deviceInfoIndex, &spDevinfoData)) {
        BYTE buffer[256] = {0};

        std::wcout << std::wstring(reinterpret_cast<WCHAR*>(buffer), StringFromGUID2(spDevinfoData.ClassGuid, reinterpret_cast<WCHAR*>(buffer), sizeof(buffer) / sizeof(WCHAR))) << ":";

        DEVPROPTYPE dataType;
        if (SetupDiGetDevicePropertyW(deviceInfoSet, &spDevinfoData, &DEVPKEY_Device_FriendlyName, &dataType, buffer, sizeof(buffer) / sizeof(WCHAR), NULL, 0)) {
            std::wcout << std::wstring(reinterpret_cast<WCHAR*>(buffer));
        }
 
        if (SetupDiGetDevicePropertyW(deviceInfoSet, &spDevinfoData, &DEVPKEY_Device_SafeRemovalRequiredOverride, &dataType, buffer, 1, NULL, 0)) {
            std::cout << ": Override=" << (buffer[0] != DEVPROP_FALSE);
        }

        dataType = DEVPROP_TYPE_BOOLEAN;
        buffer[0] = DEVPROP_FALSE;
        if (SetupDiSetDevicePropertyW(deviceInfoSet, &spDevinfoData, &DEVPKEY_Device_SafeRemovalRequiredOverride, dataType, buffer, 1, 0)) {
            std::cout << ": Override set";
        } else {
            std::cout << ": Override not set (" << GetLastError() << ")";
        }

        std::cout << std::endl;

        ++deviceInfoIndex;
    }

    SetupDiDestroyDeviceInfoList(deviceInfoSet);

    return 0;
}