1 /*
2 * Copyright 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 #include "android-base/file.h"
18 #include "fuzzer/FuzzedDataProvider.h"
19 #include "input/PropertyMap.h"
20
21 static constexpr int MAX_FILE_SIZE = 256;
22 static constexpr int MAX_STR_LEN = 2048;
23 static constexpr int MAX_OPERATIONS = 1000;
24
25 static const std::vector<std::function<void(FuzzedDataProvider*, android::PropertyMap&)>>
26 operations = {
__anon7da9061f0102() 27 [](FuzzedDataProvider*, android::PropertyMap& propertyMap) -> void {
28 propertyMap.clear();
29 },
__anon7da9061f0202() 30 [](FuzzedDataProvider* dataProvider, android::PropertyMap& propertyMap) -> void {
31 std::string key = dataProvider->ConsumeRandomLengthString(MAX_STR_LEN);
32 propertyMap.getString(key);
33 },
__anon7da9061f0302() 34 [](FuzzedDataProvider* dataProvider, android::PropertyMap& /*unused*/) -> void {
35 TemporaryFile tf;
36 // Generate file contents
37 std::string contents = dataProvider->ConsumeRandomLengthString(MAX_FILE_SIZE);
38 // If we have string contents, dump them into the file.
39 // Otherwise, just leave it as an empty file.
40 if (contents.length() > 0) {
41 const char* bytes = contents.c_str();
42 android::base::WriteStringToFd(bytes, tf.fd);
43 }
44 android::PropertyMap::load(tf.path);
45 },
__anon7da9061f0402() 46 [](FuzzedDataProvider* dataProvider, android::PropertyMap& propertyMap) -> void {
47 std::string key = dataProvider->ConsumeRandomLengthString(MAX_STR_LEN);
48 std::string val = dataProvider->ConsumeRandomLengthString(MAX_STR_LEN);
49 propertyMap.addProperty(key, val);
50 },
51 };
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)52 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
53 FuzzedDataProvider dataProvider(data, size);
54 android::PropertyMap propertyMap;
55
56 int opsRun = 0;
57 while (dataProvider.remaining_bytes() > 0 && opsRun++ < MAX_OPERATIONS) {
58 uint8_t op = dataProvider.ConsumeIntegralInRange<uint8_t>(0, operations.size() - 1);
59 operations[op](&dataProvider, propertyMap);
60 }
61 return 0;
62 }
63