3 files changed, 20 insertions, 13 deletions

diff --git a/tools/usb/ffs-aio-example/multibuff/host_app/test.c b/tools/usb/ffs-aio-example/multibuff/host_app/test.c
index daa3abe6bebd..2cbcce6e8dd7 100644
--- a/tools/usb/ffs-aio-example/multibuff/host_app/test.c
+++ b/tools/usb/ffs-aio-example/multibuff/host_app/test.c
@@ -33,11 +33,6 @@
 #define VENDOR  0x1d6b
 #define PRODUCT 0x0105
 
-/* endpoints indexes */
-
-#define EP_BULK_IN      (1 | LIBUSB_ENDPOINT_IN)
-#define EP_BULK_OUT     (2 | LIBUSB_ENDPOINT_OUT)
-
 #define BUF_LEN         8192
 
 /*
@@ -159,14 +154,21 @@ void test_exit(struct test_state *state)
 int main(void)
 {
         struct test_state state;
+        struct libusb_config_descriptor *conf;
+        struct libusb_interface_descriptor const *iface;
+        unsigned char addr;
 
         if (test_init(&state))
                 return 1;
 
+        libusb_get_config_descriptor(state.found, 0, &conf);
+        iface = &conf->interface[0].altsetting[0];
+        addr = iface->endpoint[0].bEndpointAddress;
+
         while (1) {
                 static unsigned char buffer[BUF_LEN];
                 int bytes;
-                libusb_bulk_transfer(state.handle, EP_BULK_IN, buffer, BUF_LEN,
+                libusb_bulk_transfer(state.handle, addr, buffer, BUF_LEN,
                                      &bytes, 500);
         }
         test_exit(&state);
diff --git a/tools/usb/ffs-aio-example/simple/device_app/aio_simple.c b/tools/usb/ffs-aio-example/simple/device_app/aio_simple.c
index adc310a6d489..1f44a29818bf 100644
--- a/tools/usb/ffs-aio-example/simple/device_app/aio_simple.c
+++ b/tools/usb/ffs-aio-example/simple/device_app/aio_simple.c
@@ -103,12 +103,14 @@ static const struct {
                         .bDescriptorType = USB_DT_ENDPOINT,
                         .bEndpointAddress = 1 | USB_DIR_IN,
                         .bmAttributes = USB_ENDPOINT_XFER_BULK,
+                        .wMaxPacketSize = htole16(512),
                 },
                 .bulk_source = {
                         .bLength = sizeof(descriptors.hs_descs.bulk_source),
                         .bDescriptorType = USB_DT_ENDPOINT,
                         .bEndpointAddress = 2 | USB_DIR_OUT,
                         .bmAttributes = USB_ENDPOINT_XFER_BULK,
+                        .wMaxPacketSize = htole16(512),
                 },
         },
 };
diff --git a/tools/usb/ffs-aio-example/simple/host_app/test.c b/tools/usb/ffs-aio-example/simple/host_app/test.c
index acd6332811f3..aed86ffff280 100644
--- a/tools/usb/ffs-aio-example/simple/host_app/test.c
+++ b/tools/usb/ffs-aio-example/simple/host_app/test.c
@@ -33,11 +33,6 @@
 #define VENDOR  0x1d6b
 #define PRODUCT 0x0105
 
-/* endpoints indexes */
-
-#define EP_BULK_IN      (1 | LIBUSB_ENDPOINT_IN)
-#define EP_BULK_OUT     (2 | LIBUSB_ENDPOINT_OUT)
-
 #define BUF_LEN         8192
 
 /*
@@ -159,16 +154,24 @@ void test_exit(struct test_state *state)
 int main(void)
 {
         struct test_state state;
+        struct libusb_config_descriptor *conf;
+        struct libusb_interface_descriptor const *iface;
+        unsigned char in_addr, out_addr;
 
         if (test_init(&state))
                 return 1;
 
+        libusb_get_config_descriptor(state.found, 0, &conf);
+        iface = &conf->interface[0].altsetting[0];
+        in_addr = iface->endpoint[0].bEndpointAddress;
+        out_addr = iface->endpoint[1].bEndpointAddress;
+
         while (1) {
                 static unsigned char buffer[BUF_LEN];
                 int bytes;
-                libusb_bulk_transfer(state.handle, EP_BULK_IN, buffer, BUF_LEN,
+                libusb_bulk_transfer(state.handle, in_addr, buffer, BUF_LEN,
                                      &bytes, 500);
-                libusb_bulk_transfer(state.handle, EP_BULK_OUT, buffer, BUF_LEN,
+                libusb_bulk_transfer(state.handle, out_addr, buffer, BUF_LEN,
                                      &bytes, 500);
         }
         test_exit(&state);