基于 Linux 的轻量级多线程 HTTP 服务

使用基本的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服务。它的主要逻辑如下:

  1. 引入所需的头文件和库:包括iostream(用于输入输出)、string(用于处理字符串)、cstring(用于字符串操作)、sys/socket.h和arpa/inet.h(用于socket编程)、unistd.h(用于关闭连接)和pthread.h(用于多线程编程)。

  2. 定义常量和变量:将缓冲区大小定义为BUFFER_SIZE,并声明服务器socket文件描述符(server_fd)和客户端socket文件描述符(client_sock),以及服务器地址(server_addr)和客户端地址(client_addr)。

  3. 创建连接处理函数:定义了一个名为connection_handler的函数,该函数作为每个连接的处理函数。它接受一个void*类型的参数,将其转换为int*类型,表示客户端socket文件描述符。在该函数中,首先从客户端读取请求并打印到控制台,然后向客户端发送固定的HTTP响应。最后关闭客户端socket连接,并释放分配的内存。

  4. 主函数:在主函数中,首先创建服务器socket(server_fd)。然后设置服务器地址和端口,并将其与socket绑定。接下来,开始监听连接请求。在一个无限循环中,使用accept函数接受客户端连接,并创建一个新的线程来处理连接。为了能够将客户端socket传递给新线程,使用new运算符动态分配了一个int类型的指针new_sock,并将客户端socket文件描述符赋值给它。然后,使用pthread_create函数创建一个新的线程,将新线程的执行函数设置为connection_handler,并将new_sock作为参数传递给新线程。最后,使用pthread_detach函数将新线程设置为分离状态,使其在完成后自动释放资源。

  5. 关闭服务器socket:在无限循环中无法达到的代码,当服务端结束时,关闭服务器socket(server_fd)。

这段代码使用了基本的Linux系统调用和pthread库来实现一个简单的多线程HTTP服务。它接受客户端连接,并为每个连接创建一个独立的线程来处理请求和发送响应。需要注意的是,这只是一个简化的示例,实际的HTTP服务器需要考虑更多的功能和错误处理,例如处理HTTP请求的各种方法、路径解析、错误处理等。

相关推荐
远游客07132 小时前
centos stream 8下载安装遇到的坑
linux·服务器·centos
马甲是掉不了一点的<.<2 小时前
本地电脑使用命令行上传文件至远程服务器
linux·scp·cmd·远程文件上传
jingyu飞鸟2 小时前
centos-stream9系统安装docker
linux·docker·centos
XH华2 小时前
初识C语言之二维数组(下)
c语言·算法
超爱吃士力架2 小时前
邀请逻辑
java·linux·后端
南宫生2 小时前
力扣-图论-17【算法学习day.67】
java·学习·算法·leetcode·图论
不想当程序猿_3 小时前
【蓝桥杯每日一题】求和——前缀和
算法·前缀和·蓝桥杯
落魄君子3 小时前
GA-BP分类-遗传算法(Genetic Algorithm)和反向传播算法(Backpropagation)
算法·分类·数据挖掘
菜鸡中的奋斗鸡→挣扎鸡3 小时前
滑动窗口 + 算法复习
数据结构·算法
Lenyiin3 小时前
第146场双周赛:统计符合条件长度为3的子数组数目、统计异或值为给定值的路径数目、判断网格图能否被切割成块、唯一中间众数子序列 Ⅰ
c++·算法·leetcode·周赛·lenyiin