FreeRDP
Loading...
Searching...
No Matches
libfreerdp/core/client.c
1
22#include <freerdp/config.h>
23
24#include "settings.h"
25
26#include <winpr/assert.h>
27
28#include <freerdp/log.h>
29#include <freerdp/channels/drdynvc.h>
30
31#include "rdp.h"
32#include "client.h"
33
34#define TAG FREERDP_TAG("core.client")
35
36typedef struct
37{
38 freerdp_channel_handle_fkt_t fkt;
39 void* userdata;
40} ChannelEventEntry;
41
42/* Use this instance to get access to channels in VirtualChannelInit. It is set during
43 * freerdp_connect so channels that use VirtualChannelInit must be initialized from the same thread
44 * as freerdp_connect was called */
45static WINPR_TLS freerdp* g_Instance = NULL;
46
47/* use global counter to ensure uniqueness across channel manager instances */
48static volatile LONG g_OpenHandleSeq = 1;
49
50/* HashTable mapping channel handles to CHANNEL_OPEN_DATA */
51static INIT_ONCE g_ChannelHandlesOnce = INIT_ONCE_STATIC_INIT;
52static wHashTable* g_ChannelHandles = NULL;
53
54static BOOL freerdp_channels_process_message_free(wMessage* message, DWORD type);
55
56static CHANNEL_OPEN_DATA* freerdp_channels_find_channel_open_data_by_name(rdpChannels* channels,
57 const char* name)
58{
59 for (int index = 0; index < channels->openDataCount; index++)
60 {
61 CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[index];
62
63 if (strncmp(name, pChannelOpenData->name, CHANNEL_NAME_LEN + 1) == 0)
64 return pChannelOpenData;
65 }
66
67 return NULL;
68}
69
70/* returns rdpChannel for the channel name passed in */
71static rdpMcsChannel* freerdp_channels_find_channel_by_name(rdpRdp* rdp, const char* name)
72{
73 rdpMcs* mcs = NULL;
74
75 if (!rdp)
76 return NULL;
77
78 mcs = rdp->mcs;
79
80 for (UINT32 index = 0; index < mcs->channelCount; index++)
81 {
82 rdpMcsChannel* channel = &mcs->channels[index];
83
84 if (strncmp(name, channel->Name, CHANNEL_NAME_LEN + 1) == 0)
85 {
86 return channel;
87 }
88 }
89
90 return NULL;
91}
92
93static rdpMcsChannel* freerdp_channels_find_channel_by_id(rdpRdp* rdp, UINT16 channel_id)
94{
95 rdpMcsChannel* channel = NULL;
96 rdpMcs* mcs = NULL;
97
98 if (!rdp)
99 return NULL;
100
101 mcs = rdp->mcs;
102
103 for (UINT32 index = 0; index < mcs->channelCount; index++)
104 {
105 channel = &mcs->channels[index];
106
107 if (channel->ChannelId == channel_id)
108 {
109 return channel;
110 }
111 }
112
113 return NULL;
114}
115
116static void channel_queue_message_free(wMessage* msg)
117{
118 CHANNEL_OPEN_EVENT* ev = NULL;
119
120 if (!msg || (msg->id != 0))
121 return;
122
123 ev = (CHANNEL_OPEN_EVENT*)msg->wParam;
124 free(ev);
125}
126
127static void channel_queue_free(void* obj)
128{
129 wMessage* msg = (wMessage*)obj;
130 freerdp_channels_process_message_free(msg, CHANNEL_EVENT_WRITE_CANCELLED);
131 channel_queue_message_free(msg);
132}
133
134static BOOL CALLBACK init_channel_handles_table(WINPR_ATTR_UNUSED PINIT_ONCE once,
135 WINPR_ATTR_UNUSED PVOID param,
136 WINPR_ATTR_UNUSED PVOID* context)
137{
138 g_ChannelHandles = HashTable_New(TRUE);
139 return TRUE;
140}
141
142static void* channel_event_entry_clone(const void* data)
143{
144 const ChannelEventEntry* entry = data;
145 if (!entry)
146 return NULL;
147
148 ChannelEventEntry* copy = calloc(1, sizeof(ChannelEventEntry));
149 if (!copy)
150 return NULL;
151 *copy = *entry;
152 return copy;
153}
154
155rdpChannels* freerdp_channels_new(freerdp* instance)
156{
157 wObject* obj = NULL;
158 rdpChannels* channels = NULL;
159 channels = (rdpChannels*)calloc(1, sizeof(rdpChannels));
160
161 if (!channels)
162 return NULL;
163
164 InitOnceExecuteOnce(&g_ChannelHandlesOnce, init_channel_handles_table, NULL, NULL);
165
166 if (!g_ChannelHandles)
167 goto error;
168 if (!InitializeCriticalSectionAndSpinCount(&channels->channelsLock, 4000))
169 goto error;
170
171 channels->instance = instance;
172 channels->queue = MessageQueue_New(NULL);
173
174 if (!channels->queue)
175 goto error;
176
177 obj = MessageQueue_Object(channels->queue);
178 obj->fnObjectFree = channel_queue_free;
179
180 channels->channelEvents = HashTable_New(FALSE);
181 if (!channels->channelEvents)
182 goto error;
183
184 obj = HashTable_ValueObject(channels->channelEvents);
185 WINPR_ASSERT(obj);
186 obj->fnObjectFree = free;
187 obj->fnObjectNew = channel_event_entry_clone;
188 return channels;
189error:
190 WINPR_PRAGMA_DIAG_PUSH
191 WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC
192 freerdp_channels_free(channels);
193 WINPR_PRAGMA_DIAG_POP
194 return NULL;
195}
196
197void freerdp_channels_free(rdpChannels* channels)
198{
199 if (!channels)
200 return;
201
202 HashTable_Free(channels->channelEvents);
203
204 DeleteCriticalSection(&channels->channelsLock);
205
206 if (channels->queue)
207 {
208 MessageQueue_Free(channels->queue);
209 channels->queue = NULL;
210 }
211
212 free(channels);
213}
214
220static UINT freerdp_drdynvc_on_channel_connected(DrdynvcClientContext* context, const char* name,
221 void* pInterface)
222{
223 UINT status = CHANNEL_RC_OK;
224 ChannelConnectedEventArgs e = { 0 };
225 rdpChannels* channels = (rdpChannels*)context->custom;
226 freerdp* instance = channels->instance;
227 EventArgsInit(&e, "freerdp");
228 e.name = name;
229 e.pInterface = pInterface;
230 PubSub_OnChannelConnected(instance->context->pubSub, instance->context, &e);
231 return status;
232}
233
239static UINT freerdp_drdynvc_on_channel_disconnected(DrdynvcClientContext* context, const char* name,
240 void* pInterface)
241{
242 UINT status = CHANNEL_RC_OK;
243 ChannelDisconnectedEventArgs e = { 0 };
244 rdpChannels* channels = (rdpChannels*)context->custom;
245 freerdp* instance = channels->instance;
246 EventArgsInit(&e, "freerdp");
247 e.name = name;
248 e.pInterface = pInterface;
249 PubSub_OnChannelDisconnected(instance->context->pubSub, instance->context, &e);
250 return status;
251}
252
253static UINT freerdp_drdynvc_on_channel_attached(DrdynvcClientContext* context, const char* name,
254 void* pInterface)
255{
256 UINT status = CHANNEL_RC_OK;
257 ChannelAttachedEventArgs e = { 0 };
258 rdpChannels* channels = (rdpChannels*)context->custom;
259 freerdp* instance = channels->instance;
260 EventArgsInit(&e, "freerdp");
261 e.name = name;
262 e.pInterface = pInterface;
263 PubSub_OnChannelAttached(instance->context->pubSub, instance->context, &e);
264 return status;
265}
266
267static UINT freerdp_drdynvc_on_channel_detached(DrdynvcClientContext* context, const char* name,
268 void* pInterface)
269{
270 UINT status = CHANNEL_RC_OK;
271 ChannelDetachedEventArgs e = { 0 };
272 rdpChannels* channels = (rdpChannels*)context->custom;
273 freerdp* instance = channels->instance;
274 EventArgsInit(&e, "freerdp");
275 e.name = name;
276 e.pInterface = pInterface;
277 PubSub_OnChannelDetached(instance->context->pubSub, instance->context, &e);
278 return status;
279}
280
281void freerdp_channels_register_instance(WINPR_ATTR_UNUSED rdpChannels* channels, freerdp* instance)
282{
283 /* store instance in TLS so future VirtualChannelInit calls can use it */
284 g_Instance = instance;
285}
286
291UINT freerdp_channels_pre_connect(rdpChannels* channels, freerdp* instance)
292{
293 UINT error = CHANNEL_RC_OK;
294 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
295
296 MessageQueue_Clear(channels->queue);
297
298 for (int index = 0; index < channels->clientDataCount; index++)
299 {
300 pChannelClientData = &channels->clientDataList[index];
301
302 if (pChannelClientData->pChannelInitEventProc)
303 {
304 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
305 CHANNEL_EVENT_INITIALIZED, 0, 0);
306 }
307 else if (pChannelClientData->pChannelInitEventProcEx)
308 {
309 pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
310 pChannelClientData->pInitHandle,
311 CHANNEL_EVENT_INITIALIZED, 0, 0);
312 }
313
314 if (CHANNEL_RC_OK != getChannelError(instance->context))
315 break;
316 }
317
318 return error;
319}
320
321UINT freerdp_channels_attach(freerdp* instance)
322{
323 UINT error = CHANNEL_RC_OK;
324 const char* hostname = NULL;
325 size_t hostnameLength = 0;
326 rdpChannels* channels = NULL;
327 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
328
329 WINPR_ASSERT(instance);
330 WINPR_ASSERT(instance->context);
331 WINPR_ASSERT(instance->context->settings);
332
333 channels = instance->context->channels;
334 hostname = freerdp_settings_get_string(instance->context->settings, FreeRDP_ServerHostname);
335 WINPR_ASSERT(hostname);
336 hostnameLength = strnlen(hostname, MAX_PATH);
337
338 for (int index = 0; index < channels->clientDataCount; index++)
339 {
340 union
341 {
342 const void* cpv;
343 void* pv;
344 } cnv;
345 ChannelAttachedEventArgs e = { 0 };
346 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
347
348 cnv.cpv = hostname;
349 pChannelClientData = &channels->clientDataList[index];
350
351 if (pChannelClientData->pChannelInitEventProc)
352 {
353
354 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
355 CHANNEL_EVENT_ATTACHED, cnv.pv,
356 (UINT)hostnameLength);
357 }
358 else if (pChannelClientData->pChannelInitEventProcEx)
359 {
360 pChannelClientData->pChannelInitEventProcEx(
361 pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
362 CHANNEL_EVENT_ATTACHED, cnv.pv, (UINT)hostnameLength);
363 }
364
365 if (getChannelError(instance->context) != CHANNEL_RC_OK)
366 goto fail;
367
368 pChannelOpenData = &channels->openDataList[index];
369 EventArgsInit(&e, "freerdp");
370 e.name = pChannelOpenData->name;
371 e.pInterface = pChannelOpenData->pInterface;
372 PubSub_OnChannelAttached(instance->context->pubSub, instance->context, &e);
373 }
374
375fail:
376 return error;
377}
378
379UINT freerdp_channels_detach(freerdp* instance)
380{
381 UINT error = CHANNEL_RC_OK;
382 const char* hostname = NULL;
383 size_t hostnameLength = 0;
384 rdpChannels* channels = NULL;
385 rdpContext* context = NULL;
386 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
387
388 WINPR_ASSERT(instance);
389
390 context = instance->context;
391 WINPR_ASSERT(context);
392
393 channels = context->channels;
394 WINPR_ASSERT(channels);
395
396 WINPR_ASSERT(context->settings);
397 hostname = freerdp_settings_get_string(context->settings, FreeRDP_ServerHostname);
398 WINPR_ASSERT(hostname);
399 hostnameLength = strnlen(hostname, MAX_PATH);
400
401 for (int index = 0; index < channels->clientDataCount; index++)
402 {
403 union
404 {
405 const void* cpv;
406 void* pv;
407 } cnv;
408
409 ChannelDetachedEventArgs e = { 0 };
410 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
411
412 cnv.cpv = hostname;
413 pChannelClientData = &channels->clientDataList[index];
414
415 if (pChannelClientData->pChannelInitEventProc)
416 {
417 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
418 CHANNEL_EVENT_DETACHED, cnv.pv,
419 (UINT)hostnameLength);
420 }
421 else if (pChannelClientData->pChannelInitEventProcEx)
422 {
423 pChannelClientData->pChannelInitEventProcEx(
424 pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
425 CHANNEL_EVENT_DETACHED, cnv.pv, (UINT)hostnameLength);
426 }
427
428 if (getChannelError(context) != CHANNEL_RC_OK)
429 goto fail;
430
431 pChannelOpenData = &channels->openDataList[index];
432 EventArgsInit(&e, "freerdp");
433 e.name = pChannelOpenData->name;
434 e.pInterface = pChannelOpenData->pInterface;
435 PubSub_OnChannelDetached(context->pubSub, context, &e);
436 }
437
438fail:
439 return error;
440}
441
447UINT freerdp_channels_post_connect(rdpChannels* channels, freerdp* instance)
448{
449 UINT error = CHANNEL_RC_OK;
450 const char* hostname = NULL;
451 size_t hostnameLength = 0;
452 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
453
454 WINPR_ASSERT(channels);
455 WINPR_ASSERT(instance);
456 WINPR_ASSERT(instance->context);
457 WINPR_ASSERT(instance->context->settings);
458
459 channels->connected = TRUE;
460 hostname = freerdp_settings_get_string(instance->context->settings, FreeRDP_ServerHostname);
461 WINPR_ASSERT(hostname);
462 hostnameLength = strnlen(hostname, MAX_PATH);
463
464 for (int index = 0; index < channels->clientDataCount; index++)
465 {
466 union
467 {
468 const void* pcb;
469 void* pb;
470 } cnv;
471 ChannelConnectedEventArgs e = { 0 };
472 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
473 pChannelClientData = &channels->clientDataList[index];
474
475 cnv.pcb = hostname;
476 if (pChannelClientData->pChannelInitEventProc)
477 {
478 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
479 CHANNEL_EVENT_CONNECTED, cnv.pb,
480 (UINT)hostnameLength);
481 }
482 else if (pChannelClientData->pChannelInitEventProcEx)
483 {
484 pChannelClientData->pChannelInitEventProcEx(
485 pChannelClientData->lpUserParam, pChannelClientData->pInitHandle,
486 CHANNEL_EVENT_CONNECTED, cnv.pb, (UINT)hostnameLength);
487 }
488
489 error = getChannelError(instance->context);
490 if (error != CHANNEL_RC_OK)
491 goto fail;
492
493 pChannelOpenData = &channels->openDataList[index];
494 EventArgsInit(&e, "freerdp");
495 e.name = pChannelOpenData->name;
496 e.pInterface = pChannelOpenData->pInterface;
497 PubSub_OnChannelConnected(instance->context->pubSub, instance->context, &e);
498 }
499
500 channels->drdynvc = (DrdynvcClientContext*)freerdp_channels_get_static_channel_interface(
501 channels, DRDYNVC_SVC_CHANNEL_NAME);
502
503 if (channels->drdynvc)
504 {
505 channels->drdynvc->custom = (void*)channels;
506 channels->drdynvc->OnChannelConnected = freerdp_drdynvc_on_channel_connected;
507 channels->drdynvc->OnChannelDisconnected = freerdp_drdynvc_on_channel_disconnected;
508 channels->drdynvc->OnChannelAttached = freerdp_drdynvc_on_channel_attached;
509 channels->drdynvc->OnChannelDetached = freerdp_drdynvc_on_channel_detached;
510 }
511
512fail:
513 return error;
514}
515
516BOOL freerdp_channels_data(freerdp* instance, UINT16 channelId, const BYTE* cdata, size_t dataSize,
517 UINT32 flags, size_t totalSize)
518{
519 rdpMcs* mcs = NULL;
520 rdpChannels* channels = NULL;
521 rdpMcsChannel* channel = NULL;
522 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
523 union
524 {
525 const BYTE* pcb;
526 BYTE* pb;
527 } data;
528
529 data.pcb = cdata;
530 if (!instance || !data.pcb)
531 {
532 WLog_ERR(TAG, "(%p, %" PRIu16 ", %p, 0x%08x): Invalid arguments",
533 WINPR_CXX_COMPAT_CAST(const void*, instance), channelId,
534 WINPR_CXX_COMPAT_CAST(const void*, data.pcb), flags);
535 return FALSE;
536 }
537
538 mcs = instance->context->rdp->mcs;
539 channels = instance->context->channels;
540
541 if (!channels || !mcs)
542 {
543 return FALSE;
544 }
545
546 for (UINT32 index = 0; index < mcs->channelCount; index++)
547 {
548 rdpMcsChannel* cur = &mcs->channels[index];
549
550 if (cur->ChannelId == channelId)
551 {
552 channel = cur;
553 break;
554 }
555 }
556
557 if (!channel)
558 {
559 return FALSE;
560 }
561
562 pChannelOpenData = freerdp_channels_find_channel_open_data_by_name(channels, channel->Name);
563
564 if (!pChannelOpenData)
565 {
566 return FALSE;
567 }
568
569 if (pChannelOpenData->pChannelOpenEventProc)
570 {
571 pChannelOpenData->pChannelOpenEventProc(pChannelOpenData->OpenHandle,
572 CHANNEL_EVENT_DATA_RECEIVED, data.pb,
573 (UINT32)dataSize, (UINT32)totalSize, flags);
574 }
575 else if (pChannelOpenData->pChannelOpenEventProcEx)
576 {
577 pChannelOpenData->pChannelOpenEventProcEx(
578 pChannelOpenData->lpUserParam, pChannelOpenData->OpenHandle,
579 CHANNEL_EVENT_DATA_RECEIVED, data.pb, (UINT32)dataSize, (UINT32)totalSize, flags);
580 }
581
582 return TRUE;
583}
584
585UINT16 freerdp_channels_get_id_by_name(freerdp* instance, const char* channel_name)
586{
587 if (!instance || !channel_name)
588 return 0;
589
590 rdpMcsChannel* mcsChannel =
591 freerdp_channels_find_channel_by_name(instance->context->rdp, channel_name);
592 if (!mcsChannel)
593 return 0;
594
595 return mcsChannel->ChannelId;
596}
597
598const char* freerdp_channels_get_name_by_id(freerdp* instance, UINT16 channelId)
599{
600 rdpMcsChannel* mcsChannel = NULL;
601 if (!instance)
602 return NULL;
603
604 mcsChannel = freerdp_channels_find_channel_by_id(instance->context->rdp, channelId);
605 if (!mcsChannel)
606 return NULL;
607
608 return mcsChannel->Name;
609}
610
611BOOL freerdp_channels_process_message_free(wMessage* message, DWORD type)
612{
613 if (message->id == WMQ_QUIT)
614 {
615 return FALSE;
616 }
617
618 if (message->id == 0)
619 {
620 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
621 CHANNEL_OPEN_EVENT* item = (CHANNEL_OPEN_EVENT*)message->wParam;
622
623 if (!item)
624 return FALSE;
625
626 pChannelOpenData = item->pChannelOpenData;
627
628 if (pChannelOpenData->pChannelOpenEventProc)
629 {
630 pChannelOpenData->pChannelOpenEventProc(pChannelOpenData->OpenHandle, type,
631 item->UserData, item->DataLength,
632 item->DataLength, 0);
633 }
634 else if (pChannelOpenData->pChannelOpenEventProcEx)
635 {
636 pChannelOpenData->pChannelOpenEventProcEx(
637 pChannelOpenData->lpUserParam, pChannelOpenData->OpenHandle, type, item->UserData,
638 item->DataLength, item->DataLength, 0);
639 }
640 }
641
642 return TRUE;
643}
644
645static BOOL freerdp_channels_process_message(freerdp* instance, wMessage* message)
646{
647 BOOL ret = TRUE;
648 BOOL rc = FALSE;
649
650 WINPR_ASSERT(instance);
651 WINPR_ASSERT(message);
652
653 if (message->id == WMQ_QUIT)
654 goto fail;
655 else if (message->id == 0)
656 {
657 rdpMcsChannel* channel = NULL;
658 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
659 CHANNEL_OPEN_EVENT* item = (CHANNEL_OPEN_EVENT*)message->wParam;
660
661 if (!item)
662 goto fail;
663
664 pChannelOpenData = item->pChannelOpenData;
665 if (pChannelOpenData->flags != 2)
666 {
667 freerdp_channels_process_message_free(message, CHANNEL_EVENT_WRITE_CANCELLED);
668 goto fail;
669 }
670 channel =
671 freerdp_channels_find_channel_by_name(instance->context->rdp, pChannelOpenData->name);
672
673 if (channel)
674 ret = instance->SendChannelData(instance, channel->ChannelId, item->Data,
675 item->DataLength);
676 }
677
678 if (!freerdp_channels_process_message_free(message, CHANNEL_EVENT_WRITE_COMPLETE))
679 goto fail;
680
681 rc = ret;
682
683fail:
684 IFCALL(message->Free, message);
685 return rc;
686}
687
691static BOOL freerdp_channels_process_sync(rdpChannels* channels, freerdp* instance)
692{
693 BOOL status = TRUE;
694 wMessage message = { 0 };
695
696 WINPR_ASSERT(channels);
697
698 while (MessageQueue_Peek(channels->queue, &message, TRUE))
699 {
700 if (!freerdp_channels_process_message(instance, &message))
701 status = FALSE;
702 }
703
704 return status;
705}
706
710#if defined(WITH_FREERDP_DEPRECATED)
711BOOL freerdp_channels_get_fds(rdpChannels* channels, WINPR_ATTR_UNUSED freerdp* instance,
712 void** read_fds, int* read_count, WINPR_ATTR_UNUSED void** write_fds,
713 WINPR_ATTR_UNUSED int* write_count)
714{
715 void* pfd = NULL;
716 pfd = GetEventWaitObject(MessageQueue_Event(channels->queue));
717
718 if (pfd)
719 {
720 read_fds[*read_count] = pfd;
721 (*read_count)++;
722 }
723
724 return TRUE;
725}
726#endif
727
728void* freerdp_channels_get_static_channel_interface(rdpChannels* channels, const char* name)
729{
730 void* pInterface = NULL;
731 CHANNEL_OPEN_DATA* pChannelOpenData =
732 freerdp_channels_find_channel_open_data_by_name(channels, name);
733
734 if (pChannelOpenData)
735 pInterface = pChannelOpenData->pInterface;
736
737 return pInterface;
738}
739
740HANDLE freerdp_channels_get_event_handle(freerdp* instance)
741{
742 if (!instance)
743 return INVALID_HANDLE_VALUE;
744
745 WINPR_ASSERT(instance->context);
746
747 rdpChannels* channels = instance->context->channels;
748 WINPR_ASSERT(channels);
749
750 return MessageQueue_Event(channels->queue);
751}
752
753static BOOL channels_process(const void* key, void* value, void* arg)
754{
755 ChannelEventEntry* entry = value;
756 rdpContext* context = arg;
757
758 WINPR_UNUSED(key);
759
760 if (!entry->fkt)
761 return FALSE;
762 return entry->fkt(context, entry->userdata);
763}
764
765int freerdp_channels_process_pending_messages(freerdp* instance)
766{
767 if (!instance)
768 return -1;
769
770 WINPR_ASSERT(instance->context);
771
772 rdpChannels* channels = instance->context->channels;
773 WINPR_ASSERT(channels);
774
775 const DWORD status = WaitForSingleObject(MessageQueue_Event(channels->queue), 0);
776 if (status == WAIT_OBJECT_0)
777 {
778 if (!freerdp_channels_process_sync(channels, instance))
779 return -1;
780 }
781
782 if (!HashTable_Foreach(channels->channelEvents, channels_process, instance->context))
783 return -1;
784
785 return 1;
786}
787
791BOOL freerdp_channels_check_fds(rdpChannels* channels, freerdp* instance)
792{
793 WINPR_ASSERT(channels);
794 WINPR_UNUSED(channels);
795
796 const int rc = freerdp_channels_process_pending_messages(instance);
797 return rc == 1;
798}
799
800BOOL freerdp_client_channel_register(rdpChannels* channels, HANDLE handle,
801 freerdp_channel_handle_fkt_t fkt, void* userdata)
802{
803 if (!channels || (handle == INVALID_HANDLE_VALUE) || !fkt)
804 {
805 WLog_ERR(TAG, "Invalid function arguments (channels=%p, handle=%p, fkt=%p, userdata=%p",
806 WINPR_CXX_COMPAT_CAST(const void*, channels), handle,
807 WINPR_CXX_COMPAT_CAST(const void*, fkt), userdata);
808 return FALSE;
809 }
810
811 ChannelEventEntry entry = { .fkt = fkt, .userdata = userdata };
812 return HashTable_Insert(channels->channelEvents, handle, &entry);
813}
814
815BOOL freerdp_client_channel_unregister(rdpChannels* channels, HANDLE handle)
816{
817 if (!channels || (handle == INVALID_HANDLE_VALUE))
818 {
819 WLog_ERR(TAG, "Invalid function arguments (channels=%p, handle=%p",
820 WINPR_CXX_COMPAT_CAST(const void*, channels), handle);
821 return FALSE;
822 }
823
824 return HashTable_Remove(channels->channelEvents, handle);
825}
826
827SSIZE_T freerdp_client_channel_get_registered_event_handles(rdpChannels* channels, HANDLE* events,
828 DWORD count)
829{
830 SSIZE_T rc = -1;
831
832 WINPR_ASSERT(channels);
833 WINPR_ASSERT(events || (count == 0));
834
835 HashTable_Lock(channels->channelEvents);
836 size_t len = HashTable_Count(channels->channelEvents);
837 if (len <= count)
838 {
839 ULONG_PTR* keys = NULL;
840 const size_t nrKeys = HashTable_GetKeys(channels->channelEvents, &keys);
841 if ((nrKeys <= SSIZE_MAX) && (nrKeys == len))
842 {
843 for (size_t x = 0; x < nrKeys; x++)
844 {
845 HANDLE cur = (HANDLE)keys[x];
846 events[x] = cur;
847 }
848 rc = (SSIZE_T)nrKeys;
849 }
850 free(keys);
851 }
852 HashTable_Unlock(channels->channelEvents);
853 return rc;
854}
855
856UINT freerdp_channels_disconnect(rdpChannels* channels, freerdp* instance)
857{
858 UINT error = CHANNEL_RC_OK;
859 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
860 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
861
862 WINPR_ASSERT(channels);
863
864 if (!channels->connected)
865 return 0;
866
867 (void)freerdp_channels_check_fds(channels, instance);
868
869 /* tell all libraries we are shutting down */
870 for (int index = 0; index < channels->clientDataCount; index++)
871 {
872 ChannelDisconnectedEventArgs e = { 0 };
873 pChannelClientData = &channels->clientDataList[index];
874
875 if (pChannelClientData->pChannelInitEventProc)
876 {
877 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
878 CHANNEL_EVENT_DISCONNECTED, 0, 0);
879 }
880 else if (pChannelClientData->pChannelInitEventProcEx)
881 {
882 pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
883 pChannelClientData->pInitHandle,
884 CHANNEL_EVENT_DISCONNECTED, 0, 0);
885 }
886
887 pChannelOpenData = &channels->openDataList[index];
888 EventArgsInit(&e, "freerdp");
889 e.name = pChannelOpenData->name;
890 e.pInterface = pChannelOpenData->pInterface;
891 PubSub_OnChannelDisconnected(instance->context->pubSub, instance->context, &e);
892 }
893
894 channels->connected = FALSE;
895
896 /* Flush pending messages */
897 (void)freerdp_channels_check_fds(channels, instance);
898 return error;
899}
900
901void freerdp_channels_close(rdpChannels* channels, freerdp* instance)
902{
903 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
904 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
905
906 WINPR_ASSERT(channels);
907 WINPR_ASSERT(instance);
908
909 MessageQueue_PostQuit(channels->queue, 0);
910 (void)freerdp_channels_check_fds(channels, instance);
911
912 /* tell all libraries we are shutting down */
913 for (int index = 0; index < channels->clientDataCount; index++)
914 {
915 pChannelClientData = &channels->clientDataList[index];
916
917 if (pChannelClientData->pChannelInitEventProc)
918 {
919 pChannelClientData->pChannelInitEventProc(pChannelClientData->pInitHandle,
920 CHANNEL_EVENT_TERMINATED, 0, 0);
921 }
922 else if (pChannelClientData->pChannelInitEventProcEx)
923 {
924 pChannelClientData->pChannelInitEventProcEx(pChannelClientData->lpUserParam,
925 pChannelClientData->pInitHandle,
926 CHANNEL_EVENT_TERMINATED, 0, 0);
927 }
928 }
929
930 for (int index = 0; index < channels->openDataCount; index++)
931 {
932 pChannelOpenData = &channels->openDataList[index];
933 HashTable_Remove(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->OpenHandle);
934 }
935
936 channels->openDataCount = 0;
937 channels->initDataCount = 0;
938 channels->clientDataCount = 0;
939
940 WINPR_ASSERT(instance->context);
941 WINPR_ASSERT(instance->context->settings);
942 instance->context->settings->ChannelCount = 0;
943 g_Instance = NULL;
944}
945
946static UINT VCAPITYPE FreeRDP_VirtualChannelInitEx(
947 LPVOID lpUserParam, LPVOID clientContext, LPVOID pInitHandle, PCHANNEL_DEF pChannel,
948 INT channelCount, ULONG versionRequested, PCHANNEL_INIT_EVENT_EX_FN pChannelInitEventProcEx)
949{
950 rdpSettings* settings = NULL;
951 CHANNEL_INIT_DATA* pChannelInitData = NULL;
952 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
953 rdpChannels* channels = NULL;
954
955 if (!pInitHandle)
956 return CHANNEL_RC_BAD_INIT_HANDLE;
957
958 if (!pChannel)
959 return CHANNEL_RC_BAD_CHANNEL;
960
961 if ((channelCount <= 0) || !pChannelInitEventProcEx)
962 return CHANNEL_RC_INITIALIZATION_ERROR;
963
964 pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
965 WINPR_ASSERT(pChannelInitData);
966
967 channels = pChannelInitData->channels;
968 WINPR_ASSERT(channels);
969
970 if (!channels->can_call_init)
971 return CHANNEL_RC_NOT_IN_VIRTUALCHANNELENTRY;
972
973 if ((channels->openDataCount + channelCount) > CHANNEL_MAX_COUNT)
974 return CHANNEL_RC_TOO_MANY_CHANNELS;
975
976 if (channels->connected)
977 return CHANNEL_RC_ALREADY_CONNECTED;
978
979 if (versionRequested != VIRTUAL_CHANNEL_VERSION_WIN2000)
980 {
981 }
982
983 for (int index = 0; index < channelCount; index++)
984 {
985 const PCHANNEL_DEF pChannelDef = &pChannel[index];
986
987 if (freerdp_channels_find_channel_open_data_by_name(channels, pChannelDef->name) != 0)
988 {
989 return CHANNEL_RC_BAD_CHANNEL;
990 }
991 }
992
993 pChannelInitData->pInterface = clientContext;
994 pChannelClientData = &channels->clientDataList[channels->clientDataCount];
995 pChannelClientData->pChannelInitEventProcEx = pChannelInitEventProcEx;
996 pChannelClientData->pInitHandle = pInitHandle;
997 pChannelClientData->lpUserParam = lpUserParam;
998 channels->clientDataCount++;
999
1000 WINPR_ASSERT(channels->instance);
1001 WINPR_ASSERT(channels->instance->context);
1002 settings = channels->instance->context->settings;
1003 WINPR_ASSERT(settings);
1004
1005 for (int index = 0; index < channelCount; index++)
1006 {
1007 const PCHANNEL_DEF pChannelDef = &pChannel[index];
1008 CHANNEL_OPEN_DATA* pChannelOpenData = &channels->openDataList[channels->openDataCount];
1009
1010 WINPR_ASSERT(pChannelOpenData);
1011
1012 const LONG hdl = InterlockedIncrement(&g_OpenHandleSeq);
1013 pChannelOpenData->OpenHandle = WINPR_ASSERTING_INT_CAST(uint32_t, hdl);
1014 pChannelOpenData->channels = channels;
1015 pChannelOpenData->lpUserParam = lpUserParam;
1016 if (!HashTable_Insert(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->OpenHandle,
1017 (void*)pChannelOpenData))
1018 {
1019 pChannelInitData->pInterface = NULL;
1020 return CHANNEL_RC_INITIALIZATION_ERROR;
1021 }
1022 pChannelOpenData->flags = 1; /* init */
1023 strncpy(pChannelOpenData->name, pChannelDef->name, CHANNEL_NAME_LEN);
1024 pChannelOpenData->options = pChannelDef->options;
1025
1026 const UINT32 max = freerdp_settings_get_uint32(settings, FreeRDP_ChannelDefArraySize);
1027 WINPR_ASSERT(max >= CHANNEL_MAX_COUNT);
1028 if (settings->ChannelCount < max)
1029 {
1030 CHANNEL_DEF* channel = freerdp_settings_get_pointer_array_writable(
1031 settings, FreeRDP_ChannelDefArray, settings->ChannelCount);
1032 if (!channel)
1033 continue;
1034 strncpy(channel->name, pChannelDef->name, CHANNEL_NAME_LEN);
1035 channel->options = pChannelDef->options;
1036 settings->ChannelCount++;
1037 }
1038
1039 channels->openDataCount++;
1040 }
1041
1042 return CHANNEL_RC_OK;
1043}
1044
1045static UINT VCAPITYPE FreeRDP_VirtualChannelInit(LPVOID* ppInitHandle, PCHANNEL_DEF pChannel,
1046 INT channelCount, ULONG versionRequested,
1047 PCHANNEL_INIT_EVENT_FN pChannelInitEventProc)
1048{
1049 CHANNEL_DEF* channel = NULL;
1050 rdpSettings* settings = NULL;
1051 PCHANNEL_DEF pChannelDef = NULL;
1052 CHANNEL_INIT_DATA* pChannelInitData = NULL;
1053 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1054 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
1055 rdpChannels* channels = NULL;
1056
1057 /* g_Instance should have been set during freerdp_connect - otherwise VirtualChannelInit was
1058 * called from a different thread */
1059 if (!g_Instance || !g_Instance->context)
1060 return CHANNEL_RC_NOT_INITIALIZED;
1061
1062 channels = g_Instance->context->channels;
1063
1064 if (!ppInitHandle || !channels)
1065 return CHANNEL_RC_BAD_INIT_HANDLE;
1066
1067 if (!pChannel)
1068 return CHANNEL_RC_BAD_CHANNEL;
1069
1070 if ((channelCount <= 0) || !pChannelInitEventProc)
1071 return CHANNEL_RC_INITIALIZATION_ERROR;
1072
1073 pChannelInitData = &(channels->initDataList[channels->initDataCount]);
1074 *ppInitHandle = pChannelInitData;
1075 channels->initDataCount++;
1076 pChannelInitData->channels = channels;
1077 pChannelInitData->pInterface = NULL;
1078
1079 if (!channels->can_call_init)
1080 return CHANNEL_RC_NOT_IN_VIRTUALCHANNELENTRY;
1081
1082 if (channels->openDataCount + channelCount > CHANNEL_MAX_COUNT)
1083 return CHANNEL_RC_TOO_MANY_CHANNELS;
1084
1085 if (channels->connected)
1086 return CHANNEL_RC_ALREADY_CONNECTED;
1087
1088 if (versionRequested != VIRTUAL_CHANNEL_VERSION_WIN2000)
1089 {
1090 }
1091
1092 for (int index = 0; index < channelCount; index++)
1093 {
1094 pChannelDef = &pChannel[index];
1095
1096 if (freerdp_channels_find_channel_open_data_by_name(channels, pChannelDef->name) != 0)
1097 {
1098 return CHANNEL_RC_BAD_CHANNEL;
1099 }
1100 }
1101
1102 pChannelClientData = &channels->clientDataList[channels->clientDataCount];
1103 pChannelClientData->pChannelInitEventProc = pChannelInitEventProc;
1104 pChannelClientData->pInitHandle = *ppInitHandle;
1105 channels->clientDataCount++;
1106 settings = channels->instance->context->settings;
1107
1108 for (int index = 0; index < channelCount; index++)
1109 {
1110 UINT32 ChannelCount = freerdp_settings_get_uint32(settings, FreeRDP_ChannelCount);
1111
1112 pChannelDef = &pChannel[index];
1113
1114 pChannelOpenData = &channels->openDataList[channels->openDataCount];
1115
1116 const LONG hdl = InterlockedIncrement(&g_OpenHandleSeq);
1117 pChannelOpenData->OpenHandle = WINPR_ASSERTING_INT_CAST(uint32_t, hdl);
1118 pChannelOpenData->channels = channels;
1119 if (!HashTable_Insert(g_ChannelHandles, (void*)(UINT_PTR)pChannelOpenData->OpenHandle,
1120 (void*)pChannelOpenData))
1121 return CHANNEL_RC_INITIALIZATION_ERROR;
1122 pChannelOpenData->flags = 1; /* init */
1123 strncpy(pChannelOpenData->name, pChannelDef->name, CHANNEL_NAME_LEN);
1124 pChannelOpenData->options = pChannelDef->options;
1125
1126 if (ChannelCount < CHANNEL_MAX_COUNT)
1127 {
1128 channel = freerdp_settings_get_pointer_array_writable(settings, FreeRDP_ChannelDefArray,
1129 ChannelCount++);
1130 strncpy(channel->name, pChannelDef->name, CHANNEL_NAME_LEN);
1131 channel->options = pChannelDef->options;
1132 if (!freerdp_settings_set_uint32(settings, FreeRDP_ChannelCount, ChannelCount))
1133 return ERROR_INTERNAL_ERROR;
1134 }
1135
1136 channels->openDataCount++;
1137 }
1138
1139 return CHANNEL_RC_OK;
1140}
1141
1142static UINT VCAPITYPE
1143FreeRDP_VirtualChannelOpenEx(LPVOID pInitHandle, LPDWORD pOpenHandle, PCHAR pChannelName,
1144 PCHANNEL_OPEN_EVENT_EX_FN pChannelOpenEventProcEx)
1145{
1146 void* pInterface = NULL;
1147 rdpChannels* channels = NULL;
1148 CHANNEL_INIT_DATA* pChannelInitData = NULL;
1149 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1150 pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
1151 channels = pChannelInitData->channels;
1152 pInterface = pChannelInitData->pInterface;
1153
1154 if (!pOpenHandle)
1155 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1156
1157 if (!pChannelOpenEventProcEx)
1158 return CHANNEL_RC_BAD_PROC;
1159
1160 if (!channels->connected)
1161 return CHANNEL_RC_NOT_CONNECTED;
1162
1163 pChannelOpenData = freerdp_channels_find_channel_open_data_by_name(channels, pChannelName);
1164
1165 if (!pChannelOpenData)
1166 return CHANNEL_RC_UNKNOWN_CHANNEL_NAME;
1167
1168 if (pChannelOpenData->flags == 2)
1169 return CHANNEL_RC_ALREADY_OPEN;
1170
1171 pChannelOpenData->flags = 2; /* open */
1172 pChannelOpenData->pInterface = pInterface;
1173 pChannelOpenData->pChannelOpenEventProcEx = pChannelOpenEventProcEx;
1174 *pOpenHandle = pChannelOpenData->OpenHandle;
1175 return CHANNEL_RC_OK;
1176}
1177
1178static UINT VCAPITYPE FreeRDP_VirtualChannelOpen(LPVOID pInitHandle, LPDWORD pOpenHandle,
1179 PCHAR pChannelName,
1180 PCHANNEL_OPEN_EVENT_FN pChannelOpenEventProc)
1181{
1182 void* pInterface = NULL;
1183 rdpChannels* channels = NULL;
1184 CHANNEL_INIT_DATA* pChannelInitData = NULL;
1185 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1186 pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
1187 channels = pChannelInitData->channels;
1188 pInterface = pChannelInitData->pInterface;
1189
1190 if (!pOpenHandle)
1191 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1192
1193 if (!pChannelOpenEventProc)
1194 return CHANNEL_RC_BAD_PROC;
1195
1196 if (!channels->connected)
1197 return CHANNEL_RC_NOT_CONNECTED;
1198
1199 pChannelOpenData = freerdp_channels_find_channel_open_data_by_name(channels, pChannelName);
1200
1201 if (!pChannelOpenData)
1202 return CHANNEL_RC_UNKNOWN_CHANNEL_NAME;
1203
1204 if (pChannelOpenData->flags == 2)
1205 return CHANNEL_RC_ALREADY_OPEN;
1206
1207 pChannelOpenData->flags = 2; /* open */
1208 pChannelOpenData->pInterface = pInterface;
1209 pChannelOpenData->pChannelOpenEventProc = pChannelOpenEventProc;
1210 *pOpenHandle = pChannelOpenData->OpenHandle;
1211 return CHANNEL_RC_OK;
1212}
1213
1214static UINT VCAPITYPE FreeRDP_VirtualChannelCloseEx(LPVOID pInitHandle, DWORD openHandle)
1215{
1216 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1217
1218 if (!pInitHandle)
1219 return CHANNEL_RC_BAD_INIT_HANDLE;
1220
1221 pChannelOpenData = HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);
1222
1223 if (!pChannelOpenData)
1224 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1225
1226 if (pChannelOpenData->flags != 2)
1227 return CHANNEL_RC_NOT_OPEN;
1228
1229 pChannelOpenData->flags = 0;
1230 return CHANNEL_RC_OK;
1231}
1232
1233static UINT VCAPITYPE FreeRDP_VirtualChannelClose(DWORD openHandle)
1234{
1235 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1236
1237 pChannelOpenData = HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);
1238
1239 if (!pChannelOpenData)
1240 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1241
1242 if (pChannelOpenData->flags != 2)
1243 return CHANNEL_RC_NOT_OPEN;
1244
1245 pChannelOpenData->flags = 0;
1246 return CHANNEL_RC_OK;
1247}
1248
1249static UINT VCAPITYPE FreeRDP_VirtualChannelWriteEx(LPVOID pInitHandle, DWORD openHandle,
1250 LPVOID pData, ULONG dataLength,
1251 LPVOID pUserData)
1252{
1253 rdpChannels* channels = NULL;
1254 CHANNEL_INIT_DATA* pChannelInitData = NULL;
1255 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1256 CHANNEL_OPEN_EVENT* pChannelOpenEvent = NULL;
1257 wMessage message = { 0 };
1258
1259 if (!pInitHandle)
1260 return CHANNEL_RC_BAD_INIT_HANDLE;
1261
1262 pChannelInitData = (CHANNEL_INIT_DATA*)pInitHandle;
1263 channels = pChannelInitData->channels;
1264
1265 if (!channels)
1266 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1267
1268 pChannelOpenData = HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);
1269
1270 if (!pChannelOpenData)
1271 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1272
1273 if (!channels->connected)
1274 return CHANNEL_RC_NOT_CONNECTED;
1275
1276 if (!pData)
1277 return CHANNEL_RC_NULL_DATA;
1278
1279 if (!dataLength)
1280 return CHANNEL_RC_ZERO_LENGTH;
1281
1282 if (pChannelOpenData->flags != 2)
1283 return CHANNEL_RC_NOT_OPEN;
1284
1285 pChannelOpenEvent = (CHANNEL_OPEN_EVENT*)malloc(sizeof(CHANNEL_OPEN_EVENT));
1286
1287 if (!pChannelOpenEvent)
1288 return CHANNEL_RC_NO_MEMORY;
1289
1290 pChannelOpenEvent->Data = pData;
1291 pChannelOpenEvent->DataLength = dataLength;
1292 pChannelOpenEvent->UserData = pUserData;
1293 pChannelOpenEvent->pChannelOpenData = pChannelOpenData;
1294 message.context = channels;
1295 message.id = 0;
1296 message.wParam = pChannelOpenEvent;
1297 message.lParam = NULL;
1298 message.Free = channel_queue_message_free;
1299
1300 if (!MessageQueue_Dispatch(channels->queue, &message))
1301 {
1302 free(pChannelOpenEvent);
1303 return CHANNEL_RC_NO_MEMORY;
1304 }
1305
1306 return CHANNEL_RC_OK;
1307}
1308
1309static UINT VCAPITYPE FreeRDP_VirtualChannelWrite(DWORD openHandle, LPVOID pData, ULONG dataLength,
1310 LPVOID pUserData)
1311{
1312 wMessage message = { 0 };
1313 CHANNEL_OPEN_DATA* pChannelOpenData = NULL;
1314 CHANNEL_OPEN_EVENT* pChannelOpenEvent = NULL;
1315 rdpChannels* channels = NULL;
1316
1317 pChannelOpenData = HashTable_GetItemValue(g_ChannelHandles, (void*)(UINT_PTR)openHandle);
1318
1319 if (!pChannelOpenData)
1320 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1321
1322 channels = pChannelOpenData->channels;
1323 if (!channels)
1324 return CHANNEL_RC_BAD_CHANNEL_HANDLE;
1325
1326 if (!channels->connected)
1327 return CHANNEL_RC_NOT_CONNECTED;
1328
1329 if (!pData)
1330 return CHANNEL_RC_NULL_DATA;
1331
1332 if (!dataLength)
1333 return CHANNEL_RC_ZERO_LENGTH;
1334
1335 if (pChannelOpenData->flags != 2)
1336 return CHANNEL_RC_NOT_OPEN;
1337
1338 pChannelOpenEvent = (CHANNEL_OPEN_EVENT*)malloc(sizeof(CHANNEL_OPEN_EVENT));
1339
1340 if (!pChannelOpenEvent)
1341 return CHANNEL_RC_NO_MEMORY;
1342
1343 pChannelOpenEvent->Data = pData;
1344 pChannelOpenEvent->DataLength = dataLength;
1345 pChannelOpenEvent->UserData = pUserData;
1346 pChannelOpenEvent->pChannelOpenData = pChannelOpenData;
1347 message.context = channels;
1348 message.id = 0;
1349 message.wParam = pChannelOpenEvent;
1350 message.lParam = NULL;
1351 message.Free = channel_queue_message_free;
1352
1353 if (!MessageQueue_Dispatch(channels->queue, &message))
1354 {
1355 free(pChannelOpenEvent);
1356 return CHANNEL_RC_NO_MEMORY;
1357 }
1358
1359 return CHANNEL_RC_OK;
1360}
1361
1362static BOOL freerdp_channels_is_loaded(rdpChannels* channels, PVIRTUALCHANNELENTRY entry)
1363{
1364 for (int i = 0; i < channels->clientDataCount; i++)
1365 {
1366 CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[i];
1367
1368 if (pChannelClientData->entry == entry)
1369 return TRUE;
1370 }
1371
1372 return FALSE;
1373}
1374
1375static BOOL freerdp_channels_is_loaded_ex(rdpChannels* channels, PVIRTUALCHANNELENTRYEX entryEx)
1376{
1377 for (int i = 0; i < channels->clientDataCount; i++)
1378 {
1379 CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[i];
1380
1381 if (pChannelClientData->entryEx == entryEx)
1382 return TRUE;
1383 }
1384
1385 return FALSE;
1386}
1387
1388int freerdp_channels_client_load(rdpChannels* channels, WINPR_ATTR_UNUSED rdpSettings* settings,
1389 PVIRTUALCHANNELENTRY entry, void* data)
1390{
1391 CHANNEL_CLIENT_DATA* pChannelClientData = NULL;
1392
1393 WINPR_ASSERT(channels);
1394 WINPR_ASSERT(channels->instance);
1395 WINPR_ASSERT(channels->instance->context);
1396 WINPR_ASSERT(entry);
1397
1398 if (channels->clientDataCount + 1 > CHANNEL_MAX_COUNT)
1399 {
1400 WLog_ERR(TAG, "error: too many channels");
1401 return 1;
1402 }
1403
1404 if (freerdp_channels_is_loaded(channels, entry))
1405 {
1406 WLog_WARN(TAG, "Skipping, channel already loaded");
1407 return 0;
1408 }
1409
1410 pChannelClientData = &channels->clientDataList[channels->clientDataCount];
1411 pChannelClientData->entry = entry;
1412
1413 CHANNEL_ENTRY_POINTS_FREERDP EntryPoints = { .cbSize = sizeof(EntryPoints),
1414 .protocolVersion = VIRTUAL_CHANNEL_VERSION_WIN2000,
1415 .pVirtualChannelInit = FreeRDP_VirtualChannelInit,
1416 .pVirtualChannelOpen = FreeRDP_VirtualChannelOpen,
1417 .pVirtualChannelClose =
1418 FreeRDP_VirtualChannelClose,
1419 .pVirtualChannelWrite =
1420 FreeRDP_VirtualChannelWrite,
1421 .MagicNumber = FREERDP_CHANNEL_MAGIC_NUMBER,
1422 .pExtendedData = data,
1423 .context = channels->instance->context };
1424 /* enable VirtualChannelInit */
1425 union
1426 {
1428 PCHANNEL_ENTRY_POINTS px;
1429 } ptr;
1430
1431 ptr.pfx = &EntryPoints;
1432
1433 channels->can_call_init = TRUE;
1434 EnterCriticalSection(&channels->channelsLock);
1435 const BOOL status = pChannelClientData->entry(ptr.px);
1436 LeaveCriticalSection(&channels->channelsLock);
1437 /* disable MyVirtualChannelInit */
1438 channels->can_call_init = FALSE;
1439
1440 if (!status)
1441 {
1442 WLog_ERR(TAG, "error: channel export function call failed");
1443 return 1;
1444 }
1445
1446 return 0;
1447}
1448
1449int freerdp_channels_client_load_ex(rdpChannels* channels, WINPR_ATTR_UNUSED rdpSettings* settings,
1450 PVIRTUALCHANNELENTRYEX entryEx, void* data)
1451{
1452 WINPR_ASSERT(channels);
1453 WINPR_ASSERT(channels->instance);
1454 WINPR_ASSERT(channels->instance->context);
1455 WINPR_ASSERT(entryEx);
1456
1457 if (channels->clientDataCount + 1 > CHANNEL_MAX_COUNT)
1458 {
1459 WLog_ERR(TAG, "error: too many channels");
1460 return 1;
1461 }
1462
1463 if (freerdp_channels_is_loaded_ex(channels, entryEx))
1464 {
1465 WLog_WARN(TAG, "Skipping, channel already loaded");
1466 return 0;
1467 }
1468
1469 CHANNEL_CLIENT_DATA* pChannelClientData = &channels->clientDataList[channels->clientDataCount];
1470 pChannelClientData->entryEx = entryEx;
1471
1472 CHANNEL_INIT_DATA* pChannelInitData = &(channels->initDataList[channels->initDataCount++]);
1473 void* pInitHandle = pChannelInitData;
1474 pChannelInitData->channels = channels;
1475
1476 CHANNEL_ENTRY_POINTS_FREERDP_EX EntryPointsEx = {
1477 .cbSize = sizeof(EntryPointsEx),
1478 .protocolVersion = VIRTUAL_CHANNEL_VERSION_WIN2000,
1479 .pVirtualChannelInitEx = FreeRDP_VirtualChannelInitEx,
1480 .pVirtualChannelOpenEx = FreeRDP_VirtualChannelOpenEx,
1481 .pVirtualChannelCloseEx = FreeRDP_VirtualChannelCloseEx,
1482 .pVirtualChannelWriteEx = FreeRDP_VirtualChannelWriteEx,
1483 .MagicNumber = FREERDP_CHANNEL_MAGIC_NUMBER,
1484 .pExtendedData = data,
1485 .context = channels->instance->context
1486 };
1487
1488 union
1489 {
1491 PCHANNEL_ENTRY_POINTS_EX px;
1492 } ptr;
1493
1494 ptr.pfx = &EntryPointsEx;
1495
1496 /* enable VirtualChannelInit */
1497 channels->can_call_init = TRUE;
1498 EnterCriticalSection(&channels->channelsLock);
1499 const BOOL status = pChannelClientData->entryEx(ptr.px, pInitHandle);
1500 LeaveCriticalSection(&channels->channelsLock);
1501 /* disable MyVirtualChannelInit */
1502 channels->can_call_init = FALSE;
1503
1504 if (!status)
1505 {
1506 WLog_ERR(TAG, "error: channel export function call failed");
1507 return 1;
1508 }
1509
1510 return 0;
1511}
1512
1517int freerdp_channels_load_plugin(rdpChannels* channels, rdpSettings* settings, const char* name,
1518 void* data)
1519{
1520 PVIRTUALCHANNELENTRY entry =
1521 freerdp_load_channel_addin_entry(name, NULL, NULL, FREERDP_ADDIN_CHANNEL_STATIC);
1522
1523 if (!entry)
1524 return 1;
1525
1526 return freerdp_channels_client_load(channels, settings, entry, data);
1527}
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_set_uint32(rdpSettings *settings, FreeRDP_Settings_Keys_UInt32 id, UINT32 param)
Sets a UINT32 settings value.
FREERDP_API const char * freerdp_settings_get_string(const rdpSettings *settings, FreeRDP_Settings_Keys_String id)
Returns a immutable string settings value.
Definition svc.h:60
Definition svc.h:43
This struct contains function pointer to initialize/free objects.
Definition collections.h:57