【分享】本人写的一个基于C#的驱动管理类

JerryAJ

最近打算用C#做界面，突发奇想就弄了这个驱动管理类，原理和c++那些没什么不同，步骤还是老样子，只是有些地方做了修改
小生也是刚入门，技术有限O(∩_∩)O
目前在WIN7 x64和WIN10 1024 x64下测试通过

具体使用方法，我会上传一个项目文件，里面关于这个类的使用和驱动通信有一个实例

说明：里面的NativeApi和NativeApiEx 2个类是用于声明需要用到的API函数和相关常量


贴上类的主要代码：

1. using Microsoft.Win32.SafeHandles;
   
2. using System;
   
3. using System.IO;
   
4. using System.ServiceProcess;
   
5. /*
   
6.     .Net driver manager class template
   
7.     Version：1.0
   
8.     Autor：JerryAJ
   
9. */
   
10. namespace KernelBox
    
11. {
    
12.     public class DriverManager
    
13.     {
    
14.         // 驱动文件的名称
    
15.         public static string g_strSysName = "MainDriver.sys";
    
16.         // 驱动文件的路径
    
17.         public static string g_strSysPath = Directory.GetCurrentDirectory() + "\" + g_strSysName;
    
18.         // 驱动符号链接名称
    
19.         public static string g_strSysLinkName = "\\\\.\\AJ_Driver"; //格式：\\\\.\\xxoo
    
20.         // 驱动服务名称
    
21.         public static string g_strSysSvcLinkName = "AJ_Driver";
    
22.         // 驱动句柄
    
23.         public SafeFileHandle hDrv;
    
24.         // SCM句柄
    
25.         private IntPtr hSCManager;
    
26.         // 驱动服务句柄
    
27.         private IntPtr hService;
    
28.         
    
29. 
    
30.         // 获取驱动服务句柄
    
31.         public bool GetSvcHandle()
    
32.         {
    
33.             hSCManager = NativeApi.OpenSCManager(null, null, (uint)NativeApiEx.SCM_ACCESS.SC_MANAGER_ALL_ACCESS);
    
34.             if (IntPtr.Zero == hSCManager)
    
35.                 return false;
    
36.             hService = NativeApi.OpenService(hSCManager, g_strSysSvcLinkName, (uint)NativeApiEx.SERVICE_ACCESS.SERVICE_ALL_ACCESS);
    
37.             if (IntPtr.Zero == hService)
    
38.             {
    
39.                 NativeApi.CloseServiceHandle(hService);
    
40.                 return false;
    
41.             }
    
42.             return true;
    
43.         }
    
44. 
    
45.         // 安装驱动服务
    
46.         public bool Install()
    
47.         {
    
48.             hSCManager = NativeApi.OpenSCManager(null, null, (uint)NativeApiEx.SCM_ACCESS.SC_MANAGER_ALL_ACCESS);
    
49.             if (IntPtr.Zero == hSCManager)
    
50.                 return false;
    
51.             hService = NativeApi.CreateService(hSCManager, g_strSysSvcLinkName, g_strSysSvcLinkName,
    
52.                 (uint)NativeApiEx.SERVICE_ACCESS.SERVICE_ALL_ACCESS, (uint)NativeApiEx.SERVICE_TYPE.SERVICE_KERNEL_DRIVER,
    
53.                 (uint)NativeApiEx.SERVICE_START.SERVICE_DEMAND_START, (uint)NativeApiEx.SERVICE_ERROR.SERVICE_ERROR_NORMAL,
    
54.                 g_strSysPath, null, null, null, null, null);
    
55. 
    
56.             if (IntPtr.Zero == hService)
    
57.             {
    
58.                 NativeApi.GetLastError();
    
59.                 NativeApi.CloseServiceHandle(hService);
    
60.                 return false;
    
61.             }
    
62.             return true;
    
63.         }
    
64. 
    
65.         // 启动驱动服务
    
66.         public bool Start()
    
67.         {
    
68.             if (!NativeApi.StartService(hService, 0x0, null))
    
69.                 return false;
    
70.             return true;
    
71.         }
    
72. 
    
73.         // 停止驱动服务
    
74.         public bool Stop()
    
75.         {
    
76.             try
    
77.             {
    
78.                 ServiceController service = new ServiceController(g_strSysSvcLinkName);
    
79.                 if (service.Status == ServiceControllerStatus.Stopped)
    
80.                     return true;
    
81.                 else
    
82.                 {
    
83.                     TimeSpan timeout = TimeSpan.FromMilliseconds(1000 * 10);
    
84.                     service.Stop();
    
85.                     service.WaitForStatus(ServiceControllerStatus.Stopped, timeout);
    
86.                 }
    
87.             }
    
88.             catch (Exception)
    
89.             {
    
90.                 return false;
    
91.                 throw;
    
92.             }
    
93.             return true;
    
94.         }
    
95. 
    
96.         // 卸载驱动
    
97.         public bool Remove()
    
98.         {
    
99.             if (!NativeApi.DeleteService(hService))
    
100.                 return false;
     
101.             else
     
102.             {
     
103.                 NativeApi.CloseServiceHandle(hService);
     
104.                 NativeApi.CloseServiceHandle(hSCManager);
     
105.             }
     
106.             return true;
     
107.         }
     
108. 
     
109.         // 打开当前驱动
     
110.         public SafeFileHandle OpenDriver(string strLinkName)
     
111.         {
     
112.             return NativeApi.CreateFile(strLinkName, FileAccess.ReadWrite, FileShare.None, IntPtr.Zero, FileMode.Open, FileAttributes.Normal, IntPtr.Zero);
     
113.         }
     
114. 
     
115.         // 用于驱动通信的IO控制器
     
116.         public bool IoControl(SafeFileHandle hDriver, uint nIoCode, object InBuffer, uint nInBufferSize, object OutBuffer, uint nOutBufferSize, ref uint pBytesReturned, ref System.Threading.NativeOverlapped Overlapped)
     
117.         {
     
118.             const uint FILE_ANY_ACCESS = 0;
     
119.             const uint METHOD_BUFFERED = 0;
     
120.             bool bRet;
     
121.             nIoCode = ((int)NativeApiEx.DEVICE_TYPE.FILE_DEVICE_UNKNOWN * 65536) | (FILE_ANY_ACCESS * 16384) | (nIoCode * 4) | METHOD_BUFFERED;
     
122.             bRet = NativeApi.DeviceIoControl(hDriver, nIoCode, InBuffer, nInBufferSize, OutBuffer, nOutBufferSize, ref pBytesReturned, ref Overlapped);
     
123.             return bRet;
     
124.         }
     
125. 
     
126.         // 初始化
     
127.         public bool InitializeDriver()
     
128.         {
     
129.             bool bRet = false;
     
130.             hDrv = OpenDriver(g_strSysLinkName);
     
131.             NativeApi.GetLastError();
     
132.             if (!hDrv.IsInvalid)
     
133.                 bRet = GetSvcHandle();
     
134.             else
     
135.             {
     
136.                 bRet = Install();
     
137.                 if (!bRet)
     
138.                     return bRet;
     
139.                 else
     
140.                 {
     
141.                     bRet = Start();
     
142.                     hDrv = OpenDriver(g_strSysLinkName);
     
143.                 }
     
144.             }
     
145.             return bRet;
     
146.         }
     
147. 
     
148.         // 卸载驱动
     
149.         public bool RemoveDrvier()
     
150.         {
     
151.             hDrv.Close();
     
152.             hDrv.Dispose();
     
153.             hDrv = null;
     
154. 
     
155.             bool bRet = false;
     
156.             bRet = Stop();
     
157.             if (!bRet)
     
158.                 return bRet;
     
159. 
     
160.             bRet = Remove();
     
161. 
     
162.             return bRet;
     
163.         }
     
164. 
     
165.     }
     
166. }

复制代码

完毕！

Tesla.Angela

写的不错，加分支持。

JerryAJ

谢谢TA大大支持
努力学习，天天向上