/src/FreeRDP/channels/drdynvc/client/drdynvc_main.c
Line | Count | Source |
1 | | /** |
2 | | * FreeRDP: A Remote Desktop Protocol Implementation |
3 | | * Dynamic Virtual Channel |
4 | | * |
5 | | * Copyright 2010-2011 Vic Lee |
6 | | * Copyright 2015 Thincast Technologies GmbH |
7 | | * Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com> |
8 | | * |
9 | | * Licensed under the Apache License, Version 2.0 (the "License"); |
10 | | * you may not use this file except in compliance with the License. |
11 | | * You may obtain a copy of the License at |
12 | | * |
13 | | * http://www.apache.org/licenses/LICENSE-2.0 |
14 | | * |
15 | | * Unless required by applicable law or agreed to in writing, software |
16 | | * distributed under the License is distributed on an "AS IS" BASIS, |
17 | | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
18 | | * See the License for the specific language governing permissions and |
19 | | * limitations under the License. |
20 | | */ |
21 | | |
22 | | #include <freerdp/config.h> |
23 | | |
24 | | #include <winpr/crt.h> |
25 | | #include <winpr/cast.h> |
26 | | #include <winpr/stream.h> |
27 | | #include <winpr/interlocked.h> |
28 | | |
29 | | #include <freerdp/freerdp.h> |
30 | | #include <freerdp/multitransport.h> |
31 | | #if defined(WITH_RDPEUDP) |
32 | | #include <freerdp/channels/channels.h> |
33 | | #endif |
34 | | #include <freerdp/channels/drdynvc.h> |
35 | | #include <freerdp/utils/drdynvc.h> |
36 | | #include <freerdp/codec/zgfx.h> |
37 | | |
38 | | #include "drdynvc_main.h" |
39 | | |
40 | 2.14k | #define TAG CHANNELS_TAG("drdynvc.client") |
41 | | |
42 | | /* drdynvc message queue ids */ |
43 | | typedef enum WINPR_C23_ENUM_TYPE(uint32_t) |
44 | | { |
45 | | DRDYNVC_MSG_DATA = 0, |
46 | | DRDYNVC_MSG_TUNNEL_DATA = 1, |
47 | | DRDYNVC_MSG_TUNNEL_STATE = 2 |
48 | | } DRDYNVC_MSG_ID; |
49 | | |
50 | | /* Tunnel data held back until the Soft-Sync Request arrives, see drdynvc_receive_tunnel_pdu */ |
51 | 0 | #define DRDYNVC_MAX_TUNNEL_BACKLOG 4096 |
52 | | |
53 | | /* Distinct tunnel types a Soft-Sync Response can name */ |
54 | 703 | #define DRDYNVC_MAX_SOFT_SYNC_TUNNELS 8 |
55 | | |
56 | | WINPR_ATTR_NODISCARD |
57 | | static const char* channel_state2str(DVC_CHANNEL_STATE state) |
58 | 0 | { |
59 | 0 | switch (state) |
60 | 0 | { |
61 | 0 | case DVC_CHANNEL_INIT: |
62 | 0 | return "DVC_CHANNEL_INIT"; |
63 | 0 | case DVC_CHANNEL_RUNNING: |
64 | 0 | return "DVC_CHANNEL_RUNNING"; |
65 | 0 | case DVC_CHANNEL_CLOSED: |
66 | 0 | return "DVC_CHANNEL_CLOSED"; |
67 | 0 | default: |
68 | 0 | return "DVC_CHANNEL_UNKNOWN"; |
69 | 0 | } |
70 | 0 | } |
71 | | |
72 | | static void dvcman_channel_free(DVCMAN_CHANNEL* channel); |
73 | | |
74 | | static UINT dvcman_channel_close(DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn); |
75 | | static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr); |
76 | | |
77 | | WINPR_ATTR_NODISCARD |
78 | | static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId, UINT32 tunnelType, |
79 | | const BYTE* data, UINT32 dataSize, BOOL* close, |
80 | | DVCMAN_CHANNEL_STATS* stats); |
81 | | |
82 | | WINPR_ATTR_NODISCARD |
83 | | static UINT drdynvc_send(drdynvcPlugin* drdynvc, wStream* s, DVCMAN_CHANNEL_STATS* stats); |
84 | | WINPR_ATTR_NODISCARD |
85 | | static UINT drdynvc_send_on(drdynvcPlugin* drdynvc, wStream* s, DVCMAN_CHANNEL_STATS* stats, |
86 | | UINT32* tunnelType, BOOL continuation); |
87 | | |
88 | | static void dvcman_wtslistener_free(DVCMAN_LISTENER* listener) |
89 | 4.13k | { |
90 | 4.13k | if (listener) |
91 | 2.06k | free(listener->channel_name); |
92 | 4.13k | free(listener); |
93 | 4.13k | } |
94 | | |
95 | | /** |
96 | | * Function description |
97 | | * |
98 | | * @return 0 on success, otherwise a Win32 error code |
99 | | */ |
100 | | WINPR_ATTR_NODISCARD |
101 | | static UINT dvcman_get_configuration(IWTSListener* pListener, void** ppPropertyBag) |
102 | 0 | { |
103 | 0 | WINPR_ASSERT(ppPropertyBag); |
104 | 0 | WINPR_UNUSED(pListener); |
105 | 0 | *ppPropertyBag = nullptr; |
106 | 0 | return ERROR_INTERNAL_ERROR; |
107 | 0 | } |
108 | | |
109 | | /** |
110 | | * Function description |
111 | | * |
112 | | * @return 0 on success, otherwise a Win32 error code |
113 | | */ |
114 | | WINPR_ATTR_NODISCARD |
115 | | static UINT dvcman_create_listener(IWTSVirtualChannelManager* pChannelMgr, |
116 | | const char* pszChannelName, ULONG ulFlags, |
117 | | IWTSListenerCallback* pListenerCallback, |
118 | | IWTSListener** ppListener) |
119 | 2.06k | { |
120 | 2.06k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
121 | 2.06k | DVCMAN_LISTENER* listener = nullptr; |
122 | | |
123 | 2.06k | WINPR_ASSERT(dvcman); |
124 | 2.06k | WLog_DBG(TAG, "create_listener: %" PRIuz ".%s.", HashTable_Count(dvcman->listeners) + 1, |
125 | 2.06k | pszChannelName); |
126 | 2.06k | listener = (DVCMAN_LISTENER*)calloc(1, sizeof(DVCMAN_LISTENER)); |
127 | | |
128 | 2.06k | if (!listener) |
129 | 0 | { |
130 | 0 | WLog_ERR(TAG, "calloc failed!"); |
131 | 0 | return CHANNEL_RC_NO_MEMORY; |
132 | 0 | } |
133 | | |
134 | 2.06k | listener->iface.GetConfiguration = dvcman_get_configuration; |
135 | 2.06k | listener->iface.pInterface = nullptr; |
136 | 2.06k | listener->dvcman = dvcman; |
137 | 2.06k | listener->channel_name = _strdup(pszChannelName); |
138 | | |
139 | 2.06k | if (!listener->channel_name) |
140 | 0 | { |
141 | 0 | WLog_ERR(TAG, "_strdup failed!"); |
142 | 0 | dvcman_wtslistener_free(listener); |
143 | 0 | return CHANNEL_RC_NO_MEMORY; |
144 | 0 | } |
145 | | |
146 | 2.06k | listener->flags = ulFlags; |
147 | 2.06k | listener->listener_callback = pListenerCallback; |
148 | | |
149 | 2.06k | if (ppListener) |
150 | 0 | *ppListener = (IWTSListener*)listener; |
151 | | |
152 | 2.06k | if (!HashTable_Insert(dvcman->listeners, listener->channel_name, listener)) |
153 | 0 | { |
154 | 0 | dvcman_wtslistener_free(listener); |
155 | 0 | return ERROR_INTERNAL_ERROR; |
156 | 0 | } |
157 | | |
158 | | // NOLINTNEXTLINE(clang-analyzer-unix.Malloc): HashTable_Insert takes ownership of listener |
159 | 2.06k | return CHANNEL_RC_OK; |
160 | 2.06k | } |
161 | | |
162 | | WINPR_ATTR_NODISCARD |
163 | | static UINT dvcman_destroy_listener(IWTSVirtualChannelManager* pChannelMgr, IWTSListener* pListener) |
164 | 0 | { |
165 | 0 | DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)pListener; |
166 | |
|
167 | 0 | WINPR_UNUSED(pChannelMgr); |
168 | |
|
169 | 0 | if (listener) |
170 | 0 | { |
171 | 0 | DVCMAN* dvcman = listener->dvcman; |
172 | 0 | if (dvcman) |
173 | 0 | HashTable_Remove(dvcman->listeners, listener->channel_name); |
174 | 0 | } |
175 | |
|
176 | 0 | return CHANNEL_RC_OK; |
177 | 0 | } |
178 | | |
179 | | /** |
180 | | * Function description |
181 | | * |
182 | | * @return 0 on success, otherwise a Win32 error code |
183 | | */ |
184 | | WINPR_ATTR_NODISCARD |
185 | | static UINT dvcman_register_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name, |
186 | | IWTSPlugin* pPlugin) |
187 | 0 | { |
188 | 0 | WINPR_ASSERT(pEntryPoints); |
189 | 0 | DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman; |
190 | |
|
191 | 0 | WINPR_ASSERT(dvcman); |
192 | 0 | if (!ArrayList_Append(dvcman->plugin_names, name)) |
193 | 0 | return ERROR_INTERNAL_ERROR; |
194 | 0 | if (!ArrayList_Append(dvcman->plugins, pPlugin)) |
195 | 0 | return ERROR_INTERNAL_ERROR; |
196 | | |
197 | 0 | WLog_DBG(TAG, "register_plugin: num_plugins %" PRIuz, ArrayList_Count(dvcman->plugins)); |
198 | 0 | return CHANNEL_RC_OK; |
199 | 0 | } |
200 | | |
201 | | WINPR_ATTR_NODISCARD |
202 | | static IWTSPlugin* dvcman_get_plugin(IDRDYNVC_ENTRY_POINTS* pEntryPoints, const char* name) |
203 | 0 | { |
204 | 0 | IWTSPlugin* plugin = nullptr; |
205 | 0 | size_t nc = 0; |
206 | 0 | size_t pc = 0; |
207 | 0 | WINPR_ASSERT(pEntryPoints); |
208 | 0 | DVCMAN* dvcman = ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->dvcman; |
209 | 0 | if (!dvcman || !pEntryPoints || !name) |
210 | 0 | return nullptr; |
211 | | |
212 | 0 | nc = ArrayList_Count(dvcman->plugin_names); |
213 | 0 | pc = ArrayList_Count(dvcman->plugins); |
214 | 0 | if (nc != pc) |
215 | 0 | return nullptr; |
216 | | |
217 | 0 | ArrayList_Lock(dvcman->plugin_names); |
218 | 0 | ArrayList_Lock(dvcman->plugins); |
219 | 0 | for (size_t i = 0; i < pc; i++) |
220 | 0 | { |
221 | 0 | const char* cur = ArrayList_GetItem(dvcman->plugin_names, i); |
222 | 0 | if (strcmp(cur, name) == 0) |
223 | 0 | { |
224 | 0 | plugin = ArrayList_GetItem(dvcman->plugins, i); |
225 | 0 | break; |
226 | 0 | } |
227 | 0 | } |
228 | 0 | ArrayList_Unlock(dvcman->plugin_names); |
229 | 0 | ArrayList_Unlock(dvcman->plugins); |
230 | 0 | return plugin; |
231 | 0 | } |
232 | | |
233 | | WINPR_ATTR_NODISCARD |
234 | | static const ADDIN_ARGV* dvcman_get_plugin_data(IDRDYNVC_ENTRY_POINTS* pEntryPoints) |
235 | 0 | { |
236 | 0 | WINPR_ASSERT(pEntryPoints); |
237 | 0 | return ((DVCMAN_ENTRY_POINTS*)pEntryPoints)->args; |
238 | 0 | } |
239 | | |
240 | | WINPR_ATTR_NODISCARD |
241 | | static rdpContext* dvcman_get_rdp_context(IDRDYNVC_ENTRY_POINTS* pEntryPoints) |
242 | 0 | { |
243 | 0 | DVCMAN_ENTRY_POINTS* entry = (DVCMAN_ENTRY_POINTS*)pEntryPoints; |
244 | 0 | WINPR_ASSERT(entry); |
245 | 0 | return entry->context; |
246 | 0 | } |
247 | | |
248 | | WINPR_ATTR_NODISCARD |
249 | | static rdpSettings* dvcman_get_rdp_settings(IDRDYNVC_ENTRY_POINTS* pEntryPoints) |
250 | 0 | { |
251 | 0 | rdpContext* context = dvcman_get_rdp_context(pEntryPoints); |
252 | 0 | WINPR_ASSERT(context); |
253 | |
|
254 | 0 | return context->settings; |
255 | 0 | } |
256 | | |
257 | | WINPR_ATTR_NODISCARD |
258 | | static UINT32 dvcman_get_channel_id(IWTSVirtualChannel* channel) |
259 | 0 | { |
260 | 0 | DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel; |
261 | 0 | WINPR_ASSERT(dvc); |
262 | 0 | return dvc->channel_id; |
263 | 0 | } |
264 | | |
265 | | WINPR_ATTR_NODISCARD |
266 | | static const char* dvcman_get_channel_name(IWTSVirtualChannel* channel) |
267 | 0 | { |
268 | 0 | DVCMAN_CHANNEL* dvc = (DVCMAN_CHANNEL*)channel; |
269 | 0 | WINPR_ASSERT(dvc); |
270 | 0 | return dvc->channel_name; |
271 | 0 | } |
272 | | |
273 | | WINPR_ATTR_NODISCARD |
274 | | static DVCMAN_CHANNEL* dvcman_get_channel_by_id(IWTSVirtualChannelManager* pChannelMgr, |
275 | | UINT32 ChannelId, BOOL doRef) |
276 | 6.35k | { |
277 | 6.35k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
278 | 6.35k | DVCMAN_CHANNEL* dvcChannel = nullptr; |
279 | | |
280 | 6.35k | WINPR_ASSERT(dvcman); |
281 | 6.35k | HashTable_Lock(dvcman->channelsById); |
282 | 6.35k | dvcChannel = HashTable_GetItemValue(dvcman->channelsById, &ChannelId); |
283 | 6.35k | if (dvcChannel) |
284 | 3.61k | { |
285 | 3.61k | if (doRef) |
286 | 2.58k | InterlockedIncrement(&dvcChannel->refCounter); |
287 | 3.61k | } |
288 | 2.73k | else |
289 | 2.73k | HashTable_Unlock(dvcman->channelsById); |
290 | | |
291 | 6.35k | return dvcChannel; |
292 | 6.35k | } |
293 | | |
294 | | static void dvcman_return_channel(IWTSVirtualChannelManager* pChannelMgr, DVCMAN_CHANNEL* channel) |
295 | 2.58k | { |
296 | 2.58k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
297 | | |
298 | 2.58k | if (!channel) |
299 | 0 | return; |
300 | | |
301 | 2.58k | WINPR_ASSERT(dvcman); |
302 | 2.58k | HashTable_Unlock(dvcman->channelsById); |
303 | 2.58k | } |
304 | | |
305 | | WINPR_ATTR_NODISCARD |
306 | | static IWTSVirtualChannel* dvcman_find_channel_by_id(IWTSVirtualChannelManager* pChannelMgr, |
307 | | UINT32 ChannelId) |
308 | 0 | { |
309 | 0 | DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE); |
310 | 0 | if (!channel) |
311 | 0 | return nullptr; |
312 | 0 | dvcman_return_channel(pChannelMgr, channel); |
313 | |
|
314 | 0 | return &channel->iface; |
315 | 0 | } |
316 | | |
317 | | static void dvcman_plugin_terminate(void* plugin) |
318 | 0 | { |
319 | 0 | IWTSPlugin* pPlugin = plugin; |
320 | |
|
321 | 0 | WINPR_ASSERT(pPlugin); |
322 | 0 | UINT error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Terminated, pPlugin); |
323 | 0 | if (error != CHANNEL_RC_OK) |
324 | 0 | WLog_ERR(TAG, "Terminated failed with error %" PRIu32 "!", error); |
325 | 0 | } |
326 | | |
327 | | static void wts_listener_free(void* arg) |
328 | 4.13k | { |
329 | 4.13k | DVCMAN_LISTENER* listener = (DVCMAN_LISTENER*)arg; |
330 | 4.13k | dvcman_wtslistener_free(listener); |
331 | 4.13k | } |
332 | | |
333 | | WINPR_ATTR_NODISCARD |
334 | | static BOOL channelIdMatch(const void* k1, const void* k2) |
335 | 7.95k | { |
336 | 7.95k | WINPR_ASSERT(k1); |
337 | 7.95k | WINPR_ASSERT(k2); |
338 | 7.95k | return *((const UINT32*)k1) == *((const UINT32*)k2); |
339 | 7.95k | } |
340 | | |
341 | | WINPR_ATTR_NODISCARD |
342 | | static UINT32 channelIdHash(const void* id) |
343 | 8.15k | { |
344 | 8.15k | WINPR_ASSERT(id); |
345 | 8.15k | return *((const UINT32*)id); |
346 | 8.15k | } |
347 | | |
348 | | static void channelByIdCleanerFn(void* value) |
349 | 2.88k | { |
350 | 2.88k | DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)value; |
351 | 2.88k | if (channel) |
352 | 1.44k | { |
353 | 1.44k | dvcman_channel_close(channel, FALSE, TRUE); |
354 | 1.44k | dvcman_channel_free(channel); |
355 | 1.44k | } |
356 | 2.88k | } |
357 | | |
358 | | WINPR_ATTR_MALLOC(dvcman_free, 2) |
359 | | static IWTSVirtualChannelManager* dvcman_new(drdynvcPlugin* plugin) |
360 | 2.14k | { |
361 | 2.14k | wObject* obj = nullptr; |
362 | 2.14k | DVCMAN* dvcman = (DVCMAN*)calloc(1, sizeof(DVCMAN)); |
363 | | |
364 | 2.14k | if (!dvcman) |
365 | 0 | return nullptr; |
366 | | |
367 | 2.14k | dvcman->iface.CreateListener = dvcman_create_listener; |
368 | 2.14k | dvcman->iface.DestroyListener = dvcman_destroy_listener; |
369 | 2.14k | dvcman->iface.FindChannelById = dvcman_find_channel_by_id; |
370 | 2.14k | dvcman->iface.GetChannelId = dvcman_get_channel_id; |
371 | 2.14k | dvcman->iface.GetChannelName = dvcman_get_channel_name; |
372 | 2.14k | dvcman->drdynvc = plugin; |
373 | 2.14k | dvcman->channelsById = HashTable_New(TRUE); |
374 | | |
375 | 2.14k | if (!dvcman->channelsById) |
376 | 0 | goto fail; |
377 | | |
378 | 2.14k | if (!HashTable_SetHashFunction(dvcman->channelsById, channelIdHash)) |
379 | 0 | goto fail; |
380 | | |
381 | 2.14k | obj = HashTable_KeyObject(dvcman->channelsById); |
382 | 2.14k | WINPR_ASSERT(obj); |
383 | 2.14k | obj->fnObjectEquals = channelIdMatch; |
384 | | |
385 | 2.14k | obj = HashTable_ValueObject(dvcman->channelsById); |
386 | 2.14k | WINPR_ASSERT(obj); |
387 | 2.14k | obj->fnObjectFree = channelByIdCleanerFn; |
388 | | |
389 | 2.14k | dvcman->pool = StreamPool_New(TRUE, 10); |
390 | 2.14k | if (!dvcman->pool) |
391 | 0 | goto fail; |
392 | | |
393 | 2.14k | dvcman->listeners = HashTable_New(TRUE); |
394 | 2.14k | if (!dvcman->listeners) |
395 | 0 | goto fail; |
396 | | |
397 | 2.14k | if (!HashTable_SetHashFunction(dvcman->listeners, HashTable_StringHash)) |
398 | 0 | goto fail; |
399 | | |
400 | 2.14k | obj = HashTable_KeyObject(dvcman->listeners); |
401 | 2.14k | obj->fnObjectEquals = HashTable_StringCompare; |
402 | | |
403 | 2.14k | obj = HashTable_ValueObject(dvcman->listeners); |
404 | 2.14k | obj->fnObjectFree = wts_listener_free; |
405 | | |
406 | 2.14k | dvcman->plugin_names = ArrayList_New(TRUE); |
407 | 2.14k | if (!dvcman->plugin_names) |
408 | 0 | goto fail; |
409 | 2.14k | obj = ArrayList_Object(dvcman->plugin_names); |
410 | 2.14k | obj->fnObjectNew = winpr_ObjectStringClone; |
411 | 2.14k | obj->fnObjectFree = winpr_ObjectStringFree; |
412 | | |
413 | 2.14k | dvcman->plugins = ArrayList_New(TRUE); |
414 | 2.14k | if (!dvcman->plugins) |
415 | 0 | goto fail; |
416 | 2.14k | obj = ArrayList_Object(dvcman->plugins); |
417 | 2.14k | obj->fnObjectFree = dvcman_plugin_terminate; |
418 | 2.14k | return &dvcman->iface; |
419 | 0 | fail: |
420 | 0 | dvcman_free(plugin, &dvcman->iface); |
421 | 0 | return nullptr; |
422 | 2.14k | } |
423 | | |
424 | | /** |
425 | | * Function description |
426 | | * |
427 | | * @return 0 on success, otherwise a Win32 error code |
428 | | */ |
429 | | WINPR_ATTR_NODISCARD |
430 | | static UINT dvcman_load_addin(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr, |
431 | | const ADDIN_ARGV* args, rdpContext* context) |
432 | 0 | { |
433 | 0 | WINPR_ASSERT(drdynvc); |
434 | 0 | WINPR_ASSERT(pChannelMgr); |
435 | 0 | WINPR_ASSERT(args); |
436 | 0 | WINPR_ASSERT(context); |
437 | |
|
438 | 0 | WLog_Print(drdynvc->log, WLOG_INFO, "Loading Dynamic Virtual Channel %s", args->argv[0]); |
439 | |
|
440 | 0 | PVIRTUALCHANNELENTRY pvce = freerdp_load_channel_addin_entry(args->argv[0], nullptr, nullptr, |
441 | 0 | FREERDP_ADDIN_CHANNEL_DYNAMIC); |
442 | 0 | PDVC_PLUGIN_ENTRY pDVCPluginEntry = WINPR_FUNC_PTR_CAST(pvce, PDVC_PLUGIN_ENTRY); |
443 | |
|
444 | 0 | if (pDVCPluginEntry) |
445 | 0 | { |
446 | 0 | DVCMAN_ENTRY_POINTS entryPoints = WINPR_C_ARRAY_INIT; |
447 | |
|
448 | 0 | entryPoints.iface.RegisterPlugin = dvcman_register_plugin; |
449 | 0 | entryPoints.iface.GetPlugin = dvcman_get_plugin; |
450 | 0 | entryPoints.iface.GetPluginData = dvcman_get_plugin_data; |
451 | 0 | entryPoints.iface.GetRdpSettings = dvcman_get_rdp_settings; |
452 | 0 | entryPoints.iface.GetRdpContext = dvcman_get_rdp_context; |
453 | 0 | entryPoints.dvcman = (DVCMAN*)pChannelMgr; |
454 | 0 | entryPoints.args = args; |
455 | 0 | entryPoints.context = context; |
456 | 0 | return pDVCPluginEntry(&entryPoints.iface); |
457 | 0 | } |
458 | | |
459 | 0 | return ERROR_INVALID_FUNCTION; |
460 | 0 | } |
461 | | |
462 | | static void dvcman_channel_free(DVCMAN_CHANNEL* channel) |
463 | 1.44k | { |
464 | 1.44k | if (!channel) |
465 | 0 | return; |
466 | | |
467 | 1.44k | if (channel->dvcman) |
468 | 1.44k | { |
469 | 1.44k | drdynvcPlugin* plugin = channel->dvcman->drdynvc; |
470 | 1.44k | if (plugin) |
471 | 1.44k | { |
472 | 1.44k | rdpContext* context = plugin->rdpcontext; |
473 | 1.44k | if (context) |
474 | 1.44k | { |
475 | 1.44k | ChannelTerminatedEventArgs e = WINPR_C_ARRAY_INIT; |
476 | 1.44k | EventArgsInit(&e, "freerdp"); |
477 | 1.44k | e.name = channel->channel_name; |
478 | 1.44k | e.pInterface = channel->pInterface; |
479 | | |
480 | 1.44k | const int rc = PubSub_OnChannelTerminated(context->pubSub, context, &e); |
481 | 1.44k | if (rc < 0) |
482 | 0 | WLog_WARN(TAG, "PubSub_OnChannelTerminated(%s) failed", channel->channel_name); |
483 | 1.44k | } |
484 | 1.44k | } |
485 | 1.44k | } |
486 | | |
487 | 1.44k | if (channel->dvc_data) |
488 | 85 | Stream_Release(channel->dvc_data); |
489 | | |
490 | 1.44k | zgfx_context_free(channel->decompressor); |
491 | 1.44k | DeleteCriticalSection(&(channel->lock)); |
492 | 1.44k | free(channel->channel_name); |
493 | 1.44k | free(channel); |
494 | 1.44k | } |
495 | | |
496 | | static void dvcman_channel_unref(DVCMAN_CHANNEL* channel) |
497 | 2.94k | { |
498 | 2.94k | WINPR_ASSERT(channel); |
499 | 2.94k | if (InterlockedDecrement(&channel->refCounter)) |
500 | 2.58k | return; |
501 | | |
502 | 359 | DVCMAN* dvcman = channel->dvcman; |
503 | 359 | if (dvcman) |
504 | 359 | HashTable_Remove(dvcman->channelsById, &channel->channel_id); |
505 | 359 | } |
506 | | |
507 | | WINPR_ATTR_NODISCARD |
508 | | static UINT32 dvcman_channel_get_tunnel(DVCMAN_CHANNEL* channel) |
509 | 1.44k | { |
510 | 1.44k | WINPR_ASSERT(channel); |
511 | 1.44k | EnterCriticalSection(&channel->lock); |
512 | 1.44k | const UINT32 tunnelType = channel->tunnelType; |
513 | 1.44k | LeaveCriticalSection(&channel->lock); |
514 | 1.44k | return tunnelType; |
515 | 1.44k | } |
516 | | |
517 | | /* Waits for a message the channel is writing to be out, see DVCMAN_CHANNEL::tunnelType. */ |
518 | | static void dvcman_channel_set_tunnel(DVCMAN_CHANNEL* channel, UINT32 tunnelType) |
519 | 1.44k | { |
520 | 1.44k | WINPR_ASSERT(channel); |
521 | 1.44k | EnterCriticalSection(&channel->lock); |
522 | 1.44k | channel->tunnelType = tunnelType; |
523 | 1.44k | LeaveCriticalSection(&channel->lock); |
524 | 1.44k | } |
525 | | |
526 | | WINPR_ATTR_NODISCARD |
527 | | static UINT dvcchannel_send_close(DVCMAN_CHANNEL* channel) |
528 | 1.44k | { |
529 | 1.44k | WINPR_ASSERT(channel); |
530 | 1.44k | DVCMAN* dvcman = channel->dvcman; |
531 | 1.44k | drdynvcPlugin* drdynvc = dvcman->drdynvc; |
532 | 1.44k | wStream* s = StreamPool_Take(dvcman->pool, 5); |
533 | | |
534 | 1.44k | if (!s) |
535 | 0 | { |
536 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
537 | 0 | return CHANNEL_RC_NO_MEMORY; |
538 | 0 | } |
539 | | |
540 | 1.44k | Stream_Write_UINT8(s, (CLOSE_REQUEST_PDU << 4) | 0x02); |
541 | 1.44k | Stream_Write_UINT32(s, channel->channel_id); |
542 | 1.44k | UINT32 tunnelType = dvcman_channel_get_tunnel(channel); |
543 | 1.44k | return drdynvc_send_on(drdynvc, s, &channel->stats, &tunnelType, FALSE); |
544 | 1.44k | } |
545 | | |
546 | | static void check_open_close_receive(DVCMAN_CHANNEL* channel) |
547 | 3.39k | { |
548 | 3.39k | WINPR_ASSERT(channel); |
549 | | |
550 | 3.39k | IWTSVirtualChannelCallback* cb = channel->channel_callback; |
551 | 3.39k | const char* name = channel->channel_name; |
552 | 3.39k | const UINT32 id = channel->channel_id; |
553 | | |
554 | 3.39k | WINPR_ASSERT(cb); |
555 | 3.39k | if (!cb->OnOpen || !cb->OnClose || !cb->OnDataReceived) |
556 | 0 | WLog_VRB(TAG, "{%s:%" PRIu32 "} OnOpen=%p, OnClose=%p, OnDataReceived=%p", name, id, |
557 | 3.39k | WINPR_FUNC_PTR_CAST(cb->OnOpen, const void*), |
558 | 3.39k | WINPR_FUNC_PTR_CAST(cb->OnClose, const void*), |
559 | 3.39k | WINPR_FUNC_PTR_CAST(cb->OnDataReceived, const void*)); |
560 | 3.39k | } |
561 | | |
562 | | WINPR_ATTR_NODISCARD |
563 | | static UINT dvcman_call_on_receive(DVCMAN_CHANNEL* channel, wStream* data) |
564 | 515 | { |
565 | 515 | WINPR_ASSERT(channel); |
566 | 515 | WINPR_ASSERT(data); |
567 | | |
568 | 515 | channel->stats.packetsIn++; |
569 | | |
570 | 515 | IWTSVirtualChannelCallback* cb = channel->channel_callback; |
571 | 515 | WINPR_ASSERT(cb); |
572 | | |
573 | 515 | check_open_close_receive(channel); |
574 | 515 | WINPR_ASSERT(cb->OnDataReceived); |
575 | 515 | return cb->OnDataReceived(cb, data); |
576 | 515 | } |
577 | | |
578 | | static UINT dvcman_channel_close(DVCMAN_CHANNEL* channel, BOOL perRequest, BOOL fromHashTableFn) |
579 | 1.80k | { |
580 | 1.80k | UINT error = CHANNEL_RC_OK; |
581 | 1.80k | DrdynvcClientContext* context = nullptr; |
582 | | |
583 | 1.80k | WINPR_ASSERT(channel); |
584 | 1.80k | switch (channel->state) |
585 | 1.80k | { |
586 | 0 | case DVC_CHANNEL_INIT: |
587 | 0 | break; |
588 | 1.44k | case DVC_CHANNEL_RUNNING: |
589 | 1.44k | if (channel->dvcman) |
590 | 1.44k | { |
591 | 1.44k | drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; |
592 | 1.44k | WINPR_ASSERT(drdynvc); |
593 | 1.44k | context = drdynvc->context; |
594 | 1.44k | if (perRequest) |
595 | 319 | WLog_Print(drdynvc->log, WLOG_DEBUG, "sending close confirm for '%s'", |
596 | 1.44k | channel->channel_name); |
597 | | |
598 | 1.44k | error = dvcchannel_send_close(channel); |
599 | 1.44k | if (error != CHANNEL_RC_OK) |
600 | 0 | { |
601 | 0 | if (perRequest) |
602 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, |
603 | 0 | "error when sending closeRequest for '%s'", |
604 | 0 | channel->channel_name); |
605 | 0 | else |
606 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, |
607 | 0 | "error when sending close confirm for '%s'", |
608 | 0 | channel->channel_name); |
609 | 0 | } |
610 | 1.44k | WLog_Print(drdynvc->log, WLOG_DEBUG, "listener %s destroyed channel %" PRIu32 "", |
611 | 1.44k | channel->channel_name, channel->channel_id); |
612 | 1.44k | } |
613 | | |
614 | 1.44k | channel->state = DVC_CHANNEL_CLOSED; |
615 | | |
616 | 1.44k | { |
617 | 1.44k | check_open_close_receive(channel); |
618 | | |
619 | 1.44k | IWTSVirtualChannelCallback* cb = channel->channel_callback; |
620 | 1.44k | channel->channel_callback = nullptr; |
621 | 1.44k | if (cb) |
622 | 1.44k | error = IFCALLRESULT(CHANNEL_RC_OK, cb->OnClose, cb); |
623 | 1.44k | } |
624 | | |
625 | | #if defined(WITH_RDPEUDP) |
626 | | /* not announced to the application, see dvcman_create_channel */ |
627 | | if (channel->staticBridge) |
628 | | context = nullptr; |
629 | | #endif |
630 | 1.44k | if (channel->dvcman && channel->dvcman->drdynvc) |
631 | 1.44k | { |
632 | 1.44k | if (context) |
633 | 1.44k | { |
634 | 1.44k | IFCALLRET(context->OnChannelDisconnected, error, context, channel->channel_name, |
635 | 1.44k | channel->pInterface); |
636 | 1.44k | } |
637 | 1.44k | } |
638 | | |
639 | 1.44k | if (!fromHashTableFn) |
640 | 359 | dvcman_channel_unref(channel); |
641 | 1.44k | break; |
642 | 359 | case DVC_CHANNEL_CLOSED: |
643 | 359 | break; |
644 | 0 | default: |
645 | 0 | break; |
646 | 1.80k | } |
647 | | |
648 | 1.80k | return error; |
649 | 1.80k | } |
650 | | |
651 | | WINPR_ATTR_MALLOC(dvcman_channel_free, 1) |
652 | | static DVCMAN_CHANNEL* dvcman_channel_new(drdynvcPlugin* drdynvc, |
653 | | IWTSVirtualChannelManager* pChannelMgr, UINT32 ChannelId, |
654 | | const char* ChannelName) |
655 | 1.44k | { |
656 | 1.44k | WINPR_ASSERT(drdynvc); |
657 | 1.44k | WINPR_ASSERT(pChannelMgr); |
658 | 1.44k | DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)calloc(1, sizeof(DVCMAN_CHANNEL)); |
659 | | |
660 | 1.44k | if (!channel) |
661 | 0 | return nullptr; |
662 | | |
663 | 1.44k | channel->dvcman = (DVCMAN*)pChannelMgr; |
664 | 1.44k | channel->channel_id = ChannelId; |
665 | 1.44k | channel->refCounter = 1; |
666 | 1.44k | channel->state = DVC_CHANNEL_INIT; |
667 | 1.44k | channel->channel_name = _strdup(ChannelName); |
668 | 1.44k | if (!channel->channel_name) |
669 | 0 | goto fail; |
670 | | |
671 | 1.44k | channel->decompressor = zgfx_context_new(FALSE); |
672 | 1.44k | if (!channel->decompressor) |
673 | 0 | goto fail; |
674 | | |
675 | 1.44k | if (!InitializeCriticalSectionEx(&(channel->lock), 0, 0)) |
676 | 0 | goto fail; |
677 | | |
678 | 1.44k | if (drdynvc) |
679 | 1.44k | { |
680 | 1.44k | rdpContext* context = drdynvc->rdpcontext; |
681 | 1.44k | if (context) |
682 | 1.44k | { |
683 | 1.44k | ChannelInitializedEventArgs e = WINPR_C_ARRAY_INIT; |
684 | 1.44k | EventArgsInit(&e, "freerdp"); |
685 | 1.44k | e.name = channel->channel_name; |
686 | 1.44k | e.pInterface = channel->pInterface; |
687 | | |
688 | 1.44k | const int rc = PubSub_OnChannelInitialized(context->pubSub, context, &e); |
689 | 1.44k | if (rc < 0) |
690 | 0 | WLog_WARN(TAG, "PubSub_OnChannelInitialized(%s) failed", channel->channel_name); |
691 | 1.44k | } |
692 | 1.44k | } |
693 | | |
694 | 1.44k | return channel; |
695 | 0 | fail: |
696 | 0 | dvcman_channel_free(channel); |
697 | 0 | return nullptr; |
698 | 1.44k | } |
699 | | |
700 | | static void dvcman_clear(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) |
701 | 2.06k | { |
702 | 2.06k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
703 | | |
704 | 2.06k | WINPR_ASSERT(dvcman); |
705 | 2.06k | WINPR_UNUSED(drdynvc); |
706 | | |
707 | 2.06k | HashTable_Clear(dvcman->channelsById); |
708 | 2.06k | ArrayList_Clear(dvcman->plugins); |
709 | 2.06k | ArrayList_Clear(dvcman->plugin_names); |
710 | 2.06k | HashTable_Clear(dvcman->listeners); |
711 | 2.06k | } |
712 | | |
713 | | static void dvcman_free(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) |
714 | 2.06k | { |
715 | 2.06k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
716 | | |
717 | 2.06k | WINPR_ASSERT(dvcman); |
718 | 2.06k | WINPR_UNUSED(drdynvc); |
719 | | |
720 | 2.06k | HashTable_Free(dvcman->channelsById); |
721 | 2.06k | ArrayList_Free(dvcman->plugins); |
722 | 2.06k | ArrayList_Free(dvcman->plugin_names); |
723 | 2.06k | HashTable_Free(dvcman->listeners); |
724 | | |
725 | 2.06k | StreamPool_Free(dvcman->pool); |
726 | 2.06k | free(dvcman); |
727 | 2.06k | } |
728 | | |
729 | | /** |
730 | | * Function description |
731 | | * |
732 | | * @return 0 on success, otherwise a Win32 error code |
733 | | */ |
734 | | WINPR_ATTR_NODISCARD |
735 | | static UINT dvcman_init(drdynvcPlugin* drdynvc, IWTSVirtualChannelManager* pChannelMgr) |
736 | 2.06k | { |
737 | 2.06k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
738 | 2.06k | UINT error = CHANNEL_RC_OK; |
739 | | |
740 | 2.06k | WINPR_ASSERT(dvcman); |
741 | 2.06k | ArrayList_Lock(dvcman->plugins); |
742 | 2.06k | for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) |
743 | 0 | { |
744 | 0 | IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); |
745 | |
|
746 | 0 | error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Initialize, pPlugin, pChannelMgr); |
747 | 0 | if (error != CHANNEL_RC_OK) |
748 | 0 | { |
749 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "Initialize failed with error %" PRIu32 "!", |
750 | 0 | error); |
751 | 0 | goto fail; |
752 | 0 | } |
753 | 0 | } |
754 | | |
755 | 2.06k | fail: |
756 | 2.06k | ArrayList_Unlock(dvcman->plugins); |
757 | 2.06k | return error; |
758 | 2.06k | } |
759 | | |
760 | | /** |
761 | | * Function description |
762 | | * |
763 | | * @return 0 on success, otherwise a Win32 error code |
764 | | */ |
765 | | WINPR_ATTR_NODISCARD |
766 | | static UINT dvcman_write_channel(IWTSVirtualChannel* pChannel, ULONG cbSize, const BYTE* pBuffer, |
767 | | void* pReserved) |
768 | 503 | { |
769 | 503 | BOOL close = FALSE; |
770 | 503 | UINT status = 0; |
771 | 503 | DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel; |
772 | | |
773 | 503 | WINPR_UNUSED(pReserved); |
774 | 503 | if (!channel || !channel->dvcman) |
775 | 0 | return CHANNEL_RC_BAD_CHANNEL; |
776 | | |
777 | 503 | EnterCriticalSection(&(channel->lock)); |
778 | 503 | status = drdynvc_write_data(channel->dvcman->drdynvc, channel->channel_id, channel->tunnelType, |
779 | 503 | pBuffer, cbSize, &close, &channel->stats); |
780 | 503 | LeaveCriticalSection(&(channel->lock)); |
781 | | /* Close delayed, it removes the channel struct */ |
782 | 503 | if (close) |
783 | 0 | dvcman_channel_close(channel, FALSE, FALSE); |
784 | | |
785 | 503 | return status; |
786 | 503 | } |
787 | | |
788 | | /** |
789 | | * Function description |
790 | | * |
791 | | * @return 0 on success, otherwise a Win32 error code |
792 | | */ |
793 | | WINPR_ATTR_NODISCARD |
794 | | static UINT dvcman_close_channel_iface(IWTSVirtualChannel* pChannel) |
795 | 0 | { |
796 | 0 | DVCMAN_CHANNEL* channel = (DVCMAN_CHANNEL*)pChannel; |
797 | |
|
798 | 0 | if (!channel) |
799 | 0 | return CHANNEL_RC_BAD_CHANNEL; |
800 | | |
801 | 0 | WLog_DBG(TAG, "close_channel_iface: id=%" PRIu32 "", channel->channel_id); |
802 | 0 | return dvcman_channel_close(channel, FALSE, FALSE); |
803 | 0 | } |
804 | | |
805 | | struct stats_collector_argument |
806 | | { |
807 | | DrdynvcClientChannelStat* stats; |
808 | | size_t count; |
809 | | size_t used; |
810 | | }; |
811 | | |
812 | | WINPR_ATTR_NODISCARD |
813 | | static BOOL stats_collector(WINPR_ATTR_UNUSED const void* key, void* value, void* arg) |
814 | 0 | { |
815 | 0 | struct stats_collector_argument* args = arg; |
816 | 0 | WINPR_ASSERT(args); |
817 | |
|
818 | 0 | DVCMAN_CHANNEL* channel = value; |
819 | 0 | WINPR_ASSERT(channel); |
820 | 0 | if (args->used >= args->count) |
821 | 0 | return FALSE; |
822 | | |
823 | 0 | DrdynvcClientChannelStat* stat = &args->stats[args->used++]; |
824 | |
|
825 | 0 | if (channel->channel_name) |
826 | 0 | { |
827 | 0 | const size_t len = strnlen(channel->channel_name, ARRAYSIZE(stat->channelName) - 1); |
828 | 0 | strncpy(stat->channelName, channel->channel_name, len); |
829 | 0 | } |
830 | 0 | else |
831 | 0 | memset(stat->channelName, 0, sizeof(stat->channelName)); |
832 | 0 | stat->channelId = channel->channel_id; |
833 | 0 | stat->bytesIn = channel->stats.bytesIn; |
834 | 0 | stat->bytesOut = channel->stats.bytesOut; |
835 | 0 | stat->fragmentsIn = channel->stats.fragmentsIn; |
836 | 0 | stat->fragmentsOut = channel->stats.fragmentsOut; |
837 | 0 | stat->packetsIn = channel->stats.packetsIn; |
838 | 0 | stat->packetsOut = channel->stats.packetsOut; |
839 | 0 | return TRUE; |
840 | 0 | } |
841 | | |
842 | | WINPR_ATTR_MALLOC(free, 1) |
843 | | static DrdynvcClientChannelStat* drdynvc_get_channel_stats(DrdynvcClientContext* context, |
844 | | size_t* pCount) |
845 | 0 | { |
846 | 0 | WINPR_ASSERT(context); |
847 | 0 | WINPR_ASSERT(pCount); |
848 | |
|
849 | 0 | drdynvcPlugin* drdynvc = (drdynvcPlugin*)context->handle; |
850 | 0 | WINPR_ASSERT(drdynvc); |
851 | |
|
852 | 0 | DVCMAN* dvcman = (DVCMAN*)drdynvc->channel_mgr; |
853 | 0 | *pCount = 0; |
854 | 0 | if (!dvcman) |
855 | 0 | return nullptr; |
856 | | |
857 | 0 | struct stats_collector_argument args = { .stats = nullptr, .count = 0 }; |
858 | |
|
859 | 0 | HashTable_Lock(dvcman->channelsById); |
860 | 0 | const size_t count = HashTable_Count(dvcman->channelsById); |
861 | 0 | if (count > 0) |
862 | 0 | { |
863 | 0 | args.stats = calloc(count, sizeof(DrdynvcClientChannelStat)); |
864 | 0 | if (args.stats) |
865 | 0 | { |
866 | 0 | args.count = count; |
867 | 0 | const BOOL rc = HashTable_Foreach(dvcman->channelsById, stats_collector, &args); |
868 | 0 | if (!rc) |
869 | 0 | { |
870 | 0 | HashTable_Unlock(dvcman->channelsById); |
871 | 0 | free(args.stats); |
872 | 0 | return nullptr; |
873 | 0 | } |
874 | 0 | } |
875 | 0 | } |
876 | 0 | HashTable_Unlock(dvcman->channelsById); |
877 | 0 | *pCount = args.used; |
878 | 0 | return args.stats; |
879 | 0 | } |
880 | | |
881 | | #if defined(WITH_RDPEUDP) |
882 | | /* [MS-RDPEDYC] 3.1.5.4: a server tunnelling static virtual channel traffic opens a dynamic |
883 | | * channel with the name of a static one for it, and both ends then send and receive that static |
884 | | * channel's data on the dynamic channel. Servers have been seen doing so also when the client did |
885 | | * not advertise TRANSPORTTYPE_UDP_PREFERRED, so with UDP multitransport any such channel is |
886 | | * taken: what arrives on it is handed to the static channel's plugin as if it had come on the |
887 | | * static channel, and what the plugin writes goes out on it (freerdp_channels_set_static_bridge). |
888 | | * Windows 11 reads the clipboard's answers only there. Without UDP multitransport such a channel |
889 | | * is refused, as any channel without a listener. */ |
890 | | typedef struct |
891 | | { |
892 | | IWTSVirtualChannelCallback iface; |
893 | | freerdp* instance; |
894 | | UINT16 staticChannelId; |
895 | | char staticChannelName[CHANNEL_NAME_LEN + 1]; |
896 | | |
897 | | /* the custom of the sends it set; the static channel owns it, here it is only compared */ |
898 | | void* target; |
899 | | /* target until dvcman_static_bridge_publish sets it, owned here until then */ |
900 | | void* pending; |
901 | | } DVCMAN_STATIC_BRIDGE; |
902 | | |
903 | | /* What the static channel's sends are passed, see dvcman_static_bridge_send. */ |
904 | | typedef struct |
905 | | { |
906 | | drdynvcPlugin* drdynvc; |
907 | | UINT32 channelId; |
908 | | } DVCMAN_STATIC_BRIDGE_TARGET; |
909 | | |
910 | | /* [MS-RDPEDYC] 3.1.5.4.2: the static channel's data is sent on the dynamic channel that carries |
911 | | * it. Called for each message the static channel's plugin writes; one for a dynamic channel that |
912 | | * closed since is discarded. */ |
913 | | WINPR_ATTR_NODISCARD |
914 | | static BOOL dvcman_static_bridge_send(void* custom, const BYTE* data, size_t length) |
915 | | { |
916 | | const DVCMAN_STATIC_BRIDGE_TARGET* target = custom; |
917 | | WINPR_ASSERT(target); |
918 | | WINPR_ASSERT(target->drdynvc); |
919 | | |
920 | | if (length > UINT32_MAX) |
921 | | return FALSE; |
922 | | |
923 | | /* the reference keeps the channel, the table is not held while it writes */ |
924 | | IWTSVirtualChannelManager* mgr = target->drdynvc->channel_mgr; |
925 | | DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(mgr, target->channelId, TRUE); |
926 | | if (!channel) |
927 | | return TRUE; |
928 | | dvcman_return_channel(mgr, channel); |
929 | | |
930 | | UINT rc = CHANNEL_RC_OK; |
931 | | if ((channel->state == DVC_CHANNEL_RUNNING) && channel->staticBridge) |
932 | | rc = dvcman_write_channel(&channel->iface, (ULONG)length, data, nullptr); |
933 | | dvcman_channel_unref(channel); |
934 | | return rc == CHANNEL_RC_OK; |
935 | | } |
936 | | |
937 | | /* One chunk for the static channel, with the Channel PDU Header flags and total length it would |
938 | | * have had there. A plugin that closed its static channel takes no data, which is then |
939 | | * discarded ([MS-RDPEDYC] 3.1.5.4.5). */ |
940 | | static void dvcman_static_bridge_deliver(DVCMAN_STATIC_BRIDGE* bridge, const BYTE* data, |
941 | | size_t length, UINT32 flags, size_t totalLength) |
942 | | { |
943 | | WINPR_ASSERT(bridge); |
944 | | |
945 | | if ((length > 0) && !freerdp_channels_data(bridge->instance, bridge->staticChannelId, data, |
946 | | length, flags, totalLength)) |
947 | | WLog_DBG(TAG, "static channel %" PRIu16 " took no data, %" PRIuz " bytes discarded", |
948 | | bridge->staticChannelId, length); |
949 | | } |
950 | | |
951 | | /* dvcman_receive_channel_data hands a bridge its chunks without this, see |
952 | | * dvcman_static_bridge_receive; anything else is one whole message. */ |
953 | | WINPR_ATTR_NODISCARD |
954 | | static UINT dvcman_static_bridge_on_data(IWTSVirtualChannelCallback* pChannelCallback, |
955 | | wStream* data) |
956 | | { |
957 | | WINPR_ASSERT(data); |
958 | | const size_t length = Stream_GetRemainingLength(data); |
959 | | dvcman_static_bridge_deliver((DVCMAN_STATIC_BRIDGE*)pChannelCallback, Stream_ConstPointer(data), |
960 | | length, CHANNEL_FLAG_FIRST | CHANNEL_FLAG_LAST, length); |
961 | | return CHANNEL_RC_OK; |
962 | | } |
963 | | |
964 | | /* [MS-RDPEDYC] 3.1.5.4.2: the data is not reassembled, each chunk of the data PDUs goes to the |
965 | | * static channel as it came so that its endpoint reassembles it. 3.1.5.4.4: CHANNEL_FLAG_FIRST |
966 | | * comes from a DATA_FIRST PDU or a DATA PDU with a whole message, CHANNEL_FLAG_LAST from the DATA |
967 | | * PDU that completes the message. */ |
968 | | WINPR_ATTR_NODISCARD |
969 | | static UINT dvcman_static_bridge_receive(DVCMAN_CHANNEL* channel, wStream* data) |
970 | | { |
971 | | WINPR_ASSERT(channel); |
972 | | WINPR_ASSERT(data); |
973 | | |
974 | | const size_t length = Stream_GetRemainingLength(data); |
975 | | UINT32 flags = CHANNEL_FLAG_FIRST | CHANNEL_FLAG_LAST; |
976 | | size_t totalLength = length; |
977 | | |
978 | | if (channel->bridgeFragmented) |
979 | | { |
980 | | if (length > channel->dvc_data_length - channel->bridgeReceived) |
981 | | { |
982 | | WLog_Print(channel->dvcman->drdynvc->log, WLOG_ERROR, |
983 | | "data exceeding declared length!"); |
984 | | channel->bridgeFragmented = FALSE; |
985 | | return ERROR_INVALID_DATA; |
986 | | } |
987 | | |
988 | | flags = (channel->bridgeReceived == 0) ? CHANNEL_FLAG_FIRST : 0; |
989 | | channel->bridgeReceived += (UINT32)length; |
990 | | totalLength = channel->dvc_data_length; |
991 | | channel->stats.fragmentsIn++; |
992 | | if (channel->bridgeReceived >= channel->dvc_data_length) |
993 | | { |
994 | | flags |= CHANNEL_FLAG_LAST; |
995 | | channel->bridgeFragmented = FALSE; |
996 | | } |
997 | | } |
998 | | if (flags & CHANNEL_FLAG_LAST) |
999 | | channel->stats.packetsIn++; |
1000 | | |
1001 | | dvcman_static_bridge_deliver((DVCMAN_STATIC_BRIDGE*)channel->channel_callback, |
1002 | | Stream_ConstPointer(data), length, flags, totalLength); |
1003 | | return CHANNEL_RC_OK; |
1004 | | } |
1005 | | |
1006 | | WINPR_ATTR_NODISCARD |
1007 | | static UINT dvcman_static_bridge_on_close(IWTSVirtualChannelCallback* pChannelCallback) |
1008 | | { |
1009 | | DVCMAN_STATIC_BRIDGE* bridge = (DVCMAN_STATIC_BRIDGE*)pChannelCallback; |
1010 | | WINPR_ASSERT(bridge); |
1011 | | |
1012 | | /* the static channel sends on itself again (3.1.5.4.5), unless it is gone already or another |
1013 | | * dynamic channel carries it since */ |
1014 | | if (bridge->target && |
1015 | | !freerdp_channels_set_static_bridge(bridge->instance, bridge->staticChannelName, nullptr, |
1016 | | nullptr, bridge->target)) |
1017 | | WLog_DBG(TAG, "static channel %s is gone already", bridge->staticChannelName); |
1018 | | free(bridge->pending); |
1019 | | free(bridge); |
1020 | | return CHANNEL_RC_OK; |
1021 | | } |
1022 | | |
1023 | | /* [MS-RDPEDYC] 3.1.5.4.2: the static channel's sends move to the dynamic channel once the |
1024 | | * Create Response is out, so none reaches the server ahead of it. */ |
1025 | | static void dvcman_static_bridge_publish(drdynvcPlugin* drdynvc, DVCMAN_CHANNEL* channel) |
1026 | | { |
1027 | | WINPR_ASSERT(drdynvc); |
1028 | | WINPR_ASSERT(channel); |
1029 | | |
1030 | | if (!channel->staticBridge) |
1031 | | return; |
1032 | | DVCMAN_STATIC_BRIDGE* bridge = (DVCMAN_STATIC_BRIDGE*)channel->channel_callback; |
1033 | | WINPR_ASSERT(bridge); |
1034 | | void* target = bridge->pending; |
1035 | | bridge->pending = nullptr; |
1036 | | if (target && freerdp_channels_set_static_bridge(bridge->instance, bridge->staticChannelName, |
1037 | | dvcman_static_bridge_send, free, target)) |
1038 | | bridge->target = target; |
1039 | | else |
1040 | | { |
1041 | | WLog_Print(drdynvc->log, WLOG_WARN, |
1042 | | "static channel %s goes on sending on itself, the server may not read it", |
1043 | | bridge->staticChannelName); |
1044 | | free(target); |
1045 | | } |
1046 | | } |
1047 | | |
1048 | | /* A bridge for a dynamic channel named like a joined static channel, nullptr for any other. */ |
1049 | | WINPR_ATTR_NODISCARD |
1050 | | static DVCMAN_STATIC_BRIDGE* dvcman_static_bridge_new(drdynvcPlugin* drdynvc, |
1051 | | const char* ChannelName) |
1052 | | { |
1053 | | WINPR_ASSERT(drdynvc); |
1054 | | WINPR_ASSERT(ChannelName); |
1055 | | |
1056 | | /* the channel that carries the dynamic ones stays static (3.1.5.4.2) */ |
1057 | | if (_stricmp(ChannelName, DRDYNVC_SVC_CHANNEL_NAME) == 0) |
1058 | | return nullptr; |
1059 | | if (!drdynvc->rdpcontext || !drdynvc->rdpcontext->instance) |
1060 | | return nullptr; |
1061 | | |
1062 | | /* only along with UDP multitransport: Soft-Sync, which the tunnelling goes with (3.1.5.4.1), |
1063 | | * is negotiated when both ends do it */ |
1064 | | const rdpSettings* settings = drdynvc->rdpcontext->settings; |
1065 | | if (!settings || !freerdp_settings_get_bool(settings, FreeRDP_SupportMultitransport) || |
1066 | | ((freerdp_settings_get_uint32(settings, FreeRDP_MultitransportFlags) & |
1067 | | SOFTSYNC_TCP_TO_UDP) == 0)) |
1068 | | return nullptr; |
1069 | | |
1070 | | freerdp* instance = drdynvc->rdpcontext->instance; |
1071 | | const UINT16 id = freerdp_channels_get_id_by_name(instance, ChannelName); |
1072 | | if (id == 0) |
1073 | | return nullptr; |
1074 | | |
1075 | | DVCMAN_STATIC_BRIDGE* bridge = calloc(1, sizeof(DVCMAN_STATIC_BRIDGE)); |
1076 | | if (!bridge) |
1077 | | return nullptr; |
1078 | | bridge->iface.OnDataReceived = dvcman_static_bridge_on_data; |
1079 | | bridge->iface.OnClose = dvcman_static_bridge_on_close; |
1080 | | bridge->instance = instance; |
1081 | | bridge->staticChannelId = id; |
1082 | | (void)_snprintf(bridge->staticChannelName, sizeof(bridge->staticChannelName), "%s", |
1083 | | ChannelName); |
1084 | | return bridge; |
1085 | | } |
1086 | | #endif |
1087 | | |
1088 | | /** |
1089 | | * Function description |
1090 | | * |
1091 | | * @return 0 on success, otherwise a Win32 error code |
1092 | | */ |
1093 | | WINPR_ATTR_NODISCARD |
1094 | | static DVCMAN_CHANNEL* dvcman_create_channel(drdynvcPlugin* drdynvc, |
1095 | | IWTSVirtualChannelManager* pChannelMgr, |
1096 | | UINT32 ChannelId, const char* ChannelName, UINT* res) |
1097 | 3.63k | { |
1098 | 3.63k | BOOL bAccept = 0; |
1099 | 3.63k | DrdynvcClientContext* context = nullptr; |
1100 | 3.63k | DVCMAN* dvcman = (DVCMAN*)pChannelMgr; |
1101 | 3.63k | IWTSVirtualChannelCallback* pCallback = nullptr; |
1102 | | |
1103 | 3.63k | WINPR_ASSERT(dvcman); |
1104 | 3.63k | WINPR_ASSERT(res); |
1105 | | |
1106 | 3.63k | HashTable_Lock(dvcman->listeners); |
1107 | | /* Refuse channels without a listener before registering anything, so a refused |
1108 | | * ChannelId leaves no state behind and the server may reuse it. */ |
1109 | 3.63k | DVCMAN_LISTENER* listener = |
1110 | 3.63k | (DVCMAN_LISTENER*)HashTable_GetItemValue(dvcman->listeners, ChannelName); |
1111 | | #if defined(WITH_RDPEUDP) |
1112 | | /* or a static channel to carry */ |
1113 | | DVCMAN_STATIC_BRIDGE* bridge = |
1114 | | listener ? nullptr : dvcman_static_bridge_new(drdynvc, ChannelName); |
1115 | | if (!listener && !bridge) |
1116 | | #else |
1117 | 3.63k | if (!listener) |
1118 | 1.16k | #endif |
1119 | 1.16k | { |
1120 | 1.16k | HashTable_Unlock(dvcman->listeners); |
1121 | 1.16k | *res = ERROR_NOT_FOUND; |
1122 | 1.16k | return nullptr; |
1123 | 1.16k | } |
1124 | | |
1125 | 2.47k | DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(pChannelMgr, ChannelId, FALSE); |
1126 | 2.47k | if (channel) |
1127 | 1.03k | { |
1128 | 1.03k | switch (channel->state) |
1129 | 1.03k | { |
1130 | 1.03k | case DVC_CHANNEL_RUNNING: |
1131 | 1.03k | WLog_Print(drdynvc->log, WLOG_ERROR, |
1132 | 1.03k | "Protocol error: Duplicated ChannelId %" PRIu32 " (%s)!", ChannelId, |
1133 | 1.03k | ChannelName); |
1134 | 1.03k | *res = CHANNEL_RC_ALREADY_OPEN; |
1135 | 1.03k | goto out; |
1136 | | |
1137 | 0 | case DVC_CHANNEL_CLOSED: |
1138 | 0 | case DVC_CHANNEL_INIT: |
1139 | 0 | default: |
1140 | 0 | { |
1141 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "not expecting a createChannel from state %s", |
1142 | 0 | channel_state2str(channel->state)); |
1143 | 0 | *res = CHANNEL_RC_INITIALIZATION_ERROR; |
1144 | 0 | goto out; |
1145 | 0 | } |
1146 | 1.03k | } |
1147 | 1.03k | } |
1148 | 1.44k | else |
1149 | 1.44k | { |
1150 | 1.44k | HashTable_Lock(dvcman->channelsById); |
1151 | 1.44k | channel = dvcman_channel_new(drdynvc, pChannelMgr, ChannelId, ChannelName); |
1152 | 1.44k | if (!channel) |
1153 | 0 | { |
1154 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_channel_new failed!"); |
1155 | 0 | *res = CHANNEL_RC_NO_MEMORY; |
1156 | 0 | goto out; |
1157 | 0 | } |
1158 | 1.44k | } |
1159 | | |
1160 | 1.44k | if (!HashTable_Insert(dvcman->channelsById, &channel->channel_id, channel)) |
1161 | 0 | { |
1162 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "unable to register channel in our channel list"); |
1163 | 0 | *res = ERROR_INTERNAL_ERROR; |
1164 | 0 | dvcman_channel_free(channel); |
1165 | 0 | channel = nullptr; |
1166 | 0 | goto out; |
1167 | 0 | } |
1168 | | |
1169 | 1.44k | channel->iface.Write = dvcman_write_channel; |
1170 | 1.44k | channel->iface.Close = dvcman_close_channel_iface; |
1171 | 1.44k | bAccept = TRUE; |
1172 | | |
1173 | | #if defined(WITH_RDPEUDP) |
1174 | | if (bridge) |
1175 | | { |
1176 | | /* No OnChannelConnected: clients take the pInterface of a channel named like a static |
1177 | | * one for that channel's interface, and a bridge has none. */ |
1178 | | WLog_Print(drdynvc->log, WLOG_INFO, |
1179 | | "dynamic channel %" PRIu32 " carries the data of static channel %s", |
1180 | | channel->channel_id, ChannelName); |
1181 | | channel->staticBridge = TRUE; |
1182 | | channel->state = DVC_CHANNEL_RUNNING; |
1183 | | channel->channel_callback = &bridge->iface; |
1184 | | |
1185 | | /* published once the Create Response is out, see dvcman_static_bridge_publish */ |
1186 | | DVCMAN_STATIC_BRIDGE_TARGET* target = calloc(1, sizeof(DVCMAN_STATIC_BRIDGE_TARGET)); |
1187 | | if (target) |
1188 | | { |
1189 | | target->drdynvc = drdynvc; |
1190 | | target->channelId = channel->channel_id; |
1191 | | } |
1192 | | bridge->pending = target; |
1193 | | bridge = nullptr; |
1194 | | *res = CHANNEL_RC_OK; |
1195 | | goto out; |
1196 | | } |
1197 | | #endif |
1198 | | |
1199 | 1.44k | *res = listener->listener_callback->OnNewChannelConnection( |
1200 | 1.44k | listener->listener_callback, &channel->iface, nullptr, &bAccept, &pCallback); |
1201 | | |
1202 | 1.44k | if (*res != CHANNEL_RC_OK) |
1203 | 0 | { |
1204 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
1205 | 0 | "OnNewChannelConnection failed with error %" PRIu32 "!", *res); |
1206 | 0 | *res = ERROR_INTERNAL_ERROR; |
1207 | 0 | dvcman_channel_unref(channel); |
1208 | 0 | channel = nullptr; |
1209 | 0 | goto out; |
1210 | 0 | } |
1211 | | |
1212 | 1.44k | if (!bAccept) |
1213 | 0 | { |
1214 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "OnNewChannelConnection returned with bAccept FALSE!"); |
1215 | 0 | *res = ERROR_INTERNAL_ERROR; |
1216 | 0 | dvcman_channel_unref(channel); |
1217 | 0 | channel = nullptr; |
1218 | 0 | goto out; |
1219 | 0 | } |
1220 | | |
1221 | 1.44k | WLog_Print(drdynvc->log, WLOG_DEBUG, "listener %s created new channel %" PRIu32 "", |
1222 | 1.44k | listener->channel_name, channel->channel_id); |
1223 | 1.44k | channel->state = DVC_CHANNEL_RUNNING; |
1224 | 1.44k | channel->channel_callback = pCallback; |
1225 | 1.44k | channel->pInterface = listener->iface.pInterface; |
1226 | 1.44k | context = dvcman->drdynvc->context; |
1227 | | |
1228 | 1.44k | IFCALLRET(context->OnChannelConnected, *res, context, ChannelName, listener->iface.pInterface); |
1229 | 1.44k | if (*res != CHANNEL_RC_OK) |
1230 | 0 | { |
1231 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
1232 | 0 | "context.OnChannelConnected failed with error %" PRIu32 "", *res); |
1233 | 0 | } |
1234 | | |
1235 | 2.47k | out: |
1236 | 2.47k | HashTable_Unlock(dvcman->listeners); |
1237 | 2.47k | HashTable_Unlock(dvcman->channelsById); |
1238 | | #if defined(WITH_RDPEUDP) |
1239 | | free(bridge); |
1240 | | #endif |
1241 | | |
1242 | 2.47k | return channel; |
1243 | 1.44k | } |
1244 | | |
1245 | | /** |
1246 | | * Function description |
1247 | | * |
1248 | | * @return 0 on success, otherwise a Win32 error code |
1249 | | */ |
1250 | | WINPR_ATTR_NODISCARD |
1251 | | static UINT dvcman_open_channel(drdynvcPlugin* drdynvc, DVCMAN_CHANNEL* channel) |
1252 | 1.44k | { |
1253 | 1.44k | UINT error = CHANNEL_RC_OK; |
1254 | | |
1255 | 1.44k | WINPR_ASSERT(drdynvc); |
1256 | 1.44k | WINPR_ASSERT(channel); |
1257 | 1.44k | if (channel->state == DVC_CHANNEL_RUNNING) |
1258 | 1.44k | { |
1259 | 1.44k | IWTSVirtualChannelCallback* pCallback = channel->channel_callback; |
1260 | | |
1261 | 1.44k | if (pCallback->OnOpen) |
1262 | 1.44k | { |
1263 | 1.44k | check_open_close_receive(channel); |
1264 | 1.44k | error = pCallback->OnOpen(pCallback); |
1265 | 1.44k | if (error) |
1266 | 0 | { |
1267 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "OnOpen failed with error %" PRIu32 "!", |
1268 | 0 | error); |
1269 | 0 | goto out; |
1270 | 0 | } |
1271 | 1.44k | } |
1272 | | |
1273 | 1.44k | WLog_Print(drdynvc->log, WLOG_DEBUG, "open_channel: ChannelId %" PRIu32 "", |
1274 | 1.44k | channel->channel_id); |
1275 | 1.44k | } |
1276 | | |
1277 | 1.44k | out: |
1278 | 1.44k | return error; |
1279 | 1.44k | } |
1280 | | |
1281 | | /** |
1282 | | * Function description |
1283 | | * |
1284 | | * @return 0 on success, otherwise a Win32 error code |
1285 | | */ |
1286 | | WINPR_ATTR_NODISCARD |
1287 | | static UINT dvcman_receive_channel_data_first(DVCMAN_CHANNEL* channel, UINT32 length) |
1288 | 271 | { |
1289 | 271 | WINPR_ASSERT(channel); |
1290 | 271 | WINPR_ASSERT(channel->dvcman); |
1291 | | #if defined(WITH_RDPEUDP) |
1292 | | if (channel->staticBridge) |
1293 | | { |
1294 | | /* not reassembled here, see dvcman_static_bridge_receive */ |
1295 | | channel->dvc_data_length = length; |
1296 | | channel->bridgeReceived = 0; |
1297 | | channel->bridgeFragmented = TRUE; |
1298 | | return CHANNEL_RC_OK; |
1299 | | } |
1300 | | #endif |
1301 | 271 | if (channel->dvc_data) |
1302 | 111 | Stream_Release(channel->dvc_data); |
1303 | | |
1304 | 271 | channel->dvc_data = StreamPool_Take(channel->dvcman->pool, 1600); |
1305 | | |
1306 | 271 | if (!channel->dvc_data) |
1307 | 0 | { |
1308 | 0 | drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; |
1309 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
1310 | 0 | return CHANNEL_RC_NO_MEMORY; |
1311 | 0 | } |
1312 | | |
1313 | 271 | channel->dvc_data_length = length; |
1314 | 271 | return CHANNEL_RC_OK; |
1315 | 271 | } |
1316 | | |
1317 | | /** |
1318 | | * Function description |
1319 | | * |
1320 | | * @return 0 on success, otherwise a Win32 error code |
1321 | | */ |
1322 | | WINPR_ATTR_NODISCARD |
1323 | | static UINT dvcman_receive_channel_data(DVCMAN_CHANNEL* channel, wStream* data, |
1324 | | WINPR_ATTR_UNUSED UINT32 ThreadingFlags) |
1325 | 973 | { |
1326 | 973 | UINT status = CHANNEL_RC_OK; |
1327 | 973 | size_t dataSize = Stream_GetRemainingLength(data); |
1328 | | |
1329 | 973 | WINPR_ASSERT(channel); |
1330 | 973 | WINPR_ASSERT(channel->dvcman); |
1331 | | |
1332 | 973 | channel->stats.bytesIn += Stream_Length(data); |
1333 | | #if defined(WITH_RDPEUDP) |
1334 | | if (channel->staticBridge) |
1335 | | return dvcman_static_bridge_receive(channel, data); |
1336 | | #endif |
1337 | 973 | if (channel->dvc_data) |
1338 | 493 | { |
1339 | 493 | drdynvcPlugin* drdynvc = channel->dvcman->drdynvc; |
1340 | | |
1341 | | /* Fragmented data */ |
1342 | 493 | if (Stream_GetPosition(channel->dvc_data) + dataSize > channel->dvc_data_length) |
1343 | 40 | { |
1344 | 40 | WLog_Print(drdynvc->log, WLOG_ERROR, "data exceeding declared length!"); |
1345 | 40 | Stream_Release(channel->dvc_data); |
1346 | 40 | channel->dvc_data = nullptr; |
1347 | 40 | status = ERROR_INVALID_DATA; |
1348 | 40 | goto out; |
1349 | 40 | } |
1350 | 453 | if (!Stream_EnsureRemainingCapacity(channel->dvc_data, dataSize)) |
1351 | 0 | { |
1352 | 0 | Stream_Release(channel->dvc_data); |
1353 | 0 | channel->dvc_data = nullptr; |
1354 | 0 | status = ERROR_INVALID_DATA; |
1355 | 0 | goto out; |
1356 | 0 | } |
1357 | | |
1358 | 453 | Stream_Copy(data, channel->dvc_data, dataSize); |
1359 | 453 | channel->stats.fragmentsIn++; |
1360 | | |
1361 | 453 | if (Stream_GetPosition(channel->dvc_data) >= channel->dvc_data_length) |
1362 | 35 | { |
1363 | 35 | Stream_SealLength(channel->dvc_data); |
1364 | 35 | Stream_ResetPosition(channel->dvc_data); |
1365 | | |
1366 | 35 | status = dvcman_call_on_receive(channel, channel->dvc_data); |
1367 | 35 | Stream_Release(channel->dvc_data); |
1368 | 35 | channel->dvc_data = nullptr; |
1369 | 35 | } |
1370 | 453 | } |
1371 | 480 | else |
1372 | 480 | status = dvcman_call_on_receive(channel, data); |
1373 | | |
1374 | 973 | out: |
1375 | 973 | return status; |
1376 | 973 | } |
1377 | | |
1378 | | WINPR_ATTR_NODISCARD |
1379 | | static UINT8 drdynvc_write_variable_uint(wStream* s, UINT32 val) |
1380 | 3.24k | { |
1381 | 3.24k | UINT8 cb = 0; |
1382 | | |
1383 | 3.24k | if (val <= 0xFF) |
1384 | 62 | { |
1385 | 62 | cb = 0; |
1386 | 62 | Stream_Write_UINT8(s, (UINT8)val); |
1387 | 62 | } |
1388 | 3.17k | else if (val <= 0xFFFF) |
1389 | 188 | { |
1390 | 188 | cb = 1; |
1391 | 188 | Stream_Write_UINT16(s, (UINT16)val); |
1392 | 188 | } |
1393 | 2.99k | else |
1394 | 2.99k | { |
1395 | 2.99k | cb = 2; |
1396 | 2.99k | Stream_Write_UINT32(s, val); |
1397 | 2.99k | } |
1398 | | |
1399 | 3.24k | return cb; |
1400 | 3.24k | } |
1401 | | |
1402 | | /** |
1403 | | * Function description |
1404 | | * |
1405 | | * @return 0 on success, otherwise a Win32 error code |
1406 | | */ |
1407 | | WINPR_ATTR_NODISCARD |
1408 | | static UINT drdynvc_send(drdynvcPlugin* drdynvc, wStream* s, DVCMAN_CHANNEL_STATS* stats) |
1409 | 10.4k | { |
1410 | 10.4k | UINT status = 0; |
1411 | | |
1412 | 10.4k | if (!drdynvc) |
1413 | 0 | status = CHANNEL_RC_BAD_CHANNEL_HANDLE; |
1414 | 10.4k | else |
1415 | 10.4k | { |
1416 | 10.4k | const size_t len = Stream_GetPosition(s); |
1417 | | |
1418 | 10.4k | if (stats) |
1419 | 4.51k | stats->bytesOut += len; |
1420 | | |
1421 | 10.4k | WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelWriteEx); |
1422 | 10.4k | status = drdynvc->channelEntryPoints.pVirtualChannelWriteEx( |
1423 | 10.4k | drdynvc->InitHandle, drdynvc->OpenHandle, Stream_Buffer(s), (UINT32)len, s); |
1424 | 10.4k | } |
1425 | | |
1426 | 10.4k | switch (status) |
1427 | 10.4k | { |
1428 | 10.4k | case CHANNEL_RC_OK: |
1429 | 10.4k | return CHANNEL_RC_OK; |
1430 | | |
1431 | 0 | case CHANNEL_RC_NOT_CONNECTED: |
1432 | 0 | Stream_Release(s); |
1433 | 0 | return CHANNEL_RC_OK; |
1434 | | |
1435 | 0 | case CHANNEL_RC_BAD_CHANNEL_HANDLE: |
1436 | 0 | Stream_Release(s); |
1437 | 0 | WLog_ERR(TAG, "VirtualChannelWriteEx failed with CHANNEL_RC_BAD_CHANNEL_HANDLE"); |
1438 | 0 | return status; |
1439 | | |
1440 | 0 | default: |
1441 | 0 | Stream_Release(s); |
1442 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
1443 | 0 | "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", |
1444 | 0 | WTSErrorToString(status), status); |
1445 | 0 | return status; |
1446 | 10.4k | } |
1447 | 10.4k | } |
1448 | | |
1449 | | /** |
1450 | | * Ends the session over data of a tunnelled channel that can no longer be sent in order. The |
1451 | | * client then reconnects, which is how the main connection handles a broken transport too. |
1452 | | */ |
1453 | | static void drdynvc_tunnel_abort(drdynvcPlugin* drdynvc, const char* why) |
1454 | 0 | { |
1455 | 0 | WINPR_ASSERT(drdynvc); |
1456 | 0 | WINPR_ASSERT(why); |
1457 | |
|
1458 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "%s, ending the session", why); |
1459 | 0 | if (!drdynvc->rdpcontext) |
1460 | 0 | return; |
1461 | 0 | freerdp_set_last_error_if_not(drdynvc->rdpcontext, FREERDP_ERROR_CONNECT_TRANSPORT_FAILED); |
1462 | 0 | (void)freerdp_abort_connect_context(drdynvc->rdpcontext); |
1463 | 0 | } |
1464 | | |
1465 | | /** |
1466 | | * Sends a PDU of a channel the way that channel's data travels: on its multitransport tunnel, |
1467 | | * or on the drdynvc static channel. |
1468 | | * |
1469 | | * A tunnel that is gone falls back to the static channel, and tunnelType is set to 0 so the |
1470 | | * rest of the message follows. A continuation, a later fragment of a message whose start went |
1471 | | * over the tunnel, cannot: on TCP it could overtake that start or be read without it. |
1472 | | * |
1473 | | * @return 0 on success, otherwise a Win32 error code |
1474 | | */ |
1475 | | WINPR_ATTR_NODISCARD |
1476 | | static UINT drdynvc_send_on(drdynvcPlugin* drdynvc, wStream* s, DVCMAN_CHANNEL_STATS* stats, |
1477 | | UINT32* tunnelType, BOOL continuation) |
1478 | 8.14k | { |
1479 | 8.14k | WINPR_ASSERT(s); |
1480 | 8.14k | WINPR_ASSERT(tunnelType); |
1481 | 8.14k | if (drdynvc && (*tunnelType != 0)) |
1482 | 0 | { |
1483 | 0 | const size_t len = Stream_GetPosition(s); |
1484 | 0 | if (freerdp_multitransport_send_dvc(drdynvc->rdpcontext, *tunnelType, Stream_Buffer(s), |
1485 | 0 | len)) |
1486 | 0 | { |
1487 | 0 | if (stats) |
1488 | 0 | stats->bytesOut += len; |
1489 | 0 | Stream_Release(s); |
1490 | 0 | return CHANNEL_RC_OK; |
1491 | 0 | } |
1492 | 0 | if (continuation) |
1493 | 0 | { |
1494 | 0 | Stream_Release(s); |
1495 | 0 | drdynvc_tunnel_abort(drdynvc, "the tunnel failed in the middle of a message"); |
1496 | 0 | return ERROR_BROKEN_PIPE; |
1497 | 0 | } |
1498 | 0 | *tunnelType = 0; |
1499 | 0 | } |
1500 | 8.14k | return drdynvc_send(drdynvc, s, stats); |
1501 | 8.14k | } |
1502 | | |
1503 | | /** |
1504 | | * Function description |
1505 | | * |
1506 | | * @return 0 on success, otherwise a Win32 error code |
1507 | | */ |
1508 | | WINPR_ATTR_NODISCARD |
1509 | | static UINT drdynvc_write_data(drdynvcPlugin* drdynvc, UINT32 ChannelId, UINT32 tunnelType, |
1510 | | const BYTE* data, UINT32 dataSize, BOOL* close, |
1511 | | DVCMAN_CHANNEL_STATS* stats) |
1512 | 503 | { |
1513 | 503 | size_t pos = 0; |
1514 | 503 | UINT8 cbChId = 0; |
1515 | 503 | UINT8 cbLen = 0; |
1516 | 503 | UINT status = CHANNEL_RC_BAD_INIT_HANDLE; |
1517 | 503 | DVCMAN* dvcman = nullptr; |
1518 | | |
1519 | 503 | if (!drdynvc) |
1520 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
1521 | | |
1522 | 503 | dvcman = (DVCMAN*)drdynvc->channel_mgr; |
1523 | 503 | WINPR_ASSERT(dvcman); |
1524 | | |
1525 | | /* All fragments of one message take the same way, so a message never starts on one |
1526 | | * transport and continues on the other. */ |
1527 | 503 | if ((tunnelType != 0) && |
1528 | 0 | (freerdp_multitransport_get_tunnel_state(drdynvc->rdpcontext, tunnelType) != |
1529 | 0 | FREERDP_MULTITRANSPORT_TUNNEL_READY)) |
1530 | 0 | tunnelType = 0; |
1531 | | |
1532 | 503 | WLog_Print(drdynvc->log, WLOG_TRACE, "write_data: ChannelId=%" PRIu32 " size=%" PRIu32 "", |
1533 | 503 | ChannelId, dataSize); |
1534 | 503 | wStream* data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH); |
1535 | | |
1536 | 503 | if (!data_out) |
1537 | 0 | { |
1538 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
1539 | 0 | return CHANNEL_RC_NO_MEMORY; |
1540 | 0 | } |
1541 | | |
1542 | 503 | if (!Stream_SetPosition(data_out, 1)) |
1543 | 0 | { |
1544 | 0 | Stream_Release(data_out); |
1545 | 0 | return ERROR_INVALID_DATA; |
1546 | 0 | } |
1547 | 503 | cbChId = drdynvc_write_variable_uint(data_out, ChannelId); |
1548 | 503 | pos = Stream_GetPosition(data_out); |
1549 | | |
1550 | 503 | if (dataSize == 0) |
1551 | 0 | { |
1552 | | /* TODO: shall treat that case with write(0) that do a close */ |
1553 | 0 | *close = TRUE; |
1554 | 0 | Stream_Release(data_out); |
1555 | 0 | } |
1556 | 503 | else if (dataSize <= CHANNEL_CHUNK_LENGTH - pos) |
1557 | 332 | { |
1558 | 332 | Stream_ResetPosition(data_out); |
1559 | 332 | Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId); |
1560 | 332 | if (!Stream_SetPosition(data_out, pos)) |
1561 | 0 | { |
1562 | 0 | Stream_Release(data_out); |
1563 | 0 | return ERROR_INVALID_DATA; |
1564 | 0 | } |
1565 | 332 | Stream_Write(data_out, data, dataSize); |
1566 | 332 | stats->packetsOut++; |
1567 | 332 | status = drdynvc_send_on(drdynvc, data_out, stats, &tunnelType, FALSE); |
1568 | 332 | } |
1569 | 171 | else |
1570 | 171 | { |
1571 | | /* Fragment the data */ |
1572 | 171 | cbLen = drdynvc_write_variable_uint(data_out, dataSize); |
1573 | 171 | pos = Stream_GetPosition(data_out); |
1574 | 171 | Stream_ResetPosition(data_out); |
1575 | | |
1576 | 171 | const INT32 pdu = (DATA_FIRST_PDU << 4) | cbChId | (cbLen << 2); |
1577 | 171 | Stream_Write_UINT8(data_out, WINPR_ASSERTING_INT_CAST(UINT8, pdu)); |
1578 | 171 | if (!Stream_SetPosition(data_out, pos)) |
1579 | 0 | { |
1580 | 0 | Stream_Release(data_out); |
1581 | 0 | return ERROR_INVALID_DATA; |
1582 | 0 | } |
1583 | | |
1584 | 171 | { |
1585 | 171 | WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH); |
1586 | 171 | const uint32_t chunkLength = |
1587 | 171 | CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos); |
1588 | 171 | Stream_Write(data_out, data, chunkLength); |
1589 | | |
1590 | 171 | data += chunkLength; |
1591 | 171 | dataSize -= chunkLength; |
1592 | 171 | } |
1593 | 171 | if (dataSize > 0) |
1594 | 171 | stats->fragmentsOut++; |
1595 | | |
1596 | 171 | status = drdynvc_send_on(drdynvc, data_out, stats, &tunnelType, FALSE); |
1597 | | |
1598 | 2.73k | while (status == CHANNEL_RC_OK && dataSize > 0) |
1599 | 2.56k | { |
1600 | 2.56k | data_out = StreamPool_Take(dvcman->pool, CHANNEL_CHUNK_LENGTH); |
1601 | | |
1602 | 2.56k | if (!data_out) |
1603 | 0 | { |
1604 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
1605 | 0 | return CHANNEL_RC_NO_MEMORY; |
1606 | 0 | } |
1607 | | |
1608 | 2.56k | if (!Stream_SetPosition(data_out, 1)) |
1609 | 0 | { |
1610 | 0 | Stream_Release(data_out); |
1611 | 0 | return ERROR_INVALID_DATA; |
1612 | 0 | } |
1613 | | |
1614 | 2.56k | cbChId = drdynvc_write_variable_uint(data_out, ChannelId); |
1615 | 2.56k | pos = Stream_GetPosition(data_out); |
1616 | 2.56k | Stream_ResetPosition(data_out); |
1617 | 2.56k | Stream_Write_UINT8(data_out, (DATA_PDU << 4) | cbChId); |
1618 | 2.56k | if (!Stream_SetPosition(data_out, pos)) |
1619 | 0 | { |
1620 | 0 | Stream_Release(data_out); |
1621 | 0 | return ERROR_INVALID_DATA; |
1622 | 0 | } |
1623 | | |
1624 | 2.56k | uint32_t chunkLength = dataSize; |
1625 | | |
1626 | 2.56k | WINPR_ASSERT(pos <= CHANNEL_CHUNK_LENGTH); |
1627 | 2.56k | const uint32_t clen = CHANNEL_CHUNK_LENGTH - WINPR_ASSERTING_INT_CAST(uint32_t, pos); |
1628 | 2.56k | if (chunkLength > clen) |
1629 | 2.39k | { |
1630 | 2.39k | stats->fragmentsOut++; |
1631 | 2.39k | chunkLength = clen; |
1632 | 2.39k | } |
1633 | 171 | else |
1634 | 171 | stats->packetsOut++; |
1635 | | |
1636 | 2.56k | Stream_Write(data_out, data, chunkLength); |
1637 | 2.56k | data += chunkLength; |
1638 | 2.56k | dataSize -= chunkLength; |
1639 | | |
1640 | 2.56k | status = drdynvc_send_on(drdynvc, data_out, stats, &tunnelType, TRUE); |
1641 | 2.56k | } |
1642 | 171 | } |
1643 | | |
1644 | 503 | if (status != CHANNEL_RC_OK) |
1645 | 0 | { |
1646 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", |
1647 | 0 | WTSErrorToString(status), status); |
1648 | 0 | return status; |
1649 | 0 | } |
1650 | | |
1651 | 503 | return CHANNEL_RC_OK; |
1652 | 503 | } |
1653 | | |
1654 | | /** |
1655 | | * Function description |
1656 | | * |
1657 | | * @return 0 on success, otherwise a Win32 error code |
1658 | | */ |
1659 | | WINPR_ATTR_NODISCARD |
1660 | | static UINT drdynvc_send_capability_response(drdynvcPlugin* drdynvc) |
1661 | 1.60k | { |
1662 | 1.60k | UINT status = 0; |
1663 | 1.60k | wStream* s = nullptr; |
1664 | 1.60k | DVCMAN* dvcman = nullptr; |
1665 | | |
1666 | 1.60k | if (!drdynvc) |
1667 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
1668 | | |
1669 | 1.60k | dvcman = (DVCMAN*)drdynvc->channel_mgr; |
1670 | 1.60k | WINPR_ASSERT(dvcman); |
1671 | | |
1672 | 1.60k | WLog_Print(drdynvc->log, WLOG_TRACE, "capability_response"); |
1673 | 1.60k | s = StreamPool_Take(dvcman->pool, 4); |
1674 | | |
1675 | 1.60k | if (!s) |
1676 | 0 | { |
1677 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_New failed!"); |
1678 | 0 | return CHANNEL_RC_NO_MEMORY; |
1679 | 0 | } |
1680 | | |
1681 | 1.60k | Stream_Write_UINT16(s, 0x0050); /* Cmd+Sp+cbChId+Pad. Note: MSTSC sends 0x005c */ |
1682 | 1.60k | Stream_Write_UINT16(s, drdynvc->version); |
1683 | 1.60k | status = drdynvc_send(drdynvc, s, nullptr); |
1684 | | |
1685 | 1.60k | if (status != CHANNEL_RC_OK) |
1686 | 0 | { |
1687 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", |
1688 | 0 | WTSErrorToString(status), status); |
1689 | 0 | } |
1690 | | |
1691 | 1.60k | return status; |
1692 | 1.60k | } |
1693 | | |
1694 | | /** |
1695 | | * Function description |
1696 | | * |
1697 | | * @return 0 on success, otherwise a Win32 error code |
1698 | | */ |
1699 | | WINPR_ATTR_NODISCARD |
1700 | | static UINT drdynvc_process_capability_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, |
1701 | | wStream* s) |
1702 | 1.11k | { |
1703 | 1.11k | UINT status = 0; |
1704 | | |
1705 | 1.11k | if (!drdynvc) |
1706 | 0 | return CHANNEL_RC_BAD_INIT_HANDLE; |
1707 | | |
1708 | 1.11k | if (!Stream_CheckAndLogRequiredLength(TAG, s, 3)) |
1709 | 470 | return ERROR_INVALID_DATA; |
1710 | | |
1711 | 644 | WLog_Print(drdynvc->log, WLOG_TRACE, "capability_request Sp=%d cbChId=%d", Sp, cbChId); |
1712 | 644 | Stream_Seek(s, 1); /* pad */ |
1713 | 644 | Stream_Read_UINT16(s, drdynvc->version); |
1714 | | |
1715 | | /* RDP8 servers offer version 3, though Microsoft forgot to document it |
1716 | | * in their early documents. It behaves the same as version 2. |
1717 | | */ |
1718 | 644 | if ((drdynvc->version == 2) || (drdynvc->version == 3)) |
1719 | 458 | { |
1720 | 458 | if (!Stream_CheckAndLogRequiredLength(TAG, s, 8)) |
1721 | 209 | return ERROR_INVALID_DATA; |
1722 | | |
1723 | 249 | Stream_Read_UINT16(s, drdynvc->PriorityCharge0); |
1724 | 249 | Stream_Read_UINT16(s, drdynvc->PriorityCharge1); |
1725 | 249 | Stream_Read_UINT16(s, drdynvc->PriorityCharge2); |
1726 | 249 | Stream_Read_UINT16(s, drdynvc->PriorityCharge3); |
1727 | 249 | } |
1728 | | |
1729 | 435 | status = drdynvc_send_capability_response(drdynvc); |
1730 | 435 | drdynvc->state = DRDYNVC_STATE_READY; |
1731 | 435 | return status; |
1732 | 644 | } |
1733 | | |
1734 | | WINPR_ATTR_NODISCARD |
1735 | | static UINT32 drdynvc_cblen_to_bytes(int cbLen) |
1736 | 11.2k | { |
1737 | 11.2k | switch (cbLen) |
1738 | 11.2k | { |
1739 | 1.48k | case 0: |
1740 | 1.48k | return 1; |
1741 | | |
1742 | 1.92k | case 1: |
1743 | 1.92k | return 2; |
1744 | | |
1745 | 7.82k | default: |
1746 | 7.82k | return 4; |
1747 | 11.2k | } |
1748 | 11.2k | } |
1749 | | |
1750 | | WINPR_ATTR_NODISCARD |
1751 | | static UINT32 drdynvc_read_variable_uint(wStream* s, int cbLen) |
1752 | 9.28k | { |
1753 | 9.28k | UINT32 val = 0; |
1754 | | |
1755 | 9.28k | switch (cbLen) |
1756 | 9.28k | { |
1757 | 842 | case 0: |
1758 | 842 | Stream_Read_UINT8(s, val); |
1759 | 842 | break; |
1760 | | |
1761 | 930 | case 1: |
1762 | 930 | Stream_Read_UINT16(s, val); |
1763 | 930 | break; |
1764 | | |
1765 | 7.50k | default: |
1766 | 7.50k | Stream_Read_UINT32(s, val); |
1767 | 7.50k | break; |
1768 | 9.28k | } |
1769 | | |
1770 | 9.28k | return val; |
1771 | 9.28k | } |
1772 | | |
1773 | | /** |
1774 | | * Function description |
1775 | | * |
1776 | | * @return 0 on success, otherwise a Win32 error code |
1777 | | */ |
1778 | | WINPR_ATTR_NODISCARD |
1779 | | static UINT drdynvc_process_create_request(drdynvcPlugin* drdynvc, UINT8 Sp, UINT8 cbChId, |
1780 | | wStream* s, UINT32 tunnelType) |
1781 | 5.05k | { |
1782 | 5.05k | UINT status = 0; |
1783 | 5.05k | wStream* data_out = nullptr; |
1784 | 5.05k | UINT channel_status = 0; |
1785 | 5.05k | DVCMAN* dvcman = nullptr; |
1786 | 5.05k | DVCMAN_CHANNEL* channel = nullptr; |
1787 | 5.05k | INT32 retStatus = 0; |
1788 | | |
1789 | 5.05k | WINPR_UNUSED(Sp); |
1790 | 5.05k | if (!drdynvc) |
1791 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
1792 | | |
1793 | 5.05k | dvcman = (DVCMAN*)drdynvc->channel_mgr; |
1794 | 5.05k | WINPR_ASSERT(dvcman); |
1795 | | |
1796 | 5.05k | if (drdynvc->state == DRDYNVC_STATE_CAPABILITIES) |
1797 | 1.17k | { |
1798 | | /** |
1799 | | * For some reason the server does not always send the |
1800 | | * capabilities pdu as it should. When this happens, |
1801 | | * send a capabilities response. |
1802 | | */ |
1803 | 1.17k | drdynvc->version = 3; |
1804 | | |
1805 | 1.17k | if ((status = drdynvc_send_capability_response(drdynvc))) |
1806 | 0 | { |
1807 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_send_capability_response failed!"); |
1808 | 0 | return status; |
1809 | 0 | } |
1810 | | |
1811 | 1.17k | drdynvc->state = DRDYNVC_STATE_READY; |
1812 | 1.17k | } |
1813 | | |
1814 | 5.05k | if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) |
1815 | 616 | return ERROR_INVALID_DATA; |
1816 | | |
1817 | 4.43k | const UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId); |
1818 | 4.43k | const size_t pos = Stream_GetPosition(s); |
1819 | 4.43k | const char* name = Stream_ConstPointer(s); |
1820 | 4.43k | const size_t length = Stream_GetRemainingLength(s); |
1821 | | |
1822 | 4.43k | if (strnlen(name, length) >= length) |
1823 | 798 | return ERROR_INVALID_DATA; |
1824 | | |
1825 | 3.63k | WLog_Print(drdynvc->log, WLOG_DEBUG, |
1826 | 3.63k | "process_create_request: ChannelId=%" PRIu32 " ChannelName=%s", ChannelId, name); |
1827 | | |
1828 | 3.63k | data_out = StreamPool_Take(dvcman->pool, pos + 4); |
1829 | 3.63k | if (!data_out) |
1830 | 0 | { |
1831 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
1832 | 0 | return CHANNEL_RC_NO_MEMORY; |
1833 | 0 | } |
1834 | | |
1835 | 3.63k | Stream_Write_UINT8(data_out, (CREATE_REQUEST_PDU << 4) | cbChId); |
1836 | 3.63k | if (!Stream_SetPosition(s, 1)) |
1837 | 0 | return ERROR_INVALID_DATA; |
1838 | 3.63k | Stream_Copy(s, data_out, pos - 1); |
1839 | | |
1840 | 3.63k | channel = |
1841 | 3.63k | dvcman_create_channel(drdynvc, drdynvc->channel_mgr, ChannelId, name, &channel_status); |
1842 | 3.63k | switch (channel_status) |
1843 | 3.63k | { |
1844 | 1.44k | case CHANNEL_RC_OK: |
1845 | 1.44k | WLog_Print(drdynvc->log, WLOG_DEBUG, "channel created"); |
1846 | 1.44k | retStatus = 0; |
1847 | 1.44k | break; |
1848 | 0 | case CHANNEL_RC_NO_MEMORY: |
1849 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, "not enough memory for channel creation"); |
1850 | 0 | retStatus = STATUS_NO_MEMORY; |
1851 | 0 | break; |
1852 | 1.16k | case ERROR_NOT_FOUND: |
1853 | 1.16k | WLog_Print(drdynvc->log, WLOG_DEBUG, "no listener for '%s'", name); |
1854 | 1.16k | retStatus = STATUS_NOT_FOUND; /* same code used by mstsc, STATUS_UNSUCCESSFUL */ |
1855 | 1.16k | break; |
1856 | 1.03k | default: |
1857 | 1.03k | WLog_Print(drdynvc->log, WLOG_DEBUG, "channel creation error"); |
1858 | 1.03k | retStatus = STATUS_UNSUCCESSFUL; /* same code used by mstsc, STATUS_UNSUCCESSFUL */ |
1859 | 1.03k | break; |
1860 | 3.63k | } |
1861 | 3.63k | Stream_Write_INT32(data_out, retStatus); |
1862 | | |
1863 | | /* A channel created on a tunnel is read there by the server, so it is answered there. */ |
1864 | 3.63k | if (channel && (channel_status == CHANNEL_RC_OK)) |
1865 | 1.44k | dvcman_channel_set_tunnel(channel, tunnelType); |
1866 | 3.63k | UINT32 responseTunnel = tunnelType; |
1867 | 3.63k | status = drdynvc_send_on(drdynvc, data_out, nullptr, &responseTunnel, FALSE); |
1868 | 3.63k | if (status != CHANNEL_RC_OK) |
1869 | 0 | { |
1870 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "VirtualChannelWriteEx failed with %s [%08" PRIX32 "]", |
1871 | 0 | WTSErrorToString(status), status); |
1872 | 0 | if (channel_status == CHANNEL_RC_OK) |
1873 | 0 | dvcman_channel_unref(channel); |
1874 | 0 | return status; |
1875 | 0 | } |
1876 | | |
1877 | 3.63k | if (channel_status == CHANNEL_RC_OK) |
1878 | 1.44k | { |
1879 | | #if defined(WITH_RDPEUDP) |
1880 | | dvcman_static_bridge_publish(drdynvc, channel); |
1881 | | #endif |
1882 | 1.44k | if ((status = dvcman_open_channel(drdynvc, channel))) |
1883 | 0 | { |
1884 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
1885 | 0 | "dvcman_open_channel failed with error %" PRIu32 "!", status); |
1886 | 0 | return status; |
1887 | 0 | } |
1888 | 1.44k | } |
1889 | | |
1890 | 3.63k | return status; |
1891 | 3.63k | } |
1892 | | |
1893 | | /** |
1894 | | * Function description |
1895 | | * |
1896 | | * @return 0 on success, otherwise a Win32 error code |
1897 | | */ |
1898 | | WINPR_ATTR_NODISCARD |
1899 | | static UINT drdynvc_process_data_first(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s, |
1900 | | BOOL compressed, UINT32 ThreadingFlags, UINT32 tunnelType) |
1901 | 1.31k | { |
1902 | 1.31k | WINPR_ASSERT(drdynvc); |
1903 | 1.31k | if (!Stream_CheckAndLogRequiredLength( |
1904 | 1.31k | TAG, s, drdynvc_cblen_to_bytes(cbChId) + drdynvc_cblen_to_bytes(Sp))) |
1905 | 352 | return ERROR_INVALID_DATA; |
1906 | | |
1907 | 967 | UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId); |
1908 | 967 | UINT32 Length = drdynvc_read_variable_uint(s, Sp); |
1909 | 967 | WLog_Print(drdynvc->log, WLOG_TRACE, |
1910 | 967 | "process_data_first: Sp=%d cbChId=%d, ChannelId=%" PRIu32 " Length=%" PRIu32 "", Sp, |
1911 | 967 | cbChId, ChannelId, Length); |
1912 | | |
1913 | 967 | DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); |
1914 | 967 | if (!channel) |
1915 | 492 | { |
1916 | | /** |
1917 | | * Windows Server 2012 R2 can send some messages over |
1918 | | * Microsoft::Windows::RDS::Geometry::v08.01 even if the dynamic virtual channel wasn't |
1919 | | * registered on our side. Ignoring it works. |
1920 | | */ |
1921 | 492 | WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId); |
1922 | 492 | return CHANNEL_RC_OK; |
1923 | 492 | } |
1924 | | |
1925 | 475 | UINT status = CHANNEL_RC_OK; |
1926 | 475 | BOOL shouldFree = FALSE; |
1927 | 475 | if (tunnelType != 0) |
1928 | 0 | dvcman_channel_set_tunnel(channel, tunnelType); |
1929 | 475 | if (channel->state != DVC_CHANNEL_RUNNING) |
1930 | 0 | goto out; |
1931 | | |
1932 | 475 | if (compressed) |
1933 | 236 | { |
1934 | 236 | BYTE* data = nullptr; |
1935 | 236 | UINT32 dataSize = 0; |
1936 | 236 | if (zgfx_decompress(channel->decompressor, Stream_Pointer(s), |
1937 | 236 | WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data, |
1938 | 236 | &dataSize, 0) < 0) |
1939 | 194 | { |
1940 | 194 | status = ERROR_INVALID_DATA; |
1941 | 194 | WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing first packet"); |
1942 | 194 | goto out; |
1943 | 194 | } |
1944 | | |
1945 | 42 | if (dataSize == 0) |
1946 | 10 | { |
1947 | 10 | status = ERROR_INVALID_DATA; |
1948 | 10 | WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing first packet"); |
1949 | 10 | free(data); |
1950 | 10 | goto out; |
1951 | 10 | } |
1952 | | |
1953 | 32 | s = Stream_New(data, dataSize); |
1954 | 32 | if (!s) |
1955 | 0 | { |
1956 | 0 | status = CHANNEL_RC_NO_MEMORY; |
1957 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "error allocating new Stream(len=%" PRIu32 ")", |
1958 | 0 | dataSize); |
1959 | 0 | free(data); |
1960 | 0 | goto out; |
1961 | 0 | } |
1962 | 32 | shouldFree = TRUE; |
1963 | 32 | } |
1964 | | |
1965 | 271 | status = dvcman_receive_channel_data_first(channel, Length); |
1966 | | |
1967 | 271 | if (status == CHANNEL_RC_OK) |
1968 | 271 | status = dvcman_receive_channel_data(channel, s, ThreadingFlags); |
1969 | | |
1970 | 271 | if (status != CHANNEL_RC_OK) |
1971 | 19 | status = dvcman_channel_close(channel, FALSE, FALSE); |
1972 | | |
1973 | 475 | out: |
1974 | 475 | if (shouldFree) |
1975 | 32 | Stream_Free(s, TRUE); |
1976 | 475 | dvcman_channel_unref(channel); |
1977 | 475 | dvcman_return_channel(drdynvc->channel_mgr, channel); |
1978 | 475 | return status; |
1979 | 271 | } |
1980 | | |
1981 | | /** |
1982 | | * Function description |
1983 | | * |
1984 | | * @return 0 on success, otherwise a Win32 error code |
1985 | | */ |
1986 | | WINPR_ATTR_NODISCARD |
1987 | | static UINT drdynvc_process_data(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s, |
1988 | | BOOL compressed, UINT32 ThreadingFlags, UINT32 tunnelType) |
1989 | 2.96k | { |
1990 | 2.96k | WINPR_ASSERT(drdynvc); |
1991 | 2.96k | if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) |
1992 | 546 | return ERROR_INVALID_DATA; |
1993 | | |
1994 | 2.41k | UINT32 ChannelId = drdynvc_read_variable_uint(s, cbChId); |
1995 | 2.41k | WLog_Print(drdynvc->log, WLOG_TRACE, "process_data: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp, |
1996 | 2.41k | cbChId, ChannelId); |
1997 | | |
1998 | 2.41k | DVCMAN_CHANNEL* channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); |
1999 | 2.41k | if (!channel) |
2000 | 620 | { |
2001 | | /** |
2002 | | * Windows Server 2012 R2 can send some messages over |
2003 | | * Microsoft::Windows::RDS::Geometry::v08.01 even if the dynamic virtual channel wasn't |
2004 | | * registered on our side. Ignoring it works. |
2005 | | */ |
2006 | 620 | WLog_Print(drdynvc->log, WLOG_ERROR, "ChannelId %" PRIu32 " not found!", ChannelId); |
2007 | 620 | return CHANNEL_RC_OK; |
2008 | 620 | } |
2009 | | |
2010 | 1.79k | BOOL shouldFree = FALSE; |
2011 | 1.79k | UINT status = CHANNEL_RC_OK; |
2012 | 1.79k | if (tunnelType != 0) |
2013 | 0 | dvcman_channel_set_tunnel(channel, tunnelType); |
2014 | 1.79k | if (channel->state != DVC_CHANNEL_RUNNING) |
2015 | 0 | goto out; |
2016 | | |
2017 | 1.79k | if (compressed) |
2018 | 1.59k | { |
2019 | 1.59k | BYTE* data = nullptr; |
2020 | 1.59k | UINT32 dataSize = 0; |
2021 | | |
2022 | 1.59k | if (zgfx_decompress(channel->decompressor, Stream_Pointer(s), |
2023 | 1.59k | WINPR_ASSERTING_INT_CAST(UINT32, Stream_GetRemainingLength(s)), &data, |
2024 | 1.59k | &dataSize, 0) < 0) |
2025 | 997 | { |
2026 | 997 | status = ERROR_INVALID_DATA; |
2027 | 997 | WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing data packet"); |
2028 | 997 | free(data); |
2029 | 997 | goto out; |
2030 | 997 | } |
2031 | | |
2032 | 594 | if (dataSize == 0) |
2033 | 95 | { |
2034 | 95 | status = ERROR_INVALID_DATA; |
2035 | 95 | WLog_Print(drdynvc->log, WLOG_ERROR, "error de-compressing data packet"); |
2036 | 95 | free(data); |
2037 | 95 | goto out; |
2038 | 95 | } |
2039 | | |
2040 | 499 | s = Stream_New(data, dataSize); |
2041 | 499 | if (!s) |
2042 | 0 | { |
2043 | 0 | status = CHANNEL_RC_NO_MEMORY; |
2044 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "error allocating new Stream(len=%" PRIu32 ")", |
2045 | 0 | dataSize); |
2046 | 0 | free(data); |
2047 | 0 | goto out; |
2048 | 0 | } |
2049 | 499 | shouldFree = TRUE; |
2050 | 499 | } |
2051 | | |
2052 | 702 | status = dvcman_receive_channel_data(channel, s, ThreadingFlags); |
2053 | 702 | if (status != CHANNEL_RC_OK) |
2054 | 21 | status = dvcman_channel_close(channel, FALSE, FALSE); |
2055 | | |
2056 | 1.79k | out: |
2057 | 1.79k | if (shouldFree) |
2058 | 499 | Stream_Free(s, TRUE); |
2059 | 1.79k | dvcman_channel_unref(channel); |
2060 | 1.79k | dvcman_return_channel(drdynvc->channel_mgr, channel); |
2061 | 1.79k | return status; |
2062 | 702 | } |
2063 | | |
2064 | | /** |
2065 | | * Function description |
2066 | | * |
2067 | | * @return 0 on success, otherwise a Win32 error code |
2068 | | */ |
2069 | | WINPR_ATTR_NODISCARD |
2070 | | static UINT drdynvc_process_close_request(drdynvcPlugin* drdynvc, int Sp, int cbChId, wStream* s, |
2071 | | UINT32 tunnelType) |
2072 | 582 | { |
2073 | 582 | UINT32 ChannelId = 0; |
2074 | 582 | DVCMAN_CHANNEL* channel = nullptr; |
2075 | | |
2076 | 582 | WINPR_ASSERT(drdynvc); |
2077 | 582 | if (!Stream_CheckAndLogRequiredLength(TAG, s, drdynvc_cblen_to_bytes(cbChId))) |
2078 | 84 | return ERROR_INVALID_DATA; |
2079 | | |
2080 | 498 | ChannelId = drdynvc_read_variable_uint(s, cbChId); |
2081 | 498 | WLog_Print(drdynvc->log, WLOG_DEBUG, |
2082 | 498 | "process_close_request: Sp=%d cbChId=%d, ChannelId=%" PRIu32 "", Sp, cbChId, |
2083 | 498 | ChannelId); |
2084 | | |
2085 | 498 | channel = dvcman_get_channel_by_id(drdynvc->channel_mgr, ChannelId, TRUE); |
2086 | 498 | if (!channel) |
2087 | 179 | { |
2088 | 179 | WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_close_request channel %" PRIu32 " not present", |
2089 | 179 | ChannelId); |
2090 | 179 | return CHANNEL_RC_OK; |
2091 | 179 | } |
2092 | | |
2093 | 319 | if (tunnelType != 0) |
2094 | 0 | dvcman_channel_set_tunnel(channel, tunnelType); |
2095 | 319 | dvcman_channel_close(channel, TRUE, FALSE); |
2096 | 319 | dvcman_channel_unref(channel); |
2097 | 319 | dvcman_return_channel(drdynvc->channel_mgr, channel); |
2098 | 319 | return CHANNEL_RC_OK; |
2099 | 498 | } |
2100 | | |
2101 | | WINPR_ATTR_NODISCARD |
2102 | | static BOOL drdynvc_soft_sync_negotiated(drdynvcPlugin* drdynvc) |
2103 | 0 | { |
2104 | 0 | WINPR_ASSERT(drdynvc); |
2105 | 0 | if (!drdynvc->rdpcontext) |
2106 | 0 | return FALSE; |
2107 | 0 | const UINT32 flags = |
2108 | 0 | freerdp_settings_get_uint32(drdynvc->rdpcontext->settings, FreeRDP_MultitransportFlags); |
2109 | 0 | return (flags & SOFTSYNC_TCP_TO_UDP) != 0; |
2110 | 0 | } |
2111 | | |
2112 | | WINPR_ATTR_NODISCARD |
2113 | | static UINT drdynvc_order_recv(drdynvcPlugin* drdynvc, wStream* s, UINT32 ThreadingFlags, |
2114 | | UINT32 tunnelType); |
2115 | | |
2116 | | static void drdynvc_tunnel_backlog_clear(drdynvcPlugin* drdynvc) |
2117 | 6.20k | { |
2118 | 6.20k | WINPR_ASSERT(drdynvc); |
2119 | 6.20k | for (size_t x = 0; x < drdynvc->tunnelBacklogCount; x++) |
2120 | 0 | Stream_Release(drdynvc->tunnelBacklog[x]); |
2121 | 6.20k | drdynvc->tunnelBacklogCount = 0; |
2122 | 6.20k | } |
2123 | | |
2124 | | /* Hands up tunnel data that arrived before the Soft-Sync Request, in arrival order. The backlog |
2125 | | * is taken out first, so a PDU handed up that leads back here finds it empty instead of entries |
2126 | | * this loop has handed up already. */ |
2127 | | WINPR_ATTR_NODISCARD |
2128 | | static UINT drdynvc_tunnel_backlog_flush(drdynvcPlugin* drdynvc) |
2129 | 703 | { |
2130 | 703 | WINPR_ASSERT(drdynvc); |
2131 | | |
2132 | 703 | wStream** backlog = drdynvc->tunnelBacklog; |
2133 | 703 | UINT32* types = drdynvc->tunnelBacklogTypes; |
2134 | 703 | const size_t count = drdynvc->tunnelBacklogCount; |
2135 | 703 | drdynvc->tunnelBacklog = nullptr; |
2136 | 703 | drdynvc->tunnelBacklogTypes = nullptr; |
2137 | 703 | drdynvc->tunnelBacklogCount = 0; |
2138 | 703 | drdynvc->tunnelBacklogCapacity = 0; |
2139 | | |
2140 | 703 | UINT error = CHANNEL_RC_OK; |
2141 | 703 | for (size_t x = 0; x < count; x++) |
2142 | 0 | { |
2143 | 0 | const UINT rc = drdynvc_order_recv(drdynvc, backlog[x], TRUE, types[x]); |
2144 | 0 | Stream_Release(backlog[x]); |
2145 | 0 | if (rc != CHANNEL_RC_OK) |
2146 | 0 | error = rc; |
2147 | 0 | } |
2148 | 703 | free((void*)backlog); |
2149 | 703 | free(types); |
2150 | 703 | return error; |
2151 | 703 | } |
2152 | | |
2153 | | WINPR_ATTR_NODISCARD |
2154 | | static BOOL drdynvc_tunnel_backlog_add(drdynvcPlugin* drdynvc, wStream* s, UINT32 tunnelType) |
2155 | 0 | { |
2156 | 0 | WINPR_ASSERT(drdynvc); |
2157 | 0 | WINPR_ASSERT(s); |
2158 | |
|
2159 | 0 | if (drdynvc->tunnelBacklogCount == drdynvc->tunnelBacklogCapacity) |
2160 | 0 | { |
2161 | | /* bounded by DRDYNVC_MAX_TUNNEL_BACKLOG, checked anyway */ |
2162 | 0 | if (drdynvc->tunnelBacklogCapacity > SIZE_MAX / 2 / sizeof(wStream*)) |
2163 | 0 | return FALSE; |
2164 | 0 | const size_t capacity = |
2165 | 0 | (drdynvc->tunnelBacklogCapacity == 0) ? 32 : drdynvc->tunnelBacklogCapacity * 2; |
2166 | 0 | wStream** streams = |
2167 | 0 | (wStream**)realloc((void*)drdynvc->tunnelBacklog, capacity * sizeof(wStream*)); |
2168 | 0 | if (!streams) |
2169 | 0 | return FALSE; |
2170 | 0 | drdynvc->tunnelBacklog = streams; |
2171 | 0 | UINT32* types = realloc(drdynvc->tunnelBacklogTypes, capacity * sizeof(UINT32)); |
2172 | 0 | if (!types) |
2173 | 0 | return FALSE; |
2174 | 0 | drdynvc->tunnelBacklogTypes = types; |
2175 | 0 | drdynvc->tunnelBacklogCapacity = capacity; |
2176 | 0 | } |
2177 | 0 | drdynvc->tunnelBacklog[drdynvc->tunnelBacklogCount] = s; |
2178 | 0 | drdynvc->tunnelBacklogTypes[drdynvc->tunnelBacklogCount] = tunnelType; |
2179 | 0 | drdynvc->tunnelBacklogCount++; |
2180 | 0 | return TRUE; |
2181 | 0 | } |
2182 | | |
2183 | | /** |
2184 | | * A PDU that arrived on a multitransport tunnel. Takes ownership of the stream. |
2185 | | * |
2186 | | * [MS-RDPEDYC] 3.2.5.3.1: the client must not read dynamic channel data from a tunnel before it |
2187 | | * has the Soft-Sync Request, which tells it the server has flushed those channels on TCP. Data |
2188 | | * that overtakes the request on the faster path waits for it. |
2189 | | */ |
2190 | | WINPR_ATTR_NODISCARD |
2191 | | static UINT drdynvc_receive_tunnel_pdu(drdynvcPlugin* drdynvc, wStream* s, UINT32 tunnelType) |
2192 | 0 | { |
2193 | 0 | WINPR_ASSERT(drdynvc); |
2194 | 0 | WINPR_ASSERT(s); |
2195 | |
|
2196 | 0 | if (drdynvc_soft_sync_negotiated(drdynvc) && !drdynvc->softSyncDone) |
2197 | 0 | { |
2198 | | /* Reading on without the request would interleave tunnel and TCP data in no defined |
2199 | | * order, so a server that never sends it ends the session. */ |
2200 | 0 | if (drdynvc->tunnelBacklogCount >= DRDYNVC_MAX_TUNNEL_BACKLOG) |
2201 | 0 | { |
2202 | 0 | Stream_Release(s); |
2203 | 0 | drdynvc_tunnel_backlog_clear(drdynvc); |
2204 | 0 | drdynvc_tunnel_abort(drdynvc, |
2205 | 0 | "no Soft-Sync Request after the tunnel backlog filled up"); |
2206 | 0 | return ERROR_INVALID_DATA; |
2207 | 0 | } |
2208 | 0 | if (!drdynvc_tunnel_backlog_add(drdynvc, s, tunnelType)) |
2209 | 0 | { |
2210 | | /* acknowledged to the server already, dropping it leaves a hole in the stream */ |
2211 | 0 | Stream_Release(s); |
2212 | 0 | drdynvc_tunnel_abort(drdynvc, "no memory to hold tunnel data until the Soft-Sync"); |
2213 | 0 | return CHANNEL_RC_NO_MEMORY; |
2214 | 0 | } |
2215 | 0 | return CHANNEL_RC_OK; |
2216 | 0 | } |
2217 | | |
2218 | 0 | const UINT error = drdynvc_order_recv(drdynvc, s, TRUE, tunnelType); |
2219 | 0 | Stream_Release(s); |
2220 | 0 | return error; |
2221 | 0 | } |
2222 | | |
2223 | | /* Moves a channel's data to a tunnel, if the channel exists. The lookup and the move happen under |
2224 | | * the channel table's lock, so the channel cannot go away in between. */ |
2225 | | static void dvcman_move_channel_to_tunnel(drdynvcPlugin* drdynvc, UINT32 channelId, |
2226 | | UINT32 tunnelType) |
2227 | 0 | { |
2228 | 0 | WINPR_ASSERT(drdynvc); |
2229 | 0 | DVCMAN* dvcman = (DVCMAN*)drdynvc->channel_mgr; |
2230 | 0 | WINPR_ASSERT(dvcman); |
2231 | |
|
2232 | 0 | HashTable_Lock(dvcman->channelsById); |
2233 | 0 | DVCMAN_CHANNEL* channel = HashTable_GetItemValue(dvcman->channelsById, &channelId); |
2234 | 0 | if (channel) |
2235 | 0 | { |
2236 | 0 | dvcman_channel_set_tunnel(channel, tunnelType); |
2237 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, |
2238 | 0 | "channel %" PRIu32 " (%s) moves to tunnel 0x%08" PRIx32, channelId, |
2239 | 0 | channel->channel_name, tunnelType); |
2240 | 0 | } |
2241 | 0 | HashTable_Unlock(dvcman->channelsById); |
2242 | 0 | } |
2243 | | |
2244 | | WINPR_ATTR_NODISCARD |
2245 | | static UINT drdynvc_send_soft_sync_response(drdynvcPlugin* drdynvc, const UINT32* tunnels, |
2246 | | UINT32 count) |
2247 | 703 | { |
2248 | 703 | WINPR_ASSERT(drdynvc); |
2249 | 703 | WINPR_ASSERT(tunnels || (count == 0)); |
2250 | 703 | DVCMAN* dvcman = (DVCMAN*)drdynvc->channel_mgr; |
2251 | 703 | WINPR_ASSERT(dvcman); |
2252 | | |
2253 | | /* count is the number of tunnels we accept, never a value from the wire */ |
2254 | 703 | if (count > DRDYNVC_MAX_SOFT_SYNC_TUNNELS) |
2255 | 0 | return ERROR_INVALID_PARAMETER; |
2256 | 703 | wStream* s = StreamPool_Take(dvcman->pool, 6 + sizeof(UINT32) * count); |
2257 | 703 | if (!s) |
2258 | 0 | return CHANNEL_RC_NO_MEMORY; |
2259 | | |
2260 | | /* [MS-RDPEDYC] 2.2.5.2 DYNVC_SOFT_SYNC_RESPONSE */ |
2261 | 703 | Stream_Write_UINT8(s, SOFT_SYNC_RESPONSE_PDU << 4); |
2262 | 703 | Stream_Write_UINT8(s, 0); /* Pad */ |
2263 | 703 | Stream_Write_UINT32(s, count); |
2264 | 703 | for (UINT32 x = 0; x < count; x++) |
2265 | 0 | Stream_Write_UINT32(s, tunnels[x]); |
2266 | | |
2267 | | /* always on the drdynvc static channel */ |
2268 | 703 | return drdynvc_send(drdynvc, s, nullptr); |
2269 | 703 | } |
2270 | | |
2271 | | /** |
2272 | | * [MS-RDPEDYC] 2.2.5.1 DYNVC_SOFT_SYNC_REQUEST, positioned after the header byte. |
2273 | | */ |
2274 | | WINPR_ATTR_NODISCARD |
2275 | | static UINT drdynvc_process_soft_sync_request(drdynvcPlugin* drdynvc, wStream* s) |
2276 | 1.57k | { |
2277 | 1.57k | WINPR_ASSERT(drdynvc); |
2278 | 1.57k | WINPR_ASSERT(s); |
2279 | | |
2280 | | /* the header byte was read already */ |
2281 | 1.57k | const size_t position = Stream_GetPosition(s); |
2282 | 1.57k | if (position < 1) |
2283 | 0 | return ERROR_INVALID_DATA; |
2284 | 1.57k | const size_t start = position - 1; |
2285 | 1.57k | if (!Stream_CheckAndLogRequiredLength(TAG, s, 9)) |
2286 | 77 | return ERROR_INVALID_DATA; |
2287 | | |
2288 | 1.49k | Stream_Seek_UINT8(s); /* Pad */ |
2289 | 1.49k | const UINT32 length = Stream_Get_UINT32(s); |
2290 | 1.49k | const UINT16 flags = Stream_Get_UINT16(s); |
2291 | 1.49k | const UINT16 numberOfTunnels = Stream_Get_UINT16(s); |
2292 | | |
2293 | | /* Windows counts Length from the Length field on, the header byte and Pad are not in it. The |
2294 | | * channel lists must fit in it, nothing after it belongs to them. */ |
2295 | 1.49k | const size_t total = Stream_Length(s); |
2296 | 1.49k | if ((total < start) || (total - start < 2)) |
2297 | 0 | return ERROR_INVALID_DATA; |
2298 | 1.49k | const size_t available = total - start - 2; |
2299 | 1.49k | if ((length < 8) || (length > available)) |
2300 | 388 | { |
2301 | 388 | WLog_Print(drdynvc->log, WLOG_ERROR, |
2302 | 388 | "Soft-Sync Request Length %" PRIu32 " with %" PRIuz " bytes present", length, |
2303 | 388 | available); |
2304 | 388 | return ERROR_INVALID_DATA; |
2305 | 388 | } |
2306 | | /* length <= available, so the request ends inside the stream */ |
2307 | 1.10k | if (!Stream_SetLength(s, total - (available - length))) |
2308 | 0 | return ERROR_INVALID_DATA; |
2309 | | |
2310 | 1.10k | WLog_Print(drdynvc->log, WLOG_DEBUG, |
2311 | 1.10k | "Soft-Sync Request: flags 0x%04" PRIx16 ", %" PRIu16 " tunnel(s)", flags, |
2312 | 1.10k | numberOfTunnels); |
2313 | | |
2314 | | /* [MS-RDPEDYC] 2.2.5.1: SOFT_SYNC_TCP_FLUSHED must be set. Without it the server has not |
2315 | | * said its TCP data for these channels is out, so none of them moves: an empty response |
2316 | | * keeps them all on TCP. */ |
2317 | 1.10k | if (!(flags & SOFT_SYNC_TCP_FLUSHED)) |
2318 | 204 | { |
2319 | 204 | WLog_Print(drdynvc->log, WLOG_WARN, |
2320 | 204 | "Soft-Sync Request without SOFT_SYNC_TCP_FLUSHED, channels stay on TCP"); |
2321 | 204 | const UINT error = drdynvc_send_soft_sync_response(drdynvc, nullptr, 0); |
2322 | 204 | if (error != CHANNEL_RC_OK) |
2323 | 0 | return error; |
2324 | 204 | drdynvc->softSyncDone = TRUE; |
2325 | 204 | return drdynvc_tunnel_backlog_flush(drdynvc); |
2326 | 204 | } |
2327 | | |
2328 | 903 | const size_t lists = Stream_GetPosition(s); |
2329 | | |
2330 | | /* A tunnel that is still being set up gets its answer once it is up, or has failed. The |
2331 | | * response must not name a tunnel before the Initiate Multitransport Response for it went |
2332 | | * out ([MS-RDPEDYC] 3.2.5.3.2). */ |
2333 | 903 | if (flags & SOFT_SYNC_CHANNEL_LIST_PRESENT) |
2334 | 828 | { |
2335 | 3.78k | for (UINT16 x = 0; x < numberOfTunnels; x++) |
2336 | 3.36k | { |
2337 | 3.36k | if (!Stream_CheckAndLogRequiredLength(TAG, s, 6)) |
2338 | 289 | return ERROR_INVALID_DATA; |
2339 | 3.07k | const UINT32 tunnelType = Stream_Get_UINT32(s); |
2340 | 3.07k | const UINT16 count = Stream_Get_UINT16(s); |
2341 | 3.07k | if (!Stream_CheckAndLogRequiredLengthOfSize(TAG, s, count, sizeof(UINT32))) |
2342 | 115 | return ERROR_INVALID_DATA; |
2343 | 2.96k | Stream_Seek(s, 4ull * count); |
2344 | | |
2345 | 2.96k | if (freerdp_multitransport_get_tunnel_state(drdynvc->rdpcontext, tunnelType) == |
2346 | 2.96k | FREERDP_MULTITRANSPORT_TUNNEL_PENDING) |
2347 | 0 | { |
2348 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, |
2349 | 0 | "tunnel 0x%08" PRIx32 " is not up yet, answering Soft-Sync later", |
2350 | 0 | tunnelType); |
2351 | 0 | const size_t size = Stream_Length(s) - start; |
2352 | 0 | Stream_Free(drdynvc->pendingSoftSync, TRUE); |
2353 | 0 | drdynvc->pendingSoftSync = Stream_New(nullptr, size); |
2354 | 0 | if (!drdynvc->pendingSoftSync) |
2355 | 0 | return CHANNEL_RC_NO_MEMORY; |
2356 | 0 | Stream_Write(drdynvc->pendingSoftSync, Stream_Buffer(s) + start, size); |
2357 | 0 | Stream_SealLength(drdynvc->pendingSoftSync); |
2358 | 0 | return CHANNEL_RC_OK; |
2359 | 0 | } |
2360 | 2.96k | } |
2361 | 828 | } |
2362 | | |
2363 | 499 | UINT32 accepted[DRDYNVC_MAX_SOFT_SYNC_TUNNELS] = WINPR_C_ARRAY_INIT; |
2364 | 499 | UINT32 acceptedCount = 0; |
2365 | 499 | if (!Stream_SetPosition(s, lists)) |
2366 | 0 | return ERROR_INVALID_DATA; |
2367 | | |
2368 | 499 | if (flags & SOFT_SYNC_CHANNEL_LIST_PRESENT) |
2369 | 424 | { |
2370 | 3.04k | for (UINT16 x = 0; x < numberOfTunnels; x++) |
2371 | 2.61k | { |
2372 | 2.61k | const UINT32 tunnelType = Stream_Get_UINT32(s); |
2373 | 2.61k | const UINT16 count = Stream_Get_UINT16(s); |
2374 | 2.61k | const BOOL ready = |
2375 | 2.61k | freerdp_multitransport_get_tunnel_state(drdynvc->rdpcontext, tunnelType) == |
2376 | 2.61k | FREERDP_MULTITRANSPORT_TUNNEL_READY; |
2377 | | |
2378 | 4.60k | for (UINT16 y = 0; y < count; y++) |
2379 | 1.98k | { |
2380 | 1.98k | const UINT32 channelId = Stream_Get_UINT32(s); |
2381 | 1.98k | if (ready) |
2382 | 0 | dvcman_move_channel_to_tunnel(drdynvc, channelId, tunnelType); |
2383 | 1.98k | } |
2384 | | |
2385 | 2.61k | if (ready && (acceptedCount < ARRAYSIZE(accepted))) |
2386 | 0 | { |
2387 | 0 | BOOL known = FALSE; |
2388 | 0 | for (UINT32 z = 0; z < acceptedCount; z++) |
2389 | 0 | known |= (accepted[z] == tunnelType); |
2390 | 0 | if (!known) |
2391 | 0 | accepted[acceptedCount++] = tunnelType; |
2392 | 0 | } |
2393 | 2.61k | } |
2394 | 424 | } |
2395 | 75 | else if (freerdp_multitransport_get_tunnel_state(drdynvc->rdpcontext, TUNNELTYPE_UDPFECR) == |
2396 | 75 | FREERDP_MULTITRANSPORT_TUNNEL_READY) |
2397 | 0 | { |
2398 | 0 | accepted[acceptedCount++] = TUNNELTYPE_UDPFECR; |
2399 | 0 | } |
2400 | | |
2401 | 499 | WLog_Print(drdynvc->log, WLOG_INFO, |
2402 | 499 | "Soft-Sync: %" PRIu32 " tunnel(s) will carry dynamic channel data", acceptedCount); |
2403 | | |
2404 | 499 | const UINT error = drdynvc_send_soft_sync_response(drdynvc, accepted, acceptedCount); |
2405 | 499 | if (error != CHANNEL_RC_OK) |
2406 | 0 | return error; |
2407 | | |
2408 | 499 | drdynvc->softSyncDone = TRUE; |
2409 | 499 | return drdynvc_tunnel_backlog_flush(drdynvc); |
2410 | 499 | } |
2411 | | |
2412 | | /* A tunnel came up or went away: a deferred Soft-Sync Request can be answered now. */ |
2413 | | WINPR_ATTR_NODISCARD |
2414 | | static UINT drdynvc_tunnel_state_changed(drdynvcPlugin* drdynvc) |
2415 | 0 | { |
2416 | 0 | WINPR_ASSERT(drdynvc); |
2417 | 0 | if (drdynvc->pendingSoftSync) |
2418 | 0 | { |
2419 | 0 | wStream* s = drdynvc->pendingSoftSync; |
2420 | 0 | drdynvc->pendingSoftSync = nullptr; |
2421 | 0 | Stream_ResetPosition(s); |
2422 | 0 | const UINT error = drdynvc_order_recv(drdynvc, s, TRUE, 0); |
2423 | 0 | Stream_Free(s, TRUE); |
2424 | 0 | if (error != CHANNEL_RC_OK) |
2425 | 0 | return error; |
2426 | 0 | } |
2427 | | |
2428 | | /* without a tunnel there is nothing to wait for */ |
2429 | 0 | if (!drdynvc->softSyncDone && |
2430 | 0 | (freerdp_multitransport_get_tunnel_state(drdynvc->rdpcontext, TUNNELTYPE_UDPFECR) == |
2431 | 0 | FREERDP_MULTITRANSPORT_TUNNEL_NONE)) |
2432 | 0 | return drdynvc_tunnel_backlog_flush(drdynvc); |
2433 | 0 | return CHANNEL_RC_OK; |
2434 | 0 | } |
2435 | | |
2436 | | /** |
2437 | | * Function description |
2438 | | * |
2439 | | * @return 0 on success, otherwise a Win32 error code |
2440 | | */ |
2441 | | WINPR_ATTR_NODISCARD |
2442 | | static UINT drdynvc_order_recv(drdynvcPlugin* drdynvc, wStream* s, UINT32 ThreadingFlags, |
2443 | | UINT32 tunnelType) |
2444 | 19.7k | { |
2445 | 19.7k | WINPR_ASSERT(drdynvc); |
2446 | 19.7k | if (!Stream_CheckAndLogRequiredLength(TAG, s, 1)) |
2447 | 6.67k | return ERROR_INVALID_DATA; |
2448 | | |
2449 | 13.1k | UINT8 value = Stream_Get_UINT8(s); |
2450 | 13.1k | const UINT8 Cmd = (value & 0xf0) >> 4; |
2451 | 13.1k | const UINT8 Sp = (value & 0x0c) >> 2; |
2452 | 13.1k | const UINT8 cbChId = (value & 0x03) >> 0; |
2453 | 13.1k | WLog_Print(drdynvc->log, WLOG_TRACE, "order_recv: Cmd=%s, Sp=%" PRIu8 " cbChId=%" PRIu8, |
2454 | 13.1k | drdynvc_get_packet_type(Cmd), Sp, cbChId); |
2455 | | |
2456 | | /* Capabilities and Soft-Sync travel on the drdynvc static channel only, a tunnel carries the |
2457 | | * channels themselves. Accepting a Soft-Sync Request from the tunnel would also let it run |
2458 | | * while the tunnel backlog is being handed up. */ |
2459 | 13.1k | if ((tunnelType != 0) && ((Cmd == CAPABILITY_REQUEST_PDU) || (Cmd == SOFT_SYNC_REQUEST_PDU))) |
2460 | 0 | { |
2461 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "%s on tunnel 0x%08" PRIx32 ", rejected", |
2462 | 0 | drdynvc_get_packet_type(Cmd), tunnelType); |
2463 | 0 | return ERROR_INVALID_DATA; |
2464 | 0 | } |
2465 | | |
2466 | 13.1k | switch (Cmd) |
2467 | 13.1k | { |
2468 | 1.11k | case CAPABILITY_REQUEST_PDU: |
2469 | 1.11k | return drdynvc_process_capability_request(drdynvc, Sp, cbChId, s); |
2470 | | |
2471 | 5.05k | case CREATE_REQUEST_PDU: |
2472 | 5.05k | return drdynvc_process_create_request(drdynvc, Sp, cbChId, s, tunnelType); |
2473 | | |
2474 | 635 | case DATA_FIRST_PDU: |
2475 | 1.31k | case DATA_FIRST_COMPRESSED_PDU: |
2476 | 1.31k | return drdynvc_process_data_first(drdynvc, Sp, cbChId, s, |
2477 | 1.31k | (Cmd == DATA_FIRST_COMPRESSED_PDU), ThreadingFlags, |
2478 | 1.31k | tunnelType); |
2479 | | |
2480 | 623 | case DATA_PDU: |
2481 | 2.96k | case DATA_COMPRESSED_PDU: |
2482 | 2.96k | return drdynvc_process_data(drdynvc, Sp, cbChId, s, (Cmd == DATA_COMPRESSED_PDU), |
2483 | 2.96k | ThreadingFlags, tunnelType); |
2484 | | |
2485 | 582 | case CLOSE_REQUEST_PDU: |
2486 | 582 | return drdynvc_process_close_request(drdynvc, Sp, cbChId, s, tunnelType); |
2487 | | |
2488 | 1.57k | case SOFT_SYNC_REQUEST_PDU: |
2489 | 1.57k | return drdynvc_process_soft_sync_request(drdynvc, s); |
2490 | | |
2491 | 201 | case SOFT_SYNC_RESPONSE_PDU: |
2492 | 201 | WLog_Print(drdynvc->log, WLOG_ERROR, |
2493 | 201 | "not expecting a SOFT_SYNC_RESPONSE_PDU as a client"); |
2494 | 201 | return ERROR_INTERNAL_ERROR; |
2495 | | |
2496 | 330 | default: |
2497 | 330 | WLog_Print(drdynvc->log, WLOG_ERROR, "unknown drdynvc cmd 0x%x", Cmd); |
2498 | 330 | return ERROR_INTERNAL_ERROR; |
2499 | 13.1k | } |
2500 | 13.1k | } |
2501 | | |
2502 | | /** |
2503 | | * Function description |
2504 | | * |
2505 | | * @return 0 on success, otherwise a Win32 error code |
2506 | | */ |
2507 | | WINPR_ATTR_NODISCARD |
2508 | | static UINT drdynvc_virtual_channel_event_data_received(drdynvcPlugin* drdynvc, void* pData, |
2509 | | UINT32 dataLength, UINT32 totalLength, |
2510 | | UINT32 dataFlags) |
2511 | 32.6k | { |
2512 | 32.6k | wStream* data_in = nullptr; |
2513 | | |
2514 | 32.6k | WINPR_ASSERT(drdynvc); |
2515 | 32.6k | if ((dataFlags & CHANNEL_FLAG_SUSPEND) || (dataFlags & CHANNEL_FLAG_RESUME)) |
2516 | 0 | { |
2517 | 0 | return CHANNEL_RC_OK; |
2518 | 0 | } |
2519 | | |
2520 | 32.6k | if (dataFlags & CHANNEL_FLAG_FIRST) |
2521 | 23.7k | { |
2522 | 23.7k | if (drdynvc->firstFlagReceived) |
2523 | 1.24k | return ERROR_INVALID_DATA; |
2524 | 22.5k | drdynvc->firstFlagReceived = TRUE; |
2525 | | |
2526 | 22.5k | DVCMAN* mgr = (DVCMAN*)drdynvc->channel_mgr; |
2527 | 22.5k | if (drdynvc->data_in) |
2528 | 1.35k | Stream_Release(drdynvc->data_in); |
2529 | | |
2530 | 22.5k | drdynvc->data_in = StreamPool_Take(mgr->pool, dataLength); |
2531 | 22.5k | drdynvc->totalLength = totalLength; |
2532 | 22.5k | } |
2533 | | |
2534 | 31.3k | if (!(data_in = drdynvc->data_in)) |
2535 | 7.15k | { |
2536 | 7.15k | WLog_Print(drdynvc->log, WLOG_ERROR, "StreamPool_Take failed!"); |
2537 | 7.15k | return CHANNEL_RC_NO_MEMORY; |
2538 | 7.15k | } |
2539 | | |
2540 | 24.2k | if (!Stream_EnsureRemainingCapacity(data_in, dataLength)) |
2541 | 0 | { |
2542 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "Stream_EnsureRemainingCapacity failed!"); |
2543 | 0 | Stream_Release(drdynvc->data_in); |
2544 | 0 | drdynvc->data_in = nullptr; |
2545 | 0 | return ERROR_INTERNAL_ERROR; |
2546 | 0 | } |
2547 | | |
2548 | 24.2k | Stream_Write(data_in, pData, dataLength); |
2549 | | |
2550 | 24.2k | if ((Stream_GetPosition(data_in) > totalLength) || (drdynvc->totalLength != totalLength)) |
2551 | 936 | { |
2552 | 936 | Stream_Release(drdynvc->data_in); |
2553 | 936 | drdynvc->data_in = nullptr; |
2554 | 936 | return ERROR_INVALID_DATA; |
2555 | 936 | } |
2556 | | |
2557 | 23.3k | if (dataFlags & CHANNEL_FLAG_LAST) |
2558 | 22.4k | { |
2559 | 22.4k | if (!drdynvc->firstFlagReceived) |
2560 | 305 | return ERROR_INVALID_DATA; |
2561 | 22.1k | drdynvc->firstFlagReceived = FALSE; |
2562 | | |
2563 | 22.1k | const size_t pos = Stream_GetPosition(data_in); |
2564 | 22.1k | if (drdynvc->totalLength != pos) |
2565 | 2.38k | { |
2566 | 2.38k | WLog_Print(drdynvc->log, WLOG_ERROR, "drdynvc_plugin_process_received: read error"); |
2567 | 2.38k | return ERROR_INVALID_DATA; |
2568 | 2.38k | } |
2569 | | |
2570 | 19.7k | drdynvc->totalLength = 0; |
2571 | 19.7k | drdynvc->data_in = nullptr; |
2572 | 19.7k | Stream_SealLength(data_in); |
2573 | 19.7k | Stream_ResetPosition(data_in); |
2574 | | |
2575 | 19.7k | if (drdynvc->async) |
2576 | 0 | { |
2577 | 0 | if (!MessageQueue_Post(drdynvc->queue, nullptr, DRDYNVC_MSG_DATA, (void*)data_in, |
2578 | 0 | nullptr)) |
2579 | 0 | { |
2580 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Post failed!"); |
2581 | 0 | return ERROR_INTERNAL_ERROR; |
2582 | 0 | } |
2583 | 0 | } |
2584 | 19.7k | else |
2585 | 19.7k | { |
2586 | 19.7k | UINT error = drdynvc_order_recv(drdynvc, data_in, TRUE, 0); |
2587 | 19.7k | Stream_Release(data_in); |
2588 | | |
2589 | 19.7k | if (error) |
2590 | 12.4k | { |
2591 | 12.4k | WLog_Print(drdynvc->log, WLOG_WARN, |
2592 | 12.4k | "drdynvc_order_recv failed with error %" PRIu32 "!", error); |
2593 | 12.4k | return error; |
2594 | 12.4k | } |
2595 | 19.7k | } |
2596 | 19.7k | } |
2597 | | |
2598 | 8.17k | return CHANNEL_RC_OK; |
2599 | 23.3k | } |
2600 | | |
2601 | | static void VCAPITYPE drdynvc_virtual_channel_open_event_ex(LPVOID lpUserParam, DWORD openHandle, |
2602 | | UINT event, LPVOID pData, |
2603 | | UINT32 dataLength, UINT32 totalLength, |
2604 | | UINT32 dataFlags) |
2605 | 43.1k | { |
2606 | 43.1k | UINT error = CHANNEL_RC_OK; |
2607 | 43.1k | drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam; |
2608 | | |
2609 | 43.1k | WINPR_ASSERT(drdynvc); |
2610 | 43.1k | switch (event) |
2611 | 43.1k | { |
2612 | 32.6k | case CHANNEL_EVENT_DATA_RECEIVED: |
2613 | 32.6k | if (!drdynvc || (drdynvc->OpenHandle != openHandle)) |
2614 | 0 | { |
2615 | 0 | WLog_ERR(TAG, "drdynvc_virtual_channel_open_event: error no match"); |
2616 | 0 | return; |
2617 | 0 | } |
2618 | 32.6k | if ((error = drdynvc_virtual_channel_event_data_received(drdynvc, pData, dataLength, |
2619 | 32.6k | totalLength, dataFlags))) |
2620 | 24.4k | WLog_Print(drdynvc->log, WLOG_ERROR, |
2621 | 32.6k | "drdynvc_virtual_channel_event_data_received failed with error %" PRIu32 |
2622 | 32.6k | "", |
2623 | 32.6k | error); |
2624 | | |
2625 | 32.6k | break; |
2626 | | |
2627 | 0 | case CHANNEL_EVENT_WRITE_CANCELLED: |
2628 | 10.4k | case CHANNEL_EVENT_WRITE_COMPLETE: |
2629 | 10.4k | { |
2630 | 10.4k | wStream* s = (wStream*)pData; |
2631 | 10.4k | Stream_Release(s); |
2632 | 10.4k | } |
2633 | 10.4k | break; |
2634 | | |
2635 | 0 | case CHANNEL_EVENT_USER: |
2636 | 0 | break; |
2637 | 0 | default: |
2638 | 0 | break; |
2639 | 43.1k | } |
2640 | | |
2641 | 43.1k | if (error && drdynvc && drdynvc->rdpcontext) |
2642 | 24.4k | setChannelError(drdynvc->rdpcontext, error, |
2643 | 24.4k | "drdynvc_virtual_channel_open_event reported an error"); |
2644 | 43.1k | } |
2645 | | |
2646 | | WINPR_ATTR_NODISCARD |
2647 | | static DWORD WINAPI drdynvc_virtual_channel_client_thread(LPVOID arg) |
2648 | 0 | { |
2649 | | /* TODO: rewrite this */ |
2650 | 0 | wStream* data = nullptr; |
2651 | 0 | wMessage message = WINPR_C_ARRAY_INIT; |
2652 | 0 | UINT error = CHANNEL_RC_OK; |
2653 | 0 | drdynvcPlugin* drdynvc = (drdynvcPlugin*)arg; |
2654 | |
|
2655 | 0 | if (!drdynvc) |
2656 | 0 | { |
2657 | 0 | ExitThread((DWORD)CHANNEL_RC_BAD_CHANNEL_HANDLE); |
2658 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
2659 | 0 | } |
2660 | | |
2661 | 0 | while (1) |
2662 | 0 | { |
2663 | 0 | if (!MessageQueue_Wait(drdynvc->queue)) |
2664 | 0 | { |
2665 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Wait failed!"); |
2666 | 0 | error = ERROR_INTERNAL_ERROR; |
2667 | 0 | break; |
2668 | 0 | } |
2669 | | |
2670 | 0 | if (!MessageQueue_Peek(drdynvc->queue, &message, TRUE)) |
2671 | 0 | { |
2672 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Peek failed!"); |
2673 | 0 | error = ERROR_INTERNAL_ERROR; |
2674 | 0 | break; |
2675 | 0 | } |
2676 | | |
2677 | 0 | if (message.id == WMQ_QUIT) |
2678 | 0 | break; |
2679 | | |
2680 | 0 | if (message.id == DRDYNVC_MSG_DATA) |
2681 | 0 | { |
2682 | 0 | UINT32 ThreadingFlags = TRUE; |
2683 | 0 | data = (wStream*)message.wParam; |
2684 | |
|
2685 | 0 | if ((error = drdynvc_order_recv(drdynvc, data, ThreadingFlags, 0))) |
2686 | 0 | { |
2687 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2688 | 0 | "drdynvc_order_recv failed with error %" PRIu32 "!", error); |
2689 | 0 | } |
2690 | |
|
2691 | 0 | Stream_Release(data); |
2692 | 0 | } |
2693 | 0 | else if (message.id == DRDYNVC_MSG_TUNNEL_DATA) |
2694 | 0 | { |
2695 | 0 | data = (wStream*)message.wParam; |
2696 | 0 | const UINT32 tunnelType = WINPR_ASSERTING_INT_CAST(UINT32, (UINT_PTR)message.lParam); |
2697 | |
|
2698 | 0 | if ((error = drdynvc_receive_tunnel_pdu(drdynvc, data, tunnelType))) |
2699 | 0 | { |
2700 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2701 | 0 | "drdynvc_receive_tunnel_pdu failed with error %" PRIu32 "!", error); |
2702 | 0 | } |
2703 | 0 | } |
2704 | 0 | else if (message.id == DRDYNVC_MSG_TUNNEL_STATE) |
2705 | 0 | { |
2706 | 0 | if ((error = drdynvc_tunnel_state_changed(drdynvc))) |
2707 | 0 | { |
2708 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2709 | 0 | "drdynvc_tunnel_state_changed failed with error %" PRIu32 "!", error); |
2710 | 0 | } |
2711 | 0 | } |
2712 | 0 | } |
2713 | |
|
2714 | 0 | { |
2715 | | /* Disconnect remaining dynamic channels that the server did not. |
2716 | | * This is required to properly shut down channels by calling the appropriate |
2717 | | * event handlers. */ |
2718 | 0 | DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr; |
2719 | |
|
2720 | 0 | HashTable_Clear(drdynvcMgr->channelsById); |
2721 | 0 | } |
2722 | |
|
2723 | 0 | if (error && drdynvc->rdpcontext) |
2724 | 0 | setChannelError(drdynvc->rdpcontext, error, |
2725 | 0 | "drdynvc_virtual_channel_client_thread reported an error"); |
2726 | |
|
2727 | 0 | ExitThread((DWORD)error); |
2728 | 0 | return error; |
2729 | 0 | } |
2730 | | |
2731 | | static void drdynvc_queue_object_free(void* obj) |
2732 | 2.06k | { |
2733 | 2.06k | wStream* s = nullptr; |
2734 | 2.06k | wMessage* msg = (wMessage*)obj; |
2735 | | |
2736 | 2.06k | if (!msg || ((msg->id != DRDYNVC_MSG_DATA) && (msg->id != DRDYNVC_MSG_TUNNEL_DATA))) |
2737 | 2.06k | return; |
2738 | | |
2739 | 0 | s = (wStream*)msg->wParam; |
2740 | |
|
2741 | 0 | if (s) |
2742 | 0 | Stream_Release(s); |
2743 | 0 | } |
2744 | | |
2745 | | /* Multitransport hooks, called on the thread running the main loop, and tunnel data on the |
2746 | | * tunnel thread when the plugin has its own thread. Like data from the static channel, a failure |
2747 | | * here ends the connection. */ |
2748 | | WINPR_ATTR_NODISCARD |
2749 | | static BOOL drdynvc_on_tunnel_data(void* custom, UINT32 tunnelType, wStream* s) |
2750 | 0 | { |
2751 | 0 | drdynvcPlugin* drdynvc = custom; |
2752 | 0 | WINPR_ASSERT(drdynvc); |
2753 | 0 | WINPR_ASSERT(s); |
2754 | |
|
2755 | 0 | if (!drdynvc->channel_mgr || (Stream_Length(s) == 0)) |
2756 | 0 | { |
2757 | 0 | Stream_Release(s); |
2758 | 0 | return TRUE; |
2759 | 0 | } |
2760 | | |
2761 | 0 | if (drdynvc->async) |
2762 | 0 | { |
2763 | 0 | if (!MessageQueue_Post(drdynvc->queue, nullptr, DRDYNVC_MSG_TUNNEL_DATA, s, |
2764 | 0 | (void*)(UINT_PTR)tunnelType)) |
2765 | 0 | { |
2766 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Post failed!"); |
2767 | 0 | Stream_Release(s); |
2768 | 0 | return FALSE; |
2769 | 0 | } |
2770 | 0 | return TRUE; |
2771 | 0 | } |
2772 | | |
2773 | 0 | const UINT error = drdynvc_receive_tunnel_pdu(drdynvc, s, tunnelType); |
2774 | 0 | if (error) |
2775 | 0 | { |
2776 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2777 | 0 | "drdynvc_receive_tunnel_pdu failed with error %" PRIu32 "!", error); |
2778 | 0 | return FALSE; |
2779 | 0 | } |
2780 | 0 | return TRUE; |
2781 | 0 | } |
2782 | | |
2783 | | WINPR_ATTR_NODISCARD |
2784 | | static BOOL drdynvc_on_tunnel_state(void* custom, UINT32 tunnelType, |
2785 | | FreeRDP_MultitransportTunnelState state) |
2786 | 0 | { |
2787 | 0 | drdynvcPlugin* drdynvc = custom; |
2788 | 0 | WINPR_ASSERT(drdynvc); |
2789 | |
|
2790 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, "tunnel 0x%08" PRIx32 " is %s", tunnelType, |
2791 | 0 | (state == FREERDP_MULTITRANSPORT_TUNNEL_READY) ? "ready" : "gone"); |
2792 | |
|
2793 | 0 | if (drdynvc->async) |
2794 | 0 | { |
2795 | 0 | if (!MessageQueue_Post(drdynvc->queue, nullptr, DRDYNVC_MSG_TUNNEL_STATE, nullptr, nullptr)) |
2796 | 0 | { |
2797 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_Post failed!"); |
2798 | 0 | return FALSE; |
2799 | 0 | } |
2800 | 0 | return TRUE; |
2801 | 0 | } |
2802 | | |
2803 | 0 | const UINT error = drdynvc_tunnel_state_changed(drdynvc); |
2804 | 0 | if (error) |
2805 | 0 | { |
2806 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2807 | 0 | "drdynvc_tunnel_state_changed failed with error %" PRIu32 "!", error); |
2808 | 0 | return FALSE; |
2809 | 0 | } |
2810 | 0 | return TRUE; |
2811 | 0 | } |
2812 | | |
2813 | | static void drdynvc_soft_sync_reset(drdynvcPlugin* drdynvc) |
2814 | 6.20k | { |
2815 | 6.20k | WINPR_ASSERT(drdynvc); |
2816 | 6.20k | drdynvc->softSyncDone = FALSE; |
2817 | 6.20k | Stream_Free(drdynvc->pendingSoftSync, TRUE); |
2818 | 6.20k | drdynvc->pendingSoftSync = nullptr; |
2819 | 6.20k | drdynvc_tunnel_backlog_clear(drdynvc); |
2820 | 6.20k | } |
2821 | | |
2822 | | /* After this no tunnel event reaches the plugin. */ |
2823 | | static void drdynvc_tunnel_hooks_remove(drdynvcPlugin* drdynvc) |
2824 | 4.13k | { |
2825 | 4.13k | WINPR_ASSERT(drdynvc); |
2826 | | |
2827 | | /* fails only without a multitransport, then nothing was registered */ |
2828 | 4.13k | if (!freerdp_multitransport_set_dvc_callbacks(drdynvc->rdpcontext, nullptr, nullptr)) |
2829 | 0 | WLog_Print(drdynvc->log, WLOG_DEBUG, "no multitransport tunnel hooks to remove"); |
2830 | 4.13k | } |
2831 | | |
2832 | | WINPR_ATTR_NODISCARD |
2833 | | static UINT drdynvc_virtual_channel_event_initialized(drdynvcPlugin* drdynvc, LPVOID pData, |
2834 | | UINT32 dataLength) |
2835 | 2.14k | { |
2836 | 2.14k | wObject* obj = nullptr; |
2837 | 2.14k | WINPR_UNUSED(pData); |
2838 | 2.14k | WINPR_UNUSED(dataLength); |
2839 | | |
2840 | 2.14k | if (!drdynvc) |
2841 | 0 | goto error; |
2842 | | |
2843 | 2.14k | drdynvc->queue = MessageQueue_New(nullptr); |
2844 | | |
2845 | 2.14k | if (!drdynvc->queue) |
2846 | 0 | { |
2847 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "MessageQueue_New failed!"); |
2848 | 0 | goto error; |
2849 | 0 | } |
2850 | | |
2851 | 2.14k | obj = MessageQueue_Object(drdynvc->queue); |
2852 | 2.14k | obj->fnObjectFree = drdynvc_queue_object_free; |
2853 | 2.14k | drdynvc->channel_mgr = dvcman_new(drdynvc); |
2854 | | |
2855 | 2.14k | if (!drdynvc->channel_mgr) |
2856 | 0 | { |
2857 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_new failed!"); |
2858 | 0 | goto error; |
2859 | 0 | } |
2860 | | |
2861 | 2.14k | return CHANNEL_RC_OK; |
2862 | 0 | error: |
2863 | 0 | return ERROR_INTERNAL_ERROR; |
2864 | 2.14k | } |
2865 | | |
2866 | | /** |
2867 | | * Function description |
2868 | | * |
2869 | | * @return 0 on success, otherwise a Win32 error code |
2870 | | */ |
2871 | | WINPR_ATTR_NODISCARD |
2872 | | static UINT drdynvc_virtual_channel_event_connected(drdynvcPlugin* drdynvc, LPVOID pData, |
2873 | | UINT32 dataLength) |
2874 | 2.06k | { |
2875 | 2.06k | UINT error = 0; |
2876 | 2.06k | UINT32 status = 0; |
2877 | 2.06k | rdpSettings* settings = nullptr; |
2878 | | |
2879 | 2.06k | WINPR_ASSERT(drdynvc); |
2880 | 2.06k | WINPR_UNUSED(pData); |
2881 | 2.06k | WINPR_UNUSED(dataLength); |
2882 | | |
2883 | 2.06k | if (!drdynvc) |
2884 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
2885 | | |
2886 | 2.06k | WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelOpenEx); |
2887 | 2.06k | status = drdynvc->channelEntryPoints.pVirtualChannelOpenEx( |
2888 | 2.06k | drdynvc->InitHandle, &drdynvc->OpenHandle, drdynvc->channelDef.name, |
2889 | 2.06k | drdynvc_virtual_channel_open_event_ex); |
2890 | | |
2891 | 2.06k | if (status != CHANNEL_RC_OK) |
2892 | 0 | { |
2893 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelOpen failed with %s [%08" PRIX32 "]", |
2894 | 0 | WTSErrorToString(status), status); |
2895 | 0 | return status; |
2896 | 0 | } |
2897 | | |
2898 | 2.06k | WINPR_ASSERT(drdynvc->rdpcontext); |
2899 | 2.06k | settings = drdynvc->rdpcontext->settings; |
2900 | 2.06k | WINPR_ASSERT(settings); |
2901 | | |
2902 | 2.06k | for (UINT32 index = 0; |
2903 | 2.06k | index < freerdp_settings_get_uint32(settings, FreeRDP_DynamicChannelCount); index++) |
2904 | 0 | { |
2905 | 0 | const ADDIN_ARGV* args = |
2906 | 0 | freerdp_settings_get_pointer_array(settings, FreeRDP_DynamicChannelArray, index); |
2907 | 0 | error = dvcman_load_addin(drdynvc, drdynvc->channel_mgr, args, drdynvc->rdpcontext); |
2908 | |
|
2909 | 0 | if (CHANNEL_RC_OK != error) |
2910 | 0 | goto error; |
2911 | 0 | } |
2912 | | |
2913 | 2.06k | if ((error = dvcman_init(drdynvc, drdynvc->channel_mgr))) |
2914 | 0 | { |
2915 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "dvcman_init failed with error %" PRIu32 "!", error); |
2916 | 0 | goto error; |
2917 | 0 | } |
2918 | | |
2919 | 2.06k | drdynvc->state = DRDYNVC_STATE_CAPABILITIES; |
2920 | | |
2921 | 2.06k | if (drdynvc->async) |
2922 | 0 | { |
2923 | 0 | if (!(drdynvc->thread = CreateThread(nullptr, 0, drdynvc_virtual_channel_client_thread, |
2924 | 0 | (void*)drdynvc, 0, nullptr))) |
2925 | 0 | { |
2926 | 0 | error = ERROR_INTERNAL_ERROR; |
2927 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "CreateThread failed!"); |
2928 | 0 | goto error; |
2929 | 0 | } |
2930 | | |
2931 | 0 | if (!SetThreadPriority(drdynvc->thread, THREAD_PRIORITY_HIGHEST)) |
2932 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, "SetThreadPriority failed, ignoring."); |
2933 | 0 | } |
2934 | | |
2935 | 2.06k | { |
2936 | 2.06k | drdynvc_soft_sync_reset(drdynvc); |
2937 | 2.06k | const FreeRDP_MultitransportDvcCallbacks callbacks = { |
2938 | 2.06k | .TunnelDataReceived = drdynvc_on_tunnel_data, |
2939 | 2.06k | .TunnelStateChanged = drdynvc_on_tunnel_state, |
2940 | 2.06k | .DataOnTunnelThread = drdynvc->async |
2941 | 2.06k | }; |
2942 | 2.06k | if (!freerdp_multitransport_set_dvc_callbacks(drdynvc->rdpcontext, &callbacks, drdynvc)) |
2943 | 0 | WLog_Print(drdynvc->log, WLOG_WARN, |
2944 | 2.06k | "multitransport tunnels unavailable, using the static channel only"); |
2945 | 2.06k | } |
2946 | | |
2947 | 2.06k | error: |
2948 | 2.06k | return error; |
2949 | 2.06k | } |
2950 | | |
2951 | | /** |
2952 | | * Function description |
2953 | | * |
2954 | | * @return 0 on success, otherwise a Win32 error code |
2955 | | */ |
2956 | | WINPR_ATTR_NODISCARD |
2957 | | static UINT drdynvc_virtual_channel_event_disconnected(drdynvcPlugin* drdynvc) |
2958 | 2.06k | { |
2959 | 2.06k | UINT status = 0; |
2960 | | |
2961 | 2.06k | if (!drdynvc) |
2962 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
2963 | | |
2964 | | /* registered on initialization, whether the channel opened or not */ |
2965 | 2.06k | drdynvc_tunnel_hooks_remove(drdynvc); |
2966 | | |
2967 | 2.06k | if (drdynvc->OpenHandle == 0) |
2968 | 0 | return CHANNEL_RC_OK; |
2969 | | |
2970 | 2.06k | if (drdynvc->queue) |
2971 | 2.06k | { |
2972 | 2.06k | if (!MessageQueue_PostQuit(drdynvc->queue, 0)) |
2973 | 0 | { |
2974 | 0 | status = GetLastError(); |
2975 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
2976 | 0 | "MessageQueue_PostQuit failed with error %" PRIu32 "", status); |
2977 | 0 | return status; |
2978 | 0 | } |
2979 | 2.06k | } |
2980 | | |
2981 | 2.06k | if (drdynvc->thread) |
2982 | 0 | { |
2983 | 0 | if (WaitForSingleObject(drdynvc->thread, INFINITE) != WAIT_OBJECT_0) |
2984 | 0 | { |
2985 | 0 | status = GetLastError(); |
2986 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
2987 | 0 | "WaitForSingleObject failed with error %" PRIu32 "", status); |
2988 | 0 | return status; |
2989 | 0 | } |
2990 | | |
2991 | 0 | (void)CloseHandle(drdynvc->thread); |
2992 | 0 | drdynvc->thread = nullptr; |
2993 | 0 | } |
2994 | 2.06k | else |
2995 | 2.06k | { |
2996 | 2.06k | { |
2997 | | /* Disconnect remaining dynamic channels that the server did not. |
2998 | | * This is required to properly shut down channels by calling the appropriate |
2999 | | * event handlers. */ |
3000 | 2.06k | DVCMAN* drdynvcMgr = (DVCMAN*)drdynvc->channel_mgr; |
3001 | | |
3002 | 2.06k | HashTable_Clear(drdynvcMgr->channelsById); |
3003 | 2.06k | } |
3004 | 2.06k | } |
3005 | | |
3006 | 2.06k | WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelCloseEx); |
3007 | 2.06k | status = drdynvc->channelEntryPoints.pVirtualChannelCloseEx(drdynvc->InitHandle, |
3008 | 2.06k | drdynvc->OpenHandle); |
3009 | | |
3010 | 2.06k | if (status != CHANNEL_RC_OK) |
3011 | 0 | { |
3012 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelClose failed with %s [%08" PRIX32 "]", |
3013 | 0 | WTSErrorToString(status), status); |
3014 | 0 | } |
3015 | | |
3016 | 2.06k | drdynvc_soft_sync_reset(drdynvc); |
3017 | 2.06k | dvcman_clear(drdynvc, drdynvc->channel_mgr); |
3018 | 2.06k | if (drdynvc->queue) |
3019 | 2.06k | MessageQueue_Clear(drdynvc->queue); |
3020 | 2.06k | drdynvc->OpenHandle = 0; |
3021 | | |
3022 | 2.06k | if (drdynvc->data_in) |
3023 | 409 | { |
3024 | 409 | Stream_Release(drdynvc->data_in); |
3025 | 409 | drdynvc->data_in = nullptr; |
3026 | 409 | } |
3027 | | |
3028 | 2.06k | return status; |
3029 | 2.06k | } |
3030 | | |
3031 | | /** |
3032 | | * Function description |
3033 | | * |
3034 | | * @return 0 on success, otherwise a Win32 error code |
3035 | | */ |
3036 | | WINPR_ATTR_NODISCARD |
3037 | | static UINT drdynvc_virtual_channel_event_terminated(drdynvcPlugin* drdynvc) |
3038 | 2.06k | { |
3039 | 2.06k | if (!drdynvc) |
3040 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
3041 | | |
3042 | | /* the hooks point at this plugin, which is freed below */ |
3043 | 2.06k | drdynvc_tunnel_hooks_remove(drdynvc); |
3044 | | |
3045 | 2.06k | MessageQueue_Free(drdynvc->queue); |
3046 | 2.06k | drdynvc->queue = nullptr; |
3047 | | |
3048 | 2.06k | drdynvc_soft_sync_reset(drdynvc); |
3049 | 2.06k | free((void*)drdynvc->tunnelBacklog); |
3050 | 2.06k | free(drdynvc->tunnelBacklogTypes); |
3051 | | |
3052 | 2.06k | if (drdynvc->channel_mgr) |
3053 | 2.06k | { |
3054 | 2.06k | dvcman_free(drdynvc, drdynvc->channel_mgr); |
3055 | 2.06k | drdynvc->channel_mgr = nullptr; |
3056 | 2.06k | } |
3057 | 2.06k | drdynvc->InitHandle = nullptr; |
3058 | 2.06k | free(drdynvc->context); |
3059 | 2.06k | free(drdynvc); |
3060 | 2.06k | return CHANNEL_RC_OK; |
3061 | 2.06k | } |
3062 | | |
3063 | | WINPR_ATTR_NODISCARD |
3064 | | static UINT drdynvc_virtual_channel_event_attached(drdynvcPlugin* drdynvc) |
3065 | 0 | { |
3066 | 0 | UINT error = CHANNEL_RC_OK; |
3067 | 0 | DVCMAN* dvcman = nullptr; |
3068 | |
|
3069 | 0 | if (!drdynvc) |
3070 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
3071 | | |
3072 | 0 | dvcman = (DVCMAN*)drdynvc->channel_mgr; |
3073 | |
|
3074 | 0 | if (!dvcman) |
3075 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
3076 | | |
3077 | 0 | ArrayList_Lock(dvcman->plugins); |
3078 | 0 | for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) |
3079 | 0 | { |
3080 | 0 | IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); |
3081 | |
|
3082 | 0 | error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Attached, pPlugin); |
3083 | 0 | if (error != CHANNEL_RC_OK) |
3084 | 0 | { |
3085 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "Attach failed with error %" PRIu32 "!", error); |
3086 | 0 | goto fail; |
3087 | 0 | } |
3088 | 0 | } |
3089 | | |
3090 | 0 | fail: |
3091 | 0 | ArrayList_Unlock(dvcman->plugins); |
3092 | 0 | return error; |
3093 | 0 | } |
3094 | | |
3095 | | WINPR_ATTR_NODISCARD |
3096 | | static UINT drdynvc_virtual_channel_event_detached(drdynvcPlugin* drdynvc) |
3097 | 0 | { |
3098 | 0 | UINT error = CHANNEL_RC_OK; |
3099 | 0 | DVCMAN* dvcman = nullptr; |
3100 | |
|
3101 | 0 | if (!drdynvc) |
3102 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
3103 | | |
3104 | 0 | dvcman = (DVCMAN*)drdynvc->channel_mgr; |
3105 | |
|
3106 | 0 | if (!dvcman) |
3107 | 0 | return CHANNEL_RC_BAD_CHANNEL_HANDLE; |
3108 | | |
3109 | 0 | ArrayList_Lock(dvcman->plugins); |
3110 | 0 | for (size_t i = 0; i < ArrayList_Count(dvcman->plugins); i++) |
3111 | 0 | { |
3112 | 0 | IWTSPlugin* pPlugin = ArrayList_GetItem(dvcman->plugins, i); |
3113 | |
|
3114 | 0 | error = IFCALLRESULT(CHANNEL_RC_OK, pPlugin->Detached, pPlugin); |
3115 | 0 | if (error != CHANNEL_RC_OK) |
3116 | 0 | { |
3117 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "Detach failed with error %" PRIu32 "!", error); |
3118 | 0 | goto fail; |
3119 | 0 | } |
3120 | 0 | } |
3121 | | |
3122 | 0 | fail: |
3123 | 0 | ArrayList_Unlock(dvcman->plugins); |
3124 | |
|
3125 | 0 | return error; |
3126 | 0 | } |
3127 | | |
3128 | | static VOID VCAPITYPE drdynvc_virtual_channel_init_event_ex(LPVOID lpUserParam, LPVOID pInitHandle, |
3129 | | UINT event, LPVOID pData, |
3130 | | UINT dataLength) |
3131 | 8.34k | { |
3132 | 8.34k | UINT error = CHANNEL_RC_OK; |
3133 | 8.34k | drdynvcPlugin* drdynvc = (drdynvcPlugin*)lpUserParam; |
3134 | | |
3135 | 8.34k | if (!drdynvc || (drdynvc->InitHandle != pInitHandle)) |
3136 | 0 | { |
3137 | 0 | WLog_ERR(TAG, "drdynvc_virtual_channel_init_event: error no match"); |
3138 | 0 | return; |
3139 | 0 | } |
3140 | | |
3141 | 8.34k | switch (event) |
3142 | 8.34k | { |
3143 | 2.14k | case CHANNEL_EVENT_INITIALIZED: |
3144 | 2.14k | error = drdynvc_virtual_channel_event_initialized(drdynvc, pData, dataLength); |
3145 | 2.14k | break; |
3146 | 2.06k | case CHANNEL_EVENT_CONNECTED: |
3147 | 2.06k | if ((error = drdynvc_virtual_channel_event_connected(drdynvc, pData, dataLength))) |
3148 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
3149 | 2.06k | "drdynvc_virtual_channel_event_connected failed with error %" PRIu32 "", |
3150 | 2.06k | error); |
3151 | | |
3152 | 2.06k | break; |
3153 | | |
3154 | 2.06k | case CHANNEL_EVENT_DISCONNECTED: |
3155 | 2.06k | if ((error = drdynvc_virtual_channel_event_disconnected(drdynvc))) |
3156 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
3157 | 2.06k | "drdynvc_virtual_channel_event_disconnected failed with error %" PRIu32 |
3158 | 2.06k | "", |
3159 | 2.06k | error); |
3160 | | |
3161 | 2.06k | break; |
3162 | | |
3163 | 2.06k | case CHANNEL_EVENT_TERMINATED: |
3164 | 2.06k | if ((error = drdynvc_virtual_channel_event_terminated(drdynvc))) |
3165 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
3166 | 2.06k | "drdynvc_virtual_channel_event_terminated failed with error %" PRIu32 "", |
3167 | 2.06k | error); |
3168 | | |
3169 | 2.06k | break; |
3170 | | |
3171 | 0 | case CHANNEL_EVENT_ATTACHED: |
3172 | 0 | if ((error = drdynvc_virtual_channel_event_attached(drdynvc))) |
3173 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
3174 | 0 | "drdynvc_virtual_channel_event_attached failed with error %" PRIu32 "", |
3175 | 0 | error); |
3176 | |
|
3177 | 0 | break; |
3178 | | |
3179 | 0 | case CHANNEL_EVENT_DETACHED: |
3180 | 0 | if ((error = drdynvc_virtual_channel_event_detached(drdynvc))) |
3181 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, |
3182 | 0 | "drdynvc_virtual_channel_event_detached failed with error %" PRIu32 "", |
3183 | 0 | error); |
3184 | |
|
3185 | 0 | break; |
3186 | | |
3187 | 0 | default: |
3188 | 0 | break; |
3189 | 8.34k | } |
3190 | | |
3191 | 8.34k | if (error && drdynvc->rdpcontext) |
3192 | 0 | setChannelError(drdynvc->rdpcontext, error, |
3193 | 0 | "drdynvc_virtual_channel_init_event_ex reported an error"); |
3194 | 8.34k | } |
3195 | | |
3196 | | /** |
3197 | | * Channel Client Interface |
3198 | | */ |
3199 | | |
3200 | | WINPR_ATTR_NODISCARD |
3201 | | static int drdynvc_get_version(DrdynvcClientContext* context) |
3202 | 0 | { |
3203 | 0 | WINPR_ASSERT(context); |
3204 | 0 | drdynvcPlugin* drdynvc = (drdynvcPlugin*)context->handle; |
3205 | 0 | WINPR_ASSERT(drdynvc); |
3206 | 0 | return drdynvc->version; |
3207 | 0 | } |
3208 | | |
3209 | | /* drdynvc is always built-in */ |
3210 | | #define VirtualChannelEntryEx drdynvc_VirtualChannelEntryEx |
3211 | | |
3212 | | FREERDP_ENTRY_POINT(BOOL VCAPITYPE VirtualChannelEntryEx(PCHANNEL_ENTRY_POINTS_EX pEntryPoints, |
3213 | | PVOID pInitHandle)) |
3214 | 2.14k | { |
3215 | 2.14k | UINT rc = 0; |
3216 | 2.14k | drdynvcPlugin* drdynvc = nullptr; |
3217 | 2.14k | DrdynvcClientContext* context = nullptr; |
3218 | 2.14k | CHANNEL_ENTRY_POINTS_FREERDP_EX* pEntryPointsEx = nullptr; |
3219 | 2.14k | drdynvc = (drdynvcPlugin*)calloc(1, sizeof(drdynvcPlugin)); |
3220 | | |
3221 | 2.14k | WINPR_ASSERT(pEntryPoints); |
3222 | 2.14k | if (!drdynvc) |
3223 | 0 | { |
3224 | 0 | WLog_ERR(TAG, "calloc failed!"); |
3225 | 0 | return FALSE; |
3226 | 0 | } |
3227 | | |
3228 | 2.14k | drdynvc->channelDef.options = |
3229 | 2.14k | CHANNEL_OPTION_INITIALIZED | CHANNEL_OPTION_ENCRYPT_RDP | CHANNEL_OPTION_COMPRESS_RDP; |
3230 | 2.14k | (void)sprintf_s(drdynvc->channelDef.name, ARRAYSIZE(drdynvc->channelDef.name), |
3231 | 2.14k | DRDYNVC_SVC_CHANNEL_NAME); |
3232 | 2.14k | drdynvc->state = DRDYNVC_STATE_INITIAL; |
3233 | 2.14k | pEntryPointsEx = (CHANNEL_ENTRY_POINTS_FREERDP_EX*)pEntryPoints; |
3234 | | |
3235 | 2.14k | if ((pEntryPointsEx->cbSize >= sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)) && |
3236 | 2.14k | (pEntryPointsEx->MagicNumber == FREERDP_CHANNEL_MAGIC_NUMBER)) |
3237 | 2.14k | { |
3238 | 2.14k | context = (DrdynvcClientContext*)calloc(1, sizeof(DrdynvcClientContext)); |
3239 | | |
3240 | 2.14k | if (!context) |
3241 | 0 | { |
3242 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "calloc failed!"); |
3243 | 0 | free(drdynvc); |
3244 | 0 | return FALSE; |
3245 | 0 | } |
3246 | | |
3247 | 2.14k | context->handle = (void*)drdynvc; |
3248 | 2.14k | context->custom = nullptr; |
3249 | 2.14k | drdynvc->context = context; |
3250 | 2.14k | context->GetVersion = drdynvc_get_version; |
3251 | 2.14k | context->GetChannelStats = drdynvc_get_channel_stats; |
3252 | 2.14k | drdynvc->rdpcontext = pEntryPointsEx->context; |
3253 | 2.14k | if (!freerdp_settings_get_bool(drdynvc->rdpcontext->settings, |
3254 | 2.14k | FreeRDP_TransportDumpReplay) && |
3255 | 2.14k | !freerdp_settings_get_bool(drdynvc->rdpcontext->settings, |
3256 | 2.14k | FreeRDP_SynchronousDynamicChannels)) |
3257 | 81 | drdynvc->async = TRUE; |
3258 | 2.14k | } |
3259 | | |
3260 | 2.14k | drdynvc->log = WLog_Get(TAG); |
3261 | 2.14k | WLog_Print(drdynvc->log, WLOG_DEBUG, "VirtualChannelEntryEx"); |
3262 | 2.14k | CopyMemory(&(drdynvc->channelEntryPoints), pEntryPoints, |
3263 | 2.14k | sizeof(CHANNEL_ENTRY_POINTS_FREERDP_EX)); |
3264 | 2.14k | drdynvc->InitHandle = pInitHandle; |
3265 | | |
3266 | 2.14k | WINPR_ASSERT(drdynvc->channelEntryPoints.pVirtualChannelInitEx); |
3267 | 2.14k | rc = drdynvc->channelEntryPoints.pVirtualChannelInitEx( |
3268 | 2.14k | drdynvc, context, pInitHandle, &drdynvc->channelDef, 1, VIRTUAL_CHANNEL_VERSION_WIN2000, |
3269 | 2.14k | drdynvc_virtual_channel_init_event_ex); |
3270 | | |
3271 | 2.14k | if (CHANNEL_RC_OK != rc) |
3272 | 0 | { |
3273 | 0 | WLog_Print(drdynvc->log, WLOG_ERROR, "pVirtualChannelInit failed with %s [%08" PRIX32 "]", |
3274 | 0 | WTSErrorToString(rc), rc); |
3275 | 0 | free(drdynvc->context); |
3276 | 0 | free(drdynvc); |
3277 | 0 | return FALSE; |
3278 | 0 | } |
3279 | | |
3280 | 2.14k | drdynvc->channelEntryPoints.pInterface = context; |
3281 | 2.14k | return TRUE; |
3282 | 2.14k | } |