From 4bf631e34ddcec9529058c8b780aa2b6e740ae60 Mon Sep 17 00:00:00 2001
From: Lex Lim <hyperzlib@outlook.com>
Date: Sat, 6 May 2023 17:50:05 +0800
Subject: [PATCH] Add feature to custom bypass process list

1. Add feature to custom bypass process list
2. Add example config form winsw
---
 EnergyStar/EnergyManager.cs | 13 ++++++++-
 EnergyStar/Program.cs       | 57 +++++++++++++++++++++++++++++++++++++
 examples/winsw/winsw.xml    | 53 ++++++++++++++++++++++++++++++++++
 3 files changed, 122 insertions(+), 1 deletion(-)
 create mode 100644 examples/winsw/winsw.xml

diff --git a/EnergyStar/EnergyManager.cs b/EnergyStar/EnergyManager.cs
index 0b93002..2b7bc68 100644
--- a/EnergyStar/EnergyManager.cs
+++ b/EnergyStar/EnergyManager.cs
@@ -7,7 +7,7 @@ namespace EnergyStar
 {
     public unsafe class EnergyManager
     {
-        public static readonly HashSet<string> BypassProcessList = new HashSet<string>
+        public static readonly HashSet<string> DefaultBypassProcessList = new HashSet<string>
         {
             // Not ourselves
             "EnergyStar.exe".ToLowerInvariant(),
@@ -44,6 +44,7 @@ public unsafe class EnergyManager
             // WUDF
             "WUDFRd.exe".ToLowerInvariant(),
         };
+        public static HashSet<string> BypassProcessList = DefaultBypassProcessList;
         // Speical handling needs for UWP to get the child window process
         public const string UWPFrameHostApp = "ApplicationFrameHost.exe";
 
@@ -184,5 +185,15 @@ public static void ThrottleAllUserBackgroundProcesses()
                 Win32Api.CloseHandle(hProcess);
             }
         }
+
+        public static void SetBypassProcessList(IEnumerable<string> newBypassList)
+        {
+            var list = new List<string>();
+
+            list.AddRange(DefaultBypassProcessList);
+            list.AddRange(newBypassList);
+
+            BypassProcessList = list.Distinct().ToHashSet();
+        }
     }
 }
diff --git a/EnergyStar/Program.cs b/EnergyStar/Program.cs
index 6f791a4..186ee85 100644
--- a/EnergyStar/Program.cs
+++ b/EnergyStar/Program.cs
@@ -24,6 +24,46 @@ static async void HouseKeepingThreadProc()
             }
         }
 
+        static void OnBypassListChanged(object sender, FileSystemEventArgs e)
+        {
+            var path = e.FullPath;
+            if (path == null)
+                return;
+
+            LoadBypassList(path);
+        }
+
+        static void LoadBypassList(string path)
+        {
+            if (File.Exists(path))
+            {
+                var file = File.OpenText(path);
+                var content = file.ReadToEnd();
+                file.Close();
+
+                if (content != null)
+                {
+                    var srcBypassList = content.Replace("\r\n", "\n").Split('\n');
+                    var bypassList = new List<string> { };
+                    // Remove empty items
+                    foreach (var item in srcBypassList)
+                    {
+                        var trimed = item.Trim().ToLowerInvariant();
+                        if (trimed.Length > 0)
+                        {
+                            bypassList.Add(trimed);
+                        }
+                    }
+
+                    if (bypassList != null)
+                    {
+                        EnergyManager.SetBypassProcessList(bypassList);
+                        Console.WriteLine("Bypass list updated");
+                    }
+                }
+            }
+        }
+
         static void Main(string[] args)
         {
             // Well, this program only works for Windows Version starting with Cobalt...
@@ -38,6 +78,23 @@ static void Main(string[] args)
                 Environment.Exit(120);
             }
 
+            // Load bypass list and watch for changes.
+            var baseDir = AppDomain.CurrentDomain.BaseDirectory;
+            var bypassListFile = Path.Combine(baseDir, "bypass.txt");
+            using var bypassListWatcher = new FileSystemWatcher(baseDir);
+
+            bypassListWatcher.Changed += OnBypassListChanged;
+            bypassListWatcher.Created += OnBypassListChanged;
+            bypassListWatcher.Deleted += OnBypassListChanged;
+            bypassListWatcher.Renamed += OnBypassListChanged;
+
+            bypassListWatcher.Filter = "bypass.txt";
+            bypassListWatcher.IncludeSubdirectories = true;
+            bypassListWatcher.EnableRaisingEvents = true;
+
+            LoadBypassList(bypassListFile);
+
+
             HookManager.SubscribeToWindowEvents();
             EnergyManager.ThrottleAllUserBackgroundProcesses();
 
diff --git a/examples/winsw/winsw.xml b/examples/winsw/winsw.xml
new file mode 100644
index 0000000..75d950c
--- /dev/null
+++ b/examples/winsw/winsw.xml
@@ -0,0 +1,53 @@
+<!--
+  MIT License
+
+  Copyright (c) 2008-2020 Kohsuke Kawaguchi, Sun Microsystems, Inc., CloudBees,
+  Inc., Oleg Nenashev and other contributors
+
+  Permission is hereby granted, free of charge, to any person obtaining a copy
+  of this software and associated documentation files (the "Software"), to deal
+  in the Software without restriction, including without limitation the rights
+  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+  copies of the Software, and to permit persons to whom the Software is
+  furnished to do so, subject to the following conditions:
+
+  The above copyright notice and this permission notice shall be included in all
+  copies or substantial portions of the Software.
+
+  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+  SOFTWARE.
+-->
+
+<!--
+ This is an example of a minimal Windows Service Wrapper configuration, which includes only mandatory options.
+ 
+ This configuration file should be placed near the WinSW executable, the name should be the same.
+ E.g. for myapp.exe the configuration file name should be myapp.xml
+ 
+ You can find more information about the configuration options here: https://github.com/kohsuke/winsw/blob/master/doc/xmlConfigFile.md
+ Full example: https://github.com/kohsuke/winsw/blob/master/examples/sample-allOptions.xml
+-->
+<service>
+  
+  <!-- ID of the service. It should be unique across the Windows system-->
+  <id>energystar</id>
+  <!-- Display name of the service -->
+  <name>EnergyStar</name>
+  <!-- Service description -->
+  <description>A terrible application setting SV2 Efficiency Mode for inactive Windows apps and user background apps</description>
+  
+  <!-- Path to the executable, which should be started -->
+  <executable>%BASE%\EnergyStar.exe</executable>
+
+  <logpath>%BASE%\logs</logpath>
+
+  <log mode="roll">
+    <sizeThreshold>1024</sizeThreshold>
+    <keepFiles>1</keepFiles>
+  </log>
+</service>