Merge "Add cputimeinstate_fuzzer"

/******************************************************************************

 *

 * Copyright (C) 2022 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.

 *

 *****************************************************************************

 */

cc_fuzz {

    name: "cputimeinstate_fuzzer",

    srcs: [

        "cputimeinstate_fuzzer.cpp",

    ],

    static_libs: [

        "libtimeinstate",

    ],

    shared_libs: [

        "libbpf_bcc",

        "libbase",

        "libbpf_minimal",

    ],

}

/******************************************************************************

 *

 * Copyright (C) 2022 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 <fuzzer/FuzzedDataProvider.h>

#include <android-base/unique_fd.h>

#include <cputimeinstate.h>

using namespace android::bpf;

static const uint16_t MAX_VEC_SIZE = 500;

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {

    FuzzedDataProvider fdp(data, size);

    uint32_t uid = fdp.ConsumeIntegral<uint32_t>();

    uint64_t lastUpdate = fdp.ConsumeIntegral<uint64_t>();

    uint16_t aggregationKey = fdp.ConsumeIntegral<uint16_t>();

    pid_t pid = fdp.ConsumeIntegral<pid_t>();

    std::vector<uint16_t> aggregationKeys;

    uint16_t aggregationKeysSize = fdp.ConsumeIntegralInRange<size_t>(0, MAX_VEC_SIZE);

    for (uint16_t i = 0; i < aggregationKeysSize; i++) {

        aggregationKeys.push_back(fdp.ConsumeIntegral<uint16_t>());

    }

    // To randomize the API calls

     while (fdp.remaining_bytes() > 0) {

        auto func = fdp.PickValueInArray<const std::function<void()>>({

                [&]() { getUidCpuFreqTimes(uid); },

                [&]() { getUidsUpdatedCpuFreqTimes(&lastUpdate); },

                [&]() { getUidConcurrentTimes(uid);},

                [&]() { getUidsUpdatedConcurrentTimes(&lastUpdate); },

                [&]() { startAggregatingTaskCpuTimes(pid, aggregationKey); },

                [&]() { getAggregatedTaskCpuFreqTimes(pid, aggregationKeys); },

        });

        func();

    }

    return 0;

}