mirror of
https://github.com/netbirdio/netbird.git
synced 2025-07-19 07:34:28 +02:00
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"syscall"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/util"
|
|
)
|
|
|
|
const (
|
|
// STD_ERROR_HANDLE ((DWORD)-12) = 4294967284
|
|
stdErrorHandle = ^uintptr(11)
|
|
)
|
|
|
|
var (
|
|
kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
|
|
|
// https://learn.microsoft.com/en-us/windows/console/setstdhandle
|
|
setStdHandleFn = kernel32.NewProc("SetStdHandle")
|
|
)
|
|
|
|
func handlePanicLog() error {
|
|
// TODO: move this to a central location
|
|
logDir := path.Join(os.Getenv("PROGRAMDATA"), "Netbird")
|
|
logPath := path.Join(logDir, "netbird.err")
|
|
|
|
if err := os.MkdirAll(logDir, 0750); err != nil {
|
|
return fmt.Errorf("create panic log directory: %w", err)
|
|
}
|
|
if err := util.EnforcePermission(logPath); err != nil {
|
|
return fmt.Errorf("enforce permission on panic log file: %w", err)
|
|
}
|
|
|
|
f, err := os.OpenFile(logPath, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("open panic log file: %w", err)
|
|
}
|
|
|
|
if err = redirectStderr(f); err != nil {
|
|
if closeErr := f.Close(); closeErr != nil {
|
|
log.Warnf("failed to close file after redirect error: %v", closeErr)
|
|
}
|
|
return fmt.Errorf("redirect stderr: %w", err)
|
|
}
|
|
|
|
log.Infof("successfully configured panic logging to: %s", logPath)
|
|
return nil
|
|
}
|
|
|
|
// redirectStderr redirects stderr to the provided file
|
|
func redirectStderr(f *os.File) error {
|
|
if err := setStdHandle(f); err != nil {
|
|
return fmt.Errorf("failed to set stderr handle: %w", err)
|
|
}
|
|
|
|
// Also set os.Stderr for Go's standard library
|
|
os.Stderr = f
|
|
|
|
return nil
|
|
}
|
|
|
|
func setStdHandle(f *os.File) error {
|
|
handle := f.Fd()
|
|
r0, _, e1 := setStdHandleFn.Call(stdErrorHandle, handle)
|
|
if r0 == 0 {
|
|
if e1 != nil {
|
|
return e1
|
|
}
|
|
return syscall.EINVAL
|
|
}
|
|
return nil
|
|
}
|