summaryrefslogtreecommitdiffstats
path: root/channels/client/addin.c
diff options
context:
space:
mode:
Diffstat (limited to 'channels/client/addin.c')
-rw-r--r--channels/client/addin.c761
1 files changed, 761 insertions, 0 deletions
diff --git a/channels/client/addin.c b/channels/client/addin.c
new file mode 100644
index 0000000..6d87f6c
--- /dev/null
+++ b/channels/client/addin.c
@@ -0,0 +1,761 @@
+/**
+ * FreeRDP: A Remote Desktop Protocol Implementation
+ * Channel Addins
+ *
+ * Copyright 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 <winpr/crt.h>
+#include <winpr/assert.h>
+#include <winpr/path.h>
+#include <winpr/string.h>
+#include <winpr/file.h>
+#include <winpr/synch.h>
+#include <winpr/library.h>
+#include <winpr/collections.h>
+
+#include <freerdp/freerdp.h>
+#include <freerdp/addin.h>
+#include <freerdp/build-config.h>
+#include <freerdp/client/channels.h>
+
+#include "tables.h"
+
+#include "addin.h"
+
+#include <freerdp/channels/log.h>
+#define TAG CHANNELS_TAG("addin")
+
+extern const STATIC_ENTRY_TABLE CLIENT_STATIC_ENTRY_TABLES[];
+
+static void* freerdp_channels_find_static_entry_in_table(const STATIC_ENTRY_TABLE* table,
+ const char* identifier)
+{
+ size_t index = 0;
+ const STATIC_ENTRY* pEntry = (const STATIC_ENTRY*)&table->table[index++];
+
+ while (pEntry->entry != NULL)
+ {
+ if (strcmp(pEntry->name, identifier) == 0)
+ {
+ return (void*)pEntry->entry;
+ }
+
+ pEntry = (const STATIC_ENTRY*)&table->table[index++];
+ }
+
+ return NULL;
+}
+
+void* freerdp_channels_client_find_static_entry(const char* name, const char* identifier)
+{
+ size_t index = 0;
+ const STATIC_ENTRY_TABLE* pEntry = &CLIENT_STATIC_ENTRY_TABLES[index++];
+
+ while (pEntry->table != NULL)
+ {
+ if (strcmp(pEntry->name, name) == 0)
+ {
+ return freerdp_channels_find_static_entry_in_table(pEntry, identifier);
+ }
+
+ pEntry = &CLIENT_STATIC_ENTRY_TABLES[index++];
+ }
+
+ return NULL;
+}
+
+extern const STATIC_ADDIN_TABLE CLIENT_STATIC_ADDIN_TABLE[];
+
+static FREERDP_ADDIN** freerdp_channels_list_client_static_addins(LPCSTR pszName,
+ LPCSTR pszSubsystem,
+ LPCSTR pszType, DWORD dwFlags)
+{
+ DWORD nAddins = 0;
+ FREERDP_ADDIN** ppAddins = NULL;
+ const STATIC_SUBSYSTEM_ENTRY* subsystems = NULL;
+ nAddins = 0;
+ ppAddins = (FREERDP_ADDIN**)calloc(128, sizeof(FREERDP_ADDIN*));
+
+ if (!ppAddins)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ return NULL;
+ }
+
+ ppAddins[nAddins] = NULL;
+
+ for (size_t i = 0; CLIENT_STATIC_ADDIN_TABLE[i].name != NULL; i++)
+ {
+ FREERDP_ADDIN* pAddin = (FREERDP_ADDIN*)calloc(1, sizeof(FREERDP_ADDIN));
+
+ if (!pAddin)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ goto error_out;
+ }
+
+ sprintf_s(pAddin->cName, ARRAYSIZE(pAddin->cName), "%s", CLIENT_STATIC_ADDIN_TABLE[i].name);
+ pAddin->dwFlags = FREERDP_ADDIN_CLIENT;
+ pAddin->dwFlags |= FREERDP_ADDIN_STATIC;
+ pAddin->dwFlags |= FREERDP_ADDIN_NAME;
+ ppAddins[nAddins++] = pAddin;
+ subsystems = (const STATIC_SUBSYSTEM_ENTRY*)CLIENT_STATIC_ADDIN_TABLE[i].table;
+
+ for (size_t j = 0; subsystems[j].name != NULL; j++)
+ {
+ pAddin = (FREERDP_ADDIN*)calloc(1, sizeof(FREERDP_ADDIN));
+
+ if (!pAddin)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ goto error_out;
+ }
+
+ sprintf_s(pAddin->cName, ARRAYSIZE(pAddin->cName), "%s",
+ CLIENT_STATIC_ADDIN_TABLE[i].name);
+ sprintf_s(pAddin->cSubsystem, ARRAYSIZE(pAddin->cSubsystem), "%s", subsystems[j].name);
+ pAddin->dwFlags = FREERDP_ADDIN_CLIENT;
+ pAddin->dwFlags |= FREERDP_ADDIN_STATIC;
+ pAddin->dwFlags |= FREERDP_ADDIN_NAME;
+ pAddin->dwFlags |= FREERDP_ADDIN_SUBSYSTEM;
+ ppAddins[nAddins++] = pAddin;
+ }
+ }
+
+ return ppAddins;
+error_out:
+ freerdp_channels_addin_list_free(ppAddins);
+ return NULL;
+}
+
+static HANDLE FindFirstFileUTF8(LPCSTR pszSearchPath, WIN32_FIND_DATAW* FindData)
+{
+ HANDLE hdl = INVALID_HANDLE_VALUE;
+ if (!pszSearchPath)
+ return hdl;
+ WCHAR* wpath = ConvertUtf8ToWCharAlloc(pszSearchPath, NULL);
+ if (!wpath)
+ return hdl;
+
+ hdl = FindFirstFileW(wpath, FindData);
+ free(wpath);
+
+ return hdl;
+}
+
+static FREERDP_ADDIN** freerdp_channels_list_dynamic_addins(LPCSTR pszName, LPCSTR pszSubsystem,
+ LPCSTR pszType, DWORD dwFlags)
+{
+ int nDashes = 0;
+ HANDLE hFind = NULL;
+ DWORD nAddins = 0;
+ LPSTR pszPattern = NULL;
+ size_t cchPattern = 0;
+ LPCSTR pszAddinPath = FREERDP_ADDIN_PATH;
+ LPCSTR pszInstallPrefix = FREERDP_INSTALL_PREFIX;
+ LPCSTR pszExtension = NULL;
+ LPSTR pszSearchPath = NULL;
+ size_t cchSearchPath = 0;
+ size_t cchAddinPath = 0;
+ size_t cchInstallPrefix = 0;
+ FREERDP_ADDIN** ppAddins = NULL;
+ WIN32_FIND_DATAW FindData = { 0 };
+ cchAddinPath = strnlen(pszAddinPath, sizeof(FREERDP_ADDIN_PATH));
+ cchInstallPrefix = strnlen(pszInstallPrefix, sizeof(FREERDP_INSTALL_PREFIX));
+ pszExtension = PathGetSharedLibraryExtensionA(0);
+ cchPattern = 128 + strnlen(pszExtension, MAX_PATH) + 2;
+ pszPattern = (LPSTR)malloc(cchPattern + 1);
+
+ if (!pszPattern)
+ {
+ WLog_ERR(TAG, "malloc failed!");
+ return NULL;
+ }
+
+ if (pszName && pszSubsystem && pszType)
+ {
+ sprintf_s(pszPattern, cchPattern, FREERDP_SHARED_LIBRARY_PREFIX "%s-client-%s-%s.%s",
+ pszName, pszSubsystem, pszType, pszExtension);
+ }
+ else if (pszName && pszType)
+ {
+ sprintf_s(pszPattern, cchPattern, FREERDP_SHARED_LIBRARY_PREFIX "%s-client-?-%s.%s",
+ pszName, pszType, pszExtension);
+ }
+ else if (pszName)
+ {
+ sprintf_s(pszPattern, cchPattern, FREERDP_SHARED_LIBRARY_PREFIX "%s-client*.%s", pszName,
+ pszExtension);
+ }
+ else
+ {
+ sprintf_s(pszPattern, cchPattern, FREERDP_SHARED_LIBRARY_PREFIX "?-client*.%s",
+ pszExtension);
+ }
+
+ cchPattern = strnlen(pszPattern, cchPattern);
+ cchSearchPath = cchInstallPrefix + cchAddinPath + cchPattern + 3;
+ pszSearchPath = (LPSTR)calloc(cchSearchPath + 1, sizeof(char));
+
+ if (!pszSearchPath)
+ {
+ WLog_ERR(TAG, "malloc failed!");
+ free(pszPattern);
+ return NULL;
+ }
+
+ CopyMemory(pszSearchPath, pszInstallPrefix, cchInstallPrefix);
+ pszSearchPath[cchInstallPrefix] = '\0';
+ const HRESULT hr1 = NativePathCchAppendA(pszSearchPath, cchSearchPath + 1, pszAddinPath);
+ const HRESULT hr2 = NativePathCchAppendA(pszSearchPath, cchSearchPath + 1, pszPattern);
+ free(pszPattern);
+
+ if (FAILED(hr1) || FAILED(hr2))
+ {
+ free(pszSearchPath);
+ return NULL;
+ }
+
+ hFind = FindFirstFileUTF8(pszSearchPath, &FindData);
+
+ free(pszSearchPath);
+ nAddins = 0;
+ ppAddins = (FREERDP_ADDIN**)calloc(128, sizeof(FREERDP_ADDIN*));
+
+ if (!ppAddins)
+ {
+ FindClose(hFind);
+ WLog_ERR(TAG, "calloc failed!");
+ return NULL;
+ }
+
+ if (hFind == INVALID_HANDLE_VALUE)
+ return ppAddins;
+
+ do
+ {
+ char* cFileName = NULL;
+ BOOL used = FALSE;
+ FREERDP_ADDIN* pAddin = (FREERDP_ADDIN*)calloc(1, sizeof(FREERDP_ADDIN));
+
+ if (!pAddin)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ goto error_out;
+ }
+
+ cFileName =
+ ConvertWCharNToUtf8Alloc(FindData.cFileName, ARRAYSIZE(FindData.cFileName), NULL);
+ if (!cFileName)
+ goto skip;
+
+ nDashes = 0;
+ for (size_t index = 0; cFileName[index]; index++)
+ nDashes += (cFileName[index] == '-') ? 1 : 0;
+
+ if (nDashes == 1)
+ {
+ size_t len = 0;
+ char* p[2] = { 0 };
+ /* <name>-client.<extension> */
+ p[0] = cFileName;
+ p[1] = strchr(p[0], '-');
+ if (!p[1])
+ goto skip;
+ p[1] += 1;
+
+ len = (size_t)(p[1] - p[0]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cName, p[0], MIN(ARRAYSIZE(pAddin->cName), len - 1));
+
+ pAddin->dwFlags = FREERDP_ADDIN_CLIENT;
+ pAddin->dwFlags |= FREERDP_ADDIN_DYNAMIC;
+ pAddin->dwFlags |= FREERDP_ADDIN_NAME;
+ ppAddins[nAddins++] = pAddin;
+
+ used = TRUE;
+ }
+ else if (nDashes == 2)
+ {
+ size_t len = 0;
+ char* p[4] = { 0 };
+ /* <name>-client-<subsystem>.<extension> */
+ p[0] = cFileName;
+ p[1] = strchr(p[0], '-');
+ if (!p[1])
+ goto skip;
+ p[1] += 1;
+ p[2] = strchr(p[1], '-');
+ if (!p[2])
+ goto skip;
+ p[2] += 1;
+ p[3] = strchr(p[2], '.');
+ if (!p[3])
+ goto skip;
+ p[3] += 1;
+
+ len = (size_t)(p[1] - p[0]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cName, p[0], MIN(ARRAYSIZE(pAddin->cName), len - 1));
+
+ len = (size_t)(p[3] - p[2]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cSubsystem, p[2], MIN(ARRAYSIZE(pAddin->cSubsystem), len - 1));
+
+ pAddin->dwFlags = FREERDP_ADDIN_CLIENT;
+ pAddin->dwFlags |= FREERDP_ADDIN_DYNAMIC;
+ pAddin->dwFlags |= FREERDP_ADDIN_NAME;
+ pAddin->dwFlags |= FREERDP_ADDIN_SUBSYSTEM;
+ ppAddins[nAddins++] = pAddin;
+
+ used = TRUE;
+ }
+ else if (nDashes == 3)
+ {
+ size_t len = 0;
+ char* p[5] = { 0 };
+ /* <name>-client-<subsystem>-<type>.<extension> */
+ p[0] = cFileName;
+ p[1] = strchr(p[0], '-');
+ if (!p[1])
+ goto skip;
+ p[1] += 1;
+ p[2] = strchr(p[1], '-');
+ if (!p[2])
+ goto skip;
+ p[2] += 1;
+ p[3] = strchr(p[2], '-');
+ if (!p[3])
+ goto skip;
+ p[3] += 1;
+ p[4] = strchr(p[3], '.');
+ if (!p[4])
+ goto skip;
+ p[4] += 1;
+
+ len = (size_t)(p[1] - p[0]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cName, p[0], MIN(ARRAYSIZE(pAddin->cName), len - 1));
+
+ len = (size_t)(p[3] - p[2]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cSubsystem, p[2], MIN(ARRAYSIZE(pAddin->cSubsystem), len - 1));
+
+ len = (size_t)(p[4] - p[3]);
+ if (len < 1)
+ {
+ WLog_WARN(TAG, "Skipping file '%s', invalid format", cFileName);
+ goto skip;
+ }
+ strncpy(pAddin->cType, p[3], MIN(ARRAYSIZE(pAddin->cType), len - 1));
+
+ pAddin->dwFlags = FREERDP_ADDIN_CLIENT;
+ pAddin->dwFlags |= FREERDP_ADDIN_DYNAMIC;
+ pAddin->dwFlags |= FREERDP_ADDIN_NAME;
+ pAddin->dwFlags |= FREERDP_ADDIN_SUBSYSTEM;
+ pAddin->dwFlags |= FREERDP_ADDIN_TYPE;
+ ppAddins[nAddins++] = pAddin;
+
+ used = TRUE;
+ }
+
+ skip:
+ free(cFileName);
+ if (!used)
+ free(pAddin);
+
+ } while (FindNextFileW(hFind, &FindData));
+
+ FindClose(hFind);
+ ppAddins[nAddins] = NULL;
+ return ppAddins;
+error_out:
+ FindClose(hFind);
+ freerdp_channels_addin_list_free(ppAddins);
+ return NULL;
+}
+
+FREERDP_ADDIN** freerdp_channels_list_addins(LPCSTR pszName, LPCSTR pszSubsystem, LPCSTR pszType,
+ DWORD dwFlags)
+{
+ if (dwFlags & FREERDP_ADDIN_STATIC)
+ return freerdp_channels_list_client_static_addins(pszName, pszSubsystem, pszType, dwFlags);
+ else if (dwFlags & FREERDP_ADDIN_DYNAMIC)
+ return freerdp_channels_list_dynamic_addins(pszName, pszSubsystem, pszType, dwFlags);
+
+ return NULL;
+}
+
+void freerdp_channels_addin_list_free(FREERDP_ADDIN** ppAddins)
+{
+ if (!ppAddins)
+ return;
+
+ for (size_t index = 0; ppAddins[index] != NULL; index++)
+ free(ppAddins[index]);
+
+ free(ppAddins);
+}
+
+extern const STATIC_ENTRY CLIENT_VirtualChannelEntryEx_TABLE[];
+
+static BOOL freerdp_channels_is_virtual_channel_entry_ex(LPCSTR pszName)
+{
+ for (size_t i = 0; CLIENT_VirtualChannelEntryEx_TABLE[i].name != NULL; i++)
+ {
+ const STATIC_ENTRY* entry = &CLIENT_VirtualChannelEntryEx_TABLE[i];
+
+ if (!strncmp(entry->name, pszName, MAX_PATH))
+ return TRUE;
+ }
+
+ return FALSE;
+}
+
+PVIRTUALCHANNELENTRY freerdp_channels_load_static_addin_entry(LPCSTR pszName, LPCSTR pszSubsystem,
+ LPCSTR pszType, DWORD dwFlags)
+{
+ const STATIC_ADDIN_TABLE* table = CLIENT_STATIC_ADDIN_TABLE;
+ const char* type = NULL;
+
+ if (!pszName)
+ return NULL;
+
+ if (dwFlags & FREERDP_ADDIN_CHANNEL_DYNAMIC)
+ type = "DVCPluginEntry";
+ else if (dwFlags & FREERDP_ADDIN_CHANNEL_DEVICE)
+ type = "DeviceServiceEntry";
+ else if (dwFlags & FREERDP_ADDIN_CHANNEL_STATIC)
+ {
+ if (dwFlags & FREERDP_ADDIN_CHANNEL_ENTRYEX)
+ type = "VirtualChannelEntryEx";
+ else
+ type = "VirtualChannelEntry";
+ }
+
+ for (; table->name != NULL; table++)
+ {
+ if (strncmp(table->name, pszName, MAX_PATH) == 0)
+ {
+ if (type && strncmp(table->type, type, MAX_PATH))
+ continue;
+
+ if (pszSubsystem != NULL)
+ {
+ const STATIC_SUBSYSTEM_ENTRY* subsystems = table->table;
+
+ for (; subsystems->name != NULL; subsystems++)
+ {
+ /* If the pszSubsystem is an empty string use the default backend. */
+ if ((strnlen(pszSubsystem, 1) ==
+ 0) || /* we only want to know if strnlen is > 0 */
+ (strncmp(subsystems->name, pszSubsystem, MAX_PATH) == 0))
+ {
+ if (pszType)
+ {
+ if (strncmp(subsystems->type, pszType, MAX_PATH) == 0)
+ return (PVIRTUALCHANNELENTRY)subsystems->entry;
+ }
+ else
+ {
+ return (PVIRTUALCHANNELENTRY)subsystems->entry;
+ }
+ }
+ }
+ }
+ else
+ {
+ if (dwFlags & FREERDP_ADDIN_CHANNEL_ENTRYEX)
+ {
+ if (!freerdp_channels_is_virtual_channel_entry_ex(pszName))
+ return NULL;
+ }
+
+ return (PVIRTUALCHANNELENTRY)table->entry;
+ }
+ }
+ }
+
+ return NULL;
+}
+
+typedef struct
+{
+ wMessageQueue* queue;
+ wStream* data_in;
+ HANDLE thread;
+ char* channel_name;
+ rdpContext* ctx;
+ LPVOID userdata;
+ MsgHandler msg_handler;
+} msg_proc_internals;
+
+static DWORD WINAPI channel_client_thread_proc(LPVOID userdata)
+{
+ UINT error = CHANNEL_RC_OK;
+ wStream* data = NULL;
+ wMessage message = { 0 };
+ msg_proc_internals* internals = userdata;
+
+ WINPR_ASSERT(internals);
+
+ while (1)
+ {
+ if (!MessageQueue_Wait(internals->queue))
+ {
+ WLog_ERR(TAG, "MessageQueue_Wait failed!");
+ error = ERROR_INTERNAL_ERROR;
+ break;
+ }
+ if (!MessageQueue_Peek(internals->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 = internals->msg_handler(internals->userdata, data)))
+ {
+ WLog_ERR(TAG, "msg_handler failed with error %" PRIu32 "!", error);
+ break;
+ }
+ }
+ }
+ if (error && internals->ctx)
+ {
+ char msg[128];
+ _snprintf(msg, 127,
+ "%s_virtual_channel_client_thread reported an"
+ " error",
+ internals->channel_name);
+ setChannelError(internals->ctx, error, msg);
+ }
+ ExitThread(error);
+ return error;
+}
+
+static void free_msg(void* obj)
+{
+ wMessage* msg = (wMessage*)obj;
+
+ if (msg && (msg->id == 0))
+ {
+ wStream* s = (wStream*)msg->wParam;
+ Stream_Free(s, TRUE);
+ }
+}
+
+static void channel_client_handler_free(msg_proc_internals* internals)
+{
+ if (!internals)
+ return;
+
+ if (internals->thread)
+ CloseHandle(internals->thread);
+ MessageQueue_Free(internals->queue);
+ Stream_Free(internals->data_in, TRUE);
+ free(internals->channel_name);
+ free(internals);
+}
+
+/* Create message queue and thread or not, depending on settings */
+void* channel_client_create_handler(rdpContext* ctx, LPVOID userdata, MsgHandler msg_handler,
+ const char* channel_name)
+{
+ msg_proc_internals* internals = calloc(1, sizeof(msg_proc_internals));
+ if (!internals)
+ {
+ WLog_ERR(TAG, "calloc failed!");
+ return NULL;
+ }
+ internals->msg_handler = msg_handler;
+ internals->userdata = userdata;
+ if (channel_name)
+ {
+ internals->channel_name = _strdup(channel_name);
+ if (!internals->channel_name)
+ goto fail;
+ }
+ WINPR_ASSERT(ctx);
+ WINPR_ASSERT(ctx->settings);
+ internals->ctx = ctx;
+ if ((freerdp_settings_get_uint32(ctx->settings, FreeRDP_ThreadingFlags) &
+ THREADING_FLAGS_DISABLE_THREADS) == 0)
+ {
+ wObject obj = { 0 };
+ obj.fnObjectFree = free_msg;
+ internals->queue = MessageQueue_New(&obj);
+ if (!internals->queue)
+ {
+ WLog_ERR(TAG, "MessageQueue_New failed!");
+ goto fail;
+ }
+
+ if (!(internals->thread =
+ CreateThread(NULL, 0, channel_client_thread_proc, (void*)internals, 0, NULL)))
+ {
+ WLog_ERR(TAG, "CreateThread failed!");
+ goto fail;
+ }
+ }
+ return internals;
+
+fail:
+ channel_client_handler_free(internals);
+ return NULL;
+}
+/* post a message in the queue or directly call the processing handler */
+UINT channel_client_post_message(void* MsgsHandle, LPVOID pData, UINT32 dataLength,
+ UINT32 totalLength, UINT32 dataFlags)
+{
+ msg_proc_internals* internals = MsgsHandle;
+ wStream* data_in = NULL;
+
+ if (!internals)
+ {
+ /* TODO: return some error here */
+ return CHANNEL_RC_OK;
+ }
+
+ if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
+ {
+ return CHANNEL_RC_OK;
+ }
+
+ if (dataFlags & CHANNEL_FLAG_FIRST)
+ {
+ if (internals->data_in)
+ {
+ if (!Stream_EnsureCapacity(internals->data_in, totalLength))
+ return CHANNEL_RC_NO_MEMORY;
+ }
+ else
+ internals->data_in = Stream_New(NULL, totalLength);
+ }
+
+ if (!(data_in = internals->data_in))
+ {
+ WLog_ERR(TAG, "Stream_New failed!");
+ return CHANNEL_RC_NO_MEMORY;
+ }
+
+ if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
+ {
+ Stream_Free(internals->data_in, TRUE);
+ internals->data_in = NULL;
+ return CHANNEL_RC_NO_MEMORY;
+ }
+
+ Stream_Write(data_in, pData, dataLength);
+
+ if (dataFlags & CHANNEL_FLAG_LAST)
+ {
+ if (Stream_Capacity(data_in) != Stream_GetPosition(data_in))
+ {
+ char msg[128];
+ _snprintf(msg, 127, "%s_plugin_process_received: read error", internals->channel_name);
+ WLog_ERR(TAG, msg);
+ return ERROR_INTERNAL_ERROR;
+ }
+
+ internals->data_in = NULL;
+ Stream_SealLength(data_in);
+ Stream_SetPosition(data_in, 0);
+
+ if ((freerdp_settings_get_uint32(internals->ctx->settings, FreeRDP_ThreadingFlags) &
+ THREADING_FLAGS_DISABLE_THREADS) != 0)
+ {
+ UINT error = CHANNEL_RC_OK;
+ if ((error = internals->msg_handler(internals->userdata, data_in)))
+ {
+ WLog_ERR(TAG,
+ "msg_handler failed with error"
+ " %" PRIu32 "!",
+ error);
+ return ERROR_INTERNAL_ERROR;
+ }
+ }
+ else if (!MessageQueue_Post(internals->queue, NULL, 0, (void*)data_in, NULL))
+ {
+ WLog_ERR(TAG, "MessageQueue_Post failed!");
+ return ERROR_INTERNAL_ERROR;
+ }
+ }
+ return CHANNEL_RC_OK;
+}
+/* Tear down queue and thread */
+UINT channel_client_quit_handler(void* MsgsHandle)
+{
+ msg_proc_internals* internals = MsgsHandle;
+ UINT rc = 0;
+ if (!internals)
+ {
+ /* TODO: return some error here */
+ return CHANNEL_RC_OK;
+ }
+
+ WINPR_ASSERT(internals->ctx);
+ WINPR_ASSERT(internals->ctx->settings);
+
+ if ((freerdp_settings_get_uint32(internals->ctx->settings, FreeRDP_ThreadingFlags) &
+ THREADING_FLAGS_DISABLE_THREADS) == 0)
+ {
+ if (internals->queue && internals->thread)
+ {
+ if (MessageQueue_PostQuit(internals->queue, 0) &&
+ (WaitForSingleObject(internals->thread, INFINITE) == WAIT_FAILED))
+ {
+ rc = GetLastError();
+ WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "", rc);
+ return rc;
+ }
+ }
+ }
+
+ channel_client_handler_free(internals);
+ return CHANNEL_RC_OK;
+}