1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef __NETWORK_HELPERS_H
3 #define __NETWORK_HELPERS_H
4 #include <sys/socket.h>
5 #include <sys/types.h>
6 #include <linux/types.h>
7 typedef __u16 __sum16;
8 #include <linux/if_ether.h>
9 #include <linux/if_packet.h>
10 #include <linux/ip.h>
11 #include <linux/ipv6.h>
12 #include <netinet/tcp.h>
13 #include <bpf/bpf_endian.h>
14
15 #define MAGIC_VAL 0x1234
16 #define NUM_ITER 100000
17 #define VIP_NUM 5
18 #define MAGIC_BYTES 123
19
20 struct network_helper_opts {
21 const char *cc;
22 int timeout_ms;
23 bool must_fail;
24 bool noconnect;
25 int type;
26 int proto;
27 };
28
29 /* ipv4 test vector */
30 struct ipv4_packet {
31 struct ethhdr eth;
32 struct iphdr iph;
33 struct tcphdr tcp;
34 } __packed;
35 extern struct ipv4_packet pkt_v4;
36
37 /* ipv6 test vector */
38 struct ipv6_packet {
39 struct ethhdr eth;
40 struct ipv6hdr iph;
41 struct tcphdr tcp;
42 } __packed;
43 extern struct ipv6_packet pkt_v6;
44
45 int settimeo(int fd, int timeout_ms);
46 int start_server(int family, int type, const char *addr, __u16 port,
47 int timeout_ms);
48 int start_mptcp_server(int family, const char *addr, __u16 port,
49 int timeout_ms);
50 int *start_reuseport_server(int family, int type, const char *addr_str,
51 __u16 port, int timeout_ms,
52 unsigned int nr_listens);
53 void free_fds(int *fds, unsigned int nr_close_fds);
54 int connect_to_addr(const struct sockaddr_storage *addr, socklen_t len, int type);
55 int connect_to_fd(int server_fd, int timeout_ms);
56 int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts);
57 int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms);
58 int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
59 int timeout_ms);
60 int make_sockaddr(int family, const char *addr_str, __u16 port,
61 struct sockaddr_storage *addr, socklen_t *len);
62 char *ping_command(int family);
63 int get_socket_local_port(int sock_fd);
64
65 struct nstoken;
66 /**
67 * open_netns() - Switch to specified network namespace by name.
68 *
69 * Returns token with which to restore the original namespace
70 * using close_netns().
71 */
72 struct nstoken *open_netns(const char *name);
73 void close_netns(struct nstoken *token);
74
csum_fold(__u32 csum)75 static __u16 csum_fold(__u32 csum)
76 {
77 csum = (csum & 0xffff) + (csum >> 16);
78 csum = (csum & 0xffff) + (csum >> 16);
79
80 return (__u16)~csum;
81 }
82
csum_tcpudp_magic(__be32 saddr,__be32 daddr,__u32 len,__u8 proto,__wsum csum)83 static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr,
84 __u32 len, __u8 proto,
85 __wsum csum)
86 {
87 __u64 s = csum;
88
89 s += (__u32)saddr;
90 s += (__u32)daddr;
91 s += htons(proto + len);
92 s = (s & 0xffffffff) + (s >> 32);
93 s = (s & 0xffffffff) + (s >> 32);
94
95 return csum_fold((__u32)s);
96 }
97
csum_ipv6_magic(const struct in6_addr * saddr,const struct in6_addr * daddr,__u32 len,__u8 proto,__wsum csum)98 static inline __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
99 const struct in6_addr *daddr,
100 __u32 len, __u8 proto,
101 __wsum csum)
102 {
103 __u64 s = csum;
104 int i;
105
106 for (i = 0; i < 4; i++)
107 s += (__u32)saddr->s6_addr32[i];
108 for (i = 0; i < 4; i++)
109 s += (__u32)daddr->s6_addr32[i];
110 s += htons(proto + len);
111 s = (s & 0xffffffff) + (s >> 32);
112 s = (s & 0xffffffff) + (s >> 32);
113
114 return csum_fold((__u32)s);
115 }
116
117 #endif
118