--test-retimer: Allow testing retimer update mode

Only on systems with Intel retimer (Framework 13 with Intel CPU)

This is what it's supposed to look like.
Update mode for each port: false, true, false

```
> framework_tool --test-retimer
Retimer Self-Test
  Retimers on PD Controller 0
    In update mode: false
    Enabling update mode
    In update mode: true
    Disabling update mode
    In update mode: false
  Retimers on PD Controller 1
    In update mode: false
    Enabling update mode
    In update mode: true
    Disabling update mode
    In update mode: false
```

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,67 @@ 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: {:?}",
+            ec.retimer_in_fwupd_mode(i).unwrap()
+        );
+        if update_mode? {
+            println!("      Disabling it to start test.");
+            ec.retimer_enable_fwupd(i, false)?;
+        }
+
+        println!("    Enabling update mode");
+        let success = ec.retimer_enable_fwupd(i, true);
+        println!(
+            "    In update mode: {:?}",
+            ec.retimer_in_fwupd_mode(i).unwrap()
+        );
+
+        println!("    Disabling update mode");
+        ec.retimer_enable_fwupd(i, false)?;
+        os_specific::sleep(100);
+        println!(
+            "    In update mode: {:?}",
+            ec.retimer_in_fwupd_mode(i).unwrap()
+        );
+        success?;
+    }
+    Ok(())
+}
+
 fn smbios_info() {
     println!("Summary");
     println!("  Is Framework: {}", is_framework());

framework_lib/src/commandline/uefi.rs

@@ -84,6 +84,7 @@ pub fn parse(args: &[String]) -> Cli {
         pd_addrs: None,
         pd_ports: None,
         test: false,
+        test_retimer: false,
         dry_run: false,
         force: false,
         help: false,
@@ -495,6 +496,9 @@ pub fn parse(args: &[String]) -> Cli {
         } else if arg == "-t" || arg == "--test" {
             cli.test = true;
             found_an_option = true;
+        } else if arg == "-t" || arg == "--test-retimer" {
+            cli.test_retimer = true;
+            found_an_option = true;
         } else if arg == "-f" || arg == "--force" {
             cli.force = true;
             found_an_option = true;