162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * mmconfig.c - Low-level direct PCI config space access via MMCONFIG
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * This is an 64bit optimized version that always keeps the full mmconfig
662306a36Sopenharmony_ci * space mapped. This allows lockless config space operation.
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <linux/pci.h>
1062306a36Sopenharmony_ci#include <linux/init.h>
1162306a36Sopenharmony_ci#include <linux/acpi.h>
1262306a36Sopenharmony_ci#include <linux/bitmap.h>
1362306a36Sopenharmony_ci#include <linux/rcupdate.h>
1462306a36Sopenharmony_ci#include <asm/e820/api.h>
1562306a36Sopenharmony_ci#include <asm/pci_x86.h>
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_ci#define PREFIX "PCI: "
1862306a36Sopenharmony_ci
1962306a36Sopenharmony_cistatic char __iomem *pci_dev_base(unsigned int seg, unsigned int bus, unsigned int devfn)
2062306a36Sopenharmony_ci{
2162306a36Sopenharmony_ci	struct pci_mmcfg_region *cfg = pci_mmconfig_lookup(seg, bus);
2262306a36Sopenharmony_ci
2362306a36Sopenharmony_ci	if (cfg && cfg->virt)
2462306a36Sopenharmony_ci		return cfg->virt + (PCI_MMCFG_BUS_OFFSET(bus) | (devfn << 12));
2562306a36Sopenharmony_ci	return NULL;
2662306a36Sopenharmony_ci}
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_cistatic int pci_mmcfg_read(unsigned int seg, unsigned int bus,
2962306a36Sopenharmony_ci			  unsigned int devfn, int reg, int len, u32 *value)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	char __iomem *addr;
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	/* Why do we have this when nobody checks it. How about a BUG()!? -AK */
3462306a36Sopenharmony_ci	if (unlikely((bus > 255) || (devfn > 255) || (reg > 4095))) {
3562306a36Sopenharmony_cierr:		*value = -1;
3662306a36Sopenharmony_ci		return -EINVAL;
3762306a36Sopenharmony_ci	}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ci	rcu_read_lock();
4062306a36Sopenharmony_ci	addr = pci_dev_base(seg, bus, devfn);
4162306a36Sopenharmony_ci	if (!addr) {
4262306a36Sopenharmony_ci		rcu_read_unlock();
4362306a36Sopenharmony_ci		goto err;
4462306a36Sopenharmony_ci	}
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ci	switch (len) {
4762306a36Sopenharmony_ci	case 1:
4862306a36Sopenharmony_ci		*value = mmio_config_readb(addr + reg);
4962306a36Sopenharmony_ci		break;
5062306a36Sopenharmony_ci	case 2:
5162306a36Sopenharmony_ci		*value = mmio_config_readw(addr + reg);
5262306a36Sopenharmony_ci		break;
5362306a36Sopenharmony_ci	case 4:
5462306a36Sopenharmony_ci		*value = mmio_config_readl(addr + reg);
5562306a36Sopenharmony_ci		break;
5662306a36Sopenharmony_ci	}
5762306a36Sopenharmony_ci	rcu_read_unlock();
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci	return 0;
6062306a36Sopenharmony_ci}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_cistatic int pci_mmcfg_write(unsigned int seg, unsigned int bus,
6362306a36Sopenharmony_ci			   unsigned int devfn, int reg, int len, u32 value)
6462306a36Sopenharmony_ci{
6562306a36Sopenharmony_ci	char __iomem *addr;
6662306a36Sopenharmony_ci
6762306a36Sopenharmony_ci	/* Why do we have this when nobody checks it. How about a BUG()!? -AK */
6862306a36Sopenharmony_ci	if (unlikely((bus > 255) || (devfn > 255) || (reg > 4095)))
6962306a36Sopenharmony_ci		return -EINVAL;
7062306a36Sopenharmony_ci
7162306a36Sopenharmony_ci	rcu_read_lock();
7262306a36Sopenharmony_ci	addr = pci_dev_base(seg, bus, devfn);
7362306a36Sopenharmony_ci	if (!addr) {
7462306a36Sopenharmony_ci		rcu_read_unlock();
7562306a36Sopenharmony_ci		return -EINVAL;
7662306a36Sopenharmony_ci	}
7762306a36Sopenharmony_ci
7862306a36Sopenharmony_ci	switch (len) {
7962306a36Sopenharmony_ci	case 1:
8062306a36Sopenharmony_ci		mmio_config_writeb(addr + reg, value);
8162306a36Sopenharmony_ci		break;
8262306a36Sopenharmony_ci	case 2:
8362306a36Sopenharmony_ci		mmio_config_writew(addr + reg, value);
8462306a36Sopenharmony_ci		break;
8562306a36Sopenharmony_ci	case 4:
8662306a36Sopenharmony_ci		mmio_config_writel(addr + reg, value);
8762306a36Sopenharmony_ci		break;
8862306a36Sopenharmony_ci	}
8962306a36Sopenharmony_ci	rcu_read_unlock();
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ci	return 0;
9262306a36Sopenharmony_ci}
9362306a36Sopenharmony_ci
9462306a36Sopenharmony_ciconst struct pci_raw_ops pci_mmcfg = {
9562306a36Sopenharmony_ci	.read =		pci_mmcfg_read,
9662306a36Sopenharmony_ci	.write =	pci_mmcfg_write,
9762306a36Sopenharmony_ci};
9862306a36Sopenharmony_ci
9962306a36Sopenharmony_cistatic void __iomem *mcfg_ioremap(struct pci_mmcfg_region *cfg)
10062306a36Sopenharmony_ci{
10162306a36Sopenharmony_ci	void __iomem *addr;
10262306a36Sopenharmony_ci	u64 start, size;
10362306a36Sopenharmony_ci	int num_buses;
10462306a36Sopenharmony_ci
10562306a36Sopenharmony_ci	start = cfg->address + PCI_MMCFG_BUS_OFFSET(cfg->start_bus);
10662306a36Sopenharmony_ci	num_buses = cfg->end_bus - cfg->start_bus + 1;
10762306a36Sopenharmony_ci	size = PCI_MMCFG_BUS_OFFSET(num_buses);
10862306a36Sopenharmony_ci	addr = ioremap(start, size);
10962306a36Sopenharmony_ci	if (addr)
11062306a36Sopenharmony_ci		addr -= PCI_MMCFG_BUS_OFFSET(cfg->start_bus);
11162306a36Sopenharmony_ci	return addr;
11262306a36Sopenharmony_ci}
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ciint __init pci_mmcfg_arch_init(void)
11562306a36Sopenharmony_ci{
11662306a36Sopenharmony_ci	struct pci_mmcfg_region *cfg;
11762306a36Sopenharmony_ci
11862306a36Sopenharmony_ci	list_for_each_entry(cfg, &pci_mmcfg_list, list)
11962306a36Sopenharmony_ci		if (pci_mmcfg_arch_map(cfg)) {
12062306a36Sopenharmony_ci			pci_mmcfg_arch_free();
12162306a36Sopenharmony_ci			return 0;
12262306a36Sopenharmony_ci		}
12362306a36Sopenharmony_ci
12462306a36Sopenharmony_ci	raw_pci_ext_ops = &pci_mmcfg;
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_ci	return 1;
12762306a36Sopenharmony_ci}
12862306a36Sopenharmony_ci
12962306a36Sopenharmony_civoid __init pci_mmcfg_arch_free(void)
13062306a36Sopenharmony_ci{
13162306a36Sopenharmony_ci	struct pci_mmcfg_region *cfg;
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	list_for_each_entry(cfg, &pci_mmcfg_list, list)
13462306a36Sopenharmony_ci		pci_mmcfg_arch_unmap(cfg);
13562306a36Sopenharmony_ci}
13662306a36Sopenharmony_ci
13762306a36Sopenharmony_ciint pci_mmcfg_arch_map(struct pci_mmcfg_region *cfg)
13862306a36Sopenharmony_ci{
13962306a36Sopenharmony_ci	cfg->virt = mcfg_ioremap(cfg);
14062306a36Sopenharmony_ci	if (!cfg->virt) {
14162306a36Sopenharmony_ci		pr_err(PREFIX "can't map MMCONFIG at %pR\n", &cfg->res);
14262306a36Sopenharmony_ci		return -ENOMEM;
14362306a36Sopenharmony_ci	}
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	return 0;
14662306a36Sopenharmony_ci}
14762306a36Sopenharmony_ci
14862306a36Sopenharmony_civoid pci_mmcfg_arch_unmap(struct pci_mmcfg_region *cfg)
14962306a36Sopenharmony_ci{
15062306a36Sopenharmony_ci	if (cfg && cfg->virt) {
15162306a36Sopenharmony_ci		iounmap(cfg->virt + PCI_MMCFG_BUS_OFFSET(cfg->start_bus));
15262306a36Sopenharmony_ci		cfg->virt = NULL;
15362306a36Sopenharmony_ci	}
15462306a36Sopenharmony_ci}
155