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 "tink/hybrid/internal/hpke_context_boringssl.h"
18
19 #include <memory>
20 #include <string>
21 #include <utility>
22
23 #include "gtest/gtest.h"
24 #include "absl/status/status.h"
25 #include "tink/hybrid/internal/hpke_test_util.h"
26 #include "tink/hybrid/internal/hpke_util.h"
27 #include "tink/hybrid/internal/test_hpke_context_boringssl.h"
28 #include "tink/util/secret_data.h"
29 #include "tink/util/test_matchers.h"
30
31 namespace crypto {
32 namespace tink {
33 namespace internal {
34 namespace {
35
36 using ::crypto::tink::test::IsOk;
37 using ::crypto::tink::test::IsOkAndHolds;
38 using ::crypto::tink::test::StatusIs;
39 using ::testing::Values;
40
41 class HpkeContextBoringSslTest : public testing::TestWithParam<HpkeParams> {};
42
43 INSTANTIATE_TEST_SUITE_P(
44 HpkeContextBoringSslTestSuite, HpkeContextBoringSslTest,
45 Values(HpkeParams{HpkeKem::kX25519HkdfSha256, HpkeKdf::kHkdfSha256,
46 HpkeAead::kAes128Gcm},
47 HpkeParams{HpkeKem::kX25519HkdfSha256, HpkeKdf::kHkdfSha256,
48 HpkeAead::kChaCha20Poly1305}));
49
TEST_P(HpkeContextBoringSslTest,Seal)50 TEST_P(HpkeContextBoringSslTest, Seal) {
51 HpkeParams hpke_params = GetParam();
52 util::StatusOr<HpkeTestParams> params = CreateHpkeTestParams(hpke_params);
53 ASSERT_THAT(params, IsOk());
54 util::StatusOr<SenderHpkeContextBoringSsl> hpke_context =
55 TestHpkeContextBoringSsl::SetupSender(
56 hpke_params, params->recipient_public_key, params->application_info,
57 params->seed_for_testing);
58 ASSERT_THAT(hpke_context, IsOk());
59 util::StatusOr<std::string> ciphertext =
60 hpke_context->context->Seal(params->plaintext, params->associated_data);
61 ASSERT_THAT(ciphertext, IsOkAndHolds(params->ciphertext));
62 }
63
TEST_P(HpkeContextBoringSslTest,Open)64 TEST_P(HpkeContextBoringSslTest, Open) {
65 HpkeParams hpke_params = GetParam();
66 util::StatusOr<HpkeTestParams> params = CreateHpkeTestParams(hpke_params);
67 ASSERT_THAT(params, IsOk());
68 util::StatusOr<std::unique_ptr<HpkeContextBoringSsl>> hpke_context =
69 HpkeContextBoringSsl::SetupRecipient(
70 hpke_params,
71 util::SecretDataFromStringView(params->recipient_private_key),
72 params->encapsulated_key, params->application_info);
73 ASSERT_THAT(hpke_context, IsOk());
74 util::StatusOr<std::string> plaintext =
75 (*hpke_context)->Open(params->ciphertext, params->associated_data);
76 ASSERT_THAT(plaintext, IsOkAndHolds(params->plaintext));
77 }
78
TEST_P(HpkeContextBoringSslTest,SenderExport)79 TEST_P(HpkeContextBoringSslTest, SenderExport) {
80 HpkeParams hpke_params = GetParam();
81 util::StatusOr<HpkeTestParams> params = CreateHpkeTestParams(hpke_params);
82 ASSERT_THAT(params, IsOk());
83
84 util::StatusOr<SenderHpkeContextBoringSsl> sender_hpke_context =
85 TestHpkeContextBoringSsl::SetupSender(
86 hpke_params, params->recipient_public_key, params->application_info,
87 params->seed_for_testing);
88 ASSERT_THAT(sender_hpke_context, IsOk());
89
90 for (int i = 0; i < params->exported_contexts.size(); ++i) {
91 util::StatusOr<util::SecretData> sender_secret =
92 sender_hpke_context->context->Export(params->exported_contexts[i], 32);
93 ASSERT_THAT(sender_secret, IsOkAndHolds(util::SecretDataFromStringView(
94 params->exported_values[i])));
95 }
96 }
97
TEST_P(HpkeContextBoringSslTest,RecipientExport)98 TEST_P(HpkeContextBoringSslTest, RecipientExport) {
99 HpkeParams hpke_params = GetParam();
100 util::StatusOr<HpkeTestParams> params = CreateHpkeTestParams(hpke_params);
101 ASSERT_THAT(params, IsOk());
102
103 util::StatusOr<std::unique_ptr<HpkeContextBoringSsl>> recipient_hpke_context =
104 HpkeContextBoringSsl::SetupRecipient(
105 hpke_params,
106 util::SecretDataFromStringView(params->recipient_private_key),
107 params->encapsulated_key, params->application_info);
108 ASSERT_THAT(recipient_hpke_context, IsOk());
109
110 for (int i = 0; i < params->exported_contexts.size(); ++i) {
111 util::StatusOr<util::SecretData> recipient_secret =
112 (*recipient_hpke_context)->Export(params->exported_contexts[i], 32);
113 ASSERT_THAT(recipient_secret, IsOkAndHolds(util::SecretDataFromStringView(
114 params->exported_values[i])));
115 }
116 }
117
118 class HpkeContextBoringSslWithBadParamTest
119 : public testing::TestWithParam<HpkeParams> {};
120
121 INSTANTIATE_TEST_SUITE_P(
122 HpkeContextBoringSslWithBadParamTestSuite,
123 HpkeContextBoringSslWithBadParamTest,
124 Values(HpkeParams{HpkeKem::kUnknownKem, HpkeKdf::kHkdfSha256,
125 HpkeAead::kAes128Gcm},
126 HpkeParams{HpkeKem::kX25519HkdfSha256, HpkeKdf::kUnknownKdf,
127 HpkeAead::kAes128Gcm},
128 HpkeParams{HpkeKem::kX25519HkdfSha256, HpkeKdf::kHkdfSha256,
129 HpkeAead::kUnknownAead}));
130
TEST_P(HpkeContextBoringSslWithBadParamTest,BadSenderParamsFail)131 TEST_P(HpkeContextBoringSslWithBadParamTest, BadSenderParamsFail) {
132 HpkeParams hpke_params = GetParam();
133 HpkeTestParams params = DefaultHpkeTestParams();
134 util::StatusOr<SenderHpkeContextBoringSsl> result =
135 TestHpkeContextBoringSsl::SetupSender(
136 hpke_params, params.recipient_public_key, params.application_info,
137 params.seed_for_testing);
138 ASSERT_THAT(result.status(), StatusIs(absl::StatusCode::kInvalidArgument));
139 }
140
TEST_P(HpkeContextBoringSslWithBadParamTest,BadRecipientParamsFail)141 TEST_P(HpkeContextBoringSslWithBadParamTest, BadRecipientParamsFail) {
142 HpkeParams hpke_params = GetParam();
143 HpkeTestParams params = DefaultHpkeTestParams();
144 util::StatusOr<std::unique_ptr<HpkeContextBoringSsl>> hpke_context =
145 HpkeContextBoringSsl::SetupRecipient(
146 hpke_params,
147 util::SecretDataFromStringView(params.recipient_private_key),
148 params.encapsulated_key, params.application_info);
149 ASSERT_THAT(hpke_context.status(),
150 StatusIs(absl::StatusCode::kInvalidArgument));
151 }
152
TEST(HpkeContextBoringSslWithMissingPublicKeyTest,ZeroLengthPublicKeyFails)153 TEST(HpkeContextBoringSslWithMissingPublicKeyTest, ZeroLengthPublicKeyFails) {
154 HpkeParams hpke_params = {HpkeKem::kX25519HkdfSha256, HpkeKdf::kHkdfSha256,
155 HpkeAead::kAes128Gcm};
156 HpkeTestParams params = DefaultHpkeTestParams();
157 util::StatusOr<SenderHpkeContextBoringSsl> result =
158 TestHpkeContextBoringSsl::SetupSender(
159 hpke_params, /*recipient_public_key=*/"", params.application_info,
160 params.seed_for_testing);
161 ASSERT_THAT(result.status(), StatusIs(absl::StatusCode::kUnknown));
162 }
163
TEST(HpkeContextBoringSslWithMissingPrivateKeyTest,ZeroLengthPrivateKeyFails)164 TEST(HpkeContextBoringSslWithMissingPrivateKeyTest, ZeroLengthPrivateKeyFails) {
165 HpkeParams hpke_params = {HpkeKem::kX25519HkdfSha256, HpkeKdf::kHkdfSha256,
166 HpkeAead::kAes128Gcm};
167 HpkeTestParams params = DefaultHpkeTestParams();
168 util::StatusOr<std::unique_ptr<HpkeContextBoringSsl>> hpke_context =
169 HpkeContextBoringSsl::SetupRecipient(
170 hpke_params, util::SecretDataFromStringView(/*secret=*/""),
171 params.encapsulated_key, params.application_info);
172 ASSERT_THAT(hpke_context.status(),
173 StatusIs(absl::StatusCode::kInvalidArgument));
174 }
175
176 } // namespace
177 } // namespace internal
178 } // namespace tink
179 } // namespace crypto
180