使用基本的Linux系统调用来创建一个TCP socket,监听端口8080,并在接受到客户端连接时创建一个新的线程来处理连接。每个连接处理函数都是一个独立的线程,读取客户端请求并发送固定的HTTP响应。
代码:
cpp
#include <iostream>
#include <string>
#include <cstring>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <pthread.h>
const int BUFFER_SIZE = 1024;
void *connection_handler(void *socket_desc) {
int sock = *(int *)socket_desc;
char buffer[BUFFER_SIZE];
std::string response = "HTTP/1.1 200 OK\r\nContent-Type: text/html\r\n\r\nHello, World!";
// 读取客户端请求
ssize_t read_size = read(sock, buffer, BUFFER_SIZE - 1);
if (read_size > 0) {
buffer[read_size] = '\0';
std::cout << "Received request:\n" << buffer << std::endl;
// 发送响应
write(sock, response.c_str(), response.length());
}
// 关闭连接
close(sock);
delete (int *)socket_desc;
return NULL;
}
int main() {
int server_fd, client_sock;
struct sockaddr_in server_addr, client_addr;
socklen_t addr_len = sizeof(struct sockaddr_in);
// 创建 socket
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if (server_fd == -1) {
std::cerr << "Failed to create socket." << std::endl;
return 1;
}
// 设置服务器地址和端口
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(8080);
// 绑定地址和端口
if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
std::cerr << "Failed to bind." << std::endl;
return 1;
}
// 监听连接
if (listen(server_fd, 10) < 0) {
std::cerr << "Failed to listen." << std::endl;
return 1;
}
std::cout << "Server started. Listening on port 8080..." << std::endl;
while (true) {
// 接受客户端连接
client_sock = accept(server_fd, (struct sockaddr *)&client_addr, &addr_len);
if (client_sock < 0) {
std::cerr << "Failed to accept connection." << std::endl;
continue;
}
std::cout << "New connection accepted." << std::endl;
// 创建线程处理连接
pthread_t thread;
int *new_sock = new int;
*new_sock = client_sock;
if (pthread_create(&thread, NULL, connection_handler, (void *)new_sock) < 0) {
std::cerr << "Failed to create thread." << std::endl;
return 1;
}
// 分离线程,使其自行释放资源
pthread_detach(thread);
}
// 关闭服务器 socket
close(server_fd);
return 0;
}
代码解析:
这段代码实现了一个简单的基于Linux的轻量级多线程HTTP服务。它的主要逻辑如下:
-
引入所需的头文件和库:包括iostream(用于输入输出)、string(用于处理字符串)、cstring(用于字符串操作)、sys/socket.h和arpa/inet.h(用于socket编程)、unistd.h(用于关闭连接)和pthread.h(用于多线程编程)。
-
定义常量和变量:将缓冲区大小定义为BUFFER_SIZE,并声明服务器socket文件描述符(server_fd)和客户端socket文件描述符(client_sock),以及服务器地址(server_addr)和客户端地址(client_addr)。
-
创建连接处理函数:定义了一个名为
connection_handler
的函数,该函数作为每个连接的处理函数。它接受一个void*
类型的参数,将其转换为int*
类型,表示客户端socket文件描述符。在该函数中,首先从客户端读取请求并打印到控制台,然后向客户端发送固定的HTTP响应。最后关闭客户端socket连接,并释放分配的内存。 -
主函数:在主函数中,首先创建服务器socket(server_fd)。然后设置服务器地址和端口,并将其与socket绑定。接下来,开始监听连接请求。在一个无限循环中,使用
accept
函数接受客户端连接,并创建一个新的线程来处理连接。为了能够将客户端socket传递给新线程,使用new
运算符动态分配了一个int类型的指针new_sock,并将客户端socket文件描述符赋值给它。然后,使用pthread_create
函数创建一个新的线程,将新线程的执行函数设置为connection_handler
,并将new_sock作为参数传递给新线程。最后,使用pthread_detach
函数将新线程设置为分离状态,使其在完成后自动释放资源。 -
关闭服务器socket:在无限循环中无法达到的代码,当服务端结束时,关闭服务器socket(server_fd)。
这段代码使用了基本的Linux系统调用和pthread库来实现一个简单的多线程HTTP服务。它接受客户端连接,并为每个连接创建一个独立的线程来处理请求和发送响应。需要注意的是,这只是一个简化的示例,实际的HTTP服务器需要考虑更多的功能和错误处理,例如处理HTTP请求的各种方法、路径解析、错误处理等。