mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
470 lines
12 KiB
470 lines
12 KiB
/****************************** Module Header ******************************\
|
|
* Module Name: shutdown.c
|
|
*
|
|
* Copyright (c) 1992, Microsoft Corporation
|
|
*
|
|
* Handles shutdown dialog.
|
|
*
|
|
* History:
|
|
* 2-25-92 JohanneC Created - extracted from old pmdlgs.c with security
|
|
* added.
|
|
\***************************************************************************/
|
|
|
|
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include "progman.h"
|
|
|
|
#define LOGOFF_SETTING L"Logoff Setting"
|
|
|
|
INT_PTR WINAPI
|
|
ShutdownDlgProc(
|
|
HWND hDlg,
|
|
UINT message,
|
|
WPARAM wParam,
|
|
LPARAM lParam
|
|
);
|
|
|
|
VOID
|
|
CentreWindow(
|
|
HWND hwnd
|
|
);
|
|
|
|
|
|
/***************************************************************************\
|
|
* ShutdownDialog
|
|
*
|
|
* Creates the shutdown dialog
|
|
*
|
|
*
|
|
* History:
|
|
* 2-28-92 Johannec Created
|
|
*
|
|
\***************************************************************************/
|
|
BOOL
|
|
ShutdownDialog(
|
|
HANDLE hInst,
|
|
HWND hwnd
|
|
)
|
|
{
|
|
INT_PTR nResult;
|
|
BOOLEAN WasEnabled;
|
|
NTSTATUS Status;
|
|
TCHAR szMessage[MAXMESSAGELEN];
|
|
TCHAR szTitle[MAXMESSAGELEN];
|
|
|
|
Status = RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE,
|
|
(BOOLEAN)TRUE,
|
|
(BOOLEAN)FALSE,
|
|
&WasEnabled);
|
|
|
|
if (!NT_SUCCESS(Status)) {
|
|
|
|
//
|
|
// We don't have permission to shutdown
|
|
//
|
|
|
|
LoadString(hInst, IDS_NO_PERMISSION_SHUTDOWN, szMessage, CharSizeOf(szMessage));
|
|
LoadString(hInst, IDS_SHUTDOWN_MESSAGE, szTitle, CharSizeOf(szTitle));
|
|
|
|
nResult = MessageBox(hwnd, szMessage, szTitle, MB_OK | MB_ICONSTOP);
|
|
|
|
return(FALSE);
|
|
}
|
|
|
|
//
|
|
// Put up the shutdown dialog
|
|
//
|
|
|
|
nResult = DialogBox(hInst, (LPTSTR) MAKEINTRESOURCE(IDD_SHUTDOWN_QUERY), hwnd, ShutdownDlgProc);
|
|
|
|
//
|
|
// Restore the shutdown privilege state
|
|
//
|
|
|
|
if (!WasEnabled) {
|
|
|
|
Status = RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE,
|
|
(BOOLEAN)WasEnabled,
|
|
(BOOLEAN)FALSE,
|
|
&WasEnabled);
|
|
}
|
|
|
|
return(FALSE);
|
|
}
|
|
|
|
|
|
/***************************************************************************\
|
|
* FUNCTION: ShutdownDlgProc
|
|
*
|
|
* PURPOSE: Processes messages for shutdown confirmation dialog
|
|
*
|
|
* RETURNS: if user decided not to shutdown or reboot.
|
|
*
|
|
* HISTORY:
|
|
*
|
|
* 05-17-92 Davidc Created.
|
|
*
|
|
\***************************************************************************/
|
|
|
|
INT_PTR WINAPI
|
|
ShutdownDlgProc(
|
|
HWND hDlg,
|
|
UINT message,
|
|
WPARAM wParam,
|
|
LPARAM lParam
|
|
)
|
|
{
|
|
UINT uiOptions;
|
|
static DWORD dwShutdown = 1;
|
|
static HKEY hkeyShutdown = NULL;
|
|
|
|
switch (message) {
|
|
|
|
case WM_INITDIALOG:
|
|
{
|
|
DWORD dwType;
|
|
DWORD cbData;
|
|
DWORD dwDisposition;
|
|
BOOL bPowerdown;
|
|
|
|
bPowerdown = GetProfileInt(TEXT("Winlogon"), TEXT("PowerdownAfterShutdown"), 0);
|
|
if (!bPowerdown) {
|
|
ShowWindow(GetDlgItem(hDlg, IDD_POWEROFF), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDOK2), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDCANCEL2), SW_HIDE);
|
|
SendMessage(hDlg, DM_SETDEFID, IDOK, 0);
|
|
}
|
|
else {
|
|
ShowWindow(GetDlgItem(hDlg, IDOK), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDCANCEL), SW_HIDE);
|
|
}
|
|
|
|
//
|
|
// Check the button that was the user's last shutdown selection.
|
|
//
|
|
if (RegCreateKeyEx(HKEY_CURRENT_USER, SHUTDOWN_SETTING_KEY, 0, 0, 0,
|
|
KEY_READ | KEY_WRITE | DELETE,
|
|
NULL, &hkeyShutdown, &dwDisposition) == ERROR_SUCCESS) {
|
|
cbData = sizeof(dwShutdown);
|
|
RegQueryValueEx(hkeyShutdown, SHUTDOWN_SETTING, 0, &dwType, (LPBYTE)&dwShutdown, &cbData);
|
|
}
|
|
switch(dwShutdown) {
|
|
case DLGSEL_SHUTDOWN_AND_RESTART:
|
|
CheckDlgButton(hDlg, IDD_RESTART, 1);
|
|
break;
|
|
case DLGSEL_SHUTDOWN_AND_POWEROFF:
|
|
if (bPowerdown) {
|
|
CheckDlgButton(hDlg, IDD_POWEROFF, 1);
|
|
break;
|
|
}
|
|
//
|
|
// Fall thru.
|
|
// Default to shutdown.
|
|
//
|
|
default:
|
|
CheckDlgButton(hDlg, IDD_SHUTDOWN, 1);
|
|
break;
|
|
}
|
|
|
|
//
|
|
// Position ourselves
|
|
//
|
|
CentreWindow(hDlg);
|
|
|
|
return(TRUE);
|
|
}
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam)) {
|
|
|
|
case IDOK:
|
|
case IDOK2:
|
|
if (IsDlgButtonChecked(hDlg, IDD_SHUTDOWN)) {
|
|
uiOptions = EWX_SHUTDOWN;
|
|
dwShutdown = DLGSEL_SHUTDOWN;
|
|
}
|
|
else if (IsDlgButtonChecked(hDlg, IDD_RESTART)) {
|
|
uiOptions = EWX_SHUTDOWN | EWX_REBOOT;
|
|
dwShutdown = DLGSEL_SHUTDOWN_AND_RESTART;
|
|
}
|
|
else {
|
|
uiOptions = EWX_SHUTDOWN | EWX_POWEROFF;
|
|
dwShutdown = DLGSEL_SHUTDOWN_AND_POWEROFF;
|
|
}
|
|
|
|
if (hkeyShutdown) {
|
|
RegSetValueEx(hkeyShutdown, SHUTDOWN_SETTING, 0, REG_DWORD, (LPBYTE)&dwShutdown, sizeof(dwShutdown));
|
|
RegCloseKey(hkeyShutdown);
|
|
hkeyShutdown = NULL;
|
|
}
|
|
|
|
ExitWindowsEx(uiOptions, (DWORD)-1);
|
|
|
|
case IDCANCEL2:
|
|
case IDCANCEL:
|
|
if (hkeyShutdown) {
|
|
RegCloseKey(hkeyShutdown);
|
|
hkeyShutdown = NULL;
|
|
}
|
|
EndDialog(hDlg, 0);
|
|
return(TRUE);
|
|
}
|
|
break;
|
|
}
|
|
|
|
// We didn't process the message
|
|
return(FALSE);
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* FUNCTION: NewLogoffDlgProc
|
|
*
|
|
* PURPOSE: Processes messages for logoff/shutdown confirmation dialog
|
|
*
|
|
* RETURNS: if user decided not to logoff, shutdown or reboot.
|
|
*
|
|
* HISTORY:
|
|
*
|
|
* 10-05-93 Johannec Created.
|
|
*
|
|
\***************************************************************************/
|
|
|
|
INT_PTR APIENTRY
|
|
NewLogoffDlgProc(
|
|
HWND hDlg,
|
|
UINT message,
|
|
WPARAM wParam,
|
|
LPARAM lParam
|
|
)
|
|
{
|
|
UINT uiOptions;
|
|
BOOLEAN WasEnabled = FALSE;
|
|
NTSTATUS Status;
|
|
static DWORD dwShutdown = 0;
|
|
static HKEY hkeyShutdown = NULL;
|
|
|
|
switch (message) {
|
|
|
|
case WM_INITDIALOG:
|
|
{
|
|
DWORD dwType;
|
|
BOOL bPowerdown;
|
|
DWORD cbData;
|
|
DWORD dwDisposition;
|
|
|
|
bPowerdown = GetProfileInt(TEXT("Winlogon"), TEXT("PowerdownAfterShutdown"), 0);
|
|
if (!bPowerdown) {
|
|
ShowWindow(GetDlgItem(hDlg, IDD_POWEROFF), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDOK2), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDCANCEL2), SW_HIDE);
|
|
SendMessage(hDlg, DM_SETDEFID, IDOK, 0);
|
|
}
|
|
else {
|
|
ShowWindow(GetDlgItem(hDlg, IDOK), SW_HIDE);
|
|
ShowWindow(GetDlgItem(hDlg, IDCANCEL), SW_HIDE);
|
|
}
|
|
|
|
//
|
|
// Initial setting is User's last selection
|
|
//
|
|
if (RegCreateKeyEx(HKEY_CURRENT_USER, SHUTDOWN_SETTING_KEY, 0, 0, 0,
|
|
KEY_READ | KEY_WRITE | DELETE,
|
|
NULL, &hkeyShutdown, &dwDisposition) == ERROR_SUCCESS) {
|
|
cbData = sizeof(dwShutdown);
|
|
RegQueryValueEx(hkeyShutdown, LOGOFF_SETTING, 0, &dwType, (LPBYTE)&dwShutdown, &cbData);
|
|
}
|
|
switch(dwShutdown) {
|
|
case DLGSEL_SHUTDOWN:
|
|
CheckDlgButton(hDlg, IDD_SHUTDOWN, 1);
|
|
break;
|
|
case DLGSEL_SHUTDOWN_AND_RESTART:
|
|
CheckDlgButton(hDlg, IDD_RESTART, 1);
|
|
break;
|
|
case DLGSEL_SHUTDOWN_AND_POWEROFF:
|
|
if (bPowerdown) {
|
|
CheckDlgButton(hDlg, IDD_POWEROFF, 1);
|
|
}
|
|
else {
|
|
CheckDlgButton(hDlg, IDD_SHUTDOWN, 1);
|
|
}
|
|
break;
|
|
default:
|
|
CheckDlgButton(hDlg, IDD_LOGOFF, 1);
|
|
break;
|
|
}
|
|
|
|
//
|
|
// Make sure the user has the privileges to shutdown the computer.
|
|
//
|
|
|
|
Status = RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE,
|
|
(BOOLEAN)TRUE,
|
|
(BOOLEAN)FALSE,
|
|
&WasEnabled);
|
|
|
|
if (!NT_SUCCESS(Status)) {
|
|
|
|
//
|
|
// We don't have permission to shutdown
|
|
//
|
|
EnableWindow(GetDlgItem(hDlg, IDD_SHUTDOWN), 0);
|
|
EnableWindow(GetDlgItem(hDlg, IDD_RESTART), 0);
|
|
EnableWindow(GetDlgItem(hDlg, IDD_POWEROFF), 0);
|
|
}
|
|
|
|
//
|
|
// Position ourselves
|
|
//
|
|
CentreWindow(hDlg);
|
|
|
|
return(TRUE);
|
|
}
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam)) {
|
|
case IDOK:
|
|
case IDOK2:
|
|
if (IsDlgButtonChecked(hDlg, IDD_LOGOFF)) {
|
|
uiOptions = EWX_LOGOFF;
|
|
dwShutdown = DLGSEL_LOGOFF;
|
|
}
|
|
else if (IsDlgButtonChecked(hDlg, IDD_SHUTDOWN)) {
|
|
uiOptions = EWX_SHUTDOWN;
|
|
dwShutdown = DLGSEL_SHUTDOWN;
|
|
}
|
|
else if (IsDlgButtonChecked(hDlg, IDD_RESTART)) {
|
|
uiOptions = EWX_SHUTDOWN | EWX_REBOOT;
|
|
dwShutdown = DLGSEL_SHUTDOWN_AND_RESTART;
|
|
}
|
|
else {
|
|
uiOptions = EWX_SHUTDOWN | EWX_POWEROFF;
|
|
dwShutdown = DLGSEL_SHUTDOWN_AND_POWEROFF;
|
|
}
|
|
|
|
//
|
|
// Save user's shutdown selection.
|
|
//
|
|
if (hkeyShutdown) {
|
|
RegSetValueEx(hkeyShutdown, LOGOFF_SETTING, 0, REG_DWORD, (LPBYTE)&dwShutdown, sizeof(dwShutdown));
|
|
RegCloseKey(hkeyShutdown);
|
|
hkeyShutdown = NULL;
|
|
}
|
|
|
|
//
|
|
// Make sure the user has the privileges to shutdown the computer.
|
|
//
|
|
|
|
ExitWindowsEx(uiOptions, (DWORD)-1);
|
|
//
|
|
// Restore the shutdown privilege state
|
|
//
|
|
|
|
if (!WasEnabled) {
|
|
|
|
Status = RtlAdjustPrivilege(SE_SHUTDOWN_PRIVILEGE,
|
|
(BOOLEAN)WasEnabled,
|
|
(BOOLEAN)FALSE,
|
|
&WasEnabled);
|
|
}
|
|
|
|
|
|
case IDCANCEL:
|
|
case IDCANCEL2:
|
|
if (hkeyShutdown) {
|
|
RegCloseKey(hkeyShutdown);
|
|
hkeyShutdown = NULL;
|
|
}
|
|
EndDialog(hDlg, 0);
|
|
return(TRUE);
|
|
}
|
|
break;
|
|
}
|
|
|
|
// We didn't process the message
|
|
return(FALSE);
|
|
}
|
|
|
|
/***************************************************************************\
|
|
* CentreWindow
|
|
*
|
|
* Purpose : Positions a window so that it is centred in its parent
|
|
*
|
|
* !!! WARNING this code is duplicated in winlogon\winutil.c
|
|
*
|
|
* History:
|
|
* 12-09-91 Davidc Created.
|
|
\***************************************************************************/
|
|
VOID
|
|
CentreWindow(
|
|
HWND hwnd
|
|
)
|
|
{
|
|
RECT rect;
|
|
RECT rectParent;
|
|
HWND hwndParent;
|
|
LONG dx, dy;
|
|
LONG dxParent, dyParent;
|
|
LONG Style;
|
|
|
|
// Get window rect
|
|
GetWindowRect(hwnd, &rect);
|
|
|
|
dx = rect.right - rect.left;
|
|
dy = rect.bottom - rect.top;
|
|
|
|
// Get parent rect
|
|
Style = GetWindowLong(hwnd, GWL_STYLE);
|
|
if ((Style & WS_CHILD) == 0) {
|
|
|
|
// Return the desktop windows size (size of main screen)
|
|
dxParent = GetSystemMetrics(SM_CXSCREEN);
|
|
dyParent = GetSystemMetrics(SM_CYSCREEN);
|
|
} else {
|
|
hwndParent = GetParent(hwnd);
|
|
if (hwndParent == NULL) {
|
|
hwndParent = GetDesktopWindow();
|
|
}
|
|
GetWindowRect(hwndParent, &rectParent);
|
|
|
|
dxParent = rectParent.right - rectParent.left;
|
|
dyParent = rectParent.bottom - rectParent.top;
|
|
}
|
|
|
|
// Centre the child in the parent
|
|
rect.left = (dxParent - dx) / 2;
|
|
rect.top = (dyParent - dy) / 3;
|
|
|
|
// Move the child into position
|
|
SetWindowPos(hwnd, NULL, rect.left, rect.top, 0, 0, SWP_NOSIZE | SWP_NOZORDER);
|
|
|
|
SetForegroundWindow(hwnd);
|
|
}
|
|
|
|
INT_PTR APIENTRY ExitDlgProc(HWND hwnd, UINT uiMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
switch (uiMsg) {
|
|
case WM_INITDIALOG:
|
|
CentreWindow(hwnd);
|
|
break;
|
|
|
|
case WM_COMMAND:
|
|
switch (LOWORD(wParam)) {
|
|
|
|
case IDOK:
|
|
|
|
// fall through...
|
|
|
|
case IDCANCEL:
|
|
EndDialog(hwnd, LOWORD(wParam) == IDOK);
|
|
break;
|
|
|
|
default:
|
|
return(FALSE);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
return(FALSE);
|
|
}
|
|
UNREFERENCED_PARAMETER(lParam);
|
|
return(TRUE);
|
|
}
|