|
|
/////////////////////////////////////////////////////////////////////////////////
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 2000-2002.
//
// File: SelectTemplateDlg.cpp
//
// Contents: Implementation of CSelectTemplateDlg
//
//----------------------------------------------------------------------------
//
#include "stdafx.h"
#include "CompData.h"
#include "SelectTemplateDlg.h"
#include "CertTemplate.h"
#include "TemplateGeneralPropertyPage.h"
#include "TemplateV1RequestPropertyPage.h"
#include "TemplateV2RequestPropertyPage.h"
#include "TemplateV1SubjectNamePropertyPage.h"
#include "TemplateV2SubjectNamePropertyPage.h"
#include "TemplateV2AuthenticationPropertyPage.h"
#include "TemplateV2SupercedesPropertyPage.h"
#include "TemplateExtensionsPropertyPage.h"
#include "PolicyOID.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;#endif
/////////////////////////////////////////////////////////////////////////////
// CSelectTemplateDlg dialog
CSelectTemplateDlg::CSelectTemplateDlg(CWnd* pParent, const CCertTmplComponentData* pCompData, const CStringList& supercededNameList) : CHelpDialog(CSelectTemplateDlg::IDD, pParent), m_supercededTemplateNameList (supercededNameList), m_pCompData (pCompData){ //{{AFX_DATA_INIT(CSelectTemplateDlg)
//}}AFX_DATA_INIT
}
void CSelectTemplateDlg::DoDataExchange(CDataExchange* pDX){ CHelpDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CSelectTemplateDlg)
DDX_Control(pDX, IDC_TEMPLATE_LIST, m_templateList); //}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CSelectTemplateDlg, CHelpDialog) //{{AFX_MSG_MAP(CSelectTemplateDlg)
ON_BN_CLICKED(IDC_TEMPLATE_PROPERTIES, OnTemplateProperties) ON_NOTIFY(LVN_ITEMCHANGED, IDC_TEMPLATE_LIST, OnItemchangedTemplateList) ON_NOTIFY(NM_DBLCLK, IDC_TEMPLATE_LIST, OnDblclkTemplateList) ON_NOTIFY(LVN_DELETEITEM, IDC_TEMPLATE_LIST, OnDeleteitemTemplateList) ON_WM_DESTROY() //}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CSelectTemplateDlg message handlers
BOOL CSelectTemplateDlg::OnInitDialog() { _TRACE (1, L"Entering CSelectTemplateDlg::OnInitDialog\n"); CHelpDialog::OnInitDialog(); CWaitCursor cursor; // Set up list controls
COLORREF cr = RGB (255, 0, 255); CThemeContextActivator activator; VERIFY (m_imageListNormal.Create (IDB_TEMPLATES, 32, 0, cr)); VERIFY (m_imageListSmall.Create (IDB_TEMPLATES, 16, 0, cr)); m_templateList.SetImageList (CImageList::FromHandle (m_imageListSmall), LVSIL_SMALL); m_templateList.SetImageList (CImageList::FromHandle (m_imageListNormal), LVSIL_NORMAL);
int colWidths[NUM_COLS] = {200, 200};
// Add "Certificate Extension" column
CString szText; VERIFY (szText.LoadString (IDS_CERTIFICATE_TEMPLATES)); VERIFY (m_templateList.InsertColumn (COL_CERT_TEMPLATE, (LPCWSTR) szText, LVCFMT_LEFT, colWidths[COL_CERT_TEMPLATE], COL_CERT_TEMPLATE) != -1);
VERIFY (szText.LoadString (IDS_COLUMN_SUPPORTED_CAS)); VERIFY (m_templateList.InsertColumn (COL_CERT_VERSION, (LPCWSTR) szText, LVCFMT_LEFT, colWidths[COL_CERT_VERSION], COL_CERT_VERSION) != -1);
ASSERT (m_pCompData); if ( m_pCompData ) { POSITION pos = m_pCompData->m_globalTemplateNameList.GetHeadPosition (); CString szTemplateName;
for (; pos; ) { szTemplateName = m_pCompData->m_globalTemplateNameList.GetNext (pos);
// #NTRAID 363879 Certtmpl: Certificate Template Snapin must not
// allow the Subordinate CA template to be Superceded
if ( wszCERTTYPE_SUBORDINATE_CA == szTemplateName ) continue;
// Only add those templates which are not already superceded
if ( !m_supercededTemplateNameList.Find (szTemplateName) ) { HCERTTYPE hCertType = 0; HRESULT hr = CAFindCertTypeByName (szTemplateName, NULL, CT_ENUM_MACHINE_TYPES | CT_ENUM_USER_TYPES | CT_FLAG_NO_CACHE_LOOKUP, &hCertType); _ASSERT (SUCCEEDED (hr)); if ( SUCCEEDED (hr) ) { // NTRAID# 660535 Certtmpl: Supercedes list should filter
// out any certificate templates which have the TYPE == CA
DWORD dwGeneralFlags = 0;
hr = CAGetCertTypeFlagsEx (hCertType, CERTTYPE_GENERAL_FLAG, &dwGeneralFlags); if ( FAILED (hr) ) continue;
if ( dwGeneralFlags & (CT_FLAG_IS_CA | CT_FLAG_IS_CROSS_CA) ) continue;
PWSTR* rgwszProp = 0;
hr = CAGetCertTypePropertyEx (hCertType, CERTTYPE_PROP_FRIENDLY_NAME, &rgwszProp); if ( SUCCEEDED (hr) ) { DWORD dwVersion = 0; hr = CAGetCertTypePropertyEx (hCertType, CERTTYPE_PROP_SCHEMA_VERSION, &dwVersion); if ( SUCCEEDED (hr) ) { LV_ITEM lvItem; int iItem = m_templateList.GetItemCount (); int iResult = 0;
// security review 2/21/2002 BryanWal ok
::ZeroMemory (&lvItem, sizeof (lvItem)); lvItem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM; lvItem.iItem = iItem; lvItem.iSubItem = COL_CERT_TEMPLATE; lvItem.pszText = rgwszProp[0]; if ( 1 == dwVersion ) lvItem.iImage = 0; // version is 1
else lvItem.iImage = 1; // version is 2
lvItem.lParam = (LPARAM) new CString (szTemplateName); iItem = m_templateList.InsertItem (&lvItem); ASSERT (-1 != iItem); if ( -1 == iItem ) break; else { // security review 2/21/2002 BryanWal ok
::ZeroMemory (&lvItem, sizeof (lvItem)); lvItem.mask = LVIF_TEXT; lvItem.iItem = iItem; lvItem.iSubItem = COL_CERT_VERSION; CString text; if ( 1 == dwVersion ) VERIFY (text.LoadString (IDS_WINDOWS_2000_AND_LATER)); else VERIFY (text.LoadString (IDS_WINDOWS_2002_AND_LATER)); lvItem.pszText = (PWSTR)(PCWSTR) text; iResult = m_templateList.SetItem (&lvItem); ASSERT (-1 != iResult); } } else { _TRACE (0, L"CAGetCertTypePropertyEx (CERTTYPE_PROP_SCHEMA_VERSION) failed: 0x%x\n", hr); }
CAFreeCertTypeProperty (hCertType, rgwszProp); } else { _TRACE (0, L"CAGetCertTypePropertyEx (CERTTYPE_PROP_FRIENDLY_NAME) failed: 0x%x\n", hr); }
hr = CACloseCertType (hCertType); _ASSERT (SUCCEEDED (hr)); if ( !SUCCEEDED (hr) ) { _TRACE (0, L"CACloseCertType (%s) failed: 0x%x\n", hr); } } else { _TRACE (0, L"CAFindCertTypeByName (%s) failed: 0x%x\n", (PCWSTR) szTemplateName, hr); } } } }
EnableControls ();
_TRACE (-1, L"Leaving CSelectTemplateDlg::OnInitDialog\n"); return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
void CSelectTemplateDlg::OnTemplateProperties() { int nSelCnt = m_templateList.GetSelectedCount (); int nSelItem = GetSelectedListItem ();
if ( 1 == nSelCnt ) { CString szFriendlyName = m_templateList.GetItemText (nSelItem, COL_CERT_TEMPLATE); CString* pszTemplateName = (CString*) m_templateList.GetItemData (nSelItem); HCERTTYPE hCertType = 0; HRESULT hr = CAFindCertTypeByName (*pszTemplateName, NULL, CT_ENUM_MACHINE_TYPES | CT_ENUM_USER_TYPES | CT_FLAG_NO_CACHE_LOOKUP, &hCertType); _ASSERT (SUCCEEDED (hr)); if ( SUCCEEDED (hr) ) { CCertTemplate certTemplate (szFriendlyName, *pszTemplateName, L"", true, m_pCompData->m_fUseCache); CString title;
// security review 2/21/2002 BryanWal ok
title.FormatMessage (IDS_PROPERTIES_OF_TEMPLATE_X, szFriendlyName); CTemplatePropertySheet propSheet (title, certTemplate, this);
if ( 1 == certTemplate.GetType () ) { CTemplateGeneralPropertyPage* pGeneralPage = new CTemplateGeneralPropertyPage (certTemplate, m_pCompData); if ( pGeneralPage ) { // Add General page
propSheet.AddPage (pGeneralPage);
// Add Request and Subject Name page only if subject is not a CA
if ( !certTemplate.SubjectIsCA () ) { propSheet.AddPage (new CTemplateV1RequestPropertyPage ( certTemplate)); propSheet.AddPage (new CTemplateV1SubjectNamePropertyPage ( certTemplate)); }
// Add extensions page
propSheet.AddPage (new CTemplateExtensionsPropertyPage ( certTemplate, pGeneralPage->m_bIsDirty)); } } else // version is 2
{ CTemplateGeneralPropertyPage* pGeneralPage = new CTemplateGeneralPropertyPage (certTemplate, m_pCompData); if ( pGeneralPage ) { propSheet.AddPage (pGeneralPage);
// Add Request and Subject pages if subject is not a CA
if ( !certTemplate.SubjectIsCA () ) { propSheet.AddPage (new CTemplateV2RequestPropertyPage ( certTemplate, pGeneralPage->m_bIsDirty)); propSheet.AddPage (new CTemplateV2SubjectNamePropertyPage ( certTemplate, pGeneralPage->m_bIsDirty)); } propSheet.AddPage (new CTemplateV2AuthenticationPropertyPage ( certTemplate, pGeneralPage->m_bIsDirty)); propSheet.AddPage (new CTemplateV2SupercedesPropertyPage ( certTemplate, pGeneralPage->m_bIsDirty, m_pCompData)); propSheet.AddPage (new CTemplateExtensionsPropertyPage ( certTemplate, pGeneralPage->m_bIsDirty)); } }
CThemeContextActivator activator; propSheet.DoModal ();
hr = CACloseCertType (hCertType); _ASSERT (SUCCEEDED (hr)); if ( !SUCCEEDED (hr) ) { _TRACE (0, L"CACloseCertType (%s) failed: 0x%x\n", hr); } } else { _TRACE (0, L"CAFindCertTypeByName (%s) failed: 0x%x\n", (PCWSTR) pszTemplateName, hr); } }}
void CSelectTemplateDlg::OnOK() { int nCnt = m_templateList.GetItemCount (); ASSERT (nCnt >= 1); UINT flag = 0; while (--nCnt >= 0) { flag = ListView_GetItemState (m_templateList.m_hWnd, nCnt, LVIS_SELECTED); if ( flag & LVNI_SELECTED ) { CString* pszTemplateName = (CString*) m_templateList.GetItemData (nCnt); if ( pszTemplateName ) m_returnedTemplates.AddTail (*pszTemplateName); } } CHelpDialog::OnOK();}
void CSelectTemplateDlg::EnableControls(){ int nSelCnt = m_templateList.GetSelectedCount ();
GetDlgItem (IDC_TEMPLATE_PROPERTIES)->EnableWindow (1 == nSelCnt); GetDlgItem (IDOK)->EnableWindow (nSelCnt > 0);}
void CSelectTemplateDlg::OnItemchangedTemplateList(NMHDR* /*pNMHDR*/, LRESULT* pResult) { EnableControls (); *pResult = 0;}
int CSelectTemplateDlg::GetSelectedListItem(){ int nSelItem = -1;
if ( m_templateList.m_hWnd && m_templateList.GetSelectedCount () > 0 ) { int nCnt = m_templateList.GetItemCount (); ASSERT (nCnt >= 1); UINT flag = 0; while (--nCnt >= 0) { flag = ListView_GetItemState (m_templateList.m_hWnd, nCnt, LVIS_SELECTED); if ( flag & LVNI_SELECTED ) { nSelItem = nCnt; break; } } }
return nSelItem;}
void CSelectTemplateDlg::OnDblclkTemplateList(NMHDR* /*pNMHDR*/, LRESULT* pResult) { OnTemplateProperties (); *pResult = 0;}
void CSelectTemplateDlg::OnDeleteitemTemplateList(NMHDR* pNMHDR, LRESULT* pResult) { NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR; CString* pszTemplateName = (CString*) m_templateList.GetItemData (pNMListView->iItem); if ( pszTemplateName ) delete pszTemplateName; *pResult = 0;}
void CSelectTemplateDlg::DoContextHelp (HWND hWndControl){ _TRACE(1, L"Entering CSelectTemplateDlg::DoContextHelp\n"); switch (::GetDlgCtrlID (hWndControl)) { case IDC_STATIC: break;
default: // Display context help for a control
if ( !::WinHelp ( hWndControl, GetContextHelpFile (), HELP_WM_HELP, (DWORD_PTR) g_aHelpIDs_IDD_SELECT_TEMPLATE) ) { _TRACE(0, L"WinHelp () failed: 0x%x\n", GetLastError ()); } break; } _TRACE(-1, L"Leaving CSelectTemplateDlg::DoContextHelp\n");}
void CSelectTemplateDlg::OnDestroy() { CHelpDialog::OnDestroy(); m_imageListNormal.Destroy (); m_imageListSmall.Destroy ();}
|
