多路複用

Jiangson發表於2024-08-30
#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);
 }

以上就是多路複用的程式碼

相關文章