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 "dumpheapsnapshot2_fuzzer.h"
17 
18 #include "ecmascript/napi/include/jsnapi.h"
19 #include "ecmascript/napi/include/dfx_jsnapi.h"
20 #include "ecmascript/ecma_string-inl.h"
21 
22 using namespace panda;
23 using namespace panda::ecmascript;
24 
25 #define MAXBYTELEN sizeof(int)
26 
27 namespace OHOS {
DumpHeapSnapshot2FuzzTest(const uint8_t* data, size_t size)28     void DumpHeapSnapshot2FuzzTest(const uint8_t* data, size_t size)
29     {
30         RuntimeOption option;
31         option.SetLogLevel(RuntimeOption::LOG_LEVEL::ERROR);
32         EcmaVM *vm = JSNApi::CreateJSVM(option);
33         DumpFormat dumpFormat = DumpFormat::JSON;
34         if (size <= 0) {
35             return;
36         }
37         if (size > MAXBYTELEN) {
38             size = MAXBYTELEN;
39         }
40         if (memcpy_s(&dumpFormat, MAXBYTELEN, data, size) != 0) {
41             std::cout << "memcpy_s failed!";
42             UNREACHABLE();
43         }
44         DumpSnapShotOption dumpOption;
45         dumpOption.dumpFormat = dumpFormat;
46         dumpOption.isVmMode = true;
47         dumpOption.isPrivate = false;
48         dumpOption.captureNumericValue = false;
49         std::string path(data, data + size);
50         DFXJSNApi::DumpHeapSnapshot(vm, path, dumpOption);
51         JSNApi::DestroyJSVM(vm);
52     }
53 }
54 
55 // Fuzzer entry point.
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)56 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
57 {
58     // Run your code on data.
59     OHOS::DumpHeapSnapshot2FuzzTest(data, size);
60     return 0;
61 }