领购物优惠券9块9包邮
首页 > 编程语言 > C/C++教程

C++之WSAAsyncSelect模型实例

   2015-06-28 0
核心提示：这篇文章主要介绍了C++的WSAAsyncSelect模型,实例讲述了socket与Windows消息机制的用法,需要的朋友可以参考下

本文实例讲述了C++中WSAAsyncSelect模型的用法。分享给大家供大家参考。具体实现方法如下：

TCPServer.cpp源文件如下：

复制代码 代码如下:
#include "TCPServer.h" 
#include "resource.h" 
 
#define WM_SOCKET WM_USER+1 
 
CMyApp theApp; 
 
BOOL CMyApp::InitInstance() 
{ 
    //初始化套接字 
    WSADATA wsaData; 
    WORD wVersionRequested = MAKEWORD(2,0); 
    ::WSAStartup(wVersionRequested, &wsaData); 
    //显示对话框 
    CMainDialog dlg; 
    m_pMainWnd = &dlg; 
    dlg.DoModal(); 
    //释放套接字 
    ::WSACleanup(); 
    return FALSE; 
} 
 
//CMainDialog 
CMainDialog::CMainDialog(CWnd* pParentWnd):CDialog(IDD_MAINDIALOG,pParentWnd) 
{ 
 
} 
BEGIN_MESSAGE_MAP(CMainDialog, CDialog) 
ON_BN_CLICKED(IDC_START, on Start) 
ON_BN_CLICKED(IDC_CLEAR, OnClear) 
ON_MESSAGE(WM_SOCKET, OnSocket) 
END_MESSAGE_MAP() 
 
void CMainDialog::OnCancel() 
{ 
    this->CloseAllSocket(); 
    CDialog::OnCancel(); 
} 
 
BOOL CMainDialog::OnInitDialog() 
{ 
    CDialog::OnInitDialog(); 
 
    //设置图标 
    SetIcon(theApp.LoadIconA(IDI_MAIN), FALSE); 
 
    //创建状态栏并设置其属性 
    m_bar.Create(WS_CHILD|WS_VISIBLE|SBS_SIZEGRIP, CRect(0,0,0,0), this, 101); 
    m_bar.SetBkColor(RGB(0xa6, 0xca, 0xfa)); 
    int arWidth[]={200,-1}; 
    m_bar.SetParts(2, arWidth); 
    m_bar.SetText("windows程序设计", 1, 0); 
    m_bar.SetText("空闲", 0, 0); 
    //关联列表控件 
    m_listInfo.SubclassDlgItem(IDC_LIST, this); 
 
    //初始化套接字和连接列表 
    m_socket = INVALID_SOCKET; 
    m_nClient = 0; 
 
    //取得本机IP，在状态栏中显示 
    char szHostName[MAX_PATH] = {0}; 
    ::gethostname(szHostName, MAX_PATH); 
    hostent *pHost = gethostbyname(szHostName); 
    if (pHost != NULL) 
    { 
        CString strIP; 
        in_addr* addr = (in_addr*)*pHost->h_addr_list; 
        strIP.Format("本机IP:%s",inet_ntoa(addr[0])); 
        m_bar.SetText(strIP, 0, 0); 
    } 
    return TRUE; 
} 
 
BOOL CMainDialog::CreateAndListen(int nPort) 
{ 
    if (m_socket == INVALID_SOCKET) 
    { 
        ::closesocket(m_socket); 
    } 
    //创建套接字 
    m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); 
    if (m_socket == INVALID_SOCKET) 
    { 
        return FALSE; 
    } 
    //绑定端口 
    sockaddr_in sin; 
    sin.sin_family = AF_INET; 
    sin.sin_port = htons(nPort); 
    //sin.sin_addr.S_un.S_addr = INADDR_ANY; 
    sin.sin_addr.s_addr = INADDR_ANY; 
    int nErr = GetLastError(); 
    if (::bind(m_socket, (sockaddr*)&sin, sizeof(sin)) == SOCKET_ERROR) 
    { 
        nErr = GetLastError(); 
        return FALSE; 
    } 
    ::WSAAsyncSelect(m_socket, m_hWnd, WM_SOCKET, FD_ACCEPT|FD_CLOSE|FD_READ); 
 
    //进入监听模式 
    ::listen(m_socket, 5); 
 
    return TRUE; 
} 
 
BOOL CMainDialog::AddClient(SOCKET s) 
{ 
     
    if (m_nClient < MAX_SOCKET) 
    { 
        m_arClient[m_nClient++] = s; 
        return TRUE; 
    } 
    return FALSE; 
     
} 
 
void CMainDialog::RemoveClient(SOCKET s) 
{ 
    BOOL bFound = FALSE; 
    int i; 
    for (i=0;i<m_nClient;i++) 
    { 
        if (m_arClient[i] == s) 
        { 
            bFound = TRUE; 
            break; 
        } 
    } 
 
    //找到 
    if (bFound) 
    { 
        m_nClient--; 
        for (int j=i;j<m_nClient;j++) 
        { 
            m_arClient[j] = m_arClient[j+1]; 
        } 
    } 
} 
void CMainDialog::CloseAllSocket() 
{ 
    if (m_socket != INVALID_SOCKET) 
    { 
        ::closesocket(m_socket); 
        m_socket = INVALID_SOCKET; 
    } 
    for (int i=0;i<m_nClient;i++) 
    { 
        ::closesocket(m_arClient[i]); 
    } 
    m_nClient = 0; 
} 
 
void CMainDialog::on Start() 
{ 
    if (m_socket == INVALID_SOCKET) //开启服务 
    { 
        CString strPort; 
        GetDlgItem(IDC_PORT)->GetWindowText(strPort); 
        int nPort = atoi(strPort); 
        if (nPort < 1 || nPort >65535) 
        { 
            MessageBox("port error"); 
            return; 
        } 
        //创建套接字 
        if (!this->CreateAndListen(nPort)) 
        { 
            MessageBox("create socket error"); 
            return; 
        } 
        //设置控件状态 
        GetDlgItem(IDC_START)->SetWindowTextA("停止服务"); 
        m_bar.SetText("正在监听...", 0, 0); 
        GetDlgItem(IDC_PORT)->EnableWindow(FALSE); 
    } 
    else //关闭服务 
    { 
        CloseAllSocket(); 
        GetDlgItem(IDC_START)->SetWindowTextA("开启服务"); 
        m_bar.SetText("空闲", 0, 0); 
        GetDlgItem(IDC_PORT)->EnableWindow(TRUE); 
    } 
    return ; 
} 
void CMainDialog::OnClear() 
{ 
    m_listInfo.ResetContent(); 
    return ; 
} 
 
long CMainDialog::OnSocket(WPARAM wParam, LPARAM lParam) 
{ 
    //得到句柄  
    SOCKET s = wParam; 
    //查看是否出错 
    if (WSAGETSELECTERROR(lParam)) 
    { 
        RemoveClient(s); 
        ::closesocket(s); 
        return 0; 
    } 
    //处理发生的事件 
    switch (WSAGETSELECTEVENT(lParam)) 
    { 
    case FD_ACCEPT: //监听到有套接字中有连接进入 
        { 
            MessageBox("server:accept"); 
            if (m_nClient < MAX_SOCKET) 
            { 
                SOCKET client = ::accept(s, NULL, NULL); 
                this->AddClient(client); 
            } 
            else 
            { 
                MessageBox("too many connection"); 
            } 
        } 
        break; 
    case FD_CLOSE: 
        { 
            MessageBox("server:close"); 
            RemoveClient(s); 
            closesocket(s); 
        } 
        break; 
    case FD_READ: //接收到对方发来的数据包 
        { 
            MessageBox("server:read"); 
            //得到对方的地址 
            sockaddr_in sockAddr; 
            memset(&sockAddr, 0, sizeof(sockAddr)); 
            int nSockAddrLength = sizeof(sockAddr); 
            ::getpeername(s, (sockaddr*)&sockAddr, &nSockAddrLength); 
 
            int nPeerPort = ntohs(sockAddr.sin_port); 
            CString strIP = inet_ntoa(sockAddr.sin_addr);  // strIP 
 
            //获得主机名称 
            DWORD dwIP = ::inet_addr(strIP); 
            hostent* pHost = ::gethostbyaddr((LPSTR)&dwIP, 4, AF_INET); 
            char szHostName[256]={0}; 
            strncpy(szHostName, pHost->h_name, 256); 
 
            //得到网络数据 
            char szContent[1024]={0}; 
            ::recv(s, szContent, 1024, 0); 
 
            //显示 
            CString strItem = CString(szHostName) + "[" + strIP + "]:" + CString(szContent); 
            m_listInfo.InsertString(0, strItem); 
        } 
        break; 
    } 
    return 0; 
}

TCPServer.h头文件如下：

复制代码 代码如下:
#include <afxwin.h> 
#include <afxext.h>  //CStatusBar 
#include <WinSock2.h> 
#include <afxcmn.h> 
 
#pragma comment(lib, "WS2_32.lib") 
#define  MAX_SOCKET 56 //最大客户量 
 
class CMyApp:public  CWinApp 
{ 
public: 
    BOOL InitInstance(); 
}; 
 
//CMainDialog 
class CMainDialog:public CDialog 
{ 
public: 
    CMainDialog(CWnd* pParentWnd=NULL); 
 
protected: 
    virtual BOOL OnInitDialog(); 
    virtual void OnCancel(); 
    //开启或停止服务 
    afx_msg void on Start(); 
    afx_msg void OnClear(); 
    afx_msg long OnSocket(WPARAM wParam, LPARAM lParam); 
 
    BOOL CreateAndListen(int nPort); 
 
    //向客户连接列表中加一个客户 
    BOOL AddClient(SOCKET s); 
    //从客户连接列表中移除一个客户 
    void RemoveClient(SOCKET s); 
    //关闭所有连接 
    void CloseAllSocket(); 
 
protected: 
    SOCKET m_socket; 
    //两个子窗口控件 
    CListBox m_listInfo; 
    CStatusBarCtrl m_bar; 
 
    //客户连接列表 
    SOCKET m_arClient[MAX_SOCKET]; //套接字列表 
    int m_nClient; //上述数组的大小 
 
    DECLARE_MESSAGE_MAP() 
};

TCPClient.cpp源文件如下：

复制代码 代码如下:
#include "TCPClient.h" 
#include "resource.h" 
 
#define WM_SOCKET WM_USER+1 
 
CMyApp theApp; 
 
BOOL CMyApp::InitInstance() 
{ 
    //初始化套接字 
    WSADATA wsaData; 
    WORD wVersionRequested = MAKEWORD(2,0); 
    ::WSAStartup(wVersionRequested, &wsaData); 
    //显示对话框 
    CMainDialog dlg; 
    m_pMainWnd = &dlg; 
    dlg.DoModal(); 
    //释放套接字 
    ::WSACleanup(); 
    return FALSE; 
} 
 
//CMainDialog 
CMainDialog::CMainDialog(CWnd* pParentWnd):CDialog(IDD_MAINDIALOG,pParentWnd) 
{ 
 
} 
BEGIN_MESSAGE_MAP(CMainDialog, CDialog) 
    ON_BN_CLICKED(IDC_CONNECT, OnConnect) 
    ON_BN_CLICKED(IDC_SEND, OnSend) 
    ON_MESSAGE(WM_SOCKET, OnSocket) 
END_MESSAGE_MAP() 
 
void CMainDialog::OnCancel() 
{ 
     
    CDialog::OnCancel(); 
} 
 
BOOL CMainDialog::OnInitDialog() 
{ 
    CDialog::OnInitDialog(); 
 
    //设置图标 
    SetIcon(theApp.LoadIconA(IDI_MAIN), FALSE); 
 
    //关联控件 
    m_edit_text.SubclassDlgItem(IDC_EDIT_CONTENT, this); 
    //状态栏 
    m_bar.Create(WS_CHILD|WS_VISIBLE|SBS_SIZEGRIP, CRect(0, 0, 0,0), this, NULL); 
    int nWidth[]={100,-1}; 
    m_bar.SetParts(2, nWidth); 
    m_bar.SetText("windows程序设计", 1, 0); 
    m_bar.SetText("空闲", 0, 0); 
 
    GetDlgItem(IDC_ADDR)->SetWindowTextA("192.168.19.143"); 
    GetDlgItem(IDC_PORT)->SetWindowTextA("9999"); 
 
    // 
    m_socket = INVALID_SOCKET; 
     
    return TRUE; 
} 
void CMainDialog::AddStringToList(CString strText) 
{ 
    CString strContent; 
    GetDlgItem(IDC_EDIT_CONTENT)->GetWindowText(strContent); 
    GetDlgItem(IDC_EDIT_CONTENT)->SetWindowText(strContent+strText); 
 
} 
long CMainDialog::OnSocket(WPARAM wParam, LPARAM lParam) 
{ 
    SOCKET  s = wParam; 
    if (WSAGETSELECTERROR(lParam)) 
    { 
        ::closesocket(m_socket); 
        m_socket = INVALID_SOCKET; 
        return 0; 
    } 
    switch (WSAGETSELECTEVENT(lParam)) 
    { 
    case FD_READ: 
        { 
            MessageBox("client:read"); 
            char szText[1024]={0}; 
            ::recv(s, szText, 1024, 0); 
            AddStringToList(CString(szText)+"\r\n"); 
        } 
        break; 
    case FD_CONNECT: 
        { 
            MessageBox("client:connect"); 
            GetDlgItem(IDC_CONNECT)->SetWindowTextA("断开连接"); 
            GetDlgItem(IDC_ADDR)->EnableWindow(FALSE); 
            GetDlgItem(IDC_PORT)->EnableWindow(FALSE); 
            GetDlgItem(IDC_TEXT)->EnableWindow(TRUE); 
            GetDlgItem(IDC_SEND)->EnableWindow(TRUE); 
            m_bar.SetText("已经连接到服务器", 0, 0); 
        } 
        break; 
    case FD_CLOSE: 
        { 
            MessageBox("client:close"); 
            OnConnect(); 
        } 
        break; 
    } 
    return 0; 
} 
 
BOOL CMainDialog::Connect(LPCTSTR pszRemoteAddr, u_short nPort) 
{ 
    //创建套接字 
    m_socket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); 
    if (INVALID_SOCKET == m_socket) 
    { 
        return FALSE; 
    } 
    ::WSAAsyncSelect(m_socket, m_hWnd, WM_SOCKET, FD_READ|FD_WRITE|FD_CONNECT|FD_CLOSE); 
 
    ULONG uAddr = ::inet_addr(pszRemoteAddr); 
    if (uAddr == INADDR_NONE) 
    { 
        //不是IP地址，就认为是主机名称 
        //从主机名得到IP 
        hostent* pHost = ::gethostbyname(pszRemoteAddr); 
        if (pHost == NULL) 
        { 
            ::closesocket(m_socket); 
            m_socket = INVALID_SOCKET; 
            return FALSE; 
        } 
        uAddr = ((struct in_addr*)*(pHost->h_addr_list))->s_addr; 
    } 
 
    //填写服务器信息 
    sockaddr_in remote; 
    remote.sin_family = AF_INET; 
    remote.sin_addr.S_un.S_addr = uAddr; 
    remote.sin_port = ::htons(nPort); 
    //连接 
    ::connect(m_socket, (sockaddr*)&remote, sizeof(sockaddr)); 
    return TRUE; 
} 
 
void CMainDialog::OnConnect() 
{ 
    if (INVALID_SOCKET == m_socket) //连接服务器 
    { 
        CString strAddr; 
        GetDlgItem(IDC_ADDR)->GetWindowText(strAddr); 
        if (strAddr.IsEmpty()) 
        { 
            MessageBox("the servers IP is empty"); 
            return; 
        } 
        CString strPort; 
        GetDlgItem(IDC_PORT)->GetWindowTextA(strPort); 
        int nPort = atoi(strPort); 
        if (nPort < 1 || nPort > 65535) 
        { 
            MessageBox("port error"); 
            return; 
        } 
        if (Connect(strAddr, nPort) == FALSE) 
        { 
            MessageBox("connect to servers error..."); 
            return; 
        } 
        //设置用户界面 
        GetDlgItem(IDC_CONNECT)->SetWindowText("取消"); 
        m_bar.SetText("正在连接..", 0, 0); 
         
    } 
    else //断开服务器 
    { 
        ::closesocket(m_socket); 
        m_socket = INVALID_SOCKET; 
        //设置用户界面 
        GetDlgItem(IDC_CONNECT)->SetWindowTextA("连接服务器"); 
        m_bar.SetText("空闲", 0, 0); 
        GetDlgItem(IDC_ADDR)->EnableWindow(TRUE); 
        GetDlgItem(IDC_PORT)->EnableWindow(TRUE); 
        GetDlgItem(IDC_SEND)->EnableWindow(FALSE); 
        GetDlgItem(IDC_TEXT)->EnableWindow(FALSE); 
    } 
     
    //this->Connect(szAddr, ) 
} 
void CMainDialog::OnSend() 
{ 
    CString strSendContent; 
    GetDlgItem(IDC_TEXT)->GetWindowTextA(strSendContent); 
    ::send(m_socket, strSendContent, strSendContent.GetLength(), 0); 
    GetDlgItem(IDC_TEXT)->SetWindowTextA(""); 
}

TCPClient.h头文件如下：

复制代码 代码如下:
#include <afxwin.h> 
#include <afxext.h>  //CStatusBar 
#include <WinSock2.h> 
#include <afxcmn.h> 
 
#pragma comment(lib, "WS2_32.lib") 
#define  MAX_SOCKET 56 //最大客户量 
 
class CMyApp:public  CWinApp 
{ 
public: 
    BOOL InitInstance(); 
}; 
 
 
//CMainDialog 
class CMainDialog:public CDialog 
{ 
public: 
    CMainDialog(CWnd* pParentWnd=NULL); 
 
protected: 
    virtual BOOL OnInitDialog(); 
    virtual void OnCancel(); 
    ////开启或停止服务 
    //afx_msg void on Start(); 
    afx_msg void OnSend(); 
    afx_msg long OnSocket(WPARAM wParam, LPARAM lParam); 
    void OnConnect(); 
 
    BOOL Connect(LPCTSTR pszRemoteAddr, u_short nPort); 
    SOCKET m_socket; 
 
    // 控件 
    CStatusBarCtrl m_bar; 
    CEdit m_edit_text; 
 
    void AddStringToList(CString strText); 
    //BOOL CreateAndListen(int nPort); 
 
    ////向客户连接列表中加一个客户 
    //BOOL AddClient(SOCKET s); 
    ////从客户连接列表中移除一个客户 
    //void RemoveClient(SOCKET s); 
    ////关闭所有连接 
    //void CloseAllSocket(); 
 
 
    DECLARE_MESSAGE_MAP() 
};

希望本文所述对大家的C++程序设计有所帮助。

 
标签： C++
反对 0举报 0 评论 0
 

免责声明：本文仅代表作者个人观点，与乐学笔记（本网）无关。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。
    本网站有部分内容均转载自其它媒体，转载目的在于传递更多信息，并不代表本网赞同其观点和对其真实性负责，若因作品内容、知识产权、版权和其他问题，请及时提供相关证明等材料并与我们留言联系，本网站将在规定时间内给予删除等相关处理.

  • Aurelius vs mORMot vs EntityDAC Delphi 的
    Aurelius vs mORMot vs EntityDAC   Delphi 的 ORM框架：http://www.tmssoftware.com/site/aurelius.asp#product-buy-onlinehttps://synopse.info/fossil/wiki/Synopse+OpenSourcehttps://www.devart.com/entitydac/download.htmlkbmMW  http://www.compo
    02-09
  • 【Ruby】Mac gem的一些坑
    前言自上一次升级MacOS系统后出现jekyll无法构建的问题，当时处理半天。谁知道最近又升级了MacOS，荒废博客多时，今天吝啬写了一篇准备发布，构建报错，问题重新。还是记录下，以防下次升级出问题。问题描述安装jekyll静态博客需要在Ruby环境下运行，于是参照
    02-09
  • iOS oc 调用 swift
    如股票oc要调用swift里面的代码 需要包含固定这个头文件项目名称 LiqunSwiftDemo-Swift.h         #ProjectName#-Swift.h固定的写法swift 目的 是取代oc 但是 不会完全取代 只是前端的替换LiqunSwiftDemo-Swift 点进去 可以看到 所有的swift代码 都产生
    02-09
  • objective-c NSTimer 定时器
    -(void)initTimer{//时间间隔NSTimeInterval timeInterval =3.0 ;//定时器repeats 表示是否需要重复，NO为只重复一次NSTimer *timer = [NSTimer scheduledTimerWithTimeInterval:timeInterval target:self selector:@selector(mobileAnimation) userInfo:nil
    02-09
  • Objective-C 日记③ 字符串
    Objective-C 日记③ 字符串
    一、创建字符串、类方法　　　公式创建NSString　　+(id) stringWithFormat:(NSString *) format,……;eg:　　NSString *height;　　height=[NSString stringWithFormat:@"高度是: %d 长度: %d",10,20];得到的字符串：“高度是: 10 长度: 20” 注意：　　省
    02-09
  • Objective-C KVC机制
    Objective-C KVC机制http://blog.csdn.net/omegayy/article/details/7381301全部推翻重写一个版本，这是我在公司内做技术分享的文档总结，对结构、条理做了更清晰的调整。 1.    基本概念MODEL主要是英文文档里面经常出现的一些概念，讲解一下，方便英文
    02-09
  • objective-c 加号 减号 - +
    “加号代表static”是错误的说法，可能跟你那样表达的人其实意思是：“前置加号的方法相当于Java 里面的静态方法”。在Oc中，方法分为类方法和实例方法。前置加号（+）的方法为类方法，这类方法是可以直接用类名来调用的，它的作用主要是创建一个实例。有人把
    02-09
  • Objective-C 日记①
    Objective-C 日记①
    1、Xcode的.m扩展名表示文件含有Object-C代码，C以.c文件，C++以.cpp文件2、头文件声明：C使用：#include，O-C使用#import（当然你也可以使用#include） 3、输出方式：　　C：printf(""，参数);　　O-C:NSLog(@""，参数); 4、布尔类型　　C：bool 具有true
    02-09
  • ASP.NET MVC 操作AD 获取域服务器当前用户姓
    #region 根据当前登录域账号 获取AD用户姓名和所在OU目录/// summary/// 根据当前登录域账号 获取AD用户姓名和所在OU目录/// /summary/// param name="searchUser"要搜索的当前用户名/param/// param name="paths"out返回该用户所在OU目录/param/// param nam
    02-09
  • swift和OC - 拆分数组 和 拆分字符串
    1. 拆分数组 /// 根据 数组 截取 指定个数返回 多个数组的集合func splitArray( array: [Date], withSubSize subSize: Int) - [[Date]] {//数组将被拆分成指定长度数组的个数let count = array.count% subSize == 0 ? (array.count/ subSize) : (array.count
    02-08
更多>PHP教程
php-fpm进程管理的三种模式 phpfpm子进程 [PHP8] 我参加了PHP8工程师认证初学者考试beta考试
更多>C/C++教程
VScode运行C++中文终端乱码的解决方案 vscode错误提示中文 C++ LeetCode1945题解字符串转化后的各位数字之和
更多>Java教程
Java项目导出数据为 PDF 文件的操作代码 盘点Java中延时任务的多种实现方式 java 延时队列怎么实现
更多>ASP.NET
使用WebClient自动填写并提交ASP.NET页面表单的源代码 asp.net mvc多条件+分页查询解决方案 asp.net mvc多条件+分页查询解决方案
更多>JSP编程
JSP通用高大上分页代码(超管用) JSP验证码动态生成方法
点击排行
网站首页  |  关于我们  |  联系方式  |  使用协议  |  版权隐私  |  网站地图  |  网站留言  |  违规举报