/* 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
+{
+ 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: ./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: ./st-flash {read|write} /dev/sgX path addr <size> */
+ /* stlinkv2 command line: ./st-flash {read|write} path addr <size> */
+
+ unsigned int i = 0;
+
+ if (ac < 1) return -1;
+
+ 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], "--reset") == 0)
+ {
+ o->reset = 1;
+ ac--;
+ av++;
+ }
+ else
+ {
+ o->reset = 0;
+ }
+
+ 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->serial = NULL;
+ }
+
+ if (ac < 1) return -1;
+
+ /* stlinkv2 */
+ o->devname = NULL;
+
+ if (strcmp(av[0], "erase") == 0)
+ {
+ o->cmd = DO_ERASE;
+
+ /* stlinkv1 mode */
+ if (ac == 2)
+ {
+ o->devname = av[1];
+ i = 1;
+ }
+ }
+ else {
+ if (ac < 3) return -1;
+ if (strcmp(av[0], "read") == 0)
+ {
+ o->cmd = DO_READ;
+
+ /* 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;
+
+ /* 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)
{
- /* stlinkv1 command line: ./flash /dev/sgX path addr */
- /* stlinkv2 command line: ./flash path addr */
-
- stlink_t* sl = NULL;
- stm32_addr_t addr;
- const char* path;
- int err = -1;
-
- if (ac == 4) /* stlinkv1 */
- {
- static const int scsi_verbose = 2;
- sl = stlink_quirk_open(av[1], scsi_verbose);
- path = av[2];
- addr = strtoul(av[3], NULL, 16);
- }
- else if (ac == 3) /* stlinkv2 */
- {
- sl = stlink_open_usb(NULL, 10);
- path = av[1];
- addr = strtoul(av[2], NULL, 16);
- }
- else /* invalid */
- {
- printf("invalid command line\n");
- goto on_error;
- }
-
- if (sl == NULL) goto on_error;
-
- if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE)
- stlink_exit_dfu_mode(sl);
- stlink_enter_swd_mode(sl);
- stlink_reset(sl);
-
- err = stlink_fwrite_flash(sl, path, addr);
- if (err == -1)
- {
- printf("stlink_fwrite_flash() == -1\n");
- goto on_error;
- }
-
- /* success */
- err = 0;
-
- on_error:
- if (sl != NULL) stlink_close(sl);
-
- return err;
+ 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)
+ {
+ err = stlink_erase_flash_mass(sl);
+ if (err == -1)
+ {
+ printf("stlink_erase_flash_mass() == -1\n");
+ goto on_error;
+ }
+ }
+ else /* read */
+ {
+ 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;
+
+on_error:
+ stlink_exit_debug_mode(sl);
+ stlink_close(sl);
+
+ return err;
}