mirror of
https://github.com/siderolabs/talos.git
synced 2025-08-20 14:11:11 +02:00
d430a37e46
This removes the github.com/pkg/errors package in favor of the official error wrapping in go 1.13. Signed-off-by: Andrew Rynhard <andrew@andrewrynhard.com>
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
package reaper
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"syscall"
|
|
)
|
|
|
|
// WaitWrapper emulates os/exec.Command.Wait() when reaper is running.
|
|
//
|
|
// WaitWrapper(true, cmd) should be equivalent to cmd.Wait().
|
|
func WaitWrapper(usingReaper bool, notifyCh <-chan ProcessInfo, cmd *exec.Cmd) error {
|
|
if !usingReaper {
|
|
return cmd.Wait()
|
|
}
|
|
|
|
var info ProcessInfo
|
|
|
|
for info = range notifyCh {
|
|
if info.Pid == cmd.Process.Pid && (info.Status.Exited() || info.Status.Signaled()) {
|
|
break
|
|
}
|
|
}
|
|
|
|
err := convertWaitStatus(info.Status)
|
|
|
|
// still do cmd.Wait() to release any resources
|
|
waitErr := cmd.Wait()
|
|
if err == nil && waitErr != nil && waitErr.Error() != "waitid: no child processes" {
|
|
err = waitErr
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func convertWaitStatus(status syscall.WaitStatus) error {
|
|
if status.Signaled() {
|
|
return fmt.Errorf("signal: %s", status.Signal())
|
|
}
|
|
|
|
if status.Exited() && status.ExitStatus() != 0 {
|
|
return fmt.Errorf("exit status %d", status.ExitStatus())
|
|
}
|
|
|
|
return nil
|
|
}
|