1 /* 2 * INET An implementation of the TCP/IP protocol suite for the LINUX 3 * operating system. INET is implemented using the BSD Socket 4 * interface as the means of communication with the user level. 5 * 6 * Authors: Lotsa people, from code originally in tcp 7 * 8 * This program is free software; you can redistribute it and/or 9 * modify it under the terms of the GNU General Public License 10 * as published by the Free Software Foundation; either version 11 * 2 of the License, or (at your option) any later version. 12 */ 13 14 #ifndef _INET6_HASHTABLES_H 15 #define _INET6_HASHTABLES_H 16 17 18 #if IS_ENABLED(CONFIG_IPV6) 19 #include <linux/in6.h> 20 #include <linux/ipv6.h> 21 #include <linux/types.h> 22 #include <linux/jhash.h> 23 24 #include <net/inet_sock.h> 25 26 #include <net/ipv6.h> 27 #include <net/netns/hash.h> 28 29 struct inet_hashinfo; 30 31 static inline unsigned int __inet6_ehashfn(const u32 lhash, 32 const u16 lport, 33 const u32 fhash, 34 const __be16 fport, 35 const u32 initval) 36 { 37 const u32 ports = (((u32)lport) << 16) | (__force u32)fport; 38 return jhash_3words(lhash, fhash, ports, initval); 39 } 40 41 /* 42 * Sockets in TCP_CLOSE state are _always_ taken out of the hash, so 43 * we need not check it for TCP lookups anymore, thanks Alexey. -DaveM 44 * 45 * The sockhash lock must be held as a reader here. 46 */ 47 struct sock *__inet6_lookup_established(struct net *net, 48 struct inet_hashinfo *hashinfo, 49 const struct in6_addr *saddr, 50 const __be16 sport, 51 const struct in6_addr *daddr, 52 const u16 hnum, const int dif, 53 const int sdif); 54 55 struct sock *inet6_lookup_listener(struct net *net, 56 struct inet_hashinfo *hashinfo, 57 struct sk_buff *skb, int doff, 58 const struct in6_addr *saddr, 59 const __be16 sport, 60 const struct in6_addr *daddr, 61 const unsigned short hnum, 62 const int dif, const int sdif); 63 64 static inline struct sock *__inet6_lookup(struct net *net, 65 struct inet_hashinfo *hashinfo, 66 struct sk_buff *skb, int doff, 67 const struct in6_addr *saddr, 68 const __be16 sport, 69 const struct in6_addr *daddr, 70 const u16 hnum, 71 const int dif, const int sdif, 72 bool *refcounted) 73 { 74 struct sock *sk = __inet6_lookup_established(net, hashinfo, saddr, 75 sport, daddr, hnum, 76 dif, sdif); 77 *refcounted = true; 78 if (sk) 79 return sk; 80 *refcounted = false; 81 return inet6_lookup_listener(net, hashinfo, skb, doff, saddr, sport, 82 daddr, hnum, dif, sdif); 83 } 84 85 static inline struct sock *__inet6_lookup_skb(struct inet_hashinfo *hashinfo, 86 struct sk_buff *skb, int doff, 87 const __be16 sport, 88 const __be16 dport, 89 int iif, int sdif, 90 bool *refcounted) 91 { 92 struct sock *sk = skb_steal_sock(skb); 93 94 *refcounted = true; 95 if (sk) 96 return sk; 97 98 return __inet6_lookup(dev_net(skb_dst(skb)->dev), hashinfo, skb, 99 doff, &ipv6_hdr(skb)->saddr, sport, 100 &ipv6_hdr(skb)->daddr, ntohs(dport), 101 iif, sdif, refcounted); 102 } 103 104 struct sock *inet6_lookup(struct net *net, struct inet_hashinfo *hashinfo, 105 struct sk_buff *skb, int doff, 106 const struct in6_addr *saddr, const __be16 sport, 107 const struct in6_addr *daddr, const __be16 dport, 108 const int dif); 109 110 int inet6_hash(struct sock *sk); 111 #endif /* IS_ENABLED(CONFIG_IPV6) */ 112 113 #define INET6_MATCH(__sk, __net, __saddr, __daddr, __ports, __dif, __sdif) \ 114 (((__sk)->sk_portpair == (__ports)) && \ 115 ((__sk)->sk_family == AF_INET6) && \ 116 ipv6_addr_equal(&(__sk)->sk_v6_daddr, (__saddr)) && \ 117 ipv6_addr_equal(&(__sk)->sk_v6_rcv_saddr, (__daddr)) && \ 118 (!(__sk)->sk_bound_dev_if || \ 119 ((__sk)->sk_bound_dev_if == (__dif)) || \ 120 ((__sk)->sk_bound_dev_if == (__sdif))) && \ 121 net_eq(sock_net(__sk), (__net))) 122 123 #endif /* _INET6_HASHTABLES_H */ 124