From 88373387b8cf909bdfd2c3f04363703929b481a2 Mon Sep 17 00:00:00 2001
From: nareix <hi.nareix@gmail.com>
Date: Fri, 1 Jul 2016 19:20:07 +0800
Subject: [PATCH] change to h264parser.FindDataNALUInAVCCNALUs()

---
 rtmp.go | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/rtmp.go b/rtmp.go
index c1a2d79..1790f4a 100644
--- a/rtmp.go
+++ b/rtmp.go
@@ -708,11 +708,11 @@ func (self *Conn) ReadPacket() (pkt av.Packet, err error) {
 		switch tag := _tag.(type) {
 		case *flvio.Videodata:
 			pkt.CompositionTime = tsToTime(uint32(tag.CompositionTime))
-			if typ := h264parser.CheckNALUsType(tag.Data); typ != h264parser.NALU_AVCC {
-				err = fmt.Errorf("rtmp: input h264 nalu format=%d invalid", typ)
+			var ok bool
+			if pkt.Data, ok = h264parser.FindDataNALUInAVCCNALUs(tag.Data); !ok {
+				err = fmt.Errorf("rtmp: input h264 format invalid")
 				return
 			}
-			pkt.Data = tag.Data[4:]
 			pkt.IsKeyFrame = tag.FrameType == flvio.FRAME_KEY
 			pkt.Idx = int8(self.videostreamidx)
 			break poll