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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 系统编程 > 查看源码
cobjsafe.cpp
资源名称：shell.rar　[点击查看]
上传用户：xhy777
上传日期：2007-02-14
资源大小：24088k
文件大小：5k
源码类别：

系统编程

开发平台：

Visual C++

cobjsafe.cpp：源码内容
  1. #include "stdafx.h"
  2. #pragma hdrstop
  4. // a default isafetyobject that we generally would use...  marks 
  5. // deals with IDispatch 
  8. HRESULT CObjectSafety::GetInterfaceSafetyOptions(REFIID riid, DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
  9. {
  10.     if (IsEqualIID(riid, IID_IDispatch))
  11.     {
  12.         *pdwEnabledOptions = _dwSafetyOptions;
  13.     }
  14.     else
  15.     {
  16.         ::DefaultGetSafetyOptions(riid, pdwSupportedOptions, pdwEnabledOptions);
  17.     }
  19.     return S_OK;
  20. }
  23. HRESULT CObjectSafety::SetInterfaceSafetyOptions(REFIID riid, DWORD dwOptionSetMask, DWORD dwEnabledOptions)
  24. {
  25.     if (dwOptionSetMask & ~(INTERFACESAFE_FOR_UNTRUSTED_CALLER |
  26.                             INTERFACESAFE_FOR_UNTRUSTED_DATA))
  27.     {
  28.         return E_INVALIDARG;
  29.     }
  31.     if (IsEqualIID(riid, IID_IDispatch))
  32.     {
  33.         _dwSafetyOptions = (_dwSafetyOptions & ~dwOptionSetMask) |
  34.                            (dwEnabledOptions & dwOptionSetMask);
  35.         return S_OK;
  36.     }
  37.     else
  38.     {
  39.         return ::DefaultSetSafetyOptions(riid, dwOptionSetMask, dwEnabledOptions);
  40.     }
  41. }
  45. // *** IObjectSafety
  46. //
  47. // A couple static functions called by sitemap (and webbrowser).
  48. // These are static so anyone else in this dll who has an OC
  49. // that's always safe can just call them.
  50. //
  51. // These functions say we are safe for these three interfaces we implement
  52. //  IID_IDispatch
  53. //  IID_IPersistStream
  54. //  IID_IPersistPropertyBag
  55. //
  56. // The WebBrowser OC handles IDispatch differently.
  57. //
  58. HRESULT DefaultGetSafetyOptions(REFIID riid, DWORD *pdwSupportedOptions, DWORD *pdwEnabledOptions)
  59. {
  60.     *pdwSupportedOptions = 0;
  61.     *pdwEnabledOptions = 0;
  63.     if (IsEqualIID(riid, IID_IDispatch) ||
  64.         IsEqualIID(riid, IID_IPersistStream) ||
  65.         IsEqualIID(riid, IID_IPersistStreamInit) ||
  66.         IsEqualIID(riid, IID_IPersistPropertyBag))
  67.     {
  68.         *pdwSupportedOptions = INTERFACESAFE_FOR_UNTRUSTED_CALLER | INTERFACESAFE_FOR_UNTRUSTED_DATA;
  69.         *pdwEnabledOptions = INTERFACESAFE_FOR_UNTRUSTED_CALLER | INTERFACESAFE_FOR_UNTRUSTED_DATA;
  70.     }
  72.     return S_OK;
  73. }
  75. HRESULT DefaultSetSafetyOptions(REFIID riid, DWORD dwOptionSetMask, DWORD dwEnabledOptions)
  76. {
  77.     if (dwOptionSetMask & ~(INTERFACESAFE_FOR_UNTRUSTED_CALLER |
  78.                             INTERFACESAFE_FOR_UNTRUSTED_DATA))
  79.     {
  80.         return E_INVALIDARG;
  81.     }
  83.     if (IsEqualIID(riid, IID_IDispatch) ||
  84.         IsEqualIID(riid, IID_IPersistStream) ||
  85.         IsEqualIID(riid, IID_IPersistStreamInit) ||
  86.         IsEqualIID(riid, IID_IPersistPropertyBag))
  87.     {
  88.         return S_OK;
  89.     }
  91.     return E_FAIL;
  92. }
  95. // When CWebBrowserOC is in the safe for scripting mode, we can't give out
  96. // anyone else's IDispatch that is not also safe for scripting.
  97. // This function encapsulates the basic functionality needed by both
  98. // MakeSafeScripting and MakeSafeForInitializing (which we don't use)
  99. BOOL MakeSafeFor(
  100. IUnknown *punk,                 // object to test for safety
  101. REFCATID catid,                 // category of safety
  102. REFIID riid,                    // interface on which safety is desired
  103. DWORD dwXSetMask,               // options to set
  104. DWORD dwXOptions                // options to make safe for
  105.                                     // (either INTERFACESAFE_FOR_UNTRUSTED_CALLER or
  106.                                     //  INTERFACESAFE_FOR_UNTRUSTED_DATA)
  107. )
  108. {
  109.     HRESULT hres;
  111.     // first try IObjectSafety
  112.     IObjectSafety *posafe;
  113.     if (SUCCEEDED(punk->QueryInterface(IID_IObjectSafety, (LPVOID*) &posafe)))
  114.     {
  115.         hres = posafe->SetInterfaceSafetyOptions(riid, dwXSetMask, dwXOptions);
  116.         posafe->Release();
  118.         if (SUCCEEDED(hres))
  119.             return TRUE;
  120.     }
  122.     // check the registry for "safe for scripting" component category
  124.     // we need the classid -- get it thru IPersist
  125.     CLSID clsid;
  126.     IPersist *ppersist;
  127.     hres = punk->QueryInterface(IID_IPersist, (LPVOID*) &ppersist);
  128.     if (SUCCEEDED(hres))
  129.     {
  130.         hres = ppersist->GetClassID(&clsid);
  131.         ppersist->Release();
  132.     }
  133.     if (FAILED(hres))
  134.     {
  135.         return FALSE;
  136.     }
  138.     // Create the category manager
  139.     ICatInformation *pcatinfo;
  140.     hres = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
  141.                             NULL, CLSCTX_INPROC_SERVER,
  142.                             IID_ICatInformation, (LPVOID*) &pcatinfo);
  143.     if (FAILED(hres))
  144.         return FALSE;
  146.     // Ask if the object belongs to the specified category
  147.     CATID rgcatid[1];
  148.     rgcatid[0] = catid;
  150.     hres = pcatinfo->IsClassOfCategories(clsid, 1, rgcatid, 0, NULL);
  151.     pcatinfo->Release();
  153.     return (hres==S_OK) ? TRUE : FALSE;;
  154. }
  156. HRESULT MakeSafeForScripting(IUnknown** ppDisp)
  157. {
  158.     HRESULT hres = S_OK;
  160.     if (!MakeSafeFor(*ppDisp, CATID_SafeForScripting, IID_IDispatch,
  161.                        INTERFACESAFE_FOR_UNTRUSTED_CALLER,
  162.                        INTERFACESAFE_FOR_UNTRUSTED_CALLER))
  163.     {
  164.         (*ppDisp)->Release();
  165.         *ppDisp = NULL;
  166.         hres = E_FAIL;
  167.     }
  169.     return hres;
  170. }