Create udp4_lib_lookup_noref and udp6_lib_lookup_noref. These perfrom a socket lookup on addresses and ports without taking a reference.
Signed-off-by: Tom Herbert <t...@herbertland.com> --- include/net/udp.h | 8 ++++++++ net/ipv4/udp.c | 8 ++++++++ net/ipv6/udp.c | 10 ++++++++++ 3 files changed, 26 insertions(+) diff --git a/include/net/udp.h b/include/net/udp.h index ea53a87..717a972 100644 --- a/include/net/udp.h +++ b/include/net/udp.h @@ -275,6 +275,10 @@ struct sock *__udp4_lib_lookup(struct net *net, __be32 saddr, __be16 sport, struct udp_table *tbl, struct sk_buff *skb); struct sock *udp4_lib_lookup_skb(struct sk_buff *skb, __be16 sport, __be16 dport); +struct sock *udp4_lib_lookup_noref(struct net *net, + __be32 saddr, __be16 sport, + __be32 daddr, __be16 dport, + int dif); struct sock *udp6_lib_lookup(struct net *net, const struct in6_addr *saddr, __be16 sport, const struct in6_addr *daddr, __be16 dport, @@ -286,6 +290,10 @@ struct sock *__udp6_lib_lookup(struct net *net, struct sk_buff *skb); struct sock *udp6_lib_lookup_skb(struct sk_buff *skb, __be16 sport, __be16 dport); +struct sock *udp6_lib_lookup_noref(struct net *net, + const struct in6_addr *saddr, __be16 sport, + const struct in6_addr *daddr, __be16 dport, + int dif); /* * SNMP statistics for UDP and UDP-Lite diff --git a/net/ipv4/udp.c b/net/ipv4/udp.c index 7d96dc2..7f84c51 100644 --- a/net/ipv4/udp.c +++ b/net/ipv4/udp.c @@ -595,6 +595,14 @@ struct sock *udp4_lib_lookup(struct net *net, __be32 saddr, __be16 sport, EXPORT_SYMBOL_GPL(udp4_lib_lookup); #endif +struct sock *udp4_lib_lookup_noref(struct net *net, __be32 saddr, __be16 sport, + __be32 daddr, __be16 dport, int dif) +{ + return __udp4_lib_lookup(net, saddr, sport, daddr, dport, + dif, &udp_table, NULL); +} +EXPORT_SYMBOL_GPL(udp4_lib_lookup_noref); + static inline bool __udp_is_mcast_sock(struct net *net, struct sock *sk, __be16 loc_port, __be32 loc_addr, __be16 rmt_port, __be32 rmt_addr, diff --git a/net/ipv6/udp.c b/net/ipv6/udp.c index 9aa7c1c..6e382d9 100644 --- a/net/ipv6/udp.c +++ b/net/ipv6/udp.c @@ -317,6 +317,16 @@ struct sock *udp6_lib_lookup(struct net *net, const struct in6_addr *saddr, __be EXPORT_SYMBOL_GPL(udp6_lib_lookup); #endif +struct sock *udp6_lib_lookup_noref(struct net *net, + const struct in6_addr *saddr, __be16 sport, + const struct in6_addr *daddr, __be16 dport, + int dif) +{ + return __udp6_lib_lookup(net, saddr, sport, daddr, dport, + dif, &udp_table, NULL); +} +EXPORT_SYMBOL_GPL(udp6_lib_lookup_noref); + /* * This should be easy, if there is something there we * return it, otherwise we block. -- 2.9.3