server/drdynvc_main.c

 
#include <freerdp/config.h>
 
#include <winpr/crt.h>
#include <winpr/print.h>
#include <winpr/stream.h>
#include <freerdp/channels/log.h>
#include <freerdp/channels/drdynvc.h>
 
#include "drdynvc_main.h"
 
#define TAG CHANNELS_TAG("drdynvc.server")
 
static UINT drdynvc_server_start(DrdynvcServerContext* context)
{
  context->priv->ChannelHandle =
      WTSVirtualChannelOpen(context->vcm, WTS_CURRENT_SESSION, DRDYNVC_SVC_CHANNEL_NAME);
 
  if (!context->priv->ChannelHandle)
  {
    WLog_ERR(TAG, "WTSVirtualChannelOpen failed!");
    return CHANNEL_RC_NO_MEMORY;
  }
 
  if (!(context->priv->StopEvent = CreateEvent(nullptr, TRUE, FALSE, nullptr)))
  {
    WLog_ERR(TAG, "CreateEvent failed!");
    return ERROR_INTERNAL_ERROR;
  }
 
  return ERROR_INTERNAL_ERROR;
}
 
static UINT drdynvc_server_stop(DrdynvcServerContext* context)
{
  UINT error = 0;
  (void)SetEvent(context->priv->StopEvent);
 
  if (WaitForSingleObject(context->priv->Thread, INFINITE) == WAIT_FAILED)
  {
    error = GetLastError();
    WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "!", error);
    return error;
  }
 
  (void)CloseHandle(context->priv->Thread);
  return CHANNEL_RC_OK;
}
 
DrdynvcServerContext* drdynvc_server_context_new(HANDLE vcm)
{
  DrdynvcServerContext* context = nullptr;
  context = (DrdynvcServerContext*)calloc(1, sizeof(DrdynvcServerContext));
 
  if (context)
  {
    context->vcm = vcm;
    context->Start = drdynvc_server_start;
    context->Stop = drdynvc_server_stop;
    context->priv = (DrdynvcServerPrivate*)calloc(1, sizeof(DrdynvcServerPrivate));
 
    if (!context->priv)
    {
      WLog_ERR(TAG, "calloc failed!");
      free(context);
      return nullptr;
    }
  }
  else
  {
    WLog_ERR(TAG, "calloc failed!");
  }
 
  return context;
}
 
void drdynvc_server_context_free(DrdynvcServerContext* context)
{
  if (context)
  {
    free(context->priv);
    free(context);
  }
}