Closing the program now triggers a forced kill of compact.exe if it's active - with a warning.

Added option to restart as administrator if you encounter an UnauthorizedAccessException

Author: Iridium-IO

WindowsApp1/Compact.vb

@@ -3,6 +3,7 @@ Imports System.Runtime.InteropServices
 Imports System.Text
 Imports System.Text.RegularExpressions
 Imports Ookii.Dialogs                                                                          'Uses Ookii Dialogs for the non-archaic filebrowser dialog. http://www.ookii.org/Software/Dialogs
+Imports System.Management
 
 Public Class Compact
     Dim version = "1.3.6"
@@ -274,7 +275,7 @@ Public Class Compact
 
 
                 Dim DIwDString = New DirectoryInfo(wDString)
-
+                directorysizeexceptionCount = 0
                 workingDir = Chr(34) + wDString.ToString() + Chr(34)
                 chosenDirDisplay.Text = DIwDString.Parent.ToString + " ❯ " + DIwDString.Name.ToString
                 uncompressedfoldersize = Math.Round(DirectorySize(DIwDString, True), 0)
@@ -362,6 +363,7 @@ Public Class Compact
             MyProcess.BeginErrorReadLine()
             MyProcess.BeginOutputReadLine()
 
+
             Try
 
                 MyProcess.StandardInput.WriteLine("cd /d " + workingDir)
@@ -466,21 +468,17 @@ Public Class Compact
 
             Else
                 Try
-                    MyProcess.Kill()
+                    LetsKillStuff()
                 Catch ex As Exception
                 End Try
-
             End If
         Else
-
             Try
-                MyProcess.Kill()
+                LetsKillStuff()
             Catch ex As Exception
             End Try
-
         End If
 
-
     End Sub
 
 
@@ -608,8 +606,6 @@ Public Class Compact
     Dim hasqueryfinished = 0
 
 
-
-
     Private Sub RunCompact(desiredfunction As String)
 
         If desiredfunction = "compact" Then
@@ -668,7 +664,6 @@ Public Class Compact
 
         End If
 
-
     End Sub
 
 
@@ -706,7 +701,9 @@ Public Class Compact
     Private Function DirectorySize _
         (ByVal dInfo As IO.DirectoryInfo, ByVal includeSubdirectories As Boolean) As Long
 
+
         Try
+
             Dim totalSize As Long = dInfo.EnumerateFiles().Sum(Function(file) file.Length)
 
             If includeSubdirectories Then
@@ -715,18 +712,31 @@ Public Class Compact
 
             Return totalSize
 
-            directorysizeexceptionCount = 0
-
         Catch generatedexceptionname As UnauthorizedAccessException
+            directorysizeexceptionCount += 1
 
-            If directorysizeexceptionCount = 0 Then
-                MsgBox("This directory contains a subfolder that you do not have permission to access. Please try running the program again as an Administrator." _
-                       & vbCrLf & vbCrLf & "If the problem persists, the subfolder is most likely protected by the System, and by design this program will refuse to let you proceed.")
+
+            If directorysizeexceptionCount = 1 Then
 
                 overrideCompressFolderButton = 1
-                directorysizeexceptionCount = 1
-            End If
+                directorysizeexceptionCount += 1
+
+                If My.User.IsInRole(ApplicationServices.BuiltInRole.Administrator) = False Then
+                    If MessageBox.Show("This directory contains a subfolder that you do not have permission to access. Please try running the program again as an Administrator." _
+                         & vbCrLf & vbCrLf & "If the problem persists, the subfolder is most likely protected by the System, and by design this program will refuse to let you proceed." _
+                        & vbCrLf & vbCrLf & " Would you like to restart the program as an Administrator?", "Permission Error", MessageBoxButtons.YesNo, MessageBoxIcon.Exclamation) = DialogResult.Yes Then
+
+                        RCMenu.RunAsAdmin()
+                        Me.Close()
 
+                    End If
+
+                Else
+                    MsgBox("This directory contains a subfolder that you do not have permission To access." _
+                       & vbCrLf & vbCrLf & "The subfolder is most likely protected by the System, and by design this program will refuse to let you proceed.")
+                End If
+
+            End If
 
         Catch ex As Exception
 
@@ -737,6 +747,24 @@ Public Class Compact
 
 
 
+    Private WithEvents KillProc As Process
+    Private Sub LetsKillStuff()
+
+        KillProc = New Process
+
+        With KillProc.StartInfo
+            .FileName = "powershell.exe"
+            .Arguments = "taskkill /f /im compact.exe"
+            .UseShellExecute = False
+            .CreateNoWindow = True
+        End With
+
+        KillProc.Start()
+
+    End Sub
+
+
+
 
     '////////////////////TESTING////////////////////
 
@@ -796,5 +824,4 @@ Public Class Compact
     End Sub
 
 
-
 End Class

WindowsApp1/CompactGUI.vbproj

@@ -62,6 +62,7 @@
     <Reference Include="System.Data" />
     <Reference Include="System.Deployment" />
     <Reference Include="System.Drawing" />
+    <Reference Include="System.Management" />
     <Reference Include="System.Windows.Forms" />
     <Reference Include="System.Xml" />
     <Reference Include="System.Core" />