netfilter: conntrack: move extension sizes into core
authorFlorian Westphal <fw@strlen.de>
Thu, 20 Jan 2022 12:07:00 +0000 (13:07 +0100)
committerPablo Neira Ayuso <pablo@netfilter.org>
Fri, 4 Feb 2022 05:30:28 +0000 (06:30 +0100)
No need to specify this in the registration modules, we already
collect all sizes for build-time checks on the maximum combined size.

After this change, all extensions except nat have no meaningful content
in their nf_ct_ext_type struct definition.

Next patch handles nat, this will then allow to remove the dynamic
register api completely.

Signed-off-by: Florian Westphal <fw@strlen.de>
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
13 files changed:
include/net/netfilter/nf_conntrack_extend.h
net/netfilter/nf_conntrack_acct.c
net/netfilter/nf_conntrack_core.c
net/netfilter/nf_conntrack_ecache.c
net/netfilter/nf_conntrack_extend.c
net/netfilter/nf_conntrack_helper.c
net/netfilter/nf_conntrack_labels.c
net/netfilter/nf_conntrack_seqadj.c
net/netfilter/nf_conntrack_timeout.c
net/netfilter/nf_conntrack_timestamp.c
net/netfilter/nf_nat_core.c
net/netfilter/nf_synproxy_core.c
net/sched/act_ct.c

index 705a4487f023326c7f37d1cc1f1e2e7391481f55..87d818414092a1fe28ccaa69c4541cfe82585c79 100644 (file)
@@ -83,7 +83,6 @@ struct nf_ct_ext_type {
        void (*destroy)(struct nf_conn *ct);
 
        enum nf_ct_ext_id id;
-       u8 len;
 };
 
 int nf_ct_extend_register(const struct nf_ct_ext_type *type);
index c9b20b86711ca24d56d8fe20da0643c565ab83c9..4b5048ee84f29c9062c72dd6dcbd570821688edb 100644 (file)
@@ -23,7 +23,6 @@ module_param_named(acct, nf_ct_acct, bool, 0644);
 MODULE_PARM_DESC(acct, "Enable connection tracking flow accounting.");
 
 static const struct nf_ct_ext_type acct_extend = {
-       .len    = sizeof(struct nf_conn_acct),
        .id     = NF_CT_EXT_ACCT,
 };
 
index d38d689de23c8b688cbdb588ad608b96ed76dc9f..9edd3ae8d62edf56185e93a15a709b39692e4368 100644 (file)
@@ -48,7 +48,6 @@
 #include <net/netfilter/nf_conntrack_timeout.h>
 #include <net/netfilter/nf_conntrack_labels.h>
 #include <net/netfilter/nf_conntrack_synproxy.h>
-#include <net/netfilter/nf_conntrack_act_ct.h>
 #include <net/netfilter/nf_nat.h>
 #include <net/netfilter/nf_nat_helper.h>
 #include <net/netns/hash.h>
@@ -2629,39 +2628,6 @@ int nf_conntrack_set_hashsize(const char *val, const struct kernel_param *kp)
        return nf_conntrack_hash_resize(hashsize);
 }
 
-static __always_inline unsigned int total_extension_size(void)
-{
-       /* remember to add new extensions below */
-       BUILD_BUG_ON(NF_CT_EXT_NUM > 10);
-
-       return sizeof(struct nf_ct_ext) +
-              sizeof(struct nf_conn_help)
-#if IS_ENABLED(CONFIG_NF_NAT)
-               + sizeof(struct nf_conn_nat)
-#endif
-               + sizeof(struct nf_conn_seqadj)
-               + sizeof(struct nf_conn_acct)
-#ifdef CONFIG_NF_CONNTRACK_EVENTS
-               + sizeof(struct nf_conntrack_ecache)
-#endif
-#ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
-               + sizeof(struct nf_conn_tstamp)
-#endif
-#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
-               + sizeof(struct nf_conn_timeout)
-#endif
-#ifdef CONFIG_NF_CONNTRACK_LABELS
-               + sizeof(struct nf_conn_labels)
-#endif
-#if IS_ENABLED(CONFIG_NETFILTER_SYNPROXY)
-               + sizeof(struct nf_conn_synproxy)
-#endif
-#if IS_ENABLED(CONFIG_NET_ACT_CT)
-               + sizeof(struct nf_conn_act_ct_ext)
-#endif
-       ;
-};
-
 int nf_conntrack_init_start(void)
 {
        unsigned long nr_pages = totalram_pages();
@@ -2669,9 +2635,6 @@ int nf_conntrack_init_start(void)
        int ret = -ENOMEM;
        int i;
 
-       /* struct nf_ct_ext uses u8 to store offsets/size */
-       BUILD_BUG_ON(total_extension_size() > 255u);
-
        seqcount_spinlock_init(&nf_conntrack_generation,
                               &nf_conntrack_locks_all_lock);
 
index 1cf2c8cd6a4a695df747937edac6a4b358a36ab7..9ececc9b45f985385544765037fa520f16775417 100644 (file)
@@ -305,7 +305,6 @@ void nf_conntrack_ecache_work(struct net *net, enum nf_ct_ecache_state state)
 static int nf_ct_events __read_mostly = NF_CT_EVENTS_DEFAULT;
 
 static const struct nf_ct_ext_type event_extend = {
-       .len    = sizeof(struct nf_conntrack_ecache),
        .id     = NF_CT_EXT_ECACHE,
 };
 
index c62f477c6533b401fcb18bf09e6dc93a50fb9579..69a6cafcb045c04c2cfac7bababfeb00be6fcace 100644 (file)
 #include <linux/skbuff.h>
 #include <net/netfilter/nf_conntrack_extend.h>
 
+#include <net/netfilter/nf_conntrack_helper.h>
+#include <net/netfilter/nf_conntrack_acct.h>
+#include <net/netfilter/nf_conntrack_seqadj.h>
+#include <net/netfilter/nf_conntrack_ecache.h>
+#include <net/netfilter/nf_conntrack_zones.h>
+#include <net/netfilter/nf_conntrack_timestamp.h>
+#include <net/netfilter/nf_conntrack_timeout.h>
+#include <net/netfilter/nf_conntrack_labels.h>
+#include <net/netfilter/nf_conntrack_synproxy.h>
+#include <net/netfilter/nf_conntrack_act_ct.h>
+#include <net/netfilter/nf_nat.h>
+
 static struct nf_ct_ext_type __rcu *nf_ct_ext_types[NF_CT_EXT_NUM];
 static DEFINE_MUTEX(nf_ct_ext_type_mutex);
 #define NF_CT_EXT_PREALLOC     128u /* conntrack events are on by default */
 
+static const u8 nf_ct_ext_type_len[NF_CT_EXT_NUM] = {
+       [NF_CT_EXT_HELPER] = sizeof(struct nf_conn_help),
+#if IS_ENABLED(CONFIG_NF_NAT)
+       [NF_CT_EXT_NAT] = sizeof(struct nf_conn_nat),
+#endif
+       [NF_CT_EXT_SEQADJ] = sizeof(struct nf_conn_seqadj),
+       [NF_CT_EXT_ACCT] = sizeof(struct nf_conn_acct),
+#ifdef CONFIG_NF_CONNTRACK_EVENTS
+       [NF_CT_EXT_ECACHE] = sizeof(struct nf_conntrack_ecache),
+#endif
+#ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
+       [NF_CT_EXT_TSTAMP] = sizeof(struct nf_conn_acct),
+#endif
+#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
+       [NF_CT_EXT_TIMEOUT] = sizeof(struct nf_conn_tstamp),
+#endif
+#ifdef CONFIG_NF_CONNTRACK_LABELS
+       [NF_CT_EXT_LABELS] = sizeof(struct nf_conn_labels),
+#endif
+#if IS_ENABLED(CONFIG_NETFILTER_SYNPROXY)
+       [NF_CT_EXT_SYNPROXY] = sizeof(struct nf_conn_synproxy),
+#endif
+#if IS_ENABLED(CONFIG_NET_ACT_CT)
+       [NF_CT_EXT_ACT_CT] = sizeof(struct nf_conn_act_ct_ext),
+#endif
+};
+
+static __always_inline unsigned int total_extension_size(void)
+{
+       /* remember to add new extensions below */
+       BUILD_BUG_ON(NF_CT_EXT_NUM > 10);
+
+       return sizeof(struct nf_ct_ext) +
+              sizeof(struct nf_conn_help)
+#if IS_ENABLED(CONFIG_NF_NAT)
+               + sizeof(struct nf_conn_nat)
+#endif
+               + sizeof(struct nf_conn_seqadj)
+               + sizeof(struct nf_conn_acct)
+#ifdef CONFIG_NF_CONNTRACK_EVENTS
+               + sizeof(struct nf_conntrack_ecache)
+#endif
+#ifdef CONFIG_NF_CONNTRACK_TIMESTAMP
+               + sizeof(struct nf_conn_tstamp)
+#endif
+#ifdef CONFIG_NF_CONNTRACK_TIMEOUT
+               + sizeof(struct nf_conn_timeout)
+#endif
+#ifdef CONFIG_NF_CONNTRACK_LABELS
+               + sizeof(struct nf_conn_labels)
+#endif
+#if IS_ENABLED(CONFIG_NETFILTER_SYNPROXY)
+               + sizeof(struct nf_conn_synproxy)
+#endif
+#if IS_ENABLED(CONFIG_NET_ACT_CT)
+               + sizeof(struct nf_conn_act_ct_ext)
+#endif
+       ;
+}
+
 void nf_ct_ext_destroy(struct nf_conn *ct)
 {
        unsigned int i;
@@ -41,7 +113,6 @@ void nf_ct_ext_destroy(struct nf_conn *ct)
 void *nf_ct_ext_add(struct nf_conn *ct, enum nf_ct_ext_id id, gfp_t gfp)
 {
        unsigned int newlen, newoff, oldlen, alloc;
-       struct nf_ct_ext_type *t;
        struct nf_ct_ext *new;
 
        /* Conntrack must not be confirmed to avoid races on reallocation. */
@@ -58,16 +129,8 @@ void *nf_ct_ext_add(struct nf_conn *ct, enum nf_ct_ext_id id, gfp_t gfp)
                oldlen = sizeof(*new);
        }
 
-       rcu_read_lock();
-       t = rcu_dereference(nf_ct_ext_types[id]);
-       if (!t) {
-               rcu_read_unlock();
-               return NULL;
-       }
-
        newoff = ALIGN(oldlen, __alignof__(struct nf_ct_ext));
-       newlen = newoff + t->len;
-       rcu_read_unlock();
+       newlen = newoff + nf_ct_ext_type_len[id];
 
        alloc = max(newlen, NF_CT_EXT_PREALLOC);
        new = krealloc(ct->ext, alloc, gfp);
@@ -91,6 +154,9 @@ int nf_ct_extend_register(const struct nf_ct_ext_type *type)
 {
        int ret = 0;
 
+       /* struct nf_ct_ext uses u8 to store offsets/size */
+       BUILD_BUG_ON(total_extension_size() > 255u);
+
        mutex_lock(&nf_ct_ext_type_mutex);
        if (nf_ct_ext_types[type->id]) {
                ret = -EBUSY;
index e8f6a389bd017c0f2e7cc2317875f7db5ffa1feb..6fe94f18a4ac93ab3893be08e1f0af37df8c55d4 100644 (file)
@@ -551,7 +551,6 @@ void nf_nat_helper_unregister(struct nf_conntrack_nat_helper *nat)
 EXPORT_SYMBOL_GPL(nf_nat_helper_unregister);
 
 static const struct nf_ct_ext_type helper_extend = {
-       .len    = sizeof(struct nf_conn_help),
        .id     = NF_CT_EXT_HELPER,
 };
 
index 6323358dbe73d8e9097d9f517cc775e3b56363e8..0cd99535122bd685aa21b3837ec4fc91d334bc2b 100644 (file)
@@ -80,7 +80,6 @@ void nf_connlabels_put(struct net *net)
 EXPORT_SYMBOL_GPL(nf_connlabels_put);
 
 static const struct nf_ct_ext_type labels_extend = {
-       .len    = sizeof(struct nf_conn_labels),
        .id     = NF_CT_EXT_LABELS,
 };
 
index b13b3a8a1082ac2e77ef31ef78fec72a4c95d7f1..b9629916e53d9543413fcd39f6da28c6b3ec1b56 100644 (file)
@@ -234,7 +234,6 @@ s32 nf_ct_seq_offset(const struct nf_conn *ct,
 EXPORT_SYMBOL_GPL(nf_ct_seq_offset);
 
 static const struct nf_ct_ext_type nf_ct_seqadj_extend = {
-       .len    = sizeof(struct nf_conn_seqadj),
        .id     = NF_CT_EXT_SEQADJ,
 };
 
index 816fe680375d6882cba77afd9f182e54039aa649..ac99a0083156eb7e6dbf5bf1cabcce5561654867 100644 (file)
@@ -136,7 +136,6 @@ void nf_ct_destroy_timeout(struct nf_conn *ct)
 EXPORT_SYMBOL_GPL(nf_ct_destroy_timeout);
 
 static const struct nf_ct_ext_type timeout_extend = {
-       .len    = sizeof(struct nf_conn_timeout),
        .id     = NF_CT_EXT_TIMEOUT,
 };
 
index 81878d9786ba6d469ddfa2ed3e661db91eec0d2c..c696ca19dcb10371338f4428170d26dcf528379f 100644 (file)
@@ -20,7 +20,6 @@ module_param_named(tstamp, nf_ct_tstamp, bool, 0644);
 MODULE_PARM_DESC(tstamp, "Enable connection tracking flow timestamping.");
 
 static const struct nf_ct_ext_type tstamp_extend = {
-       .len    = sizeof(struct nf_conn_tstamp),
        .id     = NF_CT_EXT_TSTAMP,
 };
 
index d0000f63b0af99052894970a8d1f7a32864dbb84..2ff20d6a5afb417ce59d96ec843f646b99c4d87e 100644 (file)
@@ -876,7 +876,6 @@ static void nf_nat_cleanup_conntrack(struct nf_conn *ct)
 }
 
 static struct nf_ct_ext_type nat_extend __read_mostly = {
-       .len            = sizeof(struct nf_conn_nat),
        .destroy        = nf_nat_cleanup_conntrack,
        .id             = NF_CT_EXT_NAT,
 };
index d5c1e93c4ba3dcbb4da05d6a5daa9c8882729f1b..6d328f7bb32388283ab2cbe86b102b93c1b97988 100644 (file)
@@ -237,7 +237,6 @@ synproxy_tstamp_adjust(struct sk_buff *skb, unsigned int protoff,
 }
 
 static struct nf_ct_ext_type nf_ct_synproxy_extend __read_mostly = {
-       .len            = sizeof(struct nf_conn_synproxy),
        .id             = NF_CT_EXT_SYNPROXY,
 };
 
index 78ccd16be05e02e0dc08c8b9782e2a7ac08f8230..774e32fab5cfb66ddd54ca74816ef25f3231b0d3 100644 (file)
@@ -58,7 +58,6 @@ static const struct rhashtable_params zones_params = {
 };
 
 static struct nf_ct_ext_type act_ct_extend __read_mostly = {
-       .len            = sizeof(struct nf_conn_act_ct_ext),
        .id             = NF_CT_EXT_ACT_CT,
 };