FreeRDP/channels/urbdrc/client/searchman.c

226 lines
5.1 KiB
C
Raw Normal View History

/**
2012-10-09 07:02:04 +04:00
* FreeRDP: A Remote Desktop Protocol Implementation
* 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>
2012-12-03 19:32:04 +04:00
#include <winpr/crt.h>
#include <winpr/synch.h>
2012-12-03 19:32:04 +04:00
#include "searchman.h"
2012-10-03 01:24:52 +04:00
static void searchman_rewind(USB_SEARCHMAN* searchman)
{
searchman->idev = searchman->head;
}
2012-10-03 01:24:52 +04:00
static int searchman_has_next(USB_SEARCHMAN* searchman)
{
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)
{
USB_SEARCHDEV* search;
search = searchman->idev;
2012-10-03 01:24:52 +04:00
searchman->idev = (USB_SEARCHDEV*) searchman->idev->next;
return search;
}
static int searchman_list_add(USB_SEARCHMAN* searchman, UINT16 idVendor, UINT16 idProduct)
{
USB_SEARCHDEV* search;
2012-10-03 01:24:52 +04:00
search = (USB_SEARCHDEV*) malloc(sizeof(USB_SEARCHDEV));
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;
}
static int searchman_list_remove(USB_SEARCHMAN* searchman, UINT16 idVendor, UINT16 idProduct)
{
USB_SEARCHDEV* search;
USB_SEARCHDEV* point;
searchman_rewind(searchman);
2012-10-03 01:24:52 +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)
{
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)
{
SetEvent(self->term_event);
}
2012-10-03 01:24:52 +04:00
static void searchman_list_show(USB_SEARCHMAN* self)
{
int num = 0;
2012-10-03 01:24:52 +04:00
USB_SEARCHDEV* usb;
fprintf(stderr, "=========== Usb Search List ========= \n");
self->rewind(self);
while (self->has_next(self))
{
usb = self->get_next(self);
fprintf(stderr, " USB %d: \n", num++);
fprintf(stderr, " idVendor: 0x%04X \n", usb->idVendor);
fprintf(stderr, " idProduct: 0x%04X \n", usb->idProduct);
}
fprintf(stderr, "================= END =============== \n");
}
2012-10-03 01:24:52 +04:00
void searchman_free(USB_SEARCHMAN* self)
{
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);
CloseHandle(self->term_event);
free(self);
}
2012-10-09 11:26:39 +04:00
USB_SEARCHMAN* searchman_new(void * urbdrc, UINT32 UsbDevice)
{
int ret;
2012-10-03 01:24:52 +04:00
USB_SEARCHMAN* searchman;
2012-10-03 01:24:52 +04:00
searchman = (USB_SEARCHMAN*) malloc(sizeof(USB_SEARCHMAN));
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
if (ret != 0)
{
fprintf(stderr, "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 = CreateEvent(NULL, TRUE, FALSE, NULL);
sem_init(&searchman->sem_term, 0, 0);
return searchman;
}