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