1// SPDX-License-Identifier: GPL-2.0-only
  2/* Copyright (c) 2017 Facebook
  3 */
  4#include <uapi/linux/bpf.h>
  5#include <bpf/bpf_helpers.h>
  6
  7#if !defined(__aarch64__)
  8struct syscalls_enter_open_args {
  9	unsigned long long unused;
 10	long syscall_nr;
 11	long filename_ptr;
 12	long flags;
 13	long mode;
 14};
 15#endif
 16
 17struct syscalls_exit_open_args {
 18	unsigned long long unused;
 19	long syscall_nr;
 20	long ret;
 21};
 22
 23struct syscalls_enter_open_at_args {
 24	unsigned long long unused;
 25	long syscall_nr;
 26	long long dfd;
 27	long filename_ptr;
 28	long flags;
 29	long mode;
 30};
 31
 32struct {
 33	__uint(type, BPF_MAP_TYPE_ARRAY);
 34	__type(key, u32);
 35	__type(value, u32);
 36	__uint(max_entries, 1);
 37} enter_open_map SEC(".maps");
 38
 39struct {
 40	__uint(type, BPF_MAP_TYPE_ARRAY);
 41	__type(key, u32);
 42	__type(value, u32);
 43	__uint(max_entries, 1);
 44} exit_open_map SEC(".maps");
 45
 46static __always_inline void count(void *map)
 47{
 48	u32 key = 0;
 49	u32 *value, init_val = 1;
 50
 51	value = bpf_map_lookup_elem(map, &key);
 52	if (value)
 53		*value += 1;
 54	else
 55		bpf_map_update_elem(map, &key, &init_val, BPF_NOEXIST);
 56}
 57
 58#if !defined(__aarch64__)
 59SEC("tracepoint/syscalls/sys_enter_open")
 60int trace_enter_open(struct syscalls_enter_open_args *ctx)
 61{
 62	count(&enter_open_map);
 63	return 0;
 64}
 65#endif
 66
 67SEC("tracepoint/syscalls/sys_enter_openat")
 68int trace_enter_open_at(struct syscalls_enter_open_at_args *ctx)
 69{
 70	count(&enter_open_map);
 71	return 0;
 72}
 73
 74SEC("tracepoint/syscalls/sys_enter_openat2")
 75int trace_enter_open_at2(struct syscalls_enter_open_at_args *ctx)
 76{
 77	count(&enter_open_map);
 78	return 0;
 79}
 80
 81#if !defined(__aarch64__)
 82SEC("tracepoint/syscalls/sys_exit_open")
 83int trace_enter_exit(struct syscalls_exit_open_args *ctx)
 84{
 85	count(&exit_open_map);
 86	return 0;
 87}
 88#endif
 89
 90SEC("tracepoint/syscalls/sys_exit_openat")
 91int trace_enter_exit_at(struct syscalls_exit_open_args *ctx)
 92{
 93	count(&exit_open_map);
 94	return 0;
 95}
 96
 97SEC("tracepoint/syscalls/sys_exit_openat2")
 98int trace_enter_exit_at2(struct syscalls_exit_open_args *ctx)
 99{
100	count(&exit_open_map);
101	return 0;
102}