[linux-block.git] / net / smc / smc_close.c

// SPDX-License-Identifier: GPL-2.0
/*
 *  Shared Memory Communications over RDMA (SMC-R) and RoCE
 *
 *  Socket Closing - normal and abnormal
 *
 *  Copyright IBM Corp. 2016
 *
 *  Author(s):  Ursula Braun <ubraun@linux.vnet.ibm.com>
 */

#include <linux/workqueue.h>
#include <linux/sched/signal.h>

#include <net/sock.h>

#include "smc.h"
#include "smc_tx.h"
#include "smc_cdc.h"
#include "smc_close.h"

static void smc_close_cleanup_listen(struct sock *parent)
{
	struct sock *sk;

	/* Close non-accepted connections */
	while ((sk = smc_accept_dequeue(parent, NULL)))
		smc_close_non_accepted(sk);
}

/* wait for sndbuf data being transmitted */
static void smc_close_stream_wait(struct smc_sock *smc, long timeout)
{
	DEFINE_WAIT_FUNC(wait, woken_wake_function);
	struct sock *sk = &smc->sk;

	if (!timeout)
		return;

	if (!smc_tx_prepared_sends(&smc->conn))
		return;

	smc->wait_close_tx_prepared = 1;
	add_wait_queue(sk_sleep(sk), &wait);
	while (!signal_pending(current) && timeout) {
		int rc;

		rc = sk_wait_event(sk, &timeout,
				   !smc_tx_prepared_sends(&smc->conn) ||
				   (sk->sk_err == ECONNABORTED) ||
				   (sk->sk_err == ECONNRESET),
				   &wait);
		if (rc)
			break;
	}
	remove_wait_queue(sk_sleep(sk), &wait);
	smc->wait_close_tx_prepared = 0;
}

void smc_close_wake_tx_prepared(struct smc_sock *smc)
{
	if (smc->wait_close_tx_prepared)
		/* wake up socket closing */
		smc->sk.sk_state_change(&smc->sk);
}

static int smc_close_wr(struct smc_connection *conn)
{
	conn->local_tx_ctrl.conn_state_flags.peer_done_writing = 1;

	return smc_cdc_get_slot_and_msg_send(conn);
}

static int smc_close_final(struct smc_connection *conn)
{
	if (atomic_read(&conn->bytes_to_rcv))
		conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1;
	else
		conn->local_tx_ctrl.conn_state_flags.peer_conn_closed = 1;

	return smc_cdc_get_slot_and_msg_send(conn);
}

static int smc_close_abort(struct smc_connection *conn)
{
	conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1;

	return smc_cdc_get_slot_and_msg_send(conn);
}

/* terminate smc socket abnormally - active abort
 * link group is terminated, i.e. RDMA communication no longer possible
 */
static void smc_close_active_abort(struct smc_sock *smc)
{
	struct sock *sk = &smc->sk;

	struct smc_cdc_conn_state_flags *txflags =
		&smc->conn.local_tx_ctrl.conn_state_flags;

	sk->sk_err = ECONNABORTED;
	if (smc->clcsock && smc->clcsock->sk) {
		smc->clcsock->sk->sk_err = ECONNABORTED;
		smc->clcsock->sk->sk_state_change(smc->clcsock->sk);
	}
	switch (sk->sk_state) {
	case SMC_INIT:
	case SMC_ACTIVE:
		sk->sk_state = SMC_PEERABORTWAIT;
		release_sock(sk);
		cancel_delayed_work_sync(&smc->conn.tx_work);
		lock_sock(sk);
		break;
	case SMC_APPCLOSEWAIT1:
	case SMC_APPCLOSEWAIT2:
		sock_release(smc->clcsock);
		if (!smc_cdc_rxed_any_close(&smc->conn))
			sk->sk_state = SMC_PEERABORTWAIT;
		else
			sk->sk_state = SMC_CLOSED;
		release_sock(sk);
		cancel_delayed_work_sync(&smc->conn.tx_work);
		lock_sock(sk);
		break;
	case SMC_PEERCLOSEWAIT1:
	case SMC_PEERCLOSEWAIT2:
		if (!txflags->peer_conn_closed) {
			sk->sk_state = SMC_PEERABORTWAIT;
			sock_release(smc->clcsock);
		} else {
			sk->sk_state = SMC_CLOSED;
		}
		break;
	case SMC_PROCESSABORT:
	case SMC_APPFINCLOSEWAIT:
		if (!txflags->peer_conn_closed)
			sock_release(smc->clcsock);
		sk->sk_state = SMC_CLOSED;
		break;
	case SMC_PEERFINCLOSEWAIT:
	case SMC_PEERABORTWAIT:
	case SMC_CLOSED:
		break;
	}

	sock_set_flag(sk, SOCK_DEAD);
	sk->sk_state_change(sk);
}

static inline bool smc_close_sent_any_close(struct smc_connection *conn)
{
	return conn->local_tx_ctrl.conn_state_flags.peer_conn_abort ||
	       conn->local_tx_ctrl.conn_state_flags.peer_conn_closed;
}

int smc_close_active(struct smc_sock *smc)
{
	struct smc_cdc_conn_state_flags *txflags =
		&smc->conn.local_tx_ctrl.conn_state_flags;
	struct smc_connection *conn = &smc->conn;
	struct sock *sk = &smc->sk;
	int old_state;
	long timeout;
	int rc = 0;

	timeout = current->flags & PF_EXITING ?
		  0 : sock_flag(sk, SOCK_LINGER) ?
		      sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT;

	old_state = sk->sk_state;
again:
	switch (sk->sk_state) {
	case SMC_INIT:
		sk->sk_state = SMC_CLOSED;
		if (smc->smc_listen_work.func)
			cancel_work_sync(&smc->smc_listen_work);
		break;
	case SMC_LISTEN:
		sk->sk_state = SMC_CLOSED;
		sk->sk_state_change(sk); /* wake up accept */
		if (smc->clcsock && smc->clcsock->sk) {
			rc = kernel_sock_shutdown(smc->clcsock, SHUT_RDWR);
			/* wake up kernel_accept of smc_tcp_listen_worker */
			smc->clcsock->sk->sk_data_ready(smc->clcsock->sk);
		}
		release_sock(sk);
		smc_close_cleanup_listen(sk);
		cancel_work_sync(&smc->smc_listen_work);
		lock_sock(sk);
		break;
	case SMC_ACTIVE:
		smc_close_stream_wait(smc, timeout);
		release_sock(sk);
		cancel_delayed_work_sync(&conn->tx_work);
		lock_sock(sk);
		if (sk->sk_state == SMC_ACTIVE) {
			/* send close request */
			rc = smc_close_final(conn);
			if (rc)
				break;
			sk->sk_state = SMC_PEERCLOSEWAIT1;
		} else {
			/* peer event has changed the state */
			goto again;
		}
		break;
	case SMC_APPFINCLOSEWAIT:
		/* socket already shutdown wr or both (active close) */
		if (txflags->peer_done_writing &&
		    !smc_close_sent_any_close(conn)) {
			/* just shutdown wr done, send close request */
			rc = smc_close_final(conn);
			if (rc)
				break;
		}
		sk->sk_state = SMC_CLOSED;
		break;
	case SMC_APPCLOSEWAIT1:
	case SMC_APPCLOSEWAIT2:
		if (!smc_cdc_rxed_any_close(conn))
			smc_close_stream_wait(smc, timeout);
		release_sock(sk);
		cancel_delayed_work_sync(&conn->tx_work);
		lock_sock(sk);
		if (sk->sk_state != SMC_APPCLOSEWAIT1 &&
		    sk->sk_state != SMC_APPCLOSEWAIT2)
			goto again;
		/* confirm close from peer */
		rc = smc_close_final(conn);
		if (rc)
			break;
		if (smc_cdc_rxed_any_close(conn))
			/* peer has closed the socket already */
			sk->sk_state = SMC_CLOSED;
		else
			/* peer has just issued a shutdown write */
			sk->sk_state = SMC_PEERFINCLOSEWAIT;
		break;
	case SMC_PEERCLOSEWAIT1:
	case SMC_PEERCLOSEWAIT2:
		if (txflags->peer_done_writing &&
		    !smc_close_sent_any_close(conn)) {
			/* just shutdown wr done, send close request */
			rc = smc_close_final(conn);
			if (rc)
				break;
		}
		/* peer sending PeerConnectionClosed will cause transition */
		break;
	case SMC_PEERFINCLOSEWAIT:
		/* peer sending PeerConnectionClosed will cause transition */
		break;
	case SMC_PROCESSABORT:
		smc_close_abort(conn);
		sk->sk_state = SMC_CLOSED;
		break;
	case SMC_PEERABORTWAIT:
	case SMC_CLOSED:
		/* nothing to do, add tracing in future patch */
		break;
	}

	if (old_state != sk->sk_state)
		sk->sk_state_change(sk);
	return rc;
}

static void smc_close_passive_abort_received(struct smc_sock *smc)
{
	struct smc_cdc_conn_state_flags *txflags =
		&smc->conn.local_tx_ctrl.conn_state_flags;
	struct sock *sk = &smc->sk;

	switch (sk->sk_state) {
	case SMC_ACTIVE:
	case SMC_APPFINCLOSEWAIT:
	case SMC_APPCLOSEWAIT1:
	case SMC_APPCLOSEWAIT2:
		sk->sk_state = SMC_PROCESSABORT;
		break;
	case SMC_PEERCLOSEWAIT1:
	case SMC_PEERCLOSEWAIT2:
		if (txflags->peer_done_writing &&
		    !smc_close_sent_any_close(&smc->conn)) {
			/* just shutdown, but not yet closed locally */
			sk->sk_state = SMC_PROCESSABORT;
		} else {
			sk->sk_state = SMC_CLOSED;
		}
		break;
	case SMC_PEERFINCLOSEWAIT:
	case SMC_PEERABORTWAIT:
		sk->sk_state = SMC_CLOSED;
		break;
	case SMC_INIT:
	case SMC_PROCESSABORT:
	/* nothing to do, add tracing in future patch */
		break;
	}
}

/* Either some kind of closing has been received: peer_conn_closed,
 * peer_conn_abort, or peer_done_writing
 * or the link group of the connection terminates abnormally.
 */
static void smc_close_passive_work(struct work_struct *work)
{
	struct smc_connection *conn = container_of(work,
						   struct smc_connection,
						   close_work);
	struct smc_sock *smc = container_of(conn, struct smc_sock, conn);
	struct smc_cdc_conn_state_flags *rxflags;
	struct sock *sk = &smc->sk;
	int old_state;

	lock_sock(sk);
	old_state = sk->sk_state;

	if (!conn->alert_token_local) {
		/* abnormal termination */
		smc_close_active_abort(smc);
		goto wakeup;
	}

	rxflags = &conn->local_rx_ctrl.conn_state_flags;
	if (rxflags->peer_conn_abort) {
		/* peer has not received all data */
		smc_close_passive_abort_received(smc);
		release_sock(&smc->sk);
		cancel_delayed_work_sync(&conn->tx_work);
		lock_sock(&smc->sk);
		goto wakeup;
	}

	switch (sk->sk_state) {
	case SMC_INIT:
		if (atomic_read(&conn->bytes_to_rcv) ||
		    (rxflags->peer_done_writing &&
		     !smc_cdc_rxed_any_close(conn)))
			sk->sk_state = SMC_APPCLOSEWAIT1;
		else
			sk->sk_state = SMC_CLOSED;
		break;
	case SMC_ACTIVE:
		sk->sk_state = SMC_APPCLOSEWAIT1;
		break;
	case SMC_PEERCLOSEWAIT1:
		if (rxflags->peer_done_writing)
			sk->sk_state = SMC_PEERCLOSEWAIT2;
		/* fall through */
		/* to check for closing */
	case SMC_PEERCLOSEWAIT2:
		if (!smc_cdc_rxed_any_close(conn))
			break;
		if (sock_flag(sk, SOCK_DEAD) &&
		    smc_close_sent_any_close(conn)) {
			/* smc_release has already been called locally */
			sk->sk_state = SMC_CLOSED;
		} else {
			/* just shutdown, but not yet closed locally */
			sk->sk_state = SMC_APPFINCLOSEWAIT;
		}
		break;
	case SMC_PEERFINCLOSEWAIT:
		if (smc_cdc_rxed_any_close(conn))
			sk->sk_state = SMC_CLOSED;
		break;
	case SMC_APPCLOSEWAIT1:
	case SMC_APPCLOSEWAIT2:
	case SMC_APPFINCLOSEWAIT:
	case SMC_PEERABORTWAIT:
	case SMC_PROCESSABORT:
	case SMC_CLOSED:
		/* nothing to do, add tracing in future patch */
		break;
	}

wakeup:
	sk->sk_data_ready(sk); /* wakeup blocked rcvbuf consumers */
	sk->sk_write_space(sk); /* wakeup blocked sndbuf producers */

	if (old_state != sk->sk_state) {
		sk->sk_state_change(sk);
		if ((sk->sk_state == SMC_CLOSED) &&
		    (sock_flag(sk, SOCK_DEAD) || !sk->sk_socket)) {
			smc_conn_free(conn);
			schedule_delayed_work(&smc->sock_put_work,
					      SMC_CLOSE_SOCK_PUT_DELAY);
		}
	}
	release_sock(sk);
}

void smc_close_sock_put_work(struct work_struct *work)
{
	struct smc_sock *smc = container_of(to_delayed_work(work),
					    struct smc_sock,
					    sock_put_work);

	smc->sk.sk_prot->unhash(&smc->sk);
	sock_put(&smc->sk);
}

int smc_close_shutdown_write(struct smc_sock *smc)
{
	struct smc_connection *conn = &smc->conn;
	struct sock *sk = &smc->sk;
	int old_state;
	long timeout;
	int rc = 0;

	timeout = current->flags & PF_EXITING ?
		  0 : sock_flag(sk, SOCK_LINGER) ?
		      sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT;

	old_state = sk->sk_state;
again:
	switch (sk->sk_state) {
	case SMC_ACTIVE:
		smc_close_stream_wait(smc, timeout);
		release_sock(sk);
		cancel_delayed_work_sync(&conn->tx_work);
		lock_sock(sk);
		if (sk->sk_state != SMC_ACTIVE)
			goto again;
		/* send close wr request */
		rc = smc_close_wr(conn);
		if (rc)
			break;
		sk->sk_state = SMC_PEERCLOSEWAIT1;
		break;
	case SMC_APPCLOSEWAIT1:
		/* passive close */
		if (!smc_cdc_rxed_any_close(conn))
			smc_close_stream_wait(smc, timeout);
		release_sock(sk);
		cancel_delayed_work_sync(&conn->tx_work);
		lock_sock(sk);
		if (sk->sk_state != SMC_APPCLOSEWAIT1)
			goto again;
		/* confirm close from peer */
		rc = smc_close_wr(conn);
		if (rc)
			break;
		sk->sk_state = SMC_APPCLOSEWAIT2;
		break;
	case SMC_APPCLOSEWAIT2:
	case SMC_PEERFINCLOSEWAIT:
	case SMC_PEERCLOSEWAIT1:
	case SMC_PEERCLOSEWAIT2:
	case SMC_APPFINCLOSEWAIT:
	case SMC_PROCESSABORT:
	case SMC_PEERABORTWAIT:
		/* nothing to do, add tracing in future patch */
		break;
	}

	if (old_state != sk->sk_state)
		sk->sk_state_change(sk);
	return rc;
}

/* Initialize close properties on connection establishment. */
void smc_close_init(struct smc_sock *smc)
{
	INIT_WORK(&smc->conn.close_work, smc_close_passive_work);
}
Commit	Line	Data
b2441318	1	// SPDX-License-Identifier: GPL-2.0
b38d7324 UB	2	/*
	3	* Shared Memory Communications over RDMA (SMC-R) and RoCE
	4	*
	5	* Socket Closing - normal and abnormal
	6	*
	7	* Copyright IBM Corp. 2016
	8	*
	9	* Author(s): Ursula Braun <ubraun@linux.vnet.ibm.com>
	10	*/
	11
	12	#include <linux/workqueue.h>
c3edc401 IM	13	#include <linux/sched/signal.h>
c3edc401 IM	14
b38d7324 UB	15	#include <net/sock.h>
	16
	17	#include "smc.h"
	18	#include "smc_tx.h"
	19	#include "smc_cdc.h"
	20	#include "smc_close.h"
	21
b38d7324 UB	22	static void smc_close_cleanup_listen(struct sock *parent)
	23	{
	24	struct sock *sk;
	25
	26	/* Close non-accepted connections */
	27	while ((sk = smc_accept_dequeue(parent, NULL)))
	28	smc_close_non_accepted(sk);
	29	}
	30
b38d7324 UB	31	/* wait for sndbuf data being transmitted */
	32	static void smc_close_stream_wait(struct smc_sock *smc, long timeout)
	33	{
	34	DEFINE_WAIT_FUNC(wait, woken_wake_function);
	35	struct sock *sk = &smc->sk;
	36
	37	if (!timeout)
	38	return;
	39
	40	if (!smc_tx_prepared_sends(&smc->conn))
	41	return;
	42
	43	smc->wait_close_tx_prepared = 1;
	44	add_wait_queue(sk_sleep(sk), &wait);
	45	while (!signal_pending(current) && timeout) {
	46	int rc;
	47
	48	rc = sk_wait_event(sk, &timeout,
	49	!smc_tx_prepared_sends(&smc->conn) \|\|
	50	(sk->sk_err == ECONNABORTED) \|\|
	51	(sk->sk_err == ECONNRESET),
	52	&wait);
	53	if (rc)
	54	break;
	55	}
	56	remove_wait_queue(sk_sleep(sk), &wait);
	57	smc->wait_close_tx_prepared = 0;
	58	}
	59
	60	void smc_close_wake_tx_prepared(struct smc_sock *smc)
	61	{
	62	if (smc->wait_close_tx_prepared)
	63	/* wake up socket closing */
	64	smc->sk.sk_state_change(&smc->sk);
	65	}
	66
	67	static int smc_close_wr(struct smc_connection *conn)
	68	{
	69	conn->local_tx_ctrl.conn_state_flags.peer_done_writing = 1;
	70
	71	return smc_cdc_get_slot_and_msg_send(conn);
	72	}
	73
	74	static int smc_close_final(struct smc_connection *conn)
	75	{
	76	if (atomic_read(&conn->bytes_to_rcv))
	77	conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1;
	78	else
	79	conn->local_tx_ctrl.conn_state_flags.peer_conn_closed = 1;
	80
	81	return smc_cdc_get_slot_and_msg_send(conn);
	82	}
	83
	84	static int smc_close_abort(struct smc_connection *conn)
	85	{
	86	conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1;
	87
	88	return smc_cdc_get_slot_and_msg_send(conn);
	89	}
	90
	91	/* terminate smc socket abnormally - active abort
732720fa	92	* link group is terminated, i.e. RDMA communication no longer possible
b38d7324	93	*/
a8ae890b	94	static void smc_close_active_abort(struct smc_sock *smc)
b38d7324	95	{
3163c507 UB	96	struct sock *sk = &smc->sk;
3163c507 UB	97
b38d7324 UB	98	struct smc_cdc_conn_state_flags *txflags =
	99	&smc->conn.local_tx_ctrl.conn_state_flags;
	100
3163c507	101	sk->sk_err = ECONNABORTED;
b38d7324 UB	102	if (smc->clcsock && smc->clcsock->sk) {
	103	smc->clcsock->sk->sk_err = ECONNABORTED;
	104	smc->clcsock->sk->sk_state_change(smc->clcsock->sk);
	105	}
3163c507	106	switch (sk->sk_state) {
b38d7324	107	case SMC_INIT:
46c28dbd	108	case SMC_ACTIVE:
3163c507	109	sk->sk_state = SMC_PEERABORTWAIT;
611b63a1 UB	110	release_sock(sk);
	111	cancel_delayed_work_sync(&smc->conn.tx_work);
	112	lock_sock(sk);
b38d7324 UB	113	break;
	114	case SMC_APPCLOSEWAIT1:
	115	case SMC_APPCLOSEWAIT2:
b38d7324 UB	116	sock_release(smc->clcsock);
b38d7324 UB	117	if (!smc_cdc_rxed_any_close(&smc->conn))
3163c507	118	sk->sk_state = SMC_PEERABORTWAIT;
b38d7324	119	else
3163c507	120	sk->sk_state = SMC_CLOSED;
611b63a1 UB	121	release_sock(sk);
	122	cancel_delayed_work_sync(&smc->conn.tx_work);
	123	lock_sock(sk);
b38d7324 UB	124	break;
	125	case SMC_PEERCLOSEWAIT1:
	126	case SMC_PEERCLOSEWAIT2:
	127	if (!txflags->peer_conn_closed) {
3163c507	128	sk->sk_state = SMC_PEERABORTWAIT;
b38d7324 UB	129	sock_release(smc->clcsock);
b38d7324 UB	130	} else {
3163c507	131	sk->sk_state = SMC_CLOSED;
b38d7324 UB	132	}
	133	break;
	134	case SMC_PROCESSABORT:
	135	case SMC_APPFINCLOSEWAIT:
732720fa	136	if (!txflags->peer_conn_closed)
b38d7324	137	sock_release(smc->clcsock);
3163c507	138	sk->sk_state = SMC_CLOSED;
b38d7324 UB	139	break;
	140	case SMC_PEERFINCLOSEWAIT:
	141	case SMC_PEERABORTWAIT:
	142	case SMC_CLOSED:
	143	break;
	144	}
	145
3163c507 UB	146	sock_set_flag(sk, SOCK_DEAD);
3163c507 UB	147	sk->sk_state_change(sk);
b38d7324 UB	148	}
b38d7324 UB	149
a98bf8c0 UB	150	static inline bool smc_close_sent_any_close(struct smc_connection *conn)
	151	{
	152	return conn->local_tx_ctrl.conn_state_flags.peer_conn_abort \|\|
	153	conn->local_tx_ctrl.conn_state_flags.peer_conn_closed;
	154	}
	155
b38d7324 UB	156	int smc_close_active(struct smc_sock *smc)
	157	{
	158	struct smc_cdc_conn_state_flags *txflags =
	159	&smc->conn.local_tx_ctrl.conn_state_flags;
b38d7324 UB	160	struct smc_connection *conn = &smc->conn;
	161	struct sock *sk = &smc->sk;
	162	int old_state;
8c96feee	163	long timeout;
b38d7324 UB	164	int rc = 0;
b38d7324 UB	165
8c96feee UB	166	timeout = current->flags & PF_EXITING ?
	167	0 : sock_flag(sk, SOCK_LINGER) ?
	168	sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT;
b38d7324	169
b38d7324	170	old_state = sk->sk_state;
bbb96bf2 UB	171	again:
bbb96bf2 UB	172	switch (sk->sk_state) {
b38d7324 UB	173	case SMC_INIT:
	174	sk->sk_state = SMC_CLOSED;
	175	if (smc->smc_listen_work.func)
46c28dbd	176	cancel_work_sync(&smc->smc_listen_work);
b38d7324 UB	177	break;
	178	case SMC_LISTEN:
	179	sk->sk_state = SMC_CLOSED;
	180	sk->sk_state_change(sk); /* wake up accept */
	181	if (smc->clcsock && smc->clcsock->sk) {
	182	rc = kernel_sock_shutdown(smc->clcsock, SHUT_RDWR);
	183	/* wake up kernel_accept of smc_tcp_listen_worker */
	184	smc->clcsock->sk->sk_data_ready(smc->clcsock->sk);
	185	}
	186	release_sock(sk);
	187	smc_close_cleanup_listen(sk);
46c28dbd	188	cancel_work_sync(&smc->smc_listen_work);
b38d7324 UB	189	lock_sock(sk);
	190	break;
	191	case SMC_ACTIVE:
	192	smc_close_stream_wait(smc, timeout);
	193	release_sock(sk);
18e537cd	194	cancel_delayed_work_sync(&conn->tx_work);
b38d7324 UB	195	lock_sock(sk);
	196	if (sk->sk_state == SMC_ACTIVE) {
	197	/* send close request */
	198	rc = smc_close_final(conn);
bbb96bf2 UB	199	if (rc)
bbb96bf2 UB	200	break;
b38d7324 UB	201	sk->sk_state = SMC_PEERCLOSEWAIT1;
	202	} else {
	203	/* peer event has changed the state */
	204	goto again;
	205	}
	206	break;
	207	case SMC_APPFINCLOSEWAIT:
	208	/* socket already shutdown wr or both (active close) */
	209	if (txflags->peer_done_writing &&
a98bf8c0	210	!smc_close_sent_any_close(conn)) {
b38d7324 UB	211	/* just shutdown wr done, send close request */
b38d7324 UB	212	rc = smc_close_final(conn);
bbb96bf2 UB	213	if (rc)
bbb96bf2 UB	214	break;
b38d7324 UB	215	}
b38d7324 UB	216	sk->sk_state = SMC_CLOSED;
b38d7324 UB	217	break;
	218	case SMC_APPCLOSEWAIT1:
	219	case SMC_APPCLOSEWAIT2:
	220	if (!smc_cdc_rxed_any_close(conn))
	221	smc_close_stream_wait(smc, timeout);
	222	release_sock(sk);
18e537cd	223	cancel_delayed_work_sync(&conn->tx_work);
b38d7324	224	lock_sock(sk);
bbb96bf2 UB	225	if (sk->sk_state != SMC_APPCLOSEWAIT1 &&
	226	sk->sk_state != SMC_APPCLOSEWAIT2)
	227	goto again;
	228	/* confirm close from peer */
	229	rc = smc_close_final(conn);
	230	if (rc)
	231	break;
b38d7324 UB	232	if (smc_cdc_rxed_any_close(conn))
	233	/* peer has closed the socket already */
	234	sk->sk_state = SMC_CLOSED;
	235	else
	236	/* peer has just issued a shutdown write */
	237	sk->sk_state = SMC_PEERFINCLOSEWAIT;
b38d7324 UB	238	break;
	239	case SMC_PEERCLOSEWAIT1:
	240	case SMC_PEERCLOSEWAIT2:
a98bf8c0 UB	241	if (txflags->peer_done_writing &&
	242	!smc_close_sent_any_close(conn)) {
	243	/* just shutdown wr done, send close request */
	244	rc = smc_close_final(conn);
bbb96bf2 UB	245	if (rc)
bbb96bf2 UB	246	break;
a98bf8c0 UB	247	}
	248	/* peer sending PeerConnectionClosed will cause transition */
	249	break;
b38d7324 UB	250	case SMC_PEERFINCLOSEWAIT:
	251	/* peer sending PeerConnectionClosed will cause transition */
	252	break;
	253	case SMC_PROCESSABORT:
b38d7324 UB	254	smc_close_abort(conn);
b38d7324 UB	255	sk->sk_state = SMC_CLOSED;
b38d7324 UB	256	break;
	257	case SMC_PEERABORTWAIT:
	258	case SMC_CLOSED:
	259	/* nothing to do, add tracing in future patch */
	260	break;
	261	}
	262
	263	if (old_state != sk->sk_state)
3163c507	264	sk->sk_state_change(sk);
b38d7324 UB	265	return rc;
	266	}
	267
	268	static void smc_close_passive_abort_received(struct smc_sock *smc)
	269	{
	270	struct smc_cdc_conn_state_flags *txflags =
	271	&smc->conn.local_tx_ctrl.conn_state_flags;
	272	struct sock *sk = &smc->sk;
	273
	274	switch (sk->sk_state) {
	275	case SMC_ACTIVE:
	276	case SMC_APPFINCLOSEWAIT:
	277	case SMC_APPCLOSEWAIT1:
	278	case SMC_APPCLOSEWAIT2:
b38d7324 UB	279	sk->sk_state = SMC_PROCESSABORT;
	280	break;
	281	case SMC_PEERCLOSEWAIT1:
	282	case SMC_PEERCLOSEWAIT2:
	283	if (txflags->peer_done_writing &&
a98bf8c0	284	!smc_close_sent_any_close(&smc->conn)) {
b38d7324	285	/* just shutdown, but not yet closed locally */
b38d7324 UB	286	sk->sk_state = SMC_PROCESSABORT;
	287	} else {
	288	sk->sk_state = SMC_CLOSED;
	289	}
	290	break;
	291	case SMC_PEERFINCLOSEWAIT:
	292	case SMC_PEERABORTWAIT:
	293	sk->sk_state = SMC_CLOSED;
	294	break;
	295	case SMC_INIT:
	296	case SMC_PROCESSABORT:
	297	/* nothing to do, add tracing in future patch */
	298	break;
	299	}
	300	}
	301
732720fa UB	302	/* Either some kind of closing has been received: peer_conn_closed,
	303	* peer_conn_abort, or peer_done_writing
	304	* or the link group of the connection terminates abnormally.
b38d7324	305	*/
46c28dbd	306	static void smc_close_passive_work(struct work_struct *work)
b38d7324	307	{
46c28dbd UB	308	struct smc_connection *conn = container_of(work,
	309	struct smc_connection,
	310	close_work);
	311	struct smc_sock *smc = container_of(conn, struct smc_sock, conn);
	312	struct smc_cdc_conn_state_flags *rxflags;
b38d7324 UB	313	struct sock *sk = &smc->sk;
	314	int old_state;
	315
3163c507	316	lock_sock(sk);
b38d7324 UB	317	old_state = sk->sk_state;
b38d7324 UB	318
46c28dbd UB	319	if (!conn->alert_token_local) {
	320	/* abnormal termination */
	321	smc_close_active_abort(smc);
	322	goto wakeup;
	323	}
	324
3163c507	325	rxflags = &conn->local_rx_ctrl.conn_state_flags;
b38d7324	326	if (rxflags->peer_conn_abort) {
732720fa	327	/* peer has not received all data */
b38d7324	328	smc_close_passive_abort_received(smc);
611b63a1 UB	329	release_sock(&smc->sk);
	330	cancel_delayed_work_sync(&conn->tx_work);
	331	lock_sock(&smc->sk);
b38d7324 UB	332	goto wakeup;
	333	}
	334
	335	switch (sk->sk_state) {
	336	case SMC_INIT:
3163c507	337	if (atomic_read(&conn->bytes_to_rcv) \|\|
b38d7324	338	(rxflags->peer_done_writing &&
a98bf8c0	339	!smc_cdc_rxed_any_close(conn)))
b38d7324 UB	340	sk->sk_state = SMC_APPCLOSEWAIT1;
	341	else
	342	sk->sk_state = SMC_CLOSED;
	343	break;
	344	case SMC_ACTIVE:
	345	sk->sk_state = SMC_APPCLOSEWAIT1;
	346	break;
	347	case SMC_PEERCLOSEWAIT1:
	348	if (rxflags->peer_done_writing)
	349	sk->sk_state = SMC_PEERCLOSEWAIT2;
7f6b437e GS	350	/* fall through */
7f6b437e GS	351	/* to check for closing */
b38d7324	352	case SMC_PEERCLOSEWAIT2:
3163c507	353	if (!smc_cdc_rxed_any_close(conn))
b38d7324 UB	354	break;
b38d7324 UB	355	if (sock_flag(sk, SOCK_DEAD) &&
a98bf8c0	356	smc_close_sent_any_close(conn)) {
b38d7324 UB	357	/* smc_release has already been called locally */
	358	sk->sk_state = SMC_CLOSED;
	359	} else {
	360	/* just shutdown, but not yet closed locally */
	361	sk->sk_state = SMC_APPFINCLOSEWAIT;
	362	}
	363	break;
5ac92a00 UB	364	case SMC_PEERFINCLOSEWAIT:
	365	if (smc_cdc_rxed_any_close(conn))
	366	sk->sk_state = SMC_CLOSED;
	367	break;
b38d7324 UB	368	case SMC_APPCLOSEWAIT1:
	369	case SMC_APPCLOSEWAIT2:
	370	case SMC_APPFINCLOSEWAIT:
	371	case SMC_PEERABORTWAIT:
	372	case SMC_PROCESSABORT:
	373	case SMC_CLOSED:
	374	/* nothing to do, add tracing in future patch */
	375	break;
	376	}
	377
	378	wakeup:
b38d7324 UB	379	sk->sk_data_ready(sk); /* wakeup blocked rcvbuf consumers */
	380	sk->sk_write_space(sk); /* wakeup blocked sndbuf producers */
	381
a98bf8c0 UB	382	if (old_state != sk->sk_state) {
	383	sk->sk_state_change(sk);
	384	if ((sk->sk_state == SMC_CLOSED) &&
	385	(sock_flag(sk, SOCK_DEAD) \|\| !sk->sk_socket)) {
3163c507	386	smc_conn_free(conn);
a98bf8c0 UB	387	schedule_delayed_work(&smc->sock_put_work,
	388	SMC_CLOSE_SOCK_PUT_DELAY);
	389	}
b38d7324	390	}
3163c507	391	release_sock(sk);
b38d7324 UB	392	}
	393
	394	void smc_close_sock_put_work(struct work_struct *work)
	395	{
	396	struct smc_sock *smc = container_of(to_delayed_work(work),
	397	struct smc_sock,
	398	sock_put_work);
	399
f16a7dd5	400	smc->sk.sk_prot->unhash(&smc->sk);
b38d7324 UB	401	sock_put(&smc->sk);
	402	}
	403
	404	int smc_close_shutdown_write(struct smc_sock *smc)
	405	{
	406	struct smc_connection *conn = &smc->conn;
b38d7324 UB	407	struct sock *sk = &smc->sk;
b38d7324 UB	408	int old_state;
8c96feee	409	long timeout;
b38d7324 UB	410	int rc = 0;
b38d7324 UB	411
8c96feee UB	412	timeout = current->flags & PF_EXITING ?
	413	0 : sock_flag(sk, SOCK_LINGER) ?
	414	sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT;
b38d7324	415
b38d7324	416	old_state = sk->sk_state;
bbb96bf2 UB	417	again:
bbb96bf2 UB	418	switch (sk->sk_state) {
b38d7324 UB	419	case SMC_ACTIVE:
	420	smc_close_stream_wait(smc, timeout);
	421	release_sock(sk);
18e537cd	422	cancel_delayed_work_sync(&conn->tx_work);
b38d7324	423	lock_sock(sk);
bbb96bf2 UB	424	if (sk->sk_state != SMC_ACTIVE)
bbb96bf2 UB	425	goto again;
b38d7324 UB	426	/* send close wr request */
b38d7324 UB	427	rc = smc_close_wr(conn);
bbb96bf2 UB	428	if (rc)
	429	break;
	430	sk->sk_state = SMC_PEERCLOSEWAIT1;
b38d7324 UB	431	break;
	432	case SMC_APPCLOSEWAIT1:
	433	/* passive close */
	434	if (!smc_cdc_rxed_any_close(conn))
	435	smc_close_stream_wait(smc, timeout);
	436	release_sock(sk);
18e537cd	437	cancel_delayed_work_sync(&conn->tx_work);
b38d7324	438	lock_sock(sk);
bbb96bf2 UB	439	if (sk->sk_state != SMC_APPCLOSEWAIT1)
bbb96bf2 UB	440	goto again;
b38d7324 UB	441	/* confirm close from peer */
b38d7324 UB	442	rc = smc_close_wr(conn);
bbb96bf2 UB	443	if (rc)
bbb96bf2 UB	444	break;
b38d7324 UB	445	sk->sk_state = SMC_APPCLOSEWAIT2;
	446	break;
	447	case SMC_APPCLOSEWAIT2:
	448	case SMC_PEERFINCLOSEWAIT:
	449	case SMC_PEERCLOSEWAIT1:
	450	case SMC_PEERCLOSEWAIT2:
	451	case SMC_APPFINCLOSEWAIT:
	452	case SMC_PROCESSABORT:
	453	case SMC_PEERABORTWAIT:
	454	/* nothing to do, add tracing in future patch */
	455	break;
	456	}
	457
	458	if (old_state != sk->sk_state)
3163c507	459	sk->sk_state_change(sk);
b38d7324 UB	460	return rc;
b38d7324 UB	461	}
46c28dbd UB	462
	463	/* Initialize close properties on connection establishment. */
	464	void smc_close_init(struct smc_sock *smc)
	465	{
	466	INIT_WORK(&smc->conn.close_work, smc_close_passive_work);
	467	}