diff options
Diffstat (limited to 'kernel/dma/swiotlb.c')
| -rw-r--r-- | kernel/dma/swiotlb.c | 37 | 
1 files changed, 22 insertions, 15 deletions
diff --git a/kernel/dma/swiotlb.c b/kernel/dma/swiotlb.c index 9de232229063..673a2cdb2656 100644 --- a/kernel/dma/swiotlb.c +++ b/kernel/dma/swiotlb.c @@ -444,7 +444,9 @@ static void swiotlb_bounce(phys_addr_t orig_addr, phys_addr_t tlb_addr,  phys_addr_t swiotlb_tbl_map_single(struct device *hwdev,  				   dma_addr_t tbl_dma_addr, -				   phys_addr_t orig_addr, size_t size, +				   phys_addr_t orig_addr, +				   size_t mapping_size, +				   size_t alloc_size,  				   enum dma_data_direction dir,  				   unsigned long attrs)  { @@ -461,8 +463,13 @@ phys_addr_t swiotlb_tbl_map_single(struct device *hwdev,  		panic("Can not allocate SWIOTLB buffer earlier and can't now provide you with the DMA bounce buffer");  	if (mem_encrypt_active()) -		pr_warn_once("%s is active and system is using DMA bounce buffers\n", -			     sme_active() ? "SME" : "SEV"); +		pr_warn_once("Memory encryption is active and system is using DMA bounce buffers\n"); + +	if (mapping_size > alloc_size) { +		dev_warn_once(hwdev, "Invalid sizes (mapping: %zd bytes, alloc: %zd bytes)", +			      mapping_size, alloc_size); +		return (phys_addr_t)DMA_MAPPING_ERROR; +	}  	mask = dma_get_seg_boundary(hwdev); @@ -471,8 +478,8 @@ phys_addr_t swiotlb_tbl_map_single(struct device *hwdev,  	offset_slots = ALIGN(tbl_dma_addr, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT;  	/* - 	 * Carefully handle integer overflow which can occur when mask == ~0UL. - 	 */ +	 * Carefully handle integer overflow which can occur when mask == ~0UL. +	 */  	max_slots = mask + 1  		    ? ALIGN(mask + 1, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT  		    : 1UL << (BITS_PER_LONG - IO_TLB_SHIFT); @@ -481,8 +488,8 @@ phys_addr_t swiotlb_tbl_map_single(struct device *hwdev,  	 * For mappings greater than or equal to a page, we limit the stride  	 * (and hence alignment) to a page size.  	 */ -	nslots = ALIGN(size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT; -	if (size >= PAGE_SIZE) +	nslots = ALIGN(alloc_size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT; +	if (alloc_size >= PAGE_SIZE)  		stride = (1 << (PAGE_SHIFT - IO_TLB_SHIFT));  	else  		stride = 1; @@ -547,7 +554,7 @@ not_found:  	spin_unlock_irqrestore(&io_tlb_lock, flags);  	if (!(attrs & DMA_ATTR_NO_WARN) && printk_ratelimit())  		dev_warn(hwdev, "swiotlb buffer is full (sz: %zd bytes), total %lu (slots), used %lu (slots)\n", -			 size, io_tlb_nslabs, tmp_io_tlb_used); +			 alloc_size, io_tlb_nslabs, tmp_io_tlb_used);  	return (phys_addr_t)DMA_MAPPING_ERROR;  found:  	io_tlb_used += nslots; @@ -562,7 +569,7 @@ found:  		io_tlb_orig_addr[index+i] = orig_addr + (i << IO_TLB_SHIFT);  	if (!(attrs & DMA_ATTR_SKIP_CPU_SYNC) &&  	    (dir == DMA_TO_DEVICE || dir == DMA_BIDIRECTIONAL)) -		swiotlb_bounce(orig_addr, tlb_addr, size, DMA_TO_DEVICE); +		swiotlb_bounce(orig_addr, tlb_addr, mapping_size, DMA_TO_DEVICE);  	return tlb_addr;  } @@ -571,11 +578,11 @@ found:   * tlb_addr is the physical address of the bounce buffer to unmap.   */  void swiotlb_tbl_unmap_single(struct device *hwdev, phys_addr_t tlb_addr, -			      size_t size, enum dma_data_direction dir, -			      unsigned long attrs) +			      size_t mapping_size, size_t alloc_size, +			      enum dma_data_direction dir, unsigned long attrs)  {  	unsigned long flags; -	int i, count, nslots = ALIGN(size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT; +	int i, count, nslots = ALIGN(alloc_size, 1 << IO_TLB_SHIFT) >> IO_TLB_SHIFT;  	int index = (tlb_addr - io_tlb_start) >> IO_TLB_SHIFT;  	phys_addr_t orig_addr = io_tlb_orig_addr[index]; @@ -585,7 +592,7 @@ void swiotlb_tbl_unmap_single(struct device *hwdev, phys_addr_t tlb_addr,  	if (orig_addr != INVALID_PHYS_ADDR &&  	    !(attrs & DMA_ATTR_SKIP_CPU_SYNC) &&  	    ((dir == DMA_FROM_DEVICE) || (dir == DMA_BIDIRECTIONAL))) -		swiotlb_bounce(orig_addr, tlb_addr, size, DMA_FROM_DEVICE); +		swiotlb_bounce(orig_addr, tlb_addr, mapping_size, DMA_FROM_DEVICE);  	/*  	 * Return the buffer to the free list by setting the corresponding @@ -665,14 +672,14 @@ bool swiotlb_map(struct device *dev, phys_addr_t *phys, dma_addr_t *dma_addr,  	/* Oh well, have to allocate and map a bounce buffer. */  	*phys = swiotlb_tbl_map_single(dev, __phys_to_dma(dev, io_tlb_start), -			*phys, size, dir, attrs); +			*phys, size, size, dir, attrs);  	if (*phys == (phys_addr_t)DMA_MAPPING_ERROR)  		return false;  	/* Ensure that the address returned is DMA'ble */  	*dma_addr = __phys_to_dma(dev, *phys);  	if (unlikely(!dma_capable(dev, *dma_addr, size))) { -		swiotlb_tbl_unmap_single(dev, *phys, size, dir, +		swiotlb_tbl_unmap_single(dev, *phys, size, size, dir,  			attrs | DMA_ATTR_SKIP_CPU_SYNC);  		return false;  	}  |