altos: Add actual flashing functions to stm-flash app
[fw/altos] / src / stm-flash / ao_stm_flash.c
1 /*
2  * Copyright © 2011 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include "ao.h"
19 #include <ao_exti.h>
20 #include <ao_boot.h>
21 #include <ao_flash_stm.h>
22
23 void
24 ao_panic(uint8_t reason)
25 {
26         for (;;);
27 }
28
29 void
30 ao_application(void)
31 {
32         ao_boot_reboot(AO_BOOT_APPLICATION_BASE);
33 }
34
35 static uint32_t
36 ao_cmd_hex32(void)
37 {
38         __pdata uint8_t r = ao_cmd_lex_error;
39         int8_t  n;
40         uint32_t v = 0;
41
42         ao_cmd_white();
43         for(;;) {
44                 n = ao_cmd_hexchar(ao_cmd_lex_c);
45                 if (n < 0)
46                         break;
47                 v = (v << 4) | n;
48                 r = ao_cmd_success;
49                 ao_cmd_lex();
50         }
51         if (r != ao_cmd_success)
52                 ao_cmd_status = r;
53         return v;
54 }
55
56 void
57 ao_block_write(void)
58 {
59         uint32_t        addr = ao_cmd_hex32();
60         uint32_t        *p = (uint32_t *) addr;
61         union {
62                 uint8_t         data8[256];
63                 uint32_t        data32[64];
64         } u;
65         uint16_t        i;
66
67         for (i = 0; i < 256; i++)
68                 u.data8[i] = getchar();
69         ao_flash_page(p, u.data32);
70 }
71
72 void
73 ao_block_read(void)
74 {
75         uint32_t        addr = ao_cmd_hex32();
76         uint8_t         *p = (uint8_t *) addr;
77         uint16_t        i;
78
79         for (i = 0; i < 256; i++)
80                 putchar(*p++);
81 }
82
83 __code struct ao_cmds ao_flash_cmds[] = {
84         { ao_application, "A\0Switch to application" },
85         { ao_block_write, "W <addr>\0Write block. 256 binary bytes follow newline" },
86         { ao_block_read, "R <addr>\0Read block. Returns 256 bytes" },
87         { 0, NULL },
88 };
89
90
91 int
92 main(void)
93 {
94         ao_clock_init();
95
96         ao_task_init();
97
98         ao_timer_init();
99 //      ao_dma_init();
100         ao_cmd_init();
101 //      ao_exti_init();
102         ao_usb_init();
103
104         ao_cmd_register(&ao_flash_cmds[0]);
105         ao_start_scheduler();
106         return 0;
107 }