1 /*
2  * Copyright (C) 2020 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 #define LOG_TAG "keymaster_hidl_hal_test"
18 #include <cutils/log.h>
19 
20 #include "Keymaster4_1HidlTest.h"
21 
22 #include <cutils/properties.h>
23 
24 #include <openssl/x509.h>
25 
26 #include <keymasterV4_1/attestation_record.h>
27 #include <keymasterV4_1/authorization_set.h>
28 
29 // Not to dump the attestation by default. Can enable by specify the parameter
30 // "--dump_attestations" on lunching VTS
31 static bool dumpAttestations = false;
32 
33 namespace android::hardware::keymaster::V4_0 {
34 
operator ==(const AuthorizationSet & a,const AuthorizationSet & b)35 bool operator==(const AuthorizationSet& a, const AuthorizationSet& b) {
36     return std::equal(a.begin(), a.end(), b.begin(), b.end());
37 }
38 
39 }  // namespace android::hardware::keymaster::V4_0
40 
41 namespace android::hardware::keymaster::V4_1 {
42 
operator <<(::std::ostream & os,Tag tag)43 inline ::std::ostream& operator<<(::std::ostream& os, Tag tag) {
44     return os << toString(tag);
45 }
46 
47 namespace test {
48 
49 using std::string;
50 using std::tuple;
51 
52 namespace {
53 
54 char nibble2hex[16] = {'0', '1', '2', '3', '4', '5', '6', '7',
55                        '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
56 
bin2hex(const hidl_vec<uint8_t> & data)57 string bin2hex(const hidl_vec<uint8_t>& data) {
58     string retval;
59     retval.reserve(data.size() * 2 + 1);
60     for (uint8_t byte : data) {
61         retval.push_back(nibble2hex[0x0F & (byte >> 4)]);
62         retval.push_back(nibble2hex[0x0F & byte]);
63     }
64     return retval;
65 }
66 
dumpContent(string content)67 inline void dumpContent(string content) {
68     std::cout << content << std::endl;
69 }
70 
71 struct AuthorizationSetDifferences {
72     string aName;
73     string bName;
74     AuthorizationSet aWhackB;
75     AuthorizationSet bWhackA;
76 };
77 
operator <<(std::ostream & o,const AuthorizationSetDifferences & diffs)78 std::ostream& operator<<(std::ostream& o, const AuthorizationSetDifferences& diffs) {
79     if (!diffs.aWhackB.empty()) {
80         o << "Set " << diffs.aName << " contains the following that " << diffs.bName << " does not"
81           << diffs.aWhackB;
82         if (!diffs.bWhackA.empty()) o << std::endl;
83     }
84 
85     if (!diffs.bWhackA.empty()) {
86         o << "Set " << diffs.bName << " contains the following that " << diffs.aName << " does not"
87           << diffs.bWhackA;
88     }
89     return o;
90 }
91 
92 // Computes and returns a \ b and b \ a ('\' is the set-difference operator, a \ b means all the
93 // elements that are in a but not b, i.e. take a and whack all the elements in b) to the provided
94 // stream.  The sets must be sorted.
95 //
96 // This provides a simple and clear view of how the two sets differ, generally much
97 // easier than scrutinizing printouts of the two sets.
difference(string aName,const AuthorizationSet & a,string bName,const AuthorizationSet & b)98 AuthorizationSetDifferences difference(string aName, const AuthorizationSet& a, string bName,
99                                        const AuthorizationSet& b) {
100     AuthorizationSetDifferences diffs = {std::move(aName), std::move(bName), {}, {}};
101     std::set_difference(a.begin(), a.end(), b.begin(), b.end(), std::back_inserter(diffs.aWhackB));
102     std::set_difference(b.begin(), b.end(), a.begin(), a.end(), std::back_inserter(diffs.bWhackA));
103     return diffs;
104 }
105 
106 #define DIFFERENCE(a, b) difference(#a, a, #b, b)
107 
check_root_of_trust(const RootOfTrust & root_of_trust)108 void check_root_of_trust(const RootOfTrust& root_of_trust) {
109     char vb_meta_device_state[PROPERTY_VALUE_MAX];
110     if (property_get("ro.boot.vbmeta.device_state", vb_meta_device_state, "") == 0) return;
111 
112     char vb_meta_digest[PROPERTY_VALUE_MAX];
113     EXPECT_GT(property_get("ro.boot.vbmeta.digest", vb_meta_digest, ""), 0);
114     EXPECT_EQ(vb_meta_digest, bin2hex(root_of_trust.verified_boot_hash));
115 
116     // Verified boot key should be all 0's if the boot state is not verified or self signed
117     HidlBuf empty_boot_key(string(32, '\0'));
118 
119     char vb_meta_bootstate[PROPERTY_VALUE_MAX];
120     auto& verified_boot_key = root_of_trust.verified_boot_key;
121     auto& verified_boot_state = root_of_trust.verified_boot_state;
122     EXPECT_GT(property_get("ro.boot.verifiedbootstate", vb_meta_bootstate, ""), 0);
123     if (!strcmp(vb_meta_bootstate, "green")) {
124         EXPECT_EQ(verified_boot_state, V4_0::KM_VERIFIED_BOOT_VERIFIED);
125         EXPECT_NE(verified_boot_key, empty_boot_key);
126     } else if (!strcmp(vb_meta_bootstate, "yellow")) {
127         EXPECT_EQ(verified_boot_state, V4_0::KM_VERIFIED_BOOT_SELF_SIGNED);
128         EXPECT_NE(verified_boot_key, empty_boot_key);
129     } else if (!strcmp(vb_meta_bootstate, "orange")) {
130         EXPECT_EQ(verified_boot_state, V4_0::KM_VERIFIED_BOOT_UNVERIFIED);
131         EXPECT_EQ(verified_boot_key, empty_boot_key);
132     } else if (!strcmp(vb_meta_bootstate, "red")) {
133         EXPECT_EQ(verified_boot_state, V4_0::KM_VERIFIED_BOOT_FAILED);
134     } else {
135         EXPECT_EQ(verified_boot_state, V4_0::KM_VERIFIED_BOOT_UNVERIFIED);
136         EXPECT_EQ(verified_boot_key, empty_boot_key);
137     }
138 }
139 
tag_in_list(const KeyParameter & entry)140 bool tag_in_list(const KeyParameter& entry) {
141     // Attestations don't contain everything in key authorization lists, so we need to filter
142     // the key lists to produce the lists that we expect to match the attestations.
143     auto tag_list = {
144             Tag::INCLUDE_UNIQUE_ID, Tag::BLOB_USAGE_REQUIREMENTS, Tag::EC_CURVE,
145             Tag::HARDWARE_TYPE,     Tag::VENDOR_PATCHLEVEL,       Tag::BOOT_PATCHLEVEL,
146             Tag::CREATION_DATETIME,
147     };
148     return std::find(tag_list.begin(), tag_list.end(), (V4_1::Tag)entry.tag) != tag_list.end();
149 }
150 
filter_tags(const AuthorizationSet & set)151 AuthorizationSet filter_tags(const AuthorizationSet& set) {
152     AuthorizationSet filtered;
153     std::remove_copy_if(set.begin(), set.end(), std::back_inserter(filtered), tag_in_list);
154     return filtered;
155 }
156 
check_attestation_record(AttestationRecord attestation,const HidlBuf & challenge,AuthorizationSet expected_sw_enforced,AuthorizationSet expected_hw_enforced,SecurityLevel expected_security_level)157 void check_attestation_record(AttestationRecord attestation, const HidlBuf& challenge,
158                               AuthorizationSet expected_sw_enforced,
159                               AuthorizationSet expected_hw_enforced,
160                               SecurityLevel expected_security_level) {
161     EXPECT_EQ(41U, attestation.keymaster_version);
162     EXPECT_EQ(4U, attestation.attestation_version);
163     EXPECT_EQ(expected_security_level, attestation.attestation_security_level);
164     EXPECT_EQ(expected_security_level, attestation.keymaster_security_level);
165     EXPECT_EQ(challenge, attestation.attestation_challenge);
166 
167     check_root_of_trust(attestation.root_of_trust);
168 
169     // Sort all of the authorization lists, so that equality matching works.
170     expected_sw_enforced.Sort();
171     expected_hw_enforced.Sort();
172     attestation.software_enforced.Sort();
173     attestation.hardware_enforced.Sort();
174 
175     EXPECT_EQ(filter_tags(expected_sw_enforced), filter_tags(attestation.software_enforced))
176             << DIFFERENCE(expected_sw_enforced, attestation.software_enforced);
177     EXPECT_EQ(filter_tags(expected_hw_enforced), filter_tags(attestation.hardware_enforced))
178             << DIFFERENCE(expected_hw_enforced, attestation.hardware_enforced);
179 }
180 
181 }  // namespace
182 
183 using std::string;
184 using DeviceUniqueAttestationTest = Keymaster4_1HidlTest;
185 
TEST_P(DeviceUniqueAttestationTest,NonStrongBoxOnly)186 TEST_P(DeviceUniqueAttestationTest, NonStrongBoxOnly) {
187     if (SecLevel() == SecurityLevel::STRONGBOX) return;
188 
189     ASSERT_EQ(ErrorCode::OK, convert(GenerateKey(AuthorizationSetBuilder()
190                                                          .Authorization(TAG_NO_AUTH_REQUIRED)
191                                                          .RsaSigningKey(2048, 65537)
192                                                          .Digest(Digest::SHA_2_256)
193                                                          .Padding(PaddingMode::RSA_PKCS1_1_5_SIGN)
194                                                          .Authorization(TAG_INCLUDE_UNIQUE_ID))));
195 
196     hidl_vec<hidl_vec<uint8_t>> cert_chain;
197     EXPECT_EQ(ErrorCode::UNIMPLEMENTED,
198               convert(AttestKey(
199                       AuthorizationSetBuilder()
200                               .Authorization(TAG_DEVICE_UNIQUE_ATTESTATION)
201                               .Authorization(TAG_ATTESTATION_CHALLENGE, HidlBuf("challenge"))
202                               .Authorization(TAG_ATTESTATION_APPLICATION_ID, HidlBuf("foo")),
203                       &cert_chain)));
204     CheckedDeleteKey();
205 
206     ASSERT_EQ(ErrorCode::OK, convert(GenerateKey(AuthorizationSetBuilder()
207                                                          .Authorization(TAG_NO_AUTH_REQUIRED)
208                                                          .EcdsaSigningKey(EcCurve::P_256)
209                                                          .Digest(Digest::SHA_2_256)
210                                                          .Authorization(TAG_INCLUDE_UNIQUE_ID))));
211 
212     EXPECT_EQ(ErrorCode::UNIMPLEMENTED,
213               convert(AttestKey(
214                       AuthorizationSetBuilder()
215                               .Authorization(TAG_ATTESTATION_CHALLENGE, HidlBuf("challenge"))
216                               .Authorization(TAG_ATTESTATION_APPLICATION_ID, HidlBuf("foo")),
217                       &cert_chain)));
218 }
219 
TEST_P(DeviceUniqueAttestationTest,Rsa)220 TEST_P(DeviceUniqueAttestationTest, Rsa) {
221     if (SecLevel() != SecurityLevel::STRONGBOX) return;
222     ASSERT_EQ(ErrorCode::OK,
223               convert(GenerateKey(AuthorizationSetBuilder()
224                                   .Authorization(TAG_NO_AUTH_REQUIRED)
225                                   .RsaSigningKey(2048, 65537)
226                                   .Digest(Digest::SHA_2_256)
227                                   .Padding(PaddingMode::RSA_PKCS1_1_5_SIGN)
228                                   .Authorization(TAG_INCLUDE_UNIQUE_ID))));
229 
230     hidl_vec<hidl_vec<uint8_t>> cert_chain;
231     HidlBuf challenge("challenge");
232     HidlBuf app_id("foo");
233     EXPECT_EQ(ErrorCode::OK,
234               convert(AttestKey(AuthorizationSetBuilder()
235                                         .Authorization(TAG_DEVICE_UNIQUE_ATTESTATION)
236                                         .Authorization(TAG_ATTESTATION_CHALLENGE, challenge)
237                                         .Authorization(TAG_ATTESTATION_APPLICATION_ID, app_id),
238                                 &cert_chain)));
239 
240     EXPECT_EQ(2U, cert_chain.size());
241     if (dumpAttestations) dumpContent(bin2hex(cert_chain[0]));
242     auto [err, attestation] = parse_attestation_record(cert_chain[0]);
243     ASSERT_EQ(ErrorCode::OK, err);
244 
245     check_attestation_record(attestation, challenge,
246                              /* sw_enforced */
247                              AuthorizationSetBuilder()
248                                      .Authorization(TAG_ATTESTATION_APPLICATION_ID, app_id),
249                              /* hw_enforced */
250                              AuthorizationSetBuilder()
251                                      .Authorization(TAG_DEVICE_UNIQUE_ATTESTATION)
252                                      .Authorization(TAG_NO_AUTH_REQUIRED)
253                                      .RsaSigningKey(2048, 65537)
254                                      .Digest(Digest::SHA_2_256)
255                                      .Padding(PaddingMode::RSA_PKCS1_1_5_SIGN)
256                                      .Authorization(TAG_ORIGIN, KeyOrigin::GENERATED)
257                                      .Authorization(TAG_OS_VERSION, os_version())
258                                      .Authorization(TAG_OS_PATCHLEVEL, os_patch_level()),
259                              SecLevel());
260 }
261 
TEST_P(DeviceUniqueAttestationTest,Ecdsa)262 TEST_P(DeviceUniqueAttestationTest, Ecdsa) {
263     if (SecLevel() != SecurityLevel::STRONGBOX) return;
264     ASSERT_EQ(ErrorCode::OK,
265               convert(GenerateKey(AuthorizationSetBuilder()
266                                           .Authorization(TAG_NO_AUTH_REQUIRED)
267                                           .EcdsaSigningKey(256)
268                                           .Digest(Digest::SHA_2_256)
269                                           .Authorization(TAG_INCLUDE_UNIQUE_ID))));
270 
271     hidl_vec<hidl_vec<uint8_t>> cert_chain;
272     HidlBuf challenge("challenge");
273     HidlBuf app_id("foo");
274     EXPECT_EQ(ErrorCode::OK,
275               convert(AttestKey(AuthorizationSetBuilder()
276                                         .Authorization(TAG_DEVICE_UNIQUE_ATTESTATION)
277                                         .Authorization(TAG_ATTESTATION_CHALLENGE, challenge)
278                                         .Authorization(TAG_ATTESTATION_APPLICATION_ID, app_id),
279                                 &cert_chain)));
280 
281     EXPECT_EQ(2U, cert_chain.size());
282     if (dumpAttestations) dumpContent(bin2hex(cert_chain[0]));
283     auto [err, attestation] = parse_attestation_record(cert_chain[0]);
284     ASSERT_EQ(ErrorCode::OK, err);
285 
286     check_attestation_record(attestation, challenge,
287             /* sw_enforced */
288             AuthorizationSetBuilder().Authorization(TAG_ATTESTATION_APPLICATION_ID, app_id),
289             /* hw_enforced */
290             AuthorizationSetBuilder()
291                     .Authorization(TAG_DEVICE_UNIQUE_ATTESTATION)
292                     .Authorization(TAG_NO_AUTH_REQUIRED)
293                     .EcdsaSigningKey(256)
294                     .Digest(Digest::SHA_2_256)
295                     .Authorization(TAG_EC_CURVE, EcCurve::P_256)
296                     .Authorization(TAG_ORIGIN, KeyOrigin::GENERATED)
297                     .Authorization(TAG_OS_VERSION, os_version())
298                     .Authorization(TAG_OS_PATCHLEVEL, os_patch_level()),
299             SecLevel());
300 }
301 
302 INSTANTIATE_KEYMASTER_4_1_HIDL_TEST(DeviceUniqueAttestationTest);
303 
304 }  // namespace test
305 }  // namespace android::hardware::keymaster::V4_1
306 
main(int argc,char ** argv)307 int main(int argc, char** argv) {
308     ::testing::InitGoogleTest(&argc, argv);
309     for (int i = 1; i < argc; ++i) {
310         if (argv[i][0] == '-') {
311             if (std::string(argv[i]) == "--dump_attestations") {
312                 dumpAttestations = true;
313             }
314         }
315     }
316     int status = RUN_ALL_TESTS();
317     ALOGI("Test result = %d", status);
318     return status;
319 }
320