HLS: support kick-off hls client

trunk/src/app/srs_app_http_static.cpp

@@ -44,13 +44,23 @@ using namespace std;
 SrsM3u8CtxInfo::SrsM3u8CtxInfo()
 {
     req = NULL;
+    interrupt = false;
 }
 
 SrsM3u8CtxInfo::~SrsM3u8CtxInfo()
 {
     srs_freep(req);
 }
 
+void SrsM3u8CtxInfo::expire()
+{
+    interrupt = true;
+
+    // remove statistic quickly
+    SrsStatistic* stat = SrsStatistic::instance();
+    stat->on_disconnect(ctx, srs_success);
+}
+
 SrsHlsStream::SrsHlsStream()
 {
     _srs_hybrid->timer5s()->subscribe(this);
@@ -94,6 +104,12 @@ srs_error_t SrsHlsStream::serve_m3u8_ctx(ISrsHttpResponseWriter* w, ISrsHttpMess
             *served = false;
             return srs_success;
         }
+
+        if (is_interrupt(ctx)) {
+            srs_warn("Reject: HLS stream is EOF, ctx=%s", ctx.c_str());
+            return srs_go_http_error(w, SRS_CONSTS_HTTP_NotFound, srs_fmt("HLS stream %s is EOF", ctx.c_str()));
+        }
+
         err = serve_exists_session(w, r, factory, fullpath);
     } else {
         // Create a m3u8 in memory, contains the session id(ctx).
@@ -244,14 +260,25 @@ void SrsHlsStream::alive(std::string ctx, SrsRequest* req)
     if (it == map_ctx_info_.end()) {
         SrsM3u8CtxInfo *info = new SrsM3u8CtxInfo();
         info->req = req->copy();
+        info->ctx = ctx;
         info->request_time = srs_get_system_time();
         map_ctx_info_.insert(make_pair(ctx, info));
+
+        // Update the conn of stat client, which is used for receiving the event of kickoff.
+        SrsStatistic* stat = SrsStatistic::instance();
+        SrsStatisticClient* client = stat->find_client(ctx);
+        if (client) {
+            client->conn = info;
+        }
+
         return;
     }
 
     // Update alive time of context.
     SrsM3u8CtxInfo* info = it->second;
-    info->request_time = srs_get_system_time();
+    if (!info->interrupt) {
+        info->request_time = srs_get_system_time();
+    }
 }
 
 srs_error_t SrsHlsStream::http_hooks_on_play(SrsRequest* req)
@@ -347,6 +374,14 @@ srs_error_t SrsHlsStream::on_timer(srs_utime_t interval)
     return err;
 }
 
+bool SrsHlsStream::is_interrupt(std::string id) {
+    std::map<std::string, SrsM3u8CtxInfo*>::iterator it = map_ctx_info_.find(id);
+    if (it != map_ctx_info_.end()) {
+        return it->second->interrupt;
+    }
+    return false;
+}
+
 SrsVodStream::SrsVodStream(string root_dir) : SrsHttpFileServer(root_dir)
 {
 }

trunk/src/app/srs_app_http_static.hpp

@@ -13,12 +13,17 @@
 
 class ISrsFileReaderFactory;
 
-struct SrsM3u8CtxInfo
+struct SrsM3u8CtxInfo: public ISrsExpire
 {
     srs_utime_t request_time;
     SrsRequest* req;
+    std::string ctx;
+    bool interrupt;
     SrsM3u8CtxInfo();
     virtual ~SrsM3u8CtxInfo();
+// Interface ISrsExpire.
+public:
+    virtual void expire();
 };
 
 // Server HLS streaming.
@@ -40,6 +45,7 @@ class SrsHlsStream : public ISrsFastTimer
     void alive(std::string ctx, SrsRequest* req);
     srs_error_t http_hooks_on_play(SrsRequest* req);
     void http_hooks_on_stop(SrsRequest* req);
+    bool is_interrupt(std::string id);
 // interface ISrsFastTimer
 private:
     srs_error_t on_timer(srs_utime_t interval);