|
@@ -83,21 +83,29 @@ assign:
|
|
|
|
|
|
static int ops_init(const struct pernet_operations *ops, struct net *net)
|
|
static int ops_init(const struct pernet_operations *ops, struct net *net)
|
|
{
|
|
{
|
|
- int err;
|
|
|
|
|
|
+ int err = -ENOMEM;
|
|
|
|
+ void *data = NULL;
|
|
|
|
+
|
|
if (ops->id && ops->size) {
|
|
if (ops->id && ops->size) {
|
|
- void *data = kzalloc(ops->size, GFP_KERNEL);
|
|
|
|
|
|
+ data = kzalloc(ops->size, GFP_KERNEL);
|
|
if (!data)
|
|
if (!data)
|
|
- return -ENOMEM;
|
|
|
|
|
|
+ goto out;
|
|
|
|
|
|
err = net_assign_generic(net, *ops->id, data);
|
|
err = net_assign_generic(net, *ops->id, data);
|
|
- if (err) {
|
|
|
|
- kfree(data);
|
|
|
|
- return err;
|
|
|
|
- }
|
|
|
|
|
|
+ if (err)
|
|
|
|
+ goto cleanup;
|
|
}
|
|
}
|
|
|
|
+ err = 0;
|
|
if (ops->init)
|
|
if (ops->init)
|
|
- return ops->init(net);
|
|
|
|
- return 0;
|
|
|
|
|
|
+ err = ops->init(net);
|
|
|
|
+ if (!err)
|
|
|
|
+ return 0;
|
|
|
|
+
|
|
|
|
+cleanup:
|
|
|
|
+ kfree(data);
|
|
|
|
+
|
|
|
|
+out:
|
|
|
|
+ return err;
|
|
}
|
|
}
|
|
|
|
|
|
static void ops_free(const struct pernet_operations *ops, struct net *net)
|
|
static void ops_free(const struct pernet_operations *ops, struct net *net)
|
|
@@ -448,12 +456,7 @@ static void __unregister_pernet_operations(struct pernet_operations *ops)
|
|
static int __register_pernet_operations(struct list_head *list,
|
|
static int __register_pernet_operations(struct list_head *list,
|
|
struct pernet_operations *ops)
|
|
struct pernet_operations *ops)
|
|
{
|
|
{
|
|
- int err = 0;
|
|
|
|
- err = ops_init(ops, &init_net);
|
|
|
|
- if (err)
|
|
|
|
- ops_free(ops, &init_net);
|
|
|
|
- return err;
|
|
|
|
-
|
|
|
|
|
|
+ return ops_init(ops, &init_net);
|
|
}
|
|
}
|
|
|
|
|
|
static void __unregister_pernet_operations(struct pernet_operations *ops)
|
|
static void __unregister_pernet_operations(struct pernet_operations *ops)
|