1 /*
2  * Copyright (C) 2021 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 "keystoreCommon.h"
18 #include <keystore/KeyAttestationApplicationId.h>
19 
20 using ::security::keymaster::KeyAttestationApplicationId;
21 
22 constexpr size_t kPackageVectorSizeMin = 1;
23 constexpr size_t kPackageVectorSizeMax = 10;
24 
25 class KeystoreApplicationId {
26   public:
27     void process(const uint8_t* data, size_t size);
~KeystoreApplicationId()28     ~KeystoreApplicationId() {}
29 
30   private:
31     void invokeApplicationId();
32     std::unique_ptr<FuzzedDataProvider> mFdp;
33 };
34 
invokeApplicationId()35 void KeystoreApplicationId::invokeApplicationId() {
36     std::optional<KeyAttestationApplicationId> applicationId;
37     bool shouldUsePackageInfoVector = mFdp->ConsumeBool();
38     if (shouldUsePackageInfoVector) {
39         KeyAttestationApplicationId::PackageInfoVector packageInfoVector;
40         int32_t packageVectorSize =
41             mFdp->ConsumeIntegralInRange<int32_t>(kPackageVectorSizeMin, kPackageVectorSizeMax);
42         for (int32_t packageSize = 0; packageSize < packageVectorSize; ++packageSize) {
43             auto packageInfoData = initPackageInfoData(mFdp.get());
44             packageInfoVector.push_back(make_optional<KeyAttestationPackageInfo>(
45                 String16((packageInfoData.packageName).c_str()), packageInfoData.versionCode,
46                 packageInfoData.sharedSignaturesVector));
47         }
48         applicationId = KeyAttestationApplicationId(std::move(packageInfoVector));
49     } else {
50         auto packageInfoData = initPackageInfoData(mFdp.get());
51         applicationId = KeyAttestationApplicationId(make_optional<KeyAttestationPackageInfo>(
52             String16((packageInfoData.packageName).c_str()), packageInfoData.versionCode,
53             packageInfoData.sharedSignaturesVector));
54     }
55     invokeReadWriteParcel(&applicationId.value());
56 }
57 
process(const uint8_t * data,size_t size)58 void KeystoreApplicationId::process(const uint8_t* data, size_t size) {
59     mFdp = std::make_unique<FuzzedDataProvider>(data, size);
60     invokeApplicationId();
61 }
62 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)63 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
64     KeystoreApplicationId keystoreApplicationId;
65     keystoreApplicationId.process(data, size);
66     return 0;
67 }
68