c2f9eafee9
They don't belong to the family definition, move them to the filter chain type definition instead. Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
153 lines
3.5 KiB
C
153 lines
3.5 KiB
C
/*
|
|
* Copyright (c) 2012-2014 Patrick McHardy <kaber@trash.net>
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#include <linux/init.h>
|
|
#include <linux/module.h>
|
|
#include <linux/ip.h>
|
|
#include <linux/ipv6.h>
|
|
#include <linux/netfilter_ipv4.h>
|
|
#include <linux/netfilter_ipv6.h>
|
|
#include <net/netfilter/nf_tables.h>
|
|
#include <net/netfilter/nf_tables_ipv4.h>
|
|
#include <net/netfilter/nf_tables_ipv6.h>
|
|
#include <net/ip.h>
|
|
|
|
static unsigned int nft_do_chain_inet(void *priv, struct sk_buff *skb,
|
|
const struct nf_hook_state *state)
|
|
{
|
|
struct nft_pktinfo pkt;
|
|
|
|
nft_set_pktinfo(&pkt, skb, state);
|
|
|
|
switch (state->pf) {
|
|
case NFPROTO_IPV4:
|
|
nft_set_pktinfo_ipv4(&pkt, skb);
|
|
break;
|
|
case NFPROTO_IPV6:
|
|
nft_set_pktinfo_ipv6(&pkt, skb);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return nft_do_chain(&pkt, priv);
|
|
}
|
|
|
|
static unsigned int nft_inet_output(void *priv, struct sk_buff *skb,
|
|
const struct nf_hook_state *state)
|
|
{
|
|
struct nft_pktinfo pkt;
|
|
|
|
nft_set_pktinfo(&pkt, skb, state);
|
|
|
|
switch (state->pf) {
|
|
case NFPROTO_IPV4:
|
|
if (unlikely(skb->len < sizeof(struct iphdr) ||
|
|
ip_hdr(skb)->ihl < sizeof(struct iphdr) / 4)) {
|
|
if (net_ratelimit())
|
|
pr_info("ignoring short SOCK_RAW packet\n");
|
|
return NF_ACCEPT;
|
|
}
|
|
nft_set_pktinfo_ipv4(&pkt, skb);
|
|
break;
|
|
case NFPROTO_IPV6:
|
|
if (unlikely(skb->len < sizeof(struct ipv6hdr))) {
|
|
if (net_ratelimit())
|
|
pr_info("ignoring short SOCK_RAW packet\n");
|
|
return NF_ACCEPT;
|
|
}
|
|
nft_set_pktinfo_ipv6(&pkt, skb);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return nft_do_chain(&pkt, priv);
|
|
}
|
|
|
|
static struct nft_af_info nft_af_inet __read_mostly = {
|
|
.family = NFPROTO_INET,
|
|
.nhooks = NF_INET_NUMHOOKS,
|
|
.owner = THIS_MODULE,
|
|
};
|
|
|
|
static int __net_init nf_tables_inet_init_net(struct net *net)
|
|
{
|
|
net->nft.inet = kmalloc(sizeof(struct nft_af_info), GFP_KERNEL);
|
|
if (net->nft.inet == NULL)
|
|
return -ENOMEM;
|
|
memcpy(net->nft.inet, &nft_af_inet, sizeof(nft_af_inet));
|
|
|
|
if (nft_register_afinfo(net, net->nft.inet) < 0)
|
|
goto err;
|
|
|
|
return 0;
|
|
|
|
err:
|
|
kfree(net->nft.inet);
|
|
return -ENOMEM;
|
|
}
|
|
|
|
static void __net_exit nf_tables_inet_exit_net(struct net *net)
|
|
{
|
|
nft_unregister_afinfo(net, net->nft.inet);
|
|
kfree(net->nft.inet);
|
|
}
|
|
|
|
static struct pernet_operations nf_tables_inet_net_ops = {
|
|
.init = nf_tables_inet_init_net,
|
|
.exit = nf_tables_inet_exit_net,
|
|
};
|
|
|
|
static const struct nf_chain_type filter_inet = {
|
|
.name = "filter",
|
|
.type = NFT_CHAIN_T_DEFAULT,
|
|
.family = NFPROTO_INET,
|
|
.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),
|
|
.hooks = {
|
|
[NF_INET_LOCAL_IN] = nft_do_chain_inet,
|
|
[NF_INET_LOCAL_OUT] = nft_inet_output,
|
|
[NF_INET_FORWARD] = nft_do_chain_inet,
|
|
[NF_INET_PRE_ROUTING] = nft_do_chain_inet,
|
|
[NF_INET_POST_ROUTING] = nft_do_chain_inet,
|
|
},
|
|
};
|
|
|
|
static int __init nf_tables_inet_init(void)
|
|
{
|
|
int ret;
|
|
|
|
ret = nft_register_chain_type(&filter_inet);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
ret = register_pernet_subsys(&nf_tables_inet_net_ops);
|
|
if (ret < 0)
|
|
nft_unregister_chain_type(&filter_inet);
|
|
|
|
return ret;
|
|
}
|
|
|
|
static void __exit nf_tables_inet_exit(void)
|
|
{
|
|
unregister_pernet_subsys(&nf_tables_inet_net_ops);
|
|
nft_unregister_chain_type(&filter_inet);
|
|
}
|
|
|
|
module_init(nf_tables_inet_init);
|
|
module_exit(nf_tables_inet_exit);
|
|
|
|
MODULE_LICENSE("GPL");
|
|
MODULE_AUTHOR("Patrick McHardy <kaber@trash.net>");
|
|
MODULE_ALIAS_NFT_FAMILY(1);
|