415 lines
11 KiB
C
415 lines
11 KiB
C
|
/*++
|
|||
|
|
|
|||
|
|
Copyright (c) 1994 Microsoft Corporation
|
|||
|
|
|
|||
|
|
Module Name:
|
|||
|
|
|
|||
|
|
drvsnprt.C
|
|||
|
|
|
|||
|
|
Abstract:
|
|||
|
|
|
|||
|
|
functions used to set the Drive Letter and Printer Assignment
|
|||
|
|
on the current system.
|
|||
|
|
|
|||
|
|
Author:
|
|||
|
|
|
|||
|
|
Bob Watson (a-robw)
|
|||
|
|
|
|||
|
|
Revision History:
|
|||
|
|
|
|||
|
|
3 Jul 95
|
|||
|
|
|
|||
|
|
--*/
|
|||
|
|
#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_DRIVES_AND_PRINTER_NOCHANGE 0
|
|||
|
|
#define AC_DRIVES_AND_PRINTER_SET_ADMIN 1
|
|||
|
|
#define AC_DRIVES_AND_PRINTER_SET_ANYONE 2
|
|||
|
|
|
|||
|
|
#define SECURE C2DLL_SECURE
|
|||
|
|
|
|||
|
|
|
|||
|
|
static
|
|||
|
|
BOOL
|
|||
|
|
SetDrivesAndPrinterSetting (
|
|||
|
|
DWORD dwNewValue
|
|||
|
|
)
|
|||
|
|
{
|
|||
|
|
HKEY hKeySM = NULL;
|
|||
|
|
LONG lStatus = ERROR_SUCCESS;
|
|||
|
|
BOOL bReturn = FALSE;
|
|||
|
|
|
|||
|
|
SET_WAIT_CURSOR;
|
|||
|
|
|
|||
|
|
lStatus = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
|||
|
|
GetStringResource (GetDllInstance(), IDS_SESSION_MANAGER_KEY),
|
|||
|
|
0L,
|
|||
|
|
KEY_SET_VALUE,
|
|||
|
|
&hKeySM);
|
|||
|
|
|
|||
|
|
if (lStatus == ERROR_SUCCESS) {
|
|||
|
|
// key opened OK so set value
|
|||
|
|
lStatus = RegSetValueEx (
|
|||
|
|
hKeySM,
|
|||
|
|
GetStringResource (GetDllInstance(), IDS_PROTECTION_MODE_VALUE),
|
|||
|
|
0L,
|
|||
|
|
REG_DWORD,
|
|||
|
|
(CONST LPBYTE)&dwNewValue,
|
|||
|
|
sizeof(DWORD));
|
|||
|
|
|
|||
|
|
if (lStatus == ERROR_SUCCESS) {
|
|||
|
|
bReturn = TRUE;
|
|||
|
|
} else {
|
|||
|
|
bReturn = FALSE;
|
|||
|
|
}
|
|||
|
|
RegCloseKey (hKeySM);
|
|||
|
|
} else {
|
|||
|
|
bReturn = FALSE;
|
|||
|
|
SetLastError (ERROR_BADKEY);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
SET_ARROW_CURSOR;
|
|||
|
|
|
|||
|
|
return bReturn;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
static
|
|||
|
|
BOOL
|
|||
|
|
GetDrivesAndPrinterSetting (
|
|||
|
|
)
|
|||
|
|
{
|
|||
|
|
HKEY hKeySM = NULL;
|
|||
|
|
LONG lStatus = ERROR_SUCCESS;
|
|||
|
|
DWORD dwType = 0;
|
|||
|
|
DWORD dwValue = 0;
|
|||
|
|
DWORD dwValueSize = sizeof(DWORD);
|
|||
|
|
BOOL bReturn = FALSE;
|
|||
|
|
|
|||
|
|
SET_WAIT_CURSOR;
|
|||
|
|
|
|||
|
|
lStatus = RegOpenKeyEx (HKEY_LOCAL_MACHINE,
|
|||
|
|
GetStringResource (GetDllInstance(), IDS_SESSION_MANAGER_KEY),
|
|||
|
|
0L,
|
|||
|
|
KEY_READ,
|
|||
|
|
&hKeySM);
|
|||
|
|
|
|||
|
|
if (lStatus == ERROR_SUCCESS) {
|
|||
|
|
// key opened OK so check value
|
|||
|
|
lStatus = RegQueryValueEx (
|
|||
|
|
hKeySM,
|
|||
|
|
(LPTSTR)GetStringResource (GetDllInstance(), IDS_PROTECTION_MODE_VALUE),
|
|||
|
|
(LPDWORD)NULL,
|
|||
|
|
&dwType,
|
|||
|
|
(LPBYTE)&dwValue,
|
|||
|
|
&dwValueSize);
|
|||
|
|
|
|||
|
|
if (lStatus == ERROR_SUCCESS) {
|
|||
|
|
// value read successfully so check it out
|
|||
|
|
if (dwType == REG_DWORD) {
|
|||
|
|
// check value. The "C2" value is 1, any
|
|||
|
|
// other value is NOT C2
|
|||
|
|
if (dwValue == 1) {
|
|||
|
|
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 (hKeySM);
|
|||
|
|
} else {
|
|||
|
|
bReturn = FALSE;
|
|||
|
|
SetLastError (ERROR_BADKEY);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
SET_ARROW_CURSOR;
|
|||
|
|
|
|||
|
|
return bReturn;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
BOOL CALLBACK
|
|||
|
|
C2DrivesAndPrinterDlgProc(
|
|||
|
|
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 Drives and Printer assignment 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 Audit failure settings
|
|||
|
|
|
|||
|
|
CheckDlgButton (hDlg, IDC_ADMIN_ASSIGN_DRIVES,
|
|||
|
|
(GetDrivesAndPrinterSetting() ? CHECKED : UNCHECKED));
|
|||
|
|
|
|||
|
|
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_ADMIN_ASSIGN_DRIVES) == CHECKED) {
|
|||
|
|
*lpdwNewValue = TRUE;
|
|||
|
|
EndDialog (hDlg, (int)LOWORD(wParam));
|
|||
|
|
} else {
|
|||
|
|
*lpdwNewValue = FALSE;
|
|||
|
|
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_ADMIN_ASSIGN_DRIVES, 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
|
|||
|
|
C2QueryAssignDrivesAndPrinter (
|
|||
|
|
IN LPARAM lParam
|
|||
|
|
)
|
|||
|
|
/*++
|
|||
|
|
|
|||
|
|
Routine Description:
|
|||
|
|
|
|||
|
|
Function called to find out if only the Admin can assign drives and printer
|
|||
|
|
on the system. For C2 compliance,
|
|||
|
|
|
|||
|
|
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 (GetDrivesAndPrinterSetting()) {
|
|||
|
|
pC2Data->lC2Compliance = SECURE;
|
|||
|
|
nMsgString = IDS_ONLY_ADMINS_ASSIGN_DRIVE;
|
|||
|
|
} else {
|
|||
|
|
pC2Data->lC2Compliance = C2DLL_NOT_SECURE;
|
|||
|
|
nMsgString = IDS_ANYONE_CAN_ASSIGN_DRIVE;
|
|||
|
|
}
|
|||
|
|
lstrcpy (pC2Data->szStatusName,
|
|||
|
|
GetStringResource (GetDllInstance(), nMsgString));
|
|||
|
|
} else {
|
|||
|
|
return ERROR_BAD_ARGUMENTS;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return ERROR_SUCCESS;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
LONG
|
|||
|
|
C2SetAssignDrivesAndPrinter (
|
|||
|
|
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_DRIVES_AND_PRINTER_NOCHANGE) {
|
|||
|
|
nMsgString = 0;
|
|||
|
|
if (pC2Data->lActionCode == AC_DRIVES_AND_PRINTER_SET_ADMIN) {
|
|||
|
|
if (SetDrivesAndPrinterSetting (1)) {
|
|||
|
|
pC2Data->lC2Compliance = SECURE;
|
|||
|
|
nMsgString = IDS_ONLY_ADMINS_ASSIGN_DRIVE;
|
|||
|
|
} else {
|
|||
|
|
DisplayDllMessageBox (
|
|||
|
|
pC2Data->hWnd,
|
|||
|
|
IDS_AUDIT_ERROR_NO_SET,
|
|||
|
|
IDS_AUDIT_CAPTION,
|
|||
|
|
MBOK_EXCLAIM);
|
|||
|
|
}
|
|||
|
|
} else if (pC2Data->lActionCode == AC_DRIVES_AND_PRINTER_SET_ANYONE) {
|
|||
|
|
if (SetDrivesAndPrinterSetting (0)) {
|
|||
|
|
pC2Data->lC2Compliance = C2DLL_NOT_SECURE;
|
|||
|
|
nMsgString = IDS_ANYONE_CAN_ASSIGN_DRIVE;
|
|||
|
|
} else {
|
|||
|
|
DisplayDllMessageBox (
|
|||
|
|
pC2Data->hWnd,
|
|||
|
|
IDS_AUDIT_ERROR_NO_SET,
|
|||
|
|
IDS_AUDIT_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
|
|||
|
|
C2DisplayAssignDrivesAndPrinter (
|
|||
|
|
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_DRIVES_AND_PRINTERS),
|
|||
|
|
pC2Data->hWnd,
|
|||
|
|
C2DrivesAndPrinterDlgProc,
|
|||
|
|
(LPARAM)&dwNewValue) == IDOK) {
|
|||
|
|
pC2Data->lActionValue = 0;
|
|||
|
|
if (dwNewValue) {
|
|||
|
|
pC2Data->lActionCode = AC_DRIVES_AND_PRINTER_SET_ADMIN;
|
|||
|
|
} else {
|
|||
|
|
pC2Data->lActionCode = AC_DRIVES_AND_PRINTER_SET_ANYONE;
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// no action
|
|||
|
|
pC2Data->lActionCode = AC_DRIVES_AND_PRINTER_NOCHANGE;
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
return ERROR_BAD_ARGUMENTS;
|
|||
|
|
}
|
|||
|
|
return ERROR_SUCCESS;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|