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 49 /* Block size in bytes of a SHA-256 digest. */ 50 #define AVB_SHA256_BLOCK_SIZE 64 51 52 53 /* Block size in bytes of a SHA-512 digest. */ 54 #define AVB_SHA512_BLOCK_SIZE 128 55 56 /* Data structure used for SHA-256. */ 57 typedef struct { 58 uint32_t h[8]; 59 uint64_t tot_len; 60 size_t len; 61 uint8_t block[2 * AVB_SHA256_BLOCK_SIZE]; 62 uint8_t buf[AVB_SHA256_DIGEST_SIZE]; /* Used for storing the final digest. */ 63 #ifdef CONFIG_DM_CRYPTO 64 struct udevice *crypto_dev; 65 sha_context crypto_ctx; 66 #endif 67 } AvbSHA256Ctx; 68 69 /* Data structure used for SHA-512. */ 70 typedef struct { 71 uint64_t h[8]; 72 uint64_t tot_len; 73 size_t len; 74 uint8_t block[2 * AVB_SHA512_BLOCK_SIZE]; 75 uint8_t buf[AVB_SHA512_DIGEST_SIZE]; /* Used for storing the final digest. */ 76 #ifdef CONFIG_DM_CRYPTO 77 struct udevice *crypto_dev; 78 sha_context crypto_ctx; 79 #endif 80 } AvbSHA512Ctx; 81 82 /* Initializes the SHA-256 context. */ 83 void avb_sha256_init(AvbSHA256Ctx* ctx); 84 85 /* Updates the SHA-256 context with |len| bytes from |data|. */ 86 void avb_sha256_update(AvbSHA256Ctx* ctx, const uint8_t* data, size_t len); 87 88 /* Returns the SHA-256 digest. */ 89 uint8_t* avb_sha256_final(AvbSHA256Ctx* ctx) AVB_ATTR_WARN_UNUSED_RESULT; 90 91 /* Initializes the SHA-512 context. */ 92 void avb_sha512_init(AvbSHA512Ctx* ctx); 93 94 /* Updates the SHA-512 context with |len| bytes from |data|. */ 95 void avb_sha512_update(AvbSHA512Ctx* ctx, const uint8_t* data, size_t len); 96 97 /* Returns the SHA-512 digest. */ 98 uint8_t* avb_sha512_final(AvbSHA512Ctx* ctx) AVB_ATTR_WARN_UNUSED_RESULT; 99 100 #ifdef __cplusplus 101 } 102 #endif 103 104 #endif /* AVB_SHA_H_ */ 105