inetpeer: Add v6 peers tree, abstract root properly.
Add the ipv6 peer tree instance, and adapt remaining
direct references to 'v4_peers' as needed.
Signed-off-by: David S. Miller <davem@davemloft.net>
diff --git a/net/ipv4/inetpeer.c b/net/ipv4/inetpeer.c
index c96dc51..1c1335b 100644
--- a/net/ipv4/inetpeer.c
+++ b/net/ipv4/inetpeer.c
@@ -79,15 +79,24 @@
.avl_height = 0
};
-static struct inet_peer_base {
+struct inet_peer_base {
struct inet_peer __rcu *root;
spinlock_t lock;
int total;
-} v4_peers = {
+};
+
+static struct inet_peer_base v4_peers = {
.root = peer_avl_empty_rcu,
.lock = __SPIN_LOCK_UNLOCKED(v4_peers.lock),
.total = 0,
};
+
+static struct inet_peer_base v6_peers = {
+ .root = peer_avl_empty_rcu,
+ .lock = __SPIN_LOCK_UNLOCKED(v6_peers.lock),
+ .total = 0,
+};
+
#define PEER_MAXDEPTH 40 /* sufficient for about 2^27 nodes */
/* Exported for sysctl_net_ipv4. */
@@ -415,9 +424,14 @@
inet_putpeer(p);
}
+static struct inet_peer_base *family_to_base(int family)
+{
+ return (family == AF_INET ? &v4_peers : &v6_peers);
+}
+
static struct inet_peer_base *peer_to_base(struct inet_peer *p)
{
- return &v4_peers;
+ return family_to_base(p->daddr.family);
}
/* May be called with local BH enabled. */
@@ -457,11 +471,6 @@
return 0;
}
-static struct inet_peer_base *family_to_base(int family)
-{
- return &v4_peers;
-}
-
/* Called with or without local BH being disabled. */
struct inet_peer *inet_getpeer(inet_peer_address_t *daddr, int create)
{
@@ -521,7 +530,7 @@
static int compute_total(void)
{
- return v4_peers.total;
+ return v4_peers.total + v6_peers.total;
}
/* Called with local BH disabled. */