Alexey Khit
45 lines
855 B
Go
45 lines
855 B
Go
package rtsp
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"net"
|
|
"net/url"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func Dial(uri string) (net.Conn, error) {
|
|
u, err := url.Parse(uri)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch u.Scheme {
|
|
case "rtsp":
|
|
return dialTCP(u.Host, nil)
|
|
case "rtsps":
|
|
tlsConf := &tls.Config{ServerName: u.Hostname()}
|
|
return dialTCP(u.Host, tlsConf)
|
|
case "rtspx":
|
|
tlsConf := &tls.Config{InsecureSkipVerify: true}
|
|
return dialTCP(u.Host, tlsConf)
|
|
}
|
|
|
|
return nil, errors.New("unsupported scheme: " + u.Scheme)
|
|
}
|
|
|
|
func dialTCP(address string, tlsConf *tls.Config) (net.Conn, error) {
|
|
if strings.IndexByte(address, ':') < 0 {
|
|
address += ":554"
|
|
}
|
|
|
|
conn, err := net.DialTimeout("tcp", address, time.Second*5)
|
|
if tlsConf == nil || err != nil {
|
|
return conn, err
|
|
}
|
|
|
|
tlsConn := tls.Client(conn, tlsConf)
|
|
return tlsConn, tlsConn.Handshake()
|
|
}
|