| /* |
| * Copyright 2020 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "IoPerfCollection.h" |
| #include "UidIoStats.h" |
| |
| #include <android-base/file.h> |
| #include <fuzzer/FuzzedDataProvider.h> |
| #include <iostream> |
| |
| namespace android { |
| namespace automotive { |
| namespace watchdog { |
| |
| using ::android::base::WriteStringToFile; |
| |
| struct statsInfo { |
| std::string instr[13]; |
| int len[13]; |
| }; |
| |
| statsInfo info = {{"cpu", "procs_running", "procs_blocked", "cpu0", "cpu1", "cpu2", "cpu3", "intr", |
| "ctxt", "btime", "process", "procs_sleeping", "softirq"}, |
| {10, 1, 1, 10, 10, 10, 10, 31, 1, 1, 1, 1, 11}}; |
| |
| extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, std::size_t size) { |
| FuzzedDataProvider fdp(data, size); |
| std::string uidIoStatsSnapshot; |
| std::string procStatsSnapshot; |
| |
| // Prepare for UidIOStats data |
| int count = 0; |
| while (fdp.remaining_bytes() > (size / 2)) { |
| uint64_t val = fdp.ConsumeIntegral<uint64_t>(); |
| uidIoStatsSnapshot += (((count % 11) == 0 ? "" : " ") + std::to_string(val)); |
| if (count > 0 && count % 11 == 10) { |
| uidIoStatsSnapshot += "\n"; |
| } |
| count++; |
| } |
| |
| // Prepare for ProcStats data |
| while (fdp.remaining_bytes() > 5) { |
| uint32_t choose = fdp.ConsumeIntegralInRange<uint32_t>(0, 3); |
| uint32_t idx = choose; |
| switch (choose) { |
| case 0: |
| case 1: |
| case 2: |
| break; |
| case 3: |
| idx = fdp.ConsumeIntegralInRange<uint32_t>(3, 12); |
| break; |
| default: |
| break; |
| } |
| procStatsSnapshot += info.instr[idx]; |
| for (int i = 0; i < info.len[idx]; i++) { |
| uint64_t val = fdp.ConsumeIntegral<uint64_t>(); |
| procStatsSnapshot += " " + std::to_string(val); |
| } |
| procStatsSnapshot += "\n"; |
| } |
| |
| if (uidIoStatsSnapshot.size() > 0 && procStatsSnapshot.size() > 0) { |
| // Test UidIoStats |
| TemporaryFile tf1; |
| WriteStringToFile(uidIoStatsSnapshot, tf1.path); |
| UidIoStats uidIoStats(tf1.path); |
| assert(uidIoStats.enabled() == true); |
| uidIoStats.collect(); |
| // Test ProcStats |
| TemporaryFile tf2; |
| WriteStringToFile(procStatsSnapshot, tf2.path); |
| ProcStat procStat(tf2.path); |
| assert(procStat.enabled() == true); |
| procStat.collect(); |
| } |
| return 0; |
| } |
| |
| } // namespace watchdog |
| } // namespace automotive |
| } // namespace android |