Lines Matching refs:queue

66  * queue before determining it to be idle.  This optional module behavior
110 struct nvmet_tcp_queue *queue;
221 static inline u16 nvmet_tcp_cmd_tag(struct nvmet_tcp_queue *queue,
224 if (unlikely(!queue->nr_cmds)) {
229 return cmd - queue->cmds;
257 nvmet_tcp_get_cmd(struct nvmet_tcp_queue *queue)
261 cmd = list_first_entry_or_null(&queue->free_list,
277 if (unlikely(cmd == &cmd->queue->connect))
280 list_add_tail(&cmd->entry, &cmd->queue->free_list);
283 static inline int queue_cpu(struct nvmet_tcp_queue *queue)
285 return queue->sock->sk->sk_incoming_cpu;
288 static inline u8 nvmet_tcp_hdgst_len(struct nvmet_tcp_queue *queue)
290 return queue->hdr_digest ? NVME_TCP_DIGEST_LENGTH : 0;
293 static inline u8 nvmet_tcp_ddgst_len(struct nvmet_tcp_queue *queue)
295 return queue->data_digest ? NVME_TCP_DIGEST_LENGTH : 0;
308 static int nvmet_tcp_verify_hdgst(struct nvmet_tcp_queue *queue,
316 pr_err("queue %d: header digest enabled but no header digest\n",
317 queue->idx);
322 nvmet_tcp_hdgst(queue->rcv_hash, pdu, len);
325 pr_err("queue %d: header digest error: recv %#x expected %#x\n",
326 queue->idx, le32_to_cpu(recv_digest),
334 static int nvmet_tcp_check_ddgst(struct nvmet_tcp_queue *queue, void *pdu)
337 u8 digest_len = nvmet_tcp_hdgst_len(queue);
344 pr_err("queue %d: data digest flag is cleared\n", queue->idx);
390 static void nvmet_tcp_fatal_error(struct nvmet_tcp_queue *queue)
392 queue->rcv_state = NVMET_TCP_RECV_ERR;
393 if (queue->nvme_sq.ctrl)
394 nvmet_ctrl_fatal_error(queue->nvme_sq.ctrl);
396 kernel_sock_shutdown(queue->sock, SHUT_RDWR);
399 static void nvmet_tcp_socket_error(struct nvmet_tcp_queue *queue, int status)
401 queue->rcv_state = NVMET_TCP_RECV_ERR;
403 kernel_sock_shutdown(queue->sock, SHUT_RDWR);
405 nvmet_tcp_fatal_error(queue);
456 struct nvmet_tcp_queue *queue = cmd->queue;
457 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
458 u8 ddgst = nvmet_tcp_ddgst_len(cmd->queue);
464 pdu->hdr.flags = NVME_TCP_F_DATA_LAST | (queue->nvme_sq.sqhd_disabled ?
475 if (queue->data_digest) {
477 nvmet_tcp_calc_ddgst(queue->snd_hash, cmd);
480 if (cmd->queue->hdr_digest) {
482 nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));
489 struct nvmet_tcp_queue *queue = cmd->queue;
490 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
502 pdu->ttag = nvmet_tcp_cmd_tag(cmd->queue, cmd);
505 if (cmd->queue->hdr_digest) {
507 nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));
514 struct nvmet_tcp_queue *queue = cmd->queue;
515 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
525 if (cmd->queue->hdr_digest) {
527 nvmet_tcp_hdgst(queue->snd_hash, pdu, sizeof(*pdu));
531 static void nvmet_tcp_process_resp_list(struct nvmet_tcp_queue *queue)
536 for (node = llist_del_all(&queue->resp_list); node; node = node->next) {
538 list_add(&cmd->entry, &queue->resp_send_list);
539 queue->send_list_len++;
543 static struct nvmet_tcp_cmd *nvmet_tcp_fetch_cmd(struct nvmet_tcp_queue *queue)
545 queue->snd_cmd = list_first_entry_or_null(&queue->resp_send_list,
547 if (!queue->snd_cmd) {
548 nvmet_tcp_process_resp_list(queue);
549 queue->snd_cmd =
550 list_first_entry_or_null(&queue->resp_send_list,
552 if (unlikely(!queue->snd_cmd))
556 list_del_init(&queue->snd_cmd->entry);
557 queue->send_list_len--;
559 if (nvmet_tcp_need_data_out(queue->snd_cmd))
560 nvmet_setup_c2h_data_pdu(queue->snd_cmd);
561 else if (nvmet_tcp_need_data_in(queue->snd_cmd))
562 nvmet_setup_r2t_pdu(queue->snd_cmd);
564 nvmet_setup_response_pdu(queue->snd_cmd);
566 return queue->snd_cmd;
573 struct nvmet_tcp_queue *queue = cmd->queue;
577 if (unlikely(cmd == queue->cmd)) {
586 if (queue->rcv_state == NVMET_TCP_RECV_PDU &&
592 llist_add(&cmd->lentry, &queue->resp_list);
593 queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &cmd->queue->io_work);
610 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
616 ret = sock_sendmsg(cmd->queue->sock, &msg);
633 struct nvmet_tcp_queue *queue = cmd->queue;
644 if ((!last_in_batch && cmd->queue->send_list_len) ||
646 queue->data_digest || !queue->nvme_sq.sqhd_disabled)
651 ret = sock_sendmsg(cmd->queue->sock, &msg);
665 if (queue->data_digest) {
669 if (queue->nvme_sq.sqhd_disabled) {
670 cmd->queue->snd_cmd = NULL;
677 if (queue->nvme_sq.sqhd_disabled)
689 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
693 if (!last_in_batch && cmd->queue->send_list_len)
700 ret = sock_sendmsg(cmd->queue->sock, &msg);
710 cmd->queue->snd_cmd = NULL;
719 u8 hdgst = nvmet_tcp_hdgst_len(cmd->queue);
723 if (!last_in_batch && cmd->queue->send_list_len)
730 ret = sock_sendmsg(cmd->queue->sock, &msg);
739 cmd->queue->snd_cmd = NULL;
745 struct nvmet_tcp_queue *queue = cmd->queue;
754 if (!last_in_batch && cmd->queue->send_list_len)
759 ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);
769 if (queue->nvme_sq.sqhd_disabled) {
770 cmd->queue->snd_cmd = NULL;
778 static int nvmet_tcp_try_send_one(struct nvmet_tcp_queue *queue,
781 struct nvmet_tcp_cmd *cmd = queue->snd_cmd;
784 if (!cmd || queue->state == NVMET_TCP_Q_DISCONNECTING) {
785 cmd = nvmet_tcp_fetch_cmd(queue);
827 static int nvmet_tcp_try_send(struct nvmet_tcp_queue *queue,
833 ret = nvmet_tcp_try_send_one(queue, i == budget - 1);
835 nvmet_tcp_socket_error(queue, ret);
846 static void nvmet_prepare_receive_pdu(struct nvmet_tcp_queue *queue)
848 queue->offset = 0;
849 queue->left = sizeof(struct nvme_tcp_hdr);
850 queue->cmd = NULL;
851 queue->rcv_state = NVMET_TCP_RECV_PDU;
854 static void nvmet_tcp_free_crypto(struct nvmet_tcp_queue *queue)
856 struct crypto_ahash *tfm = crypto_ahash_reqtfm(queue->rcv_hash);
858 ahash_request_free(queue->rcv_hash);
859 ahash_request_free(queue->snd_hash);
863 static int nvmet_tcp_alloc_crypto(struct nvmet_tcp_queue *queue)
871 queue->snd_hash = ahash_request_alloc(tfm, GFP_KERNEL);
872 if (!queue->snd_hash)
874 ahash_request_set_callback(queue->snd_hash, 0, NULL, NULL);
876 queue->rcv_hash = ahash_request_alloc(tfm, GFP_KERNEL);
877 if (!queue->rcv_hash)
879 ahash_request_set_callback(queue->rcv_hash, 0, NULL, NULL);
883 ahash_request_free(queue->snd_hash);
890 static int nvmet_tcp_handle_icreq(struct nvmet_tcp_queue *queue)
892 struct nvme_tcp_icreq_pdu *icreq = &queue->pdu.icreq;
893 struct nvme_tcp_icresp_pdu *icresp = &queue->pdu.icresp;
901 nvmet_tcp_fatal_error(queue);
906 pr_err("queue %d: bad pfv %d\n", queue->idx, icreq->pfv);
911 pr_err("queue %d: unsupported hpda %d\n", queue->idx,
916 queue->hdr_digest = !!(icreq->digest & NVME_TCP_HDR_DIGEST_ENABLE);
917 queue->data_digest = !!(icreq->digest & NVME_TCP_DATA_DIGEST_ENABLE);
918 if (queue->hdr_digest || queue->data_digest) {
919 ret = nvmet_tcp_alloc_crypto(queue);
932 if (queue->hdr_digest)
934 if (queue->data_digest)
939 ret = kernel_sendmsg(queue->sock, &msg, &iov, 1, iov.iov_len);
941 queue->state = NVMET_TCP_Q_FAILED;
942 return ret; /* queue removal will cleanup */
945 queue->state = NVMET_TCP_Q_LIVE;
946 nvmet_prepare_receive_pdu(queue);
950 static void nvmet_tcp_handle_req_failure(struct nvmet_tcp_queue *queue,
965 nvmet_prepare_receive_pdu(queue);
971 pr_err("queue %d: failed to map data\n", queue->idx);
972 nvmet_tcp_fatal_error(queue);
976 queue->rcv_state = NVMET_TCP_RECV_DATA;
981 static int nvmet_tcp_handle_h2c_data_pdu(struct nvmet_tcp_queue *queue)
983 struct nvme_tcp_data_pdu *data = &queue->pdu.data;
987 if (likely(queue->nr_cmds)) {
988 if (unlikely(data->ttag >= queue->nr_cmds)) {
989 pr_err("queue %d: received out of bound ttag %u, nr_cmds %u\n",
990 queue->idx, data->ttag, queue->nr_cmds);
993 cmd = &queue->cmds[data->ttag];
995 cmd = &queue->connect;
1006 nvmet_tcp_hdgst_len(queue) -
1007 nvmet_tcp_ddgst_len(queue) -
1019 queue->cmd = cmd;
1020 queue->rcv_state = NVMET_TCP_RECV_DATA;
1026 nvmet_tcp_fatal_error(queue);
1030 static int nvmet_tcp_done_recv_pdu(struct nvmet_tcp_queue *queue)
1032 struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;
1033 struct nvme_command *nvme_cmd = &queue->pdu.cmd.cmd;
1037 if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {
1041 nvmet_tcp_fatal_error(queue);
1044 return nvmet_tcp_handle_icreq(queue);
1048 pr_err("queue %d: received icreq pdu in state %d\n",
1049 queue->idx, queue->state);
1050 nvmet_tcp_fatal_error(queue);
1055 ret = nvmet_tcp_handle_h2c_data_pdu(queue);
1061 queue->cmd = nvmet_tcp_get_cmd(queue);
1062 if (unlikely(!queue->cmd)) {
1064 pr_err("queue %d: out of commands (%d) send_list_len: %d, opcode: %d",
1065 queue->idx, queue->nr_cmds, queue->send_list_len,
1067 nvmet_tcp_fatal_error(queue);
1071 req = &queue->cmd->req;
1074 if (unlikely(!nvmet_req_init(req, &queue->nvme_cq,
1075 &queue->nvme_sq, &nvmet_tcp_ops))) {
1081 nvmet_tcp_handle_req_failure(queue, queue->cmd, req);
1085 ret = nvmet_tcp_map_data(queue->cmd);
1087 pr_err("queue %d: failed to map data\n", queue->idx);
1088 if (nvmet_tcp_has_inline_data(queue->cmd))
1089 nvmet_tcp_fatal_error(queue);
1096 if (nvmet_tcp_need_data_in(queue->cmd)) {
1097 if (nvmet_tcp_has_inline_data(queue->cmd)) {
1098 queue->rcv_state = NVMET_TCP_RECV_DATA;
1099 nvmet_tcp_build_pdu_iovec(queue->cmd);
1103 nvmet_tcp_queue_response(&queue->cmd->req);
1107 queue->cmd->req.execute(&queue->cmd->req);
1109 nvmet_prepare_receive_pdu(queue);
1141 static int nvmet_tcp_tls_record_ok(struct nvmet_tcp_queue *queue,
1148 ctype = tls_get_record_type(queue->sock->sk, cmsg);
1155 tls_alert_recv(queue->sock->sk, msg, &level, &description);
1157 pr_err("queue %d: TLS Alert desc %u\n",
1158 queue->idx, description);
1161 pr_warn("queue %d: TLS Alert desc %u\n",
1162 queue->idx, description);
1168 pr_err("queue %d: TLS record %d unhandled\n",
1169 queue->idx, ctype);
1176 static int nvmet_tcp_try_recv_pdu(struct nvmet_tcp_queue *queue)
1178 struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;
1185 iov.iov_base = (void *)&queue->pdu + queue->offset;
1186 iov.iov_len = queue->left;
1187 if (queue->tls_pskid) {
1191 len = kernel_recvmsg(queue->sock, &msg, &iov, 1,
1195 if (queue->tls_pskid) {
1196 ret = nvmet_tcp_tls_record_ok(queue, &msg, cbuf);
1201 queue->offset += len;
1202 queue->left -= len;
1203 if (queue->left)
1206 if (queue->offset == sizeof(struct nvme_tcp_hdr)) {
1207 u8 hdgst = nvmet_tcp_hdgst_len(queue);
1211 nvmet_tcp_fatal_error(queue);
1220 queue->left = hdr->hlen - queue->offset + hdgst;
1224 if (queue->hdr_digest &&
1225 nvmet_tcp_verify_hdgst(queue, &queue->pdu, hdr->hlen)) {
1226 nvmet_tcp_fatal_error(queue); /* fatal */
1230 if (queue->data_digest &&
1231 nvmet_tcp_check_ddgst(queue, &queue->pdu)) {
1232 nvmet_tcp_fatal_error(queue); /* fatal */
1236 return nvmet_tcp_done_recv_pdu(queue);
1241 struct nvmet_tcp_queue *queue = cmd->queue;
1243 nvmet_tcp_calc_ddgst(queue->rcv_hash, cmd);
1244 queue->offset = 0;
1245 queue->left = NVME_TCP_DIGEST_LENGTH;
1246 queue->rcv_state = NVMET_TCP_RECV_DDGST;
1249 static int nvmet_tcp_try_recv_data(struct nvmet_tcp_queue *queue)
1251 struct nvmet_tcp_cmd *cmd = queue->cmd;
1255 len = sock_recvmsg(cmd->queue->sock, &cmd->recv_msg,
1259 if (queue->tls_pskid) {
1260 ret = nvmet_tcp_tls_record_ok(cmd->queue,
1270 if (queue->data_digest) {
1278 nvmet_prepare_receive_pdu(queue);
1282 static int nvmet_tcp_try_recv_ddgst(struct nvmet_tcp_queue *queue)
1284 struct nvmet_tcp_cmd *cmd = queue->cmd;
1289 .iov_base = (void *)&cmd->recv_ddgst + queue->offset,
1290 .iov_len = queue->left
1293 if (queue->tls_pskid) {
1297 len = kernel_recvmsg(queue->sock, &msg, &iov, 1,
1301 if (queue->tls_pskid) {
1302 ret = nvmet_tcp_tls_record_ok(queue, &msg, cbuf);
1307 queue->offset += len;
1308 queue->left -= len;
1309 if (queue->left)
1312 if (queue->data_digest && cmd->exp_ddgst != cmd->recv_ddgst) {
1313 pr_err("queue %d: cmd %d pdu (%d) data digest error: recv %#x expected %#x\n",
1314 queue->idx, cmd->req.cmd->common.command_id,
1315 queue->pdu.cmd.hdr.type, le32_to_cpu(cmd->recv_ddgst),
1319 nvmet_tcp_fatal_error(queue);
1329 nvmet_prepare_receive_pdu(queue);
1333 static int nvmet_tcp_try_recv_one(struct nvmet_tcp_queue *queue)
1337 if (unlikely(queue->rcv_state == NVMET_TCP_RECV_ERR))
1340 if (queue->rcv_state == NVMET_TCP_RECV_PDU) {
1341 result = nvmet_tcp_try_recv_pdu(queue);
1346 if (queue->rcv_state == NVMET_TCP_RECV_DATA) {
1347 result = nvmet_tcp_try_recv_data(queue);
1352 if (queue->rcv_state == NVMET_TCP_RECV_DDGST) {
1353 result = nvmet_tcp_try_recv_ddgst(queue);
1367 static int nvmet_tcp_try_recv(struct nvmet_tcp_queue *queue,
1373 ret = nvmet_tcp_try_recv_one(queue);
1375 nvmet_tcp_socket_error(queue, ret);
1388 struct nvmet_tcp_queue *queue =
1391 WARN_ON(queue->state != NVMET_TCP_Q_DISCONNECTING);
1392 queue_work(nvmet_wq, &queue->release_work);
1395 static void nvmet_tcp_schedule_release_queue(struct nvmet_tcp_queue *queue)
1397 spin_lock_bh(&queue->state_lock);
1398 if (queue->state == NVMET_TCP_Q_TLS_HANDSHAKE) {
1400 tls_handshake_cancel(queue->sock->sk);
1402 if (queue->state != NVMET_TCP_Q_DISCONNECTING) {
1403 queue->state = NVMET_TCP_Q_DISCONNECTING;
1404 kref_put(&queue->kref, nvmet_tcp_release_queue);
1406 spin_unlock_bh(&queue->state_lock);
1409 static inline void nvmet_tcp_arm_queue_deadline(struct nvmet_tcp_queue *queue)
1411 queue->poll_end = jiffies + usecs_to_jiffies(idle_poll_period_usecs);
1414 static bool nvmet_tcp_check_queue_deadline(struct nvmet_tcp_queue *queue,
1421 nvmet_tcp_arm_queue_deadline(queue);
1423 return !time_after(jiffies, queue->poll_end);
1428 struct nvmet_tcp_queue *queue =
1436 ret = nvmet_tcp_try_recv(queue, NVMET_TCP_RECV_BUDGET, &ops);
1442 ret = nvmet_tcp_try_send(queue, NVMET_TCP_SEND_BUDGET, &ops);
1454 if (nvmet_tcp_check_queue_deadline(queue, ops) || pending)
1455 queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);
1458 static int nvmet_tcp_alloc_cmd(struct nvmet_tcp_queue *queue,
1461 u8 hdgst = nvmet_tcp_hdgst_len(queue);
1463 c->queue = queue;
1464 c->req.port = queue->port->nport;
1466 c->cmd_pdu = page_frag_alloc(&queue->pf_cache,
1472 c->rsp_pdu = page_frag_alloc(&queue->pf_cache,
1478 c->data_pdu = page_frag_alloc(&queue->pf_cache,
1483 c->r2t_pdu = page_frag_alloc(&queue->pf_cache,
1488 if (queue->state == NVMET_TCP_Q_TLS_HANDSHAKE) {
1494 list_add_tail(&c->entry, &queue->free_list);
1514 static int nvmet_tcp_alloc_cmds(struct nvmet_tcp_queue *queue)
1517 int i, ret = -EINVAL, nr_cmds = queue->nr_cmds;
1524 ret = nvmet_tcp_alloc_cmd(queue, cmds + i);
1529 queue->cmds = cmds;
1540 static void nvmet_tcp_free_cmds(struct nvmet_tcp_queue *queue)
1542 struct nvmet_tcp_cmd *cmds = queue->cmds;
1545 for (i = 0; i < queue->nr_cmds; i++)
1548 nvmet_tcp_free_cmd(&queue->connect);
1552 static void nvmet_tcp_restore_socket_callbacks(struct nvmet_tcp_queue *queue)
1554 struct socket *sock = queue->sock;
1557 sock->sk->sk_data_ready = queue->data_ready;
1558 sock->sk->sk_state_change = queue->state_change;
1559 sock->sk->sk_write_space = queue->write_space;
1564 static void nvmet_tcp_uninit_data_in_cmds(struct nvmet_tcp_queue *queue)
1566 struct nvmet_tcp_cmd *cmd = queue->cmds;
1569 for (i = 0; i < queue->nr_cmds; i++, cmd++) {
1574 if (!queue->nr_cmds && nvmet_tcp_need_data_in(&queue->connect)) {
1576 nvmet_req_uninit(&queue->connect.req);
1580 static void nvmet_tcp_free_cmd_data_in_buffers(struct nvmet_tcp_queue *queue)
1582 struct nvmet_tcp_cmd *cmd = queue->cmds;
1585 for (i = 0; i < queue->nr_cmds; i++, cmd++)
1587 nvmet_tcp_free_cmd_buffers(&queue->connect);
1592 struct nvmet_tcp_queue *queue =
1596 list_del_init(&queue->queue_list);
1599 nvmet_tcp_restore_socket_callbacks(queue);
1600 cancel_delayed_work_sync(&queue->tls_handshake_tmo_work);
1601 cancel_work_sync(&queue->io_work);
1603 queue->rcv_state = NVMET_TCP_RECV_ERR;
1605 nvmet_tcp_uninit_data_in_cmds(queue);
1606 nvmet_sq_destroy(&queue->nvme_sq);
1607 cancel_work_sync(&queue->io_work);
1608 nvmet_tcp_free_cmd_data_in_buffers(queue);
1610 fput(queue->sock->file);
1611 nvmet_tcp_free_cmds(queue);
1612 if (queue->hdr_digest || queue->data_digest)
1613 nvmet_tcp_free_crypto(queue);
1614 ida_free(&nvmet_tcp_queue_ida, queue->idx);
1615 page_frag_cache_drain(&queue->pf_cache);
1616 kfree(queue);
1621 struct nvmet_tcp_queue *queue;
1626 queue = sk->sk_user_data;
1627 if (likely(queue)) {
1628 if (queue->data_ready)
1629 queue->data_ready(sk);
1630 if (queue->state != NVMET_TCP_Q_TLS_HANDSHAKE)
1631 queue_work_on(queue_cpu(queue), nvmet_tcp_wq,
1632 &queue->io_work);
1639 struct nvmet_tcp_queue *queue;
1642 queue = sk->sk_user_data;
1643 if (unlikely(!queue))
1646 if (unlikely(queue->state == NVMET_TCP_Q_CONNECTING)) {
1647 queue->write_space(sk);
1653 queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);
1661 struct nvmet_tcp_queue *queue;
1664 queue = sk->sk_user_data;
1665 if (!queue)
1676 nvmet_tcp_schedule_release_queue(queue);
1679 pr_warn("queue %d unhandled state %d\n",
1680 queue->idx, sk->sk_state);
1686 static int nvmet_tcp_set_queue_sock(struct nvmet_tcp_queue *queue)
1688 struct socket *sock = queue->sock;
1693 (struct sockaddr *)&queue->sockaddr);
1698 (struct sockaddr *)&queue->sockaddr_peer);
1703 * Cleanup whatever is sitting in the TCP transmit queue on socket
1725 sock->sk->sk_user_data = queue;
1726 queue->data_ready = sock->sk->sk_data_ready;
1728 queue->state_change = sock->sk->sk_state_change;
1730 queue->write_space = sock->sk->sk_write_space;
1733 nvmet_tcp_arm_queue_deadline(queue);
1734 queue_work_on(queue_cpu(queue), nvmet_tcp_wq, &queue->io_work);
1742 static int nvmet_tcp_try_peek_pdu(struct nvmet_tcp_queue *queue)
1744 struct nvme_tcp_hdr *hdr = &queue->pdu.cmd.hdr;
1747 .iov_base = (u8 *)&queue->pdu + queue->offset,
1757 if (nvmet_port_secure_channel_required(queue->port->nport))
1760 len = kernel_recvmsg(queue->sock, &msg, &iov, 1,
1763 pr_debug("queue %d: peek error %d\n",
1764 queue->idx, len);
1768 ret = nvmet_tcp_tls_record_ok(queue, &msg, cbuf);
1773 pr_debug("queue %d: short read, %d bytes missing\n",
1774 queue->idx, (int)iov.iov_len - len);
1777 pr_debug("queue %d: hdr type %d hlen %d plen %d size %d\n",
1778 queue->idx, hdr->type, hdr->hlen, hdr->plen,
1783 pr_debug("queue %d: icreq detected\n",
1784 queue->idx);
1793 struct nvmet_tcp_queue *queue = data;
1795 pr_debug("queue %d: TLS handshake done, key %x, status %d\n",
1796 queue->idx, peerid, status);
1797 spin_lock_bh(&queue->state_lock);
1798 if (WARN_ON(queue->state != NVMET_TCP_Q_TLS_HANDSHAKE)) {
1799 spin_unlock_bh(&queue->state_lock);
1803 queue->tls_pskid = peerid;
1804 queue->state = NVMET_TCP_Q_CONNECTING;
1806 queue->state = NVMET_TCP_Q_FAILED;
1807 spin_unlock_bh(&queue->state_lock);
1809 cancel_delayed_work_sync(&queue->tls_handshake_tmo_work);
1811 nvmet_tcp_schedule_release_queue(queue);
1813 nvmet_tcp_set_queue_sock(queue);
1814 kref_put(&queue->kref, nvmet_tcp_release_queue);
1819 struct nvmet_tcp_queue *queue = container_of(to_delayed_work(w),
1822 pr_warn("queue %d: TLS handshake timeout\n", queue->idx);
1826 if (!tls_handshake_cancel(queue->sock->sk))
1828 spin_lock_bh(&queue->state_lock);
1829 if (WARN_ON(queue->state != NVMET_TCP_Q_TLS_HANDSHAKE)) {
1830 spin_unlock_bh(&queue->state_lock);
1833 queue->state = NVMET_TCP_Q_FAILED;
1834 spin_unlock_bh(&queue->state_lock);
1835 nvmet_tcp_schedule_release_queue(queue);
1836 kref_put(&queue->kref, nvmet_tcp_release_queue);
1839 static int nvmet_tcp_tls_handshake(struct nvmet_tcp_queue *queue)
1844 if (queue->state != NVMET_TCP_Q_TLS_HANDSHAKE) {
1845 pr_warn("cannot start TLS in state %d\n", queue->state);
1849 kref_get(&queue->kref);
1850 pr_debug("queue %d: TLS ServerHello\n", queue->idx);
1852 args.ta_sock = queue->sock;
1854 args.ta_data = queue;
1855 args.ta_keyring = key_serial(queue->port->nport->keyring);
1860 kref_put(&queue->kref, nvmet_tcp_release_queue);
1863 queue_delayed_work(nvmet_wq, &queue->tls_handshake_tmo_work,
1875 struct nvmet_tcp_queue *queue;
1879 queue = kzalloc(sizeof(*queue), GFP_KERNEL);
1880 if (!queue) {
1885 INIT_WORK(&queue->release_work, nvmet_tcp_release_queue_work);
1886 INIT_WORK(&queue->io_work, nvmet_tcp_io_work);
1887 kref_init(&queue->kref);
1888 queue->sock = newsock;
1889 queue->port = port;
1890 queue->nr_cmds = 0;
1891 spin_lock_init(&queue->state_lock);
1892 if (queue->port->nport->disc_addr.tsas.tcp.sectype ==
1894 queue->state = NVMET_TCP_Q_TLS_HANDSHAKE;
1896 queue->state = NVMET_TCP_Q_CONNECTING;
1897 INIT_LIST_HEAD(&queue->free_list);
1898 init_llist_head(&queue->resp_list);
1899 INIT_LIST_HEAD(&queue->resp_send_list);
1901 sock_file = sock_alloc_file(queue->sock, O_CLOEXEC, NULL);
1907 queue->idx = ida_alloc(&nvmet_tcp_queue_ida, GFP_KERNEL);
1908 if (queue->idx < 0) {
1909 ret = queue->idx;
1913 ret = nvmet_tcp_alloc_cmd(queue, &queue->connect);
1917 ret = nvmet_sq_init(&queue->nvme_sq);
1921 nvmet_prepare_receive_pdu(queue);
1924 list_add_tail(&queue->queue_list, &nvmet_tcp_queue_list);
1927 INIT_DELAYED_WORK(&queue->tls_handshake_tmo_work,
1930 if (queue->state == NVMET_TCP_Q_TLS_HANDSHAKE) {
1931 struct sock *sk = queue->sock->sk;
1938 if (!nvmet_tcp_try_peek_pdu(queue)) {
1939 if (!nvmet_tcp_tls_handshake(queue))
1945 queue->state = NVMET_TCP_Q_CONNECTING;
1949 ret = nvmet_tcp_set_queue_sock(queue);
1956 list_del_init(&queue->queue_list);
1958 nvmet_sq_destroy(&queue->nvme_sq);
1960 nvmet_tcp_free_cmd(&queue->connect);
1962 ida_free(&nvmet_tcp_queue_ida, queue->idx);
1964 fput(queue->sock->file);
1966 kfree(queue);
1968 pr_err("failed to allocate queue, error %d\n", ret);
2088 struct nvmet_tcp_queue *queue;
2091 list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)
2092 if (queue->port == port)
2093 kernel_sock_shutdown(queue->sock, SHUT_RDWR);
2118 struct nvmet_tcp_queue *queue;
2121 list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)
2122 if (queue->nvme_sq.ctrl == ctrl)
2123 kernel_sock_shutdown(queue->sock, SHUT_RDWR);
2129 struct nvmet_tcp_queue *queue =
2148 queue->nr_cmds = sq->size * 2;
2149 if (nvmet_tcp_alloc_cmds(queue))
2162 struct nvmet_tcp_queue *queue = cmd->queue;
2164 sprintf(traddr, "%pISc", (struct sockaddr *)&queue->sockaddr);
2203 struct nvmet_tcp_queue *queue;
2209 list_for_each_entry(queue, &nvmet_tcp_queue_list, queue_list)
2210 kernel_sock_shutdown(queue->sock, SHUT_RDWR);