1 files changed, 32 insertions, 20 deletions

diff --git a/src/block/lion/lion.cpp b/src/block/lion/lion.cpp
index c7cdf6d13..83c1e3aa3 100644
--- a/src/block/lion/lion.cpp
+++ b/src/block/lion/lion.cpp
@@ -14,41 +14,53 @@ namespace Botan {
 /*
 * Lion Encryption
 */
-void Lion::enc(const byte in[], byte out[]) const
+void Lion::encrypt_n(const byte in[], byte out[], u32bit blocks) const
    {
    SecureVector<byte> buffer(LEFT_SIZE);
 
-   xor_buf(buffer, in, key1, LEFT_SIZE);
-   cipher->set_key(buffer, LEFT_SIZE);
-   cipher->encrypt(in + LEFT_SIZE, out + LEFT_SIZE, RIGHT_SIZE);
+   for(u32bit i = 0; i != blocks; ++i)
+      {
+      xor_buf(buffer, in, key1, LEFT_SIZE);
+      cipher->set_key(buffer, LEFT_SIZE);
+      cipher->encrypt(in + LEFT_SIZE, out + LEFT_SIZE, RIGHT_SIZE);
 
-   hash->update(out + LEFT_SIZE, RIGHT_SIZE);
-   hash->final(buffer);
-   xor_buf(out, in, buffer, LEFT_SIZE);
+      hash->update(out + LEFT_SIZE, RIGHT_SIZE);
+      hash->final(buffer);
+      xor_buf(out, in, buffer, LEFT_SIZE);
 
-   xor_buf(buffer, out, key2, LEFT_SIZE);
-   cipher->set_key(buffer, LEFT_SIZE);
-   cipher->encrypt(out + LEFT_SIZE, RIGHT_SIZE);
+      xor_buf(buffer, out, key2, LEFT_SIZE);
+      cipher->set_key(buffer, LEFT_SIZE);
+      cipher->encrypt(out + LEFT_SIZE, RIGHT_SIZE);
+
+      in += BLOCK_SIZE;
+      out += BLOCK_SIZE;
+      }
    }
 
 /*
 * Lion Decryption
 */
-void Lion::dec(const byte in[], byte out[]) const
+void Lion::decrypt_n(const byte in[], byte out[], u32bit blocks) const
    {
    SecureVector<byte> buffer(LEFT_SIZE);
 
-   xor_buf(buffer, in, key2, LEFT_SIZE);
-   cipher->set_key(buffer, LEFT_SIZE);
-   cipher->encrypt(in + LEFT_SIZE, out + LEFT_SIZE, RIGHT_SIZE);
+   for(u32bit i = 0; i != blocks; ++i)
+      {
+      xor_buf(buffer, in, key2, LEFT_SIZE);
+      cipher->set_key(buffer, LEFT_SIZE);
+      cipher->encrypt(in + LEFT_SIZE, out + LEFT_SIZE, RIGHT_SIZE);
+
+      hash->update(out + LEFT_SIZE, RIGHT_SIZE);
+      hash->final(buffer);
+      xor_buf(out, in, buffer, LEFT_SIZE);
 
-   hash->update(out + LEFT_SIZE, RIGHT_SIZE);
-   hash->final(buffer);
-   xor_buf(out, in, buffer, LEFT_SIZE);
+      xor_buf(buffer, out, key1, LEFT_SIZE);
+      cipher->set_key(buffer, LEFT_SIZE);
+      cipher->encrypt(out + LEFT_SIZE, RIGHT_SIZE);
 
-   xor_buf(buffer, out, key1, LEFT_SIZE);
-   cipher->set_key(buffer, LEFT_SIZE);
-   cipher->encrypt(out + LEFT_SIZE, RIGHT_SIZE);
+      in += BLOCK_SIZE;
+      out += BLOCK_SIZE;
+      }
    }
 
 /*