altos/lpc: Stop sending SETUP IN when the requested size is reached

[fw/altos] / src / lpc / ao_usb_lpc.c

diff --git a/src/lpc/ao_usb_lpc.c b/src/lpc/ao_usb_lpc.c
index 144d10752b3ba2eb8642c1173586b4bb0eac71db..686dc3a4bd8019d6c2e44251c1bbb10a01f586be 100644 (file)
--- a/src/lpc/ao_usb_lpc.c
+++ b/src/lpc/ao_usb_lpc.c
@@ -58,6 +58,7 @@ static uint8_t        ao_usb_ep0_state;
 /* Pending EP0 IN data */
 static const uint8_t   *ao_usb_ep0_in_data;    /* Remaining data */
 static uint8_t                 ao_usb_ep0_in_len;      /* Remaining amount */
+static uint16_t                ao_usb_ep0_in_max;      /* Requested amount from host */
 
 /* Temp buffer for smaller EP0 in data */
 static uint8_t ao_usb_ep0_in_buf[2];
@@ -316,6 +317,13 @@ ao_usb_disable_epn(uint8_t n)
        ao_usb_disable_ep(&lpc_usb_endpoint.epn[n-1].in[1]);
 }
 
+static void
+ao_usb_reset(void)
+{
+       ao_usb_set_address(0);
+       ao_usb_configuration = 0;
+}
+
 static void
 ao_usb_set_ep0(void)
 {
@@ -326,11 +334,8 @@ ao_usb_set_ep0(void)
        lpc_usb.epinuse = 0;
        lpc_usb.epskip = 0xffffffff;
 
-       ao_usb_set_address(0);
        lpc_usb.intstat = 0xc00003ff;
 
-       ao_usb_configuration = 0;
-
        ao_usb_sram = lpc_usb_sram;
 
        lpc_usb.epliststart = (uint32_t) (intptr_t) &lpc_usb_endpoint;
@@ -346,7 +351,7 @@ ao_usb_set_ep0(void)
        /* Clear all of the other endpoints */
        for (e = 1; e <= 4; e++)
                ao_usb_disable_epn(e);
-
+       ao_usb_reset();
 }
 
 static void
@@ -376,10 +381,11 @@ ao_usb_ep0_flush(void)
        if (this_len > AO_USB_CONTROL_SIZE)
                this_len = AO_USB_CONTROL_SIZE;
 
-       if (this_len < AO_USB_CONTROL_SIZE)
-               ao_usb_ep0_state = AO_USB_EP0_IDLE;
-
        ao_usb_ep0_in_len -= this_len;
+       ao_usb_ep0_in_max -= this_len;
+
+       if (this_len < AO_USB_CONTROL_SIZE || ao_usb_ep0_in_max == 0)
+               ao_usb_ep0_state = AO_USB_EP0_IDLE;
 
        debug_data ("Flush EP0 len %d:", this_len);
        memcpy(ao_usb_ep0_tx_buffer, ao_usb_ep0_in_data, this_len);
@@ -450,8 +456,9 @@ ao_usb_ep0_out_set(uint8_t *data, uint8_t len)
 }
 
 static void
-ao_usb_ep0_in_start(uint8_t max)
+ao_usb_ep0_in_start(uint16_t max)
 {
+       ao_usb_ep0_in_max = max;
        /* Don't send more than asked for */
        if (ao_usb_ep0_in_len > max)
                ao_usb_ep0_in_len = max;
@@ -590,7 +597,7 @@ ao_usb_ep0_handle(uint8_t receive)
 
        if (receive & AO_USB_EP0_GOT_RESET) {
                debug ("\treset\n");
-               ao_usb_set_ep0();
+               ao_usb_reset();
                return;
        }
        if (receive & AO_USB_EP0_GOT_SETUP) {
@@ -936,11 +943,11 @@ ao_usb_enable(void)
        for (t = 0; t < 1000; t++)
                ao_arch_nop();
 
+       ao_usb_set_ep0();
+
 #if HAS_USB_PULLUP
        ao_gpio_set(AO_USB_PULLUP_PORT, AO_USB_PULLUP_PIN, AO_USB_PULLUP, 1);
 #endif
-
-       ao_usb_set_ep0();
 }
 
 #if USB_ECHO