Files
rtsp-simple-server/internal/protocols/rtmp/message/command_amf0_test.go
Alessandro Ros da7c5744b1 rtmp: add new AMF0 encoder and encoder (#3069)
This improves performance, security and removes a dependency.
2024-02-24 19:11:42 +01:00

74 lines
2.2 KiB
Go

package message
import (
"testing"
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/amf0"
"github.com/bluenviron/mediamtx/internal/protocols/rtmp/rawmessage"
)
func BenchmarkCommandAMF0Marshal(b *testing.B) {
msg := &CommandAMF0{
ChunkStreamID: 3,
Name: "connect",
CommandID: 1,
Arguments: []interface{}{
amf0.Object{
{Key: "app", Value: "/stream"},
{Key: "flashVer", Value: "LNX 9,0,124,2"},
{Key: "tcUrl", Value: "http://example.com"},
{Key: "fpad", Value: false},
{Key: "capabilities", Value: 15},
{Key: "audioCodecs", Value: 4071},
{Key: "videoCodecs", Value: 252},
{Key: "videoFunction", Value: 1},
},
},
}
for n := 0; n < b.N; n++ {
msg.marshal() //nolint:errcheck
}
}
func BenchmarkCommandAMF0Unmarshal(b *testing.B) {
raw := &rawmessage.Message{
ChunkStreamID: 0x3,
Timestamp: 0,
Type: 0x14,
MessageStreamID: 0x0,
Body: []uint8{
0x02, 0x00, 0x07, 0x63, 0x6f, 0x6e, 0x6e, 0x65,
0x63, 0x74, 0x00, 0x3f, 0xf0, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x03, 0x00, 0x03, 0x61, 0x70,
0x70, 0x02, 0x00, 0x07, 0x2f, 0x73, 0x74, 0x72,
0x65, 0x61, 0x6d, 0x00, 0x08, 0x66, 0x6c, 0x61,
0x73, 0x68, 0x56, 0x65, 0x72, 0x02, 0x00, 0x0d,
0x4c, 0x4e, 0x58, 0x20, 0x39, 0x2c, 0x30, 0x2c,
0x31, 0x32, 0x34, 0x2c, 0x32, 0x00, 0x05, 0x74,
0x63, 0x55, 0x72, 0x6c, 0x02, 0x00, 0x12, 0x68,
0x74, 0x74, 0x70, 0x3a, 0x2f, 0x2f, 0x65, 0x78,
0x61, 0x6d, 0x70, 0x6c, 0x65, 0x2e, 0x63, 0x6f,
0x6d, 0x00, 0x04, 0x66, 0x70, 0x61, 0x64, 0x01,
0x00, 0x00, 0x0c, 0x63, 0x61, 0x70, 0x61, 0x62,
0x69, 0x6c, 0x69, 0x74, 0x69, 0x65, 0x73, 0x00,
0x40, 0x2e, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x0b, 0x61, 0x75, 0x64, 0x69, 0x6f, 0x43,
0x6f, 0x64, 0x65, 0x63, 0x73, 0x00, 0x40, 0xaf,
0xce, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0b,
0x76, 0x69, 0x64, 0x65, 0x6f, 0x43, 0x6f, 0x64,
0x65, 0x63, 0x73, 0x00, 0x40, 0x6f, 0x80, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x0d, 0x76, 0x69,
0x64, 0x65, 0x6f, 0x46, 0x75, 0x6e, 0x63, 0x74,
0x69, 0x6f, 0x6e, 0x00, 0x3f, 0xf0, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x09,
},
}
msg := &CommandAMF0{}
for n := 0; n < b.N; n++ {
msg.unmarshal(raw) //nolint:errcheck
}
}