Add Windows update install feature

ocdtrekkie · Aug 4, 2024 · 98a0a90 · 98a0a90
1 parent d769fd5
commit 98a0a90
Show file tree

Hide file tree

Showing 9 changed files with 84 additions and 4 deletions.
diff --git a/App.config b/App.config
@@ -39,6 +39,9 @@
       <setting name="Scripts_FolderURI" serializeAs="String">
         <value>C:\HAC\scripts\</value>
       </setting>
+      <setting name="Tools_FolderURI" serializeAs="String">
+        <value>C:\Program Files\XRFAgent\tools\</value>
+      </setting>
     </XRFAgent.Properties.Settings>
   </applicationSettings>
 </configuration>
diff --git a/Properties/AssemblyInfo.cs b/Properties/AssemblyInfo.cs
@@ -32,5 +32,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.0.0.10")]
-[assembly: AssemblyFileVersion("1.0.0.10")]
+[assembly: AssemblyVersion("1.0.0.11")]
+[assembly: AssemblyFileVersion("1.0.0.11")]
diff --git a/Properties/Settings.Designer.cs b/Properties/Settings.Designer.cs
diff --git a/Properties/Settings.settings b/Properties/Settings.settings
@@ -11,5 +11,8 @@
     <Setting Name="Scripts_FolderURI" Type="System.String" Scope="Application">
       <Value Profile="(Default)">C:\HAC\scripts\</Value>
     </Setting>
+    <Setting Name="Tools_FolderURI" Type="System.String" Scope="Application">
+      <Value Profile="(Default)">C:\Program Files\XRFAgent\tools\</Value>
+    </Setting>
   </Settings>
 </SettingsFile>
diff --git a/XRFAgent.csproj b/XRFAgent.csproj
@@ -25,7 +25,7 @@
     <UpdatePeriodically>false</UpdatePeriodically>
     <UpdateRequired>false</UpdateRequired>
     <MapFileExtensions>true</MapFileExtensions>
-    <ApplicationRevision>10</ApplicationRevision>
+    <ApplicationRevision>11</ApplicationRevision>
     <ApplicationVersion>1.0.0.%2a</ApplicationVersion>
     <UseApplicationTrust>false</UseApplicationTrust>
     <BootstrapperEnabled>true</BootstrapperEnabled>

diff --git a/docs/ErrorCodes.md b/docs/ErrorCodes.md
@@ -9,9 +9,13 @@
 6021 - Update check failed
 6022 - Update failed
 6023 - Update starting (informational)
+6024 - Tool installation failed
 
 6031 - External process start error
 6032 - Registry access error
 
+6041 - Windows Update error
+6042 - Windows Update results (informational)
+
 6051 - Detected new software installed (informational)
 6052 - Truncated installed software inventory (informational)
diff --git a/modCommand.cs b/modCommand.cs
@@ -63,6 +63,21 @@ public static void Handle(string inputCommand, string inputSource, string reques
                                     outputResponse = "Updating"; break;
                             }
                             break;
+                        case "windows":
+                            result = modSystem.InstallWindowsUpdates();
+                            switch (result)
+                            {
+                                case 0:
+                                    outputResponse = "Update successful"; break;
+                                case -1:
+                                case 31:
+                                    outputResponse = "Update error"; break;
+                                case 3010:
+                                    outputResponse = "Reboot required"; break;
+                                default:
+                                    outputResponse = "Updating"; break;
+                            }
+                            break;
                         default: break;
                     }
                     break;

diff --git a/modSystem.cs b/modSystem.cs
@@ -181,6 +181,35 @@ public static string GetSystemDetails()
             }
         }
 
+        public static int InstallWindowsUpdates()
+        {
+            try
+            {
+                if (File.Exists(Properties.Settings.Default.Tools_FolderURI + "WindowsUpdatePush.exe") == false)
+                {
+                    int installResult = modUpdate.InstallWindowsUpdatePush();
+                    if (installResult == -1)
+                    {
+                        return -1;
+                    }
+                }
+                Process UpdateRunner = new Process();
+                UpdateRunner.StartInfo.UseShellExecute = false;
+                UpdateRunner.StartInfo.RedirectStandardOutput = true;
+                UpdateRunner.StartInfo.WindowStyle = ProcessWindowStyle.Hidden;
+                UpdateRunner.StartInfo.FileName = Properties.Settings.Default.Tools_FolderURI + "WindowsUpdatePush.exe";
+                UpdateRunner.Start();
+                UpdateRunner.WaitForExit();
+                modLogging.LogEvent(UpdateRunner.StandardOutput.ReadToEnd(), EventLogEntryType.Information, 6042);
+                return UpdateRunner.ExitCode;
+            }
+            catch(Exception err)
+            {
+                modLogging.LogEvent("Windows Update error: " + err.Message, EventLogEntryType.Error, 6041);
+                return -1;
+            }
+        }
+
         /// <summary>
         /// Reboots the host computer
         /// </summary>

diff --git a/modUpdate.cs b/modUpdate.cs
@@ -99,6 +99,23 @@ public static int UpdateAgent()
             return updateNeeded;
         }
 
+        public static int InstallWindowsUpdatePush()
+        {
+            try
+            {
+                Directory.CreateDirectory(Properties.Settings.Default.Tools_FolderURI);
+                UpdateDownloadClient.DownloadFile(Properties.Settings.Default.Update_SourceURI + "windowsupdatepush.zip", Properties.Settings.Default.Tools_FolderURI + "windowsupdatepush.zip");
+                ZipFile.ExtractToDirectory(Properties.Settings.Default.Tools_FolderURI + "windowsupdatepush.zip", Properties.Settings.Default.Tools_FolderURI);
+                File.Delete(Properties.Settings.Default.Tools_FolderURI + "windowsupdatepush.zip");
+                return 0;
+            }
+            catch (Exception err)
+            {
+                modLogging.LogEvent(err.Message, EventLogEntryType.Error, 6024);
+                return -1;
+            }
+        }
+
         public static int Autoupdate()
         {
             if (modDatabase.GetConfig("Update_Autoupdate") == "enabled")