1 // SPDX-License-Identifier: GPL-2.0-or-later 1 // SPDX-License-Identifier: GPL-2.0-or-later 2 /* Client connection-specific management code. 2 /* Client connection-specific management code. 3 * 3 * 4 * Copyright (C) 2016, 2020 Red Hat, Inc. All 4 * Copyright (C) 2016, 2020 Red Hat, Inc. All Rights Reserved. 5 * Written by David Howells (dhowells@redhat.c 5 * Written by David Howells (dhowells@redhat.com) 6 * 6 * 7 * Client connections need to be cached for a 7 * Client connections need to be cached for a little while after they've made a 8 * call so as to handle retransmitted DATA pac 8 * call so as to handle retransmitted DATA packets in case the server didn't 9 * receive the final ACK or terminating ABORT 9 * receive the final ACK or terminating ABORT we sent it. 10 * 10 * 11 * There are flags of relevance to the cache: 11 * There are flags of relevance to the cache: 12 * 12 * 13 * (2) DONT_REUSE - The connection should be 13 * (2) DONT_REUSE - The connection should be discarded as soon as possible and 14 * should not be reused. This is set whe 14 * should not be reused. This is set when an exclusive connection is used 15 * or a call ID counter overflows. 15 * or a call ID counter overflows. 16 * 16 * 17 * The caching state may only be changed if th 17 * The caching state may only be changed if the cache lock is held. 18 * 18 * 19 * There are two idle client connection expiry 19 * There are two idle client connection expiry durations. If the total number 20 * of connections is below the reap threshold, 20 * of connections is below the reap threshold, we use the normal duration; if 21 * it's above, we use the fast duration. 21 * it's above, we use the fast duration. 22 */ 22 */ 23 23 24 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 24 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt 25 25 26 #include <linux/slab.h> 26 #include <linux/slab.h> 27 #include <linux/idr.h> 27 #include <linux/idr.h> 28 #include <linux/timer.h> 28 #include <linux/timer.h> 29 #include <linux/sched/signal.h> 29 #include <linux/sched/signal.h> 30 30 31 #include "ar-internal.h" 31 #include "ar-internal.h" 32 32 33 __read_mostly unsigned int rxrpc_reap_client_c 33 __read_mostly unsigned int rxrpc_reap_client_connections = 900; 34 __read_mostly unsigned long rxrpc_conn_idle_cl 34 __read_mostly unsigned long rxrpc_conn_idle_client_expiry = 2 * 60 * HZ; 35 __read_mostly unsigned long rxrpc_conn_idle_cl 35 __read_mostly unsigned long rxrpc_conn_idle_client_fast_expiry = 2 * HZ; 36 36 37 static void rxrpc_activate_bundle(struct rxrpc 37 static void rxrpc_activate_bundle(struct rxrpc_bundle *bundle) 38 { 38 { 39 atomic_inc(&bundle->active); 39 atomic_inc(&bundle->active); 40 } 40 } 41 41 42 /* 42 /* 43 * Release a connection ID for a client connec 43 * Release a connection ID for a client connection. 44 */ 44 */ 45 static void rxrpc_put_client_connection_id(str 45 static void rxrpc_put_client_connection_id(struct rxrpc_local *local, 46 str 46 struct rxrpc_connection *conn) 47 { 47 { 48 idr_remove(&local->conn_ids, conn->pro 48 idr_remove(&local->conn_ids, conn->proto.cid >> RXRPC_CIDSHIFT); 49 } 49 } 50 50 51 /* 51 /* 52 * Destroy the client connection ID tree. 52 * Destroy the client connection ID tree. 53 */ 53 */ 54 static void rxrpc_destroy_client_conn_ids(stru 54 static void rxrpc_destroy_client_conn_ids(struct rxrpc_local *local) 55 { 55 { 56 struct rxrpc_connection *conn; 56 struct rxrpc_connection *conn; 57 int id; 57 int id; 58 58 59 if (!idr_is_empty(&local->conn_ids)) { 59 if (!idr_is_empty(&local->conn_ids)) { 60 idr_for_each_entry(&local->con 60 idr_for_each_entry(&local->conn_ids, conn, id) { 61 pr_err("AF_RXRPC: Leak 61 pr_err("AF_RXRPC: Leaked client conn %p {%d}\n", 62 conn, refcount_ 62 conn, refcount_read(&conn->ref)); 63 } 63 } 64 BUG(); 64 BUG(); 65 } 65 } 66 66 67 idr_destroy(&local->conn_ids); 67 idr_destroy(&local->conn_ids); 68 } 68 } 69 69 70 /* 70 /* 71 * Allocate a connection bundle. 71 * Allocate a connection bundle. 72 */ 72 */ 73 static struct rxrpc_bundle *rxrpc_alloc_bundle 73 static struct rxrpc_bundle *rxrpc_alloc_bundle(struct rxrpc_call *call, 74 74 gfp_t gfp) 75 { 75 { 76 static atomic_t rxrpc_bundle_id; 76 static atomic_t rxrpc_bundle_id; 77 struct rxrpc_bundle *bundle; 77 struct rxrpc_bundle *bundle; 78 78 79 bundle = kzalloc(sizeof(*bundle), gfp) 79 bundle = kzalloc(sizeof(*bundle), gfp); 80 if (bundle) { 80 if (bundle) { 81 bundle->local = call 81 bundle->local = call->local; 82 bundle->peer = rxrp 82 bundle->peer = rxrpc_get_peer(call->peer, rxrpc_peer_get_bundle); 83 bundle->key = key_ 83 bundle->key = key_get(call->key); 84 bundle->security = call 84 bundle->security = call->security; 85 bundle->exclusive = test 85 bundle->exclusive = test_bit(RXRPC_CALL_EXCLUSIVE, &call->flags); 86 bundle->upgrade = test 86 bundle->upgrade = test_bit(RXRPC_CALL_UPGRADE, &call->flags); 87 bundle->service_id = call 87 bundle->service_id = call->dest_srx.srx_service; 88 bundle->security_level = call 88 bundle->security_level = call->security_level; 89 bundle->debug_id = atom 89 bundle->debug_id = atomic_inc_return(&rxrpc_bundle_id); 90 refcount_set(&bundle->ref, 1); 90 refcount_set(&bundle->ref, 1); 91 atomic_set(&bundle->active, 1) 91 atomic_set(&bundle->active, 1); 92 INIT_LIST_HEAD(&bundle->waitin 92 INIT_LIST_HEAD(&bundle->waiting_calls); 93 trace_rxrpc_bundle(bundle->deb 93 trace_rxrpc_bundle(bundle->debug_id, 1, rxrpc_bundle_new); 94 94 95 write_lock(&bundle->local->rxn 95 write_lock(&bundle->local->rxnet->conn_lock); 96 list_add_tail(&bundle->proc_li 96 list_add_tail(&bundle->proc_link, &bundle->local->rxnet->bundle_proc_list); 97 write_unlock(&bundle->local->r 97 write_unlock(&bundle->local->rxnet->conn_lock); 98 } 98 } 99 return bundle; 99 return bundle; 100 } 100 } 101 101 102 struct rxrpc_bundle *rxrpc_get_bundle(struct r 102 struct rxrpc_bundle *rxrpc_get_bundle(struct rxrpc_bundle *bundle, 103 enum rxr 103 enum rxrpc_bundle_trace why) 104 { 104 { 105 int r; 105 int r; 106 106 107 __refcount_inc(&bundle->ref, &r); 107 __refcount_inc(&bundle->ref, &r); 108 trace_rxrpc_bundle(bundle->debug_id, r 108 trace_rxrpc_bundle(bundle->debug_id, r + 1, why); 109 return bundle; 109 return bundle; 110 } 110 } 111 111 112 static void rxrpc_free_bundle(struct rxrpc_bun 112 static void rxrpc_free_bundle(struct rxrpc_bundle *bundle) 113 { 113 { 114 trace_rxrpc_bundle(bundle->debug_id, r 114 trace_rxrpc_bundle(bundle->debug_id, refcount_read(&bundle->ref), 115 rxrpc_bundle_free); 115 rxrpc_bundle_free); 116 write_lock(&bundle->local->rxnet->conn 116 write_lock(&bundle->local->rxnet->conn_lock); 117 list_del(&bundle->proc_link); 117 list_del(&bundle->proc_link); 118 write_unlock(&bundle->local->rxnet->co 118 write_unlock(&bundle->local->rxnet->conn_lock); 119 rxrpc_put_peer(bundle->peer, rxrpc_pee 119 rxrpc_put_peer(bundle->peer, rxrpc_peer_put_bundle); 120 key_put(bundle->key); 120 key_put(bundle->key); 121 kfree(bundle); 121 kfree(bundle); 122 } 122 } 123 123 124 void rxrpc_put_bundle(struct rxrpc_bundle *bun 124 void rxrpc_put_bundle(struct rxrpc_bundle *bundle, enum rxrpc_bundle_trace why) 125 { 125 { 126 unsigned int id; 126 unsigned int id; 127 bool dead; 127 bool dead; 128 int r; 128 int r; 129 129 130 if (bundle) { 130 if (bundle) { 131 id = bundle->debug_id; 131 id = bundle->debug_id; 132 dead = __refcount_dec_and_test 132 dead = __refcount_dec_and_test(&bundle->ref, &r); 133 trace_rxrpc_bundle(id, r - 1, 133 trace_rxrpc_bundle(id, r - 1, why); 134 if (dead) 134 if (dead) 135 rxrpc_free_bundle(bund 135 rxrpc_free_bundle(bundle); 136 } 136 } 137 } 137 } 138 138 139 /* 139 /* 140 * Get rid of outstanding client connection pr 140 * Get rid of outstanding client connection preallocations when a local 141 * endpoint is destroyed. 141 * endpoint is destroyed. 142 */ 142 */ 143 void rxrpc_purge_client_connections(struct rxr 143 void rxrpc_purge_client_connections(struct rxrpc_local *local) 144 { 144 { 145 rxrpc_destroy_client_conn_ids(local); 145 rxrpc_destroy_client_conn_ids(local); 146 } 146 } 147 147 148 /* 148 /* 149 * Allocate a client connection. 149 * Allocate a client connection. 150 */ 150 */ 151 static struct rxrpc_connection * 151 static struct rxrpc_connection * 152 rxrpc_alloc_client_connection(struct rxrpc_bun 152 rxrpc_alloc_client_connection(struct rxrpc_bundle *bundle) 153 { 153 { 154 struct rxrpc_connection *conn; 154 struct rxrpc_connection *conn; 155 struct rxrpc_local *local = bundle->lo 155 struct rxrpc_local *local = bundle->local; 156 struct rxrpc_net *rxnet = local->rxnet 156 struct rxrpc_net *rxnet = local->rxnet; 157 int id; 157 int id; 158 158 159 _enter(""); 159 _enter(""); 160 160 161 conn = rxrpc_alloc_connection(rxnet, G 161 conn = rxrpc_alloc_connection(rxnet, GFP_ATOMIC | __GFP_NOWARN); 162 if (!conn) 162 if (!conn) 163 return ERR_PTR(-ENOMEM); 163 return ERR_PTR(-ENOMEM); 164 164 165 id = idr_alloc_cyclic(&local->conn_ids 165 id = idr_alloc_cyclic(&local->conn_ids, conn, 1, 0x40000000, 166 GFP_ATOMIC | __G 166 GFP_ATOMIC | __GFP_NOWARN); 167 if (id < 0) { 167 if (id < 0) { 168 kfree(conn); 168 kfree(conn); 169 return ERR_PTR(id); 169 return ERR_PTR(id); 170 } 170 } 171 171 172 refcount_set(&conn->ref, 1); 172 refcount_set(&conn->ref, 1); 173 conn->proto.cid = id << RXRPC_ 173 conn->proto.cid = id << RXRPC_CIDSHIFT; 174 conn->proto.epoch = local->rxnet 174 conn->proto.epoch = local->rxnet->epoch; 175 conn->out_clientflag = RXRPC_CLIENT 175 conn->out_clientflag = RXRPC_CLIENT_INITIATED; 176 conn->bundle = rxrpc_get_bu 176 conn->bundle = rxrpc_get_bundle(bundle, rxrpc_bundle_get_client_conn); 177 conn->local = rxrpc_get_lo 177 conn->local = rxrpc_get_local(bundle->local, rxrpc_local_get_client_conn); 178 conn->peer = rxrpc_get_pe 178 conn->peer = rxrpc_get_peer(bundle->peer, rxrpc_peer_get_client_conn); 179 conn->key = key_get(bund 179 conn->key = key_get(bundle->key); 180 conn->security = bundle->secu 180 conn->security = bundle->security; 181 conn->exclusive = bundle->excl 181 conn->exclusive = bundle->exclusive; 182 conn->upgrade = bundle->upgr 182 conn->upgrade = bundle->upgrade; 183 conn->orig_service_id = bundle->serv 183 conn->orig_service_id = bundle->service_id; 184 conn->security_level = bundle->secu 184 conn->security_level = bundle->security_level; 185 conn->state = RXRPC_CONN_C 185 conn->state = RXRPC_CONN_CLIENT_UNSECURED; 186 conn->service_id = conn->orig_s 186 conn->service_id = conn->orig_service_id; 187 187 188 if (conn->security == &rxrpc_no_securi 188 if (conn->security == &rxrpc_no_security) 189 conn->state = RXRPC_CONN_C 189 conn->state = RXRPC_CONN_CLIENT; 190 190 191 atomic_inc(&rxnet->nr_conns); 191 atomic_inc(&rxnet->nr_conns); 192 write_lock(&rxnet->conn_lock); 192 write_lock(&rxnet->conn_lock); 193 list_add_tail(&conn->proc_link, &rxnet 193 list_add_tail(&conn->proc_link, &rxnet->conn_proc_list); 194 write_unlock(&rxnet->conn_lock); 194 write_unlock(&rxnet->conn_lock); 195 195 196 rxrpc_see_connection(conn, rxrpc_conn_ 196 rxrpc_see_connection(conn, rxrpc_conn_new_client); 197 197 198 atomic_inc(&rxnet->nr_client_conns); 198 atomic_inc(&rxnet->nr_client_conns); 199 trace_rxrpc_client(conn, -1, rxrpc_cli 199 trace_rxrpc_client(conn, -1, rxrpc_client_alloc); 200 return conn; 200 return conn; 201 } 201 } 202 202 203 /* 203 /* 204 * Determine if a connection may be reused. 204 * Determine if a connection may be reused. 205 */ 205 */ 206 static bool rxrpc_may_reuse_conn(struct rxrpc_ 206 static bool rxrpc_may_reuse_conn(struct rxrpc_connection *conn) 207 { 207 { 208 struct rxrpc_net *rxnet; 208 struct rxrpc_net *rxnet; 209 int id_cursor, id, distance, limit; 209 int id_cursor, id, distance, limit; 210 210 211 if (!conn) 211 if (!conn) 212 goto dont_reuse; 212 goto dont_reuse; 213 213 214 rxnet = conn->rxnet; 214 rxnet = conn->rxnet; 215 if (test_bit(RXRPC_CONN_DONT_REUSE, &c 215 if (test_bit(RXRPC_CONN_DONT_REUSE, &conn->flags)) 216 goto dont_reuse; 216 goto dont_reuse; 217 217 218 if ((conn->state != RXRPC_CONN_CLIENT_ 218 if ((conn->state != RXRPC_CONN_CLIENT_UNSECURED && 219 conn->state != RXRPC_CONN_CLIENT) 219 conn->state != RXRPC_CONN_CLIENT) || 220 conn->proto.epoch != rxnet->epoch) 220 conn->proto.epoch != rxnet->epoch) 221 goto mark_dont_reuse; 221 goto mark_dont_reuse; 222 222 223 /* The IDR tree gets very expensive on 223 /* The IDR tree gets very expensive on memory if the connection IDs are 224 * widely scattered throughout the num 224 * widely scattered throughout the number space, so we shall want to 225 * kill off connections that, say, hav 225 * kill off connections that, say, have an ID more than about four 226 * times the maximum number of client 226 * times the maximum number of client conns away from the current 227 * allocation point to try and keep th 227 * allocation point to try and keep the IDs concentrated. 228 */ 228 */ 229 id_cursor = idr_get_cursor(&conn->loca 229 id_cursor = idr_get_cursor(&conn->local->conn_ids); 230 id = conn->proto.cid >> RXRPC_CIDSHIFT 230 id = conn->proto.cid >> RXRPC_CIDSHIFT; 231 distance = id - id_cursor; 231 distance = id - id_cursor; 232 if (distance < 0) 232 if (distance < 0) 233 distance = -distance; 233 distance = -distance; 234 limit = max_t(unsigned long, atomic_re 234 limit = max_t(unsigned long, atomic_read(&rxnet->nr_conns) * 4, 1024); 235 if (distance > limit) 235 if (distance > limit) 236 goto mark_dont_reuse; 236 goto mark_dont_reuse; 237 237 238 return true; 238 return true; 239 239 240 mark_dont_reuse: 240 mark_dont_reuse: 241 set_bit(RXRPC_CONN_DONT_REUSE, &conn-> 241 set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags); 242 dont_reuse: 242 dont_reuse: 243 return false; 243 return false; 244 } 244 } 245 245 246 /* 246 /* 247 * Look up the conn bundle that matches the co 247 * Look up the conn bundle that matches the connection parameters, adding it if 248 * it doesn't yet exist. 248 * it doesn't yet exist. 249 */ 249 */ 250 int rxrpc_look_up_bundle(struct rxrpc_call *ca 250 int rxrpc_look_up_bundle(struct rxrpc_call *call, gfp_t gfp) 251 { 251 { 252 struct rxrpc_bundle *bundle, *candidat 252 struct rxrpc_bundle *bundle, *candidate; 253 struct rxrpc_local *local = call->loca 253 struct rxrpc_local *local = call->local; 254 struct rb_node *p, **pp, *parent; 254 struct rb_node *p, **pp, *parent; 255 long diff; 255 long diff; 256 bool upgrade = test_bit(RXRPC_CALL_UPG 256 bool upgrade = test_bit(RXRPC_CALL_UPGRADE, &call->flags); 257 257 258 _enter("{%px,%x,%u,%u}", 258 _enter("{%px,%x,%u,%u}", 259 call->peer, key_serial(call->ke 259 call->peer, key_serial(call->key), call->security_level, 260 upgrade); 260 upgrade); 261 261 262 if (test_bit(RXRPC_CALL_EXCLUSIVE, &ca 262 if (test_bit(RXRPC_CALL_EXCLUSIVE, &call->flags)) { 263 call->bundle = rxrpc_alloc_bun 263 call->bundle = rxrpc_alloc_bundle(call, gfp); 264 return call->bundle ? 0 : -ENO 264 return call->bundle ? 0 : -ENOMEM; 265 } 265 } 266 266 267 /* First, see if the bundle is already 267 /* First, see if the bundle is already there. */ 268 _debug("search 1"); 268 _debug("search 1"); 269 spin_lock(&local->client_bundles_lock) 269 spin_lock(&local->client_bundles_lock); 270 p = local->client_bundles.rb_node; 270 p = local->client_bundles.rb_node; 271 while (p) { 271 while (p) { 272 bundle = rb_entry(p, struct rx 272 bundle = rb_entry(p, struct rxrpc_bundle, local_node); 273 273 274 #define cmp(X, Y) ((long)(X) - (long)(Y)) 274 #define cmp(X, Y) ((long)(X) - (long)(Y)) 275 diff = (cmp(bundle->peer, call 275 diff = (cmp(bundle->peer, call->peer) ?: 276 cmp(bundle->key, call- 276 cmp(bundle->key, call->key) ?: 277 cmp(bundle->security_l 277 cmp(bundle->security_level, call->security_level) ?: 278 cmp(bundle->upgrade, u 278 cmp(bundle->upgrade, upgrade)); 279 #undef cmp 279 #undef cmp 280 if (diff < 0) 280 if (diff < 0) 281 p = p->rb_left; 281 p = p->rb_left; 282 else if (diff > 0) 282 else if (diff > 0) 283 p = p->rb_right; 283 p = p->rb_right; 284 else 284 else 285 goto found_bundle; 285 goto found_bundle; 286 } 286 } 287 spin_unlock(&local->client_bundles_loc 287 spin_unlock(&local->client_bundles_lock); 288 _debug("not found"); 288 _debug("not found"); 289 289 290 /* It wasn't. We need to add one. */ 290 /* It wasn't. We need to add one. */ 291 candidate = rxrpc_alloc_bundle(call, g 291 candidate = rxrpc_alloc_bundle(call, gfp); 292 if (!candidate) 292 if (!candidate) 293 return -ENOMEM; 293 return -ENOMEM; 294 294 295 _debug("search 2"); 295 _debug("search 2"); 296 spin_lock(&local->client_bundles_lock) 296 spin_lock(&local->client_bundles_lock); 297 pp = &local->client_bundles.rb_node; 297 pp = &local->client_bundles.rb_node; 298 parent = NULL; 298 parent = NULL; 299 while (*pp) { 299 while (*pp) { 300 parent = *pp; 300 parent = *pp; 301 bundle = rb_entry(parent, stru 301 bundle = rb_entry(parent, struct rxrpc_bundle, local_node); 302 302 303 #define cmp(X, Y) ((long)(X) - (long)(Y)) 303 #define cmp(X, Y) ((long)(X) - (long)(Y)) 304 diff = (cmp(bundle->peer, call 304 diff = (cmp(bundle->peer, call->peer) ?: 305 cmp(bundle->key, call- 305 cmp(bundle->key, call->key) ?: 306 cmp(bundle->security_l 306 cmp(bundle->security_level, call->security_level) ?: 307 cmp(bundle->upgrade, u 307 cmp(bundle->upgrade, upgrade)); 308 #undef cmp 308 #undef cmp 309 if (diff < 0) 309 if (diff < 0) 310 pp = &(*pp)->rb_left; 310 pp = &(*pp)->rb_left; 311 else if (diff > 0) 311 else if (diff > 0) 312 pp = &(*pp)->rb_right; 312 pp = &(*pp)->rb_right; 313 else 313 else 314 goto found_bundle_free 314 goto found_bundle_free; 315 } 315 } 316 316 317 _debug("new bundle"); 317 _debug("new bundle"); 318 rb_link_node(&candidate->local_node, p 318 rb_link_node(&candidate->local_node, parent, pp); 319 rb_insert_color(&candidate->local_node 319 rb_insert_color(&candidate->local_node, &local->client_bundles); 320 call->bundle = rxrpc_get_bundle(candid 320 call->bundle = rxrpc_get_bundle(candidate, rxrpc_bundle_get_client_call); 321 spin_unlock(&local->client_bundles_loc 321 spin_unlock(&local->client_bundles_lock); 322 _leave(" = B=%u [new]", call->bundle-> 322 _leave(" = B=%u [new]", call->bundle->debug_id); 323 return 0; 323 return 0; 324 324 325 found_bundle_free: 325 found_bundle_free: 326 rxrpc_free_bundle(candidate); 326 rxrpc_free_bundle(candidate); 327 found_bundle: 327 found_bundle: 328 call->bundle = rxrpc_get_bundle(bundle 328 call->bundle = rxrpc_get_bundle(bundle, rxrpc_bundle_get_client_call); 329 rxrpc_activate_bundle(bundle); 329 rxrpc_activate_bundle(bundle); 330 spin_unlock(&local->client_bundles_loc 330 spin_unlock(&local->client_bundles_lock); 331 _leave(" = B=%u [found]", call->bundle 331 _leave(" = B=%u [found]", call->bundle->debug_id); 332 return 0; 332 return 0; 333 } 333 } 334 334 335 /* 335 /* 336 * Allocate a new connection and add it into a 336 * Allocate a new connection and add it into a bundle. 337 */ 337 */ 338 static bool rxrpc_add_conn_to_bundle(struct rx 338 static bool rxrpc_add_conn_to_bundle(struct rxrpc_bundle *bundle, 339 unsigned 339 unsigned int slot) 340 { 340 { 341 struct rxrpc_connection *conn, *old; 341 struct rxrpc_connection *conn, *old; 342 unsigned int shift = slot * RXRPC_MAXC 342 unsigned int shift = slot * RXRPC_MAXCALLS; 343 unsigned int i; 343 unsigned int i; 344 344 345 old = bundle->conns[slot]; 345 old = bundle->conns[slot]; 346 if (old) { 346 if (old) { 347 bundle->conns[slot] = NULL; 347 bundle->conns[slot] = NULL; 348 bundle->conn_ids[slot] = 0; 348 bundle->conn_ids[slot] = 0; 349 trace_rxrpc_client(old, -1, rx 349 trace_rxrpc_client(old, -1, rxrpc_client_replace); 350 rxrpc_put_connection(old, rxrp 350 rxrpc_put_connection(old, rxrpc_conn_put_noreuse); 351 } 351 } 352 352 353 conn = rxrpc_alloc_client_connection(b 353 conn = rxrpc_alloc_client_connection(bundle); 354 if (IS_ERR(conn)) { 354 if (IS_ERR(conn)) { 355 bundle->alloc_error = PTR_ERR( 355 bundle->alloc_error = PTR_ERR(conn); 356 return false; 356 return false; 357 } 357 } 358 358 359 rxrpc_activate_bundle(bundle); 359 rxrpc_activate_bundle(bundle); 360 conn->bundle_shift = shift; 360 conn->bundle_shift = shift; 361 bundle->conns[slot] = conn; 361 bundle->conns[slot] = conn; 362 bundle->conn_ids[slot] = conn->debug_i 362 bundle->conn_ids[slot] = conn->debug_id; 363 for (i = 0; i < RXRPC_MAXCALLS; i++) 363 for (i = 0; i < RXRPC_MAXCALLS; i++) 364 set_bit(shift + i, &bundle->av 364 set_bit(shift + i, &bundle->avail_chans); 365 return true; 365 return true; 366 } 366 } 367 367 368 /* 368 /* 369 * Add a connection to a bundle if there are n 369 * Add a connection to a bundle if there are no usable connections or we have 370 * connections waiting for extra capacity. 370 * connections waiting for extra capacity. 371 */ 371 */ 372 static bool rxrpc_bundle_has_space(struct rxrp 372 static bool rxrpc_bundle_has_space(struct rxrpc_bundle *bundle) 373 { 373 { 374 int slot = -1, i, usable; 374 int slot = -1, i, usable; 375 375 376 _enter(""); 376 _enter(""); 377 377 378 bundle->alloc_error = 0; 378 bundle->alloc_error = 0; 379 379 380 /* See if there are any usable connect 380 /* See if there are any usable connections. */ 381 usable = 0; 381 usable = 0; 382 for (i = 0; i < ARRAY_SIZE(bundle->con 382 for (i = 0; i < ARRAY_SIZE(bundle->conns); i++) { 383 if (rxrpc_may_reuse_conn(bundl 383 if (rxrpc_may_reuse_conn(bundle->conns[i])) 384 usable++; 384 usable++; 385 else if (slot == -1) 385 else if (slot == -1) 386 slot = i; 386 slot = i; 387 } 387 } 388 388 389 if (!usable && bundle->upgrade) 389 if (!usable && bundle->upgrade) 390 bundle->try_upgrade = true; 390 bundle->try_upgrade = true; 391 391 392 if (!usable) 392 if (!usable) 393 goto alloc_conn; 393 goto alloc_conn; 394 394 395 if (!bundle->avail_chans && 395 if (!bundle->avail_chans && 396 !bundle->try_upgrade && 396 !bundle->try_upgrade && 397 usable < ARRAY_SIZE(bundle->conns) 397 usable < ARRAY_SIZE(bundle->conns)) 398 goto alloc_conn; 398 goto alloc_conn; 399 399 400 _leave(""); 400 _leave(""); 401 return usable; 401 return usable; 402 402 403 alloc_conn: 403 alloc_conn: 404 return slot >= 0 ? rxrpc_add_conn_to_b 404 return slot >= 0 ? rxrpc_add_conn_to_bundle(bundle, slot) : false; 405 } 405 } 406 406 407 /* 407 /* 408 * Assign a channel to the call at the front o 408 * Assign a channel to the call at the front of the queue and wake the call up. 409 * We don't increment the callNumber counter u 409 * We don't increment the callNumber counter until this number has been exposed 410 * to the world. 410 * to the world. 411 */ 411 */ 412 static void rxrpc_activate_one_channel(struct 412 static void rxrpc_activate_one_channel(struct rxrpc_connection *conn, 413 unsigne 413 unsigned int channel) 414 { 414 { 415 struct rxrpc_channel *chan = &conn->ch 415 struct rxrpc_channel *chan = &conn->channels[channel]; 416 struct rxrpc_bundle *bundle = conn->bu 416 struct rxrpc_bundle *bundle = conn->bundle; 417 struct rxrpc_call *call = list_entry(b 417 struct rxrpc_call *call = list_entry(bundle->waiting_calls.next, 418 s 418 struct rxrpc_call, wait_link); 419 u32 call_id = chan->call_counter + 1; 419 u32 call_id = chan->call_counter + 1; 420 420 421 _enter("C=%x,%u", conn->debug_id, chan 421 _enter("C=%x,%u", conn->debug_id, channel); 422 422 423 list_del_init(&call->wait_link); 423 list_del_init(&call->wait_link); 424 424 425 trace_rxrpc_client(conn, channel, rxrp 425 trace_rxrpc_client(conn, channel, rxrpc_client_chan_activate); 426 426 427 /* Cancel the final ACK on the previou 427 /* Cancel the final ACK on the previous call if it hasn't been sent yet 428 * as the DATA packet will implicitly 428 * as the DATA packet will implicitly ACK it. 429 */ 429 */ 430 clear_bit(RXRPC_CONN_FINAL_ACK_0 + cha 430 clear_bit(RXRPC_CONN_FINAL_ACK_0 + channel, &conn->flags); 431 clear_bit(conn->bundle_shift + channel 431 clear_bit(conn->bundle_shift + channel, &bundle->avail_chans); 432 432 433 rxrpc_see_call(call, rxrpc_call_see_ac 433 rxrpc_see_call(call, rxrpc_call_see_activate_client); 434 call->conn = rxrpc_get_connection 434 call->conn = rxrpc_get_connection(conn, rxrpc_conn_get_activate_call); 435 call->cid = conn->proto.cid | ch 435 call->cid = conn->proto.cid | channel; 436 call->call_id = call_id; 436 call->call_id = call_id; 437 call->dest_srx.srx_service = conn->ser 437 call->dest_srx.srx_service = conn->service_id; 438 call->cong_ssthresh = call->peer->cong 438 call->cong_ssthresh = call->peer->cong_ssthresh; 439 if (call->cong_cwnd >= call->cong_ssth 439 if (call->cong_cwnd >= call->cong_ssthresh) 440 call->cong_mode = RXRPC_CALL_C 440 call->cong_mode = RXRPC_CALL_CONGEST_AVOIDANCE; 441 else 441 else 442 call->cong_mode = RXRPC_CALL_S 442 call->cong_mode = RXRPC_CALL_SLOW_START; 443 443 444 chan->call_id = call_id; 444 chan->call_id = call_id; 445 chan->call_debug_id = call->debug_ 445 chan->call_debug_id = call->debug_id; 446 chan->call = call; 446 chan->call = call; 447 447 448 rxrpc_see_call(call, rxrpc_call_see_co 448 rxrpc_see_call(call, rxrpc_call_see_connected); 449 trace_rxrpc_connect_call(call); 449 trace_rxrpc_connect_call(call); 450 call->tx_last_sent = ktime_get_real(); 450 call->tx_last_sent = ktime_get_real(); 451 rxrpc_start_call_timer(call); 451 rxrpc_start_call_timer(call); 452 rxrpc_set_call_state(call, RXRPC_CALL_ 452 rxrpc_set_call_state(call, RXRPC_CALL_CLIENT_SEND_REQUEST); 453 wake_up(&call->waitq); 453 wake_up(&call->waitq); 454 } 454 } 455 455 456 /* 456 /* 457 * Remove a connection from the idle list if i 457 * Remove a connection from the idle list if it's on it. 458 */ 458 */ 459 static void rxrpc_unidle_conn(struct rxrpc_con 459 static void rxrpc_unidle_conn(struct rxrpc_connection *conn) 460 { 460 { 461 if (!list_empty(&conn->cache_link)) { 461 if (!list_empty(&conn->cache_link)) { 462 list_del_init(&conn->cache_lin 462 list_del_init(&conn->cache_link); 463 rxrpc_put_connection(conn, rxr 463 rxrpc_put_connection(conn, rxrpc_conn_put_unidle); 464 } 464 } 465 } 465 } 466 466 467 /* 467 /* 468 * Assign channels and callNumbers to waiting 468 * Assign channels and callNumbers to waiting calls. 469 */ 469 */ 470 static void rxrpc_activate_channels(struct rxr 470 static void rxrpc_activate_channels(struct rxrpc_bundle *bundle) 471 { 471 { 472 struct rxrpc_connection *conn; 472 struct rxrpc_connection *conn; 473 unsigned long avail, mask; 473 unsigned long avail, mask; 474 unsigned int channel, slot; 474 unsigned int channel, slot; 475 475 476 trace_rxrpc_client(NULL, -1, rxrpc_cli 476 trace_rxrpc_client(NULL, -1, rxrpc_client_activate_chans); 477 477 478 if (bundle->try_upgrade) 478 if (bundle->try_upgrade) 479 mask = 1; 479 mask = 1; 480 else 480 else 481 mask = ULONG_MAX; 481 mask = ULONG_MAX; 482 482 483 while (!list_empty(&bundle->waiting_ca 483 while (!list_empty(&bundle->waiting_calls)) { 484 avail = bundle->avail_chans & 484 avail = bundle->avail_chans & mask; 485 if (!avail) 485 if (!avail) 486 break; 486 break; 487 channel = __ffs(avail); 487 channel = __ffs(avail); 488 clear_bit(channel, &bundle->av 488 clear_bit(channel, &bundle->avail_chans); 489 489 490 slot = channel / RXRPC_MAXCALL 490 slot = channel / RXRPC_MAXCALLS; 491 conn = bundle->conns[slot]; 491 conn = bundle->conns[slot]; 492 if (!conn) 492 if (!conn) 493 break; 493 break; 494 494 495 if (bundle->try_upgrade) 495 if (bundle->try_upgrade) 496 set_bit(RXRPC_CONN_PRO 496 set_bit(RXRPC_CONN_PROBING_FOR_UPGRADE, &conn->flags); 497 rxrpc_unidle_conn(conn); 497 rxrpc_unidle_conn(conn); 498 498 499 channel &= (RXRPC_MAXCALLS - 1 499 channel &= (RXRPC_MAXCALLS - 1); 500 conn->act_chans |= 1 << channe 500 conn->act_chans |= 1 << channel; 501 rxrpc_activate_one_channel(con 501 rxrpc_activate_one_channel(conn, channel); 502 } 502 } 503 } 503 } 504 504 505 /* 505 /* 506 * Connect waiting channels (called from the I 506 * Connect waiting channels (called from the I/O thread). 507 */ 507 */ 508 void rxrpc_connect_client_calls(struct rxrpc_l 508 void rxrpc_connect_client_calls(struct rxrpc_local *local) 509 { 509 { 510 struct rxrpc_call *call; 510 struct rxrpc_call *call; 511 511 512 while ((call = list_first_entry_or_nul 512 while ((call = list_first_entry_or_null(&local->new_client_calls, 513 513 struct rxrpc_call, wait_link)) 514 ) { 514 ) { 515 struct rxrpc_bundle *bundle = 515 struct rxrpc_bundle *bundle = call->bundle; 516 516 517 spin_lock(&local->client_call_ 517 spin_lock(&local->client_call_lock); 518 list_move_tail(&call->wait_lin 518 list_move_tail(&call->wait_link, &bundle->waiting_calls); 519 rxrpc_see_call(call, rxrpc_cal 519 rxrpc_see_call(call, rxrpc_call_see_waiting_call); 520 spin_unlock(&local->client_cal 520 spin_unlock(&local->client_call_lock); 521 521 522 if (rxrpc_bundle_has_space(bun 522 if (rxrpc_bundle_has_space(bundle)) 523 rxrpc_activate_channel 523 rxrpc_activate_channels(bundle); 524 } 524 } 525 } 525 } 526 526 527 /* 527 /* 528 * Note that a call, and thus a connection, is 528 * Note that a call, and thus a connection, is about to be exposed to the 529 * world. 529 * world. 530 */ 530 */ 531 void rxrpc_expose_client_call(struct rxrpc_cal 531 void rxrpc_expose_client_call(struct rxrpc_call *call) 532 { 532 { 533 unsigned int channel = call->cid & RXR 533 unsigned int channel = call->cid & RXRPC_CHANNELMASK; 534 struct rxrpc_connection *conn = call-> 534 struct rxrpc_connection *conn = call->conn; 535 struct rxrpc_channel *chan = &conn->ch 535 struct rxrpc_channel *chan = &conn->channels[channel]; 536 536 537 if (!test_and_set_bit(RXRPC_CALL_EXPOS 537 if (!test_and_set_bit(RXRPC_CALL_EXPOSED, &call->flags)) { 538 /* Mark the call ID as being u 538 /* Mark the call ID as being used. If the callNumber counter 539 * exceeds ~2 billion, we kill 539 * exceeds ~2 billion, we kill the connection after its 540 * outstanding calls have fini 540 * outstanding calls have finished so that the counter doesn't 541 * wrap. 541 * wrap. 542 */ 542 */ 543 chan->call_counter++; 543 chan->call_counter++; 544 if (chan->call_counter >= INT_ 544 if (chan->call_counter >= INT_MAX) 545 set_bit(RXRPC_CONN_DON 545 set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags); 546 trace_rxrpc_client(conn, chann 546 trace_rxrpc_client(conn, channel, rxrpc_client_exposed); 547 547 548 spin_lock(&call->peer->lock); 548 spin_lock(&call->peer->lock); 549 hlist_add_head(&call->error_li 549 hlist_add_head(&call->error_link, &call->peer->error_targets); 550 spin_unlock(&call->peer->lock) 550 spin_unlock(&call->peer->lock); 551 } 551 } 552 } 552 } 553 553 554 /* 554 /* 555 * Set the reap timer. 555 * Set the reap timer. 556 */ 556 */ 557 static void rxrpc_set_client_reap_timer(struct 557 static void rxrpc_set_client_reap_timer(struct rxrpc_local *local) 558 { 558 { 559 if (!local->kill_all_client_conns) { 559 if (!local->kill_all_client_conns) { 560 unsigned long now = jiffies; 560 unsigned long now = jiffies; 561 unsigned long reap_at = now + 561 unsigned long reap_at = now + rxrpc_conn_idle_client_expiry; 562 562 563 if (local->rxnet->live) 563 if (local->rxnet->live) 564 timer_reduce(&local->c 564 timer_reduce(&local->client_conn_reap_timer, reap_at); 565 } 565 } 566 } 566 } 567 567 568 /* 568 /* 569 * Disconnect a client call. 569 * Disconnect a client call. 570 */ 570 */ 571 void rxrpc_disconnect_client_call(struct rxrpc 571 void rxrpc_disconnect_client_call(struct rxrpc_bundle *bundle, struct rxrpc_call *call) 572 { 572 { 573 struct rxrpc_connection *conn; 573 struct rxrpc_connection *conn; 574 struct rxrpc_channel *chan = NULL; 574 struct rxrpc_channel *chan = NULL; 575 struct rxrpc_local *local = bundle->lo 575 struct rxrpc_local *local = bundle->local; 576 unsigned int channel; 576 unsigned int channel; 577 bool may_reuse; 577 bool may_reuse; 578 u32 cid; 578 u32 cid; 579 579 580 _enter("c=%x", call->debug_id); 580 _enter("c=%x", call->debug_id); 581 581 582 /* Calls that have never actually been 582 /* Calls that have never actually been assigned a channel can simply be 583 * discarded. 583 * discarded. 584 */ 584 */ 585 conn = call->conn; 585 conn = call->conn; 586 if (!conn) { 586 if (!conn) { 587 _debug("call is waiting"); 587 _debug("call is waiting"); 588 ASSERTCMP(call->call_id, ==, 0 588 ASSERTCMP(call->call_id, ==, 0); 589 ASSERT(!test_bit(RXRPC_CALL_EX 589 ASSERT(!test_bit(RXRPC_CALL_EXPOSED, &call->flags)); 590 /* May still be on ->new_clien 590 /* May still be on ->new_client_calls. */ 591 spin_lock(&local->client_call_ 591 spin_lock(&local->client_call_lock); 592 list_del_init(&call->wait_link 592 list_del_init(&call->wait_link); 593 spin_unlock(&local->client_cal 593 spin_unlock(&local->client_call_lock); 594 return; 594 return; 595 } 595 } 596 596 597 cid = call->cid; 597 cid = call->cid; 598 channel = cid & RXRPC_CHANNELMASK; 598 channel = cid & RXRPC_CHANNELMASK; 599 chan = &conn->channels[channel]; 599 chan = &conn->channels[channel]; 600 trace_rxrpc_client(conn, channel, rxrp 600 trace_rxrpc_client(conn, channel, rxrpc_client_chan_disconnect); 601 601 602 if (WARN_ON(chan->call != call)) 602 if (WARN_ON(chan->call != call)) 603 return; 603 return; 604 604 605 may_reuse = rxrpc_may_reuse_conn(conn) 605 may_reuse = rxrpc_may_reuse_conn(conn); 606 606 607 /* If a client call was exposed to the 607 /* If a client call was exposed to the world, we save the result for 608 * retransmission. 608 * retransmission. 609 * 609 * 610 * We use a barrier here so that the c 610 * We use a barrier here so that the call number and abort code can be 611 * read without needing to take a lock 611 * read without needing to take a lock. 612 * 612 * 613 * TODO: Make the incoming packet hand 613 * TODO: Make the incoming packet handler check this and handle 614 * terminal retransmission without req 614 * terminal retransmission without requiring access to the call. 615 */ 615 */ 616 if (test_bit(RXRPC_CALL_EXPOSED, &call 616 if (test_bit(RXRPC_CALL_EXPOSED, &call->flags)) { 617 _debug("exposed %u,%u", call-> 617 _debug("exposed %u,%u", call->call_id, call->abort_code); 618 __rxrpc_disconnect_call(conn, 618 __rxrpc_disconnect_call(conn, call); 619 619 620 if (test_and_clear_bit(RXRPC_C 620 if (test_and_clear_bit(RXRPC_CONN_PROBING_FOR_UPGRADE, &conn->flags)) { 621 trace_rxrpc_client(con 621 trace_rxrpc_client(conn, channel, rxrpc_client_to_active); 622 bundle->try_upgrade = 622 bundle->try_upgrade = false; 623 if (may_reuse) 623 if (may_reuse) 624 rxrpc_activate 624 rxrpc_activate_channels(bundle); 625 } 625 } 626 } 626 } 627 627 628 /* See if we can pass the channel dire 628 /* See if we can pass the channel directly to another call. */ 629 if (may_reuse && !list_empty(&bundle-> 629 if (may_reuse && !list_empty(&bundle->waiting_calls)) { 630 trace_rxrpc_client(conn, chann 630 trace_rxrpc_client(conn, channel, rxrpc_client_chan_pass); 631 rxrpc_activate_one_channel(con 631 rxrpc_activate_one_channel(conn, channel); 632 return; 632 return; 633 } 633 } 634 634 635 /* Schedule the final ACK to be transm 635 /* Schedule the final ACK to be transmitted in a short while so that it 636 * can be skipped if we find a follow- 636 * can be skipped if we find a follow-on call. The first DATA packet 637 * of the follow on call will implicit 637 * of the follow on call will implicitly ACK this call. 638 */ 638 */ 639 if (call->completion == RXRPC_CALL_SUC 639 if (call->completion == RXRPC_CALL_SUCCEEDED && 640 test_bit(RXRPC_CALL_EXPOSED, &call 640 test_bit(RXRPC_CALL_EXPOSED, &call->flags)) { 641 unsigned long final_ack_at = j 641 unsigned long final_ack_at = jiffies + 2; 642 642 643 chan->final_ack_at = final_ack 643 chan->final_ack_at = final_ack_at; 644 smp_wmb(); /* vs rxrpc_process 644 smp_wmb(); /* vs rxrpc_process_delayed_final_acks() */ 645 set_bit(RXRPC_CONN_FINAL_ACK_0 645 set_bit(RXRPC_CONN_FINAL_ACK_0 + channel, &conn->flags); 646 rxrpc_reduce_conn_timer(conn, 646 rxrpc_reduce_conn_timer(conn, final_ack_at); 647 } 647 } 648 648 649 /* Deactivate the channel. */ 649 /* Deactivate the channel. */ 650 chan->call = NULL; 650 chan->call = NULL; 651 set_bit(conn->bundle_shift + channel, 651 set_bit(conn->bundle_shift + channel, &conn->bundle->avail_chans); 652 conn->act_chans &= ~(1 << channel); 652 conn->act_chans &= ~(1 << channel); 653 653 654 /* If no channels remain active, then 654 /* If no channels remain active, then put the connection on the idle 655 * list for a short while. Give it a 655 * list for a short while. Give it a ref to stop it going away if it 656 * becomes unbundled. 656 * becomes unbundled. 657 */ 657 */ 658 if (!conn->act_chans) { 658 if (!conn->act_chans) { 659 trace_rxrpc_client(conn, chann 659 trace_rxrpc_client(conn, channel, rxrpc_client_to_idle); 660 conn->idle_timestamp = jiffies 660 conn->idle_timestamp = jiffies; 661 661 662 rxrpc_get_connection(conn, rxr 662 rxrpc_get_connection(conn, rxrpc_conn_get_idle); 663 list_move_tail(&conn->cache_li 663 list_move_tail(&conn->cache_link, &local->idle_client_conns); 664 664 665 rxrpc_set_client_reap_timer(lo 665 rxrpc_set_client_reap_timer(local); 666 } 666 } 667 } 667 } 668 668 669 /* 669 /* 670 * Remove a connection from a bundle. 670 * Remove a connection from a bundle. 671 */ 671 */ 672 static void rxrpc_unbundle_conn(struct rxrpc_c 672 static void rxrpc_unbundle_conn(struct rxrpc_connection *conn) 673 { 673 { 674 struct rxrpc_bundle *bundle = conn->bu 674 struct rxrpc_bundle *bundle = conn->bundle; 675 unsigned int bindex; 675 unsigned int bindex; 676 int i; 676 int i; 677 677 678 _enter("C=%x", conn->debug_id); 678 _enter("C=%x", conn->debug_id); 679 679 680 if (conn->flags & RXRPC_CONN_FINAL_ACK 680 if (conn->flags & RXRPC_CONN_FINAL_ACK_MASK) 681 rxrpc_process_delayed_final_ac 681 rxrpc_process_delayed_final_acks(conn, true); 682 682 683 bindex = conn->bundle_shift / RXRPC_MA 683 bindex = conn->bundle_shift / RXRPC_MAXCALLS; 684 if (bundle->conns[bindex] == conn) { 684 if (bundle->conns[bindex] == conn) { 685 _debug("clear slot %u", bindex 685 _debug("clear slot %u", bindex); 686 bundle->conns[bindex] = NULL; 686 bundle->conns[bindex] = NULL; 687 bundle->conn_ids[bindex] = 0; 687 bundle->conn_ids[bindex] = 0; 688 for (i = 0; i < RXRPC_MAXCALLS 688 for (i = 0; i < RXRPC_MAXCALLS; i++) 689 clear_bit(conn->bundle 689 clear_bit(conn->bundle_shift + i, &bundle->avail_chans); 690 rxrpc_put_client_connection_id 690 rxrpc_put_client_connection_id(bundle->local, conn); 691 rxrpc_deactivate_bundle(bundle 691 rxrpc_deactivate_bundle(bundle); 692 rxrpc_put_connection(conn, rxr 692 rxrpc_put_connection(conn, rxrpc_conn_put_unbundle); 693 } 693 } 694 } 694 } 695 695 696 /* 696 /* 697 * Drop the active count on a bundle. 697 * Drop the active count on a bundle. 698 */ 698 */ 699 void rxrpc_deactivate_bundle(struct rxrpc_bund 699 void rxrpc_deactivate_bundle(struct rxrpc_bundle *bundle) 700 { 700 { 701 struct rxrpc_local *local; 701 struct rxrpc_local *local; 702 bool need_put = false; 702 bool need_put = false; 703 703 704 if (!bundle) 704 if (!bundle) 705 return; 705 return; 706 706 707 local = bundle->local; 707 local = bundle->local; 708 if (atomic_dec_and_lock(&bundle->activ 708 if (atomic_dec_and_lock(&bundle->active, &local->client_bundles_lock)) { 709 if (!bundle->exclusive) { 709 if (!bundle->exclusive) { 710 _debug("erase bundle") 710 _debug("erase bundle"); 711 rb_erase(&bundle->loca 711 rb_erase(&bundle->local_node, &local->client_bundles); 712 need_put = true; 712 need_put = true; 713 } 713 } 714 714 715 spin_unlock(&local->client_bun 715 spin_unlock(&local->client_bundles_lock); 716 if (need_put) 716 if (need_put) 717 rxrpc_put_bundle(bundl 717 rxrpc_put_bundle(bundle, rxrpc_bundle_put_discard); 718 } 718 } 719 } 719 } 720 720 721 /* 721 /* 722 * Clean up a dead client connection. 722 * Clean up a dead client connection. 723 */ 723 */ 724 void rxrpc_kill_client_conn(struct rxrpc_conne 724 void rxrpc_kill_client_conn(struct rxrpc_connection *conn) 725 { 725 { 726 struct rxrpc_local *local = conn->loca 726 struct rxrpc_local *local = conn->local; 727 struct rxrpc_net *rxnet = local->rxnet 727 struct rxrpc_net *rxnet = local->rxnet; 728 728 729 _enter("C=%x", conn->debug_id); 729 _enter("C=%x", conn->debug_id); 730 730 731 trace_rxrpc_client(conn, -1, rxrpc_cli 731 trace_rxrpc_client(conn, -1, rxrpc_client_cleanup); 732 atomic_dec(&rxnet->nr_client_conns); 732 atomic_dec(&rxnet->nr_client_conns); 733 733 734 rxrpc_put_client_connection_id(local, 734 rxrpc_put_client_connection_id(local, conn); 735 } 735 } 736 736 737 /* 737 /* 738 * Discard expired client connections from the 738 * Discard expired client connections from the idle list. Each conn in the 739 * idle list has been exposed and holds an ext 739 * idle list has been exposed and holds an extra ref because of that. 740 * 740 * 741 * This may be called from conn setup or from 741 * This may be called from conn setup or from a work item so cannot be 742 * considered non-reentrant. 742 * considered non-reentrant. 743 */ 743 */ 744 void rxrpc_discard_expired_client_conns(struct 744 void rxrpc_discard_expired_client_conns(struct rxrpc_local *local) 745 { 745 { 746 struct rxrpc_connection *conn; 746 struct rxrpc_connection *conn; 747 unsigned long expiry, conn_expires_at, 747 unsigned long expiry, conn_expires_at, now; 748 unsigned int nr_conns; 748 unsigned int nr_conns; 749 749 750 _enter(""); 750 _enter(""); 751 751 752 /* We keep an estimate of what the num 752 /* We keep an estimate of what the number of conns ought to be after 753 * we've discarded some so that we don 753 * we've discarded some so that we don't overdo the discarding. 754 */ 754 */ 755 nr_conns = atomic_read(&local->rxnet-> 755 nr_conns = atomic_read(&local->rxnet->nr_client_conns); 756 756 757 next: 757 next: 758 conn = list_first_entry_or_null(&local 758 conn = list_first_entry_or_null(&local->idle_client_conns, 759 struct 759 struct rxrpc_connection, cache_link); 760 if (!conn) 760 if (!conn) 761 return; 761 return; 762 762 763 if (!local->kill_all_client_conns) { 763 if (!local->kill_all_client_conns) { 764 /* If the number of connection 764 /* If the number of connections is over the reap limit, we 765 * expedite discard by reducin 765 * expedite discard by reducing the expiry timeout. We must, 766 * however, have at least a sh 766 * however, have at least a short grace period to be able to do 767 * final-ACK or ABORT retransm 767 * final-ACK or ABORT retransmission. 768 */ 768 */ 769 expiry = rxrpc_conn_idle_clien 769 expiry = rxrpc_conn_idle_client_expiry; 770 if (nr_conns > rxrpc_reap_clie 770 if (nr_conns > rxrpc_reap_client_connections) 771 expiry = rxrpc_conn_id 771 expiry = rxrpc_conn_idle_client_fast_expiry; 772 if (conn->local->service_close 772 if (conn->local->service_closed) 773 expiry = rxrpc_closed_ 773 expiry = rxrpc_closed_conn_expiry * HZ; 774 774 775 conn_expires_at = conn->idle_t 775 conn_expires_at = conn->idle_timestamp + expiry; 776 776 777 now = jiffies; 777 now = jiffies; 778 if (time_after(conn_expires_at 778 if (time_after(conn_expires_at, now)) 779 goto not_yet_expired; 779 goto not_yet_expired; 780 } 780 } 781 781 782 atomic_dec(&conn->active); 782 atomic_dec(&conn->active); 783 trace_rxrpc_client(conn, -1, rxrpc_cli 783 trace_rxrpc_client(conn, -1, rxrpc_client_discard); 784 list_del_init(&conn->cache_link); 784 list_del_init(&conn->cache_link); 785 785 786 rxrpc_unbundle_conn(conn); 786 rxrpc_unbundle_conn(conn); 787 /* Drop the ->cache_link ref */ 787 /* Drop the ->cache_link ref */ 788 rxrpc_put_connection(conn, rxrpc_conn_ 788 rxrpc_put_connection(conn, rxrpc_conn_put_discard_idle); 789 789 790 nr_conns--; 790 nr_conns--; 791 goto next; 791 goto next; 792 792 793 not_yet_expired: 793 not_yet_expired: 794 /* The connection at the front of the 794 /* The connection at the front of the queue hasn't yet expired, so 795 * schedule the work item for that poi 795 * schedule the work item for that point if we discarded something. 796 * 796 * 797 * We don't worry if the work item is 797 * We don't worry if the work item is already scheduled - it can look 798 * after rescheduling itself at a late 798 * after rescheduling itself at a later time. We could cancel it, but 799 * then things get messier. 799 * then things get messier. 800 */ 800 */ 801 _debug("not yet"); 801 _debug("not yet"); 802 if (!local->kill_all_client_conns) 802 if (!local->kill_all_client_conns) 803 timer_reduce(&local->client_co 803 timer_reduce(&local->client_conn_reap_timer, conn_expires_at); 804 804 805 _leave(""); 805 _leave(""); 806 } 806 } 807 807 808 /* 808 /* 809 * Clean up the client connections on a local 809 * Clean up the client connections on a local endpoint. 810 */ 810 */ 811 void rxrpc_clean_up_local_conns(struct rxrpc_l 811 void rxrpc_clean_up_local_conns(struct rxrpc_local *local) 812 { 812 { 813 struct rxrpc_connection *conn; 813 struct rxrpc_connection *conn; 814 814 815 _enter(""); 815 _enter(""); 816 816 817 local->kill_all_client_conns = true; 817 local->kill_all_client_conns = true; 818 818 819 del_timer_sync(&local->client_conn_rea 819 del_timer_sync(&local->client_conn_reap_timer); 820 820 821 while ((conn = list_first_entry_or_nul 821 while ((conn = list_first_entry_or_null(&local->idle_client_conns, 822 822 struct rxrpc_connection, cache_link))) { 823 list_del_init(&conn->cache_lin 823 list_del_init(&conn->cache_link); 824 atomic_dec(&conn->active); 824 atomic_dec(&conn->active); 825 trace_rxrpc_client(conn, -1, r 825 trace_rxrpc_client(conn, -1, rxrpc_client_discard); 826 rxrpc_unbundle_conn(conn); 826 rxrpc_unbundle_conn(conn); 827 rxrpc_put_connection(conn, rxr 827 rxrpc_put_connection(conn, rxrpc_conn_put_local_dead); 828 } 828 } 829 829 830 _leave(" [culled]"); 830 _leave(" [culled]"); 831 } 831 } 832 832
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.