2019-11-30 11:38:00 +08:00
|
|
|
|
/*
|
2020-04-04 20:30:09 +08:00
|
|
|
|
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
2019-11-30 11:38:00 +08:00
|
|
|
|
*
|
|
|
|
|
* This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
|
|
|
|
|
*
|
2020-04-04 20:30:09 +08:00
|
|
|
|
* Use of this source code is governed by MIT 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.
|
2019-11-30 11:38:00 +08:00
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include <sys/stat.h>
|
|
|
|
|
#if !defined(_WIN32)
|
|
|
|
|
#include <dirent.h>
|
|
|
|
|
#endif //!defined(_WIN32)
|
|
|
|
|
#include <iomanip>
|
|
|
|
|
#include "HttpFileManager.h"
|
|
|
|
|
#include "Util/File.h"
|
|
|
|
|
#include "HttpSession.h"
|
2019-12-28 18:50:56 +08:00
|
|
|
|
#include "Record/HlsMediaSource.h"
|
2019-11-30 11:38:00 +08:00
|
|
|
|
|
|
|
|
|
namespace mediakit {
|
|
|
|
|
|
2019-11-30 14:29:44 +08:00
|
|
|
|
// hls的播放cookie缓存时间默认60秒,
|
|
|
|
|
// 每次访问一次该cookie,那么将重新刷新cookie有效期
|
|
|
|
|
// 假如播放器在60秒内都未访问该cookie,那么将重新触发hls播放鉴权
|
|
|
|
|
static int kHlsCookieSecond = 60;
|
2019-11-30 11:38:00 +08:00
|
|
|
|
static const string kCookieName = "ZL_COOKIE";
|
2019-11-30 14:29:44 +08:00
|
|
|
|
static const string kHlsSuffix = "/hls.m3u8";
|
2020-01-02 17:46:20 +08:00
|
|
|
|
|
|
|
|
|
class HttpCookieAttachment{
|
|
|
|
|
public:
|
|
|
|
|
HttpCookieAttachment() {};
|
|
|
|
|
~HttpCookieAttachment() {};
|
|
|
|
|
public:
|
|
|
|
|
//cookie生效作用域,本cookie只对该目录下的文件生效
|
|
|
|
|
string _path;
|
|
|
|
|
//上次鉴权失败信息,为空则上次鉴权成功
|
|
|
|
|
string _err_msg;
|
|
|
|
|
//本cookie是否为hls直播的
|
|
|
|
|
bool _is_hls = false;
|
|
|
|
|
//hls直播时的其他一些信息,主要用于播放器个数计数以及流量计数
|
|
|
|
|
HlsCookieData::Ptr _hls_data;
|
|
|
|
|
//如果是hls直播,那么判断该cookie是否使用过MediaSource::findAsync查找过
|
|
|
|
|
//如果程序未正常退出,会残余上次的hls文件,所以判断hls直播是否存在的关键不是文件存在与否
|
|
|
|
|
//而是应该判断HlsMediaSource是否已注册,但是这样会每次获取m3u8文件时都会用MediaSource::findAsync判断一次
|
|
|
|
|
//会导致程序性能低下,所以我们应该在cookie声明周期的第一次判断HlsMediaSource是否已经注册,后续通过文件存在与否判断
|
|
|
|
|
bool _have_find_media_source = false;
|
|
|
|
|
};
|
2019-11-30 11:38:00 +08:00
|
|
|
|
|
2020-01-02 18:24:11 +08:00
|
|
|
|
static const char *s_mime_src[][2] = {
|
|
|
|
|
{"html", "text/html"},
|
|
|
|
|
{"htm", "text/html"},
|
|
|
|
|
{"shtml", "text/html"},
|
|
|
|
|
{"css", "text/css"},
|
|
|
|
|
{"xml", "text/xml"},
|
|
|
|
|
{"gif", "image/gif"},
|
|
|
|
|
{"jpeg", "image/jpeg"},
|
|
|
|
|
{"jpg", "image/jpeg"},
|
|
|
|
|
{"js", "application/javascript"},
|
|
|
|
|
{"map", "application/javascript" },
|
|
|
|
|
{"atom", "application/atom+xml"},
|
|
|
|
|
{"rss", "application/rss+xml"},
|
|
|
|
|
{"mml", "text/mathml"},
|
|
|
|
|
{"txt", "text/plain"},
|
|
|
|
|
{"jad", "text/vnd.sun.j2me.app-descriptor"},
|
|
|
|
|
{"wml", "text/vnd.wap.wml"},
|
|
|
|
|
{"htc", "text/x-component"},
|
|
|
|
|
{"png", "image/png"},
|
|
|
|
|
{"tif", "image/tiff"},
|
|
|
|
|
{"tiff", "image/tiff"},
|
|
|
|
|
{"wbmp", "image/vnd.wap.wbmp"},
|
|
|
|
|
{"ico", "image/x-icon"},
|
|
|
|
|
{"jng", "image/x-jng"},
|
|
|
|
|
{"bmp", "image/x-ms-bmp"},
|
|
|
|
|
{"svg", "image/svg+xml"},
|
|
|
|
|
{"svgz", "image/svg+xml"},
|
|
|
|
|
{"webp", "image/webp"},
|
|
|
|
|
{"woff", "application/font-woff"},
|
|
|
|
|
{"woff2","application/font-woff" },
|
|
|
|
|
{"jar", "application/java-archive"},
|
|
|
|
|
{"war", "application/java-archive"},
|
|
|
|
|
{"ear", "application/java-archive"},
|
|
|
|
|
{"json", "application/json"},
|
|
|
|
|
{"hqx", "application/mac-binhex40"},
|
|
|
|
|
{"doc", "application/msword"},
|
|
|
|
|
{"pdf", "application/pdf"},
|
|
|
|
|
{"ps", "application/postscript"},
|
|
|
|
|
{"eps", "application/postscript"},
|
|
|
|
|
{"ai", "application/postscript"},
|
|
|
|
|
{"rtf", "application/rtf"},
|
|
|
|
|
{"m3u8", "application/vnd.apple.mpegurl"},
|
|
|
|
|
{"xls", "application/vnd.ms-excel"},
|
|
|
|
|
{"eot", "application/vnd.ms-fontobject"},
|
|
|
|
|
{"ppt", "application/vnd.ms-powerpoint"},
|
|
|
|
|
{"wmlc", "application/vnd.wap.wmlc"},
|
|
|
|
|
{"kml", "application/vnd.google-earth.kml+xml"},
|
|
|
|
|
{"kmz", "application/vnd.google-earth.kmz"},
|
|
|
|
|
{"7z", "application/x-7z-compressed"},
|
|
|
|
|
{"cco", "application/x-cocoa"},
|
|
|
|
|
{"jardiff", "application/x-java-archive-diff"},
|
|
|
|
|
{"jnlp", "application/x-java-jnlp-file"},
|
|
|
|
|
{"run", "application/x-makeself"},
|
|
|
|
|
{"pl", "application/x-perl"},
|
|
|
|
|
{"pm", "application/x-perl"},
|
|
|
|
|
{"prc", "application/x-pilot"},
|
|
|
|
|
{"pdb", "application/x-pilot"},
|
|
|
|
|
{"rar", "application/x-rar-compressed"},
|
|
|
|
|
{"rpm", "application/x-redhat-package-manager"},
|
|
|
|
|
{"sea", "application/x-sea"},
|
|
|
|
|
{"swf", "application/x-shockwave-flash"},
|
|
|
|
|
{"sit", "application/x-stuffit"},
|
|
|
|
|
{"tcl", "application/x-tcl"},
|
|
|
|
|
{"tk", "application/x-tcl"},
|
|
|
|
|
{"der", "application/x-x509-ca-cert"},
|
|
|
|
|
{"pem", "application/x-x509-ca-cert"},
|
|
|
|
|
{"crt", "application/x-x509-ca-cert"},
|
|
|
|
|
{"xpi", "application/x-xpinstall"},
|
|
|
|
|
{"xhtml", "application/xhtml+xml"},
|
|
|
|
|
{"xspf", "application/xspf+xml"},
|
|
|
|
|
{"zip", "application/zip"},
|
|
|
|
|
{"bin", "application/octet-stream"},
|
|
|
|
|
{"exe", "application/octet-stream"},
|
|
|
|
|
{"dll", "application/octet-stream"},
|
|
|
|
|
{"deb", "application/octet-stream"},
|
|
|
|
|
{"dmg", "application/octet-stream"},
|
|
|
|
|
{"iso", "application/octet-stream"},
|
|
|
|
|
{"img", "application/octet-stream"},
|
|
|
|
|
{"msi", "application/octet-stream"},
|
|
|
|
|
{"msp", "application/octet-stream"},
|
|
|
|
|
{"msm", "application/octet-stream"},
|
|
|
|
|
{"docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document"},
|
|
|
|
|
{"xlsx", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"},
|
|
|
|
|
{"pptx", "application/vnd.openxmlformats-officedocument.presentationml.presentation"},
|
|
|
|
|
{"mid", "audio/midi"},
|
|
|
|
|
{"midi", "audio/midi"},
|
|
|
|
|
{"kar", "audio/midi"},
|
|
|
|
|
{"mp3", "audio/mpeg"},
|
|
|
|
|
{"ogg", "audio/ogg"},
|
|
|
|
|
{"m4a", "audio/x-m4a"},
|
|
|
|
|
{"ra", "audio/x-realaudio"},
|
|
|
|
|
{"3gpp", "video/3gpp"},
|
|
|
|
|
{"3gp", "video/3gpp"},
|
|
|
|
|
{"ts", "video/mp2t"},
|
|
|
|
|
{"mp4", "video/mp4"},
|
|
|
|
|
{"mpeg", "video/mpeg"},
|
|
|
|
|
{"mpg", "video/mpeg"},
|
|
|
|
|
{"mov", "video/quicktime"},
|
|
|
|
|
{"webm", "video/webm"},
|
|
|
|
|
{"flv", "video/x-flv"},
|
|
|
|
|
{"m4v", "video/x-m4v"},
|
|
|
|
|
{"mng", "video/x-mng"},
|
|
|
|
|
{"asx", "video/x-ms-asf"},
|
|
|
|
|
{"asf", "video/x-ms-asf"},
|
|
|
|
|
{"wmv", "video/x-ms-wmv"},
|
|
|
|
|
{"avi", "video/x-msvideo"},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
const string &HttpFileManager::getContentType(const char *name) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
const char *dot;
|
|
|
|
|
dot = strrchr(name, '.');
|
|
|
|
|
static StrCaseMap mapType;
|
|
|
|
|
static onceToken token([&]() {
|
2020-01-02 18:24:11 +08:00
|
|
|
|
for (unsigned int i = 0; i < sizeof (s_mime_src) / sizeof (s_mime_src[0]); ++i) {
|
|
|
|
|
mapType.emplace(s_mime_src[i][0], s_mime_src[i][1]);
|
|
|
|
|
}
|
2019-11-30 11:38:00 +08:00
|
|
|
|
});
|
|
|
|
|
static string defaultType = "text/plain";
|
|
|
|
|
if (!dot) {
|
|
|
|
|
return defaultType;
|
|
|
|
|
}
|
2020-01-02 18:24:11 +08:00
|
|
|
|
auto it = mapType.find(dot + 1);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
if (it == mapType.end()) {
|
|
|
|
|
return defaultType;
|
|
|
|
|
}
|
|
|
|
|
return it->second;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static string searchIndexFile(const string &dir){
|
|
|
|
|
DIR *pDir;
|
|
|
|
|
dirent *pDirent;
|
|
|
|
|
if ((pDir = opendir(dir.data())) == NULL) {
|
|
|
|
|
return "";
|
|
|
|
|
}
|
|
|
|
|
set<string> setFile;
|
|
|
|
|
while ((pDirent = readdir(pDir)) != NULL) {
|
|
|
|
|
static set<const char *,StrCaseCompare> indexSet = {"index.html","index.htm","index"};
|
|
|
|
|
if(indexSet.find(pDirent->d_name) != indexSet.end()){
|
|
|
|
|
string ret = pDirent->d_name;
|
|
|
|
|
closedir(pDir);
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
closedir(pDir);
|
|
|
|
|
return "";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static bool makeFolderMenu(const string &httpPath, const string &strFullPath, string &strRet) {
|
|
|
|
|
string strPathPrefix(strFullPath);
|
|
|
|
|
string last_dir_name;
|
|
|
|
|
if(strPathPrefix.back() == '/'){
|
|
|
|
|
strPathPrefix.pop_back();
|
|
|
|
|
}else{
|
|
|
|
|
last_dir_name = split(strPathPrefix,"/").back();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (!File::is_dir(strPathPrefix.data())) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
stringstream ss;
|
|
|
|
|
ss << "<html>\r\n"
|
|
|
|
|
"<head>\r\n"
|
|
|
|
|
"<title>文件索引</title>\r\n"
|
|
|
|
|
"</head>\r\n"
|
|
|
|
|
"<body>\r\n"
|
|
|
|
|
"<h1>文件索引:";
|
|
|
|
|
|
|
|
|
|
ss << httpPath;
|
|
|
|
|
ss << "</h1>\r\n";
|
|
|
|
|
if (httpPath != "/") {
|
|
|
|
|
ss << "<li><a href=\"";
|
|
|
|
|
ss << "/";
|
|
|
|
|
ss << "\">";
|
|
|
|
|
ss << "根目录";
|
|
|
|
|
ss << "</a></li>\r\n";
|
|
|
|
|
|
|
|
|
|
ss << "<li><a href=\"";
|
|
|
|
|
if(!last_dir_name.empty()){
|
|
|
|
|
ss << "./";
|
|
|
|
|
}else{
|
|
|
|
|
ss << "../";
|
|
|
|
|
}
|
|
|
|
|
ss << "\">";
|
|
|
|
|
ss << "上级目录";
|
|
|
|
|
ss << "</a></li>\r\n";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
DIR *pDir;
|
|
|
|
|
dirent *pDirent;
|
|
|
|
|
if ((pDir = opendir(strPathPrefix.data())) == NULL) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
set<string> setFile;
|
|
|
|
|
while ((pDirent = readdir(pDir)) != NULL) {
|
|
|
|
|
if (File::is_special_dir(pDirent->d_name)) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
if(pDirent->d_name[0] == '.'){
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
setFile.emplace(pDirent->d_name);
|
|
|
|
|
}
|
|
|
|
|
int i = 0;
|
|
|
|
|
for(auto &strFile :setFile ){
|
|
|
|
|
string strAbsolutePath = strPathPrefix + "/" + strFile;
|
|
|
|
|
bool isDir = File::is_dir(strAbsolutePath.data());
|
|
|
|
|
ss << "<li><span>" << i++ << "</span>\t";
|
|
|
|
|
ss << "<a href=\"";
|
|
|
|
|
if(!last_dir_name.empty()){
|
|
|
|
|
ss << last_dir_name << "/" << strFile;
|
|
|
|
|
}else{
|
|
|
|
|
ss << strFile;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if(isDir){
|
|
|
|
|
ss << "/";
|
|
|
|
|
}
|
|
|
|
|
ss << "\">";
|
|
|
|
|
ss << strFile;
|
|
|
|
|
if (isDir) {
|
|
|
|
|
ss << "/</a></li>\r\n";
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
//是文件
|
|
|
|
|
struct stat fileData;
|
|
|
|
|
if (0 == stat(strAbsolutePath.data(), &fileData)) {
|
|
|
|
|
auto &fileSize = fileData.st_size;
|
|
|
|
|
if (fileSize < 1024) {
|
|
|
|
|
ss << " (" << fileData.st_size << "B)" << endl;
|
|
|
|
|
} else if (fileSize < 1024 * 1024) {
|
|
|
|
|
ss << fixed << setprecision(2) << " (" << fileData.st_size / 1024.0 << "KB)";
|
|
|
|
|
} else if (fileSize < 1024 * 1024 * 1024) {
|
|
|
|
|
ss << fixed << setprecision(2) << " (" << fileData.st_size / 1024 / 1024.0 << "MB)";
|
|
|
|
|
} else {
|
|
|
|
|
ss << fixed << setprecision(2) << " (" << fileData.st_size / 1024 / 1024 / 1024.0 << "GB)";
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
ss << "</a></li>\r\n";
|
|
|
|
|
}
|
|
|
|
|
closedir(pDir);
|
|
|
|
|
ss << "<ul>\r\n";
|
|
|
|
|
ss << "</ul>\r\n</body></html>";
|
|
|
|
|
ss.str().swap(strRet);
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//字符串是否以xx结尾
|
|
|
|
|
static bool end_of(const string &str, const string &substr){
|
|
|
|
|
auto pos = str.rfind(substr);
|
|
|
|
|
return pos != string::npos && pos == str.size() - substr.size();
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
//拦截hls的播放请求
|
2019-12-29 11:52:02 +08:00
|
|
|
|
static bool emitHlsPlayed(const Parser &parser, const MediaInfo &mediaInfo, const HttpSession::HttpAccessPathInvoker &invoker,TcpSession &sender){
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//访问的hls.m3u8结尾,我们转换成kBroadcastMediaPlayed事件
|
2019-12-29 11:52:02 +08:00
|
|
|
|
Broadcast::AuthInvoker mediaAuthInvoker = [invoker](const string &err){
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//cookie有效期为kHlsCookieSecond
|
|
|
|
|
invoker(err,"",kHlsCookieSecond);
|
|
|
|
|
};
|
2019-12-29 11:52:02 +08:00
|
|
|
|
return NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,mediaInfo,mediaAuthInvoker,sender);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 判断http客户端是否有权限访问文件的逻辑步骤
|
|
|
|
|
* 1、根据http请求头查找cookie,找到进入步骤3
|
|
|
|
|
* 2、根据http url参数查找cookie,如果还是未找到cookie则进入步骤5
|
|
|
|
|
* 3、cookie标记是否有权限访问文件,如果有权限,直接返回文件
|
|
|
|
|
* 4、cookie中记录的url参数是否跟本次url参数一致,如果一致直接返回客户端错误码
|
|
|
|
|
* 5、触发kBroadcastHttpAccess事件
|
|
|
|
|
*/
|
|
|
|
|
static void canAccessPath(TcpSession &sender, const Parser &parser, const MediaInfo &mediaInfo, bool is_dir,
|
|
|
|
|
const function<void(const string &errMsg, const HttpServerCookie::Ptr &cookie)> &callback) {
|
|
|
|
|
//获取用户唯一id
|
|
|
|
|
auto uid = parser.Params();
|
|
|
|
|
auto path = parser.Url();
|
|
|
|
|
|
|
|
|
|
//先根据http头中的cookie字段获取cookie
|
|
|
|
|
HttpServerCookie::Ptr cookie = HttpCookieManager::Instance().getCookie(kCookieName, parser.getValues());
|
|
|
|
|
//如果不是从http头中找到的cookie,我们让http客户端设置下cookie
|
|
|
|
|
bool cookie_from_header = true;
|
|
|
|
|
if (!cookie && !uid.empty()) {
|
|
|
|
|
//客户端请求中无cookie,再根据该用户的用户id获取cookie
|
|
|
|
|
cookie = HttpCookieManager::Instance().getCookieByUid(kCookieName, uid);
|
|
|
|
|
cookie_from_header = false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if (cookie) {
|
|
|
|
|
//找到了cookie,对cookie上锁先
|
|
|
|
|
auto lck = cookie->getLock();
|
2020-01-02 17:46:20 +08:00
|
|
|
|
auto attachment = (*cookie)[kCookieName].get<HttpCookieAttachment>();
|
|
|
|
|
if (path.find(attachment._path) == 0) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//上次cookie是限定本目录
|
2020-01-02 17:46:20 +08:00
|
|
|
|
if (attachment._err_msg.empty()) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//上次鉴权成功
|
2020-01-02 17:46:20 +08:00
|
|
|
|
if(attachment._is_hls){
|
2019-12-29 17:55:02 +08:00
|
|
|
|
//如果播放的是hls,那么刷新hls的cookie(获取ts文件也会刷新)
|
2019-11-30 14:29:44 +08:00
|
|
|
|
cookie->updateTime();
|
|
|
|
|
cookie_from_header = false;
|
|
|
|
|
}
|
2019-11-30 11:38:00 +08:00
|
|
|
|
callback("", cookie_from_header ? nullptr : cookie);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
//上次鉴权失败,但是如果url参数发生变更,那么也重新鉴权下
|
|
|
|
|
if (parser.Params().empty() || parser.Params() == cookie->getUid()) {
|
|
|
|
|
//url参数未变,或者本来就没有url参数,那么判断本次请求为重复请求,无访问权限
|
2020-01-02 17:46:20 +08:00
|
|
|
|
callback(attachment._err_msg, cookie_from_header ? nullptr : cookie);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
//如果url参数变了或者不是限定本目录,那么旧cookie失效,重新鉴权
|
|
|
|
|
HttpCookieManager::Instance().delCookie(cookie);
|
|
|
|
|
}
|
|
|
|
|
|
2019-12-29 11:52:02 +08:00
|
|
|
|
bool is_hls = mediaInfo._schema == HLS_SCHEMA;
|
2020-02-13 12:10:08 +08:00
|
|
|
|
string identifier = sender.getIdentifier();
|
2020-02-13 11:33:59 +08:00
|
|
|
|
string peer_ip = sender.get_peer_ip();
|
|
|
|
|
uint16_t peer_port = sender.get_peer_port();
|
|
|
|
|
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//该用户从来未获取过cookie,这个时候我们广播是否允许该用户访问该http目录
|
2020-02-13 12:10:08 +08:00
|
|
|
|
HttpSession::HttpAccessPathInvoker accessPathInvoker = [callback, uid, path, is_dir, is_hls, mediaInfo, identifier, peer_ip, peer_port]
|
2019-12-29 17:55:02 +08:00
|
|
|
|
(const string &errMsg, const string &cookie_path_in, int cookieLifeSecond) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
HttpServerCookie::Ptr cookie;
|
|
|
|
|
if (cookieLifeSecond) {
|
|
|
|
|
//本次鉴权设置了有效期,我们把鉴权结果缓存在cookie中
|
|
|
|
|
string cookie_path = cookie_path_in;
|
|
|
|
|
if (cookie_path.empty()) {
|
|
|
|
|
//如果未设置鉴权目录,那么我们采用当前目录
|
|
|
|
|
cookie_path = is_dir ? path : path.substr(0, path.rfind("/") + 1);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
cookie = HttpCookieManager::Instance().addCookie(kCookieName, uid, cookieLifeSecond);
|
|
|
|
|
//对cookie上锁
|
|
|
|
|
auto lck = cookie->getLock();
|
2020-01-02 17:46:20 +08:00
|
|
|
|
HttpCookieAttachment attachment;
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//记录用户能访问的路径
|
2020-01-02 17:46:20 +08:00
|
|
|
|
attachment._path = cookie_path;
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//记录能否访问
|
2020-01-02 17:46:20 +08:00
|
|
|
|
attachment._err_msg = errMsg;
|
2019-11-30 14:29:44 +08:00
|
|
|
|
//记录访问的是否为hls
|
2020-01-02 17:46:20 +08:00
|
|
|
|
attachment._is_hls = is_hls;
|
2019-12-28 16:48:11 +08:00
|
|
|
|
if(is_hls){
|
|
|
|
|
//hls相关信息
|
2020-02-13 12:10:08 +08:00
|
|
|
|
attachment._hls_data = std::make_shared<HlsCookieData>(mediaInfo, identifier, peer_ip, peer_port);
|
2019-12-29 15:38:29 +08:00
|
|
|
|
//hls未查找MediaSource
|
2020-01-02 17:46:20 +08:00
|
|
|
|
attachment._have_find_media_source = false;
|
2019-12-28 16:48:11 +08:00
|
|
|
|
}
|
2020-01-02 17:46:20 +08:00
|
|
|
|
(*cookie)[kCookieName].set<HttpCookieAttachment>(std::move(attachment));
|
2019-12-28 16:48:11 +08:00
|
|
|
|
callback(errMsg, cookie);
|
|
|
|
|
}else{
|
|
|
|
|
callback(errMsg, nullptr);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
2019-12-29 11:52:02 +08:00
|
|
|
|
if (is_hls && emitHlsPlayed(parser, mediaInfo, accessPathInvoker, sender)) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//是hls的播放鉴权,拦截之
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//事件未被拦截,则认为是http下载请求
|
2019-12-29 11:52:02 +08:00
|
|
|
|
bool flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastHttpAccess, parser, path, is_dir, accessPathInvoker, sender);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
if (!flag) {
|
|
|
|
|
//此事件无人监听,我们默认都有权限访问
|
|
|
|
|
callback("", nullptr);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 发送404 Not Found
|
|
|
|
|
*/
|
|
|
|
|
static void sendNotFound(const HttpFileManager::invoker &cb) {
|
|
|
|
|
GET_CONFIG(string,notFound,Http::kNotFound);
|
|
|
|
|
cb("404 Not Found","text/html",StrCaseMap(),std::make_shared<HttpStringBody>(notFound));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 拼接文件路径
|
|
|
|
|
*/
|
|
|
|
|
static string pathCat(const string &a, const string &b){
|
|
|
|
|
if(a.back() == '/'){
|
|
|
|
|
return a + b;
|
|
|
|
|
}
|
|
|
|
|
return a + '/' + b;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 访问文件
|
|
|
|
|
* @param sender 事件触发者
|
|
|
|
|
* @param parser http请求
|
|
|
|
|
* @param mediaInfo http url信息
|
|
|
|
|
* @param strFile 文件绝对路径
|
|
|
|
|
* @param cb 回调对象
|
|
|
|
|
*/
|
|
|
|
|
static void accessFile(TcpSession &sender, const Parser &parser, const MediaInfo &mediaInfo, const string &strFile, const HttpFileManager::invoker &cb) {
|
2019-12-29 11:52:02 +08:00
|
|
|
|
bool is_hls = end_of(strFile, kHlsSuffix);
|
2019-12-29 14:00:20 +08:00
|
|
|
|
bool file_exist = File::is_file(strFile.data());
|
|
|
|
|
if (!is_hls && !file_exist) {
|
2019-12-29 11:52:02 +08:00
|
|
|
|
//文件不存在且不是hls,那么直接返回404
|
2019-11-30 11:38:00 +08:00
|
|
|
|
sendNotFound(cb);
|
|
|
|
|
return;
|
|
|
|
|
}
|
2019-12-29 11:52:02 +08:00
|
|
|
|
|
|
|
|
|
if(is_hls){
|
|
|
|
|
//hls,那么移除掉后缀获取真实的stream_id并且修改协议为HLS
|
|
|
|
|
const_cast<string &>(mediaInfo._schema) = HLS_SCHEMA;
|
|
|
|
|
replace(const_cast<string &>(mediaInfo._streamid), kHlsSuffix, "");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
weak_ptr<TcpSession> weakSession = sender.shared_from_this();
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//判断是否有权限访问该文件
|
2019-12-29 14:00:20 +08:00
|
|
|
|
canAccessPath(sender, parser, mediaInfo, false, [cb, strFile, parser, is_hls, mediaInfo, weakSession , file_exist](const string &errMsg, const HttpServerCookie::Ptr &cookie) {
|
2019-12-29 15:38:29 +08:00
|
|
|
|
auto strongSession = weakSession.lock();
|
|
|
|
|
if(!strongSession){
|
|
|
|
|
//http客户端已经断开,不需要回复
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
if (!errMsg.empty()) {
|
|
|
|
|
//文件鉴权失败
|
|
|
|
|
StrCaseMap headerOut;
|
|
|
|
|
if (cookie) {
|
2020-01-02 17:46:20 +08:00
|
|
|
|
headerOut["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
}
|
2019-12-29 15:38:29 +08:00
|
|
|
|
cb("401 Unauthorized", "text/html", headerOut, std::make_shared<HttpStringBody>(errMsg));
|
|
|
|
|
return;
|
|
|
|
|
}
|
2019-11-30 11:38:00 +08:00
|
|
|
|
|
2019-12-29 17:55:02 +08:00
|
|
|
|
auto response_file = [file_exist](const HttpServerCookie::Ptr &cookie, const HttpFileManager::invoker &cb, const string &strFile, const Parser &parser) {
|
2019-12-29 15:38:29 +08:00
|
|
|
|
StrCaseMap httpHeader;
|
|
|
|
|
if (cookie) {
|
2020-01-02 17:46:20 +08:00
|
|
|
|
httpHeader["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
|
2019-12-29 15:38:29 +08:00
|
|
|
|
}
|
|
|
|
|
HttpSession::HttpResponseInvoker invoker = [&](const string &codeOut, const StrCaseMap &headerOut, const HttpBody::Ptr &body) {
|
2019-12-29 17:55:02 +08:00
|
|
|
|
if (cookie && file_exist) {
|
2019-12-29 15:38:29 +08:00
|
|
|
|
cookie->getLock();
|
2020-01-02 17:46:20 +08:00
|
|
|
|
auto is_hls = (*cookie)[kCookieName].get<HttpCookieAttachment>()._is_hls;
|
2019-12-29 15:38:29 +08:00
|
|
|
|
if (is_hls) {
|
2020-01-02 17:46:20 +08:00
|
|
|
|
(*cookie)[kCookieName].get<HttpCookieAttachment>()._hls_data->addByteUsage(body->remainSize());
|
2019-12-29 11:52:02 +08:00
|
|
|
|
}
|
2019-12-29 15:38:29 +08:00
|
|
|
|
}
|
2020-01-02 18:24:11 +08:00
|
|
|
|
cb(codeOut.data(), HttpFileManager::getContentType(strFile.data()), headerOut, body);
|
2019-12-29 11:52:02 +08:00
|
|
|
|
};
|
2019-12-29 15:38:29 +08:00
|
|
|
|
invoker.responseFile(parser.getValues(), httpHeader, strFile);
|
|
|
|
|
};
|
2019-12-29 11:52:02 +08:00
|
|
|
|
|
2019-12-29 15:38:29 +08:00
|
|
|
|
if (!is_hls) {
|
|
|
|
|
//不是hls,直接回复文件或404
|
|
|
|
|
response_file(cookie, cb, strFile, parser);
|
|
|
|
|
} else {
|
2020-01-02 17:46:20 +08:00
|
|
|
|
//是hls直播,判断是否存在
|
2019-12-29 15:38:29 +08:00
|
|
|
|
bool have_find_media_src = false;
|
|
|
|
|
if(cookie){
|
2020-01-02 17:46:20 +08:00
|
|
|
|
have_find_media_src = (*cookie)[kCookieName].get<HttpCookieAttachment>()._have_find_media_source;
|
2019-12-29 15:38:29 +08:00
|
|
|
|
if(!have_find_media_src){
|
2020-01-02 17:46:20 +08:00
|
|
|
|
(*cookie)[kCookieName].get<HttpCookieAttachment>()._have_find_media_source = true;
|
2019-12-28 16:48:11 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
2019-12-29 15:38:29 +08:00
|
|
|
|
if(have_find_media_src){
|
|
|
|
|
//之前该cookie已经通过MediaSource::findAsync查找过了,所以现在只以文件系统查找结果为准
|
|
|
|
|
response_file(cookie, cb, strFile, parser);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
//hls文件不存在,我们等待其生成并延后回复
|
|
|
|
|
MediaSource::findAsync(mediaInfo, strongSession, [response_file, cookie, cb, strFile, parser](const MediaSource::Ptr &src) {
|
|
|
|
|
//hls已经生成或者超时后仍未生成,那么不管怎么样都返回客户端
|
|
|
|
|
response_file(cookie, cb, strFile, parser);
|
|
|
|
|
});
|
|
|
|
|
}
|
2019-11-30 11:38:00 +08:00
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
|
2019-12-23 12:47:04 +08:00
|
|
|
|
static string getFilePath(const Parser &parser,const MediaInfo &mediaInfo, TcpSession &sender){
|
|
|
|
|
GET_CONFIG(bool, enableVhost, General::kEnableVhost);
|
|
|
|
|
GET_CONFIG(string, rootPath, Http::kRootPath);
|
|
|
|
|
auto ret = File::absolutePath(enableVhost ? mediaInfo._vhost + parser.Url() : parser.Url(), rootPath);
|
2019-12-29 11:52:02 +08:00
|
|
|
|
NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastHttpBeforeAccess, parser, ret, sender);
|
2019-12-23 12:47:04 +08:00
|
|
|
|
return std::move(ret);
|
|
|
|
|
}
|
|
|
|
|
|
2019-11-30 11:38:00 +08:00
|
|
|
|
/**
|
|
|
|
|
* 访问文件或文件夹
|
|
|
|
|
* @param sender 事件触发者
|
|
|
|
|
* @param parser http请求
|
|
|
|
|
* @param cb 回调对象
|
|
|
|
|
*/
|
|
|
|
|
void HttpFileManager::onAccessPath(TcpSession &sender, Parser &parser, const HttpFileManager::invoker &cb) {
|
|
|
|
|
auto fullUrl = string(HTTP_SCHEMA) + "://" + parser["Host"] + parser.FullUrl();
|
|
|
|
|
MediaInfo mediaInfo(fullUrl);
|
2019-12-23 12:47:04 +08:00
|
|
|
|
auto strFile = getFilePath(parser, mediaInfo, sender);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
//访问的是文件夹
|
2019-11-30 11:56:40 +08:00
|
|
|
|
if (File::is_dir(strFile.data())) {
|
2019-11-30 11:38:00 +08:00
|
|
|
|
auto indexFile = searchIndexFile(strFile);
|
|
|
|
|
if (!indexFile.empty()) {
|
|
|
|
|
//发现该文件夹下有index文件
|
|
|
|
|
strFile = pathCat(strFile, indexFile);
|
|
|
|
|
parser.setUrl(pathCat(parser.Url(), indexFile));
|
|
|
|
|
accessFile(sender, parser, mediaInfo, strFile, cb);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
string strMenu;
|
|
|
|
|
//生成文件夹菜单索引
|
|
|
|
|
if (!makeFolderMenu(parser.Url(), strFile, strMenu)) {
|
|
|
|
|
//文件夹不存在
|
|
|
|
|
sendNotFound(cb);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
//判断是否有权限访问该目录
|
|
|
|
|
canAccessPath(sender, parser, mediaInfo, true, [strMenu, cb](const string &errMsg, const HttpServerCookie::Ptr &cookie) {
|
|
|
|
|
if (!errMsg.empty()) {
|
|
|
|
|
const_cast<string &>(strMenu) = errMsg;
|
|
|
|
|
}
|
|
|
|
|
StrCaseMap headerOut;
|
|
|
|
|
if (cookie) {
|
2020-01-02 17:46:20 +08:00
|
|
|
|
headerOut["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
|
2019-11-30 11:38:00 +08:00
|
|
|
|
}
|
|
|
|
|
cb(errMsg.empty() ? "200 OK" : "401 Unauthorized", "text/html", headerOut, std::make_shared<HttpStringBody>(strMenu));
|
|
|
|
|
});
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//访问的是文件
|
|
|
|
|
accessFile(sender, parser, mediaInfo, strFile, cb);
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
////////////////////////////////////HttpResponseInvokerImp//////////////////////////////////////
|
|
|
|
|
|
|
|
|
|
void HttpResponseInvokerImp::operator()(const string &codeOut, const StrCaseMap &headerOut, const HttpBody::Ptr &body) const{
|
|
|
|
|
if(_lambad){
|
|
|
|
|
_lambad(codeOut,headerOut,body);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void HttpResponseInvokerImp::operator()(const string &codeOut, const StrCaseMap &headerOut, const string &body) const{
|
|
|
|
|
this->operator()(codeOut,headerOut,std::make_shared<HttpStringBody>(body));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
HttpResponseInvokerImp::HttpResponseInvokerImp(const HttpResponseInvokerImp::HttpResponseInvokerLambda0 &lambda){
|
|
|
|
|
_lambad = lambda;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
HttpResponseInvokerImp::HttpResponseInvokerImp(const HttpResponseInvokerImp::HttpResponseInvokerLambda1 &lambda){
|
|
|
|
|
if(!lambda){
|
|
|
|
|
_lambad = nullptr;
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
_lambad = [lambda](const string &codeOut, const StrCaseMap &headerOut, const HttpBody::Ptr &body){
|
|
|
|
|
string str;
|
|
|
|
|
if(body && body->remainSize()){
|
|
|
|
|
str = body->readData(body->remainSize())->toString();
|
|
|
|
|
}
|
|
|
|
|
lambda(codeOut,headerOut,str);
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void HttpResponseInvokerImp::responseFile(const StrCaseMap &requestHeader,
|
|
|
|
|
const StrCaseMap &responseHeader,
|
|
|
|
|
const string &filePath) const {
|
|
|
|
|
StrCaseMap &httpHeader = const_cast<StrCaseMap&>(responseHeader);
|
|
|
|
|
std::shared_ptr<FILE> fp(fopen(filePath.data(), "rb"), [](FILE *fp) {
|
|
|
|
|
if (fp) {
|
|
|
|
|
fclose(fp);
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
if (!fp) {
|
|
|
|
|
//打开文件失败
|
|
|
|
|
GET_CONFIG(string,notFound,Http::kNotFound);
|
|
|
|
|
GET_CONFIG(string,charSet,Http::kCharSet);
|
|
|
|
|
|
|
|
|
|
auto strContentType = StrPrinter << "text/html; charset=" << charSet << endl;
|
|
|
|
|
httpHeader["Content-Type"] = strContentType;
|
|
|
|
|
(*this)("404 Not Found", httpHeader, notFound);
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
auto &strRange = const_cast<StrCaseMap &>(requestHeader)["Range"];
|
|
|
|
|
int64_t iRangeStart = 0;
|
|
|
|
|
int64_t iRangeEnd = 0 ;
|
|
|
|
|
int64_t fileSize = HttpMultiFormBody::fileSize(fp.get());
|
|
|
|
|
|
|
|
|
|
const char *pcHttpResult = NULL;
|
|
|
|
|
if (strRange.size() == 0) {
|
|
|
|
|
//全部下载
|
|
|
|
|
pcHttpResult = "200 OK";
|
|
|
|
|
iRangeEnd = fileSize - 1;
|
|
|
|
|
} else {
|
|
|
|
|
//分节下载
|
|
|
|
|
pcHttpResult = "206 Partial Content";
|
|
|
|
|
iRangeStart = atoll(FindField(strRange.data(), "bytes=", "-").data());
|
|
|
|
|
iRangeEnd = atoll(FindField(strRange.data(), "-", "\r\n").data());
|
|
|
|
|
if (iRangeEnd == 0) {
|
|
|
|
|
iRangeEnd = fileSize - 1;
|
|
|
|
|
}
|
|
|
|
|
//分节下载返回Content-Range头
|
|
|
|
|
httpHeader.emplace("Content-Range", StrPrinter << "bytes " << iRangeStart << "-" << iRangeEnd << "/" << fileSize << endl);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//回复文件
|
|
|
|
|
HttpBody::Ptr fileBody = std::make_shared<HttpFileBody>(fp, iRangeStart, iRangeEnd - iRangeStart + 1);
|
|
|
|
|
(*this)(pcHttpResult, httpHeader, fileBody);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
HttpResponseInvokerImp::operator bool(){
|
|
|
|
|
return _lambad.operator bool();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}//namespace mediakit
|