自己写的一个使用CSocket的实例

图片借用一下别人的，我的程序是根据这个改的，因为感觉其CNewSocket耦合性太强，所以自己重新写了CNewSocket类。原代码中使用了CSocketFile和CArchive，我认为使用这2个类显得累赘，于是去掉了。

------------------------------------------------NewSocket.h-----------------------------------

#define  WM_FD_ACCEPT WM_USER+1
#define  WM_FD_RECEIVE WM_USER+2
#define  WM_FD_CLOSE WM_USER+3

class  CNewSocket :  public  CSocket
... {
// Attributes
public:

// Operations
public:
    CNewSocket(HWND hwnd);
    virtual ~CNewSocket();

// Overrides
public:

    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CNewSocket)
    //}}AFX_VIRTUAL

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

    // Generated message map functions
    //{{AFX_MSG(CNewSocket)
        // NOTE - the ClassWizard will add and remove member functions here.
    //}}AFX_MSG
private:
    HWND m_hWnd;

// Implementation
protected:
} ;

 

 ------------------------------------------------NewSocket.CPP-----------------------------------

 

CNewSocket::CNewSocket(HWND hwnd)
... {
    m_hWnd = hwnd;
}

CNewSocket:: ~ CNewSocket()
... {
}

void  CNewSocket::OnAccept( int  nErrorCode)
... {
    AfxMessageBox("OnAccept");
    PostMessage(m_hWnd, WM_FD_ACCEPT, 0, 0);
    CSocket::OnAccept( nErrorCode );
}

void  CNewSocket::OnReceive( int  nErrorCode)
... {
    AfxMessageBox("OnReceive");
    PostMessage(m_hWnd, WM_FD_RECEIVE, 0, 0);
    
    CSocket::OnReceive( nErrorCode );
}

void  CNewSocket::OnClose( int  nErrorCode)
... {

    PostMessage(m_hWnd, WM_FD_CLOSE, 0, 0);
    CSocket::OnClose( nErrorCode );
}

 

 ------------------------------------------------SocketTestDlg.h-----------------------------------

CString    m_IpAddress;   // IP输入Edit框变量
CString    m_Connect;     // 提示信息框
int                   m_nStatus;      // 状态变量
UINT    m_Port;            // 端口输入Edit框变量
CString    m_InPut;          // 用于接收的Edit框
CString    m_OutPut;       // 用于发送的EDit框

// {{AFX_MSG(CSocketTestDlg)
…………
afx_msg  void  OnReceive();
afx_msg  void  OnAccept();
afx_msg  void  OnClose();
// }}AFX_MSG
DECLARE_MESSAGE_MAP()

 ------------------------------------------------SocketTestDlg.cpp-----------------------------------

CSocketTestDlg::CSocketTestDlg(CWnd *  pParent  /**/ /*=NULL*/ )
    : CDialog(CSocketTestDlg::IDD, pParent)
... {
    //{{AFX_DATA_INIT(CSocketTestDlg)
    m_IpAddress = _T("");
    m_Connect = _T("");
    m_nStatus = -1;
    m_Port = 0;
    m_InPut = _T("");
    m_OutPut = _T("");
    //}}AFX_DATA_INIT
    // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
BEGIN_MESSAGE_MAP(CSocketTestDlg, CDialog)
     // {{AFX_MSG_MAP(CSocketTestDlg)
// ………………
    ON_MESSAGE(WM_FD_ACCEPT, OnAccept)
    ON_MESSAGE(WM_FD_RECEIVE, OnReceive)
    ON_MESSAGE(WM_FD_CLOSE, OnClose)
     // }}AFX_MSG_MAP
END_MESSAGE_MAP()

BOOL CSocketTestDlg::OnInitDialog()
... {
     ………………
    m_nStatus = -1;
    m_ServerSocket = NULL;
    m_ClientSocket = NULL;
    m_Connect = "";
    m_IpAddress = "127.0.0.1";
    m_Port = 7897;
    
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(FALSE);
    GetDlgItem(IDC_PORTEDIT)->EnableWindow(FALSE);
}

void  CSocketTestDlg::OnConnect()
... {
    CString str_Msg;
    UpdateData(TRUE);

    if (0 == m_nStatus)         //如果为服务器
    ...{
        if (m_ServerSocket != NULL)     //服务器Socket是否已经创建
        ...{
            m_Connect = "Please disconnect!";
            UpdateData(FALSE);
        }
        else
        ...{
            m_Connect = "Wait for client...";
            UpdateData(FALSE);
            
            if (!AfxSocketInit())   //Socket初始化
            ...{
                MessageBox("WindowSocket initial failed!", "Send", MB_ICONSTOP);
                return;
            }
            //将Socket与窗口关联
            m_ServerSocket = new CNewSocket(this->m_hWnd);
            
            //如果创建Socket失败则提示，成功则开始监听
            if (!m_ServerSocket->Create(m_Port))
            ...{
                MessageBox("SendSocket Create failed!", "Send", MB_ICONSTOP);
            }
            else
            ...{
                m_ServerSocket->Listen();
            }
        }
    }
    else if (1 == m_nStatus)        //如果为客户端
    ...{
        if (m_ClientSocket != NULL) //判断客户端Socket是否已经创建
        ...{
            m_Connect = "Please disconnect!";
            UpdateData(FALSE);
        }
        else
        ...{
            m_Connect = "Connect to the server...";
            UpdateData(FALSE);
            if (!AfxSocketInit())
            ...{
                MessageBox("WindowSocket initial failed!", "Receive", MB_ICONSTOP);
                return;
            }
            m_ClientSocket = new CNewSocket(this->m_hWnd);; 
            if(!m_ClientSocket->Create())
            ...{ 
                MessageBox("ReceiveSocket create failed!","Receive",MB_ICONSTOP); 
                return; 
            }
            else
            ...{
                if (!m_ClientSocket->Connect(m_IpAddress, m_Port))
                ...{
                    CString strTmp = m_Connect;
                    SocketReset();
                    m_Connect = strTmp;
                    m_Connect += "Error!";
                    UpdateData(FALSE);
                }
                else
                ...{
                    m_Connect += "OK!";
                }
                
                UpdateData(FALSE);
            }
        }
    }

    if (-1 == m_nStatus)
    ...{
        str_Msg = "Please choose the status!";
        AfxMessageBox(str_Msg);
    }
}

void  CSocketTestDlg::OnDisconnect() 
... {
    SocketReset();
    m_Connect = "Disconnect!";
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnSend() 
... {
    UpdateData(TRUE);
    this->m_ClientSocket->Send(m_OutPut.GetBuffer(0), m_OutPut.GetLength());
}

void  CSocketTestDlg::OnSendclear() 
... {
    m_OutPut = "";
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnReceiveclear() 
... {
    m_InPut = "";
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnExit() 
... {
    CDialog::OnOK();
}

void  CSocketTestDlg::SocketReset()
... {
    if (m_ServerSocket != NULL)
    ...{
        delete m_ServerSocket;
        m_ServerSocket = NULL;
    }

    if (m_ClientSocket != NULL)
    ...{
        delete m_ClientSocket;
        m_ClientSocket = NULL;
    }

    m_Connect = "";
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnClose()
... {
    if (0 == m_nStatus)
    ...{
        m_Connect = "Client";
    }
    else
    ...{
        m_Connect = "Server";
    }
    m_Connect += "has disconnected";
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnAccept()
... {
    m_Connect += "OK!";
    UpdateData(FALSE);
    
    m_ClientSocket = new CNewSocket(this->m_hWnd);

    m_ServerSocket->Accept(*m_ClientSocket);
}

void  CSocketTestDlg::OnReceive()
... {
    char *buff = new char[1024];
    memset(buff, 0, 1024);
    m_ClientSocket->Receive(buff, 1024);
    AfxMessageBox(buff);
    delete []buff;
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnServerradio() 
... {
    UpdateData(TRUE);
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(FALSE);
    GetDlgItem(IDC_PORTEDIT)->EnableWindow(TRUE);
    UpdateData(FALSE);
}

void  CSocketTestDlg::OnClientradio() 
... {
    UpdateData(TRUE);
    GetDlgItem(IDC_IPADDRESS)->EnableWindow(TRUE);
    GetDlgItem(IDC_PORTEDIT)->EnableWindow(TRUE);
    UpdateData(FALSE);
}

 

使用方法：开启2个应用程序，一个选择Server，点击Connect，一个选择Cleint，点击Connect，然后就可以收发了消息了。如果不是本机测试，就改一下IP和端口。