2012-04-25 22:26:35 +04:00
|
|
|
/**
|
2012-10-09 07:02:04 +04:00
|
|
|
* FreeRDP: A Remote Desktop Protocol Implementation
|
2012-04-25 22:26:35 +04:00
|
|
|
* RemoteFX USB Redirection
|
|
|
|
*
|
|
|
|
* Copyright 2012 Atrust corp.
|
|
|
|
* Copyright 2012 Alfred Liu <alfred.liu@atruscorp.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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include "searchman.h"
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
static void searchman_rewind(USB_SEARCHMAN* searchman)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
searchman->idev = searchman->head;
|
|
|
|
}
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
static int searchman_has_next(USB_SEARCHMAN* searchman)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
if (searchman->idev == NULL)
|
|
|
|
return 0;
|
|
|
|
else
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
static USB_SEARCHDEV* searchman_get_next(USB_SEARCHMAN* searchman)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
USB_SEARCHDEV* search;
|
|
|
|
|
|
|
|
search = searchman->idev;
|
2012-10-03 01:24:52 +04:00
|
|
|
searchman->idev = (USB_SEARCHDEV*) searchman->idev->next;
|
2012-04-25 22:26:35 +04:00
|
|
|
|
|
|
|
return search;
|
|
|
|
}
|
|
|
|
|
2012-10-09 11:01:37 +04:00
|
|
|
static int searchman_list_add(USB_SEARCHMAN* searchman, UINT16 idVendor, UINT16 idProduct)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
USB_SEARCHDEV* search;
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
search = (USB_SEARCHDEV*) malloc(sizeof(USB_SEARCHDEV));
|
2012-04-25 22:26:35 +04:00
|
|
|
|
|
|
|
search->prev = NULL;
|
|
|
|
search->next = NULL;
|
|
|
|
search->idVendor = idVendor;
|
|
|
|
search->idProduct = idProduct;
|
|
|
|
|
|
|
|
if (searchman->head == NULL)
|
|
|
|
{
|
|
|
|
/* linked list is empty */
|
|
|
|
searchman->head = search;
|
|
|
|
searchman->tail = search;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* append device to the end of the linked list */
|
|
|
|
searchman->tail->next = (void*)search;
|
|
|
|
search->prev = (void*)searchman->tail;
|
|
|
|
searchman->tail = search;
|
|
|
|
}
|
|
|
|
searchman->usb_numbers += 1;
|
|
|
|
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2012-10-09 11:01:37 +04:00
|
|
|
static int searchman_list_remove(USB_SEARCHMAN* searchman, UINT16 idVendor, UINT16 idProduct)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
USB_SEARCHDEV* search;
|
|
|
|
USB_SEARCHDEV* point;
|
|
|
|
|
|
|
|
searchman_rewind(searchman);
|
2012-10-03 01:24:52 +04:00
|
|
|
|
2012-04-25 22:26:35 +04:00
|
|
|
while (searchman_has_next(searchman) != 0)
|
|
|
|
{
|
|
|
|
point = searchman_get_next(searchman);
|
|
|
|
|
|
|
|
if (point->idVendor == idVendor &&
|
|
|
|
point->idProduct == idProduct)
|
|
|
|
{
|
|
|
|
/* set previous device to point to next device */
|
|
|
|
|
|
|
|
search = point;
|
|
|
|
if (search->prev != NULL)
|
|
|
|
{
|
|
|
|
/* unregistered device is not the head */
|
|
|
|
point = (USB_SEARCHDEV*)search->prev;
|
|
|
|
point->next = search->next;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* unregistered device is the head, update head */
|
|
|
|
searchman->head = (USB_SEARCHDEV*)search->next;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* set next device to point to previous device */
|
|
|
|
|
|
|
|
if (search->next != NULL)
|
|
|
|
{
|
|
|
|
/* unregistered device is not the tail */
|
|
|
|
point = (USB_SEARCHDEV*)search->next;
|
|
|
|
point->prev = search->prev;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
/* unregistered device is the tail, update tail */
|
|
|
|
searchman->tail = (USB_SEARCHDEV*)search->prev;
|
|
|
|
}
|
|
|
|
searchman->usb_numbers--;
|
|
|
|
|
|
|
|
free(search);
|
|
|
|
|
|
|
|
return 1; /* unregistration successful */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* if we reach this point, the device wasn't found */
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
static void searchman_start(USB_SEARCHMAN* self, void* func)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
pthread_t thread;
|
|
|
|
|
|
|
|
/* create search thread */
|
|
|
|
pthread_create(&thread, 0, func, self);
|
|
|
|
pthread_detach(thread);
|
|
|
|
self->strated = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* close thread */
|
2012-10-03 01:24:52 +04:00
|
|
|
static void searchman_close(USB_SEARCHMAN* self)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
wait_obj_set(self->term_event);
|
|
|
|
}
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
static void searchman_list_show(USB_SEARCHMAN* self)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
int num = 0;
|
2012-10-03 01:24:52 +04:00
|
|
|
USB_SEARCHDEV* usb;
|
2012-04-25 22:26:35 +04:00
|
|
|
|
|
|
|
printf("=========== Usb Search List ========= \n");
|
|
|
|
self->rewind(self);
|
|
|
|
while (self->has_next(self))
|
|
|
|
{
|
|
|
|
usb = self->get_next(self);
|
|
|
|
printf(" USB %d: \n", num++);
|
|
|
|
printf(" idVendor: 0x%04X \n", usb->idVendor);
|
|
|
|
printf(" idProduct: 0x%04X \n", usb->idProduct);
|
|
|
|
}
|
|
|
|
printf("================= END =============== \n");
|
|
|
|
}
|
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
void searchman_free(USB_SEARCHMAN* self)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
USB_SEARCHDEV * dev;
|
|
|
|
|
|
|
|
while (self->head != NULL)
|
|
|
|
{
|
|
|
|
dev = (USB_SEARCHDEV *)self->head;
|
|
|
|
self->remove (self, dev->idVendor, dev->idProduct);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* free searchman */
|
|
|
|
sem_destroy(&self->sem_term);
|
|
|
|
wait_obj_free(self->term_event);
|
|
|
|
free(self);
|
|
|
|
}
|
|
|
|
|
2012-10-09 11:26:39 +04:00
|
|
|
USB_SEARCHMAN* searchman_new(void * urbdrc, UINT32 UsbDevice)
|
2012-04-25 22:26:35 +04:00
|
|
|
{
|
|
|
|
int ret;
|
2012-10-03 01:24:52 +04:00
|
|
|
USB_SEARCHMAN* searchman;
|
2012-04-25 22:26:35 +04:00
|
|
|
|
2012-10-03 01:24:52 +04:00
|
|
|
searchman = (USB_SEARCHMAN*) malloc(sizeof(USB_SEARCHMAN));
|
|
|
|
|
2012-04-25 22:26:35 +04:00
|
|
|
searchman->idev = NULL;
|
|
|
|
searchman->head = NULL;
|
|
|
|
searchman->tail = NULL;
|
|
|
|
searchman->usb_numbers = 0;
|
|
|
|
searchman->urbdrc = urbdrc;
|
|
|
|
searchman->UsbDevice = UsbDevice;
|
|
|
|
|
|
|
|
ret = pthread_mutex_init(&searchman->mutex, NULL);
|
2012-10-03 01:24:52 +04:00
|
|
|
|
2012-04-25 22:26:35 +04:00
|
|
|
if (ret != 0)
|
|
|
|
{
|
|
|
|
printf("searchman mutex initialization: searchman->mutex failed");
|
|
|
|
exit(EXIT_FAILURE);
|
|
|
|
}
|
|
|
|
|
|
|
|
/* load service */
|
|
|
|
searchman->add = searchman_list_add;
|
|
|
|
searchman->remove = searchman_list_remove;
|
|
|
|
searchman->rewind = searchman_rewind;
|
|
|
|
searchman->get_next = searchman_get_next;
|
|
|
|
searchman->has_next = searchman_has_next;
|
|
|
|
searchman->show = searchman_list_show;
|
|
|
|
searchman->start = searchman_start;
|
|
|
|
searchman->close = searchman_close;
|
|
|
|
searchman->free = searchman_free;
|
|
|
|
|
|
|
|
searchman->strated = 0;
|
|
|
|
searchman->term_event = wait_obj_new();
|
|
|
|
sem_init(&searchman->sem_term, 0, 0);
|
|
|
|
|
|
|
|
return searchman;
|
|
|
|
}
|