1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright Amazon.com Inc. or its affiliates. */
3
4 #ifndef _TEST_TCP_SYNCOOKIE_H
5 #define _TEST_TCP_SYNCOOKIE_H
6
7 #define __packed __attribute__((__packed__))
8 #define __force
9
10 #define swap(a, b) \
11 do { \
12 typeof(a) __tmp = (a); \
13 (a) = (b); \
14 (b) = __tmp; \
15 } while (0)
16
17 #define swap_array(a, b) \
18 do { \
19 typeof(a) __tmp[sizeof(a)]; \
20 __builtin_memcpy(__tmp, a, sizeof(a)); \
21 __builtin_memcpy(a, b, sizeof(a)); \
22 __builtin_memcpy(b, __tmp, sizeof(a)); \
23 } while (0)
24
25 /* linux/unaligned.h */
26 #define __get_unaligned_t(type, ptr) ({ \
27 const struct { type x; } __packed * __pptr = (typeof(__pptr))(ptr); \
28 __pptr->x; \
29 })
30
31 #define get_unaligned(ptr) __get_unaligned_t(typeof(*(ptr)), (ptr))
32
get_unaligned_be16(const void * p)33 static inline u16 get_unaligned_be16(const void *p)
34 {
35 return bpf_ntohs(__get_unaligned_t(__be16, p));
36 }
37
get_unaligned_be32(const void * p)38 static inline u32 get_unaligned_be32(const void *p)
39 {
40 return bpf_ntohl(__get_unaligned_t(__be32, p));
41 }
42
43 /* lib/checksum.c */
from64to32(u64 x)44 static inline u32 from64to32(u64 x)
45 {
46 /* add up 32-bit and 32-bit for 32+c bit */
47 x = (x & 0xffffffff) + (x >> 32);
48 /* add up carry.. */
49 x = (x & 0xffffffff) + (x >> 32);
50 return (u32)x;
51 }
52
csum_tcpudp_nofold(__be32 saddr,__be32 daddr,__u32 len,__u8 proto,__wsum sum)53 static inline __wsum csum_tcpudp_nofold(__be32 saddr, __be32 daddr,
54 __u32 len, __u8 proto, __wsum sum)
55 {
56 unsigned long long s = (__force u32)sum;
57
58 s += (__force u32)saddr;
59 s += (__force u32)daddr;
60 #ifdef __BIG_ENDIAN
61 s += proto + len;
62 #else
63 s += (proto + len) << 8;
64 #endif
65 return (__force __wsum)from64to32(s);
66 }
67
68 /* asm-generic/checksum.h */
csum_fold(__wsum csum)69 static inline __sum16 csum_fold(__wsum csum)
70 {
71 u32 sum = (__force u32)csum;
72
73 sum = (sum & 0xffff) + (sum >> 16);
74 sum = (sum & 0xffff) + (sum >> 16);
75 return (__force __sum16)~sum;
76 }
77
csum_tcpudp_magic(__be32 saddr,__be32 daddr,__u32 len,__u8 proto,__wsum sum)78 static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr, __u32 len,
79 __u8 proto, __wsum sum)
80 {
81 return csum_fold(csum_tcpudp_nofold(saddr, daddr, len, proto, sum));
82 }
83
84 /* net/ipv6/ip6_checksum.c */
csum_ipv6_magic(const struct in6_addr * saddr,const struct in6_addr * daddr,__u32 len,__u8 proto,__wsum csum)85 static inline __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
86 const struct in6_addr *daddr,
87 __u32 len, __u8 proto, __wsum csum)
88 {
89 int carry;
90 __u32 ulen;
91 __u32 uproto;
92 __u32 sum = (__force u32)csum;
93
94 sum += (__force u32)saddr->in6_u.u6_addr32[0];
95 carry = (sum < (__force u32)saddr->in6_u.u6_addr32[0]);
96 sum += carry;
97
98 sum += (__force u32)saddr->in6_u.u6_addr32[1];
99 carry = (sum < (__force u32)saddr->in6_u.u6_addr32[1]);
100 sum += carry;
101
102 sum += (__force u32)saddr->in6_u.u6_addr32[2];
103 carry = (sum < (__force u32)saddr->in6_u.u6_addr32[2]);
104 sum += carry;
105
106 sum += (__force u32)saddr->in6_u.u6_addr32[3];
107 carry = (sum < (__force u32)saddr->in6_u.u6_addr32[3]);
108 sum += carry;
109
110 sum += (__force u32)daddr->in6_u.u6_addr32[0];
111 carry = (sum < (__force u32)daddr->in6_u.u6_addr32[0]);
112 sum += carry;
113
114 sum += (__force u32)daddr->in6_u.u6_addr32[1];
115 carry = (sum < (__force u32)daddr->in6_u.u6_addr32[1]);
116 sum += carry;
117
118 sum += (__force u32)daddr->in6_u.u6_addr32[2];
119 carry = (sum < (__force u32)daddr->in6_u.u6_addr32[2]);
120 sum += carry;
121
122 sum += (__force u32)daddr->in6_u.u6_addr32[3];
123 carry = (sum < (__force u32)daddr->in6_u.u6_addr32[3]);
124 sum += carry;
125
126 ulen = (__force u32)bpf_htonl((__u32)len);
127 sum += ulen;
128 carry = (sum < ulen);
129 sum += carry;
130
131 uproto = (__force u32)bpf_htonl(proto);
132 sum += uproto;
133 carry = (sum < uproto);
134 sum += carry;
135
136 return csum_fold((__force __wsum)sum);
137 }
138 #endif
139