/src/FreeRDP/channels/rdpdr/client/rdpdr_main.c

Source (jump to first uncovered line)
/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Device Redirection Virtual Channel
 *
 * Copyright 2010-2011 Vic Lee
 * Copyright 2010-2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
 * Copyright 2015-2016 Thincast Technologies GmbH
 * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
 * Copyright 2016 Armin Novak <armin.novak@thincast.com>
 * Copyright 2016 David PHAM-VAN <d.phamvan@inuvika.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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>

#include <winpr/crt.h>
#include <winpr/sysinfo.h>
#include <winpr/assert.h>
#include <winpr/stream.h>

#include <winpr/print.h>
#include <winpr/sspicli.h>

#include <freerdp/types.h>
#include <freerdp/freerdp.h>
#include <freerdp/constants.h>
#include <freerdp/channels/log.h>
#include <freerdp/channels/rdpdr.h>
#include <freerdp/utils/rdpdr_utils.h>

#ifdef _WIN32
#include <windows.h>
#include <dbt.h>
#else
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#endif

#ifdef __MACOSX__
#include <CoreFoundation/CoreFoundation.h>
#include <stdio.h>
#include <dirent.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#endif

#include "rdpdr_capabilities.h"

#include "devman.h"
#include "irp.h"

#include "rdpdr_main.h"

#define TAG CHANNELS_TAG("rdpdr.client")

/* IMPORTANT: Keep in sync with DRIVE_DEVICE */
typedef struct
{
  DEVICE device;
  WCHAR* path;
  BOOL automount;
} DEVICE_DRIVE_EXT;

static const char* rdpdr_state_str(enum RDPDR_CHANNEL_STATE state)
{
  switch (state)
  {
    case RDPDR_CHANNEL_STATE_INITIAL:
      return "RDPDR_CHANNEL_STATE_INITIAL";
    case RDPDR_CHANNEL_STATE_ANNOUNCE:
      return "RDPDR_CHANNEL_STATE_ANNOUNCE";
    case RDPDR_CHANNEL_STATE_ANNOUNCE_REPLY:
      return "RDPDR_CHANNEL_STATE_ANNOUNCE_REPLY";
    case RDPDR_CHANNEL_STATE_NAME_REQUEST:
      return "RDPDR_CHANNEL_STATE_NAME_REQUEST";
    case RDPDR_CHANNEL_STATE_SERVER_CAPS:
      return "RDPDR_CHANNEL_STATE_SERVER_CAPS";
    case RDPDR_CHANNEL_STATE_CLIENT_CAPS:
      return "RDPDR_CHANNEL_STATE_CLIENT_CAPS";
    case RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM:
      return "RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM";
    case RDPDR_CHANNEL_STATE_READY:
      return "RDPDR_CHANNEL_STATE_READY";
    case RDPDR_CHANNEL_STATE_USER_LOGGEDON:
      return "RDPDR_CHANNEL_STATE_USER_LOGGEDON";
    default:
      return "RDPDR_CHANNEL_STATE_UNKNOWN";
  }
}

static const char* rdpdr_device_type_string(UINT32 type)
{
  switch (type)
  {
    case RDPDR_DTYP_SERIAL:
      return "serial";
    case RDPDR_DTYP_PRINT:
      return "printer";
    case RDPDR_DTYP_FILESYSTEM:
      return "drive";
    case RDPDR_DTYP_SMARTCARD:
      return "smartcard";
    case RDPDR_DTYP_PARALLEL:
      return "parallel";
    default:
      return "UNKNOWN";
  }
}

static const char* support_str(BOOL val)
{
  if (val)
    return "supported";
  return "not found";
}

static const char* rdpdr_caps_pdu_str(UINT32 flag)
{
  switch (flag)
  {
    case RDPDR_DEVICE_REMOVE_PDUS:
      return "RDPDR_USER_LOGGEDON_PDU";
    case RDPDR_CLIENT_DISPLAY_NAME_PDU:
      return "RDPDR_CLIENT_DISPLAY_NAME_PDU";
    case RDPDR_USER_LOGGEDON_PDU:
      return "RDPDR_USER_LOGGEDON_PDU";
    default:
      return "RDPDR_UNKNONW";
  }
}

static BOOL rdpdr_check_extended_pdu_flag(rdpdrPlugin* rdpdr, UINT32 flag)
{
  WINPR_ASSERT(rdpdr);

  const BOOL client = (rdpdr->clientExtendedPDU & flag) != 0;
  const BOOL server = (rdpdr->serverExtendedPDU & flag) != 0;

  if (!client || !server)
  {
    WLog_Print(rdpdr->log, WLOG_WARN, "Checking ExtendedPDU::%s, client %s, server %s",
               rdpdr_caps_pdu_str(flag), support_str(client), support_str(server));
    return FALSE;
  }
  return TRUE;
}

BOOL rdpdr_state_advance(rdpdrPlugin* rdpdr, enum RDPDR_CHANNEL_STATE next)
{
  WINPR_ASSERT(rdpdr);

  if (next != rdpdr->state)
    WLog_Print(rdpdr->log, WLOG_DEBUG, "[RDPDR] transition from %s to %s",
               rdpdr_state_str(rdpdr->state), rdpdr_state_str(next));
  rdpdr->state = next;
  return TRUE;
}

static BOOL device_foreach(rdpdrPlugin* rdpdr, BOOL abortOnFail,
                           BOOL (*fkt)(ULONG_PTR key, void* element, void* data), void* data)
{
  BOOL rc = TRUE;
  ULONG_PTR* keys = NULL;

  ListDictionary_Lock(rdpdr->devman->devices);
  const size_t count = ListDictionary_GetKeys(rdpdr->devman->devices, &keys);
  for (size_t x = 0; x < count; x++)
  {
    void* element = ListDictionary_GetItemValue(rdpdr->devman->devices, (void*)keys[x]);
    if (!fkt(keys[x], element, data))
    {
      rc = FALSE;
      if (abortOnFail)
        break;
    }
  }
  free(keys);
  ListDictionary_Unlock(rdpdr->devman->devices);
  return rc;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_try_send_device_list_announce_request(rdpdrPlugin* rdpdr);

static BOOL rdpdr_load_drive(rdpdrPlugin* rdpdr, const char* name, const char* path, BOOL automount)
{
  UINT rc = ERROR_INTERNAL_ERROR;
  union
  {
    RDPDR_DRIVE* drive;
    RDPDR_DEVICE* device;
  } drive;
  const char* args[] = { name, path, automount ? NULL : name };

  drive.device = freerdp_device_new(RDPDR_DTYP_FILESYSTEM, ARRAYSIZE(args), args);
  if (!drive.device)
    goto fail;

  WINPR_ASSERT(rdpdr->context.RdpdrRegisterDevice);
  rc = rdpdr->context.RdpdrRegisterDevice(&rdpdr->context, drive.device, &drive.device->Id);
  if (rc != CHANNEL_RC_OK)
    goto fail;

fail:
  freerdp_device_free(drive.device);
  return rc == CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_send_device_list_remove_request(rdpdrPlugin* rdpdr, UINT32 count,
                                                  const UINT32 ids[])
{
  wStream* s = NULL;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(ids || (count == 0));

  if (count == 0)
    return CHANNEL_RC_OK;

  if (!rdpdr_check_extended_pdu_flag(rdpdr, RDPDR_DEVICE_REMOVE_PDUS))
    return CHANNEL_RC_OK;

  s = StreamPool_Take(rdpdr->pool, count * sizeof(UINT32) + 8);

  if (!s)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_Write_UINT16(s, RDPDR_CTYP_CORE);
  Stream_Write_UINT16(s, PAKID_CORE_DEVICELIST_REMOVE);
  Stream_Write_UINT32(s, count);

  for (UINT32 i = 0; i < count; i++)
    Stream_Write_UINT32(s, ids[i]);

  Stream_SealLength(s);
  return rdpdr_send(rdpdr, s);
}

#if defined(_UWP) || defined(__IOS__)

static UINT handle_hotplug(WINPR_ATTR_UNUSED RdpdrClientContext* context,
                           WINPR_ATTR_UNUSED RdpdrHotplugEventType type)
{
  return ERROR_CALL_NOT_IMPLEMENTED;
}

static void first_hotplug(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr)
{
}

static DWORD WINAPI drive_hotplug_thread_func(WINPR_ATTR_UNUSED LPVOID arg)
{
  return CHANNEL_RC_OK;
}

static UINT drive_hotplug_thread_terminate(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr)
{
  return CHANNEL_RC_OK;
}

#elif defined(_WIN32)

static UINT handle_hotplug(WINPR_ATTR_UNUSED RdpdrClientContext* context,
                           WINPR_ATTR_UNUSED RdpdrHotplugEventType type)
{
  return CHANNEL_RC_OK;
}

static BOOL check_path(const char* path)
{
  UINT type = GetDriveTypeA(path);

  if (!(type == DRIVE_FIXED || type == DRIVE_REMOVABLE || type == DRIVE_CDROM ||
        type == DRIVE_REMOTE))
    return FALSE;

  return GetVolumeInformationA(path, NULL, 0, NULL, NULL, NULL, NULL, 0);
}

static void first_hotplug(rdpdrPlugin* rdpdr)
{
  DWORD unitmask = GetLogicalDrives();

  for (size_t i = 0; i < 26; i++)
  {
    if (unitmask & 0x01)
    {
      char drive_path[] = { 'c', ':', '\\', '\0' };
      char drive_name[] = { 'c', '\0' };
      drive_path[0] = 'A' + (char)i;
      drive_name[0] = 'A' + (char)i;

      if (check_path(drive_path))
      {
        rdpdr_load_drive(rdpdr, drive_name, drive_path, TRUE);
      }
    }

    unitmask = unitmask >> 1;
  }
}

static LRESULT CALLBACK hotplug_proc(HWND hWnd, UINT Msg, WPARAM wParam, LPARAM lParam)
{
  rdpdrPlugin* rdpdr;
  PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)lParam;
  UINT error;
  rdpdr = (rdpdrPlugin*)GetWindowLongPtr(hWnd, GWLP_USERDATA);

  switch (Msg)
  {
    case WM_DEVICECHANGE:
      switch (wParam)
      {
        case DBT_DEVICEARRIVAL:
          if (lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
          {
            PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
            DWORD unitmask = lpdbv->dbcv_unitmask;

            for (int i = 0; i < 26; i++)
            {
              if (unitmask & 0x01)
              {
                char drive_path[] = { 'c', ':', '/', '\0' };
                char drive_name[] = { 'c', '\0' };
                drive_path[0] = 'A' + (char)i;
                drive_name[0] = 'A' + (char)i;

                if (check_path(drive_path))
                {
                  rdpdr_load_drive(rdpdr, drive_name, drive_path, TRUE);
                }
              }

              unitmask = unitmask >> 1;
            }
          }

          break;

        case DBT_DEVICEREMOVECOMPLETE:
          if (lpdb->dbch_devicetype == DBT_DEVTYP_VOLUME)
          {
            PDEV_BROADCAST_VOLUME lpdbv = (PDEV_BROADCAST_VOLUME)lpdb;
            DWORD unitmask = lpdbv->dbcv_unitmask;
            char drive_name_upper, drive_name_lower;
            ULONG_PTR* keys = NULL;
            DEVICE_DRIVE_EXT* device_ext;
            UINT32 ids[1];

            for (int i = 0; i < 26; i++)
            {
              if (unitmask & 0x01)
              {
                drive_name_upper = 'A' + i;
                drive_name_lower = 'a' + i;
                const size_t count =
                    ListDictionary_GetKeys(rdpdr->devman->devices, &keys);

                for (size_t j = 0; j < count; j++)
                {
                  device_ext = (DEVICE_DRIVE_EXT*)ListDictionary_GetItemValue(
                      rdpdr->devman->devices, (void*)keys[j]);

                  if (device_ext->device.type != RDPDR_DTYP_FILESYSTEM)
                    continue;

                  if (device_ext->path[0] == drive_name_upper ||
                      device_ext->path[0] == drive_name_lower)
                  {
                    if (device_ext->automount)
                    {
                      const uint32_t ids[] = { keys[j] };
                      WINPR_ASSERT(rdpdr->context.RdpdrUnregisterDevice);
                      error = rdpdr->context.RdpdrUnregisterDevice(
                          &rdpdr->context, ARRAYSIZE(ids), ids);
                      if (error)
                      {
                        // don't end on error, just report ?
                        WLog_Print(
                            rdpdr->log, WLOG_ERROR,
                            "rdpdr_send_device_list_remove_request failed "
                            "with error %" PRIu32 "!",
                            error);
                      }

                      break;
                    }
                  }
                }

                free(keys);
              }

              unitmask = unitmask >> 1;
            }
          }

          break;

        default:
          break;
      }

      break;

    default:
      return DefWindowProc(hWnd, Msg, wParam, lParam);
  }

  return DefWindowProc(hWnd, Msg, wParam, lParam);
}

static DWORD WINAPI drive_hotplug_thread_func(LPVOID arg)
{
  rdpdrPlugin* rdpdr;
  WNDCLASSEX wnd_cls;
  HWND hwnd;
  MSG msg;
  BOOL bRet;
  DEV_BROADCAST_HANDLE NotificationFilter;
  HDEVNOTIFY hDevNotify;
  rdpdr = (rdpdrPlugin*)arg;
  /* init windows class */
  wnd_cls.cbSize = sizeof(WNDCLASSEX);
  wnd_cls.style = CS_HREDRAW | CS_VREDRAW;
  wnd_cls.lpfnWndProc = hotplug_proc;
  wnd_cls.cbClsExtra = 0;
  wnd_cls.cbWndExtra = 0;
  wnd_cls.hIcon = LoadIcon(NULL, IDI_APPLICATION);
  wnd_cls.hCursor = NULL;
  wnd_cls.hbrBackground = (HBRUSH)GetStockObject(WHITE_BRUSH);
  wnd_cls.lpszMenuName = NULL;
  wnd_cls.lpszClassName = L"DRIVE_HOTPLUG";
  wnd_cls.hInstance = NULL;
  wnd_cls.hIconSm = LoadIcon(NULL, IDI_APPLICATION);
  RegisterClassEx(&wnd_cls);
  /* create window */
  hwnd = CreateWindowEx(0, L"DRIVE_HOTPLUG", NULL, 0, 0, 0, 0, 0, NULL, NULL, NULL, NULL);
  SetWindowLongPtr(hwnd, GWLP_USERDATA, (LONG_PTR)rdpdr);
  rdpdr->hotplug_wnd = hwnd;
  /* register device interface to hwnd */
  NotificationFilter.dbch_size = sizeof(DEV_BROADCAST_HANDLE);
  NotificationFilter.dbch_devicetype = DBT_DEVTYP_HANDLE;
  hDevNotify = RegisterDeviceNotification(hwnd, &NotificationFilter, DEVICE_NOTIFY_WINDOW_HANDLE);

  /* message loop */
  while ((bRet = GetMessage(&msg, 0, 0, 0)) != 0)
  {
    if (bRet == -1)
    {
      break;
    }
    else
    {
      TranslateMessage(&msg);
      DispatchMessage(&msg);
    }
  }

  UnregisterDeviceNotification(hDevNotify);
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT drive_hotplug_thread_terminate(rdpdrPlugin* rdpdr)
{
  UINT error = CHANNEL_RC_OK;

  if (rdpdr->hotplug_wnd && !PostMessage(rdpdr->hotplug_wnd, WM_QUIT, 0, 0))
  {
    error = GetLastError();
    WLog_Print(rdpdr->log, WLOG_ERROR, "PostMessage failed with error %" PRIu32 "", error);
  }

  return error;
}

#elif defined(__MACOSX__)

#define MAX_USB_DEVICES 100

typedef struct
{
  char* path;
  BOOL to_add;
} hotplug_dev;

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT handle_hotplug(WINPR_ATTR_UNUSED RdpdrClientContext* context,
                           WINPR_ATTR_UNUSED RdpdrHotplugEventType type)
{
  WINPR_ASSERT(context);
  rdpdrPlugin* rdpdr = context->handle;

  struct dirent* pDirent = NULL;
  char fullpath[PATH_MAX] = { 0 };
  char* szdir = (char*)"/Volumes";
  struct stat buf = { 0 };
  hotplug_dev dev_array[MAX_USB_DEVICES] = { 0 };
  int count = 0;
  DEVICE_DRIVE_EXT* device_ext = NULL;
  ULONG_PTR* keys = NULL;
  int size = 0;
  UINT error = ERROR_INTERNAL_ERROR;
  UINT32 ids[1];

  DIR* pDir = opendir(szdir);

  if (pDir == NULL)
  {
    printf("Cannot open directory\n");
    return ERROR_OPEN_FAILED;
  }

  while ((pDirent = readdir(pDir)) != NULL)
  {
    if (pDirent->d_name[0] != '.')
    {
      (void)sprintf_s(fullpath, ARRAYSIZE(fullpath), "%s/%s", szdir, pDirent->d_name);
      if (stat(fullpath, &buf) != 0)
        continue;

      if (S_ISDIR(buf.st_mode))
      {
        dev_array[size].path = _strdup(fullpath);

        if (!dev_array[size].path)
        {
          closedir(pDir);
          error = CHANNEL_RC_NO_MEMORY;
          goto cleanup;
        }

        dev_array[size++].to_add = TRUE;
      }
    }
  }

  closedir(pDir);
  /* delete removed devices */
  count = ListDictionary_GetKeys(rdpdr->devman->devices, &keys);

  for (size_t j = 0; j < count; j++)
  {
    char* path = NULL;
    BOOL dev_found = FALSE;
    device_ext =
        (DEVICE_DRIVE_EXT*)ListDictionary_GetItemValue(rdpdr->devman->devices, (void*)keys[j]);

    if (!device_ext || !device_ext->automount)
      continue;

    if (device_ext->device.type != RDPDR_DTYP_FILESYSTEM)
      continue;

    if (device_ext->path == NULL)
      continue;

    path = ConvertWCharToUtf8Alloc(device_ext->path, NULL);
    if (!path)
      continue;

    /* not pluggable device */
    if (strstr(path, "/Volumes/") == NULL)
    {
      free(path);
      continue;
    }

    for (size_t i = 0; i < size; i++)
    {
      if (strstr(path, dev_array[i].path) != NULL)
      {
        dev_found = TRUE;
        dev_array[i].to_add = FALSE;
        break;
      }
    }

    free(path);

    if (!dev_found)
    {
      const uint32_t ids[] = { keys[j] };
      WINPR_ASSERT(rdpdr->context.RdpdrUnregisterDevice);
      error = rdpdr->context.RdpdrUnregisterDevice(&rdpdr->context, ARRAYSIZE(ids), ids);
      if (error)
      {
        WLog_Print(rdpdr->log, WLOG_ERROR,
                   "rdpdr_send_device_list_remove_request failed with error %" PRIu32 "!",
                   error);
        goto cleanup;
      }
    }
  }

  /* add new devices */
  for (size_t i = 0; i < size; i++)
  {
    const hotplug_dev* dev = &dev_array[i];
    if (dev->to_add)
    {
      const char* path = dev->path;
      const char* name = strrchr(path, '/') + 1;
      error = rdpdr_load_drive(rdpdr, name, path, TRUE);
      if (error)
        goto cleanup;
    }
  }

cleanup:
  free(keys);

  for (size_t i = 0; i < size; i++)
    free(dev_array[i].path);

  return error;
}

static void drive_hotplug_fsevent_callback(ConstFSEventStreamRef streamRef,
                                           void* clientCallBackInfo, size_t numEvents,
                                           void* eventPaths,
                                           const FSEventStreamEventFlags eventFlags[],
                                           const FSEventStreamEventId eventIds[])
{
  rdpdrPlugin* rdpdr;
  UINT error;
  char** paths = (char**)eventPaths;
  rdpdr = (rdpdrPlugin*)clientCallBackInfo;

  for (size_t i = 0; i < numEvents; i++)
  {
    if (strcmp(paths[i], "/Volumes/") == 0)
    {
      UINT error = ERROR_CALL_NOT_IMPLEMENTED;
      if (rdpdr->context.RdpdrHotplugDevice)
        error = rdpdr->context.RdpdrHotplugDevice(&rdpdr->context,
                                                  RDPDR_HOTPLUG_CHECK_FOR_CHANGES);
      switch (error)
      {
        case ERROR_DISK_CHANGE:
        case CHANNEL_RC_OK:
          break;
        case ERROR_CALL_NOT_IMPLEMENTED:
          break;
        default:
          WLog_Print(rdpdr->log, WLOG_ERROR,
                     "handle_hotplug failed with error %" PRIu32 "!", error);
          break;
      }
    }
  }
}

static void first_hotplug(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);
  UINT error = ERROR_CALL_NOT_IMPLEMENTED;
  if (rdpdr->context.RdpdrHotplugDevice)
    error = rdpdr->context.RdpdrHotplugDevice(&rdpdr->context, RDPDR_HOTPLUG_FIRST_CHECK);

  switch (error)
  {
    case ERROR_DISK_CHANGE:
    case CHANNEL_RC_OK:
    case ERROR_CALL_NOT_IMPLEMENTED:
      break;
    default:
      WLog_Print(rdpdr->log, WLOG_ERROR, "handle_hotplug failed with error %" PRIu32 "!",
                 error);
      break;
  }
}

static DWORD WINAPI drive_hotplug_thread_func(LPVOID arg)
{
  rdpdrPlugin* rdpdr = (rdpdrPlugin*)arg;
  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->stopEvent);

  CFStringRef path = CFSTR("/Volumes/");
  CFArrayRef pathsToWatch = CFArrayCreate(kCFAllocatorMalloc, (const void**)&path, 1, NULL);
  FSEventStreamContext ctx = {
    .copyDescription = NULL, .info = arg, .release = NULL, .retain = NULL, .version = 0
  };
  FSEventStreamRef fsev =
      FSEventStreamCreate(kCFAllocatorMalloc, drive_hotplug_fsevent_callback, &ctx, pathsToWatch,
                          kFSEventStreamEventIdSinceNow, 1, kFSEventStreamCreateFlagNone);

  dispatch_queue_t queue = dispatch_queue_create(TAG, NULL);
  FSEventStreamSetDispatchQueue(fsev, queue);
  FSEventStreamStart(fsev);
  WLog_Print(rdpdr->log, WLOG_DEBUG, "Started hotplug watcher");
  HANDLE handles[] = { rdpdr->stopEvent, freerdp_abort_event(rdpdr->rdpcontext) };
  WaitForMultipleObjects(ARRAYSIZE(handles), handles, FALSE, INFINITE);
  WLog_Print(rdpdr->log, WLOG_DEBUG, "Stopped hotplug watcher");
  FSEventStreamStop(fsev);
  FSEventStreamRelease(fsev);
  dispatch_release(queue);
out:
  ExitThread(CHANNEL_RC_OK);
  return CHANNEL_RC_OK;
}

#else

static const char* automountLocations[] = { "/run/user/%lu/gvfs", "/run/media/%s", "/media/%s",
                                          "/media", "/mnt" };

static BOOL isAutomountLocation(const char* path)
{
  const size_t nrLocations = sizeof(automountLocations) / sizeof(automountLocations[0]);
  char buffer[MAX_PATH] = { 0 };
  uid_t uid = getuid();
  char uname[MAX_PATH] = { 0 };
  ULONG size = sizeof(uname) - 1;

  if (!GetUserNameExA(NameSamCompatible, uname, &size))
    return FALSE;

  if (!path)
    return FALSE;

  for (size_t x = 0; x < nrLocations; x++)
  {
    const char* location = automountLocations[x];
    size_t length = 0;

    WINPR_PRAGMA_DIAG_PUSH
    WINPR_PRAGMA_DIAG_IGNORED_FORMAT_NONLITERAL
    if (strstr(location, "%lu"))
      (void)snprintf(buffer, sizeof(buffer), location, (unsigned long)uid);
    else if (strstr(location, "%s"))
      (void)snprintf(buffer, sizeof(buffer), location, uname);
    else
      (void)snprintf(buffer, sizeof(buffer), "%s", location);
    WINPR_PRAGMA_DIAG_POP

    length = strnlen(buffer, sizeof(buffer));

    if (strncmp(buffer, path, length) == 0)
    {
      const char* rest = &path[length];

      /* Only consider mount locations with max depth of 1 below the
       * base path or the base path itself. */
      if (*rest == '\0')
        return TRUE;
      else if (*rest == '/')
      {
        const char* token = strstr(&rest[1], "/");

        if (!token || (token[1] == '\0'))
          return TRUE;
      }
    }
  }

  return FALSE;
}

#define MAX_USB_DEVICES 100

typedef struct
{
  char* path;
  BOOL to_add;
} hotplug_dev;

static void handle_mountpoint(hotplug_dev* dev_array, size_t* size, const char* mountpoint)
{
  if (!mountpoint)
    return;
  /* copy hotpluged device mount point to the dev_array */
  if (isAutomountLocation(mountpoint) && (*size < MAX_USB_DEVICES))
  {
    dev_array[*size].path = _strdup(mountpoint);
    dev_array[*size].to_add = TRUE;
    (*size)++;
  }
}

#ifdef __sun
#include <sys/mnttab.h>
static UINT handle_platform_mounts_sun(wLog* log, hotplug_dev* dev_array, size_t* size)
{
  FILE* f;
  struct mnttab ent;
  f = winpr_fopen("/etc/mnttab", "r");
  if (f == NULL)
  {
    WLog_Print(log, WLOG_ERROR, "fopen failed!");
    return ERROR_OPEN_FAILED;
  }
  while (getmntent(f, &ent) == 0)
  {
    handle_mountpoint(dev_array, size, ent.mnt_mountp);
  }
  fclose(f);
  return ERROR_SUCCESS;
}
#endif

#if defined(__FreeBSD__) || defined(__OpenBSD__)
#include <sys/mount.h>
static UINT handle_platform_mounts_bsd(wLog* log, hotplug_dev* dev_array, size_t* size)
{
  int mntsize;
  struct statfs* mntbuf = NULL;

  mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
  if (!mntsize)
  {
    /* TODO: handle 'errno' */
    WLog_Print(log, WLOG_ERROR, "getmntinfo failed!");
    return ERROR_OPEN_FAILED;
  }
  for (size_t idx = 0; idx < (size_t)mntsize; idx++)
  {
    handle_mountpoint(dev_array, size, mntbuf[idx].f_mntonname);
  }
  free(mntbuf);
  return ERROR_SUCCESS;
}
#endif

#if defined(__LINUX__) || defined(__linux__)
#include <mntent.h>
static struct mntent* getmntent_x(FILE* f, struct mntent* buffer, char* pathbuffer,
                                  size_t pathbuffersize)
{
#if defined(FREERDP_HAVE_GETMNTENT_R)
  WINPR_ASSERT(pathbuffersize <= INT32_MAX);
  return getmntent_r(f, buffer, pathbuffer, (int)pathbuffersize);
#else
  (void)buffer;
  (void)pathbuffer;
  (void)pathbuffersize;
  return getmntent(f);
#endif
}

static UINT handle_platform_mounts_linux(wLog* log, hotplug_dev* dev_array, size_t* size)
{
  FILE* f = NULL;
  struct mntent mnt = { 0 };
  char pathbuffer[PATH_MAX] = { 0 };
  struct mntent* ent = NULL;
  f = winpr_fopen("/proc/mounts", "r");
  if (f == NULL)
  {
    WLog_Print(log, WLOG_ERROR, "fopen failed!");
    return ERROR_OPEN_FAILED;
  }
  while ((ent = getmntent_x(f, &mnt, pathbuffer, sizeof(pathbuffer))) != NULL)
  {
    handle_mountpoint(dev_array, size, ent->mnt_dir);
  }
  (void)fclose(f);
  return ERROR_SUCCESS;
}
#endif

static UINT handle_platform_mounts(wLog* log, hotplug_dev* dev_array, size_t* size)
{
#ifdef __sun
  return handle_platform_mounts_sun(log, dev_array, size);
#elif defined(__FreeBSD__) || defined(__OpenBSD__)
  return handle_platform_mounts_bsd(log, dev_array, size);
#elif defined(__LINUX__) || defined(__linux__)
  return handle_platform_mounts_linux(log, dev_array, size);
#endif
  return ERROR_CALL_NOT_IMPLEMENTED;
}

static BOOL device_not_plugged(ULONG_PTR key, void* element, void* data)
{
  const WCHAR* path = (const WCHAR*)data;
  DEVICE_DRIVE_EXT* device_ext = (DEVICE_DRIVE_EXT*)element;

  WINPR_UNUSED(key);
  WINPR_ASSERT(path);

  if (!device_ext || (device_ext->device.type != RDPDR_DTYP_FILESYSTEM) || !device_ext->path)
    return TRUE;
  if (_wcscmp(device_ext->path, path) != 0)
    return TRUE;
  return FALSE;
}

static BOOL device_already_plugged(rdpdrPlugin* rdpdr, const hotplug_dev* device)
{
  BOOL rc = FALSE;
  WCHAR* path = NULL;

  if (!rdpdr || !device)
    return TRUE;
  if (!device->to_add)
    return TRUE;

  WINPR_ASSERT(rdpdr->devman);
  WINPR_ASSERT(device->path);

  path = ConvertUtf8ToWCharAlloc(device->path, NULL);
  if (!path)
    return TRUE;

  rc = device_foreach(rdpdr, TRUE, device_not_plugged, path);
  free(path);
  return !rc;
}

struct hotplug_delete_arg
{
  hotplug_dev* dev_array;
  size_t dev_array_size;
  rdpdrPlugin* rdpdr;
};

static BOOL hotplug_delete_foreach(ULONG_PTR key, void* element, void* data)
{
  char* path = NULL;
  BOOL dev_found = FALSE;
  struct hotplug_delete_arg* arg = (struct hotplug_delete_arg*)data;
  DEVICE_DRIVE_EXT* device_ext = (DEVICE_DRIVE_EXT*)element;

  WINPR_ASSERT(arg);
  WINPR_ASSERT(arg->rdpdr);
  WINPR_ASSERT(arg->dev_array || (arg->dev_array_size == 0));
  WINPR_ASSERT(key <= UINT32_MAX);

  if (!device_ext || (device_ext->device.type != RDPDR_DTYP_FILESYSTEM) || !device_ext->path ||
      !device_ext->automount)
    return TRUE;

  WINPR_ASSERT(device_ext->path);
  path = ConvertWCharToUtf8Alloc(device_ext->path, NULL);
  if (!path)
    return FALSE;

  /* not pluggable device */
  if (isAutomountLocation(path))
  {
    for (size_t i = 0; i < arg->dev_array_size; i++)
    {
      hotplug_dev* cur = &arg->dev_array[i];
      if (cur->path && strstr(path, cur->path) != NULL)
      {
        dev_found = TRUE;
        cur->to_add = FALSE;
        break;
      }
    }
  }

  free(path);

  if (!dev_found)
  {
    const UINT32 ids[1] = { (UINT32)key };
    WINPR_ASSERT(arg->rdpdr->context.RdpdrUnregisterDevice);
    const UINT error =
        arg->rdpdr->context.RdpdrUnregisterDevice(&arg->rdpdr->context, ARRAYSIZE(ids), ids);

    if (error)
    {
      WLog_Print(arg->rdpdr->log, WLOG_ERROR,
                 "rdpdr_send_device_list_remove_request failed with error %" PRIu32 "!",
                 error);
      return FALSE;
    }
  }

  return TRUE;
}

static UINT handle_hotplug(RdpdrClientContext* context,
                           WINPR_ATTR_UNUSED RdpdrHotplugEventType type)
{
  WINPR_ASSERT(context);
  rdpdrPlugin* rdpdr = context->handle;

  hotplug_dev dev_array[MAX_USB_DEVICES] = { 0 };
  size_t size = 0;
  UINT error = ERROR_SUCCESS;
  struct hotplug_delete_arg arg = { dev_array, ARRAYSIZE(dev_array), rdpdr };

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->devman);

  error = handle_platform_mounts(rdpdr->log, dev_array, &size);

  /* delete removed devices */
  /* Ignore result */ device_foreach(rdpdr, FALSE, hotplug_delete_foreach, &arg);

  /* add new devices */
  for (size_t i = 0; i < size; i++)
  {
    hotplug_dev* cur = &dev_array[i];
    if (!device_already_plugged(rdpdr, cur))
    {
      const char* path = cur->path;
      const char* name = strrchr(path, '/') + 1;

      rdpdr_load_drive(rdpdr, name, path, TRUE);
      error = ERROR_DISK_CHANGE;
    }
  }

  for (size_t i = 0; i < size; i++)
    free(dev_array[i].path);

  return error;
}

static void first_hotplug(rdpdrPlugin* rdpdr)
{
  UINT error = ERROR_CALL_NOT_IMPLEMENTED;

  WINPR_ASSERT(rdpdr);
  if (rdpdr->context.RdpdrHotplugDevice)
    error = rdpdr->context.RdpdrHotplugDevice(&rdpdr->context, RDPDR_HOTPLUG_FIRST_CHECK);

  switch (error)
  {
    case ERROR_DISK_CHANGE:
    case CHANNEL_RC_OK:
    case ERROR_OPEN_FAILED:
    case ERROR_CALL_NOT_IMPLEMENTED:
      break;
    default:
      WLog_Print(rdpdr->log, WLOG_ERROR, "handle_hotplug failed with error %" PRIu32 "!",
                 error);
      break;
  }
}

static DWORD WINAPI drive_hotplug_thread_func(LPVOID arg)
{
  rdpdrPlugin* rdpdr = (rdpdrPlugin*)arg;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->stopEvent);

  while (WaitForSingleObject(rdpdr->stopEvent, 1000) == WAIT_TIMEOUT)
  {
    UINT error = ERROR_CALL_NOT_IMPLEMENTED;
    if (rdpdr->context.RdpdrHotplugDevice)
      error =
          rdpdr->context.RdpdrHotplugDevice(&rdpdr->context, RDPDR_HOTPLUG_CHECK_FOR_CHANGES);
    switch (error)
    {
      case ERROR_DISK_CHANGE:
        break;
      case CHANNEL_RC_OK:
      case ERROR_OPEN_FAILED:
      case ERROR_CALL_NOT_IMPLEMENTED:
        break;
      default:
        WLog_Print(rdpdr->log, WLOG_ERROR, "handle_hotplug failed with error %" PRIu32 "!",
                   error);
        goto out;
    }
  }

out:
{
  const UINT error = GetLastError();
  if (error && rdpdr->rdpcontext)
    setChannelError(rdpdr->rdpcontext, error, "reported an error");

  ExitThread(error);
  return error;
}
}

#endif

#if !defined(_WIN32) && !defined(__IOS__)
/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT drive_hotplug_thread_terminate(rdpdrPlugin* rdpdr)
{
  UINT error = 0;

  WINPR_ASSERT(rdpdr);

  if (rdpdr->hotplugThread)
  {
#if !defined(_WIN32)
    if (rdpdr->stopEvent)
      (void)SetEvent(rdpdr->stopEvent);
#endif

    if (WaitForSingleObject(rdpdr->hotplugThread, INFINITE) == WAIT_FAILED)
    {
      error = GetLastError();
      WLog_Print(rdpdr->log, WLOG_ERROR, "WaitForSingleObject failed with error %" PRIu32 "!",
                 error);
      return error;
    }

    (void)CloseHandle(rdpdr->hotplugThread);
    rdpdr->hotplugThread = NULL;
  }

  return CHANNEL_RC_OK;
}

#endif

static UINT rdpdr_add_devices(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->rdpcontext);

  rdpSettings* settings = rdpdr->rdpcontext->settings;
  WINPR_ASSERT(settings);

  for (UINT32 index = 0; index < freerdp_settings_get_uint32(settings, FreeRDP_DeviceCount);
       index++)
  {
    RDPDR_DEVICE* device =
        freerdp_settings_get_pointer_array_writable(settings, FreeRDP_DeviceArray, index);
    WINPR_ASSERT(device);

    if (device->Type == RDPDR_DTYP_FILESYSTEM)
    {
      const char DynamicDrives[] = "DynamicDrives";
      const RDPDR_DRIVE* drive = (const RDPDR_DRIVE*)device;
      if (!drive->Path)
        continue;

      const char wildcard[] = "*";
      BOOL hotplugAll = strncmp(drive->Path, wildcard, sizeof(wildcard)) == 0;
      BOOL hotplugLater = strncmp(drive->Path, DynamicDrives, sizeof(DynamicDrives)) == 0;

      if (hotplugAll || hotplugLater)
      {
        if (!rdpdr->async)
        {
          WLog_Print(rdpdr->log, WLOG_WARN,
                     "Drive hotplug is not supported in synchronous mode!");
          continue;
        }

        if (hotplugAll)
          first_hotplug(rdpdr);

        /* There might be multiple hotplug related device entries.
         * Ensure the thread is only started once
         */
        if (!rdpdr->hotplugThread)
        {
          rdpdr->hotplugThread =
              CreateThread(NULL, 0, drive_hotplug_thread_func, rdpdr, 0, NULL);
          if (!rdpdr->hotplugThread)
          {
            WLog_Print(rdpdr->log, WLOG_ERROR, "CreateThread failed!");
            return ERROR_INTERNAL_ERROR;
          }
        }

        continue;
      }
    }

    const UINT error = devman_load_device_service(rdpdr->devman, device, rdpdr->rdpcontext);
    if (error)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR,
                 "devman_load_device_service failed with error %" PRIu32 "!", error);
      return error;
    }
  }
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_process_connect(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);

  rdpdr->devman = devman_new(rdpdr);

  if (!rdpdr->devman)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "devman_new failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  WINPR_ASSERT(rdpdr->rdpcontext);

  rdpSettings* settings = rdpdr->rdpcontext->settings;
  WINPR_ASSERT(settings);

  rdpdr->ignoreInvalidDevices = freerdp_settings_get_bool(settings, FreeRDP_IgnoreInvalidDevices);

  const char* name = freerdp_settings_get_string(settings, FreeRDP_ClientHostname);
  if (!name)
    name = freerdp_settings_get_string(settings, FreeRDP_ComputerName);
  if (!name)
  {
    DWORD size = ARRAYSIZE(rdpdr->computerName);
    if (!GetComputerNameExA(ComputerNameNetBIOS, rdpdr->computerName, &size))
      return ERROR_INTERNAL_ERROR;
  }
  else
    strncpy(rdpdr->computerName, name, strnlen(name, sizeof(rdpdr->computerName)));

  return rdpdr_add_devices(rdpdr);
}

static UINT rdpdr_process_server_announce_request(rdpdrPlugin* rdpdr, wStream* s)
{
  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(s);

  if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 8))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT16(s, rdpdr->serverVersionMajor);
  Stream_Read_UINT16(s, rdpdr->serverVersionMinor);
  Stream_Read_UINT32(s, rdpdr->clientID);
  rdpdr->sequenceId++;

  rdpdr->clientVersionMajor = MIN(RDPDR_VERSION_MAJOR, rdpdr->serverVersionMajor);
  rdpdr->clientVersionMinor = MIN(RDPDR_VERSION_MINOR_RDP10X, rdpdr->serverVersionMinor);
  WLog_Print(rdpdr->log, WLOG_DEBUG,
             "[rdpdr] server announces version %" PRIu32 ".%" PRIu32 ", client uses %" PRIu32
             ".%" PRIu32,
             rdpdr->serverVersionMajor, rdpdr->serverVersionMinor, rdpdr->clientVersionMajor,
             rdpdr->clientVersionMinor);
  return CHANNEL_RC_OK;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_send_client_announce_reply(rdpdrPlugin* rdpdr)
{
  wStream* s = NULL;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->state == RDPDR_CHANNEL_STATE_ANNOUNCE);
  rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_ANNOUNCE_REPLY);

  s = StreamPool_Take(rdpdr->pool, 12);

  if (!s)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_Write_UINT16(s, RDPDR_CTYP_CORE);             /* Component (2 bytes) */
  Stream_Write_UINT16(s, PAKID_CORE_CLIENTID_CONFIRM); /* PacketId (2 bytes) */
  Stream_Write_UINT16(s, rdpdr->clientVersionMajor);
  Stream_Write_UINT16(s, rdpdr->clientVersionMinor);
  Stream_Write_UINT32(s, rdpdr->clientID);
  return rdpdr_send(rdpdr, s);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_send_client_name_request(rdpdrPlugin* rdpdr)
{
  wStream* s = NULL;
  WCHAR* computerNameW = NULL;
  size_t computerNameLenW = 0;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->state == RDPDR_CHANNEL_STATE_ANNOUNCE_REPLY);
  rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_NAME_REQUEST);

  const size_t len = strnlen(rdpdr->computerName, sizeof(rdpdr->computerName));
  if (len == 0)
    return ERROR_INTERNAL_ERROR;

  WINPR_ASSERT(rdpdr->computerName);
  computerNameW = ConvertUtf8NToWCharAlloc(rdpdr->computerName, len, &computerNameLenW);
  computerNameLenW *= sizeof(WCHAR);

  if (computerNameLenW > 0)
    computerNameLenW += sizeof(WCHAR); // also write '\0'

  s = StreamPool_Take(rdpdr->pool, 16U + computerNameLenW);

  if (!s)
  {
    free(computerNameW);
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_Write_UINT16(s, RDPDR_CTYP_CORE);        /* Component (2 bytes) */
  Stream_Write_UINT16(s, PAKID_CORE_CLIENT_NAME); /* PacketId (2 bytes) */
  Stream_Write_UINT32(s, 1);                      /* unicodeFlag, 0 for ASCII and 1 for Unicode */
  Stream_Write_UINT32(s, 0);                      /* codePage, must be set to zero */
  Stream_Write_UINT32(s,
                      (UINT32)computerNameLenW); /* computerNameLen, including null terminator */
  Stream_Write(s, computerNameW, computerNameLenW);
  free(computerNameW);
  return rdpdr_send(rdpdr, s);
}

static UINT rdpdr_process_server_clientid_confirm(rdpdrPlugin* rdpdr, wStream* s)
{
  UINT16 versionMajor = 0;
  UINT16 versionMinor = 0;
  UINT32 clientID = 0;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(s);

  if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 8))
    return ERROR_INVALID_DATA;

  Stream_Read_UINT16(s, versionMajor);
  Stream_Read_UINT16(s, versionMinor);
  Stream_Read_UINT32(s, clientID);

  if (versionMajor != rdpdr->clientVersionMajor || versionMinor != rdpdr->clientVersionMinor)
  {
    WLog_Print(rdpdr->log, WLOG_WARN,
               "[rdpdr] server announced version %" PRIu32 ".%" PRIu32 ", client uses %" PRIu32
               ".%" PRIu32 " but clientid confirm requests version %" PRIu32 ".%" PRIu32,
               rdpdr->serverVersionMajor, rdpdr->serverVersionMinor, rdpdr->clientVersionMajor,
               rdpdr->clientVersionMinor, versionMajor, versionMinor);
    rdpdr->clientVersionMajor = versionMajor;
    rdpdr->clientVersionMinor = versionMinor;
  }

  if (clientID != rdpdr->clientID)
    rdpdr->clientID = clientID;

  return CHANNEL_RC_OK;
}

struct device_announce_arg
{
  rdpdrPlugin* rdpdr;
  wStream* s;
  BOOL userLoggedOn;
  UINT32 count;
};

static BOOL device_announce(ULONG_PTR key, void* element, void* data)
{
  struct device_announce_arg* arg = data;
  rdpdrPlugin* rdpdr = NULL;
  DEVICE* device = (DEVICE*)element;

  WINPR_UNUSED(key);

  WINPR_ASSERT(arg);
  WINPR_ASSERT(device);
  WINPR_ASSERT(arg->rdpdr);
  WINPR_ASSERT(arg->s);

  rdpdr = arg->rdpdr;

  /**
   * 1. versionMinor 0x0005 doesn't send PAKID_CORE_USER_LOGGEDON
   *    so all devices should be sent regardless of user_loggedon
   * 2. smartcard devices should be always sent
   * 3. other devices are sent only after user_loggedon
   */

  if ((rdpdr->clientVersionMinor == RDPDR_VERSION_MINOR_RDP51) ||
      (device->type == RDPDR_DTYP_SMARTCARD) || arg->userLoggedOn)
  {
    size_t data_len = (device->data == NULL ? 0 : Stream_GetPosition(device->data));

    if (!Stream_EnsureRemainingCapacity(arg->s, 20 + data_len))
    {
      Stream_Release(arg->s);
      WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_EnsureRemainingCapacity failed!");
      return FALSE;
    }

    Stream_Write_UINT32(arg->s, device->type); /* deviceType */
    Stream_Write_UINT32(arg->s, device->id);   /* deviceID */
    strncpy(Stream_Pointer(arg->s), device->name, 8);

    for (size_t i = 0; i < 8; i++)
    {
      BYTE c = 0;
      Stream_Peek_UINT8(arg->s, c);

      if (c > 0x7F)
        Stream_Write_UINT8(arg->s, '_');
      else
        Stream_Seek_UINT8(arg->s);
    }

    WINPR_ASSERT(data_len <= UINT32_MAX);
    Stream_Write_UINT32(arg->s, (UINT32)data_len);

    if (data_len > 0)
      Stream_Write(arg->s, Stream_Buffer(device->data), data_len);

    arg->count++;
    WLog_Print(rdpdr->log, WLOG_INFO,
               "registered [%09s] device #%" PRIu32 ": %05s (type=%2" PRIu32 " id=%2" PRIu32
               ")",
               rdpdr_device_type_string(device->type), arg->count, device->name, device->type,
               device->id);
  }
  return TRUE;
}

static UINT rdpdr_send_device_list_announce_request(rdpdrPlugin* rdpdr, BOOL userLoggedOn)
{
  size_t pos = 0;
  wStream* s = NULL;
  size_t count_pos = 0;
  struct device_announce_arg arg = { 0 };

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->devman);

  if (userLoggedOn)
  {
    rdpdr->userLoggedOn = TRUE;
  }

  s = StreamPool_Take(rdpdr->pool, 256);

  if (!s)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_Write_UINT16(s, RDPDR_CTYP_CORE);                /* Component (2 bytes) */
  Stream_Write_UINT16(s, PAKID_CORE_DEVICELIST_ANNOUNCE); /* PacketId (2 bytes) */
  count_pos = Stream_GetPosition(s);
  Stream_Seek_UINT32(s); /* deviceCount */

  arg.rdpdr = rdpdr;
  arg.userLoggedOn = userLoggedOn;
  arg.s = s;
  if (!device_foreach(rdpdr, TRUE, device_announce, &arg))
    return ERROR_INVALID_DATA;

  if (arg.count == 0)
  {
    Stream_Release(s);
    return CHANNEL_RC_OK;
  }
  pos = Stream_GetPosition(s);
  Stream_SetPosition(s, count_pos);
  Stream_Write_UINT32(s, arg.count);
  Stream_SetPosition(s, pos);
  Stream_SealLength(s);
  return rdpdr_send(rdpdr, s);
}

UINT rdpdr_try_send_device_list_announce_request(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);
  if (rdpdr->state != RDPDR_CHANNEL_STATE_READY)
  {
    WLog_Print(rdpdr->log, WLOG_DEBUG,
               "hotplug event received, but channel [RDPDR] is not ready (state %s), ignoring.",
               rdpdr_state_str(rdpdr->state));
    return CHANNEL_RC_OK;
  }
  return rdpdr_send_device_list_announce_request(rdpdr, rdpdr->userLoggedOn);
}

static UINT dummy_irp_response(rdpdrPlugin* rdpdr, wStream* s)
{
  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(s);

  wStream* output = StreamPool_Take(rdpdr->pool, 256); // RDPDR_DEVICE_IO_RESPONSE_LENGTH
  if (!output)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
    return CHANNEL_RC_NO_MEMORY;
  }

  Stream_SetPosition(s, 4); /* see "rdpdr_process_receive" */

  const uint32_t DeviceId = Stream_Get_UINT32(s);     /* DeviceId (4 bytes) */
  const uint32_t FileId = Stream_Get_UINT32(s);       /* FileId (4 bytes) */
  const uint32_t CompletionId = Stream_Get_UINT32(s); /* CompletionId (4 bytes) */

  WLog_Print(rdpdr->log, WLOG_WARN,
             "Dummy response {DeviceId=%" PRIu32 ", FileId=%" PRIu32 ", CompletionId=%" PRIu32
             "}",
             DeviceId, FileId, CompletionId);
  if (!rdpdr_write_iocompletion_header(output, DeviceId, CompletionId, STATUS_UNSUCCESSFUL))
    return CHANNEL_RC_NO_MEMORY;

  return rdpdr_send(rdpdr, output);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_process_irp(rdpdrPlugin* rdpdr, wStream* s)
{
  IRP* irp = NULL;
  UINT error = CHANNEL_RC_OK;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(s);

  irp = irp_new(rdpdr->devman, rdpdr->pool, s, rdpdr->log, &error);

  if (!irp)
  {
    if ((error == CHANNEL_RC_OK) ||
        (error == ERROR_DEV_NOT_EXIST && rdpdr->ignoreInvalidDevices))
    {
      return dummy_irp_response(rdpdr, s);
    }

    WLog_Print(rdpdr->log, WLOG_ERROR, "irp_new failed with %" PRIu32 "!", error);
    return error;
  }

  if (irp->device->IRPRequest)
    IFCALLRET(irp->device->IRPRequest, error, irp->device, irp);
  else
    irp->Discard(irp);

  if (error != CHANNEL_RC_OK)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "device->IRPRequest failed with error %" PRIu32 "",
               error);
    irp->Discard(irp);
  }

  return error;
}

static UINT rdpdr_process_component(rdpdrPlugin* rdpdr, UINT16 component, UINT16 packetId,
                                    wStream* s)
{
  UINT32 type = 0;
  DEVICE* device = NULL;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(s);

  switch (component)
  {
    case RDPDR_CTYP_PRN:
      type = RDPDR_DTYP_PRINT;
      break;

    default:
      return ERROR_INVALID_DATA;
  }

  device = devman_get_device_by_type(rdpdr->devman, type);

  if (!device)
    return ERROR_DEV_NOT_EXIST;

  return IFCALLRESULT(ERROR_INVALID_PARAMETER, device->CustomComponentRequest, device, component,
                      packetId, s);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static BOOL device_init(ULONG_PTR key, void* element, void* data)
{
  wLog* log = data;
  UINT error = CHANNEL_RC_OK;
  DEVICE* device = element;

  WINPR_UNUSED(key);
  WINPR_UNUSED(data);

  IFCALLRET(device->Init, error, device);

  if (error != CHANNEL_RC_OK)
  {
    WLog_Print(log, WLOG_ERROR, "Device init failed with %s", WTSErrorToString(error));
    return FALSE;
  }
  return TRUE;
}

static UINT rdpdr_process_init(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(rdpdr->devman);

  rdpdr->userLoggedOn = FALSE; /* reset possible received state */
  if (!device_foreach(rdpdr, TRUE, device_init, rdpdr->log))
    return ERROR_INTERNAL_ERROR;
  return CHANNEL_RC_OK;
}

static BOOL state_match(enum RDPDR_CHANNEL_STATE state, size_t count, va_list ap)
{
  for (size_t x = 0; x < count; x++)
  {
    enum RDPDR_CHANNEL_STATE cur = va_arg(ap, enum RDPDR_CHANNEL_STATE);
    if (state == cur)
      return TRUE;
  }
  return FALSE;
}

static const char* state_str(size_t count, va_list ap, char* buffer, size_t size)
{
  for (size_t x = 0; x < count; x++)
  {
    enum RDPDR_CHANNEL_STATE cur = va_arg(ap, enum RDPDR_CHANNEL_STATE);
    const char* curstr = rdpdr_state_str(cur);
    winpr_str_append(curstr, buffer, size, "|");
  }
  return buffer;
}

static BOOL rdpdr_state_check(rdpdrPlugin* rdpdr, UINT16 packetid, enum RDPDR_CHANNEL_STATE next,
                              size_t count, ...)
{
  va_list ap = { 0 };
  WINPR_ASSERT(rdpdr);

  va_start(ap, count);
  BOOL rc = state_match(rdpdr->state, count, ap);
  va_end(ap);

  if (!rc)
  {
    const char* strstate = rdpdr_state_str(rdpdr->state);
    char buffer[256] = { 0 };

    va_start(ap, count);
    state_str(count, ap, buffer, sizeof(buffer));
    va_end(ap);

    WLog_Print(rdpdr->log, WLOG_ERROR,
               "channel [RDPDR] received %s, expected states [%s] but have state %s, aborting.",
               rdpdr_packetid_string(packetid), buffer, strstate);

    rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_INITIAL);
    return FALSE;
  }
  return rdpdr_state_advance(rdpdr, next);
}

static BOOL rdpdr_check_channel_state(rdpdrPlugin* rdpdr, UINT16 packetid)
{
  WINPR_ASSERT(rdpdr);

  switch (packetid)
  {
    case PAKID_CORE_SERVER_ANNOUNCE:
      /* windows servers sometimes send this message.
       * it seems related to session login (e.g. first initialization for RDP/TLS style login,
       * then reinitialize the channel after login successful
       */
      rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_INITIAL);
      return rdpdr_state_check(rdpdr, packetid, RDPDR_CHANNEL_STATE_ANNOUNCE, 1,
                               RDPDR_CHANNEL_STATE_INITIAL);
    case PAKID_CORE_SERVER_CAPABILITY:
      return rdpdr_state_check(
          rdpdr, packetid, RDPDR_CHANNEL_STATE_SERVER_CAPS, 6,
          RDPDR_CHANNEL_STATE_NAME_REQUEST, RDPDR_CHANNEL_STATE_SERVER_CAPS,
          RDPDR_CHANNEL_STATE_READY, RDPDR_CHANNEL_STATE_CLIENT_CAPS,
          RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM, RDPDR_CHANNEL_STATE_USER_LOGGEDON);
    case PAKID_CORE_CLIENTID_CONFIRM:
      return rdpdr_state_check(rdpdr, packetid, RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM, 5,
                               RDPDR_CHANNEL_STATE_NAME_REQUEST,
                               RDPDR_CHANNEL_STATE_SERVER_CAPS,
                               RDPDR_CHANNEL_STATE_CLIENT_CAPS, RDPDR_CHANNEL_STATE_READY,
                               RDPDR_CHANNEL_STATE_USER_LOGGEDON);
    case PAKID_CORE_USER_LOGGEDON:
      if (!rdpdr_check_extended_pdu_flag(rdpdr, RDPDR_USER_LOGGEDON_PDU))
        return FALSE;

      return rdpdr_state_check(
          rdpdr, packetid, RDPDR_CHANNEL_STATE_USER_LOGGEDON, 4,
          RDPDR_CHANNEL_STATE_NAME_REQUEST, RDPDR_CHANNEL_STATE_CLIENT_CAPS,
          RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM, RDPDR_CHANNEL_STATE_READY);
    default:
    {
      enum RDPDR_CHANNEL_STATE state = RDPDR_CHANNEL_STATE_READY;
      return rdpdr_state_check(rdpdr, packetid, state, 1, state);
    }
  }
}

static BOOL tryAdvance(rdpdrPlugin* rdpdr)
{
  if (rdpdr->haveClientId && rdpdr->haveServerCaps)
  {
    const UINT error = rdpdr_send_device_list_announce_request(rdpdr, FALSE);
    if (error)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR,
                 "rdpdr_send_device_list_announce_request failed with error %" PRIu32 "",
                 error);
      return FALSE;
    }
    if (!rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_READY))
      return FALSE;
  }
  return TRUE;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_process_receive(rdpdrPlugin* rdpdr, wStream* s)
{
  UINT16 component = 0;
  UINT16 packetId = 0;
  UINT32 deviceId = 0;
  UINT32 status = 0;
  UINT error = ERROR_INVALID_DATA;

  if (!rdpdr || !s)
    return CHANNEL_RC_NULL_DATA;

  rdpdr_dump_received_packet(rdpdr->log, WLOG_TRACE, s, "[rdpdr-channel] receive");
  if (Stream_GetRemainingLength(s) >= 4)
  {
    Stream_Read_UINT16(s, component); /* Component (2 bytes) */
    Stream_Read_UINT16(s, packetId);  /* PacketId (2 bytes) */

    if (component == RDPDR_CTYP_CORE)
    {
      if (!rdpdr_check_channel_state(rdpdr, packetId))
        return CHANNEL_RC_OK;

      switch (packetId)
      {
        case PAKID_CORE_SERVER_ANNOUNCE:
          rdpdr->haveClientId = FALSE;
          rdpdr->haveServerCaps = FALSE;
          if ((error = rdpdr_process_server_announce_request(rdpdr, s)))
          {
          }
          else if ((error = rdpdr_send_client_announce_reply(rdpdr)))
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "rdpdr_send_client_announce_reply failed with error %" PRIu32 "",
                       error);
          }
          else if ((error = rdpdr_send_client_name_request(rdpdr)))
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "rdpdr_send_client_name_request failed with error %" PRIu32 "",
                       error);
          }
          else if ((error = rdpdr_process_init(rdpdr)))
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "rdpdr_process_init failed with error %" PRIu32 "", error);
          }

          break;

        case PAKID_CORE_SERVER_CAPABILITY:
          if ((error = rdpdr_process_capability_request(rdpdr, s)))
          {
          }
          else if ((error = rdpdr_send_capability_response(rdpdr)))
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "rdpdr_send_capability_response failed with error %" PRIu32 "",
                       error);
          }
          else
          {
            rdpdr->haveServerCaps = TRUE;
            if (!tryAdvance(rdpdr))
              error = ERROR_INTERNAL_ERROR;
          }

          break;

        case PAKID_CORE_CLIENTID_CONFIRM:
          if ((error = rdpdr_process_server_clientid_confirm(rdpdr, s)))
          {
          }
          else
          {
            rdpdr->haveClientId = TRUE;
            if (!tryAdvance(rdpdr))
              error = ERROR_INTERNAL_ERROR;
          }
          break;

        case PAKID_CORE_USER_LOGGEDON:
          if (!rdpdr->haveServerCaps)
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "Wrong state %s for %s. [serverCaps=%d, clientId=%d]",
                       rdpdr_state_str(rdpdr->state), rdpdr_packetid_string(packetId),
                       rdpdr->haveServerCaps, rdpdr->haveClientId);
            error = ERROR_INTERNAL_ERROR;
          }
          else if ((error = rdpdr_send_device_list_announce_request(rdpdr, TRUE)))
          {
            WLog_Print(
                rdpdr->log, WLOG_ERROR,
                "rdpdr_send_device_list_announce_request failed with error %" PRIu32 "",
                error);
          }
          else if (!tryAdvance(rdpdr))
          {
            error = ERROR_INTERNAL_ERROR;
          }

          break;

        case PAKID_CORE_DEVICE_REPLY:

          /* connect to a specific resource */
          if (Stream_GetRemainingLength(s) >= 8)
          {
            Stream_Read_UINT32(s, deviceId);
            Stream_Read_UINT32(s, status);

            if (status != 0)
              devman_unregister_device(rdpdr->devman, (void*)((size_t)deviceId));
            error = CHANNEL_RC_OK;
          }

          break;

        case PAKID_CORE_DEVICE_IOREQUEST:
          if ((error = rdpdr_process_irp(rdpdr, s)))
          {
            WLog_Print(rdpdr->log, WLOG_ERROR,
                       "rdpdr_process_irp failed with error %" PRIu32 "", error);
            return error;
          }
          else
            s = NULL;

          break;

        default:
          WLog_Print(rdpdr->log, WLOG_ERROR,
                     "RDPDR_CTYP_CORE unknown PacketId: 0x%04" PRIX16 "", packetId);
          error = ERROR_INVALID_DATA;
          break;
      }
    }
    else
    {
      error = rdpdr_process_component(rdpdr, component, packetId, s);

      if (error != CHANNEL_RC_OK)
      {
        DWORD level = WLOG_ERROR;
        if (rdpdr->ignoreInvalidDevices)
        {
          if (error == ERROR_DEV_NOT_EXIST)
          {
            level = WLOG_WARN;
            error = CHANNEL_RC_OK;
          }
        }
        WLog_Print(rdpdr->log, level,
                   "Unknown message: Component: %s [0x%04" PRIX16
                   "] PacketId: %s [0x%04" PRIX16 "]",
                   rdpdr_component_string(component), component,
                   rdpdr_packetid_string(packetId), packetId);
      }
    }
  }

  return error;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
UINT rdpdr_send(rdpdrPlugin* rdpdr, wStream* s)
{
  rdpdrPlugin* plugin = rdpdr;

  if (!s)
  {
    Stream_Release(s);
    return CHANNEL_RC_NULL_DATA;
  }

  if (!plugin)
  {
    Stream_Release(s);
    return CHANNEL_RC_BAD_INIT_HANDLE;
  }

  const size_t pos = Stream_GetPosition(s);
  UINT status = ERROR_INTERNAL_ERROR;
  if (pos <= UINT32_MAX)
  {
    rdpdr_dump_send_packet(rdpdr->log, WLOG_TRACE, s, "[rdpdr-channel] send");
    status = plugin->channelEntryPoints.pVirtualChannelWriteEx(
        plugin->InitHandle, plugin->OpenHandle, Stream_Buffer(s), (UINT32)pos, s);
  }

  if (status != CHANNEL_RC_OK)
  {
    Stream_Release(s);
    WLog_Print(rdpdr->log, WLOG_ERROR, "pVirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
               WTSErrorToString(status), status);
  }

  return status;
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_virtual_channel_event_data_received(rdpdrPlugin* rdpdr, void* pData,
                                                      UINT32 dataLength, UINT32 totalLength,
                                                      UINT32 dataFlags)
{
  wStream* data_in = NULL;

  WINPR_ASSERT(rdpdr);
  WINPR_ASSERT(pData || (dataLength == 0));

  if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
  {
    /*
     * According to MS-RDPBCGR 2.2.6.1, "All virtual channel traffic MUST be suspended.
     * This flag is only valid in server-to-client virtual channel traffic. It MUST be
     * ignored in client-to-server data." Thus it would be best practice to cease data
     * transmission. However, simply returning here avoids a crash.
     */
    return CHANNEL_RC_OK;
  }

  if (dataFlags & CHANNEL_FLAG_FIRST)
  {
    if (rdpdr->data_in != NULL)
      Stream_Release(rdpdr->data_in);

    rdpdr->data_in = StreamPool_Take(rdpdr->pool, totalLength);

    if (!rdpdr->data_in)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
      return CHANNEL_RC_NO_MEMORY;
    }
  }

  data_in = rdpdr->data_in;

  if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_EnsureRemainingCapacity failed!");
    return ERROR_INVALID_DATA;
  }

  Stream_Write(data_in, pData, dataLength);

  if (dataFlags & CHANNEL_FLAG_LAST)
  {
    const size_t pos = Stream_GetPosition(data_in);
    const size_t cap = Stream_Capacity(data_in);
    if (cap < pos)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR,
                 "rdpdr_virtual_channel_event_data_received: read error");
      return ERROR_INTERNAL_ERROR;
    }

    Stream_SealLength(data_in);
    Stream_SetPosition(data_in, 0);

    if (rdpdr->async)
    {
      if (!MessageQueue_Post(rdpdr->queue, NULL, 0, (void*)data_in, NULL))
      {
        WLog_Print(rdpdr->log, WLOG_ERROR, "MessageQueue_Post failed!");
        return ERROR_INTERNAL_ERROR;
      }
      rdpdr->data_in = NULL;
    }
    else
    {
      UINT error = rdpdr_process_receive(rdpdr, data_in);
      Stream_Release(data_in);
      rdpdr->data_in = NULL;
      if (error)
        return error;
    }
  }

  return CHANNEL_RC_OK;
}

static VOID VCAPITYPE rdpdr_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle,
                                                          UINT event, LPVOID pData,
                                                          UINT32 dataLength, UINT32 totalLength,
                                                          UINT32 dataFlags)
{
  UINT error = CHANNEL_RC_OK;
  rdpdrPlugin* rdpdr = (rdpdrPlugin*)lpUserParam;

  WINPR_ASSERT(rdpdr);
  switch (event)
  {
    case CHANNEL_EVENT_DATA_RECEIVED:
      if (!rdpdr || !pData || (rdpdr->OpenHandle != openHandle))
      {
        WLog_Print(rdpdr->log, WLOG_ERROR, "error no match");
        return;
      }
      if ((error = rdpdr_virtual_channel_event_data_received(rdpdr, pData, dataLength,
                                                             totalLength, dataFlags)))
        WLog_Print(rdpdr->log, WLOG_ERROR,
                   "rdpdr_virtual_channel_event_data_received failed with error %" PRIu32
                   "!",
                   error);

      break;

    case CHANNEL_EVENT_WRITE_CANCELLED:
    case CHANNEL_EVENT_WRITE_COMPLETE:
    {
      wStream* s = (wStream*)pData;
      Stream_Release(s);
    }
    break;

    case CHANNEL_EVENT_USER:
      break;
    default:
      break;
  }

  if (error && rdpdr && rdpdr->rdpcontext)
    setChannelError(rdpdr->rdpcontext, error,
                    "rdpdr_virtual_channel_open_event_ex reported an error");
}

static DWORD WINAPI rdpdr_virtual_channel_client_thread(LPVOID arg)
{
  rdpdrPlugin* rdpdr = (rdpdrPlugin*)arg;
  UINT error = 0;

  if (!rdpdr)
  {
    ExitThread((DWORD)CHANNEL_RC_NULL_DATA);
    return CHANNEL_RC_NULL_DATA;
  }

  if ((error = rdpdr_process_connect(rdpdr)))
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "rdpdr_process_connect failed with error %" PRIu32 "!",
               error);

    if (rdpdr->rdpcontext)
      setChannelError(rdpdr->rdpcontext, error,
                      "rdpdr_virtual_channel_client_thread reported an error");

    ExitThread(error);
    return error;
  }

  while (1)
  {
    wMessage message = { 0 };
    WINPR_ASSERT(rdpdr);

    if (!MessageQueue_Wait(rdpdr->queue))
      break;

    if (MessageQueue_Peek(rdpdr->queue, &message, TRUE))
    {
      if (message.id == WMQ_QUIT)
        break;

      if (message.id == 0)
      {
        wStream* data = (wStream*)message.wParam;

        error = rdpdr_process_receive(rdpdr, data);

        Stream_Release(data);
        if (error)
        {
          WLog_Print(rdpdr->log, WLOG_ERROR,
                     "rdpdr_process_receive failed with error %" PRIu32 "!", error);

          if (rdpdr->rdpcontext)
            setChannelError(rdpdr->rdpcontext, error,
                            "rdpdr_virtual_channel_client_thread reported an error");

          goto fail;
        }
      }
    }
  }

fail:
  if ((error = drive_hotplug_thread_terminate(rdpdr)))
    WLog_Print(rdpdr->log, WLOG_ERROR,
               "drive_hotplug_thread_terminate failed with error %" PRIu32 "!", error);

  ExitThread(error);
  return error;
}

static void queue_free(void* obj)
{
  wStream* s = NULL;
  wMessage* msg = (wMessage*)obj;

  if (!msg || (msg->id != 0))
    return;

  s = (wStream*)msg->wParam;
  WINPR_ASSERT(s);
  Stream_Release(s);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_virtual_channel_event_connected(rdpdrPlugin* rdpdr, LPVOID pData,
                                                  UINT32 dataLength)
{
  wObject* obj = NULL;

  WINPR_ASSERT(rdpdr);
  WINPR_UNUSED(pData);
  WINPR_UNUSED(dataLength);

  if (rdpdr->async)
  {
    rdpdr->queue = MessageQueue_New(NULL);

    if (!rdpdr->queue)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR, "MessageQueue_New failed!");
      return CHANNEL_RC_NO_MEMORY;
    }

    obj = MessageQueue_Object(rdpdr->queue);
    obj->fnObjectFree = queue_free;

    if (!(rdpdr->thread = CreateThread(NULL, 0, rdpdr_virtual_channel_client_thread,
                                       (void*)rdpdr, 0, NULL)))
    {
      WLog_Print(rdpdr->log, WLOG_ERROR, "CreateThread failed!");
      return ERROR_INTERNAL_ERROR;
    }
  }
  else
  {
    UINT error = rdpdr_process_connect(rdpdr);
    if (error)
    {
      WLog_Print(rdpdr->log, WLOG_ERROR,
                 "rdpdr_process_connect failed with error %" PRIu32 "!", error);
      return error;
    }
  }

  return rdpdr->channelEntryPoints.pVirtualChannelOpenEx(rdpdr->InitHandle, &rdpdr->OpenHandle,
                                                         rdpdr->channelDef.name,
                                                         rdpdr_virtual_channel_open_event_ex);
}

/**
 * Function description
 *
 * @return 0 on success, otherwise a Win32 error code
 */
static UINT rdpdr_virtual_channel_event_disconnected(rdpdrPlugin* rdpdr)
{
  UINT error = 0;

  WINPR_ASSERT(rdpdr);

  if (rdpdr->OpenHandle == 0)
    return CHANNEL_RC_OK;

  if (rdpdr->queue && rdpdr->thread)
  {
    if (MessageQueue_PostQuit(rdpdr->queue, 0) &&
        (WaitForSingleObject(rdpdr->thread, INFINITE) == WAIT_FAILED))
    {
      error = GetLastError();
      WLog_Print(rdpdr->log, WLOG_ERROR, "WaitForSingleObject failed with error %" PRIu32 "!",
                 error);
      return error;
    }
  }

  if (rdpdr->thread)
    (void)CloseHandle(rdpdr->thread);
  MessageQueue_Free(rdpdr->queue);
  rdpdr->queue = NULL;
  rdpdr->thread = NULL;

  WINPR_ASSERT(rdpdr->channelEntryPoints.pVirtualChannelCloseEx);
  error = rdpdr->channelEntryPoints.pVirtualChannelCloseEx(rdpdr->InitHandle, rdpdr->OpenHandle);

  if (CHANNEL_RC_OK != error)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "pVirtualChannelCloseEx failed with %s [%08" PRIX32 "]",
               WTSErrorToString(error), error);
  }

  rdpdr->OpenHandle = 0;

  if (rdpdr->data_in)
  {
    Stream_Release(rdpdr->data_in);
    rdpdr->data_in = NULL;
  }

  if (rdpdr->devman)
  {
    devman_free(rdpdr->devman);
    rdpdr->devman = NULL;
  }

  return error;
}

static void rdpdr_virtual_channel_event_terminated(rdpdrPlugin* rdpdr)
{
  WINPR_ASSERT(rdpdr);
#if !defined(_WIN32)
  if (rdpdr->stopEvent)
  {
    (void)CloseHandle(rdpdr->stopEvent);
    rdpdr->stopEvent = NULL;
  }
#endif
  rdpdr->InitHandle = 0;
  StreamPool_Free(rdpdr->pool);
  free(rdpdr);
}

static UINT rdpdr_register_device(RdpdrClientContext* context, const RDPDR_DEVICE* device,
                                  uint32_t* pid)
{
  WINPR_ASSERT(context);
  WINPR_ASSERT(device);
  WINPR_ASSERT(pid);

  rdpdrPlugin* rdpdr = context->handle;
  WINPR_ASSERT(rdpdr);

  RDPDR_DEVICE* copy = freerdp_device_clone(device);
  if (!copy)
    return ERROR_INVALID_DATA;
  UINT rc = devman_load_device_service(rdpdr->devman, copy, rdpdr->rdpcontext);
  *pid = copy->Id;
  freerdp_device_free(copy);
  if (rc == CHANNEL_RC_OK)
    rc = rdpdr_try_send_device_list_announce_request(rdpdr);
  return rc;
}

static UINT rdpdr_unregister_device(RdpdrClientContext* context, size_t count, const uint32_t ids[])
{
  WINPR_ASSERT(context);

  rdpdrPlugin* rdpdr = context->handle;
  WINPR_ASSERT(rdpdr);

  for (size_t x = 0; x < count; x++)
  {
    const uintptr_t id = ids[x];
    devman_unregister_device(rdpdr->devman, (void*)id);
  }
  return rdpdr_send_device_list_remove_request(rdpdr, WINPR_ASSERTING_INT_CAST(uint32_t, count),
                                               ids);
}

static UINT rdpdr_virtual_channel_event_initialized(rdpdrPlugin* rdpdr,
                                                    WINPR_ATTR_UNUSED LPVOID pData,
                                                    WINPR_ATTR_UNUSED UINT32 dataLength)
{
  WINPR_ASSERT(rdpdr);
#if !defined(_WIN32)
  WINPR_ASSERT(!rdpdr->stopEvent);
  rdpdr->stopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
  WINPR_ASSERT(rdpdr->stopEvent);
#endif

  rdpdr->context.handle = rdpdr;
  rdpdr->context.RdpdrHotplugDevice = handle_hotplug;
  rdpdr->context.RdpdrRegisterDevice = rdpdr_register_device;
  rdpdr->context.RdpdrUnregisterDevice = rdpdr_unregister_device;
  return CHANNEL_RC_OK;
}

static VOID VCAPITYPE rdpdr_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle,
                                                          UINT event, LPVOID pData, UINT dataLength)
{
  UINT error = CHANNEL_RC_OK;
  rdpdrPlugin* rdpdr = (rdpdrPlugin*)lpUserParam;

  if (!rdpdr || (rdpdr->InitHandle != pInitHandle))
  {
    WLog_ERR(TAG, "error no match");
    return;
  }

  WINPR_ASSERT(pData || (dataLength == 0));

  switch (event)
  {
    case CHANNEL_EVENT_INITIALIZED:
      error = rdpdr_virtual_channel_event_initialized(rdpdr, pData, dataLength);
      break;

    case CHANNEL_EVENT_CONNECTED:
      if ((error = rdpdr_virtual_channel_event_connected(rdpdr, pData, dataLength)))
        WLog_Print(rdpdr->log, WLOG_ERROR,
                   "rdpdr_virtual_channel_event_connected failed with error %" PRIu32 "!",
                   error);

      break;

    case CHANNEL_EVENT_DISCONNECTED:
      if ((error = rdpdr_virtual_channel_event_disconnected(rdpdr)))
        WLog_Print(rdpdr->log, WLOG_ERROR,
                   "rdpdr_virtual_channel_event_disconnected failed with error %" PRIu32
                   "!",
                   error);

      break;

    case CHANNEL_EVENT_TERMINATED:
      rdpdr_virtual_channel_event_terminated(rdpdr);
      rdpdr = NULL;
      break;

    case CHANNEL_EVENT_ATTACHED:
    case CHANNEL_EVENT_DETACHED:
    default:
      WLog_Print(rdpdr->log, WLOG_ERROR, "unknown event %" PRIu32 "!", event);
      break;
  }

  if (error && rdpdr && rdpdr->rdpcontext)
    setChannelError(rdpdr->rdpcontext, error,
                    "rdpdr_virtual_channel_init_event_ex reported an error");
}

/* rdpdr is always built-in */
#define VirtualChannelEntryEx rdpdr_VirtualChannelEntryEx

FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS pEntryPoints,
                                                         PVOID pInitHandle))
{
  UINT rc = 0;
  rdpdrPlugin* rdpdr = NULL;
  CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx = NULL;

  WINPR_ASSERT(pEntryPoints);
  WINPR_ASSERT(pInitHandle);

  rdpdr = (rdpdrPlugin*)calloc(1, sizeof(rdpdrPlugin));

  if (!rdpdr)
  {
    WLog_ERR(TAG, "calloc failed!");
    return FALSE;
  }
  rdpdr->log = WLog_Get(TAG);

  rdpdr->clientExtendedPDU =
      RDPDR_DEVICE_REMOVE_PDUS | RDPDR_CLIENT_DISPLAY_NAME_PDU | RDPDR_USER_LOGGEDON_PDU;
  rdpdr->clientIOCode1 =
      RDPDR_IRP_MJ_CREATE | RDPDR_IRP_MJ_CLEANUP | RDPDR_IRP_MJ_CLOSE | RDPDR_IRP_MJ_READ |
      RDPDR_IRP_MJ_WRITE | RDPDR_IRP_MJ_FLUSH_BUFFERS | RDPDR_IRP_MJ_SHUTDOWN |
      RDPDR_IRP_MJ_DEVICE_CONTROL | RDPDR_IRP_MJ_QUERY_VOLUME_INFORMATION |
      RDPDR_IRP_MJ_SET_VOLUME_INFORMATION | RDPDR_IRP_MJ_QUERY_INFORMATION |
      RDPDR_IRP_MJ_SET_INFORMATION | RDPDR_IRP_MJ_DIRECTORY_CONTROL | RDPDR_IRP_MJ_LOCK_CONTROL |
      RDPDR_IRP_MJ_QUERY_SECURITY | RDPDR_IRP_MJ_SET_SECURITY;

  rdpdr->clientExtraFlags1 = ENABLE_ASYNCIO;

  rdpdr->pool = StreamPool_New(TRUE, 1024);
  if (!rdpdr->pool)
  {
    free(rdpdr);
    return FALSE;
  }

  rdpdr->channelDef.options =
      CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP | CHANNEL_OPTION_COMPRESS_RDP;
  (void)sprintf_s(rdpdr->channelDef.name, ARRAYSIZE(rdpdr->channelDef.name),
                  RDPDR_SVC_CHANNEL_NAME);
  rdpdr->sequenceId = 0;
  pEntryPointsEx = (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints;

  if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) &&
      (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER))
  {
    rdpdr->rdpcontext = pEntryPointsEx->context;
    if (!freerdp_settings_get_bool(rdpdr->rdpcontext->settings,
                                   FreeRDP_SynchronousStaticChannels))
      rdpdr->async = TRUE;
  }

  CopyMemory(&(rdpdr->channelEntryPoints), pEntryPoints, sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX));
  rdpdr->InitHandle = pInitHandle;
  rc = rdpdr->channelEntryPoints.pVirtualChannelInitEx(
      rdpdr, &rdpdr->context, pInitHandle, &rdpdr->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
      rdpdr_virtual_channel_init_event_ex);

  if (CHANNEL_RC_OK != rc)
  {
    WLog_Print(rdpdr->log, WLOG_ERROR, "pVirtualChannelInitEx failed with %s [%08" PRIX32 "]",
               WTSErrorToString(rc), rc);
    free(rdpdr);
    return FALSE;
  }

  return TRUE;
}

Coverage Report

Created: 2025-07-01 06:46