improve types

1 files changed, 8 insertions, 1 deletions

diff --git a/Cost.hs b/Cost.hs
index 0425707..5fd003f 100644
--- a/Cost.hs
+++ b/Cost.hs
@@ -1,4 +1,4 @@
-{-# LANGUAGE GeneralizedNewtypeDeriving #-}
+{-# LANGUAGE GeneralizedNewtypeDeriving, MultiParamTypeClasses #-}
 
 module Cost where
 
@@ -74,3 +74,10 @@ data UnknownPassword
 bruteForceLinearSearch :: Cost step -> CostCalc BruteForceOp t
 bruteForceLinearSearch stepcost e = 
 	castCost stepcost `raiseCostPower` reduceEntropy e 1
+
+-- | Things that can be brute-forced expose a CostCalc.
+class Bruteforceable t a where
+	getBruteCostCalc :: t -> CostCalc BruteForceOp a
+
+estimateBruteforceOf :: Bruteforceable t a => t -> Entropy a -> Cost BruteForceOp
+estimateBruteforceOf t e = getBruteCostCalc t e