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 "PerfFileFormat_fuzzer.h"
17 #include "perf_file_format.h"
18
19 using namespace OHOS::Developtools::HiPerf;
20
21 namespace OHOS {
FuzzPerfFileFormat(const uint8_t *data, size_t size)22 bool FuzzPerfFileFormat(const uint8_t *data, size_t size)
23 {
24 char buf[DATA_MAX_SIZE] = { 0 };
25 if (memcpy_s(buf, sizeof(buf) - 1, data, size) != 0) { // 1 ： make sure end of '\0'
26 return false;
27 }
28 std::unique_ptr<PerfFileSectionString> perfFileSectionString =
29 std::make_unique<PerfFileSectionString>(FEATURE::HIPERF_FILES_SYMBOL, buf, size);
30 perfFileSectionString->GetBinary(buf, size);
31 std::unique_ptr<PerfFileSectionNrCpus> perfFileSectionNrCpus =
32 std::make_unique<PerfFileSectionNrCpus>(FEATURE::HIPERF_FILES_SYMBOL, buf, size);
33 perfFileSectionNrCpus->GetBinary(buf, size);
34 std::unique_ptr<PerfFileSectionU64> perfFileSectionU64 =
35 std::make_unique<PerfFileSectionU64>(FEATURE::HIPERF_FILES_SYMBOL, buf, size);
36 perfFileSectionU64->GetBinary(buf, size);
37 return 0;
38 }
39 } // namespace OHOS
40
41 /* Fuzzer entry point */
LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)42 extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
43 {
44 /* Run your code on data */
45 #ifdef DEBUG_HIPERF_FUZZ
46 ScopeDebugLevel mix(LEVEL_DEBUG, true);
47 DebugLogger::GetInstance()->Disable(false);
48 #else
49 OHOS::Developtools::HiPerf::StdoutRecord noStdOut("/dev/null", "w");
50 #endif
51
52 OHOS::FuzzPerfFileFormat(data, size);
53 return 0;
54 }
55