From bc3ad0a54f4d0a4e87f71516a3b5c2e8b50185a7 Mon Sep 17 00:00:00 2001
From: Jeremy Whiting <jeremy.whiting@collabora.com>
Date: Wed, 18 Oct 2023 15:09:16 -0600
Subject: [PATCH] Implement SetTdpLimit.

Write given limit to /sys/class/hwmon/hwmon5/power[12]_cap if in the
range of 3..15.
Return false if out of range or unable to write.
---
 com.steampowered.SteamOSManager1.xml |  4 +--
 src/manager.rs                       | 37 ++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 2 deletions(-)

diff --git a/com.steampowered.SteamOSManager1.xml b/com.steampowered.SteamOSManager1.xml
index 7354b17..8e61396 100644
--- a/com.steampowered.SteamOSManager1.xml
+++ b/com.steampowered.SteamOSManager1.xml
@@ -160,8 +160,8 @@
     </method>
 
     <!--
-      SetTDPLimit:
-      @limit: The upper limit to use.
+      SetTdpLimit:
+      @limit: The upper limit to use (between 3 and 15).
       @success: True on success. False otherwise.
 
       Set the TDP limit to the given value.
diff --git a/src/manager.rs b/src/manager.rs
index 42873c0..7ad00a9 100644
--- a/src/manager.rs
+++ b/src/manager.rs
@@ -200,6 +200,43 @@ impl SMManager {
         }
     }
     
+    async fn set_tdp_limit(&self, limit: i32) -> bool {
+        // Set TDP limit given if within range (3-15)
+        // Returns false on error or out of range
+        // Writes value to /sys/class/hwmon/hwmon5/power[12]_cap
+        if !(3..=15).contains(&limit) {
+            return false;
+        }
+
+        let result = File::create("/sys/class/hwmon/hwmon5/power1_cap").await;
+        let mut power1file;
+        match result {
+            Ok(f) => power1file = f,
+            Err(message) => { println!("Error opening sysfs power1_cap file for writing TDP limits {message}"); return false; }
+        };
+
+        let result = File::create("/sys/class/hwmon/hwmon5/power2_cap").await;
+        let mut power2file;
+        match result {
+            Ok(f) => power2file = f,
+            Err(message) => { println!("Error opening sysfs power2_cap file for wtriting TDP limits {message}"); return false; }
+        };
+
+        // Now write the value * 1,000,000
+        let data = format!("{limit}000000");
+        let result = power1file.write(data.as_bytes()).await;
+        match result {
+            Ok(_worked) => {
+                let result = power2file.write(data.as_bytes()).await;
+                match result {
+                    Ok(_worked) => true,
+                    Err(message) => { println!("Error writing to power2_cap file: {message}"); false }
+                }
+            },
+            Err(message) => { println!("Error writing to power1_cap file: {message}"); false }
+        }
+    }
+
     /// A version property.
     #[dbus_interface(property)]
     async fn version(&self) -> u32 {