Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +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 <linux/list.h> |
Tejun Heo | 5a0e3ad | 2010-03-24 17:04:11 +0900 | [diff] [blame] | 35 | #include <linux/slab.h> |
Paul Gortmaker | bc3b2d7 | 2011-07-15 11:47:34 -0400 | [diff] [blame] | 36 | #include <linux/export.h> |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 37 | #include <net/inet_hashtables.h> |
| 38 | |
| 39 | #include "rds.h" |
| 40 | #include "loop.h" |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 41 | |
| 42 | #define RDS_CONNECTION_HASH_BITS 12 |
| 43 | #define RDS_CONNECTION_HASH_ENTRIES (1 << RDS_CONNECTION_HASH_BITS) |
| 44 | #define RDS_CONNECTION_HASH_MASK (RDS_CONNECTION_HASH_ENTRIES - 1) |
| 45 | |
| 46 | /* converting this to RCU is a chore for another day.. */ |
| 47 | static DEFINE_SPINLOCK(rds_conn_lock); |
| 48 | static unsigned long rds_conn_count; |
| 49 | static struct hlist_head rds_conn_hash[RDS_CONNECTION_HASH_ENTRIES]; |
| 50 | static struct kmem_cache *rds_conn_slab; |
| 51 | |
| 52 | static struct hlist_head *rds_conn_bucket(__be32 laddr, __be32 faddr) |
| 53 | { |
Hannes Frederic Sowa | 1bbdcee | 2013-10-19 21:48:57 +0200 | [diff] [blame] | 54 | static u32 rds_hash_secret __read_mostly; |
| 55 | |
| 56 | unsigned long hash; |
| 57 | |
| 58 | net_get_random_once(&rds_hash_secret, sizeof(rds_hash_secret)); |
| 59 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 60 | /* Pass NULL, don't need struct net for hash */ |
Hannes Frederic Sowa | 1bbdcee | 2013-10-19 21:48:57 +0200 | [diff] [blame] | 61 | hash = __inet_ehashfn(be32_to_cpu(laddr), 0, |
| 62 | be32_to_cpu(faddr), 0, |
| 63 | rds_hash_secret); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 64 | return &rds_conn_hash[hash & RDS_CONNECTION_HASH_MASK]; |
| 65 | } |
| 66 | |
| 67 | #define rds_conn_info_set(var, test, suffix) do { \ |
| 68 | if (test) \ |
| 69 | var |= RDS_INFO_CONNECTION_FLAG_##suffix; \ |
| 70 | } while (0) |
| 71 | |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 72 | /* rcu read lock must be held or the connection spinlock */ |
Sowmini Varadhan | 8f384c0 | 2015-09-03 16:24:52 -0400 | [diff] [blame] | 73 | static struct rds_connection *rds_conn_lookup(struct net *net, |
| 74 | struct hlist_head *head, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 75 | __be32 laddr, __be32 faddr, |
| 76 | struct rds_transport *trans) |
| 77 | { |
| 78 | struct rds_connection *conn, *ret = NULL; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 79 | |
Sasha Levin | b67bfe0 | 2013-02-27 17:06:00 -0800 | [diff] [blame] | 80 | hlist_for_each_entry_rcu(conn, head, c_hash_node) { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 81 | if (conn->c_faddr == faddr && conn->c_laddr == laddr && |
Sowmini Varadhan | 8f384c0 | 2015-09-03 16:24:52 -0400 | [diff] [blame] | 82 | conn->c_trans == trans && net == rds_conn_net(conn)) { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 83 | ret = conn; |
| 84 | break; |
| 85 | } |
| 86 | } |
| 87 | rdsdebug("returning conn %p for %pI4 -> %pI4\n", ret, |
| 88 | &laddr, &faddr); |
| 89 | return ret; |
| 90 | } |
| 91 | |
| 92 | /* |
| 93 | * This is called by transports as they're bringing down a connection. |
| 94 | * It clears partial message state so that the transport can start sending |
| 95 | * and receiving over this connection again in the future. It is up to |
| 96 | * the transport to have serialized this call with its send and recv. |
| 97 | */ |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 98 | static void rds_conn_path_reset(struct rds_conn_path *cp) |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 99 | { |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 100 | struct rds_connection *conn = cp->cp_conn; |
| 101 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 102 | rdsdebug("connection %pI4 to %pI4 reset\n", |
| 103 | &conn->c_laddr, &conn->c_faddr); |
| 104 | |
| 105 | rds_stats_inc(s_conn_reset); |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 106 | rds_send_path_reset(cp); |
| 107 | cp->cp_flags = 0; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 108 | |
| 109 | /* Do not clear next_rx_seq here, else we cannot distinguish |
| 110 | * retransmitted packets from new packets, and will hand all |
| 111 | * of them to the application. That is not consistent with the |
| 112 | * reliability guarantees of RDS. */ |
| 113 | } |
| 114 | |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 115 | static void __rds_conn_path_init(struct rds_connection *conn, |
| 116 | struct rds_conn_path *cp, bool is_outgoing) |
| 117 | { |
| 118 | spin_lock_init(&cp->cp_lock); |
| 119 | cp->cp_next_tx_seq = 1; |
| 120 | init_waitqueue_head(&cp->cp_waitq); |
| 121 | INIT_LIST_HEAD(&cp->cp_send_queue); |
| 122 | INIT_LIST_HEAD(&cp->cp_retrans); |
| 123 | |
| 124 | cp->cp_conn = conn; |
| 125 | atomic_set(&cp->cp_state, RDS_CONN_DOWN); |
| 126 | cp->cp_send_gen = 0; |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 127 | cp->cp_reconnect_jiffies = 0; |
| 128 | INIT_DELAYED_WORK(&cp->cp_send_w, rds_send_worker); |
| 129 | INIT_DELAYED_WORK(&cp->cp_recv_w, rds_recv_worker); |
| 130 | INIT_DELAYED_WORK(&cp->cp_conn_w, rds_connect_worker); |
| 131 | INIT_WORK(&cp->cp_down_w, rds_shutdown_worker); |
| 132 | mutex_init(&cp->cp_cm_lock); |
| 133 | cp->cp_flags = 0; |
| 134 | } |
| 135 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 136 | /* |
| 137 | * There is only every one 'conn' for a given pair of addresses in the |
| 138 | * system at a time. They contain messages to be retransmitted and so |
| 139 | * span the lifetime of the actual underlying transport connections. |
| 140 | * |
| 141 | * For now they are not garbage collected once they're created. They |
| 142 | * are torn down as the module is removed, if ever. |
| 143 | */ |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 144 | static struct rds_connection *__rds_conn_create(struct net *net, |
| 145 | __be32 laddr, __be32 faddr, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 146 | struct rds_transport *trans, gfp_t gfp, |
| 147 | int is_outgoing) |
| 148 | { |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 149 | struct rds_connection *conn, *parent = NULL; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 150 | struct hlist_head *head = rds_conn_bucket(laddr, faddr); |
Zach Brown | 5adb5bc | 2010-07-23 10:32:31 -0700 | [diff] [blame] | 151 | struct rds_transport *loop_trans; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 152 | unsigned long flags; |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 153 | int ret, i; |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 154 | int npaths = (trans->t_mp_capable ? RDS_MPATH_WORKERS : 1); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 155 | |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 156 | rcu_read_lock(); |
Sowmini Varadhan | 8f384c0 | 2015-09-03 16:24:52 -0400 | [diff] [blame] | 157 | conn = rds_conn_lookup(net, head, laddr, faddr, trans); |
Joe Perches | f64f9e7 | 2009-11-29 16:55:45 -0800 | [diff] [blame] | 158 | if (conn && conn->c_loopback && conn->c_trans != &rds_loop_transport && |
Sowmini Varadhan | 1789b2c07 | 2015-04-08 12:33:46 -0400 | [diff] [blame] | 159 | laddr == faddr && !is_outgoing) { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 160 | /* This is a looped back IB connection, and we're |
| 161 | * called by the code handling the incoming connect. |
| 162 | * We need a second connection object into which we |
| 163 | * can stick the other QP. */ |
| 164 | parent = conn; |
| 165 | conn = parent->c_passive; |
| 166 | } |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 167 | rcu_read_unlock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 168 | if (conn) |
| 169 | goto out; |
| 170 | |
Wei Yongjun | 05a178e | 2009-04-09 14:09:44 +0000 | [diff] [blame] | 171 | conn = kmem_cache_zalloc(rds_conn_slab, gfp); |
Andy Grover | 8690bfa | 2010-01-12 11:56:44 -0800 | [diff] [blame] | 172 | if (!conn) { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 173 | conn = ERR_PTR(-ENOMEM); |
| 174 | goto out; |
| 175 | } |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 176 | conn->c_path = kcalloc(npaths, sizeof(struct rds_conn_path), gfp); |
| 177 | if (!conn->c_path) { |
| 178 | kmem_cache_free(rds_conn_slab, conn); |
| 179 | conn = ERR_PTR(-ENOMEM); |
| 180 | goto out; |
| 181 | } |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 182 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 183 | INIT_HLIST_NODE(&conn->c_hash_node); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 184 | conn->c_laddr = laddr; |
| 185 | conn->c_faddr = faddr; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 186 | |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 187 | rds_conn_net_set(conn, net); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 188 | |
| 189 | ret = rds_cong_get_maps(conn); |
| 190 | if (ret) { |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 191 | kfree(conn->c_path); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 192 | kmem_cache_free(rds_conn_slab, conn); |
| 193 | conn = ERR_PTR(ret); |
| 194 | goto out; |
| 195 | } |
| 196 | |
| 197 | /* |
| 198 | * This is where a connection becomes loopback. If *any* RDS sockets |
| 199 | * can bind to the destination address then we'd rather the messages |
| 200 | * flow through loopback rather than either transport. |
| 201 | */ |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 202 | loop_trans = rds_trans_get_preferred(net, faddr); |
Zach Brown | 5adb5bc | 2010-07-23 10:32:31 -0700 | [diff] [blame] | 203 | if (loop_trans) { |
| 204 | rds_trans_put(loop_trans); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 205 | conn->c_loopback = 1; |
| 206 | if (is_outgoing && trans->t_prefer_loopback) { |
| 207 | /* "outgoing" connection - and the transport |
| 208 | * says it wants the connection handled by the |
| 209 | * loopback transport. This is what TCP does. |
| 210 | */ |
| 211 | trans = &rds_loop_transport; |
| 212 | } |
| 213 | } |
| 214 | |
| 215 | conn->c_trans = trans; |
| 216 | |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 217 | init_waitqueue_head(&conn->c_hs_waitq); |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 218 | for (i = 0; i < npaths; i++) { |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 219 | __rds_conn_path_init(conn, &conn->c_path[i], |
| 220 | is_outgoing); |
| 221 | conn->c_path[i].cp_index = i; |
| 222 | } |
Sowmini Varadhan | ebeeb1a | 2018-02-03 04:26:51 -0800 | [diff] [blame] | 223 | rcu_read_lock(); |
| 224 | if (rds_destroy_pending(conn)) |
| 225 | ret = -ENETDOWN; |
| 226 | else |
Sowmini Varadhan | d4014d8 | 2018-02-13 09:46:16 -0800 | [diff] [blame] | 227 | ret = trans->conn_alloc(conn, GFP_ATOMIC); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 228 | if (ret) { |
Sowmini Varadhan | ebeeb1a | 2018-02-03 04:26:51 -0800 | [diff] [blame] | 229 | rcu_read_unlock(); |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 230 | kfree(conn->c_path); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 231 | kmem_cache_free(rds_conn_slab, conn); |
| 232 | conn = ERR_PTR(ret); |
| 233 | goto out; |
| 234 | } |
| 235 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 236 | rdsdebug("allocated conn %p for %pI4 -> %pI4 over %s %s\n", |
| 237 | conn, &laddr, &faddr, |
Prashant Bhole | d9b8693 | 2017-12-06 10:47:04 +0900 | [diff] [blame] | 238 | strnlen(trans->t_name, sizeof(trans->t_name)) ? trans->t_name : |
| 239 | "[unknown]", is_outgoing ? "(outgoing)" : ""); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 240 | |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 241 | /* |
| 242 | * Since we ran without holding the conn lock, someone could |
| 243 | * have created the same conn (either normal or passive) in the |
| 244 | * interim. We check while holding the lock. If we won, we complete |
| 245 | * init and return our conn. If we lost, we rollback and return the |
| 246 | * other one. |
| 247 | */ |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 248 | spin_lock_irqsave(&rds_conn_lock, flags); |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 249 | if (parent) { |
| 250 | /* Creating passive conn */ |
| 251 | if (parent->c_passive) { |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 252 | trans->conn_free(conn->c_path[0].cp_transport_data); |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 253 | kfree(conn->c_path); |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 254 | kmem_cache_free(rds_conn_slab, conn); |
| 255 | conn = parent->c_passive; |
| 256 | } else { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 257 | parent->c_passive = conn; |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 258 | rds_cong_add_conn(conn); |
| 259 | rds_conn_count++; |
| 260 | } |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 261 | } else { |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 262 | /* Creating normal conn */ |
| 263 | struct rds_connection *found; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 264 | |
Sowmini Varadhan | 3b20fc3 | 2015-09-30 16:54:07 -0400 | [diff] [blame] | 265 | found = rds_conn_lookup(net, head, laddr, faddr, trans); |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 266 | if (found) { |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 267 | struct rds_conn_path *cp; |
| 268 | int i; |
| 269 | |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 270 | for (i = 0; i < npaths; i++) { |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 271 | cp = &conn->c_path[i]; |
Sowmini Varadhan | 02105b2 | 2016-06-30 16:11:12 -0700 | [diff] [blame] | 272 | /* The ->conn_alloc invocation may have |
| 273 | * allocated resource for all paths, so all |
| 274 | * of them may have to be freed here. |
| 275 | */ |
| 276 | if (cp->cp_transport_data) |
| 277 | trans->conn_free(cp->cp_transport_data); |
Sowmini Varadhan | 1c5113c | 2016-06-13 09:44:40 -0700 | [diff] [blame] | 278 | } |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 279 | kfree(conn->c_path); |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 280 | kmem_cache_free(rds_conn_slab, conn); |
| 281 | conn = found; |
| 282 | } else { |
Sowmini Varadhan | 905dd41 | 2016-11-16 13:29:49 -0800 | [diff] [blame] | 283 | conn->c_my_gen_num = rds_gen_num; |
| 284 | conn->c_peer_gen_num = 0; |
Sowmini Varadhan | 3b20fc3 | 2015-09-30 16:54:07 -0400 | [diff] [blame] | 285 | hlist_add_head_rcu(&conn->c_hash_node, head); |
Andy Grover | cb24405 | 2009-07-17 13:13:36 +0000 | [diff] [blame] | 286 | rds_cong_add_conn(conn); |
| 287 | rds_conn_count++; |
| 288 | } |
| 289 | } |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 290 | spin_unlock_irqrestore(&rds_conn_lock, flags); |
Sowmini Varadhan | ebeeb1a | 2018-02-03 04:26:51 -0800 | [diff] [blame] | 291 | rcu_read_unlock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 292 | |
| 293 | out: |
| 294 | return conn; |
| 295 | } |
| 296 | |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 297 | struct rds_connection *rds_conn_create(struct net *net, |
| 298 | __be32 laddr, __be32 faddr, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 299 | struct rds_transport *trans, gfp_t gfp) |
| 300 | { |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 301 | return __rds_conn_create(net, laddr, faddr, trans, gfp, 0); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 302 | } |
Andy Grover | 616b757 | 2009-08-21 12:28:32 +0000 | [diff] [blame] | 303 | EXPORT_SYMBOL_GPL(rds_conn_create); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 304 | |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 305 | struct rds_connection *rds_conn_create_outgoing(struct net *net, |
| 306 | __be32 laddr, __be32 faddr, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 307 | struct rds_transport *trans, gfp_t gfp) |
| 308 | { |
Sowmini Varadhan | d5a8ac2 | 2015-08-05 01:43:25 -0400 | [diff] [blame] | 309 | return __rds_conn_create(net, laddr, faddr, trans, gfp, 1); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 310 | } |
Andy Grover | 616b757 | 2009-08-21 12:28:32 +0000 | [diff] [blame] | 311 | EXPORT_SYMBOL_GPL(rds_conn_create_outgoing); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 312 | |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 313 | void rds_conn_shutdown(struct rds_conn_path *cp) |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 314 | { |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 315 | struct rds_connection *conn = cp->cp_conn; |
| 316 | |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 317 | /* shut it down unless it's down already */ |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 318 | if (!rds_conn_path_transition(cp, RDS_CONN_DOWN, RDS_CONN_DOWN)) { |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 319 | /* |
| 320 | * Quiesce the connection mgmt handlers before we start tearing |
| 321 | * things down. We don't hold the mutex for the entire |
| 322 | * duration of the shutdown operation, else we may be |
| 323 | * deadlocking with the CM handler. Instead, the CM event |
| 324 | * handler is supposed to check for state DISCONNECTING |
| 325 | */ |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 326 | mutex_lock(&cp->cp_cm_lock); |
| 327 | if (!rds_conn_path_transition(cp, RDS_CONN_UP, |
| 328 | RDS_CONN_DISCONNECTING) && |
| 329 | !rds_conn_path_transition(cp, RDS_CONN_ERROR, |
| 330 | RDS_CONN_DISCONNECTING)) { |
| 331 | rds_conn_path_error(cp, |
| 332 | "shutdown called in state %d\n", |
| 333 | atomic_read(&cp->cp_state)); |
| 334 | mutex_unlock(&cp->cp_cm_lock); |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 335 | return; |
| 336 | } |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 337 | mutex_unlock(&cp->cp_cm_lock); |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 338 | |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 339 | wait_event(cp->cp_waitq, |
| 340 | !test_bit(RDS_IN_XMIT, &cp->cp_flags)); |
| 341 | wait_event(cp->cp_waitq, |
| 342 | !test_bit(RDS_RECV_REFILL, &cp->cp_flags)); |
Chris Mason | 7e3f295 | 2010-05-11 15:11:11 -0700 | [diff] [blame] | 343 | |
Sowmini Varadhan | 226f7a7 | 2016-06-30 16:11:10 -0700 | [diff] [blame] | 344 | conn->c_trans->conn_path_shutdown(cp); |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 345 | rds_conn_path_reset(cp); |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 346 | |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 347 | if (!rds_conn_path_transition(cp, RDS_CONN_DISCONNECTING, |
Sowmini Varadhan | e97656d | 2017-03-31 15:56:30 -0700 | [diff] [blame] | 348 | RDS_CONN_DOWN) && |
| 349 | !rds_conn_path_transition(cp, RDS_CONN_ERROR, |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 350 | RDS_CONN_DOWN)) { |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 351 | /* This can happen - eg when we're in the middle of tearing |
| 352 | * down the connection, and someone unloads the rds module. |
Sowmini Varadhan | e97656d | 2017-03-31 15:56:30 -0700 | [diff] [blame] | 353 | * Quite reproducible with loopback connections. |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 354 | * Mostly harmless. |
Sowmini Varadhan | e97656d | 2017-03-31 15:56:30 -0700 | [diff] [blame] | 355 | * |
| 356 | * Note that this also happens with rds-tcp because |
| 357 | * we could have triggered rds_conn_path_drop in irq |
| 358 | * mode from rds_tcp_state change on the receipt of |
| 359 | * a FIN, thus we need to recheck for RDS_CONN_ERROR |
| 360 | * here. |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 361 | */ |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 362 | rds_conn_path_error(cp, "%s: failed to transition " |
| 363 | "to state DOWN, current state " |
| 364 | "is %d\n", __func__, |
| 365 | atomic_read(&cp->cp_state)); |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 366 | return; |
| 367 | } |
| 368 | } |
| 369 | |
| 370 | /* Then reconnect if it's still live. |
| 371 | * The passive side of an IB loopback connection is never added |
| 372 | * to the conn hash, so we never trigger a reconnect on this |
| 373 | * conn - the reconnect is always triggered by the active peer. */ |
Sowmini Varadhan | d769ef8 | 2016-06-13 09:44:41 -0700 | [diff] [blame] | 374 | cancel_delayed_work_sync(&cp->cp_conn_w); |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 375 | rcu_read_lock(); |
| 376 | if (!hlist_unhashed(&conn->c_hash_node)) { |
| 377 | rcu_read_unlock(); |
Sowmini Varadhan | 8315011 | 2016-06-30 16:11:17 -0700 | [diff] [blame] | 378 | rds_queue_reconnect(cp); |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 379 | } else { |
| 380 | rcu_read_unlock(); |
| 381 | } |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 382 | } |
| 383 | |
Sowmini Varadhan | 3ecc569 | 2016-06-13 09:44:42 -0700 | [diff] [blame] | 384 | /* destroy a single rds_conn_path. rds_conn_destroy() iterates over |
| 385 | * all paths using rds_conn_path_destroy() |
| 386 | */ |
| 387 | static void rds_conn_path_destroy(struct rds_conn_path *cp) |
| 388 | { |
| 389 | struct rds_message *rm, *rtmp; |
| 390 | |
Sowmini Varadhan | 02105b2 | 2016-06-30 16:11:12 -0700 | [diff] [blame] | 391 | if (!cp->cp_transport_data) |
| 392 | return; |
| 393 | |
Sowmini Varadhan | 3ecc569 | 2016-06-13 09:44:42 -0700 | [diff] [blame] | 394 | /* make sure lingering queued work won't try to ref the conn */ |
| 395 | cancel_delayed_work_sync(&cp->cp_send_w); |
| 396 | cancel_delayed_work_sync(&cp->cp_recv_w); |
| 397 | |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 398 | rds_conn_path_drop(cp, true); |
| 399 | flush_work(&cp->cp_down_w); |
| 400 | |
Sowmini Varadhan | 3ecc569 | 2016-06-13 09:44:42 -0700 | [diff] [blame] | 401 | /* tear down queued messages */ |
| 402 | list_for_each_entry_safe(rm, rtmp, |
| 403 | &cp->cp_send_queue, |
| 404 | m_conn_item) { |
| 405 | list_del_init(&rm->m_conn_item); |
| 406 | BUG_ON(!list_empty(&rm->m_sock_item)); |
| 407 | rds_message_put(rm); |
| 408 | } |
| 409 | if (cp->cp_xmit_rm) |
| 410 | rds_message_put(cp->cp_xmit_rm); |
| 411 | |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 412 | WARN_ON(delayed_work_pending(&cp->cp_send_w)); |
| 413 | WARN_ON(delayed_work_pending(&cp->cp_recv_w)); |
| 414 | WARN_ON(delayed_work_pending(&cp->cp_conn_w)); |
| 415 | WARN_ON(work_pending(&cp->cp_down_w)); |
| 416 | |
Sowmini Varadhan | 3ecc569 | 2016-06-13 09:44:42 -0700 | [diff] [blame] | 417 | cp->cp_conn->c_trans->conn_free(cp->cp_transport_data); |
| 418 | } |
| 419 | |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 420 | /* |
| 421 | * Stop and free a connection. |
Zach Brown | ffcec0e | 2010-07-23 10:36:58 -0700 | [diff] [blame] | 422 | * |
| 423 | * This can only be used in very limited circumstances. It assumes that once |
| 424 | * the conn has been shutdown that no one else is referencing the connection. |
| 425 | * We can only ensure this in the rmmod path in the current code. |
Andy Grover | 2dc3935 | 2010-06-11 13:49:13 -0700 | [diff] [blame] | 426 | */ |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 427 | void rds_conn_destroy(struct rds_connection *conn) |
| 428 | { |
Zach Brown | fe8ff6b | 2010-07-23 10:30:45 -0700 | [diff] [blame] | 429 | unsigned long flags; |
Sowmini Varadhan | 02105b2 | 2016-06-30 16:11:12 -0700 | [diff] [blame] | 430 | int i; |
| 431 | struct rds_conn_path *cp; |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 432 | int npaths = (conn->c_trans->t_mp_capable ? RDS_MPATH_WORKERS : 1); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 433 | |
| 434 | rdsdebug("freeing conn %p for %pI4 -> " |
| 435 | "%pI4\n", conn, &conn->c_laddr, |
| 436 | &conn->c_faddr); |
| 437 | |
Chris Mason | abf4543 | 2010-05-11 15:14:52 -0700 | [diff] [blame] | 438 | /* Ensure conn will not be scheduled for reconnect */ |
| 439 | spin_lock_irq(&rds_conn_lock); |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 440 | hlist_del_init_rcu(&conn->c_hash_node); |
Chris Mason | abf4543 | 2010-05-11 15:14:52 -0700 | [diff] [blame] | 441 | spin_unlock_irq(&rds_conn_lock); |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 442 | synchronize_rcu(); |
| 443 | |
Zach Brown | ffcec0e | 2010-07-23 10:36:58 -0700 | [diff] [blame] | 444 | /* shut the connection down */ |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 445 | for (i = 0; i < npaths; i++) { |
Sowmini Varadhan | 02105b2 | 2016-06-30 16:11:12 -0700 | [diff] [blame] | 446 | cp = &conn->c_path[i]; |
| 447 | rds_conn_path_destroy(cp); |
| 448 | BUG_ON(!list_empty(&cp->cp_retrans)); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 449 | } |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 450 | |
| 451 | /* |
| 452 | * The congestion maps aren't freed up here. They're |
| 453 | * freed by rds_cong_exit() after all the connections |
| 454 | * have been freed. |
| 455 | */ |
| 456 | rds_cong_remove_conn(conn); |
| 457 | |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 458 | kfree(conn->c_path); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 459 | kmem_cache_free(rds_conn_slab, conn); |
| 460 | |
Zach Brown | fe8ff6b | 2010-07-23 10:30:45 -0700 | [diff] [blame] | 461 | spin_lock_irqsave(&rds_conn_lock, flags); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 462 | rds_conn_count--; |
Zach Brown | fe8ff6b | 2010-07-23 10:30:45 -0700 | [diff] [blame] | 463 | spin_unlock_irqrestore(&rds_conn_lock, flags); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 464 | } |
Andy Grover | 616b757 | 2009-08-21 12:28:32 +0000 | [diff] [blame] | 465 | EXPORT_SYMBOL_GPL(rds_conn_destroy); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 466 | |
| 467 | static void rds_conn_message_info(struct socket *sock, unsigned int len, |
| 468 | struct rds_info_iterator *iter, |
| 469 | struct rds_info_lengths *lens, |
| 470 | int want_send) |
| 471 | { |
| 472 | struct hlist_head *head; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 473 | struct list_head *list; |
| 474 | struct rds_connection *conn; |
| 475 | struct rds_message *rm; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 476 | unsigned int total = 0; |
Zach Brown | 501dccc | 2010-06-04 14:25:27 -0700 | [diff] [blame] | 477 | unsigned long flags; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 478 | size_t i; |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 479 | int j; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 480 | |
| 481 | len /= sizeof(struct rds_info_message); |
| 482 | |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 483 | rcu_read_lock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 484 | |
| 485 | for (i = 0, head = rds_conn_hash; i < ARRAY_SIZE(rds_conn_hash); |
| 486 | i++, head++) { |
Sasha Levin | b67bfe0 | 2013-02-27 17:06:00 -0800 | [diff] [blame] | 487 | hlist_for_each_entry_rcu(conn, head, c_hash_node) { |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 488 | struct rds_conn_path *cp; |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 489 | int npaths; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 490 | |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 491 | npaths = (conn->c_trans->t_mp_capable ? |
| 492 | RDS_MPATH_WORKERS : 1); |
| 493 | |
| 494 | for (j = 0; j < npaths; j++) { |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 495 | cp = &conn->c_path[j]; |
| 496 | if (want_send) |
| 497 | list = &cp->cp_send_queue; |
| 498 | else |
| 499 | list = &cp->cp_retrans; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 500 | |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 501 | spin_lock_irqsave(&cp->cp_lock, flags); |
| 502 | |
| 503 | /* XXX too lazy to maintain counts.. */ |
| 504 | list_for_each_entry(rm, list, m_conn_item) { |
| 505 | total++; |
| 506 | if (total <= len) |
| 507 | rds_inc_info_copy(&rm->m_inc, |
| 508 | iter, |
| 509 | conn->c_laddr, |
| 510 | conn->c_faddr, |
| 511 | 0); |
| 512 | } |
| 513 | |
| 514 | spin_unlock_irqrestore(&cp->cp_lock, flags); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 515 | } |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 516 | } |
| 517 | } |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 518 | rcu_read_unlock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 519 | |
| 520 | lens->nr = total; |
| 521 | lens->each = sizeof(struct rds_info_message); |
| 522 | } |
| 523 | |
| 524 | static void rds_conn_message_info_send(struct socket *sock, unsigned int len, |
| 525 | struct rds_info_iterator *iter, |
| 526 | struct rds_info_lengths *lens) |
| 527 | { |
| 528 | rds_conn_message_info(sock, len, iter, lens, 1); |
| 529 | } |
| 530 | |
| 531 | static void rds_conn_message_info_retrans(struct socket *sock, |
| 532 | unsigned int len, |
| 533 | struct rds_info_iterator *iter, |
| 534 | struct rds_info_lengths *lens) |
| 535 | { |
| 536 | rds_conn_message_info(sock, len, iter, lens, 0); |
| 537 | } |
| 538 | |
| 539 | void rds_for_each_conn_info(struct socket *sock, unsigned int len, |
| 540 | struct rds_info_iterator *iter, |
| 541 | struct rds_info_lengths *lens, |
| 542 | int (*visitor)(struct rds_connection *, void *), |
Salvatore Mesoraca | f1cb9d6 | 2018-03-11 22:07:49 +0100 | [diff] [blame] | 543 | u64 *buffer, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 544 | size_t item_len) |
| 545 | { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 546 | struct hlist_head *head; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 547 | struct rds_connection *conn; |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 548 | size_t i; |
| 549 | |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 550 | rcu_read_lock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 551 | |
| 552 | lens->nr = 0; |
| 553 | lens->each = item_len; |
| 554 | |
| 555 | for (i = 0, head = rds_conn_hash; i < ARRAY_SIZE(rds_conn_hash); |
| 556 | i++, head++) { |
Sasha Levin | b67bfe0 | 2013-02-27 17:06:00 -0800 | [diff] [blame] | 557 | hlist_for_each_entry_rcu(conn, head, c_hash_node) { |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 558 | |
| 559 | /* XXX no c_lock usage.. */ |
| 560 | if (!visitor(conn, buffer)) |
| 561 | continue; |
| 562 | |
| 563 | /* We copy as much as we can fit in the buffer, |
| 564 | * but we count all items so that the caller |
| 565 | * can resize the buffer. */ |
| 566 | if (len >= item_len) { |
| 567 | rds_info_copy(iter, buffer, item_len); |
| 568 | len -= item_len; |
| 569 | } |
| 570 | lens->nr++; |
| 571 | } |
| 572 | } |
Chris Mason | bcf50ef | 2010-05-11 15:15:15 -0700 | [diff] [blame] | 573 | rcu_read_unlock(); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 574 | } |
Andy Grover | 616b757 | 2009-08-21 12:28:32 +0000 | [diff] [blame] | 575 | EXPORT_SYMBOL_GPL(rds_for_each_conn_info); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 576 | |
Santosh Shilimkar | bb78976 | 2016-12-04 16:41:29 -0800 | [diff] [blame] | 577 | static void rds_walk_conn_path_info(struct socket *sock, unsigned int len, |
| 578 | struct rds_info_iterator *iter, |
| 579 | struct rds_info_lengths *lens, |
| 580 | int (*visitor)(struct rds_conn_path *, void *), |
Salvatore Mesoraca | b2c9272 | 2018-03-11 22:07:50 +0100 | [diff] [blame] | 581 | u64 *buffer, |
Santosh Shilimkar | bb78976 | 2016-12-04 16:41:29 -0800 | [diff] [blame] | 582 | size_t item_len) |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 583 | { |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 584 | struct hlist_head *head; |
| 585 | struct rds_connection *conn; |
| 586 | size_t i; |
| 587 | int j; |
| 588 | |
| 589 | rcu_read_lock(); |
| 590 | |
| 591 | lens->nr = 0; |
| 592 | lens->each = item_len; |
| 593 | |
| 594 | for (i = 0, head = rds_conn_hash; i < ARRAY_SIZE(rds_conn_hash); |
| 595 | i++, head++) { |
| 596 | hlist_for_each_entry_rcu(conn, head, c_hash_node) { |
| 597 | struct rds_conn_path *cp; |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 598 | int npaths; |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 599 | |
Sowmini Varadhan | 840df16 | 2017-08-02 10:34:31 -0700 | [diff] [blame] | 600 | npaths = (conn->c_trans->t_mp_capable ? |
| 601 | RDS_MPATH_WORKERS : 1); |
| 602 | for (j = 0; j < npaths; j++) { |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 603 | cp = &conn->c_path[j]; |
| 604 | |
| 605 | /* XXX no cp_lock usage.. */ |
| 606 | if (!visitor(cp, buffer)) |
| 607 | continue; |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 608 | } |
| 609 | |
| 610 | /* We copy as much as we can fit in the buffer, |
| 611 | * but we count all items so that the caller |
| 612 | * can resize the buffer. |
| 613 | */ |
| 614 | if (len >= item_len) { |
| 615 | rds_info_copy(iter, buffer, item_len); |
| 616 | len -= item_len; |
| 617 | } |
| 618 | lens->nr++; |
| 619 | } |
| 620 | } |
| 621 | rcu_read_unlock(); |
| 622 | } |
| 623 | |
| 624 | static int rds_conn_info_visitor(struct rds_conn_path *cp, void *buffer) |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 625 | { |
| 626 | struct rds_info_connection *cinfo = buffer; |
| 627 | |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 628 | cinfo->next_tx_seq = cp->cp_next_tx_seq; |
| 629 | cinfo->next_rx_seq = cp->cp_next_rx_seq; |
| 630 | cinfo->laddr = cp->cp_conn->c_laddr; |
| 631 | cinfo->faddr = cp->cp_conn->c_faddr; |
| 632 | strncpy(cinfo->transport, cp->cp_conn->c_trans->t_name, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 633 | sizeof(cinfo->transport)); |
| 634 | cinfo->flags = 0; |
| 635 | |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 636 | rds_conn_info_set(cinfo->flags, test_bit(RDS_IN_XMIT, &cp->cp_flags), |
Zach Brown | 0f4b1c7 | 2010-06-04 14:41:41 -0700 | [diff] [blame] | 637 | SENDING); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 638 | /* XXX Future: return the state rather than these funky bits */ |
| 639 | rds_conn_info_set(cinfo->flags, |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 640 | atomic_read(&cp->cp_state) == RDS_CONN_CONNECTING, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 641 | CONNECTING); |
| 642 | rds_conn_info_set(cinfo->flags, |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 643 | atomic_read(&cp->cp_state) == RDS_CONN_UP, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 644 | CONNECTED); |
| 645 | return 1; |
| 646 | } |
| 647 | |
| 648 | static void rds_conn_info(struct socket *sock, unsigned int len, |
| 649 | struct rds_info_iterator *iter, |
| 650 | struct rds_info_lengths *lens) |
| 651 | { |
Salvatore Mesoraca | b2c9272 | 2018-03-11 22:07:50 +0100 | [diff] [blame] | 652 | u64 buffer[(sizeof(struct rds_info_connection) + 7) / 8]; |
| 653 | |
Sowmini Varadhan | 992c9ec | 2016-06-13 09:44:38 -0700 | [diff] [blame] | 654 | rds_walk_conn_path_info(sock, len, iter, lens, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 655 | rds_conn_info_visitor, |
Salvatore Mesoraca | b2c9272 | 2018-03-11 22:07:50 +0100 | [diff] [blame] | 656 | buffer, |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 657 | sizeof(struct rds_info_connection)); |
| 658 | } |
| 659 | |
Zach Brown | ef87b7e | 2010-07-09 12:26:20 -0700 | [diff] [blame] | 660 | int rds_conn_init(void) |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 661 | { |
| 662 | rds_conn_slab = kmem_cache_create("rds_connection", |
| 663 | sizeof(struct rds_connection), |
| 664 | 0, 0, NULL); |
Andy Grover | 8690bfa | 2010-01-12 11:56:44 -0800 | [diff] [blame] | 665 | if (!rds_conn_slab) |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 666 | return -ENOMEM; |
| 667 | |
| 668 | rds_info_register_func(RDS_INFO_CONNECTIONS, rds_conn_info); |
| 669 | rds_info_register_func(RDS_INFO_SEND_MESSAGES, |
| 670 | rds_conn_message_info_send); |
| 671 | rds_info_register_func(RDS_INFO_RETRANS_MESSAGES, |
| 672 | rds_conn_message_info_retrans); |
| 673 | |
| 674 | return 0; |
| 675 | } |
| 676 | |
| 677 | void rds_conn_exit(void) |
| 678 | { |
| 679 | rds_loop_exit(); |
| 680 | |
| 681 | WARN_ON(!hlist_empty(rds_conn_hash)); |
| 682 | |
| 683 | kmem_cache_destroy(rds_conn_slab); |
| 684 | |
| 685 | rds_info_deregister_func(RDS_INFO_CONNECTIONS, rds_conn_info); |
| 686 | rds_info_deregister_func(RDS_INFO_SEND_MESSAGES, |
| 687 | rds_conn_message_info_send); |
| 688 | rds_info_deregister_func(RDS_INFO_RETRANS_MESSAGES, |
| 689 | rds_conn_message_info_retrans); |
| 690 | } |
| 691 | |
| 692 | /* |
| 693 | * Force a disconnect |
| 694 | */ |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 695 | void rds_conn_path_drop(struct rds_conn_path *cp, bool destroy) |
Sowmini Varadhan | 0cb4396 | 2016-06-13 09:44:26 -0700 | [diff] [blame] | 696 | { |
| 697 | atomic_set(&cp->cp_state, RDS_CONN_ERROR); |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 698 | |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 699 | rcu_read_lock(); |
Sowmini Varadhan | ebeeb1a | 2018-02-03 04:26:51 -0800 | [diff] [blame] | 700 | if (!destroy && rds_destroy_pending(cp->cp_conn)) { |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 701 | rcu_read_unlock(); |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 702 | return; |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 703 | } |
Sowmini Varadhan | 0cb4396 | 2016-06-13 09:44:26 -0700 | [diff] [blame] | 704 | queue_work(rds_wq, &cp->cp_down_w); |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 705 | rcu_read_unlock(); |
Sowmini Varadhan | 0cb4396 | 2016-06-13 09:44:26 -0700 | [diff] [blame] | 706 | } |
| 707 | EXPORT_SYMBOL_GPL(rds_conn_path_drop); |
| 708 | |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 709 | void rds_conn_drop(struct rds_connection *conn) |
| 710 | { |
Sowmini Varadhan | 5916e2c | 2016-07-14 03:51:03 -0700 | [diff] [blame] | 711 | WARN_ON(conn->c_trans->t_mp_capable); |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 712 | rds_conn_path_drop(&conn->c_path[0], false); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 713 | } |
Andy Grover | 616b757 | 2009-08-21 12:28:32 +0000 | [diff] [blame] | 714 | EXPORT_SYMBOL_GPL(rds_conn_drop); |
Andy Grover | 00e0f34 | 2009-02-24 15:30:23 +0000 | [diff] [blame] | 715 | |
| 716 | /* |
Zach Brown | f3c6808 | 2010-05-24 13:14:36 -0700 | [diff] [blame] | 717 | * If the connection is down, trigger a connect. We may have scheduled a |
| 718 | * delayed reconnect however - in this case we should not interfere. |
| 719 | */ |
Sowmini Varadhan | 3c0a590 | 2016-06-13 09:44:37 -0700 | [diff] [blame] | 720 | void rds_conn_path_connect_if_down(struct rds_conn_path *cp) |
| 721 | { |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 722 | rcu_read_lock(); |
Sowmini Varadhan | ebeeb1a | 2018-02-03 04:26:51 -0800 | [diff] [blame] | 723 | if (rds_destroy_pending(cp->cp_conn)) { |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 724 | rcu_read_unlock(); |
| 725 | return; |
| 726 | } |
Sowmini Varadhan | 3c0a590 | 2016-06-13 09:44:37 -0700 | [diff] [blame] | 727 | if (rds_conn_path_state(cp) == RDS_CONN_DOWN && |
| 728 | !test_and_set_bit(RDS_RECONNECT_PENDING, &cp->cp_flags)) |
| 729 | queue_delayed_work(rds_wq, &cp->cp_conn_w, 0); |
Sowmini Varadhan | 3db6e0d | 2018-01-04 06:53:00 -0800 | [diff] [blame] | 730 | rcu_read_unlock(); |
Sowmini Varadhan | 3c0a590 | 2016-06-13 09:44:37 -0700 | [diff] [blame] | 731 | } |
Sowmini Varadhan | 1a0e100 | 2016-11-16 13:29:50 -0800 | [diff] [blame] | 732 | EXPORT_SYMBOL_GPL(rds_conn_path_connect_if_down); |
Sowmini Varadhan | 3c0a590 | 2016-06-13 09:44:37 -0700 | [diff] [blame] | 733 | |
Zach Brown | f3c6808 | 2010-05-24 13:14:36 -0700 | [diff] [blame] | 734 | void rds_conn_connect_if_down(struct rds_connection *conn) |
| 735 | { |
Sowmini Varadhan | 3c0a590 | 2016-06-13 09:44:37 -0700 | [diff] [blame] | 736 | WARN_ON(conn->c_trans->t_mp_capable); |
| 737 | rds_conn_path_connect_if_down(&conn->c_path[0]); |
Zach Brown | f3c6808 | 2010-05-24 13:14:36 -0700 | [diff] [blame] | 738 | } |
| 739 | EXPORT_SYMBOL_GPL(rds_conn_connect_if_down); |
| 740 | |
Sowmini Varadhan | fb1b3dc | 2016-06-13 09:44:39 -0700 | [diff] [blame] | 741 | void |
| 742 | __rds_conn_path_error(struct rds_conn_path *cp, const char *fmt, ...) |
| 743 | { |
| 744 | va_list ap; |
| 745 | |
| 746 | va_start(ap, fmt); |
| 747 | vprintk(fmt, ap); |
| 748 | va_end(ap); |
| 749 | |
Sowmini Varadhan | aed20a5 | 2017-07-16 16:43:46 -0700 | [diff] [blame] | 750 | rds_conn_path_drop(cp, false); |
Sowmini Varadhan | fb1b3dc | 2016-06-13 09:44:39 -0700 | [diff] [blame] | 751 | } |