repos_ecs/hp-socket/HP-Socket/Demo/TestEcho-Agent/Agent-4C/ClientDlg.cpp

// ClientDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Client.h"
#include "ClientDlg.h"
#include "afxdialogex.h"


// CClientDlg dialog

#define DEFAULT_CONTENT	_T("text to be sent")
#define DEFAULT_ADDRESS	_T("127.0.0.1")
#define DEFAULT_PORT	_T("5555")
#define LOCAL_ADDRESS	_T("0.0.0.0")

// link: ..\..\Bin\HPSocket\x86\HPSocket_UD.lib

#ifdef _WIN64
	#ifdef _DEBUG
		#pragma comment(lib, "../../../Bin/HPSocket4C/x64/HPSocket4C_UD.lib")
	#else
		#pragma comment(lib, "../../../Bin/HPSocket4C/x64/HPSocket4C_U.lib")
	#endif
#else
	#ifdef _DEBUG
		#pragma comment(lib, "../../../Bin/HPSocket4C/x86/HPSocket4C_UD.lib")
	#else
		#pragma comment(lib, "../../../Bin/HPSocket4C/x86/HPSocket4C_U.lib")
	#endif
#endif

CClientDlg* CClientDlg::m_spThis					= nullptr;
HP_TcpPullAgent CClientDlg::m_spAgent				= nullptr;
HP_TcpPullAgentListener CClientDlg::m_spListener	= nullptr;

CClientDlg::CClientDlg(CWnd* pParent /*=NULL*/)
	: CDialogEx(CClientDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);

	m_spThis		= this;
	// 创建监听器对象
	m_spListener	= ::Create_HP_TcpPullAgentListener();
	// 创建 Socket 对象
	m_spAgent		= ::Create_HP_TcpPullAgent(m_spListener);
	// 设置 Socket 监听器回调函数
	::HP_Set_FN_Agent_OnConnect(m_spListener, OnConnect);
	::HP_Set_FN_Agent_OnSend(m_spListener, OnSend);
	::HP_Set_FN_Agent_OnPullReceive(m_spListener, OnReceive);
	::HP_Set_FN_Agent_OnClose(m_spListener, OnClose);
	::HP_Set_FN_Agent_OnShutdown(m_spListener, OnShutdown);
}

CClientDlg::~CClientDlg()
{
	// 销毁 Socket 对象
	::Destroy_HP_TcpPullAgent(m_spAgent);
	// 销毁监听器对象
	::Destroy_HP_TcpPullAgentListener(m_spListener);
}

void CClientDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_CONTENT, m_Content);
	DDX_Control(pDX, IDC_SEND, m_Send);
	DDX_Control(pDX, IDC_INFO, m_Info);
	DDX_Control(pDX, IDC_ADDRESS, m_Address);
	DDX_Control(pDX, IDC_PORT, m_Port);
	DDX_Control(pDX, IDC_START, m_Start);
	DDX_Control(pDX, IDC_CONNECT, m_Connect);
	DDX_Control(pDX, IDC_STOP, m_Stop);
}

BEGIN_MESSAGE_MAP(CClientDlg, CDialogEx)
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_SEND, &CClientDlg::OnBnClickedSend)
	ON_BN_CLICKED(IDC_START, &CClientDlg::OnBnClickedStart)
	ON_BN_CLICKED(IDC_CONNECT, &CClientDlg::OnBnClickedConnect)
	ON_BN_CLICKED(IDC_STOP, &CClientDlg::OnBnClickedStop)
	ON_MESSAGE(USER_INFO_MSG, OnUserInfoMsg)
	ON_WM_VKEYTOITEM()
END_MESSAGE_MAP()


// CClientDlg message handlers

BOOL CClientDlg::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon

	// TODO: Add extra initialization here

	m_Content.SetWindowText(DEFAULT_CONTENT);
	m_Address.SetWindowText(DEFAULT_ADDRESS);
	m_Port.SetWindowText(DEFAULT_PORT);

	::SetMainWnd(this);
	::SetInfoList(&m_Info);
	SetAppState(ST_STOPPED);

	return TRUE;  // return TRUE  unless you set the focus to a control
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CClientDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialogEx::OnPaint();
	}
}

// The system calls this function to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CClientDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

BOOL CClientDlg::PreTranslateMessage(MSG* pMsg)
{
	if (
			pMsg->message == WM_KEYDOWN		
			&&(	pMsg->wParam == VK_ESCAPE	 
			||	pMsg->wParam == VK_CANCEL	
			||	pMsg->wParam == VK_RETURN	
		))
		return TRUE;

	return CDialog::PreTranslateMessage(pMsg);
}

void CClientDlg::SetAppState(EnAppState state)
{
	m_enState = state;

	if(this->GetSafeHwnd() == nullptr)
		return;

	m_Connect.SetWindowText(m_enState == ST_CONNECTED ? _T("Dis Connect") : _T("Connect"));

	m_Start.EnableWindow(m_enState == ST_STOPPED);
	m_Connect.EnableWindow(m_enState == ST_STARTED || m_enState == ST_CONNECTED);
	m_Stop.EnableWindow(m_enState == ST_STARTED || m_enState == ST_CONNECTED);
	m_Send.EnableWindow(m_enState == ST_CONNECTED);
	m_Address.EnableWindow(m_enState == ST_STOPPED);
	m_Port.EnableWindow(m_enState == ST_STOPPED);
}

void CClientDlg::OnBnClickedSend()
{
	USES_CONVERSION;

	static DWORD SEQ = 0;

	CString strContent;
	m_Content.GetWindowText(strContent);

	smart_simple_ptr<CBufferPtr> buffer = ::GeneratePkgBuffer(++SEQ, _T("伤神小怪兽"), 23, strContent);

	if(::HP_Agent_Send(m_spAgent, m_dwConnID, buffer->Ptr(), (int)buffer->Size()))
		::LogSend(m_dwConnID, strContent);
	else
		::LogSendFail(m_dwConnID, SE_DATA_SEND, ::HP_GetSocketErrorDesc(SE_DATA_SEND));
}


void CClientDlg::OnBnClickedStart()
{
	SetAppState(ST_STARTING);

	CString strPort;

	m_Address.GetWindowText(m_strAddress);
	m_Port.GetWindowText(strPort);

	m_usPort	= (USHORT)_ttoi(strPort);
	m_dwConnID	= 0;

	m_pkgInfo.Reset();

	::LogClientStarting(m_strAddress, m_usPort);

	::HP_TcpAgent_SetReuseAddress(m_spAgent, TRUE);
	
	if(::HP_Agent_Start(m_spAgent, LOCAL_ADDRESS, FALSE))
	{
		SetAppState(ST_STARTED);
	}
	else
	{
		::LogClientStartFail(::SYS_GetLastError(), ::HP_GetSocketErrorDesc(SE_DATA_SEND));
		SetAppState(ST_STOPPED);
	}
}

void CClientDlg::OnBnClickedConnect()
{
	EnAppState enState = m_enState;

	SetAppState(ST_CONNECTING);

	if(enState != ST_CONNECTED)
	{
		if(::HP_Agent_Connect(m_spAgent, m_strAddress, m_usPort, &m_dwConnID))
		{
		}
		else
		{
			::LogClientStartFail(::SYS_GetLastError(), ::HP_GetSocketErrorDesc(SE_CONNECT_SERVER));
			SetAppState(ST_STARTED);
		}
	}
	else
	{
		::HP_Agent_Disconnect(m_spAgent, m_dwConnID, FALSE);
	}
}

void CClientDlg::OnBnClickedStop()
{
	SetAppState(ST_STOPPING);

	if(!::HP_Agent_Stop(m_spAgent))
		ASSERT(FALSE);
}

int CClientDlg::OnVKeyToItem(UINT nKey, CListBox* pListBox, UINT nIndex)
{
	if(nKey == 'C')
		pListBox->ResetContent();

	return __super::OnVKeyToItem(nKey, pListBox, nIndex);
}

LRESULT CClientDlg::OnUserInfoMsg(WPARAM wp, LPARAM lp)
{
	info_msg* msg = (info_msg*)wp;

	::LogInfoMsg(msg);

	return 0;
}

En_HP_HandleResult CClientDlg::OnConnect(HP_CONNID m_dwConnID)
{
	TCHAR szAddress[40];
	int iAddressLen = sizeof(szAddress) / sizeof(TCHAR);
	USHORT usPort;

	::HP_Agent_GetLocalAddress(m_spAgent, m_dwConnID, szAddress, &iAddressLen, &usPort);

	::PostOnConnect(m_dwConnID, szAddress, usPort);
	m_spThis->SetAppState(ST_CONNECTED);

	return HR_OK;
}

En_HP_HandleResult CClientDlg::OnSend(HP_CONNID m_dwConnID, const BYTE* pData, int iLength)
{
	::PostOnSend(m_dwConnID, pData, iLength);
	return HR_OK;
}

En_HP_HandleResult CClientDlg::OnReceive(HP_CONNID m_dwConnID, int iLength)
{
	int required = m_spThis->m_pkgInfo.length;
	int remain = iLength;

	while(remain >= required)
	{
		remain -= required;
		CBufferPtr buffer(required);

		En_HP_FetchResult result = ::HP_TcpPullAgent_Fetch(m_spAgent, m_dwConnID, buffer, (int)buffer.Size());

		if(result == FR_OK)
		{
			if(m_spThis->m_pkgInfo.is_header)
			{
				TPkgHeader* pHeader = (TPkgHeader*)buffer.Ptr();
				TRACE("[Agent] head -> seq: %d, body_len: %d\n", pHeader->seq, pHeader->body_len);

				required = pHeader->body_len;
			}
			else
			{
				TPkgBody* pBody = (TPkgBody*)buffer.Ptr();
				TRACE("[Agent] body -> name: %s, age: %d, desc: %s\n", pBody->name, pBody->age, pBody->desc);

				required = sizeof(TPkgHeader);
			}

			m_spThis->m_pkgInfo.is_header	= !m_spThis->m_pkgInfo.is_header;
			m_spThis->m_pkgInfo.length		= required;

			::PostOnReceive(m_dwConnID, buffer, (int)buffer.Size());
		}
	}

	return HR_OK;
}

En_HP_HandleResult CClientDlg::OnClose(HP_CONNID m_dwConnID, En_HP_SocketOperation enOperation, int iErrorCode)
{
	iErrorCode == SE_OK ? ::PostOnClose(m_dwConnID):
	::PostOnError(m_dwConnID, enOperation, iErrorCode);

	m_spThis->SetAppState(ST_STARTED);

	return HR_OK;
}

En_HP_HandleResult CClientDlg::OnShutdown()
{
	::PostOnShutdown();
	m_spThis->SetAppState(ST_STOPPED);

	return HR_OK;
}