263 lines
7.0 KiB
C
263 lines
7.0 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
* Device Redirection Virtual Channel
|
|
*
|
|
* Copyright 2010-2011 Vic Lee
|
|
* Copyright 2010-2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>
|
|
* Copyright 2015-2016 Thincast Technologies GmbH
|
|
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
|
|
* Copyright 2016 Armin Novak <armin.novak@thincast.com>
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <winpr/crt.h>
|
|
#include <winpr/stream.h>
|
|
|
|
#include "rdpdr_main.h"
|
|
#include "rdpdr_capabilities.h"
|
|
|
|
/* Output device redirection capability set header */
|
|
static void rdpdr_write_capset_header(wStream* s, UINT16 capabilityType, UINT16 capabilityLength, UINT32 version)
|
|
{
|
|
Stream_Write_UINT16(s, capabilityType);
|
|
Stream_Write_UINT16(s, capabilityLength);
|
|
Stream_Write_UINT32(s, version);
|
|
}
|
|
|
|
/* Output device direction general capability set */
|
|
static void rdpdr_write_general_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
rdpdr_write_capset_header(s, CAP_GENERAL_TYPE, 44, GENERAL_CAPABILITY_VERSION_02);
|
|
|
|
Stream_Write_UINT32(s, 0); /* osType, ignored on receipt */
|
|
Stream_Write_UINT32(s, 0); /* osVersion, unused and must be set to zero */
|
|
Stream_Write_UINT16(s, 1); /* protocolMajorVersion, must be set to 1 */
|
|
Stream_Write_UINT16(s, RDPDR_MINOR_RDP_VERSION_5_2); /* protocolMinorVersion */
|
|
Stream_Write_UINT32(s, 0x0000FFFF); /* ioCode1 */
|
|
Stream_Write_UINT32(s, 0); /* ioCode2, must be set to zero, reserved for future use */
|
|
Stream_Write_UINT32(s, RDPDR_DEVICE_REMOVE_PDUS | RDPDR_CLIENT_DISPLAY_NAME_PDU | RDPDR_USER_LOGGEDON_PDU); /* extendedPDU */
|
|
Stream_Write_UINT32(s, ENABLE_ASYNCIO); /* extraFlags1 */
|
|
Stream_Write_UINT32(s, 0); /* extraFlags2, must be set to zero, reserved for future use */
|
|
Stream_Write_UINT32(s, 0); /* SpecialTypeDeviceCap, number of special devices to be redirected before logon */
|
|
}
|
|
|
|
/* Process device direction general capability set */
|
|
static UINT rdpdr_process_general_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT16 capabilityLength;
|
|
|
|
if (Stream_GetRemainingLength(s) < 2)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityLength);
|
|
|
|
if (Stream_GetRemainingLength(s) < capabilityLength - 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(s, capabilityLength - 4);
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/* Output printer direction capability set */
|
|
static void rdpdr_write_printer_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
rdpdr_write_capset_header(s, CAP_PRINTER_TYPE, 8, PRINT_CAPABILITY_VERSION_01);
|
|
}
|
|
|
|
/* Process printer direction capability set */
|
|
static UINT rdpdr_process_printer_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT16 capabilityLength;
|
|
|
|
if (Stream_GetRemainingLength(s) < 2)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityLength);
|
|
|
|
if (Stream_GetRemainingLength(s) < capabilityLength - 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(s, capabilityLength - 4);
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/* Output port redirection capability set */
|
|
static void rdpdr_write_port_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
rdpdr_write_capset_header(s, CAP_PORT_TYPE, 8, PORT_CAPABILITY_VERSION_01);
|
|
}
|
|
|
|
/* Process port redirection capability set */
|
|
static UINT rdpdr_process_port_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT16 capabilityLength;
|
|
|
|
if (Stream_GetRemainingLength(s) < 2)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityLength);
|
|
|
|
if (Stream_GetRemainingLength(s) < capabilityLength - 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(s, capabilityLength - 4);
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/* Output drive redirection capability set */
|
|
static void rdpdr_write_drive_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
rdpdr_write_capset_header(s, CAP_DRIVE_TYPE, 8, DRIVE_CAPABILITY_VERSION_02);
|
|
}
|
|
|
|
/* Process drive redirection capability set */
|
|
static UINT rdpdr_process_drive_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT16 capabilityLength;
|
|
|
|
if (Stream_GetRemainingLength(s) < 2)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityLength);
|
|
|
|
if (Stream_GetRemainingLength(s) < capabilityLength - 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(s, capabilityLength - 4);
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/* Output smart card redirection capability set */
|
|
static void rdpdr_write_smartcard_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
rdpdr_write_capset_header(s, CAP_SMARTCARD_TYPE, 8, SMARTCARD_CAPABILITY_VERSION_01);
|
|
}
|
|
|
|
/* Process smartcard redirection capability set */
|
|
static UINT rdpdr_process_smartcard_capset(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT16 capabilityLength;
|
|
|
|
if (Stream_GetRemainingLength(s) < 2)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityLength);
|
|
|
|
if (Stream_GetRemainingLength(s) < capabilityLength - 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(s, capabilityLength - 4);
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
UINT rdpdr_process_capability_request(rdpdrPlugin* rdpdr, wStream* s)
|
|
{
|
|
UINT status = CHANNEL_RC_OK;
|
|
UINT16 i;
|
|
UINT16 numCapabilities;
|
|
UINT16 capabilityType;
|
|
|
|
if (!rdpdr || !s)
|
|
return CHANNEL_RC_NULL_DATA;
|
|
|
|
if (Stream_GetRemainingLength(s) < 4)
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, numCapabilities);
|
|
Stream_Seek(s, 2); /* pad (2 bytes) */
|
|
|
|
for (i = 0; i < numCapabilities; i++)
|
|
{
|
|
if (Stream_GetRemainingLength(s) < sizeof(UINT16))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT16(s, capabilityType);
|
|
|
|
switch (capabilityType)
|
|
{
|
|
case CAP_GENERAL_TYPE:
|
|
status = rdpdr_process_general_capset(rdpdr, s);
|
|
break;
|
|
|
|
case CAP_PRINTER_TYPE:
|
|
status = rdpdr_process_printer_capset(rdpdr, s);
|
|
break;
|
|
|
|
case CAP_PORT_TYPE:
|
|
status = rdpdr_process_port_capset(rdpdr, s);
|
|
break;
|
|
|
|
case CAP_DRIVE_TYPE:
|
|
status = rdpdr_process_drive_capset(rdpdr, s);
|
|
break;
|
|
|
|
case CAP_SMARTCARD_TYPE:
|
|
status = rdpdr_process_smartcard_capset(rdpdr, s);
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if (status != CHANNEL_RC_OK)
|
|
return status;
|
|
}
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
UINT rdpdr_send_capability_response(rdpdrPlugin* rdpdr)
|
|
{
|
|
wStream* s;
|
|
|
|
s = Stream_New(NULL, 256);
|
|
if (!s)
|
|
{
|
|
WLog_ERR(TAG, "Stream_New failed!");
|
|
return CHANNEL_RC_NO_MEMORY;
|
|
}
|
|
|
|
Stream_Write_UINT16(s, RDPDR_CTYP_CORE);
|
|
Stream_Write_UINT16(s, PAKID_CORE_CLIENT_CAPABILITY);
|
|
|
|
Stream_Write_UINT16(s, 5); /* numCapabilities */
|
|
Stream_Write_UINT16(s, 0); /* pad */
|
|
|
|
rdpdr_write_general_capset(rdpdr, s);
|
|
rdpdr_write_printer_capset(rdpdr, s);
|
|
rdpdr_write_port_capset(rdpdr, s);
|
|
rdpdr_write_drive_capset(rdpdr, s);
|
|
rdpdr_write_smartcard_capset(rdpdr, s);
|
|
|
|
return rdpdr_send(rdpdr, s);
|
|
}
|