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_jtag_reset(sl,2);
 142         stlink_reset(sl);
 143     }
 144
 145     // Disable DMA - Set All DMA CCR Registers to zero. - AKS 1/7/2013
 146     if (sl->chip_id == STM32_CHIPID_F4)
 147     {
 148         memset(sl->q_buf,0,4);
 149         for (int i=0;i<8;i++) {
 150             stlink_write_mem32(sl,0x40026000+0x10+0x18*i,4);
 151             stlink_write_mem32(sl,0x40026400+0x10+0x18*i,4);
 152             stlink_write_mem32(sl,0x40026000+0x24+0x18*i,4);
 153             stlink_write_mem32(sl,0x40026400+0x24+0x18*i,4);
 154         }
 155     }
 156     if (o.cmd == DO_WRITE) /* write */
 157     {
 158         if ((o.addr >= sl->flash_base) &&
 159                 (o.addr < sl->flash_base + sl->flash_size)) {
 160             err = stlink_fwrite_flash(sl, o.filename, o.addr);
 161             if (err == -1)
 162             {
 163                 printf("stlink_fwrite_flash() == -1\n");
 164                 goto on_error;
 165             }
 166         }
 167         else if ((o.addr >= sl->sram_base) &&
 168                 (o.addr < sl->sram_base + sl->sram_size)) {
 169             err = stlink_fwrite_sram(sl, o.filename, o.addr);
 170             if (err == -1)
 171             {
 172                 printf("stlink_sram_flash() == -1\n");
 173                 goto on_error;
 174             }
 175         }
 176     } else if (o.cmd == DO_ERASE)
 177     {
 178         err = stlink_erase_flash_mass(sl);
 179         if (err == -1)
 180         {
 181             printf("stlink_fwrite_flash() == -1\n");
 182             goto on_error;
 183         }
 184     }
 185     else /* read */
 186     {
 187         if ((o.addr >= sl->flash_base) && (o.size == 0) &&
 188                 (o.addr < sl->flash_base + sl->flash_size))
 189             o.size = sl->flash_size;
 190         else if ((o.addr >= sl->sram_base) && (o.size == 0) &&
 191                 (o.addr < sl->sram_base + sl->sram_size))
 192             o.size = sl->sram_size;
 193         err = stlink_fread(sl, o.filename, o.addr, o.size);
 194         if (err == -1)
 195         {
 196             printf("stlink_fread() == -1\n");
 197             goto on_error;
 198         }
 199     }
 200
 201     if (o.reset){
 202         stlink_jtag_reset(sl,2);
 203         stlink_reset(sl);
 204     }
 205
 206     /* success */
 207     err = 0;
 208
 209 on_error:
 210     if (sl != NULL)
 211     {
 212         stlink_exit_debug_mode(sl);
 213         stlink_close(sl);
 214     }
 215
 216     return err;
 217 }