驱动更新实现

Author: JusterZhu

src/c#/GeneralUpdate.Bowl/Bowl.cs

@@ -7,7 +7,7 @@ namespace GeneralUpdate.Bowl;
 /// <summary>
 /// Surveillance Main Program.
 /// </summary>
-public class Bowl
+public sealed class Bowl
 {
     private IStrategy _strategy;
 
@@ -27,6 +27,9 @@ private void CreateStrategy()
         {
             _strategy = new LinuxStrategy();
         }
+        
+        if (_strategy == null)
+            throw new PlatformNotSupportedException("Unsupported operating system");
     }
 
     public Bowl SetParameter(MonitorParameter parameter)

src/c#/GeneralUpdate.Bowl/Strategys/AbstractStrategy.cs

@@ -11,7 +11,8 @@ public abstract class AbstractStrategy : IStrategy
     private readonly IReadOnlyList<string> _sensitiveCharacter = new List<string>
     {
         "Exit",
-        "exit"
+        "exit",
+        "EXIT"
     };
 
     public virtual void Launch()

src/c#/GeneralUpdate.Bowl/Strategys/LinuxStrategy.cs

@@ -66,8 +66,8 @@ private void Install()
 
     private string GetPacketName()
     {
-        string packageFileName = string.Empty;
-        LinuxSystem system = GetSystem();
+        var packageFileName = string.Empty;
+        var system = GetSystem();
         if (_rocdumpAmd64.Contains(system.Name))
         {
             packageFileName = $"procdump_3.3.0_amd64.deb";
@@ -107,9 +107,7 @@ private LinuxSystem GetSystem()
 
             return new LinuxSystem(distro, version);
         }
-        else
-        {
-            throw new FileNotFoundException("Cannot determine the Linux distribution. The /etc/os-release file does not exist.");
-        }
+        
+        throw new FileNotFoundException("Cannot determine the Linux distribution. The /etc/os-release file does not exist.");
     }
 }

src/c#/GeneralUpdate.Core/Driver/BackupDriverCommand.cs

@@ -1,29 +1,39 @@
-using System.IO;
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
+using System.IO;
+using System.Linq;
 using System.Text;
 
 namespace GeneralUpdate.Core.Driver
 {
     /// <summary>
     /// When the /export-driver command backs up a driver, it backs up the driver package along with all its dependencies, such as associated library files and other related files.
     /// </summary>
-    public class BackupDriverCommand : IDriverCommand
+    public class BackupDriverCommand(DriverInformation information) : DriverCommand
     {
-        private DriverInformation _information;
+        private readonly string _driverExtension = $"*{information.DriverFileExtension}";
 
-        public BackupDriverCommand(DriverInformation information) => _information = information;
-
-        public void Execute()
+        public override void Execute()
         {
+            var uninstalledDrivers = Directory.GetFiles(information.DriverDirectory, _driverExtension, SearchOption.AllDirectories).ToList();
+            var installedDrivers = GetInstalledDrivers(information.FieldMappings);
+            var tempDrivers = installedDrivers.Where(a => uninstalledDrivers.Any(b => string.Equals(a.OriginalName, Path.GetFileName(b)))).ToList();
+            information.Drivers = tempDrivers;
+            
+            //Export the backup according to the driver name.
+            if (Directory.Exists(information.OutPutDirectory))
+            {
+                Directory.Delete(information.OutPutDirectory, true);
+            }
+            
+            Directory.CreateDirectory(information.OutPutDirectory);
+            
             /*
              * Back up the specified list of drives.
              */
-            foreach (var driver in _information.Drivers)
+            foreach (var driver in tempDrivers)
             {
-                //Export the backup according to the driver name.
-                if (Directory.Exists(_information.OutPutDirectory))
-                    Directory.Delete(_information.OutPutDirectory, true);
-
-                Directory.CreateDirectory(_information.OutPutDirectory);
                 /*
                  * If no test driver files are available, you can run the following command to export all installed driver files.
                  *  (1) dism /online /export-driver /destination:"D:\packet\cache\"
@@ -32,14 +42,82 @@ public void Execute()
                  *  The following code example exports the specified driver to the specified directory.
                  *  pnputil /export-driver oemXX.inf D:\packet\cache
                  */
+                var path = Path.Combine(information.OutPutDirectory, driver.PublishedName);
                 var command = new StringBuilder("/c pnputil /export-driver ")
-                .Append(driver)
+                .Append(driver.PublishedName)
                 .Append(' ')
-                .Append(_information.OutPutDirectory)
+                .Append(path)
                 .ToString();
-
+                
+                if (!Directory.Exists(path))
+                {
+                    Directory.CreateDirectory(path);
+                }
+                
                 CommandExecutor.ExecuteCommand(command);
             }
         }
+
+        private IEnumerable<DriverInfo> GetInstalledDrivers(Dictionary<string, string> fieldMappings)
+        {
+            var drivers = new List<DriverInfo>();
+            var process = new Process();
+            process.StartInfo.FileName = "pnputil";
+            process.StartInfo.Arguments = "/enum-drivers";
+            process.StartInfo.RedirectStandardOutput = true;
+            process.StartInfo.UseShellExecute = false;
+            process.StartInfo.CreateNoWindow = true;
+            process.Start();
+
+            var output = process.StandardOutput.ReadToEnd();
+            process.WaitForExit();
+
+            var lines = output.Split(new[] { Environment.NewLine }, StringSplitOptions.RemoveEmptyEntries);
+        
+            DriverInfo currentDriver = null;
+            foreach (var line in lines)
+            {
+                if (line.StartsWith(fieldMappings["PublishedName"]))
+                {
+                    if (currentDriver != null)
+                    {
+                        drivers.Add(currentDriver);
+                    }
+                    currentDriver = new ();
+                    currentDriver.PublishedName = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["OriginalName"]) && currentDriver != null)
+                {
+                    currentDriver.OriginalName = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["Provider"]) && currentDriver != null)
+                {
+                    currentDriver.Provider = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["ClassName"]) && currentDriver != null)
+                {
+                    currentDriver.ClassName = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["ClassGUID"]) && currentDriver != null)
+                {
+                    currentDriver.ClassGUID = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["Version"]) && currentDriver != null)
+                {
+                    currentDriver.Version = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+                else if (line.StartsWith(fieldMappings["Signer"]) && currentDriver != null)
+                {
+                    currentDriver.Signer = line.Split(new[] { ':' }, 2)[1].Trim();
+                }
+            }
+
+            if (currentDriver != null)
+            {
+                drivers.Add(currentDriver);
+            }
+
+            return drivers;
+        }
     }
 }

src/c#/GeneralUpdate.Core/Driver/CommandExecutor.cs

@@ -8,6 +8,9 @@ namespace GeneralUpdate.Core.Driver
     /// </summary>
     public class CommandExecutor
     {
+        private CommandExecutor()
+        { }
+
         public static void ExecuteCommand(string command)
         {
             /*
@@ -28,7 +31,9 @@ Update the driver regularly to ensure that the driver is compatible with the cur
                 WindowStyle = ProcessWindowStyle.Hidden,
                 FileName = "cmd.exe",
                 Arguments = command,
-                UseShellExecute = true,
+                UseShellExecute = false,
+                RedirectStandardOutput = true,
+                RedirectStandardError = true,
                 Verb = "runas"
             };
 
@@ -39,6 +44,17 @@ Update the driver regularly to ensure that the driver is compatible with the cur
                 process.Start();
                 process.WaitForExit();
 
+                // 读取标准输出
+                var output = process.StandardOutput.ReadToEnd();
+                Debug.WriteLine(output);
+
+                // 读取错误输出
+                var error = process.StandardError.ReadToEnd();
+                if (!string.IsNullOrEmpty(error))
+                {
+                    Debug.WriteLine("Error: " + error);
+                }
+                
                 if (process.ExitCode != 0)
                     throw new ApplicationException($"Operation failed code: {process.ExitCode}");
             }

src/c#/GeneralUpdate.Core/Driver/DeleteDriverCommand.cs

@@ -1,23 +1,18 @@
 using System.Text;
 
-namespace GeneralUpdate.Core.Driver
+namespace GeneralUpdate.Core.Driver;
+
+public class DeleteDriverCommand(DriverInformation information) : DriverCommand
 {
-    public class DeleteDriverCommand : IDriverCommand
+    public override void Execute()
     {
-        private DriverInformation _information;
-
-        public DeleteDriverCommand(DriverInformation information) => _information = information;
-
-        public void Execute()
+        //Before installing the driver, delete the driver that has been installed on the local system. Otherwise, an exception may occur.
+        foreach (var driver in information.Drivers)
         {
-            //Before installing the driver, delete the driver that has been installed on the local system. Otherwise, an exception may occur.
-            foreach (var driver in _information.Drivers)
-            {
-                var command = new StringBuilder("/c pnputil /delete-driver ")
-                                  .Append(driver)
-                                  .ToString();
-                CommandExecutor.ExecuteCommand(command);
-            }
+            var command = new StringBuilder("/c pnputil /delete-driver ")
+                .Append(driver.PublishedName)
+                .ToString();
+            CommandExecutor.ExecuteCommand(command);
         }
     }
 }

src/c#/GeneralUpdate.Core/Driver/DriverCommand.cs

@@ -0,0 +1,24 @@
+using System.Collections;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using GeneralUpdate.Common;
+
+namespace GeneralUpdate.Core.Driver
+{
+    public abstract class DriverCommand
+    {
+        public abstract void Execute();
+
+        /// <summary>
+        /// Search for driver files.
+        /// </summary>
+        /// <param name="patchPath"></param>
+        /// <returns></returns>
+        protected static IEnumerable<FileInfo> SearchDrivers(string patchPath, string fileExtension)
+        {
+            var files = GeneralFileManager.GetAllfiles(patchPath);
+            return files.Where(x => x.FullName.EndsWith(fileExtension)).ToList();
+        }
+    }
+}

src/c#/GeneralUpdate.Core/Driver/DriverInfo.cs

@@ -0,0 +1,12 @@
+namespace GeneralUpdate.Core.Driver;
+
+public class DriverInfo
+{
+    public string PublishedName { get; set; }
+    public string OriginalName { get; set; }
+    public string Provider { get; set; }
+    public string ClassName { get; set; }
+    public string ClassGUID { get; set; }
+    public string Version { get; set; }
+    public string Signer { get; set; }
+}