nvme: add support for the Write Zeroes command (6d31e3ba) · Commits · 戴 / test

drivers/nvme/host/core.c

+21 −0

Original line number	Diff line number	Diff line
		@@ -272,6 +272,21 @@ static inline int nvme_setup_discard(struct nvme_ns ns, struct request req,
		return BLK_MQ_RQ_QUEUE_OK;
		}

		static inline void nvme_setup_write_zeroes(struct nvme_ns *ns,
		struct request req, struct nvme_command cmnd)
		{
		struct nvme_write_zeroes_cmd *write_zeroes = &cmnd->write_zeroes;

		memset(cmnd, 0, sizeof(*cmnd));
		write_zeroes->opcode = nvme_cmd_write_zeroes;
		write_zeroes->nsid = cpu_to_le32(ns->ns_id);
		write_zeroes->slba =
		cpu_to_le64(nvme_block_nr(ns, blk_rq_pos(req)));
		write_zeroes->length =
		cpu_to_le16((blk_rq_bytes(req) >> ns->lba_shift) - 1);
		write_zeroes->control = 0;
		}

		static inline void nvme_setup_rw(struct nvme_ns ns, struct request req,
		struct nvme_command *cmnd)
		{
		@@ -325,6 +340,8 @@ int nvme_setup_cmd(struct nvme_ns ns, struct request req,
		nvme_setup_flush(ns, cmd);
		else if (req_op(req) == REQ_OP_DISCARD)
		ret = nvme_setup_discard(ns, req, cmd);
		else if (req_op(req) == REQ_OP_WRITE_ZEROES)
		nvme_setup_write_zeroes(ns, req, cmd);
		else
		nvme_setup_rw(ns, req, cmd);

		@@ -943,6 +960,10 @@ static void __nvme_revalidate_disk(struct gendisk disk, struct nvme_id_ns id)

		if (ns->ctrl->oncs & NVME_CTRL_ONCS_DSM)
		nvme_config_discard(ns);
		if (ns->ctrl->oncs & NVME_CTRL_ONCS_WRITE_ZEROES)
		blk_queue_max_write_zeroes_sectors(ns->queue,
		((u32)(USHRT_MAX + 1) * bs) >> 9);

		blk_mq_unfreeze_queue(disk->queue);
		}