开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
PassThruUserIo.cpp
资源名称：MyFireWall.rar　[点击查看]
上传用户：zhuzhu0204
上传日期：2020-07-13
资源大小：13165k
文件大小：8k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

PassThruUserIo.cpp：源码内容
  1. //PassThruUserIo.cpp
  2. //提供有关PassThru与应用程序交互的支持函数
  4. #include "stdafx.h"
  6. #include <winioctl.h>
  7. #include <ntddndis.h>
  8. #include <stdio.h>
  9. #include <tchar.h>
  11. #include "PassThruUserIo.h"
  13. extern ARPRule *pARPRuleList;
  15. HANDLE PtOpenControlDevice()
  16. {
  17. // 打开到驱动程序所控制设备的句柄
  18. HANDLE hFile = ::CreateFile(
  19. _T("\\.\PassThru"),
  20. GENERIC_READ | GENERIC_WRITE,
  21. 0,
  22. NULL,
  23. OPEN_EXISTING,
  24. FILE_ATTRIBUTE_NORMAL,
  25. NULL);
  27. return hFile;
  28. }
  30. HANDLE PtOpenAdapter(PWSTR pszAdapterName)
  31. {
  32. // 打开控制设备对象句柄
  33. HANDLE hAdapter = PtOpenControlDevice();
  34. if(hAdapter == INVALID_HANDLE_VALUE)
  35. return INVALID_HANDLE_VALUE;
  38. // 确定适配器名称的长度
  39. int nBufferLength = wcslen((PWSTR)pszAdapterName) * sizeof(WCHAR);
  41. // 发送IOCTL_PTUSERIO_OPEN_ADAPTER控制代码，打开适配器上下文
  42. DWORD dwBytesReturn;
  43. BOOL bOK = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_OPEN_ADAPTER, 
  44. pszAdapterName, nBufferLength, NULL, 0, &dwBytesReturn, NULL);
  45. // 检查结果
  46. if(!bOK)
  47. {
  48. ::CloseHandle(hAdapter);
  49. return INVALID_HANDLE_VALUE;
  50. }
  51. return hAdapter;
  52. }
  54. BOOL PtAdapterRequest(HANDLE hAdapter, PPTUSERIO_OID_DATA pOidData, BOOL bQuery)
  55. {
  56. if(hAdapter == INVALID_HANDLE_VALUE)
  57. return FALSE;
  58. // 发送IOCTL
  59. DWORD dw;
  60. int bRet = ::DeviceIoControl(
  61. hAdapter, bQuery ? IOCTL_PTUSERIO_QUERY_OID : IOCTL_PTUSERIO_SET_OID,
  62. pOidData, sizeof(PTUSERIO_OID_DATA) -1 + pOidData->Length,
  63. pOidData, sizeof(PTUSERIO_OID_DATA) -1 + pOidData->Length, &dw, NULL);
  65. return bRet;
  66. }
  68. //
  70. // 设置工作模式
  71. BOOL PtSetWorkMode(int *pWorkMode)
  72. {
  73. BOOL bRet = TRUE;
  74. // 打开PassThruEx驱动的控制设备对象，枚举下层绑定
  75. HANDLE hControlDevice = PtOpenControlDevice();
  76. CIMAdapters adapters;
  77. if(!adapters.EnumAdapters(hControlDevice))
  78. {
  79. AfxMessageBox(L"枚举适配器失败！");
  80. return FALSE;
  81. }
  83. // 将工作模式设置到每个下层适配器
  84. HANDLE hAdapter;
  85. for(int i=0; i<adapters.m_nAdapters; i++)
  86. {
  87. // 打开下层适配器
  88. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  89. if(hAdapter != INVALID_HANDLE_VALUE)
  90. {
  91. ULONG nLen = sizeof(int);
  92. BOOL bRet = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_SET_WORKMODE, pWorkMode, nLen, NULL, 0, &nLen, NULL);
  93. ::CloseHandle(hAdapter);
  94. }
  95. else
  96. {
  97. bRet = FALSE;
  98. break;
  99. }
  100. }
  101. ::CloseHandle(hControlDevice);
  102. return bRet;
  103. }
  105. // 设Ping
  106. BOOL PtSetPing(int *pPing)
  107. {
  108. BOOL bRet = TRUE;
  109. // 打开PassThruEx驱动的控制设备对象，枚举下层绑定
  110. HANDLE hControlDevice = PtOpenControlDevice();
  111. CIMAdapters adapters;
  112. if(!adapters.EnumAdapters(hControlDevice))
  113. {
  114. AfxMessageBox(L"枚举适配器失败！");
  115. return FALSE;
  116. }
  118. // 将工作模式设置到每个下层适配器
  119. HANDLE hAdapter;
  120. for(int i=0; i<adapters.m_nAdapters; i++)
  121. {
  122. // 打开下层适配器
  123. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  124. if(hAdapter != INVALID_HANDLE_VALUE)
  125. {
  126. ULONG nLen = sizeof(int); // 需要设置的是两个BOOL值，一个表示Ping入，另一个表示Ping出
  127. BOOL bRet = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_SET_PING, pPing, nLen, NULL, 0, &nLen, NULL);
  128. ::CloseHandle(hAdapter);
  129. }
  130. else
  131. {
  132. bRet = FALSE;
  133. break;
  134. }
  135. }
  136. ::CloseHandle(hControlDevice);
  137. return bRet;
  138. }
  140. // 设置ARP过滤规则
  141. BOOL PtSetARPRules(BOOL bIsARPRuleOpened)
  142. {
  143. BOOL bRet = TRUE;
  144. // 打开PassThruEx驱动的控制设备对象，枚举下层绑定
  145. HANDLE hControlDevice = PtOpenControlDevice();
  146. CIMAdapters adapters;
  147. if(!adapters.EnumAdapters(hControlDevice))
  148. {
  149. AfxMessageBox(L"枚举适配器失败！");
  150. return FALSE;
  151. }
  153. // 将工作模式设置到每个下层适配器
  154. HANDLE hAdapter;
  155. for(int i=0; i<adapters.m_nAdapters; i++)
  156. {
  157. // 打开下层适配器
  158. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  159. if(hAdapter != INVALID_HANDLE_VALUE)
  160. {
  161. if (bIsARPRuleOpened == FALSE)
  162. {
  163. // 删除ARP过滤规则
  164. DWORD dwBytes;
  165. bRet = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_CLEAR_ARP_RULE, NULL, 0, NULL, 0, &dwBytes, NULL);
  166. }
  167. else
  168. {
  169. // 增加ARP过滤规则
  170. for (ARPRule *p=pARPRuleList; p!=NULL; p=p->pNext)
  171. {
  172. ULONG nLen = sizeof(ARPRule);
  173. bRet = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_SET_ARP_RULE, p, nLen, NULL, 0, &nLen, NULL);
  174. }
  175. }
  176. ::CloseHandle(hAdapter);
  177. }
  178. else
  179. {
  180. bRet = FALSE;
  181. break;
  182. }
  183. }
  184. ::CloseHandle(hControlDevice);
  185. return bRet;
  186. }
  188. // 查询网络活动状态
  189. BOOL PtQueryStatistics(HANDLE hAdapter, PPassthruStatistics pStats)
  190. {
  191. ULONG nStatsLen = sizeof(PassthruStatistics);
  193. BOOL bRet = ::DeviceIoControl(hAdapter, 
  194. IOCTL_PTUSERIO_QUERY_STATISTICS, NULL, 0, pStats, nStatsLen, &nStatsLen, NULL);
  196. return bRet;
  197. }
  199. // 重置统计数字
  200. BOOL PtResetStatistics(HANDLE hAdapter)
  201. {
  202. DWORD dwBytes;
  203. BOOL bRet = ::DeviceIoControl(hAdapter, 
  204. IOCTL_PTUSERIO_RESET_STATISTICS, NULL, 0, NULL, 0, &dwBytes, NULL);
  205. return bRet;
  206. }
  208. // 向适配器添加一个过滤规则
  209. BOOL PtAddFilter(HANDLE hAdapter, PPassthruFilter pFilter)
  210. {
  211. ULONG nFilterLen = sizeof(PassthruFilter);
  212. BOOL bRet = ::DeviceIoControl(hAdapter, IOCTL_PTUSERIO_ADD_FILTER, 
  213. pFilter, nFilterLen, NULL, 0, &nFilterLen, NULL);
  214. return bRet;
  215. }
  217. // 清除适配器上的过滤规则
  218. BOOL PtClearFilter(HANDLE hAdapter)
  219. {
  220. DWORD dwBytes;
  221. BOOL bRet = ::DeviceIoControl(hAdapter, 
  222. IOCTL_PTUSERIO_CLEAR_FILTER, NULL, 0, NULL, 0, &dwBytes, NULL);
  223. return bRet;
  224. }
  226. //////////////////////////////////////////////////////////
  228. BOOL CIMAdapters::EnumAdapters(HANDLE hControlDevice)
  229. {
  230. DWORD dwBufferLength = sizeof(m_buffer);
  231. BOOL bRet = ::DeviceIoControl(hControlDevice, IOCTL_PTUSERIO_ENUMERATE, 
  232. NULL, 0, m_buffer, dwBufferLength, &dwBufferLength, NULL);
  233. if(!bRet)
  234. {
  235. AfxMessageBox(L"::DeviceIoControl失败！");
  236. return FALSE;
  237. }
  239. // 保存适配器数量
  240. m_nAdapters = (ULONG)((ULONG*)m_buffer)[0];
  242. // 下面从m_buffer中获取适配器名称和符号连接名称
  243. // 指向设备名称
  244. WCHAR *pwsz = (WCHAR *)((ULONG *)m_buffer + 1);
  245. int i = 0;
  246. m_pwszVirtualName[i] = pwsz;
  247. while(*(pwsz++) != NULL)
  248. {
  249. while(*(pwsz++) != NULL)
  250. { ; }
  252. m_pwszAdapterName[i] = pwsz;
  254. while(*(pwsz++) != NULL)
  255. { ; }
  257. if(++i >= MAX_ADAPTERS)
  258. break;
  260. m_pwszVirtualName[i] = pwsz;
  261. }
  263. return TRUE;
  264. }
  266. // 设置IM驱动的过滤规则
  267. BOOL IMSetRules(PPassthruFilter pRules, int nRuleCount)
  268. {
  269. BOOL bRet = TRUE;
  270. // 打开PassThruEx驱动的控制设备对象，枚举下层绑定
  271. HANDLE hControlDevice = PtOpenControlDevice();
  272. CIMAdapters adapters;
  273. if(!adapters.EnumAdapters(hControlDevice))
  274. {
  275. AfxMessageBox(L"枚举适配器失败！");
  276. return FALSE;
  277. }
  279. // 将过滤规则设置到每个下层适配器
  280. HANDLE hAdapter;
  281. int i, j;
  282. for(i=0; i<adapters.m_nAdapters; i++)
  283. {
  284. // 打开下层适配器
  285. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  286. if(hAdapter != INVALID_HANDLE_VALUE)
  287. {
  288. for(j=0; j<nRuleCount; j++)
  289. {
  290. PassthruFilter rule = pRules[j];
  291. // 注意在这里转化字节顺序
  292. rule.sourcePort = htons(rule.sourcePort);
  293. rule.sourceIP = htonl(rule.sourceIP);
  294. rule.sourceMask = htonl(rule.sourceMask);
  296. rule.destinationPort = htons(rule.destinationPort);
  297. rule.destinationIP = htonl(rule.destinationIP);
  298. rule.destinationMask = htonl(rule.destinationMask);
  299. // 添加过滤规则
  300. if(!PtAddFilter(hAdapter, &rule))
  301. {
  302. bRet = FALSE;
  303. break;
  304. }
  305. }
  306. ::CloseHandle(hAdapter);
  307. }
  308. else
  309. {
  310. bRet = FALSE;
  311. break;
  312. }
  313. }
  314. ::CloseHandle(hControlDevice);
  315. return bRet;
  316. }
  318. // 清除IM驱动的过滤规则
  319. BOOL IMClearRules()
  320. {
  321. BOOL bRet = TRUE;
  322. // 打开PassThruEx驱动的控制设备对象，枚举下层绑定
  323. HANDLE hControlDevice = PtOpenControlDevice();
  324. CIMAdapters adapters;
  325. if(!adapters.EnumAdapters(hControlDevice))
  326. return FALSE;
  328. // 将过滤规则设置到每个下层适配器
  329. HANDLE hAdapter;
  330. for(int i=0; i<adapters.m_nAdapters; i++)
  331. {
  332. // 打开下层适配器
  333. hAdapter = PtOpenAdapter(adapters.m_pwszAdapterName[i]);
  334. if(hAdapter != INVALID_HANDLE_VALUE)
  335. {
  336. // 清除过滤规则
  337. PtClearFilter(hAdapter);
  338. ::CloseHandle(hAdapter);
  339. }
  340. else
  341. {
  342. bRet = FALSE;
  343. break;
  344. }
  345. }
  346. ::CloseHandle(hControlDevice);
  347. return bRet;
  348. }