* Public Domain.

/*
* 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; }

extern const npf_mbufops_t npftest_mbufops;
extern 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);
struct mbuf * mbuf_get_frame(const char *, const char *, uint16_t);
npf_cache_t * get_cached_pkt(struct mbuf *, const char *, uint32_t);
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_layer2_rule_test(bool);
bool npf_nat_test(bool);
bool npf_guid_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