1406c557eSAlexei Starovoitov // SPDX-License-Identifier: GPL-2.0
2406c557eSAlexei Starovoitov /* Copyright (c) 2021 Facebook */
3406c557eSAlexei Starovoitov 
4406c557eSAlexei Starovoitov #include "vmlinux.h"
5406c557eSAlexei Starovoitov #include <bpf/bpf_helpers.h>
6406c557eSAlexei Starovoitov #include <bpf/bpf_tracing.h>
7406c557eSAlexei Starovoitov 
8406c557eSAlexei Starovoitov char _license[] SEC("license") = "GPL";
9406c557eSAlexei Starovoitov 
10406c557eSAlexei Starovoitov struct {
11406c557eSAlexei Starovoitov 	__uint(type, BPF_MAP_TYPE_HASH);
12406c557eSAlexei Starovoitov 	__uint(max_entries, 1);
13406c557eSAlexei Starovoitov 	__type(key, int);
14406c557eSAlexei Starovoitov 	__type(value, long);
15406c557eSAlexei Starovoitov } hash1 SEC(".maps");
16406c557eSAlexei Starovoitov 
17406c557eSAlexei Starovoitov struct {
18406c557eSAlexei Starovoitov 	__uint(type, BPF_MAP_TYPE_HASH);
19406c557eSAlexei Starovoitov 	__uint(max_entries, 1);
20406c557eSAlexei Starovoitov 	__type(key, int);
21406c557eSAlexei Starovoitov 	__type(value, long);
22406c557eSAlexei Starovoitov } hash2 SEC(".maps");
23406c557eSAlexei Starovoitov 
24406c557eSAlexei Starovoitov int pass1 = 0;
25406c557eSAlexei Starovoitov int pass2 = 0;
26406c557eSAlexei Starovoitov 
27*189c83bdSJiri Olsa SEC("fentry/htab_map_delete_elem")
BPF_PROG(on_delete,struct bpf_map * map)28*189c83bdSJiri Olsa int BPF_PROG(on_delete, struct bpf_map *map)
29406c557eSAlexei Starovoitov {
30406c557eSAlexei Starovoitov 	int key = 0;
31406c557eSAlexei Starovoitov 
32406c557eSAlexei Starovoitov 	if (map == (void *)&hash1) {
33406c557eSAlexei Starovoitov 		pass1++;
34406c557eSAlexei Starovoitov 		return 0;
35406c557eSAlexei Starovoitov 	}
36406c557eSAlexei Starovoitov 	if (map == (void *)&hash2) {
37406c557eSAlexei Starovoitov 		pass2++;
38*189c83bdSJiri Olsa 		bpf_map_delete_elem(&hash2, &key);
39406c557eSAlexei Starovoitov 		return 0;
40406c557eSAlexei Starovoitov 	}
41406c557eSAlexei Starovoitov 
42406c557eSAlexei Starovoitov 	return 0;
43406c557eSAlexei Starovoitov }
44