1 /*
2  * Copyright (c) 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 "mergewithexistprofile_fuzzer.h"
17 #include <fuzzer/FuzzedDataProvider.h>
18 #include "ecmascript/pgo_profiler/pgo_profiler_encoder.h"
19 #include "ecmascript/pgo_profiler/pgo_profiler_decoder.h"
20 
21 using namespace panda;
22 using namespace panda::ecmascript;
23 using namespace panda::ecmascript::pgo;
24 
25 namespace OHOS {
MergeWithExistProfileFuzzTest(const uint8_t* data, size_t size)26     void MergeWithExistProfileFuzzTest(const uint8_t* data, size_t size)
27     {
28         FuzzedDataProvider dataProvider(data, size);
29 
30         // generate random output path and boudlename
31         std::string outDir = "/tmp/fuzz_test_" + std::to_string(dataProvider.ConsumeIntegral<uint32_t>());
32         std::string bundleName = dataProvider.ConsumeRandomLengthString(20);
33 
34         // generat random hotnessThreshold and ApGenMode
35         uint32_t hotnessThreshold = dataProvider.ConsumeIntegral<uint32_t>();
36         PGOProfilerEncoder::ApGenMode mode = static_cast<PGOProfilerEncoder::ApGenMode>(
37             dataProvider.ConsumeIntegralInRange<int>(0, 1)
38         );
39 
40         PGOProfilerEncoder encoder(outDir, hotnessThreshold, mode);
41 
42         encoder.SetBundleName(bundleName);
43         encoder.Save();
44 
45         // cleanup
46         encoder.Destroy();
47         std::remove(outDir.c_str());
48     }
49 }
50 
51 // Fuzzer entry point.
LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)52 extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size)
53 {
54     // Run your code on data.
55     OHOS::MergeWithExistProfileFuzzTest(data, size);
56     return 0;
57 }