[linux-2.6-block.git] / net / ipv6 / netfilter / ip6table_nat.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2011 Patrick McHardy <kaber@trash.net>
 *
 * Based on Rusty Russell's IPv4 NAT code. Development of IPv6 NAT
 * funded by Astaro.
 */

#include <linux/module.h>
#include <linux/netfilter.h>
#include <linux/netfilter_ipv6.h>
#include <linux/netfilter_ipv6/ip6_tables.h>
#include <linux/ipv6.h>
#include <net/ipv6.h>

#include <net/netfilter/nf_nat.h>

static int __net_init ip6table_nat_table_init(struct net *net);

static const struct xt_table nf_nat_ipv6_table = {
	.name		= "nat",
	.valid_hooks	= (1 << NF_INET_PRE_ROUTING) |
			  (1 << NF_INET_POST_ROUTING) |
			  (1 << NF_INET_LOCAL_OUT) |
			  (1 << NF_INET_LOCAL_IN),
	.me		= THIS_MODULE,
	.af		= NFPROTO_IPV6,
	.table_init	= ip6table_nat_table_init,
};

static unsigned int ip6table_nat_do_chain(void *priv,
					  struct sk_buff *skb,
					  const struct nf_hook_state *state)
{
	return ip6t_do_table(skb, state, state->net->ipv6.ip6table_nat);
}

static const struct nf_hook_ops nf_nat_ipv6_ops[] = {
	{
		.hook		= ip6table_nat_do_chain,
		.pf		= NFPROTO_IPV6,
		.hooknum	= NF_INET_PRE_ROUTING,
		.priority	= NF_IP6_PRI_NAT_DST,
	},
	{
		.hook		= ip6table_nat_do_chain,
		.pf		= NFPROTO_IPV6,
		.hooknum	= NF_INET_POST_ROUTING,
		.priority	= NF_IP6_PRI_NAT_SRC,
	},
	{
		.hook		= ip6table_nat_do_chain,
		.pf		= NFPROTO_IPV6,
		.hooknum	= NF_INET_LOCAL_OUT,
		.priority	= NF_IP6_PRI_NAT_DST,
	},
	{
		.hook		= ip6table_nat_do_chain,
		.pf		= NFPROTO_IPV6,
		.hooknum	= NF_INET_LOCAL_IN,
		.priority	= NF_IP6_PRI_NAT_SRC,
	},
};

static int ip6t_nat_register_lookups(struct net *net)
{
	int i, ret;

	for (i = 0; i < ARRAY_SIZE(nf_nat_ipv6_ops); i++) {
		ret = nf_nat_ipv6_register_fn(net, &nf_nat_ipv6_ops[i]);
		if (ret) {
			while (i)
				nf_nat_ipv6_unregister_fn(net, &nf_nat_ipv6_ops[--i]);

			return ret;
		}
	}

	return 0;
}

static void ip6t_nat_unregister_lookups(struct net *net)
{
	int i;

	for (i = 0; i < ARRAY_SIZE(nf_nat_ipv6_ops); i++)
		nf_nat_ipv6_unregister_fn(net, &nf_nat_ipv6_ops[i]);
}

static int __net_init ip6table_nat_table_init(struct net *net)
{
	struct ip6t_replace *repl;
	int ret;

	if (net->ipv6.ip6table_nat)
		return 0;

	repl = ip6t_alloc_initial_table(&nf_nat_ipv6_table);
	if (repl == NULL)
		return -ENOMEM;
	ret = ip6t_register_table(net, &nf_nat_ipv6_table, repl,
				  NULL, &net->ipv6.ip6table_nat);
	if (ret < 0) {
		kfree(repl);
		return ret;
	}

	ret = ip6t_nat_register_lookups(net);
	if (ret < 0) {
		ip6t_unregister_table(net, net->ipv6.ip6table_nat, NULL);
		net->ipv6.ip6table_nat = NULL;
	}
	kfree(repl);
	return ret;
}

static void __net_exit ip6table_nat_net_exit(struct net *net)
{
	if (!net->ipv6.ip6table_nat)
		return;
	ip6t_nat_unregister_lookups(net);
	ip6t_unregister_table(net, net->ipv6.ip6table_nat, NULL);
	net->ipv6.ip6table_nat = NULL;
}

static struct pernet_operations ip6table_nat_net_ops = {
	.exit	= ip6table_nat_net_exit,
};

static int __init ip6table_nat_init(void)
{
	int ret = register_pernet_subsys(&ip6table_nat_net_ops);

	if (ret)
		return ret;

	ret = ip6table_nat_table_init(&init_net);
	if (ret)
		unregister_pernet_subsys(&ip6table_nat_net_ops);
	return ret;
}

static void __exit ip6table_nat_exit(void)
{
	unregister_pernet_subsys(&ip6table_nat_net_ops);
}

module_init(ip6table_nat_init);
module_exit(ip6table_nat_exit);

MODULE_LICENSE("GPL");
Commit	Line	Data
d2912cb1	1	// SPDX-License-Identifier: GPL-2.0-only
58a317f1 PM	2	/*
	3	* Copyright (c) 2011 Patrick McHardy <kaber@trash.net>
	4	*
58a317f1 PM	5	* Based on Rusty Russell's IPv4 NAT code. Development of IPv6 NAT
	6	* funded by Astaro.
	7	*/
	8
	9	#include <linux/module.h>
	10	#include <linux/netfilter.h>
	11	#include <linux/netfilter_ipv6.h>
	12	#include <linux/netfilter_ipv6/ip6_tables.h>
	13	#include <linux/ipv6.h>
	14	#include <net/ipv6.h>
	15
	16	#include <net/netfilter/nf_nat.h>
58a317f1	17
b9e69e12 FW	18	static int __net_init ip6table_nat_table_init(struct net *net);
b9e69e12 FW	19
58a317f1 PM	20	static const struct xt_table nf_nat_ipv6_table = {
	21	.name = "nat",
	22	.valid_hooks = (1 << NF_INET_PRE_ROUTING) \|
	23	(1 << NF_INET_POST_ROUTING) \|
	24	(1 << NF_INET_LOCAL_OUT) \|
	25	(1 << NF_INET_LOCAL_IN),
	26	.me = THIS_MODULE,
	27	.af = NFPROTO_IPV6,
b9e69e12	28	.table_init = ip6table_nat_table_init,
58a317f1 PM	29	};
58a317f1 PM	30
06198b34	31	static unsigned int ip6table_nat_do_chain(void *priv,
2a5538e9	32	struct sk_buff *skb,
3a2e86f6	33	const struct nf_hook_state *state)
58a317f1	34	{
6cb8ff3f	35	return ip6t_do_table(skb, state, state->net->ipv6.ip6table_nat);
58a317f1 PM	36	}
58a317f1 PM	37
591bb278	38	static const struct nf_hook_ops nf_nat_ipv6_ops[] = {
58a317f1	39	{
9971a514	40	.hook = ip6table_nat_do_chain,
58a317f1 PM	41	.pf = NFPROTO_IPV6,
	42	.hooknum = NF_INET_PRE_ROUTING,
	43	.priority = NF_IP6_PRI_NAT_DST,
	44	},
58a317f1	45	{
9971a514	46	.hook = ip6table_nat_do_chain,
58a317f1 PM	47	.pf = NFPROTO_IPV6,
	48	.hooknum = NF_INET_POST_ROUTING,
	49	.priority = NF_IP6_PRI_NAT_SRC,
	50	},
58a317f1	51	{
9971a514	52	.hook = ip6table_nat_do_chain,
58a317f1 PM	53	.pf = NFPROTO_IPV6,
	54	.hooknum = NF_INET_LOCAL_OUT,
	55	.priority = NF_IP6_PRI_NAT_DST,
	56	},
58a317f1	57	{
9971a514	58	.hook = ip6table_nat_do_chain,
58a317f1 PM	59	.pf = NFPROTO_IPV6,
	60	.hooknum = NF_INET_LOCAL_IN,
	61	.priority = NF_IP6_PRI_NAT_SRC,
	62	},
	63	};
	64
9971a514 FW	65	static int ip6t_nat_register_lookups(struct net *net)
	66	{
	67	int i, ret;
	68
	69	for (i = 0; i < ARRAY_SIZE(nf_nat_ipv6_ops); i++) {
d2c5c103	70	ret = nf_nat_ipv6_register_fn(net, &nf_nat_ipv6_ops[i]);
9971a514 FW	71	if (ret) {
9971a514 FW	72	while (i)
d2c5c103	73	nf_nat_ipv6_unregister_fn(net, &nf_nat_ipv6_ops[--i]);
9971a514 FW	74
	75	return ret;
	76	}
	77	}
	78
	79	return 0;
	80	}
	81
	82	static void ip6t_nat_unregister_lookups(struct net *net)
	83	{
	84	int i;
	85
	86	for (i = 0; i < ARRAY_SIZE(nf_nat_ipv6_ops); i++)
d2c5c103	87	nf_nat_ipv6_unregister_fn(net, &nf_nat_ipv6_ops[i]);
9971a514 FW	88	}
9971a514 FW	89
b9e69e12	90	static int __net_init ip6table_nat_table_init(struct net *net)
58a317f1 PM	91	{
58a317f1 PM	92	struct ip6t_replace *repl;
a67dd266	93	int ret;
58a317f1	94
b9e69e12 FW	95	if (net->ipv6.ip6table_nat)
	96	return 0;
	97
58a317f1 PM	98	repl = ip6t_alloc_initial_table(&nf_nat_ipv6_table);
	99	if (repl == NULL)
	100	return -ENOMEM;
a67dd266	101	ret = ip6t_register_table(net, &nf_nat_ipv6_table, repl,
9971a514 FW	102	NULL, &net->ipv6.ip6table_nat);
	103	if (ret < 0) {
	104	kfree(repl);
	105	return ret;
	106	}
	107
	108	ret = ip6t_nat_register_lookups(net);
	109	if (ret < 0) {
	110	ip6t_unregister_table(net, net->ipv6.ip6table_nat, NULL);
	111	net->ipv6.ip6table_nat = NULL;
	112	}
58a317f1	113	kfree(repl);
a67dd266	114	return ret;
58a317f1 PM	115	}
	116
	117	static void __net_exit ip6table_nat_net_exit(struct net *net)
	118	{
b9e69e12 FW	119	if (!net->ipv6.ip6table_nat)
b9e69e12 FW	120	return;
9971a514 FW	121	ip6t_nat_unregister_lookups(net);
9971a514 FW	122	ip6t_unregister_table(net, net->ipv6.ip6table_nat, NULL);
b9e69e12	123	net->ipv6.ip6table_nat = NULL;
58a317f1 PM	124	}
	125
	126	static struct pernet_operations ip6table_nat_net_ops = {
58a317f1 PM	127	.exit = ip6table_nat_net_exit,
	128	};
	129
	130	static int __init ip6table_nat_init(void)
	131	{
b9e69e12	132	int ret = register_pernet_subsys(&ip6table_nat_net_ops);
58a317f1	133
b9e69e12 FW	134	if (ret)
b9e69e12 FW	135	return ret;
58a317f1	136
b9e69e12 FW	137	ret = ip6table_nat_table_init(&init_net);
	138	if (ret)
	139	unregister_pernet_subsys(&ip6table_nat_net_ops);
	140	return ret;
58a317f1 PM	141	}
	142
	143	static void __exit ip6table_nat_exit(void)
	144	{
58a317f1 PM	145	unregister_pernet_subsys(&ip6table_nat_net_ops);
	146	}
	147
	148	module_init(ip6table_nat_init);
	149	module_exit(ip6table_nat_exit);
	150
	151	MODULE_LICENSE("GPL");