xref: /aosp_15_r20/system/security/keystore/tests/fuzzer/keystoreSignature_fuzzer.cpp (revision e1997b9af69e3155ead6e072d106a0077849ffba)
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 #include "keystoreCommon.h"
17 #include <android/security/keystore/Signature.h>
18 
19 using ::android::security::keystore::Signature;
20 
21 class KeystoreSignatureFuzzer {
22   public:
23     void process(const uint8_t* data, size_t size);
~KeystoreSignatureFuzzer()24     ~KeystoreSignatureFuzzer() {}
25 
26   private:
27     void invokeSignature();
28     std::unique_ptr<FuzzedDataProvider> mFdp;
29 };
30 
invokeSignature()31 void KeystoreSignatureFuzzer::invokeSignature() {
32     Signature signature;
33     bool shouldUseParameterizedConstructor = mFdp->ConsumeBool();
34     if (shouldUseParameterizedConstructor) {
35         std::vector<uint8_t> signatureData = initSignatureData(mFdp.get());
36         signature.data = signatureData;
37     } else {
38         signature = Signature();
39     }
40     invokeReadWriteParcel(&signature);
41 }
42 
process(const uint8_t * data,size_t size)43 void KeystoreSignatureFuzzer::process(const uint8_t* data, size_t size) {
44     mFdp = std::make_unique<FuzzedDataProvider>(data, size);
45     invokeSignature();
46 }
47 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)48 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
49     KeystoreSignatureFuzzer keystoreSignatureFuzzer;
50     keystoreSignatureFuzzer.process(data, size);
51     return 0;
52 }
53