blob: de06d25f9b15e450b0688b1399c8565aaa8d091d [file] [log] [blame]
/*
* Copyright (c) 2019, ARM Limited and Contributors. All rights reserved.
* Copyright (c) 2019-2023, Intel Corporation. All rights reserved.
* Copyright (c) 2024, Altera Corporation. All rights reserved.
*
* SPDX-License-Identifier: BSD-3-Clause
*/
#include <assert.h>
#include <errno.h>
#include <arch_helpers.h>
#include <common/debug.h>
#include <common/tbbr/tbbr_img_def.h>
#include <drivers/delay_timer.h>
#include <lib/mmio.h>
#include <lib/utils.h>
#include <tools_share/firmware_image_package.h>
#include "socfpga_mailbox.h"
#include "socfpga_vab.h"
static size_t get_img_size(uint8_t *img_buf, size_t img_buf_sz)
{
uint8_t *img_buf_end = img_buf + img_buf_sz;
uint32_t cert_sz = get_unaligned_le32(img_buf_end - sizeof(uint32_t));
uint8_t *p = img_buf_end - cert_sz - sizeof(uint32_t);
/* Ensure p is pointing within the img_buf */
if (p < img_buf || p > (img_buf_end - VAB_CERT_HEADER_SIZE))
return 0;
if (get_unaligned_le32(p) == SDM_CERT_MAGIC_NUM)
return (size_t)(p - img_buf);
return 0;
}
int socfpga_vendor_authentication(void **p_image, size_t *p_size)
{
int retry_count = 20;
uint8_t hash384[FCS_SHA384_WORD_SIZE];
uint64_t img_addr, mbox_data_addr;
uint32_t img_sz, mbox_data_sz;
uint8_t *cert_hash_ptr, *mbox_relocate_data_addr;
uint32_t resp = 0, resp_len = 1;
int ret = 0;
img_addr = (uintptr_t)*p_image;
img_sz = get_img_size((uint8_t *)img_addr, *p_size);
if (!img_sz) {
NOTICE("VAB certificate not found in image!\n");
return -ENOVABCERT;
}
if (!IS_BYTE_ALIGNED(img_sz, sizeof(uint32_t))) {
NOTICE("Image size (%d bytes) not aliged to 4 bytes!\n", img_sz);
return -EIMGERR;
}
/* Generate HASH384 from the image */
/* TODO: This part need to cross check !!!!!! */
sha384_csum_wd((uint8_t *)img_addr, img_sz, hash384, CHUNKSZ_PER_WD_RESET);
cert_hash_ptr = (uint8_t *)(img_addr + img_sz +
VAB_CERT_MAGIC_OFFSET + VAB_CERT_FIT_SHA384_OFFSET);
/*
* Compare the SHA384 found in certificate against the SHA384
* calculated from image
*/
if (memcmp(hash384, cert_hash_ptr, FCS_SHA384_WORD_SIZE)) {
NOTICE("SHA384 does not match!\n");
return -EKEYREJECTED;
}
mbox_data_addr = img_addr + img_sz - sizeof(uint32_t);
/* Size in word (32bits) */
mbox_data_sz = (BYTE_ALIGN(*p_size - img_sz, sizeof(uint32_t))) >> 2;
NOTICE("mbox_data_addr = %lx mbox_data_sz = %d\n", mbox_data_addr, mbox_data_sz);
/* TODO: This part need to cross check !!!!!! */
// mbox_relocate_data_addr = (uint8_t *)malloc(mbox_data_sz * sizeof(uint32_t));
// if (!mbox_relocate_data_addr) {
// NOTICE("Cannot allocate memory for VAB certificate relocation!\n");
// return -ENOMEM;
// }
memcpy(mbox_relocate_data_addr, (uint8_t *)mbox_data_addr, mbox_data_sz * sizeof(uint32_t));
*(uint32_t *)mbox_relocate_data_addr = 0;
do {
/* Invoke SMC call to ATF to send the VAB certificate to SDM */
ret = mailbox_send_cmd(MBOX_JOB_ID, MBOX_CMD_VAB_SRC_CERT,
(uint32_t *)mbox_relocate_data_addr, mbox_data_sz, 0, &resp, &resp_len);
/* If SDM is not available, just delay 50ms and retry again */
/* 0x1FF = The device is busy */
if (ret == MBOX_RESP_ERR(0x1FF)) {
mdelay(50);
} else {
break;
}
} while (--retry_count);
/* Free the relocate certificate memory space */
zeromem((void *)&mbox_relocate_data_addr, sizeof(uint32_t));
/* Exclude the size of the VAB certificate from image size */
*p_size = img_sz;
if (ret) {
/*
* Unsupported mailbox command or device not in the
* owned/secure state
*/
/* 0x85 = Not allowed under current security setting */
if (ret == MBOX_RESP_ERR(0x85)) {
/* SDM bypass authentication */
NOTICE("Image Authentication bypassed at address\n");
return 0;
}
NOTICE("VAB certificate authentication failed in SDM\n");
/* 0x1FF = The device is busy */
if (ret == MBOX_RESP_ERR(0x1FF)) {
NOTICE("Operation timed out\n");
return -ETIMEOUT;
} else if (ret == MBOX_WRONG_ID) {
NOTICE("No such process\n");
return -EPROCESS;
}
} else {
/* If Certificate Process Status has error */
if (resp) {
NOTICE("VAB certificate execution format error\n");
return -EIMGERR;
}
}
NOTICE("Image Authentication bypassed at address\n");
return ret;
}
static uint32_t get_unaligned_le32(const void *p)
{
/* TODO: Temp for testing */
//return le32_to_cpup((__le32 *)p);
return 0;
}
static void sha512_transform(uint64_t *state, const uint8_t *input)
{
uint64_t a, b, c, d, e, f, g, h, t1, t2;
int i;
uint64_t W[16];
/* load the state into our registers */
a = state[0]; b = state[1]; c = state[2]; d = state[3];
e = state[4]; f = state[5]; g = state[6]; h = state[7];
/* now iterate */
for (i = 0 ; i < 80; i += 8) {
if (!(i & 8)) {
int j;
if (i < 16) {
/* load the input */
for (j = 0; j < 16; j++)
LOAD_OP(i + j, W, input);
} else {
for (j = 0; j < 16; j++) {
BLEND_OP(i + j, W);
}
}
}
t1 = h + e1(e) + Ch(e, f, g) + sha512_K[i] + W[(i & 15)];
t2 = e0(a) + Maj(a, b, c); d += t1; h = t1 + t2;
t1 = g + e1(d) + Ch(d, e, f) + sha512_K[i+1] + W[(i & 15) + 1];
t2 = e0(h) + Maj(h, a, b); c += t1; g = t1 + t2;
t1 = f + e1(c) + Ch(c, d, e) + sha512_K[i+2] + W[(i & 15) + 2];
t2 = e0(g) + Maj(g, h, a); b += t1; f = t1 + t2;
t1 = e + e1(b) + Ch(b, c, d) + sha512_K[i+3] + W[(i & 15) + 3];
t2 = e0(f) + Maj(f, g, h); a += t1; e = t1 + t2;
t1 = d + e1(a) + Ch(a, b, c) + sha512_K[i+4] + W[(i & 15) + 4];
t2 = e0(e) + Maj(e, f, g); h += t1; d = t1 + t2;
t1 = c + e1(h) + Ch(h, a, b) + sha512_K[i+5] + W[(i & 15) + 5];
t2 = e0(d) + Maj(d, e, f); g += t1; c = t1 + t2;
t1 = b + e1(g) + Ch(g, h, a) + sha512_K[i+6] + W[(i & 15) + 6];
t2 = e0(c) + Maj(c, d, e); f += t1; b = t1 + t2;
t1 = a + e1(f) + Ch(f, g, h) + sha512_K[i+7] + W[(i & 15) + 7];
t2 = e0(b) + Maj(b, c, d); e += t1; a = t1 + t2;
}
state[0] += a; state[1] += b; state[2] += c; state[3] += d;
state[4] += e; state[5] += f; state[6] += g; state[7] += h;
/* erase our data */
a = b = c = d = e = f = g = h = t1 = t2 = 0;
}
static void sha512_block_fn(sha512_context *sst, const uint8_t *src,
int blocks)
{
while (blocks--) {
sha512_transform(sst->state, src);
src += SHA512_BLOCK_SIZE;
}
}
static void sha512_base_do_finalize(sha512_context *sctx)
{
const int bit_offset = SHA512_BLOCK_SIZE - sizeof(uint64_t[2]);
uint64_t *bits = (uint64_t *)(sctx->buf + bit_offset);
unsigned int partial = sctx->count[0] % SHA512_BLOCK_SIZE;
sctx->buf[partial++] = 0x80;
if (partial > bit_offset) {
memset(sctx->buf + partial, 0x0, SHA512_BLOCK_SIZE - partial);
partial = 0;
sha512_block_fn(sctx, sctx->buf, 1);
}
memset(sctx->buf + partial, 0x0, bit_offset - partial);
//fixme bits[0] = cpu_to_be64(sctx->count[1] << 3 | sctx->count[0] >> 61);
//fixme bits[1] = cpu_to_be64(sctx->count[0] << 3);
bits[0] = (sctx->count[1] << 3 | sctx->count[0] >> 61);
bits[1] = (sctx->count[0] << 3);
sha512_block_fn(sctx, sctx->buf, 1);
}
static void sha512_base_do_update(sha512_context *sctx,
const uint8_t *data,
unsigned int len)
{
unsigned int partial = sctx->count[0] % SHA512_BLOCK_SIZE;
sctx->count[0] += len;
if (sctx->count[0] < len)
sctx->count[1]++;
if (((partial + len) >= SHA512_BLOCK_SIZE)) {
int blocks;
if (partial) {
int p = SHA512_BLOCK_SIZE - partial;
memcpy(sctx->buf + partial, data, p);
data += p;
len -= p;
sha512_block_fn(sctx, sctx->buf, 1);
}
blocks = len / SHA512_BLOCK_SIZE;
len %= SHA512_BLOCK_SIZE;
if (blocks) {
sha512_block_fn(sctx, data, blocks);
data += blocks * SHA512_BLOCK_SIZE;
}
partial = 0;
}
if (len)
memcpy(sctx->buf + partial, data, len);
}
void sha384_starts(sha512_context *ctx)
{
ctx->state[0] = SHA384_H0;
ctx->state[1] = SHA384_H1;
ctx->state[2] = SHA384_H2;
ctx->state[3] = SHA384_H3;
ctx->state[4] = SHA384_H4;
ctx->state[5] = SHA384_H5;
ctx->state[6] = SHA384_H6;
ctx->state[7] = SHA384_H7;
ctx->count[0] = ctx->count[1] = 0;
}
void sha384_update(sha512_context *ctx, const uint8_t *input, uint32_t length)
{
sha512_base_do_update(ctx, input, length);
}
void sha384_finish(sha512_context *ctx, uint8_t digest[SHA384_SUM_LEN])
{
int i;
sha512_base_do_finalize(ctx);
for (i = 0; i < SHA384_SUM_LEN / sizeof(uint64_t); i++)
PUT_UINT64_BE(ctx->state[i], digest, i * 8);
}
void sha384_csum_wd(const unsigned char *input, unsigned int ilen,
unsigned char *output, unsigned int chunk_sz)
{
sha512_context ctx;
// #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
// const unsigned char *end;
// unsigned char *curr;
// int chunk;
// #endif
sha384_starts(&ctx);
// #if defined(CONFIG_HW_WATCHDOG) || defined(CONFIG_WATCHDOG)
// curr = (unsigned char *)input;
// end = input + ilen;
// while (curr < end) {
// chunk = end - curr;
// if (chunk > chunk_sz)
// chunk = chunk_sz;
// sha384_update(&ctx, curr, chunk);
// curr += chunk;
// schedule();
// }
// #else
sha384_update(&ctx, input, ilen);
// #endif
sha384_finish(&ctx, output);
}