// ClientSocket.cpp : implementation file
//

#include "stdafx.h"
#include "SecretChat.h"		//應(yīng)用程序類頭文件
#include "ClientSocket.h"
#include "SecretChatDlg.h"	//主窗口類頭文件

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

#define WM_CANCELINSTANCE ( WM_USER + 2)	//關(guān)閉窗口消息
/////////////////////////////////////////////////////////////////////////////
// CClientSocket


CClientSocket::CClientSocket()
{
}

CClientSocket::~CClientSocket()
{
}


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

/////////////////////////////////////////////////////////////////////////////
// CClientSocket member functions

void CClientSocket::OnReceive(int nErrorCode)	//接收服務(wù)器數(shù)據(jù)事件
{	
	if(0 != nErrorCode)
	{	//發(fā)送錯(cuò)誤
		AfxMessageBox("接收數(shù)據(jù)出錯(cuò)-1");
		return;
	}

	CSecretChatDlg * pMain = (CSecretChatDlg *)AfxGetMainWnd();
	MessagePackage msg;
	int n;					//實(shí)際接收到數(shù)據(jù)的字節(jié)數(shù)
	n = Receive(&msg, sizeof(MessagePackage));	//接收客戶機(jī)數(shù)據(jù)

	if(0 > n)
	{	//發(fā)送錯(cuò)誤
		AfxMessageBox("接收數(shù)據(jù)出錯(cuò)-2");
		return;
	}

	//錯(cuò)誤提示
	pMain->CuoWuTiShi();

	/**********當(dāng)收到的是對(duì)方發(fā)過來的消息************/
	if(msg.head == HEAD_TEXT) //收到的是正文
	{
		pMain->Receive(msg);	//在這里會(huì)調(diào)用接收消息的線程
	}

	else if(msg.head == HEAD_REVERT_TEXT) //發(fā)出的正文就收到的是回復(fù)
	{
		pMain->ReceiveRevert(msg);
	}

	else if(msg.head == HEAD_DISCONNECTION) //收到的是斷開提示
	{
		pMain->disconnection_message(msg);
	}

	else if(msg.head == HEAD_ZAIXIANBIAOJI)		//在線通知
	{
		pMain->m_zai_xian_biao_ji = 0;	//取消等待時(shí)間記錄
	}

	else if(msg.head == HEAD_VERSION)		//對(duì)方版本號(hào)和一些其他信息
	{	
		pMain->message_version(msg);
	}

	else if(msg.head == HEAD_CLAIM_PUBLIC_KEY) //對(duì)方的請(qǐng)求獲得公鑰
	{
		pMain->ClaimPublicKey(msg);
	}

	else if(msg.head == HEAD_REVERT_PUBLIC_KEY) //收到對(duì)方發(fā)過來的公鑰
	{		
		pMain->RevertPublicKey(msg);
	}

	else if(msg.head == HEAD_DIGITAL_SIGNATURE) //對(duì)方的數(shù)字簽名，看看我這里有沒有它的公鑰，可以在請(qǐng)求獲得它的公鑰
	{
		if(msg.ID == HEAD_DIGITAL_SIGNATURE_NO)
		{	//對(duì)方還沒識(shí)別我這條消息的身份
			MessageBox(
				NULL,
				"對(duì)方不是 " 
				+ pMain->GetFriendName() 
				+ " 好友或沒有 " 
				+ pMain->GetUserName() 
				+ " 用戶的公鑰",
				"發(fā)送失敗身份不確認(rèn)",
				MB_ICONEXCLAMATION);
		}
	}

	else if(msg.head == HEAD_SENDFILE) //對(duì)方發(fā)來了文件及相關(guān)信息
	{
		pMain->receive_file(msg);
	}

	/******************************/

	//要求讀、寫、關(guān)閉通知，會(huì)出錯(cuò)
//	AsyncSelect(FD_READ | FD_WRITE | FD_CLOSE);
}