ActiveX控件实现IobjectSafety接口

如果没有实现IobjectSafety接口浏览器会提示交互不安全等信息

 

打开工程名Ctl.h
搜索    class C工程名Ctrl : public COleControl
在其上面添加   #include <objsafe.h>
搜索    DECLARE_DYNCREATE(C工程名Ctrl)


下面添加

    DECLARE_INTERFACE_MAP()

    BEGIN_INTERFACE_PART(ObjSafe, IObjectSafety)
        STDMETHOD_(HRESULT, GetInterfaceSafetyOptions) (
        REFIID riid,
        DWORD __RPC_FAR *pdwSupportedOptions,
        DWORD __RPC_FAR *pdwEnabledOptions
        );

        STDMETHOD_(HRESULT, SetInterfaceSafetyOptions) (
            REFIID riid,
            DWORD dwOptionSetMask,
            DWORD dwEnabledOptions
            );
    END_INTERFACE_PART(ObjSafe);


打开工程名Ctl.cpp
搜索    BOOL C工程名Ctrl::C工程名CtrlFactory::UpdateRegistry(BOOL bRegister)

/
// Interface map for IObjectSafety

BEGIN_INTERFACE_MAP(CDxrOcxCtrl, COleControl)
    INTERFACE_PART(CDxrOcxCtrl, IID_IObjectSafety, ObjSafe)
END_INTERFACE_MAP()

/
// IObjectSafety member functions

// Delegate AddRef, Release, QueryInterface

ULONG FAR EXPORT CDxrOcxCtrl::XObjSafe::AddRef()
{
    METHOD_PROLOGUE(CDxrOcxCtrl, ObjSafe)
        return pThis->ExternalAddRef();
}

ULONG FAR EXPORT CDxrOcxCtrl::XObjSafe::Release()
{
    METHOD_PROLOGUE(CDxrOcxCtrl, ObjSafe)
        return pThis->ExternalRelease();
}

HRESULT FAR EXPORT CDxrOcxCtrl::XObjSafe::QueryInterface(
    REFIID iid, void FAR* FAR* ppvObj)
{
    METHOD_PROLOGUE(CDxrOcxCtrl, ObjSafe)
        return (HRESULT)pThis->ExternalQueryInterface(&iid, ppvObj);
}

const DWORD dwSupportedBits =
INTERFACESAFE_FOR_UNTRUSTED_CALLER |
INTERFACESAFE_FOR_UNTRUSTED_DATA;
const DWORD dwNotSupportedBits = ~dwSupportedBits;

/
// CStopLiteCtrl::XObjSafe::GetInterfaceSafetyOptions
// Allows container to query what interfaces are safe for what. We're
// optimizing significantly by ignoring which interface the caller is
// asking for.
HRESULT STDMETHODCALLTYPE
CDxrOcxCtrl::XObjSafe::GetInterfaceSafetyOptions(
    REFIID riid,
    DWORD __RPC_FAR *pdwSupportedOptions,
    DWORD __RPC_FAR *pdwEnabledOptions)
{
    METHOD_PROLOGUE(CDxrOcxCtrl, ObjSafe)

    HRESULT retval = ResultFromScode(S_OK);

    // does interface exist?
    IUnknown FAR* punkInterface;
    retval = pThis->ExternalQueryInterface(&riid,
        (void * *)&punkInterface);
    if (retval != E_NOINTERFACE) { // interface exists
        punkInterface->Release(); // release it--just checking!
    }

    // we support both kinds of safety and have always both set,
    // regardless of interface
    *pdwSupportedOptions = *pdwEnabledOptions = dwSupportedBits;

    return retval; // E_NOINTERFACE if QI failed
}

/
// CStopLiteCtrl::XObjSafe::SetInterfaceSafetyOptions
// Since we're always safe, this is a no-brainer--but we do check to make
// sure the interface requested exists and that the options we're asked to
// set exist and are set on (we don't support unsafe mode).
HRESULT STDMETHODCALLTYPE
CDxrOcxCtrl::XObjSafe::SetInterfaceSafetyOptions(
    REFIID riid,
    DWORD dwOptionSetMask,
    DWORD dwEnabledOptions)
{
    METHOD_PROLOGUE(CDxrOcxCtrl, ObjSafe)

    // does interface exist?
    IUnknown FAR* punkInterface;
    pThis->ExternalQueryInterface(&riid, (void * *)&punkInterface);
    if (punkInterface) { // interface exists
        punkInterface->Release(); // release it--just checking!
    }
    else { // interface doesn't exist
        return ResultFromScode(E_NOINTERFACE);
    }

    // can't set bits we don't support
    if (dwOptionSetMask & dwNotSupportedBits) {
        return ResultFromScode(E_FAIL);
    }

    // can't set bits we do support to zero
    dwEnabledOptions &= dwSupportedBits;
    // (we already know there are no extra bits in mask )
    if ((dwOptionSetMask & dwEnabledOptions) !=
        dwOptionSetMask) {
            return ResultFromScode(E_FAIL);
    }       

    // don't need to change anything since we're always safe
    return ResultFromScode(S_OK);
}