143 lines
4 KiB
C
143 lines
4 KiB
C
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
// HardwareDevices Registration
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
#ifndef _REGNOTIF_H
|
||
|
|
#define _REGNOTIF_H
|
||
|
|
|
||
|
|
#include "unk.h"
|
||
|
|
#include "namellst.h"
|
||
|
|
|
||
|
|
#include <shpriv.h>
|
||
|
|
|
||
|
|
extern const CLSID CLSID_HardwareDevices;
|
||
|
|
|
||
|
|
class CHardwareDevicesImpl : public CCOMBase, public IHardwareDevices
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// Interface IHardwareDevices
|
||
|
|
STDMETHODIMP EnumVolumes(DWORD dwFlags, IHardwareDevicesVolumesEnum** ppenum);
|
||
|
|
STDMETHODIMP EnumMountPoints(IHardwareDevicesMountPointsEnum** ppenum);
|
||
|
|
STDMETHODIMP EnumDevices(IHardwareDevicesEnum** ppenum);
|
||
|
|
|
||
|
|
STDMETHODIMP Advise(DWORD dwProcessID, ULONG_PTR hThread,
|
||
|
|
ULONG_PTR pfctCallback, DWORD* pdwToken);
|
||
|
|
STDMETHODIMP Unadvise(DWORD dwToken);
|
||
|
|
|
||
|
|
public:
|
||
|
|
static HRESULT _AdviseVolumeArrivedOrUpdated(
|
||
|
|
VOLUMEINFO2* pvolinfo2, LPCWSTR pszMtPts, DWORD cchMtPts, BOOL fAdded);
|
||
|
|
static HRESULT _AdviseVolumeRemoved(LPCWSTR pszDeviceIDVolume,
|
||
|
|
LPCWSTR pszMtPts, DWORD cchMtPts);
|
||
|
|
static HRESULT _AdviseMountPointHelper(LPCWSTR pszMtPt,
|
||
|
|
LPCWSTR pszDeviceIDVolume, BOOL fAdded);
|
||
|
|
static HRESULT _AdviseVolumeMountingEvent(LPCWSTR pszDeviceIDVolume,
|
||
|
|
DWORD dwEvent);
|
||
|
|
static HRESULT _AdviseCheckClients(void);
|
||
|
|
static HRESULT _AdviseDeviceArrivedOrRemoved(LPCWSTR pszDeviceIntfID,
|
||
|
|
GUID* pguidInterface, DWORD dwDeviceFlags, LPCWSTR pszEventType);
|
||
|
|
|
||
|
|
public:
|
||
|
|
CHardwareDevicesImpl();
|
||
|
|
~CHardwareDevicesImpl();
|
||
|
|
|
||
|
|
private:
|
||
|
|
static DWORD _chwdevcb;
|
||
|
|
static LONG _lAdviseToken;
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CUnkTmpl<CHardwareDevicesImpl> CHardwareDevices;
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
class CHardwareDevicesEnumImpl : public CCOMBase, public IHardwareDevicesEnum
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// Interface IHardwareDevicesEnum
|
||
|
|
STDMETHODIMP Next(
|
||
|
|
LPWSTR* ppszDeviceID,
|
||
|
|
GUID* pguidDeviceID);
|
||
|
|
|
||
|
|
public:
|
||
|
|
CHardwareDevicesEnumImpl();
|
||
|
|
~CHardwareDevicesEnumImpl();
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CUnkTmpl<CHardwareDevicesEnumImpl> CHardwareDevicesEnum;
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
class CHardwareDevicesVolumesEnumImpl : public CCOMBase,
|
||
|
|
public IHardwareDevicesVolumesEnum
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// Interface IHardwareDevicesVolumesEnum
|
||
|
|
STDMETHODIMP Next(VOLUMEINFO* pvolinfo);
|
||
|
|
|
||
|
|
public: // for CHardwareDevicesImpl usage only
|
||
|
|
HRESULT _Init(DWORD dwFlags);
|
||
|
|
|
||
|
|
public:
|
||
|
|
CHardwareDevicesVolumesEnumImpl();
|
||
|
|
~CHardwareDevicesVolumesEnumImpl();
|
||
|
|
|
||
|
|
private:
|
||
|
|
class CNamedElemEnum* _penum;
|
||
|
|
DWORD _dwFlags;
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CUnkTmpl<CHardwareDevicesVolumesEnumImpl> CHardwareDevicesVolumesEnum;
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
class CHardwareDevicesMountPointsEnumImpl : public CCOMBase,
|
||
|
|
public IHardwareDevicesMountPointsEnum
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
// Interface IHardwareDevicesMountPointsEnum
|
||
|
|
STDMETHODIMP Next(
|
||
|
|
LPWSTR* ppszMountPoint, // "c:\", or "d:\MountFolder\"
|
||
|
|
LPWSTR* ppszDeviceIDVolume);// \\?\STORAGE#Volume#...{...GUID...}
|
||
|
|
|
||
|
|
public: // for CHardwareDevicesImpl usage only
|
||
|
|
HRESULT _Init();
|
||
|
|
|
||
|
|
public:
|
||
|
|
CHardwareDevicesMountPointsEnumImpl();
|
||
|
|
~CHardwareDevicesMountPointsEnumImpl();
|
||
|
|
|
||
|
|
private:
|
||
|
|
class CNamedElemEnum* _penum;
|
||
|
|
};
|
||
|
|
|
||
|
|
typedef CUnkTmpl<CHardwareDevicesMountPointsEnumImpl>
|
||
|
|
CHardwareDevicesMountPointsEnum;
|
||
|
|
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
class CAdviseClient : public CNamedElem
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
HRESULT Init(LPCWSTR pszElemName);
|
||
|
|
|
||
|
|
HRESULT _Init(DWORD dwProcessID, ULONG_PTR hThread,
|
||
|
|
ULONG_PTR pfctCallback);
|
||
|
|
HRESULT _Cleanup();
|
||
|
|
HRESULT WriteMemoryChunkInOtherProcess(SHHARDWAREEVENT* pshhe,
|
||
|
|
DWORD cbSize, void** ppv);
|
||
|
|
HRESULT QueueUserAPC(void* pv);
|
||
|
|
HRESULT IsProcessStillAlive(void);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
CAdviseClient();
|
||
|
|
~CAdviseClient();
|
||
|
|
|
||
|
|
public:
|
||
|
|
static HRESULT Create(CNamedElem** ppelem);
|
||
|
|
|
||
|
|
private:
|
||
|
|
PAPCFUNC _pfct;
|
||
|
|
HANDLE _hProcess;
|
||
|
|
HANDLE _hThread;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // _REGNOTIF_H
|