TCP服务器的演变过程：多进程实现一对多的TCP服务器

Lion Long2023-12-26 4:06

使用多进程实现一对多的TCP服务器

一、前言

手把手教你从0开始编写TCP服务器程序，体验开局一块砖，大厦全靠垒。

为了避免篇幅过长使读者感到乏味，对【TCP服务器的开发】进行分阶段实现，一步步进行优化升级。本节在上一章节的基础上，改为多进程方式实现TCP服务器，为每个新接入的客户端分配进程，实现一个服务器程序处理多个客户端连接。主要目的是比较不同方式的利弊关系。

二、新增使用的fork()函数

函数原型：

c 复制代码 
#include <unistd.h>
pid_t fork(void);

fork（）通过复制调用进程来创建一个新进程。新进程被称为子进程。调用进程被称为父进程。

子进程和父进程在单独的内存空间中运行。在执行fork（）时，两个内存空间都具有相同的内容。其中一个进程执行的内存写入、文件映射（mmap）和取消映射（munmap），不会影响另一个进程。

返回值：

  • 返回0，代表子进程。
  • 返回非零，代表是父进程。

三、实现步骤

使用多进程方案，来一个连接请求则克隆一个子进程。

（1）创建socket。

c 复制代码 
int listenfd=socket(AF_INET,SOCK_STREAM,0);
if(listenfd==-1){
    printf("errno = %d, %s\n",errno,strerror(errno));
    return SOCKET_CREATE_FAILED;
}

（2）绑定地址。

c 复制代码 
struct sockaddr_in server;
memset(&server,0,sizeof(server));

server.sin_family=AF_INET;
server.sin_addr.s_addr=htonl(INADDR_ANY);
server.sin_port=htons(LISTEN_PORT);

if(-1==bind(listenfd,(struct sockaddr*)&server,sizeof(server))){
    printf("errno = %d, %s\n",errno,strerror(errno));
    close(listenfd);
    return SOCKET_BIND_FAILED;
}

（3）设置监听。

c 复制代码 
if(-1==listen(listenfd,BLOCK_SIZE)){
   printf("errno = %d, %s\n",errno,strerror(errno));
   close(listenfd);
   return SOCKET_LISTEN_FAILED;
}

（4）接收连接。

c 复制代码 
struct sockaddr_in client;
memset(&client,0,sizeof(client));
socklen_t len=sizeof(client);
    
int clientfd=accept(listenfd,(struct sockaddr*)&client,&len);
if(clientfd==-1){
    printf("errno = %d, %s\n",errno,strerror(errno));
    close(listenfd);
    return SOCKET_ACCEPT_FAILED;
}

（5）为每个连接克隆子进程。

c 复制代码 
pid_t pid=fork();
if(pid==0)
{
     routine(clientfd);
     break;
}
else
{
     printf("pid = %d\n",pid);
}

（6）在子进程里面接收数据。

c 复制代码 
char buf[BUFFER_LENGTH]={0};
ret=recv(clientfd,buf,BUFFER_LENGTH,0);
if(ret==0) {
     printf("connection dropped\n");

}
printf("recv --> %s\n",buf);

（7）在子进程里面发送数据。

c 复制代码 
if(-1==send(clientfd,buf,ret,0))
{
    printf("errno = %d, %s\n",errno,strerror(errno));
}

（8）关闭文件描述符。

c 复制代码 
close(listenfd);

四、完整代码

c 复制代码 
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>

#include <errno.h>
#include <string.h>
#include <unistd.h>

#include <pthread.h>

#define LISTEN_PORT     8888
#define BLOCK_SIZE      10
#define BUFFER_LENGTH   1024

enum ERROR_CODE{
    SOCKET_CREATE_FAILED=-1,
    SOCKET_BIND_FAILED=-2,
    SOCKET_LISTEN_FAILED=-3,
    SOCKET_ACCEPT_FAILED=-4
};

void routine(int clientfd)
{
    while(1)
    {
        // 5.
        char buf[BUFFER_LENGTH]={0};
        int ret=recv(clientfd,buf,BUFFER_LENGTH,0);
        if(ret==0) {
            printf("connection dropped\n");
            break;
        }
        
        printf("fd=%d recv --> %s\n",clientfd,buf);
        send(clientfd,buf,ret,0);
        
    }
    

    close(clientfd);
}

int main(int argc,char **argv)
{
    // 1.
    int listenfd=socket(AF_INET,SOCK_STREAM,0);
    if(listenfd==-1){
        printf("errno = %d, %s\n",errno,strerror(errno));
        return SOCKET_CREATE_FAILED;
    }

    // 2.
    struct sockaddr_in server;
    memset(&server,0,sizeof(server));

    server.sin_family=AF_INET;
    server.sin_addr.s_addr=htonl(INADDR_ANY);
    server.sin_port=htons(LISTEN_PORT);

    if(-1==bind(listenfd,(struct sockaddr*)&server,sizeof(server))){
        printf("errno = %d, %s\n",errno,strerror(errno));
        close(listenfd);
        return SOCKET_BIND_FAILED;
    }

    // 3.
    if(-1==listen(listenfd,BLOCK_SIZE)){
        printf("errno = %d, %s\n",errno,strerror(errno));
        close(listenfd);
        return SOCKET_LISTEN_FAILED;
    }

    printf("listen port: %d\n",LISTEN_PORT);
    struct sockaddr_in client;
    socklen_t len=sizeof(client);
    int clientfd=-1;
    
    while(1)
    {
        // 4.
        memset(&client,0,sizeof(client));
        clientfd=accept(listenfd,(struct sockaddr*)&client,&len);
        if(clientfd==-1){
            printf("errno = %d, %s\n",errno,strerror(errno));
            continue;
        }

        printf("accept successdul, fd = %d\n",clientfd);
        pid_t pid=fork();
        if(pid==0)
        {
            routine(clientfd);
            break;
        }
        else
        {
            printf("pid = %d\n",pid);
        }
    }

    
    close(listenfd);

    return 0;
}

编译：

bash 复制代码 
gcc -o server server.c

五、TCP客户端

5.1、自己实现一个TCP客户端

自己实现一个TCP客户端连接TCP服务器的代码：

c 复制代码 
#include <stdio.h>
#include <sys/socket.h>

#include <netinet/in.h>
#include <arpa/inet.h>

#include <errno.h>
#include <string.h>

#include <unistd.h>
#include <stdlib.h>

#define BUFFER_LENGTH   1024

enum ERROR_CODE{
    SOCKET_CREATE_FAILED=-1,
    SOCKET_CONN_FAILED=-2,
    SOCKET_LISTEN_FAILED=-3,
    SOCKET_ACCEPT_FAILED=-4
};

int main(int argc,char** argv)
{
    if(argc<3)
    {
        printf("Please enter the server IP and port.");
        return 0;
    }
    printf("connect to %s, port=%s\n",argv[1],argv[2]);

    int connfd=socket(AF_INET,SOCK_STREAM,0);
    if(connfd==-1)
    {
        printf("errno = %d, %s\n",errno,strerror(errno));
        return SOCKET_CREATE_FAILED;

    }
    struct sockaddr_in serv;
    serv.sin_family=AF_INET;
    serv.sin_addr.s_addr=inet_addr(argv[1]);
    serv.sin_port=htons(atoi(argv[2]));
    socklen_t len=sizeof(serv);
    int rwfd=connect(connfd,(struct sockaddr*)&serv,len);
    if(rwfd==-1)
    {
        printf("errno = %d, %s\n",errno,strerror(errno));
        close(rwfd);
        return SOCKET_CONN_FAILED;
    }
    int ret=1;
    while(ret>0)
    {
        char buf[BUFFER_LENGTH]={0};
        printf("Please enter the string to send:\n");
        scanf("%s",buf);
        send(connfd,buf,strlen(buf),0);

        memset(buf,0,BUFFER_LENGTH);
        printf("recv:\n");
        ret=recv(connfd,buf,BUFFER_LENGTH,0);
        printf("%s\n",buf);
        
    }
    close(rwfd);
    return 0;
}

编译：

bash 复制代码 
gcc -o client client.c

5.2、Windows下可以使用NetAssist的网络助手工具

下载地址：http://old.tpyboard.com/downloads/NetAssist.exe

小结

这里基于上一个章节的内容进行了升级，可以接受多个客户端同时连接，并为每个连接克隆一个子进程进行通信。

但是，使用多进程会非常消耗资源，特别是高并发的时候，会是系统内存爆满，开销巨大。那么有没有办法在一个线程中就可以完成高并发通信呢？答案是可以的，下一章节将介绍使用select实现一个线程完成多个连接的通信，也就是IO多路复用技术。

相关推荐
vip4512 分钟前
Linux 经典面试八股文
linux
大霞上仙4 分钟前
Ubuntu系统电脑没有WiFi适配器
linux·运维·电脑
weixin_4426434223 分钟前
推荐FileLink数据跨网摆渡系统 — 安全、高效的数据传输解决方案
服务器·网络·安全·filelink数据摆渡系统
阑梦清川37 分钟前
JavaEE初阶---网络原理（五）---HTTP协议
网络·http·java-ee
Karoku06640 分钟前
【企业级分布式系统】Zabbix监控系统与部署安装
运维·服务器·数据库·redis·mysql·zabbix
半桶水专家1 小时前
用go实现创建WebSocket服务器
服务器·websocket·golang
布值倒区什么name1 小时前
bug日常记录responded with a status of 413 (Request Entity Too Large)
运维·服务器·bug
～yY…s<#>1 小时前
【刷题17】最小栈、栈的压入弹出、逆波兰表达式
c语言·数据结构·c++·算法·leetcode
阿尔帕兹1 小时前
构建 HTTP 服务端与 Docker 镜像：从开发到测试
网络协议·http·docker
孤客网络科技工作室1 小时前
VMware 虚拟机使用教程及 Kali Linux 安装指南
linux·虚拟机·kali linux
热门推荐
01【HarmonyOS】HUAWEI DevEco Studio 下载地址汇总02(欧拉)openEuler系统添加网卡文件配置流程、(欧拉)openEuler系统手动配置ipv6地址流程、(欧拉)openEuler系统网络管理说明03组基轨迹建模 GBTM的介绍与实现（Stata 或 R）04【AIGC】重塑未来的科技巨轮05全面解析：构建基于深度学习的安全帽检测系统（UI界面+YOLO代码+数据集）06【经验分享】Ubuntu22.04安装微信(linux官方版)07基于YOLOv10深度学习的CT扫描图像肾结石智能检测系统【python源码+Pyqt5界面+数据集+训练代码】深度学习实战、目标检测08Ubuntu 20.04使用Livox mid 360 测试 FAST_LIO09RAG 实践- Ollama+RagFlow 部署本地知识库10【TC3xx芯片】TC3xx芯片电源管理系统PMS详解