2 * INETPEER - A storage for permanent information about peers
4 * Authors: Andrey V. Savochkin <saw@msu.ru>
7 #ifndef _NET_INETPEER_H
8 #define _NET_INETPEER_H
10 #include <linux/types.h>
11 #include <linux/init.h>
12 #include <linux/jiffies.h>
13 #include <linux/spinlock.h>
14 #include <linux/rtnetlink.h>
16 #include <linux/atomic.h>
18 struct inetpeer_addr_base
{
25 struct inetpeer_addr
{
26 struct inetpeer_addr_base addr
;
31 /* group together avl_left,avl_right,v4daddr to speedup lookups */
32 struct inet_peer __rcu
*avl_left
, *avl_right
;
33 struct inetpeer_addr daddr
;
36 u32 metrics
[RTAX_MAX
];
37 u32 rate_tokens
; /* rate limiting for ICMP */
38 unsigned long rate_last
;
40 struct list_head gc_list
;
41 struct rcu_head gc_rcu
;
44 * Once inet_peer is queued for deletion (refcnt == -1), following field
45 * is not available: rid
46 * We can share memory with rcu_head to help keep inet_peer small.
50 atomic_t rid
; /* Frag reception counter */
53 struct inet_peer
*gc_next
;
56 /* following fields might be frequently dirtied */
57 __u32 dtime
; /* the time of last use of not referenced entries */
61 struct inet_peer_base
{
62 struct inet_peer __rcu
*root
;
68 #define INETPEER_BASE_BIT 0x1UL
70 static inline struct inet_peer
*inetpeer_ptr(unsigned long val
)
72 BUG_ON(val
& INETPEER_BASE_BIT
);
73 return (struct inet_peer
*) val
;
76 static inline struct inet_peer_base
*inetpeer_base_ptr(unsigned long val
)
78 if (!(val
& INETPEER_BASE_BIT
))
80 val
&= ~INETPEER_BASE_BIT
;
81 return (struct inet_peer_base
*) val
;
84 static inline bool inetpeer_ptr_is_peer(unsigned long val
)
86 return !(val
& INETPEER_BASE_BIT
);
89 static inline void __inetpeer_ptr_set_peer(unsigned long *val
, struct inet_peer
*peer
)
91 /* This implicitly clears INETPEER_BASE_BIT */
92 *val
= (unsigned long) peer
;
95 static inline bool inetpeer_ptr_set_peer(unsigned long *ptr
, struct inet_peer
*peer
)
97 unsigned long val
= (unsigned long) peer
;
98 unsigned long orig
= *ptr
;
100 if (!(orig
& INETPEER_BASE_BIT
) ||
101 cmpxchg(ptr
, orig
, val
) != orig
)
106 static inline void inetpeer_init_ptr(unsigned long *ptr
, struct inet_peer_base
*base
)
108 *ptr
= (unsigned long) base
| INETPEER_BASE_BIT
;
111 static inline void inetpeer_transfer_peer(unsigned long *to
, unsigned long *from
)
113 unsigned long val
= *from
;
116 if (inetpeer_ptr_is_peer(val
)) {
117 struct inet_peer
*peer
= inetpeer_ptr(val
);
118 atomic_inc(&peer
->refcnt
);
122 extern void inet_peer_base_init(struct inet_peer_base
*);
124 void inet_initpeers(void) __init
;
126 #define INETPEER_METRICS_NEW (~(u32) 0)
128 static inline bool inet_metrics_new(const struct inet_peer
*p
)
130 return p
->metrics
[RTAX_LOCK
-1] == INETPEER_METRICS_NEW
;
133 /* can be called with or without local BH being disabled */
134 struct inet_peer
*inet_getpeer(struct inet_peer_base
*base
,
135 const struct inetpeer_addr
*daddr
,
138 static inline struct inet_peer
*inet_getpeer_v4(struct inet_peer_base
*base
,
142 struct inetpeer_addr daddr
;
144 daddr
.addr
.a4
= v4daddr
;
145 daddr
.family
= AF_INET
;
146 return inet_getpeer(base
, &daddr
, create
);
149 static inline struct inet_peer
*inet_getpeer_v6(struct inet_peer_base
*base
,
150 const struct in6_addr
*v6daddr
,
153 struct inetpeer_addr daddr
;
155 *(struct in6_addr
*)daddr
.addr
.a6
= *v6daddr
;
156 daddr
.family
= AF_INET6
;
157 return inet_getpeer(base
, &daddr
, create
);
160 /* can be called from BH context or outside */
161 extern void inet_putpeer(struct inet_peer
*p
);
162 extern bool inet_peer_xrlim_allow(struct inet_peer
*peer
, int timeout
);
164 extern void inetpeer_invalidate_tree(struct inet_peer_base
*);
165 extern void inetpeer_invalidate_family(int family
);
168 * temporary check to make sure we dont access rid, tcp_ts,
169 * tcp_ts_stamp if no refcount is taken on inet_peer
171 static inline void inet_peer_refcheck(const struct inet_peer
*p
)
173 WARN_ON_ONCE(atomic_read(&p
->refcnt
) <= 0);
176 #endif /* _NET_INETPEER_H */