mirror of
https://github.com/Telecominfraproject/OpenCellular.git
synced 2025-11-23 17:55:01 +00:00
e166d04e79
This is the first of several CLs adding a more memory- and code-efficient firmware verification library. This CL adds the crypto library (modified from firmware/lib/cryptolib) and unit tests for it. BUG=chromium:370082 BRANCH=none TEST=make clean && VBOOT2=1 COV=1 make Change-Id: I4240eab227bb197cacc6c8e7a6397127d74414a2 Signed-off-by: Randall Spangler <rspangler@chromium.org> Reviewed-on: https://chromium-review.googlesource.com/199578 Reviewed-by: Bill Richardson <wfrichar@chromium.org>
160 lines
3.8 KiB
C
160 lines
3.8 KiB
C
/* Copyright (c) 2014 The Chromium OS Authors. All rights reserved.
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#ifndef VBOOT_REFERENCE_2SHA_H_
|
|
#define VBOOT_REFERENCE_2SHA_H_
|
|
|
|
/* Hash algorithms may be disabled individually to save code space */
|
|
|
|
#ifndef VB2_SUPPORT_SHA1
|
|
#define VB2_SUPPORT_SHA1 1
|
|
#endif
|
|
|
|
#ifndef VB2_SUPPORT_SHA256
|
|
#define VB2_SUPPORT_SHA256 1
|
|
#endif
|
|
|
|
#ifndef VB2_SUPPORT_SHA512
|
|
#define VB2_SUPPORT_SHA512 1
|
|
#endif
|
|
|
|
#define VB2_SHA1_DIGEST_SIZE 20
|
|
#define VB2_SHA1_BLOCK_SIZE 64
|
|
|
|
/* Context structs for hash algorithms */
|
|
|
|
struct vb2_sha1_context {
|
|
uint32_t count;
|
|
uint32_t state[5];
|
|
#if defined(HAVE_ENDIAN_H) && defined(HAVE_LITTLE_ENDIAN)
|
|
union {
|
|
uint8_t b[VB2_SHA1_BLOCK_SIZE];
|
|
uint32_t w[VB2_SHA1_BLOCK_SIZE / sizeof(uint32_t)];
|
|
} buf;
|
|
#else
|
|
uint8_t buf[VB2_SHA1_BLOCK_SIZE];
|
|
#endif
|
|
};
|
|
|
|
#define VB2_SHA256_DIGEST_SIZE 32
|
|
#define VB2_SHA256_BLOCK_SIZE 64
|
|
|
|
struct vb2_sha256_context {
|
|
uint32_t h[8];
|
|
uint32_t total_size;
|
|
uint32_t size;
|
|
uint8_t block[2 * VB2_SHA256_BLOCK_SIZE];
|
|
};
|
|
|
|
#define VB2_SHA512_DIGEST_SIZE 64
|
|
#define VB2_SHA512_BLOCK_SIZE 128
|
|
|
|
struct vb2_sha512_context {
|
|
uint64_t h[8];
|
|
uint32_t total_size;
|
|
uint32_t size;
|
|
uint8_t block[2 * VB2_SHA512_BLOCK_SIZE];
|
|
};
|
|
|
|
/* Hash algorithm independent digest context; includes all of the above. */
|
|
struct vb2_digest_context {
|
|
/* Context union for all algorithms */
|
|
union {
|
|
#if VB2_SUPPORT_SHA1
|
|
struct vb2_sha1_context sha1;
|
|
#endif
|
|
#if VB2_SUPPORT_SHA256
|
|
struct vb2_sha256_context sha256;
|
|
#endif
|
|
#if VB2_SUPPORT_SHA512
|
|
struct vb2_sha512_context sha512;
|
|
#endif
|
|
};
|
|
|
|
/* Current hash algorithms */
|
|
uint32_t algorithm;
|
|
};
|
|
|
|
/**
|
|
* Initialize a hash context.
|
|
*
|
|
* @param ctx Hash context
|
|
*/
|
|
void vb2_sha1_init(struct vb2_sha1_context *ctx);
|
|
void vb2_sha256_init(struct vb2_sha256_context *ctx);
|
|
void vb2_sha512_init(struct vb2_sha512_context *ctx);
|
|
|
|
/**
|
|
* Update (extend) a hash.
|
|
*
|
|
* @param ctx Hash context
|
|
* @param data Data to hash
|
|
* @param size Length of data in bytes
|
|
*/
|
|
void vb2_sha1_update(struct vb2_sha1_context *ctx,
|
|
const uint8_t *data,
|
|
uint32_t size);
|
|
void vb2_sha256_update(struct vb2_sha256_context *ctx,
|
|
const uint8_t *data,
|
|
uint32_t size);
|
|
void vb2_sha512_update(struct vb2_sha512_context *ctx,
|
|
const uint8_t *data,
|
|
uint32_t size);
|
|
|
|
/**
|
|
* Finalize a hash digest.
|
|
*
|
|
* @param ctx Hash context
|
|
* @param digest Destination for hash; must be VB_SHA*_DIGEST_SIZE bytes
|
|
*/
|
|
void vb2_sha1_finalize(struct vb2_sha1_context *ctx, uint8_t *digest);
|
|
void vb2_sha256_finalize(struct vb2_sha256_context *ctx, uint8_t *digest);
|
|
void vb2_sha512_finalize(struct vb2_sha512_context *ctx, uint8_t *digest);
|
|
|
|
/**
|
|
* Return the size of the digest for a key algorithm.
|
|
*
|
|
* @param algorithm Key algorithm
|
|
* @return The size of the digest, or 0 if error.
|
|
*/
|
|
int vb2_digest_size(uint32_t algorithm);
|
|
|
|
/**
|
|
* Initialize a digest context for doing block-style digesting.
|
|
*
|
|
* @param dc Digest context
|
|
* @param algorithm Key algorithm
|
|
* @return VB2_SUCCESS, or non-zero on error.
|
|
*/
|
|
int vb2_digest_init(struct vb2_digest_context *dc, uint32_t algorithm);
|
|
|
|
/**
|
|
* Extend a digest's hash with another block of data.
|
|
*
|
|
* @param dc Digest context
|
|
* @param buf Data to hash
|
|
* @param size Length of data in bytes
|
|
* @return VB2_SUCCESS, or non-zero on error.
|
|
*/
|
|
int vb2_digest_extend(struct vb2_digest_context *dc,
|
|
const uint8_t *buf,
|
|
uint32_t size);
|
|
|
|
/**
|
|
* Finalize a digest and store the result.
|
|
*
|
|
* The destination digest should be at least vb2_digest_size(algorithm).
|
|
*
|
|
* @param dc Digest context
|
|
* @param digest Destination for digest
|
|
* @param digest_size Length of digest buffer in bytes.
|
|
* @return VB2_SUCCESS, or non-zero on error.
|
|
*/
|
|
int vb2_digest_finalize(struct vb2_digest_context *dc,
|
|
uint8_t *digest,
|
|
uint32_t digest_size);
|
|
|
|
#endif /* VBOOT_REFERENCE_2SHA_H_ */
|