162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/* Glue code for SHA512 hashing optimized for sparc64 crypto opcodes.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * This is based largely upon crypto/sha512_generic.c
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci * Copyright (c) Jean-Luc Cooke <jlcooke@certainkey.com>
762306a36Sopenharmony_ci * Copyright (c) Andrew McDonald <andrew@mcdonald.org.uk>
862306a36Sopenharmony_ci * Copyright (c) 2003 Kyle McMartin <kyle@debian.org>
962306a36Sopenharmony_ci */
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_ci#define pr_fmt(fmt)	KBUILD_MODNAME ": " fmt
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ci#include <crypto/internal/hash.h>
1462306a36Sopenharmony_ci#include <linux/init.h>
1562306a36Sopenharmony_ci#include <linux/module.h>
1662306a36Sopenharmony_ci#include <linux/mm.h>
1762306a36Sopenharmony_ci#include <linux/types.h>
1862306a36Sopenharmony_ci#include <crypto/sha2.h>
1962306a36Sopenharmony_ci#include <crypto/sha512_base.h>
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci#include <asm/pstate.h>
2262306a36Sopenharmony_ci#include <asm/elf.h>
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci#include "opcodes.h"
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ciasmlinkage void sha512_sparc64_transform(u64 *digest, const char *data,
2762306a36Sopenharmony_ci					 unsigned int rounds);
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_cistatic void __sha512_sparc64_update(struct sha512_state *sctx, const u8 *data,
3062306a36Sopenharmony_ci				    unsigned int len, unsigned int partial)
3162306a36Sopenharmony_ci{
3262306a36Sopenharmony_ci	unsigned int done = 0;
3362306a36Sopenharmony_ci
3462306a36Sopenharmony_ci	if ((sctx->count[0] += len) < len)
3562306a36Sopenharmony_ci		sctx->count[1]++;
3662306a36Sopenharmony_ci	if (partial) {
3762306a36Sopenharmony_ci		done = SHA512_BLOCK_SIZE - partial;
3862306a36Sopenharmony_ci		memcpy(sctx->buf + partial, data, done);
3962306a36Sopenharmony_ci		sha512_sparc64_transform(sctx->state, sctx->buf, 1);
4062306a36Sopenharmony_ci	}
4162306a36Sopenharmony_ci	if (len - done >= SHA512_BLOCK_SIZE) {
4262306a36Sopenharmony_ci		const unsigned int rounds = (len - done) / SHA512_BLOCK_SIZE;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci		sha512_sparc64_transform(sctx->state, data + done, rounds);
4562306a36Sopenharmony_ci		done += rounds * SHA512_BLOCK_SIZE;
4662306a36Sopenharmony_ci	}
4762306a36Sopenharmony_ci
4862306a36Sopenharmony_ci	memcpy(sctx->buf, data + done, len - done);
4962306a36Sopenharmony_ci}
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_cistatic int sha512_sparc64_update(struct shash_desc *desc, const u8 *data,
5262306a36Sopenharmony_ci				 unsigned int len)
5362306a36Sopenharmony_ci{
5462306a36Sopenharmony_ci	struct sha512_state *sctx = shash_desc_ctx(desc);
5562306a36Sopenharmony_ci	unsigned int partial = sctx->count[0] % SHA512_BLOCK_SIZE;
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_ci	/* Handle the fast case right here */
5862306a36Sopenharmony_ci	if (partial + len < SHA512_BLOCK_SIZE) {
5962306a36Sopenharmony_ci		if ((sctx->count[0] += len) < len)
6062306a36Sopenharmony_ci			sctx->count[1]++;
6162306a36Sopenharmony_ci		memcpy(sctx->buf + partial, data, len);
6262306a36Sopenharmony_ci	} else
6362306a36Sopenharmony_ci		__sha512_sparc64_update(sctx, data, len, partial);
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ci	return 0;
6662306a36Sopenharmony_ci}
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_cistatic int sha512_sparc64_final(struct shash_desc *desc, u8 *out)
6962306a36Sopenharmony_ci{
7062306a36Sopenharmony_ci	struct sha512_state *sctx = shash_desc_ctx(desc);
7162306a36Sopenharmony_ci	unsigned int i, index, padlen;
7262306a36Sopenharmony_ci	__be64 *dst = (__be64 *)out;
7362306a36Sopenharmony_ci	__be64 bits[2];
7462306a36Sopenharmony_ci	static const u8 padding[SHA512_BLOCK_SIZE] = { 0x80, };
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci	/* Save number of bits */
7762306a36Sopenharmony_ci	bits[1] = cpu_to_be64(sctx->count[0] << 3);
7862306a36Sopenharmony_ci	bits[0] = cpu_to_be64(sctx->count[1] << 3 | sctx->count[0] >> 61);
7962306a36Sopenharmony_ci
8062306a36Sopenharmony_ci	/* Pad out to 112 mod 128 and append length */
8162306a36Sopenharmony_ci	index = sctx->count[0] % SHA512_BLOCK_SIZE;
8262306a36Sopenharmony_ci	padlen = (index < 112) ? (112 - index) : ((SHA512_BLOCK_SIZE+112) - index);
8362306a36Sopenharmony_ci
8462306a36Sopenharmony_ci	/* We need to fill a whole block for __sha512_sparc64_update() */
8562306a36Sopenharmony_ci	if (padlen <= 112) {
8662306a36Sopenharmony_ci		if ((sctx->count[0] += padlen) < padlen)
8762306a36Sopenharmony_ci			sctx->count[1]++;
8862306a36Sopenharmony_ci		memcpy(sctx->buf + index, padding, padlen);
8962306a36Sopenharmony_ci	} else {
9062306a36Sopenharmony_ci		__sha512_sparc64_update(sctx, padding, padlen, index);
9162306a36Sopenharmony_ci	}
9262306a36Sopenharmony_ci	__sha512_sparc64_update(sctx, (const u8 *)&bits, sizeof(bits), 112);
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ci	/* Store state in digest */
9562306a36Sopenharmony_ci	for (i = 0; i < 8; i++)
9662306a36Sopenharmony_ci		dst[i] = cpu_to_be64(sctx->state[i]);
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_ci	/* Wipe context */
9962306a36Sopenharmony_ci	memset(sctx, 0, sizeof(*sctx));
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_ci	return 0;
10262306a36Sopenharmony_ci}
10362306a36Sopenharmony_ci
10462306a36Sopenharmony_cistatic int sha384_sparc64_final(struct shash_desc *desc, u8 *hash)
10562306a36Sopenharmony_ci{
10662306a36Sopenharmony_ci	u8 D[64];
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_ci	sha512_sparc64_final(desc, D);
10962306a36Sopenharmony_ci
11062306a36Sopenharmony_ci	memcpy(hash, D, 48);
11162306a36Sopenharmony_ci	memzero_explicit(D, 64);
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	return 0;
11462306a36Sopenharmony_ci}
11562306a36Sopenharmony_ci
11662306a36Sopenharmony_cistatic struct shash_alg sha512 = {
11762306a36Sopenharmony_ci	.digestsize	=	SHA512_DIGEST_SIZE,
11862306a36Sopenharmony_ci	.init		=	sha512_base_init,
11962306a36Sopenharmony_ci	.update		=	sha512_sparc64_update,
12062306a36Sopenharmony_ci	.final		=	sha512_sparc64_final,
12162306a36Sopenharmony_ci	.descsize	=	sizeof(struct sha512_state),
12262306a36Sopenharmony_ci	.base		=	{
12362306a36Sopenharmony_ci		.cra_name	=	"sha512",
12462306a36Sopenharmony_ci		.cra_driver_name=	"sha512-sparc64",
12562306a36Sopenharmony_ci		.cra_priority	=	SPARC_CR_OPCODE_PRIORITY,
12662306a36Sopenharmony_ci		.cra_blocksize	=	SHA512_BLOCK_SIZE,
12762306a36Sopenharmony_ci		.cra_module	=	THIS_MODULE,
12862306a36Sopenharmony_ci	}
12962306a36Sopenharmony_ci};
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_cistatic struct shash_alg sha384 = {
13262306a36Sopenharmony_ci	.digestsize	=	SHA384_DIGEST_SIZE,
13362306a36Sopenharmony_ci	.init		=	sha384_base_init,
13462306a36Sopenharmony_ci	.update		=	sha512_sparc64_update,
13562306a36Sopenharmony_ci	.final		=	sha384_sparc64_final,
13662306a36Sopenharmony_ci	.descsize	=	sizeof(struct sha512_state),
13762306a36Sopenharmony_ci	.base		=	{
13862306a36Sopenharmony_ci		.cra_name	=	"sha384",
13962306a36Sopenharmony_ci		.cra_driver_name=	"sha384-sparc64",
14062306a36Sopenharmony_ci		.cra_priority	=	SPARC_CR_OPCODE_PRIORITY,
14162306a36Sopenharmony_ci		.cra_blocksize	=	SHA384_BLOCK_SIZE,
14262306a36Sopenharmony_ci		.cra_module	=	THIS_MODULE,
14362306a36Sopenharmony_ci	}
14462306a36Sopenharmony_ci};
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_cistatic bool __init sparc64_has_sha512_opcode(void)
14762306a36Sopenharmony_ci{
14862306a36Sopenharmony_ci	unsigned long cfr;
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_ci	if (!(sparc64_elf_hwcap & HWCAP_SPARC_CRYPTO))
15162306a36Sopenharmony_ci		return false;
15262306a36Sopenharmony_ci
15362306a36Sopenharmony_ci	__asm__ __volatile__("rd %%asr26, %0" : "=r" (cfr));
15462306a36Sopenharmony_ci	if (!(cfr & CFR_SHA512))
15562306a36Sopenharmony_ci		return false;
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	return true;
15862306a36Sopenharmony_ci}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_cistatic int __init sha512_sparc64_mod_init(void)
16162306a36Sopenharmony_ci{
16262306a36Sopenharmony_ci	if (sparc64_has_sha512_opcode()) {
16362306a36Sopenharmony_ci		int ret = crypto_register_shash(&sha384);
16462306a36Sopenharmony_ci		if (ret < 0)
16562306a36Sopenharmony_ci			return ret;
16662306a36Sopenharmony_ci
16762306a36Sopenharmony_ci		ret = crypto_register_shash(&sha512);
16862306a36Sopenharmony_ci		if (ret < 0) {
16962306a36Sopenharmony_ci			crypto_unregister_shash(&sha384);
17062306a36Sopenharmony_ci			return ret;
17162306a36Sopenharmony_ci		}
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_ci		pr_info("Using sparc64 sha512 opcode optimized SHA-512/SHA-384 implementation\n");
17462306a36Sopenharmony_ci		return 0;
17562306a36Sopenharmony_ci	}
17662306a36Sopenharmony_ci	pr_info("sparc64 sha512 opcode not available.\n");
17762306a36Sopenharmony_ci	return -ENODEV;
17862306a36Sopenharmony_ci}
17962306a36Sopenharmony_ci
18062306a36Sopenharmony_cistatic void __exit sha512_sparc64_mod_fini(void)
18162306a36Sopenharmony_ci{
18262306a36Sopenharmony_ci	crypto_unregister_shash(&sha384);
18362306a36Sopenharmony_ci	crypto_unregister_shash(&sha512);
18462306a36Sopenharmony_ci}
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_cimodule_init(sha512_sparc64_mod_init);
18762306a36Sopenharmony_cimodule_exit(sha512_sparc64_mod_fini);
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_ciMODULE_LICENSE("GPL");
19062306a36Sopenharmony_ciMODULE_DESCRIPTION("SHA-384 and SHA-512 Secure Hash Algorithm, sparc64 sha512 opcode accelerated");
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ciMODULE_ALIAS_CRYPTO("sha384");
19362306a36Sopenharmony_ciMODULE_ALIAS_CRYPTO("sha512");
19462306a36Sopenharmony_ci
19562306a36Sopenharmony_ci#include "crop_devid.c"
196