golanggohlsrtmpwebrtcmedia-serverobs-studiortcprtmp-proxyrtmp-serverrtprtsprtsp-proxyrtsp-relayrtsp-serversrtstreamingwebrtc-proxy
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
791 lines
19 KiB
791 lines
19 KiB
package rtmp |
|
|
|
import ( |
|
"bufio" |
|
"net" |
|
"net/url" |
|
"strings" |
|
"testing" |
|
|
|
"github.com/aler9/gortsplib" |
|
"github.com/aler9/gortsplib/pkg/aac" |
|
nh264 "github.com/notedit/rtmp/codec/h264" |
|
"github.com/notedit/rtmp/format/flv/flvio" |
|
"github.com/stretchr/testify/require" |
|
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/base" |
|
) |
|
|
|
func splitPath(u *url.URL) (app, stream string) { |
|
nu := *u |
|
nu.ForceQuery = false |
|
|
|
pathsegs := strings.Split(nu.RequestURI(), "/") |
|
if len(pathsegs) == 2 { |
|
app = pathsegs[1] |
|
} |
|
if len(pathsegs) == 3 { |
|
app = pathsegs[1] |
|
stream = pathsegs[2] |
|
} |
|
if len(pathsegs) > 3 { |
|
app = strings.Join(pathsegs[1:3], "/") |
|
stream = strings.Join(pathsegs[3:], "/") |
|
} |
|
return |
|
} |
|
|
|
func getTcURL(u string) string { |
|
ur, err := url.Parse(u) |
|
if err != nil { |
|
panic(err) |
|
} |
|
app, _ := splitPath(ur) |
|
nu := *ur |
|
nu.RawQuery = "" |
|
nu.Path = "/" |
|
return nu.String() + app |
|
} |
|
|
|
func TestReadTracks(t *testing.T) { |
|
sps := []byte{ |
|
0x67, 0x64, 0x00, 0x0c, 0xac, 0x3b, 0x50, 0xb0, |
|
0x4b, 0x42, 0x00, 0x00, 0x03, 0x00, 0x02, 0x00, |
|
0x00, 0x03, 0x00, 0x3d, 0x08, |
|
} |
|
|
|
pps := []byte{ |
|
0x68, 0xee, 0x3c, 0x80, |
|
} |
|
|
|
for _, ca := range []string{ |
|
"standard", |
|
"metadata without codec id", |
|
"no metadata", |
|
} { |
|
t.Run(ca, func(t *testing.T) { |
|
ln, err := net.Listen("tcp", "127.0.0.1:9121") |
|
require.NoError(t, err) |
|
defer ln.Close() |
|
|
|
done := make(chan struct{}) |
|
|
|
go func() { |
|
conn, err := ln.Accept() |
|
require.NoError(t, err) |
|
defer conn.Close() |
|
|
|
rconn := NewServerConn(conn) |
|
err = rconn.ServerHandshake() |
|
require.NoError(t, err) |
|
|
|
videoTrack, audioTrack, err := rconn.ReadTracks() |
|
require.NoError(t, err) |
|
|
|
switch ca { |
|
case "standard": |
|
videoTrack2, err := gortsplib.NewTrackH264(96, sps, pps, nil) |
|
require.NoError(t, err) |
|
require.Equal(t, videoTrack2, videoTrack) |
|
|
|
audioTrack2, err := gortsplib.NewTrackAAC(96, 2, 44100, 2, nil, 13, 3, 3) |
|
require.NoError(t, err) |
|
require.Equal(t, audioTrack2, audioTrack) |
|
|
|
case "metadata without codec id": |
|
videoTrack2, err := gortsplib.NewTrackH264(96, sps, pps, nil) |
|
require.NoError(t, err) |
|
require.Equal(t, videoTrack2, videoTrack) |
|
|
|
require.Equal(t, (*gortsplib.TrackAAC)(nil), audioTrack) |
|
|
|
case "no metadata": |
|
videoTrack2, err := gortsplib.NewTrackH264(96, sps, pps, nil) |
|
require.NoError(t, err) |
|
require.Equal(t, videoTrack2, videoTrack) |
|
|
|
require.Equal(t, (*gortsplib.TrackAAC)(nil), audioTrack) |
|
} |
|
|
|
close(done) |
|
}() |
|
|
|
conn, err := net.Dial("tcp", "127.0.0.1:9121") |
|
require.NoError(t, err) |
|
defer conn.Close() |
|
|
|
// C->S handshake C0 |
|
err = base.HandshakeC0{}.Write(conn) |
|
require.NoError(t, err) |
|
|
|
// C->S handshake C1 |
|
err = base.HandshakeC1{}.Write(conn) |
|
require.NoError(t, err) |
|
|
|
// S->C handshake S0 |
|
err = base.HandshakeS0{}.Read(conn) |
|
require.NoError(t, err) |
|
|
|
// S->C handshake S1+S2 |
|
s1s2 := make([]byte, 1536*2) |
|
_, err = conn.Read(s1s2) |
|
require.NoError(t, err) |
|
|
|
// C->S handshake C2 |
|
err = base.HandshakeC2{}.Write(conn, s1s2) |
|
require.NoError(t, err) |
|
|
|
mw := base.NewMessageWriter(conn) |
|
mr := base.NewMessageReader(bufio.NewReader(conn)) |
|
|
|
// C->S connect |
|
byts := flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"connect", |
|
1, |
|
flvio.AMFMap{ |
|
{K: "app", V: "/stream"}, |
|
{K: "flashVer", V: "LNX 9,0,124,2"}, |
|
{K: "tcUrl", V: getTcURL("rtmp://127.0.0.1:9121/stream")}, |
|
{K: "fpad", V: false}, |
|
{K: "capabilities", V: 15}, |
|
{K: "audioCodecs", V: 4071}, |
|
{K: "videoCodecs", V: 252}, |
|
{K: "videoFunction", V: 1}, |
|
}, |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C window acknowledgement size |
|
msg, err := mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetWindowAckSize, |
|
Body: []byte{0x00, 38, 37, 160}, |
|
}, msg) |
|
|
|
// S->C set peer bandwidth |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetPeerBandwidth, |
|
Body: []byte{0x00, 0x26, 0x25, 0xa0, 0x02}, |
|
}, msg) |
|
|
|
// S->C set chunk size |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetChunkSize, |
|
Body: []byte{0x00, 0x01, 0x00, 0x00}, |
|
}, msg) |
|
|
|
mr.SetChunkSize(65536) |
|
|
|
// S->C result |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(3), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err := flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"_result", |
|
float64(1), |
|
flvio.AMFMap{ |
|
{K: "fmsVer", V: "LNX 9,0,124,2"}, |
|
{K: "capabilities", V: float64(31)}, |
|
}, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetConnection.Connect.Success"}, |
|
{K: "description", V: "Connection succeeded."}, |
|
{K: "objectEncoding", V: float64(0)}, |
|
}, |
|
}, arr) |
|
|
|
// C->S set chunk size |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetChunkSize, |
|
Body: []byte{0x00, 0x01, 0x00, 0x00}, |
|
}) |
|
require.NoError(t, err) |
|
|
|
mw.SetChunkSize(65536) |
|
|
|
// C->S releaseStream |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"releaseStream", |
|
float64(2), |
|
nil, |
|
"", |
|
}), |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S FCPublish |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"FCPublish", |
|
float64(3), |
|
nil, |
|
"", |
|
}), |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S createStream |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"createStream", |
|
float64(4), |
|
nil, |
|
}), |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C result |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(3), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"_result", |
|
float64(4), |
|
nil, |
|
float64(1), |
|
}, arr) |
|
|
|
// C->S publish |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 8, |
|
Type: base.MessageTypeCommandAMF0, |
|
MessageStreamID: 1, |
|
Body: flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"publish", |
|
float64(5), |
|
nil, |
|
"", |
|
"live", |
|
}), |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C onStatus |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(5), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onStatus", |
|
float64(5), |
|
nil, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetStream.Publish.Start"}, |
|
{K: "description", V: "publish start"}, |
|
}, |
|
}, arr) |
|
|
|
switch ca { |
|
case "standard": |
|
// C->S metadata |
|
byts = flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"@setDataFrame", |
|
"onMetaData", |
|
flvio.AMFMap{ |
|
{ |
|
K: "videodatarate", |
|
V: float64(0), |
|
}, |
|
{ |
|
K: "videocodecid", |
|
V: float64(codecH264), |
|
}, |
|
{ |
|
K: "audiodatarate", |
|
V: float64(0), |
|
}, |
|
{ |
|
K: "audiocodecid", |
|
V: float64(codecAAC), |
|
}, |
|
}, |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 4, |
|
Type: base.MessageTypeDataAMF0, |
|
MessageStreamID: 1, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S H264 decoder config |
|
codec := nh264.Codec{ |
|
SPS: map[int][]byte{ |
|
0: sps, |
|
}, |
|
PPS: map[int][]byte{ |
|
0: pps, |
|
}, |
|
} |
|
b := make([]byte, 128) |
|
var n int |
|
codec.ToConfig(b, &n) |
|
body := append([]byte{flvio.FRAME_KEY<<4 | flvio.VIDEO_H264, 0, 0, 0, 0}, b[:n]...) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 6, |
|
Type: base.MessageTypeVideo, |
|
MessageStreamID: 1, |
|
Body: body, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S AAC decoder config |
|
enc, err := aac.MPEG4AudioConfig{ |
|
Type: 2, |
|
SampleRate: 44100, |
|
ChannelCount: 2, |
|
}.Encode() |
|
require.NoError(t, err) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 4, |
|
Type: base.MessageTypeAudio, |
|
MessageStreamID: 1, |
|
Body: append([]byte{ |
|
flvio.SOUND_AAC<<4 | flvio.SOUND_44Khz<<2 | flvio.SOUND_16BIT<<1 | flvio.SOUND_STEREO, |
|
flvio.AAC_SEQHDR, |
|
}, enc...), |
|
}) |
|
require.NoError(t, err) |
|
|
|
case "metadata without codec id": |
|
// C->S metadata |
|
byts = flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"@setDataFrame", |
|
"onMetaData", |
|
flvio.AMFMap{ |
|
{ |
|
K: "width", |
|
V: float64(2688), |
|
}, |
|
{ |
|
K: "height", |
|
V: float64(1520), |
|
}, |
|
{ |
|
K: "framerate", |
|
V: float64(0o25), |
|
}, |
|
}, |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 4, |
|
Type: base.MessageTypeDataAMF0, |
|
MessageStreamID: 1, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S H264 decoder config |
|
codec := nh264.Codec{ |
|
SPS: map[int][]byte{ |
|
0: sps, |
|
}, |
|
PPS: map[int][]byte{ |
|
0: pps, |
|
}, |
|
} |
|
b := make([]byte, 128) |
|
var n int |
|
codec.ToConfig(b, &n) |
|
body := append([]byte{flvio.FRAME_KEY<<4 | flvio.VIDEO_H264, 0, 0, 0, 0}, b[:n]...) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 6, |
|
Type: base.MessageTypeVideo, |
|
MessageStreamID: 1, |
|
Body: body, |
|
}) |
|
require.NoError(t, err) |
|
|
|
case "no metadata": |
|
// C->S H264 decoder config |
|
codec := nh264.Codec{ |
|
SPS: map[int][]byte{ |
|
0: sps, |
|
}, |
|
PPS: map[int][]byte{ |
|
0: pps, |
|
}, |
|
} |
|
b := make([]byte, 128) |
|
var n int |
|
codec.ToConfig(b, &n) |
|
body := append([]byte{flvio.FRAME_KEY<<4 | flvio.VIDEO_H264, 0, 0, 0, 0}, b[:n]...) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 6, |
|
Type: base.MessageTypeVideo, |
|
MessageStreamID: 1, |
|
Body: body, |
|
}) |
|
require.NoError(t, err) |
|
} |
|
|
|
<-done |
|
}) |
|
} |
|
} |
|
|
|
func TestWriteTracks(t *testing.T) { |
|
ln, err := net.Listen("tcp", "127.0.0.1:9121") |
|
require.NoError(t, err) |
|
defer ln.Close() |
|
|
|
go func() { |
|
conn, err := ln.Accept() |
|
require.NoError(t, err) |
|
defer conn.Close() |
|
|
|
rconn := NewServerConn(conn) |
|
err = rconn.ServerHandshake() |
|
require.NoError(t, err) |
|
|
|
videoTrack, err := gortsplib.NewTrackH264(96, |
|
[]byte{ |
|
0x67, 0x64, 0x00, 0x0c, 0xac, 0x3b, 0x50, 0xb0, |
|
0x4b, 0x42, 0x00, 0x00, 0x03, 0x00, 0x02, 0x00, |
|
0x00, 0x03, 0x00, 0x3d, 0x08, |
|
}, |
|
[]byte{ |
|
0x68, 0xee, 0x3c, 0x80, |
|
}, |
|
nil) |
|
require.NoError(t, err) |
|
|
|
audioTrack, err := gortsplib.NewTrackAAC(96, 2, 44100, 2, nil, 13, 3, 3) |
|
require.NoError(t, err) |
|
|
|
err = rconn.WriteTracks(videoTrack, audioTrack) |
|
require.NoError(t, err) |
|
}() |
|
|
|
conn, err := net.Dial("tcp", "127.0.0.1:9121") |
|
require.NoError(t, err) |
|
defer conn.Close() |
|
|
|
// C->S handshake C0 |
|
err = base.HandshakeC0{}.Write(conn) |
|
require.NoError(t, err) |
|
|
|
// C-> handshake C1 |
|
err = base.HandshakeC1{}.Write(conn) |
|
require.NoError(t, err) |
|
|
|
// S->C handshake S0 |
|
err = base.HandshakeS0{}.Read(conn) |
|
require.NoError(t, err) |
|
|
|
// S->C handshake S1+S2 |
|
s1s2 := make([]byte, 1536*2) |
|
_, err = conn.Read(s1s2) |
|
require.NoError(t, err) |
|
|
|
// C->S handshake C2 |
|
err = base.HandshakeC2{}.Write(conn, s1s2) |
|
require.NoError(t, err) |
|
|
|
mw := base.NewMessageWriter(conn) |
|
mr := base.NewMessageReader(bufio.NewReader(conn)) |
|
|
|
// C->S connect |
|
byts := flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"connect", |
|
1, |
|
flvio.AMFMap{ |
|
{K: "app", V: "/stream"}, |
|
{K: "flashVer", V: "LNX 9,0,124,2"}, |
|
{K: "tcUrl", V: getTcURL("rtmp://127.0.0.1:9121/stream")}, |
|
{K: "fpad", V: false}, |
|
{K: "capabilities", V: 15}, |
|
{K: "audioCodecs", V: 4071}, |
|
{K: "videoCodecs", V: 252}, |
|
{K: "videoFunction", V: 1}, |
|
}, |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C window acknowledgement size |
|
msg, err := mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetWindowAckSize, |
|
Body: []byte{0x00, 38, 37, 160}, |
|
}, msg) |
|
|
|
// S->C set peer bandwidth |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetPeerBandwidth, |
|
Body: []byte{0x00, 0x26, 0x25, 0xa0, 0x02}, |
|
}, msg) |
|
|
|
// S->C set chunk size |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetChunkSize, |
|
Body: []byte{0x00, 0x01, 0x00, 0x00}, |
|
}, msg) |
|
|
|
mr.SetChunkSize(65536) |
|
|
|
// S->C result |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(3), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err := flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"_result", |
|
float64(1), |
|
flvio.AMFMap{ |
|
{K: "fmsVer", V: "LNX 9,0,124,2"}, |
|
{K: "capabilities", V: float64(31)}, |
|
}, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetConnection.Connect.Success"}, |
|
{K: "description", V: "Connection succeeded."}, |
|
{K: "objectEncoding", V: float64(0)}, |
|
}, |
|
}, arr) |
|
|
|
// C->S window acknowledgement size |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetWindowAckSize, |
|
Body: []byte{0x00, 0x26, 0x25, 0xa0}, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S set chunk size |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeSetChunkSize, |
|
Body: []byte{0x00, 0x01, 0x00, 0x00}, |
|
}) |
|
require.NoError(t, err) |
|
|
|
mw.SetChunkSize(65536) |
|
|
|
// C->S createStream |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 3, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"createStream", |
|
float64(2), |
|
nil, |
|
}), |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C result |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(3), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"_result", |
|
float64(2), |
|
nil, |
|
float64(1), |
|
}, arr) |
|
|
|
// C->S getStreamLength |
|
byts = flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"getStreamLength", |
|
float64(3), |
|
nil, |
|
"", |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 8, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// C->S play |
|
byts = flvio.FillAMF0ValsMalloc([]interface{}{ |
|
"play", |
|
float64(4), |
|
nil, |
|
"", |
|
float64(-2000), |
|
}) |
|
err = mw.Write(&base.Message{ |
|
ChunkStreamID: 8, |
|
Type: base.MessageTypeCommandAMF0, |
|
Body: byts, |
|
}) |
|
require.NoError(t, err) |
|
|
|
// S->C event "stream is recorded" |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeUserControl, |
|
Body: []byte{0x00, 0x04, 0x00, 0x00, 0x00, 0x01}, |
|
}, msg) |
|
|
|
// S->C event "stream begin 1" |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, &base.Message{ |
|
ChunkStreamID: base.ControlChunkStreamID, |
|
Type: base.MessageTypeUserControl, |
|
Body: []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x01}, |
|
}, msg) |
|
|
|
// S->C onStatus |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(5), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onStatus", |
|
float64(4), |
|
nil, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetStream.Play.Reset"}, |
|
{K: "description", V: "play reset"}, |
|
}, |
|
}, arr) |
|
|
|
// S->C onStatus |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(5), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onStatus", |
|
float64(4), |
|
nil, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetStream.Play.Start"}, |
|
{K: "description", V: "play start"}, |
|
}, |
|
}, arr) |
|
|
|
// S->C onStatus |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(5), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onStatus", |
|
float64(4), |
|
nil, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetStream.Data.Start"}, |
|
{K: "description", V: "data start"}, |
|
}, |
|
}, arr) |
|
|
|
// S->C onStatus |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(5), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageTypeCommandAMF0, msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onStatus", |
|
float64(4), |
|
nil, |
|
flvio.AMFMap{ |
|
{K: "level", V: "status"}, |
|
{K: "code", V: "NetStream.Play.PublishNotify"}, |
|
{K: "description", V: "publish notify"}, |
|
}, |
|
}, arr) |
|
|
|
// S->C onMetadata |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(4), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageType(0x12), msg.Type) |
|
arr, err = flvio.ParseAMFVals(msg.Body, false) |
|
require.NoError(t, err) |
|
require.Equal(t, []interface{}{ |
|
"onMetaData", |
|
flvio.AMFMap{ |
|
{K: "videodatarate", V: float64(0)}, |
|
{K: "videocodecid", V: float64(7)}, |
|
{K: "audiodatarate", V: float64(0)}, |
|
{K: "audiocodecid", V: float64(10)}, |
|
}, |
|
}, arr) |
|
|
|
// S->C H264 decoder config |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(6), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageType(0x09), msg.Type) |
|
require.Equal(t, []byte{ |
|
0x17, 0x0, 0x0, 0x0, 0x0, 0x1, 0x64, 0x0, |
|
0xc, 0xff, 0xe1, 0x0, 0x15, 0x67, 0x64, 0x0, |
|
0xc, 0xac, 0x3b, 0x50, 0xb0, 0x4b, 0x42, 0x0, |
|
0x0, 0x3, 0x0, 0x2, 0x0, 0x0, 0x3, 0x0, |
|
0x3d, 0x8, 0x1, 0x0, 0x4, 0x68, 0xee, 0x3c, |
|
0x80, |
|
}, msg.Body) |
|
|
|
// S->C AAC decoder config |
|
msg, err = mr.Read() |
|
require.NoError(t, err) |
|
require.Equal(t, uint8(4), msg.ChunkStreamID) |
|
require.Equal(t, base.MessageType(0x08), msg.Type) |
|
require.Equal(t, []byte{0xae, 0x0, 0x12, 0x10}, msg.Body) |
|
}
|
|
|