Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 2 | /** |
| 3 | * Copyright (c) 2017-2020, Regents of the University of California. |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 4 | * |
| 5 | * This file is part of ndncert, a certificate management system based on NDN. |
| 6 | * |
| 7 | * ndncert is free software: you can redistribute it and/or modify it under the terms |
| 8 | * of the GNU General Public License as published by the Free Software Foundation, either |
| 9 | * version 3 of the License, or (at your option) any later version. |
| 10 | * |
| 11 | * ndncert is distributed in the hope that it will be useful, but WITHOUT ANY |
| 12 | * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A |
| 13 | * PARTICULAR PURPOSE. See the GNU General Public License for more details. |
| 14 | * |
| 15 | * You should have received copies of the GNU General Public License along with |
| 16 | * ndncert, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| 17 | * |
| 18 | * See AUTHORS.md for complete list of ndncert authors and contributors. |
| 19 | */ |
| 20 | |
Zhiyi Zhang | b041d44 | 2020-10-22 21:57:11 -0700 | [diff] [blame] | 21 | #include "detail/crypto-helpers.hpp" |
Zhiyi Zhang | dc25ddf | 2020-10-20 14:28:55 -0700 | [diff] [blame] | 22 | |
| 23 | #include <boost/endian/conversion.hpp> |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 24 | #include <cstring> |
Davide Pesavento | 368341b | 2019-08-13 23:57:50 -0400 | [diff] [blame] | 25 | #include <ndn-cxx/encoding/buffer-stream.hpp> |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 26 | #include <ndn-cxx/security/transform/base64-decode.hpp> |
| 27 | #include <ndn-cxx/security/transform/base64-encode.hpp> |
| 28 | #include <ndn-cxx/security/transform/buffer-source.hpp> |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 29 | #include <ndn-cxx/security/transform/stream-sink.hpp> |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 30 | #include <ndn-cxx/util/random.hpp> |
Zhiyi Zhang | dc25ddf | 2020-10-20 14:28:55 -0700 | [diff] [blame] | 31 | #include <openssl/ec.h> |
| 32 | #include <openssl/err.h> |
Zhiyi Zhang | dc25ddf | 2020-10-20 14:28:55 -0700 | [diff] [blame] | 33 | #include <openssl/hmac.h> |
| 34 | #include <openssl/kdf.h> |
| 35 | #include <openssl/pem.h> |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 36 | |
| 37 | namespace ndn { |
| 38 | namespace ndncert { |
| 39 | |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 40 | ECDHState::ECDHState() |
| 41 | { |
Zhiyi Zhang | f12ee30 | 2020-10-14 17:46:44 -0700 | [diff] [blame] | 42 | auto EC_NID = NID_X9_62_prime256v1; |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 43 | // params context |
| 44 | EVP_PKEY_CTX* ctx_params = EVP_PKEY_CTX_new_id(EVP_PKEY_EC, nullptr); |
| 45 | if (ctx_params == nullptr) { |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 46 | NDN_THROW(std::runtime_error("Could not create context")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 47 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 48 | if (EVP_PKEY_paramgen_init(ctx_params) != 1) { |
| 49 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 50 | NDN_THROW(std::runtime_error("Could not initialize parameter generation")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 51 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 52 | if (1 != EVP_PKEY_CTX_set_ec_paramgen_curve_nid(ctx_params, EC_NID)) { |
| 53 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 54 | NDN_THROW(std::runtime_error("Likely unknown elliptical curve ID specified")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 55 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 56 | // generate params |
| 57 | EVP_PKEY* params = nullptr; |
| 58 | if (!EVP_PKEY_paramgen(ctx_params, ¶ms)) { |
| 59 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 60 | NDN_THROW(std::runtime_error("Could not create parameter object parameters")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 61 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 62 | // key generation context |
Zhiyi Zhang | f908f66 | 2020-10-21 12:55:10 -0700 | [diff] [blame] | 63 | EVP_PKEY_CTX* ctx_keygen = EVP_PKEY_CTX_new(params, nullptr); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 64 | if (ctx_keygen == nullptr) { |
| 65 | EVP_PKEY_free(params); |
| 66 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5c05945 | 2020-10-16 17:43:31 -0700 | [diff] [blame] | 67 | NDN_THROW(std::runtime_error("Could not create the context for the key generation")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 68 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 69 | if (1 != EVP_PKEY_keygen_init(ctx_keygen)) { |
| 70 | EVP_PKEY_CTX_free(ctx_keygen); |
| 71 | EVP_PKEY_free(params); |
| 72 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 73 | NDN_THROW(std::runtime_error("Could not init context for key generation")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 74 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 75 | if (1 != EVP_PKEY_keygen(ctx_keygen, &m_privkey)) { |
| 76 | EVP_PKEY_CTX_free(ctx_keygen); |
| 77 | EVP_PKEY_free(params); |
| 78 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | 5c05945 | 2020-10-16 17:43:31 -0700 | [diff] [blame] | 79 | NDN_THROW(std::runtime_error("Could not generate DHE keys in final step")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 80 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 81 | EVP_PKEY_CTX_free(ctx_keygen); |
| 82 | EVP_PKEY_free(params); |
| 83 | EVP_PKEY_CTX_free(ctx_params); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 84 | } |
| 85 | |
| 86 | ECDHState::~ECDHState() |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 87 | { |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 88 | if (m_privkey != nullptr) { |
| 89 | EVP_PKEY_free(m_privkey); |
| 90 | } |
| 91 | } |
| 92 | |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 93 | const std::vector <uint8_t>& |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 94 | ECDHState::getSelfPubKey() |
| 95 | { |
| 96 | auto privECKey = EVP_PKEY_get1_EC_KEY(m_privkey); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 97 | auto ecPoint = EC_KEY_get0_public_key(privECKey); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 98 | auto group = EC_KEY_get0_group(privECKey); |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 99 | auto requiredBufLen = EC_POINT_point2oct(group, ecPoint, POINT_CONVERSION_UNCOMPRESSED, nullptr, 0, nullptr); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 100 | m_pubKey.resize(requiredBufLen); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 101 | auto resultCode = EC_POINT_point2oct(group, ecPoint, POINT_CONVERSION_UNCOMPRESSED, |
| 102 | m_pubKey.data(), requiredBufLen, nullptr); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 103 | EC_KEY_free(privECKey); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 104 | if (resultCode == 0) { |
| 105 | NDN_THROW(std::runtime_error("Error in getting EC Public Key in the format of octet string")); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 106 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 107 | return m_pubKey; |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 108 | } |
| 109 | |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 110 | const std::vector<uint8_t>& |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 111 | ECDHState::deriveSecret(const std::vector <uint8_t>& peerKey) |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 112 | { |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 113 | // prepare self private key |
| 114 | auto privECKey = EVP_PKEY_get1_EC_KEY(m_privkey); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 115 | auto group = EC_KEY_get0_group(privECKey); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 116 | EC_KEY_free(privECKey); |
| 117 | // prepare the peer public key |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 118 | auto peerPoint = EC_POINT_new(group); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 119 | EC_POINT_oct2point(group, peerPoint, peerKey.data(), peerKey.size(), nullptr); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 120 | EC_KEY* ecPeerkey = EC_KEY_new(); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 121 | EC_KEY_set_group(ecPeerkey, group); |
| 122 | EC_KEY_set_public_key(ecPeerkey, peerPoint); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 123 | EVP_PKEY* evpPeerkey = EVP_PKEY_new(); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 124 | EVP_PKEY_set1_EC_KEY(evpPeerkey, ecPeerkey); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 125 | EC_KEY_free(ecPeerkey); |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 126 | EC_POINT_free(peerPoint); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 127 | // ECDH context |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 128 | EVP_PKEY_CTX* ctx = EVP_PKEY_CTX_new(m_privkey, nullptr); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 129 | // Initialize |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 130 | EVP_PKEY_derive_init(ctx); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 131 | // Provide the peer public key |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 132 | EVP_PKEY_derive_set_peer(ctx, evpPeerkey); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 133 | // Determine buffer length for shared secret |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 134 | size_t secretLen = 0; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 135 | EVP_PKEY_derive(ctx, nullptr, &secretLen); |
Zhiyi Zhang | def00e4 | 2020-10-20 22:41:56 -0700 | [diff] [blame] | 136 | m_secret.resize(secretLen); |
| 137 | // Derive the shared secret |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 138 | auto resultCode = EVP_PKEY_derive(ctx, m_secret.data(), &secretLen); |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 139 | EVP_PKEY_CTX_free(ctx); |
| 140 | EVP_PKEY_free(evpPeerkey); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 141 | if (resultCode == 0) { |
| 142 | NDN_THROW(std::runtime_error("Error when calling ECDH")); |
| 143 | } |
Zhiyi Zhang | bed854c | 2020-10-20 18:25:35 -0700 | [diff] [blame] | 144 | return m_secret; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 145 | |
| 146 | // // prepare self private key |
| 147 | // auto privECKey = EVP_PKEY_get1_EC_KEY(m_privkey); |
| 148 | // if (privECKey == nullptr) { |
| 149 | // NDN_THROW(std::runtime_error("Cannot not get key when calling EVP_PKEY_get1_EC_KEY()")); |
| 150 | // } |
| 151 | // auto group = EC_KEY_get0_group(privECKey); |
| 152 | // EC_KEY_free(privECKey); |
| 153 | // // prepare the peer public key |
| 154 | // auto peerPoint = EC_POINT_new(group); |
| 155 | // if (peerPoint == nullptr) { |
| 156 | // NDN_THROW(std::runtime_error("Cannot create the EC_POINT for peer key when calling EC_POINT_new()")); |
| 157 | // } |
| 158 | // if (EC_POINT_oct2point(group, peerPoint, peerKey.data(), peerKey.size(), nullptr) == 0) { |
| 159 | // EC_POINT_free(peerPoint); |
| 160 | // NDN_THROW(std::runtime_error("Cannot convert peer's key into a EC point when calling EC_POINT_oct2point()")); |
| 161 | // } |
| 162 | // EC_KEY* ecPeerkey = EC_KEY_new(); |
| 163 | // if (ecPeerkey == nullptr) { |
| 164 | // EC_POINT_free(peerPoint); |
| 165 | // NDN_THROW(std::runtime_error("Cannot create EC_KEY for peer key when calling EC_KEY_new()")); |
| 166 | // } |
| 167 | // if (EC_KEY_set_group(ecPeerkey, group) != 1) { |
| 168 | // EC_POINT_free(peerPoint); |
| 169 | // NDN_THROW(std::runtime_error("Cannot set group for peer key's EC_KEY when calling EC_KEY_set_group()")); |
| 170 | // } |
| 171 | // if (EC_KEY_set_public_key(ecPeerkey, peerPoint) == 0) { |
| 172 | // EC_KEY_free(ecPeerkey); |
| 173 | // EC_POINT_free(peerPoint); |
| 174 | // NDN_THROW( |
| 175 | // std::runtime_error("Cannot initialize peer EC_KEY with the EC_POINT when calling EC_KEY_set_public_key()")); |
| 176 | // } |
| 177 | // EVP_PKEY* evpPeerkey = EVP_PKEY_new(); |
| 178 | // if (EVP_PKEY_set1_EC_KEY(evpPeerkey, ecPeerkey) == 0) { |
| 179 | // EC_KEY_free(ecPeerkey); |
| 180 | // EC_POINT_free(peerPoint); |
| 181 | // NDN_THROW(std::runtime_error("Cannot create EVP_PKEY for peer key when calling EVP_PKEY_new()")); |
| 182 | // } |
| 183 | // EC_KEY_free(ecPeerkey); |
| 184 | // EC_POINT_free(peerPoint); |
| 185 | // // ECDH context |
| 186 | // EVP_PKEY_CTX* ctx = EVP_PKEY_CTX_new(m_privkey, nullptr); |
| 187 | // if (ctx == nullptr) { |
| 188 | // EVP_PKEY_free(evpPeerkey); |
| 189 | // NDN_THROW(std::runtime_error("Cannot create context for ECDH when calling EVP_PKEY_CTX_new()")); |
| 190 | // } |
| 191 | // // Initialize |
| 192 | // if (1 != EVP_PKEY_derive_init(ctx)) { |
| 193 | // EVP_PKEY_CTX_free(ctx); |
| 194 | // EVP_PKEY_free(evpPeerkey); |
| 195 | // NDN_THROW(std::runtime_error("Cannot initialize context for ECDH when calling EVP_PKEY_derive_init()")); |
| 196 | // } |
| 197 | // // Provide the peer public key |
| 198 | // if (1 != EVP_PKEY_derive_set_peer(ctx, evpPeerkey)) { |
| 199 | // EVP_PKEY_CTX_free(ctx); |
| 200 | // EVP_PKEY_free(evpPeerkey); |
| 201 | // NDN_THROW(std::runtime_error("Cannot set peer key for ECDH when calling EVP_PKEY_derive_set_peer()")); |
| 202 | // } |
| 203 | // // Determine buffer length for shared secret |
| 204 | // size_t secretLen = 0; |
| 205 | // if (1 != EVP_PKEY_derive(ctx, nullptr, &secretLen)) { |
| 206 | // EVP_PKEY_CTX_free(ctx); |
| 207 | // EVP_PKEY_free(evpPeerkey); |
| 208 | // NDN_THROW(std::runtime_error("Cannot determine the needed buffer length when calling EVP_PKEY_derive()")); |
| 209 | // } |
| 210 | // m_secret.resize(secretLen); |
| 211 | // // Derive the shared secret |
| 212 | // if (1 != (EVP_PKEY_derive(ctx, m_secret.data(), &secretLen))) { |
| 213 | // EVP_PKEY_CTX_free(ctx); |
| 214 | // EVP_PKEY_free(evpPeerkey); |
| 215 | // NDN_THROW(std::runtime_error("Cannot derive ECDH secret when calling EVP_PKEY_derive()")); |
| 216 | // } |
| 217 | // EVP_PKEY_CTX_free(ctx); |
| 218 | // EVP_PKEY_free(evpPeerkey); |
| 219 | // return m_secret; |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 220 | } |
| 221 | |
Zhiyi Zhang | f12ee30 | 2020-10-14 17:46:44 -0700 | [diff] [blame] | 222 | void |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 223 | hmacSha256(const uint8_t* data, size_t dataLen, |
Zhiyi Zhang | 199508a | 2020-10-21 10:45:50 -0700 | [diff] [blame] | 224 | const uint8_t* key, size_t keyLen, |
| 225 | uint8_t* result) |
Davide Pesavento | 368341b | 2019-08-13 23:57:50 -0400 | [diff] [blame] | 226 | { |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 227 | auto ret = HMAC(EVP_sha256(), key, keyLen, data, dataLen, result, nullptr); |
Zhiyi Zhang | f12ee30 | 2020-10-14 17:46:44 -0700 | [diff] [blame] | 228 | if (ret == nullptr) { |
Zhiyi Zhang | 5c05945 | 2020-10-16 17:43:31 -0700 | [diff] [blame] | 229 | NDN_THROW(std::runtime_error("Error computing HMAC when calling HMAC()")); |
Zhiyi Zhang | f12ee30 | 2020-10-14 17:46:44 -0700 | [diff] [blame] | 230 | } |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 231 | } |
| 232 | |
Zhiyi Zhang | 159ba63 | 2020-10-20 15:09:12 -0700 | [diff] [blame] | 233 | size_t |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 234 | hkdf(const uint8_t* secret, size_t secretLen, const uint8_t* salt, |
| 235 | size_t saltLen, uint8_t* output, size_t outputLen, |
| 236 | const uint8_t* info, size_t infoLen) |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 237 | { |
Zhiyi Zhang | f908f66 | 2020-10-21 12:55:10 -0700 | [diff] [blame] | 238 | EVP_PKEY_CTX* pctx = EVP_PKEY_CTX_new_id(EVP_PKEY_HKDF, nullptr); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 239 | EVP_PKEY_derive_init(pctx); |
| 240 | EVP_PKEY_CTX_set_hkdf_md(pctx, EVP_sha256()); |
| 241 | EVP_PKEY_CTX_set1_hkdf_salt(pctx, salt, saltLen); |
| 242 | EVP_PKEY_CTX_set1_hkdf_key(pctx, secret, secretLen); |
| 243 | EVP_PKEY_CTX_add1_hkdf_info(pctx, info, infoLen); |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 244 | size_t outLen = outputLen; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 245 | auto resultCode = EVP_PKEY_derive(pctx, output, &outLen); |
Zhiyi Zhang | 222810b | 2020-10-16 21:50:35 -0700 | [diff] [blame] | 246 | EVP_PKEY_CTX_free(pctx); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 247 | if (resultCode == 0) { |
| 248 | NDN_THROW(std::runtime_error("Error when calling HKDF")); |
| 249 | } |
Zhiyi Zhang | 159ba63 | 2020-10-20 15:09:12 -0700 | [diff] [blame] | 250 | return outLen; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 251 | |
| 252 | // if (EVP_PKEY_derive_init(pctx) <= 0) { |
| 253 | // EVP_PKEY_CTX_free(pctx); |
| 254 | // NDN_THROW(std::runtime_error("HKDF: Cannot init ctx when calling EVP_PKEY_derive_init()")); |
| 255 | // } |
| 256 | // if (EVP_PKEY_CTX_set_hkdf_md(pctx, EVP_sha256()) <= 0) { |
| 257 | // EVP_PKEY_CTX_free(pctx); |
| 258 | // NDN_THROW(std::runtime_error("HKDF: Cannot set md when calling EVP_PKEY_CTX_set_hkdf_md()")); |
| 259 | // } |
| 260 | // if (EVP_PKEY_CTX_set1_hkdf_salt(pctx, salt, saltLen) <= 0) { |
| 261 | // EVP_PKEY_CTX_free(pctx); |
| 262 | // NDN_THROW(std::runtime_error("HKDF: Cannot set salt when calling EVP_PKEY_CTX_set1_hkdf_salt()")); |
| 263 | // } |
| 264 | // if (EVP_PKEY_CTX_set1_hkdf_key(pctx, secret, secretLen) <= 0) { |
| 265 | // EVP_PKEY_CTX_free(pctx); |
| 266 | // NDN_THROW(std::runtime_error("HKDF: Cannot set secret when calling EVP_PKEY_CTX_set1_hkdf_key()")); |
| 267 | // } |
| 268 | // if (EVP_PKEY_CTX_add1_hkdf_info(pctx, info, infoLen) <= 0) { |
| 269 | // EVP_PKEY_CTX_free(pctx); |
| 270 | // NDN_THROW(std::runtime_error("HKDF: Cannot set info when calling EVP_PKEY_CTX_add1_hkdf_info()")); |
| 271 | // } |
| 272 | // size_t outLen = outputLen; |
| 273 | // if (EVP_PKEY_derive(pctx, output, &outLen) <= 0) { |
| 274 | // EVP_PKEY_CTX_free(pctx); |
| 275 | // NDN_THROW(std::runtime_error("HKDF: Cannot derive result when calling EVP_PKEY_derive()")); |
| 276 | // } |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 277 | } |
| 278 | |
Zhiyi Zhang | e5a07c9 | 2020-10-21 10:51:37 -0700 | [diff] [blame] | 279 | size_t |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 280 | aesGcm128Encrypt(const uint8_t* plaintext, size_t plaintextLen, const uint8_t* associated, size_t associatedLen, |
Zhiyi Zhang | f908f66 | 2020-10-21 12:55:10 -0700 | [diff] [blame] | 281 | const uint8_t* key, const uint8_t* iv, uint8_t* ciphertext, uint8_t* tag) |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 282 | { |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 283 | int len = 0; |
| 284 | size_t ciphertextLen = 0; |
| 285 | EVP_CIPHER_CTX* ctx = EVP_CIPHER_CTX_new(); |
| 286 | EVP_EncryptInit_ex(ctx, EVP_aes_128_gcm(), nullptr, nullptr, nullptr); |
| 287 | EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, 12, nullptr); |
| 288 | EVP_EncryptInit_ex(ctx, nullptr, nullptr, key, iv); |
| 289 | EVP_EncryptUpdate(ctx, nullptr, &len, associated, associatedLen); |
| 290 | EVP_EncryptUpdate(ctx, ciphertext, &len, plaintext, plaintextLen); |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 291 | ciphertextLen = len; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 292 | EVP_EncryptFinal_ex(ctx, ciphertext + len, &len); |
| 293 | auto resultCode = EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_GET_TAG, 16, tag); |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 294 | EVP_CIPHER_CTX_free(ctx); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 295 | if (resultCode == 0) { |
| 296 | NDN_THROW(std::runtime_error("Error in encryption plaintext with AES GCM")); |
| 297 | } |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 298 | return ciphertextLen; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 299 | |
| 300 | // if (!(ctx = EVP_CIPHER_CTX_new())) { |
| 301 | // NDN_THROW(std::runtime_error("Cannot create and initialise the context when calling EVP_CIPHER_CTX_new()")); |
| 302 | // } |
| 303 | // if (1 != EVP_EncryptInit_ex(ctx, EVP_aes_128_gcm(), nullptr, nullptr, nullptr)) { |
| 304 | // EVP_CIPHER_CTX_free(ctx); |
| 305 | // NDN_THROW(std::runtime_error("Cannot initialize the encryption operation when calling EVP_EncryptInit_ex()")); |
| 306 | // } |
| 307 | // if (1 != EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, 12, nullptr)) { |
| 308 | // EVP_CIPHER_CTX_free(ctx); |
| 309 | // NDN_THROW(std::runtime_error("Cannot set IV length when calling EVP_CIPHER_CTX_ctrl()")); |
| 310 | // } |
| 311 | // if (1 != EVP_EncryptInit_ex(ctx, nullptr, nullptr, key, iv)) { |
| 312 | // EVP_CIPHER_CTX_free(ctx); |
| 313 | // NDN_THROW(std::runtime_error("Cannot initialize key and IV when calling EVP_EncryptInit_ex()")); |
| 314 | // } |
| 315 | // if (1 != EVP_EncryptUpdate(ctx, nullptr, &len, associated, associatedLen)) { |
| 316 | // EVP_CIPHER_CTX_free(ctx); |
| 317 | // NDN_THROW(std::runtime_error("Cannot set associated authentication data when calling EVP_EncryptUpdate()")); |
| 318 | // } |
| 319 | // if (1 != EVP_EncryptUpdate(ctx, ciphertext, &len, plaintext, plaintextLen)) { |
| 320 | // EVP_CIPHER_CTX_free(ctx); |
| 321 | // NDN_THROW(std::runtime_error("Cannot encrypt when calling EVP_EncryptUpdate()")); |
| 322 | // } |
| 323 | // ciphertextLen = len; |
| 324 | // if (1 != EVP_EncryptFinal_ex(ctx, ciphertext + len, &len)) { |
| 325 | // EVP_CIPHER_CTX_free(ctx); |
| 326 | // NDN_THROW(std::runtime_error("Cannot finalise the encryption when calling EVP_EncryptFinal_ex()")); |
| 327 | // } |
| 328 | // ciphertextLen += len; |
| 329 | // if (1 != EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_GET_TAG, 16, tag)) { |
| 330 | // EVP_CIPHER_CTX_free(ctx); |
| 331 | // NDN_THROW(std::runtime_error("Cannot get tag when calling EVP_CIPHER_CTX_ctrl()")); |
| 332 | // } |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 333 | } |
| 334 | |
Zhiyi Zhang | e5a07c9 | 2020-10-21 10:51:37 -0700 | [diff] [blame] | 335 | size_t |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 336 | aesGcm128Decrypt(const uint8_t* ciphertext, size_t ciphertextLen, const uint8_t* associated, size_t associatedLen, |
Zhiyi Zhang | f908f66 | 2020-10-21 12:55:10 -0700 | [diff] [blame] | 337 | const uint8_t* tag, const uint8_t* key, const uint8_t* iv, uint8_t* plaintext) |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 338 | { |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 339 | int len = 0; |
| 340 | size_t plaintextLen = 0; |
| 341 | EVP_CIPHER_CTX* ctx = EVP_CIPHER_CTX_new(); |
| 342 | EVP_DecryptInit_ex(ctx, EVP_aes_128_gcm(), nullptr, nullptr, nullptr); |
| 343 | EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, 12, nullptr); |
| 344 | EVP_DecryptInit_ex(ctx, nullptr, nullptr, key, iv); |
| 345 | EVP_DecryptUpdate(ctx, nullptr, &len, associated, associatedLen); |
| 346 | EVP_DecryptUpdate(ctx, plaintext, &len, ciphertext, ciphertextLen); |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 347 | plaintextLen = len; |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 348 | EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_TAG, 16, const_cast<void*>(reinterpret_cast<const void*>(tag))); |
| 349 | auto resultCode = EVP_DecryptFinal_ex(ctx, plaintext + len, &len); |
| 350 | plaintextLen += len; |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 351 | EVP_CIPHER_CTX_free(ctx); |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 352 | if (resultCode == 0) { |
| 353 | NDN_THROW(std::runtime_error("Error in decrypting ciphertext with AES GCM")); |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 354 | } |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 355 | return plaintextLen; |
| 356 | |
| 357 | // if (!(ctx = EVP_CIPHER_CTX_new())) { |
| 358 | // NDN_THROW(std::runtime_error("Cannot create and initialise the context when calling EVP_CIPHER_CTX_new()")); |
| 359 | // } |
| 360 | // if (!EVP_DecryptInit_ex(ctx, EVP_aes_128_gcm(), nullptr, nullptr, nullptr)) { |
| 361 | // EVP_CIPHER_CTX_free(ctx); |
| 362 | // NDN_THROW(std::runtime_error("Cannot initialise the decryption operation when calling EVP_DecryptInit_ex()")); |
| 363 | // } |
| 364 | // if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_IVLEN, 12, nullptr)) { |
| 365 | // EVP_CIPHER_CTX_free(ctx); |
| 366 | // NDN_THROW(std::runtime_error("Cannot set IV length when calling EVP_CIPHER_CTX_ctrl")); |
| 367 | // } |
| 368 | // if (!EVP_DecryptInit_ex(ctx, nullptr, nullptr, key, iv)) { |
| 369 | // EVP_CIPHER_CTX_free(ctx); |
| 370 | // NDN_THROW(std::runtime_error("Cannot initialise key and IV when calling EVP_DecryptInit_ex()")); |
| 371 | // } |
| 372 | // if (!EVP_DecryptUpdate(ctx, nullptr, &len, associated, associatedLen)) { |
| 373 | // EVP_CIPHER_CTX_free(ctx); |
| 374 | // NDN_THROW(std::runtime_error("Cannot set associated authentication data when calling EVP_EncryptUpdate()")); |
| 375 | // } |
| 376 | // if (!EVP_DecryptUpdate(ctx, plaintext, &len, ciphertext, ciphertextLen)) { |
| 377 | // EVP_CIPHER_CTX_free(ctx); |
| 378 | // NDN_THROW(std::runtime_error("Cannot decrypt when calling EVP_DecryptUpdate()")); |
| 379 | // } |
| 380 | // plaintextLen = len; |
| 381 | // if (!EVP_CIPHER_CTX_ctrl(ctx, EVP_CTRL_GCM_SET_TAG, 16, const_cast<void*>(reinterpret_cast<const void*>(tag)))) { |
| 382 | // EVP_CIPHER_CTX_free(ctx); |
| 383 | // NDN_THROW(std::runtime_error("Cannot set tag value when calling EVP_CIPHER_CTX_ctrl()")); |
| 384 | // } |
Zhiyi Zhang | a67fa46 | 2020-04-19 13:48:03 -0700 | [diff] [blame] | 385 | } |
| 386 | |
Zhiyi Zhang | 886a226 | 2020-12-24 15:19:33 -0800 | [diff] [blame] | 387 | // Can be removed after boost version 1.72, replaced by boost::endian::load_big_u32 |
| 388 | static uint32_t |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 389 | loadBigU32(const std::vector <uint8_t>& iv, size_t pos) |
Zhiyi Zhang | 886a226 | 2020-12-24 15:19:33 -0800 | [diff] [blame] | 390 | { |
| 391 | uint32_t result = iv[pos] << 24 | iv[pos + 1] << 16 | iv[pos + 2] << 8 | iv[pos + 3]; |
| 392 | return result; |
| 393 | } |
| 394 | |
| 395 | // Can be removed after boost version 1.72, replaced by boost::endian::store_big_u32 |
| 396 | static void |
| 397 | storeBigU32(uint8_t* iv, uint32_t counter) |
| 398 | { |
| 399 | uint32_t temp = boost::endian::native_to_big(counter); |
| 400 | std::memcpy(iv, reinterpret_cast<const uint8_t*>(&temp), 4); |
| 401 | return; |
| 402 | } |
| 403 | |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 404 | static void |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 405 | updateIv(std::vector <uint8_t>& iv, size_t payloadSize) |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 406 | { |
Zhiyi Zhang | 886a226 | 2020-12-24 15:19:33 -0800 | [diff] [blame] | 407 | // uint32_t counter = boost::endian::load_big_u32(&iv[8]); |
| 408 | uint32_t counter = loadBigU32(iv, 8); |
Zhiyi Zhang | 199508a | 2020-10-21 10:45:50 -0700 | [diff] [blame] | 409 | uint32_t increment = (payloadSize + 15) / 16; |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 410 | if (std::numeric_limits<uint32_t>::max() - counter <= increment) { |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 411 | NDN_THROW(std::runtime_error("Error incrementing the AES block counter: " |
Zhiyi Zhang | e11a2d6 | 2020-10-22 09:59:12 -0700 | [diff] [blame] | 412 | "too many blocks have been encrypted for the same request instance")); |
Zhiyi Zhang | 159ba63 | 2020-10-20 15:09:12 -0700 | [diff] [blame] | 413 | } |
| 414 | else { |
| 415 | counter += increment; |
| 416 | } |
Zhiyi Zhang | 886a226 | 2020-12-24 15:19:33 -0800 | [diff] [blame] | 417 | // boost::endian::store_big_u32(&iv[8], counter); |
| 418 | storeBigU32(&iv[8], counter); |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 419 | } |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 420 | |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 421 | Block |
Zhiyi Zhang | c1e9815 | 2020-12-21 16:15:39 -0800 | [diff] [blame] | 422 | encodeBlockWithAesGcm128(uint32_t tlvType, const uint8_t* key, |
| 423 | const uint8_t* payload, size_t payloadSize, |
| 424 | const uint8_t* associatedData, size_t associatedDataSize, |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 425 | std::vector <uint8_t>& encryptionIv) |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 426 | { |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 427 | // The spec of AES encrypted payload TLV used in NDNCERT: |
| 428 | // https://github.com/named-data/ndncert/wiki/NDNCERT-Protocol-0.3#242-aes-gcm-encryption |
Zhiyi Zhang | 222810b | 2020-10-16 21:50:35 -0700 | [diff] [blame] | 429 | Buffer encryptedPayload(payloadSize); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 430 | uint8_t tag[16]; |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 431 | if (encryptionIv.empty()) { |
| 432 | encryptionIv.resize(12, 0); |
| 433 | random::generateSecureBytes(encryptionIv.data(), 8); |
| 434 | } |
Zhiyi Zhang | 11cf7eb | 2020-10-20 15:43:36 -0700 | [diff] [blame] | 435 | size_t encryptedPayloadLen = aesGcm128Encrypt(payload, payloadSize, associatedData, associatedDataSize, |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 436 | key, encryptionIv.data(), encryptedPayload.data(), tag); |
tylerliu | 1f480be | 2020-11-10 13:02:53 -0800 | [diff] [blame] | 437 | Block content(tlvType); |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 438 | content.push_back(makeBinaryBlock(tlv::InitializationVector, encryptionIv.data(), encryptionIv.size())); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 439 | content.push_back(makeBinaryBlock(tlv::AuthenticationTag, tag, 16)); |
Zhiyi Zhang | 222810b | 2020-10-16 21:50:35 -0700 | [diff] [blame] | 440 | content.push_back(makeBinaryBlock(tlv::EncryptedPayload, encryptedPayload.data(), encryptedPayloadLen)); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 441 | content.encode(); |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 442 | // update IV's counter |
| 443 | updateIv(encryptionIv, payloadSize); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 444 | return content; |
| 445 | } |
| 446 | |
| 447 | Buffer |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 448 | decodeBlockWithAesGcm128(const Block& block, const uint8_t* key, |
| 449 | const uint8_t* associatedData, size_t associatedDataSize, |
Zhiyi Zhang | 1e5d677 | 2021-02-18 17:35:46 -0800 | [diff] [blame^] | 450 | std::vector <uint8_t>& decryptionIv, const std::vector <uint8_t>& encryptionIv) |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 451 | { |
Zhiyi Zhang | 5e74ecd | 2020-10-21 12:52:25 -0700 | [diff] [blame] | 452 | // The spec of AES encrypted payload TLV used in NDNCERT: |
| 453 | // https://github.com/named-data/ndncert/wiki/NDNCERT-Protocol-0.3#242-aes-gcm-encryption |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 454 | block.parse(); |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 455 | const auto& encryptedPayloadBlock = block.get(tlv::EncryptedPayload); |
| 456 | Buffer result(encryptedPayloadBlock.value_size()); |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 457 | if (block.get(tlv::InitializationVector).value_size() != 12 || |
| 458 | block.get(tlv::AuthenticationTag).value_size() != 16) { |
Zhiyi Zhang | de58e60 | 2021-02-17 23:09:02 -0800 | [diff] [blame] | 459 | NDN_THROW(std::runtime_error("Error when decrypting the AES Encrypted Block: " |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 460 | "The observed IV or Authentication Tag is of an unexpected size.")); |
Zhiyi Zhang | de58e60 | 2021-02-17 23:09:02 -0800 | [diff] [blame] | 461 | } |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 462 | std::vector <uint8_t> observedDecryptionIv(block.get(tlv::InitializationVector).value(), |
| 463 | block.get(tlv::InitializationVector).value() + 12); |
| 464 | if (!encryptionIv.empty()) { |
| 465 | if (std::equal(observedDecryptionIv.begin(), observedDecryptionIv.begin() + 8, encryptionIv.begin())) { |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 466 | NDN_THROW(std::runtime_error("Error when decrypting the AES Encrypted Block: " |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 467 | "The observed IV's the random component should be different from ours.")); |
Zhiyi Zhang | de58e60 | 2021-02-17 23:09:02 -0800 | [diff] [blame] | 468 | } |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 469 | } |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 470 | if (!decryptionIv.empty()) { |
| 471 | if (loadBigU32(observedDecryptionIv, 8) < loadBigU32(decryptionIv, 8) || |
| 472 | !std::equal(observedDecryptionIv.begin(), observedDecryptionIv.begin() + 8, decryptionIv.begin())) { |
| 473 | NDN_THROW(std::runtime_error("Error when decrypting the AES Encrypted Block: " |
| 474 | "The observed IV's counter should be monotonically increasing " |
| 475 | "and the random component must be the same from the requester.")); |
| 476 | } |
| 477 | } |
| 478 | decryptionIv = observedDecryptionIv; |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 479 | auto resultLen = aesGcm128Decrypt(encryptedPayloadBlock.value(), encryptedPayloadBlock.value_size(), |
Zhiyi Zhang | f908f66 | 2020-10-21 12:55:10 -0700 | [diff] [blame] | 480 | associatedData, associatedDataSize, block.get(tlv::AuthenticationTag).value(), |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 481 | key, decryptionIv.data(), result.data()); |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 482 | if (resultLen != encryptedPayloadBlock.value_size()) { |
Zhiyi Zhang | aeab497 | 2020-10-22 22:20:40 -0700 | [diff] [blame] | 483 | NDN_THROW(std::runtime_error("Error when decrypting the AES Encrypted Block: " |
Zhiyi Zhang | 3b9a503 | 2021-02-18 11:15:09 -0800 | [diff] [blame] | 484 | "Decrypted payload is of an unexpected size.")); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 485 | } |
Zhiyi Zhang | 4f1c010 | 2020-12-21 15:08:09 -0800 | [diff] [blame] | 486 | updateIv(decryptionIv, resultLen); |
Zhiyi Zhang | c5d93a9 | 2020-10-14 17:07:35 -0700 | [diff] [blame] | 487 | return result; |
Zhiyi Zhang | af7c290 | 2019-03-14 22:13:21 -0700 | [diff] [blame] | 488 | } |
| 489 | |
Zhiyi Zhang | e4891b7 | 2020-10-10 15:11:57 -0700 | [diff] [blame] | 490 | } // namespace ndncert |
| 491 | } // namespace ndn |