【在Linux世界中追寻伟大的One Piece】HTTP Session

#pragma once
#include <iostream>
#include <string>
#include <memory>
#include <ctime>
#include <unistd.h>
#include <unordered_map>
<p>// 用来进行测试说明
class Session {
public:
Session(const std::string& username, const std::string& status)
: _username(username), _status(status) {
_create_time = time(nullptr); // 获取时间戳就行了，后面实际需要，就转化就转换一下
}
~Session() {}</p><p>public:
std::string _username;
std::string _status;
uint64_t _create_time;
// 当然还可以再加任何其他信息，看你的需求
};</p><p>using session_ptr = std::shared_ptr<Session>;</p><p>class SessionManager {
public:
SessionManager() {
srand(time(nullptr) ^ getpid());
}</p><pre class="brush:php;toolbar:false">std::string AddSession(session_ptr s) {
    uint32_t randomid = rand() + time(nullptr); // 随机数+时间戳，实际有形成 sessionid 的库，比如 boost uuid 库，或者其他第三方库等
    std::string sessionid = std::to_string(randomid);
    _sessions.insert(std::make_pair(sessionid, s));
    return sessionid;
}

session_ptr GetSession(const std::string sessionid) {
    if (_sessions.find(sessionid) == _sessions.end()) return nullptr;
    return _sessions[sessionid];
}

~SessionManager() {}

#pragma once</p><h1>include <iostream></h1><h1>include <string></h1><h1>include <sstream></h1><h1>include <vector></h1><h1>include <memory></h1><h1>include <ctime></h1><h1>include <functional></h1><h1>include "TcpServer.hpp"</h1><h1>include "Session.hpp" // 引入 session</h1><p>const std::string HttpSep = "\r\n";</p><p>// 可以配置的
const std::string homepage = "index.html";
const std::string wwwroot = "./wwwroot";</p><p>class HttpRequest {
public:
HttpRequest() : _req_blank(HttpSep), _path(wwwroot) {}</p><pre class="brush:php;toolbar:false">bool GetLine(std::string& str, std::string* line) {
    auto pos = str.find(HttpSep);
    if (pos == std::string::npos) return false;
    *line = str.substr(0, pos); // \r\n
    str.erase(0, pos + HttpSep.size());
    return true;
}

void Parse() {
    // 解析出来 url
    std::stringstream ss(_req_line);
    ss >> _method >> _url >> _http_version;

    // 查找 cookies
    std::string prefix = "Cookie: ";
    for (auto& line : _req_header) {
        std::string cookie;
        if (strncmp(line.c_str(), prefix.c_str(), prefix.size()) == 0) // 找到了
        {
            cookie = line.substr(prefix.size()); // 截取"Cookie: "之后的就行了
            _cookies.emplace_back(cookie);
            break;
        }
    }

    // 查找 sessionid
    prefix = "sessionid=";
    for (const auto& cookie : _cookies) {
        if (strncmp(cookie.c_str(), prefix.c_str(), prefix.size()) == 0) {
            _sessionid = cookie.substr(prefix.size()); // 截取"sessionid="之后的就行了
            break;
        }
    }
}

// 其他成员变量和方法...

std::string _req_line;
std::vector<std::string> _req_header;
std::string _req_blank;
std::string _req_content;

// 解析之后的内容
std::string _method;
std::string _url; // / /dira/dirb/x.html / dira / dirb / XX ? usrname = 100 && password = 1234 / dira / dirb
std::string _http_version;
std::string _path; // "./wwwroot"
std::string _suffix; // 请求资源的后缀
std::vector<std::string> _cookies; // 其实 cookie 可以有多个，因为 Set - Cookie 可以被写多条，测试，一条够了。
std::string _sessionid; // 请求携带的 sessionid，仅仅用来测试

void SetCode(int code) {
    _status_code = code;
}

void SetDesc(const std::string& desc) {
    _status_code_desc = desc;
}

void MakeStatusLine() {
    _status_line = _http_version + BlankSep +
        std::to_string(_status_code) + BlankSep + _status_code_desc +
        LineSep;
}

void AddHeader(const std::string& header) {
    _resp_header.push_back(header + LineSep);
}

void AddContent(const std::string& content) {
    _resp_content = content;
}

std::string Serialize() {
    MakeStatusLine();
    std::string response_str = _status_line;
    for (auto& header : _resp_header) {
        response_str += header;
    }
    response_str += _resp_blank;
    response_str += _resp_content;
    return response_str;
}

~HttpResponse() {}

// httpversion StatusCode StatusCodeDesc
std::string _http_version;
int _status_code;
std::string _status_code_desc;

std::string GetWeekDayName(int day) {
    std::vector<std::string> weekdays = { "Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat" };
    return weekdays[day];
}

std::string ExpireTimeUseRfc1123(int t) // 秒级别的未来 UTC 时间
{
    time_t timeout = time(nullptr) + t;
    struct tm* tm = gmtime(&timeout); // 这里不能用 localtime，因为 localtime 是默认带了时区的.gmtime 获取的就是 UTC 统一时间
    char timebuffer[1024];
    // 时间格式如: expires=Thu, 18 Dec 2024 12:00:00 UTC
    snprintf(timebuffer, sizeof(timebuffer), "%s, %02d %s %d %02d:%02d:%02d UTC",
        GetWeekDayName(tm->tm_wday).c_str(),
        tm->tm_mday,
        GetMonthName(tm->tm_mon).c_str(),
        tm->tm_year + 1900,
        tm->tm_hour,
        tm->tm_min,
        tm->tm_sec);
    return timebuffer;
}

std::string ProveCookieWrite() // 证明 cookie 能被写入浏览器
{
    return "Set-Cookie: username=zhangsan;";
}

std::string ProveCookieTimeOut() {
    return "Set-Cookie: username=zhangsan; expires=" + ExpireTimeUseRfc1123(60) + ";"; // 让 cookie 1min 后过期
}

std::string ProvePath() {
    return "Set-Cookie: username=zhangsan; path=/a/b;";
}

std::string ProveSession(const std::string& session_id) {
    return "Set-Cookie: sessionid=" + session_id + ";";
}

std::string HandlerHttp(std::string request) {
    HttpRequest req;
    HttpResponse resp;
    req.Deserialize(request);
    req.Parse();

    // req.DebugHttp();
    // std::cout << "用户" << user << "已登录" << std::endl;
    std::string sessionid = _session_manager->AddSession(s);
    lg.LogMessage(Debug, "%s 被添加, sessionid是: %s\n", user.c_str(), sessionid.c_str());
    resp.AddHeader(ProveSession(sessionid));
} else {
    // 当浏览器在本站点任何路径中活跃，都会自动提交 sessionid, 我们就能知道谁活跃了.
    std::string sessionid = req.SessionId();
    if (!sessionid.empty()) {
        session_ptr s = _session_manager->GetSession(sessionid);
        // 这个地方有坑，一定要判断服务器端 session 对象是否存在，因为可能测试的时候
        // 浏览器还有历史 sessionid，但是服务器重启之后，session 对象没有了.
        if (s != nullptr)
            lg.LogMessage(Debug, "%s 正在活跃.\n", s->_username.c_str());
        else
            lg.LogMessage(Debug, "cookie : %s 已经过期, 需要清理\n", sessionid.c_str());
    }
}

resp.SetCode(200);
resp.SetDesc("OK");
resp.AddHeader("Content-Type: text/html");
// resp.AddHeader(ProveCookieWrite()); //测试 cookie 被写入与自动提交
// resp.AddHeader(ProveCookieTimeOut()); //测试过期时间的写入
// resp.AddHeader(ProvePath()); // 测试路径
resp.AddContent("<h1>helloworld</h1>");
return resp.Serialize();