BHO (Browser Helper Objects)

//MainModule.cpp — главный модуль

#include <windows.h>
#include "IEClassFactory.h"

#include "debug.h"

extern WORD g_nSrvRefCount=0;


bool CanUnload = true;

BOOL WINAPI DllMain(HINSTANCE instance, DWORD fdwReason, LPVOID lpvReserved)
{
return TRUE;
}

STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
{
debug("DllGetClassObject");

if (!ppv)
return E_INVALIDARG;

*ppv=NULL;
HRESULT hr=E_FAIL;

CIEClassFactory *pObj=new CIEClassFactory;
if (!pObj)
return E_OUTOFMEMORY;

hr=pObj->QueryInterface(riid, ppv);

if (FAILED(hr))
{
debug("getclass error");
delete pObj;
return CLASS_E_CLASSNOTAVAILABLE;
}
return S_OK;
}

STDAPI DllCanUnloadNow(void)
{
debug("DllCanUnloadNow");
if(g_nSrvRefCount)
return S_FALSE;
else
return S_OK;
}
//////////////////-----------------------------------------

//IEClassFactory.h
#include "unknwn.h"

class CIEClassFactory: public IClassFactory
{
protected:
long m_lRef;
public:
CIEClassFactory();
~CIEClassFactory();

HRESULT STDMETHODCALLTYPE QueryInterface (REFIID riid, void** ppv);
ULONG STDMETHODCALLTYPE AddRef();
ULONG STDMETHODCALLTYPE Release();
HRESULT STDMETHODCALLTYPE CreateInstance(LPUNKNOWN pUnk, REFIID riid, void** ppv);
HRESULT STDMETHODCALLTYPE LockServer(BOOL fLock) ;
};
//////////////////-----------------------------------------

//IEClassFactory.cpp
#include "IEClassFactory.h"
#include "debug.h"
#include "IEClass.h"

extern WORD g_nSrvRefCount;

CIEClassFactory::CIEClassFactory()
{
m_lRef = 0;
}

CIEClassFactory::~CIEClassFactory()
{
debug("IEClassFactory ~");

}

HRESULT STDMETHODCALLTYPE CIEClassFactory::QueryInterface (REFIID riid, void** ppv)
{
debug("IEClassFactory QueryInterface");
if (!ppv)
return E_INVALIDARG;

*ppv=NULL;

if (riid == IID_IUnknown)
*ppv=(IUnknown*)this;
else if (riid == IID_IClassFactory)
*ppv=(IClassFactory*)this;

if (*ppv)
{
debug("IEClassFactory QueryInterface AddRef");
AddRef();
return S_OK;
}

debug("IEClassFactory QueryInterface error");
return E_NOINTERFACE;
}

ULONG STDMETHODCALLTYPE CIEClassFactory::AddRef()
{
debug("IEClassFactory AddRef");
g_nSrvRefCount++;
return ++m_lRef;
}

ULONG STDMETHODCALLTYPE CIEClassFactory::Release()
{
debug("IEClassFactory Release");
g_nSrvRefCount--;
if (!--m_lRef)
{
delete this;
return 0;
}
else return m_lRef;
}

HRESULT STDMETHODCALLTYPE CIEClassFactory::CreateInstance(LPUNKNOWN pUnk, REFIID riid, void** ppv)
{
debug("IEClassFactory CreateInstance");

if (!ppv)
{
debug("IEClassFactory CreateInstance E_INVALIDARG");
return E_INVALIDARG;
}
if (pUnk)
{
debug("IEClassFactory CreateInstance CLASS_E_NOAGGREGATION");
return CLASS_E_NOAGGREGATION;
}

*ppv=NULL;
HRESULT hr=E_NOINTERFACE;

CIEClass *pExt=new CIEClass;
if (!pExt)
return E_OUTOFMEMORY;

hr=pExt->QueryInterface(riid, ppv);
if (SUCCEEDED(hr))
{
debug("IEClassFactory CreateInstance OK");
return S_OK;
}

debug("IEClassFactory CreateInstance delete");
delete pExt;
return hr;
}

HRESULT STDMETHODCALLTYPE CIEClassFactory::LockServer(BOOL fLock)
{
debug("IEClassFactory LockServer");
if (fLock)
g_nSrvRefCount++;
else
g_nSrvRefCount--;
return S_OK;

}
//-----------------------------------------------

//IEClass.h
#include "exdisp.h"

class CIEClass: public IObjectWithSite, public IDispatch
{

int m_nRefCount;
public:
CIEClass();
~CIEClass();
HRESULT Connect();

ULONG STDMETHODCALLTYPE AddRef();
ULONG STDMETHODCALLTYPE Release();
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID iid, //Identifier of the requested interface
void ** ppvObject);

HRESULT STDMETHODCALLTYPE GetSite(REFIID riid, void** ppvSite);
HRESULT STDMETHODCALLTYPE SetSite(IUnknown *pUnkSite);

HRESULT STDMETHODCALLTYPE GetTypeInfoCount(unsigned int FAR* pctinfo);
HRESULT STDMETHODCALLTYPE GetTypeInfo(unsigned int iTInfo, LCID lcid, ITypeInfo FAR* FAR* ppTInfo);
HRESULT STDMETHODCALLTYPE GetIDsOfNames(
REFIID riid,
OLECHAR FAR* FAR* rgszNames,
unsigned int cNames,
LCID lcid,
DISPID FAR* rgDispId
);
HRESULT STDMETHODCALLTYPE Invoke(
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS FAR* pDispParams,
VARIANT FAR* pVarResult,
EXCEPINFO FAR* pExcepInfo,
unsigned int FAR* puArgErr
);



private:
IDispatch* document_;
IWebBrowser2* webBrowser;
IConnectionPointContainer* connectionPointContainer;
unsigned long cookie;


};
//------------------------------------------------

//IEClass.cpp
#include "IEClass.h"
#include "debug.h"

extern WORD g_nSrvRefCount;

CIEClass::CIEClass()
{
m_nRefCount = 0;
}

CIEClass::~CIEClass()
{
debug("CIEClass ~");

}

STDMETHODIMP CIEClass::GetSite(REFIID riid, void** ppvSite)
{
debug("CIEClass GetSite");
return 0;
}

STDMETHODIMP CIEClass::SetSite(IUnknown *pUnkSite)
{
debug("CIEClass SetSite");
//до вызова дело даже не доходит
return S_OK;
}

ULONG STDMETHODCALLTYPE CIEClass::AddRef()
{
debug("CIEClass AddRef");
g_nSrvRefCount++;
return ++m_nRefCount;
}


ULONG STDMETHODCALLTYPE CIEClass::Release()
{
debug("CIEClass Release");
g_nSrvRefCount--;
if (!--m_nRefCount)
{
debug("CIEClass Release delete");
delete this;
return 0;
}
else return m_nRefCount;
}

STDMETHODIMP CIEClass::QueryInterface(REFIID iid, void ** ppvObject)
{
debug("CIEClass QueryInterface");

if(IID_IWebBrowser2 == iid)
{
debug("IID_IWebBrowser2");
*ppvObject = (IWebBrowser2*)this;
}
else if(IID_IObjectWithSite == iid)
{
debug("IID_IObjectWithSite");
*ppvObject = (IObjectWithSite*)this;
} else
if(IID_IClassFactory == iid)
{
debug("IID_IClassFactory");
*ppvObject = (IClassFactory*)this;
}
else
{
debug("E_NOINTERFACE");
return E_NOINTERFACE;
}

return 0;
}

STDMETHODIMP CIEClass::GetTypeInfoCount(unsigned int FAR* pctinfo)
{
debug("CIEClass GetTypeInfoCount");
return 0;
}

STDMETHODIMP CIEClass::GetTypeInfo(unsigned int iTInfo, LCID lcid, ITypeInfo FAR* FAR* ppTInfo)
{
debug("CIEClass GetTypeInfo");
return 0;
}

STDMETHODIMP CIEClass::GetIDsOfNames(
REFIID riid,
OLECHAR FAR* FAR* rgszNames,
unsigned int cNames,
LCID lcid,
DISPID FAR* rgDispId
)
{
debug("CIEClass GetIDsOfNames");
return 0;
}

STDMETHODIMP CIEClass::Invoke(
DISPID dispIdMember,
REFIID riid,
LCID lcid,
WORD wFlags,
DISPPARAMS FAR* pDispParams,
VARIANT FAR* pVarResult,
EXCEPINFO FAR* pExcepInfo,
unsigned int FAR* puArgErr
)
{
debug("CIEClass Invoke");
return 0;
}