From 613c8fec1a25b132acf49bde3d17cb8355aa4b54 Mon Sep 17 00:00:00 2001
From: Oystein Tveit <oystein.tveit@nordicsemi.no>
Date: Wed, 28 Jun 2023 13:12:32 +0200
Subject: [PATCH] Add support for editing device strings

---
 src/device.rs | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/src/device.rs b/src/device.rs
index 4bacc3f..9ed6de0 100644
--- a/src/device.rs
+++ b/src/device.rs
@@ -85,6 +85,30 @@ impl UsbDevice {
         res
     }
 
+    /// Returns the old value, if present.
+    pub fn set_configuration_name(&mut self, name: &str) -> Option<String> {
+        self.string_pool
+            .insert(self.string_configuration, name.to_string())
+    }
+
+    /// Returns the old value, if present.
+    pub fn set_serial_number(&mut self, name: &str) -> Option<String> {
+        self.string_pool
+            .insert(self.string_serial, name.to_string())
+    }
+
+    /// Returns the old value, if present.
+    pub fn set_product_name(&mut self, name: &str) -> Option<String> {
+        self.string_pool
+            .insert(self.string_product, name.to_string())
+    }
+
+    /// Returns the old value, if present.
+    pub fn set_manufacturer_name(&mut self, name: &str) -> Option<String> {
+        self.string_pool
+            .insert(self.string_manufacturer, name.to_string())
+    }
+
     pub fn with_interface(
         mut self,
         interface_class: u8,
@@ -453,3 +477,27 @@ pub trait UsbDeviceHandler {
     /// ```
     fn as_any(&mut self) -> &mut dyn Any;
 }
+
+#[cfg(test)]
+mod test {
+    use super::*;
+
+    #[test]
+    fn test_set_string_descriptors() {
+        let mut device = UsbDevice::new(0);
+
+        assert_eq!(device.string_pool.len(), 4);
+
+        assert!(device.set_configuration_name("test").is_some());
+        assert!(device.set_manufacturer_name("test").is_some());
+        assert!(device.set_product_name("test").is_some());
+        assert!(device.set_serial_number("test").is_some());
+
+        assert_eq!(device.string_pool.len(), 4);
+
+        assert_eq!(device.string_pool[&1], "test");
+        assert_eq!(device.string_pool[&2], "test");
+        assert_eq!(device.string_pool[&3], "test");
+        assert_eq!(device.string_pool[&4], "test");
+    }
+}