batt.c File Reference

#include <stdarg.h>
#include <windef.h>
#include <winbase.h>
#include <winreg.h>
#include <winuser.h>
#include <setupapi.h>
#include <initguid.h>
#include <devguid.h>
#include <debug.h>

Include dependency graph for batt.c:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS
#define	NDEBUG

Functions
static DWORD	InstallCompositeBattery (_In_ HDEVINFO DeviceInfoSet, _In_opt_ PSP_DEVINFO_DATA DeviceInfoData, _In_ PSP_DEVINSTALL_PARAMS_W DeviceInstallParams)
DWORD WINAPI	BatteryClassInstall (_In_ DI_FUNCTION InstallFunction, _In_ HDEVINFO DeviceInfoSet, _In_opt_ PSP_DEVINFO_DATA DeviceInfoData)
DWORD WINAPI	BatteryClassCoInstaller (IN DI_FUNCTION InstallFunction, IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN OUT PCOINSTALLER_CONTEXT_DATA Context)
BOOL WINAPI	DllMain (_In_ HINSTANCE hinstDll, _In_ DWORD dwReason, _In_ LPVOID reserved)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 21 of file batt.c.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 10 of file batt.c.

Function Documentation

BatteryClassCoInstaller()

DWORD WINAPI BatteryClassCoInstaller	(	IN DI_FUNCTION	InstallFunction,
		IN HDEVINFO	DeviceInfoSet,
		IN PSP_DEVINFO_DATA DeviceInfoData	OPTIONAL,
		IN OUT PCOINSTALLER_CONTEXT_DATA	Context
	)

Definition at line 209 of file batt.c.

{
    switch (InstallFunction)
    {
        default:
            DPRINT("Install function %u ignored\n", InstallFunction);
            return ERROR_DI_DO_DEFAULT;
    }
}

BatteryClassInstall()

DWORD WINAPI BatteryClassInstall	(	_In_ DI_FUNCTION	InstallFunction,
		_In_ HDEVINFO	DeviceInfoSet,
		_In_opt_ PSP_DEVINFO_DATA	DeviceInfoData
	)

Definition at line 170 of file batt.c.

{
    SP_DEVINSTALL_PARAMS_W DeviceInstallParams;
    DWORD dwError;
 
    DPRINT("BatteryClassInstall(%u %p %p)\n",
           InstallFunction, DeviceInfoSet, DeviceInfoData);
 
    if (InstallFunction != DIF_INSTALLDEVICE)
        return ERROR_DI_DO_DEFAULT;
 
    DeviceInstallParams.cbSize = sizeof(DeviceInstallParams);
    if (!SetupDiGetDeviceInstallParamsW(DeviceInfoSet,
                                        DeviceInfoData,
                                        &DeviceInstallParams))
    {
        DPRINT1("SetupDiGetDeviceInstallParamsW() failed (Error %lu)\n", GetLastError());
        return GetLastError();
    }
 
    /* Install the composite battery device */
    dwError = InstallCompositeBattery(DeviceInfoSet,
                                      DeviceInfoData,
                                      &DeviceInstallParams);
    if (dwError == ERROR_SUCCESS)
    {
        /* Install the battery device */
        dwError = ERROR_DI_DO_DEFAULT;
    }
 
    return dwError;
}

DllMain()

BOOL WINAPI DllMain	(	_In_ HINSTANCE	hinstDll,
		_In_ DWORD	dwReason,
		_In_ LPVOID	reserved
	)

Definition at line 225 of file batt.c.

{
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
            DisableThreadLibraryCalls(hinstDll);
            break;
 
        case DLL_PROCESS_DETACH:
            break;
    }
 
   return TRUE;
}

InstallCompositeBattery()

static DWORD InstallCompositeBattery ( _In_ HDEVINFO  DeviceInfoSet, _In_opt_ PSP_DEVINFO_DATA  DeviceInfoData, _In_ PSP_DEVINSTALL_PARAMS_W  DeviceInstallParams  )

static

Definition at line 26 of file batt.c.

{
    WCHAR szDeviceId[32];
    SP_DRVINFO_DATA DriverInfoData;
    HDEVINFO NewDeviceInfoSet = INVALID_HANDLE_VALUE;
    PSP_DEVINFO_DATA NewDeviceInfoData = NULL;
    BOOL  bDeviceRegistered = FALSE, bHaveDriverInfoList = FALSE;
    DWORD dwError = ERROR_SUCCESS;
 
    DPRINT("InstallCompositeBattery(%p %p %p)\n",
           DeviceInfoSet, DeviceInfoData, DeviceInstallParams);
 
    NewDeviceInfoSet = SetupDiCreateDeviceInfoList(&GUID_DEVCLASS_SYSTEM,
                                                   DeviceInstallParams->hwndParent);
    if (NewDeviceInfoSet == INVALID_HANDLE_VALUE)
    {
        DPRINT1("SetupDiCreateDeviceInfoList() failed (Error %lu)\n", GetLastError());
        return GetLastError();
    }
 
    NewDeviceInfoData = HeapAlloc(GetProcessHeap(),
                                  HEAP_ZERO_MEMORY,
                                  sizeof(SP_DEVINFO_DATA));
    if (NewDeviceInfoData == NULL)
    {
        dwError = ERROR_OUTOFMEMORY;
        goto done;
    }
 
    NewDeviceInfoData->cbSize = sizeof(SP_DEVINFO_DATA);
    if (!SetupDiCreateDeviceInfoW(NewDeviceInfoSet,
                                  L"Root\\COMPOSITE_BATTERY\\0000",
                                  &GUID_DEVCLASS_SYSTEM,
                                  NULL,
                                  DeviceInstallParams->hwndParent,
                                  0,
                                  NewDeviceInfoData))
    {
        dwError = GetLastError();
        if (dwError == ERROR_DEVINST_ALREADY_EXISTS)
        {
            dwError = ERROR_SUCCESS;
            goto done;
        }
 
        DPRINT1("SetupDiCreateDeviceInfoW() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    if (!SetupDiRegisterDeviceInfo(NewDeviceInfoSet,
                                   NewDeviceInfoData,
                                   0,
                                   NULL,
                                   NULL,
                                   NULL))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiRegisterDeviceInfo() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    bDeviceRegistered = TRUE;
 
    ZeroMemory(szDeviceId, sizeof(szDeviceId));
    wcscpy(szDeviceId, L"COMPOSITE_BATTERY");
 
    if (!SetupDiSetDeviceRegistryPropertyW(NewDeviceInfoSet,
                                           NewDeviceInfoData,
                                           SPDRP_HARDWAREID,
                                           (PBYTE)szDeviceId,
                                           (wcslen(szDeviceId) + 2) * sizeof(WCHAR)))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiSetDeviceRegistryPropertyW() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    if (!SetupDiBuildDriverInfoList(NewDeviceInfoSet,
                                    NewDeviceInfoData,
                                    SPDIT_COMPATDRIVER))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiBuildDriverInfoList() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    bHaveDriverInfoList = TRUE;
 
    DriverInfoData.cbSize = sizeof(SP_DRVINFO_DATA);
    if (!SetupDiEnumDriverInfo(NewDeviceInfoSet,
                               NewDeviceInfoData,
                               SPDIT_COMPATDRIVER,
                               0,
                               &DriverInfoData))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiEnumDriverInfo() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    if (!SetupDiSetSelectedDriver(NewDeviceInfoSet,
                                  NewDeviceInfoData,
                                  &DriverInfoData))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiSetSelectedDriver() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    if (!SetupDiInstallDevice(NewDeviceInfoSet,
                              NewDeviceInfoData))
    {
        dwError = GetLastError();
        DPRINT1("SetupDiInstallDevice() failed (Error %lu 0x%08lx)\n", dwError, dwError);
        goto done;
    }
 
    dwError = ERROR_SUCCESS;
 
done:
    if (bHaveDriverInfoList)
        SetupDiDestroyDriverInfoList(NewDeviceInfoSet,
                                     NewDeviceInfoData,
                                     SPDIT_COMPATDRIVER);
 
    if (bDeviceRegistered)
        SetupDiDeleteDeviceInfo(NewDeviceInfoSet,
                                NewDeviceInfoData);
 
    if (NewDeviceInfoData != NULL)
        HeapFree(GetProcessHeap(), 0, NewDeviceInfoData);
 
    if (NewDeviceInfoSet != INVALID_HANDLE_VALUE)
        SetupDiDestroyDeviceInfoList(NewDeviceInfoSet);
 
    return dwError;
}

Referenced by BatteryClassInstall().