add SetupTcp, SetupUdp

examples/client-tcp.go

@@ -39,11 +39,7 @@ func main() {
 	}
 
 	for i, media := range sdpd.MediaDescriptions {
-		_, err := rconn.Setup(u, media, []string{
-			"RTP/AVP/TCP",
-			"unicast",
-			fmt.Sprintf("interleaved=%d-%d", (i * 2), (i*2)+1),
-		})
+		_, err := rconn.SetupTcp(u, media, i)
 		if err != nil {
 			panic(err)
 		}
@@ -54,9 +50,7 @@ func main() {
 		panic(err)
 	}
 
-	frame := &gortsplib.InterleavedFrame{
-		Content: make([]byte, 512*1024),
-	}
+	frame := &gortsplib.InterleavedFrame{Content: make([]byte, 512*1024)}
 
 	for {
 		err := rconn.ReadFrame(frame)
@@ -64,6 +58,7 @@ func main() {
 			panic(err)
 		}
 
-		fmt.Println("incoming", frame.Channel, frame.Content)
+		trackId, streamType := gortsplib.ConvChannelToTrackIdAndStreamType(frame.Channel)
+		fmt.Printf("packet from track %d, type %v: %v\n", trackId, streamType, frame.Content)
 	}
 }

examples/client-udp.go

@@ -0,0 +1,100 @@
+// +build ignore
+
+package main
+
+import (
+	"fmt"
+	"net"
+	"net/url"
+	"strconv"
+	"time"
+
+	"github.com/aler9/gortsplib"
+)
+
+func main() {
+	u, err := url.Parse("rtsp://user:pass@example.com/mystream")
+	if err != nil {
+		panic(err)
+	}
+
+	conn, err := net.DialTimeout("tcp", u.Host, 5*time.Second)
+	if err != nil {
+		panic(err)
+	}
+	defer conn.Close()
+
+	rconn, err := gortsplib.NewConnClient(gortsplib.ConnClientConf{Conn: conn})
+	if err != nil {
+		panic(err)
+	}
+
+	_, err = rconn.Options(u)
+	if err != nil {
+		panic(err)
+	}
+
+	sdpd, _, err := rconn.Describe(u)
+	if err != nil {
+		panic(err)
+	}
+
+	var rtpListeners []net.PacketConn
+	var rtcpListeners []net.PacketConn
+
+	for i, media := range sdpd.MediaDescriptions {
+		rtpPort := 9000 + i*2
+		rtpl, err := net.ListenPacket("udp", ":"+strconv.FormatInt(int64(rtpPort), 10))
+		if err != nil {
+			panic(err)
+		}
+		rtpListeners = append(rtpListeners, rtpl)
+
+		rtcpPort := 9001 + i*2
+		rtcpl, err := net.ListenPacket("udp", ":"+strconv.FormatInt(int64(rtcpPort), 10))
+		if err != nil {
+			panic(err)
+		}
+		rtcpListeners = append(rtcpListeners, rtcpl)
+
+		_, _, _, err = rconn.SetupUdp(u, media, rtpPort, rtcpPort)
+		if err != nil {
+			panic(err)
+		}
+	}
+
+	_, err = rconn.Play(u)
+	if err != nil {
+		panic(err)
+	}
+
+	for trackId, l := range rtpListeners {
+		go func(trackId int, l net.PacketConn) {
+			buf := make([]byte, 2048)
+			for {
+				n, _, err := l.ReadFrom(buf)
+				if err != nil {
+					break
+				}
+
+				fmt.Printf("packet from track %d, type RTP: %v\n", trackId, buf[:n])
+			}
+		}(trackId, l)
+	}
+
+	for trackId, l := range rtcpListeners {
+		go func(trackId int, l net.PacketConn) {
+			buf := make([]byte, 2048)
+			for {
+				n, _, err := l.ReadFrom(buf)
+				if err != nil {
+					break
+				}
+
+				fmt.Printf("packet from track %d, type RTCP: %v\n", trackId, buf[:n])
+			}
+		}(trackId, l)
+	}
+
+	select {}
+}