mediamtx/internal/rtmp/conn_test.go

package rtmp

import (
	"bytes"
	"crypto/hmac"
	"crypto/rand"
	"crypto/sha256"
	"fmt"
	"io"
	"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"
)

var (
	hsClientFullKey = []byte{
		'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
		'F', 'l', 'a', 's', 'h', ' ', 'P', 'l', 'a', 'y', 'e', 'r', ' ',
		'0', '0', '1',
		0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1,
		0x02, 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB,
		0x93, 0xB8, 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE,
	}
	hsServerFullKey = []byte{
		'G', 'e', 'n', 'u', 'i', 'n', 'e', ' ', 'A', 'd', 'o', 'b', 'e', ' ',
		'F', 'l', 'a', 's', 'h', ' ', 'M', 'e', 'd', 'i', 'a', ' ',
		'S', 'e', 'r', 'v', 'e', 'r', ' ',
		'0', '0', '1',
		0xF0, 0xEE, 0xC2, 0x4A, 0x80, 0x68, 0xBE, 0xE8, 0x2E, 0x00, 0xD0, 0xD1,
		0x02, 0x9E, 0x7E, 0x57, 0x6E, 0xEC, 0x5D, 0x2D, 0x29, 0x80, 0x6F, 0xAB,
		0x93, 0xB8, 0xE6, 0x36, 0xCF, 0xEB, 0x31, 0xAE,
	}
	hsClientPartialKey = hsClientFullKey[:30]
	hsServerPartialKey = hsServerFullKey[:36]
)

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 hsMakeDigest(key []byte, src []byte, gap int) (dst []byte) {
	h := hmac.New(sha256.New, key)
	if gap <= 0 {
		h.Write(src)
	} else {
		h.Write(src[:gap])
		h.Write(src[gap+32:])
	}
	return h.Sum(nil)
}

func hsCalcDigestPos(p []byte, base int) (pos int) {
	for i := 0; i < 4; i++ {
		pos += int(p[base+i])
	}
	pos = (pos % 728) + base + 4
	return
}

func hsFindDigest(p []byte, key []byte, base int) int {
	gap := hsCalcDigestPos(p, base)
	digest := hsMakeDigest(key, p, gap)
	if !bytes.Equal(p[gap:gap+32], digest) {
		return -1
	}
	return gap
}

func hsParse1(p []byte, peerkey []byte, key []byte) (ok bool, digest []byte) {
	var pos int
	if pos = hsFindDigest(p, peerkey, 772); pos == -1 {
		if pos = hsFindDigest(p, peerkey, 8); pos == -1 {
			return
		}
	}
	ok = true
	digest = hsMakeDigest(key, p[pos:pos+32], -1)
	return
}

func writeHandshakeC2(w io.Writer, s0s1s2 []byte) error {
	ok, key := hsParse1(s0s1s2[1:1537], hsServerPartialKey, hsClientFullKey)
	if !ok {
		panic("unable to parse s0s1s2")
	}

	buf := make([]byte, 1536)
	rand.Read(buf)
	gap := len(buf) - 32
	digest := hsMakeDigest(key, buf, gap)
	copy(buf[gap:], digest)
	_, err := w.Write(buf)
	return err
}

func writeHandshakeC0C1(w io.Writer) error {
	buf := make([]byte, 1537)
	buf[0] = 0x03
	copy(buf[1:5], []byte{0x00, 0x00, 0x00, 0x00})
	copy(buf[5:9], []byte{0x09, 0x00, 0x7c, 0x02})
	rand.Read(buf[1+8:])
	gap := hsCalcDigestPos(buf[1:], 8)
	digest := hsMakeDigest(hsClientPartialKey, buf[1:], gap)
	copy(buf[gap+1:], digest)

	_, err := w.Write(buf)
	return err
}

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
}

type chunk0 struct {
	chunkStreamID byte
	timestamp     uint32
	typ           byte
	streamID      uint32
	bodyLen       uint32
	body          []byte
}

func (m *chunk0) read(r io.Reader, chunkMaxBodyLen int) error {
	header := make([]byte, 12)
	_, err := r.Read(header)
	if err != nil {
		return err
	}

	if header[0]>>6 != 0 {
		return fmt.Errorf("wrong chunk header type")
	}

	m.chunkStreamID = header[0] & 0x3F
	m.timestamp = uint32(header[3])<<16 | uint32(header[2])<<8 | uint32(header[1])
	m.bodyLen = uint32(header[4])<<16 | uint32(header[5])<<8 | uint32(header[6])
	m.typ = header[7]
	m.streamID = uint32(header[8])<<24 | uint32(header[9])<<16 | uint32(header[10])<<8 | uint32(header[11])

	chunkBodyLen := int(m.bodyLen)
	if chunkBodyLen > chunkMaxBodyLen {
		chunkBodyLen = chunkMaxBodyLen
	}

	m.body = make([]byte, chunkBodyLen)
	_, err = r.Read(m.body)
	return err
}

func (m chunk0) write(w io.Writer) error {
	header := make([]byte, 12)
	header[0] = m.chunkStreamID
	header[1] = byte(m.timestamp >> 16)
	header[2] = byte(m.timestamp >> 8)
	header[3] = byte(m.timestamp)
	header[4] = byte(m.bodyLen >> 16)
	header[5] = byte(m.bodyLen >> 8)
	header[6] = byte(m.bodyLen)
	header[7] = m.typ
	header[8] = byte(m.streamID)
	_, err := w.Write(header)
	if err != nil {
		return err
	}

	_, err = w.Write(m.body)
	return err
}

type chunk1 struct {
	chunkStreamID byte
	typ           byte
	body          []byte
}

func (m chunk1) write(w io.Writer) error {
	header := make([]byte, 8)
	header[0] = 1<<6 | m.chunkStreamID
	l := uint32(len(m.body))
	header[4] = byte(l >> 16)
	header[5] = byte(l >> 8)
	header[6] = byte(l)
	header[7] = m.typ
	_, err := w.Write(header)
	if err != nil {
		return err
	}

	_, err = w.Write(m.body)
	return err
}

type chunk3 struct {
	chunkStreamID byte
	body          []byte
}

func (m chunk3) write(w io.Writer) error {
	header := make([]byte, 1)
	header[0] = 3<<6 | m.chunkStreamID
	_, err := w.Write(header)
	if err != nil {
		return err
	}

	_, err = w.Write(m.body)
	return err
}

func TestReadMetadata(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.ReadMetadata()
		require.NoError(t, err)

		videoTrack2, 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)
		require.Equal(t, videoTrack2, videoTrack)

		audioTrack2, err := gortsplib.NewTrackAAC(96, 2, 44100, 2, nil)
		require.NoError(t, err)
		require.Equal(t, audioTrack2, audioTrack)

		close(done)
	}()

	conn, err := net.Dial("tcp", "127.0.0.1:9121")
	require.NoError(t, err)
	defer conn.Close()

	// C->S handshake C0+C1
	err = writeHandshakeC0C1(conn)
	require.NoError(t, err)

	// S->C handshake S0+S1+S2
	s0s1s2 := make([]byte, 1536*2+1)
	_, err = conn.Read(s0s1s2)
	require.NoError(t, err)

	// C->S handshake C2
	err = writeHandshakeC2(conn, s0s1s2)
	require.NoError(t, err)

	// 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 = chunk0{
		chunkStreamID: 3,
		typ:           0x14,
		bodyLen:       uint32(len(byts)),
		body:          byts[:128],
	}.write(conn)
	require.NoError(t, err)
	err = chunk3{
		chunkStreamID: 3,
		body:          byts[128:],
	}.write(conn)
	require.NoError(t, err)

	// S->C window acknowledgement size
	var c0 chunk0
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           5,
		bodyLen:       4,
		body:          []byte{0x00, 38, 37, 160},
	}, c0)

	// S->C set peer bandwidth
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           6,
		bodyLen:       5,
		body:          []byte{0x00, 0x26, 0x25, 0xa0, 0x02},
	}, c0)

	// S->C set chunk size
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           1,
		bodyLen:       4,
		body:          []byte{0x00, 0x01, 0x00, 0x00},
	}, c0)

	// S->C result
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(3), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err := flvio.ParseAMFVals(c0.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 = chunk0{
		chunkStreamID: 2,
		typ:           1,
		bodyLen:       4,
		body:          []byte{0x00, 0x01, 0x00, 0x00},
	}.write(conn)
	require.NoError(t, err)

	// C->S releaseStream
	err = chunk1{
		chunkStreamID: 3,
		typ:           0x14,
		body: flvio.FillAMF0ValsMalloc([]interface{}{
			"releaseStream",
			float64(2),
			nil,
			"",
		}),
	}.write(conn)
	require.NoError(t, err)

	// C->S FCPublish
	err = chunk1{
		chunkStreamID: 3,
		typ:           0x14,
		body: flvio.FillAMF0ValsMalloc([]interface{}{
			"FCPublish",
			float64(3),
			nil,
			"",
		}),
	}.write(conn)
	require.NoError(t, err)

	// C->S createStream
	err = chunk3{
		chunkStreamID: 3,
		body: flvio.FillAMF0ValsMalloc([]interface{}{
			"createStream",
			float64(4),
			nil,
		}),
	}.write(conn)
	require.NoError(t, err)

	// S->C result
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(3), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.body, false)
	require.NoError(t, err)
	require.Equal(t, []interface{}{
		"_result",
		float64(4),
		nil,
		float64(1),
	}, arr)

	// C->S publish
	byts = flvio.FillAMF0ValsMalloc([]interface{}{
		"publish",
		float64(5),
		nil,
		"",
		"live",
	})
	err = chunk0{
		chunkStreamID: 8,
		typ:           0x14,
		streamID:      1,
		bodyLen:       uint32(len(byts)),
		body:          byts,
	}.write(conn)
	require.NoError(t, err)

	// S->C onStatus
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(5), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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)

	// 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 = chunk0{
		chunkStreamID: 4,
		typ:           0x12,
		streamID:      1,
		bodyLen:       uint32(len(byts)),
		body:          byts,
	}.write(conn)
	require.NoError(t, err)

	// C->S H264 decoder config
	codec := nh264.Codec{
		SPS: map[int][]byte{
			0: {
				0x67, 0x64, 0x00, 0x0c, 0xac, 0x3b, 0x50, 0xb0,
				0x4b, 0x42, 0x00, 0x00, 0x03, 0x00, 0x02, 0x00,
				0x00, 0x03, 0x00, 0x3d, 0x08,
			},
		},
		PPS: map[int][]byte{
			0: {
				0x68, 0xee, 0x3c, 0x80,
			},
		},
	}
	b := make([]byte, 128)
	var n int
	codec.ToConfig(b, &n)
	decConfig := b[:n]
	err = chunk0{
		chunkStreamID: 6,
		typ:           flvio.TAG_VIDEO,
		streamID:      1,
		bodyLen:       uint32(len(decConfig) + 5),
		body:          append([]byte{flvio.FRAME_KEY<<4 | flvio.VIDEO_H264, 0, 0, 0, 0}, decConfig...),
	}.write(conn)
	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 = chunk0{
		chunkStreamID: 4,
		typ:           flvio.TAG_AUDIO,
		streamID:      1,
		bodyLen:       uint32(len(enc) + 2),
		body: append([]byte{
			flvio.SOUND_AAC<<4 | flvio.SOUND_44Khz<<2 | flvio.SOUND_16BIT<<1 | flvio.SOUND_STEREO,
			flvio.AAC_SEQHDR,
		}, enc...),
	}.write(conn)
	require.NoError(t, err)

	<-done
}

func TestWriteMetadata(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)
		require.NoError(t, err)

		err = rconn.WriteMetadata(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+C1
	err = writeHandshakeC0C1(conn)
	require.NoError(t, err)

	// S->C handshake S0+S1+S2
	s0s1s2 := make([]byte, 1536*2+1)
	_, err = conn.Read(s0s1s2)
	require.NoError(t, err)

	// C->S handshake C2
	err = writeHandshakeC2(conn, s0s1s2)
	require.NoError(t, err)

	// 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 = chunk0{
		chunkStreamID: 3,
		typ:           0x14,
		bodyLen:       uint32(len(byts)),
		body:          byts[:128],
	}.write(conn)
	require.NoError(t, err)
	err = chunk3{
		chunkStreamID: 3,
		body:          byts[128:],
	}.write(conn)
	require.NoError(t, err)

	// S->C window acknowledgement size
	var c0 chunk0
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           5,
		bodyLen:       4,
		body:          []byte{0x00, 38, 37, 160},
	}, c0)

	// S->C set peer bandwidth
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           6,
		bodyLen:       5,
		body:          []byte{0x00, 0x26, 0x25, 0xa0, 0x02},
	}, c0)

	// S->C set chunk size
	err = c0.read(conn, 128)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           1,
		bodyLen:       4,
		body:          []byte{0x00, 0x01, 0x00, 0x00},
	}, c0)

	// S->C result
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(3), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err := flvio.ParseAMFVals(c0.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 = chunk0{
		chunkStreamID: 2,
		typ:           0x05,
		bodyLen:       4,
		body:          []byte{0x00, 0x26, 0x25, 0xa0},
	}.write(conn)
	require.NoError(t, err)

	// C->S set chunk size
	err = chunk0{
		chunkStreamID: 2,
		typ:           1,
		bodyLen:       4,
		body:          []byte{0x00, 0x01, 0x00, 0x00},
	}.write(conn)
	require.NoError(t, err)

	// C->S createStream
	err = chunk1{
		chunkStreamID: 3,
		typ:           0x14,
		body: flvio.FillAMF0ValsMalloc([]interface{}{
			"createStream",
			float64(2),
			nil,
		}),
	}.write(conn)
	require.NoError(t, err)

	// S->C result
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(3), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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 = chunk0{
		chunkStreamID: 8,
		bodyLen:       uint32(len(byts)),
		body:          byts,
	}.write(conn)
	require.NoError(t, err)

	// C->S play
	byts = flvio.FillAMF0ValsMalloc([]interface{}{
		"play",
		float64(4),
		nil,
		"",
		float64(-2000),
	})
	err = chunk0{
		chunkStreamID: 8,
		typ:           0x14,
		bodyLen:       uint32(len(byts)),
		body:          byts,
	}.write(conn)
	require.NoError(t, err)

	// S->C event "stream is recorded"
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           4,
		bodyLen:       6,
		body:          []byte{0x00, 0x04, 0x00, 0x00, 0x00, 0x01},
	}, c0)

	// S->C event "stream begin 1"
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, chunk0{
		chunkStreamID: 2,
		typ:           4,
		bodyLen:       6,
		body:          []byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x01},
	}, c0)

	// S->C onStatus
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(5), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(5), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(5), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(5), c0.chunkStreamID)
	require.Equal(t, uint8(0x14), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(4), c0.chunkStreamID)
	require.Equal(t, uint8(0x12), c0.typ)
	arr, err = flvio.ParseAMFVals(c0.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
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(6), c0.chunkStreamID)
	require.Equal(t, uint8(0x09), c0.typ)
	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,
	}, c0.body)

	// S->C AAC decoder config
	err = c0.read(conn, 65536)
	require.NoError(t, err)
	require.Equal(t, uint8(4), c0.chunkStreamID)
	require.Equal(t, uint8(0x08), c0.typ)
	require.Equal(t, []byte{0xae, 0x0, 0x12, 0x10}, c0.body)
}