include/trace/events/sock.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #undef TRACE_SYSTEM
   3 #define TRACE_SYSTEM sock
   4
   5 #if !defined(_TRACE_SOCK_H) || defined(TRACE_HEADER_MULTI_READ)
   6 #define _TRACE_SOCK_H
   7
   8 #include <net/sock.h>
   9 #include <net/ipv6.h>
  10 #include <linux/tracepoint.h>
  11 #include <linux/ipv6.h>
  12 #include <linux/tcp.h>
  13 #include <trace/events/net_probe_common.h>
  14
  15 #define family_names                    \
  16                 EM(AF_INET)                             \
  17                 EMe(AF_INET6)
  18
  19 /* The protocol traced by inet_sock_set_state */
  20 #define inet_protocol_names             \
  21                 EM(IPPROTO_TCP)                 \
  22                 EM(IPPROTO_SCTP)                \
  23                 EMe(IPPROTO_MPTCP)
  24
  25 #define tcp_state_names                 \
  26                 EM(TCP_ESTABLISHED)             \
  27                 EM(TCP_SYN_SENT)                \
  28                 EM(TCP_SYN_RECV)                \
  29                 EM(TCP_FIN_WAIT1)               \
  30                 EM(TCP_FIN_WAIT2)               \
  31                 EM(TCP_TIME_WAIT)               \
  32                 EM(TCP_CLOSE)                   \
  33                 EM(TCP_CLOSE_WAIT)              \
  34                 EM(TCP_LAST_ACK)                \
  35                 EM(TCP_LISTEN)                  \
  36                 EM(TCP_CLOSING)                 \
  37                 EMe(TCP_NEW_SYN_RECV)
  38
  39 #define skmem_kind_names                        \
  40                 EM(SK_MEM_SEND)                 \
  41                 EMe(SK_MEM_RECV)
  42
  43 /* enums need to be exported to user space */
  44 #undef EM
  45 #undef EMe
  46 #define EM(a)       TRACE_DEFINE_ENUM(a);
  47 #define EMe(a)      TRACE_DEFINE_ENUM(a);
  48
  49 family_names
  50 inet_protocol_names
  51 tcp_state_names
  52 skmem_kind_names
  53
  54 #undef EM
  55 #undef EMe
  56 #define EM(a)       { a, #a },
  57 #define EMe(a)      { a, #a }
  58
  59 #define show_family_name(val)                   \
  60         __print_symbolic(val, family_names)
  61
  62 #define show_inet_protocol_name(val)    \
  63         __print_symbolic(val, inet_protocol_names)
  64
  65 #define show_tcp_state_name(val)        \
  66         __print_symbolic(val, tcp_state_names)
  67
  68 #define show_skmem_kind_names(val)      \
  69         __print_symbolic(val, skmem_kind_names)
  70
  71 TRACE_EVENT(sock_rcvqueue_full,
  72
  73         TP_PROTO(struct sock *sk, struct sk_buff *skb),
  74
  75         TP_ARGS(sk, skb),
  76
  77         TP_STRUCT__entry(
  78                 __field(int, rmem_alloc)
  79                 __field(unsigned int, truesize)
  80                 __field(int, sk_rcvbuf)
  81         ),
  82
  83         TP_fast_assign(
  84                 __entry->rmem_alloc = atomic_read(&sk->sk_rmem_alloc);
  85                 __entry->truesize   = skb->truesize;
  86                 __entry->sk_rcvbuf  = READ_ONCE(sk->sk_rcvbuf);
  87         ),
  88
  89         TP_printk("rmem_alloc=%d truesize=%u sk_rcvbuf=%d",
  90                 __entry->rmem_alloc, __entry->truesize, __entry->sk_rcvbuf)
  91 );
  92
  93 TRACE_EVENT(sock_exceed_buf_limit,
  94
  95         TP_PROTO(struct sock *sk, struct proto *prot, long allocated, int kind),
  96
  97         TP_ARGS(sk, prot, allocated, kind),
  98
  99         TP_STRUCT__entry(
 100                 __array(char, name, 32)
 101                 __array(long, sysctl_mem, 3)
 102                 __field(long, allocated)
 103                 __field(int, sysctl_rmem)
 104                 __field(int, rmem_alloc)
 105                 __field(int, sysctl_wmem)
 106                 __field(int, wmem_alloc)
 107                 __field(int, wmem_queued)
 108                 __field(int, kind)
 109         ),
 110
 111         TP_fast_assign(
 112                 strscpy(__entry->name, prot->name, 32);
 113                 __entry->sysctl_mem[0] = READ_ONCE(prot->sysctl_mem[0]);
 114                 __entry->sysctl_mem[1] = READ_ONCE(prot->sysctl_mem[1]);
 115                 __entry->sysctl_mem[2] = READ_ONCE(prot->sysctl_mem[2]);
 116                 __entry->allocated = allocated;
 117                 __entry->sysctl_rmem = sk_get_rmem0(sk, prot);
 118                 __entry->rmem_alloc = atomic_read(&sk->sk_rmem_alloc);
 119                 __entry->sysctl_wmem = sk_get_wmem0(sk, prot);
 120                 __entry->wmem_alloc = refcount_read(&sk->sk_wmem_alloc);
 121                 __entry->wmem_queued = READ_ONCE(sk->sk_wmem_queued);
 122                 __entry->kind = kind;
 123         ),
 124
 125         TP_printk("proto:%s sysctl_mem=%ld,%ld,%ld allocated=%ld sysctl_rmem=%d rmem_alloc=%d sysctl_wmem=%d wmem_alloc=%d wmem_queued=%d kind=%s",
 126                 __entry->name,
 127                 __entry->sysctl_mem[0],
 128                 __entry->sysctl_mem[1],
 129                 __entry->sysctl_mem[2],
 130                 __entry->allocated,
 131                 __entry->sysctl_rmem,
 132                 __entry->rmem_alloc,
 133                 __entry->sysctl_wmem,
 134                 __entry->wmem_alloc,
 135                 __entry->wmem_queued,
 136                 show_skmem_kind_names(__entry->kind)
 137         )
 138 );
 139
 140 TRACE_EVENT(inet_sock_set_state,
 141
 142         TP_PROTO(const struct sock *sk, const int oldstate, const int newstate),
 143
 144         TP_ARGS(sk, oldstate, newstate),
 145
 146         TP_STRUCT__entry(
 147                 __field(const void *, skaddr)
 148                 __field(int, oldstate)
 149                 __field(int, newstate)
 150                 __field(__u16, sport)
 151                 __field(__u16, dport)
 152                 __field(__u16, family)
 153                 __field(__u16, protocol)
 154                 __array(__u8, saddr, 4)
 155                 __array(__u8, daddr, 4)
 156                 __array(__u8, saddr_v6, 16)
 157                 __array(__u8, daddr_v6, 16)
 158         ),
 159
 160         TP_fast_assign(
 161                 const struct inet_sock *inet = inet_sk(sk);
 162                 __be32 *p32;
 163
 164                 __entry->skaddr = sk;
 165                 __entry->oldstate = oldstate;
 166                 __entry->newstate = newstate;
 167
 168                 __entry->family = sk->sk_family;
 169                 __entry->protocol = sk->sk_protocol;
 170                 __entry->sport = ntohs(inet->inet_sport);
 171                 __entry->dport = ntohs(inet->inet_dport);
 172
 173                 p32 = (__be32 *) __entry->saddr;
 174                 *p32 = inet->inet_saddr;
 175
 176                 p32 = (__be32 *) __entry->daddr;
 177                 *p32 =  inet->inet_daddr;
 178
 179                 TP_STORE_ADDRS(__entry, inet->inet_saddr, inet->inet_daddr,
 180                                sk->sk_v6_rcv_saddr, sk->sk_v6_daddr);
 181         ),
 182
 183         TP_printk("family=%s protocol=%s sport=%hu dport=%hu saddr=%pI4 daddr=%pI4 saddrv6=%pI6c daddrv6=%pI6c oldstate=%s newstate=%s",
 184                         show_family_name(__entry->family),
 185                         show_inet_protocol_name(__entry->protocol),
 186                         __entry->sport, __entry->dport,
 187                         __entry->saddr, __entry->daddr,
 188                         __entry->saddr_v6, __entry->daddr_v6,
 189                         show_tcp_state_name(__entry->oldstate),
 190                         show_tcp_state_name(__entry->newstate))
 191 );
 192
 193 TRACE_EVENT(inet_sk_error_report,
 194
 195         TP_PROTO(const struct sock *sk),
 196
 197         TP_ARGS(sk),
 198
 199         TP_STRUCT__entry(
 200                 __field(int, error)
 201                 __field(__u16, sport)
 202                 __field(__u16, dport)
 203                 __field(__u16, family)
 204                 __field(__u16, protocol)
 205                 __array(__u8, saddr, 4)
 206                 __array(__u8, daddr, 4)
 207                 __array(__u8, saddr_v6, 16)
 208                 __array(__u8, daddr_v6, 16)
 209         ),
 210
 211         TP_fast_assign(
 212                 const struct inet_sock *inet = inet_sk(sk);
 213                 __be32 *p32;
 214
 215                 __entry->error = sk->sk_err;
 216                 __entry->family = sk->sk_family;
 217                 __entry->protocol = sk->sk_protocol;
 218                 __entry->sport = ntohs(inet->inet_sport);
 219                 __entry->dport = ntohs(inet->inet_dport);
 220
 221                 p32 = (__be32 *) __entry->saddr;
 222                 *p32 = inet->inet_saddr;
 223
 224                 p32 = (__be32 *) __entry->daddr;
 225                 *p32 =  inet->inet_daddr;
 226
 227                 TP_STORE_ADDRS(__entry, inet->inet_saddr, inet->inet_daddr,
 228                                sk->sk_v6_rcv_saddr, sk->sk_v6_daddr);
 229         ),
 230
 231         TP_printk("family=%s protocol=%s sport=%hu dport=%hu saddr=%pI4 daddr=%pI4 saddrv6=%pI6c daddrv6=%pI6c error=%d",
 232                   show_family_name(__entry->family),
 233                   show_inet_protocol_name(__entry->protocol),
 234                   __entry->sport, __entry->dport,
 235                   __entry->saddr, __entry->daddr,
 236                   __entry->saddr_v6, __entry->daddr_v6,
 237                   __entry->error)
 238 );
 239
 240 TRACE_EVENT(sk_data_ready,
 241
 242         TP_PROTO(const struct sock *sk),
 243
 244         TP_ARGS(sk),
 245
 246         TP_STRUCT__entry(
 247                 __field(const void *, skaddr)
 248                 __field(__u16, family)
 249                 __field(__u16, protocol)
 250                 __field(unsigned long, ip)
 251         ),
 252
 253         TP_fast_assign(
 254                 __entry->skaddr = sk;
 255                 __entry->family = sk->sk_family;
 256                 __entry->protocol = sk->sk_protocol;
 257                 __entry->ip = _RET_IP_;
 258         ),
 259
 260         TP_printk("family=%u protocol=%u func=%ps",
 261                   __entry->family, __entry->protocol, (void *)__entry->ip)
 262 );
 263
 264 /*
 265  * sock send/recv msg length
 266  */
 267 DECLARE_EVENT_CLASS(sock_msg_length,
 268
 269         TP_PROTO(struct sock *sk, int ret, int flags),
 270
 271         TP_ARGS(sk, ret, flags),
 272
 273         TP_STRUCT__entry(
 274                 __field(void *, sk)
 275                 __field(__u16, family)
 276                 __field(__u16, protocol)
 277                 __field(int, ret)
 278                 __field(int, flags)
 279         ),
 280
 281         TP_fast_assign(
 282                 __entry->sk = sk;
 283                 __entry->family = sk->sk_family;
 284                 __entry->protocol = sk->sk_protocol;
 285                 __entry->ret = ret;
 286                 __entry->flags = flags;
 287         ),
 288
 289         TP_printk("sk address = %p, family = %s protocol = %s, length = %d, error = %d, flags = 0x%x",
 290                   __entry->sk, show_family_name(__entry->family),
 291                   show_inet_protocol_name(__entry->protocol),
 292                   !(__entry->flags & MSG_PEEK) ?
 293                   (__entry->ret > 0 ? __entry->ret : 0) : 0,
 294                   __entry->ret < 0 ? __entry->ret : 0,
 295                   __entry->flags)
 296 );
 297
 298 DEFINE_EVENT(sock_msg_length, sock_send_length,
 299         TP_PROTO(struct sock *sk, int ret, int flags),
 300
 301         TP_ARGS(sk, ret, flags)
 302 );
 303
 304 DEFINE_EVENT(sock_msg_length, sock_recv_length,
 305         TP_PROTO(struct sock *sk, int ret, int flags),
 306
 307         TP_ARGS(sk, ret, flags)
 308 );
 309 #endif /* _TRACE_SOCK_H */
 310
 311 /* This part must be outside protection */
 312 #include <trace/define_trace.h>