19#include <winpr/assert.h>
21#include <freerdp/channels/drdynvc.h>
22#include <freerdp/utils/drdynvc.h>
23#include <freerdp/server/proxy/proxy_log.h>
25#include "pf_channel_drdynvc.h"
26#include "../pf_channel.h"
27#include "../proxy_modules.h"
28#include "../pf_utils.h"
30#define DTAG PROXY_TAG("drdynvc")
32#define Stream_CheckAndLogRequiredLengthWLogWithBackend(log, s, nmemb, backdata) \
33 Stream_CheckAndLogRequiredLengthWLogEx(log, WLOG_WARN, s, nmemb, 1, "%s(%s:%" PRIuz ")[%s]", \
34 __func__, __FILE__, (size_t)__LINE__, \
35 getDirection(backdata))
40 CHANNEL_OPENSTATE_WAITING_OPEN_STATUS,
41 CHANNEL_OPENSTATE_OPENED,
42 CHANNEL_OPENSTATE_CLOSED
43} PfDynChannelOpenStatus;
45typedef struct p_server_dynamic_channel_context pServerDynamicChannelContext;
46typedef struct DynChannelTrackerState DynChannelTrackerState;
48typedef PfChannelResult (*dynamic_channel_on_data_fn)(pServerContext* ps,
49 pServerDynamicChannelContext* channel,
50 BOOL isBackData, ChannelStateTracker* tracker,
51 BOOL firstPacket, BOOL lastPacket);
54struct DynChannelTrackerState
56 UINT32 currentDataLength;
57 UINT32 CurrentDataReceived;
58 UINT32 CurrentDataFragments;
60 dynamic_channel_on_data_fn dataCallback;
63typedef void (*channel_data_dtor_fn)(
void** user_data);
65struct p_server_dynamic_channel_context
69 PfDynChannelOpenStatus openStatus;
70 pf_utils_channel_mode channelMode;
71 BOOL packetReassembly;
72 DynChannelTrackerState backTracker;
73 DynChannelTrackerState frontTracker;
76 channel_data_dtor_fn channelDataDtor;
83 ChannelStateTracker* backTracker;
84 ChannelStateTracker* frontTracker;
93 DYNCVC_READ_INCOMPLETE
96static const char* openstatus2str(PfDynChannelOpenStatus status)
100 case CHANNEL_OPENSTATE_WAITING_OPEN_STATUS:
101 return "CHANNEL_OPENSTATE_WAITING_OPEN_STATUS";
102 case CHANNEL_OPENSTATE_CLOSED:
103 return "CHANNEL_OPENSTATE_CLOSED";
104 case CHANNEL_OPENSTATE_OPENED:
105 return "CHANNEL_OPENSTATE_OPENED";
107 return "CHANNEL_OPENSTATE_UNKNOWN";
111#define DynvcTrackerLog(log, level, dynChannel, cmd, isBackData, ...) \
112 dyn_log_((log), (level), (dynChannel), (cmd), (isBackData), __func__, __FILE__, __LINE__, \
115static const char* getDirection(BOOL isBackData)
117 return isBackData ?
"B->F" :
"F->B";
120static void dyn_log_(wLog* log, DWORD level,
const pServerDynamicChannelContext* dynChannel,
121 BYTE cmd, BOOL isBackData,
const char* fkt,
const char* file,
size_t line,
122 const char* fmt, ...)
124 if (!WLog_IsLevelActive(log, level))
129 size_t prefixlen = 0;
132 uint32_t channelId = dynChannel ? dynChannel->channelId : UINT32_MAX;
133 const char* channelName = dynChannel ? dynChannel->channelName :
"<NULL>";
134 (void)winpr_asprintf(&prefix, &prefixlen,
"DynvcTracker[%s](%s [%s:%" PRIu32
"])",
135 getDirection(isBackData), channelName, drdynvc_get_packet_type(cmd),
140 (void)winpr_vasprintf(&msg, &msglen, fmt, ap);
143 WLog_PrintTextMessage(log, level, line, file, fkt,
"%s: %s", prefix, msg);
148static PfChannelResult data_cb(pServerContext* ps, pServerDynamicChannelContext* channel,
149 BOOL isBackData, ChannelStateTracker* tracker, BOOL firstPacket,
153 WINPR_ASSERT(channel);
154 WINPR_ASSERT(tracker);
155 WINPR_ASSERT(ps->pdata);
157 wStream* currentPacket = channelTracker_getCurrentPacket(tracker);
159 .channelId = channel->channelId,
160 .data = currentPacket,
161 .isBackData = isBackData,
162 .first = firstPacket,
165 .packetSize = channelTracker_getCurrentPacketSize(tracker),
166 .result = PF_CHANNEL_RESULT_ERROR };
167 Stream_SealLength(dyn.data);
168 if (!pf_modules_run_filter(ps->pdata->module, FILTER_TYPE_INTERCEPT_CHANNEL, ps->pdata, &dyn))
169 return PF_CHANNEL_RESULT_ERROR;
171 channelTracker_setCurrentPacketSize(tracker, dyn.packetSize);
173 return channelTracker_flushCurrent(tracker, firstPacket, lastPacket, !isBackData);
177static pServerDynamicChannelContext* DynamicChannelContext_new(wLog* log, pServerContext* ps,
178 const char* name, UINT32
id)
182 pServerDynamicChannelContext* ret = calloc(1,
sizeof(*ret));
185 WLog_Print(log, WLOG_ERROR,
"error allocating dynamic channel context '%s'", name);
190 ret->channelName = _strdup(name);
191 if (!ret->channelName)
193 WLog_Print(log, WLOG_ERROR,
"error allocating name in dynamic channel context '%s'", name);
198 ret->frontTracker.dataCallback = data_cb;
199 ret->backTracker.dataCallback = data_cb;
202 if (pf_modules_run_filter(ps->pdata->module, FILTER_TYPE_DYN_INTERCEPT_LIST, ps->pdata, &dyn) &&
204 ret->channelMode = PF_UTILS_CHANNEL_INTERCEPT;
206 ret->channelMode = pf_utils_get_channel_mode(ps->pdata->config, name);
207 ret->openStatus = CHANNEL_OPENSTATE_OPENED;
208 ret->packetReassembly = (ret->channelMode == PF_UTILS_CHANNEL_INTERCEPT);
213static void DynamicChannelContext_free(
void* ptr)
215 pServerDynamicChannelContext* c = (pServerDynamicChannelContext*)ptr;
219 if (c->backTracker.currentPacket)
220 Stream_Free(c->backTracker.currentPacket, TRUE);
222 if (c->frontTracker.currentPacket)
223 Stream_Free(c->frontTracker.currentPacket, TRUE);
225 if (c->channelDataDtor)
226 c->channelDataDtor(&c->channelData);
228 free(c->channelName);
232static UINT32 ChannelId_Hash(
const void* key)
234 const UINT32* v = (
const UINT32*)key;
238static BOOL ChannelId_Compare(
const void* objA,
const void* objB)
240 const UINT32* v1 = objA;
241 const UINT32* v2 = objB;
245static DynvcReadResult dynvc_read_varInt(wLog* log,
wStream* s,
size_t len, UINT64* varInt,
248 WINPR_ASSERT(varInt);
252 if (!Stream_CheckAndLogRequiredLengthWLog(log, s, 1))
253 return last ? DYNCVC_READ_ERROR : DYNCVC_READ_INCOMPLETE;
254 Stream_Read_UINT8(s, *varInt);
257 if (!Stream_CheckAndLogRequiredLengthWLog(log, s, 2))
258 return last ? DYNCVC_READ_ERROR : DYNCVC_READ_INCOMPLETE;
259 Stream_Read_UINT16(s, *varInt);
262 if (!Stream_CheckAndLogRequiredLengthWLog(log, s, 4))
263 return last ? DYNCVC_READ_ERROR : DYNCVC_READ_INCOMPLETE;
264 Stream_Read_UINT32(s, *varInt);
268 WLog_Print(log, WLOG_ERROR,
"Unknown int len %" PRIuz, len);
269 return DYNCVC_READ_ERROR;
271 return DYNCVC_READ_OK;
274static PfChannelResult DynvcTrackerPeekHandleByMode(ChannelStateTracker* tracker,
275 DynChannelTrackerState* trackerState,
276 pServerDynamicChannelContext* dynChannel,
277 BYTE cmd, BOOL firstPacket, BOOL lastPacket)
279 WINPR_ASSERT(tracker);
280 WINPR_ASSERT(trackerState);
281 WINPR_ASSERT(dynChannel);
282 PfChannelResult result = PF_CHANNEL_RESULT_ERROR;
284 DynChannelContext* dynChannelContext =
285 (DynChannelContext*)channelTracker_getCustomData(tracker);
286 WINPR_ASSERT(dynChannelContext);
288 proxyData* pdata = channelTracker_getPData(tracker);
291 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
292 switch (dynChannel->channelMode)
294 case PF_UTILS_CHANNEL_PASSTHROUGH:
295 result = channelTracker_flushCurrent(tracker, firstPacket, lastPacket, !isBackData);
297 case PF_UTILS_CHANNEL_BLOCK:
298 channelTracker_setMode(tracker, CHANNEL_TRACKER_DROP);
299 result = PF_CHANNEL_RESULT_DROP;
301 case PF_UTILS_CHANNEL_INTERCEPT:
302 if (trackerState->dataCallback)
304 result = trackerState->dataCallback(pdata->ps, dynChannel, isBackData, tracker,
305 firstPacket, lastPacket);
309 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
310 "no intercept callback for channel, dropping packet");
311 result = PF_CHANNEL_RESULT_DROP;
315 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
316 "unknown channel mode %u", dynChannel->channelMode);
317 result = PF_CHANNEL_RESULT_ERROR;
321 if (!trackerState->currentDataLength ||
322 (trackerState->CurrentDataReceived == trackerState->currentDataLength))
324 trackerState->currentDataLength = 0;
325 trackerState->CurrentDataFragments = 0;
326 trackerState->CurrentDataReceived = 0;
328 if (dynChannel->packetReassembly && trackerState->currentPacket)
329 Stream_SetPosition(trackerState->currentPacket, 0);
335static PfChannelResult DynvcTrackerHandleClose(ChannelStateTracker* tracker,
336 pServerDynamicChannelContext* dynChannel,
337 DynChannelContext* dynChannelContext,
338 BOOL firstPacket, BOOL lastPacket)
340 WINPR_ASSERT(dynChannelContext);
342 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
344 if (!lastPacket || !dynChannel)
345 return PF_CHANNEL_RESULT_DROP;
347 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, CLOSE_REQUEST_PDU, isBackData,
349 channelTracker_setMode(tracker, CHANNEL_TRACKER_PASS);
350 if (dynChannel->openStatus != CHANNEL_OPENSTATE_OPENED)
352 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, CLOSE_REQUEST_PDU,
353 isBackData,
"is in state %s, expected %s",
354 openstatus2str(dynChannel->openStatus),
355 openstatus2str(CHANNEL_OPENSTATE_OPENED));
357 dynChannel->openStatus = CHANNEL_OPENSTATE_CLOSED;
358 return channelTracker_flushCurrent(tracker, firstPacket, lastPacket, !isBackData);
361static PfChannelResult DynvcTrackerHandleCreateBack(ChannelStateTracker* tracker,
wStream* s,
362 DWORD flags, proxyData* pdata,
363 pServerDynamicChannelContext* dynChannel,
364 DynChannelContext* dynChannelContext,
368 const char* name = Stream_ConstPointer(s);
369 const size_t nameLen = Stream_GetRemainingLength(s);
370 const size_t len = strnlen(name, nameLen);
371 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
372 const BYTE cmd = CREATE_REQUEST_PDU;
374 if ((len == 0) || (len == nameLen) || (dynChannelId > UINT16_MAX))
376 char namebuffer[64] = { 0 };
377 (void)_snprintf(namebuffer,
sizeof(namebuffer) - 1,
"%s", name);
379 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
380 "channel id %" PRIu64
", name=%s [%" PRIuz
"|%" PRIuz
"], status=%s",
381 dynChannelId, namebuffer, len, nameLen,
382 dynChannel ? openstatus2str(dynChannel->openStatus) :
"NULL");
383 return PF_CHANNEL_RESULT_ERROR;
386 wStream* currentPacket = channelTracker_getCurrentPacket(tracker);
387 dev.channel_id = (UINT16)dynChannelId;
388 dev.channel_name = name;
389 dev.data = Stream_Buffer(s);
390 dev.data_len = Stream_GetPosition(currentPacket);
392 dev.total_size = Stream_GetPosition(currentPacket);
396 DynvcTrackerLog(dynChannelContext->log, WLOG_WARN, dynChannel, cmd, isBackData,
397 "Reusing channel id, now %s", name);
399 HashTable_Remove(dynChannelContext->channels, &dynChannel->channelId);
402 if (!pf_modules_run_filter(pdata->module, FILTER_TYPE_CLIENT_PASSTHROUGH_DYN_CHANNEL_CREATE,
404 return PF_CHANNEL_RESULT_DROP;
407 DynamicChannelContext_new(dynChannelContext->log, pdata->ps, name, (UINT32)dynChannelId);
410 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
411 "unable to create dynamic channel context data");
412 return PF_CHANNEL_RESULT_ERROR;
415 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
417 if (!HashTable_Insert(dynChannelContext->channels, &dynChannel->channelId, dynChannel))
419 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
420 "unable register dynamic channel context data");
421 DynamicChannelContext_free(dynChannel);
422 return PF_CHANNEL_RESULT_ERROR;
425 dynChannel->openStatus = CHANNEL_OPENSTATE_WAITING_OPEN_STATUS;
427 const BOOL firstPacket = (flags & CHANNEL_FLAG_FIRST) ? TRUE : FALSE;
428 const BOOL lastPacket = (flags & CHANNEL_FLAG_LAST) ? TRUE : FALSE;
431 return channelTracker_flushCurrent(tracker, firstPacket, lastPacket, FALSE);
434static PfChannelResult DynvcTrackerHandleCreateFront(ChannelStateTracker* tracker,
wStream* s,
436 WINPR_ATTR_UNUSED proxyData* pdata,
437 pServerDynamicChannelContext* dynChannel,
438 DynChannelContext* dynChannelContext,
439 WINPR_ATTR_UNUSED UINT64 dynChannelId)
441 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
442 const BYTE cmd = CREATE_REQUEST_PDU;
445 if (!Stream_CheckAndLogRequiredLengthWLogWithBackend(dynChannelContext->log, s, 4, FALSE))
446 return PF_CHANNEL_RESULT_ERROR;
448 const UINT32 creationStatus = Stream_Get_UINT32(s);
449 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
450 "CREATE_RESPONSE openStatus=%" PRIu32, creationStatus);
452 if (dynChannel && (creationStatus == 0))
453 dynChannel->openStatus = CHANNEL_OPENSTATE_OPENED;
455 const BOOL firstPacket = (flags & CHANNEL_FLAG_FIRST) ? TRUE : FALSE;
456 const BOOL lastPacket = (flags & CHANNEL_FLAG_LAST) ? TRUE : FALSE;
458 return channelTracker_flushCurrent(tracker, firstPacket, lastPacket, TRUE);
461static PfChannelResult DynvcTrackerHandleCreate(ChannelStateTracker* tracker,
wStream* s,
463 pServerDynamicChannelContext* dynChannel,
466 WINPR_ASSERT(tracker);
469 DynChannelContext* dynChannelContext =
470 (DynChannelContext*)channelTracker_getCustomData(tracker);
471 WINPR_ASSERT(dynChannelContext);
473 const BOOL lastPacket = (flags & CHANNEL_FLAG_LAST) ? TRUE : FALSE;
474 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
476 proxyData* pdata = channelTracker_getPData(tracker);
481 return PF_CHANNEL_RESULT_DROP;
484 return DynvcTrackerHandleCreateBack(tracker, s, flags, pdata, dynChannel, dynChannelContext,
487 return DynvcTrackerHandleCreateFront(tracker, s, flags, pdata, dynChannel, dynChannelContext,
491static PfChannelResult DynvcTrackerHandleCmdDATA(ChannelStateTracker* tracker,
492 pServerDynamicChannelContext* dynChannel,
493 wStream* s, BYTE cmd, UINT64 Length,
494 BOOL firstPacket, BOOL lastPacket)
496 WINPR_ASSERT(tracker);
499 DynChannelContext* dynChannelContext =
500 (DynChannelContext*)channelTracker_getCustomData(tracker);
501 WINPR_ASSERT(dynChannelContext);
503 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
507 DynvcTrackerLog(dynChannelContext->log, WLOG_WARN, dynChannel, cmd, isBackData,
508 "channel is NULL, dropping packet");
509 return PF_CHANNEL_RESULT_DROP;
512 DynChannelTrackerState* trackerState =
513 isBackData ? &dynChannel->backTracker : &dynChannel->frontTracker;
514 if (dynChannel->openStatus != CHANNEL_OPENSTATE_OPENED)
516 DynvcTrackerLog(dynChannelContext->log, WLOG_WARN, dynChannel, cmd, isBackData,
517 "channel is not opened, dropping packet");
518 return PF_CHANNEL_RESULT_DROP;
524 case DATA_FIRST_COMPRESSED_PDU:
526 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
527 "DATA_FIRST currentPacketLength=%" PRIu64
"", Length);
528 if (Length > UINT32_MAX)
530 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
531 "Length out of bounds: %" PRIu64, Length);
532 return PF_CHANNEL_RESULT_ERROR;
534 trackerState->currentDataLength = (UINT32)Length;
535 trackerState->CurrentDataReceived = 0;
536 trackerState->CurrentDataFragments = 0;
538 if (dynChannel->packetReassembly)
540 if (trackerState->currentPacket)
541 Stream_SetPosition(trackerState->currentPacket, 0);
554 size_t extraSize = Stream_GetRemainingLength(s);
556 trackerState->CurrentDataFragments++;
557 trackerState->CurrentDataReceived += WINPR_ASSERTING_INT_CAST(uint32_t, extraSize);
559 if (dynChannel->packetReassembly)
561 if (!trackerState->currentPacket)
563 trackerState->currentPacket = Stream_New(NULL, 1024);
564 if (!trackerState->currentPacket)
566 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd,
567 isBackData,
"unable to create current packet",
568 getDirection(isBackData), dynChannel->channelName,
569 drdynvc_get_packet_type(cmd));
570 return PF_CHANNEL_RESULT_ERROR;
574 if (!Stream_EnsureRemainingCapacity(trackerState->currentPacket, extraSize))
576 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
577 "unable to grow current packet", getDirection(isBackData),
578 dynChannel->channelName, drdynvc_get_packet_type(cmd));
579 return PF_CHANNEL_RESULT_ERROR;
582 Stream_Write(trackerState->currentPacket, Stream_ConstPointer(s), extraSize);
584 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
585 "frags=%" PRIu32
" received=%" PRIu32
"(%" PRIu32
")",
586 trackerState->CurrentDataFragments, trackerState->CurrentDataReceived,
587 trackerState->currentDataLength);
598 if (trackerState->currentDataLength)
600 if (trackerState->CurrentDataReceived > trackerState->currentDataLength)
602 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
603 "reassembled packet (%" PRIu32
604 ") is bigger than announced length (%" PRIu32
")",
605 trackerState->CurrentDataReceived,
606 trackerState->currentDataLength);
607 return PF_CHANNEL_RESULT_ERROR;
612 trackerState->CurrentDataFragments = 0;
613 trackerState->CurrentDataReceived = 0;
621 return DynvcTrackerPeekHandleByMode(tracker, trackerState, dynChannel, cmd, firstPacket,
625static PfChannelResult DynvcTrackerHandleCmd(ChannelStateTracker* tracker,
626 pServerDynamicChannelContext* dynChannel,
wStream* s,
627 BYTE cmd, UINT32 flags, UINT64 Length,
628 UINT64 dynChannelId, BOOL firstPacket, BOOL lastPacket)
630 WINPR_ASSERT(tracker);
633 DynChannelContext* dynChannelContext =
634 (DynChannelContext*)channelTracker_getCustomData(tracker);
635 WINPR_ASSERT(dynChannelContext);
637 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
640 case CAPABILITY_REQUEST_PDU:
641 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
642 "CAPABILITY_%s", isBackData ?
"REQUEST" :
"RESPONSE");
643 channelTracker_setMode(tracker, CHANNEL_TRACKER_PASS);
644 return PF_CHANNEL_RESULT_PASS;
646 case CREATE_REQUEST_PDU:
647 return DynvcTrackerHandleCreate(tracker, s, flags, dynChannel, dynChannelId);
649 case CLOSE_REQUEST_PDU:
650 return DynvcTrackerHandleClose(tracker, dynChannel, dynChannelContext, firstPacket,
653 case SOFT_SYNC_REQUEST_PDU:
655 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
656 "SOFT_SYNC_REQUEST_PDU");
657 channelTracker_setMode(tracker, CHANNEL_TRACKER_PASS);
659 return PF_CHANNEL_RESULT_PASS;
661 case SOFT_SYNC_RESPONSE_PDU:
663 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
664 "SOFT_SYNC_RESPONSE_PDU");
665 channelTracker_setMode(tracker, CHANNEL_TRACKER_PASS);
666 return PF_CHANNEL_RESULT_PASS;
670 return DynvcTrackerHandleCmdDATA(tracker, dynChannel, s, cmd, Length, firstPacket,
673 case DATA_FIRST_COMPRESSED_PDU:
674 case DATA_COMPRESSED_PDU:
675 DynvcTrackerLog(dynChannelContext->log, WLOG_DEBUG, dynChannel, cmd, isBackData,
676 "TODO: compressed data packets, pass them as is for now");
677 channelTracker_setMode(tracker, CHANNEL_TRACKER_PASS);
678 return channelTracker_flushCurrent(tracker, firstPacket, lastPacket, !isBackData);
681 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
682 "Invalid command ID");
683 return PF_CHANNEL_RESULT_ERROR;
687static PfChannelResult DynvcTrackerPeekFn(ChannelStateTracker* tracker, BOOL firstPacket,
692 BOOL haveChannelId = 0;
694 UINT64 dynChannelId = 0;
696 pServerDynamicChannelContext* dynChannel = NULL;
698 WINPR_ASSERT(tracker);
700 DynChannelContext* dynChannelContext =
701 (DynChannelContext*)channelTracker_getCustomData(tracker);
702 WINPR_ASSERT(dynChannelContext);
704 const BOOL isBackData = (tracker == dynChannelContext->backTracker);
706 UINT32 flags = lastPacket ? CHANNEL_FLAG_LAST : 0;
708 flags |= CHANNEL_FLAG_FIRST;
709 proxyData* pdata = channelTracker_getPData(tracker);
713 wStream* currentPacket = channelTracker_getCurrentPacket(tracker);
714 s = Stream_StaticConstInit(&sbuffer, Stream_Buffer(currentPacket),
715 Stream_GetPosition(currentPacket));
718 if (!Stream_CheckAndLogRequiredLengthWLogWithBackend(dynChannelContext->log, s, 1, isBackData))
719 return PF_CHANNEL_RESULT_ERROR;
721 const BYTE byte0 = Stream_Get_UINT8(s);
722 const BYTE cmd = byte0 >> 4;
726 case CREATE_REQUEST_PDU:
727 case CLOSE_REQUEST_PDU:
729 case DATA_COMPRESSED_PDU:
730 haveChannelId = TRUE;
734 case DATA_FIRST_COMPRESSED_PDU:
736 haveChannelId = TRUE;
739 haveChannelId = FALSE;
746 BYTE cbId = byte0 & 0x03;
748 switch (dynvc_read_varInt(dynChannelContext->log, s, cbId, &dynChannelId, lastPacket))
752 case DYNCVC_READ_INCOMPLETE:
753 return PF_CHANNEL_RESULT_DROP;
754 case DYNCVC_READ_ERROR:
756 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
757 "invalid channelId field");
758 return PF_CHANNEL_RESULT_ERROR;
764 dynChannel = (pServerDynamicChannelContext*)HashTable_GetItemValue(
765 dynChannelContext->channels, &dynChannelId);
766 if ((cmd != CREATE_REQUEST_PDU) || !isBackData)
768 if (!dynChannel || (dynChannel->openStatus == CHANNEL_OPENSTATE_CLOSED))
772 channelTracker_setMode(tracker, CHANNEL_TRACKER_DROP);
773 return PF_CHANNEL_RESULT_DROP;
780 BYTE lenLen = (byte0 >> 2) & 0x03;
781 switch (dynvc_read_varInt(dynChannelContext->log, s, lenLen, &Length, lastPacket))
785 case DYNCVC_READ_INCOMPLETE:
786 return PF_CHANNEL_RESULT_DROP;
787 case DYNCVC_READ_ERROR:
789 DynvcTrackerLog(dynChannelContext->log, WLOG_ERROR, dynChannel, cmd, isBackData,
790 "invalid length field");
791 return PF_CHANNEL_RESULT_ERROR;
795 return DynvcTrackerHandleCmd(tracker, dynChannel, s, cmd, flags, Length, dynChannelId,
796 firstPacket, lastPacket);
799static void DynChannelContext_free(
void* context)
801 DynChannelContext* c = context;
804 channelTracker_free(c->backTracker);
805 channelTracker_free(c->frontTracker);
806 HashTable_Free(c->channels);
810static const char* dynamic_context(
void* arg)
812 proxyData* pdata = arg;
815 return pdata->session_id;
818static DynChannelContext* DynChannelContext_new(proxyData* pdata,
819 pServerStaticChannelContext* channel)
821 DynChannelContext* dyn = calloc(1,
sizeof(DynChannelContext));
825 dyn->log = WLog_Get(DTAG);
826 WINPR_ASSERT(dyn->log);
827 WLog_SetContext(dyn->log, dynamic_context, pdata);
829 dyn->backTracker = channelTracker_new(channel, DynvcTrackerPeekFn, dyn);
830 if (!dyn->backTracker)
832 if (!channelTracker_setPData(dyn->backTracker, pdata))
835 dyn->frontTracker = channelTracker_new(channel, DynvcTrackerPeekFn, dyn);
836 if (!dyn->frontTracker)
838 if (!channelTracker_setPData(dyn->frontTracker, pdata))
841 dyn->channels = HashTable_New(FALSE);
845 if (!HashTable_SetHashFunction(dyn->channels, ChannelId_Hash))
848 wObject* kobj = HashTable_KeyObject(dyn->channels);
850 kobj->fnObjectEquals = ChannelId_Compare;
852 wObject* vobj = HashTable_ValueObject(dyn->channels);
854 vobj->fnObjectFree = DynamicChannelContext_free;
859 DynChannelContext_free(dyn);
863static PfChannelResult pf_dynvc_back_data(proxyData* pdata,
864 const pServerStaticChannelContext* channel,
865 const BYTE* xdata,
size_t xsize, UINT32 flags,
868 WINPR_ASSERT(channel);
870 DynChannelContext* dyn = (DynChannelContext*)channel->context;
874 return channelTracker_update(dyn->backTracker, xdata, xsize, flags, totalSize);
877static PfChannelResult pf_dynvc_front_data(proxyData* pdata,
878 const pServerStaticChannelContext* channel,
879 const BYTE* xdata,
size_t xsize, UINT32 flags,
882 WINPR_ASSERT(channel);
884 DynChannelContext* dyn = (DynChannelContext*)channel->context;
888 return channelTracker_update(dyn->frontTracker, xdata, xsize, flags, totalSize);
891BOOL pf_channel_setup_drdynvc(proxyData* pdata, pServerStaticChannelContext* channel)
893 DynChannelContext* ret = DynChannelContext_new(pdata, channel);
897 channel->onBackData = pf_dynvc_back_data;
898 channel->onFrontData = pf_dynvc_front_data;
899 channel->contextDtor = DynChannelContext_free;
900 channel->context = ret;
This struct contains function pointer to initialize/free objects.