1 files changed, 12 insertions, 7 deletions
diff --git a/crypto_aead/lilliputaei128v1/ref/tweakey.h b/crypto_aead/lilliputaei128v1/ref/tweakey.h
index 8018f9c..d5d2be3 100644
--- a/crypto_aead/lilliputaei128v1/ref/tweakey.h
+++ b/crypto_aead/lilliputaei128v1/ref/tweakey.h
@@ -5,14 +5,19 @@
 #include <stdint.h>
 
 
-#define TWEAK_LENGTH_BITS 192
-#define KEY_LENGTH_BITS 128
-#define ROUND_TWEAKEY_BITS 64
+#define TWEAK_LENGTH_BITS         192
+#define KEY_LENGTH_BITS           128
+#define ROUND_TWEAKEY_LENGTH_BITS 64
+
+#define TWEAK_BYTES         (TWEAK_LENGTH_BITS/8)
+#define KEY_BYTES           (KEY_LENGTH_BITS/8)
+#define ROUND_TWEAKEY_BYTES (ROUND_TWEAKEY_LENGTH_BITS/8)
 
 
 struct tweakey_state
 {
     FILE* debug;
+    uint8_t TK[TWEAK_BYTES+KEY_BYTES];
 };
 
 typedef struct tweakey_state tweakey_state;
@@ -20,15 +25,15 @@ typedef struct tweakey_state tweakey_state;
 
 void tweakey_state_init(
     tweakey_state *TK,
-    const uint8_t key[KEY_LENGTH_BITS/8],
-    const uint8_t tweak[TWEAK_LENGTH_BITS/8],
+    const uint8_t key[KEY_BYTES],
+    const uint8_t tweak[TWEAK_BYTES],
     FILE* debug
 );
 
 void tweakey_state_extract(
     const tweakey_state *TK,
-    uint8_t round_tweakey[ROUND_TWEAKEY_BITS/8], /* output */
-    uint8_t i                                    /* round constant */
+    uint8_t round_tweakey[ROUND_TWEAKEY_BYTES], /* output */
+    uint8_t i                                   /* round constant */
 );
 
 void tweakey_state_update(tweakey_state *TK);