saramonteiro commented on a change in pull request #3882: URL: https://github.com/apache/incubator-nuttx/pull/3882#discussion_r651797430
########## File path: arch/risc-v/src/esp32c3/esp32c3_sha.c ########## @@ -0,0 +1,1666 @@ +/**************************************************************************** + * arch/risc-v/src/esp32c3/esp32c3_sha.c + * + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. The + * ASF licenses this file to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance with the + * License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT + * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the + * License for the specific language governing permissions and limitations + * under the License. + * + ****************************************************************************/ + +/**************************************************************************** + * Included Files + ****************************************************************************/ + +#include <nuttx/config.h> + +#ifdef CONFIG_ESP32C3_SHA_ACCELERATOR + +#include <stdint.h> +#include <stdio.h> +#include <string.h> +#include <stdlib.h> +#include <inttypes.h> +#include <debug.h> +#include <semaphore.h> + +#include "riscv_arch.h" +#include "hardware/esp32c3_sha.h" +#include "hardware/esp32c3_system.h" + +#include "esp32c3_sha.h" + +/**************************************************************************** + * Pre-processor Definitions + ****************************************************************************/ + +#define PUT_UINT32_BE(n,b,i) \ +{ \ + (b)[(i) ] = (unsigned char) ((n) >> 24); \ + (b)[(i) + 1] = (unsigned char) ((n) >> 16); \ + (b)[(i) + 2] = (unsigned char) ((n) >> 8); \ + (b)[(i) + 3] = (unsigned char) ((n) ); \ +} + +#define GET_UINT64_BE(n,b,i) \ +{ \ + (n) = ((uint64_t) (b)[(i) ] << 56) \ + | ((uint64_t) (b)[(i) + 1] << 48) \ + | ((uint64_t) (b)[(i) + 2] << 40) \ + | ((uint64_t) (b)[(i) + 3] << 32) \ + | ((uint64_t) (b)[(i) + 4] << 24) \ + | ((uint64_t) (b)[(i) + 5] << 16) \ + | ((uint64_t) (b)[(i) + 6] << 8) \ + | ((uint64_t) (b)[(i) + 7] ); \ +} + +#define PUT_UINT64_BE(n,b,i) \ +{ \ + (b)[(i) ] = (uint8_t) ((n) >> 56); \ + (b)[(i) + 1] = (uint8_t) ((n) >> 48); \ + (b)[(i) + 2] = (uint8_t) ((n) >> 40); \ + (b)[(i) + 3] = (uint8_t) ((n) >> 32); \ + (b)[(i) + 4] = (uint8_t) ((n) >> 24); \ + (b)[(i) + 5] = (uint8_t) ((n) >> 16); \ + (b)[(i) + 6] = (uint8_t) ((n) >> 8); \ + (b)[(i) + 7] = (uint8_t) ((n) ); \ +} + +#define SHR(x,n) ((x) >> (n)) +#define ROTR(x,n) (SHR((x),(n)) | ((x) << (64 - (n)))) + +#define S0(x) (ROTR(x, 1) ^ ROTR(x, 8) ^ SHR(x, 7)) +#define S1(x) (ROTR(x,19) ^ ROTR(x,61) ^ SHR(x, 6)) + +#define S2(x) (ROTR(x,28) ^ ROTR(x,34) ^ ROTR(x,39)) +#define S3(x) (ROTR(x,14) ^ ROTR(x,18) ^ ROTR(x,41)) + +#define F0(x,y,z) (((x) & (y)) | ((z) & ((x) | (y)))) +#define F1(x,y,z) ((z) ^ ((x) & ((y) ^ (z)))) + +#define P(a,b,c,d,e,f,g,h,x,K) \ + do \ + { \ + temp1 = (h) + S3(e) + F1((e),(f),(g)) + (K) + (x); \ + temp2 = S2(a) + F0((a),(b),(c)); \ + (d) += temp1; \ + (h) = temp1 + temp2; \ + } while(0) + +#define SHA1_BLK_SIZE (20) +#define SHA2_BLK_SIZE (32) + +/**************************************************************************** + * Private Data + ****************************************************************************/ + +static bool g_sha_inited; +static sem_t g_sha_sem = SEM_INITIALIZER(1); +static const unsigned char esp32c3_sha_padding[64] = +{ + 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, + 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 +}; + +static const uint64_t K[80] = +{ + UINT64_C(0x428a2f98d728ae22), UINT64_C(0x7137449123ef65cd), + UINT64_C(0xb5c0fbcfec4d3b2f), UINT64_C(0xe9b5dba58189dbbc), + UINT64_C(0x3956c25bf348b538), UINT64_C(0x59f111f1b605d019), + UINT64_C(0x923f82a4af194f9b), UINT64_C(0xab1c5ed5da6d8118), + UINT64_C(0xd807aa98a3030242), UINT64_C(0x12835b0145706fbe), + UINT64_C(0x243185be4ee4b28c), UINT64_C(0x550c7dc3d5ffb4e2), + UINT64_C(0x72be5d74f27b896f), UINT64_C(0x80deb1fe3b1696b1), + UINT64_C(0x9bdc06a725c71235), UINT64_C(0xc19bf174cf692694), + UINT64_C(0xe49b69c19ef14ad2), UINT64_C(0xefbe4786384f25e3), + UINT64_C(0x0fc19dc68b8cd5b5), UINT64_C(0x240ca1cc77ac9c65), + UINT64_C(0x2de92c6f592b0275), UINT64_C(0x4a7484aa6ea6e483), + UINT64_C(0x5cb0a9dcbd41fbd4), UINT64_C(0x76f988da831153b5), + UINT64_C(0x983e5152ee66dfab), UINT64_C(0xa831c66d2db43210), + UINT64_C(0xb00327c898fb213f), UINT64_C(0xbf597fc7beef0ee4), + UINT64_C(0xc6e00bf33da88fc2), UINT64_C(0xd5a79147930aa725), + UINT64_C(0x06ca6351e003826f), UINT64_C(0x142929670a0e6e70), + UINT64_C(0x27b70a8546d22ffc), UINT64_C(0x2e1b21385c26c926), + UINT64_C(0x4d2c6dfc5ac42aed), UINT64_C(0x53380d139d95b3df), + UINT64_C(0x650a73548baf63de), UINT64_C(0x766a0abb3c77b2a8), + UINT64_C(0x81c2c92e47edaee6), UINT64_C(0x92722c851482353b), + UINT64_C(0xa2bfe8a14cf10364), UINT64_C(0xa81a664bbc423001), + UINT64_C(0xc24b8b70d0f89791), UINT64_C(0xc76c51a30654be30), + UINT64_C(0xd192e819d6ef5218), UINT64_C(0xd69906245565a910), + UINT64_C(0xf40e35855771202a), UINT64_C(0x106aa07032bbd1b8), + UINT64_C(0x19a4c116b8d2d0c8), UINT64_C(0x1e376c085141ab53), + UINT64_C(0x2748774cdf8eeb99), UINT64_C(0x34b0bcb5e19b48a8), + UINT64_C(0x391c0cb3c5c95a63), UINT64_C(0x4ed8aa4ae3418acb), + UINT64_C(0x5b9cca4f7763e373), UINT64_C(0x682e6ff3d6b2b8a3), + UINT64_C(0x748f82ee5defb2fc), UINT64_C(0x78a5636f43172f60), + UINT64_C(0x84c87814a1f0ab72), UINT64_C(0x8cc702081a6439ec), + UINT64_C(0x90befffa23631e28), UINT64_C(0xa4506cebde82bde9), + UINT64_C(0xbef9a3f7b2c67915), UINT64_C(0xc67178f2e372532b), + UINT64_C(0xca273eceea26619c), UINT64_C(0xd186b8c721c0c207), + UINT64_C(0xeada7dd6cde0eb1e), UINT64_C(0xf57d4f7fee6ed178), + UINT64_C(0x06f067aa72176fba), UINT64_C(0x0a637dc5a2c898a6), + UINT64_C(0x113f9804bef90dae), UINT64_C(0x1b710b35131c471b), + UINT64_C(0x28db77f523047d84), UINT64_C(0x32caab7b40c72493), + UINT64_C(0x3c9ebe0a15c9bebc), UINT64_C(0x431d67c49c100d4c), + UINT64_C(0x4cc5d4becb3e42b6), UINT64_C(0x597f299cfc657e2a), + UINT64_C(0x5fcb6fab3ad6faec), UINT64_C(0x6c44198c4a475817) +}; + +/**************************************************************************** + * Private Functions + ****************************************************************************/ + +static int esp32c3_sha1_block(struct esp32c3_sha1_context_s *ctx, + const uint8_t *data, size_t len, + uint8_t *buf, size_t buf_len) +{ + uint32_t *data_words = NULL; + + size_t blk_len = 0; + size_t blk_word_len = 0; + int num_block = 0; + int i; + int j; + + blk_len = 64; + blk_word_len = blk_len / 4; + num_block = len / blk_len; + + putreg32(ctx->mode, SHA_MODE_REG); + + if (buf_len != 0) + { + data_words = (uint32_t *)buf; + + while (getreg32(SHA_BUSY_REG)) + { + } + + for (i = 0; i < blk_word_len; i++) + { + putreg32(data_words[i], SHA_M_0_REG + i * 4); + } + + if (ctx->first_block) + { + putreg32(1, SHA_START_REG); + } + else + { + putreg32(1, SHA_CONTINUE_REG); + } + + ctx->first_block = false; + } + + for (j = 0; j < num_block; j++) + { + data_words = (uint32_t *)(data + blk_len * j); + + while (getreg32(SHA_BUSY_REG)) + { + } + + for (i = 0; i < blk_word_len; i++) + { + putreg32(data_words[i], SHA_M_0_REG + i * 4); + } + + if (ctx->first_block) + { + putreg32(1, SHA_START_REG); + } + else + { + putreg32(1, SHA_CONTINUE_REG); + } + + ctx->first_block = false; + } + + while (getreg32(SHA_BUSY_REG)) + { + } + + ctx->state[0] = getreg32(SHA_H_0_REG + 0); + ctx->state[1] = getreg32(SHA_H_0_REG + 4); + ctx->state[2] = getreg32(SHA_H_0_REG + 8); + ctx->state[3] = getreg32(SHA_H_0_REG + 12); + ctx->state[4] = getreg32(SHA_H_0_REG + 16); + + return 0; +} + +static int esp32c3_sha256_block(struct esp32c3_sha256_context_s *ctx, + const uint8_t *data, size_t len, + uint8_t *buf, size_t buf_len) +{ + uint32_t *data_words = NULL; + + size_t blk_len = 0; + size_t blk_word_len = 0; + int num_block = 0; + int i; + int j; + + blk_len = 64; + blk_word_len = blk_len / 4; + num_block = len / blk_len; + + putreg32(ctx->mode, SHA_MODE_REG); + + if (buf_len != 0) + { + data_words = (uint32_t *)buf; + + while (getreg32(SHA_BUSY_REG)) + { + } + + for (i = 0; i < blk_word_len; i++) + { + putreg32(data_words[i], SHA_M_0_REG + i * 4); + } + + if (ctx->first_block) + { + putreg32(1, SHA_START_REG); + } + else + { + putreg32(1, SHA_CONTINUE_REG); + } + + ctx->first_block = false; + } + + for (j = 0; j < num_block; j++) + { + data_words = (uint32_t *)(data + blk_len * j); + + while (getreg32(SHA_BUSY_REG)) + { + } + + for (i = 0; i < blk_word_len; i++) + { + putreg32(data_words[i], SHA_M_0_REG + i * 4); + } + + if (ctx->first_block) + { + putreg32(1, SHA_START_REG); + } + else + { + putreg32(1, SHA_CONTINUE_REG); + } + + ctx->first_block = false; + } + + while (getreg32(SHA_BUSY_REG)) + { + } + + ctx->state[0] = getreg32(SHA_H_0_REG + 0); + ctx->state[1] = getreg32(SHA_H_0_REG + 4); + ctx->state[2] = getreg32(SHA_H_0_REG + 8); + ctx->state[3] = getreg32(SHA_H_0_REG + 12); + ctx->state[4] = getreg32(SHA_H_0_REG + 16); + ctx->state[5] = getreg32(SHA_H_0_REG + 20); + ctx->state[6] = getreg32(SHA_H_0_REG + 24); Review comment: why not to use a loop here? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
