Add serial ID filter

Author: tomas-pecserke

main.cpp

@@ -25,6 +25,7 @@
 #include <memory>
 #include <functional>
 #include <ctime>
+#include <sstream>
 
 #include "boot/uf2.h"
 #include "picoboot_connection_cxx.h"
@@ -255,6 +256,7 @@ struct _settings {
     uint32_t binary_start = FLASH_START;
     int bus=-1;
     int address=-1;
+    string serial;
     uint32_t offset = 0;
     uint32_t from = 0;
     uint32_t to = 0;
@@ -292,17 +294,31 @@ struct _settings {
 _settings settings;
 std::shared_ptr<cmd> selected_cmd;
 
-auto device_selection =
+auto device_selection = (
     (
         (option("--bus") & integer("bus").min_value(0).max_value(255).set(settings.bus)
-            .if_missing([] { return "missing bus number"; })) % "Filter devices by USB bus number" +
-        (option("--address") & integer("addr").min_value(1).max_value(127).set(settings.address)
-            .if_missing([] { return "missing address"; })) % "Filter devices by USB device address"
+            .if_missing([] { return "missing bus number"; }))
+            % "Filter devices by USB bus number"
+        + (option("--address") & integer("addr").min_value(1).max_value(127).set(settings.address)
+            .if_missing([] { return "missing address"; }))
+            % "Filter devices by USB device address"
+    ).min(0).doc_non_optional(true) +
+    (
+        (option("--serial") & value("serial").set(settings.serial)
+            .if_missing([] { return "missing serial id"; }))
+            % "Filter devices by serial id"
 #if !defined(_WIN32)
-        + option('f', "--force").set(settings.force) % "Force a device not in BOOTSEL mode but running compatible code to reset so the command can be executed. After executing the command (unless the command itself is a 'reboot') the device will be rebooted back to application mode" +
-                option('F', "--force-no-reboot").set(settings.force_no_reboot) % "Force a device not in BOOTSEL mode but running compatible code to reset so the command can be executed. After executing the command (unless the command itself is a 'reboot') the device will be left connected and accessible to picotool, but without the RPI-RP2 drive mounted"
+        + option('f', "--force").set(settings.force)
+            % "Force a device not in BOOTSEL mode but running compatible code to reset so the command can be executed. "
+              "After executing the command (unless the command itself is a 'reboot') "
+              "the device will be rebooted back to application mode"
+        + option('F', "--force-no-reboot").set(settings.force_no_reboot)
+            % "Force a device not in BOOTSEL mode but running compatible code to reset so the command can be executed. "
+              "After executing the command (unless the command itself is a 'reboot') the device will be left connected "
+              "and accessible to picotool, but without the RPI-RP2 drive mounted"
 #endif
-    ).min(0).doc_non_optional(true);
+    ).min(0).doc_non_optional(true)
+).min(0).doc_non_optional(true);
 
 auto file_types = (option ('t', "--type") & value("type").set(settings.file_type))
             % "Specify file type (uf2 | elf | bin) explicitly, ignoring file extension";
@@ -1581,28 +1597,25 @@ void info_guts(memory_access &raw_access) {
 }
 
 string missing_device_string(bool wasRetry) {
-    char b[256];
-    if (wasRetry) {
-        strcpy(b, "Despite the reboot attempt, no ");
-    } else {
-        strcpy(b, "No ");
+    std::ostringstream oss;
+    oss << (wasRetry ? "Despite the reboot attempt, no" : "No") << " accessible RP2040 device";
+    if (settings.bus != -1 && settings.serial.empty()) {
+        oss << 's';
+    }
+    oss << " in BOOTSEL mode";
+    if (!settings.serial.empty()) {
+        oss << " with serial ID '" << settings.serial << '\'';
+    }
+    oss << " was found";
+    if (settings.bus != -1) {
+        oss << " at bus " << settings.bus;
     }
-    char *buf = b + strlen(b);
-    int buf_len = b + sizeof(b) - buf;
     if (settings.address != -1) {
-        if (settings.bus != -1) {
-            snprintf(buf, buf_len, "accessible RP2040 device in BOOTSEL mode was found at bus %d, address %d.", settings.bus, settings.address);
-        } else {
-            snprintf(buf, buf_len, "accessible RP2040 devices in BOOTSEL mode were found with address %d.", settings.address);
-        }
-    } else {
-        if (settings.bus != -1) {
-            snprintf(buf, buf_len, "accessible RP2040 devices in BOOTSEL mode were found found on bus %d.", settings.bus);
-        } else {
-            snprintf(buf, buf_len,"accessible RP2040 devices in BOOTSEL mode were found.");
-        }
+        oss << (settings.bus == -1 ? " with" : ",") << " address " << settings.address;
     }
-    return b;
+    oss << '.';
+
+    return oss.str();
 }
 
 bool help_command::execute(device_map &devices) {
@@ -2179,6 +2192,20 @@ void cancelled(int) {
     throw cancelled_exception();
 }
 
+string get_usb_device_serial(libusb_device *device, libusb_device_handle *handle) {
+    struct libusb_device_descriptor desc{};
+    int ret = libusb_get_device_descriptor(device, &desc);
+    if (!ret) {
+        unsigned char data[33];
+        if (libusb_get_string_descriptor_ascii(handle, desc.iSerialNumber, data, 31)) {
+            data[32] = '\0';
+            return reinterpret_cast<char const*>(data);
+        }
+    }
+
+    return "";
+}
+
 int main(int argc, char **argv) {
     libusb_context *ctx = nullptr;
 
@@ -2229,6 +2256,7 @@ int main(int argc, char **argv) {
                     if (handle) {
                         to_close.push_back(handle);
                     }
+                    if (!settings.serial.empty() && (!handle || settings.serial != get_usb_device_serial(*dev, handle))) continue;
                     if (result != dr_error) {
                         devices[result].push_back(std::make_pair(*dev, handle));
                     }