Alexis Lothoré (eBPF Foundation)
e6ef0b5463
BugLink: https://bugs.launchpad.net/bugs/2101915
[ Upstream commit bc9b3fb827fceec4e05564d6e668280f4470ab5b ]
Including the network_helpers.h header in tests can lead to the following
build error:
./network_helpers.h: In function ‘csum_tcpudp_magic’:
./network_helpers.h:116:14: error: implicit declaration of function \
‘htons’ [-Werror=implicit-function-declaration]
116 | s += htons(proto + len);
The error is avoided in many cases thanks to some other headers included
earlier and bringing in arpa/inet.h (ie: test_progs.h).
Make sure that test_progs build success does not depend on header ordering
by adding the missing header include in network_helpers.h
Fixes: f6642de0c3 ("selftests/bpf: Add csum helpers")
Signed-off-by: Alexis Lothoré (eBPF Foundation) <alexis.lothore@bootlin.com>
Link: https://lore.kernel.org/r/20241008-network_helpers_fix-v1-1-2c2ae03df7ef@bootlin.com
Signed-off-by: Martin KaFai Lau <martin.lau@kernel.org>
Signed-off-by: Sasha Levin <sashal@kernel.org>
Signed-off-by: Koichiro Den <koichiro.den@canonical.com>
Signed-off-by: Stefan Bader <stefan.bader@canonical.com>
121 lines
3.0 KiB
C
121 lines
3.0 KiB
C
/* SPDX-License-Identifier: GPL-2.0 */
|
|
#ifndef __NETWORK_HELPERS_H
|
|
#define __NETWORK_HELPERS_H
|
|
#include <arpa/inet.h>
|
|
#include <sys/socket.h>
|
|
#include <sys/types.h>
|
|
#include <linux/types.h>
|
|
typedef __u16 __sum16;
|
|
#include <linux/if_ether.h>
|
|
#include <linux/if_packet.h>
|
|
#include <linux/ip.h>
|
|
#include <linux/ipv6.h>
|
|
#include <netinet/tcp.h>
|
|
#include <bpf/bpf_endian.h>
|
|
|
|
#define MAGIC_VAL 0x1234
|
|
#define NUM_ITER 100000
|
|
#define VIP_NUM 5
|
|
#define MAGIC_BYTES 123
|
|
|
|
struct network_helper_opts {
|
|
const char *cc;
|
|
int timeout_ms;
|
|
bool must_fail;
|
|
bool noconnect;
|
|
int type;
|
|
int proto;
|
|
};
|
|
|
|
/* ipv4 test vector */
|
|
struct ipv4_packet {
|
|
struct ethhdr eth;
|
|
struct iphdr iph;
|
|
struct tcphdr tcp;
|
|
} __packed;
|
|
extern struct ipv4_packet pkt_v4;
|
|
|
|
/* ipv6 test vector */
|
|
struct ipv6_packet {
|
|
struct ethhdr eth;
|
|
struct ipv6hdr iph;
|
|
struct tcphdr tcp;
|
|
} __packed;
|
|
extern struct ipv6_packet pkt_v6;
|
|
|
|
int settimeo(int fd, int timeout_ms);
|
|
int start_server(int family, int type, const char *addr, __u16 port,
|
|
int timeout_ms);
|
|
int start_mptcp_server(int family, const char *addr, __u16 port,
|
|
int timeout_ms);
|
|
int *start_reuseport_server(int family, int type, const char *addr_str,
|
|
__u16 port, int timeout_ms,
|
|
unsigned int nr_listens);
|
|
void free_fds(int *fds, unsigned int nr_close_fds);
|
|
int connect_to_addr(const struct sockaddr_storage *addr, socklen_t len, int type);
|
|
int connect_to_fd(int server_fd, int timeout_ms);
|
|
int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts);
|
|
int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms);
|
|
int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
|
|
int timeout_ms);
|
|
int make_sockaddr(int family, const char *addr_str, __u16 port,
|
|
struct sockaddr_storage *addr, socklen_t *len);
|
|
char *ping_command(int family);
|
|
int get_socket_local_port(int sock_fd);
|
|
|
|
struct nstoken;
|
|
/**
|
|
* open_netns() - Switch to specified network namespace by name.
|
|
*
|
|
* Returns token with which to restore the original namespace
|
|
* using close_netns().
|
|
*/
|
|
struct nstoken *open_netns(const char *name);
|
|
void close_netns(struct nstoken *token);
|
|
int make_netns(const char *name);
|
|
int remove_netns(const char *name);
|
|
|
|
static __u16 csum_fold(__u32 csum)
|
|
{
|
|
csum = (csum & 0xffff) + (csum >> 16);
|
|
csum = (csum & 0xffff) + (csum >> 16);
|
|
|
|
return (__u16)~csum;
|
|
}
|
|
|
|
static inline __sum16 csum_tcpudp_magic(__be32 saddr, __be32 daddr,
|
|
__u32 len, __u8 proto,
|
|
__wsum csum)
|
|
{
|
|
__u64 s = csum;
|
|
|
|
s += (__u32)saddr;
|
|
s += (__u32)daddr;
|
|
s += htons(proto + len);
|
|
s = (s & 0xffffffff) + (s >> 32);
|
|
s = (s & 0xffffffff) + (s >> 32);
|
|
|
|
return csum_fold((__u32)s);
|
|
}
|
|
|
|
static inline __sum16 csum_ipv6_magic(const struct in6_addr *saddr,
|
|
const struct in6_addr *daddr,
|
|
__u32 len, __u8 proto,
|
|
__wsum csum)
|
|
{
|
|
__u64 s = csum;
|
|
int i;
|
|
|
|
for (i = 0; i < 4; i++)
|
|
s += (__u32)saddr->s6_addr32[i];
|
|
for (i = 0; i < 4; i++)
|
|
s += (__u32)daddr->s6_addr32[i];
|
|
s += htons(proto + len);
|
|
s = (s & 0xffffffff) + (s >> 32);
|
|
s = (s & 0xffffffff) + (s >> 32);
|
|
|
|
return csum_fold((__u32)s);
|
|
}
|
|
|
|
#endif
|