多进程/多线程两种方式实现简单socket通信

多进程服务端

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <sys/time.h>
#include <time.h>
#include <pthread.h> // 第三方库,编译需要加上-l pthread参数
#include <semaphore.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>

void recyleChild(int arg) { // 父进程回收子进程资源
    while (1) {
        int ret = waitpid(-1, NULL, WNOHANG); // 设置为非阻塞
        if (ret == -1) break;
        else if (ret > 0) printf("子进程%d被回收了\n", ret);
    }
}

int main() {
    // 注册SIGCHLD信号捕捉
    struct sigaction act;
    act.sa_handler = recyleChild;
    act.sa_flags = 0;
    sigemptyset(&act.sa_mask);
    sigaction(SIGCHLD, &act, NULL);

    // 创建socket
    int lfd = socket(AF_INET, SOCK_STREAM, 0);

    // 绑定服务器地址
    struct sockaddr_in Saddr;
    Saddr.sin_family = AF_INET;
    // inet_pton(AF_INET, "10.16.90.125", Saddr.sin_addr.s_addr);
    Saddr.sin_addr.s_addr = INADDR_ANY;
    Saddr.sin_port = htons(9999);
    bind(lfd, (struct sockaddr*)&Saddr, sizeof Saddr);

    // 监听
    int ret = listen(lfd, 128);
    if (ret == -1) {
        perror("listen");
        exit(-1);
    }

    while (1) { // 每收到一个连接, 就fork一个子进程去通信
        struct sockaddr_in cliaddr;
        int len = sizeof cliaddr;
        int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len); // 正常返回connect socket
        if (cfd == -1) {
            // 当accept阻塞时, 若有子进程终止, 则主进程会去调用recyleChild(相当于一个软中断), 调用完成后, 这里的accept会变成非阻塞, 但此时并没有客户连接, 因此需要重新accept
            if (errno == EINTR) continue; // 当accept被系统调用打断时, error会被设置为EINTR, 并且accept会变成非阻塞
            else { // 这里是正常的调用失败分支
                perror("accept");
                exit(-1);
            }
        }

        pid_t pid = fork();
        if (pid == 0) { // 子进程
            char cliIp[16]; // 客户端IP
            inet_ntop(AF_INET, &cliaddr.sin_addr.s_addr, cliIp, sizeof cliIp);
            unsigned short cliPort = ntohs(cliaddr.sin_port); // 客户端port
            printf("client ip is : %s, port is %d\n", cliIp, cliPort);

            char recvBuf[1024]; 
            while (1) { // 不断与客户端通信
                int len = read(cfd, recvBuf, sizeof recvBuf);
                if (len == -1) {
                    perror("read");
                    exit(-1);
                } else if (len > 0) {
                    printf("recv client data: %s\n", recvBuf);
                } else if (len == 0) { // 客户关闭了, 需要break避免执行下面的write
                    printf("client closed...\n"); 
                    break;
                }

                write(cfd, recvBuf, sizeof recvBuf);

            }

            close(cfd); // 关闭connect socket
        }
    }

    close(lfd); // 关闭监听socket

    return 0;
}

多线程服务端

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <unistd.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>
#include <sys/time.h>
#include <time.h>
#include <pthread.h> // 第三方库,编译需要加上-l pthread参数
#include <semaphore.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <errno.h>

struct sockInfo {
    int fd; // connect socket
    pthread_t tid; // 子线程的tid
    struct sockaddr_in addr; // 与之通信的客户端信息(主要是IP,port)
} sockInfos[128]; // 可以支持128个线程同时通信

void* working(void* args) { // 子线程与客户端通信
    struct sockInfo* pinfo = (struct sockInfo*)args;
    pinfo->tid = pthread_self();

    // 与之通信的客户端信息(IP, port)
    char cliIp[16];
    inet_ntop(AF_INET, &pinfo->addr.sin_addr.s_addr, cliIp, sizeof cliIp);
    unsigned short cliPort = ntohs(pinfo->addr.sin_port);
    printf("client ip is : %s, port is %d\n", cliIp, cliPort);

    char recvBuf[1024];
    while (1) { // 与客户端通信
        int len = read(pinfo->fd, recvBuf, sizeof recvBuf);
        if (len == -1) {
            perror("read");
            exit(-1);
        }
        else if (len > 0) {
            printf("recv client data: %s\n", recvBuf);
        }
        else {
            printf("client closed...\n");
            break;
        }

        write(pinfo->fd, recvBuf, sizeof recvBuf);

    }

    close(pinfo->fd); // 通信结束, 关闭connect socket
}

int main() {
    // 创建socket
    int lfd = socket(AF_INET, SOCK_STREAM, 0);

    // 绑定服务器地址
    struct sockaddr_in Saddr;
    Saddr.sin_family = AF_INET;
    // inet_pton(AF_INET, "10.16.90.125", Saddr.sin_addr.s_addr);
    Saddr.sin_addr.s_addr = INADDR_ANY;
    Saddr.sin_port = htons(9999);
    bind(lfd, (struct sockaddr*)&Saddr, sizeof Saddr);

    // 监听
    int ret = listen(lfd, 128);
    if (ret == -1) {
        perror("listen");
        exit(-1);
    }

    // 初始化数据
    int max = sizeof(sockInfos) / sizeof(sockInfos[0]) ;
    for (int i = 0; i < max; ++i) {
        bzero(&sockInfos[i], sizeof(sockInfos[i]));
        sockInfos[i].fd = -1; // -1表示还没使用
        sockInfos[i].tid = -1;
    }

    while (1) {// 每收到连接就创建一个新的子线程去通信
        // 等待连接
        struct sockaddr_in cliaddr;
        int len = sizeof cliaddr;
        int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len);

        // 创建子线程的准备工作
        struct sockInfo* pinfo; // 作用: 将主线程中的一些信息传入working, 给子线程用
        for (int i = 0; i < max; ++i) { // 找到一个可用的sockInfos[i]
            if(sockInfos[i].fd == -1) {
                pinfo = &sockInfos[i];
                break;
            }
            if (i == max - 1) { // 如果128个都不能用了, 就等1s, 再去尝试
                sleep(1);
                i = -1;
            }
        }

        pinfo->fd = cfd;
        memcpy(&pinfo->addr, &cliaddr, len);
        // 这里pinfo->tid可以等进入子线程执行的代码段, 即working中通过pthread_self()知道

        // 创建子线程
        pthread_t tid;
        pthread_create(&pinfo->tid, NULL, working, pinfo);

        // 回收子线程资源
        pthread_detach(pinfo->tid);
    }

    close(lfd); // 关闭监听socket

    return 0;
}

客户端

#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>

int main() {
    // 创建socket
    int fd = socket(AF_INET, SOCK_STREAM, 0);
    if(fd == -1) {
        perror("socket");
        exit(-1);
    }

    // 连接服务器
    struct sockaddr_in serveraddr;
    serveraddr.sin_family = AF_INET;
    inet_pton(AF_INET, "10.16.90.125", &serveraddr.sin_addr.s_addr);
    serveraddr.sin_port = htons(9999);
    int ret = connect(fd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));
    if(ret == -1) {
        perror("connect");
        exit(-1);
    }
    
    // 通信
    char recvBuf[1024];
    int i = 0;
    while(1) {
        sprintf(recvBuf, "data : %d\n", i++);
        
        // 给服务器端发送数据
        write(fd, recvBuf, strlen(recvBuf)+1);

        int len = read(fd, recvBuf, sizeof(recvBuf));
        if(len == -1) {
            perror("read");
            exit(-1);
        } else if(len > 0) {
            printf("recv server : %s\n", recvBuf);
        } else if(len == 0) { // 表示服务器端断开连接
            printf("server closed...");
            break;
        }

        sleep(1); // 这里不太懂? https://blog.csdn.net/xc_zhou/article/details/80950753
    }

    // 关闭连接
    close(fd);

    return 0;
}
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值