207 lines
4.1 KiB
C++
207 lines
4.1 KiB
C++
extern "C" {
|
|
#include <shimdb.h>
|
|
}
|
|
|
|
|
|
#ifndef __BLOCKEDDRIVERS_H_
|
|
#define __BLOCKEDDRIVERS_H_
|
|
|
|
#include "resource.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CBlockedDrivers
|
|
class ATL_NO_VTABLE CBlockedDrivers :
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
public CComCoClass<CBlockedDrivers, &CLSID_BlockedDrivers>,
|
|
public IDispatchImpl<IBlockedDrivers, &IID_IBlockedDrivers, &LIBID_BLOCKDRIVERLib>
|
|
{
|
|
protected:
|
|
LPGUID m_guidIDs;
|
|
|
|
|
|
public:
|
|
CBlockedDrivers()
|
|
{
|
|
m_guidIDs = NULL;
|
|
}
|
|
~CBlockedDrivers()
|
|
{
|
|
if (m_guidIDs) {
|
|
delete m_guidIDs;
|
|
}
|
|
}
|
|
|
|
DECLARE_REGISTRY_RESOURCEID(IDR_BLKDRV)
|
|
DECLARE_NOT_AGGREGATABLE(CBlockedDrivers)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CBlockedDrivers)
|
|
COM_INTERFACE_ENTRY(IBlockedDrivers)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
END_COM_MAP()
|
|
|
|
// IBlockedDrivers
|
|
public:
|
|
STDMETHOD(BlockedDrivers)(/*[out,retval]*/ LPDISPATCH * pCollection);
|
|
};
|
|
|
|
#endif //__BLOCKEDDRIVERS_H_
|
|
|
|
|
|
#ifndef __DRIVER_H_
|
|
#define __DRIVER_H_
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDriver
|
|
class ATL_NO_VTABLE CDriver :
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
public IDispatchImpl<IDriver, &IID_IDriver, &LIBID_BLOCKDRIVERLib>
|
|
{
|
|
protected:
|
|
HAPPHELPINFOCONTEXT m_hAppHelpInfoContext;
|
|
BSTR m_Name;
|
|
BSTR m_Description;
|
|
BSTR m_Manufacturer;
|
|
BSTR m_HelpFile;
|
|
public:
|
|
CDriver()
|
|
{
|
|
m_hAppHelpInfoContext = NULL;
|
|
m_Name = NULL;
|
|
m_Description = NULL;
|
|
m_Manufacturer = NULL;
|
|
m_HelpFile = NULL;
|
|
}
|
|
~CDriver();
|
|
|
|
DECLARE_NOT_AGGREGATABLE(CDriver)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CDriver)
|
|
COM_INTERFACE_ENTRY(IDriver)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
END_COM_MAP()
|
|
|
|
// IDriver
|
|
public:
|
|
BOOL Init(GUID *pguidDB, GUID *pguidID);
|
|
BSTR GetValueFromDatabase(APPHELPINFORMATIONCLASS InfoClass);
|
|
STDMETHOD(get_Name)(/*[out, retval]*/ BSTR *pVal);
|
|
STDMETHOD(get_Description)(/*[out, retval]*/ BSTR *pVal);
|
|
STDMETHOD(get_Manufacturer)(/*[out, retval]*/ BSTR *pVal);
|
|
STDMETHOD(get_HelpFile)(/*[out, retval]*/ BSTR *pVal);
|
|
};
|
|
|
|
#endif //__DRIVER_H_
|
|
|
|
|
|
#ifndef __DRIVERS_H_
|
|
#define __DRIVERS_H_
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDrivers
|
|
class CDriver;
|
|
|
|
class ATL_NO_VTABLE CDrivers :
|
|
public CComObjectRootEx<CComSingleThreadModel>,
|
|
public IDispatchImpl<IDrivers, &IID_IDrivers, &LIBID_BLOCKDRIVERLib>
|
|
{
|
|
protected:
|
|
CDriver ** pDrivers;
|
|
long Count;
|
|
|
|
public:
|
|
BOOL SetDriver(long index,CDriver *pDriver);
|
|
BOOL InitDriverList(long NewCount);
|
|
|
|
CDrivers()
|
|
{
|
|
pDrivers = NULL;
|
|
Count = 0;
|
|
}
|
|
~CDrivers();
|
|
|
|
DECLARE_NOT_AGGREGATABLE(CDrivers)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CDrivers)
|
|
COM_INTERFACE_ENTRY(IDrivers)
|
|
COM_INTERFACE_ENTRY(IDispatch)
|
|
END_COM_MAP()
|
|
|
|
// IDrivers
|
|
public:
|
|
STDMETHOD(get__NewEnum)(/*[out, retval]*/ IUnknown** ppUnk);
|
|
STDMETHOD(Item)(/*[in]*/ long Index,/*[out, retval]*/ LPDISPATCH * ppVal);
|
|
STDMETHOD(get_Count)(/*[out, retval]*/ long *pVal);
|
|
|
|
};
|
|
|
|
#endif //__DRIVERS_H_
|
|
|
|
|
|
#ifndef __DRIVERSENUM_H_
|
|
#define __DRIVERSENUM_H_
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDriversEnum
|
|
class CDriver;
|
|
|
|
class ATL_NO_VTABLE CDriversEnum :
|
|
public IDriversEnum,
|
|
public CComObjectRootEx<CComSingleThreadModel>
|
|
{
|
|
protected:
|
|
CDriver ** pDrivers;
|
|
long Count;
|
|
long Position;
|
|
|
|
public:
|
|
BOOL InternalCopyDrivers(CDriver **pArray,long Count);
|
|
|
|
CDriversEnum()
|
|
{
|
|
pDrivers = NULL;
|
|
Count = 0;
|
|
Position = 0;
|
|
}
|
|
|
|
~CDriversEnum();
|
|
|
|
|
|
|
|
DECLARE_NOT_AGGREGATABLE(CDriversEnum)
|
|
|
|
DECLARE_PROTECT_FINAL_CONSTRUCT()
|
|
|
|
BEGIN_COM_MAP(CDriversEnum)
|
|
COM_INTERFACE_ENTRY(IDriversEnum)
|
|
COM_INTERFACE_ENTRY(IEnumVARIANT)
|
|
END_COM_MAP()
|
|
|
|
// IEnumVARIANT
|
|
public:
|
|
STDMETHOD(Next)(
|
|
/*[in]*/ ULONG celt,
|
|
/*[out, size_is(celt), length_is(*pCeltFetched)]*/ VARIANT * rgVar,
|
|
/*[out]*/ ULONG * pCeltFetched
|
|
);
|
|
STDMETHOD(Skip)(
|
|
/*[in]*/ ULONG celt
|
|
);
|
|
|
|
STDMETHOD(Reset)(
|
|
);
|
|
|
|
STDMETHOD(Clone)(
|
|
/*[out]*/ IEnumVARIANT ** ppEnum
|
|
);
|
|
};
|
|
|
|
#endif //__DRIVERSENUM_H_
|
|
|
|
|