Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2014 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 <UniquePtr.h> |
| 18 | |
| 19 | #include <gtest/gtest.h> |
| 20 | |
Shawn Willden | 98d9b92 | 2014-08-26 08:14:10 -0600 | [diff] [blame] | 21 | #include <keymaster/keymaster_tags.h> |
| 22 | #include <keymaster/google_keymaster_utils.h> |
| 23 | |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 24 | #include "google_keymaster_test_utils.h" |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 25 | #include "google_softkeymaster.h" |
| 26 | |
| 27 | int main(int argc, char** argv) { |
| 28 | ::testing::InitGoogleTest(&argc, argv); |
| 29 | int result = RUN_ALL_TESTS(); |
| 30 | return result; |
| 31 | } |
| 32 | |
| 33 | namespace keymaster { |
| 34 | namespace test { |
| 35 | |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 36 | /** |
| 37 | * Serialize and deserialize a message. |
| 38 | */ |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 39 | template <typename Message> |
| 40 | Message* round_trip(int32_t ver, const Message& message, size_t expected_size) { |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 41 | size_t size = message.SerializedSize(); |
| 42 | EXPECT_EQ(expected_size, size); |
| 43 | if (size == 0) |
| 44 | return NULL; |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 45 | |
| 46 | UniquePtr<uint8_t[]> buf(new uint8_t[size]); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 47 | EXPECT_EQ(buf.get() + size, message.Serialize(buf.get(), buf.get() + size)); |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 48 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 49 | Message* deserialized = new Message(ver); |
Shawn Willden | 58e1a54 | 2014-08-08 21:58:29 -0600 | [diff] [blame] | 50 | const uint8_t* p = buf.get(); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 51 | EXPECT_TRUE(deserialized->Deserialize(&p, p + size)); |
| 52 | EXPECT_EQ((ptrdiff_t)size, p - buf.get()); |
| 53 | return deserialized; |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 54 | } |
| 55 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 56 | struct EmptyKeymasterResponse : public KeymasterResponse { |
| 57 | EmptyKeymasterResponse(int32_t ver) : KeymasterResponse(ver) {} |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 58 | size_t NonErrorSerializedSize() const { return 1; } |
| 59 | uint8_t* NonErrorSerialize(uint8_t* buf, const uint8_t* /* end */) const { |
| 60 | *buf++ = 0; |
| 61 | return buf; |
| 62 | } |
| 63 | bool NonErrorDeserialize(const uint8_t** buf_ptr, const uint8_t* end) { |
| 64 | if (*buf_ptr >= end) |
| 65 | return false; |
| 66 | EXPECT_EQ(0, **buf_ptr); |
| 67 | (*buf_ptr)++; |
| 68 | return true; |
| 69 | } |
| 70 | }; |
| 71 | |
| 72 | TEST(RoundTrip, EmptyKeymasterResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 73 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 74 | EmptyKeymasterResponse msg(ver); |
| 75 | msg.error = KM_ERROR_OK; |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 76 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 77 | UniquePtr<EmptyKeymasterResponse> deserialized(round_trip(ver, msg, 5)); |
| 78 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 79 | } |
| 80 | |
| 81 | TEST(RoundTrip, EmptyKeymasterResponseError) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 82 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 83 | EmptyKeymasterResponse msg(ver); |
| 84 | msg.error = KM_ERROR_MEMORY_ALLOCATION_FAILED; |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 85 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 86 | UniquePtr<EmptyKeymasterResponse> deserialized(round_trip(ver, msg, 4)); |
| 87 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 88 | } |
| 89 | |
| 90 | TEST(RoundTrip, SupportedAlgorithmsResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 91 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 92 | SupportedAlgorithmsResponse rsp(ver); |
| 93 | keymaster_algorithm_t algorithms[] = {KM_ALGORITHM_RSA, KM_ALGORITHM_DSA, |
| 94 | KM_ALGORITHM_ECDSA}; |
| 95 | rsp.error = KM_ERROR_OK; |
| 96 | rsp.algorithms = dup_array(algorithms); |
| 97 | rsp.algorithms_length = array_length(algorithms); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 98 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 99 | UniquePtr<SupportedAlgorithmsResponse> deserialized(round_trip(ver, rsp, 20)); |
| 100 | EXPECT_EQ(array_length(algorithms), deserialized->algorithms_length); |
| 101 | EXPECT_EQ(0, memcmp(deserialized->algorithms, algorithms, array_size(algorithms))); |
| 102 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 103 | } |
| 104 | |
| 105 | TEST(RoundTrip, SupportedResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 106 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 107 | SupportedResponse<keymaster_digest_t> rsp(ver); |
| 108 | keymaster_digest_t digests[] = {KM_DIGEST_NONE, KM_DIGEST_MD5, KM_DIGEST_SHA1}; |
| 109 | rsp.error = KM_ERROR_OK; |
| 110 | rsp.SetResults(digests); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 111 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 112 | UniquePtr<SupportedResponse<keymaster_digest_t>> deserialized(round_trip(ver, rsp, 20)); |
| 113 | EXPECT_EQ(array_length(digests), deserialized->results_length); |
| 114 | EXPECT_EQ(0, memcmp(deserialized->results, digests, array_size(digests))); |
| 115 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 116 | } |
| 117 | |
| 118 | static keymaster_key_param_t params[] = { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 119 | Authorization(TAG_PURPOSE, KM_PURPOSE_SIGN), Authorization(TAG_PURPOSE, KM_PURPOSE_VERIFY), |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 120 | Authorization(TAG_ALGORITHM, KM_ALGORITHM_RSA), Authorization(TAG_USER_ID, 7), |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 121 | Authorization(TAG_USER_AUTH_ID, 8), Authorization(TAG_APPLICATION_ID, "app_id", 6), |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 122 | Authorization(TAG_AUTH_TIMEOUT, 300), |
| 123 | }; |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 124 | uint8_t TEST_DATA[] = "a key blob"; |
| 125 | |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 126 | TEST(RoundTrip, GenerateKeyRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 127 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 128 | GenerateKeyRequest req(ver); |
| 129 | req.key_description.Reinitialize(params, array_length(params)); |
| 130 | UniquePtr<GenerateKeyRequest> deserialized(round_trip(ver, req, 78)); |
| 131 | EXPECT_EQ(deserialized->key_description, req.key_description); |
| 132 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 133 | } |
| 134 | |
| 135 | TEST(RoundTrip, GenerateKeyResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 136 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 137 | GenerateKeyResponse rsp(ver); |
| 138 | rsp.error = KM_ERROR_OK; |
| 139 | rsp.key_blob.key_material = dup_array(TEST_DATA); |
| 140 | rsp.key_blob.key_material_size = array_length(TEST_DATA); |
| 141 | rsp.enforced.Reinitialize(params, array_length(params)); |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 142 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 143 | UniquePtr<GenerateKeyResponse> deserialized(round_trip(ver, rsp, 109)); |
| 144 | EXPECT_EQ(KM_ERROR_OK, deserialized->error); |
| 145 | EXPECT_EQ(deserialized->enforced, rsp.enforced); |
| 146 | EXPECT_EQ(deserialized->unenforced, rsp.unenforced); |
| 147 | } |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 148 | } |
| 149 | |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 150 | TEST(RoundTrip, GenerateKeyResponseTestError) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 151 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 152 | GenerateKeyResponse rsp(ver); |
| 153 | rsp.error = KM_ERROR_UNSUPPORTED_ALGORITHM; |
| 154 | rsp.key_blob.key_material = dup_array(TEST_DATA); |
| 155 | rsp.key_blob.key_material_size = array_length(TEST_DATA); |
| 156 | rsp.enforced.Reinitialize(params, array_length(params)); |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 157 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 158 | UniquePtr<GenerateKeyResponse> deserialized(round_trip(ver, rsp, 4)); |
| 159 | EXPECT_EQ(KM_ERROR_UNSUPPORTED_ALGORITHM, deserialized->error); |
| 160 | EXPECT_EQ(0U, deserialized->enforced.size()); |
| 161 | EXPECT_EQ(0U, deserialized->unenforced.size()); |
| 162 | EXPECT_EQ(0U, deserialized->key_blob.key_material_size); |
| 163 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 164 | } |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 165 | |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 166 | TEST(RoundTrip, GetKeyCharacteristicsRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 167 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 168 | GetKeyCharacteristicsRequest req(ver); |
| 169 | req.additional_params.Reinitialize(params, array_length(params)); |
| 170 | req.SetKeyMaterial("foo", 3); |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 171 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 172 | UniquePtr<GetKeyCharacteristicsRequest> deserialized(round_trip(ver, req, 85)); |
| 173 | EXPECT_EQ(7U, deserialized->additional_params.size()); |
| 174 | EXPECT_EQ(3U, deserialized->key_blob.key_material_size); |
| 175 | EXPECT_EQ(0, memcmp(deserialized->key_blob.key_material, "foo", 3)); |
| 176 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 177 | } |
| 178 | |
| 179 | TEST(RoundTrip, GetKeyCharacteristicsResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 180 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 181 | GetKeyCharacteristicsResponse msg(ver); |
| 182 | msg.error = KM_ERROR_OK; |
| 183 | msg.enforced.Reinitialize(params, array_length(params)); |
| 184 | msg.unenforced.Reinitialize(params, array_length(params)); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 185 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 186 | UniquePtr<GetKeyCharacteristicsResponse> deserialized(round_trip(ver, msg, 160)); |
| 187 | EXPECT_EQ(msg.enforced, deserialized->enforced); |
| 188 | EXPECT_EQ(msg.unenforced, deserialized->unenforced); |
| 189 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 190 | } |
| 191 | |
| 192 | TEST(RoundTrip, BeginOperationRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 193 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 194 | BeginOperationRequest msg(ver); |
| 195 | msg.purpose = KM_PURPOSE_SIGN; |
| 196 | msg.SetKeyMaterial("foo", 3); |
| 197 | msg.additional_params.Reinitialize(params, array_length(params)); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 198 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 199 | UniquePtr<BeginOperationRequest> deserialized(round_trip(ver, msg, 89)); |
| 200 | EXPECT_EQ(KM_PURPOSE_SIGN, deserialized->purpose); |
| 201 | EXPECT_EQ(3U, deserialized->key_blob.key_material_size); |
| 202 | EXPECT_EQ(0, memcmp(deserialized->key_blob.key_material, "foo", 3)); |
| 203 | EXPECT_EQ(msg.additional_params, deserialized->additional_params); |
| 204 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 205 | } |
| 206 | |
| 207 | TEST(RoundTrip, BeginOperationResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 208 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 209 | BeginOperationResponse msg(ver); |
| 210 | msg.error = KM_ERROR_OK; |
| 211 | msg.op_handle = 0xDEADBEEF; |
Shawn Willden | 7b38f44 | 2015-01-22 13:07:34 -0700 | [diff] [blame] | 212 | msg.output_params.push_back(Authorization(TAG_NONCE, "foo", 3)); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 213 | |
Shawn Willden | 7b38f44 | 2015-01-22 13:07:34 -0700 | [diff] [blame] | 214 | UniquePtr<BeginOperationResponse> deserialized; |
| 215 | switch (ver) { |
| 216 | case 0: |
| 217 | deserialized.reset(round_trip(ver, msg, 12)); |
| 218 | break; |
| 219 | case 1: |
| 220 | deserialized.reset(round_trip(ver, msg, 39)); |
| 221 | break; |
| 222 | default: |
| 223 | FAIL(); |
| 224 | } |
| 225 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 226 | EXPECT_EQ(KM_ERROR_OK, deserialized->error); |
| 227 | EXPECT_EQ(0xDEADBEEF, deserialized->op_handle); |
Shawn Willden | 7b38f44 | 2015-01-22 13:07:34 -0700 | [diff] [blame] | 228 | |
| 229 | switch (ver) { |
| 230 | case 0: |
| 231 | EXPECT_EQ(0, deserialized->output_params.size()); |
| 232 | break; |
| 233 | case 1: |
| 234 | EXPECT_EQ(msg.output_params, deserialized->output_params); |
| 235 | break; |
| 236 | default: |
| 237 | FAIL(); |
| 238 | } |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 239 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 240 | } |
| 241 | |
| 242 | TEST(RoundTrip, BeginOperationResponseError) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 243 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 244 | BeginOperationResponse msg(ver); |
| 245 | msg.error = KM_ERROR_INVALID_OPERATION_HANDLE; |
| 246 | msg.op_handle = 0xDEADBEEF; |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 247 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 248 | UniquePtr<BeginOperationResponse> deserialized(round_trip(ver, msg, 4)); |
| 249 | EXPECT_EQ(KM_ERROR_INVALID_OPERATION_HANDLE, deserialized->error); |
| 250 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 251 | } |
| 252 | |
| 253 | TEST(RoundTrip, UpdateOperationRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 254 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 255 | UpdateOperationRequest msg(ver); |
| 256 | msg.op_handle = 0xDEADBEEF; |
| 257 | msg.input.Reinitialize("foo", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 258 | |
Shawn Willden | ba5e94d | 2015-02-06 17:09:23 -0700 | [diff] [blame] | 259 | UniquePtr<UpdateOperationRequest> deserialized; |
| 260 | switch(ver) { |
| 261 | case 0: |
| 262 | deserialized.reset(round_trip(ver, msg, 15)); |
| 263 | break; |
| 264 | case 1: |
| 265 | deserialized.reset(round_trip(ver, msg, 27)); |
| 266 | break; |
| 267 | default: |
| 268 | FAIL(); |
| 269 | } |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 270 | EXPECT_EQ(3U, deserialized->input.available_read()); |
| 271 | EXPECT_EQ(0, memcmp(deserialized->input.peek_read(), "foo", 3)); |
| 272 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 273 | } |
| 274 | |
| 275 | TEST(RoundTrip, UpdateOperationResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 276 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 277 | UpdateOperationResponse msg(ver); |
| 278 | msg.error = KM_ERROR_OK; |
| 279 | msg.output.Reinitialize("foo", 3); |
Shawn Willden | b736113 | 2014-12-08 08:15:14 -0700 | [diff] [blame] | 280 | msg.input_consumed = 99; |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 281 | |
Shawn Willden | b736113 | 2014-12-08 08:15:14 -0700 | [diff] [blame] | 282 | UniquePtr<UpdateOperationResponse> deserialized; |
| 283 | switch (ver) { |
| 284 | case 0: |
| 285 | deserialized.reset(round_trip(ver, msg, 11)); |
| 286 | break; |
| 287 | case 1: |
| 288 | deserialized.reset(round_trip(ver, msg, 15)); |
| 289 | break; |
| 290 | default: |
| 291 | FAIL(); |
| 292 | } |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 293 | EXPECT_EQ(KM_ERROR_OK, deserialized->error); |
| 294 | EXPECT_EQ(3U, deserialized->output.available_read()); |
| 295 | EXPECT_EQ(0, memcmp(deserialized->output.peek_read(), "foo", 3)); |
Shawn Willden | b736113 | 2014-12-08 08:15:14 -0700 | [diff] [blame] | 296 | |
| 297 | switch (ver) { |
| 298 | case 0: |
| 299 | EXPECT_EQ(0, deserialized->input_consumed); |
| 300 | break; |
| 301 | case 1: |
| 302 | EXPECT_EQ(99, deserialized->input_consumed); |
| 303 | break; |
| 304 | default: |
| 305 | FAIL(); |
| 306 | } |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 307 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 308 | } |
| 309 | |
| 310 | TEST(RoundTrip, FinishOperationRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 311 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 312 | FinishOperationRequest msg(ver); |
| 313 | msg.op_handle = 0xDEADBEEF; |
| 314 | msg.signature.Reinitialize("bar", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 315 | |
Shawn Willden | ba5e94d | 2015-02-06 17:09:23 -0700 | [diff] [blame] | 316 | UniquePtr<FinishOperationRequest> deserialized; |
| 317 | switch(ver) { |
| 318 | case 0: |
| 319 | deserialized.reset(round_trip(ver, msg, 15)); |
| 320 | break; |
| 321 | case 1: |
| 322 | deserialized.reset(round_trip(ver, msg, 27)); |
| 323 | break; |
| 324 | default: |
| 325 | FAIL(); |
| 326 | } |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 327 | EXPECT_EQ(0xDEADBEEF, deserialized->op_handle); |
| 328 | EXPECT_EQ(3U, deserialized->signature.available_read()); |
| 329 | EXPECT_EQ(0, memcmp(deserialized->signature.peek_read(), "bar", 3)); |
| 330 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 331 | } |
| 332 | |
| 333 | TEST(Round_Trip, FinishOperationResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 334 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 335 | FinishOperationResponse msg(ver); |
| 336 | msg.error = KM_ERROR_OK; |
| 337 | msg.output.Reinitialize("foo", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 338 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 339 | UniquePtr<FinishOperationResponse> deserialized(round_trip(ver, msg, 11)); |
| 340 | EXPECT_EQ(msg.error, deserialized->error); |
| 341 | EXPECT_EQ(msg.output.available_read(), deserialized->output.available_read()); |
| 342 | EXPECT_EQ(0, memcmp(msg.output.peek_read(), deserialized->output.peek_read(), |
| 343 | msg.output.available_read())); |
| 344 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 345 | } |
| 346 | |
| 347 | TEST(RoundTrip, ImportKeyRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 348 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 349 | ImportKeyRequest msg(ver); |
| 350 | msg.key_description.Reinitialize(params, array_length(params)); |
| 351 | msg.key_format = KM_KEY_FORMAT_X509; |
| 352 | msg.SetKeyMaterial("foo", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 353 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 354 | UniquePtr<ImportKeyRequest> deserialized(round_trip(ver, msg, 89)); |
| 355 | EXPECT_EQ(msg.key_description, deserialized->key_description); |
| 356 | EXPECT_EQ(msg.key_format, deserialized->key_format); |
| 357 | EXPECT_EQ(msg.key_data_length, deserialized->key_data_length); |
| 358 | EXPECT_EQ(0, memcmp(msg.key_data, deserialized->key_data, msg.key_data_length)); |
| 359 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 360 | } |
| 361 | |
| 362 | TEST(RoundTrip, ImportKeyResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 363 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 364 | ImportKeyResponse msg(ver); |
| 365 | msg.error = KM_ERROR_OK; |
| 366 | msg.SetKeyMaterial("foo", 3); |
| 367 | msg.enforced.Reinitialize(params, array_length(params)); |
| 368 | msg.unenforced.Reinitialize(params, array_length(params)); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 369 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 370 | UniquePtr<ImportKeyResponse> deserialized(round_trip(ver, msg, 167)); |
| 371 | EXPECT_EQ(msg.error, deserialized->error); |
| 372 | EXPECT_EQ(msg.key_blob.key_material_size, deserialized->key_blob.key_material_size); |
| 373 | EXPECT_EQ(0, memcmp(msg.key_blob.key_material, deserialized->key_blob.key_material, |
| 374 | msg.key_blob.key_material_size)); |
| 375 | EXPECT_EQ(msg.enforced, deserialized->enforced); |
| 376 | EXPECT_EQ(msg.unenforced, deserialized->unenforced); |
| 377 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 378 | } |
| 379 | |
| 380 | TEST(RoundTrip, ExportKeyRequest) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 381 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 382 | ExportKeyRequest msg(ver); |
| 383 | msg.additional_params.Reinitialize(params, array_length(params)); |
| 384 | msg.key_format = KM_KEY_FORMAT_X509; |
| 385 | msg.SetKeyMaterial("foo", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 386 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 387 | UniquePtr<ExportKeyRequest> deserialized(round_trip(ver, msg, 89)); |
| 388 | EXPECT_EQ(msg.additional_params, deserialized->additional_params); |
| 389 | EXPECT_EQ(msg.key_format, deserialized->key_format); |
| 390 | EXPECT_EQ(3U, deserialized->key_blob.key_material_size); |
| 391 | EXPECT_EQ(0, memcmp("foo", deserialized->key_blob.key_material, 3)); |
| 392 | } |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 393 | } |
| 394 | |
| 395 | TEST(RoundTrip, ExportKeyResponse) { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 396 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 397 | ExportKeyResponse msg(ver); |
| 398 | msg.error = KM_ERROR_OK; |
| 399 | msg.SetKeyMaterial("foo", 3); |
Shawn Willden | da8485e | 2014-08-17 08:00:01 -0600 | [diff] [blame] | 400 | |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 401 | UniquePtr<ExportKeyResponse> deserialized(round_trip(ver, msg, 11)); |
| 402 | EXPECT_EQ(3U, deserialized->key_data_length); |
| 403 | EXPECT_EQ(0, memcmp("foo", deserialized->key_data, 3)); |
| 404 | } |
| 405 | } |
| 406 | |
| 407 | TEST(RoundTrip, GetVersionRequest) { |
| 408 | GetVersionRequest msg; |
| 409 | |
| 410 | size_t size = msg.SerializedSize(); |
| 411 | ASSERT_EQ(0, size); |
| 412 | |
| 413 | UniquePtr<uint8_t[]> buf(new uint8_t[size]); |
| 414 | EXPECT_EQ(buf.get() + size, msg.Serialize(buf.get(), buf.get() + size)); |
| 415 | |
| 416 | GetVersionRequest deserialized; |
| 417 | const uint8_t* p = buf.get(); |
| 418 | EXPECT_TRUE(deserialized.Deserialize(&p, p + size)); |
| 419 | EXPECT_EQ((ptrdiff_t)size, p - buf.get()); |
| 420 | } |
| 421 | |
| 422 | TEST(RoundTrip, GetVersionResponse) { |
| 423 | GetVersionResponse msg; |
| 424 | msg.error = KM_ERROR_OK; |
| 425 | msg.major_ver = 9; |
| 426 | msg.minor_ver = 98; |
| 427 | msg.subminor_ver = 38; |
| 428 | |
| 429 | size_t size = msg.SerializedSize(); |
| 430 | ASSERT_EQ(7, size); |
| 431 | |
| 432 | UniquePtr<uint8_t[]> buf(new uint8_t[size]); |
| 433 | EXPECT_EQ(buf.get() + size, msg.Serialize(buf.get(), buf.get() + size)); |
| 434 | |
| 435 | GetVersionResponse deserialized; |
| 436 | const uint8_t* p = buf.get(); |
| 437 | EXPECT_TRUE(deserialized.Deserialize(&p, p + size)); |
| 438 | EXPECT_EQ((ptrdiff_t)size, p - buf.get()); |
| 439 | EXPECT_EQ(9, msg.major_ver); |
| 440 | EXPECT_EQ(98, msg.minor_ver); |
| 441 | EXPECT_EQ(38, msg.subminor_ver); |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 442 | } |
| 443 | |
Shawn Willden | cd69582 | 2015-01-26 14:06:32 -0700 | [diff] [blame] | 444 | TEST(RoundTrip, AddEntropyRequest) { |
| 445 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 446 | AddEntropyRequest msg(ver); |
| 447 | msg.random_data.Reinitialize("foo", 3); |
| 448 | |
| 449 | UniquePtr<AddEntropyRequest> deserialized(round_trip(ver, msg, 7)); |
| 450 | EXPECT_EQ(3U, deserialized->random_data.available_read()); |
| 451 | EXPECT_EQ(0, memcmp("foo", deserialized->random_data.peek_read(), 3)); |
| 452 | } |
| 453 | } |
| 454 | |
Shawn Willden | 9c9dd9a | 2015-01-14 08:31:49 -0700 | [diff] [blame] | 455 | TEST(RoundTrip, RescopeRequest) { |
| 456 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 457 | RescopeRequest msg(ver); |
| 458 | msg.additional_params.Reinitialize(params, array_length(params)); |
| 459 | msg.new_authorizations.Reinitialize(params, array_length(params)); |
| 460 | |
| 461 | UniquePtr<RescopeRequest> deserialized(round_trip(ver, msg, 160)); |
| 462 | EXPECT_EQ(msg.new_authorizations, deserialized->new_authorizations); |
| 463 | } |
| 464 | } |
| 465 | |
| 466 | TEST(RoundTrip, RescopeResponse) { |
| 467 | for (int ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 468 | RescopeResponse msg(ver); |
| 469 | msg.error = KM_ERROR_OK; |
| 470 | msg.enforced.Reinitialize(params, array_length(params)); |
| 471 | msg.unenforced.Reinitialize(params, array_length(params)); |
| 472 | msg.SetKeyMaterial("foo", 3); |
| 473 | |
| 474 | UniquePtr<RescopeResponse> deserialized(round_trip(ver, msg, 167)); |
| 475 | EXPECT_EQ(msg.enforced, deserialized->enforced); |
| 476 | EXPECT_EQ(msg.unenforced, deserialized->unenforced); |
| 477 | EXPECT_EQ(3U, deserialized->key_blob.key_material_size); |
| 478 | EXPECT_EQ(0, memcmp("foo", deserialized->key_blob.key_material, 3)); |
| 479 | } |
| 480 | } |
| 481 | |
Shawn Willden | f2282b3 | 2014-08-25 06:49:54 -0600 | [diff] [blame] | 482 | uint8_t msgbuf[] = { |
| 483 | 220, 88, 183, 255, 71, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 484 | 0, 173, 0, 0, 0, 228, 174, 98, 187, 191, 135, 253, 200, 51, 230, 114, 247, 151, 109, |
| 485 | 237, 79, 87, 32, 94, 5, 204, 46, 154, 30, 91, 6, 103, 148, 254, 129, 65, 171, 228, |
| 486 | 167, 224, 163, 9, 15, 206, 90, 58, 11, 205, 55, 211, 33, 87, 178, 149, 91, 28, 236, |
| 487 | 218, 112, 231, 34, 82, 82, 134, 103, 137, 115, 27, 156, 102, 159, 220, 226, 89, 42, 25, |
| 488 | 37, 9, 84, 239, 76, 161, 198, 72, 167, 163, 39, 91, 148, 191, 17, 191, 87, 169, 179, |
| 489 | 136, 10, 194, 154, 4, 40, 107, 109, 61, 161, 20, 176, 247, 13, 214, 106, 229, 45, 17, |
| 490 | 5, 60, 189, 64, 39, 166, 208, 14, 57, 25, 140, 148, 25, 177, 246, 189, 43, 181, 88, |
| 491 | 204, 29, 126, 224, 100, 143, 93, 60, 57, 249, 55, 0, 87, 83, 227, 224, 166, 59, 214, |
| 492 | 81, 144, 129, 58, 6, 57, 46, 254, 232, 41, 220, 209, 230, 167, 138, 158, 94, 180, 125, |
| 493 | 247, 26, 162, 116, 238, 202, 187, 100, 65, 13, 180, 44, 245, 159, 83, 161, 176, 58, 72, |
| 494 | 236, 109, 105, 160, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 495 | 0, 11, 0, 0, 0, 98, 0, 0, 0, 1, 0, 0, 32, 2, 0, 0, 0, 1, 0, |
| 496 | 0, 32, 3, 0, 0, 0, 2, 0, 0, 16, 1, 0, 0, 0, 3, 0, 0, 48, 0, |
| 497 | 1, 0, 0, 200, 0, 0, 80, 3, 0, 0, 0, 0, 0, 0, 0, 244, 1, 0, 112, |
| 498 | 1, 246, 1, 0, 112, 1, 189, 2, 0, 96, 144, 178, 236, 250, 255, 255, 255, 255, 145, |
| 499 | 1, 0, 96, 144, 226, 33, 60, 222, 2, 0, 0, 189, 2, 0, 96, 0, 0, 0, 0, |
| 500 | 0, 0, 0, 0, 190, 2, 0, 16, 1, 0, 0, 0, 12, 0, 0, 0, 0, 0, 0, |
| 501 | 0, 0, 0, 0, 0, 0, 0, 0, 0, 110, 0, 0, 0, 0, 0, 0, 0, 11, 0, |
| 502 | 0, 0, 98, 0, 0, 0, 1, 0, 0, 32, 2, 0, 0, 0, 1, 0, 0, 32, 3, |
| 503 | 0, 0, 0, 2, 0, 0, 16, 1, 0, 0, 0, 3, 0, 0, 48, 0, 1, 0, 0, |
| 504 | 200, 0, 0, 80, 3, 0, 0, 0, 0, 0, 0, 0, 244, 1, 0, 112, 1, 246, 1, |
| 505 | 0, 112, 1, 189, 2, 0, 96, 144, 178, 236, 250, 255, 255, 255, 255, 145, 1, 0, 96, |
| 506 | 144, 226, 33, 60, 222, 2, 0, 0, 189, 2, 0, 96, 0, 0, 0, 0, 0, 0, 0, |
| 507 | 0, 190, 2, 0, 16, 1, 0, 0, 0, |
| 508 | }; |
| 509 | |
| 510 | /* |
| 511 | * These tests don't have any assertions or expectations. They just try to parse garbage, to see if |
| 512 | * the result will be a crash. This is especially informative when run under Valgrind memcheck. |
| 513 | */ |
| 514 | |
| 515 | template <typename Message> void parse_garbage() { |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 516 | for (int32_t ver = 0; ver <= MAX_MESSAGE_VERSION; ++ver) { |
| 517 | Message msg(ver); |
| 518 | const uint8_t* end = msgbuf + array_length(msgbuf); |
| 519 | for (size_t i = 0; i < array_length(msgbuf); ++i) { |
| 520 | const uint8_t* begin = msgbuf + i; |
| 521 | const uint8_t* p = begin; |
| 522 | msg.Deserialize(&p, end); |
| 523 | } |
Shawn Willden | f2282b3 | 2014-08-25 06:49:54 -0600 | [diff] [blame] | 524 | } |
| 525 | } |
| 526 | |
| 527 | #define GARBAGE_TEST(Message) \ |
| 528 | TEST(GarbageTest, Message) { parse_garbage<Message>(); } |
| 529 | |
| 530 | GARBAGE_TEST(SupportedAlgorithmsResponse) |
| 531 | GARBAGE_TEST(GenerateKeyRequest); |
| 532 | GARBAGE_TEST(GenerateKeyResponse); |
| 533 | GARBAGE_TEST(GetKeyCharacteristicsRequest); |
| 534 | GARBAGE_TEST(GetKeyCharacteristicsResponse); |
| 535 | GARBAGE_TEST(BeginOperationRequest); |
| 536 | GARBAGE_TEST(BeginOperationResponse); |
| 537 | GARBAGE_TEST(UpdateOperationRequest); |
| 538 | GARBAGE_TEST(UpdateOperationResponse); |
| 539 | GARBAGE_TEST(FinishOperationRequest); |
| 540 | GARBAGE_TEST(FinishOperationResponse); |
Shawn Willden | cd69582 | 2015-01-26 14:06:32 -0700 | [diff] [blame] | 541 | GARBAGE_TEST(AddEntropyRequest); |
Shawn Willden | f2282b3 | 2014-08-25 06:49:54 -0600 | [diff] [blame] | 542 | GARBAGE_TEST(ImportKeyRequest); |
| 543 | GARBAGE_TEST(ImportKeyResponse); |
| 544 | GARBAGE_TEST(ExportKeyRequest); |
| 545 | GARBAGE_TEST(ExportKeyResponse); |
Shawn Willden | 9c9dd9a | 2015-01-14 08:31:49 -0700 | [diff] [blame] | 546 | GARBAGE_TEST(RescopeRequest); |
| 547 | GARBAGE_TEST(RescopeResponse); |
Shawn Willden | f2282b3 | 2014-08-25 06:49:54 -0600 | [diff] [blame] | 548 | |
| 549 | // The macro doesn't work on this one. |
| 550 | TEST(GarbageTest, SupportedResponse) { |
| 551 | parse_garbage<SupportedResponse<keymaster_digest_t>>(); |
| 552 | } |
| 553 | |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 554 | } // namespace test |
Shawn Willden | 2665e86 | 2014-11-24 14:46:21 -0700 | [diff] [blame] | 555 | |
Shawn Willden | 128ffe0 | 2014-08-06 12:31:33 -0600 | [diff] [blame] | 556 | } // namespace keymaster |