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   /* stlinkv2 */
  53   o->devname = NULL;
  54
  55   if (strcmp(av[0], "erase") == 0)
  56   {
  57     o->cmd = DO_ERASE;
  58
  59     /* stlinkv1 mode */
  60     if (ac == 2)
  61     {
  62       o->devname = av[1];
  63       i = 1;
  64     }
  65   }
  66   else {
  67       if (ac < 3) return -1;
  68       if (strcmp(av[0], "read") == 0)
  69       {
  70           o->cmd = DO_READ;
  71
  72           /* stlinkv1 mode */
  73           if (ac == 5)
  74           {
  75               o->devname = av[1];
  76               i = 1;
  77           }
  78           if (ac > 3)
  79               o->size = strtoul(av[i + 3], NULL, 16);
  80       }
  81       else if (strcmp(av[0], "write") == 0)
  82       {
  83           o->cmd = DO_WRITE;
  84
  85           /* stlinkv1 mode */
  86           if (ac == 4)
  87           {
  88               o->devname = av[1];
  89               i = 1;
  90           }
  91       }
  92       else
  93       {
  94           return -1;
  95       }
  96   }
  97
  98   o->filename = av[i + 1];
  99   o->addr = strtoul(av[i + 2], NULL, 16);
 100
 101   return 0;
 102 }
 103
 104
 105 int main(int ac, char** av)
 106 {
 107   stlink_t* sl = NULL;
 108   struct opts o;
 109   int err = -1;
 110
 111   o.size = 0;
 112   if (get_opts(&o, ac - 1, av + 1) == -1)
 113   {
 114     printf("invalid command line\n");
 115     usage();
 116     goto on_error;
 117   }
 118
 119   if (o.devname != NULL) /* stlinkv1 */
 120   {
 121     sl = stlink_v1_open(50, 1);
 122     if (sl == NULL) goto on_error;
 123     sl->verbose = 50;
 124   }
 125   else /* stlinkv2 */
 126   {
 127     sl = stlink_open_usb(50, 1);
 128     if (sl == NULL) goto on_error;
 129     sl->verbose = 50;
 130   }
 131
 132   if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE)
 133     stlink_exit_dfu_mode(sl);
 134
 135   if (stlink_current_mode(sl) != STLINK_DEV_DEBUG_MODE)
 136     stlink_enter_swd_mode(sl);
 137
 138   if (o.reset)
 139     stlink_reset(sl);
 140
 141 // Disable DMA - Set All DMA CCR Registers to zero. - AKS 1/7/2013
 142   if (sl->chip_id == STM32_CHIPID_F4)
 143   {
 144     memset(sl->q_buf,0,4);
 145         for (int i=0;i<8;i++) {
 146             stlink_write_mem32(sl,0x40026000+0x10+0x18*i,4);
 147             stlink_write_mem32(sl,0x40026400+0x10+0x18*i,4);
 148             stlink_write_mem32(sl,0x40026000+0x24+0x18*i,4);
 149             stlink_write_mem32(sl,0x40026400+0x24+0x18*i,4);
 150         }
 151   }
 152   if (o.cmd == DO_WRITE) /* write */
 153   {
 154     if ((o.addr >= sl->flash_base) &&
 155         (o.addr < sl->flash_base + sl->flash_size)) {
 156         err = stlink_fwrite_flash(sl, o.filename, o.addr);
 157         if (err == -1)
 158         {
 159             printf("stlink_fwrite_flash() == -1\n");
 160             goto on_error;
 161         }
 162     }
 163    else if ((o.addr >= sl->sram_base) &&
 164              (o.addr < sl->sram_base + sl->sram_size)) {
 165         err = stlink_fwrite_sram(sl, o.filename, o.addr);
 166         if (err == -1)
 167         {
 168             printf("stlink_sram_flash() == -1\n");
 169             goto on_error;
 170         }
 171    }
 172   } else if (o.cmd == DO_ERASE)
 173   {
 174      err = stlink_erase_flash_mass(sl);
 175     if (err == -1)
 176     {
 177       printf("stlink_fwrite_flash() == -1\n");
 178       goto on_error;
 179     }
 180   }
 181   else /* read */
 182   {
 183     if ((o.addr >= sl->flash_base) && (o.size == 0) &&
 184         (o.addr < sl->flash_base + sl->flash_size))
 185         o.size = sl->flash_size;
 186     else if ((o.addr >= sl->sram_base) && (o.size == 0) &&
 187              (o.addr < sl->sram_base + sl->sram_size))
 188         o.size = sl->sram_size;
 189     err = stlink_fread(sl, o.filename, o.addr, o.size);
 190     if (err == -1)
 191     {
 192       printf("stlink_fread() == -1\n");
 193       goto on_error;
 194     }
 195   }
 196
 197   if (o.reset)
 198     stlink_reset(sl);
 199
 200   /* success */
 201   err = 0;
 202
 203  on_error:
 204   if (sl != NULL)
 205   {
 206     stlink_exit_debug_mode(sl);
 207     stlink_close(sl);
 208   }
 209
 210   return err;
 211 }