SetupAPI.h

/*
    SPDX-License-Identifier: MIT
    Copyright © 1991-2022 Amebis
    Copyright © 2016 GÉANT
*/
 
#pragma once
 
#include "Common.h"
#include <SetupAPI.h>
 
namespace winstd
{
 
    class setup_device_info_list : public handle<HDEVINFO, INVALID_HANDLE_VALUE>
    {
        WINSTD_HANDLE_IMPL(setup_device_info_list, INVALID_HANDLE_VALUE)
 
    public:
        virtual ~setup_device_info_list()
        {
            if (m_h != invalid)
                free_internal();
        }
 
        bool create(
            _In_opt_ const GUID * ClassGuid,
            _In_opt_ HWND         hwndParent) noexcept
        {
            handle_type h = SetupDiCreateDeviceInfoList(ClassGuid, hwndParent);
            if (h != invalid) {
                attach(h);
                return true;
            }
            else
                return false;
        }
 
        bool create(
            _In_opt_   const GUID * ClassGuid,
            _In_opt_   PCTSTR       Enumerator,
            _In_opt_   HWND         hwndParent,
            _In_       DWORD        Flags,
            _In_opt_   HDEVINFO     DeviceInfoSet,
            _In_opt_   PCTSTR       MachineName,
            _Reserved_ PVOID        Reserved) noexcept
        {
            handle_type h = SetupDiGetClassDevsEx(ClassGuid, Enumerator, hwndParent, Flags, DeviceInfoSet, MachineName, Reserved);
            if (h != invalid) {
                attach(h);
                return true;
            }
            else
                return false;
        }
 
    protected:
        void free_internal() noexcept override
        {
            SetupDiDestroyDeviceInfoList(m_h);
        }
    };
 
    class setup_driver_info_list_builder
    {
        WINSTD_NONCOPYABLE(setup_driver_info_list_builder)
        WINSTD_NONMOVABLE(setup_driver_info_list_builder)
 
    public:
        setup_driver_info_list_builder(
            _In_        HDEVINFO         DeviceInfoSet,
            _Inout_opt_ PSP_DEVINFO_DATA DeviceInfoData,
            _In_        DWORD            DriverType) noexcept :
            m_DeviceInfoSet (DeviceInfoSet),
            m_DeviceInfoData(DeviceInfoData),
            m_DriverType    (DriverType)
        {
            m_result = SetupDiBuildDriverInfoList(m_DeviceInfoSet, m_DeviceInfoData, m_DriverType);
        }
 
        virtual ~setup_driver_info_list_builder()
        {
            if (m_result)
                SetupDiDestroyDriverInfoList(m_DeviceInfoSet, m_DeviceInfoData, m_DriverType);
        }
 
        BOOL status() const noexcept
        {
            return m_result;
        }
 
    protected:
        HDEVINFO         m_DeviceInfoSet;
        PSP_DEVINFO_DATA m_DeviceInfoData;
        DWORD            m_DriverType;
        BOOL             m_result;
    };
 
}