mirror of
https://github.com/safing/portbase
synced 2025-09-01 18:19:57 +00:00
48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package osdetail
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
// RunPowershellCmd runs a powershell command and returns its output.
|
|
func RunPowershellCmd(script string) (output string, err error) {
|
|
// Create command to execute.
|
|
cmd := exec.Command(
|
|
"powershell.exe",
|
|
"-NoProfile",
|
|
"-NonInteractive",
|
|
"[System.Console]::OutputEncoding = [System.Text.Encoding]::UTF8\n"+script,
|
|
)
|
|
|
|
// Create and assign output buffers.
|
|
var stdoutBuf bytes.Buffer
|
|
var stderrBuf bytes.Buffer
|
|
cmd.Stdout = &stdoutBuf
|
|
cmd.Stderr = &stderrBuf
|
|
|
|
// Run command and collect output.
|
|
err = cmd.Run()
|
|
stdout, stderr := stdoutBuf.String(), stderrBuf.String()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
// Powershell might not return an error, but just write to stdout instead.
|
|
if stderr != "" {
|
|
return "", errors.New(strings.SplitN(stderr, "\n", 2)[0])
|
|
}
|
|
|
|
// Debugging output:
|
|
// fmt.Printf("powershell stdout: %s\n", stdout)
|
|
// fmt.Printf("powershell stderr: %s\n", stderr)
|
|
|
|
// Finalize stdout.
|
|
cleanedOutput := strings.TrimSpace(stdout)
|
|
if cleanedOutput == "" {
|
|
return "", ErrEmptyOutput
|
|
}
|
|
|
|
return cleanedOutput, nil
|
|
}
|