Skip to content

Commit 7e210d5

Browse files
committed
Merge tag 'nvme-6.19-2026-02-05' of git://git.infradead.org/nvme into block-6.19
Pull NVMe fixes from Keith: "- Fix NULL pointer access setting up dma mappings (Keith) - Fix invalid memory access from malformed TCP PDU (YunJe)" * tag 'nvme-6.19-2026-02-05' of git://git.infradead.org/nvme: nvmet-tcp: add bounds checks in nvmet_tcp_build_pdu_iovec nvme-pci: handle changing device dma map requirements
2 parents 4da7c5c + 52a0a98 commit 7e210d5

2 files changed

Lines changed: 47 additions & 15 deletions

File tree

drivers/nvme/host/pci.c

Lines changed: 30 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -816,6 +816,32 @@ static void nvme_unmap_data(struct request *req)
816816
nvme_free_descriptors(req);
817817
}
818818

819+
static bool nvme_pci_prp_save_mapping(struct request *req,
820+
struct device *dma_dev,
821+
struct blk_dma_iter *iter)
822+
{
823+
struct nvme_iod *iod = blk_mq_rq_to_pdu(req);
824+
825+
if (dma_use_iova(&iod->dma_state) || !dma_need_unmap(dma_dev))
826+
return true;
827+
828+
if (!iod->nr_dma_vecs) {
829+
struct nvme_queue *nvmeq = req->mq_hctx->driver_data;
830+
831+
iod->dma_vecs = mempool_alloc(nvmeq->dev->dmavec_mempool,
832+
GFP_ATOMIC);
833+
if (!iod->dma_vecs) {
834+
iter->status = BLK_STS_RESOURCE;
835+
return false;
836+
}
837+
}
838+
839+
iod->dma_vecs[iod->nr_dma_vecs].addr = iter->addr;
840+
iod->dma_vecs[iod->nr_dma_vecs].len = iter->len;
841+
iod->nr_dma_vecs++;
842+
return true;
843+
}
844+
819845
static bool nvme_pci_prp_iter_next(struct request *req, struct device *dma_dev,
820846
struct blk_dma_iter *iter)
821847
{
@@ -825,12 +851,7 @@ static bool nvme_pci_prp_iter_next(struct request *req, struct device *dma_dev,
825851
return true;
826852
if (!blk_rq_dma_map_iter_next(req, dma_dev, &iod->dma_state, iter))
827853
return false;
828-
if (!dma_use_iova(&iod->dma_state) && dma_need_unmap(dma_dev)) {
829-
iod->dma_vecs[iod->nr_dma_vecs].addr = iter->addr;
830-
iod->dma_vecs[iod->nr_dma_vecs].len = iter->len;
831-
iod->nr_dma_vecs++;
832-
}
833-
return true;
854+
return nvme_pci_prp_save_mapping(req, dma_dev, iter);
834855
}
835856

836857
static blk_status_t nvme_pci_setup_data_prp(struct request *req,
@@ -843,15 +864,8 @@ static blk_status_t nvme_pci_setup_data_prp(struct request *req,
843864
unsigned int prp_len, i;
844865
__le64 *prp_list;
845866

846-
if (!dma_use_iova(&iod->dma_state) && dma_need_unmap(nvmeq->dev->dev)) {
847-
iod->dma_vecs = mempool_alloc(nvmeq->dev->dmavec_mempool,
848-
GFP_ATOMIC);
849-
if (!iod->dma_vecs)
850-
return BLK_STS_RESOURCE;
851-
iod->dma_vecs[0].addr = iter->addr;
852-
iod->dma_vecs[0].len = iter->len;
853-
iod->nr_dma_vecs = 1;
854-
}
867+
if (!nvme_pci_prp_save_mapping(req, nvmeq->dev->dev, iter))
868+
return iter->status;
855869

856870
/*
857871
* PRP1 always points to the start of the DMA transfers.
@@ -1219,6 +1233,7 @@ static blk_status_t nvme_prep_rq(struct request *req)
12191233
iod->nr_descriptors = 0;
12201234
iod->total_len = 0;
12211235
iod->meta_total_len = 0;
1236+
iod->nr_dma_vecs = 0;
12221237

12231238
ret = nvme_setup_cmd(req->q->queuedata, req);
12241239
if (ret)

drivers/nvme/target/tcp.c

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -349,28 +349,45 @@ static void nvmet_tcp_free_cmd_buffers(struct nvmet_tcp_cmd *cmd)
349349
cmd->req.sg = NULL;
350350
}
351351

352+
static void nvmet_tcp_fatal_error(struct nvmet_tcp_queue *queue);
353+
352354
static void nvmet_tcp_build_pdu_iovec(struct nvmet_tcp_cmd *cmd)
353355
{
354356
struct bio_vec *iov = cmd->iov;
355357
struct scatterlist *sg;
356358
u32 length, offset, sg_offset;
359+
unsigned int sg_remaining;
357360
int nr_pages;
358361

359362
length = cmd->pdu_len;
360363
nr_pages = DIV_ROUND_UP(length, PAGE_SIZE);
361364
offset = cmd->rbytes_done;
362365
cmd->sg_idx = offset / PAGE_SIZE;
363366
sg_offset = offset % PAGE_SIZE;
367+
if (!cmd->req.sg_cnt || cmd->sg_idx >= cmd->req.sg_cnt) {
368+
nvmet_tcp_fatal_error(cmd->queue);
369+
return;
370+
}
364371
sg = &cmd->req.sg[cmd->sg_idx];
372+
sg_remaining = cmd->req.sg_cnt - cmd->sg_idx;
365373

366374
while (length) {
375+
if (!sg_remaining) {
376+
nvmet_tcp_fatal_error(cmd->queue);
377+
return;
378+
}
379+
if (!sg->length || sg->length <= sg_offset) {
380+
nvmet_tcp_fatal_error(cmd->queue);
381+
return;
382+
}
367383
u32 iov_len = min_t(u32, length, sg->length - sg_offset);
368384

369385
bvec_set_page(iov, sg_page(sg), iov_len,
370386
sg->offset + sg_offset);
371387

372388
length -= iov_len;
373389
sg = sg_next(sg);
390+
sg_remaining--;
374391
iov++;
375392
sg_offset = 0;
376393
}

0 commit comments

Comments
 (0)