Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

518 lines
16 KiB

// This is a part of the Microsoft Management Console.
// Copyright (C) Microsoft Corporation, 1995 - 1999
// All rights reserved.
//
// This source code is only intended as a supplement to the
// Microsoft Management Console and related
// electronic documentation provided with the interfaces.
#include "stdafx.h"
#include "certsrv.h"
#include "csprop.h"
#include "setupids.h"
#include "misc.h"
#define __dwFILE__ __dwFILE_CERTMMC_EVENTS_CPP__
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// Event handlers for IFrame::Notify
HRESULT
CSnapin::OnAddImages(
MMC_COOKIE, // cookie
LPARAM arg,
LPARAM /* param */ )
{
if (arg == 0)
return E_INVALIDARG;
ASSERT(m_pImageResult != NULL);
ASSERT((IImageList*)arg == m_pImageResult);
CBitmap bmpResultStrip16x16, bmpResultStrip32x32;
if (NULL == bmpResultStrip16x16.LoadBitmap(IDB_16x16))
return S_FALSE;
if (NULL == bmpResultStrip32x32.LoadBitmap(IDB_32x32))
return S_FALSE;
// Set the images
m_pImageResult->ImageListSetStrip(reinterpret_cast<LONG_PTR*>(static_cast<HBITMAP>(bmpResultStrip16x16)),
reinterpret_cast<LONG_PTR*>(static_cast<HBITMAP>(bmpResultStrip32x32)),
0, RGB(255, 0, 255));
return S_OK;
}
HRESULT
CSnapin::OnShow(
MMC_COOKIE cookie,
LPARAM arg,
LPARAM /* param */ )
{
HRESULT hr;
CFolder* pFolder = dynamic_cast<CComponentDataImpl*>(m_pComponentData)->FindObject(cookie);
// Note - arg is TRUE when it is time to enumerate
if (arg == TRUE)
{
m_pCurrentlySelectedScopeFolder = pFolder;
// if list view on display
if (m_CustomViewID == VIEW_DEFAULT_LV)
{
// Show the headers for this nodetype
hr = InitializeHeaders(cookie);
// virtual list support
if (m_bVirtualView)
m_pResult->SetItemCount(1, 0);
}
}
else
{
// if list view is on display
if (m_CustomViewID == VIEW_DEFAULT_LV)
{
// Free data associated with the result pane items, because
// your node is no longer being displayed.
}
// Note: The console will remove the items from the result pane
}
return S_OK;
}
HRESULT CSnapin::GetRowColContents(CFolder* pFolder, LONG idxRow, LPCWSTR szColHead, PBYTE* ppbData, DWORD* pcbData, BOOL fStringFmt /*FALSE*/)
{
HRESULT hr;
LONG idxCol;
IEnumCERTVIEWROW* pRow = NULL;
ICertView* pView = NULL;
#if DBG
DWORD dwVerifySize;
#endif
CComponentDataImpl* pCompData = dynamic_cast<CComponentDataImpl*>(m_pComponentData);
ASSERT(pCompData != NULL);
if (pCompData == NULL)
return E_POINTER;
ASSERT(pFolder != NULL);
// pollute the row enumerator we've got
hr = m_RowEnum.GetRowEnum(pFolder->GetCA(), &pRow);
_JumpIfError(hr, Ret, "GetRowEnum");
hr = m_RowEnum.SetRowEnumPos(idxRow);
_JumpIfError(hr, Ret, "SetRowEnumPos");
// now we have the correct row; siphon data out of the correct column
hr = m_RowEnum.GetView(pFolder->GetCA(), &pView);
_JumpIfError(hr, Ret, "GetView");
// get column number in schema
idxCol = pCompData->FindColIdx(szColHead);
// retrieve and alloc
*pcbData = 0;
hr = GetCellContents(&m_RowEnum, pFolder->GetCA(), idxRow, idxCol, NULL, pcbData, fStringFmt);
_JumpIfError(hr, Ret, "GetCellContents");
*ppbData = new BYTE[*pcbData];
_JumpIfOutOfMemory(hr, Ret, *ppbData);
#if DBG
dwVerifySize = *pcbData;
#endif
hr = GetCellContents(&m_RowEnum, pFolder->GetCA(), idxRow, idxCol, *ppbData, pcbData, fStringFmt);
_JumpIfError(hr, Ret, "GetCellContents");
#if DBG
ASSERT(dwVerifySize == *pcbData);
#endif
Ret:
// catch column inclusion errors, handle in a smart way
if (hr == HRESULT_FROM_WIN32(ERROR_INVALID_INDEX) ||
hr == HRESULT_FROM_WIN32(ERROR_CONTINUE))
{
CString cstrFormat;
cstrFormat.LoadString(IDS_COLUMN_INCLUSION_ERROR);
LPCWSTR pszLocalizedCol = NULL;
hr = myGetColumnDisplayName(szColHead, &pszLocalizedCol);
ASSERT ((hr == S_OK) && (NULL != pszLocalizedCol));
CString cstrTmp;
cstrTmp.Format(cstrFormat, pszLocalizedCol);
cstrFormat.Empty();
cstrFormat.LoadString(IDS_MSG_TITLE);
m_pConsole->MessageBoxW(cstrTmp, cstrFormat, MB_OK, NULL);
hr = ERROR_CANCELLED; // this is a cancellation so bail silently, we've shown error
}
return hr;
}
HRESULT
GetBinaryColumnFormat(
IN WCHAR const *pwszColumnName,
OUT LONG *pFormat)
{
LONG Format = CV_OUT_BINARY;
if (0 == LSTRCMPIS(
pwszColumnName,
wszPROPREQUESTDOT wszPROPREQUESTRAWREQUEST))
{
Format = CV_OUT_BASE64REQUESTHEADER;
}
else
if (0 == LSTRCMPIS(pwszColumnName, wszPROPRAWCERTIFICATE) ||
0 == LSTRCMPIS(
pwszColumnName,
wszPROPREQUESTDOT wszPROPREQUESTRAWOLDCERTIFICATE))
{
Format = CV_OUT_BASE64HEADER;
}
else
if (0 == LSTRCMPIS(pwszColumnName, wszPROPCRLRAWCRL))
{
Format = CV_OUT_BASE64X509CRLHEADER;
}
else
{
Format = CV_OUT_HEX;
}
*pFormat = Format;
return(S_OK);
}
// Build the display name for templates: "friendly name (internal name)"
HRESULT CSnapin::BuildTemplateDisplayName(
LPCWSTR pcwszFriendlyName,
LPCWSTR pcwszTemplateName,
VARIANT& varDisplayName)
{
CString strName;
strName = pcwszFriendlyName;
strName += L" (";
strName += pcwszTemplateName;
strName += L")";
V_VT(&varDisplayName) = VT_BSTR;
V_BSTR(&varDisplayName) = ::SysAllocString(strName);
if(!V_BSTR(&varDisplayName))
return E_OUTOFMEMORY;
return S_OK;
}
// copies cell to pbData, truncates if necessary. Real size passed out in pcbData
HRESULT CSnapin::GetCellContents(CertViewRowEnum* pCRowEnum, CertSvrCA* pCA, LONG idxRow, LONG idxCol, PBYTE pbData, DWORD* pcbData, BOOL fStringFmt)
{
HRESULT hr;
CComponentDataImpl* pCompData = dynamic_cast<CComponentDataImpl*>(m_pComponentData);
if (NULL == pCompData)
return E_POINTER;
VARIANT varCert;
VariantInit(&varCert);
LONG idxViewCol;
IEnumCERTVIEWROW* pRow;
IEnumCERTVIEWCOLUMN* pCol = NULL;
hr = pCRowEnum->GetRowEnum(pCA, &pRow);
if (hr != S_OK)
return hr;
do
{
hr = pCRowEnum->SetRowEnumPos(idxRow);
if (hr != S_OK)
break;
LONG lType;
LPCWSTR szColHead; // no free needed
hr = pCRowEnum->GetColumnCacheInfo(idxCol, (int*)&idxViewCol);
if (hr != S_OK)
break;
// get col enumerator object
hr = pRow->EnumCertViewColumn(&pCol);
if (hr != S_OK)
break;
hr = pCol->Skip(idxViewCol);
if (hr != S_OK)
break;
// get value there
hr = pCol->Next(&idxViewCol);
if (hr != S_OK)
break;
if (fStringFmt)
{
LONG lFormat = CV_OUT_BINARY;
VARIANT varTmp;
VariantInit(&varTmp);
hr = pCompData->GetDBSchemaEntry(idxCol, &szColHead, &lType, NULL);
if (hr != S_OK)
break;
// New: translate _some_ cols to readable strings
if (PROPTYPE_BINARY == lType)
{
hr = GetBinaryColumnFormat(szColHead, &lFormat);
if (hr != S_OK)
break;
}
hr = pCol->GetValue(lFormat, &varTmp);
if (hr != S_OK)
break;
if (0 == LSTRCMPIS(szColHead, wszPROPREQUESTDOT wszPROPREQUESTRAWARCHIVEDKEY))
{
if (VT_EMPTY != varTmp.vt)
{
varCert.bstrVal = ::SysAllocString(g_pResources->m_szYes);
varCert.vt = VT_BSTR;
}
}
else if (0 == LSTRCMPIS(szColHead, wszPROPCERTTEMPLATE))
{
LPCWSTR pcwszOID = NULL;
if (VT_BSTR == varTmp.vt)
{
// Map OID or template name to friendly name
// Try name first
HCERTTYPE hCertType;
LPWSTR *pwszCertTypeName;
hr = FindCertType(
varTmp.bstrVal,
hCertType);
if(S_OK==hr)
{
hr = CAGetCertTypeProperty(
hCertType,
CERTTYPE_PROP_FRIENDLY_NAME,
&pwszCertTypeName);
if(S_OK==hr)
{
hr = S_FALSE;
if (NULL != pwszCertTypeName)
{
if (NULL != pwszCertTypeName[0])
{
BuildTemplateDisplayName(
pwszCertTypeName[0],
varTmp.bstrVal,
varCert);
hr = S_OK;
}
}
CAFreeCertTypeProperty(
hCertType,
pwszCertTypeName);
}
CACloseCertType(hCertType);
}
// Failed to find by name, try OID
if(S_OK!=hr)
{
pcwszOID = myGetOIDName(varTmp.bstrVal);
if(!pcwszOID)
{
hr = E_OUTOFMEMORY;
break;
}
hr = S_OK;
varCert.vt = VT_BSTR;
if (EmptyString(pcwszOID))
{
varCert.bstrVal = ::SysAllocString(varTmp.bstrVal);
if(!varCert.bstrVal)
{
hr = E_OUTOFMEMORY;
break;
}
}
else
{
hr = BuildTemplateDisplayName(
pcwszOID,
varTmp.bstrVal,
varCert);
if(S_OK != hr)
break;
}
}
}
}
else if (0 == LSTRCMPIS(szColHead, wszPROPREQUESTDOT wszPROPREQUESTSTATUSCODE) ||
0 == LSTRCMPIS(szColHead, wszPROPCRLPUBLISHSTATUSCODE))
{
if (VT_I4 == varTmp.vt) // don't be empty
{
WCHAR const *pwszError = myGetErrorMessageText(varTmp.lVal, TRUE);
varCert.bstrVal = ::SysAllocString(pwszError);
varCert.vt = VT_BSTR;
if (NULL != pwszError)
{
LocalFree(const_cast<WCHAR *>(pwszError));
}
}
}
else if (0 == LSTRCMPIS(szColHead, wszPROPREQUESTDOT wszPROPREQUESTREVOKEDREASON))
{
if (VT_I4 == varTmp.vt) // don't be empty
{
// Request.Disposition
ASSERT(VT_I4 == varTmp.vt); // we'd better be looking at a dword
switch(varTmp.lVal)
{
case CRL_REASON_KEY_COMPROMISE:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_KeyCompromise);
break;
case CRL_REASON_CA_COMPROMISE:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_CaCompromise);
break;
case CRL_REASON_AFFILIATION_CHANGED:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_Affiliation);
break;
case CRL_REASON_SUPERSEDED:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_Superseded);
break;
case CRL_REASON_CESSATION_OF_OPERATION:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_Cessatation);
break;
case CRL_REASON_CERTIFICATE_HOLD:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_CertHold);
break;
case CRL_REASON_UNSPECIFIED:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_Unspecified);
break;
case CRL_REASON_REMOVE_FROM_CRL:
varCert.bstrVal = ::SysAllocString(g_pResources->m_szRevokeReason_RemoveFromCRL);
break;
default:
{
// sprint this into a buffer for display
CString cstrSprintVal;
cstrSprintVal.Format(L"%i", varTmp.lVal);
varCert.bstrVal = cstrSprintVal.AllocSysString();
break;
}
}
if (varCert.bstrVal == NULL)
{
hr = E_OUTOFMEMORY;
break;
}
varCert.vt = VT_BSTR;
}
}
else if (0 == LSTRCMPIS(szColHead, wszPROPCERTIFICATEISSUERNAMEID) ||
0 == LSTRCMPIS(szColHead, wszPROPCRLNAMEID))
{
if (VT_I4 == varTmp.vt) // don't be empty
{
// sprint this into a buffer for display
CString cstrSprintVal;
cstrSprintVal.Format(L"V%u.%u", CANAMEIDTOICERT(varTmp.lVal), CANAMEIDTOIKEY(varTmp.lVal));
varCert.bstrVal = cstrSprintVal.AllocSysString();
varCert.vt = VT_BSTR;
}
}
if (varCert.vt != VT_BSTR) // if this hasn't been converted yet
{
// default: conversion to string
// returns localized string time (even for date!)
VERIFY( MakeDisplayStrFromDBVariant(&varTmp, &varCert) ); // variant type change failed!?
InplaceStripControlChars(varCert.bstrVal);
}
VariantClear(&varTmp);
}
else
{
hr = pCol->GetValue(CV_OUT_BINARY, &varCert);
if (hr != S_OK)
break;
if (VT_EMPTY == varCert.vt)
{
hr = CERTSRV_E_PROPERTY_EMPTY;
break;
}
}
// finally, copy this value out to pb
// copy, truncate if necessary
DWORD cbTruncate = *pcbData;
if (varCert.vt == VT_BSTR)
{
*pcbData = SysStringByteLen(varCert.bstrVal) + ((fStringFmt)? sizeof(WCHAR):0);
CopyMemory(pbData, varCert.bstrVal, min(cbTruncate, *pcbData));
}
else if (varCert.vt == VT_I4)
{
*pcbData = sizeof(LONG);
if (pbData != NULL)
*(DWORD*)pbData = varCert.lVal;
}
else
{
hr = E_INVALIDARG;
break;
}
}while(0);
VariantClear(&varCert);
if (pCol)
pCol->Release();
return hr;
}