2023-03-09 11:41:50 +00:00
|
|
|
//go:build windows
|
|
|
|
|
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"os"
|
|
|
|
"syscall"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
kernel32 = syscall.MustLoadDLL("kernel32.dll")
|
|
|
|
procSetStdHandle = kernel32.MustFindProc("SetStdHandle")
|
|
|
|
)
|
|
|
|
|
|
|
|
func setStdHandle(stdhandle int32, handle syscall.Handle) error {
|
2024-02-25 15:56:37 +00:00
|
|
|
r0, _, e1 := syscall.SyscallN(procSetStdHandle.Addr(), uintptr(stdhandle), uintptr(handle), 0)
|
2023-03-09 11:41:50 +00:00
|
|
|
if r0 == 0 {
|
|
|
|
if e1 != 0 {
|
|
|
|
return error(e1)
|
|
|
|
}
|
|
|
|
return syscall.EINVAL
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func RedirectStderr(f *os.File) {
|
|
|
|
err := setStdHandle(syscall.STD_ERROR_HANDLE, syscall.Handle(f.Fd()))
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Failed to redirect stderr to file: %v", err)
|
|
|
|
}
|
|
|
|
// SetStdHandle does not affect prior references to stderr
|
|
|
|
os.Stderr = f
|
|
|
|
}
|