94 lines
2.4 KiB
C++
94 lines
2.4 KiB
C++
#include "ScriptHookPlugin.h"
|
|
|
|
#include "Logging.h"
|
|
#include "RegistryConfig.h"
|
|
|
|
ScriptHookPlugin::ScriptHookPlugin() = default;
|
|
ScriptHookPlugin::~ScriptHookPlugin() = default;
|
|
|
|
HRESULT STDMETHODCALLTYPE ScriptHookPlugin::QueryInterface(REFIID riid, void** ppvObject)
|
|
{
|
|
if (ppvObject == nullptr)
|
|
{
|
|
return E_POINTER;
|
|
}
|
|
|
|
*ppvObject = nullptr;
|
|
if (riid == __uuidof(IUnknown) || riid == __uuidof(IWTSPlugin))
|
|
{
|
|
*ppvObject = static_cast<IWTSPlugin*>(this);
|
|
AddRef();
|
|
return S_OK;
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE ScriptHookPlugin::AddRef()
|
|
{
|
|
return static_cast<ULONG>(InterlockedIncrement(&refCount_));
|
|
}
|
|
|
|
ULONG STDMETHODCALLTYPE ScriptHookPlugin::Release()
|
|
{
|
|
ULONG remaining = static_cast<ULONG>(InterlockedDecrement(&refCount_));
|
|
if (remaining == 0)
|
|
{
|
|
delete this;
|
|
}
|
|
return remaining;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE ScriptHookPlugin::Initialize(IWTSVirtualChannelManager* channelManager)
|
|
{
|
|
UNREFERENCED_PARAMETER(channelManager);
|
|
const hook::HookConfig config = hook::LoadConfig();
|
|
hook::Log(config.enableLogging, L"IWTSPlugin::Initialize aufgerufen.");
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE ScriptHookPlugin::Connected()
|
|
{
|
|
const hook::HookConfig config = hook::LoadConfig();
|
|
hook::Log(config.enableLogging, L"RDP-Verbindung hergestellt (IWTSPlugin::Connected).");
|
|
|
|
if (!config.enabled || !config.startOnConnect)
|
|
{
|
|
hook::Log(config.enableLogging, L"Start ist in der Konfiguration deaktiviert.");
|
|
return S_OK;
|
|
}
|
|
|
|
// Nur Registry lesen + CreateProcess aufrufen. Es wird niemals auf das Script gewartet.
|
|
launcher_.Start(config);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE ScriptHookPlugin::Disconnected(DWORD disconnectCode)
|
|
{
|
|
const hook::HookConfig config = hook::LoadConfig();
|
|
hook::Log(config.enableLogging,
|
|
L"RDP-Verbindung getrennt (IWTSPlugin::Disconnected), Code=" + std::to_wstring(disconnectCode) + L".");
|
|
|
|
if (config.stopOnDisconnect)
|
|
{
|
|
launcher_.Stop(config.enableLogging);
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE ScriptHookPlugin::Terminated()
|
|
{
|
|
const hook::HookConfig config = hook::LoadConfig();
|
|
hook::Log(config.enableLogging, L"IWTSPlugin::Terminated aufgerufen.");
|
|
|
|
if (config.stopOnDisconnect)
|
|
{
|
|
launcher_.Stop(config.enableLogging);
|
|
}
|
|
else
|
|
{
|
|
launcher_.Detach(config.enableLogging);
|
|
}
|
|
return S_OK;
|
|
}
|