1 // Copyright 2018 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "quiche/quic/core/crypto/quic_hkdf.h"
6
7 #include <string>
8
9 #include "absl/base/macros.h"
10 #include "absl/strings/escaping.h"
11 #include "quiche/quic/platform/api/quic_test.h"
12
13 namespace quic {
14 namespace test {
15 namespace {
16
17 struct HKDFInput {
18 const char* key_hex;
19 const char* salt_hex;
20 const char* info_hex;
21 const char* output_hex;
22 };
23
24 // These test cases are taken from
25 // https://tools.ietf.org/html/rfc5869#appendix-A.
26 static const HKDFInput kHKDFInputs[] = {
27 {
28 "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b",
29 "000102030405060708090a0b0c",
30 "f0f1f2f3f4f5f6f7f8f9",
31 "3cb25f25faacd57a90434f64d0362f2a2d2d0a90cf1a5a4c5db02d56ecc4c5bf340072"
32 "08d5"
33 "b887185865",
34 },
35 {
36 "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122"
37 "2324"
38 "25262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f4041424344454647"
39 "4849"
40 "4a4b4c4d4e4f",
41 "606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182"
42 "8384"
43 "85868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7"
44 "a8a9"
45 "aaabacadaeaf",
46 "b0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2"
47 "d3d4"
48 "d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5f6f7"
49 "f8f9"
50 "fafbfcfdfeff",
51 "b11e398dc80327a1c8e7f78c596a49344f012eda2d4efad8a050cc4c19afa97c59045a"
52 "99ca"
53 "c7827271cb41c65e590e09da3275600c2f09b8367793a9aca3db71cc30c58179ec3e87"
54 "c14c"
55 "01d5c1f3434f1d87",
56 },
57 {
58 "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b",
59 "",
60 "",
61 "8da4e775a563c18f715f802a063c5a31b8a11f5c5ee1879ec3454e5f3c738d2d9d2013"
62 "95fa"
63 "a4b61a96c8",
64 },
65 };
66
67 class QuicHKDFTest : public QuicTest {};
68
TEST_F(QuicHKDFTest,HKDF)69 TEST_F(QuicHKDFTest, HKDF) {
70 for (size_t i = 0; i < ABSL_ARRAYSIZE(kHKDFInputs); i++) {
71 const HKDFInput& test(kHKDFInputs[i]);
72 SCOPED_TRACE(i);
73
74 std::string key;
75 std::string salt;
76 std::string info;
77 std::string expected;
78 ASSERT_TRUE(absl::HexStringToBytes(test.key_hex, &key));
79 ASSERT_TRUE(absl::HexStringToBytes(test.salt_hex, &salt));
80 ASSERT_TRUE(absl::HexStringToBytes(test.info_hex, &info));
81 ASSERT_TRUE(absl::HexStringToBytes(test.output_hex, &expected));
82
83 // We set the key_length to the length of the expected output and then take
84 // the result from the first key, which is the client write key.
85 QuicHKDF hkdf(key, salt, info, expected.size(), 0, 0);
86
87 ASSERT_EQ(expected.size(), hkdf.client_write_key().size());
88 EXPECT_EQ(0, memcmp(expected.data(), hkdf.client_write_key().data(),
89 expected.size()));
90 }
91 }
92
93 } // namespace
94 } // namespace test
95 } // namespace quic
96