1 /*
2  * Copyright (c) 2023-2024 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 "jsnapiset_fuzzer.h"
17 #include "ecmascript/base/string_helper.h"
18 #include "ecmascript/ecma_vm.h"
19 #include "ecmascript/napi/include/dfx_jsnapi.h"
20 #include "ecmascript/napi/include/jsnapi.h"
21 
22 using namespace panda;
23 using namespace panda::ecmascript;
24 
25 namespace OHOS {
JSNApiSetAssetPathFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)26 void JSNApiSetAssetPathFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)
27 {
28     RuntimeOption option;
29     option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
30     EcmaVM *vm = JSNApi::CreateJSVM(option);
31     if (size <= 0) {
32         return;
33     }
34     std::string str = "/data/storage/el1/bundle/moduleName/ets/modules.abc";
35     JSNApi::SetAssetPath(vm, str);
36     JSNApi::DestroyJSVM(vm);
37 }
38 
JSNApiSetBundleFuzzTest(const uint8_t *data, size_t size)39 void JSNApiSetBundleFuzzTest(const uint8_t *data, size_t size)
40 {
41     RuntimeOption option;
42     option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
43     EcmaVM *vm = JSNApi::CreateJSVM(option);
44     if (data == nullptr || size <= 0) {
45         LOG_ECMA(ERROR) << "illegal input!";
46         return;
47     }
48     uint8_t *ptr = nullptr;
49     ptr = const_cast<uint8_t *>(data);
50     JSNApi::SetBundle(vm, (bool)size);
51     JSNApi::DestroyJSVM(vm);
52 }
53 
JSNApiSetHostEnqueueJobFuzzTest(const uint8_t *data, size_t size)54 void JSNApiSetHostEnqueueJobFuzzTest(const uint8_t *data, size_t size)
55 {
56     RuntimeOption option;
57     option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
58     EcmaVM *vm = JSNApi::CreateJSVM(option);
59     if (data == nullptr || size <= 0) {
60         LOG_ECMA(ERROR) << "illegal input!";
61         return;
62     }
63     Local<JSValueRef> key = StringRef::NewFromUtf8(vm, (char *)data, (int)size);
64     JSNApi::SetHostEnqueueJob(vm, key);
65     JSNApi::DestroyJSVM(vm);
66 }
67 
JSNApiSetMockModuleListFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)68 void JSNApiSetMockModuleListFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)
69 {
70     RuntimeOption option;
71     option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
72     EcmaVM *vm = JSNApi::CreateJSVM(option);
73     if (size <= 0) {
74         return;
75     }
76     std::map<std::string, std::string> str = { { "10", "20" } };
77     JSNApi::SetMockModuleList(vm, str);
78     JSNApi::DestroyJSVM(vm);
79 }
80 
JSNApiSetSourceMapTranslateCallbackFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)81 void JSNApiSetSourceMapTranslateCallbackFuzzTest([[maybe_unused]]const uint8_t *data, size_t size)
82 {
83     RuntimeOption option;
84     option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
85     EcmaVM *vm = JSNApi::CreateJSVM(option);
86     if (size <= 0) {
87         return;
88     }
89     SourceMapTranslateCallback tag { nullptr };
90     JSNApi::SetSourceMapTranslateCallback(vm, tag);
91     vm->GetSourceMapTranslateCallback();
92     JSNApi::DestroyJSVM(vm);
93 }
94 }
95 
96 // Fuzzer entry point.
LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)97 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
98 {
99     // Run your code on data.
100     OHOS::JSNApiSetAssetPathFuzzTest(data, size);
101     OHOS::JSNApiSetBundleFuzzTest(data, size);
102     OHOS::JSNApiSetHostEnqueueJobFuzzTest(data, size);
103     OHOS::JSNApiSetMockModuleListFuzzTest(data, size);
104     OHOS::JSNApiSetSourceMapTranslateCallbackFuzzTest(data, size);
105     return 0;
106 }