mirror of
https://github.com/aler9/gortsplib
synced 2025-10-03 22:36:31 +08:00
f3b0fc69b4
Read() / Write() are used to read / write from streams, while Unmarshal() / Marshal() are used to decode / encode from / to bytes.
276 lines
6.5 KiB
Go
276 lines
6.5 KiB
Go
package base
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/aler9/gortsplib/pkg/url"
|
|
)
|
|
|
|
func mustParseURL(s string) *url.URL {
|
|
u, err := url.Parse(s)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return u
|
|
}
|
|
|
|
var casesRequest = []struct {
|
|
name string
|
|
byts []byte
|
|
req Request
|
|
}{
|
|
{
|
|
"options",
|
|
[]byte("OPTIONS rtsp://example.com/media.mp4 RTSP/1.0\r\n" +
|
|
"CSeq: 1\r\n" +
|
|
"Proxy-Require: gzipped-messages\r\n" +
|
|
"Require: implicit-play\r\n" +
|
|
"\r\n"),
|
|
Request{
|
|
Method: "OPTIONS",
|
|
URL: mustParseURL("rtsp://example.com/media.mp4"),
|
|
Header: Header{
|
|
"CSeq": HeaderValue{"1"},
|
|
"Require": HeaderValue{"implicit-play"},
|
|
"Proxy-Require": HeaderValue{"gzipped-messages"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"describe",
|
|
[]byte("DESCRIBE rtsp://example.com/media.mp4 RTSP/1.0\r\n" +
|
|
"Accept: application/sdp\r\n" +
|
|
"CSeq: 2\r\n" +
|
|
"\r\n"),
|
|
Request{
|
|
Method: "DESCRIBE",
|
|
URL: mustParseURL("rtsp://example.com/media.mp4"),
|
|
Header: Header{
|
|
"Accept": HeaderValue{"application/sdp"},
|
|
"CSeq": HeaderValue{"2"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"describe with special chars",
|
|
[]byte("DESCRIBE rtsp://192.168.1.99:554/user=tmp&password=BagRep1!&channel=1&stream=0.sdp RTSP/1.0\r\n" +
|
|
"Accept: application/sdp\r\n" +
|
|
"CSeq: 3\r\n" +
|
|
"\r\n"),
|
|
Request{
|
|
Method: "DESCRIBE",
|
|
URL: mustParseURL("rtsp://192.168.1.99:554/user=tmp&password=BagRep1!&channel=1&stream=0.sdp"),
|
|
Header: Header{
|
|
"Accept": HeaderValue{"application/sdp"},
|
|
"CSeq": HeaderValue{"3"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
"announce",
|
|
[]byte("ANNOUNCE rtsp://example.com/media.mp4 RTSP/1.0\r\n" +
|
|
"CSeq: 7\r\n" +
|
|
"Content-Length: 306\r\n" +
|
|
"Content-Type: application/sdp\r\n" +
|
|
"Date: 23 Jan 1997 15:35:06 GMT\r\n" +
|
|
"Session: 12345678\r\n" +
|
|
"\r\n" +
|
|
"v=0\n" +
|
|
"o=mhandley 2890844526 2890845468 IN IP4 126.16.64.4\n" +
|
|
"s=SDP Seminar\n" +
|
|
"i=A Seminar on the session description protocol\n" +
|
|
"u=http://www.cs.ucl.ac.uk/staff/M.Handley/sdp.03.ps\n" +
|
|
"e=mjh@isi.edu (Mark Handley)\n" +
|
|
"c=IN IP4 224.2.17.12/127\n" +
|
|
"t=2873397496 2873404696\n" +
|
|
"a=recvonly\n" +
|
|
"m=audio 3456 RTP/AVP 0\n" +
|
|
"m=video 2232 RTP/AVP 31\n"),
|
|
Request{
|
|
Method: "ANNOUNCE",
|
|
URL: mustParseURL("rtsp://example.com/media.mp4"),
|
|
Header: Header{
|
|
"CSeq": HeaderValue{"7"},
|
|
"Date": HeaderValue{"23 Jan 1997 15:35:06 GMT"},
|
|
"Session": HeaderValue{"12345678"},
|
|
"Content-Type": HeaderValue{"application/sdp"},
|
|
"Content-Length": HeaderValue{"306"},
|
|
},
|
|
Body: []byte("v=0\n" +
|
|
"o=mhandley 2890844526 2890845468 IN IP4 126.16.64.4\n" +
|
|
"s=SDP Seminar\n" +
|
|
"i=A Seminar on the session description protocol\n" +
|
|
"u=http://www.cs.ucl.ac.uk/staff/M.Handley/sdp.03.ps\n" +
|
|
"e=mjh@isi.edu (Mark Handley)\n" +
|
|
"c=IN IP4 224.2.17.12/127\n" +
|
|
"t=2873397496 2873404696\n" +
|
|
"a=recvonly\n" +
|
|
"m=audio 3456 RTP/AVP 0\n" +
|
|
"m=video 2232 RTP/AVP 31\n",
|
|
),
|
|
},
|
|
},
|
|
{
|
|
"get_parameter",
|
|
[]byte("GET_PARAMETER rtsp://example.com/media.mp4 RTSP/1.0\r\n" +
|
|
"CSeq: 9\r\n" +
|
|
"Content-Length: 24\r\n" +
|
|
"Content-Type: text/parameters\r\n" +
|
|
"Session: 12345678\r\n" +
|
|
"\r\n" +
|
|
"packets_received\n" +
|
|
"jitter\n"),
|
|
Request{
|
|
Method: "GET_PARAMETER",
|
|
URL: mustParseURL("rtsp://example.com/media.mp4"),
|
|
Header: Header{
|
|
"CSeq": HeaderValue{"9"},
|
|
"Content-Type": HeaderValue{"text/parameters"},
|
|
"Session": HeaderValue{"12345678"},
|
|
"Content-Length": HeaderValue{"24"},
|
|
},
|
|
Body: []byte("packets_received\n" +
|
|
"jitter\n",
|
|
),
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestRequestRead(t *testing.T) {
|
|
// keep req global to make sure that all its fields are overridden.
|
|
var req Request
|
|
|
|
for _, ca := range casesRequest {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
err := req.Read(bufio.NewReader(bytes.NewBuffer(ca.byts)))
|
|
require.NoError(t, err)
|
|
require.Equal(t, ca.req, req)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRequestReadErrors(t *testing.T) {
|
|
for _, ca := range []struct {
|
|
name string
|
|
byts []byte
|
|
err string
|
|
}{
|
|
{
|
|
"empty",
|
|
[]byte{},
|
|
"EOF",
|
|
},
|
|
{
|
|
"missing url, protocol, r, n",
|
|
[]byte("GET"),
|
|
"EOF",
|
|
},
|
|
{
|
|
"missing protocol, r, n",
|
|
[]byte("GET rtsp://testing123/test"),
|
|
"EOF",
|
|
},
|
|
{
|
|
"missing r, n",
|
|
[]byte("GET rtsp://testing123/test RTSP/1.0"),
|
|
"EOF",
|
|
},
|
|
{
|
|
"missing n",
|
|
[]byte("GET rtsp://testing123/test RTSP/1.0\r"),
|
|
"EOF",
|
|
},
|
|
{
|
|
"empty method",
|
|
[]byte(" rtsp://testing123 RTSP/1.0\r\n"),
|
|
"empty method",
|
|
},
|
|
{
|
|
"empty URL",
|
|
[]byte("GET RTSP/1.0\r\n"),
|
|
"invalid URL ()",
|
|
},
|
|
{
|
|
"empty protocol",
|
|
[]byte("GET rtsp://testing123 \r\n"),
|
|
"expected 'RTSP/1.0', got []",
|
|
},
|
|
{
|
|
"invalid URL",
|
|
[]byte("GET http://testing123 RTSP/1.0\r\n"),
|
|
"invalid URL (http://testing123)",
|
|
},
|
|
{
|
|
"invalid protocol",
|
|
[]byte("GET rtsp://testing123 RTSP/2.0\r\n"),
|
|
"expected 'RTSP/1.0', got [82 84 83 80 47 50 46 48]",
|
|
},
|
|
{
|
|
"invalid header",
|
|
[]byte("GET rtsp://testing123 RTSP/1.0\r\nTesting: val\r"),
|
|
"EOF",
|
|
},
|
|
{
|
|
"invalid body",
|
|
[]byte("GET rtsp://testing123 RTSP/1.0\r\nContent-Length: 17\r\n\r\n123"),
|
|
"unexpected EOF",
|
|
},
|
|
} {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
var req Request
|
|
err := req.Read(bufio.NewReader(bytes.NewBuffer(ca.byts)))
|
|
require.EqualError(t, err, ca.err)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRequestMarshal(t *testing.T) {
|
|
for _, ca := range casesRequest {
|
|
t.Run(ca.name, func(t *testing.T) {
|
|
buf, err := ca.req.Marshal()
|
|
require.NoError(t, err)
|
|
require.Equal(t, ca.byts, buf)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestRequestReadIgnoreFrames(t *testing.T) {
|
|
byts := []byte{0x24, 0x6, 0x0, 0x4, 0x1, 0x2, 0x3, 0x4}
|
|
byts = append(byts, []byte("OPTIONS rtsp://example.com/media.mp4 RTSP/1.0\r\n"+
|
|
"CSeq: 1\r\n"+
|
|
"Proxy-Require: gzipped-messages\r\n"+
|
|
"Require: implicit-play\r\n"+
|
|
"\r\n")...)
|
|
|
|
rb := bufio.NewReader(bytes.NewBuffer(byts))
|
|
var req Request
|
|
err := req.ReadIgnoreFrames(10, rb)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestRequestReadIgnoreFramesErrors(t *testing.T) {
|
|
byts := []byte{0x25}
|
|
|
|
rb := bufio.NewReader(bytes.NewBuffer(byts))
|
|
var req Request
|
|
err := req.ReadIgnoreFrames(10, rb)
|
|
require.EqualError(t, err, "EOF")
|
|
}
|
|
|
|
func TestRequestString(t *testing.T) {
|
|
byts := []byte("OPTIONS rtsp://example.com/media.mp4 RTSP/1.0\r\n" +
|
|
"CSeq: 1\r\n" +
|
|
"Content-Length: 7\r\n" +
|
|
"\r\n" +
|
|
"testing")
|
|
|
|
var req Request
|
|
err := req.Read(bufio.NewReader(bytes.NewBuffer(byts)))
|
|
require.NoError(t, err)
|
|
require.Equal(t, string(byts), req.String())
|
|
}
|