From b5702fa62ad3b1329de26e7451d383d11ad3dfc5 Mon Sep 17 00:00:00 2001
From: Andreas Karner <andreaskarner@outlook.com>
Date: Mon, 27 Oct 2025 09:23:55 +0100
Subject: [PATCH] feat: probe-rs: added speed configuration option

Signed-off-by: Andreas Karner <andreaskarner@outlook.com>
---
 packages/jumpstarter-driver-probe-rs/README.md                 | 2 ++
 packages/jumpstarter-driver-probe-rs/examples/exporter.yaml    | 1 +
 .../jumpstarter_driver_probe_rs/driver.py                      | 3 +++
 3 files changed, 6 insertions(+)

diff --git a/packages/jumpstarter-driver-probe-rs/README.md b/packages/jumpstarter-driver-probe-rs/README.md
index cd104fe35..b477253b7 100644
--- a/packages/jumpstarter-driver-probe-rs/README.md
+++ b/packages/jumpstarter-driver-probe-rs/README.md
@@ -24,6 +24,7 @@ export:
       chip: "RP2350"
       protocol: "swd"
       connect_under_reset: false
+      speed: 4000
 ```
 
 ### Config parameters
@@ -35,6 +36,7 @@ export:
 | chip                | The target chip                                                | str             | no       |          |
 | protocol            | The target protocol                                            | "swd" or "jtag" | no       |          |
 | connect_under_reset | Connect to the target while asserting reset                    | bool            | no       | false    |
+| speed               | Connection speed in kHz                                        | int             | no       |          |
 
 ## API Reference
 
diff --git a/packages/jumpstarter-driver-probe-rs/examples/exporter.yaml b/packages/jumpstarter-driver-probe-rs/examples/exporter.yaml
index cfb800d52..d510a6b98 100644
--- a/packages/jumpstarter-driver-probe-rs/examples/exporter.yaml
+++ b/packages/jumpstarter-driver-probe-rs/examples/exporter.yaml
@@ -11,4 +11,5 @@ export:
     config:
       probe_rs_path: /home/majopela/.cargo/bin/probe-rs
       protocol: "swd"
+      speed: 4000
 
diff --git a/packages/jumpstarter-driver-probe-rs/jumpstarter_driver_probe_rs/driver.py b/packages/jumpstarter-driver-probe-rs/jumpstarter_driver_probe_rs/driver.py
index 808e4b52e..bce4d4af0 100644
--- a/packages/jumpstarter-driver-probe-rs/jumpstarter_driver_probe_rs/driver.py
+++ b/packages/jumpstarter-driver-probe-rs/jumpstarter_driver_probe_rs/driver.py
@@ -17,6 +17,7 @@ class ProbeRs(Driver):
     chip: str | None = None
     protocol: str | None = None
     connect_under_reset: bool = False
+    speed: int | None = None
 
     def __post_init__(self):
         if hasattr(super(), "__post_init__"):
@@ -83,6 +84,8 @@ def env_from_cfg(self):
             env["PROBE_RS_PROTOCOL"] = self.protocol
         if self.connect_under_reset:
             env["PROBE_RS_CONNECT_UNDER_RESET"] = "true"
+        if self.speed:
+            env["PROBE_RS_SPEED"] = str(self.speed)
         return env