// CreateKeyDlg.cpp : implementation file
//
#include "stdafx.h"
#include "KeyRing.h"
extern "C"
{
#include <wincrypt.h>
#include <sslsp.h>
}
#include "NwKeyDlg.h"
#include "PassDlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CCreateKeyDlg dialog
CCreateKeyDlg::CCreateKeyDlg(CWnd* pParent /*=NULL*/)
: CDialog(CCreateKeyDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CCreateKeyDlg)
m_szNetAddress = _T("");
m_szCountry = _T("");
m_szLocality = _T("");
m_szOrganization = _T("");
m_szUnit = _T("");
m_szState = _T("");
m_szKeyName = _T("");
m_szCertificateFile = _T("");
m_szPassword = _T("");
//}}AFX_DATA_INIT
}
void CCreateKeyDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CCreateKeyDlg)
DDX_Control(pDX, IDC_NEW_KEY_PASSWORD, m_ceditPassword);
DDX_Control(pDX, IDOK, m_btnOK);
DDX_Control(pDX, IDC_NEW_KEY_BITS, m_comboBits);
DDX_Text(pDX, IDC_NEWKEY_COMMONNAME, m_szNetAddress);
DDX_Text(pDX, IDC_NEWKEY_COUNTRY, m_szCountry);
DDV_MaxChars(pDX, m_szCountry, 2);
DDX_Text(pDX, IDC_NEWKEY_LOCALITY, m_szLocality);
DDX_Text(pDX, IDC_NEWKEY_ORG, m_szOrganization);
DDX_Text(pDX, IDC_NEWKEY_ORGUNIT, m_szUnit);
DDX_Text(pDX, IDC_NEWKEY_STATE, m_szState);
DDX_Text(pDX, IDC_NEW_KEY_NAME, m_szKeyName);
DDX_Text(pDX, IDC_NEW_KEY_REQUEST_FILE, m_szCertificateFile);
DDX_Text(pDX, IDC_NEW_KEY_PASSWORD, m_szPassword);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CCreateKeyDlg, CDialog)
//{{AFX_MSG_MAP(CCreateKeyDlg)
ON_EN_CHANGE(IDC_NEW_KEY_NAME, OnChangeNewKeyName)
ON_BN_CLICKED(IDC_NEW_KEY_BROWSE, OnNewKeyBrowse)
ON_EN_CHANGE(IDC_NEW_KEY_REQUEST_FILE, OnChangeNewKeyRequestFile)
ON_EN_CHANGE(IDC_NEW_KEY_PASSWORD, OnChangeNewKeyPassword)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CCreateKeyDlg message handlers
/////////////////////////////////////////////////////////////////////////////
// members
//----------------------------------------------------------------
// override virtual oninitdialog
BOOL CCreateKeyDlg::OnInitDialog( )
{
// set the initial strings
m_szKeyName.LoadString( IDS_CREATE_KEY_NEW_NAME );
m_szOrganization.LoadString( IDS_CREATE_KEY_YOUR_COMPANY );
m_szUnit.LoadString( IDS_CREATE_KEY_YOUR_UNIT );
m_szNetAddress.LoadString( IDS_CREATE_KEY_YOUR_ADDRESS );
m_szCountry.LoadString( IDS_LOCALIZED_DEFAULT_COUNTRY_CODE );
m_szState.LoadString( IDS_CREATE_KEY_YOUR_STATE );
m_szLocality.LoadString( IDS_CREATE_KEY_YOUR_LOCALITY );
// call the base oninit
CDialog::OnInitDialog();
// to comply with the munitions export laws, we need to limit the max bits available
m_nMaxBits = 1024;
// LOOK HERE KIM
m_nMaxBits = SslGetMaximumKeySize(NULL);
// set the default bit size
m_nBits = m_nMaxBits;
m_comboBits.SetCurSel( 2 );
// if necessary, remove items from the bits combo box
if ( m_nMaxBits < 1024 )
{
m_comboBits.DeleteString(2);
m_comboBits.SetCurSel( 1 );
}
if ( m_nMaxBits < 768 )
{
m_comboBits.DeleteString(1);
m_comboBits.SetCurSel( 0 );
}
// any other defaults
m_fKeyNameChangedFile = FALSE;
m_fSpecifiedFile = FALSE;
OnChangeNewKeyName();
// we start with no password, so diable the ok window
m_btnOK.EnableWindow( FALSE );
// return 0 to say we set the default item
// return 1 to just select the default default item
return 1;
}
//----------------------------------------------------------------
void CCreateKeyDlg::OnChangeNewKeyName()
{
// if the user has not specifically chosen a file, update the
// path of the request file to reflect the new file name
if ( !m_fSpecifiedFile )
{
UpdateData( TRUE );
m_szCertificateFile = _T("C:\\");
m_szCertificateFile += m_szKeyName;
m_szCertificateFile += _T(".req");
UpdateData( FALSE );
}
}
//----------------------------------------------------------------
void CCreateKeyDlg::OnChangeNewKeyRequestFile()
{
if ( m_fKeyNameChangedFile )
{
// the change is because of a key name change. No big deal.
// reset the flag and return
m_fKeyNameChangedFile = FALSE;
return;
}
// the user has been typing in the file box, or chose something in
// the dialog. Either way, set the m_fSpecifiedFile flag so that
// we stop changing the path with the key name is changed
m_fSpecifiedFile = TRUE;
}
//----------------------------------------------------------------
void CCreateKeyDlg::OnNewKeyBrowse()
{
CFileDialog cfdlg(FALSE, _T("*.req"), m_szCertificateFile);
CString szFilter;
WORD i = 0;
LPSTR lpszBuffer;
// prepare the filter string
szFilter.LoadString( IDS_CERTIFICATE_FILTER );
// replace the "!" characters with nulls
lpszBuffer = szFilter.GetBuffer(MAX_PATH+1);
while( lpszBuffer[i] )
{
if ( lpszBuffer[i] == _T('!') )
lpszBuffer[i] = _T('\0'); // yes, set \0 on purpose
i++;
}
// prep the dialog
cfdlg.m_ofn.lpstrFilter = lpszBuffer;
// run the dialog
if ( cfdlg.DoModal() == IDOK )
{
// get the current data out of the dialog
UpdateData( TRUE );
// get the path for the file from the dialog
m_szCertificateFile = cfdlg.GetPathName();
UpdateData( FALSE );
// the user has been typing in the file box, or chose something in
// the dialog. Either way, set the m_fSpecifiedFile flag so that
// we stop changing the path with the key name is changed
m_fSpecifiedFile = TRUE;
}
// release the buffer in the filter string
szFilter.ReleaseBuffer(60);
}
//----------------------------------------------------------------
void CCreateKeyDlg::OnOK()
{
HANDLE hFile;
// get the data out of the dialog
UpdateData( TRUE );
// if the file already exists, ask the user if they want to overwrite it
// also, make sure that it is a valid pathname
hFile = CreateFile( m_szCertificateFile, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
// if we opened the file, we have an error
if ( hFile != INVALID_HANDLE_VALUE )
{
// well, first we close the handle, we were only checking if the file was there after all
CloseHandle( hFile );
// ask the user if they really want to overwrite the file
try
{
// get the second part of the message from the resources
CString szMessageNote;
szMessageNote.LoadString( IDS_CERT_FILE_EXISTS );
// next, build the string we will use in the message box
CString szMessage;
szMessage = m_szCertificateFile;
szMessage += szMessageNote;
// put up the message box, if the user choose no, they do not want to overwrite the
// file. Then we should exit now
if ( AfxMessageBox(szMessage, MB_ICONQUESTION|MB_YESNO) == IDNO )
return;
}
catch( CException e )
{
return;
}
}
// first, we need to confirm that the password is alright
// the user must re-enter the password to confirm it
CConfirmPassDlg dlgconfirm;
if ( dlgconfirm.DoModal() != IDOK )
return;
// confirm the password
if ( dlgconfirm.m_szPassword != m_szPassword )
{
AfxMessageBox( IDS_INCORRECT_PASSWORD );
return;
}
// set the default bit size
switch( m_comboBits.GetCurSel() )
{
case 0: // bits == 512
m_nBits = 512;
break;
case 1: // bits == 768
m_nBits = 768;
break;
case 2: // bits == 1024
m_nBits = 1024;
break;
};
// call the inherited OnOK
CDialog::OnOK();
}
//----------------------------------------------------------------
void CCreateKeyDlg::OnChangeNewKeyPassword()
{
// if there is no password, disable the ok button.
// otherwise, enable it
UpdateData( TRUE );
m_btnOK.EnableWindow( !m_szPassword.IsEmpty() );
}
//----------------------------------------------------------------
BOOL CCreateKeyDlg::PreTranslateMessage(MSG* pMsg)
{
// filter commas out of all edit fields except the password field.
// commas would interfere with the distinguishing name information
if ( (pMsg->message == WM_CHAR) &&
(pMsg->hwnd != m_ceditPassword) &&
((TCHAR)pMsg->wParam == _T(',')) )
{
MessageBeep(0);
return TRUE;
}
// translate normally
return CDialog::PreTranslateMessage(pMsg);
}