1 /* SPDX-License-Identifier: GPL-2.0-only */ 2 /* 3 * MMU-based software IOTLB. 4 * 5 * Copyright (C) 2020-2021 Bytedance Inc. and/or its affiliates. All rights reserved. 6 * 7 * Author: Xie Yongji <xieyongji@bytedance.com> 8 * 9 */ 10 11 #ifndef _VDUSE_IOVA_DOMAIN_H 12 #define _VDUSE_IOVA_DOMAIN_H 13 14 #include <linux/iova.h> 15 #include <linux/dma-mapping.h> 16 #include <linux/vhost_iotlb.h> 17 18 #define IOVA_START_PFN 1 19 20 #define INVALID_PHYS_ADDR (~(phys_addr_t)0) 21 22 #define BOUNCE_MAP_SHIFT 12 23 #define BOUNCE_MAP_SIZE (1 << BOUNCE_MAP_SHIFT) 24 #define BOUNCE_MAP_MASK (~(BOUNCE_MAP_SIZE - 1)) 25 #define BOUNCE_MAP_ALIGN(addr) (((addr) + BOUNCE_MAP_SIZE - 1) & ~(BOUNCE_MAP_SIZE - 1)) 26 27 struct vduse_bounce_map { 28 struct page *bounce_page; 29 struct page *user_bounce_page; 30 u64 orig_phys; 31 }; 32 33 struct vduse_iova_domain { 34 struct iova_domain stream_iovad; 35 struct iova_domain consistent_iovad; 36 struct vduse_bounce_map *bounce_maps; 37 size_t bounce_size; 38 unsigned long iova_limit; 39 int bounce_map; 40 struct vhost_iotlb *iotlb; 41 spinlock_t iotlb_lock; 42 struct file *file; 43 bool user_bounce_pages; 44 rwlock_t bounce_lock; 45 }; 46 47 int vduse_domain_set_map(struct vduse_iova_domain *domain, 48 struct vhost_iotlb *iotlb); 49 50 void vduse_domain_clear_map(struct vduse_iova_domain *domain, 51 struct vhost_iotlb *iotlb); 52 53 void vduse_domain_sync_single_for_device(struct vduse_iova_domain *domain, 54 dma_addr_t dma_addr, size_t size, 55 enum dma_data_direction dir); 56 57 void vduse_domain_sync_single_for_cpu(struct vduse_iova_domain *domain, 58 dma_addr_t dma_addr, size_t size, 59 enum dma_data_direction dir); 60 61 dma_addr_t vduse_domain_map_page(struct vduse_iova_domain *domain, 62 struct page *page, unsigned long offset, 63 size_t size, enum dma_data_direction dir, 64 unsigned long attrs); 65 66 void vduse_domain_unmap_page(struct vduse_iova_domain *domain, 67 dma_addr_t dma_addr, size_t size, 68 enum dma_data_direction dir, unsigned long attrs); 69 70 void *vduse_domain_alloc_coherent(struct vduse_iova_domain *domain, 71 size_t size, dma_addr_t *dma_addr, 72 gfp_t flag); 73 74 void vduse_domain_free_coherent(struct vduse_iova_domain *domain, size_t size, 75 void *vaddr, dma_addr_t dma_addr, 76 unsigned long attrs); 77 78 void vduse_domain_reset_bounce_map(struct vduse_iova_domain *domain); 79 80 int vduse_domain_add_user_bounce_pages(struct vduse_iova_domain *domain, 81 struct page **pages, int count); 82 83 void vduse_domain_remove_user_bounce_pages(struct vduse_iova_domain *domain); 84 85 void vduse_domain_destroy(struct vduse_iova_domain *domain); 86 87 struct vduse_iova_domain *vduse_domain_create(unsigned long iova_limit, 88 size_t bounce_size); 89 90 int vduse_domain_init(void); 91 92 void vduse_domain_exit(void); 93 94 #endif /* _VDUSE_IOVA_DOMAIN_H */ 95