9.4练习

多进程多线程并发服务器

多进程并发执行

模型:

定义信号处理函数,非阻塞回收僵尸进程。

绑定子进程退出时的信号。

  1. 创建套接字
  2. 绑定
  3. 监听
  4. 循环接收客户端信息
  5. 让父进程接收客户端请求并关闭上次用于通话的新文件描述符,子进程负责数据收发。
  6. 子进程关闭新描述符并退出
  7. 关闭旧的文件描述符。
#include <myhead.h> // 包含自定义头文件

#define BACKLOG 20
#define SERPORT 8888
#define SERIP "192.168.0.132"

void handle(int a) {
    if (a == SIGCHLD) {
        while (waitpid(-1, NULL, WNOHANG) != -1);
    }
}

int main() {
    // 注册信号处理函数
    if (signal(SIGCHLD, handle) == SIG_ERR) {
        perror("signal");
        return -1;
    }

    // 创建套接字
    int oldfd = socket(AF_INET, SOCK_STREAM, 0);
    if (oldfd == -1) {
        perror("socket");
        return -1;
    }

    // 设置端口快速复用
    int k = 1;
    if (setsockopt(oldfd, SOL_SOCKET, SO_REUSEPORT, &k, sizeof(k)) == -1) {
        perror("setsockopt");
        return -1;
    }
    printf("端口快速复用成功\n");

    // 绑定地址和端口
    struct sockaddr_in sin = {
        .sin_family = AF_INET,
        .sin_port = htons(SERPORT),
        .sin_addr.s_addr = inet_addr(SERIP)
    };

    if (bind(oldfd, (struct sockaddr *)&sin, sizeof(sin)) == -1) {
        perror("bind");
        return -1;
    }

    // 监听连接
    if (listen(oldfd, BACKLOG) == -1) {
        perror("listen");
        return -1;
    }

    // 接收和处理客户端连接
    struct sockaddr_in cin;
    int cinlen = sizeof(cin);

    while (1) {
        int newfd = accept(oldfd, (struct sockaddr *)&cin, &cinlen);
        if (newfd == -1) {
            perror("accept");
            return -1;
        }
        printf("%s:%d 链接成功\n", inet_ntoa(cin.sin_addr), ntohs(sin.sin_port));

        pid_t pid = fork();
        if (pid > 0) { // 父进程
            close(newfd);
        } else if (pid == 0) { // 子进程
            close(oldfd);

            char buff[1024];
            while (1) {
                memset(buff, 0, sizeof(buff));
                int len = recv(newfd, buff, sizeof(buff), 0);
                if (len == 0) {
                    printf("客户端退出\n");
                    break;
                }
                printf("服务器收到消息:%s\n", buff);
                strcat(buff, "^_^");
                send(newfd, buff, strlen(buff), 0);
                printf("发送成功\n");
                memset(buff, 0, sizeof(buff));
            }

            close(newfd);
            exit(0); // 退出子进程
        } else {
            perror("fork");
            return -1;
        }
    }

    close(oldfd);

    return 0;
}

代码讲解

1. 宏定义
#define BACKLOG 20
#define SERPORT 8888
#define SERIP "192.168.0.132"
2. 信号处理函数

接下来,定义了一个信号处理函数 handle,用于处理 SIGCHLD 信号:

void handle(int a) {
    if (a == SIGCHLD) {
        while (waitpid(-1, NULL, WNOHANG) != -1);
    }
}

当子进程终止时,父进程会收到 SIGCHLD 信号。在信号处理函数中调用 waitpid,以非阻塞方式回收所有已终止的子进程,防止产生僵尸进程。

3. 主函数

在主函数中,首先注册信号处理函数:

if (signal(SIGCHLD, handle) == SIG_ERR) {
    perror("signal");
    return -1;
}

创建一个 TCP 套接字:

int oldfd = socket(AF_INET, SOCK_STREAM, 0);
if (oldfd == -1) {
    perror("socket");
    return -1;
}

设置端口快速复用选项:

int k = 1;
if (setsockopt(oldfd, SOL_SOCKET, SO_REUSEPORT, &k, sizeof(k)) == -1) {
    perror("setsockopt");
    return -1;
}
printf("端口快速复用成功\n");
4.绑定套接字到指定的 IP 地址和端口
struct sockaddr_in sin = {
    .sin_family = AF_INET,
    .sin_port = htons(SERPORT),
    .sin_addr.s_addr = inet_addr(SERIP)
};
if (bind(oldfd, (struct sockaddr )&sin, sizeof(sin)) == -1) {
    perror("bind");
    return -1;
}
5.监听连接请求
if (listen(oldfd, BACKLOG) == -1) {
    perror("listen");
    return -1;
}
6. 接收和处理客户端连接

在主循环中,接受客户端连接,并创建子进程处理每个连接:

struct sockaddr_in cin;
int cinlen = sizeof(cin);
while (1) {
    int newfd = accept(oldfd, (struct sockaddr )&cin, &cinlen);
    if (newfd == -1) {
        perror("accept");
        return -1;
    }
    printf("%s:%d 链接成功\n", inet_ntoa(cin.sin_addr), ntohs(sin.sin_port));
    pid_t pid = fork();
    if (pid > 0) { // 父进程
        close(newfd);
    } else if (pid == 0) { // 子进程
        close(oldfd);
        char buff[1024];
        while (1) {
            memset(buff, 0, sizeof(buff));
            int len = recv(newfd, buff, sizeof(buff), 0);
            if (len == 0) {
                printf("客户端退出\n");
                break;
            }
            printf("服务器收到消息:%s\n", buff);
            strcat(buff, "^^");
            send(newfd, buff, strlen(buff), 0);
            printf("发送成功\n");
            memset(buff, 0, sizeof(buff));
        }
        close(newfd);
        exit(0); // 退出子进程
    } else {
        perror("fork");
        return -1;
    }
}

父进程负责关闭新套接字,子进程负责接收和发送数据。子进程接收客户端消息,添加表情符号后发送回去,直到客户端退出。

7. 关闭套接字

关闭套接字:

close(oldfd);

多线程并发执行

在这篇博客中,将详细讲解如何实现一个简单的多线程TCP服务器。这个服务器能够接受多个客户端的连接,并处理每个客户端发送的数据。将使用C语言和POSIX标准库来实现这个服务器。

模型:
线程函数:
    收发消息
    关闭新描述符
    子线程退出
 
    
     建立原始套接字
     绑定主机IP
     监听客户端
     循环: 
            accept:获取客户端请求
            建立子线程
            线程挂起                
    关闭旧的文件描述符

代码示例

首先,来看一下完整的代码示例:

#include <myhead.h>
#define BACKLOG 20
#define SERPORT 9999
#define SERIP "192.168.0.132"
// 定义一个结构体,用于传递新文件描述符和客户端信息
typedef struct {
    int newfd; // 存储新的文件描述符
    struct sockaddr_in cin; // 存储客户端信息
} SIG;
// 线程处理函数
void fun1(void as) {
    int newfd = ((SIG )as)->newfd;
    struct sockaddr_in cin = ((SIG )as)->cin;
    char buff[1024];
    while (1) {
        memset(buff, 0, sizeof(buff));
        int len = recv(newfd, buff, sizeof(buff), 0);
        printf("收到消息: %s\n", buff);
        if (len == 0) {
            printf("客户端退出\n");
            break;
        }
        strcat(buff, "Received");
        send(newfd, buff, strlen(buff), 0);
        printf("发送成功\n");
        memset(buff, 0, sizeof(buff));
    }
    // 关闭新的描述符
    close(newfd);
    pthread_exit(NULL);
}
int main(int argc, const char argv[]) {
    // 创建套接字
    int oldfd = socket(AF_INET, SOCK_STREAM, 0);
    if (oldfd == -1) {
        perror("socket");
        return -1;
    }
    // 设置端口号复用
    int aoto = 1;
    if (setsockopt(oldfd, SOL_SOCKET, SO_REUSEADDR, &aoto, sizeof(aoto)) == -1) {
        perror("setsockopt");
        return -1;
    }
    printf("端口号复用成功\n");
    // 绑定地址和端口
    struct sockaddr_in sin = {
        .sin_family = AF_INET,
        .sin_port = htons(SERPORT),
        .sin_addr.s_addr = inet_addr(SERIP)
    };
    if (bind(oldfd, (struct sockaddr )&sin, sizeof(sin)) == -1) {
        perror("bind");
        return -1;
    }
    // 监听连接
    if (listen(oldfd, BACKLOG) == -1) {
        perror("listen");
        return -1;
    }
    printf("监听成功\n");
    struct sockaddr_in cin;
    int cinlen = sizeof(cin);
    SIG MSG;
    while (1) {
        // 接收客户端的请求
        int newfd = accept(oldfd, (struct sockaddr )&cin, &cinlen);
        if (newfd == -1) {
            perror("accept");
            return -1;
        }
        printf("%s:%d 发来请求\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
        MSG.newfd = newfd;
        MSG.cin = cin;
        // 创建子线程与客户端交流
        pthread_t tid;
        if (pthread_create(&tid, NULL, fun1, &MSG) != 0) {
            perror("pthread_create");
            return -1;
        }
        // 分离子线程,回收资源
        pthread_detach(tid);
    }
    // 关闭旧的描述符
    close(oldfd);
    return 0;
}

代码讲解

1. 定义结构体

定义了一个结构体 SIG,用于传递新文件描述符和客户端信息:

typedef struct {
    int newfd; // 存储新的文件描述符
    struct sockaddr_in cin; // 存储客户端信息
} SIG;
2. 线程处理函数

定义了一个线程处理函数 fun1,用于处理客户端的通信:

void fun1(void as) {
    int newfd = ((SIG )as)->newfd;
    struct sockaddr_in cin = ((SIG )as)->cin;
    char buff[1024];
    while (1) {
        memset(buff, 0, sizeof(buff));
        int len = recv(newfd, buff, sizeof(buff), 0);
        printf("收到消息: %s\n", buff);
        if (len == 0) {
            printf("客户端退出\n");
            break;
        }
        strcat(buff, "Received");
        send(newfd, buff, strlen(buff), 0);
        printf("发送成功\n");
        memset(buff, 0, sizeof(buff));
    }
    // 关闭新的描述符
    close(newfd);
    pthread_exit(NULL);
}
3. 主函数

在主函数中,创建一个 TCP 套接字:

int oldfd = socket(AF_INET, SOCK_STREAM, 0);
if (oldfd == -1) {
    perror("socket");
    return -1;
}

设置端口号复用选项:

int aoto = 1;
if (setsockopt(oldfd, SOL_SOCKET, SO_REUSEADDR, &aoto, sizeof(aoto)) == -1) {
    perror("setsockopt");
    return -1;
}
printf("端口号复用成功\n");

初始化服务器地址结构体 sin,并绑定套接字到指定的 IP 地址和端口:

struct sockaddr_in sin = {
    .sin_family = AF_INET,
    .sin_port = htons(SERPORT),
    .sin_addr.s_addr = inet_addr(SERIP)
};
if (bind(oldfd, (struct sockaddr )&sin, sizeof(sin)) == -1) {
    perror("bind");
    return -1;
}

开始监听连接请求,最大连接数为 BACKLOG

if (listen(oldfd, BACKLOG) == -1) {
    perror("listen");
    return -1;
}
printf("监听成功\n");
4. 接收和处理客户端连接

在主循环中,接受客户端连接,并创建子线程处理每个连接:

struct sockaddr_in cin;
int cinlen = sizeof(cin);
SIG MSG;
while (1) {
    // 接收客户端的请求
    int newfd = accept(oldfd, (struct sockaddr )&cin, &cinlen);
    if (newfd == -1) {
        perror("accept");
        return -1;
    }
    printf("%s:%d 发来请求\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
    MSG.newfd = newfd;
    MSG.cin = cin;
    // 创建子线程与客户端交流
    pthread_t tid;
    if (pthread_create(&tid, NULL, fun1, &MSG) != 0) {
        perror("pthread_create");
        return -1;
    }
    // 分离子线程,回收资源
    pthread_detach(tid);
}

父进程负责接受客户端连接,并创建子线程处理每个连接。子线程负责接收和发送数据,直到客户端退出。

5. 关闭套接字

最后,关闭套接字:

close(oldfd);
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值