package/gortsplib
1
0
Fork 0
mirror of https://github.com/aler9/gortsplib synced 2025-10-05 15:16:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

server: decode and encode RTP/RTCP packets

Browse Source
This commit is contained in:
aler9
2021-12-21 23:26:58 +01:00
committed by Alessandro Ros
parent b7df36d4ad
commit 3a2f9adbd5
12 changed files with 197 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
server_read_test.go
View File

@@ -289,8 +289,8 @@ func TestServerRead(t *testing.T) {
onPlay: func(ctx *ServerHandlerOnPlayCtx) (*base.Response, error) {
go func() {
time.Sleep(1 * time.Second)
stream.WritePacketRTP(0, []byte{0x01, 0x02, 0x03, 0x04})
stream.WritePacketRTCP(0, []byte{0x05, 0x06, 0x07, 0x08})
stream.WritePacketRTP(0, &testRTPPacket)
stream.WritePacketRTCP(0, &testRTCPPacket)
}()
return &base.Response{
@@ -304,7 +304,7 @@ func TestServerRead(t *testing.T) {
}
require.Equal(t, 0, ctx.TrackID)
require.Equal(t, []byte{0x01, 0x02, 0x03, 0x04}, ctx.Payload)
require.Equal(t, &testRTCPPacket, ctx.Packet)
close(framesReceived)
},
onGetParameter: func(ctx *ServerHandlerOnGetParameterCtx) (*base.Response, error) {
@@ -466,7 +466,7 @@ func TestServerRead(t *testing.T) {
buf := make([]byte, 2048)
n, _, err := l1.ReadFrom(buf)
require.NoError(t, err)
require.Equal(t, []byte{0x01, 0x02, 0x03, 0x04}, buf[:n])
require.Equal(t, testRTPPacketMarshaled, buf[:n])
// skip firewall opening
if transport == "udp" {
@@ -478,33 +478,33 @@ func TestServerRead(t *testing.T) {
buf = make([]byte, 2048)
n, _, err = l2.ReadFrom(buf)
require.NoError(t, err)
require.Equal(t, []byte{0x05, 0x06, 0x07, 0x08}, buf[:n])
require.Equal(t, testRTCPPacketMarshaled, buf[:n])
} else {
var f base.InterleavedFrame
f.Payload = make([]byte, 2048)
err := f.Read(br)
require.NoError(t, err)
require.Equal(t, 4, f.Channel)
require.Equal(t, []byte{0x01, 0x02, 0x03, 0x04}, f.Payload)
require.Equal(t, testRTPPacketMarshaled, f.Payload)
f.Payload = make([]byte, 2048)
err = f.Read(br)
require.NoError(t, err)
require.Equal(t, 5, f.Channel)
require.Equal(t, []byte{0x05, 0x06, 0x07, 0x08}, f.Payload)
require.Equal(t, testRTCPPacketMarshaled, f.Payload)
}
// client -> server (RTCP)
switch transport {
case "udp":
l2.WriteTo([]byte{0x01, 0x02, 0x03, 0x04}, &net.UDPAddr{
l2.WriteTo(testRTCPPacketMarshaled, &net.UDPAddr{
IP: net.ParseIP("127.0.0.1"),
Port: th.ServerPorts[1],
})
<-framesReceived
case "multicast":
l2.WriteTo([]byte{0x01, 0x02, 0x03, 0x04}, &net.UDPAddr{
l2.WriteTo(testRTCPPacketMarshaled, &net.UDPAddr{
IP: *th.Destination,
Port: th.Ports[1],
})
@@ -513,7 +513,7 @@ func TestServerRead(t *testing.T) {
default:
base.InterleavedFrame{
Channel: 5,
Payload: []byte{0x01, 0x02, 0x03, 0x04},
Payload: testRTCPPacketMarshaled,
}.Write(&bb)
_, err = conn.Write(bb.Bytes())
require.NoError(t, err)
@@ -614,13 +614,21 @@ func TestServerReadVLCMulticast(t *testing.T) {
}
func TestServerReadNonStandardFrameSize(t *testing.T) {
packet := rtp.Packet{
Header: rtp.Header{
Version: 2,
PayloadType: 97,
CSRC: []uint32{},
},
Payload: bytes.Repeat([]byte{0x01, 0x02, 0x03, 0x04, 0x05}, 4096/5),
}
packetMarshaled, _ := packet.Marshal()
track, err := NewTrackH264(96, []byte{0x01, 0x02, 0x03, 0x04}, []byte{0x01, 0x02, 0x03, 0x04}, nil)
require.NoError(t, err)
stream := NewServerStream(Tracks{track})
payload := bytes.Repeat([]byte{0x01, 0x02, 0x03, 0x04, 0x05}, 4096/5)
s := &Server{
Handler: &testServerHandler{
onSetup: func(ctx *ServerHandlerOnSetupCtx) (*base.Response, *ServerStream, error) {
@@ -631,7 +639,7 @@ func TestServerReadNonStandardFrameSize(t *testing.T) {
onPlay: func(ctx *ServerHandlerOnPlayCtx) (*base.Response, error) {
go func() {
time.Sleep(1 * time.Second)
stream.WritePacketRTP(0, payload)
stream.WritePacketRTP(0, &packet)
}()
return &base.Response{
@@ -694,7 +702,7 @@ func TestServerReadNonStandardFrameSize(t *testing.T) {
err = f.Read(br)
require.NoError(t, err)
require.Equal(t, 0, f.Channel)
require.Equal(t, payload, f.Payload)
require.Equal(t, packetMarshaled, f.Payload)
}
func TestServerReadTCPResponseBeforeFrames(t *testing.T) {
@@ -722,7 +730,7 @@ func TestServerReadTCPResponseBeforeFrames(t *testing.T) {
go func() {
defer close(writerDone)
stream.WritePacketRTP(0, []byte("\x00\x00\x00\x00"))
stream.WritePacketRTP(0, &testRTPPacket)
t := time.NewTicker(50 * time.Millisecond)
defer t.Stop()
@@ -730,7 +738,7 @@ func TestServerReadTCPResponseBeforeFrames(t *testing.T) {
for {
select {
case <-t.C:
stream.WritePacketRTP(0, []byte("\x00\x00\x00\x00"))
stream.WritePacketRTP(0, &testRTPPacket)
case <-writerTerminate:
return
}
@@ -913,7 +921,7 @@ func TestServerReadPlayPausePlay(t *testing.T) {
for {
select {
case <-t.C:
stream.WritePacketRTP(0, []byte("\x00\x00\x00\x00"))
stream.WritePacketRTP(0, &testRTPPacket)
case <-writerTerminate:
return
}
@@ -1033,7 +1041,7 @@ func TestServerReadPlayPausePause(t *testing.T) {
for {
select {
case <-t.C:
stream.WritePacketRTP(0, []byte("\x00\x00\x00\x00"))
stream.WritePacketRTP(0, &testRTPPacket)
case <-writerTerminate:
return
}
@@ -1457,8 +1465,8 @@ func TestServerReadPartialTracks(t *testing.T) {
onPlay: func(ctx *ServerHandlerOnPlayCtx) (*base.Response, error) {
go func() {
time.Sleep(1 * time.Second)
stream.WritePacketRTP(0, []byte{0x01, 0x02, 0x03, 0x04})
stream.WritePacketRTP(1, []byte{0x05, 0x06, 0x07, 0x08})
stream.WritePacketRTP(0, &testRTPPacket)
stream.WritePacketRTP(1, &testRTPPacket)
}()
return &base.Response{
@@ -1522,7 +1530,7 @@ func TestServerReadPartialTracks(t *testing.T) {
err = f.Read(br)
require.NoError(t, err)
require.Equal(t, 4, f.Channel)
require.Equal(t, []byte{0x05, 0x06, 0x07, 0x08}, f.Payload)
require.Equal(t, testRTPPacketMarshaled, f.Payload)
}
func TestServerReadAdditionalInfos(t *testing.T) {
@@ -1630,8 +1638,8 @@ func TestServerReadAdditionalInfos(t *testing.T) {
onPlay: func(ctx *ServerHandlerOnPlayCtx) (*base.Response, error) {
go func() {
time.Sleep(1 * time.Second)
stream.WritePacketRTP(1, []byte{0x01, 0x02, 0x03, 0x04})
stream.WritePacketRTP(0, []byte{0x05, 0x06, 0x07, 0x08})
stream.WritePacketRTP(1, &testRTPPacket)
stream.WritePacketRTP(0, &testRTPPacket)
}()
return &base.Response{
@@ -1646,7 +1654,7 @@ func TestServerReadAdditionalInfos(t *testing.T) {
require.NoError(t, err)
defer s.Close()
buf, err := (&rtp.Packet{
stream.WritePacketRTP(0, &rtp.Packet{
Header: rtp.Header{
Version: 2,
PayloadType: 96,
@@ -1655,9 +1663,7 @@ func TestServerReadAdditionalInfos(t *testing.T) {
SSRC: 96342362,
},
Payload: []byte{0x01, 0x02, 0x03, 0x04},
}).Marshal()
require.NoError(t, err)
stream.WritePacketRTP(0, buf)
})
rtpInfo, ssrcs := getInfos()
require.Equal(t, &headers.RTPInfo{
@@ -1682,7 +1688,7 @@ func TestServerReadAdditionalInfos(t *testing.T) {
nil,
}, ssrcs)
buf, err = (&rtp.Packet{
stream.WritePacketRTP(1, &rtp.Packet{
Header: rtp.Header{
Version: 2,
PayloadType: 96,
@@ -1691,9 +1697,7 @@ func TestServerReadAdditionalInfos(t *testing.T) {
SSRC: 536474323,
},
Payload: []byte{0x01, 0x02, 0x03, 0x04},
}).Marshal()
require.NoError(t, err)
stream.WritePacketRTP(1, buf)
})
rtpInfo, ssrcs = getInfos()
require.Equal(t, &headers.RTPInfo{