From 7b8c8c4d7973f17bbbee45d4718224c5853d9dc7 Mon Sep 17 00:00:00 2001
From: Andy Polyakov <appro@openssl.org>
Date: Wed, 25 Jun 2014 22:10:45 +0200
Subject: [PATCH] aesv8-armx.pl: rigid input verification in key setup.

---
 crypto/aes/asm/aesv8-armx.pl | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/crypto/aes/asm/aesv8-armx.pl b/crypto/aes/asm/aesv8-armx.pl
index 87fe446b30..923c7f62d5 100755
--- a/crypto/aes/asm/aesv8-armx.pl
+++ b/crypto/aes/asm/aesv8-armx.pl
@@ -70,6 +70,19 @@ $code.=<<___	if ($flavour =~ /64/);
 	add	x29,sp,#0
 ___
 $code.=<<___;
+	mov	$ptr,#-1
+	cmp	$inp,#0
+	b.eq	.Lenc_key_abort
+	cmp	$out,#0
+	b.eq	.Lenc_key_abort
+	mov	$ptr,#-2
+	cmp	$bits,#128
+	b.lt	.Lenc_key_abort
+	cmp	$bits,#256
+	b.gt	.Lenc_key_abort
+	tst	$bits,#0x3f
+	b.ne	.Lenc_key_abort
+
 	adr	$ptr,rcon
 	cmp	$bits,#192
 
@@ -209,8 +222,10 @@ $code.=<<___;
 
 .Ldone:
 	str	$rounds,[$out]
+	mov	$ptr,#0
 
-	eor	x0,x0,x0		// return value
+.Lenc_key_abort:
+	mov	x0,$ptr			// return value
 	`"ldr	x29,[sp],#16"		if ($flavour =~ /64/)`
 	ret
 .size	${prefix}_set_encrypt_key,.-${prefix}_set_encrypt_key
@@ -230,6 +245,9 @@ ___
 $code.=<<___;
 	bl	.Lenc_key
 
+	cmp	x0,#0
+	b.ne	.Ldec_key_abort
+
 	sub	$out,$out,#240		// restore original $out
 	mov	x4,#-16
 	add	$inp,$out,x12,lsl#4	// end of key schedule
@@ -254,6 +272,7 @@ $code.=<<___;
 	vst1.32	{v0.16b},[$inp]
 
 	eor	x0,x0,x0		// return value
+.Ldec_key_abort:
 ___
 $code.=<<___	if ($flavour !~ /64/);
 	ldmia	sp!,{r4,pc}
-- 
2.34.1