87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
/*
|
|
© Copyright IBM Corporation 2017, 2018
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License 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.
|
|
*/
|
|
|
|
// Package command contains code to run external commands
|
|
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"os/user"
|
|
"runtime"
|
|
"strconv"
|
|
"syscall"
|
|
)
|
|
|
|
// RunCmd runs an OS command. On Linux it waits for the command to
|
|
// complete and returns the exit status (return code).
|
|
// Do not use this function to run shell built-ins (like "cd"), because
|
|
// the error handling works differently
|
|
func RunCmd(cmd *exec.Cmd) (string, int, error) {
|
|
// Run the command and wait for completion
|
|
out, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
// Assert that this is an ExitError
|
|
exiterr, ok := err.(*exec.ExitError)
|
|
// If the type assertion was correct, and we're on Linux
|
|
if ok && runtime.GOOS == "linux" {
|
|
status, ok := exiterr.Sys().(syscall.WaitStatus)
|
|
if ok {
|
|
return string(out), status.ExitStatus(), fmt.Errorf("%v: %v", cmd.Path, err)
|
|
}
|
|
}
|
|
return string(out), -1, err
|
|
}
|
|
return string(out), 0, nil
|
|
}
|
|
|
|
// Run runs an OS command. On Linux it waits for the command to
|
|
// complete and returns the exit status (return code).
|
|
// Do not use this function to run shell built-ins (like "cd"), because
|
|
// the error handling works differently
|
|
func Run(name string, arg ...string) (string, int, error) {
|
|
return RunCmd(exec.Command(name, arg...))
|
|
}
|
|
|
|
// RunAsMQM runs the specified command as the mqm user
|
|
func RunAsMQM(name string, arg ...string) (string, int, error) {
|
|
cmd := exec.Command(name, arg...)
|
|
cmd.SysProcAttr = &syscall.SysProcAttr{}
|
|
uid, gid, err := LookupMQM()
|
|
if err != nil {
|
|
return "", 0, err
|
|
}
|
|
cmd.SysProcAttr.Credential = &syscall.Credential{Uid: uint32(uid), Gid: uint32(gid)}
|
|
return RunCmd(cmd)
|
|
}
|
|
|
|
// LookupMQM looks up the UID & GID of the mqm user
|
|
func LookupMQM() (int, int, error) {
|
|
mqm, err := user.Lookup("mqm")
|
|
if err != nil {
|
|
return -1, -1, err
|
|
}
|
|
mqmUID, err := strconv.Atoi(mqm.Uid)
|
|
if err != nil {
|
|
return -1, -1, err
|
|
}
|
|
mqmGID, err := strconv.Atoi(mqm.Gid)
|
|
if err != nil {
|
|
return -1, -1, err
|
|
}
|
|
return mqmUID, mqmGID, nil
|
|
}
|