Merge git://git.kernel.org/pub/scm/linux/kernel/git/davem/net

[linux-2.6-microblaze.git] / net / ipv4 / inetpeer.c

diff --git a/net/ipv4/inetpeer.c b/net/ipv4/inetpeer.c
index b20c8ac..914d569 100644 (file)
--- a/net/ipv4/inetpeer.c
+++ b/net/ipv4/inetpeer.c
@@ -284,14 +284,17 @@ EXPORT_SYMBOL(inet_peer_xrlim_allow);
 
 void inetpeer_invalidate_tree(struct inet_peer_base *base)
 {
-       struct inet_peer *p, *n;
+       struct rb_node *p = rb_first(&base->rb_root);
 
-       rbtree_postorder_for_each_entry_safe(p, n, &base->rb_root, rb_node) {
-               inet_putpeer(p);
+       while (p) {
+               struct inet_peer *peer = rb_entry(p, struct inet_peer, rb_node);
+
+               p = rb_next(p);
+               rb_erase(&peer->rb_node, &base->rb_root);
+               inet_putpeer(peer);
                cond_resched();
        }
 
-       base->rb_root = RB_ROOT;
        base->total = 0;
 }
 EXPORT_SYMBOL(inetpeer_invalidate_tree);