From 16c3a3fba2d52f68558d5c7446ba282ec7264285 Mon Sep 17 00:00:00 2001
From: Boshi Lian <farmer1992@gmail.com>
Date: Tue, 20 Feb 2018 13:02:00 +0800
Subject: [PATCH] better handle unknow cmd

---
 sshpiperd/cmd.go | 34 +++++++++++++++++++++++-----------
 1 file changed, 23 insertions(+), 11 deletions(-)

diff --git a/sshpiperd/cmd.go b/sshpiperd/cmd.go
index ef2c70a8..9463b13b 100644
--- a/sshpiperd/cmd.go
+++ b/sshpiperd/cmd.go
@@ -98,6 +98,29 @@ func main() {
 	addPlugins(parser, "challenger", challenger.All(), func(n string) registry.Plugin { return challenger.Get(n) })
 	addPlugins(parser, "auditor", auditor.All(), func(n string) registry.Plugin { return auditor.Get(n) })
 
+	parser.CommandHandler = func(command flags.Commander, args []string) error {
+
+		// no subcommand called, start to serve
+		if command == nil {
+
+			if len(args) > 0 {
+				return fmt.Errorf("Unknown command %v", args)
+			}
+
+			// init log
+			initLogger(config.Logfile)
+
+			showVersion()
+			dumpConfig()
+
+			startPiper(&config.piperdConfig)
+
+			return nil
+		}
+
+		return command.Execute(args)
+	}
+
 	if _, err := parser.Parse(); err != nil {
 		return
 	}
@@ -115,15 +138,4 @@ func main() {
 		}
 	}
 
-	// init log
-	initLogger(config.Logfile)
-
-	// no subcommand called, start to serve
-	if parser.Active == nil {
-		showVersion()
-		dumpConfig()
-
-		startPiper(&config.piperdConfig)
-	}
-
 }