/
executers_unix.go
75 lines (66 loc) · 2.33 KB
/
executers_unix.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
// Copyright 2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License"). You may not
// use this file except in compliance with the License. A copy of the
// License is located at
//
// http://aws.amazon.com/apache2.0/
//
// or in the "license" file accompanying this file. This file is distributed
// on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
// either express or implied. See the License for the specific language governing
// permissions and limitations under the License.
//go:build darwin || freebsd || linux || netbsd || openbsd
// +build darwin freebsd linux netbsd openbsd
package executers
import (
"os"
"os/exec"
"runtime"
"strings"
"syscall"
"github.com/aws/amazon-ssm-agent/agent/appconfig"
)
func prepareProcess(command *exec.Cmd) {
// make the process the leader of its process group
// (otherwise we cannot kill it properly)
command.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
}
func quiesce() {
if runtime.GOOS != "darwin" {
return
}
syscall.Sync() // workaround for https://github.com/golang/go/issues/33565
}
func killProcess(process *os.Process, signal *timeoutSignal) error {
// NOTE: go only kills the process but not its sub processes.
// The consequence is that command.Wait() does not return, for some reason.
// As a workaround we use some (platform specific) magic:
// syscall.Kill(-pid, syscall.SIGKILL)
// Here '-pid' means that the KILL signal is sent to all processes
// in the process group whose id is 'pid'. 'prepareProcess' makes
// the shell we spawn the leader of its own process group and so
// the kill here not just kills the shell but all its descendant
// processes. [See manpage for kill(2)]
return syscall.Kill(-process.Pid, syscall.SIGKILL) // note the minus sign
}
// Running powershell on linux erquired the HOME env variable to be set and to remove the TERM env variable
func validateEnvironmentVariables(command *exec.Cmd) {
if command.Path == appconfig.PowerShellPluginCommandName {
env := command.Env
env = append(env, fmtEnvVariable("HOME", "/"))
i := 0
for _, a := range env {
if strings.Contains(a, "TERM") {
if i == len(env)-1 {
env = env[:i]
} else {
env = append(env[:i], env[i+1:]...)
}
break
}
i++
}
command.Env = env
}
}