Add UsbHostDeviceHandler for #7

src/host.rs

@@ -3,17 +3,17 @@ use super::*;
 
 /// A handler to pass requests to a USB device of the host
 #[derive(Clone)]
-pub struct UsbHostHandler {
+pub struct UsbHostInterfaceHandler {
     handle: Arc<Mutex<DeviceHandle<GlobalContext>>>,
 }
 
-impl UsbHostHandler {
+impl UsbHostInterfaceHandler {
     pub fn new(handle: Arc<Mutex<DeviceHandle<GlobalContext>>>) -> Self {
         Self { handle }
     }
 }
 
-impl UsbInterfaceHandler for UsbHostHandler {
+impl UsbInterfaceHandler for UsbHostInterfaceHandler {
     fn handle_urb(
         &mut self,
         _interface: &UsbInterface,
@@ -90,3 +90,55 @@ impl UsbInterfaceHandler for UsbHostHandler {
         self
     }
 }
+
+/// A handler to pass requests to a USB device of the host
+#[derive(Clone)]
+pub struct UsbHostDeviceHandler {
+    handle: Arc<Mutex<DeviceHandle<GlobalContext>>>,
+}
+
+impl UsbHostDeviceHandler {
+    pub fn new(handle: Arc<Mutex<DeviceHandle<GlobalContext>>>) -> Self {
+        Self { handle }
+    }
+}
+
+impl UsbDeviceHandler for UsbHostDeviceHandler {
+    fn handle_urb(&mut self, setup: SetupPacket, req: &[u8]) -> Result<Vec<u8>> {
+        debug!("To host device: setup={:?} req={:?}", setup, req);
+        let mut buffer = [0u8; 1024];
+        let timeout = std::time::Duration::new(1, 0);
+        let handle = self.handle.lock().unwrap();
+        // control
+        if setup.request_type & 0x80 == 0 {
+            // control out
+            handle
+                .write_control(
+                    setup.request_type,
+                    setup.request,
+                    setup.value,
+                    setup.index,
+                    req,
+                    timeout,
+                )
+                .ok();
+        } else {
+            // control in
+            if let Ok(len) = handle.read_control(
+                setup.request_type,
+                setup.request,
+                setup.value,
+                setup.index,
+                &mut buffer,
+                timeout,
+            ) {
+                return Ok(Vec::from(&buffer[..len]));
+            }
+        }
+        Ok(vec![])
+    }
+
+    fn as_any(&mut self) -> &mut dyn Any {
+        self
+    }
+}

src/lib.rs

@@ -81,9 +81,10 @@ impl UsbIpServer {
                         });
                     }
 
-                    let handler =
-                        Arc::new(Mutex::new(Box::new(UsbHostHandler::new(handle.clone()))
-                            as Box<dyn UsbInterfaceHandler + Send>));
+                    let handler = Arc::new(Mutex::new(Box::new(UsbHostInterfaceHandler::new(
+                        handle.clone(),
+                    ))
+                        as Box<dyn UsbInterfaceHandler + Send>));
                     interfaces.push(UsbInterface {
                         interface_class: intf_desc.class_code(),
                         interface_subclass: intf_desc.sub_class_code(),
@@ -130,6 +131,9 @@ impl UsbIpServer {
                         interval: 0,
                     },
                     interfaces,
+                    device_handler: Some(Arc::new(Mutex::new(Box::new(
+                        UsbHostDeviceHandler::new(handle.clone()),
+                    )))),
                     ..UsbDevice::default()
                 };