7 changed files with 115 additions and 70 deletions
@ -0,0 +1,41 @@
@@ -0,0 +1,41 @@
|
||||
// +build !windows
|
||||
|
||||
package externalcmd |
||||
|
||||
import ( |
||||
"os" |
||||
"os/exec" |
||||
) |
||||
|
||||
func (e *ExternalCmd) runInner() bool { |
||||
cmd := exec.Command("/bin/sh", "-c", "exec "+e.cmdstr) |
||||
|
||||
cmd.Env = append(os.Environ(), |
||||
"RTSP_PATH="+e.env.Path, |
||||
"RTSP_PORT="+e.env.Port, |
||||
) |
||||
|
||||
cmd.Stdout = os.Stdout |
||||
cmd.Stderr = os.Stderr |
||||
|
||||
err := cmd.Start() |
||||
if err != nil { |
||||
return true |
||||
} |
||||
|
||||
cmdDone := make(chan struct{}) |
||||
go func() { |
||||
defer close(cmdDone) |
||||
cmd.Wait() |
||||
}() |
||||
|
||||
select { |
||||
case <-e.terminate: |
||||
cmd.Process.Signal(os.Interrupt) |
||||
<-cmdDone |
||||
return false |
||||
|
||||
case <-cmdDone: |
||||
return true |
||||
} |
||||
} |
||||
@ -0,0 +1,57 @@
@@ -0,0 +1,57 @@
|
||||
// +build windows
|
||||
|
||||
package externalcmd |
||||
|
||||
import ( |
||||
"os" |
||||
"os/exec" |
||||
"strings" |
||||
|
||||
"github.com/kballard/go-shellquote" |
||||
) |
||||
|
||||
func (e *ExternalCmd) runInner() bool { |
||||
// on Windows the shell is not used and command is started directly
|
||||
// variables are replaced manually in order to guarantee compatibility
|
||||
// with Linux commands
|
||||
tmp := strings.ReplaceAll(e.cmdstr, "$RTSP_PATH", e.env.Path) |
||||
tmp = strings.ReplaceAll(tmp, "$RTSP_PORT", e.env.Port) |
||||
|
||||
parts, err := shellquote.Split(tmp) |
||||
if err != nil { |
||||
return true |
||||
} |
||||
|
||||
cmd := exec.Command(parts[0], parts[1:]...) |
||||
|
||||
cmd.Env = append(os.Environ(), |
||||
"RTSP_PATH="+e.env.Path, |
||||
"RTSP_PORT="+e.env.Port, |
||||
) |
||||
|
||||
cmd.Stdout = os.Stdout |
||||
cmd.Stderr = os.Stderr |
||||
|
||||
err := cmd.Start() |
||||
if err != nil { |
||||
return true |
||||
} |
||||
|
||||
cmdDone := make(chan struct{}) |
||||
go func() { |
||||
defer close(cmdDone) |
||||
cmd.Wait() |
||||
}() |
||||
|
||||
select { |
||||
case <-e.terminate: |
||||
// on Windows it's not possible to send os.Interrupt to a process
|
||||
// Kill() is the only supported way
|
||||
cmd.Process.Kill() |
||||
<-cmdDone |
||||
return false |
||||
|
||||
case <-cmdDone: |
||||
return true |
||||
} |
||||
} |
||||
Loading…
Reference in new issue