st-info: Add support for OpenOCD hla_serial printing

[fw/stlink] / src / st-info.c

diff --git a/src/st-info.c b/src/st-info.c
index 6fe2f8d7a5ab70ab56597b77dd0291a3da548d79..cf61bccbf2c1b57ea0aff4c6e537397824b5176e 100644 (file)
--- a/src/st-info.c
+++ b/src/st-info.c
@@ -1,13 +1,9 @@
-/* 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 <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
-#include "stlink-common.h"
+
+#include <stlink-common.h>
 
 static void usage(void)
 {
@@ -16,11 +12,75 @@ static void usage(void)
     puts("st-info --descr");
     puts("st-info --pagesize");
     puts("st-info --chipid");
+    puts("st-info --serial");
+    puts("st-info --hla-serial");
+    puts("st-info --probe");
+}
+
+/* Print normal or OpenOCD hla_serial with newline */
+static void stlink_print_serial(stlink_t *sl, bool openocd)
+{
+    const char *fmt;
+
+    if (openocd) {
+       printf("\"");
+       fmt = "\\x%02x";
+    } else {
+       fmt = "%02x";
+    }
+
+    for (int n = 0; n < sl->serial_size; n++)
+        printf(fmt, sl->serial[n]);
+
+    if (openocd)
+       printf("\"");
+    printf("\n");
 }
 
-static int print_data(stlink_t* sl, char** av)
+static void stlink_print_info(stlink_t *sl)
+{
+    const chip_params_t *params = NULL;
+
+    if (!sl)
+        return;
+
+    printf(" serial: ");
+    stlink_print_serial(sl, false);
+    printf("openocd: ");
+    stlink_print_serial(sl, true);
+
+    printf("  flash: %zu (pagesize: %zu)\n", sl->flash_size, sl->flash_pgsz);
+    printf("   sram: %zu\n",       sl->sram_size);
+    printf(" chipid: 0x%.4x\n",    sl->chip_id);
+
+    for (size_t i = 0; i < sizeof(devices) / sizeof(devices[0]); i++) {
+        if (devices[i].chip_id == sl->chip_id) {
+            params = &devices[i];
+            break;
+        }
+    }
+
+    if (params)
+        printf(" descr: %s\n", params->description);
+}
+
+static void stlink_probe(void)
+{
+    stlink_t **stdevs;
+    size_t size;
+
+    size = stlink_probe_usb(&stdevs);
+
+    printf("Found %zu stlink programmers\n", size);
+
+    for (size_t n = 0; n < size; n++)
+        stlink_print_info(stdevs[n]);
+
+    stlink_probe_usb_free(&stdevs, size);
+}
+
+static int print_data(stlink_t *sl, char **av)
 {
-    int ret = 0;
     if (strcmp(av[1], "--flash") == 0)
         printf("0x%zx\n", sl->flash_size);
     else if (strcmp(av[1], "--sram") == 0)
@@ -29,7 +89,13 @@ static int print_data(stlink_t* sl, char** av)
         printf("0x%zx\n", sl->flash_pgsz);
     else if (strcmp(av[1], "--chipid") == 0)
         printf("0x%.4x\n", sl->chip_id);
-    else if (strcmp(av[1], "--descr")==0) {
+    else if (strcmp(av[1], "--probe") == 0)
+        stlink_probe();
+    else if (strcmp(av[1], "--serial") == 0)
+        stlink_print_serial(sl, false);
+    else if (strcmp(av[1], "--hla-serial") == 0)
+        stlink_print_serial(sl, true);
+    else if (strcmp(av[1], "--descr") == 0) {
         const chip_params_t *params = NULL;
         for (size_t i = 0; i < sizeof(devices) / sizeof(devices[0]); i++) {
             if(devices[i].chip_id == sl->chip_id) {
@@ -42,7 +108,8 @@ static int print_data(stlink_t* sl, char** av)
         }
         printf("%s\n", params->description);
     }
-    return ret;
+
+    return 0;
 }
 
 
@@ -51,7 +118,7 @@ stlink_t* open_sl(void)
     stlink_t* sl;
     sl = stlink_v1_open(0, 1);
     if (sl == NULL)
-        sl = stlink_open_usb(0, 1);
+        sl = stlink_open_usb(0, 1, NULL);
     return sl;
 }
 
@@ -70,7 +137,9 @@ int main(int ac, char** av)
     if (sl == NULL) {
         return -1;
     }
-    sl->verbose=0;
+
+    sl->verbose = 0;
+
     if (stlink_current_mode(sl) == STLINK_DEV_DFU_MODE)
         stlink_exit_dfu_mode(sl);