?? internet.cpp
字號:
//=============================================================================================
/*
文件： Internet.cpp
說明:
---------------------------------------------------
網絡數據傳輸操作，用來完成用戶注冊信息的上載和
網絡命令的下載。
---------------------------------------------------
工程： Xfilter 個人防火墻
作者： 朱雁輝，朱雁冰
創建日期： 2001/08/26
網址： http://www.xfilt.com
電子郵件： xstudio@xfilt.com
版權所有 (c) 2001-2002 X 工作室
警告:
---------------------------------------------------
本電腦程序受著作權法的保護。未經授權，不能使用
和修改本軟件全部或部分源代碼。凡擅自復制、盜用或散
布此程序或部分程序或者有其它任何越權行為，將遭到民
事賠償及刑事的處罰，并將依法以最高刑罰進行追訴。
凡通過合法途徑購買本軟件源代碼的用戶被默認授權
可以在自己的程序中使用本軟件的部分代碼，但作者不對
代碼產生的任何后果負責。
使用了本軟件代碼的程序只能以可執行文件形式發布，
未經特別許可，不能將含有本軟件源代碼的源程序以任何
形式發布。
---------------------------------------------------
*/
//=============================================================================================
#include "stdafx.h"
#include "..\guires.h"
#include "..\Property.h" // in the header #include "internet.h"
#include "..\Register.h"
const TCHAR szHeaders[] =
_T("Accept: text/*\r\nUser-Agent: XFILTER\r\n");
extern CAclFile m_AclFile;
extern CPropertyApp theApp;
CHttpRequest::CHttpRequest()
{
m_pUrlRequest[0] = '\0';
m_DownloadThread = NULL;
m_UploadThread = NULL;
m_IsUploaded = TRUE;
lCommandId = NULL;
m_IsConnecting = FALSE;
}
CHttpRequest::~CHttpRequest()
{
Close();
}
int CHttpRequest::ConnectUrl(TCHAR *sUrl, TCHAR *sReturn, long *lVersion, int *Count)
{
ODS(_T("XFILTER.EXE: GetFromUrl Begin... "));
if(sUrl == NULL)
return XERR_INVALID_PARAMETER;
CString strServerName;
CString strObject;
INTERNET_PORT nPort;
DWORD dwServiceType;
if (!AfxParseURL(sUrl, dwServiceType, strServerName, strObject, nPort) ||
dwServiceType != INTERNET_SERVICE_HTTP)
{
ODS(_T("XFILTER.EXE: Internet Invalid Url ..."));
return XERR_INTERNET_URL_ERROR;
}
CInternetSession session(GUI_APP_CLASS_NAME);
CHttpConnection *pServer = NULL;
CHttpFile *pFile = NULL;
int iRet = XERR_SUCCESS;
m_IsConnecting = TRUE;
try
{
pServer = session.GetHttpConnection(strServerName, nPort);
pFile = pServer->OpenRequest(CHttpConnection::HTTP_VERB_GET, strObject);
pFile->AddRequestHeaders(szHeaders);
pFile->SendRequest();
DWORD dwRet;
pFile->QueryInfoStatusCode(dwRet);
if (dwRet >= 400 && dwRet <= 499)
{
ODS(_T("XFILTER.EXE: Internet Request Error ..."));
iRet = XERR_INTERNET_REQUEST_ERROR;
}
else if(dwRet >= 500 && dwRet <= 599)
{
ODS(_T("XFILTER.EXE: Internet Server Error ..."));
iRet = XERR_INTERNET_SERVER_ERROR;
}
else if(sReturn != NULL)
{
pFile->ReadString(sReturn, MAX_NET_COMMAND_LENTH - 1);
ODS(sReturn);
CString tmpStr = sReturn;
long lVer = atol(tmpStr.Left(MAX_NET_COMMAND_VERSION_LENTH));
if(lVer > *lVersion)
{
*lVersion = lVer;
int i = 1;
while (i < MAX_NET_COMMAND
&& pFile->ReadString((sReturn + MAX_NET_COMMAND_LENTH * i), MAX_NET_COMMAND_LENTH - 1))
{
ODS(sReturn + i * MAX_NET_COMMAND_LENTH);
i ++;
}
*Count = i;
}
}
else
{
CString sRet;
pFile->ReadString(sRet);
if(sRet.GetAt(0) != '1')
iRet = XERR_INTERNET_REG_ERROR;
ODS2(_T("XFILTER.EXE: Internet User Register Return Value "),sRet);
}
pFile->Close();
pServer->Close();
}
catch(CInternetException* pEx)
{
#if _DEBUG
TCHAR sError[1024];
pEx->GetErrorMessage(sError, 1024);
ODS(sError);
#endif
pEx->Delete();
iRet = XERR_INTERNET_CONNECT_ERROR;
}
if (pFile != NULL)
delete pFile;
if (pServer != NULL)
delete pServer;
session.Close();
m_IsConnecting = FALSE;
ODS(_T("XFILTER.EXE: GetFromUrl End... "));
return iRet;
}
void CHttpRequest::Close()
{
if(lCommandId != NULL)
{
delete lCommandId;
lCommandId = NULL;
}
if(m_UploadThread != NULL)
{
TerminateThread(m_UploadThread, 0);
m_UploadThread = NULL;
}
if(m_DownloadThread != NULL)
{
TerminateThread(m_DownloadThread, 0);
m_DownloadThread = NULL;
}
ODS(_T("XFILTER.EXE: Internet Close free lCommandId and exit thread..."));
}
BOOL CHttpRequest::IsConnected()
{
ODS(_T("XFILTER.EXE: Internet Check Connected..."));
RASCONN lpRasConn ;
RASCONNSTATUS rasStatus;
DWORD cbBuf = 0;
DWORD cConn = 0;
DWORD dwRet = 0;
cbBuf = sizeof(RASCONN);
lpRasConn.dwSize = sizeof(RASCONN );
dwRet = RasEnumConnections(&lpRasConn, &cbBuf, &cConn );
if ( dwRet != 0 )
return FALSE;
else
{
rasStatus.dwSize = sizeof(RASCONNSTATUS);
RasGetConnectStatus(lpRasConn.hrasconn,&rasStatus);
if (rasStatus.rasconnstate==RASCS_Connected)
return TRUE;
else
return FALSE;
}
return TRUE;
}
BOOL CHttpRequest::UserReg()
{
CRegister dlgReg;
int iRet = dlgReg.DoModal();
if(iRet == IDCANCEL)
return FALSE;
m_UserInfo.iStatus = REG_STATUS_REGISTERING;
return theApp.m_Install.SaveReg(REG_INFO_ITEM, (BYTE*)&m_UserInfo, sizeof(XUSER_INFO));
}
BOOL CHttpRequest::InitRegister()
{
if(m_AclFile.mAclHeader.sSignature[0] == '\0' || m_AclFile.mAclHeader.sWebURL[0] == '\0')
return FALSE;
theApp.m_Install.ReadReg(REG_INFO_ITEM, (BYTE*)&m_UserInfo, sizeof(XUSER_INFO));
if(m_UserInfo.iStatus == REG_STATUS_NO_REGISTER)
{
theApp.SetAutoStart();
if(!UserReg())
return FALSE;
m_IsUploaded = PreUpload();
}
else if(m_UserInfo.iStatus == REG_STATUS_REGISTERING
|| m_UserInfo.iStatus == REG_STATUS_INFO_CHANGED)
m_IsUploaded = PreUpload();
return TRUE;
}
BOOL CHttpRequest::PreUpload()
{
DWORD dwThreadId;
DWORD dwFlags;
if(!InternetGetConnectedState(&dwFlags, 0))
return FALSE;
SetRegisterUrl();
m_UploadThread = ::CreateThread(NULL, 0, UploadUserInfo, 0, 0, &dwThreadId);
return TRUE;
}
void CHttpRequest::SetRegisterUrl()
{
CString tmpStr;
tmpStr.Format(
_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")
, m_AclFile.mAclHeader.sUserRegisterURL
, m_UserInfo.sEmail
, m_UserInfo.sName
, m_UserInfo.bGender
, m_UserInfo.sQQ
, m_UserInfo.sInc
, m_UserInfo.tBirthday.Format("%Y-%m-%d")
, m_UserInfo.bDegree
, m_UserInfo.bMetier
, m_UserInfo.sDuty
, m_UserInfo.sZip
, m_UserInfo.bSalary
, m_UserInfo.sAddress
, m_AclFile.mAclHeader.uiSerial
, m_UserInfo.sPassword
);
_tcscpy(m_pUrlRequest, tmpStr);
ODS(tmpStr);
}
BOOL CHttpRequest::InitNetCommand()
{
if(m_AclFile.mAclHeader.sSignature[0] == '\0'
|| m_AclFile.mAclHeader.sCommandURL[0] == '\0')
return FALSE;
DWORD dwThreadId;
theApp.m_Install.ReadReg(REG_NET_COMMAND_HEADER_ITEM, (BYTE*)&m_CommandHeader, sizeof(XNET_COMMAND_HEADER));
if(m_AclFile.mAclHeader.bUpdateInterval > 100)
return TRUE;
if(m_AclFile.mAclHeader.bUpdateInterval == 0
|| m_CommandHeader.tCheckTime == 0
|| (CTime::GetCurrentTime() - m_CommandHeader.tCheckTime)
>= ACL_MAX_TIME * m_AclFile.mAclHeader.bUpdateInterval)
{
m_DownloadThread = ::CreateThread(NULL, 0, DownloadCommandFile, 0, 0, &dwThreadId);
}
return TRUE;
}
DWORD WINAPI DownloadCommandFile(LPVOID pVoid)
{
if(m_AclFile.mAclHeader.sCommandURL[0] == '\0')
return 0;
BOOL IsSuccess = FALSE;
BOOL IsAclChange = FALSE;
int iCount = 0;
int tmpCount = 0;
TCHAR sCommand[MAX_NET_COMMAND][MAX_NET_COMMAND_LENTH];
if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
{
theApp.m_DownloadCommand.lCommandId = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount];
theApp.m_Install.ReadReg(REG_NET_COMMAND_ITEM
, (BYTE*)theApp.m_DownloadCommand.lCommandId, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
}
while(!IsSuccess)
{
if(++iCount > 3)
break;
if(theApp.m_DownloadCommand.ConnectUrl(m_AclFile.mAclHeader.sCommandURL
, (TCHAR*)sCommand, &theApp.m_DownloadCommand.m_CommandHeader.lVersion,&tmpCount) != XERR_SUCCESS)
{
Sleep(180000);
continue;
}
if(tmpCount <= 1)
break;
CString tmpStr;
long tmpLong;
for(int i = 1; i < tmpCount; i++)
{
tmpStr = sCommand[i];
tmpLong = atol(tmpStr.Left(MAX_NET_COMMAND_VERSION_LENTH));
for(int j = 0; j < theApp.m_DownloadCommand.m_CommandHeader.lCount; j++)
{
if(tmpLong == theApp.m_DownloadCommand.lCommandId [j])
break;
}
if(j < theApp.m_DownloadCommand.m_CommandHeader.lCount)
continue;
long *pLong = NULL;
if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
{
pLong = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount];
memcpy(pLong, theApp.m_DownloadCommand.lCommandId, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
free(theApp.m_DownloadCommand.lCommandId);
}
theApp.m_DownloadCommand.lCommandId = new long[theApp.m_DownloadCommand.m_CommandHeader.lCount + 1];
if(theApp.m_DownloadCommand.m_CommandHeader.lCount > 0)
{
memcpy(theApp.m_DownloadCommand.lCommandId, pLong, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
free(pLong);
}
theApp.m_DownloadCommand.lCommandId[theApp.m_DownloadCommand.m_CommandHeader.lCount] = tmpLong;
theApp.m_DownloadCommand.m_CommandHeader.lCount ++;
int iCommand = atoi(tmpStr.Mid(MAX_NET_COMMAND_VERSION_LENTH + 1
, MAX_NET_COMMAND_COMMAND_LENTH));
CString sMessage= tmpStr.Mid(MAX_NET_COMMAND_VERSION_LENTH
+ MAX_NET_COMMAND_COMMAND_LENTH + 2);
sMessage.Replace(13, '\0');
sMessage.Replace(10, '\0');
if(sMessage.GetAt(0) == '\0')
break;
BYTE bInterval;
switch(iCommand)
{
case NET_COMMAND_CHANGE_WEB_STATION_URL:
if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sWebURL) == 0)
break;
_tcscpy(m_AclFile.mAclHeader.sWebURL, sMessage);
IsAclChange = TRUE;
break;
case NET_COMMAND_CHANGE_NET_COMMAND_URL:
if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sCommandURL) == 0)
break;
_tcscpy(m_AclFile.mAclHeader.sCommandURL, sMessage);
IsAclChange = TRUE;
break;
case NET_COMMAND_CHANGE_USER_REGISTER_URL:
if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sUserRegisterURL) == 0)
break;
_tcscpy(m_AclFile.mAclHeader.sUserRegisterURL, sMessage);
IsAclChange = TRUE;
break;
case NET_COMMAND_CHANGE_EMAIL_ADDRESS:
if(sMessage.CompareNoCase(m_AclFile.mAclHeader.sEmail) == 0)
break;
_tcscpy(m_AclFile.mAclHeader.sEmail, sMessage);
IsAclChange = TRUE;
break;
case NET_COMMAND_CHANGE_UPDATE_INTERVAL_DAYS:
bInterval = BYTE(atoi(sMessage));
if(bInterval== m_AclFile.mAclHeader.bUpdateInterval)
break;
m_AclFile.mAclHeader.bUpdateInterval = bInterval;
IsAclChange = TRUE;
break;
case NET_COMMAND_CHANGE_POST_MESSAGE:
_tcscpy(theApp.m_sMessage[i], sMessage);
::SendMessage(theApp.m_pMainWnd->m_hWnd, WM_NET_MESSAGE, i, NULL);
break;
default:
break;
}
}
theApp.m_DownloadCommand.m_CommandHeader.tCheckTime = CTime::GetCurrentTime();
theApp.m_Install.SaveReg(REG_NET_COMMAND_HEADER_ITEM,
(BYTE*)&theApp.m_DownloadCommand.m_CommandHeader, sizeof(XNET_COMMAND_HEADER));
theApp.m_Install.SaveReg(REG_NET_COMMAND_ITEM, (BYTE*)theApp.m_DownloadCommand.lCommandId
, sizeof(long) * theApp.m_DownloadCommand.m_CommandHeader.lCount);
if(IsAclChange)
m_AclFile.SaveAcl();
IsSuccess = TRUE;
break;
}
if(theApp.m_DownloadCommand.lCommandId != NULL)
{
delete theApp.m_DownloadCommand.lCommandId;
theApp.m_DownloadCommand.lCommandId = NULL;
}
return 0;
}
DWORD WINAPI UploadUserInfo(LPVOID pVoid)
{
if(theApp.m_RegisterRequest.m_pUrlRequest[0] == '\0')
return 0;
BOOL IsUploaded = FALSE;
BOOL IsSaved = FALSE;
int iCount = 0;
while(!IsUploaded || !IsSaved)
{
if(++iCount > 3)
break;
if(!IsUploaded && theApp.m_RegisterRequest.ConnectUrl(
theApp.m_RegisterRequest.m_pUrlRequest
) == XERR_SUCCESS)
{
theApp.m_RegisterRequest.m_UserInfo.iStatus = REG_STATUS_REGISTERED;
IsUploaded = TRUE;
}
if(theApp.m_Install.SaveReg(REG_INFO_ITEM, (BYTE*)&theApp.m_RegisterRequest.m_UserInfo, sizeof(XUSER_INFO)))
{
IsSaved = TRUE;
break;
}
Sleep(180000);
}
return 0;
}
?? 快捷鍵說明
復制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號
Ctrl + =
減小字號
Ctrl + -