mirror of
https://github.com/gnif/LookingGlass.git
synced 2024-12-23 06:11:46 +00:00
202 lines
No EOL
4.6 KiB
C++
202 lines
No EOL
4.6 KiB
C++
/*
|
|
Looking Glass - KVM FrameRelay (KVMFR) Client
|
|
Copyright (C) 2017 Geoffrey McRae <geoff@hostfission.com>
|
|
|
|
This program is free software; you can redistribute it and/or modify it under
|
|
the terms of the GNU General Public License as published by the Free Software
|
|
Foundation; either version 2 of the License, or (at your option) any later
|
|
version.
|
|
|
|
This program is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along with
|
|
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
|
|
Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "Service.h"
|
|
#include "IVSHMEM.h"
|
|
|
|
#include "common\debug.h"
|
|
#include "common\KVMFR.h"
|
|
|
|
#include "CaptureFactory.h"
|
|
|
|
Service * Service::m_instance = NULL;
|
|
|
|
Service::Service() :
|
|
m_initialized(false),
|
|
m_readyEvent(INVALID_HANDLE_VALUE),
|
|
m_capture(NULL),
|
|
m_header(NULL),
|
|
m_frameIndex(0)
|
|
{
|
|
m_ivshmem = IVSHMEM::Get();
|
|
}
|
|
|
|
Service::~Service()
|
|
{
|
|
}
|
|
|
|
bool Service::Initialize(ICapture * captureDevice)
|
|
{
|
|
if (m_initialized)
|
|
DeInitialize();
|
|
|
|
m_capture = captureDevice;
|
|
if (!m_ivshmem->Initialize())
|
|
{
|
|
DEBUG_ERROR("IVSHMEM failed to initalize");
|
|
DeInitialize();
|
|
return false;
|
|
}
|
|
|
|
if (m_ivshmem->GetSize() < sizeof(KVMFRHeader))
|
|
{
|
|
DEBUG_ERROR("Shared memory is not large enough for the KVMFRHeader");
|
|
DeInitialize();
|
|
return false;
|
|
}
|
|
|
|
uint8_t * memory = static_cast<uint8_t*>(m_ivshmem->GetMemory());
|
|
if (!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;
|
|
}
|
|
|
|
m_header = reinterpret_cast<KVMFRHeader *>(memory);
|
|
m_frame[0] = (uint8_t *)(((uintptr_t)memory + sizeof(KVMFRHeader *) + 0x7F) & ~0x7F);
|
|
m_frameSize = ((m_ivshmem->GetSize() - (m_frame[0] - memory)) & ~0x7F) >> 1;
|
|
m_frame[1] = m_frame[0] + m_frameSize;
|
|
m_dataOffset[0] = m_frame[0] - memory;
|
|
m_dataOffset[1] = m_frame[1] - memory;
|
|
|
|
if (m_capture->GetMaxFrameSize() > m_frameSize)
|
|
{
|
|
DEBUG_ERROR("Frame can exceed buffer size!");
|
|
DeInitialize();
|
|
return false;
|
|
}
|
|
|
|
// we save this as it might actually be valid
|
|
UINT16 hostID = m_header->hostID;
|
|
|
|
ZeroMemory(m_header, sizeof(KVMFRHeader));
|
|
memcpy(m_header->magic, KVMFR_HEADER_MAGIC, sizeof(KVMFR_HEADER_MAGIC));
|
|
|
|
m_header->version = KVMFR_HEADER_VERSION;
|
|
m_header->guestID = m_ivshmem->GetPeerID();
|
|
m_header->hostID = hostID;
|
|
m_header->frameType = m_capture->GetFrameType();
|
|
|
|
m_initialized = true;
|
|
return true;
|
|
}
|
|
|
|
void Service::DeInitialize()
|
|
{
|
|
if (m_readyEvent != INVALID_HANDLE_VALUE)
|
|
CloseHandle(m_readyEvent);
|
|
|
|
m_header = NULL;
|
|
m_frame[0] = NULL;
|
|
m_frame[1] = NULL;
|
|
m_dataOffset[0] = 0;
|
|
m_dataOffset[1] = 0;
|
|
|
|
m_ivshmem->DeInitialize();
|
|
|
|
if (m_capture)
|
|
{
|
|
m_capture->DeInitialize();
|
|
m_capture = NULL;
|
|
}
|
|
|
|
m_initialized = false;
|
|
}
|
|
|
|
bool Service::Process()
|
|
{
|
|
if (!m_initialized)
|
|
return false;
|
|
|
|
FrameInfo frame;
|
|
frame.buffer = m_frame[m_frameIndex];
|
|
frame.bufferSize = m_frameSize;
|
|
frame.hasMousePos = false;
|
|
|
|
// wait for the host to notify that is it is ready to proceed
|
|
bool eventDone = false;
|
|
while (!eventDone)
|
|
{
|
|
switch (WaitForSingleObject(m_readyEvent, 1000))
|
|
{
|
|
case WAIT_ABANDONED:
|
|
DEBUG_ERROR("Wait abandoned");
|
|
return false;
|
|
|
|
case WAIT_OBJECT_0:
|
|
eventDone = true;
|
|
break;
|
|
|
|
case WAIT_TIMEOUT:
|
|
continue;
|
|
|
|
case WAIT_FAILED:
|
|
DEBUG_ERROR("Wait failed");
|
|
return false;
|
|
|
|
default:
|
|
DEBUG_ERROR("Unknown error");
|
|
return false;
|
|
}
|
|
}
|
|
ResetEvent(m_readyEvent);
|
|
|
|
// capture a frame of data
|
|
if (!m_capture->GrabFrame(frame))
|
|
{
|
|
m_header->dataLen = 0;
|
|
DEBUG_ERROR("Capture failed");
|
|
return false;
|
|
}
|
|
|
|
// copy the frame details into the header
|
|
// setup the header
|
|
m_header->width = frame.width;
|
|
m_header->height = frame.height;
|
|
m_header->stride = frame.stride;
|
|
m_header->dataPos = m_dataOffset[m_frameIndex];
|
|
m_header->dataLen = frame.outSize;
|
|
|
|
// tell the host where the cursor is
|
|
if (frame.hasMousePos)
|
|
{
|
|
m_header->mouseX = frame.mouseX;
|
|
m_header->mouseY = frame.mouseY;
|
|
}
|
|
else
|
|
{
|
|
POINT cursorPos;
|
|
GetCursorPos(&cursorPos);
|
|
m_header->mouseX = cursorPos.x;
|
|
m_header->mouseY = cursorPos.y;
|
|
}
|
|
|
|
if (++m_frameIndex == 2)
|
|
m_frameIndex = 0;
|
|
|
|
return true;
|
|
} |