126 lines
3.1 KiB
C
126 lines
3.1 KiB
C
/*
|
|
* Copyright (c) 2008 Patrick McHardy <kaber@trash.net>
|
|
* Copyright (c) 2012-2013 Pablo Neira Ayuso <pablo@netfilter.org>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* Development of this code funded by Astaro AG (http://www.astaro.com/)
|
|
*/
|
|
|
|
#include <linux/init.h>
|
|
#include <linux/module.h>
|
|
#include <linux/ip.h>
|
|
#include <linux/netfilter_ipv4.h>
|
|
#include <net/netfilter/nf_tables.h>
|
|
#include <net/net_namespace.h>
|
|
#include <net/ip.h>
|
|
#include <net/netfilter/nf_tables_ipv4.h>
|
|
|
|
static unsigned int nft_do_chain_ipv4(const struct nf_hook_ops *ops,
|
|
struct sk_buff *skb,
|
|
const struct nf_hook_state *state)
|
|
{
|
|
struct nft_pktinfo pkt;
|
|
|
|
nft_set_pktinfo_ipv4(&pkt, ops, skb, state);
|
|
|
|
return nft_do_chain(&pkt, ops);
|
|
}
|
|
|
|
static unsigned int nft_ipv4_output(const struct nf_hook_ops *ops,
|
|
struct sk_buff *skb,
|
|
const struct nf_hook_state *state)
|
|
{
|
|
if (unlikely(skb->len < sizeof(struct iphdr) ||
|
|
ip_hdr(skb)->ihl < sizeof(struct iphdr) / 4)) {
|
|
if (net_ratelimit())
|
|
pr_info("nf_tables_ipv4: ignoring short SOCK_RAW "
|
|
"packet\n");
|
|
return NF_ACCEPT;
|
|
}
|
|
|
|
return nft_do_chain_ipv4(ops, skb, state);
|
|
}
|
|
|
|
struct nft_af_info nft_af_ipv4 __read_mostly = {
|
|
.family = NFPROTO_IPV4,
|
|
.nhooks = NF_INET_NUMHOOKS,
|
|
.owner = THIS_MODULE,
|
|
.nops = 1,
|
|
.hooks = {
|
|
[NF_INET_LOCAL_IN] = nft_do_chain_ipv4,
|
|
[NF_INET_LOCAL_OUT] = nft_ipv4_output,
|
|
[NF_INET_FORWARD] = nft_do_chain_ipv4,
|
|
[NF_INET_PRE_ROUTING] = nft_do_chain_ipv4,
|
|
[NF_INET_POST_ROUTING] = nft_do_chain_ipv4,
|
|
},
|
|
};
|
|
EXPORT_SYMBOL_GPL(nft_af_ipv4);
|
|
|
|
static int nf_tables_ipv4_init_net(struct net *net)
|
|
{
|
|
net->nft.ipv4 = kmalloc(sizeof(struct nft_af_info), GFP_KERNEL);
|
|
if (net->nft.ipv4 == NULL)
|
|
return -ENOMEM;
|
|
|
|
memcpy(net->nft.ipv4, &nft_af_ipv4, sizeof(nft_af_ipv4));
|
|
|
|
if (nft_register_afinfo(net, net->nft.ipv4) < 0)
|
|
goto err;
|
|
|
|
return 0;
|
|
err:
|
|
kfree(net->nft.ipv4);
|
|
return -ENOMEM;
|
|
}
|
|
|
|
static void nf_tables_ipv4_exit_net(struct net *net)
|
|
{
|
|
nft_unregister_afinfo(net->nft.ipv4);
|
|
kfree(net->nft.ipv4);
|
|
}
|
|
|
|
static struct pernet_operations nf_tables_ipv4_net_ops = {
|
|
.init = nf_tables_ipv4_init_net,
|
|
.exit = nf_tables_ipv4_exit_net,
|
|
};
|
|
|
|
static const struct nf_chain_type filter_ipv4 = {
|
|
.name = "filter",
|
|
.type = NFT_CHAIN_T_DEFAULT,
|
|
.family = NFPROTO_IPV4,
|
|
.owner = THIS_MODULE,
|
|
.hook_mask = (1 << NF_INET_LOCAL_IN) |
|
|
(1 << NF_INET_LOCAL_OUT) |
|
|
(1 << NF_INET_FORWARD) |
|
|
(1 << NF_INET_PRE_ROUTING) |
|
|
(1 << NF_INET_POST_ROUTING),
|
|
};
|
|
|
|
static int __init nf_tables_ipv4_init(void)
|
|
{
|
|
int ret;
|
|
|
|
nft_register_chain_type(&filter_ipv4);
|
|
ret = register_pernet_subsys(&nf_tables_ipv4_net_ops);
|
|
if (ret < 0)
|
|
nft_unregister_chain_type(&filter_ipv4);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void __exit nf_tables_ipv4_exit(void)
|
|
{
|
|
unregister_pernet_subsys(&nf_tables_ipv4_net_ops);
|
|
nft_unregister_chain_type(&filter_ipv4);
|
|
}
|
|
|
|
module_init(nf_tables_ipv4_init);
|
|
module_exit(nf_tables_ipv4_exit);
|
|
|
|
MODULE_LICENSE("GPL");
|
|
MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
|
|
MODULE_ALIAS_NFT_FAMILY(AF_INET);
|