/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2015-2019 Yandex LLC
 * Copyright (c) 2015-2019 Andrey V. Elsukov <ae@FreeBSD.org>
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#ifndef	_IP_FW_NAT64_H_
#define	_IP_FW_NAT64_H_

#define	DPRINTF(mask, fmt, ...)	\
    if (V_nat64_debug & (mask))	\
	printf("NAT64: %s: " fmt "\n", __func__, ## __VA_ARGS__)
#define	DP_GENERIC	0x0001
#define	DP_OBJ		0x0002
#define	DP_JQUEUE	0x0004
#define	DP_STATE	0x0008
#define	DP_DROPS	0x0010
#define	DP_ALL		0xFFFF

VNET_DECLARE(int, nat64_debug);
#define	V_nat64_debug		VNET(nat64_debug)

#if 0
#define	NAT64NOINLINE	__noinline
#else
#define	NAT64NOINLINE
#endif

int	nat64stl_init(struct ip_fw_chain *ch, int first);
void	nat64stl_uninit(struct ip_fw_chain *ch, int last);
int	nat64lsn_init(struct ip_fw_chain *ch, int first);
void	nat64lsn_uninit(struct ip_fw_chain *ch, int last);
int	nat64clat_init(struct ip_fw_chain *ch, int first);
void	nat64clat_uninit(struct ip_fw_chain *ch, int last);

#define	NAT64_DEFINE_OPCODE_REWRITER(mod, name, ops)			\
static int								\
mod ## _classify(ipfw_insn *cmd0, uint32_t *puidx, uint8_t *ptype)	\
{									\
	ipfw_insn *icmd;						\
	icmd = cmd0 - F_LEN(cmd0);					\
	if (icmd->opcode != O_EXTERNAL_ACTION ||			\
	    insntod(icmd, kidx)->kidx != V_ ## mod ## _eid)		\
		return (1);						\
	*puidx = insntod(cmd0, kidx)->kidx;				\
	*ptype = 0;							\
	return (0);							\
}									\
static void								\
mod ## _update_kidx(ipfw_insn *cmd0, uint32_t idx)			\
{									\
	insntod(cmd0, kidx)->kidx = idx;				\
}									\
static int								\
mod ## _findbyname(struct ip_fw_chain *ch, struct tid_info *ti,		\
    struct named_object **pno)						\
{									\
	return (ipfw_objhash_find_type(CHAIN_TO_SRV(ch), ti,		\
	    IPFW_TLV_## name ## _NAME, pno));				\
}									\
static struct named_object *						\
mod ## _findbykidx(struct ip_fw_chain *ch, uint32_t idx)		\
{									\
	struct namedobj_instance *ni;					\
	struct named_object *no;					\
	IPFW_UH_WLOCK_ASSERT(ch);					\
	ni = CHAIN_TO_SRV(ch);						\
	no = ipfw_objhash_lookup_kidx(ni, idx);				\
	KASSERT(no != NULL, ("NAT with index %u not found", idx));	\
	return (no);							\
}									\
static struct opcode_obj_rewrite ops[] = {				\
	{								\
		.opcode = O_EXTERNAL_INSTANCE,				\
		.etlv = IPFW_TLV_EACTION /* just show it isn't table */,\
		.classifier = mod ## _classify,				\
		.update = mod ## _update_kidx,				\
		.find_byname = mod ## _findbyname,			\
		.find_bykidx = mod ## _findbykidx,			\
		.manage_sets = mod ## _manage_sets,			\
	},								\
}

#endif /* _IP_FW_NAT64_H_ */