iommu/dma: force bouncing if the size is not cacheline-aligned
Similarly to the direct DMA, bounce small allocations as they may have originated from a kmalloc() cache not safe for DMA. Unlike the direct DMA, iommu_dma_map_sg() cannot call iommu_dma_map_sg_swiotlb() for all non-coherent devices as this would break some cases where the iova is expected to be contiguous (dmabuf). Instead, scan the scatterlist for any small sizes and only go the swiotlb path if any element of the list needs bouncing (note that iommu_dma_map_page() would still only bounce those buffers which are not DMA-aligned). To avoid scanning the scatterlist on the 'sync' operations, introduce an SG_DMA_SWIOTLB flag set by iommu_dma_map_sg_swiotlb(). The dev_use_swiotlb() function together with the newly added dev_use_sg_swiotlb() now check for both untrusted devices and unaligned kmalloc() buffers (suggested by Robin Murphy). Link: https://lkml.kernel.org/r/20230612153201.554742-16-catalin.marinas@arm.com Signed-off-by: Catalin Marinas <catalin.marinas@arm.com> Reviewed-by: Robin Murphy <robin.murphy@arm.com> Tested-by: Isaac J. Manjarres <isaacmanjarres@google.com> Cc: Joerg Roedel <joro@8bytes.org> Cc: Christoph Hellwig <hch@lst.de> Cc: Alasdair Kergon <agk@redhat.com> Cc: Ard Biesheuvel <ardb@kernel.org> Cc: Arnd Bergmann <arnd@arndb.de> Cc: Daniel Vetter <daniel@ffwll.ch> Cc: Greg Kroah-Hartman <gregkh@linuxfoundation.org> Cc: Herbert Xu <herbert@gondor.apana.org.au> Cc: Jerry Snitselaar <jsnitsel@redhat.com> Cc: Jonathan Cameron <jic23@kernel.org> Cc: Jonathan Cameron <Jonathan.Cameron@huawei.com> Cc: Lars-Peter Clausen <lars@metafoo.de> Cc: Logan Gunthorpe <logang@deltatee.com> Cc: Marc Zyngier <maz@kernel.org> Cc: Mark Brown <broonie@kernel.org> Cc: Mike Snitzer <snitzer@kernel.org> Cc: "Rafael J. Wysocki" <rafael@kernel.org> Cc: Saravana Kannan <saravanak@google.com> Cc: Vlastimil Babka <vbabka@suse.cz> Cc: Will Deacon <will@kernel.org> Signed-off-by: Andrew Morton <akpm@linux-foundation.org>
This commit is contained in:
parent
370645f41e
commit
861370f49c
|
@ -152,6 +152,7 @@ config IOMMU_DMA
|
|||
select IOMMU_IOVA
|
||||
select IRQ_MSI_IOMMU
|
||||
select NEED_SG_DMA_LENGTH
|
||||
select NEED_SG_DMA_FLAGS if SWIOTLB
|
||||
|
||||
# Shared Virtual Addressing
|
||||
config IOMMU_SVA
|
||||
|
|
|
@ -520,9 +520,38 @@ static bool dev_is_untrusted(struct device *dev)
|
|||
return dev_is_pci(dev) && to_pci_dev(dev)->untrusted;
|
||||
}
|
||||
|
||||
static bool dev_use_swiotlb(struct device *dev)
|
||||
static bool dev_use_swiotlb(struct device *dev, size_t size,
|
||||
enum dma_data_direction dir)
|
||||
{
|
||||
return IS_ENABLED(CONFIG_SWIOTLB) && dev_is_untrusted(dev);
|
||||
return IS_ENABLED(CONFIG_SWIOTLB) &&
|
||||
(dev_is_untrusted(dev) ||
|
||||
dma_kmalloc_needs_bounce(dev, size, dir));
|
||||
}
|
||||
|
||||
static bool dev_use_sg_swiotlb(struct device *dev, struct scatterlist *sg,
|
||||
int nents, enum dma_data_direction dir)
|
||||
{
|
||||
struct scatterlist *s;
|
||||
int i;
|
||||
|
||||
if (!IS_ENABLED(CONFIG_SWIOTLB))
|
||||
return false;
|
||||
|
||||
if (dev_is_untrusted(dev))
|
||||
return true;
|
||||
|
||||
/*
|
||||
* If kmalloc() buffers are not DMA-safe for this device and
|
||||
* direction, check the individual lengths in the sg list. If any
|
||||
* element is deemed unsafe, use the swiotlb for bouncing.
|
||||
*/
|
||||
if (!dma_kmalloc_safe(dev, dir)) {
|
||||
for_each_sg(sg, s, nents, i)
|
||||
if (!dma_kmalloc_size_aligned(s->length))
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -922,7 +951,7 @@ static void iommu_dma_sync_single_for_cpu(struct device *dev,
|
|||
{
|
||||
phys_addr_t phys;
|
||||
|
||||
if (dev_is_dma_coherent(dev) && !dev_use_swiotlb(dev))
|
||||
if (dev_is_dma_coherent(dev) && !dev_use_swiotlb(dev, size, dir))
|
||||
return;
|
||||
|
||||
phys = iommu_iova_to_phys(iommu_get_dma_domain(dev), dma_handle);
|
||||
|
@ -938,7 +967,7 @@ static void iommu_dma_sync_single_for_device(struct device *dev,
|
|||
{
|
||||
phys_addr_t phys;
|
||||
|
||||
if (dev_is_dma_coherent(dev) && !dev_use_swiotlb(dev))
|
||||
if (dev_is_dma_coherent(dev) && !dev_use_swiotlb(dev, size, dir))
|
||||
return;
|
||||
|
||||
phys = iommu_iova_to_phys(iommu_get_dma_domain(dev), dma_handle);
|
||||
|
@ -956,7 +985,7 @@ static void iommu_dma_sync_sg_for_cpu(struct device *dev,
|
|||
struct scatterlist *sg;
|
||||
int i;
|
||||
|
||||
if (dev_use_swiotlb(dev))
|
||||
if (sg_dma_is_swiotlb(sgl))
|
||||
for_each_sg(sgl, sg, nelems, i)
|
||||
iommu_dma_sync_single_for_cpu(dev, sg_dma_address(sg),
|
||||
sg->length, dir);
|
||||
|
@ -972,7 +1001,7 @@ static void iommu_dma_sync_sg_for_device(struct device *dev,
|
|||
struct scatterlist *sg;
|
||||
int i;
|
||||
|
||||
if (dev_use_swiotlb(dev))
|
||||
if (sg_dma_is_swiotlb(sgl))
|
||||
for_each_sg(sgl, sg, nelems, i)
|
||||
iommu_dma_sync_single_for_device(dev,
|
||||
sg_dma_address(sg),
|
||||
|
@ -998,7 +1027,8 @@ static dma_addr_t iommu_dma_map_page(struct device *dev, struct page *page,
|
|||
* If both the physical buffer start address and size are
|
||||
* page aligned, we don't need to use a bounce page.
|
||||
*/
|
||||
if (dev_use_swiotlb(dev) && iova_offset(iovad, phys | size)) {
|
||||
if (dev_use_swiotlb(dev, size, dir) &&
|
||||
iova_offset(iovad, phys | size)) {
|
||||
void *padding_start;
|
||||
size_t padding_size, aligned_size;
|
||||
|
||||
|
@ -1166,6 +1196,8 @@ static int iommu_dma_map_sg_swiotlb(struct device *dev, struct scatterlist *sg,
|
|||
struct scatterlist *s;
|
||||
int i;
|
||||
|
||||
sg_dma_mark_swiotlb(sg);
|
||||
|
||||
for_each_sg(sg, s, nents, i) {
|
||||
sg_dma_address(s) = iommu_dma_map_page(dev, sg_page(s),
|
||||
s->offset, s->length, dir, attrs);
|
||||
|
@ -1210,7 +1242,7 @@ static int iommu_dma_map_sg(struct device *dev, struct scatterlist *sg,
|
|||
goto out;
|
||||
}
|
||||
|
||||
if (dev_use_swiotlb(dev))
|
||||
if (dev_use_sg_swiotlb(dev, sg, nents, dir))
|
||||
return iommu_dma_map_sg_swiotlb(dev, sg, nents, dir, attrs);
|
||||
|
||||
if (!(attrs & DMA_ATTR_SKIP_CPU_SYNC))
|
||||
|
@ -1315,7 +1347,7 @@ static void iommu_dma_unmap_sg(struct device *dev, struct scatterlist *sg,
|
|||
struct scatterlist *tmp;
|
||||
int i;
|
||||
|
||||
if (dev_use_swiotlb(dev)) {
|
||||
if (sg_dma_is_swiotlb(sg)) {
|
||||
iommu_dma_unmap_sg_swiotlb(dev, sg, nents, dir, attrs);
|
||||
return;
|
||||
}
|
||||
|
|
|
@ -251,11 +251,13 @@ static inline void sg_unmark_end(struct scatterlist *sg)
|
|||
/*
|
||||
* One 64-bit architectures there is a 4-byte padding in struct scatterlist
|
||||
* (assuming also CONFIG_NEED_SG_DMA_LENGTH is set). Use this padding for DMA
|
||||
* flags bits to indicate when a specific dma address is a bus address.
|
||||
* flags bits to indicate when a specific dma address is a bus address or the
|
||||
* buffer may have been bounced via SWIOTLB.
|
||||
*/
|
||||
#ifdef CONFIG_NEED_SG_DMA_FLAGS
|
||||
|
||||
#define SG_DMA_BUS_ADDRESS (1 << 0)
|
||||
#define SG_DMA_BUS_ADDRESS (1 << 0)
|
||||
#define SG_DMA_SWIOTLB (1 << 1)
|
||||
|
||||
/**
|
||||
* sg_dma_is_bus_address - Return whether a given segment was marked
|
||||
|
@ -298,6 +300,34 @@ static inline void sg_dma_unmark_bus_address(struct scatterlist *sg)
|
|||
sg->dma_flags &= ~SG_DMA_BUS_ADDRESS;
|
||||
}
|
||||
|
||||
/**
|
||||
* sg_dma_is_swiotlb - Return whether the scatterlist was marked for SWIOTLB
|
||||
* bouncing
|
||||
* @sg: SG entry
|
||||
*
|
||||
* Description:
|
||||
* Returns true if the scatterlist was marked for SWIOTLB bouncing. Not all
|
||||
* elements may have been bounced, so the caller would have to check
|
||||
* individual SG entries with is_swiotlb_buffer().
|
||||
*/
|
||||
static inline bool sg_dma_is_swiotlb(struct scatterlist *sg)
|
||||
{
|
||||
return sg->dma_flags & SG_DMA_SWIOTLB;
|
||||
}
|
||||
|
||||
/**
|
||||
* sg_dma_mark_swiotlb - Mark the scatterlist for SWIOTLB bouncing
|
||||
* @sg: SG entry
|
||||
*
|
||||
* Description:
|
||||
* Marks a a scatterlist for SWIOTLB bounce. Not all SG entries may be
|
||||
* bounced.
|
||||
*/
|
||||
static inline void sg_dma_mark_swiotlb(struct scatterlist *sg)
|
||||
{
|
||||
sg->dma_flags |= SG_DMA_SWIOTLB;
|
||||
}
|
||||
|
||||
#else
|
||||
|
||||
static inline bool sg_dma_is_bus_address(struct scatterlist *sg)
|
||||
|
@ -310,6 +340,13 @@ static inline void sg_dma_mark_bus_address(struct scatterlist *sg)
|
|||
static inline void sg_dma_unmark_bus_address(struct scatterlist *sg)
|
||||
{
|
||||
}
|
||||
static inline bool sg_dma_is_swiotlb(struct scatterlist *sg)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
static inline void sg_dma_mark_swiotlb(struct scatterlist *sg)
|
||||
{
|
||||
}
|
||||
|
||||
#endif /* CONFIG_NEED_SG_DMA_FLAGS */
|
||||
|
||||
|
|
Loading…
Reference in New Issue