290 static void nbd_mark_nsock_dead(struct nbd_device *nbd, struct nbd_sock *nsock,  in nbd_mark_nsock_dead()  argument
293 	if (!nsock->dead && notify && !nbd_disconnected(nbd->config)) {  in nbd_mark_nsock_dead()
302 	if (!nsock->dead) {  in nbd_mark_nsock_dead()
303 		kernel_sock_shutdown(nsock->sock, SHUT_RDWR);  in nbd_mark_nsock_dead()
314 	nsock->dead = true;  in nbd_mark_nsock_dead()
315 	nsock->pending = NULL;  in nbd_mark_nsock_dead()
316 	nsock->sent = 0;  in nbd_mark_nsock_dead()
372 		struct nbd_sock *nsock = config->socks[i];  in sock_shutdown()  local
373 		mutex_lock(&nsock->tx_lock);  in sock_shutdown()
374 		nbd_mark_nsock_dead(nbd, nsock, 0);  in sock_shutdown()
375 		mutex_unlock(&nsock->tx_lock);  in sock_shutdown()
432 				struct nbd_sock *nsock =  in nbd_xmit_timeout()  local
434 				mutex_lock(&nsock->tx_lock);  in nbd_xmit_timeout()
441 				if (cmd->cookie == nsock->cookie)  in nbd_xmit_timeout()
442 					nbd_mark_nsock_dead(nbd, nsock, 1);  in nbd_xmit_timeout()
443 				mutex_unlock(&nsock->tx_lock);  in nbd_xmit_timeout()
457 		struct nbd_sock *nsock = config->socks[cmd->index];  in nbd_xmit_timeout()  local
464 		mutex_lock(&nsock->tx_lock);  in nbd_xmit_timeout()
465 		if (cmd->cookie != nsock->cookie) {  in nbd_xmit_timeout()
467 			mutex_unlock(&nsock->tx_lock);  in nbd_xmit_timeout()
472 		mutex_unlock(&nsock->tx_lock);  in nbd_xmit_timeout()
554 	struct nbd_sock *nsock = config->socks[index];  in nbd_send_cmd()  local
564 	int sent = nsock->sent, skip = 0;  in nbd_send_cmd()
600 	cmd->cookie = nsock->cookie;  in nbd_send_cmd()
626 				nsock->pending = req;  in nbd_send_cmd()
627 				nsock->sent = sent;  in nbd_send_cmd()
668 					nsock->pending = req;  in nbd_send_cmd()
669 					nsock->sent = sent;  in nbd_send_cmd()
691 	nsock->pending = NULL;  in nbd_send_cmd()
692 	nsock->sent = 0;  in nbd_send_cmd()
829 	struct nbd_sock *nsock;  in recv_work()  local
871 	nsock = config->socks[args->index];  in recv_work()
872 	mutex_lock(&nsock->tx_lock);  in recv_work()
873 	nbd_mark_nsock_dead(nbd, nsock, 1);  in recv_work()
874 	mutex_unlock(&nsock->tx_lock);  in recv_work()
914 	struct nbd_sock *nsock = config->socks[index];  in find_fallback()  local
915 	int fallback = nsock->fallback_index;  in find_fallback()
930 	if (nsock->fallback_index < 0 ||  in find_fallback()
931 	    nsock->fallback_index >= config->num_connections ||  in find_fallback()
932 	    config->socks[nsock->fallback_index]->dead) {  in find_fallback()
942 		nsock->fallback_index = new_index;  in find_fallback()
949 	new_index = nsock->fallback_index;  in find_fallback()
974 	struct nbd_sock *nsock;  in nbd_handle_cmd()  local
992 	nsock = config->socks[index];  in nbd_handle_cmd()
993 	mutex_lock(&nsock->tx_lock);  in nbd_handle_cmd()
994 	if (nsock->dead) {  in nbd_handle_cmd()
997 		mutex_unlock(&nsock->tx_lock);  in nbd_handle_cmd()
1022 	if (unlikely(nsock->pending && nsock->pending != req)) {  in nbd_handle_cmd()
1041 		nbd_mark_nsock_dead(nbd, nsock, 1);  in nbd_handle_cmd()
1046 	mutex_unlock(&nsock->tx_lock);  in nbd_handle_cmd()
1110 	struct nbd_sock *nsock;  in nbd_add_socket()  local
1136 	nsock = kzalloc(sizeof(*nsock), GFP_KERNEL);  in nbd_add_socket()
1137 	if (!nsock) {  in nbd_add_socket()
1145 		kfree(nsock);  in nbd_add_socket()
1152 	nsock->fallback_index = -1;  in nbd_add_socket()
1153 	nsock->dead = false;  in nbd_add_socket()
1154 	mutex_init(&nsock->tx_lock);  in nbd_add_socket()
1155 	nsock->sock = sock;  in nbd_add_socket()
1156 	nsock->pending = NULL;  in nbd_add_socket()
1157 	nsock->sent = 0;  in nbd_add_socket()
1158 	nsock->cookie = 0;  in nbd_add_socket()
1159 	socks[config->num_connections++] = nsock;  in nbd_add_socket()
1190 		struct nbd_sock *nsock = config->socks[i];  in nbd_reconnect_socket()  local
1192 		if (!nsock->dead)  in nbd_reconnect_socket()
1195 		mutex_lock(&nsock->tx_lock);  in nbd_reconnect_socket()
1196 		if (!nsock->dead) {  in nbd_reconnect_socket()
1197 			mutex_unlock(&nsock->tx_lock);  in nbd_reconnect_socket()
1205 		old = nsock->sock;  in nbd_reconnect_socket()
1206 		nsock->fallback_index = -1;  in nbd_reconnect_socket()
1207 		nsock->sock = sock;  in nbd_reconnect_socket()
1208 		nsock->dead = false;  in nbd_reconnect_socket()
1212 		nsock->cookie++;  in nbd_reconnect_socket()
1213 		mutex_unlock(&nsock->tx_lock);  in nbd_reconnect_socket()
1268 		struct nbd_sock *nsock = config->socks[i];  in send_disconnects()  local
1271 		mutex_lock(&nsock->tx_lock);  in send_disconnects()
1276 		mutex_unlock(&nsock->tx_lock);  in send_disconnects()