add --benchmark

1 files changed, 6 insertions, 2 deletions

diff --git a/CmdLine.hs b/CmdLine.hs
index b47d609..8e3040a 100644
--- a/CmdLine.hs
+++ b/CmdLine.hs
@@ -17,12 +17,12 @@ data CmdLine = CmdLine
 	}
 	deriving (Show)
 
-data Mode = Backup | Restore
+data Mode = Backup | Restore | Benchmark
 	deriving (Show)
 
 parse :: Parser CmdLine
 parse = CmdLine
-	<$> (backup <|> restore)
+	<$> (backup <|> restore <|> benchmark)
 	<*> keytypeopt
 	<*> testmodeswitch
 	<*> guiswitch
@@ -35,6 +35,10 @@ parse = CmdLine
 		( long "restore"
 		<> help "Retrieve a secret key from keysafe."
 		)
+	benchmark = flag' Benchmark
+		( long "benchmark"
+		<> help "Benchmark speed of keysafe's cryptographic primitives."
+		)
 	keytypeopt = KeyType . BU8.fromString <$> strOption
 		( long "type"
 		<> help "Type of key (eg, \"gpg\")."