clean up

1 files changed, 10 insertions, 9 deletions

diff --git a/src/device/usbd_control.c b/src/device/usbd_control.c
index 58b78ff53..b14d08a9c 100644
--- a/src/device/usbd_control.c
+++ b/src/device/usbd_control.c
@@ -71,13 +71,12 @@ uint8_t* usbd_get_ctrl_buf(void) {
 // Per USB 2.0 §9.3.1, when wLength == 0 the Direction bit is ignored and the Status stage
 // is always IN. Otherwise the Status stage is opposite to the Data stage direction.
 TU_ATTR_ALWAYS_INLINE static inline uint8_t status_stage_ep(const tusb_control_request_t* request) {
-  if (request->wLength == 0) return TU_EP0_IN;
-  return request->bmRequestType_bit.direction ? TU_EP0_OUT : TU_EP0_IN;
+  return (request->wLength != 0 && request->bmRequestType_bit.direction) ? TU_EP0_OUT : TU_EP0_IN;
 }
 
 // Queue ZLP status transaction
-TU_ATTR_ALWAYS_INLINE static inline bool status_stage_xact(uint8_t rhport, const tusb_control_request_t* request) {
-  return usbd_edpt_xfer(rhport, status_stage_ep(request), NULL, 0, false);
+TU_ATTR_ALWAYS_INLINE static inline  bool status_stage_xact(uint8_t rhport, uint8_t ep_status) {
+  return usbd_edpt_xfer(rhport, ep_status, NULL, 0, false);
 }
 
 // Status phase
@@ -87,7 +86,7 @@ bool tud_control_status(uint8_t rhport, const tusb_control_request_t* request) {
   _ctrl_xfer.total_xferred = 0;
   _ctrl_xfer.data_len = 0;
 
-  return status_stage_xact(rhport, request);
+  return status_stage_xact(rhport, status_stage_ep(request));
 }
 
 // Queue a transaction in Data Stage
@@ -121,7 +120,7 @@ bool tud_control_xfer(uint8_t rhport, const tusb_control_request_t* request, voi
     }
     TU_ASSERT(data_stage_xact(rhport));
   } else {
-    TU_ASSERT(status_stage_xact(rhport, request));
+    TU_ASSERT(status_stage_xact(rhport, TU_EP0_IN));
   }
 
   return true;
@@ -158,8 +157,9 @@ void usbd_control_set_request(const tusb_control_request_t* request) {
 bool usbd_control_xfer_cb(uint8_t rhport, uint8_t ep_addr, xfer_result_t result, uint32_t xferred_bytes) {
   (void) result;
 
-  // Status Stage complete: callback endpoint matches the Status stage endpoint
-  if (ep_addr == status_stage_ep(&_ctrl_xfer.request)) {
+  // Status Stage complete: endpoint matches the Status stage endpoint
+  uint8_t const ep_status = status_stage_ep(&_ctrl_xfer.request);
+  if (ep_addr == ep_status) {
     TU_ASSERT(0 == xferred_bytes);
 
     // invoke optional dcd hook if available
@@ -173,6 +173,7 @@ bool usbd_control_xfer_cb(uint8_t rhport, uint8_t ep_addr, xfer_result_t result,
     return true;
   }
 
+  // Data stage complete
   if (_ctrl_xfer.request.bmRequestType_bit.direction == TUSB_DIR_OUT) {
     TU_VERIFY(_ctrl_xfer.buffer);
     if (_ctrl_xfer.buffer != _ctrl_epbuf.buf) {
@@ -202,7 +203,7 @@ bool usbd_control_xfer_cb(uint8_t rhport, uint8_t ep_addr, xfer_result_t result,
     }
 
     if (is_ok) {
-      TU_ASSERT(status_stage_xact(rhport, &_ctrl_xfer.request));
+      TU_ASSERT(status_stage_xact(rhport, ep_status));
     } else {
       // Stall both IN and OUT control endpoint
       dcd_edpt_stall(rhport, TU_EP0_OUT);