1 /* SPDX-License-Identifier: GPL-2.0 */
2 #ifndef __IO_PGTABLE_H
3 #define __IO_PGTABLE_H
4
5 #include <linux/bitops.h>
6 #include <linux/iommu.h>
7
8 /*
9 * Public API for use by IOMMU drivers
10 */
11 enum io_pgtable_fmt {
12 ARM_32_LPAE_S1,
13 ARM_32_LPAE_S2,
14 ARM_64_LPAE_S1,
15 ARM_64_LPAE_S2,
16 ARM_V7S,
17 ARM_MALI_LPAE,
18 AMD_IOMMU_V1,
19 AMD_IOMMU_V2,
20 APPLE_DART,
21 APPLE_DART2,
22 IO_PGTABLE_NUM_FMTS,
23 };
24
25 /**
26 * struct iommu_flush_ops - IOMMU callbacks for TLB and page table management.
27 *
28 * @tlb_flush_all: Synchronously invalidate the entire TLB context.
29 * @tlb_flush_walk: Synchronously invalidate all intermediate TLB state
30 * (sometimes referred to as the "walk cache") for a virtual
31 * address range.
32 * @tlb_add_page: Optional callback to queue up leaf TLB invalidation for a
33 * single page. IOMMUs that cannot batch TLB invalidation
34 * operations efficiently will typically issue them here, but
35 * others may decide to update the iommu_iotlb_gather structure
36 * and defer the invalidation until iommu_iotlb_sync() instead.
37 *
38 * Note that these can all be called in atomic context and must therefore
39 * not block.
40 */
41 struct iommu_flush_ops {
42 void (*tlb_flush_all)(void *cookie);
43 void (*tlb_flush_walk)(unsigned long iova, size_t size, size_t granule,
44 void *cookie);
45 void (*tlb_add_page)(struct iommu_iotlb_gather *gather,
46 unsigned long iova, size_t granule, void *cookie);
47 };
48
49 /**
50 * struct io_pgtable_cfg - Configuration data for a set of page tables.
51 *
52 * @quirks: A bitmap of hardware quirks that require some special
53 * action by the low-level page table allocator.
54 * @pgsize_bitmap: A bitmap of page sizes supported by this set of page
55 * tables.
56 * @ias: Input address (iova) size, in bits.
57 * @oas: Output address (paddr) size, in bits.
58 * @coherent_walk A flag to indicate whether or not page table walks made
59 * by the IOMMU are coherent with the CPU caches.
60 * @tlb: TLB management callbacks for this set of tables.
61 * @iommu_dev: The device representing the DMA configuration for the
62 * page table walker.
63 */
64 struct io_pgtable_cfg {
65 /*
66 * IO_PGTABLE_QUIRK_ARM_NS: (ARM formats) Set NS and NSTABLE bits in
67 * stage 1 PTEs, for hardware which insists on validating them
68 * even in non-secure state where they should normally be ignored.
69 *
70 * IO_PGTABLE_QUIRK_NO_PERMS: Ignore the IOMMU_READ, IOMMU_WRITE and
71 * IOMMU_NOEXEC flags and map everything with full access, for
72 * hardware which does not implement the permissions of a given
73 * format, and/or requires some format-specific default value.
74 *
75 * IO_PGTABLE_QUIRK_ARM_MTK_EXT: (ARM v7s format) MediaTek IOMMUs extend
76 * to support up to 35 bits PA where the bit32, bit33 and bit34 are
77 * encoded in the bit9, bit4 and bit5 of the PTE respectively.
78 *
79 * IO_PGTABLE_QUIRK_ARM_MTK_TTBR_EXT: (ARM v7s format) MediaTek IOMMUs
80 * extend the translation table base support up to 35 bits PA, the
81 * encoding format is same with IO_PGTABLE_QUIRK_ARM_MTK_EXT.
82 *
83 * IO_PGTABLE_QUIRK_ARM_TTBR1: (ARM LPAE format) Configure the table
84 * for use in the upper half of a split address space.
85 *
86 * IO_PGTABLE_QUIRK_ARM_OUTER_WBWA: Override the outer-cacheability
87 * attributes set in the TCR for a non-coherent page-table walker.
88 *
89 * IO_PGTABLE_QUIRK_ARM_HD: Enables dirty tracking in stage 1 pagetable.
90 * IO_PGTABLE_QUIRK_ARM_S2FWB: Use the FWB format for the MemAttrs bits
91 */
92 #define IO_PGTABLE_QUIRK_ARM_NS BIT(0)
93 #define IO_PGTABLE_QUIRK_NO_PERMS BIT(1)
94 #define IO_PGTABLE_QUIRK_ARM_MTK_EXT BIT(3)
95 #define IO_PGTABLE_QUIRK_ARM_MTK_TTBR_EXT BIT(4)
96 #define IO_PGTABLE_QUIRK_ARM_TTBR1 BIT(5)
97 #define IO_PGTABLE_QUIRK_ARM_OUTER_WBWA BIT(6)
98 #define IO_PGTABLE_QUIRK_ARM_HD BIT(7)
99 #define IO_PGTABLE_QUIRK_ARM_S2FWB BIT(8)
100 unsigned long quirks;
101 unsigned long pgsize_bitmap;
102 unsigned int ias;
103 unsigned int oas;
104 bool coherent_walk;
105 const struct iommu_flush_ops *tlb;
106 struct device *iommu_dev;
107
108 /**
109 * @alloc: Custom page allocator.
110 *
111 * Optional hook used to allocate page tables. If this function is NULL,
112 * @free must be NULL too.
113 *
114 * Memory returned should be zeroed and suitable for dma_map_single() and
115 * virt_to_phys().
116 *
117 * Not all formats support custom page allocators. Before considering
118 * passing a non-NULL value, make sure the chosen page format supports
119 * this feature.
120 */
121 void *(*alloc)(void *cookie, size_t size, gfp_t gfp);
122
123 /**
124 * @free: Custom page de-allocator.
125 *
126 * Optional hook used to free page tables allocated with the @alloc
127 * hook. Must be non-NULL if @alloc is not NULL, must be NULL
128 * otherwise.
129 */
130 void (*free)(void *cookie, void *pages, size_t size);
131
132 /* Low-level data specific to the table format */
133 union {
134 struct {
135 u64 ttbr;
136 struct {
137 u32 ips:3;
138 u32 tg:2;
139 u32 sh:2;
140 u32 orgn:2;
141 u32 irgn:2;
142 u32 tsz:6;
143 } tcr;
144 u64 mair;
145 } arm_lpae_s1_cfg;
146
147 struct {
148 u64 vttbr;
149 struct {
150 u32 ps:3;
151 u32 tg:2;
152 u32 sh:2;
153 u32 orgn:2;
154 u32 irgn:2;
155 u32 sl:2;
156 u32 tsz:6;
157 } vtcr;
158 } arm_lpae_s2_cfg;
159
160 struct {
161 u32 ttbr;
162 u32 tcr;
163 u32 nmrr;
164 u32 prrr;
165 } arm_v7s_cfg;
166
167 struct {
168 u64 transtab;
169 u64 memattr;
170 } arm_mali_lpae_cfg;
171
172 struct {
173 u64 ttbr[4];
174 u32 n_ttbrs;
175 } apple_dart_cfg;
176
177 struct {
178 int nid;
179 } amd;
180 };
181 };
182
183 /**
184 * struct arm_lpae_io_pgtable_walk_data - information from a pgtable walk
185 *
186 * @ptes: The recorded PTE values from the walk
187 */
188 struct arm_lpae_io_pgtable_walk_data {
189 u64 ptes[4];
190 };
191
192 /**
193 * struct io_pgtable_ops - Page table manipulation API for IOMMU drivers.
194 *
195 * @map_pages: Map a physically contiguous range of pages of the same size.
196 * @unmap_pages: Unmap a range of virtually contiguous pages of the same size.
197 * @iova_to_phys: Translate iova to physical address.
198 * @pgtable_walk: (optional) Perform a page table walk for a given iova.
199 *
200 * These functions map directly onto the iommu_ops member functions with
201 * the same names.
202 */
203 struct io_pgtable_ops {
204 int (*map_pages)(struct io_pgtable_ops *ops, unsigned long iova,
205 phys_addr_t paddr, size_t pgsize, size_t pgcount,
206 int prot, gfp_t gfp, size_t *mapped);
207 size_t (*unmap_pages)(struct io_pgtable_ops *ops, unsigned long iova,
208 size_t pgsize, size_t pgcount,
209 struct iommu_iotlb_gather *gather);
210 phys_addr_t (*iova_to_phys)(struct io_pgtable_ops *ops,
211 unsigned long iova);
212 int (*pgtable_walk)(struct io_pgtable_ops *ops, unsigned long iova, void *wd);
213 int (*read_and_clear_dirty)(struct io_pgtable_ops *ops,
214 unsigned long iova, size_t size,
215 unsigned long flags,
216 struct iommu_dirty_bitmap *dirty);
217 };
218
219 /**
220 * alloc_io_pgtable_ops() - Allocate a page table allocator for use by an IOMMU.
221 *
222 * @fmt: The page table format.
223 * @cfg: The page table configuration. This will be modified to represent
224 * the configuration actually provided by the allocator (e.g. the
225 * pgsize_bitmap may be restricted).
226 * @cookie: An opaque token provided by the IOMMU driver and passed back to
227 * the callback routines in cfg->tlb.
228 */
229 struct io_pgtable_ops *alloc_io_pgtable_ops(enum io_pgtable_fmt fmt,
230 struct io_pgtable_cfg *cfg,
231 void *cookie);
232
233 /**
234 * free_io_pgtable_ops() - Free an io_pgtable_ops structure. The caller
235 * *must* ensure that the page table is no longer
236 * live, but the TLB can be dirty.
237 *
238 * @ops: The ops returned from alloc_io_pgtable_ops.
239 */
240 void free_io_pgtable_ops(struct io_pgtable_ops *ops);
241
242
243 /*
244 * Internal structures for page table allocator implementations.
245 */
246
247 /**
248 * struct io_pgtable - Internal structure describing a set of page tables.
249 *
250 * @fmt: The page table format.
251 * @cookie: An opaque token provided by the IOMMU driver and passed back to
252 * any callback routines.
253 * @cfg: A copy of the page table configuration.
254 * @ops: The page table operations in use for this set of page tables.
255 */
256 struct io_pgtable {
257 enum io_pgtable_fmt fmt;
258 void *cookie;
259 struct io_pgtable_cfg cfg;
260 struct io_pgtable_ops ops;
261 };
262
263 #define io_pgtable_ops_to_pgtable(x) container_of((x), struct io_pgtable, ops)
264
io_pgtable_tlb_flush_all(struct io_pgtable * iop)265 static inline void io_pgtable_tlb_flush_all(struct io_pgtable *iop)
266 {
267 if (iop->cfg.tlb && iop->cfg.tlb->tlb_flush_all)
268 iop->cfg.tlb->tlb_flush_all(iop->cookie);
269 }
270
271 static inline void
io_pgtable_tlb_flush_walk(struct io_pgtable * iop,unsigned long iova,size_t size,size_t granule)272 io_pgtable_tlb_flush_walk(struct io_pgtable *iop, unsigned long iova,
273 size_t size, size_t granule)
274 {
275 if (iop->cfg.tlb && iop->cfg.tlb->tlb_flush_walk)
276 iop->cfg.tlb->tlb_flush_walk(iova, size, granule, iop->cookie);
277 }
278
279 static inline void
io_pgtable_tlb_add_page(struct io_pgtable * iop,struct iommu_iotlb_gather * gather,unsigned long iova,size_t granule)280 io_pgtable_tlb_add_page(struct io_pgtable *iop,
281 struct iommu_iotlb_gather * gather, unsigned long iova,
282 size_t granule)
283 {
284 if (iop->cfg.tlb && iop->cfg.tlb->tlb_add_page)
285 iop->cfg.tlb->tlb_add_page(gather, iova, granule, iop->cookie);
286 }
287
288 /**
289 * enum io_pgtable_caps - IO page table backend capabilities.
290 */
291 enum io_pgtable_caps {
292 /** @IO_PGTABLE_CAP_CUSTOM_ALLOCATOR: Backend accepts custom page table allocators. */
293 IO_PGTABLE_CAP_CUSTOM_ALLOCATOR = BIT(0),
294 };
295
296 /**
297 * struct io_pgtable_init_fns - Alloc/free a set of page tables for a
298 * particular format.
299 *
300 * @alloc: Allocate a set of page tables described by cfg.
301 * @free: Free the page tables associated with iop.
302 * @caps: Combination of @io_pgtable_caps flags encoding the backend capabilities.
303 */
304 struct io_pgtable_init_fns {
305 struct io_pgtable *(*alloc)(struct io_pgtable_cfg *cfg, void *cookie);
306 void (*free)(struct io_pgtable *iop);
307 u32 caps;
308 };
309
310 extern struct io_pgtable_init_fns io_pgtable_arm_32_lpae_s1_init_fns;
311 extern struct io_pgtable_init_fns io_pgtable_arm_32_lpae_s2_init_fns;
312 extern struct io_pgtable_init_fns io_pgtable_arm_64_lpae_s1_init_fns;
313 extern struct io_pgtable_init_fns io_pgtable_arm_64_lpae_s2_init_fns;
314 extern struct io_pgtable_init_fns io_pgtable_arm_v7s_init_fns;
315 extern struct io_pgtable_init_fns io_pgtable_arm_mali_lpae_init_fns;
316 extern struct io_pgtable_init_fns io_pgtable_amd_iommu_v1_init_fns;
317 extern struct io_pgtable_init_fns io_pgtable_amd_iommu_v2_init_fns;
318 extern struct io_pgtable_init_fns io_pgtable_apple_dart_init_fns;
319
320 #endif /* __IO_PGTABLE_H */
321