UI: UAC Elevation

Rans4ckeR · Jan 1, 2024 · 3cd56a3 · 3cd56a3
1 parent 69ffca8
commit 3cd56a3
Show file tree

Hide file tree

Showing 9 changed files with 91 additions and 5 deletions.
diff --git a/CipherPunk.UI/App.xaml b/CipherPunk.UI/App.xaml
@@ -103,5 +103,8 @@
         <DataTemplate DataType="{x:Type ui:DefaultEllipticCurvesViewModel}">
             <ui:DefaultEllipticCurvesView/>
         </DataTemplate>
+        <DataTemplate DataType="{x:Type ui:ElevationViewModel}">
+            <ui:ElevationView/>
+        </DataTemplate>
     </Application.Resources>
 </Application>
diff --git a/CipherPunk.UI/App.xaml.cs b/CipherPunk.UI/App.xaml.cs
@@ -39,6 +39,7 @@ public App()
                     .AddSingleton<LoggingViewModel>()
                     .AddSingleton<DefaultCipherSuitesViewModel>()
                     .AddSingleton<DefaultEllipticCurvesViewModel>()
+                    .AddSingleton<ElevationViewModel>()
                     .AddCipherPunk()
                     .AddCipherSuiteInfoApi();
             }).Build();

diff --git a/CipherPunk.UI/Infrastructure/BaseViewModel.cs b/CipherPunk.UI/Infrastructure/BaseViewModel.cs
@@ -9,7 +9,6 @@
 
 internal abstract class BaseViewModel : ObservableRecipient
 {
-    private readonly IUacService uacService;
     private bool defaultCommandActive;
     private bool canExecuteDefaultCommand;
     private BitmapSource? uacIcon;
@@ -18,7 +17,7 @@ internal abstract class BaseViewModel : ObservableRecipient
     protected BaseViewModel(ILogger logger, IUacService uacService)
         : base(StrongReferenceMessenger.Default)
     {
-        this.uacService = uacService;
+        UacService = uacService;
         IsActive = true;
         Logger = logger;
         DefaultCommand = new AsyncRelayCommand<bool?>(ExecuteDefaultCommandAsync, _ => CanExecuteDefaultCommand);
@@ -39,12 +38,14 @@ public bool DefaultCommandActive
         }
     }
 
-    public BitmapSource UacIcon => uacIcon ??= uacService.GetShieldIcon();
+    public BitmapSource UacIcon => uacIcon ??= UacService.GetShieldIcon();
 
-    public bool Elevated => elevated ??= uacService.GetIntegrityLevel().Elevated;
+    public bool Elevated => elevated ??= UacService.GetIntegrityLevel().Elevated;
 
     protected ILogger Logger { get; }
 
+    protected IUacService UacService { get; }
+
     protected bool CanExecuteDefaultCommand
     {
         get => canExecuteDefaultCommand;

diff --git a/CipherPunk.UI/MainWindow.xaml b/CipherPunk.UI/MainWindow.xaml
@@ -76,6 +76,9 @@
                         <RibbonGroup>
                             <RibbonButton Label="Windows Default Elliptic Curves" LargeImageSource="{Binding EllipticCurvesButtonImage}" Command="{Binding DefaultEllipticCurvesViewModel.DefaultCommand}"/>
                         </RibbonGroup>
+                        <RibbonGroup>
+                            <RibbonButton Label="UAC Elevation" LargeImageSource="{Binding CipherSuitesButtonImage}" Command="{Binding ElevationViewModel.DefaultCommand}"/>
+                        </RibbonGroup>
                     </RibbonTab>
                 </Ribbon>
                 <ScrollViewer Grid.Row="1" Grid.Column="0" Background="{StaticResource WindowBackgroundSolidColorBrush}" HorizontalScrollBarVisibility="Auto" VerticalScrollBarVisibility="Auto">

diff --git a/CipherPunk.UI/MainWindowViewModel.cs b/CipherPunk.UI/MainWindowViewModel.cs
@@ -31,7 +31,8 @@ public MainWindowViewModel(
         RemoteServerTestViewModel remoteServerTestViewModel,
         LoggingViewModel loggingViewModel,
         DefaultCipherSuitesViewModel defaultCipherSuitesViewModel,
-        DefaultEllipticCurvesViewModel defaultEllipticCurvesViewModel)
+        DefaultEllipticCurvesViewModel defaultEllipticCurvesViewModel,
+        ElevationViewModel elevationViewModel)
         : base(logger, uacService)
     {
         IsActive = true;
@@ -46,6 +47,7 @@ public MainWindowViewModel(
         LoggingViewModel = loggingViewModel;
         DefaultCipherSuitesViewModel = defaultCipherSuitesViewModel;
         DefaultEllipticCurvesViewModel = defaultEllipticCurvesViewModel;
+        ElevationViewModel = elevationViewModel;
         CopyMessageCommand = new RelayCommand(ExecuteCopyMessageCommand);
         CloseMessageCommand = new RelayCommand(ExecuteCloseMessageCommand);
 
@@ -82,6 +84,8 @@ public MainWindowViewModel(
 
     public DefaultEllipticCurvesViewModel DefaultEllipticCurvesViewModel { get; }
 
+    public ElevationViewModel ElevationViewModel { get; }
+
     public double MainContentOpacity
     {
         get => mainContentOpacity; set => _ = SetProperty(ref mainContentOpacity, value);

diff --git a/CipherPunk.UI/UserControls/Elevation/ElevationView.xaml b/CipherPunk.UI/UserControls/Elevation/ElevationView.xaml
@@ -0,0 +1,32 @@
+<UserControl x:Class="CipherPunk.UI.ElevationView"
+             x:ClassModifier="internal"
+             xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:ui="clr-namespace:CipherPunk.UI"
+             d:DataContext="{d:DesignInstance Type=ui:ElevationViewModel}"
+             mc:Ignorable="d"
+             d:DesignHeight="450" d:DesignWidth="800">
+    <Grid>
+        <Grid.RowDefinitions>
+            <RowDefinition Height="Auto"/>
+        </Grid.RowDefinitions>
+        <Grid.ColumnDefinitions>
+            <ColumnDefinition Width="*"/>
+        </Grid.ColumnDefinitions>
+        <Border Style="{StaticResource BorderStyle}" Grid.Row="0" Grid.Column="0">
+            <Expander Header="Current Process UAC Elevation Level">
+                <Grid>
+                    <Grid.RowDefinitions>
+                        <RowDefinition Height="Auto"/>
+                    </Grid.RowDefinitions>
+                    <Grid.ColumnDefinitions>
+                        <ColumnDefinition Width="Auto"/>
+                    </Grid.ColumnDefinitions>
+                    <ListBox Grid.Row="0" Grid.Column="0" ItemsSource="{Binding MandatoryLevels}" SelectedItem="{Binding MandatoryLevel, Mode=OneWay}" IsEnabled="False"/>
+                </Grid>
+            </Expander>
+        </Border>
+    </Grid>
+</UserControl>
diff --git a/CipherPunk.UI/UserControls/Elevation/ElevationView.xaml.cs b/CipherPunk.UI/UserControls/Elevation/ElevationView.xaml.cs
@@ -0,0 +1,6 @@
+namespace CipherPunk.UI;
+
+internal sealed partial class ElevationView
+{
+    public ElevationView() => InitializeComponent();
+}
diff --git a/CipherPunk.UI/UserControls/Elevation/ElevationViewModel.cs b/CipherPunk.UI/UserControls/Elevation/ElevationViewModel.cs
@@ -0,0 +1,33 @@
+namespace CipherPunk.UI;
+
+using System.Collections.ObjectModel;
+
+internal sealed class ElevationViewModel : BaseViewModel
+{
+    private ObservableCollection<MandatoryLevel>? mandatoryLevels;
+    private MandatoryLevel? mandatoryLevel;
+
+    public ElevationViewModel(ILogger logger, IUacService uacService)
+        : base(logger, uacService)
+        => UpdateCanExecuteDefaultCommand();
+
+    public ObservableCollection<MandatoryLevel>? MandatoryLevels
+    {
+        get => mandatoryLevels;
+        private set => _ = SetProperty(ref mandatoryLevels, value);
+    }
+
+    public MandatoryLevel? MandatoryLevel
+    {
+        get => mandatoryLevel;
+        private set => _ = SetProperty(ref mandatoryLevel, value);
+    }
+
+    protected override Task DoExecuteDefaultCommandAsync(CancellationToken cancellationToken)
+    {
+        MandatoryLevels ??= new(Enum.GetValues<MandatoryLevel>().OrderByDescending(q => (int)q));
+        (MandatoryLevel, _) = UacService.GetIntegrityLevel();
+
+        return Task.CompletedTask;
+    }
+}
diff --git a/CipherPunk/Services/GroupPolicyService.cs b/CipherPunk/Services/GroupPolicyService.cs
@@ -183,6 +183,9 @@ private static void UpdateOrderPolicy(string valueData, string valueName, REG_VA
 
             using (phkResult)
             {
+                if (regOpenKeyExResult is WIN32_ERROR.ERROR_FILE_NOT_FOUND)
+                    return null;
+
                 if (regOpenKeyExResult is not WIN32_ERROR.ERROR_SUCCESS)
                     throw new Win32Exception((int)regOpenKeyExResult);