tcpsedlg.cpp

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

// TCPSedlg.cpp : implementation file
//

#include "stdafx.h"
#include "..\cwserver.h"
#include "..\testinfo.h"
#include "TCPServer.h"
#include "TCPSedlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CTCPServerDlg dialog

CTCPServerDlg::CTCPServerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CTCPServerDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CTCPServerDlg)
	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 CTCPServerDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CTCPServerDlg)
	DDX_Control(pDX, IDC_START_BUTTON, m_StartButton);
	DDX_Text(pDX, IDC_STATUS, m_strStatus);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CTCPServerDlg, 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)
	ON_MESSAGE(WSA_ACCEPT,OnAccept)
	//{{AFX_MSG_MAP(CTCPServerDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_DESTROY()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_START_BUTTON, OnStartButton)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CTCPServerDlg message handlers

BOOL CTCPServerDlg::OnInitDialog()
{
	CDialog::OnInitDialog();
	RECT rectDialog;
	GetWindowRect(&rectDialog);
	rectDialog.left = 0;
	rectDialog.top  = 0;
	ClientToScreen(&rectDialog);
	SetWindowPos(NULL, rectDialog.left, rectDialog.top, 0,0, SWP_NOSIZE | SWP_NOZORDER);

	for (int i=0;i<MaxServerConnections;i++)
		m_ServerSockets[i].m_bSocketAvailable = TRUE;

    m_ulConnectionCount = 0l;
	m_pServer = NULL;		
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

void CTCPServerDlg::OnDestroy()
{
	for (int i=0; i<MaxServerConnections;i++){
		if (m_ServerSockets[i].m_bSocketAvailable == FALSE){
			WS_TRACE1("Destroy: Closesocket %u\n",i);
			m_pServer->closesocket(m_ServerSockets[i].m_Socket);
			m_ServerSockets[i].m_bSocketAvailable = TRUE;
		}
	}

	if (m_pServer != NULL)
		delete m_pServer;
			
	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 CTCPServerDlg::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 CTCPServerDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CTCPServerDlg::OnStartButton() 
{
	m_pServer = new CWServer;
	if (m_pServer->GetCurrentSocketState() == eWSInitialized){
		m_strStatus = "Initialized";
		SetDlgItemText(IDC_STATUS,m_strStatus);
	}

	if (m_pServer != NULL){
		m_StartButton.EnableWindow(FALSE);
	}

	int iRet = m_pServer->WaitForClient(constPortNo,NULL,m_hWnd);
	if (iRet == 0){
		m_strStatus = "Waiting to connect";
		SetDlgItemText(IDC_STATUS,m_strStatus);
	}	
	
	SetDlgItemInt(IDC_WINSOCK_CNT,m_ulConnectionCount);

}

void CTCPServerDlg::OnCancel() 
{
	// TODO: Add extra cleanup here

	CDialog::OnCancel();
}

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

	WS_TRACE0("Event Received");

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

	CString cstr = m_pServer->StringError(iError);
	WS_TRACE1(" (error code: %s) ",cstr);
	
	m_pServer->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 if (iEvent == FD_ACCEPT)
		lresultRet = OnAccept(wParam, lParam);
	else
		return -1;

	cstr = m_pServer->StringError();

	SetDlgItemText(IDC_WINSOCK_ERROR,cstr);
	SetDlgItemInt(IDC_WINSOCK_CNT,m_ulConnectionCount);
	
	return lresultRet;
}	

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

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

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

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

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

	return 0;
}

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

	struct linger slinger;
	slinger.l_onoff = 0;
	slinger.l_linger = 0;

	for (int i=0;i<MaxServerConnections;i++){
		if (m_ServerSockets[i].m_bSocketAvailable == FALSE){
			if (m_ServerSockets[i].m_Socket == wParam){
				//This will cause the socket to be freed immediately when
				//we call closesocket
				m_pServer->setsockopt(m_ServerSockets[i].m_Socket,
									  SOL_SOCKET,
									  SO_LINGER,
									  &slinger,
									  sizeof(slinger));

				m_pServer->closesocket(m_ServerSockets[i].m_Socket);
				WS_TRACE1("Closing using socket in %u\n",i);

				if (m_pServer->WSAGetLastError() == 0){
					m_ServerSockets[i].m_bSocketAvailable = TRUE;
                    m_ulConnectionCount--;
                }
				break;
			}
		}
	}

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

	return 0;
}
LRESULT CTCPServerDlg::OnRead(WPARAM wParam, LPARAM)
{
	TCHAR	buf[254];
    SOCKET  sock = (SOCKET)wParam;

	WS_TRACE0("Read Event\n");

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

	if (err < 0){
   		WS_TRACE0("ConnectToServer: receive error-");
 		m_pServer->PrintError();
		return m_pServer->WSAGetLastError();
	} else
        buf[err] = 0x00;    //Just for NULL termination
	
	WS_TRACE0("ConnectToServer: received");


	m_strStatus = "Read:";
    m_strStatus+= buf;
	SetDlgItemText(IDC_STATUS,m_strStatus);

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

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

	return 0;
}

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

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

	return 0;
}

LRESULT CTCPServerDlg::OnAccept(WPARAM, LPARAM)
{
	BOOL	bSlotFree = FALSE;
	WS_TRACE0("Accept Event\n");

	for (int i=0;i<MaxServerConnections;i++){
		if (m_ServerSockets[i].m_bSocketAvailable == TRUE){
			m_ServerSockets[i].m_Socket = m_pServer->accept();
			WS_TRACE1("Accept storing socket in %u\n",i);
			if (m_pServer->WSAGetLastError() == 0){
				m_ServerSockets[i].m_bSocketAvailable = FALSE;
                m_ulConnectionCount++;
            }
			bSlotFree = TRUE;
			break;
		}
	}

	if (bSlotFree == FALSE){
		WS_TRACE0("No free slots\n");
		closesocket(m_pServer->accept());
		m_strStatus = "Not accepting";
	} else
		m_strStatus = "Accepting";

	SetDlgItemText(IDC_STATUS,m_strStatus);

	return 0;
}