| // Copyright 2017 Google Inc. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| // |
| /////////////////////////////////////////////////////////////////////////////// |
| |
| #ifndef TINK_AEAD_AEAD_SET_WRAPPER_H_ |
| #define TINK_AEAD_AEAD_SET_WRAPPER_H_ |
| |
| #include "absl/strings/string_view.h" |
| #include "tink/aead.h" |
| #include "tink/primitive_set.h" |
| #include "tink/util/statusor.h" |
| #include "proto/tink.pb.h" |
| |
| namespace crypto { |
| namespace tink { |
| |
| // Wraps a set of Aead-instances that correspond to a keyset, |
| // and combines them into a single Aead-primitive, that uses the provided |
| // instances, depending on the context: |
| // * Aead::Encrypt(...) uses the primary instance from the set |
| // * Aead::Decrypt(...) uses the instance that matches the ciphertext prefix. |
| class AeadSetWrapper : public Aead { |
| public: |
| // Returns an Aead-primitive that uses Aead-instances provided in 'aead_set', |
| // which must be non-NULL and must contain a primary instance. |
| static crypto::tink::util::StatusOr<std::unique_ptr<Aead>> NewAead( |
| std::unique_ptr<PrimitiveSet<Aead>> aead_set); |
| |
| crypto::tink::util::StatusOr<std::string> Encrypt( |
| absl::string_view plaintext, |
| absl::string_view associated_data) const override; |
| |
| crypto::tink::util::StatusOr<std::string> Decrypt( |
| absl::string_view ciphertext, |
| absl::string_view associated_data) const override; |
| |
| virtual ~AeadSetWrapper() {} |
| |
| private: |
| std::unique_ptr<PrimitiveSet<Aead>> aead_set_; |
| |
| AeadSetWrapper(std::unique_ptr<PrimitiveSet<Aead>> aead_set) |
| : aead_set_(std::move(aead_set)) {} |
| }; |
| |
| } // namespace tink |
| } // namespace crypto |
| |
| #endif // TINK_AEAD_AEAD_SET_WRAPPER_H_ |