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 }