net/netfilter/nf_flow_table_inet.c

   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 struct nf_flowtable_type flowtable_ipv4 = {
  58         .family         = NFPROTO_IPV4,
  59         .init           = nf_flow_table_init,
  60         .setup          = nf_flow_table_offload_setup,
  61         .action         = nf_flow_rule_route_ipv4,
  62         .free           = nf_flow_table_free,
  63         .hook           = nf_flow_offload_ip_hook,
  64         .owner          = THIS_MODULE,
  65 };
  66
  67 static struct nf_flowtable_type flowtable_ipv6 = {
  68         .family         = NFPROTO_IPV6,
  69         .init           = nf_flow_table_init,
  70         .setup          = nf_flow_table_offload_setup,
  71         .action         = nf_flow_rule_route_ipv6,
  72         .free           = nf_flow_table_free,
  73         .hook           = nf_flow_offload_ipv6_hook,
  74         .owner          = THIS_MODULE,
  75 };
  76
  77 static int __init nf_flow_inet_module_init(void)
  78 {
  79         nft_register_flowtable_type(&flowtable_ipv4);
  80         nft_register_flowtable_type(&flowtable_ipv6);
  81         nft_register_flowtable_type(&flowtable_inet);
  82
  83         return 0;
  84 }
  85
  86 static void __exit nf_flow_inet_module_exit(void)
  87 {
  88         nft_unregister_flowtable_type(&flowtable_inet);
  89         nft_unregister_flowtable_type(&flowtable_ipv6);
  90         nft_unregister_flowtable_type(&flowtable_ipv4);
  91 }
  92
  93 module_init(nf_flow_inet_module_init);
  94 module_exit(nf_flow_inet_module_exit);
  95
  96 MODULE_LICENSE("GPL");
  97 MODULE_AUTHOR("Pablo Neira Ayuso <pablo@netfilter.org>");
  98 MODULE_ALIAS_NF_FLOWTABLE(AF_INET);
  99 MODULE_ALIAS_NF_FLOWTABLE(AF_INET6);
 100 MODULE_ALIAS_NF_FLOWTABLE(1); /* NFPROTO_INET */
 101 MODULE_DESCRIPTION("Netfilter flow table mixed IPv4/IPv6 module");