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>
7typedef __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
20struct 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 */
30struct ipv4_packet {
31	struct ethhdr eth;
32	struct iphdr iph;
33	struct tcphdr tcp;
34} __packed;
35extern struct ipv4_packet pkt_v4;
36
37/* ipv6 test vector */
38struct ipv6_packet {
39	struct ethhdr eth;
40	struct ipv6hdr iph;
41	struct tcphdr tcp;
42} __packed;
43extern struct ipv6_packet pkt_v6;
44
45int settimeo(int fd, int timeout_ms);
46int start_server(int family, int type, const char *addr, __u16 port,
47		 int timeout_ms);
48int start_mptcp_server(int family, const char *addr, __u16 port,
49		       int timeout_ms);
50int *start_reuseport_server(int family, int type, const char *addr_str,
51			    __u16 port, int timeout_ms,
52			    unsigned int nr_listens);
53void free_fds(int *fds, unsigned int nr_close_fds);
54int connect_to_addr(const struct sockaddr_storage *addr, socklen_t len, int type);
55int connect_to_fd(int server_fd, int timeout_ms);
56int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts);
57int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms);
58int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
59		     int timeout_ms);
60int make_sockaddr(int family, const char *addr_str, __u16 port,
61		  struct sockaddr_storage *addr, socklen_t *len);
62char *ping_command(int family);
63int get_socket_local_port(int sock_fd);
64
65struct 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 */
72struct nstoken *open_netns(const char *name);
73void close_netns(struct nstoken *token);
74
75static __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
83static 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
98static 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