libtock-sync: console-write: use yield-waitfor

tock · Jun 14, 2024 · 48d5105 · 48d5105
1 parent c734ec5
commit 48d5105
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/libtock-sync/interface/console.c b/libtock-sync/interface/console.c
@@ -22,10 +22,11 @@ returncode_t libtocksync_console_write(const uint8_t* buffer, uint32_t length, i
   if (err != RETURNCODE_SUCCESS) return err;
 
   // Wait for the callback.
-  yield_for(&result.fired);
-  if (result.result != RETURNCODE_SUCCESS) return result.result;
+  libtock_console_write_done_set_upcall(NULL, NULL);
+  yield_waitfor_return_t yval = yield_wait_for(DRIVER_NUM_CONSOLE, 1);
+  if (yval.data0 != RETURNCODE_SUCCESS) return yval.data0;
 
-  *written = result.length;
+  *written = yval.data1;
   return RETURNCODE_SUCCESS;
 }