| 1 |
|
| 2 |
|
| 3 | #if !defined(_TRACE_NET_PROBE_COMMON_H) || defined(TRACE_HEADER_MULTI_READ)
|
| 4 | #define _TRACE_NET_PROBE_COMMON_H
|
| 5 |
|
| 6 | #define TP_STORE_ADDR_PORTS_V4(__entry, inet, sk) \
|
| 7 | do { \
|
| 8 | struct sockaddr_in *v4 = (void *)__entry->saddr; \
|
| 9 | \
|
| 10 | v4->sin_family = AF_INET; \
|
| 11 | v4->sin_port = inet->inet_sport; \
|
| 12 | v4->sin_addr.s_addr = inet->inet_saddr; \
|
| 13 | v4 = (void *)__entry->daddr; \
|
| 14 | v4->sin_family = AF_INET; \
|
| 15 | v4->sin_port = inet->inet_dport; \
|
| 16 | v4->sin_addr.s_addr = inet->inet_daddr; \
|
| 17 | } while (0)
|
| 18 |
|
| 19 | #if IS_ENABLED(CONFIG_IPV6)
|
| 20 |
|
| 21 | #define TP_STORE_ADDR_PORTS(__entry, inet, sk) \
|
| 22 | do { \
|
| 23 | if (sk->sk_family == AF_INET6) { \
|
| 24 | struct sockaddr_in6 *v6 = (void *)__entry->saddr; \
|
| 25 | \
|
| 26 | v6->sin6_family = AF_INET6; \
|
| 27 | v6->sin6_port = inet->inet_sport; \
|
| 28 | v6->sin6_addr = inet6_sk(sk)->saddr; \
|
| 29 | v6 = (void *)__entry->daddr; \
|
| 30 | v6->sin6_family = AF_INET6; \
|
| 31 | v6->sin6_port = inet->inet_dport; \
|
| 32 | v6->sin6_addr = sk->sk_v6_daddr; \
|
| 33 | } else \
|
| 34 | TP_STORE_ADDR_PORTS_V4(__entry, inet, sk); \
|
| 35 | } while (0)
|
| 36 |
|
| 37 | #else
|
| 38 |
|
| 39 | #define TP_STORE_ADDR_PORTS(__entry, inet, sk) \
|
| 40 | TP_STORE_ADDR_PORTS_V4(__entry, inet, sk);
|
| 41 |
|
| 42 | #endif
|
| 43 |
|
| 44 | #define TP_STORE_V4MAPPED(__entry, saddr, daddr) \
|
| 45 | do { \
|
| 46 | struct in6_addr *pin6; \
|
| 47 | \
|
| 48 | pin6 = (struct in6_addr *)__entry->saddr_v6; \
|
| 49 | ipv6_addr_set_v4mapped(saddr, pin6); \
|
| 50 | pin6 = (struct in6_addr *)__entry->daddr_v6; \
|
| 51 | ipv6_addr_set_v4mapped(daddr, pin6); \
|
| 52 | } while (0)
|
| 53 |
|
| 54 | #if IS_ENABLED(CONFIG_IPV6)
|
| 55 | #define TP_STORE_ADDRS(__entry, saddr, daddr, saddr6, daddr6) \
|
| 56 | do { \
|
| 57 | if (sk->sk_family == AF_INET6) { \
|
| 58 | struct in6_addr *pin6; \
|
| 59 | \
|
| 60 | pin6 = (struct in6_addr *)__entry->saddr_v6; \
|
| 61 | *pin6 = saddr6; \
|
| 62 | pin6 = (struct in6_addr *)__entry->daddr_v6; \
|
| 63 | *pin6 = daddr6; \
|
| 64 | } else { \
|
| 65 | TP_STORE_V4MAPPED(__entry, saddr, daddr); \
|
| 66 | } \
|
| 67 | } while (0)
|
| 68 | #else
|
| 69 | #define TP_STORE_ADDRS(__entry, saddr, daddr, saddr6, daddr6) \
|
| 70 | TP_STORE_V4MAPPED(__entry, saddr, daddr)
|
| 71 | #endif
|
| 72 |
|
| 73 | #define TP_STORE_ADDR_PORTS_SKB_V4(skb, protoh, entry_saddr, entry_daddr) \
|
| 74 | do { \
|
| 75 | struct sockaddr_in *v4 = (void *)entry_saddr; \
|
| 76 | \
|
| 77 | v4->sin_family = AF_INET; \
|
| 78 | v4->sin_port = protoh->source; \
|
| 79 | v4->sin_addr.s_addr = ip_hdr(skb)->saddr; \
|
| 80 | v4 = (void *)entry_daddr; \
|
| 81 | v4->sin_family = AF_INET; \
|
| 82 | v4->sin_port = protoh->dest; \
|
| 83 | v4->sin_addr.s_addr = ip_hdr(skb)->daddr; \
|
| 84 | } while (0)
|
| 85 |
|
| 86 | #if IS_ENABLED(CONFIG_IPV6)
|
| 87 |
|
| 88 | #define TP_STORE_ADDR_PORTS_SKB(skb, protoh, entry_saddr, entry_daddr) \
|
| 89 | do { \
|
| 90 | const struct iphdr *iph = ip_hdr(skb); \
|
| 91 | \
|
| 92 | if (iph->version == 6) { \
|
| 93 | struct sockaddr_in6 *v6 = (void *)entry_saddr; \
|
| 94 | \
|
| 95 | v6->sin6_family = AF_INET6; \
|
| 96 | v6->sin6_port = protoh->source; \
|
| 97 | v6->sin6_addr = ipv6_hdr(skb)->saddr; \
|
| 98 | v6 = (void *)entry_daddr; \
|
| 99 | v6->sin6_family = AF_INET6; \
|
| 100 | v6->sin6_port = protoh->dest; \
|
| 101 | v6->sin6_addr = ipv6_hdr(skb)->daddr; \
|
| 102 | } else \
|
| 103 | TP_STORE_ADDR_PORTS_SKB_V4(skb, protoh, \
|
| 104 | entry_saddr, \
|
| 105 | entry_daddr); \
|
| 106 | } while (0)
|
| 107 |
|
| 108 | #else
|
| 109 |
|
| 110 | #define TP_STORE_ADDR_PORTS_SKB(skb, protoh, entry_saddr, entry_daddr) \
|
| 111 | TP_STORE_ADDR_PORTS_SKB_V4(skb, protoh, entry_saddr, entry_daddr)
|
| 112 |
|
| 113 | #endif
|
| 114 |
|
| 115 | #endif
|