并发TCP服务器/客户端demo

tcpsvr.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <wait.h>
#include <errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
void sigchld(int signum) {
    for (;;) {
        pid_t pid = waitpid(-1, NULL, WNOHANG);
        if (pid == -1) {
            if (errno != ECHILD) {
                perror("waitpid");
                exit(-1);
            }
            printf("服务器："
                "全部子进程都已退出\n");
            break;
        }
        if (pid)
            printf("服务器："
                "发现%d子进程退出\n", pid);
        else {
            printf("服务器："
                "暂时没有子进程退出\n");
            break;
        }
    }
}
int client(int connfd) {
    for (;;) {
        printf("%d：接收请求\n", getpid());
        char buf[1024];
        ssize_t rb = recv(connfd, buf,
            sizeof(buf), 0);
        if (rb == -1) {
            perror("recv");
            return -1;
        }
        if (rb == 0) {
            printf("%d：客户机已关闭\n",
                getpid());
            break;
        }
        printf("%d：发送响应\n", getpid());
        if (send(connfd, buf, rb, 0) == -1) {
            perror("send");
            return -1;
        }
    }
    return 0;
}
int main(int argc, char* argv[]) {
    if (argc < 2) {
        fprintf(stderr, "用法：%s <端口号>\n",
            argv[0]);
        return -1;
    }
    if (signal(SIGCHLD, sigchld) == SIG_ERR) {
        perror("signal");
        return -1;
    }
    printf("服务器：创建网络套接字\n");
    int sockfd = socket(PF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("socket");
        return -1;
    }
    printf("服务器：准备地址并绑定\n");
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(atoi(argv[1]));
    addr.sin_addr.s_addr = INADDR_ANY;
    if (bind(sockfd, (struct sockaddr*)&addr,
        sizeof(addr)) == -1) {
        perror("bind");
        return -1;
    }
    printf("服务器：侦听套接字\n");
    if (listen(sockfd, 1024) == -1) {
        perror("listen");
        return -1;
    }
    for (;;) {
        printf("服务器：等待连接\n");
        struct sockaddr_in addrcli = {};
        socklen_t addrlen = sizeof(addrcli);
        int connfd = accept(sockfd,
            (struct sockaddr*)&addrcli,
            &addrlen);
        if (connfd == -1) {
            perror("accept");
            return -1;
        }
        printf("服务器：客户机%s:%u\n",
            inet_ntoa(addrcli.sin_addr),
            ntohs(addrcli.sin_port));
        pid_t pid = fork();
        if (pid == -1) {
            perror("fork");
            return -1;
        }
        if (pid == 0) {
            if (close(sockfd) == -1) {
                perror("close");
                return -1;
            }
            if (client(connfd) == -1)
                return -1;
            if (close(connfd) == -1) {
                perror("close");
                return -1;
            }
            printf("%d：完成\n", getpid());
            return 0;
        }
        if (close(connfd) == -1) {
            perror("close");
            return -1;
        }
    }
    return 0;
}

tcpcli.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <errno.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
int main(int argc, char* argv[]) {
    if (argc < 3) {
        fprintf(stderr, "用法：%s <IP地址> "
            "<端口号>\n", argv[0]);
        return -1;
    }
    printf("客户机：创建网络套接字\n");
    int sockfd = socket(PF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("socket");
        return -1;
    }
    printf("客户机：准备地址并连接\n");
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(atoi(argv[2]));
    addr.sin_addr.s_addr = inet_addr(argv[1]);
    if (connect(sockfd, (struct sockaddr*)&addr,
        sizeof(addr)) == -1) {
        perror("connect");
        return -1;
    }
    for (;;) {
        printf("> ");
        char buf[1024];
        fgets(buf, sizeof(buf) / sizeof(buf[0]),
            stdin);
        if (!strcmp(buf, "!\n"))
            break;
        printf("客户机：发送请求\n");
        if (send(sockfd, buf, strlen(buf) *
            sizeof(buf[0]), 0) == -1) {
            perror("send");
            return -1;
        }
        printf("客户机：接收响应\n");
        ssize_t rb = recv(sockfd, buf,
            sizeof(buf) - sizeof(buf[0]), 0);
        if (rb == 0) {
            printf("客户机：服务器已关闭\n");
            break;
        }
        buf[rb / sizeof(buf[0])] = '\0';
        printf("< %s", buf);
    }
    printf("客户机：关闭套接字\n");
    if (close(sockfd) == -1) {
        perror("close");
        return -1;
    }
    printf("完成\n");
    return 0;
}