xref: /src/usr.sbin/npf/npftest/libnpftest/npf_test.h

/*
 * Public Domain.
 */

#ifndef _LIB_NPF_TEST_H_
#define _LIB_NPF_TEST_H_

#ifdef _KERNEL
#include <sys/types.h>
#include <sys/mbuf.h>

#include <netinet/in_systm.h>
#include <netinet/in.h>
#include <netinet6/in6.h>

#include <netinet/ip.h>
#include <netinet/ip6.h>
#include <netinet/tcp.h>
#include <netinet/udp.h>
#include <netinet/ip_icmp.h>

#include <net/if.h>
#include <net/if_ether.h>
#include <net/ethertypes.h>
#endif

#define	IFNAME_DUMMY	"npftest999"

/* Test interfaces and IP addresses. */
#define	IFNAME_EXT	"npftest0"
#define	IFNAME_INT	"npftest1"
#define	IFNAME_TEST	"npftest2"

#define	LOCAL_IP1	"10.1.1.1"
#define	LOCAL_IP2	"10.1.1.2"
#define	LOCAL_IP3	"10.1.1.3"

/* Note: RFC 5737 compliant addresses. */
#define	PUB_IP1		"192.0.2.1"
#define	PUB_IP2		"192.0.2.2"
#define	PUB_IP3		"192.0.2.3"

#define	REMOTE_IP1	"192.0.2.101"
#define	REMOTE_IP2	"192.0.2.102"
#define	REMOTE_IP3	"192.0.2.103"
#define	REMOTE_IP4	"192.0.2.104"

#define	LOCAL_IP6	"fd01:203:405:1::1234"
#define	REMOTE_IP6	"2001:db8:fefe::1010"
#define	EXPECTED_IP6	"2001:db8:1:d550::1234"

#define	NET_A_IP1	"10.100.7.126"
#define	NET_B_IP1	"10.255.7.126"

#if defined(_NPF_STANDALONE)

#define	MLEN		512

struct mbuf {
	unsigned	m_flags;
	int		m_type;
	unsigned	m_len;
	void *		m_next;
	struct {
		int	len;
	} m_pkthdr;
	void *		m_data;
	unsigned char	m_data0[MLEN];
};

#define	MT_FREE			0
#define	M_UNWRITABLE(m, l)	false
#define	M_NOWAIT		0x00001
#define M_PKTHDR		0x00002

#define	m_get(x, y)		npfkern_m_get(NULL, 0, MLEN)
#define	m_gethdr(x, y)		npfkern_m_get(NULL, M_PKTHDR, MLEN)
#define	m_length(m)		npfkern_m_length(m)
#define	m_freem(m)		npfkern_m_freem(m)
#define	mtod(m, t)		((t)((m)->m_data))

#endif

#define	CHECK_TRUE(x)	\
    if (!(x)) { printf("FAIL: %s line %d\n", __func__, __LINE__); return 0; }

const npf_mbufops_t	npftest_mbufops;
const npf_ifops_t	npftest_ifops;

struct mbuf *	npfkern_m_get(npf_t *, unsigned, size_t);
size_t		npfkern_m_length(const struct mbuf *);
void		npfkern_m_freem(struct mbuf *);

void		npf_test_init(int (*)(int, const char *, void *),
		    const char *(*)(int, const void *, char *, socklen_t),
		    long (*)(void));
void		npf_test_fini(void);
int		npf_test_load(const void *, size_t, bool);
ifnet_t *	npf_test_addif(const char *, bool, bool);
ifnet_t *	npf_test_getif(const char *);

int		npf_test_statetrack(const void *, size_t, ifnet_t *,
		    bool, int64_t *);
void		npf_test_conc(bool, unsigned);

struct mbuf *	mbuf_getwithdata(const void *, size_t);
struct mbuf *	mbuf_construct_ether(int);
struct mbuf *	mbuf_construct(int);
struct mbuf *	mbuf_construct6(int);
void *		mbuf_return_hdrs(struct mbuf *, bool, struct ip **);
void *		mbuf_return_hdrs6(struct mbuf *, struct ip6_hdr **);
void		mbuf_icmp_append(struct mbuf *, struct mbuf *);

struct mbuf *	mbuf_get_pkt(int, int, const char *, const char *, int, int);
npf_cache_t *	get_cached_pkt(struct mbuf *, const char *);
void		put_cached_pkt(npf_cache_t *);

bool		npf_nbuf_test(bool);
bool		npf_bpf_test(bool);
bool		npf_table_test(bool, void *, size_t);
bool		npf_state_test(bool);

bool		npf_rule_test(bool);
bool		npf_nat_test(bool);
bool		npf_gc_test(bool);

int		npf_inet_pton(int, const char *, void *);
const char *	npf_inet_ntop(int, const void *, char *, socklen_t);

#endif