Merge pull request #17 from jianglieshan/aler9

fix:修改rtsp插件作为服务端出流时，ssrc为0的bug

client.go

@@ -151,7 +151,7 @@ func (client *RTSPClient) pullStream() {
 				f(append(clone, payload...))
 			}
 		},
-		ReadBufferSize: 10000,
+		ReadBufferSize: config.ReadBufferSize,
 		Transport:      &client.Transport,
 	}
 	// parse URL
@@ -165,6 +165,9 @@ func (client *RTSPClient) pullStream() {
 		Printf("connect:%s error:%v", client.URL, err)
 		return
 	}
+	client.OnClose = func() {
+		client.Client.Close()
+	}
 	//client.close should be after connected!
 	defer client.Client.Close()
 	var res *base.Response

main.go

@@ -13,14 +13,15 @@ import (
 )
 
 var config = struct {
-	ListenAddr   string
-	UDPAddr      string
-	RTCPAddr     string
-	Timeout      int
-	Reconnect    bool
-	AutoPullList map[string]string
-	AutoPushList map[string]string
-}{":554", ":8000", ":8001", 0, false, nil, nil}
+	ListenAddr     string
+	UDPAddr        string
+	RTCPAddr       string
+	Timeout        int
+	Reconnect      bool
+	AutoPullList   map[string]string
+	AutoPushList   map[string]string
+	ReadBufferSize int
+}{":554", ":8000", ":8001", 0, false, nil, nil, 2048}
 
 var pconfig = PluginConfig{
 	Name:   "RTSP",
@@ -84,7 +85,7 @@ func runPlugin() {
 			Println(err)
 		}
 	}
-	
+
 	go AddHook(HOOK_PUBLISH, func(s *Stream) {
 		for streamPath, url := range config.AutoPushList {
 			if s.StreamPath == streamPath {

server.go

@@ -75,7 +75,7 @@ func (sh *RTSPServer) OnDescribe(ctx *gortsplib.ServerHandlerOnDescribeCtx) (*ba
 		sub.Type = "RTSP pull"
 		sub.vt = s.WaitVideoTrack("h264", "h265")
 		sub.at = s.WaitAudioTrack("aac", "pcma", "pcmu")
-		ssrc := uintptr(unsafe.Pointer(stream))
+		ssrc := uintptr(unsafe.Pointer(&stream))
 		var trackIds = 0
 		if sub.vt != nil {
 			trackId := trackIds
@@ -98,8 +98,15 @@ func (sh *RTSPServer) OnDescribe(ctx *gortsplib.ServerHandlerOnDescribeCtx) (*ba
 			var st uint32
 			onVideo := func(ts uint32, pack *engine.VideoPack) {
 				for i, nalu := range pack.NALUs {
-					for _, rtpack := range vpacketer.Packetize(nalu, (ts-st)*90) {
-						rtpack.Marker = i == len(pack.NALUs)-1
+					var samples uint32
+					if i == len(pack.NALUs)-1 {
+						samples = (ts - st) * 90
+					} else {
+						samples = 0
+					}
+					packs := vpacketer.Packetize(nalu, samples)
+					for j, rtpack := range packs {
+						rtpack.Marker = i == len(pack.NALUs)-1 && j == len(packs)-1
 						rtp, _ := rtpack.Marshal()
 						stream.WritePacketRTP(trackId, rtp)
 					}