1 // Copyright 2022 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 #include "walkthrough/test_util.h" 18 19 #include <memory> 20 21 #include "absl/strings/match.h" 22 #include "walkthrough/load_cleartext_keyset.h" 23 #include "tink/keyset_handle.h" 24 25 namespace tink_walkthrough { 26 27 using ::crypto::tink::Aead; 28 using ::crypto::tink::KeysetHandle; 29 using ::crypto::tink::util::Status; 30 using ::crypto::tink::util::StatusOr; 31 DoesSupport(absl::string_view key_uri) const32bool FakeKmsClient::DoesSupport(absl::string_view key_uri) const { 33 return absl::StartsWith(key_uri, "fake://"); 34 } 35 GetAead(absl::string_view key_uri) const36StatusOr<std::unique_ptr<Aead>> FakeKmsClient::GetAead( 37 absl::string_view key_uri) const { 38 StatusOr<std::unique_ptr<KeysetHandle>> master_key_keyset = 39 LoadKeyset(serialized_master_key_keyset_); 40 if (!master_key_keyset.ok()) return master_key_keyset.status(); 41 return (*master_key_keyset)->GetPrimitive<Aead>(); 42 } 43 DoesSupport(absl::string_view key_uri) const44bool AlwaysFailingFakeKmsClient::DoesSupport(absl::string_view key_uri) const { 45 return absl::StartsWith(key_uri, "failing://"); 46 } 47 GetAead(absl::string_view key_uri) const48StatusOr<std::unique_ptr<Aead>> AlwaysFailingFakeKmsClient::GetAead( 49 absl::string_view key_uri) const { 50 return Status(absl::StatusCode::kUnimplemented, "Unimplemented"); 51 } 52 53 } // namespace tink_walkthrough 54