把TA教程的SCM驱动加载代码改成了“面向过程”版

gfw

原版的代码比较啰嗦 于是抽空改写了份 现在看来舒爽多了 方便大家 不隐藏不卖B了 哈哈

1. #pragma comment(lib,"advapi32.lib") //Finishing by GFW@vbasm.com
   
2. 
   
3. SC_HANDLE InstallDriver(PCHAR m_pSysPath, PCHAR m_pServiceName)
   
4. {
   
5.         PCHAR m_pDisplayName = m_pServiceName;
   
6.         SC_HANDLE m_hSCManager = OpenSCManagerA(NULL,NULL,SC_MANAGER_ALL_ACCESS);
   
7.         if (NULL == m_hSCManager)
   
8.         {
   
9.                 return 0;
   
10.         }
    
11.         SC_HANDLE m_hService = CreateServiceA(m_hSCManager,
    
12.                                                                                         m_pServiceName,
    
13.                                                                                         m_pDisplayName,
    
14.                                                                                         SERVICE_ALL_ACCESS,
    
15.                                                                                         SERVICE_KERNEL_DRIVER,
    
16.                                                                                         SERVICE_DEMAND_START,
    
17.                                                                                         SERVICE_ERROR_NORMAL,
    
18.                                                                                         m_pSysPath,
    
19.                                                                                         NULL,NULL,NULL,NULL,NULL);
    
20.         if (NULL == m_hService)
    
21.         {
    
22.                 if (ERROR_SERVICE_EXISTS == GetLastError())
    
23.                 {
    
24.                         m_hService = OpenServiceA(m_hSCManager,m_pServiceName,SERVICE_ALL_ACCESS);
    
25.                 }
    
26.         }
    
27.         CloseServiceHandle(m_hSCManager);
    
28.         return m_hService;
    
29. }
    
30. 
    
31. BOOL StartDriver(SC_HANDLE m_hService)
    
32. {
    
33.         return StartServiceA(m_hService,NULL,NULL);
    
34. }
    
35. 
    
36. BOOL StopDriver(SC_HANDLE m_hService)
    
37. {
    
38.         SERVICE_STATUS ss = {0};
    
39.         return ControlService(m_hService,SERVICE_CONTROL_STOP,&ss);
    
40. }
    
41. 
    
42. BOOL RemoveDriver(SC_HANDLE m_hService)
    
43. {
    
44.         return DeleteService(m_hService);
    
45. }
    
46. 
    
47. HANDLE OpenDriver(PCHAR pLinkName)/* \\\\.\\test */
    
48. {
    
49.         HANDLE m_hDriver = CreateFileA(pLinkName, GENERIC_READ | GENERIC_WRITE, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);
    
50.         if(m_hDriver != INVALID_HANDLE_VALUE)
    
51.                 return m_hDriver;
    
52.         else
    
53.                 return 0;
    
54. }
    
55. 
    
56. BOOL CloseDriver(HANDLE m_hDriver)
    
57. {
    
58.         __try
    
59.         {
    
60.                 return CloseHandle(m_hDriver);
    
61.         }
    
62.         __except(1)
    
63.         {
    
64.                 return 0;
    
65.         }
    
66. }
    
67. 
    
68. _inline DWORD CTL_CODE_GEN(DWORD lngFunction)
    
69. {
    
70.         return (FILE_DEVICE_UNKNOWN * 65536) | (FILE_ANY_ACCESS * 16384) | (lngFunction * 4) | METHOD_BUFFERED;
    
71. }
    
72. 
    
73. BOOL ControlDriver(HANDLE m_hDriver, DWORD dwIoCode, PVOID InBuff, DWORD InBuffLen, PVOID OutBuff, DWORD OutBuffLen, DWORD *RealRetBytes)
    
74. {
    
75.         DWORD dw=0;
    
76.         BOOL b=DeviceIoControl(m_hDriver,CTL_CODE_GEN(dwIoCode),InBuff,InBuffLen,OutBuff,OutBuffLen,&dw,NULL);
    
77.         if(RealRetBytes)
    
78.         {
    
79.                 *RealRetBytes=dw;
    
80.         }
    
81.         return b;
    
82. }
    
83. 
    
84. void GetAppPath(char *szPathString)
    
85. {
    
86.         GetModuleFileNameA(0,szPathString,MAX_PATH);
    
87.         for(SIZE_T i=strlen(szPathString)-1;i>=0;i--)
    
88.         {
    
89.                 if(szPathString[i]=='\\')
    
90.                 {
    
91.                         szPathString[i+1]='\0';
    
92.                         break;
    
93.                 }
    
94.         }
    
95. }
    
96. 
    
97. SC_HANDLE GetServiceHandle(PCHAR m_pServiceName)
    
98. {
    
99.         SC_HANDLE m_hSCManager = OpenSCManagerA(NULL,NULL,SC_MANAGER_ALL_ACCESS);
    
100.         if (NULL == m_hSCManager)
     
101.         {
     
102.                 return 0;
     
103.         }
     
104.         SC_HANDLE m_hService = OpenServiceA(m_hSCManager,m_pServiceName,SERVICE_ALL_ACCESS);
     
105.         if (NULL == m_hService)
     
106.         {
     
107.                 CloseServiceHandle(m_hSCManager);
     
108.                 return 0;
     
109.         }
     
110.         CloseServiceHandle(m_hSCManager);
     
111.         return m_hService;
     
112. }

复制代码

如何使用？

1. void test()
   
2. {
   
3.         CHAR szDrvFile[MAX_PATH]={0};
   
4.         GetAppPath(szDrvFile);
   
5.         strcat(szDrvFile,"KrnlHW64.sys");
   
6.         SC_HANDLE hSc=InstallDriver(szDrvFile,"KrnlHW64");
   
7.         if(hSc)
   
8.         {
   
9.                 StartDriver(hSc);
   
10.                 HANDLE hDrv=OpenDriver("\\\\.\\KrnlHW64");
    
11.                 ControlDriver(hDrv,0x800,0,0,0,0,0);
    
12.                 CloseDriver(hDrv);
    
13.                 StopDriver(hSc);
    
14.                 RemoveDriver(hSc);
    
15.         }
    
16. }

复制代码

Tesla.Angela

嗯，不错。已将链接加入教程帖目录。此外最后应该加上：CloseServiceHandle(hSC)，防止句柄泄漏。

又把你的例子加上了点内容，『驱动本体』和『加载器』完全模板化了。WIN32+WIN64都可以用。

1. SC_HANDLE g_hSc;
   
2. HANDLE g_hDrv;
   
3. 
   
4. BOOLEAN DriverInit(BOOLEAN IsInit)
   
5. {
   
6.         if(IsInit)
   
7.         {
   
8.                 char szFileName[]="XXXX.sys";//TODO: CHANGE THIS VALUE
   
9.                 char szLinkName[]="\\\\.\\XXXX";//TODO: CHANGE THIS VALUE
   
10.                 CHAR szDrvFile[MAX_PATH]={0};
    
11.                 GetAppPath(szDrvFile);
    
12.                 strcat(szDrvFile,szFileName);
    
13.                 g_hSc=InstallDriver(szDrvFile,szFileName);
    
14.                 if(g_hSc)
    
15.                 {
    
16.                         StartDriver(g_hSc);
    
17.                         g_hDrv=OpenDriver(szLinkName);
    
18.                         if(g_hDrv)
    
19.                                 return 1;
    
20.                         else
    
21.                                 return 0;
    
22.                 }
    
23.                 return 0;
    
24.         }
    
25.         else
    
26.         {
    
27.                 BOOLEAN b;
    
28.                 b=CloseDriver(g_hDrv);if(!b) return 0;
    
29.                 b=StopDriver(g_hSc);if(!b) return 0;
    
30.                 b=RemoveDriver(g_hSc);if(!b) return 0;
    
31.                 b=CloseServiceHandle(g_hSc);if(!b) return 0;
    
32.                 return 1;
    
33.         }
    
34. }
    
35. 
    
36. void main()
    
37. {
    
38.         BOOLEAN b;
    
39.         b=DriverInit(1);if(b){puts("load driver OK!");}else{puts("load driver failed!");return;}
    
40.         system("pause");
    
41.         //ControlDriver(g_hDrv,0x800,0,0,0,0,0);
    
42.         b=DriverInit(0);if(b){puts("unload driver OK!");}else{puts("unload driver failed!");}
    
43.         system("pause");
    
44. }

复制代码