chore(pe): Route saferwall/pe log messages to logrus

rabbitstack · rabbitstack · commit c16127377d51 · 2024-10-16T22:18:58.000+02:00
diff --git a/pkg/pe/parser.go b/pkg/pe/parser.go
@@ -28,6 +28,8 @@ import (
 	"github.com/rabbitstack/fibratus/pkg/util/format"
 	"github.com/rabbitstack/fibratus/pkg/util/va"
 	peparser "github.com/saferwall/pe"
+	peparserlog "github.com/saferwall/pe/log"
+	log "github.com/sirupsen/logrus"
 	"golang.org/x/sys/windows"
 	"os"
 	"path/filepath"
@@ -220,9 +222,29 @@ func newParserOpts(opts opts) *peparser.Options {
 		OmitLoadConfigDirectory:   true,
 		OmitGlobalPtrDirectory:    true,
 		SectionEntropy:            opts.sectionEntropy,
+		Logger:                    &Logger{},
 	}
 }
 
+// Logger is the adapter for routing PE package logs to logrus.
+type Logger struct{}
+
+func (l Logger) Log(level peparserlog.Level, keyvals ...interface{}) error {
+	switch level {
+	case peparserlog.LevelDebug:
+		log.Debug(keyvals[1:]...)
+	case peparserlog.LevelInfo:
+		log.Info(keyvals[1:]...)
+	case peparserlog.LevelWarn:
+		log.Warn(keyvals[1:]...)
+	case peparserlog.LevelError, peparserlog.LevelFatal:
+		log.Error(keyvals[1:]...)
+	default:
+		log.Info(keyvals[1:]...)
+	}
+	return nil
+}
+
 func parse(path string, data []byte, options ...Option) (*PE, error) {
 	var opts opts
 	for _, opt := range options {
