1919
2020#include " absl/base/attributes.h"
2121#include " absl/base/call_once.h"
22- #include " absl/log/absl_check.h"
23- #include " absl/status/status.h"
2422#include " absl/status/statusor.h"
2523#include " absl/strings/string_view.h"
2624#include " dpf/status_macros.h"
2725#include " pir/testing/data/embedded_private_key.h"
2826#include " pir/testing/data/embedded_public_key.h"
2927#include " tink/cleartext_keyset_handle.h"
30- #include " tink/hybrid/hybrid_config.h"
28+ #include " tink/hybrid/config_v0.h"
29+ #include " tink/hybrid_decrypt.h"
30+ #include " tink/hybrid_encrypt.h"
3131#include " tink/json/json_keyset_reader.h"
3232#include " tink/keyset_handle.h"
3333#include " tink/keyset_reader.h"
@@ -45,10 +45,6 @@ using ::crypto::tink::KeysetReader;
4545absl::once_flag register_tink_once ABSL_ATTRIBUTE_UNUSED;
4646
4747absl::StatusOr<std::unique_ptr<HybridDecrypt>> CreateFakeHybridDecrypt () {
48- absl::call_once (register_tink_once, []() {
49- ABSL_CHECK_OK (crypto::tink::HybridConfig::Register ());
50- });
51-
5248 const auto * const toc = embedded_private_key_create ();
5349 absl::string_view private_key_json (toc->data , toc->size );
5450
@@ -59,14 +55,11 @@ absl::StatusOr<std::unique_ptr<HybridDecrypt>> CreateFakeHybridDecrypt() {
5955 std::unique_ptr<KeysetHandle> private_key_handle,
6056 CleartextKeysetHandle::Read (std::move (private_key_reader)));
6157
62- return private_key_handle->GetPrimitive <HybridDecrypt>();
58+ return private_key_handle->GetPrimitive <HybridDecrypt>(
59+ crypto::tink::ConfigHybridV0 ());
6360}
6461
6562absl::StatusOr<std::unique_ptr<HybridEncrypt>> CreateFakeHybridEncrypt () {
66- absl::call_once (register_tink_once, []() {
67- ABSL_CHECK_OK (crypto::tink::HybridConfig::Register ());
68- });
69-
7063 const auto * const toc = embedded_public_key_create ();
7164 absl::string_view public_key_json (toc->data , toc->size );
7265
@@ -77,7 +70,8 @@ absl::StatusOr<std::unique_ptr<HybridEncrypt>> CreateFakeHybridEncrypt() {
7770 std::unique_ptr<KeysetHandle> public_key_handle,
7871 CleartextKeysetHandle::Read (std::move (public_key_reader)));
7972
80- return public_key_handle->GetPrimitive <HybridEncrypt>();
73+ return public_key_handle->GetPrimitive <HybridEncrypt>(
74+ crypto::tink::ConfigHybridV0 ());
8175}
8276
8377} // namespace pir_testing
0 commit comments