22#include <freerdp/config.h>
25#include <winpr/cast.h>
26#include <winpr/stream.h>
27#include <winpr/interlocked.h>
29#include <freerdp/freerdp.h>
30#include <freerdp/channels/drdynvc.h>
31#include <freerdp/utils/drdynvc.h>
32#include <freerdp/codec/zgfx.h>
34#include "drdynvc_main.h"
36#define TAG CHANNELS_TAG("drdynvc.client")
38static const char* channel_state2str(DVC_CHANNEL_STATE state)
42 case DVC_CHANNEL_INIT:
43 return "DVC_CHANNEL_INIT";
44 case DVC_CHANNEL_RUNNING:
45 return "DVC_CHANNEL_RUNNING";
46 case DVC_CHANNEL_CLOSED:
47 return "DVC_CHANNEL_CLOSED";
49 return "DVC_CHANNEL_UNKNOWN";
54static UINT dvcman_channel_close(
DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn);
55static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr);
56static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId,
const BYTE* data,
57 UINT32 dataSize, BOOL* close);
58static UINT drdynvc_send(drdynvcPlugin* drdynvc,
wStream* s);
63 free(listener->channel_name);
72static UINT dvcman_get_configuration(IWTSListener* pListener,
void** ppPropertyBag)
74 WINPR_ASSERT(ppPropertyBag);
75 WINPR_UNUSED(pListener);
76 *ppPropertyBag =
nullptr;
77 return ERROR_INTERNAL_ERROR;
85static UINT dvcman_create_listener(IWTSVirtualChannelManager* pChannelMgr,
86 const char* pszChannelName, ULONG ulFlags,
87 IWTSListenerCallback* pListenerCallback,
88 IWTSListener** ppListener)
94 WLog_DBG(TAG,
"create_listener: %" PRIuz
".%s.", HashTable_Count(dvcman->listeners) + 1,
100 WLog_ERR(TAG,
"calloc failed!");
101 return CHANNEL_RC_NO_MEMORY;
104 listener->iface.GetConfiguration = dvcman_get_configuration;
105 listener->iface.pInterface =
nullptr;
106 listener->dvcman = dvcman;
107 listener->channel_name = _strdup(pszChannelName);
109 if (!listener->channel_name)
111 WLog_ERR(TAG,
"_strdup failed!");
112 dvcman_wtslistener_free(listener);
113 return CHANNEL_RC_NO_MEMORY;
116 listener->flags = ulFlags;
117 listener->listener_callback = pListenerCallback;
120 *ppListener = (IWTSListener*)listener;
122 if (!HashTable_Insert(dvcman->listeners, listener->channel_name, listener))
124 dvcman_wtslistener_free(listener);
125 return ERROR_INTERNAL_ERROR;
129 return CHANNEL_RC_OK;
132static UINT dvcman_destroy_listener(IWTSVirtualChannelManager* pChannelMgr, IWTSListener* pListener)
136 WINPR_UNUSED(pChannelMgr);
140 DVCMAN* dvcman = listener->dvcman;
142 HashTable_Remove(dvcman->listeners, listener->channel_name);
145 return CHANNEL_RC_OK;
153static UINT dvcman_register_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints,
const char* name,
156 WINPR_ASSERT(pEntryPoints);
159 WINPR_ASSERT(dvcman);
160 if (!ArrayList_Append(dvcman->plugin_names, name))
161 return ERROR_INTERNAL_ERROR;
162 if (!ArrayList_Append(dvcman->plugins, pPlugin))
163 return ERROR_INTERNAL_ERROR;
165 WLog_DBG(TAG,
"register_plugin: num_plugins %" PRIuz, ArrayList_Count(dvcman->plugins));
166 return CHANNEL_RC_OK;
169static IWTSPlugin* dvcman_get_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints,
const char* name)
171 IWTSPlugin* plugin =
nullptr;
174 WINPR_ASSERT(pEntryPoints);
176 if (!dvcman || !pEntryPoints || !name)
179 nc = ArrayList_Count(dvcman->plugin_names);
180 pc = ArrayList_Count(dvcman->plugins);
184 ArrayList_Lock(dvcman->plugin_names);
185 ArrayList_Lock(dvcman->plugins);
186 for (
size_t i = 0; i < pc; i++)
188 const char* cur = ArrayList_GetItem(dvcman->plugin_names, i);
189 if (strcmp(cur, name) == 0)
191 plugin = ArrayList_GetItem(dvcman->plugins, i);
195 ArrayList_Unlock(dvcman->plugin_names);
196 ArrayList_Unlock(dvcman->plugins);
200static const ADDIN_ARGV* dvcman_get_plugin_data(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
202 WINPR_ASSERT(pEntryPoints);
206static rdpContext* dvcman_get_rdp_context(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
210 return entry->context;
213static rdpSettings* dvcman_get_rdp_settings(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
215 rdpContext* context = dvcman_get_rdp_context(pEntryPoints);
216 WINPR_ASSERT(context);
218 return context->settings;
221static UINT32 dvcman_get_channel_id(IWTSVirtualChannel* channel)
225 return dvc->channel_id;
228static const char* dvcman_get_channel_name(IWTSVirtualChannel* channel)
232 return dvc->channel_name;
235static DVCMAN_CHANNEL* dvcman_get_channel_by_id(IWTSVirtualChannelManager* pChannelMgr,
236 UINT32 ChannelId, BOOL doRef)
241 WINPR_ASSERT(dvcman);
242 HashTable_Lock(dvcman->channelsById);
243 dvcChannel = HashTable_GetItemValue(dvcman->channelsById, &ChannelId);
247 InterlockedIncrement(&dvcChannel->refCounter);
250 HashTable_Unlock(dvcman->channelsById);
254static IWTSVirtualChannel* dvcman_find_channel_by_id(IWTSVirtualChannelManager* pChannelMgr,
257 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE);
261 return &channel->iface;
264static void dvcman_plugin_terminate(
void* plugin)
266 IWTSPlugin* pPlugin = plugin;
268 WINPR_ASSERT(pPlugin);
269 UINT error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Terminated, pPlugin);
270 if (error != CHANNEL_RC_OK)
271 WLog_ERR(TAG,
"Terminated failed with error %" PRIu32
"!", error);
274static void wts_listener_free(
void* arg)
277 dvcman_wtslistener_free(listener);
280static BOOL channelIdMatch(
const void* k1,
const void* k2)
284 return *((
const UINT32*)k1) == *((
const UINT32*)k2);
287static UINT32 channelIdHash(
const void*
id)
290 return *((
const UINT32*)
id);
293static void channelByIdCleanerFn(
void* value)
298 dvcman_channel_close(channel, FALSE, TRUE);
299 dvcman_channel_free(channel);
303static IWTSVirtualChannelManager* dvcman_new(drdynvcPlugin* plugin)
311 dvcman->iface.CreateListener = dvcman_create_listener;
312 dvcman->iface.DestroyListener = dvcman_destroy_listener;
313 dvcman->iface.FindChannelById = dvcman_find_channel_by_id;
314 dvcman->iface.GetChannelId = dvcman_get_channel_id;
315 dvcman->iface.GetChannelName = dvcman_get_channel_name;
316 dvcman->drdynvc = plugin;
317 dvcman->channelsById = HashTable_New(TRUE);
319 if (!dvcman->channelsById)
322 if (!HashTable_SetHashFunction(dvcman->channelsById, channelIdHash))
325 obj = HashTable_KeyObject(dvcman->channelsById);
329 obj = HashTable_ValueObject(dvcman->channelsById);
333 dvcman->pool = StreamPool_New(TRUE, 10);
337 dvcman->listeners = HashTable_New(TRUE);
338 if (!dvcman->listeners)
341 if (!HashTable_SetHashFunction(dvcman->listeners, HashTable_StringHash))
344 obj = HashTable_KeyObject(dvcman->listeners);
347 obj = HashTable_ValueObject(dvcman->listeners);
350 dvcman->plugin_names = ArrayList_New(TRUE);
351 if (!dvcman->plugin_names)
353 obj = ArrayList_Object(dvcman->plugin_names);
357 dvcman->plugins = ArrayList_New(TRUE);
358 if (!dvcman->plugins)
360 obj = ArrayList_Object(dvcman->plugins);
362 return &dvcman->iface;
364 dvcman_free(plugin, &dvcman->iface);
373static UINT dvcman_load_addin(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr,
376 WINPR_ASSERT(drdynvc);
377 WINPR_ASSERT(pChannelMgr);
379 WINPR_ASSERT(context);
381 WLog_Print(drdynvc->log, WLOG_INFO,
"Loading Dynamic Virtual Channel %s", args->argv[0]);
383 PVIRTUALCHANNELENTRY pvce = freerdp_load_channel_addin_entry(args->argv[0],
nullptr,
nullptr,
384 FREERDP_ADDIN_CHANNEL_DYNAMIC);
385 PDVC_PLUGIN_ENTRY pDVCPluginEntry = WINPR_FUNC_PTR_CAST(pvce, PDVC_PLUGIN_ENTRY);
391 entryPoints.iface.RegisterPlugin = dvcman_register_plugin;
392 entryPoints.iface.GetPlugin = dvcman_get_plugin;
393 entryPoints.iface.GetPluginData = dvcman_get_plugin_data;
394 entryPoints.iface.GetRdpSettings = dvcman_get_rdp_settings;
395 entryPoints.iface.GetRdpContext = dvcman_get_rdp_context;
396 entryPoints.dvcman = (
DVCMAN*)pChannelMgr;
397 entryPoints.args = args;
398 entryPoints.context = context;
399 return pDVCPluginEntry(&entryPoints.iface);
402 return ERROR_INVALID_FUNCTION;
410 if (channel->dvc_data)
411 Stream_Release(channel->dvc_data);
413 zgfx_context_free(channel->decompressor);
414 DeleteCriticalSection(&(channel->lock));
415 free(channel->channel_name);
421 WINPR_ASSERT(channel);
422 if (InterlockedDecrement(&channel->refCounter))
425 DVCMAN* dvcman = channel->dvcman;
427 HashTable_Remove(dvcman->channelsById, &channel->channel_id);
432 WINPR_ASSERT(channel);
433 DVCMAN* dvcman = channel->dvcman;
434 drdynvcPlugin* drdynvc = dvcman->drdynvc;
435 wStream* s = StreamPool_Take(dvcman->pool, 5);
439 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
440 return CHANNEL_RC_NO_MEMORY;
443 Stream_Write_UINT8(s, (CLOSE_REQUEST_PDU << 4) | 0x02);
444 Stream_Write_UINT32(s, channel->channel_id);
445 return drdynvc_send(drdynvc, s);
450 WINPR_ASSERT(channel);
452 IWTSVirtualChannelCallback* cb = channel->channel_callback;
453 const char* name = channel->channel_name;
454 const UINT32
id = channel->channel_id;
457 if (!cb->OnOpen || !cb->OnClose || !cb->OnDataReceived)
458 WLog_VRB(TAG,
"{%s:%" PRIu32
"} OnOpen=%p, OnClose=%p, OnDataReceived=%p", name,
id,
459 WINPR_FUNC_PTR_CAST(cb->OnOpen,
const void*),
460 WINPR_FUNC_PTR_CAST(cb->OnClose,
const void*),
461 WINPR_FUNC_PTR_CAST(cb->OnDataReceived,
const void*));
466 WINPR_ASSERT(channel);
469 IWTSVirtualChannelCallback* cb = channel->channel_callback;
472 check_open_close_receive(channel);
473 WINPR_ASSERT(cb->OnDataReceived);
474 return cb->OnDataReceived(cb, data);
477static UINT dvcman_channel_close(
DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn)
479 UINT error = CHANNEL_RC_OK;
480 DrdynvcClientContext* context =
nullptr;
482 WINPR_ASSERT(channel);
483 switch (channel->state)
485 case DVC_CHANNEL_INIT:
487 case DVC_CHANNEL_RUNNING:
490 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
491 WINPR_ASSERT(drdynvc);
492 context = drdynvc->context;
494 WLog_Print(drdynvc->log, WLOG_DEBUG,
"sending close confirm for '%s'",
495 channel->channel_name);
497 error = dvcchannel_send_close(channel);
498 if (error != CHANNEL_RC_OK)
501 WLog_Print(drdynvc->log, WLOG_DEBUG,
502 "error when sending closeRequest for '%s'",
503 channel->channel_name);
505 WLog_Print(drdynvc->log, WLOG_DEBUG,
506 "error when sending close confirm for '%s'",
507 channel->channel_name);
509 WLog_Print(drdynvc->log, WLOG_DEBUG,
"listener %s destroyed channel %" PRIu32
"",
510 channel->channel_name, channel->channel_id);
513 channel->state = DVC_CHANNEL_CLOSED;
516 check_open_close_receive(channel);
518 IWTSVirtualChannelCallback* cb = channel->channel_callback;
519 channel->channel_callback =
nullptr;
521 error = IFCALLRESULT(CHANNEL_RC_OK, cb->OnClose, cb);
524 if (channel->dvcman && channel->dvcman->drdynvc)
528 IFCALLRET(context->OnChannelDisconnected, error, context, channel->channel_name,
529 channel->pInterface);
533 if (!fromHashTableFn)
534 dvcman_channel_unref(channel);
536 case DVC_CHANNEL_CLOSED:
545static DVCMAN_CHANNEL* dvcman_channel_new(WINPR_ATTR_UNUSED drdynvcPlugin* drdynvc,
546 IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId,
547 const char* ChannelName)
551 WINPR_ASSERT(drdynvc);
552 WINPR_ASSERT(pChannelMgr);
558 channel->dvcman = (
DVCMAN*)pChannelMgr;
559 channel->channel_id = ChannelId;
560 channel->refCounter = 1;
561 channel->state = DVC_CHANNEL_INIT;
562 channel->channel_name = _strdup(ChannelName);
563 if (!channel->channel_name)
566 channel->decompressor = zgfx_context_new(FALSE);
567 if (!channel->decompressor)
570 if (!InitializeCriticalSectionEx(&(channel->lock), 0, 0))
575 dvcman_channel_free(channel);
579static void dvcman_clear(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
583 WINPR_ASSERT(dvcman);
584 WINPR_UNUSED(drdynvc);
586 HashTable_Clear(dvcman->channelsById);
587 ArrayList_Clear(dvcman->plugins);
588 ArrayList_Clear(dvcman->plugin_names);
589 HashTable_Clear(dvcman->listeners);
591static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
595 WINPR_ASSERT(dvcman);
596 WINPR_UNUSED(drdynvc);
598 HashTable_Free(dvcman->channelsById);
599 ArrayList_Free(dvcman->plugins);
600 ArrayList_Free(dvcman->plugin_names);
601 HashTable_Free(dvcman->listeners);
603 StreamPool_Free(dvcman->pool);
612static UINT dvcman_init(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
615 UINT error = CHANNEL_RC_OK;
617 WINPR_ASSERT(dvcman);
618 ArrayList_Lock(dvcman->plugins);
619 for (
size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
621 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
623 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Initialize, pPlugin, pChannelMgr);
624 if (error != CHANNEL_RC_OK)
626 WLog_Print(drdynvc->log, WLOG_ERROR,
"Initialize failed with error %" PRIu32
"!",
633 ArrayList_Unlock(dvcman->plugins);
642static UINT dvcman_write_channel(IWTSVirtualChannel* pChannel, ULONG cbSize,
const BYTE* pBuffer,
649 WINPR_UNUSED(pReserved);
650 if (!channel || !channel->dvcman)
651 return CHANNEL_RC_BAD_CHANNEL;
653 EnterCriticalSection(&(channel->lock));
655 drdynvc_write_data(channel->dvcman->drdynvc, channel->channel_id, pBuffer, cbSize, &close);
656 LeaveCriticalSection(&(channel->lock));
659 dvcman_channel_close(channel, FALSE, FALSE);
669static UINT dvcman_close_channel_iface(IWTSVirtualChannel* pChannel)
674 return CHANNEL_RC_BAD_CHANNEL;
676 WLog_DBG(TAG,
"close_channel_iface: id=%" PRIu32
"", channel->channel_id);
677 return dvcman_channel_close(channel, FALSE, FALSE);
685static DVCMAN_CHANNEL* dvcman_create_channel(drdynvcPlugin* drdynvc,
686 IWTSVirtualChannelManager* pChannelMgr,
687 UINT32 ChannelId,
const char* ChannelName, UINT* res)
691 DrdynvcClientContext* context =
nullptr;
694 IWTSVirtualChannelCallback* pCallback =
nullptr;
696 WINPR_ASSERT(dvcman);
699 HashTable_Lock(dvcman->listeners);
700 listener = (
DVCMAN_LISTENER*)HashTable_GetItemValue(dvcman->listeners, ChannelName);
703 *res = ERROR_NOT_FOUND;
707 channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE);
710 switch (channel->state)
712 case DVC_CHANNEL_RUNNING:
713 WLog_Print(drdynvc->log, WLOG_ERROR,
714 "Protocol error: Duplicated ChannelId %" PRIu32
" (%s)!", ChannelId,
716 *res = CHANNEL_RC_ALREADY_OPEN;
719 case DVC_CHANNEL_CLOSED:
720 case DVC_CHANNEL_INIT:
723 WLog_Print(drdynvc->log, WLOG_ERROR,
"not expecting a createChannel from state %s",
724 channel_state2str(channel->state));
725 *res = CHANNEL_RC_INITIALIZATION_ERROR;
732 if (!(channel = dvcman_channel_new(drdynvc, pChannelMgr, ChannelId, ChannelName)))
734 WLog_Print(drdynvc->log, WLOG_ERROR,
"dvcman_channel_new failed!");
735 *res = CHANNEL_RC_NO_MEMORY;
740 if (!HashTable_Insert(dvcman->channelsById, &channel->channel_id, channel))
742 WLog_Print(drdynvc->log, WLOG_ERROR,
"unable to register channel in our channel list");
743 *res = ERROR_INTERNAL_ERROR;
744 dvcman_channel_free(channel);
749 channel->iface.Write = dvcman_write_channel;
750 channel->iface.Close = dvcman_close_channel_iface;
753 *res = listener->listener_callback->OnNewChannelConnection(
754 listener->listener_callback, &channel->iface,
nullptr, &bAccept, &pCallback);
756 if (*res != CHANNEL_RC_OK)
758 WLog_Print(drdynvc->log, WLOG_ERROR,
759 "OnNewChannelConnection failed with error %" PRIu32
"!", *res);
760 *res = ERROR_INTERNAL_ERROR;
761 dvcman_channel_unref(channel);
767 WLog_Print(drdynvc->log, WLOG_ERROR,
"OnNewChannelConnection returned with bAccept FALSE!");
768 *res = ERROR_INTERNAL_ERROR;
769 dvcman_channel_unref(channel);
774 WLog_Print(drdynvc->log, WLOG_DEBUG,
"listener %s created new channel %" PRIu32
"",
775 listener->channel_name, channel->channel_id);
776 channel->state = DVC_CHANNEL_RUNNING;
777 channel->channel_callback = pCallback;
778 channel->pInterface = listener->iface.pInterface;
779 context = dvcman->drdynvc->context;
781 IFCALLRET(context->OnChannelConnected, *res, context, ChannelName, listener->iface.pInterface);
782 if (*res != CHANNEL_RC_OK)
784 WLog_Print(drdynvc->log, WLOG_ERROR,
785 "context.OnChannelConnected failed with error %" PRIu32
"", *res);
789 HashTable_Unlock(dvcman->listeners);
799static UINT dvcman_open_channel(drdynvcPlugin* drdynvc,
DVCMAN_CHANNEL* channel)
801 UINT error = CHANNEL_RC_OK;
803 WINPR_ASSERT(drdynvc);
804 WINPR_ASSERT(channel);
805 if (channel->state == DVC_CHANNEL_RUNNING)
807 IWTSVirtualChannelCallback* pCallback = channel->channel_callback;
809 if (pCallback->OnOpen)
811 check_open_close_receive(channel);
812 error = pCallback->OnOpen(pCallback);
815 WLog_Print(drdynvc->log, WLOG_ERROR,
"OnOpen failed with error %" PRIu32
"!",
821 WLog_Print(drdynvc->log, WLOG_DEBUG,
"open_channel: ChannelId %" PRIu32
"",
822 channel->channel_id);
834static UINT dvcman_receive_channel_data_first(
DVCMAN_CHANNEL* channel, UINT32 length)
836 WINPR_ASSERT(channel);
837 WINPR_ASSERT(channel->dvcman);
838 if (channel->dvc_data)
839 Stream_Release(channel->dvc_data);
841 channel->dvc_data = StreamPool_Take(channel->dvcman->pool, length);
843 if (!channel->dvc_data)
845 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
846 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
847 return CHANNEL_RC_NO_MEMORY;
850 channel->dvc_data_length = length;
851 return CHANNEL_RC_OK;
860 WINPR_ATTR_UNUSED UINT32 ThreadingFlags)
862 UINT status = CHANNEL_RC_OK;
863 size_t dataSize = Stream_GetRemainingLength(data);
865 WINPR_ASSERT(channel);
866 WINPR_ASSERT(channel->dvcman);
867 if (channel->dvc_data)
869 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
872 if (Stream_GetPosition(channel->dvc_data) + dataSize > channel->dvc_data_length)
874 WLog_Print(drdynvc->log, WLOG_ERROR,
"data exceeding declared length!");
875 Stream_Release(channel->dvc_data);
876 channel->dvc_data =
nullptr;
877 status = ERROR_INVALID_DATA;
881 Stream_Copy(data, channel->dvc_data, dataSize);
883 if (Stream_GetPosition(channel->dvc_data) >= channel->dvc_data_length)
885 Stream_SealLength(channel->dvc_data);
886 Stream_ResetPosition(channel->dvc_data);
888 status = dvcman_call_on_receive(channel, channel->dvc_data);
889 Stream_Release(channel->dvc_data);
890 channel->dvc_data =
nullptr;
894 status = dvcman_call_on_receive(channel, data);
900static UINT8 drdynvc_write_variable_uint(
wStream* s, UINT32 val)
907 Stream_Write_UINT8(s, (UINT8)val);
909 else if (val <= 0xFFFF)
912 Stream_Write_UINT16(s, (UINT16)val);
917 Stream_Write_UINT32(s, val);
928static UINT drdynvc_send(drdynvcPlugin* drdynvc,
wStream* s)
933 status = CHANNEL_RC_BAD_CHANNEL_HANDLE;
936 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelWriteEx);
937 status = drdynvc->channelEntryPoints.pVirtualChannelWriteEx(
938 drdynvc->InitHandle, drdynvc->OpenHandle, Stream_Buffer(s),
939 (UINT32)Stream_GetPosition(s), s);
945 return CHANNEL_RC_OK;
947 case CHANNEL_RC_NOT_CONNECTED:
949 return CHANNEL_RC_OK;
951 case CHANNEL_RC_BAD_CHANNEL_HANDLE:
953 WLog_ERR(TAG,
"VirtualChannelWriteEx failed with CHANNEL_RC_BAD_CHANNEL_HANDLE");
958 WLog_Print(drdynvc->log, WLOG_ERROR,
959 "VirtualChannelWriteEx failed with %s [%08" PRIX32
"]",
960 WTSErrorToString(status), status);
970static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId,
const BYTE* data,
971 UINT32 dataSize, BOOL* close)
976 UINT status = CHANNEL_RC_BAD_INIT_HANDLE;
980 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
982 dvcman = (
DVCMAN*)drdynvc->channel_mgr;
983 WINPR_ASSERT(dvcman);
985 WLog_Print(drdynvc->log, WLOG_TRACE,
"write_data: ChannelId=%" PRIu32
" size=%" PRIu32
"",
986 ChannelId, dataSize);
987 wStream* data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH);
991 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
992 return CHANNEL_RC_NO_MEMORY;
995 if (!Stream_SetPosition(data_out, 1))
997 Stream_Release(data_out);
998 return ERROR_INVALID_DATA;
1000 cbChId = drdynvc_write_variable_uint(data_out, ChannelId);
1001 pos = Stream_GetPosition(data_out);
1007 Stream_Release(data_out);
1009 else if (dataSize <= CHANNEL_CHUNK_LENGTH - pos)
1011 Stream_ResetPosition(data_out);
1012 Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId);
1013 if (!Stream_SetPosition(data_out, pos))
1015 Stream_Release(data_out);
1016 return ERROR_INVALID_DATA;
1018 Stream_Write(data_out, data, dataSize);
1019 status = drdynvc_send(drdynvc, data_out);
1024 cbLen = drdynvc_write_variable_uint(data_out, dataSize);
1025 pos = Stream_GetPosition(data_out);
1026 Stream_ResetPosition(data_out);
1028 const INT32 pdu = (DATA_FIRST_PDU << 4) | cbChId | (cbLen << 2);
1029 Stream_Write_UINT8(data_out, WINPR_ASSERTING_INT_CAST(UINT8, pdu));
1030 if (!Stream_SetPosition(data_out, pos))
1032 Stream_Release(data_out);
1033 return ERROR_INVALID_DATA;
1037 WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH);
1038 const uint32_t chunkLength =
1039 CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos);
1040 Stream_Write(data_out, data, chunkLength);
1042 data += chunkLength;
1043 dataSize -= chunkLength;
1045 status = drdynvc_send(drdynvc, data_out);
1047 while (status == CHANNEL_RC_OK && dataSize > 0)
1049 data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH);
1053 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
1054 return CHANNEL_RC_NO_MEMORY;
1057 if (!Stream_SetPosition(data_out, 1))
1059 Stream_Release(data_out);
1060 return ERROR_INVALID_DATA;
1063 cbChId = drdynvc_write_variable_uint(data_out, ChannelId);
1064 pos = Stream_GetPosition(data_out);
1065 Stream_ResetPosition(data_out);
1066 Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId);
1067 if (!Stream_SetPosition(data_out, pos))
1069 Stream_Release(data_out);
1070 return ERROR_INVALID_DATA;
1073 uint32_t chunkLength = dataSize;
1075 WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH);
1076 if (chunkLength > CHANNEL_CHUNK_LENGTH - pos)
1077 chunkLength = CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos);
1079 Stream_Write(data_out, data, chunkLength);
1080 data += chunkLength;
1081 dataSize -= chunkLength;
1082 status = drdynvc_send(drdynvc, data_out);
1086 if (status != CHANNEL_RC_OK)
1088 WLog_Print(drdynvc->log, WLOG_ERROR,
"VirtualChannelWriteEx failed with %s [%08" PRIX32
"]",
1089 WTSErrorToString(status), status);
1093 return CHANNEL_RC_OK;
1101static UINT drdynvc_send_capability_response(drdynvcPlugin* drdynvc)
1105 DVCMAN* dvcman =
nullptr;
1108 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1110 dvcman = (
DVCMAN*)drdynvc->channel_mgr;
1111 WINPR_ASSERT(dvcman);
1113 WLog_Print(drdynvc->log, WLOG_TRACE,
"capability_response");
1114 s = StreamPool_Take(dvcman->pool, 4);
1118 WLog_Print(drdynvc->log, WLOG_ERROR,
"Stream_Ndrdynvc_write_variable_uintew failed!");
1119 return CHANNEL_RC_NO_MEMORY;
1122 Stream_Write_UINT16(s, 0x0050);
1123 Stream_Write_UINT16(s, drdynvc->version);
1124 status = drdynvc_send(drdynvc, s);
1126 if (status != CHANNEL_RC_OK)
1128 WLog_Print(drdynvc->log, WLOG_ERROR,
"VirtualChannelWriteEx failed with %s [%08" PRIX32
"]",
1129 WTSErrorToString(status), status);
1140static UINT drdynvc_process_capability_request(drdynvcPlugin* drdynvc,
int Sp,
int cbChId,
1146 return CHANNEL_RC_BAD_INIT_HANDLE;
1148 if (!Stream_CheckAndLogRequiredLength(TAG, s, 3))
1149 return ERROR_INVALID_DATA;
1151 WLog_Print(drdynvc->log, WLOG_TRACE,
"capability_request Sp=%d cbChId=%d", Sp, cbChId);
1153 Stream_Read_UINT16(s, drdynvc->version);
1158 if ((drdynvc->version == 2) || (drdynvc->version == 3))
1160 if (!Stream_CheckAndLogRequiredLength(TAG, s, 8))
1161 return ERROR_INVALID_DATA;
1163 Stream_Read_UINT16(s, drdynvc->PriorityCharge0);
1164 Stream_Read_UINT16(s, drdynvc->PriorityCharge1);
1165 Stream_Read_UINT16(s, drdynvc->PriorityCharge2);
1166 Stream_Read_UINT16(s, drdynvc->PriorityCharge3);
1169 status = drdynvc_send_capability_response(drdynvc);
1170 drdynvc->state = DRDYNVC_STATE_READY;
1174static UINT32 drdynvc_cblen_to_bytes(
int cbLen)
1189static UINT32 drdynvc_read_variable_uint(
wStream* s,
int cbLen)
1196 Stream_Read_UINT8(s, val);
1200 Stream_Read_UINT16(s, val);
1204 Stream_Read_UINT32(s, val);
1216static UINT drdynvc_process_create_request(drdynvcPlugin* drdynvc, UINT8 Sp, UINT8 cbChId,
1221 UINT channel_status = 0;
1222 DVCMAN* dvcman =
nullptr;
1224 INT32 retStatus = 0;
1228 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1230 dvcman = (
DVCMAN*)drdynvc->channel_mgr;
1231 WINPR_ASSERT(dvcman);
1233 if (drdynvc->state == DRDYNVC_STATE_CAPABILITIES)
1240 drdynvc->version = 3;
1242 if ((status = drdynvc_send_capability_response(drdynvc)))
1244 WLog_Print(drdynvc->log, WLOG_ERROR,
"drdynvc_send_capability_response failed!");
1248 drdynvc->state = DRDYNVC_STATE_READY;
1251 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1252 return ERROR_INVALID_DATA;
1254 const UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1255 const size_t pos = Stream_GetPosition(s);
1256 const char* name = Stream_ConstPointer(s);
1257 const size_t length = Stream_GetRemainingLength(s);
1259 if (strnlen(name, length) >= length)
1260 return ERROR_INVALID_DATA;
1262 WLog_Print(drdynvc->log, WLOG_DEBUG,
1263 "process_create_request: ChannelId=%" PRIu32
" ChannelName=%s", ChannelId, name);
1265 data_out = StreamPool_Take(dvcman->pool, pos + 4);
1268 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
1269 return CHANNEL_RC_NO_MEMORY;
1272 Stream_Write_UINT8(data_out, (CREATE_REQUEST_PDU << 4) | cbChId);
1273 if (!Stream_SetPosition(s, 1))
1274 return ERROR_INVALID_DATA;
1275 Stream_Copy(s, data_out, pos - 1);
1278 dvcman_create_channel(drdynvc, drdynvc->channel_mgr, ChannelId, name, &channel_status);
1279 switch (channel_status)
1282 WLog_Print(drdynvc->log, WLOG_DEBUG,
"channel created");
1285 case CHANNEL_RC_NO_MEMORY:
1286 WLog_Print(drdynvc->log, WLOG_DEBUG,
"not enough memory for channel creation");
1287 retStatus = STATUS_NO_MEMORY;
1289 case ERROR_NOT_FOUND:
1290 WLog_Print(drdynvc->log, WLOG_DEBUG,
"no listener for '%s'", name);
1291 retStatus = STATUS_NOT_FOUND;
1294 WLog_Print(drdynvc->log, WLOG_DEBUG,
"channel creation error");
1295 retStatus = STATUS_UNSUCCESSFUL;
1298 Stream_Write_INT32(data_out, retStatus);
1300 status = drdynvc_send(drdynvc, data_out);
1301 if (status != CHANNEL_RC_OK)
1303 WLog_Print(drdynvc->log, WLOG_ERROR,
"VirtualChannelWriteEx failed with %s [%08" PRIX32
"]",
1304 WTSErrorToString(status), status);
1305 dvcman_channel_unref(channel);
1309 if (channel_status == CHANNEL_RC_OK)
1311 if ((status = dvcman_open_channel(drdynvc, channel)))
1313 WLog_Print(drdynvc->log, WLOG_ERROR,
1314 "dvcman_open_channel failed with error %" PRIu32
"!", status);
1327static UINT drdynvc_process_data_first(drdynvcPlugin* drdynvc,
int Sp,
int cbChId,
wStream* s,
1328 BOOL compressed, UINT32 ThreadingFlags)
1330 WINPR_ASSERT(drdynvc);
1331 if (!Stream_CheckAndLogRequiredLength(
1332 TAG, s, drdynvc_cblen_to_bytes(cbChId) + drdynvc_cblen_to_bytes(Sp)))
1333 return ERROR_INVALID_DATA;
1335 UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1336 UINT32 Length = drdynvc_read_variable_uint(s, Sp);
1337 WLog_Print(drdynvc->log, WLOG_TRACE,
1338 "process_data_first: Sp=%d cbChId=%d, ChannelId=%" PRIu32
" Length=%" PRIu32
"", Sp,
1339 cbChId, ChannelId, Length);
1341 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1349 WLog_Print(drdynvc->log, WLOG_ERROR,
"ChannelId %" PRIu32
" not found!", ChannelId);
1350 return CHANNEL_RC_OK;
1353 UINT status = CHANNEL_RC_OK;
1354 BOOL shouldFree = FALSE;
1355 if (channel->state != DVC_CHANNEL_RUNNING)
1360 BYTE* data =
nullptr;
1361 UINT32 dataSize = 0;
1362 if (zgfx_decompress(channel->decompressor, Stream_Pointer(s),
1363 WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data,
1366 status = ERROR_INVALID_DATA;
1367 WLog_Print(drdynvc->log, WLOG_ERROR,
"error de-compressing first packet");
1371 s = Stream_New(data, dataSize);
1374 status = CHANNEL_RC_NO_MEMORY;
1375 WLog_Print(drdynvc->log, WLOG_ERROR,
"error allocating new Stream(len=%" PRIu32
")",
1383 status = dvcman_receive_channel_data_first(channel, Length);
1385 if (status == CHANNEL_RC_OK)
1386 status = dvcman_receive_channel_data(channel, s, ThreadingFlags);
1388 if (status != CHANNEL_RC_OK)
1389 status = dvcman_channel_close(channel, FALSE, FALSE);
1393 Stream_Free(s, TRUE);
1394 dvcman_channel_unref(channel);
1403static UINT drdynvc_process_data(drdynvcPlugin* drdynvc,
int Sp,
int cbChId,
wStream* s,
1404 BOOL compressed, UINT32 ThreadingFlags)
1406 WINPR_ASSERT(drdynvc);
1407 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1408 return ERROR_INVALID_DATA;
1410 UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1411 WLog_Print(drdynvc->log, WLOG_TRACE,
"process_data: Sp=%d cbChId=%d, ChannelId=%" PRIu32
"", Sp,
1414 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1422 WLog_Print(drdynvc->log, WLOG_ERROR,
"ChannelId %" PRIu32
" not found!", ChannelId);
1423 return CHANNEL_RC_OK;
1426 BOOL shouldFree = FALSE;
1427 UINT status = CHANNEL_RC_OK;
1428 if (channel->state != DVC_CHANNEL_RUNNING)
1433 BYTE* data =
nullptr;
1434 UINT32 dataSize = 0;
1436 if (zgfx_decompress(channel->decompressor, Stream_Pointer(s),
1437 WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data,
1440 status = ERROR_INVALID_DATA;
1441 WLog_Print(drdynvc->log, WLOG_ERROR,
"error de-compressing data packet");
1445 s = Stream_New(data, dataSize);
1448 status = CHANNEL_RC_NO_MEMORY;
1449 WLog_Print(drdynvc->log, WLOG_ERROR,
"error allocating new Stream(len=%" PRIu32
")",
1457 status = dvcman_receive_channel_data(channel, s, ThreadingFlags);
1458 if (status != CHANNEL_RC_OK)
1459 status = dvcman_channel_close(channel, FALSE, FALSE);
1463 Stream_Free(s, TRUE);
1464 dvcman_channel_unref(channel);
1473static UINT drdynvc_process_close_request(drdynvcPlugin* drdynvc,
int Sp,
int cbChId,
wStream* s)
1475 UINT32 ChannelId = 0;
1478 WINPR_ASSERT(drdynvc);
1479 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1480 return ERROR_INVALID_DATA;
1482 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1483 WLog_Print(drdynvc->log, WLOG_DEBUG,
1484 "process_close_request: Sp=%d cbChId=%d, ChannelId=%" PRIu32
"", Sp, cbChId,
1487 channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1490 WLog_Print(drdynvc->log, WLOG_ERROR,
"dvcman_close_request channel %" PRIu32
" not present",
1492 return CHANNEL_RC_OK;
1495 dvcman_channel_close(channel, TRUE, FALSE);
1496 dvcman_channel_unref(channel);
1497 return CHANNEL_RC_OK;
1505static UINT drdynvc_order_recv(drdynvcPlugin* drdynvc,
wStream* s, UINT32 ThreadingFlags)
1507 WINPR_ASSERT(drdynvc);
1508 if (!Stream_CheckAndLogRequiredLength(TAG, s, 1))
1509 return ERROR_INVALID_DATA;
1511 UINT8 value = Stream_Get_UINT8(s);
1512 const UINT8 Cmd = (value & 0xf0) >> 4;
1513 const UINT8 Sp = (value & 0x0c) >> 2;
1514 const UINT8 cbChId = (value & 0x03) >> 0;
1515 WLog_Print(drdynvc->log, WLOG_TRACE,
"order_recv: Cmd=%s, Sp=%" PRIu8
" cbChId=%" PRIu8,
1516 drdynvc_get_packet_type(Cmd), Sp, cbChId);
1520 case CAPABILITY_REQUEST_PDU:
1521 return drdynvc_process_capability_request(drdynvc, Sp, cbChId, s);
1523 case CREATE_REQUEST_PDU:
1524 return drdynvc_process_create_request(drdynvc, Sp, cbChId, s);
1526 case DATA_FIRST_PDU:
1527 case DATA_FIRST_COMPRESSED_PDU:
1528 return drdynvc_process_data_first(drdynvc, Sp, cbChId, s,
1529 (Cmd == DATA_FIRST_COMPRESSED_PDU), ThreadingFlags);
1532 case DATA_COMPRESSED_PDU:
1533 return drdynvc_process_data(drdynvc, Sp, cbChId, s, (Cmd == DATA_COMPRESSED_PDU),
1536 case CLOSE_REQUEST_PDU:
1537 return drdynvc_process_close_request(drdynvc, Sp, cbChId, s);
1539 case SOFT_SYNC_RESPONSE_PDU:
1540 WLog_Print(drdynvc->log, WLOG_ERROR,
1541 "not expecting a SOFT_SYNC_RESPONSE_PDU as a client");
1542 return ERROR_INTERNAL_ERROR;
1545 WLog_Print(drdynvc->log, WLOG_ERROR,
"unknown drdynvc cmd 0x%x", Cmd);
1546 return ERROR_INTERNAL_ERROR;
1555static UINT drdynvc_virtual_channel_event_data_received(drdynvcPlugin* drdynvc,
void* pData,
1556 UINT32 dataLength, UINT32 totalLength,
1561 WINPR_ASSERT(drdynvc);
1562 if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
1564 return CHANNEL_RC_OK;
1567 if (dataFlags & CHANNEL_FLAG_FIRST)
1570 if (drdynvc->data_in)
1571 Stream_Release(drdynvc->data_in);
1573 drdynvc->data_in = StreamPool_Take(mgr->pool, totalLength);
1576 if (!(data_in = drdynvc->data_in))
1578 WLog_Print(drdynvc->log, WLOG_ERROR,
"StreamPool_Take failed!");
1579 return CHANNEL_RC_NO_MEMORY;
1582 if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
1584 WLog_Print(drdynvc->log, WLOG_ERROR,
"Stream_EnsureRemainingCapacity failed!");
1585 Stream_Release(drdynvc->data_in);
1586 drdynvc->data_in =
nullptr;
1587 return ERROR_INTERNAL_ERROR;
1590 Stream_Write(data_in, pData, dataLength);
1592 if (dataFlags & CHANNEL_FLAG_LAST)
1594 const size_t cap = Stream_Capacity(data_in);
1595 const size_t pos = Stream_GetPosition(data_in);
1598 WLog_Print(drdynvc->log, WLOG_ERROR,
"drdynvc_plugin_process_received: read error");
1599 return ERROR_INVALID_DATA;
1602 drdynvc->data_in =
nullptr;
1603 Stream_SealLength(data_in);
1604 Stream_ResetPosition(data_in);
1608 if (!MessageQueue_Post(drdynvc->queue,
nullptr, 0, (
void*)data_in,
nullptr))
1610 WLog_Print(drdynvc->log, WLOG_ERROR,
"MessageQueue_Post failed!");
1611 return ERROR_INTERNAL_ERROR;
1616 UINT error = drdynvc_order_recv(drdynvc, data_in, TRUE);
1617 Stream_Release(data_in);
1621 WLog_Print(drdynvc->log, WLOG_WARN,
1622 "drdynvc_order_recv failed with error %" PRIu32
"!", error);
1628 return CHANNEL_RC_OK;
1631static void VCAPITYPE drdynvc_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle,
1632 UINT event, LPVOID pData,
1633 UINT32 dataLength, UINT32 totalLength,
1636 UINT error = CHANNEL_RC_OK;
1637 drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam;
1639 WINPR_ASSERT(drdynvc);
1642 case CHANNEL_EVENT_DATA_RECEIVED:
1643 if (!drdynvc || (drdynvc->OpenHandle != openHandle))
1645 WLog_ERR(TAG,
"drdynvc_virtual_channel_open_event: error no match");
1648 if ((error = drdynvc_virtual_channel_event_data_received(drdynvc, pData, dataLength,
1649 totalLength, dataFlags)))
1650 WLog_Print(drdynvc->log, WLOG_ERROR,
1651 "drdynvc_virtual_channel_event_data_received failed with error %" PRIu32
1657 case CHANNEL_EVENT_WRITE_CANCELLED:
1658 case CHANNEL_EVENT_WRITE_COMPLETE:
1665 case CHANNEL_EVENT_USER:
1671 if (error && drdynvc && drdynvc->rdpcontext)
1672 setChannelError(drdynvc->rdpcontext, error,
1673 "drdynvc_virtual_channel_open_event reported an error");
1676static DWORD WINAPI drdynvc_virtual_channel_client_thread(LPVOID arg)
1680 wMessage message = WINPR_C_ARRAY_INIT;
1681 UINT error = CHANNEL_RC_OK;
1682 drdynvcPlugin* drdynvc = (drdynvcPlugin*)arg;
1686 ExitThread((DWORD)CHANNEL_RC_BAD_CHANNEL_HANDLE);
1687 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1692 if (!MessageQueue_Wait(drdynvc->queue))
1694 WLog_Print(drdynvc->log, WLOG_ERROR,
"MessageQueue_Wait failed!");
1695 error = ERROR_INTERNAL_ERROR;
1699 if (!MessageQueue_Peek(drdynvc->queue, &message, TRUE))
1701 WLog_Print(drdynvc->log, WLOG_ERROR,
"MessageQueue_Peek failed!");
1702 error = ERROR_INTERNAL_ERROR;
1706 if (message.id == WMQ_QUIT)
1709 if (message.id == 0)
1711 UINT32 ThreadingFlags = TRUE;
1712 data = (
wStream*)message.wParam;
1714 if ((error = drdynvc_order_recv(drdynvc, data, ThreadingFlags)))
1716 WLog_Print(drdynvc->log, WLOG_WARN,
1717 "drdynvc_order_recv failed with error %" PRIu32
"!", error);
1720 Stream_Release(data);
1730 HashTable_Clear(drdynvcMgr->channelsById);
1733 if (error && drdynvc->rdpcontext)
1734 setChannelError(drdynvc->rdpcontext, error,
1735 "drdynvc_virtual_channel_client_thread reported an error");
1737 ExitThread((DWORD)error);
1741static void drdynvc_queue_object_free(
void* obj)
1744 wMessage* msg = (wMessage*)obj;
1746 if (!msg || (msg->id != 0))
1755static UINT drdynvc_virtual_channel_event_initialized(drdynvcPlugin* drdynvc, LPVOID pData,
1759 WINPR_UNUSED(pData);
1760 WINPR_UNUSED(dataLength);
1765 drdynvc->queue = MessageQueue_New(
nullptr);
1767 if (!drdynvc->queue)
1769 WLog_Print(drdynvc->log, WLOG_ERROR,
"MessageQueue_New failed!");
1773 obj = MessageQueue_Object(drdynvc->queue);
1775 drdynvc->channel_mgr = dvcman_new(drdynvc);
1777 if (!drdynvc->channel_mgr)
1779 WLog_Print(drdynvc->log, WLOG_ERROR,
"dvcman_new failed!");
1783 return CHANNEL_RC_OK;
1785 return ERROR_INTERNAL_ERROR;
1793static UINT drdynvc_virtual_channel_event_connected(drdynvcPlugin* drdynvc, LPVOID pData,
1798 rdpSettings* settings =
nullptr;
1800 WINPR_ASSERT(drdynvc);
1801 WINPR_UNUSED(pData);
1802 WINPR_UNUSED(dataLength);
1805 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1807 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelOpenEx);
1808 status = drdynvc->channelEntryPoints.pVirtualChannelOpenEx(
1809 drdynvc->InitHandle, &drdynvc->OpenHandle, drdynvc->channelDef.name,
1810 drdynvc_virtual_channel_open_event_ex);
1812 if (status != CHANNEL_RC_OK)
1814 WLog_Print(drdynvc->log, WLOG_ERROR,
"pVirtualChannelOpen failed with %s [%08" PRIX32
"]",
1815 WTSErrorToString(status), status);
1819 WINPR_ASSERT(drdynvc->rdpcontext);
1820 settings = drdynvc->rdpcontext->settings;
1821 WINPR_ASSERT(settings);
1823 for (UINT32 index = 0;
1827 freerdp_settings_get_pointer_array(settings, FreeRDP_DynamicChannelArray, index);
1828 error = dvcman_load_addin(drdynvc, drdynvc->channel_mgr, args, drdynvc->rdpcontext);
1830 if (CHANNEL_RC_OK != error)
1834 if ((error = dvcman_init(drdynvc, drdynvc->channel_mgr)))
1836 WLog_Print(drdynvc->log, WLOG_ERROR,
"dvcman_init failed with error %" PRIu32
"!", error);
1840 drdynvc->state = DRDYNVC_STATE_CAPABILITIES;
1844 if (!(drdynvc->thread = CreateThread(
nullptr, 0, drdynvc_virtual_channel_client_thread,
1845 (
void*)drdynvc, 0,
nullptr)))
1847 error = ERROR_INTERNAL_ERROR;
1848 WLog_Print(drdynvc->log, WLOG_ERROR,
"CreateThread failed!");
1852 if (!SetThreadPriority(drdynvc->thread, THREAD_PRIORITY_HIGHEST))
1853 WLog_Print(drdynvc->log, WLOG_WARN,
"SetThreadPriority failed, ignoring.");
1865static UINT drdynvc_virtual_channel_event_disconnected(drdynvcPlugin* drdynvc)
1870 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1872 if (drdynvc->OpenHandle == 0)
1873 return CHANNEL_RC_OK;
1877 if (!MessageQueue_PostQuit(drdynvc->queue, 0))
1879 status = GetLastError();
1880 WLog_Print(drdynvc->log, WLOG_ERROR,
1881 "MessageQueue_PostQuit failed with error %" PRIu32
"", status);
1886 if (drdynvc->thread)
1888 if (WaitForSingleObject(drdynvc->thread, INFINITE) != WAIT_OBJECT_0)
1890 status = GetLastError();
1891 WLog_Print(drdynvc->log, WLOG_ERROR,
1892 "WaitForSingleObject failed with error %" PRIu32
"", status);
1896 (void)CloseHandle(drdynvc->thread);
1897 drdynvc->thread =
nullptr;
1907 HashTable_Clear(drdynvcMgr->channelsById);
1911 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelCloseEx);
1912 status = drdynvc->channelEntryPoints.pVirtualChannelCloseEx(drdynvc->InitHandle,
1913 drdynvc->OpenHandle);
1915 if (status != CHANNEL_RC_OK)
1917 WLog_Print(drdynvc->log, WLOG_ERROR,
"pVirtualChannelClose failed with %s [%08" PRIX32
"]",
1918 WTSErrorToString(status), status);
1921 dvcman_clear(drdynvc, drdynvc->channel_mgr);
1923 MessageQueue_Clear(drdynvc->queue);
1924 drdynvc->OpenHandle = 0;
1926 if (drdynvc->data_in)
1928 Stream_Release(drdynvc->data_in);
1929 drdynvc->data_in =
nullptr;
1940static UINT drdynvc_virtual_channel_event_terminated(drdynvcPlugin* drdynvc)
1943 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1945 MessageQueue_Free(drdynvc->queue);
1946 drdynvc->queue =
nullptr;
1948 if (drdynvc->channel_mgr)
1950 dvcman_free(drdynvc, drdynvc->channel_mgr);
1951 drdynvc->channel_mgr =
nullptr;
1953 drdynvc->InitHandle =
nullptr;
1954 free(drdynvc->context);
1956 return CHANNEL_RC_OK;
1959static UINT drdynvc_virtual_channel_event_attached(drdynvcPlugin* drdynvc)
1961 UINT error = CHANNEL_RC_OK;
1962 DVCMAN* dvcman =
nullptr;
1965 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1967 dvcman = (
DVCMAN*)drdynvc->channel_mgr;
1970 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1972 ArrayList_Lock(dvcman->plugins);
1973 for (
size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
1975 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
1977 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Attached, pPlugin);
1978 if (error != CHANNEL_RC_OK)
1980 WLog_Print(drdynvc->log, WLOG_ERROR,
"Attach failed with error %" PRIu32
"!", error);
1986 ArrayList_Unlock(dvcman->plugins);
1990static UINT drdynvc_virtual_channel_event_detached(drdynvcPlugin* drdynvc)
1992 UINT error = CHANNEL_RC_OK;
1993 DVCMAN* dvcman =
nullptr;
1996 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1998 dvcman = (
DVCMAN*)drdynvc->channel_mgr;
2001 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
2003 ArrayList_Lock(dvcman->plugins);
2004 for (
size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
2006 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
2008 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Detached, pPlugin);
2009 if (error != CHANNEL_RC_OK)
2011 WLog_Print(drdynvc->log, WLOG_ERROR,
"Detach failed with error %" PRIu32
"!", error);
2017 ArrayList_Unlock(dvcman->plugins);
2022static VOID VCAPITYPE drdynvc_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle,
2023 UINT event, LPVOID pData,
2026 UINT error = CHANNEL_RC_OK;
2027 drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam;
2029 if (!drdynvc || (drdynvc->InitHandle != pInitHandle))
2031 WLog_ERR(TAG,
"drdynvc_virtual_channel_init_event: error no match");
2037 case CHANNEL_EVENT_INITIALIZED:
2038 error = drdynvc_virtual_channel_event_initialized(drdynvc, pData, dataLength);
2040 case CHANNEL_EVENT_CONNECTED:
2041 if ((error = drdynvc_virtual_channel_event_connected(drdynvc, pData, dataLength)))
2042 WLog_Print(drdynvc->log, WLOG_ERROR,
2043 "drdynvc_virtual_channel_event_connected failed with error %" PRIu32
"",
2048 case CHANNEL_EVENT_DISCONNECTED:
2049 if ((error = drdynvc_virtual_channel_event_disconnected(drdynvc)))
2050 WLog_Print(drdynvc->log, WLOG_ERROR,
2051 "drdynvc_virtual_channel_event_disconnected failed with error %" PRIu32
2057 case CHANNEL_EVENT_TERMINATED:
2058 if ((error = drdynvc_virtual_channel_event_terminated(drdynvc)))
2059 WLog_Print(drdynvc->log, WLOG_ERROR,
2060 "drdynvc_virtual_channel_event_terminated failed with error %" PRIu32
"",
2065 case CHANNEL_EVENT_ATTACHED:
2066 if ((error = drdynvc_virtual_channel_event_attached(drdynvc)))
2067 WLog_Print(drdynvc->log, WLOG_ERROR,
2068 "drdynvc_virtual_channel_event_attached failed with error %" PRIu32
"",
2073 case CHANNEL_EVENT_DETACHED:
2074 if ((error = drdynvc_virtual_channel_event_detached(drdynvc)))
2075 WLog_Print(drdynvc->log, WLOG_ERROR,
2076 "drdynvc_virtual_channel_event_detached failed with error %" PRIu32
"",
2085 if (error && drdynvc->rdpcontext)
2086 setChannelError(drdynvc->rdpcontext, error,
2087 "drdynvc_virtual_channel_init_event_ex reported an error");
2094static int drdynvc_get_version(DrdynvcClientContext* context)
2096 WINPR_ASSERT(context);
2097 drdynvcPlugin* drdynvc = (drdynvcPlugin*)context->handle;
2098 WINPR_ASSERT(drdynvc);
2099 return drdynvc->version;
2103#define VirtualChannelEntryEx drdynvc_VirtualChannelEntryEx
2105FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS_EX pEntryPoints,
2109 drdynvcPlugin* drdynvc =
nullptr;
2110 DrdynvcClientContext* context =
nullptr;
2112 drdynvc = (drdynvcPlugin*)calloc(1,
sizeof(drdynvcPlugin));
2114 WINPR_ASSERT(pEntryPoints);
2117 WLog_ERR(TAG,
"calloc failed!");
2121 drdynvc->channelDef.options =
2122 CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP | CHANNEL_OPTION_COMPRESS_RDP;
2123 (void)sprintf_s(drdynvc->channelDef.name, ARRAYSIZE(drdynvc->channelDef.name),
2124 DRDYNVC_SVC_CHANNEL_NAME);
2125 drdynvc->state = DRDYNVC_STATE_INITIAL;
2129 (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER))
2131 context = (DrdynvcClientContext*)calloc(1,
sizeof(DrdynvcClientContext));
2135 WLog_Print(drdynvc->log, WLOG_ERROR,
"calloc failed!");
2140 context->handle = (
void*)drdynvc;
2141 context->custom =
nullptr;
2142 drdynvc->context = context;
2143 context->GetVersion = drdynvc_get_version;
2144 drdynvc->rdpcontext = pEntryPointsEx->context;
2146 FreeRDP_TransportDumpReplay) &&
2148 FreeRDP_SynchronousDynamicChannels))
2149 drdynvc->async = TRUE;
2152 drdynvc->log = WLog_Get(TAG);
2153 WLog_Print(drdynvc->log, WLOG_DEBUG,
"VirtualChannelEntryEx");
2154 CopyMemory(&(drdynvc->channelEntryPoints), pEntryPoints,
2156 drdynvc->InitHandle = pInitHandle;
2158 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelInitEx);
2159 rc = drdynvc->channelEntryPoints.pVirtualChannelInitEx(
2160 drdynvc, context, pInitHandle, &drdynvc->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
2161 drdynvc_virtual_channel_init_event_ex);
2163 if (CHANNEL_RC_OK != rc)
2165 WLog_Print(drdynvc->log, WLOG_ERROR,
"pVirtualChannelInit failed with %s [%08" PRIX32
"]",
2166 WTSErrorToString(rc), rc);
2167 free(drdynvc->context);
2172 drdynvc->channelEntryPoints.pInterface = context;
WINPR_ATTR_NODISCARD FREERDP_API UINT32 freerdp_settings_get_uint32(const rdpSettings *settings, FreeRDP_Settings_Keys_UInt32 id)
Returns a UINT32 settings value.
WINPR_ATTR_NODISCARD FREERDP_API BOOL freerdp_settings_get_bool(const rdpSettings *settings, FreeRDP_Settings_Keys_Bool id)
Returns a boolean settings value.
This struct contains function pointer to initialize/free objects.
OBJECT_FREE_FN fnObjectFree
WINPR_ATTR_NODISCARD OBJECT_EQUALS_FN fnObjectEquals
WINPR_ATTR_NODISCARD OBJECT_NEW_FN fnObjectNew