Skip to content

Commit

Permalink
Log service panics on windows
Browse files Browse the repository at this point in the history
  • Loading branch information
lixmal committed Nov 1, 2024
1 parent ec5095b commit c42c709
Show file tree
Hide file tree
Showing 3 changed files with 89 additions and 0 deletions.
7 changes: 7 additions & 0 deletions client/server/panic_generic.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
//go:build !windows

package server

func handlePanicLog() error {
return nil
}
78 changes: 78 additions & 0 deletions client/server/panic_windows.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
package server

import (
"fmt"
"os"
"path/filepath"
"syscall"

log "github.com/sirupsen/logrus"
)

const (
windowsPanicLogEnvVar = "NB_WINDOWS_PANIC_LOG"
// 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 {
logPath := os.Getenv(windowsPanicLogEnvVar)
if logPath == "" {
return nil
}

// Ensure the directory exists
logDir := filepath.Dir(logPath)
if err := os.MkdirAll(logDir, 0755); err != nil {
return fmt.Errorf("create panic log directory: %v", err)
}

// Open log file with append mode
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: %v", err)
}

// Redirect stderr to the file
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: %v", 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 {
// Get the current process's stderr handle
if err := setStdHandle(f); err != nil {
return fmt.Errorf("failed to set stderr handle: %v", 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
}
4 changes: 4 additions & 0 deletions client/server/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,10 @@ func (s *Server) Start() error {
defer s.mutex.Unlock()
state := internal.CtxGetState(s.rootCtx)

if err := handlePanicLog(); err != nil {
log.Warnf("failed to redirect stderr: %v", err)
}

if err := restoreResidualState(s.rootCtx); err != nil {
log.Warnf(errRestoreResidualState, err)
}
Expand Down

0 comments on commit c42c709

Please sign in to comment.