1 // Copyright 2021 Google LLC 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 // 15 /////////////////////////////////////////////////////////////////////////////// 16 17 #ifndef TINK_AEAD_INTERNAL_ZERO_COPY_AEAD_WRAPPER_H_ 18 #define TINK_AEAD_INTERNAL_ZERO_COPY_AEAD_WRAPPER_H_ 19 20 #include <memory> 21 22 #include "tink/aead.h" 23 #include "tink/aead/internal/zero_copy_aead.h" 24 #include "tink/primitive_set.h" 25 #include "tink/primitive_wrapper.h" 26 #include "tink/util/statusor.h" 27 28 namespace crypto { 29 namespace tink { 30 namespace internal { 31 32 // Wraps a set of ZeroCopyAead-instances that correspond to a keyset and 33 // combines them into a single Aead-primitive that uses the provided instances, 34 // depending on the context: 35 // * ZeroCopyAeadSetWrapper::Encrypt(...) uses the primary instance from the 36 // set. 37 // * ZeroCopyAeadSetWrapper::Decrypt(...) uses the instance that matches the 38 // ciphertext prefix. 39 class ZeroCopyAeadWrapper : public PrimitiveWrapper<ZeroCopyAead, Aead> { 40 public: 41 // Returns an Aead-primitive that uses ZeroCopyAead-instances provided in 42 // 'aead_set', which must be non-NULL and must contain a primary instance. 43 util::StatusOr<std::unique_ptr<Aead>> Wrap( 44 std::unique_ptr<PrimitiveSet<ZeroCopyAead>> aead_set) const override; 45 }; 46 47 } // namespace internal 48 } // namespace tink 49 } // namespace crypto 50 51 #endif // TINK_AEAD_INTERNAL_ZERO_COPY_AEAD_WRAPPER_H_ 52