1 /*
2 * Copyright (C) 2021 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "core_jni_helpers.h"
18
19 #include <cputimeinstate.h>
20
21 namespace android {
22
KernelCpuBpfTracking_isSupported(JNIEnv *,jobject)23 static jboolean KernelCpuBpfTracking_isSupported(JNIEnv *, jobject) {
24 return android::bpf::isTrackingUidTimesSupported() ? JNI_TRUE : JNI_FALSE;
25 }
26
KernelCpuBpfTracking_startTrackingInternal(JNIEnv *,jobject)27 static jboolean KernelCpuBpfTracking_startTrackingInternal(JNIEnv *, jobject) {
28 return android::bpf::startTrackingUidTimes();
29 }
30
KernelCpuBpfTracking_getFreqsInternal(JNIEnv * env,jobject)31 static jlongArray KernelCpuBpfTracking_getFreqsInternal(JNIEnv *env, jobject) {
32 auto freqs = android::bpf::getCpuFreqs();
33 if (!freqs) return nullptr;
34
35 std::vector<uint64_t> allFreqs;
36 for (const auto &vec : *freqs) std::copy(vec.begin(), vec.end(), std::back_inserter(allFreqs));
37
38 auto ar = env->NewLongArray(allFreqs.size());
39 if (ar != nullptr) {
40 env->SetLongArrayRegion(ar, 0, allFreqs.size(),
41 reinterpret_cast<const jlong *>(allFreqs.data()));
42 }
43 return ar;
44 }
45
KernelCpuBpfTracking_getFreqsClustersInternal(JNIEnv * env,jobject)46 static jintArray KernelCpuBpfTracking_getFreqsClustersInternal(JNIEnv *env, jobject) {
47 auto freqs = android::bpf::getCpuFreqs();
48 if (!freqs) return nullptr;
49
50 std::vector<uint32_t> freqsClusters;
51 uint32_t clusters = freqs->size();
52 for (uint32_t c = 0; c < clusters; ++c) {
53 freqsClusters.insert(freqsClusters.end(), (*freqs)[c].size(), c);
54 }
55
56 auto ar = env->NewIntArray(freqsClusters.size());
57 if (ar != nullptr) {
58 env->SetIntArrayRegion(ar, 0, freqsClusters.size(),
59 reinterpret_cast<const jint *>(freqsClusters.data()));
60 }
61 return ar;
62 }
63
64 static const JNINativeMethod methods[] = {
65 {"isSupported", "()Z", (void *)KernelCpuBpfTracking_isSupported},
66 {"startTrackingInternal", "()Z", (void *)KernelCpuBpfTracking_startTrackingInternal},
67 {"getFreqsInternal", "()[J", (void *)KernelCpuBpfTracking_getFreqsInternal},
68 {"getFreqsClustersInternal", "()[I", (void *)KernelCpuBpfTracking_getFreqsClustersInternal},
69 };
70
register_com_android_internal_os_KernelCpuBpfTracking(JNIEnv * env)71 int register_com_android_internal_os_KernelCpuBpfTracking(JNIEnv *env) {
72 return RegisterMethodsOrDie(env, "com/android/internal/os/KernelCpuBpfTracking", methods,
73 NELEM(methods));
74 }
75
76 } // namespace android
77