html5 api activex <objiect,【ActiveX】实现安全接口

1 #include //for IObjectSafety; in ActiveX SDK

2 class CtryISafeObjectCtrl : publicCOleControl3 {4 DECLARE_DYNCREATE(CtryISafeObjectCtrl)5 //........................................................................6 //ISafeObject

7 DECLARE_INTERFACE_MAP()8 BEGIN_INTERFACE_PART(ObjSafe, IObjectSafety)9 STDMETHOD_(HRESULT, GetInterfaceSafetyOptions) (10 /*[in]*/REFIID riid,11 /*[out]*/ DWORD __RPC_FAR *pdwSupportedOptions,12 /*[out]*/ DWORD __RPC_FAR *pdwEnabledOptions13 );14

15 STDMETHOD_(HRESULT, SetInterfaceSafetyOptions) (16 /*[in]*/REFIID riid,17 /*[in]*/DWORD dwOptionSetMask,18 /*[in]*/DWORD dwEnabledOptions19 );20 END_INTERFACE_PART(ObjSafe);21 //ISafeObject22 //........................................................................

23 。。。。。24 };25 在objsafe.h头文件中有ISafeObject接口的相关定义26 3。在tryISafeObjectCtrl.cpp中ISafeObject接口的相关实现：27 //.............................................................................28 //Interface map for IObjectSafety

29 BEGIN_INTERFACE_MAP( CtryISafeObjectCtrl, COleControl )30 INTERFACE_PART(CtryISafeObjectCtrl, IID_IObjectSafety, ObjSafe)31 END_INTERFACE_MAP()32 //.............................................................................33 //IObjectSafety member functions34 //Delegate AddRef, Release, QueryInterface

35 ULONG FAR EXPORT CtryISafeObjectCtrl::XObjSafe::AddRef()36 {37 METHOD_PROLOGUE(CtryISafeObjectCtrl, ObjSafe)38 return pThis->ExternalAddRef();39 }40 ULONG FAR EXPORT CtryISafeObjectCtrl::XObjSafe::Release()41 {42 METHOD_PROLOGUE(CtryISafeObjectCtrl, ObjSafe)43 return pThis->ExternalRelease();44 }45 HRESULT FAR EXPORT CtryISafeObjectCtrl::XObjSafe::QueryInterface(46 REFIID iid, void FAR* FAR*ppvObj)47 {48 METHOD_PROLOGUE(CtryISafeObjectCtrl, ObjSafe)49 return (HRESULT)pThis->ExternalQueryInterface(&iid, ppvObj);50 }51 const DWORD dwSupportedBits =

52 INTERFACESAFE_FOR_UNTRUSTED_CALLER |

53 INTERFACESAFE_FOR_UNTRUSTED_DATA;54 const DWORD dwNotSupportedBits = ~dwSupportedBits;55 //.............................................................................56 //CStopLiteCtrl::XObjSafe::GetInterfaceSafetyOptions57 //Allows container to query what interfaces are safe for what. We're58 //optimizing significantly by ignoring which interface the caller is59 //asking for.

60 HRESULT STDMETHODCALLTYPE61 CtryISafeObjectCtrl::XObjSafe::GetInterfaceSafetyOptions(62 /*[in]*/REFIID riid,63 /*[out]*/ DWORD __RPC_FAR *pdwSupportedOptions,64 /*[out]*/ DWORD __RPC_FAR *pdwEnabledOptions)65 {66 METHOD_PROLOGUE(CtryISafeObjectCtrl, ObjSafe)67 HRESULT retval =ResultFromScode(S_OK);68 //does interface exist?

69 IUnknown FAR*punkInterface;70 retval = pThis->ExternalQueryInterface(&riid,71 (void * *)&punkInterface);72 if (retval != E_NOINTERFACE) { //interface exists

73 punkInterface->Release(); //release it--just checking!

74 }75

76 //we support both kinds of safety and have always both set,77 //regardless of interface

78 *pdwSupportedOptions = *pdwEnabledOptions =dwSupportedBits;79 return retval; //E_NOINTERFACE if QI failed

80 }81 /////////////////////////////////////////////////////////////////////////////

82 //CStopLiteCtrl::XObjSafe::SetInterfaceSafetyOptions83 //Since we're always safe, this is a no-brainer--but we do check to make84 //sure the interface requested exists and that the options we're asked to85 //set exist and are set on (we don't support unsafe mode).

86 HRESULT STDMETHODCALLTYPE87 CtryISafeObjectCtrl::XObjSafe::SetInterfaceSafetyOptions(88 /*[in]*/REFIID riid,89 /*[in]*/DWORD dwOptionSetMask,90 /*[in]*/DWORD dwEnabledOptions)91 {92 METHOD_PROLOGUE(CtryISafeObjectCtrl, ObjSafe)93

94 //does interface exist?

95 IUnknown FAR*punkInterface;96 pThis->ExternalQueryInterface(&riid, (void * *)&punkInterface);97 if (punkInterface) { //interface exists

98 punkInterface->Release(); //release it--just checking!

99 }100 else { //interface doesn't exist

101 returnResultFromScode(E_NOINTERFACE);102 }103 //can't set bits we don't support

104 if (dwOptionSetMask &dwNotSupportedBits) {105 returnResultFromScode(E_FAIL);106 }107

108 //can't set bits we do support to zero

109 dwEnabledOptions &=dwSupportedBits;110 //(we already know there are no extra bits in mask )

111 if ((dwOptionSetMask & dwEnabledOptions) !=

112 dwOptionSetMask) {113 returnResultFromScode(E_FAIL);114 }115

116 //don't need to change anything since we're always safe

117 returnResultFromScode(S_OK);118 }