From f53eb8e260d174153bb3fc24ff1fff7966dcfbee Mon Sep 17 00:00:00 2001
From: Mounir IDRASSI <mounir.idrassi@idrix.fr>
Date: Mon, 27 Nov 2017 09:10:17 +0200
Subject: SIMD speed optimization for Kuznyechik cipher implementation (up to
 2x speedup). Based on https://github.com/aprelev/libgost15.

---
 src/Volume/Cipher.cpp  | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 src/Volume/Cipher.h    |  2 +-
 src/Volume/Volume.make |  1 +
 3 files changed, 50 insertions(+), 1 deletion(-)

(limited to 'src/Volume')

diff --git a/src/Volume/Cipher.cpp b/src/Volume/Cipher.cpp
index de351ad4..32f61b76 100644
--- a/src/Volume/Cipher.cpp
+++ b/src/Volume/Cipher.cpp
@@ -462,5 +462,53 @@ namespace VeraCrypt
 	{
 		kuznyechik_set_key (key, (kuznyechik_kds *) ScheduledKey.Ptr());
 	}
+	void CipherKuznyechik::EncryptBlocks (byte *data, size_t blockCount) const
+	{
+		if (!Initialized)
+			throw NotInitialized (SRC_POS);
+
+#if CRYPTOPP_BOOL_SSE2_INTRINSICS_AVAILABLE
+		if ((blockCount >= 4)
+			&& IsHwSupportAvailable())
+		{
+			kuznyechik_encrypt_blocks (data, data, blockCount, (kuznyechik_kds *) ScheduledKey.Ptr());
+		}
+		else
+#endif
+			Cipher::EncryptBlocks (data, blockCount);
+	}
+	
+	void CipherKuznyechik::DecryptBlocks (byte *data, size_t blockCount) const
+	{
+		if (!Initialized)
+			throw NotInitialized (SRC_POS);
+
+#if CRYPTOPP_BOOL_SSE2_INTRINSICS_AVAILABLE
+		if ((blockCount >= 4)
+			&& IsHwSupportAvailable())
+		{
+			kuznyechik_decrypt_blocks (data, data, blockCount, (kuznyechik_kds *) ScheduledKey.Ptr());
+		}
+		else
+#endif
+			Cipher::DecryptBlocks (data, blockCount);
+	}
+	
+	bool CipherKuznyechik::IsHwSupportAvailable () const
+	{
+#if CRYPTOPP_BOOL_SSE2_INTRINSICS_AVAILABLE
+		static bool state = false;
+		static bool stateValid = false;
+
+		if (!stateValid)
+		{
+			state = HasSSE2() ? true : false;
+			stateValid = true;
+		}
+		return state;
+#else
+		return false;
+#endif
+	}
 	bool Cipher::HwSupportEnabled = true;
 }
diff --git a/src/Volume/Cipher.h b/src/Volume/Cipher.h
index 27925b7f..18d1d2ed 100644
--- a/src/Volume/Cipher.h
+++ b/src/Volume/Cipher.h
@@ -104,13 +104,13 @@ namespace VeraCrypt
 	TC_CIPHER (Serpent, 16, 32);
 	TC_CIPHER (Twofish, 16, 32);
 	TC_CIPHER (Camellia, 16, 32);
+	TC_CIPHER (Kuznyechik, 16, 32);
 
 #undef TC_CIPHER_ADD_METHODS
 #define TC_CIPHER_ADD_METHODS
 
 	TC_CIPHER (Gost89, 16, 32);
 	TC_CIPHER (Gost89StaticSBOX, 16, 32);
-	TC_CIPHER (Kuznyechik, 16, 32);
 
 #undef TC_CIPHER
 
diff --git a/src/Volume/Volume.make b/src/Volume/Volume.make
index b8f69429..7b5cb4f1 100644
--- a/src/Volume/Volume.make
+++ b/src/Volume/Volume.make
@@ -79,6 +79,7 @@ OBJS += ../Crypto/Camellia.o
 OBJS += ../Crypto/GostCipher.o
 OBJS += ../Crypto/Streebog.o
 OBJS += ../Crypto/kuznyechik.o
+OBJS += ../Crypto/kuznyechik_simd.o
 
 OBJS += ../Common/Crc.o
 OBJS += ../Common/Endian.o
-- 
cgit v1.2.3