From dec7c50cc458695b7112a72d6c393db68db7507c Mon Sep 17 00:00:00 2001
From: jakcron <jkrcarr@gmail.com>
Date: Sat, 7 Apr 2018 15:58:07 +0800
Subject: [PATCH] [crypto] Properly define sAesXts128Key

---
 lib/libcrypto/include/crypto/aes.h | 27 +++++++++++++++++++++++++--
 1 file changed, 25 insertions(+), 2 deletions(-)

diff --git a/lib/libcrypto/include/crypto/aes.h b/lib/libcrypto/include/crypto/aes.h
index b955a2c..bd77797 100644
--- a/lib/libcrypto/include/crypto/aes.h
+++ b/lib/libcrypto/include/crypto/aes.h
@@ -42,8 +42,31 @@ namespace aes
 
 	struct sAesXts128Key
 	{
-		sAes128Key key1;
-		sAes128Key key2;
+		uint8_t key[2][kAes128KeySize];
+
+		void set(const uint8_t key_0[kAes128KeySize], const uint8_t key_1[kAes128KeySize])
+		{
+			memcpy(this->key[0], key_0, kAes128KeySize);
+			memcpy(this->key[1], key_1, kAes128KeySize);
+		}
+
+		bool compare(const sAesXts128Key& other) const
+		{
+			return memcmp(this->key[0], other.key[0], kAes128KeySize) == 0 && memcmp(this->key[1], other.key[1], kAes128KeySize) == 0;
+		}
+
+		void operator=(const sAesXts128Key& other)
+		{
+			set(other.key[0], other.key[1]);
+		}
+		bool operator==(const sAesXts128Key& other) const
+		{
+			return compare(other);
+		}
+		bool operator!=(const sAesXts128Key& other) const
+		{
+			return !compare(other);
+		}
 	};
 
 	struct sAesIvCtr