cmaindialog.cpp

vc环境下的pgp源码

//////////////////////////////////////////////////////////////////////////////
// CMainDialog.cpp
//
// Implementation of the CMainDialog class.
//////////////////////////////////////////////////////////////////////////////

// $Id: CMainDialog.cpp,v 1.14 1999/02/26 04:09:57 nryan Exp $

// Copyright (C) 1998 by Network Associates, Inc.
// All rights reserved.

#include "StdAfx.h"

#include "Required.h"
#include "PGPdiskPrefs.h"

#include "CMainDialog.h"
#include "Globals.h"
#include "PGPdiskHelpIds.h"


////////////
// Constants
////////////

// Size constants for the main window.
const PGPUInt32	kWidthLargeMainDialog	= 268;
const PGPUInt32	kHeightLargeMainDialog	= 69;
const PGPUInt32	kWidthSmallMainDialog	= 132;
const PGPUInt32	kHeightSmallMainDialog	= 36;

// Horizontal bias constants for the main window buttons.
const PGPUInt32	kBigButtonXBias			= 1;
const PGPUInt32	kBigButtonYBias			= 1;
const PGPUInt32	kSmallButtonXBias		= 1;
const PGPUInt32	kSmallButtonYBias		= 2;

// Size constants for the main window buttons
const PGPUInt32	kSideLengthBigButton	= 66;
const PGPUInt32	kSideLengthSmallButton	= 32;

// Size constants for the main window button icons.
const PGPUInt32	kSideLengthBigIcon		= 60;
const PGPUInt32	kSideLengthSmallIcon	= 25;


/////////////////////
// Context help array
/////////////////////

static PGPUInt32 HelpIds[] =
{
	IDC_NEWPGPDISK,			IDH_PGPDISKAPP_MAINNEW, 
	IDC_MOUNTPGPDISK,		IDH_PGPDISKAPP_MAINMOUNT, 
	IDC_UNMOUNTPGPDISK,		IDH_PGPDISKAPP_MAINUNMOUNT, 
	IDC_PREFERENCES,		IDH_PGPDISKAPP_MAINPREFS, 
    0,0 
};


///////////////////////////
// MFC specific definitions
///////////////////////////

BEGIN_MESSAGE_MAP(CMainDialog, CDialog)
	//{{AFX_MSG_MAP(CMainDialog)
	ON_WM_DESTROY()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_COMMAND(IDM_ABOUT, OnAboutBox)
	ON_COMMAND(IDM_ADDPASS, OnAddPassphrase)
	ON_COMMAND(IDM_CHANGEPASS, OnChangePassphrase)
	ON_WM_DROPFILES()
	ON_COMMAND(IDM_REMOVEALL, OnRemoveAlternates)
	ON_COMMAND(IDM_REMOVEPASS, OnRemovePassphrase)
	ON_COMMAND(IDM_EXIT, OnExit)
	ON_BN_CLICKED(IDC_MOUNTPGPDISK, OnMountPGPdisk)
	ON_COMMAND(IDM_UNMOUNTPGPDISK, OnUnmountPGPdisk)
	ON_BN_CLICKED(IDC_NEWPGPDISK, OnNewPGPdisk)
	ON_BN_CLICKED(IDC_PREFERENCES, OnPreferences)
	ON_COMMAND(IDM_HELPCONTENTS, OnHelpContents)
	ON_WM_CLOSE()
	ON_WM_INITMENU()
	ON_WM_HELPINFO()
	ON_WM_SYSCOMMAND()
	ON_WM_SIZING()
	ON_WM_SIZE()
	ON_COMMAND(IDM_MOUNTPGPDISK, OnMountPGPdisk)
	ON_COMMAND(IDM_UNMOUNTPGPDISK, OnUnmountPGPdisk)
	ON_COMMAND(IDM_NEWPGPDISK, OnNewPGPdisk)
	ON_COMMAND(IDM_PREFS, OnPreferences)
	ON_COMMAND(IDM_PUBLICKEYS, OnAddRemovePublicKeys)
	ON_COMMAND(IDM_GLOBALCONVERT, OnGlobalConvert)
	ON_BN_CLICKED(IDC_UNMOUNTPGPDISK, OnUnmountPGPdisk)
	//}}AFX_MSG_MAP

END_MESSAGE_MAP()


///////////////////////////////////////////////////////////////////////
// CMainDialog public custom functions and non-default message handlers
///////////////////////////////////////////////////////////////////////

// The CMainDialog constructor.

CMainDialog::CMainDialog(CWnd *pParent) : CDialog(CMainDialog::IDD, pParent)
{
	//{{AFX_DATA_INIT(CMainDialog)
	//}}AFX_DATA_INIT

	mAccelTable = NULL;
	mHIcon = App->LoadIcon(IDI_MAINFRAME);

	mArePastInitDialog = FALSE;

}

// The CMainDialog destructor.
CMainDialog::~CMainDialog()
{
	if (IsntNull(m_hWnd))
		DestroyWindow();
}

// We override PreTranslateMessage so we can handle accelerators, the escape
// key, and the return key properly.

BOOL 
CMainDialog::PreTranslateMessage(MSG *pMsg) 
{
	PGPBoolean weProcessed = FALSE;

	if (mAccelTable && TranslateAccelerator(m_hWnd, mAccelTable, pMsg))
	{
		weProcessed = TRUE;
	}
	else if (pMsg->message == WM_KEYDOWN)
	{
		PGPUInt8 vCode = pMsg->wParam & 0x7F;

		switch (vCode)
		{
		case VK_ESCAPE:
			// Ignore the escape key.
			weProcessed = TRUE;
			break;

		case VK_RETURN:
			// Letting MFC handle return results in main dialog going away.
			if (HasFocus(&mNewButton))
				PostMessage(WM_COMMAND, IDM_NEWPGPDISK, NULL);
			else if (HasFocus(&mMountButton))
				PostMessage(WM_COMMAND, IDM_MOUNTPGPDISK, NULL);
			else if (HasFocus(&mUnmountButton))
				PostMessage(WM_COMMAND, IDM_UNMOUNTPGPDISK, NULL);
			else if (HasFocus(&mPrefsButton))
				PostMessage(WM_COMMAND, IDM_PREFS, NULL);

			weProcessed = TRUE;
			break;
		}
	}

	if (weProcessed)
		return TRUE;
	else
		return CDialog::PreTranslateMessage(pMsg);
}


//////////////////////////////////////////////////////////////////////////
// CMainDialog protected custom functions and non-default message handlers
//////////////////////////////////////////////////////////////////////////

// ShowMainMenu shows the main menu, loading it if necessary.

DualErr 
CMainDialog::ShowMainMenu()
{
	DualErr derr;

	// Load the main menu if it's not loaded already
	if (IsNull(mMainMenu.GetSafeHmenu()))
	{
		if (!mMainMenu.LoadMenu(IDR_MAIN_MENU))
		{
			mInitErr = kPGDMinorError_DialogDisplayFailed;
		}
	}

	// Show the main menu if it's not shown already.
	if (derr.IsntError())
	{
		if (IsNull(GetMenu()))
		{
			// Display the main menu.
			if (!SetMenu(&mMainMenu))
			{
				derr = kPGDMinorError_DialogDisplayFailed;
			}
		}
	}

	return derr;
}

// HideMainMenu hides the main menu.

void 
CMainDialog::HideMainMenu()
{
	pgpAssert(IsntNull(mMainMenu.GetSafeHmenu()));

	SetMenu(NULL);
}

// ConvertMainDialogToLarge enlarges the dialog and its buttons, and shows the
// menubar.

void 
CMainDialog::ConvertMainDialogToLarge()
{
	PGPInt32	edgeX		= GetSystemMetrics(SM_CXFIXEDFRAME);
	PGPInt32	edgeY		= GetSystemMetrics(SM_CYFIXEDFRAME);
	PGPInt32	captionY	= GetSystemMetrics(SM_CYCAPTION);
	PGPInt32	menuY		= GetSystemMetrics(SM_CYMENU);
	PGPInt32	largeSizeX, largeSizeY;

	mAreWeInLargeMode = TRUE;

	// Show the main menu.
	ShowMainMenu();

	// Resize the dialog.
	largeSizeX	= kWidthLargeMainDialog + edgeX * 2;
	largeSizeY	= kHeightLargeMainDialog + edgeY * 2 + captionY + menuY;

	SetWindowPos(NULL, 0, 0, largeSizeX, largeSizeY, 
		SWP_NOMOVE | SWP_NOZORDER);

	// Move and resize the buttons.
	mNewButton.MoveWindow(kBigButtonXBias, kBigButtonYBias, 
		kSideLengthBigButton, kSideLengthBigButton);
	mMountButton.MoveWindow(kSideLengthBigButton + kBigButtonXBias, 
		kBigButtonYBias, kSideLengthBigButton, kSideLengthBigButton);
	mUnmountButton.MoveWindow(kSideLengthBigButton*2 + kBigButtonXBias, 
		kBigButtonYBias, kSideLengthBigButton, kSideLengthBigButton);
	mPrefsButton.MoveWindow(kSideLengthBigButton*3 + kBigButtonXBias, 
		kBigButtonYBias, kSideLengthBigButton, kSideLengthBigButton);

	// Load the big icons.
	mMountButton.SetIcon(mMountBigIcon);
	mNewButton.SetIcon(mNewBigIcon);
	mPrefsButton.SetIcon(mPrefsBigIcon);
	mUnmountButton.SetIcon(mUnmountBigIcon);
}

// ConvertMainDialogToSmall shrinks the dialog and its buttons, and hides the
// the menubar.

void 
CMainDialog::ConvertMainDialogToSmall()
{
	PGPInt32	edgeX		= GetSystemMetrics(SM_CXFIXEDFRAME);
	PGPInt32	edgeY		= GetSystemMetrics(SM_CYFIXEDFRAME);
	PGPInt32	captionY	= GetSystemMetrics(SM_CYCAPTION);
	PGPInt32	smallSizeX, smallSizeY;

	mAreWeInLargeMode = FALSE;

	// Load the small icons.
	mMountButton.SetIcon(mMountSmallIcon);
	mNewButton.SetIcon(mNewSmallIcon);
	mPrefsButton.SetIcon(mPrefsSmallIcon);
	mUnmountButton.SetIcon(mUnmountSmallIcon);

	// Move and resize the buttons.
	mNewButton.MoveWindow(kSmallButtonXBias, kSmallButtonYBias, 
		kSideLengthSmallButton, kSideLengthSmallButton);
	mMountButton.MoveWindow(kSideLengthSmallButton + kSmallButtonXBias, 
		kSmallButtonYBias, kSideLengthSmallButton, kSideLengthSmallButton);
	mUnmountButton.MoveWindow(kSideLengthSmallButton*2 + kSmallButtonXBias, 
		kSmallButtonYBias, kSideLengthSmallButton, kSideLengthSmallButton);
	mPrefsButton.MoveWindow(kSideLengthSmallButton*3 + kSmallButtonXBias, 
		kSmallButtonYBias, kSideLengthSmallButton, kSideLengthSmallButton);

	// Hide the main menu.
	HideMainMenu();

	// Resize the dialog.
	smallSizeX	= kWidthSmallMainDialog + edgeX * 2;
	smallSizeY	= kHeightSmallMainDialog + edgeY * 2 + captionY;

	SetWindowPos(NULL, 0, 0, smallSizeX, smallSizeY, 
		SWP_NOMOVE | SWP_NOZORDER);
}

// RecallMainWndOnTop recalls the 'window on top' position of the main window
// from the registry and checks the menu item to reflect this.

void 
CMainDialog::RecallMainWndOnTop()
{
	CMenu				*pSysMenu;
	PGPdiskWin32Prefs	prefs;
	PGPUInt32			checkState;

	if (GetPGPdiskWin32Prefs(prefs).IsntError())
	{
		pSysMenu = GetSystemMenu(FALSE);
		pgpAssertAddrValid(pSysMenu, CMenu);

		checkState = prefs.mainStayOnTop;

		pSysMenu->CheckMenuItem(IDM_STAYONTOP, MF_BYCOMMAND | checkState);

		if (checkState == MF_CHECKED)
		{
			SetWindowPos(&CWnd::wndTopMost, 0, 0, 0, 0, 
				SWP_NOMOVE | SWP_NOSIZE);
		}
	}
}

// SaveMainWndOnTop saves the 'window on top' position of the main window in
// the registry.

void 
CMainDialog::SaveMainWndOnTop()
{
	CMenu			*pSysMenu;
	PGPUInt32		checkState;
	PGPdiskWin32Prefs	prefs;

	pSysMenu = GetSystemMenu(FALSE);
	pgpAssertAddrValid(pSysMenu, CMenu);

	checkState = pSysMenu->GetMenuState(IDM_STAYONTOP, MF_BYCOMMAND);

	if (GetPGPdiskWin32Prefs(prefs).IsntError())
	{
		prefs.mainStayOnTop = checkState;
		SetPGPdiskWin32Prefs(prefs);
	}
}

// The CMainDialog data exchange function.

void 
CMainDialog::DoDataExchange(CDataExchange *pDX)
{
	try
	{
		CDialog::DoDataExchange(pDX);

		//{{AFX_DATA_MAP(CMainDialog)
		DDX_Control(pDX, IDC_UNMOUNTPGPDISK, mUnmountButton);
		DDX_Control(pDX, IDC_MOUNTPGPDISK, mMountButton);
		DDX_Control(pDX, IDC_PREFERENCES, mPrefsButton);
		DDX_Control(pDX, IDC_NEWPGPDISK, mNewButton);
		//}}AFX_DATA_MAP
	}
	catch (CMemoryException *ex)
	{
		ex->Delete();
	}
}

// OnAboutBox displays the about box.

void 
CMainDialog::OnAboutBox()
{
	App->CallShowAboutBox();
}

// OnAddPassphrase adds a passphrase to a PGPdisk.

void 
CMainDialog::OnAddPassphrase()
{
	App->AskUserAddPassphrase();
}

// OnChangePassphrase changes a passphrase on a PGPdisk.

void 
CMainDialog::OnChangePassphrase()
{
	App->AskUserChangePassphrase();
}

// OnExit is called when the user chooses Exit.

void 
CMainDialog::OnExit() 
{
	PostMessage(WM_CLOSE);
}

// OnHelpContents displays the PGPdisk help file.

void 
CMainDialog::OnHelpContents()
{
	App->CallShowHelp();
}

// OnAddRemovePublicKeys is called when the user selects the "Manage Public
// Keys" menu option.

void 
CMainDialog::OnAddRemovePublicKeys()
{
	App->AskUserAddRemovePublicKeys();
}

// OnGlobalConvert is called when the user selects the "Convert Old PGPdisks"
// menu item.

void 
CMainDialog::OnGlobalConvert()
{
	App->AskUserGlobalConvertPGPdisks();
}

// OnMountPGPdisk is called when the user presses the "Mount PGPdisk" button.

void 
CMainDialog::OnMountPGPdisk()
{
	App->AskUserMountPGPdisk();
}

// OnNewPGPdisk is called when the user presses the New PGPdisk button. We
// display the "Create New PGPdisk" dialog box.

void 
CMainDialog::OnNewPGPdisk()
{
	App->AskUserNewPGPdisk();
}

// OnPreferences handles the 'Preferences' button.

void 
CMainDialog::OnPreferences() 
{
	App->CallShowPrefs();
}

// OnRemoveAlternates removes all alternate passphrases from a PGPdisk.

void 
CMainDialog::OnRemoveAlternates()
{
	App->AskUserRemovePassphrase(TRUE);
}

// OnRemovePassphrase removes a passphrase from a PGPdisk.

void 
CMainDialog::OnRemovePassphrase()
{
	App->AskUserRemovePassphrase();
}

// OnUnmountPGPdisk is called when the user selects the "Unmount PGPdisk" menu
// item.

void 
CMainDialog::OnUnmountPGPdisk()
{
	App->AskUserUnmountPGPdisk();
}


/////////////////////////////////////////////////
// CMainDialog protected default message handlers
/////////////////////////////////////////////////

// OnClose is called before the window is destroyed.

void 
CMainDialog::OnClose() 
{
	PGPdiskWin32Prefs	prefs;

	// Save the big/small window pref in the registry.
	if (GetPGPdiskWin32Prefs(prefs).IsntError())
	{
		prefs.wasDialogSmall = !mAreWeInLargeMode;
		SetPGPdiskWin32Prefs(prefs);
	}

	// Save the window coords in the registry.
	SaveWindowPos(this);

	DestroyWindow();
}

// OnCommand is overridden so we can make sure that spurious commands aren't
// executed in command-line mode.

BOOL 
CMainDialog::OnCommand(WPARAM wParam, LPARAM lParam) 
{
	if (App->mCommandLineMode)			// ignore if in command line mode
		return TRUE;
	
	return CDialog::OnCommand(wParam, lParam);
}

// OnDestroy is called when the dialog is going away.

void 
CMainDialog::OnDestroy()
{
	CDialog::OnDestroy();
}

// OnDropFiles is called to handle files dropped on the dialog which the user
// dragged there from the desktop.

void 
CMainDialog::OnDropFiles(HDROP hDrop)
{
	PGPUInt32 numDropped;

	numDropped = DragQueryFile(hDrop, 0xFFFFFFFF, NULL, NULL);