Add support custom timeout for RTSP source

internal/rtsp/rtsp.go

@@ -102,6 +102,7 @@ func rtspHandler(rawURL string) (core.Producer, error) {
 		query := streams.ParseQuery(rawQuery)
 		conn.Backchannel = query.Get("backchannel") == "1"
 		conn.Media = query.Get("media")
+		conn.Timeout = core.Atoi(query.Get("timeout"))
 		conn.Transport = query.Get("transport")
 	}
 

pkg/core/helpers.go

@@ -63,7 +63,9 @@ func Between(s, sub1, sub2 string) string {
 }
 
 func Atoi(s string) (i int) {
-	i, _ = strconv.Atoi(s)
+	if s != "" {
+		i, _ = strconv.Atoi(s)
+	}
 	return
 }
 

pkg/rtmp/producer.go

@@ -14,7 +14,7 @@ func Dial(rawURL string) (core.Producer, error) {
 		return nil, err
 	}
 
-	conn, err := tcp.Dial(u, "1935")
+	conn, err := tcp.Dial(u, "1935", core.ConnDialTimeout)
 	if err != nil {
 		return nil, err
 	}

pkg/rtsp/client.go

@@ -31,7 +31,11 @@ func (c *Conn) Dial() (err error) {
 	var conn net.Conn
 
 	if c.Transport == "" {
-		conn, err = tcp.Dial(c.URL, "554")
+		timeout := core.ConnDialTimeout
+		if c.Timeout != 0 {
+			timeout = time.Second * time.Duration(c.Timeout)
+		}
+		conn, err = tcp.Dial(c.URL, "554", timeout)
 	} else {
 		conn, err = websocket.Dial(c.Transport)
 	}

pkg/rtsp/conn.go

@@ -26,6 +26,7 @@ type Conn struct {
 	Media       string
 	PacketSize  uint16
 	SessionName string
+	Timeout     int
 	Transport   string // custom transport support, ex. RTSP over WebSocket
 
 	Medias    []*core.Media
@@ -108,13 +109,17 @@ func (c *Conn) Handle() (err error) {
 		}
 		keepaliveTS = time.Now().Add(keepaliveDT)
 
-		// polling frames from remote RTSP Server (ex Camera)
-		if len(c.receivers) > 0 {
-			// if we receiving video/audio from camera
-			timeout = time.Second * 5
+		if c.Timeout == 0 {
+			// polling frames from remote RTSP Server (ex Camera)
+			if len(c.receivers) > 0 {
+				// if we receiving video/audio from camera
+				timeout = time.Second * 5
+			} else {
+				// if we only send audio to camera
+				timeout = time.Second * 30
+			}
 		} else {
-			// if we only send audio to camera
-			timeout = time.Second * 30
+			timeout = time.Second * time.Duration(c.Timeout)
 		}
 
 	case core.ModePassiveProducer:

pkg/tcp/dial.go

@@ -6,12 +6,11 @@ import (
 	"net"
 	"net/url"
 	"strings"
-
-	"github.com/AlexxIT/go2rtc/pkg/core"
+	"time"
 )
 
 // Dial - for RTSP(S|X) and RTMP(S|X)
-func Dial(u *url.URL, port string) (net.Conn, error) {
+func Dial(u *url.URL, port string, timeout time.Duration) (net.Conn, error) {
 	var hostname string // without port
 	if i := strings.IndexByte(u.Host, ':'); i > 0 {
 		hostname = u.Host[:i]
@@ -34,7 +33,7 @@ func Dial(u *url.URL, port string) (net.Conn, error) {
 		return nil, errors.New("unsupported scheme: " + u.Scheme)
 	}
 
-	conn, err := net.DialTimeout("tcp", u.Host, core.ConnDialTimeout)
+	conn, err := net.DialTimeout("tcp", u.Host, timeout)
 	if err != nil {
 		return nil, err
 	}