X-Git-Url: https://git.gag.com/?p=fw%2Faltos;a=blobdiff_plain;f=src%2Fstm-flash%2Fao_stm_flash.c;h=f85807350763628f596958f401dba528951e3bfc;hp=4a91ef1fa4a5e1d234b003ae464e85b8305f1cb8;hb=b131c5ac59bbd339a724892586023a43f97c7f90;hpb=4bc55ee8fabc9f4c997c2a515d74baada590c93e

diff --git a/src/stm-flash/ao_stm_flash.c b/src/stm-flash/ao_stm_flash.c
index 4a91ef1f..f8580735 100644
--- a/src/stm-flash/ao_stm_flash.c
+++ b/src/stm-flash/ao_stm_flash.c
@@ -26,6 +26,17 @@ ao_panic(uint8_t reason)
 	for (;;);
 }
 
+void
+ao_put_string(__code char *s)
+{
+	char	c;
+	while ((c = *s++)) {
+		if (c == '\n')
+			ao_usb_putchar('\r');
+		ao_usb_putchar(c);
+	}
+}
+
 void
 ao_application(void)
 {
@@ -33,30 +44,44 @@ ao_application(void)
 }
 
 static uint32_t
-ao_cmd_hex32(void)
+ao_get_hex32(void)
 {
-	__pdata uint8_t	r = ao_cmd_lex_error;
 	int8_t	n;
 	uint32_t v = 0;
 
-	ao_cmd_white();
+	for (;;) {
+		n = ao_usb_getchar();
+		if (n != ' ')
+			break;
+	}
 	for(;;) {
-		n = ao_cmd_hexchar(ao_cmd_lex_c);
-		if (n < 0)
+		if ('0' <= n && n <= '9')
+			n = n - '0';
+		else if ('a' <= n && n <= 'f')
+			n = n - ('a' - 10);
+		else if ('A' <= n && n <= 'F')
+			n = n - ('A' - 10);
+		else
 			break;
 		v = (v << 4) | n;
-		r = ao_cmd_success;
-		ao_cmd_lex();
+		n = ao_usb_getchar();
 	}
-	if (r != ao_cmd_success)
-		ao_cmd_status = r;
 	return v;
 }
 
+void
+ao_block_erase(void)
+{
+	uint32_t	addr = ao_get_hex32();
+	uint32_t	*p = (uint32_t *) addr;
+
+	ao_flash_erase_page(p);
+}
+
 void
 ao_block_write(void)
 {
-	uint32_t	addr = ao_cmd_hex32();
+	uint32_t	addr = ao_get_hex32();
 	uint32_t	*p = (uint32_t *) addr;
 	union {
 		uint8_t		data8[256];
@@ -64,28 +89,52 @@ ao_block_write(void)
 	} u;
 	uint16_t	i;
 
+	if (addr < (uint32_t) AO_BOOT_APPLICATION_BASE) {
+		ao_put_string("Invalid address\n");
+		return;
+	}
 	for (i = 0; i < 256; i++)
-		u.data8[i] = getchar();
+		u.data8[i] = ao_usb_getchar();
 	ao_flash_page(p, u.data32);
 }
 
 void
 ao_block_read(void)
 {
-	uint32_t	addr = ao_cmd_hex32();
+	uint32_t	addr = ao_get_hex32();
 	uint8_t		*p = (uint8_t *) addr;
 	uint16_t	i;
+	uint8_t		c;
 
-	for (i = 0; i < 256; i++)
-		putchar(*p++);
+	for (i = 0; i < 256; i++) {
+		c = *p++;
+		ao_usb_putchar(c);
+	}
 }
 
-__code struct ao_cmds ao_flash_cmds[] = {
-	{ ao_application, "A\0Switch to application" },
-	{ ao_block_write, "W <addr>\0Write block. 256 binary bytes follow newline" },
-	{ ao_block_read, "R <addr>\0Read block. Returns 256 bytes" },
-	{ 0, NULL },
-};
+static void
+ao_show_version(void)
+{
+	ao_put_string("altos-loader");
+	ao_put_string("\nmanufacturer     "); ao_put_string(ao_manufacturer);
+	ao_put_string("\nproduct          "); ao_put_string(ao_product);
+	ao_put_string("\nsoftware-version "); ao_put_string(ao_version);
+	ao_put_string("\n");
+}
+
+static void
+ao_flash_task(void) {
+	for (;;) {
+		ao_usb_flush();
+		switch (ao_usb_getchar()) {
+		case 'v': ao_show_version(); break;
+		case 'a': ao_application(); break;
+		case 'X': ao_block_erase(); break;
+		case 'W': ao_block_write(); break;
+		case 'R': ao_block_read(); break;
+		}
+	}
+}
 
 
 int
@@ -93,15 +142,11 @@ main(void)
 {
 	ao_clock_init();
 
-	ao_task_init();
-
-	ao_timer_init();
+//	ao_timer_init();
 //	ao_dma_init();
-	ao_cmd_init();
 //	ao_exti_init();
 	ao_usb_init();
 
-	ao_cmd_register(&ao_flash_cmds[0]);
-	ao_start_scheduler();
+	ao_flash_task();
 	return 0;
 }