115 lines
3.2 KiB
C++
115 lines
3.2 KiB
C++
/*
|
|
* This file is part of the Metasploit Exploit Framework
|
|
* and is subject to the same licenses and copyrights as
|
|
* the rest of this package.
|
|
*/
|
|
#ifndef _CPASSIVEX_H
|
|
#define _CPASSIVEX_H
|
|
|
|
#include <windows.h>
|
|
#include <atlbase.h>
|
|
|
|
extern CComModule _Module;
|
|
|
|
#include <atlcom.h>
|
|
#include <ocidl.h>
|
|
|
|
#include "HttpTunnel.h"
|
|
|
|
class ATL_NO_VTABLE CPassiveX :
|
|
public CComObjectRootEx<CComMultiThreadModel>,
|
|
public CComCoClass<CPassiveX, &CLSID_PassiveX>,
|
|
public CComControl<CPassiveX>,
|
|
public IOleObjectImpl<CPassiveX>,
|
|
public IOleControlImpl<CPassiveX>,
|
|
public IOleInPlaceActiveObjectImpl<CPassiveX>,
|
|
public IOleInPlaceObjectWindowlessImpl<CPassiveX>,
|
|
public IObjectWithSiteImpl<CPassiveX>,
|
|
public IProvideClassInfo2Impl<&CLSID_PassiveX, &DIID_PassiveXEvents, &LIBID_PassiveXCOM>,
|
|
public IConnectionPointContainerImpl<CPassiveX>,
|
|
public IDispatchImpl<IPassiveX, &IID_IPassiveX, &LIBID_PassiveXCOM>,
|
|
public IConnectionPointImpl<CPassiveX, &DIID_PassiveXEvents, CComDynamicUnkArray>,
|
|
public IPersistPropertyBagImpl<CPassiveX>,
|
|
public ISupportErrorInfo
|
|
{ public:
|
|
CPassiveX();
|
|
~CPassiveX();
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_PASSIVEX)
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
BEGIN_COM_MAP(CPassiveX)
|
|
COM_INTERFACE_ENTRY(IPassiveX)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
COM_INTERFACE_ENTRY(ISupportErrorInfo)
|
|
COM_INTERFACE_ENTRY(IProvideClassInfo)
|
|
COM_INTERFACE_ENTRY(IProvideClassInfo2)
|
|
COM_INTERFACE_ENTRY(IObjectWithSite)
|
|
COM_INTERFACE_ENTRY(IOleInPlaceObjectWindowless)
|
|
COM_INTERFACE_ENTRY(IOleInPlaceObject)
|
|
COM_INTERFACE_ENTRY2(IOleWindow, IOleInPlaceObject)
|
|
COM_INTERFACE_ENTRY(IOleInPlaceActiveObject)
|
|
COM_INTERFACE_ENTRY(IOleControl)
|
|
COM_INTERFACE_ENTRY(IOleObject)
|
|
COM_INTERFACE_ENTRY(IPersistPropertyBag)
|
|
COM_INTERFACE_ENTRY(IConnectionPointContainer)
|
|
COM_INTERFACE_ENTRY_IMPL(IConnectionPointContainer)
|
|
END_COM_MAP()
|
|
|
|
// We are a singleton
|
|
DECLARE_CLASSFACTORY_SINGLETON(CPassiveX);
|
|
|
|
// Messages
|
|
BEGIN_MSG_MAP(CPassiveX)
|
|
CHAIN_MSG_MAP(CComControl<CPassiveX>)
|
|
DEFAULT_REFLECTION_HANDLER()
|
|
END_MSG_MAP()
|
|
|
|
// Connections
|
|
BEGIN_CONNECTION_POINT_MAP(CPassiveX)
|
|
CONNECTION_POINT_ENTRY(DIID_PassiveXEvents)
|
|
END_CONNECTION_POINT_MAP()
|
|
|
|
// Properties
|
|
BEGIN_PROPERTY_MAP(CPassiveX)
|
|
PROP_ENTRY("HttpHost", PASSIVEX_PROPERTY_HTTP_HOST, CLSID_NULL)
|
|
PROP_ENTRY("HttpPort", PASSIVEX_PROPERTY_HTTP_PORT, CLSID_NULL)
|
|
PROP_ENTRY("DownloadSecondStage", PASSIVEX_PROPERTY_DOWNLOAD_SECOND_STAGE, CLSID_NULL)
|
|
END_PROPERTY_MAP()
|
|
|
|
// ISupportErrorInfo
|
|
STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid);
|
|
|
|
// IPassiveX
|
|
STDMETHOD(get_HttpHost)(BSTR *Host);
|
|
STDMETHOD(put_HttpHost)(BSTR Host);
|
|
STDMETHOD(get_HttpPort)(ULONG *Port);
|
|
STDMETHOD(put_HttpPort)(ULONG Port);
|
|
STDMETHOD(get_DownloadSecondStage)(ULONG *Port);
|
|
STDMETHOD(put_DownloadSecondStage)(ULONG Port);
|
|
|
|
#ifdef PXDEBUG
|
|
// Debug logging
|
|
static VOID Log(LPCTSTR fmt, ...);
|
|
#else
|
|
static VOID Log(LPCTSTR fmt, ...) { }
|
|
#endif
|
|
protected:
|
|
|
|
VOID Initialize();
|
|
|
|
VOID ResetExplorerZoneRestrictions();
|
|
|
|
/**************
|
|
* Attributes *
|
|
**************/
|
|
|
|
// Properties
|
|
CComBSTR PropHttpHost;
|
|
ULONG PropHttpPort;
|
|
|
|
// Tunnel
|
|
HttpTunnel Tunnel;
|
|
};
|
|
|
|
#endif
|