/* LIBUSB-WIN32, Generic Windows USB Library * Copyright (c) 2002-2005 Stephan Meyer <ste_meyer@web.de> * * 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;  DEBUG_PRINT_NL();  DEBUG_MESSAGE("get_descriptor(): buffer size %d", size);  DEBUG_MESSAGE("get_descriptor(): type %04d", type);  DEBUG_MESSAGE("get_descriptor(): recipient %04d", recipient);  DEBUG_MESSAGE("get_descriptor(): index %04d", index);  DEBUG_MESSAGE("get_descriptor(): language id %04d", language_id);  DEBUG_MESSAGE("get_descriptor(): timeout %d", 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:      DEBUG_ERROR("get_descriptor(): invalid recipient");      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;    status = call_usbd(dev, &urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout);        if(!NT_SUCCESS(status) || !USBD_SUCCESS(urb.UrbHeader.Status))    {      DEBUG_ERROR("get_descriptor(): getting descriptor "                  "failed: status: 0x%x, urb-status: 0x%x",                   status, urb.UrbHeader.Status);      *received = 0;    }  else    {      *received = urb.UrbControlDescriptorRequest.TransferBufferLength;    }  return status;}USB_CONFIGURATION_DESCRIPTOR *get_config_descriptor(libusb_device_t *dev, int value, int *size){  NTSTATUS status;  USB_CONFIGURATION_DESCRIPTOR *desc = NULL;  USB_DEVICE_DESCRIPTOR device_descriptor;  int i;  volatile int desc_size;  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))    {      DEBUG_ERROR("get_config_descriptor(): getting device descriptor failed");      return NULL;    }  if(!(desc = ExAllocatePool(NonPagedPool,                              sizeof(USB_CONFIGURATION_DESCRIPTOR))))    {      DEBUG_ERROR("get_config_descriptor(): memory allocation error");      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)))        {          DEBUG_ERROR("get_config_descriptor(): getting configuration "                      "descriptor failed");          break;        }      if(desc->bConfigurationValue == value)        {          desc_size = desc->wTotalLength;          ExFreePool(desc);          if(!(desc = ExAllocatePool(NonPagedPool, desc_size)))            {              DEBUG_ERROR("get_config_descriptor(): memory allocation error");              break;            }                    if(!NT_SUCCESS(get_descriptor(dev, desc, desc_size,                                        USB_CONFIGURATION_DESCRIPTOR_TYPE,                                        USB_RECIP_DEVICE,                                        i, 0, size, LIBUSB_DEFAULT_TIMEOUT)))            {              DEBUG_ERROR("get_config_descriptor(): getting configuration "                          "descriptor failed");              break;            }                    return desc;        }    }  if(desc)    {      ExFreePool(desc);    }  return NULL;}