#include "StdAfx.h"
|
#include "MBThreadCheckServer.h"
|
#include "MBBaseSvrInfo.h"
|
|
BOOL EstablishConnection(CString strExterFServer, u_short nPort)
|
{
|
// Create a stream socket
|
SOCKET sd = socket(AF_INET, SOCK_STREAM, 0);
|
if (sd != INVALID_SOCKET)
|
{
|
struct sockaddr_in sockaddr;
|
LPSTR pDest ;
|
memset( &sockaddr, 0, sizeof(sockaddr));
|
if ( !CStrFileUtils::CStringToAnsi(strExterFServer, &pDest))
|
{
|
return FALSE;
|
}
|
memset(&sockaddr, 0, sizeof(sockaddr));
|
sockaddr.sin_addr.s_addr = inet_addr( pDest );
|
sockaddr.sin_family = AF_INET;
|
sockaddr.sin_port = htons(nPort);
|
if (::connect(sd, (struct sockaddr*)&sockaddr, sizeof(sockaddr_in)) == SOCKET_ERROR)
|
{
|
::closesocket(sd);
|
return FALSE;
|
}
|
::closesocket(sd);
|
return TRUE;
|
}
|
return FALSE;
|
}
|
|
CMBThreadCheckServer::CMBThreadCheckServer(void)
|
{
|
m_bStop = false;
|
m_asUserIP.RemoveAll();
|
m_pSvrInfo = NULL;
|
}
|
|
|
CMBThreadCheckServer::~CMBThreadCheckServer(void)
|
{
|
}
|
|
void CMBThreadCheckServer::SetLoginUserIPInfo( CStringArray asUserIP, CString strUserConnIP, CString strOrgServer)
|
{
|
asUserIP.RemoveAll();
|
asUserIP.Copy(asUserIP);
|
m_strUserConnIP = strUserConnIP;
|
m_strOrgServer= strOrgServer;
|
}
|
|
// ¿ªÊ¼Ïß³Ì
|
void CMBThreadCheckServer::StartThread()
|
{
|
if(IsRunning())
|
return;
|
BeginThread();
|
}
|
|
// ½áÊøÏß³Ì
|
void CMBThreadCheckServer::OverThread()
|
{
|
m_bStop = true;
|
|
if(!IsRunning())
|
return;
|
EndThread();
|
}
|
|
|
|
UINT CMBThreadCheckServer::Run()
|
{
|
try
|
{
|
if ( m_pSvrInfo )
|
{
|
CheckFileServerValid(m_pSvrInfo->m_strExterIP,_tstol(m_pSvrInfo->m_strExterPort),m_pSvrInfo->m_strInterIP,_tstol(m_pSvrInfo->m_strInterPort));
|
}
|
}
|
catch(...)
|
{
|
}
|
|
return 0;
|
}
|
BOOL CMBThreadCheckServer::CheckFileServerValid(CString strExterFServer, UINT nExterFSPort,CString strInterFServer, UINT nInterFSPort)
|
{
|
int nPos;
|
CString strOrgSubIp1 = _T("");
|
CString strOrgSubIp2= _T("");
|
CString strConnSubIp1= _T("");
|
CString strConnSubIp2= _T("");
|
CString strTempOrgSubIp;
|
CString strTempConnSubIp;
|
CString strUserIP;
|
BOOL bInExIPSame = FALSE;
|
for(int i=0; i< m_asUserIP.GetCount();i++)
|
{
|
strUserIP = m_asUserIP.GetAt(i);
|
if(strUserIP == m_strUserConnIP)
|
{
|
bInExIPSame = TRUE;
|
break;
|
}
|
}
|
if( bInExIPSame )
|
{
|
if( EstablishConnection(strInterFServer,nInterFSPort))
|
{
|
SetValidSvrInfo(strInterFServer, nInterFSPort );
|
}
|
else
|
{
|
SetValidSvrInfo(strExterFServer, nExterFSPort );
|
}
|
return TRUE;
|
}
|
nPos = m_strOrgServer.Find(_T("."));
|
if(nPos > 0 )
|
{
|
strOrgSubIp1 = m_strOrgServer.Left(nPos);
|
strTempOrgSubIp = m_strOrgServer.Mid(nPos + 1);
|
nPos = strTempOrgSubIp.Find(_T("."));
|
if(nPos > 0 )
|
strOrgSubIp2 = strTempOrgSubIp.Left(nPos);
|
}
|
nPos = m_strUserConnIP.Find(_T("."));
|
if(nPos > 0 )
|
{
|
strConnSubIp1 = m_strUserConnIP.Left(nPos);
|
strTempConnSubIp = m_strUserConnIP.Mid(nPos + 1);
|
nPos = strTempConnSubIp.Find(_T("."));
|
if(nPos > 0 )
|
strConnSubIp2 = strTempConnSubIp.Left(nPos);
|
}
|
if(strConnSubIp1 == strOrgSubIp1 && strConnSubIp2 == strOrgSubIp2 )
|
{
|
if( EstablishConnection(strInterFServer,nInterFSPort))
|
{
|
SetValidSvrInfo(strInterFServer, nInterFSPort );
|
}
|
else
|
{
|
SetValidSvrInfo(strExterFServer, nExterFSPort );
|
}
|
}
|
else
|
{
|
if( EstablishConnection(strExterFServer,nExterFSPort))
|
{
|
SetValidSvrInfo(strExterFServer, nExterFSPort );
|
}
|
else
|
{
|
SetValidSvrInfo(strInterFServer, nInterFSPort );
|
}
|
}
|
return TRUE;
|
}
|
void CMBThreadCheckServer::SetValidSvrInfo( CString strSvrIP, int nSvrPort)
|
{
|
if ( m_pSvrInfo )
|
{
|
CString strPort;
|
strPort.Format(_T("%d"),nSvrPort);
|
m_pSvrInfo->SetValidSvrInfo(strSvrIP,strPort);
|
}
|
}
|
