[host] initial service framework implemented

host/Service.cpp

@@ -0,0 +1,165 @@
+#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();
+
+  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;
+  }
+
+  m_capture = CaptureFactory::GetCaptureDevice();
+  if (!m_capture || !m_capture->Initialize())
+  {
+    DEBUG_ERROR("Failed to initialize capture interface");
+    DeInitialize();
+    return false;
+  }
+
+  KVMGFXHeader * header = static_cast<KVMGFXHeader*>(m_memory);
+  ZeroMemory(header, sizeof(KVMGFXHeader));
+  memcpy(header->magic, KVMGFX_HEADER_MAGIC, sizeof(KVMGFX_HEADER_MAGIC));
+
+  header->version   = 2;
+  header->guestID   = m_ivshmem->GetPeerID();
+
+  m_initialized = true;
+  return true;
+}
+
+void Service::DeInitialize()
+{
+  if (m_capture)
+  {
+    m_capture->DeInitialize();
+    m_capture = NULL;
+  }
+
+  if (m_readyEvent != INVALID_HANDLE_VALUE)
+    CloseHandle(m_readyEvent);
+
+  m_memory = NULL;
+  m_ivshmem->DeInitialize();
+  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();
+
+  // capture a frame of data
+  if (!m_capture->GrabFrame(
+    data,
+    m_ivshmem->GetSize() - sizeof(KVMGFXHeader),
+    &header->dataLen))
+  {
+    DEBUG_ERROR("Capture failed");
+    return false;
+  }
+
+  // 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;
+}