device-agent: capture panics, log their stacks

Author: sechmann

cmd/device-agent/main.go

@@ -67,6 +67,7 @@ func main() {
 
 	logDir := filepath.Join(cfg.ConfigDir, logger.LogDir)
 	log := logger.SetupLogger(cfg.LogLevel, logDir, logger.Agent).WithField("component", "main")
+	defer logger.CapturePanic(log)
 
 	programContext, programCancel := context.WithCancel(context.Background())
 	handleSignals(log, programCancel)

internal/logger/logger.go

@@ -4,6 +4,7 @@ import (
 	"io"
 	"os"
 	"path/filepath"
+	"runtime/debug"
 	"time"
 
 	"github.com/sirupsen/logrus"
@@ -74,3 +75,9 @@ func Setup(level string) *logrus.Logger {
 
 	return log
 }
+
+func CapturePanic(log logrus.FieldLogger) {
+	if err := recover(); err != nil {
+		log.WithField("stack", string(debug.Stack())).Errorf("recovered from panic, %T: %x", err, err)
+	}
+}