Visual C++网络编程经典案例详解 第2章 Winsock网络程序开发流程 TCP服务器程序 功能实现

TCP服务器应该具有监听 发生和接收数据的功能
首先 用户应该为程序添加监听功能
因为 服务器必须等待客户端的连接请求到来之后
才能实现接收和发送数据
将服务器创建的套接字设置为异步模式
并将套接字事件设置为连接和读取事件

#define WM_SOCKET WM_USER+10//自定义套接字消息
class CTCPDlg : public CDialog
{
// Construction
public:
    CTCPDlg(CWnd* pParent = NULL);    // standard constructor
SOCKET s,s1;
sockaddr_in addr,add1;
CString str,str1;
int n;
// Dialog Data
    //{{AFX_DATA(CTCPDlg)
    enum { IDD = IDD_TCP_DIALOG };
        // NOTE: the ClassWizard will add data members here
    //}}AFX_DATA

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

// Implementation
protected:
    HICON m_hIcon;

    // Generated message map functions
    //{{AFX_MSG(CTCPDlg)
    virtual BOOL OnInitDialog();
    afx_msg void OnSysCommand(UINT nID, LPARAM lParam);
    afx_msg void OnPaint();
    afx_msg HCURSOR OnQueryDragIcon();
    afx_msg void OnSocket(WPARAM wParam,LPARAM lParam); //自定义消息响应函数
    afx_msg void OnSend();
    //}}AFX_MSG
    DECLARE_MESSAGE_MAP()
};

BOOL CTCPDlg::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
    SetIcon(m_hIcon, TRUE);            // Set big icon
    SetIcon(m_hIcon, FALSE);        // Set small icon
    
    // TODO: Add extra initialization here
    addr.sin_family=AF_INET;//填充套接字地址结构
    addr.sin_port=htons(80);
    addr.sin_addr.S_un.S_addr=INADDR_ANY;
    s=::socket(AF_INET,SOCK_STREAM,0);
    ::bind(s,(sockaddr*)&addr,sizeof(addr));//绑定本地地址
    ::listen(s,5);//监听端口
    ::WSAAsyncSelect(s,this->m_hWnd,WM_SOCKET,FD_ACCEPT|FD_READ);//设置异步套接字
    GetDlgItem(IDC_TEXT)->EnableWindow(false);
//    GetDlgItem(IDC_SENDTEXT)->SetWindowText("客户端，你好！");
    GetDlgItem(IDC_ADDR)->SetWindowText("服务器监听已经启动！");
//        GetDlgItem(IDC_TEXT)->SetWindowText("已经有1个客户端连接上服务器了 \r\n 客户端1说：你好！");
        
    GetDlgItem(IDC_TEXT)->GetWindowText(str);
        str+="\r\n";
        GetDlgItem(IDC_SENDTEXT)->GetWindowText(str1);
        str+=str1;
            GetDlgItem(IDC_TEXT)->SetWindowText(str);
    return TRUE;  // return TRUE  unless you set the focus to a control
}

当服务器监听套接字上有关的事件发生时
程序便会调用自定义的消息响应函数OnSocket()对该事件进行处理

void CTCPDlg::OnSocket(WPARAM wParam,LPARAM lParam)
{
    CString str13;//定义字符串变量
    char cs[100]={0};//定义字符数组
switch (lParam)
{
  case FD_ACCEPT://如果事件为连接事件
            {
         int lenth=sizeof(add1);     
         s1=::accept(s,(sockaddr*)&add1,&lenth);//应答客户端的连接请求
         n=n+1;//连接的客户端数目
         str13.Format("有%d客户已经连接上了",n);//格式化字符串
         GetDlgItem(IDC_TEXT)->SetWindowText(str13);//设置消息框内容
         str13+=::inet_ntoa(add1.sin_addr);//转换IP地址
         str13+="\r\n";
         str13+="登陆\r\n";
         GetDlgItem(IDC_TEXT)->SetWindowText(str13);//设置显示消息
             GetDlgItem(IDC_TEXT)->SetWindowText("\r\n客户端1说：你好！");
                            }
            break;//跳出该函数
        
        case FD_READ://处理读取事件
               { 
               CString num="";//定义字符串
               ::recv(s1,cs,100,0);//接收消息
                       GetDlgItem(IDC_TEXT)->GetWindowText(num);
                        num+="\r\n";
                        num+=(LPTSTR)::inet_ntoa(add1.sin_addr);//转换IP地址
                        num+="对您说：";
                         num+=(LPTSTR)cs;
                   GetDlgItem(IDC_TEXT)->SetWindowText(num);}
               
            break;}
      
}

用户在代码中
实现了服务器应答客户端的连接请求以及显示信息等功能
当有客户端向服务器发送连接请求时
服务器将显示相关信息

服务器显示接收到的信息标题

 

现在 服务器端已经应答服务器连接请求和接收客户端信息的功能
服务器还需要具有发送消息的功能
为“发送”按钮添加消息响应函数

函数OnSend()中
服务器程序应该将发送到客户端的消息也显示在服务器端界面中
用户在使用该功能时
对于信息的发送和接收理解比较直观

void CTCPDlg::OnSend() 
{
    // TODO: Add your control notification handler code here
    CString str,str1; //定义字符串
    GetDlgItem(IDC_SENDTEXT)->GetWindowText(str); //获取发送信息
    if(str=="")//消息不能为空
    {
        GetDlgItem(IDC_TEXT)->GetWindowText(str1);
        //str1+="\r\n";
        str1+="消息不能为空\r\n"; //显示提示信息
        GetDlgItem(IDC_TEXT)->SetWindowText(str1);
    }
    else
    {
        ::send(s,str.GetBuffer(1),sizeof(str),0);
        GetDlgItem(IDC_TEXT)->GetWindowText(str1);
        str1+="\r\n"; //在消息框中显示提示信息
        str1+=str;
        GetDlgItem(IDC_TEXT)->SetWindowText(str1);
    }
}

用户先调用函数获取发送消息框中的内容并将其存放在字符串变量str
如果发送的消息是空字符串
则提示用户不能发送空消息