| 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> |