Mirrored_Repos/rmmagent
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add env to exec

Browse Source
This commit is contained in:
wh1te909 2022-09-11 01:34:04 +00:00
parent 89d7ec8242
commit e8c743c197
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
agent/agent.go
View File

@ -205,6 +205,7 @@ type CmdOptions struct {
IsScript bool IsScript bool
IsExecutable bool IsExecutable bool
Detached bool Detached bool
Env []string
} }
func (a *Agent) NewCMDOpts() *CmdOptions { func (a *Agent) NewCMDOpts() *CmdOptions {
@ -228,11 +229,16 @@ func (a *Agent) CmdV2(c *CmdOptions) CmdStatus {
// have a child process that is in a different process group so that // have a child process that is in a different process group so that
// parent terminating doesn't kill child // parent terminating doesn't kill child
if c.Detached { if c.Detached {
cmdOptions.BeforeExec = []func(cmd *exec.Cmd){ cmdOptions.BeforeExec = append(cmdOptions.BeforeExec, func(cmd *exec.Cmd) {
func(cmd *exec.Cmd) {
cmd.SysProcAttr = SetDetached() cmd.SysProcAttr = SetDetached()
}, })
} }
if len(c.Env) > 0 {
cmdOptions.BeforeExec = append(cmdOptions.BeforeExec, func(cmd *exec.Cmd) {
cmd.Env = os.Environ()
cmd.Env = append(cmd.Env, c.Env...)
})
} }
var envCmd *gocmd.Cmd var envCmd *gocmd.Cmd