[BATT] Add the composite battery installer

dll/win32/batt/CMakeLists.txt

@@ -7,5 +7,5 @@ add_library(batt MODULE
     ${CMAKE_CURRENT_BINARY_DIR}/batt.def)
 
 set_module_type(batt win32dll UNICODE)
-add_importlibs(batt msvcrt kernel32 ntdll)
+add_importlibs(batt setupapi msvcrt kernel32 ntdll)
 add_cd_file(TARGET batt DESTINATION reactos/system32 FOR all)

dll/win32/batt/batt.c

@@ -15,27 +15,192 @@
 #include <winuser.h>
 #include <setupapi.h>
 
+#include <initguid.h>
+#include <devguid.h>
+
 #define NDEBUG
 #include <debug.h>
 
-
+static
-BOOL
+DWORD
-WINAPI
+InstallCompositeBattery(
-DllMain(HINSTANCE hinstDll,
+    _In_ HDEVINFO DeviceInfoSet,
-        DWORD dwReason,
+    _In_opt_ PSP_DEVINFO_DATA DeviceInfoData,
-        LPVOID reserved)
+    _In_ PSP_DEVINSTALL_PARAMS_W DeviceInstallParams)
 {
-    switch (dwReason)
+    WCHAR szDeviceId[32];
-    {
+    SP_DRVINFO_DATA DriverInfoData;
-        case DLL_PROCESS_ATTACH:
+    HDEVINFO NewDeviceInfoSet = INVALID_HANDLE_VALUE;
-            DisableThreadLibraryCalls(hinstDll);
+    PSP_DEVINFO_DATA NewDeviceInfoData = NULL;
-            break;
+    BOOL  bDeviceRegistered = FALSE, bHaveDriverInfoList = FALSE;
+    DWORD dwError = ERROR_SUCCESS;
 
-        case DLL_PROCESS_DETACH:
+    DPRINT("InstallCompositeBattery(%p %p %p)\n",
-            break;
+           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;
     }
 
-   return TRUE;
+    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;
+}
+
+
+DWORD
+WINAPI
+BatteryClassInstall(
+    _In_ DI_FUNCTION InstallFunction,
+    _In_ HDEVINFO DeviceInfoSet,
+    _In_opt_ PSP_DEVINFO_DATA DeviceInfoData)
+{
+    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;
 }
 
 
@@ -55,18 +220,24 @@ BatteryClassCoInstaller(IN DI_FUNCTION InstallFunction,
 }
 
 
-DWORD
+BOOL
 WINAPI
-BatteryClassInstall(IN DI_FUNCTION InstallFunction,
+DllMain(
-                    IN HDEVINFO DeviceInfoSet,
+    _In_ HINSTANCE hinstDll,
-                    IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL)
+    _In_ DWORD dwReason,
+    _In_ LPVOID reserved)
 {
-    switch (InstallFunction)
+    switch (dwReason)
     {
-        default:
+        case DLL_PROCESS_ATTACH:
-            DPRINT("Install function %u ignored\n", InstallFunction);
+            DisableThreadLibraryCalls(hinstDll);
-            return ERROR_DI_DO_DEFAULT;
+            break;
+
+        case DLL_PROCESS_DETACH:
+            break;
     }
+
+   return TRUE;
 }
 
 /* EOF */