Merge tag 'xfs-4.20-merge-1' of git://git.kernel.org/pub/scm/fs/xfs/xfs-linux

[linux-2.6-block.git] / net / netfilter / nf_tables_core.c

diff --git a/net/netfilter/nf_tables_core.c b/net/netfilter/nf_tables_core.c
index ffd5c0f9412b4dab673b3294292a2fa6a7da404b..3fbce3b9c5ec0f51c9841aca36c0e5fc48b233c6 100644 (file)
--- a/net/netfilter/nf_tables_core.c
+++ b/net/netfilter/nf_tables_core.c
@@ -249,12 +249,24 @@ static struct nft_expr_type *nft_basic_types[] = {
        &nft_exthdr_type,
 };
 
+static struct nft_object_type *nft_basic_objects[] = {
+#ifdef CONFIG_NETWORK_SECMARK
+       &nft_secmark_obj_type,
+#endif
+};
+
 int __init nf_tables_core_module_init(void)
 {
-       int err, i;
+       int err, i, j = 0;
+
+       for (i = 0; i < ARRAY_SIZE(nft_basic_objects); i++) {
+               err = nft_register_obj(nft_basic_objects[i]);
+               if (err)
+                       goto err;
+       }
 
-       for (i = 0; i < ARRAY_SIZE(nft_basic_types); i++) {
-               err = nft_register_expr(nft_basic_types[i]);
+       for (j = 0; j < ARRAY_SIZE(nft_basic_types); j++) {
+               err = nft_register_expr(nft_basic_types[j]);
                if (err)
                        goto err;
        }
@@ -262,8 +274,12 @@ int __init nf_tables_core_module_init(void)
        return 0;
 
 err:
+       while (j-- > 0)
+               nft_unregister_expr(nft_basic_types[j]);
+
        while (i-- > 0)
-               nft_unregister_expr(nft_basic_types[i]);
+               nft_unregister_obj(nft_basic_objects[i]);
+
        return err;
 }
 
@@ -274,4 +290,8 @@ void nf_tables_core_module_exit(void)
        i = ARRAY_SIZE(nft_basic_types);
        while (i-- > 0)
                nft_unregister_expr(nft_basic_types[i]);
+
+       i = ARRAY_SIZE(nft_basic_objects);
+       while (i-- > 0)
+               nft_unregister_obj(nft_basic_objects[i]);
 }