/** * FreeRDP: A Remote Desktop Protocol Implementation * Dynamic Virtual Channel * * Copyright 2010-2011 Vic Lee * Copyright 2015 Thincast Technologies GmbH * Copyright 2015 DI (FH) Martin Haimberger * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include #include #include #include #include #include #include "drdynvc_main.h" #define TAG CHANNELS_TAG("drdynvc.client") static void dvcman_channel_free(DVCMAN_CHANNEL* channel); static UINT dvcman_channel_close(DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn); static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr); static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId, const BYTE* data, UINT32 dataSize, BOOL* close); static UINT drdynvc_send(drdynvcPlugin* drdynvc, wStream* s); static void dvcman_wtslistener_free(DVCMAN_LISTENER* listener) { if (listener) free(listener->channel_name); free(listener); } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_get_configuration(IWTSListener* pListener, void** ppPropertyBag) { WINPR_ASSERT(ppPropertyBag); WINPR_UNUSED(pListener); *ppPropertyBag = NULL; return ERROR_INTERNAL_ERROR; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_create_listener(IWTSVirtualChannelManager* pChannelMgr, const char* pszChannelName, ULONG ulFlags, IWTSListenerCallback* pListenerCallback, IWTSListener** ppListener) { DVCMAN* dvcman = (DVCMAN*)pChannelMgr; DVCMAN_LISTENER* listener = NULL; WINPR_ASSERT(dvcman); WLog_DBG(TAG, "create_listener: %" PRIuz ".%s.", HashTable_Count(dvcman->listeners) + 1, pszChannelName); listener = (DVCMAN_LISTENER*)calloc(1, sizeof(DVCMAN_LISTENER)); if (!listener) { WLog_ERR(TAG, "calloc failed!"); return CHANNEL_RC_NO_MEMORY; } listener->iface.GetConfiguration = dvcman_get_configuration; listener->iface.pInterface = NULL; listener->dvcman = dvcman; listener->channel_name = _strdup(pszChannelName); if (!listener->channel_name) { WLog_ERR(TAG, "_strdup failed!"); dvcman_wtslistener_free(listener); return CHANNEL_RC_NO_MEMORY; } listener->flags = ulFlags; listener->listener_callback = pListenerCallback; if (ppListener) *ppListener = (IWTSListener*)listener; if (!HashTable_Insert(dvcman->listeners, listener->channel_name, listener)) { dvcman_wtslistener_free(listener); return ERROR_INTERNAL_ERROR; } // NOLINTNEXTLINE(clang-analyzer-unix.Malloc): HashTable_Insert takes ownership of listener return CHANNEL_RC_OK; } static UINT dvcman_destroy_listener(IWTSVirtualChannelManager* pChannelMgr, IWTSListener* pListener) { DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)pListener; WINPR_UNUSED(pChannelMgr); if (listener) { DVCMAN* dvcman = listener->dvcman; if (dvcman) HashTable_Remove(dvcman->listeners, listener->channel_name); } return CHANNEL_RC_OK; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_register_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name, IWTSPlugin* pPlugin) { WINPR_ASSERT(pEntryPoints); DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman; WINPR_ASSERT(dvcman); if (!ArrayList_Append(dvcman->plugin_names, name)) return ERROR_INTERNAL_ERROR; if (!ArrayList_Append(dvcman->plugins, pPlugin)) return ERROR_INTERNAL_ERROR; WLog_DBG(TAG, "register_plugin: num_plugins %" PRIuz, ArrayList_Count(dvcman->plugins)); return CHANNEL_RC_OK; } static IWTSPlugin* dvcman_get_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name) { IWTSPlugin* plugin = NULL; size_t nc = 0; size_t pc = 0; WINPR_ASSERT(pEntryPoints); DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman; if (!dvcman || !pEntryPoints || !name) return NULL; nc = ArrayList_Count(dvcman->plugin_names); pc = ArrayList_Count(dvcman->plugins); if (nc != pc) return NULL; ArrayList_Lock(dvcman->plugin_names); ArrayList_Lock(dvcman->plugins); for (size_t i = 0; i < pc; i++) { const char* cur = ArrayList_GetItem(dvcman->plugin_names, i); if (strcmp(cur, name) == 0) { plugin = ArrayList_GetItem(dvcman->plugins, i); break; } } ArrayList_Unlock(dvcman->plugin_names); ArrayList_Unlock(dvcman->plugins); return plugin; } static const ADDIN_ARGV* dvcman_get_plugin_data(IDRDYNVC_ENTRY_POINTS* pEntryPoints) { WINPR_ASSERT(pEntryPoints); return ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->args; } static rdpContext* dvcman_get_rdp_context(IDRDYNVC_ENTRY_POINTS* pEntryPoints) { DVCMAN_ENTRY_POINTS* entry = (DVCMAN_ENTRY_POINTS*)pEntryPoints; WINPR_ASSERT(entry); return entry->context; } static rdpSettings* dvcman_get_rdp_settings(IDRDYNVC_ENTRY_POINTS* pEntryPoints) { rdpContext* context = dvcman_get_rdp_context(pEntryPoints); WINPR_ASSERT(context); return context->settings; } static UINT32 dvcman_get_channel_id(IWTSVirtualChannel* channel) { DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel; WINPR_ASSERT(dvc); return dvc->channel_id; } static const char* dvcman_get_channel_name(IWTSVirtualChannel* channel) { DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel; WINPR_ASSERT(dvc); return dvc->channel_name; } static DVCMAN_CHANNEL* dvcman_get_channel_by_id(IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId, BOOL doRef) { DVCMAN* dvcman = (DVCMAN*)pChannelMgr; DVCMAN_CHANNEL* dvcChannel = NULL; WINPR_ASSERT(dvcman); HashTable_Lock(dvcman->channelsById); dvcChannel = HashTable_GetItemValue(dvcman->channelsById, &ChannelId); if (dvcChannel) { if (doRef) InterlockedIncrement(&dvcChannel->refCounter); } HashTable_Unlock(dvcman->channelsById); return dvcChannel; } static IWTSVirtualChannel* dvcman_find_channel_by_id(IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId) { DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE); if (!channel) return NULL; return &channel->iface; } static void dvcman_plugin_terminate(void* plugin) { IWTSPlugin* pPlugin = plugin; WINPR_ASSERT(pPlugin); UINT error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Terminated, pPlugin); if (error != CHANNEL_RC_OK) WLog_ERR(TAG, "Terminated failed with error %" PRIu32 "!", error); } static void wts_listener_free(void* arg) { DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)arg; dvcman_wtslistener_free(listener); } static BOOL channelIdMatch(const void* k1, const void* k2) { WINPR_ASSERT(k1); WINPR_ASSERT(k2); return *((const UINT32*)k1) == *((const UINT32*)k2); } static UINT32 channelIdHash(const void* id) { WINPR_ASSERT(id); return *((const UINT32*)id); } static void channelByIdCleanerFn(void* value) { DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)value; if (channel) { dvcman_channel_close(channel, FALSE, TRUE); dvcman_channel_free(channel); } } static IWTSVirtualChannelManager* dvcman_new(drdynvcPlugin* plugin) { wObject* obj = NULL; DVCMAN* dvcman = (DVCMAN*)calloc(1, sizeof(DVCMAN)); if (!dvcman) return NULL; dvcman->iface.CreateListener = dvcman_create_listener; dvcman->iface.DestroyListener = dvcman_destroy_listener; dvcman->iface.FindChannelById = dvcman_find_channel_by_id; dvcman->iface.GetChannelId = dvcman_get_channel_id; dvcman->iface.GetChannelName = dvcman_get_channel_name; dvcman->drdynvc = plugin; dvcman->channelsById = HashTable_New(TRUE); if (!dvcman->channelsById) goto fail; HashTable_SetHashFunction(dvcman->channelsById, channelIdHash); obj = HashTable_KeyObject(dvcman->channelsById); WINPR_ASSERT(obj); obj->fnObjectEquals = channelIdMatch; obj = HashTable_ValueObject(dvcman->channelsById); WINPR_ASSERT(obj); obj->fnObjectFree = channelByIdCleanerFn; dvcman->pool = StreamPool_New(TRUE, 10); if (!dvcman->pool) goto fail; dvcman->listeners = HashTable_New(TRUE); if (!dvcman->listeners) goto fail; HashTable_SetHashFunction(dvcman->listeners, HashTable_StringHash); obj = HashTable_KeyObject(dvcman->listeners); obj->fnObjectEquals = HashTable_StringCompare; obj = HashTable_ValueObject(dvcman->listeners); obj->fnObjectFree = wts_listener_free; dvcman->plugin_names = ArrayList_New(TRUE); if (!dvcman->plugin_names) goto fail; obj = ArrayList_Object(dvcman->plugin_names); obj->fnObjectNew = winpr_ObjectStringClone; obj->fnObjectFree = winpr_ObjectStringFree; dvcman->plugins = ArrayList_New(TRUE); if (!dvcman->plugins) goto fail; obj = ArrayList_Object(dvcman->plugins); obj->fnObjectFree = dvcman_plugin_terminate; return &dvcman->iface; fail: dvcman_free(plugin, &dvcman->iface); return NULL; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_load_addin(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr, const ADDIN_ARGV* args, rdpContext* context) { PDVC_PLUGIN_ENTRY pDVCPluginEntry = NULL; WINPR_ASSERT(drdynvc); WINPR_ASSERT(pChannelMgr); WINPR_ASSERT(args); WINPR_ASSERT(context); WLog_Print(drdynvc->log, WLOG_INFO, "Loading Dynamic Virtual Channel %s", args->argv[0]); pDVCPluginEntry = (PDVC_PLUGIN_ENTRY)freerdp_load_channel_addin_entry( args->argv[0], NULL, NULL, FREERDP_ADDIN_CHANNEL_DYNAMIC); if (pDVCPluginEntry) { DVCMAN_ENTRY_POINTS entryPoints = { 0 }; entryPoints.iface.RegisterPlugin = dvcman_register_plugin; entryPoints.iface.GetPlugin = dvcman_get_plugin; entryPoints.iface.GetPluginData = dvcman_get_plugin_data; entryPoints.iface.GetRdpSettings = dvcman_get_rdp_settings; entryPoints.iface.GetRdpContext = dvcman_get_rdp_context; entryPoints.dvcman = (DVCMAN*)pChannelMgr; entryPoints.args = args; entryPoints.context = context; return pDVCPluginEntry(&entryPoints.iface); } return ERROR_INVALID_FUNCTION; } static void dvcman_channel_free(DVCMAN_CHANNEL* channel) { if (!channel) return; if (channel->dvc_data) Stream_Release(channel->dvc_data); DeleteCriticalSection(&(channel->lock)); free(channel->channel_name); free(channel); } static void dvcman_channel_unref(DVCMAN_CHANNEL* channel) { WINPR_ASSERT(channel); if (InterlockedDecrement(&channel->refCounter)) return; DVCMAN* dvcman = channel->dvcman; if (dvcman) HashTable_Remove(dvcman->channelsById, &channel->channel_id); } static UINT dvcchannel_send_close(DVCMAN_CHANNEL* channel) { WINPR_ASSERT(channel); DVCMAN* dvcman = channel->dvcman; drdynvcPlugin* drdynvc = dvcman->drdynvc; wStream* s = StreamPool_Take(dvcman->pool, 5); if (!s) { WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } Stream_Write_UINT8(s, (CLOSE_REQUEST_PDU << 4) | 0x02); Stream_Write_UINT32(s, channel->channel_id); return drdynvc_send(drdynvc, s); } static void check_open_close_receive(DVCMAN_CHANNEL* channel) { WINPR_ASSERT(channel); IWTSVirtualChannelCallback* cb = channel->channel_callback; const char* name = channel->channel_name; const UINT32 id = channel->channel_id; WINPR_ASSERT(cb); if (cb->OnOpen || cb->OnClose) { if (!cb->OnOpen || !cb->OnClose) WLog_WARN(TAG, "{%s:%" PRIu32 "} OnOpen=%p, OnClose=%p", name, id, cb->OnOpen, cb->OnClose); } } static UINT dvcman_call_on_receive(DVCMAN_CHANNEL* channel, wStream* data) { WINPR_ASSERT(channel); WINPR_ASSERT(data); IWTSVirtualChannelCallback* cb = channel->channel_callback; WINPR_ASSERT(cb); check_open_close_receive(channel); WINPR_ASSERT(cb->OnDataReceived); return cb->OnDataReceived(cb, data); } static UINT dvcman_channel_close(DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn) { UINT error = CHANNEL_RC_OK; DrdynvcClientContext* context = NULL; WINPR_ASSERT(channel); switch (channel->state) { case DVC_CHANNEL_INIT: break; case DVC_CHANNEL_RUNNING: if (channel->dvcman) { drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; WINPR_ASSERT(drdynvc); context = drdynvc->context; if (perRequest) WLog_Print(drdynvc->log, WLOG_DEBUG, "sending close confirm for '%s'", channel->channel_name); error = dvcchannel_send_close(channel); if (error != CHANNEL_RC_OK) { const char* msg = "error when sending close confirm for '%s'"; if (perRequest) msg = "error when sending closeRequest for '%s'"; WLog_Print(drdynvc->log, WLOG_DEBUG, msg, channel->channel_name); } } channel->state = DVC_CHANNEL_CLOSED; IWTSVirtualChannelCallback* cb = channel->channel_callback; if (cb) { check_open_close_receive(channel); IFCALL(cb->OnClose, cb); } channel->channel_callback = NULL; if (channel->dvcman && channel->dvcman->drdynvc) { if (context) { IFCALLRET(context->OnChannelDisconnected, error, context, channel->channel_name, channel->pInterface); } } if (!fromHashTableFn) dvcman_channel_unref(channel); break; case DVC_CHANNEL_CLOSED: break; } return error; } static DVCMAN_CHANNEL* dvcman_channel_new(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId, const char* ChannelName) { DVCMAN_CHANNEL* channel = NULL; WINPR_ASSERT(drdynvc); WINPR_ASSERT(pChannelMgr); channel = (DVCMAN_CHANNEL*)calloc(1, sizeof(DVCMAN_CHANNEL)); if (!channel) return NULL; channel->dvcman = (DVCMAN*)pChannelMgr; channel->channel_id = ChannelId; channel->refCounter = 1; channel->state = DVC_CHANNEL_INIT; channel->channel_name = _strdup(ChannelName); if (!channel->channel_name) goto fail; if (!InitializeCriticalSectionEx(&(channel->lock), 0, 0)) goto fail; return channel; fail: dvcman_channel_free(channel); return NULL; } static void dvcman_clear(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) { DVCMAN* dvcman = (DVCMAN*)pChannelMgr; WINPR_ASSERT(dvcman); WINPR_UNUSED(drdynvc); HashTable_Clear(dvcman->channelsById); ArrayList_Clear(dvcman->plugins); ArrayList_Clear(dvcman->plugin_names); HashTable_Clear(dvcman->listeners); } static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) { DVCMAN* dvcman = (DVCMAN*)pChannelMgr; WINPR_ASSERT(dvcman); WINPR_UNUSED(drdynvc); HashTable_Free(dvcman->channelsById); ArrayList_Free(dvcman->plugins); ArrayList_Free(dvcman->plugin_names); HashTable_Free(dvcman->listeners); StreamPool_Free(dvcman->pool); free(dvcman); } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_init(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) { DVCMAN* dvcman = (DVCMAN*)pChannelMgr; UINT error = CHANNEL_RC_OK; WINPR_ASSERT(dvcman); ArrayList_Lock(dvcman->plugins); for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) { IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Initialize, pPlugin, pChannelMgr); if (error != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "Initialize failed with error %" PRIu32 "!", error); goto fail; } } fail: ArrayList_Unlock(dvcman->plugins); return error; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_write_channel(IWTSVirtualChannel* pChannel, ULONG cbSize, const BYTE* pBuffer, void* pReserved) { BOOL close = FALSE; UINT status = 0; DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel; WINPR_UNUSED(pReserved); if (!channel || !channel->dvcman) return CHANNEL_RC_BAD_CHANNEL; EnterCriticalSection(&(channel->lock)); status = drdynvc_write_data(channel->dvcman->drdynvc, channel->channel_id, pBuffer, cbSize, &close); LeaveCriticalSection(&(channel->lock)); /* Close delayed, it removes the channel struct */ if (close) dvcman_channel_close(channel, FALSE, FALSE); return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_close_channel_iface(IWTSVirtualChannel* pChannel) { DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel; if (!channel) return CHANNEL_RC_BAD_CHANNEL; WLog_DBG(TAG, "close_channel_iface: id=%" PRIu32 "", channel->channel_id); return dvcman_channel_close(channel, FALSE, FALSE); } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static DVCMAN_CHANNEL* dvcman_create_channel(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId, const char* ChannelName, UINT* res) { BOOL bAccept = 0; DVCMAN_CHANNEL* channel = NULL; DrdynvcClientContext* context = NULL; DVCMAN* dvcman = (DVCMAN*)pChannelMgr; DVCMAN_LISTENER* listener = NULL; IWTSVirtualChannelCallback* pCallback = NULL; WINPR_ASSERT(dvcman); WINPR_ASSERT(res); HashTable_Lock(dvcman->listeners); listener = (DVCMAN_LISTENER*)HashTable_GetItemValue(dvcman->listeners, ChannelName); if (!listener) { *res = ERROR_NOT_FOUND; goto out; } channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE); if (channel) { switch (channel->state) { case DVC_CHANNEL_RUNNING: WLog_Print(drdynvc->log, WLOG_ERROR, "Protocol error: Duplicated ChannelId %" PRIu32 " (%s)!", ChannelId, ChannelName); *res = CHANNEL_RC_ALREADY_OPEN; goto out; case DVC_CHANNEL_CLOSED: case DVC_CHANNEL_INIT: default: WLog_Print(drdynvc->log, WLOG_ERROR, "not expecting a createChannel from state %d", channel->state); *res = CHANNEL_RC_INITIALIZATION_ERROR; goto out; } } else { if (!(channel = dvcman_channel_new(drdynvc, pChannelMgr, ChannelId, ChannelName))) { WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_channel_new failed!"); *res = CHANNEL_RC_NO_MEMORY; goto out; } } if (!HashTable_Insert(dvcman->channelsById, &channel->channel_id, channel)) { WLog_Print(drdynvc->log, WLOG_ERROR, "unable to register channel in our channel list"); *res = ERROR_INTERNAL_ERROR; dvcman_channel_free(channel); channel = NULL; goto out; } channel->iface.Write = dvcman_write_channel; channel->iface.Close = dvcman_close_channel_iface; bAccept = TRUE; *res = listener->listener_callback->OnNewChannelConnection( listener->listener_callback, &channel->iface, NULL, &bAccept, &pCallback); if (*res != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "OnNewChannelConnection failed with error %" PRIu32 "!", *res); *res = ERROR_INTERNAL_ERROR; dvcman_channel_unref(channel); goto out; } if (!bAccept) { WLog_Print(drdynvc->log, WLOG_ERROR, "OnNewChannelConnection returned with bAccept FALSE!"); *res = ERROR_INTERNAL_ERROR; dvcman_channel_unref(channel); channel = NULL; goto out; } WLog_Print(drdynvc->log, WLOG_DEBUG, "listener %s created new channel %" PRIu32 "", listener->channel_name, channel->channel_id); channel->state = DVC_CHANNEL_RUNNING; channel->channel_callback = pCallback; channel->pInterface = listener->iface.pInterface; context = dvcman->drdynvc->context; IFCALLRET(context->OnChannelConnected, *res, context, ChannelName, listener->iface.pInterface); if (*res != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "context.OnChannelConnected failed with error %" PRIu32 "", *res); } out: HashTable_Unlock(dvcman->listeners); return channel; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_open_channel(drdynvcPlugin* drdynvc, DVCMAN_CHANNEL* channel) { IWTSVirtualChannelCallback* pCallback = NULL; UINT error = CHANNEL_RC_OK; WINPR_ASSERT(drdynvc); WINPR_ASSERT(channel); if (channel->state == DVC_CHANNEL_RUNNING) { pCallback = channel->channel_callback; if (pCallback->OnOpen) { check_open_close_receive(channel); error = pCallback->OnOpen(pCallback); if (error) { WLog_Print(drdynvc->log, WLOG_ERROR, "OnOpen failed with error %" PRIu32 "!", error); goto out; } } WLog_Print(drdynvc->log, WLOG_DEBUG, "open_channel: ChannelId %" PRIu32 "", channel->channel_id); } out: return error; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_receive_channel_data_first(DVCMAN_CHANNEL* channel, UINT32 length) { WINPR_ASSERT(channel); WINPR_ASSERT(channel->dvcman); if (channel->dvc_data) Stream_Release(channel->dvc_data); channel->dvc_data = StreamPool_Take(channel->dvcman->pool, length); if (!channel->dvc_data) { drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } channel->dvc_data_length = length; return CHANNEL_RC_OK; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT dvcman_receive_channel_data(DVCMAN_CHANNEL* channel, wStream* data, UINT32 ThreadingFlags) { UINT status = CHANNEL_RC_OK; size_t dataSize = Stream_GetRemainingLength(data); WINPR_ASSERT(channel); WINPR_ASSERT(channel->dvcman); if (channel->dvc_data) { drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; /* Fragmented data */ if (Stream_GetPosition(channel->dvc_data) + dataSize > channel->dvc_data_length) { WLog_Print(drdynvc->log, WLOG_ERROR, "data exceeding declared length!"); Stream_Release(channel->dvc_data); channel->dvc_data = NULL; status = ERROR_INVALID_DATA; goto out; } Stream_Copy(data, channel->dvc_data, dataSize); if (Stream_GetPosition(channel->dvc_data) >= channel->dvc_data_length) { Stream_SealLength(channel->dvc_data); Stream_SetPosition(channel->dvc_data, 0); status = dvcman_call_on_receive(channel, channel->dvc_data); Stream_Release(channel->dvc_data); channel->dvc_data = NULL; } } else status = dvcman_call_on_receive(channel, data); out: return status; } static UINT8 drdynvc_write_variable_uint(wStream* s, UINT32 val) { UINT8 cb = 0; if (val <= 0xFF) { cb = 0; Stream_Write_UINT8(s, (UINT8)val); } else if (val <= 0xFFFF) { cb = 1; Stream_Write_UINT16(s, (UINT16)val); } else { cb = 2; Stream_Write_UINT32(s, val); } return cb; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_send(drdynvcPlugin* drdynvc, wStream* s) { UINT status = 0; if (!drdynvc) status = CHANNEL_RC_BAD_CHANNEL_HANDLE; else { WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelWriteEx); status = drdynvc->channelEntryPoints.pVirtualChannelWriteEx( drdynvc->InitHandle, drdynvc->OpenHandle, Stream_Buffer(s), (UINT32)Stream_GetPosition(s), s); } switch (status) { case CHANNEL_RC_OK: return CHANNEL_RC_OK; case CHANNEL_RC_NOT_CONNECTED: Stream_Release(s); return CHANNEL_RC_OK; case CHANNEL_RC_BAD_CHANNEL_HANDLE: Stream_Release(s); WLog_ERR(TAG, "VirtualChannelWriteEx failed with CHANNEL_RC_BAD_CHANNEL_HANDLE"); return status; default: Stream_Release(s); WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); return status; } } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId, const BYTE* data, UINT32 dataSize, BOOL* close) { wStream* data_out = NULL; size_t pos = 0; UINT8 cbChId = 0; UINT8 cbLen = 0; unsigned long chunkLength = 0; UINT status = CHANNEL_RC_BAD_INIT_HANDLE; DVCMAN* dvcman = NULL; if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; dvcman = (DVCMAN*)drdynvc->channel_mgr; WINPR_ASSERT(dvcman); WLog_Print(drdynvc->log, WLOG_TRACE, "write_data: ChannelId=%" PRIu32 " size=%" PRIu32 "", ChannelId, dataSize); data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH); if (!data_out) { WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } Stream_SetPosition(data_out, 1); cbChId = drdynvc_write_variable_uint(data_out, ChannelId); pos = Stream_GetPosition(data_out); if (dataSize == 0) { /* TODO: shall treat that case with write(0) that do a close */ *close = TRUE; Stream_Release(data_out); } else if (dataSize <= CHANNEL_CHUNK_LENGTH - pos) { Stream_SetPosition(data_out, 0); Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId); Stream_SetPosition(data_out, pos); Stream_Write(data_out, data, dataSize); status = drdynvc_send(drdynvc, data_out); } else { /* Fragment the data */ cbLen = drdynvc_write_variable_uint(data_out, dataSize); pos = Stream_GetPosition(data_out); Stream_SetPosition(data_out, 0); Stream_Write_UINT8(data_out, (DATA_FIRST_PDU << 4) | cbChId | (cbLen << 2)); Stream_SetPosition(data_out, pos); chunkLength = CHANNEL_CHUNK_LENGTH - pos; Stream_Write(data_out, data, chunkLength); data += chunkLength; dataSize -= chunkLength; status = drdynvc_send(drdynvc, data_out); while (status == CHANNEL_RC_OK && dataSize > 0) { data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH); if (!data_out) { WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } Stream_SetPosition(data_out, 1); cbChId = drdynvc_write_variable_uint(data_out, ChannelId); pos = Stream_GetPosition(data_out); Stream_SetPosition(data_out, 0); Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId); Stream_SetPosition(data_out, pos); chunkLength = dataSize; if (chunkLength > CHANNEL_CHUNK_LENGTH - pos) chunkLength = CHANNEL_CHUNK_LENGTH - pos; Stream_Write(data_out, data, chunkLength); data += chunkLength; dataSize -= chunkLength; status = drdynvc_send(drdynvc, data_out); } } if (status != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); return status; } return CHANNEL_RC_OK; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_send_capability_response(drdynvcPlugin* drdynvc) { UINT status = 0; wStream* s = NULL; DVCMAN* dvcman = NULL; if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; dvcman = (DVCMAN*)drdynvc->channel_mgr; WINPR_ASSERT(dvcman); WLog_Print(drdynvc->log, WLOG_TRACE, "capability_response"); s = StreamPool_Take(dvcman->pool, 4); if (!s) { WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_Ndrdynvc_write_variable_uintew failed!"); return CHANNEL_RC_NO_MEMORY; } Stream_Write_UINT16(s, 0x0050); /* Cmd+Sp+cbChId+Pad. Note: MSTSC sends 0x005c */ Stream_Write_UINT16(s, drdynvc->version); status = drdynvc_send(drdynvc, s); if (status != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); } return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_process_capability_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s) { UINT status = 0; if (!drdynvc) return CHANNEL_RC_BAD_INIT_HANDLE; if (!Stream_CheckAndLogRequiredLength(TAG, s, 3)) return ERROR_INVALID_DATA; WLog_Print(drdynvc->log, WLOG_TRACE, "capability_request Sp=%d cbChId=%d", Sp, cbChId); Stream_Seek(s, 1); /* pad */ Stream_Read_UINT16(s, drdynvc->version); /* RDP8 servers offer version 3, though Microsoft forgot to document it * in their early documents. It behaves the same as version 2. */ if ((drdynvc->version == 2) || (drdynvc->version == 3)) { if (!Stream_CheckAndLogRequiredLength(TAG, s, 8)) return ERROR_INVALID_DATA; Stream_Read_UINT16(s, drdynvc->PriorityCharge0); Stream_Read_UINT16(s, drdynvc->PriorityCharge1); Stream_Read_UINT16(s, drdynvc->PriorityCharge2); Stream_Read_UINT16(s, drdynvc->PriorityCharge3); } status = drdynvc_send_capability_response(drdynvc); drdynvc->state = DRDYNVC_STATE_READY; return status; } static UINT32 drdynvc_cblen_to_bytes(int cbLen) { switch (cbLen) { case 0: return 1; case 1: return 2; default: return 4; } } static UINT32 drdynvc_read_variable_uint(wStream* s, int cbLen) { UINT32 val = 0; switch (cbLen) { case 0: Stream_Read_UINT8(s, val); break; case 1: Stream_Read_UINT16(s, val); break; default: Stream_Read_UINT32(s, val); break; } return val; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_process_create_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s) { UINT status = 0; wStream* data_out = NULL; UINT channel_status = 0; DVCMAN* dvcman = NULL; DVCMAN_CHANNEL* channel = NULL; UINT32 retStatus = 0; WINPR_UNUSED(Sp); if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; dvcman = (DVCMAN*)drdynvc->channel_mgr; WINPR_ASSERT(dvcman); if (drdynvc->state == DRDYNVC_STATE_CAPABILITIES) { /** * For some reason the server does not always send the * capabilities pdu as it should. When this happens, * send a capabilities response. */ drdynvc->version = 3; if ((status = drdynvc_send_capability_response(drdynvc))) { WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_send_capability_response failed!"); return status; } drdynvc->state = DRDYNVC_STATE_READY; } if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) return ERROR_INVALID_DATA; const UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId); const size_t pos = Stream_GetPosition(s); const char* name = Stream_ConstPointer(s); const size_t length = Stream_GetRemainingLength(s); if (strnlen(name, length) >= length) return ERROR_INVALID_DATA; WLog_Print(drdynvc->log, WLOG_DEBUG, "process_create_request: ChannelId=%" PRIu32 " ChannelName=%s", ChannelId, name); data_out = StreamPool_Take(dvcman->pool, pos + 4); if (!data_out) { WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } Stream_Write_UINT8(data_out, (CREATE_REQUEST_PDU << 4) | cbChId); Stream_SetPosition(s, 1); Stream_Copy(s, data_out, pos - 1); channel = dvcman_create_channel(drdynvc, drdynvc->channel_mgr, ChannelId, name, &channel_status); switch (channel_status) { case CHANNEL_RC_OK: WLog_Print(drdynvc->log, WLOG_DEBUG, "channel created"); retStatus = 0; break; case CHANNEL_RC_NO_MEMORY: WLog_Print(drdynvc->log, WLOG_DEBUG, "not enough memory for channel creation"); retStatus = STATUS_NO_MEMORY; break; case ERROR_NOT_FOUND: WLog_Print(drdynvc->log, WLOG_DEBUG, "no listener for '%s'", name); retStatus = (UINT32)0xC0000001; /* same code used by mstsc, STATUS_UNSUCCESSFUL */ break; default: WLog_Print(drdynvc->log, WLOG_DEBUG, "channel creation error"); retStatus = (UINT32)0xC0000001; /* same code used by mstsc, STATUS_UNSUCCESSFUL */ break; } Stream_Write_UINT32(data_out, retStatus); status = drdynvc_send(drdynvc, data_out); if (status != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); dvcman_channel_unref(channel); return status; } if (channel_status == CHANNEL_RC_OK) { if ((status = dvcman_open_channel(drdynvc, channel))) { WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_open_channel failed with error %" PRIu32 "!", status); return status; } } return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_process_data_first(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s, UINT32 ThreadingFlags) { UINT status = CHANNEL_RC_OK; UINT32 Length = 0; UINT32 ChannelId = 0; DVCMAN_CHANNEL* channel = NULL; WINPR_ASSERT(drdynvc); if (!Stream_CheckAndLogRequiredLength( TAG, s, drdynvc_cblen_to_bytes(cbChId) + drdynvc_cblen_to_bytes(Sp))) return ERROR_INVALID_DATA; ChannelId = drdynvc_read_variable_uint(s, cbChId); Length = drdynvc_read_variable_uint(s, Sp); WLog_Print(drdynvc->log, WLOG_TRACE, "process_data_first: Sp=%d cbChId=%d, ChannelId=%" PRIu32 " Length=%" PRIu32 "", Sp, cbChId, ChannelId, Length); channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); if (!channel) { /** * Windows Server 2012 R2 can send some messages over * Microsoft::Windows::RDS::Geometry::v08.01 even if the dynamic virtual channel wasn't * registered on our side. Ignoring it works. */ WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId); return CHANNEL_RC_OK; } if (channel->state != DVC_CHANNEL_RUNNING) goto out; status = dvcman_receive_channel_data_first(channel, Length); if (status == CHANNEL_RC_OK) status = dvcman_receive_channel_data(channel, s, ThreadingFlags); if (status != CHANNEL_RC_OK) status = dvcman_channel_close(channel, FALSE, FALSE); out: dvcman_channel_unref(channel); return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_process_data(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s, UINT32 ThreadingFlags) { UINT32 ChannelId = 0; DVCMAN_CHANNEL* channel = NULL; UINT status = CHANNEL_RC_OK; WINPR_ASSERT(drdynvc); if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) return ERROR_INVALID_DATA; ChannelId = drdynvc_read_variable_uint(s, cbChId); WLog_Print(drdynvc->log, WLOG_TRACE, "process_data: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp, cbChId, ChannelId); channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); if (!channel) { /** * Windows Server 2012 R2 can send some messages over * Microsoft::Windows::RDS::Geometry::v08.01 even if the dynamic virtual channel wasn't * registered on our side. Ignoring it works. */ WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId); return CHANNEL_RC_OK; } if (channel->state != DVC_CHANNEL_RUNNING) goto out; status = dvcman_receive_channel_data(channel, s, ThreadingFlags); if (status != CHANNEL_RC_OK) status = dvcman_channel_close(channel, FALSE, FALSE); out: dvcman_channel_unref(channel); return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_process_close_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s) { UINT32 ChannelId = 0; DVCMAN_CHANNEL* channel = NULL; WINPR_ASSERT(drdynvc); if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) return ERROR_INVALID_DATA; ChannelId = drdynvc_read_variable_uint(s, cbChId); WLog_Print(drdynvc->log, WLOG_DEBUG, "process_close_request: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp, cbChId, ChannelId); channel = (DVCMAN_CHANNEL*)dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); if (!channel) { WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_close_request channel %" PRIu32 " not present", ChannelId); return CHANNEL_RC_OK; } dvcman_channel_close(channel, TRUE, FALSE); dvcman_channel_unref(channel); return CHANNEL_RC_OK; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_order_recv(drdynvcPlugin* drdynvc, wStream* s, UINT32 ThreadingFlags) { UINT8 value = 0; WINPR_ASSERT(drdynvc); if (!Stream_CheckAndLogRequiredLength(TAG, s, 1)) return ERROR_INVALID_DATA; Stream_Read_UINT8(s, value); const UINT8 Cmd = (value & 0xf0) >> 4; const UINT8 Sp = (value & 0x0c) >> 2; const UINT8 cbChId = (value & 0x03) >> 0; WLog_Print(drdynvc->log, WLOG_TRACE, "order_recv: Cmd=%s, Sp=%" PRIu8 " cbChId=%" PRIu8, drdynvc_get_packet_type(Cmd), Sp, cbChId); switch (Cmd) { case CAPABILITY_REQUEST_PDU: return drdynvc_process_capability_request(drdynvc, Sp, cbChId, s); case CREATE_REQUEST_PDU: return drdynvc_process_create_request(drdynvc, Sp, cbChId, s); case DATA_FIRST_PDU: return drdynvc_process_data_first(drdynvc, Sp, cbChId, s, ThreadingFlags); case DATA_PDU: return drdynvc_process_data(drdynvc, Sp, cbChId, s, ThreadingFlags); case CLOSE_REQUEST_PDU: return drdynvc_process_close_request(drdynvc, Sp, cbChId, s); default: WLog_Print(drdynvc->log, WLOG_ERROR, "unknown drdynvc cmd 0x%x", Cmd); return ERROR_INTERNAL_ERROR; } } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_virtual_channel_event_data_received(drdynvcPlugin* drdynvc, void* pData, UINT32 dataLength, UINT32 totalLength, UINT32 dataFlags) { wStream* data_in = NULL; WINPR_ASSERT(drdynvc); if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME)) { return CHANNEL_RC_OK; } if (dataFlags & CHANNEL_FLAG_FIRST) { DVCMAN* mgr = (DVCMAN*)drdynvc->channel_mgr; if (drdynvc->data_in) Stream_Release(drdynvc->data_in); drdynvc->data_in = StreamPool_Take(mgr->pool, totalLength); } if (!(data_in = drdynvc->data_in)) { WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); return CHANNEL_RC_NO_MEMORY; } if (!Stream_EnsureRemainingCapacity(data_in, dataLength)) { WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_EnsureRemainingCapacity failed!"); Stream_Release(drdynvc->data_in); drdynvc->data_in = NULL; return ERROR_INTERNAL_ERROR; } Stream_Write(data_in, pData, dataLength); if (dataFlags & CHANNEL_FLAG_LAST) { const size_t cap = Stream_Capacity(data_in); const size_t pos = Stream_GetPosition(data_in); if (cap < pos) { WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_plugin_process_received: read error"); return ERROR_INVALID_DATA; } drdynvc->data_in = NULL; Stream_SealLength(data_in); Stream_SetPosition(data_in, 0); if (drdynvc->async) { if (!MessageQueue_Post(drdynvc->queue, NULL, 0, (void*)data_in, NULL)) { WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Post failed!"); return ERROR_INTERNAL_ERROR; } } else { UINT error = drdynvc_order_recv(drdynvc, data_in, TRUE); Stream_Release(data_in); if (error) { WLog_Print(drdynvc->log, WLOG_WARN, "drdynvc_order_recv failed with error %" PRIu32 "!", error); } } } return CHANNEL_RC_OK; } static void VCAPITYPE drdynvc_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle, UINT event, LPVOID pData, UINT32 dataLength, UINT32 totalLength, UINT32 dataFlags) { UINT error = CHANNEL_RC_OK; drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam; WINPR_ASSERT(drdynvc); switch (event) { case CHANNEL_EVENT_DATA_RECEIVED: if (!drdynvc || (drdynvc->OpenHandle != openHandle)) { WLog_ERR(TAG, "drdynvc_virtual_channel_open_event: error no match"); return; } if ((error = drdynvc_virtual_channel_event_data_received(drdynvc, pData, dataLength, totalLength, dataFlags))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_data_received failed with error %" PRIu32 "", error); break; case CHANNEL_EVENT_WRITE_CANCELLED: case CHANNEL_EVENT_WRITE_COMPLETE: { wStream* s = (wStream*)pData; Stream_Release(s); } break; case CHANNEL_EVENT_USER: break; } if (error && drdynvc && drdynvc->rdpcontext) setChannelError(drdynvc->rdpcontext, error, "drdynvc_virtual_channel_open_event reported an error"); } static DWORD WINAPI drdynvc_virtual_channel_client_thread(LPVOID arg) { /* TODO: rewrite this */ wStream* data = NULL; wMessage message = { 0 }; UINT error = CHANNEL_RC_OK; drdynvcPlugin* drdynvc = (drdynvcPlugin*)arg; if (!drdynvc) { ExitThread((DWORD)CHANNEL_RC_BAD_CHANNEL_HANDLE); return CHANNEL_RC_BAD_CHANNEL_HANDLE; } while (1) { if (!MessageQueue_Wait(drdynvc->queue)) { WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Wait failed!"); error = ERROR_INTERNAL_ERROR; break; } if (!MessageQueue_Peek(drdynvc->queue, &message, TRUE)) { WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Peek failed!"); error = ERROR_INTERNAL_ERROR; break; } if (message.id == WMQ_QUIT) break; if (message.id == 0) { UINT32 ThreadingFlags = TRUE; data = (wStream*)message.wParam; if ((error = drdynvc_order_recv(drdynvc, data, ThreadingFlags))) { WLog_Print(drdynvc->log, WLOG_WARN, "drdynvc_order_recv failed with error %" PRIu32 "!", error); } Stream_Release(data); } } { /* Disconnect remaining dynamic channels that the server did not. * This is required to properly shut down channels by calling the appropriate * event handlers. */ DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr; HashTable_Clear(drdynvcMgr->channelsById); } if (error && drdynvc->rdpcontext) setChannelError(drdynvc->rdpcontext, error, "drdynvc_virtual_channel_client_thread reported an error"); ExitThread((DWORD)error); return error; } static void drdynvc_queue_object_free(void* obj) { wStream* s = NULL; wMessage* msg = (wMessage*)obj; if (!msg || (msg->id != 0)) return; s = (wStream*)msg->wParam; if (s) Stream_Release(s); } static UINT drdynvc_virtual_channel_event_initialized(drdynvcPlugin* drdynvc, LPVOID pData, UINT32 dataLength) { wObject* obj = NULL; WINPR_UNUSED(pData); WINPR_UNUSED(dataLength); if (!drdynvc) goto error; drdynvc->queue = MessageQueue_New(NULL); if (!drdynvc->queue) { WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_New failed!"); goto error; } obj = MessageQueue_Object(drdynvc->queue); obj->fnObjectFree = drdynvc_queue_object_free; drdynvc->channel_mgr = dvcman_new(drdynvc); if (!drdynvc->channel_mgr) { WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_new failed!"); goto error; } return CHANNEL_RC_OK; error: return ERROR_INTERNAL_ERROR; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_virtual_channel_event_connected(drdynvcPlugin* drdynvc, LPVOID pData, UINT32 dataLength) { UINT error = 0; UINT32 status = 0; rdpSettings* settings = NULL; WINPR_ASSERT(drdynvc); WINPR_UNUSED(pData); WINPR_UNUSED(dataLength); if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelOpenEx); status = drdynvc->channelEntryPoints.pVirtualChannelOpenEx( drdynvc->InitHandle, &drdynvc->OpenHandle, drdynvc->channelDef.name, drdynvc_virtual_channel_open_event_ex); if (status != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelOpen failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); return status; } WINPR_ASSERT(drdynvc->rdpcontext); settings = drdynvc->rdpcontext->settings; WINPR_ASSERT(settings); for (UINT32 index = 0; index < freerdp_settings_get_uint32(settings, FreeRDP_DynamicChannelCount); index++) { const ADDIN_ARGV* args = freerdp_settings_get_pointer_array(settings, FreeRDP_DynamicChannelArray, index); error = dvcman_load_addin(drdynvc, drdynvc->channel_mgr, args, drdynvc->rdpcontext); if (CHANNEL_RC_OK != error) goto error; } if ((error = dvcman_init(drdynvc, drdynvc->channel_mgr))) { WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_init failed with error %" PRIu32 "!", error); goto error; } drdynvc->state = DRDYNVC_STATE_CAPABILITIES; if (drdynvc->async) { if (!(drdynvc->thread = CreateThread(NULL, 0, drdynvc_virtual_channel_client_thread, (void*)drdynvc, 0, NULL))) { error = ERROR_INTERNAL_ERROR; WLog_Print(drdynvc->log, WLOG_ERROR, "CreateThread failed!"); goto error; } } error: return error; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_virtual_channel_event_disconnected(drdynvcPlugin* drdynvc) { UINT status = 0; if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; if (drdynvc->OpenHandle == 0) return CHANNEL_RC_OK; if (drdynvc->queue) { if (!MessageQueue_PostQuit(drdynvc->queue, 0)) { status = GetLastError(); WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_PostQuit failed with error %" PRIu32 "", status); return status; } } if (drdynvc->thread) { if (WaitForSingleObject(drdynvc->thread, INFINITE) != WAIT_OBJECT_0) { status = GetLastError(); WLog_Print(drdynvc->log, WLOG_ERROR, "WaitForSingleObject failed with error %" PRIu32 "", status); return status; } CloseHandle(drdynvc->thread); drdynvc->thread = NULL; } else { { /* Disconnect remaining dynamic channels that the server did not. * This is required to properly shut down channels by calling the appropriate * event handlers. */ DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr; HashTable_Clear(drdynvcMgr->channelsById); } } WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelCloseEx); status = drdynvc->channelEntryPoints.pVirtualChannelCloseEx(drdynvc->InitHandle, drdynvc->OpenHandle); if (status != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelClose failed with %s [%08" PRIX32 "]", WTSErrorToString(status), status); } dvcman_clear(drdynvc, drdynvc->channel_mgr); if (drdynvc->queue) MessageQueue_Clear(drdynvc->queue); drdynvc->OpenHandle = 0; if (drdynvc->data_in) { Stream_Release(drdynvc->data_in); drdynvc->data_in = NULL; } return status; } /** * Function description * * @return 0 on success, otherwise a Win32 error code */ static UINT drdynvc_virtual_channel_event_terminated(drdynvcPlugin* drdynvc) { if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; MessageQueue_Free(drdynvc->queue); drdynvc->queue = NULL; if (drdynvc->channel_mgr) { dvcman_free(drdynvc, drdynvc->channel_mgr); drdynvc->channel_mgr = NULL; } drdynvc->InitHandle = 0; free(drdynvc->context); free(drdynvc); return CHANNEL_RC_OK; } static UINT drdynvc_virtual_channel_event_attached(drdynvcPlugin* drdynvc) { UINT error = CHANNEL_RC_OK; DVCMAN* dvcman = NULL; if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; dvcman = (DVCMAN*)drdynvc->channel_mgr; if (!dvcman) return CHANNEL_RC_BAD_CHANNEL_HANDLE; ArrayList_Lock(dvcman->plugins); for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) { IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Attached, pPlugin); if (error != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "Attach failed with error %" PRIu32 "!", error); goto fail; } } fail: ArrayList_Unlock(dvcman->plugins); return error; } static UINT drdynvc_virtual_channel_event_detached(drdynvcPlugin* drdynvc) { UINT error = CHANNEL_RC_OK; DVCMAN* dvcman = NULL; if (!drdynvc) return CHANNEL_RC_BAD_CHANNEL_HANDLE; dvcman = (DVCMAN*)drdynvc->channel_mgr; if (!dvcman) return CHANNEL_RC_BAD_CHANNEL_HANDLE; ArrayList_Lock(dvcman->plugins); for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) { IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Detached, pPlugin); if (error != CHANNEL_RC_OK) { WLog_Print(drdynvc->log, WLOG_ERROR, "Detach failed with error %" PRIu32 "!", error); goto fail; } } fail: ArrayList_Unlock(dvcman->plugins); return error; } static VOID VCAPITYPE drdynvc_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle, UINT event, LPVOID pData, UINT dataLength) { UINT error = CHANNEL_RC_OK; drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam; if (!drdynvc || (drdynvc->InitHandle != pInitHandle)) { WLog_ERR(TAG, "drdynvc_virtual_channel_init_event: error no match"); return; } switch (event) { case CHANNEL_EVENT_INITIALIZED: error = drdynvc_virtual_channel_event_initialized(drdynvc, pData, dataLength); break; case CHANNEL_EVENT_CONNECTED: if ((error = drdynvc_virtual_channel_event_connected(drdynvc, pData, dataLength))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_connected failed with error %" PRIu32 "", error); break; case CHANNEL_EVENT_DISCONNECTED: if ((error = drdynvc_virtual_channel_event_disconnected(drdynvc))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_disconnected failed with error %" PRIu32 "", error); break; case CHANNEL_EVENT_TERMINATED: if ((error = drdynvc_virtual_channel_event_terminated(drdynvc))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_terminated failed with error %" PRIu32 "", error); break; case CHANNEL_EVENT_ATTACHED: if ((error = drdynvc_virtual_channel_event_attached(drdynvc))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_attached failed with error %" PRIu32 "", error); break; case CHANNEL_EVENT_DETACHED: if ((error = drdynvc_virtual_channel_event_detached(drdynvc))) WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_virtual_channel_event_detached failed with error %" PRIu32 "", error); break; default: break; } if (error && drdynvc->rdpcontext) setChannelError(drdynvc->rdpcontext, error, "drdynvc_virtual_channel_init_event_ex reported an error"); } /** * Channel Client Interface */ static int drdynvc_get_version(DrdynvcClientContext* context) { WINPR_ASSERT(context); drdynvcPlugin* drdynvc = (drdynvcPlugin*)context->handle; WINPR_ASSERT(drdynvc); return drdynvc->version; } /* drdynvc is always built-in */ #define VirtualChannelEntryEx drdynvc_VirtualChannelEntryEx FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS_EX pEntryPoints, PVOID pInitHandle)) { UINT rc = 0; drdynvcPlugin* drdynvc = NULL; DrdynvcClientContext* context = NULL; CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx = NULL; drdynvc = (drdynvcPlugin*)calloc(1, sizeof(drdynvcPlugin)); WINPR_ASSERT(pEntryPoints); if (!drdynvc) { WLog_ERR(TAG, "calloc failed!"); return FALSE; } drdynvc->channelDef.options = CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP | CHANNEL_OPTION_COMPRESS_RDP; sprintf_s(drdynvc->channelDef.name, ARRAYSIZE(drdynvc->channelDef.name), DRDYNVC_SVC_CHANNEL_NAME); drdynvc->state = DRDYNVC_STATE_INITIAL; pEntryPointsEx = (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints; if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) && (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER)) { context = (DrdynvcClientContext*)calloc(1, sizeof(DrdynvcClientContext)); if (!context) { WLog_Print(drdynvc->log, WLOG_ERROR, "calloc failed!"); free(drdynvc); return FALSE; } context->handle = (void*)drdynvc; context->custom = NULL; drdynvc->context = context; context->GetVersion = drdynvc_get_version; drdynvc->rdpcontext = pEntryPointsEx->context; if (!freerdp_settings_get_bool(drdynvc->rdpcontext->settings, FreeRDP_TransportDumpReplay) && !freerdp_settings_get_bool(drdynvc->rdpcontext->settings, FreeRDP_SynchronousDynamicChannels)) drdynvc->async = TRUE; } drdynvc->log = WLog_Get(TAG); WLog_Print(drdynvc->log, WLOG_DEBUG, "VirtualChannelEntryEx"); CopyMemory(&(drdynvc->channelEntryPoints), pEntryPoints, sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)); drdynvc->InitHandle = pInitHandle; WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelInitEx); rc = drdynvc->channelEntryPoints.pVirtualChannelInitEx( drdynvc, context, pInitHandle, &drdynvc->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000, drdynvc_virtual_channel_init_event_ex); if (CHANNEL_RC_OK != rc) { WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelInit failed with %s [%08" PRIX32 "]", WTSErrorToString(rc), rc); free(drdynvc->context); free(drdynvc); return FALSE; } drdynvc->channelEntryPoints.pInterface = context; return TRUE; }