tcpcldlg.cpp

早期的WINDOWS编程,适合刚接触WINDOWS编程的初学者

// TCPCldlg.cpp : implementation file
//

#include "stdafx.h"
#include "..\cwclient.h"
#include "..\testinfo.h"
#include "TCPClient.h"
#include "TCPCldlg.h"

#ifdef _DEBUG
#undef THIS_FILE
static char BASED_CODE THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CTCPClientDlg dialog

CTCPClientDlg::CTCPClientDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTCPClientDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CTCPClientDlg)
	m_strStatus = _T("Uninitialized");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CTCPClientDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CTCPClientDlg)
	DDX_Control(pDX, IDC_CONNNECT, m_ConnectButton);
	DDX_Text(pDX, IDC_STATUS, m_strStatus);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTCPClientDlg, CDialog)
	ON_MESSAGE(WSA_EVENT,  OnEvent)
	ON_MESSAGE(WSA_CONNECT,OnConnect)
	ON_MESSAGE(WSA_CLOSE, OnClose)
	ON_MESSAGE(WSA_READ,  OnRead)
	ON_MESSAGE(WSA_WRITE, OnWrite)
	ON_MESSAGE(WSA_OOB,   OnOOB)
	//{{AFX_MSG_MAP(CTCPClientDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_DESTROY()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_CONNNECT, OnConnnectButton)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTCPClientDlg message handlers

BOOL CTCPClientDlg::OnInitDialog()
{
	CDialog::OnInitDialog();
	CenterWindow();

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

void CTCPClientDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	CDialog::OnSysCommand(nID, lParam);
}

void CTCPClientDlg::OnDestroy()
{
	WinHelp(0L, HELP_QUIT);
	CDialog::OnDestroy();
}

// 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 CTCPClientDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (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
	{
		CDialog::OnPaint();
	}
}

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


void CTCPClientDlg::OnConnnectButton() 
{
	m_pClient = new CWClient;
	if (m_pClient->GetCurrentSocketState() == eWSInitialized){
		m_strStatus = "Initialized";
		SetDlgItemText(IDC_STATUS, m_strStatus);
	}
	
	if (m_pClient != NULL){
		m_ConnectButton.EnableWindow(FALSE);
	}
	
	int iRet = m_pClient->ConnectToServer(constPortNo,NULL,m_hWnd);
	if (iRet == 0){
		m_strStatus = "Connected to Server";
		SetDlgItemText(IDC_STATUS,m_strStatus);
	}	
}

void CTCPClientDlg::OnCancel() 
{
	if (m_pClient != NULL)
		delete m_pClient;
			
	CDialog::OnCancel();
}

LRESULT CTCPClientDlg::OnEvent(WPARAM wParam, LPARAM lParam)
{
	LRESULT lresultRet;


	int iEvent = WSAGETSELECTEVENT(lParam);
	int iError = WSAGETSELECTERROR(lParam);

    WS_TRACE1("Event Received %u", iEvent);

	CString cstr = m_pClient->StringError(iError);
	WS_TRACE1(" (error code: %s) ",cstr);

	m_pClient->WSASetLastError(iError);

	if (iEvent == FD_READ)
		lresultRet = OnRead(wParam, lParam);
	else if (iEvent == FD_WRITE)
		lresultRet = OnWrite(wParam, lParam);
	else if (iEvent == FD_CLOSE)
		lresultRet = OnClose(wParam, lParam);
	else if (iEvent == FD_OOB)
		lresultRet = OnOOB(wParam, lParam);
	else if (iEvent == FD_CONNECT)
		lresultRet = OnConnect(wParam, lParam);
	else
		return -1;

	cstr = m_pClient->StringError();

	SetDlgItemText(IDC_WINSOCK_ERROR,"                           ");
	SetDlgItemText(IDC_WINSOCK_ERROR,cstr);
	return lresultRet;
}	

LRESULT CTCPClientDlg::OnConnect(WPARAM wParam, LPARAM lParam)
{
	WS_TRACE0("Connect Event\n");

	if (m_pClient->WSAGetLastError() != 0)
		return m_pClient->WSAGetLastError();

	m_strStatus = "Connect received";
	SetDlgItemText(IDC_STATUS, m_strStatus);

	int err = m_pClient->connect(m_pClient->GetDestSin(), sizeof(SOCKADDR_IN));

	if (err < 0){
   		WS_TRACE0("ConnectToServer: connect error-");
 		m_pClient->PrintError();
		return m_pClient->WSAGetLastError();
	}
	
	WS_TRACE0("ConnectToServer: connected");

	m_strStatus = "Connected";
	SetDlgItemText(IDC_STATUS,m_strStatus);

	return 0;
}

LRESULT CTCPClientDlg::OnClose(WPARAM, LPARAM)
{
	WS_TRACE0("Close Event\n");

	m_strStatus = "Closed";
	SetDlgItemText(IDC_STATUS,m_strStatus);

	return 0;
}
LRESULT CTCPClientDlg::OnRead(WPARAM, LPARAM)
{
	TCHAR	buf[254];

	WS_TRACE0("Read Event\n");

	int err = m_pClient->recv(buf,sizeof(buf),0);

	if (err < 0){
   		WS_TRACE0("ConnectToServer: receive error-");
 		m_pClient->PrintError();
		return m_pClient->WSAGetLastError();
	}
	
	WS_TRACE0("ConnectToServer: received");


	m_strStatus = "Read";
	SetDlgItemText(IDC_STATUS,m_strStatus);

	return 0;
}
LRESULT CTCPClientDlg::OnWrite(WPARAM, LPARAM)
{
	WS_TRACE0("Write Event\n");

	m_strStatus = "Write";
	SetDlgItemText(IDC_STATUS,m_strStatus);

    char szTestData[250];
    int err;

   for (int i=0;i<10;i++){
    wsprintf(szTestData,"Test Data %u",i);
	err = m_pClient->send(szTestData,lstrlen(szTestData),0);

	if (err < 0){
   		WS_TRACE0("ConnectToServer: send error-");
 		m_pClient->PrintError();
		return m_pClient->WSAGetLastError();
	}
	
	WS_TRACE0("ConnectToServer: sent");
   }
	return 0;
}

LRESULT CTCPClientDlg::OnOOB(WPARAM, LPARAM)
{
	WS_TRACE0("Out of Band Event\n");

	m_strStatus = "Out of Band";
	SetDlgItemText(IDC_STATUS,m_strStatus);

	return 0;
}