fix: add rtsp/rtsps const

This commit is contained in:
Jake Daynes
2026-03-13 04:15:26 -07:00
parent 8d216370d3
commit f715379a44
+10 -5
View File
@@ -21,6 +21,11 @@ import (
"github.com/bluenviron/gortsplib/v5/pkg/liberrors" "github.com/bluenviron/gortsplib/v5/pkg/liberrors"
) )
const (
rtsp = "rtsp"
rtsps = "rtsps"
)
func (a Attacker) newRTSPClient(u *base.URL) (*gortsplib.Client, error) { func (a Attacker) newRTSPClient(u *base.URL) (*gortsplib.Client, error) {
client := &gortsplib.Client{ client := &gortsplib.Client{
ReadTimeout: a.timeout, ReadTimeout: a.timeout,
@@ -69,7 +74,7 @@ func (a Attacker) probeDescribeHeaders(ctx context.Context, u *base.URL, urlStr
var conn net.Conn var conn net.Conn
var err error var err error
if u.Scheme == "rtsps" { if u.Scheme == rtsps {
tlsDialer := &tls.Dialer{NetDialer: dialer, Config: &tls.Config{InsecureSkipVerify: true}} tlsDialer := &tls.Dialer{NetDialer: dialer, Config: &tls.Config{InsecureSkipVerify: true}}
conn, err = tlsDialer.DialContext(ctx, "tcp", u.Host) conn, err = tlsDialer.DialContext(ctx, "tcp", u.Host)
} else { } else {
@@ -140,9 +145,9 @@ func (a Attacker) handleRedirect(stream *cameradar.Stream, resHeaders base.Heade
} }
switch location.Scheme { switch location.Scheme {
case "rtsps": case rtsps:
stream.Secure = true stream.Secure = true
case "rtsp": case rtsp:
stream.Secure = false stream.Secure = false
} }
@@ -210,9 +215,9 @@ func buildRTSPURL(stream cameradar.Stream, route, username, password string) (*b
host := net.JoinHostPort(stream.Address.String(), strconv.Itoa(int(stream.Port))) host := net.JoinHostPort(stream.Address.String(), strconv.Itoa(int(stream.Port)))
path := "/" + strings.TrimLeft(strings.TrimSpace(route), "/") // Ensure path starts with a single "/" path := "/" + strings.TrimLeft(strings.TrimSpace(route), "/") // Ensure path starts with a single "/"
scheme := "rtsp" scheme := rtsp
if stream.Secure { if stream.Secure {
scheme = "rtsps" scheme = rtsps
} }
u := &url.URL{ u := &url.URL{