hc
2023-12-06 d38611ca164021d018c1b23eee65bbebc09c63e0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2020 Facebook */
#include "bpf_iter.h"
#include "bpf_tracing_net.h"
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
 
char _license[] SEC("license") = "GPL";
 
static __attribute__((noinline)) struct inode *SOCK_INODE(struct socket *socket)
{
   return &container_of(socket, struct socket_alloc, socket)->vfs_inode;
}
 
SEC("iter/netlink")
int dump_netlink(struct bpf_iter__netlink *ctx)
{
   struct seq_file *seq = ctx->meta->seq;
   struct netlink_sock *nlk = ctx->sk;
   unsigned long group, ino;
   struct inode *inode;
   struct socket *sk;
   struct sock *s;
 
   if (nlk == (void *)0)
       return 0;
 
   if (ctx->meta->seq_num == 0)
       BPF_SEQ_PRINTF(seq, "sk               Eth Pid        Groups   "
                   "Rmem     Wmem     Dump  Locks    Drops    "
                   "Inode\n");
 
   s = &nlk->sk;
   BPF_SEQ_PRINTF(seq, "%pK %-3d ", s, s->sk_protocol);
 
   if (!nlk->groups)  {
       group = 0;
   } else {
       /* FIXME: temporary use bpf_probe_read_kernel here, needs
        * verifier support to do direct access.
        */
       bpf_probe_read_kernel(&group, sizeof(group), &nlk->groups[0]);
   }
   BPF_SEQ_PRINTF(seq, "%-10u %08x %-8d %-8d %-5d %-8d ",
              nlk->portid, (u32)group,
              s->sk_rmem_alloc.counter,
              s->sk_wmem_alloc.refs.counter - 1,
              nlk->cb_running, s->sk_refcnt.refs.counter);
 
   sk = s->sk_socket;
   if (!sk) {
       ino = 0;
   } else {
       /* FIXME: container_of inside SOCK_INODE has a forced
        * type conversion, and direct access cannot be used
        * with current verifier.
        */
       inode = SOCK_INODE(sk);
       bpf_probe_read_kernel(&ino, sizeof(ino), &inode->i_ino);
   }
   BPF_SEQ_PRINTF(seq, "%-8u %-8lu\n", s->sk_drops.counter, ino);
 
   return 0;
}