mirror of
https://github.com/bluenviron/mediamtx
synced 2024-12-17 04:04:41 +00:00
258 lines
5.9 KiB
Go
258 lines
5.9 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/aler9/gortsplib"
|
|
"github.com/aler9/gortsplib/pkg/base"
|
|
"github.com/aler9/gortsplib/pkg/headers"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestClientRTMPPublish(t *testing.T) {
|
|
for _, source := range []string{
|
|
"videoaudio",
|
|
"video",
|
|
} {
|
|
t.Run(source, func(t *testing.T) {
|
|
p, ok := testProgram("")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "empty" + source + ".mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + ":1935/test1/test2",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-rtsp_transport", "udp",
|
|
"-i", "rtsp://" + ownDockerIP + ":8554/test1/test2",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.Equal(t, 0, cnt2.wait())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestClientRTMPRead(t *testing.T) {
|
|
p, ok := testProgram("")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "rtsp",
|
|
"rtsp://" + ownDockerIP + ":8554/teststream",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "rtmp://" + ownDockerIP + ":1935/teststream",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.Equal(t, 0, cnt2.wait())
|
|
}
|
|
|
|
func TestClientRTMPAuth(t *testing.T) {
|
|
t.Run("publish", func(t *testing.T) {
|
|
p, ok := testProgram("rtspDisable: yes\n" +
|
|
"paths:\n" +
|
|
" all:\n" +
|
|
" publishUser: testuser\n" +
|
|
" publishPass: testpass\n")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + "/teststream?user=testuser&pass=testpass",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "rtmp://" + ownDockerIP + "/teststream",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.Equal(t, 0, cnt2.wait())
|
|
})
|
|
|
|
t.Run("read", func(t *testing.T) {
|
|
p, ok := testProgram("rtspDisable: yes\n" +
|
|
"paths:\n" +
|
|
" all:\n" +
|
|
" readUser: testuser\n" +
|
|
" readPass: testpass\n")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + "/teststream",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "rtmp://" + ownDockerIP + "/teststream?user=testuser&pass=testpass",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.Equal(t, 0, cnt2.wait())
|
|
})
|
|
}
|
|
|
|
func TestClientRTMPAuthFail(t *testing.T) {
|
|
t.Run("publish", func(t *testing.T) {
|
|
p, ok := testProgram("rtspDisable: yes\n" +
|
|
"paths:\n" +
|
|
" all:\n" +
|
|
" publishUser: testuser2\n" +
|
|
" publishPass: testpass\n")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + "/teststream?user=testuser&pass=testpass",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "rtmp://" + ownDockerIP + "/teststream",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.NotEqual(t, 0, cnt2.wait())
|
|
})
|
|
|
|
t.Run("read", func(t *testing.T) {
|
|
p, ok := testProgram("rtspDisable: yes\n" +
|
|
"paths:\n" +
|
|
" all:\n" +
|
|
" readUser: testuser2\n" +
|
|
" readPass: testpass\n")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideo.mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + "/teststream",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
cnt2, err := newContainer("ffmpeg", "dest", []string{
|
|
"-i", "rtmp://" + ownDockerIP + "/teststream?user=testuser&pass=testpass",
|
|
"-vframes", "1",
|
|
"-f", "image2",
|
|
"-y", "/dev/null",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt2.close()
|
|
require.NotEqual(t, 0, cnt2.wait())
|
|
})
|
|
}
|
|
|
|
func TestClientRTMPRTPInfo(t *testing.T) {
|
|
p, ok := testProgram("")
|
|
require.Equal(t, true, ok)
|
|
defer p.close()
|
|
|
|
cnt1, err := newContainer("ffmpeg", "source", []string{
|
|
"-re",
|
|
"-stream_loop", "-1",
|
|
"-i", "emptyvideoaudio.mkv",
|
|
"-c", "copy",
|
|
"-f", "flv",
|
|
"rtmp://" + ownDockerIP + ":1935/teststream",
|
|
})
|
|
require.NoError(t, err)
|
|
defer cnt1.close()
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
dest, err := gortsplib.DialRead("rtsp://" + ownDockerIP + ":8554/teststream")
|
|
require.NoError(t, err)
|
|
defer dest.Close()
|
|
|
|
require.Equal(t, &headers.RTPInfo{
|
|
&headers.RTPInfoEntry{
|
|
URL: (&base.URL{
|
|
Scheme: "rtsp",
|
|
Host: ownDockerIP + ":8554",
|
|
Path: "/teststream/trackID=0",
|
|
}).String(),
|
|
SequenceNumber: (*dest.RTPInfo())[0].SequenceNumber,
|
|
Timestamp: (*dest.RTPInfo())[0].Timestamp,
|
|
},
|
|
&headers.RTPInfoEntry{
|
|
URL: (&base.URL{
|
|
Scheme: "rtsp",
|
|
Host: ownDockerIP + ":8554",
|
|
Path: "/teststream/trackID=1",
|
|
}).String(),
|
|
SequenceNumber: (*dest.RTPInfo())[1].SequenceNumber,
|
|
Timestamp: (*dest.RTPInfo())[1].Timestamp,
|
|
},
|
|
}, dest.RTPInfo())
|
|
}
|