26-LINUX--I/O复用-select

一.I/O复用概述

        /O复用使得多个程序能够同时监听多个文件描述符,对提高程序的性能有很大帮助。以下情况适用于I/O复用技术:

TCP 服务器同时要处理监听套接字和连接套接字。
服务器要同时处理 TCP 请求和 UDP 请求。
程序要同时处理多个套接字。
客户端程序要同时处理用户输入和网络连接。
服务器要同时监听多个端口
        需要指出的是,I/O 复用虽然能同时监听多个文件描述符,但它本身是阻塞的。并且当
多个文件描述符同时就绪时,如果不采取额外的措施,程序就只能按顺序依处理其中的每一
个文件描述符,这使得服务器看起来好像是串行工作的。如果要提高并发处理的能力,可以
配合使用多线程或多进程等编程方法

二.select机制

1.select接口介绍

        select 系统调用的用途是:在一段指定时间内,监听用户感兴趣的文件描述符的可读、
可写和异常等事件。
        select 系统调用的原型如下:
 #include <sys/select.h>

 int select(int maxfd, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct ti
meval *timeout);
 /*
 select 成功时返回就绪(可读、可写和异常)文件描述符的总数。如果在超时时间内
没有任何文件描述符就绪,select 将返回 0。select 失败是返回-1.如果在 select 等待
期间,程序接收到信号,则 select 立即返回-1,并设置 errno 为 EINTR。

 maxfd 参数指定的被监听的文件描述符的总数。它通常被设置为 select 监听的所
有文件描述符中的最大值+1
 readfds、writefds 和 exceptfds 参数分别指向可读、可写和异常等事件对应的文件
描述符集合。应用程序调用 select 函数时,通过这 3 个参数传入自己感兴趣的文件
描述符。select 返回时,内核将修改它们来通知应用程序哪些文件描述符已经就绪
fd_set 结构如下:
 #define __FD_SETSIZE 1024
 typedef long int __fd_mask;
 #define __NFDBITS (8 * (int) sizeof (__fd_mask))
 typedef struct
 {
 #ifdef __USE_XOPEN
 __fd_mask fds_bits[__FD_SETSIZE / __NFDBITS];
 # define __FDS_BITS(set) ((set)->fds_bits)
 #else
 __fd_mask __fds_bits[__FD_SETSIZE / __NFDBITS];
 # define __FDS_BITS(set) ((set)->__fds_bits)
 #endif
 } fd_set;
 通过下列宏可以访问 fd_set 结构中的位:
 FD_ZERO(fd_set *fdset); // 清除 fdset 的所有位
 FD_SET(int fd, fd_set *fdset); // 设置 fdset 的位 fd

 FD_CLR(int fd, fd_set *fdset); // 清除 fdset 的位 fd
 int FD_ISSET(int fd, fd_set *fdset);// 测试 fdset 的位 fd 是否被设置
 timeout 参数用来设置 select 函数的超时时间。它是一个 timeval 结构类型的指
 针,采用指针参数是因为内核将修改它以告诉应用程序 select 等待了多久。timeval
 结构的定义如下:
 struct timeval
 {
 long tv_sec; //秒数
 long tv_usec; // 微秒数
 };//struct timeval tv = {5,0};
 如果给 timeout 的两个成员都是 0,则 select 将立即返回。如果 timeout 传递
NULL,则 select 将一直阻塞,直到某个文件描述符就绪
 */

2.设计思路图解

3.测试代码

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<sys/select.h>
#include<time.h>

#define STDIN 0
int main()
{
        int fd = STDIN;//键盘
        fd_set fdset;//集合,收集描述符

        while(1)//因为不止检测一次
        {
                FD_ZERO(&fdset);//清空集合,每个位置0:FD_ZERO
                FD_SET(fd,&fdset);//将描述符fd添加到集合fdset

                struct timeval tv = {5,0};//超时时间

                int n = select(fd+1,&fdset,NULL,NULL,&tv);//可能阻塞
                if(n ==-1)//select执行失败
                {
                        printf("select err\n");
                }
                else if(n==0)//超市,没有找到可用事件描述符
                {
                        printf("tme out\n");
                }
                else
                {
                        if(FD_ISSET(fd,&fdset))
                        {
                                char buff[128]={0};
                                int num = read(fd,buff,127);
                                printf("num=%d,buff=%s\n",num,buff);
                        }
                }

        }
}
~                                                                                                                                                                                            
~                                                                                                                                                                                            
~                         

4.tcp通过select实现并发连接

SER.C

#include<stdio.h>      // 标准输入输出库
#include<stdlib.h>     // 标准库,提供一些函数如malloc, free, rand等
#include<string.h>    // 字符串操作库
#include<unistd.h>    // UNIX标准函数库
#include<sys/select.h>// 选择库,提供select函数
#include<time.h>      // 时间库
#include<sys/socket.h>// 套接字库
#include<arpa/inet.h> // 提供inet_addr等函数
#include<netinet/in.h>// 提供一些网络相关的宏

#define MAXFD 10       // 定义最大文件描述符数量

// 初始化socket函数
int socket_init();

// 初始化文件描述符数组
void fds_init(int fds[]){
    for(int i=0; i<MAXFD; i++){
        fds[i] = -1; // 将所有文件描述符初始化为-1,表示未被使用
    }
}

// 将新的文件描述符添加到数组中
void fds_add(int fds[], int fd){
    for(int i=0; i<MAXFD; i++){
        if(fds[i] == -1){ // 找到数组中第一个未使用的文件描述符位置
            fds[i] = fd;  // 添加文件描述符
            break;        // 退出循环
        }
    }
}

// 从未使用的文件描述符数组中删除指定的文件描述符
void fds_del(int fds[], int fd){
    for(int i=0; i<MAXFD; i++){
        if(fds[i] == fd){ // 找到要删除的文件描述符
            fds[i] = -1;   // 将其设置为-1,表示未使用
            break;         // 退出循环
        }
    }
}

// 接受客户端连接请求并添加到文件描述符数组
void accept_client(int sockfd, int fds[]){
    int c = accept(sockfd, NULL, NULL); // 接受连接
    if(c < 0){
        return; // 如果返回-1,表示出错
    }
    printf("accept c = %d\n", c);
    fds_add(fds, c); // 添加到文件描述符数组
}

// 接收客户端发送的数据
void recv_date(int c, int fds[]){
    char buff[128] = {0}; // 创建缓冲区
    int n = recv(c, buff, 127, 0); // 接收数据
    if(n < 0){
        printf("cli close\n");
        close(c);          // 如果接收失败,关闭连接
        fds_del(fds, c);   // 从数组中删除该文件描述符
        return;
    }
    if(n == 0){
        printf("time out(%d)\n", n); // 如果超时
    }

    printf("buff(c=%d)=%s\n", c, buff); // 打印接收到的数据
    send(c, "ok", 2, 0);             // 发送确认消息
}

// 主函数
int main(){
    int sockfd = socket_init(); // 初始化socket
    if(sockfd == -1){
        exit(1); // 如果初始化失败,退出程序
    }

    int fds[MAXFD]; // 文件描述符数组
    fds_init(fds);  // 初始化数组
    fds_add(fds, sockfd); // 将监听的socket添加到数组

    fd_set fdset; // 创建文件描述符集合
    while(1){ // 无限循环等待事件
        FD_ZERO(&fdset); // 清空文件描述符集合
        int maxfd = -1; // 存储最大的文件描述符

        // 遍历文件描述符数组,将所有文件描述符添加到集合中
        for(int i=0; i<MAXFD; i++){
            if(fds[i] == -1){
                continue; // 如果文件描述符未使用,跳过
            }
            FD_SET(fds[i], &fdset); // 添加到集合
            if(fds[i] > maxfd){ // 更新最大文件描述符
                maxfd = fds[i];
            }
        }
        struct timeval tv = {5,0}; // 设置超时时间

        // 使用select等待直到有文件描述符准备好IO操作或超时
        int n = select(maxfd+1, &fdset, NULL, NULL, &tv);
        if(n == -1){
            printf("select err\n"); // 错误
        } else if(n == 0){
            printf("time out\n"); // 超时
        } else{
            // 遍历文件描述符数组,检查哪些文件描述符准备好了IO操作
            for(int i=0; i<MAXFD; i++){
                if(fds[i] == -1){
                    continue; // 如果文件描述符未使用,跳过
                }
                if(FD_ISSET(fds[i], &fdset)){ // 检查文件描述符是否被设置
                    if(fds[i] == sockfd){ // 如果是监听的socket
                        accept_client(sockfd, fds); // 接受新的连接
                    } else{ // 如果是已连接的客户端
                        recv_date(fds[i], fds); // 接收数据
                    }
                }
            }
        }
    }
}

// 创建socket并绑定到端口
int socket_init(){
    int sockfd = socket(AF_INET, SOCK_STREAM, 0); // 创建socket
    if(sockfd == -1){
        return -1; // 创建失败返回-1
    }
    struct sockaddr_in saddr; // 服务器地址结构
    memset(&saddr, 0, sizeof(saddr)); // 清零
    saddr.sin_family = AF_INET; // 地址族
    saddr.sin_port = htons(6000); // 端口
    saddr.sin_addr.s_addr = inet_addr("127.0.0.1"); // IP地址

    int res = bind(sockfd, (struct sockaddr*)&saddr, sizeof(saddr)); // 绑定
    if(res == -1){
        printf("bind err\n");
        return -1; // 绑定失败返回-1
    }
    if(listen(sockfd, 5) == -1){ // 开始监听,设置队列长度为5
        return -1; // 监听失败返回-1
    }
    return sockfd; // 返回socket文件描述符
}
                 

CLI.C

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

int main()
{
        int sockfd = socket(AF_INET,SOCK_STREAM,0);
        if(sockfd == -1)
        {
                exit(1);
        }
        struct sockaddr_in saddr;//代表服务器的端口
        memset(&saddr,0,sizeof(saddr));
        saddr.sin_family = AF_INET;
        saddr.sin_port = htons(6000);
        saddr.sin_addr.s_addr = inet_addr("127.0.0.1");

        int res = connect(sockfd,(struct sockaddr*)&saddr,sizeof(saddr));
        if(res == -1)
        {
                printf("connct err\n");
                exit(1);
        }

        while(1)
        {
                printf("input: ");
                char buff[128]={0};
                fgets(buff,128,stdin);
                if(strncmp(buff,"end",3)==0)
                {
                        break;
                }
                send(sockfd,buff,strlen(buff)-1,0);
                memset(buff,0,128);
                recv(sockfd,buff,127,0);
                printf("buff=%s\n",buff);
        }
        close(sockfd);
        exit(0);
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

果蛋蛋

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值