2 * Copyright © 2014 Keith Packard <keithp@keithp.com>
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; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
34 #include "ao-verbose.h"
36 static const struct option options[] = {
37 { .name = "tty", .has_arg = 1, .val = 'T' },
38 { .name = "device", .has_arg = 1, .val = 'D' },
39 { .name = "raw", .has_arg = 0, .val = 'r' },
40 { .name = "verbose", .has_arg = 1, .val = 'v' },
44 static void usage(char *program)
46 fprintf(stderr, "usage: %s [--verbose=<verbose>] [--device=<device>] [-tty=<tty>] <expected-size>\n", program);
51 done(struct cc_usb *cc, int code)
58 ends_with(char *whole, char *suffix)
60 int whole_len = strlen(whole);
61 int suffix_len = strlen(suffix);
63 if (suffix_len > whole_len)
65 return strcmp(whole + whole_len - suffix_len, suffix) == 0;
69 starts_with(char *whole, char *prefix)
71 int whole_len = strlen(whole);
72 int prefix_len = strlen(prefix);
74 if (prefix_len > whole_len)
76 return strncmp(whole, prefix, prefix_len) == 0;
81 char **strs = malloc (sizeof (char *)), *str;
84 while ((str = strtok(line, " \t"))) {
86 strs = realloc(strs, (n + 2) * sizeof (char *));
87 strs[n] = strdup(str);
95 free_strs(char **strs) {
99 for (i = 0; (str = strs[i]) != NULL; i++)
110 static struct flash *
111 flash(struct cc_usb *usb)
113 struct flash *head = NULL, **tail = &head;
114 cc_usb_printf(usb, "f\nv\n");
119 cc_usb_getline(usb, line, sizeof (line));
120 b = malloc (sizeof (struct flash));
121 strcpy(b->line, line);
126 if (strstr(line, "software-version"))
133 free_flash(struct flash *b) {
145 find_flash(struct flash *b, char *word0) {
147 for (;b; b = b->next) {
148 if (strstr(b->line, word0))
155 do_flash(struct cc_usb *usb, int expected_size) {
156 struct flash *b = flash(usb);
157 char **size = find_flash(b, "Storage size:");
158 char **erase = find_flash(b, "Storage erase unit:");
160 if (!size || !erase) {
161 printf("no response\n");
166 int actual_size = atoi(size[2]);
168 if (actual_size != expected_size) {
169 printf ("weird flash size %d != %d\n", actual_size, expected_size);
174 int actual_erase = atoi(erase[3]);
176 if (actual_erase != 65536) {
177 printf ("weird erase size %d\n", actual_erase);
182 printf ("flash size %d erase block %d\n", actual_size, actual_erase);
188 main (int argc, char **argv)
197 struct cc_usb *cc = NULL;
204 while ((c = getopt_long(argc, argv, "rT:D:c:s:v:", options, NULL)) != -1) {
224 ao_verbose = verbose;
227 ccdbg_add_debug(CC_DEBUG_BITBANG);
230 tty = cc_usbdevs_find_by_arg(device, "AltosFlash");
232 tty = cc_usbdevs_find_by_arg(device, "TeleMega");
234 tty = getenv("ALTOS_TTY");
238 cc = cc_usb_open(tty);
243 if (!do_flash(cc, atoi(argv[optind])))