[cmd/tunneld] add syslog option

1 files changed, 15 insertions, 6 deletions

diff --git a/cmd/tunneld/main.go b/cmd/tunneld/main.go
index cd8afda..6a16dec 100644
--- a/cmd/tunneld/main.go
+++ b/cmd/tunneld/main.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"flag"
 	"log"
+	"log/syslog"
 	"os"
 	"os/signal"
 	"path"
@@ -12,20 +13,28 @@ import (
 	"tunnel/pkg/server"
 )
 
-var debugVar bool
-
-func init() {
-	flag.BoolVar(&debugVar, "d", false, "debug: print time and source info")
-}
+var (
+	debugFlag  = flag.Bool("d", false, "debug: print time and source info")
+	syslogFlag = flag.Bool("s", false, "log output to syslog instead of stdout")
+)
 
 func initLog() {
 	var logFlags int
 
-	if debugVar {
+	if *debugFlag {
 		logFlags |= log.Ldate | log.Ltime | log.Lshortfile
 	}
 
 	log.SetFlags(logFlags)
+
+	if *syslogFlag {
+		sysLog, err := syslog.New(syslog.LOG_INFO, "tunneld")
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		log.SetOutput(sysLog)
+	}
 }
 
 func sighandler(c chan os.Signal, s *server.Server) {