56 lines
1.2 KiB
C++
56 lines
1.2 KiB
C++
#ifndef _HOOKPROC_WINDOWSTATIONENUMERATOR_HPP_
|
|
#define _HOOKPROC_WINDOWSTATIONENUMERATOR_HPP_
|
|
#ifndef _HOOKPROC_ENUMWINDOWSTATIONHOOK_HPP_
|
|
#include <hookproc/enumstation.hpp>
|
|
#endif
|
|
#ifndef _COMMON_STRING_HPP_
|
|
#include <common/string.hpp>
|
|
#endif
|
|
#ifndef _COMMON_BLOCK_HPP_
|
|
#include <common/block.hpp>
|
|
#endif
|
|
|
|
class WindowStationEnumerator : public EnumWindowStationHook, public Block<String>
|
|
{
|
|
public:
|
|
WindowStationEnumerator();
|
|
virtual ~WindowStationEnumerator();
|
|
bool enumerateWindowStations(void);
|
|
DWORD size(void)const;
|
|
protected:
|
|
virtual bool enumWindowStationProc(const String &strWindowStation);
|
|
private:
|
|
};
|
|
|
|
inline
|
|
WindowStationEnumerator::WindowStationEnumerator()
|
|
{
|
|
}
|
|
|
|
inline
|
|
WindowStationEnumerator::~WindowStationEnumerator()
|
|
{
|
|
}
|
|
|
|
inline
|
|
bool WindowStationEnumerator::enumerateWindowStations(void)
|
|
{
|
|
remove();
|
|
::EnumWindowStations((LPFNENUMWINDOWSTATIONPROC)getHookAddress(),0L);
|
|
return Block<String>::size()?true:false;
|
|
}
|
|
|
|
inline
|
|
bool WindowStationEnumerator::enumWindowStationProc(const String &strWindowStation)
|
|
{
|
|
insert(&strWindowStation);
|
|
return true;
|
|
}
|
|
|
|
inline
|
|
DWORD WindowStationEnumerator::size(void)const
|
|
{
|
|
return Block<String>::size();
|
|
}
|
|
#endif
|