Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2017 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 <gtest/gtest.h> |
| 18 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 19 | #include <endian.h> |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 20 | #include <hidl/HidlSupport.h> |
| 21 | #include <keymaster/logger.h> |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 22 | |
Shawn Willden | bb22a6c | 2017-12-06 19:35:28 -0700 | [diff] [blame^] | 23 | #include <keystore/keymaster_types.h> |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 24 | #include <keystore/keystore_hidl_support.h> |
| 25 | |
Shawn Willden | bb22a6c | 2017-12-06 19:35:28 -0700 | [diff] [blame^] | 26 | #include "../auth_token_table.h" |
| 27 | |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 28 | using std::vector; |
| 29 | |
| 30 | namespace keystore { |
Shawn Willden | bb22a6c | 2017-12-06 19:35:28 -0700 | [diff] [blame^] | 31 | |
| 32 | using android::hardware::hidl_array; |
| 33 | using android::hardware::hidl_vec; |
| 34 | |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 35 | namespace test { |
| 36 | |
| 37 | namespace { |
| 38 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 39 | class StdoutLogger : public ::keymaster::Logger { |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 40 | public: |
| 41 | StdoutLogger() { set_instance(this); } |
| 42 | |
| 43 | int log_msg(LogLevel level, const char* fmt, va_list args) const { |
| 44 | int output_len = 0; |
| 45 | switch (level) { |
| 46 | case DEBUG_LVL: |
| 47 | output_len = printf("DEBUG: "); |
| 48 | break; |
| 49 | case INFO_LVL: |
| 50 | output_len = printf("INFO: "); |
| 51 | break; |
| 52 | case WARNING_LVL: |
| 53 | output_len = printf("WARNING: "); |
| 54 | break; |
| 55 | case ERROR_LVL: |
| 56 | output_len = printf("ERROR: "); |
| 57 | break; |
| 58 | case SEVERE_LVL: |
| 59 | output_len = printf("SEVERE: "); |
| 60 | break; |
| 61 | } |
| 62 | |
| 63 | output_len += vprintf(fmt, args); |
| 64 | output_len += printf("\n"); |
| 65 | return output_len; |
| 66 | } |
| 67 | }; |
| 68 | |
| 69 | StdoutLogger logger; |
| 70 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 71 | } // namespace |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 72 | |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 73 | constexpr const uint8_t test_token[69] = { |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 74 | 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, |
| 75 | 0x0e, 0x0f, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18, 0x19, 0x1a, 0x1b, |
| 76 | 0x1c, 0x1d, 0x1e, 0x1f, 0x20, 0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28, 0x29, |
| 77 | 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, |
| 78 | 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f, 0x40, 0x41, 0x42, 0x43, 0x44}; |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 79 | |
| 80 | constexpr const uint8_t test_hmac_data[] = { |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 81 | 0x25, 0x26, 0x27, 0x28, 0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30, 0x31, 0x32, 0x33, 0x34, |
| 82 | 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f, 0x40, 0x41, 0x42, 0x43, 0x44}; |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 83 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 84 | static const Km3HardwareAuthToken km3_hidl_test_token_little_endian = { |
| 85 | UINT64_C(0x0807060504030201), UINT64_C(0x100f0e0d0c0b0a09), |
| 86 | UINT64_C(0x1817161514131211), UINT32_C(0x1c1b1a19), |
| 87 | UINT64_C(0x24232221201f1e1d), hidl_array<uint8_t, 32>(test_hmac_data)}; |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 88 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 89 | static const HardwareAuthToken km4_hidl_test_token = { |
| 90 | UINT64_C(0x0807060504030201), UINT64_C(0x100f0e0d0c0b0a09), |
| 91 | UINT64_C(0x1817161514131211), static_cast<HardwareAuthenticatorType>(UINT32_C(0x191a1b1c)), |
| 92 | UINT64_C(0x1d1e1f2021222324), hidl_vec<uint8_t>(test_hmac_data, test_hmac_data + 32)}; |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 93 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 94 | TEST(AuthenticationTokenFormattingTest, hidlVec2Km3AuthToken) { |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 95 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 96 | hidl_vec<uint8_t> hidl_test_token; |
| 97 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 98 | ASSERT_EQ(km3_hidl_test_token_little_endian, hidlVec2Km3AuthToken(hidl_test_token)); |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 99 | } |
| 100 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 101 | TEST(AuthenticationTokenFormattingTest, hidlVec2Km4AuthToken) { |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 102 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 103 | hidl_vec<uint8_t> hidl_test_token; |
| 104 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 105 | ASSERT_EQ(km4_hidl_test_token, hidlVec2AuthToken(hidl_test_token)); |
| 106 | } |
| 107 | |
| 108 | TEST(AuthenticationTokenFormattingTest, km3AuthToken2HidlVec) { |
| 109 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 110 | hidl_vec<uint8_t> hidl_test_token; |
| 111 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
| 112 | ASSERT_EQ(hidl_test_token, authToken2HidlVec(km3_hidl_test_token_little_endian)); |
| 113 | } |
| 114 | |
| 115 | TEST(AuthenticationTokenFormattingTest, km4AuthToken2HidlVec) { |
| 116 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 117 | hidl_vec<uint8_t> hidl_test_token; |
| 118 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
| 119 | ASSERT_EQ(hidl_test_token, authToken2HidlVec(km4_hidl_test_token)); |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 120 | } |
| 121 | |
| 122 | TEST(AuthenticationTokenFormattingTest, backAndForth) { |
| 123 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 124 | hidl_vec<uint8_t> hidl_test_token; |
| 125 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 126 | ASSERT_EQ(km3_hidl_test_token_little_endian, |
| 127 | hidlVec2Km3AuthToken(authToken2HidlVec(km3_hidl_test_token_little_endian))); |
| 128 | ASSERT_EQ(km4_hidl_test_token, hidlVec2AuthToken(authToken2HidlVec(km4_hidl_test_token))); |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 129 | } |
| 130 | |
| 131 | TEST(AuthenticationTokenFormattingTest, forthAndBack) { |
| 132 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 133 | hidl_vec<uint8_t> hidl_test_token; |
| 134 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 135 | ASSERT_EQ(hidl_test_token, authToken2HidlVec(hidlVec2Km3AuthToken(hidl_test_token))); |
| 136 | ASSERT_EQ(hidl_test_token, authToken2HidlVec(hidlVec2Km3AuthToken(hidl_test_token))); |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 137 | } |
| 138 | |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 139 | TEST(AuthenticationTokenFormattingTest, roundAndRound) { |
| 140 | static_assert(sizeof(hw_auth_token_t) == sizeof(test_token), "test_token has wrong size"); |
| 141 | hidl_vec<uint8_t> hidl_test_token; |
| 142 | hidl_test_token.setToExternal(const_cast<unsigned char*>(test_token), sizeof(test_token)); |
| 143 | HardwareAuthToken km4_from_hidl = hidlVec2AuthToken(hidl_test_token); |
| 144 | hidl_vec<uint8_t> hidl_from_km4 = authToken2HidlVec(km4_from_hidl); |
| 145 | Km3HardwareAuthToken km3_from_hidl = hidlVec2Km3AuthToken(hidl_from_km4); |
| 146 | hidl_vec<uint8_t> hidl_from_km3 = authToken2HidlVec(km3_from_hidl); |
| 147 | |
| 148 | ASSERT_EQ(hidl_from_km4, hidl_test_token); |
| 149 | ASSERT_EQ(hidl_from_km3, hidl_test_token); |
| 150 | ASSERT_NE(km4_from_hidl.timestamp, km3_from_hidl.timestamp); |
| 151 | ASSERT_NE(static_cast<uint32_t>(km4_from_hidl.authenticatorType), |
| 152 | km3_from_hidl.authenticatorType); |
| 153 | } |
| 154 | |
Janis Danisevskis | 9a390a7 | 2017-11-29 12:02:07 -0800 | [diff] [blame] | 155 | } // namespace test |
Shawn Willden | 0329a82 | 2017-12-04 13:55:14 -0700 | [diff] [blame] | 156 | } // namespace keystore |