/* $OpenBSD: sha512.c,v 1.38 2023/05/19 00:54:28 deraadt Exp $ */
/* ====================================================================
 * Copyright (c) 1998-2011 The OpenSSL Project.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. All advertising materials mentioning features or use of this
 *    software must display the following acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit. (http://www.openssl.org/)"
 *
 * 4. The names "OpenSSL Toolkit" and "OpenSSL Project" must not be used to
 *    endorse or promote products derived from this software without
 *    prior written permission. For written permission, please contact
 *    openssl-core@openssl.org.
 *
 * 5. Products derived from this software may not be called "OpenSSL"
 *    nor may "OpenSSL" appear in their names without prior written
 *    permission of the OpenSSL Project.
 *
 * 6. Redistributions of any form whatsoever must retain the following
 *    acknowledgment:
 *    "This product includes software developed by the OpenSSL Project
 *    for use in the OpenSSL Toolkit (http://www.openssl.org/)"
 *
 * THIS SOFTWARE IS PROVIDED BY THE OpenSSL PROJECT ``AS IS'' AND ANY
 * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE OpenSSL PROJECT OR
 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
 * OF THE POSSIBILITY OF SUCH DAMAGE.
 * ====================================================================
 *
 * This product includes cryptographic software written by Eric Young
 * (eay@cryptsoft.com).  This product includes software written by Tim
 * Hudson (tjh@cryptsoft.com).
 */

#include <endian.h>
#include <stdlib.h>
#include <string.h>

#include <openssl/opensslconf.h>

#include <openssl/crypto.h>
#include <openssl/sha.h>

#include "crypto_internal.h"
#include "sha_internal.h"

#if !defined(OPENSSL_NO_SHA) && !defined(OPENSSL_NO_SHA512)

#if !defined(__STRICT_ALIGNMENT) || defined(SHA512_ASM)
#define SHA512_BLOCK_CAN_MANAGE_UNALIGNED_DATA
#endif

#ifdef SHA512_ASM
void sha512_block_data_order(SHA512_CTX *ctx, const void *in, size_t num);
#endif

#ifndef SHA512_ASM
static const SHA_LONG64 K512[80] = {
	U64(0x428a2f98d728ae22), U64(0x7137449123ef65cd),
	U64(0xb5c0fbcfec4d3b2f), U64(0xe9b5dba58189dbbc),
	U64(0x3956c25bf348b538), U64(0x59f111f1b605d019),
	U64(0x923f82a4af194f9b), U64(0xab1c5ed5da6d8118),
	U64(0xd807aa98a3030242), U64(0x12835b0145706fbe),
	U64(0x243185be4ee4b28c), U64(0x550c7dc3d5ffb4e2),
	U64(0x72be5d74f27b896f), U64(0x80deb1fe3b1696b1),
	U64(0x9bdc06a725c71235), U64(0xc19bf174cf692694),
	U64(0xe49b69c19ef14ad2), U64(0xefbe4786384f25e3),
	U64(0x0fc19dc68b8cd5b5), U64(0x240ca1cc77ac9c65),
	U64(0x2de92c6f592b0275), U64(0x4a7484aa6ea6e483),
	U64(0x5cb0a9dcbd41fbd4), U64(0x76f988da831153b5),
	U64(0x983e5152ee66dfab), U64(0xa831c66d2db43210),
	U64(0xb00327c898fb213f), U64(0xbf597fc7beef0ee4),
	U64(0xc6e00bf33da88fc2), U64(0xd5a79147930aa725),
	U64(0x06ca6351e003826f), U64(0x142929670a0e6e70),
	U64(0x27b70a8546d22ffc), U64(0x2e1b21385c26c926),
	U64(0x4d2c6dfc5ac42aed), U64(0x53380d139d95b3df),
	U64(0x650a73548baf63de), U64(0x766a0abb3c77b2a8),
	U64(0x81c2c92e47edaee6), U64(0x92722c851482353b),
	U64(0xa2bfe8a14cf10364), U64(0xa81a664bbc423001),
	U64(0xc24b8b70d0f89791), U64(0xc76c51a30654be30),
	U64(0xd192e819d6ef5218), U64(0xd69906245565a910),
	U64(0xf40e35855771202a), U64(0x106aa07032bbd1b8),
	U64(0x19a4c116b8d2d0c8), U64(0x1e376c085141ab53),
	U64(0x2748774cdf8eeb99), U64(0x34b0bcb5e19b48a8),
	U64(0x391c0cb3c5c95a63), U64(0x4ed8aa4ae3418acb),
	U64(0x5b9cca4f7763e373), U64(0x682e6ff3d6b2b8a3),
	U64(0x748f82ee5defb2fc), U64(0x78a5636f43172f60),
	U64(0x84c87814a1f0ab72), U64(0x8cc702081a6439ec),
	U64(0x90befffa23631e28), U64(0xa4506cebde82bde9),
	U64(0xbef9a3f7b2c67915), U64(0xc67178f2e372532b),
	U64(0xca273eceea26619c), U64(0xd186b8c721c0c207),
	U64(0xeada7dd6cde0eb1e), U64(0xf57d4f7fee6ed178),
	U64(0x06f067aa72176fba), U64(0x0a637dc5a2c898a6),
	U64(0x113f9804bef90dae), U64(0x1b710b35131c471b),
	U64(0x28db77f523047d84), U64(0x32caab7b40c72493),
	U64(0x3c9ebe0a15c9bebc), U64(0x431d67c49c100d4c),
	U64(0x4cc5d4becb3e42b6), U64(0x597f299cfc657e2a),
	U64(0x5fcb6fab3ad6faec), U64(0x6c44198c4a475817),
};

#if defined(__GNUC__) && __GNUC__>=2 && !defined(OPENSSL_NO_ASM) && !defined(OPENSSL_NO_INLINE_ASM)
# if defined(__x86_64) || defined(__x86_64__)
#   define PULL64(x) ({ SHA_LONG64 ret=*((const SHA_LONG64 *)(&(x)));	\
				asm ("bswapq	%0"		\
				: "=r"(ret)			\
				: "0"(ret)); ret;		})
# elif (defined(__i386) || defined(__i386__))
#   define PULL64(x) ({ const unsigned int *p=(const unsigned int *)(&(x));\
			 unsigned int hi=p[0],lo=p[1];		\
				asm ("bswapl %0; bswapl %1;"	\
				: "=r"(lo),"=r"(hi)		\
				: "0"(lo),"1"(hi));		\
				((SHA_LONG64)hi)<<32|lo;	})
# endif
#endif

#ifndef PULL64
#if BYTE_ORDER == BIG_ENDIAN
#define PULL64(x)	(x)
#else
#define B(x, j)		(((SHA_LONG64)(*(((const unsigned char *)(&x))+j)))<<((7-j)*8))
#define PULL64(x)	(B(x,0)|B(x,1)|B(x,2)|B(x,3)|B(x,4)|B(x,5)|B(x,6)|B(x,7))
#endif
#endif

#define ROTR(x, s)	crypto_ror_u64(x, s)

#define Sigma0(x)	(ROTR((x),28) ^ ROTR((x),34) ^ ROTR((x),39))
#define Sigma1(x)	(ROTR((x),14) ^ ROTR((x),18) ^ ROTR((x),41))
#define sigma0(x)	(ROTR((x),1)  ^ ROTR((x),8)  ^ ((x)>>7))
#define sigma1(x)	(ROTR((x),19) ^ ROTR((x),61) ^ ((x)>>6))

#define Ch(x, y, z)	(((x) & (y)) ^ ((~(x)) & (z)))
#define Maj(x, y, z)	(((x) & (y)) ^ ((x) & (z)) ^ ((y) & (z)))

#define	ROUND_00_15(i, a, b, c, d, e, f, g, h, Wt)		do {	\
	T1 = h + Sigma1(e) + Ch(e, f, g) + K512[i] + Wt;		\
	T2 = Sigma0(a) + Maj(a, b, c);					\
	d += T1;							\
	h = T1 + T2;							\
								} while (0)

#define	ROUND_16_80(i, j, a, b, c, d, e, f, g, h, X)		do {	\
	s0 = sigma0(X[(j + 1) & 0x0f]);					\
	s1 = sigma1(X[(j + 14) & 0x0f]);				\
	X[(j) & 0x0f] += s0 + s1 + X[(j + 9) & 0x0f];			\
	ROUND_00_15(i + j, a, b, c, d, e, f, g, h, X[(j) & 0x0f]);	\
								} while (0)

static void
sha512_block_data_order(SHA512_CTX *ctx, const void *_in, size_t num)
{
	const SHA_LONG64 *in = _in;
	SHA_LONG64 a, b, c, d, e, f, g, h, s0, s1, T1, T2;
	SHA_LONG64 X[16];
	int i;

	while (num--) {
		a = ctx->h[0];
		b = ctx->h[1];
		c = ctx->h[2];
		d = ctx->h[3];
		e = ctx->h[4];
		f = ctx->h[5];
		g = ctx->h[6];
		h = ctx->h[7];

		X[0] = PULL64(in[0]);
		ROUND_00_15(0, a, b, c, d, e, f, g, h, X[0]);
		X[1] = PULL64(in[1]);
		ROUND_00_15(1, h, a, b, c, d, e, f, g, X[1]);
		X[2] = PULL64(in[2]);
		ROUND_00_15(2, g, h, a, b, c, d, e, f, X[2]);
		X[3] = PULL64(in[3]);
		ROUND_00_15(3, f, g, h, a, b, c, d, e, X[3]);
		X[4] = PULL64(in[4]);
		ROUND_00_15(4, e, f, g, h, a, b, c, d, X[4]);
		X[5] = PULL64(in[5]);
		ROUND_00_15(5, d, e, f, g, h, a, b, c, X[5]);
		X[6] = PULL64(in[6]);
		ROUND_00_15(6, c, d, e, f, g, h, a, b, X[6]);
		X[7] = PULL64(in[7]);
		ROUND_00_15(7, b, c, d, e, f, g, h, a, X[7]);
		X[8] = PULL64(in[8]);
		ROUND_00_15(8, a, b, c, d, e, f, g, h, X[8]);
		X[9] = PULL64(in[9]);
		ROUND_00_15(9, h, a, b, c, d, e, f, g, X[9]);
		X[10] = PULL64(in[10]);
		ROUND_00_15(10, g, h, a, b, c, d, e, f, X[10]);
		X[11] = PULL64(in[11]);
		ROUND_00_15(11, f, g, h, a, b, c, d, e, X[11]);
		X[12] = PULL64(in[12]);
		ROUND_00_15(12, e, f, g, h, a, b, c, d, X[12]);
		X[13] = PULL64(in[13]);
		ROUND_00_15(13, d, e, f, g, h, a, b, c, X[13]);
		X[14] = PULL64(in[14]);
		ROUND_00_15(14, c, d, e, f, g, h, a, b, X[14]);
		X[15] = PULL64(in[15]);
		ROUND_00_15(15, b, c, d, e, f, g, h, a, X[15]);

		for (i = 16; i < 80; i += 16) {
			ROUND_16_80(i, 0, a, b, c, d, e, f, g, h, X);
			ROUND_16_80(i, 1, h, a, b, c, d, e, f, g, X);
			ROUND_16_80(i, 2, g, h, a, b, c, d, e, f, X);
			ROUND_16_80(i, 3, f, g, h, a, b, c, d, e, X);
			ROUND_16_80(i, 4, e, f, g, h, a, b, c, d, X);
			ROUND_16_80(i, 5, d, e, f, g, h, a, b, c, X);
			ROUND_16_80(i, 6, c, d, e, f, g, h, a, b, X);
			ROUND_16_80(i, 7, b, c, d, e, f, g, h, a, X);
			ROUND_16_80(i, 8, a, b, c, d, e, f, g, h, X);
			ROUND_16_80(i, 9, h, a, b, c, d, e, f, g, X);
			ROUND_16_80(i, 10, g, h, a, b, c, d, e, f, X);
			ROUND_16_80(i, 11, f, g, h, a, b, c, d, e, X);
			ROUND_16_80(i, 12, e, f, g, h, a, b, c, d, X);
			ROUND_16_80(i, 13, d, e, f, g, h, a, b, c, X);
			ROUND_16_80(i, 14, c, d, e, f, g, h, a, b, X);
			ROUND_16_80(i, 15, b, c, d, e, f, g, h, a, X);
		}

		ctx->h[0] += a;
		ctx->h[1] += b;
		ctx->h[2] += c;
		ctx->h[3] += d;
		ctx->h[4] += e;
		ctx->h[5] += f;
		ctx->h[6] += g;
		ctx->h[7] += h;

		in += SHA_LBLOCK;
	}
}

#endif /* SHA512_ASM */

int
SHA384_Init(SHA512_CTX *c)
{
	memset(c, 0, sizeof(*c));

	c->h[0] = U64(0xcbbb9d5dc1059ed8);
	c->h[1] = U64(0x629a292a367cd507);
	c->h[2] = U64(0x9159015a3070dd17);
	c->h[3] = U64(0x152fecd8f70e5939);
	c->h[4] = U64(0x67332667ffc00b31);
	c->h[5] = U64(0x8eb44a8768581511);
	c->h[6] = U64(0xdb0c2e0d64f98fa7);
	c->h[7] = U64(0x47b5481dbefa4fa4);

	c->md_len = SHA384_DIGEST_LENGTH;

	return 1;
}

int
SHA384_Update(SHA512_CTX *c, const void *data, size_t len)
{
	return SHA512_Update(c, data, len);
}

int
SHA384_Final(unsigned char *md, SHA512_CTX *c)
{
	return SHA512_Final(md, c);
}

unsigned char *
SHA384(const unsigned char *d, size_t n, unsigned char *md)
{
	SHA512_CTX c;
	static unsigned char m[SHA384_DIGEST_LENGTH];

	if (md == NULL)
		md = m;

	SHA384_Init(&c);
	SHA512_Update(&c, d, n);
	SHA512_Final(md, &c);

	explicit_bzero(&c, sizeof(c));

	return (md);
}

int
SHA512_Init(SHA512_CTX *c)
{
	memset(c, 0, sizeof(*c));

	c->h[0] = U64(0x6a09e667f3bcc908);
	c->h[1] = U64(0xbb67ae8584caa73b);
	c->h[2] = U64(0x3c6ef372fe94f82b);
	c->h[3] = U64(0xa54ff53a5f1d36f1);
	c->h[4] = U64(0x510e527fade682d1);
	c->h[5] = U64(0x9b05688c2b3e6c1f);
	c->h[6] = U64(0x1f83d9abfb41bd6b);
	c->h[7] = U64(0x5be0cd19137e2179);

	c->md_len = SHA512_DIGEST_LENGTH;

	return 1;
}

void
SHA512_Transform(SHA512_CTX *c, const unsigned char *data)
{
#ifndef SHA512_BLOCK_CAN_MANAGE_UNALIGNED_DATA
	if ((size_t)data % sizeof(c->u.d[0]) != 0) {
		memcpy(c->u.p, data, sizeof(c->u.p));
		data = c->u.p;
	}
#endif
	sha512_block_data_order(c, data, 1);
}

int
SHA512_Update(SHA512_CTX *c, const void *_data, size_t len)
{
	SHA_LONG64	l;
	unsigned char  *p = c->u.p;
	const unsigned char *data = (const unsigned char *)_data;

	if (len == 0)
		return 1;

	l = (c->Nl + (((SHA_LONG64)len) << 3))&U64(0xffffffffffffffff);
	if (l < c->Nl)
		c->Nh++;
	if (sizeof(len) >= 8)
		c->Nh += (((SHA_LONG64)len) >> 61);
	c->Nl = l;

	if (c->num != 0) {
		size_t n = sizeof(c->u) - c->num;

		if (len < n) {
			memcpy(p + c->num, data, len);
			c->num += (unsigned int)len;
			return 1;
		} else{
			memcpy(p + c->num, data, n);
			c->num = 0;
			len -= n;
			data += n;
			sha512_block_data_order(c, p, 1);
		}
	}

	if (len >= sizeof(c->u)) {
#ifndef SHA512_BLOCK_CAN_MANAGE_UNALIGNED_DATA
		if ((size_t)data % sizeof(c->u.d[0]) != 0) {
			while (len >= sizeof(c->u)) {
				memcpy(p, data, sizeof(c->u));
				sha512_block_data_order(c, p, 1);
				len -= sizeof(c->u);
				data += sizeof(c->u);
			}
		} else
#endif
		{
			sha512_block_data_order(c, data, len/sizeof(c->u));
			data += len;
			len %= sizeof(c->u);
			data -= len;
		}
	}

	if (len != 0) {
		memcpy(p, data, len);
		c->num = (int)len;
	}

	return 1;
}

int
SHA512_Final(unsigned char *md, SHA512_CTX *c)
{
	unsigned char *p = (unsigned char *)c->u.p;
	size_t n = c->num;

	p[n]=0x80;	/* There always is a room for one */
	n++;
	if (n > (sizeof(c->u) - 16)) {
		memset(p + n, 0, sizeof(c->u) - n);
		n = 0;
		sha512_block_data_order(c, p, 1);
	}

	memset(p + n, 0, sizeof(c->u) - 16 - n);
	c->u.d[SHA_LBLOCK - 2] = htobe64(c->Nh);
	c->u.d[SHA_LBLOCK - 1] = htobe64(c->Nl);

	sha512_block_data_order(c, p, 1);

	if (md == NULL)
		return 0;

	/* Let compiler decide if it's appropriate to unroll... */
	switch (c->md_len) {
	case SHA512_224_DIGEST_LENGTH:
		for (n = 0; n < SHA512_224_DIGEST_LENGTH/8; n++) {
			crypto_store_htobe64(md, c->h[n]);
			md += 8;
		}
		crypto_store_htobe32(md, c->h[n] >> 32);
		break;
	case SHA512_256_DIGEST_LENGTH:
		for (n = 0; n < SHA512_256_DIGEST_LENGTH/8; n++) {
			crypto_store_htobe64(md, c->h[n]);
			md += 8;
		}
		break;
	case SHA384_DIGEST_LENGTH:
		for (n = 0; n < SHA384_DIGEST_LENGTH/8; n++) {
			crypto_store_htobe64(md, c->h[n]);
			md += 8;
		}
		break;
	case SHA512_DIGEST_LENGTH:
		for (n = 0; n < SHA512_DIGEST_LENGTH/8; n++) {
			crypto_store_htobe64(md, c->h[n]);
			md += 8;
		}
		break;
	default:
		return 0;
	}

	return 1;
}

unsigned char *
SHA512(const unsigned char *d, size_t n, unsigned char *md)
{
	SHA512_CTX c;
	static unsigned char m[SHA512_DIGEST_LENGTH];

	if (md == NULL)
		md = m;

	SHA512_Init(&c);
	SHA512_Update(&c, d, n);
	SHA512_Final(md, &c);

	explicit_bzero(&c, sizeof(c));

	return (md);
}

int
SHA512_224_Init(SHA512_CTX *c)
{
	memset(c, 0, sizeof(*c));

	/* FIPS 180-4 section 5.3.6.1. */
	c->h[0] = U64(0x8c3d37c819544da2);
	c->h[1] = U64(0x73e1996689dcd4d6);
	c->h[2] = U64(0x1dfab7ae32ff9c82);
	c->h[3] = U64(0x679dd514582f9fcf);
	c->h[4] = U64(0x0f6d2b697bd44da8);
	c->h[5] = U64(0x77e36f7304c48942);
	c->h[6] = U64(0x3f9d85a86a1d36c8);
	c->h[7] = U64(0x1112e6ad91d692a1);

	c->md_len = SHA512_224_DIGEST_LENGTH;

	return 1;
}

int
SHA512_224_Update(SHA512_CTX *c, const void *data, size_t len)
{
	return SHA512_Update(c, data, len);
}

int
SHA512_224_Final(unsigned char *md, SHA512_CTX *c)
{
	return SHA512_Final(md, c);
}

int
SHA512_256_Init(SHA512_CTX *c)
{
	memset(c, 0, sizeof(*c));

	/* FIPS 180-4 section 5.3.6.2. */
	c->h[0] = U64(0x22312194fc2bf72c);
	c->h[1] = U64(0x9f555fa3c84c64c2);
	c->h[2] = U64(0x2393b86b6f53b151);
	c->h[3] = U64(0x963877195940eabd);
	c->h[4] = U64(0x96283ee2a88effe3);
	c->h[5] = U64(0xbe5e1e2553863992);
	c->h[6] = U64(0x2b0199fc2c85b8aa);
	c->h[7] = U64(0x0eb72ddc81c52ca2);

	c->md_len = SHA512_256_DIGEST_LENGTH;

	return 1;
}

int
SHA512_256_Update(SHA512_CTX *c, const void *data, size_t len)
{
	return SHA512_Update(c, data, len);
}

int
SHA512_256_Final(unsigned char *md, SHA512_CTX *c)
{
	return SHA512_Final(md, c);
}

#endif /* !OPENSSL_NO_SHA512 */