From 3b643b39276d76505c7d6caaf34e6fbf882ca124 Mon Sep 17 00:00:00 2001
From: Bernhard Froehlich <decke@bluelife.at>
Date: Thu, 20 Dec 2018 09:44:41 +0000
Subject: [PATCH] Improve logging and make sure smtpd is also logging

---
 main.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 9c677bb..dc43d4b 100644
--- a/main.go
+++ b/main.go
@@ -52,14 +52,17 @@ func main() {
 
 	iniflags.Parse()
 
+	logwriter := io.Writer(os.Stdout)
+
 	if *logFile != "" {
-		f, err := os.OpenFile(*logFile, os.O_RDWR | os.O_CREATE | os.O_APPEND, 0600)
+		f, err := os.OpenFile(*logFile, os.O_WRONLY | os.O_CREATE | os.O_APPEND, 0600)
 		if err != nil {
 			log.Fatalf("Error opening logfile: %v", err)
 		}
 		defer f.Close()
-		wrt := io.MultiWriter(os.Stdout, f)
-		log.SetOutput(wrt)
+
+		logwriter = io.MultiWriter(os.Stdout, f)
+		log.SetOutput(logwriter)
 	}
 
 	listeners := strings.Split(*listen, " ")
@@ -71,6 +74,7 @@ func main() {
 			Hostname:	*hostName,
 			WelcomeMessage: *welcomeMsg,
 			Handler:        handler,
+			ProtocolLogger: log.New(logwriter, "INBOUND: ", log.Lshortfile),
 		}
 
 		if strings.Index(listeners[i], "://") == -1 {