sock.h 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. #undef TRACE_SYSTEM
  3. #define TRACE_SYSTEM sock
  4. #if !defined(_TRACE_SOCK_H) || defined(TRACE_HEADER_MULTI_READ)
  5. #define _TRACE_SOCK_H
  6. #include <net/sock.h>
  7. #include <net/ipv6.h>
  8. #include <linux/tracepoint.h>
  9. #include <linux/ipv6.h>
  10. #include <linux/tcp.h>
  11. #define family_names \
  12. EM(AF_INET) \
  13. EMe(AF_INET6)
  14. /* The protocol traced by inet_sock_set_state */
  15. #define inet_protocol_names \
  16. EM(IPPROTO_TCP) \
  17. EM(IPPROTO_DCCP) \
  18. EM(IPPROTO_SCTP) \
  19. EMe(IPPROTO_MPTCP)
  20. #define tcp_state_names \
  21. EM(TCP_ESTABLISHED) \
  22. EM(TCP_SYN_SENT) \
  23. EM(TCP_SYN_RECV) \
  24. EM(TCP_FIN_WAIT1) \
  25. EM(TCP_FIN_WAIT2) \
  26. EM(TCP_TIME_WAIT) \
  27. EM(TCP_CLOSE) \
  28. EM(TCP_CLOSE_WAIT) \
  29. EM(TCP_LAST_ACK) \
  30. EM(TCP_LISTEN) \
  31. EM(TCP_CLOSING) \
  32. EMe(TCP_NEW_SYN_RECV)
  33. #define skmem_kind_names \
  34. EM(SK_MEM_SEND) \
  35. EMe(SK_MEM_RECV)
  36. /* enums need to be exported to user space */
  37. #undef EM
  38. #undef EMe
  39. #define EM(a) TRACE_DEFINE_ENUM(a);
  40. #define EMe(a) TRACE_DEFINE_ENUM(a);
  41. family_names
  42. inet_protocol_names
  43. tcp_state_names
  44. skmem_kind_names
  45. #undef EM
  46. #undef EMe
  47. #define EM(a) { a, #a },
  48. #define EMe(a) { a, #a }
  49. #define show_family_name(val) \
  50. __print_symbolic(val, family_names)
  51. #define show_inet_protocol_name(val) \
  52. __print_symbolic(val, inet_protocol_names)
  53. #define show_tcp_state_name(val) \
  54. __print_symbolic(val, tcp_state_names)
  55. #define show_skmem_kind_names(val) \
  56. __print_symbolic(val, skmem_kind_names)
  57. TRACE_EVENT(sock_rcvqueue_full,
  58. TP_PROTO(struct sock *sk, struct sk_buff *skb),
  59. TP_ARGS(sk, skb),
  60. TP_STRUCT__entry(
  61. __field(int, rmem_alloc)
  62. __field(unsigned int, truesize)
  63. __field(int, sk_rcvbuf)
  64. ),
  65. TP_fast_assign(
  66. __entry->rmem_alloc = atomic_read(&sk->sk_rmem_alloc);
  67. __entry->truesize = skb->truesize;
  68. __entry->sk_rcvbuf = READ_ONCE(sk->sk_rcvbuf);
  69. ),
  70. TP_printk("rmem_alloc=%d truesize=%u sk_rcvbuf=%d",
  71. __entry->rmem_alloc, __entry->truesize, __entry->sk_rcvbuf)
  72. );
  73. TRACE_EVENT(sock_exceed_buf_limit,
  74. TP_PROTO(struct sock *sk, struct proto *prot, long allocated, int kind),
  75. TP_ARGS(sk, prot, allocated, kind),
  76. TP_STRUCT__entry(
  77. __array(char, name, 32)
  78. __array(long, sysctl_mem, 3)
  79. __field(long, allocated)
  80. __field(int, sysctl_rmem)
  81. __field(int, rmem_alloc)
  82. __field(int, sysctl_wmem)
  83. __field(int, wmem_alloc)
  84. __field(int, wmem_queued)
  85. __field(int, kind)
  86. ),
  87. TP_fast_assign(
  88. strncpy(__entry->name, prot->name, 32);
  89. __entry->sysctl_mem[0] = READ_ONCE(prot->sysctl_mem[0]);
  90. __entry->sysctl_mem[1] = READ_ONCE(prot->sysctl_mem[1]);
  91. __entry->sysctl_mem[2] = READ_ONCE(prot->sysctl_mem[2]);
  92. __entry->allocated = allocated;
  93. __entry->sysctl_rmem = sk_get_rmem0(sk, prot);
  94. __entry->rmem_alloc = atomic_read(&sk->sk_rmem_alloc);
  95. __entry->sysctl_wmem = sk_get_wmem0(sk, prot);
  96. __entry->wmem_alloc = refcount_read(&sk->sk_wmem_alloc);
  97. __entry->wmem_queued = READ_ONCE(sk->sk_wmem_queued);
  98. __entry->kind = kind;
  99. ),
  100. 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",
  101. __entry->name,
  102. __entry->sysctl_mem[0],
  103. __entry->sysctl_mem[1],
  104. __entry->sysctl_mem[2],
  105. __entry->allocated,
  106. __entry->sysctl_rmem,
  107. __entry->rmem_alloc,
  108. __entry->sysctl_wmem,
  109. __entry->wmem_alloc,
  110. __entry->wmem_queued,
  111. show_skmem_kind_names(__entry->kind)
  112. )
  113. );
  114. TRACE_EVENT(inet_sock_set_state,
  115. TP_PROTO(const struct sock *sk, const int oldstate, const int newstate),
  116. TP_ARGS(sk, oldstate, newstate),
  117. TP_STRUCT__entry(
  118. __field(const void *, skaddr)
  119. __field(int, oldstate)
  120. __field(int, newstate)
  121. __field(__u16, sport)
  122. __field(__u16, dport)
  123. __field(__u16, family)
  124. __field(__u16, protocol)
  125. __array(__u8, saddr, 4)
  126. __array(__u8, daddr, 4)
  127. __array(__u8, saddr_v6, 16)
  128. __array(__u8, daddr_v6, 16)
  129. ),
  130. TP_fast_assign(
  131. struct inet_sock *inet = inet_sk(sk);
  132. struct in6_addr *pin6;
  133. __be32 *p32;
  134. __entry->skaddr = sk;
  135. __entry->oldstate = oldstate;
  136. __entry->newstate = newstate;
  137. __entry->family = sk->sk_family;
  138. __entry->protocol = sk->sk_protocol;
  139. __entry->sport = ntohs(inet->inet_sport);
  140. __entry->dport = ntohs(inet->inet_dport);
  141. p32 = (__be32 *) __entry->saddr;
  142. *p32 = inet->inet_saddr;
  143. p32 = (__be32 *) __entry->daddr;
  144. *p32 = inet->inet_daddr;
  145. #if IS_ENABLED(CONFIG_IPV6)
  146. if (sk->sk_family == AF_INET6) {
  147. pin6 = (struct in6_addr *)__entry->saddr_v6;
  148. *pin6 = sk->sk_v6_rcv_saddr;
  149. pin6 = (struct in6_addr *)__entry->daddr_v6;
  150. *pin6 = sk->sk_v6_daddr;
  151. } else
  152. #endif
  153. {
  154. pin6 = (struct in6_addr *)__entry->saddr_v6;
  155. ipv6_addr_set_v4mapped(inet->inet_saddr, pin6);
  156. pin6 = (struct in6_addr *)__entry->daddr_v6;
  157. ipv6_addr_set_v4mapped(inet->inet_daddr, pin6);
  158. }
  159. ),
  160. TP_printk("family=%s protocol=%s sport=%hu dport=%hu saddr=%pI4 daddr=%pI4 saddrv6=%pI6c daddrv6=%pI6c oldstate=%s newstate=%s",
  161. show_family_name(__entry->family),
  162. show_inet_protocol_name(__entry->protocol),
  163. __entry->sport, __entry->dport,
  164. __entry->saddr, __entry->daddr,
  165. __entry->saddr_v6, __entry->daddr_v6,
  166. show_tcp_state_name(__entry->oldstate),
  167. show_tcp_state_name(__entry->newstate))
  168. );
  169. TRACE_EVENT(inet_sk_error_report,
  170. TP_PROTO(const struct sock *sk),
  171. TP_ARGS(sk),
  172. TP_STRUCT__entry(
  173. __field(int, error)
  174. __field(__u16, sport)
  175. __field(__u16, dport)
  176. __field(__u16, family)
  177. __field(__u16, protocol)
  178. __array(__u8, saddr, 4)
  179. __array(__u8, daddr, 4)
  180. __array(__u8, saddr_v6, 16)
  181. __array(__u8, daddr_v6, 16)
  182. ),
  183. TP_fast_assign(
  184. struct inet_sock *inet = inet_sk(sk);
  185. struct in6_addr *pin6;
  186. __be32 *p32;
  187. __entry->error = sk->sk_err;
  188. __entry->family = sk->sk_family;
  189. __entry->protocol = sk->sk_protocol;
  190. __entry->sport = ntohs(inet->inet_sport);
  191. __entry->dport = ntohs(inet->inet_dport);
  192. p32 = (__be32 *) __entry->saddr;
  193. *p32 = inet->inet_saddr;
  194. p32 = (__be32 *) __entry->daddr;
  195. *p32 = inet->inet_daddr;
  196. #if IS_ENABLED(CONFIG_IPV6)
  197. if (sk->sk_family == AF_INET6) {
  198. pin6 = (struct in6_addr *)__entry->saddr_v6;
  199. *pin6 = sk->sk_v6_rcv_saddr;
  200. pin6 = (struct in6_addr *)__entry->daddr_v6;
  201. *pin6 = sk->sk_v6_daddr;
  202. } else
  203. #endif
  204. {
  205. pin6 = (struct in6_addr *)__entry->saddr_v6;
  206. ipv6_addr_set_v4mapped(inet->inet_saddr, pin6);
  207. pin6 = (struct in6_addr *)__entry->daddr_v6;
  208. ipv6_addr_set_v4mapped(inet->inet_daddr, pin6);
  209. }
  210. ),
  211. TP_printk("family=%s protocol=%s sport=%hu dport=%hu saddr=%pI4 daddr=%pI4 saddrv6=%pI6c daddrv6=%pI6c error=%d",
  212. show_family_name(__entry->family),
  213. show_inet_protocol_name(__entry->protocol),
  214. __entry->sport, __entry->dport,
  215. __entry->saddr, __entry->daddr,
  216. __entry->saddr_v6, __entry->daddr_v6,
  217. __entry->error)
  218. );
  219. #endif /* _TRACE_SOCK_H */
  220. /* This part must be outside protection */
  221. #include <trace/define_trace.h>