add --benchmark

1 files changed, 6 insertions, 4 deletions

diff --git a/keysafe.hs b/keysafe.hs
index 4058487..7de3079 100644
--- a/keysafe.hs
+++ b/keysafe.hs
@@ -13,6 +13,7 @@ import qualified CmdLine
 import UI
 import Encryption
 import Entropy
+import ExpensiveHash
 import Cost
 import Shard
 import Storage
@@ -30,12 +31,13 @@ main = do
 	-- TODO determine gpg key id by examining secret key,
 	-- or retrieving public key from keyserver and examining it.
 	let keyid = KeyId keytype "dummy key id"
+	let tunables = if CmdLine.testMode cmdline
+		then testModeTunables
+		else defaultTunables
 	case CmdLine.mode cmdline of
-		CmdLine.Backup -> storedemo ui keyid $
-			if CmdLine.testMode cmdline
-				then testModeTunables
-				else defaultTunables
+		CmdLine.Backup -> storedemo ui keyid tunables
 		CmdLine.Restore -> retrievedemo ui keyid
+		CmdLine.Benchmark -> benchmarkTunables tunables
 
 storedemo :: UI -> KeyId -> Tunables -> IO ()
 storedemo ui keyid tunables = do