git.gag.com Git - fw/stlink/blob - flash/main.c

   1 /* simple wrapper around the stlink_flash_write function */
   2
   3 // TODO - this should be done as just a simple flag to the st-util command line...
   4
   5
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <sys/types.h>
  10 #include "stlink-common.h"
  11
  12 enum st_cmds {DO_WRITE = 0, DO_READ = 1, DO_ERASE = 2};
  13 struct opts
  14 {
  15     enum st_cmds cmd;
  16     const char* devname;
  17     const char* filename;
  18     stm32_addr_t addr;
  19     size_t size;
  20     int reset;
  21 };
  22
  23 static void usage(void)
  24 {
  25     puts("stlinkv1 command line: ./flash [--reset] {read|write} /dev/sgX path addr <size>");
  26     puts("stlinkv1 command line: ./flash /dev/sgX erase");
  27     puts("stlinkv2 command line: ./flash [--reset] {read|write} path addr <size>");
  28     puts("stlinkv2 command line: ./flash erase");
  29     puts("                       use hex format for addr and <size>");
  30 }
  31
  32 static int get_opts(struct opts* o, int ac, char** av)
  33 {
  34     /* stlinkv1 command line: ./flash {read|write} /dev/sgX path addr <size> */
  35     /* stlinkv2 command line: ./flash {read|write} path addr <size> */
  36
  37     unsigned int i = 0;
  38
  39     if (ac < 1) return -1;
  40
  41     if (strcmp(av[0], "--reset") == 0)
  42     {
  43         o->reset = 1;
  44         ac--;
  45         av++;
  46     }
  47     else
  48     {
  49         o->reset = 0;
  50     }
  51
  52     if (ac < 1) return -1;
  53
  54     /* stlinkv2 */
  55     o->devname = NULL;
  56
  57     if (strcmp(av[0], "erase") == 0)
  58     {
  59         o->cmd = DO_ERASE;
  60
  61         /* stlinkv1 mode */
  62         if (ac == 2)
  63         {
  64             o->devname = av[1];
  65             i = 1;
  66         }
  67     }
  68     else {
  69         if (ac < 3) return -1;
  70         if (strcmp(av[0], "read") == 0)
  71         {
  72             o->cmd = DO_READ;
  73
  74             /* stlinkv1 mode */
  75             if (ac == 5)
  76             {
  77                 o->devname = av[1];
  78                 i = 1;
  79             }
  80             if (ac > 3)
  81                 o->size = strtoul(av[i + 3], NULL, 16);
  82         }
  83         else if (strcmp(av[0], "write") == 0)
  84         {
  85             o->cmd = DO_WRITE;
  86
  87             /* stlinkv1 mode */
  88             if (ac == 4)
  89             {
  90                 o->devname = av[1];
  91                 i = 1;
  92             }
  93         }
  94         else
  95         {
  96             return -1;
  97         }
  98     }
  99
 100     o->filename = av[i + 1];
 101     o->addr = strtoul(av[i + 2], NULL, 16);
 102
 103     return 0;
 104 }
 105
 106
 107 int main(int ac, char** av)
 108 {
 109     stlink_t* sl = NULL;
 110     struct opts o;
 111     int err = -1;
 112
 113     o.size = 0;
 114     if (get_opts(&o, ac - 1, av + 1) == -1)
 115     {
 116         printf("invalid command line\n");
 117         usage();
 118         goto on_error;
 119     }
 120
 121     if (o.devname != NULL) /* stlinkv1 */
 122     {
 123         sl = stlink_v1_open(50, 1);
 124         if (sl == NULL) goto on_error;
 125         sl->verbose = 50;
 126     }
 127     else /* stlinkv2 */
 128     {
 129         sl = stlink_open_usb(50, 1);
 130         if (sl == NULL) goto on_error;
 131         sl->verbose = 50;
 132     }
 133
 134     if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE)
 135         stlink_exit_dfu_mode(sl);
 136
 137     if (stlink_current_mode(sl) != STLINK_DEV_DEBUG_MODE)
 138         stlink_enter_swd_mode(sl);
 139
 140     if (o.reset)
 141         stlink_reset(sl);
 142
 143     // Disable DMA - Set All DMA CCR Registers to zero. - AKS 1/7/2013
 144     if (sl->chip_id == STM32_CHIPID_F4)
 145     {
 146         memset(sl->q_buf,0,4);
 147         for (int i=0;i<8;i++) {
 148             stlink_write_mem32(sl,0x40026000+0x10+0x18*i,4);
 149             stlink_write_mem32(sl,0x40026400+0x10+0x18*i,4);
 150             stlink_write_mem32(sl,0x40026000+0x24+0x18*i,4);
 151             stlink_write_mem32(sl,0x40026400+0x24+0x18*i,4);
 152         }
 153     }
 154     if (o.cmd == DO_WRITE) /* write */
 155     {
 156         if ((o.addr >= sl->flash_base) &&
 157                 (o.addr < sl->flash_base + sl->flash_size)) {
 158             err = stlink_fwrite_flash(sl, o.filename, o.addr);
 159             if (err == -1)
 160             {
 161                 printf("stlink_fwrite_flash() == -1\n");
 162                 goto on_error;
 163             }
 164         }
 165         else if ((o.addr >= sl->sram_base) &&
 166                 (o.addr < sl->sram_base + sl->sram_size)) {
 167             err = stlink_fwrite_sram(sl, o.filename, o.addr);
 168             if (err == -1)
 169             {
 170                 printf("stlink_sram_flash() == -1\n");
 171                 goto on_error;
 172             }
 173         }
 174     } else if (o.cmd == DO_ERASE)
 175     {
 176         err = stlink_erase_flash_mass(sl);
 177         if (err == -1)
 178         {
 179             printf("stlink_fwrite_flash() == -1\n");
 180             goto on_error;
 181         }
 182     }
 183     else /* read */
 184     {
 185         if ((o.addr >= sl->flash_base) && (o.size == 0) &&
 186                 (o.addr < sl->flash_base + sl->flash_size))
 187             o.size = sl->flash_size;
 188         else if ((o.addr >= sl->sram_base) && (o.size == 0) &&
 189                 (o.addr < sl->sram_base + sl->sram_size))
 190             o.size = sl->sram_size;
 191         err = stlink_fread(sl, o.filename, o.addr, o.size);
 192         if (err == -1)
 193         {
 194             printf("stlink_fread() == -1\n");
 195             goto on_error;
 196         }
 197     }
 198
 199     if (o.reset)
 200         stlink_reset(sl);
 201
 202     /* success */
 203     err = 0;
 204
 205 on_error:
 206     if (sl != NULL)
 207     {
 208         stlink_exit_debug_mode(sl);
 209         stlink_close(sl);
 210     }
 211
 212     return err;
 213 }