1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * IOMMU API for ARM architected SMMU implementations.
4 *
5 * Copyright (C) 2013 ARM Limited
6 *
7 * Author: Will Deacon <will.deacon@arm.com>
8 */
9
10#ifndef _ARM_SMMU_H
11#define _ARM_SMMU_H
12
13#include <linux/atomic.h>
14#include <linux/bitfield.h>
15#include <linux/bits.h>
16#include <linux/clk.h>
17#include <linux/device.h>
18#include <linux/io-64-nonatomic-hi-lo.h>
19#include <linux/io-pgtable.h>
20#include <linux/iommu.h>
21#include <linux/irqreturn.h>
22#include <linux/mutex.h>
23#include <linux/spinlock.h>
24#include <linux/types.h>
25
26/* Configuration registers */
27#define ARM_SMMU_GR0_sCR0		0x0
28#define ARM_SMMU_sCR0_VMID16EN		BIT(31)
29#define ARM_SMMU_sCR0_BSU		GENMASK(15, 14)
30#define ARM_SMMU_sCR0_FB		BIT(13)
31#define ARM_SMMU_sCR0_PTM		BIT(12)
32#define ARM_SMMU_sCR0_VMIDPNE		BIT(11)
33#define ARM_SMMU_sCR0_USFCFG		BIT(10)
34#define ARM_SMMU_sCR0_GCFGFIE		BIT(5)
35#define ARM_SMMU_sCR0_GCFGFRE		BIT(4)
36#define ARM_SMMU_sCR0_EXIDENABLE	BIT(3)
37#define ARM_SMMU_sCR0_GFIE		BIT(2)
38#define ARM_SMMU_sCR0_GFRE		BIT(1)
39#define ARM_SMMU_sCR0_CLIENTPD		BIT(0)
40
41/* Auxiliary Configuration register */
42#define ARM_SMMU_GR0_sACR		0x10
43
44/* Identification registers */
45#define ARM_SMMU_GR0_ID0		0x20
46#define ARM_SMMU_ID0_S1TS		BIT(30)
47#define ARM_SMMU_ID0_S2TS		BIT(29)
48#define ARM_SMMU_ID0_NTS		BIT(28)
49#define ARM_SMMU_ID0_SMS		BIT(27)
50#define ARM_SMMU_ID0_ATOSNS		BIT(26)
51#define ARM_SMMU_ID0_PTFS_NO_AARCH32	BIT(25)
52#define ARM_SMMU_ID0_PTFS_NO_AARCH32S	BIT(24)
53#define ARM_SMMU_ID0_NUMIRPT		GENMASK(23, 16)
54#define ARM_SMMU_ID0_CTTW		BIT(14)
55#define ARM_SMMU_ID0_NUMSIDB		GENMASK(12, 9)
56#define ARM_SMMU_ID0_EXIDS		BIT(8)
57#define ARM_SMMU_ID0_NUMSMRG		GENMASK(7, 0)
58
59#define ARM_SMMU_GR0_ID1		0x24
60#define ARM_SMMU_ID1_PAGESIZE		BIT(31)
61#define ARM_SMMU_ID1_NUMPAGENDXB	GENMASK(30, 28)
62#define ARM_SMMU_ID1_NUMS2CB		GENMASK(23, 16)
63#define ARM_SMMU_ID1_NUMCB		GENMASK(7, 0)
64
65#define ARM_SMMU_GR0_ID2		0x28
66#define ARM_SMMU_ID2_VMID16		BIT(15)
67#define ARM_SMMU_ID2_PTFS_64K		BIT(14)
68#define ARM_SMMU_ID2_PTFS_16K		BIT(13)
69#define ARM_SMMU_ID2_PTFS_4K		BIT(12)
70#define ARM_SMMU_ID2_UBS		GENMASK(11, 8)
71#define ARM_SMMU_ID2_OAS		GENMASK(7, 4)
72#define ARM_SMMU_ID2_IAS		GENMASK(3, 0)
73
74#define ARM_SMMU_GR0_ID3		0x2c
75#define ARM_SMMU_GR0_ID4		0x30
76#define ARM_SMMU_GR0_ID5		0x34
77#define ARM_SMMU_GR0_ID6		0x38
78
79#define ARM_SMMU_GR0_ID7		0x3c
80#define ARM_SMMU_ID7_MAJOR		GENMASK(7, 4)
81#define ARM_SMMU_ID7_MINOR		GENMASK(3, 0)
82
83#define ARM_SMMU_GR0_sGFSR		0x48
84#define ARM_SMMU_sGFSR_USF		BIT(1)
85
86#define ARM_SMMU_GR0_sGFSYNR0		0x50
87#define ARM_SMMU_GR0_sGFSYNR1		0x54
88#define ARM_SMMU_GR0_sGFSYNR2		0x58
89
90/* Global TLB invalidation */
91#define ARM_SMMU_GR0_TLBIVMID		0x64
92#define ARM_SMMU_GR0_TLBIALLNSNH	0x68
93#define ARM_SMMU_GR0_TLBIALLH		0x6c
94#define ARM_SMMU_GR0_sTLBGSYNC		0x70
95
96#define ARM_SMMU_GR0_sTLBGSTATUS	0x74
97#define ARM_SMMU_sTLBGSTATUS_GSACTIVE	BIT(0)
98
99/* Stream mapping registers */
100#define ARM_SMMU_GR0_SMR(n)		(0x800 + ((n) << 2))
101#define ARM_SMMU_SMR_VALID		BIT(31)
102#define ARM_SMMU_SMR_MASK		GENMASK(31, 16)
103#define ARM_SMMU_SMR_ID			GENMASK(15, 0)
104
105#define ARM_SMMU_GR0_S2CR(n)		(0xc00 + ((n) << 2))
106#define ARM_SMMU_S2CR_PRIVCFG		GENMASK(25, 24)
107enum arm_smmu_s2cr_privcfg {
108	S2CR_PRIVCFG_DEFAULT,
109	S2CR_PRIVCFG_DIPAN,
110	S2CR_PRIVCFG_UNPRIV,
111	S2CR_PRIVCFG_PRIV,
112};
113#define ARM_SMMU_S2CR_TYPE		GENMASK(17, 16)
114enum arm_smmu_s2cr_type {
115	S2CR_TYPE_TRANS,
116	S2CR_TYPE_BYPASS,
117	S2CR_TYPE_FAULT,
118};
119#define ARM_SMMU_S2CR_EXIDVALID		BIT(10)
120#define ARM_SMMU_S2CR_CBNDX		GENMASK(7, 0)
121
122/* Context bank attribute registers */
123#define ARM_SMMU_GR1_CBAR(n)		(0x0 + ((n) << 2))
124#define ARM_SMMU_CBAR_IRPTNDX		GENMASK(31, 24)
125#define ARM_SMMU_CBAR_TYPE		GENMASK(17, 16)
126enum arm_smmu_cbar_type {
127	CBAR_TYPE_S2_TRANS,
128	CBAR_TYPE_S1_TRANS_S2_BYPASS,
129	CBAR_TYPE_S1_TRANS_S2_FAULT,
130	CBAR_TYPE_S1_TRANS_S2_TRANS,
131};
132#define ARM_SMMU_CBAR_S1_MEMATTR	GENMASK(15, 12)
133#define ARM_SMMU_CBAR_S1_MEMATTR_WB	0xf
134#define ARM_SMMU_CBAR_S1_BPSHCFG	GENMASK(9, 8)
135#define ARM_SMMU_CBAR_S1_BPSHCFG_NSH	3
136#define ARM_SMMU_CBAR_VMID		GENMASK(7, 0)
137
138#define ARM_SMMU_GR1_CBFRSYNRA(n)	(0x400 + ((n) << 2))
139
140#define ARM_SMMU_GR1_CBA2R(n)		(0x800 + ((n) << 2))
141#define ARM_SMMU_CBA2R_VMID16		GENMASK(31, 16)
142#define ARM_SMMU_CBA2R_VA64		BIT(0)
143
144#define ARM_SMMU_CB_SCTLR		0x0
145#define ARM_SMMU_SCTLR_S1_ASIDPNE	BIT(12)
146#define ARM_SMMU_SCTLR_CFCFG		BIT(7)
147#define ARM_SMMU_SCTLR_HUPCF		BIT(8)
148#define ARM_SMMU_SCTLR_CFIE		BIT(6)
149#define ARM_SMMU_SCTLR_CFRE		BIT(5)
150#define ARM_SMMU_SCTLR_E		BIT(4)
151#define ARM_SMMU_SCTLR_AFE		BIT(2)
152#define ARM_SMMU_SCTLR_TRE		BIT(1)
153#define ARM_SMMU_SCTLR_M		BIT(0)
154
155#define ARM_SMMU_CB_ACTLR		0x4
156
157#define ARM_SMMU_CB_RESUME		0x8
158#define ARM_SMMU_RESUME_TERMINATE	BIT(0)
159
160#define ARM_SMMU_CB_TCR2		0x10
161#define ARM_SMMU_TCR2_SEP		GENMASK(17, 15)
162#define ARM_SMMU_TCR2_SEP_UPSTREAM	0x7
163#define ARM_SMMU_TCR2_AS		BIT(4)
164#define ARM_SMMU_TCR2_PASIZE		GENMASK(3, 0)
165
166#define ARM_SMMU_CB_TTBR0		0x20
167#define ARM_SMMU_CB_TTBR1		0x28
168#define ARM_SMMU_TTBRn_ASID		GENMASK_ULL(63, 48)
169
170#define ARM_SMMU_CB_TCR			0x30
171#define ARM_SMMU_TCR_EAE		BIT(31)
172#define ARM_SMMU_TCR_EPD1		BIT(23)
173#define ARM_SMMU_TCR_A1			BIT(22)
174#define ARM_SMMU_TCR_TG0		GENMASK(15, 14)
175#define ARM_SMMU_TCR_SH0		GENMASK(13, 12)
176#define ARM_SMMU_TCR_ORGN0		GENMASK(11, 10)
177#define ARM_SMMU_TCR_IRGN0		GENMASK(9, 8)
178#define ARM_SMMU_TCR_EPD0		BIT(7)
179#define ARM_SMMU_TCR_T0SZ		GENMASK(5, 0)
180
181#define ARM_SMMU_VTCR_RES1		BIT(31)
182#define ARM_SMMU_VTCR_PS		GENMASK(18, 16)
183#define ARM_SMMU_VTCR_TG0		ARM_SMMU_TCR_TG0
184#define ARM_SMMU_VTCR_SH0		ARM_SMMU_TCR_SH0
185#define ARM_SMMU_VTCR_ORGN0		ARM_SMMU_TCR_ORGN0
186#define ARM_SMMU_VTCR_IRGN0		ARM_SMMU_TCR_IRGN0
187#define ARM_SMMU_VTCR_SL0		GENMASK(7, 6)
188#define ARM_SMMU_VTCR_T0SZ		ARM_SMMU_TCR_T0SZ
189
190#define ARM_SMMU_CB_CONTEXTIDR		0x34
191#define ARM_SMMU_CB_S1_MAIR0		0x38
192#define ARM_SMMU_CB_S1_MAIR1		0x3c
193
194#define ARM_SMMU_CB_PAR			0x50
195#define ARM_SMMU_CB_PAR_F		BIT(0)
196
197#define ARM_SMMU_CB_FSR			0x58
198#define ARM_SMMU_FSR_MULTI		BIT(31)
199#define ARM_SMMU_FSR_SS			BIT(30)
200#define ARM_SMMU_FSR_UUT		BIT(8)
201#define ARM_SMMU_FSR_ASF		BIT(7)
202#define ARM_SMMU_FSR_TLBLKF		BIT(6)
203#define ARM_SMMU_FSR_TLBMCF		BIT(5)
204#define ARM_SMMU_FSR_EF			BIT(4)
205#define ARM_SMMU_FSR_PF			BIT(3)
206#define ARM_SMMU_FSR_AFF		BIT(2)
207#define ARM_SMMU_FSR_TF			BIT(1)
208
209#define ARM_SMMU_FSR_IGN		(ARM_SMMU_FSR_AFF |		\
210					 ARM_SMMU_FSR_ASF |		\
211					 ARM_SMMU_FSR_TLBMCF |		\
212					 ARM_SMMU_FSR_TLBLKF)
213
214#define ARM_SMMU_FSR_FAULT		(ARM_SMMU_FSR_MULTI |		\
215					 ARM_SMMU_FSR_SS |		\
216					 ARM_SMMU_FSR_UUT |		\
217					 ARM_SMMU_FSR_EF |		\
218					 ARM_SMMU_FSR_PF |		\
219					 ARM_SMMU_FSR_TF |		\
220					 ARM_SMMU_FSR_IGN)
221
222#define ARM_SMMU_CB_FAR			0x60
223
224#define ARM_SMMU_CB_FSYNR0		0x68
225#define ARM_SMMU_FSYNR0_WNR		BIT(4)
226
227#define ARM_SMMU_CB_FSYNR1		0x6c
228
229#define ARM_SMMU_CB_S1_TLBIVA		0x600
230#define ARM_SMMU_CB_S1_TLBIASID		0x610
231#define ARM_SMMU_CB_S1_TLBIVAL		0x620
232#define ARM_SMMU_CB_S2_TLBIIPAS2	0x630
233#define ARM_SMMU_CB_S2_TLBIIPAS2L	0x638
234#define ARM_SMMU_CB_TLBSYNC		0x7f0
235#define ARM_SMMU_CB_TLBSTATUS		0x7f4
236#define ARM_SMMU_CB_ATS1PR		0x800
237
238#define ARM_SMMU_CB_ATSR		0x8f0
239#define ARM_SMMU_ATSR_ACTIVE		BIT(0)
240
241
242/* Maximum number of context banks per SMMU */
243#define ARM_SMMU_MAX_CBS		128
244
245#define TLB_LOOP_TIMEOUT		1000000	/* 1s! */
246#define TLB_SPIN_COUNT			10
247
248/* Shared driver definitions */
249enum arm_smmu_arch_version {
250	ARM_SMMU_V1,
251	ARM_SMMU_V1_64K,
252	ARM_SMMU_V2,
253};
254
255enum arm_smmu_implementation {
256	GENERIC_SMMU,
257	ARM_MMU500,
258	CAVIUM_SMMUV2,
259	QCOM_SMMUV2,
260};
261
262struct arm_smmu_s2cr {
263	struct iommu_group		*group;
264	int				count;
265	enum arm_smmu_s2cr_type		type;
266	enum arm_smmu_s2cr_privcfg	privcfg;
267	u8				cbndx;
268};
269
270struct arm_smmu_smr {
271	u16				mask;
272	u16				id;
273	bool				valid;
274	bool				pinned;
275};
276
277struct arm_smmu_device {
278	struct device			*dev;
279
280	void __iomem			*base;
281	phys_addr_t			ioaddr;
282	unsigned int			numpage;
283	unsigned int			pgshift;
284
285#define ARM_SMMU_FEAT_COHERENT_WALK	(1 << 0)
286#define ARM_SMMU_FEAT_STREAM_MATCH	(1 << 1)
287#define ARM_SMMU_FEAT_TRANS_S1		(1 << 2)
288#define ARM_SMMU_FEAT_TRANS_S2		(1 << 3)
289#define ARM_SMMU_FEAT_TRANS_NESTED	(1 << 4)
290#define ARM_SMMU_FEAT_TRANS_OPS		(1 << 5)
291#define ARM_SMMU_FEAT_VMID16		(1 << 6)
292#define ARM_SMMU_FEAT_FMT_AARCH64_4K	(1 << 7)
293#define ARM_SMMU_FEAT_FMT_AARCH64_16K	(1 << 8)
294#define ARM_SMMU_FEAT_FMT_AARCH64_64K	(1 << 9)
295#define ARM_SMMU_FEAT_FMT_AARCH32_L	(1 << 10)
296#define ARM_SMMU_FEAT_FMT_AARCH32_S	(1 << 11)
297#define ARM_SMMU_FEAT_EXIDS		(1 << 12)
298	u32				features;
299
300	enum arm_smmu_arch_version	version;
301	enum arm_smmu_implementation	model;
302	const struct arm_smmu_impl	*impl;
303
304	u32				num_context_banks;
305	u32				num_s2_context_banks;
306	DECLARE_BITMAP(context_map, ARM_SMMU_MAX_CBS);
307	struct arm_smmu_cb		*cbs;
308	atomic_t			irptndx;
309
310	u32				num_mapping_groups;
311	u16				streamid_mask;
312	u16				smr_mask_mask;
313	struct arm_smmu_smr		*smrs;
314	struct arm_smmu_s2cr		*s2crs;
315	struct mutex			stream_map_mutex;
316
317	unsigned long			va_size;
318	unsigned long			ipa_size;
319	unsigned long			pa_size;
320	unsigned long			pgsize_bitmap;
321
322	int				num_context_irqs;
323	int				num_clks;
324	unsigned int			*irqs;
325	struct clk_bulk_data		*clks;
326
327	spinlock_t			global_sync_lock;
328
329	/* IOMMU core code handle */
330	struct iommu_device		iommu;
331};
332
333enum arm_smmu_context_fmt {
334	ARM_SMMU_CTX_FMT_NONE,
335	ARM_SMMU_CTX_FMT_AARCH64,
336	ARM_SMMU_CTX_FMT_AARCH32_L,
337	ARM_SMMU_CTX_FMT_AARCH32_S,
338};
339
340struct arm_smmu_cfg {
341	u8				cbndx;
342	u8				irptndx;
343	union {
344		u16			asid;
345		u16			vmid;
346	};
347	enum arm_smmu_cbar_type		cbar;
348	enum arm_smmu_context_fmt	fmt;
349	bool				flush_walk_prefer_tlbiasid;
350};
351#define ARM_SMMU_INVALID_IRPTNDX	0xff
352
353struct arm_smmu_cb {
354	u64				ttbr[2];
355	u32				tcr[2];
356	u32				mair[2];
357	struct arm_smmu_cfg		*cfg;
358};
359
360enum arm_smmu_domain_stage {
361	ARM_SMMU_DOMAIN_S1 = 0,
362	ARM_SMMU_DOMAIN_S2,
363	ARM_SMMU_DOMAIN_NESTED,
364};
365
366struct arm_smmu_domain {
367	struct arm_smmu_device		*smmu;
368	struct io_pgtable_ops		*pgtbl_ops;
369	unsigned long			pgtbl_quirks;
370	const struct iommu_flush_ops	*flush_ops;
371	struct arm_smmu_cfg		cfg;
372	enum arm_smmu_domain_stage	stage;
373	struct mutex			init_mutex; /* Protects smmu pointer */
374	spinlock_t			cb_lock; /* Serialises ATS1* ops and TLB syncs */
375	struct iommu_domain		domain;
376};
377
378struct arm_smmu_master_cfg {
379	struct arm_smmu_device		*smmu;
380	s16				smendx[];
381};
382
383static inline u32 arm_smmu_lpae_tcr(const struct io_pgtable_cfg *cfg)
384{
385	u32 tcr = FIELD_PREP(ARM_SMMU_TCR_TG0, cfg->arm_lpae_s1_cfg.tcr.tg) |
386		FIELD_PREP(ARM_SMMU_TCR_SH0, cfg->arm_lpae_s1_cfg.tcr.sh) |
387		FIELD_PREP(ARM_SMMU_TCR_ORGN0, cfg->arm_lpae_s1_cfg.tcr.orgn) |
388		FIELD_PREP(ARM_SMMU_TCR_IRGN0, cfg->arm_lpae_s1_cfg.tcr.irgn) |
389		FIELD_PREP(ARM_SMMU_TCR_T0SZ, cfg->arm_lpae_s1_cfg.tcr.tsz);
390
391       /*
392	* When TTBR1 is selected shift the TCR fields by 16 bits and disable
393	* translation in TTBR0
394	*/
395	if (cfg->quirks & IO_PGTABLE_QUIRK_ARM_TTBR1) {
396		tcr = (tcr << 16) & ~ARM_SMMU_TCR_A1;
397		tcr |= ARM_SMMU_TCR_EPD0;
398	} else
399		tcr |= ARM_SMMU_TCR_EPD1;
400
401	return tcr;
402}
403
404static inline u32 arm_smmu_lpae_tcr2(const struct io_pgtable_cfg *cfg)
405{
406	return FIELD_PREP(ARM_SMMU_TCR2_PASIZE, cfg->arm_lpae_s1_cfg.tcr.ips) |
407	       FIELD_PREP(ARM_SMMU_TCR2_SEP, ARM_SMMU_TCR2_SEP_UPSTREAM);
408}
409
410static inline u32 arm_smmu_lpae_vtcr(const struct io_pgtable_cfg *cfg)
411{
412	return ARM_SMMU_VTCR_RES1 |
413	       FIELD_PREP(ARM_SMMU_VTCR_PS, cfg->arm_lpae_s2_cfg.vtcr.ps) |
414	       FIELD_PREP(ARM_SMMU_VTCR_TG0, cfg->arm_lpae_s2_cfg.vtcr.tg) |
415	       FIELD_PREP(ARM_SMMU_VTCR_SH0, cfg->arm_lpae_s2_cfg.vtcr.sh) |
416	       FIELD_PREP(ARM_SMMU_VTCR_ORGN0, cfg->arm_lpae_s2_cfg.vtcr.orgn) |
417	       FIELD_PREP(ARM_SMMU_VTCR_IRGN0, cfg->arm_lpae_s2_cfg.vtcr.irgn) |
418	       FIELD_PREP(ARM_SMMU_VTCR_SL0, cfg->arm_lpae_s2_cfg.vtcr.sl) |
419	       FIELD_PREP(ARM_SMMU_VTCR_T0SZ, cfg->arm_lpae_s2_cfg.vtcr.tsz);
420}
421
422/* Implementation details, yay! */
423struct arm_smmu_impl {
424	u32 (*read_reg)(struct arm_smmu_device *smmu, int page, int offset);
425	void (*write_reg)(struct arm_smmu_device *smmu, int page, int offset,
426			  u32 val);
427	u64 (*read_reg64)(struct arm_smmu_device *smmu, int page, int offset);
428	void (*write_reg64)(struct arm_smmu_device *smmu, int page, int offset,
429			    u64 val);
430	int (*cfg_probe)(struct arm_smmu_device *smmu);
431	int (*reset)(struct arm_smmu_device *smmu);
432	int (*init_context)(struct arm_smmu_domain *smmu_domain,
433			struct io_pgtable_cfg *cfg, struct device *dev);
434	void (*tlb_sync)(struct arm_smmu_device *smmu, int page, int sync,
435			 int status);
436	int (*def_domain_type)(struct device *dev);
437	irqreturn_t (*global_fault)(int irq, void *dev);
438	irqreturn_t (*context_fault)(int irq, void *dev);
439	int (*alloc_context_bank)(struct arm_smmu_domain *smmu_domain,
440				  struct arm_smmu_device *smmu,
441				  struct device *dev, int start);
442	void (*write_s2cr)(struct arm_smmu_device *smmu, int idx);
443	void (*write_sctlr)(struct arm_smmu_device *smmu, int idx, u32 reg);
444	void (*probe_finalize)(struct arm_smmu_device *smmu, struct device *dev);
445};
446
447#define INVALID_SMENDX			-1
448#define cfg_smendx(cfg, fw, i) \
449	(i >= fw->num_ids ? INVALID_SMENDX : cfg->smendx[i])
450#define for_each_cfg_sme(cfg, fw, i, idx) \
451	for (i = 0; idx = cfg_smendx(cfg, fw, i), i < fw->num_ids; ++i)
452
453static inline int __arm_smmu_alloc_bitmap(unsigned long *map, int start, int end)
454{
455	int idx;
456
457	do {
458		idx = find_next_zero_bit(map, end, start);
459		if (idx == end)
460			return -ENOSPC;
461	} while (test_and_set_bit(idx, map));
462
463	return idx;
464}
465
466static inline void __iomem *arm_smmu_page(struct arm_smmu_device *smmu, int n)
467{
468	return smmu->base + (n << smmu->pgshift);
469}
470
471static inline u32 arm_smmu_readl(struct arm_smmu_device *smmu, int page, int offset)
472{
473	if (smmu->impl && unlikely(smmu->impl->read_reg))
474		return smmu->impl->read_reg(smmu, page, offset);
475	return readl_relaxed(arm_smmu_page(smmu, page) + offset);
476}
477
478static inline void arm_smmu_writel(struct arm_smmu_device *smmu, int page,
479				   int offset, u32 val)
480{
481	if (smmu->impl && unlikely(smmu->impl->write_reg))
482		smmu->impl->write_reg(smmu, page, offset, val);
483	else
484		writel_relaxed(val, arm_smmu_page(smmu, page) + offset);
485}
486
487static inline u64 arm_smmu_readq(struct arm_smmu_device *smmu, int page, int offset)
488{
489	if (smmu->impl && unlikely(smmu->impl->read_reg64))
490		return smmu->impl->read_reg64(smmu, page, offset);
491	return readq_relaxed(arm_smmu_page(smmu, page) + offset);
492}
493
494static inline void arm_smmu_writeq(struct arm_smmu_device *smmu, int page,
495				   int offset, u64 val)
496{
497	if (smmu->impl && unlikely(smmu->impl->write_reg64))
498		smmu->impl->write_reg64(smmu, page, offset, val);
499	else
500		writeq_relaxed(val, arm_smmu_page(smmu, page) + offset);
501}
502
503#define ARM_SMMU_GR0		0
504#define ARM_SMMU_GR1		1
505#define ARM_SMMU_CB(s, n)	((s)->numpage + (n))
506
507#define arm_smmu_gr0_read(s, o)		\
508	arm_smmu_readl((s), ARM_SMMU_GR0, (o))
509#define arm_smmu_gr0_write(s, o, v)	\
510	arm_smmu_writel((s), ARM_SMMU_GR0, (o), (v))
511
512#define arm_smmu_gr1_read(s, o)		\
513	arm_smmu_readl((s), ARM_SMMU_GR1, (o))
514#define arm_smmu_gr1_write(s, o, v)	\
515	arm_smmu_writel((s), ARM_SMMU_GR1, (o), (v))
516
517#define arm_smmu_cb_read(s, n, o)	\
518	arm_smmu_readl((s), ARM_SMMU_CB((s), (n)), (o))
519#define arm_smmu_cb_write(s, n, o, v)	\
520	arm_smmu_writel((s), ARM_SMMU_CB((s), (n)), (o), (v))
521#define arm_smmu_cb_readq(s, n, o)	\
522	arm_smmu_readq((s), ARM_SMMU_CB((s), (n)), (o))
523#define arm_smmu_cb_writeq(s, n, o, v)	\
524	arm_smmu_writeq((s), ARM_SMMU_CB((s), (n)), (o), (v))
525
526struct arm_smmu_device *arm_smmu_impl_init(struct arm_smmu_device *smmu);
527struct arm_smmu_device *nvidia_smmu_impl_init(struct arm_smmu_device *smmu);
528struct arm_smmu_device *qcom_smmu_impl_init(struct arm_smmu_device *smmu);
529
530void arm_smmu_write_context_bank(struct arm_smmu_device *smmu, int idx);
531int arm_mmu500_reset(struct arm_smmu_device *smmu);
532
533#endif /* _ARM_SMMU_H */
534