From 22f703cab05b7cd368f4de9e03991b7664dc5022 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?J=C3=B6rg=20Frings-F=C3=BCrst?= Date: Mon, 1 Sep 2014 13:56:46 +0200 Subject: Initial import of argyll version 1.5.1-8 --- usb/driver/get_descriptor.c | 289 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 289 insertions(+) create mode 100644 usb/driver/get_descriptor.c (limited to 'usb/driver/get_descriptor.c') diff --git a/usb/driver/get_descriptor.c b/usb/driver/get_descriptor.c new file mode 100644 index 0000000..2b5172e --- /dev/null +++ b/usb/driver/get_descriptor.c @@ -0,0 +1,289 @@ +/* libusb-win32, Generic Windows USB Library + * Copyright (c) 2002-2005 Stephan Meyer + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + + +#include "libusb_driver.h" + + + +NTSTATUS get_descriptor(libusb_device_t *dev, + void *buffer, int size, int type, int recipient, + int index, int language_id, int *received, int timeout) +{ + NTSTATUS status = STATUS_SUCCESS; + URB urb; + + USBMSG("buffer size: %d type: %04d recipient: %04d index: %04d language id: %04d timeout: %d\n", + size, type, recipient, index, language_id, timeout); + + memset(&urb, 0, sizeof(struct _URB_CONTROL_DESCRIPTOR_REQUEST)); + + switch (recipient) + { + case USB_RECIP_DEVICE: + urb.UrbHeader.Function = URB_FUNCTION_GET_DESCRIPTOR_FROM_DEVICE; + break; + case USB_RECIP_INTERFACE: + urb.UrbHeader.Function = URB_FUNCTION_GET_DESCRIPTOR_FROM_INTERFACE; + break; + case USB_RECIP_ENDPOINT: + urb.UrbHeader.Function = URB_FUNCTION_GET_DESCRIPTOR_FROM_ENDPOINT; + break; + default: + USBERR0("invalid recipient\n"); + return STATUS_INVALID_PARAMETER; + } + + urb.UrbHeader.Length = sizeof(struct _URB_CONTROL_DESCRIPTOR_REQUEST); + urb.UrbControlDescriptorRequest.TransferBufferLength = size; + urb.UrbControlDescriptorRequest.TransferBuffer = buffer; + urb.UrbControlDescriptorRequest.DescriptorType = (UCHAR)type; + urb.UrbControlDescriptorRequest.Index = (UCHAR)index; + urb.UrbControlDescriptorRequest.LanguageId = (USHORT)language_id; + + // the device and active config descriptors are cached. If the request + // is for one of these and we have already retrieved it, then this is + // a non-blocking non-i/o call. + + if (type == USB_DEVICE_DESCRIPTOR_TYPE && + recipient == USB_RECIP_DEVICE && + index == 0 && + language_id == 0 && + size >= sizeof(USB_DEVICE_DESCRIPTOR)) + { + // this is a device descriptor request. + if (dev->device_descriptor.bLength == 0) + { + // this is the first request made for the device descriptor. + // Cache it now and forever + status = call_usbd(dev, &urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout); + if (NT_SUCCESS(status) && urb.UrbControlDescriptorRequest.TransferBufferLength < sizeof(USB_DEVICE_DESCRIPTOR)) + { + USBERR("Invalid device decriptor length %d\n", urb.UrbControlDescriptorRequest.TransferBufferLength); + status = STATUS_BAD_DEVICE_TYPE; + *received = 0; + goto Done; + } + + if (!NT_SUCCESS(status) || !USBD_SUCCESS(urb.UrbHeader.Status)) + { + USBERR("getting descriptor failed: status: 0x%x, urb-status: 0x%x\n", status, urb.UrbHeader.Status); + *received = 0; + goto Done; + } + + // valid device descriptor + size = sizeof(USB_DEVICE_DESCRIPTOR); + RtlCopyMemory(&dev->device_descriptor, buffer, size); + *received = size; + + } + else + { + // device descriptor is already cached. + size = sizeof(USB_DEVICE_DESCRIPTOR); + RtlCopyMemory(buffer, &dev->device_descriptor, size); + *received = size; + } + + goto Done; + } + + if (type == USB_CONFIGURATION_DESCRIPTOR_TYPE && + recipient == USB_RECIP_DEVICE && + language_id == 0 && + size >= sizeof(USB_CONFIGURATION_DESCRIPTOR)) + { + if (dev->device_descriptor.bLength != 0 && index >=dev->device_descriptor.bNumConfigurations) + { + USBWRN("config descriptor index %d out of range.\n", index); + status = STATUS_NO_MORE_ENTRIES; + *received = 0; + goto Done; + } + + // this is a config descriptor request. + if (!dev->config.descriptor || dev->config.index != index) + { + // this is either: + // * The first request made for a config descriptor. + // * A request for a config descriptor other than the cached index. + status = call_usbd(dev, &urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout); + + if (!NT_SUCCESS(status) || !USBD_SUCCESS(urb.UrbHeader.Status)) + { + USBERR("getting descriptor failed: status: 0x%x, urb-status: 0x%x\n", status, urb.UrbHeader.Status); + *received = 0; + goto Done; + } + + if (!dev->config.descriptor && + urb.UrbControlDescriptorRequest.TransferBufferLength >= ((PUSB_CONFIGURATION_DESCRIPTOR)buffer)->wTotalLength) + { + // + // This is a special case scenario where we cache the first + // config deescriptor requested when dev->config.descriptor == NULL + // and this was a request for the *entire* descriptor. + // + // Nearly all windows USB devices will only have one configuration. + // In the case a the filter driver, it does *not* auto configure the + // device which is where the *active* config descriptor caching occurs. + // + // This code ensures the first config descriptor requested is always cached + // even if the device is not configured yet. + // + + PUSB_CONFIGURATION_DESCRIPTOR config_desc; + size = ((PUSB_CONFIGURATION_DESCRIPTOR)buffer)->wTotalLength; + + if (!( config_desc = ExAllocatePool(NonPagedPool, size))) + { + USBERR0("memory allocation error\n"); + status = STATUS_NO_MEMORY; + goto Done; + } + + dev->config.descriptor=config_desc; + + RtlCopyMemory(dev->config.descriptor,buffer,size); + dev->config.value=0; + dev->config.total_size=size; + dev->config.index=index; + + *received = size; + } + else + { + *received = urb.UrbControlDescriptorRequest.TransferBufferLength; + } + goto Done; + } + else + { + // This is a request for the active configuration descriptor. + // This is only updated upon a successful set_configuration(). + size = (size > dev->config.descriptor->wTotalLength) ? dev->config.descriptor->wTotalLength : size; + RtlCopyMemory(buffer, dev->config.descriptor, size); + + *received = size; + goto Done; + } + } + + // this is not a device or config descriptor reequest + status = call_usbd(dev, &urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout); + + + if (!NT_SUCCESS(status) || !USBD_SUCCESS(urb.UrbHeader.Status)) + { + USBERR("getting descriptor failed: status: 0x%x, urb-status: 0x%x\n", status, urb.UrbHeader.Status); + *received = 0; + } + else + { + *received = urb.UrbControlDescriptorRequest.TransferBufferLength; + } + +Done: + return status; +} + +PUSB_CONFIGURATION_DESCRIPTOR get_config_descriptor( + libusb_device_t *dev, + int value, + int *size, + int* index) +{ + NTSTATUS status; + USB_CONFIGURATION_DESCRIPTOR *desc = NULL; + USB_DEVICE_DESCRIPTOR device_descriptor; + int i; + volatile int desc_size; + + *index = 0; + + status = get_descriptor(dev, &device_descriptor, + sizeof(USB_DEVICE_DESCRIPTOR), + USB_DEVICE_DESCRIPTOR_TYPE, + USB_RECIP_DEVICE, + 0, 0, size, LIBUSB_DEFAULT_TIMEOUT); + + if (!NT_SUCCESS(status) || *size != sizeof(USB_DEVICE_DESCRIPTOR)) + { + USBERR0("getting device descriptor failed\n"); + return NULL; + } + + if (!(desc = ExAllocatePool(NonPagedPool, + sizeof(USB_CONFIGURATION_DESCRIPTOR)))) + { + USBERR0("memory allocation error\n"); + return NULL; + } + + for (i = 0; i < device_descriptor.bNumConfigurations; i++) + { + + if (!NT_SUCCESS(get_descriptor(dev, desc, + sizeof(USB_CONFIGURATION_DESCRIPTOR), + USB_CONFIGURATION_DESCRIPTOR_TYPE, + USB_RECIP_DEVICE, + i, 0, size, LIBUSB_DEFAULT_TIMEOUT))) + { + USBERR0("getting configuration descriptor failed\n"); + break; + } + + // if value is negative, get the descriptor by index + // if positive, get it by value. + if ((value > 0 && desc->bConfigurationValue == value) || + (value < 0 && -(i+1) == (value))) + { + desc_size = desc->wTotalLength; + ExFreePool(desc); + + if (!(desc = ExAllocatePool(NonPagedPool, desc_size))) + { + USBERR0("memory allocation error\n"); + break; + } + + if (!NT_SUCCESS(get_descriptor(dev, desc, desc_size, + USB_CONFIGURATION_DESCRIPTOR_TYPE, + USB_RECIP_DEVICE, + i, 0, size, LIBUSB_DEFAULT_TIMEOUT))) + { + USBERR0("getting configuration descriptor failed\n"); + break; + } + else + { + *index = i; + } + + return desc; + } + } + + if (desc) + { + ExFreePool(desc); + } + + return NULL; +} -- cgit v1.2.3