LinuxC 网络 -- TCP（C/S）循环服务器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;
    }
}

mian.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 cfd;
    int sockfd;

    socklen_t c_size;
    struct sockaddr_in s_addr;
    struct sockaddr_in c_addr;

    sockfd = tcp_server_create();
    set_nonblock(sockfd);
    char message[1024];

    while (1)
    {
        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));

        set_nonblock(cfd);

        while (1)
        {
            memset(message, 0, sizeof(message));

            if (read_nonblock(cfd, message, sizeof(message) - 1) == 0)
            {
                break;
            }

            printf("server recv msg = %s\n", message);
        }

        shutdown(cfd, SHUT_RDWR);
    }
    return 0;
}

运行结果：