1 /*
2  * Copyright (c) 2023 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 "jsnapisethostpromiserejectiontracker_fuzzer.h"
17 
18 #include "ecmascript/napi/include/jsnapi.h"
19 
20 using namespace panda;
21 using namespace panda::ecmascript;
22 
23 namespace OHOS {
JSNApiSetHostPromiseRejectionTrackerFuzzTest(const uint8_t* data, size_t size)24     void JSNApiSetHostPromiseRejectionTrackerFuzzTest(const uint8_t* data, size_t size)
25     {
26         RuntimeOption option;
27         option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
28         EcmaVM *vm = JSNApi::CreateJSVM(option);
29         if (size <= 0) {
30             return;
31         }
32         JSNApi::SetHostPromiseRejectionTracker(vm, (void *)(data + size), (void *)data);
33         JSNApi::DestroyJSVM(vm);
34     }
35 
JSNApiSetNativePtrGetterFuzzTest(const uint8_t* data, size_t size)36     void JSNApiSetNativePtrGetterFuzzTest(const uint8_t* data, size_t size)
37     {
38         RuntimeOption option;
39         option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
40         EcmaVM *vm = JSNApi::CreateJSVM(option);
41         if (size <= 0) {
42             return;
43         }
44         JSNApi::SetNativePtrGetter(vm, (void*)data);
45         JSNApi::DestroyJSVM(vm);
46     }
47 
JSNApiSetHostResolveBufferTrackerFuzzTest(const uint8_t* data, size_t size)48     void JSNApiSetHostResolveBufferTrackerFuzzTest(const uint8_t* data, size_t size)
49     {
50         RuntimeOption option;
51         option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
52         EcmaVM *vm = JSNApi::CreateJSVM(option);
53         if (size <= 0) {
54             return;
55         }
56         uint8_t* ptr = nullptr;
57         ptr = const_cast<uint8_t*>(data);
58         JSNApi::SetHostResolveBufferTracker(vm,
59             [&](std::string, uint8_t **, size_t *, std::string &) -> bool { return true; });
60         JSNApi::DestroyJSVM(vm);
61     }
62 
JSNApiSetUnloalNativeModuleCallbackFuzzTest(const uint8_t* data, size_t size)63     void JSNApiSetUnloalNativeModuleCallbackFuzzTest(const uint8_t* data, size_t size)
64     {
65         RuntimeOption option;
66         option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
67         EcmaVM *vm = JSNApi::CreateJSVM(option);
68         if (size <= 0) {
69             return;
70         }
71         uint8_t* ptr = nullptr;
72         ptr = const_cast<uint8_t*>(data);
73         JSNApi::SetUnloadNativeModuleCallback(vm, [&](const std::string &) -> bool { return true; });
74         JSNApi::DestroyJSVM(vm);
75     }
76 }
77 
78 // Fuzzer entry point.
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)79 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
80 {
81     // Run your code on data.
82     OHOS::JSNApiSetHostPromiseRejectionTrackerFuzzTest(data, size);
83     OHOS::JSNApiSetNativePtrGetterFuzzTest(data, size);
84     OHOS::JSNApiSetHostResolveBufferTrackerFuzzTest(data, size);
85     OHOS::JSNApiSetUnloalNativeModuleCallbackFuzzTest(data, size);
86     return 0;
87 }