MFC Socket双向通信

记录点滴。

 

服务端部分程序：

//MySocket.h

// CMySocket 命令目标

#pragma once

#define SOCKET_EVENT    WM_USER + 11
#define WM_MYMESSAGE    WM_USER + 12 

// 同意连接        发送            接收            关闭
enum { ACCEPT = 0, SEND = 1, RETR = 2, CLOSE = 3 };


class CMySocket : public CSocket
{
public:
    CWnd *pWnd;

    CMySocket();
    virtual ~CMySocket();

    void AttachCWnd(CWnd *pWnd);
    virtual void OnSend(int nErrorCode);
    virtual void OnReceive(int nErrorCode);    
    virtual void OnClose(int nErrorCode);    
    virtual void OnAccept(int nErrorCode);    //接受连接

};

// MySocket.cpp : 实现文件
//

#include "stdafx.h"
#include "Socket_Server_HT.h"
#include "MySocket.h"


// CMySocket

CMySocket::CMySocket()
{
    pWnd = NULL;

}

CMySocket::~CMySocket()
{
}


// CMySocket 成员函数

void CMySocket::AttachCWnd(CWnd *pW)
{
    pWnd = pW;
}

void CMySocket::OnSend(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (WPARAM)this, SEND);

    CSocket::OnSend(nErrorCode);
}

void CMySocket::OnReceive(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (WPARAM)this, RETR);    

    // CSocket收到数据触发 OnReceive()函数
    CSocket::OnReceive(nErrorCode);
}

void CMySocket::OnClose(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (WPARAM)this, CLOSE);

    CSocket::OnClose(nErrorCode);
}

void CMySocket::OnAccept(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (WPARAM)this, ACCEPT);

    CSocket::OnAccept(nErrorCode);
}

// Socket_Server_HTDlg.h : 头文件
//

#pragma once

#include "MySocket.h"

// CSocket_Server_HTDlg 对话框
class CSocket_Server_HTDlg : public CDialogEx
{
// 构造
public:
    CSocket_Server_HTDlg(CWnd* pParent = NULL);    // 标准构造函数

// 对话框数据
    enum { IDD = IDD_SOCKET_SERVER_HT_DIALOG };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

    // 主 Socket
    CMySocket server_socket;

    // 接收发送数据 Socket
    CMySocket sock;

    void RecvText(CMySocket &from);
    void SendText(CMySocket &from);

// 实现
protected:
    HICON m_hIcon;

    // 生成的消息映射函数
    virtual BOOL OnInitDialog();
    afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
    afx_msg void OnPaint();

    // 根据消息参数解析网络事件类型
    LRESULT OnSocket(WPARAM wParam, LPARAM lParam);
    //LRESULT OnAccept(WPARAM wParam, LPARAM lParam);

    afx_msg HCURSOR OnQueryDragIcon();
    DECLARE_MESSAGE_MAP()
public:
    CString m_ReceiveStr;
    CString m_ReceiveStr_ID;
    CString m_SendStr;
    afx_msg void OnBnClickedButton1();
    afx_msg void OnEnChangeEdit2();
    
    afx_msg void OnBnClickedButton2();
    afx_msg void OnBnClickedButton3();
};

// Socket_Server_HTDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "Socket_Server_HT.h"
#include "Socket_Server_HTDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

// 对话框数据
    enum { IDD = IDD_ABOUTBOX };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CSocket_Server_HTDlg 对话框



CSocket_Server_HTDlg::CSocket_Server_HTDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CSocket_Server_HTDlg::IDD, pParent)
    , m_ReceiveStr(_T(""))
    , m_ReceiveStr_ID(_T(""))
    , m_SendStr(_T(""))
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSocket_Server_HTDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_EDIT1, m_ReceiveStr);
    DDX_Text(pDX, IDC_EDIT2, m_ReceiveStr_ID);
    DDX_Text(pDX, IDC_EDIT3, m_SendStr);
}

BEGIN_MESSAGE_MAP(CSocket_Server_HTDlg, CDialogEx)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_MESSAGE(SOCKET_EVENT, &CSocket_Server_HTDlg::OnSocket)
    ON_BN_CLICKED(IDC_BUTTON1, &CSocket_Server_HTDlg::OnBnClickedButton1)
    ON_EN_CHANGE(IDC_EDIT2, &CSocket_Server_HTDlg::OnEnChangeEdit2)
    ON_BN_CLICKED(IDC_BUTTON2, &CSocket_Server_HTDlg::OnBnClickedButton2)
    ON_BN_CLICKED(IDC_BUTTON3, &CSocket_Server_HTDlg::OnBnClickedButton3)
END_MESSAGE_MAP()


// CSocket_Server_HTDlg 消息处理程序

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

    // 将“关于...”菜单项添加到系统菜单中。

    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

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

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO:  在此添加额外的初始化代码

    server_socket.AttachCWnd(this);
    sock.AttachCWnd(this);    //将连接绑定到本窗体
    server_socket.Create(8001, SOCK_STREAM);// SOCK_STREAM表示TCP连接
    server_socket.Listen();        // 监听


    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CSocket_Server_HTDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialogEx::OnSysCommand(nID, lParam);
    }
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CSocket_Server_HTDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

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

        // 使图标在工作区矩形中居中
        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;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialogEx::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CSocket_Server_HTDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}

//PMSG pMsg = AfxGetCurrentMessage();

// wParam对应与相应客户端通信的SOCKET指针，也有可能是主SOCKET，lParam对应事件类型
LRESULT CSocket_Server_HTDlg::OnSocket(WPARAM wParam, LPARAM lParam)
{    
    SOCKADDR_IN sockAddr;//IP地址、端口号等
    int            nSize = sizeof(sockAddr);
    BOOL        res;

    switch (lParam)
    {
        //新的连接消息
    case ACCEPT:
        //主SOCKET指派新创建的SOCKET与客户端通信
        res = server_socket.Accept(sock,(SOCKADDR *)&sockAddr, &nSize);
        if (res == FALSE)
            MessageBox(_T("Accept Error!"));
        break;

    case SEND:                        //发送消息
        SendText(sock);
        break;

    case RETR:                        //收到数据消息
        RecvText(sock);                //解析消息
        break;

    case CLOSE:                        //连接关闭消息
        break;

    }
    return TRUE;
}


void CSocket_Server_HTDlg::RecvText(CMySocket &from)
{
    TCHAR RecvBuff[256];        //缓冲
    TCHAR RecvBuffName[256];
    TCHAR RecvBuffID[256];
    int len, i, j;                    //长度    
    CString temp1, temp2;

    memset(RecvBuff, 0, 256);
    memset(RecvBuffName, 0, 256);
    memset(RecvBuffID, 0, 256);

    // 读取数据
    len = from.Receive(RecvBuff, 256);    

    // 测试用，可以发送
    //from->Send(RecvBuff, 256);

    if (len < 1)                    
        return;

    for (i = 0; i < 256; i++)
    {// 为了使发过来的姓名和学号分开显示
        if (RecvBuff[i] != '@')
        {
            RecvBuffName[i] = RecvBuff[i];
        }
        else
        {
            break;
        }
    }

    ++i;
    for (j = 0; j < (256-i); j++)
    {
        RecvBuffID[j] = RecvBuff[i];
        ++i;
    }

    temp1.Format(_T("%s"), RecvBuffName);
    m_ReceiveStr += temp1;
    m_ReceiveStr.Append(_T("\r\n"));

    temp2.Format(_T("%s"), RecvBuffID);
    m_ReceiveStr_ID += temp2;
    m_ReceiveStr_ID.Append(_T("\r\n"));

    UpdateData(FALSE);
}

void CSocket_Server_HTDlg::SendText(CMySocket &from)
{
    UpdateData(TRUE);
    int len1 = m_SendStr.GetLength();
    char* p = (char*)m_SendStr.GetBuffer(0);

    from.Send(p, len1*2);

    //测试用
    //MessageBox(_T("Send OK"));
}

void CSocket_Server_HTDlg::OnBnClickedButton1()
{
    // TODO:  在此添加控件通知处理程序代码

    m_ReceiveStr = _T("");
    m_ReceiveStr_ID = _T("");
    UpdateData(FALSE);
}


void CSocket_Server_HTDlg::OnEnChangeEdit2()
{
    // TODO:  如果该控件是 RICHEDIT 控件，它将不
    // 发送此通知，除非重写 CDialogEx::OnInitDialog()
    // 函数并调用 CRichEditCtrl().SetEventMask()，
    // 同时将 ENM_CHANGE 标志“或”运算到掩码中。

    // TODO:  在此添加控件通知处理程序代码
}


void CSocket_Server_HTDlg::OnBnClickedButton2()
{
    // TODO:  在此添加控件通知处理程序代码

    // 模拟发送消息
    server_socket.OnSend(0);

}


void CSocket_Server_HTDlg::OnBnClickedButton3()
{
    // TODO:  在此添加控件通知处理程序代码

    m_SendStr = _T("");
    UpdateData(FALSE);
}

 

 

客户端部分程序：

/*
2015.4 HT
将Socket封装成适合本工程的模块

MySocket.h
*/

#pragma once

#define SOCKET_EVENT WM_USER + 10

enum { ACCEPT = 0, SEND = 1, RETR = 2, CLOSE = 3 };

class MySocket : public CSocket
{
public:
    MySocket();
    virtual ~MySocket();

    // 绑定的窗体
    CWnd *pWnd;

    // 窗体绑定
    void AttachCWnd(CWnd *pWnd);

    virtual void OnReceive(int nErrorCode);
    virtual void OnClose(int nErrorCode);

};

// MySocket.cpp : 实现文件
//

#include "stdafx.h"
#include "Socket_Client_HT.h"
#include "MySocket.h"


// MySocket

MySocket::MySocket()
{
    pWnd = NULL;
}

MySocket::~MySocket()
{
}


// MySocket 成员函数

// CWnd是WINDOWS的一个基类，对话框，按钮等都是从它派生过来
void MySocket::AttachCWnd(CWnd *pW)    
{
    pWnd = pW;
}

void MySocket::OnReceive(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (long)this, RETR);

    CSocket::OnReceive(nErrorCode);
}

void MySocket::OnClose(int nErrorCode)
{
    if (pWnd)
        pWnd->SendMessage(SOCKET_EVENT, (long)this, CLOSE);

    CSocket::OnClose(nErrorCode);
}

// Socket_Client_HTDlg.h : 头文件
//

#pragma once

#include "MySocket.h"
#include "afxcmn.h"


// CSocket_Client_HTDlg 对话框
class CSocket_Client_HTDlg : public CDialogEx
{
// 构造
public:

    
    MySocket clientSocket;    // 创建自定义客户端socket
    CString m_ipAddr;        // IP 地址

    CSocket_Client_HTDlg(CWnd* pParent = NULL);    // 标准构造函数

// 对话框数据
    enum { IDD = IDD_SOCKET_CLIENT_HT_DIALOG };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持


// 实现
protected:
    HICON m_hIcon;

    // 生成的消息映射函数
    virtual BOOL OnInitDialog();
    afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
    afx_msg void OnPaint();

    // 自定义消息接收函数
    afx_msg LRESULT RecvText(WPARAM wParam, LPARAM lParam);

    afx_msg HCURSOR OnQueryDragIcon();
    DECLARE_MESSAGE_MAP()
public:
    CIPAddressCtrl m_IP;
    CString m_SendStr;
    CString m_SendStrNum;
    CString m_ReceiveStr;
    afx_msg void OnBnClickedButton1();
    afx_msg void OnBnClickedButton2();
    afx_msg void OnBnClickedButton3();
    afx_msg void OnBnClickedButton4();
    
};

// Socket_Client_HTDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "Socket_Client_HT.h"
#include "Socket_Client_HTDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

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

// 对话框数据
    enum { IDD = IDD_ABOUTBOX };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()


// CSocket_Client_HTDlg 对话框



CSocket_Client_HTDlg::CSocket_Client_HTDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CSocket_Client_HTDlg::IDD, pParent)
    , m_SendStr(_T(""))
    , m_ReceiveStr(_T(""))
    , m_SendStrNum(_T(""))
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSocket_Client_HTDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Control(pDX, IDC_IPADDRESS1, m_IP);
    DDX_Text(pDX, IDC_EDIT1, m_SendStr);
    DDX_Text(pDX, IDC_EDIT2, m_ReceiveStr);
    DDX_Text(pDX, IDC_EDIT3, m_SendStrNum);
}

BEGIN_MESSAGE_MAP(CSocket_Client_HTDlg, CDialogEx)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_MESSAGE(SOCKET_EVENT, &CSocket_Client_HTDlg::RecvText)
    ON_BN_CLICKED(IDC_BUTTON1, &CSocket_Client_HTDlg::OnBnClickedButton1)
    ON_BN_CLICKED(IDC_BUTTON2, &CSocket_Client_HTDlg::OnBnClickedButton2)
    ON_BN_CLICKED(IDC_BUTTON3, &CSocket_Client_HTDlg::OnBnClickedButton3)
    ON_BN_CLICKED(IDC_BUTTON4, &CSocket_Client_HTDlg::OnBnClickedButton4)
END_MESSAGE_MAP()


// CSocket_Client_HTDlg 消息处理程序

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

    // 将“关于...”菜单项添加到系统菜单中。

    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

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

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO:  在此添加额外的初始化代码

    
    m_IP.SetWindowText(_T("127.0.0.1"));

    // 把SOCKET与对话框联系起来，SOCKET有消息就通知本对话框
    clientSocket.AttachCWnd(this);        
    // SOCKET初始化
    clientSocket.Create();

    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CSocket_Client_HTDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
    if ((nID & 0xFFF0) == IDM_ABOUTBOX)
    {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    }
    else
    {
        CDialogEx::OnSysCommand(nID, lParam);
    }
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CSocket_Client_HTDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

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

        // 使图标在工作区矩形中居中
        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;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialogEx::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CSocket_Client_HTDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}

int Alen = 0;            //总长度
LRESULT CSocket_Client_HTDlg::RecvText(WPARAM wParam, LPARAM lParam)
{
    TCHAR RecvBuff[256];        //缓冲
    int len;                    //长度    
    CString temp;
    memset(RecvBuff, 0, 256);
    if (RETR == lParam)
    {
        // 读取数据
        len = clientSocket.Receive(RecvBuff, 256);
        Alen += len;
        if (len < 1)
            return TRUE;
        temp.Format(_T("%s"), RecvBuff);
        m_ReceiveStr += temp;
        //m_ReceiveStr.Append(_T("\r\n"));
        UpdateData(FALSE);
    }

    if (Alen > 56)
    {
        Alen = 0;
        m_ReceiveStr.Append(_T("\r\n"));
        UpdateData(FALSE);
    }

    //UpdateData(FALSE);
    return TRUE;
}



void CSocket_Client_HTDlg::OnBnClickedButton1()
{
    // TODO:  在此添加控件通知处理程序代码

    // 发送框
    UpdateData(TRUE);

    int len1 = m_SendStr.GetLength();
    int len2 = m_SendStrNum.GetLength();
    char* p = (char*)m_SendStr.GetBuffer(0);
    char* q = (char*)m_SendStrNum.GetBuffer(0);

    clientSocket.Send(p, len1*2);
    clientSocket.Send("@",2);
    clientSocket.Send(q, len2*2);

    UpdateData(FALSE);
}


void CSocket_Client_HTDlg::OnBnClickedButton2()
{
    // TODO:  在此添加控件通知处理程序代码

    m_SendStr = _T("");
    m_SendStrNum = _T("");
    UpdateData(FALSE);
}


void CSocket_Client_HTDlg::OnBnClickedButton3()
{
    // TODO:  在此添加控件通知处理程序代码

    m_ReceiveStr = _T("");
    UpdateData(FALSE);
}


void CSocket_Client_HTDlg::OnBnClickedButton4()
{
    // TODO:  在此添加控件通知处理程序代码

    UpdateData(TRUE);
    LPTSTR lpstr = 0;
    m_IP.GetWindowText(lpstr, 16);
    // 获得IP地址
    m_ipAddr = lpstr;

    // Socket连接服务器
    if (clientSocket.Connect(m_ipAddr, 8001) == FALSE)
    {    
        MessageBox(_T("连接失败！"));
    }
    
}