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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 防火墙与安全工具 > 查看源码
XInstall.cpp
资源名称：FireWall-C++.rar　[点击查看]
上传用户：kklily621
上传日期：2013-06-25
资源大小：252k
文件大小：8k
源码类别：

防火墙与安全工具

开发平台：

Visual C++

XInstall.cpp：源码内容
  1. //=============================================================================================
  2. /*
  3. 文件： XInstall.cpp
  5. 说明:
  6. ---------------------------------------------------
  7. 用来安装 Xfilter 个人防火墙。
  8. ---------------------------------------------------
  10. 工程： Xfilter 个人防火墙
  11. 作者： 朱雁辉，朱雁冰
  12. 创建日期： 2001/08/28
  13. 网址： http://www.xfilt.com
  14. 电子邮件： xstudio@xfilt.com
  15. 版权所有 (c) 2001-2002 X 工作室
  17. 警告:
  18. ---------------------------------------------------
  19. 本电脑程序受著作权法的保护。未经授权，不能使用
  20. 和修改本软件全部或部分源代码。凡擅自复制、盗用或散
  21. 布此程序或部分程序或者有其它任何越权行为，将遭到民
  22. 事赔偿及刑事的处罚，并将依法以最高刑罚进行追诉。
  24. 凡通过合法途径购买本软件源代码的用户被默认授权
  25. 可以在自己的程序中使用本软件的部分代码，但作者不对
  26. 代码产生的任何后果负责。
  28. 使用了本软件代码的程序只能以可执行文件形式发布，
  29. 未经特别许可，不能将含有本软件源代码的源程序以任何
  30. 形式发布。
  31. ---------------------------------------------------
  32. */
  33. //=============================================================================================
  35. #include "stdafx.h"
  36. #include "XInstall.h"
  38. BOOL CXInstall::IsWinsock2()
  39. {
  40. WORD wVersionRequested = MAKEWORD(2, 0);
  41. WSADATA wsaData;
  43. if(WSAStartup(wVersionRequested, &wsaData) != 0)
  44. return FALSE;
  45.  
  46. if (LOBYTE(wsaData.wVersion) != 2)
  47. {
  48. WSACleanup();
  49. return FALSE; 
  50. }
  52. return TRUE;
  53. }
  55. BOOL CXInstall::IsInstalled(TCHAR *sPathName)
  56. {
  57. TCHAR tsPathName[MAX_PATH];
  59. if( ReadReg(REG_INSTALL_PATH_ITEM, 
  60. (BYTE*)tsPathName, 
  61. MAX_PATH, 
  62. HKEY_LOCAL_MACHINE, 
  63. REG_INSTALL_KEY, REG_SZ
  64. )
  65. )
  66. {
  67. if(sPathName != NULL)
  68. _tcscpy(sPathName, tsPathName);
  69. return TRUE;
  70. }
  72. return FALSE;
  73. }
  75. int CXInstall::InstallProvider(TCHAR *sPathName)
  76. {
  77. if(IsInstalled())
  78. return XERR_PROVIDER_ALREADY_INSTALL;
  80. _tcscpy(m_sPathName, sPathName);
  82. int iRet;
  83. if((iRet = EnumHookKey()) != XERR_SUCCESS)
  84. return iRet;
  86. if(!SaveReg(
  87. REG_INSTALL_PATH_ITEM, 
  88. (BYTE*)sPathName, 
  89. _tcslen(sPathName),
  90. HKEY_LOCAL_MACHINE, 
  91. REG_INSTALL_KEY, 
  92. REG_SZ
  93. )
  94. )
  95. return XERR_PROVIDER_SAVE_PATH_FAILED;
  97. return XERR_SUCCESS;
  98. }
  100. BOOL CXInstall::RemoveProvider()
  101. {
  102. int iRet = XERR_SUCCESS;
  104. if(!IsInstalled())
  105. return XERR_PROVIDER_NOT_INSTALL;
  107. if(iRet = EnumHookKey(TRUE) != XERR_SUCCESS)
  108. return iRet;
  110. if(!DeleteReg())
  111. return XERR_PROVIDER_REG_DELETE_FAILED;
  113. return XERR_SUCCESS;
  114. }
  116. //=============================================================================================
  117. // private install function
  119. int CXInstall::EnumHookKey(BOOL IsRemove)
  120. {
  121. HKEY hkey = NULL;
  123. if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, REG_PROTOCOL_CATALOG_KEY, 0, KEY_READ, &hkey) != ERROR_SUCCESS)
  124. return XERR_PROVIDER_OPEN_REG_FAILED;
  126. __try
  127. {
  128. TCHAR sSubKey[MAX_PATH];
  129. DWORD dwIndex = 0;
  130. int   iRet = 0;
  132. while(RegEnumKey(hkey, dwIndex, sSubKey, MAX_PATH) == ERROR_SUCCESS)
  133. {
  134. if((iRet = SaveHookKey(hkey, sSubKey, IsRemove)) != XERR_SUCCESS)
  135. return iRet;
  137. dwIndex ++;
  138. }
  139. }
  140. __finally
  141. {
  142. RegCloseKey(hkey);
  143. }
  145. return XERR_SUCCESS;
  146. }
  148. int CXInstall::SaveHookKey(HKEY hkey, LPCTSTR sSubKey, BOOL IsRemove)
  149. {
  150. HKEY hSubKey = NULL;
  151. BYTE ItemValue   [MAX_PROTOCOL_CATALOG_LENTH];
  152. DWORD ItemSize = MAX_PROTOCOL_CATALOG_LENTH;
  154. if(RegOpenKeyEx(hkey, sSubKey, 0, KEY_ALL_ACCESS, &hSubKey) != ERROR_SUCCESS)
  155. return XERR_PROVIDER_OPEN_REG_FAILED;
  157. __try
  158. {
  159. if(RegQueryValueEx(hSubKey, REG_PROTOCOL_CATALOG_ITEM, 0, NULL, ItemValue, &ItemSize) != ERROR_SUCCESS
  160. || (ItemSize != MAX_PROTOCOL_CATALOG_LENTH))
  161. return XERR_PROVIDER_READ_VALUE_FAILED;
  163. WSAPROTOCOL_INFOW *mProtocolInfo = (WSAPROTOCOL_INFOW*)(ItemValue + MAX_PATH);
  165. #if _DEBUG
  166. TCHAR sProtocolInfo[1024];
  167. _stprintf(sProtocolInfo
  168. , _T("dwServiceFlags1: %u;tdwProviderFlags: %u;tProviderId: %X-%X-%X-%X;t
  169. dwCatalogEntryId: %u;tProtocolChain->ChainLen: %i;tiVersion: %i;t
  170. iAddressFamily: %i;tiMaxSockAddr: %i;tiMinSockAddr: %i;tiSocketType: %i;t
  171. iProtocol: %i;tiProtocolMaxOffset: %i;tiNetworkByteOrder: %i;tiSecurityScheme: %i;t
  172. dwMessageSize: %u;tdwProviderReserved: %u;tszProtocol: %s; ") 
  173. , mProtocolInfo->dwServiceFlags1, mProtocolInfo->dwProviderFlags 
  174. , mProtocolInfo->ProviderId.Data1, mProtocolInfo->ProviderId.Data2 
  175. , mProtocolInfo->ProviderId.Data3, mProtocolInfo->ProviderId.Data4 
  176. , mProtocolInfo->dwCatalogEntryId, mProtocolInfo->ProtocolChain.ChainLen 
  177. , mProtocolInfo->iVersion, mProtocolInfo->iAddressFamily
  178. , mProtocolInfo->iMaxSockAddr, mProtocolInfo->iMinSockAddr
  179. , mProtocolInfo->iSocketType, mProtocolInfo->iProtocol
  180. , mProtocolInfo->iProtocolMaxOffset, mProtocolInfo->iNetworkByteOrder
  181. , mProtocolInfo->iSecurityScheme, mProtocolInfo->dwMessageSize
  182. , mProtocolInfo->dwProviderReserved, mProtocolInfo->szProtocol
  183. );
  184. ODS(sProtocolInfo);
  185. #endif
  187. if(mProtocolInfo->ProtocolChain.ChainLen == 1 && mProtocolInfo->iMaxSockAddr == 16 && mProtocolInfo->iMinSockAddr == 16)
  188. {
  189. TCHAR sItem[21];
  190. _stprintf(sItem, _T("%u"), mProtocolInfo->dwCatalogEntryId);
  191. if(!IsRemove)
  192. {
  193. if(!SaveReg(
  194. sItem, 
  195. ItemValue,
  196. _tcslen((TCHAR*)ItemValue), 
  197. HKEY_LOCAL_MACHINE, 
  198. REG_INSTALL_KEY, 
  199. REG_SZ
  200. )
  201. )
  202. return XERR_PROVIDER_CREATE_ITEM_FAILED;
  204. _tcscpy((TCHAR*)ItemValue, m_sPathName);
  206. if(RegSetValueEx(hSubKey, REG_PROTOCOL_CATALOG_ITEM, 0, REG_BINARY, ItemValue, ItemSize) != ERROR_SUCCESS)
  207. return XERR_PROVIDER_SET_VALUE_FAILED;
  208. }
  209. else
  210. {
  211. TCHAR sProvider[MAX_PATH];
  213. int iRet = ReadReg(
  214. sItem, 
  215. (BYTE*)sProvider, 
  216. MAX_PATH, 
  217. HKEY_LOCAL_MACHINE, 
  218. REG_INSTALL_KEY, REG_SZ
  219. );
  220. #if _DEBUG
  221. if(!iRet)
  222. return XERR_PROVIDER_READ_VALUE_FAILED;
  223. #endif
  224. _tcscpy((TCHAR*)ItemValue, sProvider);
  226. iRet = RegSetValueEx(hSubKey, REG_PROTOCOL_CATALOG_ITEM, 0, REG_BINARY, ItemValue, ItemSize);
  227. #if _DEBUG
  228. if(iRet != ERROR_SUCCESS)
  229. return XERR_PROVIDER_SET_VALUE_FAILED;
  230. #endif
  231. }
  232. }
  233. }
  234. __finally
  235. {
  236. RegCloseKey(hSubKey);
  237. }
  239. return XERR_SUCCESS;
  240. }
  242. //=============================================================================================
  243. // registry operator function
  245. BOOL CXInstall::ReadReg(
  246. TCHAR *sKey, 
  247. BYTE *pBuffer,
  248. DWORD dwBufSize,
  249. HKEY hkey, 
  250. TCHAR *sSubKey, 
  251. DWORD ulType
  252. )
  253. {
  254. HKEY hSubkey;
  256. if(RegOpenKeyEx(hkey, sSubKey, 0, KEY_ALL_ACCESS, &hSubkey) != ERROR_SUCCESS)
  257. return FALSE;
  259. __try
  260. {
  261. DWORD dwType;
  263. if (RegQueryValueEx(hSubkey, sKey, 0, &dwType, pBuffer, &dwBufSize) == ERROR_SUCCESS
  264. && dwType == ulType)
  265. return TRUE;
  266. }
  267. __finally
  268. {
  269. RegCloseKey(hSubkey);
  270. }
  272. return FALSE;
  273. }
  275. BOOL CXInstall::SaveReg(
  276. TCHAR *sKey, 
  277. BYTE *pBuffer,
  278. DWORD dwBufSize,
  279. HKEY hkey, 
  280. TCHAR *sSubKey, 
  281. DWORD ulType
  282. )
  283. {
  284. HKEY hSubkey;
  285. DWORD dwDisposition;
  287. if (RegCreateKeyEx(hkey, sSubKey, 0, NULL, REG_OPTION_NON_VOLATILE
  288. , KEY_ALL_ACCESS, NULL, &hSubkey, &dwDisposition) != ERROR_SUCCESS)
  289. return FALSE;
  291. if (RegSetValueEx(hSubkey, sKey, 0, ulType, pBuffer, dwBufSize) != ERROR_SUCCESS)
  292. {
  293. RegCloseKey(hSubkey);
  294. return FALSE;
  295. }
  297. RegCloseKey(hSubkey);
  299. return TRUE;
  300. }
  302. BOOL CXInstall::DeleteReg(
  303. HKEY hkey,
  304. TCHAR *sSubKey, 
  305. TCHAR *sItem
  306. )
  307. {
  308. if(hkey == NULL || sSubKey == NULL)
  309. return FALSE;
  311. if(sItem == NULL)
  312. {
  313. if(RegDeleteKey(hkey,sSubKey) == ERROR_SUCCESS)
  314. return TRUE;
  315. else
  316. return FALSE;
  317. }
  319. HKEY hSubKey;
  321. if(RegOpenKeyEx(hkey, sSubKey, 0, KEY_ALL_ACCESS, &hSubKey) != ERROR_SUCCESS)
  322. return FALSE;
  324. __try
  325. {
  326. if(RegDeleteValue(hSubKey, sItem) == ERROR_SUCCESS)
  327. return TRUE;
  328. }
  329. __finally
  330. {
  331. RegCloseKey(hSubKey);
  332. }
  334. return FALSE;
  335. }