开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 查看源码
EXIT.CPP
资源名称：MSDN_VC98.zip　[点击查看]
上传用户：bangxh
上传日期：2007-01-31
资源大小：42235k
文件大小：8k
源码类别：

Windows编程

开发平台：

Visual C++

EXIT.CPP：源码内容
  1. //+--------------------------------------------------------------------------
  2. //
  3. // Microsoft Windows
  4. // Copyright (C) Microsoft Corporation, 1996-1997
  5. //
  6. // File:        exit.cpp
  7. //
  8. // Contents:    CCertExit implementation
  9. //
  10. //---------------------------------------------------------------------------
  12. #include "pch.cpp"
  13. #pragma hdrstop
  15. #include <stdio.h>
  16. #include "exit.h"
  18. #define myEXITEVENTS 
  19. EXITEVENT_CERTISSUED | 
  20. EXITEVENT_CERTPENDING | 
  21. EXITEVENT_CERTDENIED | 
  22. EXITEVENT_CERTREVOKED | 
  23. EXITEVENT_CERTRETRIEVEPENDING | 
  24. EXITEVENT_CRLISSUED | 
  25. EXITEVENT_SHUTDOWN
  27. #ifndef DBG_CERTSRV
  28. #error -- DBG_CERTSRV not defined!
  29. #endif
  30. BOOL fDebug = DBG_CERTSRV;
  32. #define wszDESCRIPTION L"Sample Exit Module"
  35. //+--------------------------------------------------------------------------
  36. // CCertExit::~CCertExit -- destructor
  37. //
  38. // free memory associated with this instance
  39. //+--------------------------------------------------------------------------
  41. CCertExit::~CCertExit()
  42. {
  43.     if (NULL != m_strConfig)
  44.     {
  45. SysFreeString(m_strConfig);
  46.     }
  47. }
  50. //+--------------------------------------------------------------------------
  51. // CCertExit::Initialize -- initialize for a CA & return interesting Event Mask
  52. //
  53. // Returns S_OK on success.
  54. //+--------------------------------------------------------------------------
  56. STDMETHODIMP
  57. CCertExit::Initialize(
  58.     /* [in] */ BSTR const strConfig,
  59.     /* [retval][out] */ LONG __RPC_FAR *pEventMask)
  60. {
  61.     HRESULT hr = S_OK;
  63.     m_strConfig = SysAllocString(strConfig);
  64.     if (NULL == m_strConfig)
  65.     {
  66. hr = E_OUTOFMEMORY;
  67.     }
  68.     else
  69.     {
  70. *pEventMask = myEXITEVENTS;
  71. printf("Exit::Initialize(%ws) ==> %xn", m_strConfig, *pEventMask);
  72.     }
  73.     return(hr);
  74. }
  77. HRESULT
  78. EnumerateExtensions(
  79.     IN ICertServerExit *pServer)
  80. {
  81.     HRESULT hr;
  82.     HRESULT hr2;
  83.     BSTR strName = NULL;
  84.     LONG ExtFlags;
  85.     VARIANT varValue;
  86.     BOOL fClose = FALSE;
  88.     VariantInit(&varValue);
  89.     hr = pServer->EnumerateExtensionsSetup(0);
  90.     if (S_OK != hr)
  91.     {
  92. if (fDebug)
  93. {
  94.     printf("Exit:EnumerateExtensionsSetup: %x", hr);
  95. }
  96. goto error;
  97.     }
  98.     fClose = TRUE;
  99.     while (TRUE)
  100.     {
  101. hr = pServer->EnumerateExtensions(&strName);
  102. if (S_OK != hr)
  103. {
  104.     if (S_FALSE == hr)
  105.     {
  106. hr = S_OK;
  107. break;
  108.     }
  109.     if (fDebug)
  110.     {
  111. printf("Exit:EnumerateExtensions: %x", hr);
  112.     }
  113.     goto error;
  114. }
  115. hr = pServer->GetCertificateExtension(
  116. strName,
  117. PROPTYPE_BINARY,
  118. &varValue);
  119. if (S_OK != hr)
  120. {
  121.     if (fDebug)
  122.     {
  123. printf("Exit:GetCertificateExtension: %x", hr);
  124.     }
  125.     goto error;
  126. }
  127. hr = pServer->GetCertificateExtensionFlags(&ExtFlags);
  128. if (S_OK != hr)
  129. {
  130.     if (fDebug)
  131.     {
  132. printf("Exit:GetCertificateExtensionFlags: %x", hr);
  133.     }
  134.     goto error;
  135. }
  136. if (fDebug)
  137. {
  138.     printf(
  139. "Exit:EnumerateExtensions(%ws, Flags=%x, %x bytes)n",
  140. strName,
  141. ExtFlags,
  142. SysStringByteLen(varValue.bstrVal));
  143. }
  144. VariantClear(&varValue);
  145.     }
  147. error:
  148.     if (fClose)
  149.     {
  150. hr2 = pServer->EnumerateExtensionsClose();
  151. if (S_OK != hr2)
  152. {
  153.     if (fDebug)
  154.     {
  155. printf("Exit:EnumerateExtensionsClose: %x", hr2);
  156.     }
  157.     if (S_OK == hr)
  158.     {
  159. hr = hr2;
  160.     }
  161.     goto error;
  162. }
  163.     }
  164.     if (NULL != strName)
  165.     {
  166. SysFreeString(strName);
  167.     }
  168.     VariantClear(&varValue);
  169.     return(hr);
  170. }
  173. HRESULT
  174. EnumerateAttributes(
  175.     IN ICertServerExit *pServer)
  176. {
  177.     HRESULT hr;
  178.     HRESULT hr2;
  179.     BSTR strName = NULL;
  180.     BOOL fClose = FALSE;
  181.     BSTR strValue = NULL;
  183.     hr = pServer->EnumerateAttributesSetup(0);
  184.     if (S_OK != hr)
  185.     {
  186. if (fDebug)
  187. {
  188.     printf("Exit:EnumerateAttributesSetup: %x", hr);
  189. }
  190. goto error;
  191.     }
  192.     fClose = TRUE;
  193.     while (TRUE)
  194.     {
  195. hr = pServer->EnumerateAttributes(&strName);
  196. if (S_OK != hr)
  197. {
  198.     if (S_FALSE == hr)
  199.     {
  200. hr = S_OK;
  201. break;
  202.     }
  203.     if (fDebug)
  204.     {
  205. printf("Exit:EnumerateAttributes: %x", hr);
  206.     }
  207.     goto error;
  208. }
  209. hr = pServer->GetRequestAttribute(strName, &strValue);
  210. if (S_OK != hr)
  211. {
  212.     if (fDebug)
  213.     {
  214. printf("Exit:GetRequestAttribute: %x", hr);
  215.     }
  216.     goto error;
  217. }
  218. if (fDebug)
  219. {
  220.     printf("Exit:EnumerateAttributes(%ws = %ws)n", strName, strValue);
  221. }
  222. if (NULL != strValue)
  223. {
  224.     SysFreeString(strValue);
  225.     strValue = NULL;
  226. }
  227.     }
  229. error:
  230.     if (fClose)
  231.     {
  232. hr2 = pServer->EnumerateAttributesClose();
  233. if (S_OK != hr2)
  234. {
  235.     if (fDebug)
  236.     {
  237. printf("Exit:EnumerateAttributesClose: %x", hr2);
  238.     }
  239.     if (S_OK == hr)
  240.     {
  241. hr = hr2;
  242.     }
  243.     goto error;
  244. }
  245.     }
  247.     if (NULL != strName)
  248.     {
  249. SysFreeString(strName);
  250.     }
  251.     if (NULL != strValue)
  252.     {
  253. SysFreeString(strValue);
  254.     }
  255.     return(hr);
  256. }
  259. HRESULT
  260. CheckRequestProperties(
  261.     IN ICertServerExit *pServer)
  262. {
  263.     HRESULT hr;
  264.     VARIANT varValue;
  265.     BSTR strName = NULL;
  267.     VariantInit(&varValue);
  269.     strName = SysAllocString(wszPROPREQUESTREQUESTID);
  270.     if (NULL == strName)
  271.     {
  272. hr = E_OUTOFMEMORY;
  273. goto error;
  274.     }
  276.     hr = pServer->GetRequestProperty(strName, PROPTYPE_LONG, &varValue);
  277.     if (S_OK != hr)
  278.     {
  279. if (fDebug)
  280. {
  281.     printf("Exit:GetRequestProperty: %x", hr);
  282. }
  283. goto error;
  284.     }
  285.     if (fDebug)
  286.     {
  287. printf(
  288.     "Exit:CheckRequestProperties(%ws = %x)n",
  289.     strName,
  290.     varValue.lVal);
  291.     }
  292.     VariantClear(&varValue);
  294. error:
  295.     if (NULL != strName)
  296.     {
  297. SysFreeString(strName);
  298.     }
  299.     return(hr);
  300. }
  303. HRESULT
  304. CheckCert(
  305.     LONG Context)
  306. {
  307.     HRESULT hr;
  308.     ICertServerExit *pServer = NULL;
  310.     hr = CoCreateInstance(
  311.     CLSID_CCertServerExit,
  312.     NULL, // pUnkOuter
  313.     CLSCTX_INPROC_SERVER,
  314.     IID_ICertServerExit,
  315.     (VOID **) &pServer);
  316.     if (S_OK != hr)
  317.     {
  318. goto exit;
  319.     }
  321.     hr = pServer->SetContext(Context);
  322.     if (S_OK != hr)
  323.     {
  324. goto exit;
  325.     }
  326.     hr = EnumerateExtensions(pServer);
  327.     if (S_OK != hr)
  328.     {
  329. goto exit;
  330.     }
  332.     hr = EnumerateAttributes(pServer);
  333.     if (S_OK != hr)
  334.     {
  335. goto exit;
  336.     }
  338.     hr = CheckRequestProperties(pServer);
  339.     if (S_OK != hr)
  340.     {
  341. goto exit;
  342.     }
  344. exit:
  345.     if (NULL != pServer)
  346.     {
  347. pServer->Release();
  348.     }
  349.     return(hr);
  350. }
  353. //+--------------------------------------------------------------------------
  354. // CCertExit::Notify -- Notify the exit module of an event
  355. //
  356. // Returns S_OK.
  357. //+--------------------------------------------------------------------------
  359. STDMETHODIMP
  360. CCertExit::Notify(
  361.     /* [in] */ LONG Event,
  362.     /* [in] */ LONG Context)
  363. {
  364.     HRESULT hr = S_OK;
  365.     char *psz = "UNKNOWN EVENT";
  367.     switch (Event)
  368.     {
  369. case EXITEVENT_CERTISSUED:
  370.     psz = "certissued";
  371.     break;
  373. case EXITEVENT_CERTPENDING:
  374.     psz = "certpending";
  375.     break;
  377. case EXITEVENT_CERTDENIED:
  378.     psz = "certdenied";
  379.     break;
  381. case EXITEVENT_CERTREVOKED:
  382.     psz = "certrevoked";
  383.     break;
  385. case EXITEVENT_CERTRETRIEVEPENDING:
  386.     psz = "retrievepending";
  387.     break;
  389. case EXITEVENT_CRLISSUED:
  390.     psz = "crlissued";
  391.     break;
  393. case EXITEVENT_SHUTDOWN:
  394.     psz = "shutdown";
  395.     break;
  396.     }
  397.     printf("Exit::Notify(%s=%x, ctx=%u)n", psz, Event, Context);
  399.     if (EXITEVENT_CERTISSUED == Event)
  400.     {
  401. hr = CheckCert(Context);
  402.     }
  403.     return(hr);
  404. }
  407. STDMETHODIMP
  408. CCertExit::GetDescription(
  409.     /* [retval][out] */ BSTR *pstrDescription)
  410. {
  411.     HRESULT hr = S_OK;
  413.     *pstrDescription = SysAllocString(wszDESCRIPTION);
  414.     if (NULL == *pstrDescription)
  415.     {
  416. hr = E_OUTOFMEMORY;
  417.     }
  418.     return(hr);
  419. }
  422. /////////////////////////////////////////////////////////////////////////////
  423. //
  425. STDMETHODIMP
  426. CCertExit::InterfaceSupportsErrorInfo(REFIID riid)
  427. {
  428.     int i;
  429.     static const IID *arr[] =
  430.     {
  431. &IID_ICertExit,
  432.     };
  434.     for (i = 0; i < sizeof(arr)/sizeof(arr[0]); i++)
  435.     {
  436. if (InlineIsEqualGUID(*arr[i],riid))
  437. {
  438.     return(S_OK);
  439. }
  440.     }
  441.     return(S_FALSE);
  442. }