Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-08-02 17:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity

[DEVMGR]

Browse source 
CORE-6350
Hack: Call the "Install hardware" wizard when the user clicks the "update driver" button.

svn path=/trunk/; revision=61270
This commit is contained in:
Eric Kohl 2013-12-14 15:58:35 +00:00
parent 952460b582
commit 671f721ef1
2 changed files with 69 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

119
reactos/dll/win32/devmgr/advprop.c
View file

@ -398,6 +398,73 @@ DriverDetailsDlgProc(IN HWND hwndDlg,
return Ret; return Ret;
} }
BOOL
WINAPI
DevInstallW(
IN HWND hWndParent,
IN HINSTANCE hInstance,
IN LPCWSTR InstanceId,
IN INT Show);
static
VOID
UpdateDriver(
IN HWND hwndDlg,
IN PDEVADVPROP_INFO dap)
{
TOKEN_PRIVILEGES Privileges;
HANDLE hToken;
BOOL NeedReboot = FALSE;
// Better use InstallDevInst:
// BOOL
// WINAPI
// InstallDevInst(
// HWND hWnd,
// LPWSTR wszDeviceId,
// BOOL bUpdate,
// DWORD *dwReboot);
// See: http://comp.os.ms-windows.programmer.win32.narkive.com/J8FTd4KK/signature-of-undocumented-installdevinstex
if (!DevInstallW(hwndDlg, NULL, dap->szDeviceID, SW_SHOWNOACTIVATE))
return;
if (NeedReboot == FALSE)
return;
//FIXME: load text from resource file
if (MessageBoxW(hwndDlg, L"Reboot now?", L"Reboot required", MB_YESNO | MB_ICONQUESTION) != IDYES)
return;
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
{
DPRINT("OpenProcessToken failed\n");
return;
}
/* Get the LUID for the Shutdown privilege */
if (!LookupPrivilegeValueW(NULL, SE_SHUTDOWN_NAME, &Privileges.Privileges[0].Luid))
{
DPRINT("LookupPrivilegeValue failed\n");
return;
}
/* Assign the Shutdown privilege to our process */
Privileges.PrivilegeCount = 1;
Privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
if (!AdjustTokenPrivileges(hToken, FALSE, &Privileges, 0, NULL, NULL))
{
DPRINT("AdjustTokenPrivileges failed\n");
return;
}
/* Finally shut down the system */
if (!ExitWindowsEx(EWX_REBOOT, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER | SHTDN_REASON_FLAG_PLANNED))
{
DPRINT("ExitWindowsEx failed\n");
}
}
static VOID static VOID
UpdateDriverDlg(IN HWND hwndDlg, UpdateDriverDlg(IN HWND hwndDlg,
@ -486,64 +553,16 @@ AdvProcDriverDlgProc(IN HWND hwndDlg,
switch (LOWORD(wParam)) switch (LOWORD(wParam))
{ {
case IDC_DRIVERDETAILS: case IDC_DRIVERDETAILS:
{
DialogBoxParam(hDllInstance, DialogBoxParam(hDllInstance,
MAKEINTRESOURCE(IDD_DRIVERDETAILS), MAKEINTRESOURCE(IDD_DRIVERDETAILS),
hwndDlg, hwndDlg,
DriverDetailsDlgProc, DriverDetailsDlgProc,
(ULONG_PTR)dap); (ULONG_PTR)dap);
break; break;
}
case IDC_UPDATEDRIVER: case IDC_UPDATEDRIVER:
{ UpdateDriver(hwndDlg, dap);
if (dap->CurrentDeviceInfoSet != INVALID_HANDLE_VALUE)
{
BOOL NeedReboot;
if (DiShowUpdateDevice(hwndDlg, dap->CurrentDeviceInfoSet, &dap->CurrentDeviceInfoData, 0, &NeedReboot))
{
if (NeedReboot)
{
//FIXME: load text from resource file
if(MessageBoxW(hwndDlg, L"Reboot now?", L"Reboot required", MB_YESNO | MB_ICONQUESTION) == IDYES)
{
HANDLE hToken;
TOKEN_PRIVILEGES Privileges;
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
{
DPRINT("OpenProcessToken failed\n");
break;
}
/* Get the LUID for the Shutdown privilege */
if (!LookupPrivilegeValueW(NULL, SE_SHUTDOWN_NAME, &Privileges.Privileges[0].Luid))
{
DPRINT("LookupPrivilegeValue failed\n");
break;
}
/* Assign the Shutdown privilege to our process */
Privileges.PrivilegeCount = 1;
Privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
if (!AdjustTokenPrivileges(hToken, FALSE, &Privileges, 0, NULL, NULL))
{
DPRINT("AdjustTokenPrivileges failed\n");
break;
}
/* Finally shut down the system */
if(!ExitWindowsEx(EWX_REBOOT, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER | SHTDN_REASON_FLAG_PLANNED))
{
DPRINT("ExitWindowsEx failed\n");
break;
}
}
}
}
}
break; break;
}
} }
break; break;
} }

2
reactos/dll/win32/devmgr/precomp.h
View file

@ -8,8 +8,6 @@
#include <regstr.h> #include <regstr.h>
#include <setupapi.h> #include <setupapi.h>
#include <cfgmgr32.h> #include <cfgmgr32.h>
#undef WINVER
#define WINVER _WIN32_WINNT_VISTA
#include <newdev.h> #include <newdev.h>
#include <dll/devmgr/devmgr.h> #include <dll/devmgr/devmgr.h>