Align loader to 32-bit boundary
[fw/stlink] / flash / main.c
index 278d63d1878b5100985300867967f9a082d67c80..0ba429c669dbd0f5d4a6c14f0e90149c236b0704 100644 (file)
 /* simple wrapper around the stlink_flash_write function */
 
+// TODO - this should be done as just a simple flag to the st-util command line...
 
+
+#include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
 #include "stlink-common.h"
 
+#define DEBUG_LOG_LEVEL 100
+#define STND_LOG_LEVEL  50
+
+stlink_t *connected_stlink = NULL;
+
+static void cleanup(int signal __attribute__((unused))) {
+    if (connected_stlink) {
+        /* Switch back to mass storage mode before closing. */
+        stlink_run(connected_stlink);
+        stlink_exit_debug_mode(connected_stlink);
+        stlink_close(connected_stlink);
+    }
+
+    exit(1);
+}
 
+enum st_cmds {DO_WRITE = 0, DO_READ = 1, DO_ERASE = 2};
 struct opts
 {
-  unsigned int do_read;
-  const char* devname;
-  const char* filename;
-  stm32_addr_t addr;
-  size_t size;
+    enum st_cmds cmd;
+    const char* devname;
+       char *serial;
+    const char* filename;
+    stm32_addr_t addr;
+    size_t size;
+    int reset;
+    int log_level;
 };
 
 static void usage(void)
 {
-    puts("stlinkv1 command line: ./flash {read|write} /dev/sgX path addr <size>");
-    puts("stlinkv2 command line: ./flash {read|write} path addr <size>");
+    puts("stlinkv1 command line: ./st-flash [--debug] [--reset] [--serial <iSerial>] {read|write} /dev/sgX path addr <size>");
+    puts("stlinkv1 command line: ./st-flash [--debug] /dev/sgX erase");
+    puts("stlinkv2 command line: ./st-flash [--debug] [--reset] [--serial <iSerial>] {read|write} path addr <size>");
+    puts("stlinkv2 command line: ./st-flash [--debug] [--serial <iSerial>] erase");
+    puts("                       use hex format for addr, <iSerial> and <size>");
 }
 
 static int get_opts(struct opts* o, int ac, char** av)
 {
-  /* stlinkv1 command line: ./flash {read|write} /dev/sgX path addr <size> */
-  /* stlinkv2 command line: ./flash {read|write} path addr <size> */
+    /* stlinkv1 command line: ./st-flash {read|write} /dev/sgX path addr <size> */
+    /* stlinkv2 command line: ./st-flash {read|write} path addr <size> */
 
-  unsigned int i = 0;
+    unsigned int i = 0;
 
-  if (ac < 3) return -1;
+    if (ac < 1) return -1;
 
-  /* stlinkv2 */
-  o->devname = NULL;
+    if (strcmp(av[0], "--debug") == 0)
+    {
+        o->log_level = DEBUG_LOG_LEVEL;
+        ac--;
+        av++;
+    }
+    else
+    {
+        o->log_level = STND_LOG_LEVEL;
+    }
 
-  if (strcmp(av[0], "read") == 0)
-  {
-    o->do_read = 1;
+    if (strcmp(av[0], "--reset") == 0)
+    {
+        o->reset = 1;
+        ac--;
+        av++;
+    }
+    else
+    {
+        o->reset = 0;
+    }
 
-    /* stlinkv1 mode */
-    if (ac == 5)
+    if (strcmp(av[0], "--serial") == 0)
+    {
+        ac--;
+        av++;
+           int i=strlen(av[0]);
+           if(i%2 != 0){
+                   puts("no valid hex value, length must be multiple of 2\n");
+                   return -1;
+           }
+           int j=0;
+           while(i>=0 && j<=13){
+                   char buffer[3]={0};
+                   memcpy(buffer,&av[0][i],2);
+                   o->serial[12-j] = (char)strtol((const char*)buffer,NULL, 16);
+                   j++;
+                   i-=2;
+           }
+        ac--;
+        av++;
+    }
+    else
     {
-      o->devname = av[1];
-      i = 1;
+        o->serial = NULL;
     }
 
-    o->size = strtoul(av[i + 3], NULL, 10);
-  }
-  else if (strcmp(av[0], "write") == 0)
-  {
-    o->do_read = 0;
+    if (ac < 1) return -1;
+
+    /* stlinkv2 */
+    o->devname = NULL;
 
-    /* stlinkv1 mode */
-    if (ac == 4)
+    if (strcmp(av[0], "erase") == 0)
     {
-      o->devname = av[1];
-      i = 1;
+        o->cmd = DO_ERASE;
+
+        /* stlinkv1 mode */
+        if (ac == 2)
+        {
+            o->devname = av[1];
+            i = 1;
+        }
     }
-  }
-  else
-  {
-    return -1;
-  }
+    else {
+        if (ac < 3) return -1;
+        if (strcmp(av[0], "read") == 0)
+        {
+            o->cmd = DO_READ;
 
-  o->filename = av[i + 1];
-  o->addr = strtoul(av[i + 2], NULL, 16);
+            /* stlinkv1 mode */
+            if (ac == 5)
+            {
+                o->devname = av[1];
+                i = 1;
+            }
+            if (ac > 3)
+                o->size = strtoul(av[i + 3], NULL, 16);
+        }
+        else if (strcmp(av[0], "write") == 0)
+        {
+            o->cmd = DO_WRITE;
 
-  return 0;
-} 
+            /* stlinkv1 mode */
+            if (ac == 4)
+            {
+                o->devname = av[1];
+                i = 1;
+            }
+        }
+        else
+        {
+            return -1;
+        }
+    }
+
+    o->filename = av[i + 1];
+    o->addr = strtoul(av[i + 2], NULL, 16);
+
+    return 0;
+}
 
 
 int main(int ac, char** av)
 {
-  stlink_t* sl = NULL;
-  struct opts o;
-  int err = -1;
-
-  if (get_opts(&o, ac - 1, av + 1) == -1)
-  {
-    printf("invalid command line\n");
-    usage();
-    goto on_error;
-  }
-
-  if (o.devname != NULL) /* stlinkv1 */
-  {
-#if CONFIG_USE_LIBSG
-    static const int scsi_verbose = 2;
-    sl = stlink_quirk_open(o.devname, scsi_verbose);
-    if (sl == NULL) goto on_error;
-#else
-    printf("not compiled for use with STLink/V1");
-    goto on_error;
-#endif
-  }
-  else /* stlinkv2 */
-  {
-    sl = stlink_open_usb(1);
-    if (sl == NULL) goto on_error;
-  }
-
-  if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE)
-    stlink_exit_dfu_mode(sl);
-
-  if (stlink_current_mode(sl) != STLINK_DEV_DEBUG_MODE)
-    stlink_enter_swd_mode(sl);
-
-  stlink_reset(sl);
-
-  if (o.do_read == 0) /* write */
-  {
-
-    err = stlink_fwrite_flash(sl, o.filename, o.addr);
-    if (err == -1)
+    stlink_t* sl = NULL;
+    struct opts o;
+    char serial_buffer[13] = {0};
+    o.serial = serial_buffer;
+    int err = -1;
+
+    o.size = 0;
+    if (get_opts(&o, ac - 1, av + 1) == -1)
+    {
+        printf("invalid command line\n");
+        usage();
+        return -1;
+    }
+
+    if (o.devname != NULL) /* stlinkv1 */
+        sl = stlink_v1_open(o.log_level, 1);
+    else /* stlinkv2 */
+           sl = stlink_open_usb(o.log_level, 1, o.serial);
+
+    if (sl == NULL)
+        return -1;
+
+    sl->verbose = o.log_level;
+
+    connected_stlink = sl;
+    signal(SIGINT, &cleanup);
+    signal(SIGTERM, &cleanup);
+    signal(SIGSEGV, &cleanup);
+
+    if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE) {
+        if (stlink_exit_dfu_mode(sl)) {
+            printf("Failed to exit DFU mode\n");
+            goto on_error;
+        }
+    }
+
+    if (stlink_current_mode(sl) != STLINK_DEV_DEBUG_MODE) {
+        if (stlink_enter_swd_mode(sl)) {
+            printf("Failed to enter SWD mode\n");
+            goto on_error;
+        }
+    }
+
+    if (o.reset){
+        if (stlink_jtag_reset(sl, 2)) {
+            printf("Failed to reset JTAG\n");
+            goto on_error;
+        }
+
+        if (stlink_reset(sl)) {
+            printf("Failed to reset device\n");
+            goto on_error;
+        }
+    }
+
+    // Disable DMA - Set All DMA CCR Registers to zero. - AKS 1/7/2013
+    if (sl->chip_id == STM32_CHIPID_F4)
+    {
+        memset(sl->q_buf,0,4);
+        for (int i=0;i<8;i++) {
+            stlink_write_mem32(sl,0x40026000+0x10+0x18*i,4);
+            stlink_write_mem32(sl,0x40026400+0x10+0x18*i,4);
+            stlink_write_mem32(sl,0x40026000+0x24+0x18*i,4);
+            stlink_write_mem32(sl,0x40026400+0x24+0x18*i,4);
+        }
+    }
+
+    // Core must be halted to use RAM based flashloaders
+    if (stlink_force_debug(sl)) {
+        printf("Failed to halt the core\n");
+        goto on_error;
+    }
+
+    if (stlink_status(sl)) {
+        printf("Failed to get Core's status\n");
+        goto on_error;
+    }
+
+    if (o.cmd == DO_WRITE) /* write */
+    {
+        if ((o.addr >= sl->flash_base) &&
+                (o.addr < sl->flash_base + sl->flash_size)) {
+            err = stlink_fwrite_flash(sl, o.filename, o.addr);
+            if (err == -1)
+            {
+                printf("stlink_fwrite_flash() == -1\n");
+                goto on_error;
+            }
+        }
+        else if ((o.addr >= sl->sram_base) &&
+                (o.addr < sl->sram_base + sl->sram_size)) {
+            err = stlink_fwrite_sram(sl, o.filename, o.addr);
+            if (err == -1)
+            {
+                printf("stlink_fwrite_sram() == -1\n");
+                goto on_error;
+            }
+        }
+    } else if (o.cmd == DO_ERASE)
     {
-      printf("stlink_fwrite_flash() == -1\n");
-      goto on_error;
-    }
-  }
-  else /* read */
-  {
-    err = stlink_fread(sl, o.filename, o.addr, o.size);
-    if (err == -1)
+        err = stlink_erase_flash_mass(sl);
+        if (err == -1)
+        {
+            printf("stlink_erase_flash_mass() == -1\n");
+            goto on_error;
+        }
+    }
+    else /* read */
     {
-      printf("stlink_fread() == -1\n");
-      goto on_error;
+        if ((o.addr >= sl->flash_base) && (o.size == 0) &&
+                (o.addr < sl->flash_base + sl->flash_size))
+            o.size = sl->flash_size;
+        else if ((o.addr >= sl->sram_base) && (o.size == 0) &&
+                (o.addr < sl->sram_base + sl->sram_size))
+            o.size = sl->sram_size;
+        err = stlink_fread(sl, o.filename, o.addr, o.size);
+        if (err == -1)
+        {
+            printf("stlink_fread() == -1\n");
+            goto on_error;
+        }
+    }
+
+    if (o.reset){
+        stlink_jtag_reset(sl,2);
+        stlink_reset(sl);
     }
-  }
 
-  /* success */
-  err = 0;
+    /* success */
+    err = 0;
 
- on_error:
-  if (sl != NULL) stlink_close(sl);
+on_error:
+    stlink_exit_debug_mode(sl);
+    stlink_close(sl);
 
-  return err;
+    return err;
 }