1 // SPDX-License-Identifier: GPL-2.0 1 // SPDX-License-Identifier: GPL-2.0 2 /* 2 /* 3 * Shared Memory Communications over RDMA (SM 3 * Shared Memory Communications over RDMA (SMC-R) and RoCE 4 * 4 * 5 * Socket Closing - normal and abnormal 5 * Socket Closing - normal and abnormal 6 * 6 * 7 * Copyright IBM Corp. 2016 7 * Copyright IBM Corp. 2016 8 * 8 * 9 * Author(s): Ursula Braun <ubraun@linux.vne 9 * Author(s): Ursula Braun <ubraun@linux.vnet.ibm.com> 10 */ 10 */ 11 11 12 #include <linux/workqueue.h> 12 #include <linux/workqueue.h> 13 #include <linux/sched/signal.h> 13 #include <linux/sched/signal.h> 14 14 15 #include <net/sock.h> 15 #include <net/sock.h> 16 #include <net/tcp.h> << 17 16 18 #include "smc.h" 17 #include "smc.h" 19 #include "smc_tx.h" 18 #include "smc_tx.h" 20 #include "smc_cdc.h" 19 #include "smc_cdc.h" 21 #include "smc_close.h" 20 #include "smc_close.h" 22 21 23 /* release the clcsock that is assigned to the !! 22 #define SMC_CLOSE_WAIT_LISTEN_CLCSOCK_TIME (5 * HZ) 24 void smc_clcsock_release(struct smc_sock *smc) << 25 { << 26 struct socket *tcp; << 27 << 28 if (smc->listen_smc && current_work() << 29 cancel_work_sync(&smc->smc_lis << 30 mutex_lock(&smc->clcsock_release_lock) << 31 if (smc->clcsock) { << 32 tcp = smc->clcsock; << 33 smc->clcsock = NULL; << 34 sock_release(tcp); << 35 } << 36 mutex_unlock(&smc->clcsock_release_loc << 37 } << 38 23 39 static void smc_close_cleanup_listen(struct so 24 static void smc_close_cleanup_listen(struct sock *parent) 40 { 25 { 41 struct sock *sk; 26 struct sock *sk; 42 27 43 /* Close non-accepted connections */ 28 /* Close non-accepted connections */ 44 while ((sk = smc_accept_dequeue(parent 29 while ((sk = smc_accept_dequeue(parent, NULL))) 45 smc_close_non_accepted(sk); 30 smc_close_non_accepted(sk); 46 } 31 } 47 32 48 /* wait for sndbuf data being transmitted */ 33 /* wait for sndbuf data being transmitted */ 49 static void smc_close_stream_wait(struct smc_s 34 static void smc_close_stream_wait(struct smc_sock *smc, long timeout) 50 { 35 { 51 DEFINE_WAIT_FUNC(wait, woken_wake_func 36 DEFINE_WAIT_FUNC(wait, woken_wake_function); 52 struct sock *sk = &smc->sk; 37 struct sock *sk = &smc->sk; 53 38 54 if (!timeout) 39 if (!timeout) 55 return; 40 return; 56 41 57 if (!smc_tx_prepared_sends(&smc->conn) 42 if (!smc_tx_prepared_sends(&smc->conn)) 58 return; 43 return; 59 44 60 /* Send out corked data remaining in s << 61 smc_tx_pending(&smc->conn); << 62 << 63 smc->wait_close_tx_prepared = 1; 45 smc->wait_close_tx_prepared = 1; 64 add_wait_queue(sk_sleep(sk), &wait); 46 add_wait_queue(sk_sleep(sk), &wait); 65 while (!signal_pending(current) && tim 47 while (!signal_pending(current) && timeout) { 66 int rc; 48 int rc; 67 49 68 rc = sk_wait_event(sk, &timeou 50 rc = sk_wait_event(sk, &timeout, 69 !smc_tx_pre 51 !smc_tx_prepared_sends(&smc->conn) || 70 READ_ONCE(s !! 52 (sk->sk_err == ECONNABORTED) || 71 READ_ONCE(s !! 53 (sk->sk_err == ECONNRESET), 72 smc->conn.k << 73 &wait); 54 &wait); 74 if (rc) 55 if (rc) 75 break; 56 break; 76 } 57 } 77 remove_wait_queue(sk_sleep(sk), &wait) 58 remove_wait_queue(sk_sleep(sk), &wait); 78 smc->wait_close_tx_prepared = 0; 59 smc->wait_close_tx_prepared = 0; 79 } 60 } 80 61 81 void smc_close_wake_tx_prepared(struct smc_soc 62 void smc_close_wake_tx_prepared(struct smc_sock *smc) 82 { 63 { 83 if (smc->wait_close_tx_prepared) 64 if (smc->wait_close_tx_prepared) 84 /* wake up socket closing */ 65 /* wake up socket closing */ 85 smc->sk.sk_state_change(&smc-> 66 smc->sk.sk_state_change(&smc->sk); 86 } 67 } 87 68 88 static int smc_close_wr(struct smc_connection 69 static int smc_close_wr(struct smc_connection *conn) 89 { 70 { 90 conn->local_tx_ctrl.conn_state_flags.p 71 conn->local_tx_ctrl.conn_state_flags.peer_done_writing = 1; 91 72 92 return smc_cdc_get_slot_and_msg_send(c 73 return smc_cdc_get_slot_and_msg_send(conn); 93 } 74 } 94 75 95 static int smc_close_final(struct smc_connecti 76 static int smc_close_final(struct smc_connection *conn) 96 { 77 { 97 if (atomic_read(&conn->bytes_to_rcv)) 78 if (atomic_read(&conn->bytes_to_rcv)) 98 conn->local_tx_ctrl.conn_state 79 conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1; 99 else 80 else 100 conn->local_tx_ctrl.conn_state 81 conn->local_tx_ctrl.conn_state_flags.peer_conn_closed = 1; 101 if (conn->killed) << 102 return -EPIPE; << 103 82 104 return smc_cdc_get_slot_and_msg_send(c 83 return smc_cdc_get_slot_and_msg_send(conn); 105 } 84 } 106 85 107 int smc_close_abort(struct smc_connection *con !! 86 static int smc_close_abort(struct smc_connection *conn) 108 { 87 { 109 conn->local_tx_ctrl.conn_state_flags.p 88 conn->local_tx_ctrl.conn_state_flags.peer_conn_abort = 1; 110 89 111 return smc_cdc_get_slot_and_msg_send(c 90 return smc_cdc_get_slot_and_msg_send(conn); 112 } 91 } 113 92 114 static void smc_close_cancel_work(struct smc_s << 115 { << 116 struct sock *sk = &smc->sk; << 117 << 118 release_sock(sk); << 119 if (cancel_work_sync(&smc->conn.close_ << 120 sock_put(sk); << 121 cancel_delayed_work_sync(&smc->conn.tx << 122 lock_sock(sk); << 123 } << 124 << 125 /* terminate smc socket abnormally - active ab 93 /* terminate smc socket abnormally - active abort 126 * link group is terminated, i.e. RDMA communi 94 * link group is terminated, i.e. RDMA communication no longer possible 127 */ 95 */ 128 void smc_close_active_abort(struct smc_sock *s !! 96 static void smc_close_active_abort(struct smc_sock *smc) 129 { 97 { 130 struct sock *sk = &smc->sk; 98 struct sock *sk = &smc->sk; 131 bool release_clcsock = false; !! 99 >> 100 struct smc_cdc_conn_state_flags *txflags = >> 101 &smc->conn.local_tx_ctrl.conn_state_flags; 132 102 133 if (sk->sk_state != SMC_INIT && smc->c 103 if (sk->sk_state != SMC_INIT && smc->clcsock && smc->clcsock->sk) { 134 sk->sk_err = ECONNABORTED; 104 sk->sk_err = ECONNABORTED; 135 if (smc->clcsock && smc->clcso !! 105 if (smc->clcsock && smc->clcsock->sk) { 136 tcp_abort(smc->clcsock !! 106 smc->clcsock->sk->sk_err = ECONNABORTED; >> 107 smc->clcsock->sk->sk_state_change(smc->clcsock->sk); >> 108 } 137 } 109 } 138 switch (sk->sk_state) { 110 switch (sk->sk_state) { 139 case SMC_ACTIVE: 111 case SMC_ACTIVE: >> 112 sk->sk_state = SMC_PEERABORTWAIT; >> 113 release_sock(sk); >> 114 cancel_delayed_work_sync(&smc->conn.tx_work); >> 115 lock_sock(sk); >> 116 sock_put(sk); /* passive closing */ >> 117 break; 140 case SMC_APPCLOSEWAIT1: 118 case SMC_APPCLOSEWAIT1: 141 case SMC_APPCLOSEWAIT2: 119 case SMC_APPCLOSEWAIT2: 142 sk->sk_state = SMC_PEERABORTWA !! 120 if (!smc_cdc_rxed_any_close(&smc->conn)) 143 smc_close_cancel_work(smc); !! 121 sk->sk_state = SMC_PEERABORTWAIT; 144 if (sk->sk_state != SMC_PEERAB !! 122 else 145 break; !! 123 sk->sk_state = SMC_CLOSED; 146 sk->sk_state = SMC_CLOSED; !! 124 release_sock(sk); 147 sock_put(sk); /* (postponed) p !! 125 cancel_delayed_work_sync(&smc->conn.tx_work); >> 126 lock_sock(sk); 148 break; 127 break; 149 case SMC_PEERCLOSEWAIT1: 128 case SMC_PEERCLOSEWAIT1: 150 case SMC_PEERCLOSEWAIT2: 129 case SMC_PEERCLOSEWAIT2: 151 case SMC_PEERFINCLOSEWAIT: !! 130 if (!txflags->peer_conn_closed) { 152 sk->sk_state = SMC_PEERABORTWA !! 131 /* just SHUTDOWN_SEND done */ 153 smc_close_cancel_work(smc); !! 132 sk->sk_state = SMC_PEERABORTWAIT; 154 if (sk->sk_state != SMC_PEERAB !! 133 } else { 155 break; !! 134 sk->sk_state = SMC_CLOSED; 156 sk->sk_state = SMC_CLOSED; !! 135 } 157 smc_conn_free(&smc->conn); << 158 release_clcsock = true; << 159 sock_put(sk); /* passive closi 136 sock_put(sk); /* passive closing */ 160 break; 137 break; 161 case SMC_PROCESSABORT: 138 case SMC_PROCESSABORT: 162 case SMC_APPFINCLOSEWAIT: 139 case SMC_APPFINCLOSEWAIT: 163 sk->sk_state = SMC_PEERABORTWA << 164 smc_close_cancel_work(smc); << 165 if (sk->sk_state != SMC_PEERAB << 166 break; << 167 sk->sk_state = SMC_CLOSED; 140 sk->sk_state = SMC_CLOSED; 168 smc_conn_free(&smc->conn); !! 141 break; 169 release_clcsock = true; !! 142 case SMC_PEERFINCLOSEWAIT: >> 143 sock_put(sk); /* passive closing */ 170 break; 144 break; 171 case SMC_INIT: 145 case SMC_INIT: 172 case SMC_PEERABORTWAIT: 146 case SMC_PEERABORTWAIT: 173 case SMC_CLOSED: 147 case SMC_CLOSED: 174 break; 148 break; 175 } 149 } 176 150 177 smc_sock_set_flag(sk, SOCK_DEAD); !! 151 sock_set_flag(sk, SOCK_DEAD); 178 sk->sk_state_change(sk); 152 sk->sk_state_change(sk); 179 << 180 if (release_clcsock) { << 181 release_sock(sk); << 182 smc_clcsock_release(smc); << 183 lock_sock(sk); << 184 } << 185 } 153 } 186 154 187 static inline bool smc_close_sent_any_close(st 155 static inline bool smc_close_sent_any_close(struct smc_connection *conn) 188 { 156 { 189 return conn->local_tx_ctrl.conn_state_ 157 return conn->local_tx_ctrl.conn_state_flags.peer_conn_abort || 190 conn->local_tx_ctrl.conn_state_ 158 conn->local_tx_ctrl.conn_state_flags.peer_conn_closed; 191 } 159 } 192 160 193 int smc_close_active(struct smc_sock *smc) 161 int smc_close_active(struct smc_sock *smc) 194 { 162 { 195 struct smc_cdc_conn_state_flags *txfla 163 struct smc_cdc_conn_state_flags *txflags = 196 &smc->conn.local_tx_ctrl.conn_ 164 &smc->conn.local_tx_ctrl.conn_state_flags; 197 struct smc_connection *conn = &smc->co 165 struct smc_connection *conn = &smc->conn; 198 struct sock *sk = &smc->sk; 166 struct sock *sk = &smc->sk; 199 int old_state; 167 int old_state; 200 long timeout; 168 long timeout; 201 int rc = 0; 169 int rc = 0; 202 int rc1 = 0; << 203 170 204 timeout = current->flags & PF_EXITING 171 timeout = current->flags & PF_EXITING ? 205 0 : sock_flag(sk, SOCK_LINGE 172 0 : sock_flag(sk, SOCK_LINGER) ? 206 sk->sk_lingertime : SMC_ 173 sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT; 207 174 208 old_state = sk->sk_state; 175 old_state = sk->sk_state; 209 again: 176 again: 210 switch (sk->sk_state) { 177 switch (sk->sk_state) { 211 case SMC_INIT: 178 case SMC_INIT: 212 sk->sk_state = SMC_CLOSED; 179 sk->sk_state = SMC_CLOSED; 213 break; 180 break; 214 case SMC_LISTEN: 181 case SMC_LISTEN: 215 sk->sk_state = SMC_CLOSED; 182 sk->sk_state = SMC_CLOSED; 216 sk->sk_state_change(sk); /* wa 183 sk->sk_state_change(sk); /* wake up accept */ 217 if (smc->clcsock && smc->clcso 184 if (smc->clcsock && smc->clcsock->sk) { 218 write_lock_bh(&smc->cl << 219 smc_clcsock_restore_cb << 220 << 221 smc->clcsock->sk->sk_u << 222 write_unlock_bh(&smc-> << 223 rc = kernel_sock_shutd 185 rc = kernel_sock_shutdown(smc->clcsock, SHUT_RDWR); >> 186 /* wake up kernel_accept of smc_tcp_listen_worker */ >> 187 smc->clcsock->sk->sk_data_ready(smc->clcsock->sk); 224 } 188 } 225 smc_close_cleanup_listen(sk); 189 smc_close_cleanup_listen(sk); 226 release_sock(sk); 190 release_sock(sk); 227 flush_work(&smc->tcp_listen_wo 191 flush_work(&smc->tcp_listen_work); 228 lock_sock(sk); 192 lock_sock(sk); 229 break; 193 break; 230 case SMC_ACTIVE: 194 case SMC_ACTIVE: 231 smc_close_stream_wait(smc, tim 195 smc_close_stream_wait(smc, timeout); 232 release_sock(sk); 196 release_sock(sk); 233 cancel_delayed_work_sync(&conn 197 cancel_delayed_work_sync(&conn->tx_work); 234 lock_sock(sk); 198 lock_sock(sk); 235 if (sk->sk_state == SMC_ACTIVE 199 if (sk->sk_state == SMC_ACTIVE) { 236 /* send close request 200 /* send close request */ 237 rc = smc_close_final(c 201 rc = smc_close_final(conn); >> 202 if (rc) >> 203 break; 238 sk->sk_state = SMC_PEE 204 sk->sk_state = SMC_PEERCLOSEWAIT1; 239 << 240 /* actively shutdown c << 241 * prevent peer from e << 242 */ << 243 if (smc->clcsock && sm << 244 rc1 = kernel_s << 245 << 246 rc = rc ? rc : << 247 } << 248 } else { 205 } else { 249 /* peer event has chan 206 /* peer event has changed the state */ 250 goto again; 207 goto again; 251 } 208 } 252 break; 209 break; 253 case SMC_APPFINCLOSEWAIT: 210 case SMC_APPFINCLOSEWAIT: 254 /* socket already shutdown wr 211 /* socket already shutdown wr or both (active close) */ 255 if (txflags->peer_done_writing 212 if (txflags->peer_done_writing && 256 !smc_close_sent_any_close( 213 !smc_close_sent_any_close(conn)) { 257 /* just shutdown wr do 214 /* just shutdown wr done, send close request */ 258 rc = smc_close_final(c 215 rc = smc_close_final(conn); >> 216 if (rc) >> 217 break; 259 } 218 } 260 sk->sk_state = SMC_CLOSED; 219 sk->sk_state = SMC_CLOSED; 261 break; 220 break; 262 case SMC_APPCLOSEWAIT1: 221 case SMC_APPCLOSEWAIT1: 263 case SMC_APPCLOSEWAIT2: 222 case SMC_APPCLOSEWAIT2: 264 if (!smc_cdc_rxed_any_close(co 223 if (!smc_cdc_rxed_any_close(conn)) 265 smc_close_stream_wait( 224 smc_close_stream_wait(smc, timeout); 266 release_sock(sk); 225 release_sock(sk); 267 cancel_delayed_work_sync(&conn 226 cancel_delayed_work_sync(&conn->tx_work); 268 lock_sock(sk); 227 lock_sock(sk); 269 if (sk->sk_state != SMC_APPCLO 228 if (sk->sk_state != SMC_APPCLOSEWAIT1 && 270 sk->sk_state != SMC_APPCLO 229 sk->sk_state != SMC_APPCLOSEWAIT2) 271 goto again; 230 goto again; 272 /* confirm close from peer */ 231 /* confirm close from peer */ 273 rc = smc_close_final(conn); 232 rc = smc_close_final(conn); >> 233 if (rc) >> 234 break; 274 if (smc_cdc_rxed_any_close(con 235 if (smc_cdc_rxed_any_close(conn)) { 275 /* peer has closed the 236 /* peer has closed the socket already */ 276 sk->sk_state = SMC_CLO 237 sk->sk_state = SMC_CLOSED; 277 sock_put(sk); /* postp 238 sock_put(sk); /* postponed passive closing */ 278 } else { 239 } else { 279 /* peer has just issue 240 /* peer has just issued a shutdown write */ 280 sk->sk_state = SMC_PEE 241 sk->sk_state = SMC_PEERFINCLOSEWAIT; 281 } 242 } 282 break; 243 break; 283 case SMC_PEERCLOSEWAIT1: 244 case SMC_PEERCLOSEWAIT1: 284 case SMC_PEERCLOSEWAIT2: 245 case SMC_PEERCLOSEWAIT2: 285 if (txflags->peer_done_writing 246 if (txflags->peer_done_writing && 286 !smc_close_sent_any_close( 247 !smc_close_sent_any_close(conn)) { 287 /* just shutdown wr do 248 /* just shutdown wr done, send close request */ 288 rc = smc_close_final(c 249 rc = smc_close_final(conn); >> 250 if (rc) >> 251 break; 289 } 252 } 290 /* peer sending PeerConnection 253 /* peer sending PeerConnectionClosed will cause transition */ 291 break; 254 break; 292 case SMC_PEERFINCLOSEWAIT: 255 case SMC_PEERFINCLOSEWAIT: 293 /* peer sending PeerConnection 256 /* peer sending PeerConnectionClosed will cause transition */ 294 break; 257 break; 295 case SMC_PROCESSABORT: 258 case SMC_PROCESSABORT: 296 rc = smc_close_abort(conn); !! 259 smc_close_abort(conn); 297 sk->sk_state = SMC_CLOSED; 260 sk->sk_state = SMC_CLOSED; 298 break; 261 break; 299 case SMC_PEERABORTWAIT: 262 case SMC_PEERABORTWAIT: 300 sk->sk_state = SMC_CLOSED; << 301 break; << 302 case SMC_CLOSED: 263 case SMC_CLOSED: 303 /* nothing to do, add tracing 264 /* nothing to do, add tracing in future patch */ 304 break; 265 break; 305 } 266 } 306 267 307 if (old_state != sk->sk_state) 268 if (old_state != sk->sk_state) 308 sk->sk_state_change(sk); 269 sk->sk_state_change(sk); 309 return rc; 270 return rc; 310 } 271 } 311 272 312 static void smc_close_passive_abort_received(s 273 static void smc_close_passive_abort_received(struct smc_sock *smc) 313 { 274 { 314 struct smc_cdc_conn_state_flags *txfla 275 struct smc_cdc_conn_state_flags *txflags = 315 &smc->conn.local_tx_ctrl.conn_ 276 &smc->conn.local_tx_ctrl.conn_state_flags; 316 struct sock *sk = &smc->sk; 277 struct sock *sk = &smc->sk; 317 278 318 switch (sk->sk_state) { 279 switch (sk->sk_state) { 319 case SMC_INIT: 280 case SMC_INIT: 320 case SMC_ACTIVE: 281 case SMC_ACTIVE: 321 case SMC_APPCLOSEWAIT1: 282 case SMC_APPCLOSEWAIT1: 322 sk->sk_state = SMC_PROCESSABOR 283 sk->sk_state = SMC_PROCESSABORT; 323 sock_put(sk); /* passive closi 284 sock_put(sk); /* passive closing */ 324 break; 285 break; 325 case SMC_APPFINCLOSEWAIT: 286 case SMC_APPFINCLOSEWAIT: 326 sk->sk_state = SMC_PROCESSABOR 287 sk->sk_state = SMC_PROCESSABORT; 327 break; 288 break; 328 case SMC_PEERCLOSEWAIT1: 289 case SMC_PEERCLOSEWAIT1: 329 case SMC_PEERCLOSEWAIT2: 290 case SMC_PEERCLOSEWAIT2: 330 if (txflags->peer_done_writing 291 if (txflags->peer_done_writing && 331 !smc_close_sent_any_close( 292 !smc_close_sent_any_close(&smc->conn)) 332 /* just shutdown, but 293 /* just shutdown, but not yet closed locally */ 333 sk->sk_state = SMC_PRO 294 sk->sk_state = SMC_PROCESSABORT; 334 else 295 else 335 sk->sk_state = SMC_CLO 296 sk->sk_state = SMC_CLOSED; 336 sock_put(sk); /* passive closi 297 sock_put(sk); /* passive closing */ 337 break; 298 break; 338 case SMC_APPCLOSEWAIT2: 299 case SMC_APPCLOSEWAIT2: 339 case SMC_PEERFINCLOSEWAIT: 300 case SMC_PEERFINCLOSEWAIT: 340 sk->sk_state = SMC_CLOSED; 301 sk->sk_state = SMC_CLOSED; 341 sock_put(sk); /* passive closi 302 sock_put(sk); /* passive closing */ 342 break; 303 break; 343 case SMC_PEERABORTWAIT: 304 case SMC_PEERABORTWAIT: 344 sk->sk_state = SMC_CLOSED; 305 sk->sk_state = SMC_CLOSED; 345 break; 306 break; 346 case SMC_PROCESSABORT: 307 case SMC_PROCESSABORT: 347 /* nothing to do, add tracing in futur 308 /* nothing to do, add tracing in future patch */ 348 break; 309 break; 349 } 310 } 350 } 311 } 351 312 352 /* Either some kind of closing has been receiv 313 /* Either some kind of closing has been received: peer_conn_closed, 353 * peer_conn_abort, or peer_done_writing 314 * peer_conn_abort, or peer_done_writing 354 * or the link group of the connection termina 315 * or the link group of the connection terminates abnormally. 355 */ 316 */ 356 static void smc_close_passive_work(struct work 317 static void smc_close_passive_work(struct work_struct *work) 357 { 318 { 358 struct smc_connection *conn = containe 319 struct smc_connection *conn = container_of(work, 359 320 struct smc_connection, 360 321 close_work); 361 struct smc_sock *smc = container_of(co 322 struct smc_sock *smc = container_of(conn, struct smc_sock, conn); 362 struct smc_cdc_conn_state_flags *rxfla 323 struct smc_cdc_conn_state_flags *rxflags; 363 bool release_clcsock = false; << 364 struct sock *sk = &smc->sk; 324 struct sock *sk = &smc->sk; 365 int old_state; 325 int old_state; 366 326 367 lock_sock(sk); 327 lock_sock(sk); 368 old_state = sk->sk_state; 328 old_state = sk->sk_state; 369 329 >> 330 if (!conn->alert_token_local) { >> 331 /* abnormal termination */ >> 332 smc_close_active_abort(smc); >> 333 goto wakeup; >> 334 } >> 335 370 rxflags = &conn->local_rx_ctrl.conn_st 336 rxflags = &conn->local_rx_ctrl.conn_state_flags; 371 if (rxflags->peer_conn_abort) { 337 if (rxflags->peer_conn_abort) { 372 /* peer has not received all d 338 /* peer has not received all data */ 373 smc_close_passive_abort_receiv 339 smc_close_passive_abort_received(smc); 374 release_sock(sk); !! 340 release_sock(&smc->sk); 375 cancel_delayed_work_sync(&conn 341 cancel_delayed_work_sync(&conn->tx_work); 376 lock_sock(sk); !! 342 lock_sock(&smc->sk); 377 goto wakeup; 343 goto wakeup; 378 } 344 } 379 345 380 switch (sk->sk_state) { 346 switch (sk->sk_state) { 381 case SMC_INIT: 347 case SMC_INIT: 382 sk->sk_state = SMC_APPCLOSEWAI 348 sk->sk_state = SMC_APPCLOSEWAIT1; 383 break; 349 break; 384 case SMC_ACTIVE: 350 case SMC_ACTIVE: 385 sk->sk_state = SMC_APPCLOSEWAI 351 sk->sk_state = SMC_APPCLOSEWAIT1; 386 /* postpone sock_put() for pas 352 /* postpone sock_put() for passive closing to cover 387 * received SEND_SHUTDOWN as w 353 * received SEND_SHUTDOWN as well 388 */ 354 */ 389 break; 355 break; 390 case SMC_PEERCLOSEWAIT1: 356 case SMC_PEERCLOSEWAIT1: 391 if (rxflags->peer_done_writing 357 if (rxflags->peer_done_writing) 392 sk->sk_state = SMC_PEE 358 sk->sk_state = SMC_PEERCLOSEWAIT2; 393 fallthrough; !! 359 /* fall through */ 394 /* to check for closing */ 360 /* to check for closing */ 395 case SMC_PEERCLOSEWAIT2: 361 case SMC_PEERCLOSEWAIT2: 396 if (!smc_cdc_rxed_any_close(co 362 if (!smc_cdc_rxed_any_close(conn)) 397 break; 363 break; 398 if (sock_flag(sk, SOCK_DEAD) & 364 if (sock_flag(sk, SOCK_DEAD) && 399 smc_close_sent_any_close(c 365 smc_close_sent_any_close(conn)) { 400 /* smc_release has alr 366 /* smc_release has already been called locally */ 401 sk->sk_state = SMC_CLO 367 sk->sk_state = SMC_CLOSED; 402 } else { 368 } else { 403 /* just shutdown, but 369 /* just shutdown, but not yet closed locally */ 404 sk->sk_state = SMC_APP 370 sk->sk_state = SMC_APPFINCLOSEWAIT; 405 } 371 } 406 sock_put(sk); /* passive closi 372 sock_put(sk); /* passive closing */ 407 break; 373 break; 408 case SMC_PEERFINCLOSEWAIT: 374 case SMC_PEERFINCLOSEWAIT: 409 if (smc_cdc_rxed_any_close(con 375 if (smc_cdc_rxed_any_close(conn)) { 410 sk->sk_state = SMC_CLO 376 sk->sk_state = SMC_CLOSED; 411 sock_put(sk); /* passi 377 sock_put(sk); /* passive closing */ 412 } 378 } 413 break; 379 break; 414 case SMC_APPCLOSEWAIT1: 380 case SMC_APPCLOSEWAIT1: 415 case SMC_APPCLOSEWAIT2: 381 case SMC_APPCLOSEWAIT2: 416 /* postpone sock_put() for pas 382 /* postpone sock_put() for passive closing to cover 417 * received SEND_SHUTDOWN as w 383 * received SEND_SHUTDOWN as well 418 */ 384 */ 419 break; 385 break; 420 case SMC_APPFINCLOSEWAIT: 386 case SMC_APPFINCLOSEWAIT: 421 case SMC_PEERABORTWAIT: 387 case SMC_PEERABORTWAIT: 422 case SMC_PROCESSABORT: 388 case SMC_PROCESSABORT: 423 case SMC_CLOSED: 389 case SMC_CLOSED: 424 /* nothing to do, add tracing 390 /* nothing to do, add tracing in future patch */ 425 break; 391 break; 426 } 392 } 427 393 428 wakeup: 394 wakeup: 429 sk->sk_data_ready(sk); /* wakeup block 395 sk->sk_data_ready(sk); /* wakeup blocked rcvbuf consumers */ 430 sk->sk_write_space(sk); /* wakeup bloc 396 sk->sk_write_space(sk); /* wakeup blocked sndbuf producers */ 431 397 432 if (old_state != sk->sk_state) { 398 if (old_state != sk->sk_state) { 433 sk->sk_state_change(sk); 399 sk->sk_state_change(sk); 434 if ((sk->sk_state == SMC_CLOSE 400 if ((sk->sk_state == SMC_CLOSED) && 435 (sock_flag(sk, SOCK_DEAD) !! 401 (sock_flag(sk, SOCK_DEAD) || !sk->sk_socket)) 436 smc_conn_free(conn); 402 smc_conn_free(conn); 437 if (smc->clcsock) << 438 release_clcsoc << 439 } << 440 } 403 } 441 release_sock(sk); 404 release_sock(sk); 442 if (release_clcsock) << 443 smc_clcsock_release(smc); << 444 sock_put(sk); /* sock_hold done by sch 405 sock_put(sk); /* sock_hold done by schedulers of close_work */ 445 } 406 } 446 407 447 int smc_close_shutdown_write(struct smc_sock * 408 int smc_close_shutdown_write(struct smc_sock *smc) 448 { 409 { 449 struct smc_connection *conn = &smc->co 410 struct smc_connection *conn = &smc->conn; 450 struct sock *sk = &smc->sk; 411 struct sock *sk = &smc->sk; 451 int old_state; 412 int old_state; 452 long timeout; 413 long timeout; 453 int rc = 0; 414 int rc = 0; 454 415 455 timeout = current->flags & PF_EXITING 416 timeout = current->flags & PF_EXITING ? 456 0 : sock_flag(sk, SOCK_LINGE 417 0 : sock_flag(sk, SOCK_LINGER) ? 457 sk->sk_lingertime : SMC_ 418 sk->sk_lingertime : SMC_MAX_STREAM_WAIT_TIMEOUT; 458 419 459 old_state = sk->sk_state; 420 old_state = sk->sk_state; 460 again: 421 again: 461 switch (sk->sk_state) { 422 switch (sk->sk_state) { 462 case SMC_ACTIVE: 423 case SMC_ACTIVE: 463 smc_close_stream_wait(smc, tim 424 smc_close_stream_wait(smc, timeout); 464 release_sock(sk); 425 release_sock(sk); 465 cancel_delayed_work_sync(&conn 426 cancel_delayed_work_sync(&conn->tx_work); 466 lock_sock(sk); 427 lock_sock(sk); 467 if (sk->sk_state != SMC_ACTIVE 428 if (sk->sk_state != SMC_ACTIVE) 468 goto again; 429 goto again; 469 /* send close wr request */ 430 /* send close wr request */ 470 rc = smc_close_wr(conn); 431 rc = smc_close_wr(conn); >> 432 if (rc) >> 433 break; 471 sk->sk_state = SMC_PEERCLOSEWA 434 sk->sk_state = SMC_PEERCLOSEWAIT1; 472 break; 435 break; 473 case SMC_APPCLOSEWAIT1: 436 case SMC_APPCLOSEWAIT1: 474 /* passive close */ 437 /* passive close */ 475 if (!smc_cdc_rxed_any_close(co 438 if (!smc_cdc_rxed_any_close(conn)) 476 smc_close_stream_wait( 439 smc_close_stream_wait(smc, timeout); 477 release_sock(sk); 440 release_sock(sk); 478 cancel_delayed_work_sync(&conn 441 cancel_delayed_work_sync(&conn->tx_work); 479 lock_sock(sk); 442 lock_sock(sk); 480 if (sk->sk_state != SMC_APPCLO 443 if (sk->sk_state != SMC_APPCLOSEWAIT1) 481 goto again; 444 goto again; 482 /* confirm close from peer */ 445 /* confirm close from peer */ 483 rc = smc_close_wr(conn); 446 rc = smc_close_wr(conn); >> 447 if (rc) >> 448 break; 484 sk->sk_state = SMC_APPCLOSEWAI 449 sk->sk_state = SMC_APPCLOSEWAIT2; 485 break; 450 break; 486 case SMC_APPCLOSEWAIT2: 451 case SMC_APPCLOSEWAIT2: 487 case SMC_PEERFINCLOSEWAIT: 452 case SMC_PEERFINCLOSEWAIT: 488 case SMC_PEERCLOSEWAIT1: 453 case SMC_PEERCLOSEWAIT1: 489 case SMC_PEERCLOSEWAIT2: 454 case SMC_PEERCLOSEWAIT2: 490 case SMC_APPFINCLOSEWAIT: 455 case SMC_APPFINCLOSEWAIT: 491 case SMC_PROCESSABORT: 456 case SMC_PROCESSABORT: 492 case SMC_PEERABORTWAIT: 457 case SMC_PEERABORTWAIT: 493 /* nothing to do, add tracing 458 /* nothing to do, add tracing in future patch */ 494 break; 459 break; 495 } 460 } 496 461 497 if (old_state != sk->sk_state) 462 if (old_state != sk->sk_state) 498 sk->sk_state_change(sk); 463 sk->sk_state_change(sk); 499 return rc; 464 return rc; 500 } 465 } 501 466 502 /* Initialize close properties on connection e 467 /* Initialize close properties on connection establishment. */ 503 void smc_close_init(struct smc_sock *smc) 468 void smc_close_init(struct smc_sock *smc) 504 { 469 { 505 INIT_WORK(&smc->conn.close_work, smc_c 470 INIT_WORK(&smc->conn.close_work, smc_close_passive_work); 506 } 471 } 507 472
Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.