package rtmp import ( "crypto/tls" "net" "os" "testing" "time" "github.com/stretchr/testify/require" "github.com/bluenviron/mediamtx/internal/conf" "github.com/bluenviron/mediamtx/internal/defs" "github.com/bluenviron/mediamtx/internal/protocols/rtmp" "github.com/bluenviron/mediamtx/internal/test" ) func TestSource(t *testing.T) { for _, encryption := range []string{ "plain", "tls", } { for _, auth := range []string{ "no auth", "auth", } { t.Run(encryption+"_"+auth, func(t *testing.T) { var ln net.Listener if encryption == "plain" { var err error ln, err = net.Listen("tcp", "127.0.0.1:1935") require.NoError(t, err) } else { serverCertFpath, err := test.CreateTempFile(test.TLSCertPub) require.NoError(t, err) defer os.Remove(serverCertFpath) serverKeyFpath, err := test.CreateTempFile(test.TLSCertKey) require.NoError(t, err) defer os.Remove(serverKeyFpath) var cert tls.Certificate cert, err = tls.LoadX509KeyPair(serverCertFpath, serverKeyFpath) require.NoError(t, err) ln, err = tls.Listen("tcp", "127.0.0.1:1936", &tls.Config{Certificates: []tls.Certificate{cert}}) require.NoError(t, err) } defer ln.Close() go func() { for { nconn, err := ln.Accept() require.NoError(t, err) defer nconn.Close() conn := &rtmp.ServerConn{ RW: nconn, } err = conn.Initialize() require.NoError(t, err) if auth == "auth" { err = conn.CheckCredentials("myuser", "mypass") if err != nil { continue } } err = conn.Accept() require.NoError(t, err) w := &rtmp.Writer{ Conn: conn, VideoTrack: test.FormatH264, AudioTrack: test.FormatMPEG4Audio, } err = w.Initialize() require.NoError(t, err) err = w.WriteH264(2*time.Second, 2*time.Second, [][]byte{{5, 2, 3, 4}}) require.NoError(t, err) break } }() var source string if encryption == "plain" { source = "rtmp://" } else { source = "rtmps://" } if auth == "auth" { source += "myuser:mypass@" } source += "localhost/teststream" n := 0 timeNow = func() time.Time { d := time.Date(2009, 5, 20, 22, 15, 25, 427000, time.Local).Add(time.Duration(n) * 2 * time.Second) n++ return d } defer func() { timeNow = time.Now }() te := test.NewSourceTester( func(p defs.StaticSourceParent) defs.StaticSource { return &Source{ ReadTimeout: conf.Duration(10 * time.Second), WriteTimeout: conf.Duration(10 * time.Second), Parent: p, } }, source, &conf.Path{ SourceFingerprint: "33949E05FFFB5FF3E8AA16F8213A6251B4D9363804BA53233C4DA9A46D6F2739", }, ) defer te.Close() <-te.Unit }) } } }