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