NVMe: Fix DMA mapping for admin commands

[firefly-linux-kernel-4.4.55.git] / drivers / block / nvme.c

diff --git a/drivers/block/nvme.c b/drivers/block/nvme.c
index 1cc01872f6dcf1eb28c51fd22d891d002fe918dc..3f8cae9dc96023e475c0a7cb69a5cf892efb37ed 100644 (file)
--- a/drivers/block/nvme.c
+++ b/drivers/block/nvme.c
@@ -1165,7 +1165,8 @@ static int nvme_user_admin_cmd(struct nvme_ns *ns,
 
        length = cmd.data_len;
        if (cmd.data_len) {
-               iod = nvme_map_user_pages(dev, 1, cmd.addr, length);
+               iod = nvme_map_user_pages(dev, cmd.opcode & 1, cmd.addr,
+                                                               length);
                if (IS_ERR(iod))
                        return PTR_ERR(iod);
                length = nvme_setup_prps(dev, &c.common, iod, length,
@@ -1178,7 +1179,8 @@ static int nvme_user_admin_cmd(struct nvme_ns *ns,
                status = nvme_submit_admin_cmd(dev, &c, NULL);
 
        if (cmd.data_len) {
-               nvme_unmap_user_pages(dev, 0, cmd.addr, cmd.data_len, iod);
+               nvme_unmap_user_pages(dev, cmd.opcode & 1, cmd.addr,
+                                                       cmd.data_len, iod);
                nvme_free_iod(dev, iod);
        }
        return status;