Alexey Khit
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"errors"
|
|
"net"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
|
"github.com/AlexxIT/go2rtc/pkg/flv"
|
|
)
|
|
|
|
func Dial(rawURL string) (core.Producer, error) {
|
|
u, err := url.Parse(rawURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var hostname string // without port
|
|
if i := strings.IndexByte(u.Host, ':'); i > 0 {
|
|
hostname = u.Host[:i]
|
|
} else {
|
|
hostname = u.Host
|
|
u.Host += ":1935"
|
|
}
|
|
|
|
conn, err := net.DialTimeout("tcp", u.Host, core.ConnDialTimeout)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if u.Scheme != "rtmp" {
|
|
var conf *tls.Config
|
|
|
|
switch {
|
|
case u.Scheme == "rtmpx" || net.ParseIP(hostname) != nil:
|
|
conf = &tls.Config{InsecureSkipVerify: true}
|
|
case u.Scheme == "rtmps":
|
|
conf = &tls.Config{ServerName: hostname}
|
|
default:
|
|
return nil, errors.New("unsupported scheme: " + u.Scheme)
|
|
}
|
|
|
|
tlsConn := tls.Client(conn, conf)
|
|
if err = tlsConn.Handshake(); err != nil {
|
|
return nil, err
|
|
}
|
|
conn = tlsConn
|
|
}
|
|
|
|
rd, err := NewReader(u, conn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return flv.Open(rd)
|
|
}
|