WIP: Retimer test

Signed-off-by: Daniel Schaefer <[email protected]>

Author: JohnAZoidberg

framework_lib/src/commandline/clap_std.rs

@@ -263,6 +263,10 @@ struct ClapCli {
     #[arg(long, short)]
     test: bool,
 
+    /// Run self-test to check if interaction with retimers is possible
+    #[arg(long)]
+    test_retimer: bool,
+
     /// Force execution of an unsafe command - may render your hardware unbootable!
     #[arg(long, short)]
     force: bool,
@@ -455,6 +459,7 @@ pub fn parse(args: &[String]) -> Cli {
         pd_addrs,
         pd_ports,
         test: args.test,
+        test_retimer: args.test_retimer,
         dry_run: args.dry_run,
         force: args.force,
         // TODO: Set help. Not very important because Clap handles this by itself

framework_lib/src/commandline/mod.rs

@@ -188,6 +188,7 @@ pub struct Cli {
     pub flash_rw_ec: Option<String>,
     pub driver: Option<CrosEcDriverType>,
     pub test: bool,
+    pub test_retimer: bool,
     pub dry_run: bool,
     pub force: bool,
     pub intrusion: bool,
@@ -271,6 +272,7 @@ pub fn parse(args: &[String]) -> Cli {
             // flash_rw_ec
             driver: cli.driver,
             test: cli.test,
+            test_retimer: cli.test_retimer,
             dry_run: cli.dry_run,
             // force
             intrusion: cli.intrusion,
@@ -1174,6 +1176,11 @@ pub fn run_with_args(args: &Cli, _allupdate: bool) -> i32 {
             println!("FAILED!!");
             return 1;
         }
+    } else if args.test_retimer {
+        println!("Retimer Self-Test");
+        if let Err(err) = selftest_retimer(&ec) {
+            println!("  Failed: {:?}", err);
+        }
     } else if args.power {
         return power::get_and_print_power_info(&ec);
     } else if args.thermal {
@@ -1606,6 +1613,55 @@ fn selftest(ec: &CrosEc) -> Option<()> {
     Some(())
 }
 
+// Platforms that have Retimers
+// Retimer I2C is always connected to the CPU, except for the Framework 16 dGPU retimer.
+//
+// - Framework 12
+//   - No Retimer, only retimer for both left ports (no firmware)
+// - Framework 13 Intel
+//   - One Intel retimer for each port (with firmware)
+// - Framework 13 AMD 7040
+//   - Kandou Retimer on top two ports (no firmware)
+//   - Analogix Retimer on bottom two ports (no firmware)
+// - Framework 13 AMD AI 300
+//   - Parade Retimer on top two ports (with firmware)
+//   - Analogix Retimer on bottom two ports (no firmware)
+// - Framework 16 AMD 7040
+//   - Kandou Retimer on top two ports (no firmware)
+//   - Analogix Retimer on lower and middle left ports (no firmware)
+// - Framework 16 AMD AI 300
+//   - Parade Retimer on top two ports (with firmware)
+// - Framework 16 AMD dGPU
+//   - None
+// - Framework 16 NVIDIA dGPU
+//   - Parade Retimer
+// - Framework Desktop
+//   - Parade Retimer on both back ports (with firmware)
+fn selftest_retimer(ec: &CrosEc) -> EcResult<()> {
+    // TODO: Make sure that it can work for the NVIDIA dGPU retimer and increase to 3
+    for i in 0..2 {
+        let update_mode = ec.retimer_in_fwupd_mode(i);
+        if update_mode == Err(EcError::Response(EcResponseStatus::InvalidParameter)) {
+            println!("  Retimer status not supported on this platform. Cannot test");
+            return Ok(());
+        }
+        println!("  Retimers on PD Controller {}", i);
+        println!("    In update mode: {:?}", update_mode);
+        if update_mode? {
+            println!("    Unexpected! During normal operaton it should not be in update mode! Disabling...");
+            ec.retimer_enable_fwupd(i, false)?;
+        }
+        println!("    Enabling update mode");
+        let success = ec.retimer_enable_fwupd(i, true);
+        println!("    Success: {:?}", success);
+
+        println!("    Disabling again");
+        ec.retimer_enable_fwupd(i, false)?;
+        success?;
+    }
+    Ok(())
+}
+
 fn smbios_info() {
     println!("Summary");
     println!("  Is Framework: {}", is_framework());