24#include <freerdp/config.h>
31#include <winpr/stream.h>
33#include <freerdp/freerdp.h>
34#include <freerdp/utils/rdpdr_utils.h>
36#include "rdpdr_main.h"
37#include "rdpdr_capabilities.h"
39#define RDPDR_CAPABILITY_HEADER_LENGTH 8
46 GENERAL_CAPABILITY_VERSION_02 };
48 const UINT32 ioCode1 = rdpdr->clientIOCode1 & rdpdr->serverIOCode1;
49 const UINT32 ioCode2 = rdpdr->clientIOCode2 & rdpdr->serverIOCode2;
51 if (rdpdr_write_capset_header(rdpdr->log, s, &header) != CHANNEL_RC_OK)
54 if (!Stream_EnsureRemainingCapacity(s, 36))
57 Stream_Write_UINT32(s, rdpdr->clientOsType);
58 Stream_Write_UINT32(s, rdpdr->clientOsVersion);
59 Stream_Write_UINT16(s, rdpdr->clientVersionMajor);
60 Stream_Write_UINT16(s, rdpdr->clientVersionMinor);
61 Stream_Write_UINT32(s, ioCode1);
62 Stream_Write_UINT32(s, ioCode2);
63 Stream_Write_UINT32(s, rdpdr->clientExtendedPDU);
64 Stream_Write_UINT32(s, rdpdr->clientExtraFlags1);
67 rdpdr->clientExtraFlags2);
69 s, rdpdr->clientSpecialTypeDeviceCap);
80 const BOOL gotV1 = header->Version == GENERAL_CAPABILITY_VERSION_01;
81 const size_t expect = gotV1 ? 32 : 36;
82 if (header->CapabilityLength != expect)
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;
89 if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, expect))
90 return ERROR_INVALID_DATA;
92 rdpdr->serverOsType = Stream_Get_UINT32(s);
93 rdpdr->serverOsVersion = Stream_Get_UINT32(s);
94 rdpdr->serverVersionMajor = Stream_Get_UINT16(s);
95 rdpdr->serverVersionMinor = Stream_Get_UINT16(s);
96 rdpdr->serverIOCode1 = Stream_Get_UINT32(s);
97 rdpdr->serverIOCode2 =
99 rdpdr->serverExtendedPDU = Stream_Get_UINT32(s);
100 rdpdr->serverExtraFlags1 = Stream_Get_UINT32(s);
101 rdpdr->serverExtraFlags2 =
102 Stream_Get_UINT32(s);
104 rdpdr->serverSpecialTypeDeviceCap = 0;
106 rdpdr->serverSpecialTypeDeviceCap = Stream_Get_UINT32(
109 return CHANNEL_RC_OK;
117 PRINT_CAPABILITY_VERSION_01 };
118 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
122static UINT rdpdr_process_printer_capset(WINPR_ATTR_UNUSED
rdpdrPlugin* rdpdr,
wStream* s,
125 WINPR_ASSERT(header);
126 Stream_Seek(s, header->CapabilityLength);
127 return CHANNEL_RC_OK;
135 PORT_CAPABILITY_VERSION_01 };
136 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
140static UINT rdpdr_process_port_capset(WINPR_ATTR_UNUSED
rdpdrPlugin* rdpdr,
wStream* s,
143 WINPR_ASSERT(header);
144 Stream_Seek(s, header->CapabilityLength);
145 return CHANNEL_RC_OK;
153 DRIVE_CAPABILITY_VERSION_02 };
154 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
158static UINT rdpdr_process_drive_capset(WINPR_ATTR_UNUSED
rdpdrPlugin* rdpdr,
wStream* s,
161 WINPR_ASSERT(header);
162 Stream_Seek(s, header->CapabilityLength);
163 return CHANNEL_RC_OK;
171 SMARTCARD_CAPABILITY_VERSION_01 };
172 return rdpdr_write_capset_header(rdpdr->log, s, &header) == CHANNEL_RC_OK;
176static UINT rdpdr_process_smartcard_capset(WINPR_ATTR_UNUSED
rdpdrPlugin* rdpdr,
wStream* s,
179 WINPR_ASSERT(header);
180 Stream_Seek(s, header->CapabilityLength);
181 return CHANNEL_RC_OK;
186 UINT status = CHANNEL_RC_OK;
187 UINT16 numCapabilities = 0;
190 return CHANNEL_RC_NULL_DATA;
192 WINPR_ASSERT(rdpdr->state == RDPDR_CHANNEL_STATE_SERVER_CAPS);
193 rdpdr_state_advance(rdpdr, RDPDR_CHANNEL_STATE_CLIENT_CAPS);
195 if (!Stream_CheckAndLogRequiredLengthWLog(rdpdr->log, s, 4))
196 return ERROR_INVALID_DATA;
198 Stream_Read_UINT16(s, numCapabilities);
201 memset(rdpdr->capabilities, 0,
sizeof(rdpdr->capabilities));
202 for (UINT16 i = 0; i < numCapabilities; i++)
205 UINT error = rdpdr_read_capset_header(rdpdr->log, s, &header);
206 if (error != CHANNEL_RC_OK)
209 switch (header.CapabilityType)
211 case CAP_GENERAL_TYPE:
212 rdpdr->capabilities[header.CapabilityType] = TRUE;
213 status = rdpdr_process_general_capset(rdpdr, s, &header);
216 case CAP_PRINTER_TYPE:
217 rdpdr->capabilities[header.CapabilityType] = TRUE;
218 status = rdpdr_process_printer_capset(rdpdr, s, &header);
222 rdpdr->capabilities[header.CapabilityType] = TRUE;
223 status = rdpdr_process_port_capset(rdpdr, s, &header);
227 rdpdr->capabilities[header.CapabilityType] = TRUE;
228 status = rdpdr_process_drive_capset(rdpdr, s, &header);
231 case CAP_SMARTCARD_TYPE:
232 rdpdr->capabilities[header.CapabilityType] = TRUE;
233 status = rdpdr_process_smartcard_capset(rdpdr, s, &header);
240 if (status != CHANNEL_RC_OK)
244 return CHANNEL_RC_OK;
252UINT rdpdr_send_capability_response(
rdpdrPlugin* rdpdr)
255 WINPR_ASSERT(rdpdr->rdpcontext);
257 rdpSettings* settings = rdpdr->rdpcontext->settings;
258 WINPR_ASSERT(settings);
260 wStream* s = StreamPool_Take(rdpdr->pool, 256);
264 WLog_Print(rdpdr->log, WLOG_ERROR,
"Stream_New failed!");
265 return CHANNEL_RC_NO_MEMORY;
269 freerdp_device_collection_find_type(settings, RDPDR_DTYP_FILESYSTEM);
270 const RDPDR_DEVICE* cserial = freerdp_device_collection_find_type(settings, RDPDR_DTYP_SERIAL);
272 freerdp_device_collection_find_type(settings, RDPDR_DTYP_PARALLEL);
274 freerdp_device_collection_find_type(settings, RDPDR_DTYP_SMARTCARD);
275 const RDPDR_DEVICE* cprinter = freerdp_device_collection_find_type(settings, RDPDR_DTYP_PRINT);
278 const BOOL drives = cdrives && rdpdr->capabilities[CAP_DRIVE_TYPE];
279 const BOOL serial = cserial && rdpdr->capabilities[CAP_PORT_TYPE];
280 const BOOL parallel = cparallel && rdpdr->capabilities[CAP_PORT_TYPE];
281 const BOOL smart = csmart && rdpdr->capabilities[CAP_SMARTCARD_TYPE];
282 const BOOL printer = cprinter && rdpdr->capabilities[CAP_PRINTER_TYPE];
286 if (serial || parallel)
293 Stream_Write_UINT16(s, RDPDR_CTYP_CORE);
294 Stream_Write_UINT16(s, PAKID_CORE_CLIENT_CAPABILITY);
295 Stream_Write_UINT16(s, count);
296 Stream_Write_UINT16(s, 0);
298 if (!rdpdr_write_general_capset(rdpdr, s))
303 if (!rdpdr_write_printer_capset(rdpdr, s))
306 if (serial || parallel)
308 if (!rdpdr_write_port_capset(rdpdr, s))
313 if (!rdpdr_write_drive_capset(rdpdr, s))
318 if (!rdpdr_write_smartcard_capset(rdpdr, s))
321 return rdpdr_send(rdpdr, s);
325 return ERROR_OUTOFMEMORY;