1 /*
2  * Copyright (c) 2022 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #include "loadfromconfigfile_fuzzer.h"
17 
18 #include "plugin_switch.h"
19 #include "plugin_mgr.h"
20 
21 namespace OHOS {
22 namespace ResourceSchedule {
23     const std::string fuzzedFile = "mySwitchFuzzed.xml";
24     const int32_t FILE_RETURN_SUCCESS = 1;
25 
DoSomethingInterestingWithMyAPI(const uint8_t * data,size_t size)26     bool DoSomethingInterestingWithMyAPI(const uint8_t* data, size_t size)
27     {
28         // generate fuzzed xml
29         FILE *pFile = fopen(fuzzedFile.c_str(), "wb");
30         if (!pFile) {
31             return false;
32         }
33 
34         int32_t retCode = fwrite(reinterpret_cast<const void*>(data), size, 1, pFile); // 1 means count=1
35         if (retCode < FILE_RETURN_SUCCESS) {
36             (void)fclose(pFile);
37             pFile = nullptr;
38             return false;
39         }
40 
41         (void)fclose(pFile);
42         pFile = nullptr;
43 
44         std::shared_ptr<PluginSwitch> pluginSwitch = std::make_shared<PluginSwitch>();
45         std::vector<std::string> contents;
46         PluginMgr::GetInstance().GetConfigContent(-1, fuzzedFile, contents);
47         bool ret = false;
48         for (auto content : contents) {
49             if (pluginSwitch->LoadFromConfigContent(content)) {
50                 ret = true;
51             }
52         }
53         return ret;
54     }
55 } // namespace ResourceSchedule
56 } // namespace OHOS
57 
58 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t * data,size_t size)59 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
60 {
61     /* Run your code on data */
62     OHOS::ResourceSchedule::DoSomethingInterestingWithMyAPI(data, size);
63     return 0;
64 }
65