18c2ecf20Sopenharmony_ci/* Copyright 2016 Netflix, Inc. 28c2ecf20Sopenharmony_ci * 38c2ecf20Sopenharmony_ci * This program is free software; you can redistribute it and/or 48c2ecf20Sopenharmony_ci * modify it under the terms of version 2 of the GNU General Public 58c2ecf20Sopenharmony_ci * License as published by the Free Software Foundation. 68c2ecf20Sopenharmony_ci */ 78c2ecf20Sopenharmony_ci#include <linux/version.h> 88c2ecf20Sopenharmony_ci#include <linux/ptrace.h> 98c2ecf20Sopenharmony_ci#include <uapi/linux/bpf.h> 108c2ecf20Sopenharmony_ci#include <uapi/linux/bpf_perf_event.h> 118c2ecf20Sopenharmony_ci#include <bpf/bpf_helpers.h> 128c2ecf20Sopenharmony_ci#include <bpf/bpf_tracing.h> 138c2ecf20Sopenharmony_ci 148c2ecf20Sopenharmony_ci#define MAX_IPS 8192 158c2ecf20Sopenharmony_ci 168c2ecf20Sopenharmony_cistruct { 178c2ecf20Sopenharmony_ci __uint(type, BPF_MAP_TYPE_HASH); 188c2ecf20Sopenharmony_ci __type(key, u64); 198c2ecf20Sopenharmony_ci __type(value, u32); 208c2ecf20Sopenharmony_ci __uint(max_entries, MAX_IPS); 218c2ecf20Sopenharmony_ci} ip_map SEC(".maps"); 228c2ecf20Sopenharmony_ci 238c2ecf20Sopenharmony_ciSEC("perf_event") 248c2ecf20Sopenharmony_ciint do_sample(struct bpf_perf_event_data *ctx) 258c2ecf20Sopenharmony_ci{ 268c2ecf20Sopenharmony_ci u64 ip; 278c2ecf20Sopenharmony_ci u32 *value, init_val = 1; 288c2ecf20Sopenharmony_ci 298c2ecf20Sopenharmony_ci ip = PT_REGS_IP(&ctx->regs); 308c2ecf20Sopenharmony_ci value = bpf_map_lookup_elem(&ip_map, &ip); 318c2ecf20Sopenharmony_ci if (value) 328c2ecf20Sopenharmony_ci *value += 1; 338c2ecf20Sopenharmony_ci else 348c2ecf20Sopenharmony_ci /* E2BIG not tested for this example only */ 358c2ecf20Sopenharmony_ci bpf_map_update_elem(&ip_map, &ip, &init_val, BPF_NOEXIST); 368c2ecf20Sopenharmony_ci 378c2ecf20Sopenharmony_ci return 0; 388c2ecf20Sopenharmony_ci} 398c2ecf20Sopenharmony_cichar _license[] SEC("license") = "GPL"; 40