LinuxC 网络 -- TCP（C/S）IO多路转接（poll）demo

nc -- 作为客户端

代码如下

TcpServer.h 文件：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include <unistd.h>

#include <netinet/in.h> //描述网络的

#include <ctype.h>

#include <sys/socket.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/select.h>
#include <sys/epoll.h>
#include <poll.h>

#include <fcntl.h>
#include <arpa/inet.h>
#include <signal.h>
#include <errno.h>
#include <pthread.h>

#define PORT 5555

int tcp_server_create();
void set_nonblock(int cfd);
int read_nonblock(int cfd, char *message, size_t size);
int read_block(int cfd, char *message, size_t size);

TcpServer.c

#include "tcp_server.h"

void set_nonblock(int cfd)
{
    int flags;
    flags = fcntl(cfd, F_GETFL);
    flags = flags | O_NONBLOCK; // 把 cfd 设置成非阻塞
    fcntl(cfd, F_SETFL, flags);
}

int tcp_server_create()
{
    signal(SIGPIPE, SIG_IGN);

    struct sockaddr_in s_addr;
    struct sockaddr_in c_addr;
    int sockfd;
    int cfd;
    socklen_t c_size;
    int i;

    // 启连接 加 SOCK_NONBLOCK accept 不会阻塞
    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    {
        perror("socket error");
        exit(1);
    }
    printf("success socket!\n");

    int opt = 1;
    setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    s_addr.sin_family = AF_INET;
    s_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    s_addr.sin_port = htons(PORT);

    if (bind(sockfd, (struct sockaddr *)&s_addr, sizeof(struct sockaddr_in)) != 0)
    {
        perror("bind error");
        exit(1);
    }

    printf("bind successuful!\n");

    if (listen(sockfd, 20) != 0)
    {
        perror("listen error");
        exit(1);
    }
    printf("listen successuful!\n");

    return sockfd;
}

int read_nonblock(int cfd, char *message, size_t size)
{
    int r_n;
    while (r_n = read(cfd, message, size))
    {
        if (r_n < 0)
        {
            if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR)
            {
                perror("recv message error");
                exit(1);
            }

            continue;
        }

        if (r_n == 0)
        {
            printf("client is close!\n");
        }
        else
        {
            break;
        }

        return r_n;
    }
}

int read_block(int cfd, char *message, size_t size)
{
    int r_n;
    while (r_n = read(cfd, message, size))
    {
        if (r_n < 0)
        {
            perror("recv message error");
            exit(1);
        }
        if (r_n == 0)
        {
            printf("client is close!\n");
            return r_n;
        }
        else
        {
            break;
        }

        return r_n;
    }
}

poll_main.c

#include "tcp_server.h"

void conver_char(char *p)
{
    if (p == NULL)
    {
        return;
    }
    for (; *p != '\0'; p++)
    {
        if (*p >= 'a' && *p <= 'z')
        {
            *p = *p - 'a' + 'A';
        }
    }
}

int main(int argc, char const *argv[])
{
    int i;
    int cfd;
    int sockfd;

    socklen_t c_size;
    struct sockaddr_in s_addr;
    struct sockaddr_in c_addr;

    struct pollfd fds[100];

    for (int i = 0; i < 100; i++)
    {
        fds[i].fd = -1;
    }

    sockfd = tcp_server_create();
    set_nonblock(sockfd);

    fds[0].fd = sockfd;
    fds[0].events = POLLIN; // 监听的事件为可读

    char message[1024];

    while (1)
    {
        int rnum = poll(fds, 100, -1);

        if (fds[0].events == fds[0].revents)
        {
            c_size = sizeof(struct sockaddr_in);
            cfd = accept(sockfd, (struct sockaddr *)&c_addr, &c_size);

            if (cfd < 0)
            {
                if (errno != EAGAIN && errno != EWOULDBLOCK && errno != EINTR)
                {
                    perror("accept error");
                    exit(1);
                }

                continue; // 返回while(1)
            }
            printf("info client: ip = %s port = %d\n", inet_ntoa(c_addr.sin_addr), ntohs(c_addr.sin_port));

            for (i = 1; i < 100; i++)
            {
                if (fds[i].fd == -1)
                {
                    fds[i].fd = cfd;
                    fds[i].events = POLLIN;
                    break;
                }

                if (i == 100)
                {
                    printf("too many client!\n");
                    exit(1);
                }
            }

            if (--rnum <= 0)
            {
                continue;
            }
        }

        for (int i = 1; i < 100; i++)
        {
            if (fds[i].fd != -1)
            {
                if (fds[i].events == fds[i].revents)
                {
                    int fd = fds[i].fd;
                    memset(message, 0, sizeof(message));
                    int r_n = read(fd, message, sizeof(message));

                    if (r_n < 0)
                    {
                        perror("read cfd data error");
                        fds[i].fd = -1;
                    }
                    if (r_n == 0)
                    {
                        printf("client is close!\n");
                        fds[i].fd = -1;
                        break;
                    }
                    printf("recv data: %s\n", message);
                    conver_char(message);
                    write(cfd, message, strlen(message));

                    if (--rnum <= 0)
                    {
                        break;
                    }
                }
            }
        }
    }
    return 0;
}