[PATCH] add MIPS assembler version of twofish crypto algorithm

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

 



Hi,

this patch adds a MIPS assembler version of the twofish cipher
algorithm.  x86(_64) had an assembler version of twofish for some time
now, giving it an "unfair" advantage against the not so common
architectures.

The code wast put through various tests and benchmarks in a user-space
testbench, testing its correctness (also via qemu) for 32bit vs. 64bit
and big vs. little endian MIPS.

http://mosquito.dyndns.tv/freesvn/trunk/linux/twofish-mips/

Also the same code backported to a 2.6.39 kernel (64bit) has been
working flawlessly on my loongson-2f machine (with a twofish dm-crypt
root) for a few weeks now.

Performance gain for the pure crypto code in 32-bit mode is about 15% on
both Ingenic Xburst and Loongson-2f.  With 64-bit it is 24-29%, as the C
code suffers from 32->64 bit type conversions inserted by gcc.  The code
uses less unrolling than the C version, and should be smaller by at
least a factor of 8.  Full results in file Benchres.txt at the SVN URL
given above.  

Signed-off-by: David Kühling <dvdkhlng@xxxxxx>
---
 arch/mips/Kbuild                    |    2 +
 arch/mips/crypto/Makefile           |    7 +
 arch/mips/crypto/twofish-mips-asm.S |  346 +++++++++++++++++++++++++++++++++++
 arch/mips/crypto/twofish_glue.c     |   97 ++++++++++
 arch/mips/kernel/asm-offsets.c      |    8 +
 crypto/Kconfig                      |   16 ++
 6 files changed, 476 insertions(+), 0 deletions(-)
 create mode 100644 arch/mips/crypto/Makefile
 create mode 100644 arch/mips/crypto/twofish-mips-asm.S
 create mode 100644 arch/mips/crypto/twofish_glue.c

diff --git a/arch/mips/Kbuild b/arch/mips/Kbuild
index 7dd65cf..fc88443 100644
--- a/arch/mips/Kbuild
+++ b/arch/mips/Kbuild
@@ -17,3 +17,5 @@ obj- := $(platform-)
 obj-y += kernel/
 obj-y += mm/
 obj-y += math-emu/
+
+obj-y += crypto/
diff --git a/arch/mips/crypto/Makefile b/arch/mips/crypto/Makefile
new file mode 100644
index 0000000..6069354
--- /dev/null
+++ b/arch/mips/crypto/Makefile
@@ -0,0 +1,7 @@
+#
+# Arch-specific CryptoAPI modules.
+#
+
+obj-$(CONFIG_CRYPTO_TWOFISH_MIPS) += twofish-mips.o
+
+twofish-mips-y := twofish-mips-asm.o twofish_glue.o
diff --git a/arch/mips/crypto/twofish-mips-asm.S b/arch/mips/crypto/twofish-mips-asm.S
new file mode 100644
index 0000000..4918008
--- /dev/null
+++ b/arch/mips/crypto/twofish-mips-asm.S
@@ -0,0 +1,346 @@
+/*
+ * Copyright (C) 2011 David Kuehling <dvdkhlng AT gmx DOT de>
+ *
+ * (Dervived from twofish-x86_64-asm_64.S by Joachim Fritschi)
+ *		
+ * This file is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.        
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.
+ */
+
+/*
+ * In this file we reference MIPS registers only via symbolic names $aN $tN,
+ * $taN etc. which makes it work with any of o32/n32/o64/n64 ABIs.  Some
+ * additonal #ifdef'ing is used for handling 32 vs. 64-bit and little
+ * vs. big-endian issues.
+ */	
+	
+.file "twofish-mips-asm.S"
+.text
+
+#include <asm/asm-offsets.h>
+
+#define a_offset	0
+#define b_offset	4
+#define c_offset	8
+#define d_offset	12
+
+/* Structure of the crypto context struct*/
+
+#define s0	0	/* S0 Array 256 Words each */
+#define s1	1024	/* S1 Array */
+#define s2	2048	/* S2 Array */
+#define s3	3072	/* S3 Array */
+#define w	4096	/* 8 whitening keys (word) */
+#define k	4128	/* key 1-32 ( word ) */
+
+/*
+ * Abstractions for 32 vs. 64-bit code and big/little endian
+ */
+
+#if __mips64
+#  define HAVE_64BIT
+#endif
+
+#if _MIPS_SZPTR == 32
+#  define PTRADD addu
+#  warning "detected pointer-size as 32-bit"
+#else
+#  ifndef HAVE_64BIT
+#  error "something is broken: detected pointer size > GPR size"
+#  endif
+#  define PTRADD daddu
+#  warning "detected pointer-size as 64-bit"
+#endif
+
+#if MIPSEL
+#  define LITTLEENDIAN 1
+#elif MIPSEB
+#  define BIGENDIAN 1
+#else
+#  error "endianess undefined"
+#endif
+	
+#ifdef LITTLEENDIAN
+#  define le32_fromto_cpu(a)
+#else
+#  warning "adding endianess conversion code"
+	/* todo: we could pair two conversions for better scheduling */
+#  define le32_fromto_cpu(a)			\
+	sll	$t8, a, 24 ;			\
+	srl	$t9, a, 24 ;			\
+	srl	$ta3, a, 8 ;			\
+	and	a, 0xff00   ;			\
+	or	$t8, $t9    ;			\
+	sll	a, a, 8    ;			\
+	and 	$ta3, 0xff00 ;			\
+	or	a, $ta3	    ;			\
+	or	a, $t8	    ;
+#endif
+
+	
+ /* perform input whitening of 16-byte block */
+#define input_whitening(a,b,c,d,context)	\
+	lw	$ta2, w+a_offset(context)	; \
+	lw	$ta3, w+b_offset(context)	; \
+	lw	$t8, w+c_offset(context)	; \
+	lw	$t9, w+d_offset(context)	; \
+	xor	a, $ta2				; \
+	xor	b, $ta3				; \
+	xor	c, $t8				; \
+	xor	d, $t9	
+
+	
+ /* perform output whitening of 16-byte block */
+#define output_whitening(a,b,c,d,context)	\
+	lw	$ta2, w+16+a_offset(context)	; \
+	lw	$ta3, w+16+b_offset(context)	; \
+	lw	$t8, w+16+c_offset(context)	; \
+	lw	$t9, w+16+d_offset(context)	; \
+	xor	a, $ta2				; \
+	xor	b, $ta3				; \
+	xor	c, $t8				; \
+	xor	d, $t9	
+
+/*
+ * a input register containing a
+ * b input register containing b
+ * c input register containing c
+ * d input register containing d
+ * koffs: koffs($a3) is the element in the 'k' array to use
+ */
+#define encrypt_round(a,b,c,d,koffs)				\
+	sll	$ta0, b, 2					; \
+	sll	$ta1, a, 2					; \
+	andi	$ta0, 0x3FC					; \
+	andi	$ta1, 0x3FC					; \
+	PTRADD	$ta0, $a0					; \
+	PTRADD	$ta1, $a0					; \
+	lw	$t8, s1($ta0)					; \
+	lw	$t9, s0($ta1)					; \
+								; \
+	srl	$ta2, b, 8-2					; \
+	srl	$ta3, a, 8-2					; \
+	andi	$ta2, 0x3FC					; \
+	andi	$ta3, 0x3FC					; \
+	PTRADD	$ta2, $a0					; \
+	PTRADD	$ta3, $a0					; \
+	lw	$ta2, s2($ta2)					; \
+	lw	$ta3, s1($ta3)					; \
+								; \
+	srl	$ta0, b, 16-2					; \
+	srl	$ta1, a, 16-2					; \
+	andi	$ta0, 0x3FC					; \
+	andi	$ta1, 0x3FC					; \
+	PTRADD	$ta0, $a0					; \
+	PTRADD	$ta1, $a0					; \
+	lw	$ta0, s3($ta0)					; \
+	lw	$ta1, s2($ta1)					; \
+								; \
+	srl	$v0, b, 24-2					; \
+	srl	$v1, a, 24-2					; \
+	andi	$v0, 0x3FC					; \
+	andi	$v1, 0x3FC					; \
+	PTRADD	$v0, $a0					; \
+	PTRADD	$v1, $a0					; \
+	xor	$t8, $ta2					; \
+	xor	$t9, $ta3					; \
+	lw	$v0, s0($v0)					; \
+	xor	$t8, $ta0					; \
+	xor	$t9, $ta1					; \
+	lw	$v1, s3($v1)					; \
+								; \
+	/* rotate d left by 1, interleaved with xoring, and	\
+	  loading from the 'k' array */				\
+	sll	$ta0, d, 1					; \
+	xor	$t8, $v0					; \
+	lw	$ta2, 0+koffs($a3)				; \
+	srl	d, 32-1						; \
+	xor	$t9, $v1					; \
+	lw	$ta3, 4+koffs($a3)				; \
+	or	d, $ta0						; \
+								; \
+	addu	$t9, $t8					; \
+	addu	$v0, $t8, $ta3					; \
+	addu	$v1, $t9, $ta2					; \
+	addu	$v0, $t9					; \
+	xor	c, $v1						; \
+	xor	d, $v0						; \
+								; \
+	srl	$ta3, c, 1	/* rotate c right 1 */		; \
+	sll	$ta2, c, 32-1	/* ... rotate */		; \
+	or	c, $ta2, $ta3	/* ... rotate */		; \
+
+/*
+ * a input register containing a
+ * b input register containing b 
+ * c input register containing c
+ * d input register containing d
+ * koffs: koffs($a3) is the element in the 'k' array to use
+ */
+#define decrypt_round(a,b,c,d,koffs)				\
+	sll	$ta0, a, 2					; \
+	sll	$ta1, b, 2					; \
+	andi	$ta0, 0x3FC					; \
+	andi	$ta1, 0x3FC					; \
+	PTRADD	$ta0, $a0					; \
+	PTRADD	$ta1, $a0					; \
+	lw	$t9, s0($ta0)					; \
+	lw	$t8, s1($ta1)					; \
+								; \
+	srl	$ta2, a, 8-2					; \
+	srl	$ta3, b, 8-2					; \
+	andi	$ta2, 0x3FC					; \
+	andi	$ta3, 0x3FC					; \
+	PTRADD	$ta2, $a0					; \
+	PTRADD	$ta3, $a0					; \
+	lw	$ta2, s1($ta2)					; \
+	lw	$ta3, s2($ta3)					; \
+								; \
+	srl	$ta0, a, 16-2					; \
+	srl	$ta1, b, 16-2					; \
+	andi	$ta1, 0x3FC					; \
+	andi	$ta0, 0x3FC					; \
+	PTRADD	$ta0, $a0					; \
+	PTRADD	$ta1, $a0					; \
+	lw	$ta0, s2($ta0)					; \
+	lw	$ta1, s3($ta1)					; \
+								; \
+	srl	$v0, a, 24-2					; \
+	srl	$v1, b, 24-2					; \
+	andi	$v0, 0x3FC					; \
+	andi	$v1, 0x3FC					; \
+	PTRADD	$v0, $a0					; \
+	PTRADD	$v1, $a0					; \
+	xor	$t9, $ta2					; \
+	xor	$t8, $ta3					; \
+	lw	$v0, s3($v0)					; \
+	xor	$t9, $ta0					; \
+	xor	$t8, $ta1					; \
+	lw	$v1, s0($v1)					; \
+								; \
+	/* rotate c left by 1, interleaved with xoring, and	\
+	  loading from the 'k' array */				\
+	sll	$ta0, c, 1					; \
+	xor	$t9, $v0					; \
+	lw	$ta2, 0+koffs($a3)				; \
+	srl	c, 32-1						; \
+	xor	$t8, $v1					; \
+	lw	$ta3, 4+koffs($a3)				; \
+	or	c, $ta0						; \
+								; \
+	addu	$t9, $t8					; \
+	addu	$v0, $t8, $ta3					; \
+	addu	$v1, $t9, $ta2					; \
+	addu	$v0, $t9					; \
+	xor	c, $v1						; \
+	xor	d, $v0						; \
+								; \
+	srl	$ta3, d, 1	/* rotate d right by 1 */	; \
+	sll	$ta2, d, 32-1	/* ...rotate */			; \
+	or	d, $ta2, $ta3	/* ...rotate */			; \
+
+
+.global twofish_enc_blk
+.global twofish_dec_blk
+
+twofish_enc_blk:
+	/* $a0 contains the crypto tfm address */
+	/* $a1 contains the output address */
+	/* $a2 contains the input address */
+
+	PTRADD	$a0, crypto_tfm_ctx_offset	/* set ctx address */
+
+	/* if we turn this into 64-bit ops, we get endianess issues on
+	   big-endian mips, plus alignment problems */
+	lw	$t0, a_offset($a2)
+	lw	$t1, b_offset($a2)
+	lw	$t2, c_offset($a2)
+	lw	$t3, d_offset($a2)
+	
+	le32_fromto_cpu($t0)
+	le32_fromto_cpu($t1)
+	le32_fromto_cpu($t2)
+	le32_fromto_cpu($t3)
+
+	input_whitening($t0,$t1,$t2,$t3,$a0)
+	
+	/* If we unrolled this loop, we gain < 5% performance, but function
+	   size reaches almost 4kB, vs. 0.5kB as implemented below.  During
+	   loop $a3 points into the 'k' array, '$a2' denotes the position in
+	   the k array where the loop should exit.  'a3' is incremented at the
+	   start of loop, to help branch prediction and scheduling */
+	PTRADD	$a3, $a0, k
+	PTRADD	$a2, $a0, k+8*16
+1:	PTRADD	$a3, $a3, 16
+ 	encrypt_round($t0,$t1,$t2,$t3,-16);
+	encrypt_round($t2,$t3,$t0,$t1,-8);
+	bne	$a2, $a3, 1b
+
+	output_whitening($t2,$t3,$t0,$t1,$a0)
+
+	le32_fromto_cpu($t2)
+	le32_fromto_cpu($t3)
+	le32_fromto_cpu($t0)
+	le32_fromto_cpu($t1)
+
+	sw	$t2, a_offset($a1)
+	sw	$t3, b_offset($a1)
+	sw	$t0, c_offset($a1)
+	sw	$t1, d_offset($a1)
+
+	jr	$ra
+	
+twofish_dec_blk:
+	/* $a0 contains the crypto tfm address */
+	/* $a1 contains the output address */
+	/* $a2 contains the input address */
+
+	PTRADD	$a0, crypto_tfm_ctx_offset	/* set ctx address */
+
+	/* if we turned this into 64-bit ops, we get endianess issues on
+	   big-endian mips, plus alignment problems */
+	lw	$t0, a_offset($a2)
+	lw	$t1, b_offset($a2)
+	lw	$t2, c_offset($a2)
+	lw	$t3, d_offset($a2)
+	
+	le32_fromto_cpu($t0)
+	le32_fromto_cpu($t1)
+	le32_fromto_cpu($t2)
+	le32_fromto_cpu($t3)
+
+	output_whitening($t0,$t1,$t2,$t3,$a0)
+
+	/* for details about loop below see comment in twofish_enc_blk */
+	PTRADD	$a3, $a0, k+16*8
+	PTRADD	$a2, $a0, k
+1:	PTRADD	$a3, $a3, -16
+	decrypt_round($t0,$t1,$t2,$t3,8);
+	decrypt_round($t2,$t3,$t0,$t1,0);
+	bne	$a2, $a3, 1b
+	
+	input_whitening($t2,$t3,$t0,$t1,$a0)
+
+	le32_fromto_cpu($t2)
+	le32_fromto_cpu($t3)
+	le32_fromto_cpu($t0)
+	le32_fromto_cpu($t1)
+
+	sw	$t2, a_offset($a1)
+	sw	$t3, b_offset($a1)
+	sw	$t0, c_offset($a1)
+	sw	$t1, d_offset($a1)
+
+	jr	$ra
+
diff --git a/arch/mips/crypto/twofish_glue.c b/arch/mips/crypto/twofish_glue.c
new file mode 100644
index 0000000..cefaf8b
--- /dev/null
+++ b/arch/mips/crypto/twofish_glue.c
@@ -0,0 +1,97 @@
+/*
+ * Glue Code for assembler optimized version of TWOFISH
+ *
+ * Originally Twofish for GPG
+ * By Matthew Skala <mskala@xxxxxxxxxxxxxxxxx>, July 26, 1998
+ * 256-bit key length added March 20, 1999
+ * Some modifications to reduce the text size by Werner Koch, April, 1998
+ * Ported to the kerneli patch by Marc Mutz <Marc@xxxxxxxx>
+ * Ported to CryptoAPI by Colin Slater <hoho@xxxxxxxxxxxx>
+ *
+ * The original author has disclaimed all copyright interest in this
+ * code and thus put it in the public domain. The subsequent authors
+ * have put this under the GNU General Public License.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
+ * USA
+ *
+ * This code is a "clean room" implementation, written from the paper
+ * _Twofish: A 128-Bit Block Cipher_ by Bruce Schneier, John Kelsey,
+ * Doug Whiting, David Wagner, Chris Hall, and Niels Ferguson, available
+ * through http://www.counterpane.com/twofish.html
+ *
+ * For background information on multiplication in finite fields, used for
+ * the matrix operations in the key schedule, see the book _Contemporary
+ * Abstract Algebra_ by Joseph A. Gallian, especially chapter 22 in the
+ * Third Edition.
+ */
+
+#include <crypto/twofish.h>
+#include <linux/crypto.h>
+#include <linux/init.h>
+#include <linux/module.h>
+#include <linux/types.h>
+
+asmlinkage void twofish_enc_blk(struct crypto_tfm *tfm, u8 *dst, const u8 *src);
+asmlinkage void twofish_dec_blk(struct crypto_tfm *tfm, u8 *dst, const u8 *src);
+
+static void twofish_encrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
+{
+	twofish_enc_blk(tfm, dst, src);
+}
+
+static void twofish_decrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
+{
+	twofish_dec_blk(tfm, dst, src);
+}
+
+static struct crypto_alg alg = {
+	.cra_name		=	"twofish",
+	.cra_driver_name	=	"twofish-asm",
+	.cra_priority		=	200,
+	.cra_flags		=	CRYPTO_ALG_TYPE_CIPHER,
+	.cra_blocksize		=	TF_BLOCK_SIZE,
+	.cra_ctxsize		=	sizeof(struct twofish_ctx),
+	.cra_alignmask		=	3,
+	.cra_module		=	THIS_MODULE,
+	.cra_list		=	LIST_HEAD_INIT(alg.cra_list),
+	.cra_u			=	{
+		.cipher = {
+			.cia_min_keysize	=	TF_MIN_KEY_SIZE,
+			.cia_max_keysize	=	TF_MAX_KEY_SIZE,
+			.cia_setkey		=	twofish_setkey,
+			.cia_encrypt		=	twofish_encrypt,
+			.cia_decrypt		=	twofish_decrypt
+		}
+	}
+};
+
+static int __init init(void)
+{
+	return crypto_register_alg(&alg);
+}
+
+static void __exit fini(void)
+{
+	crypto_unregister_alg(&alg);
+}
+
+module_init(init);
+module_exit(fini);
+
+MODULE_LICENSE("GPL");
+MODULE_DESCRIPTION ("Twofish Cipher Algorithm, asm optimized");
+MODULE_ALIAS("twofish");
+MODULE_ALIAS("twofish-asm");
diff --git a/arch/mips/kernel/asm-offsets.c b/arch/mips/kernel/asm-offsets.c
index 6b30fb2..9d91a6d 100644
--- a/arch/mips/kernel/asm-offsets.c
+++ b/arch/mips/kernel/asm-offsets.c
@@ -15,6 +15,7 @@
 #include <linux/interrupt.h>
 #include <linux/kbuild.h>
 #include <linux/suspend.h>
+#include <linux/crypto.h>
 #include <asm/ptrace.h>
 #include <asm/processor.h>
 
@@ -339,3 +340,10 @@ void output_pbe_defines(void)
 	BLANK();
 }
 #endif
+
+void output_crypto_defines(void)
+{
+	COMMENT(" Linux crypto API offsets. ");
+	OFFSET(crypto_tfm_ctx_offset, crypto_tfm, __crt_ctx);
+	BLANK();
+}
diff --git a/crypto/Kconfig b/crypto/Kconfig
index ae27b753..fa1b2b2 100644
--- a/crypto/Kconfig
+++ b/crypto/Kconfig
@@ -793,6 +793,22 @@ config CRYPTO_TWOFISH_X86_64
 	  See also:
 	  <http://www.schneier.com/twofish.html>
 
+config CRYPTO_TWOFISH_MIPS
+	tristate "Twofish cipher algorithm (MIPS)"
+	depends on MIPS
+	select CRYPTO_ALGAPI
+	select CRYPTO_TWOFISH_COMMON
+	help
+	  Twofish cipher algorithm (MIPS).
+
+	  Twofish was submitted as an AES (Advanced Encryption Standard)
+	  candidate cipher by researchers at CounterPane Systems.  It is a
+	  16 round block cipher supporting key sizes of 128, 192, and 256
+	  bits.
+
+	  See also:
+	  <http://www.schneier.com/twofish.html>
+
 comment "Compression"
 
 config CRYPTO_DEFLATE
-- 
1.7.4.1

Attachment: pgpZsCUmEqMcI.pgp
Description: PGP signature


[Index of Archives]     [Linux MIPS Home]     [LKML Archive]     [Linux ARM Kernel]     [Linux ARM]     [Linux]     [Git]     [Yosemite News]     [Linux SCSI]     [Linux Hams]

  Powered by Linux