Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 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> |
Chris Mason | 38a4e5e | 2010-05-11 15:09:45 -0700 | [diff] [blame] | 37 | #include <linux/jhash.h> |
Manuel Zerpies | cb0a605 | 2011-06-16 02:09:57 +0000 | [diff] [blame] | 38 | #include <linux/ratelimit.h> |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 39 | #include "rds.h" |
| 40 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 41 | static struct rhashtable bind_hash_table; |
| 42 | |
| 43 | static struct rhashtable_params ht_parms = { |
| 44 | .nelem_hint = 768, |
| 45 | .key_len = sizeof(u64), |
| 46 | .key_offset = offsetof(struct rds_sock, rs_bound_key), |
| 47 | .head_offset = offsetof(struct rds_sock, rs_bound_node), |
| 48 | .max_size = 16384, |
| 49 | .min_size = 1024, |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 50 | }; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 51 | |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 52 | /* |
| 53 | * Return the rds_sock bound at the given local address. |
| 54 | * |
| 55 | * The rx path can race with rds_release. We notice if rds_release() has |
| 56 | * marked this socket and don't return a rs ref to the rx path. |
| 57 | */ |
| 58 | struct rds_sock *rds_find_bound(__be32 addr, __be16 port) |
| 59 | { |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 60 | u64 key = ((u64)addr << 32) | port; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 61 | struct rds_sock *rs; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 62 | |
Cong Wang | 53a3bad | 2018-09-10 18:27:26 -0700 | [diff] [blame] | 63 | rcu_read_lock(); |
| 64 | rs = rhashtable_lookup(&bind_hash_table, &key, ht_parms); |
Eric Dumazet | 44476fd | 2019-01-31 08:47:10 -0800 | [diff] [blame] | 65 | if (rs && (sock_flag(rds_rs_to_sk(rs), SOCK_DEAD) || |
| 66 | !atomic_inc_not_zero(&rds_rs_to_sk(rs)->sk_refcnt))) |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 67 | rs = NULL; |
Eric Dumazet | 44476fd | 2019-01-31 08:47:10 -0800 | [diff] [blame] | 68 | |
Cong Wang | 53a3bad | 2018-09-10 18:27:26 -0700 | [diff] [blame] | 69 | rcu_read_unlock(); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 70 | |
| 71 | rdsdebug("returning rs %p for %pI4:%u\n", rs, &addr, |
| 72 | ntohs(port)); |
Santosh Shilimkar | 8b0a6b4 | 2012-02-03 11:09:23 -0500 | [diff] [blame] | 73 | |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 74 | return rs; |
| 75 | } |
| 76 | |
| 77 | /* returns -ve errno or +ve port */ |
| 78 | static int rds_add_bound(struct rds_sock *rs, __be32 addr, __be16 *port) |
| 79 | { |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 80 | int ret = -EADDRINUSE; |
| 81 | u16 rover, last; |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 82 | u64 key; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 83 | |
| 84 | if (*port != 0) { |
| 85 | rover = be16_to_cpu(*port); |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 86 | if (rover == RDS_FLAG_PROBE_PORT) |
| 87 | return -EINVAL; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 88 | last = rover; |
| 89 | } else { |
Aruna-Hewapathirane | 63862b5 | 2014-01-11 07:15:59 -0500 | [diff] [blame] | 90 | rover = max_t(u16, prandom_u32(), 2); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 91 | last = rover - 1; |
| 92 | } |
| 93 | |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 94 | do { |
| 95 | if (rover == 0) |
| 96 | rover++; |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 97 | |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 98 | if (rover == RDS_FLAG_PROBE_PORT) |
| 99 | continue; |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 100 | key = ((u64)addr << 32) | cpu_to_be16(rover); |
| 101 | if (rhashtable_lookup_fast(&bind_hash_table, &key, ht_parms)) |
| 102 | continue; |
| 103 | |
| 104 | rs->rs_bound_key = key; |
| 105 | rs->rs_bound_addr = addr; |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 106 | net_get_random_once(&rs->rs_hash_initval, |
| 107 | sizeof(rs->rs_hash_initval)); |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 108 | rs->rs_bound_port = cpu_to_be16(rover); |
| 109 | rs->rs_bound_node.next = NULL; |
| 110 | rds_sock_addref(rs); |
| 111 | if (!rhashtable_insert_fast(&bind_hash_table, |
| 112 | &rs->rs_bound_node, ht_parms)) { |
Chris Mason | 38a4e5e | 2010-05-11 15:09:45 -0700 | [diff] [blame] | 113 | *port = rs->rs_bound_port; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 114 | ret = 0; |
Chris Mason | 38a4e5e | 2010-05-11 15:09:45 -0700 | [diff] [blame] | 115 | rdsdebug("rs %p binding to %pI4:%d\n", |
| 116 | rs, &addr, (int)ntohs(*port)); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 117 | break; |
Santosh Shilimkar | 2812695 | 2012-02-03 11:08:50 -0500 | [diff] [blame] | 118 | } else { |
Sowmini Varadhan | 59bf036 | 2017-12-22 09:38:59 -0800 | [diff] [blame] | 119 | rs->rs_bound_addr = 0; |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 120 | rds_sock_put(rs); |
| 121 | ret = -ENOMEM; |
| 122 | break; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 123 | } |
| 124 | } while (rover++ != last); |
| 125 | |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 126 | return ret; |
| 127 | } |
| 128 | |
| 129 | void rds_remove_bound(struct rds_sock *rs) |
| 130 | { |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 131 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 132 | if (!rs->rs_bound_addr) |
| 133 | return; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 134 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 135 | rdsdebug("rs %p unbinding from %pI4:%d\n", |
| 136 | rs, &rs->rs_bound_addr, |
| 137 | ntohs(rs->rs_bound_port)); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 138 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 139 | rhashtable_remove_fast(&bind_hash_table, &rs->rs_bound_node, ht_parms); |
| 140 | rds_sock_put(rs); |
| 141 | rs->rs_bound_addr = 0; |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 142 | } |
| 143 | |
| 144 | int rds_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len) |
| 145 | { |
| 146 | struct sock *sk = sock->sk; |
| 147 | struct sockaddr_in *sin = (struct sockaddr_in *)uaddr; |
| 148 | struct rds_sock *rs = rds_sk_to_rs(sk); |
| 149 | struct rds_transport *trans; |
| 150 | int ret = 0; |
| 151 | |
| 152 | lock_sock(sk); |
| 153 | |
| 154 | if (addr_len != sizeof(struct sockaddr_in) || |
| 155 | sin->sin_family != AF_INET || |
| 156 | rs->rs_bound_addr || |
| 157 | sin->sin_addr.s_addr == htonl(INADDR_ANY)) { |
| 158 | ret = -EINVAL; |
| 159 | goto out; |
| 160 | } |
| 161 | |
Cong Wang | 53a3bad | 2018-09-10 18:27:26 -0700 | [diff] [blame] | 162 | sock_set_flag(sk, SOCK_RCU_FREE); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 163 | ret = rds_add_bound(rs, sin->sin_addr.s_addr, &sin->sin_port); |
| 164 | if (ret) |
| 165 | goto out; |
| 166 | |
Sowmini Varadhan | d97dac5 | 2015-05-29 17:28:08 -0400 | [diff] [blame] | 167 | if (rs->rs_transport) { /* previously bound */ |
Sowmini Varadhan | 4867980 | 2015-10-11 16:46:03 -0400 | [diff] [blame] | 168 | trans = rs->rs_transport; |
| 169 | if (trans->laddr_check(sock_net(sock->sk), |
| 170 | sin->sin_addr.s_addr) != 0) { |
| 171 | ret = -ENOPROTOOPT; |
| 172 | rds_remove_bound(rs); |
| 173 | } else { |
| 174 | ret = 0; |
| 175 | } |
Sowmini Varadhan | d97dac5 | 2015-05-29 17:28:08 -0400 | [diff] [blame] | 176 | goto out; |
| 177 | } |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 178 | trans = rds_trans_get_preferred(sock_net(sock->sk), |
| 179 | sin->sin_addr.s_addr); |
Andy Grover | 8690bfa | 2010-01-12 11:56:44 -0800 | [diff] [blame] | 180 | if (!trans) { |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 181 | ret = -EADDRNOTAVAIL; |
| 182 | rds_remove_bound(rs); |
Manuel Zerpies | cb0a605 | 2011-06-16 02:09:57 +0000 | [diff] [blame] | 183 | printk_ratelimited(KERN_INFO "RDS: rds_bind() could not find a transport, " |
Andy Grover | f2c4493 | 2009-08-21 12:28:35 +0000 | [diff] [blame] | 184 | "load rds_tcp or rds_rdma?\n"); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 185 | goto out; |
| 186 | } |
| 187 | |
| 188 | rs->rs_transport = trans; |
| 189 | ret = 0; |
| 190 | |
| 191 | out: |
| 192 | release_sock(sk); |
Andy Grover | 639b321 | 2009-02-24 15:30:18 +0000 | [diff] [blame] | 193 | return ret; |
| 194 | } |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 195 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 196 | void rds_bind_lock_destroy(void) |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 197 | { |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 198 | rhashtable_destroy(&bind_hash_table); |
| 199 | } |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 200 | |
santosh.shilimkar@oracle.com | 7b56543 | 2015-10-30 08:49:10 -0700 | [diff] [blame] | 201 | int rds_bind_lock_init(void) |
| 202 | { |
| 203 | return rhashtable_init(&bind_hash_table, &ht_parms); |
Santosh Shilimkar | 9b9acde | 2014-02-11 19:34:25 -0800 | [diff] [blame] | 204 | } |