summaryrefslogtreecommitdiffstats
path: root/channels/rdpdr/client
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-04 01:24:41 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-05-04 01:24:41 +0000
commita9bcc81f821d7c66f623779fa5147e728eb3c388 (patch)
tree98676963bcdd537ae5908a067a8eb110b93486a6 /channels/rdpdr/client
parentInitial commit. (diff)
downloadfreerdp3-a9bcc81f821d7c66f623779fa5147e728eb3c388.tar.xz
freerdp3-a9bcc81f821d7c66f623779fa5147e728eb3c388.zip
Adding upstream version 3.3.0+dfsg1.upstream/3.3.0+dfsg1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'channels/rdpdr/client')
-rw-r--r--channels/rdpdr/client/CMakeLists.txt42
-rw-r--r--channels/rdpdr/client/devman.c236
-rw-r--r--channels/rdpdr/client/devman.h36
-rw-r--r--channels/rdpdr/client/irp.c165
-rw-r--r--channels/rdpdr/client/irp.h29
-rw-r--r--channels/rdpdr/client/rdpdr_capabilities.c259
-rw-r--r--channels/rdpdr/client/rdpdr_capabilities.h31
-rw-r--r--channels/rdpdr/client/rdpdr_main.c2342
-rw-r--r--channels/rdpdr/client/rdpdr_main.h121
9 files changed, 3261 insertions, 0 deletions
diff --git a/channels/rdpdr/client/CMakeLists.txt b/channels/rdpdr/client/CMakeLists.txt
new file mode 100644
index 0000000..7d11574
--- /dev/null
+++ b/channels/rdpdr/client/CMakeLists.txt
@@ -0,0 +1,42 @@
+# FreeRDP: A Remote Desktop Protocol Implementation
+# FreeRDP cmake build script
+#
+# Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
+# Copyright 2016 Inuvika Inc.
+# 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.
+
+define_channel_client("rdpdr")
+
+set(${MODULE_PREFIX}_SRCS
+ irp.c
+ irp.h
+ devman.c
+ devman.h
+ rdpdr_main.c
+ rdpdr_main.h
+ rdpdr_capabilities.c
+ rdpdr_capabilities.h
+)
+
+set(${MODULE_PREFIX}_LIBS
+ winpr freerdp
+)
+if(APPLE AND (NOT IOS))
+ find_library(CORESERVICES_LIBRARY CoreServices)
+ list(APPEND ${MODULE_PREFIX}_LIBS
+ ${CORESERVICES_LIBRARY}
+ )
+endif()
+add_channel_client_library(${MODULE_PREFIX} ${MODULE_NAME} ${CHANNEL_NAME} FALSE "VirtualChannelEntryEx")
diff --git a/channels/rdpdr/client/devman.c b/channels/rdpdr/client/devman.c
new file mode 100644
index 0000000..7c9e3ff
--- /dev/null
+++ b/channels/rdpdr/client/devman.c
@@ -0,0 +1,236 @@
+/**
+ * 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 Thincast Technologies GmbH
+ * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
+ * Copyright 2016 Armin Novak <armin.novak@gmail.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 <winpr/crt.h>
+#include <winpr/stream.h>
+
+#include <freerdp/types.h>
+#include <freerdp/addin.h>
+#include <freerdp/client/channels.h>
+#include <freerdp/channels/log.h>
+
+#include "rdpdr_main.h"
+
+#include "devman.h"
+
+#define TAG CHANNELS_TAG("rdpdr.client")
+
+static void devman_device_free(void* obj)
+{
+ DEVICE* device = (DEVICE*)obj;
+
+ if (!device)
+ return;
+
+ IFCALL(device->Free, device);
+}
+
+DEVMAN* devman_new(rdpdrPlugin* rdpdr)
+{
+ DEVMAN* devman = NULL;
+
+ if (!rdpdr)
+ return NULL;
+
+ devman = (DEVMAN*)calloc(1, sizeof(DEVMAN));
+
+ if (!devman)
+ {
+ WLog_Print(rdpdr->log, WLOG_INFO, "calloc failed!");
+ return NULL;
+ }
+
+ devman->plugin = (void*)rdpdr;
+ devman->id_sequence = 1;
+ devman->devices = ListDictionary_New(TRUE);
+
+ if (!devman->devices)
+ {
+ WLog_Print(rdpdr->log, WLOG_INFO, "ListDictionary_New failed!");
+ free(devman);
+ return NULL;
+ }
+
+ ListDictionary_ValueObject(devman->devices)->fnObjectFree = devman_device_free;
+ return devman;
+}
+
+void devman_free(DEVMAN* devman)
+{
+ ListDictionary_Free(devman->devices);
+ free(devman);
+}
+
+void devman_unregister_device(DEVMAN* devman, void* key)
+{
+ DEVICE* device = NULL;
+
+ if (!devman || !key)
+ return;
+
+ device = (DEVICE*)ListDictionary_Take(devman->devices, key);
+
+ if (device)
+ devman_device_free(device);
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT devman_register_device(DEVMAN* devman, DEVICE* device)
+{
+ void* key = NULL;
+
+ if (!devman || !device)
+ return ERROR_INVALID_PARAMETER;
+
+ device->id = devman->id_sequence++;
+ key = (void*)(size_t)device->id;
+
+ if (!ListDictionary_Add(devman->devices, key, device))
+ {
+ WLog_INFO(TAG, "ListDictionary_Add failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+DEVICE* devman_get_device_by_id(DEVMAN* devman, UINT32 id)
+{
+ DEVICE* device = NULL;
+ void* key = (void*)(size_t)id;
+
+ if (!devman)
+ {
+ WLog_ERR(TAG, "device manager=%p", devman);
+ return NULL;
+ }
+
+ device = (DEVICE*)ListDictionary_GetItemValue(devman->devices, key);
+ if (!device)
+ WLog_WARN(TAG, "could not find device ID 0x%08" PRIx32, id);
+ return device;
+}
+
+DEVICE* devman_get_device_by_type(DEVMAN* devman, UINT32 type)
+{
+ DEVICE* device = NULL;
+ ULONG_PTR* keys = NULL;
+
+ if (!devman)
+ return NULL;
+
+ ListDictionary_Lock(devman->devices);
+ const size_t count = ListDictionary_GetKeys(devman->devices, &keys);
+
+ for (size_t x = 0; x < count; x++)
+ {
+ DEVICE* cur = (DEVICE*)ListDictionary_GetItemValue(devman->devices, (void*)keys[x]);
+
+ if (!cur)
+ continue;
+
+ if (cur->type != type)
+ continue;
+
+ device = cur;
+ break;
+ }
+
+ free(keys);
+ ListDictionary_Unlock(devman->devices);
+ return device;
+}
+
+static const char DRIVE_SERVICE_NAME[] = "drive";
+static const char PRINTER_SERVICE_NAME[] = "printer";
+static const char SMARTCARD_SERVICE_NAME[] = "smartcard";
+static const char SERIAL_SERVICE_NAME[] = "serial";
+static const char PARALLEL_SERVICE_NAME[] = "parallel";
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+UINT devman_load_device_service(DEVMAN* devman, const RDPDR_DEVICE* device, rdpContext* rdpcontext)
+{
+ const char* ServiceName = NULL;
+ DEVICE_SERVICE_ENTRY_POINTS ep;
+ PDEVICE_SERVICE_ENTRY entry = NULL;
+ union
+ {
+ const RDPDR_DEVICE* cdp;
+ RDPDR_DEVICE* dp;
+ } devconv;
+
+ devconv.cdp = device;
+ if (!devman || !device || !rdpcontext)
+ return ERROR_INVALID_PARAMETER;
+
+ if (device->Type == RDPDR_DTYP_FILESYSTEM)
+ ServiceName = DRIVE_SERVICE_NAME;
+ else if (device->Type == RDPDR_DTYP_PRINT)
+ ServiceName = PRINTER_SERVICE_NAME;
+ else if (device->Type == RDPDR_DTYP_SMARTCARD)
+ ServiceName = SMARTCARD_SERVICE_NAME;
+ else if (device->Type == RDPDR_DTYP_SERIAL)
+ ServiceName = SERIAL_SERVICE_NAME;
+ else if (device->Type == RDPDR_DTYP_PARALLEL)
+ ServiceName = PARALLEL_SERVICE_NAME;
+
+ if (!ServiceName)
+ {
+ WLog_INFO(TAG, "ServiceName %s did not match!", ServiceName);
+ return ERROR_INVALID_NAME;
+ }
+
+ if (device->Name)
+ WLog_INFO(TAG, "Loading device service %s [%s] (static)", ServiceName, device->Name);
+ else
+ WLog_INFO(TAG, "Loading device service %s (static)", ServiceName);
+
+ entry = (PDEVICE_SERVICE_ENTRY)freerdp_load_channel_addin_entry(ServiceName, NULL,
+ "DeviceServiceEntry", 0);
+
+ if (!entry)
+ {
+ WLog_INFO(TAG, "freerdp_load_channel_addin_entry failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ ep.devman = devman;
+ ep.RegisterDevice = devman_register_device;
+ ep.device = devconv.dp;
+ ep.rdpcontext = rdpcontext;
+ return entry(&ep);
+}
diff --git a/channels/rdpdr/client/devman.h b/channels/rdpdr/client/devman.h
new file mode 100644
index 0000000..2e6019e
--- /dev/null
+++ b/channels/rdpdr/client/devman.h
@@ -0,0 +1,36 @@
+/**
+ * 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 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.
+ */
+
+#ifndef FREERDP_CHANNEL_RDPDR_CLIENT_DEVMAN_H
+#define FREERDP_CHANNEL_RDPDR_CLIENT_DEVMAN_H
+
+#include "rdpdr_main.h"
+
+void devman_unregister_device(DEVMAN* devman, void* key);
+UINT devman_load_device_service(DEVMAN* devman, const RDPDR_DEVICE* device, rdpContext* rdpcontext);
+DEVICE* devman_get_device_by_id(DEVMAN* devman, UINT32 id);
+DEVICE* devman_get_device_by_type(DEVMAN* devman, UINT32 type);
+
+DEVMAN* devman_new(rdpdrPlugin* rdpdr);
+void devman_free(DEVMAN* devman);
+
+#endif /* FREERDP_CHANNEL_RDPDR_CLIENT_DEVMAN_H */
diff --git a/channels/rdpdr/client/irp.c b/channels/rdpdr/client/irp.c
new file mode 100644
index 0000000..eeba80d
--- /dev/null
+++ b/channels/rdpdr/client/irp.c
@@ -0,0 +1,165 @@
+/**
+ * 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 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 <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <winpr/crt.h>
+#include <winpr/stream.h>
+
+#include <freerdp/utils/rdpdr_utils.h>
+
+#include "rdpdr_main.h"
+#include "devman.h"
+#include "irp.h"
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT irp_free(IRP* irp)
+{
+ if (!irp)
+ return CHANNEL_RC_OK;
+
+ if (irp->input)
+ Stream_Release(irp->input);
+ if (irp->output)
+ Stream_Release(irp->output);
+
+ winpr_aligned_free(irp);
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT irp_complete(IRP* irp)
+{
+ size_t pos = 0;
+ rdpdrPlugin* rdpdr = NULL;
+ UINT error = 0;
+
+ WINPR_ASSERT(irp);
+ WINPR_ASSERT(irp->output);
+ WINPR_ASSERT(irp->devman);
+
+ rdpdr = (rdpdrPlugin*)irp->devman->plugin;
+ WINPR_ASSERT(rdpdr);
+
+ pos = Stream_GetPosition(irp->output);
+ Stream_SetPosition(irp->output, RDPDR_DEVICE_IO_RESPONSE_LENGTH - 4);
+ Stream_Write_UINT32(irp->output, irp->IoStatus); /* IoStatus (4 bytes) */
+ Stream_SetPosition(irp->output, pos);
+
+ error = rdpdr_send(rdpdr, irp->output);
+ irp->output = NULL;
+
+ irp_free(irp);
+ return error;
+}
+
+IRP* irp_new(DEVMAN* devman, wStreamPool* pool, wStream* s, wLog* log, UINT* error)
+{
+ IRP* irp = NULL;
+ DEVICE* device = NULL;
+ UINT32 DeviceId = 0;
+
+ WINPR_ASSERT(devman);
+ WINPR_ASSERT(pool);
+ WINPR_ASSERT(s);
+
+ if (!Stream_CheckAndLogRequiredLengthWLog(log, s, 20))
+ {
+ if (error)
+ *error = ERROR_INVALID_DATA;
+ return NULL;
+ }
+
+ Stream_Read_UINT32(s, DeviceId); /* DeviceId (4 bytes) */
+ device = devman_get_device_by_id(devman, DeviceId);
+
+ if (!device)
+ {
+ if (error)
+ *error = ERROR_DEV_NOT_EXIST;
+
+ return NULL;
+ }
+
+ irp = (IRP*)winpr_aligned_malloc(sizeof(IRP), MEMORY_ALLOCATION_ALIGNMENT);
+
+ if (!irp)
+ {
+ WLog_Print(log, WLOG_ERROR, "_aligned_malloc failed!");
+ if (error)
+ *error = CHANNEL_RC_NO_MEMORY;
+ return NULL;
+ }
+
+ ZeroMemory(irp, sizeof(IRP));
+
+ Stream_Read_UINT32(s, irp->FileId); /* FileId (4 bytes) */
+ Stream_Read_UINT32(s, irp->CompletionId); /* CompletionId (4 bytes) */
+ Stream_Read_UINT32(s, irp->MajorFunction); /* MajorFunction (4 bytes) */
+ Stream_Read_UINT32(s, irp->MinorFunction); /* MinorFunction (4 bytes) */
+
+ Stream_AddRef(s);
+ irp->input = s;
+ irp->device = device;
+ irp->devman = devman;
+
+ irp->output = StreamPool_Take(pool, 256);
+ if (!irp->output)
+ {
+ WLog_Print(log, WLOG_ERROR, "Stream_New failed!");
+ irp_free(irp);
+ if (error)
+ *error = CHANNEL_RC_NO_MEMORY;
+ return NULL;
+ }
+
+ if (!rdpdr_write_iocompletion_header(irp->output, DeviceId, irp->CompletionId, 0))
+ {
+ irp_free(irp);
+ if (error)
+ *error = CHANNEL_RC_NO_MEMORY;
+ return NULL;
+ }
+
+ irp->Complete = irp_complete;
+ irp->Discard = irp_free;
+
+ irp->thread = NULL;
+ irp->cancelled = FALSE;
+
+ if (error)
+ *error = CHANNEL_RC_OK;
+
+ return irp;
+}
diff --git a/channels/rdpdr/client/irp.h b/channels/rdpdr/client/irp.h
new file mode 100644
index 0000000..0538295
--- /dev/null
+++ b/channels/rdpdr/client/irp.h
@@ -0,0 +1,29 @@
+/**
+ * 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>
+ *
+ * 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.
+ */
+
+#ifndef FREERDP_CHANNEL_RDPDR_CLIENT_IRP_H
+#define FREERDP_CHANNEL_RDPDR_CLIENT_IRP_H
+
+#include <winpr/wlog.h>
+#include "rdpdr_main.h"
+
+IRP* irp_new(DEVMAN* devman, wStreamPool* pool, wStream* s, wLog* log, UINT* error);
+
+#endif /* FREERDP_CHANNEL_RDPDR_CLIENT_IRP_H */
diff --git a/channels/rdpdr/client/rdpdr_capabilities.c b/channels/rdpdr/client/rdpdr_capabilities.c
new file mode 100644
index 0000000..e094cc7
--- /dev/null
+++ b/channels/rdpdr/client/rdpdr_capabilities.c
@@ -0,0 +1,259 @@
+/**
+ * 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>
+ *
+ * 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 <winpr/crt.h>
+#include <winpr/stream.h>
+
+#include <freerdp/utils/rdpdr_utils.h>
+
+#include "rdpdr_main.h"
+#include "rdpdr_capabilities.h"
+
+#define RDPDR_CAPABILITY_HEADER_LENGTH 8
+
+/* Output device direction general capability set */
+static void rdpdr_write_general_capset(rdpdrPlugin* rdpdr, wStream* s)
+{
+ WINPR_UNUSED(rdpdr);
+ const RDPDR_CAPABILITY_HEADER header = { CAP_GENERAL_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH + 36,
+ GENERAL_CAPABILITY_VERSION_02 };
+
+ const UINT32 ioCode1 = rdpdr->clientIOCode1 & rdpdr->serverIOCode1;
+ const UINT32 ioCode2 = rdpdr->clientIOCode2 & rdpdr->serverIOCode2;
+
+ rdpdr_write_capset_header(rdpdr->log, s, &header);
+ Stream_Write_UINT32(s, rdpdr->clientOsType); /* osType, ignored on receipt */
+ Stream_Write_UINT32(s, rdpdr->clientOsVersion); /* osVersion, unused and must be set to zero */
+ Stream_Write_UINT16(s, rdpdr->clientVersionMajor); /* protocolMajorVersion, must be set to 1 */
+ Stream_Write_UINT16(s, rdpdr->clientVersionMinor); /* protocolMinorVersion */
+ Stream_Write_UINT32(s, ioCode1); /* ioCode1 */
+ Stream_Write_UINT32(s, ioCode2); /* ioCode2, must be set to zero, reserved for future use */
+ Stream_Write_UINT32(s, rdpdr->clientExtendedPDU); /* extendedPDU */
+ Stream_Write_UINT32(s, rdpdr->clientExtraFlags1); /* extraFlags1 */
+ Stream_Write_UINT32(
+ s,
+ rdpdr->clientExtraFlags2); /* extraFlags2, must be set to zero, reserved for future use */
+ Stream_Write_UINT32(
+ s, rdpdr->clientSpecialTypeDeviceCap); /* SpecialTypeDeviceCap, number of special devices to
+ be redirected before logon */
+}
+
+/* Process device direction general capability set */
+static UINT rdpdr_process_general_capset(rdpdrPlugin* rdpdr, wStream* s,
+ const RDPDR_CAPABILITY_HEADER* header)
+{
+ WINPR_ASSERT(header);
+
+ if (header->CapabilityLength != 36)
+ {
+ WLog_Print(rdpdr->log, WLOG_ERROR,
+ "CAP_GENERAL_TYPE::CapabilityLength expected 36, got %" PRIu32,
+ header->CapabilityLength);
+ return ERROR_INVALID_DATA;
+ }
+ if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 36))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT32(s, rdpdr->serverOsType); /* osType, ignored on receipt */
+ Stream_Read_UINT32(s, rdpdr->serverOsVersion); /* osVersion, unused and must be set to zero */
+ Stream_Read_UINT16(s, rdpdr->serverVersionMajor); /* protocolMajorVersion, must be set to 1 */
+ Stream_Read_UINT16(s, rdpdr->serverVersionMinor); /* protocolMinorVersion */
+ Stream_Read_UINT32(s, rdpdr->serverIOCode1); /* ioCode1 */
+ Stream_Read_UINT32(
+ s, rdpdr->serverIOCode2); /* ioCode2, must be set to zero, reserved for future use */
+ Stream_Read_UINT32(s, rdpdr->serverExtendedPDU); /* extendedPDU */
+ Stream_Read_UINT32(s, rdpdr->serverExtraFlags1); /* extraFlags1 */
+ Stream_Read_UINT32(
+ s,
+ rdpdr->serverExtraFlags2); /* extraFlags2, must be set to zero, reserved for future use */
+ Stream_Read_UINT32(
+ s, rdpdr->serverSpecialTypeDeviceCap); /* SpecialTypeDeviceCap, number of special devices to
+ be redirected before logon */
+ return CHANNEL_RC_OK;
+}
+
+/* Output printer direction capability set */
+static void rdpdr_write_printer_capset(rdpdrPlugin* rdpdr, wStream* s)
+{
+ WINPR_UNUSED(rdpdr);
+ const RDPDR_CAPABILITY_HEADER header = { CAP_PRINTER_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
+ PRINT_CAPABILITY_VERSION_01 };
+ rdpdr_write_capset_header(rdpdr->log, s, &header);
+}
+
+/* Process printer direction capability set */
+static UINT rdpdr_process_printer_capset(rdpdrPlugin* rdpdr, wStream* s,
+ const RDPDR_CAPABILITY_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ Stream_Seek(s, header->CapabilityLength);
+ return CHANNEL_RC_OK;
+}
+
+/* Output port redirection capability set */
+static void rdpdr_write_port_capset(rdpdrPlugin* rdpdr, wStream* s)
+{
+ WINPR_UNUSED(rdpdr);
+ const RDPDR_CAPABILITY_HEADER header = { CAP_PORT_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
+ PORT_CAPABILITY_VERSION_01 };
+ rdpdr_write_capset_header(rdpdr->log, s, &header);
+}
+
+/* Process port redirection capability set */
+static UINT rdpdr_process_port_capset(rdpdrPlugin* rdpdr, wStream* s,
+ const RDPDR_CAPABILITY_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ Stream_Seek(s, header->CapabilityLength);
+ return CHANNEL_RC_OK;
+}
+
+/* Output drive redirection capability set */
+static void rdpdr_write_drive_capset(rdpdrPlugin* rdpdr, wStream* s)
+{
+ WINPR_UNUSED(rdpdr);
+ const RDPDR_CAPABILITY_HEADER header = { CAP_DRIVE_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
+ DRIVE_CAPABILITY_VERSION_02 };
+ rdpdr_write_capset_header(rdpdr->log, s, &header);
+}
+
+/* Process drive redirection capability set */
+static UINT rdpdr_process_drive_capset(rdpdrPlugin* rdpdr, wStream* s,
+ const RDPDR_CAPABILITY_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ Stream_Seek(s, header->CapabilityLength);
+ return CHANNEL_RC_OK;
+}
+
+/* Output smart card redirection capability set */
+static void rdpdr_write_smartcard_capset(rdpdrPlugin* rdpdr, wStream* s)
+{
+ WINPR_UNUSED(rdpdr);
+ const RDPDR_CAPABILITY_HEADER header = { CAP_SMARTCARD_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
+ SMARTCARD_CAPABILITY_VERSION_01 };
+ rdpdr_write_capset_header(rdpdr->log, s, &header);
+}
+
+/* Process smartcard redirection capability set */
+static UINT rdpdr_process_smartcard_capset(rdpdrPlugin* rdpdr, wStream* s,
+ const RDPDR_CAPABILITY_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ Stream_Seek(s, header->CapabilityLength);
+ return CHANNEL_RC_OK;
+}
+
+UINT rdpdr_process_capability_request(rdpdrPlugin* rdpdr, wStream* s)
+{
+ UINT status = CHANNEL_RC_OK;
+ UINT16 numCapabilities = 0;
+
+ if (!rdpdr || !s)
+ return CHANNEL_RC_NULL_DATA;
+
+ WINPR_ASSERT(rdpdr->state == RDPDR_CHANNEL_STATE_SERVER_CAPS);
+ rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_CLIENT_CAPS);
+
+ if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 4))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, numCapabilities);
+ Stream_Seek(s, 2); /* pad (2 bytes) */
+
+ for (UINT16 i = 0; i < numCapabilities; i++)
+ {
+ RDPDR_CAPABILITY_HEADER header = { 0 };
+ UINT error = rdpdr_read_capset_header(rdpdr->log, s, &header);
+ if (error != CHANNEL_RC_OK)
+ return error;
+
+ switch (header.CapabilityType)
+ {
+ case CAP_GENERAL_TYPE:
+ status = rdpdr_process_general_capset(rdpdr, s, &header);
+ break;
+
+ case CAP_PRINTER_TYPE:
+ status = rdpdr_process_printer_capset(rdpdr, s, &header);
+ break;
+
+ case CAP_PORT_TYPE:
+ status = rdpdr_process_port_capset(rdpdr, s, &header);
+ break;
+
+ case CAP_DRIVE_TYPE:
+ status = rdpdr_process_drive_capset(rdpdr, s, &header);
+ break;
+
+ case CAP_SMARTCARD_TYPE:
+ status = rdpdr_process_smartcard_capset(rdpdr, s, &header);
+ break;
+
+ default:
+
+ break;
+ }
+
+ if (status != CHANNEL_RC_OK)
+ return status;
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+UINT rdpdr_send_capability_response(rdpdrPlugin* rdpdr)
+{
+ wStream* s = NULL;
+
+ WINPR_ASSERT(rdpdr);
+ 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);
+ Stream_Write_UINT16(s, PAKID_CORE_CLIENT_CAPABILITY);
+ Stream_Write_UINT16(s, 5); /* numCapabilities */
+ Stream_Write_UINT16(s, 0); /* pad */
+ rdpdr_write_general_capset(rdpdr, s);
+ rdpdr_write_printer_capset(rdpdr, s);
+ rdpdr_write_port_capset(rdpdr, s);
+ rdpdr_write_drive_capset(rdpdr, s);
+ rdpdr_write_smartcard_capset(rdpdr, s);
+ return rdpdr_send(rdpdr, s);
+}
diff --git a/channels/rdpdr/client/rdpdr_capabilities.h b/channels/rdpdr/client/rdpdr_capabilities.h
new file mode 100644
index 0000000..d4e1ecb
--- /dev/null
+++ b/channels/rdpdr/client/rdpdr_capabilities.h
@@ -0,0 +1,31 @@
+/**
+ * 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 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.
+ */
+
+#ifndef FREERDP_CHANNEL_RDPDR_CLIENT_CAPABILITIES_H
+#define FREERDP_CHANNEL_RDPDR_CLIENT_CAPABILITIES_H
+
+#include "rdpdr_main.h"
+
+UINT rdpdr_process_capability_request(rdpdrPlugin* rdpdr, wStream* s);
+UINT rdpdr_send_capability_response(rdpdrPlugin* rdpdr);
+
+#endif /* FREERDP_CHANNEL_RDPDR_CLIENT_CAPABILITIES_H */
diff --git a/channels/rdpdr/client/rdpdr_main.c b/channels/rdpdr/client/rdpdr_main.c
new file mode 100644
index 0000000..53f5011
--- /dev/null
+++ b/channels/rdpdr/client/rdpdr_main.c
@@ -0,0 +1,2342 @@
+/**
+ * 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 <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;
+
+ rc = devman_load_device_service(rdpdr->devman, drive.device, rdpdr->rdpcontext);
+ if (rc != CHANNEL_RC_OK)
+ goto fail;
+
+fail:
+ freerdp_device_free(drive.device);
+ return rc;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT rdpdr_send_device_list_remove_request(rdpdrPlugin* rdpdr, UINT32 count, 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 void first_hotplug(rdpdrPlugin* rdpdr)
+{
+}
+
+static DWORD WINAPI drive_hotplug_thread_func(LPVOID arg)
+{
+ return CHANNEL_RC_OK;
+}
+
+static UINT drive_hotplug_thread_terminate(rdpdrPlugin* rdpdr)
+{
+ return CHANNEL_RC_OK;
+}
+
+#elif defined(_WIN32)
+
+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);
+ rdpdr_try_send_device_list_announce_request(rdpdr);
+ }
+ }
+
+ 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;
+ int count;
+ 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;
+ count = ListDictionary_GetKeys(rdpdr->devman->devices, &keys);
+
+ for (int 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)
+ {
+ devman_unregister_device(rdpdr->devman, (void*)keys[j]);
+ ids[0] = keys[j];
+
+ if ((error = rdpdr_send_device_list_remove_request(
+ rdpdr, 1, ids)))
+ {
+ // dont 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(rdpdrPlugin* rdpdr)
+{
+ struct dirent* pDirent;
+ DIR* pDir;
+ char fullpath[PATH_MAX];
+ char* szdir = (char*)"/Volumes";
+ struct stat buf;
+ hotplug_dev dev_array[MAX_USB_DEVICES];
+ int count;
+ DEVICE_DRIVE_EXT* device_ext;
+ ULONG_PTR* keys = NULL;
+ int size = 0;
+ UINT error;
+ UINT32 ids[1];
+ 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] != '.')
+ {
+ 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 plugable 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)
+ {
+ devman_unregister_device(rdpdr->devman, (void*)keys[j]);
+ ids[0] = keys[j];
+
+ if ((error = rdpdr_send_device_list_remove_request(rdpdr, 1, ids)))
+ {
+ 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)
+ {
+ if ((error = handle_hotplug(rdpdr)))
+ {
+ WLog_Print(rdpdr->log, WLOG_ERROR, "handle_hotplug failed with error %" PRIu32 "!",
+ error);
+ }
+ else
+ rdpdr_try_send_device_list_announce_request(rdpdr);
+
+ return;
+ }
+ }
+}
+
+static void first_hotplug(rdpdrPlugin* rdpdr)
+{
+ UINT error;
+
+ if ((error = handle_hotplug(rdpdr)))
+ {
+ WLog_Print(rdpdr->log, WLOG_ERROR, "handle_hotplug failed with error %" PRIu32 "!", error);
+ }
+}
+
+static DWORD WINAPI drive_hotplug_thread_func(LPVOID arg)
+{
+ rdpdrPlugin* rdpdr;
+ FSEventStreamRef fsev;
+ rdpdr = (rdpdrPlugin*)arg;
+ CFStringRef path = CFSTR("/Volumes/");
+ CFArrayRef pathsToWatch = CFArrayCreate(kCFAllocatorMalloc, (const void**)&path, 1, NULL);
+ FSEventStreamContext ctx = { 0 };
+
+ ctx.info = arg;
+
+ WINPR_ASSERT(!rdpdr->stopEvent);
+ rdpdr->stopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
+ if (!rdpdr->stopEvent)
+ goto out;
+
+ fsev =
+ FSEventStreamCreate(kCFAllocatorMalloc, drive_hotplug_fsevent_callback, &ctx, pathsToWatch,
+ kFSEventStreamEventIdSinceNow, 1, kFSEventStreamCreateFlagNone);
+
+ rdpdr->runLoop = CFRunLoopGetCurrent();
+ FSEventStreamScheduleWithRunLoop(fsev, rdpdr->runLoop, kCFRunLoopDefaultMode);
+ FSEventStreamStart(fsev);
+ CFRunLoopRun();
+ FSEventStreamStop(fsev);
+ FSEventStreamRelease(fsev);
+out:
+ if (rdpdr->stopEvent)
+ {
+ CloseHandle(rdpdr->stopEvent);
+ rdpdr->stopEvent = NULL;
+ }
+ 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;
+
+ if (strstr(location, "%lu"))
+ snprintf(buffer, sizeof(buffer), location, (unsigned long)uid);
+ else if (strstr(location, "%s"))
+ snprintf(buffer, sizeof(buffer), location, uname);
+ else
+ snprintf(buffer, sizeof(buffer), "%s", location);
+
+ 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 UINT handle_platform_mounts_linux(wLog* log, hotplug_dev* dev_array, size_t* size)
+{
+ FILE* f = NULL;
+ 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(f)) != NULL)
+ {
+ handle_mountpoint(dev_array, size, ent->mnt_dir);
+ }
+ 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));
+
+ 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 plugable 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)
+ {
+ UINT error = 0;
+ UINT32 ids[1] = { key };
+
+ WINPR_ASSERT(arg->rdpdr->devman);
+ devman_unregister_device(arg->rdpdr->devman, (void*)key);
+ WINPR_ASSERT(key <= UINT32_MAX);
+
+ error = rdpdr_send_device_list_remove_request(arg->rdpdr, 1, 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(rdpdrPlugin* rdpdr)
+{
+ 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 = 0;
+
+ WINPR_ASSERT(rdpdr);
+ if ((error = handle_hotplug(rdpdr)))
+ {
+ 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 = NULL;
+ UINT error = 0;
+ rdpdr = (rdpdrPlugin*)arg;
+
+ WINPR_ASSERT(rdpdr);
+
+ WINPR_ASSERT(!rdpdr->stopEvent);
+ rdpdr->stopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
+ if (!rdpdr->stopEvent)
+ goto out;
+
+ while (WaitForSingleObject(rdpdr->stopEvent, 1000) == WAIT_TIMEOUT)
+ {
+ error = handle_hotplug(rdpdr);
+ switch (error)
+ {
+ case ERROR_DISK_CHANGE:
+ rdpdr_try_send_device_list_announce_request(rdpdr);
+ 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:
+ error = GetLastError();
+ if (error && rdpdr->rdpcontext)
+ setChannelError(rdpdr->rdpcontext, error, "reported an error");
+
+ if (rdpdr->stopEvent)
+ {
+ CloseHandle(rdpdr->stopEvent);
+ rdpdr->stopEvent = NULL;
+ }
+
+ 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)
+ SetEvent(rdpdr->stopEvent);
+#endif
+#ifdef __MACOSX__
+ CFRunLoopStop(rdpdr->runLoop);
+#endif
+
+ if (WaitForSingleObject(rdpdr->hotplugThread, INFINITE) == WAIT_FAILED)
+ {
+ error = GetLastError();
+ WLog_Print(rdpdr->log, WLOG_ERROR, "WaitForSingleObject failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ CloseHandle(rdpdr->hotplugThread);
+ rdpdr->hotplugThread = NULL;
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+#endif
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT rdpdr_process_connect(rdpdrPlugin* rdpdr)
+{
+ UINT error = CHANNEL_RC_OK;
+
+ 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);
+ strncpy(rdpdr->computerName, name, sizeof(rdpdr->computerName) - 1);
+
+ for (UINT32 index = 0; index < freerdp_settings_get_uint32(settings, FreeRDP_DeviceCount);
+ index++)
+ {
+ const RDPDR_DEVICE* device =
+ freerdp_settings_get_pointer_array(settings, FreeRDP_DeviceArray, index);
+
+ if (device->Type == RDPDR_DTYP_FILESYSTEM)
+ {
+ const char DynamicDrives[] = "DynamicDrives";
+ const RDPDR_DRIVE* drive = (const RDPDR_DRIVE*)device;
+ BOOL hotplugAll = strncmp(drive->Path, "*", 2) == 0;
+ BOOL hotplugLater = strncmp(drive->Path, DynamicDrives, sizeof(DynamicDrives)) == 0;
+
+ if (drive->Path && (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;
+ }
+ }
+
+ if ((error = devman_load_device_service(rdpdr->devman, device, rdpdr->rdpcontext)))
+ {
+ WLog_Print(rdpdr->log, WLOG_ERROR,
+ "devman_load_device_service failed with error %" PRIu32 "!", error);
+ return error;
+ }
+ }
+
+ return error;
+}
+
+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, (UINT32)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);
+
+ if (!rdpdr->computerName[0])
+ {
+ DWORD size = sizeof(rdpdr->computerName) - 1;
+ GetComputerNameA(rdpdr->computerName, &size);
+ }
+
+ WINPR_ASSERT(rdpdr->computerName);
+ computerNameW = ConvertUtf8ToWCharAlloc(rdpdr->computerName, &computerNameLenW);
+ computerNameLenW *= sizeof(WCHAR);
+ WINPR_ASSERT(computerNameLenW >= 0);
+
+ 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, (size_t)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 ": %s (type=%" PRIu32 " id=%" 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_USER_LOGGEDON)
+ {
+ 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, TRUE);
+}
+
+static UINT dummy_irp_response(rdpdrPlugin* rdpdr, wStream* s)
+{
+ wStream* output = NULL;
+ UINT32 DeviceId = 0;
+ UINT32 FileId = 0;
+ UINT32 CompletionId = 0;
+
+ WINPR_ASSERT(rdpdr);
+ WINPR_ASSERT(s);
+
+ 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" */
+
+ Stream_Read_UINT32(s, DeviceId); /* DeviceId (4 bytes) */
+ Stream_Read_UINT32(s, FileId); /* FileId (4 bytes) */
+ Stream_Read_UINT32(s, CompletionId); /* CompletionId (4 bytes) */
+
+ if (!rdpdr_write_iocompletion_header(output, DeviceId, CompletionId,
+ (UINT32)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)
+ {
+ WLog_Print(rdpdr->log, WLOG_ERROR, "irp_new failed with %" PRIu32 "!", error);
+
+ if (error == CHANNEL_RC_OK || (error == ERROR_DEV_NOT_EXIST && rdpdr->ignoreInvalidDevices))
+ {
+ return dummy_irp_response(rdpdr, s);
+ }
+
+ 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;
+ 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, PAKID_CORE_CLIENTID_CONFIRM,
+ PAKID_CORE_USER_LOGGEDON);
+ case PAKID_CORE_CLIENTID_CONFIRM:
+ return rdpdr_state_check(rdpdr, packetid, RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM, 3,
+ 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);
+ }
+ }
+}
+
+/**
+ * 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:
+ 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);
+ }
+
+ break;
+
+ case PAKID_CORE_CLIENTID_CONFIRM:
+ if ((error = rdpdr_process_server_clientid_confirm(rdpdr, s)))
+ {
+ }
+ else if ((error = rdpdr_send_device_list_announce_request(rdpdr, FALSE)))
+ {
+ WLog_Print(
+ rdpdr->log, WLOG_ERROR,
+ "rdpdr_send_device_list_announce_request failed with error %" PRIu32 "",
+ error);
+ }
+ else if (!rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_READY))
+ {
+ error = ERROR_INTERNAL_ERROR;
+ }
+ break;
+
+ case PAKID_CORE_USER_LOGGEDON:
+ 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 (!rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_READY))
+ {
+ 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)
+{
+ UINT status = 0;
+ rdpdrPlugin* plugin = (rdpdrPlugin*)rdpdr;
+
+ if (!rdpdr || !s)
+ {
+ Stream_Release(s);
+ return CHANNEL_RC_NULL_DATA;
+ }
+
+ if (!plugin)
+ {
+ Stream_Release(s);
+ status = CHANNEL_RC_BAD_INIT_HANDLE;
+ }
+ else
+ {
+ const size_t pos = Stream_GetPosition(s);
+ rdpdr_dump_send_packet(rdpdr->log, WLOG_TRACE, s, "[rdpdr-channel] send");
+ status = plugin->channelEntryPoints.pVirtualChannelWriteEx(
+ plugin->InitHandle, plugin->OpenHandle, Stream_Buffer(s), 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;
+ }
+
+ if (error && rdpdr && rdpdr->rdpcontext)
+ setChannelError(rdpdr->rdpcontext, error,
+ "rdpdr_virtual_channel_open_event_ex reported an error");
+
+ return;
+}
+
+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)
+ 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);
+ rdpdr->InitHandle = 0;
+ StreamPool_Free(rdpdr->pool);
+ free(rdpdr);
+}
+
+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:
+ 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;
+ 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, NULL, 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;
+}
diff --git a/channels/rdpdr/client/rdpdr_main.h b/channels/rdpdr/client/rdpdr_main.h
new file mode 100644
index 0000000..91131cb
--- /dev/null
+++ b/channels/rdpdr/client/rdpdr_main.h
@@ -0,0 +1,121 @@
+/**
+ * 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 Thincast Technologies GmbH
+ * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
+ * Copyright 2016 Inuvika Inc.
+ * 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.
+ */
+
+#ifndef FREERDP_CHANNEL_RDPDR_CLIENT_MAIN_H
+#define FREERDP_CHANNEL_RDPDR_CLIENT_MAIN_H
+
+#include <winpr/crt.h>
+#include <winpr/synch.h>
+#include <winpr/thread.h>
+#include <winpr/stream.h>
+#include <winpr/collections.h>
+
+#include <freerdp/api.h>
+#include <freerdp/svc.h>
+#include <freerdp/addin.h>
+
+#include <freerdp/channels/rdpdr.h>
+#include <freerdp/channels/log.h>
+
+#ifdef __MACOSX__
+#include <CoreServices/CoreServices.h>
+#endif
+
+enum RDPDR_CHANNEL_STATE
+{
+ RDPDR_CHANNEL_STATE_INITIAL = 0,
+ RDPDR_CHANNEL_STATE_ANNOUNCE,
+ RDPDR_CHANNEL_STATE_ANNOUNCE_REPLY,
+ RDPDR_CHANNEL_STATE_NAME_REQUEST,
+ RDPDR_CHANNEL_STATE_SERVER_CAPS,
+ RDPDR_CHANNEL_STATE_CLIENT_CAPS,
+ RDPDR_CHANNEL_STATE_CLIENTID_CONFIRM,
+ RDPDR_CHANNEL_STATE_READY,
+ RDPDR_CHANNEL_STATE_USER_LOGGEDON
+};
+
+typedef struct
+{
+ CHANNEL_DEF channelDef;
+ CHANNEL_ENTRY_POINTS_FREERDP_EX channelEntryPoints;
+
+ enum RDPDR_CHANNEL_STATE state;
+ HANDLE thread;
+ wStream* data_in;
+ void* InitHandle;
+ DWORD OpenHandle;
+ wMessageQueue* queue;
+
+ DEVMAN* devman;
+ BOOL ignoreInvalidDevices;
+
+ UINT32 serverOsType;
+ UINT32 serverOsVersion;
+ UINT16 serverVersionMajor;
+ UINT16 serverVersionMinor;
+ UINT32 serverExtendedPDU;
+ UINT32 serverIOCode1;
+ UINT32 serverIOCode2;
+ UINT32 serverExtraFlags1;
+ UINT32 serverExtraFlags2;
+ UINT32 serverSpecialTypeDeviceCap;
+
+ UINT32 clientOsType;
+ UINT32 clientOsVersion;
+ UINT16 clientVersionMajor;
+ UINT16 clientVersionMinor;
+ UINT32 clientExtendedPDU;
+ UINT32 clientIOCode1;
+ UINT32 clientIOCode2;
+ UINT32 clientExtraFlags1;
+ UINT32 clientExtraFlags2;
+ UINT32 clientSpecialTypeDeviceCap;
+
+ UINT32 clientID;
+ char computerName[256];
+
+ UINT32 sequenceId;
+ BOOL userLoggedOn;
+
+ /* hotplug support */
+ HANDLE hotplugThread;
+#ifdef _WIN32
+ HWND hotplug_wnd;
+#endif
+#ifdef __MACOSX__
+ CFRunLoopRef runLoop;
+#endif
+#ifndef _WIN32
+ HANDLE stopEvent;
+#endif
+ rdpContext* rdpcontext;
+ wStreamPool* pool;
+ wLog* log;
+ BOOL async;
+} rdpdrPlugin;
+
+BOOL rdpdr_state_advance(rdpdrPlugin* rdpdr, enum RDPDR_CHANNEL_STATE next);
+UINT rdpdr_send(rdpdrPlugin* rdpdr, wStream* s);
+
+#endif /* FREERDP_CHANNEL_RDPDR_CLIENT_MAIN_H */