clientsocket.cpp

游戏框架

#include "Stdafx.h"
#include "ClientSocket.h"

//////////////////////////////////////////////////////////////////////////

//宏定义
#define ID_SOCKET_WND				10								//SOCKET 窗口 ID
#define WM_SOCKET_NOTIFY			WM_USER+10						//SOCKET 消息

//////////////////////////////////////////////////////////////////////////


//////////////////////////////////////////////////////////////////////////

//发送字节映射表
BYTE CClientSocket::m_SendByteMap[256]=				
{
	0x70,0x2F,0x40,0x5F,0x44,0x8E,0x6E,0x45,0x7E,0xAB,0x2C,0x1F,0xB4,0xAC,0x9D,0x91,
		0x0D,0x36,0x9B,0x0B,0xD4,0xC4,0x39,0x74,0xBF,0x23,0x16,0x14,0x06,0xEB,0x04,0x3E,
		0x12,0x5C,0x8B,0xBC,0x61,0x63,0xF6,0xA5,0xE1,0x65,0xD8,0xF5,0x5A,0x07,0xF0,0x13,
		0xF2,0x20,0x6B,0x4A,0x24,0x59,0x89,0x64,0xD7,0x42,0x6A,0x5E,0x3D,0x0A,0x77,0xE0,
		0x80,0x27,0xB8,0xC5,0x8C,0x0E,0xFA,0x8A,0xD5,0x29,0x56,0x57,0x6C,0x53,0x67,0x41,
		0xE8,0x00,0x1A,0xCE,0x86,0x83,0xB0,0x22,0x28,0x4D,0x3F,0x26,0x46,0x4F,0x6F,0x2B,
		0x72,0x3A,0xF1,0x8D,0x97,0x95,0x49,0x84,0xE5,0xE3,0x79,0x8F,0x51,0x10,0xA8,0x82,
		0xC6,0xDD,0xFF,0xFC,0xE4,0xCF,0xB3,0x09,0x5D,0xEA,0x9C,0x34,0xF9,0x17,0x9F,0xDA,
		0x87,0xF8,0x15,0x05,0x3C,0xD3,0xA4,0x85,0x2E,0xFB,0xEE,0x47,0x3B,0xEF,0x37,0x7F,
		0x93,0xAF,0x69,0x0C,0x71,0x31,0xDE,0x21,0x75,0xA0,0xAA,0xBA,0x7C,0x38,0x02,0xB7,
		0x81,0x01,0xFD,0xE7,0x1D,0xCC,0xCD,0xBD,0x1B,0x7A,0x2A,0xAD,0x66,0xBE,0x55,0x33,
		0x03,0xDB,0x88,0xB2,0x1E,0x4E,0xB9,0xE6,0xC2,0xF7,0xCB,0x7D,0xC9,0x62,0xC3,0xA6,
		0xDC,0xA7,0x50,0xB5,0x4B,0x94,0xC0,0x92,0x4C,0x11,0x5B,0x78,0xD9,0xB1,0xED,0x19,
		0xE9,0xA1,0x1C,0xB6,0x32,0x99,0xA3,0x76,0x9E,0x7B,0x6D,0x9A,0x30,0xD6,0xA9,0x25,
		0xC7,0xAE,0x96,0x35,0xD0,0xBB,0xD2,0xC8,0xA2,0x08,0xF3,0xD1,0x73,0xF4,0x48,0x2D,
		0x90,0xCA,0xE2,0x58,0xC1,0x18,0x52,0xFE,0xDF,0x68,0x98,0x54,0xEC,0x60,0x43,0x0F
};

//接收字节映射表
BYTE CClientSocket::m_RecvByteMap[256]=				
{
	0x51,0xA1,0x9E,0xB0,0x1E,0x83,0x1C,0x2D,0xE9,0x77,0x3D,0x13,0x93,0x10,0x45,0xFF,
		0x6D,0xC9,0x20,0x2F,0x1B,0x82,0x1A,0x7D,0xF5,0xCF,0x52,0xA8,0xD2,0xA4,0xB4,0x0B,
		0x31,0x97,0x57,0x19,0x34,0xDF,0x5B,0x41,0x58,0x49,0xAA,0x5F,0x0A,0xEF,0x88,0x01,
		0xDC,0x95,0xD4,0xAF,0x7B,0xE3,0x11,0x8E,0x9D,0x16,0x61,0x8C,0x84,0x3C,0x1F,0x5A,
		0x02,0x4F,0x39,0xFE,0x04,0x07,0x5C,0x8B,0xEE,0x66,0x33,0xC4,0xC8,0x59,0xB5,0x5D,
		0xC2,0x6C,0xF6,0x4D,0xFB,0xAE,0x4A,0x4B,0xF3,0x35,0x2C,0xCA,0x21,0x78,0x3B,0x03,
		0xFD,0x24,0xBD,0x25,0x37,0x29,0xAC,0x4E,0xF9,0x92,0x3A,0x32,0x4C,0xDA,0x06,0x5E,
		0x00,0x94,0x60,0xEC,0x17,0x98,0xD7,0x3E,0xCB,0x6A,0xA9,0xD9,0x9C,0xBB,0x08,0x8F,
		0x40,0xA0,0x6F,0x55,0x67,0x87,0x54,0x80,0xB2,0x36,0x47,0x22,0x44,0x63,0x05,0x6B,
		0xF0,0x0F,0xC7,0x90,0xC5,0x65,0xE2,0x64,0xFA,0xD5,0xDB,0x12,0x7A,0x0E,0xD8,0x7E,
		0x99,0xD1,0xE8,0xD6,0x86,0x27,0xBF,0xC1,0x6E,0xDE,0x9A,0x09,0x0D,0xAB,0xE1,0x91,
		0x56,0xCD,0xB3,0x76,0x0C,0xC3,0xD3,0x9F,0x42,0xB6,0x9B,0xE5,0x23,0xA7,0xAD,0x18,
		0xC6,0xF4,0xB8,0xBE,0x15,0x43,0x70,0xE0,0xE7,0xBC,0xF1,0xBA,0xA5,0xA6,0x53,0x75,
		0xE4,0xEB,0xE6,0x85,0x14,0x48,0xDD,0x38,0x2A,0xCC,0x7F,0xB1,0xC0,0x71,0x96,0xF8,
		0x3F,0x28,0xF2,0x69,0x74,0x68,0xB7,0xA3,0x50,0xD0,0x79,0x1D,0xFC,0xCE,0x8A,0x8D,
		0x2E,0x62,0x30,0xEA,0xED,0x2B,0x26,0xB9,0x81,0x7C,0x46,0x89,0x73,0xA2,0xF7,0x72
};

//数据加密密钥
const static DWORD g_dwPacketKey=0xA55AA55A;

//////////////////////////////////////////////////////////////////////////

//构造函数
CClientSocket::CClientSocket()
{
	m_wRecvSize=0;
	m_cbSendRound=0;
	m_cbRecvRound=0;
	m_dwSendXorKey=0;
	m_dwRecvXorKey=0;
	m_dwSendTickCount=0;
	m_dwRecvTickCount=0;
	m_dwSendPacketCount=0;
	m_dwRecvPacketCount=0;
	m_bCloseByServer=false;
	m_hSocket=INVALID_SOCKET;
	//m_pIClientSocketSink=NULL;
	m_ProxyInfo.wProxyPort=0;
	m_ProxyInfo.cbProxyType=PROXY_NONE;
	m_SocketState=SocketState_NoConnect;
	m_hWnd=NULL;
	m_bEncrypted=true;
}

//析构函数
CClientSocket::~CClientSocket()
{
	CloseSocket(false);
}

//接口查询
void * __cdecl CClientSocket::QueryInterface(const IID & Guid, DWORD dwQueryVer)
{
	QUERYINTERFACE(IClientSocket,Guid,dwQueryVer);
	QUERYINTERFACE_IUNKNOWNEX(IClientSocket,Guid,dwQueryVer);
	return NULL;
}

//取消代理
void __cdecl CClientSocket::CancelProxyServer()
{
	m_ProxyInfo.wProxyPort=0;
	m_ProxyInfo.strHttpDomain.IsNull();
	m_ProxyInfo.strProxyName.IsNull();
	m_ProxyInfo.strProxyPass.IsNull();
	m_ProxyInfo.strProxyServer.IsNull();
	m_ProxyInfo.cbProxyType=PROXY_NONE;
	return;
}

//代理服务器
void __cdecl CClientSocket::SetProxyServer(tagProxyInfo & ProxyInfo)
{
	m_ProxyInfo=ProxyInfo;
	return;
}

//设置加密
void __cdecl CClientSocket::SetEncrypted(bool bEncrypted){
	m_bEncrypted = bEncrypted;
}

//是否加密
bool __cdecl CClientSocket::IsEncrypted(){
	return m_bEncrypted;
}

//设置接口
bool __cdecl CClientSocket::SetSocketSink(IUnknownEx * pIUnknownEx)
{
	//效验参数
	_ASSERT(pIUnknownEx!=NULL);
	if (pIUnknownEx==NULL) return false;

	//设置接口
	if (m_AttemperEvent.SetQueueService(pIUnknownEx)==false)
	{
		return false;
	}
	
	return true;
}

//获取接口
void * __cdecl CClientSocket::GetSocketSink(const IID & Guid, DWORD dwQueryVer)
{
	//if (m_pIClientSocketSink==NULL) return NULL;
	//return m_pIClientSocketSink->QueryInterface(Guid,dwQueryVer);
	return NULL;
}

//连接服务器
bool __cdecl CClientSocket::Connect(DWORD dwServerIP, WORD wPort)
{
	//效验参数
	_ASSERT(m_hSocket==INVALID_SOCKET);
	_ASSERT(m_SocketState==SocketState_NoConnect);

	//效验状态
	if (m_hSocket!=INVALID_SOCKET) throw TEXT("连接 SOCKET 句柄已经存在");
	if (m_SocketState!=SocketState_NoConnect) throw TEXT("连接状态不是等待连接状态");
	if (dwServerIP==INADDR_NONE) throw TEXT("目标服务器地址格式不正确，请检查后再次尝试！");

	//设置参数
	m_wRecvSize=0;
	m_cbSendRound=0;
	m_cbRecvRound=0;
	m_dwSendXorKey=0x12345678;
	m_dwRecvXorKey=0x12345678;
	m_dwSendTickCount=GetTickCount()/1000L;
	m_dwRecvTickCount=GetTickCount()/1000L;

	try
	{
		//建立 SOCKET
        WSADATA   wsaData;   
        if(::WSAStartup(MAKEWORD(2,2),&wsaData)!=0)
			throw TEXT("WSAStartup 执行失败");

		m_hSocket=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
		int nError = WSAGetLastError();
		if (m_hSocket==INVALID_SOCKET) throw TEXT("SOCKET 创建失败");
		
		if ((m_hWnd==NULL)&&(m_hWnd = MakeWorkerWindow())==NULL)
			throw TEXT("SOCKET 内部窗口创建失败");

		//填写服务器地址
		SOCKADDR_IN SocketAddr;
		memset(&SocketAddr,0,sizeof(SocketAddr));
		SocketAddr.sin_family=AF_INET;
		SocketAddr.sin_port=htons(wPort);
		SocketAddr.sin_addr.S_un.S_addr=dwServerIP;

		//连接服务器
		int iErrorCode=0;
		WSASetLastError(0);
		if (m_ProxyInfo.cbProxyType!=PROXY_NONE)
		{
			//填写服务器地址
			sockaddr_in SocketProxyAddr;
			memset(&SocketProxyAddr,0,sizeof(SocketProxyAddr));
			SocketProxyAddr.sin_family=AF_INET;
			SocketProxyAddr.sin_port=htons(m_ProxyInfo.wProxyPort);
			SocketProxyAddr.sin_addr.S_un.S_addr=TranslateAddr(m_ProxyInfo.strProxyServer);
			iErrorCode=connect(m_hSocket,(sockaddr *)&SocketProxyAddr,sizeof(SocketProxyAddr));
			if (iErrorCode==SOCKET_ERROR) 
			{
				WSASetLastError(0);
				//AfxMessageBox("代理服务器连接失败，请联系网络管理员！",MB_ICONSTOP);
				return false;
			}

			DWORD arg = 0;
			int retv=ioctlsocket(m_hSocket,FIONBIO,&arg);		// 有代理服务器时设为Blocking方式
			if (retv!=0) return false;

			//与代理服务器通讯
			int nOff=0;
			char cbRecvBuffer[SOCKET_PACKAGE];
			if (m_ProxyInfo.cbProxyType==PROXY_SOCKS5)
			{
				// 1、请求代理服务器服务方式，包括：无密码方式和需密码方式
				cbRecvBuffer[nOff++]=5;		// ver
				cbRecvBuffer[nOff++]=1;		// method count
				cbRecvBuffer[nOff++]=(m_ProxyInfo.strProxyName.IsNull())?0:2;	// method
				if (send(m_hSocket,cbRecvBuffer,nOff,0)!=nOff)
				{
					//请求代理服务器失败
					return false;
				}
				int iRec=recv(m_hSocket,cbRecvBuffer,2,0);
				if (iRec!=2)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器应答失败",MB_ICONSTOP);
					return false;
				}
				if (cbRecvBuffer[0]!=5)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器应答失败,Error 1",MB_ICONSTOP);
					return false;
				}
				if ((cbRecvBuffer[1]!=0)&&(cbRecvBuffer[1]!=2))
				{
					WSASetLastError(0);
					//AfxMessageBox("代理服务器拒绝请求",MB_ICONSTOP);
					return false;
				}

				nOff=0;
				if (cbRecvBuffer[1]==2)
				{
					// 2、身份验证请求
					/*+----+------+----------+------+----------+
					|VER | ULEN |  UNAME   | PLEN |  PASSWD  |
					+----+------+----------+------+----------+
					| 1  |  1   | 1 to 255 |  1   | 1 to 255 |
					+----+------+----------+------+----------+*/
					// version of the subnegotiation=1
					cbRecvBuffer[nOff++]=1; 
					int nStrLen=(int)m_ProxyInfo.strProxyName.GetLength();
					cbRecvBuffer[nOff++]=(char)nStrLen;
					memcpy(cbRecvBuffer+nOff,(LPCTSTR)m_ProxyInfo.strProxyName,nStrLen);
					nOff+=nStrLen;

					nStrLen=(int)m_ProxyInfo.strProxyPass.GetLength();
					cbRecvBuffer[nOff++]=(char)nStrLen;
					memcpy(cbRecvBuffer+nOff,(LPCTSTR)m_ProxyInfo.strProxyPass,nStrLen);
					nOff+=nStrLen;

					int iSend=send(m_hSocket,cbRecvBuffer,nOff,0);
					if (iSend!=nOff)
					{
						WSASetLastError(0);
						//AfxMessageBox("向代理服务器发送身份验证失败",MB_ICONSTOP);
						return false;
					}

					int iRecv=recv(m_hSocket,cbRecvBuffer,2,0);
					if (iRecv!=2)
					{
						WSASetLastError(0);
						//AfxMessageBox("接收代理服务器身份验证应答失败",MB_ICONSTOP);
						return false;
					}

					// 3、proxy 应答身份验证请求
					/*+----+--------+
					|VER | STATUS |
					+----+--------+
					| 1  |   1    |
					+----+--------+*/
					// version of the subnegotiation=1
					if (cbRecvBuffer[0]!=1)
					{
						WSASetLastError(0);
						//AfxMessageBox("您的通往代理服务器的身份可能有错",MB_ICONSTOP);
						return false;
					}
					if (cbRecvBuffer[1]!=0)
					{
						WSASetLastError(0);
						//AfxMessageBox("请重新输入您通向代理服务器的用户名和密码",MB_ICONSTOP);
						return false;
					}
				}

				// 2、请求连接远程服务器
				/*+----+-----+-------+------+----------+--------+
				|VER | CMD |  RSV  | ATYP | DST.ADDR | DST.PORT |
				+----+-----+-------+------+----------+----------+
				| 1  |  1  | X'00' |  1   | Variable |    2     |
				+----+-----+-------+------+----------+----------*/
				nOff=0;
				cbRecvBuffer[nOff++]=5;	// ver
				cbRecvBuffer[nOff++]=1;	// connect;
				cbRecvBuffer[nOff++]=0;	// RSV
				cbRecvBuffer[nOff++]=1;	// Adress type : IP V4
				memcpy(cbRecvBuffer+nOff,(char*)&SocketAddr.sin_addr,4);
				nOff+=4;
				*(WORD*)(cbRecvBuffer+nOff)=SocketAddr.sin_port;
				nOff+=2;

				if (send(m_hSocket,cbRecvBuffer,nOff,0)!=nOff)
				{
					WSASetLastError(0);
					//AfxMessageBox("请求代理服务器与远程连接失败",MB_ICONSTOP);
					return false;
				}

				// 接收PROXY对connect的reply
				// reply的数据结构与connect请求的一样
				if (recv(m_hSocket,cbRecvBuffer,4,0)!=4)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器对于连接的应答失败",MB_ICONSTOP);
					return false;
				}

				if (cbRecvBuffer[0]!=5)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器对于连接的应答失败,Error 1",MB_ICONSTOP);
					return false;
				}
				if (cbRecvBuffer[1]!=0)
				{
					WSASetLastError(0);
					//AfxMessageBox("代理服务器拒绝向远程连接的请求",MB_ICONSTOP);
					return false;
				}

				int nLeftDataLen=(cbRecvBuffer[3]==1)?6:cbRecvBuffer[4]+2;
				int nRecvLen=recv(m_hSocket,cbRecvBuffer,nLeftDataLen,0);
				if (nRecvLen!=nLeftDataLen)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器数据过程中出现异常,Error 1",MB_ICONSTOP);
					return false;
				}
			}
			else
			{
				/*----+----+----+----+----+----+----+----+----+----+....+----+
				|VER | CMD| DSTPORT |      DSTIP        | USERID       |NULL|
				+----+----+----+----+----+----+----+----+----+----+....+----*/
				nOff=0;
				cbRecvBuffer[nOff++]=4;	// VER
				cbRecvBuffer[nOff++]=1;	// connect
				*(WORD*)(cbRecvBuffer+nOff)=SocketAddr.sin_port;
				nOff+=2;
				memcpy(cbRecvBuffer+nOff,(char*)&SocketAddr.sin_addr,4);
				nOff+=4;
				cbRecvBuffer[nOff++]=0;	// no userid
				if (send(m_hSocket,cbRecvBuffer,nOff,0)!=nOff)
				{
					WSASetLastError(0);
					//AfxMessageBox("请求代理服务器与远程连接失败",MB_ICONSTOP);
					return false;
				}
				if (recv(m_hSocket,cbRecvBuffer,8,0)!=8)
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器对于连接的应答失败",MB_ICONSTOP);
					return false;
				}
				if (cbRecvBuffer[0]!=0)		// the version of the reply code
				{
					WSASetLastError(0);
					//AfxMessageBox("接收代理服务器对于连接的应答失败",MB_ICONSTOP);
					return false;
				}
				if (cbRecvBuffer[1]!=90)	// request granted
				{
					WSASetLastError(0);
					//AfxMessageBox("代理服务器拒绝连接或连接失败",MB_ICONSTOP);
					return false;
				}
			}
			if (WSAAsyncSelect(m_hSocket,m_hWnd,WM_SOCKET_NOTIFY,FD_READ|FD_CONNECT|FD_CLOSE)==SOCKET_ERROR) return false;
			m_SocketState=SocketState_Connected;
		}
		else
		{
			//绑定窗口
			iErrorCode=WSAAsyncSelect(m_hSocket,m_hWnd,WM_SOCKET_NOTIFY,FD_READ|FD_CONNECT|FD_CLOSE);
			if (iErrorCode==SOCKET_ERROR) throw TEXT("绑定内部窗口错误");

			//连接服务器
			iErrorCode=connect(m_hSocket,(SOCKADDR *)&SocketAddr,sizeof(SocketAddr));
			if (iErrorCode==SOCKET_ERROR)
			{
				iErrorCode=WSAGetLastError();
				if (iErrorCode!=WSAEWOULDBLOCK)
				{
					static TCHAR szBuffer[64];
					_snprintf(szBuffer,sizeof(szBuffer),TEXT("连接发生错误，错误代码 [ %d ]"),iErrorCode);
					throw szBuffer;
				}
			}

			//设置变量
			m_SocketState=SocketState_Connecting;
		}

		return true;
	}
	catch (LPCTSTR pszError)
	{
		CloseSocket(false);
		throw pszError;
	}
	catch (...)
	{
		CloseSocket(false);
		throw TEXT("连接产生未知异常错误");
	}

	return false;
}

//连接服务器
bool __cdecl CClientSocket::Connect(const LPCTSTR szServerIP, WORD wPort)
{
	//效验数据
	_ASSERT(wPort!=0);
	_ASSERT(szServerIP!=NULL);
	if ((szServerIP==NULL)||(wPort==0)) return false;
	return Connect(TranslateAddr(szServerIP),wPort);
}

//发送函数
bool __cdecl CClientSocket::SendData(WORD wMainCmdID, WORD wSubCmdID)
{
	//效验状态
	if (m_hSocket==INVALID_SOCKET) return false;
	if (m_SocketState!=SocketState_Connected) return false;

	//构造数据
	BYTE cbDataBuffer[SOCKET_BUFFER];
	CMD_Head * pHead=(CMD_Head *)cbDataBuffer;
	pHead->CommandInfo.wMainCmdID=wMainCmdID;
	pHead->CommandInfo.wSubCmdID=wSubCmdID;

	WORD wSendSize = sizeof(CMD_Head);
	pHead->CmdInfo.wDataSize=wSendSize;
	pHead->CmdInfo.cbMessageVer=SOCKET_VER;
	if(m_bEncrypted){
		//加密数据
		wSendSize=EncryptBuffer(cbDataBuffer,sizeof(CMD_Head),sizeof(cbDataBuffer));
	}

	m_dwSendPacketCount++;

	//发送数据
	return SendBuffer(cbDataBuffer,wSendSize);
}

//发送函数