FreeRDP
Loading...
Searching...
No Matches
client/drdynvc_main.c
1
22#include <freerdp/config.h>
23
24#include <winpr/crt.h>
25#include <winpr/cast.h>
26#include <winpr/stream.h>
27#include <winpr/interlocked.h>
28
29#include <freerdp/freerdp.h>
30#include <freerdp/channels/drdynvc.h>
31#include <freerdp/utils/drdynvc.h>
32#include <freerdp/codec/zgfx.h>
33
34#include "drdynvc_main.h"
35
36#define TAG CHANNELS_TAG("drdynvc.client")
37
38static const char* channel_state2str(DVC_CHANNEL_STATE state)
39{
40 switch (state)
41 {
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";
48 default:
49 return "DVC_CHANNEL_UNKNOWN";
50 }
51}
52
53static void dvcman_channel_free(DVCMAN_CHANNEL* channel);
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);
59
60static void dvcman_wtslistener_free(DVCMAN_LISTENER* listener)
61{
62 if (listener)
63 free(listener->channel_name);
64 free(listener);
65}
66
72static UINT dvcman_get_configuration(IWTSListener* pListener, void** ppPropertyBag)
73{
74 WINPR_ASSERT(ppPropertyBag);
75 WINPR_UNUSED(pListener);
76 *ppPropertyBag = nullptr;
77 return ERROR_INTERNAL_ERROR;
78}
79
85static UINT dvcman_create_listener(IWTSVirtualChannelManager* pChannelMgr,
86 const char* pszChannelName, ULONG ulFlags,
87 IWTSListenerCallback* pListenerCallback,
88 IWTSListener** ppListener)
89{
90 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
91 DVCMAN_LISTENER* listener = nullptr;
92
93 WINPR_ASSERT(dvcman);
94 WLog_DBG(TAG, "create_listener: %" PRIuz ".%s.", HashTable_Count(dvcman->listeners) + 1,
95 pszChannelName);
96 listener = (DVCMAN_LISTENER*)calloc(1, sizeof(DVCMAN_LISTENER));
97
98 if (!listener)
99 {
100 WLog_ERR(TAG, "calloc failed!");
101 return CHANNEL_RC_NO_MEMORY;
102 }
103
104 listener->iface.GetConfiguration = dvcman_get_configuration;
105 listener->iface.pInterface = nullptr;
106 listener->dvcman = dvcman;
107 listener->channel_name = _strdup(pszChannelName);
108
109 if (!listener->channel_name)
110 {
111 WLog_ERR(TAG, "_strdup failed!");
112 dvcman_wtslistener_free(listener);
113 return CHANNEL_RC_NO_MEMORY;
114 }
115
116 listener->flags = ulFlags;
117 listener->listener_callback = pListenerCallback;
118
119 if (ppListener)
120 *ppListener = (IWTSListener*)listener;
121
122 if (!HashTable_Insert(dvcman->listeners, listener->channel_name, listener))
123 {
124 dvcman_wtslistener_free(listener);
125 return ERROR_INTERNAL_ERROR;
126 }
127
128 // NOLINTNEXTLINE(clang-analyzer-unix.Malloc): HashTable_Insert takes ownership of listener
129 return CHANNEL_RC_OK;
130}
131
132static UINT dvcman_destroy_listener(IWTSVirtualChannelManager* pChannelMgr, IWTSListener* pListener)
133{
134 DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)pListener;
135
136 WINPR_UNUSED(pChannelMgr);
137
138 if (listener)
139 {
140 DVCMAN* dvcman = listener->dvcman;
141 if (dvcman)
142 HashTable_Remove(dvcman->listeners, listener->channel_name);
143 }
144
145 return CHANNEL_RC_OK;
146}
147
153static UINT dvcman_register_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name,
154 IWTSPlugin* pPlugin)
155{
156 WINPR_ASSERT(pEntryPoints);
157 DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman;
158
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;
164
165 WLog_DBG(TAG, "register_plugin: num_plugins %" PRIuz, ArrayList_Count(dvcman->plugins));
166 return CHANNEL_RC_OK;
167}
168
169static IWTSPlugin* dvcman_get_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name)
170{
171 IWTSPlugin* plugin = nullptr;
172 size_t nc = 0;
173 size_t pc = 0;
174 WINPR_ASSERT(pEntryPoints);
175 DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman;
176 if (!dvcman || !pEntryPoints || !name)
177 return nullptr;
178
179 nc = ArrayList_Count(dvcman->plugin_names);
180 pc = ArrayList_Count(dvcman->plugins);
181 if (nc != pc)
182 return nullptr;
183
184 ArrayList_Lock(dvcman->plugin_names);
185 ArrayList_Lock(dvcman->plugins);
186 for (size_t i = 0; i < pc; i++)
187 {
188 const char* cur = ArrayList_GetItem(dvcman->plugin_names, i);
189 if (strcmp(cur, name) == 0)
190 {
191 plugin = ArrayList_GetItem(dvcman->plugins, i);
192 break;
193 }
194 }
195 ArrayList_Unlock(dvcman->plugin_names);
196 ArrayList_Unlock(dvcman->plugins);
197 return plugin;
198}
199
200static const ADDIN_ARGV* dvcman_get_plugin_data(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
201{
202 WINPR_ASSERT(pEntryPoints);
203 return ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->args;
204}
205
206static rdpContext* dvcman_get_rdp_context(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
207{
208 DVCMAN_ENTRY_POINTS* entry = (DVCMAN_ENTRY_POINTS*)pEntryPoints;
209 WINPR_ASSERT(entry);
210 return entry->context;
211}
212
213static rdpSettings* dvcman_get_rdp_settings(IDRDYNVC_ENTRY_POINTS* pEntryPoints)
214{
215 rdpContext* context = dvcman_get_rdp_context(pEntryPoints);
216 WINPR_ASSERT(context);
217
218 return context->settings;
219}
220
221static UINT32 dvcman_get_channel_id(IWTSVirtualChannel* channel)
222{
223 DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel;
224 WINPR_ASSERT(dvc);
225 return dvc->channel_id;
226}
227
228static const char* dvcman_get_channel_name(IWTSVirtualChannel* channel)
229{
230 DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel;
231 WINPR_ASSERT(dvc);
232 return dvc->channel_name;
233}
234
235static DVCMAN_CHANNEL* dvcman_get_channel_by_id(IWTSVirtualChannelManager* pChannelMgr,
236 UINT32 ChannelId, BOOL doRef)
237{
238 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
239 DVCMAN_CHANNEL* dvcChannel = nullptr;
240
241 WINPR_ASSERT(dvcman);
242 HashTable_Lock(dvcman->channelsById);
243 dvcChannel = HashTable_GetItemValue(dvcman->channelsById, &ChannelId);
244 if (dvcChannel)
245 {
246 if (doRef)
247 InterlockedIncrement(&dvcChannel->refCounter);
248 }
249
250 HashTable_Unlock(dvcman->channelsById);
251 return dvcChannel;
252}
253
254static IWTSVirtualChannel* dvcman_find_channel_by_id(IWTSVirtualChannelManager* pChannelMgr,
255 UINT32 ChannelId)
256{
257 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE);
258 if (!channel)
259 return nullptr;
260
261 return &channel->iface;
262}
263
264static void dvcman_plugin_terminate(void* plugin)
265{
266 IWTSPlugin* pPlugin = plugin;
267
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);
272}
273
274static void wts_listener_free(void* arg)
275{
276 DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)arg;
277 dvcman_wtslistener_free(listener);
278}
279
280static BOOL channelIdMatch(const void* k1, const void* k2)
281{
282 WINPR_ASSERT(k1);
283 WINPR_ASSERT(k2);
284 return *((const UINT32*)k1) == *((const UINT32*)k2);
285}
286
287static UINT32 channelIdHash(const void* id)
288{
289 WINPR_ASSERT(id);
290 return *((const UINT32*)id);
291}
292
293static void channelByIdCleanerFn(void* value)
294{
295 DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)value;
296 if (channel)
297 {
298 dvcman_channel_close(channel, FALSE, TRUE);
299 dvcman_channel_free(channel);
300 }
301}
302
303static IWTSVirtualChannelManager* dvcman_new(drdynvcPlugin* plugin)
304{
305 wObject* obj = nullptr;
306 DVCMAN* dvcman = (DVCMAN*)calloc(1, sizeof(DVCMAN));
307
308 if (!dvcman)
309 return nullptr;
310
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);
318
319 if (!dvcman->channelsById)
320 goto fail;
321
322 if (!HashTable_SetHashFunction(dvcman->channelsById, channelIdHash))
323 goto fail;
324
325 obj = HashTable_KeyObject(dvcman->channelsById);
326 WINPR_ASSERT(obj);
327 obj->fnObjectEquals = channelIdMatch;
328
329 obj = HashTable_ValueObject(dvcman->channelsById);
330 WINPR_ASSERT(obj);
331 obj->fnObjectFree = channelByIdCleanerFn;
332
333 dvcman->pool = StreamPool_New(TRUE, 10);
334 if (!dvcman->pool)
335 goto fail;
336
337 dvcman->listeners = HashTable_New(TRUE);
338 if (!dvcman->listeners)
339 goto fail;
340
341 if (!HashTable_SetHashFunction(dvcman->listeners, HashTable_StringHash))
342 goto fail;
343
344 obj = HashTable_KeyObject(dvcman->listeners);
345 obj->fnObjectEquals = HashTable_StringCompare;
346
347 obj = HashTable_ValueObject(dvcman->listeners);
348 obj->fnObjectFree = wts_listener_free;
349
350 dvcman->plugin_names = ArrayList_New(TRUE);
351 if (!dvcman->plugin_names)
352 goto fail;
353 obj = ArrayList_Object(dvcman->plugin_names);
354 obj->fnObjectNew = winpr_ObjectStringClone;
355 obj->fnObjectFree = winpr_ObjectStringFree;
356
357 dvcman->plugins = ArrayList_New(TRUE);
358 if (!dvcman->plugins)
359 goto fail;
360 obj = ArrayList_Object(dvcman->plugins);
361 obj->fnObjectFree = dvcman_plugin_terminate;
362 return &dvcman->iface;
363fail:
364 dvcman_free(plugin, &dvcman->iface);
365 return nullptr;
366}
367
373static UINT dvcman_load_addin(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr,
374 const ADDIN_ARGV* args, rdpContext* context)
375{
376 WINPR_ASSERT(drdynvc);
377 WINPR_ASSERT(pChannelMgr);
378 WINPR_ASSERT(args);
379 WINPR_ASSERT(context);
380
381 WLog_Print(drdynvc->log, WLOG_INFO, "Loading Dynamic Virtual Channel %s", args->argv[0]);
382
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);
386
387 if (pDVCPluginEntry)
388 {
389 DVCMAN_ENTRY_POINTS entryPoints = WINPR_C_ARRAY_INIT;
390
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);
400 }
401
402 return ERROR_INVALID_FUNCTION;
403}
404
405static void dvcman_channel_free(DVCMAN_CHANNEL* channel)
406{
407 if (!channel)
408 return;
409
410 if (channel->dvcman)
411 {
412 drdynvcPlugin* plugin = channel->dvcman->drdynvc;
413 if (plugin)
414 {
415 rdpContext* context = plugin->rdpcontext;
416 if (context)
417 {
418 ChannelTerminatedEventArgs e = WINPR_C_ARRAY_INIT;
419 EventArgsInit(&e, "freerdp");
420 e.name = channel->channel_name;
421 e.pInterface = channel->pInterface;
422
423 const int rc = PubSub_OnChannelTerminated(context->pubSub, context, &e);
424 if (rc < 0)
425 WLog_WARN(TAG, "PubSub_OnChannelTerminated(%s) failed", channel->channel_name);
426 }
427 }
428 }
429
430 if (channel->dvc_data)
431 Stream_Release(channel->dvc_data);
432
433 zgfx_context_free(channel->decompressor);
434 DeleteCriticalSection(&(channel->lock));
435 free(channel->channel_name);
436 free(channel);
437}
438
439static void dvcman_channel_unref(DVCMAN_CHANNEL* channel)
440{
441 WINPR_ASSERT(channel);
442 if (InterlockedDecrement(&channel->refCounter))
443 return;
444
445 DVCMAN* dvcman = channel->dvcman;
446 if (dvcman)
447 HashTable_Remove(dvcman->channelsById, &channel->channel_id);
448}
449
450static UINT dvcchannel_send_close(DVCMAN_CHANNEL* channel)
451{
452 WINPR_ASSERT(channel);
453 DVCMAN* dvcman = channel->dvcman;
454 drdynvcPlugin* drdynvc = dvcman->drdynvc;
455 wStream* s = StreamPool_Take(dvcman->pool, 5);
456
457 if (!s)
458 {
459 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
460 return CHANNEL_RC_NO_MEMORY;
461 }
462
463 Stream_Write_UINT8(s, (CLOSE_REQUEST_PDU << 4) | 0x02);
464 Stream_Write_UINT32(s, channel->channel_id);
465 return drdynvc_send(drdynvc, s);
466}
467
468static void check_open_close_receive(DVCMAN_CHANNEL* channel)
469{
470 WINPR_ASSERT(channel);
471
472 IWTSVirtualChannelCallback* cb = channel->channel_callback;
473 const char* name = channel->channel_name;
474 const UINT32 id = channel->channel_id;
475
476 WINPR_ASSERT(cb);
477 if (!cb->OnOpen || !cb->OnClose || !cb->OnDataReceived)
478 WLog_VRB(TAG, "{%s:%" PRIu32 "} OnOpen=%p, OnClose=%p, OnDataReceived=%p", name, id,
479 WINPR_FUNC_PTR_CAST(cb->OnOpen, const void*),
480 WINPR_FUNC_PTR_CAST(cb->OnClose, const void*),
481 WINPR_FUNC_PTR_CAST(cb->OnDataReceived, const void*));
482}
483
484static UINT dvcman_call_on_receive(DVCMAN_CHANNEL* channel, wStream* data)
485{
486 WINPR_ASSERT(channel);
487 WINPR_ASSERT(data);
488
489 IWTSVirtualChannelCallback* cb = channel->channel_callback;
490 WINPR_ASSERT(cb);
491
492 check_open_close_receive(channel);
493 WINPR_ASSERT(cb->OnDataReceived);
494 return cb->OnDataReceived(cb, data);
495}
496
497static UINT dvcman_channel_close(DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn)
498{
499 UINT error = CHANNEL_RC_OK;
500 DrdynvcClientContext* context = nullptr;
501
502 WINPR_ASSERT(channel);
503 switch (channel->state)
504 {
505 case DVC_CHANNEL_INIT:
506 break;
507 case DVC_CHANNEL_RUNNING:
508 if (channel->dvcman)
509 {
510 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
511 WINPR_ASSERT(drdynvc);
512 context = drdynvc->context;
513 if (perRequest)
514 WLog_Print(drdynvc->log, WLOG_DEBUG, "sending close confirm for '%s'",
515 channel->channel_name);
516
517 error = dvcchannel_send_close(channel);
518 if (error != CHANNEL_RC_OK)
519 {
520 if (perRequest)
521 WLog_Print(drdynvc->log, WLOG_DEBUG,
522 "error when sending closeRequest for '%s'",
523 channel->channel_name);
524 else
525 WLog_Print(drdynvc->log, WLOG_DEBUG,
526 "error when sending close confirm for '%s'",
527 channel->channel_name);
528 }
529 WLog_Print(drdynvc->log, WLOG_DEBUG, "listener %s destroyed channel %" PRIu32 "",
530 channel->channel_name, channel->channel_id);
531 }
532
533 channel->state = DVC_CHANNEL_CLOSED;
534
535 {
536 check_open_close_receive(channel);
537
538 IWTSVirtualChannelCallback* cb = channel->channel_callback;
539 channel->channel_callback = nullptr;
540 if (cb)
541 error = IFCALLRESULT(CHANNEL_RC_OK, cb->OnClose, cb);
542 }
543
544 if (channel->dvcman && channel->dvcman->drdynvc)
545 {
546 if (context)
547 {
548 IFCALLRET(context->OnChannelDisconnected, error, context, channel->channel_name,
549 channel->pInterface);
550 }
551 }
552
553 if (!fromHashTableFn)
554 dvcman_channel_unref(channel);
555 break;
556 case DVC_CHANNEL_CLOSED:
557 break;
558 default:
559 break;
560 }
561
562 return error;
563}
564
565static DVCMAN_CHANNEL* dvcman_channel_new(drdynvcPlugin* drdynvc,
566 IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId,
567 const char* ChannelName)
568{
569 WINPR_ASSERT(drdynvc);
570 WINPR_ASSERT(pChannelMgr);
571 DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)calloc(1, sizeof(DVCMAN_CHANNEL));
572
573 if (!channel)
574 return nullptr;
575
576 channel->dvcman = (DVCMAN*)pChannelMgr;
577 channel->channel_id = ChannelId;
578 channel->refCounter = 1;
579 channel->state = DVC_CHANNEL_INIT;
580 channel->channel_name = _strdup(ChannelName);
581 if (!channel->channel_name)
582 goto fail;
583
584 channel->decompressor = zgfx_context_new(FALSE);
585 if (!channel->decompressor)
586 goto fail;
587
588 if (!InitializeCriticalSectionEx(&(channel->lock), 0, 0))
589 goto fail;
590
591 if (drdynvc)
592 {
593 rdpContext* context = drdynvc->rdpcontext;
594 if (context)
595 {
596 ChannelInitializedEventArgs e = WINPR_C_ARRAY_INIT;
597 EventArgsInit(&e, "freerdp");
598 e.name = channel->channel_name;
599 e.pInterface = channel->pInterface;
600
601 const int rc = PubSub_OnChannelInitialized(context->pubSub, context, &e);
602 if (rc < 0)
603 WLog_WARN(TAG, "PubSub_OnChannelInitialized(%s) failed", channel->channel_name);
604 }
605 }
606
607 return channel;
608fail:
609 dvcman_channel_free(channel);
610 return nullptr;
611}
612
613static void dvcman_clear(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
614{
615 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
616
617 WINPR_ASSERT(dvcman);
618 WINPR_UNUSED(drdynvc);
619
620 HashTable_Clear(dvcman->channelsById);
621 ArrayList_Clear(dvcman->plugins);
622 ArrayList_Clear(dvcman->plugin_names);
623 HashTable_Clear(dvcman->listeners);
624}
625static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
626{
627 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
628
629 WINPR_ASSERT(dvcman);
630 WINPR_UNUSED(drdynvc);
631
632 HashTable_Free(dvcman->channelsById);
633 ArrayList_Free(dvcman->plugins);
634 ArrayList_Free(dvcman->plugin_names);
635 HashTable_Free(dvcman->listeners);
636
637 StreamPool_Free(dvcman->pool);
638 free(dvcman);
639}
640
646static UINT dvcman_init(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr)
647{
648 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
649 UINT error = CHANNEL_RC_OK;
650
651 WINPR_ASSERT(dvcman);
652 ArrayList_Lock(dvcman->plugins);
653 for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
654 {
655 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
656
657 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Initialize, pPlugin, pChannelMgr);
658 if (error != CHANNEL_RC_OK)
659 {
660 WLog_Print(drdynvc->log, WLOG_ERROR, "Initialize failed with error %" PRIu32 "!",
661 error);
662 goto fail;
663 }
664 }
665
666fail:
667 ArrayList_Unlock(dvcman->plugins);
668 return error;
669}
670
676static UINT dvcman_write_channel(IWTSVirtualChannel* pChannel, ULONG cbSize, const BYTE* pBuffer,
677 void* pReserved)
678{
679 BOOL close = FALSE;
680 UINT status = 0;
681 DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel;
682
683 WINPR_UNUSED(pReserved);
684 if (!channel || !channel->dvcman)
685 return CHANNEL_RC_BAD_CHANNEL;
686
687 EnterCriticalSection(&(channel->lock));
688 status =
689 drdynvc_write_data(channel->dvcman->drdynvc, channel->channel_id, pBuffer, cbSize, &close);
690 LeaveCriticalSection(&(channel->lock));
691 /* Close delayed, it removes the channel struct */
692 if (close)
693 dvcman_channel_close(channel, FALSE, FALSE);
694
695 return status;
696}
697
703static UINT dvcman_close_channel_iface(IWTSVirtualChannel* pChannel)
704{
705 DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel;
706
707 if (!channel)
708 return CHANNEL_RC_BAD_CHANNEL;
709
710 WLog_DBG(TAG, "close_channel_iface: id=%" PRIu32 "", channel->channel_id);
711 return dvcman_channel_close(channel, FALSE, FALSE);
712}
713
719static DVCMAN_CHANNEL* dvcman_create_channel(drdynvcPlugin* drdynvc,
720 IWTSVirtualChannelManager* pChannelMgr,
721 UINT32 ChannelId, const char* ChannelName, UINT* res)
722{
723 BOOL bAccept = 0;
724 DVCMAN_CHANNEL* channel = nullptr;
725 DrdynvcClientContext* context = nullptr;
726 DVCMAN* dvcman = (DVCMAN*)pChannelMgr;
727 DVCMAN_LISTENER* listener = nullptr;
728 IWTSVirtualChannelCallback* pCallback = nullptr;
729
730 WINPR_ASSERT(dvcman);
731 WINPR_ASSERT(res);
732
733 HashTable_Lock(dvcman->listeners);
734 listener = (DVCMAN_LISTENER*)HashTable_GetItemValue(dvcman->listeners, ChannelName);
735 if (!listener)
736 {
737 *res = ERROR_NOT_FOUND;
738 goto out;
739 }
740
741 channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE);
742 if (channel)
743 {
744 switch (channel->state)
745 {
746 case DVC_CHANNEL_RUNNING:
747 WLog_Print(drdynvc->log, WLOG_ERROR,
748 "Protocol error: Duplicated ChannelId %" PRIu32 " (%s)!", ChannelId,
749 ChannelName);
750 *res = CHANNEL_RC_ALREADY_OPEN;
751 goto out;
752
753 case DVC_CHANNEL_CLOSED:
754 case DVC_CHANNEL_INIT:
755 default:
756 {
757 WLog_Print(drdynvc->log, WLOG_ERROR, "not expecting a createChannel from state %s",
758 channel_state2str(channel->state));
759 *res = CHANNEL_RC_INITIALIZATION_ERROR;
760 goto out;
761 }
762 }
763 }
764 else
765 {
766 if (!(channel = dvcman_channel_new(drdynvc, pChannelMgr, ChannelId, ChannelName)))
767 {
768 WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_channel_new failed!");
769 *res = CHANNEL_RC_NO_MEMORY;
770 goto out;
771 }
772 }
773
774 if (!HashTable_Insert(dvcman->channelsById, &channel->channel_id, channel))
775 {
776 WLog_Print(drdynvc->log, WLOG_ERROR, "unable to register channel in our channel list");
777 *res = ERROR_INTERNAL_ERROR;
778 dvcman_channel_free(channel);
779 channel = nullptr;
780 goto out;
781 }
782
783 channel->iface.Write = dvcman_write_channel;
784 channel->iface.Close = dvcman_close_channel_iface;
785 bAccept = TRUE;
786
787 *res = listener->listener_callback->OnNewChannelConnection(
788 listener->listener_callback, &channel->iface, nullptr, &bAccept, &pCallback);
789
790 if (*res != CHANNEL_RC_OK)
791 {
792 WLog_Print(drdynvc->log, WLOG_ERROR,
793 "OnNewChannelConnection failed with error %" PRIu32 "!", *res);
794 *res = ERROR_INTERNAL_ERROR;
795 dvcman_channel_unref(channel);
796 goto out;
797 }
798
799 if (!bAccept)
800 {
801 WLog_Print(drdynvc->log, WLOG_ERROR, "OnNewChannelConnection returned with bAccept FALSE!");
802 *res = ERROR_INTERNAL_ERROR;
803 dvcman_channel_unref(channel);
804 channel = nullptr;
805 goto out;
806 }
807
808 WLog_Print(drdynvc->log, WLOG_DEBUG, "listener %s created new channel %" PRIu32 "",
809 listener->channel_name, channel->channel_id);
810 channel->state = DVC_CHANNEL_RUNNING;
811 channel->channel_callback = pCallback;
812 channel->pInterface = listener->iface.pInterface;
813 context = dvcman->drdynvc->context;
814
815 IFCALLRET(context->OnChannelConnected, *res, context, ChannelName, listener->iface.pInterface);
816 if (*res != CHANNEL_RC_OK)
817 {
818 WLog_Print(drdynvc->log, WLOG_ERROR,
819 "context.OnChannelConnected failed with error %" PRIu32 "", *res);
820 }
821
822out:
823 HashTable_Unlock(dvcman->listeners);
824
825 return channel;
826}
827
833static UINT dvcman_open_channel(drdynvcPlugin* drdynvc, DVCMAN_CHANNEL* channel)
834{
835 UINT error = CHANNEL_RC_OK;
836
837 WINPR_ASSERT(drdynvc);
838 WINPR_ASSERT(channel);
839 if (channel->state == DVC_CHANNEL_RUNNING)
840 {
841 IWTSVirtualChannelCallback* pCallback = channel->channel_callback;
842
843 if (pCallback->OnOpen)
844 {
845 check_open_close_receive(channel);
846 error = pCallback->OnOpen(pCallback);
847 if (error)
848 {
849 WLog_Print(drdynvc->log, WLOG_ERROR, "OnOpen failed with error %" PRIu32 "!",
850 error);
851 goto out;
852 }
853 }
854
855 WLog_Print(drdynvc->log, WLOG_DEBUG, "open_channel: ChannelId %" PRIu32 "",
856 channel->channel_id);
857 }
858
859out:
860 return error;
861}
862
868static UINT dvcman_receive_channel_data_first(DVCMAN_CHANNEL* channel, UINT32 length)
869{
870 WINPR_ASSERT(channel);
871 WINPR_ASSERT(channel->dvcman);
872 if (channel->dvc_data)
873 Stream_Release(channel->dvc_data);
874
875 channel->dvc_data = StreamPool_Take(channel->dvcman->pool, length);
876
877 if (!channel->dvc_data)
878 {
879 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
880 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
881 return CHANNEL_RC_NO_MEMORY;
882 }
883
884 channel->dvc_data_length = length;
885 return CHANNEL_RC_OK;
886}
887
893static UINT dvcman_receive_channel_data(DVCMAN_CHANNEL* channel, wStream* data,
894 WINPR_ATTR_UNUSED UINT32 ThreadingFlags)
895{
896 UINT status = CHANNEL_RC_OK;
897 size_t dataSize = Stream_GetRemainingLength(data);
898
899 WINPR_ASSERT(channel);
900 WINPR_ASSERT(channel->dvcman);
901 if (channel->dvc_data)
902 {
903 drdynvcPlugin* drdynvc = channel->dvcman->drdynvc;
904
905 /* Fragmented data */
906 if (Stream_GetPosition(channel->dvc_data) + dataSize > channel->dvc_data_length)
907 {
908 WLog_Print(drdynvc->log, WLOG_ERROR, "data exceeding declared length!");
909 Stream_Release(channel->dvc_data);
910 channel->dvc_data = nullptr;
911 status = ERROR_INVALID_DATA;
912 goto out;
913 }
914
915 Stream_Copy(data, channel->dvc_data, dataSize);
916
917 if (Stream_GetPosition(channel->dvc_data) >= channel->dvc_data_length)
918 {
919 Stream_SealLength(channel->dvc_data);
920 Stream_ResetPosition(channel->dvc_data);
921
922 status = dvcman_call_on_receive(channel, channel->dvc_data);
923 Stream_Release(channel->dvc_data);
924 channel->dvc_data = nullptr;
925 }
926 }
927 else
928 status = dvcman_call_on_receive(channel, data);
929
930out:
931 return status;
932}
933
934static UINT8 drdynvc_write_variable_uint(wStream* s, UINT32 val)
935{
936 UINT8 cb = 0;
937
938 if (val <= 0xFF)
939 {
940 cb = 0;
941 Stream_Write_UINT8(s, (UINT8)val);
942 }
943 else if (val <= 0xFFFF)
944 {
945 cb = 1;
946 Stream_Write_UINT16(s, (UINT16)val);
947 }
948 else
949 {
950 cb = 2;
951 Stream_Write_UINT32(s, val);
952 }
953
954 return cb;
955}
956
962static UINT drdynvc_send(drdynvcPlugin* drdynvc, wStream* s)
963{
964 UINT status = 0;
965
966 if (!drdynvc)
967 status = CHANNEL_RC_BAD_CHANNEL_HANDLE;
968 else
969 {
970 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelWriteEx);
971 status = drdynvc->channelEntryPoints.pVirtualChannelWriteEx(
972 drdynvc->InitHandle, drdynvc->OpenHandle, Stream_Buffer(s),
973 (UINT32)Stream_GetPosition(s), s);
974 }
975
976 switch (status)
977 {
978 case CHANNEL_RC_OK:
979 return CHANNEL_RC_OK;
980
981 case CHANNEL_RC_NOT_CONNECTED:
982 Stream_Release(s);
983 return CHANNEL_RC_OK;
984
985 case CHANNEL_RC_BAD_CHANNEL_HANDLE:
986 Stream_Release(s);
987 WLog_ERR(TAG, "VirtualChannelWriteEx failed with CHANNEL_RC_BAD_CHANNEL_HANDLE");
988 return status;
989
990 default:
991 Stream_Release(s);
992 WLog_Print(drdynvc->log, WLOG_ERROR,
993 "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
994 WTSErrorToString(status), status);
995 return status;
996 }
997}
998
1004static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId, const BYTE* data,
1005 UINT32 dataSize, BOOL* close)
1006{
1007 size_t pos = 0;
1008 UINT8 cbChId = 0;
1009 UINT8 cbLen = 0;
1010 UINT status = CHANNEL_RC_BAD_INIT_HANDLE;
1011 DVCMAN* dvcman = nullptr;
1012
1013 if (!drdynvc)
1014 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1015
1016 dvcman = (DVCMAN*)drdynvc->channel_mgr;
1017 WINPR_ASSERT(dvcman);
1018
1019 WLog_Print(drdynvc->log, WLOG_TRACE, "write_data: ChannelId=%" PRIu32 " size=%" PRIu32 "",
1020 ChannelId, dataSize);
1021 wStream* data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH);
1022
1023 if (!data_out)
1024 {
1025 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
1026 return CHANNEL_RC_NO_MEMORY;
1027 }
1028
1029 if (!Stream_SetPosition(data_out, 1))
1030 {
1031 Stream_Release(data_out);
1032 return ERROR_INVALID_DATA;
1033 }
1034 cbChId = drdynvc_write_variable_uint(data_out, ChannelId);
1035 pos = Stream_GetPosition(data_out);
1036
1037 if (dataSize == 0)
1038 {
1039 /* TODO: shall treat that case with write(0) that do a close */
1040 *close = TRUE;
1041 Stream_Release(data_out);
1042 }
1043 else if (dataSize <= CHANNEL_CHUNK_LENGTH - pos)
1044 {
1045 Stream_ResetPosition(data_out);
1046 Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId);
1047 if (!Stream_SetPosition(data_out, pos))
1048 {
1049 Stream_Release(data_out);
1050 return ERROR_INVALID_DATA;
1051 }
1052 Stream_Write(data_out, data, dataSize);
1053 status = drdynvc_send(drdynvc, data_out);
1054 }
1055 else
1056 {
1057 /* Fragment the data */
1058 cbLen = drdynvc_write_variable_uint(data_out, dataSize);
1059 pos = Stream_GetPosition(data_out);
1060 Stream_ResetPosition(data_out);
1061
1062 const INT32 pdu = (DATA_FIRST_PDU << 4) | cbChId | (cbLen << 2);
1063 Stream_Write_UINT8(data_out, WINPR_ASSERTING_INT_CAST(UINT8, pdu));
1064 if (!Stream_SetPosition(data_out, pos))
1065 {
1066 Stream_Release(data_out);
1067 return ERROR_INVALID_DATA;
1068 }
1069
1070 {
1071 WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH);
1072 const uint32_t chunkLength =
1073 CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos);
1074 Stream_Write(data_out, data, chunkLength);
1075
1076 data += chunkLength;
1077 dataSize -= chunkLength;
1078 }
1079 status = drdynvc_send(drdynvc, data_out);
1080
1081 while (status == CHANNEL_RC_OK && dataSize > 0)
1082 {
1083 data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH);
1084
1085 if (!data_out)
1086 {
1087 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
1088 return CHANNEL_RC_NO_MEMORY;
1089 }
1090
1091 if (!Stream_SetPosition(data_out, 1))
1092 {
1093 Stream_Release(data_out);
1094 return ERROR_INVALID_DATA;
1095 }
1096
1097 cbChId = drdynvc_write_variable_uint(data_out, ChannelId);
1098 pos = Stream_GetPosition(data_out);
1099 Stream_ResetPosition(data_out);
1100 Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId);
1101 if (!Stream_SetPosition(data_out, pos))
1102 {
1103 Stream_Release(data_out);
1104 return ERROR_INVALID_DATA;
1105 }
1106
1107 uint32_t chunkLength = dataSize;
1108
1109 WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH);
1110 if (chunkLength > CHANNEL_CHUNK_LENGTH - pos)
1111 chunkLength = CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos);
1112
1113 Stream_Write(data_out, data, chunkLength);
1114 data += chunkLength;
1115 dataSize -= chunkLength;
1116 status = drdynvc_send(drdynvc, data_out);
1117 }
1118 }
1119
1120 if (status != CHANNEL_RC_OK)
1121 {
1122 WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
1123 WTSErrorToString(status), status);
1124 return status;
1125 }
1126
1127 return CHANNEL_RC_OK;
1128}
1129
1135static UINT drdynvc_send_capability_response(drdynvcPlugin* drdynvc)
1136{
1137 UINT status = 0;
1138 wStream* s = nullptr;
1139 DVCMAN* dvcman = nullptr;
1140
1141 if (!drdynvc)
1142 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1143
1144 dvcman = (DVCMAN*)drdynvc->channel_mgr;
1145 WINPR_ASSERT(dvcman);
1146
1147 WLog_Print(drdynvc->log, WLOG_TRACE, "capability_response");
1148 s = StreamPool_Take(dvcman->pool, 4);
1149
1150 if (!s)
1151 {
1152 WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_New failed!");
1153 return CHANNEL_RC_NO_MEMORY;
1154 }
1155
1156 Stream_Write_UINT16(s, 0x0050); /* Cmd+Sp+cbChId+Pad. Note: MSTSC sends 0x005c */
1157 Stream_Write_UINT16(s, drdynvc->version);
1158 status = drdynvc_send(drdynvc, s);
1159
1160 if (status != CHANNEL_RC_OK)
1161 {
1162 WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
1163 WTSErrorToString(status), status);
1164 }
1165
1166 return status;
1167}
1168
1174static UINT drdynvc_process_capability_request(drdynvcPlugin* drdynvc, int Sp, int cbChId,
1175 wStream* s)
1176{
1177 UINT status = 0;
1178
1179 if (!drdynvc)
1180 return CHANNEL_RC_BAD_INIT_HANDLE;
1181
1182 if (!Stream_CheckAndLogRequiredLength(TAG, s, 3))
1183 return ERROR_INVALID_DATA;
1184
1185 WLog_Print(drdynvc->log, WLOG_TRACE, "capability_request Sp=%d cbChId=%d", Sp, cbChId);
1186 Stream_Seek(s, 1); /* pad */
1187 Stream_Read_UINT16(s, drdynvc->version);
1188
1189 /* RDP8 servers offer version 3, though Microsoft forgot to document it
1190 * in their early documents. It behaves the same as version 2.
1191 */
1192 if ((drdynvc->version == 2) || (drdynvc->version == 3))
1193 {
1194 if (!Stream_CheckAndLogRequiredLength(TAG, s, 8))
1195 return ERROR_INVALID_DATA;
1196
1197 Stream_Read_UINT16(s, drdynvc->PriorityCharge0);
1198 Stream_Read_UINT16(s, drdynvc->PriorityCharge1);
1199 Stream_Read_UINT16(s, drdynvc->PriorityCharge2);
1200 Stream_Read_UINT16(s, drdynvc->PriorityCharge3);
1201 }
1202
1203 status = drdynvc_send_capability_response(drdynvc);
1204 drdynvc->state = DRDYNVC_STATE_READY;
1205 return status;
1206}
1207
1208static UINT32 drdynvc_cblen_to_bytes(int cbLen)
1209{
1210 switch (cbLen)
1211 {
1212 case 0:
1213 return 1;
1214
1215 case 1:
1216 return 2;
1217
1218 default:
1219 return 4;
1220 }
1221}
1222
1223static UINT32 drdynvc_read_variable_uint(wStream* s, int cbLen)
1224{
1225 UINT32 val = 0;
1226
1227 switch (cbLen)
1228 {
1229 case 0:
1230 Stream_Read_UINT8(s, val);
1231 break;
1232
1233 case 1:
1234 Stream_Read_UINT16(s, val);
1235 break;
1236
1237 default:
1238 Stream_Read_UINT32(s, val);
1239 break;
1240 }
1241
1242 return val;
1243}
1244
1250static UINT drdynvc_process_create_request(drdynvcPlugin* drdynvc, UINT8 Sp, UINT8 cbChId,
1251 wStream* s)
1252{
1253 UINT status = 0;
1254 wStream* data_out = nullptr;
1255 UINT channel_status = 0;
1256 DVCMAN* dvcman = nullptr;
1257 DVCMAN_CHANNEL* channel = nullptr;
1258 INT32 retStatus = 0;
1259
1260 WINPR_UNUSED(Sp);
1261 if (!drdynvc)
1262 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1263
1264 dvcman = (DVCMAN*)drdynvc->channel_mgr;
1265 WINPR_ASSERT(dvcman);
1266
1267 if (drdynvc->state == DRDYNVC_STATE_CAPABILITIES)
1268 {
1274 drdynvc->version = 3;
1275
1276 if ((status = drdynvc_send_capability_response(drdynvc)))
1277 {
1278 WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_send_capability_response failed!");
1279 return status;
1280 }
1281
1282 drdynvc->state = DRDYNVC_STATE_READY;
1283 }
1284
1285 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1286 return ERROR_INVALID_DATA;
1287
1288 const UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1289 const size_t pos = Stream_GetPosition(s);
1290 const char* name = Stream_ConstPointer(s);
1291 const size_t length = Stream_GetRemainingLength(s);
1292
1293 if (strnlen(name, length) >= length)
1294 return ERROR_INVALID_DATA;
1295
1296 WLog_Print(drdynvc->log, WLOG_DEBUG,
1297 "process_create_request: ChannelId=%" PRIu32 " ChannelName=%s", ChannelId, name);
1298
1299 data_out = StreamPool_Take(dvcman->pool, pos + 4);
1300 if (!data_out)
1301 {
1302 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
1303 return CHANNEL_RC_NO_MEMORY;
1304 }
1305
1306 Stream_Write_UINT8(data_out, (CREATE_REQUEST_PDU << 4) | cbChId);
1307 if (!Stream_SetPosition(s, 1))
1308 return ERROR_INVALID_DATA;
1309 Stream_Copy(s, data_out, pos - 1);
1310
1311 channel =
1312 dvcman_create_channel(drdynvc, drdynvc->channel_mgr, ChannelId, name, &channel_status);
1313 switch (channel_status)
1314 {
1315 case CHANNEL_RC_OK:
1316 WLog_Print(drdynvc->log, WLOG_DEBUG, "channel created");
1317 retStatus = 0;
1318 break;
1319 case CHANNEL_RC_NO_MEMORY:
1320 WLog_Print(drdynvc->log, WLOG_DEBUG, "not enough memory for channel creation");
1321 retStatus = STATUS_NO_MEMORY;
1322 break;
1323 case ERROR_NOT_FOUND:
1324 WLog_Print(drdynvc->log, WLOG_DEBUG, "no listener for '%s'", name);
1325 retStatus = STATUS_NOT_FOUND; /* same code used by mstsc, STATUS_UNSUCCESSFUL */
1326 break;
1327 default:
1328 WLog_Print(drdynvc->log, WLOG_DEBUG, "channel creation error");
1329 retStatus = STATUS_UNSUCCESSFUL; /* same code used by mstsc, STATUS_UNSUCCESSFUL */
1330 break;
1331 }
1332 Stream_Write_INT32(data_out, retStatus);
1333
1334 status = drdynvc_send(drdynvc, data_out);
1335 if (status != CHANNEL_RC_OK)
1336 {
1337 WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]",
1338 WTSErrorToString(status), status);
1339 dvcman_channel_unref(channel);
1340 return status;
1341 }
1342
1343 if (channel_status == CHANNEL_RC_OK)
1344 {
1345 if ((status = dvcman_open_channel(drdynvc, channel)))
1346 {
1347 WLog_Print(drdynvc->log, WLOG_ERROR,
1348 "dvcman_open_channel failed with error %" PRIu32 "!", status);
1349 return status;
1350 }
1351 }
1352
1353 return status;
1354}
1355
1361static UINT drdynvc_process_data_first(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s,
1362 BOOL compressed, UINT32 ThreadingFlags)
1363{
1364 WINPR_ASSERT(drdynvc);
1365 if (!Stream_CheckAndLogRequiredLength(
1366 TAG, s, drdynvc_cblen_to_bytes(cbChId) + drdynvc_cblen_to_bytes(Sp)))
1367 return ERROR_INVALID_DATA;
1368
1369 UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1370 UINT32 Length = drdynvc_read_variable_uint(s, Sp);
1371 WLog_Print(drdynvc->log, WLOG_TRACE,
1372 "process_data_first: Sp=%d cbChId=%d, ChannelId=%" PRIu32 " Length=%" PRIu32 "", Sp,
1373 cbChId, ChannelId, Length);
1374
1375 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1376 if (!channel)
1377 {
1383 WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId);
1384 return CHANNEL_RC_OK;
1385 }
1386
1387 UINT status = CHANNEL_RC_OK;
1388 BOOL shouldFree = FALSE;
1389 if (channel->state != DVC_CHANNEL_RUNNING)
1390 goto out;
1391
1392 if (compressed)
1393 {
1394 BYTE* data = nullptr;
1395 UINT32 dataSize = 0;
1396 if (zgfx_decompress(channel->decompressor, Stream_Pointer(s),
1397 WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data,
1398 &dataSize, 0) < 0)
1399 {
1400 status = ERROR_INVALID_DATA;
1401 WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing first packet");
1402 goto out;
1403 }
1404
1405 s = Stream_New(data, dataSize);
1406 if (!s)
1407 {
1408 status = CHANNEL_RC_NO_MEMORY;
1409 WLog_Print(drdynvc->log, WLOG_ERROR, "error allocating new Stream(len=%" PRIu32 ")",
1410 dataSize);
1411 free(data);
1412 goto out;
1413 }
1414 shouldFree = TRUE;
1415 }
1416
1417 status = dvcman_receive_channel_data_first(channel, Length);
1418
1419 if (status == CHANNEL_RC_OK)
1420 status = dvcman_receive_channel_data(channel, s, ThreadingFlags);
1421
1422 if (status != CHANNEL_RC_OK)
1423 status = dvcman_channel_close(channel, FALSE, FALSE);
1424
1425out:
1426 if (shouldFree)
1427 Stream_Free(s, TRUE);
1428 dvcman_channel_unref(channel);
1429 return status;
1430}
1431
1437static UINT drdynvc_process_data(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s,
1438 BOOL compressed, UINT32 ThreadingFlags)
1439{
1440 WINPR_ASSERT(drdynvc);
1441 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1442 return ERROR_INVALID_DATA;
1443
1444 UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1445 WLog_Print(drdynvc->log, WLOG_TRACE, "process_data: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp,
1446 cbChId, ChannelId);
1447
1448 DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1449 if (!channel)
1450 {
1456 WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId);
1457 return CHANNEL_RC_OK;
1458 }
1459
1460 BOOL shouldFree = FALSE;
1461 UINT status = CHANNEL_RC_OK;
1462 if (channel->state != DVC_CHANNEL_RUNNING)
1463 goto out;
1464
1465 if (compressed)
1466 {
1467 BYTE* data = nullptr;
1468 UINT32 dataSize = 0;
1469
1470 if (zgfx_decompress(channel->decompressor, Stream_Pointer(s),
1471 WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data,
1472 &dataSize, 0) < 0)
1473 {
1474 status = ERROR_INVALID_DATA;
1475 WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing data packet");
1476 goto out;
1477 }
1478
1479 s = Stream_New(data, dataSize);
1480 if (!s)
1481 {
1482 status = CHANNEL_RC_NO_MEMORY;
1483 WLog_Print(drdynvc->log, WLOG_ERROR, "error allocating new Stream(len=%" PRIu32 ")",
1484 dataSize);
1485 free(data);
1486 goto out;
1487 }
1488 shouldFree = TRUE;
1489 }
1490
1491 status = dvcman_receive_channel_data(channel, s, ThreadingFlags);
1492 if (status != CHANNEL_RC_OK)
1493 status = dvcman_channel_close(channel, FALSE, FALSE);
1494
1495out:
1496 if (shouldFree)
1497 Stream_Free(s, TRUE);
1498 dvcman_channel_unref(channel);
1499 return status;
1500}
1501
1507static UINT drdynvc_process_close_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s)
1508{
1509 UINT32 ChannelId = 0;
1510 DVCMAN_CHANNEL* channel = nullptr;
1511
1512 WINPR_ASSERT(drdynvc);
1513 if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId)))
1514 return ERROR_INVALID_DATA;
1515
1516 ChannelId = drdynvc_read_variable_uint(s, cbChId);
1517 WLog_Print(drdynvc->log, WLOG_DEBUG,
1518 "process_close_request: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp, cbChId,
1519 ChannelId);
1520
1521 channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE);
1522 if (!channel)
1523 {
1524 WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_close_request channel %" PRIu32 " not present",
1525 ChannelId);
1526 return CHANNEL_RC_OK;
1527 }
1528
1529 dvcman_channel_close(channel, TRUE, FALSE);
1530 dvcman_channel_unref(channel);
1531 return CHANNEL_RC_OK;
1532}
1533
1539static UINT drdynvc_order_recv(drdynvcPlugin* drdynvc, wStream* s, UINT32 ThreadingFlags)
1540{
1541 WINPR_ASSERT(drdynvc);
1542 if (!Stream_CheckAndLogRequiredLength(TAG, s, 1))
1543 return ERROR_INVALID_DATA;
1544
1545 UINT8 value = Stream_Get_UINT8(s);
1546 const UINT8 Cmd = (value & 0xf0) >> 4;
1547 const UINT8 Sp = (value & 0x0c) >> 2;
1548 const UINT8 cbChId = (value & 0x03) >> 0;
1549 WLog_Print(drdynvc->log, WLOG_TRACE, "order_recv: Cmd=%s, Sp=%" PRIu8 " cbChId=%" PRIu8,
1550 drdynvc_get_packet_type(Cmd), Sp, cbChId);
1551
1552 switch (Cmd)
1553 {
1554 case CAPABILITY_REQUEST_PDU:
1555 return drdynvc_process_capability_request(drdynvc, Sp, cbChId, s);
1556
1557 case CREATE_REQUEST_PDU:
1558 return drdynvc_process_create_request(drdynvc, Sp, cbChId, s);
1559
1560 case DATA_FIRST_PDU:
1561 case DATA_FIRST_COMPRESSED_PDU:
1562 return drdynvc_process_data_first(drdynvc, Sp, cbChId, s,
1563 (Cmd == DATA_FIRST_COMPRESSED_PDU), ThreadingFlags);
1564
1565 case DATA_PDU:
1566 case DATA_COMPRESSED_PDU:
1567 return drdynvc_process_data(drdynvc, Sp, cbChId, s, (Cmd == DATA_COMPRESSED_PDU),
1568 ThreadingFlags);
1569
1570 case CLOSE_REQUEST_PDU:
1571 return drdynvc_process_close_request(drdynvc, Sp, cbChId, s);
1572
1573 case SOFT_SYNC_RESPONSE_PDU:
1574 WLog_Print(drdynvc->log, WLOG_ERROR,
1575 "not expecting a SOFT_SYNC_RESPONSE_PDU as a client");
1576 return ERROR_INTERNAL_ERROR;
1577
1578 default:
1579 WLog_Print(drdynvc->log, WLOG_ERROR, "unknown drdynvc cmd 0x%x", Cmd);
1580 return ERROR_INTERNAL_ERROR;
1581 }
1582}
1583
1589static UINT drdynvc_virtual_channel_event_data_received(drdynvcPlugin* drdynvc, void* pData,
1590 UINT32 dataLength, UINT32 totalLength,
1591 UINT32 dataFlags)
1592{
1593 wStream* data_in = nullptr;
1594
1595 WINPR_ASSERT(drdynvc);
1596 if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME))
1597 {
1598 return CHANNEL_RC_OK;
1599 }
1600
1601 if (dataFlags & CHANNEL_FLAG_FIRST)
1602 {
1603 DVCMAN* mgr = (DVCMAN*)drdynvc->channel_mgr;
1604 if (drdynvc->data_in)
1605 Stream_Release(drdynvc->data_in);
1606
1607 drdynvc->data_in = StreamPool_Take(mgr->pool, totalLength);
1608 }
1609
1610 if (!(data_in = drdynvc->data_in))
1611 {
1612 WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!");
1613 return CHANNEL_RC_NO_MEMORY;
1614 }
1615
1616 if (!Stream_EnsureRemainingCapacity(data_in, dataLength))
1617 {
1618 WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_EnsureRemainingCapacity failed!");
1619 Stream_Release(drdynvc->data_in);
1620 drdynvc->data_in = nullptr;
1621 return ERROR_INTERNAL_ERROR;
1622 }
1623
1624 Stream_Write(data_in, pData, dataLength);
1625
1626 if (dataFlags & CHANNEL_FLAG_LAST)
1627 {
1628 const size_t cap = Stream_Capacity(data_in);
1629 const size_t pos = Stream_GetPosition(data_in);
1630 if (cap < pos)
1631 {
1632 WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_plugin_process_received: read error");
1633 return ERROR_INVALID_DATA;
1634 }
1635
1636 drdynvc->data_in = nullptr;
1637 Stream_SealLength(data_in);
1638 Stream_ResetPosition(data_in);
1639
1640 if (drdynvc->async)
1641 {
1642 if (!MessageQueue_Post(drdynvc->queue, nullptr, 0, (void*)data_in, nullptr))
1643 {
1644 WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Post failed!");
1645 return ERROR_INTERNAL_ERROR;
1646 }
1647 }
1648 else
1649 {
1650 UINT error = drdynvc_order_recv(drdynvc, data_in, TRUE);
1651 Stream_Release(data_in);
1652
1653 if (error)
1654 {
1655 WLog_Print(drdynvc->log, WLOG_WARN,
1656 "drdynvc_order_recv failed with error %" PRIu32 "!", error);
1657 return error;
1658 }
1659 }
1660 }
1661
1662 return CHANNEL_RC_OK;
1663}
1664
1665static void VCAPITYPE drdynvc_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle,
1666 UINT event, LPVOID pData,
1667 UINT32 dataLength, UINT32 totalLength,
1668 UINT32 dataFlags)
1669{
1670 UINT error = CHANNEL_RC_OK;
1671 drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam;
1672
1673 WINPR_ASSERT(drdynvc);
1674 switch (event)
1675 {
1676 case CHANNEL_EVENT_DATA_RECEIVED:
1677 if (!drdynvc || (drdynvc->OpenHandle != openHandle))
1678 {
1679 WLog_ERR(TAG, "drdynvc_virtual_channel_open_event: error no match");
1680 return;
1681 }
1682 if ((error = drdynvc_virtual_channel_event_data_received(drdynvc, pData, dataLength,
1683 totalLength, dataFlags)))
1684 WLog_Print(drdynvc->log, WLOG_ERROR,
1685 "drdynvc_virtual_channel_event_data_received failed with error %" PRIu32
1686 "",
1687 error);
1688
1689 break;
1690
1691 case CHANNEL_EVENT_WRITE_CANCELLED:
1692 case CHANNEL_EVENT_WRITE_COMPLETE:
1693 {
1694 wStream* s = (wStream*)pData;
1695 Stream_Release(s);
1696 }
1697 break;
1698
1699 case CHANNEL_EVENT_USER:
1700 break;
1701 default:
1702 break;
1703 }
1704
1705 if (error && drdynvc && drdynvc->rdpcontext)
1706 setChannelError(drdynvc->rdpcontext, error,
1707 "drdynvc_virtual_channel_open_event reported an error");
1708}
1709
1710static DWORD WINAPI drdynvc_virtual_channel_client_thread(LPVOID arg)
1711{
1712 /* TODO: rewrite this */
1713 wStream* data = nullptr;
1714 wMessage message = WINPR_C_ARRAY_INIT;
1715 UINT error = CHANNEL_RC_OK;
1716 drdynvcPlugin* drdynvc = (drdynvcPlugin*)arg;
1717
1718 if (!drdynvc)
1719 {
1720 ExitThread((DWORD)CHANNEL_RC_BAD_CHANNEL_HANDLE);
1721 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1722 }
1723
1724 while (1)
1725 {
1726 if (!MessageQueue_Wait(drdynvc->queue))
1727 {
1728 WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Wait failed!");
1729 error = ERROR_INTERNAL_ERROR;
1730 break;
1731 }
1732
1733 if (!MessageQueue_Peek(drdynvc->queue, &message, TRUE))
1734 {
1735 WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Peek failed!");
1736 error = ERROR_INTERNAL_ERROR;
1737 break;
1738 }
1739
1740 if (message.id == WMQ_QUIT)
1741 break;
1742
1743 if (message.id == 0)
1744 {
1745 UINT32 ThreadingFlags = TRUE;
1746 data = (wStream*)message.wParam;
1747
1748 if ((error = drdynvc_order_recv(drdynvc, data, ThreadingFlags)))
1749 {
1750 WLog_Print(drdynvc->log, WLOG_WARN,
1751 "drdynvc_order_recv failed with error %" PRIu32 "!", error);
1752 }
1753
1754 Stream_Release(data);
1755 }
1756 }
1757
1758 {
1759 /* Disconnect remaining dynamic channels that the server did not.
1760 * This is required to properly shut down channels by calling the appropriate
1761 * event handlers. */
1762 DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr;
1763
1764 HashTable_Clear(drdynvcMgr->channelsById);
1765 }
1766
1767 if (error && drdynvc->rdpcontext)
1768 setChannelError(drdynvc->rdpcontext, error,
1769 "drdynvc_virtual_channel_client_thread reported an error");
1770
1771 ExitThread((DWORD)error);
1772 return error;
1773}
1774
1775static void drdynvc_queue_object_free(void* obj)
1776{
1777 wStream* s = nullptr;
1778 wMessage* msg = (wMessage*)obj;
1779
1780 if (!msg || (msg->id != 0))
1781 return;
1782
1783 s = (wStream*)msg->wParam;
1784
1785 if (s)
1786 Stream_Release(s);
1787}
1788
1789static UINT drdynvc_virtual_channel_event_initialized(drdynvcPlugin* drdynvc, LPVOID pData,
1790 UINT32 dataLength)
1791{
1792 wObject* obj = nullptr;
1793 WINPR_UNUSED(pData);
1794 WINPR_UNUSED(dataLength);
1795
1796 if (!drdynvc)
1797 goto error;
1798
1799 drdynvc->queue = MessageQueue_New(nullptr);
1800
1801 if (!drdynvc->queue)
1802 {
1803 WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_New failed!");
1804 goto error;
1805 }
1806
1807 obj = MessageQueue_Object(drdynvc->queue);
1808 obj->fnObjectFree = drdynvc_queue_object_free;
1809 drdynvc->channel_mgr = dvcman_new(drdynvc);
1810
1811 if (!drdynvc->channel_mgr)
1812 {
1813 WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_new failed!");
1814 goto error;
1815 }
1816
1817 return CHANNEL_RC_OK;
1818error:
1819 return ERROR_INTERNAL_ERROR;
1820}
1821
1827static UINT drdynvc_virtual_channel_event_connected(drdynvcPlugin* drdynvc, LPVOID pData,
1828 UINT32 dataLength)
1829{
1830 UINT error = 0;
1831 UINT32 status = 0;
1832 rdpSettings* settings = nullptr;
1833
1834 WINPR_ASSERT(drdynvc);
1835 WINPR_UNUSED(pData);
1836 WINPR_UNUSED(dataLength);
1837
1838 if (!drdynvc)
1839 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1840
1841 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelOpenEx);
1842 status = drdynvc->channelEntryPoints.pVirtualChannelOpenEx(
1843 drdynvc->InitHandle, &drdynvc->OpenHandle, drdynvc->channelDef.name,
1844 drdynvc_virtual_channel_open_event_ex);
1845
1846 if (status != CHANNEL_RC_OK)
1847 {
1848 WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelOpen failed with %s [%08" PRIX32 "]",
1849 WTSErrorToString(status), status);
1850 return status;
1851 }
1852
1853 WINPR_ASSERT(drdynvc->rdpcontext);
1854 settings = drdynvc->rdpcontext->settings;
1855 WINPR_ASSERT(settings);
1856
1857 for (UINT32 index = 0;
1858 index < freerdp_settings_get_uint32(settings, FreeRDP_DynamicChannelCount); index++)
1859 {
1860 const ADDIN_ARGV* args =
1861 freerdp_settings_get_pointer_array(settings, FreeRDP_DynamicChannelArray, index);
1862 error = dvcman_load_addin(drdynvc, drdynvc->channel_mgr, args, drdynvc->rdpcontext);
1863
1864 if (CHANNEL_RC_OK != error)
1865 goto error;
1866 }
1867
1868 if ((error = dvcman_init(drdynvc, drdynvc->channel_mgr)))
1869 {
1870 WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_init failed with error %" PRIu32 "!", error);
1871 goto error;
1872 }
1873
1874 drdynvc->state = DRDYNVC_STATE_CAPABILITIES;
1875
1876 if (drdynvc->async)
1877 {
1878 if (!(drdynvc->thread = CreateThread(nullptr, 0, drdynvc_virtual_channel_client_thread,
1879 (void*)drdynvc, 0, nullptr)))
1880 {
1881 error = ERROR_INTERNAL_ERROR;
1882 WLog_Print(drdynvc->log, WLOG_ERROR, "CreateThread failed!");
1883 goto error;
1884 }
1885
1886 if (!SetThreadPriority(drdynvc->thread, THREAD_PRIORITY_HIGHEST))
1887 WLog_Print(drdynvc->log, WLOG_WARN, "SetThreadPriority failed, ignoring.");
1888 }
1889
1890error:
1891 return error;
1892}
1893
1899static UINT drdynvc_virtual_channel_event_disconnected(drdynvcPlugin* drdynvc)
1900{
1901 UINT status = 0;
1902
1903 if (!drdynvc)
1904 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1905
1906 if (drdynvc->OpenHandle == 0)
1907 return CHANNEL_RC_OK;
1908
1909 if (drdynvc->queue)
1910 {
1911 if (!MessageQueue_PostQuit(drdynvc->queue, 0))
1912 {
1913 status = GetLastError();
1914 WLog_Print(drdynvc->log, WLOG_ERROR,
1915 "MessageQueue_PostQuit failed with error %" PRIu32 "", status);
1916 return status;
1917 }
1918 }
1919
1920 if (drdynvc->thread)
1921 {
1922 if (WaitForSingleObject(drdynvc->thread, INFINITE) != WAIT_OBJECT_0)
1923 {
1924 status = GetLastError();
1925 WLog_Print(drdynvc->log, WLOG_ERROR,
1926 "WaitForSingleObject failed with error %" PRIu32 "", status);
1927 return status;
1928 }
1929
1930 (void)CloseHandle(drdynvc->thread);
1931 drdynvc->thread = nullptr;
1932 }
1933 else
1934 {
1935 {
1936 /* Disconnect remaining dynamic channels that the server did not.
1937 * This is required to properly shut down channels by calling the appropriate
1938 * event handlers. */
1939 DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr;
1940
1941 HashTable_Clear(drdynvcMgr->channelsById);
1942 }
1943 }
1944
1945 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelCloseEx);
1946 status = drdynvc->channelEntryPoints.pVirtualChannelCloseEx(drdynvc->InitHandle,
1947 drdynvc->OpenHandle);
1948
1949 if (status != CHANNEL_RC_OK)
1950 {
1951 WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelClose failed with %s [%08" PRIX32 "]",
1952 WTSErrorToString(status), status);
1953 }
1954
1955 dvcman_clear(drdynvc, drdynvc->channel_mgr);
1956 if (drdynvc->queue)
1957 MessageQueue_Clear(drdynvc->queue);
1958 drdynvc->OpenHandle = 0;
1959
1960 if (drdynvc->data_in)
1961 {
1962 Stream_Release(drdynvc->data_in);
1963 drdynvc->data_in = nullptr;
1964 }
1965
1966 return status;
1967}
1968
1974static UINT drdynvc_virtual_channel_event_terminated(drdynvcPlugin* drdynvc)
1975{
1976 if (!drdynvc)
1977 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1978
1979 MessageQueue_Free(drdynvc->queue);
1980 drdynvc->queue = nullptr;
1981
1982 if (drdynvc->channel_mgr)
1983 {
1984 dvcman_free(drdynvc, drdynvc->channel_mgr);
1985 drdynvc->channel_mgr = nullptr;
1986 }
1987 drdynvc->InitHandle = nullptr;
1988 free(drdynvc->context);
1989 free(drdynvc);
1990 return CHANNEL_RC_OK;
1991}
1992
1993static UINT drdynvc_virtual_channel_event_attached(drdynvcPlugin* drdynvc)
1994{
1995 UINT error = CHANNEL_RC_OK;
1996 DVCMAN* dvcman = nullptr;
1997
1998 if (!drdynvc)
1999 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
2000
2001 dvcman = (DVCMAN*)drdynvc->channel_mgr;
2002
2003 if (!dvcman)
2004 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
2005
2006 ArrayList_Lock(dvcman->plugins);
2007 for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
2008 {
2009 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
2010
2011 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Attached, pPlugin);
2012 if (error != CHANNEL_RC_OK)
2013 {
2014 WLog_Print(drdynvc->log, WLOG_ERROR, "Attach failed with error %" PRIu32 "!", error);
2015 goto fail;
2016 }
2017 }
2018
2019fail:
2020 ArrayList_Unlock(dvcman->plugins);
2021 return error;
2022}
2023
2024static UINT drdynvc_virtual_channel_event_detached(drdynvcPlugin* drdynvc)
2025{
2026 UINT error = CHANNEL_RC_OK;
2027 DVCMAN* dvcman = nullptr;
2028
2029 if (!drdynvc)
2030 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
2031
2032 dvcman = (DVCMAN*)drdynvc->channel_mgr;
2033
2034 if (!dvcman)
2035 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
2036
2037 ArrayList_Lock(dvcman->plugins);
2038 for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++)
2039 {
2040 IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i);
2041
2042 error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Detached, pPlugin);
2043 if (error != CHANNEL_RC_OK)
2044 {
2045 WLog_Print(drdynvc->log, WLOG_ERROR, "Detach failed with error %" PRIu32 "!", error);
2046 goto fail;
2047 }
2048 }
2049
2050fail:
2051 ArrayList_Unlock(dvcman->plugins);
2052
2053 return error;
2054}
2055
2056static VOID VCAPITYPE drdynvc_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle,
2057 UINT event, LPVOID pData,
2058 UINT dataLength)
2059{
2060 UINT error = CHANNEL_RC_OK;
2061 drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam;
2062
2063 if (!drdynvc || (drdynvc->InitHandle != pInitHandle))
2064 {
2065 WLog_ERR(TAG, "drdynvc_virtual_channel_init_event: error no match");
2066 return;
2067 }
2068
2069 switch (event)
2070 {
2071 case CHANNEL_EVENT_INITIALIZED:
2072 error = drdynvc_virtual_channel_event_initialized(drdynvc, pData, dataLength);
2073 break;
2074 case CHANNEL_EVENT_CONNECTED:
2075 if ((error = drdynvc_virtual_channel_event_connected(drdynvc, pData, dataLength)))
2076 WLog_Print(drdynvc->log, WLOG_ERROR,
2077 "drdynvc_virtual_channel_event_connected failed with error %" PRIu32 "",
2078 error);
2079
2080 break;
2081
2082 case CHANNEL_EVENT_DISCONNECTED:
2083 if ((error = drdynvc_virtual_channel_event_disconnected(drdynvc)))
2084 WLog_Print(drdynvc->log, WLOG_ERROR,
2085 "drdynvc_virtual_channel_event_disconnected failed with error %" PRIu32
2086 "",
2087 error);
2088
2089 break;
2090
2091 case CHANNEL_EVENT_TERMINATED:
2092 if ((error = drdynvc_virtual_channel_event_terminated(drdynvc)))
2093 WLog_Print(drdynvc->log, WLOG_ERROR,
2094 "drdynvc_virtual_channel_event_terminated failed with error %" PRIu32 "",
2095 error);
2096
2097 break;
2098
2099 case CHANNEL_EVENT_ATTACHED:
2100 if ((error = drdynvc_virtual_channel_event_attached(drdynvc)))
2101 WLog_Print(drdynvc->log, WLOG_ERROR,
2102 "drdynvc_virtual_channel_event_attached failed with error %" PRIu32 "",
2103 error);
2104
2105 break;
2106
2107 case CHANNEL_EVENT_DETACHED:
2108 if ((error = drdynvc_virtual_channel_event_detached(drdynvc)))
2109 WLog_Print(drdynvc->log, WLOG_ERROR,
2110 "drdynvc_virtual_channel_event_detached failed with error %" PRIu32 "",
2111 error);
2112
2113 break;
2114
2115 default:
2116 break;
2117 }
2118
2119 if (error && drdynvc->rdpcontext)
2120 setChannelError(drdynvc->rdpcontext, error,
2121 "drdynvc_virtual_channel_init_event_ex reported an error");
2122}
2123
2128static int drdynvc_get_version(DrdynvcClientContext* context)
2129{
2130 WINPR_ASSERT(context);
2131 drdynvcPlugin* drdynvc = (drdynvcPlugin*)context->handle;
2132 WINPR_ASSERT(drdynvc);
2133 return drdynvc->version;
2134}
2135
2136/* drdynvc is always built-in */
2137#define VirtualChannelEntryEx drdynvc_VirtualChannelEntryEx
2138
2139FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS_EX pEntryPoints,
2140 PVOID pInitHandle))
2141{
2142 UINT rc = 0;
2143 drdynvcPlugin* drdynvc = nullptr;
2144 DrdynvcClientContext* context = nullptr;
2145 CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx = nullptr;
2146 drdynvc = (drdynvcPlugin*)calloc(1, sizeof(drdynvcPlugin));
2147
2148 WINPR_ASSERT(pEntryPoints);
2149 if (!drdynvc)
2150 {
2151 WLog_ERR(TAG, "calloc failed!");
2152 return FALSE;
2153 }
2154
2155 drdynvc->channelDef.options =
2156 CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP | CHANNEL_OPTION_COMPRESS_RDP;
2157 (void)sprintf_s(drdynvc->channelDef.name, ARRAYSIZE(drdynvc->channelDef.name),
2158 DRDYNVC_SVC_CHANNEL_NAME);
2159 drdynvc->state = DRDYNVC_STATE_INITIAL;
2160 pEntryPointsEx = (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints;
2161
2162 if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) &&
2163 (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER))
2164 {
2165 context = (DrdynvcClientContext*)calloc(1, sizeof(DrdynvcClientContext));
2166
2167 if (!context)
2168 {
2169 WLog_Print(drdynvc->log, WLOG_ERROR, "calloc failed!");
2170 free(drdynvc);
2171 return FALSE;
2172 }
2173
2174 context->handle = (void*)drdynvc;
2175 context->custom = nullptr;
2176 drdynvc->context = context;
2177 context->GetVersion = drdynvc_get_version;
2178 drdynvc->rdpcontext = pEntryPointsEx->context;
2179 if (!freerdp_settings_get_bool(drdynvc->rdpcontext->settings,
2180 FreeRDP_TransportDumpReplay) &&
2181 !freerdp_settings_get_bool(drdynvc->rdpcontext->settings,
2182 FreeRDP_SynchronousDynamicChannels))
2183 drdynvc->async = TRUE;
2184 }
2185
2186 drdynvc->log = WLog_Get(TAG);
2187 WLog_Print(drdynvc->log, WLOG_DEBUG, "VirtualChannelEntryEx");
2188 CopyMemory(&(drdynvc->channelEntryPoints), pEntryPoints,
2190 drdynvc->InitHandle = pInitHandle;
2191
2192 WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelInitEx);
2193 rc = drdynvc->channelEntryPoints.pVirtualChannelInitEx(
2194 drdynvc, context, pInitHandle, &drdynvc->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000,
2195 drdynvc_virtual_channel_init_event_ex);
2196
2197 if (CHANNEL_RC_OK != rc)
2198 {
2199 WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelInit failed with %s [%08" PRIX32 "]",
2200 WTSErrorToString(rc), rc);
2201 free(drdynvc->context);
2202 free(drdynvc);
2203 return FALSE;
2204 }
2205
2206 drdynvc->channelEntryPoints.pInterface = context;
2207 return TRUE;
2208}
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.
Definition svc.h:60
This struct contains function pointer to initialize/free objects.
Definition collections.h:52
OBJECT_FREE_FN fnObjectFree
Definition collections.h:59
WINPR_ATTR_NODISCARD OBJECT_EQUALS_FN fnObjectEquals
Definition collections.h:61
WINPR_ATTR_NODISCARD OBJECT_NEW_FN fnObjectNew
Definition collections.h:54