X-Git-Url: https://git.gag.com/?p=fw%2Faltos;a=blobdiff_plain;f=src%2Flpc%2Fao_usb_lpc.c;h=686dc3a4bd8019d6c2e44251c1bbb10a01f586be;hp=144d10752b3ba2eb8642c1173586b4bb0eac71db;hb=a04c1dd5df76c9127615bc797a9d9f764eec1234;hpb=4ff54bb96f6c00c0c2c7dd32f81403bac331621a

diff --git a/src/lpc/ao_usb_lpc.c b/src/lpc/ao_usb_lpc.c
index 144d1075..686dc3a4 100644
--- 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