1 files changed, 23 insertions, 2 deletions

diff --git a/Crypto/Cipher/ThomasToVincent.hs b/Crypto/Cipher/ThomasToVincent.hs
index 5a68cf3..cf5cdee 100644
--- a/Crypto/Cipher/ThomasToVincent.hs
+++ b/Crypto/Cipher/ThomasToVincent.hs
@@ -5,7 +5,12 @@ module Crypto.Cipher.ThomasToVincent where
 import qualified Data.ByteString as S
 import Crypto.Cipher.Types
 import Crypto.Cipher.Cast5
+#if defined(VERSION_cryptonite)
+import qualified Data.ByteArray as Bytes
+import Crypto.Error
+#else
 import Data.Byteable
+#endif
 
 import Data.Tagged
 {-
@@ -19,9 +24,14 @@ endif
 type ThomasToVincent b = b
 
 instance Cast5Bits size => Cipher (Cast5 size) where
-    cipherName _ = "CAST-5"
+    cipherName _ = "CAST-"++show (cast5bits (undefined :: size))
+#if defined(VERSION_cryptonite)
+    cipherInit k = CryptoPassed b
+      where Just b = buildKey (Bytes.convert k)
+#else
     cipherInit k = b
       where Just b = buildKey (toBytes k)
+#endif
     cipherKeySize _ =  KeySizeFixed (bitlen `div` 8)
       where Tagged bitlen = keyLength :: Tagged (Cast5 size) Int
 
@@ -42,14 +52,25 @@ instance Cast5Bits size => BlockCipher (Cast5 size) where
     blockSize _ = bitlen `div` 8
      where Tagged bitlen = Crypto.Cipher.Cast5.blockSize :: Tagged (Cast5 size) Int
 
+    -- ecbEncrypt :: (BlockCipher cipher, ByteArray ba) => cipher -> ba -> ba
     -- modeEcb' :: BlockCipher k => k -> B.ByteString -> B.ByteString
     ecbEncrypt k msg =
-            let chunks = chunkFor' k msg
+#if defined(VERSION_cryptonite)
+            let chunks = chunkFor' k $ Bytes.convert msg
+                    in Bytes.convert $ S.concat $ map (encryptBlock k) chunks
+#else
+            let chunks = chunkFor' k $ msg
                     in S.concat $ map (encryptBlock k) chunks
+#endif
 
     ecbDecrypt k ct =
+#if defined(VERSION_cryptonite)
+        let chunks = chunkFor' k $ Bytes.convert ct
+        in Bytes.convert $ S.concat $ map (decryptBlock k) chunks
+#else
         let chunks = chunkFor' k ct
         in S.concat $ map (decryptBlock k) chunks
+#endif
 
 
 {-