Label binaries with product and serial info
[fw/altos] / ao_cmd.c
1 /*
2  * Copyright © 2009 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
20 __xdata uint16_t ao_cmd_lex_i;
21 __xdata uint8_t ao_cmd_lex_c;
22 __xdata enum ao_cmd_status ao_cmd_status;
23 static __xdata uint8_t  lex_echo;
24
25 #define CMD_LEN 32
26
27 static __xdata uint8_t  cmd_line[CMD_LEN];
28 static __xdata uint8_t  cmd_len;
29 static __xdata uint8_t  cmd_i;
30
31 static void
32 put_string(char *s)
33 {
34         __xdata uint8_t c;
35         while (c = *s++)
36                 putchar(c);
37 }
38
39 static void
40 readline(void)
41 {
42         __xdata uint8_t c;
43         if (lex_echo)
44                 put_string("> ");
45         cmd_len = 0;
46         for (;;) {
47                 flush();
48                 c = getchar();
49                 /* backspace/delete */
50                 if (c == '\010' || c == '\177') {
51                         if (cmd_len != 0) {
52                                 if (lex_echo)
53                                         put_string("\010 \010");
54                                 --cmd_len;
55                         }
56                         continue;
57                 }
58
59                 /* ^U */
60                 if (c == '\025') {
61                         while (cmd_len != 0) {
62                                 if (lex_echo)
63                                         put_string("\010 \010");
64                                 --cmd_len;
65                         }
66                         continue;
67                 }
68
69                 /* map CR to NL */
70                 if (c == '\r')
71                         c = '\n';
72                 
73                 if (c == '\n') {
74                         if (lex_echo)
75                                 putchar('\n');
76                         break;
77                 }
78
79                 if (cmd_len >= CMD_LEN - 2) {
80                         if (lex_echo)
81                                 putchar('\007');
82                         continue;
83                 }
84                 cmd_line[cmd_len++] = c;
85                 if (lex_echo)
86                         putchar(c);
87         }
88         cmd_line[cmd_len++] = '\n';
89         cmd_line[cmd_len++] = '\0';
90         cmd_i = 0;
91 }
92
93 void
94 ao_cmd_lex(void)
95 {
96         ao_cmd_lex_c = '\n';
97         if (cmd_i < cmd_len)
98                 ao_cmd_lex_c = cmd_line[cmd_i++];
99 }
100
101 static void
102 putnibble(uint8_t v)
103 {
104         if (v < 10)
105                 putchar(v + '0');
106         else
107                 putchar(v + ('a' - 10));
108 }
109
110 void
111 ao_cmd_put16(uint16_t v)
112 {
113         int8_t i;
114         for (i = 3; i >= 0; i--)
115                 putnibble((v >> (i << 2)) & 0xf);
116 }
117
118 void
119 ao_cmd_put8(uint8_t v)
120 {
121         putnibble((v >> 4) & 0xf);
122         putnibble(v & 0xf);
123 }
124
125 void
126 ao_cmd_white(void)
127 {
128         while (ao_cmd_lex_c == ' ' || ao_cmd_lex_c == '\t')
129                 ao_cmd_lex();
130 }
131
132 void
133 ao_cmd_hex(void)
134 {
135         __xdata uint8_t r = ao_cmd_lex_error;
136         
137         ao_cmd_lex_i = 0;
138         ao_cmd_white();
139         for(;;) {
140                 if ('0' <= ao_cmd_lex_c && ao_cmd_lex_c <= '9')
141                         ao_cmd_lex_i = (ao_cmd_lex_i << 4) | (ao_cmd_lex_c - '0');
142                 else if ('a' <= ao_cmd_lex_c && ao_cmd_lex_c <= 'f')
143                         ao_cmd_lex_i = (ao_cmd_lex_i << 4) | (ao_cmd_lex_c - 'a' + 10);
144                 else if ('A' <= ao_cmd_lex_c && ao_cmd_lex_c <= 'F')
145                         ao_cmd_lex_i = (ao_cmd_lex_i << 4) | (ao_cmd_lex_c - 'A' + 10);
146                 else
147                         break;
148                 r = ao_cmd_success;
149                 ao_cmd_lex();
150         }
151         if (r != ao_cmd_success)
152                 ao_cmd_status = r;
153 }
154
155 void
156 ao_cmd_decimal(void)
157 {
158         __xdata uint8_t r = ao_cmd_lex_error;
159         
160         ao_cmd_lex_i = 0;
161         ao_cmd_white();
162         for(;;) {
163                 if ('0' <= ao_cmd_lex_c && ao_cmd_lex_c <= '9')
164                         ao_cmd_lex_i = (ao_cmd_lex_i * 10) + (ao_cmd_lex_c - '0');
165                 else
166                         break;
167                 r = ao_cmd_success;
168                 ao_cmd_lex();
169         }
170         if (r != ao_cmd_success)
171                 ao_cmd_status = r;
172 }
173
174 static void
175 eol(void)
176 {
177         while (ao_cmd_lex_c != '\n')
178                 ao_cmd_lex();
179 }
180
181 static void
182 dump(void)
183 {
184         __xdata uint16_t c;
185         __xdata uint8_t * __xdata start, * __xdata end;
186
187         ao_cmd_hex();
188         start = (uint8_t __xdata *) ao_cmd_lex_i;
189         ao_cmd_hex();
190         end = (uint8_t __xdata *) ao_cmd_lex_i;
191         if (ao_cmd_status != ao_cmd_success)
192                 return;
193         c = 0;
194         while (start <= end) {
195                 if ((c & 7) == 0) {
196                         if (c)
197                                 putchar('\n');
198                         ao_cmd_put16((uint16_t) start);
199                 }
200                 putchar(' ');
201                 ao_cmd_put8(*start);
202                 ++c;
203                 start++;
204         }
205         putchar('\n');
206 }
207
208 static void
209 echo(void)
210 {
211         ao_cmd_hex();
212         lex_echo = ao_cmd_lex_i != 0;
213 }
214
215 static const uint8_t help_txt[] = "All numbers are in hex";
216
217 #define NUM_CMDS        10
218
219 static __code struct ao_cmds    *__xdata (ao_cmds[NUM_CMDS]);
220 static __xdata uint8_t          ao_ncmds;
221
222 static void
223 help(void)
224 {
225         __xdata uint8_t cmds;
226         __xdata uint8_t cmd;
227         __code struct ao_cmds * __xdata cs;
228         puts(help_txt);
229         for (cmds = 0; cmds < ao_ncmds; cmds++) {
230                 cs = ao_cmds[cmds];
231                 for (cmd = 0; cs[cmd].cmd != '\0'; cmd++)
232                         puts(cs[cmd].help);
233         }
234 }
235
236 static void
237 report(void)
238 {
239         switch(ao_cmd_status) {
240         case ao_cmd_lex_error:
241         case ao_cmd_syntax_error:
242                 puts("Syntax error");
243                 ao_cmd_status = 0;
244                 break;
245         }
246 }
247
248 void
249 ao_cmd_register(__code struct ao_cmds *cmds)
250 {
251         if (ao_ncmds >= NUM_CMDS)
252                 ao_panic(AO_PANIC_CMD);
253         ao_cmds[ao_ncmds++] = cmds;
254 }
255
256 void
257 ao_cmd(void *parameters)
258 {
259         __xdata uint8_t c;
260         __xdata uint8_t cmd, cmds;
261         __code struct ao_cmds * __xdata cs;
262         void (*__xdata func)(void);
263         (void) parameters;
264
265         lex_echo = 1;
266         for (;;) {
267                 readline();
268                 ao_cmd_lex();
269                 ao_cmd_white();
270                 c = ao_cmd_lex_c;
271                 ao_cmd_lex();
272                 if (c == '\r' || c == '\n')
273                         continue;
274                 func = (void (*)(void)) NULL;
275                 for (cmds = 0; cmds < ao_ncmds; cmds++) {
276                         cs = ao_cmds[cmds];
277                         for (cmd = 0; cs[cmd].cmd != '\0'; cmd++)
278                                 if (cs[cmd].cmd == c) {
279                                         func = cs[cmd].func;
280                                         break;
281                                 }
282                         if (func)
283                                 break;
284                 }
285                 if (func)
286                         (*func)();
287                 else
288                         ao_cmd_status = ao_cmd_syntax_error;
289                 report();
290         }
291 }
292
293 __xdata struct ao_task ao_cmd_task;
294
295 __code struct ao_cmds   ao_base_cmds[] = {
296         { '?', help,            "?                                  Print this message" },
297         { 'T', ao_task_info,    "T                                  Show task states" },
298         { 'E', echo,            "E <0 off, 1 on>                    Set command echo mode" },
299         { 'd', dump,            "d <start> <end>                    Dump memory" },
300         { 0,    help,   NULL },
301 };
302
303 void
304 ao_cmd_init(void)
305 {
306         ao_cmd_register(&ao_base_cmds[0]);
307         ao_add_task(&ao_cmd_task, ao_cmd, "cmd");
308 }