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