/* SPDX-License-Identifier: GPL-2.0 */
/*
 * Hardware-accelerated CRC-32 variants for Linux on z Systems
 *
 * Use the z/Architecture Vector Extension Facility to accelerate the
 * computing of bitreflected CRC-32 checksums for IEEE 802.3 Ethernet
 * and Castagnoli.
 *
 * This CRC-32 implementation algorithm is bitreflected and processes
 * the least-significant bit first (Little-Endian).
 *
 * Copyright IBM Corp. 2015
 * Author(s): Hendrik Brueckner <brueckner@linux.vnet.ibm.com>
 */

#include <linux/types.h>
#include <asm/fpu.h>
#include "crc32-vx.h"

/* Vector register range containing CRC-32 constants */
#define CONST_PERM_LE2BE	9
#define CONST_R2R1		10
#define CONST_R4R3		11
#define CONST_R5		12
#define CONST_RU_POLY		13
#define CONST_CRC_POLY		14

/*
 * The CRC-32 constant block contains reduction constants to fold and
 * process particular chunks of the input data stream in parallel.
 *
 * For the CRC-32 variants, the constants are precomputed according to
 * these definitions:
 *
 *	R1 = [(x4*128+32 mod P'(x) << 32)]' << 1
 *	R2 = [(x4*128-32 mod P'(x) << 32)]' << 1
 *	R3 = [(x128+32 mod P'(x) << 32)]'   << 1
 *	R4 = [(x128-32 mod P'(x) << 32)]'   << 1
 *	R5 = [(x64 mod P'(x) << 32)]'	    << 1
 *	R6 = [(x32 mod P'(x) << 32)]'	    << 1
 *
 *	The bitreflected Barret reduction constant, u', is defined as
 *	the bit reversal of floor(x**64 / P(x)).
 *
 *	where P(x) is the polynomial in the normal domain and the P'(x) is the
 *	polynomial in the reversed (bitreflected) domain.
 *
 * CRC-32 (IEEE 802.3 Ethernet, ...) polynomials:
 *
 *	P(x)  = 0x04C11DB7
 *	P'(x) = 0xEDB88320
 *
 * CRC-32C (Castagnoli) polynomials:
 *
 *	P(x)  = 0x1EDC6F41
 *	P'(x) = 0x82F63B78
 */

static unsigned long constants_CRC_32_LE[] = {
	0x0f0e0d0c0b0a0908, 0x0706050403020100,	/* BE->LE mask */
	0x1c6e41596, 0x154442bd4,		/* R2, R1 */
	0x0ccaa009e, 0x1751997d0,		/* R4, R3 */
	0x0, 0x163cd6124,			/* R5 */
	0x0, 0x1f7011641,			/* u' */
	0x0, 0x1db710641			/* P'(x) << 1 */
};

static unsigned long constants_CRC_32C_LE[] = {
	0x0f0e0d0c0b0a0908, 0x0706050403020100,	/* BE->LE mask */
	0x09e4addf8, 0x740eef02,		/* R2, R1 */
	0x14cd00bd6, 0xf20c0dfe,		/* R4, R3 */
	0x0, 0x0dd45aab8,			/* R5 */
	0x0, 0x0dea713f1,			/* u' */
	0x0, 0x105ec76f0			/* P'(x) << 1 */
};

/**
 * crc32_le_vgfm_generic - Compute CRC-32 (LE variant) with vector registers
 * @crc: Initial CRC value, typically ~0.
 * @buf: Input buffer pointer, performance might be improved if the
 *	 buffer is on a doubleword boundary.
 * @size: Size of the buffer, must be 64 bytes or greater.
 * @constants: CRC-32 constant pool base pointer.
 *
 * Register usage:
 *	V0:	  Initial CRC value and intermediate constants and results.
 *	V1..V4:	  Data for CRC computation.
 *	V5..V8:	  Next data chunks that are fetched from the input buffer.
 *	V9:	  Constant for BE->LE conversion and shift operations
 *	V10..V14: CRC-32 constants.
 */
static u32 crc32_le_vgfm_generic(u32 crc, unsigned char const *buf, size_t size, unsigned long *constants)
{
	/* Load CRC-32 constants */
	fpu_vlm(CONST_PERM_LE2BE, CONST_CRC_POLY, constants);

	/*
	 * Load the initial CRC value.
	 *
	 * The CRC value is loaded into the rightmost word of the
	 * vector register and is later XORed with the LSB portion
	 * of the loaded input data.
	 */
	fpu_vzero(0);			/* Clear V0 */
	fpu_vlvgf(0, crc, 3);		/* Load CRC into rightmost word */

	/* Load a 64-byte data chunk and XOR with CRC */
	fpu_vlm(1, 4, buf);
	fpu_vperm(1, 1, 1, CONST_PERM_LE2BE);
	fpu_vperm(2, 2, 2, CONST_PERM_LE2BE);
	fpu_vperm(3, 3, 3, CONST_PERM_LE2BE);
	fpu_vperm(4, 4, 4, CONST_PERM_LE2BE);

	fpu_vx(1, 0, 1);		/* V1 ^= CRC */
	buf += 64;
	size -= 64;

	while (size >= 64) {
		fpu_vlm(5, 8, buf);
		fpu_vperm(5, 5, 5, CONST_PERM_LE2BE);
		fpu_vperm(6, 6, 6, CONST_PERM_LE2BE);
		fpu_vperm(7, 7, 7, CONST_PERM_LE2BE);
		fpu_vperm(8, 8, 8, CONST_PERM_LE2BE);
		/*
		 * Perform a GF(2) multiplication of the doublewords in V1 with
		 * the R1 and R2 reduction constants in V0.  The intermediate
		 * result is then folded (accumulated) with the next data chunk
		 * in V5 and stored in V1. Repeat this step for the register
		 * contents in V2, V3, and V4 respectively.
		 */
		fpu_vgfmag(1, CONST_R2R1, 1, 5);
		fpu_vgfmag(2, CONST_R2R1, 2, 6);
		fpu_vgfmag(3, CONST_R2R1, 3, 7);
		fpu_vgfmag(4, CONST_R2R1, 4, 8);
		buf += 64;
		size -= 64;
	}

	/*
	 * Fold V1 to V4 into a single 128-bit value in V1.  Multiply V1 with R3
	 * and R4 and accumulating the next 128-bit chunk until a single 128-bit
	 * value remains.
	 */
	fpu_vgfmag(1, CONST_R4R3, 1, 2);
	fpu_vgfmag(1, CONST_R4R3, 1, 3);
	fpu_vgfmag(1, CONST_R4R3, 1, 4);

	while (size >= 16) {
		fpu_vl(2, buf);
		fpu_vperm(2, 2, 2, CONST_PERM_LE2BE);
		fpu_vgfmag(1, CONST_R4R3, 1, 2);
		buf += 16;
		size -= 16;
	}

	/*
	 * Set up a vector register for byte shifts.  The shift value must
	 * be loaded in bits 1-4 in byte element 7 of a vector register.
	 * Shift by 8 bytes: 0x40
	 * Shift by 4 bytes: 0x20
	 */
	fpu_vleib(9, 0x40, 7);

	/*
	 * Prepare V0 for the next GF(2) multiplication: shift V0 by 8 bytes
	 * to move R4 into the rightmost doubleword and set the leftmost
	 * doubleword to 0x1.
	 */
	fpu_vsrlb(0, CONST_R4R3, 9);
	fpu_vleig(0, 1, 0);

	/*
	 * Compute GF(2) product of V1 and V0.	The rightmost doubleword
	 * of V1 is multiplied with R4.  The leftmost doubleword of V1 is
	 * multiplied by 0x1 and is then XORed with rightmost product.
	 * Implicitly, the intermediate leftmost product becomes padded
	 */
	fpu_vgfmg(1, 0, 1);

	/*
	 * Now do the final 32-bit fold by multiplying the rightmost word
	 * in V1 with R5 and XOR the result with the remaining bits in V1.
	 *
	 * To achieve this by a single VGFMAG, right shift V1 by a word
	 * and store the result in V2 which is then accumulated.  Use the
	 * vector unpack instruction to load the rightmost half of the
	 * doubleword into the rightmost doubleword element of V1; the other
	 * half is loaded in the leftmost doubleword.
	 * The vector register with CONST_R5 contains the R5 constant in the
	 * rightmost doubleword and the leftmost doubleword is zero to ignore
	 * the leftmost product of V1.
	 */
	fpu_vleib(9, 0x20, 7);		  /* Shift by words */
	fpu_vsrlb(2, 1, 9);		  /* Store remaining bits in V2 */
	fpu_vupllf(1, 1);		  /* Split rightmost doubleword */
	fpu_vgfmag(1, CONST_R5, 1, 2);	  /* V1 = (V1 * R5) XOR V2 */

	/*
	 * Apply a Barret reduction to compute the final 32-bit CRC value.
	 *
	 * The input values to the Barret reduction are the degree-63 polynomial
	 * in V1 (R(x)), degree-32 generator polynomial, and the reduction
	 * constant u.	The Barret reduction result is the CRC value of R(x) mod
	 * P(x).
	 *
	 * The Barret reduction algorithm is defined as:
	 *
	 *    1. T1(x) = floor( R(x) / x^32 ) GF2MUL u
	 *    2. T2(x) = floor( T1(x) / x^32 ) GF2MUL P(x)
	 *    3. C(x)  = R(x) XOR T2(x) mod x^32
	 *
	 *  Note: The leftmost doubleword of vector register containing
	 *  CONST_RU_POLY is zero and, thus, the intermediate GF(2) product
	 *  is zero and does not contribute to the final result.
	 */

	/* T1(x) = floor( R(x) / x^32 ) GF2MUL u */
	fpu_vupllf(2, 1);
	fpu_vgfmg(2, CONST_RU_POLY, 2);

	/*
	 * Compute the GF(2) product of the CRC polynomial with T1(x) in
	 * V2 and XOR the intermediate result, T2(x), with the value in V1.
	 * The final result is stored in word element 2 of V2.
	 */
	fpu_vupllf(2, 2);
	fpu_vgfmag(2, CONST_CRC_POLY, 2, 1);

	return fpu_vlgvf(2, 2);
}

u32 crc32_le_vgfm_16(u32 crc, unsigned char const *buf, size_t size)
{
	return crc32_le_vgfm_generic(crc, buf, size, &constants_CRC_32_LE[0]);
}

u32 crc32c_le_vgfm_16(u32 crc, unsigned char const *buf, size_t size)
{
	return crc32_le_vgfm_generic(crc, buf, size, &constants_CRC_32C_LE[0]);
}