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

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

防火墙与安全工具

开发平台：

Visual C++

Internet.cpp：源码内容
  1. //=============================================================================================
  2. /*
  3. 文件： Internet.cpp
  5. 说明:
  6. ---------------------------------------------------
  7. 网络数据传输操作，用来完成用户注册信息的上载和
  8. 网络命令的下载。
  9. ---------------------------------------------------
  11. 工程： Xfilter 个人防火墙
  12. 作者： 朱雁辉，朱雁冰
  13. 创建日期： 2001/08/26
  14. 网址： http://www.xfilt.com
  15. 电子邮件： xstudio@xfilt.com
  16. 版权所有 (c) 2001-2002 X 工作室
  18. 警告:
  19. ---------------------------------------------------
  20. 本电脑程序受著作权法的保护。未经授权，不能使用
  21. 和修改本软件全部或部分源代码。凡擅自复制、盗用或散
  22. 布此程序或部分程序或者有其它任何越权行为，将遭到民
  23. 事赔偿及刑事的处罚，并将依法以最高刑罚进行追诉。
  25. 凡通过合法途径购买本软件源代码的用户被默认授权
  26. 可以在自己的程序中使用本软件的部分代码，但作者不对
  27. 代码产生的任何后果负责。
  29. 使用了本软件代码的程序只能以可执行文件形式发布，
  30. 未经特别许可，不能将含有本软件源代码的源程序以任何
  31. 形式发布。
  32. ---------------------------------------------------
  33. */
  34. //=============================================================================================
  36. #include "stdafx.h"
  38. #include "..guires.h"
  39. #include "..Property.h" // in the header #include "internet.h"
  40. #include "..Register.h"
  42. const TCHAR szHeaders[] =
  43. _T("Accept: text/*rnUser-Agent: XFILTERrn");
  45. extern CAclFile m_AclFile;
  46. extern CPropertyApp theApp;
  48. CHttpRequest::CHttpRequest()
  49. {
  50. m_pUrlRequest[0] = '';
  51. m_DownloadThread = NULL;
  52. m_UploadThread = NULL;
  53. m_IsUploaded = TRUE;
  54. lCommandId = NULL;
  55. m_IsConnecting = FALSE;
  56. }
  58. CHttpRequest::~CHttpRequest()
  59. {
  60. Close();
  61. }
  63. int CHttpRequest::ConnectUrl(TCHAR *sUrl, TCHAR *sReturn, long *lVersion, int *Count)
  64. {
  65. ODS(_T("XFILTER.EXE: GetFromUrl Begin... "));
  67. if(sUrl == NULL)
  68. return XERR_INVALID_PARAMETER;
  70. CString strServerName;
  71. CString strObject;
  72. INTERNET_PORT nPort;
  73. DWORD dwServiceType;
  75. if (!AfxParseURL(sUrl, dwServiceType, strServerName, strObject, nPort) ||
  76. dwServiceType != INTERNET_SERVICE_HTTP)
  77. {
  78. ODS(_T("XFILTER.EXE: Internet Invalid Url ..."));
  79. return XERR_INTERNET_URL_ERROR;
  80. }
  82. CInternetSession session(GUI_APP_CLASS_NAME);
  83. CHttpConnection *pServer = NULL;
  84. CHttpFile *pFile = NULL;
  85. int iRet = XERR_SUCCESS;
  87. m_IsConnecting = TRUE;
  89. try
  90. {
  91. pServer = session.GetHttpConnection(strServerName, nPort);
  92. pFile = pServer->OpenRequest(CHttpConnection::HTTP_VERB_GET, strObject);
  94. pFile->AddRequestHeaders(szHeaders);
  95. pFile->SendRequest();
  97. DWORD dwRet;
  98. pFile->QueryInfoStatusCode(dwRet);
  100. if (dwRet >= 400 && dwRet <= 499)
  101. {
  102. ODS(_T("XFILTER.EXE: Internet Request Error ..."));
  103. iRet = XERR_INTERNET_REQUEST_ERROR;
  104. }
  105. else if(dwRet >= 500 && dwRet <= 599)
  106. {
  107. ODS(_T("XFILTER.EXE: Internet Server Error ..."));
  108. iRet = XERR_INTERNET_SERVER_ERROR;
  109. }
  110. else if(sReturn != NULL)
  111. {
  112. pFile->ReadString(sReturn, MAX_NET_COMMAND_LENTH - 1);
  113. ODS(sReturn);
  115. CString tmpStr = sReturn;
  116. long lVer = atol(tmpStr.Left(MAX_NET_COMMAND_VERSION_LENTH));
  118. if(lVer > *lVersion)
  119. {
  120. *lVersion = lVer;
  121. int i = 1;
  123. while (i < MAX_NET_COMMAND
  124. && pFile->ReadString((sReturn + MAX_NET_COMMAND_LENTH * i), MAX_NET_COMMAND_LENTH - 1))
  125. {
  126. ODS(sReturn + i * MAX_NET_COMMAND_LENTH);
  127. i ++;
  128. }
  129. *Count = i;
  130. }
  131. }
  132. else
  133. {
  134. CString sRet;
  135. pFile->ReadString(sRet);
  136. if(sRet.GetAt(0) != '1')
  137. iRet = XERR_INTERNET_REG_ERROR;
  139. ODS2(_T("XFILTER.EXE: Internet User Register Return Value "),sRet);
  140. }
  142. pFile->Close();
  143. pServer->Close();
  144. }
  145. catch(CInternetException* pEx)
  146. {
  148. #if _DEBUG
  149. TCHAR sError[1024];
  150. pEx->GetErrorMessage(sError, 1024);
  151. ODS(sError);
  152. #endif
  153. pEx->Delete();
  154. iRet = XERR_INTERNET_CONNECT_ERROR;
  155. }
  157. if (pFile != NULL)
  158. delete pFile;
  159. if (pServer != NULL)
  160. delete pServer;
  161. session.Close();
  163. m_IsConnecting = FALSE;
  165. ODS(_T("XFILTER.EXE: GetFromUrl End... "));
  166. return iRet;
  167. }
  169. void CHttpRequest::Close()
  170. {
  171. if(lCommandId != NULL)
  172. {
  173. delete lCommandId;
  174. lCommandId = NULL;
  175. }
  177. if(m_UploadThread != NULL)
  178. {
  179. TerminateThread(m_UploadThread, 0);
  180. m_UploadThread = NULL;
  181. }
  183. if(m_DownloadThread != NULL)
  184. {
  185. TerminateThread(m_DownloadThread, 0);
  186. m_DownloadThread = NULL;
  187. }
  189. ODS(_T("XFILTER.EXE: Internet Close free lCommandId and exit thread..."));
  190. }
  192. BOOL CHttpRequest::IsConnected()
  193. {
  194. ODS(_T("XFILTER.EXE: Internet Check Connected..."));
  196.     RASCONN lpRasConn ;
  197. RASCONNSTATUS rasStatus;     
  198. DWORD cbBuf = 0;     
  199. DWORD cConn = 0;     
  200. DWORD dwRet = 0; 
  201. cbBuf = sizeof(RASCONN);
  202. lpRasConn.dwSize = sizeof(RASCONN );
  203. dwRet = RasEnumConnections(&lpRasConn, &cbBuf, &cConn );
  204. if ( dwRet != 0 )   
  205. return FALSE;
  206. else
  207. {
  208. rasStatus.dwSize = sizeof(RASCONNSTATUS);
  209. RasGetConnectStatus(lpRasConn.hrasconn,&rasStatus);
  210. if (rasStatus.rasconnstate==RASCS_Connected)
  211. return TRUE;
  212. else
  213. return FALSE;
  214. }
  215. return TRUE;
  216. }
  218. BOOL CHttpRequest::UserReg()
  219. {
  220. CRegister dlgReg;
  222. int iRet = dlgReg.DoModal();
  224. if(iRet == IDCANCEL)
  225. return FALSE;
  227. m_UserInfo.iStatus = REG_STATUS_REGISTERING;
  229. return theApp.m_Install.SaveReg(REG_INFO_ITEM, (BYTE*)&m_UserInfo, sizeof(XUSER_INFO));
  230. }
  232. BOOL CHttpRequest::InitRegister()
  233. {
  234. if(m_AclFile.mAclHeader.sSignature[0] == '' || m_AclFile.mAclHeader.sWebURL[0] == '')
  235. return FALSE;
  237. theApp.m_Install.ReadReg(REG_INFO_ITEM, (BYTE*)&m_UserInfo, sizeof(XUSER_INFO));
  239. if(m_UserInfo.iStatus == REG_STATUS_NO_REGISTER)
  240. {
  241. theApp.SetAutoStart();
  243. if(!UserReg())
  244. return FALSE;
  246. m_IsUploaded = PreUpload();
  247. }
  248. else if(m_UserInfo.iStatus == REG_STATUS_REGISTERING 
  249. || m_UserInfo.iStatus == REG_STATUS_INFO_CHANGED)
  250. m_IsUploaded = PreUpload();
  252. return TRUE;
  253. }
  255. BOOL CHttpRequest::PreUpload()
  256. {
  257. DWORD dwThreadId;
  258. DWORD dwFlags;
  260. if(!InternetGetConnectedState(&dwFlags, 0))
  261. return FALSE;
  263. SetRegisterUrl();
  264. m_UploadThread = ::CreateThread(NULL, 0, UploadUserInfo, 0, 0, &dwThreadId);
  266. return TRUE;
  267. }
  269. void CHttpRequest::SetRegisterUrl()
  270. {
  271. CString tmpStr;
  273. tmpStr.Format(
  274. _T("%s?sEmail=%s&sName=%s&bGender=%u&sQQ=%s&sInc=%s&sBirthday=%s&bDegree=%u&bMetier=%u&sDuty=%s&sZip=%s&bSalary=%u&sAddress=%s&iProductId=%u&sPassword=%s")
  275. , m_AclFile.mAclHeader.sUserRegisterURL
  276. , m_UserInfo.sEmail
  277. , m_UserInfo.sName
  278. , m_UserInfo.bGender 
  279. , m_UserInfo.sQQ 
  280. , m_UserInfo.sInc
  281. , m_UserInfo.tBirthday.Format("%Y-%m-%d")
  282. , m_UserInfo.bDegree 
  283. , m_UserInfo.bMetier
  284. , m_UserInfo.sDuty 
  285. , m_UserInfo.sZip  
  286. , m_UserInfo.bSalary 
  287. , m_UserInfo.sAddress
  288. , m_AclFile.mAclHeader.uiSerial
  289. , m_UserInfo.sPassword 
  290. );
  292. _tcscpy(m_pUrlRequest, tmpStr);
  294. ODS(tmpStr);
  295. }
  297. BOOL CHttpRequest::InitNetCommand()
  298. {
  299. if(m_AclFile.mAclHeader.sSignature[0] == '' 
  300. || m_AclFile.mAclHeader.sCommandURL[0] == '')
  301. return FALSE;
  303. DWORD dwThreadId;
  304. theApp.m_Install.ReadReg(REG_NET_COMMAND_HEADER_ITEM, (BYTE*)&m_CommandHeader, sizeof(XNET_COMMAND_HEADER));
  306. if(m_AclFile.mAclHeader.bUpdateInterval > 100)
  307. return TRUE;
  309. if(m_AclFile.mAclHeader.bUpdateInterval == 0 
  310. || m_CommandHeader.tCheckTime == 0
  311. || (CTime::GetCurrentTime() - m_CommandHeader.tCheckTime) 
  312. >= ACL_MAX_TIME * m_AclFile.mAclHeader.bUpdateInterval) 
  313. {
  314. m_DownloadThread = ::CreateThread(NULL, 0, DownloadCommandFile, 0, 0, &dwThreadId);
  315. }
  317. return TRUE;
  318. }
  320. DWORD WINAPI DownloadCommandFile(LPVOID pVoid)
  321. {
  322. if(m_AclFile.mAclHeader.sCommandURL[0] == '')
  323. return 0;
  325. BOOL IsSuccess = FALSE;
  326. BOOL IsAclChange = FALSE;
  327. int iCount = 0;
  328. int tmpCount = 0;
  329. TCHAR sCommand[MAX_NET_COMMAND][MAX_NET_COMMAND_LENTH];
  331. if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
  332. {
  333. theApp.m_DownloadCommand.lCommandId = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount];
  334. theApp.m_Install.ReadReg(REG_NET_COMMAND_ITEM
  335. , (BYTE*)theApp.m_DownloadCommand.lCommandId, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
  336. }
  338. while(!IsSuccess)
  339. {
  340. if(++iCount > 3)
  341. break;
  343. if(theApp.m_DownloadCommand.ConnectUrl(m_AclFile.mAclHeader.sCommandURL
  344. , (TCHAR*)sCommand, &theApp.m_DownloadCommand.m_CommandHeader.lVersion,&tmpCount) != XERR_SUCCESS)
  345. {
  346. Sleep(180000);
  347. continue;
  348. }
  350. if(tmpCount <= 1)
  351. break;
  353. CString tmpStr;
  354. long tmpLong;
  356. for(int i = 1; i < tmpCount; i++)
  357. {
  358. tmpStr = sCommand[i];
  359. tmpLong = atol(tmpStr.Left(MAX_NET_COMMAND_VERSION_LENTH));
  361. for(int j = 0; j < theApp.m_DownloadCommand.m_CommandHeader.lCount; j++)
  362. {
  363. if(tmpLong == theApp.m_DownloadCommand.lCommandId [j])
  364. break;
  365. }
  367. if(j < theApp.m_DownloadCommand.m_CommandHeader.lCount)
  368. continue;
  370. long *pLong = NULL;
  372. if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
  373. {
  374. pLong = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount];
  375. memcpy(pLong, theApp.m_DownloadCommand.lCommandId, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
  376. free(theApp.m_DownloadCommand.lCommandId);
  377. }
  379. theApp.m_DownloadCommand.lCommandId = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount + 1];
  381. if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
  382. {
  383. memcpy(theApp.m_DownloadCommand.lCommandId, pLong, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
  384. free(pLong);
  385. }
  387. theApp.m_DownloadCommand.lCommandId[theApp.m_DownloadCommand.m_CommandHeader.lCount] = tmpLong;
  388. theApp.m_DownloadCommand.m_CommandHeader.lCount ++;
  390. int iCommand = atoi(tmpStr.Mid(MAX_NET_COMMAND_VERSION_LENTH + 1
  391. , MAX_NET_COMMAND_COMMAND_LENTH));
  392. CString sMessage= tmpStr.Mid(MAX_NET_COMMAND_VERSION_LENTH 
  393. + MAX_NET_COMMAND_COMMAND_LENTH + 2);
  395. sMessage.Replace(13, '');
  396. sMessage.Replace(10, '');
  398. if(sMessage.GetAt(0) == '')
  399. break;
  401. BYTE bInterval;
  403. switch(iCommand)
  404. {
  405. case NET_COMMAND_CHANGE_WEB_STATION_URL:
  406. if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sWebURL) == 0)
  407. break;
  408. _tcscpy(m_AclFile.mAclHeader.sWebURL, sMessage);
  409. IsAclChange = TRUE;
  410. break;
  412. case NET_COMMAND_CHANGE_NET_COMMAND_URL:
  413. if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sCommandURL) == 0)
  414. break;
  415. _tcscpy(m_AclFile.mAclHeader.sCommandURL, sMessage);
  416. IsAclChange = TRUE;
  417. break;
  419. case NET_COMMAND_CHANGE_USER_REGISTER_URL:
  420. if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sUserRegisterURL) == 0)
  421. break;
  422. _tcscpy(m_AclFile.mAclHeader.sUserRegisterURL, sMessage);
  423. IsAclChange = TRUE;
  424. break;
  426. case NET_COMMAND_CHANGE_EMAIL_ADDRESS:
  427. if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sEmail) == 0)
  428. break;
  429. _tcscpy(m_AclFile.mAclHeader.sEmail, sMessage);
  430. IsAclChange = TRUE;
  431. break;
  433. case NET_COMMAND_CHANGE_UPDATE_INTERVAL_DAYS:
  434. bInterval = BYTE(atoi(sMessage));
  435. if(bInterval== m_AclFile.mAclHeader.bUpdateInterval)
  436. break;
  437. m_AclFile.mAclHeader.bUpdateInterval = bInterval;
  438. IsAclChange = TRUE;
  439. break;
  441. case NET_COMMAND_CHANGE_POST_MESSAGE:
  442. _tcscpy(theApp.m_sMessage[i], sMessage);
  443. ::SendMessage(theApp.m_pMainWnd->m_hWnd, WM_NET_MESSAGE, i, NULL);
  444. break;
  446. default:
  447. break;
  448. }
  449. }
  451. theApp.m_DownloadCommand.m_CommandHeader.tCheckTime = CTime::GetCurrentTime();
  453. theApp.m_Install.SaveReg(REG_NET_COMMAND_HEADER_ITEM, 
  454. (BYTE*)&theApp.m_DownloadCommand.m_CommandHeader, sizeof(XNET_COMMAND_HEADER));
  456. theApp.m_Install.SaveReg(REG_NET_COMMAND_ITEM, (BYTE*)theApp.m_DownloadCommand.lCommandId
  457. , sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
  459. if(IsAclChange)
  460. m_AclFile.SaveAcl();
  462. IsSuccess = TRUE;
  464. break;
  465. }
  467. if(theApp.m_DownloadCommand.lCommandId != NULL)
  468. {
  469. delete theApp.m_DownloadCommand.lCommandId;
  470. theApp.m_DownloadCommand.lCommandId = NULL;
  471. }
  473. return 0;
  474. }
  476. DWORD WINAPI UploadUserInfo(LPVOID pVoid)
  477. {
  478. if(theApp.m_RegisterRequest.m_pUrlRequest[0] == '')
  479. return 0;
  481. BOOL IsUploaded = FALSE;
  482. BOOL IsSaved = FALSE;
  483. int iCount = 0;
  485. while(!IsUploaded || !IsSaved)
  486. {
  487. if(++iCount > 3)
  488. break;
  490. if(!IsUploaded && theApp.m_RegisterRequest.ConnectUrl(
  491. theApp.m_RegisterRequest.m_pUrlRequest
  492. ) == XERR_SUCCESS)
  493. {
  494. theApp.m_RegisterRequest.m_UserInfo.iStatus = REG_STATUS_REGISTERED;
  495. IsUploaded = TRUE;
  496. }
  498. if(theApp.m_Install.SaveReg(REG_INFO_ITEM, (BYTE*)&theApp.m_RegisterRequest.m_UserInfo, sizeof(XUSER_INFO)))
  499. {
  500. IsSaved = TRUE;
  501. break;
  502. }
  504. Sleep(180000);
  505. }
  507. return 0;
  508. }