2 files changed, 54 insertions, 42 deletions

diff --git a/src/block/noekeon/noekeon.cpp b/src/block/noekeon/noekeon.cpp
index 90eb9ad2b..1b327aa47 100644
--- a/src/block/noekeon/noekeon.cpp
+++ b/src/block/noekeon/noekeon.cpp
@@ -84,65 +84,77 @@ const byte Noekeon::RC[] = {
 /*
 * Noekeon Encryption
 */
-void Noekeon::enc(const byte in[], byte out[]) const
+void Noekeon::encrypt_n(const byte in[], byte out[], u32bit blocks) const
    {
-   u32bit A0 = load_be<u32bit>(in, 0);
-   u32bit A1 = load_be<u32bit>(in, 1);
-   u32bit A2 = load_be<u32bit>(in, 2);
-   u32bit A3 = load_be<u32bit>(in, 3);
-
-   for(u32bit j = 0; j != 16; ++j)
+   for(u32bit i = 0; i != blocks; ++i)
       {
-      A0 ^= RC[j];
-      theta(A0, A1, A2, A3, EK);
+      u32bit A0 = load_be<u32bit>(in, 0);
+      u32bit A1 = load_be<u32bit>(in, 1);
+      u32bit A2 = load_be<u32bit>(in, 2);
+      u32bit A3 = load_be<u32bit>(in, 3);
 
-      A1 = rotate_left(A1, 1);
-      A2 = rotate_left(A2, 5);
-      A3 = rotate_left(A3, 2);
+      for(u32bit j = 0; j != 16; ++j)
+         {
+         A0 ^= RC[j];
+         theta(A0, A1, A2, A3, EK);
 
-      gamma(A0, A1, A2, A3);
+         A1 = rotate_left(A1, 1);
+         A2 = rotate_left(A2, 5);
+         A3 = rotate_left(A3, 2);
 
-      A1 = rotate_right(A1, 1);
-      A2 = rotate_right(A2, 5);
-      A3 = rotate_right(A3, 2);
-      }
+         gamma(A0, A1, A2, A3);
 
-   A0 ^= RC[16];
-   theta(A0, A1, A2, A3, EK);
+         A1 = rotate_right(A1, 1);
+         A2 = rotate_right(A2, 5);
+         A3 = rotate_right(A3, 2);
+         }
+
+      A0 ^= RC[16];
+      theta(A0, A1, A2, A3, EK);
+
+      store_be(out, A0, A1, A2, A3);
 
-   store_be(out, A0, A1, A2, A3);
+      in += BLOCK_SIZE;
+      out += BLOCK_SIZE;
+      }
    }
 
 /*
 * Noekeon Encryption
 */
-void Noekeon::dec(const byte in[], byte out[]) const
+void Noekeon::decrypt_n(const byte in[], byte out[], u32bit blocks) const
    {
-   u32bit A0 = load_be<u32bit>(in, 0);
-   u32bit A1 = load_be<u32bit>(in, 1);
-   u32bit A2 = load_be<u32bit>(in, 2);
-   u32bit A3 = load_be<u32bit>(in, 3);
-
-   for(u32bit j = 16; j != 0; --j)
+   for(u32bit i = 0; i != blocks; ++i)
       {
-      theta(A0, A1, A2, A3, DK);
-      A0 ^= RC[j];
+      u32bit A0 = load_be<u32bit>(in, 0);
+      u32bit A1 = load_be<u32bit>(in, 1);
+      u32bit A2 = load_be<u32bit>(in, 2);
+      u32bit A3 = load_be<u32bit>(in, 3);
 
-      A1 = rotate_left(A1, 1);
-      A2 = rotate_left(A2, 5);
-      A3 = rotate_left(A3, 2);
+      for(u32bit j = 16; j != 0; --j)
+         {
+         theta(A0, A1, A2, A3, DK);
+         A0 ^= RC[j];
 
-      gamma(A0, A1, A2, A3);
+         A1 = rotate_left(A1, 1);
+         A2 = rotate_left(A2, 5);
+         A3 = rotate_left(A3, 2);
 
-      A1 = rotate_right(A1, 1);
-      A2 = rotate_right(A2, 5);
-      A3 = rotate_right(A3, 2);
-      }
+         gamma(A0, A1, A2, A3);
 
-   theta(A0, A1, A2, A3, DK);
-   A0 ^= RC[0];
+         A1 = rotate_right(A1, 1);
+         A2 = rotate_right(A2, 5);
+         A3 = rotate_right(A3, 2);
+         }
 
-   store_be(out, A0, A1, A2, A3);
+      theta(A0, A1, A2, A3, DK);
+      A0 ^= RC[0];
+
+      store_be(out, A0, A1, A2, A3);
+
+      in += BLOCK_SIZE;
+      out += BLOCK_SIZE;
+      }
    }
 
 /*
diff --git a/src/block/noekeon/noekeon.h b/src/block/noekeon/noekeon.h
index 893892446..b4f3980d7 100644
--- a/src/block/noekeon/noekeon.h
+++ b/src/block/noekeon/noekeon.h
@@ -23,8 +23,8 @@ class BOTAN_DLL Noekeon : public BlockCipher
       BlockCipher* clone() const { return new Noekeon; }
       Noekeon() : BlockCipher(16, 16) {}
    private:
-      void enc(const byte[], byte[]) const;
-      void dec(const byte[], byte[]) const;
+      void encrypt_n(const byte in[], byte out[], u32bit blocks) const;
+      void decrypt_n(const byte in[], byte out[], u32bit blocks) const;
       void key_schedule(const byte[], u32bit);
 
       static const byte RC[17];