添加 http-flv/rtmp/rtsp通用鉴权接口

src/Common/MediaSource.cpp

@@ -150,8 +150,8 @@ void MediaInfo::parse(const string &url){
         auto pos = steamid.find("?");
         if(pos != string::npos){
             m_streamid = steamid.substr(0,pos);
-            auto params = steamid.substr(pos + 1);
+            m_param_strs = steamid.substr(pos + 1);
-            m_params = Parser::parseArgs(params);
+            m_params = Parser::parseArgs(m_param_strs);
             if(m_params.find(VHOST_KEY) != m_params.end()){
                 m_vhost = m_params[VHOST_KEY];
             }

src/Common/MediaSource.h

@@ -83,6 +83,8 @@ public:
     string m_app;
     string m_streamid;
     StrCaseMap m_params;
+    string m_param_strs;
+
 };
 
 

src/Common/config.cpp

@@ -47,13 +47,13 @@ void loadIniConfig(){
 }
 ////////////广播名称///////////
 namespace Broadcast {
-const char kBroadcastMediaPlayed[] = "kBroadcastMediaPlayed";
 const char kBroadcastMediaChanged[] = "kBroadcastMediaChanged";
 const char kBroadcastRecordMP4[] = "kBroadcastRecordMP4";
 const char kBroadcastHttpRequest[] = "kBroadcastHttpRequest";
 const char kBroadcastOnGetRtspRealm[] = "kBroadcastOnGetRtspRealm";
 const char kBroadcastOnRtspAuth[] = "kBroadcastOnRtspAuth";
-
+const char kBroadcastMediaPlayed[] = "kBroadcastMediaPlayed";
+const char kBroadcastRtmpPublish[] = "kBroadcastRtmpPublish";
 } //namespace Broadcast
 
 //代理失败最大重试次数

src/Common/config.h

@@ -51,6 +51,7 @@ void loadIniConfig();
 
 #define SERVER_NAME "ZLMediaKit"
 #define VHOST_KEY "vhost"
+#define HTTP_SCHEMA "http"
 #define RTSP_SCHEMA "rtsp"
 #define RTMP_SCHEMA "rtmp"
 #define DEFAULT_VHOST "__defaultVhost__"
@@ -59,15 +60,16 @@ void loadIniConfig();
 
 ////////////广播名称///////////
 namespace Broadcast {
-extern const char kBroadcastMediaPlayed[];
-#define BroadcastMediaPlayedArgs const char *schema,const char *vhost,const char *app,const char *stream
 
+//注册或反注册MediaSource事件广播
 extern const char kBroadcastMediaChanged[];
 #define BroadcastMediaChangedArgs bool bRegist, const char *schema,const char *vhost,const char *app,const char *stream
 
+//录制mp4文件成功后广播
 extern const char kBroadcastRecordMP4[];
 #define BroadcastRecordMP4Args const Mp4Info &info
 
+//收到http api请求广播
 extern const char kBroadcastHttpRequest[];
 #define BroadcastHttpRequestArgs const Parser &parser,HttpSession::HttpResponseInvoker &invoker,bool &consumed
 
@@ -80,6 +82,18 @@ extern const char kBroadcastOnGetRtspRealm[];
 extern const char kBroadcastOnRtspAuth[];
 #define BroadcastOnRtspAuthArgs const char *user_name,bool must_no_encrypt,const RtspSession::onAuth &invoker
 
+
+//鉴权结果回调对象
+typedef std::function<void(bool success)> AuthInvoker;
+
+//收到rtmp推流事件广播，通过该事件控制推流鉴权
+extern const char kBroadcastRtmpPublish[];
+#define BroadcastRtmpPublishArgs MediaInfo &args,Broadcast::AuthInvoker &invoker
+
+//播放rtsp或rtmp事件广播，通过该事件控制播放鉴权
+extern const char kBroadcastMediaPlayed[];
+#define BroadcastMediaPlayedArgs MediaInfo &args,Broadcast::AuthInvoker &invoker
+
 } //namespace Broadcast
 
 //代理失败最大重试次数

src/Http/HttpSession.cpp

@@ -171,7 +171,8 @@ void HttpSession::onManager() {
 		shutdown();
 	}
 }
-
+//http-flv 链接格式:http://vhost-url:port/app/streamid.flv?key1=value1&key2=value2
+//如果url(除去?以及后面的参数)后缀是.flv,那么表明该url是一个http-flv直播。
 inline bool HttpSession::checkLiveFlvStream(){
 	auto pos = strrchr(m_parser.Url().data(),'.');
 	if(!pos){
@@ -182,23 +183,34 @@ inline bool HttpSession::checkLiveFlvStream(){
 		//未找到".flv"后缀
 		return false;
 	}
-    auto fullUrl = string("http://") + m_parser["Host"] + FindField(m_parser.Url().data(),NULL,pos);
+    //拼接成完整url
+    auto fullUrl = string(HTTP_SCHEMA) + "://" + m_parser["Host"] + m_parser.FullUrl();
     MediaInfo info(fullUrl);
-	if(!m_parser.getUrlArgs()[VHOST_KEY].empty()){
+    info.m_streamid.erase(info.m_streamid.size() - 4);//去除.flv后缀
-		info.m_vhost = m_parser.getUrlArgs()[VHOST_KEY];
-	}
 
 	auto mediaSrc = dynamic_pointer_cast<RtmpMediaSource>(MediaSource::find(RTMP_SCHEMA,info.m_vhost,info.m_app,info.m_streamid));
 	if(!mediaSrc){
 		//该rtmp源不存在
-		return false;
+        sendNotFound(true);
+        shutdown();
+		return true;
 	}
 	if(!mediaSrc->ready()){
 		//未准备好
-		return false;
+        sendNotFound(true);
+        shutdown();
+		return true;
+	}
+
+    auto onRes = [this,mediaSrc](bool authSuccess){
+        if(!authSuccess){
+            string status = "401 Unauthorized";
+            sendResponse(status.data(), makeHttpHeader(true,status.size()),status);
+            shutdown();
+            return ;
         }
         //找到rtmp源，发送http头，负载后续发送
-	sendResponse("200 OK", makeHttpHeader(false,0,get_mime_type(m_parser.Url().data())), "");
+        sendResponse("200 OK", makeHttpHeader(false,0,get_mime_type(".flv")), "");
         //发送flv文件头
         char flv_file_header[] = "FLV\x1\x5\x0\x0\x0\x9"; // have audio and have video
         bool is_have_audio = false,is_have_video = false;
@@ -261,8 +273,28 @@ inline bool HttpSession::checkLiveFlvStream(){
                 strongSelf->shutdown();
             });
         });
-	return true;
+    };
 
+    weak_ptr<HttpSession> weakSelf = dynamic_pointer_cast<HttpSession>(shared_from_this());
+    Broadcast::AuthInvoker invoker = [weakSelf,onRes](bool authSuccess){
+        auto strongSelf = weakSelf.lock();
+        if(!strongSelf){
+            return;
+        }
+        strongSelf->async([weakSelf,onRes,authSuccess](){
+            auto strongSelf = weakSelf.lock();
+            if(!strongSelf){
+                return;
+            }
+            onRes(authSuccess);
+        });
+    };
+    auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,info,invoker);
+    if(!flag){
+        //该事件无人监听,默认不鉴权
+        onRes(true);
+    }
+    return true;
 }
 inline HttpSession::HttpCode HttpSession::Handle_Req_GET() {
 	//先看看该http事件是否被拦截
@@ -274,11 +306,9 @@ inline HttpSession::HttpCode HttpSession::Handle_Req_GET() {
 	}
 	//事件未被拦截，则认为是http下载请求
 
-	auto fullUrl = string("http://") + m_parser["Host"] + m_parser.Url();
+	auto fullUrl = string(HTTP_SCHEMA) + "://" + m_parser["Host"] + m_parser.FullUrl();
 	MediaInfo info(fullUrl);
-	if(!m_parser.getUrlArgs()[VHOST_KEY].empty()){
+
-		info.m_vhost = m_parser.getUrlArgs()[VHOST_KEY];
-	}
 	string strFile = m_strPath + "/" + info.m_vhost + m_parser.Url();
 	/////////////HTTP连接是否需要被关闭////////////////
 	static uint32_t reqCnt =  mINI::Instance()[Config::Http::kMaxReqCount].as<uint32_t>();

src/Rtmp/RtmpSession.cpp

@@ -101,7 +101,7 @@ void RtmpSession::onCmd_connect(AMFDecoder &dec) {
     m_strTcUrl = params["tcUrl"].as_string();
     if(m_strTcUrl.empty()){
         //defaultVhost:默认vhost
-        m_strTcUrl = "rtmp://127.0.0.1/" + m_mediaInfo.m_app;
+        m_strTcUrl = string(RTMP_SCHEMA) + "://" + DEFAULT_VHOST + "/" + m_mediaInfo.m_app;
     }
 	bool ok = true; //(app == APP_NAME);
 	AMFValue version(AMF_OBJECT);
@@ -129,26 +129,54 @@ void RtmpSession::onCmd_createStream(AMFDecoder &dec) {
 void RtmpSession::onCmd_publish(AMFDecoder &dec) {
 	dec.load<AMFValue>();/* NULL */
     m_mediaInfo.parse(m_strTcUrl + "/" + dec.load<std::string>());
+
+    auto onRes = [this](bool authSuccess){
         auto src = dynamic_pointer_cast<RtmpMediaSource>(MediaSource::find(RTMP_SCHEMA,
                                                                            m_mediaInfo.m_vhost,
                                                                            m_mediaInfo.m_app,
                                                                            m_mediaInfo.m_streamid,
                                                                            false));
-	bool ok = (!src && !m_pPublisherSrc);
+        bool ok = (!src && !m_pPublisherSrc && authSuccess);
         AMFValue status(AMF_OBJECT);
         status.set("level", ok ? "status" : "error");
-	status.set("code", ok ? "NetStream.Publish.Start" : "NetStream.Publish.BadName");
+        status.set("code", ok ? "NetStream.Publish.Start" : (authSuccess ? "NetStream.Publish.BadName" : "NetStream.Publish.BadAuth"));
-	status.set("description", ok ? "Started publishing stream." : "Already publishing.");
+        status.set("description", ok ? "Started publishing stream." : (authSuccess ? "Already publishing." : "Auth failed"));
         status.set("clientid", "0");
         sendReply("onStatus", nullptr, status);
         if (!ok) {
-		throw std::runtime_error( StrPrinter << "Already publishing:"
+            WarnL << "onPublish:"
+                  << (authSuccess ? "Already publishing:" : "Auth failed:")
                   << m_mediaInfo.m_vhost << " "
                   << m_mediaInfo.m_app << " "
-                                             << m_mediaInfo.m_streamid << endl);
+                  << m_mediaInfo.m_streamid << endl;
+            shutdown();
+            return;
         }
         m_bPublisherSrcRegisted = false;
         m_pPublisherSrc.reset(new RtmpToRtspMediaSource(m_mediaInfo.m_vhost,m_mediaInfo.m_app,m_mediaInfo.m_streamid));
+    };
+
+    weak_ptr<RtmpSession> weakSelf = dynamic_pointer_cast<RtmpSession>(shared_from_this());
+    Broadcast::AuthInvoker invoker = [weakSelf,onRes](bool success){
+        auto strongSelf = weakSelf.lock();
+        if(!strongSelf){
+            return;
+        }
+        strongSelf->async([weakSelf,onRes,success](){
+            auto strongSelf = weakSelf.lock();
+            if(!strongSelf){
+                return;
+            }
+            onRes(success);
+        });
+    };
+    auto flag = NoticeCenter::Instance().emitEvent(Config::Broadcast::kBroadcastRtmpPublish,
+                                                   m_mediaInfo,
+                                                   invoker);
+    if(!flag){
+        //该事件无人监听，默认鉴权成功
+        onRes(true);
+    }
 }
 
 void RtmpSession::onCmd_deleteStream(AMFDecoder &dec) {
@@ -161,31 +189,37 @@ void RtmpSession::onCmd_deleteStream(AMFDecoder &dec) {
 }
 
 void  RtmpSession::doPlay(AMFDecoder &dec){
-    m_mediaInfo.parse(m_strTcUrl + "/" + dec.load<std::string>());
+    auto onRes = [this](bool authSuccess) {
         auto src = dynamic_pointer_cast<RtmpMediaSource>(MediaSource::find(RTMP_SCHEMA,
                                                                            m_mediaInfo.m_vhost,
                                                                            m_mediaInfo.m_app,
                                                                            m_mediaInfo.m_streamid,
                                                                            true));
-	bool ok = (src.operator bool());
+        bool ok = (src.operator bool() && authSuccess);
+        if(ok){
             ok = ok && src->ready();
+        }
+        if (ok) {
             //stream begin
            sendUserControl(CONTROL_STREAM_BEGIN, STREAM_MEDIA);
-
+        }
         // onStatus(NetStream.Play.Reset)
         AMFValue status(AMF_OBJECT);
         status.set("level", ok ? "status" : "error");
-	status.set("code", ok ? "NetStream.Play.Reset" : "NetStream.Play.StreamNotFound");
+        status.set("code", ok ? "NetStream.Play.Reset" : (authSuccess ? "NetStream.Play.StreamNotFound" : "NetStream.Play.BadAuth"));
-	status.set("description", ok ? "Resetting and playing." : "No such stream.");
+        status.set("description", ok ? "Resetting and playing." : (authSuccess ? "No such stream." : "Auth failed"));
         status.set("details", m_mediaInfo.m_streamid);
         status.set("clientid", "0");
         sendReply("onStatus", nullptr, status);
         if (!ok) {
-		throw std::runtime_error( StrPrinter << "No such stream:"
+            WarnL << "onPlayed:"
+                  << (authSuccess ? "No such stream:" : "Auth failed:")
                   << m_mediaInfo.m_vhost << " "
                   << m_mediaInfo.m_app << " "
                   << m_mediaInfo.m_streamid
-                                             << endl);
+                  << endl;
+            shutdown();
+            return;
         }
 
         // onStatus(NetStream.Play.Start)
@@ -255,18 +289,34 @@ void  RtmpSession::doPlay(AMFDecoder &dec){
         if (src->getRing()->readerCount() == 1) {
             src->seekTo(0);
         }
+    };
 
-    NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,
+    weak_ptr<RtmpSession> weakSelf = dynamic_pointer_cast<RtmpSession>(shared_from_this());
-                                       RTMP_SCHEMA,
+    Broadcast::AuthInvoker invoker = [weakSelf,onRes](bool authSuccess){
-                                       m_mediaInfo.m_vhost.data(),
+        auto strongSelf = weakSelf.lock();
-                                       m_mediaInfo.m_app.data(),
+        if(!strongSelf){
-                                       m_mediaInfo.m_streamid.data());
+            return;
+        }
+        strongSelf->async([weakSelf,onRes,authSuccess](){
+            auto strongSelf = weakSelf.lock();
+            if(!strongSelf){
+                return;
+            }
+            onRes(authSuccess);
+        });
+    };
+    auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,m_mediaInfo,invoker);
+    if(!flag){
+        //该事件无人监听,默认不鉴权
+        onRes(true);
+    }
 }
 void RtmpSession::onCmd_play2(AMFDecoder &dec) {
 	doPlay(dec);
 }
 void RtmpSession::onCmd_play(AMFDecoder &dec) {
 	dec.load<AMFValue>();/* NULL */
+    m_mediaInfo.parse(m_strTcUrl + "/" + dec.load<std::string>());
 	doPlay(dec);
 }
 

src/Rtsp/Rtsp.h

@@ -101,15 +101,15 @@ public:
 			}
 			if (start == buf) {
 				m_strMethod = FindField(line.c_str(), NULL, " ");
-				auto full_url = FindField(line.c_str(), " ", " ");
+                m_strFullUrl = FindField(line.c_str(), " ", " ");
-				auto args_pos =  full_url.find('?');
+				auto args_pos =  m_strFullUrl.find('?');
 				if(args_pos != string::npos){
-					m_strUrl = full_url.substr(0,args_pos);
+					m_strUrl = m_strFullUrl.substr(0,args_pos);
-					m_mapUrlArgs = parseArgs(full_url.substr(args_pos + 1 ));
+					m_mapUrlArgs = parseArgs(m_strFullUrl.substr(args_pos + 1 ));
 				}else{
-					m_strUrl = full_url;
+					m_strUrl = m_strFullUrl;
 				}
-				m_strTail = FindField(line.c_str(), (full_url + " ").c_str(), NULL);
+				m_strTail = FindField(line.c_str(), (m_strFullUrl + " ").c_str(), NULL);
 			} else {
 				auto field = FindField(line.c_str(), NULL, ": ");
 				auto value = FindField(line.c_str(), ": ", NULL);
@@ -132,6 +132,10 @@ public:
 		//rtsp url
 		return m_strUrl;
 	}
+    const string& FullUrl() const {
+        //rtsp url with args
+        return m_strFullUrl;
+    }
 	const string& Tail() const {
 		//RTSP/1.0
 		return m_strTail;
@@ -150,6 +154,7 @@ public:
 	void Clear() {
 		m_strMethod.clear();
 		m_strUrl.clear();
+        m_strFullUrl.clear();
 		m_strTail.clear();
 		m_strContent.clear();
 		m_mapValues.clear();
@@ -189,6 +194,7 @@ private:
 	string m_strTail;
 	string m_strContent;
 	string m_strNull;
+    string m_strFullUrl;
 	mutable StrCaseMap m_mapValues;
 	mutable StrCaseMap m_mapUrlArgs;
 };

src/Rtsp/RtspSession.cpp

@@ -193,10 +193,7 @@ bool RtspSession::handleReq_Describe() {
     {
         //解析url获取媒体名称
         m_strUrl = m_parser.Url();
-        m_mediaInfo.parse(m_strUrl);
+        m_mediaInfo.parse(m_parser.FullUrl());
-        if(!m_parser.getUrlArgs()[VHOST_KEY].empty()){
-            m_mediaInfo.m_vhost = m_parser.getUrlArgs()[VHOST_KEY];
-        }
     }
 
 	if (!findStream()) {
@@ -644,7 +641,21 @@ bool RtspSession::handleReq_Play() {
 		send_SessionNotFound();
 		return false;
 	}
-
+    auto onRes = [this](bool authSuccess){
+        char response[2 * 1024];
+        m_pcBuf = response;
+        if(!authSuccess && m_bFirstPlay){
+            //第一次play是播放，否则是恢复播放。只对播放鉴权
+            int n = sprintf(m_pcBuf,
+                            "RTSP/1.0 401 Unauthorized\r\n"
+                            "CSeq: %d\r\n"
+                            "Server: %s-%0.2f(build in %s)\r\n"
+                            "%s\r\n",
+                            m_iCseq, SERVER_NAME, RTSP_VERSION, RTSP_BUILDTIME, dateHeader().data());
+            send(m_pcBuf,n);
+            shutdown();
+            return;
+        }
         if(m_pRtpReader){
             weak_ptr<RtspSession> weakSelf = dynamic_pointer_cast<RtspSession>(shared_from_this());
             SockUtil::setNoDelay(m_pSender->rawFD(), false);
@@ -705,7 +716,8 @@ bool RtspSession::handleReq_Play() {
             auto &track = m_aTrackInfo[i];
             if (track.inited == false) {
                 //还有track没有setup
-			return false;
+                shutdown();
+                return;
             }
             iLen += sprintf(m_pcBuf + iLen, "url=%s/%s%d;seq=%d;rtptime=%u,",
                             m_strUrl.data(), track.trackStyle.data(),
@@ -715,11 +727,27 @@ bool RtspSession::handleReq_Play() {
         (m_pcBuf)[iLen] = '\0';
         iLen += sprintf(m_pcBuf + iLen, "\r\n\r\n");
         send(m_pcBuf, iLen);
-	NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,
+    };
-                                       RTSP_SCHEMA,
+
-                                       m_mediaInfo.m_vhost.data(),
+    weak_ptr<RtspSession> weakSelf = dynamic_pointer_cast<RtspSession>(shared_from_this());
-                                       m_mediaInfo.m_app.data(),
+    Broadcast::AuthInvoker invoker = [weakSelf,onRes](bool authSuccess){
-                                       m_mediaInfo.m_streamid.data());
+        auto strongSelf = weakSelf.lock();
+        if(!strongSelf){
+            return;
+        }
+        strongSelf->async([weakSelf,onRes,authSuccess](){
+            auto strongSelf = weakSelf.lock();
+            if(!strongSelf){
+                return;
+            }
+            onRes(authSuccess);
+        });
+    };
+    auto flag = NoticeCenter::Instance().emitEvent(Broadcast::kBroadcastMediaPlayed,m_mediaInfo,invoker);
+    if(!flag){
+        //该事件无人监听,默认不鉴权
+        onRes(true);
+    }
 	return true;
 }
 

tests/test_server.cpp

@@ -107,6 +107,22 @@ static onceToken s_token([](){
 			invoker(false,user);
 		});
 	});
+
+
+	NoticeCenter::Instance().addListener(nullptr,Config::Broadcast::kBroadcastRtmpPublish,[](BroadcastRtmpPublishArgs){
+        InfoL << args.m_vhost << " " << args.m_app << " " << args.m_streamid << " " << args.m_param_strs ;
+        EventPoller::Instance().async([invoker](){
+            invoker(true);
+        });
+    });
+
+    NoticeCenter::Instance().addListener(nullptr,Config::Broadcast::kBroadcastMediaPlayed,[](BroadcastMediaPlayedArgs){
+        InfoL << args.m_schema << " " << args.m_vhost << " " << args.m_app << " " << args.m_streamid << " " << args.m_param_strs ;
+        EventPoller::Instance().async([invoker](){
+            invoker(true);
+        });
+    });
+
 }, nullptr);
 
 int main(int argc,char *argv[]){