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