serverdlg.cpp

网络编程最基本最常用的网络端和客户端的程序实现

// ServerDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Server.h"
#include "ServerDlg.h"
#include "setup.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
//
/****************************************************************************
*   显示错误信息
*   输入参数
*         hDlg : 窗口的Handle
*   nErrorCode : 调用AsyncSocket成员函数时, 发生错误的错误值
*****************************************************************************/
void ShowErrorMessage(HWND hDlg, int nErrorCode)
{
    LPVOID lpMsgBuf;
//
    FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | 
                  FORMAT_MESSAGE_IGNORE_INSERTS,
                  NULL,
                  nErrorCode,
                  MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
                  (LPTSTR)&lpMsgBuf,
                  0, NULL);
    ::MessageBox(hDlg, (LPCTSTR)lpMsgBuf, "错误", MB_OK | MB_ICONSTOP);
    LocalFree(lpMsgBuf);
};
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
    CAboutDlg();

// Dialog Data
    //{{AFX_DATA(CAboutDlg)
    enum { IDD = IDD_ABOUTBOX };
    //}}AFX_DATA

    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CAboutDlg)
    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
    //}}AFX_VIRTUAL

// Implementation
protected:
    //{{AFX_MSG(CAboutDlg)
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};
//
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
    //{{AFX_DATA_INIT(CAboutDlg)
    //}}AFX_DATA_INIT
};
//
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CAboutDlg)
    //}}AFX_DATA_MAP
};
//
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
    //{{AFX_MSG_MAP(CAboutDlg)
        // No message handlers
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CServerDlg dialog

CServerDlg::CServerDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CServerDlg::IDD, pParent)
{
    //{{AFX_DATA_INIT(CServerDlg)
    // NOTE: the ClassWizard will add member initialization here
    //}}AFX_DATA_INIT
    // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
    m_lpszName = _T("");
    m_pListen = NULL;
    m_pSocket = NULL;
};
//
void CServerDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialog::DoDataExchange(pDX);
    //{{AFX_DATA_MAP(CServerDlg)
    DDX_Control(pDX, IDC_SENDMSG, m_Sendmsg);
    DDX_Control(pDX, IDC_LISTMSG, m_Listmsg);
    //}}AFX_DATA_MAP
};
//
BEGIN_MESSAGE_MAP(CServerDlg, CDialog)
    //{{AFX_MSG_MAP(CServerDlg)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_WM_DESTROY()
    ON_BN_CLICKED(IDC_DISCONNECT, OnDisconnect)
    ON_BN_CLICKED(IDC_SENDBTN, OnSendbtn)
    ON_COMMAND(IDM_SETUPCONNECT, OnSetupconnect)
    ON_EN_CHANGE(IDC_SENDMSG, OnChange)
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CServerDlg message handlers

BOOL CServerDlg::OnInitDialog()
{
    CDialog::OnInitDialog();

    // Add "About..." menu item to system menu.

    // IDM_ABOUTBOX must be in the system command range.
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != NULL)
    {
        CString strAboutMenu;
        strAboutMenu.LoadString(IDS_ABOUTBOX);
        if (!strAboutMenu.IsEmpty())
        {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        };
    };

    // Set the icon for this dialog.  The framework does this automatically
    //  when the application's main window is not a dialog
    SetupIcon(0);
	
    // TODO: Add extra initialization here
    DWORD dwStyle = ::GetClassLong(m_hWnd, GCL_STYLE);
    dwStyle |= CS_NOCLOSE;
    ::SetClassLong(m_hWnd, GCL_STYLE, dwStyle);
    m_Sendmsg.SetEventMask(m_Sendmsg.GetEventMask() | ENM_CHANGE);
    // 设置处理CHANGE的事件
//
    return TRUE;  // return TRUE  unless you set the focus to a control
};
//
void CServerDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialog::OnSysCommand(nID, lParam);
    };
};

// 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 CServerDlg::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 CServerDlg::OnQueryDragIcon()
{
    return (HCURSOR) m_hIcon;
};
//
void CServerDlg::OnDestroy() 
{
    CDialog::OnDestroy();
    if (m_pSocket)
        m_pSocket->ShutDown();
    if (m_pListen)
        m_pListen->Close();
    delete m_pSocket;
    delete m_pListen;
};
//
/********************************************************************************
*  终止目前的连接
*  输入参数 : 无
*********************************************************************************/
void CServerDlg::OnDisconnect() 
{
    if (m_pSocket)
    {
        m_pSocket->ShutDown();
        // 停止连接并释放资源, Socket 并未关闭
        delete m_pSocket;
        m_pSocket = NULL;
    };
    CMenu *menu = GetMenu();  // 更改选项(MENU) 的状态
    if (menu)
        menu->EnableMenuItem(IDM_SETUPCONNECT, MF_BYCOMMAND | MF_ENABLED);
    CString str = _T("Server - [未连接]");
    SetWindowText(str);
    SetupIcon(0);
    m_Sendmsg.SetOptions(ECOOP_SET, ECO_READONLY);
    // 因为未连接,所以将的CRichEditCtrl控制项设成只读
};
//
/********************************************************************************
*                         按下传送信息的按钮
*     输入参数 : 无 
*********************************************************************************/

void CServerDlg::OnSendbtn() 
{
    if (!m_pSocket)
        return;
    int len;
    int sent;
    CString s, s1 = m_lpszName;
//
    m_Sendmsg.GetWindowText(s);
    s1 += _T(" Say :\n") + s;
    len = s1.GetLength();
    sent = m_pSocket->Send((LPCTSTR)s1, len);
    // 调用 CAsyncSocket 的成员函数 Sned()来传送信息
    // 如果成功传送, 则传回传送的字符数,
    // 否则传回 SOCKET_ERROR 此时可用 GetLastError() 取得信息的错误值 
    if (sent == SOCKET_ERROR)
    {
        ShowErrorMessage(GetSafeHwnd(), GetLastError());
        return;
    }
    else
    {
        m_Listmsg.GetWindowText(s1);
        if (s1.IsEmpty())
            s1 = CString(_T("----->")) + s;
        else
            s1 += CString(_T("\n----->")) + s;
        m_Listmsg.SetWindowText(s1);    // 将传送的信息显示在窗口中
        m_Sendmsg.SetWindowText(_T(""));
    };
};
//
/*******************************************************************************
*                          设置连接(主机及连接端口)
*      输入参数 : 无
********************************************************************************/

void CServerDlg::OnSetupconnect() 
{
    CSetupDlg dlg;
//
    if (dlg.DoModal() == IDOK)   // 如果以设置完成
    {
        m_pListen = new _tagAsyncSocket(this);
        // 配置 ASyncSocket 的资源
        if (!m_pListen->Create(dlg.m_nPort))
        // 服务器端使用时, 一定要指定连接端口
        // 如果无法开启WinSock
        {
            ShowErrorMessage(m_hWnd, GetLastError());
            delete m_pListen;
            m_pListen = NULL;
        };
        if (!m_pListen->Listen())
        // 如果无法等候连接
        {
            ShowErrorMessage(m_hWnd, GetLastError());
            delete m_pListen;
            m_pListen = NULL;
        };
        CString str;
        GetWindowText(str);
        str += _T(" - ") + dlg.m_lpszName + _T("[等待连接中...]");
        SetWindowText(str);
        m_lpszName = dlg.m_lpszName;
        CMenu* menu = GetMenu();
        if (menu)
            menu->EnableMenuItem(IDM_SETUPCONNECT, MF_BYCOMMAND | MF_DISABLED | MF_GRAYED);
        SetupIcon();
        //更换图示， 代表目前正在等候连接
        m_Sendmsg.SetOptions(ECOOP_XOR, ECO_READONLY);
        // 已经连接,所以取消CRichEditCtrl控制项的只读设置
    };
};
//
/*********************************************************************************
*            接受连接的处理程序 : 此函数是由被 Overlap 函数 OnAccept() 所调用
*       输入参数 :
*          nErrorCode : 0 为正确 
*                       非 "零" 值错误
*     注意 : 此参数在此函数中不需处理
**********************************************************************************/

void CServerDlg::OnAccept(int nErrorCode)
{
    m_pSocket = new _tagAsyncSocket(this);
    // 配置 ASyncSocket 的资源
    // 要让 Accept()这个CAsyncSocket的成员函数使用
    if (!m_pListen->Accept(*m_pSocket))
    // 如果发生错误
    {
        ShowErrorMessage(m_hWnd, GetLastError());
        delete m_pSocket;   // 释放之前所配置的资源
        m_pSocket = NULL;
        return;
    };
    CString str = _T("Server - ") + m_lpszName;
    str += _T("[连接交谈中...]");
    SetWindowText(str);
    SetupIcon(1);
};
//
/*************************************************************************************
*           接收信息的处理程序 : 此函数是由被 Overlap 函数 OnReceive() 所调用
*     输入参数 :
*          nErrorCode : 0 为正确 
*                       非 "零" 值错误
*     注意 : 此参数在此函数中不需处理
**************************************************************************************/

void CServerDlg::OnReceive(int nErrorCode)
{
    int len = 0, size = 1024;
    _TCHAR *buf = new _TCHAR[1024];
//
    ::memset(buf, 0, sizeof(_TCHAR)*1024);
    len = m_pSocket->Receive(buf, size);
    // 设置接受信息的最大长度
    // 如果成功则传回接收的数据长度, 否则传回 SOCKET_ERROR
    if (len == SOCKET_ERROR)
    {
        ShowErrorMessage(m_hWnd, GetLastError());
        return;
    };
//
    CString s;
    m_Listmsg.GetWindowText(s);
    s += _T("\n") + CString(buf);
    m_Listmsg.SetWindowText(s);
    // 将接收的信息加至窗口中
};
//
/**********************************************************************************
*                   设置连接时的图示
*      输入参数 :
*           bID : 0 未启动连接服务
*               : 1 连接中 
*               : 2 等候连接, 此值为内定值
***********************************************************************************/

void CServerDlg::SetupIcon(int nID)
{
    UINT nIcon;
//
    switch(nID)
    {
        case 0:
            nIcon = IDI_TRFF10C;
            break;
        case 1:
            nIcon = IDI_NET08;
            break;
        case 2:
            nIcon = IDI_NET10A;
            break;
    };
//
    m_hIcon = AfxGetApp()->LoadIcon(nIcon);
    SetIcon(m_hIcon, TRUE);
    SetIcon(m_hIcon, FALSE);
};
//
/*********************************************************************************
*                    输入的信息窗口内容被改变时
*       输入参数 : 无
**********************************************************************************/
void CServerDlg::OnChange() 
{
    CString str;
//
    m_Sendmsg.GetWindowText(str);
    str.IsEmpty() ? GetDlgItem(IDC_SENDBTN)->EnableWindow(FALSE)
                  : GetDlgItem(IDC_SENDBTN)->EnableWindow();
};