1/* Copyright (c) 2011 The Chromium OS Authors. All rights reserved. 2 * Use of this source code is governed by a BSD-style license that can be 3 * found in the LICENSE file. 4 */ 5 6/* SHA-1, 256 and 512 functions. */ 7 8#ifndef VBOOT_REFERENCE_SHA_H_ 9#define VBOOT_REFERENCE_SHA_H_ 10 11#ifndef VBOOT_REFERENCE_CRYPTOLIB_H_ 12#error "Do not include this file directly. Use cryptolib.h instead." 13#endif 14 15#include "sysincludes.h" 16 17#define SHA1_DIGEST_SIZE 20 18#define SHA1_BLOCK_SIZE 64 19 20#define SHA256_DIGEST_SIZE 32 21#define SHA256_BLOCK_SIZE 64 22 23#define SHA512_DIGEST_SIZE 64 24#define SHA512_BLOCK_SIZE 128 25 26typedef struct SHA1_CTX { 27 uint64_t count; 28 uint32_t state[5]; 29#if defined(HAVE_ENDIAN_H) && defined(HAVE_LITTLE_ENDIAN) 30 union { 31 uint8_t b[64]; 32 uint32_t w[16]; 33 } buf; 34#else 35 uint8_t buf[64]; 36#endif 37} SHA1_CTX; 38 39typedef struct { 40 uint32_t h[8]; 41 uint32_t tot_len; 42 uint32_t len; 43 uint8_t block[2 * SHA256_BLOCK_SIZE]; 44 uint8_t buf[SHA256_DIGEST_SIZE]; /* Used for storing the final digest. */ 45} VB_SHA256_CTX; 46 47typedef struct { 48 uint64_t h[8]; 49 uint32_t tot_len; 50 uint32_t len; 51 uint8_t block[2 * SHA512_BLOCK_SIZE]; 52 uint8_t buf[SHA512_DIGEST_SIZE]; /* Used for storing the final digest. */ 53} VB_SHA512_CTX; 54 55 56void SHA1_init(SHA1_CTX* ctx); 57void SHA1_update(SHA1_CTX* ctx, const uint8_t* data, uint64_t len); 58uint8_t* SHA1_final(SHA1_CTX* ctx); 59 60void SHA256_init(VB_SHA256_CTX* ctx); 61void SHA256_update(VB_SHA256_CTX* ctx, const uint8_t* data, uint32_t len); 62uint8_t* SHA256_final(VB_SHA256_CTX* ctx); 63 64void SHA512_init(VB_SHA512_CTX* ctx); 65void SHA512_update(VB_SHA512_CTX* ctx, const uint8_t* data, uint32_t len); 66uint8_t* SHA512_final(VB_SHA512_CTX* ctx); 67 68/* Convenience function for SHA-1. Computes hash on [data] of length [len]. 69 * and stores it into [digest]. [digest] should be pre-allocated to 70 * SHA1_DIGEST_SIZE bytes. 71 */ 72uint8_t* internal_SHA1(const uint8_t* data, uint64_t len, uint8_t* digest); 73 74/* Convenience function for SHA-256. Computes hash on [data] of length [len]. 75 * and stores it into [digest]. [digest] should be pre-allocated to 76 * SHA256_DIGEST_SIZE bytes. 77 */ 78uint8_t* internal_SHA256(const uint8_t* data, uint64_t len, uint8_t* digest); 79 80/* Convenience function for SHA-512. Computes hash on [data] of length [len]. 81 * and stores it into [digest]. [digest] should be pre-allocated to 82 * SHA512_DIGEST_SIZE bytes. 83 */ 84uint8_t* internal_SHA512(const uint8_t* data, uint64_t len, uint8_t* digest); 85 86 87/*---- Utility functions/wrappers for message digests. */ 88 89#define SHA1_DIGEST_ALGORITHM 0 90#define SHA256_DIGEST_ALGORITHM 1 91#define SHA512_DIGEST_ALGORITHM 2 92 93/* A generic digest context structure which can be used to represent 94 * the SHA*_CTX for multiple digest algorithms. 95 */ 96typedef struct DigestContext { 97 SHA1_CTX* sha1_ctx; 98 VB_SHA256_CTX* sha256_ctx; 99 VB_SHA512_CTX* sha512_ctx; 100 int algorithm; /* Hashing algorithm to use. */ 101} DigestContext; 102 103/* Wrappers for message digest algorithms. These are useful when the hashing 104 * operation is being done in parallel with something else. DigestContext tracks 105 * and stores the state of any digest algorithm (one at any given time). 106 */ 107 108/* Initialize a digest context for use with signature algorithm [algorithm]. */ 109void DigestInit(DigestContext* ctx, int sig_algorithm); 110void DigestUpdate(DigestContext* ctx, const uint8_t* data, uint32_t len); 111 112/* Caller owns the returned digest and must free it. */ 113uint8_t* DigestFinal(DigestContext* ctx); 114 115/* Returns the appropriate digest for the data in [input_file] 116 * based on the signature [algorithm]. 117 * Caller owns the returned digest and must free it. 118 */ 119uint8_t* DigestFile(char* input_file, int sig_algorithm); 120 121/* Returns the appropriate digest of [buf] of length 122 * [len] based on the signature [algorithm]. 123 * Caller owns the returned digest and must free it. 124 */ 125uint8_t* DigestBuf(const uint8_t* buf, uint64_t len, int sig_algorithm); 126 127 128#endif /* VBOOT_REFERENCE_SHA_H_ */ 129