From 35380300e3ebd99cd991daf308c2d3a0292e74ca Mon Sep 17 00:00:00 2001
From: Roman Arutyunyan <arut@qip.ru>
Date: Wed, 14 Nov 2012 16:35:59 +0400
Subject: [PATCH] implemented live pause

---
 ngx_rtmp_live_module.c | 52 ++++++++++++++++++++++++++++++++++++++++--
 ngx_rtmp_live_module.h |  1 +
 2 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/ngx_rtmp_live_module.c b/ngx_rtmp_live_module.c
index ca35751..16b6697 100644
--- a/ngx_rtmp_live_module.c
+++ b/ngx_rtmp_live_module.c
@@ -11,6 +11,7 @@
 static ngx_rtmp_publish_pt              next_publish;
 static ngx_rtmp_play_pt                 next_play;
 static ngx_rtmp_close_stream_pt         next_close_stream;
+static ngx_rtmp_pause_pt                next_pause;
 static ngx_rtmp_stream_begin_pt         next_stream_begin;
 static ngx_rtmp_stream_eof_pt           next_stream_eof;
 
@@ -556,7 +557,51 @@ ngx_rtmp_live_close_stream(ngx_rtmp_session_t *s, ngx_rtmp_close_stream_t *v)
 next:
     return next_close_stream(s, v);
 }
-                           
+
+
+static ngx_int_t
+ngx_rtmp_live_pause(ngx_rtmp_session_t *s, ngx_rtmp_pause_t *v)
+{
+    ngx_rtmp_live_ctx_t            *ctx;
+
+    ctx = ngx_rtmp_get_module_ctx(s, ngx_rtmp_live_module);
+
+    if (ctx == NULL || ctx->stream == NULL) {
+        goto next;
+    }
+
+    ngx_log_debug2(NGX_LOG_DEBUG_RTMP, s->connection->log, 0,
+                   "live: pause=%i timestamp=%f",
+                   (ngx_int_t) v->pause, v->position);
+
+    if (v->pause) {
+        if (ngx_rtmp_send_status(s, "NetStream.Pause.Notify", "status",
+                                 "Paused live")
+            != NGX_OK)
+        {
+            return NGX_ERROR;
+        }
+
+        ctx->paused = 1;
+
+        ngx_rtmp_live_stop(s);
+
+    } else {
+        if (ngx_rtmp_send_status(s, "NetStream.Unpause.Notify", "status",
+                                 "Unpaused live")
+            != NGX_OK)
+        {
+            return NGX_ERROR;
+        }
+
+        ctx->paused = 0;
+
+        ngx_rtmp_live_start(s);
+    }
+
+next:
+    return next_pause(s, v);
+}
 
 static ngx_int_t
 ngx_rtmp_live_av(ngx_rtmp_session_t *s, ngx_rtmp_header_t *h, 
@@ -693,7 +738,7 @@ ngx_rtmp_live_av(ngx_rtmp_session_t *s, ngx_rtmp_header_t *h,
     /* broadcast to all subscribers */
 
     for (pctx = ctx->stream->ctx; pctx; pctx = pctx->next) {
-        if (pctx == ctx) {
+        if (pctx == ctx || pctx->paused) {
             continue;
         }
 
@@ -922,6 +967,9 @@ ngx_rtmp_live_postconfiguration(ngx_conf_t *cf)
     next_close_stream = ngx_rtmp_close_stream;
     ngx_rtmp_close_stream = ngx_rtmp_live_close_stream;
 
+    next_pause = ngx_rtmp_pause;
+    ngx_rtmp_pause = ngx_rtmp_live_pause;
+
     next_stream_begin = ngx_rtmp_stream_begin;
     ngx_rtmp_stream_begin = ngx_rtmp_live_stream_begin;
 
diff --git a/ngx_rtmp_live_module.h b/ngx_rtmp_live_module.h
index f95f468..5a2a2c4 100644
--- a/ngx_rtmp_live_module.h
+++ b/ngx_rtmp_live_module.h
@@ -35,6 +35,7 @@ struct ngx_rtmp_live_ctx_s {
     unsigned                            active:1;
     unsigned                            publishing:1;
     unsigned                            silent:1;
+    unsigned                            paused:1;
 };