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);
}
相关推荐
xx155802862xx8 分钟前
centos转移mysql的数据存储目录
linux·mysql·centos
独行soc11 分钟前
2025年渗透测试面试题总结-字节跳动[实习]安全研发员（题目+回答）
linux·科技·安全·面试·职场和发展·渗透测试
chian-ocean13 分钟前
永不休眠：Linux 守护进程的工作原理
linux·运维·服务器
00后程序员张1 小时前
实战解析：如何用克魔(KeyMob)等工具构建iOS应用稳定性与数据可观测体系
websocket·网络协议·tcp/ip·http·网络安全·https·udp
GalaxyPokemon1 小时前
LeetCode - 76. 最小覆盖子串
运维·服务器·数据结构·算法·leetcode
嵌入式@秋刀鱼1 小时前
《 第三章-招式初成》 C++修炼生涯笔记（基础篇）程序流程结构
linux·开发语言·数据结构·c++·笔记·visual studio code
楽码1 小时前
安装和编写grpc协议文件
服务器·后端·grpc
qq_290606271 小时前
监听本地tcp,udp端口,通过socks5转发至远端
网络协议·tcp/ip·udp
q567315232 小时前
Linux搭建爬虫ip与私有IP池教程
linux·爬虫·tcp/ip
Arrtoria2 小时前
Kernel2.X的内存世界
linux
热门推荐
01零代码入门 | Coze——让大模型接入自己的数据库02【图像处理与机器视觉】XJTU期末考点03KGG转MP3工具|非KGM文件|解密音频04Coze扣子平台完整体验和实践（附国内和国际版对比）05从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑06YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】07扣子空间的使用教程与大模型技术思考08海康Visionmaster-常见问题排查方法-启动阶段09【SpeedAI科研小助手】2分钟极速解决知网维普重复率、AIGC率过高，一键全文降！文件格式不变，公式都保留的！10苹果iPhone 16 Pro Max上手体验:性能极致释放