select监听多个socket的例子

         先贴上代码,回头再仔细编辑。

         server.cpp:

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <fcntl.h>

#include <vector>
using namespace std;

int main(int argc,char* argv[])
{
    printf("server start up\n");

    if(argc <= 3)
    {
        printf("usage:%s ip port backlog\n",basename(argv[0]));
        return 1;
    }

    //IP地址
    const char* ip = argv[1];
    //端口号
    int port = atoi(argv[2]);
    //内核监听队列的最大长度(完全连接的socket)
    int backlog = atoi(argv[3]);

    //创建socket	(TIP/IP协议族,流式socket)
    int server_sockfd = socket(PF_INET,SOCK_STREAM,0);

    //最大连接数,初始化为server_sockfd
    int max_fd = server_sockfd;

    //
    vector<int>	 connFds;
    connFds.reserve(backlog);

    //TCP/IP协议族的socket地址结构体
    struct sockaddr_in server_addr;
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family = AF_INET;			//TCP/IPv4的地址族
    inet_pton(AF_INET,ip,&server_addr.sin_addr);		//将IP地址字符串转换为二进制的整数并赋给addr.sin_server_addr
    server_addr.sin_port = htons(port);			//端口,host to net,将主机字节序(小端)转换为网络字节序(大端)

    //将文件描述符sock和socket地址关联,仅服务端需要,客户端自动绑定地址
    //注意需要强制转换为 struct sockaddr*
    int ret = bind(server_sockfd,(struct sockaddr*)&server_addr,sizeof(server_addr));
    assert(ret != -1);

    //监听
    ret = listen(server_sockfd,backlog);
    assert(ret != -1);

    //客户端地址信息
    struct sockaddr_in client_addr;
    socklen_t client_addrlength = sizeof(client_addr);

    char buf[1024];
    memset(buf,0,sizeof(buf));

        //文件描述符集合
    fd_set read_fds;
    fd_set  exception_fds;
    FD_ZERO(&read_fds);
    FD_ZERO(&exception_fds);

        //超时
    struct timeval timeout;
    timeout.tv_sec = 30;             //妙
    timeout.tv_usec = 0 ;            //微秒

    while(1)
    {
        FD_SET(server_sockfd,&read_fds);
        for(int i =0;i<connFds.size();++i)
        {
            FD_SET(connFds[i],&read_fds);
            FD_SET(connFds[i],&exception_fds);
        }

        int ret = select(max_fd+1,&read_fds,NULL,&exception_fds,NULL);
        if( ret  ==  -1)
        {
            perror("select");
            break;
        }
        else if( ret == 0)
        {
            perror("timeout");
            break;
        }

        for(vector<int>::iterator it = connFds.begin();it != connFds.end();)
        {
            memset(buf,0,sizeof(buf));

            if( FD_ISSET(*it,&read_fds) )
            {
                int ret = recv(*it,buf,sizeof(buf)-1,0);
                if(ret <= 0)
                {
                    perror("recv");
                    close(*it);
                    FD_CLR(*it,&read_fds);
                    it = connFds.erase(it);
                    printf("A client disconnectd\n");
                    continue;
                }

                    buf[ret] = '\0';
                    printf("%s",buf);
            }
            else  if(FD_ISSET(*it,&exception_fds))
            {
                int ret = recv(*it,buf,sizeof(buf)-1,MSG_OOB);
                if(ret < 0)
                {
                    perror("exception");
                }
                else
                {
                    buf[ret] = '\0';
                    printf("%s",buf);
                }

            }

            ++it;
        }

        if( FD_ISSET(server_sockfd,&read_fds) )
        {
            //接受连接,并将被接受的远端sock地址信息保存在第二个参数中
            int newfd = accept(server_sockfd,(struct sockaddr*)&client_addr,&client_addrlength);

            if(newfd < 0)
            {
                perror("accept");
                return -1;
            }
            else
            {
                connFds.push_back(newfd);
                max_fd = newfd;
                //inet_ntoa(struct addr_in) 将IP地址转换为字符串并返回
                //只是从监听队列中取出连接,即使客户端已经断开网络连接也会accept成功
                printf("accept client_addr %s\n",inet_ntoa(client_addr.sin_addr));
            }
        }

    }


    //关闭连接,实际只是socket的引用-1,必须引用为0才会真正关闭
    for(int i= 0;i< connFds.size();++i)
    {
        close(connFds[i]);
    }

    close(server_sockfd);
    return 0;
}

在 Python 中,使用 `select` 模块可以实现对多个 socket 连接的同时监听,从而实现高效的网络编程。 下面是一个基本的使用 `select` 的例子: ```python import select import socket # 创建 TCP socket server = socket.socket(socket.AF_INET, socket.SOCK_STREAM) server.setblocking(False) # 绑定 IP 和端口 server_address = ('localhost', 10000) server.bind(server_address) # 启动监听 server.listen(5) # 创建 epoll 对象 epoll = select.epoll() # 注册 server socket 到 epoll 中 epoll.register(server.fileno(), select.EPOLLIN) # 事件循环 while True: # 等待 epoll 事件 events = epoll.poll() # 遍历事件 for fileno, event in events: # 如果是 server socket 有新连接到来 if fileno == server.fileno(): connection, client_address = server.accept() connection.setblocking(False) epoll.register(connection.fileno(), select.EPOLLIN) # 如果是已连接的 socket 有数据到达 elif event & select.EPOLLIN: data = connection.recv(1024) if data: print(f'Received data: {data.decode()}') else: epoll.unregister(fileno) connection.close() ``` 上述代码中,我们首先创建了一个 TCP socket,然后将其绑定到指定的 IP 和端口,并启动监听。接下来,我们创建了一个 epoll 对象,并将 server socket 注册到 epoll 中。然后进入事件循环,等待 epoll 事件,如果是 server socket 有新连接到来,我们就接受连接,并将新连接的 socket 也注册到 epoll 中。如果是已连接的 socket 有数据到达,我们就接收数据并处理。如果数据为空,说明连接已关闭,我们就将该 socket 从 epoll 中注销,并关闭该连接。 需要注意的是,在使用 `select` 和 `epoll` 时,需要确保所有 socket 都是非阻塞的,否则会阻塞整个程序。在上述代码中,我们通过 `setblocking(False)` 方法将 server socket 和连接 socket 都设置为非阻塞的。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值