Linux Audio

Check our new training course

Loading...
v6.8
   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Rockchip AXI PCIe host controller driver
   4 *
   5 * Copyright (c) 2016 Rockchip, Inc.
   6 *
   7 * Author: Shawn Lin <shawn.lin@rock-chips.com>
   8 *         Wenrui Li <wenrui.li@rock-chips.com>
   9 *
  10 * Bits taken from Synopsys DesignWare Host controller driver and
  11 * ARM PCI Host generic driver.
  12 */
  13
  14#include <linux/bitrev.h>
  15#include <linux/clk.h>
  16#include <linux/delay.h>
  17#include <linux/gpio/consumer.h>
  18#include <linux/init.h>
  19#include <linux/interrupt.h>
  20#include <linux/iopoll.h>
  21#include <linux/irq.h>
  22#include <linux/irqchip/chained_irq.h>
  23#include <linux/irqdomain.h>
  24#include <linux/kernel.h>
  25#include <linux/mfd/syscon.h>
  26#include <linux/module.h>
  27#include <linux/of.h>
 
  28#include <linux/of_pci.h>
 
 
  29#include <linux/pci.h>
  30#include <linux/pci_ids.h>
  31#include <linux/phy/phy.h>
  32#include <linux/platform_device.h>
  33#include <linux/reset.h>
  34#include <linux/regmap.h>
  35
  36#include "../pci.h"
  37#include "pcie-rockchip.h"
  38
  39static void rockchip_pcie_enable_bw_int(struct rockchip_pcie *rockchip)
  40{
  41	u32 status;
  42
  43	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
  44	status |= (PCI_EXP_LNKCTL_LBMIE | PCI_EXP_LNKCTL_LABIE);
  45	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
  46}
  47
  48static void rockchip_pcie_clr_bw_int(struct rockchip_pcie *rockchip)
  49{
  50	u32 status;
  51
  52	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
  53	status |= (PCI_EXP_LNKSTA_LBMS | PCI_EXP_LNKSTA_LABS) << 16;
  54	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
  55}
  56
  57static void rockchip_pcie_update_txcredit_mui(struct rockchip_pcie *rockchip)
  58{
  59	u32 val;
  60
  61	/* Update Tx credit maximum update interval */
  62	val = rockchip_pcie_read(rockchip, PCIE_CORE_TXCREDIT_CFG1);
  63	val &= ~PCIE_CORE_TXCREDIT_CFG1_MUI_MASK;
  64	val |= PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(24000);	/* ns */
  65	rockchip_pcie_write(rockchip, val, PCIE_CORE_TXCREDIT_CFG1);
  66}
  67
  68static int rockchip_pcie_valid_device(struct rockchip_pcie *rockchip,
  69				      struct pci_bus *bus, int dev)
  70{
  71	/*
  72	 * Access only one slot on each root port.
  73	 * Do not read more than one device on the bus directly attached
  74	 * to RC's downstream side.
  75	 */
  76	if (pci_is_root_bus(bus) || pci_is_root_bus(bus->parent))
  77		return dev == 0;
  78
  79	return 1;
  80}
  81
  82static u8 rockchip_pcie_lane_map(struct rockchip_pcie *rockchip)
  83{
  84	u32 val;
  85	u8 map;
  86
  87	if (rockchip->legacy_phy)
  88		return GENMASK(MAX_LANE_NUM - 1, 0);
  89
  90	val = rockchip_pcie_read(rockchip, PCIE_CORE_LANE_MAP);
  91	map = val & PCIE_CORE_LANE_MAP_MASK;
  92
  93	/* The link may be using a reverse-indexed mapping. */
  94	if (val & PCIE_CORE_LANE_MAP_REVERSE)
  95		map = bitrev8(map) >> 4;
  96
  97	return map;
  98}
  99
 100static int rockchip_pcie_rd_own_conf(struct rockchip_pcie *rockchip,
 101				     int where, int size, u32 *val)
 102{
 103	void __iomem *addr;
 104
 105	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + where;
 106
 107	if (!IS_ALIGNED((uintptr_t)addr, size)) {
 108		*val = 0;
 109		return PCIBIOS_BAD_REGISTER_NUMBER;
 110	}
 111
 112	if (size == 4) {
 113		*val = readl(addr);
 114	} else if (size == 2) {
 115		*val = readw(addr);
 116	} else if (size == 1) {
 117		*val = readb(addr);
 118	} else {
 119		*val = 0;
 120		return PCIBIOS_BAD_REGISTER_NUMBER;
 121	}
 122	return PCIBIOS_SUCCESSFUL;
 123}
 124
 125static int rockchip_pcie_wr_own_conf(struct rockchip_pcie *rockchip,
 126				     int where, int size, u32 val)
 127{
 128	u32 mask, tmp, offset;
 129	void __iomem *addr;
 130
 131	offset = where & ~0x3;
 132	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + offset;
 133
 134	if (size == 4) {
 135		writel(val, addr);
 136		return PCIBIOS_SUCCESSFUL;
 137	}
 138
 139	mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
 140
 141	/*
 142	 * N.B. This read/modify/write isn't safe in general because it can
 143	 * corrupt RW1C bits in adjacent registers.  But the hardware
 144	 * doesn't support smaller writes.
 145	 */
 146	tmp = readl(addr) & mask;
 147	tmp |= val << ((where & 0x3) * 8);
 148	writel(tmp, addr);
 149
 150	return PCIBIOS_SUCCESSFUL;
 151}
 152
 153static int rockchip_pcie_rd_other_conf(struct rockchip_pcie *rockchip,
 154				       struct pci_bus *bus, u32 devfn,
 155				       int where, int size, u32 *val)
 156{
 157	void __iomem *addr;
 158
 159	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
 160
 161	if (!IS_ALIGNED((uintptr_t)addr, size)) {
 162		*val = 0;
 163		return PCIBIOS_BAD_REGISTER_NUMBER;
 164	}
 165
 166	if (pci_is_root_bus(bus->parent))
 167		rockchip_pcie_cfg_configuration_accesses(rockchip,
 168						AXI_WRAPPER_TYPE0_CFG);
 169	else
 170		rockchip_pcie_cfg_configuration_accesses(rockchip,
 171						AXI_WRAPPER_TYPE1_CFG);
 172
 173	if (size == 4) {
 174		*val = readl(addr);
 175	} else if (size == 2) {
 176		*val = readw(addr);
 177	} else if (size == 1) {
 178		*val = readb(addr);
 179	} else {
 180		*val = 0;
 181		return PCIBIOS_BAD_REGISTER_NUMBER;
 182	}
 183	return PCIBIOS_SUCCESSFUL;
 184}
 185
 186static int rockchip_pcie_wr_other_conf(struct rockchip_pcie *rockchip,
 187				       struct pci_bus *bus, u32 devfn,
 188				       int where, int size, u32 val)
 189{
 190	void __iomem *addr;
 191
 192	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
 193
 194	if (!IS_ALIGNED((uintptr_t)addr, size))
 195		return PCIBIOS_BAD_REGISTER_NUMBER;
 196
 197	if (pci_is_root_bus(bus->parent))
 198		rockchip_pcie_cfg_configuration_accesses(rockchip,
 199						AXI_WRAPPER_TYPE0_CFG);
 200	else
 201		rockchip_pcie_cfg_configuration_accesses(rockchip,
 202						AXI_WRAPPER_TYPE1_CFG);
 203
 204	if (size == 4)
 205		writel(val, addr);
 206	else if (size == 2)
 207		writew(val, addr);
 208	else if (size == 1)
 209		writeb(val, addr);
 210	else
 211		return PCIBIOS_BAD_REGISTER_NUMBER;
 212
 213	return PCIBIOS_SUCCESSFUL;
 214}
 215
 216static int rockchip_pcie_rd_conf(struct pci_bus *bus, u32 devfn, int where,
 217				 int size, u32 *val)
 218{
 219	struct rockchip_pcie *rockchip = bus->sysdata;
 220
 221	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
 
 222		return PCIBIOS_DEVICE_NOT_FOUND;
 
 223
 224	if (pci_is_root_bus(bus))
 225		return rockchip_pcie_rd_own_conf(rockchip, where, size, val);
 226
 227	return rockchip_pcie_rd_other_conf(rockchip, bus, devfn, where, size,
 228					   val);
 229}
 230
 231static int rockchip_pcie_wr_conf(struct pci_bus *bus, u32 devfn,
 232				 int where, int size, u32 val)
 233{
 234	struct rockchip_pcie *rockchip = bus->sysdata;
 235
 236	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
 237		return PCIBIOS_DEVICE_NOT_FOUND;
 238
 239	if (pci_is_root_bus(bus))
 240		return rockchip_pcie_wr_own_conf(rockchip, where, size, val);
 241
 242	return rockchip_pcie_wr_other_conf(rockchip, bus, devfn, where, size,
 243					   val);
 244}
 245
 246static struct pci_ops rockchip_pcie_ops = {
 247	.read = rockchip_pcie_rd_conf,
 248	.write = rockchip_pcie_wr_conf,
 249};
 250
 251static void rockchip_pcie_set_power_limit(struct rockchip_pcie *rockchip)
 252{
 253	int curr;
 254	u32 status, scale, power;
 255
 256	if (IS_ERR(rockchip->vpcie3v3))
 257		return;
 258
 259	/*
 260	 * Set RC's captured slot power limit and scale if
 261	 * vpcie3v3 available. The default values are both zero
 262	 * which means the software should set these two according
 263	 * to the actual power supply.
 264	 */
 265	curr = regulator_get_current_limit(rockchip->vpcie3v3);
 266	if (curr <= 0)
 267		return;
 268
 269	scale = 3; /* 0.001x */
 270	curr = curr / 1000; /* convert to mA */
 271	power = (curr * 3300) / 1000; /* milliwatt */
 272	while (power > PCIE_RC_CONFIG_DCR_CSPL_LIMIT) {
 273		if (!scale) {
 274			dev_warn(rockchip->dev, "invalid power supply\n");
 275			return;
 276		}
 277		scale--;
 278		power = power / 10;
 279	}
 280
 281	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCR);
 282	status |= (power << PCIE_RC_CONFIG_DCR_CSPL_SHIFT) |
 283		  (scale << PCIE_RC_CONFIG_DCR_CPLS_SHIFT);
 284	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCR);
 285}
 286
 287/**
 288 * rockchip_pcie_host_init_port - Initialize hardware
 289 * @rockchip: PCIe port information
 290 */
 291static int rockchip_pcie_host_init_port(struct rockchip_pcie *rockchip)
 292{
 293	struct device *dev = rockchip->dev;
 294	int err, i = MAX_LANE_NUM;
 295	u32 status;
 296
 297	gpiod_set_value_cansleep(rockchip->ep_gpio, 0);
 298
 299	err = rockchip_pcie_init_port(rockchip);
 300	if (err)
 301		return err;
 302
 303	/* Fix the transmitted FTS count desired to exit from L0s. */
 304	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL_PLC1);
 305	status = (status & ~PCIE_CORE_CTRL_PLC1_FTS_MASK) |
 306		 (PCIE_CORE_CTRL_PLC1_FTS_CNT << PCIE_CORE_CTRL_PLC1_FTS_SHIFT);
 307	rockchip_pcie_write(rockchip, status, PCIE_CORE_CTRL_PLC1);
 308
 309	rockchip_pcie_set_power_limit(rockchip);
 310
 311	/* Set RC's clock architecture as common clock */
 312	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 313	status |= PCI_EXP_LNKSTA_SLC << 16;
 314	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 315
 316	/* Set RC's RCB to 128 */
 317	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 318	status |= PCI_EXP_LNKCTL_RCB;
 319	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 320
 321	/* Enable Gen1 training */
 322	rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
 323			    PCIE_CLIENT_CONFIG);
 324
 325	gpiod_set_value_cansleep(rockchip->ep_gpio, 1);
 326
 327	/* 500ms timeout value should be enough for Gen1/2 training */
 328	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_BASIC_STATUS1,
 329				 status, PCIE_LINK_UP(status), 20,
 330				 500 * USEC_PER_MSEC);
 331	if (err) {
 332		dev_err(dev, "PCIe link training gen1 timeout!\n");
 333		goto err_power_off_phy;
 334	}
 335
 336	if (rockchip->link_gen == 2) {
 337		/*
 338		 * Enable retrain for gen2. This should be configured only after
 339		 * gen1 finished.
 340		 */
 341		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 342		status |= PCI_EXP_LNKCTL_RL;
 343		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 344
 345		err = readl_poll_timeout(rockchip->apb_base + PCIE_CORE_CTRL,
 346					 status, PCIE_LINK_IS_GEN2(status), 20,
 347					 500 * USEC_PER_MSEC);
 348		if (err)
 349			dev_dbg(dev, "PCIe link training gen2 timeout, fall back to gen1!\n");
 350	}
 351
 352	/* Check the final link width from negotiated lane counter from MGMT */
 353	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL);
 354	status = 0x1 << ((status & PCIE_CORE_PL_CONF_LANE_MASK) >>
 355			  PCIE_CORE_PL_CONF_LANE_SHIFT);
 356	dev_dbg(dev, "current link width is x%d\n", status);
 357
 358	/* Power off unused lane(s) */
 359	rockchip->lanes_map = rockchip_pcie_lane_map(rockchip);
 360	for (i = 0; i < MAX_LANE_NUM; i++) {
 361		if (!(rockchip->lanes_map & BIT(i))) {
 362			dev_dbg(dev, "idling lane %d\n", i);
 363			phy_power_off(rockchip->phys[i]);
 364		}
 365	}
 366
 367	rockchip_pcie_write(rockchip, ROCKCHIP_VENDOR_ID,
 368			    PCIE_CORE_CONFIG_VENDOR);
 369	rockchip_pcie_write(rockchip,
 370			    PCI_CLASS_BRIDGE_PCI_NORMAL << 8,
 371			    PCIE_RC_CONFIG_RID_CCR);
 372
 373	/* Clear THP cap's next cap pointer to remove L1 substate cap */
 374	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_THP_CAP);
 375	status &= ~PCIE_RC_CONFIG_THP_CAP_NEXT_MASK;
 376	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_THP_CAP);
 377
 378	/* Clear L0s from RC's link cap */
 379	if (of_property_read_bool(dev->of_node, "aspm-no-l0s")) {
 380		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LINK_CAP);
 381		status &= ~PCIE_RC_CONFIG_LINK_CAP_L0S;
 382		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LINK_CAP);
 383	}
 384
 385	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCSR);
 386	status &= ~PCIE_RC_CONFIG_DCSR_MPS_MASK;
 387	status |= PCIE_RC_CONFIG_DCSR_MPS_256;
 388	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCSR);
 389
 390	return 0;
 391err_power_off_phy:
 392	while (i--)
 393		phy_power_off(rockchip->phys[i]);
 394	i = MAX_LANE_NUM;
 395	while (i--)
 396		phy_exit(rockchip->phys[i]);
 397	return err;
 398}
 399
 400static irqreturn_t rockchip_pcie_subsys_irq_handler(int irq, void *arg)
 401{
 402	struct rockchip_pcie *rockchip = arg;
 403	struct device *dev = rockchip->dev;
 404	u32 reg;
 405	u32 sub_reg;
 406
 407	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 408	if (reg & PCIE_CLIENT_INT_LOCAL) {
 409		dev_dbg(dev, "local interrupt received\n");
 410		sub_reg = rockchip_pcie_read(rockchip, PCIE_CORE_INT_STATUS);
 411		if (sub_reg & PCIE_CORE_INT_PRFPE)
 412			dev_dbg(dev, "parity error detected while reading from the PNP receive FIFO RAM\n");
 413
 414		if (sub_reg & PCIE_CORE_INT_CRFPE)
 415			dev_dbg(dev, "parity error detected while reading from the Completion Receive FIFO RAM\n");
 416
 417		if (sub_reg & PCIE_CORE_INT_RRPE)
 418			dev_dbg(dev, "parity error detected while reading from replay buffer RAM\n");
 419
 420		if (sub_reg & PCIE_CORE_INT_PRFO)
 421			dev_dbg(dev, "overflow occurred in the PNP receive FIFO\n");
 422
 423		if (sub_reg & PCIE_CORE_INT_CRFO)
 424			dev_dbg(dev, "overflow occurred in the completion receive FIFO\n");
 425
 426		if (sub_reg & PCIE_CORE_INT_RT)
 427			dev_dbg(dev, "replay timer timed out\n");
 428
 429		if (sub_reg & PCIE_CORE_INT_RTR)
 430			dev_dbg(dev, "replay timer rolled over after 4 transmissions of the same TLP\n");
 431
 432		if (sub_reg & PCIE_CORE_INT_PE)
 433			dev_dbg(dev, "phy error detected on receive side\n");
 434
 435		if (sub_reg & PCIE_CORE_INT_MTR)
 436			dev_dbg(dev, "malformed TLP received from the link\n");
 437
 438		if (sub_reg & PCIE_CORE_INT_UCR)
 439			dev_dbg(dev, "malformed TLP received from the link\n");
 440
 441		if (sub_reg & PCIE_CORE_INT_FCE)
 442			dev_dbg(dev, "an error was observed in the flow control advertisements from the other side\n");
 443
 444		if (sub_reg & PCIE_CORE_INT_CT)
 445			dev_dbg(dev, "a request timed out waiting for completion\n");
 446
 447		if (sub_reg & PCIE_CORE_INT_UTC)
 448			dev_dbg(dev, "unmapped TC error\n");
 449
 450		if (sub_reg & PCIE_CORE_INT_MMVC)
 451			dev_dbg(dev, "MSI mask register changes\n");
 452
 453		rockchip_pcie_write(rockchip, sub_reg, PCIE_CORE_INT_STATUS);
 454	} else if (reg & PCIE_CLIENT_INT_PHY) {
 455		dev_dbg(dev, "phy link changes\n");
 456		rockchip_pcie_update_txcredit_mui(rockchip);
 457		rockchip_pcie_clr_bw_int(rockchip);
 458	}
 459
 460	rockchip_pcie_write(rockchip, reg & PCIE_CLIENT_INT_LOCAL,
 461			    PCIE_CLIENT_INT_STATUS);
 462
 463	return IRQ_HANDLED;
 464}
 465
 466static irqreturn_t rockchip_pcie_client_irq_handler(int irq, void *arg)
 467{
 468	struct rockchip_pcie *rockchip = arg;
 469	struct device *dev = rockchip->dev;
 470	u32 reg;
 471
 472	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 473	if (reg & PCIE_CLIENT_INT_LEGACY_DONE)
 474		dev_dbg(dev, "legacy done interrupt received\n");
 475
 476	if (reg & PCIE_CLIENT_INT_MSG)
 477		dev_dbg(dev, "message done interrupt received\n");
 478
 479	if (reg & PCIE_CLIENT_INT_HOT_RST)
 480		dev_dbg(dev, "hot reset interrupt received\n");
 481
 482	if (reg & PCIE_CLIENT_INT_DPA)
 483		dev_dbg(dev, "dpa interrupt received\n");
 484
 485	if (reg & PCIE_CLIENT_INT_FATAL_ERR)
 486		dev_dbg(dev, "fatal error interrupt received\n");
 487
 488	if (reg & PCIE_CLIENT_INT_NFATAL_ERR)
 489		dev_dbg(dev, "no fatal error interrupt received\n");
 490
 491	if (reg & PCIE_CLIENT_INT_CORR_ERR)
 492		dev_dbg(dev, "correctable error interrupt received\n");
 493
 494	if (reg & PCIE_CLIENT_INT_PHY)
 495		dev_dbg(dev, "phy interrupt received\n");
 496
 497	rockchip_pcie_write(rockchip, reg & (PCIE_CLIENT_INT_LEGACY_DONE |
 498			      PCIE_CLIENT_INT_MSG | PCIE_CLIENT_INT_HOT_RST |
 499			      PCIE_CLIENT_INT_DPA | PCIE_CLIENT_INT_FATAL_ERR |
 500			      PCIE_CLIENT_INT_NFATAL_ERR |
 501			      PCIE_CLIENT_INT_CORR_ERR |
 502			      PCIE_CLIENT_INT_PHY),
 503		   PCIE_CLIENT_INT_STATUS);
 504
 505	return IRQ_HANDLED;
 506}
 507
 508static void rockchip_pcie_intx_handler(struct irq_desc *desc)
 509{
 510	struct irq_chip *chip = irq_desc_get_chip(desc);
 511	struct rockchip_pcie *rockchip = irq_desc_get_handler_data(desc);
 512	struct device *dev = rockchip->dev;
 513	u32 reg;
 514	u32 hwirq;
 515	int ret;
 516
 517	chained_irq_enter(chip, desc);
 518
 519	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 520	reg = (reg & PCIE_CLIENT_INTR_MASK) >> PCIE_CLIENT_INTR_SHIFT;
 521
 522	while (reg) {
 523		hwirq = ffs(reg) - 1;
 524		reg &= ~BIT(hwirq);
 525
 526		ret = generic_handle_domain_irq(rockchip->irq_domain, hwirq);
 527		if (ret)
 
 
 528			dev_err(dev, "unexpected IRQ, INT%d\n", hwirq);
 529	}
 530
 531	chained_irq_exit(chip, desc);
 532}
 533
 534static int rockchip_pcie_setup_irq(struct rockchip_pcie *rockchip)
 535{
 536	int irq, err;
 537	struct device *dev = rockchip->dev;
 538	struct platform_device *pdev = to_platform_device(dev);
 539
 540	irq = platform_get_irq_byname(pdev, "sys");
 541	if (irq < 0)
 542		return irq;
 543
 544	err = devm_request_irq(dev, irq, rockchip_pcie_subsys_irq_handler,
 545			       IRQF_SHARED, "pcie-sys", rockchip);
 546	if (err) {
 547		dev_err(dev, "failed to request PCIe subsystem IRQ\n");
 548		return err;
 549	}
 550
 551	irq = platform_get_irq_byname(pdev, "legacy");
 552	if (irq < 0)
 553		return irq;
 554
 555	irq_set_chained_handler_and_data(irq,
 556					 rockchip_pcie_intx_handler,
 557					 rockchip);
 558
 559	irq = platform_get_irq_byname(pdev, "client");
 560	if (irq < 0)
 561		return irq;
 562
 563	err = devm_request_irq(dev, irq, rockchip_pcie_client_irq_handler,
 564			       IRQF_SHARED, "pcie-client", rockchip);
 565	if (err) {
 566		dev_err(dev, "failed to request PCIe client IRQ\n");
 567		return err;
 568	}
 569
 570	return 0;
 571}
 572
 573/**
 574 * rockchip_pcie_parse_host_dt - Parse Device Tree
 575 * @rockchip: PCIe port information
 576 *
 577 * Return: '0' on success and error value on failure
 578 */
 579static int rockchip_pcie_parse_host_dt(struct rockchip_pcie *rockchip)
 580{
 581	struct device *dev = rockchip->dev;
 582	int err;
 583
 584	err = rockchip_pcie_parse_dt(rockchip);
 585	if (err)
 586		return err;
 587
 588	rockchip->vpcie12v = devm_regulator_get_optional(dev, "vpcie12v");
 589	if (IS_ERR(rockchip->vpcie12v)) {
 590		if (PTR_ERR(rockchip->vpcie12v) != -ENODEV)
 591			return PTR_ERR(rockchip->vpcie12v);
 592		dev_info(dev, "no vpcie12v regulator found\n");
 593	}
 594
 595	rockchip->vpcie3v3 = devm_regulator_get_optional(dev, "vpcie3v3");
 596	if (IS_ERR(rockchip->vpcie3v3)) {
 597		if (PTR_ERR(rockchip->vpcie3v3) != -ENODEV)
 598			return PTR_ERR(rockchip->vpcie3v3);
 599		dev_info(dev, "no vpcie3v3 regulator found\n");
 600	}
 601
 602	rockchip->vpcie1v8 = devm_regulator_get(dev, "vpcie1v8");
 603	if (IS_ERR(rockchip->vpcie1v8))
 604		return PTR_ERR(rockchip->vpcie1v8);
 605
 606	rockchip->vpcie0v9 = devm_regulator_get(dev, "vpcie0v9");
 607	if (IS_ERR(rockchip->vpcie0v9))
 608		return PTR_ERR(rockchip->vpcie0v9);
 609
 610	return 0;
 611}
 612
 613static int rockchip_pcie_set_vpcie(struct rockchip_pcie *rockchip)
 614{
 615	struct device *dev = rockchip->dev;
 616	int err;
 617
 618	if (!IS_ERR(rockchip->vpcie12v)) {
 619		err = regulator_enable(rockchip->vpcie12v);
 620		if (err) {
 621			dev_err(dev, "fail to enable vpcie12v regulator\n");
 622			goto err_out;
 623		}
 624	}
 625
 626	if (!IS_ERR(rockchip->vpcie3v3)) {
 627		err = regulator_enable(rockchip->vpcie3v3);
 628		if (err) {
 629			dev_err(dev, "fail to enable vpcie3v3 regulator\n");
 630			goto err_disable_12v;
 631		}
 632	}
 633
 634	err = regulator_enable(rockchip->vpcie1v8);
 635	if (err) {
 636		dev_err(dev, "fail to enable vpcie1v8 regulator\n");
 637		goto err_disable_3v3;
 638	}
 639
 640	err = regulator_enable(rockchip->vpcie0v9);
 641	if (err) {
 642		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
 643		goto err_disable_1v8;
 644	}
 645
 646	return 0;
 647
 648err_disable_1v8:
 649	regulator_disable(rockchip->vpcie1v8);
 650err_disable_3v3:
 651	if (!IS_ERR(rockchip->vpcie3v3))
 652		regulator_disable(rockchip->vpcie3v3);
 653err_disable_12v:
 654	if (!IS_ERR(rockchip->vpcie12v))
 655		regulator_disable(rockchip->vpcie12v);
 656err_out:
 657	return err;
 658}
 659
 660static void rockchip_pcie_enable_interrupts(struct rockchip_pcie *rockchip)
 661{
 662	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) &
 663			    (~PCIE_CLIENT_INT_CLI), PCIE_CLIENT_INT_MASK);
 664	rockchip_pcie_write(rockchip, (u32)(~PCIE_CORE_INT),
 665			    PCIE_CORE_INT_MASK);
 666
 667	rockchip_pcie_enable_bw_int(rockchip);
 668}
 669
 670static int rockchip_pcie_intx_map(struct irq_domain *domain, unsigned int irq,
 671				  irq_hw_number_t hwirq)
 672{
 673	irq_set_chip_and_handler(irq, &dummy_irq_chip, handle_simple_irq);
 674	irq_set_chip_data(irq, domain->host_data);
 675
 676	return 0;
 677}
 678
 679static const struct irq_domain_ops intx_domain_ops = {
 680	.map = rockchip_pcie_intx_map,
 681};
 682
 683static int rockchip_pcie_init_irq_domain(struct rockchip_pcie *rockchip)
 684{
 685	struct device *dev = rockchip->dev;
 686	struct device_node *intc = of_get_next_child(dev->of_node, NULL);
 687
 688	if (!intc) {
 689		dev_err(dev, "missing child interrupt-controller node\n");
 690		return -EINVAL;
 691	}
 692
 693	rockchip->irq_domain = irq_domain_add_linear(intc, PCI_NUM_INTX,
 694						    &intx_domain_ops, rockchip);
 695	of_node_put(intc);
 696	if (!rockchip->irq_domain) {
 697		dev_err(dev, "failed to get a INTx IRQ domain\n");
 698		return -EINVAL;
 699	}
 700
 701	return 0;
 702}
 703
 704static int rockchip_pcie_prog_ob_atu(struct rockchip_pcie *rockchip,
 705				     int region_no, int type, u8 num_pass_bits,
 706				     u32 lower_addr, u32 upper_addr)
 707{
 708	u32 ob_addr_0;
 709	u32 ob_addr_1;
 710	u32 ob_desc_0;
 711	u32 aw_offset;
 712
 713	if (region_no >= MAX_AXI_WRAPPER_REGION_NUM)
 714		return -EINVAL;
 715	if (num_pass_bits + 1 < 8)
 716		return -EINVAL;
 717	if (num_pass_bits > 63)
 718		return -EINVAL;
 719	if (region_no == 0) {
 720		if (AXI_REGION_0_SIZE < (2ULL << num_pass_bits))
 721			return -EINVAL;
 722	}
 723	if (region_no != 0) {
 724		if (AXI_REGION_SIZE < (2ULL << num_pass_bits))
 725			return -EINVAL;
 726	}
 727
 728	aw_offset = (region_no << OB_REG_SIZE_SHIFT);
 729
 730	ob_addr_0 = num_pass_bits & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS;
 731	ob_addr_0 |= lower_addr & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR;
 732	ob_addr_1 = upper_addr;
 733	ob_desc_0 = (1 << 23 | type);
 734
 735	rockchip_pcie_write(rockchip, ob_addr_0,
 736			    PCIE_CORE_OB_REGION_ADDR0 + aw_offset);
 737	rockchip_pcie_write(rockchip, ob_addr_1,
 738			    PCIE_CORE_OB_REGION_ADDR1 + aw_offset);
 739	rockchip_pcie_write(rockchip, ob_desc_0,
 740			    PCIE_CORE_OB_REGION_DESC0 + aw_offset);
 741	rockchip_pcie_write(rockchip, 0,
 742			    PCIE_CORE_OB_REGION_DESC1 + aw_offset);
 743
 744	return 0;
 745}
 746
 747static int rockchip_pcie_prog_ib_atu(struct rockchip_pcie *rockchip,
 748				     int region_no, u8 num_pass_bits,
 749				     u32 lower_addr, u32 upper_addr)
 750{
 751	u32 ib_addr_0;
 752	u32 ib_addr_1;
 753	u32 aw_offset;
 754
 755	if (region_no > MAX_AXI_IB_ROOTPORT_REGION_NUM)
 756		return -EINVAL;
 757	if (num_pass_bits + 1 < MIN_AXI_ADDR_BITS_PASSED)
 758		return -EINVAL;
 759	if (num_pass_bits > 63)
 760		return -EINVAL;
 761
 762	aw_offset = (region_no << IB_ROOT_PORT_REG_SIZE_SHIFT);
 763
 764	ib_addr_0 = num_pass_bits & PCIE_CORE_IB_REGION_ADDR0_NUM_BITS;
 765	ib_addr_0 |= (lower_addr << 8) & PCIE_CORE_IB_REGION_ADDR0_LO_ADDR;
 766	ib_addr_1 = upper_addr;
 767
 768	rockchip_pcie_write(rockchip, ib_addr_0, PCIE_RP_IB_ADDR0 + aw_offset);
 769	rockchip_pcie_write(rockchip, ib_addr_1, PCIE_RP_IB_ADDR1 + aw_offset);
 770
 771	return 0;
 772}
 773
 774static int rockchip_pcie_cfg_atu(struct rockchip_pcie *rockchip)
 775{
 776	struct device *dev = rockchip->dev;
 777	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
 778	struct resource_entry *entry;
 779	u64 pci_addr, size;
 780	int offset;
 781	int err;
 782	int reg_no;
 783
 784	rockchip_pcie_cfg_configuration_accesses(rockchip,
 785						 AXI_WRAPPER_TYPE0_CFG);
 786	entry = resource_list_first_type(&bridge->windows, IORESOURCE_MEM);
 787	if (!entry)
 788		return -ENODEV;
 789
 790	size = resource_size(entry->res);
 791	pci_addr = entry->res->start - entry->offset;
 792	rockchip->msg_bus_addr = pci_addr;
 793
 794	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
 795		err = rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1,
 796						AXI_WRAPPER_MEM_WRITE,
 797						20 - 1,
 798						pci_addr + (reg_no << 20),
 799						0);
 800		if (err) {
 801			dev_err(dev, "program RC mem outbound ATU failed\n");
 802			return err;
 803		}
 804	}
 805
 806	err = rockchip_pcie_prog_ib_atu(rockchip, 2, 32 - 1, 0x0, 0);
 807	if (err) {
 808		dev_err(dev, "program RC mem inbound ATU failed\n");
 809		return err;
 810	}
 811
 812	entry = resource_list_first_type(&bridge->windows, IORESOURCE_IO);
 813	if (!entry)
 814		return -ENODEV;
 815
 816	/* store the register number offset to program RC io outbound ATU */
 817	offset = size >> 20;
 818
 819	size = resource_size(entry->res);
 820	pci_addr = entry->res->start - entry->offset;
 821
 822	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
 823		err = rockchip_pcie_prog_ob_atu(rockchip,
 824						reg_no + 1 + offset,
 825						AXI_WRAPPER_IO_WRITE,
 826						20 - 1,
 827						pci_addr + (reg_no << 20),
 828						0);
 829		if (err) {
 830			dev_err(dev, "program RC io outbound ATU failed\n");
 831			return err;
 832		}
 833	}
 834
 835	/* assign message regions */
 836	rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1 + offset,
 837				  AXI_WRAPPER_NOR_MSG,
 838				  20 - 1, 0, 0);
 839
 840	rockchip->msg_bus_addr += ((reg_no + offset) << 20);
 841	return err;
 842}
 843
 844static int rockchip_pcie_wait_l2(struct rockchip_pcie *rockchip)
 845{
 846	u32 value;
 847	int err;
 848
 849	/* send PME_TURN_OFF message */
 850	writel(0x0, rockchip->msg_region + PCIE_RC_SEND_PME_OFF);
 851
 852	/* read LTSSM and wait for falling into L2 link state */
 853	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_DEBUG_OUT_0,
 854				 value, PCIE_LINK_IS_L2(value), 20,
 855				 jiffies_to_usecs(5 * HZ));
 856	if (err) {
 857		dev_err(rockchip->dev, "PCIe link enter L2 timeout!\n");
 858		return err;
 859	}
 860
 861	return 0;
 862}
 863
 864static int rockchip_pcie_suspend_noirq(struct device *dev)
 865{
 866	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
 867	int ret;
 868
 869	/* disable core and cli int since we don't need to ack PME_ACK */
 870	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) |
 871			    PCIE_CLIENT_INT_CLI, PCIE_CLIENT_INT_MASK);
 872	rockchip_pcie_write(rockchip, (u32)PCIE_CORE_INT, PCIE_CORE_INT_MASK);
 873
 874	ret = rockchip_pcie_wait_l2(rockchip);
 875	if (ret) {
 876		rockchip_pcie_enable_interrupts(rockchip);
 877		return ret;
 878	}
 879
 880	rockchip_pcie_deinit_phys(rockchip);
 881
 882	rockchip_pcie_disable_clocks(rockchip);
 883
 884	regulator_disable(rockchip->vpcie0v9);
 885
 886	return ret;
 887}
 888
 889static int rockchip_pcie_resume_noirq(struct device *dev)
 890{
 891	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
 892	int err;
 893
 894	err = regulator_enable(rockchip->vpcie0v9);
 895	if (err) {
 896		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
 897		return err;
 898	}
 899
 900	err = rockchip_pcie_enable_clocks(rockchip);
 901	if (err)
 902		goto err_disable_0v9;
 903
 904	err = rockchip_pcie_host_init_port(rockchip);
 905	if (err)
 906		goto err_pcie_resume;
 907
 908	err = rockchip_pcie_cfg_atu(rockchip);
 909	if (err)
 910		goto err_err_deinit_port;
 911
 912	/* Need this to enter L1 again */
 913	rockchip_pcie_update_txcredit_mui(rockchip);
 914	rockchip_pcie_enable_interrupts(rockchip);
 915
 916	return 0;
 917
 918err_err_deinit_port:
 919	rockchip_pcie_deinit_phys(rockchip);
 920err_pcie_resume:
 921	rockchip_pcie_disable_clocks(rockchip);
 922err_disable_0v9:
 923	regulator_disable(rockchip->vpcie0v9);
 924	return err;
 925}
 926
 927static int rockchip_pcie_probe(struct platform_device *pdev)
 928{
 929	struct rockchip_pcie *rockchip;
 930	struct device *dev = &pdev->dev;
 931	struct pci_host_bridge *bridge;
 932	int err;
 933
 934	if (!dev->of_node)
 935		return -ENODEV;
 936
 937	bridge = devm_pci_alloc_host_bridge(dev, sizeof(*rockchip));
 938	if (!bridge)
 939		return -ENOMEM;
 940
 941	rockchip = pci_host_bridge_priv(bridge);
 942
 943	platform_set_drvdata(pdev, rockchip);
 944	rockchip->dev = dev;
 945	rockchip->is_rc = true;
 946
 947	err = rockchip_pcie_parse_host_dt(rockchip);
 948	if (err)
 949		return err;
 950
 951	err = rockchip_pcie_enable_clocks(rockchip);
 952	if (err)
 953		return err;
 954
 955	err = rockchip_pcie_set_vpcie(rockchip);
 956	if (err) {
 957		dev_err(dev, "failed to set vpcie regulator\n");
 958		goto err_set_vpcie;
 959	}
 960
 961	err = rockchip_pcie_host_init_port(rockchip);
 962	if (err)
 963		goto err_vpcie;
 964
 965	err = rockchip_pcie_init_irq_domain(rockchip);
 966	if (err < 0)
 967		goto err_deinit_port;
 968
 969	err = rockchip_pcie_cfg_atu(rockchip);
 970	if (err)
 971		goto err_remove_irq_domain;
 972
 973	rockchip->msg_region = devm_ioremap(dev, rockchip->msg_bus_addr, SZ_1M);
 974	if (!rockchip->msg_region) {
 975		err = -ENOMEM;
 976		goto err_remove_irq_domain;
 977	}
 978
 979	bridge->sysdata = rockchip;
 980	bridge->ops = &rockchip_pcie_ops;
 981
 982	err = rockchip_pcie_setup_irq(rockchip);
 983	if (err)
 984		goto err_remove_irq_domain;
 985
 986	rockchip_pcie_enable_interrupts(rockchip);
 987
 988	err = pci_host_probe(bridge);
 989	if (err < 0)
 990		goto err_remove_irq_domain;
 991
 992	return 0;
 993
 994err_remove_irq_domain:
 995	irq_domain_remove(rockchip->irq_domain);
 996err_deinit_port:
 997	rockchip_pcie_deinit_phys(rockchip);
 998err_vpcie:
 999	if (!IS_ERR(rockchip->vpcie12v))
1000		regulator_disable(rockchip->vpcie12v);
1001	if (!IS_ERR(rockchip->vpcie3v3))
1002		regulator_disable(rockchip->vpcie3v3);
1003	regulator_disable(rockchip->vpcie1v8);
1004	regulator_disable(rockchip->vpcie0v9);
1005err_set_vpcie:
1006	rockchip_pcie_disable_clocks(rockchip);
1007	return err;
1008}
1009
1010static void rockchip_pcie_remove(struct platform_device *pdev)
1011{
1012	struct device *dev = &pdev->dev;
1013	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
1014	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
1015
1016	pci_stop_root_bus(bridge->bus);
1017	pci_remove_root_bus(bridge->bus);
1018	irq_domain_remove(rockchip->irq_domain);
1019
1020	rockchip_pcie_deinit_phys(rockchip);
1021
1022	rockchip_pcie_disable_clocks(rockchip);
1023
1024	if (!IS_ERR(rockchip->vpcie12v))
1025		regulator_disable(rockchip->vpcie12v);
1026	if (!IS_ERR(rockchip->vpcie3v3))
1027		regulator_disable(rockchip->vpcie3v3);
1028	regulator_disable(rockchip->vpcie1v8);
1029	regulator_disable(rockchip->vpcie0v9);
 
 
1030}
1031
1032static const struct dev_pm_ops rockchip_pcie_pm_ops = {
1033	NOIRQ_SYSTEM_SLEEP_PM_OPS(rockchip_pcie_suspend_noirq,
1034				  rockchip_pcie_resume_noirq)
1035};
1036
1037static const struct of_device_id rockchip_pcie_of_match[] = {
1038	{ .compatible = "rockchip,rk3399-pcie", },
1039	{}
1040};
1041MODULE_DEVICE_TABLE(of, rockchip_pcie_of_match);
1042
1043static struct platform_driver rockchip_pcie_driver = {
1044	.driver = {
1045		.name = "rockchip-pcie",
1046		.of_match_table = rockchip_pcie_of_match,
1047		.pm = &rockchip_pcie_pm_ops,
1048	},
1049	.probe = rockchip_pcie_probe,
1050	.remove_new = rockchip_pcie_remove,
1051};
1052module_platform_driver(rockchip_pcie_driver);
1053
1054MODULE_AUTHOR("Rockchip Inc");
1055MODULE_DESCRIPTION("Rockchip AXI PCIe driver");
1056MODULE_LICENSE("GPL v2");
v5.14.15
   1// SPDX-License-Identifier: GPL-2.0+
   2/*
   3 * Rockchip AXI PCIe host controller driver
   4 *
   5 * Copyright (c) 2016 Rockchip, Inc.
   6 *
   7 * Author: Shawn Lin <shawn.lin@rock-chips.com>
   8 *         Wenrui Li <wenrui.li@rock-chips.com>
   9 *
  10 * Bits taken from Synopsys DesignWare Host controller driver and
  11 * ARM PCI Host generic driver.
  12 */
  13
  14#include <linux/bitrev.h>
  15#include <linux/clk.h>
  16#include <linux/delay.h>
  17#include <linux/gpio/consumer.h>
  18#include <linux/init.h>
  19#include <linux/interrupt.h>
  20#include <linux/iopoll.h>
  21#include <linux/irq.h>
  22#include <linux/irqchip/chained_irq.h>
  23#include <linux/irqdomain.h>
  24#include <linux/kernel.h>
  25#include <linux/mfd/syscon.h>
  26#include <linux/module.h>
  27#include <linux/of_address.h>
  28#include <linux/of_device.h>
  29#include <linux/of_pci.h>
  30#include <linux/of_platform.h>
  31#include <linux/of_irq.h>
  32#include <linux/pci.h>
  33#include <linux/pci_ids.h>
  34#include <linux/phy/phy.h>
  35#include <linux/platform_device.h>
  36#include <linux/reset.h>
  37#include <linux/regmap.h>
  38
  39#include "../pci.h"
  40#include "pcie-rockchip.h"
  41
  42static void rockchip_pcie_enable_bw_int(struct rockchip_pcie *rockchip)
  43{
  44	u32 status;
  45
  46	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
  47	status |= (PCI_EXP_LNKCTL_LBMIE | PCI_EXP_LNKCTL_LABIE);
  48	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
  49}
  50
  51static void rockchip_pcie_clr_bw_int(struct rockchip_pcie *rockchip)
  52{
  53	u32 status;
  54
  55	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
  56	status |= (PCI_EXP_LNKSTA_LBMS | PCI_EXP_LNKSTA_LABS) << 16;
  57	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
  58}
  59
  60static void rockchip_pcie_update_txcredit_mui(struct rockchip_pcie *rockchip)
  61{
  62	u32 val;
  63
  64	/* Update Tx credit maximum update interval */
  65	val = rockchip_pcie_read(rockchip, PCIE_CORE_TXCREDIT_CFG1);
  66	val &= ~PCIE_CORE_TXCREDIT_CFG1_MUI_MASK;
  67	val |= PCIE_CORE_TXCREDIT_CFG1_MUI_ENCODE(24000);	/* ns */
  68	rockchip_pcie_write(rockchip, val, PCIE_CORE_TXCREDIT_CFG1);
  69}
  70
  71static int rockchip_pcie_valid_device(struct rockchip_pcie *rockchip,
  72				      struct pci_bus *bus, int dev)
  73{
  74	/*
  75	 * Access only one slot on each root port.
  76	 * Do not read more than one device on the bus directly attached
  77	 * to RC's downstream side.
  78	 */
  79	if (pci_is_root_bus(bus) || pci_is_root_bus(bus->parent))
  80		return dev == 0;
  81
  82	return 1;
  83}
  84
  85static u8 rockchip_pcie_lane_map(struct rockchip_pcie *rockchip)
  86{
  87	u32 val;
  88	u8 map;
  89
  90	if (rockchip->legacy_phy)
  91		return GENMASK(MAX_LANE_NUM - 1, 0);
  92
  93	val = rockchip_pcie_read(rockchip, PCIE_CORE_LANE_MAP);
  94	map = val & PCIE_CORE_LANE_MAP_MASK;
  95
  96	/* The link may be using a reverse-indexed mapping. */
  97	if (val & PCIE_CORE_LANE_MAP_REVERSE)
  98		map = bitrev8(map) >> 4;
  99
 100	return map;
 101}
 102
 103static int rockchip_pcie_rd_own_conf(struct rockchip_pcie *rockchip,
 104				     int where, int size, u32 *val)
 105{
 106	void __iomem *addr;
 107
 108	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + where;
 109
 110	if (!IS_ALIGNED((uintptr_t)addr, size)) {
 111		*val = 0;
 112		return PCIBIOS_BAD_REGISTER_NUMBER;
 113	}
 114
 115	if (size == 4) {
 116		*val = readl(addr);
 117	} else if (size == 2) {
 118		*val = readw(addr);
 119	} else if (size == 1) {
 120		*val = readb(addr);
 121	} else {
 122		*val = 0;
 123		return PCIBIOS_BAD_REGISTER_NUMBER;
 124	}
 125	return PCIBIOS_SUCCESSFUL;
 126}
 127
 128static int rockchip_pcie_wr_own_conf(struct rockchip_pcie *rockchip,
 129				     int where, int size, u32 val)
 130{
 131	u32 mask, tmp, offset;
 132	void __iomem *addr;
 133
 134	offset = where & ~0x3;
 135	addr = rockchip->apb_base + PCIE_RC_CONFIG_NORMAL_BASE + offset;
 136
 137	if (size == 4) {
 138		writel(val, addr);
 139		return PCIBIOS_SUCCESSFUL;
 140	}
 141
 142	mask = ~(((1 << (size * 8)) - 1) << ((where & 0x3) * 8));
 143
 144	/*
 145	 * N.B. This read/modify/write isn't safe in general because it can
 146	 * corrupt RW1C bits in adjacent registers.  But the hardware
 147	 * doesn't support smaller writes.
 148	 */
 149	tmp = readl(addr) & mask;
 150	tmp |= val << ((where & 0x3) * 8);
 151	writel(tmp, addr);
 152
 153	return PCIBIOS_SUCCESSFUL;
 154}
 155
 156static int rockchip_pcie_rd_other_conf(struct rockchip_pcie *rockchip,
 157				       struct pci_bus *bus, u32 devfn,
 158				       int where, int size, u32 *val)
 159{
 160	void __iomem *addr;
 161
 162	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
 163
 164	if (!IS_ALIGNED((uintptr_t)addr, size)) {
 165		*val = 0;
 166		return PCIBIOS_BAD_REGISTER_NUMBER;
 167	}
 168
 169	if (pci_is_root_bus(bus->parent))
 170		rockchip_pcie_cfg_configuration_accesses(rockchip,
 171						AXI_WRAPPER_TYPE0_CFG);
 172	else
 173		rockchip_pcie_cfg_configuration_accesses(rockchip,
 174						AXI_WRAPPER_TYPE1_CFG);
 175
 176	if (size == 4) {
 177		*val = readl(addr);
 178	} else if (size == 2) {
 179		*val = readw(addr);
 180	} else if (size == 1) {
 181		*val = readb(addr);
 182	} else {
 183		*val = 0;
 184		return PCIBIOS_BAD_REGISTER_NUMBER;
 185	}
 186	return PCIBIOS_SUCCESSFUL;
 187}
 188
 189static int rockchip_pcie_wr_other_conf(struct rockchip_pcie *rockchip,
 190				       struct pci_bus *bus, u32 devfn,
 191				       int where, int size, u32 val)
 192{
 193	void __iomem *addr;
 194
 195	addr = rockchip->reg_base + PCIE_ECAM_OFFSET(bus->number, devfn, where);
 196
 197	if (!IS_ALIGNED((uintptr_t)addr, size))
 198		return PCIBIOS_BAD_REGISTER_NUMBER;
 199
 200	if (pci_is_root_bus(bus->parent))
 201		rockchip_pcie_cfg_configuration_accesses(rockchip,
 202						AXI_WRAPPER_TYPE0_CFG);
 203	else
 204		rockchip_pcie_cfg_configuration_accesses(rockchip,
 205						AXI_WRAPPER_TYPE1_CFG);
 206
 207	if (size == 4)
 208		writel(val, addr);
 209	else if (size == 2)
 210		writew(val, addr);
 211	else if (size == 1)
 212		writeb(val, addr);
 213	else
 214		return PCIBIOS_BAD_REGISTER_NUMBER;
 215
 216	return PCIBIOS_SUCCESSFUL;
 217}
 218
 219static int rockchip_pcie_rd_conf(struct pci_bus *bus, u32 devfn, int where,
 220				 int size, u32 *val)
 221{
 222	struct rockchip_pcie *rockchip = bus->sysdata;
 223
 224	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn))) {
 225		*val = 0xffffffff;
 226		return PCIBIOS_DEVICE_NOT_FOUND;
 227	}
 228
 229	if (pci_is_root_bus(bus))
 230		return rockchip_pcie_rd_own_conf(rockchip, where, size, val);
 231
 232	return rockchip_pcie_rd_other_conf(rockchip, bus, devfn, where, size,
 233					   val);
 234}
 235
 236static int rockchip_pcie_wr_conf(struct pci_bus *bus, u32 devfn,
 237				 int where, int size, u32 val)
 238{
 239	struct rockchip_pcie *rockchip = bus->sysdata;
 240
 241	if (!rockchip_pcie_valid_device(rockchip, bus, PCI_SLOT(devfn)))
 242		return PCIBIOS_DEVICE_NOT_FOUND;
 243
 244	if (pci_is_root_bus(bus))
 245		return rockchip_pcie_wr_own_conf(rockchip, where, size, val);
 246
 247	return rockchip_pcie_wr_other_conf(rockchip, bus, devfn, where, size,
 248					   val);
 249}
 250
 251static struct pci_ops rockchip_pcie_ops = {
 252	.read = rockchip_pcie_rd_conf,
 253	.write = rockchip_pcie_wr_conf,
 254};
 255
 256static void rockchip_pcie_set_power_limit(struct rockchip_pcie *rockchip)
 257{
 258	int curr;
 259	u32 status, scale, power;
 260
 261	if (IS_ERR(rockchip->vpcie3v3))
 262		return;
 263
 264	/*
 265	 * Set RC's captured slot power limit and scale if
 266	 * vpcie3v3 available. The default values are both zero
 267	 * which means the software should set these two according
 268	 * to the actual power supply.
 269	 */
 270	curr = regulator_get_current_limit(rockchip->vpcie3v3);
 271	if (curr <= 0)
 272		return;
 273
 274	scale = 3; /* 0.001x */
 275	curr = curr / 1000; /* convert to mA */
 276	power = (curr * 3300) / 1000; /* milliwatt */
 277	while (power > PCIE_RC_CONFIG_DCR_CSPL_LIMIT) {
 278		if (!scale) {
 279			dev_warn(rockchip->dev, "invalid power supply\n");
 280			return;
 281		}
 282		scale--;
 283		power = power / 10;
 284	}
 285
 286	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCR);
 287	status |= (power << PCIE_RC_CONFIG_DCR_CSPL_SHIFT) |
 288		  (scale << PCIE_RC_CONFIG_DCR_CPLS_SHIFT);
 289	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCR);
 290}
 291
 292/**
 293 * rockchip_pcie_host_init_port - Initialize hardware
 294 * @rockchip: PCIe port information
 295 */
 296static int rockchip_pcie_host_init_port(struct rockchip_pcie *rockchip)
 297{
 298	struct device *dev = rockchip->dev;
 299	int err, i = MAX_LANE_NUM;
 300	u32 status;
 301
 302	gpiod_set_value_cansleep(rockchip->ep_gpio, 0);
 303
 304	err = rockchip_pcie_init_port(rockchip);
 305	if (err)
 306		return err;
 307
 308	/* Fix the transmitted FTS count desired to exit from L0s. */
 309	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL_PLC1);
 310	status = (status & ~PCIE_CORE_CTRL_PLC1_FTS_MASK) |
 311		 (PCIE_CORE_CTRL_PLC1_FTS_CNT << PCIE_CORE_CTRL_PLC1_FTS_SHIFT);
 312	rockchip_pcie_write(rockchip, status, PCIE_CORE_CTRL_PLC1);
 313
 314	rockchip_pcie_set_power_limit(rockchip);
 315
 316	/* Set RC's clock architecture as common clock */
 317	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 318	status |= PCI_EXP_LNKSTA_SLC << 16;
 319	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 320
 321	/* Set RC's RCB to 128 */
 322	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 323	status |= PCI_EXP_LNKCTL_RCB;
 324	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 325
 326	/* Enable Gen1 training */
 327	rockchip_pcie_write(rockchip, PCIE_CLIENT_LINK_TRAIN_ENABLE,
 328			    PCIE_CLIENT_CONFIG);
 329
 330	gpiod_set_value_cansleep(rockchip->ep_gpio, 1);
 331
 332	/* 500ms timeout value should be enough for Gen1/2 training */
 333	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_BASIC_STATUS1,
 334				 status, PCIE_LINK_UP(status), 20,
 335				 500 * USEC_PER_MSEC);
 336	if (err) {
 337		dev_err(dev, "PCIe link training gen1 timeout!\n");
 338		goto err_power_off_phy;
 339	}
 340
 341	if (rockchip->link_gen == 2) {
 342		/*
 343		 * Enable retrain for gen2. This should be configured only after
 344		 * gen1 finished.
 345		 */
 346		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LCS);
 347		status |= PCI_EXP_LNKCTL_RL;
 348		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LCS);
 349
 350		err = readl_poll_timeout(rockchip->apb_base + PCIE_CORE_CTRL,
 351					 status, PCIE_LINK_IS_GEN2(status), 20,
 352					 500 * USEC_PER_MSEC);
 353		if (err)
 354			dev_dbg(dev, "PCIe link training gen2 timeout, fall back to gen1!\n");
 355	}
 356
 357	/* Check the final link width from negotiated lane counter from MGMT */
 358	status = rockchip_pcie_read(rockchip, PCIE_CORE_CTRL);
 359	status = 0x1 << ((status & PCIE_CORE_PL_CONF_LANE_MASK) >>
 360			  PCIE_CORE_PL_CONF_LANE_SHIFT);
 361	dev_dbg(dev, "current link width is x%d\n", status);
 362
 363	/* Power off unused lane(s) */
 364	rockchip->lanes_map = rockchip_pcie_lane_map(rockchip);
 365	for (i = 0; i < MAX_LANE_NUM; i++) {
 366		if (!(rockchip->lanes_map & BIT(i))) {
 367			dev_dbg(dev, "idling lane %d\n", i);
 368			phy_power_off(rockchip->phys[i]);
 369		}
 370	}
 371
 372	rockchip_pcie_write(rockchip, ROCKCHIP_VENDOR_ID,
 373			    PCIE_CORE_CONFIG_VENDOR);
 374	rockchip_pcie_write(rockchip,
 375			    PCI_CLASS_BRIDGE_PCI << PCIE_RC_CONFIG_SCC_SHIFT,
 376			    PCIE_RC_CONFIG_RID_CCR);
 377
 378	/* Clear THP cap's next cap pointer to remove L1 substate cap */
 379	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_THP_CAP);
 380	status &= ~PCIE_RC_CONFIG_THP_CAP_NEXT_MASK;
 381	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_THP_CAP);
 382
 383	/* Clear L0s from RC's link cap */
 384	if (of_property_read_bool(dev->of_node, "aspm-no-l0s")) {
 385		status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_LINK_CAP);
 386		status &= ~PCIE_RC_CONFIG_LINK_CAP_L0S;
 387		rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_LINK_CAP);
 388	}
 389
 390	status = rockchip_pcie_read(rockchip, PCIE_RC_CONFIG_DCSR);
 391	status &= ~PCIE_RC_CONFIG_DCSR_MPS_MASK;
 392	status |= PCIE_RC_CONFIG_DCSR_MPS_256;
 393	rockchip_pcie_write(rockchip, status, PCIE_RC_CONFIG_DCSR);
 394
 395	return 0;
 396err_power_off_phy:
 397	while (i--)
 398		phy_power_off(rockchip->phys[i]);
 399	i = MAX_LANE_NUM;
 400	while (i--)
 401		phy_exit(rockchip->phys[i]);
 402	return err;
 403}
 404
 405static irqreturn_t rockchip_pcie_subsys_irq_handler(int irq, void *arg)
 406{
 407	struct rockchip_pcie *rockchip = arg;
 408	struct device *dev = rockchip->dev;
 409	u32 reg;
 410	u32 sub_reg;
 411
 412	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 413	if (reg & PCIE_CLIENT_INT_LOCAL) {
 414		dev_dbg(dev, "local interrupt received\n");
 415		sub_reg = rockchip_pcie_read(rockchip, PCIE_CORE_INT_STATUS);
 416		if (sub_reg & PCIE_CORE_INT_PRFPE)
 417			dev_dbg(dev, "parity error detected while reading from the PNP receive FIFO RAM\n");
 418
 419		if (sub_reg & PCIE_CORE_INT_CRFPE)
 420			dev_dbg(dev, "parity error detected while reading from the Completion Receive FIFO RAM\n");
 421
 422		if (sub_reg & PCIE_CORE_INT_RRPE)
 423			dev_dbg(dev, "parity error detected while reading from replay buffer RAM\n");
 424
 425		if (sub_reg & PCIE_CORE_INT_PRFO)
 426			dev_dbg(dev, "overflow occurred in the PNP receive FIFO\n");
 427
 428		if (sub_reg & PCIE_CORE_INT_CRFO)
 429			dev_dbg(dev, "overflow occurred in the completion receive FIFO\n");
 430
 431		if (sub_reg & PCIE_CORE_INT_RT)
 432			dev_dbg(dev, "replay timer timed out\n");
 433
 434		if (sub_reg & PCIE_CORE_INT_RTR)
 435			dev_dbg(dev, "replay timer rolled over after 4 transmissions of the same TLP\n");
 436
 437		if (sub_reg & PCIE_CORE_INT_PE)
 438			dev_dbg(dev, "phy error detected on receive side\n");
 439
 440		if (sub_reg & PCIE_CORE_INT_MTR)
 441			dev_dbg(dev, "malformed TLP received from the link\n");
 442
 443		if (sub_reg & PCIE_CORE_INT_UCR)
 444			dev_dbg(dev, "malformed TLP received from the link\n");
 445
 446		if (sub_reg & PCIE_CORE_INT_FCE)
 447			dev_dbg(dev, "an error was observed in the flow control advertisements from the other side\n");
 448
 449		if (sub_reg & PCIE_CORE_INT_CT)
 450			dev_dbg(dev, "a request timed out waiting for completion\n");
 451
 452		if (sub_reg & PCIE_CORE_INT_UTC)
 453			dev_dbg(dev, "unmapped TC error\n");
 454
 455		if (sub_reg & PCIE_CORE_INT_MMVC)
 456			dev_dbg(dev, "MSI mask register changes\n");
 457
 458		rockchip_pcie_write(rockchip, sub_reg, PCIE_CORE_INT_STATUS);
 459	} else if (reg & PCIE_CLIENT_INT_PHY) {
 460		dev_dbg(dev, "phy link changes\n");
 461		rockchip_pcie_update_txcredit_mui(rockchip);
 462		rockchip_pcie_clr_bw_int(rockchip);
 463	}
 464
 465	rockchip_pcie_write(rockchip, reg & PCIE_CLIENT_INT_LOCAL,
 466			    PCIE_CLIENT_INT_STATUS);
 467
 468	return IRQ_HANDLED;
 469}
 470
 471static irqreturn_t rockchip_pcie_client_irq_handler(int irq, void *arg)
 472{
 473	struct rockchip_pcie *rockchip = arg;
 474	struct device *dev = rockchip->dev;
 475	u32 reg;
 476
 477	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 478	if (reg & PCIE_CLIENT_INT_LEGACY_DONE)
 479		dev_dbg(dev, "legacy done interrupt received\n");
 480
 481	if (reg & PCIE_CLIENT_INT_MSG)
 482		dev_dbg(dev, "message done interrupt received\n");
 483
 484	if (reg & PCIE_CLIENT_INT_HOT_RST)
 485		dev_dbg(dev, "hot reset interrupt received\n");
 486
 487	if (reg & PCIE_CLIENT_INT_DPA)
 488		dev_dbg(dev, "dpa interrupt received\n");
 489
 490	if (reg & PCIE_CLIENT_INT_FATAL_ERR)
 491		dev_dbg(dev, "fatal error interrupt received\n");
 492
 493	if (reg & PCIE_CLIENT_INT_NFATAL_ERR)
 494		dev_dbg(dev, "no fatal error interrupt received\n");
 495
 496	if (reg & PCIE_CLIENT_INT_CORR_ERR)
 497		dev_dbg(dev, "correctable error interrupt received\n");
 498
 499	if (reg & PCIE_CLIENT_INT_PHY)
 500		dev_dbg(dev, "phy interrupt received\n");
 501
 502	rockchip_pcie_write(rockchip, reg & (PCIE_CLIENT_INT_LEGACY_DONE |
 503			      PCIE_CLIENT_INT_MSG | PCIE_CLIENT_INT_HOT_RST |
 504			      PCIE_CLIENT_INT_DPA | PCIE_CLIENT_INT_FATAL_ERR |
 505			      PCIE_CLIENT_INT_NFATAL_ERR |
 506			      PCIE_CLIENT_INT_CORR_ERR |
 507			      PCIE_CLIENT_INT_PHY),
 508		   PCIE_CLIENT_INT_STATUS);
 509
 510	return IRQ_HANDLED;
 511}
 512
 513static void rockchip_pcie_legacy_int_handler(struct irq_desc *desc)
 514{
 515	struct irq_chip *chip = irq_desc_get_chip(desc);
 516	struct rockchip_pcie *rockchip = irq_desc_get_handler_data(desc);
 517	struct device *dev = rockchip->dev;
 518	u32 reg;
 519	u32 hwirq;
 520	u32 virq;
 521
 522	chained_irq_enter(chip, desc);
 523
 524	reg = rockchip_pcie_read(rockchip, PCIE_CLIENT_INT_STATUS);
 525	reg = (reg & PCIE_CLIENT_INTR_MASK) >> PCIE_CLIENT_INTR_SHIFT;
 526
 527	while (reg) {
 528		hwirq = ffs(reg) - 1;
 529		reg &= ~BIT(hwirq);
 530
 531		virq = irq_find_mapping(rockchip->irq_domain, hwirq);
 532		if (virq)
 533			generic_handle_irq(virq);
 534		else
 535			dev_err(dev, "unexpected IRQ, INT%d\n", hwirq);
 536	}
 537
 538	chained_irq_exit(chip, desc);
 539}
 540
 541static int rockchip_pcie_setup_irq(struct rockchip_pcie *rockchip)
 542{
 543	int irq, err;
 544	struct device *dev = rockchip->dev;
 545	struct platform_device *pdev = to_platform_device(dev);
 546
 547	irq = platform_get_irq_byname(pdev, "sys");
 548	if (irq < 0)
 549		return irq;
 550
 551	err = devm_request_irq(dev, irq, rockchip_pcie_subsys_irq_handler,
 552			       IRQF_SHARED, "pcie-sys", rockchip);
 553	if (err) {
 554		dev_err(dev, "failed to request PCIe subsystem IRQ\n");
 555		return err;
 556	}
 557
 558	irq = platform_get_irq_byname(pdev, "legacy");
 559	if (irq < 0)
 560		return irq;
 561
 562	irq_set_chained_handler_and_data(irq,
 563					 rockchip_pcie_legacy_int_handler,
 564					 rockchip);
 565
 566	irq = platform_get_irq_byname(pdev, "client");
 567	if (irq < 0)
 568		return irq;
 569
 570	err = devm_request_irq(dev, irq, rockchip_pcie_client_irq_handler,
 571			       IRQF_SHARED, "pcie-client", rockchip);
 572	if (err) {
 573		dev_err(dev, "failed to request PCIe client IRQ\n");
 574		return err;
 575	}
 576
 577	return 0;
 578}
 579
 580/**
 581 * rockchip_pcie_parse_host_dt - Parse Device Tree
 582 * @rockchip: PCIe port information
 583 *
 584 * Return: '0' on success and error value on failure
 585 */
 586static int rockchip_pcie_parse_host_dt(struct rockchip_pcie *rockchip)
 587{
 588	struct device *dev = rockchip->dev;
 589	int err;
 590
 591	err = rockchip_pcie_parse_dt(rockchip);
 592	if (err)
 593		return err;
 594
 595	rockchip->vpcie12v = devm_regulator_get_optional(dev, "vpcie12v");
 596	if (IS_ERR(rockchip->vpcie12v)) {
 597		if (PTR_ERR(rockchip->vpcie12v) != -ENODEV)
 598			return PTR_ERR(rockchip->vpcie12v);
 599		dev_info(dev, "no vpcie12v regulator found\n");
 600	}
 601
 602	rockchip->vpcie3v3 = devm_regulator_get_optional(dev, "vpcie3v3");
 603	if (IS_ERR(rockchip->vpcie3v3)) {
 604		if (PTR_ERR(rockchip->vpcie3v3) != -ENODEV)
 605			return PTR_ERR(rockchip->vpcie3v3);
 606		dev_info(dev, "no vpcie3v3 regulator found\n");
 607	}
 608
 609	rockchip->vpcie1v8 = devm_regulator_get(dev, "vpcie1v8");
 610	if (IS_ERR(rockchip->vpcie1v8))
 611		return PTR_ERR(rockchip->vpcie1v8);
 612
 613	rockchip->vpcie0v9 = devm_regulator_get(dev, "vpcie0v9");
 614	if (IS_ERR(rockchip->vpcie0v9))
 615		return PTR_ERR(rockchip->vpcie0v9);
 616
 617	return 0;
 618}
 619
 620static int rockchip_pcie_set_vpcie(struct rockchip_pcie *rockchip)
 621{
 622	struct device *dev = rockchip->dev;
 623	int err;
 624
 625	if (!IS_ERR(rockchip->vpcie12v)) {
 626		err = regulator_enable(rockchip->vpcie12v);
 627		if (err) {
 628			dev_err(dev, "fail to enable vpcie12v regulator\n");
 629			goto err_out;
 630		}
 631	}
 632
 633	if (!IS_ERR(rockchip->vpcie3v3)) {
 634		err = regulator_enable(rockchip->vpcie3v3);
 635		if (err) {
 636			dev_err(dev, "fail to enable vpcie3v3 regulator\n");
 637			goto err_disable_12v;
 638		}
 639	}
 640
 641	err = regulator_enable(rockchip->vpcie1v8);
 642	if (err) {
 643		dev_err(dev, "fail to enable vpcie1v8 regulator\n");
 644		goto err_disable_3v3;
 645	}
 646
 647	err = regulator_enable(rockchip->vpcie0v9);
 648	if (err) {
 649		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
 650		goto err_disable_1v8;
 651	}
 652
 653	return 0;
 654
 655err_disable_1v8:
 656	regulator_disable(rockchip->vpcie1v8);
 657err_disable_3v3:
 658	if (!IS_ERR(rockchip->vpcie3v3))
 659		regulator_disable(rockchip->vpcie3v3);
 660err_disable_12v:
 661	if (!IS_ERR(rockchip->vpcie12v))
 662		regulator_disable(rockchip->vpcie12v);
 663err_out:
 664	return err;
 665}
 666
 667static void rockchip_pcie_enable_interrupts(struct rockchip_pcie *rockchip)
 668{
 669	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) &
 670			    (~PCIE_CLIENT_INT_CLI), PCIE_CLIENT_INT_MASK);
 671	rockchip_pcie_write(rockchip, (u32)(~PCIE_CORE_INT),
 672			    PCIE_CORE_INT_MASK);
 673
 674	rockchip_pcie_enable_bw_int(rockchip);
 675}
 676
 677static int rockchip_pcie_intx_map(struct irq_domain *domain, unsigned int irq,
 678				  irq_hw_number_t hwirq)
 679{
 680	irq_set_chip_and_handler(irq, &dummy_irq_chip, handle_simple_irq);
 681	irq_set_chip_data(irq, domain->host_data);
 682
 683	return 0;
 684}
 685
 686static const struct irq_domain_ops intx_domain_ops = {
 687	.map = rockchip_pcie_intx_map,
 688};
 689
 690static int rockchip_pcie_init_irq_domain(struct rockchip_pcie *rockchip)
 691{
 692	struct device *dev = rockchip->dev;
 693	struct device_node *intc = of_get_next_child(dev->of_node, NULL);
 694
 695	if (!intc) {
 696		dev_err(dev, "missing child interrupt-controller node\n");
 697		return -EINVAL;
 698	}
 699
 700	rockchip->irq_domain = irq_domain_add_linear(intc, PCI_NUM_INTX,
 701						    &intx_domain_ops, rockchip);
 702	of_node_put(intc);
 703	if (!rockchip->irq_domain) {
 704		dev_err(dev, "failed to get a INTx IRQ domain\n");
 705		return -EINVAL;
 706	}
 707
 708	return 0;
 709}
 710
 711static int rockchip_pcie_prog_ob_atu(struct rockchip_pcie *rockchip,
 712				     int region_no, int type, u8 num_pass_bits,
 713				     u32 lower_addr, u32 upper_addr)
 714{
 715	u32 ob_addr_0;
 716	u32 ob_addr_1;
 717	u32 ob_desc_0;
 718	u32 aw_offset;
 719
 720	if (region_no >= MAX_AXI_WRAPPER_REGION_NUM)
 721		return -EINVAL;
 722	if (num_pass_bits + 1 < 8)
 723		return -EINVAL;
 724	if (num_pass_bits > 63)
 725		return -EINVAL;
 726	if (region_no == 0) {
 727		if (AXI_REGION_0_SIZE < (2ULL << num_pass_bits))
 728			return -EINVAL;
 729	}
 730	if (region_no != 0) {
 731		if (AXI_REGION_SIZE < (2ULL << num_pass_bits))
 732			return -EINVAL;
 733	}
 734
 735	aw_offset = (region_no << OB_REG_SIZE_SHIFT);
 736
 737	ob_addr_0 = num_pass_bits & PCIE_CORE_OB_REGION_ADDR0_NUM_BITS;
 738	ob_addr_0 |= lower_addr & PCIE_CORE_OB_REGION_ADDR0_LO_ADDR;
 739	ob_addr_1 = upper_addr;
 740	ob_desc_0 = (1 << 23 | type);
 741
 742	rockchip_pcie_write(rockchip, ob_addr_0,
 743			    PCIE_CORE_OB_REGION_ADDR0 + aw_offset);
 744	rockchip_pcie_write(rockchip, ob_addr_1,
 745			    PCIE_CORE_OB_REGION_ADDR1 + aw_offset);
 746	rockchip_pcie_write(rockchip, ob_desc_0,
 747			    PCIE_CORE_OB_REGION_DESC0 + aw_offset);
 748	rockchip_pcie_write(rockchip, 0,
 749			    PCIE_CORE_OB_REGION_DESC1 + aw_offset);
 750
 751	return 0;
 752}
 753
 754static int rockchip_pcie_prog_ib_atu(struct rockchip_pcie *rockchip,
 755				     int region_no, u8 num_pass_bits,
 756				     u32 lower_addr, u32 upper_addr)
 757{
 758	u32 ib_addr_0;
 759	u32 ib_addr_1;
 760	u32 aw_offset;
 761
 762	if (region_no > MAX_AXI_IB_ROOTPORT_REGION_NUM)
 763		return -EINVAL;
 764	if (num_pass_bits + 1 < MIN_AXI_ADDR_BITS_PASSED)
 765		return -EINVAL;
 766	if (num_pass_bits > 63)
 767		return -EINVAL;
 768
 769	aw_offset = (region_no << IB_ROOT_PORT_REG_SIZE_SHIFT);
 770
 771	ib_addr_0 = num_pass_bits & PCIE_CORE_IB_REGION_ADDR0_NUM_BITS;
 772	ib_addr_0 |= (lower_addr << 8) & PCIE_CORE_IB_REGION_ADDR0_LO_ADDR;
 773	ib_addr_1 = upper_addr;
 774
 775	rockchip_pcie_write(rockchip, ib_addr_0, PCIE_RP_IB_ADDR0 + aw_offset);
 776	rockchip_pcie_write(rockchip, ib_addr_1, PCIE_RP_IB_ADDR1 + aw_offset);
 777
 778	return 0;
 779}
 780
 781static int rockchip_pcie_cfg_atu(struct rockchip_pcie *rockchip)
 782{
 783	struct device *dev = rockchip->dev;
 784	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
 785	struct resource_entry *entry;
 786	u64 pci_addr, size;
 787	int offset;
 788	int err;
 789	int reg_no;
 790
 791	rockchip_pcie_cfg_configuration_accesses(rockchip,
 792						 AXI_WRAPPER_TYPE0_CFG);
 793	entry = resource_list_first_type(&bridge->windows, IORESOURCE_MEM);
 794	if (!entry)
 795		return -ENODEV;
 796
 797	size = resource_size(entry->res);
 798	pci_addr = entry->res->start - entry->offset;
 799	rockchip->msg_bus_addr = pci_addr;
 800
 801	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
 802		err = rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1,
 803						AXI_WRAPPER_MEM_WRITE,
 804						20 - 1,
 805						pci_addr + (reg_no << 20),
 806						0);
 807		if (err) {
 808			dev_err(dev, "program RC mem outbound ATU failed\n");
 809			return err;
 810		}
 811	}
 812
 813	err = rockchip_pcie_prog_ib_atu(rockchip, 2, 32 - 1, 0x0, 0);
 814	if (err) {
 815		dev_err(dev, "program RC mem inbound ATU failed\n");
 816		return err;
 817	}
 818
 819	entry = resource_list_first_type(&bridge->windows, IORESOURCE_IO);
 820	if (!entry)
 821		return -ENODEV;
 822
 823	/* store the register number offset to program RC io outbound ATU */
 824	offset = size >> 20;
 825
 826	size = resource_size(entry->res);
 827	pci_addr = entry->res->start - entry->offset;
 828
 829	for (reg_no = 0; reg_no < (size >> 20); reg_no++) {
 830		err = rockchip_pcie_prog_ob_atu(rockchip,
 831						reg_no + 1 + offset,
 832						AXI_WRAPPER_IO_WRITE,
 833						20 - 1,
 834						pci_addr + (reg_no << 20),
 835						0);
 836		if (err) {
 837			dev_err(dev, "program RC io outbound ATU failed\n");
 838			return err;
 839		}
 840	}
 841
 842	/* assign message regions */
 843	rockchip_pcie_prog_ob_atu(rockchip, reg_no + 1 + offset,
 844				  AXI_WRAPPER_NOR_MSG,
 845				  20 - 1, 0, 0);
 846
 847	rockchip->msg_bus_addr += ((reg_no + offset) << 20);
 848	return err;
 849}
 850
 851static int rockchip_pcie_wait_l2(struct rockchip_pcie *rockchip)
 852{
 853	u32 value;
 854	int err;
 855
 856	/* send PME_TURN_OFF message */
 857	writel(0x0, rockchip->msg_region + PCIE_RC_SEND_PME_OFF);
 858
 859	/* read LTSSM and wait for falling into L2 link state */
 860	err = readl_poll_timeout(rockchip->apb_base + PCIE_CLIENT_DEBUG_OUT_0,
 861				 value, PCIE_LINK_IS_L2(value), 20,
 862				 jiffies_to_usecs(5 * HZ));
 863	if (err) {
 864		dev_err(rockchip->dev, "PCIe link enter L2 timeout!\n");
 865		return err;
 866	}
 867
 868	return 0;
 869}
 870
 871static int __maybe_unused rockchip_pcie_suspend_noirq(struct device *dev)
 872{
 873	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
 874	int ret;
 875
 876	/* disable core and cli int since we don't need to ack PME_ACK */
 877	rockchip_pcie_write(rockchip, (PCIE_CLIENT_INT_CLI << 16) |
 878			    PCIE_CLIENT_INT_CLI, PCIE_CLIENT_INT_MASK);
 879	rockchip_pcie_write(rockchip, (u32)PCIE_CORE_INT, PCIE_CORE_INT_MASK);
 880
 881	ret = rockchip_pcie_wait_l2(rockchip);
 882	if (ret) {
 883		rockchip_pcie_enable_interrupts(rockchip);
 884		return ret;
 885	}
 886
 887	rockchip_pcie_deinit_phys(rockchip);
 888
 889	rockchip_pcie_disable_clocks(rockchip);
 890
 891	regulator_disable(rockchip->vpcie0v9);
 892
 893	return ret;
 894}
 895
 896static int __maybe_unused rockchip_pcie_resume_noirq(struct device *dev)
 897{
 898	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
 899	int err;
 900
 901	err = regulator_enable(rockchip->vpcie0v9);
 902	if (err) {
 903		dev_err(dev, "fail to enable vpcie0v9 regulator\n");
 904		return err;
 905	}
 906
 907	err = rockchip_pcie_enable_clocks(rockchip);
 908	if (err)
 909		goto err_disable_0v9;
 910
 911	err = rockchip_pcie_host_init_port(rockchip);
 912	if (err)
 913		goto err_pcie_resume;
 914
 915	err = rockchip_pcie_cfg_atu(rockchip);
 916	if (err)
 917		goto err_err_deinit_port;
 918
 919	/* Need this to enter L1 again */
 920	rockchip_pcie_update_txcredit_mui(rockchip);
 921	rockchip_pcie_enable_interrupts(rockchip);
 922
 923	return 0;
 924
 925err_err_deinit_port:
 926	rockchip_pcie_deinit_phys(rockchip);
 927err_pcie_resume:
 928	rockchip_pcie_disable_clocks(rockchip);
 929err_disable_0v9:
 930	regulator_disable(rockchip->vpcie0v9);
 931	return err;
 932}
 933
 934static int rockchip_pcie_probe(struct platform_device *pdev)
 935{
 936	struct rockchip_pcie *rockchip;
 937	struct device *dev = &pdev->dev;
 938	struct pci_host_bridge *bridge;
 939	int err;
 940
 941	if (!dev->of_node)
 942		return -ENODEV;
 943
 944	bridge = devm_pci_alloc_host_bridge(dev, sizeof(*rockchip));
 945	if (!bridge)
 946		return -ENOMEM;
 947
 948	rockchip = pci_host_bridge_priv(bridge);
 949
 950	platform_set_drvdata(pdev, rockchip);
 951	rockchip->dev = dev;
 952	rockchip->is_rc = true;
 953
 954	err = rockchip_pcie_parse_host_dt(rockchip);
 955	if (err)
 956		return err;
 957
 958	err = rockchip_pcie_enable_clocks(rockchip);
 959	if (err)
 960		return err;
 961
 962	err = rockchip_pcie_set_vpcie(rockchip);
 963	if (err) {
 964		dev_err(dev, "failed to set vpcie regulator\n");
 965		goto err_set_vpcie;
 966	}
 967
 968	err = rockchip_pcie_host_init_port(rockchip);
 969	if (err)
 970		goto err_vpcie;
 971
 972	err = rockchip_pcie_init_irq_domain(rockchip);
 973	if (err < 0)
 974		goto err_deinit_port;
 975
 976	err = rockchip_pcie_cfg_atu(rockchip);
 977	if (err)
 978		goto err_remove_irq_domain;
 979
 980	rockchip->msg_region = devm_ioremap(dev, rockchip->msg_bus_addr, SZ_1M);
 981	if (!rockchip->msg_region) {
 982		err = -ENOMEM;
 983		goto err_remove_irq_domain;
 984	}
 985
 986	bridge->sysdata = rockchip;
 987	bridge->ops = &rockchip_pcie_ops;
 988
 989	err = rockchip_pcie_setup_irq(rockchip);
 990	if (err)
 991		goto err_remove_irq_domain;
 992
 993	rockchip_pcie_enable_interrupts(rockchip);
 994
 995	err = pci_host_probe(bridge);
 996	if (err < 0)
 997		goto err_remove_irq_domain;
 998
 999	return 0;
1000
1001err_remove_irq_domain:
1002	irq_domain_remove(rockchip->irq_domain);
1003err_deinit_port:
1004	rockchip_pcie_deinit_phys(rockchip);
1005err_vpcie:
1006	if (!IS_ERR(rockchip->vpcie12v))
1007		regulator_disable(rockchip->vpcie12v);
1008	if (!IS_ERR(rockchip->vpcie3v3))
1009		regulator_disable(rockchip->vpcie3v3);
1010	regulator_disable(rockchip->vpcie1v8);
1011	regulator_disable(rockchip->vpcie0v9);
1012err_set_vpcie:
1013	rockchip_pcie_disable_clocks(rockchip);
1014	return err;
1015}
1016
1017static int rockchip_pcie_remove(struct platform_device *pdev)
1018{
1019	struct device *dev = &pdev->dev;
1020	struct rockchip_pcie *rockchip = dev_get_drvdata(dev);
1021	struct pci_host_bridge *bridge = pci_host_bridge_from_priv(rockchip);
1022
1023	pci_stop_root_bus(bridge->bus);
1024	pci_remove_root_bus(bridge->bus);
1025	irq_domain_remove(rockchip->irq_domain);
1026
1027	rockchip_pcie_deinit_phys(rockchip);
1028
1029	rockchip_pcie_disable_clocks(rockchip);
1030
1031	if (!IS_ERR(rockchip->vpcie12v))
1032		regulator_disable(rockchip->vpcie12v);
1033	if (!IS_ERR(rockchip->vpcie3v3))
1034		regulator_disable(rockchip->vpcie3v3);
1035	regulator_disable(rockchip->vpcie1v8);
1036	regulator_disable(rockchip->vpcie0v9);
1037
1038	return 0;
1039}
1040
1041static const struct dev_pm_ops rockchip_pcie_pm_ops = {
1042	SET_NOIRQ_SYSTEM_SLEEP_PM_OPS(rockchip_pcie_suspend_noirq,
1043				      rockchip_pcie_resume_noirq)
1044};
1045
1046static const struct of_device_id rockchip_pcie_of_match[] = {
1047	{ .compatible = "rockchip,rk3399-pcie", },
1048	{}
1049};
1050MODULE_DEVICE_TABLE(of, rockchip_pcie_of_match);
1051
1052static struct platform_driver rockchip_pcie_driver = {
1053	.driver = {
1054		.name = "rockchip-pcie",
1055		.of_match_table = rockchip_pcie_of_match,
1056		.pm = &rockchip_pcie_pm_ops,
1057	},
1058	.probe = rockchip_pcie_probe,
1059	.remove = rockchip_pcie_remove,
1060};
1061module_platform_driver(rockchip_pcie_driver);
1062
1063MODULE_AUTHOR("Rockchip Inc");
1064MODULE_DESCRIPTION("Rockchip AXI PCIe driver");
1065MODULE_LICENSE("GPL v2");