Commit | Line | Data |
---|---|---|
639b321b AG |
1 | /* |
2 | * Copyright (c) 2006 Oracle. All rights reserved. | |
3 | * | |
4 | * This software is available to you under a choice of one of two | |
5 | * licenses. You may choose to be licensed under the terms of the GNU | |
6 | * General Public License (GPL) Version 2, available from the file | |
7 | * COPYING in the main directory of this source tree, or the | |
8 | * OpenIB.org BSD license below: | |
9 | * | |
10 | * Redistribution and use in source and binary forms, with or | |
11 | * without modification, are permitted provided that the following | |
12 | * conditions are met: | |
13 | * | |
14 | * - Redistributions of source code must retain the above | |
15 | * copyright notice, this list of conditions and the following | |
16 | * disclaimer. | |
17 | * | |
18 | * - Redistributions in binary form must reproduce the above | |
19 | * copyright notice, this list of conditions and the following | |
20 | * disclaimer in the documentation and/or other materials | |
21 | * provided with the distribution. | |
22 | * | |
23 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |
24 | * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |
25 | * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | |
26 | * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS | |
27 | * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN | |
28 | * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN | |
29 | * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
30 | * SOFTWARE. | |
31 | * | |
32 | */ | |
33 | #include <linux/kernel.h> | |
34 | #include <net/sock.h> | |
35 | #include <linux/in.h> | |
36 | #include <linux/if_arp.h> | |
38a4e5e6 | 37 | #include <linux/jhash.h> |
cb0a6056 | 38 | #include <linux/ratelimit.h> |
639b321b AG |
39 | #include "rds.h" |
40 | ||
9b9acde7 SS |
41 | struct bind_bucket { |
42 | rwlock_t lock; | |
43 | struct hlist_head head; | |
44 | }; | |
45 | ||
38a4e5e6 | 46 | #define BIND_HASH_SIZE 1024 |
9b9acde7 | 47 | static struct bind_bucket bind_hash_table[BIND_HASH_SIZE]; |
38a4e5e6 | 48 | |
9b9acde7 | 49 | static struct bind_bucket *hash_to_bucket(__be32 addr, __be16 port) |
38a4e5e6 CM |
50 | { |
51 | return bind_hash_table + (jhash_2words((u32)addr, (u32)port, 0) & | |
52 | (BIND_HASH_SIZE - 1)); | |
53 | } | |
639b321b | 54 | |
8b0a6b46 | 55 | /* must hold either read or write lock (write lock for insert != NULL) */ |
9b9acde7 SS |
56 | static struct rds_sock *rds_bind_lookup(struct bind_bucket *bucket, |
57 | __be32 addr, __be16 port, | |
38a4e5e6 | 58 | struct rds_sock *insert) |
639b321b | 59 | { |
639b321b | 60 | struct rds_sock *rs; |
9b9acde7 | 61 | struct hlist_head *head = &bucket->head; |
639b321b AG |
62 | u64 cmp; |
63 | u64 needle = ((u64)be32_to_cpu(addr) << 32) | be16_to_cpu(port); | |
64 | ||
8b0a6b46 | 65 | hlist_for_each_entry(rs, head, rs_bound_node) { |
639b321b AG |
66 | cmp = ((u64)be32_to_cpu(rs->rs_bound_addr) << 32) | |
67 | be16_to_cpu(rs->rs_bound_port); | |
68 | ||
28126959 SS |
69 | if (cmp == needle) { |
70 | rds_sock_addref(rs); | |
639b321b | 71 | return rs; |
28126959 | 72 | } |
639b321b AG |
73 | } |
74 | ||
75 | if (insert) { | |
38a4e5e6 CM |
76 | /* |
77 | * make sure our addr and port are set before | |
8b0a6b46 | 78 | * we are added to the list. |
38a4e5e6 CM |
79 | */ |
80 | insert->rs_bound_addr = addr; | |
81 | insert->rs_bound_port = port; | |
82 | rds_sock_addref(insert); | |
83 | ||
8b0a6b46 | 84 | hlist_add_head(&insert->rs_bound_node, head); |
639b321b AG |
85 | } |
86 | return NULL; | |
87 | } | |
88 | ||
89 | /* | |
90 | * Return the rds_sock bound at the given local address. | |
91 | * | |
92 | * The rx path can race with rds_release. We notice if rds_release() has | |
93 | * marked this socket and don't return a rs ref to the rx path. | |
94 | */ | |
95 | struct rds_sock *rds_find_bound(__be32 addr, __be16 port) | |
96 | { | |
97 | struct rds_sock *rs; | |
8b0a6b46 | 98 | unsigned long flags; |
9b9acde7 | 99 | struct bind_bucket *bucket = hash_to_bucket(addr, port); |
639b321b | 100 | |
9b9acde7 SS |
101 | read_lock_irqsave(&bucket->lock, flags); |
102 | rs = rds_bind_lookup(bucket, addr, port, NULL); | |
103 | read_unlock_irqrestore(&bucket->lock, flags); | |
38a4e5e6 | 104 | |
28126959 SS |
105 | if (rs && sock_flag(rds_rs_to_sk(rs), SOCK_DEAD)) { |
106 | rds_sock_put(rs); | |
639b321b | 107 | rs = NULL; |
28126959 | 108 | } |
639b321b AG |
109 | |
110 | rdsdebug("returning rs %p for %pI4:%u\n", rs, &addr, | |
111 | ntohs(port)); | |
8b0a6b46 | 112 | |
639b321b AG |
113 | return rs; |
114 | } | |
115 | ||
116 | /* returns -ve errno or +ve port */ | |
117 | static int rds_add_bound(struct rds_sock *rs, __be32 addr, __be16 *port) | |
118 | { | |
119 | unsigned long flags; | |
120 | int ret = -EADDRINUSE; | |
121 | u16 rover, last; | |
9b9acde7 | 122 | struct bind_bucket *bucket; |
639b321b AG |
123 | |
124 | if (*port != 0) { | |
125 | rover = be16_to_cpu(*port); | |
126 | last = rover; | |
127 | } else { | |
63862b5b | 128 | rover = max_t(u16, prandom_u32(), 2); |
639b321b AG |
129 | last = rover - 1; |
130 | } | |
131 | ||
639b321b | 132 | do { |
28126959 | 133 | struct rds_sock *rrs; |
639b321b AG |
134 | if (rover == 0) |
135 | rover++; | |
9b9acde7 SS |
136 | |
137 | bucket = hash_to_bucket(addr, cpu_to_be16(rover)); | |
138 | write_lock_irqsave(&bucket->lock, flags); | |
139 | rrs = rds_bind_lookup(bucket, addr, cpu_to_be16(rover), rs); | |
140 | write_unlock_irqrestore(&bucket->lock, flags); | |
28126959 | 141 | if (!rrs) { |
38a4e5e6 | 142 | *port = rs->rs_bound_port; |
639b321b | 143 | ret = 0; |
38a4e5e6 CM |
144 | rdsdebug("rs %p binding to %pI4:%d\n", |
145 | rs, &addr, (int)ntohs(*port)); | |
639b321b | 146 | break; |
28126959 SS |
147 | } else { |
148 | rds_sock_put(rrs); | |
639b321b AG |
149 | } |
150 | } while (rover++ != last); | |
151 | ||
639b321b AG |
152 | return ret; |
153 | } | |
154 | ||
155 | void rds_remove_bound(struct rds_sock *rs) | |
156 | { | |
157 | unsigned long flags; | |
9b9acde7 SS |
158 | struct bind_bucket *bucket = |
159 | hash_to_bucket(rs->rs_bound_addr, rs->rs_bound_port); | |
639b321b | 160 | |
9b9acde7 | 161 | write_lock_irqsave(&bucket->lock, flags); |
639b321b AG |
162 | |
163 | if (rs->rs_bound_addr) { | |
164 | rdsdebug("rs %p unbinding from %pI4:%d\n", | |
165 | rs, &rs->rs_bound_addr, | |
166 | ntohs(rs->rs_bound_port)); | |
167 | ||
8b0a6b46 | 168 | hlist_del_init(&rs->rs_bound_node); |
639b321b AG |
169 | rds_sock_put(rs); |
170 | rs->rs_bound_addr = 0; | |
171 | } | |
172 | ||
9b9acde7 | 173 | write_unlock_irqrestore(&bucket->lock, flags); |
639b321b AG |
174 | } |
175 | ||
176 | int rds_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len) | |
177 | { | |
178 | struct sock *sk = sock->sk; | |
179 | struct sockaddr_in *sin = (struct sockaddr_in *)uaddr; | |
180 | struct rds_sock *rs = rds_sk_to_rs(sk); | |
181 | struct rds_transport *trans; | |
182 | int ret = 0; | |
183 | ||
184 | lock_sock(sk); | |
185 | ||
186 | if (addr_len != sizeof(struct sockaddr_in) || | |
187 | sin->sin_family != AF_INET || | |
188 | rs->rs_bound_addr || | |
189 | sin->sin_addr.s_addr == htonl(INADDR_ANY)) { | |
190 | ret = -EINVAL; | |
191 | goto out; | |
192 | } | |
193 | ||
194 | ret = rds_add_bound(rs, sin->sin_addr.s_addr, &sin->sin_port); | |
195 | if (ret) | |
196 | goto out; | |
197 | ||
d97dac54 | 198 | if (rs->rs_transport) { /* previously bound */ |
48679800 SV |
199 | trans = rs->rs_transport; |
200 | if (trans->laddr_check(sock_net(sock->sk), | |
201 | sin->sin_addr.s_addr) != 0) { | |
202 | ret = -ENOPROTOOPT; | |
203 | rds_remove_bound(rs); | |
204 | } else { | |
205 | ret = 0; | |
206 | } | |
d97dac54 SV |
207 | goto out; |
208 | } | |
d5a8ac28 SV |
209 | trans = rds_trans_get_preferred(sock_net(sock->sk), |
210 | sin->sin_addr.s_addr); | |
8690bfa1 | 211 | if (!trans) { |
639b321b AG |
212 | ret = -EADDRNOTAVAIL; |
213 | rds_remove_bound(rs); | |
cb0a6056 | 214 | printk_ratelimited(KERN_INFO "RDS: rds_bind() could not find a transport, " |
f2c44932 | 215 | "load rds_tcp or rds_rdma?\n"); |
639b321b AG |
216 | goto out; |
217 | } | |
218 | ||
219 | rs->rs_transport = trans; | |
220 | ret = 0; | |
221 | ||
222 | out: | |
223 | release_sock(sk); | |
224 | return ret; | |
225 | } | |
9b9acde7 SS |
226 | |
227 | void rds_bind_lock_init(void) | |
228 | { | |
229 | int i; | |
230 | ||
231 | for (i = 0; i < BIND_HASH_SIZE; i++) | |
232 | rwlock_init(&bind_hash_table[i].lock); | |
233 | } |