mirror of https://bitbucket.org/ausocean/av.git
102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
|
package packet
|
||
|
|
|
||
|
|
import (
|
||
|
|
//"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"log"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/beatgammit/rtsp"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
rtpPort = 17300
|
||
|
|
rtcpPort = 17319
|
||
|
|
rtspUrl = "rtsp://192.168.0.50:8554/CH002.sdp"
|
||
|
|
rtpUrl = "rtsp://192.168.0.50:8554/CH002.sdp/track1"
|
||
|
|
)
|
||
|
|
|
||
|
|
/* Let's see if we can connect to an rtsp device then read an rtp stream,
|
||
|
|
and then convert the rtp packets to mpegts packets and output. */
|
||
|
|
func TestMain(t *testing.T) {
|
||
|
|
sess := rtsp.NewSession()
|
||
|
|
res, err := sess.Options(rtspUrl)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Shouldn't have got error: %v\n",err)
|
||
|
|
}
|
||
|
|
fmt.Println("Options:")
|
||
|
|
fmt.Println(res)
|
||
|
|
|
||
|
|
res, err = sess.Describe(rtspUrl)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalln(err)
|
||
|
|
t.Errorf("Shouldn't have got error: %v\n", err)
|
||
|
|
}
|
||
|
|
fmt.Println("Describe:")
|
||
|
|
fmt.Println(res)
|
||
|
|
|
||
|
|
p, err := rtsp.ParseSdp(&io.LimitedReader{R: res.Body, N: res.ContentLength})
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Shouldn't have got error: %v\n", err)
|
||
|
|
}
|
||
|
|
log.Printf("%+v", p)
|
||
|
|
|
||
|
|
fmt.Println("Setting up!")
|
||
|
|
res, err = sess.Setup(rtpUrl, fmt.Sprintf("RTP/AVP;unicast;client_port=%d-%d", rtpPort, rtcpPort))
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Shouldn't have got error: %v\n", err)
|
||
|
|
}
|
||
|
|
log.Println(res)
|
||
|
|
|
||
|
|
fmt.Println("Playing !")
|
||
|
|
res, err = sess.Play(rtspUrl, res.Header.Get("Session"))
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Shouldn't have got error: %v\n", err)
|
||
|
|
}
|
||
|
|
log.Println(res)
|
||
|
|
|
||
|
|
// create udp connection for rtp stuff
|
||
|
|
rtpLaddr, err := net.ResolveUDPAddr("udp", "192.168.0.109:17300")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Local rtp addr not set! %v\n",err)
|
||
|
|
}
|
||
|
|
rtpAddr, err := net.ResolveUDPAddr("udp", "192.168.0.50:17300")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Resolving rtp address didn't work! %v\n", err)
|
||
|
|
}
|
||
|
|
rtpConn, err := net.DialUDP("udp", rtpLaddr, rtpAddr)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Conncection not established! %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Create udp connection for rtcp stuff
|
||
|
|
rtcpLaddr, err := net.ResolveUDPAddr("udp", "192.168.0.109:17319")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Local RTCP address not resolved! %v\n", err)
|
||
|
|
}
|
||
|
|
rtcpAddr, err := net.ResolveUDPAddr("udp", "192.168.0.50:17301")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Remote RTCP address not resolved! %v\n", err)
|
||
|
|
}
|
||
|
|
rtcpConn, err := net.DialUDP("udp", rtcpLaddr, rtcpAddr)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Connection not established! %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// let's create a session that will store useful stuff from the connections
|
||
|
|
rtpSession := NewSession(rtpConn, rtcpConn)
|
||
|
|
converter := NewRtpToTsConverter()
|
||
|
|
go converter.Convert()
|
||
|
|
for ii:=0;ii<10000000;ii++{
|
||
|
|
select{
|
||
|
|
default:
|
||
|
|
case rtpPacket := <-rtpSession.RtpChan:
|
||
|
|
converter.RtpChan<-rtpPacket
|
||
|
|
case tsPacket:=<-converter.TsChan:
|
||
|
|
fmt.Println(tsPacket.ToByteSlice())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|