419 lines
11 KiB
C
419 lines
11 KiB
C
/*++
|
||
|
||
Copyright (c) 1994 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
shutdown.C
|
||
|
||
Abstract:
|
||
|
||
functions used to enable/disable the shutdown button in the logon screen
|
||
on the current system.
|
||
|
||
Author:
|
||
|
||
Bob Watson (a-robw)
|
||
|
||
Revision History:
|
||
|
||
23 Dec 94
|
||
|
||
--*/
|
||
#include <windows.h>
|
||
#include <stdlib.h>
|
||
#include <stdio.h>
|
||
#include <c2dll.h>
|
||
#include <c2inc.h>
|
||
#include <c2utils.h>
|
||
#include "c2funcs.h"
|
||
#include "c2funres.h"
|
||
|
||
// local constants
|
||
|
||
// define action codes here. They are only meaningful in the
|
||
// context of this module.
|
||
|
||
#define AC_SHUTDOWN_BTN_NOCHANGE 0
|
||
#define AC_SHUTDOWN_BTN_UPDATE 1
|
||
|
||
#define SECURE C2DLL_SECURE
|
||
|
||
static
|
||
BOOL
|
||
SetShutDownButton (
|
||
DWORD dwNewValue
|
||
)
|
||
{
|
||
HKEY hKeyWinlogon = NULL;
|
||
LONG lStatus = ERROR_SUCCESS;
|
||
BOOL bReturn = FALSE;
|
||
UINT nNewData;
|
||
LPCTSTR szNewString;
|
||
|
||
SET_WAIT_CURSOR;
|
||
|
||
lStatus = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
||
GetStringResource (GetDllInstance(), IDS_WINLOGON_KEY),
|
||
0L,
|
||
KEY_SET_VALUE,
|
||
&hKeyWinlogon);
|
||
|
||
if (lStatus == ERROR_SUCCESS) {
|
||
// key opened OK so set value
|
||
nNewData = (dwNewValue ? IDS_1 : IDS_0);
|
||
szNewString = GetStringResource (GetDllInstance(), nNewData);
|
||
lStatus = RegSetValueEx (
|
||
hKeyWinlogon,
|
||
GetStringResource (GetDllInstance(), IDS_SHUTDOWN_BTN_VALUE),
|
||
0L,
|
||
REG_SZ,
|
||
(CONST LPBYTE)szNewString,
|
||
(lstrlen (szNewString) + 1) * sizeof(TCHAR));
|
||
|
||
if (lStatus == ERROR_SUCCESS) {
|
||
bReturn = TRUE;
|
||
} else {
|
||
bReturn = FALSE;
|
||
}
|
||
RegCloseKey (hKeyWinlogon);
|
||
} else {
|
||
bReturn = FALSE;
|
||
SetLastError (ERROR_BADKEY);
|
||
}
|
||
|
||
SET_ARROW_CURSOR;
|
||
|
||
return bReturn;
|
||
}
|
||
|
||
static
|
||
BOOL
|
||
GetShutDownButton (
|
||
)
|
||
{
|
||
HKEY hKeyWinlogon = NULL;
|
||
LONG lStatus = ERROR_SUCCESS;
|
||
DWORD dwType = 0;
|
||
BOOL bReturn = FALSE;
|
||
TCHAR szValue[MAX_PATH];
|
||
DWORD dwValueSize = MAX_PATH * sizeof(TCHAR);
|
||
|
||
SET_WAIT_CURSOR;
|
||
|
||
lStatus = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
||
GetStringResource (GetDllInstance(), IDS_WINLOGON_KEY),
|
||
0L,
|
||
KEY_READ,
|
||
&hKeyWinlogon);
|
||
|
||
if (lStatus == ERROR_SUCCESS) {
|
||
// key opened OK so check value
|
||
lStatus = RegQueryValueEx (
|
||
hKeyWinlogon,
|
||
(LPTSTR)GetStringResource (GetDllInstance(), IDS_SHUTDOWN_BTN_VALUE),
|
||
(LPDWORD)NULL,
|
||
&dwType,
|
||
(LPBYTE)szValue,
|
||
&dwValueSize);
|
||
|
||
if (lStatus == ERROR_SUCCESS) {
|
||
// value read successfully so check it out
|
||
if (dwType == REG_SZ) {
|
||
// if the button is visible, then this value is 1
|
||
if (lstrcmp(szValue, GetStringResource(GetDllInstance(), IDS_1)) == 0) {
|
||
bReturn = TRUE;
|
||
} else {
|
||
bReturn = FALSE;
|
||
}
|
||
SetLastError (ERROR_SUCCESS);
|
||
} else {
|
||
// wrong data type returned
|
||
bReturn = FALSE;
|
||
SetLastError (ERROR_CANTREAD);
|
||
}
|
||
} else {
|
||
// no value present
|
||
bReturn = FALSE;
|
||
SetLastError (ERROR_CANTREAD);
|
||
}
|
||
RegCloseKey (hKeyWinlogon);
|
||
} else {
|
||
bReturn = FALSE;
|
||
SetLastError (ERROR_BADKEY);
|
||
}
|
||
|
||
SET_ARROW_CURSOR;
|
||
|
||
return bReturn;
|
||
}
|
||
|
||
BOOL CALLBACK
|
||
C2ShutDownButtonDlgProc(
|
||
IN HWND hDlg, // window handle of the dialog box
|
||
IN UINT message, // type of message
|
||
IN WPARAM wParam,
|
||
IN LPARAM lParam
|
||
)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Window procedure for Last User Display dialog box
|
||
|
||
Arguments:
|
||
|
||
Standard DlgProc arguments
|
||
|
||
ReturnValue:
|
||
|
||
TRUE the message was handled by this routine
|
||
FALSE DefDialogProc should handle the message
|
||
|
||
--*/
|
||
{
|
||
static LPDWORD lpdwNewValue;
|
||
DWORD dwLogSetting = 0;
|
||
|
||
switch (message) {
|
||
case WM_INITDIALOG:
|
||
// save the pointer to the new value
|
||
lpdwNewValue = (LPDWORD)lParam;
|
||
|
||
// get Shutdown button state for check box
|
||
CheckDlgButton (hDlg, IDC_HIDE_SHUTDOWN_BUTTON,
|
||
(GetShutDownButton() ? UNCHECKED : CHECKED));
|
||
|
||
SetFocus (GetDlgItem (hDlg, IDOK)); // set focus to OK Button
|
||
return FALSE; // we don't want Windows to set the focus
|
||
|
||
case WM_COMMAND:
|
||
switch (LOWORD(wParam)){
|
||
case IDOK:
|
||
if (HIWORD(wParam) == BN_CLICKED) {
|
||
// exit and return button that caused exit
|
||
if (IsDlgButtonChecked (hDlg, IDC_HIDE_SHUTDOWN_BUTTON) == CHECKED) {
|
||
*lpdwNewValue = FALSE; // hide the button
|
||
EndDialog (hDlg, (int)LOWORD(wParam));
|
||
} else {
|
||
*lpdwNewValue = TRUE; // show the button
|
||
EndDialog (hDlg, (int)LOWORD(wParam));
|
||
}
|
||
return TRUE;
|
||
} else {
|
||
return FALSE;
|
||
}
|
||
|
||
case IDCANCEL:
|
||
if (HIWORD(wParam) == BN_CLICKED) {
|
||
// exit and return button that caused exit
|
||
*lpdwNewValue = 0;
|
||
EndDialog (hDlg, (int)LOWORD(wParam));
|
||
return TRUE;
|
||
} else {
|
||
return FALSE;
|
||
}
|
||
|
||
case IDC_C2:
|
||
if (HIWORD(wParam) == BN_CLICKED) {
|
||
CheckDlgButton (hDlg, IDC_HIDE_SHUTDOWN_BUTTON, CHECKED);
|
||
return TRUE;
|
||
} else {
|
||
return FALSE;
|
||
}
|
||
|
||
case IDC_HELP:
|
||
PostMessage (GetParent(hDlg), UM_SHOW_CONTEXT_HELP, 0, 0);
|
||
return TRUE;
|
||
|
||
default:
|
||
return FALSE;
|
||
}
|
||
|
||
default:
|
||
return (FALSE); // Didn't process the message
|
||
}
|
||
}
|
||
|
||
LONG
|
||
C2QueryShutDownButton (
|
||
IN LPARAM lParam
|
||
)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Function called to find out if the OS/2 subsystem is installed
|
||
on the system. For C2 compliance, OS/2 must not be
|
||
allowed on the system.
|
||
|
||
Arguments:
|
||
|
||
Pointer to the Dll data block passed as an LPARAM.
|
||
|
||
ReturnValue:
|
||
|
||
ERROR_SUCCESS if the function succeeds otherwise a
|
||
WIN32 error is returned if an error occurs
|
||
|
||
--*/
|
||
{
|
||
PC2DLL_DATA pC2Data;
|
||
DWORD dwLogSetting = 0;
|
||
UINT nMsgString;
|
||
|
||
if (lParam != 0) {
|
||
pC2Data = (PC2DLL_DATA)lParam;
|
||
pC2Data->lC2Compliance = SECURE; // assume true for now
|
||
if (!GetShutDownButton()) {
|
||
// shutdown button is not displayed on the logon dialog
|
||
pC2Data->lC2Compliance = SECURE;
|
||
nMsgString = IDS_SHUTDOWN_BTN_HIDDEN;
|
||
} else {
|
||
// shutdown button is displayed on the logon dialog
|
||
pC2Data->lC2Compliance = C2DLL_NOT_SECURE;
|
||
nMsgString = IDS_SHUTDOWN_BTN_DISPLAYED;
|
||
}
|
||
lstrcpy (pC2Data->szStatusName,
|
||
GetStringResource (GetDllInstance(), nMsgString));
|
||
} else {
|
||
return ERROR_BAD_ARGUMENTS;
|
||
}
|
||
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
LONG
|
||
C2SetShutDownButton (
|
||
IN LPARAM lParam
|
||
)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Function called to change the current state of this configuration
|
||
item based on an action code passed in the DLL data block. If
|
||
this function successfully sets the state of the configuration
|
||
item, then the C2 Compliance flag and the Status string to reflect
|
||
the new value of the configuration item.
|
||
|
||
Arguments:
|
||
|
||
Pointer to the Dll data block passed as an LPARAM.
|
||
|
||
ReturnValue:
|
||
|
||
ERROR_SUCCESS if the function succeeds otherwise a
|
||
WIN32 error is returned if an error occurs
|
||
|
||
--*/
|
||
{
|
||
PC2DLL_DATA pC2Data;
|
||
UINT nMsgString;
|
||
|
||
if (lParam != 0) {
|
||
pC2Data = (PC2DLL_DATA)lParam;
|
||
// action valie = the new value of the wrap setting
|
||
if (pC2Data->lActionCode != AC_SHUTDOWN_BTN_NOCHANGE) {
|
||
nMsgString = 0;
|
||
if (pC2Data->lActionCode == AC_SHUTDOWN_BTN_UPDATE) {
|
||
if (pC2Data->lActionValue) {
|
||
if (SetShutDownButton (TRUE)) {
|
||
pC2Data->lC2Compliance = C2DLL_NOT_SECURE;
|
||
nMsgString = IDS_SHUTDOWN_BTN_DISPLAYED;
|
||
} else {
|
||
DisplayDllMessageBox (
|
||
pC2Data->hWnd,
|
||
IDS_SHUTDOWN_ERROR_NOT_SET,
|
||
IDS_SHUTDOWN_CAPTION,
|
||
MBOK_EXCLAIM);
|
||
}
|
||
} else {
|
||
if (SetShutDownButton (FALSE)) {
|
||
pC2Data->lC2Compliance = SECURE;
|
||
nMsgString = IDS_SHUTDOWN_BTN_HIDDEN;
|
||
} else {
|
||
DisplayDllMessageBox (
|
||
pC2Data->hWnd,
|
||
IDS_SHUTDOWN_ERROR_NOT_SET,
|
||
IDS_SHUTDOWN_CAPTION,
|
||
MBOK_EXCLAIM);
|
||
}
|
||
}
|
||
}
|
||
if (nMsgString != 0) {
|
||
// update status string if set was successful
|
||
lstrcpy (pC2Data->szStatusName,
|
||
GetStringResource (GetDllInstance(), nMsgString));
|
||
}
|
||
}
|
||
// update action values
|
||
pC2Data->lActionCode = 0;
|
||
pC2Data->lActionValue = 0;
|
||
} else {
|
||
return ERROR_BAD_ARGUMENTS;
|
||
}
|
||
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
LONG
|
||
C2DisplayShutDownButton (
|
||
IN LPARAM lParam
|
||
)
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Function called to display more information on the configuration
|
||
item and provide the user with the option to change the current
|
||
setting (if appropriate). If the User "OK's" out of the UI,
|
||
then the action code field in the DLL data block is set to the
|
||
appropriate (and configuration item-specific) action code so the
|
||
"Set" function can be called to perform the desired action. If
|
||
the user Cancels out of the UI, then the Action code field is
|
||
set to 0 (no action) and no action is performed.
|
||
|
||
Arguments:
|
||
|
||
Pointer to the Dll data block passed as an LPARAM.
|
||
|
||
ReturnValue:
|
||
|
||
ERROR_SUCCESS if the function succeeds otherwise a
|
||
WIN32 error is returned if an error occurs
|
||
|
||
--*/
|
||
{
|
||
PC2DLL_DATA pC2Data;
|
||
DWORD dwNewValue;
|
||
|
||
if (lParam != 0) {
|
||
pC2Data = (PC2DLL_DATA)lParam;
|
||
if (DialogBoxParam (
|
||
GetDllInstance(),
|
||
MAKEINTRESOURCE (IDD_SHUTDOWN_BUTTON),
|
||
pC2Data->hWnd,
|
||
C2ShutDownButtonDlgProc,
|
||
(LPARAM)&dwNewValue) == IDOK) {
|
||
pC2Data->lActionValue = dwNewValue;
|
||
pC2Data->lActionCode = AC_SHUTDOWN_BTN_UPDATE;
|
||
} else {
|
||
// no action
|
||
pC2Data->lActionCode = AC_SHUTDOWN_BTN_NOCHANGE;
|
||
}
|
||
} else {
|
||
return ERROR_BAD_ARGUMENTS;
|
||
}
|
||
return ERROR_SUCCESS;
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|