mysocket.cpp

扫描局域网

// MySocket.cpp : implementation file
//

#include "stdafx.h"
#include "MySocket.h"
#include <fstream>
#include <utility>

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif



/////////////////////////////////////////////////////////////////////////////
// MySocket

// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(MySocket, CAsyncSocket)
	//{{AFX_MSG_MAP(MySocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0


MySocket::MySocket()
{
	m_bAccept = FALSE;
}

MySocket::~MySocket()
{
	if(m_hSocket != INVALID_SOCKET)	Close();
}

void MySocket::OnReceive(int nErrorCode) 
{
	int m_nLength = Receive(m_szBuffer, 4096);
	m_szBuffer[m_nLength]='\0';
	m_MessStr = m_szBuffer;
	m_pWnd->SendMessage(m_UserMsg, MYNET_SERVERREAD);

//	CAsyncSocket::OnReceive(nErrorCode);
}

void MySocket::OnSend(int nErrorCode) 
{
	while(Send(m_szBuffer, strlen(m_szBuffer)) == SOCKET_ERROR);
	m_MessStr = "Message Sended !";
	m_pWnd->SendMessage(m_UserMsg, MYNET_SERVERSEND);
	//继续提请一个“读”的网络事件，接收Server消息
	AsyncSelect(FD_READ | FD_CLOSE);

//	CAsyncSocket::OnSend(nErrorCode);
}

BOOL MySocket::SendStr(CString m_str)
{
	if(m_bAccept)
	{
		strcpy(m_szBuffer, LPCTSTR(m_str));
		AsyncSelect(FD_WRITE);
		return TRUE;
	}

	return FALSE;
}

CString MySocket::GetMessStr(void)
{
	return m_MessStr;
}

void MySocket::SetMessagePara(CWnd* pParent,UINT m_nMessageID)
{
	m_UserMsg = m_nMessageID;
	m_pWnd = pParent;
}

void MySocket::OnClose(int nErrorCode) 
{
	m_pWnd->SendMessage(m_UserMsg, MYNET_CLIENTCLOSE);

	CAsyncSocket::OnClose(nErrorCode);
}




/////////////////////////////////////////////////////////////////////////////
// MyServerSocket

// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(MyServerSocket, CAsyncSocket)
	//{{AFX_MSG_MAP(MyServerSocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0


MyServerSocket::MyServerSocket()
{
	m_pSocket = new MySocket();
}

MyServerSocket::~MyServerSocket()
{
}

void MyServerSocket::OnAccept(int nErrorCode) 
{
	//侦听到连接请求，调用Accept函数
	MySocket* pSocket = new MySocket();
	if(Accept(*pSocket))
	{
		delete m_pSocket;

	    pSocket->AsyncSelect(FD_READ | FD_CLOSE);
		m_pSocket = pSocket;

		m_pSocket->SetMessagePara(m_pWnd,m_UserMsg);

		m_pSocket->m_bAccept = TRUE;
		m_pWnd->SendMessage(m_UserMsg, MYNET_ACCEPT);
	}
	else
	    delete pSocket;
	
//	CAsyncSocket::OnAccept(nErrorCode);
}

BOOL MyServerSocket::SendStr(CString m_str)
{
	return m_pSocket->SendStr(m_str);
}

CString MyServerSocket::GetMessStr(void)
{
	return m_pSocket->GetMessStr();
}

void MyServerSocket::SetMessagePara(CWnd* pParent,UINT m_nMessageID)
{
	m_UserMsg = m_nMessageID;
	m_pWnd = pParent;
	m_pSocket->SetMessagePara(m_pWnd,m_UserMsg);
}

void MyServerSocket::CloseServer(void) 
{
	m_pSocket->Close();
	delete m_pSocket;

	Close();
}



/////////////////////////////////////////////////////////////////////////////
// MyClientSocket

// Do not edit the following lines, which are needed by ClassWizard.
#if 0
BEGIN_MESSAGE_MAP(MyClientSocket, CAsyncSocket)
	//{{AFX_MSG_MAP(MyClientSocket)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
#endif	// 0


MyClientSocket::MyClientSocket()
{
	m_bConnected = FALSE;
	InitMap();
}

MyClientSocket::~MyClientSocket()
{
	//关闭套接字
	if(m_hSocket != INVALID_SOCKET)	Close();
}

void MyClientSocket::OnConnect(int nErrorCode) 
{
	if (nErrorCode == 0)
	{
	    m_bConnected = TRUE;
		m_pWnd->SendMessage(m_UserMsg, MYNET_CONNECT);

		//	提请一个“读”的网络事件，准备接收
		AsyncSelect(FD_READ  | FD_CLOSE);
	}

//	CAsyncSocket::OnConnect(nErrorCode);
}

void MyClientSocket::OnReceive(int nErrorCode) 
{
	int m_nLength = Receive(m_szBuffer, 4096);
	m_szBuffer[m_nLength]='\0';
	m_MessStr = m_szBuffer;
	m_pWnd->SendMessage(m_UserMsg, MYNET_CLIENTREAD);
	
//	CAsyncSocket::OnReceive(nErrorCode);
}

void MyClientSocket::OnSend(int nErrorCode) 
{
	while(Send(m_szBuffer, strlen(m_szBuffer)) == SOCKET_ERROR);
	m_MessStr = "Message Sended !";
	m_pWnd->SendMessage(m_UserMsg, MYNET_CLIENTSEND);
	//继续提请一个“读”的网络事件，接收Server消息
	AsyncSelect(FD_READ | FD_CLOSE);
	
//	CAsyncSocket::OnSend(nErrorCode);
}

BOOL MyClientSocket::IsConnect(void)
{
	return m_bConnected;
}

BOOL MyClientSocket::SendStr(CString m_str)
{
	if(m_bConnected)
	{
		strcpy(m_szBuffer, LPCTSTR(m_str));
		AsyncSelect(FD_WRITE);
		return TRUE;
	}

	return FALSE;
}

CString MyClientSocket::GetMessStr(void)
{
	return m_MessStr;
}

void MyClientSocket::Rest(void)
{
	m_hSocket = INVALID_SOCKET;
	m_bConnected = FALSE;
}

void MyClientSocket::SetMessagePara(CWnd* pParent,UINT m_nMessageID)
{
	m_UserMsg = m_nMessageID;
	m_pWnd = pParent;
}

void MyClientSocket::OnClose(int nErrorCode) 
{
	m_pWnd->SendMessage(m_UserMsg, MYNET_SERVERCLOSE);
	
	CAsyncSocket::OnClose(nErrorCode);
}

BOOL MyServerSocket::CreateAndListen(CWnd *pWnd, UINT messageID, UINT nSocketPort)
{
	SetMessagePara(pWnd, messageID);
	if(m_hSocket == INVALID_SOCKET)
	{
		BOOL bFlag = Create(nSocketPort, SOCK_STREAM, FD_ACCEPT);
		if(!bFlag)
		{
    		AfxMessageBox("网络建立错误 !");
		   	Close();
	    	return FALSE;
		}
	}
	//“侦听”成功，等待连接请求
	if(!Listen(1))
	{
	    int nErrorCode = GetLastError();
	    if(nErrorCode = WSAEWOULDBLOCK)
		{
	    	AfxMessageBox("网络侦听错误 !");
	    	Close();
	    	return FALSE;
	    }
	}
	return TRUE;
}

BOOL MyClientSocket::CreateAndConnect(CWnd *pWnd, UINT messageID, LPCTSTR devName, UINT nSocketPort)
{
	SetMessagePara(pWnd, messageID);
	ShutDown(2);
	Rest();
	if (m_hSocket == INVALID_SOCKET)
	{
		BOOL bFlag=Create(0,SOCK_STREAM,FD_CONNECT);
		if(!bFlag)
		{	
			AfxMessageBox("网络请求失败 !");
			Close();
	   		return FALSE;
		}
	}
	BOOL bflag = Connect(devIpMap[devName].c_str(),nSocketPort);
	return bflag;
}

//DEL void MyServerSocket::InitMap()
//DEL {
//DEL 
//DEL }

void MyClientSocket::InitMap()
{
	std::ifstream in;
	in.open("DevIPMap.txt");
	std::string dev, IP;
	while(in >> dev >> IP)
		devIpMap.insert(std::make_pair(dev, IP));
}

BOOL MyServerSocket::SendMess(UINT nMode, CString strMessage)
{
	CString str;
	str.Format("%d:",nMode);
	str+=strMessage;
	return SendStr(str);
}

BOOL MyClientSocket::SendMess(UINT nMode, CString strMessage)
{
	CString str;
	str.Format("%d:",nMode);
	str+=strMessage;
	str+=";";
	return SendStr(str);
}
CString MDD2CStr(const MessageDD &mess)
{
	CString str;
	str.Format("%d_%d_%d_%d_%d_%d",mess.addSrc,mess.addDes,mess.portSrc,mess.portDes,mess.bDrop,mess.hideMethod);
	return str;
}
CString MDefVC2CStr(const MessDef2VC &mess)
{
	CString str;
	str.Format("%d_%d_%d_%d_%d_%d_%d_%d_%d_%d",mess.addSrc,mess.addDes,mess.portSrc,mess.portDes,mess.hideMethod
		,mess.cut,mess.set,mess.insert,mess.sum,mess.defNum);
	return str;
}

MessDef2VC CStr2MDefVC(const CString &str, int &nMode)
{
	MessDef2VC mess;
	int indexFrom=0;
	int indexTo=str.Find(':');
	nMode=atoi(str.Mid(indexFrom, indexTo));
	indexFrom=indexTo+1;
	indexTo = str.Find('_');
	mess.addSrc=(unsigned long)atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.addDes =(unsigned long)atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.portSrc = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.portDes = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.hideMethod = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.cut = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.set = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.insert = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.sum = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	mess.defNum=atoi(str.Mid(indexFrom, str.GetLength()-indexFrom));
	return mess;	
}

MessageDD CStr2MDD(const CString &str, int &nMode)
{
	MessageDD mess;
	int indexFrom=0;
	int indexTo=str.Find(':');
	nMode=atoi(str.Mid(indexFrom, indexTo));
	indexFrom=indexTo+1;
	indexTo = str.Find('_');
	mess.addSrc=(unsigned long)atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.addDes =(unsigned long)atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.portSrc = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.portDes = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	indexTo = str.Find('_',indexTo+1);
	mess.bDrop = atoi(str.Mid(indexFrom,indexTo-indexFrom+1));
	indexFrom=indexTo+1;
	mess.hideMethod=atoi(str.Mid(indexFrom, str.GetLength()-indexFrom));
	return mess;
}