FreeRDP
Loading...
Searching...
No Matches
rdpdr_capabilities.c
1
24#include <freerdp/config.h>
25
26#include <stdio.h>
27#include <stdlib.h>
28#include <string.h>
29
30#include <winpr/crt.h>
31#include <winpr/stream.h>
32
33#include <freerdp/freerdp.h>
34#include <freerdp/utils/rdpdr_utils.h>
35
36#include "rdpdr_main.h"
37#include "rdpdr_capabilities.h"
38
39#define RDPDR_CAPABILITY_HEADER_LENGTH 8
40
41/* Output device direction general capability set */
42static BOOL rdpdr_write_general_capset(rdpdrPlugin* rdpdr, wStream* s)
43{
44 WINPR_ASSERT(rdpdr);
45 const RDPDR_CAPABILITY_HEADER header = { CAP_GENERAL_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH + 36,
46 GENERAL_CAPABILITY_VERSION_02 };
47
48 const UINT32 ioCode1 = rdpdr->clientIOCode1 & rdpdr->serverIOCode1;
49 const UINT32 ioCode2 = rdpdr->clientIOCode2 & rdpdr->serverIOCode2;
50
51 if (rdpdr_write_capset_header(rdpdr->log, s, &header) != CHANNEL_RC_OK)
52 return FALSE;
53
54 if (!Stream_EnsureRemainingCapacity(s, 36))
55 return FALSE;
56
57 Stream_Write_UINT32(s, rdpdr->clientOsType); /* osType, ignored on receipt */
58 Stream_Write_UINT32(s, rdpdr->clientOsVersion); /* osVersion, unused and must be set to zero */
59 Stream_Write_UINT16(s, rdpdr->clientVersionMajor); /* protocolMajorVersion, must be set to 1 */
60 Stream_Write_UINT16(s, rdpdr->clientVersionMinor); /* protocolMinorVersion */
61 Stream_Write_UINT32(s, ioCode1); /* ioCode1 */
62 Stream_Write_UINT32(s, ioCode2); /* ioCode2, must be set to zero, reserved for future use */
63 Stream_Write_UINT32(s, rdpdr->clientExtendedPDU); /* extendedPDU */
64 Stream_Write_UINT32(s, rdpdr->clientExtraFlags1); /* extraFlags1 */
65 Stream_Write_UINT32(
66 s,
67 rdpdr->clientExtraFlags2); /* extraFlags2, must be set to zero, reserved for future use */
68 Stream_Write_UINT32(
69 s, rdpdr->clientSpecialTypeDeviceCap); /* SpecialTypeDeviceCap, number of special devices to
70 be redirected before logon */
71 return TRUE;
72}
73
74/* Process device direction general capability set */
75static UINT rdpdr_process_general_capset(rdpdrPlugin* rdpdr, wStream* s,
76 const RDPDR_CAPABILITY_HEADER* header)
77{
78 WINPR_ASSERT(header);
79
80 const BOOL gotV1 = header->Version == GENERAL_CAPABILITY_VERSION_01;
81 const size_t expect = gotV1 ? 32 : 36;
82 if (header->CapabilityLength != expect)
83 {
84 WLog_Print(rdpdr->log, WLOG_ERROR,
85 "CAP_GENERAL_TYPE::CapabilityLength expected %" PRIuz ", got %" PRIu32, expect,
86 header->CapabilityLength);
87 return ERROR_INVALID_DATA;
88 }
89 if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, expect))
90 return ERROR_INVALID_DATA;
91
92 rdpdr->serverOsType = Stream_Get_UINT32(s); /* osType, ignored on receipt */
93 rdpdr->serverOsVersion = Stream_Get_UINT32(s); /* osVersion, unused and must be set to zero */
94 rdpdr->serverVersionMajor = Stream_Get_UINT16(s); /* protocolMajorVersion, must be set to 1 */
95 rdpdr->serverVersionMinor = Stream_Get_UINT16(s); /* protocolMinorVersion */
96 rdpdr->serverIOCode1 = Stream_Get_UINT32(s); /* ioCode1 */
97 rdpdr->serverIOCode2 =
98 Stream_Get_UINT32(s); /* ioCode2, must be set to zero, reserved for future use */
99 rdpdr->serverExtendedPDU = Stream_Get_UINT32(s); /* extendedPDU */
100 rdpdr->serverExtraFlags1 = Stream_Get_UINT32(s); /* extraFlags1 */
101 rdpdr->serverExtraFlags2 =
102 Stream_Get_UINT32(s); /* extraFlags2, must be set to zero, reserved for future use */
103 if (gotV1)
104 rdpdr->serverSpecialTypeDeviceCap = 0;
105 else
106 rdpdr->serverSpecialTypeDeviceCap = Stream_Get_UINT32(
107 s); /* SpecialTypeDeviceCap, number of special devices to
108 be redirected before logon */
109 return CHANNEL_RC_OK;
110}
111
112/* Output printer direction capability set */
113static BOOL rdpdr_write_printer_capset(rdpdrPlugin* rdpdr, wStream* s)
114{
115 WINPR_UNUSED(rdpdr);
116 const RDPDR_CAPABILITY_HEADER header = { CAP_PRINTER_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
117 PRINT_CAPABILITY_VERSION_01 };
118 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
119}
120
121/* Process printer direction capability set */
122static UINT rdpdr_process_printer_capset(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr, wStream* s,
123 const RDPDR_CAPABILITY_HEADER* header)
124{
125 WINPR_ASSERT(header);
126 Stream_Seek(s, header->CapabilityLength);
127 return CHANNEL_RC_OK;
128}
129
130/* Output port redirection capability set */
131static BOOL rdpdr_write_port_capset(rdpdrPlugin* rdpdr, wStream* s)
132{
133 WINPR_UNUSED(rdpdr);
134 const RDPDR_CAPABILITY_HEADER header = { CAP_PORT_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
135 PORT_CAPABILITY_VERSION_01 };
136 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
137}
138
139/* Process port redirection capability set */
140static UINT rdpdr_process_port_capset(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr, wStream* s,
141 const RDPDR_CAPABILITY_HEADER* header)
142{
143 WINPR_ASSERT(header);
144 Stream_Seek(s, header->CapabilityLength);
145 return CHANNEL_RC_OK;
146}
147
148/* Output drive redirection capability set */
149static BOOL rdpdr_write_drive_capset(rdpdrPlugin* rdpdr, wStream* s)
150{
151 WINPR_UNUSED(rdpdr);
152 const RDPDR_CAPABILITY_HEADER header = { CAP_DRIVE_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
153 DRIVE_CAPABILITY_VERSION_02 };
154 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
155}
156
157/* Process drive redirection capability set */
158static UINT rdpdr_process_drive_capset(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr, wStream* s,
159 const RDPDR_CAPABILITY_HEADER* header)
160{
161 WINPR_ASSERT(header);
162 Stream_Seek(s, header->CapabilityLength);
163 return CHANNEL_RC_OK;
164}
165
166/* Output smart card redirection capability set */
167static BOOL rdpdr_write_smartcard_capset(rdpdrPlugin* rdpdr, wStream* s)
168{
169 WINPR_UNUSED(rdpdr);
170 const RDPDR_CAPABILITY_HEADER header = { CAP_SMARTCARD_TYPE, RDPDR_CAPABILITY_HEADER_LENGTH,
171 SMARTCARD_CAPABILITY_VERSION_01 };
172 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
173}
174
175/* Process smartcard redirection capability set */
176static UINT rdpdr_process_smartcard_capset(WINPR_ATTR_UNUSED rdpdrPlugin* rdpdr, wStream* s,
177 const RDPDR_CAPABILITY_HEADER* header)
178{
179 WINPR_ASSERT(header);
180 Stream_Seek(s, header->CapabilityLength);
181 return CHANNEL_RC_OK;
182}
183
184UINT rdpdr_process_capability_request(rdpdrPlugin* rdpdr, wStream* s)
185{
186 UINT status = CHANNEL_RC_OK;
187 UINT16 numCapabilities = 0;
188
189 if (!rdpdr || !s)
190 return CHANNEL_RC_NULL_DATA;
191
192 rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_CLIENT_CAPS);
193
194 if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 4))
195 return ERROR_INVALID_DATA;
196
197 Stream_Read_UINT16(s, numCapabilities);
198 Stream_Seek(s, 2); /* pad (2 bytes) */
199
200 memset(rdpdr->capabilities, 0, sizeof(rdpdr->capabilities));
201 for (UINT16 i = 0; i < numCapabilities; i++)
202 {
203 RDPDR_CAPABILITY_HEADER header = { 0 };
204 UINT error = rdpdr_read_capset_header(rdpdr->log, s, &header);
205 if (error != CHANNEL_RC_OK)
206 return error;
207
208 switch (header.CapabilityType)
209 {
210 case CAP_GENERAL_TYPE:
211 rdpdr->capabilities[header.CapabilityType] = TRUE;
212 status = rdpdr_process_general_capset(rdpdr, s, &header);
213 break;
214
215 case CAP_PRINTER_TYPE:
216 rdpdr->capabilities[header.CapabilityType] = TRUE;
217 status = rdpdr_process_printer_capset(rdpdr, s, &header);
218 break;
219
220 case CAP_PORT_TYPE:
221 rdpdr->capabilities[header.CapabilityType] = TRUE;
222 status = rdpdr_process_port_capset(rdpdr, s, &header);
223 break;
224
225 case CAP_DRIVE_TYPE:
226 rdpdr->capabilities[header.CapabilityType] = TRUE;
227 status = rdpdr_process_drive_capset(rdpdr, s, &header);
228 break;
229
230 case CAP_SMARTCARD_TYPE:
231 rdpdr->capabilities[header.CapabilityType] = TRUE;
232 status = rdpdr_process_smartcard_capset(rdpdr, s, &header);
233 break;
234
235 default:
236 break;
237 }
238
239 if (status != CHANNEL_RC_OK)
240 return status;
241 }
242
243 return CHANNEL_RC_OK;
244}
245
251UINT rdpdr_send_capability_response(rdpdrPlugin* rdpdr)
252{
253 WINPR_ASSERT(rdpdr);
254 WINPR_ASSERT(rdpdr->rdpcontext);
255
256 rdpSettings* settings = rdpdr->rdpcontext->settings;
257 WINPR_ASSERT(settings);
258
259 wStream* s = StreamPool_Take(rdpdr->pool, 256);
260
261 if (!s)
262 {
263 WLog_Print(rdpdr->log, WLOG_ERROR, "Stream_New failed!");
264 return CHANNEL_RC_NO_MEMORY;
265 }
266
267 const RDPDR_DEVICE* cdrives =
268 freerdp_device_collection_find_type(settings, RDPDR_DTYP_FILESYSTEM);
269 const RDPDR_DEVICE* cserial = freerdp_device_collection_find_type(settings, RDPDR_DTYP_SERIAL);
270 const RDPDR_DEVICE* cparallel =
271 freerdp_device_collection_find_type(settings, RDPDR_DTYP_PARALLEL);
272 const RDPDR_DEVICE* csmart =
273 freerdp_device_collection_find_type(settings, RDPDR_DTYP_SMARTCARD);
274 const RDPDR_DEVICE* cprinter = freerdp_device_collection_find_type(settings, RDPDR_DTYP_PRINT);
275
276 /* Only send capabilities the server announced */
277 const BOOL drives = cdrives && rdpdr->capabilities[CAP_DRIVE_TYPE];
278 const BOOL serial = cserial && rdpdr->capabilities[CAP_PORT_TYPE];
279 const BOOL parallel = cparallel && rdpdr->capabilities[CAP_PORT_TYPE];
280 const BOOL smart = csmart && rdpdr->capabilities[CAP_SMARTCARD_TYPE];
281 const BOOL printer = cprinter && rdpdr->capabilities[CAP_PRINTER_TYPE];
282 UINT16 count = 1;
283 if (drives)
284 count++;
285 if (serial || parallel)
286 count++;
287 if (smart)
288 count++;
289 if (printer)
290 count++;
291
292 Stream_Write_UINT16(s, RDPDR_CTYP_CORE);
293 Stream_Write_UINT16(s, PAKID_CORE_CLIENT_CAPABILITY);
294 Stream_Write_UINT16(s, count); /* numCapabilities */
295 Stream_Write_UINT16(s, 0); /* pad */
296
297 if (!rdpdr_write_general_capset(rdpdr, s))
298 goto fail;
299
300 if (printer)
301 {
302 if (!rdpdr_write_printer_capset(rdpdr, s))
303 goto fail;
304 }
305 if (serial || parallel)
306 {
307 if (!rdpdr_write_port_capset(rdpdr, s))
308 goto fail;
309 }
310 if (drives)
311 {
312 if (!rdpdr_write_drive_capset(rdpdr, s))
313 goto fail;
314 }
315 if (smart)
316 {
317 if (!rdpdr_write_smartcard_capset(rdpdr, s))
318 goto fail;
319 }
320 return rdpdr_send(rdpdr, s);
321
322fail:
323 Stream_Release(s);
324 return ERROR_OUTOFMEMORY;
325}
RDPDR_DEVICE
Definition settings_types.h:441
RDPDR_CAPABILITY_HEADER
Definition rdpdr_utils.h:42