33  * queue before determining it to be idle.  This optional module behavior
66 	struct nvmet_tcp_queue		*queue;  member
170 static inline u16 nvmet_tcp_cmd_tag(struct nvmet_tcp_queue *queue,  in nvmet_tcp_cmd_tag()  argument
173 	if (unlikely(!queue->nr_cmds)) {  in nvmet_tcp_cmd_tag()
178 	return cmd - queue->cmds;  in nvmet_tcp_cmd_tag()
206 nvmet_tcp_get_cmd(struct nvmet_tcp_queue *queue)  in nvmet_tcp_get_cmd()  argument
210 	cmd = list_first_entry_or_null(&queue->free_list,  in nvmet_tcp_get_cmd()
226 	if (unlikely(cmd == &cmd->queue->connect))  in nvmet_tcp_put_cmd()
229 	list_add_tail(&cmd->entry, &cmd->queue->free_list);  in nvmet_tcp_put_cmd()
232 static inline int queue_cpu(struct nvmet_tcp_queue *queue)  in queue_cpu()  argument
234 	return queue->sock->sk->sk_incoming_cpu;  in queue_cpu()
237 static inline u8 nvmet_tcp_hdgst_len(struct nvmet_tcp_queue *queue)  in nvmet_tcp_hdgst_len()  argument
239 	return queue->hdr_digest ? NVME_TCP_DIGEST_LENGTH : 0;  in nvmet_tcp_hdgst_len()
242 static inline u8 nvmet_tcp_ddgst_len(struct nvmet_tcp_queue *queue)  in nvmet_tcp_ddgst_len()  argument
244 	return queue->data_digest ? NVME_TCP_DIGEST_LENGTH : 0;  in nvmet_tcp_ddgst_len()
257 static int nvmet_tcp_verify_hdgst(struct nvmet_tcp_queue *queue,  in nvmet_tcp_verify_hdgst()  argument
265 		pr_err("queue %d: header digest enabled but no header digest\n",  in nvmet_tcp_verify_hdgst()
266 			queue->idx);  in nvmet_tcp_verify_hdgst()
271 	nvmet_tcp_hdgst(queue->rcv_hash, pdu, len);  in nvmet_tcp_verify_hdgst()
274 		pr_err("queue %d: header digest error: recv %#x expected %#x\n",  in nvmet_tcp_verify_hdgst()
275 			queue->idx, le32_to_cpu(recv_digest),  in nvmet_tcp_verify_hdgst()
283 static int nvmet_tcp_check_ddgst(struct nvmet_tcp_queue *queue, void *pdu)  in nvmet_tcp_check_ddgst()  argument
286 	u8 digest_len = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_check_ddgst()
293 		pr_err("queue %d: data digest flag is cleared\n", queue->idx);  in nvmet_tcp_check_ddgst()
340 static void nvmet_tcp_fatal_error(struct nvmet_tcp_queue *queue)  in nvmet_tcp_fatal_error()  argument
342 	queue->rcv_state = NVMET_TCP_RECV_ERR;  in nvmet_tcp_fatal_error()
343 	if (queue->nvme_sq.ctrl)  in nvmet_tcp_fatal_error()
344 		nvmet_ctrl_fatal_error(queue->nvme_sq.ctrl);  in nvmet_tcp_fatal_error()
346 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_fatal_error()
349 static void nvmet_tcp_socket_error(struct nvmet_tcp_queue *queue, int status)  in nvmet_tcp_socket_error()  argument
352 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_socket_error()
354 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_socket_error()
422 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_c2h_data_pdu()  local
423 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_c2h_data_pdu()
424 	u8 ddgst = nvmet_tcp_ddgst_len(cmd->queue);  in nvmet_setup_c2h_data_pdu()
430 	pdu->hdr.flags = NVME_TCP_F_DATA_LAST | (queue->nvme_sq.sqhd_disabled ?  in nvmet_setup_c2h_data_pdu()
441 	if (queue->data_digest) {  in nvmet_setup_c2h_data_pdu()
443 		nvmet_tcp_send_ddgst(queue->snd_hash, cmd);  in nvmet_setup_c2h_data_pdu()
446 	if (cmd->queue->hdr_digest) {  in nvmet_setup_c2h_data_pdu()
448 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_c2h_data_pdu()
455 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_r2t_pdu()  local
456 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_r2t_pdu()
468 	pdu->ttag = nvmet_tcp_cmd_tag(cmd->queue, cmd);  in nvmet_setup_r2t_pdu()
471 	if (cmd->queue->hdr_digest) {  in nvmet_setup_r2t_pdu()
473 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_r2t_pdu()
480 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_response_pdu()  local
481 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_response_pdu()
491 	if (cmd->queue->hdr_digest) {  in nvmet_setup_response_pdu()
493 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_response_pdu()
497 static void nvmet_tcp_process_resp_list(struct nvmet_tcp_queue *queue)  in nvmet_tcp_process_resp_list()  argument
502 	for (node = llist_del_all(&queue->resp_list); node; node = node->next) {  in nvmet_tcp_process_resp_list()
504 		list_add(&cmd->entry, &queue->resp_send_list);  in nvmet_tcp_process_resp_list()
505 		queue->send_list_len++;  in nvmet_tcp_process_resp_list()
509 static struct nvmet_tcp_cmd *nvmet_tcp_fetch_cmd(struct nvmet_tcp_queue *queue)  in nvmet_tcp_fetch_cmd()  argument
511 	queue->snd_cmd = list_first_entry_or_null(&queue->resp_send_list,  in nvmet_tcp_fetch_cmd()
513 	if (!queue->snd_cmd) {  in nvmet_tcp_fetch_cmd()
514 		nvmet_tcp_process_resp_list(queue);  in nvmet_tcp_fetch_cmd()
515 		queue->snd_cmd =  in nvmet_tcp_fetch_cmd()
516 			list_first_entry_or_null(&queue->resp_send_list,  in nvmet_tcp_fetch_cmd()
518 		if (unlikely(!queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
522 	list_del_init(&queue->snd_cmd->entry);  in nvmet_tcp_fetch_cmd()
523 	queue->send_list_len--;  in nvmet_tcp_fetch_cmd()
525 	if (nvmet_tcp_need_data_out(queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
526 		nvmet_setup_c2h_data_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
527 	else if (nvmet_tcp_need_data_in(queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
528 		nvmet_setup_r2t_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
530 		nvmet_setup_response_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
532 	return queue->snd_cmd;  in nvmet_tcp_fetch_cmd()
539 	struct nvmet_tcp_queue	*queue = cmd->queue;  in nvmet_tcp_queue_response()  local
543 	if (unlikely(cmd == queue->cmd)) {  in nvmet_tcp_queue_response()
552 		if (queue->rcv_state == NVMET_TCP_RECV_PDU &&  in nvmet_tcp_queue_response()
558 	llist_add(&cmd->lentry, &queue->resp_list);  in nvmet_tcp_queue_response()
559 	queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &cmd->queue->io_work);  in nvmet_tcp_queue_response()
572 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_data_pdu()
576 	ret = kernel_sendpage(cmd->queue->sock, virt_to_page(cmd->data_pdu),  in nvmet_try_send_data_pdu()
595 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_try_send_data()  local
603 		if ((!last_in_batch && cmd->queue->send_list_len) ||  in nvmet_try_send_data()
605 		    queue->data_digest || !queue->nvme_sq.sqhd_disabled)  in nvmet_try_send_data()
608 		ret = kernel_sendpage(cmd->queue->sock, page, cmd->offset,  in nvmet_try_send_data()
623 	if (queue->data_digest) {  in nvmet_try_send_data()
627 		if (queue->nvme_sq.sqhd_disabled) {  in nvmet_try_send_data()
628 			cmd->queue->snd_cmd = NULL;  in nvmet_try_send_data()
635 	if (queue->nvme_sq.sqhd_disabled) {  in nvmet_try_send_data()
647 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_response()
652 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_response()
657 	ret = kernel_sendpage(cmd->queue->sock, virt_to_page(cmd->rsp_pdu),  in nvmet_try_send_response()
669 	cmd->queue->snd_cmd = NULL;  in nvmet_try_send_response()
676 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_r2t()
681 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_r2t()
686 	ret = kernel_sendpage(cmd->queue->sock, virt_to_page(cmd->r2t_pdu),  in nvmet_try_send_r2t()
696 	cmd->queue->snd_cmd = NULL;  in nvmet_try_send_r2t()
702 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_try_send_ddgst()  local
710 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_ddgst()
715 	ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);  in nvmet_try_send_ddgst()
721 	if (queue->nvme_sq.sqhd_disabled) {  in nvmet_try_send_ddgst()
722 		cmd->queue->snd_cmd = NULL;  in nvmet_try_send_ddgst()
730 static int nvmet_tcp_try_send_one(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_send_one()  argument
733 	struct nvmet_tcp_cmd *cmd = queue->snd_cmd;  in nvmet_tcp_try_send_one()
736 	if (!cmd || queue->state == NVMET_TCP_Q_DISCONNECTING) {  in nvmet_tcp_try_send_one()
737 		cmd = nvmet_tcp_fetch_cmd(queue);  in nvmet_tcp_try_send_one()
779 static int nvmet_tcp_try_send(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_send()  argument
785 		ret = nvmet_tcp_try_send_one(queue, i == budget - 1);  in nvmet_tcp_try_send()
787 			nvmet_tcp_socket_error(queue, ret);  in nvmet_tcp_try_send()
798 static void nvmet_prepare_receive_pdu(struct nvmet_tcp_queue *queue)  in nvmet_prepare_receive_pdu()  argument
800 	queue->offset = 0;  in nvmet_prepare_receive_pdu()
801 	queue->left = sizeof(struct nvme_tcp_hdr);  in nvmet_prepare_receive_pdu()
802 	queue->cmd = NULL;  in nvmet_prepare_receive_pdu()
803 	queue->rcv_state = NVMET_TCP_RECV_PDU;  in nvmet_prepare_receive_pdu()
806 static void nvmet_tcp_free_crypto(struct nvmet_tcp_queue *queue)  in nvmet_tcp_free_crypto()  argument
808 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(queue->rcv_hash);  in nvmet_tcp_free_crypto()
810 	ahash_request_free(queue->rcv_hash);  in nvmet_tcp_free_crypto()
811 	ahash_request_free(queue->snd_hash);  in nvmet_tcp_free_crypto()
815 static int nvmet_tcp_alloc_crypto(struct nvmet_tcp_queue *queue)  in nvmet_tcp_alloc_crypto()  argument
823 	queue->snd_hash = ahash_request_alloc(tfm, GFP_KERNEL);  in nvmet_tcp_alloc_crypto()
824 	if (!queue->snd_hash)  in nvmet_tcp_alloc_crypto()
826 	ahash_request_set_callback(queue->snd_hash, 0, NULL, NULL);  in nvmet_tcp_alloc_crypto()
828 	queue->rcv_hash = ahash_request_alloc(tfm, GFP_KERNEL);  in nvmet_tcp_alloc_crypto()
829 	if (!queue->rcv_hash)  in nvmet_tcp_alloc_crypto()
831 	ahash_request_set_callback(queue->rcv_hash, 0, NULL, NULL);  in nvmet_tcp_alloc_crypto()
835 	ahash_request_free(queue->snd_hash);  in nvmet_tcp_alloc_crypto()
842 static int nvmet_tcp_handle_icreq(struct nvmet_tcp_queue *queue)  in nvmet_tcp_handle_icreq()  argument
844 	struct nvme_tcp_icreq_pdu *icreq = &queue->pdu.icreq;  in nvmet_tcp_handle_icreq()
845 	struct nvme_tcp_icresp_pdu *icresp = &queue->pdu.icresp;  in nvmet_tcp_handle_icreq()
853 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_handle_icreq()
857 		pr_err("queue %d: bad pfv %d\n", queue->idx, icreq->pfv);  in nvmet_tcp_handle_icreq()
862 		pr_err("queue %d: unsupported hpda %d\n", queue->idx,  in nvmet_tcp_handle_icreq()
867 	queue->hdr_digest = !!(icreq->digest & NVME_TCP_HDR_DIGEST_ENABLE);  in nvmet_tcp_handle_icreq()
868 	queue->data_digest = !!(icreq->digest & NVME_TCP_DATA_DIGEST_ENABLE);  in nvmet_tcp_handle_icreq()
869 	if (queue->hdr_digest || queue->data_digest) {  in nvmet_tcp_handle_icreq()
870 		ret = nvmet_tcp_alloc_crypto(queue);  in nvmet_tcp_handle_icreq()
883 	if (queue->hdr_digest)  in nvmet_tcp_handle_icreq()
885 	if (queue->data_digest)  in nvmet_tcp_handle_icreq()
890 	ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);  in nvmet_tcp_handle_icreq()
894 	queue->state = NVMET_TCP_Q_LIVE;  in nvmet_tcp_handle_icreq()
895 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_handle_icreq()
898 	if (queue->hdr_digest || queue->data_digest)  in nvmet_tcp_handle_icreq()
899 		nvmet_tcp_free_crypto(queue);  in nvmet_tcp_handle_icreq()
903 static void nvmet_tcp_handle_req_failure(struct nvmet_tcp_queue *queue,  in nvmet_tcp_handle_req_failure()  argument
911 		nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_handle_req_failure()
917 		pr_err("queue %d: failed to map data\n", queue->idx);  in nvmet_tcp_handle_req_failure()
918 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_handle_req_failure()
922 	queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_handle_req_failure()
927 static int nvmet_tcp_handle_h2c_data_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_handle_h2c_data_pdu()  argument
929 	struct nvme_tcp_data_pdu *data = &queue->pdu.data;  in nvmet_tcp_handle_h2c_data_pdu()
932 	if (likely(queue->nr_cmds))  in nvmet_tcp_handle_h2c_data_pdu()
933 		cmd = &queue->cmds[data->ttag];  in nvmet_tcp_handle_h2c_data_pdu()
935 		cmd = &queue->connect;  in nvmet_tcp_handle_h2c_data_pdu()
950 	queue->cmd = cmd;  in nvmet_tcp_handle_h2c_data_pdu()
951 	queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_handle_h2c_data_pdu()
956 static int nvmet_tcp_done_recv_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_done_recv_pdu()  argument
958 	struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;  in nvmet_tcp_done_recv_pdu()
959 	struct nvme_command *nvme_cmd = &queue->pdu.cmd.cmd;  in nvmet_tcp_done_recv_pdu()
963 	if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {  in nvmet_tcp_done_recv_pdu()
967 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
970 		return nvmet_tcp_handle_icreq(queue);  in nvmet_tcp_done_recv_pdu()
974 		ret = nvmet_tcp_handle_h2c_data_pdu(queue);  in nvmet_tcp_done_recv_pdu()
980 	queue->cmd = nvmet_tcp_get_cmd(queue);  in nvmet_tcp_done_recv_pdu()
981 	if (unlikely(!queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
983 		pr_err("queue %d: out of commands (%d) send_list_len: %d, opcode: %d",  in nvmet_tcp_done_recv_pdu()
984 			queue->idx, queue->nr_cmds, queue->send_list_len,  in nvmet_tcp_done_recv_pdu()
986 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
990 	req = &queue->cmd->req;  in nvmet_tcp_done_recv_pdu()
993 	if (unlikely(!nvmet_req_init(req, &queue->nvme_cq,  in nvmet_tcp_done_recv_pdu()
994 			&queue->nvme_sq, &nvmet_tcp_ops))) {  in nvmet_tcp_done_recv_pdu()
1000 		nvmet_tcp_handle_req_failure(queue, queue->cmd, req);  in nvmet_tcp_done_recv_pdu()
1004 	ret = nvmet_tcp_map_data(queue->cmd);  in nvmet_tcp_done_recv_pdu()
1006 		pr_err("queue %d: failed to map data\n", queue->idx);  in nvmet_tcp_done_recv_pdu()
1007 		if (nvmet_tcp_has_inline_data(queue->cmd))  in nvmet_tcp_done_recv_pdu()
1008 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
1015 	if (nvmet_tcp_need_data_in(queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
1016 		if (nvmet_tcp_has_inline_data(queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
1017 			queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_done_recv_pdu()
1018 			nvmet_tcp_map_pdu_iovec(queue->cmd);  in nvmet_tcp_done_recv_pdu()
1022 		nvmet_tcp_queue_response(&queue->cmd->req);  in nvmet_tcp_done_recv_pdu()
1026 	queue->cmd->req.execute(&queue->cmd->req);  in nvmet_tcp_done_recv_pdu()
1028 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_done_recv_pdu()
1060 static int nvmet_tcp_try_recv_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_pdu()  argument
1062 	struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;  in nvmet_tcp_try_recv_pdu()
1068 	iov.iov_base = (void *)&queue->pdu + queue->offset;  in nvmet_tcp_try_recv_pdu()
1069 	iov.iov_len = queue->left;  in nvmet_tcp_try_recv_pdu()
1070 	len = kernel_recvmsg(queue->sock, &msg, &iov, 1,  in nvmet_tcp_try_recv_pdu()
1075 	queue->offset += len;  in nvmet_tcp_try_recv_pdu()
1076 	queue->left -= len;  in nvmet_tcp_try_recv_pdu()
1077 	if (queue->left)  in nvmet_tcp_try_recv_pdu()
1080 	if (queue->offset == sizeof(struct nvme_tcp_hdr)) {  in nvmet_tcp_try_recv_pdu()
1081 		u8 hdgst = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_try_recv_pdu()
1085 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_try_recv_pdu()
1094 		queue->left = hdr->hlen - queue->offset + hdgst;  in nvmet_tcp_try_recv_pdu()
1098 	if (queue->hdr_digest &&  in nvmet_tcp_try_recv_pdu()
1099 	    nvmet_tcp_verify_hdgst(queue, &queue->pdu, hdr->hlen)) {  in nvmet_tcp_try_recv_pdu()
1100 		nvmet_tcp_fatal_error(queue); /* fatal */  in nvmet_tcp_try_recv_pdu()
1104 	if (queue->data_digest &&  in nvmet_tcp_try_recv_pdu()
1105 	    nvmet_tcp_check_ddgst(queue, &queue->pdu)) {  in nvmet_tcp_try_recv_pdu()
1106 		nvmet_tcp_fatal_error(queue); /* fatal */  in nvmet_tcp_try_recv_pdu()
1110 	return nvmet_tcp_done_recv_pdu(queue);  in nvmet_tcp_try_recv_pdu()
1115 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_tcp_prep_recv_ddgst()  local
1117 	nvmet_tcp_recv_ddgst(queue->rcv_hash, cmd);  in nvmet_tcp_prep_recv_ddgst()
1118 	queue->offset = 0;  in nvmet_tcp_prep_recv_ddgst()
1119 	queue->left = NVME_TCP_DIGEST_LENGTH;  in nvmet_tcp_prep_recv_ddgst()
1120 	queue->rcv_state = NVMET_TCP_RECV_DDGST;  in nvmet_tcp_prep_recv_ddgst()
1123 static int nvmet_tcp_try_recv_data(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_data()  argument
1125 	struct nvmet_tcp_cmd  *cmd = queue->cmd;  in nvmet_tcp_try_recv_data()
1129 		ret = sock_recvmsg(cmd->queue->sock, &cmd->recv_msg,  in nvmet_tcp_try_recv_data()
1139 	if (queue->data_digest) {  in nvmet_tcp_try_recv_data()
1147 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_try_recv_data()
1151 static int nvmet_tcp_try_recv_ddgst(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_ddgst()  argument
1153 	struct nvmet_tcp_cmd *cmd = queue->cmd;  in nvmet_tcp_try_recv_ddgst()
1157 		.iov_base = (void *)&cmd->recv_ddgst + queue->offset,  in nvmet_tcp_try_recv_ddgst()
1158 		.iov_len = queue->left  in nvmet_tcp_try_recv_ddgst()
1161 	ret = kernel_recvmsg(queue->sock, &msg, &iov, 1,  in nvmet_tcp_try_recv_ddgst()
1166 	queue->offset += ret;  in nvmet_tcp_try_recv_ddgst()
1167 	queue->left -= ret;  in nvmet_tcp_try_recv_ddgst()
1168 	if (queue->left)  in nvmet_tcp_try_recv_ddgst()
1171 	if (queue->data_digest && cmd->exp_ddgst != cmd->recv_ddgst) {  in nvmet_tcp_try_recv_ddgst()
1172 		pr_err("queue %d: cmd %d pdu (%d) data digest error: recv %#x expected %#x\n",  in nvmet_tcp_try_recv_ddgst()
1173 			queue->idx, cmd->req.cmd->common.command_id,  in nvmet_tcp_try_recv_ddgst()
1174 			queue->pdu.cmd.hdr.type, le32_to_cpu(cmd->recv_ddgst),  in nvmet_tcp_try_recv_ddgst()
1177 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_try_recv_ddgst()
1187 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_try_recv_ddgst()
1191 static int nvmet_tcp_try_recv_one(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_one()  argument
1195 	if (unlikely(queue->rcv_state == NVMET_TCP_RECV_ERR))  in nvmet_tcp_try_recv_one()
1198 	if (queue->rcv_state == NVMET_TCP_RECV_PDU) {  in nvmet_tcp_try_recv_one()
1199 		result = nvmet_tcp_try_recv_pdu(queue);  in nvmet_tcp_try_recv_one()
1204 	if (queue->rcv_state == NVMET_TCP_RECV_DATA) {  in nvmet_tcp_try_recv_one()
1205 		result = nvmet_tcp_try_recv_data(queue);  in nvmet_tcp_try_recv_one()
1210 	if (queue->rcv_state == NVMET_TCP_RECV_DDGST) {  in nvmet_tcp_try_recv_one()
1211 		result = nvmet_tcp_try_recv_ddgst(queue);  in nvmet_tcp_try_recv_one()
1225 static int nvmet_tcp_try_recv(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_recv()  argument
1231 		ret = nvmet_tcp_try_recv_one(queue);  in nvmet_tcp_try_recv()
1233 			nvmet_tcp_socket_error(queue, ret);  in nvmet_tcp_try_recv()
1244 static void nvmet_tcp_schedule_release_queue(struct nvmet_tcp_queue *queue)  in nvmet_tcp_schedule_release_queue()  argument
1246 	spin_lock(&queue->state_lock);  in nvmet_tcp_schedule_release_queue()
1247 	if (queue->state != NVMET_TCP_Q_DISCONNECTING) {  in nvmet_tcp_schedule_release_queue()
1248 		queue->state = NVMET_TCP_Q_DISCONNECTING;  in nvmet_tcp_schedule_release_queue()
1249 		schedule_work(&queue->release_work);  in nvmet_tcp_schedule_release_queue()
1251 	spin_unlock(&queue->state_lock);  in nvmet_tcp_schedule_release_queue()
1254 static inline void nvmet_tcp_arm_queue_deadline(struct nvmet_tcp_queue *queue)  in nvmet_tcp_arm_queue_deadline()  argument
1256 	queue->poll_end = jiffies + usecs_to_jiffies(idle_poll_period_usecs);  in nvmet_tcp_arm_queue_deadline()
1259 static bool nvmet_tcp_check_queue_deadline(struct nvmet_tcp_queue *queue,  in nvmet_tcp_check_queue_deadline()  argument
1266 		nvmet_tcp_arm_queue_deadline(queue);  in nvmet_tcp_check_queue_deadline()
1268 	return !time_after(jiffies, queue->poll_end);  in nvmet_tcp_check_queue_deadline()
1273 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_io_work()  local
1281 		ret = nvmet_tcp_try_recv(queue, NVMET_TCP_RECV_BUDGET, &ops);  in nvmet_tcp_io_work()
1287 		ret = nvmet_tcp_try_send(queue, NVMET_TCP_SEND_BUDGET, &ops);  in nvmet_tcp_io_work()
1299 	if (nvmet_tcp_check_queue_deadline(queue, ops) || pending)  in nvmet_tcp_io_work()
1300 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_io_work()
1303 static int nvmet_tcp_alloc_cmd(struct nvmet_tcp_queue *queue,  in nvmet_tcp_alloc_cmd()  argument
1306 	u8 hdgst = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_alloc_cmd()
1308 	c->queue = queue;  in nvmet_tcp_alloc_cmd()
1309 	c->req.port = queue->port->nport;  in nvmet_tcp_alloc_cmd()
1311 	c->cmd_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1317 	c->rsp_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1323 	c->data_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1328 	c->r2t_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1335 	list_add_tail(&c->entry, &queue->free_list);  in nvmet_tcp_alloc_cmd()
1355 static int nvmet_tcp_alloc_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_alloc_cmds()  argument
1358 	int i, ret = -EINVAL, nr_cmds = queue->nr_cmds;  in nvmet_tcp_alloc_cmds()
1365 		ret = nvmet_tcp_alloc_cmd(queue, cmds + i);  in nvmet_tcp_alloc_cmds()
1370 	queue->cmds = cmds;  in nvmet_tcp_alloc_cmds()
1381 static void nvmet_tcp_free_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_free_cmds()  argument
1383 	struct nvmet_tcp_cmd *cmds = queue->cmds;  in nvmet_tcp_free_cmds()
1386 	for (i = 0; i < queue->nr_cmds; i++)  in nvmet_tcp_free_cmds()
1389 	nvmet_tcp_free_cmd(&queue->connect);  in nvmet_tcp_free_cmds()
1393 static void nvmet_tcp_restore_socket_callbacks(struct nvmet_tcp_queue *queue)  in nvmet_tcp_restore_socket_callbacks()  argument
1395 	struct socket *sock = queue->sock;  in nvmet_tcp_restore_socket_callbacks()
1398 	sock->sk->sk_data_ready =  queue->data_ready;  in nvmet_tcp_restore_socket_callbacks()
1399 	sock->sk->sk_state_change = queue->state_change;  in nvmet_tcp_restore_socket_callbacks()
1400 	sock->sk->sk_write_space = queue->write_space;  in nvmet_tcp_restore_socket_callbacks()
1413 static void nvmet_tcp_uninit_data_in_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_uninit_data_in_cmds()  argument
1415 	struct nvmet_tcp_cmd *cmd = queue->cmds;  in nvmet_tcp_uninit_data_in_cmds()
1418 	for (i = 0; i < queue->nr_cmds; i++, cmd++) {  in nvmet_tcp_uninit_data_in_cmds()
1423 	if (!queue->nr_cmds && nvmet_tcp_need_data_in(&queue->connect)) {  in nvmet_tcp_uninit_data_in_cmds()
1425 		nvmet_tcp_finish_cmd(&queue->connect);  in nvmet_tcp_uninit_data_in_cmds()
1432 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_release_queue_work()  local
1436 	list_del_init(&queue->queue_list);  in nvmet_tcp_release_queue_work()
1439 	nvmet_tcp_restore_socket_callbacks(queue);  in nvmet_tcp_release_queue_work()
1440 	flush_work(&queue->io_work);  in nvmet_tcp_release_queue_work()
1442 	nvmet_tcp_uninit_data_in_cmds(queue);  in nvmet_tcp_release_queue_work()
1443 	nvmet_sq_destroy(&queue->nvme_sq);  in nvmet_tcp_release_queue_work()
1444 	cancel_work_sync(&queue->io_work);  in nvmet_tcp_release_queue_work()
1445 	sock_release(queue->sock);  in nvmet_tcp_release_queue_work()
1446 	nvmet_tcp_free_cmds(queue);  in nvmet_tcp_release_queue_work()
1447 	if (queue->hdr_digest || queue->data_digest)  in nvmet_tcp_release_queue_work()
1448 		nvmet_tcp_free_crypto(queue);  in nvmet_tcp_release_queue_work()
1449 	ida_simple_remove(&nvmet_tcp_queue_ida, queue->idx);  in nvmet_tcp_release_queue_work()
1451 	page = virt_to_head_page(queue->pf_cache.va);  in nvmet_tcp_release_queue_work()
1452 	__page_frag_cache_drain(page, queue->pf_cache.pagecnt_bias);  in nvmet_tcp_release_queue_work()
1453 	kfree(queue);  in nvmet_tcp_release_queue_work()
1458 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_data_ready()  local
1461 	queue = sk->sk_user_data;  in nvmet_tcp_data_ready()
1462 	if (likely(queue))  in nvmet_tcp_data_ready()
1463 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_data_ready()
1469 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_write_space()  local
1472 	queue = sk->sk_user_data;  in nvmet_tcp_write_space()
1473 	if (unlikely(!queue))  in nvmet_tcp_write_space()
1476 	if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {  in nvmet_tcp_write_space()
1477 		queue->write_space(sk);  in nvmet_tcp_write_space()
1483 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_write_space()
1491 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_state_change()  local
1494 	queue = sk->sk_user_data;  in nvmet_tcp_state_change()
1495 	if (!queue)  in nvmet_tcp_state_change()
1503 		nvmet_tcp_schedule_release_queue(queue);  in nvmet_tcp_state_change()
1506 		pr_warn("queue %d unhandled state %d\n",  in nvmet_tcp_state_change()
1507 			queue->idx, sk->sk_state);  in nvmet_tcp_state_change()
1513 static int nvmet_tcp_set_queue_sock(struct nvmet_tcp_queue *queue)  in nvmet_tcp_set_queue_sock()  argument
1515 	struct socket *sock = queue->sock;  in nvmet_tcp_set_queue_sock()
1520 		(struct sockaddr *)&queue->sockaddr);  in nvmet_tcp_set_queue_sock()
1525 		(struct sockaddr *)&queue->sockaddr_peer);  in nvmet_tcp_set_queue_sock()
1530 	 * Cleanup whatever is sitting in the TCP transmit queue on socket  in nvmet_tcp_set_queue_sock()
1552 		sock->sk->sk_user_data = queue;  in nvmet_tcp_set_queue_sock()
1553 		queue->data_ready = sock->sk->sk_data_ready;  in nvmet_tcp_set_queue_sock()
1555 		queue->state_change = sock->sk->sk_state_change;  in nvmet_tcp_set_queue_sock()
1557 		queue->write_space = sock->sk->sk_write_space;  in nvmet_tcp_set_queue_sock()
1560 			nvmet_tcp_arm_queue_deadline(queue);  in nvmet_tcp_set_queue_sock()
1561 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_set_queue_sock()
1571 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_alloc_queue()  local
1574 	queue = kzalloc(sizeof(*queue), GFP_KERNEL);  in nvmet_tcp_alloc_queue()
1575 	if (!queue)  in nvmet_tcp_alloc_queue()
1578 	INIT_WORK(&queue->release_work, nvmet_tcp_release_queue_work);  in nvmet_tcp_alloc_queue()
1579 	INIT_WORK(&queue->io_work, nvmet_tcp_io_work);  in nvmet_tcp_alloc_queue()
1580 	queue->sock = newsock;  in nvmet_tcp_alloc_queue()
1581 	queue->port = port;  in nvmet_tcp_alloc_queue()
1582 	queue->nr_cmds = 0;  in nvmet_tcp_alloc_queue()
1583 	spin_lock_init(&queue->state_lock);  in nvmet_tcp_alloc_queue()
1584 	queue->state = NVMET_TCP_Q_CONNECTING;  in nvmet_tcp_alloc_queue()
1585 	INIT_LIST_HEAD(&queue->free_list);  in nvmet_tcp_alloc_queue()
1586 	init_llist_head(&queue->resp_list);  in nvmet_tcp_alloc_queue()
1587 	INIT_LIST_HEAD(&queue->resp_send_list);  in nvmet_tcp_alloc_queue()
1589 	queue->idx = ida_simple_get(&nvmet_tcp_queue_ida, 0, 0, GFP_KERNEL);  in nvmet_tcp_alloc_queue()
1590 	if (queue->idx < 0) {  in nvmet_tcp_alloc_queue()
1591 		ret = queue->idx;  in nvmet_tcp_alloc_queue()
1595 	ret = nvmet_tcp_alloc_cmd(queue, &queue->connect);  in nvmet_tcp_alloc_queue()
1599 	ret = nvmet_sq_init(&queue->nvme_sq);  in nvmet_tcp_alloc_queue()
1603 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_alloc_queue()
1606 	list_add_tail(&queue->queue_list, &nvmet_tcp_queue_list);  in nvmet_tcp_alloc_queue()
1609 	ret = nvmet_tcp_set_queue_sock(queue);  in nvmet_tcp_alloc_queue()
1616 	list_del_init(&queue->queue_list);  in nvmet_tcp_alloc_queue()
1618 	nvmet_sq_destroy(&queue->nvme_sq);  in nvmet_tcp_alloc_queue()
1620 	nvmet_tcp_free_cmd(&queue->connect);  in nvmet_tcp_alloc_queue()
1622 	ida_simple_remove(&nvmet_tcp_queue_ida, queue->idx);  in nvmet_tcp_alloc_queue()
1624 	kfree(queue);  in nvmet_tcp_alloc_queue()
1644 			pr_err("failed to allocate queue\n");  in nvmet_tcp_accept_work()
1759 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_delete_ctrl()  local
1762 	list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)  in nvmet_tcp_delete_ctrl()
1763 		if (queue->nvme_sq.ctrl == ctrl)  in nvmet_tcp_delete_ctrl()
1764 			kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_delete_ctrl()
1770 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_install_queue()  local
1778 	queue->nr_cmds = sq->size * 2;  in nvmet_tcp_install_queue()
1779 	if (nvmet_tcp_alloc_cmds(queue))  in nvmet_tcp_install_queue()
1792 		struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_tcp_disc_port_addr()  local
1794 		sprintf(traddr, "%pISc", (struct sockaddr *)&queue->sockaddr);  in nvmet_tcp_disc_port_addr()
1832 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_exit()  local
1838 	list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)  in nvmet_tcp_exit()
1839 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_exit()