162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Accelerated CRC-T10DIF using ARM NEON and Crypto Extensions instructions
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2016 Linaro Ltd <ard.biesheuvel@linaro.org>
662306a36Sopenharmony_ci */
762306a36Sopenharmony_ci
862306a36Sopenharmony_ci#include <linux/crc-t10dif.h>
962306a36Sopenharmony_ci#include <linux/init.h>
1062306a36Sopenharmony_ci#include <linux/kernel.h>
1162306a36Sopenharmony_ci#include <linux/module.h>
1262306a36Sopenharmony_ci#include <linux/string.h>
1362306a36Sopenharmony_ci
1462306a36Sopenharmony_ci#include <crypto/internal/hash.h>
1562306a36Sopenharmony_ci#include <crypto/internal/simd.h>
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_ci#include <asm/neon.h>
1862306a36Sopenharmony_ci#include <asm/simd.h>
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_ci#define CRC_T10DIF_PMULL_CHUNK_SIZE	16U
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ciasmlinkage u16 crc_t10dif_pmull(u16 init_crc, const u8 *buf, size_t len);
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic int crct10dif_init(struct shash_desc *desc)
2562306a36Sopenharmony_ci{
2662306a36Sopenharmony_ci	u16 *crc = shash_desc_ctx(desc);
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci	*crc = 0;
2962306a36Sopenharmony_ci	return 0;
3062306a36Sopenharmony_ci}
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_cistatic int crct10dif_update(struct shash_desc *desc, const u8 *data,
3362306a36Sopenharmony_ci			    unsigned int length)
3462306a36Sopenharmony_ci{
3562306a36Sopenharmony_ci	u16 *crc = shash_desc_ctx(desc);
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_ci	if (length >= CRC_T10DIF_PMULL_CHUNK_SIZE && crypto_simd_usable()) {
3862306a36Sopenharmony_ci		kernel_neon_begin();
3962306a36Sopenharmony_ci		*crc = crc_t10dif_pmull(*crc, data, length);
4062306a36Sopenharmony_ci		kernel_neon_end();
4162306a36Sopenharmony_ci	} else {
4262306a36Sopenharmony_ci		*crc = crc_t10dif_generic(*crc, data, length);
4362306a36Sopenharmony_ci	}
4462306a36Sopenharmony_ci
4562306a36Sopenharmony_ci	return 0;
4662306a36Sopenharmony_ci}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_cistatic int crct10dif_final(struct shash_desc *desc, u8 *out)
4962306a36Sopenharmony_ci{
5062306a36Sopenharmony_ci	u16 *crc = shash_desc_ctx(desc);
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci	*(u16 *)out = *crc;
5362306a36Sopenharmony_ci	return 0;
5462306a36Sopenharmony_ci}
5562306a36Sopenharmony_ci
5662306a36Sopenharmony_cistatic struct shash_alg crc_t10dif_alg = {
5762306a36Sopenharmony_ci	.digestsize		= CRC_T10DIF_DIGEST_SIZE,
5862306a36Sopenharmony_ci	.init			= crct10dif_init,
5962306a36Sopenharmony_ci	.update			= crct10dif_update,
6062306a36Sopenharmony_ci	.final			= crct10dif_final,
6162306a36Sopenharmony_ci	.descsize		= CRC_T10DIF_DIGEST_SIZE,
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	.base.cra_name		= "crct10dif",
6462306a36Sopenharmony_ci	.base.cra_driver_name	= "crct10dif-arm-ce",
6562306a36Sopenharmony_ci	.base.cra_priority	= 200,
6662306a36Sopenharmony_ci	.base.cra_blocksize	= CRC_T10DIF_BLOCK_SIZE,
6762306a36Sopenharmony_ci	.base.cra_module	= THIS_MODULE,
6862306a36Sopenharmony_ci};
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_cistatic int __init crc_t10dif_mod_init(void)
7162306a36Sopenharmony_ci{
7262306a36Sopenharmony_ci	if (!(elf_hwcap2 & HWCAP2_PMULL))
7362306a36Sopenharmony_ci		return -ENODEV;
7462306a36Sopenharmony_ci
7562306a36Sopenharmony_ci	return crypto_register_shash(&crc_t10dif_alg);
7662306a36Sopenharmony_ci}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_cistatic void __exit crc_t10dif_mod_exit(void)
7962306a36Sopenharmony_ci{
8062306a36Sopenharmony_ci	crypto_unregister_shash(&crc_t10dif_alg);
8162306a36Sopenharmony_ci}
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_cimodule_init(crc_t10dif_mod_init);
8462306a36Sopenharmony_cimodule_exit(crc_t10dif_mod_exit);
8562306a36Sopenharmony_ci
8662306a36Sopenharmony_ciMODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>");
8762306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
8862306a36Sopenharmony_ciMODULE_ALIAS_CRYPTO("crct10dif");
89