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.
165 lines
4.4 KiB
165 lines
4.4 KiB
/*++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
|
|
Microsoft Windows, Copyright (C) Microsoft Corporation, 2000
|
|
|
|
File: Store.h
|
|
|
|
Content: Declaration of CStore.
|
|
|
|
History: 11-15-99 dsie created
|
|
|
|
------------------------------------------------------------------------------*/
|
|
|
|
#ifndef __STORE_H_
|
|
#define __STORE_H_
|
|
|
|
#include "Resource.h"
|
|
#include "Error.h"
|
|
#include "Lock.h"
|
|
#include "Debug.h"
|
|
#include "DialogUI.h"
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CStore
|
|
//
|
|
|
|
class ATL_NO_VTABLE CStore :
|
|
public ICertStore,
|
|
public CComObjectRootEx<CComMultiThreadModel>,
|
|
public CComCoClass<CStore, &CLSID_Store>,
|
|
public ICAPICOMError<CStore, &IID_IStore>,
|
|
public IPromptUser<CStore>,
|
|
public IDispatchImpl<IStore2, &IID_IStore2, &LIBID_CAPICOM,
|
|
CAPICOM_MAJOR_VERSION, CAPICOM_MINOR_VERSION>,
|
|
public IObjectSafetyImpl<CStore, INTERFACESAFE_FOR_UNTRUSTED_CALLER |
|
|
INTERFACESAFE_FOR_UNTRUSTED_DATA>
|
|
{
|
|
public:
|
|
CStore()
|
|
{
|
|
}
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_STORE)
|
|
|
|
DECLARE_GET_CONTROLLING_UNKNOWN()
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CStore)
|
|
COM_INTERFACE_ENTRY(IStore)
|
|
COM_INTERFACE_ENTRY(IStore2)
|
|
COM_INTERFACE_ENTRY(ICertStore)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
COM_INTERFACE_ENTRY(IObjectSafety)
|
|
COM_INTERFACE_ENTRY(IObjectWithSite)
|
|
COM_INTERFACE_ENTRY(ISupportErrorInfo)
|
|
END_COM_MAP()
|
|
|
|
BEGIN_CATEGORY_MAP(CStore)
|
|
IMPLEMENTED_CATEGORY(CATID_SafeForScripting)
|
|
IMPLEMENTED_CATEGORY(CATID_SafeForInitializing)
|
|
END_CATEGORY_MAP()
|
|
|
|
HRESULT FinalConstruct()
|
|
{
|
|
HRESULT hr;
|
|
|
|
if (FAILED(hr = m_Lock.Initialized()))
|
|
{
|
|
DebugTrace("Error [%#x]: Critical section could not be created for Store object.\n", hr);
|
|
return hr;
|
|
}
|
|
|
|
m_hCertStore = NULL;
|
|
m_StoreLocation = CAPICOM_CURRENT_USER_STORE;
|
|
m_bIsProtected = TRUE;
|
|
m_cKeyProvInfo = 0;
|
|
m_rgpKeyProvInfo = NULL;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
void FinalRelease()
|
|
{
|
|
if (m_hCertStore)
|
|
{
|
|
Close();
|
|
}
|
|
}
|
|
|
|
//
|
|
// IStore
|
|
//
|
|
public:
|
|
STDMETHOD(Import)
|
|
(/*[in]*/ BSTR EncodedStore);
|
|
|
|
STDMETHOD(Export)
|
|
(/*[in, defaultvalue(CAPICOM_STORE_SAVE_AS_SERIALIZED)]*/ CAPICOM_STORE_SAVE_AS_TYPE SaveAs,
|
|
/*[in, defaultvalue(CAPICOM_BASE64_ENCODE)]*/ CAPICOM_ENCODING_TYPE EncodingType,
|
|
/*[out, retval]*/ BSTR * pVal);
|
|
|
|
STDMETHOD(Remove)
|
|
(/*[in]*/ ICertificate * pVal);
|
|
|
|
STDMETHOD(Add)
|
|
(/*[in]*/ ICertificate * pVal);
|
|
|
|
STDMETHOD(Open)
|
|
(/*[in]*/ CAPICOM_STORE_LOCATION StoreLocation,
|
|
/*[in, defaultvalue("My")]*/ BSTR StoreName,
|
|
/*[in, defaultvalue(CAPICOM_STORE_OPEN_MAXIMUM_ALLOWED)]*/ CAPICOM_STORE_OPEN_MODE OpenMode);
|
|
|
|
STDMETHOD(get_Certificates)
|
|
(/*[out, retval]*/ ICertificates ** pVal);
|
|
|
|
//
|
|
// v2.0
|
|
//
|
|
STDMETHOD(Load)
|
|
(/*[in]*/ BSTR FileName,
|
|
/*[in, defaultvalue("")]*/ BSTR Password,
|
|
/*[in, defaultvalue(CAPICOM_KEY_STORAGE_DEFAULT)]*/ CAPICOM_KEY_STORAGE_FLAG KeyStorageFlag);
|
|
|
|
//
|
|
// Custom interfaces.
|
|
//
|
|
STDMETHOD(get_StoreHandle)
|
|
(/*[out, retval]*/ long * phCertStore);
|
|
|
|
STDMETHOD(put_StoreHandle)
|
|
(/*[in]*/ long hCertStore);
|
|
|
|
STDMETHOD(get_StoreLocation)
|
|
(/*[out, retval]*/ CAPICOM_STORE_LOCATION * pStoreLocation);
|
|
|
|
STDMETHOD(put_StoreLocation)
|
|
(/*[in]*/ CAPICOM_STORE_LOCATION StoreLocation);
|
|
|
|
STDMETHOD(CloseHandle)
|
|
(/*[in]*/ long hCertStore);
|
|
|
|
//
|
|
// None COM functions.
|
|
//
|
|
STDMETHOD(ImportCertObject)
|
|
(DWORD dwObjectType,
|
|
LPVOID pvObject,
|
|
BOOL bAllowPfx,
|
|
LPWSTR pwszPassword,
|
|
CAPICOM_KEY_STORAGE_FLAG KeyStorageFlag);
|
|
|
|
STDMETHOD(Close)
|
|
(void);
|
|
|
|
private:
|
|
CLock m_Lock;
|
|
HCERTSTORE m_hCertStore;
|
|
CAPICOM_STORE_LOCATION m_StoreLocation;
|
|
BOOL m_bIsProtected;
|
|
DWORD m_cKeyProvInfo;
|
|
PCRYPT_KEY_PROV_INFO * m_rgpKeyProvInfo;
|
|
};
|
|
|
|
#endif //__STORE_H_
|