# HG changeset patch
# User mstorsjo
# Date 1283077516 0
# Node ID 2685bf2b0b784efd982a7ae3c674a59932f388f6
# Parent  cb5f5fc848b019f3a5f1190540a2cd293c48726a
rtsp: Return AVERROR_EOF when all streams have received an RTCP BYE packet

Patch by Josh Allmann, joshua dot allmann at gmail

diff -r cb5f5fc848b0 -r 2685bf2b0b78 rtpdec.c
--- a/rtpdec.c	Sun Aug 29 10:20:18 2010 +0000
+++ b/rtpdec.c	Sun Aug 29 10:25:16 2010 +0000
@@ -92,11 +92,13 @@
             buf += payload_len;
             len -= payload_len;
             break;
+        case RTCP_BYE:
+            return -RTCP_BYE;
         default:
             return -1;
         }
     }
-    return 0;
+    return -1;
 }
 
 #define RTP_SEQ_MOD (1<<16)
@@ -451,8 +453,7 @@
     if ((buf[0] & 0xc0) != (RTP_VERSION << 6))
         return -1;
     if (buf[1] >= RTCP_SR && buf[1] <= RTCP_APP) {
-        rtcp_parse_packet(s, buf, len);
-        return -1;
+        return rtcp_parse_packet(s, buf, len);
     }
     payload_type = buf[1] & 0x7f;
     if (buf[1] & 0x80)
diff -r cb5f5fc848b0 -r 2685bf2b0b78 rtsp.c
--- a/rtsp.c	Sun Aug 29 10:20:18 2010 +0000
+++ b/rtsp.c	Sun Aug 29 10:25:16 2010 +0000
@@ -1226,6 +1226,7 @@
     char cmd[1024];
 
     av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
+    rt->nb_byes = 0;
 
     if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
         if (rt->state == RTSP_STATE_PAUSED) {
@@ -1777,6 +1778,9 @@
     uint8_t buf[10 * RTP_MAX_PACKET_LENGTH];
     RTSPStream *rtsp_st;
 
+    if (rt->nb_byes == rt->nb_rtsp_streams)
+        return AVERROR_EOF;
+
     /* get next frames from the same RTP packet */
     if (rt->cur_transport_priv) {
         if (rt->transport == RTSP_TRANSPORT_RDT) {
@@ -1833,6 +1837,15 @@
                         rtpctx2->first_rtcp_ntp_time = rtpctx->first_rtcp_ntp_time;
                 }
             }
+            if (ret == -RTCP_BYE) {
+                rt->nb_byes++;
+
+                av_log(s, AV_LOG_DEBUG, "Received BYE for stream %d (%d/%d)\n",
+                       rtsp_st->stream_index, rt->nb_byes, rt->nb_rtsp_streams);
+
+                if (rt->nb_byes == rt->nb_rtsp_streams)
+                    return AVERROR_EOF;
+            }
         }
     }
     if (ret < 0)
diff -r cb5f5fc848b0 -r 2685bf2b0b78 rtsp.h
--- a/rtsp.h	Sun Aug 29 10:20:18 2010 +0000
+++ b/rtsp.h	Sun Aug 29 10:25:16 2010 +0000
@@ -303,6 +303,11 @@
 
     /** RTSP transport mode, such as plain or tunneled. */
     enum RTSPControlTransport control_transport;
+
+    /* Number of RTCP BYE packets the RTSP session has received.
+     * An EOF is propagated back if nb_byes == nb_streams.
+     * This is reset after a seek. */
+    int nb_byes;
 } RTSPState;
 
 /**