/*++

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;
}