Juliusz Chroboczek
329 lines
9.0 KiB
Go
329 lines
9.0 KiB
Go
package sdpfrag
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/pion/sdp/v3"
|
|
)
|
|
|
|
var sdpOffer = `v=0
|
|
o=- 999478090166020379 2 IN IP4 127.0.0.1
|
|
s=-
|
|
t=0 0
|
|
a=group:BUNDLE 0 1
|
|
a=extmap-allow-mixed
|
|
a=msid-semantic: WMS ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe
|
|
m=audio 9 UDP/TLS/RTP/SAVPF 111 63 9 0 8 13 110 126
|
|
c=IN IP4 0.0.0.0
|
|
a=rtcp:9 IN IP4 0.0.0.0
|
|
a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
a=ice-options:trickle
|
|
a=fingerprint:sha-256 EF:FE:1C:DA:83:C0:AF:B3:12:31:42:32:A4:37:04:5A:BE:7A:8D:BA:9D:0B:F2:A0:81:17:51:60:F4:96:11:5D
|
|
a=setup:actpass
|
|
a=mid:0
|
|
a=extmap:1 urn:ietf:params:rtp-hdrext:ssrc-audio-level
|
|
a=extmap:2 http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time
|
|
a=extmap:3 http://www.ietf.org/id/draft-holmer-rmcat-transport-wide-cc-extensions-01
|
|
a=extmap:4 urn:ietf:params:rtp-hdrext:sdes:mid
|
|
a=sendrecv
|
|
a=msid:ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe d71aa4c4-6f2a-480a-9c83-75e95ef1519c
|
|
a=rtcp-mux
|
|
a=rtcp-rsize
|
|
a=rtpmap:111 opus/48000/2
|
|
a=rtcp-fb:111 transport-cc
|
|
a=fmtp:111 minptime=10;useinbandfec=1
|
|
a=rtpmap:63 red/48000/2
|
|
a=fmtp:63 111/111
|
|
a=rtpmap:9 G722/8000
|
|
a=rtpmap:0 PCMU/8000
|
|
a=rtpmap:8 PCMA/8000
|
|
a=rtpmap:13 CN/8000
|
|
a=rtpmap:110 telephone-event/48000
|
|
a=rtpmap:126 telephone-event/8000
|
|
a=ssrc:870576534 cname:VkgQMWvIx9VqAt+A
|
|
a=ssrc:870576534 msid:ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe d71aa4c4-6f2a-480a-9c83-75e95ef1519c
|
|
m=video 9 UDP/TLS/RTP/SAVPF 96 97 103 104 107 108 109 114 115 116 117 118 39 40 45 46 98 99 100 101 119 120 121
|
|
c=IN IP4 0.0.0.0
|
|
a=rtcp:9 IN IP4 0.0.0.0
|
|
a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
a=ice-options:trickle
|
|
a=fingerprint:sha-256 EF:FE:1C:DA:83:C0:AF:B3:12:31:42:32:A4:37:04:5A:BE:7A:8D:BA:9D:0B:F2:A0:81:17:51:60:F4:96:11:5D
|
|
a=setup:actpass
|
|
a=mid:1
|
|
a=extmap:14 urn:ietf:params:rtp-hdrext:toffset
|
|
a=extmap:2 http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time
|
|
a=extmap:13 urn:3gpp:video-orientation
|
|
a=extmap:3 http://www.ietf.org/id/draft-holmer-rmcat-transport-wide-cc-extensions-01
|
|
a=extmap:5 http://www.webrtc.org/experiments/rtp-hdrext/playout-delay
|
|
a=extmap:6 http://www.webrtc.org/experiments/rtp-hdrext/video-content-type
|
|
a=extmap:7 http://www.webrtc.org/experiments/rtp-hdrext/video-timing
|
|
a=extmap:8 http://www.webrtc.org/experiments/rtp-hdrext/color-space
|
|
a=extmap:4 urn:ietf:params:rtp-hdrext:sdes:mid
|
|
a=extmap:10 urn:ietf:params:rtp-hdrext:sdes:rtp-stream-id
|
|
a=extmap:11 urn:ietf:params:rtp-hdrext:sdes:repaired-rtp-stream-id
|
|
a=sendrecv
|
|
a=msid:ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe 4fada03b-35ae-435d-b08e-001b6ce8362f
|
|
a=rtcp-mux
|
|
a=rtcp-rsize
|
|
a=rtpmap:96 VP8/90000
|
|
a=rtcp-fb:96 goog-remb
|
|
a=rtcp-fb:96 transport-cc
|
|
a=rtcp-fb:96 ccm fir
|
|
a=rtcp-fb:96 nack
|
|
a=rtcp-fb:96 nack pli
|
|
a=rtpmap:97 rtx/90000
|
|
a=fmtp:97 apt=96
|
|
a=rtpmap:103 H264/90000
|
|
a=rtcp-fb:103 goog-remb
|
|
a=rtcp-fb:103 transport-cc
|
|
a=rtcp-fb:103 ccm fir
|
|
a=rtcp-fb:103 nack
|
|
a=rtcp-fb:103 nack pli
|
|
a=fmtp:103 level-asymmetry-allowed=1;packetization-mode=1;profile-level-id=42001f
|
|
a=rtpmap:104 rtx/90000
|
|
a=fmtp:104 apt=103
|
|
a=rtpmap:107 H264/90000
|
|
a=rtcp-fb:107 goog-remb
|
|
a=rtcp-fb:107 transport-cc
|
|
a=rtcp-fb:107 ccm fir
|
|
a=rtcp-fb:107 nack
|
|
a=rtcp-fb:107 nack pli
|
|
a=fmtp:107 level-asymmetry-allowed=1;packetization-mode=0;profile-level-id=42001f
|
|
a=rtpmap:108 rtx/90000
|
|
a=fmtp:108 apt=107
|
|
a=rtpmap:109 H264/90000
|
|
a=rtcp-fb:109 goog-remb
|
|
a=rtcp-fb:109 transport-cc
|
|
a=rtcp-fb:109 ccm fir
|
|
a=rtcp-fb:109 nack
|
|
a=rtcp-fb:109 nack pli
|
|
a=fmtp:109 level-asymmetry-allowed=1;packetization-mode=1;profile-level-id=42e01f
|
|
a=rtpmap:114 rtx/90000
|
|
a=fmtp:114 apt=109
|
|
a=rtpmap:115 H264/90000
|
|
a=rtcp-fb:115 goog-remb
|
|
a=rtcp-fb:115 transport-cc
|
|
a=rtcp-fb:115 ccm fir
|
|
a=rtcp-fb:115 nack
|
|
a=rtcp-fb:115 nack pli
|
|
a=fmtp:115 level-asymmetry-allowed=1;packetization-mode=0;profile-level-id=42e01f
|
|
a=rtpmap:116 rtx/90000
|
|
a=fmtp:116 apt=115
|
|
a=rtpmap:117 H264/90000
|
|
a=rtcp-fb:117 goog-remb
|
|
a=rtcp-fb:117 transport-cc
|
|
a=rtcp-fb:117 ccm fir
|
|
a=rtcp-fb:117 nack
|
|
a=rtcp-fb:117 nack pli
|
|
a=fmtp:117 level-asymmetry-allowed=1;packetization-mode=1;profile-level-id=4d001f
|
|
a=rtpmap:118 rtx/90000
|
|
a=fmtp:118 apt=117
|
|
a=rtpmap:39 H264/90000
|
|
a=rtcp-fb:39 goog-remb
|
|
a=rtcp-fb:39 transport-cc
|
|
a=rtcp-fb:39 ccm fir
|
|
a=rtcp-fb:39 nack
|
|
a=rtcp-fb:39 nack pli
|
|
a=fmtp:39 level-asymmetry-allowed=1;packetization-mode=0;profile-level-id=4d001f
|
|
a=rtpmap:40 rtx/90000
|
|
a=fmtp:40 apt=39
|
|
a=rtpmap:45 AV1/90000
|
|
a=rtcp-fb:45 goog-remb
|
|
a=rtcp-fb:45 transport-cc
|
|
a=rtcp-fb:45 ccm fir
|
|
a=rtcp-fb:45 nack
|
|
a=rtcp-fb:45 nack pli
|
|
a=fmtp:45 level-idx=5;profile=0;tier=0
|
|
a=rtpmap:46 rtx/90000
|
|
a=fmtp:46 apt=45
|
|
a=rtpmap:98 VP9/90000
|
|
a=rtcp-fb:98 goog-remb
|
|
a=rtcp-fb:98 transport-cc
|
|
a=rtcp-fb:98 ccm fir
|
|
a=rtcp-fb:98 nack
|
|
a=rtcp-fb:98 nack pli
|
|
a=fmtp:98 profile-id=0
|
|
a=rtpmap:99 rtx/90000
|
|
a=fmtp:99 apt=98
|
|
a=rtpmap:100 VP9/90000
|
|
a=rtcp-fb:100 goog-remb
|
|
a=rtcp-fb:100 transport-cc
|
|
a=rtcp-fb:100 ccm fir
|
|
a=rtcp-fb:100 nack
|
|
a=rtcp-fb:100 nack pli
|
|
a=fmtp:100 profile-id=2
|
|
a=rtpmap:101 rtx/90000
|
|
a=fmtp:101 apt=100
|
|
a=rtpmap:119 red/90000
|
|
a=rtpmap:120 rtx/90000
|
|
a=fmtp:120 apt=119
|
|
a=rtpmap:121 ulpfec/90000
|
|
a=ssrc-group:FID 18877847 3714356054
|
|
a=ssrc:18877847 cname:VkgQMWvIx9VqAt+A
|
|
a=ssrc:18877847 msid:ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe 4fada03b-35ae-435d-b08e-001b6ce8362f
|
|
a=ssrc:3714356054 cname:VkgQMWvIx9VqAt+A
|
|
a=ssrc:3714356054 msid:ef5d4db1-5c64-4b6e-9f97-8b1f3e598bfe 4fada03b-35ae-435d-b08e-001b6ce8362f
|
|
`
|
|
|
|
var sdpfragTrickle = `a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
m=audio 9 UDP/TLS/RTP/SAVPF 0
|
|
a=mid:0
|
|
a=candidate:1937612317 1 udp 2113937151 40a192af-d71f-4cf2-b5b4-494a95dcf739.local 44501 typ host generation 0 ufrag qiKa network-cost 999
|
|
a=candidate:670660484 1 udp 2113939711 47da31ec-00cb-4c68-bc90-c403b203d8d8.local 45749 typ host generation 0 ufrag qiKa network-cost 999
|
|
`
|
|
|
|
var sdpfragEndOfCandidates = `a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
m=audio 9 UDP/TLS/RTP/SAVPF 0
|
|
a=mid:0
|
|
a=end-of-candidates
|
|
`
|
|
|
|
var sdpfragRestart = `a=ice-ufrag:HWmk
|
|
a=ice-pwd:6kyIjoI1lVhYJUeo1EyUq0Ei
|
|
`
|
|
|
|
func TestParseSDPFrag(t *testing.T) {
|
|
var frag SDPFrag
|
|
err := frag.Unmarshal([]byte(sdpfragTrickle))
|
|
if err != nil {
|
|
t.Errorf("Unmarshal: %v", err)
|
|
}
|
|
if frag.UsernameFragment != "qiKa" ||
|
|
frag.Password != "bcfs93hb/+ZLuUE2K50HVbkr" {
|
|
t.Errorf("Ufrag %v, pwd %v", frag.UsernameFragment, frag.Password)
|
|
}
|
|
if len(frag.Candidates) != 0 {
|
|
t.Errorf("Expected 0, got %v", len(frag.Candidates))
|
|
}
|
|
if len(frag.MediaDescriptions) != 1 {
|
|
t.Errorf("Expected 1, got %v", len(frag.MediaDescriptions))
|
|
}
|
|
if len(frag.MediaDescriptions[0].Candidates) != 2 {
|
|
t.Errorf("Expected 1, got %v", len(frag.MediaDescriptions[0].Candidates))
|
|
}
|
|
c := frag.MediaDescriptions[0].Candidates[0]
|
|
if *c.SDPMLineIndex != 0 ||
|
|
*c.SDPMid != "0" ||
|
|
*c.UsernameFragment != "qiKa" {
|
|
t.Errorf("Got %v", c)
|
|
}
|
|
}
|
|
|
|
func testRoundtrip(t *testing.T, sdpfrag string) {
|
|
var frag SDPFrag
|
|
err := frag.Unmarshal([]byte(sdpfrag))
|
|
if err != nil {
|
|
t.Errorf("Unmarshal: %v", err)
|
|
}
|
|
|
|
sdpfrag2, err := frag.Marshal()
|
|
if err != nil {
|
|
t.Errorf("Marshal: %v", err)
|
|
}
|
|
|
|
var frag2 SDPFrag
|
|
err = frag2.Unmarshal(sdpfrag2)
|
|
if err != nil {
|
|
t.Errorf("Unmarshal: %v", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(frag, frag2) {
|
|
t.Errorf("Not equal: %v %v", frag, frag2)
|
|
}
|
|
}
|
|
|
|
func TestRoundtrip(t *testing.T) {
|
|
type test struct{ name, frag string }
|
|
for _, tt := range []test{
|
|
{"sdpfragTrickle", sdpfragTrickle},
|
|
{"sdpfragEndOfCandidates", sdpfragEndOfCandidates},
|
|
{"sdpfragRestart", sdpfragRestart},
|
|
} {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
testRoundtrip(t, tt.frag)
|
|
})
|
|
}
|
|
}
|
|
|
|
func testPatchSDP(t *testing.T, sdpOffer, sdpFrag, ufrag string, ncandidates int) {
|
|
var s sdp.SessionDescription
|
|
err := s.Unmarshal([]byte(sdpOffer))
|
|
if err != nil {
|
|
t.Fatalf("Unmarshal SDP: %v", err)
|
|
}
|
|
|
|
var f SDPFrag
|
|
err = f.Unmarshal([]byte(sdpFrag))
|
|
if err != nil {
|
|
t.Fatalf("Unmarshal frag: %v", err)
|
|
}
|
|
|
|
ss, _ := PatchSDP(s, f)
|
|
u, _ := ss.MediaDescriptions[0].Attribute("ice-ufrag")
|
|
if u != ufrag {
|
|
t.Errorf("UFrag mismatch: %v and %v", u, ufrag)
|
|
}
|
|
|
|
nc := 0
|
|
for _, a := range ss.Attributes {
|
|
if a.Key == "candidate" {
|
|
nc++
|
|
}
|
|
}
|
|
|
|
for _, m := range ss.MediaDescriptions {
|
|
for _, a := range m.Attributes {
|
|
if a.Key == "candidate" {
|
|
nc++
|
|
}
|
|
}
|
|
}
|
|
if nc != ncandidates {
|
|
t.Errorf("Candidate mismatch: got %v, expected %v",
|
|
nc, ncandidates)
|
|
}
|
|
}
|
|
|
|
func TestPatchSDP(t *testing.T) {
|
|
t.Run("trickle", func(t *testing.T) {
|
|
testPatchSDP(t, sdpOffer, sdpfragTrickle, "qiKa", 2)
|
|
})
|
|
t.Run("end-of-candidates", func(t *testing.T) {
|
|
testPatchSDP(t, sdpOffer, sdpfragEndOfCandidates, "qiKa", 0)
|
|
})
|
|
t.Run("restart", func(t *testing.T) {
|
|
testPatchSDP(t, sdpOffer, sdpfragRestart, "HWmk", 0)
|
|
})
|
|
}
|
|
|
|
func TestFromSDP(t *testing.T) {
|
|
frag := `m=audio 9 UDP/TLS/RTP/SAVPF 111 63 9 0 8 13 110 126
|
|
a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
a=mid:0
|
|
m=video 9 UDP/TLS/RTP/SAVPF 96 97 103 104 107 108 109 114 115 116 117 118 39 40 45 46 98 99 100 101 119 120 121
|
|
a=ice-ufrag:qiKa
|
|
a=ice-pwd:bcfs93hb/+ZLuUE2K50HVbkr
|
|
a=mid:1
|
|
`
|
|
var f SDPFrag
|
|
err := f.Unmarshal([]byte(frag))
|
|
if err != nil {
|
|
t.Fatalf("Unmarshal: %v", err)
|
|
}
|
|
var s sdp.SessionDescription
|
|
err = s.Unmarshal([]byte(sdpOffer))
|
|
if err != nil {
|
|
t.Fatalf("Unmarshal: %v", err)
|
|
}
|
|
ff := FromSDP(s)
|
|
if !reflect.DeepEqual(ff, f) {
|
|
t.Errorf("Not equal: %v %v", ff, f)
|
|
}
|
|
}
|