summaryrefslogtreecommitdiffstats
path: root/channels/encomsp
diff options
context:
space:
mode:
Diffstat (limited to 'channels/encomsp')
-rw-r--r--channels/encomsp/CMakeLists.txt26
-rw-r--r--channels/encomsp/ChannelOptions.cmake13
-rw-r--r--channels/encomsp/client/CMakeLists.txt27
-rw-r--r--channels/encomsp/client/encomsp_main.c1304
-rw-r--r--channels/encomsp/client/encomsp_main.h42
-rw-r--r--channels/encomsp/server/CMakeLists.txt30
-rw-r--r--channels/encomsp/server/encomsp_main.c372
-rw-r--r--channels/encomsp/server/encomsp_main.h36
8 files changed, 1850 insertions, 0 deletions
diff --git a/channels/encomsp/CMakeLists.txt b/channels/encomsp/CMakeLists.txt
new file mode 100644
index 0000000..be2d374
--- /dev/null
+++ b/channels/encomsp/CMakeLists.txt
@@ -0,0 +1,26 @@
+# FreeRDP: A Remote Desktop Protocol Implementation
+# FreeRDP cmake build script
+#
+# Copyright 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.
+
+define_channel("encomsp")
+
+if(WITH_CLIENT_CHANNELS)
+ add_channel_client(${MODULE_PREFIX} ${CHANNEL_NAME})
+endif()
+
+if(WITH_SERVER_CHANNELS)
+ add_channel_server(${MODULE_PREFIX} ${CHANNEL_NAME})
+endif()
diff --git a/channels/encomsp/ChannelOptions.cmake b/channels/encomsp/ChannelOptions.cmake
new file mode 100644
index 0000000..82ef07e
--- /dev/null
+++ b/channels/encomsp/ChannelOptions.cmake
@@ -0,0 +1,13 @@
+
+set(OPTION_DEFAULT OFF)
+set(OPTION_CLIENT_DEFAULT ON)
+set(OPTION_SERVER_DEFAULT ON)
+
+define_channel_options(NAME "encomsp" TYPE "static"
+ DESCRIPTION "Multiparty Virtual Channel Extension"
+ SPECIFICATIONS "[MS-RDPEMC]"
+ DEFAULT ${OPTION_DEFAULT})
+
+define_channel_client_options(${OPTION_CLIENT_DEFAULT})
+define_channel_server_options(${OPTION_SERVER_DEFAULT})
+
diff --git a/channels/encomsp/client/CMakeLists.txt b/channels/encomsp/client/CMakeLists.txt
new file mode 100644
index 0000000..59eaa8f
--- /dev/null
+++ b/channels/encomsp/client/CMakeLists.txt
@@ -0,0 +1,27 @@
+# FreeRDP: A Remote Desktop Protocol Implementation
+# FreeRDP cmake build script
+#
+# Copyright 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.
+
+define_channel_client("encomsp")
+
+include_directories(..)
+
+set(${MODULE_PREFIX}_SRCS
+ encomsp_main.c
+ encomsp_main.h
+)
+
+add_channel_client_library(${MODULE_PREFIX} ${MODULE_NAME} ${CHANNEL_NAME} FALSE "VirtualChannelEntryEx")
diff --git a/channels/encomsp/client/encomsp_main.c b/channels/encomsp/client/encomsp_main.c
new file mode 100644
index 0000000..18b856e
--- /dev/null
+++ b/channels/encomsp/client/encomsp_main.c
@@ -0,0 +1,1304 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Multiparty Virtual Channel
+ *
+ * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
+ * Copyright 2015 Thincast Technologies GmbH
+ * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <freerdp/config.h>
+
+#include <winpr/crt.h>
+#include <winpr/assert.h>
+#include <winpr/print.h>
+
+#include <freerdp/freerdp.h>
+#include <freerdp/channels/log.h>
+#include <freerdp/client/encomsp.h>
+
+#include "encomsp_main.h"
+
+struct encomsp_plugin
+{
+ CHANNEL_DEF channelDef;
+ CHANNEL_ENTRY_POINTS_FREERDP_EX channelEntryPoints;
+
+ EncomspClientContext* context;
+
+ HANDLE thread;
+ wStream* data_in;
+ void* InitHandle;
+ DWORD OpenHandle;
+ wMessageQueue* queue;
+ rdpContext* rdpcontext;
+};
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_read_header(wStream* s, ENCOMSP_ORDER_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, ENCOMSP_ORDER_HEADER_SIZE))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, header->Type); /* Type (2 bytes) */
+ Stream_Read_UINT16(s, header->Length); /* Length (2 bytes) */
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_write_header(wStream* s, const ENCOMSP_ORDER_HEADER* header)
+{
+ WINPR_ASSERT(header);
+ Stream_Write_UINT16(s, header->Type); /* Type (2 bytes) */
+ Stream_Write_UINT16(s, header->Length); /* Length (2 bytes) */
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_read_unicode_string(wStream* s, ENCOMSP_UNICODE_STRING* str)
+{
+ WINPR_ASSERT(str);
+ const ENCOMSP_UNICODE_STRING empty = { 0 };
+ *str = empty;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 2))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, str->cchString); /* cchString (2 bytes) */
+
+ if (str->cchString > 1024)
+ {
+ WLog_ERR(TAG, "cchString was %" PRIu16 " but has to be < 1025!", str->cchString);
+ return ERROR_INVALID_DATA;
+ }
+
+ if (!Stream_CheckAndLogRequiredLengthOfSize(TAG, s, str->cchString, sizeof(WCHAR)))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read(s, &(str->wString), (str->cchString * 2)); /* String (variable) */
+ return CHANNEL_RC_OK;
+}
+
+static EncomspClientContext* encomsp_get_client_interface(encomspPlugin* encomsp)
+{
+ WINPR_ASSERT(encomsp);
+ return (EncomspClientContext*)encomsp->channelEntryPoints.pInterface;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_virtual_channel_write(encomspPlugin* encomsp, wStream* s)
+{
+ if (!encomsp)
+ {
+ Stream_Free(s, TRUE);
+ return ERROR_INVALID_HANDLE;
+ }
+
+#if 0
+ WLog_INFO(TAG, "EncomspWrite (%"PRIuz")", Stream_Length(s));
+ winpr_HexDump(Stream_Buffer(s), Stream_Length(s));
+#endif
+ const UINT status = encomsp->channelEntryPoints.pVirtualChannelWriteEx(
+ encomsp->InitHandle, encomsp->OpenHandle, Stream_Buffer(s), (UINT32)Stream_Length(s), s);
+
+ if (status != CHANNEL_RC_OK)
+ {
+ Stream_Free(s, TRUE);
+ WLog_ERR(TAG, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
+ WTSErrorToString(status), status);
+ }
+ return status;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_filter_updated_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_FILTER_UPDATED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ WINPR_ASSERT(header);
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 1))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT8(s, pdu.Flags); /* Flags (1 byte) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->FilterUpdated, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->FilterUpdated failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_application_created_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_APPLICATION_CREATED_PDU pdu = { 0 };
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
+ return ERROR_INVALID_DATA;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ Stream_Read_UINT16(s, pdu.Flags); /* Flags (2 bytes) */
+ Stream_Read_UINT32(s, pdu.AppId); /* AppId (4 bytes) */
+
+ UINT error = encomsp_read_unicode_string(s, &(pdu.Name));
+ if (error != CHANNEL_RC_OK)
+ {
+ WLog_ERR(TAG, "encomsp_read_unicode_string failed with error %" PRIu32 "", error);
+ return error;
+ }
+
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ApplicationCreated, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ApplicationCreated failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_application_removed_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_APPLICATION_REMOVED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT32(s, pdu.AppId); /* AppId (4 bytes) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ApplicationRemoved, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ApplicationRemoved failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_window_created_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_WINDOW_CREATED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 10))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, pdu.Flags); /* Flags (2 bytes) */
+ Stream_Read_UINT32(s, pdu.AppId); /* AppId (4 bytes) */
+ Stream_Read_UINT32(s, pdu.WndId); /* WndId (4 bytes) */
+
+ if ((error = encomsp_read_unicode_string(s, &(pdu.Name))))
+ {
+ WLog_ERR(TAG, "encomsp_read_unicode_string failed with error %" PRIu32 "", error);
+ return error;
+ }
+
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->WindowCreated, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->WindowCreated failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_window_removed_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_WINDOW_REMOVED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT32(s, pdu.WndId); /* WndId (4 bytes) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->WindowRemoved, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->WindowRemoved failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_show_window_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_SHOW_WINDOW_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 4))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT32(s, pdu.WndId); /* WndId (4 bytes) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ShowWindow, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ShowWindow failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_participant_created_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_PARTICIPANT_CREATED_PDU pdu = { 0 };
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 10))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT32(s, pdu.ParticipantId); /* ParticipantId (4 bytes) */
+ Stream_Read_UINT32(s, pdu.GroupId); /* GroupId (4 bytes) */
+ Stream_Read_UINT16(s, pdu.Flags); /* Flags (2 bytes) */
+
+ UINT error = encomsp_read_unicode_string(s, &(pdu.FriendlyName));
+ if (error != CHANNEL_RC_OK)
+ {
+ WLog_ERR(TAG, "encomsp_read_unicode_string failed with error %" PRIu32 "", error);
+ return error;
+ }
+
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ParticipantCreated, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ParticipantCreated failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_participant_removed_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_PARTICIPANT_REMOVED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 12))
+ return ERROR_INVALID_DATA;
+
+ const size_t beg = (Stream_GetPosition(s)) - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ Stream_Read_UINT32(s, pdu.ParticipantId); /* ParticipantId (4 bytes) */
+ Stream_Read_UINT32(s, pdu.DiscType); /* DiscType (4 bytes) */
+ Stream_Read_UINT32(s, pdu.DiscCode); /* DiscCode (4 bytes) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ParticipantRemoved, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ParticipantRemoved failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_change_participant_control_level_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_CHANGE_PARTICIPANT_CONTROL_LEVEL_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, pdu.Flags); /* Flags (2 bytes) */
+ Stream_Read_UINT32(s, pdu.ParticipantId); /* ParticipantId (4 bytes) */
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->ChangeParticipantControlLevel, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ChangeParticipantControlLevel failed with error %" PRIu32 "",
+ error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_send_change_participant_control_level_pdu(
+ EncomspClientContext* context, const ENCOMSP_CHANGE_PARTICIPANT_CONTROL_LEVEL_PDU* pdu)
+{
+ ENCOMSP_ORDER_HEADER header = { 0 };
+
+ WINPR_ASSERT(context);
+ encomspPlugin* encomsp = (encomspPlugin*)context->handle;
+
+ header.Type = ODTYPE_PARTICIPANT_CTRL_CHANGED;
+ header.Length = ENCOMSP_ORDER_HEADER_SIZE + 6;
+
+ wStream* s = Stream_New(NULL, header.Length);
+
+ if (!s)
+ {
+ WLog_ERR(TAG, "Stream_New failed!");
+ return CHANNEL_RC_NO_MEMORY;
+ }
+
+ const UINT error = encomsp_write_header(s, &header);
+ if (error != CHANNEL_RC_OK)
+ {
+ WLog_ERR(TAG, "encomsp_write_header failed with error %" PRIu32 "!", error);
+ return error;
+ }
+
+ Stream_Write_UINT16(s, pdu->Flags); /* Flags (2 bytes) */
+ Stream_Write_UINT32(s, pdu->ParticipantId); /* ParticipantId (4 bytes) */
+ Stream_SealLength(s);
+ return encomsp_virtual_channel_write(encomsp, s);
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_graphics_stream_paused_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_GRAPHICS_STREAM_PAUSED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->GraphicsStreamPaused, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->GraphicsStreamPaused failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_graphics_stream_resumed_pdu(encomspPlugin* encomsp, wStream* s,
+ const ENCOMSP_ORDER_HEADER* header)
+{
+ ENCOMSP_GRAPHICS_STREAM_RESUMED_PDU pdu = { 0 };
+ UINT error = CHANNEL_RC_OK;
+ EncomspClientContext* context = encomsp_get_client_interface(encomsp);
+
+ if (!context)
+ return ERROR_INVALID_HANDLE;
+
+ const size_t pos = Stream_GetPosition(s);
+ if (pos < ENCOMSP_ORDER_HEADER_SIZE)
+ return ERROR_INVALID_DATA;
+ const size_t beg = pos - ENCOMSP_ORDER_HEADER_SIZE;
+
+ WINPR_ASSERT(header);
+ pdu.Length = header->Length;
+ pdu.Type = header->Type;
+
+ const size_t end = Stream_GetPosition(s);
+ const size_t body = beg + header->Length;
+
+ if (body < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if (body > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)(body - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, body);
+ }
+
+ IFCALLRET(context->GraphicsStreamResumed, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->GraphicsStreamResumed failed with error %" PRIu32 "", error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_process_receive(encomspPlugin* encomsp, wStream* s)
+{
+ UINT error = CHANNEL_RC_OK;
+ ENCOMSP_ORDER_HEADER header = { 0 };
+
+ WINPR_ASSERT(encomsp);
+ while (Stream_GetRemainingLength(s) > 0)
+ {
+ if ((error = encomsp_read_header(s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_read_header failed with error %" PRIu32 "!", error);
+ return error;
+ }
+
+ // WLog_DBG(TAG, "EncomspReceive: Type: %"PRIu16" Length: %"PRIu16"", header.Type,
+ // header.Length);
+
+ switch (header.Type)
+ {
+ case ODTYPE_FILTER_STATE_UPDATED:
+ if ((error = encomsp_recv_filter_updated_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_recv_filter_updated_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_APP_REMOVED:
+ if ((error = encomsp_recv_application_removed_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_application_removed_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_APP_CREATED:
+ if ((error = encomsp_recv_application_created_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_application_removed_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_WND_REMOVED:
+ if ((error = encomsp_recv_window_removed_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_recv_window_removed_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_WND_CREATED:
+ if ((error = encomsp_recv_window_created_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_recv_window_created_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_WND_SHOW:
+ if ((error = encomsp_recv_show_window_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_recv_show_window_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_PARTICIPANT_REMOVED:
+ if ((error = encomsp_recv_participant_removed_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_participant_removed_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_PARTICIPANT_CREATED:
+ if ((error = encomsp_recv_participant_created_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_participant_created_pdu failed with error %" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_PARTICIPANT_CTRL_CHANGED:
+ if ((error =
+ encomsp_recv_change_participant_control_level_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_change_participant_control_level_pdu failed with error "
+ "%" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_GRAPHICS_STREAM_PAUSED:
+ if ((error = encomsp_recv_graphics_stream_paused_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_graphics_stream_paused_pdu failed with error %" PRIu32
+ "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ case ODTYPE_GRAPHICS_STREAM_RESUMED:
+ if ((error = encomsp_recv_graphics_stream_resumed_pdu(encomsp, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_graphics_stream_resumed_pdu failed with error %" PRIu32
+ "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ default:
+ WLog_ERR(TAG, "header.Type %" PRIu16 " not found", header.Type);
+ return ERROR_INVALID_DATA;
+ }
+ }
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_virtual_channel_event_data_received(encomspPlugin* encomsp, const void* pData,
+ UINT32 dataLength, UINT32 totalLength,
+ UINT32 dataFlags)
+{
+ WINPR_ASSERT(encomsp);
+
+ if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
+ return CHANNEL_RC_OK;
+
+ if (dataFlags & CHANNEL_FLAG_FIRST)
+ {
+ if (encomsp->data_in)
+ Stream_Free(encomsp->data_in, TRUE);
+
+ encomsp->data_in = Stream_New(NULL, totalLength);
+
+ if (!encomsp->data_in)
+ {
+ WLog_ERR(TAG, "Stream_New failed!");
+ return CHANNEL_RC_NO_MEMORY;
+ }
+ }
+
+ wStream* data_in = encomsp->data_in;
+
+ if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
+ {
+ WLog_ERR(TAG, "Stream_EnsureRemainingCapacity failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ Stream_Write(data_in, pData, dataLength);
+
+ if (dataFlags & CHANNEL_FLAG_LAST)
+ {
+ if (Stream_Capacity(data_in) != Stream_GetPosition(data_in))
+ {
+ WLog_ERR(TAG, "encomsp_plugin_process_received: read error");
+ return ERROR_INVALID_DATA;
+ }
+
+ encomsp->data_in = NULL;
+ Stream_SealLength(data_in);
+ Stream_SetPosition(data_in, 0);
+
+ if (!MessageQueue_Post(encomsp->queue, NULL, 0, (void*)data_in, NULL))
+ {
+ WLog_ERR(TAG, "MessageQueue_Post failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+static VOID VCAPITYPE encomsp_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle,
+ UINT event, LPVOID pData,
+ UINT32 dataLength, UINT32 totalLength,
+ UINT32 dataFlags)
+{
+ UINT error = CHANNEL_RC_OK;
+ encomspPlugin* encomsp = (encomspPlugin*)lpUserParam;
+
+ switch (event)
+ {
+ case CHANNEL_EVENT_DATA_RECEIVED:
+ if (!encomsp || (encomsp->OpenHandle != openHandle))
+ {
+ WLog_ERR(TAG, "error no match");
+ return;
+ }
+ if ((error = encomsp_virtual_channel_event_data_received(encomsp, pData, dataLength,
+ totalLength, dataFlags)))
+ WLog_ERR(TAG,
+ "encomsp_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_Free(s, TRUE);
+ }
+ break;
+
+ case CHANNEL_EVENT_USER:
+ break;
+ }
+
+ if (error && encomsp && encomsp->rdpcontext)
+ setChannelError(encomsp->rdpcontext, error,
+ "encomsp_virtual_channel_open_event reported an error");
+
+ return;
+}
+
+static DWORD WINAPI encomsp_virtual_channel_client_thread(LPVOID arg)
+{
+ wStream* data = NULL;
+ wMessage message = { 0 };
+ encomspPlugin* encomsp = (encomspPlugin*)arg;
+ UINT error = CHANNEL_RC_OK;
+
+ WINPR_ASSERT(encomsp);
+ while (1)
+ {
+ if (!MessageQueue_Wait(encomsp->queue))
+ {
+ WLog_ERR(TAG, "MessageQueue_Wait failed!");
+ error = ERROR_INTERNAL_ERROR;
+ break;
+ }
+
+ if (!MessageQueue_Peek(encomsp->queue, &message, TRUE))
+ {
+ WLog_ERR(TAG, "MessageQueue_Peek failed!");
+ error = ERROR_INTERNAL_ERROR;
+ break;
+ }
+
+ if (message.id == WMQ_QUIT)
+ break;
+
+ if (message.id == 0)
+ {
+ data = (wStream*)message.wParam;
+
+ if ((error = encomsp_process_receive(encomsp, data)))
+ {
+ WLog_ERR(TAG, "encomsp_process_receive failed with error %" PRIu32 "!", error);
+ Stream_Free(data, TRUE);
+ break;
+ }
+
+ Stream_Free(data, TRUE);
+ }
+ }
+
+ if (error && encomsp->rdpcontext)
+ setChannelError(encomsp->rdpcontext, error,
+ "encomsp_virtual_channel_client_thread reported an error");
+
+ ExitThread(error);
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_virtual_channel_event_connected(encomspPlugin* encomsp, LPVOID pData,
+ UINT32 dataLength)
+{
+ WINPR_ASSERT(encomsp);
+
+ encomsp->queue = MessageQueue_New(NULL);
+
+ if (!encomsp->queue)
+ {
+ WLog_ERR(TAG, "MessageQueue_New failed!");
+ return CHANNEL_RC_NO_MEMORY;
+ }
+
+ if (!(encomsp->thread = CreateThread(NULL, 0, encomsp_virtual_channel_client_thread,
+ (void*)encomsp, 0, NULL)))
+ {
+ WLog_ERR(TAG, "CreateThread failed!");
+ MessageQueue_Free(encomsp->queue);
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ return encomsp->channelEntryPoints.pVirtualChannelOpenEx(
+ encomsp->InitHandle, &encomsp->OpenHandle, encomsp->channelDef.name,
+ encomsp_virtual_channel_open_event_ex);
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_virtual_channel_event_disconnected(encomspPlugin* encomsp)
+{
+ WINPR_ASSERT(encomsp);
+ if (encomsp->OpenHandle == 0)
+ return CHANNEL_RC_OK;
+
+ if (encomsp->queue && encomsp->thread)
+ {
+ if (MessageQueue_PostQuit(encomsp->queue, 0) &&
+ (WaitForSingleObject(encomsp->thread, INFINITE) == WAIT_FAILED))
+ {
+ const UINT rc = GetLastError();
+ WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", rc);
+ return rc;
+ }
+ }
+
+ MessageQueue_Free(encomsp->queue);
+ CloseHandle(encomsp->thread);
+ encomsp->queue = NULL;
+ encomsp->thread = NULL;
+
+ WINPR_ASSERT(encomsp->channelEntryPoints.pVirtualChannelCloseEx);
+ const UINT rc = encomsp->channelEntryPoints.pVirtualChannelCloseEx(encomsp->InitHandle,
+ encomsp->OpenHandle);
+
+ if (CHANNEL_RC_OK != rc)
+ {
+ WLog_ERR(TAG, "pVirtualChannelClose failed with %s [%08" PRIX32 "]", WTSErrorToString(rc),
+ rc);
+ return rc;
+ }
+
+ encomsp->OpenHandle = 0;
+
+ if (encomsp->data_in)
+ {
+ Stream_Free(encomsp->data_in, TRUE);
+ encomsp->data_in = NULL;
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_virtual_channel_event_terminated(encomspPlugin* encomsp)
+{
+ WINPR_ASSERT(encomsp);
+
+ encomsp->InitHandle = 0;
+ free(encomsp->context);
+ free(encomsp);
+ return CHANNEL_RC_OK;
+}
+
+static VOID VCAPITYPE encomsp_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle,
+ UINT event, LPVOID pData,
+ UINT dataLength)
+{
+ UINT error = CHANNEL_RC_OK;
+ encomspPlugin* encomsp = (encomspPlugin*)lpUserParam;
+
+ if (!encomsp || (encomsp->InitHandle != pInitHandle))
+ {
+ WLog_ERR(TAG, "error no match");
+ return;
+ }
+
+ switch (event)
+ {
+ case CHANNEL_EVENT_INITIALIZED:
+ break;
+
+ case CHANNEL_EVENT_CONNECTED:
+ if ((error = encomsp_virtual_channel_event_connected(encomsp, pData, dataLength)))
+ WLog_ERR(TAG,
+ "encomsp_virtual_channel_event_connected failed with error %" PRIu32 "",
+ error);
+
+ break;
+
+ case CHANNEL_EVENT_DISCONNECTED:
+ if ((error = encomsp_virtual_channel_event_disconnected(encomsp)))
+ WLog_ERR(TAG,
+ "encomsp_virtual_channel_event_disconnected failed with error %" PRIu32 "",
+ error);
+
+ break;
+
+ case CHANNEL_EVENT_TERMINATED:
+ encomsp_virtual_channel_event_terminated(encomsp);
+ break;
+
+ default:
+ break;
+ }
+
+ if (error && encomsp->rdpcontext)
+ setChannelError(encomsp->rdpcontext, error,
+ "encomsp_virtual_channel_init_event reported an error");
+}
+
+/* encomsp is always built-in */
+#define VirtualChannelEntryEx encomsp_VirtualChannelEntryEx
+
+FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS_EX pEntryPoints,
+ PVOID pInitHandle))
+{
+ BOOL isFreerdp = FALSE;
+ encomspPlugin* encomsp = (encomspPlugin*)calloc(1, sizeof(encomspPlugin));
+
+ if (!encomsp)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ return FALSE;
+ }
+
+ encomsp->channelDef.options = CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP |
+ CHANNEL_OPTION_COMPRESS_RDP | CHANNEL_OPTION_SHOW_PROTOCOL;
+ sprintf_s(encomsp->channelDef.name, ARRAYSIZE(encomsp->channelDef.name),
+ ENCOMSP_SVC_CHANNEL_NAME);
+ CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx =
+ (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints;
+ WINPR_ASSERT(pEntryPointsEx);
+
+ EncomspClientContext* context = NULL;
+ if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) &&
+ (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER))
+ {
+ context = (EncomspClientContext*)calloc(1, sizeof(EncomspClientContext));
+
+ if (!context)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ goto error_out;
+ }
+
+ context->handle = (void*)encomsp;
+ context->FilterUpdated = NULL;
+ context->ApplicationCreated = NULL;
+ context->ApplicationRemoved = NULL;
+ context->WindowCreated = NULL;
+ context->WindowRemoved = NULL;
+ context->ShowWindow = NULL;
+ context->ParticipantCreated = NULL;
+ context->ParticipantRemoved = NULL;
+ context->ChangeParticipantControlLevel = encomsp_send_change_participant_control_level_pdu;
+ context->GraphicsStreamPaused = NULL;
+ context->GraphicsStreamResumed = NULL;
+ encomsp->context = context;
+ encomsp->rdpcontext = pEntryPointsEx->context;
+ isFreerdp = TRUE;
+ }
+
+ CopyMemory(&(encomsp->channelEntryPoints), pEntryPoints,
+ sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX));
+ encomsp->InitHandle = pInitHandle;
+ const UINT rc = encomsp->channelEntryPoints.pVirtualChannelInitEx(
+ encomsp, context, pInitHandle, &encomsp->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
+ encomsp_virtual_channel_init_event_ex);
+
+ if (CHANNEL_RC_OK != rc)
+ {
+ WLog_ERR(TAG, "failed with %s [%08" PRIX32 "]", WTSErrorToString(rc), rc);
+ goto error_out;
+ }
+
+ encomsp->channelEntryPoints.pInterface = context;
+ return TRUE;
+error_out:
+
+ if (isFreerdp)
+ free(encomsp->context);
+
+ free(encomsp);
+ return FALSE;
+}
diff --git a/channels/encomsp/client/encomsp_main.h b/channels/encomsp/client/encomsp_main.h
new file mode 100644
index 0000000..ad43cea
--- /dev/null
+++ b/channels/encomsp/client/encomsp_main.h
@@ -0,0 +1,42 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Multiparty Virtual Channel
+ *
+ * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
+ * Copyright 2015 Thincast Technologies GmbH
+ * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef FREERDP_CHANNEL_ENCOMSP_CLIENT_MAIN_H
+#define FREERDP_CHANNEL_ENCOMSP_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/channels/log.h>
+#include <freerdp/svc.h>
+#include <freerdp/addin.h>
+
+#include <freerdp/client/encomsp.h>
+
+#define TAG CHANNELS_TAG("encomsp.client")
+
+typedef struct encomsp_plugin encomspPlugin;
+
+#endif /* FREERDP_CHANNEL_ENCOMSP_CLIENT_MAIN_H */
diff --git a/channels/encomsp/server/CMakeLists.txt b/channels/encomsp/server/CMakeLists.txt
new file mode 100644
index 0000000..096e6d0
--- /dev/null
+++ b/channels/encomsp/server/CMakeLists.txt
@@ -0,0 +1,30 @@
+# FreeRDP: A Remote Desktop Protocol Implementation
+# FreeRDP cmake build script
+#
+# Copyright 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.
+
+define_channel_server("encomsp")
+
+include_directories(..)
+
+set(${MODULE_PREFIX}_SRCS
+ encomsp_main.c
+ encomsp_main.h
+)
+
+set(${MODULE_PREFIX}_LIBS
+ winpr
+)
+add_channel_server_library(${MODULE_PREFIX} ${MODULE_NAME} ${CHANNEL_NAME} FALSE "VirtualChannelEntry")
diff --git a/channels/encomsp/server/encomsp_main.c b/channels/encomsp/server/encomsp_main.c
new file mode 100644
index 0000000..f6390df
--- /dev/null
+++ b/channels/encomsp/server/encomsp_main.c
@@ -0,0 +1,372 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Multiparty Virtual Channel
+ *
+ * Copyright 2014 Marc-Andre Moreau <marcandre.moreau@gmail.com>
+ * Copyright 2015 Thincast Technologies GmbH
+ * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <freerdp/config.h>
+
+#include <winpr/crt.h>
+#include <winpr/print.h>
+#include <winpr/stream.h>
+
+#include <freerdp/freerdp.h>
+#include <freerdp/channels/log.h>
+
+#include "encomsp_main.h"
+
+#define TAG CHANNELS_TAG("encomsp.server")
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_read_header(wStream* s, ENCOMSP_ORDER_HEADER* header)
+{
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, ENCOMSP_ORDER_HEADER_SIZE))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, header->Type); /* Type (2 bytes) */
+ Stream_Read_UINT16(s, header->Length); /* Length (2 bytes) */
+ return CHANNEL_RC_OK;
+}
+
+#if 0
+
+static int encomsp_write_header(wStream* s, ENCOMSP_ORDER_HEADER* header)
+{
+ Stream_Write_UINT16(s, header->Type); /* Type (2 bytes) */
+ Stream_Write_UINT16(s, header->Length); /* Length (2 bytes) */
+ return 1;
+}
+
+static int encomsp_read_unicode_string(wStream* s, ENCOMSP_UNICODE_STRING* str)
+{
+ ZeroMemory(str, sizeof(ENCOMSP_UNICODE_STRING));
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 2))
+ return -1;
+
+ Stream_Read_UINT16(s, str->cchString); /* cchString (2 bytes) */
+
+ if (str->cchString > 1024)
+ return -1;
+
+ if (!Stream_CheckAndLogRequiredLengthOfSize(TAG, s, str->cchString, sizeof(WCHAR)))
+ return -1;
+
+ Stream_Read(s, &(str->wString), (str->cchString * 2)); /* String (variable) */
+ return 1;
+}
+
+#endif
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_recv_change_participant_control_level_pdu(EncomspServerContext* context,
+ wStream* s,
+ ENCOMSP_ORDER_HEADER* header)
+{
+ int beg = 0;
+ int end = 0;
+ ENCOMSP_CHANGE_PARTICIPANT_CONTROL_LEVEL_PDU pdu;
+ UINT error = CHANNEL_RC_OK;
+ beg = ((int)Stream_GetPosition(s)) - ENCOMSP_ORDER_HEADER_SIZE;
+ CopyMemory(&pdu, header, sizeof(ENCOMSP_ORDER_HEADER));
+
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, 6))
+ return ERROR_INVALID_DATA;
+
+ Stream_Read_UINT16(s, pdu.Flags); /* Flags (2 bytes) */
+ Stream_Read_UINT32(s, pdu.ParticipantId); /* ParticipantId (4 bytes) */
+ end = (int)Stream_GetPosition(s);
+
+ if ((beg + header->Length) < end)
+ {
+ WLog_ERR(TAG, "Not enough data!");
+ return ERROR_INVALID_DATA;
+ }
+
+ if ((beg + header->Length) > end)
+ {
+ if (!Stream_CheckAndLogRequiredLength(TAG, s, (size_t)((beg + header->Length) - end)))
+ return ERROR_INVALID_DATA;
+
+ Stream_SetPosition(s, (beg + header->Length));
+ }
+
+ IFCALLRET(context->ChangeParticipantControlLevel, error, context, &pdu);
+
+ if (error)
+ WLog_ERR(TAG, "context->ChangeParticipantControlLevel failed with error %" PRIu32 "",
+ error);
+
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_server_receive_pdu(EncomspServerContext* context, wStream* s)
+{
+ UINT error = CHANNEL_RC_OK;
+ ENCOMSP_ORDER_HEADER header;
+
+ while (Stream_GetRemainingLength(s) > 0)
+ {
+ if ((error = encomsp_read_header(s, &header)))
+ {
+ WLog_ERR(TAG, "encomsp_read_header failed with error %" PRIu32 "!", error);
+ return error;
+ }
+
+ WLog_INFO(TAG, "EncomspReceive: Type: %" PRIu16 " Length: %" PRIu16 "", header.Type,
+ header.Length);
+
+ switch (header.Type)
+ {
+ case ODTYPE_PARTICIPANT_CTRL_CHANGED:
+ if ((error =
+ encomsp_recv_change_participant_control_level_pdu(context, s, &header)))
+ {
+ WLog_ERR(TAG,
+ "encomsp_recv_change_participant_control_level_pdu failed with error "
+ "%" PRIu32 "!",
+ error);
+ return error;
+ }
+
+ break;
+
+ default:
+ WLog_ERR(TAG, "header.Type unknown %" PRIu16 "!", header.Type);
+ return ERROR_INVALID_DATA;
+ }
+ }
+
+ return error;
+}
+
+static DWORD WINAPI encomsp_server_thread(LPVOID arg)
+{
+ wStream* s = NULL;
+ DWORD nCount = 0;
+ void* buffer = NULL;
+ HANDLE events[8];
+ HANDLE ChannelEvent = NULL;
+ DWORD BytesReturned = 0;
+ ENCOMSP_ORDER_HEADER* header = NULL;
+ EncomspServerContext* context = NULL;
+ UINT error = CHANNEL_RC_OK;
+ DWORD status = 0;
+ context = (EncomspServerContext*)arg;
+
+ buffer = NULL;
+ BytesReturned = 0;
+ ChannelEvent = NULL;
+ s = Stream_New(NULL, 4096);
+
+ if (!s)
+ {
+ WLog_ERR(TAG, "Stream_New failed!");
+ error = CHANNEL_RC_NO_MEMORY;
+ goto out;
+ }
+
+ if (WTSVirtualChannelQuery(context->priv->ChannelHandle, WTSVirtualEventHandle, &buffer,
+ &BytesReturned) == TRUE)
+ {
+ if (BytesReturned == sizeof(HANDLE))
+ CopyMemory(&ChannelEvent, buffer, sizeof(HANDLE));
+
+ WTSFreeMemory(buffer);
+ }
+
+ nCount = 0;
+ events[nCount++] = ChannelEvent;
+ events[nCount++] = context->priv->StopEvent;
+
+ while (1)
+ {
+ status = WaitForMultipleObjects(nCount, events, FALSE, INFINITE);
+
+ if (status == WAIT_FAILED)
+ {
+ error = GetLastError();
+ WLog_ERR(TAG, "WaitForMultipleObjects failed with error %" PRIu32 "", error);
+ break;
+ }
+
+ status = WaitForSingleObject(context->priv->StopEvent, 0);
+
+ if (status == WAIT_FAILED)
+ {
+ error = GetLastError();
+ WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", error);
+ break;
+ }
+
+ if (status == WAIT_OBJECT_0)
+ {
+ break;
+ }
+
+ WTSVirtualChannelRead(context->priv->ChannelHandle, 0, NULL, 0, &BytesReturned);
+
+ if (BytesReturned < 1)
+ continue;
+
+ if (!Stream_EnsureRemainingCapacity(s, BytesReturned))
+ {
+ WLog_ERR(TAG, "Stream_EnsureRemainingCapacity failed!");
+ error = CHANNEL_RC_NO_MEMORY;
+ break;
+ }
+
+ if (!WTSVirtualChannelRead(context->priv->ChannelHandle, 0, (PCHAR)Stream_Buffer(s),
+ Stream_Capacity(s), &BytesReturned))
+ {
+ WLog_ERR(TAG, "WTSVirtualChannelRead failed!");
+ error = ERROR_INTERNAL_ERROR;
+ break;
+ }
+
+ if (Stream_GetPosition(s) >= ENCOMSP_ORDER_HEADER_SIZE)
+ {
+ header = (ENCOMSP_ORDER_HEADER*)Stream_Buffer(s);
+
+ if (header->Length >= Stream_GetPosition(s))
+ {
+ Stream_SealLength(s);
+ Stream_SetPosition(s, 0);
+
+ if ((error = encomsp_server_receive_pdu(context, s)))
+ {
+ WLog_ERR(TAG, "encomsp_server_receive_pdu failed with error %" PRIu32 "!",
+ error);
+ break;
+ }
+
+ Stream_SetPosition(s, 0);
+ }
+ }
+ }
+
+ Stream_Free(s, TRUE);
+out:
+
+ if (error && context->rdpcontext)
+ setChannelError(context->rdpcontext, error, "encomsp_server_thread reported an error");
+
+ ExitThread(error);
+ return error;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_server_start(EncomspServerContext* context)
+{
+ context->priv->ChannelHandle =
+ WTSVirtualChannelOpen(context->vcm, WTS_CURRENT_SESSION, ENCOMSP_SVC_CHANNEL_NAME);
+
+ if (!context->priv->ChannelHandle)
+ return CHANNEL_RC_BAD_CHANNEL;
+
+ if (!(context->priv->StopEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))
+ {
+ WLog_ERR(TAG, "CreateEvent failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ if (!(context->priv->Thread =
+ CreateThread(NULL, 0, encomsp_server_thread, (void*)context, 0, NULL)))
+ {
+ WLog_ERR(TAG, "CreateThread failed!");
+ CloseHandle(context->priv->StopEvent);
+ context->priv->StopEvent = NULL;
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ return CHANNEL_RC_OK;
+}
+
+/**
+ * Function description
+ *
+ * @return 0 on success, otherwise a Win32 error code
+ */
+static UINT encomsp_server_stop(EncomspServerContext* context)
+{
+ UINT error = CHANNEL_RC_OK;
+ SetEvent(context->priv->StopEvent);
+
+ if (WaitForSingleObject(context->priv->Thread, INFINITE) == WAIT_FAILED)
+ {
+ error = GetLastError();
+ WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", error);
+ return error;
+ }
+
+ CloseHandle(context->priv->Thread);
+ CloseHandle(context->priv->StopEvent);
+ return error;
+}
+
+EncomspServerContext* encomsp_server_context_new(HANDLE vcm)
+{
+ EncomspServerContext* context = NULL;
+ context = (EncomspServerContext*)calloc(1, sizeof(EncomspServerContext));
+
+ if (context)
+ {
+ context->vcm = vcm;
+ context->Start = encomsp_server_start;
+ context->Stop = encomsp_server_stop;
+ context->priv = (EncomspServerPrivate*)calloc(1, sizeof(EncomspServerPrivate));
+
+ if (!context->priv)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ free(context);
+ return NULL;
+ }
+ }
+
+ return context;
+}
+
+void encomsp_server_context_free(EncomspServerContext* context)
+{
+ if (context)
+ {
+ if (context->priv->ChannelHandle != INVALID_HANDLE_VALUE)
+ WTSVirtualChannelClose(context->priv->ChannelHandle);
+
+ free(context->priv);
+ free(context);
+ }
+}
diff --git a/channels/encomsp/server/encomsp_main.h b/channels/encomsp/server/encomsp_main.h
new file mode 100644
index 0000000..37d8c71
--- /dev/null
+++ b/channels/encomsp/server/encomsp_main.h
@@ -0,0 +1,36 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Multiparty Virtual Channel
+ *
+ * Copyright 2014 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_ENCOMSP_SERVER_MAIN_H
+#define FREERDP_CHANNEL_ENCOMSP_SERVER_MAIN_H
+
+#include <winpr/crt.h>
+#include <winpr/synch.h>
+#include <winpr/thread.h>
+
+#include <freerdp/server/encomsp.h>
+
+struct s_encomsp_server_private
+{
+ HANDLE Thread;
+ HANDLE StopEvent;
+ void* ChannelHandle;
+};
+
+#endif /* FREERDP_CHANNEL_ENCOMSP_SERVER_MAIN_H */
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-24 09:52:08 +0000