Alessandro Ros
3 years ago
committed by
aler9
18 changed files with 650 additions and 259 deletions
@ -1,56 +1,147 @@
@@ -1,56 +1,147 @@
|
||||
package core |
||||
|
||||
import ( |
||||
"crypto/tls" |
||||
"net" |
||||
"os" |
||||
"testing" |
||||
"time" |
||||
|
||||
"github.com/aler9/gortsplib" |
||||
"github.com/aler9/gortsplib/pkg/mpeg4audio" |
||||
"github.com/aler9/gortsplib/pkg/url" |
||||
"github.com/notedit/rtmp/format/flv/flvio" |
||||
"github.com/stretchr/testify/require" |
||||
|
||||
"github.com/aler9/rtsp-simple-server/internal/rtmp" |
||||
"github.com/aler9/rtsp-simple-server/internal/rtmp/message" |
||||
) |
||||
|
||||
func TestRTMPSource(t *testing.T) { |
||||
for _, source := range []string{ |
||||
"videoaudio", |
||||
"video", |
||||
for _, ca := range []string{ |
||||
"plain", |
||||
"tls", |
||||
} { |
||||
t.Run(source, func(t *testing.T) { |
||||
cnt1, err := newContainer("nginx-rtmp", "rtmpserver", []string{}) |
||||
t.Run(ca, func(t *testing.T) { |
||||
ln, err := func() (net.Listener, error) { |
||||
if ca == "plain" { |
||||
return net.Listen("tcp", "127.0.0.1:1937") |
||||
} |
||||
|
||||
serverCertFpath, err := writeTempFile(serverCert) |
||||
require.NoError(t, err) |
||||
defer os.Remove(serverCertFpath) |
||||
|
||||
serverKeyFpath, err := writeTempFile(serverKey) |
||||
require.NoError(t, err) |
||||
defer os.Remove(serverKeyFpath) |
||||
|
||||
var cert tls.Certificate |
||||
cert, err = tls.LoadX509KeyPair(serverCertFpath, serverKeyFpath) |
||||
require.NoError(t, err) |
||||
|
||||
return tls.Listen("tcp", "127.0.0.1:1937", &tls.Config{Certificates: []tls.Certificate{cert}}) |
||||
}() |
||||
require.NoError(t, err) |
||||
defer cnt1.close() |
||||
|
||||
cnt2, err := newContainer("ffmpeg", "source", []string{ |
||||
"-re", |
||||
"-stream_loop", "-1", |
||||
"-i", "empty" + source + ".mkv", |
||||
"-c", "copy", |
||||
"-f", "flv", |
||||
"rtmp://" + cnt1.ip() + "/stream/test", |
||||
}) |
||||
defer ln.Close() |
||||
|
||||
connected := make(chan struct{}) |
||||
received := make(chan struct{}) |
||||
done := make(chan struct{}) |
||||
|
||||
go func() { |
||||
nconn, err := ln.Accept() |
||||
require.NoError(t, err) |
||||
defer nconn.Close() |
||||
conn := rtmp.NewConn(nconn) |
||||
|
||||
_, _, err = conn.InitializeServer() |
||||
require.NoError(t, err) |
||||
|
||||
videoTrack := &gortsplib.TrackH264{ |
||||
PayloadType: 96, |
||||
SPS: []byte{ // 1920x1080 baseline
|
||||
0x67, 0x42, 0xc0, 0x28, 0xd9, 0x00, 0x78, 0x02, |
||||
0x27, 0xe5, 0x84, 0x00, 0x00, 0x03, 0x00, 0x04, |
||||
0x00, 0x00, 0x03, 0x00, 0xf0, 0x3c, 0x60, 0xc9, 0x20, |
||||
}, |
||||
PPS: []byte{0x08, 0x06, 0x07, 0x08}, |
||||
} |
||||
|
||||
audioTrack := &gortsplib.TrackMPEG4Audio{ |
||||
PayloadType: 96, |
||||
Config: &mpeg4audio.Config{ |
||||
Type: 2, |
||||
SampleRate: 44100, |
||||
ChannelCount: 2, |
||||
}, |
||||
SizeLength: 13, |
||||
IndexLength: 3, |
||||
IndexDeltaLength: 3, |
||||
} |
||||
|
||||
err = conn.WriteTracks(videoTrack, audioTrack) |
||||
require.NoError(t, err) |
||||
|
||||
<-connected |
||||
|
||||
err = conn.WriteMessage(&message.MsgVideo{ |
||||
ChunkStreamID: 6, |
||||
MessageStreamID: 0x1000000, |
||||
IsKeyFrame: true, |
||||
H264Type: flvio.AVC_NALU, |
||||
Payload: []byte{0x00, 0x00, 0x00, 0x04, 0x05, 0x02, 0x03, 0x04}, |
||||
}) |
||||
require.NoError(t, err) |
||||
|
||||
<-done |
||||
}() |
||||
|
||||
if ca == "plain" { |
||||
p, ok := newInstance("paths:\n" + |
||||
" proxied:\n" + |
||||
" source: rtmp://localhost:1937/teststream\n" + |
||||
" sourceOnDemand: yes\n") |
||||
require.Equal(t, true, ok) |
||||
defer p.close() |
||||
} else { |
||||
p, ok := newInstance("paths:\n" + |
||||
" proxied:\n" + |
||||
" source: rtmps://localhost:1937/teststream\n" + |
||||
" sourceFingerprint: 33949E05FFFB5FF3E8AA16F8213A6251B4D9363804BA53233C4DA9A46D6F2739\n" + |
||||
" sourceOnDemand: yes\n") |
||||
require.Equal(t, true, ok) |
||||
defer p.close() |
||||
} |
||||
|
||||
c := gortsplib.Client{ |
||||
OnPacketRTP: func(ctx *gortsplib.ClientOnPacketRTPCtx) { |
||||
require.Equal(t, []byte{ |
||||
0x18, 0x0, 0x19, 0x67, 0x42, 0xc0, 0x28, 0xd9, |
||||
0x0, 0x78, 0x2, 0x27, 0xe5, 0x84, 0x0, 0x0, |
||||
0x3, 0x0, 0x4, 0x0, 0x0, 0x3, 0x0, 0xf0, |
||||
0x3c, 0x60, 0xc9, 0x20, 0x0, 0x4, 0x8, 0x6, |
||||
0x7, 0x8, 0x0, 0x4, 0x5, 0x2, 0x3, 0x4, |
||||
}, ctx.Packet.Payload) |
||||
close(received) |
||||
}, |
||||
} |
||||
|
||||
u, err := url.Parse("rtsp://127.0.0.1:8554/proxied") |
||||
require.NoError(t, err) |
||||
defer cnt2.close() |
||||
|
||||
time.Sleep(1 * time.Second) |
||||
|
||||
p, ok := newInstance("hlsDisable: yes\n" + |
||||
"rtmpDisable: yes\n" + |
||||
"paths:\n" + |
||||
" proxied:\n" + |
||||
" source: rtmp://localhost/stream/test\n" + |
||||
" sourceOnDemand: yes\n") |
||||
require.Equal(t, true, ok) |
||||
defer p.close() |
||||
|
||||
time.Sleep(1 * time.Second) |
||||
|
||||
cnt3, err := newContainer("ffmpeg", "dest", []string{ |
||||
"-rtsp_transport", "udp", |
||||
"-i", "rtsp://localhost:8554/proxied", |
||||
"-vframes", "1", |
||||
"-f", "image2", |
||||
"-y", "/dev/null", |
||||
}) |
||||
|
||||
err = c.Start(u.Scheme, u.Host) |
||||
require.NoError(t, err) |
||||
defer cnt3.close() |
||||
require.Equal(t, 0, cnt3.wait()) |
||||
defer c.Close() |
||||
|
||||
tracks, baseURL, _, err := c.Describe(u) |
||||
require.NoError(t, err) |
||||
|
||||
err = c.SetupAndPlay(tracks, baseURL) |
||||
require.NoError(t, err) |
||||
|
||||
close(connected) |
||||
<-received |
||||
close(done) |
||||
}) |
||||
} |
||||
} |
||||
|
||||
@ -1,8 +0,0 @@
@@ -1,8 +0,0 @@
|
||||
FROM amd64/alpine:3.14 |
||||
|
||||
RUN apk add --no-cache \ |
||||
nginx-mod-rtmp |
||||
|
||||
COPY nginx.conf /etc/nginx/ |
||||
|
||||
ENTRYPOINT [ "nginx", "-g", "daemon off;" ] |
||||
@ -1,23 +0,0 @@
@@ -1,23 +0,0 @@
|
||||
|
||||
pid /run/nginx.pid; |
||||
worker_processes auto; |
||||
pcre_jit on; |
||||
|
||||
error_log /dev/null; |
||||
|
||||
include /etc/nginx/modules/*.conf; |
||||
|
||||
events { |
||||
worker_connections 20000; |
||||
} |
||||
|
||||
rtmp { |
||||
server { |
||||
listen 1935; |
||||
access_log /dev/null; |
||||
|
||||
application stream { |
||||
live on; |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue