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);
 110     if (sl == NULL) goto on_error;
 111     sl->verbose = 50;
 112   }
 113   else /* stlinkv2 */
 114   {
 115     sl = stlink_open_usb(50);
 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   if (o.cmd == DO_WRITE) /* write */
 127   {
 128     if ((o.addr >= sl->flash_base) &&
 129         (o.addr < sl->flash_base + sl->flash_size))
 130         err = stlink_fwrite_flash(sl, o.filename, o.addr);
 131     else if ((o.addr >= sl->sram_base) &&
 132              (o.addr < sl->sram_base + sl->sram_size))
 133         err = stlink_fwrite_sram(sl, o.filename, o.addr);
 134     if (err == -1)
 135     {
 136       printf("stlink_fwrite_flash() == -1\n");
 137       goto on_error;
 138     }
 139   }
 140   else if (o.cmd == DO_ERASE)
 141   {
 142      err = stlink_erase_flash_mass(sl);
 143     if (err == -1)
 144     {
 145       printf("stlink_fwrite_flash() == -1\n");
 146       goto on_error;
 147     }
 148   }
 149   else /* read */
 150   {
 151     err = stlink_fread(sl, o.filename, o.addr, o.size);
 152     if (err == -1)
 153     {
 154       printf("stlink_fread() == -1\n");
 155       goto on_error;
 156     }
 157   }
 158
 159   /* success */
 160   err = 0;
 161
 162  on_error:
 163   if (sl != NULL)
 164   {
 165     stlink_exit_debug_mode(sl);
 166     stlink_close(sl);
 167   }
 168
 169   return err;
 170 }