Handle Stlink V21.

Author: UweBonnes

src/main.c

@@ -28,6 +28,7 @@
 
 #define STLINK_VID 0x0483
 #define STLINK_PID 0x3748
+#define STLINK_PIDV21 0x374b
 
 void print_help(char *argv[]) {
   printf("Usage: %s [options] [firmware.bin]\n", argv[0]);
@@ -37,6 +38,63 @@ void print_help(char *argv[]) {
   printf("\tApplication is started when called without argument or after firmware load\n\n");
 }
 
+#include <string.h>
+#define USB_TIMEOUT 5000
+int stlink_dfu_mode(libusb_device_handle *dev_handle, int trigger) {
+    unsigned char data[16];
+    int rw_bytes, res;
+
+    memset(data, 0, sizeof(data));
+
+    data[0] = 0xF9;
+    if (trigger) data[1] = 0x10;
+    /* Write */
+    res = libusb_bulk_transfer(dev_handle,
+                               1 | LIBUSB_ENDPOINT_OUT,
+                               data,
+                               sizeof(data),
+                               &rw_bytes,
+                               USB_TIMEOUT);
+    if (res) {
+        fprintf(stderr, "USB transfer failure\n");
+        return -1;
+    }
+    if (!trigger) {
+        /* Read */
+        libusb_bulk_transfer(dev_handle,
+                             1 | LIBUSB_ENDPOINT_IN,data,
+                             2,
+                             &rw_bytes,
+                             USB_TIMEOUT);
+        if (res) {
+            fprintf(stderr, "stlink_read_infos() failure\n");
+            return -1;
+        }
+    }
+    return data[0] << 8 | data[1];
+}
+
+int test_v21(libusb_context *usb_ctx)
+{
+    libusb_device_handle *dev_handle;
+    int res = 0;
+    dev_handle = libusb_open_device_with_vid_pid(usb_ctx,
+                                                 STLINK_VID,
+                                                 STLINK_PIDV21);
+    if (dev_handle) {
+        fprintf(stderr, "StlinkV21 found\n");
+        res = stlink_dfu_mode(dev_handle, 0);
+        if (res != 0x8000) {
+            libusb_release_interface(dev_handle, 0);
+            return 0;
+        }
+        stlink_dfu_mode(dev_handle, 1);
+        libusb_release_interface(dev_handle, 0);
+    }
+    usleep(1000000);
+    return 1;
+}
+
 int main(int argc, char *argv[]) {
   libusb_context *usb_ctx;
   libusb_device_handle *dev_handle;
@@ -63,7 +121,8 @@ int main(int argc, char *argv[]) {
 
   res = libusb_init(&usb_ctx);
   (void)res;
-
+  /* Test for V21 device and sitch to DFU mode */
+  test_v21(usb_ctx);
   dev_handle = libusb_open_device_with_vid_pid(usb_ctx,
 					       STLINK_VID,
 					       STLINK_PID);
@@ -103,7 +162,7 @@ int main(int argc, char *argv[]) {
   }
   printf("Current mode : %d\n", res);
 
-  if (res != 1) {
+  if (res & 0xfffc) {
     printf("ST-Link dongle is not in the correct mode. Please unplug and plug the dongle again.\n");
     libusb_release_interface(dev_handle, 0);
     return EXIT_SUCCESS;