#! /usr/bin/env perl

# Copyright 2016-2020 The OpenSSL Project Authors. All Rights Reserved.

#

# Licensed under the OpenSSL license (the "License").  You may not use

# this file except in compliance with the License.  You can obtain a copy

# in the file LICENSE in the source distribution or at

# https://www.openssl.org/source/license.html

#

# ====================================================================

# Written by Andy Polyakov <appro@openssl.org> for the OpenSSL

# project. The module is, however, dual licensed under OpenSSL and

# CRYPTOGAMS licenses depending on where you obtain it. For further

# details see http://www.openssl.org/~appro/cryptogams/.

# ====================================================================

#

# June 2015

#

# ChaCha20 for ARMv8.

#

# April 2019

#

# Replace 3xNEON+1xIALU code path with 4+1. 4+1 is actually fastest

# option on most(*), but not all, processors, yet 6+2 is retained.

# This is because penalties are considered tolerable in comparison to

# improvement on processors where 6+2 helps. Most notably +37% on

# ThunderX2. It's server-oriented processor which will have to serve

# as many requests as possible. While others are mostly clients, when

# performance doesn't have to be absolute top-notch, just fast enough,

# as majority of time is spent "entertaining" relatively slow human.

#

# Performance in cycles per byte out of large buffer.

#

#			IALU/gcc-4.9	4xNEON+1xIALU	6xNEON+2xIALU

#

# Apple A7		5.50/+49%	2.72		1.60

# Cortex-A53		8.40/+80%	4.06		4.45(*)

# Cortex-A57		8.06/+43%	4.15		4.40(*)

# Denver		4.50/+82%	2.30		2.70(*)

# X-Gene		9.50/+46%	8.20		8.90(*)

# Mongoose		8.00/+44%	2.74		3.12(*)

# Kryo			8.17/+50%	4.47		4.65(*)

# ThunderX2		7.22/+48%	5.64		4.10

#

# (*)	slower than 4+1:-(

# $output is the last argument if it looks like a file (it has an extension)

# $flavour is the first argument if it doesn't look like a file

$output = $#ARGV >= 0 && $ARGV[$#ARGV] =~ m|\.\w+$| ? pop : undef;

$flavour = $#ARGV >= 0 && $ARGV[0] !~ m|\.| ? shift : undef;

$0 =~ m/(.*[\/\\])[^\/\\]+$/; $dir=$1;

( $xlate="${dir}arm-xlate.pl" and -f $xlate ) or

( $xlate="${dir}../../perlasm/arm-xlate.pl" and -f $xlate) or

die "can't locate arm-xlate.pl";

open OUT,"| \"$^X\" $xlate $flavour \"$output\""

    or die "can't call $xlate: $!";

*STDOUT=*OUT;

sub AUTOLOAD()		# thunk [simplified] x86-style perlasm

{ my $opcode = $AUTOLOAD; $opcode =~ s/.*:://; $opcode =~ s/_/\./;

  my $arg = pop;

    $arg = "#$arg" if ($arg*1 eq $arg);

    $code .= "\t$opcode\t".join(',',@_,$arg)."\n";

}

my ($out,$inp,$len,$key,$ctr) = map("x$_",(0..4));

my @x=map("x$_",(5..17,19..21));

my @d=map("x$_",(22..28,30));

sub ROUND {

my ($a0,$b0,$c0,$d0)=@_;

my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));

my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));

my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));

    (

	"&add_32	(@x[$a0],@x[$a0],@x[$b0])",

	 "&add_32	(@x[$a1],@x[$a1],@x[$b1])",

	  "&add_32	(@x[$a2],@x[$a2],@x[$b2])",

	   "&add_32	(@x[$a3],@x[$a3],@x[$b3])",

	"&eor_32	(@x[$d0],@x[$d0],@x[$a0])",

	 "&eor_32	(@x[$d1],@x[$d1],@x[$a1])",

	  "&eor_32	(@x[$d2],@x[$d2],@x[$a2])",

	   "&eor_32	(@x[$d3],@x[$d3],@x[$a3])",

	"&ror_32	(@x[$d0],@x[$d0],16)",

	 "&ror_32	(@x[$d1],@x[$d1],16)",

	  "&ror_32	(@x[$d2],@x[$d2],16)",

	   "&ror_32	(@x[$d3],@x[$d3],16)",

	"&add_32	(@x[$c0],@x[$c0],@x[$d0])",

	 "&add_32	(@x[$c1],@x[$c1],@x[$d1])",

	  "&add_32	(@x[$c2],@x[$c2],@x[$d2])",

	   "&add_32	(@x[$c3],@x[$c3],@x[$d3])",

	"&eor_32	(@x[$b0],@x[$b0],@x[$c0])",

	 "&eor_32	(@x[$b1],@x[$b1],@x[$c1])",

	  "&eor_32	(@x[$b2],@x[$b2],@x[$c2])",

	   "&eor_32	(@x[$b3],@x[$b3],@x[$c3])",

	"&ror_32	(@x[$b0],@x[$b0],20)",

	 "&ror_32	(@x[$b1],@x[$b1],20)",

	  "&ror_32	(@x[$b2],@x[$b2],20)",

	   "&ror_32	(@x[$b3],@x[$b3],20)",

	"&add_32	(@x[$a0],@x[$a0],@x[$b0])",

	 "&add_32	(@x[$a1],@x[$a1],@x[$b1])",

	  "&add_32	(@x[$a2],@x[$a2],@x[$b2])",

	   "&add_32	(@x[$a3],@x[$a3],@x[$b3])",

	"&eor_32	(@x[$d0],@x[$d0],@x[$a0])",

	 "&eor_32	(@x[$d1],@x[$d1],@x[$a1])",

	  "&eor_32	(@x[$d2],@x[$d2],@x[$a2])",

	   "&eor_32	(@x[$d3],@x[$d3],@x[$a3])",

	"&ror_32	(@x[$d0],@x[$d0],24)",

	 "&ror_32	(@x[$d1],@x[$d1],24)",

	  "&ror_32	(@x[$d2],@x[$d2],24)",

	   "&ror_32	(@x[$d3],@x[$d3],24)",

	"&add_32	(@x[$c0],@x[$c0],@x[$d0])",

	 "&add_32	(@x[$c1],@x[$c1],@x[$d1])",

	  "&add_32	(@x[$c2],@x[$c2],@x[$d2])",

	   "&add_32	(@x[$c3],@x[$c3],@x[$d3])",

	"&eor_32	(@x[$b0],@x[$b0],@x[$c0])",

	 "&eor_32	(@x[$b1],@x[$b1],@x[$c1])",

	  "&eor_32	(@x[$b2],@x[$b2],@x[$c2])",

	   "&eor_32	(@x[$b3],@x[$b3],@x[$c3])",

	"&ror_32	(@x[$b0],@x[$b0],25)",

	 "&ror_32	(@x[$b1],@x[$b1],25)",

	  "&ror_32	(@x[$b2],@x[$b2],25)",

	   "&ror_32	(@x[$b3],@x[$b3],25)"

    );

}

$code.=<<___;

#ifndef	__KERNEL__

# include "arm_arch.h"

.extern	OPENSSL_armcap_P

#endif

.text

.align	5

.Lsigma:

.quad	0x3320646e61707865,0x6b20657479622d32		// endian-neutral

.Lone:

.long	1,2,3,4

.Lrot24:

.long	0x02010003,0x06050407,0x0a09080b,0x0e0d0c0f

.asciz	"ChaCha20 for ARMv8, CRYPTOGAMS by \@dot-asm"

.globl	ChaCha20_ctr32

.type	ChaCha20_ctr32,%function

.align	5

ChaCha20_ctr32:

	cbz	$len,.Labort

	cmp	$len,#192

	b.lo	.Lshort

#ifndef	__KERNEL__

	adrp	x17,OPENSSL_armcap_P

	ldr	w17,[x17,#:lo12:OPENSSL_armcap_P]

	tst	w17,#ARMV7_NEON

	b.ne	.LChaCha20_neon

#endif

.Lshort:

	.inst	0xd503233f			// paciasp

	stp	x29,x30,[sp,#-96]!

	add	x29,sp,#0

	adr	@x[0],.Lsigma

	stp	x19,x20,[sp,#16]

	stp	x21,x22,[sp,#32]

	stp	x23,x24,[sp,#48]

	stp	x25,x26,[sp,#64]

	stp	x27,x28,[sp,#80]

	sub	sp,sp,#64

	ldp	@d[0],@d[1],[@x[0]]		// load sigma

	ldp	@d[2],@d[3],[$key]		// load key

	ldp	@d[4],@d[5],[$key,#16]

	ldp	@d[6],@d[7],[$ctr]		// load counter

#ifdef	__AARCH64EB__

	ror	@d[2],@d[2],#32

	ror	@d[3],@d[3],#32

	ror	@d[4],@d[4],#32

	ror	@d[5],@d[5],#32

	ror	@d[6],@d[6],#32

	ror	@d[7],@d[7],#32

#endif

.Loop_outer:

	mov.32	@x[0],@d[0]			// unpack key block

	lsr	@x[1],@d[0],#32

	mov.32	@x[2],@d[1]

	lsr	@x[3],@d[1],#32

	mov.32	@x[4],@d[2]

	lsr	@x[5],@d[2],#32

	mov.32	@x[6],@d[3]

	lsr	@x[7],@d[3],#32

	mov.32	@x[8],@d[4]

	lsr	@x[9],@d[4],#32

	mov.32	@x[10],@d[5]

	lsr	@x[11],@d[5],#32

	mov.32	@x[12],@d[6]

	lsr	@x[13],@d[6],#32

	mov.32	@x[14],@d[7]

	lsr	@x[15],@d[7],#32

	mov	$ctr,#10

	subs	$len,$len,#64

.Loop:

	sub	$ctr,$ctr,#1

___

	foreach (&ROUND(0, 4, 8,12)) { eval; }

	foreach (&ROUND(0, 5,10,15)) { eval; }

$code.=<<___;

	cbnz	$ctr,.Loop

	add.32	@x[0],@x[0],@d[0]		// accumulate key block

	add	@x[1],@x[1],@d[0],lsr#32

	add.32	@x[2],@x[2],@d[1]

	add	@x[3],@x[3],@d[1],lsr#32

	add.32	@x[4],@x[4],@d[2]

	add	@x[5],@x[5],@d[2],lsr#32

	add.32	@x[6],@x[6],@d[3]

	add	@x[7],@x[7],@d[3],lsr#32

	add.32	@x[8],@x[8],@d[4]

	add	@x[9],@x[9],@d[4],lsr#32

	add.32	@x[10],@x[10],@d[5]

	add	@x[11],@x[11],@d[5],lsr#32

	add.32	@x[12],@x[12],@d[6]

	add	@x[13],@x[13],@d[6],lsr#32

	add.32	@x[14],@x[14],@d[7]

	add	@x[15],@x[15],@d[7],lsr#32

	b.lo	.Ltail

	add	@x[0],@x[0],@x[1],lsl#32	// pack

	add	@x[2],@x[2],@x[3],lsl#32

	ldp	@x[1],@x[3],[$inp,#0]		// load input

	add	@x[4],@x[4],@x[5],lsl#32

	add	@x[6],@x[6],@x[7],lsl#32

	ldp	@x[5],@x[7],[$inp,#16]

	add	@x[8],@x[8],@x[9],lsl#32

	add	@x[10],@x[10],@x[11],lsl#32

	ldp	@x[9],@x[11],[$inp,#32]

	add	@x[12],@x[12],@x[13],lsl#32

	add	@x[14],@x[14],@x[15],lsl#32

	ldp	@x[13],@x[15],[$inp,#48]

	add	$inp,$inp,#64

#ifdef	__AARCH64EB__

	rev	@x[0],@x[0]

	rev	@x[2],@x[2]

	rev	@x[4],@x[4]

	rev	@x[6],@x[6]

	rev	@x[8],@x[8]

	rev	@x[10],@x[10]

	rev	@x[12],@x[12]

	rev	@x[14],@x[14]

#endif

	eor	@x[0],@x[0],@x[1]

	eor	@x[2],@x[2],@x[3]

	eor	@x[4],@x[4],@x[5]

	eor	@x[6],@x[6],@x[7]

	eor	@x[8],@x[8],@x[9]

	eor	@x[10],@x[10],@x[11]

	eor	@x[12],@x[12],@x[13]

	eor	@x[14],@x[14],@x[15]

	stp	@x[0],@x[2],[$out,#0]		// store output

	 add	@d[6],@d[6],#1			// increment counter

	stp	@x[4],@x[6],[$out,#16]

	stp	@x[8],@x[10],[$out,#32]

	stp	@x[12],@x[14],[$out,#48]

	add	$out,$out,#64

	b.hi	.Loop_outer

	ldp	x19,x20,[x29,#16]

	add	sp,sp,#64

	ldp	x21,x22,[x29,#32]

	ldp	x23,x24,[x29,#48]

	ldp	x25,x26,[x29,#64]

	ldp	x27,x28,[x29,#80]

	ldp	x29,x30,[sp],#96

	.inst	0xd50323bf			// autiasp

.Labort:

	ret

.align	4

.Ltail:

	add	$len,$len,#64

.Less_than_64:

	sub	$out,$out,#1

	add	$inp,$inp,$len

	add	$out,$out,$len

	add	$ctr,sp,$len

	neg	$len,$len

	add	@x[0],@x[0],@x[1],lsl#32	// pack

	add	@x[2],@x[2],@x[3],lsl#32

	add	@x[4],@x[4],@x[5],lsl#32

	add	@x[6],@x[6],@x[7],lsl#32

	add	@x[8],@x[8],@x[9],lsl#32

	add	@x[10],@x[10],@x[11],lsl#32

	add	@x[12],@x[12],@x[13],lsl#32

	add	@x[14],@x[14],@x[15],lsl#32

#ifdef	__AARCH64EB__

	rev	@x[0],@x[0]

	rev	@x[2],@x[2]

	rev	@x[4],@x[4]

	rev	@x[6],@x[6]

	rev	@x[8],@x[8]

	rev	@x[10],@x[10]

	rev	@x[12],@x[12]

	rev	@x[14],@x[14]

#endif

	stp	@x[0],@x[2],[sp,#0]

	stp	@x[4],@x[6],[sp,#16]

	stp	@x[8],@x[10],[sp,#32]

	stp	@x[12],@x[14],[sp,#48]

.Loop_tail:

	ldrb	w10,[$inp,$len]

	ldrb	w11,[$ctr,$len]

	add	$len,$len,#1

	eor	w10,w10,w11

	strb	w10,[$out,$len]

	cbnz	$len,.Loop_tail

	stp	xzr,xzr,[sp,#0]

	stp	xzr,xzr,[sp,#16]

	stp	xzr,xzr,[sp,#32]

	stp	xzr,xzr,[sp,#48]

	ldp	x19,x20,[x29,#16]

	add	sp,sp,#64

	ldp	x21,x22,[x29,#32]

	ldp	x23,x24,[x29,#48]

	ldp	x25,x26,[x29,#64]

	ldp	x27,x28,[x29,#80]

	ldp	x29,x30,[sp],#96

	.inst	0xd50323bf			// autiasp

	ret

.size	ChaCha20_ctr32,.-ChaCha20_ctr32

___

{{{

my @K = map("v$_.4s",(0..3));

my ($xt0,$xt1,$xt2,$xt3, $CTR,$ROT24) = map("v$_.4s",(4..9));

my @X = map("v$_.4s",(16,20,24,28, 17,21,25,29, 18,22,26,30, 19,23,27,31));

my ($xa0,$xa1,$xa2,$xa3, $xb0,$xb1,$xb2,$xb3,

    $xc0,$xc1,$xc2,$xc3, $xd0,$xd1,$xd2,$xd3) = @X;

sub NEON_lane_ROUND {

my ($a0,$b0,$c0,$d0)=@_;

my ($a1,$b1,$c1,$d1)=map(($_&~3)+(($_+1)&3),($a0,$b0,$c0,$d0));

my ($a2,$b2,$c2,$d2)=map(($_&~3)+(($_+1)&3),($a1,$b1,$c1,$d1));

my ($a3,$b3,$c3,$d3)=map(($_&~3)+(($_+1)&3),($a2,$b2,$c2,$d2));

my @x=map("'$_'",@X);

	(

	"&add		(@x[$a0],@x[$a0],@x[$b0])",	# Q1

	 "&add		(@x[$a1],@x[$a1],@x[$b1])",	# Q2

	  "&add		(@x[$a2],@x[$a2],@x[$b2])",	# Q3

	   "&add	(@x[$a3],@x[$a3],@x[$b3])",	# Q4

	"&eor		(@x[$d0],@x[$d0],@x[$a0])",

	 "&eor		(@x[$d1],@x[$d1],@x[$a1])",

	  "&eor		(@x[$d2],@x[$d2],@x[$a2])",

	   "&eor	(@x[$d3],@x[$d3],@x[$a3])",

	"&rev32_16	(@x[$d0],@x[$d0])",

	 "&rev32_16	(@x[$d1],@x[$d1])",

	  "&rev32_16	(@x[$d2],@x[$d2])",

	   "&rev32_16	(@x[$d3],@x[$d3])",

	"&add		(@x[$c0],@x[$c0],@x[$d0])",

	 "&add		(@x[$c1],@x[$c1],@x[$d1])",

	  "&add		(@x[$c2],@x[$c2],@x[$d2])",

	   "&add	(@x[$c3],@x[$c3],@x[$d3])",

	"&eor		('$xt0',@x[$b0],@x[$c0])",

	 "&eor		('$xt1',@x[$b1],@x[$c1])",

	  "&eor		('$xt2',@x[$b2],@x[$c2])",

	   "&eor	('$xt3',@x[$b3],@x[$c3])",

	"&ushr		(@x[$b0],'$xt0',20)",

	 "&ushr		(@x[$b1],'$xt1',20)",

	  "&ushr	(@x[$b2],'$xt2',20)",

	   "&ushr	(@x[$b3],'$xt3',20)",

	"&sli		(@x[$b0],'$xt0',12)",

	 "&sli		(@x[$b1],'$xt1',12)",

	  "&sli		(@x[$b2],'$xt2',12)",

	   "&sli	(@x[$b3],'$xt3',12)",

	"&add		(@x[$a0],@x[$a0],@x[$b0])",

	 "&add		(@x[$a1],@x[$a1],@x[$b1])",

	  "&add		(@x[$a2],@x[$a2],@x[$b2])",

	   "&add	(@x[$a3],@x[$a3],@x[$b3])",

	"&eor		('$xt0',@x[$d0],@x[$a0])",

	 "&eor		('$xt1',@x[$d1],@x[$a1])",

	  "&eor		('$xt2',@x[$d2],@x[$a2])",

	   "&eor	('$xt3',@x[$d3],@x[$a3])",

	"&tbl		(@x[$d0],'{$xt0}','$ROT24')",

	 "&tbl		(@x[$d1],'{$xt1}','$ROT24')",

	  "&tbl		(@x[$d2],'{$xt2}','$ROT24')",

	   "&tbl	(@x[$d3],'{$xt3}','$ROT24')",

	"&add		(@x[$c0],@x[$c0],@x[$d0])",

	 "&add		(@x[$c1],@x[$c1],@x[$d1])",

	  "&add		(@x[$c2],@x[$c2],@x[$d2])",

	   "&add	(@x[$c3],@x[$c3],@x[$d3])",

	"&eor		('$xt0',@x[$b0],@x[$c0])",

	 "&eor		('$xt1',@x[$b1],@x[$c1])",

	  "&eor		('$xt2',@x[$b2],@x[$c2])",

	   "&eor	('$xt3',@x[$b3],@x[$c3])",

	"&ushr		(@x[$b0],'$xt0',25)",

	 "&ushr		(@x[$b1],'$xt1',25)",

	  "&ushr	(@x[$b2],'$xt2',25)",

	   "&ushr	(@x[$b3],'$xt3',25)",

	"&sli		(@x[$b0],'$xt0',7)",

	 "&sli		(@x[$b1],'$xt1',7)",

	  "&sli		(@x[$b2],'$xt2',7)",

	   "&sli	(@x[$b3],'$xt3',7)"

	);

}

$code.=<<___;

#ifdef	__KERNEL__

.globl	ChaCha20_neon

#endif

.type	ChaCha20_neon,%function

.align	5

ChaCha20_neon:

.LChaCha20_neon:

	.inst	0xd503233f			// paciasp

	stp	x29,x30,[sp,#-96]!

	add	x29,sp,#0

	adr	@x[0],.Lsigma

	stp	x19,x20,[sp,#16]

	stp	x21,x22,[sp,#32]

	stp	x23,x24,[sp,#48]

	stp	x25,x26,[sp,#64]

	stp	x27,x28,[sp,#80]

	cmp	$len,#512

	b.hs	.L512_or_more_neon

	sub	sp,sp,#64

	ldp	@d[0],@d[1],[@x[0]]		// load sigma

	ld1	{@K[0]},[@x[0]],#16

	ldp	@d[2],@d[3],[$key]		// load key

	ldp	@d[4],@d[5],[$key,#16]

	ld1	{@K[1],@K[2]},[$key]

	ldp	@d[6],@d[7],[$ctr]		// load counter

	ld1	{@K[3]},[$ctr]

	stp	d8,d9,[sp]			// meet ABI requirements

	ld1	{$CTR,$ROT24},[@x[0]]

#ifdef	__AARCH64EB__

	rev64	@K[0],@K[0]

	ror	@d[2],@d[2],#32

	ror	@d[3],@d[3],#32

	ror	@d[4],@d[4],#32

	ror	@d[5],@d[5],#32

	ror	@d[6],@d[6],#32

	ror	@d[7],@d[7],#32

#endif

.Loop_outer_neon:

	dup	$xa0,@{K[0]}[0]			// unpack key block

	 mov.32	@x[0],@d[0]

	dup	$xa1,@{K[0]}[1]

	 lsr	@x[1],@d[0],#32

	dup	$xa2,@{K[0]}[2]

	 mov.32	@x[2],@d[1]

	dup	$xa3,@{K[0]}[3]

	 lsr	@x[3],@d[1],#32

	dup	$xb0,@{K[1]}[0]

	 mov.32	@x[4],@d[2]

	dup	$xb1,@{K[1]}[1]

	 lsr	@x[5],@d[2],#32

	dup	$xb2,@{K[1]}[2]

	 mov.32	@x[6],@d[3]

	dup	$xb3,@{K[1]}[3]

	 lsr	@x[7],@d[3],#32

	dup	$xd0,@{K[3]}[0]

	 mov.32	@x[8],@d[4]

	dup	$xd1,@{K[3]}[1]

	 lsr	@x[9],@d[4],#32

	dup	$xd2,@{K[3]}[2]

	 mov.32	@x[10],@d[5]

	dup	$xd3,@{K[3]}[3]

	 lsr	@x[11],@d[5],#32

	add	$xd0,$xd0,$CTR

	 mov.32	@x[12],@d[6]

	dup	$xc0,@{K[2]}[0]

	 lsr	@x[13],@d[6],#32

	dup	$xc1,@{K[2]}[1]

	 mov.32	@x[14],@d[7]

	dup	$xc2,@{K[2]}[2]

	 lsr	@x[15],@d[7],#32

	dup	$xc3,@{K[2]}[3]

	mov	$ctr,#10

	subs	$len,$len,#320

.Loop_neon:

	sub	$ctr,$ctr,#1

___

	my @plus_one=&ROUND(0,4,8,12);

	foreach (&NEON_lane_ROUND(0,4,8,12))  { eval; eval(shift(@plus_one)); }

	@plus_one=&ROUND(0,5,10,15);

	foreach (&NEON_lane_ROUND(0,5,10,15)) { eval; eval(shift(@plus_one)); }

$code.=<<___;

	cbnz	$ctr,.Loop_neon

	add	$xd0,$xd0,$CTR

	zip1	$xt0,$xa0,$xa1			// transpose data

	zip1	$xt1,$xa2,$xa3

	zip2	$xt2,$xa0,$xa1

	zip2	$xt3,$xa2,$xa3

	zip1.64	$xa0,$xt0,$xt1

	zip2.64	$xa1,$xt0,$xt1

	zip1.64	$xa2,$xt2,$xt3

	zip2.64	$xa3,$xt2,$xt3

	zip1	$xt0,$xb0,$xb1

	zip1	$xt1,$xb2,$xb3

	zip2	$xt2,$xb0,$xb1

	zip2	$xt3,$xb2,$xb3

	zip1.64	$xb0,$xt0,$xt1

	zip2.64	$xb1,$xt0,$xt1

	zip1.64	$xb2,$xt2,$xt3

	zip2.64	$xb3,$xt2,$xt3

	zip1	$xt0,$xc0,$xc1

	 add.32	@x[0],@x[0],@d[0]		// accumulate key block

	zip1	$xt1,$xc2,$xc3

	 add	@x[1],@x[1],@d[0],lsr#32

	zip2	$xt2,$xc0,$xc1

	 add.32	@x[2],@x[2],@d[1]

	zip2	$xt3,$xc2,$xc3

	 add	@x[3],@x[3],@d[1],lsr#32

	zip1.64	$xc0,$xt0,$xt1

	 add.32	@x[4],@x[4],@d[2]

	zip2.64	$xc1,$xt0,$xt1

	 add	@x[5],@x[5],@d[2],lsr#32

	zip1.64	$xc2,$xt2,$xt3

	 add.32	@x[6],@x[6],@d[3]

	zip2.64	$xc3,$xt2,$xt3

	 add	@x[7],@x[7],@d[3],lsr#32

	zip1	$xt0,$xd0,$xd1

	 add.32	@x[8],@x[8],@d[4]

	zip1	$xt1,$xd2,$xd3

	 add	@x[9],@x[9],@d[4],lsr#32

	zip2	$xt2,$xd0,$xd1

	 add.32	@x[10],@x[10],@d[5]

	zip2	$xt3,$xd2,$xd3

	 add	@x[11],@x[11],@d[5],lsr#32

	zip1.64	$xd0,$xt0,$xt1

	 add.32	@x[12],@x[12],@d[6]

	zip2.64	$xd1,$xt0,$xt1

	 add	@x[13],@x[13],@d[6],lsr#32

	zip1.64	$xd2,$xt2,$xt3

	 add.32	@x[14],@x[14],@d[7]

	zip2.64	$xd3,$xt2,$xt3

	 add	@x[15],@x[15],@d[7],lsr#32

	b.lo	.Ltail_neon

	add	@x[0],@x[0],@x[1],lsl#32	// pack

	add	@x[2],@x[2],@x[3],lsl#32

	ldp	@x[1],@x[3],[$inp,#0]		// load input

	 add	$xa0,$xa0,@K[0]			// accumulate key block

	add	@x[4],@x[4],@x[5],lsl#32

	add	@x[6],@x[6],@x[7],lsl#32

	ldp	@x[5],@x[7],[$inp,#16]

	 add	$xb0,$xb0,@K[1]

	add	@x[8],@x[8],@x[9],lsl#32

	add	@x[10],@x[10],@x[11],lsl#32

	ldp	@x[9],@x[11],[$inp,#32]

	 add	$xc0,$xc0,@K[2]

	add	@x[12],@x[12],@x[13],lsl#32

	add	@x[14],@x[14],@x[15],lsl#32

	ldp	@x[13],@x[15],[$inp,#48]

	 add	$xd0,$xd0,@K[3]

	add	$inp,$inp,#64

#ifdef	__AARCH64EB__

	rev	@x[0],@x[0]

	rev	@x[2],@x[2]

	rev	@x[4],@x[4]

	rev	@x[6],@x[6]

	rev	@x[8],@x[8]

	rev	@x[10],@x[10]

	rev	@x[12],@x[12]

	rev	@x[14],@x[14]

#endif

	ld1.8	{$xt0-$xt3},[$inp],#64

	eor	@x[0],@x[0],@x[1]

	 add	$xa1,$xa1,@K[0]

	eor	@x[2],@x[2],@x[3]

	 add	$xb1,$xb1,@K[1]

	eor	@x[4],@x[4],@x[5]

	 add	$xc1,$xc1,@K[2]

	eor	@x[6],@x[6],@x[7]

	 add	$xd1,$xd1,@K[3]

	eor	@x[8],@x[8],@x[9]

	 eor	$xa0,$xa0,$xt0

	 movi	$xt0,#5

	eor	@x[10],@x[10],@x[11]

	 eor	$xb0,$xb0,$xt1

	eor	@x[12],@x[12],@x[13]

	 eor	$xc0,$xc0,$xt2

	eor	@x[14],@x[14],@x[15]

	 eor	$xd0,$xd0,$xt3

	 add	$CTR,$CTR,$xt0			// += 5

	 ld1.8	{$xt0-$xt3},[$inp],#64

	stp	@x[0],@x[2],[$out,#0]		// store output

	 add	@d[6],@d[6],#5			// increment counter

	stp	@x[4],@x[6],[$out,#16]

	stp	@x[8],@x[10],[$out,#32]

	stp	@x[12],@x[14],[$out,#48]

	add	$out,$out,#64

	st1.8	{$xa0-$xd0},[$out],#64

	 add	$xa2,$xa2,@K[0]

	 add	$xb2,$xb2,@K[1]

	 add	$xc2,$xc2,@K[2]

	 add	$xd2,$xd2,@K[3]

	ld1.8	{$xa0-$xd0},[$inp],#64

	eor	$xa1,$xa1,$xt0

	eor	$xb1,$xb1,$xt1

	eor	$xc1,$xc1,$xt2

	eor	$xd1,$xd1,$xt3

	st1.8	{$xa1-$xd1},[$out],#64

	 add	$xa3,$xa3,@K[0]

	 add	$xb3,$xb3,@K[1]

	 add	$xc3,$xc3,@K[2]

	 add	$xd3,$xd3,@K[3]

	ld1.8	{$xa1-$xd1},[$inp],#64

	eor	$xa2,$xa2,$xa0

	eor	$xb2,$xb2,$xb0

	eor	$xc2,$xc2,$xc0

	eor	$xd2,$xd2,$xd0

	st1.8	{$xa2-$xd2},[$out],#64

	eor	$xa3,$xa3,$xa1

	eor	$xb3,$xb3,$xb1

	eor	$xc3,$xc3,$xc1

	eor	$xd3,$xd3,$xd1

	st1.8	{$xa3-$xd3},[$out],#64

	b.hi	.Loop_outer_neon

	ldp	d8,d9,[sp]			// meet ABI requirements

	ldp	x19,x20,[x29,#16]

	add	sp,sp,#64

	ldp	x21,x22,[x29,#32]

	ldp	x23,x24,[x29,#48]

	ldp	x25,x26,[x29,#64]

	ldp	x27,x28,[x29,#80]

	ldp	x29,x30,[sp],#96

	.inst	0xd50323bf			// autiasp

	ret

.align	4

.Ltail_neon:

	add	$len,$len,#320

	ldp	d8,d9,[sp]			// meet ABI requirements

	cmp	$len,#64

	b.lo	.Less_than_64

	add	@x[0],@x[0],@x[1],lsl#32	// pack

	add	@x[2],@x[2],@x[3],lsl#32

	ldp	@x[1],@x[3],[$inp,#0]		// load input

	add	@x[4],@x[4],@x[5],lsl#32

	add	@x[6],@x[6],@x[7],lsl#32

	ldp	@x[5],@x[7],[$inp,#16]

	add	@x[8],@x[8],@x[9],lsl#32

	add	@x[10],@x[10],@x[11],lsl#32

	ldp	@x[9],@x[11],[$inp,#32]

	add	@x[12],@x[12],@x[13],lsl#32

	add	@x[14],@x[14],@x[15],lsl#32

	ldp	@x[13],@x[15],[$inp,#48]

	add	$inp,$inp,#64

#ifdef	__AARCH64EB__

	rev	@x[0],@x[0]

	rev	@x[2],@x[2]

	rev	@x[4],@x[4]

	rev	@x[6],@x[6]

	rev	@x[8],@x[8]

	rev	@x[10],@x[10]

	rev	@x[12],@x[12]

	rev	@x[14],@x[14]

#endif

	eor	@x[0],@x[0],@x[1]

	eor	@x[2],@x[2],@x[3]

	eor	@x[4],@x[4],@x[5]

	eor	@x[6],@x[6],@x[7]

	eor	@x[8],@x[8],@x[9]

	eor	@x[10],@x[10],@x[11]

	eor	@x[12],@x[12],@x[13]

	eor	@x[14],@x[14],@x[15]

	stp	@x[0],@x[2],[$out,#0]		// store output

	 add	$xa0,$xa0,@K[0]			// accumulate key block

	stp	@x[4],@x[6],[$out,#16]

	 add	$xb0,$xb0,@K[1]

	stp	@x[8],@x[10],[$out,#32]

	 add	$xc0,$xc0,@K[2]

	stp	@x[12],@x[14],[$out,#48]

	 add	$xd0,$xd0,@K[3]

	add	$out,$out,#64

	b.eq	.Ldone_neon

	sub	$len,$len,#64

	cmp	$len,#64

	b.lo	.Last_neon

	ld1.8	{$xt0-$xt3},[$inp],#64

	eor	$xa0,$xa0,$xt0

	eor	$xb0,$xb0,$xt1

	eor	$xc0,$xc0,$xt2

	eor	$xd0,$xd0,$xt3

	st1.8	{$xa0-$xd0},[$out],#64

	b.eq	.Ldone_neon

	add	$xa0,$xa1,@K[0]

	add	$xb0,$xb1,@K[1]

	sub	$len,$len,#64

	add	$xc0,$xc1,@K[2]

	cmp	$len,#64

	add	$xd0,$xd1,@K[3]

	b.lo	.Last_neon

	ld1.8	{$xt0-$xt3},[$inp],#64

	eor	$xa1,$xa0,$xt0

	eor	$xb1,$xb0,$xt1

	eor	$xc1,$xc0,$xt2

	eor	$xd1,$xd0,$xt3

	st1.8	{$xa1-$xd1},[$out],#64

	b.eq	.Ldone_neon

	add	$xa0,$xa2,@K[0]

	add	$xb0,$xb2,@K[1]

	sub	$len,$len,#64

	add	$xc0,$xc2,@K[2]

	cmp	$len,#64

	add	$xd0,$xd2,@K[3]

	b.lo	.Last_neon

	ld1.8	{$xt0-$xt3},[$inp],#64

	eor	$xa2,$xa0,$xt0

	eor	$xb2,$xb0,$xt1

	eor	$xc2,$xc0,$xt2

	eor	$xd2,$xd0,$xt3

	st1.8	{$xa2-$xd2},[$out],#64

	b.eq	.Ldone_neon

	add	$xa0,$xa3,@K[0]

	add	$xb0,$xb3,@K[1]

	add	$xc0,$xc3,@K[2]

	add	$xd0,$xd3,@K[3]

	sub	$len,$len,#64

.Last_neon:

	st1.8	{$xa0-$xd0},[sp]

	sub	$out,$out,#1

	add	$inp,$inp,$len

	add	$out,$out,$len

	add	$ctr,sp,$len

	neg	$len,$len

.Loop_tail_neon:

	ldrb	w10,[$inp,$len]

	ldrb	w11,[$ctr,$len]

	add	$len,$len,#1

	eor	w10,w10,w11

	strb	w10,[$out,$len]

	cbnz	$len,.Loop_tail_neon

	stp	xzr,xzr,[sp,#0]

	stp	xzr,xzr,[sp,#16]

	stp	xzr,xzr,[sp,#32]

	stp	xzr,xzr,[sp,#48]

.Ldone_neon:

	ldp	x19,x20,[x29,#16]

	add	sp,sp,#64

	ldp	x21,x22,[x29,#32]

	ldp	x23,x24,[x29,#48]

	ldp	x25,x26,[x29,#64]

	ldp	x27,x28,[x29,#80]

	ldp	x29,x30,[sp],#96

	.inst	0xd50323bf			// autiasp

	ret

.size	ChaCha20_neon,.-ChaCha20_neon

___

{

my @K = map("v$_.4s",(0..6));

my ($T0,$T1,$T2,$T3,$T4,$T5)=@K;

my ($A0,$B0,$C0,$D0,$A1,$B1,$C1,$D1,$A2,$B2,$C2,$D2,

    $A3,$B3,$C3,$D3,$A4,$B4,$C4,$D4,$A5,$B5,$C5,$D5) = map("v$_.4s",(8..31));

my $rot24 = @K[6];

my $ONE = "v7.4s";

sub NEONROUND {

my $odd = pop;

my ($a,$b,$c,$d,$t)=@_;

	(

	"&add		('$a','$a','$b')",

	"&eor		('$d','$d','$a')",

	"&rev32_16	('$d','$d')",		# vrot ($d,16)

	"&add		('$c','$c','$d')",

	"&eor		('$t','$b','$c')",

	"&ushr		('$b','$t',20)",

	"&sli		('$b','$t',12)",

	"&add		('$a','$a','$b')",

	"&eor		('$d','$d','$a')",

	"&tbl		('$d','{$d}','$rot24')",

	"&add		('$c','$c','$d')",

	"&eor		('$t','$b','$c')",

	"&ushr		('$b','$t',25)",

	"&sli		('$b','$t',7)",

	"&ext		('$c','$c','$c',8)",

	"&ext		('$d','$d','$d',$odd?4:12)",

	"&ext		('$b','$b','$b',$odd?12:4)"

	);

}

$code.=<<___;

.type	ChaCha20_512_neon,%function

.align	5

ChaCha20_512_neon:

	.inst	0xd503233f			// paciasp

	stp	x29,x30,[sp,#-96]!

	add	x29,sp,#0

	adr	@x[0],.Lsigma

	stp	x19,x20,[sp,#16]

	stp	x21,x22,[sp,#32]

	stp	x23,x24,[sp,#48]

	stp	x25,x26,[sp,#64]

	stp	x27,x28,[sp,#80]

.L512_or_more_neon:

	sub	sp,sp,#128+64

	eor	$ONE,$ONE,$ONE

	ldp	@d[0],@d[1],[@x[0]]		// load sigma

	ld1	{@K[0]},[@x[0]],#16

	ldp	@d[2],@d[3],[$key]		// load key

	ldp	@d[4],@d[5],[$key,#16]

	ld1	{@K[1],@K[2]},[$key]

	ldp	@d[6],@d[7],[$ctr]		// load counter

	ld1	{@K[3]},[$ctr]

	ld1	{$ONE}[0],[@x[0]]

	add	$key,@x[0],#16			// .Lrot24

#ifdef	__AARCH64EB__

	rev64	@K[0],@K[0]

	ror	@d[2],@d[2],#32

	ror	@d[3],@d[3],#32

	ror	@d[4],@d[4],#32

	ror	@d[5],@d[5],#32

	ror	@d[6],@d[6],#32

	ror	@d[7],@d[7],#32

#endif

	add	@K[3],@K[3],$ONE		// += 1