From 50eeba6b7f6d04392a80f519dc009c9c22aba2bc Mon Sep 17 00:00:00 2001
From: Dylan Bickerstaff <dylan.bickerstaff@cedarfair.com>
Date: Thu, 11 Jul 2024 10:46:50 -0400
Subject: [PATCH] .

---
 CommonProperties.xml                         |  2 +-
 SuperLauncher/CredentialExpirationService.cs | 47 +++++++++++---------
 SuperLauncher/ModernLauncher.xaml.cs         | 35 ++++++++++++---
 SuperLauncher/ModernLauncherBadge.xaml.cs    |  1 -
 SuperLauncher/RunAsHelper.cs                 |  4 ++
 SuperLauncher/Settings.cs                    |  2 +-
 6 files changed, 61 insertions(+), 30 deletions(-)

diff --git a/CommonProperties.xml b/CommonProperties.xml
index bccf46a..1d214b8 100644
--- a/CommonProperties.xml
+++ b/CommonProperties.xml
@@ -18,7 +18,7 @@
 		<PlatformTarget>x64</PlatformTarget>
 		<Platforms>x64</Platforms>
 		<RuntimeIdentifier>win-x64</RuntimeIdentifier>
-		<NoWarn>CS0108;WFAC010</NoWarn>
+		<NoWarn>CS0108;WFAC010;CA1416</NoWarn>
 	</PropertyGroup>
 	<!--<ItemGroup>
 		<AssemblyAttribute Include="System.Reflection.AssemblyTitleAttribute">
diff --git a/SuperLauncher/CredentialExpirationService.cs b/SuperLauncher/CredentialExpirationService.cs
index 151a52d..86d14f7 100644
--- a/SuperLauncher/CredentialExpirationService.cs
+++ b/SuperLauncher/CredentialExpirationService.cs
@@ -6,20 +6,21 @@
 
 namespace SuperLauncher
 {
-    public static class CredentialExpirationService
+    public class CredentialExpirationService
     {
-        private static Timer CheckTimer = new();
-        private static Timer NotifyTimer = new();
-        public static DateTime PasswordLastSet = DateTime.MaxValue;
-        public static TimeSpan MaxPasswordAge = TimeSpan.MaxValue;
-        private static ExpStat Status = ExpStat.Unknown;
-        public static string PasswordExpirationMessage
+        private readonly Timer CheckTimer = new();
+        private readonly Timer NotifyTimer = new();
+        private DateTime PasswordLastSet = DateTime.MaxValue;
+        private TimeSpan MaxPasswordAge = TimeSpan.MaxValue;
+        private ExpStat Status = ExpStat.Unknown;
+        private readonly SecurityIdentifier SID;
+        public string PasswordExpirationMessage
         {
             get
             {
                 if (Status == ExpStat.NeverExpires) return "Password never expires.";
                 if (Status == ExpStat.DCNotResponding) return "Could not determine password expiration date, Active Directory is offline.";
-                if (Status == ExpStat.Expires) return 
+                if (Status == ExpStat.Expires) return
                 "Password expires in " +
                 ExpirationTimeSpan.Days +
                 " day(s), " +
@@ -30,7 +31,7 @@ public static string PasswordExpirationMessage
                 return "An un-known error occured when determining your password expiration date.";
             }
         }
-        public static DateTime PasswordExpirationDate
+        public DateTime PasswordExpirationDate
         {
             get
             {
@@ -38,16 +39,18 @@ public static DateTime PasswordExpirationDate
                 return PasswordLastSet.Add(MaxPasswordAge);
             }
         }
-        public static TimeSpan ExpirationTimeSpan
+        public TimeSpan ExpirationTimeSpan
         {
             get
             {
                 return PasswordExpirationDate.Subtract(DateTime.Now);
             }
         }
-        public static void Initialize()
+        public CredentialExpirationService(SecurityIdentifier SID)
         {
-            Task.Run(() =>  {
+            this.SID = SID;
+            Task.Run(() =>
+            {
                 CheckTimer.Elapsed += CheckTimer_Elapsed;
                 CheckTimer.Enabled = true;
                 CheckTimer.Interval = TimeSpan.FromMinutes(5).TotalMilliseconds;
@@ -61,7 +64,7 @@ public static void Initialize()
                 NotifyTimer_Elapsed(null, null);
             });
         }
-        private static void NotifyTimer_Elapsed(object sender, ElapsedEventArgs e)
+        private void NotifyTimer_Elapsed(object sender, ElapsedEventArgs e)
         {
             if (PasswordExpirationDate.CompareTo(DateTime.Now.AddDays(Settings.Default.CredentialExpirationWarningDays)) <= 0)
             {
@@ -70,12 +73,14 @@ private static void NotifyTimer_Elapsed(object sender, ElapsedEventArgs e)
                 ModernLauncherNotifyIcon.Icon.ShowBalloonTip(0);
             }
         }
-        private static void CheckTimer_Elapsed(object sender, ElapsedEventArgs e)
+        private void CheckTimer_Elapsed(object sender, ElapsedEventArgs e)
         {
             try
             {
-                DirectorySearcher ds = new();
-                ds.SearchScope = SearchScope.Base;
+                DirectorySearcher ds = new()
+                {
+                    SearchScope = SearchScope.Base
+                };
                 ds.PropertiesToLoad.Clear();
                 ds.PropertiesToLoad.Add("maxPwdAge");
                 ds.Filter = "";
@@ -86,7 +91,7 @@ private static void CheckTimer_Elapsed(object sender, ElapsedEventArgs e)
                 ds.PropertiesToLoad.Clear();
                 ds.PropertiesToLoad.Add("userAccountControl");
                 ds.PropertiesToLoad.Add("pwdLastSet");
-                ds.Filter = "(objectSid=" + WindowsIdentity.GetCurrent().User.Value + ")";
+                ds.Filter = "(objectSid=" + SID + ")";
                 SearchResult user = ds.FindOne();
                 PasswordLastSet = DateTime.FromFileTime((long)user.Properties["pwdLastSet"][0]);
                 bool pwdNeverExpires = (((int)user.Properties["userAccountControl"][0]) & 0x00010000) == 0x00010000; //https://learn.microsoft.com/en-us/windows/win32/api/iads/ne-iads-ads_user_flag_enum
@@ -105,10 +110,10 @@ private static void CheckTimer_Elapsed(object sender, ElapsedEventArgs e)
         }
         private enum ExpStat
         {
-            Expires          = 0,
-            NeverExpires     = 1,
-            DCNotResponding  = 2,
-            Unknown          = -1
+            Expires = 0,
+            NeverExpires = 1,
+            DCNotResponding = 2,
+            Unknown = -1
         }
     }
 }
\ No newline at end of file
diff --git a/SuperLauncher/ModernLauncher.xaml.cs b/SuperLauncher/ModernLauncher.xaml.cs
index a1b62b2..ac7d9c5 100644
--- a/SuperLauncher/ModernLauncher.xaml.cs
+++ b/SuperLauncher/ModernLauncher.xaml.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections.Generic;
 using System.Runtime.InteropServices;
 using System.Windows;
 using System.Windows.Controls;
@@ -22,9 +23,10 @@ public static DpiScale DPI
         private bool IsClosing = false;
         private static DpiScale rDPI;
         private WindowInteropHelper WIH;
-        private uint ShowSuperLauncherMessage = Win32Interop.RegisterWindowMessage("ShowSuperLauncher");
+        private readonly uint ShowSuperLauncherMessage = Win32Interop.RegisterWindowMessage("ShowSuperLauncher");
         private HwndSource HWND;
         private ModernLauncherBadge ExpirationBadge = null;
+        private readonly Dictionary<string, CredentialExpirationService> CredentialExpirationServices = [];
         private readonly DoubleAnimation RenderBoostAnimation = new()
         {
             Duration = TimeSpan.FromSeconds(0.5),
@@ -93,8 +95,10 @@ private void Icon_Click(object sender, System.Windows.Forms.MouseEventArgs e)
             if (e.Button == System.Windows.Forms.MouseButtons.Right)
             {
                 Win32Interop.GetCursorPos(out Win32Interop.POINT point);
-                ModernLauncherContextMenu menu = new();
-                menu.Topmost = true;
+                ModernLauncherContextMenu menu = new()
+                {
+                    Topmost = true
+                };
                 menu.Frame.Content = new ModernLauncherContextMenuMain();
                 menu.Top = DPI.ScalePixelsDown(point.y) - 240;
                 menu.Left = DPI.ScalePixelsDown(point.x) - 80;
@@ -149,13 +153,23 @@ private void Window_Loaded(object sender, RoutedEventArgs e)
             OpenTopAnimation.Completed += OpenTopAnimation_Completed;
             CloseTopAnimation.Completed += CloseTopAnimation_Completed;
             InitializeNotifyIcon();
-            CredentialExpirationService.Initialize();
             Win32Interop.RegisterHotKey(WIH.Handle, 0, 0x1 | 0x4000, 0x53); //Register Hot Key ALT + S
             Win32Interop.RegisterHotKey(WIH.Handle, 1, 0x1 | 0x4000, 0x45); //Register Hot Key ALT + E
             Win32Interop.RegisterHotKey(WIH.Handle, 2, 0x1 | 0x4000, 0x52); //Register Hot Key ALT + R
             HWND.AddHook(HwndSourceHook);
             _ = Win32Interop.SetWindowLong(WIH.Handle, Win32Interop.SetWindowLongIndex.GWL_EXSTYLE, Win32Interop.ExtendedWindowStyles.WS_EX_TOOLWINDOW);
             SetElevateLabels();
+            if (RunAsHelper.GetOriginalInvokerDomainWithUserName() != RunAsHelper.GetCurrentDomainWithUserName())
+            {
+                CredentialExpirationServices.Add(
+                    RunAsHelper.GetOriginalInvokerDomainWithUserName(),
+                    new(RunAsHelper.GetOriginalInvokerSID())
+                );
+            }
+            CredentialExpirationServices.Add(
+                RunAsHelper.GetCurrentDomainWithUserName(),
+                new(RunAsHelper.GetCurrentSID())
+            );
         }
         public void OpenWindow(bool Center = false)
         {
@@ -286,12 +300,21 @@ private void BtnAdd_Click(object sender, RoutedEventArgs e)
         }
         private void ElevateUser_MouseEnter(object sender, MouseEventArgs e)
         {
-            ExpirationBadge = new(CredentialExpirationService.PasswordExpirationMessage);
+            List<string> lines = [];
+            foreach (KeyValuePair<string, CredentialExpirationService> kvp in CredentialExpirationServices)
+            {
+                string message = string.Empty;
+                message += kvp.Key;
+                message += ": ";
+                message += kvp.Value.PasswordExpirationMessage;
+                lines.Add(message);
+            }
+            ExpirationBadge = new(string.Join('\n', lines));
             ExpirationBadge.Show();
         }
         private void ElevateUser_MouseLeave(object sender, MouseEventArgs e)
         {
-            if (ExpirationBadge != null) ExpirationBadge.Close();
+            ExpirationBadge?.Close();
         }
     }
 }
\ No newline at end of file
diff --git a/SuperLauncher/ModernLauncherBadge.xaml.cs b/SuperLauncher/ModernLauncherBadge.xaml.cs
index 9c7f5f6..ea3e707 100644
--- a/SuperLauncher/ModernLauncherBadge.xaml.cs
+++ b/SuperLauncher/ModernLauncherBadge.xaml.cs
@@ -1,6 +1,5 @@
 using System;
 using System.Windows;
-using static System.Net.Mime.MediaTypeNames;
 
 namespace SuperLauncher
 {
diff --git a/SuperLauncher/RunAsHelper.cs b/SuperLauncher/RunAsHelper.cs
index 85b1cfd..87fcf77 100644
--- a/SuperLauncher/RunAsHelper.cs
+++ b/SuperLauncher/RunAsHelper.cs
@@ -111,6 +111,10 @@ public static SecurityIdentifier GetOriginalInvokerSID()
         {
             return new WindowsIdentity(GetOriginalInvokerUPN()).User;
         }
+        public static SecurityIdentifier GetCurrentSID()
+        {
+            return WindowsIdentity.GetCurrent().User;
+        }
         public static void Exit()
         {
             if (Program.ModernApplication != null)
diff --git a/SuperLauncher/Settings.cs b/SuperLauncher/Settings.cs
index e7098ae..bd80502 100644
--- a/SuperLauncher/Settings.cs
+++ b/SuperLauncher/Settings.cs
@@ -14,7 +14,7 @@ class SettingsDefault
         public string configPath = Path.Combine(@"C:\Users\Public\Documents\Below Average\Super Launcher\", RunAsHelper.GetOriginalInvokerDomain(), RunAsHelper.GetOriginalInvokerUserName(), "SuperLauncherConfig.xml");
         public XmlDocument XDoc = new();
         public XmlDocument XDocDefault = new();
-        private string DefaultXML = 
+        private string DefaultXML =
         "<!-- Super Launcher Config File -->" +
         "<SuperLauncher>" +
         "   <AutoElevate>false</AutoElevate>" +