135 lines
4.2 KiB
C++
135 lines
4.2 KiB
C++
/*
|
||
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT-like license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#include "HttpCookie.h"
|
||
#include "Util/util.h"
|
||
#include "Util/onceToken.h"
|
||
|
||
#if defined(_WIN32)
|
||
#include "Util/strptime_win.h"
|
||
#endif
|
||
|
||
using namespace toolkit;
|
||
using namespace std;
|
||
|
||
namespace mediakit {
|
||
|
||
void HttpCookie::setPath(const string &path) {
|
||
_path = path;
|
||
}
|
||
|
||
void HttpCookie::setHost(const string &host) {
|
||
_host = host;
|
||
}
|
||
|
||
// from https://gmbabar.wordpress.com/2010/12/01/mktime-slow-use-custom-function/#comment-58
|
||
static time_t time_to_epoch(const struct tm *ltm, int utcdiff) {
|
||
const int mon_days[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
|
||
long tyears, tdays, leaps, utc_hrs;
|
||
int i;
|
||
|
||
tyears = ltm->tm_year - 70; // tm->tm_year is from 1900.
|
||
leaps = (tyears + 2) / 4; // no of next two lines until year 2100.
|
||
// i = (ltm->tm_year – 100) / 100; [AUTO-TRANSLATED:12beea30]
|
||
// i = (ltm->tm_year – 100) / 100;
|
||
// leaps -= ( (i/4)*3 + i%4 );
|
||
tdays = 0;
|
||
for (i = 0; i < ltm->tm_mon; i++)
|
||
tdays += mon_days[i];
|
||
|
||
tdays += ltm->tm_mday - 1; // days of month passed.
|
||
tdays = tdays + (tyears * 365) + leaps;
|
||
|
||
utc_hrs = ltm->tm_hour + utcdiff; // for your time zone.
|
||
return (tdays * 86400) + (utc_hrs * 3600) + (ltm->tm_min * 60) + ltm->tm_sec;
|
||
}
|
||
|
||
static time_t timeStrToInt(const string &date) {
|
||
struct tm tt;
|
||
strptime(date.data(), "%a, %b %d %Y %H:%M:%S %Z", &tt);
|
||
// mktime内部有使用互斥锁，非常影响性能 [AUTO-TRANSLATED:b3270635]
|
||
// mktime uses mutex internally, which significantly affects performance
|
||
return time_to_epoch(&tt, getGMTOff() / 3600); // mktime(&tt);
|
||
}
|
||
|
||
void HttpCookie::setExpires(const string &expires, const string &server_date) {
|
||
_expire = timeStrToInt(expires);
|
||
if (!server_date.empty()) {
|
||
_expire = time(NULL) + (_expire - timeStrToInt(server_date));
|
||
}
|
||
}
|
||
|
||
void HttpCookie::setKeyVal(const string &key, const string &val) {
|
||
_key = key;
|
||
_val = val;
|
||
}
|
||
|
||
HttpCookie::operator bool() {
|
||
return !_host.empty() && !_key.empty() && !_val.empty() && (_expire > time(NULL));
|
||
}
|
||
|
||
const string &HttpCookie::getVal() const {
|
||
return _val;
|
||
}
|
||
|
||
const string &HttpCookie::getKey() const {
|
||
return _key;
|
||
}
|
||
|
||
HttpCookieStorage &HttpCookieStorage::Instance() {
|
||
static HttpCookieStorage instance;
|
||
return instance;
|
||
}
|
||
|
||
void HttpCookieStorage::set(const HttpCookie::Ptr &cookie) {
|
||
lock_guard<mutex> lck(_mtx_cookie);
|
||
if (!cookie || !(*cookie)) {
|
||
return;
|
||
}
|
||
_all_cookie[cookie->_host][cookie->_path][cookie->_key] = cookie;
|
||
}
|
||
|
||
vector<HttpCookie::Ptr> HttpCookieStorage::get(const string &host, const string &path) {
|
||
vector<HttpCookie::Ptr> ret(0);
|
||
lock_guard<mutex> lck(_mtx_cookie);
|
||
auto it = _all_cookie.find(host);
|
||
if (it == _all_cookie.end()) {
|
||
// 未找到该host相关记录 [AUTO-TRANSLATED:0655542a]
|
||
// No record found for this host
|
||
return ret;
|
||
}
|
||
// 遍历该host下所有path [AUTO-TRANSLATED:94ca2180]
|
||
// Traverse all paths under this host
|
||
for (auto &pr : it->second) {
|
||
if (path.find(pr.first) != 0) {
|
||
// 这个path不匹配 [AUTO-TRANSLATED:3ec99732]
|
||
// This path does not match
|
||
continue;
|
||
}
|
||
// 遍历该path下的各个cookie [AUTO-TRANSLATED:ceab9c83]
|
||
// Traverse all cookies under this path
|
||
for (auto it_cookie = pr.second.begin(); it_cookie != pr.second.end();) {
|
||
if (!*(it_cookie->second)) {
|
||
// 该cookie已经过期，移除之 [AUTO-TRANSLATED:52762286]
|
||
// This cookie has expired, remove it
|
||
it_cookie = pr.second.erase(it_cookie);
|
||
continue;
|
||
}
|
||
// 保存有效cookie [AUTO-TRANSLATED:bd875507]
|
||
// Save valid cookies
|
||
ret.emplace_back(it_cookie->second);
|
||
++it_cookie;
|
||
}
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
} /* namespace mediakit */
|