301 lines
8.8 KiB
C++
301 lines
8.8 KiB
C++
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1997-2002.
|
|
//
|
|
// File: cookie.h
|
|
//
|
|
// Contents:
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#ifndef __COOKIE_H_INCLUDED__
|
|
#define __COOKIE_H_INCLUDED__
|
|
|
|
extern HINSTANCE g_hInstanceSave; // Instance handle of the DLL (initialized during CCertMgrComponent::Initialize)
|
|
|
|
|
|
#include "nodetype.h"
|
|
#pragma warning(push,3)
|
|
#include <efsstruc.h>
|
|
#pragma warning(pop)
|
|
|
|
// Name of Encrypting File System store
|
|
#define ACRS_SYSTEM_STORE_NAME L"ACRS"
|
|
#define EFS_SYSTEM_STORE_NAME L"EFS"
|
|
#define TRUST_SYSTEM_STORE_NAME L"Trust"
|
|
#define ROOT_SYSTEM_STORE_NAME L"Root"
|
|
#define MY_SYSTEM_STORE_NAME L"MY"
|
|
#define CA_SYSTEM_STORE_NAME L"CA"
|
|
#define USERDS_SYSTEM_STORE_NAME L"UserDS"
|
|
#define REQUEST_SYSTEM_STORE_NAME L"REQUEST"
|
|
#define SAFER_TRUSTED_PUBLISHER_STORE_NAME L"TrustedPublisher"
|
|
#define SAFER_DISALLOWED_STORE_NAME L"Disallowed"
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// cookie
|
|
|
|
//REVIEW -- using pointers to ID's is necessary because some compilers don't like
|
|
//references as template arguments.
|
|
|
|
|
|
class CCertificate; // forward declaration
|
|
|
|
class CCertMgrCookie : public CCookie,
|
|
public CStoresMachineName,
|
|
public CBaseCookieBlock
|
|
{
|
|
public:
|
|
CCertMgrCookie (CertificateManagerObjectType objecttype,
|
|
LPCWSTR lpcszMachineName = 0,
|
|
LPCWSTR objectName = 0);
|
|
|
|
virtual ~CCertMgrCookie ();
|
|
|
|
// returns <0, 0 or >0
|
|
virtual HRESULT CompareSimilarCookies( CCookie* pOtherCookie, int* pnResult );
|
|
|
|
// CBaseCookieBlock
|
|
virtual CCookie* QueryBaseCookie(int i);
|
|
virtual int QueryNumCookies();
|
|
|
|
public:
|
|
bool IsSelected () const;
|
|
void SetSelected (bool bIsSelected);
|
|
LPRESULTDATA m_resultDataID;
|
|
virtual void Refresh ();
|
|
virtual HRESULT Commit ();
|
|
CString GetServiceName () const;
|
|
void SetServiceName (CString &szManagedService);
|
|
LPCWSTR GetObjectName ();
|
|
const CertificateManagerObjectType m_objecttype;
|
|
UINT IncrementOpenPageCount ();
|
|
UINT DecrementOpenPageCount ();
|
|
bool HasOpenPropertyPages () const;
|
|
|
|
private:
|
|
bool m_bIsSelected;
|
|
CString m_objectName;
|
|
UINT m_nOpenPageCount;
|
|
};
|
|
|
|
typedef enum {
|
|
NO_SPECIAL_TYPE = 0,
|
|
MY_STORE,
|
|
CA_STORE,
|
|
ROOT_STORE,
|
|
TRUST_STORE,
|
|
USERDS_STORE,
|
|
ACRS_STORE,
|
|
EFS_STORE,
|
|
REQUEST_STORE,
|
|
SAFER_TRUSTED_PUBLISHER_STORE,
|
|
SAFER_DISALLOWED_STORE
|
|
} SPECIAL_STORE_TYPE;
|
|
SPECIAL_STORE_TYPE GetSpecialStoreType(PCWSTR pwszStoreName);
|
|
|
|
SPECIAL_STORE_TYPE StoreNameToType (const CString& szStoreName);
|
|
|
|
class CCTL; // forward declaration
|
|
class CCertStore : public CCertMgrCookie
|
|
{
|
|
friend CCTL;
|
|
public:
|
|
virtual bool IsNullEFSPolicy()
|
|
{
|
|
return false;
|
|
}
|
|
void IncrementCertCount ();
|
|
void Unlock ();
|
|
void Lock ();
|
|
void InvalidateCertCount();
|
|
BOOL AddEncodedCTL (DWORD dwMsgAndCertEncodingType,
|
|
const BYTE* pbCtlEncoded,
|
|
DWORD cbCtlEncoded,
|
|
DWORD dwAddDisposition,
|
|
PCCTL_CONTEXT* ppCtlContext);
|
|
virtual void Close (bool bForceClose = false);
|
|
void SetDirty();
|
|
HRESULT AddStoreToCollection(CCertStore& siblingStore,
|
|
DWORD dwUpdateFlags = 0,
|
|
DWORD dwPriority = 0);
|
|
HRESULT AddStoreToCollection(HCERTSTORE hSiblingStore,
|
|
DWORD dwUpdateFlags = 0,
|
|
DWORD dwPriority = 0);
|
|
int GetCTLCount ();
|
|
inline DWORD GetLocation () { return m_dwLocation;}
|
|
HRESULT Resync ();
|
|
inline bool IsOpen()
|
|
{
|
|
// If m_hCertStore is 0, then this store hasn't been used for anything
|
|
if ( !m_hCertStore )
|
|
return false;
|
|
else
|
|
return true;
|
|
}
|
|
|
|
PCCRL_CONTEXT GetCRL (
|
|
PCCERT_CONTEXT pIssuerContext,
|
|
PCCRL_CONTEXT pPrevCrlContext,
|
|
DWORD* pdwFlags);
|
|
bool AddCTLContext (PCCTL_CONTEXT pCtlContext);
|
|
bool AddCRLContext (PCCRL_CONTEXT pCrlContext);
|
|
PCCERT_CONTEXT FindCertificate (
|
|
DWORD dwFindFlags,
|
|
DWORD dwFindType,
|
|
const void *pvFindPara,
|
|
PCCERT_CONTEXT pPrevCertContext);
|
|
PCCRL_CONTEXT EnumCRLs (PCCRL_CONTEXT pPrevCrlContext);
|
|
PCCTL_CONTEXT EnumCTLs (PCCTL_CONTEXT pPrevCtlContext);
|
|
virtual PCCERT_CONTEXT EnumCertificates (PCCERT_CONTEXT pPrevCertContext);
|
|
CCertificate* GetSubjectCertificate (PCERT_INFO pCertId);
|
|
BOOL operator==(CCertStore&);
|
|
int GetCertCount ();
|
|
virtual HRESULT AddCertificateContext (
|
|
PCCERT_CONTEXT pContext,
|
|
LPCONSOLE pConsole,
|
|
bool bDeletePrivateKey,
|
|
PCCERT_CONTEXT* ppNewCertContext = 0,
|
|
bool* pbCertWasReplaced = false);
|
|
inline virtual void AddRef()
|
|
{
|
|
ASSERT (CERTMGR_LOG_STORE_GPE == m_objecttype ||
|
|
CERTMGR_LOG_STORE_RSOP == m_objecttype ||
|
|
CERTMGR_LOG_STORE == m_objecttype ||
|
|
CERTMGR_PHYS_STORE == m_objecttype);
|
|
CCertMgrCookie::AddRef ();
|
|
}
|
|
|
|
inline virtual void Release ()
|
|
{
|
|
ASSERT (CERTMGR_LOG_STORE_GPE == m_objecttype ||
|
|
CERTMGR_LOG_STORE_RSOP == m_objecttype ||
|
|
CERTMGR_LOG_STORE == m_objecttype ||
|
|
CERTMGR_PHYS_STORE == m_objecttype);
|
|
CCertMgrCookie::Release ();
|
|
}
|
|
|
|
CString GetStoreName () const;
|
|
LPCWSTR GetLocalizedName();
|
|
bool ContainsCTLs ();
|
|
bool ContainsCRLs ();
|
|
bool ContainsCertificates ();
|
|
virtual HRESULT Commit ();
|
|
virtual bool IsReadOnly ();
|
|
inline const SPECIAL_STORE_TYPE GetStoreType () const
|
|
{
|
|
ASSERT (CERTMGR_LOG_STORE_GPE == m_objecttype ||
|
|
CERTMGR_LOG_STORE_RSOP == m_objecttype ||
|
|
CERTMGR_LOG_STORE == m_objecttype ||
|
|
CERTMGR_PHYS_STORE == m_objecttype);
|
|
return m_storeType;
|
|
}
|
|
|
|
CCertStore (CertificateManagerObjectType objecttype,
|
|
LPCSTR pszStoreProv,
|
|
DWORD dwFlags,
|
|
LPCWSTR lpcszMachineName,
|
|
LPCWSTR objectName,
|
|
const CString & pcszLogStoreName,
|
|
const CString & pcszPhysStoreName,
|
|
const SPECIAL_STORE_TYPE storeType,
|
|
const DWORD dwLocation,
|
|
IConsole* pConsole,
|
|
bool fIsComputerType = false);
|
|
virtual ~CCertStore ();
|
|
virtual HCERTSTORE GetStoreHandle (BOOL bSilent = FALSE, HRESULT* phr = 0);
|
|
virtual bool CanContain (CertificateManagerObjectType /*nodeType*/)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
virtual bool IsMachineStore()
|
|
{
|
|
return false;
|
|
}
|
|
|
|
virtual bool IsComputerType ()
|
|
{
|
|
return m_fIsComputerType;
|
|
}
|
|
|
|
virtual void SetAdding ()
|
|
{
|
|
}
|
|
virtual void SetDeleting ()
|
|
{
|
|
}
|
|
|
|
protected:
|
|
virtual void FinalCommit();
|
|
HRESULT RetrieveBLOBFromFile (LPCWSTR pwszFileName, DWORD *pcb, BYTE **ppb);
|
|
|
|
bool m_fCertCountValid;
|
|
bool m_bUnableToOpenMsgDisplayed;
|
|
LPCSTR m_storeProvider;
|
|
DWORD m_dwFlags;
|
|
CString m_pcszStoreName;
|
|
bool m_bReadOnly;
|
|
HCERTSTORE m_hCertStore;
|
|
IConsole* m_pConsole;
|
|
bool m_bDirty;
|
|
bool m_fIsComputerType;
|
|
|
|
private:
|
|
int m_nCertCount;
|
|
bool m_fReadOnlyFlagChecked;
|
|
const DWORD m_dwLocation;
|
|
CString m_localizedName;
|
|
const SPECIAL_STORE_TYPE m_storeType;
|
|
int m_nLockCnt; // test
|
|
};
|
|
|
|
|
|
class CContainerCookie : public CCertMgrCookie
|
|
{
|
|
public:
|
|
virtual HRESULT Commit ();
|
|
inline const SPECIAL_STORE_TYPE GetStoreType () const
|
|
{
|
|
ASSERT (CERTMGR_CRL_CONTAINER == m_objecttype ||
|
|
CERTMGR_CTL_CONTAINER == m_objecttype ||
|
|
CERTMGR_CERT_CONTAINER == m_objecttype);
|
|
return m_rCertStore.GetStoreType ();
|
|
}
|
|
|
|
CContainerCookie (CCertStore& rStore,
|
|
CertificateManagerObjectType objecttype,
|
|
LPCWSTR lpcszMachineName,
|
|
LPCWSTR objectName);
|
|
virtual ~CContainerCookie ();
|
|
CCertStore& GetCertStore () const;
|
|
private:
|
|
CCertStore& m_rCertStore;
|
|
};
|
|
|
|
|
|
class CUsageCookie : public CCertMgrCookie
|
|
{
|
|
public:
|
|
int GetCertCount () const;
|
|
void SetCertCount (int nCertCount);
|
|
CUsageCookie (
|
|
CertificateManagerObjectType objecttype,
|
|
LPCWSTR lpcszMachineName,
|
|
LPCWSTR objectName);
|
|
void AddOID (LPCSTR pszOID);
|
|
virtual ~CUsageCookie ();
|
|
LPSTR GetFirstOID ();
|
|
LPSTR GetNextOID ();
|
|
int GetOIDCount () const;
|
|
|
|
private:
|
|
int m_nCertCount;
|
|
CTypedPtrList<CPtrList, LPSTR> m_OIDList;
|
|
POSITION m_OIDListPos;
|
|
};
|
|
|
|
#endif // ~__COOKIE_H_INCLUDED__
|