batman-adv: protect neighbor nodes with reference counters

Signed-off-by: Marek Lindner <lindner_marek@yahoo.de>

1 files changed, 7 insertions, 0 deletions

diff --git a/net/batman-adv/routing.c b/net/batman-adv/routing.c
index 827414067e46..36351d374f28 100644
--- a/net/batman-adv/routing.c
+++ b/net/batman-adv/routing.c
@@ -89,6 +89,8 @@ static void update_route(struct bat_priv *bat_priv,
                          struct neigh_node *neigh_node,
                          unsigned char *hna_buff, int hna_buff_len)
 {
+        struct neigh_node *neigh_node_tmp;
+
         /* route deleted */
         if ((orig_node->router) && (!neigh_node)) {
 
@@ -115,7 +117,12 @@ static void update_route(struct bat_priv *bat_priv,
                         orig_node->router->addr);
         }
 
+        if (neigh_node)
+                kref_get(&neigh_node->refcount);
+        neigh_node_tmp = orig_node->router;
         orig_node->router = neigh_node;
+        if (neigh_node_tmp)
+                kref_put(&neigh_node_tmp->refcount, neigh_node_free_ref);
 }