1a8e1175bSopenharmony_ci/*
2a8e1175bSopenharmony_ci *  FIPS-180-2 compliant SHA-384/512 implementation
3a8e1175bSopenharmony_ci *
4a8e1175bSopenharmony_ci *  Copyright The Mbed TLS Contributors
5a8e1175bSopenharmony_ci *  SPDX-License-Identifier: Apache-2.0 OR GPL-2.0-or-later
6a8e1175bSopenharmony_ci */
7a8e1175bSopenharmony_ci/*
8a8e1175bSopenharmony_ci *  The SHA-512 Secure Hash Standard was published by NIST in 2002.
9a8e1175bSopenharmony_ci *
10a8e1175bSopenharmony_ci *  http://csrc.nist.gov/publications/fips/fips180-2/fips180-2.pdf
11a8e1175bSopenharmony_ci */
12a8e1175bSopenharmony_ci
13a8e1175bSopenharmony_ci#if defined(__aarch64__) && !defined(__ARM_FEATURE_SHA512) && \
14a8e1175bSopenharmony_ci    defined(__clang__) && __clang_major__ >= 7
15a8e1175bSopenharmony_ci/* TODO: Re-consider above after https://reviews.llvm.org/D131064 merged.
16a8e1175bSopenharmony_ci *
17a8e1175bSopenharmony_ci * The intrinsic declaration are guarded by predefined ACLE macros in clang:
18a8e1175bSopenharmony_ci * these are normally only enabled by the -march option on the command line.
19a8e1175bSopenharmony_ci * By defining the macros ourselves we gain access to those declarations without
20a8e1175bSopenharmony_ci * requiring -march on the command line.
21a8e1175bSopenharmony_ci *
22a8e1175bSopenharmony_ci * `arm_neon.h` is included by common.h, so we put these defines
23a8e1175bSopenharmony_ci * at the top of this file, before any includes.
24a8e1175bSopenharmony_ci */
25a8e1175bSopenharmony_ci#define __ARM_FEATURE_SHA512 1
26a8e1175bSopenharmony_ci#define MBEDTLS_ENABLE_ARM_SHA3_EXTENSIONS_COMPILER_FLAG
27a8e1175bSopenharmony_ci#endif
28a8e1175bSopenharmony_ci
29a8e1175bSopenharmony_ci#include "common.h"
30a8e1175bSopenharmony_ci
31a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_C) || defined(MBEDTLS_SHA384_C)
32a8e1175bSopenharmony_ci
33a8e1175bSopenharmony_ci#include "mbedtls/sha512.h"
34a8e1175bSopenharmony_ci#include "mbedtls/platform_util.h"
35a8e1175bSopenharmony_ci#include "mbedtls/error.h"
36a8e1175bSopenharmony_ci
37a8e1175bSopenharmony_ci#if defined(_MSC_VER) || defined(__WATCOMC__)
38a8e1175bSopenharmony_ci  #define UL64(x) x##ui64
39a8e1175bSopenharmony_ci#else
40a8e1175bSopenharmony_ci  #define UL64(x) x##ULL
41a8e1175bSopenharmony_ci#endif
42a8e1175bSopenharmony_ci
43a8e1175bSopenharmony_ci#include <string.h>
44a8e1175bSopenharmony_ci
45a8e1175bSopenharmony_ci#include "mbedtls/platform.h"
46a8e1175bSopenharmony_ci
47a8e1175bSopenharmony_ci#if defined(__aarch64__)
48a8e1175bSopenharmony_ci#  if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT) || \
49a8e1175bSopenharmony_ci    defined(MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY)
50a8e1175bSopenharmony_ci/* *INDENT-OFF* */
51a8e1175bSopenharmony_ci#   if !defined(MBEDTLS_HAVE_NEON_INTRINSICS)
52a8e1175bSopenharmony_ci#       error "Target does not support NEON instructions"
53a8e1175bSopenharmony_ci#   endif
54a8e1175bSopenharmony_ci/*
55a8e1175bSopenharmony_ci * Best performance comes from most recent compilers, with intrinsics and -O3.
56a8e1175bSopenharmony_ci * Must compile with -march=armv8.2-a+sha3, but we can't detect armv8.2-a, and
57a8e1175bSopenharmony_ci * can't always detect __ARM_FEATURE_SHA512 (notably clang 7-12).
58a8e1175bSopenharmony_ci *
59a8e1175bSopenharmony_ci * GCC < 8 won't work at all (lacks the sha512 instructions)
60a8e1175bSopenharmony_ci * GCC >= 8 uses intrinsics, sets __ARM_FEATURE_SHA512
61a8e1175bSopenharmony_ci *
62a8e1175bSopenharmony_ci * Clang < 7 won't work at all (lacks the sha512 instructions)
63a8e1175bSopenharmony_ci * Clang 7-12 don't have intrinsics (but we work around that with inline
64a8e1175bSopenharmony_ci *            assembler) or __ARM_FEATURE_SHA512
65a8e1175bSopenharmony_ci * Clang == 13.0.0 same as clang 12 (only seen on macOS)
66a8e1175bSopenharmony_ci * Clang >= 13.0.1 has __ARM_FEATURE_SHA512 and intrinsics
67a8e1175bSopenharmony_ci */
68a8e1175bSopenharmony_ci#    if !defined(__ARM_FEATURE_SHA512) || defined(MBEDTLS_ENABLE_ARM_SHA3_EXTENSIONS_COMPILER_FLAG)
69a8e1175bSopenharmony_ci       /* Test Clang first, as it defines __GNUC__ */
70a8e1175bSopenharmony_ci#      if defined(__ARMCOMPILER_VERSION)
71a8e1175bSopenharmony_ci#        if __ARMCOMPILER_VERSION < 6090000
72a8e1175bSopenharmony_ci#          error "A more recent armclang is required for MBEDTLS_SHA512_USE_A64_CRYPTO_*"
73a8e1175bSopenharmony_ci#        elif __ARMCOMPILER_VERSION == 6090000
74a8e1175bSopenharmony_ci#          error "Must use minimum -march=armv8.2-a+sha3 for MBEDTLS_SHA512_USE_A64_CRYPTO_*"
75a8e1175bSopenharmony_ci#        else
76a8e1175bSopenharmony_ci#          pragma clang attribute push (__attribute__((target("sha3"))), apply_to=function)
77a8e1175bSopenharmony_ci#          define MBEDTLS_POP_TARGET_PRAGMA
78a8e1175bSopenharmony_ci#        endif
79a8e1175bSopenharmony_ci#      elif defined(__clang__)
80a8e1175bSopenharmony_ci#        if __clang_major__ < 7
81a8e1175bSopenharmony_ci#          error "A more recent Clang is required for MBEDTLS_SHA512_USE_A64_CRYPTO_*"
82a8e1175bSopenharmony_ci#        else
83a8e1175bSopenharmony_ci#          pragma clang attribute push (__attribute__((target("sha3"))), apply_to=function)
84a8e1175bSopenharmony_ci#          define MBEDTLS_POP_TARGET_PRAGMA
85a8e1175bSopenharmony_ci#        endif
86a8e1175bSopenharmony_ci#      elif defined(__GNUC__)
87a8e1175bSopenharmony_ci#        if __GNUC__ < 8
88a8e1175bSopenharmony_ci#          error "A more recent GCC is required for MBEDTLS_SHA512_USE_A64_CRYPTO_*"
89a8e1175bSopenharmony_ci#        else
90a8e1175bSopenharmony_ci#          pragma GCC push_options
91a8e1175bSopenharmony_ci#          pragma GCC target ("arch=armv8.2-a+sha3")
92a8e1175bSopenharmony_ci#          define MBEDTLS_POP_TARGET_PRAGMA
93a8e1175bSopenharmony_ci#        endif
94a8e1175bSopenharmony_ci#      else
95a8e1175bSopenharmony_ci#        error "Only GCC and Clang supported for MBEDTLS_SHA512_USE_A64_CRYPTO_*"
96a8e1175bSopenharmony_ci#      endif
97a8e1175bSopenharmony_ci#    endif
98a8e1175bSopenharmony_ci/* *INDENT-ON* */
99a8e1175bSopenharmony_ci#  endif
100a8e1175bSopenharmony_ci#  if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
101a8e1175bSopenharmony_ci#    if defined(__unix__)
102a8e1175bSopenharmony_ci#      if defined(__linux__)
103a8e1175bSopenharmony_ci/* Our preferred method of detection is getauxval() */
104a8e1175bSopenharmony_ci#        include <sys/auxv.h>
105a8e1175bSopenharmony_ci#        if !defined(HWCAP_SHA512)
106a8e1175bSopenharmony_ci/* The same header that declares getauxval() should provide the HWCAP_xxx
107a8e1175bSopenharmony_ci * constants to analyze its return value. However, the libc may be too
108a8e1175bSopenharmony_ci * old to have the constant that we need. So if it's missing, assume that
109a8e1175bSopenharmony_ci * the value is the same one used by the Linux kernel ABI.
110a8e1175bSopenharmony_ci */
111a8e1175bSopenharmony_ci#          define HWCAP_SHA512 (1 << 21)
112a8e1175bSopenharmony_ci#        endif
113a8e1175bSopenharmony_ci#      endif
114a8e1175bSopenharmony_ci/* Use SIGILL on Unix, and fall back to it on Linux */
115a8e1175bSopenharmony_ci#      include <signal.h>
116a8e1175bSopenharmony_ci#    endif
117a8e1175bSopenharmony_ci#  endif
118a8e1175bSopenharmony_ci#elif !defined(MBEDTLS_PLATFORM_IS_WINDOWS_ON_ARM64)
119a8e1175bSopenharmony_ci#  undef MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY
120a8e1175bSopenharmony_ci#  undef MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT
121a8e1175bSopenharmony_ci#endif
122a8e1175bSopenharmony_ci
123a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
124a8e1175bSopenharmony_ci/*
125a8e1175bSopenharmony_ci * Capability detection code comes early, so we can disable
126a8e1175bSopenharmony_ci * MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT if no detection mechanism found
127a8e1175bSopenharmony_ci */
128a8e1175bSopenharmony_ci#if defined(HWCAP_SHA512)
129a8e1175bSopenharmony_cistatic int mbedtls_a64_crypto_sha512_determine_support(void)
130a8e1175bSopenharmony_ci{
131a8e1175bSopenharmony_ci    return (getauxval(AT_HWCAP) & HWCAP_SHA512) ? 1 : 0;
132a8e1175bSopenharmony_ci}
133a8e1175bSopenharmony_ci#elif defined(__APPLE__)
134a8e1175bSopenharmony_ci#include <sys/types.h>
135a8e1175bSopenharmony_ci#include <sys/sysctl.h>
136a8e1175bSopenharmony_ci
137a8e1175bSopenharmony_cistatic int mbedtls_a64_crypto_sha512_determine_support(void)
138a8e1175bSopenharmony_ci{
139a8e1175bSopenharmony_ci    int value = 0;
140a8e1175bSopenharmony_ci    size_t value_len = sizeof(value);
141a8e1175bSopenharmony_ci
142a8e1175bSopenharmony_ci    int ret = sysctlbyname("hw.optional.armv8_2_sha512", &value, &value_len,
143a8e1175bSopenharmony_ci                           NULL, 0);
144a8e1175bSopenharmony_ci    return ret == 0 && value != 0;
145a8e1175bSopenharmony_ci}
146a8e1175bSopenharmony_ci#elif defined(MBEDTLS_PLATFORM_IS_WINDOWS_ON_ARM64)
147a8e1175bSopenharmony_ci/*
148a8e1175bSopenharmony_ci * As of March 2022, there don't appear to be any PF_ARM_V8_* flags
149a8e1175bSopenharmony_ci * available to pass to IsProcessorFeaturePresent() to check for
150a8e1175bSopenharmony_ci * SHA-512 support. So we fall back to the C code only.
151a8e1175bSopenharmony_ci */
152a8e1175bSopenharmony_ci#if defined(_MSC_VER)
153a8e1175bSopenharmony_ci#pragma message "No mechanism to detect A64_CRYPTO found, using C code only"
154a8e1175bSopenharmony_ci#else
155a8e1175bSopenharmony_ci#warning "No mechanism to detect A64_CRYPTO found, using C code only"
156a8e1175bSopenharmony_ci#endif
157a8e1175bSopenharmony_ci#elif defined(__unix__) && defined(SIG_SETMASK)
158a8e1175bSopenharmony_ci/* Detection with SIGILL, setjmp() and longjmp() */
159a8e1175bSopenharmony_ci#include <signal.h>
160a8e1175bSopenharmony_ci#include <setjmp.h>
161a8e1175bSopenharmony_ci
162a8e1175bSopenharmony_cistatic jmp_buf return_from_sigill;
163a8e1175bSopenharmony_ci
164a8e1175bSopenharmony_ci/*
165a8e1175bSopenharmony_ci * A64 SHA512 support detection via SIGILL
166a8e1175bSopenharmony_ci */
167a8e1175bSopenharmony_cistatic void sigill_handler(int signal)
168a8e1175bSopenharmony_ci{
169a8e1175bSopenharmony_ci    (void) signal;
170a8e1175bSopenharmony_ci    longjmp(return_from_sigill, 1);
171a8e1175bSopenharmony_ci}
172a8e1175bSopenharmony_ci
173a8e1175bSopenharmony_cistatic int mbedtls_a64_crypto_sha512_determine_support(void)
174a8e1175bSopenharmony_ci{
175a8e1175bSopenharmony_ci    struct sigaction old_action, new_action;
176a8e1175bSopenharmony_ci
177a8e1175bSopenharmony_ci    sigset_t old_mask;
178a8e1175bSopenharmony_ci    if (sigprocmask(0, NULL, &old_mask)) {
179a8e1175bSopenharmony_ci        return 0;
180a8e1175bSopenharmony_ci    }
181a8e1175bSopenharmony_ci
182a8e1175bSopenharmony_ci    sigemptyset(&new_action.sa_mask);
183a8e1175bSopenharmony_ci    new_action.sa_flags = 0;
184a8e1175bSopenharmony_ci    new_action.sa_handler = sigill_handler;
185a8e1175bSopenharmony_ci
186a8e1175bSopenharmony_ci    sigaction(SIGILL, &new_action, &old_action);
187a8e1175bSopenharmony_ci
188a8e1175bSopenharmony_ci    static int ret = 0;
189a8e1175bSopenharmony_ci
190a8e1175bSopenharmony_ci    if (setjmp(return_from_sigill) == 0) {         /* First return only */
191a8e1175bSopenharmony_ci        /* If this traps, we will return a second time from setjmp() with 1 */
192a8e1175bSopenharmony_ci        asm ("sha512h q0, q0, v0.2d" : : : "v0");
193a8e1175bSopenharmony_ci        ret = 1;
194a8e1175bSopenharmony_ci    }
195a8e1175bSopenharmony_ci
196a8e1175bSopenharmony_ci    sigaction(SIGILL, &old_action, NULL);
197a8e1175bSopenharmony_ci    sigprocmask(SIG_SETMASK, &old_mask, NULL);
198a8e1175bSopenharmony_ci
199a8e1175bSopenharmony_ci    return ret;
200a8e1175bSopenharmony_ci}
201a8e1175bSopenharmony_ci#else
202a8e1175bSopenharmony_ci#warning "No mechanism to detect A64_CRYPTO found, using C code only"
203a8e1175bSopenharmony_ci#undef MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT
204a8e1175bSopenharmony_ci#endif  /* HWCAP_SHA512, __APPLE__, __unix__ && SIG_SETMASK */
205a8e1175bSopenharmony_ci
206a8e1175bSopenharmony_ci#endif  /* MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT */
207a8e1175bSopenharmony_ci
208a8e1175bSopenharmony_ci#if !defined(MBEDTLS_SHA512_ALT)
209a8e1175bSopenharmony_ci
210a8e1175bSopenharmony_ci#define SHA512_BLOCK_SIZE 128
211a8e1175bSopenharmony_ci
212a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_SMALLER)
213a8e1175bSopenharmony_cistatic void sha512_put_uint64_be(uint64_t n, unsigned char *b, uint8_t i)
214a8e1175bSopenharmony_ci{
215a8e1175bSopenharmony_ci    MBEDTLS_PUT_UINT64_BE(n, b, i);
216a8e1175bSopenharmony_ci}
217a8e1175bSopenharmony_ci#else
218a8e1175bSopenharmony_ci#define sha512_put_uint64_be    MBEDTLS_PUT_UINT64_BE
219a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_SMALLER */
220a8e1175bSopenharmony_ci
221a8e1175bSopenharmony_civoid mbedtls_sha512_init(mbedtls_sha512_context *ctx)
222a8e1175bSopenharmony_ci{
223a8e1175bSopenharmony_ci    memset(ctx, 0, sizeof(mbedtls_sha512_context));
224a8e1175bSopenharmony_ci}
225a8e1175bSopenharmony_ci
226a8e1175bSopenharmony_civoid mbedtls_sha512_free(mbedtls_sha512_context *ctx)
227a8e1175bSopenharmony_ci{
228a8e1175bSopenharmony_ci    if (ctx == NULL) {
229a8e1175bSopenharmony_ci        return;
230a8e1175bSopenharmony_ci    }
231a8e1175bSopenharmony_ci
232a8e1175bSopenharmony_ci    mbedtls_platform_zeroize(ctx, sizeof(mbedtls_sha512_context));
233a8e1175bSopenharmony_ci}
234a8e1175bSopenharmony_ci
235a8e1175bSopenharmony_civoid mbedtls_sha512_clone(mbedtls_sha512_context *dst,
236a8e1175bSopenharmony_ci                          const mbedtls_sha512_context *src)
237a8e1175bSopenharmony_ci{
238a8e1175bSopenharmony_ci    *dst = *src;
239a8e1175bSopenharmony_ci}
240a8e1175bSopenharmony_ci
241a8e1175bSopenharmony_ci/*
242a8e1175bSopenharmony_ci * SHA-512 context setup
243a8e1175bSopenharmony_ci */
244a8e1175bSopenharmony_ciint mbedtls_sha512_starts(mbedtls_sha512_context *ctx, int is384)
245a8e1175bSopenharmony_ci{
246a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C) && defined(MBEDTLS_SHA512_C)
247a8e1175bSopenharmony_ci    if (is384 != 0 && is384 != 1) {
248a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
249a8e1175bSopenharmony_ci    }
250a8e1175bSopenharmony_ci#elif defined(MBEDTLS_SHA512_C)
251a8e1175bSopenharmony_ci    if (is384 != 0) {
252a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
253a8e1175bSopenharmony_ci    }
254a8e1175bSopenharmony_ci#else /* defined MBEDTLS_SHA384_C only */
255a8e1175bSopenharmony_ci    if (is384 == 0) {
256a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
257a8e1175bSopenharmony_ci    }
258a8e1175bSopenharmony_ci#endif
259a8e1175bSopenharmony_ci
260a8e1175bSopenharmony_ci    ctx->total[0] = 0;
261a8e1175bSopenharmony_ci    ctx->total[1] = 0;
262a8e1175bSopenharmony_ci
263a8e1175bSopenharmony_ci    if (is384 == 0) {
264a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_C)
265a8e1175bSopenharmony_ci        ctx->state[0] = UL64(0x6A09E667F3BCC908);
266a8e1175bSopenharmony_ci        ctx->state[1] = UL64(0xBB67AE8584CAA73B);
267a8e1175bSopenharmony_ci        ctx->state[2] = UL64(0x3C6EF372FE94F82B);
268a8e1175bSopenharmony_ci        ctx->state[3] = UL64(0xA54FF53A5F1D36F1);
269a8e1175bSopenharmony_ci        ctx->state[4] = UL64(0x510E527FADE682D1);
270a8e1175bSopenharmony_ci        ctx->state[5] = UL64(0x9B05688C2B3E6C1F);
271a8e1175bSopenharmony_ci        ctx->state[6] = UL64(0x1F83D9ABFB41BD6B);
272a8e1175bSopenharmony_ci        ctx->state[7] = UL64(0x5BE0CD19137E2179);
273a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_C */
274a8e1175bSopenharmony_ci    } else {
275a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C)
276a8e1175bSopenharmony_ci        ctx->state[0] = UL64(0xCBBB9D5DC1059ED8);
277a8e1175bSopenharmony_ci        ctx->state[1] = UL64(0x629A292A367CD507);
278a8e1175bSopenharmony_ci        ctx->state[2] = UL64(0x9159015A3070DD17);
279a8e1175bSopenharmony_ci        ctx->state[3] = UL64(0x152FECD8F70E5939);
280a8e1175bSopenharmony_ci        ctx->state[4] = UL64(0x67332667FFC00B31);
281a8e1175bSopenharmony_ci        ctx->state[5] = UL64(0x8EB44A8768581511);
282a8e1175bSopenharmony_ci        ctx->state[6] = UL64(0xDB0C2E0D64F98FA7);
283a8e1175bSopenharmony_ci        ctx->state[7] = UL64(0x47B5481DBEFA4FA4);
284a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA384_C */
285a8e1175bSopenharmony_ci    }
286a8e1175bSopenharmony_ci
287a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C)
288a8e1175bSopenharmony_ci    ctx->is384 = is384;
289a8e1175bSopenharmony_ci#endif
290a8e1175bSopenharmony_ci
291a8e1175bSopenharmony_ci    return 0;
292a8e1175bSopenharmony_ci}
293a8e1175bSopenharmony_ci
294a8e1175bSopenharmony_ci#if !defined(MBEDTLS_SHA512_PROCESS_ALT)
295a8e1175bSopenharmony_ci
296a8e1175bSopenharmony_ci/*
297a8e1175bSopenharmony_ci * Round constants
298a8e1175bSopenharmony_ci */
299a8e1175bSopenharmony_cistatic const uint64_t K[80] =
300a8e1175bSopenharmony_ci{
301a8e1175bSopenharmony_ci    UL64(0x428A2F98D728AE22),  UL64(0x7137449123EF65CD),
302a8e1175bSopenharmony_ci    UL64(0xB5C0FBCFEC4D3B2F),  UL64(0xE9B5DBA58189DBBC),
303a8e1175bSopenharmony_ci    UL64(0x3956C25BF348B538),  UL64(0x59F111F1B605D019),
304a8e1175bSopenharmony_ci    UL64(0x923F82A4AF194F9B),  UL64(0xAB1C5ED5DA6D8118),
305a8e1175bSopenharmony_ci    UL64(0xD807AA98A3030242),  UL64(0x12835B0145706FBE),
306a8e1175bSopenharmony_ci    UL64(0x243185BE4EE4B28C),  UL64(0x550C7DC3D5FFB4E2),
307a8e1175bSopenharmony_ci    UL64(0x72BE5D74F27B896F),  UL64(0x80DEB1FE3B1696B1),
308a8e1175bSopenharmony_ci    UL64(0x9BDC06A725C71235),  UL64(0xC19BF174CF692694),
309a8e1175bSopenharmony_ci    UL64(0xE49B69C19EF14AD2),  UL64(0xEFBE4786384F25E3),
310a8e1175bSopenharmony_ci    UL64(0x0FC19DC68B8CD5B5),  UL64(0x240CA1CC77AC9C65),
311a8e1175bSopenharmony_ci    UL64(0x2DE92C6F592B0275),  UL64(0x4A7484AA6EA6E483),
312a8e1175bSopenharmony_ci    UL64(0x5CB0A9DCBD41FBD4),  UL64(0x76F988DA831153B5),
313a8e1175bSopenharmony_ci    UL64(0x983E5152EE66DFAB),  UL64(0xA831C66D2DB43210),
314a8e1175bSopenharmony_ci    UL64(0xB00327C898FB213F),  UL64(0xBF597FC7BEEF0EE4),
315a8e1175bSopenharmony_ci    UL64(0xC6E00BF33DA88FC2),  UL64(0xD5A79147930AA725),
316a8e1175bSopenharmony_ci    UL64(0x06CA6351E003826F),  UL64(0x142929670A0E6E70),
317a8e1175bSopenharmony_ci    UL64(0x27B70A8546D22FFC),  UL64(0x2E1B21385C26C926),
318a8e1175bSopenharmony_ci    UL64(0x4D2C6DFC5AC42AED),  UL64(0x53380D139D95B3DF),
319a8e1175bSopenharmony_ci    UL64(0x650A73548BAF63DE),  UL64(0x766A0ABB3C77B2A8),
320a8e1175bSopenharmony_ci    UL64(0x81C2C92E47EDAEE6),  UL64(0x92722C851482353B),
321a8e1175bSopenharmony_ci    UL64(0xA2BFE8A14CF10364),  UL64(0xA81A664BBC423001),
322a8e1175bSopenharmony_ci    UL64(0xC24B8B70D0F89791),  UL64(0xC76C51A30654BE30),
323a8e1175bSopenharmony_ci    UL64(0xD192E819D6EF5218),  UL64(0xD69906245565A910),
324a8e1175bSopenharmony_ci    UL64(0xF40E35855771202A),  UL64(0x106AA07032BBD1B8),
325a8e1175bSopenharmony_ci    UL64(0x19A4C116B8D2D0C8),  UL64(0x1E376C085141AB53),
326a8e1175bSopenharmony_ci    UL64(0x2748774CDF8EEB99),  UL64(0x34B0BCB5E19B48A8),
327a8e1175bSopenharmony_ci    UL64(0x391C0CB3C5C95A63),  UL64(0x4ED8AA4AE3418ACB),
328a8e1175bSopenharmony_ci    UL64(0x5B9CCA4F7763E373),  UL64(0x682E6FF3D6B2B8A3),
329a8e1175bSopenharmony_ci    UL64(0x748F82EE5DEFB2FC),  UL64(0x78A5636F43172F60),
330a8e1175bSopenharmony_ci    UL64(0x84C87814A1F0AB72),  UL64(0x8CC702081A6439EC),
331a8e1175bSopenharmony_ci    UL64(0x90BEFFFA23631E28),  UL64(0xA4506CEBDE82BDE9),
332a8e1175bSopenharmony_ci    UL64(0xBEF9A3F7B2C67915),  UL64(0xC67178F2E372532B),
333a8e1175bSopenharmony_ci    UL64(0xCA273ECEEA26619C),  UL64(0xD186B8C721C0C207),
334a8e1175bSopenharmony_ci    UL64(0xEADA7DD6CDE0EB1E),  UL64(0xF57D4F7FEE6ED178),
335a8e1175bSopenharmony_ci    UL64(0x06F067AA72176FBA),  UL64(0x0A637DC5A2C898A6),
336a8e1175bSopenharmony_ci    UL64(0x113F9804BEF90DAE),  UL64(0x1B710B35131C471B),
337a8e1175bSopenharmony_ci    UL64(0x28DB77F523047D84),  UL64(0x32CAAB7B40C72493),
338a8e1175bSopenharmony_ci    UL64(0x3C9EBE0A15C9BEBC),  UL64(0x431D67C49C100D4C),
339a8e1175bSopenharmony_ci    UL64(0x4CC5D4BECB3E42B6),  UL64(0x597F299CFC657E2A),
340a8e1175bSopenharmony_ci    UL64(0x5FCB6FAB3AD6FAEC),  UL64(0x6C44198C4A475817)
341a8e1175bSopenharmony_ci};
342a8e1175bSopenharmony_ci#endif
343a8e1175bSopenharmony_ci
344a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT) || \
345a8e1175bSopenharmony_ci    defined(MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY)
346a8e1175bSopenharmony_ci
347a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY)
348a8e1175bSopenharmony_ci#  define mbedtls_internal_sha512_process_many_a64_crypto mbedtls_internal_sha512_process_many
349a8e1175bSopenharmony_ci#  define mbedtls_internal_sha512_process_a64_crypto      mbedtls_internal_sha512_process
350a8e1175bSopenharmony_ci#endif
351a8e1175bSopenharmony_ci
352a8e1175bSopenharmony_ci/* Accelerated SHA-512 implementation originally written by Simon Tatham for PuTTY,
353a8e1175bSopenharmony_ci * under the MIT licence; dual-licensed as Apache 2 with his kind permission.
354a8e1175bSopenharmony_ci */
355a8e1175bSopenharmony_ci
356a8e1175bSopenharmony_ci#if defined(__clang__) && \
357a8e1175bSopenharmony_ci    (__clang_major__ < 13 || \
358a8e1175bSopenharmony_ci     (__clang_major__ == 13 && __clang_minor__ == 0 && __clang_patchlevel__ == 0))
359a8e1175bSopenharmony_cistatic inline uint64x2_t vsha512su0q_u64(uint64x2_t x, uint64x2_t y)
360a8e1175bSopenharmony_ci{
361a8e1175bSopenharmony_ci    asm ("sha512su0 %0.2D,%1.2D" : "+w" (x) : "w" (y));
362a8e1175bSopenharmony_ci    return x;
363a8e1175bSopenharmony_ci}
364a8e1175bSopenharmony_cistatic inline uint64x2_t vsha512su1q_u64(uint64x2_t x, uint64x2_t y, uint64x2_t z)
365a8e1175bSopenharmony_ci{
366a8e1175bSopenharmony_ci    asm ("sha512su1 %0.2D,%1.2D,%2.2D" : "+w" (x) : "w" (y), "w" (z));
367a8e1175bSopenharmony_ci    return x;
368a8e1175bSopenharmony_ci}
369a8e1175bSopenharmony_cistatic inline uint64x2_t vsha512hq_u64(uint64x2_t x, uint64x2_t y, uint64x2_t z)
370a8e1175bSopenharmony_ci{
371a8e1175bSopenharmony_ci    asm ("sha512h %0,%1,%2.2D" : "+w" (x) : "w" (y), "w" (z));
372a8e1175bSopenharmony_ci    return x;
373a8e1175bSopenharmony_ci}
374a8e1175bSopenharmony_cistatic inline uint64x2_t vsha512h2q_u64(uint64x2_t x, uint64x2_t y, uint64x2_t z)
375a8e1175bSopenharmony_ci{
376a8e1175bSopenharmony_ci    asm ("sha512h2 %0,%1,%2.2D" : "+w" (x) : "w" (y), "w" (z));
377a8e1175bSopenharmony_ci    return x;
378a8e1175bSopenharmony_ci}
379a8e1175bSopenharmony_ci#endif  /* __clang__ etc */
380a8e1175bSopenharmony_ci
381a8e1175bSopenharmony_cistatic size_t mbedtls_internal_sha512_process_many_a64_crypto(
382a8e1175bSopenharmony_ci    mbedtls_sha512_context *ctx, const uint8_t *msg, size_t len)
383a8e1175bSopenharmony_ci{
384a8e1175bSopenharmony_ci    uint64x2_t ab = vld1q_u64(&ctx->state[0]);
385a8e1175bSopenharmony_ci    uint64x2_t cd = vld1q_u64(&ctx->state[2]);
386a8e1175bSopenharmony_ci    uint64x2_t ef = vld1q_u64(&ctx->state[4]);
387a8e1175bSopenharmony_ci    uint64x2_t gh = vld1q_u64(&ctx->state[6]);
388a8e1175bSopenharmony_ci
389a8e1175bSopenharmony_ci    size_t processed = 0;
390a8e1175bSopenharmony_ci
391a8e1175bSopenharmony_ci    for (;
392a8e1175bSopenharmony_ci         len >= SHA512_BLOCK_SIZE;
393a8e1175bSopenharmony_ci         processed += SHA512_BLOCK_SIZE,
394a8e1175bSopenharmony_ci         msg += SHA512_BLOCK_SIZE,
395a8e1175bSopenharmony_ci         len -= SHA512_BLOCK_SIZE) {
396a8e1175bSopenharmony_ci        uint64x2_t initial_sum, sum, intermed;
397a8e1175bSopenharmony_ci
398a8e1175bSopenharmony_ci        uint64x2_t ab_orig = ab;
399a8e1175bSopenharmony_ci        uint64x2_t cd_orig = cd;
400a8e1175bSopenharmony_ci        uint64x2_t ef_orig = ef;
401a8e1175bSopenharmony_ci        uint64x2_t gh_orig = gh;
402a8e1175bSopenharmony_ci
403a8e1175bSopenharmony_ci        uint64x2_t s0 = (uint64x2_t) vld1q_u8(msg + 16 * 0);
404a8e1175bSopenharmony_ci        uint64x2_t s1 = (uint64x2_t) vld1q_u8(msg + 16 * 1);
405a8e1175bSopenharmony_ci        uint64x2_t s2 = (uint64x2_t) vld1q_u8(msg + 16 * 2);
406a8e1175bSopenharmony_ci        uint64x2_t s3 = (uint64x2_t) vld1q_u8(msg + 16 * 3);
407a8e1175bSopenharmony_ci        uint64x2_t s4 = (uint64x2_t) vld1q_u8(msg + 16 * 4);
408a8e1175bSopenharmony_ci        uint64x2_t s5 = (uint64x2_t) vld1q_u8(msg + 16 * 5);
409a8e1175bSopenharmony_ci        uint64x2_t s6 = (uint64x2_t) vld1q_u8(msg + 16 * 6);
410a8e1175bSopenharmony_ci        uint64x2_t s7 = (uint64x2_t) vld1q_u8(msg + 16 * 7);
411a8e1175bSopenharmony_ci
412a8e1175bSopenharmony_ci#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__  /* assume LE if these not defined; untested on BE */
413a8e1175bSopenharmony_ci        s0 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s0)));
414a8e1175bSopenharmony_ci        s1 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s1)));
415a8e1175bSopenharmony_ci        s2 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s2)));
416a8e1175bSopenharmony_ci        s3 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s3)));
417a8e1175bSopenharmony_ci        s4 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s4)));
418a8e1175bSopenharmony_ci        s5 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s5)));
419a8e1175bSopenharmony_ci        s6 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s6)));
420a8e1175bSopenharmony_ci        s7 = vreinterpretq_u64_u8(vrev64q_u8(vreinterpretq_u8_u64(s7)));
421a8e1175bSopenharmony_ci#endif
422a8e1175bSopenharmony_ci
423a8e1175bSopenharmony_ci        /* Rounds 0 and 1 */
424a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s0, vld1q_u64(&K[0]));
425a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), gh);
426a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(ef, gh, 1), vextq_u64(cd, ef, 1));
427a8e1175bSopenharmony_ci        gh = vsha512h2q_u64(intermed, cd, ab);
428a8e1175bSopenharmony_ci        cd = vaddq_u64(cd, intermed);
429a8e1175bSopenharmony_ci
430a8e1175bSopenharmony_ci        /* Rounds 2 and 3 */
431a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s1, vld1q_u64(&K[2]));
432a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ef);
433a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(cd, ef, 1), vextq_u64(ab, cd, 1));
434a8e1175bSopenharmony_ci        ef = vsha512h2q_u64(intermed, ab, gh);
435a8e1175bSopenharmony_ci        ab = vaddq_u64(ab, intermed);
436a8e1175bSopenharmony_ci
437a8e1175bSopenharmony_ci        /* Rounds 4 and 5 */
438a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s2, vld1q_u64(&K[4]));
439a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), cd);
440a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(ab, cd, 1), vextq_u64(gh, ab, 1));
441a8e1175bSopenharmony_ci        cd = vsha512h2q_u64(intermed, gh, ef);
442a8e1175bSopenharmony_ci        gh = vaddq_u64(gh, intermed);
443a8e1175bSopenharmony_ci
444a8e1175bSopenharmony_ci        /* Rounds 6 and 7 */
445a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s3, vld1q_u64(&K[6]));
446a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ab);
447a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(gh, ab, 1), vextq_u64(ef, gh, 1));
448a8e1175bSopenharmony_ci        ab = vsha512h2q_u64(intermed, ef, cd);
449a8e1175bSopenharmony_ci        ef = vaddq_u64(ef, intermed);
450a8e1175bSopenharmony_ci
451a8e1175bSopenharmony_ci        /* Rounds 8 and 9 */
452a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s4, vld1q_u64(&K[8]));
453a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), gh);
454a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(ef, gh, 1), vextq_u64(cd, ef, 1));
455a8e1175bSopenharmony_ci        gh = vsha512h2q_u64(intermed, cd, ab);
456a8e1175bSopenharmony_ci        cd = vaddq_u64(cd, intermed);
457a8e1175bSopenharmony_ci
458a8e1175bSopenharmony_ci        /* Rounds 10 and 11 */
459a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s5, vld1q_u64(&K[10]));
460a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ef);
461a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(cd, ef, 1), vextq_u64(ab, cd, 1));
462a8e1175bSopenharmony_ci        ef = vsha512h2q_u64(intermed, ab, gh);
463a8e1175bSopenharmony_ci        ab = vaddq_u64(ab, intermed);
464a8e1175bSopenharmony_ci
465a8e1175bSopenharmony_ci        /* Rounds 12 and 13 */
466a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s6, vld1q_u64(&K[12]));
467a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), cd);
468a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(ab, cd, 1), vextq_u64(gh, ab, 1));
469a8e1175bSopenharmony_ci        cd = vsha512h2q_u64(intermed, gh, ef);
470a8e1175bSopenharmony_ci        gh = vaddq_u64(gh, intermed);
471a8e1175bSopenharmony_ci
472a8e1175bSopenharmony_ci        /* Rounds 14 and 15 */
473a8e1175bSopenharmony_ci        initial_sum = vaddq_u64(s7, vld1q_u64(&K[14]));
474a8e1175bSopenharmony_ci        sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ab);
475a8e1175bSopenharmony_ci        intermed = vsha512hq_u64(sum, vextq_u64(gh, ab, 1), vextq_u64(ef, gh, 1));
476a8e1175bSopenharmony_ci        ab = vsha512h2q_u64(intermed, ef, cd);
477a8e1175bSopenharmony_ci        ef = vaddq_u64(ef, intermed);
478a8e1175bSopenharmony_ci
479a8e1175bSopenharmony_ci        for (unsigned int t = 16; t < 80; t += 16) {
480a8e1175bSopenharmony_ci            /* Rounds t and t + 1 */
481a8e1175bSopenharmony_ci            s0 = vsha512su1q_u64(vsha512su0q_u64(s0, s1), s7, vextq_u64(s4, s5, 1));
482a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s0, vld1q_u64(&K[t]));
483a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), gh);
484a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(ef, gh, 1), vextq_u64(cd, ef, 1));
485a8e1175bSopenharmony_ci            gh = vsha512h2q_u64(intermed, cd, ab);
486a8e1175bSopenharmony_ci            cd = vaddq_u64(cd, intermed);
487a8e1175bSopenharmony_ci
488a8e1175bSopenharmony_ci            /* Rounds t + 2 and t + 3 */
489a8e1175bSopenharmony_ci            s1 = vsha512su1q_u64(vsha512su0q_u64(s1, s2), s0, vextq_u64(s5, s6, 1));
490a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s1, vld1q_u64(&K[t + 2]));
491a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ef);
492a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(cd, ef, 1), vextq_u64(ab, cd, 1));
493a8e1175bSopenharmony_ci            ef = vsha512h2q_u64(intermed, ab, gh);
494a8e1175bSopenharmony_ci            ab = vaddq_u64(ab, intermed);
495a8e1175bSopenharmony_ci
496a8e1175bSopenharmony_ci            /* Rounds t + 4 and t + 5 */
497a8e1175bSopenharmony_ci            s2 = vsha512su1q_u64(vsha512su0q_u64(s2, s3), s1, vextq_u64(s6, s7, 1));
498a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s2, vld1q_u64(&K[t + 4]));
499a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), cd);
500a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(ab, cd, 1), vextq_u64(gh, ab, 1));
501a8e1175bSopenharmony_ci            cd = vsha512h2q_u64(intermed, gh, ef);
502a8e1175bSopenharmony_ci            gh = vaddq_u64(gh, intermed);
503a8e1175bSopenharmony_ci
504a8e1175bSopenharmony_ci            /* Rounds t + 6 and t + 7 */
505a8e1175bSopenharmony_ci            s3 = vsha512su1q_u64(vsha512su0q_u64(s3, s4), s2, vextq_u64(s7, s0, 1));
506a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s3, vld1q_u64(&K[t + 6]));
507a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ab);
508a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(gh, ab, 1), vextq_u64(ef, gh, 1));
509a8e1175bSopenharmony_ci            ab = vsha512h2q_u64(intermed, ef, cd);
510a8e1175bSopenharmony_ci            ef = vaddq_u64(ef, intermed);
511a8e1175bSopenharmony_ci
512a8e1175bSopenharmony_ci            /* Rounds t + 8 and t + 9 */
513a8e1175bSopenharmony_ci            s4 = vsha512su1q_u64(vsha512su0q_u64(s4, s5), s3, vextq_u64(s0, s1, 1));
514a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s4, vld1q_u64(&K[t + 8]));
515a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), gh);
516a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(ef, gh, 1), vextq_u64(cd, ef, 1));
517a8e1175bSopenharmony_ci            gh = vsha512h2q_u64(intermed, cd, ab);
518a8e1175bSopenharmony_ci            cd = vaddq_u64(cd, intermed);
519a8e1175bSopenharmony_ci
520a8e1175bSopenharmony_ci            /* Rounds t + 10 and t + 11 */
521a8e1175bSopenharmony_ci            s5 = vsha512su1q_u64(vsha512su0q_u64(s5, s6), s4, vextq_u64(s1, s2, 1));
522a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s5, vld1q_u64(&K[t + 10]));
523a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ef);
524a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(cd, ef, 1), vextq_u64(ab, cd, 1));
525a8e1175bSopenharmony_ci            ef = vsha512h2q_u64(intermed, ab, gh);
526a8e1175bSopenharmony_ci            ab = vaddq_u64(ab, intermed);
527a8e1175bSopenharmony_ci
528a8e1175bSopenharmony_ci            /* Rounds t + 12 and t + 13 */
529a8e1175bSopenharmony_ci            s6 = vsha512su1q_u64(vsha512su0q_u64(s6, s7), s5, vextq_u64(s2, s3, 1));
530a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s6, vld1q_u64(&K[t + 12]));
531a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), cd);
532a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(ab, cd, 1), vextq_u64(gh, ab, 1));
533a8e1175bSopenharmony_ci            cd = vsha512h2q_u64(intermed, gh, ef);
534a8e1175bSopenharmony_ci            gh = vaddq_u64(gh, intermed);
535a8e1175bSopenharmony_ci
536a8e1175bSopenharmony_ci            /* Rounds t + 14 and t + 15 */
537a8e1175bSopenharmony_ci            s7 = vsha512su1q_u64(vsha512su0q_u64(s7, s0), s6, vextq_u64(s3, s4, 1));
538a8e1175bSopenharmony_ci            initial_sum = vaddq_u64(s7, vld1q_u64(&K[t + 14]));
539a8e1175bSopenharmony_ci            sum = vaddq_u64(vextq_u64(initial_sum, initial_sum, 1), ab);
540a8e1175bSopenharmony_ci            intermed = vsha512hq_u64(sum, vextq_u64(gh, ab, 1), vextq_u64(ef, gh, 1));
541a8e1175bSopenharmony_ci            ab = vsha512h2q_u64(intermed, ef, cd);
542a8e1175bSopenharmony_ci            ef = vaddq_u64(ef, intermed);
543a8e1175bSopenharmony_ci        }
544a8e1175bSopenharmony_ci
545a8e1175bSopenharmony_ci        ab = vaddq_u64(ab, ab_orig);
546a8e1175bSopenharmony_ci        cd = vaddq_u64(cd, cd_orig);
547a8e1175bSopenharmony_ci        ef = vaddq_u64(ef, ef_orig);
548a8e1175bSopenharmony_ci        gh = vaddq_u64(gh, gh_orig);
549a8e1175bSopenharmony_ci    }
550a8e1175bSopenharmony_ci
551a8e1175bSopenharmony_ci    vst1q_u64(&ctx->state[0], ab);
552a8e1175bSopenharmony_ci    vst1q_u64(&ctx->state[2], cd);
553a8e1175bSopenharmony_ci    vst1q_u64(&ctx->state[4], ef);
554a8e1175bSopenharmony_ci    vst1q_u64(&ctx->state[6], gh);
555a8e1175bSopenharmony_ci
556a8e1175bSopenharmony_ci    return processed;
557a8e1175bSopenharmony_ci}
558a8e1175bSopenharmony_ci
559a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
560a8e1175bSopenharmony_ci/*
561a8e1175bSopenharmony_ci * This function is for internal use only if we are building both C and A64
562a8e1175bSopenharmony_ci * versions, otherwise it is renamed to be the public mbedtls_internal_sha512_process()
563a8e1175bSopenharmony_ci */
564a8e1175bSopenharmony_cistatic
565a8e1175bSopenharmony_ci#endif
566a8e1175bSopenharmony_ciint mbedtls_internal_sha512_process_a64_crypto(mbedtls_sha512_context *ctx,
567a8e1175bSopenharmony_ci                                               const unsigned char data[SHA512_BLOCK_SIZE])
568a8e1175bSopenharmony_ci{
569a8e1175bSopenharmony_ci    return (mbedtls_internal_sha512_process_many_a64_crypto(ctx, data,
570a8e1175bSopenharmony_ci                                                            SHA512_BLOCK_SIZE) ==
571a8e1175bSopenharmony_ci            SHA512_BLOCK_SIZE) ? 0 : -1;
572a8e1175bSopenharmony_ci}
573a8e1175bSopenharmony_ci
574a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT || MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY */
575a8e1175bSopenharmony_ci
576a8e1175bSopenharmony_ci#if defined(MBEDTLS_POP_TARGET_PRAGMA)
577a8e1175bSopenharmony_ci#if defined(__clang__)
578a8e1175bSopenharmony_ci#pragma clang attribute pop
579a8e1175bSopenharmony_ci#elif defined(__GNUC__)
580a8e1175bSopenharmony_ci#pragma GCC pop_options
581a8e1175bSopenharmony_ci#endif
582a8e1175bSopenharmony_ci#undef MBEDTLS_POP_TARGET_PRAGMA
583a8e1175bSopenharmony_ci#endif
584a8e1175bSopenharmony_ci
585a8e1175bSopenharmony_ci
586a8e1175bSopenharmony_ci#if !defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
587a8e1175bSopenharmony_ci#define mbedtls_internal_sha512_process_many_c mbedtls_internal_sha512_process_many
588a8e1175bSopenharmony_ci#define mbedtls_internal_sha512_process_c      mbedtls_internal_sha512_process
589a8e1175bSopenharmony_ci#endif
590a8e1175bSopenharmony_ci
591a8e1175bSopenharmony_ci
592a8e1175bSopenharmony_ci#if !defined(MBEDTLS_SHA512_PROCESS_ALT) && !defined(MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY)
593a8e1175bSopenharmony_ci
594a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
595a8e1175bSopenharmony_ci/*
596a8e1175bSopenharmony_ci * This function is for internal use only if we are building both C and A64
597a8e1175bSopenharmony_ci * versions, otherwise it is renamed to be the public mbedtls_internal_sha512_process()
598a8e1175bSopenharmony_ci */
599a8e1175bSopenharmony_cistatic
600a8e1175bSopenharmony_ci#endif
601a8e1175bSopenharmony_ciint mbedtls_internal_sha512_process_c(mbedtls_sha512_context *ctx,
602a8e1175bSopenharmony_ci                                      const unsigned char data[SHA512_BLOCK_SIZE])
603a8e1175bSopenharmony_ci{
604a8e1175bSopenharmony_ci    int i;
605a8e1175bSopenharmony_ci    struct {
606a8e1175bSopenharmony_ci        uint64_t temp1, temp2, W[80];
607a8e1175bSopenharmony_ci        uint64_t A[8];
608a8e1175bSopenharmony_ci    } local;
609a8e1175bSopenharmony_ci
610a8e1175bSopenharmony_ci#define  SHR(x, n) ((x) >> (n))
611a8e1175bSopenharmony_ci#define ROTR(x, n) (SHR((x), (n)) | ((x) << (64 - (n))))
612a8e1175bSopenharmony_ci
613a8e1175bSopenharmony_ci#define S0(x) (ROTR(x, 1) ^ ROTR(x, 8) ^  SHR(x, 7))
614a8e1175bSopenharmony_ci#define S1(x) (ROTR(x, 19) ^ ROTR(x, 61) ^  SHR(x, 6))
615a8e1175bSopenharmony_ci
616a8e1175bSopenharmony_ci#define S2(x) (ROTR(x, 28) ^ ROTR(x, 34) ^ ROTR(x, 39))
617a8e1175bSopenharmony_ci#define S3(x) (ROTR(x, 14) ^ ROTR(x, 18) ^ ROTR(x, 41))
618a8e1175bSopenharmony_ci
619a8e1175bSopenharmony_ci#define F0(x, y, z) (((x) & (y)) | ((z) & ((x) | (y))))
620a8e1175bSopenharmony_ci#define F1(x, y, z) ((z) ^ ((x) & ((y) ^ (z))))
621a8e1175bSopenharmony_ci
622a8e1175bSopenharmony_ci#define P(a, b, c, d, e, f, g, h, x, K)                                      \
623a8e1175bSopenharmony_ci    do                                                              \
624a8e1175bSopenharmony_ci    {                                                               \
625a8e1175bSopenharmony_ci        local.temp1 = (h) + S3(e) + F1((e), (f), (g)) + (K) + (x);    \
626a8e1175bSopenharmony_ci        local.temp2 = S2(a) + F0((a), (b), (c));                      \
627a8e1175bSopenharmony_ci        (d) += local.temp1; (h) = local.temp1 + local.temp2;        \
628a8e1175bSopenharmony_ci    } while (0)
629a8e1175bSopenharmony_ci
630a8e1175bSopenharmony_ci    for (i = 0; i < 8; i++) {
631a8e1175bSopenharmony_ci        local.A[i] = ctx->state[i];
632a8e1175bSopenharmony_ci    }
633a8e1175bSopenharmony_ci
634a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_SMALLER)
635a8e1175bSopenharmony_ci    for (i = 0; i < 80; i++) {
636a8e1175bSopenharmony_ci        if (i < 16) {
637a8e1175bSopenharmony_ci            local.W[i] = MBEDTLS_GET_UINT64_BE(data, i << 3);
638a8e1175bSopenharmony_ci        } else {
639a8e1175bSopenharmony_ci            local.W[i] = S1(local.W[i -  2]) + local.W[i -  7] +
640a8e1175bSopenharmony_ci                         S0(local.W[i - 15]) + local.W[i - 16];
641a8e1175bSopenharmony_ci        }
642a8e1175bSopenharmony_ci
643a8e1175bSopenharmony_ci        P(local.A[0], local.A[1], local.A[2], local.A[3], local.A[4],
644a8e1175bSopenharmony_ci          local.A[5], local.A[6], local.A[7], local.W[i], K[i]);
645a8e1175bSopenharmony_ci
646a8e1175bSopenharmony_ci        local.temp1 = local.A[7]; local.A[7] = local.A[6];
647a8e1175bSopenharmony_ci        local.A[6] = local.A[5]; local.A[5] = local.A[4];
648a8e1175bSopenharmony_ci        local.A[4] = local.A[3]; local.A[3] = local.A[2];
649a8e1175bSopenharmony_ci        local.A[2] = local.A[1]; local.A[1] = local.A[0];
650a8e1175bSopenharmony_ci        local.A[0] = local.temp1;
651a8e1175bSopenharmony_ci    }
652a8e1175bSopenharmony_ci#else /* MBEDTLS_SHA512_SMALLER */
653a8e1175bSopenharmony_ci    for (i = 0; i < 16; i++) {
654a8e1175bSopenharmony_ci        local.W[i] = MBEDTLS_GET_UINT64_BE(data, i << 3);
655a8e1175bSopenharmony_ci    }
656a8e1175bSopenharmony_ci
657a8e1175bSopenharmony_ci    for (; i < 80; i++) {
658a8e1175bSopenharmony_ci        local.W[i] = S1(local.W[i -  2]) + local.W[i -  7] +
659a8e1175bSopenharmony_ci                     S0(local.W[i - 15]) + local.W[i - 16];
660a8e1175bSopenharmony_ci    }
661a8e1175bSopenharmony_ci
662a8e1175bSopenharmony_ci    i = 0;
663a8e1175bSopenharmony_ci    do {
664a8e1175bSopenharmony_ci        P(local.A[0], local.A[1], local.A[2], local.A[3], local.A[4],
665a8e1175bSopenharmony_ci          local.A[5], local.A[6], local.A[7], local.W[i], K[i]); i++;
666a8e1175bSopenharmony_ci        P(local.A[7], local.A[0], local.A[1], local.A[2], local.A[3],
667a8e1175bSopenharmony_ci          local.A[4], local.A[5], local.A[6], local.W[i], K[i]); i++;
668a8e1175bSopenharmony_ci        P(local.A[6], local.A[7], local.A[0], local.A[1], local.A[2],
669a8e1175bSopenharmony_ci          local.A[3], local.A[4], local.A[5], local.W[i], K[i]); i++;
670a8e1175bSopenharmony_ci        P(local.A[5], local.A[6], local.A[7], local.A[0], local.A[1],
671a8e1175bSopenharmony_ci          local.A[2], local.A[3], local.A[4], local.W[i], K[i]); i++;
672a8e1175bSopenharmony_ci        P(local.A[4], local.A[5], local.A[6], local.A[7], local.A[0],
673a8e1175bSopenharmony_ci          local.A[1], local.A[2], local.A[3], local.W[i], K[i]); i++;
674a8e1175bSopenharmony_ci        P(local.A[3], local.A[4], local.A[5], local.A[6], local.A[7],
675a8e1175bSopenharmony_ci          local.A[0], local.A[1], local.A[2], local.W[i], K[i]); i++;
676a8e1175bSopenharmony_ci        P(local.A[2], local.A[3], local.A[4], local.A[5], local.A[6],
677a8e1175bSopenharmony_ci          local.A[7], local.A[0], local.A[1], local.W[i], K[i]); i++;
678a8e1175bSopenharmony_ci        P(local.A[1], local.A[2], local.A[3], local.A[4], local.A[5],
679a8e1175bSopenharmony_ci          local.A[6], local.A[7], local.A[0], local.W[i], K[i]); i++;
680a8e1175bSopenharmony_ci    } while (i < 80);
681a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_SMALLER */
682a8e1175bSopenharmony_ci
683a8e1175bSopenharmony_ci    for (i = 0; i < 8; i++) {
684a8e1175bSopenharmony_ci        ctx->state[i] += local.A[i];
685a8e1175bSopenharmony_ci    }
686a8e1175bSopenharmony_ci
687a8e1175bSopenharmony_ci    /* Zeroise buffers and variables to clear sensitive data from memory. */
688a8e1175bSopenharmony_ci    mbedtls_platform_zeroize(&local, sizeof(local));
689a8e1175bSopenharmony_ci
690a8e1175bSopenharmony_ci    return 0;
691a8e1175bSopenharmony_ci}
692a8e1175bSopenharmony_ci
693a8e1175bSopenharmony_ci#endif /* !MBEDTLS_SHA512_PROCESS_ALT && !MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY */
694a8e1175bSopenharmony_ci
695a8e1175bSopenharmony_ci
696a8e1175bSopenharmony_ci#if !defined(MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY)
697a8e1175bSopenharmony_ci
698a8e1175bSopenharmony_cistatic size_t mbedtls_internal_sha512_process_many_c(
699a8e1175bSopenharmony_ci    mbedtls_sha512_context *ctx, const uint8_t *data, size_t len)
700a8e1175bSopenharmony_ci{
701a8e1175bSopenharmony_ci    size_t processed = 0;
702a8e1175bSopenharmony_ci
703a8e1175bSopenharmony_ci    while (len >= SHA512_BLOCK_SIZE) {
704a8e1175bSopenharmony_ci        if (mbedtls_internal_sha512_process_c(ctx, data) != 0) {
705a8e1175bSopenharmony_ci            return 0;
706a8e1175bSopenharmony_ci        }
707a8e1175bSopenharmony_ci
708a8e1175bSopenharmony_ci        data += SHA512_BLOCK_SIZE;
709a8e1175bSopenharmony_ci        len  -= SHA512_BLOCK_SIZE;
710a8e1175bSopenharmony_ci
711a8e1175bSopenharmony_ci        processed += SHA512_BLOCK_SIZE;
712a8e1175bSopenharmony_ci    }
713a8e1175bSopenharmony_ci
714a8e1175bSopenharmony_ci    return processed;
715a8e1175bSopenharmony_ci}
716a8e1175bSopenharmony_ci
717a8e1175bSopenharmony_ci#endif /* !MBEDTLS_SHA512_USE_A64_CRYPTO_ONLY */
718a8e1175bSopenharmony_ci
719a8e1175bSopenharmony_ci
720a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT)
721a8e1175bSopenharmony_ci
722a8e1175bSopenharmony_cistatic int mbedtls_a64_crypto_sha512_has_support(void)
723a8e1175bSopenharmony_ci{
724a8e1175bSopenharmony_ci    static int done = 0;
725a8e1175bSopenharmony_ci    static int supported = 0;
726a8e1175bSopenharmony_ci
727a8e1175bSopenharmony_ci    if (!done) {
728a8e1175bSopenharmony_ci        supported = mbedtls_a64_crypto_sha512_determine_support();
729a8e1175bSopenharmony_ci        done = 1;
730a8e1175bSopenharmony_ci    }
731a8e1175bSopenharmony_ci
732a8e1175bSopenharmony_ci    return supported;
733a8e1175bSopenharmony_ci}
734a8e1175bSopenharmony_ci
735a8e1175bSopenharmony_cistatic size_t mbedtls_internal_sha512_process_many(mbedtls_sha512_context *ctx,
736a8e1175bSopenharmony_ci                                                   const uint8_t *msg, size_t len)
737a8e1175bSopenharmony_ci{
738a8e1175bSopenharmony_ci    if (mbedtls_a64_crypto_sha512_has_support()) {
739a8e1175bSopenharmony_ci        return mbedtls_internal_sha512_process_many_a64_crypto(ctx, msg, len);
740a8e1175bSopenharmony_ci    } else {
741a8e1175bSopenharmony_ci        return mbedtls_internal_sha512_process_many_c(ctx, msg, len);
742a8e1175bSopenharmony_ci    }
743a8e1175bSopenharmony_ci}
744a8e1175bSopenharmony_ci
745a8e1175bSopenharmony_ciint mbedtls_internal_sha512_process(mbedtls_sha512_context *ctx,
746a8e1175bSopenharmony_ci                                    const unsigned char data[SHA512_BLOCK_SIZE])
747a8e1175bSopenharmony_ci{
748a8e1175bSopenharmony_ci    if (mbedtls_a64_crypto_sha512_has_support()) {
749a8e1175bSopenharmony_ci        return mbedtls_internal_sha512_process_a64_crypto(ctx, data);
750a8e1175bSopenharmony_ci    } else {
751a8e1175bSopenharmony_ci        return mbedtls_internal_sha512_process_c(ctx, data);
752a8e1175bSopenharmony_ci    }
753a8e1175bSopenharmony_ci}
754a8e1175bSopenharmony_ci
755a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_USE_A64_CRYPTO_IF_PRESENT */
756a8e1175bSopenharmony_ci
757a8e1175bSopenharmony_ci/*
758a8e1175bSopenharmony_ci * SHA-512 process buffer
759a8e1175bSopenharmony_ci */
760a8e1175bSopenharmony_ciint mbedtls_sha512_update(mbedtls_sha512_context *ctx,
761a8e1175bSopenharmony_ci                          const unsigned char *input,
762a8e1175bSopenharmony_ci                          size_t ilen)
763a8e1175bSopenharmony_ci{
764a8e1175bSopenharmony_ci    int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
765a8e1175bSopenharmony_ci    size_t fill;
766a8e1175bSopenharmony_ci    unsigned int left;
767a8e1175bSopenharmony_ci
768a8e1175bSopenharmony_ci    if (ilen == 0) {
769a8e1175bSopenharmony_ci        return 0;
770a8e1175bSopenharmony_ci    }
771a8e1175bSopenharmony_ci
772a8e1175bSopenharmony_ci    left = (unsigned int) (ctx->total[0] & 0x7F);
773a8e1175bSopenharmony_ci    fill = SHA512_BLOCK_SIZE - left;
774a8e1175bSopenharmony_ci
775a8e1175bSopenharmony_ci    ctx->total[0] += (uint64_t) ilen;
776a8e1175bSopenharmony_ci
777a8e1175bSopenharmony_ci    if (ctx->total[0] < (uint64_t) ilen) {
778a8e1175bSopenharmony_ci        ctx->total[1]++;
779a8e1175bSopenharmony_ci    }
780a8e1175bSopenharmony_ci
781a8e1175bSopenharmony_ci    if (left && ilen >= fill) {
782a8e1175bSopenharmony_ci        memcpy((void *) (ctx->buffer + left), input, fill);
783a8e1175bSopenharmony_ci
784a8e1175bSopenharmony_ci        if ((ret = mbedtls_internal_sha512_process(ctx, ctx->buffer)) != 0) {
785a8e1175bSopenharmony_ci            return ret;
786a8e1175bSopenharmony_ci        }
787a8e1175bSopenharmony_ci
788a8e1175bSopenharmony_ci        input += fill;
789a8e1175bSopenharmony_ci        ilen  -= fill;
790a8e1175bSopenharmony_ci        left = 0;
791a8e1175bSopenharmony_ci    }
792a8e1175bSopenharmony_ci
793a8e1175bSopenharmony_ci    while (ilen >= SHA512_BLOCK_SIZE) {
794a8e1175bSopenharmony_ci        size_t processed =
795a8e1175bSopenharmony_ci            mbedtls_internal_sha512_process_many(ctx, input, ilen);
796a8e1175bSopenharmony_ci        if (processed < SHA512_BLOCK_SIZE) {
797a8e1175bSopenharmony_ci            return MBEDTLS_ERR_ERROR_GENERIC_ERROR;
798a8e1175bSopenharmony_ci        }
799a8e1175bSopenharmony_ci
800a8e1175bSopenharmony_ci        input += processed;
801a8e1175bSopenharmony_ci        ilen  -= processed;
802a8e1175bSopenharmony_ci    }
803a8e1175bSopenharmony_ci
804a8e1175bSopenharmony_ci    if (ilen > 0) {
805a8e1175bSopenharmony_ci        memcpy((void *) (ctx->buffer + left), input, ilen);
806a8e1175bSopenharmony_ci    }
807a8e1175bSopenharmony_ci
808a8e1175bSopenharmony_ci    return 0;
809a8e1175bSopenharmony_ci}
810a8e1175bSopenharmony_ci
811a8e1175bSopenharmony_ci/*
812a8e1175bSopenharmony_ci * SHA-512 final digest
813a8e1175bSopenharmony_ci */
814a8e1175bSopenharmony_ciint mbedtls_sha512_finish(mbedtls_sha512_context *ctx,
815a8e1175bSopenharmony_ci                          unsigned char *output)
816a8e1175bSopenharmony_ci{
817a8e1175bSopenharmony_ci    int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
818a8e1175bSopenharmony_ci    unsigned used;
819a8e1175bSopenharmony_ci    uint64_t high, low;
820a8e1175bSopenharmony_ci    int truncated = 0;
821a8e1175bSopenharmony_ci
822a8e1175bSopenharmony_ci    /*
823a8e1175bSopenharmony_ci     * Add padding: 0x80 then 0x00 until 16 bytes remain for the length
824a8e1175bSopenharmony_ci     */
825a8e1175bSopenharmony_ci    used = ctx->total[0] & 0x7F;
826a8e1175bSopenharmony_ci
827a8e1175bSopenharmony_ci    ctx->buffer[used++] = 0x80;
828a8e1175bSopenharmony_ci
829a8e1175bSopenharmony_ci    if (used <= 112) {
830a8e1175bSopenharmony_ci        /* Enough room for padding + length in current block */
831a8e1175bSopenharmony_ci        memset(ctx->buffer + used, 0, 112 - used);
832a8e1175bSopenharmony_ci    } else {
833a8e1175bSopenharmony_ci        /* We'll need an extra block */
834a8e1175bSopenharmony_ci        memset(ctx->buffer + used, 0, SHA512_BLOCK_SIZE - used);
835a8e1175bSopenharmony_ci
836a8e1175bSopenharmony_ci        if ((ret = mbedtls_internal_sha512_process(ctx, ctx->buffer)) != 0) {
837a8e1175bSopenharmony_ci            goto exit;
838a8e1175bSopenharmony_ci        }
839a8e1175bSopenharmony_ci
840a8e1175bSopenharmony_ci        memset(ctx->buffer, 0, 112);
841a8e1175bSopenharmony_ci    }
842a8e1175bSopenharmony_ci
843a8e1175bSopenharmony_ci    /*
844a8e1175bSopenharmony_ci     * Add message length
845a8e1175bSopenharmony_ci     */
846a8e1175bSopenharmony_ci    high = (ctx->total[0] >> 61)
847a8e1175bSopenharmony_ci           | (ctx->total[1] <<  3);
848a8e1175bSopenharmony_ci    low  = (ctx->total[0] <<  3);
849a8e1175bSopenharmony_ci
850a8e1175bSopenharmony_ci    sha512_put_uint64_be(high, ctx->buffer, 112);
851a8e1175bSopenharmony_ci    sha512_put_uint64_be(low,  ctx->buffer, 120);
852a8e1175bSopenharmony_ci
853a8e1175bSopenharmony_ci    if ((ret = mbedtls_internal_sha512_process(ctx, ctx->buffer)) != 0) {
854a8e1175bSopenharmony_ci        goto exit;
855a8e1175bSopenharmony_ci    }
856a8e1175bSopenharmony_ci
857a8e1175bSopenharmony_ci    /*
858a8e1175bSopenharmony_ci     * Output final state
859a8e1175bSopenharmony_ci     */
860a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[0], output,  0);
861a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[1], output,  8);
862a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[2], output, 16);
863a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[3], output, 24);
864a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[4], output, 32);
865a8e1175bSopenharmony_ci    sha512_put_uint64_be(ctx->state[5], output, 40);
866a8e1175bSopenharmony_ci
867a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C)
868a8e1175bSopenharmony_ci    truncated = ctx->is384;
869a8e1175bSopenharmony_ci#endif
870a8e1175bSopenharmony_ci    if (!truncated) {
871a8e1175bSopenharmony_ci        sha512_put_uint64_be(ctx->state[6], output, 48);
872a8e1175bSopenharmony_ci        sha512_put_uint64_be(ctx->state[7], output, 56);
873a8e1175bSopenharmony_ci    }
874a8e1175bSopenharmony_ci
875a8e1175bSopenharmony_ci    ret = 0;
876a8e1175bSopenharmony_ci
877a8e1175bSopenharmony_ciexit:
878a8e1175bSopenharmony_ci    mbedtls_sha512_free(ctx);
879a8e1175bSopenharmony_ci    return ret;
880a8e1175bSopenharmony_ci}
881a8e1175bSopenharmony_ci
882a8e1175bSopenharmony_ci#endif /* !MBEDTLS_SHA512_ALT */
883a8e1175bSopenharmony_ci
884a8e1175bSopenharmony_ci/*
885a8e1175bSopenharmony_ci * output = SHA-512( input buffer )
886a8e1175bSopenharmony_ci */
887a8e1175bSopenharmony_ciint mbedtls_sha512(const unsigned char *input,
888a8e1175bSopenharmony_ci                   size_t ilen,
889a8e1175bSopenharmony_ci                   unsigned char *output,
890a8e1175bSopenharmony_ci                   int is384)
891a8e1175bSopenharmony_ci{
892a8e1175bSopenharmony_ci    int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED;
893a8e1175bSopenharmony_ci    mbedtls_sha512_context ctx;
894a8e1175bSopenharmony_ci
895a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C) && defined(MBEDTLS_SHA512_C)
896a8e1175bSopenharmony_ci    if (is384 != 0 && is384 != 1) {
897a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
898a8e1175bSopenharmony_ci    }
899a8e1175bSopenharmony_ci#elif defined(MBEDTLS_SHA512_C)
900a8e1175bSopenharmony_ci    if (is384 != 0) {
901a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
902a8e1175bSopenharmony_ci    }
903a8e1175bSopenharmony_ci#else /* defined MBEDTLS_SHA384_C only */
904a8e1175bSopenharmony_ci    if (is384 == 0) {
905a8e1175bSopenharmony_ci        return MBEDTLS_ERR_SHA512_BAD_INPUT_DATA;
906a8e1175bSopenharmony_ci    }
907a8e1175bSopenharmony_ci#endif
908a8e1175bSopenharmony_ci
909a8e1175bSopenharmony_ci    mbedtls_sha512_init(&ctx);
910a8e1175bSopenharmony_ci
911a8e1175bSopenharmony_ci    if ((ret = mbedtls_sha512_starts(&ctx, is384)) != 0) {
912a8e1175bSopenharmony_ci        goto exit;
913a8e1175bSopenharmony_ci    }
914a8e1175bSopenharmony_ci
915a8e1175bSopenharmony_ci    if ((ret = mbedtls_sha512_update(&ctx, input, ilen)) != 0) {
916a8e1175bSopenharmony_ci        goto exit;
917a8e1175bSopenharmony_ci    }
918a8e1175bSopenharmony_ci
919a8e1175bSopenharmony_ci    if ((ret = mbedtls_sha512_finish(&ctx, output)) != 0) {
920a8e1175bSopenharmony_ci        goto exit;
921a8e1175bSopenharmony_ci    }
922a8e1175bSopenharmony_ci
923a8e1175bSopenharmony_ciexit:
924a8e1175bSopenharmony_ci    mbedtls_sha512_free(&ctx);
925a8e1175bSopenharmony_ci
926a8e1175bSopenharmony_ci    return ret;
927a8e1175bSopenharmony_ci}
928a8e1175bSopenharmony_ci
929a8e1175bSopenharmony_ci#if defined(MBEDTLS_SELF_TEST)
930a8e1175bSopenharmony_ci
931a8e1175bSopenharmony_ci/*
932a8e1175bSopenharmony_ci * FIPS-180-2 test vectors
933a8e1175bSopenharmony_ci */
934a8e1175bSopenharmony_cistatic const unsigned char sha_test_buf[3][113] =
935a8e1175bSopenharmony_ci{
936a8e1175bSopenharmony_ci    { "abc" },
937a8e1175bSopenharmony_ci    {
938a8e1175bSopenharmony_ci        "abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu"
939a8e1175bSopenharmony_ci    },
940a8e1175bSopenharmony_ci    { "" }
941a8e1175bSopenharmony_ci};
942a8e1175bSopenharmony_ci
943a8e1175bSopenharmony_cistatic const size_t sha_test_buflen[3] =
944a8e1175bSopenharmony_ci{
945a8e1175bSopenharmony_ci    3, 112, 1000
946a8e1175bSopenharmony_ci};
947a8e1175bSopenharmony_ci
948a8e1175bSopenharmony_citypedef const unsigned char (sha_test_sum_t)[64];
949a8e1175bSopenharmony_ci
950a8e1175bSopenharmony_ci/*
951a8e1175bSopenharmony_ci * SHA-384 test vectors
952a8e1175bSopenharmony_ci */
953a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C)
954a8e1175bSopenharmony_cistatic sha_test_sum_t sha384_test_sum[] =
955a8e1175bSopenharmony_ci{
956a8e1175bSopenharmony_ci    { 0xCB, 0x00, 0x75, 0x3F, 0x45, 0xA3, 0x5E, 0x8B,
957a8e1175bSopenharmony_ci      0xB5, 0xA0, 0x3D, 0x69, 0x9A, 0xC6, 0x50, 0x07,
958a8e1175bSopenharmony_ci      0x27, 0x2C, 0x32, 0xAB, 0x0E, 0xDE, 0xD1, 0x63,
959a8e1175bSopenharmony_ci      0x1A, 0x8B, 0x60, 0x5A, 0x43, 0xFF, 0x5B, 0xED,
960a8e1175bSopenharmony_ci      0x80, 0x86, 0x07, 0x2B, 0xA1, 0xE7, 0xCC, 0x23,
961a8e1175bSopenharmony_ci      0x58, 0xBA, 0xEC, 0xA1, 0x34, 0xC8, 0x25, 0xA7 },
962a8e1175bSopenharmony_ci    { 0x09, 0x33, 0x0C, 0x33, 0xF7, 0x11, 0x47, 0xE8,
963a8e1175bSopenharmony_ci      0x3D, 0x19, 0x2F, 0xC7, 0x82, 0xCD, 0x1B, 0x47,
964a8e1175bSopenharmony_ci      0x53, 0x11, 0x1B, 0x17, 0x3B, 0x3B, 0x05, 0xD2,
965a8e1175bSopenharmony_ci      0x2F, 0xA0, 0x80, 0x86, 0xE3, 0xB0, 0xF7, 0x12,
966a8e1175bSopenharmony_ci      0xFC, 0xC7, 0xC7, 0x1A, 0x55, 0x7E, 0x2D, 0xB9,
967a8e1175bSopenharmony_ci      0x66, 0xC3, 0xE9, 0xFA, 0x91, 0x74, 0x60, 0x39 },
968a8e1175bSopenharmony_ci    { 0x9D, 0x0E, 0x18, 0x09, 0x71, 0x64, 0x74, 0xCB,
969a8e1175bSopenharmony_ci      0x08, 0x6E, 0x83, 0x4E, 0x31, 0x0A, 0x4A, 0x1C,
970a8e1175bSopenharmony_ci      0xED, 0x14, 0x9E, 0x9C, 0x00, 0xF2, 0x48, 0x52,
971a8e1175bSopenharmony_ci      0x79, 0x72, 0xCE, 0xC5, 0x70, 0x4C, 0x2A, 0x5B,
972a8e1175bSopenharmony_ci      0x07, 0xB8, 0xB3, 0xDC, 0x38, 0xEC, 0xC4, 0xEB,
973a8e1175bSopenharmony_ci      0xAE, 0x97, 0xDD, 0xD8, 0x7F, 0x3D, 0x89, 0x85 }
974a8e1175bSopenharmony_ci};
975a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA384_C */
976a8e1175bSopenharmony_ci
977a8e1175bSopenharmony_ci/*
978a8e1175bSopenharmony_ci * SHA-512 test vectors
979a8e1175bSopenharmony_ci */
980a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_C)
981a8e1175bSopenharmony_cistatic sha_test_sum_t sha512_test_sum[] =
982a8e1175bSopenharmony_ci{
983a8e1175bSopenharmony_ci    { 0xDD, 0xAF, 0x35, 0xA1, 0x93, 0x61, 0x7A, 0xBA,
984a8e1175bSopenharmony_ci      0xCC, 0x41, 0x73, 0x49, 0xAE, 0x20, 0x41, 0x31,
985a8e1175bSopenharmony_ci      0x12, 0xE6, 0xFA, 0x4E, 0x89, 0xA9, 0x7E, 0xA2,
986a8e1175bSopenharmony_ci      0x0A, 0x9E, 0xEE, 0xE6, 0x4B, 0x55, 0xD3, 0x9A,
987a8e1175bSopenharmony_ci      0x21, 0x92, 0x99, 0x2A, 0x27, 0x4F, 0xC1, 0xA8,
988a8e1175bSopenharmony_ci      0x36, 0xBA, 0x3C, 0x23, 0xA3, 0xFE, 0xEB, 0xBD,
989a8e1175bSopenharmony_ci      0x45, 0x4D, 0x44, 0x23, 0x64, 0x3C, 0xE8, 0x0E,
990a8e1175bSopenharmony_ci      0x2A, 0x9A, 0xC9, 0x4F, 0xA5, 0x4C, 0xA4, 0x9F },
991a8e1175bSopenharmony_ci    { 0x8E, 0x95, 0x9B, 0x75, 0xDA, 0xE3, 0x13, 0xDA,
992a8e1175bSopenharmony_ci      0x8C, 0xF4, 0xF7, 0x28, 0x14, 0xFC, 0x14, 0x3F,
993a8e1175bSopenharmony_ci      0x8F, 0x77, 0x79, 0xC6, 0xEB, 0x9F, 0x7F, 0xA1,
994a8e1175bSopenharmony_ci      0x72, 0x99, 0xAE, 0xAD, 0xB6, 0x88, 0x90, 0x18,
995a8e1175bSopenharmony_ci      0x50, 0x1D, 0x28, 0x9E, 0x49, 0x00, 0xF7, 0xE4,
996a8e1175bSopenharmony_ci      0x33, 0x1B, 0x99, 0xDE, 0xC4, 0xB5, 0x43, 0x3A,
997a8e1175bSopenharmony_ci      0xC7, 0xD3, 0x29, 0xEE, 0xB6, 0xDD, 0x26, 0x54,
998a8e1175bSopenharmony_ci      0x5E, 0x96, 0xE5, 0x5B, 0x87, 0x4B, 0xE9, 0x09 },
999a8e1175bSopenharmony_ci    { 0xE7, 0x18, 0x48, 0x3D, 0x0C, 0xE7, 0x69, 0x64,
1000a8e1175bSopenharmony_ci      0x4E, 0x2E, 0x42, 0xC7, 0xBC, 0x15, 0xB4, 0x63,
1001a8e1175bSopenharmony_ci      0x8E, 0x1F, 0x98, 0xB1, 0x3B, 0x20, 0x44, 0x28,
1002a8e1175bSopenharmony_ci      0x56, 0x32, 0xA8, 0x03, 0xAF, 0xA9, 0x73, 0xEB,
1003a8e1175bSopenharmony_ci      0xDE, 0x0F, 0xF2, 0x44, 0x87, 0x7E, 0xA6, 0x0A,
1004a8e1175bSopenharmony_ci      0x4C, 0xB0, 0x43, 0x2C, 0xE5, 0x77, 0xC3, 0x1B,
1005a8e1175bSopenharmony_ci      0xEB, 0x00, 0x9C, 0x5C, 0x2C, 0x49, 0xAA, 0x2E,
1006a8e1175bSopenharmony_ci      0x4E, 0xAD, 0xB2, 0x17, 0xAD, 0x8C, 0xC0, 0x9B }
1007a8e1175bSopenharmony_ci};
1008a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_C */
1009a8e1175bSopenharmony_ci
1010a8e1175bSopenharmony_cistatic int mbedtls_sha512_common_self_test(int verbose, int is384)
1011a8e1175bSopenharmony_ci{
1012a8e1175bSopenharmony_ci    int i, buflen, ret = 0;
1013a8e1175bSopenharmony_ci    unsigned char *buf;
1014a8e1175bSopenharmony_ci    unsigned char sha512sum[64];
1015a8e1175bSopenharmony_ci    mbedtls_sha512_context ctx;
1016a8e1175bSopenharmony_ci
1017a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C) && defined(MBEDTLS_SHA512_C)
1018a8e1175bSopenharmony_ci    sha_test_sum_t *sha_test_sum = (is384) ? sha384_test_sum : sha512_test_sum;
1019a8e1175bSopenharmony_ci#elif defined(MBEDTLS_SHA512_C)
1020a8e1175bSopenharmony_ci    sha_test_sum_t *sha_test_sum = sha512_test_sum;
1021a8e1175bSopenharmony_ci#else
1022a8e1175bSopenharmony_ci    sha_test_sum_t *sha_test_sum = sha384_test_sum;
1023a8e1175bSopenharmony_ci#endif
1024a8e1175bSopenharmony_ci
1025a8e1175bSopenharmony_ci    buf = mbedtls_calloc(1024, sizeof(unsigned char));
1026a8e1175bSopenharmony_ci    if (NULL == buf) {
1027a8e1175bSopenharmony_ci        if (verbose != 0) {
1028a8e1175bSopenharmony_ci            mbedtls_printf("Buffer allocation failed\n");
1029a8e1175bSopenharmony_ci        }
1030a8e1175bSopenharmony_ci
1031a8e1175bSopenharmony_ci        return 1;
1032a8e1175bSopenharmony_ci    }
1033a8e1175bSopenharmony_ci
1034a8e1175bSopenharmony_ci    mbedtls_sha512_init(&ctx);
1035a8e1175bSopenharmony_ci
1036a8e1175bSopenharmony_ci    for (i = 0; i < 3; i++) {
1037a8e1175bSopenharmony_ci        if (verbose != 0) {
1038a8e1175bSopenharmony_ci            mbedtls_printf("  SHA-%d test #%d: ", 512 - is384 * 128, i + 1);
1039a8e1175bSopenharmony_ci        }
1040a8e1175bSopenharmony_ci
1041a8e1175bSopenharmony_ci        if ((ret = mbedtls_sha512_starts(&ctx, is384)) != 0) {
1042a8e1175bSopenharmony_ci            goto fail;
1043a8e1175bSopenharmony_ci        }
1044a8e1175bSopenharmony_ci
1045a8e1175bSopenharmony_ci        if (i == 2) {
1046a8e1175bSopenharmony_ci            memset(buf, 'a', buflen = 1000);
1047a8e1175bSopenharmony_ci
1048a8e1175bSopenharmony_ci            for (int j = 0; j < 1000; j++) {
1049a8e1175bSopenharmony_ci                ret = mbedtls_sha512_update(&ctx, buf, buflen);
1050a8e1175bSopenharmony_ci                if (ret != 0) {
1051a8e1175bSopenharmony_ci                    goto fail;
1052a8e1175bSopenharmony_ci                }
1053a8e1175bSopenharmony_ci            }
1054a8e1175bSopenharmony_ci        } else {
1055a8e1175bSopenharmony_ci            ret = mbedtls_sha512_update(&ctx, sha_test_buf[i],
1056a8e1175bSopenharmony_ci                                        sha_test_buflen[i]);
1057a8e1175bSopenharmony_ci            if (ret != 0) {
1058a8e1175bSopenharmony_ci                goto fail;
1059a8e1175bSopenharmony_ci            }
1060a8e1175bSopenharmony_ci        }
1061a8e1175bSopenharmony_ci
1062a8e1175bSopenharmony_ci        if ((ret = mbedtls_sha512_finish(&ctx, sha512sum)) != 0) {
1063a8e1175bSopenharmony_ci            goto fail;
1064a8e1175bSopenharmony_ci        }
1065a8e1175bSopenharmony_ci
1066a8e1175bSopenharmony_ci        if (memcmp(sha512sum, sha_test_sum[i], 64 - is384 * 16) != 0) {
1067a8e1175bSopenharmony_ci            ret = 1;
1068a8e1175bSopenharmony_ci            goto fail;
1069a8e1175bSopenharmony_ci        }
1070a8e1175bSopenharmony_ci
1071a8e1175bSopenharmony_ci        if (verbose != 0) {
1072a8e1175bSopenharmony_ci            mbedtls_printf("passed\n");
1073a8e1175bSopenharmony_ci        }
1074a8e1175bSopenharmony_ci    }
1075a8e1175bSopenharmony_ci
1076a8e1175bSopenharmony_ci    if (verbose != 0) {
1077a8e1175bSopenharmony_ci        mbedtls_printf("\n");
1078a8e1175bSopenharmony_ci    }
1079a8e1175bSopenharmony_ci
1080a8e1175bSopenharmony_ci    goto exit;
1081a8e1175bSopenharmony_ci
1082a8e1175bSopenharmony_cifail:
1083a8e1175bSopenharmony_ci    if (verbose != 0) {
1084a8e1175bSopenharmony_ci        mbedtls_printf("failed\n");
1085a8e1175bSopenharmony_ci    }
1086a8e1175bSopenharmony_ci
1087a8e1175bSopenharmony_ciexit:
1088a8e1175bSopenharmony_ci    mbedtls_sha512_free(&ctx);
1089a8e1175bSopenharmony_ci    mbedtls_free(buf);
1090a8e1175bSopenharmony_ci
1091a8e1175bSopenharmony_ci    return ret;
1092a8e1175bSopenharmony_ci}
1093a8e1175bSopenharmony_ci
1094a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA512_C)
1095a8e1175bSopenharmony_ciint mbedtls_sha512_self_test(int verbose)
1096a8e1175bSopenharmony_ci{
1097a8e1175bSopenharmony_ci    return mbedtls_sha512_common_self_test(verbose, 0);
1098a8e1175bSopenharmony_ci}
1099a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_C */
1100a8e1175bSopenharmony_ci
1101a8e1175bSopenharmony_ci#if defined(MBEDTLS_SHA384_C)
1102a8e1175bSopenharmony_ciint mbedtls_sha384_self_test(int verbose)
1103a8e1175bSopenharmony_ci{
1104a8e1175bSopenharmony_ci    return mbedtls_sha512_common_self_test(verbose, 1);
1105a8e1175bSopenharmony_ci}
1106a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA384_C */
1107a8e1175bSopenharmony_ci
1108a8e1175bSopenharmony_ci#undef ARRAY_LENGTH
1109a8e1175bSopenharmony_ci
1110a8e1175bSopenharmony_ci#endif /* MBEDTLS_SELF_TEST */
1111a8e1175bSopenharmony_ci
1112a8e1175bSopenharmony_ci#endif /* MBEDTLS_SHA512_C || MBEDTLS_SHA384_C */
1113