LookingGlass/host/Service.cpp
2017-11-01 01:46:47 +11:00

181 lines
3.9 KiB
C++

#include "Service.h"
#include "IVSHMEM.h"
#include "common\debug.h"
#include "common\KVMGFXHeader.h"
#include "CaptureFactory.h"
Service * Service::m_instance = NULL;
Service::Service() :
m_initialized(false),
m_readyEvent(INVALID_HANDLE_VALUE),
m_capture(NULL),
m_memory(NULL)
{
m_ivshmem = IVSHMEM::Get();
}
Service::~Service()
{
}
bool Service::Initialize()
{
if (m_initialized)
DeInitialize();
m_capture = CaptureFactory::GetCaptureDevice();
if (!m_capture || !m_capture->Initialize())
{
DEBUG_ERROR("Failed to initialize capture interface");
DeInitialize();
return false;
}
if (!m_ivshmem->Initialize())
{
DEBUG_ERROR("IVSHMEM failed to initalize");
DeInitialize();
return false;
}
if (m_ivshmem->GetSize() < sizeof(KVMGFXHeader))
{
DEBUG_ERROR("Shared memory is not large enough for the KVMGFXHeader");
DeInitialize();
return false;
}
m_memory = m_ivshmem->GetMemory();
if (!m_memory)
{
DEBUG_ERROR("Failed to get IVSHMEM memory");
DeInitialize();
return false;
}
m_readyEvent = m_ivshmem->CreateVectorEvent(0);
if (m_readyEvent == INVALID_HANDLE_VALUE)
{
DEBUG_ERROR("Failed to get event for vector 0");
DeInitialize();
return false;
}
KVMGFXHeader * header = static_cast<KVMGFXHeader*>(m_memory);
// we save this as it might actually be valid
UINT16 hostID = header->hostID;
ZeroMemory(header, sizeof(KVMGFXHeader));
memcpy(header->magic, KVMGFX_HEADER_MAGIC, sizeof(KVMGFX_HEADER_MAGIC));
header->version = 2;
header->guestID = m_ivshmem->GetPeerID();
header->hostID = hostID;
m_initialized = true;
return true;
}
void Service::DeInitialize()
{
if (m_readyEvent != INVALID_HANDLE_VALUE)
CloseHandle(m_readyEvent);
m_memory = NULL;
m_ivshmem->DeInitialize();
if (m_capture)
{
m_capture->DeInitialize();
m_capture = NULL;
}
m_initialized = false;
}
bool Service::Process(HANDLE stopEvent)
{
if (!m_initialized)
return false;
KVMGFXHeader * header = static_cast<KVMGFXHeader *>(m_memory );
void * data = static_cast<void *>(header + 1);
const size_t available = m_ivshmem->GetSize() - sizeof(KVMGFXHeader);
if (m_capture->GetMaxFrameSize() > available)
{
DEBUG_ERROR("Frame could exceed buffer size!");
return false;
}
// setup the header
header->frameType = m_capture->GetFrameType();
header->compType = m_capture->GetFrameCompression();
header->dataLen = 0;
FrameInfo frame;
frame.buffer = data;
frame.bufferSize = m_ivshmem->GetSize() - sizeof(KVMGFXHeader);
// capture a frame of data
if (!m_capture->GrabFrame(frame))
{
header->dataLen = 0;
DEBUG_ERROR("Capture failed");
return false;
}
// copy the frame details into the header
header->width = frame.width;
header->height = frame.height;
header->stride = frame.stride;
header->dataLen = frame.outSize;
// tell the host where the cursor is
POINT cursorPos;
GetCursorPos(&cursorPos);
header->mouseX = cursorPos.x;
header->mouseY = cursorPos.y;
// wait for the host to notify that is it is ready to proceed
ResetEvent(m_readyEvent);
while(
stopEvent == INVALID_HANDLE_VALUE ||
(WaitForSingleObject(stopEvent, 0) == WAIT_OBJECT_0)
)
{
if (!m_ivshmem->RingDoorbell(header->hostID, 0))
{
DEBUG_ERROR("Failed to ring doorbell");
return false;
}
switch (WaitForSingleObject(m_readyEvent, 1000))
{
case WAIT_ABANDONED:
DEBUG_ERROR("Wait abandoned");
return false;
case WAIT_OBJECT_0:
return true;
// if we timed out we just continue to ring until we get an answer or we are stopped
case WAIT_TIMEOUT:
break;
case WAIT_FAILED:
DEBUG_ERROR("Wait failed");
return false;
default:
DEBUG_ERROR("Unknown error");
return false;
}
}
return true;
}