@ -12,7 +12,8 @@ import (
"gopkg.in/yaml.v2" "gopkg.in/yaml.v2"
) )
type confPath struct { type pathConf struct {
regexp * regexp . Regexp
Source string ` yaml:"source" ` Source string ` yaml:"source" `
sourceUrl * url . URL ` ` sourceUrl * url . URL ` `
SourceProtocol string ` yaml:"sourceProtocol" ` SourceProtocol string ` yaml:"sourceProtocol" `
@ -48,7 +49,7 @@ type conf struct {
LogDestinations [ ] string ` yaml:"logDestinations" ` LogDestinations [ ] string ` yaml:"logDestinations" `
logDestinationsParsed map [ logDestination ] struct { } ` ` logDestinationsParsed map [ logDestination ] struct { } ` `
LogFile string ` yaml:"logFile" ` LogFile string ` yaml:"logFile" `
Paths map [ string ] * confPath ` yaml:"paths" ` Paths map [ string ] * pathConf ` yaml:"paths" `
} }
func loadConf ( fpath string , stdin io . Reader ) ( * conf , error ) { func loadConf ( fpath string , stdin io . Reader ) ( * conf , error ) {
@ -172,105 +173,147 @@ func loadConf(fpath string, stdin io.Reader) (*conf, error) {
} }
if len ( conf . Paths ) == 0 { if len ( conf . Paths ) == 0 {
conf . Paths = map [ string ] * confPath { conf . Paths = map [ string ] * pathConf {
"all" : { } , "all" : { } ,
} }
} }
for name , confp := range conf . Paths { // "all" is an alias for "~^.*$"
if confp == nil { if _ , ok := conf . Paths [ "all" ] ; ok {
conf . Paths [ name ] = & confPath { } conf . Paths [ "~^.*$" ] = conf . Paths [ "all" ]
confp = conf . Paths [ name ] delete ( conf . Paths , "all" )
} }
for name , pconf := range conf . Paths {
if pconf == nil {
conf . Paths [ name ] = & pathConf { }
pconf = conf . Paths [ name ]
}
if name == "" {
return nil , fmt . Errorf ( "path name can not be empty" )
}
// normal path
if name [ 0 ] != '~' {
err := checkPathName ( name ) err := checkPathName ( name )
if err != nil { if err != nil {
return nil , fmt . Errorf ( "invalid path name: %s (%s)" , err , name ) return nil , fmt . Errorf ( "invalid path name: %s (%s)" , err , name )
} }
if confp . Source == "" { // regular expression path
confp . Source = "record" } else {
pathRegexp , err := regexp . Compile ( name [ 1 : ] )
if err != nil {
return nil , fmt . Errorf ( "invalid regular expression: %s" , name [ 1 : ] )
}
pconf . regexp = pathRegexp
}
if pconf . Source == "" {
pconf . Source = "record"
} }
if confp . Source != "record" { if p conf. Source != "record" {
if name == "all" { if pconf . regexp != nil {
return nil , fmt . Errorf ( "path 'all' cannot have a RTSP source; use another path" ) return nil , fmt . Errorf ( "a path with a regular expression cannot have a RTSP source; use another path" )
} }
confp . sourceUrl , err = url . Parse ( confp . Source ) p conf. sourceUrl , err = url . Parse ( p conf. Source )
if err != nil { if err != nil {
return nil , fmt . Errorf ( "'%s' is not a valid RTSP url" , confp . Source ) return nil , fmt . Errorf ( "'%s' is not a valid RTSP url" , p conf. Source )
} }
if confp . sourceUrl . Scheme != "rtsp" { if p conf. sourceUrl . Scheme != "rtsp" {
return nil , fmt . Errorf ( "'%s' is not a valid RTSP url" , confp . Source ) return nil , fmt . Errorf ( "'%s' is not a valid RTSP url" , p conf. Source )
} }
if confp . sourceUrl . Port ( ) == "" { if p conf. sourceUrl . Port ( ) == "" {
confp . sourceUrl . Host += ":554" p conf. sourceUrl . Host += ":554"
} }
if confp . sourceUrl . User != nil { if p conf. sourceUrl . User != nil {
pass , _ := confp . sourceUrl . User . Password ( ) pass , _ := p conf. sourceUrl . User . Password ( )
user := confp . sourceUrl . User . Username ( ) user := p conf. sourceUrl . User . Username ( )
if user != "" && pass == "" || if user != "" && pass == "" ||
user == "" && pass != "" { user == "" && pass != "" {
fmt . Errorf ( "username and password must be both provided" ) fmt . Errorf ( "username and password must be both provided" )
} }
} }
if confp . SourceProtocol == "" { if p conf. SourceProtocol == "" {
confp . SourceProtocol = "udp" p conf. SourceProtocol = "udp"
} }
switch confp . SourceProtocol { switch p conf. SourceProtocol {
case "udp" : case "udp" :
confp . sourceProtocolParsed = gortsplib . StreamProtocolUDP p conf. sourceProtocolParsed = gortsplib . StreamProtocolUDP
case "tcp" : case "tcp" :
confp . sourceProtocolParsed = gortsplib . StreamProtocolTCP p conf. sourceProtocolParsed = gortsplib . StreamProtocolTCP
default : default :
return nil , fmt . Errorf ( "unsupported protocol '%s'" , confp . SourceProtocol ) return nil , fmt . Errorf ( "unsupported protocol '%s'" , p conf. SourceProtocol )
} }
} }
if confp . PublishUser != "" { if p conf. PublishUser != "" {
if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( confp . PublishUser ) { if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( p conf. PublishUser ) {
return nil , fmt . Errorf ( "publish username must be alphanumeric" ) return nil , fmt . Errorf ( "publish username must be alphanumeric" )
} }
} }
if confp . PublishPass != "" { if p conf. PublishPass != "" {
if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( confp . PublishPass ) { if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( p conf. PublishPass ) {
return nil , fmt . Errorf ( "publish password must be alphanumeric" ) return nil , fmt . Errorf ( "publish password must be alphanumeric" )
} }
} }
confp . publishIpsParsed , err = parseIpCidrList ( confp . PublishIps ) p conf. publishIpsParsed , err = parseIpCidrList ( p conf. PublishIps )
if err != nil { if err != nil {
return nil , err return nil , err
} }
if confp . ReadUser != "" && confp . ReadPass == "" || confp . ReadUser == "" && confp . ReadPass != "" { if p conf. ReadUser != "" && p conf. ReadPass == "" || p conf. ReadUser == "" && p conf. ReadPass != "" {
return nil , fmt . Errorf ( "read username and password must be both filled" ) return nil , fmt . Errorf ( "read username and password must be both filled" )
} }
if confp . ReadUser != "" { if p conf. ReadUser != "" {
if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( confp . ReadUser ) { if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( p conf. ReadUser ) {
return nil , fmt . Errorf ( "read username must be alphanumeric" ) return nil , fmt . Errorf ( "read username must be alphanumeric" )
} }
} }
if confp . ReadPass != "" { if p conf. ReadPass != "" {
if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( confp . ReadPass ) { if ! regexp . MustCompile ( "^[a-zA-Z0-9]+$" ) . MatchString ( p conf. ReadPass ) {
return nil , fmt . Errorf ( "read password must be alphanumeric" ) return nil , fmt . Errorf ( "read password must be alphanumeric" )
} }
} }
if confp . ReadUser != "" && confp . ReadPass == "" || confp . ReadUser == "" && confp . ReadPass != "" { if p conf. ReadUser != "" && p conf. ReadPass == "" || p conf. ReadUser == "" && p conf. ReadPass != "" {
return nil , fmt . Errorf ( "read username and password must be both filled" ) return nil , fmt . Errorf ( "read username and password must be both filled" )
} }
confp . readIpsParsed , err = parseIpCidrList ( confp . ReadIps ) p conf. readIpsParsed , err = parseIpCidrList ( p conf. ReadIps )
if err != nil { if err != nil {
return nil , err return nil , err
} }
if name == "all" && confp . RunOnInit != "" { if pconf . regexp != nil && p conf. RunOnInit != "" {
return nil , fmt . Errorf ( "path 'all' does not support option 'runOnInit'; use another path" ) return nil , fmt . Errorf ( "a path with a regular expression does not support option 'runOnInit'; use another path" )
} }
} }
return conf , nil return conf , nil
} }
func ( conf * conf ) checkPathNameAndFindConf ( name string ) ( * pathConf , error ) {
err := checkPathName ( name )
if err != nil {
return nil , fmt . Errorf ( "invalid path name: %s (%s)" , err , name )
}
// normal path
if pconf , ok := conf . Paths [ name ] ; ok {
return pconf , nil
}
// regular expression path
for _ , pconf := range conf . Paths {
if pconf . regexp != nil && pconf . regexp . MatchString ( name ) {
return pconf , nil
}
}
return nil , fmt . Errorf ( "unable to find a valid configuration for path '%s'" , name )
}
aler9
5 years ago