add configuration item to specify name of variable used to pass path

This commit is contained in:
Liwei 2020-08-05 04:44:48 +08:00
parent c0b0b23c41
commit 3a508b1d42
5 changed files with 19 additions and 8 deletions

View File

@ -830,7 +830,7 @@ func (c *client) runPlay(path string) {
if confp.RunOnRead != "" {
onReadCmd = exec.Command("/bin/sh", "-c", confp.RunOnRead)
onReadCmd.Env = append(os.Environ(),
"RTSP_SERVER_PATH="+path,
c.p.conf.PathEnvVariable+"="+path,
)
onReadCmd.Stdout = os.Stdout
onReadCmd.Stderr = os.Stderr
@ -932,7 +932,7 @@ func (c *client) runRecord(path string) {
if confp.RunOnPublish != "" {
onPublishCmd = exec.Command("/bin/sh", "-c", confp.RunOnPublish)
onPublishCmd.Env = append(os.Environ(),
"RTSP_SERVER_PATH="+path,
c.p.conf.PathEnvVariable+"="+path,
)
onPublishCmd.Stdout = os.Stdout
onPublishCmd.Stderr = os.Stderr

View File

@ -41,6 +41,7 @@ type conf struct {
RunOnConnect string `yaml:"runOnConnect"`
ReadTimeout time.Duration `yaml:"readTimeout"`
WriteTimeout time.Duration `yaml:"writeTimeout"`
PathEnvVariable string `yaml:"pathEnvVariable"`
AuthMethods []string `yaml:"authMethods"`
authMethodsParsed []gortsplib.AuthMethod ``
Metrics bool `yaml:"metrics"`
@ -132,6 +133,14 @@ func loadConf(fpath string, stdin io.Reader) (*conf, error) {
conf.WriteTimeout = 5 * time.Second
}
if conf.PathEnvVariable == "" {
conf.PathEnvVariable = "RTSP_SERVER_PATH"
}
re := regexp.MustCompile("^[a-zA-Z_][a-zA-Z0-9_]*$")
if !re.MatchString(conf.PathEnvVariable) {
return nil, fmt.Errorf("pathEnvVariable must consist of only alphanumerics and underscores, and should not begin with a digit")
}
if len(conf.AuthMethods) == 0 {
conf.AuthMethods = []string{"basic", "digest"}
}

View File

@ -264,7 +264,7 @@ func newProgram(args []string, stdin io.Reader) (*program, error) {
if confp.RunOnInit != "" {
onInitCmd := exec.Command("/bin/sh", "-c", confp.RunOnInit)
onInitCmd.Env = append(os.Environ(),
"RTSP_SERVER_PATH="+path,
conf.PathEnvVariable+"="+path,
)
onInitCmd.Stdout = os.Stdout
onInitCmd.Stderr = os.Stderr

View File

@ -123,7 +123,7 @@ func (pa *path) describe(client *client) {
pa.lastActivation = time.Now()
pa.onDemandCmd = exec.Command("/bin/sh", "-c", pa.confp.RunOnDemand)
pa.onDemandCmd.Env = append(os.Environ(),
"RTSP_SERVER_PATH="+pa.id,
pa.p.conf.PathEnvVariable+"="+pa.id,
)
pa.onDemandCmd.Stdout = os.Stdout
pa.onDemandCmd.Stderr = os.Stderr

View File

@ -14,6 +14,8 @@ runOnConnect:
readTimeout: 10s
# timeout of write operations
writeTimeout: 5s
# name of environment variable used to pass the path to runOnInit/Demand/Publish/Read
pathEnvVariable: RTSP_SERVER_PATH
# supported authentication methods
# WARNING: both methods are insecure, use RTSP inside a VPN to enforce security.
authMethods: [basic, digest]
@ -43,26 +45,26 @@ paths:
# command to run when this path is loaded by the program.
# this can be used, for example, to publish a stream and keep it always opened.
# This is terminated with SIGINT when the program closes.
# The path is available as an environment variable $RTSP_SERVER_PATH
# The path is available as an environment variable configured by pathEnvVariable
runOnInit:
# command to run when this path is requested.
# This can be used, for example, to publish a stream on demand.
# This is terminated with SIGINT when the path is not requested anymore.
# The actual path from the request (useful for wildcard paths) is available as an
# environment variable $RTSP_SERVER_PATH
# environment variable configured by pathEnvVariable
runOnDemand:
# command to run when a client starts publishing.
# This is terminated with SIGINT when a client stops publishing.
# The actual path from the client (useful for wildcard paths) is available as an
# environment variable $RTSP_SERVER_PATH
# environment variable configured by pathEnvVariable
runOnPublish:
# command to run when a clients starts reading.
# This is terminated with SIGINT when a client stops reading.
# The actual path from the client (useful for wildcard paths) is available as an
# environment variable $RTSP_SERVER_PATH
# environment variable configured by pathEnvVariable
runOnRead:
# username required to publish