static void
descriptor_callback(HANDLE hub, ULONG port, USHORT deviceAddress,
PUSB_DEVICE_DESCRIPTOR desc, void *context)
{
descriptor_callback_context *ctx = (descriptor_callback_context *)context;
PUSB_DESCRIPTOR_REQUEST request;
if (!USBPcapIsDeviceFiltered(ctx->addresses, deviceAddress))
{
return;
}
write_setup_packet(ctx, deviceAddress, 0x80, 6,
USB_DEVICE_DESCRIPTOR_TYPE << 8, 0, 18, FALSE);
write_device_descriptor_data(ctx, deviceAddress, desc);
write_status_packet(ctx, deviceAddress, FALSE);
request = get_config_descriptor(hub, port, 0);
if (request)
{
PUSB_CONFIGURATION_DESCRIPTOR config;
config = (PUSB_CONFIGURATION_DESCRIPTOR)(request->Data);
write_setup_packet(ctx, deviceAddress,
request->SetupPacket.bmRequest,
request->SetupPacket.bRequest,
request->SetupPacket.wValue,
request->SetupPacket.wIndex,
request->SetupPacket.wLength,
FALSE);
write_data_packet(ctx, deviceAddress, request->Data,
request->SetupPacket.wLength, FALSE);
write_status_packet(ctx, deviceAddress, FALSE);
/* SET CONFIGURATION */
write_setup_packet(ctx, deviceAddress, 0x00, 9, config->bConfigurationValue, 0, 0, TRUE);
write_status_packet(ctx, deviceAddress, TRUE);
}
free(request);
}
NTSTATUS set_configuration(libusb_device_t *dev,
int configuration,
int timeout)
{
NTSTATUS status = STATUS_SUCCESS;
URB urb, *urb_ptr = NULL;
USB_CONFIGURATION_DESCRIPTOR *configuration_descriptor = NULL;
USB_INTERFACE_DESCRIPTOR *interface_descriptor = NULL;
USBD_INTERFACE_LIST_ENTRY *interfaces = NULL;
int i, j, interface_number, desc_size, config_index, ret;
// check if this config value is already set
if ((configuration > 0) && dev->config.value == configuration)
{
return STATUS_SUCCESS;
}
// check if this config index is already set
if ((configuration < 0) && dev->config.value && dev->config.index == (abs(configuration)-1))
{
return STATUS_SUCCESS;
}
memset(&urb, 0, sizeof(URB));
if (configuration == 0)
{
urb.UrbHeader.Function = URB_FUNCTION_SELECT_CONFIGURATION;
urb.UrbHeader.Length = sizeof(struct _URB_SELECT_CONFIGURATION);
status = call_usbd(dev, &urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout);
if (!NT_SUCCESS(status) || !USBD_SUCCESS(urb.UrbHeader.Status))
{
USBERR("setting configuration %d failed: status: 0x%x, urb-status: 0x%x\n",
configuration, status, urb.UrbHeader.Status);
return status;
}
dev->config.handle = urb.UrbSelectConfiguration.ConfigurationHandle;
dev->config.value = 0;
clear_pipe_info(dev);
return status;
}
if (configuration <= SET_CONFIG_ACTIVE_CONFIG)
{
// note: as of v1.2.4.0, the active/default configuration is
// always the first configuration the device returns. (index 0)
configuration=-1;
}
USBMSG("setting configuration %s %d timeout=%d",
(configuration < 0) ? "index" : "value",
(configuration < 0) ? abs(configuration) - 1 : configuration,
timeout);
// If configuration is negative, it is retrieved by index.
//
configuration_descriptor = get_config_descriptor(dev, configuration,
&desc_size, &config_index);
if (!configuration_descriptor)
{
USBERR0("getting configuration descriptor failed");
return STATUS_INVALID_PARAMETER;
}
// if we passed an index in we can check here to see
// if the device is already configured with this value
if (dev->config.value == configuration_descriptor->bConfigurationValue)
{
UpdateContextConfigDescriptor(
dev,
configuration_descriptor,
desc_size,
configuration_descriptor->bConfigurationValue,
config_index);
status = STATUS_SUCCESS;
goto SetConfigurationDone;
}
// MEMORY ALLOCATION BEGINS
interfaces =
ExAllocatePool(NonPagedPool,(configuration_descriptor->bNumInterfaces + 1)
* sizeof(USBD_INTERFACE_LIST_ENTRY));
if (!interfaces)
{
USBERR0("memory allocation failed\n");
status = STATUS_NO_MEMORY;
ExFreePool(configuration_descriptor);
goto SetConfigurationDone;
}
memset(interfaces, 0, (configuration_descriptor->bNumInterfaces + 1)
* sizeof(USBD_INTERFACE_LIST_ENTRY));
interface_number = 0;
for (i = 0; i < configuration_descriptor->bNumInterfaces; i++)
{
for (j = interface_number; j < LIBUSB_MAX_NUMBER_OF_INTERFACES; j++)
{
interface_descriptor =
find_interface_desc(configuration_descriptor, desc_size, j, 0);
if (interface_descriptor)
{
interface_number = ++j;
break;
}
}
if (!interface_descriptor)
{
USBERR("unable to find interface descriptor at index %d\n", i);
status = STATUS_INVALID_PARAMETER;
ExFreePool(configuration_descriptor);
goto SetConfigurationDone;
}
else
{
USBMSG("found interface %d\n",
interface_descriptor->bInterfaceNumber);
interfaces[i].InterfaceDescriptor = interface_descriptor;
}
}
urb_ptr = USBD_CreateConfigurationRequestEx(configuration_descriptor, interfaces);
if (!urb_ptr)
{
USBERR0("memory allocation failed\n");
status = STATUS_NO_MEMORY;
ExFreePool(configuration_descriptor);
goto SetConfigurationDone;
}
for (i = 0; i < configuration_descriptor->bNumInterfaces; i++)
{
for (j = 0; j < (int)interfaces[i].Interface->NumberOfPipes; j++)
{
interfaces[i].Interface->Pipes[j].MaximumTransferSize = LIBUSB_MAX_READ_WRITE;
}
}
USBDBG("#%d %s passing configuration request to target-device.",
dev->id, dev->device_id);
status = call_usbd(dev, urb_ptr, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout);
if (!NT_SUCCESS(status) || !USBD_SUCCESS(urb_ptr->UrbHeader.Status))
{
USBERR("setting configuration %d failed: status: 0x%x, urb-status: 0x%x\n",
configuration, status, urb_ptr->UrbHeader.Status);
if (NT_SUCCESS(status)) status = urb_ptr->UrbHeader.Status;
ExFreePool(configuration_descriptor);
goto SetConfigurationDone;
}
dev->config.handle = urb_ptr->UrbSelectConfiguration.ConfigurationHandle;
clear_pipe_info(dev);
for (i = 0; i < configuration_descriptor->bNumInterfaces; i++)
{
update_pipe_info(dev, interfaces[i].Interface);
}
UpdateContextConfigDescriptor(dev, configuration_descriptor, desc_size, configuration_descriptor->bConfigurationValue, config_index);
SetConfigurationDone:
if (interfaces)
ExFreePool(interfaces);
if (urb_ptr)
ExFreePool(urb_ptr);
return status;
}
RES_CODE usb_remote_hid_t::scan_hid(uint32_t port_indx, USBSubClassCode subcls, USBProtocolCode proto)
{
RES_CODE res;
// select EPT_0 and device address
ep0_hnd->mode.as_bytes[0] = EPT_0; // RX pipe
ep0_hnd->mode.as_bytes[1] = EPT_0; // TX pipe
res = hdc_init(port_indx);
if(res == RES_OK)
{
epi_hnd->mode0 = ep0_hnd->mode0; // device hub port
// TRACE Device descriptor
TRACELN("HID: dev found %x:%x", dev_descriptor.idVendor, dev_descriptor.idProduct);
trace_usb_descriptor(&dev_descriptor.as_generic);
if(dev_descriptor.bDeviceClass == INTERFACE_DEFINED_CLASS)
{
//loop the configurations
for(uint32_t cfg_indx=0; cfg_indx<dev_descriptor.bNumConfigurations; cfg_indx++)
{
res = get_config_descriptor(cfg_indx);
if(res != RES_OK)
break;
// TRACE Configuration descriptor(s)
trace_usb_descriptor(&config_descriptor->as_generic);
//loop the interfaces
for(uint32_t iface_indx=0; iface_indx<config_descriptor->bNumInterfaces; iface_indx++)
{
pid = usb_get_interface(config_descriptor, iface_indx);
if(pid && pid->bDeviceClass == HID_DEVICE_CLASS &&
pid->bDeviceSubClass == subcls && pid->bDeviceProtocol == proto)
{
// set the configuration
// set_configuration(0);
res = set_configuration(config_descriptor->bConfigurationValue);
if(res == RES_OK)
{
epi_hnd->mode.as_bytes[0] = EPT_0;
epi_hnd->mode.as_bytes[1] = EPT_0;
for(int i=0; i<pid->bNumEndpoints && i<2; i++)
{
USBEndpointDescriptor* ped;
ped = usb_get_enpoint(config_descriptor, i);
if(ped && ped->bmAttributes == ENDPOINT_TYPE_INTERRUPT)
{
if( ped->bEndpointAddress & 0x80 )
{
if(epi_hnd->mode.as_bytes[0] == EPT_0)
{
epi_hnd->mode.as_bytes[0] = ped->bEndpointAddress & 0x7F;
epi_hnd->mode.as_ushort[1] = ep0_hnd->mode.as_ushort[1]; //drv_state_cnt
usb_svc_configendpoint(epi_hnd, &ped->as_generic);
}
} else
{
if(epi_hnd->mode.as_bytes[1] == EPT_0)
{
epi_hnd->mode.as_bytes[1] = ped->bEndpointAddress ;
epi_hnd->mode.as_ushort[1] = ep0_hnd->mode.as_ushort[1]; //drv_state_cnt
usb_svc_configendpoint(epi_hnd, &ped->as_generic);
}
}
}
}
hid_idle = 255;
get_idle(HID_REPORT_ALL);
return RES_OK;
}
}
}
}
}
}
if(res == RES_OK)
res = RES_ERROR;
return res;
}
NTSTATUS set_interface(libusb_device_t *dev, int interface, int altsetting,
int timeout)
{
NTSTATUS status = STATUS_SUCCESS;
URB *urb;
int i, config_size, tmp_size;
USB_CONFIGURATION_DESCRIPTOR *configuration_descriptor = NULL;
USB_INTERFACE_DESCRIPTOR *interface_descriptor = NULL;
USBD_INTERFACE_INFORMATION *interface_information = NULL;
DEBUG_PRINT_NL();
DEBUG_MESSAGE("set_interface(): interface %d", interface);
DEBUG_MESSAGE("set_interface(): altsetting %d", altsetting);
DEBUG_MESSAGE("set_interface(): timeout %d", timeout);
if(!dev->config.value)
{
DEBUG_ERROR("release_interface(): device is not configured");
return STATUS_INVALID_DEVICE_STATE;
}
configuration_descriptor = get_config_descriptor(dev, dev->config.value,
&config_size);
if(!configuration_descriptor)
{
DEBUG_ERROR("set_interface(): memory_allocation error");
return STATUS_NO_MEMORY;
}
interface_descriptor =
find_interface_desc(configuration_descriptor, config_size,
interface, altsetting);
if(!interface_descriptor)
{
DEBUG_ERROR("set_interface(): interface %d or altsetting %d invalid",
interface, altsetting);
ExFreePool(configuration_descriptor);
return STATUS_UNSUCCESSFUL;
}
tmp_size = sizeof(struct _URB_SELECT_INTERFACE)
+ interface_descriptor->bNumEndpoints
* sizeof(USBD_PIPE_INFORMATION);
urb = ExAllocatePool(NonPagedPool, tmp_size);
if(!urb)
{
DEBUG_ERROR("set_interface(): memory_allocation error");
ExFreePool(configuration_descriptor);
return STATUS_NO_MEMORY;
}
memset(urb, 0, tmp_size);
urb->UrbHeader.Function = URB_FUNCTION_SELECT_INTERFACE;
urb->UrbHeader.Length = (USHORT)tmp_size;
urb->UrbSelectInterface.ConfigurationHandle = dev->config.handle;
urb->UrbSelectInterface.Interface.Length =
sizeof(struct _USBD_INTERFACE_INFORMATION);
urb->UrbSelectInterface.Interface.NumberOfPipes =
interface_descriptor->bNumEndpoints;
urb->UrbSelectInterface.Interface.Length +=
interface_descriptor->bNumEndpoints
* sizeof(struct _USBD_PIPE_INFORMATION);
urb->UrbSelectInterface.Interface.InterfaceNumber = (UCHAR)interface;
urb->UrbSelectInterface.Interface.AlternateSetting = (UCHAR)altsetting;
interface_information = &urb->UrbSelectInterface.Interface;
for(i = 0; i < interface_descriptor->bNumEndpoints; i++)
{
interface_information->Pipes[i].MaximumTransferSize
= LIBUSB_MAX_READ_WRITE;
}
status = call_usbd(dev, urb, IOCTL_INTERNAL_USB_SUBMIT_URB, timeout);
if(!NT_SUCCESS(status) || !USBD_SUCCESS(urb->UrbHeader.Status))
{
DEBUG_ERROR("set_interface(): setting interface failed: status: 0x%x, "
"urb-status: 0x%x", status, urb->UrbHeader.Status);
ExFreePool(configuration_descriptor);
ExFreePool(urb);
return STATUS_UNSUCCESSFUL;
}
update_pipe_info(dev, interface_information);
ExFreePool(configuration_descriptor);
ExFreePool(urb);
return status;
}
