1 /* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Permission is hereby granted, free of charge, to any person 5 * obtaining a copy of this software and associated documentation 6 * files (the "Software"), to deal in the Software without 7 * restriction, including without limitation the rights to use, copy, 8 * modify, merge, publish, distribute, sublicense, and/or sell copies 9 * of the Software, and to permit persons to whom the Software is 10 * furnished to do so, subject to the following conditions: 11 * 12 * The above copyright notice and this permission notice shall be 13 * included in all copies or substantial portions of the Software. 14 * 15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 16 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 17 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 18 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS 19 * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN 20 * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN 21 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 22 * SOFTWARE. 23 */ 24 25 /* 26 #ifdef AVB_INSIDE_LIBAVB_H 27 #error "You can't include avb_sha.h in the public header libavb.h." 28 #endif 29 30 #ifndef AVB_COMPILATION 31 #error "Never include this file, it may only be used from internal avb code." 32 #endif 33 */ 34 35 #ifndef AVB_SHA_H_ 36 #define AVB_SHA_H_ 37 38 #ifdef __cplusplus 39 extern "C" { 40 #endif 41 42 #ifdef CONFIG_DM_CRYPTO 43 #include <crypto.h> 44 #endif 45 #include <android_avb/avb_crypto.h> 46 #include <android_avb/avb_sysdeps.h> 47 #include <dm/device.h> 48 #include <u-boot/sha256.h> 49 50 /* Block size in bytes of a SHA-256 digest. */ 51 #define AVB_SHA256_BLOCK_SIZE 64 52 53 54 /* Block size in bytes of a SHA-512 digest. */ 55 #define AVB_SHA512_BLOCK_SIZE 128 56 57 /* Data structure used for SHA-256. */ 58 typedef struct { 59 uint32_t h[8]; 60 uint64_t tot_len; 61 size_t len; 62 uint8_t block[2 * AVB_SHA256_BLOCK_SIZE]; 63 uint8_t buf[AVB_SHA256_DIGEST_SIZE]; /* Used for storing the final digest. */ 64 sha256_context sha256ctx; 65 } AvbSHA256Ctx; 66 67 /* Data structure used for SHA-512. */ 68 typedef struct { 69 uint64_t h[8]; 70 uint64_t tot_len; 71 size_t len; 72 uint8_t block[2 * AVB_SHA512_BLOCK_SIZE]; 73 uint8_t buf[AVB_SHA512_DIGEST_SIZE]; /* Used for storing the final digest. */ 74 #ifdef CONFIG_DM_CRYPTO 75 struct udevice *crypto_dev; 76 sha_context crypto_ctx; 77 #endif 78 } AvbSHA512Ctx; 79 80 /* Initializes the SHA-256 context. */ 81 void avb_sha256_init(AvbSHA256Ctx* ctx); 82 83 /* Updates the SHA-256 context with |len| bytes from |data|. */ 84 void avb_sha256_update(AvbSHA256Ctx* ctx, const uint8_t* data, size_t len); 85 86 /* Returns the SHA-256 digest. */ 87 uint8_t* avb_sha256_final(AvbSHA256Ctx* ctx) AVB_ATTR_WARN_UNUSED_RESULT; 88 89 /* Initializes the SHA-512 context. */ 90 void avb_sha512_init(AvbSHA512Ctx* ctx); 91 92 /* Updates the SHA-512 context with |len| bytes from |data|. */ 93 void avb_sha512_update(AvbSHA512Ctx* ctx, const uint8_t* data, size_t len); 94 95 /* Returns the SHA-512 digest. */ 96 uint8_t* avb_sha512_final(AvbSHA512Ctx* ctx) AVB_ATTR_WARN_UNUSED_RESULT; 97 98 #ifdef __cplusplus 99 } 100 #endif 101 102 #endif /* AVB_SHA_H_ */ 103