1 // SPDX-License-Identifier: GPL-2.0-only
2 #include <linux/kernel.h>
3 #include <linux/init.h>
4 #include <linux/module.h>
5 #include <linux/netfilter.h>
6 #include <linux/rhashtable.h>
7 #include <net/netfilter/nf_flow_table.h>
8 #include <net/netfilter/nf_tables.h>
9 
10 static unsigned int
11 nf_flow_offload_inet_hook(void *priv, struct sk_buff *skb,
12 			  const struct nf_hook_state *state)
13 {
14 	switch (skb->protocol) {
15 	case htons(ETH_P_IP):
16 		return nf_flow_offload_ip_hook(priv, skb, state);
17 	case htons(ETH_P_IPV6):
18 		return nf_flow_offload_ipv6_hook(priv, skb, state);
19 	}
20 
21 	return NF_ACCEPT;
22 }
23 
24 static int nf_flow_rule_route_inet(struct net *net,
25 				   const struct flow_offload *flow,
26 				   enum flow_offload_tuple_dir dir,
27 				   struct nf_flow_rule *flow_rule)
28 {
29 	const struct flow_offload_tuple *flow_tuple = &flow->tuplehash[dir].tuple;
30 	int err;
31 
32 	switch (flow_tuple->l3proto) {
33 	case NFPROTO_IPV4:
34 		err = nf_flow_rule_route_ipv4(net, flow, dir, flow_rule);
35 		break;
36 	case NFPROTO_IPV6:
37 		err = nf_flow_rule_route_ipv6(net, flow, dir, flow_rule);
38 		break;
39 	default:
40 		err = -1;
41 		break;
42 	}
43 
44 	return err;
45 }
46 
47 static struct nf_flowtable_type flowtable_inet = {
48 	.family		= NFPROTO_INET,
49 	.init		= nf_flow_table_init,
50 	.setup		= nf_flow_table_offload_setup,
51 	.action		= nf_flow_rule_route_inet,
52 	.free		= nf_flow_table_free,
53 	.hook		= nf_flow_offload_inet_hook,
54 	.owner		= THIS_MODULE,
55 };
56 
57 static int __init nf_flow_inet_module_init(void)
58 {
59 	nft_register_flowtable_type(&flowtable_inet);
60 
61 	return 0;
62 }
63 
64 static void __exit nf_flow_inet_module_exit(void)
65 {
66 	nft_unregister_flowtable_type(&flowtable_inet);
67 }
68 
69 module_init(nf_flow_inet_module_init);
70 module_exit(nf_flow_inet_module_exit);
71 
72 MODULE_LICENSE("GPL");
73 MODULE_AUTHOR("Pablo Neira Ayuso <pablo@netfilter.org>");
74 MODULE_ALIAS_NF_FLOWTABLE(1); /* NFPROTO_INET */
75 MODULE_DESCRIPTION("Netfilter flow table mixed IPv4/IPv6 module");
76