xref: /aosp_15_r20/bootable/recovery/tests/fuzz/verify_package_fuzzer.cpp (revision e7c364b630b241adcb6c7726a21055250b91fdac)
1*e7c364b6SAndroid Build Coastguard Worker /*
2*e7c364b6SAndroid Build Coastguard Worker  * Copyright (C) 2020 The Android Open Source Project
3*e7c364b6SAndroid Build Coastguard Worker  *
4*e7c364b6SAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*e7c364b6SAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*e7c364b6SAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*e7c364b6SAndroid Build Coastguard Worker  *
8*e7c364b6SAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*e7c364b6SAndroid Build Coastguard Worker  *
10*e7c364b6SAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*e7c364b6SAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*e7c364b6SAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*e7c364b6SAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*e7c364b6SAndroid Build Coastguard Worker  * limitations under the License.
15*e7c364b6SAndroid Build Coastguard Worker  */
16*e7c364b6SAndroid Build Coastguard Worker 
17*e7c364b6SAndroid Build Coastguard Worker #include "fuzzer/FuzzedDataProvider.h"
18*e7c364b6SAndroid Build Coastguard Worker 
19*e7c364b6SAndroid Build Coastguard Worker #include "install/install.h"
20*e7c364b6SAndroid Build Coastguard Worker #include "otautil/package.h"
21*e7c364b6SAndroid Build Coastguard Worker #include "recovery_ui/stub_ui.h"
22*e7c364b6SAndroid Build Coastguard Worker 
CreatePackage(std::vector<uint8_t> & content)23*e7c364b6SAndroid Build Coastguard Worker std::unique_ptr<Package> CreatePackage(std::vector<uint8_t>& content) {
24*e7c364b6SAndroid Build Coastguard Worker   return Package::CreateMemoryPackage(content, [](float) -> void {});
25*e7c364b6SAndroid Build Coastguard Worker }
26*e7c364b6SAndroid Build Coastguard Worker 
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)27*e7c364b6SAndroid Build Coastguard Worker extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
28*e7c364b6SAndroid Build Coastguard Worker   FuzzedDataProvider data_provider(data, size);
29*e7c364b6SAndroid Build Coastguard Worker   auto package_contents = data_provider.ConsumeRemainingBytes<uint8_t>();
30*e7c364b6SAndroid Build Coastguard Worker   if (package_contents.size() == 0) {
31*e7c364b6SAndroid Build Coastguard Worker     return 0;
32*e7c364b6SAndroid Build Coastguard Worker   }
33*e7c364b6SAndroid Build Coastguard Worker   auto package = CreatePackage(package_contents);
34*e7c364b6SAndroid Build Coastguard Worker   StubRecoveryUI ui;
35*e7c364b6SAndroid Build Coastguard Worker   verify_package(package.get(), &ui);
36*e7c364b6SAndroid Build Coastguard Worker   return 0;
37*e7c364b6SAndroid Build Coastguard Worker }
38