162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-or-later
262306a36Sopenharmony_ci/* Client connection-specific management code.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Copyright (C) 2016, 2020 Red Hat, Inc. All Rights Reserved.
562306a36Sopenharmony_ci * Written by David Howells (dhowells@redhat.com)
662306a36Sopenharmony_ci *
762306a36Sopenharmony_ci * Client connections need to be cached for a little while after they've made a
862306a36Sopenharmony_ci * call so as to handle retransmitted DATA packets in case the server didn't
962306a36Sopenharmony_ci * receive the final ACK or terminating ABORT we sent it.
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * There are flags of relevance to the cache:
1262306a36Sopenharmony_ci *
1362306a36Sopenharmony_ci *  (2) DONT_REUSE - The connection should be discarded as soon as possible and
1462306a36Sopenharmony_ci *      should not be reused.  This is set when an exclusive connection is used
1562306a36Sopenharmony_ci *      or a call ID counter overflows.
1662306a36Sopenharmony_ci *
1762306a36Sopenharmony_ci * The caching state may only be changed if the cache lock is held.
1862306a36Sopenharmony_ci *
1962306a36Sopenharmony_ci * There are two idle client connection expiry durations.  If the total number
2062306a36Sopenharmony_ci * of connections is below the reap threshold, we use the normal duration; if
2162306a36Sopenharmony_ci * it's above, we use the fast duration.
2262306a36Sopenharmony_ci */
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_ci#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
2562306a36Sopenharmony_ci
2662306a36Sopenharmony_ci#include <linux/slab.h>
2762306a36Sopenharmony_ci#include <linux/idr.h>
2862306a36Sopenharmony_ci#include <linux/timer.h>
2962306a36Sopenharmony_ci#include <linux/sched/signal.h>
3062306a36Sopenharmony_ci
3162306a36Sopenharmony_ci#include "ar-internal.h"
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci__read_mostly unsigned int rxrpc_reap_client_connections = 900;
3462306a36Sopenharmony_ci__read_mostly unsigned long rxrpc_conn_idle_client_expiry = 2 * 60 * HZ;
3562306a36Sopenharmony_ci__read_mostly unsigned long rxrpc_conn_idle_client_fast_expiry = 2 * HZ;
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_cistatic void rxrpc_activate_bundle(struct rxrpc_bundle *bundle)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	atomic_inc(&bundle->active);
4062306a36Sopenharmony_ci}
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_ci/*
4362306a36Sopenharmony_ci * Release a connection ID for a client connection.
4462306a36Sopenharmony_ci */
4562306a36Sopenharmony_cistatic void rxrpc_put_client_connection_id(struct rxrpc_local *local,
4662306a36Sopenharmony_ci					   struct rxrpc_connection *conn)
4762306a36Sopenharmony_ci{
4862306a36Sopenharmony_ci	idr_remove(&local->conn_ids, conn->proto.cid >> RXRPC_CIDSHIFT);
4962306a36Sopenharmony_ci}
5062306a36Sopenharmony_ci
5162306a36Sopenharmony_ci/*
5262306a36Sopenharmony_ci * Destroy the client connection ID tree.
5362306a36Sopenharmony_ci */
5462306a36Sopenharmony_cistatic void rxrpc_destroy_client_conn_ids(struct rxrpc_local *local)
5562306a36Sopenharmony_ci{
5662306a36Sopenharmony_ci	struct rxrpc_connection *conn;
5762306a36Sopenharmony_ci	int id;
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	if (!idr_is_empty(&local->conn_ids)) {
6062306a36Sopenharmony_ci		idr_for_each_entry(&local->conn_ids, conn, id) {
6162306a36Sopenharmony_ci			pr_err("AF_RXRPC: Leaked client conn %p {%d}\n",
6262306a36Sopenharmony_ci			       conn, refcount_read(&conn->ref));
6362306a36Sopenharmony_ci		}
6462306a36Sopenharmony_ci		BUG();
6562306a36Sopenharmony_ci	}
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	idr_destroy(&local->conn_ids);
6862306a36Sopenharmony_ci}
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci/*
7162306a36Sopenharmony_ci * Allocate a connection bundle.
7262306a36Sopenharmony_ci */
7362306a36Sopenharmony_cistatic struct rxrpc_bundle *rxrpc_alloc_bundle(struct rxrpc_call *call,
7462306a36Sopenharmony_ci					       gfp_t gfp)
7562306a36Sopenharmony_ci{
7662306a36Sopenharmony_ci	static atomic_t rxrpc_bundle_id;
7762306a36Sopenharmony_ci	struct rxrpc_bundle *bundle;
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_ci	bundle = kzalloc(sizeof(*bundle), gfp);
8062306a36Sopenharmony_ci	if (bundle) {
8162306a36Sopenharmony_ci		bundle->local		= call->local;
8262306a36Sopenharmony_ci		bundle->peer		= rxrpc_get_peer(call->peer, rxrpc_peer_get_bundle);
8362306a36Sopenharmony_ci		bundle->key		= key_get(call->key);
8462306a36Sopenharmony_ci		bundle->security	= call->security;
8562306a36Sopenharmony_ci		bundle->exclusive	= test_bit(RXRPC_CALL_EXCLUSIVE, &call->flags);
8662306a36Sopenharmony_ci		bundle->upgrade		= test_bit(RXRPC_CALL_UPGRADE, &call->flags);
8762306a36Sopenharmony_ci		bundle->service_id	= call->dest_srx.srx_service;
8862306a36Sopenharmony_ci		bundle->security_level	= call->security_level;
8962306a36Sopenharmony_ci		bundle->debug_id	= atomic_inc_return(&rxrpc_bundle_id);
9062306a36Sopenharmony_ci		refcount_set(&bundle->ref, 1);
9162306a36Sopenharmony_ci		atomic_set(&bundle->active, 1);
9262306a36Sopenharmony_ci		INIT_LIST_HEAD(&bundle->waiting_calls);
9362306a36Sopenharmony_ci		trace_rxrpc_bundle(bundle->debug_id, 1, rxrpc_bundle_new);
9462306a36Sopenharmony_ci	}
9562306a36Sopenharmony_ci	return bundle;
9662306a36Sopenharmony_ci}
9762306a36Sopenharmony_ci
9862306a36Sopenharmony_cistruct rxrpc_bundle *rxrpc_get_bundle(struct rxrpc_bundle *bundle,
9962306a36Sopenharmony_ci				      enum rxrpc_bundle_trace why)
10062306a36Sopenharmony_ci{
10162306a36Sopenharmony_ci	int r;
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci	__refcount_inc(&bundle->ref, &r);
10462306a36Sopenharmony_ci	trace_rxrpc_bundle(bundle->debug_id, r + 1, why);
10562306a36Sopenharmony_ci	return bundle;
10662306a36Sopenharmony_ci}
10762306a36Sopenharmony_ci
10862306a36Sopenharmony_cistatic void rxrpc_free_bundle(struct rxrpc_bundle *bundle)
10962306a36Sopenharmony_ci{
11062306a36Sopenharmony_ci	trace_rxrpc_bundle(bundle->debug_id, refcount_read(&bundle->ref),
11162306a36Sopenharmony_ci			   rxrpc_bundle_free);
11262306a36Sopenharmony_ci	rxrpc_put_peer(bundle->peer, rxrpc_peer_put_bundle);
11362306a36Sopenharmony_ci	key_put(bundle->key);
11462306a36Sopenharmony_ci	kfree(bundle);
11562306a36Sopenharmony_ci}
11662306a36Sopenharmony_ci
11762306a36Sopenharmony_civoid rxrpc_put_bundle(struct rxrpc_bundle *bundle, enum rxrpc_bundle_trace why)
11862306a36Sopenharmony_ci{
11962306a36Sopenharmony_ci	unsigned int id;
12062306a36Sopenharmony_ci	bool dead;
12162306a36Sopenharmony_ci	int r;
12262306a36Sopenharmony_ci
12362306a36Sopenharmony_ci	if (bundle) {
12462306a36Sopenharmony_ci		id = bundle->debug_id;
12562306a36Sopenharmony_ci		dead = __refcount_dec_and_test(&bundle->ref, &r);
12662306a36Sopenharmony_ci		trace_rxrpc_bundle(id, r - 1, why);
12762306a36Sopenharmony_ci		if (dead)
12862306a36Sopenharmony_ci			rxrpc_free_bundle(bundle);
12962306a36Sopenharmony_ci	}
13062306a36Sopenharmony_ci}
13162306a36Sopenharmony_ci
13262306a36Sopenharmony_ci/*
13362306a36Sopenharmony_ci * Get rid of outstanding client connection preallocations when a local
13462306a36Sopenharmony_ci * endpoint is destroyed.
13562306a36Sopenharmony_ci */
13662306a36Sopenharmony_civoid rxrpc_purge_client_connections(struct rxrpc_local *local)
13762306a36Sopenharmony_ci{
13862306a36Sopenharmony_ci	rxrpc_destroy_client_conn_ids(local);
13962306a36Sopenharmony_ci}
14062306a36Sopenharmony_ci
14162306a36Sopenharmony_ci/*
14262306a36Sopenharmony_ci * Allocate a client connection.
14362306a36Sopenharmony_ci */
14462306a36Sopenharmony_cistatic struct rxrpc_connection *
14562306a36Sopenharmony_cirxrpc_alloc_client_connection(struct rxrpc_bundle *bundle)
14662306a36Sopenharmony_ci{
14762306a36Sopenharmony_ci	struct rxrpc_connection *conn;
14862306a36Sopenharmony_ci	struct rxrpc_local *local = bundle->local;
14962306a36Sopenharmony_ci	struct rxrpc_net *rxnet = local->rxnet;
15062306a36Sopenharmony_ci	int id;
15162306a36Sopenharmony_ci
15262306a36Sopenharmony_ci	_enter("");
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_ci	conn = rxrpc_alloc_connection(rxnet, GFP_ATOMIC | __GFP_NOWARN);
15562306a36Sopenharmony_ci	if (!conn)
15662306a36Sopenharmony_ci		return ERR_PTR(-ENOMEM);
15762306a36Sopenharmony_ci
15862306a36Sopenharmony_ci	id = idr_alloc_cyclic(&local->conn_ids, conn, 1, 0x40000000,
15962306a36Sopenharmony_ci			      GFP_ATOMIC | __GFP_NOWARN);
16062306a36Sopenharmony_ci	if (id < 0) {
16162306a36Sopenharmony_ci		kfree(conn);
16262306a36Sopenharmony_ci		return ERR_PTR(id);
16362306a36Sopenharmony_ci	}
16462306a36Sopenharmony_ci
16562306a36Sopenharmony_ci	refcount_set(&conn->ref, 1);
16662306a36Sopenharmony_ci	conn->proto.cid		= id << RXRPC_CIDSHIFT;
16762306a36Sopenharmony_ci	conn->proto.epoch	= local->rxnet->epoch;
16862306a36Sopenharmony_ci	conn->out_clientflag	= RXRPC_CLIENT_INITIATED;
16962306a36Sopenharmony_ci	conn->bundle		= rxrpc_get_bundle(bundle, rxrpc_bundle_get_client_conn);
17062306a36Sopenharmony_ci	conn->local		= rxrpc_get_local(bundle->local, rxrpc_local_get_client_conn);
17162306a36Sopenharmony_ci	conn->peer		= rxrpc_get_peer(bundle->peer, rxrpc_peer_get_client_conn);
17262306a36Sopenharmony_ci	conn->key		= key_get(bundle->key);
17362306a36Sopenharmony_ci	conn->security		= bundle->security;
17462306a36Sopenharmony_ci	conn->exclusive		= bundle->exclusive;
17562306a36Sopenharmony_ci	conn->upgrade		= bundle->upgrade;
17662306a36Sopenharmony_ci	conn->orig_service_id	= bundle->service_id;
17762306a36Sopenharmony_ci	conn->security_level	= bundle->security_level;
17862306a36Sopenharmony_ci	conn->state		= RXRPC_CONN_CLIENT_UNSECURED;
17962306a36Sopenharmony_ci	conn->service_id	= conn->orig_service_id;
18062306a36Sopenharmony_ci
18162306a36Sopenharmony_ci	if (conn->security == &rxrpc_no_security)
18262306a36Sopenharmony_ci		conn->state	= RXRPC_CONN_CLIENT;
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	atomic_inc(&rxnet->nr_conns);
18562306a36Sopenharmony_ci	write_lock(&rxnet->conn_lock);
18662306a36Sopenharmony_ci	list_add_tail(&conn->proc_link, &rxnet->conn_proc_list);
18762306a36Sopenharmony_ci	write_unlock(&rxnet->conn_lock);
18862306a36Sopenharmony_ci
18962306a36Sopenharmony_ci	rxrpc_see_connection(conn, rxrpc_conn_new_client);
19062306a36Sopenharmony_ci
19162306a36Sopenharmony_ci	atomic_inc(&rxnet->nr_client_conns);
19262306a36Sopenharmony_ci	trace_rxrpc_client(conn, -1, rxrpc_client_alloc);
19362306a36Sopenharmony_ci	return conn;
19462306a36Sopenharmony_ci}
19562306a36Sopenharmony_ci
19662306a36Sopenharmony_ci/*
19762306a36Sopenharmony_ci * Determine if a connection may be reused.
19862306a36Sopenharmony_ci */
19962306a36Sopenharmony_cistatic bool rxrpc_may_reuse_conn(struct rxrpc_connection *conn)
20062306a36Sopenharmony_ci{
20162306a36Sopenharmony_ci	struct rxrpc_net *rxnet;
20262306a36Sopenharmony_ci	int id_cursor, id, distance, limit;
20362306a36Sopenharmony_ci
20462306a36Sopenharmony_ci	if (!conn)
20562306a36Sopenharmony_ci		goto dont_reuse;
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	rxnet = conn->rxnet;
20862306a36Sopenharmony_ci	if (test_bit(RXRPC_CONN_DONT_REUSE, &conn->flags))
20962306a36Sopenharmony_ci		goto dont_reuse;
21062306a36Sopenharmony_ci
21162306a36Sopenharmony_ci	if ((conn->state != RXRPC_CONN_CLIENT_UNSECURED &&
21262306a36Sopenharmony_ci	     conn->state != RXRPC_CONN_CLIENT) ||
21362306a36Sopenharmony_ci	    conn->proto.epoch != rxnet->epoch)
21462306a36Sopenharmony_ci		goto mark_dont_reuse;
21562306a36Sopenharmony_ci
21662306a36Sopenharmony_ci	/* The IDR tree gets very expensive on memory if the connection IDs are
21762306a36Sopenharmony_ci	 * widely scattered throughout the number space, so we shall want to
21862306a36Sopenharmony_ci	 * kill off connections that, say, have an ID more than about four
21962306a36Sopenharmony_ci	 * times the maximum number of client conns away from the current
22062306a36Sopenharmony_ci	 * allocation point to try and keep the IDs concentrated.
22162306a36Sopenharmony_ci	 */
22262306a36Sopenharmony_ci	id_cursor = idr_get_cursor(&conn->local->conn_ids);
22362306a36Sopenharmony_ci	id = conn->proto.cid >> RXRPC_CIDSHIFT;
22462306a36Sopenharmony_ci	distance = id - id_cursor;
22562306a36Sopenharmony_ci	if (distance < 0)
22662306a36Sopenharmony_ci		distance = -distance;
22762306a36Sopenharmony_ci	limit = max_t(unsigned long, atomic_read(&rxnet->nr_conns) * 4, 1024);
22862306a36Sopenharmony_ci	if (distance > limit)
22962306a36Sopenharmony_ci		goto mark_dont_reuse;
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	return true;
23262306a36Sopenharmony_ci
23362306a36Sopenharmony_cimark_dont_reuse:
23462306a36Sopenharmony_ci	set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags);
23562306a36Sopenharmony_cidont_reuse:
23662306a36Sopenharmony_ci	return false;
23762306a36Sopenharmony_ci}
23862306a36Sopenharmony_ci
23962306a36Sopenharmony_ci/*
24062306a36Sopenharmony_ci * Look up the conn bundle that matches the connection parameters, adding it if
24162306a36Sopenharmony_ci * it doesn't yet exist.
24262306a36Sopenharmony_ci */
24362306a36Sopenharmony_ciint rxrpc_look_up_bundle(struct rxrpc_call *call, gfp_t gfp)
24462306a36Sopenharmony_ci{
24562306a36Sopenharmony_ci	struct rxrpc_bundle *bundle, *candidate;
24662306a36Sopenharmony_ci	struct rxrpc_local *local = call->local;
24762306a36Sopenharmony_ci	struct rb_node *p, **pp, *parent;
24862306a36Sopenharmony_ci	long diff;
24962306a36Sopenharmony_ci	bool upgrade = test_bit(RXRPC_CALL_UPGRADE, &call->flags);
25062306a36Sopenharmony_ci
25162306a36Sopenharmony_ci	_enter("{%px,%x,%u,%u}",
25262306a36Sopenharmony_ci	       call->peer, key_serial(call->key), call->security_level,
25362306a36Sopenharmony_ci	       upgrade);
25462306a36Sopenharmony_ci
25562306a36Sopenharmony_ci	if (test_bit(RXRPC_CALL_EXCLUSIVE, &call->flags)) {
25662306a36Sopenharmony_ci		call->bundle = rxrpc_alloc_bundle(call, gfp);
25762306a36Sopenharmony_ci		return call->bundle ? 0 : -ENOMEM;
25862306a36Sopenharmony_ci	}
25962306a36Sopenharmony_ci
26062306a36Sopenharmony_ci	/* First, see if the bundle is already there. */
26162306a36Sopenharmony_ci	_debug("search 1");
26262306a36Sopenharmony_ci	spin_lock(&local->client_bundles_lock);
26362306a36Sopenharmony_ci	p = local->client_bundles.rb_node;
26462306a36Sopenharmony_ci	while (p) {
26562306a36Sopenharmony_ci		bundle = rb_entry(p, struct rxrpc_bundle, local_node);
26662306a36Sopenharmony_ci
26762306a36Sopenharmony_ci#define cmp(X, Y) ((long)(X) - (long)(Y))
26862306a36Sopenharmony_ci		diff = (cmp(bundle->peer, call->peer) ?:
26962306a36Sopenharmony_ci			cmp(bundle->key, call->key) ?:
27062306a36Sopenharmony_ci			cmp(bundle->security_level, call->security_level) ?:
27162306a36Sopenharmony_ci			cmp(bundle->upgrade, upgrade));
27262306a36Sopenharmony_ci#undef cmp
27362306a36Sopenharmony_ci		if (diff < 0)
27462306a36Sopenharmony_ci			p = p->rb_left;
27562306a36Sopenharmony_ci		else if (diff > 0)
27662306a36Sopenharmony_ci			p = p->rb_right;
27762306a36Sopenharmony_ci		else
27862306a36Sopenharmony_ci			goto found_bundle;
27962306a36Sopenharmony_ci	}
28062306a36Sopenharmony_ci	spin_unlock(&local->client_bundles_lock);
28162306a36Sopenharmony_ci	_debug("not found");
28262306a36Sopenharmony_ci
28362306a36Sopenharmony_ci	/* It wasn't.  We need to add one. */
28462306a36Sopenharmony_ci	candidate = rxrpc_alloc_bundle(call, gfp);
28562306a36Sopenharmony_ci	if (!candidate)
28662306a36Sopenharmony_ci		return -ENOMEM;
28762306a36Sopenharmony_ci
28862306a36Sopenharmony_ci	_debug("search 2");
28962306a36Sopenharmony_ci	spin_lock(&local->client_bundles_lock);
29062306a36Sopenharmony_ci	pp = &local->client_bundles.rb_node;
29162306a36Sopenharmony_ci	parent = NULL;
29262306a36Sopenharmony_ci	while (*pp) {
29362306a36Sopenharmony_ci		parent = *pp;
29462306a36Sopenharmony_ci		bundle = rb_entry(parent, struct rxrpc_bundle, local_node);
29562306a36Sopenharmony_ci
29662306a36Sopenharmony_ci#define cmp(X, Y) ((long)(X) - (long)(Y))
29762306a36Sopenharmony_ci		diff = (cmp(bundle->peer, call->peer) ?:
29862306a36Sopenharmony_ci			cmp(bundle->key, call->key) ?:
29962306a36Sopenharmony_ci			cmp(bundle->security_level, call->security_level) ?:
30062306a36Sopenharmony_ci			cmp(bundle->upgrade, upgrade));
30162306a36Sopenharmony_ci#undef cmp
30262306a36Sopenharmony_ci		if (diff < 0)
30362306a36Sopenharmony_ci			pp = &(*pp)->rb_left;
30462306a36Sopenharmony_ci		else if (diff > 0)
30562306a36Sopenharmony_ci			pp = &(*pp)->rb_right;
30662306a36Sopenharmony_ci		else
30762306a36Sopenharmony_ci			goto found_bundle_free;
30862306a36Sopenharmony_ci	}
30962306a36Sopenharmony_ci
31062306a36Sopenharmony_ci	_debug("new bundle");
31162306a36Sopenharmony_ci	rb_link_node(&candidate->local_node, parent, pp);
31262306a36Sopenharmony_ci	rb_insert_color(&candidate->local_node, &local->client_bundles);
31362306a36Sopenharmony_ci	call->bundle = rxrpc_get_bundle(candidate, rxrpc_bundle_get_client_call);
31462306a36Sopenharmony_ci	spin_unlock(&local->client_bundles_lock);
31562306a36Sopenharmony_ci	_leave(" = B=%u [new]", call->bundle->debug_id);
31662306a36Sopenharmony_ci	return 0;
31762306a36Sopenharmony_ci
31862306a36Sopenharmony_cifound_bundle_free:
31962306a36Sopenharmony_ci	rxrpc_free_bundle(candidate);
32062306a36Sopenharmony_cifound_bundle:
32162306a36Sopenharmony_ci	call->bundle = rxrpc_get_bundle(bundle, rxrpc_bundle_get_client_call);
32262306a36Sopenharmony_ci	rxrpc_activate_bundle(bundle);
32362306a36Sopenharmony_ci	spin_unlock(&local->client_bundles_lock);
32462306a36Sopenharmony_ci	_leave(" = B=%u [found]", call->bundle->debug_id);
32562306a36Sopenharmony_ci	return 0;
32662306a36Sopenharmony_ci}
32762306a36Sopenharmony_ci
32862306a36Sopenharmony_ci/*
32962306a36Sopenharmony_ci * Allocate a new connection and add it into a bundle.
33062306a36Sopenharmony_ci */
33162306a36Sopenharmony_cistatic bool rxrpc_add_conn_to_bundle(struct rxrpc_bundle *bundle,
33262306a36Sopenharmony_ci				     unsigned int slot)
33362306a36Sopenharmony_ci{
33462306a36Sopenharmony_ci	struct rxrpc_connection *conn, *old;
33562306a36Sopenharmony_ci	unsigned int shift = slot * RXRPC_MAXCALLS;
33662306a36Sopenharmony_ci	unsigned int i;
33762306a36Sopenharmony_ci
33862306a36Sopenharmony_ci	old = bundle->conns[slot];
33962306a36Sopenharmony_ci	if (old) {
34062306a36Sopenharmony_ci		bundle->conns[slot] = NULL;
34162306a36Sopenharmony_ci		trace_rxrpc_client(old, -1, rxrpc_client_replace);
34262306a36Sopenharmony_ci		rxrpc_put_connection(old, rxrpc_conn_put_noreuse);
34362306a36Sopenharmony_ci	}
34462306a36Sopenharmony_ci
34562306a36Sopenharmony_ci	conn = rxrpc_alloc_client_connection(bundle);
34662306a36Sopenharmony_ci	if (IS_ERR(conn)) {
34762306a36Sopenharmony_ci		bundle->alloc_error = PTR_ERR(conn);
34862306a36Sopenharmony_ci		return false;
34962306a36Sopenharmony_ci	}
35062306a36Sopenharmony_ci
35162306a36Sopenharmony_ci	rxrpc_activate_bundle(bundle);
35262306a36Sopenharmony_ci	conn->bundle_shift = shift;
35362306a36Sopenharmony_ci	bundle->conns[slot] = conn;
35462306a36Sopenharmony_ci	for (i = 0; i < RXRPC_MAXCALLS; i++)
35562306a36Sopenharmony_ci		set_bit(shift + i, &bundle->avail_chans);
35662306a36Sopenharmony_ci	return true;
35762306a36Sopenharmony_ci}
35862306a36Sopenharmony_ci
35962306a36Sopenharmony_ci/*
36062306a36Sopenharmony_ci * Add a connection to a bundle if there are no usable connections or we have
36162306a36Sopenharmony_ci * connections waiting for extra capacity.
36262306a36Sopenharmony_ci */
36362306a36Sopenharmony_cistatic bool rxrpc_bundle_has_space(struct rxrpc_bundle *bundle)
36462306a36Sopenharmony_ci{
36562306a36Sopenharmony_ci	int slot = -1, i, usable;
36662306a36Sopenharmony_ci
36762306a36Sopenharmony_ci	_enter("");
36862306a36Sopenharmony_ci
36962306a36Sopenharmony_ci	bundle->alloc_error = 0;
37062306a36Sopenharmony_ci
37162306a36Sopenharmony_ci	/* See if there are any usable connections. */
37262306a36Sopenharmony_ci	usable = 0;
37362306a36Sopenharmony_ci	for (i = 0; i < ARRAY_SIZE(bundle->conns); i++) {
37462306a36Sopenharmony_ci		if (rxrpc_may_reuse_conn(bundle->conns[i]))
37562306a36Sopenharmony_ci			usable++;
37662306a36Sopenharmony_ci		else if (slot == -1)
37762306a36Sopenharmony_ci			slot = i;
37862306a36Sopenharmony_ci	}
37962306a36Sopenharmony_ci
38062306a36Sopenharmony_ci	if (!usable && bundle->upgrade)
38162306a36Sopenharmony_ci		bundle->try_upgrade = true;
38262306a36Sopenharmony_ci
38362306a36Sopenharmony_ci	if (!usable)
38462306a36Sopenharmony_ci		goto alloc_conn;
38562306a36Sopenharmony_ci
38662306a36Sopenharmony_ci	if (!bundle->avail_chans &&
38762306a36Sopenharmony_ci	    !bundle->try_upgrade &&
38862306a36Sopenharmony_ci	    usable < ARRAY_SIZE(bundle->conns))
38962306a36Sopenharmony_ci		goto alloc_conn;
39062306a36Sopenharmony_ci
39162306a36Sopenharmony_ci	_leave("");
39262306a36Sopenharmony_ci	return usable;
39362306a36Sopenharmony_ci
39462306a36Sopenharmony_cialloc_conn:
39562306a36Sopenharmony_ci	return slot >= 0 ? rxrpc_add_conn_to_bundle(bundle, slot) : false;
39662306a36Sopenharmony_ci}
39762306a36Sopenharmony_ci
39862306a36Sopenharmony_ci/*
39962306a36Sopenharmony_ci * Assign a channel to the call at the front of the queue and wake the call up.
40062306a36Sopenharmony_ci * We don't increment the callNumber counter until this number has been exposed
40162306a36Sopenharmony_ci * to the world.
40262306a36Sopenharmony_ci */
40362306a36Sopenharmony_cistatic void rxrpc_activate_one_channel(struct rxrpc_connection *conn,
40462306a36Sopenharmony_ci				       unsigned int channel)
40562306a36Sopenharmony_ci{
40662306a36Sopenharmony_ci	struct rxrpc_channel *chan = &conn->channels[channel];
40762306a36Sopenharmony_ci	struct rxrpc_bundle *bundle = conn->bundle;
40862306a36Sopenharmony_ci	struct rxrpc_call *call = list_entry(bundle->waiting_calls.next,
40962306a36Sopenharmony_ci					     struct rxrpc_call, wait_link);
41062306a36Sopenharmony_ci	u32 call_id = chan->call_counter + 1;
41162306a36Sopenharmony_ci
41262306a36Sopenharmony_ci	_enter("C=%x,%u", conn->debug_id, channel);
41362306a36Sopenharmony_ci
41462306a36Sopenharmony_ci	list_del_init(&call->wait_link);
41562306a36Sopenharmony_ci
41662306a36Sopenharmony_ci	trace_rxrpc_client(conn, channel, rxrpc_client_chan_activate);
41762306a36Sopenharmony_ci
41862306a36Sopenharmony_ci	/* Cancel the final ACK on the previous call if it hasn't been sent yet
41962306a36Sopenharmony_ci	 * as the DATA packet will implicitly ACK it.
42062306a36Sopenharmony_ci	 */
42162306a36Sopenharmony_ci	clear_bit(RXRPC_CONN_FINAL_ACK_0 + channel, &conn->flags);
42262306a36Sopenharmony_ci	clear_bit(conn->bundle_shift + channel, &bundle->avail_chans);
42362306a36Sopenharmony_ci
42462306a36Sopenharmony_ci	rxrpc_see_call(call, rxrpc_call_see_activate_client);
42562306a36Sopenharmony_ci	call->conn	= rxrpc_get_connection(conn, rxrpc_conn_get_activate_call);
42662306a36Sopenharmony_ci	call->cid	= conn->proto.cid | channel;
42762306a36Sopenharmony_ci	call->call_id	= call_id;
42862306a36Sopenharmony_ci	call->dest_srx.srx_service = conn->service_id;
42962306a36Sopenharmony_ci	call->cong_ssthresh = call->peer->cong_ssthresh;
43062306a36Sopenharmony_ci	if (call->cong_cwnd >= call->cong_ssthresh)
43162306a36Sopenharmony_ci		call->cong_mode = RXRPC_CALL_CONGEST_AVOIDANCE;
43262306a36Sopenharmony_ci	else
43362306a36Sopenharmony_ci		call->cong_mode = RXRPC_CALL_SLOW_START;
43462306a36Sopenharmony_ci
43562306a36Sopenharmony_ci	chan->call_id		= call_id;
43662306a36Sopenharmony_ci	chan->call_debug_id	= call->debug_id;
43762306a36Sopenharmony_ci	chan->call		= call;
43862306a36Sopenharmony_ci
43962306a36Sopenharmony_ci	rxrpc_see_call(call, rxrpc_call_see_connected);
44062306a36Sopenharmony_ci	trace_rxrpc_connect_call(call);
44162306a36Sopenharmony_ci	call->tx_last_sent = ktime_get_real();
44262306a36Sopenharmony_ci	rxrpc_start_call_timer(call);
44362306a36Sopenharmony_ci	rxrpc_set_call_state(call, RXRPC_CALL_CLIENT_SEND_REQUEST);
44462306a36Sopenharmony_ci	wake_up(&call->waitq);
44562306a36Sopenharmony_ci}
44662306a36Sopenharmony_ci
44762306a36Sopenharmony_ci/*
44862306a36Sopenharmony_ci * Remove a connection from the idle list if it's on it.
44962306a36Sopenharmony_ci */
45062306a36Sopenharmony_cistatic void rxrpc_unidle_conn(struct rxrpc_connection *conn)
45162306a36Sopenharmony_ci{
45262306a36Sopenharmony_ci	if (!list_empty(&conn->cache_link)) {
45362306a36Sopenharmony_ci		list_del_init(&conn->cache_link);
45462306a36Sopenharmony_ci		rxrpc_put_connection(conn, rxrpc_conn_put_unidle);
45562306a36Sopenharmony_ci	}
45662306a36Sopenharmony_ci}
45762306a36Sopenharmony_ci
45862306a36Sopenharmony_ci/*
45962306a36Sopenharmony_ci * Assign channels and callNumbers to waiting calls.
46062306a36Sopenharmony_ci */
46162306a36Sopenharmony_cistatic void rxrpc_activate_channels(struct rxrpc_bundle *bundle)
46262306a36Sopenharmony_ci{
46362306a36Sopenharmony_ci	struct rxrpc_connection *conn;
46462306a36Sopenharmony_ci	unsigned long avail, mask;
46562306a36Sopenharmony_ci	unsigned int channel, slot;
46662306a36Sopenharmony_ci
46762306a36Sopenharmony_ci	trace_rxrpc_client(NULL, -1, rxrpc_client_activate_chans);
46862306a36Sopenharmony_ci
46962306a36Sopenharmony_ci	if (bundle->try_upgrade)
47062306a36Sopenharmony_ci		mask = 1;
47162306a36Sopenharmony_ci	else
47262306a36Sopenharmony_ci		mask = ULONG_MAX;
47362306a36Sopenharmony_ci
47462306a36Sopenharmony_ci	while (!list_empty(&bundle->waiting_calls)) {
47562306a36Sopenharmony_ci		avail = bundle->avail_chans & mask;
47662306a36Sopenharmony_ci		if (!avail)
47762306a36Sopenharmony_ci			break;
47862306a36Sopenharmony_ci		channel = __ffs(avail);
47962306a36Sopenharmony_ci		clear_bit(channel, &bundle->avail_chans);
48062306a36Sopenharmony_ci
48162306a36Sopenharmony_ci		slot = channel / RXRPC_MAXCALLS;
48262306a36Sopenharmony_ci		conn = bundle->conns[slot];
48362306a36Sopenharmony_ci		if (!conn)
48462306a36Sopenharmony_ci			break;
48562306a36Sopenharmony_ci
48662306a36Sopenharmony_ci		if (bundle->try_upgrade)
48762306a36Sopenharmony_ci			set_bit(RXRPC_CONN_PROBING_FOR_UPGRADE, &conn->flags);
48862306a36Sopenharmony_ci		rxrpc_unidle_conn(conn);
48962306a36Sopenharmony_ci
49062306a36Sopenharmony_ci		channel &= (RXRPC_MAXCALLS - 1);
49162306a36Sopenharmony_ci		conn->act_chans	|= 1 << channel;
49262306a36Sopenharmony_ci		rxrpc_activate_one_channel(conn, channel);
49362306a36Sopenharmony_ci	}
49462306a36Sopenharmony_ci}
49562306a36Sopenharmony_ci
49662306a36Sopenharmony_ci/*
49762306a36Sopenharmony_ci * Connect waiting channels (called from the I/O thread).
49862306a36Sopenharmony_ci */
49962306a36Sopenharmony_civoid rxrpc_connect_client_calls(struct rxrpc_local *local)
50062306a36Sopenharmony_ci{
50162306a36Sopenharmony_ci	struct rxrpc_call *call;
50262306a36Sopenharmony_ci
50362306a36Sopenharmony_ci	while ((call = list_first_entry_or_null(&local->new_client_calls,
50462306a36Sopenharmony_ci						struct rxrpc_call, wait_link))
50562306a36Sopenharmony_ci	       ) {
50662306a36Sopenharmony_ci		struct rxrpc_bundle *bundle = call->bundle;
50762306a36Sopenharmony_ci
50862306a36Sopenharmony_ci		spin_lock(&local->client_call_lock);
50962306a36Sopenharmony_ci		list_move_tail(&call->wait_link, &bundle->waiting_calls);
51062306a36Sopenharmony_ci		spin_unlock(&local->client_call_lock);
51162306a36Sopenharmony_ci
51262306a36Sopenharmony_ci		if (rxrpc_bundle_has_space(bundle))
51362306a36Sopenharmony_ci			rxrpc_activate_channels(bundle);
51462306a36Sopenharmony_ci	}
51562306a36Sopenharmony_ci}
51662306a36Sopenharmony_ci
51762306a36Sopenharmony_ci/*
51862306a36Sopenharmony_ci * Note that a call, and thus a connection, is about to be exposed to the
51962306a36Sopenharmony_ci * world.
52062306a36Sopenharmony_ci */
52162306a36Sopenharmony_civoid rxrpc_expose_client_call(struct rxrpc_call *call)
52262306a36Sopenharmony_ci{
52362306a36Sopenharmony_ci	unsigned int channel = call->cid & RXRPC_CHANNELMASK;
52462306a36Sopenharmony_ci	struct rxrpc_connection *conn = call->conn;
52562306a36Sopenharmony_ci	struct rxrpc_channel *chan = &conn->channels[channel];
52662306a36Sopenharmony_ci
52762306a36Sopenharmony_ci	if (!test_and_set_bit(RXRPC_CALL_EXPOSED, &call->flags)) {
52862306a36Sopenharmony_ci		/* Mark the call ID as being used.  If the callNumber counter
52962306a36Sopenharmony_ci		 * exceeds ~2 billion, we kill the connection after its
53062306a36Sopenharmony_ci		 * outstanding calls have finished so that the counter doesn't
53162306a36Sopenharmony_ci		 * wrap.
53262306a36Sopenharmony_ci		 */
53362306a36Sopenharmony_ci		chan->call_counter++;
53462306a36Sopenharmony_ci		if (chan->call_counter >= INT_MAX)
53562306a36Sopenharmony_ci			set_bit(RXRPC_CONN_DONT_REUSE, &conn->flags);
53662306a36Sopenharmony_ci		trace_rxrpc_client(conn, channel, rxrpc_client_exposed);
53762306a36Sopenharmony_ci
53862306a36Sopenharmony_ci		spin_lock(&call->peer->lock);
53962306a36Sopenharmony_ci		hlist_add_head(&call->error_link, &call->peer->error_targets);
54062306a36Sopenharmony_ci		spin_unlock(&call->peer->lock);
54162306a36Sopenharmony_ci	}
54262306a36Sopenharmony_ci}
54362306a36Sopenharmony_ci
54462306a36Sopenharmony_ci/*
54562306a36Sopenharmony_ci * Set the reap timer.
54662306a36Sopenharmony_ci */
54762306a36Sopenharmony_cistatic void rxrpc_set_client_reap_timer(struct rxrpc_local *local)
54862306a36Sopenharmony_ci{
54962306a36Sopenharmony_ci	if (!local->kill_all_client_conns) {
55062306a36Sopenharmony_ci		unsigned long now = jiffies;
55162306a36Sopenharmony_ci		unsigned long reap_at = now + rxrpc_conn_idle_client_expiry;
55262306a36Sopenharmony_ci
55362306a36Sopenharmony_ci		if (local->rxnet->live)
55462306a36Sopenharmony_ci			timer_reduce(&local->client_conn_reap_timer, reap_at);
55562306a36Sopenharmony_ci	}
55662306a36Sopenharmony_ci}
55762306a36Sopenharmony_ci
55862306a36Sopenharmony_ci/*
55962306a36Sopenharmony_ci * Disconnect a client call.
56062306a36Sopenharmony_ci */
56162306a36Sopenharmony_civoid rxrpc_disconnect_client_call(struct rxrpc_bundle *bundle, struct rxrpc_call *call)
56262306a36Sopenharmony_ci{
56362306a36Sopenharmony_ci	struct rxrpc_connection *conn;
56462306a36Sopenharmony_ci	struct rxrpc_channel *chan = NULL;
56562306a36Sopenharmony_ci	struct rxrpc_local *local = bundle->local;
56662306a36Sopenharmony_ci	unsigned int channel;
56762306a36Sopenharmony_ci	bool may_reuse;
56862306a36Sopenharmony_ci	u32 cid;
56962306a36Sopenharmony_ci
57062306a36Sopenharmony_ci	_enter("c=%x", call->debug_id);
57162306a36Sopenharmony_ci
57262306a36Sopenharmony_ci	/* Calls that have never actually been assigned a channel can simply be
57362306a36Sopenharmony_ci	 * discarded.
57462306a36Sopenharmony_ci	 */
57562306a36Sopenharmony_ci	conn = call->conn;
57662306a36Sopenharmony_ci	if (!conn) {
57762306a36Sopenharmony_ci		_debug("call is waiting");
57862306a36Sopenharmony_ci		ASSERTCMP(call->call_id, ==, 0);
57962306a36Sopenharmony_ci		ASSERT(!test_bit(RXRPC_CALL_EXPOSED, &call->flags));
58062306a36Sopenharmony_ci		list_del_init(&call->wait_link);
58162306a36Sopenharmony_ci		return;
58262306a36Sopenharmony_ci	}
58362306a36Sopenharmony_ci
58462306a36Sopenharmony_ci	cid = call->cid;
58562306a36Sopenharmony_ci	channel = cid & RXRPC_CHANNELMASK;
58662306a36Sopenharmony_ci	chan = &conn->channels[channel];
58762306a36Sopenharmony_ci	trace_rxrpc_client(conn, channel, rxrpc_client_chan_disconnect);
58862306a36Sopenharmony_ci
58962306a36Sopenharmony_ci	if (WARN_ON(chan->call != call))
59062306a36Sopenharmony_ci		return;
59162306a36Sopenharmony_ci
59262306a36Sopenharmony_ci	may_reuse = rxrpc_may_reuse_conn(conn);
59362306a36Sopenharmony_ci
59462306a36Sopenharmony_ci	/* If a client call was exposed to the world, we save the result for
59562306a36Sopenharmony_ci	 * retransmission.
59662306a36Sopenharmony_ci	 *
59762306a36Sopenharmony_ci	 * We use a barrier here so that the call number and abort code can be
59862306a36Sopenharmony_ci	 * read without needing to take a lock.
59962306a36Sopenharmony_ci	 *
60062306a36Sopenharmony_ci	 * TODO: Make the incoming packet handler check this and handle
60162306a36Sopenharmony_ci	 * terminal retransmission without requiring access to the call.
60262306a36Sopenharmony_ci	 */
60362306a36Sopenharmony_ci	if (test_bit(RXRPC_CALL_EXPOSED, &call->flags)) {
60462306a36Sopenharmony_ci		_debug("exposed %u,%u", call->call_id, call->abort_code);
60562306a36Sopenharmony_ci		__rxrpc_disconnect_call(conn, call);
60662306a36Sopenharmony_ci
60762306a36Sopenharmony_ci		if (test_and_clear_bit(RXRPC_CONN_PROBING_FOR_UPGRADE, &conn->flags)) {
60862306a36Sopenharmony_ci			trace_rxrpc_client(conn, channel, rxrpc_client_to_active);
60962306a36Sopenharmony_ci			bundle->try_upgrade = false;
61062306a36Sopenharmony_ci			if (may_reuse)
61162306a36Sopenharmony_ci				rxrpc_activate_channels(bundle);
61262306a36Sopenharmony_ci		}
61362306a36Sopenharmony_ci	}
61462306a36Sopenharmony_ci
61562306a36Sopenharmony_ci	/* See if we can pass the channel directly to another call. */
61662306a36Sopenharmony_ci	if (may_reuse && !list_empty(&bundle->waiting_calls)) {
61762306a36Sopenharmony_ci		trace_rxrpc_client(conn, channel, rxrpc_client_chan_pass);
61862306a36Sopenharmony_ci		rxrpc_activate_one_channel(conn, channel);
61962306a36Sopenharmony_ci		return;
62062306a36Sopenharmony_ci	}
62162306a36Sopenharmony_ci
62262306a36Sopenharmony_ci	/* Schedule the final ACK to be transmitted in a short while so that it
62362306a36Sopenharmony_ci	 * can be skipped if we find a follow-on call.  The first DATA packet
62462306a36Sopenharmony_ci	 * of the follow on call will implicitly ACK this call.
62562306a36Sopenharmony_ci	 */
62662306a36Sopenharmony_ci	if (call->completion == RXRPC_CALL_SUCCEEDED &&
62762306a36Sopenharmony_ci	    test_bit(RXRPC_CALL_EXPOSED, &call->flags)) {
62862306a36Sopenharmony_ci		unsigned long final_ack_at = jiffies + 2;
62962306a36Sopenharmony_ci
63062306a36Sopenharmony_ci		WRITE_ONCE(chan->final_ack_at, final_ack_at);
63162306a36Sopenharmony_ci		smp_wmb(); /* vs rxrpc_process_delayed_final_acks() */
63262306a36Sopenharmony_ci		set_bit(RXRPC_CONN_FINAL_ACK_0 + channel, &conn->flags);
63362306a36Sopenharmony_ci		rxrpc_reduce_conn_timer(conn, final_ack_at);
63462306a36Sopenharmony_ci	}
63562306a36Sopenharmony_ci
63662306a36Sopenharmony_ci	/* Deactivate the channel. */
63762306a36Sopenharmony_ci	chan->call = NULL;
63862306a36Sopenharmony_ci	set_bit(conn->bundle_shift + channel, &conn->bundle->avail_chans);
63962306a36Sopenharmony_ci	conn->act_chans	&= ~(1 << channel);
64062306a36Sopenharmony_ci
64162306a36Sopenharmony_ci	/* If no channels remain active, then put the connection on the idle
64262306a36Sopenharmony_ci	 * list for a short while.  Give it a ref to stop it going away if it
64362306a36Sopenharmony_ci	 * becomes unbundled.
64462306a36Sopenharmony_ci	 */
64562306a36Sopenharmony_ci	if (!conn->act_chans) {
64662306a36Sopenharmony_ci		trace_rxrpc_client(conn, channel, rxrpc_client_to_idle);
64762306a36Sopenharmony_ci		conn->idle_timestamp = jiffies;
64862306a36Sopenharmony_ci
64962306a36Sopenharmony_ci		rxrpc_get_connection(conn, rxrpc_conn_get_idle);
65062306a36Sopenharmony_ci		list_move_tail(&conn->cache_link, &local->idle_client_conns);
65162306a36Sopenharmony_ci
65262306a36Sopenharmony_ci		rxrpc_set_client_reap_timer(local);
65362306a36Sopenharmony_ci	}
65462306a36Sopenharmony_ci}
65562306a36Sopenharmony_ci
65662306a36Sopenharmony_ci/*
65762306a36Sopenharmony_ci * Remove a connection from a bundle.
65862306a36Sopenharmony_ci */
65962306a36Sopenharmony_cistatic void rxrpc_unbundle_conn(struct rxrpc_connection *conn)
66062306a36Sopenharmony_ci{
66162306a36Sopenharmony_ci	struct rxrpc_bundle *bundle = conn->bundle;
66262306a36Sopenharmony_ci	unsigned int bindex;
66362306a36Sopenharmony_ci	int i;
66462306a36Sopenharmony_ci
66562306a36Sopenharmony_ci	_enter("C=%x", conn->debug_id);
66662306a36Sopenharmony_ci
66762306a36Sopenharmony_ci	if (conn->flags & RXRPC_CONN_FINAL_ACK_MASK)
66862306a36Sopenharmony_ci		rxrpc_process_delayed_final_acks(conn, true);
66962306a36Sopenharmony_ci
67062306a36Sopenharmony_ci	bindex = conn->bundle_shift / RXRPC_MAXCALLS;
67162306a36Sopenharmony_ci	if (bundle->conns[bindex] == conn) {
67262306a36Sopenharmony_ci		_debug("clear slot %u", bindex);
67362306a36Sopenharmony_ci		bundle->conns[bindex] = NULL;
67462306a36Sopenharmony_ci		for (i = 0; i < RXRPC_MAXCALLS; i++)
67562306a36Sopenharmony_ci			clear_bit(conn->bundle_shift + i, &bundle->avail_chans);
67662306a36Sopenharmony_ci		rxrpc_put_client_connection_id(bundle->local, conn);
67762306a36Sopenharmony_ci		rxrpc_deactivate_bundle(bundle);
67862306a36Sopenharmony_ci		rxrpc_put_connection(conn, rxrpc_conn_put_unbundle);
67962306a36Sopenharmony_ci	}
68062306a36Sopenharmony_ci}
68162306a36Sopenharmony_ci
68262306a36Sopenharmony_ci/*
68362306a36Sopenharmony_ci * Drop the active count on a bundle.
68462306a36Sopenharmony_ci */
68562306a36Sopenharmony_civoid rxrpc_deactivate_bundle(struct rxrpc_bundle *bundle)
68662306a36Sopenharmony_ci{
68762306a36Sopenharmony_ci	struct rxrpc_local *local;
68862306a36Sopenharmony_ci	bool need_put = false;
68962306a36Sopenharmony_ci
69062306a36Sopenharmony_ci	if (!bundle)
69162306a36Sopenharmony_ci		return;
69262306a36Sopenharmony_ci
69362306a36Sopenharmony_ci	local = bundle->local;
69462306a36Sopenharmony_ci	if (atomic_dec_and_lock(&bundle->active, &local->client_bundles_lock)) {
69562306a36Sopenharmony_ci		if (!bundle->exclusive) {
69662306a36Sopenharmony_ci			_debug("erase bundle");
69762306a36Sopenharmony_ci			rb_erase(&bundle->local_node, &local->client_bundles);
69862306a36Sopenharmony_ci			need_put = true;
69962306a36Sopenharmony_ci		}
70062306a36Sopenharmony_ci
70162306a36Sopenharmony_ci		spin_unlock(&local->client_bundles_lock);
70262306a36Sopenharmony_ci		if (need_put)
70362306a36Sopenharmony_ci			rxrpc_put_bundle(bundle, rxrpc_bundle_put_discard);
70462306a36Sopenharmony_ci	}
70562306a36Sopenharmony_ci}
70662306a36Sopenharmony_ci
70762306a36Sopenharmony_ci/*
70862306a36Sopenharmony_ci * Clean up a dead client connection.
70962306a36Sopenharmony_ci */
71062306a36Sopenharmony_civoid rxrpc_kill_client_conn(struct rxrpc_connection *conn)
71162306a36Sopenharmony_ci{
71262306a36Sopenharmony_ci	struct rxrpc_local *local = conn->local;
71362306a36Sopenharmony_ci	struct rxrpc_net *rxnet = local->rxnet;
71462306a36Sopenharmony_ci
71562306a36Sopenharmony_ci	_enter("C=%x", conn->debug_id);
71662306a36Sopenharmony_ci
71762306a36Sopenharmony_ci	trace_rxrpc_client(conn, -1, rxrpc_client_cleanup);
71862306a36Sopenharmony_ci	atomic_dec(&rxnet->nr_client_conns);
71962306a36Sopenharmony_ci
72062306a36Sopenharmony_ci	rxrpc_put_client_connection_id(local, conn);
72162306a36Sopenharmony_ci}
72262306a36Sopenharmony_ci
72362306a36Sopenharmony_ci/*
72462306a36Sopenharmony_ci * Discard expired client connections from the idle list.  Each conn in the
72562306a36Sopenharmony_ci * idle list has been exposed and holds an extra ref because of that.
72662306a36Sopenharmony_ci *
72762306a36Sopenharmony_ci * This may be called from conn setup or from a work item so cannot be
72862306a36Sopenharmony_ci * considered non-reentrant.
72962306a36Sopenharmony_ci */
73062306a36Sopenharmony_civoid rxrpc_discard_expired_client_conns(struct rxrpc_local *local)
73162306a36Sopenharmony_ci{
73262306a36Sopenharmony_ci	struct rxrpc_connection *conn;
73362306a36Sopenharmony_ci	unsigned long expiry, conn_expires_at, now;
73462306a36Sopenharmony_ci	unsigned int nr_conns;
73562306a36Sopenharmony_ci
73662306a36Sopenharmony_ci	_enter("");
73762306a36Sopenharmony_ci
73862306a36Sopenharmony_ci	/* We keep an estimate of what the number of conns ought to be after
73962306a36Sopenharmony_ci	 * we've discarded some so that we don't overdo the discarding.
74062306a36Sopenharmony_ci	 */
74162306a36Sopenharmony_ci	nr_conns = atomic_read(&local->rxnet->nr_client_conns);
74262306a36Sopenharmony_ci
74362306a36Sopenharmony_cinext:
74462306a36Sopenharmony_ci	conn = list_first_entry_or_null(&local->idle_client_conns,
74562306a36Sopenharmony_ci					struct rxrpc_connection, cache_link);
74662306a36Sopenharmony_ci	if (!conn)
74762306a36Sopenharmony_ci		return;
74862306a36Sopenharmony_ci
74962306a36Sopenharmony_ci	if (!local->kill_all_client_conns) {
75062306a36Sopenharmony_ci		/* If the number of connections is over the reap limit, we
75162306a36Sopenharmony_ci		 * expedite discard by reducing the expiry timeout.  We must,
75262306a36Sopenharmony_ci		 * however, have at least a short grace period to be able to do
75362306a36Sopenharmony_ci		 * final-ACK or ABORT retransmission.
75462306a36Sopenharmony_ci		 */
75562306a36Sopenharmony_ci		expiry = rxrpc_conn_idle_client_expiry;
75662306a36Sopenharmony_ci		if (nr_conns > rxrpc_reap_client_connections)
75762306a36Sopenharmony_ci			expiry = rxrpc_conn_idle_client_fast_expiry;
75862306a36Sopenharmony_ci		if (conn->local->service_closed)
75962306a36Sopenharmony_ci			expiry = rxrpc_closed_conn_expiry * HZ;
76062306a36Sopenharmony_ci
76162306a36Sopenharmony_ci		conn_expires_at = conn->idle_timestamp + expiry;
76262306a36Sopenharmony_ci
76362306a36Sopenharmony_ci		now = READ_ONCE(jiffies);
76462306a36Sopenharmony_ci		if (time_after(conn_expires_at, now))
76562306a36Sopenharmony_ci			goto not_yet_expired;
76662306a36Sopenharmony_ci	}
76762306a36Sopenharmony_ci
76862306a36Sopenharmony_ci	atomic_dec(&conn->active);
76962306a36Sopenharmony_ci	trace_rxrpc_client(conn, -1, rxrpc_client_discard);
77062306a36Sopenharmony_ci	list_del_init(&conn->cache_link);
77162306a36Sopenharmony_ci
77262306a36Sopenharmony_ci	rxrpc_unbundle_conn(conn);
77362306a36Sopenharmony_ci	/* Drop the ->cache_link ref */
77462306a36Sopenharmony_ci	rxrpc_put_connection(conn, rxrpc_conn_put_discard_idle);
77562306a36Sopenharmony_ci
77662306a36Sopenharmony_ci	nr_conns--;
77762306a36Sopenharmony_ci	goto next;
77862306a36Sopenharmony_ci
77962306a36Sopenharmony_cinot_yet_expired:
78062306a36Sopenharmony_ci	/* The connection at the front of the queue hasn't yet expired, so
78162306a36Sopenharmony_ci	 * schedule the work item for that point if we discarded something.
78262306a36Sopenharmony_ci	 *
78362306a36Sopenharmony_ci	 * We don't worry if the work item is already scheduled - it can look
78462306a36Sopenharmony_ci	 * after rescheduling itself at a later time.  We could cancel it, but
78562306a36Sopenharmony_ci	 * then things get messier.
78662306a36Sopenharmony_ci	 */
78762306a36Sopenharmony_ci	_debug("not yet");
78862306a36Sopenharmony_ci	if (!local->kill_all_client_conns)
78962306a36Sopenharmony_ci		timer_reduce(&local->client_conn_reap_timer, conn_expires_at);
79062306a36Sopenharmony_ci
79162306a36Sopenharmony_ci	_leave("");
79262306a36Sopenharmony_ci}
79362306a36Sopenharmony_ci
79462306a36Sopenharmony_ci/*
79562306a36Sopenharmony_ci * Clean up the client connections on a local endpoint.
79662306a36Sopenharmony_ci */
79762306a36Sopenharmony_civoid rxrpc_clean_up_local_conns(struct rxrpc_local *local)
79862306a36Sopenharmony_ci{
79962306a36Sopenharmony_ci	struct rxrpc_connection *conn;
80062306a36Sopenharmony_ci
80162306a36Sopenharmony_ci	_enter("");
80262306a36Sopenharmony_ci
80362306a36Sopenharmony_ci	local->kill_all_client_conns = true;
80462306a36Sopenharmony_ci
80562306a36Sopenharmony_ci	del_timer_sync(&local->client_conn_reap_timer);
80662306a36Sopenharmony_ci
80762306a36Sopenharmony_ci	while ((conn = list_first_entry_or_null(&local->idle_client_conns,
80862306a36Sopenharmony_ci						struct rxrpc_connection, cache_link))) {
80962306a36Sopenharmony_ci		list_del_init(&conn->cache_link);
81062306a36Sopenharmony_ci		atomic_dec(&conn->active);
81162306a36Sopenharmony_ci		trace_rxrpc_client(conn, -1, rxrpc_client_discard);
81262306a36Sopenharmony_ci		rxrpc_unbundle_conn(conn);
81362306a36Sopenharmony_ci		rxrpc_put_connection(conn, rxrpc_conn_put_local_dead);
81462306a36Sopenharmony_ci	}
81562306a36Sopenharmony_ci
81662306a36Sopenharmony_ci	_leave(" [culled]");
81762306a36Sopenharmony_ci}
818