1// SPDX-License-Identifier: GPL-2.0-only
2/*
3 * Copyright (C) 2007-2010 Advanced Micro Devices, Inc.
4 * Author: Joerg Roedel <jroedel@suse.de>
5 *         Leo Duran <leo.duran@amd.com>
6 */
7
8#define pr_fmt(fmt)     "AMD-Vi: " fmt
9#define dev_fmt(fmt)    pr_fmt(fmt)
10
11#include <linux/pci.h>
12#include <linux/acpi.h>
13#include <linux/list.h>
14#include <linux/bitmap.h>
15#include <linux/slab.h>
16#include <linux/syscore_ops.h>
17#include <linux/interrupt.h>
18#include <linux/msi.h>
19#include <linux/irq.h>
20#include <linux/amd-iommu.h>
21#include <linux/export.h>
22#include <linux/kmemleak.h>
23#include <linux/cc_platform.h>
24#include <linux/iopoll.h>
25#include <asm/pci-direct.h>
26#include <asm/iommu.h>
27#include <asm/apic.h>
28#include <asm/gart.h>
29#include <asm/x86_init.h>
30#include <asm/io_apic.h>
31#include <asm/irq_remapping.h>
32#include <asm/set_memory.h>
33#include <asm/sev.h>
34
35#include <linux/crash_dump.h>
36
37#include "amd_iommu.h"
38#include "../irq_remapping.h"
39
40/*
41 * definitions for the ACPI scanning code
42 */
43#define IVRS_HEADER_LENGTH 48
44
45#define ACPI_IVHD_TYPE_MAX_SUPPORTED	0x40
46#define ACPI_IVMD_TYPE_ALL              0x20
47#define ACPI_IVMD_TYPE                  0x21
48#define ACPI_IVMD_TYPE_RANGE            0x22
49
50#define IVHD_DEV_ALL                    0x01
51#define IVHD_DEV_SELECT                 0x02
52#define IVHD_DEV_SELECT_RANGE_START     0x03
53#define IVHD_DEV_RANGE_END              0x04
54#define IVHD_DEV_ALIAS                  0x42
55#define IVHD_DEV_ALIAS_RANGE            0x43
56#define IVHD_DEV_EXT_SELECT             0x46
57#define IVHD_DEV_EXT_SELECT_RANGE       0x47
58#define IVHD_DEV_SPECIAL		0x48
59#define IVHD_DEV_ACPI_HID		0xf0
60
61#define UID_NOT_PRESENT                 0
62#define UID_IS_INTEGER                  1
63#define UID_IS_CHARACTER                2
64
65#define IVHD_SPECIAL_IOAPIC		1
66#define IVHD_SPECIAL_HPET		2
67
68#define IVHD_FLAG_HT_TUN_EN_MASK        0x01
69#define IVHD_FLAG_PASSPW_EN_MASK        0x02
70#define IVHD_FLAG_RESPASSPW_EN_MASK     0x04
71#define IVHD_FLAG_ISOC_EN_MASK          0x08
72
73#define IVMD_FLAG_EXCL_RANGE            0x08
74#define IVMD_FLAG_IW                    0x04
75#define IVMD_FLAG_IR                    0x02
76#define IVMD_FLAG_UNITY_MAP             0x01
77
78#define ACPI_DEVFLAG_INITPASS           0x01
79#define ACPI_DEVFLAG_EXTINT             0x02
80#define ACPI_DEVFLAG_NMI                0x04
81#define ACPI_DEVFLAG_SYSMGT1            0x10
82#define ACPI_DEVFLAG_SYSMGT2            0x20
83#define ACPI_DEVFLAG_LINT0              0x40
84#define ACPI_DEVFLAG_LINT1              0x80
85#define ACPI_DEVFLAG_ATSDIS             0x10000000
86
87#define IVRS_GET_SBDF_ID(seg, bus, dev, fn)	(((seg & 0xffff) << 16) | ((bus & 0xff) << 8) \
88						 | ((dev & 0x1f) << 3) | (fn & 0x7))
89
90/*
91 * ACPI table definitions
92 *
93 * These data structures are laid over the table to parse the important values
94 * out of it.
95 */
96
97/*
98 * structure describing one IOMMU in the ACPI table. Typically followed by one
99 * or more ivhd_entrys.
100 */
101struct ivhd_header {
102	u8 type;
103	u8 flags;
104	u16 length;
105	u16 devid;
106	u16 cap_ptr;
107	u64 mmio_phys;
108	u16 pci_seg;
109	u16 info;
110	u32 efr_attr;
111
112	/* Following only valid on IVHD type 11h and 40h */
113	u64 efr_reg; /* Exact copy of MMIO_EXT_FEATURES */
114	u64 efr_reg2;
115} __attribute__((packed));
116
117/*
118 * A device entry describing which devices a specific IOMMU translates and
119 * which requestor ids they use.
120 */
121struct ivhd_entry {
122	u8 type;
123	u16 devid;
124	u8 flags;
125	struct_group(ext_hid,
126		u32 ext;
127		u32 hidh;
128	);
129	u64 cid;
130	u8 uidf;
131	u8 uidl;
132	u8 uid;
133} __attribute__((packed));
134
135/*
136 * An AMD IOMMU memory definition structure. It defines things like exclusion
137 * ranges for devices and regions that should be unity mapped.
138 */
139struct ivmd_header {
140	u8 type;
141	u8 flags;
142	u16 length;
143	u16 devid;
144	u16 aux;
145	u16 pci_seg;
146	u8  resv[6];
147	u64 range_start;
148	u64 range_length;
149} __attribute__((packed));
150
151bool amd_iommu_dump;
152bool amd_iommu_irq_remap __read_mostly;
153
154enum io_pgtable_fmt amd_iommu_pgtable = AMD_IOMMU_V1;
155/* Guest page table level */
156int amd_iommu_gpt_level = PAGE_MODE_4_LEVEL;
157
158int amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_VAPIC;
159static int amd_iommu_xt_mode = IRQ_REMAP_XAPIC_MODE;
160
161static bool amd_iommu_detected;
162static bool amd_iommu_disabled __initdata;
163static bool amd_iommu_force_enable __initdata;
164static bool amd_iommu_irtcachedis;
165static int amd_iommu_target_ivhd_type;
166
167/* Global EFR and EFR2 registers */
168u64 amd_iommu_efr;
169u64 amd_iommu_efr2;
170
171/* SNP is enabled on the system? */
172bool amd_iommu_snp_en;
173EXPORT_SYMBOL(amd_iommu_snp_en);
174
175LIST_HEAD(amd_iommu_pci_seg_list);	/* list of all PCI segments */
176LIST_HEAD(amd_iommu_list);		/* list of all AMD IOMMUs in the
177					   system */
178
179/* Array to assign indices to IOMMUs*/
180struct amd_iommu *amd_iommus[MAX_IOMMUS];
181
182/* Number of IOMMUs present in the system */
183static int amd_iommus_present;
184
185/* IOMMUs have a non-present cache? */
186bool amd_iommu_np_cache __read_mostly;
187bool amd_iommu_iotlb_sup __read_mostly = true;
188
189static bool amd_iommu_pc_present __read_mostly;
190bool amdr_ivrs_remap_support __read_mostly;
191
192bool amd_iommu_force_isolation __read_mostly;
193
194/*
195 * AMD IOMMU allows up to 2^16 different protection domains. This is a bitmap
196 * to know which ones are already in use.
197 */
198unsigned long *amd_iommu_pd_alloc_bitmap;
199
200enum iommu_init_state {
201	IOMMU_START_STATE,
202	IOMMU_IVRS_DETECTED,
203	IOMMU_ACPI_FINISHED,
204	IOMMU_ENABLED,
205	IOMMU_PCI_INIT,
206	IOMMU_INTERRUPTS_EN,
207	IOMMU_INITIALIZED,
208	IOMMU_NOT_FOUND,
209	IOMMU_INIT_ERROR,
210	IOMMU_CMDLINE_DISABLED,
211};
212
213/* Early ioapic and hpet maps from kernel command line */
214#define EARLY_MAP_SIZE		4
215static struct devid_map __initdata early_ioapic_map[EARLY_MAP_SIZE];
216static struct devid_map __initdata early_hpet_map[EARLY_MAP_SIZE];
217static struct acpihid_map_entry __initdata early_acpihid_map[EARLY_MAP_SIZE];
218
219static int __initdata early_ioapic_map_size;
220static int __initdata early_hpet_map_size;
221static int __initdata early_acpihid_map_size;
222
223static bool __initdata cmdline_maps;
224
225static enum iommu_init_state init_state = IOMMU_START_STATE;
226
227static int amd_iommu_enable_interrupts(void);
228static int __init iommu_go_to_state(enum iommu_init_state state);
229static void init_device_table_dma(struct amd_iommu_pci_seg *pci_seg);
230
231static bool amd_iommu_pre_enabled = true;
232
233static u32 amd_iommu_ivinfo __initdata;
234
235bool translation_pre_enabled(struct amd_iommu *iommu)
236{
237	return (iommu->flags & AMD_IOMMU_FLAG_TRANS_PRE_ENABLED);
238}
239
240static void clear_translation_pre_enabled(struct amd_iommu *iommu)
241{
242	iommu->flags &= ~AMD_IOMMU_FLAG_TRANS_PRE_ENABLED;
243}
244
245static void init_translation_status(struct amd_iommu *iommu)
246{
247	u64 ctrl;
248
249	ctrl = readq(iommu->mmio_base + MMIO_CONTROL_OFFSET);
250	if (ctrl & (1<<CONTROL_IOMMU_EN))
251		iommu->flags |= AMD_IOMMU_FLAG_TRANS_PRE_ENABLED;
252}
253
254static inline unsigned long tbl_size(int entry_size, int last_bdf)
255{
256	unsigned shift = PAGE_SHIFT +
257			 get_order((last_bdf + 1) * entry_size);
258
259	return 1UL << shift;
260}
261
262int amd_iommu_get_num_iommus(void)
263{
264	return amd_iommus_present;
265}
266
267/*
268 * Iterate through all the IOMMUs to get common EFR
269 * masks among all IOMMUs and warn if found inconsistency.
270 */
271static __init void get_global_efr(void)
272{
273	struct amd_iommu *iommu;
274
275	for_each_iommu(iommu) {
276		u64 tmp = iommu->features;
277		u64 tmp2 = iommu->features2;
278
279		if (list_is_first(&iommu->list, &amd_iommu_list)) {
280			amd_iommu_efr = tmp;
281			amd_iommu_efr2 = tmp2;
282			continue;
283		}
284
285		if (amd_iommu_efr == tmp &&
286		    amd_iommu_efr2 == tmp2)
287			continue;
288
289		pr_err(FW_BUG
290		       "Found inconsistent EFR/EFR2 %#llx,%#llx (global %#llx,%#llx) on iommu%d (%04x:%02x:%02x.%01x).\n",
291		       tmp, tmp2, amd_iommu_efr, amd_iommu_efr2,
292		       iommu->index, iommu->pci_seg->id,
293		       PCI_BUS_NUM(iommu->devid), PCI_SLOT(iommu->devid),
294		       PCI_FUNC(iommu->devid));
295
296		amd_iommu_efr &= tmp;
297		amd_iommu_efr2 &= tmp2;
298	}
299
300	pr_info("Using global IVHD EFR:%#llx, EFR2:%#llx\n", amd_iommu_efr, amd_iommu_efr2);
301}
302
303/*
304 * For IVHD type 0x11/0x40, EFR is also available via IVHD.
305 * Default to IVHD EFR since it is available sooner
306 * (i.e. before PCI init).
307 */
308static void __init early_iommu_features_init(struct amd_iommu *iommu,
309					     struct ivhd_header *h)
310{
311	if (amd_iommu_ivinfo & IOMMU_IVINFO_EFRSUP) {
312		iommu->features = h->efr_reg;
313		iommu->features2 = h->efr_reg2;
314	}
315	if (amd_iommu_ivinfo & IOMMU_IVINFO_DMA_REMAP)
316		amdr_ivrs_remap_support = true;
317}
318
319/* Access to l1 and l2 indexed register spaces */
320
321static u32 iommu_read_l1(struct amd_iommu *iommu, u16 l1, u8 address)
322{
323	u32 val;
324
325	pci_write_config_dword(iommu->dev, 0xf8, (address | l1 << 16));
326	pci_read_config_dword(iommu->dev, 0xfc, &val);
327	return val;
328}
329
330static void iommu_write_l1(struct amd_iommu *iommu, u16 l1, u8 address, u32 val)
331{
332	pci_write_config_dword(iommu->dev, 0xf8, (address | l1 << 16 | 1 << 31));
333	pci_write_config_dword(iommu->dev, 0xfc, val);
334	pci_write_config_dword(iommu->dev, 0xf8, (address | l1 << 16));
335}
336
337static u32 iommu_read_l2(struct amd_iommu *iommu, u8 address)
338{
339	u32 val;
340
341	pci_write_config_dword(iommu->dev, 0xf0, address);
342	pci_read_config_dword(iommu->dev, 0xf4, &val);
343	return val;
344}
345
346static void iommu_write_l2(struct amd_iommu *iommu, u8 address, u32 val)
347{
348	pci_write_config_dword(iommu->dev, 0xf0, (address | 1 << 8));
349	pci_write_config_dword(iommu->dev, 0xf4, val);
350}
351
352/****************************************************************************
353 *
354 * AMD IOMMU MMIO register space handling functions
355 *
356 * These functions are used to program the IOMMU device registers in
357 * MMIO space required for that driver.
358 *
359 ****************************************************************************/
360
361/*
362 * This function set the exclusion range in the IOMMU. DMA accesses to the
363 * exclusion range are passed through untranslated
364 */
365static void iommu_set_exclusion_range(struct amd_iommu *iommu)
366{
367	u64 start = iommu->exclusion_start & PAGE_MASK;
368	u64 limit = (start + iommu->exclusion_length - 1) & PAGE_MASK;
369	u64 entry;
370
371	if (!iommu->exclusion_start)
372		return;
373
374	entry = start | MMIO_EXCL_ENABLE_MASK;
375	memcpy_toio(iommu->mmio_base + MMIO_EXCL_BASE_OFFSET,
376			&entry, sizeof(entry));
377
378	entry = limit;
379	memcpy_toio(iommu->mmio_base + MMIO_EXCL_LIMIT_OFFSET,
380			&entry, sizeof(entry));
381}
382
383static void iommu_set_cwwb_range(struct amd_iommu *iommu)
384{
385	u64 start = iommu_virt_to_phys((void *)iommu->cmd_sem);
386	u64 entry = start & PM_ADDR_MASK;
387
388	if (!check_feature(FEATURE_SNP))
389		return;
390
391	/* Note:
392	 * Re-purpose Exclusion base/limit registers for Completion wait
393	 * write-back base/limit.
394	 */
395	memcpy_toio(iommu->mmio_base + MMIO_EXCL_BASE_OFFSET,
396		    &entry, sizeof(entry));
397
398	/* Note:
399	 * Default to 4 Kbytes, which can be specified by setting base
400	 * address equal to the limit address.
401	 */
402	memcpy_toio(iommu->mmio_base + MMIO_EXCL_LIMIT_OFFSET,
403		    &entry, sizeof(entry));
404}
405
406/* Programs the physical address of the device table into the IOMMU hardware */
407static void iommu_set_device_table(struct amd_iommu *iommu)
408{
409	u64 entry;
410	u32 dev_table_size = iommu->pci_seg->dev_table_size;
411	void *dev_table = (void *)get_dev_table(iommu);
412
413	BUG_ON(iommu->mmio_base == NULL);
414
415	entry = iommu_virt_to_phys(dev_table);
416	entry |= (dev_table_size >> 12) - 1;
417	memcpy_toio(iommu->mmio_base + MMIO_DEV_TABLE_OFFSET,
418			&entry, sizeof(entry));
419}
420
421/* Generic functions to enable/disable certain features of the IOMMU. */
422static void iommu_feature_enable(struct amd_iommu *iommu, u8 bit)
423{
424	u64 ctrl;
425
426	ctrl = readq(iommu->mmio_base +  MMIO_CONTROL_OFFSET);
427	ctrl |= (1ULL << bit);
428	writeq(ctrl, iommu->mmio_base +  MMIO_CONTROL_OFFSET);
429}
430
431static void iommu_feature_disable(struct amd_iommu *iommu, u8 bit)
432{
433	u64 ctrl;
434
435	ctrl = readq(iommu->mmio_base + MMIO_CONTROL_OFFSET);
436	ctrl &= ~(1ULL << bit);
437	writeq(ctrl, iommu->mmio_base + MMIO_CONTROL_OFFSET);
438}
439
440static void iommu_set_inv_tlb_timeout(struct amd_iommu *iommu, int timeout)
441{
442	u64 ctrl;
443
444	ctrl = readq(iommu->mmio_base + MMIO_CONTROL_OFFSET);
445	ctrl &= ~CTRL_INV_TO_MASK;
446	ctrl |= (timeout << CONTROL_INV_TIMEOUT) & CTRL_INV_TO_MASK;
447	writeq(ctrl, iommu->mmio_base + MMIO_CONTROL_OFFSET);
448}
449
450/* Function to enable the hardware */
451static void iommu_enable(struct amd_iommu *iommu)
452{
453	iommu_feature_enable(iommu, CONTROL_IOMMU_EN);
454}
455
456static void iommu_disable(struct amd_iommu *iommu)
457{
458	if (!iommu->mmio_base)
459		return;
460
461	/* Disable command buffer */
462	iommu_feature_disable(iommu, CONTROL_CMDBUF_EN);
463
464	/* Disable event logging and event interrupts */
465	iommu_feature_disable(iommu, CONTROL_EVT_INT_EN);
466	iommu_feature_disable(iommu, CONTROL_EVT_LOG_EN);
467
468	/* Disable IOMMU GA_LOG */
469	iommu_feature_disable(iommu, CONTROL_GALOG_EN);
470	iommu_feature_disable(iommu, CONTROL_GAINT_EN);
471
472	/* Disable IOMMU PPR logging */
473	iommu_feature_disable(iommu, CONTROL_PPRLOG_EN);
474	iommu_feature_disable(iommu, CONTROL_PPRINT_EN);
475
476	/* Disable IOMMU hardware itself */
477	iommu_feature_disable(iommu, CONTROL_IOMMU_EN);
478
479	/* Clear IRTE cache disabling bit */
480	iommu_feature_disable(iommu, CONTROL_IRTCACHEDIS);
481}
482
483/*
484 * mapping and unmapping functions for the IOMMU MMIO space. Each AMD IOMMU in
485 * the system has one.
486 */
487static u8 __iomem * __init iommu_map_mmio_space(u64 address, u64 end)
488{
489	if (!request_mem_region(address, end, "amd_iommu")) {
490		pr_err("Can not reserve memory region %llx-%llx for mmio\n",
491			address, end);
492		pr_err("This is a BIOS bug. Please contact your hardware vendor\n");
493		return NULL;
494	}
495
496	return (u8 __iomem *)ioremap(address, end);
497}
498
499static void __init iommu_unmap_mmio_space(struct amd_iommu *iommu)
500{
501	if (iommu->mmio_base)
502		iounmap(iommu->mmio_base);
503	release_mem_region(iommu->mmio_phys, iommu->mmio_phys_end);
504}
505
506static inline u32 get_ivhd_header_size(struct ivhd_header *h)
507{
508	u32 size = 0;
509
510	switch (h->type) {
511	case 0x10:
512		size = 24;
513		break;
514	case 0x11:
515	case 0x40:
516		size = 40;
517		break;
518	}
519	return size;
520}
521
522/****************************************************************************
523 *
524 * The functions below belong to the first pass of AMD IOMMU ACPI table
525 * parsing. In this pass we try to find out the highest device id this
526 * code has to handle. Upon this information the size of the shared data
527 * structures is determined later.
528 *
529 ****************************************************************************/
530
531/*
532 * This function calculates the length of a given IVHD entry
533 */
534static inline int ivhd_entry_length(u8 *ivhd)
535{
536	u32 type = ((struct ivhd_entry *)ivhd)->type;
537
538	if (type < 0x80) {
539		return 0x04 << (*ivhd >> 6);
540	} else if (type == IVHD_DEV_ACPI_HID) {
541		/* For ACPI_HID, offset 21 is uid len */
542		return *((u8 *)ivhd + 21) + 22;
543	}
544	return 0;
545}
546
547/*
548 * After reading the highest device id from the IOMMU PCI capability header
549 * this function looks if there is a higher device id defined in the ACPI table
550 */
551static int __init find_last_devid_from_ivhd(struct ivhd_header *h)
552{
553	u8 *p = (void *)h, *end = (void *)h;
554	struct ivhd_entry *dev;
555	int last_devid = -EINVAL;
556
557	u32 ivhd_size = get_ivhd_header_size(h);
558
559	if (!ivhd_size) {
560		pr_err("Unsupported IVHD type %#x\n", h->type);
561		return -EINVAL;
562	}
563
564	p += ivhd_size;
565	end += h->length;
566
567	while (p < end) {
568		dev = (struct ivhd_entry *)p;
569		switch (dev->type) {
570		case IVHD_DEV_ALL:
571			/* Use maximum BDF value for DEV_ALL */
572			return 0xffff;
573		case IVHD_DEV_SELECT:
574		case IVHD_DEV_RANGE_END:
575		case IVHD_DEV_ALIAS:
576		case IVHD_DEV_EXT_SELECT:
577			/* all the above subfield types refer to device ids */
578			if (dev->devid > last_devid)
579				last_devid = dev->devid;
580			break;
581		default:
582			break;
583		}
584		p += ivhd_entry_length(p);
585	}
586
587	WARN_ON(p != end);
588
589	return last_devid;
590}
591
592static int __init check_ivrs_checksum(struct acpi_table_header *table)
593{
594	int i;
595	u8 checksum = 0, *p = (u8 *)table;
596
597	for (i = 0; i < table->length; ++i)
598		checksum += p[i];
599	if (checksum != 0) {
600		/* ACPI table corrupt */
601		pr_err(FW_BUG "IVRS invalid checksum\n");
602		return -ENODEV;
603	}
604
605	return 0;
606}
607
608/*
609 * Iterate over all IVHD entries in the ACPI table and find the highest device
610 * id which we need to handle. This is the first of three functions which parse
611 * the ACPI table. So we check the checksum here.
612 */
613static int __init find_last_devid_acpi(struct acpi_table_header *table, u16 pci_seg)
614{
615	u8 *p = (u8 *)table, *end = (u8 *)table;
616	struct ivhd_header *h;
617	int last_devid, last_bdf = 0;
618
619	p += IVRS_HEADER_LENGTH;
620
621	end += table->length;
622	while (p < end) {
623		h = (struct ivhd_header *)p;
624		if (h->pci_seg == pci_seg &&
625		    h->type == amd_iommu_target_ivhd_type) {
626			last_devid = find_last_devid_from_ivhd(h);
627
628			if (last_devid < 0)
629				return -EINVAL;
630			if (last_devid > last_bdf)
631				last_bdf = last_devid;
632		}
633		p += h->length;
634	}
635	WARN_ON(p != end);
636
637	return last_bdf;
638}
639
640/****************************************************************************
641 *
642 * The following functions belong to the code path which parses the ACPI table
643 * the second time. In this ACPI parsing iteration we allocate IOMMU specific
644 * data structures, initialize the per PCI segment device/alias/rlookup table
645 * and also basically initialize the hardware.
646 *
647 ****************************************************************************/
648
649/* Allocate per PCI segment device table */
650static inline int __init alloc_dev_table(struct amd_iommu_pci_seg *pci_seg)
651{
652	pci_seg->dev_table = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO | GFP_DMA32,
653						      get_order(pci_seg->dev_table_size));
654	if (!pci_seg->dev_table)
655		return -ENOMEM;
656
657	return 0;
658}
659
660static inline void free_dev_table(struct amd_iommu_pci_seg *pci_seg)
661{
662	free_pages((unsigned long)pci_seg->dev_table,
663		    get_order(pci_seg->dev_table_size));
664	pci_seg->dev_table = NULL;
665}
666
667/* Allocate per PCI segment IOMMU rlookup table. */
668static inline int __init alloc_rlookup_table(struct amd_iommu_pci_seg *pci_seg)
669{
670	pci_seg->rlookup_table = (void *)__get_free_pages(
671						GFP_KERNEL | __GFP_ZERO,
672						get_order(pci_seg->rlookup_table_size));
673	if (pci_seg->rlookup_table == NULL)
674		return -ENOMEM;
675
676	return 0;
677}
678
679static inline void free_rlookup_table(struct amd_iommu_pci_seg *pci_seg)
680{
681	free_pages((unsigned long)pci_seg->rlookup_table,
682		   get_order(pci_seg->rlookup_table_size));
683	pci_seg->rlookup_table = NULL;
684}
685
686static inline int __init alloc_irq_lookup_table(struct amd_iommu_pci_seg *pci_seg)
687{
688	pci_seg->irq_lookup_table = (void *)__get_free_pages(
689					     GFP_KERNEL | __GFP_ZERO,
690					     get_order(pci_seg->rlookup_table_size));
691	kmemleak_alloc(pci_seg->irq_lookup_table,
692		       pci_seg->rlookup_table_size, 1, GFP_KERNEL);
693	if (pci_seg->irq_lookup_table == NULL)
694		return -ENOMEM;
695
696	return 0;
697}
698
699static inline void free_irq_lookup_table(struct amd_iommu_pci_seg *pci_seg)
700{
701	kmemleak_free(pci_seg->irq_lookup_table);
702	free_pages((unsigned long)pci_seg->irq_lookup_table,
703		   get_order(pci_seg->rlookup_table_size));
704	pci_seg->irq_lookup_table = NULL;
705}
706
707static int __init alloc_alias_table(struct amd_iommu_pci_seg *pci_seg)
708{
709	int i;
710
711	pci_seg->alias_table = (void *)__get_free_pages(GFP_KERNEL,
712					get_order(pci_seg->alias_table_size));
713	if (!pci_seg->alias_table)
714		return -ENOMEM;
715
716	/*
717	 * let all alias entries point to itself
718	 */
719	for (i = 0; i <= pci_seg->last_bdf; ++i)
720		pci_seg->alias_table[i] = i;
721
722	return 0;
723}
724
725static void __init free_alias_table(struct amd_iommu_pci_seg *pci_seg)
726{
727	free_pages((unsigned long)pci_seg->alias_table,
728		   get_order(pci_seg->alias_table_size));
729	pci_seg->alias_table = NULL;
730}
731
732/*
733 * Allocates the command buffer. This buffer is per AMD IOMMU. We can
734 * write commands to that buffer later and the IOMMU will execute them
735 * asynchronously
736 */
737static int __init alloc_command_buffer(struct amd_iommu *iommu)
738{
739	iommu->cmd_buf = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO,
740						  get_order(CMD_BUFFER_SIZE));
741
742	return iommu->cmd_buf ? 0 : -ENOMEM;
743}
744
745/*
746 * Interrupt handler has processed all pending events and adjusted head
747 * and tail pointer. Reset overflow mask and restart logging again.
748 */
749static void amd_iommu_restart_log(struct amd_iommu *iommu, const char *evt_type,
750				  u8 cntrl_intr, u8 cntrl_log,
751				  u32 status_run_mask, u32 status_overflow_mask)
752{
753	u32 status;
754
755	status = readl(iommu->mmio_base + MMIO_STATUS_OFFSET);
756	if (status & status_run_mask)
757		return;
758
759	pr_info_ratelimited("IOMMU %s log restarting\n", evt_type);
760
761	iommu_feature_disable(iommu, cntrl_log);
762	iommu_feature_disable(iommu, cntrl_intr);
763
764	writel(status_overflow_mask, iommu->mmio_base + MMIO_STATUS_OFFSET);
765
766	iommu_feature_enable(iommu, cntrl_intr);
767	iommu_feature_enable(iommu, cntrl_log);
768}
769
770/*
771 * This function restarts event logging in case the IOMMU experienced
772 * an event log buffer overflow.
773 */
774void amd_iommu_restart_event_logging(struct amd_iommu *iommu)
775{
776	amd_iommu_restart_log(iommu, "Event", CONTROL_EVT_INT_EN,
777			      CONTROL_EVT_LOG_EN, MMIO_STATUS_EVT_RUN_MASK,
778			      MMIO_STATUS_EVT_OVERFLOW_MASK);
779}
780
781/*
782 * This function restarts event logging in case the IOMMU experienced
783 * GA log overflow.
784 */
785void amd_iommu_restart_ga_log(struct amd_iommu *iommu)
786{
787	amd_iommu_restart_log(iommu, "GA", CONTROL_GAINT_EN,
788			      CONTROL_GALOG_EN, MMIO_STATUS_GALOG_RUN_MASK,
789			      MMIO_STATUS_GALOG_OVERFLOW_MASK);
790}
791
792/*
793 * This function restarts ppr logging in case the IOMMU experienced
794 * PPR log overflow.
795 */
796void amd_iommu_restart_ppr_log(struct amd_iommu *iommu)
797{
798	amd_iommu_restart_log(iommu, "PPR", CONTROL_PPRINT_EN,
799			      CONTROL_PPRLOG_EN, MMIO_STATUS_PPR_RUN_MASK,
800			      MMIO_STATUS_PPR_OVERFLOW_MASK);
801}
802
803/*
804 * This function resets the command buffer if the IOMMU stopped fetching
805 * commands from it.
806 */
807static void amd_iommu_reset_cmd_buffer(struct amd_iommu *iommu)
808{
809	iommu_feature_disable(iommu, CONTROL_CMDBUF_EN);
810
811	writel(0x00, iommu->mmio_base + MMIO_CMD_HEAD_OFFSET);
812	writel(0x00, iommu->mmio_base + MMIO_CMD_TAIL_OFFSET);
813	iommu->cmd_buf_head = 0;
814	iommu->cmd_buf_tail = 0;
815
816	iommu_feature_enable(iommu, CONTROL_CMDBUF_EN);
817}
818
819/*
820 * This function writes the command buffer address to the hardware and
821 * enables it.
822 */
823static void iommu_enable_command_buffer(struct amd_iommu *iommu)
824{
825	u64 entry;
826
827	BUG_ON(iommu->cmd_buf == NULL);
828
829	entry = iommu_virt_to_phys(iommu->cmd_buf);
830	entry |= MMIO_CMD_SIZE_512;
831
832	memcpy_toio(iommu->mmio_base + MMIO_CMD_BUF_OFFSET,
833		    &entry, sizeof(entry));
834
835	amd_iommu_reset_cmd_buffer(iommu);
836}
837
838/*
839 * This function disables the command buffer
840 */
841static void iommu_disable_command_buffer(struct amd_iommu *iommu)
842{
843	iommu_feature_disable(iommu, CONTROL_CMDBUF_EN);
844}
845
846static void __init free_command_buffer(struct amd_iommu *iommu)
847{
848	free_pages((unsigned long)iommu->cmd_buf, get_order(CMD_BUFFER_SIZE));
849}
850
851static void *__init iommu_alloc_4k_pages(struct amd_iommu *iommu,
852					 gfp_t gfp, size_t size)
853{
854	int order = get_order(size);
855	void *buf = (void *)__get_free_pages(gfp, order);
856
857	if (buf &&
858	    check_feature(FEATURE_SNP) &&
859	    set_memory_4k((unsigned long)buf, (1 << order))) {
860		free_pages((unsigned long)buf, order);
861		buf = NULL;
862	}
863
864	return buf;
865}
866
867/* allocates the memory where the IOMMU will log its events to */
868static int __init alloc_event_buffer(struct amd_iommu *iommu)
869{
870	iommu->evt_buf = iommu_alloc_4k_pages(iommu, GFP_KERNEL | __GFP_ZERO,
871					      EVT_BUFFER_SIZE);
872
873	return iommu->evt_buf ? 0 : -ENOMEM;
874}
875
876static void iommu_enable_event_buffer(struct amd_iommu *iommu)
877{
878	u64 entry;
879
880	BUG_ON(iommu->evt_buf == NULL);
881
882	entry = iommu_virt_to_phys(iommu->evt_buf) | EVT_LEN_MASK;
883
884	memcpy_toio(iommu->mmio_base + MMIO_EVT_BUF_OFFSET,
885		    &entry, sizeof(entry));
886
887	/* set head and tail to zero manually */
888	writel(0x00, iommu->mmio_base + MMIO_EVT_HEAD_OFFSET);
889	writel(0x00, iommu->mmio_base + MMIO_EVT_TAIL_OFFSET);
890
891	iommu_feature_enable(iommu, CONTROL_EVT_LOG_EN);
892}
893
894/*
895 * This function disables the event log buffer
896 */
897static void iommu_disable_event_buffer(struct amd_iommu *iommu)
898{
899	iommu_feature_disable(iommu, CONTROL_EVT_LOG_EN);
900}
901
902static void __init free_event_buffer(struct amd_iommu *iommu)
903{
904	free_pages((unsigned long)iommu->evt_buf, get_order(EVT_BUFFER_SIZE));
905}
906
907/* allocates the memory where the IOMMU will log its events to */
908static int __init alloc_ppr_log(struct amd_iommu *iommu)
909{
910	iommu->ppr_log = iommu_alloc_4k_pages(iommu, GFP_KERNEL | __GFP_ZERO,
911					      PPR_LOG_SIZE);
912
913	return iommu->ppr_log ? 0 : -ENOMEM;
914}
915
916static void iommu_enable_ppr_log(struct amd_iommu *iommu)
917{
918	u64 entry;
919
920	if (iommu->ppr_log == NULL)
921		return;
922
923	iommu_feature_enable(iommu, CONTROL_PPR_EN);
924
925	entry = iommu_virt_to_phys(iommu->ppr_log) | PPR_LOG_SIZE_512;
926
927	memcpy_toio(iommu->mmio_base + MMIO_PPR_LOG_OFFSET,
928		    &entry, sizeof(entry));
929
930	/* set head and tail to zero manually */
931	writel(0x00, iommu->mmio_base + MMIO_PPR_HEAD_OFFSET);
932	writel(0x00, iommu->mmio_base + MMIO_PPR_TAIL_OFFSET);
933
934	iommu_feature_enable(iommu, CONTROL_PPRLOG_EN);
935	iommu_feature_enable(iommu, CONTROL_PPRINT_EN);
936}
937
938static void __init free_ppr_log(struct amd_iommu *iommu)
939{
940	free_pages((unsigned long)iommu->ppr_log, get_order(PPR_LOG_SIZE));
941}
942
943static void free_ga_log(struct amd_iommu *iommu)
944{
945#ifdef CONFIG_IRQ_REMAP
946	free_pages((unsigned long)iommu->ga_log, get_order(GA_LOG_SIZE));
947	free_pages((unsigned long)iommu->ga_log_tail, get_order(8));
948#endif
949}
950
951#ifdef CONFIG_IRQ_REMAP
952static int iommu_ga_log_enable(struct amd_iommu *iommu)
953{
954	u32 status, i;
955	u64 entry;
956
957	if (!iommu->ga_log)
958		return -EINVAL;
959
960	entry = iommu_virt_to_phys(iommu->ga_log) | GA_LOG_SIZE_512;
961	memcpy_toio(iommu->mmio_base + MMIO_GA_LOG_BASE_OFFSET,
962		    &entry, sizeof(entry));
963	entry = (iommu_virt_to_phys(iommu->ga_log_tail) &
964		 (BIT_ULL(52)-1)) & ~7ULL;
965	memcpy_toio(iommu->mmio_base + MMIO_GA_LOG_TAIL_OFFSET,
966		    &entry, sizeof(entry));
967	writel(0x00, iommu->mmio_base + MMIO_GA_HEAD_OFFSET);
968	writel(0x00, iommu->mmio_base + MMIO_GA_TAIL_OFFSET);
969
970
971	iommu_feature_enable(iommu, CONTROL_GAINT_EN);
972	iommu_feature_enable(iommu, CONTROL_GALOG_EN);
973
974	for (i = 0; i < MMIO_STATUS_TIMEOUT; ++i) {
975		status = readl(iommu->mmio_base + MMIO_STATUS_OFFSET);
976		if (status & (MMIO_STATUS_GALOG_RUN_MASK))
977			break;
978		udelay(10);
979	}
980
981	if (WARN_ON(i >= MMIO_STATUS_TIMEOUT))
982		return -EINVAL;
983
984	return 0;
985}
986
987static int iommu_init_ga_log(struct amd_iommu *iommu)
988{
989	if (!AMD_IOMMU_GUEST_IR_VAPIC(amd_iommu_guest_ir))
990		return 0;
991
992	iommu->ga_log = (u8 *)__get_free_pages(GFP_KERNEL | __GFP_ZERO,
993					get_order(GA_LOG_SIZE));
994	if (!iommu->ga_log)
995		goto err_out;
996
997	iommu->ga_log_tail = (u8 *)__get_free_pages(GFP_KERNEL | __GFP_ZERO,
998					get_order(8));
999	if (!iommu->ga_log_tail)
1000		goto err_out;
1001
1002	return 0;
1003err_out:
1004	free_ga_log(iommu);
1005	return -EINVAL;
1006}
1007#endif /* CONFIG_IRQ_REMAP */
1008
1009static int __init alloc_cwwb_sem(struct amd_iommu *iommu)
1010{
1011	iommu->cmd_sem = iommu_alloc_4k_pages(iommu, GFP_KERNEL | __GFP_ZERO, 1);
1012
1013	return iommu->cmd_sem ? 0 : -ENOMEM;
1014}
1015
1016static void __init free_cwwb_sem(struct amd_iommu *iommu)
1017{
1018	if (iommu->cmd_sem)
1019		free_page((unsigned long)iommu->cmd_sem);
1020}
1021
1022static void iommu_enable_xt(struct amd_iommu *iommu)
1023{
1024#ifdef CONFIG_IRQ_REMAP
1025	/*
1026	 * XT mode (32-bit APIC destination ID) requires
1027	 * GA mode (128-bit IRTE support) as a prerequisite.
1028	 */
1029	if (AMD_IOMMU_GUEST_IR_GA(amd_iommu_guest_ir) &&
1030	    amd_iommu_xt_mode == IRQ_REMAP_X2APIC_MODE)
1031		iommu_feature_enable(iommu, CONTROL_XT_EN);
1032#endif /* CONFIG_IRQ_REMAP */
1033}
1034
1035static void iommu_enable_gt(struct amd_iommu *iommu)
1036{
1037	if (!check_feature(FEATURE_GT))
1038		return;
1039
1040	iommu_feature_enable(iommu, CONTROL_GT_EN);
1041}
1042
1043/* sets a specific bit in the device table entry. */
1044static void __set_dev_entry_bit(struct dev_table_entry *dev_table,
1045				u16 devid, u8 bit)
1046{
1047	int i = (bit >> 6) & 0x03;
1048	int _bit = bit & 0x3f;
1049
1050	dev_table[devid].data[i] |= (1UL << _bit);
1051}
1052
1053static void set_dev_entry_bit(struct amd_iommu *iommu, u16 devid, u8 bit)
1054{
1055	struct dev_table_entry *dev_table = get_dev_table(iommu);
1056
1057	return __set_dev_entry_bit(dev_table, devid, bit);
1058}
1059
1060static int __get_dev_entry_bit(struct dev_table_entry *dev_table,
1061			       u16 devid, u8 bit)
1062{
1063	int i = (bit >> 6) & 0x03;
1064	int _bit = bit & 0x3f;
1065
1066	return (dev_table[devid].data[i] & (1UL << _bit)) >> _bit;
1067}
1068
1069static int get_dev_entry_bit(struct amd_iommu *iommu, u16 devid, u8 bit)
1070{
1071	struct dev_table_entry *dev_table = get_dev_table(iommu);
1072
1073	return __get_dev_entry_bit(dev_table, devid, bit);
1074}
1075
1076static bool __copy_device_table(struct amd_iommu *iommu)
1077{
1078	u64 int_ctl, int_tab_len, entry = 0;
1079	struct amd_iommu_pci_seg *pci_seg = iommu->pci_seg;
1080	struct dev_table_entry *old_devtb = NULL;
1081	u32 lo, hi, devid, old_devtb_size;
1082	phys_addr_t old_devtb_phys;
1083	u16 dom_id, dte_v, irq_v;
1084	gfp_t gfp_flag;
1085	u64 tmp;
1086
1087	/* Each IOMMU use separate device table with the same size */
1088	lo = readl(iommu->mmio_base + MMIO_DEV_TABLE_OFFSET);
1089	hi = readl(iommu->mmio_base + MMIO_DEV_TABLE_OFFSET + 4);
1090	entry = (((u64) hi) << 32) + lo;
1091
1092	old_devtb_size = ((entry & ~PAGE_MASK) + 1) << 12;
1093	if (old_devtb_size != pci_seg->dev_table_size) {
1094		pr_err("The device table size of IOMMU:%d is not expected!\n",
1095			iommu->index);
1096		return false;
1097	}
1098
1099	/*
1100	 * When SME is enabled in the first kernel, the entry includes the
1101	 * memory encryption mask(sme_me_mask), we must remove the memory
1102	 * encryption mask to obtain the true physical address in kdump kernel.
1103	 */
1104	old_devtb_phys = __sme_clr(entry) & PAGE_MASK;
1105
1106	if (old_devtb_phys >= 0x100000000ULL) {
1107		pr_err("The address of old device table is above 4G, not trustworthy!\n");
1108		return false;
1109	}
1110	old_devtb = (cc_platform_has(CC_ATTR_HOST_MEM_ENCRYPT) && is_kdump_kernel())
1111		    ? (__force void *)ioremap_encrypted(old_devtb_phys,
1112							pci_seg->dev_table_size)
1113		    : memremap(old_devtb_phys, pci_seg->dev_table_size, MEMREMAP_WB);
1114
1115	if (!old_devtb)
1116		return false;
1117
1118	gfp_flag = GFP_KERNEL | __GFP_ZERO | GFP_DMA32;
1119	pci_seg->old_dev_tbl_cpy = (void *)__get_free_pages(gfp_flag,
1120						    get_order(pci_seg->dev_table_size));
1121	if (pci_seg->old_dev_tbl_cpy == NULL) {
1122		pr_err("Failed to allocate memory for copying old device table!\n");
1123		memunmap(old_devtb);
1124		return false;
1125	}
1126
1127	for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
1128		pci_seg->old_dev_tbl_cpy[devid] = old_devtb[devid];
1129		dom_id = old_devtb[devid].data[1] & DEV_DOMID_MASK;
1130		dte_v = old_devtb[devid].data[0] & DTE_FLAG_V;
1131
1132		if (dte_v && dom_id) {
1133			pci_seg->old_dev_tbl_cpy[devid].data[0] = old_devtb[devid].data[0];
1134			pci_seg->old_dev_tbl_cpy[devid].data[1] = old_devtb[devid].data[1];
1135			__set_bit(dom_id, amd_iommu_pd_alloc_bitmap);
1136			/* If gcr3 table existed, mask it out */
1137			if (old_devtb[devid].data[0] & DTE_FLAG_GV) {
1138				tmp = DTE_GCR3_VAL_B(~0ULL) << DTE_GCR3_SHIFT_B;
1139				tmp |= DTE_GCR3_VAL_C(~0ULL) << DTE_GCR3_SHIFT_C;
1140				pci_seg->old_dev_tbl_cpy[devid].data[1] &= ~tmp;
1141				tmp = DTE_GCR3_VAL_A(~0ULL) << DTE_GCR3_SHIFT_A;
1142				tmp |= DTE_FLAG_GV;
1143				pci_seg->old_dev_tbl_cpy[devid].data[0] &= ~tmp;
1144			}
1145		}
1146
1147		irq_v = old_devtb[devid].data[2] & DTE_IRQ_REMAP_ENABLE;
1148		int_ctl = old_devtb[devid].data[2] & DTE_IRQ_REMAP_INTCTL_MASK;
1149		int_tab_len = old_devtb[devid].data[2] & DTE_INTTABLEN_MASK;
1150		if (irq_v && (int_ctl || int_tab_len)) {
1151			if ((int_ctl != DTE_IRQ_REMAP_INTCTL) ||
1152			    (int_tab_len != DTE_INTTABLEN)) {
1153				pr_err("Wrong old irq remapping flag: %#x\n", devid);
1154				memunmap(old_devtb);
1155				return false;
1156			}
1157
1158			pci_seg->old_dev_tbl_cpy[devid].data[2] = old_devtb[devid].data[2];
1159		}
1160	}
1161	memunmap(old_devtb);
1162
1163	return true;
1164}
1165
1166static bool copy_device_table(void)
1167{
1168	struct amd_iommu *iommu;
1169	struct amd_iommu_pci_seg *pci_seg;
1170
1171	if (!amd_iommu_pre_enabled)
1172		return false;
1173
1174	pr_warn("Translation is already enabled - trying to copy translation structures\n");
1175
1176	/*
1177	 * All IOMMUs within PCI segment shares common device table.
1178	 * Hence copy device table only once per PCI segment.
1179	 */
1180	for_each_pci_segment(pci_seg) {
1181		for_each_iommu(iommu) {
1182			if (pci_seg->id != iommu->pci_seg->id)
1183				continue;
1184			if (!__copy_device_table(iommu))
1185				return false;
1186			break;
1187		}
1188	}
1189
1190	return true;
1191}
1192
1193void amd_iommu_apply_erratum_63(struct amd_iommu *iommu, u16 devid)
1194{
1195	int sysmgt;
1196
1197	sysmgt = get_dev_entry_bit(iommu, devid, DEV_ENTRY_SYSMGT1) |
1198		 (get_dev_entry_bit(iommu, devid, DEV_ENTRY_SYSMGT2) << 1);
1199
1200	if (sysmgt == 0x01)
1201		set_dev_entry_bit(iommu, devid, DEV_ENTRY_IW);
1202}
1203
1204/*
1205 * This function takes the device specific flags read from the ACPI
1206 * table and sets up the device table entry with that information
1207 */
1208static void __init set_dev_entry_from_acpi(struct amd_iommu *iommu,
1209					   u16 devid, u32 flags, u32 ext_flags)
1210{
1211	if (flags & ACPI_DEVFLAG_INITPASS)
1212		set_dev_entry_bit(iommu, devid, DEV_ENTRY_INIT_PASS);
1213	if (flags & ACPI_DEVFLAG_EXTINT)
1214		set_dev_entry_bit(iommu, devid, DEV_ENTRY_EINT_PASS);
1215	if (flags & ACPI_DEVFLAG_NMI)
1216		set_dev_entry_bit(iommu, devid, DEV_ENTRY_NMI_PASS);
1217	if (flags & ACPI_DEVFLAG_SYSMGT1)
1218		set_dev_entry_bit(iommu, devid, DEV_ENTRY_SYSMGT1);
1219	if (flags & ACPI_DEVFLAG_SYSMGT2)
1220		set_dev_entry_bit(iommu, devid, DEV_ENTRY_SYSMGT2);
1221	if (flags & ACPI_DEVFLAG_LINT0)
1222		set_dev_entry_bit(iommu, devid, DEV_ENTRY_LINT0_PASS);
1223	if (flags & ACPI_DEVFLAG_LINT1)
1224		set_dev_entry_bit(iommu, devid, DEV_ENTRY_LINT1_PASS);
1225
1226	amd_iommu_apply_erratum_63(iommu, devid);
1227
1228	amd_iommu_set_rlookup_table(iommu, devid);
1229}
1230
1231int __init add_special_device(u8 type, u8 id, u32 *devid, bool cmd_line)
1232{
1233	struct devid_map *entry;
1234	struct list_head *list;
1235
1236	if (type == IVHD_SPECIAL_IOAPIC)
1237		list = &ioapic_map;
1238	else if (type == IVHD_SPECIAL_HPET)
1239		list = &hpet_map;
1240	else
1241		return -EINVAL;
1242
1243	list_for_each_entry(entry, list, list) {
1244		if (!(entry->id == id && entry->cmd_line))
1245			continue;
1246
1247		pr_info("Command-line override present for %s id %d - ignoring\n",
1248			type == IVHD_SPECIAL_IOAPIC ? "IOAPIC" : "HPET", id);
1249
1250		*devid = entry->devid;
1251
1252		return 0;
1253	}
1254
1255	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
1256	if (!entry)
1257		return -ENOMEM;
1258
1259	entry->id	= id;
1260	entry->devid	= *devid;
1261	entry->cmd_line	= cmd_line;
1262
1263	list_add_tail(&entry->list, list);
1264
1265	return 0;
1266}
1267
1268static int __init add_acpi_hid_device(u8 *hid, u8 *uid, u32 *devid,
1269				      bool cmd_line)
1270{
1271	struct acpihid_map_entry *entry;
1272	struct list_head *list = &acpihid_map;
1273
1274	list_for_each_entry(entry, list, list) {
1275		if (strcmp(entry->hid, hid) ||
1276		    (*uid && *entry->uid && strcmp(entry->uid, uid)) ||
1277		    !entry->cmd_line)
1278			continue;
1279
1280		pr_info("Command-line override for hid:%s uid:%s\n",
1281			hid, uid);
1282		*devid = entry->devid;
1283		return 0;
1284	}
1285
1286	entry = kzalloc(sizeof(*entry), GFP_KERNEL);
1287	if (!entry)
1288		return -ENOMEM;
1289
1290	memcpy(entry->uid, uid, strlen(uid));
1291	memcpy(entry->hid, hid, strlen(hid));
1292	entry->devid = *devid;
1293	entry->cmd_line	= cmd_line;
1294	entry->root_devid = (entry->devid & (~0x7));
1295
1296	pr_info("%s, add hid:%s, uid:%s, rdevid:%d\n",
1297		entry->cmd_line ? "cmd" : "ivrs",
1298		entry->hid, entry->uid, entry->root_devid);
1299
1300	list_add_tail(&entry->list, list);
1301	return 0;
1302}
1303
1304static int __init add_early_maps(void)
1305{
1306	int i, ret;
1307
1308	for (i = 0; i < early_ioapic_map_size; ++i) {
1309		ret = add_special_device(IVHD_SPECIAL_IOAPIC,
1310					 early_ioapic_map[i].id,
1311					 &early_ioapic_map[i].devid,
1312					 early_ioapic_map[i].cmd_line);
1313		if (ret)
1314			return ret;
1315	}
1316
1317	for (i = 0; i < early_hpet_map_size; ++i) {
1318		ret = add_special_device(IVHD_SPECIAL_HPET,
1319					 early_hpet_map[i].id,
1320					 &early_hpet_map[i].devid,
1321					 early_hpet_map[i].cmd_line);
1322		if (ret)
1323			return ret;
1324	}
1325
1326	for (i = 0; i < early_acpihid_map_size; ++i) {
1327		ret = add_acpi_hid_device(early_acpihid_map[i].hid,
1328					  early_acpihid_map[i].uid,
1329					  &early_acpihid_map[i].devid,
1330					  early_acpihid_map[i].cmd_line);
1331		if (ret)
1332			return ret;
1333	}
1334
1335	return 0;
1336}
1337
1338/*
1339 * Takes a pointer to an AMD IOMMU entry in the ACPI table and
1340 * initializes the hardware and our data structures with it.
1341 */
1342static int __init init_iommu_from_acpi(struct amd_iommu *iommu,
1343					struct ivhd_header *h)
1344{
1345	u8 *p = (u8 *)h;
1346	u8 *end = p, flags = 0;
1347	u16 devid = 0, devid_start = 0, devid_to = 0, seg_id;
1348	u32 dev_i, ext_flags = 0;
1349	bool alias = false;
1350	struct ivhd_entry *e;
1351	struct amd_iommu_pci_seg *pci_seg = iommu->pci_seg;
1352	u32 ivhd_size;
1353	int ret;
1354
1355
1356	ret = add_early_maps();
1357	if (ret)
1358		return ret;
1359
1360	amd_iommu_apply_ivrs_quirks();
1361
1362	/*
1363	 * First save the recommended feature enable bits from ACPI
1364	 */
1365	iommu->acpi_flags = h->flags;
1366
1367	/*
1368	 * Done. Now parse the device entries
1369	 */
1370	ivhd_size = get_ivhd_header_size(h);
1371	if (!ivhd_size) {
1372		pr_err("Unsupported IVHD type %#x\n", h->type);
1373		return -EINVAL;
1374	}
1375
1376	p += ivhd_size;
1377
1378	end += h->length;
1379
1380
1381	while (p < end) {
1382		e = (struct ivhd_entry *)p;
1383		seg_id = pci_seg->id;
1384
1385		switch (e->type) {
1386		case IVHD_DEV_ALL:
1387
1388			DUMP_printk("  DEV_ALL\t\t\tflags: %02x\n", e->flags);
1389
1390			for (dev_i = 0; dev_i <= pci_seg->last_bdf; ++dev_i)
1391				set_dev_entry_from_acpi(iommu, dev_i, e->flags, 0);
1392			break;
1393		case IVHD_DEV_SELECT:
1394
1395			DUMP_printk("  DEV_SELECT\t\t\t devid: %04x:%02x:%02x.%x "
1396				    "flags: %02x\n",
1397				    seg_id, PCI_BUS_NUM(e->devid),
1398				    PCI_SLOT(e->devid),
1399				    PCI_FUNC(e->devid),
1400				    e->flags);
1401
1402			devid = e->devid;
1403			set_dev_entry_from_acpi(iommu, devid, e->flags, 0);
1404			break;
1405		case IVHD_DEV_SELECT_RANGE_START:
1406
1407			DUMP_printk("  DEV_SELECT_RANGE_START\t "
1408				    "devid: %04x:%02x:%02x.%x flags: %02x\n",
1409				    seg_id, PCI_BUS_NUM(e->devid),
1410				    PCI_SLOT(e->devid),
1411				    PCI_FUNC(e->devid),
1412				    e->flags);
1413
1414			devid_start = e->devid;
1415			flags = e->flags;
1416			ext_flags = 0;
1417			alias = false;
1418			break;
1419		case IVHD_DEV_ALIAS:
1420
1421			DUMP_printk("  DEV_ALIAS\t\t\t devid: %04x:%02x:%02x.%x "
1422				    "flags: %02x devid_to: %02x:%02x.%x\n",
1423				    seg_id, PCI_BUS_NUM(e->devid),
1424				    PCI_SLOT(e->devid),
1425				    PCI_FUNC(e->devid),
1426				    e->flags,
1427				    PCI_BUS_NUM(e->ext >> 8),
1428				    PCI_SLOT(e->ext >> 8),
1429				    PCI_FUNC(e->ext >> 8));
1430
1431			devid = e->devid;
1432			devid_to = e->ext >> 8;
1433			set_dev_entry_from_acpi(iommu, devid   , e->flags, 0);
1434			set_dev_entry_from_acpi(iommu, devid_to, e->flags, 0);
1435			pci_seg->alias_table[devid] = devid_to;
1436			break;
1437		case IVHD_DEV_ALIAS_RANGE:
1438
1439			DUMP_printk("  DEV_ALIAS_RANGE\t\t "
1440				    "devid: %04x:%02x:%02x.%x flags: %02x "
1441				    "devid_to: %04x:%02x:%02x.%x\n",
1442				    seg_id, PCI_BUS_NUM(e->devid),
1443				    PCI_SLOT(e->devid),
1444				    PCI_FUNC(e->devid),
1445				    e->flags,
1446				    seg_id, PCI_BUS_NUM(e->ext >> 8),
1447				    PCI_SLOT(e->ext >> 8),
1448				    PCI_FUNC(e->ext >> 8));
1449
1450			devid_start = e->devid;
1451			flags = e->flags;
1452			devid_to = e->ext >> 8;
1453			ext_flags = 0;
1454			alias = true;
1455			break;
1456		case IVHD_DEV_EXT_SELECT:
1457
1458			DUMP_printk("  DEV_EXT_SELECT\t\t devid: %04x:%02x:%02x.%x "
1459				    "flags: %02x ext: %08x\n",
1460				    seg_id, PCI_BUS_NUM(e->devid),
1461				    PCI_SLOT(e->devid),
1462				    PCI_FUNC(e->devid),
1463				    e->flags, e->ext);
1464
1465			devid = e->devid;
1466			set_dev_entry_from_acpi(iommu, devid, e->flags,
1467						e->ext);
1468			break;
1469		case IVHD_DEV_EXT_SELECT_RANGE:
1470
1471			DUMP_printk("  DEV_EXT_SELECT_RANGE\t devid: "
1472				    "%04x:%02x:%02x.%x flags: %02x ext: %08x\n",
1473				    seg_id, PCI_BUS_NUM(e->devid),
1474				    PCI_SLOT(e->devid),
1475				    PCI_FUNC(e->devid),
1476				    e->flags, e->ext);
1477
1478			devid_start = e->devid;
1479			flags = e->flags;
1480			ext_flags = e->ext;
1481			alias = false;
1482			break;
1483		case IVHD_DEV_RANGE_END:
1484
1485			DUMP_printk("  DEV_RANGE_END\t\t devid: %04x:%02x:%02x.%x\n",
1486				    seg_id, PCI_BUS_NUM(e->devid),
1487				    PCI_SLOT(e->devid),
1488				    PCI_FUNC(e->devid));
1489
1490			devid = e->devid;
1491			for (dev_i = devid_start; dev_i <= devid; ++dev_i) {
1492				if (alias) {
1493					pci_seg->alias_table[dev_i] = devid_to;
1494					set_dev_entry_from_acpi(iommu,
1495						devid_to, flags, ext_flags);
1496				}
1497				set_dev_entry_from_acpi(iommu, dev_i,
1498							flags, ext_flags);
1499			}
1500			break;
1501		case IVHD_DEV_SPECIAL: {
1502			u8 handle, type;
1503			const char *var;
1504			u32 devid;
1505			int ret;
1506
1507			handle = e->ext & 0xff;
1508			devid = PCI_SEG_DEVID_TO_SBDF(seg_id, (e->ext >> 8));
1509			type   = (e->ext >> 24) & 0xff;
1510
1511			if (type == IVHD_SPECIAL_IOAPIC)
1512				var = "IOAPIC";
1513			else if (type == IVHD_SPECIAL_HPET)
1514				var = "HPET";
1515			else
1516				var = "UNKNOWN";
1517
1518			DUMP_printk("  DEV_SPECIAL(%s[%d])\t\tdevid: %04x:%02x:%02x.%x\n",
1519				    var, (int)handle,
1520				    seg_id, PCI_BUS_NUM(devid),
1521				    PCI_SLOT(devid),
1522				    PCI_FUNC(devid));
1523
1524			ret = add_special_device(type, handle, &devid, false);
1525			if (ret)
1526				return ret;
1527
1528			/*
1529			 * add_special_device might update the devid in case a
1530			 * command-line override is present. So call
1531			 * set_dev_entry_from_acpi after add_special_device.
1532			 */
1533			set_dev_entry_from_acpi(iommu, devid, e->flags, 0);
1534
1535			break;
1536		}
1537		case IVHD_DEV_ACPI_HID: {
1538			u32 devid;
1539			u8 hid[ACPIHID_HID_LEN];
1540			u8 uid[ACPIHID_UID_LEN];
1541			int ret;
1542
1543			if (h->type != 0x40) {
1544				pr_err(FW_BUG "Invalid IVHD device type %#x\n",
1545				       e->type);
1546				break;
1547			}
1548
1549			BUILD_BUG_ON(sizeof(e->ext_hid) != ACPIHID_HID_LEN - 1);
1550			memcpy(hid, &e->ext_hid, ACPIHID_HID_LEN - 1);
1551			hid[ACPIHID_HID_LEN - 1] = '\0';
1552
1553			if (!(*hid)) {
1554				pr_err(FW_BUG "Invalid HID.\n");
1555				break;
1556			}
1557
1558			uid[0] = '\0';
1559			switch (e->uidf) {
1560			case UID_NOT_PRESENT:
1561
1562				if (e->uidl != 0)
1563					pr_warn(FW_BUG "Invalid UID length.\n");
1564
1565				break;
1566			case UID_IS_INTEGER:
1567
1568				sprintf(uid, "%d", e->uid);
1569
1570				break;
1571			case UID_IS_CHARACTER:
1572
1573				memcpy(uid, &e->uid, e->uidl);
1574				uid[e->uidl] = '\0';
1575
1576				break;
1577			default:
1578				break;
1579			}
1580
1581			devid = PCI_SEG_DEVID_TO_SBDF(seg_id, e->devid);
1582			DUMP_printk("  DEV_ACPI_HID(%s[%s])\t\tdevid: %04x:%02x:%02x.%x\n",
1583				    hid, uid, seg_id,
1584				    PCI_BUS_NUM(devid),
1585				    PCI_SLOT(devid),
1586				    PCI_FUNC(devid));
1587
1588			flags = e->flags;
1589
1590			ret = add_acpi_hid_device(hid, uid, &devid, false);
1591			if (ret)
1592				return ret;
1593
1594			/*
1595			 * add_special_device might update the devid in case a
1596			 * command-line override is present. So call
1597			 * set_dev_entry_from_acpi after add_special_device.
1598			 */
1599			set_dev_entry_from_acpi(iommu, devid, e->flags, 0);
1600
1601			break;
1602		}
1603		default:
1604			break;
1605		}
1606
1607		p += ivhd_entry_length(p);
1608	}
1609
1610	return 0;
1611}
1612
1613/* Allocate PCI segment data structure */
1614static struct amd_iommu_pci_seg *__init alloc_pci_segment(u16 id,
1615					  struct acpi_table_header *ivrs_base)
1616{
1617	struct amd_iommu_pci_seg *pci_seg;
1618	int last_bdf;
1619
1620	/*
1621	 * First parse ACPI tables to find the largest Bus/Dev/Func we need to
1622	 * handle in this PCI segment. Upon this information the shared data
1623	 * structures for the PCI segments in the system will be allocated.
1624	 */
1625	last_bdf = find_last_devid_acpi(ivrs_base, id);
1626	if (last_bdf < 0)
1627		return NULL;
1628
1629	pci_seg = kzalloc(sizeof(struct amd_iommu_pci_seg), GFP_KERNEL);
1630	if (pci_seg == NULL)
1631		return NULL;
1632
1633	pci_seg->last_bdf = last_bdf;
1634	DUMP_printk("PCI segment : 0x%0x, last bdf : 0x%04x\n", id, last_bdf);
1635	pci_seg->dev_table_size     = tbl_size(DEV_TABLE_ENTRY_SIZE, last_bdf);
1636	pci_seg->alias_table_size   = tbl_size(ALIAS_TABLE_ENTRY_SIZE, last_bdf);
1637	pci_seg->rlookup_table_size = tbl_size(RLOOKUP_TABLE_ENTRY_SIZE, last_bdf);
1638
1639	pci_seg->id = id;
1640	init_llist_head(&pci_seg->dev_data_list);
1641	INIT_LIST_HEAD(&pci_seg->unity_map);
1642	list_add_tail(&pci_seg->list, &amd_iommu_pci_seg_list);
1643
1644	if (alloc_dev_table(pci_seg))
1645		return NULL;
1646	if (alloc_alias_table(pci_seg))
1647		return NULL;
1648	if (alloc_rlookup_table(pci_seg))
1649		return NULL;
1650
1651	return pci_seg;
1652}
1653
1654static struct amd_iommu_pci_seg *__init get_pci_segment(u16 id,
1655					struct acpi_table_header *ivrs_base)
1656{
1657	struct amd_iommu_pci_seg *pci_seg;
1658
1659	for_each_pci_segment(pci_seg) {
1660		if (pci_seg->id == id)
1661			return pci_seg;
1662	}
1663
1664	return alloc_pci_segment(id, ivrs_base);
1665}
1666
1667static void __init free_pci_segments(void)
1668{
1669	struct amd_iommu_pci_seg *pci_seg, *next;
1670
1671	for_each_pci_segment_safe(pci_seg, next) {
1672		list_del(&pci_seg->list);
1673		free_irq_lookup_table(pci_seg);
1674		free_rlookup_table(pci_seg);
1675		free_alias_table(pci_seg);
1676		free_dev_table(pci_seg);
1677		kfree(pci_seg);
1678	}
1679}
1680
1681static void __init free_iommu_one(struct amd_iommu *iommu)
1682{
1683	free_cwwb_sem(iommu);
1684	free_command_buffer(iommu);
1685	free_event_buffer(iommu);
1686	free_ppr_log(iommu);
1687	free_ga_log(iommu);
1688	iommu_unmap_mmio_space(iommu);
1689}
1690
1691static void __init free_iommu_all(void)
1692{
1693	struct amd_iommu *iommu, *next;
1694
1695	for_each_iommu_safe(iommu, next) {
1696		list_del(&iommu->list);
1697		free_iommu_one(iommu);
1698		kfree(iommu);
1699	}
1700}
1701
1702/*
1703 * Family15h Model 10h-1fh erratum 746 (IOMMU Logging May Stall Translations)
1704 * Workaround:
1705 *     BIOS should disable L2B micellaneous clock gating by setting
1706 *     L2_L2B_CK_GATE_CONTROL[CKGateL2BMiscDisable](D0F2xF4_x90[2]) = 1b
1707 */
1708static void amd_iommu_erratum_746_workaround(struct amd_iommu *iommu)
1709{
1710	u32 value;
1711
1712	if ((boot_cpu_data.x86 != 0x15) ||
1713	    (boot_cpu_data.x86_model < 0x10) ||
1714	    (boot_cpu_data.x86_model > 0x1f))
1715		return;
1716
1717	pci_write_config_dword(iommu->dev, 0xf0, 0x90);
1718	pci_read_config_dword(iommu->dev, 0xf4, &value);
1719
1720	if (value & BIT(2))
1721		return;
1722
1723	/* Select NB indirect register 0x90 and enable writing */
1724	pci_write_config_dword(iommu->dev, 0xf0, 0x90 | (1 << 8));
1725
1726	pci_write_config_dword(iommu->dev, 0xf4, value | 0x4);
1727	pci_info(iommu->dev, "Applying erratum 746 workaround\n");
1728
1729	/* Clear the enable writing bit */
1730	pci_write_config_dword(iommu->dev, 0xf0, 0x90);
1731}
1732
1733/*
1734 * Family15h Model 30h-3fh (IOMMU Mishandles ATS Write Permission)
1735 * Workaround:
1736 *     BIOS should enable ATS write permission check by setting
1737 *     L2_DEBUG_3[AtsIgnoreIWDis](D0F2xF4_x47[0]) = 1b
1738 */
1739static void amd_iommu_ats_write_check_workaround(struct amd_iommu *iommu)
1740{
1741	u32 value;
1742
1743	if ((boot_cpu_data.x86 != 0x15) ||
1744	    (boot_cpu_data.x86_model < 0x30) ||
1745	    (boot_cpu_data.x86_model > 0x3f))
1746		return;
1747
1748	/* Test L2_DEBUG_3[AtsIgnoreIWDis] == 1 */
1749	value = iommu_read_l2(iommu, 0x47);
1750
1751	if (value & BIT(0))
1752		return;
1753
1754	/* Set L2_DEBUG_3[AtsIgnoreIWDis] = 1 */
1755	iommu_write_l2(iommu, 0x47, value | BIT(0));
1756
1757	pci_info(iommu->dev, "Applying ATS write check workaround\n");
1758}
1759
1760/*
1761 * This function glues the initialization function for one IOMMU
1762 * together and also allocates the command buffer and programs the
1763 * hardware. It does NOT enable the IOMMU. This is done afterwards.
1764 */
1765static int __init init_iommu_one(struct amd_iommu *iommu, struct ivhd_header *h,
1766				 struct acpi_table_header *ivrs_base)
1767{
1768	struct amd_iommu_pci_seg *pci_seg;
1769
1770	pci_seg = get_pci_segment(h->pci_seg, ivrs_base);
1771	if (pci_seg == NULL)
1772		return -ENOMEM;
1773	iommu->pci_seg = pci_seg;
1774
1775	raw_spin_lock_init(&iommu->lock);
1776	atomic64_set(&iommu->cmd_sem_val, 0);
1777
1778	/* Add IOMMU to internal data structures */
1779	list_add_tail(&iommu->list, &amd_iommu_list);
1780	iommu->index = amd_iommus_present++;
1781
1782	if (unlikely(iommu->index >= MAX_IOMMUS)) {
1783		WARN(1, "System has more IOMMUs than supported by this driver\n");
1784		return -ENOSYS;
1785	}
1786
1787	/* Index is fine - add IOMMU to the array */
1788	amd_iommus[iommu->index] = iommu;
1789
1790	/*
1791	 * Copy data from ACPI table entry to the iommu struct
1792	 */
1793	iommu->devid   = h->devid;
1794	iommu->cap_ptr = h->cap_ptr;
1795	iommu->mmio_phys = h->mmio_phys;
1796
1797	switch (h->type) {
1798	case 0x10:
1799		/* Check if IVHD EFR contains proper max banks/counters */
1800		if ((h->efr_attr != 0) &&
1801		    ((h->efr_attr & (0xF << 13)) != 0) &&
1802		    ((h->efr_attr & (0x3F << 17)) != 0))
1803			iommu->mmio_phys_end = MMIO_REG_END_OFFSET;
1804		else
1805			iommu->mmio_phys_end = MMIO_CNTR_CONF_OFFSET;
1806
1807		/*
1808		 * Note: GA (128-bit IRTE) mode requires cmpxchg16b supports.
1809		 * GAM also requires GA mode. Therefore, we need to
1810		 * check cmpxchg16b support before enabling it.
1811		 */
1812		if (!boot_cpu_has(X86_FEATURE_CX16) ||
1813		    ((h->efr_attr & (0x1 << IOMMU_FEAT_GASUP_SHIFT)) == 0))
1814			amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_LEGACY;
1815		break;
1816	case 0x11:
1817	case 0x40:
1818		if (h->efr_reg & (1 << 9))
1819			iommu->mmio_phys_end = MMIO_REG_END_OFFSET;
1820		else
1821			iommu->mmio_phys_end = MMIO_CNTR_CONF_OFFSET;
1822
1823		/*
1824		 * Note: GA (128-bit IRTE) mode requires cmpxchg16b supports.
1825		 * XT, GAM also requires GA mode. Therefore, we need to
1826		 * check cmpxchg16b support before enabling them.
1827		 */
1828		if (!boot_cpu_has(X86_FEATURE_CX16) ||
1829		    ((h->efr_reg & (0x1 << IOMMU_EFR_GASUP_SHIFT)) == 0)) {
1830			amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_LEGACY;
1831			break;
1832		}
1833
1834		if (h->efr_reg & BIT(IOMMU_EFR_XTSUP_SHIFT))
1835			amd_iommu_xt_mode = IRQ_REMAP_X2APIC_MODE;
1836
1837		early_iommu_features_init(iommu, h);
1838
1839		break;
1840	default:
1841		return -EINVAL;
1842	}
1843
1844	iommu->mmio_base = iommu_map_mmio_space(iommu->mmio_phys,
1845						iommu->mmio_phys_end);
1846	if (!iommu->mmio_base)
1847		return -ENOMEM;
1848
1849	return init_iommu_from_acpi(iommu, h);
1850}
1851
1852static int __init init_iommu_one_late(struct amd_iommu *iommu)
1853{
1854	int ret;
1855
1856	if (alloc_cwwb_sem(iommu))
1857		return -ENOMEM;
1858
1859	if (alloc_command_buffer(iommu))
1860		return -ENOMEM;
1861
1862	if (alloc_event_buffer(iommu))
1863		return -ENOMEM;
1864
1865	iommu->int_enabled = false;
1866
1867	init_translation_status(iommu);
1868	if (translation_pre_enabled(iommu) && !is_kdump_kernel()) {
1869		iommu_disable(iommu);
1870		clear_translation_pre_enabled(iommu);
1871		pr_warn("Translation was enabled for IOMMU:%d but we are not in kdump mode\n",
1872			iommu->index);
1873	}
1874	if (amd_iommu_pre_enabled)
1875		amd_iommu_pre_enabled = translation_pre_enabled(iommu);
1876
1877	if (amd_iommu_irq_remap) {
1878		ret = amd_iommu_create_irq_domain(iommu);
1879		if (ret)
1880			return ret;
1881	}
1882
1883	/*
1884	 * Make sure IOMMU is not considered to translate itself. The IVRS
1885	 * table tells us so, but this is a lie!
1886	 */
1887	iommu->pci_seg->rlookup_table[iommu->devid] = NULL;
1888
1889	return 0;
1890}
1891
1892/**
1893 * get_highest_supported_ivhd_type - Look up the appropriate IVHD type
1894 * @ivrs: Pointer to the IVRS header
1895 *
1896 * This function search through all IVDB of the maximum supported IVHD
1897 */
1898static u8 get_highest_supported_ivhd_type(struct acpi_table_header *ivrs)
1899{
1900	u8 *base = (u8 *)ivrs;
1901	struct ivhd_header *ivhd = (struct ivhd_header *)
1902					(base + IVRS_HEADER_LENGTH);
1903	u8 last_type = ivhd->type;
1904	u16 devid = ivhd->devid;
1905
1906	while (((u8 *)ivhd - base < ivrs->length) &&
1907	       (ivhd->type <= ACPI_IVHD_TYPE_MAX_SUPPORTED)) {
1908		u8 *p = (u8 *) ivhd;
1909
1910		if (ivhd->devid == devid)
1911			last_type = ivhd->type;
1912		ivhd = (struct ivhd_header *)(p + ivhd->length);
1913	}
1914
1915	return last_type;
1916}
1917
1918/*
1919 * Iterates over all IOMMU entries in the ACPI table, allocates the
1920 * IOMMU structure and initializes it with init_iommu_one()
1921 */
1922static int __init init_iommu_all(struct acpi_table_header *table)
1923{
1924	u8 *p = (u8 *)table, *end = (u8 *)table;
1925	struct ivhd_header *h;
1926	struct amd_iommu *iommu;
1927	int ret;
1928
1929	end += table->length;
1930	p += IVRS_HEADER_LENGTH;
1931
1932	/* Phase 1: Process all IVHD blocks */
1933	while (p < end) {
1934		h = (struct ivhd_header *)p;
1935		if (*p == amd_iommu_target_ivhd_type) {
1936
1937			DUMP_printk("device: %04x:%02x:%02x.%01x cap: %04x "
1938				    "flags: %01x info %04x\n",
1939				    h->pci_seg, PCI_BUS_NUM(h->devid),
1940				    PCI_SLOT(h->devid), PCI_FUNC(h->devid),
1941				    h->cap_ptr, h->flags, h->info);
1942			DUMP_printk("       mmio-addr: %016llx\n",
1943				    h->mmio_phys);
1944
1945			iommu = kzalloc(sizeof(struct amd_iommu), GFP_KERNEL);
1946			if (iommu == NULL)
1947				return -ENOMEM;
1948
1949			ret = init_iommu_one(iommu, h, table);
1950			if (ret)
1951				return ret;
1952		}
1953		p += h->length;
1954
1955	}
1956	WARN_ON(p != end);
1957
1958	/* Phase 2 : Early feature support check */
1959	get_global_efr();
1960
1961	/* Phase 3 : Enabling IOMMU features */
1962	for_each_iommu(iommu) {
1963		ret = init_iommu_one_late(iommu);
1964		if (ret)
1965			return ret;
1966	}
1967
1968	return 0;
1969}
1970
1971static void init_iommu_perf_ctr(struct amd_iommu *iommu)
1972{
1973	u64 val;
1974	struct pci_dev *pdev = iommu->dev;
1975
1976	if (!check_feature(FEATURE_PC))
1977		return;
1978
1979	amd_iommu_pc_present = true;
1980
1981	pci_info(pdev, "IOMMU performance counters supported\n");
1982
1983	val = readl(iommu->mmio_base + MMIO_CNTR_CONF_OFFSET);
1984	iommu->max_banks = (u8) ((val >> 12) & 0x3f);
1985	iommu->max_counters = (u8) ((val >> 7) & 0xf);
1986
1987	return;
1988}
1989
1990static ssize_t amd_iommu_show_cap(struct device *dev,
1991				  struct device_attribute *attr,
1992				  char *buf)
1993{
1994	struct amd_iommu *iommu = dev_to_amd_iommu(dev);
1995	return sysfs_emit(buf, "%x\n", iommu->cap);
1996}
1997static DEVICE_ATTR(cap, S_IRUGO, amd_iommu_show_cap, NULL);
1998
1999static ssize_t amd_iommu_show_features(struct device *dev,
2000				       struct device_attribute *attr,
2001				       char *buf)
2002{
2003	return sysfs_emit(buf, "%llx:%llx\n", amd_iommu_efr, amd_iommu_efr2);
2004}
2005static DEVICE_ATTR(features, S_IRUGO, amd_iommu_show_features, NULL);
2006
2007static struct attribute *amd_iommu_attrs[] = {
2008	&dev_attr_cap.attr,
2009	&dev_attr_features.attr,
2010	NULL,
2011};
2012
2013static struct attribute_group amd_iommu_group = {
2014	.name = "amd-iommu",
2015	.attrs = amd_iommu_attrs,
2016};
2017
2018static const struct attribute_group *amd_iommu_groups[] = {
2019	&amd_iommu_group,
2020	NULL,
2021};
2022
2023/*
2024 * Note: IVHD 0x11 and 0x40 also contains exact copy
2025 * of the IOMMU Extended Feature Register [MMIO Offset 0030h].
2026 * Default to EFR in IVHD since it is available sooner (i.e. before PCI init).
2027 */
2028static void __init late_iommu_features_init(struct amd_iommu *iommu)
2029{
2030	u64 features, features2;
2031
2032	if (!(iommu->cap & (1 << IOMMU_CAP_EFR)))
2033		return;
2034
2035	/* read extended feature bits */
2036	features = readq(iommu->mmio_base + MMIO_EXT_FEATURES);
2037	features2 = readq(iommu->mmio_base + MMIO_EXT_FEATURES2);
2038
2039	if (!amd_iommu_efr) {
2040		amd_iommu_efr = features;
2041		amd_iommu_efr2 = features2;
2042		return;
2043	}
2044
2045	/*
2046	 * Sanity check and warn if EFR values from
2047	 * IVHD and MMIO conflict.
2048	 */
2049	if (features != amd_iommu_efr ||
2050	    features2 != amd_iommu_efr2) {
2051		pr_warn(FW_WARN
2052			"EFR mismatch. Use IVHD EFR (%#llx : %#llx), EFR2 (%#llx : %#llx).\n",
2053			features, amd_iommu_efr,
2054			features2, amd_iommu_efr2);
2055	}
2056}
2057
2058static int __init iommu_init_pci(struct amd_iommu *iommu)
2059{
2060	int cap_ptr = iommu->cap_ptr;
2061	int ret;
2062
2063	iommu->dev = pci_get_domain_bus_and_slot(iommu->pci_seg->id,
2064						 PCI_BUS_NUM(iommu->devid),
2065						 iommu->devid & 0xff);
2066	if (!iommu->dev)
2067		return -ENODEV;
2068
2069	/* Prevent binding other PCI device drivers to IOMMU devices */
2070	iommu->dev->match_driver = false;
2071
2072	/* ACPI _PRT won't have an IRQ for IOMMU */
2073	iommu->dev->irq_managed = 1;
2074
2075	pci_read_config_dword(iommu->dev, cap_ptr + MMIO_CAP_HDR_OFFSET,
2076			      &iommu->cap);
2077
2078	if (!(iommu->cap & (1 << IOMMU_CAP_IOTLB)))
2079		amd_iommu_iotlb_sup = false;
2080
2081	late_iommu_features_init(iommu);
2082
2083	if (check_feature(FEATURE_GT)) {
2084		int glxval;
2085		u64 pasmax;
2086
2087		pasmax = amd_iommu_efr & FEATURE_PASID_MASK;
2088		pasmax >>= FEATURE_PASID_SHIFT;
2089		iommu->iommu.max_pasids = (1 << (pasmax + 1)) - 1;
2090
2091		BUG_ON(iommu->iommu.max_pasids & ~PASID_MASK);
2092
2093		glxval   = amd_iommu_efr & FEATURE_GLXVAL_MASK;
2094		glxval >>= FEATURE_GLXVAL_SHIFT;
2095
2096		if (amd_iommu_max_glx_val == -1)
2097			amd_iommu_max_glx_val = glxval;
2098		else
2099			amd_iommu_max_glx_val = min(amd_iommu_max_glx_val, glxval);
2100	}
2101
2102	if (check_feature(FEATURE_PPR) && alloc_ppr_log(iommu))
2103		return -ENOMEM;
2104
2105	if (iommu->cap & (1UL << IOMMU_CAP_NPCACHE)) {
2106		pr_info("Using strict mode due to virtualization\n");
2107		iommu_set_dma_strict();
2108		amd_iommu_np_cache = true;
2109	}
2110
2111	init_iommu_perf_ctr(iommu);
2112
2113	if (amd_iommu_pgtable == AMD_IOMMU_V2) {
2114		if (!check_feature(FEATURE_GIOSUP) ||
2115		    !check_feature(FEATURE_GT)) {
2116			pr_warn("Cannot enable v2 page table for DMA-API. Fallback to v1.\n");
2117			amd_iommu_pgtable = AMD_IOMMU_V1;
2118		}
2119	}
2120
2121	if (is_rd890_iommu(iommu->dev)) {
2122		int i, j;
2123
2124		iommu->root_pdev =
2125			pci_get_domain_bus_and_slot(iommu->pci_seg->id,
2126						    iommu->dev->bus->number,
2127						    PCI_DEVFN(0, 0));
2128
2129		/*
2130		 * Some rd890 systems may not be fully reconfigured by the
2131		 * BIOS, so it's necessary for us to store this information so
2132		 * it can be reprogrammed on resume
2133		 */
2134		pci_read_config_dword(iommu->dev, iommu->cap_ptr + 4,
2135				&iommu->stored_addr_lo);
2136		pci_read_config_dword(iommu->dev, iommu->cap_ptr + 8,
2137				&iommu->stored_addr_hi);
2138
2139		/* Low bit locks writes to configuration space */
2140		iommu->stored_addr_lo &= ~1;
2141
2142		for (i = 0; i < 6; i++)
2143			for (j = 0; j < 0x12; j++)
2144				iommu->stored_l1[i][j] = iommu_read_l1(iommu, i, j);
2145
2146		for (i = 0; i < 0x83; i++)
2147			iommu->stored_l2[i] = iommu_read_l2(iommu, i);
2148	}
2149
2150	amd_iommu_erratum_746_workaround(iommu);
2151	amd_iommu_ats_write_check_workaround(iommu);
2152
2153	ret = iommu_device_sysfs_add(&iommu->iommu, &iommu->dev->dev,
2154			       amd_iommu_groups, "ivhd%d", iommu->index);
2155	if (ret)
2156		return ret;
2157
2158	iommu_device_register(&iommu->iommu, &amd_iommu_ops, NULL);
2159
2160	return pci_enable_device(iommu->dev);
2161}
2162
2163static void print_iommu_info(void)
2164{
2165	int i;
2166	static const char * const feat_str[] = {
2167		"PreF", "PPR", "X2APIC", "NX", "GT", "[5]",
2168		"IA", "GA", "HE", "PC"
2169	};
2170
2171	if (amd_iommu_efr) {
2172		pr_info("Extended features (%#llx, %#llx):", amd_iommu_efr, amd_iommu_efr2);
2173
2174		for (i = 0; i < ARRAY_SIZE(feat_str); ++i) {
2175			if (check_feature(1ULL << i))
2176				pr_cont(" %s", feat_str[i]);
2177		}
2178
2179		if (check_feature(FEATURE_GAM_VAPIC))
2180			pr_cont(" GA_vAPIC");
2181
2182		if (check_feature(FEATURE_SNP))
2183			pr_cont(" SNP");
2184
2185		pr_cont("\n");
2186	}
2187
2188	if (irq_remapping_enabled) {
2189		pr_info("Interrupt remapping enabled\n");
2190		if (amd_iommu_xt_mode == IRQ_REMAP_X2APIC_MODE)
2191			pr_info("X2APIC enabled\n");
2192	}
2193	if (amd_iommu_pgtable == AMD_IOMMU_V2) {
2194		pr_info("V2 page table enabled (Paging mode : %d level)\n",
2195			amd_iommu_gpt_level);
2196	}
2197}
2198
2199static int __init amd_iommu_init_pci(void)
2200{
2201	struct amd_iommu *iommu;
2202	struct amd_iommu_pci_seg *pci_seg;
2203	int ret;
2204
2205	for_each_iommu(iommu) {
2206		ret = iommu_init_pci(iommu);
2207		if (ret) {
2208			pr_err("IOMMU%d: Failed to initialize IOMMU Hardware (error=%d)!\n",
2209			       iommu->index, ret);
2210			goto out;
2211		}
2212		/* Need to setup range after PCI init */
2213		iommu_set_cwwb_range(iommu);
2214	}
2215
2216	/*
2217	 * Order is important here to make sure any unity map requirements are
2218	 * fulfilled. The unity mappings are created and written to the device
2219	 * table during the iommu_init_pci() call.
2220	 *
2221	 * After that we call init_device_table_dma() to make sure any
2222	 * uninitialized DTE will block DMA, and in the end we flush the caches
2223	 * of all IOMMUs to make sure the changes to the device table are
2224	 * active.
2225	 */
2226	for_each_pci_segment(pci_seg)
2227		init_device_table_dma(pci_seg);
2228
2229	for_each_iommu(iommu)
2230		amd_iommu_flush_all_caches(iommu);
2231
2232	print_iommu_info();
2233
2234out:
2235	return ret;
2236}
2237
2238/****************************************************************************
2239 *
2240 * The following functions initialize the MSI interrupts for all IOMMUs
2241 * in the system. It's a bit challenging because there could be multiple
2242 * IOMMUs per PCI BDF but we can call pci_enable_msi(x) only once per
2243 * pci_dev.
2244 *
2245 ****************************************************************************/
2246
2247static int iommu_setup_msi(struct amd_iommu *iommu)
2248{
2249	int r;
2250
2251	r = pci_enable_msi(iommu->dev);
2252	if (r)
2253		return r;
2254
2255	r = request_threaded_irq(iommu->dev->irq,
2256				 amd_iommu_int_handler,
2257				 amd_iommu_int_thread,
2258				 0, "AMD-Vi",
2259				 iommu);
2260
2261	if (r) {
2262		pci_disable_msi(iommu->dev);
2263		return r;
2264	}
2265
2266	return 0;
2267}
2268
2269union intcapxt {
2270	u64	capxt;
2271	struct {
2272		u64	reserved_0		:  2,
2273			dest_mode_logical	:  1,
2274			reserved_1		:  5,
2275			destid_0_23		: 24,
2276			vector			:  8,
2277			reserved_2		: 16,
2278			destid_24_31		:  8;
2279	};
2280} __attribute__ ((packed));
2281
2282
2283static struct irq_chip intcapxt_controller;
2284
2285static int intcapxt_irqdomain_activate(struct irq_domain *domain,
2286				       struct irq_data *irqd, bool reserve)
2287{
2288	return 0;
2289}
2290
2291static void intcapxt_irqdomain_deactivate(struct irq_domain *domain,
2292					  struct irq_data *irqd)
2293{
2294}
2295
2296
2297static int intcapxt_irqdomain_alloc(struct irq_domain *domain, unsigned int virq,
2298				    unsigned int nr_irqs, void *arg)
2299{
2300	struct irq_alloc_info *info = arg;
2301	int i, ret;
2302
2303	if (!info || info->type != X86_IRQ_ALLOC_TYPE_AMDVI)
2304		return -EINVAL;
2305
2306	ret = irq_domain_alloc_irqs_parent(domain, virq, nr_irqs, arg);
2307	if (ret < 0)
2308		return ret;
2309
2310	for (i = virq; i < virq + nr_irqs; i++) {
2311		struct irq_data *irqd = irq_domain_get_irq_data(domain, i);
2312
2313		irqd->chip = &intcapxt_controller;
2314		irqd->hwirq = info->hwirq;
2315		irqd->chip_data = info->data;
2316		__irq_set_handler(i, handle_edge_irq, 0, "edge");
2317	}
2318
2319	return ret;
2320}
2321
2322static void intcapxt_irqdomain_free(struct irq_domain *domain, unsigned int virq,
2323				    unsigned int nr_irqs)
2324{
2325	irq_domain_free_irqs_top(domain, virq, nr_irqs);
2326}
2327
2328
2329static void intcapxt_unmask_irq(struct irq_data *irqd)
2330{
2331	struct amd_iommu *iommu = irqd->chip_data;
2332	struct irq_cfg *cfg = irqd_cfg(irqd);
2333	union intcapxt xt;
2334
2335	xt.capxt = 0ULL;
2336	xt.dest_mode_logical = apic->dest_mode_logical;
2337	xt.vector = cfg->vector;
2338	xt.destid_0_23 = cfg->dest_apicid & GENMASK(23, 0);
2339	xt.destid_24_31 = cfg->dest_apicid >> 24;
2340
2341	writeq(xt.capxt, iommu->mmio_base + irqd->hwirq);
2342}
2343
2344static void intcapxt_mask_irq(struct irq_data *irqd)
2345{
2346	struct amd_iommu *iommu = irqd->chip_data;
2347
2348	writeq(0, iommu->mmio_base + irqd->hwirq);
2349}
2350
2351
2352static int intcapxt_set_affinity(struct irq_data *irqd,
2353				 const struct cpumask *mask, bool force)
2354{
2355	struct irq_data *parent = irqd->parent_data;
2356	int ret;
2357
2358	ret = parent->chip->irq_set_affinity(parent, mask, force);
2359	if (ret < 0 || ret == IRQ_SET_MASK_OK_DONE)
2360		return ret;
2361	return 0;
2362}
2363
2364static int intcapxt_set_wake(struct irq_data *irqd, unsigned int on)
2365{
2366	return on ? -EOPNOTSUPP : 0;
2367}
2368
2369static struct irq_chip intcapxt_controller = {
2370	.name			= "IOMMU-MSI",
2371	.irq_unmask		= intcapxt_unmask_irq,
2372	.irq_mask		= intcapxt_mask_irq,
2373	.irq_ack		= irq_chip_ack_parent,
2374	.irq_retrigger		= irq_chip_retrigger_hierarchy,
2375	.irq_set_affinity       = intcapxt_set_affinity,
2376	.irq_set_wake		= intcapxt_set_wake,
2377	.flags			= IRQCHIP_MASK_ON_SUSPEND,
2378};
2379
2380static const struct irq_domain_ops intcapxt_domain_ops = {
2381	.alloc			= intcapxt_irqdomain_alloc,
2382	.free			= intcapxt_irqdomain_free,
2383	.activate		= intcapxt_irqdomain_activate,
2384	.deactivate		= intcapxt_irqdomain_deactivate,
2385};
2386
2387
2388static struct irq_domain *iommu_irqdomain;
2389
2390static struct irq_domain *iommu_get_irqdomain(void)
2391{
2392	struct fwnode_handle *fn;
2393
2394	/* No need for locking here (yet) as the init is single-threaded */
2395	if (iommu_irqdomain)
2396		return iommu_irqdomain;
2397
2398	fn = irq_domain_alloc_named_fwnode("AMD-Vi-MSI");
2399	if (!fn)
2400		return NULL;
2401
2402	iommu_irqdomain = irq_domain_create_hierarchy(x86_vector_domain, 0, 0,
2403						      fn, &intcapxt_domain_ops,
2404						      NULL);
2405	if (!iommu_irqdomain)
2406		irq_domain_free_fwnode(fn);
2407
2408	return iommu_irqdomain;
2409}
2410
2411static int __iommu_setup_intcapxt(struct amd_iommu *iommu, const char *devname,
2412				  int hwirq, irq_handler_t thread_fn)
2413{
2414	struct irq_domain *domain;
2415	struct irq_alloc_info info;
2416	int irq, ret;
2417	int node = dev_to_node(&iommu->dev->dev);
2418
2419	domain = iommu_get_irqdomain();
2420	if (!domain)
2421		return -ENXIO;
2422
2423	init_irq_alloc_info(&info, NULL);
2424	info.type = X86_IRQ_ALLOC_TYPE_AMDVI;
2425	info.data = iommu;
2426	info.hwirq = hwirq;
2427
2428	irq = irq_domain_alloc_irqs(domain, 1, node, &info);
2429	if (irq < 0) {
2430		irq_domain_remove(domain);
2431		return irq;
2432	}
2433
2434	ret = request_threaded_irq(irq, amd_iommu_int_handler,
2435				   thread_fn, 0, devname, iommu);
2436	if (ret) {
2437		irq_domain_free_irqs(irq, 1);
2438		irq_domain_remove(domain);
2439		return ret;
2440	}
2441
2442	return 0;
2443}
2444
2445static int iommu_setup_intcapxt(struct amd_iommu *iommu)
2446{
2447	int ret;
2448
2449	snprintf(iommu->evt_irq_name, sizeof(iommu->evt_irq_name),
2450		 "AMD-Vi%d-Evt", iommu->index);
2451	ret = __iommu_setup_intcapxt(iommu, iommu->evt_irq_name,
2452				     MMIO_INTCAPXT_EVT_OFFSET,
2453				     amd_iommu_int_thread_evtlog);
2454	if (ret)
2455		return ret;
2456
2457	snprintf(iommu->ppr_irq_name, sizeof(iommu->ppr_irq_name),
2458		 "AMD-Vi%d-PPR", iommu->index);
2459	ret = __iommu_setup_intcapxt(iommu, iommu->ppr_irq_name,
2460				     MMIO_INTCAPXT_PPR_OFFSET,
2461				     amd_iommu_int_thread_pprlog);
2462	if (ret)
2463		return ret;
2464
2465#ifdef CONFIG_IRQ_REMAP
2466	snprintf(iommu->ga_irq_name, sizeof(iommu->ga_irq_name),
2467		 "AMD-Vi%d-GA", iommu->index);
2468	ret = __iommu_setup_intcapxt(iommu, iommu->ga_irq_name,
2469				     MMIO_INTCAPXT_GALOG_OFFSET,
2470				     amd_iommu_int_thread_galog);
2471#endif
2472
2473	return ret;
2474}
2475
2476static int iommu_init_irq(struct amd_iommu *iommu)
2477{
2478	int ret;
2479
2480	if (iommu->int_enabled)
2481		goto enable_faults;
2482
2483	if (amd_iommu_xt_mode == IRQ_REMAP_X2APIC_MODE)
2484		ret = iommu_setup_intcapxt(iommu);
2485	else if (iommu->dev->msi_cap)
2486		ret = iommu_setup_msi(iommu);
2487	else
2488		ret = -ENODEV;
2489
2490	if (ret)
2491		return ret;
2492
2493	iommu->int_enabled = true;
2494enable_faults:
2495
2496	if (amd_iommu_xt_mode == IRQ_REMAP_X2APIC_MODE)
2497		iommu_feature_enable(iommu, CONTROL_INTCAPXT_EN);
2498
2499	iommu_feature_enable(iommu, CONTROL_EVT_INT_EN);
2500
2501	return 0;
2502}
2503
2504/****************************************************************************
2505 *
2506 * The next functions belong to the third pass of parsing the ACPI
2507 * table. In this last pass the memory mapping requirements are
2508 * gathered (like exclusion and unity mapping ranges).
2509 *
2510 ****************************************************************************/
2511
2512static void __init free_unity_maps(void)
2513{
2514	struct unity_map_entry *entry, *next;
2515	struct amd_iommu_pci_seg *p, *pci_seg;
2516
2517	for_each_pci_segment_safe(pci_seg, p) {
2518		list_for_each_entry_safe(entry, next, &pci_seg->unity_map, list) {
2519			list_del(&entry->list);
2520			kfree(entry);
2521		}
2522	}
2523}
2524
2525/* called for unity map ACPI definition */
2526static int __init init_unity_map_range(struct ivmd_header *m,
2527				       struct acpi_table_header *ivrs_base)
2528{
2529	struct unity_map_entry *e = NULL;
2530	struct amd_iommu_pci_seg *pci_seg;
2531	char *s;
2532
2533	pci_seg = get_pci_segment(m->pci_seg, ivrs_base);
2534	if (pci_seg == NULL)
2535		return -ENOMEM;
2536
2537	e = kzalloc(sizeof(*e), GFP_KERNEL);
2538	if (e == NULL)
2539		return -ENOMEM;
2540
2541	switch (m->type) {
2542	default:
2543		kfree(e);
2544		return 0;
2545	case ACPI_IVMD_TYPE:
2546		s = "IVMD_TYPEi\t\t\t";
2547		e->devid_start = e->devid_end = m->devid;
2548		break;
2549	case ACPI_IVMD_TYPE_ALL:
2550		s = "IVMD_TYPE_ALL\t\t";
2551		e->devid_start = 0;
2552		e->devid_end = pci_seg->last_bdf;
2553		break;
2554	case ACPI_IVMD_TYPE_RANGE:
2555		s = "IVMD_TYPE_RANGE\t\t";
2556		e->devid_start = m->devid;
2557		e->devid_end = m->aux;
2558		break;
2559	}
2560	e->address_start = PAGE_ALIGN(m->range_start);
2561	e->address_end = e->address_start + PAGE_ALIGN(m->range_length);
2562	e->prot = m->flags >> 1;
2563
2564	/*
2565	 * Treat per-device exclusion ranges as r/w unity-mapped regions
2566	 * since some buggy BIOSes might lead to the overwritten exclusion
2567	 * range (exclusion_start and exclusion_length members). This
2568	 * happens when there are multiple exclusion ranges (IVMD entries)
2569	 * defined in ACPI table.
2570	 */
2571	if (m->flags & IVMD_FLAG_EXCL_RANGE)
2572		e->prot = (IVMD_FLAG_IW | IVMD_FLAG_IR) >> 1;
2573
2574	DUMP_printk("%s devid_start: %04x:%02x:%02x.%x devid_end: "
2575		    "%04x:%02x:%02x.%x range_start: %016llx range_end: %016llx"
2576		    " flags: %x\n", s, m->pci_seg,
2577		    PCI_BUS_NUM(e->devid_start), PCI_SLOT(e->devid_start),
2578		    PCI_FUNC(e->devid_start), m->pci_seg,
2579		    PCI_BUS_NUM(e->devid_end),
2580		    PCI_SLOT(e->devid_end), PCI_FUNC(e->devid_end),
2581		    e->address_start, e->address_end, m->flags);
2582
2583	list_add_tail(&e->list, &pci_seg->unity_map);
2584
2585	return 0;
2586}
2587
2588/* iterates over all memory definitions we find in the ACPI table */
2589static int __init init_memory_definitions(struct acpi_table_header *table)
2590{
2591	u8 *p = (u8 *)table, *end = (u8 *)table;
2592	struct ivmd_header *m;
2593
2594	end += table->length;
2595	p += IVRS_HEADER_LENGTH;
2596
2597	while (p < end) {
2598		m = (struct ivmd_header *)p;
2599		if (m->flags & (IVMD_FLAG_UNITY_MAP | IVMD_FLAG_EXCL_RANGE))
2600			init_unity_map_range(m, table);
2601
2602		p += m->length;
2603	}
2604
2605	return 0;
2606}
2607
2608/*
2609 * Init the device table to not allow DMA access for devices
2610 */
2611static void init_device_table_dma(struct amd_iommu_pci_seg *pci_seg)
2612{
2613	u32 devid;
2614	struct dev_table_entry *dev_table = pci_seg->dev_table;
2615
2616	if (dev_table == NULL)
2617		return;
2618
2619	for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
2620		__set_dev_entry_bit(dev_table, devid, DEV_ENTRY_VALID);
2621		if (!amd_iommu_snp_en)
2622			__set_dev_entry_bit(dev_table, devid, DEV_ENTRY_TRANSLATION);
2623	}
2624}
2625
2626static void __init uninit_device_table_dma(struct amd_iommu_pci_seg *pci_seg)
2627{
2628	u32 devid;
2629	struct dev_table_entry *dev_table = pci_seg->dev_table;
2630
2631	if (dev_table == NULL)
2632		return;
2633
2634	for (devid = 0; devid <= pci_seg->last_bdf; ++devid) {
2635		dev_table[devid].data[0] = 0ULL;
2636		dev_table[devid].data[1] = 0ULL;
2637	}
2638}
2639
2640static void init_device_table(void)
2641{
2642	struct amd_iommu_pci_seg *pci_seg;
2643	u32 devid;
2644
2645	if (!amd_iommu_irq_remap)
2646		return;
2647
2648	for_each_pci_segment(pci_seg) {
2649		for (devid = 0; devid <= pci_seg->last_bdf; ++devid)
2650			__set_dev_entry_bit(pci_seg->dev_table,
2651					    devid, DEV_ENTRY_IRQ_TBL_EN);
2652	}
2653}
2654
2655static void iommu_init_flags(struct amd_iommu *iommu)
2656{
2657	iommu->acpi_flags & IVHD_FLAG_HT_TUN_EN_MASK ?
2658		iommu_feature_enable(iommu, CONTROL_HT_TUN_EN) :
2659		iommu_feature_disable(iommu, CONTROL_HT_TUN_EN);
2660
2661	iommu->acpi_flags & IVHD_FLAG_PASSPW_EN_MASK ?
2662		iommu_feature_enable(iommu, CONTROL_PASSPW_EN) :
2663		iommu_feature_disable(iommu, CONTROL_PASSPW_EN);
2664
2665	iommu->acpi_flags & IVHD_FLAG_RESPASSPW_EN_MASK ?
2666		iommu_feature_enable(iommu, CONTROL_RESPASSPW_EN) :
2667		iommu_feature_disable(iommu, CONTROL_RESPASSPW_EN);
2668
2669	iommu->acpi_flags & IVHD_FLAG_ISOC_EN_MASK ?
2670		iommu_feature_enable(iommu, CONTROL_ISOC_EN) :
2671		iommu_feature_disable(iommu, CONTROL_ISOC_EN);
2672
2673	/*
2674	 * make IOMMU memory accesses cache coherent
2675	 */
2676	iommu_feature_enable(iommu, CONTROL_COHERENT_EN);
2677
2678	/* Set IOTLB invalidation timeout to 1s */
2679	iommu_set_inv_tlb_timeout(iommu, CTRL_INV_TO_1S);
2680}
2681
2682static void iommu_apply_resume_quirks(struct amd_iommu *iommu)
2683{
2684	int i, j;
2685	u32 ioc_feature_control;
2686	struct pci_dev *pdev = iommu->root_pdev;
2687
2688	/* RD890 BIOSes may not have completely reconfigured the iommu */
2689	if (!is_rd890_iommu(iommu->dev) || !pdev)
2690		return;
2691
2692	/*
2693	 * First, we need to ensure that the iommu is enabled. This is
2694	 * controlled by a register in the northbridge
2695	 */
2696
2697	/* Select Northbridge indirect register 0x75 and enable writing */
2698	pci_write_config_dword(pdev, 0x60, 0x75 | (1 << 7));
2699	pci_read_config_dword(pdev, 0x64, &ioc_feature_control);
2700
2701	/* Enable the iommu */
2702	if (!(ioc_feature_control & 0x1))
2703		pci_write_config_dword(pdev, 0x64, ioc_feature_control | 1);
2704
2705	/* Restore the iommu BAR */
2706	pci_write_config_dword(iommu->dev, iommu->cap_ptr + 4,
2707			       iommu->stored_addr_lo);
2708	pci_write_config_dword(iommu->dev, iommu->cap_ptr + 8,
2709			       iommu->stored_addr_hi);
2710
2711	/* Restore the l1 indirect regs for each of the 6 l1s */
2712	for (i = 0; i < 6; i++)
2713		for (j = 0; j < 0x12; j++)
2714			iommu_write_l1(iommu, i, j, iommu->stored_l1[i][j]);
2715
2716	/* Restore the l2 indirect regs */
2717	for (i = 0; i < 0x83; i++)
2718		iommu_write_l2(iommu, i, iommu->stored_l2[i]);
2719
2720	/* Lock PCI setup registers */
2721	pci_write_config_dword(iommu->dev, iommu->cap_ptr + 4,
2722			       iommu->stored_addr_lo | 1);
2723}
2724
2725static void iommu_enable_ga(struct amd_iommu *iommu)
2726{
2727#ifdef CONFIG_IRQ_REMAP
2728	switch (amd_iommu_guest_ir) {
2729	case AMD_IOMMU_GUEST_IR_VAPIC:
2730	case AMD_IOMMU_GUEST_IR_LEGACY_GA:
2731		iommu_feature_enable(iommu, CONTROL_GA_EN);
2732		iommu->irte_ops = &irte_128_ops;
2733		break;
2734	default:
2735		iommu->irte_ops = &irte_32_ops;
2736		break;
2737	}
2738#endif
2739}
2740
2741static void iommu_disable_irtcachedis(struct amd_iommu *iommu)
2742{
2743	iommu_feature_disable(iommu, CONTROL_IRTCACHEDIS);
2744}
2745
2746static void iommu_enable_irtcachedis(struct amd_iommu *iommu)
2747{
2748	u64 ctrl;
2749
2750	if (!amd_iommu_irtcachedis)
2751		return;
2752
2753	/*
2754	 * Note:
2755	 * The support for IRTCacheDis feature is dertermined by
2756	 * checking if the bit is writable.
2757	 */
2758	iommu_feature_enable(iommu, CONTROL_IRTCACHEDIS);
2759	ctrl = readq(iommu->mmio_base +  MMIO_CONTROL_OFFSET);
2760	ctrl &= (1ULL << CONTROL_IRTCACHEDIS);
2761	if (ctrl)
2762		iommu->irtcachedis_enabled = true;
2763	pr_info("iommu%d (%#06x) : IRT cache is %s\n",
2764		iommu->index, iommu->devid,
2765		iommu->irtcachedis_enabled ? "disabled" : "enabled");
2766}
2767
2768static void early_enable_iommu(struct amd_iommu *iommu)
2769{
2770	iommu_disable(iommu);
2771	iommu_init_flags(iommu);
2772	iommu_set_device_table(iommu);
2773	iommu_enable_command_buffer(iommu);
2774	iommu_enable_event_buffer(iommu);
2775	iommu_set_exclusion_range(iommu);
2776	iommu_enable_gt(iommu);
2777	iommu_enable_ga(iommu);
2778	iommu_enable_xt(iommu);
2779	iommu_enable_irtcachedis(iommu);
2780	iommu_enable(iommu);
2781	amd_iommu_flush_all_caches(iommu);
2782}
2783
2784/*
2785 * This function finally enables all IOMMUs found in the system after
2786 * they have been initialized.
2787 *
2788 * Or if in kdump kernel and IOMMUs are all pre-enabled, try to copy
2789 * the old content of device table entries. Not this case or copy failed,
2790 * just continue as normal kernel does.
2791 */
2792static void early_enable_iommus(void)
2793{
2794	struct amd_iommu *iommu;
2795	struct amd_iommu_pci_seg *pci_seg;
2796
2797	if (!copy_device_table()) {
2798		/*
2799		 * If come here because of failure in copying device table from old
2800		 * kernel with all IOMMUs enabled, print error message and try to
2801		 * free allocated old_dev_tbl_cpy.
2802		 */
2803		if (amd_iommu_pre_enabled)
2804			pr_err("Failed to copy DEV table from previous kernel.\n");
2805
2806		for_each_pci_segment(pci_seg) {
2807			if (pci_seg->old_dev_tbl_cpy != NULL) {
2808				free_pages((unsigned long)pci_seg->old_dev_tbl_cpy,
2809						get_order(pci_seg->dev_table_size));
2810				pci_seg->old_dev_tbl_cpy = NULL;
2811			}
2812		}
2813
2814		for_each_iommu(iommu) {
2815			clear_translation_pre_enabled(iommu);
2816			early_enable_iommu(iommu);
2817		}
2818	} else {
2819		pr_info("Copied DEV table from previous kernel.\n");
2820
2821		for_each_pci_segment(pci_seg) {
2822			free_pages((unsigned long)pci_seg->dev_table,
2823				   get_order(pci_seg->dev_table_size));
2824			pci_seg->dev_table = pci_seg->old_dev_tbl_cpy;
2825		}
2826
2827		for_each_iommu(iommu) {
2828			iommu_disable_command_buffer(iommu);
2829			iommu_disable_event_buffer(iommu);
2830			iommu_disable_irtcachedis(iommu);
2831			iommu_enable_command_buffer(iommu);
2832			iommu_enable_event_buffer(iommu);
2833			iommu_enable_gt(iommu);
2834			iommu_enable_ga(iommu);
2835			iommu_enable_xt(iommu);
2836			iommu_enable_irtcachedis(iommu);
2837			iommu_set_device_table(iommu);
2838			amd_iommu_flush_all_caches(iommu);
2839		}
2840	}
2841}
2842
2843static void enable_iommus_v2(void)
2844{
2845	struct amd_iommu *iommu;
2846
2847	for_each_iommu(iommu)
2848		iommu_enable_ppr_log(iommu);
2849}
2850
2851static void enable_iommus_vapic(void)
2852{
2853#ifdef CONFIG_IRQ_REMAP
2854	u32 status, i;
2855	struct amd_iommu *iommu;
2856
2857	for_each_iommu(iommu) {
2858		/*
2859		 * Disable GALog if already running. It could have been enabled
2860		 * in the previous boot before kdump.
2861		 */
2862		status = readl(iommu->mmio_base + MMIO_STATUS_OFFSET);
2863		if (!(status & MMIO_STATUS_GALOG_RUN_MASK))
2864			continue;
2865
2866		iommu_feature_disable(iommu, CONTROL_GALOG_EN);
2867		iommu_feature_disable(iommu, CONTROL_GAINT_EN);
2868
2869		/*
2870		 * Need to set and poll check the GALOGRun bit to zero before
2871		 * we can set/ modify GA Log registers safely.
2872		 */
2873		for (i = 0; i < MMIO_STATUS_TIMEOUT; ++i) {
2874			status = readl(iommu->mmio_base + MMIO_STATUS_OFFSET);
2875			if (!(status & MMIO_STATUS_GALOG_RUN_MASK))
2876				break;
2877			udelay(10);
2878		}
2879
2880		if (WARN_ON(i >= MMIO_STATUS_TIMEOUT))
2881			return;
2882	}
2883
2884	if (AMD_IOMMU_GUEST_IR_VAPIC(amd_iommu_guest_ir) &&
2885	    !check_feature(FEATURE_GAM_VAPIC)) {
2886		amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_LEGACY_GA;
2887		return;
2888	}
2889
2890	if (amd_iommu_snp_en &&
2891	    !FEATURE_SNPAVICSUP_GAM(amd_iommu_efr2)) {
2892		pr_warn("Force to disable Virtual APIC due to SNP\n");
2893		amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_LEGACY_GA;
2894		return;
2895	}
2896
2897	/* Enabling GAM and SNPAVIC support */
2898	for_each_iommu(iommu) {
2899		if (iommu_init_ga_log(iommu) ||
2900		    iommu_ga_log_enable(iommu))
2901			return;
2902
2903		iommu_feature_enable(iommu, CONTROL_GAM_EN);
2904		if (amd_iommu_snp_en)
2905			iommu_feature_enable(iommu, CONTROL_SNPAVIC_EN);
2906	}
2907
2908	amd_iommu_irq_ops.capability |= (1 << IRQ_POSTING_CAP);
2909	pr_info("Virtual APIC enabled\n");
2910#endif
2911}
2912
2913static void enable_iommus(void)
2914{
2915	early_enable_iommus();
2916}
2917
2918static void disable_iommus(void)
2919{
2920	struct amd_iommu *iommu;
2921
2922	for_each_iommu(iommu)
2923		iommu_disable(iommu);
2924
2925#ifdef CONFIG_IRQ_REMAP
2926	if (AMD_IOMMU_GUEST_IR_VAPIC(amd_iommu_guest_ir))
2927		amd_iommu_irq_ops.capability &= ~(1 << IRQ_POSTING_CAP);
2928#endif
2929}
2930
2931/*
2932 * Suspend/Resume support
2933 * disable suspend until real resume implemented
2934 */
2935
2936static void amd_iommu_resume(void)
2937{
2938	struct amd_iommu *iommu;
2939
2940	for_each_iommu(iommu)
2941		iommu_apply_resume_quirks(iommu);
2942
2943	/* re-load the hardware */
2944	enable_iommus();
2945
2946	amd_iommu_enable_interrupts();
2947}
2948
2949static int amd_iommu_suspend(void)
2950{
2951	/* disable IOMMUs to go out of the way for BIOS */
2952	disable_iommus();
2953
2954	return 0;
2955}
2956
2957static struct syscore_ops amd_iommu_syscore_ops = {
2958	.suspend = amd_iommu_suspend,
2959	.resume = amd_iommu_resume,
2960};
2961
2962static void __init free_iommu_resources(void)
2963{
2964	kmem_cache_destroy(amd_iommu_irq_cache);
2965	amd_iommu_irq_cache = NULL;
2966
2967	free_iommu_all();
2968	free_pci_segments();
2969}
2970
2971/* SB IOAPIC is always on this device in AMD systems */
2972#define IOAPIC_SB_DEVID		((0x00 << 8) | PCI_DEVFN(0x14, 0))
2973
2974static bool __init check_ioapic_information(void)
2975{
2976	const char *fw_bug = FW_BUG;
2977	bool ret, has_sb_ioapic;
2978	int idx;
2979
2980	has_sb_ioapic = false;
2981	ret           = false;
2982
2983	/*
2984	 * If we have map overrides on the kernel command line the
2985	 * messages in this function might not describe firmware bugs
2986	 * anymore - so be careful
2987	 */
2988	if (cmdline_maps)
2989		fw_bug = "";
2990
2991	for (idx = 0; idx < nr_ioapics; idx++) {
2992		int devid, id = mpc_ioapic_id(idx);
2993
2994		devid = get_ioapic_devid(id);
2995		if (devid < 0) {
2996			pr_err("%s: IOAPIC[%d] not in IVRS table\n",
2997				fw_bug, id);
2998			ret = false;
2999		} else if (devid == IOAPIC_SB_DEVID) {
3000			has_sb_ioapic = true;
3001			ret           = true;
3002		}
3003	}
3004
3005	if (!has_sb_ioapic) {
3006		/*
3007		 * We expect the SB IOAPIC to be listed in the IVRS
3008		 * table. The system timer is connected to the SB IOAPIC
3009		 * and if we don't have it in the list the system will
3010		 * panic at boot time.  This situation usually happens
3011		 * when the BIOS is buggy and provides us the wrong
3012		 * device id for the IOAPIC in the system.
3013		 */
3014		pr_err("%s: No southbridge IOAPIC found\n", fw_bug);
3015	}
3016
3017	if (!ret)
3018		pr_err("Disabling interrupt remapping\n");
3019
3020	return ret;
3021}
3022
3023static void __init free_dma_resources(void)
3024{
3025	free_pages((unsigned long)amd_iommu_pd_alloc_bitmap,
3026		   get_order(MAX_DOMAIN_ID/8));
3027	amd_iommu_pd_alloc_bitmap = NULL;
3028
3029	free_unity_maps();
3030}
3031
3032static void __init ivinfo_init(void *ivrs)
3033{
3034	amd_iommu_ivinfo = *((u32 *)(ivrs + IOMMU_IVINFO_OFFSET));
3035}
3036
3037/*
3038 * This is the hardware init function for AMD IOMMU in the system.
3039 * This function is called either from amd_iommu_init or from the interrupt
3040 * remapping setup code.
3041 *
3042 * This function basically parses the ACPI table for AMD IOMMU (IVRS)
3043 * four times:
3044 *
3045 *	1 pass) Discover the most comprehensive IVHD type to use.
3046 *
3047 *	2 pass) Find the highest PCI device id the driver has to handle.
3048 *		Upon this information the size of the data structures is
3049 *		determined that needs to be allocated.
3050 *
3051 *	3 pass) Initialize the data structures just allocated with the
3052 *		information in the ACPI table about available AMD IOMMUs
3053 *		in the system. It also maps the PCI devices in the
3054 *		system to specific IOMMUs
3055 *
3056 *	4 pass) After the basic data structures are allocated and
3057 *		initialized we update them with information about memory
3058 *		remapping requirements parsed out of the ACPI table in
3059 *		this last pass.
3060 *
3061 * After everything is set up the IOMMUs are enabled and the necessary
3062 * hotplug and suspend notifiers are registered.
3063 */
3064static int __init early_amd_iommu_init(void)
3065{
3066	struct acpi_table_header *ivrs_base;
3067	int remap_cache_sz, ret;
3068	acpi_status status;
3069
3070	if (!amd_iommu_detected)
3071		return -ENODEV;
3072
3073	status = acpi_get_table("IVRS", 0, &ivrs_base);
3074	if (status == AE_NOT_FOUND)
3075		return -ENODEV;
3076	else if (ACPI_FAILURE(status)) {
3077		const char *err = acpi_format_exception(status);
3078		pr_err("IVRS table error: %s\n", err);
3079		return -EINVAL;
3080	}
3081
3082	/*
3083	 * Validate checksum here so we don't need to do it when
3084	 * we actually parse the table
3085	 */
3086	ret = check_ivrs_checksum(ivrs_base);
3087	if (ret)
3088		goto out;
3089
3090	ivinfo_init(ivrs_base);
3091
3092	amd_iommu_target_ivhd_type = get_highest_supported_ivhd_type(ivrs_base);
3093	DUMP_printk("Using IVHD type %#x\n", amd_iommu_target_ivhd_type);
3094
3095	/* Device table - directly used by all IOMMUs */
3096	ret = -ENOMEM;
3097
3098	amd_iommu_pd_alloc_bitmap = (void *)__get_free_pages(
3099					    GFP_KERNEL | __GFP_ZERO,
3100					    get_order(MAX_DOMAIN_ID/8));
3101	if (amd_iommu_pd_alloc_bitmap == NULL)
3102		goto out;
3103
3104	/*
3105	 * never allocate domain 0 because its used as the non-allocated and
3106	 * error value placeholder
3107	 */
3108	__set_bit(0, amd_iommu_pd_alloc_bitmap);
3109
3110	/*
3111	 * now the data structures are allocated and basically initialized
3112	 * start the real acpi table scan
3113	 */
3114	ret = init_iommu_all(ivrs_base);
3115	if (ret)
3116		goto out;
3117
3118	/* 5 level guest page table */
3119	if (cpu_feature_enabled(X86_FEATURE_LA57) &&
3120	    check_feature_gpt_level() == GUEST_PGTABLE_5_LEVEL)
3121		amd_iommu_gpt_level = PAGE_MODE_5_LEVEL;
3122
3123	/* Disable any previously enabled IOMMUs */
3124	if (!is_kdump_kernel() || amd_iommu_disabled)
3125		disable_iommus();
3126
3127	if (amd_iommu_irq_remap)
3128		amd_iommu_irq_remap = check_ioapic_information();
3129
3130	if (amd_iommu_irq_remap) {
3131		struct amd_iommu_pci_seg *pci_seg;
3132		/*
3133		 * Interrupt remapping enabled, create kmem_cache for the
3134		 * remapping tables.
3135		 */
3136		ret = -ENOMEM;
3137		if (!AMD_IOMMU_GUEST_IR_GA(amd_iommu_guest_ir))
3138			remap_cache_sz = MAX_IRQS_PER_TABLE * sizeof(u32);
3139		else
3140			remap_cache_sz = MAX_IRQS_PER_TABLE * (sizeof(u64) * 2);
3141		amd_iommu_irq_cache = kmem_cache_create("irq_remap_cache",
3142							remap_cache_sz,
3143							DTE_INTTAB_ALIGNMENT,
3144							0, NULL);
3145		if (!amd_iommu_irq_cache)
3146			goto out;
3147
3148		for_each_pci_segment(pci_seg) {
3149			if (alloc_irq_lookup_table(pci_seg))
3150				goto out;
3151		}
3152	}
3153
3154	ret = init_memory_definitions(ivrs_base);
3155	if (ret)
3156		goto out;
3157
3158	/* init the device table */
3159	init_device_table();
3160
3161out:
3162	/* Don't leak any ACPI memory */
3163	acpi_put_table(ivrs_base);
3164
3165	return ret;
3166}
3167
3168static int amd_iommu_enable_interrupts(void)
3169{
3170	struct amd_iommu *iommu;
3171	int ret = 0;
3172
3173	for_each_iommu(iommu) {
3174		ret = iommu_init_irq(iommu);
3175		if (ret)
3176			goto out;
3177	}
3178
3179	/*
3180	 * Interrupt handler is ready to process interrupts. Enable
3181	 * PPR and GA log interrupt for all IOMMUs.
3182	 */
3183	enable_iommus_vapic();
3184	enable_iommus_v2();
3185
3186out:
3187	return ret;
3188}
3189
3190static bool __init detect_ivrs(void)
3191{
3192	struct acpi_table_header *ivrs_base;
3193	acpi_status status;
3194	int i;
3195
3196	status = acpi_get_table("IVRS", 0, &ivrs_base);
3197	if (status == AE_NOT_FOUND)
3198		return false;
3199	else if (ACPI_FAILURE(status)) {
3200		const char *err = acpi_format_exception(status);
3201		pr_err("IVRS table error: %s\n", err);
3202		return false;
3203	}
3204
3205	acpi_put_table(ivrs_base);
3206
3207	if (amd_iommu_force_enable)
3208		goto out;
3209
3210	/* Don't use IOMMU if there is Stoney Ridge graphics */
3211	for (i = 0; i < 32; i++) {
3212		u32 pci_id;
3213
3214		pci_id = read_pci_config(0, i, 0, 0);
3215		if ((pci_id & 0xffff) == 0x1002 && (pci_id >> 16) == 0x98e4) {
3216			pr_info("Disable IOMMU on Stoney Ridge\n");
3217			return false;
3218		}
3219	}
3220
3221out:
3222	/* Make sure ACS will be enabled during PCI probe */
3223	pci_request_acs();
3224
3225	return true;
3226}
3227
3228static void iommu_snp_enable(void)
3229{
3230#ifdef CONFIG_KVM_AMD_SEV
3231	if (!cc_platform_has(CC_ATTR_HOST_SEV_SNP))
3232		return;
3233	/*
3234	 * The SNP support requires that IOMMU must be enabled, and is
3235	 * configured with V1 page table (DTE[Mode] = 0 is not supported).
3236	 */
3237	if (no_iommu || iommu_default_passthrough()) {
3238		pr_warn("SNP: IOMMU disabled or configured in passthrough mode, SNP cannot be supported.\n");
3239		goto disable_snp;
3240	}
3241
3242	if (amd_iommu_pgtable != AMD_IOMMU_V1) {
3243		pr_warn("SNP: IOMMU is configured with V2 page table mode, SNP cannot be supported.\n");
3244		goto disable_snp;
3245	}
3246
3247	amd_iommu_snp_en = check_feature(FEATURE_SNP);
3248	if (!amd_iommu_snp_en) {
3249		pr_warn("SNP: IOMMU SNP feature not enabled, SNP cannot be supported.\n");
3250		goto disable_snp;
3251	}
3252
3253	pr_info("IOMMU SNP support enabled.\n");
3254	return;
3255
3256disable_snp:
3257	cc_platform_clear(CC_ATTR_HOST_SEV_SNP);
3258#endif
3259}
3260
3261/****************************************************************************
3262 *
3263 * AMD IOMMU Initialization State Machine
3264 *
3265 ****************************************************************************/
3266
3267static int __init state_next(void)
3268{
3269	int ret = 0;
3270
3271	switch (init_state) {
3272	case IOMMU_START_STATE:
3273		if (!detect_ivrs()) {
3274			init_state	= IOMMU_NOT_FOUND;
3275			ret		= -ENODEV;
3276		} else {
3277			init_state	= IOMMU_IVRS_DETECTED;
3278		}
3279		break;
3280	case IOMMU_IVRS_DETECTED:
3281		if (amd_iommu_disabled) {
3282			init_state = IOMMU_CMDLINE_DISABLED;
3283			ret = -EINVAL;
3284		} else {
3285			ret = early_amd_iommu_init();
3286			init_state = ret ? IOMMU_INIT_ERROR : IOMMU_ACPI_FINISHED;
3287		}
3288		break;
3289	case IOMMU_ACPI_FINISHED:
3290		early_enable_iommus();
3291		x86_platform.iommu_shutdown = disable_iommus;
3292		init_state = IOMMU_ENABLED;
3293		break;
3294	case IOMMU_ENABLED:
3295		register_syscore_ops(&amd_iommu_syscore_ops);
3296		iommu_snp_enable();
3297		ret = amd_iommu_init_pci();
3298		init_state = ret ? IOMMU_INIT_ERROR : IOMMU_PCI_INIT;
3299		break;
3300	case IOMMU_PCI_INIT:
3301		ret = amd_iommu_enable_interrupts();
3302		init_state = ret ? IOMMU_INIT_ERROR : IOMMU_INTERRUPTS_EN;
3303		break;
3304	case IOMMU_INTERRUPTS_EN:
3305		init_state = IOMMU_INITIALIZED;
3306		break;
3307	case IOMMU_INITIALIZED:
3308		/* Nothing to do */
3309		break;
3310	case IOMMU_NOT_FOUND:
3311	case IOMMU_INIT_ERROR:
3312	case IOMMU_CMDLINE_DISABLED:
3313		/* Error states => do nothing */
3314		ret = -EINVAL;
3315		break;
3316	default:
3317		/* Unknown state */
3318		BUG();
3319	}
3320
3321	if (ret) {
3322		free_dma_resources();
3323		if (!irq_remapping_enabled) {
3324			disable_iommus();
3325			free_iommu_resources();
3326		} else {
3327			struct amd_iommu *iommu;
3328			struct amd_iommu_pci_seg *pci_seg;
3329
3330			for_each_pci_segment(pci_seg)
3331				uninit_device_table_dma(pci_seg);
3332
3333			for_each_iommu(iommu)
3334				amd_iommu_flush_all_caches(iommu);
3335		}
3336	}
3337	return ret;
3338}
3339
3340static int __init iommu_go_to_state(enum iommu_init_state state)
3341{
3342	int ret = -EINVAL;
3343
3344	while (init_state != state) {
3345		if (init_state == IOMMU_NOT_FOUND         ||
3346		    init_state == IOMMU_INIT_ERROR        ||
3347		    init_state == IOMMU_CMDLINE_DISABLED)
3348			break;
3349		ret = state_next();
3350	}
3351
3352	return ret;
3353}
3354
3355#ifdef CONFIG_IRQ_REMAP
3356int __init amd_iommu_prepare(void)
3357{
3358	int ret;
3359
3360	amd_iommu_irq_remap = true;
3361
3362	ret = iommu_go_to_state(IOMMU_ACPI_FINISHED);
3363	if (ret) {
3364		amd_iommu_irq_remap = false;
3365		return ret;
3366	}
3367
3368	return amd_iommu_irq_remap ? 0 : -ENODEV;
3369}
3370
3371int __init amd_iommu_enable(void)
3372{
3373	int ret;
3374
3375	ret = iommu_go_to_state(IOMMU_ENABLED);
3376	if (ret)
3377		return ret;
3378
3379	irq_remapping_enabled = 1;
3380	return amd_iommu_xt_mode;
3381}
3382
3383void amd_iommu_disable(void)
3384{
3385	amd_iommu_suspend();
3386}
3387
3388int amd_iommu_reenable(int mode)
3389{
3390	amd_iommu_resume();
3391
3392	return 0;
3393}
3394
3395int __init amd_iommu_enable_faulting(void)
3396{
3397	/* We enable MSI later when PCI is initialized */
3398	return 0;
3399}
3400#endif
3401
3402/*
3403 * This is the core init function for AMD IOMMU hardware in the system.
3404 * This function is called from the generic x86 DMA layer initialization
3405 * code.
3406 */
3407static int __init amd_iommu_init(void)
3408{
3409	struct amd_iommu *iommu;
3410	int ret;
3411
3412	ret = iommu_go_to_state(IOMMU_INITIALIZED);
3413#ifdef CONFIG_GART_IOMMU
3414	if (ret && list_empty(&amd_iommu_list)) {
3415		/*
3416		 * We failed to initialize the AMD IOMMU - try fallback
3417		 * to GART if possible.
3418		 */
3419		gart_iommu_init();
3420	}
3421#endif
3422
3423	for_each_iommu(iommu)
3424		amd_iommu_debugfs_setup(iommu);
3425
3426	return ret;
3427}
3428
3429static bool amd_iommu_sme_check(void)
3430{
3431	if (!cc_platform_has(CC_ATTR_HOST_MEM_ENCRYPT) ||
3432	    (boot_cpu_data.x86 != 0x17))
3433		return true;
3434
3435	/* For Fam17h, a specific level of support is required */
3436	if (boot_cpu_data.microcode >= 0x08001205)
3437		return true;
3438
3439	if ((boot_cpu_data.microcode >= 0x08001126) &&
3440	    (boot_cpu_data.microcode <= 0x080011ff))
3441		return true;
3442
3443	pr_notice("IOMMU not currently supported when SME is active\n");
3444
3445	return false;
3446}
3447
3448/****************************************************************************
3449 *
3450 * Early detect code. This code runs at IOMMU detection time in the DMA
3451 * layer. It just looks if there is an IVRS ACPI table to detect AMD
3452 * IOMMUs
3453 *
3454 ****************************************************************************/
3455int __init amd_iommu_detect(void)
3456{
3457	int ret;
3458
3459	if (no_iommu || (iommu_detected && !gart_iommu_aperture))
3460		return -ENODEV;
3461
3462	if (!amd_iommu_sme_check())
3463		return -ENODEV;
3464
3465	ret = iommu_go_to_state(IOMMU_IVRS_DETECTED);
3466	if (ret)
3467		return ret;
3468
3469	amd_iommu_detected = true;
3470	iommu_detected = 1;
3471	x86_init.iommu.iommu_init = amd_iommu_init;
3472
3473	return 1;
3474}
3475
3476/****************************************************************************
3477 *
3478 * Parsing functions for the AMD IOMMU specific kernel command line
3479 * options.
3480 *
3481 ****************************************************************************/
3482
3483static int __init parse_amd_iommu_dump(char *str)
3484{
3485	amd_iommu_dump = true;
3486
3487	return 1;
3488}
3489
3490static int __init parse_amd_iommu_intr(char *str)
3491{
3492	for (; *str; ++str) {
3493		if (strncmp(str, "legacy", 6) == 0) {
3494			amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_LEGACY_GA;
3495			break;
3496		}
3497		if (strncmp(str, "vapic", 5) == 0) {
3498			amd_iommu_guest_ir = AMD_IOMMU_GUEST_IR_VAPIC;
3499			break;
3500		}
3501	}
3502	return 1;
3503}
3504
3505static int __init parse_amd_iommu_options(char *str)
3506{
3507	if (!str)
3508		return -EINVAL;
3509
3510	while (*str) {
3511		if (strncmp(str, "fullflush", 9) == 0) {
3512			pr_warn("amd_iommu=fullflush deprecated; use iommu.strict=1 instead\n");
3513			iommu_set_dma_strict();
3514		} else if (strncmp(str, "force_enable", 12) == 0) {
3515			amd_iommu_force_enable = true;
3516		} else if (strncmp(str, "off", 3) == 0) {
3517			amd_iommu_disabled = true;
3518		} else if (strncmp(str, "force_isolation", 15) == 0) {
3519			amd_iommu_force_isolation = true;
3520		} else if (strncmp(str, "pgtbl_v1", 8) == 0) {
3521			amd_iommu_pgtable = AMD_IOMMU_V1;
3522		} else if (strncmp(str, "pgtbl_v2", 8) == 0) {
3523			amd_iommu_pgtable = AMD_IOMMU_V2;
3524		} else if (strncmp(str, "irtcachedis", 11) == 0) {
3525			amd_iommu_irtcachedis = true;
3526		} else {
3527			pr_notice("Unknown option - '%s'\n", str);
3528		}
3529
3530		str += strcspn(str, ",");
3531		while (*str == ',')
3532			str++;
3533	}
3534
3535	return 1;
3536}
3537
3538static int __init parse_ivrs_ioapic(char *str)
3539{
3540	u32 seg = 0, bus, dev, fn;
3541	int id, i;
3542	u32 devid;
3543
3544	if (sscanf(str, "=%d@%x:%x.%x", &id, &bus, &dev, &fn) == 4 ||
3545	    sscanf(str, "=%d@%x:%x:%x.%x", &id, &seg, &bus, &dev, &fn) == 5)
3546		goto found;
3547
3548	if (sscanf(str, "[%d]=%x:%x.%x", &id, &bus, &dev, &fn) == 4 ||
3549	    sscanf(str, "[%d]=%x:%x:%x.%x", &id, &seg, &bus, &dev, &fn) == 5) {
3550		pr_warn("ivrs_ioapic%s option format deprecated; use ivrs_ioapic=%d@%04x:%02x:%02x.%d instead\n",
3551			str, id, seg, bus, dev, fn);
3552		goto found;
3553	}
3554
3555	pr_err("Invalid command line: ivrs_ioapic%s\n", str);
3556	return 1;
3557
3558found:
3559	if (early_ioapic_map_size == EARLY_MAP_SIZE) {
3560		pr_err("Early IOAPIC map overflow - ignoring ivrs_ioapic%s\n",
3561			str);
3562		return 1;
3563	}
3564
3565	devid = IVRS_GET_SBDF_ID(seg, bus, dev, fn);
3566
3567	cmdline_maps			= true;
3568	i				= early_ioapic_map_size++;
3569	early_ioapic_map[i].id		= id;
3570	early_ioapic_map[i].devid	= devid;
3571	early_ioapic_map[i].cmd_line	= true;
3572
3573	return 1;
3574}
3575
3576static int __init parse_ivrs_hpet(char *str)
3577{
3578	u32 seg = 0, bus, dev, fn;
3579	int id, i;
3580	u32 devid;
3581
3582	if (sscanf(str, "=%d@%x:%x.%x", &id, &bus, &dev, &fn) == 4 ||
3583	    sscanf(str, "=%d@%x:%x:%x.%x", &id, &seg, &bus, &dev, &fn) == 5)
3584		goto found;
3585
3586	if (sscanf(str, "[%d]=%x:%x.%x", &id, &bus, &dev, &fn) == 4 ||
3587	    sscanf(str, "[%d]=%x:%x:%x.%x", &id, &seg, &bus, &dev, &fn) == 5) {
3588		pr_warn("ivrs_hpet%s option format deprecated; use ivrs_hpet=%d@%04x:%02x:%02x.%d instead\n",
3589			str, id, seg, bus, dev, fn);
3590		goto found;
3591	}
3592
3593	pr_err("Invalid command line: ivrs_hpet%s\n", str);
3594	return 1;
3595
3596found:
3597	if (early_hpet_map_size == EARLY_MAP_SIZE) {
3598		pr_err("Early HPET map overflow - ignoring ivrs_hpet%s\n",
3599			str);
3600		return 1;
3601	}
3602
3603	devid = IVRS_GET_SBDF_ID(seg, bus, dev, fn);
3604
3605	cmdline_maps			= true;
3606	i				= early_hpet_map_size++;
3607	early_hpet_map[i].id		= id;
3608	early_hpet_map[i].devid		= devid;
3609	early_hpet_map[i].cmd_line	= true;
3610
3611	return 1;
3612}
3613
3614#define ACPIID_LEN (ACPIHID_UID_LEN + ACPIHID_HID_LEN)
3615
3616static int __init parse_ivrs_acpihid(char *str)
3617{
3618	u32 seg = 0, bus, dev, fn;
3619	char *hid, *uid, *p, *addr;
3620	char acpiid[ACPIID_LEN] = {0};
3621	int i;
3622
3623	addr = strchr(str, '@');
3624	if (!addr) {
3625		addr = strchr(str, '=');
3626		if (!addr)
3627			goto not_found;
3628
3629		++addr;
3630
3631		if (strlen(addr) > ACPIID_LEN)
3632			goto not_found;
3633
3634		if (sscanf(str, "[%x:%x.%x]=%s", &bus, &dev, &fn, acpiid) == 4 ||
3635		    sscanf(str, "[%x:%x:%x.%x]=%s", &seg, &bus, &dev, &fn, acpiid) == 5) {
3636			pr_warn("ivrs_acpihid%s option format deprecated; use ivrs_acpihid=%s@%04x:%02x:%02x.%d instead\n",
3637				str, acpiid, seg, bus, dev, fn);
3638			goto found;
3639		}
3640		goto not_found;
3641	}
3642
3643	/* We have the '@', make it the terminator to get just the acpiid */
3644	*addr++ = 0;
3645
3646	if (strlen(str) > ACPIID_LEN + 1)
3647		goto not_found;
3648
3649	if (sscanf(str, "=%s", acpiid) != 1)
3650		goto not_found;
3651
3652	if (sscanf(addr, "%x:%x.%x", &bus, &dev, &fn) == 3 ||
3653	    sscanf(addr, "%x:%x:%x.%x", &seg, &bus, &dev, &fn) == 4)
3654		goto found;
3655
3656not_found:
3657	pr_err("Invalid command line: ivrs_acpihid%s\n", str);
3658	return 1;
3659
3660found:
3661	p = acpiid;
3662	hid = strsep(&p, ":");
3663	uid = p;
3664
3665	if (!hid || !(*hid) || !uid) {
3666		pr_err("Invalid command line: hid or uid\n");
3667		return 1;
3668	}
3669
3670	/*
3671	 * Ignore leading zeroes after ':', so e.g., AMDI0095:00
3672	 * will match AMDI0095:0 in the second strcmp in acpi_dev_hid_uid_match
3673	 */
3674	while (*uid == '0' && *(uid + 1))
3675		uid++;
3676
3677	i = early_acpihid_map_size++;
3678	memcpy(early_acpihid_map[i].hid, hid, strlen(hid));
3679	memcpy(early_acpihid_map[i].uid, uid, strlen(uid));
3680	early_acpihid_map[i].devid = IVRS_GET_SBDF_ID(seg, bus, dev, fn);
3681	early_acpihid_map[i].cmd_line	= true;
3682
3683	return 1;
3684}
3685
3686__setup("amd_iommu_dump",	parse_amd_iommu_dump);
3687__setup("amd_iommu=",		parse_amd_iommu_options);
3688__setup("amd_iommu_intr=",	parse_amd_iommu_intr);
3689__setup("ivrs_ioapic",		parse_ivrs_ioapic);
3690__setup("ivrs_hpet",		parse_ivrs_hpet);
3691__setup("ivrs_acpihid",		parse_ivrs_acpihid);
3692
3693bool amd_iommu_v2_supported(void)
3694{
3695	/* CPU page table size should match IOMMU guest page table size */
3696	if (cpu_feature_enabled(X86_FEATURE_LA57) &&
3697	    amd_iommu_gpt_level != PAGE_MODE_5_LEVEL)
3698		return false;
3699
3700	/*
3701	 * Since DTE[Mode]=0 is prohibited on SNP-enabled system
3702	 * (i.e. EFR[SNPSup]=1), IOMMUv2 page table cannot be used without
3703	 * setting up IOMMUv1 page table.
3704	 */
3705	return amd_iommu_gt_ppr_supported() && !amd_iommu_snp_en;
3706}
3707
3708struct amd_iommu *get_amd_iommu(unsigned int idx)
3709{
3710	unsigned int i = 0;
3711	struct amd_iommu *iommu;
3712
3713	for_each_iommu(iommu)
3714		if (i++ == idx)
3715			return iommu;
3716	return NULL;
3717}
3718
3719/****************************************************************************
3720 *
3721 * IOMMU EFR Performance Counter support functionality. This code allows
3722 * access to the IOMMU PC functionality.
3723 *
3724 ****************************************************************************/
3725
3726u8 amd_iommu_pc_get_max_banks(unsigned int idx)
3727{
3728	struct amd_iommu *iommu = get_amd_iommu(idx);
3729
3730	if (iommu)
3731		return iommu->max_banks;
3732
3733	return 0;
3734}
3735
3736bool amd_iommu_pc_supported(void)
3737{
3738	return amd_iommu_pc_present;
3739}
3740
3741u8 amd_iommu_pc_get_max_counters(unsigned int idx)
3742{
3743	struct amd_iommu *iommu = get_amd_iommu(idx);
3744
3745	if (iommu)
3746		return iommu->max_counters;
3747
3748	return 0;
3749}
3750
3751static int iommu_pc_get_set_reg(struct amd_iommu *iommu, u8 bank, u8 cntr,
3752				u8 fxn, u64 *value, bool is_write)
3753{
3754	u32 offset;
3755	u32 max_offset_lim;
3756
3757	/* Make sure the IOMMU PC resource is available */
3758	if (!amd_iommu_pc_present)
3759		return -ENODEV;
3760
3761	/* Check for valid iommu and pc register indexing */
3762	if (WARN_ON(!iommu || (fxn > 0x28) || (fxn & 7)))
3763		return -ENODEV;
3764
3765	offset = (u32)(((0x40 | bank) << 12) | (cntr << 8) | fxn);
3766
3767	/* Limit the offset to the hw defined mmio region aperture */
3768	max_offset_lim = (u32)(((0x40 | iommu->max_banks) << 12) |
3769				(iommu->max_counters << 8) | 0x28);
3770	if ((offset < MMIO_CNTR_REG_OFFSET) ||
3771	    (offset > max_offset_lim))
3772		return -EINVAL;
3773
3774	if (is_write) {
3775		u64 val = *value & GENMASK_ULL(47, 0);
3776
3777		writel((u32)val, iommu->mmio_base + offset);
3778		writel((val >> 32), iommu->mmio_base + offset + 4);
3779	} else {
3780		*value = readl(iommu->mmio_base + offset + 4);
3781		*value <<= 32;
3782		*value |= readl(iommu->mmio_base + offset);
3783		*value &= GENMASK_ULL(47, 0);
3784	}
3785
3786	return 0;
3787}
3788
3789int amd_iommu_pc_get_reg(struct amd_iommu *iommu, u8 bank, u8 cntr, u8 fxn, u64 *value)
3790{
3791	if (!iommu)
3792		return -EINVAL;
3793
3794	return iommu_pc_get_set_reg(iommu, bank, cntr, fxn, value, false);
3795}
3796
3797int amd_iommu_pc_set_reg(struct amd_iommu *iommu, u8 bank, u8 cntr, u8 fxn, u64 *value)
3798{
3799	if (!iommu)
3800		return -EINVAL;
3801
3802	return iommu_pc_get_set_reg(iommu, bank, cntr, fxn, value, true);
3803}
3804
3805#ifdef CONFIG_KVM_AMD_SEV
3806static int iommu_page_make_shared(void *page)
3807{
3808	unsigned long paddr, pfn;
3809
3810	paddr = iommu_virt_to_phys(page);
3811	/* Cbit maybe set in the paddr */
3812	pfn = __sme_clr(paddr) >> PAGE_SHIFT;
3813
3814	if (!(pfn % PTRS_PER_PMD)) {
3815		int ret, level;
3816		bool assigned;
3817
3818		ret = snp_lookup_rmpentry(pfn, &assigned, &level);
3819		if (ret) {
3820			pr_warn("IOMMU PFN %lx RMP lookup failed, ret %d\n", pfn, ret);
3821			return ret;
3822		}
3823
3824		if (!assigned) {
3825			pr_warn("IOMMU PFN %lx not assigned in RMP table\n", pfn);
3826			return -EINVAL;
3827		}
3828
3829		if (level > PG_LEVEL_4K) {
3830			ret = psmash(pfn);
3831			if (!ret)
3832				goto done;
3833
3834			pr_warn("PSMASH failed for IOMMU PFN %lx huge RMP entry, ret: %d, level: %d\n",
3835				pfn, ret, level);
3836			return ret;
3837		}
3838	}
3839
3840done:
3841	return rmp_make_shared(pfn, PG_LEVEL_4K);
3842}
3843
3844static int iommu_make_shared(void *va, size_t size)
3845{
3846	void *page;
3847	int ret;
3848
3849	if (!va)
3850		return 0;
3851
3852	for (page = va; page < (va + size); page += PAGE_SIZE) {
3853		ret = iommu_page_make_shared(page);
3854		if (ret)
3855			return ret;
3856	}
3857
3858	return 0;
3859}
3860
3861int amd_iommu_snp_disable(void)
3862{
3863	struct amd_iommu *iommu;
3864	int ret;
3865
3866	if (!amd_iommu_snp_en)
3867		return 0;
3868
3869	for_each_iommu(iommu) {
3870		ret = iommu_make_shared(iommu->evt_buf, EVT_BUFFER_SIZE);
3871		if (ret)
3872			return ret;
3873
3874		ret = iommu_make_shared(iommu->ppr_log, PPR_LOG_SIZE);
3875		if (ret)
3876			return ret;
3877
3878		ret = iommu_make_shared((void *)iommu->cmd_sem, PAGE_SIZE);
3879		if (ret)
3880			return ret;
3881	}
3882
3883	return 0;
3884}
3885EXPORT_SYMBOL_GPL(amd_iommu_snp_disable);
3886#endif
3887