Merge pull request #311 from LedgerHQ/pr-io-revamp

IO revamp compat

Author: bigspider

src/ui/display.c

@@ -76,12 +76,18 @@ static bool io_ui_process(dispatcher_context_t *context, bool set_dirty) {
     // We are not waiting for the client's input, nor we are doing computations on the device
     io_clear_processing_timeout();
 
+#ifdef REVAMPED_IO
+    do {
+        io_seproxyhal_io_heartbeat();
+    } while (!g_ux_flow_ended);
+#else   // !REVAMPED_IO
     io_seproxyhal_general_status();
     do {
         io_seproxyhal_spi_recv(G_io_seproxyhal_spi_buffer, sizeof(G_io_seproxyhal_spi_buffer), 0);
         io_seproxyhal_handle_event();
         io_seproxyhal_general_status();
     } while (io_seproxyhal_spi_is_status_sent() && !g_ux_flow_ended);
+#endif  // !REVAMPED_IO
 
     // We're back at work, we want to show the "Processing..." screen when appropriate
     io_start_processing_timeout();