mirror of
https://github.com/Monibuca/plugin-rtsp.git
synced 2025-10-05 07:36:57 +08:00
更新依赖库aler9 到2.1.0
This commit is contained in:
38
client.go
38
client.go
@@ -1,8 +1,8 @@
|
|||||||
package rtsp
|
package rtsp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/aler9/gortsplib"
|
"github.com/aler9/gortsplib/v2"
|
||||||
"github.com/aler9/gortsplib/pkg/url"
|
"github.com/aler9/gortsplib/v2/pkg/url"
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
"m7s.live/engine/v4"
|
"m7s.live/engine/v4"
|
||||||
)
|
)
|
||||||
@@ -28,14 +28,15 @@ func (p *RTSPPuller) Connect() error {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
p.Client = &gortsplib.Client{
|
p.Client = &gortsplib.Client{
|
||||||
OnPacketRTP: func(ctx *gortsplib.ClientOnPacketRTPCtx) {
|
// OnPacketRTP: func(ctx *gortsplib.ClientOnPacketRTPCtx) {
|
||||||
if p.RTSPPublisher.Tracks[ctx.TrackID] != nil {
|
// if p.RTSPPublisher.Tracks[ctx.TrackID] != nil {
|
||||||
p.RTSPPublisher.Tracks[ctx.TrackID].WriteRTPPack(ctx.Packet)
|
// p.RTSPPublisher.Tracks[ctx.TrackID].WriteRTPPack(ctx.Packet)
|
||||||
}
|
// }
|
||||||
},
|
// },
|
||||||
ReadBufferCount: rtspConfig.ReadBufferSize,
|
ReadBufferCount: rtspConfig.ReadBufferSize,
|
||||||
Transport: &p.Transport,
|
Transport: &p.Transport,
|
||||||
}
|
}
|
||||||
|
|
||||||
// parse URL
|
// parse URL
|
||||||
u, err := url.Parse(p.RemoteURL)
|
u, err := url.Parse(p.RemoteURL)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -63,11 +64,21 @@ func (p *RTSPPuller) Pull() (err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
p.tracks = tracks
|
p.tracks = tracks
|
||||||
p.SetTracks()
|
err = p.SetTracks()
|
||||||
if err = p.SetupAndPlay(tracks, baseURL); err != nil {
|
if err != nil {
|
||||||
|
p.Error("SetTracks", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err = p.SetupAll(tracks, baseURL); err != nil {
|
||||||
p.Error("SetupAndPlay", zap.Error(err))
|
p.Error("SetupAndPlay", zap.Error(err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
p.OnPacketRTPAny(p.OnPacket)
|
||||||
|
_, err = p.Play(nil)
|
||||||
|
if err != nil {
|
||||||
|
p.Error("Play", zap.Error(err))
|
||||||
|
return
|
||||||
|
}
|
||||||
return p.Wait()
|
return p.Wait()
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -81,9 +92,9 @@ type RTSPPusher struct {
|
|||||||
func (p *RTSPPusher) OnEvent(event any) {
|
func (p *RTSPPusher) OnEvent(event any) {
|
||||||
switch v := event.(type) {
|
switch v := event.(type) {
|
||||||
case engine.VideoRTP:
|
case engine.VideoRTP:
|
||||||
p.Client.WritePacketRTP(p.videoTrackId, &v.Packet)
|
p.Client.WritePacketRTP(p.videoTrack, &v.Packet)
|
||||||
case engine.AudioRTP:
|
case engine.AudioRTP:
|
||||||
p.Client.WritePacketRTP(p.audioTrackId, &v.Packet)
|
p.Client.WritePacketRTP(p.audioTrack, &v.Packet)
|
||||||
default:
|
default:
|
||||||
p.RTSPSubscriber.OnEvent(event)
|
p.RTSPSubscriber.OnEvent(event)
|
||||||
}
|
}
|
||||||
@@ -127,13 +138,12 @@ func (p *RTSPPusher) Push() (err error) {
|
|||||||
p.Error("Announce", zap.Error(err))
|
p.Error("Announce", zap.Error(err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
for _, track := range p.tracks {
|
err = p.SetupAll(p.tracks, u)
|
||||||
_, err = p.Setup(track, u, 0, 0)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
p.Error("Setup", zap.Error(err))
|
p.Error("Setup", zap.Error(err))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
|
||||||
if _, err = p.Record(); err != nil {
|
if _, err = p.Record(); err != nil {
|
||||||
p.Error("Record", zap.Error(err))
|
p.Error("Record", zap.Error(err))
|
||||||
return
|
return
|
||||||
|
7
go.mod
7
go.mod
@@ -3,9 +3,10 @@ module m7s.live/plugin/rtsp/v4
|
|||||||
go 1.18
|
go 1.18
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/aler9/gortsplib v0.0.0-20221115222755-87d5a512b129
|
github.com/aler9/gortsplib/v2 v2.1.0
|
||||||
|
github.com/pion/rtp v1.7.13
|
||||||
go.uber.org/zap v1.23.0
|
go.uber.org/zap v1.23.0
|
||||||
m7s.live/engine/v4 v4.10.0
|
m7s.live/engine/v4 v4.11.0
|
||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
@@ -21,10 +22,10 @@ require (
|
|||||||
github.com/lufia/plan9stats v0.0.0-20220913051719-115f729f3c8c // indirect
|
github.com/lufia/plan9stats v0.0.0-20220913051719-115f729f3c8c // indirect
|
||||||
github.com/marten-seemann/qtls-go1-18 v0.1.3 // indirect
|
github.com/marten-seemann/qtls-go1-18 v0.1.3 // indirect
|
||||||
github.com/marten-seemann/qtls-go1-19 v0.1.1 // indirect
|
github.com/marten-seemann/qtls-go1-19 v0.1.1 // indirect
|
||||||
|
github.com/mcuadros/go-defaults v1.2.0 // indirect
|
||||||
github.com/onsi/ginkgo/v2 v2.2.0 // indirect
|
github.com/onsi/ginkgo/v2 v2.2.0 // indirect
|
||||||
github.com/pion/randutil v0.1.0 // indirect
|
github.com/pion/randutil v0.1.0 // indirect
|
||||||
github.com/pion/rtcp v1.2.10 // indirect
|
github.com/pion/rtcp v1.2.10 // indirect
|
||||||
github.com/pion/rtp v1.7.13 // indirect
|
|
||||||
github.com/pion/sdp/v3 v3.0.6 // indirect
|
github.com/pion/sdp/v3 v3.0.6 // indirect
|
||||||
github.com/pion/webrtc/v3 v3.1.49 // indirect
|
github.com/pion/webrtc/v3 v3.1.49 // indirect
|
||||||
github.com/power-devops/perfstat v0.0.0-20220216144756-c35f1ee13d7c // indirect
|
github.com/power-devops/perfstat v0.0.0-20220216144756-c35f1ee13d7c // indirect
|
||||||
|
17
go.sum
17
go.sum
@@ -1,6 +1,6 @@
|
|||||||
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
|
||||||
github.com/aler9/gortsplib v0.0.0-20221115222755-87d5a512b129 h1:CG96FPsxizdlpsRIsjU2xQR6G3QC0sPK+f+AeVEr0Tg=
|
github.com/aler9/gortsplib/v2 v2.1.0 h1:HKQjFugTGH32FScSXPL2p4SGLpxGBRnyBSzR3GKYG9U=
|
||||||
github.com/aler9/gortsplib v0.0.0-20221115222755-87d5a512b129/go.mod h1:BOWNZ/QBkY/eVcRqUzJbPFEsRJshwxaxBT01K260Jeo=
|
github.com/aler9/gortsplib/v2 v2.1.0/go.mod h1:jZy239Qg+C9OSvJrPs2vIPalm9/pam1ljFir5DdydnA=
|
||||||
github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=
|
github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=
|
||||||
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
|
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
|
||||||
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
|
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
|
||||||
@@ -54,7 +54,6 @@ github.com/kelindar/process v0.0.0-20170730150328-69a29e249ec3/go.mod h1:+lTCLnZ
|
|||||||
github.com/kelindar/rate v1.0.0/go.mod h1:AjT4G+hTItNwt30lucEGZIz8y7Uk5zPho6vurIZ+1Es=
|
github.com/kelindar/rate v1.0.0/go.mod h1:AjT4G+hTItNwt30lucEGZIz8y7Uk5zPho6vurIZ+1Es=
|
||||||
github.com/kelindar/tcp v1.0.0/go.mod h1:JB5hj1cshLU60XrLij2BBxW3JQ4hOye8vqbyvuKb52k=
|
github.com/kelindar/tcp v1.0.0/go.mod h1:JB5hj1cshLU60XrLij2BBxW3JQ4hOye8vqbyvuKb52k=
|
||||||
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
|
||||||
github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
|
|
||||||
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
|
||||||
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
|
||||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||||
@@ -70,6 +69,10 @@ github.com/marten-seemann/qtls-go1-18 v0.1.3 h1:R4H2Ks8P6pAtUagjFty2p7BVHn3XiwDA
|
|||||||
github.com/marten-seemann/qtls-go1-18 v0.1.3/go.mod h1:mJttiymBAByA49mhlNZZGrH5u1uXYZJ+RW28Py7f4m4=
|
github.com/marten-seemann/qtls-go1-18 v0.1.3/go.mod h1:mJttiymBAByA49mhlNZZGrH5u1uXYZJ+RW28Py7f4m4=
|
||||||
github.com/marten-seemann/qtls-go1-19 v0.1.1 h1:mnbxeq3oEyQxQXwI4ReCgW9DPoPR94sNlqWoDZnjRIE=
|
github.com/marten-seemann/qtls-go1-19 v0.1.1 h1:mnbxeq3oEyQxQXwI4ReCgW9DPoPR94sNlqWoDZnjRIE=
|
||||||
github.com/marten-seemann/qtls-go1-19 v0.1.1/go.mod h1:5HTDWtVudo/WFsHKRNuOhWlbdjrfs5JHrYb0wIJqGpI=
|
github.com/marten-seemann/qtls-go1-19 v0.1.1/go.mod h1:5HTDWtVudo/WFsHKRNuOhWlbdjrfs5JHrYb0wIJqGpI=
|
||||||
|
github.com/mcuadros/go-defaults v1.2.0 h1:FODb8WSf0uGaY8elWJAkoLL0Ri6AlZ1bFlenk56oZtc=
|
||||||
|
github.com/mcuadros/go-defaults v1.2.0/go.mod h1:WEZtHEVIGYVDqkKSWBdWKUVdRyKlMfulPaGDWIVeCWY=
|
||||||
|
github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e h1:fD57ERR4JtEqsWbfPhv4DMiApHyliiK5xCTNVSPiaAs=
|
||||||
|
github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno=
|
||||||
github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A=
|
github.com/nxadm/tail v1.4.4/go.mod h1:kenIhsEOeOJmVchQTgglprH7qJGnHDVpk1VPCcaMI8A=
|
||||||
github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
|
github.com/nxadm/tail v1.4.8/go.mod h1:+ncqLTQzXmGhMZNUePPaPqPvBxHAIsmXswZKocGu+AU=
|
||||||
github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
github.com/onsi/ginkgo v1.6.0/go.mod h1:lLunBs/Ym6LB5Z9jYTR76FiuTmxDTDusOGeTQH+WWjE=
|
||||||
@@ -121,7 +124,6 @@ github.com/power-devops/perfstat v0.0.0-20220216144756-c35f1ee13d7c h1:NRoLoZvkB
|
|||||||
github.com/power-devops/perfstat v0.0.0-20220216144756-c35f1ee13d7c/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=
|
github.com/power-devops/perfstat v0.0.0-20220216144756-c35f1ee13d7c/go.mod h1:OmDBASR4679mdNQnz2pUhc2G8CO2JrUAVFDRBDP/hJE=
|
||||||
github.com/q191201771/naza v0.30.8 h1:Lhh29o65C4PmTDj2l+eKfsw9dddpgWZk4bFICtcnSaA=
|
github.com/q191201771/naza v0.30.8 h1:Lhh29o65C4PmTDj2l+eKfsw9dddpgWZk4bFICtcnSaA=
|
||||||
github.com/q191201771/naza v0.30.8/go.mod h1:n+dpJjQSh90PxBwxBNuifOwQttywvSIN5TkWSSYCeBk=
|
github.com/q191201771/naza v0.30.8/go.mod h1:n+dpJjQSh90PxBwxBNuifOwQttywvSIN5TkWSSYCeBk=
|
||||||
github.com/rogpeppe/go-internal v1.8.1 h1:geMPLpDpQOgVyCg5z5GoRwLHepNdb71NXb67XFkP+Eg=
|
|
||||||
github.com/sclevine/agouti v3.0.0+incompatible/go.mod h1:b4WX9W9L1sfQKXeJf1mUTLZKJ48R1S7H23Ji7oFO5Bw=
|
github.com/sclevine/agouti v3.0.0+incompatible/go.mod h1:b4WX9W9L1sfQKXeJf1mUTLZKJ48R1S7H23Ji7oFO5Bw=
|
||||||
github.com/shirou/gopsutil/v3 v3.22.10 h1:4KMHdfBRYXGF9skjDWiL4RA2N+E8dRdodU/bOZpPoVg=
|
github.com/shirou/gopsutil/v3 v3.22.10 h1:4KMHdfBRYXGF9skjDWiL4RA2N+E8dRdodU/bOZpPoVg=
|
||||||
github.com/shirou/gopsutil/v3 v3.22.10/go.mod h1:QNza6r4YQoydyCfo6rH0blGfKahgibh4dQmV5xdFkQk=
|
github.com/shirou/gopsutil/v3 v3.22.10/go.mod h1:QNza6r4YQoydyCfo6rH0blGfKahgibh4dQmV5xdFkQk=
|
||||||
@@ -255,8 +257,9 @@ google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp0
|
|||||||
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
|
google.golang.org/protobuf v1.26.0/go.mod h1:9q0QmTI4eRPtz6boOQmLYwt+qCgq0jsYwAQnmE0givc=
|
||||||
google.golang.org/protobuf v1.28.0 h1:w43yiav+6bVFTBQFZX0r7ipe9JQ1QsbMgHwbBziscLw=
|
google.golang.org/protobuf v1.28.0 h1:w43yiav+6bVFTBQFZX0r7ipe9JQ1QsbMgHwbBziscLw=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
|
|
||||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f h1:BLraFXnmrev5lT+xlilqcH8XK9/i0At2xKjWk4p6zsU=
|
||||||
|
gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
|
gopkg.in/fsnotify.v1 v1.4.7/go.mod h1:Tz8NjZHkW78fSQdbUxIjBTcgA1z1m8ZHf0WmKUhAMys=
|
||||||
gopkg.in/natefinch/lumberjack.v2 v2.0.0/go.mod h1:l0ndWWf7gzL7RNwBG7wST/UCcT4T24xpD6X8LsfU/+k=
|
gopkg.in/natefinch/lumberjack.v2 v2.0.0/go.mod h1:l0ndWWf7gzL7RNwBG7wST/UCcT4T24xpD6X8LsfU/+k=
|
||||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7/go.mod h1:dt/ZhP58zS4L8KSrWDmTeBkI65Dw0HsyUHuEVlX15mw=
|
||||||
@@ -269,5 +272,5 @@ gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C
|
|||||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
m7s.live/engine/v4 v4.10.0 h1:U47ifoIC41lXisvG69Os4GtUTnYhAOzV/dZSj4OYVZg=
|
m7s.live/engine/v4 v4.11.0 h1:YbL1H+FtVlTKfi3WE4bk3fcmf4HrkTKygupGUCLdjvo=
|
||||||
m7s.live/engine/v4 v4.10.0/go.mod h1:znB5Ux7fJ/bIVKnbHrwoInztVn7z/4CrXwjc24b1exI=
|
m7s.live/engine/v4 v4.11.0/go.mod h1:dx52RjysN9uocotAP2caLyb6j2uWwjcnstOy8C/U8sU=
|
||||||
|
2
main.go
2
main.go
@@ -6,7 +6,7 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/aler9/gortsplib"
|
"github.com/aler9/gortsplib/v2"
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
. "m7s.live/engine/v4"
|
. "m7s.live/engine/v4"
|
||||||
"m7s.live/engine/v4/config"
|
"m7s.live/engine/v4/config"
|
||||||
|
162
publisher.go
162
publisher.go
@@ -1,14 +1,10 @@
|
|||||||
package rtsp
|
package rtsp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/base64"
|
"github.com/aler9/gortsplib/v2/pkg/codecs/mpeg4audio"
|
||||||
"encoding/hex"
|
"github.com/aler9/gortsplib/v2/pkg/format"
|
||||||
"errors"
|
"github.com/aler9/gortsplib/v2/pkg/media"
|
||||||
"fmt"
|
"github.com/pion/rtp"
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/aler9/gortsplib"
|
|
||||||
"go.uber.org/zap"
|
"go.uber.org/zap"
|
||||||
. "m7s.live/engine/v4"
|
. "m7s.live/engine/v4"
|
||||||
"m7s.live/engine/v4/common"
|
"m7s.live/engine/v4/common"
|
||||||
@@ -17,131 +13,65 @@ import (
|
|||||||
|
|
||||||
type RTSPPublisher struct {
|
type RTSPPublisher struct {
|
||||||
Publisher
|
Publisher
|
||||||
Tracks []common.AVTrack `json:"-"`
|
Tracks map[*media.Media]common.AVTrack `json:"-"`
|
||||||
RTSPIO
|
RTSPIO
|
||||||
}
|
}
|
||||||
|
|
||||||
func (p *RTSPPublisher) SetTracks() error {
|
func (p *RTSPPublisher) SetTracks() error {
|
||||||
p.Tracks = make([]common.AVTrack, len(p.tracks))
|
p.Tracks = make(map[*media.Media]common.AVTrack, len(p.tracks))
|
||||||
defer func() {
|
defer func() {
|
||||||
for i, track := range p.Tracks {
|
for _, track := range p.Tracks {
|
||||||
if track == nil {
|
p.Info("set track", zap.String("name", track.GetBase().Name))
|
||||||
p.Info("unknown track", zap.String("codec", p.tracks[i].String()))
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
p.Info("set track", zap.Int("trackId", i), zap.String("name", track.GetBase().Name))
|
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
for trackId, track := range p.tracks {
|
for _, track := range p.tracks {
|
||||||
md := track.MediaDescription()
|
for _, forma := range track.Formats {
|
||||||
v, ok := md.Attribute("rtpmap")
|
switch f := forma.(type) {
|
||||||
if !ok {
|
case *format.H264:
|
||||||
return errors.New("rtpmap attribute not found")
|
vt := NewH264(p.Stream, f.PayloadType())
|
||||||
|
p.Tracks[track] = vt
|
||||||
|
if len(f.SPS) > 0 {
|
||||||
|
vt.WriteSliceBytes(f.SPS)
|
||||||
}
|
}
|
||||||
v = strings.TrimSpace(v)
|
if len(f.PPS) > 0 {
|
||||||
vals := strings.Split(v, " ")
|
vt.WriteSliceBytes(f.PPS)
|
||||||
if len(vals) != 2 {
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
fmtp := make(map[string]string)
|
case *format.H265:
|
||||||
if v, ok = md.Attribute("fmtp"); ok {
|
vt := NewH265(p.Stream, f.PayloadType())
|
||||||
if tmp := strings.SplitN(v, " ", 2); len(tmp) == 2 {
|
p.Tracks[track] = vt
|
||||||
for _, kv := range strings.Split(tmp[1], ";") {
|
if len(f.VPS) > 0 {
|
||||||
kv = strings.Trim(kv, " ")
|
vt.WriteSliceBytes(f.VPS)
|
||||||
|
|
||||||
if len(kv) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
tmp := strings.SplitN(kv, "=", 2)
|
if len(f.SPS) > 0 {
|
||||||
if len(tmp) == 2 {
|
vt.WriteSliceBytes(f.SPS)
|
||||||
fmtp[strings.TrimSpace(tmp[0])] = tmp[1]
|
|
||||||
}
|
}
|
||||||
|
if len(f.PPS) > 0 {
|
||||||
|
vt.WriteSliceBytes(f.PPS)
|
||||||
}
|
}
|
||||||
|
case *format.MPEG4Audio:
|
||||||
|
at := NewAAC(p.Stream, f.PayloadType())
|
||||||
|
p.Tracks[track] = at
|
||||||
|
at.SizeLength = f.SizeLength
|
||||||
|
if f.Config.Type == mpeg4audio.ObjectTypeAACLC {
|
||||||
|
at.Mode = 1
|
||||||
}
|
}
|
||||||
}
|
at.SampleRate = uint32(f.Config.SampleRate)
|
||||||
timeScale := 0
|
at.Channels = uint8(f.Config.ChannelCount)
|
||||||
keyval := strings.Split(vals[1], "/")
|
asc, _ := f.Config.Marshal()
|
||||||
if i, err := strconv.Atoi(keyval[1]); err == nil {
|
|
||||||
timeScale = i
|
|
||||||
}
|
|
||||||
if len(keyval) >= 2 {
|
|
||||||
switch strings.ToLower(keyval[0]) {
|
|
||||||
case "h264":
|
|
||||||
vt := NewH264(p.Stream)
|
|
||||||
if payloadType, err := strconv.Atoi(vals[0]); err == nil {
|
|
||||||
vt.PayloadType = byte(payloadType)
|
|
||||||
}
|
|
||||||
p.Tracks[trackId] = vt
|
|
||||||
t := track.(*gortsplib.TrackH264)
|
|
||||||
if len(t.SPS) > 0 {
|
|
||||||
vt.WriteSliceBytes(t.SPS)
|
|
||||||
}
|
|
||||||
if len(t.PPS) > 0 {
|
|
||||||
vt.WriteSliceBytes(t.PPS)
|
|
||||||
}
|
|
||||||
case "h265", "hevc":
|
|
||||||
vt := NewH265(p.Stream)
|
|
||||||
if payloadType, err := strconv.Atoi(vals[0]); err == nil {
|
|
||||||
vt.PayloadType = byte(payloadType)
|
|
||||||
}
|
|
||||||
p.Tracks[trackId] = vt
|
|
||||||
if v, ok := fmtp["sprop-vps"]; ok {
|
|
||||||
vps, _ := base64.StdEncoding.DecodeString(v)
|
|
||||||
vt.WriteSliceBytes(vps)
|
|
||||||
}
|
|
||||||
if v, ok := fmtp["sprop-sps"]; ok {
|
|
||||||
sps, _ := base64.StdEncoding.DecodeString(v)
|
|
||||||
vt.WriteSliceBytes(sps)
|
|
||||||
}
|
|
||||||
if v, ok := fmtp["sprop-pps"]; ok {
|
|
||||||
pps, _ := base64.StdEncoding.DecodeString(v)
|
|
||||||
vt.WriteSliceBytes(pps)
|
|
||||||
}
|
|
||||||
case "pcma":
|
|
||||||
at := NewG711(p.Stream, true)
|
|
||||||
if payloadType, err := strconv.Atoi(vals[0]); err == nil {
|
|
||||||
at.PayloadType = byte(payloadType)
|
|
||||||
}
|
|
||||||
p.Tracks[trackId] = at
|
|
||||||
at.SampleRate = uint32(timeScale)
|
|
||||||
if len(keyval) >= 3 {
|
|
||||||
x, _ := strconv.Atoi(keyval[2])
|
|
||||||
at.Channels = byte(x)
|
|
||||||
} else {
|
|
||||||
at.Channels = 1
|
|
||||||
}
|
|
||||||
at.AVCCHead = []byte{(byte(at.CodecID) << 4) | (1 << 1)}
|
|
||||||
case "pcmu":
|
|
||||||
at := NewG711(p.Stream, false)
|
|
||||||
if payloadType, err := strconv.Atoi(vals[0]); err == nil {
|
|
||||||
at.PayloadType = byte(payloadType)
|
|
||||||
}
|
|
||||||
p.Tracks[trackId] = at
|
|
||||||
at.SampleRate = uint32(timeScale)
|
|
||||||
if len(keyval) >= 3 {
|
|
||||||
x, _ := strconv.Atoi(keyval[2])
|
|
||||||
at.Channels = byte(x)
|
|
||||||
} else {
|
|
||||||
at.Channels = 1
|
|
||||||
}
|
|
||||||
at.AVCCHead = []byte{(byte(at.CodecID) << 4) | (1 << 1)}
|
|
||||||
case "mpeg4-generic":
|
|
||||||
at := NewAAC(p.Stream)
|
|
||||||
if payloadType, err := strconv.Atoi(vals[0]); err == nil {
|
|
||||||
at.PayloadType = byte(payloadType)
|
|
||||||
}
|
|
||||||
p.Tracks[trackId] = at
|
|
||||||
if config, ok := fmtp["config"]; ok {
|
|
||||||
asc, _ := hex.DecodeString(config)
|
|
||||||
// 复用AVCC写入逻辑,解析出AAC的配置信息
|
// 复用AVCC写入逻辑,解析出AAC的配置信息
|
||||||
at.WriteSequenceHead(append([]byte{0xAF, 0x00}, asc...))
|
at.WriteSequenceHead(append([]byte{0xAF, 0x00}, asc...))
|
||||||
} else {
|
case *format.G711:
|
||||||
RTSPPlugin.Warn("aac no config")
|
at := NewG711(p.Stream, !f.MULaw, f.PayloadType(), uint32(f.ClockRate()))
|
||||||
}
|
p.Tracks[track] = at
|
||||||
default:
|
at.AVCCHead = []byte{(byte(at.CodecID) << 4) | (1 << 1)}
|
||||||
return fmt.Errorf("unsupport codec:%s", keyval[0])
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (p *RTSPPublisher) OnPacket(m *media.Media, f format.Format, pack *rtp.Packet) {
|
||||||
|
if t, ok := p.Tracks[m]; ok {
|
||||||
|
t.WriteRTPPack(pack)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
29
server.go
29
server.go
@@ -1,16 +1,17 @@
|
|||||||
package rtsp
|
package rtsp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/aler9/gortsplib"
|
"github.com/aler9/gortsplib/v2"
|
||||||
"github.com/aler9/gortsplib/pkg/base"
|
"github.com/aler9/gortsplib/v2/pkg/base"
|
||||||
|
"github.com/aler9/gortsplib/v2/pkg/media"
|
||||||
. "m7s.live/engine/v4"
|
. "m7s.live/engine/v4"
|
||||||
)
|
)
|
||||||
|
|
||||||
type RTSPIO struct {
|
type RTSPIO struct {
|
||||||
tracks gortsplib.Tracks
|
tracks media.Medias
|
||||||
stream *gortsplib.ServerStream
|
stream *gortsplib.ServerStream
|
||||||
audioTrackId int
|
audioTrack *media.Media
|
||||||
videoTrackId int
|
videoTrack *media.Media
|
||||||
}
|
}
|
||||||
|
|
||||||
func (conf *RTSPConfig) OnConnOpen(ctx *gortsplib.ServerHandlerOnConnOpenCtx) {
|
func (conf *RTSPConfig) OnConnOpen(ctx *gortsplib.ServerHandlerOnConnOpenCtx) {
|
||||||
@@ -88,6 +89,9 @@ func (conf *RTSPConfig) OnPlay(ctx *gortsplib.ServerHandlerOnPlayCtx) (*base.Res
|
|||||||
return &resp, nil
|
return &resp, nil
|
||||||
}
|
}
|
||||||
func (conf *RTSPConfig) OnRecord(ctx *gortsplib.ServerHandlerOnRecordCtx) (*base.Response, error) {
|
func (conf *RTSPConfig) OnRecord(ctx *gortsplib.ServerHandlerOnRecordCtx) (*base.Response, error) {
|
||||||
|
if p, ok := conf.Load(ctx.Session); ok {
|
||||||
|
ctx.Session.OnPacketRTPAny(p.(*RTSPPublisher).OnPacket)
|
||||||
|
}
|
||||||
return &base.Response{
|
return &base.Response{
|
||||||
StatusCode: base.StatusOK,
|
StatusCode: base.StatusOK,
|
||||||
}, nil
|
}, nil
|
||||||
@@ -96,8 +100,8 @@ func (conf *RTSPConfig) OnAnnounce(ctx *gortsplib.ServerHandlerOnAnnounceCtx) (*
|
|||||||
p := &RTSPPublisher{}
|
p := &RTSPPublisher{}
|
||||||
p.SetIO(ctx.Conn.NetConn())
|
p.SetIO(ctx.Conn.NetConn())
|
||||||
if err := RTSPPlugin.Publish(ctx.Path, p); err == nil {
|
if err := RTSPPlugin.Publish(ctx.Path, p); err == nil {
|
||||||
p.tracks = ctx.Tracks
|
p.tracks = ctx.Medias
|
||||||
p.stream = gortsplib.NewServerStream(ctx.Tracks)
|
p.stream = gortsplib.NewServerStream(ctx.Medias)
|
||||||
if err = p.SetTracks(); err != nil {
|
if err = p.SetTracks(); err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -112,14 +116,3 @@ func (conf *RTSPConfig) OnAnnounce(ctx *gortsplib.ServerHandlerOnAnnounceCtx) (*
|
|||||||
StatusCode: base.StatusOK,
|
StatusCode: base.StatusOK,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (conf *RTSPConfig) OnPacketRTP(ctx *gortsplib.ServerHandlerOnPacketRTPCtx) {
|
|
||||||
if p, ok := conf.Load(ctx.Session); ok {
|
|
||||||
switch v := p.(type) {
|
|
||||||
case *RTSPPublisher:
|
|
||||||
if v.Tracks[ctx.TrackID] != nil {
|
|
||||||
v.Tracks[ctx.TrackID].WriteRTPPack(ctx.Packet)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
@@ -1,8 +1,10 @@
|
|||||||
package rtsp
|
package rtsp
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/aler9/gortsplib"
|
"github.com/aler9/gortsplib/v2"
|
||||||
"github.com/aler9/gortsplib/pkg/mpeg4audio"
|
"github.com/aler9/gortsplib/v2/pkg/codecs/mpeg4audio"
|
||||||
|
"github.com/aler9/gortsplib/v2/pkg/format"
|
||||||
|
"github.com/aler9/gortsplib/v2/pkg/media"
|
||||||
. "m7s.live/engine/v4"
|
. "m7s.live/engine/v4"
|
||||||
"m7s.live/engine/v4/codec"
|
"m7s.live/engine/v4/codec"
|
||||||
"m7s.live/engine/v4/track"
|
"m7s.live/engine/v4/track"
|
||||||
@@ -21,17 +23,29 @@ func (s *RTSPSubscriber) OnEvent(event any) {
|
|||||||
}
|
}
|
||||||
switch v.CodecID {
|
switch v.CodecID {
|
||||||
case codec.CodecID_H264:
|
case codec.CodecID_H264:
|
||||||
vtrack := &gortsplib.TrackH264{
|
video := &media.Media{
|
||||||
PayloadType: v.PayloadType, SPS: v.ParamaterSets[0], PPS: v.ParamaterSets[1],
|
Type: media.TypeVideo,
|
||||||
|
Formats: []format.Format{&format.H264{
|
||||||
|
PacketizationMode: 1,
|
||||||
|
PayloadTyp: v.PayloadType,
|
||||||
|
SPS: v.ParamaterSets[0],
|
||||||
|
PPS: v.ParamaterSets[1],
|
||||||
|
}},
|
||||||
}
|
}
|
||||||
s.videoTrackId = len(s.tracks)
|
s.videoTrack = video
|
||||||
s.tracks = append(s.tracks, vtrack)
|
s.tracks = append(s.tracks, video)
|
||||||
case codec.CodecID_H265:
|
case codec.CodecID_H265:
|
||||||
vtrack := &gortsplib.TrackH265{
|
video := &media.Media{
|
||||||
PayloadType: v.PayloadType, VPS: v.ParamaterSets[0], SPS: v.ParamaterSets[1], PPS: v.ParamaterSets[2],
|
Type: media.TypeVideo,
|
||||||
|
Formats: []format.Format{&format.H265{
|
||||||
|
PayloadTyp: v.PayloadType,
|
||||||
|
VPS: v.ParamaterSets[0],
|
||||||
|
SPS: v.ParamaterSets[1],
|
||||||
|
PPS: v.ParamaterSets[2],
|
||||||
|
}},
|
||||||
}
|
}
|
||||||
s.videoTrackId = len(s.tracks)
|
s.videoTrack = video
|
||||||
s.tracks = append(s.tracks, vtrack)
|
s.tracks = append(s.tracks, video)
|
||||||
}
|
}
|
||||||
s.AddTrack(v)
|
s.AddTrack(v)
|
||||||
case *track.Audio:
|
case *track.Audio:
|
||||||
@@ -40,27 +54,46 @@ func (s *RTSPSubscriber) OnEvent(event any) {
|
|||||||
}
|
}
|
||||||
switch v.CodecID {
|
switch v.CodecID {
|
||||||
case codec.CodecID_AAC:
|
case codec.CodecID_AAC:
|
||||||
var mpegConf mpeg4audio.Config
|
audio := &media.Media{
|
||||||
mpegConf.Unmarshal(v.SequenceHead[2:])
|
Type: media.TypeAudio,
|
||||||
atrack := &gortsplib.TrackMPEG4Audio{
|
Formats: []format.Format{&format.MPEG4Audio{
|
||||||
PayloadType: v.PayloadType, Config: &mpegConf, SizeLength: 13, IndexLength: 3, IndexDeltaLength: 3,
|
PayloadTyp: v.PayloadType,
|
||||||
|
Config: &mpeg4audio.Config{
|
||||||
|
Type: mpeg4audio.ObjectTypeAACLC,
|
||||||
|
SampleRate: int(v.SampleRate),
|
||||||
|
ChannelCount: int(v.Channels),
|
||||||
|
},
|
||||||
|
SizeLength: 13,
|
||||||
|
IndexLength: 3,
|
||||||
|
IndexDeltaLength: 3,
|
||||||
|
}},
|
||||||
}
|
}
|
||||||
s.audioTrackId = len(s.tracks)
|
s.audioTrack = audio
|
||||||
s.tracks = append(s.tracks, atrack)
|
s.tracks = append(s.tracks, audio)
|
||||||
case codec.CodecID_PCMA:
|
case codec.CodecID_PCMA:
|
||||||
s.audioTrackId = len(s.tracks)
|
audio := &media.Media{
|
||||||
s.tracks = append(s.tracks, &gortsplib.TrackPCMA{})
|
Type: media.TypeAudio,
|
||||||
|
Formats: []format.Format{&format.G711{}},
|
||||||
|
}
|
||||||
|
s.audioTrack = audio
|
||||||
|
s.tracks = append(s.tracks, audio)
|
||||||
case codec.CodecID_PCMU:
|
case codec.CodecID_PCMU:
|
||||||
s.audioTrackId = len(s.tracks)
|
audio := &media.Media{
|
||||||
s.tracks = append(s.tracks, &gortsplib.TrackPCMU{})
|
Type: media.TypeAudio,
|
||||||
|
Formats: []format.Format{&format.G711{
|
||||||
|
MULaw: true,
|
||||||
|
}},
|
||||||
|
}
|
||||||
|
s.audioTrack = audio
|
||||||
|
s.tracks = append(s.tracks, audio)
|
||||||
}
|
}
|
||||||
s.AddTrack(v)
|
s.AddTrack(v)
|
||||||
case ISubscriber:
|
case ISubscriber:
|
||||||
s.stream = gortsplib.NewServerStream(s.tracks)
|
s.stream = gortsplib.NewServerStream(s.tracks)
|
||||||
case VideoRTP:
|
case VideoRTP:
|
||||||
s.stream.WritePacketRTP(s.videoTrackId, &v.Packet)
|
s.stream.WritePacketRTP(s.videoTrack, &v.Packet)
|
||||||
case AudioRTP:
|
case AudioRTP:
|
||||||
s.stream.WritePacketRTP(s.audioTrackId, &v.Packet)
|
s.stream.WritePacketRTP(s.audioTrack, &v.Packet)
|
||||||
default:
|
default:
|
||||||
s.Subscriber.OnEvent(event)
|
s.Subscriber.OnEvent(event)
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user