git.gag.com Git - fw/altos/blob - ao-tools/ao-load/ao-load.c

   1 /*
   2  * Copyright © 2008 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; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   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.
  13  *
  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.
  17  */
  18
  19 #include <stdlib.h>
  20 #include <limits.h>
  21 #include <stdint.h>
  22 #include <unistd.h>
  23 #include <getopt.h>
  24 #include "ccdbg.h"
  25
  26 #define AO_USB_DESC_STRING              3
  27
  28 struct sym {
  29         unsigned        addr;
  30         char            *name;
  31 } serial_symbols[] = {
  32         { 0,    "_ao_serial_number" },
  33 #define AO_SERIAL_NUMBER        (serial_symbols[0].addr)
  34         { 0,    "_ao_usb_descriptors" },
  35 #define AO_USB_DESCRIPTORS      (serial_symbols[1].addr)
  36 };
  37
  38 #define NUM_SERIAL_SYMBOLS      (sizeof(serial_symbols)/sizeof(serial_symbols[0]))
  39
  40 static int
  41 find_symbols(FILE *map)
  42 {
  43         char    line[2048];
  44         char    *addr, *addr_end;
  45         char    *name;
  46         char    *save;
  47         char    *colon;
  48         unsigned long   a;
  49         int     s;
  50         int     found = 0;
  51
  52         while (fgets(line, sizeof(line), map) != NULL) {
  53                 line[sizeof(line)-1] = '\0';
  54                 addr = strtok_r(line, " \t\n", &save);
  55                 if (!addr)
  56                         continue;
  57                 name = strtok_r(NULL, " \t\n", &save);
  58                 if (!name)
  59                         continue;
  60                 colon = strchr (addr, ':');
  61                 if (!colon)
  62                         continue;
  63                 a = strtoul(colon+1, &addr_end, 16);
  64                 if (a == ULONG_MAX || addr_end == addr)
  65                         continue;
  66                 for (s = 0; s < NUM_SERIAL_SYMBOLS; s++)
  67                         if (!strcmp(serial_symbols[s].name, name)) {
  68                                 serial_symbols[s].addr = (unsigned) a;
  69                                 ++found;
  70                                 break;
  71                         }
  72         }
  73         return found == NUM_SERIAL_SYMBOLS;
  74 }
  75
  76 static int
  77 rewrite(struct hex_image *image, unsigned addr, char *data, int len)
  78 {
  79         int i;
  80         if (addr < image->address || image->address + image->length < addr + len)
  81                 return 0;
  82         printf("rewrite %04x:", addr);
  83         for (i = 0; i < len; i++)
  84                 printf (" %02x", image->data[addr - image->address + i]);
  85         printf(" ->");
  86         for (i = 0; i < len; i++)
  87                 printf (" %02x", data[i]);
  88         printf("\n");
  89         memcpy(image->data + addr - image->address, data, len);
  90 }
  91
  92 <<<<<<< HEAD:ao-tools/ao-load/ao-load.c
  93 static const struct option
  94 =======
  95 static const struct option options[] = {
  96         { .name = "tty", .has_arg = 1, .val = 'T' },
  97         { 0, 0, 0, 0},
  98 };
  99
 100 static void usage(char *program)
 101 {
 102         fprintf(stderr, "usage: %s [--tty <tty-name>] file.ihx serial-number\n", program);
 103         exit(1);
 104 }
 105
 106 >>>>>>> c29275b72438637d46d7a50742882d2736eb176a:ao-tools/ao-load/ao-load.c
 107 int
 108 main (int argc, char **argv)
 109 {
 110         struct ccdbg    *dbg;
 111         uint8_t         status;
 112         uint16_t        pc;
 113         struct hex_file *hex;
 114         struct hex_image *image;
 115         char            *filename;
 116         FILE            *file;
 117         FILE            *map;
 118         char            *serial_string;
 119         unsigned int    serial;
 120         char            *mapname, *dot;
 121         char            *serial_ucs2;
 122         int             serial_ucs2_len;
 123         char            serial_int[2];
 124         unsigned int    s;
 125         int             i;
 126         unsigned        usb_descriptors;
 127         int             string_num;
 128         char            *tty = NULL;
 129         int             c;
 130
 131         while ((c = getopt_long(argc, argv, "T:", options, NULL)) != -1) {
 132                 switch (c) {
 133                 case 'T':
 134                         tty = optarg;
 135                         break;
 136                 default:
 137                         usage(argv[0]);
 138                         break;
 139                 }
 140         }
 141         filename = argv[optind];
 142         if (filename == NULL)
 143                 usage(argv[0]);
 144         mapname = strdup(filename);
 145         dot = strrchr(mapname, '.');
 146         if (!dot || strcmp(dot, ".ihx") != 0)
 147                 usage(argv[0]);
 148         strcpy(dot, ".map");
 149
 150         serial_string = argv[optind + 1];
 151         if (serial_string == NULL)
 152                 usage(argv[0]);
 153
 154         file = fopen(filename, "r");
 155         if (!file) {
 156                 perror(filename);
 157                 exit(1);
 158         }
 159         map = fopen(mapname, "r");
 160         if (!map) {
 161                 perror(mapname);
 162                 exit(1);
 163         }
 164         if (!find_symbols(map)) {
 165                 fprintf(stderr, "Cannot find symbols in \"%s\"\n", mapname);
 166                 exit(1);
 167         }
 168         fclose(map);
 169
 170         hex = ccdbg_hex_file_read(file, filename);
 171         fclose(file);
 172         if (!hex) {
 173                 perror(filename);
 174                 exit (1);
 175         }
 176         image = ccdbg_hex_image_create(hex);
 177         if (!image) {
 178                 fprintf(stderr, "image create failed\n");
 179                 exit (1);
 180         }
 181         ccdbg_hex_file_free(hex);
 182
 183         serial = strtoul(serial_string, NULL, 0);
 184         if (!serial)
 185 (argv[0]);
 186
 187         serial_int[0] = serial & 0xff;
 188         serial_int[1] = (serial >> 8) & 0xff;
 189
 190         if (!rewrite(image, AO_SERIAL_NUMBER, serial_int, sizeof (serial_int))) {
 191                 fprintf(stderr, "Cannot rewrite serial integer at %04x\n",
 192                         AO_SERIAL_NUMBER);
 193                 exit(1);
 194         }
 195
 196         usb_descriptors = AO_USB_DESCRIPTORS - image->address;
 197         string_num = 0;
 198         while (image->data[usb_descriptors] != 0 && usb_descriptors < image->length) {
 199                 if (image->data[usb_descriptors+1] == AO_USB_DESC_STRING) {
 200                         ++string_num;
 201                         if (string_num == 4)
 202                                 break;
 203                 }
 204                 usb_descriptors += image->data[usb_descriptors];
 205         }
 206         if (usb_descriptors >= image->length || image->data[usb_descriptors] == 0 ) {
 207                 fprintf(stderr, "Cannot rewrite serial string at %04x\n", AO_USB_DESCRIPTORS);
 208                 exit(1);
 209         }
 210
 211         serial_ucs2_len = image->data[usb_descriptors] - 2;
 212         serial_ucs2 = malloc(serial_ucs2_len);
 213         if (!serial_ucs2) {
 214                 fprintf(stderr, "Malloc(%d) failed\n", serial_ucs2_len);
 215                 exit(1);
 216         }
 217         s = serial;
 218         for (i = serial_ucs2_len / 2; i; i--) {
 219                 serial_ucs2[i * 2 - 1] = 0;
 220                 serial_ucs2[i * 2 - 2] = (s % 10) + '0';
 221                 s /= 10;
 222         }
 223         if (!rewrite(image, usb_descriptors + 2 + image->address, serial_ucs2, serial_ucs2_len))
 224                 usage(argv[0]);
 225
 226         dbg = ccdbg_open(tty);
 227         if (!dbg)
 228                 exit (1);
 229
 230         ccdbg_add_debug(CC_DEBUG_FLASH);
 231
 232         ccdbg_debug_mode(dbg);
 233         ccdbg_halt(dbg);
 234         if (image->address == 0xf000) {
 235                 printf("Loading %d bytes to execute from RAM\n",
 236                        image->length);
 237                 ccdbg_write_hex_image(dbg, image, 0);
 238         } else if (image->address == 0x0000) {
 239                 printf("Loading %d bytes to execute from FLASH\n",
 240                        image->length);
 241                 ccdbg_flash_hex_image(dbg, image);
 242         } else {
 243                 printf("Cannot load code to 0x%04x\n",
 244                        image->address);
 245                 ccdbg_hex_image_free(image);
 246                 ccdbg_close(dbg);
 247                 exit(1);
 248         }
 249         ccdbg_set_pc(dbg, image->address);
 250         ccdbg_resume(dbg);
 251         ccdbg_close(dbg);
 252         exit (0);
 253 }