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