From 155540a8f70def07ae6cc30f03494b27b71ea05f Mon Sep 17 00:00:00 2001 From: Quantum Date: Fri, 12 Sep 2025 04:01:30 -0400 Subject: [PATCH] [idd] install: create new driver installer/uninstaller --- idd/LGIdd.sln | 18 + idd/LGIddInstall/LGIddInstall.c | 324 ++++++++++++++++++ idd/LGIddInstall/LGIddInstall.vcxproj | 139 ++++++++ idd/LGIddInstall/LGIddInstall.vcxproj.filters | 22 ++ 4 files changed, 503 insertions(+) create mode 100644 idd/LGIddInstall/LGIddInstall.c create mode 100644 idd/LGIddInstall/LGIddInstall.vcxproj create mode 100644 idd/LGIddInstall/LGIddInstall.vcxproj.filters diff --git a/idd/LGIdd.sln b/idd/LGIdd.sln index d819e628..018107c1 100644 --- a/idd/LGIdd.sln +++ b/idd/LGIdd.sln @@ -20,6 +20,8 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "LGCommon", "LGCommon", "{AC LGCommon\PipeMsg.h = LGCommon\PipeMsg.h EndProjectSection EndProject +Project("{8BC9CEB8-8B4A-11D0-8D11-00A0C91BC942}") = "LGIddInstall", "LGIddInstall\LGIddInstall.vcxproj", "{23EC8370-5F3B-4D18-8C31-E89A154F3666}" +EndProject Global GlobalSection(SolutionConfigurationPlatforms) = preSolution Debug|ARM = Debug|ARM @@ -86,6 +88,22 @@ Global {0045D7AD-3F26-4B87-81CB-78D18839596D}.Release|x64.Build.0 = Release|x64 {0045D7AD-3F26-4B87-81CB-78D18839596D}.Release|x86.ActiveCfg = Release|Win32 {0045D7AD-3F26-4B87-81CB-78D18839596D}.Release|x86.Build.0 = Release|Win32 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|ARM.ActiveCfg = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|ARM.Build.0 = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|ARM64.ActiveCfg = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|ARM64.Build.0 = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|x64.ActiveCfg = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|x64.Build.0 = Debug|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|x86.ActiveCfg = Debug|Win32 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Debug|x86.Build.0 = Debug|Win32 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|ARM.ActiveCfg = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|ARM.Build.0 = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|ARM64.ActiveCfg = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|ARM64.Build.0 = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|x64.ActiveCfg = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|x64.Build.0 = Release|x64 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|x86.ActiveCfg = Release|Win32 + {23EC8370-5F3B-4D18-8C31-E89A154F3666}.Release|x86.Build.0 = Release|Win32 EndGlobalSection GlobalSection(SolutionProperties) = preSolution HideSolutionNode = FALSE diff --git a/idd/LGIddInstall/LGIddInstall.c b/idd/LGIddInstall/LGIddInstall.c new file mode 100644 index 00000000..e7725bf5 --- /dev/null +++ b/idd/LGIddInstall/LGIddInstall.c @@ -0,0 +1,324 @@ +#define WIN32_LEAN_AND_MEAN + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define LGIDD_CLASS_GUID GUID_DEVCLASS_DISPLAY +#define LGIDD_CLASS_NAME L"Display" +#define LGIDD_HWID L"Root\\LGIdd" +#define LGIDD_HWID_MULTI_SZ (LGIDD_HWID "\0") +#define LGIDD_INF_NAME L"LGIdd.inf" + +void usage(wchar_t *program) +{ + wprintf(L"Usage: %s \n", program); + exit(2); +} + +void debugWinError(const wchar_t *desc, HRESULT status) +{ + wchar_t *buffer; + if (!FormatMessageW( + FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_IGNORE_INSERTS, + NULL, + status, + MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), + (LPWSTR) &buffer, + 1024, + NULL + )) + { + fwprintf(stderr, L"%s: 0x%08lx: FormatMessage failed with code 0x%08lx\n", desc, status, GetLastError()); + return; + } + + for (size_t i = wcslen(buffer) - 1; i > 0; --i) + if (buffer[i] == L'\n' || buffer[i] == L'\r') + buffer[i] = 0; + + fwprintf(stderr, L"%s: 0x%08lx: %s\n", desc, status, buffer); + LocalFree(buffer); +} + +typedef bool (*IDD_FOUND_PROC)(HDEVINFO hDevInfo, PSP_DEVINFO_DATA pDevInfo, void *pContext); + +bool findIddDevice(IDD_FOUND_PROC procFound, void *pContext) +{ + HDEVINFO hDevInfo = SetupDiGetClassDevsW(&LGIDD_CLASS_GUID, NULL, NULL, DIGCF_ALLCLASSES | DIGCF_PRESENT); + if (hDevInfo == INVALID_HANDLE_VALUE) + { + debugWinError(L"SetupDiGetClassDevsW", GetLastError()); + return false; + } + + SP_DEVINFO_DATA devInfo = { .cbSize = sizeof devInfo, 0 }; + for (DWORD dwIndex = 0; SetupDiEnumDeviceInfo(hDevInfo, dwIndex, &devInfo); ++dwIndex) + { + DWORD dwSizeRequired; + DWORD dwPropertyType; + SetupDiGetDeviceRegistryPropertyW(hDevInfo, &devInfo, SPDRP_HARDWAREID, &dwPropertyType, NULL, 0, &dwSizeRequired); + + DWORD dwLastError = GetLastError(); + if (dwLastError == ERROR_INVALID_DATA) + continue; + else if (dwLastError != ERROR_INSUFFICIENT_BUFFER) + { + debugWinError(L"SetupDiGetDeviceRegistryPropertyW(SPDRP_HARDWAREID) size calculation", dwLastError); + goto fail; + } + + if (dwPropertyType != REG_MULTI_SZ) + { + fwprintf(stderr, L"SetupDiGetDeviceRegistryPropertyW(SPDRP_HARDWAREID) returned wrong type\n"); + goto fail; + } + + LPWSTR lpBuffer = malloc(dwSizeRequired); + if (!lpBuffer) + { + fwprintf(stderr, L"failed to allocate memory for SetupDiGetDeviceRegistryPropertyW(SPDRP_HARDWAREID)\n"); + goto fail; + } + + if (!SetupDiGetDeviceRegistryPropertyW(hDevInfo, &devInfo, SPDRP_HARDWAREID, &dwPropertyType, (PBYTE)lpBuffer, dwSizeRequired, NULL)) + { + debugWinError(L"SetupDiGetDeviceRegistryPropertyW(SPDRP_HARDWAREID) for real", GetLastError()); + free(lpBuffer); + goto fail; + } + + bool found = false; + + for (LPWSTR lpHwId = lpBuffer; *lpHwId; lpHwId += wcslen(lpBuffer) + 1) + { + if (!lstrcmpiW(lpHwId, LGIDD_HWID)) + { + found = true; + break; + } + } + + free(lpBuffer); + + if (found && !procFound(hDevInfo, &devInfo, pContext)) + break; + } + + SetupDiDestroyDeviceInfoList(hDevInfo); + return true; + +fail: + SetupDiDestroyDeviceInfoList(hDevInfo); + return false; +} + +enum DeviceCreated { + DEVICE_CREATED, + DEVICE_NOT_CREATED, + DEVICE_UNKNOWN, +}; + +bool isIddDeviceCreatedEnum(HDEVINFO hDevInfo, PSP_DEVINFO_DATA pDevInfo, void *pContext) +{ + enum DeviceCreated *result = pContext; + *result = DEVICE_CREATED; + return false; +} + +enum DeviceCreated isIddDeviceCreated() +{ + enum DeviceCreated result = DEVICE_UNKNOWN; + if (findIddDevice(isIddDeviceCreatedEnum, &result) && result == DEVICE_UNKNOWN) + result = DEVICE_NOT_CREATED; + return result; +} + +bool createIddDevice(void) +{ + HDEVINFO hDevInfo = SetupDiCreateDeviceInfoList(&LGIDD_CLASS_GUID, NULL); + if (hDevInfo == INVALID_HANDLE_VALUE) + { + debugWinError(L"SetupDiCreateDeviceInfoList", GetLastError()); + return false; + } + + SP_DEVINFO_DATA devInfo = { .cbSize = sizeof devInfo, 0 }; + if (!SetupDiCreateDeviceInfoW(hDevInfo, LGIDD_CLASS_NAME, &LGIDD_CLASS_GUID, NULL, NULL, DICD_GENERATE_ID, &devInfo)) + { + debugWinError(L"SetupDiCreateDeviceInfoW", GetLastError()); + goto fail; + } + + if (!SetupDiSetDeviceRegistryPropertyW(hDevInfo, &devInfo, SPDRP_HARDWAREID, (PBYTE) LGIDD_HWID_MULTI_SZ, sizeof LGIDD_HWID_MULTI_SZ)) + { + debugWinError(L"SetupDiSetDeviceRegistryPropertyW", GetLastError()); + goto fail; + } + + if (!SetupDiCallClassInstaller(DIF_REGISTERDEVICE, hDevInfo, &devInfo)) + { + debugWinError(L"SetupDiCallClassInstaller", GetLastError()); + goto fail; + } + + return true; + +fail: + SetupDiDestroyDeviceInfoList(hDevInfo); + return false; +} + +bool destroyIddDeviceEnum(HDEVINFO hDevInfo, PSP_DEVINFO_DATA pDevInfo, void* pContext) +{ + LPBOOL pbNeedRestart = pContext; + BOOL bNeedRestart; + WCHAR szInfPath[MAX_PATH] = { 0 }; + + if (!SetupDiBuildDriverInfoList(hDevInfo, pDevInfo, SPDIT_COMPATDRIVER)) + { + debugWinError(L"SetupDiBuildDriverInfoList", GetLastError()); + goto uninstall; + } + + SP_DRVINFO_DATA_W drvInfo = { .cbSize = sizeof drvInfo }; + if (!SetupDiEnumDriverInfoW(hDevInfo, pDevInfo, SPDIT_COMPATDRIVER, 0, &drvInfo)) + { + debugWinError(L"SetupDiEnumDriverInfoW", GetLastError()); + goto uninstall; + } + + SP_DRVINFO_DETAIL_DATA_W drvInfoDetail = { .cbSize = sizeof drvInfoDetail }; + SetupDiGetDriverInfoDetailW(hDevInfo, pDevInfo, &drvInfo, &drvInfoDetail, sizeof drvInfoDetail, NULL); + + DWORD dwLastError = GetLastError(); + if (dwLastError == ERROR_INSUFFICIENT_BUFFER) + wcscpy_s(szInfPath, MAX_PATH, drvInfoDetail.InfFileName); + else + debugWinError(L"SetupDiEnumDriverInfoW", GetLastError()); + +uninstall: + if (DiUninstallDevice(NULL, hDevInfo, pDevInfo, 0, &bNeedRestart)) + *pbNeedRestart |= bNeedRestart; + else + { + debugWinError(L"DiUninstallDevice", GetLastError()); + return true; + } + + if (*szInfPath) + { + if (DiUninstallDriverW(NULL, szInfPath, 0, &bNeedRestart)) + *pbNeedRestart |= bNeedRestart; + else + debugWinError(L"DiUninstallDriverW", GetLastError()); + } + + return true; +} + +void destroyIddDevice(LPBOOL pbNeedRestart) +{ + findIddDevice(destroyIddDeviceEnum, pbNeedRestart); +} + +bool getIddInfPath(LPWSTR lpszInf) +{ + WCHAR szDir[MAX_PATH]; + WCHAR szInf[MAX_PATH]; + + if (!GetModuleFileNameW(NULL, szDir, MAX_PATH)) + { + debugWinError(L"GetModuleFileNameW", GetLastError()); + return false; + } + + *PathFindFileNameW(szDir) = 0; + if (!PathCombineW(lpszInf, szDir, LGIDD_INF_NAME)) + { + debugWinError(L"PathCombineW", GetLastError()); + return false; + } + + if (!PathFileExistsW(lpszInf)) + { + fwprintf(stderr, L"INF file does not exist: %s\n", szInf); + return false; + } + + return true; +} + +bool installIddInf(PBOOL pbNeedRestart) +{ + WCHAR szInf[MAX_PATH]; + + if (!getIddInfPath(szInf)) + return false; + + if (!DiInstallDriverW(NULL, szInf, DIIRFLAG_FORCE_INF, pbNeedRestart)) + { + debugWinError(L"DiInstallDriverW", GetLastError()); + return false; + } + + return true; +} + +void install() +{ + switch (isIddDeviceCreated()) + { + case DEVICE_NOT_CREATED: + wprintf(L"Creating LGIdd device: %s...\n", LGIDD_HWID); + if (!createIddDevice()) + return; + + // fallthrough + case DEVICE_CREATED: + _putws(L"Installing INF..."); + BOOL bNeedRestart; + if (!installIddInf(&bNeedRestart)) + return; + + if (bNeedRestart) + { + _putws(L"Restart required to complete installation"); + exit(12); + } + break; + case DEVICE_UNKNOWN: + exit(1); + } +} + +void uninstall() +{ + BOOL bNeedRestart = 0; + destroyIddDevice(&bNeedRestart); + + if (bNeedRestart) + { + _putws(L"Restart required to complete installation"); + exit(12); + } +} + +int wmain(int argc, wchar_t **argv) +{ + if (argc != 2) + usage(argv[0]); + + if (!wcscmp(argv[1], L"install")) + install(); + else if (!wcscmp(argv[1], L"uninstall")) + uninstall(); + else + usage(argv[0]); +} diff --git a/idd/LGIddInstall/LGIddInstall.vcxproj b/idd/LGIddInstall/LGIddInstall.vcxproj new file mode 100644 index 00000000..cc68e715 --- /dev/null +++ b/idd/LGIddInstall/LGIddInstall.vcxproj @@ -0,0 +1,139 @@ + + + + + Debug + Win32 + + + Release + Win32 + + + Debug + x64 + + + Release + x64 + + + + 17.0 + Win32Proj + {23ec8370-5f3b-4d18-8c31-e89a154f3666} + LGIddInstall + 10.0 + + + + Application + true + v143 + Unicode + + + Application + false + v143 + true + Unicode + + + Application + true + v143 + Unicode + + + Application + false + v143 + true + Unicode + + + + + + + + + + + + + + + + + + + + + + Level3 + true + WIN32;_DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + MultiThreadedDebug + + + Console + true + setupapi.lib;shlwapi.lib;newdev.lib;%(AdditionalDependencies) + + + + + Level3 + true + true + true + WIN32;NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + MultiThreaded + + + Console + true + setupapi.lib;shlwapi.lib;newdev.lib;%(AdditionalDependencies) + + + + + Level3 + true + _DEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + MultiThreadedDebug + + + Console + true + setupapi.lib;shlwapi.lib;newdev.lib;%(AdditionalDependencies) + + + + + Level3 + true + true + true + NDEBUG;_CONSOLE;%(PreprocessorDefinitions) + true + MultiThreaded + + + Console + true + setupapi.lib;shlwapi.lib;newdev.lib;%(AdditionalDependencies) + + + + + + + + + \ No newline at end of file diff --git a/idd/LGIddInstall/LGIddInstall.vcxproj.filters b/idd/LGIddInstall/LGIddInstall.vcxproj.filters new file mode 100644 index 00000000..c02e816c --- /dev/null +++ b/idd/LGIddInstall/LGIddInstall.vcxproj.filters @@ -0,0 +1,22 @@ + + + + + {4FC737F1-C7A5-4376-A066-2A32D752A2FF} + cpp;c;cc;cxx;c++;cppm;ixx;def;odl;idl;hpj;bat;asm;asmx + + + {93995380-89BD-4b04-88EB-625FBE52EBFB} + h;hh;hpp;hxx;h++;hm;inl;inc;ipp;xsd + + + {67DA6AB6-F800-4c08-8B7A-83BB121AAD01} + rc;ico;cur;bmp;dlg;rc2;rct;bin;rgs;gif;jpg;jpeg;jpe;resx;tiff;tif;png;wav;mfcribbon-ms + + + + + Source Files + + + \ No newline at end of file