Fix all stdio reading functions to be __critical

[fw/altos] / src / ao_usb.c

diff --git a/src/ao_usb.c b/src/ao_usb.c
index daca71a734e911995045318b6afe802ac60b3581..e4af8e454625c87e1457b44475a33608d4155acb 100644 (file)
--- a/src/ao_usb.c
+++ b/src/ao_usb.c
@@ -377,11 +377,13 @@ char
 ao_usb_pollchar(void) __critical
 {
        char c;
-       while (ao_usb_out_bytes == 0) {
+       if (ao_usb_out_bytes == 0) {
                USBINDEX = AO_USB_OUT_EP;
                if ((USBCSOL & USBCSOL_OUTPKT_RDY) == 0)
                        return AO_READ_AGAIN;
                ao_usb_out_bytes = (USBCNTH << 8) | USBCNTL;
+               if (ao_usb_out_bytes == 0)
+                       return AO_READ_AGAIN;
        }
        --ao_usb_out_bytes;
        c = USBFIFO[AO_USB_OUT_EP << 1];
@@ -393,7 +395,7 @@ ao_usb_pollchar(void) __critical
 }
 
 char
-ao_usb_getchar(void)
+ao_usb_getchar(void) __critical
 {
        char    c;