tools/testing/selftests/bpf/progs/bpf_iter_unix.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /* Copyright Amazon.com Inc. or its affiliates. */
   3 #include "bpf_iter.h"
   4 #include "bpf_tracing_net.h"
   5 #include <bpf/bpf_helpers.h>
   6 #include <bpf/bpf_endian.h>
   7
   8 char _license[] SEC("license") = "GPL";
   9
  10 static long sock_i_ino(const struct sock *sk)
  11 {
  12         const struct socket *sk_socket = sk->sk_socket;
  13         const struct inode *inode;
  14         unsigned long ino;
  15
  16         if (!sk_socket)
  17                 return 0;
  18
  19         inode = &container_of(sk_socket, struct socket_alloc, socket)->vfs_inode;
  20         bpf_probe_read_kernel(&ino, sizeof(ino), &inode->i_ino);
  21         return ino;
  22 }
  23
  24 SEC("iter/unix")
  25 int dump_unix(struct bpf_iter__unix *ctx)
  26 {
  27         struct unix_sock *unix_sk = ctx->unix_sk;
  28         struct sock *sk = (struct sock *)unix_sk;
  29         struct seq_file *seq;
  30         __u32 seq_num;
  31
  32         if (!unix_sk)
  33                 return 0;
  34
  35         seq = ctx->meta->seq;
  36         seq_num = ctx->meta->seq_num;
  37         if (seq_num == 0)
  38                 BPF_SEQ_PRINTF(seq, "Num               RefCount Protocol Flags    Type St    Inode Path\n");
  39
  40         BPF_SEQ_PRINTF(seq, "%pK: %08X %08X %08X %04X %02X %8lu",
  41                        unix_sk,
  42                        sk->sk_refcnt.refs.counter,
  43                        0,
  44                        sk->sk_state == TCP_LISTEN ? __SO_ACCEPTCON : 0,
  45                        sk->sk_type,
  46                        sk->sk_socket ?
  47                        (sk->sk_state == TCP_ESTABLISHED ? SS_CONNECTED : SS_UNCONNECTED) :
  48                        (sk->sk_state == TCP_ESTABLISHED ? SS_CONNECTING : SS_DISCONNECTING),
  49                        sock_i_ino(sk));
  50
  51         if (unix_sk->addr) {
  52                 if (unix_sk->addr->name->sun_path[0]) {
  53                         BPF_SEQ_PRINTF(seq, " %s", unix_sk->addr->name->sun_path);
  54                 } else {
  55                         /* The name of the abstract UNIX domain socket starts
  56                          * with '\0' and can contain '\0'.  The null bytes
  57                          * should be escaped as done in unix_seq_show().
  58                          */
  59                         __u64 i, len;
  60
  61                         len = unix_sk->addr->len - sizeof(short);
  62
  63                         BPF_SEQ_PRINTF(seq, " @");
  64
  65                         for (i = 1; i < len; i++) {
  66                                 /* unix_validate_addr() tests this upper bound. */
  67                                 if (i >= sizeof(struct sockaddr_un))
  68                                         break;
  69
  70                                 BPF_SEQ_PRINTF(seq, "%c",
  71                                                unix_sk->addr->name->sun_path[i] ?:
  72                                                '@');
  73                         }
  74                 }
  75         }
  76
  77         BPF_SEQ_PRINTF(seq, "\n");
  78
  79         return 0;
  80 }