59  * queue before determining it to be idle.  This optional module behavior
93 	struct nvmet_tcp_queue		*queue;  member
196 static inline u16 nvmet_tcp_cmd_tag(struct nvmet_tcp_queue *queue,  in nvmet_tcp_cmd_tag()  argument
199 	if (unlikely(!queue->nr_cmds)) {  in nvmet_tcp_cmd_tag()
204 	return cmd - queue->cmds;  in nvmet_tcp_cmd_tag()
232 nvmet_tcp_get_cmd(struct nvmet_tcp_queue *queue)  in nvmet_tcp_get_cmd()  argument
236 	cmd = list_first_entry_or_null(&queue->free_list,  in nvmet_tcp_get_cmd()
252 	if (unlikely(cmd == &cmd->queue->connect))  in nvmet_tcp_put_cmd()
255 	list_add_tail(&cmd->entry, &cmd->queue->free_list);  in nvmet_tcp_put_cmd()
258 static inline int queue_cpu(struct nvmet_tcp_queue *queue)  in queue_cpu()  argument
260 	return queue->sock->sk->sk_incoming_cpu;  in queue_cpu()
263 static inline u8 nvmet_tcp_hdgst_len(struct nvmet_tcp_queue *queue)  in nvmet_tcp_hdgst_len()  argument
265 	return queue->hdr_digest ? NVME_TCP_DIGEST_LENGTH : 0;  in nvmet_tcp_hdgst_len()
268 static inline u8 nvmet_tcp_ddgst_len(struct nvmet_tcp_queue *queue)  in nvmet_tcp_ddgst_len()  argument
270 	return queue->data_digest ? NVME_TCP_DIGEST_LENGTH : 0;  in nvmet_tcp_ddgst_len()
283 static int nvmet_tcp_verify_hdgst(struct nvmet_tcp_queue *queue,  in nvmet_tcp_verify_hdgst()  argument
291 		pr_err("queue %d: header digest enabled but no header digest\n",  in nvmet_tcp_verify_hdgst()
292 			queue->idx);  in nvmet_tcp_verify_hdgst()
297 	nvmet_tcp_hdgst(queue->rcv_hash, pdu, len);  in nvmet_tcp_verify_hdgst()
300 		pr_err("queue %d: header digest error: recv %#x expected %#x\n",  in nvmet_tcp_verify_hdgst()
301 			queue->idx, le32_to_cpu(recv_digest),  in nvmet_tcp_verify_hdgst()
309 static int nvmet_tcp_check_ddgst(struct nvmet_tcp_queue *queue, void *pdu)  in nvmet_tcp_check_ddgst()  argument
312 	u8 digest_len = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_check_ddgst()
319 		pr_err("queue %d: data digest flag is cleared\n", queue->idx);  in nvmet_tcp_check_ddgst()
364 static void nvmet_tcp_fatal_error(struct nvmet_tcp_queue *queue)  in nvmet_tcp_fatal_error()  argument
366 	queue->rcv_state = NVMET_TCP_RECV_ERR;  in nvmet_tcp_fatal_error()
367 	if (queue->nvme_sq.ctrl)  in nvmet_tcp_fatal_error()
368 		nvmet_ctrl_fatal_error(queue->nvme_sq.ctrl);  in nvmet_tcp_fatal_error()
370 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_fatal_error()
373 static void nvmet_tcp_socket_error(struct nvmet_tcp_queue *queue, int status)  in nvmet_tcp_socket_error()  argument
375 	queue->rcv_state = NVMET_TCP_RECV_ERR;  in nvmet_tcp_socket_error()
377 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_socket_error()
379 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_socket_error()
430 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_c2h_data_pdu()  local
431 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_c2h_data_pdu()
432 	u8 ddgst = nvmet_tcp_ddgst_len(cmd->queue);  in nvmet_setup_c2h_data_pdu()
438 	pdu->hdr.flags = NVME_TCP_F_DATA_LAST | (queue->nvme_sq.sqhd_disabled ?  in nvmet_setup_c2h_data_pdu()
449 	if (queue->data_digest) {  in nvmet_setup_c2h_data_pdu()
451 		nvmet_tcp_calc_ddgst(queue->snd_hash, cmd);  in nvmet_setup_c2h_data_pdu()
454 	if (cmd->queue->hdr_digest) {  in nvmet_setup_c2h_data_pdu()
456 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_c2h_data_pdu()
463 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_r2t_pdu()  local
464 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_r2t_pdu()
476 	pdu->ttag = nvmet_tcp_cmd_tag(cmd->queue, cmd);  in nvmet_setup_r2t_pdu()
479 	if (cmd->queue->hdr_digest) {  in nvmet_setup_r2t_pdu()
481 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_r2t_pdu()
488 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_setup_response_pdu()  local
489 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_setup_response_pdu()
499 	if (cmd->queue->hdr_digest) {  in nvmet_setup_response_pdu()
501 		nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));  in nvmet_setup_response_pdu()
505 static void nvmet_tcp_process_resp_list(struct nvmet_tcp_queue *queue)  in nvmet_tcp_process_resp_list()  argument
510 	for (node = llist_del_all(&queue->resp_list); node; node = node->next) {  in nvmet_tcp_process_resp_list()
512 		list_add(&cmd->entry, &queue->resp_send_list);  in nvmet_tcp_process_resp_list()
513 		queue->send_list_len++;  in nvmet_tcp_process_resp_list()
517 static struct nvmet_tcp_cmd *nvmet_tcp_fetch_cmd(struct nvmet_tcp_queue *queue)  in nvmet_tcp_fetch_cmd()  argument
519 	queue->snd_cmd = list_first_entry_or_null(&queue->resp_send_list,  in nvmet_tcp_fetch_cmd()
521 	if (!queue->snd_cmd) {  in nvmet_tcp_fetch_cmd()
522 		nvmet_tcp_process_resp_list(queue);  in nvmet_tcp_fetch_cmd()
523 		queue->snd_cmd =  in nvmet_tcp_fetch_cmd()
524 			list_first_entry_or_null(&queue->resp_send_list,  in nvmet_tcp_fetch_cmd()
526 		if (unlikely(!queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
530 	list_del_init(&queue->snd_cmd->entry);  in nvmet_tcp_fetch_cmd()
531 	queue->send_list_len--;  in nvmet_tcp_fetch_cmd()
533 	if (nvmet_tcp_need_data_out(queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
534 		nvmet_setup_c2h_data_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
535 	else if (nvmet_tcp_need_data_in(queue->snd_cmd))  in nvmet_tcp_fetch_cmd()
536 		nvmet_setup_r2t_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
538 		nvmet_setup_response_pdu(queue->snd_cmd);  in nvmet_tcp_fetch_cmd()
540 	return queue->snd_cmd;  in nvmet_tcp_fetch_cmd()
547 	struct nvmet_tcp_queue	*queue = cmd->queue;  in nvmet_tcp_queue_response()  local
551 	if (unlikely(cmd == queue->cmd)) {  in nvmet_tcp_queue_response()
560 		if (queue->rcv_state == NVMET_TCP_RECV_PDU &&  in nvmet_tcp_queue_response()
566 	llist_add(&cmd->lentry, &queue->resp_list);  in nvmet_tcp_queue_response()
567 	queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &cmd->queue->io_work);  in nvmet_tcp_queue_response()
584 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_data_pdu()
590 	ret = sock_sendmsg(cmd->queue->sock, &msg);  in nvmet_try_send_data_pdu()
607 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_try_send_data()  local
618 		if ((!last_in_batch && cmd->queue->send_list_len) ||  in nvmet_try_send_data()
620 		    queue->data_digest || !queue->nvme_sq.sqhd_disabled)  in nvmet_try_send_data()
625 		ret = sock_sendmsg(cmd->queue->sock, &msg);  in nvmet_try_send_data()
639 	if (queue->data_digest) {  in nvmet_try_send_data()
643 		if (queue->nvme_sq.sqhd_disabled) {  in nvmet_try_send_data()
644 			cmd->queue->snd_cmd = NULL;  in nvmet_try_send_data()
651 	if (queue->nvme_sq.sqhd_disabled)  in nvmet_try_send_data()
663 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_response()
667 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_response()
674 	ret = sock_sendmsg(cmd->queue->sock, &msg);  in nvmet_try_send_response()
684 	cmd->queue->snd_cmd = NULL;  in nvmet_try_send_response()
693 	u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);  in nvmet_try_send_r2t()
697 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_r2t()
704 	ret = sock_sendmsg(cmd->queue->sock, &msg);  in nvmet_try_send_r2t()
713 	cmd->queue->snd_cmd = NULL;  in nvmet_try_send_r2t()
719 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_try_send_ddgst()  local
728 	if (!last_in_batch && cmd->queue->send_list_len)  in nvmet_try_send_ddgst()
733 	ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);  in nvmet_try_send_ddgst()
743 	if (queue->nvme_sq.sqhd_disabled) {  in nvmet_try_send_ddgst()
744 		cmd->queue->snd_cmd = NULL;  in nvmet_try_send_ddgst()
752 static int nvmet_tcp_try_send_one(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_send_one()  argument
755 	struct nvmet_tcp_cmd *cmd = queue->snd_cmd;  in nvmet_tcp_try_send_one()
758 	if (!cmd || queue->state == NVMET_TCP_Q_DISCONNECTING) {  in nvmet_tcp_try_send_one()
759 		cmd = nvmet_tcp_fetch_cmd(queue);  in nvmet_tcp_try_send_one()
801 static int nvmet_tcp_try_send(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_send()  argument
807 		ret = nvmet_tcp_try_send_one(queue, i == budget - 1);  in nvmet_tcp_try_send()
809 			nvmet_tcp_socket_error(queue, ret);  in nvmet_tcp_try_send()
820 static void nvmet_prepare_receive_pdu(struct nvmet_tcp_queue *queue)  in nvmet_prepare_receive_pdu()  argument
822 	queue->offset = 0;  in nvmet_prepare_receive_pdu()
823 	queue->left = sizeof(struct nvme_tcp_hdr);  in nvmet_prepare_receive_pdu()
824 	queue->cmd = NULL;  in nvmet_prepare_receive_pdu()
825 	queue->rcv_state = NVMET_TCP_RECV_PDU;  in nvmet_prepare_receive_pdu()
828 static void nvmet_tcp_free_crypto(struct nvmet_tcp_queue *queue)  in nvmet_tcp_free_crypto()  argument
830 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(queue->rcv_hash);  in nvmet_tcp_free_crypto()
832 	ahash_request_free(queue->rcv_hash);  in nvmet_tcp_free_crypto()
833 	ahash_request_free(queue->snd_hash);  in nvmet_tcp_free_crypto()
837 static int nvmet_tcp_alloc_crypto(struct nvmet_tcp_queue *queue)  in nvmet_tcp_alloc_crypto()  argument
845 	queue->snd_hash = ahash_request_alloc(tfm, GFP_KERNEL);  in nvmet_tcp_alloc_crypto()
846 	if (!queue->snd_hash)  in nvmet_tcp_alloc_crypto()
848 	ahash_request_set_callback(queue->snd_hash, 0, NULL, NULL);  in nvmet_tcp_alloc_crypto()
850 	queue->rcv_hash = ahash_request_alloc(tfm, GFP_KERNEL);  in nvmet_tcp_alloc_crypto()
851 	if (!queue->rcv_hash)  in nvmet_tcp_alloc_crypto()
853 	ahash_request_set_callback(queue->rcv_hash, 0, NULL, NULL);  in nvmet_tcp_alloc_crypto()
857 	ahash_request_free(queue->snd_hash);  in nvmet_tcp_alloc_crypto()
864 static int nvmet_tcp_handle_icreq(struct nvmet_tcp_queue *queue)  in nvmet_tcp_handle_icreq()  argument
866 	struct nvme_tcp_icreq_pdu *icreq = &queue->pdu.icreq;  in nvmet_tcp_handle_icreq()
867 	struct nvme_tcp_icresp_pdu *icresp = &queue->pdu.icresp;  in nvmet_tcp_handle_icreq()
875 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_handle_icreq()
879 		pr_err("queue %d: bad pfv %d\n", queue->idx, icreq->pfv);  in nvmet_tcp_handle_icreq()
884 		pr_err("queue %d: unsupported hpda %d\n", queue->idx,  in nvmet_tcp_handle_icreq()
889 	queue->hdr_digest = !!(icreq->digest & NVME_TCP_HDR_DIGEST_ENABLE);  in nvmet_tcp_handle_icreq()
890 	queue->data_digest = !!(icreq->digest & NVME_TCP_DATA_DIGEST_ENABLE);  in nvmet_tcp_handle_icreq()
891 	if (queue->hdr_digest || queue->data_digest) {  in nvmet_tcp_handle_icreq()
892 		ret = nvmet_tcp_alloc_crypto(queue);  in nvmet_tcp_handle_icreq()
905 	if (queue->hdr_digest)  in nvmet_tcp_handle_icreq()
907 	if (queue->data_digest)  in nvmet_tcp_handle_icreq()
912 	ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);  in nvmet_tcp_handle_icreq()
914 		return ret; /* queue removal will cleanup */  in nvmet_tcp_handle_icreq()
916 	queue->state = NVMET_TCP_Q_LIVE;  in nvmet_tcp_handle_icreq()
917 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_handle_icreq()
921 static void nvmet_tcp_handle_req_failure(struct nvmet_tcp_queue *queue,  in nvmet_tcp_handle_req_failure()  argument
936 		nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_handle_req_failure()
942 		pr_err("queue %d: failed to map data\n", queue->idx);  in nvmet_tcp_handle_req_failure()
943 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_handle_req_failure()
947 	queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_handle_req_failure()
952 static int nvmet_tcp_handle_h2c_data_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_handle_h2c_data_pdu()  argument
954 	struct nvme_tcp_data_pdu *data = &queue->pdu.data;  in nvmet_tcp_handle_h2c_data_pdu()
957 	if (likely(queue->nr_cmds)) {  in nvmet_tcp_handle_h2c_data_pdu()
958 		if (unlikely(data->ttag >= queue->nr_cmds)) {  in nvmet_tcp_handle_h2c_data_pdu()
959 			pr_err("queue %d: received out of bound ttag %u, nr_cmds %u\n",  in nvmet_tcp_handle_h2c_data_pdu()
960 				queue->idx, data->ttag, queue->nr_cmds);  in nvmet_tcp_handle_h2c_data_pdu()
961 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_handle_h2c_data_pdu()
964 		cmd = &queue->cmds[data->ttag];  in nvmet_tcp_handle_h2c_data_pdu()
966 		cmd = &queue->connect;  in nvmet_tcp_handle_h2c_data_pdu()
982 	queue->cmd = cmd;  in nvmet_tcp_handle_h2c_data_pdu()
983 	queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_handle_h2c_data_pdu()
988 static int nvmet_tcp_done_recv_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_done_recv_pdu()  argument
990 	struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;  in nvmet_tcp_done_recv_pdu()
991 	struct nvme_command *nvme_cmd = &queue->pdu.cmd.cmd;  in nvmet_tcp_done_recv_pdu()
995 	if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {  in nvmet_tcp_done_recv_pdu()
999 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
1002 		return nvmet_tcp_handle_icreq(queue);  in nvmet_tcp_done_recv_pdu()
1006 		pr_err("queue %d: received icreq pdu in state %d\n",  in nvmet_tcp_done_recv_pdu()
1007 			queue->idx, queue->state);  in nvmet_tcp_done_recv_pdu()
1008 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
1013 		ret = nvmet_tcp_handle_h2c_data_pdu(queue);  in nvmet_tcp_done_recv_pdu()
1019 	queue->cmd = nvmet_tcp_get_cmd(queue);  in nvmet_tcp_done_recv_pdu()
1020 	if (unlikely(!queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
1022 		pr_err("queue %d: out of commands (%d) send_list_len: %d, opcode: %d",  in nvmet_tcp_done_recv_pdu()
1023 			queue->idx, queue->nr_cmds, queue->send_list_len,  in nvmet_tcp_done_recv_pdu()
1025 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
1029 	req = &queue->cmd->req;  in nvmet_tcp_done_recv_pdu()
1032 	if (unlikely(!nvmet_req_init(req, &queue->nvme_cq,  in nvmet_tcp_done_recv_pdu()
1033 			&queue->nvme_sq, &nvmet_tcp_ops))) {  in nvmet_tcp_done_recv_pdu()
1039 		nvmet_tcp_handle_req_failure(queue, queue->cmd, req);  in nvmet_tcp_done_recv_pdu()
1043 	ret = nvmet_tcp_map_data(queue->cmd);  in nvmet_tcp_done_recv_pdu()
1045 		pr_err("queue %d: failed to map data\n", queue->idx);  in nvmet_tcp_done_recv_pdu()
1046 		if (nvmet_tcp_has_inline_data(queue->cmd))  in nvmet_tcp_done_recv_pdu()
1047 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_done_recv_pdu()
1054 	if (nvmet_tcp_need_data_in(queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
1055 		if (nvmet_tcp_has_inline_data(queue->cmd)) {  in nvmet_tcp_done_recv_pdu()
1056 			queue->rcv_state = NVMET_TCP_RECV_DATA;  in nvmet_tcp_done_recv_pdu()
1057 			nvmet_tcp_build_pdu_iovec(queue->cmd);  in nvmet_tcp_done_recv_pdu()
1061 		nvmet_tcp_queue_response(&queue->cmd->req);  in nvmet_tcp_done_recv_pdu()
1065 	queue->cmd->req.execute(&queue->cmd->req);  in nvmet_tcp_done_recv_pdu()
1067 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_done_recv_pdu()
1099 static int nvmet_tcp_try_recv_pdu(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_pdu()  argument
1101 	struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;  in nvmet_tcp_try_recv_pdu()
1107 	iov.iov_base = (void *)&queue->pdu + queue->offset;  in nvmet_tcp_try_recv_pdu()
1108 	iov.iov_len = queue->left;  in nvmet_tcp_try_recv_pdu()
1109 	len = kernel_recvmsg(queue->sock, &msg, &iov, 1,  in nvmet_tcp_try_recv_pdu()
1114 	queue->offset += len;  in nvmet_tcp_try_recv_pdu()
1115 	queue->left -= len;  in nvmet_tcp_try_recv_pdu()
1116 	if (queue->left)  in nvmet_tcp_try_recv_pdu()
1119 	if (queue->offset == sizeof(struct nvme_tcp_hdr)) {  in nvmet_tcp_try_recv_pdu()
1120 		u8 hdgst = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_try_recv_pdu()
1124 			nvmet_tcp_fatal_error(queue);  in nvmet_tcp_try_recv_pdu()
1133 		queue->left = hdr->hlen - queue->offset + hdgst;  in nvmet_tcp_try_recv_pdu()
1137 	if (queue->hdr_digest &&  in nvmet_tcp_try_recv_pdu()
1138 	    nvmet_tcp_verify_hdgst(queue, &queue->pdu, hdr->hlen)) {  in nvmet_tcp_try_recv_pdu()
1139 		nvmet_tcp_fatal_error(queue); /* fatal */  in nvmet_tcp_try_recv_pdu()
1143 	if (queue->data_digest &&  in nvmet_tcp_try_recv_pdu()
1144 	    nvmet_tcp_check_ddgst(queue, &queue->pdu)) {  in nvmet_tcp_try_recv_pdu()
1145 		nvmet_tcp_fatal_error(queue); /* fatal */  in nvmet_tcp_try_recv_pdu()
1149 	return nvmet_tcp_done_recv_pdu(queue);  in nvmet_tcp_try_recv_pdu()
1154 	struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_tcp_prep_recv_ddgst()  local
1156 	nvmet_tcp_calc_ddgst(queue->rcv_hash, cmd);  in nvmet_tcp_prep_recv_ddgst()
1157 	queue->offset = 0;  in nvmet_tcp_prep_recv_ddgst()
1158 	queue->left = NVME_TCP_DIGEST_LENGTH;  in nvmet_tcp_prep_recv_ddgst()
1159 	queue->rcv_state = NVMET_TCP_RECV_DDGST;  in nvmet_tcp_prep_recv_ddgst()
1162 static int nvmet_tcp_try_recv_data(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_data()  argument
1164 	struct nvmet_tcp_cmd  *cmd = queue->cmd;  in nvmet_tcp_try_recv_data()
1168 		ret = sock_recvmsg(cmd->queue->sock, &cmd->recv_msg,  in nvmet_tcp_try_recv_data()
1177 	if (queue->data_digest) {  in nvmet_tcp_try_recv_data()
1185 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_try_recv_data()
1189 static int nvmet_tcp_try_recv_ddgst(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_ddgst()  argument
1191 	struct nvmet_tcp_cmd *cmd = queue->cmd;  in nvmet_tcp_try_recv_ddgst()
1195 		.iov_base = (void *)&cmd->recv_ddgst + queue->offset,  in nvmet_tcp_try_recv_ddgst()
1196 		.iov_len = queue->left  in nvmet_tcp_try_recv_ddgst()
1199 	ret = kernel_recvmsg(queue->sock, &msg, &iov, 1,  in nvmet_tcp_try_recv_ddgst()
1204 	queue->offset += ret;  in nvmet_tcp_try_recv_ddgst()
1205 	queue->left -= ret;  in nvmet_tcp_try_recv_ddgst()
1206 	if (queue->left)  in nvmet_tcp_try_recv_ddgst()
1209 	if (queue->data_digest && cmd->exp_ddgst != cmd->recv_ddgst) {  in nvmet_tcp_try_recv_ddgst()
1210 		pr_err("queue %d: cmd %d pdu (%d) data digest error: recv %#x expected %#x\n",  in nvmet_tcp_try_recv_ddgst()
1211 			queue->idx, cmd->req.cmd->common.command_id,  in nvmet_tcp_try_recv_ddgst()
1212 			queue->pdu.cmd.hdr.type, le32_to_cpu(cmd->recv_ddgst),  in nvmet_tcp_try_recv_ddgst()
1216 		nvmet_tcp_fatal_error(queue);  in nvmet_tcp_try_recv_ddgst()
1226 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_try_recv_ddgst()
1230 static int nvmet_tcp_try_recv_one(struct nvmet_tcp_queue *queue)  in nvmet_tcp_try_recv_one()  argument
1234 	if (unlikely(queue->rcv_state == NVMET_TCP_RECV_ERR))  in nvmet_tcp_try_recv_one()
1237 	if (queue->rcv_state == NVMET_TCP_RECV_PDU) {  in nvmet_tcp_try_recv_one()
1238 		result = nvmet_tcp_try_recv_pdu(queue);  in nvmet_tcp_try_recv_one()
1243 	if (queue->rcv_state == NVMET_TCP_RECV_DATA) {  in nvmet_tcp_try_recv_one()
1244 		result = nvmet_tcp_try_recv_data(queue);  in nvmet_tcp_try_recv_one()
1249 	if (queue->rcv_state == NVMET_TCP_RECV_DDGST) {  in nvmet_tcp_try_recv_one()
1250 		result = nvmet_tcp_try_recv_ddgst(queue);  in nvmet_tcp_try_recv_one()
1264 static int nvmet_tcp_try_recv(struct nvmet_tcp_queue *queue,  in nvmet_tcp_try_recv()  argument
1270 		ret = nvmet_tcp_try_recv_one(queue);  in nvmet_tcp_try_recv()
1272 			nvmet_tcp_socket_error(queue, ret);  in nvmet_tcp_try_recv()
1283 static void nvmet_tcp_schedule_release_queue(struct nvmet_tcp_queue *queue)  in nvmet_tcp_schedule_release_queue()  argument
1285 	spin_lock(&queue->state_lock);  in nvmet_tcp_schedule_release_queue()
1286 	if (queue->state != NVMET_TCP_Q_DISCONNECTING) {  in nvmet_tcp_schedule_release_queue()
1287 		queue->state = NVMET_TCP_Q_DISCONNECTING;  in nvmet_tcp_schedule_release_queue()
1288 		queue_work(nvmet_wq, &queue->release_work);  in nvmet_tcp_schedule_release_queue()
1290 	spin_unlock(&queue->state_lock);  in nvmet_tcp_schedule_release_queue()
1293 static inline void nvmet_tcp_arm_queue_deadline(struct nvmet_tcp_queue *queue)  in nvmet_tcp_arm_queue_deadline()  argument
1295 	queue->poll_end = jiffies + usecs_to_jiffies(idle_poll_period_usecs);  in nvmet_tcp_arm_queue_deadline()
1298 static bool nvmet_tcp_check_queue_deadline(struct nvmet_tcp_queue *queue,  in nvmet_tcp_check_queue_deadline()  argument
1305 		nvmet_tcp_arm_queue_deadline(queue);  in nvmet_tcp_check_queue_deadline()
1307 	return !time_after(jiffies, queue->poll_end);  in nvmet_tcp_check_queue_deadline()
1312 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_io_work()  local
1320 		ret = nvmet_tcp_try_recv(queue, NVMET_TCP_RECV_BUDGET, &ops);  in nvmet_tcp_io_work()
1326 		ret = nvmet_tcp_try_send(queue, NVMET_TCP_SEND_BUDGET, &ops);  in nvmet_tcp_io_work()
1338 	if (nvmet_tcp_check_queue_deadline(queue, ops) || pending)  in nvmet_tcp_io_work()
1339 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_io_work()
1342 static int nvmet_tcp_alloc_cmd(struct nvmet_tcp_queue *queue,  in nvmet_tcp_alloc_cmd()  argument
1345 	u8 hdgst = nvmet_tcp_hdgst_len(queue);  in nvmet_tcp_alloc_cmd()
1347 	c->queue = queue;  in nvmet_tcp_alloc_cmd()
1348 	c->req.port = queue->port->nport;  in nvmet_tcp_alloc_cmd()
1350 	c->cmd_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1356 	c->rsp_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1362 	c->data_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1367 	c->r2t_pdu = page_frag_alloc(&queue->pf_cache,  in nvmet_tcp_alloc_cmd()
1374 	list_add_tail(&c->entry, &queue->free_list);  in nvmet_tcp_alloc_cmd()
1394 static int nvmet_tcp_alloc_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_alloc_cmds()  argument
1397 	int i, ret = -EINVAL, nr_cmds = queue->nr_cmds;  in nvmet_tcp_alloc_cmds()
1404 		ret = nvmet_tcp_alloc_cmd(queue, cmds + i);  in nvmet_tcp_alloc_cmds()
1409 	queue->cmds = cmds;  in nvmet_tcp_alloc_cmds()
1420 static void nvmet_tcp_free_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_free_cmds()  argument
1422 	struct nvmet_tcp_cmd *cmds = queue->cmds;  in nvmet_tcp_free_cmds()
1425 	for (i = 0; i < queue->nr_cmds; i++)  in nvmet_tcp_free_cmds()
1428 	nvmet_tcp_free_cmd(&queue->connect);  in nvmet_tcp_free_cmds()
1432 static void nvmet_tcp_restore_socket_callbacks(struct nvmet_tcp_queue *queue)  in nvmet_tcp_restore_socket_callbacks()  argument
1434 	struct socket *sock = queue->sock;  in nvmet_tcp_restore_socket_callbacks()
1437 	sock->sk->sk_data_ready =  queue->data_ready;  in nvmet_tcp_restore_socket_callbacks()
1438 	sock->sk->sk_state_change = queue->state_change;  in nvmet_tcp_restore_socket_callbacks()
1439 	sock->sk->sk_write_space = queue->write_space;  in nvmet_tcp_restore_socket_callbacks()
1444 static void nvmet_tcp_uninit_data_in_cmds(struct nvmet_tcp_queue *queue)  in nvmet_tcp_uninit_data_in_cmds()  argument
1446 	struct nvmet_tcp_cmd *cmd = queue->cmds;  in nvmet_tcp_uninit_data_in_cmds()
1449 	for (i = 0; i < queue->nr_cmds; i++, cmd++) {  in nvmet_tcp_uninit_data_in_cmds()
1454 	if (!queue->nr_cmds && nvmet_tcp_need_data_in(&queue->connect)) {  in nvmet_tcp_uninit_data_in_cmds()
1456 		nvmet_req_uninit(&queue->connect.req);  in nvmet_tcp_uninit_data_in_cmds()
1460 static void nvmet_tcp_free_cmd_data_in_buffers(struct nvmet_tcp_queue *queue)  in nvmet_tcp_free_cmd_data_in_buffers()  argument
1462 	struct nvmet_tcp_cmd *cmd = queue->cmds;  in nvmet_tcp_free_cmd_data_in_buffers()
1465 	for (i = 0; i < queue->nr_cmds; i++, cmd++) {  in nvmet_tcp_free_cmd_data_in_buffers()
1470 	if (!queue->nr_cmds && nvmet_tcp_need_data_in(&queue->connect))  in nvmet_tcp_free_cmd_data_in_buffers()
1471 		nvmet_tcp_free_cmd_buffers(&queue->connect);  in nvmet_tcp_free_cmd_data_in_buffers()
1477 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_release_queue_work()  local
1481 	list_del_init(&queue->queue_list);  in nvmet_tcp_release_queue_work()
1484 	nvmet_tcp_restore_socket_callbacks(queue);  in nvmet_tcp_release_queue_work()
1485 	cancel_work_sync(&queue->io_work);  in nvmet_tcp_release_queue_work()
1487 	queue->rcv_state = NVMET_TCP_RECV_ERR;  in nvmet_tcp_release_queue_work()
1489 	nvmet_tcp_uninit_data_in_cmds(queue);  in nvmet_tcp_release_queue_work()
1490 	nvmet_sq_destroy(&queue->nvme_sq);  in nvmet_tcp_release_queue_work()
1491 	cancel_work_sync(&queue->io_work);  in nvmet_tcp_release_queue_work()
1492 	nvmet_tcp_free_cmd_data_in_buffers(queue);  in nvmet_tcp_release_queue_work()
1493 	sock_release(queue->sock);  in nvmet_tcp_release_queue_work()
1494 	nvmet_tcp_free_cmds(queue);  in nvmet_tcp_release_queue_work()
1495 	if (queue->hdr_digest || queue->data_digest)  in nvmet_tcp_release_queue_work()
1496 		nvmet_tcp_free_crypto(queue);  in nvmet_tcp_release_queue_work()
1497 	ida_free(&nvmet_tcp_queue_ida, queue->idx);  in nvmet_tcp_release_queue_work()
1499 	page = virt_to_head_page(queue->pf_cache.va);  in nvmet_tcp_release_queue_work()
1500 	__page_frag_cache_drain(page, queue->pf_cache.pagecnt_bias);  in nvmet_tcp_release_queue_work()
1501 	kfree(queue);  in nvmet_tcp_release_queue_work()
1506 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_data_ready()  local
1511 	queue = sk->sk_user_data;  in nvmet_tcp_data_ready()
1512 	if (likely(queue))  in nvmet_tcp_data_ready()
1513 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_data_ready()
1519 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_write_space()  local
1522 	queue = sk->sk_user_data;  in nvmet_tcp_write_space()
1523 	if (unlikely(!queue))  in nvmet_tcp_write_space()
1526 	if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {  in nvmet_tcp_write_space()
1527 		queue->write_space(sk);  in nvmet_tcp_write_space()
1533 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_write_space()
1541 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_state_change()  local
1544 	queue = sk->sk_user_data;  in nvmet_tcp_state_change()
1545 	if (!queue)  in nvmet_tcp_state_change()
1556 		nvmet_tcp_schedule_release_queue(queue);  in nvmet_tcp_state_change()
1559 		pr_warn("queue %d unhandled state %d\n",  in nvmet_tcp_state_change()
1560 			queue->idx, sk->sk_state);  in nvmet_tcp_state_change()
1566 static int nvmet_tcp_set_queue_sock(struct nvmet_tcp_queue *queue)  in nvmet_tcp_set_queue_sock()  argument
1568 	struct socket *sock = queue->sock;  in nvmet_tcp_set_queue_sock()
1573 		(struct sockaddr *)&queue->sockaddr);  in nvmet_tcp_set_queue_sock()
1578 		(struct sockaddr *)&queue->sockaddr_peer);  in nvmet_tcp_set_queue_sock()
1583 	 * Cleanup whatever is sitting in the TCP transmit queue on socket  in nvmet_tcp_set_queue_sock()
1605 		sock->sk->sk_user_data = queue;  in nvmet_tcp_set_queue_sock()
1606 		queue->data_ready = sock->sk->sk_data_ready;  in nvmet_tcp_set_queue_sock()
1608 		queue->state_change = sock->sk->sk_state_change;  in nvmet_tcp_set_queue_sock()
1610 		queue->write_space = sock->sk->sk_write_space;  in nvmet_tcp_set_queue_sock()
1613 			nvmet_tcp_arm_queue_deadline(queue);  in nvmet_tcp_set_queue_sock()
1614 		queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);  in nvmet_tcp_set_queue_sock()
1624 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_alloc_queue()  local
1627 	queue = kzalloc(sizeof(*queue), GFP_KERNEL);  in nvmet_tcp_alloc_queue()
1628 	if (!queue)  in nvmet_tcp_alloc_queue()
1631 	INIT_WORK(&queue->release_work, nvmet_tcp_release_queue_work);  in nvmet_tcp_alloc_queue()
1632 	INIT_WORK(&queue->io_work, nvmet_tcp_io_work);  in nvmet_tcp_alloc_queue()
1633 	queue->sock = newsock;  in nvmet_tcp_alloc_queue()
1634 	queue->port = port;  in nvmet_tcp_alloc_queue()
1635 	queue->nr_cmds = 0;  in nvmet_tcp_alloc_queue()
1636 	spin_lock_init(&queue->state_lock);  in nvmet_tcp_alloc_queue()
1637 	queue->state = NVMET_TCP_Q_CONNECTING;  in nvmet_tcp_alloc_queue()
1638 	INIT_LIST_HEAD(&queue->free_list);  in nvmet_tcp_alloc_queue()
1639 	init_llist_head(&queue->resp_list);  in nvmet_tcp_alloc_queue()
1640 	INIT_LIST_HEAD(&queue->resp_send_list);  in nvmet_tcp_alloc_queue()
1642 	queue->idx = ida_alloc(&nvmet_tcp_queue_ida, GFP_KERNEL);  in nvmet_tcp_alloc_queue()
1643 	if (queue->idx < 0) {  in nvmet_tcp_alloc_queue()
1644 		ret = queue->idx;  in nvmet_tcp_alloc_queue()
1648 	ret = nvmet_tcp_alloc_cmd(queue, &queue->connect);  in nvmet_tcp_alloc_queue()
1652 	ret = nvmet_sq_init(&queue->nvme_sq);  in nvmet_tcp_alloc_queue()
1656 	nvmet_prepare_receive_pdu(queue);  in nvmet_tcp_alloc_queue()
1659 	list_add_tail(&queue->queue_list, &nvmet_tcp_queue_list);  in nvmet_tcp_alloc_queue()
1662 	ret = nvmet_tcp_set_queue_sock(queue);  in nvmet_tcp_alloc_queue()
1669 	list_del_init(&queue->queue_list);  in nvmet_tcp_alloc_queue()
1671 	nvmet_sq_destroy(&queue->nvme_sq);  in nvmet_tcp_alloc_queue()
1673 	nvmet_tcp_free_cmd(&queue->connect);  in nvmet_tcp_alloc_queue()
1675 	ida_free(&nvmet_tcp_queue_ida, queue->idx);  in nvmet_tcp_alloc_queue()
1677 	kfree(queue);  in nvmet_tcp_alloc_queue()
1697 			pr_err("failed to allocate queue\n");  in nvmet_tcp_accept_work()
1800 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_destroy_port_queues()  local
1803 	list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)  in nvmet_tcp_destroy_port_queues()
1804 		if (queue->port == port)  in nvmet_tcp_destroy_port_queues()
1805 			kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_destroy_port_queues()
1830 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_delete_ctrl()  local
1833 	list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)  in nvmet_tcp_delete_ctrl()
1834 		if (queue->nvme_sq.ctrl == ctrl)  in nvmet_tcp_delete_ctrl()
1835 			kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_delete_ctrl()
1841 	struct nvmet_tcp_queue *queue =  in nvmet_tcp_install_queue()  local
1849 	queue->nr_cmds = sq->size * 2;  in nvmet_tcp_install_queue()
1850 	if (nvmet_tcp_alloc_cmds(queue))  in nvmet_tcp_install_queue()
1863 		struct nvmet_tcp_queue *queue = cmd->queue;  in nvmet_tcp_disc_port_addr()  local
1865 		sprintf(traddr, "%pISc", (struct sockaddr *)&queue->sockaddr);  in nvmet_tcp_disc_port_addr()
1904 	struct nvmet_tcp_queue *queue;  in nvmet_tcp_exit()  local
1910 	list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)  in nvmet_tcp_exit()
1911 		kernel_sock_shutdown(queue->sock, SHUT_RDWR);  in nvmet_tcp_exit()