From: Joerg Roedel <jroe...@suse.de>

This function converts dma_data_direction to
iommu-protection flags. This will be needed on multiple
places in the code, so this will save some code.

Signed-off-by: Joerg Roedel <jroe...@suse.de>
---
 drivers/iommu/amd_iommu.c | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/drivers/iommu/amd_iommu.c b/drivers/iommu/amd_iommu.c
index 1bb003e..78b278b 100644
--- a/drivers/iommu/amd_iommu.c
+++ b/drivers/iommu/amd_iommu.c
@@ -2260,6 +2260,17 @@ static void update_domain(struct protection_domain 
*domain)
        domain->updated = false;
 }
 
+static int dir2prot(enum dma_data_direction direction)
+{
+       if (direction == DMA_TO_DEVICE)
+               return IOMMU_PROT_IR;
+       else if (direction == DMA_FROM_DEVICE)
+               return IOMMU_PROT_IW;
+       else if (direction == DMA_BIDIRECTIONAL)
+               return IOMMU_PROT_IW | IOMMU_PROT_IR;
+       else
+               return 0;
+}
 /*
  * This function contains common code for mapping of a physically
  * contiguous memory region into DMA address space. It is used by all
@@ -2270,7 +2281,7 @@ static dma_addr_t __map_single(struct device *dev,
                               struct dma_ops_domain *dma_dom,
                               phys_addr_t paddr,
                               size_t size,
-                              int direction,
+                              enum dma_data_direction direction,
                               u64 dma_mask)
 {
        dma_addr_t offset = paddr & ~PAGE_MASK;
@@ -2286,12 +2297,7 @@ static dma_addr_t __map_single(struct device *dev,
        if (address == DMA_ERROR_CODE)
                goto out;
 
-       if (direction == DMA_TO_DEVICE)
-               prot = IOMMU_PROT_IR;
-       else if (direction == DMA_FROM_DEVICE)
-               prot = IOMMU_PROT_IW;
-       else if (direction == DMA_BIDIRECTIONAL)
-               prot = IOMMU_PROT_IW | IOMMU_PROT_IR;
+       prot = dir2prot(direction);
 
        start = address;
        for (i = 0; i < pages; ++i) {
-- 
1.9.1

_______________________________________________
iommu mailing list
iommu@lists.linux-foundation.org
https://lists.linuxfoundation.org/mailman/listinfo/iommu

Reply via email to