1 /* 2 * MM Events - eBPF programs 3 * 4 * Copyright 2024 The Android Open Source Project 5 * 6 * Licensed under the Apache License, Version 2.0 (the "License"); 7 * you may not use this file except in compliance with the License 8 * You may obtain a copy of the License at 9 * http://www.apache.org/licenses/LICENSE-2.0 10 * 11 * Unless required by applicable law or agreed to in writing, software 12 * distributed under the License is distributed on an "AS IS" BASIS, 13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14 * See the License for the specific language governing permissions and 15 * limitations under the License. 16 */ 17 #include <string.h> 18 19 #include <linux/bpf_perf_event.h> 20 21 #include <memevents/bpf_helpers.h> 22 #include <memevents/bpf_types.h> 23 24 DEFINE_BPF_RINGBUF(ams_rb, struct mem_event_t, MEM_EVENTS_RINGBUF_SIZE, DEFAULT_BPF_MAP_UID, 25 AID_SYSTEM, 0660) 26 27 DEFINE_BPF_RINGBUF(lmkd_rb, struct mem_event_t, MEM_EVENTS_RINGBUF_SIZE, DEFAULT_BPF_MAP_UID, 28 AID_SYSTEM, 0660) 29 30 DEFINE_BPF_PROG_KVER("tracepoint/oom/mark_victim/ams", AID_ROOT, AID_SYSTEM, tp_ams, KVER_5_8) 31 (struct mark_victim_args* args) { 32 unsigned long long timestamp_ns = bpf_ktime_get_ns(); 33 struct mem_event_t* data = bpf_ams_rb_reserve(); 34 if (data == NULL) return 1; 35 36 data->type = MEM_EVENT_OOM_KILL; 37 data->event_data.oom_kill.pid = args->pid; 38 data->event_data.oom_kill.oom_score_adj = args->oom_score_adj; 39 data->event_data.oom_kill.uid = args->uid; 40 data->event_data.oom_kill.timestamp_ms = timestamp_ns / 1000000; // Convert to milliseconds 41 data->event_data.oom_kill.total_vm_kb = args->total_vm; 42 data->event_data.oom_kill.anon_rss_kb = args->anon_rss; 43 data->event_data.oom_kill.file_rss_kb = args->file_rss; 44 data->event_data.oom_kill.shmem_rss_kb = args->shmem_rss; 45 data->event_data.oom_kill.pgtables_kb = args->pgtables; 46 47 read_str((char*)args, args->__data_loc_comm, data->event_data.oom_kill.process_name, 48 MEM_EVENT_PROC_NAME_LEN); 49 50 bpf_ams_rb_submit(data); 51 52 return 0; 53 } 54 55 DEFINE_BPF_PROG_KVER("tracepoint/vmscan/mm_vmscan_direct_reclaim_begin/lmkd", AID_ROOT, AID_SYSTEM, 56 tp_lmkd_dr_start, KVER_5_8) 57 (struct direct_reclaim_begin_args* __unused args) { 58 struct mem_event_t* data = bpf_lmkd_rb_reserve(); 59 if (data == NULL) return 1; 60 61 data->type = MEM_EVENT_DIRECT_RECLAIM_BEGIN; 62 63 bpf_lmkd_rb_submit(data); 64 65 return 0; 66 } 67 68 DEFINE_BPF_PROG_KVER("tracepoint/vmscan/mm_vmscan_direct_reclaim_end/lmkd", AID_ROOT, AID_SYSTEM, 69 tp_lmkd_dr_end, KVER_5_8) 70 (struct direct_reclaim_end_args* __unused args) { 71 struct mem_event_t* data = bpf_lmkd_rb_reserve(); 72 if (data == NULL) return 1; 73 74 data->type = MEM_EVENT_DIRECT_RECLAIM_END; 75 76 bpf_lmkd_rb_submit(data); 77 78 return 0; 79 } 80 81 DEFINE_BPF_PROG_KVER("tracepoint/vmscan/mm_vmscan_kswapd_wake/lmkd", AID_ROOT, AID_SYSTEM, 82 tp_lmkd_kswapd_wake, KVER_5_8) 83 (struct kswapd_wake_args* args) { 84 struct mem_event_t* data = bpf_lmkd_rb_reserve(); 85 if (data == NULL) return 1; 86 87 data->type = MEM_EVENT_KSWAPD_WAKE; 88 data->event_data.kswapd_wake.node_id = args->nid; 89 data->event_data.kswapd_wake.zone_id = args->zid; 90 data->event_data.kswapd_wake.alloc_order = args->order; 91 92 bpf_lmkd_rb_submit(data); 93 94 return 0; 95 } 96 97 DEFINE_BPF_PROG_KVER("tracepoint/vmscan/mm_vmscan_kswapd_sleep/lmkd", AID_ROOT, AID_SYSTEM, 98 tp_lmkd_kswapd_sleep, KVER_5_8) 99 (struct kswapd_sleep_args* args) { 100 struct mem_event_t* data = bpf_lmkd_rb_reserve(); 101 if (data == NULL) return 1; 102 103 data->type = MEM_EVENT_KSWAPD_SLEEP; 104 data->event_data.kswapd_wake.node_id = args->nid; 105 106 bpf_lmkd_rb_submit(data); 107 108 return 0; 109 } 110 111 // bpf_probe_read_str is GPL only symbol 112 LICENSE("GPL"); 113