estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 1 | // Copyright 2015 The Chromium 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 | #ifndef CRYPTO_AEAD_H_ |
| 6 | #define CRYPTO_AEAD_H_ |
| 7 | |
avi | dd373b8 | 2015-12-21 21:34:43 | [diff] [blame] | 8 | #include <stddef.h> |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 9 | #include <stdint.h> |
avi | dd373b8 | 2015-12-21 21:34:43 | [diff] [blame] | 10 | |
davidben | 6004dc5 | 2017-02-03 04:15:29 | [diff] [blame] | 11 | #include <string> |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 12 | #include <vector> |
davidben | 6004dc5 | 2017-02-03 04:15:29 | [diff] [blame] | 13 | |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 14 | #include "base/containers/span.h" |
Keishi Hattori | 0e45c02 | 2021-11-27 09:25:52 | [diff] [blame^] | 15 | #include "base/memory/raw_ptr.h" |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 16 | #include "base/strings/string_piece.h" |
| 17 | #include "crypto/crypto_export.h" |
Anton Bikineev | a3f961db | 2021-05-15 17:56:12 | [diff] [blame] | 18 | #include "third_party/abseil-cpp/absl/types/optional.h" |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 19 | |
| 20 | struct evp_aead_st; |
| 21 | |
| 22 | namespace crypto { |
| 23 | |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 24 | // This class exposes the AES-128-CTR-HMAC-SHA256 and AES_256_GCM AEAD. Note |
| 25 | // that there are two versions of most methods: an historical version based |
| 26 | // around |StringPiece| and a more modern version that takes |base::span|. |
| 27 | // Prefer the latter in new code. |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 28 | class CRYPTO_EXPORT Aead { |
| 29 | public: |
Leonid Baraz | 07c0577 | 2020-08-18 19:55:09 | [diff] [blame] | 30 | enum AeadAlgorithm { |
| 31 | AES_128_CTR_HMAC_SHA256, |
| 32 | AES_256_GCM, |
| 33 | AES_256_GCM_SIV, |
| 34 | CHACHA20_POLY1305 |
| 35 | }; |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 36 | |
| 37 | explicit Aead(AeadAlgorithm algorithm); |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 38 | Aead(const Aead&) = delete; |
| 39 | Aead& operator=(const Aead&) = delete; |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 40 | ~Aead(); |
| 41 | |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 42 | // Note that Init keeps a reference to the data pointed to by |key| thus that |
| 43 | // data must outlive this object. |
| 44 | void Init(base::span<const uint8_t> key); |
| 45 | |
| 46 | // Note that Init keeps a reference to the data pointed to by |key| thus that |
| 47 | // data must outlive this object. |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 48 | void Init(const std::string* key); |
| 49 | |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 50 | std::vector<uint8_t> Seal(base::span<const uint8_t> plaintext, |
| 51 | base::span<const uint8_t> nonce, |
| 52 | base::span<const uint8_t> additional_data) const; |
| 53 | |
David Benjamin | cda45eb | 2017-11-06 18:16:52 | [diff] [blame] | 54 | bool Seal(base::StringPiece plaintext, |
| 55 | base::StringPiece nonce, |
| 56 | base::StringPiece additional_data, |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 57 | std::string* ciphertext) const; |
| 58 | |
Anton Bikineev | a3f961db | 2021-05-15 17:56:12 | [diff] [blame] | 59 | absl::optional<std::vector<uint8_t>> Open( |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 60 | base::span<const uint8_t> ciphertext, |
| 61 | base::span<const uint8_t> nonce, |
| 62 | base::span<const uint8_t> additional_data) const; |
| 63 | |
David Benjamin | cda45eb | 2017-11-06 18:16:52 | [diff] [blame] | 64 | bool Open(base::StringPiece ciphertext, |
| 65 | base::StringPiece nonce, |
| 66 | base::StringPiece additional_data, |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 67 | std::string* plaintext) const; |
| 68 | |
| 69 | size_t KeyLength() const; |
| 70 | |
| 71 | size_t NonceLength() const; |
| 72 | |
| 73 | private: |
Adam Langley | f214973 | 2019-08-09 22:23:33 | [diff] [blame] | 74 | bool Seal(base::span<const uint8_t> plaintext, |
| 75 | base::span<const uint8_t> nonce, |
| 76 | base::span<const uint8_t> additional_data, |
| 77 | uint8_t* out, |
| 78 | size_t* output_length, |
| 79 | size_t max_output_length) const; |
| 80 | |
| 81 | bool Open(base::span<const uint8_t> ciphertext, |
| 82 | base::span<const uint8_t> nonce, |
| 83 | base::span<const uint8_t> additional_data, |
| 84 | uint8_t* out, |
| 85 | size_t* output_length, |
| 86 | size_t max_output_length) const; |
| 87 | |
Anton Bikineev | a3f961db | 2021-05-15 17:56:12 | [diff] [blame] | 88 | absl::optional<base::span<const uint8_t>> key_; |
Keishi Hattori | 0e45c02 | 2021-11-27 09:25:52 | [diff] [blame^] | 89 | raw_ptr<const evp_aead_st> aead_; |
estark | 03206a1 | 2015-04-25 04:52:25 | [diff] [blame] | 90 | }; |
| 91 | |
| 92 | } // namespace crypto |
| 93 | |
davidben | 6004dc5 | 2017-02-03 04:15:29 | [diff] [blame] | 94 | #endif // CRYPTO_AEAD_H_ |