/src/FreeRDP/libfreerdp/core/client.c

Source
/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Client Channels
 *
 * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2015 Thincast Technologies GmbH
 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <freerdp/config.h>

#include "settings.h"

#include <winpr/assert.h>

#include <freerdp/log.h>
#include <freerdp/channels/drdynvc.h>

#include "rdp.h"
#include "client.h"

#define TAG FREERDP_TAG("core.client")

typedef struct
{
  freerdp_channel_handle_fkt_t fkt;
  void* userdata;
} ChannelEventEntry;

/* Use this instance to get access to channels in VirtualChannelInit. It is set during
 * freerdp_connect so channels that use VirtualChannelInit must be initialized from the same thread
 * as freerdp_connect was called */
static WINPR_TLS freerdp* g_Instance = nullptr;

/* use global counter to ensure uniqueness across channel manager instances */
static volatile LONG g_OpenHandleSeq = 1;

/* HashTable mapping channel handles to CHANNEL_OPEN_DATA */
static INIT_ONCE g_ChannelHandlesOnce = INIT_ONCE_STATIC_INIT;
static wHashTable* g_ChannelHandles = nullptr;

static BOOL freerdp_channels_process_message_free(wMessage* message, DWORD type);

static CHANNEL_OPEN_DATA* freerdp_channels_find_channel_open_data_by_name(rdpChannels* channels,
                                                                          const char* name)
{
  WINPR_ASSERT(channels);
  for (int index = 0; index < channels->openDataCount; index++)
  {
    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];

    if (strncmp(name, pChannelOpenData->stats.channelName, CHANNEL_NAME_LEN + 1) == 0)
      return pChannelOpenData;
  }

  return nullptr;
}

/* returns rdpChannel for the channel name passed in */
static rdpMcsChannel* freerdp_channels_find_channel_by_name(rdpRdp* rdp, const char* name)
{
  if (!rdp)
    return nullptr;

  rdpMcs* mcs = rdp->mcs;
  WINPR_ASSERT(mcs);

  for (UINT32 index = 0; index < mcs->channelCount; index++)
  {
    rdpMcsChannel* channel = &mcs->channels[index];

    if (strncmp(name, channel->Name, CHANNEL_NAME_LEN + 1) == 0)
    {
      return channel;
    }
  }

  return nullptr;
}

static rdpMcsChannel* freerdp_channels_find_channel_by_id(rdpRdp* rdp, UINT16 channel_id)
{
  if (!rdp)
    return nullptr;

  rdpMcs* mcs = rdp->mcs;
  WINPR_ASSERT(mcs);

  for (UINT32 index = 0; index < mcs->channelCount; index++)
  {
    rdpMcsChannel* channel = &mcs->channels[index];

    if (channel->ChannelId == channel_id)
    {
      return channel;
    }
  }

  return nullptr;
}

static void channel_queue_message_free(wMessage* msg)
{
  if (!msg || (msg->id != 0))
    return;

  CHANNEL_OPEN_EVENT* ev = (CHANNEL_OPEN_EVENT*)msg->wParam;
  free(ev);
}

static void channel_queue_free(void* obj)
{
  wMessage* msg = (wMessage*)obj;
  freerdp_channels_process_message_free(msg, CHANNEL_EVENT_WRITE_CANCELLED);
  channel_queue_message_free(msg);
}

static BOOL CALLBACK init_channel_handles_table(WINPR_ATTR_UNUSED PINIT_ONCE once,
                                                WINPR_ATTR_UNUSED PVOID param,
                                                WINPR_ATTR_UNUSED PVOID* context)
{
  g_ChannelHandles = HashTable_New(TRUE);
  return TRUE;
}

static void* channel_event_entry_clone(const void* data)
{
  const ChannelEventEntry* entry = data;
  if (!entry)
    return nullptr;

  ChannelEventEntry* copy = calloc(1, sizeof(ChannelEventEntry));
  if (!copy)
    return nullptr;
  *copy = *entry;
  return copy;
}

rdpChannels* freerdp_channels_new(freerdp* instance)
{
  if (!InitOnceExecuteOnce(&g_ChannelHandlesOnce, init_channel_handles_table, nullptr, nullptr))
    return nullptr;

  wObject* obj = nullptr;
  rdpChannels* channels = nullptr;
  channels = (rdpChannels*)calloc(1, sizeof(rdpChannels));

  if (!channels)
    return nullptr;

  if (!g_ChannelHandles)
    goto error;
  if (!InitializeCriticalSectionAndSpinCount(&channels->channelsLock, 4000))
    goto error;

  channels->instance = instance;
  channels->queue = MessageQueue_New(nullptr);

  if (!channels->queue)
    goto error;

  obj = MessageQueue_Object(channels->queue);
  obj->fnObjectFree = channel_queue_free;

  channels->channelEvents = HashTable_New(FALSE);
  if (!channels->channelEvents)
    goto error;

  obj = HashTable_ValueObject(channels->channelEvents);
  WINPR_ASSERT(obj);
  obj->fnObjectFree = free;
  obj->fnObjectNew = channel_event_entry_clone;
  return channels;
error:
  WINPR_PRAGMA_DIAG_PUSH
  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
  freerdp_channels_free(channels);
  WINPR_PRAGMA_DIAG_POP
  return nullptr;
}

void freerdp_channels_free(rdpChannels* channels)
{
  if (!channels)
    return;

  HashTable_Free(channels->channelEvents);

  DeleteCriticalSection(&channels->channelsLock);

  if (channels->queue)
  {
    MessageQueue_Free(channels->queue);
    channels->queue = nullptr;
  }

  free(channels);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT freerdp_drdynvc_on_channel_connected(DrdynvcClientContext* context, const char* name,
                                                 void* pInterface)
{
  UINT status = CHANNEL_RC_OK;

  WINPR_ASSERT(context);

  rdpChannels* channels = (rdpChannels*)context->custom;
  WINPR_ASSERT(channels);

  freerdp* instance = channels->instance;
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);

  ChannelConnectedEventArgs e = WINPR_C_ARRAY_INIT;
  EventArgsInit(&e, "freerdp");
  e.name = name;
  e.pInterface = pInterface;
  if (PubSub_OnChannelConnected(instance->context->pubSub, instance->context, &e) < 0)
    return ERROR_INTERNAL_ERROR;
  return status;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT freerdp_drdynvc_on_channel_disconnected(DrdynvcClientContext* context, const char* name,
                                                    void* pInterface)
{
  UINT status = CHANNEL_RC_OK;

  WINPR_ASSERT(context);
  rdpChannels* channels = (rdpChannels*)context->custom;
  WINPR_ASSERT(channels);

  freerdp* instance = channels->instance;
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);

  ChannelDisconnectedEventArgs e = WINPR_C_ARRAY_INIT;
  EventArgsInit(&e, "freerdp");
  e.name = name;
  e.pInterface = pInterface;
  if (PubSub_OnChannelDisconnected(instance->context->pubSub, instance->context, &e) < 0)
    return ERROR_INTERNAL_ERROR;
  return status;
}

static UINT freerdp_drdynvc_on_channel_attached(DrdynvcClientContext* context, const char* name,
                                                void* pInterface)
{
  UINT status = CHANNEL_RC_OK;

  WINPR_ASSERT(context);
  rdpChannels* channels = (rdpChannels*)context->custom;
  WINPR_ASSERT(channels);

  freerdp* instance = channels->instance;
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);

  ChannelAttachedEventArgs e = WINPR_C_ARRAY_INIT;
  EventArgsInit(&e, "freerdp");
  e.name = name;
  e.pInterface = pInterface;
  if (PubSub_OnChannelAttached(instance->context->pubSub, instance->context, &e) < 0)
    return ERROR_INTERNAL_ERROR;
  return status;
}

static UINT freerdp_drdynvc_on_channel_detached(DrdynvcClientContext* context, const char* name,
                                                void* pInterface)
{
  UINT status = CHANNEL_RC_OK;

  WINPR_ASSERT(context);
  rdpChannels* channels = (rdpChannels*)context->custom;
  WINPR_ASSERT(channels);

  freerdp* instance = channels->instance;
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);

  ChannelDetachedEventArgs e = WINPR_C_ARRAY_INIT;
  EventArgsInit(&e, "freerdp");
  e.name = name;
  e.pInterface = pInterface;
  if (PubSub_OnChannelDetached(instance->context->pubSub, instance->context, &e) < 0)
    return ERROR_INTERNAL_ERROR;
  return status;
}

void freerdp_channels_register_instance(WINPR_ATTR_UNUSED rdpChannels* channels, freerdp* instance)
{
  WINPR_ASSERT(channels);
  WINPR_ASSERT(instance);
  /* store instance in TLS so future VirtualChannelInit calls can use it */
  g_Instance = instance;
}

/**
 * go through and inform all the libraries that we are initialized
 * called only from main thread
 */
UINT freerdp_channels_pre_connect(rdpChannels* channels, freerdp* instance)
{
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(channels);
  WINPR_ASSERT(instance);

  MessageQueue_Clear(channels->queue);

  for (int index = 0; index < channels->clientDataCount; index++)
  {
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];

    if (pChannelClientData->pChannelInitEventProc)
    {
      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_INITIALIZED, nullptr, 0);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
                                                  pChannelClientData->pInitHandle,
                                                  CHANNEL_EVENT_INITIALIZED, nullptr, 0);
    }

    error = getChannelError(instance->context);
    if (CHANNEL_RC_OK != error)
      break;

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    CHANNEL_INIT_DATA* pChannelInitData = &channels->initDataList[index];
    ChannelInitializedEventArgs e = WINPR_C_ARRAY_INIT;
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelInitData->pInterface;
    if (PubSub_OnChannelInitialized(instance->context->pubSub, instance->context, &e) < 0)
    {
      error = ERROR_INTERNAL_ERROR;
      break;
    }
  }

  return error;
}

UINT freerdp_channels_attach(freerdp* instance)
{
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->settings);

  rdpChannels* channels = instance->context->channels;
  const char* hostname =
      freerdp_settings_get_string(instance->context->settings, FreeRDP_ServerHostname);
  WINPR_ASSERT(hostname);
  const size_t hostnameLength = strnlen(hostname, MAX_PATH);

  for (int index = 0; index < channels->clientDataCount; index++)
  {
    union
    {
      const void* cpv;
      void* pv;
    } cnv;
    ChannelAttachedEventArgs e = WINPR_C_ARRAY_INIT;

    cnv.cpv = hostname;
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];

    if (pChannelClientData->pChannelInitEventProc)
    {

      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_ATTACHED, cnv.pv,
                                                (UINT)hostnameLength);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(
          pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
          CHANNEL_EVENT_ATTACHED, cnv.pv, (UINT)hostnameLength);
    }

    error = getChannelError(instance->context);
    if (error != CHANNEL_RC_OK)
      goto fail;

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelOpenData->pInterface;
    if (PubSub_OnChannelAttached(instance->context->pubSub, instance->context, &e) < 0)
    {
      error = ERROR_INTERNAL_ERROR;
      goto fail;
    }
  }

fail:
  return error;
}

UINT freerdp_channels_detach(freerdp* instance)
{
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(instance);

  rdpContext* context = instance->context;
  WINPR_ASSERT(context);

  rdpChannels* channels = context->channels;
  WINPR_ASSERT(channels);

  WINPR_ASSERT(context->settings);
  const char* hostname = freerdp_settings_get_string(context->settings, FreeRDP_ServerHostname);
  WINPR_ASSERT(hostname);
  const size_t hostnameLength = strnlen(hostname, MAX_PATH);

  for (int index = 0; index < channels->clientDataCount; index++)
  {
    union
    {
      const void* cpv;
      void* pv;
    } cnv;

    ChannelDetachedEventArgs e = WINPR_C_ARRAY_INIT;

    cnv.cpv = hostname;
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];

    if (pChannelClientData->pChannelInitEventProc)
    {
      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_DETACHED, cnv.pv,
                                                (UINT)hostnameLength);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(
          pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
          CHANNEL_EVENT_DETACHED, cnv.pv, (UINT)hostnameLength);
    }

    error = getChannelError(context);
    if (error != CHANNEL_RC_OK)
      goto fail;

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelOpenData->pInterface;
    if (PubSub_OnChannelDetached(context->pubSub, context, &e) < 0)
    {
      error = ERROR_INTERNAL_ERROR;
      goto fail;
    }
  }

fail:
  return error;
}

/**
 * go through and inform all the libraries that we are connected
 * this will tell the libraries that its ok to call MyVirtualChannelOpen
 * called only from main thread
 */
UINT freerdp_channels_post_connect(rdpChannels* channels, freerdp* instance)
{
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(channels);
  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->settings);

  channels->connected = TRUE;
  const char* hostname =
      freerdp_settings_get_string(instance->context->settings, FreeRDP_ServerHostname);
  WINPR_ASSERT(hostname);
  const size_t hostnameLength = strnlen(hostname, MAX_PATH);

  for (int index = 0; index < channels->clientDataCount; index++)
  {
    union
    {
      const void* pcb;
      void* pb;
    } cnv;

    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];
    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];

    cnv.pcb = hostname;
    if (pChannelClientData->pChannelInitEventProc)
    {
      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_CONNECTED, cnv.pb,
                                                (UINT)hostnameLength);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(
          pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
          CHANNEL_EVENT_CONNECTED, cnv.pb, (UINT)hostnameLength);
    }

    error = getChannelError(instance->context);
    if (error != CHANNEL_RC_OK)
      goto fail;

    ChannelConnectedEventArgs e = WINPR_C_ARRAY_INIT;
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelOpenData->pInterface;
    if (PubSub_OnChannelConnected(instance->context->pubSub, instance->context, &e) < 0)
    {
      error = ERROR_INTERNAL_ERROR;
      goto fail;
    }
  }

  channels->drdynvc = (DrdynvcClientContext*)freerdp_channels_get_static_channel_interface(
      channels, DRDYNVC_SVC_CHANNEL_NAME);

  if (channels->drdynvc)
  {
    channels->drdynvc->custom = (void*)channels;
    channels->drdynvc->OnChannelConnected = freerdp_drdynvc_on_channel_connected;
    channels->drdynvc->OnChannelDisconnected = freerdp_drdynvc_on_channel_disconnected;
    channels->drdynvc->OnChannelAttached = freerdp_drdynvc_on_channel_attached;
    channels->drdynvc->OnChannelDetached = freerdp_drdynvc_on_channel_detached;
  }

fail:
  return error;
}

BOOL freerdp_channels_data(freerdp* instance, UINT16 channelId, const BYTE* cdata, size_t dataSize,
                           UINT32 flags, size_t totalSize)
{
  union
  {
    const BYTE* pcb;
    BYTE* pb;
  } data;

  data.pcb = cdata;
  if (!instance || !data.pcb)
  {
    WLog_ERR(TAG, "(%p, %" PRIu16 ", %p, 0x%08x): Invalid arguments",
             WINPR_CXX_COMPAT_CAST(const void*, instance), channelId,
             WINPR_CXX_COMPAT_CAST(const void*, data.pcb), flags);
    return FALSE;
  }

  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);

  rdpMcs* mcs = instance->context->rdp->mcs;
  WINPR_ASSERT(mcs);

  rdpChannels* channels = instance->context->channels;
  WINPR_ASSERT(channels);

  if (!channels || !mcs)
  {
    return FALSE;
  }

  rdpMcsChannel* channel = nullptr;
  for (UINT32 index = 0; index < mcs->channelCount; index++)
  {
    rdpMcsChannel* cur = &mcs->channels[index];

    if (cur->ChannelId == channelId)
    {
      channel = cur;
      break;
    }
  }

  if (!channel)
  {
    return FALSE;
  }

  CHANNEL_OPEN_DATA* pChannelOpenData =
      freerdp_channels_find_channel_open_data_by_name(channels, channel->Name);

  if (!pChannelOpenData)
  {
    return FALSE;
  }

  pChannelOpenData->stats.bytesIn += dataSize;
  pChannelOpenData->stats.fragmentsIn++;
  if ((flags & CHANNEL_FLAG_LAST) != 0)
    pChannelOpenData->stats.packetsIn++;

  if (pChannelOpenData->pChannelOpenEventProc)
  {
    pChannelOpenData->pChannelOpenEventProc(pChannelOpenData->stats.channelId,
                                            CHANNEL_EVENT_DATA_RECEIVED, data.pb,
                                            (UINT32)dataSize, (UINT32)totalSize, flags);
  }
  else if (pChannelOpenData->pChannelOpenEventProcEx)
  {
    pChannelOpenData->pChannelOpenEventProcEx(
        pChannelOpenData->lpUserParam, pChannelOpenData->stats.channelId,
        CHANNEL_EVENT_DATA_RECEIVED, data.pb, (UINT32)dataSize, (UINT32)totalSize, flags);
  }

  return TRUE;
}

UINT16 freerdp_channels_get_id_by_name(freerdp* instance, const char* channel_name)
{
  if (!instance || !channel_name)
    return 0;

  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);
  rdpMcsChannel* mcsChannel =
      freerdp_channels_find_channel_by_name(instance->context->rdp, channel_name);
  if (!mcsChannel)
    return 0;

  return mcsChannel->ChannelId;
}

const char* freerdp_channels_get_name_by_id(freerdp* instance, UINT16 channelId)
{
  if (!instance)
    return nullptr;

  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->rdp);
  rdpMcsChannel* mcsChannel =
      freerdp_channels_find_channel_by_id(instance->context->rdp, channelId);
  if (!mcsChannel)
    return nullptr;

  return mcsChannel->Name;
}

BOOL freerdp_channels_process_message_free(wMessage* message, DWORD type)
{
  WINPR_ASSERT(message);

  if (message->id == WMQ_QUIT)
  {
    return FALSE;
  }

  if (message->id == 0)
  {
    CHANNEL_OPEN_EVENT* item = (CHANNEL_OPEN_EVENT*)message->wParam;

    if (!item)
      return FALSE;

    CHANNEL_OPEN_DATA* pChannelOpenData = item->pChannelOpenData;

    if (pChannelOpenData->pChannelOpenEventProc)
    {
      pChannelOpenData->pChannelOpenEventProc(pChannelOpenData->stats.channelId, type,
                                              item->UserData, item->DataLength,
                                              item->DataLength, 0);
    }
    else if (pChannelOpenData->pChannelOpenEventProcEx)
    {
      pChannelOpenData->pChannelOpenEventProcEx(
          pChannelOpenData->lpUserParam, pChannelOpenData->stats.channelId, type,
          item->UserData, item->DataLength, item->DataLength, 0);
    }
  }

  return TRUE;
}

static BOOL freerdp_channels_process_message(freerdp* instance, wMessage* message)
{
  BOOL ret = TRUE;
  BOOL rc = FALSE;

  WINPR_ASSERT(instance);
  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(message);

  if (message->id == WMQ_QUIT)
    goto fail;
  else if (message->id == 0)
  {
    CHANNEL_OPEN_EVENT* item = (CHANNEL_OPEN_EVENT*)message->wParam;

    if (!item)
      goto fail;

    CHANNEL_OPEN_DATA* pChannelOpenData = item->pChannelOpenData;

    if (pChannelOpenData->flags != 2)
    {
      freerdp_channels_process_message_free(message, CHANNEL_EVENT_WRITE_CANCELLED);
      goto fail;
    }
    rdpMcsChannel* channel = freerdp_channels_find_channel_by_name(
        instance->context->rdp, pChannelOpenData->stats.channelName);

    if (channel)
    {
      pChannelOpenData->stats.bytesOut += item->DataLength;
      pChannelOpenData->stats.packetsOut++;

      const UINT32 VCChunkSize =
          freerdp_settings_get_uint32(instance->context->settings, FreeRDP_VCChunkSize);
      if (VCChunkSize == 0)
        goto fail;
      const uint64_t frags = (1ull * item->DataLength + VCChunkSize - 1ull) / VCChunkSize;
      pChannelOpenData->stats.fragmentsOut += frags;
      ret = instance->SendChannelData(instance, channel->ChannelId, item->Data,
                                      item->DataLength);
    }
  }

  if (!freerdp_channels_process_message_free(message, CHANNEL_EVENT_WRITE_COMPLETE))
    goto fail;

  rc = ret;

fail:
  IFCALL(message->Free, message);
  return rc;
}

/**
 * called only from main thread
 */
static BOOL freerdp_channels_process_sync(rdpChannels* channels, freerdp* instance)
{
  BOOL status = TRUE;
  wMessage message = WINPR_C_ARRAY_INIT;

  WINPR_ASSERT(channels);

  while (MessageQueue_Peek(channels->queue, &message, TRUE))
  {
    if (!freerdp_channels_process_message(instance, &message))
      status = FALSE;
  }

  return status;
}

/**
 * called only from main thread
 */
#if defined(WITH_FREERDP_DEPRECATED)
BOOL freerdp_channels_get_fds(rdpChannels* channels, WINPR_ATTR_UNUSED freerdp* instance,
                              void** read_fds, int* read_count, WINPR_ATTR_UNUSED void** write_fds,
                              WINPR_ATTR_UNUSED int* write_count)
{
  void* pfd = nullptr;
  pfd = GetEventWaitObject(MessageQueue_Event(channels->queue));

  if (pfd)
  {
    read_fds[*read_count] = pfd;
    (*read_count)++;
  }

  return TRUE;
}
#endif

void* freerdp_channels_get_static_channel_interface(rdpChannels* channels, const char* name)
{
  void* pInterface = nullptr;
  CHANNEL_OPEN_DATA* pChannelOpenData =
      freerdp_channels_find_channel_open_data_by_name(channels, name);

  if (pChannelOpenData)
    pInterface = pChannelOpenData->pInterface;

  return pInterface;
}

HANDLE freerdp_channels_get_event_handle(freerdp* instance)
{
  if (!instance)
    return INVALID_HANDLE_VALUE;

  WINPR_ASSERT(instance->context);

  rdpChannels* channels = instance->context->channels;
  WINPR_ASSERT(channels);

  return MessageQueue_Event(channels->queue);
}

static BOOL channels_process(const void* key, void* value, void* arg)
{
  ChannelEventEntry* entry = value;
  rdpContext* context = arg;

  WINPR_UNUSED(key);
  WINPR_ASSERT(entry);

  if (!entry->fkt)
    return FALSE;
  return entry->fkt(context, entry->userdata);
}

int freerdp_channels_process_pending_messages(freerdp* instance)
{
  if (!instance)
    return -1;

  WINPR_ASSERT(instance->context);

  rdpChannels* channels = instance->context->channels;
  WINPR_ASSERT(channels);

  const DWORD status = WaitForSingleObject(MessageQueue_Event(channels->queue), 0);
  if (status == WAIT_OBJECT_0)
  {
    if (!freerdp_channels_process_sync(channels, instance))
      return -1;
  }

  if (!HashTable_Foreach(channels->channelEvents, channels_process, instance->context))
    return -1;

  return 1;
}

/**
 * called only from main thread
 */
BOOL freerdp_channels_check_fds(rdpChannels* channels, freerdp* instance)
{
  WINPR_ASSERT(channels);
  WINPR_UNUSED(channels);

  const int rc = freerdp_channels_process_pending_messages(instance);
  return rc == 1;
}

BOOL freerdp_client_channel_register(rdpChannels* channels, HANDLE handle,
                                     freerdp_channel_handle_fkt_t fkt, void* userdata)
{
  if (!channels || (handle == INVALID_HANDLE_VALUE) || !fkt)
  {
    WLog_ERR(TAG, "Invalid function arguments (channels=%p, handle=%p, fkt=%p, userdata=%p",
             WINPR_FUNC_PTR_CAST(channels, const void*), handle,
             WINPR_FUNC_PTR_CAST(fkt, const void*), userdata);
    return FALSE;
  }

  ChannelEventEntry entry = { .fkt = fkt, .userdata = userdata };
  return HashTable_Insert(channels->channelEvents, handle, &entry);
}

BOOL freerdp_client_channel_unregister(rdpChannels* channels, HANDLE handle)
{
  if (!channels || (handle == INVALID_HANDLE_VALUE))
  {
    WLog_ERR(TAG, "Invalid function arguments (channels=%p, handle=%p",
             WINPR_CXX_COMPAT_CAST(const void*, channels), handle);
    return FALSE;
  }

  return HashTable_Remove(channels->channelEvents, handle);
}

SSIZE_T freerdp_client_channel_get_registered_event_handles(rdpChannels* channels, HANDLE* events,
                                                            DWORD count)
{
  SSIZE_T rc = -1;

  WINPR_ASSERT(channels);
  WINPR_ASSERT(events || (count == 0));

  HashTable_Lock(channels->channelEvents);
  size_t len = HashTable_Count(channels->channelEvents);
  if (len <= count)
  {
    ULONG_PTR* keys = nullptr;
    const size_t nrKeys = HashTable_GetKeys(channels->channelEvents, &keys);
    if ((nrKeys <= SSIZE_MAX) && (nrKeys == len))
    {
      for (size_t x = 0; x < nrKeys; x++)
      {
        HANDLE cur = (HANDLE)keys[x];
        events[x] = cur;
      }
      rc = (SSIZE_T)nrKeys;
    }
    free(keys);
  }
  HashTable_Unlock(channels->channelEvents);
  return rc;
}

UINT freerdp_channels_disconnect(rdpChannels* channels, freerdp* instance)
{
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(channels);

  if (!channels->connected)
    return 0;

  freerdp_channels_check_fds(channels, instance);

  /* tell all libraries we are shutting down */
  for (int index = 0; index < channels->clientDataCount; index++)
  {
    ChannelDisconnectedEventArgs e = WINPR_C_ARRAY_INIT;
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];

    if (pChannelClientData->pChannelInitEventProc)
    {
      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_DISCONNECTED, nullptr, 0);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
                                                  pChannelClientData->pInitHandle,
                                                  CHANNEL_EVENT_DISCONNECTED, nullptr, 0);
    }

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelOpenData->pInterface;
    if (PubSub_OnChannelDisconnected(instance->context->pubSub, instance->context, &e) < 0)
      error = ERROR_INTERNAL_ERROR;
  }

  channels->connected = FALSE;

  /* Flush pending messages */
  freerdp_channels_check_fds(channels, instance);
  return error;
}

void freerdp_channels_close(rdpChannels* channels, freerdp* instance)
{
  WINPR_ASSERT(channels);
  WINPR_ASSERT(instance);

  MessageQueue_PostQuit(channels->queue, 0);
  freerdp_channels_check_fds(channels, instance);

  /* tell all libraries we are shutting down */
  for (int index = 0; index < channels->clientDataCount; index++)
  {
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[index];

    if (pChannelClientData->pChannelInitEventProc)
    {
      pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
                                                CHANNEL_EVENT_TERMINATED, nullptr, 0);
    }
    else if (pChannelClientData->pChannelInitEventProcEx)
    {
      pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
                                                  pChannelClientData->pInitHandle,
                                                  CHANNEL_EVENT_TERMINATED, nullptr, 0);
    }

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    CHANNEL_INIT_DATA* pChannelInitData = &channels->initDataList[index];
    ChannelTerminatedEventArgs e = WINPR_C_ARRAY_INIT;
    EventArgsInit(&e, "freerdp");
    e.name = pChannelOpenData->stats.channelName;
    e.pInterface = pChannelInitData->pInterface;
    if (PubSub_OnChannelTerminated(instance->context->pubSub, instance->context, &e) < 0)
    {
      WLog_WARN(TAG, "PubSub_OnChannelTerminated(%s) failed",
                pChannelOpenData->stats.channelName);
    }
  }

  for (int index = 0; index < channels->openDataCount; index++)
  {
    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
    HashTable_Remove(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->stats.channelId);
  }

  channels->openDataCount = 0;
  channels->initDataCount = 0;
  channels->clientDataCount = 0;

  WINPR_ASSERT(instance->context);
  WINPR_ASSERT(instance->context->settings);
  instance->context->settings->ChannelCount = 0;
  g_Instance = nullptr;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelInitEx(
    LPVOID lpUserParam, LPVOID clientContext, LPVOID pInitHandle, PCHANNEL_DEF pChannel,
    INT channelCount, ULONG versionRequested, PCHANNEL_INIT_EVENT_EX_FN pChannelInitEventProcEx)
{
  if (!pInitHandle)
    return CHANNEL_RC_BAD_INIT_HANDLE;

  if (!pChannel)
    return CHANNEL_RC_BAD_CHANNEL;

  if ((channelCount <= 0) || !pChannelInitEventProcEx)
    return CHANNEL_RC_INITIALIZATION_ERROR;

  CHANNEL_INIT_DATA* pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
  WINPR_ASSERT(pChannelInitData);

  rdpChannels* channels = pChannelInitData->channels;
  WINPR_ASSERT(channels);

  if (!channels->can_call_init)
    return CHANNEL_RC_NOT_IN_VIRTUALCHANNELENTRY;

  if ((channels->openDataCount + channelCount) > CHANNEL_MAX_COUNT)
    return CHANNEL_RC_TOO_MANY_CHANNELS;

  if (channels->connected)
    return CHANNEL_RC_ALREADY_CONNECTED;

  if (versionRequested != VIRTUAL_CHANNEL_VERSION_WIN2000)
  {
  }

  for (int index = 0; index < channelCount; index++)
  {
    const PCHANNEL_DEF pChannelDef = &pChannel[index];

    if (freerdp_channels_find_channel_open_data_by_name(channels, pChannelDef->name) != nullptr)
    {
      return CHANNEL_RC_BAD_CHANNEL;
    }
  }

  pChannelInitData->pInterface = clientContext;

  CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[channels->clientDataCount];
  pChannelClientData->pChannelInitEventProcEx = pChannelInitEventProcEx;
  pChannelClientData->pInitHandle = pInitHandle;
  pChannelClientData->lpUserParam = lpUserParam;
  channels->clientDataCount++;

  WINPR_ASSERT(channels->instance);
  WINPR_ASSERT(channels->instance->context);

  rdpSettings* settings = channels->instance->context->settings;
  WINPR_ASSERT(settings);

  for (int index = 0; index < channelCount; index++)
  {
    const PCHANNEL_DEF pChannelDef = &pChannel[index];
    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[channels->openDataCount];

    WINPR_ASSERT(pChannelOpenData);

    const LONG hdl = InterlockedIncrement(&g_OpenHandleSeq);
    pChannelOpenData->stats.channelId = WINPR_ASSERTING_INT_CAST(uint32_t, hdl);
    pChannelOpenData->channels = channels;
    pChannelOpenData->lpUserParam = lpUserParam;
    if (!HashTable_Insert(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->stats.channelId,
                          (void*)pChannelOpenData))
    {
      pChannelInitData->pInterface = nullptr;
      return CHANNEL_RC_INITIALIZATION_ERROR;
    }
    pChannelOpenData->flags = 1; /* init */
    strncpy(pChannelOpenData->stats.channelName, pChannelDef->name, CHANNEL_NAME_LEN);
    pChannelOpenData->options = pChannelDef->options;

    const UINT32 max = freerdp_settings_get_uint32(settings, FreeRDP_ChannelDefArraySize);
    WINPR_ASSERT(max >= CHANNEL_MAX_COUNT);
    if (settings->ChannelCount < max)
    {
      CHANNEL_DEF* channel = freerdp_settings_get_pointer_array_writable(
          settings, FreeRDP_ChannelDefArray, settings->ChannelCount);
      if (!channel)
        continue;
      strncpy(channel->name, pChannelDef->name, CHANNEL_NAME_LEN);
      channel->options = pChannelDef->options;
      settings->ChannelCount++;
    }

    channels->openDataCount++;
  }

  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelInit(LPVOID* ppInitHandle, PCHANNEL_DEF pChannel,
                                                 INT channelCount, ULONG versionRequested,
                                                 PCHANNEL_INIT_EVENT_FN pChannelInitEventProc)
{
  /* g_Instance should have been set during freerdp_connect - otherwise VirtualChannelInit was
   * called from a different thread */
  if (!g_Instance || !g_Instance->context)
    return CHANNEL_RC_NOT_INITIALIZED;

  rdpChannels* channels = g_Instance->context->channels;

  if (!ppInitHandle || !channels)
    return CHANNEL_RC_BAD_INIT_HANDLE;

  if (!pChannel)
    return CHANNEL_RC_BAD_CHANNEL;

  if ((channelCount <= 0) || !pChannelInitEventProc)
    return CHANNEL_RC_INITIALIZATION_ERROR;

  CHANNEL_INIT_DATA* pChannelInitData = &(channels->initDataList[channels->initDataCount]);
  *ppInitHandle = pChannelInitData;
  channels->initDataCount++;
  pChannelInitData->channels = channels;
  pChannelInitData->pInterface = nullptr;

  if (!channels->can_call_init)
    return CHANNEL_RC_NOT_IN_VIRTUALCHANNELENTRY;

  if (channels->openDataCount + channelCount > CHANNEL_MAX_COUNT)
    return CHANNEL_RC_TOO_MANY_CHANNELS;

  if (channels->connected)
    return CHANNEL_RC_ALREADY_CONNECTED;

  if (versionRequested != VIRTUAL_CHANNEL_VERSION_WIN2000)
  {
  }

  PCHANNEL_DEF pChannelDef = nullptr;
  for (int index = 0; index < channelCount; index++)
  {
    pChannelDef = &pChannel[index];

    if (freerdp_channels_find_channel_open_data_by_name(channels, pChannelDef->name) != nullptr)
    {
      return CHANNEL_RC_BAD_CHANNEL;
    }
  }

  CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[channels->clientDataCount];
  pChannelClientData->pChannelInitEventProc = pChannelInitEventProc;
  pChannelClientData->pInitHandle = *ppInitHandle;
  channels->clientDataCount++;

  rdpSettings* settings = channels->instance->context->settings;

  for (int index = 0; index < channelCount; index++)
  {
    UINT32 ChannelCount = freerdp_settings_get_uint32(settings, FreeRDP_ChannelCount);

    CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[channels->openDataCount];

    const LONG hdl = InterlockedIncrement(&g_OpenHandleSeq);
    pChannelOpenData->stats.channelId = WINPR_ASSERTING_INT_CAST(uint32_t, hdl);
    pChannelOpenData->channels = channels;
    if (!HashTable_Insert(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->stats.channelId,
                          (void*)pChannelOpenData))
      return CHANNEL_RC_INITIALIZATION_ERROR;
    pChannelOpenData->flags = 1; /* init */
    strncpy(pChannelOpenData->stats.channelName, pChannelDef->name, CHANNEL_NAME_LEN);
    pChannelOpenData->options = pChannelDef->options;

    if (ChannelCount < CHANNEL_MAX_COUNT)
    {
      CHANNEL_DEF* channel = freerdp_settings_get_pointer_array_writable(
          settings, FreeRDP_ChannelDefArray, ChannelCount++);
      strncpy(channel->name, pChannelDef->name, CHANNEL_NAME_LEN);
      channel->options = pChannelDef->options;
      if (!freerdp_settings_set_uint32(settings, FreeRDP_ChannelCount, ChannelCount))
        return ERROR_INTERNAL_ERROR;
    }

    channels->openDataCount++;
  }

  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE
FreeRDP_VirtualChannelOpenEx(LPVOID pInitHandle, LPDWORD pOpenHandle, PCHAR pChannelName,
                             PCHANNEL_OPEN_EVENT_EX_FN pChannelOpenEventProcEx)
{
  CHANNEL_INIT_DATA* pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
  WINPR_ASSERT(pChannelInitData);

  rdpChannels* channels = pChannelInitData->channels;
  WINPR_ASSERT(channels);

  void* pInterface = pChannelInitData->pInterface;

  if (!pOpenHandle)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (!pChannelOpenEventProcEx)
    return CHANNEL_RC_BAD_PROC;

  if (!channels->connected)
    return CHANNEL_RC_NOT_CONNECTED;

  CHANNEL_OPEN_DATA* pChannelOpenData =
      freerdp_channels_find_channel_open_data_by_name(channels, pChannelName);

  if (!pChannelOpenData)
    return CHANNEL_RC_UNKNOWN_CHANNEL_NAME;

  if (pChannelOpenData->flags == 2)
    return CHANNEL_RC_ALREADY_OPEN;

  pChannelOpenData->flags = 2; /* open */
  pChannelOpenData->pInterface = pInterface;
  pChannelOpenData->pChannelOpenEventProcEx = pChannelOpenEventProcEx;
  *pOpenHandle = pChannelOpenData->stats.channelId;
  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelOpen(LPVOID pInitHandle, LPDWORD pOpenHandle,
                                                 PCHAR pChannelName,
                                                 PCHANNEL_OPEN_EVENT_FN pChannelOpenEventProc)
{
  CHANNEL_INIT_DATA* pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
  WINPR_ASSERT(pChannelInitData);

  rdpChannels* channels = pChannelInitData->channels;
  void* pInterface = pChannelInitData->pInterface;

  if (!pOpenHandle)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (!pChannelOpenEventProc)
    return CHANNEL_RC_BAD_PROC;

  if (!channels->connected)
    return CHANNEL_RC_NOT_CONNECTED;

  CHANNEL_OPEN_DATA* pChannelOpenData =
      freerdp_channels_find_channel_open_data_by_name(channels, pChannelName);

  if (!pChannelOpenData)
    return CHANNEL_RC_UNKNOWN_CHANNEL_NAME;

  if (pChannelOpenData->flags == 2)
    return CHANNEL_RC_ALREADY_OPEN;

  pChannelOpenData->flags = 2; /* open */
  pChannelOpenData->pInterface = pInterface;
  pChannelOpenData->pChannelOpenEventProc = pChannelOpenEventProc;
  *pOpenHandle = pChannelOpenData->stats.channelId;
  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelCloseEx(LPVOID pInitHandle, DWORD openHandle)
{
  if (!pInitHandle)
    return CHANNEL_RC_BAD_INIT_HANDLE;

  CHANNEL_OPEN_DATA* pChannelOpenData =
      HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);

  if (!pChannelOpenData)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (pChannelOpenData->flags != 2)
    return CHANNEL_RC_NOT_OPEN;

  pChannelOpenData->flags = 0;
  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelClose(DWORD openHandle)
{
  CHANNEL_OPEN_DATA* pChannelOpenData =
      HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);

  if (!pChannelOpenData)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (pChannelOpenData->flags != 2)
    return CHANNEL_RC_NOT_OPEN;

  pChannelOpenData->flags = 0;
  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelWriteEx(LPVOID pInitHandle, DWORD openHandle,
                                                    LPVOID pData, ULONG dataLength,
                                                    LPVOID pUserData)
{
  if (!pInitHandle)
    return CHANNEL_RC_BAD_INIT_HANDLE;

  CHANNEL_INIT_DATA* pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
  rdpChannels* channels = pChannelInitData->channels;

  if (!channels)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  CHANNEL_OPEN_DATA* pChannelOpenData =
      HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);

  if (!pChannelOpenData)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (!channels->connected)
    return CHANNEL_RC_NOT_CONNECTED;

  if (!pData)
    return CHANNEL_RC_NULL_DATA;

  if (!dataLength)
    return CHANNEL_RC_ZERO_LENGTH;

  if (pChannelOpenData->flags != 2)
    return CHANNEL_RC_NOT_OPEN;

  CHANNEL_OPEN_EVENT* pChannelOpenEvent = (CHANNEL_OPEN_EVENT*)malloc(sizeof(CHANNEL_OPEN_EVENT));

  if (!pChannelOpenEvent)
    return CHANNEL_RC_NO_MEMORY;

  pChannelOpenEvent->Data = pData;
  pChannelOpenEvent->DataLength = dataLength;
  pChannelOpenEvent->UserData = pUserData;
  pChannelOpenEvent->pChannelOpenData = pChannelOpenData;

  const wMessage message = { .context = channels,
                           .id = 0,
                           .wParam = pChannelOpenEvent,
                           .lParam = nullptr,
                           .Free = channel_queue_message_free };

  if (!MessageQueue_Dispatch(channels->queue, &message))
  {
    free(pChannelOpenEvent);
    return CHANNEL_RC_NO_MEMORY;
  }

  return CHANNEL_RC_OK;
}

static UINT VCAPITYPE FreeRDP_VirtualChannelWrite(DWORD openHandle, LPVOID pData, ULONG dataLength,
                                                  LPVOID pUserData)
{
  CHANNEL_OPEN_DATA* pChannelOpenData =
      HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);

  if (!pChannelOpenData)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  rdpChannels* channels = pChannelOpenData->channels;
  if (!channels)
    return CHANNEL_RC_BAD_CHANNEL_HANDLE;

  if (!channels->connected)
    return CHANNEL_RC_NOT_CONNECTED;

  if (!pData)
    return CHANNEL_RC_NULL_DATA;

  if (!dataLength)
    return CHANNEL_RC_ZERO_LENGTH;

  if (pChannelOpenData->flags != 2)
    return CHANNEL_RC_NOT_OPEN;

  CHANNEL_OPEN_EVENT* pChannelOpenEvent = (CHANNEL_OPEN_EVENT*)malloc(sizeof(CHANNEL_OPEN_EVENT));

  if (!pChannelOpenEvent)
    return CHANNEL_RC_NO_MEMORY;

  pChannelOpenEvent->Data = pData;
  pChannelOpenEvent->DataLength = dataLength;
  pChannelOpenEvent->UserData = pUserData;
  pChannelOpenEvent->pChannelOpenData = pChannelOpenData;

  const wMessage message = { .context = channels,
                           .id = 0,
                           .wParam = pChannelOpenEvent,
                           .lParam = nullptr,
                           .Free = channel_queue_message_free };

  if (!MessageQueue_Dispatch(channels->queue, &message))
  {
    free(pChannelOpenEvent);
    return CHANNEL_RC_NO_MEMORY;
  }

  return CHANNEL_RC_OK;
}

static BOOL freerdp_channels_is_loaded(rdpChannels* channels, PVIRTUALCHANNELENTRY entry)
{
  WINPR_ASSERT(channels);
  for (int i = 0; i < channels->clientDataCount; i++)
  {
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[i];

    if (pChannelClientData->entry == entry)
      return TRUE;
  }

  return FALSE;
}

static BOOL freerdp_channels_is_loaded_ex(rdpChannels* channels, PVIRTUALCHANNELENTRYEX entryEx)
{
  WINPR_ASSERT(channels);
  for (int i = 0; i < channels->clientDataCount; i++)
  {
    CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[i];

    if (pChannelClientData->entryEx == entryEx)
      return TRUE;
  }

  return FALSE;
}

void freerdp_channel_client_stats_free(StaticClientChannelStats* stats)
{
  if (stats)
    free(stats->stats);
  free(stats);
}

StaticClientChannelStats* freerdp_channels_client_stats(rdpChannels* channels)
{
  WINPR_ASSERT(channels);

  if (channels->openDataCount <= 0)
    return nullptr;

  const size_t odc = WINPR_ASSERTING_INT_CAST(size_t, channels->openDataCount);
  StaticClientChannelStat* stats = calloc(odc, sizeof(StaticClientChannelStat));
  if (!stats)
    return nullptr;

  for (int i = 0; i < channels->openDataCount; i++)
  {
    const CHANNEL_OPEN_DATA* pChannelClientData = &channels->openDataList[i];
    StaticClientChannelStat* stat = &stats[i];

    *stat = pChannelClientData->stats;
  }

  StaticClientChannelStats* statsw = calloc(1, sizeof(StaticClientChannelStats));
  if (!statsw)
  {
    free(stats);
    return nullptr;
  }
  statsw->count = odc;
  statsw->stats = stats;
  return statsw;
}

int freerdp_channels_client_load(rdpChannels* channels, WINPR_ATTR_UNUSED rdpSettings* settings,
                                 PVIRTUALCHANNELENTRY entry, void* data)
{
  WINPR_ASSERT(channels);
  WINPR_ASSERT(channels->instance);
  WINPR_ASSERT(channels->instance->context);
  WINPR_ASSERT(entry);

  if (channels->clientDataCount + 1 > CHANNEL_MAX_COUNT)
  {
    WLog_ERR(TAG, "error: too many channels");
    return 1;
  }

  if (freerdp_channels_is_loaded(channels, entry))
  {
    WLog_WARN(TAG, "Skipping, channel already loaded");
    return 0;
  }

  CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[channels->clientDataCount];
  pChannelClientData->entry = entry;

  CHANNEL_ENTRY_POINTS_FREERDP EntryPoints = { .cbSize = sizeof(EntryPoints),
                                             .protocolVersion = VIRTUAL_CHANNEL_VERSION_WIN2000,
                                             .pVirtualChannelInit = FreeRDP_VirtualChannelInit,
                                             .pVirtualChannelOpen = FreeRDP_VirtualChannelOpen,
                                             .pVirtualChannelClose =
                                                 FreeRDP_VirtualChannelClose,
                                             .pVirtualChannelWrite =
                                                 FreeRDP_VirtualChannelWrite,
                                             .MagicNumber = FREERDP_CHANNEL_MAGIC_NUMBER,
                                             .pExtendedData = data,
                                             .context = channels->instance->context };
  /* enable VirtualChannelInit */
  union
  {
    PCHANNEL_ENTRY_POINTS_FREERDP pfx;
    PCHANNEL_ENTRY_POINTS px;
  } ptr;

  ptr.pfx = &EntryPoints;

  channels->can_call_init = TRUE;
  EnterCriticalSection(&channels->channelsLock);
  const BOOL status = pChannelClientData->entry(ptr.px);
  LeaveCriticalSection(&channels->channelsLock);
  /* disable MyVirtualChannelInit */
  channels->can_call_init = FALSE;

  if (!status)
  {
    WLog_ERR(TAG, "error: channel export function call failed");
    return 1;
  }

  return 0;
}

int freerdp_channels_client_load_ex(rdpChannels* channels, WINPR_ATTR_UNUSED rdpSettings* settings,
                                    PVIRTUALCHANNELENTRYEX entryEx, void* data)
{
  WINPR_ASSERT(channels);
  WINPR_ASSERT(channels->instance);
  WINPR_ASSERT(channels->instance->context);
  WINPR_ASSERT(entryEx);

  if (channels->clientDataCount + 1 > CHANNEL_MAX_COUNT)
  {
    WLog_ERR(TAG, "error: too many channels");
    return 1;
  }

  if (freerdp_channels_is_loaded_ex(channels, entryEx))
  {
    WLog_WARN(TAG, "Skipping, channel already loaded");
    return 0;
  }

  CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[channels->clientDataCount];
  pChannelClientData->entryEx = entryEx;

  CHANNEL_INIT_DATA* pChannelInitData = &(channels->initDataList[channels->initDataCount++]);
  void* pInitHandle = pChannelInitData;
  pChannelInitData->channels = channels;

  CHANNEL_ENTRY_POINTS_FREERDP_EX EntryPointsEx = {
    .cbSize = sizeof(EntryPointsEx),
    .protocolVersion = VIRTUAL_CHANNEL_VERSION_WIN2000,
    .pVirtualChannelInitEx = FreeRDP_VirtualChannelInitEx,
    .pVirtualChannelOpenEx = FreeRDP_VirtualChannelOpenEx,
    .pVirtualChannelCloseEx = FreeRDP_VirtualChannelCloseEx,
    .pVirtualChannelWriteEx = FreeRDP_VirtualChannelWriteEx,
    .MagicNumber = FREERDP_CHANNEL_MAGIC_NUMBER,
    .pExtendedData = data,
    .context = channels->instance->context
  };

  union
  {
    PCHANNEL_ENTRY_POINTS_FREERDP_EX pfx;
    PCHANNEL_ENTRY_POINTS_EX px;
  } ptr;

  ptr.pfx = &EntryPointsEx;

  /* enable VirtualChannelInit */
  channels->can_call_init = TRUE;
  EnterCriticalSection(&channels->channelsLock);
  const BOOL status = pChannelClientData->entryEx(ptr.px, pInitHandle);
  LeaveCriticalSection(&channels->channelsLock);
  /* disable MyVirtualChannelInit */
  channels->can_call_init = FALSE;

  if (!status)
  {
    WLog_ERR(TAG, "error: channel export function call failed");
    return 1;
  }

  return 0;
}

/**
 * this is called when processing the command line parameters
 * called only from main thread
 */
int freerdp_channels_load_plugin(rdpChannels* channels, rdpSettings* settings, const char* name,
                                 void* data)
{
  PVIRTUALCHANNELENTRY entry =
      freerdp_load_channel_addin_entry(name, nullptr, nullptr, FREERDP_ADDIN_CHANNEL_STATIC);

  if (!entry)
    return 1;

  return freerdp_channels_client_load(channels, settings, entry, data);
}

Coverage Report

Created: 2026-07-16 07:14