2021-07-24 13:55:42 +00:00
|
|
|
package core
|
2020-10-19 20:17:48 +00:00
|
|
|
|
|
|
|
import (
|
2021-05-10 21:23:56 +00:00
|
|
|
"context"
|
2021-04-16 20:46:22 +00:00
|
|
|
"crypto/sha256"
|
|
|
|
"crypto/tls"
|
|
|
|
"encoding/hex"
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
2020-10-19 20:17:48 +00:00
|
|
|
"time"
|
|
|
|
|
2023-04-01 16:39:12 +00:00
|
|
|
"github.com/bluenviron/gortsplib/v3"
|
|
|
|
"github.com/bluenviron/gortsplib/v3/pkg/base"
|
2023-05-14 15:02:03 +00:00
|
|
|
"github.com/bluenviron/gortsplib/v3/pkg/headers"
|
2022-11-02 19:52:12 +00:00
|
|
|
"github.com/pion/rtp"
|
2020-10-25 10:41:41 +00:00
|
|
|
|
2023-04-01 17:52:06 +00:00
|
|
|
"github.com/aler9/mediamtx/internal/conf"
|
|
|
|
"github.com/aler9/mediamtx/internal/logger"
|
2023-04-01 16:39:12 +00:00
|
|
|
"github.com/bluenviron/gortsplib/v3/pkg/url"
|
2020-10-19 20:17:48 +00:00
|
|
|
)
|
|
|
|
|
2021-07-24 13:55:42 +00:00
|
|
|
type rtspSourceParent interface {
|
2023-05-04 18:16:41 +00:00
|
|
|
logger.Writer
|
2022-08-04 19:07:17 +00:00
|
|
|
sourceStaticImplSetReady(req pathSourceStaticSetReadyReq) pathSourceStaticSetReadyRes
|
|
|
|
sourceStaticImplSetNotReady(req pathSourceStaticSetNotReadyReq)
|
2020-10-19 20:17:48 +00:00
|
|
|
}
|
|
|
|
|
2021-07-24 13:55:42 +00:00
|
|
|
type rtspSource struct {
|
2021-09-26 21:06:40 +00:00
|
|
|
readTimeout conf.StringDuration
|
|
|
|
writeTimeout conf.StringDuration
|
2021-02-18 22:26:45 +00:00
|
|
|
readBufferCount int
|
2021-07-24 13:55:42 +00:00
|
|
|
parent rtspSourceParent
|
2020-10-19 20:17:48 +00:00
|
|
|
}
|
|
|
|
|
2021-07-24 13:55:42 +00:00
|
|
|
func newRTSPSource(
|
2021-09-26 21:06:40 +00:00
|
|
|
readTimeout conf.StringDuration,
|
|
|
|
writeTimeout conf.StringDuration,
|
2021-02-18 22:26:45 +00:00
|
|
|
readBufferCount int,
|
2022-04-07 10:50:35 +00:00
|
|
|
parent rtspSourceParent,
|
|
|
|
) *rtspSource {
|
2022-07-30 19:51:38 +00:00
|
|
|
return &rtspSource{
|
2021-01-10 11:55:53 +00:00
|
|
|
readTimeout: readTimeout,
|
|
|
|
writeTimeout: writeTimeout,
|
|
|
|
readBufferCount: readBufferCount,
|
|
|
|
parent: parent,
|
2020-10-19 20:17:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-24 16:11:53 +00:00
|
|
|
func (s *rtspSource) Log(level logger.Level, format string, args ...interface{}) {
|
2023-05-04 18:16:41 +00:00
|
|
|
s.parent.Log(level, "[rtsp source] "+format, args...)
|
2020-12-08 11:21:06 +00:00
|
|
|
}
|
|
|
|
|
2023-05-14 15:02:03 +00:00
|
|
|
func createRangeHeader(cnf *conf.PathConf) (*headers.Range, error) {
|
|
|
|
switch cnf.RtspRangeType {
|
|
|
|
case conf.RtspRangeTypeClock:
|
|
|
|
start, err := time.Parse("20060102T150405Z", cnf.RtspRangeStart)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &headers.Range{
|
|
|
|
Value: &headers.RangeUTC{
|
|
|
|
Start: start,
|
|
|
|
},
|
|
|
|
}, nil
|
|
|
|
|
|
|
|
case conf.RtspRangeTypeNPT:
|
|
|
|
start, err := time.ParseDuration(cnf.RtspRangeStart)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &headers.Range{
|
|
|
|
Value: &headers.RangeNPT{
|
|
|
|
Start: start,
|
|
|
|
},
|
|
|
|
}, nil
|
|
|
|
|
|
|
|
case conf.RtspRangeTypeSMPTE:
|
|
|
|
start, err := time.ParseDuration(cnf.RtspRangeStart)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return &headers.Range{
|
|
|
|
Value: &headers.RangeSMPTE{
|
|
|
|
Start: headers.RangeSMPTETime{
|
|
|
|
Time: start,
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}, nil
|
|
|
|
|
|
|
|
default:
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-07-30 19:51:38 +00:00
|
|
|
// run implements sourceStaticImpl.
|
2023-02-13 11:12:04 +00:00
|
|
|
func (s *rtspSource) run(ctx context.Context, cnf *conf.PathConf, reloadConf chan *conf.PathConf) error {
|
2022-07-24 16:11:53 +00:00
|
|
|
s.Log(logger.Debug, "connecting")
|
2020-10-19 20:17:48 +00:00
|
|
|
|
2021-12-03 22:15:21 +00:00
|
|
|
var tlsConfig *tls.Config
|
2023-02-13 11:12:04 +00:00
|
|
|
if cnf.SourceFingerprint != "" {
|
2021-12-03 22:15:21 +00:00
|
|
|
tlsConfig = &tls.Config{
|
|
|
|
InsecureSkipVerify: true,
|
|
|
|
VerifyConnection: func(cs tls.ConnectionState) error {
|
|
|
|
h := sha256.New()
|
|
|
|
h.Write(cs.PeerCertificates[0].Raw)
|
|
|
|
hstr := hex.EncodeToString(h.Sum(nil))
|
2023-02-13 11:12:04 +00:00
|
|
|
fingerprintLower := strings.ToLower(cnf.SourceFingerprint)
|
2021-12-03 22:15:21 +00:00
|
|
|
|
|
|
|
if hstr != fingerprintLower {
|
|
|
|
return fmt.Errorf("server fingerprint do not match: expected %s, got %s",
|
|
|
|
fingerprintLower, hstr)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
},
|
2021-10-25 19:01:29 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-11-12 21:29:56 +00:00
|
|
|
c := &gortsplib.Client{
|
2023-02-13 11:12:04 +00:00
|
|
|
Transport: cnf.SourceProtocol.Transport,
|
2021-10-25 19:01:29 +00:00
|
|
|
TLSConfig: tlsConfig,
|
2021-09-26 21:06:40 +00:00
|
|
|
ReadTimeout: time.Duration(s.readTimeout),
|
|
|
|
WriteTimeout: time.Duration(s.writeTimeout),
|
2021-05-10 21:23:56 +00:00
|
|
|
ReadBufferCount: s.readBufferCount,
|
2023-02-13 11:12:04 +00:00
|
|
|
AnyPortEnable: cnf.SourceAnyPortEnable,
|
2021-05-10 21:23:56 +00:00
|
|
|
OnRequest: func(req *base.Request) {
|
2022-07-24 16:11:53 +00:00
|
|
|
s.Log(logger.Debug, "c->s %v", req)
|
2021-05-10 21:23:56 +00:00
|
|
|
},
|
|
|
|
OnResponse: func(res *base.Response) {
|
2022-07-24 16:11:53 +00:00
|
|
|
s.Log(logger.Debug, "s->c %v", res)
|
2021-05-10 21:23:56 +00:00
|
|
|
},
|
2023-04-09 15:11:54 +00:00
|
|
|
OnTransportSwitch: func(err error) {
|
|
|
|
s.Log(logger.Warn, err.Error())
|
|
|
|
},
|
|
|
|
OnPacketLost: func(err error) {
|
|
|
|
s.Log(logger.Warn, err.Error())
|
|
|
|
},
|
|
|
|
OnDecodeError: func(err error) {
|
|
|
|
s.Log(logger.Warn, err.Error())
|
2022-10-31 18:10:32 +00:00
|
|
|
},
|
2021-05-10 21:23:56 +00:00
|
|
|
}
|
|
|
|
|
2023-02-13 11:12:04 +00:00
|
|
|
u, err := url.Parse(cnf.Source)
|
2020-10-19 20:17:48 +00:00
|
|
|
if err != nil {
|
2022-07-24 16:11:53 +00:00
|
|
|
return err
|
2020-10-19 20:17:48 +00:00
|
|
|
}
|
|
|
|
|
2021-11-12 21:29:56 +00:00
|
|
|
err = c.Start(u.Scheme, u.Host)
|
|
|
|
if err != nil {
|
2022-07-24 16:11:53 +00:00
|
|
|
return err
|
2021-08-10 16:34:10 +00:00
|
|
|
}
|
2022-01-23 18:36:33 +00:00
|
|
|
defer c.Close()
|
2021-03-23 19:37:43 +00:00
|
|
|
|
2021-05-10 21:23:56 +00:00
|
|
|
readErr := make(chan error)
|
|
|
|
go func() {
|
2021-11-12 21:29:56 +00:00
|
|
|
readErr <- func() error {
|
2022-12-13 19:54:17 +00:00
|
|
|
medias, baseURL, _, err := c.Describe(u)
|
2021-11-12 21:29:56 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-12-13 19:54:17 +00:00
|
|
|
err = c.SetupAll(medias, baseURL)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
2021-11-12 21:29:56 +00:00
|
|
|
}
|
|
|
|
|
2022-08-14 11:01:06 +00:00
|
|
|
res := s.parent.sourceStaticImplSetReady(pathSourceStaticSetReadyReq{
|
2022-12-13 19:54:17 +00:00
|
|
|
medias: medias,
|
2022-08-14 11:01:06 +00:00
|
|
|
generateRTPPackets: false,
|
|
|
|
})
|
2022-01-14 22:42:41 +00:00
|
|
|
if res.err != nil {
|
|
|
|
return res.err
|
2021-11-12 21:29:56 +00:00
|
|
|
}
|
|
|
|
|
2022-12-13 19:54:17 +00:00
|
|
|
s.Log(logger.Info, "ready: %s", sourceMediaInfo(medias))
|
2021-11-12 21:29:56 +00:00
|
|
|
|
2023-05-14 12:18:03 +00:00
|
|
|
defer s.parent.sourceStaticImplSetNotReady(pathSourceStaticSetNotReadyReq{})
|
2021-11-12 21:29:56 +00:00
|
|
|
|
2022-12-13 19:54:17 +00:00
|
|
|
for _, medi := range medias {
|
|
|
|
for _, forma := range medi.Formats {
|
2023-05-14 12:18:03 +00:00
|
|
|
cmedi := medi
|
|
|
|
cforma := forma
|
2023-04-25 16:13:51 +00:00
|
|
|
|
2023-05-14 12:18:03 +00:00
|
|
|
c.OnPacketRTP(cmedi, cforma, func(pkt *rtp.Packet) {
|
|
|
|
res.stream.writeRTPPacket(cmedi, cforma, pkt, time.Now())
|
2023-04-25 16:13:51 +00:00
|
|
|
})
|
2022-11-02 19:52:12 +00:00
|
|
|
}
|
2021-11-12 21:29:56 +00:00
|
|
|
}
|
|
|
|
|
2023-05-14 15:02:03 +00:00
|
|
|
rangeHeader, err := createRangeHeader(cnf)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err = c.Play(rangeHeader)
|
2021-11-12 21:29:56 +00:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return c.Wait()
|
|
|
|
}()
|
2021-05-10 21:23:56 +00:00
|
|
|
}()
|
2020-10-19 20:17:48 +00:00
|
|
|
|
2023-02-13 11:12:04 +00:00
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case err := <-readErr:
|
|
|
|
return err
|
|
|
|
|
|
|
|
case <-reloadConf:
|
2021-11-12 21:29:56 +00:00
|
|
|
|
2023-02-13 11:12:04 +00:00
|
|
|
case <-ctx.Done():
|
|
|
|
c.Close()
|
|
|
|
<-readErr
|
|
|
|
return nil
|
|
|
|
}
|
2020-10-19 20:17:48 +00:00
|
|
|
}
|
|
|
|
}
|
2021-07-04 16:13:49 +00:00
|
|
|
|
2022-08-04 19:07:17 +00:00
|
|
|
// apiSourceDescribe implements sourceStaticImpl.
|
2023-05-14 12:18:03 +00:00
|
|
|
func (*rtspSource) apiSourceDescribe() pathAPISourceOrReader {
|
|
|
|
return pathAPISourceOrReader{
|
|
|
|
Type: "rtspSource",
|
|
|
|
ID: "",
|
|
|
|
}
|
2021-07-04 16:13:49 +00:00
|
|
|
}
|