blob: 3d3f092f71f350daa055d91da87dccd48aafbe0c
| 1 | /* |
| 2 | * Copyright 2015 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | #include "hkdf.h" |
| 18 | |
| 19 | #include <gtest/gtest.h> |
| 20 | #include <string.h> |
| 21 | |
| 22 | #include "android_keymaster_test_utils.h" |
| 23 | |
| 24 | using std::string; |
| 25 | |
| 26 | namespace keymaster { |
| 27 | namespace test { |
| 28 | |
| 29 | struct HkdfTest { |
| 30 | const char* key_hex; |
| 31 | const char* salt_hex; |
| 32 | const char* info_hex; |
| 33 | const char* output_hex; |
| 34 | }; |
| 35 | |
| 36 | // These test cases are taken from |
| 37 | // https://tools.ietf.org/html/rfc5869#appendix-A. |
| 38 | static const HkdfTest kHkdfTests[] = { |
| 39 | { |
| 40 | "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b", "000102030405060708090a0b0c", |
| 41 | "f0f1f2f3f4f5f6f7f8f9", |
| 42 | "3cb25f25faacd57a90434f64d0362f2a2d2d0a90cf1a5a4c5db02d56ecc4c5bf34007208d5b887185865", |
| 43 | }, |
| 44 | { |
| 45 | "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c" |
| 46 | "2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f", |
| 47 | "606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c" |
| 48 | "8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeaf", |
| 49 | "b0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdc" |
| 50 | "dddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3f4f5f6f7f8f9fafbfcfdfeff", |
| 51 | "b11e398dc80327a1c8e7f78c596a49344f012eda2d4efad8a050cc4c19afa97c59045a99cac7827271cb41c65e" |
| 52 | "590e09da3275600c2f09b8367793a9aca3db71cc30c58179ec3e87c14c01d5c1f3434f1d87", |
| 53 | }, |
| 54 | { |
| 55 | "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b", "", "", |
| 56 | "8da4e775a563c18f715f802a063c5a31b8a11f5c5ee1879ec3454e5f3c738d2d9d201395faa4b61a96c8", |
| 57 | }, |
| 58 | }; |
| 59 | |
| 60 | TEST(HkdfTest, Hkdf) { |
| 61 | for (auto& test : kHkdfTests) { |
| 62 | const string key = hex2str(test.key_hex); |
| 63 | const string salt = hex2str(test.salt_hex); |
| 64 | const string info = hex2str(test.info_hex); |
| 65 | const string expected = hex2str(test.output_hex); |
| 66 | size_t output_len = expected.size(); |
| 67 | uint8_t output[output_len]; |
| 68 | Rfc5869Sha256Kdf hkdf; |
| 69 | ASSERT_TRUE(hkdf.Init(reinterpret_cast<const uint8_t*>(key.data()), key.size(), |
| 70 | reinterpret_cast<const uint8_t*>(salt.data()), salt.size())); |
| 71 | ASSERT_TRUE(hkdf.GenerateKey(reinterpret_cast<const uint8_t*>(info.data()), info.size(), |
| 72 | output, output_len)); |
| 73 | EXPECT_EQ(0, memcmp(output, expected.data(), output_len)); |
| 74 | } |
| 75 | } |
| 76 | |
| 77 | } // namespace test |
| 78 | } // namespace keymaster |
| 79 |