From eaddc60775e7486ef677dd3c38892113a0826e6f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AE=87=E7=BF=94?= <178529795@qq.com>
Date: Thu, 21 May 2020 18:16:56 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BF=87=E6=BB=A4=E5=BC=80=E5=A4=B4=E7=9A=84No?=
 =?UTF-8?q?nIDR?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/main.go b/main.go
index ea5fd5c..944920c 100644
--- a/main.go
+++ b/main.go
@@ -234,15 +234,17 @@ func (rtsp *RTSP) handleNALU(nalType byte, payload []byte, ts int64) {
 		r.Write(payload)
 		rtsp.PushVideo(uint32(ts), r.Bytes())
 	case NALU_Non_IDR_Picture:
-		r := bytes.NewBuffer([]byte{})
-		pframeHead := []byte{0x27, 0x01, 0, 0, 0}
-		util.BigEndian.PutUint24(pframeHead[2:], 0)
-		r.Write(pframeHead)
-		nalLength := []byte{0, 0, 0, 0}
-		util.BigEndian.PutUint32(nalLength, uint32(vl))
-		r.Write(nalLength)
-		r.Write(payload)
-		rtsp.PushVideo(uint32(ts), r.Bytes())
+		if rtsp.avcsent {
+			r := bytes.NewBuffer([]byte{})
+			pframeHead := []byte{0x27, 0x01, 0, 0, 0}
+			util.BigEndian.PutUint24(pframeHead[2:], 0)
+			r.Write(pframeHead)
+			nalLength := []byte{0, 0, 0, 0}
+			util.BigEndian.PutUint32(nalLength, uint32(vl))
+			r.Write(nalLength)
+			r.Write(payload)
+			rtsp.PushVideo(uint32(ts), r.Bytes())
+		}
 	}
 }
 func (rtsp *RTSP) handleRTP(pack *RTPPack) {