aler9
5 years ago
4 changed files with 206 additions and 167 deletions
@ -0,0 +1,199 @@
@@ -0,0 +1,199 @@
|
||||
package main |
||||
|
||||
import ( |
||||
"fmt" |
||||
"io" |
||||
"os" |
||||
"regexp" |
||||
"time" |
||||
|
||||
"github.com/aler9/gortsplib" |
||||
"gopkg.in/yaml.v2" |
||||
) |
||||
|
||||
type ConfPath struct { |
||||
Source string `yaml:"source"` |
||||
SourceProtocol string `yaml:"sourceProtocol"` |
||||
PublishUser string `yaml:"publishUser"` |
||||
PublishPass string `yaml:"publishPass"` |
||||
PublishIps []string `yaml:"publishIps"` |
||||
publishIpsParsed []interface{} |
||||
ReadUser string `yaml:"readUser"` |
||||
ReadPass string `yaml:"readPass"` |
||||
ReadIps []string `yaml:"readIps"` |
||||
readIpsParsed []interface{} |
||||
} |
||||
|
||||
type conf struct { |
||||
Protocols []string `yaml:"protocols"` |
||||
protocolsParsed map[streamProtocol]struct{} |
||||
RtspPort int `yaml:"rtspPort"` |
||||
RtpPort int `yaml:"rtpPort"` |
||||
RtcpPort int `yaml:"rtcpPort"` |
||||
PreScript string `yaml:"preScript"` |
||||
PostScript string `yaml:"postScript"` |
||||
ReadTimeout time.Duration `yaml:"readTimeout"` |
||||
WriteTimeout time.Duration `yaml:"writeTimeout"` |
||||
AuthMethods []string `yaml:"authMethods"` |
||||
authMethodsParsed []gortsplib.AuthMethod |
||||
Pprof bool `yaml:"pprof"` |
||||
Paths map[string]*ConfPath `yaml:"paths"` |
||||
} |
||||
|
||||
func loadConf(fpath string, stdin io.Reader) (*conf, error) { |
||||
conf := &conf{} |
||||
|
||||
err := func() error { |
||||
if fpath == "stdin" { |
||||
err := yaml.NewDecoder(stdin).Decode(conf) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
return nil |
||||
|
||||
} else { |
||||
// conf.yml is optional
|
||||
if fpath == "conf.yml" { |
||||
if _, err := os.Stat(fpath); err != nil { |
||||
return nil |
||||
} |
||||
} |
||||
|
||||
f, err := os.Open(fpath) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
defer f.Close() |
||||
|
||||
err = yaml.NewDecoder(f).Decode(conf) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
}() |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if len(conf.Protocols) == 0 { |
||||
conf.Protocols = []string{"udp", "tcp"} |
||||
} |
||||
conf.protocolsParsed = make(map[streamProtocol]struct{}) |
||||
for _, proto := range conf.Protocols { |
||||
switch proto { |
||||
case "udp": |
||||
conf.protocolsParsed[_STREAM_PROTOCOL_UDP] = struct{}{} |
||||
|
||||
case "tcp": |
||||
conf.protocolsParsed[_STREAM_PROTOCOL_TCP] = struct{}{} |
||||
|
||||
default: |
||||
return nil, fmt.Errorf("unsupported protocol: %s", proto) |
||||
} |
||||
} |
||||
if len(conf.protocolsParsed) == 0 { |
||||
return nil, fmt.Errorf("no protocols provided") |
||||
} |
||||
|
||||
if conf.RtspPort == 0 { |
||||
conf.RtspPort = 8554 |
||||
} |
||||
if conf.RtpPort == 0 { |
||||
conf.RtpPort = 8000 |
||||
} |
||||
if (conf.RtpPort % 2) != 0 { |
||||
return nil, fmt.Errorf("rtp port must be even") |
||||
} |
||||
if conf.RtcpPort == 0 { |
||||
conf.RtcpPort = 8001 |
||||
} |
||||
if conf.RtcpPort != (conf.RtpPort + 1) { |
||||
return nil, fmt.Errorf("rtcp and rtp ports must be consecutive") |
||||
} |
||||
|
||||
if conf.ReadTimeout == 0 { |
||||
conf.ReadTimeout = 5 * time.Second |
||||
} |
||||
if conf.WriteTimeout == 0 { |
||||
conf.WriteTimeout = 5 * time.Second |
||||
} |
||||
|
||||
if len(conf.AuthMethods) == 0 { |
||||
conf.AuthMethods = []string{"basic", "digest"} |
||||
} |
||||
for _, method := range conf.AuthMethods { |
||||
switch method { |
||||
case "basic": |
||||
conf.authMethodsParsed = append(conf.authMethodsParsed, gortsplib.Basic) |
||||
|
||||
case "digest": |
||||
conf.authMethodsParsed = append(conf.authMethodsParsed, gortsplib.Digest) |
||||
|
||||
default: |
||||
return nil, fmt.Errorf("unsupported authentication method: %s", method) |
||||
} |
||||
} |
||||
|
||||
if len(conf.Paths) == 0 { |
||||
conf.Paths = map[string]*ConfPath{ |
||||
"all": {}, |
||||
} |
||||
} |
||||
|
||||
for path, pconf := range conf.Paths { |
||||
if pconf.Source == "" { |
||||
pconf.Source = "record" |
||||
} |
||||
|
||||
if pconf.PublishUser != "" { |
||||
if !regexp.MustCompile("^[a-zA-Z0-9]+$").MatchString(pconf.PublishUser) { |
||||
return nil, fmt.Errorf("publish username must be alphanumeric") |
||||
} |
||||
} |
||||
if pconf.PublishPass != "" { |
||||
if !regexp.MustCompile("^[a-zA-Z0-9]+$").MatchString(pconf.PublishPass) { |
||||
return nil, fmt.Errorf("publish password must be alphanumeric") |
||||
} |
||||
} |
||||
pconf.publishIpsParsed, err = parseIpCidrList(pconf.PublishIps) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if pconf.ReadUser != "" && pconf.ReadPass == "" || pconf.ReadUser == "" && pconf.ReadPass != "" { |
||||
return nil, fmt.Errorf("read username and password must be both filled") |
||||
} |
||||
if pconf.ReadUser != "" { |
||||
if !regexp.MustCompile("^[a-zA-Z0-9]+$").MatchString(pconf.ReadUser) { |
||||
return nil, fmt.Errorf("read username must be alphanumeric") |
||||
} |
||||
} |
||||
if pconf.ReadPass != "" { |
||||
if !regexp.MustCompile("^[a-zA-Z0-9]+$").MatchString(pconf.ReadPass) { |
||||
return nil, fmt.Errorf("read password must be alphanumeric") |
||||
} |
||||
} |
||||
if pconf.ReadUser != "" && pconf.ReadPass == "" || pconf.ReadUser == "" && pconf.ReadPass != "" { |
||||
return nil, fmt.Errorf("read username and password must be both filled") |
||||
} |
||||
pconf.readIpsParsed, err = parseIpCidrList(pconf.ReadIps) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
|
||||
if pconf.Source != "record" { |
||||
if path == "all" { |
||||
return nil, fmt.Errorf("path 'all' cannot have a RTSP source") |
||||
} |
||||
|
||||
if pconf.SourceProtocol == "" { |
||||
pconf.SourceProtocol = "udp" |
||||
} |
||||
} |
||||
} |
||||
|
||||
return conf, nil |
||||
} |
||||
Loading…
Reference in new issue