ZLMediaKit/server/WebHook.cpp

681 lines
27 KiB
C++
Raw Normal View History

2022-06-18 14:09:38 +08:00
/*
2020-04-04 20:30:09 +08:00
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
2019-06-11 09:25:54 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
2019-06-11 09:25:54 +08:00
*
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-06-11 09:25:54 +08:00
*/
2019-05-20 11:22:59 +08:00
#include <sstream>
#include "Util/logger.h"
#include "Util/onceToken.h"
#include "Util/NoticeCenter.h"
#include "Common/config.h"
#include "Common/MediaSource.h"
#include "Http/HttpRequester.h"
#include "Network/TcpSession.h"
2019-05-20 16:26:04 +08:00
#include "Rtsp/RtspSession.h"
2019-06-12 17:53:48 +08:00
#include "Http/HttpSession.h"
#include "WebHook.h"
2021-04-08 17:34:13 +08:00
#include "WebApi.h"
2019-05-20 11:22:59 +08:00
using namespace std;
using namespace Json;
2019-05-20 11:22:59 +08:00
using namespace toolkit;
using namespace mediakit;
namespace Hook {
#define HOOK_FIELD "hook."
const string kEnable = HOOK_FIELD"enable";
const string kTimeoutSec = HOOK_FIELD"timeoutSec";
const string kOnPublish = HOOK_FIELD"on_publish";
const string kOnPlay = HOOK_FIELD"on_play";
const string kOnFlowReport = HOOK_FIELD"on_flow_report";
const string kOnRtspRealm = HOOK_FIELD"on_rtsp_realm";
const string kOnRtspAuth = HOOK_FIELD"on_rtsp_auth";
const string kOnStreamChanged = HOOK_FIELD"on_stream_changed";
const string kOnStreamNotFound = HOOK_FIELD"on_stream_not_found";
const string kOnRecordMp4 = HOOK_FIELD"on_record_mp4";
2020-09-13 14:08:05 +08:00
const string kOnRecordTs = HOOK_FIELD"on_record_ts";
const string kOnShellLogin = HOOK_FIELD"on_shell_login";
const string kOnStreamNoneReader = HOOK_FIELD"on_stream_none_reader";
const string kOnHttpAccess = HOOK_FIELD"on_http_access";
2019-11-29 10:12:20 +08:00
const string kOnServerStarted = HOOK_FIELD"on_server_started";
2021-08-20 14:52:48 +08:00
const string kOnServerKeepalive = HOOK_FIELD"on_server_keepalive";
2022-08-27 10:53:47 +08:00
const string kOnSendRtpStopped = HOOK_FIELD"on_send_rtp_stopped";
const string kAdminParams = HOOK_FIELD"admin_params";
2021-08-20 14:52:48 +08:00
const string kAliveInterval = HOOK_FIELD"alive_interval";
const string kRetry = HOOK_FIELD"retry";
2022-06-18 14:09:38 +08:00
const string kRetryDelay = HOOK_FIELD"retry_delay";
2019-05-20 11:22:59 +08:00
onceToken token([](){
2019-08-16 15:34:34 +08:00
mINI::Instance()[kEnable] = false;
2019-05-20 11:22:59 +08:00
mINI::Instance()[kTimeoutSec] = 10;
//默认hook地址设置为空采用默认行为(例如不鉴权)
mINI::Instance()[kOnPublish] = "";
mINI::Instance()[kOnPlay] = "";
mINI::Instance()[kOnFlowReport] = "";
mINI::Instance()[kOnRtspRealm] = "";
mINI::Instance()[kOnRtspAuth] = "";
mINI::Instance()[kOnStreamChanged] = "";
mINI::Instance()[kOnStreamNotFound] = "";
mINI::Instance()[kOnRecordMp4] = "";
mINI::Instance()[kOnRecordTs] = "";
mINI::Instance()[kOnShellLogin] = "";
mINI::Instance()[kOnStreamNoneReader] = "";
mINI::Instance()[kOnHttpAccess] = "";
mINI::Instance()[kOnServerStarted] = "";
2021-08-20 14:52:48 +08:00
mINI::Instance()[kOnServerKeepalive] = "";
2022-08-27 10:53:47 +08:00
mINI::Instance()[kOnSendRtpStopped] = "";
2019-05-20 17:12:00 +08:00
mINI::Instance()[kAdminParams] = "secret=035c73f7-bb6b-4889-a715-d9eb2d1925cc";
2021-08-20 14:52:48 +08:00
mINI::Instance()[kAliveInterval] = 30.0;
2022-06-18 14:09:38 +08:00
mINI::Instance()[kRetry] = 1;
mINI::Instance()[kRetryDelay] = 3.0;
2019-05-20 11:22:59 +08:00
},nullptr);
}//namespace Hook
2022-01-12 16:45:47 +08:00
namespace Cluster {
#define CLUSTER_FIELD "cluster."
const string kOriginUrl = CLUSTER_FIELD "origin_url";
2022-01-12 17:58:07 +08:00
const string kTimeoutSec = CLUSTER_FIELD "timeout_sec";
const string kRetryCount = CLUSTER_FIELD "retry_count";
2022-01-12 17:58:07 +08:00
2022-01-12 16:45:47 +08:00
static onceToken token([]() {
mINI::Instance()[kOriginUrl] = "";
2022-01-12 17:58:07 +08:00
mINI::Instance()[kTimeoutSec] = 15;
2022-05-30 12:44:11 +08:00
mINI::Instance()[kRetryCount] = 3;
2022-01-12 16:45:47 +08:00
});
2019-05-20 11:22:59 +08:00
2022-01-12 16:45:47 +08:00
}//namespace Cluster
static void parse_http_response(const SockException &ex, const Parser &res,
2019-05-20 11:22:59 +08:00
const function<void(const Value &,const string &)> &fun){
2021-09-30 16:10:09 +08:00
if (ex) {
2019-05-20 11:22:59 +08:00
auto errStr = StrPrinter << "[network err]:" << ex.what() << endl;
2021-09-30 16:10:09 +08:00
fun(Json::nullValue, errStr);
2019-05-20 11:22:59 +08:00
return;
}
2021-09-30 16:10:09 +08:00
if (res.Url() != "200") {
auto errStr = StrPrinter << "[bad http status code]:" << res.Url() << endl;
fun(Json::nullValue, errStr);
2019-05-20 11:22:59 +08:00
return;
}
2021-10-14 16:35:06 +08:00
Value result;
2019-05-20 11:22:59 +08:00
try {
2021-09-30 16:10:09 +08:00
stringstream ss(res.Content());
2019-05-20 11:22:59 +08:00
ss >> result;
2021-09-30 16:10:09 +08:00
} catch (std::exception &ex) {
2019-05-20 11:22:59 +08:00
auto errStr = StrPrinter << "[parse json failed]:" << ex.what() << endl;
2021-09-30 16:10:09 +08:00
fun(Json::nullValue, errStr);
2021-10-14 16:35:06 +08:00
return;
}
if (result["code"].asInt() != 0) {
auto errStr = StrPrinter << "[json code]:" << "code=" << result["code"] << ",msg=" << result["msg"] << endl;
fun(Json::nullValue, errStr);
return;
}
try {
fun(result, "");
} catch (std::exception &ex) {
auto errStr = StrPrinter << "[do hook invoker failed]:" << ex.what() << endl;
//如果还是抛异常,那么再上抛异常
fun(Json::nullValue, errStr);
2019-05-20 11:22:59 +08:00
}
}
2019-05-20 16:26:04 +08:00
string to_string(const Value &value){
return value.toStyledString();
}
string to_string(const HttpArgs &value){
return value.make();
}
const char *getContentType(const Value &value){
return "application/json";
}
const char *getContentType(const HttpArgs &value){
return "application/x-www-form-urlencoded";
}
string getVhost(const Value &value) {
const char *key = VHOST_KEY;
auto val = value.find(key, key + sizeof(VHOST_KEY) - 1);
return val ? val->asString() : "";
}
string getVhost(const HttpArgs &value) {
auto val = value.find(VHOST_KEY);
return val != value.end() ? val->second : "";
}
2022-06-18 14:09:38 +08:00
void do_http_hook(const string &url, const ArgsType &body, const function<void(const Value &, const string &)> &func, uint32_t retry) {
2020-12-27 22:14:59 +08:00
GET_CONFIG(string, mediaServerId, General::kMediaServerId);
GET_CONFIG(float, hook_timeoutSec, Hook::kTimeoutSec);
2022-06-18 14:09:38 +08:00
GET_CONFIG(float, retry_delay, Hook::kRetryDelay);
2020-10-01 14:37:52 +08:00
2020-12-27 22:14:59 +08:00
const_cast<ArgsType &>(body)["mediaServerId"] = mediaServerId;
2019-05-20 11:22:59 +08:00
HttpRequester::Ptr requester(new HttpRequester);
requester->setMethod("POST");
2019-05-20 16:26:04 +08:00
auto bodyStr = to_string(body);
requester->setBody(bodyStr);
2020-12-27 22:14:59 +08:00
requester->addHeader("Content-Type", getContentType(body));
auto vhost = getVhost(body);
if (!vhost.empty()) {
requester->addHeader("X-VHOST", vhost);
}
2022-06-18 14:09:38 +08:00
Ticker ticker;
requester->startRequester(url, [url, func, bodyStr, body, requester, ticker, retry](const SockException &ex, const Parser &res) mutable {
onceToken token(nullptr, [&]() mutable { requester.reset(); });
parse_http_response(ex, res, [&](const Value &obj, const string &err) {
if (!err.empty()) {
// hook失败
WarnL << "hook " << url << " " << ticker.elapsedTime() << "ms,failed" << err << ":" << bodyStr;
if (retry-- > 0) {
requester->getPoller()->doDelayTask(MAX(retry_delay, 0.0) * 1000, [url, body, func, retry] {
do_http_hook(url, body, func, retry);
return 0;
});
//重试不需要触发回调
return;
}
} else if (ticker.elapsedTime() > 500) {
//hook成功但是hook响应超过500ms打印警告日志
DebugL << "hook " << url << " " << ticker.elapsedTime() << "ms,success:" << bodyStr;
}
2020-12-27 22:14:59 +08:00
if (func) {
func(obj, err);
2019-05-20 11:22:59 +08:00
}
});
2022-06-18 14:09:38 +08:00
}, hook_timeoutSec);
}
void do_http_hook(const string &url, const ArgsType &body, const function<void(const Value &, const string &)> &func) {
GET_CONFIG(uint32_t, hook_retry, Hook::kRetry);
do_http_hook(url, body, func, hook_retry);
2019-05-20 11:22:59 +08:00
}
2019-05-20 16:26:04 +08:00
static ArgsType make_json(const MediaInfo &args){
ArgsType body;
2019-05-20 11:22:59 +08:00
body["schema"] = args._schema;
body[VHOST_KEY] = args._vhost;
2019-05-20 11:22:59 +08:00
body["app"] = args._app;
body["stream"] = args._streamid;
body["params"] = args._param_strs;
2020-09-21 14:32:56 +08:00
return body;
2019-05-20 11:22:59 +08:00
}
2019-11-29 10:12:20 +08:00
static void reportServerStarted(){
GET_CONFIG(bool,hook_enable,Hook::kEnable);
GET_CONFIG(string,hook_server_started,Hook::kOnServerStarted);
if(!hook_enable || hook_server_started.empty()){
return;
}
ArgsType body;
for (auto &pr : mINI::Instance()) {
body[pr.first] = (string &) pr.second;
}
//执行hook
do_http_hook(hook_server_started,body, nullptr);
}
2019-05-20 11:22:59 +08:00
2021-08-20 14:52:48 +08:00
// 服务器定时保活定时器
static Timer::Ptr g_keepalive_timer;
2021-08-20 14:52:48 +08:00
static void reportServerKeepalive() {
GET_CONFIG(bool, hook_enable, Hook::kEnable);
2021-08-20 14:52:48 +08:00
GET_CONFIG(string, hook_server_keepalive, Hook::kOnServerKeepalive);
if (!hook_enable || hook_server_keepalive.empty()) {
2021-08-20 14:52:48 +08:00
return;
}
GET_CONFIG(float, alive_interval, Hook::kAliveInterval);
g_keepalive_timer = std::make_shared<Timer>(alive_interval, []() {
2021-12-27 17:40:15 +08:00
getStatisticJson([](const Value &data) mutable {
ArgsType body;
body["data"] = data;
//执行hook
do_http_hook(hook_server_keepalive, body, nullptr);
});
2021-08-20 14:52:48 +08:00
return true;
}, nullptr);
}
2022-01-12 17:43:07 +08:00
static const string kEdgeServerParam = "edge=1";
static string getPullUrl(const string &origin_fmt, const MediaInfo &info) {
char url[1024] = { 0 };
2022-01-17 10:35:29 +08:00
if ((ssize_t)origin_fmt.size() > snprintf(url, sizeof(url), origin_fmt.data(), info._app.data(), info._streamid.data())) {
2022-01-12 17:43:07 +08:00
WarnL << "get origin url failed, origin_fmt:" << origin_fmt;
return "";
}
2022-01-12 17:51:00 +08:00
//告知源站这是来自边沿站的拉流请求,如果未找到流请立即返回拉流失败
2022-01-12 17:43:07 +08:00
return string(url) + '?' + kEdgeServerParam + '&' + VHOST_KEY + '=' + info._vhost + '&' + info._param_strs;
}
2022-01-17 10:35:29 +08:00
static void pullStreamFromOrigin(const vector<string>& urls, size_t index, size_t failed_cnt, const MediaInfo &args,
2022-01-12 17:43:07 +08:00
const function<void()> &closePlayer) {
2022-01-12 17:58:07 +08:00
GET_CONFIG(float, cluster_timeout_sec, Cluster::kTimeoutSec);
GET_CONFIG(int, retry_count, Cluster::kRetryCount);
2022-01-12 17:43:07 +08:00
auto url = getPullUrl(urls[index % urls.size()], args);
2022-01-12 17:58:07 +08:00
auto timeout_sec = cluster_timeout_sec / urls.size();
2022-01-12 17:43:07 +08:00
InfoL << "pull stream from origin, failed_cnt: " << failed_cnt << ", timeout_sec: " << timeout_sec << ", url: " << url;
ProtocolOption option;
option.enable_hls = option.enable_hls || (args._schema == HLS_SCHEMA);
option.enable_mp4 = false;
addStreamProxy(args._vhost, args._app, args._streamid, url, retry_count, option, Rtsp::RTP_TCP, timeout_sec,
[=](const SockException &ex, const string &key) mutable {
2022-01-12 17:43:07 +08:00
if (!ex) {
return;
}
//拉流失败
if (++failed_cnt == urls.size()) {
//已经重试所有源站了
WarnL << "pull stream from origin final failed: " << url;
closePlayer();
return;
}
pullStreamFromOrigin(urls, index + 1, failed_cnt, args, closePlayer);
});
}
static void *web_hook_tag = nullptr;
2019-05-20 11:22:59 +08:00
void installWebHook(){
2019-05-28 17:14:36 +08:00
GET_CONFIG(bool,hook_enable,Hook::kEnable);
GET_CONFIG(string,hook_adminparams,Hook::kAdminParams);
2019-05-20 11:22:59 +08:00
NoticeCenter::Instance().addListener(&web_hook_tag, Broadcast::kBroadcastMediaPublish, [](BroadcastMediaPublishArgs) {
GET_CONFIG(string,hook_publish,Hook::kOnPublish);
if (!hook_enable || args._param_strs == hook_adminparams || hook_publish.empty() || sender.get_peer_ip() == "127.0.0.1") {
invoker("", ProtocolOption());
2019-05-20 11:22:59 +08:00
return;
}
//异步执行该hook api防止阻塞NoticeCenter
2019-05-20 16:26:04 +08:00
auto body = make_json(args);
2019-05-20 11:22:59 +08:00
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
2022-03-02 18:03:44 +08:00
body["originType"] = (int) type;
body["originTypeStr"] = getOriginTypeString(type);
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_publish, body, [invoker](const Value &obj, const string &err) mutable {
ProtocolOption option;
if (err.empty()) {
//推流鉴权成功
2022-03-12 14:34:48 +08:00
if (obj.isMember("enable_hls")) {
option.enable_hls = obj["enable_hls"].asBool();
}
2022-03-12 14:34:48 +08:00
if (obj.isMember("enable_mp4")) {
option.enable_mp4 = obj["enable_mp4"].asBool();
}
if (obj.isMember("enable_audio")) {
option.enable_audio = obj["enable_audio"].asBool();
}
if (obj.isMember("add_mute_audio")) {
option.add_mute_audio = obj["add_mute_audio"].asBool();
}
if (obj.isMember("mp4_save_path")) {
option.mp4_save_path = obj["mp4_save_path"].asString();
}
if (obj.isMember("mp4_max_second")) {
option.mp4_max_second = obj["mp4_max_second"].asUInt();
}
if (obj.isMember("hls_save_path")) {
option.hls_save_path = obj["hls_save_path"].asString();
}
if (obj.isMember("enable_rtsp")) {
option.enable_rtsp = obj["enable_rtsp"].asBool();
}
if (obj.isMember("enable_rtmp")) {
option.enable_rtmp = obj["enable_rtmp"].asBool();
}
if (obj.isMember("enable_ts")) {
option.enable_ts = obj["enable_ts"].asBool();
}
if (obj.isMember("enable_fmp4")) {
option.enable_fmp4 = obj["enable_fmp4"].asBool();
}
if (obj.isMember("continue_push_ms")) {
option.continue_push_ms = obj["continue_push_ms"].asUInt();
}
if (obj.isMember("mp4_as_player")) {
option.mp4_as_player = obj["mp4_as_player"].asBool();
}
if (obj.isMember("modify_stamp")) {
option.modify_stamp = obj["modify_stamp"].asBool();
}
invoker(err, option);
2020-09-12 19:09:56 +08:00
} else {
//推流鉴权失败
invoker(err, option);
}
2019-05-20 11:22:59 +08:00
});
});
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastMediaPlayed,[](BroadcastMediaPlayedArgs){
GET_CONFIG(string,hook_play,Hook::kOnPlay);
2019-06-06 15:33:11 +08:00
if(!hook_enable || args._param_strs == hook_adminparams || hook_play.empty() || sender.get_peer_ip() == "127.0.0.1"){
2019-05-20 11:22:59 +08:00
invoker("");
return;
}
2019-05-20 16:26:04 +08:00
auto body = make_json(args);
2019-05-20 11:22:59 +08:00
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_play,body,[invoker](const Value &obj,const string &err){
invoker(err);
2019-05-20 11:22:59 +08:00
});
});
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastFlowReport,[](BroadcastFlowReportArgs){
GET_CONFIG(string,hook_flowreport,Hook::kOnFlowReport);
2020-04-23 22:04:59 +08:00
if(!hook_enable || args._param_strs == hook_adminparams || hook_flowreport.empty() || sender.get_peer_ip() == "127.0.0.1"){
2019-05-20 11:22:59 +08:00
return;
}
2019-05-20 16:26:04 +08:00
auto body = make_json(args);
2019-05-20 11:22:59 +08:00
body["totalBytes"] = (Json::UInt64)totalBytes;
body["duration"] = (Json::UInt64)totalDuration;
2019-05-20 17:46:06 +08:00
body["player"] = isPlayer;
2020-04-23 22:04:59 +08:00
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_flowreport,body, nullptr);
2019-05-20 11:22:59 +08:00
});
2019-05-20 16:26:04 +08:00
static const string unAuthedRealm = "unAuthedRealm";
//监听kBroadcastOnGetRtspRealm事件决定rtsp链接是否需要鉴权(传统的rtsp鉴权方案)才能访问
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastOnGetRtspRealm,[](BroadcastOnGetRtspRealmArgs){
GET_CONFIG(string,hook_rtsp_realm,Hook::kOnRtspRealm);
2019-06-06 15:33:11 +08:00
if(!hook_enable || args._param_strs == hook_adminparams || hook_rtsp_realm.empty() || sender.get_peer_ip() == "127.0.0.1"){
2019-05-20 16:26:04 +08:00
//无需认证
invoker("");
return;
}
auto body = make_json(args);
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_rtsp_realm,body, [invoker](const Value &obj,const string &err){
if(!err.empty()){
//如果接口访问失败那么该rtsp流认证失败
invoker(unAuthedRealm);
return;
}
invoker(obj["realm"].asString());
2019-05-20 16:26:04 +08:00
});
});
//监听kBroadcastOnRtspAuth事件返回正确的rtsp鉴权用户密码
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastOnRtspAuth,[](BroadcastOnRtspAuthArgs){
GET_CONFIG(string,hook_rtsp_auth,Hook::kOnRtspAuth);
2019-05-20 16:26:04 +08:00
if(unAuthedRealm == realm || !hook_enable || hook_rtsp_auth.empty()){
//认证失败
invoker(false,makeRandStr(12));
return;
}
auto body = make_json(args);
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
body["user_name"] = user_name;
body["must_no_encrypt"] = must_no_encrypt;
body["realm"] = realm;
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_rtsp_auth,body, [invoker](const Value &obj,const string &err){
if(!err.empty()){
//认证失败
invoker(false,makeRandStr(12));
return;
}
invoker(obj["encrypted"].asBool(),obj["passwd"].asString());
2019-05-20 16:26:04 +08:00
});
});
//监听rtsp、rtmp源注册或注销事件
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastMediaChanged,[](BroadcastMediaChangedArgs){
GET_CONFIG(string,hook_stream_chaned,Hook::kOnStreamChanged);
2019-05-20 16:26:04 +08:00
if(!hook_enable || hook_stream_chaned.empty()){
return;
}
ArgsType body;
2021-04-08 17:34:13 +08:00
if (bRegist) {
body = makeMediaSourceJson(sender);
body["regist"] = bRegist;
2021-04-08 17:34:13 +08:00
} else {
body["schema"] = sender.getSchema();
body[VHOST_KEY] = sender.getVhost();
2021-04-08 17:34:13 +08:00
body["app"] = sender.getApp();
body["stream"] = sender.getId();
body["regist"] = bRegist;
2021-04-08 17:34:13 +08:00
}
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_stream_chaned,body, nullptr);
2019-05-20 16:26:04 +08:00
});
2022-01-12 17:43:07 +08:00
GET_CONFIG_FUNC(vector<string>, origin_urls, Cluster::kOriginUrl, [](const string &str) {
vector<string> ret;
for (auto &url : split(str, ";")) {
trim(url);
if (!url.empty()) {
ret.emplace_back(url);
}
2022-01-12 16:45:47 +08:00
}
2022-01-12 17:43:07 +08:00
return ret;
});
2022-01-12 16:45:47 +08:00
2019-05-20 16:26:04 +08:00
//监听播放失败(未找到特定的流)事件
NoticeCenter::Instance().addListener(&web_hook_tag, Broadcast::kBroadcastNotFoundStream, [](BroadcastNotFoundStreamArgs) {
2022-01-12 17:43:07 +08:00
if (!origin_urls.empty()) {
//设置了源站,那么尝试溯源
static atomic<uint8_t> s_index { 0 };
pullStreamFromOrigin(origin_urls, s_index.load(), 0, args, closePlayer);
++s_index;
2022-01-12 16:45:47 +08:00
return;
}
2022-01-12 20:24:19 +08:00
if (start_with(args._param_strs, kEdgeServerParam)) {
//源站收到来自边沿站的溯源请求,流不存在时立即返回拉流失败
closePlayer();
return;
}
2022-01-12 16:45:47 +08:00
GET_CONFIG(string, hook_stream_not_found, Hook::kOnStreamNotFound);
if (!hook_enable || hook_stream_not_found.empty()) {
2019-05-20 16:26:04 +08:00
return;
}
auto body = make_json(args);
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
2019-05-20 17:34:39 +08:00
//执行hook
2022-01-12 16:45:47 +08:00
do_http_hook(hook_stream_not_found, body, nullptr);
2019-05-20 16:26:04 +08:00
});
static auto getRecordInfo = [](const RecordInfo &info) {
ArgsType body;
body["start_time"] = (Json::UInt64) info.start_time;
body["file_size"] = (Json::UInt64) info.file_size;
body["time_len"] = info.time_len;
body["file_path"] = info.file_path;
body["file_name"] = info.file_name;
body["folder"] = info.folder;
body["url"] = info.url;
body["app"] = info.app;
body["stream"] = info.stream;
body[VHOST_KEY] = info.vhost;
return body;
};
2020-04-03 20:45:58 +08:00
#ifdef ENABLE_MP4
2019-05-20 16:26:04 +08:00
//录制mp4文件成功后广播
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastRecordMP4,[](BroadcastRecordMP4Args){
GET_CONFIG(string,hook_record_mp4,Hook::kOnRecordMp4);
if (!hook_enable || hook_record_mp4.empty()) {
2019-05-20 16:26:04 +08:00
return;
}
2019-05-20 17:34:39 +08:00
//执行hook
do_http_hook(hook_record_mp4, getRecordInfo(info), nullptr);
2019-05-20 16:26:04 +08:00
});
2020-04-03 20:45:58 +08:00
#endif //ENABLE_MP4
2019-05-20 16:26:04 +08:00
NoticeCenter::Instance().addListener(&web_hook_tag, Broadcast::kBroadcastRecordTs, [](BroadcastRecordTsArgs) {
GET_CONFIG(string,hook_record_ts,Hook::kOnRecordTs);
2020-09-13 14:08:05 +08:00
if (!hook_enable || hook_record_ts.empty()) {
return;
}
// 执行 hook
do_http_hook(hook_record_ts, getRecordInfo(info), nullptr);
2020-09-13 14:08:05 +08:00
});
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastShellLogin,[](BroadcastShellLoginArgs){
GET_CONFIG(string,hook_shell_login,Hook::kOnShellLogin);
2019-06-06 15:33:11 +08:00
if(!hook_enable || hook_shell_login.empty() || sender.get_peer_ip() == "127.0.0.1"){
invoker("");
return;
}
ArgsType body;
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
body["user_name"] = user_name;
body["passwd"] = passwd;
//执行hook
do_http_hook(hook_shell_login,body, [invoker](const Value &,const string &err){
invoker(err);
});
});
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastStreamNoneReader,[](BroadcastStreamNoneReaderArgs) {
2022-01-12 17:43:07 +08:00
if (!origin_urls.empty()) {
//边沿站无人观看时立即停止溯源
2022-01-12 16:45:47 +08:00
sender.close(false);
WarnL << "无人观看主动关闭流:" << sender.getOriginUrl();
return;
}
GET_CONFIG(string,hook_stream_none_reader,Hook::kOnStreamNoneReader);
2019-05-27 18:39:43 +08:00
if(!hook_enable || hook_stream_none_reader.empty()){
return;
}
ArgsType body;
body["schema"] = sender.getSchema();
body[VHOST_KEY] = sender.getVhost();
2019-05-27 18:39:43 +08:00
body["app"] = sender.getApp();
body["stream"] = sender.getId();
weak_ptr<MediaSource> weakSrc = sender.shared_from_this();
//执行hook
do_http_hook(hook_stream_none_reader,body, [weakSrc](const Value &obj,const string &err){
bool flag = obj["close"].asBool();
auto strongSrc = weakSrc.lock();
if(!flag || !err.empty() || !strongSrc){
return;
}
strongSrc->close(false);
2022-01-12 16:45:47 +08:00
WarnL << "无人观看主动关闭流:" << strongSrc->getOriginUrl();
2019-05-27 18:39:43 +08:00
});
});
2022-08-27 10:53:47 +08:00
NoticeCenter::Instance().addListener(&web_hook_tag, Broadcast::kBroadcastSendRtpStopped, [](BroadcastSendRtpStopped) {
GET_CONFIG(string, hook_send_rtp_stopped, Hook::kOnSendRtpStopped);
if (!hook_enable || hook_send_rtp_stopped.empty()) {
return;
}
ArgsType body;
body[VHOST_KEY] = sender.getVhost();
body["app"] = sender.getApp();
body["stream"] = sender.getStreamId();
body["ssrc"] = ssrc;
body["originType"] = (int)sender.getOriginType(MediaSource::NullMediaSource());
body["originTypeStr"] = getOriginTypeString(sender.getOriginType(MediaSource::NullMediaSource()));
body["originUrl"] = sender.getOriginUrl(MediaSource::NullMediaSource());
body["msg"] = ex.what();
body["err"] = ex.getErrCode();
//执行hook
do_http_hook(hook_send_rtp_stopped, body, nullptr);
});
/**
* kBroadcastHttpAccess事件触发机制
* 1http请求头查找cookie3
2019-11-30 11:44:05 +08:00
* 2http url参数查找cookiecookie则进入步骤5
* 3cookie标记是否有权限访问文件
* 4cookie中记录的url参数是否跟本次url参数一致
* 5kBroadcastHttpAccess事件
*/
//开发者应该通过该事件判定http客户端是否有权限访问http服务器上的特定文件
//ZLMediaKit会记录本次鉴权的结果至cookie
//如果鉴权成功在cookie有效期内那么下次客户端再访问授权目录时ZLMediaKit会直接返回文件
//如果鉴权失败在cookie有效期内如果http url参数不变(否则会立即再次触发鉴权事件)ZLMediaKit会直接返回错误码
//如果用户客户端不支持cookie那么ZLMediaKit会根据url参数查找cookie并追踪用户
//如果没有url参数客户端又不支持cookie那么会根据ip和端口追踪用户
//追踪用户的目的是为了缓存上次鉴权结果,减少鉴权次数,提高性能
NoticeCenter::Instance().addListener(&web_hook_tag,Broadcast::kBroadcastHttpAccess,[](BroadcastHttpAccessArgs){
GET_CONFIG(string,hook_http_access,Hook::kOnHttpAccess);
if(sender.get_peer_ip() == "127.0.0.1" || parser.Params() == hook_adminparams){
//如果是本机或超级管理员访问那么不做访问鉴权权限有效期1个小时
invoker("","",60 * 60);
return;
}
if(!hook_enable || hook_http_access.empty()){
//未开启http文件访问鉴权那么允许访问但是每次访问都要鉴权
//因为后续随时都可能开启鉴权(重载配置文件后可能重新开启鉴权)
invoker("","",0);
2019-06-12 17:53:48 +08:00
return;
}
ArgsType body;
body["ip"] = sender.get_peer_ip();
body["port"] = sender.get_peer_port();
body["id"] = sender.getIdentifier();
body["path"] = path;
body["is_dir"] = is_dir;
body["params"] = parser.Params();
2020-04-20 18:13:45 +08:00
for(auto &pr : parser.getHeader()){
body[string("header.") + pr.first] = pr.second;
2019-06-12 17:53:48 +08:00
}
//执行hook
do_http_hook(hook_http_access,body, [invoker](const Value &obj,const string &err){
if(!err.empty()){
//如果接口访问失败那么仅限本次没有访问http服务器的权限
invoker(err,"",0);
2019-06-12 17:53:48 +08:00
return;
}
//err参数代表不能访问的原因空则代表可以访问
//path参数是该客户端能访问或被禁止的顶端目录如果path为空字符串则表述为当前目录
//second参数规定该cookie超时时间如果second为0本次鉴权结果不缓存
invoker(obj["err"].asString(),obj["path"].asString(),obj["second"].asInt());
2019-06-12 17:53:48 +08:00
});
});
2019-11-29 10:12:20 +08:00
//汇报服务器重新启动
reportServerStarted();
2021-08-20 14:52:48 +08:00
//定时上报保活
reportServerKeepalive();
}
void unInstallWebHook(){
2021-08-20 14:52:48 +08:00
g_keepalive_timer.reset();
NoticeCenter::Instance().delListener(&web_hook_tag);
}