Skip to content

Z
ZLMediaKit
Switch branch/tag
Find file
Normal viewHistoryPermalink
HttpFileManager.cpp 22.2 KB
Newer Older
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
1 
/*
整理授权协议
7019e471
 
xiongziliang committed
2 
 * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
3 
 *
全面整理代码，去除编译警告
b6cbc877
 
xia-chu committed
4 
 * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
5 
 *
整理授权协议
7019e471
 
xiongziliang committed
6 7 8 
 * 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.
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
9 10 11 12 13 14 15 16 17 
 */

#include <sys/stat.h>
#if !defined(_WIN32)
#include <dirent.h>
#endif //!defined(_WIN32)
#include <iomanip>
#include "HttpFileManager.h"
#include "Util/File.h"
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
18 
#include "HttpConst.h"
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
19 
#include "HttpSession.h"
初步添加HlsMediaSource
1afacdcf
 
xiongziliang committed
20 
#include "Record/HlsMediaSource.h"
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
21 22 23 

namespace mediakit {
1、hls cookie改成60秒有效期并且访问时刷新cookie
cadff93d
 
xiongziliang committed
24 25 26 27 
// hls的播放cookie缓存时间默认60秒，
// 每次访问一次该cookie，那么将重新刷新cookie有效期
// 假如播放器在60秒内都未访问该cookie，那么将重新触发hls播放鉴权
static int kHlsCookieSecond = 60;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
28 
static const string kCookieName = "ZL_COOKIE";
1、hls cookie改成60秒有效期并且访问时刷新cookie
cadff93d
 
xiongziliang committed
29 
static const string kHlsSuffix = "/hls.m3u8";
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
30
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
31 
class HttpCookieAttachment {
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 
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;
};
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
50
完善http服务器mime类别
d482dc1c
 
xiongziliang committed
51 
const string &HttpFileManager::getContentType(const char *name) {
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
52 
    return getHttpContentType(name);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
53 54 55 56 57 58 59 60 61 62 
}

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) {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
63 64 
        static set<const char *, StrCaseCompare> indexSet = {"index.html", "index.htm", "index"};
        if (indexSet.find(pDirent->d_name) != indexSet.end()) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
65 66 67 68 69 70 71 72 73 74 
            string ret = pDirent->d_name;
            closedir(pDir);
            return ret;
        }
    }
    closedir(pDir);
    return "";
}

static bool makeFolderMenu(const string &httpPath, const string &strFullPath, string &strRet) {
新增是否开启浏览http目录的配置选项:#367
41c75fb6
 
xiongziliang committed
75 
    GET_CONFIG(bool, dirMenu, Http::kDirMenu);
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
76 
    if (!dirMenu) {
新增是否开启浏览http目录的配置选项:#367
41c75fb6
 
xiongziliang committed
77 78 79 
        //不允许浏览文件夹
        return false;
    }
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
80 81 
    string strPathPrefix(strFullPath);
    string last_dir_name;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
82 
    if (strPathPrefix.back() == '/') {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
83 
        strPathPrefix.pop_back();
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
84 85 
    } else {
        last_dir_name = split(strPathPrefix, "/").back();
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 
    }

    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;
        }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
129 
        if (pDirent->d_name[0] == '.') {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
130 131 132 133 134 
            continue;
        }
        setFile.emplace(pDirent->d_name);
    }
    int i = 0;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
135 
    for (auto &strFile :setFile) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
136 137 138 139 
        string strAbsolutePath = strPathPrefix + "/" + strFile;
        bool isDir = File::is_dir(strAbsolutePath.data());
        ss << "<li><span>" << i++ << "</span>\t";
        ss << "<a href=\"";
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
140 
        if (!last_dir_name.empty()) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
141 
            ss << last_dir_name << "/" << strFile;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
142 
        } else {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
143 144 145 
            ss << strFile;
        }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
146 
        if (isDir) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 
            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;
}

//拦截hls的播放请求
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
179 
static bool emitHlsPlayed(const Parser &parser, const MediaInfo &mediaInfo, const HttpSession::HttpAccessPathInvoker &invoker,TcpSession &sender){
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
180 
    //访问的hls.m3u8结尾，我们转换成kBroadcastMediaPlayed事件
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
181 
    Broadcast::AuthInvoker auth_invoker = [invoker](const string &err) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
182 
        //cookie有效期为kHlsCookieSecond
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
183 
        invoker(err, "", kHlsCookieSecond);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
184 
    };
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
185 186 
    bool flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed, mediaInfo, auth_invoker, static_cast<SockInfo &>(sender));
    if (!flag) {
修复hls播放鉴权相关bug
268a7fec
 
xiongziliang committed
187 
        //未开启鉴权，那么允许播放
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
188 
        auth_invoker("");
修复hls播放鉴权相关bug
268a7fec
 
xiongziliang committed
189 190 
    }
    return flag;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
191 192 
}
统一事件参数类型
eda7a59f
 
xiongziliang committed
193 194 195 196 197 198 
class SockInfoImp : public SockInfo{
public:
    typedef std::shared_ptr<SockInfoImp> Ptr;
    SockInfoImp() = default;
    ~SockInfoImp() override = default;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
199 
    string get_local_ip() override {
统一事件参数类型
eda7a59f
 
xiongziliang committed
200 201 202 
        return _local_ip;
    }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
203 
    uint16_t get_local_port() override {
统一事件参数类型
eda7a59f
 
xiongziliang committed
204 205 206 
        return _local_port;
    }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
207 
    string get_peer_ip() override {
统一事件参数类型
eda7a59f
 
xiongziliang committed
208 209 210 
        return _peer_ip;
    }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
211 
    uint16_t get_peer_port() override {
统一事件参数类型
eda7a59f
 
xiongziliang committed
212 213 214 
        return _peer_port;
    }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
215 
    string getIdentifier() const override {
统一事件参数类型
eda7a59f
 
xiongziliang committed
216 217 218 219 220 221 222 223 224 
        return _identifier;
    }

    string _local_ip;
    string _peer_ip;
    string _identifier;
    uint16_t _local_port;
    uint16_t _peer_port;
};
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 

/**
 * 判断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
整理解析器
1d5c6cb1
 
xiongziliang committed
241 
    HttpServerCookie::Ptr cookie = HttpCookieManager::Instance().getCookie(kCookieName, parser.getHeader());
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
242 243 244 245 246 247 248 249 250 251 252 
    //如果不是从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();
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
253 254 
        auto attachment = (*cookie)[kCookieName].get<HttpCookieAttachment>();
        if (path.find(attachment._path) == 0) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
255 
            //上次cookie是限定本目录
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
256 
            if (attachment._err_msg.empty()) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
257 
                //上次鉴权成功
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
258 
                if (attachment._is_hls) {
解决hls中断恢复时播放器计数不准确的问题
0063571f
 
xiongziliang committed
259 
                    //如果播放的是hls，那么刷新hls的cookie(获取ts文件也会刷新)
1、hls cookie改成60秒有效期并且访问时刷新cookie
cadff93d
 
xiongziliang committed
260 261 262 
                    cookie->updateTime();
                    cookie_from_header = false;
                }
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
263 264 265 266 267 268 
                callback("", cookie_from_header ? nullptr : cookie);
                return;
            }
            //上次鉴权失败，但是如果url参数发生变更，那么也重新鉴权下
            if (parser.Params().empty() || parser.Params() == cookie->getUid()) {
                //url参数未变，或者本来就没有url参数，那么判断本次请求为重复请求，无访问权限
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
269 
                callback(attachment._err_msg, cookie_from_header ? nullptr : cookie);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
270 271 272 273 274 275 276 
                return;
            }
        }
        //如果url参数变了或者不是限定本目录，那么旧cookie失效，重新鉴权
        HttpCookieManager::Instance().delCookie(cookie);
    }
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
277 
    bool is_hls = mediaInfo._schema == HLS_SCHEMA;
统一事件参数类型
eda7a59f
 
xiongziliang committed
278 279 280 281 282 283 284 

    SockInfoImp::Ptr info = std::make_shared<SockInfoImp>();
    info->_identifier = sender.getIdentifier();
    info->_peer_ip = sender.get_peer_ip();
    info->_peer_port = sender.get_peer_port();
    info->_local_ip = sender.get_local_ip();
    info->_local_port = sender.get_local_port();
流量汇报事件添加客户端ip和端口参数
7e255379
 
xiongziliang committed
285
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
286 
    //该用户从来未获取过cookie，这个时候我们广播是否允许该用户访问该http目录
统一事件参数类型
eda7a59f
 
xiongziliang committed
287 
    HttpSession::HttpAccessPathInvoker accessPathInvoker = [callback, uid, path, is_dir, is_hls, mediaInfo, info]
解决hls中断恢复时播放器计数不准确的问题
0063571f
 
xiongziliang committed
288 
            (const string &errMsg, const string &cookie_path_in, int cookieLifeSecond) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
289 290 291 292 293 294 295 296 297 298 299 300 
        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();
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
301 
            HttpCookieAttachment attachment;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
302 
            //记录用户能访问的路径
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
303 
            attachment._path = cookie_path;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
304 
            //记录能否访问
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
305 
            attachment._err_msg = errMsg;
1、hls cookie改成60秒有效期并且访问时刷新cookie
cadff93d
 
xiongziliang committed
306 
            //记录访问的是否为hls
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
307 
            attachment._is_hls = is_hls;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
308 
            if (is_hls) {
初步实现hls播放器个数计数
94806b2c
 
xiongziliang committed
309 
                //hls相关信息
统一事件参数类型
eda7a59f
 
xiongziliang committed
310 
                attachment._hls_data = std::make_shared<HlsCookieData>(mediaInfo, info);
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
311 
                //hls未查找MediaSource
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
312 
                attachment._have_find_media_source = false;
初步实现hls播放器个数计数
94806b2c
 
xiongziliang committed
313 
            }
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
314 
            (*cookie)[kCookieName].set<HttpCookieAttachment>(std::move(attachment));
初步实现hls播放器个数计数
94806b2c
 
xiongziliang committed
315 
            callback(errMsg, cookie);
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
316 
        } else {
初步实现hls播放器个数计数
94806b2c
 
xiongziliang committed
317 
            callback(errMsg, nullptr);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
318 319 320 
        }
    };
修复hls播放鉴权相关bug
268a7fec
 
xiongziliang committed
321 
    if (is_hls) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
322 
        //是hls的播放鉴权,拦截之
修复hls播放鉴权相关bug
268a7fec
 
xiongziliang committed
323 
        emitHlsPlayed(parser, mediaInfo, accessPathInvoker, sender);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
324 325 326 327 
        return;
    }

    //事件未被拦截，则认为是http下载请求
事件广播触发者改成SockInfo对象
b6946a59
 
xiongziliang committed
328 
    bool flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastHttpAccess, parser, path, is_dir, accessPathInvoker, static_cast<SockInfo &>(sender));
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
329 330 331 332 333 334 335 336 337 338 
    if (!flag) {
        //此事件无人监听，我们默认都有权限访问
        callback("", nullptr);
    }
}

/**
 * 发送404 Not Found
 */
static void sendNotFound(const HttpFileManager::invoker &cb) {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
339 
    GET_CONFIG(string, notFound, Http::kNotFound);
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
340 
    cb(404, "text/html", StrCaseMap(), std::make_shared<HttpStringBody>(notFound));
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
341 342 343 344 345 346 
}

/**
 * 拼接文件路径
 */
static string pathCat(const string &a, const string &b){
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
347 
    if (a.back() == '/') {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
348 349 350 351 352 353 354 355 356 357 358 359 360 361 
        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) {
复用代码
4f42f508
 
xiongziliang committed
362 
    bool is_hls = end_with(strFile, kHlsSuffix);
优化hls直播性能
e951efc6
 
xiongziliang committed
363 364 
    bool file_exist = File::is_file(strFile.data());
    if (!is_hls && !file_exist) {
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
365 
        //文件不存在且不是hls,那么直接返回404
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
366 367 368 
        sendNotFound(cb);
        return;
    }
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
369
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
370 
    if (is_hls) {
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
371 372 373 374 375 376 
        //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();
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
377 
    //判断是否有权限访问该文件
优化hls直播性能
e951efc6
 
xiongziliang committed
378 
    canAccessPath(sender, parser, mediaInfo, false, [cb, strFile, parser, is_hls, mediaInfo, weakSession , file_exist](const string &errMsg, const HttpServerCookie::Ptr &cookie) {
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
379 
        auto strongSession = weakSession.lock();
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
380 
        if (!strongSession) {
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
381 382 383 384 385 386 387 
            //http客户端已经断开，不需要回复
            return;
        }
        if (!errMsg.empty()) {
            //文件鉴权失败
            StrCaseMap headerOut;
            if (cookie) {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
388 
                auto lck = cookie->getLock();
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
389 
                headerOut["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
390 
            }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
391 
            cb(401, "text/html", headerOut, std::make_shared<HttpStringBody>(errMsg));
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
392 393 
            return;
        }
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
394
解决hls中断恢复时播放器计数不准确的问题
0063571f
 
xiongziliang committed
395 
        auto response_file = [file_exist](const HttpServerCookie::Ptr &cookie, const HttpFileManager::invoker &cb, const string &strFile, const Parser &parser) {
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
396 397 
            StrCaseMap httpHeader;
            if (cookie) {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
398 
                auto lck = cookie->getLock();
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
399 
                httpHeader["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
400 
            }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
401 
            HttpSession::HttpResponseInvoker invoker = [&](int code, const StrCaseMap &headerOut, const HttpBody::Ptr &body) {
解决hls中断恢复时播放器计数不准确的问题
0063571f
 
xiongziliang committed
402 
                if (cookie && file_exist) {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
403 
                    auto lck = cookie->getLock();
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
404 
                    auto is_hls = (*cookie)[kCookieName].get<HttpCookieAttachment>()._is_hls;
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
405 
                    if (is_hls) {
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
406 
                        (*cookie)[kCookieName].get<HttpCookieAttachment>()._hls_data->addByteUsage(body->remainSize());
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
407 
                    }
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
408 
                }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
409 
                cb(code, HttpFileManager::getContentType(strFile.data()), headerOut, body);
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
410 
            };
整理解析器
1d5c6cb1
 
xiongziliang committed
411 
            invoker.responseFile(parser.getHeader(), httpHeader, strFile);
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
412 
        };
实现hls按需拉流
4b4c4e0c
 
xiongziliang committed
413
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
414 415 416 
        if (!is_hls) {
            //不是hls,直接回复文件或404
            response_file(cookie, cb, strFile, parser);
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
417 418 419 420 421 422 423 424 425 426 
            return;
        }

        //是hls直播，判断HLS直播流是否已经注册
        bool have_find_media_src = false;
        if (cookie) {
            auto lck = cookie->getLock();
            have_find_media_src = (*cookie)[kCookieName].get<HttpCookieAttachment>()._have_find_media_source;
            if (!have_find_media_src) {
                (*cookie)[kCookieName].get<HttpCookieAttachment>()._have_find_media_source = true;
初步实现hls播放器个数计数
94806b2c
 
xiongziliang committed
427 
            }
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 
        }
        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) {
            if (cookie) {
                auto lck = cookie->getLock();
                //尝试添加HlsMediaSource的观看人数(HLS是按需生成的，这样可以触发HLS文件的生成)
                (*cookie)[kCookieName].get<HttpCookieAttachment>()._hls_data->addByteUsage(0);
            }
            if (src && File::is_file(strFile.data())) {
                //流和m3u8文件都存在，那么直接返回文件
                response_file(cookie, cb, strFile, parser);
                return;
            }
            auto hls = dynamic_pointer_cast<HlsMediaSource>(src);
            if (!hls) {
                //流不存在，那么直接返回文件(相当于纯粹的HLS文件服务器,但是会挂起播放器15秒左右(用于等待HLS流的注册))
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
449 450 451 
                response_file(cookie, cb, strFile, parser);
                return;
            }
完善按需转协议(包括hls)
be1e872f
 
xiongziliang committed
452
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
453 454 455 
            //流存在，但是m3u8文件不存在，那么等待生成m3u8文件(HLS源注册后，并不会立即生成HLS文件，有人观看才会按需生成HLS文件)
            hls->waitForFile([response_file, cookie, cb, strFile, parser]() {
                response_file(cookie, cb, strFile, parser);
实现hls的流量汇报事件
e91c26c0
 
xiongziliang committed
456 
            });
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
457 
        });
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
458 459 460 
    });
}
添加kBroadcastHttpBeforeAccess事件，拦截http文件下载事件
2da0c884
 
xiongziliang committed
461 462 463 464 
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);
事件广播触发者改成SockInfo对象
b6946a59
 
xiongziliang committed
465 
    NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastHttpBeforeAccess, parser, ret, static_cast<SockInfo &>(sender));
返回值去除std::move
416d21df
 
xiongziliang committed
466 
    return ret;
添加kBroadcastHttpBeforeAccess事件，拦截http文件下载事件
2da0c884
 
xiongziliang committed
467 468 
}
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
469 470 471 472 473 474 475 476 477 
/**
 * 访问文件或文件夹
 * @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);
添加kBroadcastHttpBeforeAccess事件，拦截http文件下载事件
2da0c884
 
xiongziliang committed
478 
    auto strFile = getFilePath(parser, mediaInfo, sender);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
479 
    //访问的是文件夹
精简代码
d46b67a5
 
xiongziliang committed
480 
    if (File::is_dir(strFile.data())) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 
        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) {
简化hls cookie相关逻辑
52d831e9
 
xiongziliang committed
503 
                headerOut["Set-Cookie"] = cookie->getCookie((*cookie)[kCookieName].get<HttpCookieAttachment>()._path);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
504 
            }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
505 
            cb(errMsg.empty() ? 200 : 401, "text/html", headerOut, std::make_shared<HttpStringBody>(strMenu));
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
506 507 508 509 510 511 512 513 514 515 516 
        });
        return;
    }

    //访问的是文件
    accessFile(sender, parser, mediaInfo, strFile, cb);
};


////////////////////////////////////HttpResponseInvokerImp//////////////////////////////////////
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
517 
void HttpResponseInvokerImp::operator()(int code, const StrCaseMap &headerOut, const HttpBody::Ptr &body) const{
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
518 
    if (_lambad) {
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
519 
        _lambad(code, headerOut, body);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
520 521 522 
    }
}
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
523 524 
void HttpResponseInvokerImp::operator()(int code, const StrCaseMap &headerOut, const string &body) const{
    this->operator()(code, headerOut, std::make_shared<HttpStringBody>(body));
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
525 526 527 528 529 530 531 
}

HttpResponseInvokerImp::HttpResponseInvokerImp(const HttpResponseInvokerImp::HttpResponseInvokerLambda0 &lambda){
    _lambad = lambda;
}

HttpResponseInvokerImp::HttpResponseInvokerImp(const HttpResponseInvokerImp::HttpResponseInvokerLambda1 &lambda){
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
532 
    if (!lambda) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
533 534 535 
        _lambad = nullptr;
        return;
    }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
536 
    _lambad = [lambda](int code, const StrCaseMap &headerOut, const HttpBody::Ptr &body) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
537 
        string str;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
538 
        if (body && body->remainSize()) {
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
539 540 
            str = body->readData(body->remainSize())->toString();
        }
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
541 
        lambda(code, headerOut, str);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
542 543 544 545 546 547 
    };
}

void HttpResponseInvokerImp::responseFile(const StrCaseMap &requestHeader,
                                          const StrCaseMap &responseHeader,
                                          const string &filePath) const {
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
548 
    StrCaseMap &httpHeader = const_cast<StrCaseMap &>(responseHeader);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
549 550 551 552 553 554 555 556 
    std::shared_ptr<FILE> fp(fopen(filePath.data(), "rb"), [](FILE *fp) {
        if (fp) {
            fclose(fp);
        }
    });

    if (!fp) {
        //打开文件失败
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
557 558 
        GET_CONFIG(string, notFound, Http::kNotFound);
        GET_CONFIG(string, charSet, Http::kCharSet);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
559 560 561 

        auto strContentType = StrPrinter << "text/html; charset=" << charSet << endl;
        httpHeader["Content-Type"] = strContentType;
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
562 
        (*this)(404, httpHeader, notFound);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
563 564 565 566 
        return;
    }

    auto &strRange = const_cast<StrCaseMap &>(requestHeader)["Range"];
全面整理代码，去除编译警告
b6cbc877
 
xia-chu committed
567 568 569 
    size_t iRangeStart = 0;
    size_t iRangeEnd = 0;
    size_t fileSize = HttpMultiFormBody::fileSize(fp.get());
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
570
修复编译警告
d3760702
 
xiongziliang committed
571 
    int code;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
572 573 
    if (strRange.size() == 0) {
        //全部下载
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
574 
        code = 200;
优化HLS直播相关代码
07ef4172
 
xiongziliang committed
575 
        iRangeEnd = fileSize - 1;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
576 577 
    } else {
        //分节下载
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
578 
        code = 206;
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
579 580 581 582 583 584 585 586 587 588 589 
        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);
http回复根据状态码自动生成status message: #602
13221ad7
 
xiongziliang committed
590 
    (*this)(code, httpHeader, fileBody);
整理http文件服务器相关代码
a44ab25d
 
xiongziliang committed
591 592 593 594 595 596 597 598 
}

HttpResponseInvokerImp::operator bool(){
    return _lambad.operator bool();
}


}//namespace mediakit