|
@@ -42,8 +42,10 @@
|
|
|
|
|
|
static void batadv_gw_node_free_ref(struct batadv_gw_node *gw_node)
|
|
|
{
|
|
|
- if (atomic_dec_and_test(&gw_node->refcount))
|
|
|
+ if (atomic_dec_and_test(&gw_node->refcount)) {
|
|
|
+ batadv_orig_node_free_ref(gw_node->orig_node);
|
|
|
kfree_rcu(gw_node, rcu);
|
|
|
+ }
|
|
|
}
|
|
|
|
|
|
static struct batadv_gw_node *
|
|
@@ -406,9 +408,14 @@ static void batadv_gw_node_add(struct batadv_priv *bat_priv,
|
|
|
if (gateway->bandwidth_down == 0)
|
|
|
return;
|
|
|
|
|
|
+ if (!atomic_inc_not_zero(&orig_node->refcount))
|
|
|
+ return;
|
|
|
+
|
|
|
gw_node = kzalloc(sizeof(*gw_node), GFP_ATOMIC);
|
|
|
- if (!gw_node)
|
|
|
+ if (!gw_node) {
|
|
|
+ batadv_orig_node_free_ref(orig_node);
|
|
|
return;
|
|
|
+ }
|
|
|
|
|
|
INIT_HLIST_NODE(&gw_node->list);
|
|
|
gw_node->orig_node = orig_node;
|