1 /* SPDX-License-Identifier: GPL-2.0 */ 2 #ifndef __NETWORK_HELPERS_H 3 #define __NETWORK_HELPERS_H 4 #include <arpa/inet.h> 5 #include <sys/socket.h> 6 #include <sys/types.h> 7 #include <linux/types.h> 8 typedef __u16 __sum16; 9 #include <linux/if_ether.h> 10 #include <linux/if_packet.h> 11 #include <linux/ip.h> 12 #include <linux/ipv6.h> 13 #include <linux/ethtool.h> 14 #include <linux/sockios.h> 15 #include <linux/err.h> 16 #include <netinet/tcp.h> 17 #include <bpf/bpf_endian.h> 18 #include <net/if.h> 19 20 #define MAGIC_VAL 0x1234 21 #define NUM_ITER 100000 22 #define VIP_NUM 5 23 #define MAGIC_BYTES 123 24 25 struct network_helper_opts { 26 int timeout_ms; 27 int proto; 28 /* +ve: Passed to listen() as-is. 29 * 0: Default when the test does not set 30 * a particular value during the struct init. 31 * It is changed to 1 before passing to listen(). 32 * Most tests only have one on-going connection. 33 * -ve: It is changed to 0 before passing to listen(). 34 * It is useful to force syncookie without 35 * changing the "tcp_syncookies" sysctl from 1 to 2. 36 */ 37 int backlog; 38 int (*post_socket_cb)(int fd, void *opts); 39 void *cb_opts; 40 }; 41 42 /* ipv4 test vector */ 43 struct ipv4_packet { 44 struct ethhdr eth; 45 struct iphdr iph; 46 struct tcphdr tcp; 47 } __packed; 48 extern struct ipv4_packet pkt_v4; 49 50 /* ipv6 test vector */ 51 struct ipv6_packet { 52 struct ethhdr eth; 53 struct ipv6hdr iph; 54 struct tcphdr tcp; 55 } __packed; 56 extern struct ipv6_packet pkt_v6; 57 58 int settimeo(int fd, int timeout_ms); 59 int start_server_str(int family, int type, const char *addr_str, __u16 port, 60 const struct network_helper_opts *opts); 61 int start_server(int family, int type, const char *addr, __u16 port, 62 int timeout_ms); 63 int *start_reuseport_server(int family, int type, const char *addr_str, 64 __u16 port, int timeout_ms, 65 unsigned int nr_listens); 66 int start_server_addr(int type, const struct sockaddr_storage *addr, socklen_t len, 67 const struct network_helper_opts *opts); 68 void free_fds(int *fds, unsigned int nr_close_fds); 69 int client_socket(int family, int type, 70 const struct network_helper_opts *opts); 71 int connect_to_addr(int type, const struct sockaddr_storage *addr, socklen_t len, 72 const struct network_helper_opts *opts); 73 int connect_to_addr_str(int family, int type, const char *addr_str, __u16 port, 74 const struct network_helper_opts *opts); 75 int connect_to_fd(int server_fd, int timeout_ms); 76 int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts); 77 int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms); 78 int fastopen_connect(int server_fd, const char *data, unsigned int data_len, 79 int timeout_ms); 80 int make_sockaddr(int family, const char *addr_str, __u16 port, 81 struct sockaddr_storage *addr, socklen_t *len); 82 char *ping_command(int family); 83 int get_socket_local_port(int sock_fd); 84 int get_hw_ring_size(char *ifname, struct ethtool_ringparam *ring_param); 85 int set_hw_ring_size(char *ifname, struct ethtool_ringparam *ring_param); 86 87 struct nstoken; 88 /** 89 * open_netns() - Switch to specified network namespace by name. 90 * 91 * Returns token with which to restore the original namespace 92 * using close_netns(). 93 */ 94 struct nstoken *open_netns(const char *name); 95 void close_netns(struct nstoken *token); 96 int send_recv_data(int lfd, int fd, uint32_t total_bytes); 97 int make_netns(const char *name); 98 int remove_netns(const char *name); 99 100 static __u16 csum_fold(__u32 csum) 101 { 102 csum = (csum & 0xffff) + (csum >> 16); 103 csum = (csum & 0xffff) + (csum >> 16); 104 105 return (__u16)~csum; 106 } 107 108 static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr, 109 __u32 len, __u8 proto, 110 __wsum csum) 111 { 112 __u64 s = csum; 113 114 s += (__u32)saddr; 115 s += (__u32)daddr; 116 s += htons(proto + len); 117 s = (s & 0xffffffff) + (s >> 32); 118 s = (s & 0xffffffff) + (s >> 32); 119 120 return csum_fold((__u32)s); 121 } 122 123 static inline __sum16 csum_ipv6_magic(const struct in6_addr *saddr, 124 const struct in6_addr *daddr, 125 __u32 len, __u8 proto, 126 __wsum csum) 127 { 128 __u64 s = csum; 129 int i; 130 131 for (i = 0; i < 4; i++) 132 s += (__u32)saddr->s6_addr32[i]; 133 for (i = 0; i < 4; i++) 134 s += (__u32)daddr->s6_addr32[i]; 135 s += htons(proto + len); 136 s = (s & 0xffffffff) + (s >> 32); 137 s = (s & 0xffffffff) + (s >> 32); 138 139 return csum_fold((__u32)s); 140 } 141 142 struct tmonitor_ctx; 143 144 #ifdef TRAFFIC_MONITOR 145 struct tmonitor_ctx *traffic_monitor_start(const char *netns, const char *test_name, 146 const char *subtest_name); 147 void traffic_monitor_stop(struct tmonitor_ctx *ctx); 148 #else 149 static inline struct tmonitor_ctx *traffic_monitor_start(const char *netns, const char *test_name, 150 const char *subtest_name) 151 { 152 return NULL; 153 } 154 155 static inline void traffic_monitor_stop(struct tmonitor_ctx *ctx) 156 { 157 } 158 #endif 159 160 #endif 161