1 /*
2  * Copyright (C) 2023 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 #pragma once
18 
19 namespace android {
20 namespace uprobestats {
21 namespace bpf {
22 
23 int bpfPerfEventOpen(const char *filename, int offset, int pid,
24                      const char *bpfProgramPath);
25 
26 std::vector<int32_t> consumeRingBuf(const char *mapPath);
27 
28 // TODO: share this struct with bpf
29 struct CallResult {
30   unsigned long pc;
31   unsigned long regs[10];
32 };
33 
34 struct CallTimestamp {
35   unsigned int event;
36   unsigned long timestampNs;
37 };
38 
39 struct SetUidTempAllowlistStateRecord {
40   __u64 uid;
41   bool onAllowlist;
42 };
43 
44 struct UpdateDeviceIdleTempAllowlistRecord {
45   int changing_uid;
46   bool adding;
47   long duration_ms;
48   int type;
49   int reason_code;
50   char reason[256];
51   int calling_uid;
52 };
53 
54 template <typename T>
55 std::vector<T> pollRingBuf(const char *mapPath, int timeoutMs);
56 
57 void printRingBuf(const char *mapPath);
58 
59 } // namespace bpf
60 } // namespace uprobestats
61 } // namespace android
62