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

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

防火墙与安全工具

开发平台：

Visual C++

Provider.cpp：源码内容
  1. // Provider.cpp
  4. // 声明要使用UNICODE字符串
  5. #define UNICODE
  6. #define _UNICODE
  8. #include "stdafx.h"
  10. #include <Ws2spi.h>
  11. #include <Sporder.h>
  12. #include "Provider.h"
  13. #pragma comment(lib, "Ws2_32.lib")
  14. #pragma comment(lib, "sporder.lib")
  16. // 要安装的LSP的硬编码，在移除的时候还要使用它
  17. GUID  ProviderGuid = {0xd3c21122, 0x85e1, 0x48f3, {0x9a,0xb6,0x23,0xd9,0x0c,0x73,0x07,0xef}};
  19. LPWSAPROTOCOL_INFOW GetProvider(LPINT lpnTotalProtocols)
  20. {
  21. DWORD dwSize = 0;
  22. int nError;
  23. LPWSAPROTOCOL_INFOW pProtoInfo = NULL;
  25. // 取得需要的长度
  26. if(::WSCEnumProtocols(NULL, pProtoInfo, &dwSize, &nError) == SOCKET_ERROR)
  27. {
  28. if(nError != WSAENOBUFS)
  29. return NULL;
  30. }
  32. pProtoInfo = (LPWSAPROTOCOL_INFOW)::GlobalAlloc(GPTR, dwSize);
  33. *lpnTotalProtocols = ::WSCEnumProtocols(NULL, pProtoInfo, &dwSize, &nError);
  34. return pProtoInfo;
  35. }
  37. void FreeProvider(LPWSAPROTOCOL_INFOW pProtoInfo)
  38. {
  39. ::GlobalFree(pProtoInfo);
  40. }
  44. BOOL InstallProvider(WCHAR *pwszPathName)
  45. {
  46. WCHAR wszLSPName[] = L"MyLsp";
  47. LPWSAPROTOCOL_INFOW pProtoInfo;
  48. int nProtocols;
  49. WSAPROTOCOL_INFOW OriginalProtocolInfo[3];
  50. DWORD  dwOrigCatalogId[3];
  51. int nArrayCount = 0;
  53. DWORD dwLayeredCatalogId; // 我们分层协议的目录ID号
  55. int nError;
  57. // 找到我们的下层协议，将信息放入数组中
  58. // 枚举所有服务程序提供者
  59. pProtoInfo = GetProvider(&nProtocols);
  60. BOOL bFindUdp = FALSE;
  61. BOOL bFindTcp = FALSE;
  62. BOOL bFindRaw = FALSE;
  63. for(int i=0; i<nProtocols; i++)
  64. {
  65. if(pProtoInfo[i].iAddressFamily == AF_INET)
  66. {
  67. if(!bFindUdp && pProtoInfo[i].iProtocol == IPPROTO_UDP)
  68. {
  69. memcpy(&OriginalProtocolInfo[nArrayCount], &pProtoInfo[i], sizeof(WSAPROTOCOL_INFOW));
  70. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 = 
  71. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 & (~XP1_IFS_HANDLES); 
  73. dwOrigCatalogId[nArrayCount++] = pProtoInfo[i].dwCatalogEntryId;
  75. bFindUdp = TRUE;
  76. }
  78. if(!bFindTcp && pProtoInfo[i].iProtocol == IPPROTO_TCP)
  79. {
  80. memcpy(&OriginalProtocolInfo[nArrayCount], &pProtoInfo[i], sizeof(WSAPROTOCOL_INFOW));
  81. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 = 
  82. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 & (~XP1_IFS_HANDLES); 
  84. dwOrigCatalogId[nArrayCount++] = pProtoInfo[i].dwCatalogEntryId;
  86. bFindTcp = TRUE;
  87. } 
  88. if(!bFindRaw && pProtoInfo[i].iProtocol == IPPROTO_IP)
  89. {
  90. memcpy(&OriginalProtocolInfo[nArrayCount], &pProtoInfo[i], sizeof(WSAPROTOCOL_INFOW));
  91. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 = 
  92. OriginalProtocolInfo[nArrayCount].dwServiceFlags1 & (~XP1_IFS_HANDLES); 
  94. dwOrigCatalogId[nArrayCount++] = pProtoInfo[i].dwCatalogEntryId;
  96. bFindRaw = TRUE;
  97. }
  98. }
  99. }  
  101. // 安装我们的分层协议，获取一个dwLayeredCatalogId
  102. // 随便找一个下层协议的结构复制过来即可
  103. WSAPROTOCOL_INFOW LayeredProtocolInfo;
  104. memcpy(&LayeredProtocolInfo, &OriginalProtocolInfo[0], sizeof(WSAPROTOCOL_INFOW));
  105. // 修改协议名称，类型，设置PFL_HIDDEN标志
  106. wcscpy(LayeredProtocolInfo.szProtocol, wszLSPName);
  107. LayeredProtocolInfo.ProtocolChain.ChainLen = LAYERED_PROTOCOL; // 0;
  108. LayeredProtocolInfo.dwProviderFlags |= PFL_HIDDEN;
  109. // 安装
  110. if(::WSCInstallProvider(&ProviderGuid, 
  111. pwszPathName, &LayeredProtocolInfo, 1, &nError) == SOCKET_ERROR)
  112. {
  113. return FALSE;
  114. }
  115. // 重新枚举协议，获取分层协议的目录ID号
  116. FreeProvider(pProtoInfo);
  117. pProtoInfo = GetProvider(&nProtocols);
  118. for(i=0; i<nProtocols; i++)
  119. {
  120. if(memcmp(&pProtoInfo[i].ProviderId, &ProviderGuid, sizeof(ProviderGuid)) == 0)
  121. {
  122. dwLayeredCatalogId = pProtoInfo[i].dwCatalogEntryId;
  123. break;
  124. }
  125. }
  127. // 安装协议链
  128. // 修改协议名称，类型
  129. WCHAR wszChainName[WSAPROTOCOL_LEN + 1];
  130. for(i=0; i<nArrayCount; i++)
  131. {
  132. swprintf(wszChainName, L"%ws", wszLSPName);
  133. wcscpy(OriginalProtocolInfo[i].szProtocol, wszChainName);
  134. if(OriginalProtocolInfo[i].ProtocolChain.ChainLen == 1)
  135. {
  136. OriginalProtocolInfo[i].ProtocolChain.ChainEntries[1] = dwOrigCatalogId[i];
  137. }
  138. else
  139. {
  140. for(int j = OriginalProtocolInfo[i].ProtocolChain.ChainLen; j>0; j--)
  141. {
  142. OriginalProtocolInfo[i].ProtocolChain.ChainEntries[j] 
  143. = OriginalProtocolInfo[i].ProtocolChain.ChainEntries[j-1];
  144. }
  145. }
  146. OriginalProtocolInfo[i].ProtocolChain.ChainLen ++;
  147. OriginalProtocolInfo[i].ProtocolChain.ChainEntries[0] = dwLayeredCatalogId;
  148. }
  149. // 获取一个Guid，安装之
  150. GUID ProviderChainGuid;
  151. if(::UuidCreate(&ProviderChainGuid) == RPC_S_OK)
  152. {
  153. if(::WSCInstallProvider(&ProviderChainGuid, 
  154. pwszPathName, OriginalProtocolInfo, nArrayCount, &nError) == SOCKET_ERROR)
  155. {
  156. return FALSE;
  157. }
  158. }
  159. else
  160. return FALSE;
  162. // 将我们的协议提前，重新排序Winsock目录
  163. // 重新枚举安装的协议
  164. FreeProvider(pProtoInfo);
  165. pProtoInfo = GetProvider(&nProtocols);
  167. DWORD dwIds[20];
  168. int nIndex = 0;
  169. // 添加我们的协议链
  170. for(i=0; i<nProtocols; i++)
  171. {
  172. if((pProtoInfo[i].ProtocolChain.ChainLen > 1) &&
  173. (pProtoInfo[i].ProtocolChain.ChainEntries[0] == dwLayeredCatalogId))
  174. dwIds[nIndex++] = pProtoInfo[i].dwCatalogEntryId;
  175. }
  176. // 添加其它协议
  177. for(i=0; i<nProtocols; i++)
  178. {
  179. if((pProtoInfo[i].ProtocolChain.ChainLen <= 1) ||
  180. (pProtoInfo[i].ProtocolChain.ChainEntries[0] != dwLayeredCatalogId))
  181. dwIds[nIndex++] = pProtoInfo[i].dwCatalogEntryId;
  182. }
  183. // 重新排序Winsock目录
  184. if(nError = ::WSCWriteProviderOrder(dwIds, nIndex) != ERROR_SUCCESS)
  185. {
  186. return FALSE;
  187. }
  188. FreeProvider(pProtoInfo);
  190. return TRUE;
  191. }
  193. BOOL RemoveProvider()
  194. {
  195. LPWSAPROTOCOL_INFOW pProtoInfo;
  196. int nProtocols;
  197. DWORD dwLayeredCatalogId;
  199. // 根据Guid取得分层协议的目录ID号
  200. pProtoInfo = GetProvider(&nProtocols);
  201. int nError;
  202. for(int i=0; i<nProtocols; i++)
  203. {
  204. if(memcmp(&ProviderGuid, &pProtoInfo[i].ProviderId, sizeof(ProviderGuid)) == 0)
  205. {
  206. dwLayeredCatalogId = pProtoInfo[i].dwCatalogEntryId;
  207. break;
  208. }
  209. }
  210. if(i < nProtocols)
  211. {
  212. // 移除协议链
  213. for(i=0; i<nProtocols; i++)
  214. {
  215. if((pProtoInfo[i].ProtocolChain.ChainLen > 1) &&
  216. (pProtoInfo[i].ProtocolChain.ChainEntries[0] == dwLayeredCatalogId))
  217. {
  218. ::WSCDeinstallProvider(&pProtoInfo[i].ProviderId, &nError);
  219. }
  220. }
  221. // 移除分层协议
  222. ::WSCDeinstallProvider(&ProviderGuid, &nError);
  223. }
  224. return TRUE;
  225. }
  227. BOOL IsProviderInstalled()
  228. {
  229. WCHAR wszLSPName[] = L"MyLsp";
  230. LPWSAPROTOCOL_INFOW pProtoInfo;
  231. int nProtocols;
  232. pProtoInfo=GetProvider(&nProtocols);
  234. for(int i=0;i<nProtocols;i++)
  235. {
  236. if (wcscmp(wszLSPName, pProtoInfo[i].szProtocol) == 0)
  237. {
  238. return TRUE;
  239. }
  240. }
  242. return FALSE;
  243. }
  254. /*///////////////////////////////////////////////////////////////////
  255. void RemoveAllLayeredEntries()
  256. {
  257.  BOOL    bLayer;
  258.     int     ErrorCode,
  259.             i;
  261. int TotalProtocols;
  263. LPWSAPROTOCOL_INFOW ProtocolInfo;
  265.     while (1)
  266.     {
  267.         bLayer = FALSE;
  268.         ProtocolInfo = GetProvider(&TotalProtocols);
  269.         if (!ProtocolInfo)
  270.         {
  271.             printf("Unable to enumerate Winsock catalog!n");
  272.             return;
  273.         }
  274.         for(i=0; i < TotalProtocols ;i++)
  275.         {
  276.             if (ProtocolInfo[i].ProtocolChain.ChainLen != BASE_PROTOCOL)
  277.             {
  278.                 bLayer = TRUE;
  279.                 printf("Removing '%S'n", ProtocolInfo[i].szProtocol);
  280.                 if (WSCDeinstallProvider(&ProtocolInfo[i].ProviderId, &ErrorCode) == SOCKET_ERROR)
  281.                 {
  282.                     printf("Failed to remove [%s]: Error %dn", ProtocolInfo[i].szProtocol, ErrorCode);
  283.                 }
  284.                 break;
  285.             }
  286.         }
  287.         FreeProvider(ProtocolInfo);
  288.         if (bLayer == FALSE)
  289.         {
  290.             break;
  291.         }
  293.     }
  294. }
  295. */