192 lines
4.4 KiB
C
192 lines
4.4 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 Thincast Technologies GmbH
|
|
* Copyright 2015 DI (FH) Martin Haimberger <martin.haimberger@thincast.com>
|
|
* Copyright 2016 Armin Novak <armin.novak@gmail.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 <freerdp/types.h>
|
|
#include <freerdp/addin.h>
|
|
#include <freerdp/client/channels.h>
|
|
#include <freerdp/channels/log.h>
|
|
|
|
#include "rdpdr_main.h"
|
|
|
|
#include "devman.h"
|
|
|
|
void devman_device_free(DEVICE* device)
|
|
{
|
|
if (!device)
|
|
return;
|
|
|
|
IFCALL(device->Free, device);
|
|
}
|
|
|
|
DEVMAN* devman_new(rdpdrPlugin* rdpdr)
|
|
{
|
|
DEVMAN* devman;
|
|
|
|
if (!rdpdr)
|
|
return NULL;
|
|
|
|
devman = (DEVMAN*) calloc(1, sizeof(DEVMAN));
|
|
|
|
if (!devman)
|
|
{
|
|
WLog_INFO(TAG, "calloc failed!");
|
|
return NULL;
|
|
}
|
|
|
|
devman->plugin = (void*) rdpdr;
|
|
devman->id_sequence = 1;
|
|
|
|
devman->devices = ListDictionary_New(TRUE);
|
|
if (!devman->devices)
|
|
{
|
|
WLog_INFO(TAG, "ListDictionary_New failed!");
|
|
free(devman);
|
|
return NULL;
|
|
}
|
|
|
|
ListDictionary_ValueObject(devman->devices)->fnObjectFree =
|
|
(OBJECT_FREE_FN) devman_device_free;
|
|
|
|
return devman;
|
|
}
|
|
|
|
void devman_free(DEVMAN* devman)
|
|
{
|
|
ListDictionary_Free(devman->devices);
|
|
free(devman);
|
|
}
|
|
|
|
void devman_unregister_device(DEVMAN* devman, void* key)
|
|
{
|
|
DEVICE* device;
|
|
|
|
if (!devman || !key)
|
|
return;
|
|
|
|
device = (DEVICE*) ListDictionary_Remove(devman->devices, key);
|
|
|
|
if (device)
|
|
devman_device_free(device);
|
|
}
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
static UINT devman_register_device(DEVMAN* devman, DEVICE* device)
|
|
{
|
|
void* key = NULL;
|
|
|
|
if (!devman || !device)
|
|
return ERROR_INVALID_PARAMETER;
|
|
|
|
device->id = devman->id_sequence++;
|
|
key = (void*) (size_t) device->id;
|
|
|
|
if (!ListDictionary_Add(devman->devices, key, device))
|
|
{
|
|
WLog_INFO(TAG, "ListDictionary_Add failed!");
|
|
return ERROR_INTERNAL_ERROR;
|
|
}
|
|
return CHANNEL_RC_OK;
|
|
}
|
|
|
|
DEVICE* devman_get_device_by_id(DEVMAN* devman, UINT32 id)
|
|
{
|
|
DEVICE* device = NULL;
|
|
void* key = (void*) (size_t) id;
|
|
|
|
if (!devman)
|
|
return NULL;
|
|
|
|
device = (DEVICE*) ListDictionary_GetItemValue(devman->devices, key);
|
|
|
|
return device;
|
|
}
|
|
|
|
static char DRIVE_SERVICE_NAME[] = "drive";
|
|
static char PRINTER_SERVICE_NAME[] = "printer";
|
|
static char SMARTCARD_SERVICE_NAME[] = "smartcard";
|
|
static char SERIAL_SERVICE_NAME[] = "serial";
|
|
static char PARALLEL_SERVICE_NAME[] = "parallel";
|
|
|
|
/**
|
|
* Function description
|
|
*
|
|
* @return 0 on success, otherwise a Win32 error code
|
|
*/
|
|
UINT devman_load_device_service(DEVMAN* devman, RDPDR_DEVICE* device, rdpContext* rdpcontext)
|
|
{
|
|
char* ServiceName = NULL;
|
|
DEVICE_SERVICE_ENTRY_POINTS ep;
|
|
PDEVICE_SERVICE_ENTRY entry = NULL;
|
|
|
|
if (!devman || !device || !rdpcontext)
|
|
return ERROR_INVALID_PARAMETER;
|
|
|
|
if (device->Type == RDPDR_DTYP_FILESYSTEM)
|
|
ServiceName = DRIVE_SERVICE_NAME;
|
|
else if (device->Type == RDPDR_DTYP_PRINT)
|
|
ServiceName = PRINTER_SERVICE_NAME;
|
|
else if (device->Type == RDPDR_DTYP_SMARTCARD)
|
|
ServiceName = SMARTCARD_SERVICE_NAME;
|
|
else if (device->Type == RDPDR_DTYP_SERIAL)
|
|
ServiceName = SERIAL_SERVICE_NAME;
|
|
else if (device->Type == RDPDR_DTYP_PARALLEL)
|
|
ServiceName = PARALLEL_SERVICE_NAME;
|
|
|
|
if (!ServiceName)
|
|
{
|
|
WLog_INFO(TAG, "ServiceName %s did not match!", ServiceName);
|
|
return ERROR_INVALID_NAME;
|
|
}
|
|
|
|
WLog_INFO(TAG, "Loading device service %s (static)", ServiceName);
|
|
entry = (PDEVICE_SERVICE_ENTRY) freerdp_load_channel_addin_entry(ServiceName, NULL, "DeviceServiceEntry", 0);
|
|
|
|
if (!entry)
|
|
{
|
|
WLog_INFO(TAG, "freerdp_load_channel_addin_entry failed!");
|
|
return ERROR_INTERNAL_ERROR;
|
|
}
|
|
|
|
ep.devman = devman;
|
|
ep.RegisterDevice = devman_register_device;
|
|
ep.device = device;
|
|
ep.rdpcontext = rdpcontext;
|
|
|
|
return entry(&ep);
|
|
}
|