Improve IN endpoint handling

greatscottgadgets · Feb 28, 2025 · b220fc6 · b220fc6
1 parent 5acf942
commit b220fc6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/facedancer/backends/hydradancer.py b/facedancer/backends/hydradancer.py
@@ -391,14 +391,20 @@ def process_events(self):
             elif event.type == HydradancerEvent.EVENT_BUS_RESET:
                 self.handle_bus_reset()
             elif event.type == HydradancerEvent.EVENT_IN_BUFFER_AVAILABLE and event.value != 0 and (event.value in self.ep_in.keys()):
-                self.connected_device.handle_buffer_empty(self.ep_in[event.value])
+                if self.api.in_buffer_empty(event.value):
+                    self.connected_device.handle_buffer_empty(self.ep_in[event.value])
             elif event.type == HydradancerEvent.EVENT_OUT_BUFFER_AVAILABLE:
                 if event.value != 0:
                     self.handle_out_data_endpoint(event.value)
                 else:
                     self.handle_control_request()
             elif event.value != 0 and (event.type == HydradancerEvent.EVENT_IN_BUFFER_AVAILABLE or event.type == HydradancerEvent.EVENT_NAK):
                 self.handle_in_data_endpoint(event.value)
+
+        for ep_num, ep in self.ep_in.items():
+            if self.api.in_buffer_empty(ep_num):
+                self.connected_device.handle_buffer_empty(self.ep_in[ep_num])
+                self.handle_in_data_endpoint(ep_num)
 
     def service_irqs(self):
         """