TCP中使用poll来进行多用户端的连接TCP+POLL

首先让我们来看一下TCP编程的基本过程以下为服务端（客户端在文章末尾）

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <string.h>
#include <poll.h>
 
int main(int argc, char const *argv[])
{
    
    if (argc != 2)
    {
        printf("please input %s <port>\n", argv[0]);
        return -1;
    }
    //1.创建套接字 socket
    int sockfd, acceptfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);
 
    //填充ipv4的通信结构体
    struct sockaddr_in serveraddr, caddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[1]));
    //serveraddr.sin_addr.s_addr = inet_addr("0.0.0.0"); //0.0.0.0 自动获取主机ip
    // serveraddr.sin_addr.s_addr = htonl(INADDR_ANY);
    serveraddr.sin_addr.s_addr = INADDR_ANY;
 
    socklen_t len = sizeof(caddr);
 
    //&serveraddr -->struct sockaddr_in *
 
    //2.绑定套接字  bind  （绑定自己的ip和端口，便于别人找到自己）
    if (bind(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("bind err.");
        return -1;
    }
    printf("bind ok.\n");
 
    //3.listen  监听 将主动套接子变为被动等待
    if (listen(sockfd, 5) < 0)
    {
        perror("listen err.");
        return -1;
    }
    printf("listen ok.\n");

接下来我们进入正题。

我们首先需要先创建一个结构体数组来存放我们接下来输入的文件描述符

struct pollfd fds[20]={};  //大小可以根据需要随意

然后我们将最基础的文件描述符添加进去

0代表标准输入，而sockfd则代表客户端的连接.

//2.添加关心的文件描述符
     fds[0].fd = 0;
     fds[0].events=POLLIN;//设置为读事件
 
     fds[1].fd=sockfd;
     fds[1].events=POLLIN;

接下来我们自定义一个整型变量n用来表示结构体数组中文件描述符的个数，ret则用来后面接收poll的返回值，以及一个用来储存数据的数组buf 

    int n = 2, ret;
    char buf[128];

接下来我们创建一个死循环来保障程序的一直运行，以下代码将全部写在while循环中.

poll函数介绍：

int poll(struct pollfd *fds, nfds_t nfds, int timeout);
   参p数：
   struct pollfd *fds
     关心的文件描述符数组struct pollfd fds[N];
   nfds:个数
   timeout： 超时检测
    毫秒级的：如果填1000，1秒
     如果-1，阻塞
 
 struct pollfd {
     int   fd;         /* 检测的文件描述符 */
     short events;     /* 检测事件 */
     short revents;    /* 调用poll函数返回填充的事件，poll函数一旦返回，将对应事件自动填充结构体这个成员。只需要判断这个成员的值就可以确定是否产生事件 */
 };
    事件： POLLIN ：读事件
                POLLOUT : 写事件
               POLLERR：异常事件

接下来我们使用poll函数来观测我们的文件描述符

        ret = poll(fds, n, -1);
        if (ret < 0)
        {
            perror("poll err.");
            return -1;
        }

接下来我们为了方便遍历我们的结构体数组，所以我们需要创建一个for循环。

由于数组的下标从零开始，所以i初始值为0，且小于数组中描述符个数（n）

for (int i = 0; i < n; i++)
{
 
}

在for循环中我为了应对各种状况的发生，所以我们利用if嵌套和else if等来进行条件的判断

            if (fds[i].revents == POLLIN)
            {
                if (fds[i].fd == 0)
                {
                    fgets(buf, sizeof(buf), stdin);
                    printf("key:%s\n", buf);
                    //将服务器段输入的数据作为通知的消息发送给所有链接的客户端
                    for(int j=2;j<n;j++)
                    {
                        send(fds[j].fd,buf,sizeof(buf),0);
                    }
                }
                else if (fds[i].fd == sockfd)
                {
                    //4.阻塞等待客户端链接，链接成功返回一个通信文件描述符 accept
                    acceptfd = accept(sockfd, (struct sockaddr *)&caddr, &len);
 
                    if (acceptfd < 0)
                    {
                        perror("accept err.");
                        return -1;
                    }
                    printf("accept ok.\n");
 
                    //输出查看链接的客户端的ip和端口
                    printf("ip:%s ,port:%d\n",
                           inet_ntoa(caddr.sin_addr), ntohs(caddr.sin_port));
 
                    //需要将产生通信文件描述符添加到表中进行检测
                    fds[n].fd=acceptfd;
                    fds[n].events=POLLIN;
                    n++;
                }
                else
                {
                    //接收消息
                    int recvbyte = recv(fds[i].fd, buf, sizeof(buf), 0);
                    if (recvbyte < 0)
                    {
                        perror("recv err.");
                        // return -1;
                    }
                    else if (recvbyte == 0)
                    {
                        printf("%d client exit.\n",fds[i].fd);
                        close(fds[i].fd);
                        fds[i]=fds[n-1];
                        n--;
                        i--;              
                        //break;
                    }
                    else
                    {
                        printf("%d buf:%s\n",fds[i].fd, buf);
                    }
                }
            }

最后在while循环外再加上关闭文件描述符即可

close(sockfd);

以下为客户端：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/wait.h>
 
 
int main(int argc, char const *argv[])
{
    if (argc != 3)
    {
        printf("please input %s <ip> <port>\n", argv[0]);
        return -1;
    }
    //1.创建套接字 socket
    int sockfd;
    sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd < 0)
    {
        perror("socket err.");
        return -1;
    }
    printf("socket ok %d\n", sockfd);
 
    //填充ipv4的通信结构体  服务器端ip和端口
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(atoi(argv[2]));
    serveraddr.sin_addr.s_addr = inet_addr(argv[1]);
 
    //&serveraddr -->struct sockaddr_in *
 
    //2.请求链接  connect
    if (connect(sockfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr)) < 0)
    {
        perror("connect err.");
        return -1;
    }
    printf("connect ok.\n");
 
    //3.循环收发消息
    char buf[128];
    int recvbyte;
    pid_t pid = fork();
    if (pid < 0)
    {
        perror("fork err.");
        return -1;
    }
    else if (pid == 0)
    {
        while (1)
        {
 
            fgets(buf, sizeof(buf), stdin);
            if (buf[strlen(buf) - 1] == '\n')
                buf[strlen(buf) - 1] = '\0';
            //发送消息
            send(sockfd, buf, sizeof(buf), 0);
        }
    }
    else
    {
        while (1)
        {
            //接收消息
            recvbyte = recv(sockfd, buf, sizeof(buf), 0);
            if (recvbyte < 0)
            {
                perror("recv err.");
                // return -1;
            }
            if (recvbyte == 0)
            break;
            if (buf[strlen(buf) - 1] == '\n')
            buf[strlen(buf) - 1] = '\0';
            printf("buf:%s\n", buf);
        }
    }
    close(sockfd);
    return 0;
}