1 files changed, 53 insertions, 41 deletions

diff --git a/src/block/noekeon/noekeon.cpp b/src/block/noekeon/noekeon.cpp
index 90eb9ad2b..0bfce1882 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;
+      }
    }
 
 /*
@@ -189,7 +201,7 @@ void Noekeon::key_schedule(const byte key[], u32bit)
 /*
 * Clear memory of sensitive data
 */
-void Noekeon::clear() throw()
+void Noekeon::clear()
    {
    EK.clear();
    DK.clear();