309 lines
10 KiB
C++
309 lines
10 KiB
C++
|
// 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);
|
||
|
|
}
|