ao-tools: Create general elf and hex library routines
[fw/altos] / ao-tools / ao-load / ao-load.c
index 642af2fed7ada02beccf08c2fe4e1fe35c9b1e3a..c1f551499ceac969b76072f417bb3f66f49e3e31 100644 (file)
 #include <unistd.h>
 #include <getopt.h>
 #include "ccdbg.h"
+#include "cc.h"
 
 #define AO_USB_DESC_STRING             3
 
 struct sym {
        unsigned        addr;
        char            *name;
-} serial_symbols[] = {
-       { 0,    "_ao_serial_number" },
-#define AO_SERIAL_NUMBER       (serial_symbols[0].addr)
-       { 0,    "_ao_usb_descriptors" },
-#define AO_USB_DESCRIPTORS     (serial_symbols[1].addr)
+       int             required;
+} ao_symbols[] = {
+       { 0,    "_ao_serial_number", 1 },
+#define AO_SERIAL_NUMBER       (ao_symbols[0].addr)
+       { 0,    "_ao_usb_descriptors", 0 },
+#define AO_USB_DESCRIPTORS     (ao_symbols[1].addr)
+       { 0,    "_ao_radio_cal", 1 },
+#define AO_RADIO_CAL           (ao_symbols[2].addr)
 };
 
-#define NUM_SERIAL_SYMBOLS     (sizeof(serial_symbols)/sizeof(serial_symbols[0]))
+#define NUM_SYMBOLS            3
+#define NUM_REQUIRED_SYMBOLS   2
 
 static int
 find_symbols(FILE *map)
@@ -47,7 +52,7 @@ find_symbols(FILE *map)
        char    *colon;
        unsigned long   a;
        int     s;
-       int     found = 0;
+       int     required = 0;
 
        while (fgets(line, sizeof(line), map) != NULL) {
                line[sizeof(line)-1] = '\0';
@@ -63,18 +68,19 @@ find_symbols(FILE *map)
                a = strtoul(colon+1, &addr_end, 16);
                if (a == ULONG_MAX || addr_end == addr)
                        continue;
-               for (s = 0; s < NUM_SERIAL_SYMBOLS; s++)
-                       if (!strcmp(serial_symbols[s].name, name)) {
-                               serial_symbols[s].addr = (unsigned) a;
-                               ++found;
+               for (s = 0; s < NUM_SYMBOLS; s++)
+                       if (!strcmp(ao_symbols[s].name, name)) {
+                               ao_symbols[s].addr = (unsigned) a;
+                               if (ao_symbols[s].required)
+                                       ++required;
                                break;
                        }
        }
-       return found == NUM_SERIAL_SYMBOLS;
+       return required >= NUM_REQUIRED_SYMBOLS;
 }
 
 static int
-rewrite(struct hex_image *image, unsigned addr, char *data, int len)
+rewrite(struct ao_hex_image *image, unsigned addr, char *data, int len)
 {
        int i;
        if (addr < image->address || image->address + image->length < addr + len)
@@ -89,29 +95,27 @@ rewrite(struct hex_image *image, unsigned addr, char *data, int len)
        memcpy(image->data + addr - image->address, data, len);
 }
 
-<<<<<<< HEAD:ao-tools/ao-load/ao-load.c
-static const struct option
-=======
 static const struct option options[] = {
        { .name = "tty", .has_arg = 1, .val = 'T' },
+       { .name = "device", .has_arg = 1, .val = 'D' },
+       { .name = "cal", .has_arg = 1, .val = 'c' },
        { 0, 0, 0, 0},
 };
 
 static void usage(char *program)
 {
-       fprintf(stderr, "usage: %s [--tty <tty-name>] file.ihx serial-number\n", program);
+       fprintf(stderr, "usage: %s [--tty=<tty-name>] [--device=<device-name>] [--cal=<radio-cal>] file.ihx serial-number\n", program);
        exit(1);
 }
 
->>>>>>> c29275b72438637d46d7a50742882d2736eb176a:ao-tools/ao-load/ao-load.c
 int
 main (int argc, char **argv)
 {
        struct ccdbg    *dbg;
        uint8_t         status;
        uint16_t        pc;
-       struct hex_file *hex;
-       struct hex_image *image;
+       struct ao_hex_file      *hex;
+       struct ao_hex_image *image;
        char            *filename;
        FILE            *file;
        FILE            *map;
@@ -123,16 +127,27 @@ main (int argc, char **argv)
        char            serial_int[2];
        unsigned int    s;
        int             i;
-       unsigned        usb_descriptors;
        int             string_num;
        char            *tty = NULL;
+       char            *device = NULL;
+       uint32_t        cal = 0;
+       char            cal_int[4];
+       char            *cal_end;
        int             c;
 
-       while ((c = getopt_long(argc, argv, "T:", options, NULL)) != -1) {
+       while ((c = getopt_long(argc, argv, "T:D:c:", options, NULL)) != -1) {
                switch (c) {
                case 'T':
                        tty = optarg;
                        break;
+               case 'D':
+                       device = optarg;
+                       break;
+               case 'c':
+                       cal = strtoul(optarg, &cal_end, 10);
+                       if (cal_end == optarg || *cal_end != '\0')
+                               usage(argv[0]);
+                       break;
                default:
                        usage(argv[0]);
                        break;
@@ -167,22 +182,22 @@ main (int argc, char **argv)
        }
        fclose(map);
 
-       hex = ccdbg_hex_file_read(file, filename);
+       hex = ao_hex_file_read(file, filename);
        fclose(file);
        if (!hex) {
                perror(filename);
                exit (1);
        }
-       image = ccdbg_hex_image_create(hex);
+       image = ao_hex_image_create(hex);
        if (!image) {
                fprintf(stderr, "image create failed\n");
                exit (1);
        }
-       ccdbg_hex_file_free(hex);
+       ao_hex_file_free(hex);
 
        serial = strtoul(serial_string, NULL, 0);
        if (!serial)
-(argv[0]);
+               usage(argv[0]);
 
        serial_int[0] = serial & 0xff;
        serial_int[1] = (serial >> 8) & 0xff;
@@ -193,36 +208,55 @@ main (int argc, char **argv)
                exit(1);
        }
 
-       usb_descriptors = AO_USB_DESCRIPTORS - image->address;
-       string_num = 0;
-       while (image->data[usb_descriptors] != 0 && usb_descriptors < image->length) {
-               if (image->data[usb_descriptors+1] == AO_USB_DESC_STRING) {
-                       ++string_num;
-                       if (string_num == 4)
-                               break;
+       if (AO_USB_DESCRIPTORS) {
+               unsigned        usb_descriptors;
+               usb_descriptors = AO_USB_DESCRIPTORS - image->address;
+               string_num = 0;
+               while (image->data[usb_descriptors] != 0 && usb_descriptors < image->length) {
+                       if (image->data[usb_descriptors+1] == AO_USB_DESC_STRING) {
+                               ++string_num;
+                               if (string_num == 4)
+                                       break;
+                       }
+                       usb_descriptors += image->data[usb_descriptors];
+               }
+               if (usb_descriptors >= image->length || image->data[usb_descriptors] == 0 ) {
+                       fprintf(stderr, "Cannot rewrite serial string at %04x\n", AO_USB_DESCRIPTORS);
+                       exit(1);
                }
-               usb_descriptors += image->data[usb_descriptors];
-       }
-       if (usb_descriptors >= image->length || image->data[usb_descriptors] == 0 ) {
-               fprintf(stderr, "Cannot rewrite serial string at %04x\n", AO_USB_DESCRIPTORS);
-               exit(1);
-       }
 
-       serial_ucs2_len = image->data[usb_descriptors] - 2;
-       serial_ucs2 = malloc(serial_ucs2_len);
-       if (!serial_ucs2) {
-               fprintf(stderr, "Malloc(%d) failed\n", serial_ucs2_len);
-               exit(1);
-       }
-       s = serial;
-       for (i = serial_ucs2_len / 2; i; i--) {
-               serial_ucs2[i * 2 - 1] = 0;
-               serial_ucs2[i * 2 - 2] = (s % 10) + '0';
-               s /= 10;
+               serial_ucs2_len = image->data[usb_descriptors] - 2;
+               serial_ucs2 = malloc(serial_ucs2_len);
+               if (!serial_ucs2) {
+                       fprintf(stderr, "Malloc(%d) failed\n", serial_ucs2_len);
+                       exit(1);
+               }
+               s = serial;
+               for (i = serial_ucs2_len / 2; i; i--) {
+                       serial_ucs2[i * 2 - 1] = 0;
+                       serial_ucs2[i * 2 - 2] = (s % 10) + '0';
+                       s /= 10;
+               }
+               if (!rewrite(image, usb_descriptors + 2 + image->address, serial_ucs2, serial_ucs2_len))
+                       usage(argv[0]);
        }
-       if (!rewrite(image, usb_descriptors + 2 + image->address, serial_ucs2, serial_ucs2_len))
-               usage(argv[0]);
 
+       if (cal) {
+               cal_int[0] = cal & 0xff;
+               cal_int[1] = (cal >> 8) & 0xff;
+               cal_int[2] = (cal >> 16) & 0xff;
+               cal_int[3] = (cal >> 24) & 0xff;
+               if (!AO_RADIO_CAL) {
+                       fprintf(stderr, "Cannot find radio calibration location in image\n");
+                       exit(1);
+               }
+               if (!rewrite(image, AO_RADIO_CAL, cal_int, sizeof (cal_int))) {
+                       fprintf(stderr, "Cannot rewrite radio calibration at %04x\n", AO_RADIO_CAL);
+                       exit(1);
+               }
+       }
+       if (!tty)
+               tty = cc_usbdevs_find_by_arg(device, "TIDongle");
        dbg = ccdbg_open(tty);
        if (!dbg)
                exit (1);
@@ -242,7 +276,7 @@ main (int argc, char **argv)
        } else {
                printf("Cannot load code to 0x%04x\n",
                       image->address);
-               ccdbg_hex_image_free(image);
+               ao_hex_image_free(image);
                ccdbg_close(dbg);
                exit(1);
        }