119 lines
3.1 KiB
C
119 lines
3.1 KiB
C
/* Copyright (c) 2011 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.
|
|
*
|
|
* Utility functions for message digest functions.
|
|
*/
|
|
|
|
#include "sysincludes.h"
|
|
|
|
#include "cryptolib.h"
|
|
#include "utility.h"
|
|
#include "vboot_api.h"
|
|
|
|
void DigestInit(DigestContext* ctx, int sig_algorithm) {
|
|
ctx->algorithm = hash_type_map[sig_algorithm];
|
|
switch(ctx->algorithm) {
|
|
#ifndef CHROMEOS_EC
|
|
case SHA1_DIGEST_ALGORITHM:
|
|
ctx->sha1_ctx = (SHA1_CTX*) VbExMalloc(sizeof(SHA1_CTX));
|
|
SHA1_init(ctx->sha1_ctx);
|
|
break;
|
|
#endif
|
|
case SHA256_DIGEST_ALGORITHM:
|
|
ctx->sha256_ctx = (VB_SHA256_CTX*) VbExMalloc(sizeof(VB_SHA256_CTX));
|
|
SHA256_init(ctx->sha256_ctx);
|
|
break;
|
|
#ifndef CHROMEOS_EC
|
|
case SHA512_DIGEST_ALGORITHM:
|
|
ctx->sha512_ctx = (VB_SHA512_CTX*) VbExMalloc(sizeof(VB_SHA512_CTX));
|
|
SHA512_init(ctx->sha512_ctx);
|
|
break;
|
|
#endif
|
|
};
|
|
}
|
|
|
|
void DigestUpdate(DigestContext* ctx, const uint8_t* data, uint32_t len) {
|
|
switch(ctx->algorithm) {
|
|
#ifndef CHROMEOS_EC
|
|
case SHA1_DIGEST_ALGORITHM:
|
|
SHA1_update(ctx->sha1_ctx, data, len);
|
|
break;
|
|
#endif
|
|
case SHA256_DIGEST_ALGORITHM:
|
|
SHA256_update(ctx->sha256_ctx, data, len);
|
|
break;
|
|
#ifndef CHROMEOS_EC
|
|
case SHA512_DIGEST_ALGORITHM:
|
|
SHA512_update(ctx->sha512_ctx, data, len);
|
|
break;
|
|
#endif
|
|
};
|
|
}
|
|
|
|
uint8_t* DigestFinal(DigestContext* ctx) {
|
|
uint8_t* digest = NULL;
|
|
switch(ctx->algorithm) {
|
|
#ifndef CHROMEOS_EC
|
|
case SHA1_DIGEST_ALGORITHM:
|
|
digest = (uint8_t*) VbExMalloc(SHA1_DIGEST_SIZE);
|
|
Memcpy(digest, SHA1_final(ctx->sha1_ctx), SHA1_DIGEST_SIZE);
|
|
VbExFree(ctx->sha1_ctx);
|
|
break;
|
|
#endif
|
|
case SHA256_DIGEST_ALGORITHM:
|
|
digest = (uint8_t*) VbExMalloc(SHA256_DIGEST_SIZE);
|
|
Memcpy(digest, SHA256_final(ctx->sha256_ctx), SHA256_DIGEST_SIZE);
|
|
VbExFree(ctx->sha256_ctx);
|
|
break;
|
|
#ifndef CHROMEOS_EC
|
|
case SHA512_DIGEST_ALGORITHM:
|
|
digest = (uint8_t*) VbExMalloc(SHA512_DIGEST_SIZE);
|
|
Memcpy(digest, SHA512_final(ctx->sha512_ctx), SHA512_DIGEST_SIZE);
|
|
VbExFree(ctx->sha512_ctx);
|
|
break;
|
|
#endif
|
|
};
|
|
return digest;
|
|
}
|
|
|
|
uint8_t* DigestBuf(const uint8_t* buf, uint64_t len, int sig_algorithm) {
|
|
/* Allocate enough space for the largest digest */
|
|
uint8_t* digest = (uint8_t*) VbExMalloc(SHA512_DIGEST_SIZE);
|
|
/* Define an array mapping [sig_algorithm] to function pointers to the
|
|
* SHA{1|256|512} functions.
|
|
*/
|
|
typedef uint8_t* (*Hash_ptr) (const uint8_t*, uint64_t, uint8_t*);
|
|
Hash_ptr hash[] = {
|
|
#ifdef CHROMEOS_EC
|
|
0, /* RSA 1024 */
|
|
0,
|
|
0,
|
|
0, /* RSA 2048 */
|
|
0,
|
|
0,
|
|
0, /* RSA 4096 */
|
|
internal_SHA256,
|
|
0,
|
|
0, /* RSA 8192 */
|
|
0,
|
|
0,
|
|
#else
|
|
internal_SHA1, /* RSA 1024 */
|
|
internal_SHA256,
|
|
internal_SHA512,
|
|
internal_SHA1, /* RSA 2048 */
|
|
internal_SHA256,
|
|
internal_SHA512,
|
|
internal_SHA1, /* RSA 4096 */
|
|
internal_SHA256,
|
|
internal_SHA512,
|
|
internal_SHA1, /* RSA 8192 */
|
|
internal_SHA256,
|
|
internal_SHA512,
|
|
#endif
|
|
};
|
|
/* Call the appropriate hash function. */
|
|
return hash[sig_algorithm](buf, len, digest);
|
|
}
|