Linux TCP 通信并发

狗狗狗乖乖2023-10-07 21:18

多进程

客户端

bash 复制代码 
#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
int main() {
    //创建套接字
    int lfd = socket(AF_INET, SOCK_STREAM, 0);
    if(lfd == -1) {
        perror("socket");
        exit(-1);
    }
    //连接服务器
    struct sockaddr_in serveraddr;
    inet_pton(AF_INET, "192.168.245.129", &serveraddr.sin_addr.s_addr);
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(9999);
    int ret = connect(lfd, (struct sockaddr *) &serveraddr, sizeof(serveraddr));
    if(ret == -1) {
        perror("connect");
        _exit(-1);
    }
    //通信
    int i = 0;
    char receiveBuf[1024] = {0};
    while(1) {
        sprintf(receiveBuf, "data: %d\n", i++);

        write(lfd, receiveBuf, strlen(receiveBuf));
        sleep(1);
        
        int len = read(lfd, receiveBuf, sizeof(receiveBuf));
        if(len == -1) {
            perror("read");
            exit(0);
        } else if(len > 0) {
            printf("receive server data: %s\n", receiveBuf);
        } else if (len == 0) {
            printf("server closed...");
        }
        
    }
    

    close(lfd);
}

服务器端

bash 复制代码 
#define _XOPEN_SOURCE
#include<stdio.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<string.h>
#include<stdlib.h>
#include<signal.h>
#include<wait.h>
#include<errno.h>

void recyleChild(int arg) {
    while(1) {
        int ret = waitpid(-1, NULL, WNOHANG);
        if(ret == -1) {
            break;
        } else if (ret == 0) {
            break;
        } else if (ret > 0) {
            printf("pid %d is resaved\n", ret);
        }
    }
}

int main() {

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


    //创建socket
    int lfd = socket(AF_INET, SOCK_STREAM, 0);
    if(lfd == -1) {
        perror("socket");
        exit(0);
    }
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(9999);
    saddr.sin_addr.s_addr = INADDR_ANY;

    //绑定
    int ret = bind(lfd, (struct sockaddr*)&saddr, sizeof(saddr));
    if(ret == -1) { 
        perror("bind");
        exit(0);
    }

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

    //接受链接
    while(1) {
        struct sockaddr_in cliaddr;
        int len = sizeof(cliaddr);
        
        int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len);   
        if(cfd == -1) {
            if(errno == EINTR) {
                continue;
            }
            perror("accept");
            exit(-1);
        }

        //每连接一个，创建一个子进程
        pid_t pid = fork();
        if(pid == 0) {
            char clientIP[16];
            inet_ntop(AF_INET, &cliaddr.sin_addr.s_addr, clientIP, 16);
            unsigned short cliPort = ntohs(cliaddr.sin_port);
            printf("Client Port is %d, client addr is %s\n", cliPort, clientIP);

            char recvBuf[1024] = {0};
            while(1) {
                int len = read(cfd, &recvBuf, sizeof(recvBuf));
                if(len == -1) {
                    perror("read");
                    return 0;
                } else if(len > 0) {
                    printf("recv client data : %s \n", recvBuf);
                    
                } else {
                    printf("closed.....");
                }

                write(cfd, &recvBuf, sizeof(recvBuf));
            }
            close(cfd);
            exit(0);
        }
    }
    close(lfd);
    return 0;
}

多线程

服务器

bash 复制代码 
#define _XOPEN_SOURCE
#include<stdio.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<string.h>
#include<stdlib.h>
#include<errno.h>
#include<pthread.h>


struct sockinfo {
    int fd;
    struct sockaddr_in saddr;
    pthread_t tid;
};

struct sockinfo sockinfos[128];

void* working(void* arg) {

    struct sockinfo* pinfo = (struct sockinfo*) arg;
    char clientIP[16];
    inet_ntop(AF_INET, &pinfo->saddr.sin_addr.s_addr, clientIP, 16);
    unsigned short cliPort = ntohs(pinfo->saddr.sin_port);
    printf("Client Port is %d, client addr is %s\n", cliPort, clientIP);

    char recvBuf[1024] = {0};
    while(1) {
        int len = read(pinfo->fd, &recvBuf, sizeof(recvBuf));
        if(len == -1) {
            perror("read");
            return 0;
        } else if(len > 0) {
            printf("recv client data : %s \n", recvBuf);
                    
        } else {
            printf("closed.....");
        }

        write(pinfo->fd, &recvBuf, sizeof(recvBuf));
    }
    close(pinfo->fd);
    exit(0);
    return NULL;
}

int main() {
 //创建socket
    int lfd = socket(AF_INET, SOCK_STREAM, 0);
    if(lfd == -1) {
        perror("socket");
        exit(0);
    }
    struct sockaddr_in saddr;
    saddr.sin_family = AF_INET;
    saddr.sin_port = htons(9999);
    saddr.sin_addr.s_addr = INADDR_ANY;

    //绑定
    int ret = bind(lfd, (struct sockaddr*)&saddr, sizeof(saddr));
    if(ret == -1) { 
        perror("bind");
        exit(0);
    }

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

    //初始化数据
    int max = sizeof(sockinfos) / sizeof(sockinfos[0]);
    for(int i = 0; i < max; i++) {
        memset(&sockinfos[i], 0, sizeof(sockinfos[i]));
        sockinfos[i].fd = -1;
        sockinfos[i].tid = -1;
    }

    //接受链接
    while(1) {
        struct sockaddr_in cliaddr;
        int len = sizeof(cliaddr);
        
        int cfd = accept(lfd, (struct sockaddr*)&cliaddr, &len);   
        if(cfd == -1) {
            perror("accept");
            exit(-1);
        }

        struct sockinfo * pinfo;
        for(int i = 0; i < max; i++) {
            if(sockinfos[i].fd == -1) {
                pinfo = &sockinfos[i];
                break;
            }
            if(i == max - 1) {
                sleep(1);
                i--;
            }
        }

        pinfo->fd = cfd;
        memcpy(&pinfo->saddr, &cliaddr, len);


        //每连接一个，创建一个子进程

        pthread_create(&pinfo->tid, NULL, working, pinfo);

        pthread_detach(pinfo->tid);
    }
    close(lfd);
    return 0;
}

客户端

bash 复制代码 
#include <stdio.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
int main() {
    //创建套接字
    int lfd = socket(AF_INET, SOCK_STREAM, 0);
    if(lfd == -1) {
        perror("socket");
        exit(-1);
    }
    //连接服务器
    struct sockaddr_in serveraddr;
    inet_pton(AF_INET, "192.168.245.129", &serveraddr.sin_addr.s_addr);
    serveraddr.sin_family = AF_INET;
    serveraddr.sin_port = htons(9999);
    int ret = connect(lfd, (struct sockaddr *) &serveraddr, sizeof(serveraddr));
    if(ret == -1) {
        perror("connect");
        _exit(-1);
    }
    //通信
    int i = 0;
    char receiveBuf[1024] = {0};
    while(1) {
        sprintf(receiveBuf, "data: %d\n", i++);

        write(lfd, receiveBuf, strlen(receiveBuf));
        sleep(1);
        
        int len = read(lfd, receiveBuf, sizeof(receiveBuf));
        if(len == -1) {
            perror("read");
            exit(0);
        } else if(len > 0) {
            printf("receive server data: %s\n", receiveBuf);
        } else if (len == 0) {
            printf("server closed...");
        }
        
    }
    

    close(lfd);
}
相关推荐
晨欣2 分钟前
Umi-OCR：Windows7和Linux上可免费离线使用的OCR应用！
linux·运维·ocr
沐雨风栉19 分钟前
自建云音乐服务器：Navidrome+cpolar让无损音乐随身听
运维·服务器·redis·缓存·docker·容器
德迅云安全-如意20 分钟前
你知道服务器和电脑主机的区别吗？
运维·服务器
广州腾科助你拿下华为认证23 分钟前
华为HCIE数通考试难度解析
运维·服务器
Clownseven27 分钟前
VPS、云服务器、独立服务器的区别是什么？新手服务器选择指南
运维·服务器
跨境小新34 分钟前
静态住宅Facebook养号难不难？
运维·服务器
siriuuus1 小时前
Linux 网络配置解析、IP地址配置及常用网络命令
linux·网络
ZYMFZ1 小时前
Linux 防火墙 Iptables
linux·运维·服务器
@CLoudbays_Martin111 小时前
CDN是否能有效检测并且同时防御Ddos 和 CC 攻击？
java·服务器·网络·数据库·git·数据库开发·时序数据库
河南博为智能科技有限公司1 小时前
单串口服务器-工业级串口联网解决方案
运维·服务器·物联网·边缘计算
热门推荐
01conda中设置镜像地址（附所有可换的地址）02UV安装并设置国内源03保姆级教程：手把手教你用Dify实现完美多轮对话（附Chatflow和提示词）04UV 工具安装与国内镜像源配置指南05A股预测还能更准？开源大模型Kronos带你跑通预测+回测全流程06GitHub 镜像站点07突破百度网盘的下载限速，两种方法教会你【超详细】08KGG转MP3工具|非KGM文件|解密音频09教你如何认证 Gemini 教育优惠的二次验证，薅个 1年的 Gemini Pro 会员1046个Nano-banana 精选提示词，持续更新中