980 lines
28 KiB
C
980 lines
28 KiB
C
/**
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
|
* Serial Port Device Service Virtual Channel
|
|
*
|
|
* Copyright 2011 O.S. Systems Software Ltda.
|
|
* Copyright 2011 Eduardo Fiss Beloni <beloni@ossystems.com.br>
|
|
* Copyright 2014 Hewlett-Packard Development Company, L.P.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <freerdp/config.h>
|
|
|
|
#include <winpr/assert.h>
|
|
#include <errno.h>
|
|
#include <stdio.h>
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <winpr/collections.h>
|
|
#include <winpr/comm.h>
|
|
#include <winpr/crt.h>
|
|
#include <winpr/stream.h>
|
|
#include <winpr/synch.h>
|
|
#include <winpr/thread.h>
|
|
#include <winpr/wlog.h>
|
|
|
|
#include <freerdp/freerdp.h>
|
|
#include <freerdp/channels/rdpdr.h>
|
|
#include <freerdp/channels/log.h>
|
|
#include <freerdp/utils/rdpdr_utils.h>
|
|
|
|
#define TAG CHANNELS_TAG("serial.client")
|
|
|
|
/* TODO: all #ifdef __linux__ could be removed once only some generic
|
|
* functions will be used. Replace CommReadFile by ReadFile,
|
|
* CommWriteFile by WriteFile etc.. */
|
|
#if defined __linux__ && !defined ANDROID
|
|
|
|
#define MAX_IRP_THREADS 5
|
|
|
|
typedef struct
|
|
{
|
|
DEVICE device;
|
|
BOOL permissive;
|
|
SERIAL_DRIVER_ID ServerSerialDriverId;
|
|
HANDLE* hComm;
|
|
|
|
wLog* log;
|
|
HANDLE MainThread;
|
|
wMessageQueue* MainIrpQueue;
|
|
|
|
/* one thread per pending IRP and indexed according their CompletionId */
|
|
wListDictionary* IrpThreads;
|
|
UINT32 IrpThreadToBeTerminatedCount;
|
|
CRITICAL_SECTION TerminatingIrpThreadsLock;
|
|
rdpContext* rdpcontext;
|
|
} SERIAL_DEVICE;
|
|
|
|
typedef struct
|
|
{
|
|
SERIAL_DEVICE* serial;
|
|
IRP* irp;
|
|
} IRP_THREAD_DATA;
|
|
|
|
static UINT32 _GetLastErrorToIoStatus(SERIAL_DEVICE* serial)
|
|
{
|
|
/* http://msdn.microsoft.com/en-us/library/ff547466%28v=vs.85%29.aspx#generic_status_values_for_serial_device_control_requests
|
|
*/
|
|
switch (GetLastError())
|
|
{
|
|
case ERROR_BAD_DEVICE:
|
|
return STATUS_INVALID_DEVICE_REQUEST;
|
|
|
|
case ERROR_CALL_NOT_IMPLEMENTED:
|
|
return STATUS_NOT_IMPLEMENTED;
|
|
|
|
case ERROR_CANCELLED:
|
|
return STATUS_CANCELLED;
|
|
|
|
case ERROR_INSUFFICIENT_BUFFER:
|
|
return STATUS_BUFFER_TOO_SMALL; /* NB: STATUS_BUFFER_SIZE_TOO_SMALL not defined */
|
|
|
|
case ERROR_INVALID_DEVICE_OBJECT_PARAMETER: /* eg: SerCx2.sys' _purge() */
|
|
return STATUS_INVALID_DEVICE_STATE;
|
|
|
|
case ERROR_INVALID_HANDLE:
|
|
return STATUS_INVALID_DEVICE_REQUEST;
|
|
|
|
case ERROR_INVALID_PARAMETER:
|
|
return STATUS_INVALID_PARAMETER;
|
|
|
|
case ERROR_IO_DEVICE:
|
|
return STATUS_IO_DEVICE_ERROR;
|
|
|
|
case ERROR_IO_PENDING:
|
|
return STATUS_PENDING;
|
|
|
|
case ERROR_NOT_SUPPORTED:
|
|
return STATUS_NOT_SUPPORTED;
|
|
|
|
case ERROR_TIMEOUT:
|
|
return STATUS_TIMEOUT;
|
|
/* no default */
|
|
}
|
|
|
|
WLog_Print(serial->log, WLOG_DEBUG, "unexpected last-error: 0x%08" PRIX32 "", GetLastError());
|
|
return STATUS_UNSUCCESSFUL;
|
|
}
|
|
|
|
static UINT serial_process_irp_create(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
DWORD DesiredAccess;
|
|
DWORD SharedAccess;
|
|
DWORD CreateDisposition;
|
|
UINT32 PathLength;
|
|
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, 32))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT32(irp->input, DesiredAccess); /* DesiredAccess (4 bytes) */
|
|
Stream_Seek_UINT64(irp->input); /* AllocationSize (8 bytes) */
|
|
Stream_Seek_UINT32(irp->input); /* FileAttributes (4 bytes) */
|
|
Stream_Read_UINT32(irp->input, SharedAccess); /* SharedAccess (4 bytes) */
|
|
Stream_Read_UINT32(irp->input, CreateDisposition); /* CreateDisposition (4 bytes) */
|
|
Stream_Seek_UINT32(irp->input); /* CreateOptions (4 bytes) */
|
|
Stream_Read_UINT32(irp->input, PathLength); /* PathLength (4 bytes) */
|
|
|
|
if (!Stream_SafeSeek(irp->input, PathLength)) /* Path (variable) */
|
|
return ERROR_INVALID_DATA;
|
|
|
|
WINPR_ASSERT(PathLength == 0); /* MS-RDPESP 2.2.2.2 */
|
|
#ifndef _WIN32
|
|
/* Windows 2012 server sends on a first call :
|
|
* DesiredAccess = 0x00100080: SYNCHRONIZE | FILE_READ_ATTRIBUTES
|
|
* SharedAccess = 0x00000007: FILE_SHARE_DELETE | FILE_SHARE_WRITE | FILE_SHARE_READ
|
|
* CreateDisposition = 0x00000001: CREATE_NEW
|
|
*
|
|
* then Windows 2012 sends :
|
|
* DesiredAccess = 0x00120089: SYNCHRONIZE | READ_CONTROL | FILE_READ_ATTRIBUTES |
|
|
* FILE_READ_EA | FILE_READ_DATA SharedAccess = 0x00000007: FILE_SHARE_DELETE |
|
|
* FILE_SHARE_WRITE | FILE_SHARE_READ CreateDisposition = 0x00000001: CREATE_NEW
|
|
*
|
|
* WINPR_ASSERT(DesiredAccess == (GENERIC_READ | GENERIC_WRITE));
|
|
* WINPR_ASSERT(SharedAccess == 0);
|
|
* WINPR_ASSERT(CreateDisposition == OPEN_EXISTING);
|
|
*
|
|
*/
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"DesiredAccess: 0x%" PRIX32 ", SharedAccess: 0x%" PRIX32
|
|
", CreateDisposition: 0x%" PRIX32 "",
|
|
DesiredAccess, SharedAccess, CreateDisposition);
|
|
/* FIXME: As of today only the flags below are supported by CommCreateFileA: */
|
|
DesiredAccess = GENERIC_READ | GENERIC_WRITE;
|
|
SharedAccess = 0;
|
|
CreateDisposition = OPEN_EXISTING;
|
|
#endif
|
|
serial->hComm =
|
|
CreateFile(serial->device.name, DesiredAccess, SharedAccess, NULL, /* SecurityAttributes */
|
|
CreateDisposition, 0, /* FlagsAndAttributes */
|
|
NULL); /* TemplateFile */
|
|
|
|
if (!serial->hComm || (serial->hComm == INVALID_HANDLE_VALUE))
|
|
{
|
|
WLog_Print(serial->log, WLOG_WARN, "CreateFile failure: %s last-error: 0x%08" PRIX32 "",
|
|
serial->device.name, GetLastError());
|
|
irp->IoStatus = STATUS_UNSUCCESSFUL;
|
|
goto error_handle;
|
|
}
|
|
|
|
_comm_setServerSerialDriver(serial->hComm, serial->ServerSerialDriverId);
|
|
_comm_set_permissive(serial->hComm, serial->permissive);
|
|
/* NOTE: binary mode/raw mode required for the redirection. On
|
|
* Linux, CommCreateFileA forces this setting.
|
|
*/
|
|
/* ZeroMemory(&dcb, sizeof(DCB)); */
|
|
/* dcb.DCBlength = sizeof(DCB); */
|
|
/* GetCommState(serial->hComm, &dcb); */
|
|
/* dcb.fBinary = TRUE; */
|
|
/* SetCommState(serial->hComm, &dcb); */
|
|
WINPR_ASSERT(irp->FileId == 0);
|
|
irp->FileId = irp->devman->id_sequence++; /* FIXME: why not ((WINPR_COMM*)hComm)->fd? */
|
|
irp->IoStatus = STATUS_SUCCESS;
|
|
WLog_Print(serial->log, WLOG_DEBUG, "%s (DeviceId: %" PRIu32 ", FileId: %" PRIu32 ") created.",
|
|
serial->device.name, irp->device->id, irp->FileId);
|
|
error_handle:
|
|
Stream_Write_UINT32(irp->output, irp->FileId); /* FileId (4 bytes) */
|
|
Stream_Write_UINT8(irp->output, 0); /* Information (1 byte) */
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
static UINT serial_process_irp_close(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, 32))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Seek(irp->input, 32); /* Padding (32 bytes) */
|
|
|
|
if (!CloseHandle(serial->hComm))
|
|
{
|
|
WLog_Print(serial->log, WLOG_WARN, "CloseHandle failure: %s (%" PRIu32 ") closed.",
|
|
serial->device.name, irp->device->id);
|
|
irp->IoStatus = STATUS_UNSUCCESSFUL;
|
|
goto error_handle;
|
|
}
|
|
|
|
WLog_Print(serial->log, WLOG_DEBUG, "%s (DeviceId: %" PRIu32 ", FileId: %" PRIu32 ") closed.",
|
|
serial->device.name, irp->device->id, irp->FileId);
|
|
serial->hComm = NULL;
|
|
irp->IoStatus = STATUS_SUCCESS;
|
|
error_handle:
|
|
Stream_Zero(irp->output, 5); /* Padding (5 bytes) */
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT serial_process_irp_read(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
UINT32 Length;
|
|
UINT64 Offset;
|
|
BYTE* buffer = NULL;
|
|
DWORD nbRead = 0;
|
|
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, 32))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT32(irp->input, Length); /* Length (4 bytes) */
|
|
Stream_Read_UINT64(irp->input, Offset); /* Offset (8 bytes) */
|
|
Stream_Seek(irp->input, 20); /* Padding (20 bytes) */
|
|
buffer = (BYTE*)calloc(Length, sizeof(BYTE));
|
|
|
|
if (buffer == NULL)
|
|
{
|
|
irp->IoStatus = STATUS_NO_MEMORY;
|
|
goto error_handle;
|
|
}
|
|
|
|
/* MS-RDPESP 3.2.5.1.4: If the Offset field is not set to 0, the value MUST be ignored
|
|
* WINPR_ASSERT(Offset == 0);
|
|
*/
|
|
WLog_Print(serial->log, WLOG_DEBUG, "reading %" PRIu32 " bytes from %s", Length,
|
|
serial->device.name);
|
|
|
|
/* FIXME: CommReadFile to be replaced by ReadFile */
|
|
if (CommReadFile(serial->hComm, buffer, Length, &nbRead, NULL))
|
|
{
|
|
irp->IoStatus = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"read failure to %s, nbRead=%" PRIu32 ", last-error: 0x%08" PRIX32 "",
|
|
serial->device.name, nbRead, GetLastError());
|
|
irp->IoStatus = _GetLastErrorToIoStatus(serial);
|
|
}
|
|
|
|
WLog_Print(serial->log, WLOG_DEBUG, "%" PRIu32 " bytes read from %s", nbRead,
|
|
serial->device.name);
|
|
error_handle:
|
|
Stream_Write_UINT32(irp->output, nbRead); /* Length (4 bytes) */
|
|
|
|
if (nbRead > 0)
|
|
{
|
|
if (!Stream_EnsureRemainingCapacity(irp->output, nbRead))
|
|
{
|
|
WLog_ERR(TAG, "Stream_EnsureRemainingCapacity failed!");
|
|
free(buffer);
|
|
return CHANNEL_RC_NO_MEMORY;
|
|
}
|
|
|
|
Stream_Write(irp->output, buffer, nbRead); /* ReadData */
|
|
}
|
|
|
|
free(buffer);
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
static UINT serial_process_irp_write(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
UINT32 Length;
|
|
UINT64 Offset;
|
|
void* ptr;
|
|
DWORD nbWritten = 0;
|
|
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, 32))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT32(irp->input, Length); /* Length (4 bytes) */
|
|
Stream_Read_UINT64(irp->input, Offset); /* Offset (8 bytes) */
|
|
if (!Stream_SafeSeek(irp->input, 20)) /* Padding (20 bytes) */
|
|
return ERROR_INVALID_DATA;
|
|
|
|
/* MS-RDPESP 3.2.5.1.5: The Offset field is ignored
|
|
* WINPR_ASSERT(Offset == 0);
|
|
*
|
|
* Using a serial printer, noticed though this field could be
|
|
* set.
|
|
*/
|
|
WLog_Print(serial->log, WLOG_DEBUG, "writing %" PRIu32 " bytes to %s", Length,
|
|
serial->device.name);
|
|
|
|
ptr = Stream_Pointer(irp->input);
|
|
if (!Stream_SafeSeek(irp->input, Length))
|
|
return ERROR_INVALID_DATA;
|
|
/* FIXME: CommWriteFile to be replaced by WriteFile */
|
|
if (CommWriteFile(serial->hComm, ptr, Length, &nbWritten, NULL))
|
|
{
|
|
irp->IoStatus = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"write failure to %s, nbWritten=%" PRIu32 ", last-error: 0x%08" PRIX32 "",
|
|
serial->device.name, nbWritten, GetLastError());
|
|
irp->IoStatus = _GetLastErrorToIoStatus(serial);
|
|
}
|
|
|
|
WLog_Print(serial->log, WLOG_DEBUG, "%" PRIu32 " bytes written to %s", nbWritten,
|
|
serial->device.name);
|
|
Stream_Write_UINT32(irp->output, nbWritten); /* Length (4 bytes) */
|
|
Stream_Write_UINT8(irp->output, 0); /* Padding (1 byte) */
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT serial_process_irp_device_control(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
UINT32 IoControlCode;
|
|
UINT32 InputBufferLength;
|
|
BYTE* InputBuffer = NULL;
|
|
UINT32 OutputBufferLength;
|
|
BYTE* OutputBuffer = NULL;
|
|
DWORD BytesReturned = 0;
|
|
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, 32))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
Stream_Read_UINT32(irp->input, OutputBufferLength); /* OutputBufferLength (4 bytes) */
|
|
Stream_Read_UINT32(irp->input, InputBufferLength); /* InputBufferLength (4 bytes) */
|
|
Stream_Read_UINT32(irp->input, IoControlCode); /* IoControlCode (4 bytes) */
|
|
Stream_Seek(irp->input, 20); /* Padding (20 bytes) */
|
|
|
|
if (!Stream_CheckAndLogRequiredLength(TAG, irp->input, InputBufferLength))
|
|
return ERROR_INVALID_DATA;
|
|
|
|
OutputBuffer = (BYTE*)calloc(OutputBufferLength, sizeof(BYTE));
|
|
|
|
if (OutputBuffer == NULL)
|
|
{
|
|
irp->IoStatus = STATUS_NO_MEMORY;
|
|
goto error_handle;
|
|
}
|
|
|
|
InputBuffer = (BYTE*)calloc(InputBufferLength, sizeof(BYTE));
|
|
|
|
if (InputBuffer == NULL)
|
|
{
|
|
irp->IoStatus = STATUS_NO_MEMORY;
|
|
goto error_handle;
|
|
}
|
|
|
|
Stream_Read(irp->input, InputBuffer, InputBufferLength);
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"CommDeviceIoControl: CompletionId=%" PRIu32 ", IoControlCode=[0x%" PRIX32 "] %s",
|
|
irp->CompletionId, IoControlCode, _comm_serial_ioctl_name(IoControlCode));
|
|
|
|
/* FIXME: CommDeviceIoControl to be replaced by DeviceIoControl() */
|
|
if (CommDeviceIoControl(serial->hComm, IoControlCode, InputBuffer, InputBufferLength,
|
|
OutputBuffer, OutputBufferLength, &BytesReturned, NULL))
|
|
{
|
|
/* WLog_Print(serial->log, WLOG_DEBUG, "CommDeviceIoControl: CompletionId=%"PRIu32",
|
|
* IoControlCode=[0x%"PRIX32"] %s done", irp->CompletionId, IoControlCode,
|
|
* _comm_serial_ioctl_name(IoControlCode)); */
|
|
irp->IoStatus = STATUS_SUCCESS;
|
|
}
|
|
else
|
|
{
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"CommDeviceIoControl failure: IoControlCode=[0x%" PRIX32
|
|
"] %s, last-error: 0x%08" PRIX32 "",
|
|
IoControlCode, _comm_serial_ioctl_name(IoControlCode), GetLastError());
|
|
irp->IoStatus = _GetLastErrorToIoStatus(serial);
|
|
}
|
|
|
|
error_handle:
|
|
/* FIXME: find out whether it's required or not to get
|
|
* BytesReturned == OutputBufferLength when
|
|
* CommDeviceIoControl returns FALSE */
|
|
WINPR_ASSERT(OutputBufferLength == BytesReturned);
|
|
Stream_Write_UINT32(irp->output, BytesReturned); /* OutputBufferLength (4 bytes) */
|
|
|
|
if (BytesReturned > 0)
|
|
{
|
|
if (!Stream_EnsureRemainingCapacity(irp->output, BytesReturned))
|
|
{
|
|
WLog_ERR(TAG, "Stream_EnsureRemainingCapacity failed!");
|
|
free(InputBuffer);
|
|
free(OutputBuffer);
|
|
return CHANNEL_RC_NO_MEMORY;
|
|
}
|
|
|
|
Stream_Write(irp->output, OutputBuffer, BytesReturned); /* OutputBuffer */
|
|
}
|
|
|
|
/* FIXME: Why at least Windows 2008R2 gets lost with this
|
|
* extra byte and likely on a IOCTL_SERIAL_SET_BAUD_RATE? The
|
|
* extra byte is well required according MS-RDPEFS
|
|
* 2.2.1.5.5 */
|
|
/* else */
|
|
/* { */
|
|
/* Stream_Write_UINT8(irp->output, 0); /\* Padding (1 byte) *\/ */
|
|
/* } */
|
|
free(InputBuffer);
|
|
free(OutputBuffer);
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT serial_process_irp(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
UINT error = CHANNEL_RC_OK;
|
|
WLog_Print(serial->log, WLOG_DEBUG, "IRP MajorFunction: s, MinorFunction: 0x%08" PRIX32 "\n",
|
|
rdpdr_irp_string(irp->MajorFunction), irp->MinorFunction);
|
|
|
|
switch (irp->MajorFunction)
|
|
{
|
|
case IRP_MJ_CREATE:
|
|
error = serial_process_irp_create(serial, irp);
|
|
break;
|
|
|
|
case IRP_MJ_CLOSE:
|
|
error = serial_process_irp_close(serial, irp);
|
|
break;
|
|
|
|
case IRP_MJ_READ:
|
|
if ((error = serial_process_irp_read(serial, irp)))
|
|
WLog_ERR(TAG, "serial_process_irp_read failed with error %" PRIu32 "!", error);
|
|
|
|
break;
|
|
|
|
case IRP_MJ_WRITE:
|
|
error = serial_process_irp_write(serial, irp);
|
|
break;
|
|
|
|
case IRP_MJ_DEVICE_CONTROL:
|
|
if ((error = serial_process_irp_device_control(serial, irp)))
|
|
WLog_ERR(TAG, "serial_process_irp_device_control failed with error %" PRIu32 "!",
|
|
error);
|
|
|
|
break;
|
|
|
|
default:
|
|
irp->IoStatus = STATUS_NOT_SUPPORTED;
|
|
break;
|
|
}
|
|
|
|
return error;
|
|
}
|
|
|
|
static DWORD WINAPI irp_thread_func(LPVOID arg)
|
|
{
|
|
IRP_THREAD_DATA* data = (IRP_THREAD_DATA*)arg;
|
|
UINT error;
|
|
|
|
/* blocks until the end of the request */
|
|
if ((error = serial_process_irp(data->serial, data->irp)))
|
|
{
|
|
WLog_ERR(TAG, "serial_process_irp failed with error %" PRIu32 "", error);
|
|
goto error_out;
|
|
}
|
|
|
|
EnterCriticalSection(&data->serial->TerminatingIrpThreadsLock);
|
|
data->serial->IrpThreadToBeTerminatedCount++;
|
|
error = data->irp->Complete(data->irp);
|
|
LeaveCriticalSection(&data->serial->TerminatingIrpThreadsLock);
|
|
error_out:
|
|
|
|
if (error && data->serial->rdpcontext)
|
|
setChannelError(data->serial->rdpcontext, error, "irp_thread_func reported an error");
|
|
|
|
/* NB: At this point, the server might already being reusing
|
|
* the CompletionId whereas the thread is not yet
|
|
* terminated */
|
|
free(data);
|
|
ExitThread(error);
|
|
return error;
|
|
}
|
|
|
|
static void create_irp_thread(SERIAL_DEVICE* serial, IRP* irp)
|
|
{
|
|
IRP_THREAD_DATA* data = NULL;
|
|
HANDLE irpThread;
|
|
HANDLE previousIrpThread;
|
|
uintptr_t key;
|
|
/* for a test/debug purpose, uncomment the code below to get a
|
|
* single thread for all IRPs. NB: two IRPs could not be
|
|
* processed at the same time, typically two concurent
|
|
* Read/Write operations could block each other. */
|
|
/* serial_process_irp(serial, irp); */
|
|
/* irp->Complete(irp); */
|
|
/* return; */
|
|
/* NOTE: for good or bad, this implementation relies on the
|
|
* server to avoid a flooding of requests. see also _purge().
|
|
*/
|
|
EnterCriticalSection(&serial->TerminatingIrpThreadsLock);
|
|
|
|
while (serial->IrpThreadToBeTerminatedCount > 0)
|
|
{
|
|
/* Cleaning up termitating and pending irp
|
|
* threads. See also: irp_thread_func() */
|
|
HANDLE cirpThread;
|
|
ULONG_PTR* ids;
|
|
int i, nbIds;
|
|
nbIds = ListDictionary_GetKeys(serial->IrpThreads, &ids);
|
|
|
|
for (i = 0; i < nbIds; i++)
|
|
{
|
|
/* Checking if ids[i] is terminating or pending */
|
|
DWORD waitResult;
|
|
ULONG_PTR id = ids[i];
|
|
cirpThread = ListDictionary_GetItemValue(serial->IrpThreads, (void*)id);
|
|
/* FIXME: not quite sure a zero timeout is a good thing to check whether a thread is
|
|
* stil alived or not */
|
|
waitResult = WaitForSingleObject(cirpThread, 0);
|
|
|
|
if (waitResult == WAIT_OBJECT_0)
|
|
{
|
|
/* terminating thread */
|
|
/* WLog_Print(serial->log, WLOG_DEBUG, "IRP thread with CompletionId=%"PRIuz"
|
|
* naturally died", id); */
|
|
CloseHandle(cirpThread);
|
|
ListDictionary_Remove(serial->IrpThreads, (void*)id);
|
|
serial->IrpThreadToBeTerminatedCount--;
|
|
}
|
|
else if (waitResult != WAIT_TIMEOUT)
|
|
{
|
|
/* unexpected thread state */
|
|
WLog_Print(serial->log, WLOG_WARN,
|
|
"WaitForSingleObject, got an unexpected result=0x%" PRIX32 "\n",
|
|
waitResult);
|
|
WINPR_ASSERT(FALSE);
|
|
}
|
|
|
|
/* pending thread (but not yet terminating thread) if waitResult == WAIT_TIMEOUT */
|
|
}
|
|
|
|
if (serial->IrpThreadToBeTerminatedCount > 0)
|
|
{
|
|
WLog_Print(serial->log, WLOG_DEBUG, "%" PRIu32 " IRP thread(s) not yet terminated",
|
|
serial->IrpThreadToBeTerminatedCount);
|
|
Sleep(1); /* 1 ms */
|
|
}
|
|
|
|
free(ids);
|
|
}
|
|
|
|
LeaveCriticalSection(&serial->TerminatingIrpThreadsLock);
|
|
/* NB: At this point and thanks to the synchronization we're
|
|
* sure that the incoming IRP uses well a recycled
|
|
* CompletionId or the server sent again an IRP already posted
|
|
* which didn't get yet a response (this later server behavior
|
|
* at least observed with IOCTL_SERIAL_WAIT_ON_MASK and
|
|
* mstsc.exe).
|
|
*
|
|
* FIXME: behavior documented somewhere? behavior not yet
|
|
* observed with FreeRDP).
|
|
*/
|
|
key = irp->CompletionId;
|
|
previousIrpThread = ListDictionary_GetItemValue(serial->IrpThreads, (void*)key);
|
|
|
|
if (previousIrpThread)
|
|
{
|
|
/* Thread still alived <=> Request still pending */
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"IRP recall: IRP with the CompletionId=%" PRIu32 " not yet completed!",
|
|
irp->CompletionId);
|
|
WINPR_ASSERT(FALSE); /* unimplemented */
|
|
/* TODO: WINPR_ASSERTs that previousIrpThread handles well
|
|
* the same request by checking more details. Need an
|
|
* access to the IRP object used by previousIrpThread
|
|
*/
|
|
/* TODO: taking over the pending IRP or sending a kind
|
|
* of wake up signal to accelerate the pending
|
|
* request
|
|
*
|
|
* To be considered:
|
|
* if (IoControlCode == IOCTL_SERIAL_WAIT_ON_MASK) {
|
|
* pComm->PendingEvents |= SERIAL_EV_FREERDP_*;
|
|
* }
|
|
*/
|
|
irp->Discard(irp);
|
|
return;
|
|
}
|
|
|
|
if (ListDictionary_Count(serial->IrpThreads) >= MAX_IRP_THREADS)
|
|
{
|
|
WLog_Print(serial->log, WLOG_WARN,
|
|
"Number of IRP threads threshold reached: %d, keep on anyway",
|
|
ListDictionary_Count(serial->IrpThreads));
|
|
WINPR_ASSERT(FALSE); /* unimplemented */
|
|
/* TODO: MAX_IRP_THREADS has been thought to avoid a
|
|
* flooding of pending requests. Use
|
|
* WaitForMultipleObjects() when available in winpr
|
|
* for threads.
|
|
*/
|
|
}
|
|
|
|
/* error_handle to be used ... */
|
|
data = (IRP_THREAD_DATA*)calloc(1, sizeof(IRP_THREAD_DATA));
|
|
|
|
if (data == NULL)
|
|
{
|
|
WLog_Print(serial->log, WLOG_WARN, "Could not allocate a new IRP_THREAD_DATA.");
|
|
goto error_handle;
|
|
}
|
|
|
|
data->serial = serial;
|
|
data->irp = irp;
|
|
/* data freed by irp_thread_func */
|
|
irpThread = CreateThread(NULL, 0, irp_thread_func, (void*)data, 0, NULL);
|
|
|
|
if (irpThread == INVALID_HANDLE_VALUE)
|
|
{
|
|
WLog_Print(serial->log, WLOG_WARN, "Could not allocate a new IRP thread.");
|
|
goto error_handle;
|
|
}
|
|
|
|
key = irp->CompletionId;
|
|
|
|
if (!ListDictionary_Add(serial->IrpThreads, (void*)key, irpThread))
|
|
{
|
|
WLog_ERR(TAG, "ListDictionary_Add failed!");
|
|
goto error_handle;
|
|
}
|
|
|
|
return;
|
|
error_handle:
|
|
irp->IoStatus = STATUS_NO_MEMORY;
|
|
irp->Complete(irp);
|
|
free(data);
|
|
}
|
|
|
|
static void terminate_pending_irp_threads(SERIAL_DEVICE* serial)
|
|
{
|
|
ULONG_PTR* ids;
|
|
int i, nbIds;
|
|
nbIds = ListDictionary_GetKeys(serial->IrpThreads, &ids);
|
|
WLog_Print(serial->log, WLOG_DEBUG, "Terminating %d IRP thread(s)", nbIds);
|
|
|
|
for (i = 0; i < nbIds; i++)
|
|
{
|
|
HANDLE irpThread;
|
|
ULONG_PTR id = ids[i];
|
|
irpThread = ListDictionary_GetItemValue(serial->IrpThreads, (void*)id);
|
|
TerminateThread(irpThread, 0);
|
|
if (WaitForSingleObject(irpThread, INFINITE) == WAIT_FAILED)
|
|
{
|
|
WLog_ERR(TAG, "WaitForSingleObject failed!");
|
|
continue;
|
|
}
|
|
|
|
CloseHandle(irpThread);
|
|
WLog_Print(serial->log, WLOG_DEBUG, "IRP thread terminated, CompletionId %p", (void*)id);
|
|
}
|
|
|
|
ListDictionary_Clear(serial->IrpThreads);
|
|
free(ids);
|
|
}
|
|
|
|
static DWORD WINAPI serial_thread_func(LPVOID arg)
|
|
{
|
|
IRP* irp;
|
|
wMessage message;
|
|
SERIAL_DEVICE* serial = (SERIAL_DEVICE*)arg;
|
|
UINT error = CHANNEL_RC_OK;
|
|
|
|
while (1)
|
|
{
|
|
if (!MessageQueue_Wait(serial->MainIrpQueue))
|
|
{
|
|
WLog_ERR(TAG, "MessageQueue_Wait failed!");
|
|
error = ERROR_INTERNAL_ERROR;
|
|
break;
|
|
}
|
|
|
|
if (!MessageQueue_Peek(serial->MainIrpQueue, &message, TRUE))
|
|
{
|
|
WLog_ERR(TAG, "MessageQueue_Peek failed!");
|
|
error = ERROR_INTERNAL_ERROR;
|
|
break;
|
|
}
|
|
|
|
if (message.id == WMQ_QUIT)
|
|
{
|
|
terminate_pending_irp_threads(serial);
|
|
break;
|
|
}
|
|
|
|
irp = (IRP*)message.wParam;
|
|
|
|
if (irp)
|
|
create_irp_thread(serial, irp);
|
|
}
|
|
|
|
if (error && serial->rdpcontext)
|
|
setChannelError(serial->rdpcontext, error, "serial_thread_func reported an error");
|
|
|
|
ExitThread(error);
|
|
return error;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT serial_irp_request(DEVICE* device, IRP* irp)
|
|
{
|
|
SERIAL_DEVICE* serial = (SERIAL_DEVICE*)device;
|
|
WINPR_ASSERT(irp != NULL);
|
|
|
|
if (irp == NULL)
|
|
return CHANNEL_RC_OK;
|
|
|
|
/* NB: ENABLE_ASYNCIO is set, (MS-RDPEFS 2.2.2.7.2) this
|
|
* allows the server to send multiple simultaneous read or
|
|
* write requests.
|
|
*/
|
|
|
|
if (!MessageQueue_Post(serial->MainIrpQueue, NULL, 0, (void*)irp, NULL))
|
|
{
|
|
WLog_ERR(TAG, "MessageQueue_Post failed!");
|
|
return ERROR_INTERNAL_ERROR;
|
|
}
|
|
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT serial_free(DEVICE* device)
|
|
{
|
|
UINT error;
|
|
SERIAL_DEVICE* serial = (SERIAL_DEVICE*)device;
|
|
WLog_Print(serial->log, WLOG_DEBUG, "freeing");
|
|
MessageQueue_PostQuit(serial->MainIrpQueue, 0);
|
|
|
|
if (WaitForSingleObject(serial->MainThread, INFINITE) == WAIT_FAILED)
|
|
{
|
|
error = GetLastError();
|
|
WLog_ERR(TAG, "WaitForSingleObject failed with error %" PRIu32 "!", error);
|
|
return error;
|
|
}
|
|
|
|
CloseHandle(serial->MainThread);
|
|
|
|
if (serial->hComm)
|
|
CloseHandle(serial->hComm);
|
|
|
|
/* Clean up resources */
|
|
Stream_Free(serial->device.data, TRUE);
|
|
MessageQueue_Free(serial->MainIrpQueue);
|
|
ListDictionary_Free(serial->IrpThreads);
|
|
DeleteCriticalSection(&serial->TerminatingIrpThreadsLock);
|
|
free(serial);
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
#endif /* __linux__ */
|
|
|
|
static void serial_message_free(void* obj)
|
|
{
|
|
wMessage* msg = obj;
|
|
if (!msg)
|
|
return;
|
|
if (msg->id != 0)
|
|
return;
|
|
|
|
IRP* irp = (IRP*)msg->wParam;
|
|
if (!irp)
|
|
return;
|
|
WINPR_ASSERT(irp->Discard);
|
|
irp->Discard(irp);
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
UINT serial_DeviceServiceEntry(PDEVICE_SERVICE_ENTRY_POINTS pEntryPoints)
|
|
{
|
|
char* name;
|
|
char* path;
|
|
char* driver;
|
|
RDPDR_SERIAL* device;
|
|
#if defined __linux__ && !defined ANDROID
|
|
size_t i, len;
|
|
SERIAL_DEVICE* serial;
|
|
#endif /* __linux__ */
|
|
UINT error = CHANNEL_RC_OK;
|
|
|
|
WINPR_ASSERT(pEntryPoints);
|
|
|
|
device = (RDPDR_SERIAL*)pEntryPoints->device;
|
|
WINPR_ASSERT(device);
|
|
|
|
name = device->device.Name;
|
|
path = device->Path;
|
|
driver = device->Driver;
|
|
|
|
if (!name || (name[0] == '*'))
|
|
{
|
|
/* TODO: implement auto detection of serial ports */
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
if ((name && name[0]) && (path && path[0]))
|
|
{
|
|
wLog* log;
|
|
log = WLog_Get("com.freerdp.channel.serial.client");
|
|
WLog_Print(log, WLOG_DEBUG, "initializing");
|
|
#ifndef __linux__ /* to be removed */
|
|
WLog_Print(log, WLOG_WARN, "Serial ports redirection not supported on this platform.");
|
|
return CHANNEL_RC_INITIALIZATION_ERROR;
|
|
#else /* __linux __ */
|
|
WLog_Print(log, WLOG_DEBUG, "Defining %s as %s", name, path);
|
|
|
|
if (!DefineCommDevice(name /* eg: COM1 */, path /* eg: /dev/ttyS0 */))
|
|
{
|
|
DWORD status = GetLastError();
|
|
WLog_ERR(TAG, "DefineCommDevice failed with %08" PRIx32, status);
|
|
return ERROR_INTERNAL_ERROR;
|
|
}
|
|
|
|
serial = (SERIAL_DEVICE*)calloc(1, sizeof(SERIAL_DEVICE));
|
|
|
|
if (!serial)
|
|
{
|
|
WLog_ERR(TAG, "calloc failed!");
|
|
return CHANNEL_RC_NO_MEMORY;
|
|
}
|
|
|
|
serial->log = log;
|
|
serial->device.type = RDPDR_DTYP_SERIAL;
|
|
serial->device.name = name;
|
|
serial->device.IRPRequest = serial_irp_request;
|
|
serial->device.Free = serial_free;
|
|
serial->rdpcontext = pEntryPoints->rdpcontext;
|
|
len = strlen(name);
|
|
serial->device.data = Stream_New(NULL, len + 1);
|
|
|
|
if (!serial->device.data)
|
|
{
|
|
WLog_ERR(TAG, "calloc failed!");
|
|
error = CHANNEL_RC_NO_MEMORY;
|
|
goto error_out;
|
|
}
|
|
|
|
for (i = 0; i <= len; i++)
|
|
Stream_Write_UINT8(serial->device.data, name[i] < 0 ? '_' : name[i]);
|
|
|
|
if (driver != NULL)
|
|
{
|
|
if (_stricmp(driver, "Serial") == 0)
|
|
serial->ServerSerialDriverId = SerialDriverSerialSys;
|
|
else if (_stricmp(driver, "SerCx") == 0)
|
|
serial->ServerSerialDriverId = SerialDriverSerCxSys;
|
|
else if (_stricmp(driver, "SerCx2") == 0)
|
|
serial->ServerSerialDriverId = SerialDriverSerCx2Sys;
|
|
else
|
|
{
|
|
WINPR_ASSERT(FALSE);
|
|
WLog_Print(serial->log, WLOG_DEBUG,
|
|
"Unknown server's serial driver: %s. SerCx2 will be used", driver);
|
|
serial->ServerSerialDriverId = SerialDriverSerialSys;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/* default driver */
|
|
serial->ServerSerialDriverId = SerialDriverSerialSys;
|
|
}
|
|
|
|
if (device->Permissive != NULL)
|
|
{
|
|
if (_stricmp(device->Permissive, "permissive") == 0)
|
|
{
|
|
serial->permissive = TRUE;
|
|
}
|
|
else
|
|
{
|
|
WLog_Print(serial->log, WLOG_DEBUG, "Unknown flag: %s", device->Permissive);
|
|
WINPR_ASSERT(FALSE);
|
|
}
|
|
}
|
|
|
|
WLog_Print(serial->log, WLOG_DEBUG, "Server's serial driver: %s (id: %d)", driver,
|
|
serial->ServerSerialDriverId);
|
|
/* TODO: implement auto detection of the server's serial driver */
|
|
serial->MainIrpQueue = MessageQueue_New(NULL);
|
|
|
|
if (!serial->MainIrpQueue)
|
|
{
|
|
WLog_ERR(TAG, "MessageQueue_New failed!");
|
|
error = CHANNEL_RC_NO_MEMORY;
|
|
goto error_out;
|
|
}
|
|
|
|
wObject* obj = MessageQueue_Object(serial->MainIrpQueue);
|
|
WINPR_ASSERT(obj);
|
|
obj->fnObjectFree = serial_message_free;
|
|
|
|
/* IrpThreads content only modified by create_irp_thread() */
|
|
serial->IrpThreads = ListDictionary_New(FALSE);
|
|
|
|
if (!serial->IrpThreads)
|
|
{
|
|
WLog_ERR(TAG, "ListDictionary_New failed!");
|
|
error = CHANNEL_RC_NO_MEMORY;
|
|
goto error_out;
|
|
}
|
|
|
|
serial->IrpThreadToBeTerminatedCount = 0;
|
|
InitializeCriticalSection(&serial->TerminatingIrpThreadsLock);
|
|
|
|
if ((error = pEntryPoints->RegisterDevice(pEntryPoints->devman, (DEVICE*)serial)))
|
|
{
|
|
WLog_ERR(TAG, "EntryPoints->RegisterDevice failed with error %" PRIu32 "!", error);
|
|
goto error_out;
|
|
}
|
|
|
|
if (!(serial->MainThread =
|
|
CreateThread(NULL, 0, serial_thread_func, (void*)serial, 0, NULL)))
|
|
{
|
|
WLog_ERR(TAG, "CreateThread failed!");
|
|
error = ERROR_INTERNAL_ERROR;
|
|
goto error_out;
|
|
}
|
|
|
|
#endif /* __linux __ */
|
|
}
|
|
|
|
return error;
|
|
error_out:
|
|
#ifdef __linux__ /* to be removed */
|
|
ListDictionary_Free(serial->IrpThreads);
|
|
MessageQueue_Free(serial->MainIrpQueue);
|
|
Stream_Free(serial->device.data, TRUE);
|
|
free(serial);
|
|
#endif /* __linux __ */
|
|
return error;
|
|
}
|