1b57dc7c1SPaul Blakey // SPDX-License-Identifier: GPL-2.0 OR Linux-OpenIB 2b57dc7c1SPaul Blakey /* - 3b57dc7c1SPaul Blakey * net/sched/act_ct.c Connection Tracking action 4b57dc7c1SPaul Blakey * 5b57dc7c1SPaul Blakey * Authors: Paul Blakey <paulb@mellanox.com> 6b57dc7c1SPaul Blakey * Yossi Kuperman <yossiku@mellanox.com> 7b57dc7c1SPaul Blakey * Marcelo Ricardo Leitner <marcelo.leitner@gmail.com> 8b57dc7c1SPaul Blakey */ 9b57dc7c1SPaul Blakey 10b57dc7c1SPaul Blakey #include <linux/module.h> 11b57dc7c1SPaul Blakey #include <linux/init.h> 12b57dc7c1SPaul Blakey #include <linux/kernel.h> 13b57dc7c1SPaul Blakey #include <linux/skbuff.h> 14b57dc7c1SPaul Blakey #include <linux/rtnetlink.h> 15b57dc7c1SPaul Blakey #include <linux/pkt_cls.h> 16b57dc7c1SPaul Blakey #include <linux/ip.h> 17b57dc7c1SPaul Blakey #include <linux/ipv6.h> 18c34b961aSPaul Blakey #include <linux/rhashtable.h> 19b57dc7c1SPaul Blakey #include <net/netlink.h> 20b57dc7c1SPaul Blakey #include <net/pkt_sched.h> 21b57dc7c1SPaul Blakey #include <net/pkt_cls.h> 22b57dc7c1SPaul Blakey #include <net/act_api.h> 23b57dc7c1SPaul Blakey #include <net/ip.h> 24b57dc7c1SPaul Blakey #include <net/ipv6_frag.h> 25b57dc7c1SPaul Blakey #include <uapi/linux/tc_act/tc_ct.h> 26b57dc7c1SPaul Blakey #include <net/tc_act/tc_ct.h> 27871cf386SPedro Tammela #include <net/tc_wrapper.h> 28b57dc7c1SPaul Blakey 29c34b961aSPaul Blakey #include <net/netfilter/nf_flow_table.h> 30b57dc7c1SPaul Blakey #include <net/netfilter/nf_conntrack.h> 31b57dc7c1SPaul Blakey #include <net/netfilter/nf_conntrack_core.h> 32b57dc7c1SPaul Blakey #include <net/netfilter/nf_conntrack_zones.h> 33b57dc7c1SPaul Blakey #include <net/netfilter/nf_conntrack_helper.h> 34beb97d3aSwenxu #include <net/netfilter/nf_conntrack_acct.h> 35b57dc7c1SPaul Blakey #include <net/netfilter/ipv6/nf_defrag_ipv6.h> 369795ded7SPaul Blakey #include <net/netfilter/nf_conntrack_act_ct.h> 37a21b06e7SXin Long #include <net/netfilter/nf_conntrack_seqadj.h> 3840d102cdSJeremy Sowden #include <uapi/linux/netfilter/nf_nat.h> 39b57dc7c1SPaul Blakey 40c34b961aSPaul Blakey static struct workqueue_struct *act_ct_wq; 41c34b961aSPaul Blakey static struct rhashtable zones_ht; 42138470a9SEric Dumazet static DEFINE_MUTEX(zones_mutex); 43c34b961aSPaul Blakey 44c34b961aSPaul Blakey struct tcf_ct_flow_table { 45c34b961aSPaul Blakey struct rhash_head node; /* In zones tables */ 46c34b961aSPaul Blakey 47c34b961aSPaul Blakey struct rcu_work rwork; 48c34b961aSPaul Blakey struct nf_flowtable nf_ft; 49138470a9SEric Dumazet refcount_t ref; 50c34b961aSPaul Blakey u16 zone; 51c34b961aSPaul Blakey 52c34b961aSPaul Blakey bool dying; 53c34b961aSPaul Blakey }; 54c34b961aSPaul Blakey 55c34b961aSPaul Blakey static const struct rhashtable_params zones_params = { 56c34b961aSPaul Blakey .head_offset = offsetof(struct tcf_ct_flow_table, node), 57c34b961aSPaul Blakey .key_offset = offsetof(struct tcf_ct_flow_table, zone), 58c34b961aSPaul Blakey .key_len = sizeof_field(struct tcf_ct_flow_table, zone), 59c34b961aSPaul Blakey .automatic_shrinking = true, 60c34b961aSPaul Blakey }; 61c34b961aSPaul Blakey 629c26ba9bSPaul Blakey static struct flow_action_entry * 639c26ba9bSPaul Blakey tcf_ct_flow_table_flow_action_get_next(struct flow_action *flow_action) 649c26ba9bSPaul Blakey { 659c26ba9bSPaul Blakey int i = flow_action->num_entries++; 669c26ba9bSPaul Blakey 679c26ba9bSPaul Blakey return &flow_action->entries[i]; 689c26ba9bSPaul Blakey } 699c26ba9bSPaul Blakey 709c26ba9bSPaul Blakey static void tcf_ct_add_mangle_action(struct flow_action *action, 719c26ba9bSPaul Blakey enum flow_action_mangle_base htype, 729c26ba9bSPaul Blakey u32 offset, 739c26ba9bSPaul Blakey u32 mask, 749c26ba9bSPaul Blakey u32 val) 759c26ba9bSPaul Blakey { 769c26ba9bSPaul Blakey struct flow_action_entry *entry; 779c26ba9bSPaul Blakey 789c26ba9bSPaul Blakey entry = tcf_ct_flow_table_flow_action_get_next(action); 799c26ba9bSPaul Blakey entry->id = FLOW_ACTION_MANGLE; 809c26ba9bSPaul Blakey entry->mangle.htype = htype; 819c26ba9bSPaul Blakey entry->mangle.mask = ~mask; 829c26ba9bSPaul Blakey entry->mangle.offset = offset; 839c26ba9bSPaul Blakey entry->mangle.val = val; 849c26ba9bSPaul Blakey } 859c26ba9bSPaul Blakey 869c26ba9bSPaul Blakey /* The following nat helper functions check if the inverted reverse tuple 879c26ba9bSPaul Blakey * (target) is different then the current dir tuple - meaning nat for ports 889c26ba9bSPaul Blakey * and/or ip is needed, and add the relevant mangle actions. 899c26ba9bSPaul Blakey */ 909c26ba9bSPaul Blakey static void 919c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_ipv4(const struct nf_conntrack_tuple *tuple, 929c26ba9bSPaul Blakey struct nf_conntrack_tuple target, 939c26ba9bSPaul Blakey struct flow_action *action) 949c26ba9bSPaul Blakey { 959c26ba9bSPaul Blakey if (memcmp(&target.src.u3, &tuple->src.u3, sizeof(target.src.u3))) 969c26ba9bSPaul Blakey tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_IP4, 979c26ba9bSPaul Blakey offsetof(struct iphdr, saddr), 989c26ba9bSPaul Blakey 0xFFFFFFFF, 999c26ba9bSPaul Blakey be32_to_cpu(target.src.u3.ip)); 1009c26ba9bSPaul Blakey if (memcmp(&target.dst.u3, &tuple->dst.u3, sizeof(target.dst.u3))) 1019c26ba9bSPaul Blakey tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_IP4, 1029c26ba9bSPaul Blakey offsetof(struct iphdr, daddr), 1039c26ba9bSPaul Blakey 0xFFFFFFFF, 1049c26ba9bSPaul Blakey be32_to_cpu(target.dst.u3.ip)); 1059c26ba9bSPaul Blakey } 1069c26ba9bSPaul Blakey 1079c26ba9bSPaul Blakey static void 1089c26ba9bSPaul Blakey tcf_ct_add_ipv6_addr_mangle_action(struct flow_action *action, 1099c26ba9bSPaul Blakey union nf_inet_addr *addr, 1109c26ba9bSPaul Blakey u32 offset) 1119c26ba9bSPaul Blakey { 1129c26ba9bSPaul Blakey int i; 1139c26ba9bSPaul Blakey 1149c26ba9bSPaul Blakey for (i = 0; i < sizeof(struct in6_addr) / sizeof(u32); i++) 1159c26ba9bSPaul Blakey tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_IP6, 1169c26ba9bSPaul Blakey i * sizeof(u32) + offset, 1179c26ba9bSPaul Blakey 0xFFFFFFFF, be32_to_cpu(addr->ip6[i])); 1189c26ba9bSPaul Blakey } 1199c26ba9bSPaul Blakey 1209c26ba9bSPaul Blakey static void 1219c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_ipv6(const struct nf_conntrack_tuple *tuple, 1229c26ba9bSPaul Blakey struct nf_conntrack_tuple target, 1239c26ba9bSPaul Blakey struct flow_action *action) 1249c26ba9bSPaul Blakey { 1259c26ba9bSPaul Blakey if (memcmp(&target.src.u3, &tuple->src.u3, sizeof(target.src.u3))) 1269c26ba9bSPaul Blakey tcf_ct_add_ipv6_addr_mangle_action(action, &target.src.u3, 1279c26ba9bSPaul Blakey offsetof(struct ipv6hdr, 1289c26ba9bSPaul Blakey saddr)); 1299c26ba9bSPaul Blakey if (memcmp(&target.dst.u3, &tuple->dst.u3, sizeof(target.dst.u3))) 1309c26ba9bSPaul Blakey tcf_ct_add_ipv6_addr_mangle_action(action, &target.dst.u3, 1319c26ba9bSPaul Blakey offsetof(struct ipv6hdr, 1329c26ba9bSPaul Blakey daddr)); 1339c26ba9bSPaul Blakey } 1349c26ba9bSPaul Blakey 1359c26ba9bSPaul Blakey static void 1369c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_tcp(const struct nf_conntrack_tuple *tuple, 1379c26ba9bSPaul Blakey struct nf_conntrack_tuple target, 1389c26ba9bSPaul Blakey struct flow_action *action) 1399c26ba9bSPaul Blakey { 1409c26ba9bSPaul Blakey __be16 target_src = target.src.u.tcp.port; 1419c26ba9bSPaul Blakey __be16 target_dst = target.dst.u.tcp.port; 1429c26ba9bSPaul Blakey 1439c26ba9bSPaul Blakey if (target_src != tuple->src.u.tcp.port) 1449c26ba9bSPaul Blakey tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_TCP, 1459c26ba9bSPaul Blakey offsetof(struct tcphdr, source), 1469c26ba9bSPaul Blakey 0xFFFF, be16_to_cpu(target_src)); 1479c26ba9bSPaul Blakey if (target_dst != tuple->dst.u.tcp.port) 1489c26ba9bSPaul Blakey tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_TCP, 1499c26ba9bSPaul Blakey offsetof(struct tcphdr, dest), 1509c26ba9bSPaul Blakey 0xFFFF, be16_to_cpu(target_dst)); 1519c26ba9bSPaul Blakey } 1529c26ba9bSPaul Blakey 1539c26ba9bSPaul Blakey static void 1549c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_udp(const struct nf_conntrack_tuple *tuple, 1559c26ba9bSPaul Blakey struct nf_conntrack_tuple target, 1569c26ba9bSPaul Blakey struct flow_action *action) 1579c26ba9bSPaul Blakey { 1589c26ba9bSPaul Blakey __be16 target_src = target.src.u.udp.port; 1599c26ba9bSPaul Blakey __be16 target_dst = target.dst.u.udp.port; 1609c26ba9bSPaul Blakey 1619c26ba9bSPaul Blakey if (target_src != tuple->src.u.udp.port) 16247b5d2a1SRoi Dayan tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_UDP, 1639c26ba9bSPaul Blakey offsetof(struct udphdr, source), 1649c26ba9bSPaul Blakey 0xFFFF, be16_to_cpu(target_src)); 1659c26ba9bSPaul Blakey if (target_dst != tuple->dst.u.udp.port) 16647b5d2a1SRoi Dayan tcf_ct_add_mangle_action(action, FLOW_ACT_MANGLE_HDR_TYPE_UDP, 1679c26ba9bSPaul Blakey offsetof(struct udphdr, dest), 1689c26ba9bSPaul Blakey 0xFFFF, be16_to_cpu(target_dst)); 1699c26ba9bSPaul Blakey } 1709c26ba9bSPaul Blakey 1719c26ba9bSPaul Blakey static void tcf_ct_flow_table_add_action_meta(struct nf_conn *ct, 1729c26ba9bSPaul Blakey enum ip_conntrack_dir dir, 1731a441a9bSVlad Buslov enum ip_conntrack_info ctinfo, 1749c26ba9bSPaul Blakey struct flow_action *action) 1759c26ba9bSPaul Blakey { 1769c26ba9bSPaul Blakey struct nf_conn_labels *ct_labels; 1779c26ba9bSPaul Blakey struct flow_action_entry *entry; 1789c26ba9bSPaul Blakey u32 *act_ct_labels; 1799c26ba9bSPaul Blakey 1809c26ba9bSPaul Blakey entry = tcf_ct_flow_table_flow_action_get_next(action); 1819c26ba9bSPaul Blakey entry->id = FLOW_ACTION_CT_METADATA; 1829c26ba9bSPaul Blakey #if IS_ENABLED(CONFIG_NF_CONNTRACK_MARK) 18352d1aa8bSDaniel Xu entry->ct_metadata.mark = READ_ONCE(ct->mark); 1849c26ba9bSPaul Blakey #endif 18530b0cf90SPaul Blakey /* aligns with the CT reference on the SKB nf_ct_set */ 18630b0cf90SPaul Blakey entry->ct_metadata.cookie = (unsigned long)ct | ctinfo; 187941eff5aSPaul Blakey entry->ct_metadata.orig_dir = dir == IP_CT_DIR_ORIGINAL; 1889c26ba9bSPaul Blakey 1899c26ba9bSPaul Blakey act_ct_labels = entry->ct_metadata.labels; 1909c26ba9bSPaul Blakey ct_labels = nf_ct_labels_find(ct); 1919c26ba9bSPaul Blakey if (ct_labels) 1929c26ba9bSPaul Blakey memcpy(act_ct_labels, ct_labels->bits, NF_CT_LABELS_MAX_SIZE); 1939c26ba9bSPaul Blakey else 1949c26ba9bSPaul Blakey memset(act_ct_labels, 0, NF_CT_LABELS_MAX_SIZE); 1959c26ba9bSPaul Blakey } 1969c26ba9bSPaul Blakey 1979c26ba9bSPaul Blakey static int tcf_ct_flow_table_add_action_nat(struct net *net, 1989c26ba9bSPaul Blakey struct nf_conn *ct, 1999c26ba9bSPaul Blakey enum ip_conntrack_dir dir, 2009c26ba9bSPaul Blakey struct flow_action *action) 2019c26ba9bSPaul Blakey { 2029c26ba9bSPaul Blakey const struct nf_conntrack_tuple *tuple = &ct->tuplehash[dir].tuple; 2039c26ba9bSPaul Blakey struct nf_conntrack_tuple target; 2049c26ba9bSPaul Blakey 20505aa69e5Swenxu if (!(ct->status & IPS_NAT_MASK)) 20605aa69e5Swenxu return 0; 20705aa69e5Swenxu 2089c26ba9bSPaul Blakey nf_ct_invert_tuple(&target, &ct->tuplehash[!dir].tuple); 2099c26ba9bSPaul Blakey 2109c26ba9bSPaul Blakey switch (tuple->src.l3num) { 2119c26ba9bSPaul Blakey case NFPROTO_IPV4: 2129c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_ipv4(tuple, target, 2139c26ba9bSPaul Blakey action); 2149c26ba9bSPaul Blakey break; 2159c26ba9bSPaul Blakey case NFPROTO_IPV6: 2169c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_ipv6(tuple, target, 2179c26ba9bSPaul Blakey action); 2189c26ba9bSPaul Blakey break; 2199c26ba9bSPaul Blakey default: 2209c26ba9bSPaul Blakey return -EOPNOTSUPP; 2219c26ba9bSPaul Blakey } 2229c26ba9bSPaul Blakey 2239c26ba9bSPaul Blakey switch (nf_ct_protonum(ct)) { 2249c26ba9bSPaul Blakey case IPPROTO_TCP: 2259c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_tcp(tuple, target, action); 2269c26ba9bSPaul Blakey break; 2279c26ba9bSPaul Blakey case IPPROTO_UDP: 2289c26ba9bSPaul Blakey tcf_ct_flow_table_add_action_nat_udp(tuple, target, action); 2299c26ba9bSPaul Blakey break; 2309c26ba9bSPaul Blakey default: 2319c26ba9bSPaul Blakey return -EOPNOTSUPP; 2329c26ba9bSPaul Blakey } 2339c26ba9bSPaul Blakey 2349c26ba9bSPaul Blakey return 0; 2359c26ba9bSPaul Blakey } 2369c26ba9bSPaul Blakey 2379c26ba9bSPaul Blakey static int tcf_ct_flow_table_fill_actions(struct net *net, 2381a441a9bSVlad Buslov struct flow_offload *flow, 2399c26ba9bSPaul Blakey enum flow_offload_tuple_dir tdir, 2409c26ba9bSPaul Blakey struct nf_flow_rule *flow_rule) 2419c26ba9bSPaul Blakey { 2429c26ba9bSPaul Blakey struct flow_action *action = &flow_rule->rule->action; 2439c26ba9bSPaul Blakey int num_entries = action->num_entries; 2449c26ba9bSPaul Blakey struct nf_conn *ct = flow->ct; 2451a441a9bSVlad Buslov enum ip_conntrack_info ctinfo; 2469c26ba9bSPaul Blakey enum ip_conntrack_dir dir; 2479c26ba9bSPaul Blakey int i, err; 2489c26ba9bSPaul Blakey 2499c26ba9bSPaul Blakey switch (tdir) { 2509c26ba9bSPaul Blakey case FLOW_OFFLOAD_DIR_ORIGINAL: 2519c26ba9bSPaul Blakey dir = IP_CT_DIR_ORIGINAL; 252d5774cb6SVlad Buslov ctinfo = test_bit(IPS_SEEN_REPLY_BIT, &ct->status) ? 253d5774cb6SVlad Buslov IP_CT_ESTABLISHED : IP_CT_NEW; 254d5774cb6SVlad Buslov if (ctinfo == IP_CT_ESTABLISHED) 2551a441a9bSVlad Buslov set_bit(NF_FLOW_HW_ESTABLISHED, &flow->flags); 2569c26ba9bSPaul Blakey break; 2579c26ba9bSPaul Blakey case FLOW_OFFLOAD_DIR_REPLY: 2589c26ba9bSPaul Blakey dir = IP_CT_DIR_REPLY; 2591a441a9bSVlad Buslov ctinfo = IP_CT_ESTABLISHED_REPLY; 2609c26ba9bSPaul Blakey break; 2619c26ba9bSPaul Blakey default: 2629c26ba9bSPaul Blakey return -EOPNOTSUPP; 2639c26ba9bSPaul Blakey } 2649c26ba9bSPaul Blakey 2659c26ba9bSPaul Blakey err = tcf_ct_flow_table_add_action_nat(net, ct, dir, action); 2669c26ba9bSPaul Blakey if (err) 2679c26ba9bSPaul Blakey goto err_nat; 2689c26ba9bSPaul Blakey 2691a441a9bSVlad Buslov tcf_ct_flow_table_add_action_meta(ct, dir, ctinfo, action); 2709c26ba9bSPaul Blakey return 0; 2719c26ba9bSPaul Blakey 2729c26ba9bSPaul Blakey err_nat: 2739c26ba9bSPaul Blakey /* Clear filled actions */ 2749c26ba9bSPaul Blakey for (i = num_entries; i < action->num_entries; i++) 2759c26ba9bSPaul Blakey memset(&action->entries[i], 0, sizeof(action->entries[i])); 2769c26ba9bSPaul Blakey action->num_entries = num_entries; 2779c26ba9bSPaul Blakey 2789c26ba9bSPaul Blakey return err; 2799c26ba9bSPaul Blakey } 2809c26ba9bSPaul Blakey 281735795f6SPablo Neira Ayuso static bool tcf_ct_flow_is_outdated(const struct flow_offload *flow) 282735795f6SPablo Neira Ayuso { 283735795f6SPablo Neira Ayuso return test_bit(IPS_SEEN_REPLY_BIT, &flow->ct->status) && 284a63b6622SVlad Buslov test_bit(IPS_HW_OFFLOAD_BIT, &flow->ct->status) && 285a63b6622SVlad Buslov !test_bit(NF_FLOW_HW_PENDING, &flow->flags) && 286735795f6SPablo Neira Ayuso !test_bit(NF_FLOW_HW_ESTABLISHED, &flow->flags); 287735795f6SPablo Neira Ayuso } 288735795f6SPablo Neira Ayuso 289*15f300edSVlad Buslov static void tcf_ct_flow_table_get_ref(struct tcf_ct_flow_table *ct_ft); 290*15f300edSVlad Buslov 291*15f300edSVlad Buslov static void tcf_ct_nf_get(struct nf_flowtable *ft) 292*15f300edSVlad Buslov { 293*15f300edSVlad Buslov struct tcf_ct_flow_table *ct_ft = 294*15f300edSVlad Buslov container_of(ft, struct tcf_ct_flow_table, nf_ft); 295*15f300edSVlad Buslov 296*15f300edSVlad Buslov tcf_ct_flow_table_get_ref(ct_ft); 297*15f300edSVlad Buslov } 298*15f300edSVlad Buslov 299*15f300edSVlad Buslov static void tcf_ct_flow_table_put(struct tcf_ct_flow_table *ct_ft); 300*15f300edSVlad Buslov 301*15f300edSVlad Buslov static void tcf_ct_nf_put(struct nf_flowtable *ft) 302*15f300edSVlad Buslov { 303*15f300edSVlad Buslov struct tcf_ct_flow_table *ct_ft = 304*15f300edSVlad Buslov container_of(ft, struct tcf_ct_flow_table, nf_ft); 305*15f300edSVlad Buslov 306*15f300edSVlad Buslov tcf_ct_flow_table_put(ct_ft); 307*15f300edSVlad Buslov } 308*15f300edSVlad Buslov 309c34b961aSPaul Blakey static struct nf_flowtable_type flowtable_ct = { 310735795f6SPablo Neira Ayuso .gc = tcf_ct_flow_is_outdated, 3119c26ba9bSPaul Blakey .action = tcf_ct_flow_table_fill_actions, 312*15f300edSVlad Buslov .get = tcf_ct_nf_get, 313*15f300edSVlad Buslov .put = tcf_ct_nf_put, 314c34b961aSPaul Blakey .owner = THIS_MODULE, 315c34b961aSPaul Blakey }; 316c34b961aSPaul Blakey 317fc54d906SVlad Buslov static int tcf_ct_flow_table_get(struct net *net, struct tcf_ct_params *params) 318c34b961aSPaul Blakey { 319c34b961aSPaul Blakey struct tcf_ct_flow_table *ct_ft; 320c34b961aSPaul Blakey int err = -ENOMEM; 321c34b961aSPaul Blakey 322138470a9SEric Dumazet mutex_lock(&zones_mutex); 323c34b961aSPaul Blakey ct_ft = rhashtable_lookup_fast(&zones_ht, ¶ms->zone, zones_params); 324138470a9SEric Dumazet if (ct_ft && refcount_inc_not_zero(&ct_ft->ref)) 325138470a9SEric Dumazet goto out_unlock; 326c34b961aSPaul Blakey 327138470a9SEric Dumazet ct_ft = kzalloc(sizeof(*ct_ft), GFP_KERNEL); 328c34b961aSPaul Blakey if (!ct_ft) 329c34b961aSPaul Blakey goto err_alloc; 330138470a9SEric Dumazet refcount_set(&ct_ft->ref, 1); 331c34b961aSPaul Blakey 332c34b961aSPaul Blakey ct_ft->zone = params->zone; 333c34b961aSPaul Blakey err = rhashtable_insert_fast(&zones_ht, &ct_ft->node, zones_params); 334c34b961aSPaul Blakey if (err) 335c34b961aSPaul Blakey goto err_insert; 336c34b961aSPaul Blakey 337c34b961aSPaul Blakey ct_ft->nf_ft.type = &flowtable_ct; 3383567e233SMarcelo Ricardo Leitner ct_ft->nf_ft.flags |= NF_FLOWTABLE_HW_OFFLOAD | 3393567e233SMarcelo Ricardo Leitner NF_FLOWTABLE_COUNTER; 340c34b961aSPaul Blakey err = nf_flow_table_init(&ct_ft->nf_ft); 341c34b961aSPaul Blakey if (err) 342c34b961aSPaul Blakey goto err_init; 343fc54d906SVlad Buslov write_pnet(&ct_ft->nf_ft.net, net); 344c34b961aSPaul Blakey 345c34b961aSPaul Blakey __module_get(THIS_MODULE); 346138470a9SEric Dumazet out_unlock: 347c34b961aSPaul Blakey params->ct_ft = ct_ft; 348edd5861eSPaul Blakey params->nf_ft = &ct_ft->nf_ft; 349138470a9SEric Dumazet mutex_unlock(&zones_mutex); 350c34b961aSPaul Blakey 351c34b961aSPaul Blakey return 0; 352c34b961aSPaul Blakey 353c34b961aSPaul Blakey err_init: 354c34b961aSPaul Blakey rhashtable_remove_fast(&zones_ht, &ct_ft->node, zones_params); 355c34b961aSPaul Blakey err_insert: 356c34b961aSPaul Blakey kfree(ct_ft); 357c34b961aSPaul Blakey err_alloc: 358138470a9SEric Dumazet mutex_unlock(&zones_mutex); 359c34b961aSPaul Blakey return err; 360c34b961aSPaul Blakey } 361c34b961aSPaul Blakey 362*15f300edSVlad Buslov static void tcf_ct_flow_table_get_ref(struct tcf_ct_flow_table *ct_ft) 363*15f300edSVlad Buslov { 364*15f300edSVlad Buslov refcount_inc(&ct_ft->ref); 365*15f300edSVlad Buslov } 366*15f300edSVlad Buslov 367c34b961aSPaul Blakey static void tcf_ct_flow_table_cleanup_work(struct work_struct *work) 368c34b961aSPaul Blakey { 369c34b961aSPaul Blakey struct tcf_ct_flow_table *ct_ft; 37077ac5e40SLouis Peens struct flow_block *block; 371c34b961aSPaul Blakey 372c34b961aSPaul Blakey ct_ft = container_of(to_rcu_work(work), struct tcf_ct_flow_table, 373c34b961aSPaul Blakey rwork); 374c34b961aSPaul Blakey nf_flow_table_free(&ct_ft->nf_ft); 37577ac5e40SLouis Peens 37677ac5e40SLouis Peens block = &ct_ft->nf_ft.flow_block; 37777ac5e40SLouis Peens down_write(&ct_ft->nf_ft.flow_block_lock); 378*15f300edSVlad Buslov WARN_ON(!list_empty(&block->cb_list)); 37977ac5e40SLouis Peens up_write(&ct_ft->nf_ft.flow_block_lock); 380c34b961aSPaul Blakey kfree(ct_ft); 381c34b961aSPaul Blakey 382c34b961aSPaul Blakey module_put(THIS_MODULE); 383c34b961aSPaul Blakey } 384c34b961aSPaul Blakey 38519138941SXin Long static void tcf_ct_flow_table_put(struct tcf_ct_flow_table *ct_ft) 386c34b961aSPaul Blakey { 38719138941SXin Long if (refcount_dec_and_test(&ct_ft->ref)) { 388c34b961aSPaul Blakey rhashtable_remove_fast(&zones_ht, &ct_ft->node, zones_params); 389c34b961aSPaul Blakey INIT_RCU_WORK(&ct_ft->rwork, tcf_ct_flow_table_cleanup_work); 390c34b961aSPaul Blakey queue_rcu_work(act_ct_wq, &ct_ft->rwork); 391c34b961aSPaul Blakey } 392c34b961aSPaul Blakey } 393c34b961aSPaul Blakey 394db6140e5SPaul Blakey static void tcf_ct_flow_tc_ifidx(struct flow_offload *entry, 395db6140e5SPaul Blakey struct nf_conn_act_ct_ext *act_ct_ext, u8 dir) 396db6140e5SPaul Blakey { 397db6140e5SPaul Blakey entry->tuplehash[dir].tuple.xmit_type = FLOW_OFFLOAD_XMIT_TC; 398db6140e5SPaul Blakey entry->tuplehash[dir].tuple.tc.iifidx = act_ct_ext->ifindex[dir]; 399db6140e5SPaul Blakey } 400db6140e5SPaul Blakey 401d5a116dbSVlad Buslov static void tcf_ct_flow_ct_ext_ifidx_update(struct flow_offload *entry) 402d5a116dbSVlad Buslov { 403d5a116dbSVlad Buslov struct nf_conn_act_ct_ext *act_ct_ext; 404d5a116dbSVlad Buslov 405d5a116dbSVlad Buslov act_ct_ext = nf_conn_act_ct_ext_find(entry->ct); 406d5a116dbSVlad Buslov if (act_ct_ext) { 407d5a116dbSVlad Buslov tcf_ct_flow_tc_ifidx(entry, act_ct_ext, FLOW_OFFLOAD_DIR_ORIGINAL); 408d5a116dbSVlad Buslov tcf_ct_flow_tc_ifidx(entry, act_ct_ext, FLOW_OFFLOAD_DIR_REPLY); 409d5a116dbSVlad Buslov } 410d5a116dbSVlad Buslov } 411d5a116dbSVlad Buslov 41264ff70b8SPaul Blakey static void tcf_ct_flow_table_add(struct tcf_ct_flow_table *ct_ft, 41364ff70b8SPaul Blakey struct nf_conn *ct, 4146a9bad00SVlad Buslov bool tcp, bool bidirectional) 41564ff70b8SPaul Blakey { 4169795ded7SPaul Blakey struct nf_conn_act_ct_ext *act_ct_ext; 41764ff70b8SPaul Blakey struct flow_offload *entry; 41864ff70b8SPaul Blakey int err; 41964ff70b8SPaul Blakey 42064ff70b8SPaul Blakey if (test_and_set_bit(IPS_OFFLOAD_BIT, &ct->status)) 42164ff70b8SPaul Blakey return; 42264ff70b8SPaul Blakey 42364ff70b8SPaul Blakey entry = flow_offload_alloc(ct); 42464ff70b8SPaul Blakey if (!entry) { 42564ff70b8SPaul Blakey WARN_ON_ONCE(1); 42664ff70b8SPaul Blakey goto err_alloc; 42764ff70b8SPaul Blakey } 42864ff70b8SPaul Blakey 42964ff70b8SPaul Blakey if (tcp) { 43064ff70b8SPaul Blakey ct->proto.tcp.seen[0].flags |= IP_CT_TCP_FLAG_BE_LIBERAL; 43164ff70b8SPaul Blakey ct->proto.tcp.seen[1].flags |= IP_CT_TCP_FLAG_BE_LIBERAL; 43264ff70b8SPaul Blakey } 4336a9bad00SVlad Buslov if (bidirectional) 4346a9bad00SVlad Buslov __set_bit(NF_FLOW_HW_BIDIRECTIONAL, &entry->flags); 43564ff70b8SPaul Blakey 4369795ded7SPaul Blakey act_ct_ext = nf_conn_act_ct_ext_find(ct); 4379795ded7SPaul Blakey if (act_ct_ext) { 438db6140e5SPaul Blakey tcf_ct_flow_tc_ifidx(entry, act_ct_ext, FLOW_OFFLOAD_DIR_ORIGINAL); 439db6140e5SPaul Blakey tcf_ct_flow_tc_ifidx(entry, act_ct_ext, FLOW_OFFLOAD_DIR_REPLY); 4409795ded7SPaul Blakey } 4419795ded7SPaul Blakey 44264ff70b8SPaul Blakey err = flow_offload_add(&ct_ft->nf_ft, entry); 44364ff70b8SPaul Blakey if (err) 44464ff70b8SPaul Blakey goto err_add; 44564ff70b8SPaul Blakey 44664ff70b8SPaul Blakey return; 44764ff70b8SPaul Blakey 44864ff70b8SPaul Blakey err_add: 44964ff70b8SPaul Blakey flow_offload_free(entry); 45064ff70b8SPaul Blakey err_alloc: 45164ff70b8SPaul Blakey clear_bit(IPS_OFFLOAD_BIT, &ct->status); 45264ff70b8SPaul Blakey } 45364ff70b8SPaul Blakey 45464ff70b8SPaul Blakey static void tcf_ct_flow_table_process_conn(struct tcf_ct_flow_table *ct_ft, 45564ff70b8SPaul Blakey struct nf_conn *ct, 45664ff70b8SPaul Blakey enum ip_conntrack_info ctinfo) 45764ff70b8SPaul Blakey { 4586a9bad00SVlad Buslov bool tcp = false, bidirectional = true; 45964ff70b8SPaul Blakey 46064ff70b8SPaul Blakey switch (nf_ct_protonum(ct)) { 46164ff70b8SPaul Blakey case IPPROTO_TCP: 4626a9bad00SVlad Buslov if ((ctinfo != IP_CT_ESTABLISHED && 4636a9bad00SVlad Buslov ctinfo != IP_CT_ESTABLISHED_REPLY) || 4646a9bad00SVlad Buslov !test_bit(IPS_ASSURED_BIT, &ct->status) || 4656a9bad00SVlad Buslov ct->proto.tcp.state != TCP_CONNTRACK_ESTABLISHED) 46664ff70b8SPaul Blakey return; 4676a9bad00SVlad Buslov 4686a9bad00SVlad Buslov tcp = true; 46964ff70b8SPaul Blakey break; 47064ff70b8SPaul Blakey case IPPROTO_UDP: 4716a9bad00SVlad Buslov if (!nf_ct_is_confirmed(ct)) 4726a9bad00SVlad Buslov return; 4736a9bad00SVlad Buslov if (!test_bit(IPS_ASSURED_BIT, &ct->status)) 4746a9bad00SVlad Buslov bidirectional = false; 47564ff70b8SPaul Blakey break; 476fcb6aa86SToshiaki Makita #ifdef CONFIG_NF_CT_PROTO_GRE 477fcb6aa86SToshiaki Makita case IPPROTO_GRE: { 478fcb6aa86SToshiaki Makita struct nf_conntrack_tuple *tuple; 479fcb6aa86SToshiaki Makita 4806a9bad00SVlad Buslov if ((ctinfo != IP_CT_ESTABLISHED && 4816a9bad00SVlad Buslov ctinfo != IP_CT_ESTABLISHED_REPLY) || 4826a9bad00SVlad Buslov !test_bit(IPS_ASSURED_BIT, &ct->status) || 4836a9bad00SVlad Buslov ct->status & IPS_NAT_MASK) 484fcb6aa86SToshiaki Makita return; 4856a9bad00SVlad Buslov 486fcb6aa86SToshiaki Makita tuple = &ct->tuplehash[IP_CT_DIR_ORIGINAL].tuple; 487fcb6aa86SToshiaki Makita /* No support for GRE v1 */ 488fcb6aa86SToshiaki Makita if (tuple->src.u.gre.key || tuple->dst.u.gre.key) 489fcb6aa86SToshiaki Makita return; 490fcb6aa86SToshiaki Makita break; 491fcb6aa86SToshiaki Makita } 492fcb6aa86SToshiaki Makita #endif 49364ff70b8SPaul Blakey default: 49464ff70b8SPaul Blakey return; 49564ff70b8SPaul Blakey } 49664ff70b8SPaul Blakey 49764ff70b8SPaul Blakey if (nf_ct_ext_exist(ct, NF_CT_EXT_HELPER) || 49864ff70b8SPaul Blakey ct->status & IPS_SEQ_ADJUST) 49964ff70b8SPaul Blakey return; 50064ff70b8SPaul Blakey 5016a9bad00SVlad Buslov tcf_ct_flow_table_add(ct_ft, ct, tcp, bidirectional); 50264ff70b8SPaul Blakey } 50364ff70b8SPaul Blakey 50446475bb2SPaul Blakey static bool 50546475bb2SPaul Blakey tcf_ct_flow_table_fill_tuple_ipv4(struct sk_buff *skb, 50607ac9d16SPaul Blakey struct flow_offload_tuple *tuple, 50707ac9d16SPaul Blakey struct tcphdr **tcph) 50846475bb2SPaul Blakey { 50946475bb2SPaul Blakey struct flow_ports *ports; 51046475bb2SPaul Blakey unsigned int thoff; 51146475bb2SPaul Blakey struct iphdr *iph; 512fcb6aa86SToshiaki Makita size_t hdrsize; 513fcb6aa86SToshiaki Makita u8 ipproto; 51446475bb2SPaul Blakey 5154cc5fdecSPaul Blakey if (!pskb_network_may_pull(skb, sizeof(*iph))) 51646475bb2SPaul Blakey return false; 51746475bb2SPaul Blakey 51846475bb2SPaul Blakey iph = ip_hdr(skb); 51946475bb2SPaul Blakey thoff = iph->ihl * 4; 52046475bb2SPaul Blakey 52146475bb2SPaul Blakey if (ip_is_fragment(iph) || 52246475bb2SPaul Blakey unlikely(thoff != sizeof(struct iphdr))) 52346475bb2SPaul Blakey return false; 52446475bb2SPaul Blakey 525fcb6aa86SToshiaki Makita ipproto = iph->protocol; 526fcb6aa86SToshiaki Makita switch (ipproto) { 527fcb6aa86SToshiaki Makita case IPPROTO_TCP: 528fcb6aa86SToshiaki Makita hdrsize = sizeof(struct tcphdr); 529fcb6aa86SToshiaki Makita break; 530fcb6aa86SToshiaki Makita case IPPROTO_UDP: 531fcb6aa86SToshiaki Makita hdrsize = sizeof(*ports); 532fcb6aa86SToshiaki Makita break; 533fcb6aa86SToshiaki Makita #ifdef CONFIG_NF_CT_PROTO_GRE 534fcb6aa86SToshiaki Makita case IPPROTO_GRE: 535fcb6aa86SToshiaki Makita hdrsize = sizeof(struct gre_base_hdr); 536fcb6aa86SToshiaki Makita break; 537fcb6aa86SToshiaki Makita #endif 538fcb6aa86SToshiaki Makita default: 53946475bb2SPaul Blakey return false; 540fcb6aa86SToshiaki Makita } 54146475bb2SPaul Blakey 54246475bb2SPaul Blakey if (iph->ttl <= 1) 54346475bb2SPaul Blakey return false; 54446475bb2SPaul Blakey 545fcb6aa86SToshiaki Makita if (!pskb_network_may_pull(skb, thoff + hdrsize)) 54646475bb2SPaul Blakey return false; 54746475bb2SPaul Blakey 548fcb6aa86SToshiaki Makita switch (ipproto) { 549fcb6aa86SToshiaki Makita case IPPROTO_TCP: 55007ac9d16SPaul Blakey *tcph = (void *)(skb_network_header(skb) + thoff); 551fcb6aa86SToshiaki Makita fallthrough; 552fcb6aa86SToshiaki Makita case IPPROTO_UDP: 55307ac9d16SPaul Blakey ports = (struct flow_ports *)(skb_network_header(skb) + thoff); 55446475bb2SPaul Blakey tuple->src_port = ports->source; 55546475bb2SPaul Blakey tuple->dst_port = ports->dest; 556fcb6aa86SToshiaki Makita break; 557fcb6aa86SToshiaki Makita case IPPROTO_GRE: { 558fcb6aa86SToshiaki Makita struct gre_base_hdr *greh; 559fcb6aa86SToshiaki Makita 560fcb6aa86SToshiaki Makita greh = (struct gre_base_hdr *)(skb_network_header(skb) + thoff); 561fcb6aa86SToshiaki Makita if ((greh->flags & GRE_VERSION) != GRE_VERSION_0) 562fcb6aa86SToshiaki Makita return false; 563fcb6aa86SToshiaki Makita break; 564fcb6aa86SToshiaki Makita } 565fcb6aa86SToshiaki Makita } 566fcb6aa86SToshiaki Makita 567fcb6aa86SToshiaki Makita iph = ip_hdr(skb); 568fcb6aa86SToshiaki Makita 569fcb6aa86SToshiaki Makita tuple->src_v4.s_addr = iph->saddr; 570fcb6aa86SToshiaki Makita tuple->dst_v4.s_addr = iph->daddr; 57146475bb2SPaul Blakey tuple->l3proto = AF_INET; 572fcb6aa86SToshiaki Makita tuple->l4proto = ipproto; 57346475bb2SPaul Blakey 57446475bb2SPaul Blakey return true; 57546475bb2SPaul Blakey } 57646475bb2SPaul Blakey 57746475bb2SPaul Blakey static bool 57846475bb2SPaul Blakey tcf_ct_flow_table_fill_tuple_ipv6(struct sk_buff *skb, 57907ac9d16SPaul Blakey struct flow_offload_tuple *tuple, 58007ac9d16SPaul Blakey struct tcphdr **tcph) 58146475bb2SPaul Blakey { 58246475bb2SPaul Blakey struct flow_ports *ports; 58346475bb2SPaul Blakey struct ipv6hdr *ip6h; 58446475bb2SPaul Blakey unsigned int thoff; 585fcb6aa86SToshiaki Makita size_t hdrsize; 586fcb6aa86SToshiaki Makita u8 nexthdr; 58746475bb2SPaul Blakey 5884cc5fdecSPaul Blakey if (!pskb_network_may_pull(skb, sizeof(*ip6h))) 58946475bb2SPaul Blakey return false; 59046475bb2SPaul Blakey 59146475bb2SPaul Blakey ip6h = ipv6_hdr(skb); 592fcb6aa86SToshiaki Makita thoff = sizeof(*ip6h); 59346475bb2SPaul Blakey 594fcb6aa86SToshiaki Makita nexthdr = ip6h->nexthdr; 595fcb6aa86SToshiaki Makita switch (nexthdr) { 596fcb6aa86SToshiaki Makita case IPPROTO_TCP: 597fcb6aa86SToshiaki Makita hdrsize = sizeof(struct tcphdr); 598fcb6aa86SToshiaki Makita break; 599fcb6aa86SToshiaki Makita case IPPROTO_UDP: 600fcb6aa86SToshiaki Makita hdrsize = sizeof(*ports); 601fcb6aa86SToshiaki Makita break; 602fcb6aa86SToshiaki Makita #ifdef CONFIG_NF_CT_PROTO_GRE 603fcb6aa86SToshiaki Makita case IPPROTO_GRE: 604fcb6aa86SToshiaki Makita hdrsize = sizeof(struct gre_base_hdr); 605fcb6aa86SToshiaki Makita break; 606fcb6aa86SToshiaki Makita #endif 607fcb6aa86SToshiaki Makita default: 60886360030SDan Carpenter return false; 609fcb6aa86SToshiaki Makita } 61046475bb2SPaul Blakey 61146475bb2SPaul Blakey if (ip6h->hop_limit <= 1) 61246475bb2SPaul Blakey return false; 61346475bb2SPaul Blakey 614fcb6aa86SToshiaki Makita if (!pskb_network_may_pull(skb, thoff + hdrsize)) 61546475bb2SPaul Blakey return false; 61646475bb2SPaul Blakey 617fcb6aa86SToshiaki Makita switch (nexthdr) { 618fcb6aa86SToshiaki Makita case IPPROTO_TCP: 61907ac9d16SPaul Blakey *tcph = (void *)(skb_network_header(skb) + thoff); 620fcb6aa86SToshiaki Makita fallthrough; 621fcb6aa86SToshiaki Makita case IPPROTO_UDP: 62207ac9d16SPaul Blakey ports = (struct flow_ports *)(skb_network_header(skb) + thoff); 62346475bb2SPaul Blakey tuple->src_port = ports->source; 62446475bb2SPaul Blakey tuple->dst_port = ports->dest; 625fcb6aa86SToshiaki Makita break; 626fcb6aa86SToshiaki Makita case IPPROTO_GRE: { 627fcb6aa86SToshiaki Makita struct gre_base_hdr *greh; 628fcb6aa86SToshiaki Makita 629fcb6aa86SToshiaki Makita greh = (struct gre_base_hdr *)(skb_network_header(skb) + thoff); 630fcb6aa86SToshiaki Makita if ((greh->flags & GRE_VERSION) != GRE_VERSION_0) 631fcb6aa86SToshiaki Makita return false; 632fcb6aa86SToshiaki Makita break; 633fcb6aa86SToshiaki Makita } 634fcb6aa86SToshiaki Makita } 635fcb6aa86SToshiaki Makita 636fcb6aa86SToshiaki Makita ip6h = ipv6_hdr(skb); 637fcb6aa86SToshiaki Makita 638fcb6aa86SToshiaki Makita tuple->src_v6 = ip6h->saddr; 639fcb6aa86SToshiaki Makita tuple->dst_v6 = ip6h->daddr; 64046475bb2SPaul Blakey tuple->l3proto = AF_INET6; 641fcb6aa86SToshiaki Makita tuple->l4proto = nexthdr; 64246475bb2SPaul Blakey 64346475bb2SPaul Blakey return true; 64446475bb2SPaul Blakey } 64546475bb2SPaul Blakey 64646475bb2SPaul Blakey static bool tcf_ct_flow_table_lookup(struct tcf_ct_params *p, 64746475bb2SPaul Blakey struct sk_buff *skb, 64846475bb2SPaul Blakey u8 family) 64946475bb2SPaul Blakey { 65046475bb2SPaul Blakey struct nf_flowtable *nf_ft = &p->ct_ft->nf_ft; 65146475bb2SPaul Blakey struct flow_offload_tuple_rhash *tuplehash; 65246475bb2SPaul Blakey struct flow_offload_tuple tuple = {}; 65346475bb2SPaul Blakey enum ip_conntrack_info ctinfo; 65407ac9d16SPaul Blakey struct tcphdr *tcph = NULL; 65541f2c7c3SPaul Blakey bool force_refresh = false; 65646475bb2SPaul Blakey struct flow_offload *flow; 65746475bb2SPaul Blakey struct nf_conn *ct; 65846475bb2SPaul Blakey u8 dir; 65946475bb2SPaul Blakey 66046475bb2SPaul Blakey switch (family) { 66146475bb2SPaul Blakey case NFPROTO_IPV4: 66207ac9d16SPaul Blakey if (!tcf_ct_flow_table_fill_tuple_ipv4(skb, &tuple, &tcph)) 66346475bb2SPaul Blakey return false; 66446475bb2SPaul Blakey break; 66546475bb2SPaul Blakey case NFPROTO_IPV6: 66607ac9d16SPaul Blakey if (!tcf_ct_flow_table_fill_tuple_ipv6(skb, &tuple, &tcph)) 66746475bb2SPaul Blakey return false; 66846475bb2SPaul Blakey break; 66946475bb2SPaul Blakey default: 67046475bb2SPaul Blakey return false; 67146475bb2SPaul Blakey } 67246475bb2SPaul Blakey 67346475bb2SPaul Blakey tuplehash = flow_offload_lookup(nf_ft, &tuple); 67446475bb2SPaul Blakey if (!tuplehash) 67546475bb2SPaul Blakey return false; 67646475bb2SPaul Blakey 67746475bb2SPaul Blakey dir = tuplehash->tuple.dir; 67846475bb2SPaul Blakey flow = container_of(tuplehash, struct flow_offload, tuplehash[dir]); 67946475bb2SPaul Blakey ct = flow->ct; 68046475bb2SPaul Blakey 6816a9bad00SVlad Buslov if (dir == FLOW_OFFLOAD_DIR_REPLY && 6826a9bad00SVlad Buslov !test_bit(NF_FLOW_HW_BIDIRECTIONAL, &flow->flags)) { 6836a9bad00SVlad Buslov /* Only offload reply direction after connection became 6846a9bad00SVlad Buslov * assured. 6856a9bad00SVlad Buslov */ 6866a9bad00SVlad Buslov if (test_bit(IPS_ASSURED_BIT, &ct->status)) 6876a9bad00SVlad Buslov set_bit(NF_FLOW_HW_BIDIRECTIONAL, &flow->flags); 6886a9bad00SVlad Buslov else if (test_bit(NF_FLOW_HW_ESTABLISHED, &flow->flags)) 6896a9bad00SVlad Buslov /* If flow_table flow has already been updated to the 6906a9bad00SVlad Buslov * established state, then don't refresh. 6916a9bad00SVlad Buslov */ 6926a9bad00SVlad Buslov return false; 69341f2c7c3SPaul Blakey force_refresh = true; 6946a9bad00SVlad Buslov } 6956a9bad00SVlad Buslov 69607ac9d16SPaul Blakey if (tcph && (unlikely(tcph->fin || tcph->rst))) { 69707ac9d16SPaul Blakey flow_offload_teardown(flow); 69807ac9d16SPaul Blakey return false; 69907ac9d16SPaul Blakey } 70007ac9d16SPaul Blakey 7016a9bad00SVlad Buslov if (dir == FLOW_OFFLOAD_DIR_ORIGINAL) 7026a9bad00SVlad Buslov ctinfo = test_bit(IPS_SEEN_REPLY_BIT, &ct->status) ? 7036a9bad00SVlad Buslov IP_CT_ESTABLISHED : IP_CT_NEW; 7046a9bad00SVlad Buslov else 7056a9bad00SVlad Buslov ctinfo = IP_CT_ESTABLISHED_REPLY; 70646475bb2SPaul Blakey 707d5a116dbSVlad Buslov nf_conn_act_ct_ext_fill(skb, ct, ctinfo); 708d5a116dbSVlad Buslov tcf_ct_flow_ct_ext_ifidx_update(flow); 70941f2c7c3SPaul Blakey flow_offload_refresh(nf_ft, flow, force_refresh); 71041f2c7c3SPaul Blakey if (!test_bit(IPS_ASSURED_BIT, &ct->status)) { 71141f2c7c3SPaul Blakey /* Process this flow in SW to allow promoting to ASSURED */ 71241f2c7c3SPaul Blakey return false; 71341f2c7c3SPaul Blakey } 71441f2c7c3SPaul Blakey 71546475bb2SPaul Blakey nf_conntrack_get(&ct->ct_general); 71646475bb2SPaul Blakey nf_ct_set(skb, ct, ctinfo); 7173567e233SMarcelo Ricardo Leitner if (nf_ft->flags & NF_FLOWTABLE_COUNTER) 718beb97d3aSwenxu nf_ct_acct_update(ct, dir, skb->len); 71946475bb2SPaul Blakey 72046475bb2SPaul Blakey return true; 72146475bb2SPaul Blakey } 72246475bb2SPaul Blakey 723c34b961aSPaul Blakey static int tcf_ct_flow_tables_init(void) 724c34b961aSPaul Blakey { 725c34b961aSPaul Blakey return rhashtable_init(&zones_ht, &zones_params); 726c34b961aSPaul Blakey } 727c34b961aSPaul Blakey 728c34b961aSPaul Blakey static void tcf_ct_flow_tables_uninit(void) 729c34b961aSPaul Blakey { 730c34b961aSPaul Blakey rhashtable_destroy(&zones_ht); 731c34b961aSPaul Blakey } 732c34b961aSPaul Blakey 733b57dc7c1SPaul Blakey static struct tc_action_ops act_ct_ops; 734b57dc7c1SPaul Blakey 735b57dc7c1SPaul Blakey struct tc_ct_action_net { 736b57dc7c1SPaul Blakey struct tc_action_net tn; /* Must be first */ 737b57dc7c1SPaul Blakey bool labels; 738b57dc7c1SPaul Blakey }; 739b57dc7c1SPaul Blakey 740b57dc7c1SPaul Blakey /* Determine whether skb->_nfct is equal to the result of conntrack lookup. */ 741b57dc7c1SPaul Blakey static bool tcf_ct_skb_nfct_cached(struct net *net, struct sk_buff *skb, 742a21b06e7SXin Long struct tcf_ct_params *p) 743b57dc7c1SPaul Blakey { 744b57dc7c1SPaul Blakey enum ip_conntrack_info ctinfo; 745b57dc7c1SPaul Blakey struct nf_conn *ct; 746b57dc7c1SPaul Blakey 747b57dc7c1SPaul Blakey ct = nf_ct_get(skb, &ctinfo); 748b57dc7c1SPaul Blakey if (!ct) 749b57dc7c1SPaul Blakey return false; 750b57dc7c1SPaul Blakey if (!net_eq(net, read_pnet(&ct->ct_net))) 751bcb74e13SMarcelo Ricardo Leitner goto drop_ct; 752a21b06e7SXin Long if (nf_ct_zone(ct)->id != p->zone) 753bcb74e13SMarcelo Ricardo Leitner goto drop_ct; 754a21b06e7SXin Long if (p->helper) { 755a21b06e7SXin Long struct nf_conn_help *help; 756a21b06e7SXin Long 757a21b06e7SXin Long help = nf_ct_ext_find(ct, NF_CT_EXT_HELPER); 758a21b06e7SXin Long if (help && rcu_access_pointer(help->helper) != p->helper) 759a21b06e7SXin Long goto drop_ct; 760a21b06e7SXin Long } 761b57dc7c1SPaul Blakey 762b57dc7c1SPaul Blakey /* Force conntrack entry direction. */ 763a21b06e7SXin Long if ((p->ct_action & TCA_CT_ACT_FORCE) && 764a21b06e7SXin Long CTINFO2DIR(ctinfo) != IP_CT_DIR_ORIGINAL) { 765b57dc7c1SPaul Blakey if (nf_ct_is_confirmed(ct)) 766b57dc7c1SPaul Blakey nf_ct_kill(ct); 767b57dc7c1SPaul Blakey 768bcb74e13SMarcelo Ricardo Leitner goto drop_ct; 769bcb74e13SMarcelo Ricardo Leitner } 770bcb74e13SMarcelo Ricardo Leitner 771bcb74e13SMarcelo Ricardo Leitner return true; 772bcb74e13SMarcelo Ricardo Leitner 773bcb74e13SMarcelo Ricardo Leitner drop_ct: 774408bdcfcSFlorian Westphal nf_ct_put(ct); 775b57dc7c1SPaul Blakey nf_ct_set(skb, NULL, IP_CT_UNTRACKED); 776b57dc7c1SPaul Blakey 777b57dc7c1SPaul Blakey return false; 778b57dc7c1SPaul Blakey } 779b57dc7c1SPaul Blakey 780b57dc7c1SPaul Blakey static u8 tcf_ct_skb_nf_family(struct sk_buff *skb) 781b57dc7c1SPaul Blakey { 782b57dc7c1SPaul Blakey u8 family = NFPROTO_UNSPEC; 783b57dc7c1SPaul Blakey 784d7bf2ebeSToke Høiland-Jørgensen switch (skb_protocol(skb, true)) { 785b57dc7c1SPaul Blakey case htons(ETH_P_IP): 786b57dc7c1SPaul Blakey family = NFPROTO_IPV4; 787b57dc7c1SPaul Blakey break; 788b57dc7c1SPaul Blakey case htons(ETH_P_IPV6): 789b57dc7c1SPaul Blakey family = NFPROTO_IPV6; 790b57dc7c1SPaul Blakey break; 791b57dc7c1SPaul Blakey default: 792b57dc7c1SPaul Blakey break; 793b57dc7c1SPaul Blakey } 794b57dc7c1SPaul Blakey 795b57dc7c1SPaul Blakey return family; 796b57dc7c1SPaul Blakey } 797b57dc7c1SPaul Blakey 798b57dc7c1SPaul Blakey static int tcf_ct_ipv4_is_fragment(struct sk_buff *skb, bool *frag) 799b57dc7c1SPaul Blakey { 800b57dc7c1SPaul Blakey unsigned int len; 801b57dc7c1SPaul Blakey 802b57dc7c1SPaul Blakey len = skb_network_offset(skb) + sizeof(struct iphdr); 803b57dc7c1SPaul Blakey if (unlikely(skb->len < len)) 804b57dc7c1SPaul Blakey return -EINVAL; 805b57dc7c1SPaul Blakey if (unlikely(!pskb_may_pull(skb, len))) 806b57dc7c1SPaul Blakey return -ENOMEM; 807b57dc7c1SPaul Blakey 808b57dc7c1SPaul Blakey *frag = ip_is_fragment(ip_hdr(skb)); 809b57dc7c1SPaul Blakey return 0; 810b57dc7c1SPaul Blakey } 811b57dc7c1SPaul Blakey 812b57dc7c1SPaul Blakey static int tcf_ct_ipv6_is_fragment(struct sk_buff *skb, bool *frag) 813b57dc7c1SPaul Blakey { 814b57dc7c1SPaul Blakey unsigned int flags = 0, len, payload_ofs = 0; 815b57dc7c1SPaul Blakey unsigned short frag_off; 816b57dc7c1SPaul Blakey int nexthdr; 817b57dc7c1SPaul Blakey 818b57dc7c1SPaul Blakey len = skb_network_offset(skb) + sizeof(struct ipv6hdr); 819b57dc7c1SPaul Blakey if (unlikely(skb->len < len)) 820b57dc7c1SPaul Blakey return -EINVAL; 821b57dc7c1SPaul Blakey if (unlikely(!pskb_may_pull(skb, len))) 822b57dc7c1SPaul Blakey return -ENOMEM; 823b57dc7c1SPaul Blakey 824b57dc7c1SPaul Blakey nexthdr = ipv6_find_hdr(skb, &payload_ofs, -1, &frag_off, &flags); 825b57dc7c1SPaul Blakey if (unlikely(nexthdr < 0)) 826b57dc7c1SPaul Blakey return -EPROTO; 827b57dc7c1SPaul Blakey 828b57dc7c1SPaul Blakey *frag = flags & IP6_FH_F_FRAG; 829b57dc7c1SPaul Blakey return 0; 830b57dc7c1SPaul Blakey } 831b57dc7c1SPaul Blakey 832b57dc7c1SPaul Blakey static int tcf_ct_handle_fragments(struct net *net, struct sk_buff *skb, 833ae372cb1Swenxu u8 family, u16 zone, bool *defrag) 834b57dc7c1SPaul Blakey { 835b57dc7c1SPaul Blakey enum ip_conntrack_info ctinfo; 836b57dc7c1SPaul Blakey struct nf_conn *ct; 837b57dc7c1SPaul Blakey int err = 0; 838b57dc7c1SPaul Blakey bool frag; 8390785407eSXin Long u8 proto; 840ec624fe7SPaul Blakey u16 mru; 841b57dc7c1SPaul Blakey 842b57dc7c1SPaul Blakey /* Previously seen (loopback)? Ignore. */ 843b57dc7c1SPaul Blakey ct = nf_ct_get(skb, &ctinfo); 844b57dc7c1SPaul Blakey if ((ct && !nf_ct_is_template(ct)) || ctinfo == IP_CT_UNTRACKED) 845b57dc7c1SPaul Blakey return 0; 846b57dc7c1SPaul Blakey 847b57dc7c1SPaul Blakey if (family == NFPROTO_IPV4) 848b57dc7c1SPaul Blakey err = tcf_ct_ipv4_is_fragment(skb, &frag); 849b57dc7c1SPaul Blakey else 850b57dc7c1SPaul Blakey err = tcf_ct_ipv6_is_fragment(skb, &frag); 851b57dc7c1SPaul Blakey if (err || !frag) 852b57dc7c1SPaul Blakey return err; 853b57dc7c1SPaul Blakey 854b57dc7c1SPaul Blakey skb_get(skb); 8550785407eSXin Long err = nf_ct_handle_fragments(net, skb, zone, family, &proto, &mru); 856558d95e7SXin Long if (err) 857eda814b9SAlaa Hleihel return err; 858ae372cb1Swenxu 859ae372cb1Swenxu *defrag = true; 860ec624fe7SPaul Blakey tc_skb_cb(skb)->mru = mru; 861b57dc7c1SPaul Blakey 862558d95e7SXin Long return 0; 863b57dc7c1SPaul Blakey } 864b57dc7c1SPaul Blakey 86519138941SXin Long static void tcf_ct_params_free(struct tcf_ct_params *params) 866b57dc7c1SPaul Blakey { 867a21b06e7SXin Long if (params->helper) { 868a21b06e7SXin Long #if IS_ENABLED(CONFIG_NF_NAT) 869a21b06e7SXin Long if (params->ct_action & TCA_CT_ACT_NAT) 870a21b06e7SXin Long nf_nat_helper_put(params->helper); 871a21b06e7SXin Long #endif 872a21b06e7SXin Long nf_conntrack_helper_put(params->helper); 873a21b06e7SXin Long } 87419138941SXin Long if (params->ct_ft) 87519138941SXin Long tcf_ct_flow_table_put(params->ct_ft); 876b57dc7c1SPaul Blakey if (params->tmpl) 877408bdcfcSFlorian Westphal nf_ct_put(params->tmpl); 878b57dc7c1SPaul Blakey kfree(params); 879b57dc7c1SPaul Blakey } 880b57dc7c1SPaul Blakey 88119138941SXin Long static void tcf_ct_params_free_rcu(struct rcu_head *head) 88219138941SXin Long { 88319138941SXin Long struct tcf_ct_params *params; 88419138941SXin Long 88519138941SXin Long params = container_of(head, struct tcf_ct_params, rcu); 88619138941SXin Long tcf_ct_params_free(params); 88719138941SXin Long } 88819138941SXin Long 889b57dc7c1SPaul Blakey static void tcf_ct_act_set_mark(struct nf_conn *ct, u32 mark, u32 mask) 890b57dc7c1SPaul Blakey { 891b57dc7c1SPaul Blakey #if IS_ENABLED(CONFIG_NF_CONNTRACK_MARK) 892b57dc7c1SPaul Blakey u32 new_mark; 893b57dc7c1SPaul Blakey 894b57dc7c1SPaul Blakey if (!mask) 895b57dc7c1SPaul Blakey return; 896b57dc7c1SPaul Blakey 89752d1aa8bSDaniel Xu new_mark = mark | (READ_ONCE(ct->mark) & ~(mask)); 89852d1aa8bSDaniel Xu if (READ_ONCE(ct->mark) != new_mark) { 89952d1aa8bSDaniel Xu WRITE_ONCE(ct->mark, new_mark); 900b57dc7c1SPaul Blakey if (nf_ct_is_confirmed(ct)) 901b57dc7c1SPaul Blakey nf_conntrack_event_cache(IPCT_MARK, ct); 902b57dc7c1SPaul Blakey } 903b57dc7c1SPaul Blakey #endif 904b57dc7c1SPaul Blakey } 905b57dc7c1SPaul Blakey 906b57dc7c1SPaul Blakey static void tcf_ct_act_set_labels(struct nf_conn *ct, 907b57dc7c1SPaul Blakey u32 *labels, 908b57dc7c1SPaul Blakey u32 *labels_m) 909b57dc7c1SPaul Blakey { 910b57dc7c1SPaul Blakey #if IS_ENABLED(CONFIG_NF_CONNTRACK_LABELS) 911c593642cSPankaj Bharadiya size_t labels_sz = sizeof_field(struct tcf_ct_params, labels); 912b57dc7c1SPaul Blakey 913b57dc7c1SPaul Blakey if (!memchr_inv(labels_m, 0, labels_sz)) 914b57dc7c1SPaul Blakey return; 915b57dc7c1SPaul Blakey 916b57dc7c1SPaul Blakey nf_connlabels_replace(ct, labels, labels_m, 4); 917b57dc7c1SPaul Blakey #endif 918b57dc7c1SPaul Blakey } 919b57dc7c1SPaul Blakey 920b57dc7c1SPaul Blakey static int tcf_ct_act_nat(struct sk_buff *skb, 921b57dc7c1SPaul Blakey struct nf_conn *ct, 922b57dc7c1SPaul Blakey enum ip_conntrack_info ctinfo, 923b57dc7c1SPaul Blakey int ct_action, 924b57dc7c1SPaul Blakey struct nf_nat_range2 *range, 925b57dc7c1SPaul Blakey bool commit) 926b57dc7c1SPaul Blakey { 927b57dc7c1SPaul Blakey #if IS_ENABLED(CONFIG_NF_NAT) 928ebddb140SXin Long int err, action = 0; 929b57dc7c1SPaul Blakey 930b57dc7c1SPaul Blakey if (!(ct_action & TCA_CT_ACT_NAT)) 931b57dc7c1SPaul Blakey return NF_ACCEPT; 932ebddb140SXin Long if (ct_action & TCA_CT_ACT_NAT_SRC) 933ebddb140SXin Long action |= BIT(NF_NAT_MANIP_SRC); 934ebddb140SXin Long if (ct_action & TCA_CT_ACT_NAT_DST) 935ebddb140SXin Long action |= BIT(NF_NAT_MANIP_DST); 936b57dc7c1SPaul Blakey 937ebddb140SXin Long err = nf_ct_nat(skb, ct, ctinfo, &action, range, commit); 938b57dc7c1SPaul Blakey 939ebddb140SXin Long if (action & BIT(NF_NAT_MANIP_SRC)) 940ebddb140SXin Long tc_skb_cb(skb)->post_ct_snat = 1; 941ebddb140SXin Long if (action & BIT(NF_NAT_MANIP_DST)) 942ebddb140SXin Long tc_skb_cb(skb)->post_ct_dnat = 1; 943b57dc7c1SPaul Blakey 94495219afbSAaron Conole return err; 945b57dc7c1SPaul Blakey #else 946b57dc7c1SPaul Blakey return NF_ACCEPT; 947b57dc7c1SPaul Blakey #endif 948b57dc7c1SPaul Blakey } 949b57dc7c1SPaul Blakey 950871cf386SPedro Tammela TC_INDIRECT_SCOPE int tcf_ct_act(struct sk_buff *skb, const struct tc_action *a, 951b57dc7c1SPaul Blakey struct tcf_result *res) 952b57dc7c1SPaul Blakey { 953b57dc7c1SPaul Blakey struct net *net = dev_net(skb->dev); 954b57dc7c1SPaul Blakey enum ip_conntrack_info ctinfo; 955b57dc7c1SPaul Blakey struct tcf_ct *c = to_ct(a); 956b57dc7c1SPaul Blakey struct nf_conn *tmpl = NULL; 957b57dc7c1SPaul Blakey struct nf_hook_state state; 958a21b06e7SXin Long bool cached, commit, clear; 959b57dc7c1SPaul Blakey int nh_ofs, err, retval; 960b57dc7c1SPaul Blakey struct tcf_ct_params *p; 961a21b06e7SXin Long bool add_helper = false; 96246475bb2SPaul Blakey bool skip_add = false; 963ae372cb1Swenxu bool defrag = false; 964b57dc7c1SPaul Blakey struct nf_conn *ct; 965b57dc7c1SPaul Blakey u8 family; 966b57dc7c1SPaul Blakey 967b57dc7c1SPaul Blakey p = rcu_dereference_bh(c->params); 968b57dc7c1SPaul Blakey 969b57dc7c1SPaul Blakey retval = READ_ONCE(c->tcf_action); 970b57dc7c1SPaul Blakey commit = p->ct_action & TCA_CT_ACT_COMMIT; 971b57dc7c1SPaul Blakey clear = p->ct_action & TCA_CT_ACT_CLEAR; 972b57dc7c1SPaul Blakey tmpl = p->tmpl; 973b57dc7c1SPaul Blakey 9748367b3abSwenxu tcf_lastuse_update(&c->tcf_tm); 9752dc4e9e8SPaul Blakey tcf_action_update_bstats(&c->common, skb); 9768367b3abSwenxu 977b57dc7c1SPaul Blakey if (clear) { 978ec624fe7SPaul Blakey tc_skb_cb(skb)->post_ct = false; 979b57dc7c1SPaul Blakey ct = nf_ct_get(skb, &ctinfo); 980b57dc7c1SPaul Blakey if (ct) { 981408bdcfcSFlorian Westphal nf_ct_put(ct); 982b57dc7c1SPaul Blakey nf_ct_set(skb, NULL, IP_CT_UNTRACKED); 983b57dc7c1SPaul Blakey } 984b57dc7c1SPaul Blakey 9858ca1b090SMarcelo Ricardo Leitner goto out_clear; 986b57dc7c1SPaul Blakey } 987b57dc7c1SPaul Blakey 988b57dc7c1SPaul Blakey family = tcf_ct_skb_nf_family(skb); 989b57dc7c1SPaul Blakey if (family == NFPROTO_UNSPEC) 990b57dc7c1SPaul Blakey goto drop; 991b57dc7c1SPaul Blakey 992b57dc7c1SPaul Blakey /* The conntrack module expects to be working at L3. 993b57dc7c1SPaul Blakey * We also try to pull the IPv4/6 header to linear area 994b57dc7c1SPaul Blakey */ 995b57dc7c1SPaul Blakey nh_ofs = skb_network_offset(skb); 996b57dc7c1SPaul Blakey skb_pull_rcsum(skb, nh_ofs); 997ae372cb1Swenxu err = tcf_ct_handle_fragments(net, skb, family, p->zone, &defrag); 998b57dc7c1SPaul Blakey if (err == -EINPROGRESS) { 999b57dc7c1SPaul Blakey retval = TC_ACT_STOLEN; 1000f77bd544SDavide Caratti goto out_clear; 1001b57dc7c1SPaul Blakey } 1002b57dc7c1SPaul Blakey if (err) 1003b57dc7c1SPaul Blakey goto drop; 1004b57dc7c1SPaul Blakey 100567fc5d7fSXin Long err = nf_ct_skb_network_trim(skb, family); 1006b57dc7c1SPaul Blakey if (err) 1007b57dc7c1SPaul Blakey goto drop; 1008b57dc7c1SPaul Blakey 1009b57dc7c1SPaul Blakey /* If we are recirculating packets to match on ct fields and 1010b57dc7c1SPaul Blakey * committing with a separate ct action, then we don't need to 1011b57dc7c1SPaul Blakey * actually run the packet through conntrack twice unless it's for a 1012b57dc7c1SPaul Blakey * different zone. 1013b57dc7c1SPaul Blakey */ 1014a21b06e7SXin Long cached = tcf_ct_skb_nfct_cached(net, skb, p); 1015b57dc7c1SPaul Blakey if (!cached) { 10160cc254e5SPaul Blakey if (tcf_ct_flow_table_lookup(p, skb, family)) { 101746475bb2SPaul Blakey skip_add = true; 101846475bb2SPaul Blakey goto do_nat; 101946475bb2SPaul Blakey } 102046475bb2SPaul Blakey 1021b57dc7c1SPaul Blakey /* Associate skb with specified zone. */ 1022b57dc7c1SPaul Blakey if (tmpl) { 1023b57dc7c1SPaul Blakey nf_conntrack_put(skb_nfct(skb)); 1024b57dc7c1SPaul Blakey nf_conntrack_get(&tmpl->ct_general); 1025b57dc7c1SPaul Blakey nf_ct_set(skb, tmpl, IP_CT_NEW); 1026b57dc7c1SPaul Blakey } 1027b57dc7c1SPaul Blakey 1028b57dc7c1SPaul Blakey state.hook = NF_INET_PRE_ROUTING; 1029b57dc7c1SPaul Blakey state.net = net; 1030b57dc7c1SPaul Blakey state.pf = family; 1031b57dc7c1SPaul Blakey err = nf_conntrack_in(skb, &state); 1032b57dc7c1SPaul Blakey if (err != NF_ACCEPT) 1033b57dc7c1SPaul Blakey goto out_push; 1034b57dc7c1SPaul Blakey } 1035b57dc7c1SPaul Blakey 103646475bb2SPaul Blakey do_nat: 1037b57dc7c1SPaul Blakey ct = nf_ct_get(skb, &ctinfo); 1038b57dc7c1SPaul Blakey if (!ct) 1039b57dc7c1SPaul Blakey goto out_push; 1040b57dc7c1SPaul Blakey nf_ct_deliver_cached_events(ct); 10419795ded7SPaul Blakey nf_conn_act_ct_ext_fill(skb, ct, ctinfo); 1042b57dc7c1SPaul Blakey 1043b57dc7c1SPaul Blakey err = tcf_ct_act_nat(skb, ct, ctinfo, p->ct_action, &p->range, commit); 1044b57dc7c1SPaul Blakey if (err != NF_ACCEPT) 1045b57dc7c1SPaul Blakey goto drop; 1046b57dc7c1SPaul Blakey 1047a21b06e7SXin Long if (!nf_ct_is_confirmed(ct) && commit && p->helper && !nfct_help(ct)) { 1048a21b06e7SXin Long err = __nf_ct_try_assign_helper(ct, p->tmpl, GFP_ATOMIC); 1049a21b06e7SXin Long if (err) 1050a21b06e7SXin Long goto drop; 1051a21b06e7SXin Long add_helper = true; 1052a21b06e7SXin Long if (p->ct_action & TCA_CT_ACT_NAT && !nfct_seqadj(ct)) { 1053a21b06e7SXin Long if (!nfct_seqadj_ext_add(ct)) 1054a21b06e7SXin Long goto drop; 1055a21b06e7SXin Long } 1056a21b06e7SXin Long } 1057a21b06e7SXin Long 1058a21b06e7SXin Long if (nf_ct_is_confirmed(ct) ? ((!cached && !skip_add) || add_helper) : commit) { 1059a21b06e7SXin Long if (nf_ct_helper(skb, ct, ctinfo, family) != NF_ACCEPT) 1060a21b06e7SXin Long goto drop; 1061a21b06e7SXin Long } 1062a21b06e7SXin Long 1063b57dc7c1SPaul Blakey if (commit) { 1064b57dc7c1SPaul Blakey tcf_ct_act_set_mark(ct, p->mark, p->mark_mask); 1065b57dc7c1SPaul Blakey tcf_ct_act_set_labels(ct, p->labels, p->labels_mask); 1066b57dc7c1SPaul Blakey 10679795ded7SPaul Blakey if (!nf_ct_is_confirmed(ct)) 1068d5a116dbSVlad Buslov nf_conn_act_ct_ext_add(skb, ct, ctinfo); 10699795ded7SPaul Blakey 1070b57dc7c1SPaul Blakey /* This will take care of sending queued events 1071b57dc7c1SPaul Blakey * even if the connection is already confirmed. 1072b57dc7c1SPaul Blakey */ 10738955b90cSwenxu if (nf_conntrack_confirm(skb) != NF_ACCEPT) 10748955b90cSwenxu goto drop; 107546475bb2SPaul Blakey } 107664ff70b8SPaul Blakey 10770cc254e5SPaul Blakey if (!skip_add) 10780cc254e5SPaul Blakey tcf_ct_flow_table_process_conn(p->ct_ft, ct, ctinfo); 10790cc254e5SPaul Blakey 1080b57dc7c1SPaul Blakey out_push: 1081b57dc7c1SPaul Blakey skb_push_rcsum(skb, nh_ofs); 1082b57dc7c1SPaul Blakey 1083ec624fe7SPaul Blakey tc_skb_cb(skb)->post_ct = true; 108438495958SPaul Blakey tc_skb_cb(skb)->zone = p->zone; 10858ca1b090SMarcelo Ricardo Leitner out_clear: 1086ae372cb1Swenxu if (defrag) 1087ae372cb1Swenxu qdisc_skb_cb(skb)->pkt_len = skb->len; 1088b57dc7c1SPaul Blakey return retval; 1089b57dc7c1SPaul Blakey 1090b57dc7c1SPaul Blakey drop: 109126b537a8SVlad Buslov tcf_action_inc_drop_qstats(&c->common); 1092b57dc7c1SPaul Blakey return TC_ACT_SHOT; 1093b57dc7c1SPaul Blakey } 1094b57dc7c1SPaul Blakey 1095b57dc7c1SPaul Blakey static const struct nla_policy ct_policy[TCA_CT_MAX + 1] = { 1096b57dc7c1SPaul Blakey [TCA_CT_ACTION] = { .type = NLA_U16 }, 10978140860cSJohannes Berg [TCA_CT_PARMS] = NLA_POLICY_EXACT_LEN(sizeof(struct tc_ct)), 1098b57dc7c1SPaul Blakey [TCA_CT_ZONE] = { .type = NLA_U16 }, 1099b57dc7c1SPaul Blakey [TCA_CT_MARK] = { .type = NLA_U32 }, 1100b57dc7c1SPaul Blakey [TCA_CT_MARK_MASK] = { .type = NLA_U32 }, 1101b57dc7c1SPaul Blakey [TCA_CT_LABELS] = { .type = NLA_BINARY, 1102b57dc7c1SPaul Blakey .len = 128 / BITS_PER_BYTE }, 1103b57dc7c1SPaul Blakey [TCA_CT_LABELS_MASK] = { .type = NLA_BINARY, 1104b57dc7c1SPaul Blakey .len = 128 / BITS_PER_BYTE }, 1105b57dc7c1SPaul Blakey [TCA_CT_NAT_IPV4_MIN] = { .type = NLA_U32 }, 1106b57dc7c1SPaul Blakey [TCA_CT_NAT_IPV4_MAX] = { .type = NLA_U32 }, 11078140860cSJohannes Berg [TCA_CT_NAT_IPV6_MIN] = NLA_POLICY_EXACT_LEN(sizeof(struct in6_addr)), 11088140860cSJohannes Berg [TCA_CT_NAT_IPV6_MAX] = NLA_POLICY_EXACT_LEN(sizeof(struct in6_addr)), 1109b57dc7c1SPaul Blakey [TCA_CT_NAT_PORT_MIN] = { .type = NLA_U16 }, 1110b57dc7c1SPaul Blakey [TCA_CT_NAT_PORT_MAX] = { .type = NLA_U16 }, 1111a21b06e7SXin Long [TCA_CT_HELPER_NAME] = { .type = NLA_STRING, .len = NF_CT_HELPER_NAME_LEN }, 1112a21b06e7SXin Long [TCA_CT_HELPER_FAMILY] = { .type = NLA_U8 }, 1113a21b06e7SXin Long [TCA_CT_HELPER_PROTO] = { .type = NLA_U8 }, 1114b57dc7c1SPaul Blakey }; 1115b57dc7c1SPaul Blakey 1116b57dc7c1SPaul Blakey static int tcf_ct_fill_params_nat(struct tcf_ct_params *p, 1117b57dc7c1SPaul Blakey struct tc_ct *parm, 1118b57dc7c1SPaul Blakey struct nlattr **tb, 1119b57dc7c1SPaul Blakey struct netlink_ext_ack *extack) 1120b57dc7c1SPaul Blakey { 1121b57dc7c1SPaul Blakey struct nf_nat_range2 *range; 1122b57dc7c1SPaul Blakey 1123b57dc7c1SPaul Blakey if (!(p->ct_action & TCA_CT_ACT_NAT)) 1124b57dc7c1SPaul Blakey return 0; 1125b57dc7c1SPaul Blakey 1126b57dc7c1SPaul Blakey if (!IS_ENABLED(CONFIG_NF_NAT)) { 1127b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Netfilter nat isn't enabled in kernel"); 1128b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1129b57dc7c1SPaul Blakey } 1130b57dc7c1SPaul Blakey 1131b57dc7c1SPaul Blakey if (!(p->ct_action & (TCA_CT_ACT_NAT_SRC | TCA_CT_ACT_NAT_DST))) 1132b57dc7c1SPaul Blakey return 0; 1133b57dc7c1SPaul Blakey 1134b57dc7c1SPaul Blakey if ((p->ct_action & TCA_CT_ACT_NAT_SRC) && 1135b57dc7c1SPaul Blakey (p->ct_action & TCA_CT_ACT_NAT_DST)) { 1136b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "dnat and snat can't be enabled at the same time"); 1137b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1138b57dc7c1SPaul Blakey } 1139b57dc7c1SPaul Blakey 1140b57dc7c1SPaul Blakey range = &p->range; 1141b57dc7c1SPaul Blakey if (tb[TCA_CT_NAT_IPV4_MIN]) { 1142b57dc7c1SPaul Blakey struct nlattr *max_attr = tb[TCA_CT_NAT_IPV4_MAX]; 1143b57dc7c1SPaul Blakey 1144b57dc7c1SPaul Blakey p->ipv4_range = true; 1145b57dc7c1SPaul Blakey range->flags |= NF_NAT_RANGE_MAP_IPS; 1146b57dc7c1SPaul Blakey range->min_addr.ip = 1147b57dc7c1SPaul Blakey nla_get_in_addr(tb[TCA_CT_NAT_IPV4_MIN]); 1148b57dc7c1SPaul Blakey 1149b57dc7c1SPaul Blakey range->max_addr.ip = max_attr ? 1150b57dc7c1SPaul Blakey nla_get_in_addr(max_attr) : 1151b57dc7c1SPaul Blakey range->min_addr.ip; 1152b57dc7c1SPaul Blakey } else if (tb[TCA_CT_NAT_IPV6_MIN]) { 1153b57dc7c1SPaul Blakey struct nlattr *max_attr = tb[TCA_CT_NAT_IPV6_MAX]; 1154b57dc7c1SPaul Blakey 1155b57dc7c1SPaul Blakey p->ipv4_range = false; 1156b57dc7c1SPaul Blakey range->flags |= NF_NAT_RANGE_MAP_IPS; 1157b57dc7c1SPaul Blakey range->min_addr.in6 = 1158b57dc7c1SPaul Blakey nla_get_in6_addr(tb[TCA_CT_NAT_IPV6_MIN]); 1159b57dc7c1SPaul Blakey 1160b57dc7c1SPaul Blakey range->max_addr.in6 = max_attr ? 1161b57dc7c1SPaul Blakey nla_get_in6_addr(max_attr) : 1162b57dc7c1SPaul Blakey range->min_addr.in6; 1163b57dc7c1SPaul Blakey } 1164b57dc7c1SPaul Blakey 1165b57dc7c1SPaul Blakey if (tb[TCA_CT_NAT_PORT_MIN]) { 1166b57dc7c1SPaul Blakey range->flags |= NF_NAT_RANGE_PROTO_SPECIFIED; 1167b57dc7c1SPaul Blakey range->min_proto.all = nla_get_be16(tb[TCA_CT_NAT_PORT_MIN]); 1168b57dc7c1SPaul Blakey 1169b57dc7c1SPaul Blakey range->max_proto.all = tb[TCA_CT_NAT_PORT_MAX] ? 1170b57dc7c1SPaul Blakey nla_get_be16(tb[TCA_CT_NAT_PORT_MAX]) : 1171b57dc7c1SPaul Blakey range->min_proto.all; 1172b57dc7c1SPaul Blakey } 1173b57dc7c1SPaul Blakey 1174b57dc7c1SPaul Blakey return 0; 1175b57dc7c1SPaul Blakey } 1176b57dc7c1SPaul Blakey 1177b57dc7c1SPaul Blakey static void tcf_ct_set_key_val(struct nlattr **tb, 1178b57dc7c1SPaul Blakey void *val, int val_type, 1179b57dc7c1SPaul Blakey void *mask, int mask_type, 1180b57dc7c1SPaul Blakey int len) 1181b57dc7c1SPaul Blakey { 1182b57dc7c1SPaul Blakey if (!tb[val_type]) 1183b57dc7c1SPaul Blakey return; 1184b57dc7c1SPaul Blakey nla_memcpy(val, tb[val_type], len); 1185b57dc7c1SPaul Blakey 1186b57dc7c1SPaul Blakey if (!mask) 1187b57dc7c1SPaul Blakey return; 1188b57dc7c1SPaul Blakey 1189b57dc7c1SPaul Blakey if (mask_type == TCA_CT_UNSPEC || !tb[mask_type]) 1190b57dc7c1SPaul Blakey memset(mask, 0xff, len); 1191b57dc7c1SPaul Blakey else 1192b57dc7c1SPaul Blakey nla_memcpy(mask, tb[mask_type], len); 1193b57dc7c1SPaul Blakey } 1194b57dc7c1SPaul Blakey 1195b57dc7c1SPaul Blakey static int tcf_ct_fill_params(struct net *net, 1196b57dc7c1SPaul Blakey struct tcf_ct_params *p, 1197b57dc7c1SPaul Blakey struct tc_ct *parm, 1198b57dc7c1SPaul Blakey struct nlattr **tb, 1199b57dc7c1SPaul Blakey struct netlink_ext_ack *extack) 1200b57dc7c1SPaul Blakey { 1201acd0a7abSZhengchao Shao struct tc_ct_action_net *tn = net_generic(net, act_ct_ops.net_id); 1202b57dc7c1SPaul Blakey struct nf_conntrack_zone zone; 1203a21b06e7SXin Long int err, family, proto, len; 1204b57dc7c1SPaul Blakey struct nf_conn *tmpl; 1205a21b06e7SXin Long char *name; 1206b57dc7c1SPaul Blakey 1207b57dc7c1SPaul Blakey p->zone = NF_CT_DEFAULT_ZONE_ID; 1208b57dc7c1SPaul Blakey 1209b57dc7c1SPaul Blakey tcf_ct_set_key_val(tb, 1210b57dc7c1SPaul Blakey &p->ct_action, TCA_CT_ACTION, 1211b57dc7c1SPaul Blakey NULL, TCA_CT_UNSPEC, 1212b57dc7c1SPaul Blakey sizeof(p->ct_action)); 1213b57dc7c1SPaul Blakey 1214b57dc7c1SPaul Blakey if (p->ct_action & TCA_CT_ACT_CLEAR) 1215b57dc7c1SPaul Blakey return 0; 1216b57dc7c1SPaul Blakey 1217b57dc7c1SPaul Blakey err = tcf_ct_fill_params_nat(p, parm, tb, extack); 1218b57dc7c1SPaul Blakey if (err) 1219b57dc7c1SPaul Blakey return err; 1220b57dc7c1SPaul Blakey 1221b57dc7c1SPaul Blakey if (tb[TCA_CT_MARK]) { 1222b57dc7c1SPaul Blakey if (!IS_ENABLED(CONFIG_NF_CONNTRACK_MARK)) { 1223b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Conntrack mark isn't enabled."); 1224b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1225b57dc7c1SPaul Blakey } 1226b57dc7c1SPaul Blakey tcf_ct_set_key_val(tb, 1227b57dc7c1SPaul Blakey &p->mark, TCA_CT_MARK, 1228b57dc7c1SPaul Blakey &p->mark_mask, TCA_CT_MARK_MASK, 1229b57dc7c1SPaul Blakey sizeof(p->mark)); 1230b57dc7c1SPaul Blakey } 1231b57dc7c1SPaul Blakey 1232b57dc7c1SPaul Blakey if (tb[TCA_CT_LABELS]) { 1233b57dc7c1SPaul Blakey if (!IS_ENABLED(CONFIG_NF_CONNTRACK_LABELS)) { 1234b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Conntrack labels isn't enabled."); 1235b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1236b57dc7c1SPaul Blakey } 1237b57dc7c1SPaul Blakey 1238b57dc7c1SPaul Blakey if (!tn->labels) { 1239b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Failed to set connlabel length"); 1240b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1241b57dc7c1SPaul Blakey } 1242b57dc7c1SPaul Blakey tcf_ct_set_key_val(tb, 1243b57dc7c1SPaul Blakey p->labels, TCA_CT_LABELS, 1244b57dc7c1SPaul Blakey p->labels_mask, TCA_CT_LABELS_MASK, 1245b57dc7c1SPaul Blakey sizeof(p->labels)); 1246b57dc7c1SPaul Blakey } 1247b57dc7c1SPaul Blakey 1248b57dc7c1SPaul Blakey if (tb[TCA_CT_ZONE]) { 1249b57dc7c1SPaul Blakey if (!IS_ENABLED(CONFIG_NF_CONNTRACK_ZONES)) { 1250b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Conntrack zones isn't enabled."); 1251b57dc7c1SPaul Blakey return -EOPNOTSUPP; 1252b57dc7c1SPaul Blakey } 1253b57dc7c1SPaul Blakey 1254b57dc7c1SPaul Blakey tcf_ct_set_key_val(tb, 1255b57dc7c1SPaul Blakey &p->zone, TCA_CT_ZONE, 1256b57dc7c1SPaul Blakey NULL, TCA_CT_UNSPEC, 1257b57dc7c1SPaul Blakey sizeof(p->zone)); 1258b57dc7c1SPaul Blakey } 1259b57dc7c1SPaul Blakey 1260b57dc7c1SPaul Blakey nf_ct_zone_init(&zone, p->zone, NF_CT_DEFAULT_ZONE_DIR, 0); 1261b57dc7c1SPaul Blakey tmpl = nf_ct_tmpl_alloc(net, &zone, GFP_KERNEL); 1262b57dc7c1SPaul Blakey if (!tmpl) { 1263b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Failed to allocate conntrack template"); 1264b57dc7c1SPaul Blakey return -ENOMEM; 1265b57dc7c1SPaul Blakey } 1266b57dc7c1SPaul Blakey p->tmpl = tmpl; 1267a21b06e7SXin Long if (tb[TCA_CT_HELPER_NAME]) { 1268a21b06e7SXin Long name = nla_data(tb[TCA_CT_HELPER_NAME]); 1269a21b06e7SXin Long len = nla_len(tb[TCA_CT_HELPER_NAME]); 1270a21b06e7SXin Long if (len > 16 || name[len - 1] != '\0') { 1271a21b06e7SXin Long NL_SET_ERR_MSG_MOD(extack, "Failed to parse helper name."); 1272a21b06e7SXin Long err = -EINVAL; 1273a21b06e7SXin Long goto err; 1274a21b06e7SXin Long } 1275a21b06e7SXin Long family = tb[TCA_CT_HELPER_FAMILY] ? nla_get_u8(tb[TCA_CT_HELPER_FAMILY]) : AF_INET; 1276a21b06e7SXin Long proto = tb[TCA_CT_HELPER_PROTO] ? nla_get_u8(tb[TCA_CT_HELPER_PROTO]) : IPPROTO_TCP; 1277a21b06e7SXin Long err = nf_ct_add_helper(tmpl, name, family, proto, 1278a21b06e7SXin Long p->ct_action & TCA_CT_ACT_NAT, &p->helper); 1279a21b06e7SXin Long if (err) { 1280a21b06e7SXin Long NL_SET_ERR_MSG_MOD(extack, "Failed to add helper"); 1281a21b06e7SXin Long goto err; 1282a21b06e7SXin Long } 1283a21b06e7SXin Long } 1284b57dc7c1SPaul Blakey 128576622cedSXin Long if (p->ct_action & TCA_CT_ACT_COMMIT) 1286a21b06e7SXin Long __set_bit(IPS_CONFIRMED_BIT, &tmpl->status); 1287b57dc7c1SPaul Blakey return 0; 1288a21b06e7SXin Long err: 1289a21b06e7SXin Long nf_ct_put(p->tmpl); 1290a21b06e7SXin Long p->tmpl = NULL; 1291a21b06e7SXin Long return err; 1292b57dc7c1SPaul Blakey } 1293b57dc7c1SPaul Blakey 1294b57dc7c1SPaul Blakey static int tcf_ct_init(struct net *net, struct nlattr *nla, 1295b57dc7c1SPaul Blakey struct nlattr *est, struct tc_action **a, 1296abbb0d33SVlad Buslov struct tcf_proto *tp, u32 flags, 1297b57dc7c1SPaul Blakey struct netlink_ext_ack *extack) 1298b57dc7c1SPaul Blakey { 1299acd0a7abSZhengchao Shao struct tc_action_net *tn = net_generic(net, act_ct_ops.net_id); 1300695176bfSCong Wang bool bind = flags & TCA_ACT_FLAGS_BIND; 1301b57dc7c1SPaul Blakey struct tcf_ct_params *params = NULL; 1302b57dc7c1SPaul Blakey struct nlattr *tb[TCA_CT_MAX + 1]; 1303b57dc7c1SPaul Blakey struct tcf_chain *goto_ch = NULL; 1304b57dc7c1SPaul Blakey struct tc_ct *parm; 1305b57dc7c1SPaul Blakey struct tcf_ct *c; 1306b57dc7c1SPaul Blakey int err, res = 0; 13077be8ef2cSDmytro Linkin u32 index; 1308b57dc7c1SPaul Blakey 1309b57dc7c1SPaul Blakey if (!nla) { 1310b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Ct requires attributes to be passed"); 1311b57dc7c1SPaul Blakey return -EINVAL; 1312b57dc7c1SPaul Blakey } 1313b57dc7c1SPaul Blakey 1314b57dc7c1SPaul Blakey err = nla_parse_nested(tb, TCA_CT_MAX, nla, ct_policy, extack); 1315b57dc7c1SPaul Blakey if (err < 0) 1316b57dc7c1SPaul Blakey return err; 1317b57dc7c1SPaul Blakey 1318b57dc7c1SPaul Blakey if (!tb[TCA_CT_PARMS]) { 1319b57dc7c1SPaul Blakey NL_SET_ERR_MSG_MOD(extack, "Missing required ct parameters"); 1320b57dc7c1SPaul Blakey return -EINVAL; 1321b57dc7c1SPaul Blakey } 1322b57dc7c1SPaul Blakey parm = nla_data(tb[TCA_CT_PARMS]); 13237be8ef2cSDmytro Linkin index = parm->index; 13247be8ef2cSDmytro Linkin err = tcf_idr_check_alloc(tn, &index, a, bind); 1325b57dc7c1SPaul Blakey if (err < 0) 1326b57dc7c1SPaul Blakey return err; 1327b57dc7c1SPaul Blakey 1328b57dc7c1SPaul Blakey if (!err) { 1329e3822678SVlad Buslov err = tcf_idr_create_from_flags(tn, index, est, a, 1330e3822678SVlad Buslov &act_ct_ops, bind, flags); 1331b57dc7c1SPaul Blakey if (err) { 13327be8ef2cSDmytro Linkin tcf_idr_cleanup(tn, index); 1333b57dc7c1SPaul Blakey return err; 1334b57dc7c1SPaul Blakey } 1335b57dc7c1SPaul Blakey res = ACT_P_CREATED; 1336b57dc7c1SPaul Blakey } else { 1337b57dc7c1SPaul Blakey if (bind) 1338b57dc7c1SPaul Blakey return 0; 1339b57dc7c1SPaul Blakey 1340695176bfSCong Wang if (!(flags & TCA_ACT_FLAGS_REPLACE)) { 1341b57dc7c1SPaul Blakey tcf_idr_release(*a, bind); 1342b57dc7c1SPaul Blakey return -EEXIST; 1343b57dc7c1SPaul Blakey } 1344b57dc7c1SPaul Blakey } 1345b57dc7c1SPaul Blakey err = tcf_action_check_ctrlact(parm->action, tp, &goto_ch, extack); 1346b57dc7c1SPaul Blakey if (err < 0) 1347b57dc7c1SPaul Blakey goto cleanup; 1348b57dc7c1SPaul Blakey 1349b57dc7c1SPaul Blakey c = to_ct(*a); 1350b57dc7c1SPaul Blakey 1351b57dc7c1SPaul Blakey params = kzalloc(sizeof(*params), GFP_KERNEL); 1352b57dc7c1SPaul Blakey if (unlikely(!params)) { 1353b57dc7c1SPaul Blakey err = -ENOMEM; 1354b57dc7c1SPaul Blakey goto cleanup; 1355b57dc7c1SPaul Blakey } 1356b57dc7c1SPaul Blakey 1357b57dc7c1SPaul Blakey err = tcf_ct_fill_params(net, params, parm, tb, extack); 1358b57dc7c1SPaul Blakey if (err) 1359b57dc7c1SPaul Blakey goto cleanup; 1360b57dc7c1SPaul Blakey 1361fc54d906SVlad Buslov err = tcf_ct_flow_table_get(net, params); 1362c34b961aSPaul Blakey if (err) 136319138941SXin Long goto cleanup; 1364c34b961aSPaul Blakey 1365b57dc7c1SPaul Blakey spin_lock_bh(&c->tcf_lock); 1366b57dc7c1SPaul Blakey goto_ch = tcf_action_set_ctrlact(*a, parm->action, goto_ch); 1367445d3749SPaul E. McKenney params = rcu_replace_pointer(c->params, params, 1368445d3749SPaul E. McKenney lockdep_is_held(&c->tcf_lock)); 1369b57dc7c1SPaul Blakey spin_unlock_bh(&c->tcf_lock); 1370b57dc7c1SPaul Blakey 1371b57dc7c1SPaul Blakey if (goto_ch) 1372b57dc7c1SPaul Blakey tcf_chain_put_by_act(goto_ch); 1373b57dc7c1SPaul Blakey if (params) 137419138941SXin Long call_rcu(¶ms->rcu, tcf_ct_params_free_rcu); 1375b57dc7c1SPaul Blakey 1376b57dc7c1SPaul Blakey return res; 1377b57dc7c1SPaul Blakey 1378b57dc7c1SPaul Blakey cleanup: 1379b57dc7c1SPaul Blakey if (goto_ch) 1380b57dc7c1SPaul Blakey tcf_chain_put_by_act(goto_ch); 138119138941SXin Long if (params) 138219138941SXin Long tcf_ct_params_free(params); 1383b57dc7c1SPaul Blakey tcf_idr_release(*a, bind); 1384b57dc7c1SPaul Blakey return err; 1385b57dc7c1SPaul Blakey } 1386b57dc7c1SPaul Blakey 1387b57dc7c1SPaul Blakey static void tcf_ct_cleanup(struct tc_action *a) 1388b57dc7c1SPaul Blakey { 1389b57dc7c1SPaul Blakey struct tcf_ct_params *params; 1390b57dc7c1SPaul Blakey struct tcf_ct *c = to_ct(a); 1391b57dc7c1SPaul Blakey 1392b57dc7c1SPaul Blakey params = rcu_dereference_protected(c->params, 1); 1393b57dc7c1SPaul Blakey if (params) 139419138941SXin Long call_rcu(¶ms->rcu, tcf_ct_params_free_rcu); 1395b57dc7c1SPaul Blakey } 1396b57dc7c1SPaul Blakey 1397b57dc7c1SPaul Blakey static int tcf_ct_dump_key_val(struct sk_buff *skb, 1398b57dc7c1SPaul Blakey void *val, int val_type, 1399b57dc7c1SPaul Blakey void *mask, int mask_type, 1400b57dc7c1SPaul Blakey int len) 1401b57dc7c1SPaul Blakey { 1402b57dc7c1SPaul Blakey int err; 1403b57dc7c1SPaul Blakey 1404b57dc7c1SPaul Blakey if (mask && !memchr_inv(mask, 0, len)) 1405b57dc7c1SPaul Blakey return 0; 1406b57dc7c1SPaul Blakey 1407b57dc7c1SPaul Blakey err = nla_put(skb, val_type, len, val); 1408b57dc7c1SPaul Blakey if (err) 1409b57dc7c1SPaul Blakey return err; 1410b57dc7c1SPaul Blakey 1411b57dc7c1SPaul Blakey if (mask_type != TCA_CT_UNSPEC) { 1412b57dc7c1SPaul Blakey err = nla_put(skb, mask_type, len, mask); 1413b57dc7c1SPaul Blakey if (err) 1414b57dc7c1SPaul Blakey return err; 1415b57dc7c1SPaul Blakey } 1416b57dc7c1SPaul Blakey 1417b57dc7c1SPaul Blakey return 0; 1418b57dc7c1SPaul Blakey } 1419b57dc7c1SPaul Blakey 1420b57dc7c1SPaul Blakey static int tcf_ct_dump_nat(struct sk_buff *skb, struct tcf_ct_params *p) 1421b57dc7c1SPaul Blakey { 1422b57dc7c1SPaul Blakey struct nf_nat_range2 *range = &p->range; 1423b57dc7c1SPaul Blakey 1424b57dc7c1SPaul Blakey if (!(p->ct_action & TCA_CT_ACT_NAT)) 1425b57dc7c1SPaul Blakey return 0; 1426b57dc7c1SPaul Blakey 1427b57dc7c1SPaul Blakey if (!(p->ct_action & (TCA_CT_ACT_NAT_SRC | TCA_CT_ACT_NAT_DST))) 1428b57dc7c1SPaul Blakey return 0; 1429b57dc7c1SPaul Blakey 1430b57dc7c1SPaul Blakey if (range->flags & NF_NAT_RANGE_MAP_IPS) { 1431b57dc7c1SPaul Blakey if (p->ipv4_range) { 1432b57dc7c1SPaul Blakey if (nla_put_in_addr(skb, TCA_CT_NAT_IPV4_MIN, 1433b57dc7c1SPaul Blakey range->min_addr.ip)) 1434b57dc7c1SPaul Blakey return -1; 1435b57dc7c1SPaul Blakey if (nla_put_in_addr(skb, TCA_CT_NAT_IPV4_MAX, 1436b57dc7c1SPaul Blakey range->max_addr.ip)) 1437b57dc7c1SPaul Blakey return -1; 1438b57dc7c1SPaul Blakey } else { 1439b57dc7c1SPaul Blakey if (nla_put_in6_addr(skb, TCA_CT_NAT_IPV6_MIN, 1440b57dc7c1SPaul Blakey &range->min_addr.in6)) 1441b57dc7c1SPaul Blakey return -1; 1442b57dc7c1SPaul Blakey if (nla_put_in6_addr(skb, TCA_CT_NAT_IPV6_MAX, 1443b57dc7c1SPaul Blakey &range->max_addr.in6)) 1444b57dc7c1SPaul Blakey return -1; 1445b57dc7c1SPaul Blakey } 1446b57dc7c1SPaul Blakey } 1447b57dc7c1SPaul Blakey 1448b57dc7c1SPaul Blakey if (range->flags & NF_NAT_RANGE_PROTO_SPECIFIED) { 1449b57dc7c1SPaul Blakey if (nla_put_be16(skb, TCA_CT_NAT_PORT_MIN, 1450b57dc7c1SPaul Blakey range->min_proto.all)) 1451b57dc7c1SPaul Blakey return -1; 1452b57dc7c1SPaul Blakey if (nla_put_be16(skb, TCA_CT_NAT_PORT_MAX, 1453b57dc7c1SPaul Blakey range->max_proto.all)) 1454b57dc7c1SPaul Blakey return -1; 1455b57dc7c1SPaul Blakey } 1456b57dc7c1SPaul Blakey 1457b57dc7c1SPaul Blakey return 0; 1458b57dc7c1SPaul Blakey } 1459b57dc7c1SPaul Blakey 1460a21b06e7SXin Long static int tcf_ct_dump_helper(struct sk_buff *skb, struct nf_conntrack_helper *helper) 1461a21b06e7SXin Long { 1462a21b06e7SXin Long if (!helper) 1463a21b06e7SXin Long return 0; 1464a21b06e7SXin Long 1465a21b06e7SXin Long if (nla_put_string(skb, TCA_CT_HELPER_NAME, helper->name) || 1466a21b06e7SXin Long nla_put_u8(skb, TCA_CT_HELPER_FAMILY, helper->tuple.src.l3num) || 1467a21b06e7SXin Long nla_put_u8(skb, TCA_CT_HELPER_PROTO, helper->tuple.dst.protonum)) 1468a21b06e7SXin Long return -1; 1469a21b06e7SXin Long 1470a21b06e7SXin Long return 0; 1471a21b06e7SXin Long } 1472a21b06e7SXin Long 1473b57dc7c1SPaul Blakey static inline int tcf_ct_dump(struct sk_buff *skb, struct tc_action *a, 1474b57dc7c1SPaul Blakey int bind, int ref) 1475b57dc7c1SPaul Blakey { 1476b57dc7c1SPaul Blakey unsigned char *b = skb_tail_pointer(skb); 1477b57dc7c1SPaul Blakey struct tcf_ct *c = to_ct(a); 1478b57dc7c1SPaul Blakey struct tcf_ct_params *p; 1479b57dc7c1SPaul Blakey 1480b57dc7c1SPaul Blakey struct tc_ct opt = { 1481b57dc7c1SPaul Blakey .index = c->tcf_index, 1482b57dc7c1SPaul Blakey .refcnt = refcount_read(&c->tcf_refcnt) - ref, 1483b57dc7c1SPaul Blakey .bindcnt = atomic_read(&c->tcf_bindcnt) - bind, 1484b57dc7c1SPaul Blakey }; 1485b57dc7c1SPaul Blakey struct tcf_t t; 1486b57dc7c1SPaul Blakey 1487b57dc7c1SPaul Blakey spin_lock_bh(&c->tcf_lock); 1488b57dc7c1SPaul Blakey p = rcu_dereference_protected(c->params, 1489b57dc7c1SPaul Blakey lockdep_is_held(&c->tcf_lock)); 1490b57dc7c1SPaul Blakey opt.action = c->tcf_action; 1491b57dc7c1SPaul Blakey 1492b57dc7c1SPaul Blakey if (tcf_ct_dump_key_val(skb, 1493b57dc7c1SPaul Blakey &p->ct_action, TCA_CT_ACTION, 1494b57dc7c1SPaul Blakey NULL, TCA_CT_UNSPEC, 1495b57dc7c1SPaul Blakey sizeof(p->ct_action))) 1496b57dc7c1SPaul Blakey goto nla_put_failure; 1497b57dc7c1SPaul Blakey 1498b57dc7c1SPaul Blakey if (p->ct_action & TCA_CT_ACT_CLEAR) 1499b57dc7c1SPaul Blakey goto skip_dump; 1500b57dc7c1SPaul Blakey 1501b57dc7c1SPaul Blakey if (IS_ENABLED(CONFIG_NF_CONNTRACK_MARK) && 1502b57dc7c1SPaul Blakey tcf_ct_dump_key_val(skb, 1503b57dc7c1SPaul Blakey &p->mark, TCA_CT_MARK, 1504b57dc7c1SPaul Blakey &p->mark_mask, TCA_CT_MARK_MASK, 1505b57dc7c1SPaul Blakey sizeof(p->mark))) 1506b57dc7c1SPaul Blakey goto nla_put_failure; 1507b57dc7c1SPaul Blakey 1508b57dc7c1SPaul Blakey if (IS_ENABLED(CONFIG_NF_CONNTRACK_LABELS) && 1509b57dc7c1SPaul Blakey tcf_ct_dump_key_val(skb, 1510b57dc7c1SPaul Blakey p->labels, TCA_CT_LABELS, 1511b57dc7c1SPaul Blakey p->labels_mask, TCA_CT_LABELS_MASK, 1512b57dc7c1SPaul Blakey sizeof(p->labels))) 1513b57dc7c1SPaul Blakey goto nla_put_failure; 1514b57dc7c1SPaul Blakey 1515b57dc7c1SPaul Blakey if (IS_ENABLED(CONFIG_NF_CONNTRACK_ZONES) && 1516b57dc7c1SPaul Blakey tcf_ct_dump_key_val(skb, 1517b57dc7c1SPaul Blakey &p->zone, TCA_CT_ZONE, 1518b57dc7c1SPaul Blakey NULL, TCA_CT_UNSPEC, 1519b57dc7c1SPaul Blakey sizeof(p->zone))) 1520b57dc7c1SPaul Blakey goto nla_put_failure; 1521b57dc7c1SPaul Blakey 1522b57dc7c1SPaul Blakey if (tcf_ct_dump_nat(skb, p)) 1523b57dc7c1SPaul Blakey goto nla_put_failure; 1524b57dc7c1SPaul Blakey 1525a21b06e7SXin Long if (tcf_ct_dump_helper(skb, p->helper)) 1526a21b06e7SXin Long goto nla_put_failure; 1527a21b06e7SXin Long 1528b57dc7c1SPaul Blakey skip_dump: 1529b57dc7c1SPaul Blakey if (nla_put(skb, TCA_CT_PARMS, sizeof(opt), &opt)) 1530b57dc7c1SPaul Blakey goto nla_put_failure; 1531b57dc7c1SPaul Blakey 1532b57dc7c1SPaul Blakey tcf_tm_dump(&t, &c->tcf_tm); 1533b57dc7c1SPaul Blakey if (nla_put_64bit(skb, TCA_CT_TM, sizeof(t), &t, TCA_CT_PAD)) 1534b57dc7c1SPaul Blakey goto nla_put_failure; 1535b57dc7c1SPaul Blakey spin_unlock_bh(&c->tcf_lock); 1536b57dc7c1SPaul Blakey 1537b57dc7c1SPaul Blakey return skb->len; 1538b57dc7c1SPaul Blakey nla_put_failure: 1539b57dc7c1SPaul Blakey spin_unlock_bh(&c->tcf_lock); 1540b57dc7c1SPaul Blakey nlmsg_trim(skb, b); 1541b57dc7c1SPaul Blakey return -1; 1542b57dc7c1SPaul Blakey } 1543b57dc7c1SPaul Blakey 15444b61d3e8SPo Liu static void tcf_stats_update(struct tc_action *a, u64 bytes, u64 packets, 15454b61d3e8SPo Liu u64 drops, u64 lastuse, bool hw) 1546b57dc7c1SPaul Blakey { 1547b57dc7c1SPaul Blakey struct tcf_ct *c = to_ct(a); 1548b57dc7c1SPaul Blakey 15494b61d3e8SPo Liu tcf_action_update_stats(a, bytes, packets, drops, hw); 1550b57dc7c1SPaul Blakey c->tcf_tm.lastuse = max_t(u64, c->tcf_tm.lastuse, lastuse); 1551b57dc7c1SPaul Blakey } 1552b57dc7c1SPaul Blakey 1553c54e1d92SBaowen Zheng static int tcf_ct_offload_act_setup(struct tc_action *act, void *entry_data, 1554c2ccf84eSIdo Schimmel u32 *index_inc, bool bind, 1555c2ccf84eSIdo Schimmel struct netlink_ext_ack *extack) 1556c54e1d92SBaowen Zheng { 1557c54e1d92SBaowen Zheng if (bind) { 1558c54e1d92SBaowen Zheng struct flow_action_entry *entry = entry_data; 1559c54e1d92SBaowen Zheng 15600a7c9d1fSXin Long if (tcf_ct_helper(act)) 15610a7c9d1fSXin Long return -EOPNOTSUPP; 15620a7c9d1fSXin Long 1563c54e1d92SBaowen Zheng entry->id = FLOW_ACTION_CT; 1564c54e1d92SBaowen Zheng entry->ct.action = tcf_ct_action(act); 1565c54e1d92SBaowen Zheng entry->ct.zone = tcf_ct_zone(act); 1566c54e1d92SBaowen Zheng entry->ct.flow_table = tcf_ct_ft(act); 1567c54e1d92SBaowen Zheng *index_inc = 1; 1568c54e1d92SBaowen Zheng } else { 15698cbfe939SBaowen Zheng struct flow_offload_action *fl_action = entry_data; 15708cbfe939SBaowen Zheng 15718cbfe939SBaowen Zheng fl_action->id = FLOW_ACTION_CT; 1572c54e1d92SBaowen Zheng } 1573c54e1d92SBaowen Zheng 1574c54e1d92SBaowen Zheng return 0; 1575c54e1d92SBaowen Zheng } 1576c54e1d92SBaowen Zheng 1577b57dc7c1SPaul Blakey static struct tc_action_ops act_ct_ops = { 1578b57dc7c1SPaul Blakey .kind = "ct", 1579b57dc7c1SPaul Blakey .id = TCA_ID_CT, 1580b57dc7c1SPaul Blakey .owner = THIS_MODULE, 1581b57dc7c1SPaul Blakey .act = tcf_ct_act, 1582b57dc7c1SPaul Blakey .dump = tcf_ct_dump, 1583b57dc7c1SPaul Blakey .init = tcf_ct_init, 1584b57dc7c1SPaul Blakey .cleanup = tcf_ct_cleanup, 1585b57dc7c1SPaul Blakey .stats_update = tcf_stats_update, 1586c54e1d92SBaowen Zheng .offload_act_setup = tcf_ct_offload_act_setup, 1587b57dc7c1SPaul Blakey .size = sizeof(struct tcf_ct), 1588b57dc7c1SPaul Blakey }; 1589b57dc7c1SPaul Blakey 1590b57dc7c1SPaul Blakey static __net_init int ct_init_net(struct net *net) 1591b57dc7c1SPaul Blakey { 1592c593642cSPankaj Bharadiya unsigned int n_bits = sizeof_field(struct tcf_ct_params, labels) * 8; 1593acd0a7abSZhengchao Shao struct tc_ct_action_net *tn = net_generic(net, act_ct_ops.net_id); 1594b57dc7c1SPaul Blakey 1595b57dc7c1SPaul Blakey if (nf_connlabels_get(net, n_bits - 1)) { 1596b57dc7c1SPaul Blakey tn->labels = false; 1597b57dc7c1SPaul Blakey pr_err("act_ct: Failed to set connlabels length"); 1598b57dc7c1SPaul Blakey } else { 1599b57dc7c1SPaul Blakey tn->labels = true; 1600b57dc7c1SPaul Blakey } 1601b57dc7c1SPaul Blakey 1602981471bdSCong Wang return tc_action_net_init(net, &tn->tn, &act_ct_ops); 1603b57dc7c1SPaul Blakey } 1604b57dc7c1SPaul Blakey 1605b57dc7c1SPaul Blakey static void __net_exit ct_exit_net(struct list_head *net_list) 1606b57dc7c1SPaul Blakey { 1607b57dc7c1SPaul Blakey struct net *net; 1608b57dc7c1SPaul Blakey 1609b57dc7c1SPaul Blakey rtnl_lock(); 1610b57dc7c1SPaul Blakey list_for_each_entry(net, net_list, exit_list) { 1611acd0a7abSZhengchao Shao struct tc_ct_action_net *tn = net_generic(net, act_ct_ops.net_id); 1612b57dc7c1SPaul Blakey 1613b57dc7c1SPaul Blakey if (tn->labels) 1614b57dc7c1SPaul Blakey nf_connlabels_put(net); 1615b57dc7c1SPaul Blakey } 1616b57dc7c1SPaul Blakey rtnl_unlock(); 1617b57dc7c1SPaul Blakey 1618acd0a7abSZhengchao Shao tc_action_net_exit(net_list, act_ct_ops.net_id); 1619b57dc7c1SPaul Blakey } 1620b57dc7c1SPaul Blakey 1621b57dc7c1SPaul Blakey static struct pernet_operations ct_net_ops = { 1622b57dc7c1SPaul Blakey .init = ct_init_net, 1623b57dc7c1SPaul Blakey .exit_batch = ct_exit_net, 1624acd0a7abSZhengchao Shao .id = &act_ct_ops.net_id, 1625b57dc7c1SPaul Blakey .size = sizeof(struct tc_ct_action_net), 1626b57dc7c1SPaul Blakey }; 1627b57dc7c1SPaul Blakey 1628b57dc7c1SPaul Blakey static int __init ct_init_module(void) 1629b57dc7c1SPaul Blakey { 1630c34b961aSPaul Blakey int err; 1631c34b961aSPaul Blakey 1632c34b961aSPaul Blakey act_ct_wq = alloc_ordered_workqueue("act_ct_workqueue", 0); 1633c34b961aSPaul Blakey if (!act_ct_wq) 1634c34b961aSPaul Blakey return -ENOMEM; 1635c34b961aSPaul Blakey 1636c34b961aSPaul Blakey err = tcf_ct_flow_tables_init(); 1637c34b961aSPaul Blakey if (err) 1638c34b961aSPaul Blakey goto err_tbl_init; 1639c34b961aSPaul Blakey 1640c34b961aSPaul Blakey err = tcf_register_action(&act_ct_ops, &ct_net_ops); 1641c34b961aSPaul Blakey if (err) 1642c34b961aSPaul Blakey goto err_register; 1643c34b961aSPaul Blakey 1644c129412fSwenxu static_branch_inc(&tcf_frag_xmit_count); 1645c129412fSwenxu 1646c34b961aSPaul Blakey return 0; 1647c34b961aSPaul Blakey 1648c34b961aSPaul Blakey err_register: 1649c34b961aSPaul Blakey tcf_ct_flow_tables_uninit(); 16508c5c51f5Sliujian err_tbl_init: 16518c5c51f5Sliujian destroy_workqueue(act_ct_wq); 1652c34b961aSPaul Blakey return err; 1653b57dc7c1SPaul Blakey } 1654b57dc7c1SPaul Blakey 1655b57dc7c1SPaul Blakey static void __exit ct_cleanup_module(void) 1656b57dc7c1SPaul Blakey { 1657c129412fSwenxu static_branch_dec(&tcf_frag_xmit_count); 1658b57dc7c1SPaul Blakey tcf_unregister_action(&act_ct_ops, &ct_net_ops); 1659c34b961aSPaul Blakey tcf_ct_flow_tables_uninit(); 1660c34b961aSPaul Blakey destroy_workqueue(act_ct_wq); 1661b57dc7c1SPaul Blakey } 1662b57dc7c1SPaul Blakey 1663b57dc7c1SPaul Blakey module_init(ct_init_module); 1664b57dc7c1SPaul Blakey module_exit(ct_cleanup_module); 1665b57dc7c1SPaul Blakey MODULE_AUTHOR("Paul Blakey <paulb@mellanox.com>"); 1666b57dc7c1SPaul Blakey MODULE_AUTHOR("Yossi Kuperman <yossiku@mellanox.com>"); 1667b57dc7c1SPaul Blakey MODULE_AUTHOR("Marcelo Ricardo Leitner <marcelo.leitner@gmail.com>"); 1668b57dc7c1SPaul Blakey MODULE_DESCRIPTION("Connection tracking action"); 1669b57dc7c1SPaul Blakey MODULE_LICENSE("GPL v2"); 1670