多进程多线程并发服务器
多进程并发执行
模型:
定义信号处理函数,非阻塞回收僵尸进程。
绑定子进程退出时的信号。
- 创建套接字
- 绑定
- 监听
- 循环接收客户端信息
- 让父进程接收客户端请求并关闭上次用于通话的新文件描述符,子进程负责数据收发。
- 子进程关闭新描述符并退出
- 关闭旧的文件描述符。
#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);