1 /*
2 * Copyright 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 <iostream>
18
19 #include <gtest/gtest.h>
20
21 #include <keymaster/wrapped_key.h>
22
23 #include "android_keymaster_test_utils.h"
24
25 using ::std::cout;
26 using ::std::endl;
27 using ::std::string;
28
29 namespace keymaster {
30 namespace test {
31
32 string test_wrapped_key = hex2str(
33 "3082015E020100048201005930C4FFE73B214575A66FB1DC07FD72F2508488F927926DB8DE8A78D780169FFC79728E"
34 "63BE14280C5481856CB51886BB1FF7D7F0BB73013DAE5386C7F63CD7D12E7FCC9AF89A7A52E68AEBB3CD3C08819FB2"
35 "A1D10EA717FF662D9FCF00194B7D7B75F6A898EF3295454642F697123758FB172EF015B515A2AC791BE35077346503"
36 "7D25B45375B7E00472C5250F7FD9053ECEA62D59EE3734C919A124A1659EF4F031F137DB661C0E846DFEE46C4CC85F"
37 "99B47708ADDEF2B21E1143F59A0EE12E0AB5ADF9E03C26642FC36905F38EE60A9B385FF4785FDF6611B60BD9DB283D"
38 "EDD4481DFCBCCBB51166F475A94898EC759BB9125520304FF82124559D27BE2B040CD796B02C370F1FA4CC0124F130"
39 "13020103300EA1023100A203020120A3030201200420CCD540855F833A5E1480BFD2D36FAF3AEEE15DF5BEABE2691B"
40 "C82DDE2A7AA910041094CD97F58DE55B737B60B3AD127B1C59");
41
42 string test_tag = hex2str("94CD97F58DE55B737B60B3AD127B1C59");
43 string test_iv = hex2str("D796B02C370F1FA4CC0124F1");
44 string test_transit_key =
45 hex2str("78421A8124D7960B4CBFC8F4F16B421B3511A3D29CFB329C3CCD90724FD9E8E440"
46 "F0058F2035645EDAD7BCF62D0ED23D39B049069B2B0F8607F32B084804824A6620"
47 "F2658FC74ECBFCE9533FE220E981EF1E05170988CA5EB42480FCD711B7668140DF"
48 "5DC5D23DCAFC536A971DDB4FD65E5B5F7C01E5C13079CA03C301A28C2463885663"
49 "BD649400113A8AF4FDF0D3A8B1964D48D4B5EF696D6CE4F7EF943966E7CAB4A9EA"
50 "88AD0364E454452D5D5A2EFB57049C5EDDF6AEFB068B4D5A739E5B9ACFB3F0891B"
51 "972B1A1F65167EAC34FD73BDB3D60CE6886293F755A3EA6D6CF216CB00E3A28A05"
52 "9A41818BEFE3A159329A335CF3BA87B65C53D691FC12FF1911");
53 string test_secure_key =
54 hex2str("CCD540855F833A5E1480BFD2D36FAF3AEEE15DF5BEABE2691BC82DDE2A7AA910");
55
blob2string(keymaster_blob_t & blob)56 string blob2string(keymaster_blob_t& blob) {
57 string s(reinterpret_cast<const char*>(blob.data), blob.data_length);
58 return s;
59 }
60
keyblob2string(keymaster_key_blob_t & blob)61 string keyblob2string(keymaster_key_blob_t& blob) {
62 string s(reinterpret_cast<const char*>(blob.key_material), blob.key_material_size);
63 return s;
64 }
65
TEST(WrappedKeyTest,Simple)66 TEST(WrappedKeyTest, Simple) {
67
68 KeymasterKeyBlob asn1;
69 size_t asn1_len;
70
71 KeymasterBlob iv = {reinterpret_cast<const uint8_t*>(test_iv.c_str()), test_iv.size()};
72 KeymasterKeyBlob tk = {reinterpret_cast<const uint8_t*>(test_transit_key.c_str()),
73 test_transit_key.size()};
74 KeymasterKeyBlob secure_key = {reinterpret_cast<const uint8_t*>(test_secure_key.c_str()),
75 test_secure_key.size()};
76 KeymasterBlob tag = {reinterpret_cast<const uint8_t*>(test_tag.c_str()), test_tag.size()};
77 AuthorizationSet authorization_list = AuthorizationSetBuilder().AesEncryptionKey(256).build();
78
79 EXPECT_EQ(
80 build_wrapped_key(tk, iv, KM_KEY_FORMAT_RAW, secure_key, tag, authorization_list, &asn1),
81 KM_ERROR_OK);
82
83 KeymasterBlob iv2;
84 KeymasterBlob tag2;
85 KeymasterBlob wrapped_key_description;
86 KeymasterKeyBlob secure_key2;
87 KeymasterKeyBlob transit_key2;
88 AuthorizationSet auth_list;
89 keymaster_key_format_t key_format;
90 EXPECT_EQ(parse_wrapped_key(asn1, &iv2, &transit_key2, &secure_key2, &tag2, &auth_list,
91 &key_format, &wrapped_key_description),
92 KM_ERROR_OK);
93
94 uint32_t key_size;
95 auth_list.GetTagValue(TAG_KEY_SIZE, &key_size);
96 EXPECT_EQ(key_size, (uint32_t)256);
97
98 keymaster_algorithm_t algorithm;
99 auth_list.GetTagValue(TAG_ALGORITHM, &algorithm);
100 EXPECT_EQ(algorithm, KM_ALGORITHM_AES);
101
102 EXPECT_EQ(key_format, (uint32_t)KM_KEY_FORMAT_RAW);
103 EXPECT_EQ(blob2string(tag2), test_tag);
104 EXPECT_EQ(blob2string(iv2), test_iv);
105 }
106
TEST(WrappedKeyTest,Unwrap)107 TEST(WrappedKeyTest, Unwrap) {
108 KeymasterKeyBlob wrapped_key = {reinterpret_cast<const uint8_t*>(test_wrapped_key.c_str()),
109 test_wrapped_key.size()};
110
111 KeymasterKeyBlob secure_key;
112 KeymasterKeyBlob transit_key;
113 KeymasterBlob iv;
114 KeymasterBlob tag;
115 KeymasterBlob wrapped_key_description;
116 AuthorizationSet auth_list;
117 keymaster_key_format_t key_format;
118 EXPECT_EQ(parse_wrapped_key(wrapped_key, &iv, &transit_key, &secure_key, &tag, &auth_list,
119 &key_format, &wrapped_key_description),
120 KM_ERROR_OK);
121
122 EXPECT_EQ(blob2string(tag), test_tag);
123 EXPECT_EQ(blob2string(iv), test_iv);
124 EXPECT_EQ(keyblob2string(secure_key), test_secure_key);
125 }
126
127 } // namespace test
128 } // namespace keymaster
129