多路复用

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <sys/time.h>
#include <sys/select.h> 
int main(void)
{
     //1.创建套接字
     int  sockfd  = socket(AF_INET, SOCK_STREAM, 0);
    if( sockfd < 0)
    {
        perror("socket");
        return -1;
    }
     //2.绑定端口和地址
    struct sockaddr_in addr;
    addr.sin_family=AF_INET;
    addr.sin_port = htons(9527);
     //转网络字节序
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
     //inet_addr("192.168.41.1");
     //网络字节序(32位整型数ip地址)
    //绑定地址为INADDR_ANY,这个个服务器程序在那个ip地址上运行就绑定那个ip地址    
     int ret = bind(sockfd, (struct sockaddr*)&addr,sizeof(addr));
    if(ret < 0)
    {
        perror("bind");
        return -1;
    }
    //3.监听
    ret  = listen(sockfd, 4);
    if(ret < 0)
    {
        perror("listen");
        return -1;
    }
     //接受连接--accept阻塞
    //接收客户端数据--recv阻塞
     //select多路复用
     fd_set readfds;
     //定义读文件描述符集合
    int  cfds[50]={0};
     //保存所有客户端的套接字
    while(1)
    {
        int maxfd = sockfd;
        //清空集合
        FD_ZERO(&readfds);
        FD_SET(sockfd,&readfds);
          //把sockfd添加到readfds集合中
          //把客户端的套接字也添加到集合中
        for(int i=0; i<50; i++)
        {
              if(cfds[i] != 0)//如果不等于0就说明这个空间存储了套接字 
           {
                FD_SET(cfds[i],&readfds);
                maxfd = maxfd>cfds[i]?maxfd:cfds[i];
            }
        }        //select用监听readfds里面的所有文件描述符,
        //如果某一个或多个描述符有响应就跳出select函数
        ret = select(maxfd+1,&readfds,NULL, NULL, NULL);
        if(ret < 0)
        {
            perror("select");
        }
        //判断是否是sockfd有数据---说明有客户端连接
        if(FD_ISSET(sockfd, &readfds))
        { 
               //接受连接
            int clientfd = accept(sockfd, NULL,NULL);
            if(clientfd < 0)
            {
                perror("accept");
            }            //把客户端套接字保存到cfds数组中
            for(int i=0; i<50; i++)
            {
                if(cfds[i] == 0)//判断放在没有使用的空间中
                {
                     cfds[i] = clientfd;
                     break;
                }
            }
         }        //判断是否是客户端有数据
        for(int i=0; i<50; i++)
        {
            //判断套接字是否存在,并且是否有数据到达
            if(cfds[i] != 0 && FD_ISSET(cfds[i], &readfds))
            { 
                     //接收客户端数据
                char buffer[128]={0};
                int size = recv(cfds[i], buffer, 128, 0);
                if(size<=0)
                {
                    printf("有客户端掉线\n");
                    close(cfds[i]);
                    cfds[i] = 0;
                }
                //把接收到的信息转发给客户端自己
                     size = send(cfds[i],buffer,size,0);

            }
        }
    }
     printf("有客户端连接\n");
    close(sockfd);
 }

以上就是多路复用的代码

posted @ 2024-08-30 21:02  Jiangson  阅读(12)  评论(0编辑  收藏  举报