rename device_info_pool to usbh_device_info_pool

update enum task to post set address

3 files changed, 45 insertions, 22 deletions

diff --git a/tinyusb/host/usbh.c b/tinyusb/host/usbh.c
index f60299a06..ffdb2769f 100644
--- a/tinyusb/host/usbh.c
+++ b/tinyusb/host/usbh.c
@@ -56,8 +56,9 @@
 //--------------------------------------------------------------------+
 // INTERNAL OBJECT & FUNCTION DECLARATION
 //--------------------------------------------------------------------+
+static inline uint8_t get_new_address(void) ATTR_ALWAYS_INLINE;
 
-STATIC_ usbh_device_info_t device_info_pool[TUSB_CFG_HOST_DEVICE_MAX];
+STATIC_ usbh_device_info_t usbh_device_info_pool[TUSB_CFG_HOST_DEVICE_MAX];
 
 //--------------------------------------------------------------------+
 // PUBLIC API (Parameter Verification is required)
@@ -65,7 +66,7 @@ STATIC_ usbh_device_info_t device_info_pool[TUSB_CFG_HOST_DEVICE_MAX];
 tusbh_device_status_t tusbh_device_status_get (tusb_handle_device_t const device_hdl)
 {
   ASSERT(device_hdl < TUSB_CFG_HOST_DEVICE_MAX, 0);
-  return device_info_pool[device_hdl].status;
+  return usbh_device_info_pool[device_hdl].status;
 }
 
 //--------------------------------------------------------------------+
@@ -76,11 +77,15 @@ OSAL_TASK_DEF(enum_task, usbh_enumeration_task, 128, OSAL_PRIO_HIGH);
 #define ENUM_QUEUE_DEPTH  5
 OSAL_QUEUE_DEF(enum_queue, ENUM_QUEUE_DEPTH, uin32_t);
 osal_queue_handle_t enum_queue_hdl;
+
 usbh_device_addr0_t device_addr0 TUSB_CFG_ATTR_USBRAM;
 uint8_t enum_data_buffer[TUSB_CFG_HOST_ENUM_BUFFER_SIZE] TUSB_CFG_ATTR_USBRAM;
+
+
 void usbh_enumeration_task(void)
 {
   tusb_error_t error;
+  static uint8_t new_addr;
 
   OSAL_TASK_LOOP_BEGIN
 
@@ -95,9 +100,8 @@ void usbh_enumeration_task(void)
     error = hcd_addr0_open(&device_addr0);
     TASK_ASSERT_STATUS(error);
 
-    {
-      tusb_std_request_t request_device_desc =
-      {
+    { // Get first 8 bytes of device descriptor to get Control Endpoint Size
+      tusb_std_request_t request_device_desc = {
           .bmRequestType = { .direction = TUSB_DIR_DEV_TO_HOST, .type = TUSB_REQUEST_TYPE_STANDARD, .recipient = TUSB_REQUEST_RECIPIENT_DEVICE },
           .bRequest = TUSB_REQUEST_GET_DESCRIPTOR,
           .wValue   = (TUSB_DESC_DEVICE << 8),
@@ -109,22 +113,14 @@ void usbh_enumeration_task(void)
       TASK_ASSERT_STATUS(error);
     }
 
-    {
-      uint8_t new_addr = 0;
-      for (new_addr=0; new_addr<TUSB_CFG_HOST_DEVICE_MAX; new_addr++)
-      {
-        if (device_info_pool[new_addr].status == TUSB_DEVICE_STATUS_UNPLUG)
-          break;
-      }
-
-      TASK_ASSERT(new_addr < TUSB_CFG_HOST_DEVICE_MAX);
+    new_addr = get_new_address();
+    TASK_ASSERT(new_addr < TUSB_CFG_HOST_DEVICE_MAX);
 
-      tusb_std_request_t request_set_address =
-      {
+    { // Set new address
+      tusb_std_request_t request_set_address = {
           .bmRequestType = { .direction = TUSB_DIR_HOST_TO_DEV, .type = TUSB_REQUEST_TYPE_STANDARD, .recipient = TUSB_REQUEST_RECIPIENT_DEVICE },
           .bRequest = TUSB_REQUEST_SET_ADDRESS,
-          .wValue   = (new_addr+1),
-          .wLength  = 8
+          .wValue   = (new_addr+1)
       };
 
       hcd_pipe_control_xfer(device_addr0.pipe_hdl, &request_set_address, NULL);
@@ -132,6 +128,15 @@ void usbh_enumeration_task(void)
       TASK_ASSERT_STATUS(error);
     }
 
+    // update data for the new device
+    usbh_device_info_pool[new_addr].core_id  = device_addr0.enum_entry.core_id;
+    usbh_device_info_pool[new_addr].hub_addr = device_addr0.enum_entry.hub_addr;
+    usbh_device_info_pool[new_addr].hub_port = device_addr0.enum_entry.hub_port;
+    usbh_device_info_pool[new_addr].speed    = device_addr0.speed;
+    usbh_device_info_pool[new_addr].status   = TUSB_DEVICE_STATUS_ADDRESSED;
+
+    hcd_addr0_close(&device_addr0);
+
 
   }else // device connect via a hub
   {
@@ -153,7 +158,7 @@ tusb_error_t usbh_init(void)
 {
   uint32_t i;
 
-  memset(device_info_pool, 0, sizeof(usbh_device_info_t)*TUSB_CFG_HOST_DEVICE_MAX);
+  memset(usbh_device_info_pool, 0, sizeof(usbh_device_info_t)*TUSB_CFG_HOST_DEVICE_MAX);
 
   for(i=0; i<TUSB_CFG_HOST_CONTROLLER_NUM; i++)
   {
@@ -167,3 +172,18 @@ tusb_error_t usbh_init(void)
   return TUSB_ERROR_NONE;
 }
 #endif
+
+//--------------------------------------------------------------------+
+// INTERNAL HELPER
+//--------------------------------------------------------------------+
+static inline uint8_t get_new_address(void)
+{
+  uint8_t new_addr;
+  for (new_addr=0; new_addr<TUSB_CFG_HOST_DEVICE_MAX; new_addr++)
+  {
+    if (usbh_device_info_pool[new_addr].status == TUSB_DEVICE_STATUS_UNPLUG)
+      break;
+  }
+  return new_addr;
+}
+
diff --git a/tinyusb/host/usbh.h b/tinyusb/host/usbh.h
index fcc43395f..03d6cbbc2 100644
--- a/tinyusb/host/usbh.h
+++ b/tinyusb/host/usbh.h
@@ -102,10 +102,12 @@ enum {
 /// Device Status
 enum {
   TUSB_DEVICE_STATUS_UNPLUG      = 0,
-  TUSB_DEVICE_STATUS_READY       = BIT_(0),
+  TUSB_DEVICE_STATUS_ADDRESSED,
 
-  TUSB_DEVICE_STATUS_REMOVING    = BIT_(2),
-  TUSB_DEVICE_STATUS_SAFE_REMOVE = BIT_(3),
+  TUSB_DEVICE_STATUS_READY, /* Configured */
+
+  TUSB_DEVICE_STATUS_REMOVING,
+  TUSB_DEVICE_STATUS_SAFE_REMOVE,
 };
 
 typedef enum {
diff --git a/tinyusb/host/usbh_hcd.h b/tinyusb/host/usbh_hcd.h
index e30bf143b..310de2060 100644
--- a/tinyusb/host/usbh_hcd.h
+++ b/tinyusb/host/usbh_hcd.h
@@ -110,6 +110,7 @@ typedef struct { // TODO internal structure, re-order members
 // ADDRESS 0 API
 //--------------------------------------------------------------------+
 tusb_error_t hcd_addr0_open(usbh_device_addr0_t *dev_addr0) ATTR_WARN_UNUSED_RESULT;
+tusb_error_t hcd_addr0_close(usbh_device_addr0_t *dev_addr0) ATTR_WARN_UNUSED_RESULT;
 
 #ifdef __cplusplus
  }