7  * RISCV IOMMU as a PCIe device
16 #include <linux/iommu.h>
20 #include "iommu-bits.h"
21 #include "iommu.h"
23 /* QEMU RISC-V IOMMU implementation */
36 	struct riscv_iommu_device *iommu;  in riscv_iommu_pci_probe()  local
53 	iommu = devm_kzalloc(dev, sizeof(*iommu), GFP_KERNEL);  in riscv_iommu_pci_probe()
54 	if (!iommu)  in riscv_iommu_pci_probe()
57 	iommu->dev = dev;  in riscv_iommu_pci_probe()
58 	iommu->reg = pcim_iomap_table(pdev)[0];  in riscv_iommu_pci_probe()
61 	dev_set_drvdata(dev, iommu);  in riscv_iommu_pci_probe()
64 	iommu->caps = riscv_iommu_readq(iommu, RISCV_IOMMU_REG_CAPABILITIES);  in riscv_iommu_pci_probe()
65 	iommu->fctl = riscv_iommu_readl(iommu, RISCV_IOMMU_REG_FCTL);  in riscv_iommu_pci_probe()
67 	/* The PCI driver only uses MSIs, make sure the IOMMU supports this */  in riscv_iommu_pci_probe()
68 	switch (FIELD_GET(RISCV_IOMMU_CAPABILITIES_IGS, iommu->caps)) {  in riscv_iommu_pci_probe()
84 	iommu->irqs_count = rc;  in riscv_iommu_pci_probe()
85 	for (vec = 0; vec < iommu->irqs_count; vec++)  in riscv_iommu_pci_probe()
86 		iommu->irqs[vec] = msi_get_virq(dev, vec);  in riscv_iommu_pci_probe()
89 	if (iommu->fctl & RISCV_IOMMU_FCTL_WSI) {  in riscv_iommu_pci_probe()
90 		iommu->fctl ^= RISCV_IOMMU_FCTL_WSI;  in riscv_iommu_pci_probe()
91 		riscv_iommu_writel(iommu, RISCV_IOMMU_REG_FCTL, iommu->fctl);  in riscv_iommu_pci_probe()
94 	return riscv_iommu_init(iommu);  in riscv_iommu_pci_probe()
99 	struct riscv_iommu_device *iommu = dev_get_drvdata(&pdev->dev);  in riscv_iommu_pci_remove()  local
101 	riscv_iommu_remove(iommu);  in riscv_iommu_pci_remove()
106 	struct riscv_iommu_device *iommu = dev_get_drvdata(&pdev->dev);  in riscv_iommu_pci_shutdown()  local
108 	riscv_iommu_disable(iommu);  in riscv_iommu_pci_shutdown()