162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * PCI Express Downstream Port Containment services driver
462306a36Sopenharmony_ci * Author: Keith Busch <keith.busch@intel.com>
562306a36Sopenharmony_ci *
662306a36Sopenharmony_ci * Copyright (C) 2016 Intel Corp.
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#define dev_fmt(fmt) "DPC: " fmt
1062306a36Sopenharmony_ci
1162306a36Sopenharmony_ci#include <linux/aer.h>
1262306a36Sopenharmony_ci#include <linux/delay.h>
1362306a36Sopenharmony_ci#include <linux/interrupt.h>
1462306a36Sopenharmony_ci#include <linux/init.h>
1562306a36Sopenharmony_ci#include <linux/pci.h>
1662306a36Sopenharmony_ci
1762306a36Sopenharmony_ci#include "portdrv.h"
1862306a36Sopenharmony_ci#include "../pci.h"
1962306a36Sopenharmony_ci
2062306a36Sopenharmony_cistatic const char * const rp_pio_error_string[] = {
2162306a36Sopenharmony_ci	"Configuration Request received UR Completion",	 /* Bit Position 0  */
2262306a36Sopenharmony_ci	"Configuration Request received CA Completion",	 /* Bit Position 1  */
2362306a36Sopenharmony_ci	"Configuration Request Completion Timeout",	 /* Bit Position 2  */
2462306a36Sopenharmony_ci	NULL,
2562306a36Sopenharmony_ci	NULL,
2662306a36Sopenharmony_ci	NULL,
2762306a36Sopenharmony_ci	NULL,
2862306a36Sopenharmony_ci	NULL,
2962306a36Sopenharmony_ci	"I/O Request received UR Completion",		 /* Bit Position 8  */
3062306a36Sopenharmony_ci	"I/O Request received CA Completion",		 /* Bit Position 9  */
3162306a36Sopenharmony_ci	"I/O Request Completion Timeout",		 /* Bit Position 10 */
3262306a36Sopenharmony_ci	NULL,
3362306a36Sopenharmony_ci	NULL,
3462306a36Sopenharmony_ci	NULL,
3562306a36Sopenharmony_ci	NULL,
3662306a36Sopenharmony_ci	NULL,
3762306a36Sopenharmony_ci	"Memory Request received UR Completion",	 /* Bit Position 16 */
3862306a36Sopenharmony_ci	"Memory Request received CA Completion",	 /* Bit Position 17 */
3962306a36Sopenharmony_ci	"Memory Request Completion Timeout",		 /* Bit Position 18 */
4062306a36Sopenharmony_ci};
4162306a36Sopenharmony_ci
4262306a36Sopenharmony_civoid pci_save_dpc_state(struct pci_dev *dev)
4362306a36Sopenharmony_ci{
4462306a36Sopenharmony_ci	struct pci_cap_saved_state *save_state;
4562306a36Sopenharmony_ci	u16 *cap;
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci	if (!pci_is_pcie(dev))
4862306a36Sopenharmony_ci		return;
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	save_state = pci_find_saved_ext_cap(dev, PCI_EXT_CAP_ID_DPC);
5162306a36Sopenharmony_ci	if (!save_state)
5262306a36Sopenharmony_ci		return;
5362306a36Sopenharmony_ci
5462306a36Sopenharmony_ci	cap = (u16 *)&save_state->cap.data[0];
5562306a36Sopenharmony_ci	pci_read_config_word(dev, dev->dpc_cap + PCI_EXP_DPC_CTL, cap);
5662306a36Sopenharmony_ci}
5762306a36Sopenharmony_ci
5862306a36Sopenharmony_civoid pci_restore_dpc_state(struct pci_dev *dev)
5962306a36Sopenharmony_ci{
6062306a36Sopenharmony_ci	struct pci_cap_saved_state *save_state;
6162306a36Sopenharmony_ci	u16 *cap;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	if (!pci_is_pcie(dev))
6462306a36Sopenharmony_ci		return;
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci	save_state = pci_find_saved_ext_cap(dev, PCI_EXT_CAP_ID_DPC);
6762306a36Sopenharmony_ci	if (!save_state)
6862306a36Sopenharmony_ci		return;
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	cap = (u16 *)&save_state->cap.data[0];
7162306a36Sopenharmony_ci	pci_write_config_word(dev, dev->dpc_cap + PCI_EXP_DPC_CTL, *cap);
7262306a36Sopenharmony_ci}
7362306a36Sopenharmony_ci
7462306a36Sopenharmony_cistatic DECLARE_WAIT_QUEUE_HEAD(dpc_completed_waitqueue);
7562306a36Sopenharmony_ci
7662306a36Sopenharmony_ci#ifdef CONFIG_HOTPLUG_PCI_PCIE
7762306a36Sopenharmony_cistatic bool dpc_completed(struct pci_dev *pdev)
7862306a36Sopenharmony_ci{
7962306a36Sopenharmony_ci	u16 status;
8062306a36Sopenharmony_ci
8162306a36Sopenharmony_ci	pci_read_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_STATUS, &status);
8262306a36Sopenharmony_ci	if ((!PCI_POSSIBLE_ERROR(status)) && (status & PCI_EXP_DPC_STATUS_TRIGGER))
8362306a36Sopenharmony_ci		return false;
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	if (test_bit(PCI_DPC_RECOVERING, &pdev->priv_flags))
8662306a36Sopenharmony_ci		return false;
8762306a36Sopenharmony_ci
8862306a36Sopenharmony_ci	return true;
8962306a36Sopenharmony_ci}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ci/**
9262306a36Sopenharmony_ci * pci_dpc_recovered - whether DPC triggered and has recovered successfully
9362306a36Sopenharmony_ci * @pdev: PCI device
9462306a36Sopenharmony_ci *
9562306a36Sopenharmony_ci * Return true if DPC was triggered for @pdev and has recovered successfully.
9662306a36Sopenharmony_ci * Wait for recovery if it hasn't completed yet.  Called from the PCIe hotplug
9762306a36Sopenharmony_ci * driver to recognize and ignore Link Down/Up events caused by DPC.
9862306a36Sopenharmony_ci */
9962306a36Sopenharmony_cibool pci_dpc_recovered(struct pci_dev *pdev)
10062306a36Sopenharmony_ci{
10162306a36Sopenharmony_ci	struct pci_host_bridge *host;
10262306a36Sopenharmony_ci
10362306a36Sopenharmony_ci	if (!pdev->dpc_cap)
10462306a36Sopenharmony_ci		return false;
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	/*
10762306a36Sopenharmony_ci	 * Synchronization between hotplug and DPC is not supported
10862306a36Sopenharmony_ci	 * if DPC is owned by firmware and EDR is not enabled.
10962306a36Sopenharmony_ci	 */
11062306a36Sopenharmony_ci	host = pci_find_host_bridge(pdev->bus);
11162306a36Sopenharmony_ci	if (!host->native_dpc && !IS_ENABLED(CONFIG_PCIE_EDR))
11262306a36Sopenharmony_ci		return false;
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_ci	/*
11562306a36Sopenharmony_ci	 * Need a timeout in case DPC never completes due to failure of
11662306a36Sopenharmony_ci	 * dpc_wait_rp_inactive().  The spec doesn't mandate a time limit,
11762306a36Sopenharmony_ci	 * but reports indicate that DPC completes within 4 seconds.
11862306a36Sopenharmony_ci	 */
11962306a36Sopenharmony_ci	wait_event_timeout(dpc_completed_waitqueue, dpc_completed(pdev),
12062306a36Sopenharmony_ci			   msecs_to_jiffies(4000));
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_ci	return test_and_clear_bit(PCI_DPC_RECOVERED, &pdev->priv_flags);
12362306a36Sopenharmony_ci}
12462306a36Sopenharmony_ci#endif /* CONFIG_HOTPLUG_PCI_PCIE */
12562306a36Sopenharmony_ci
12662306a36Sopenharmony_cistatic int dpc_wait_rp_inactive(struct pci_dev *pdev)
12762306a36Sopenharmony_ci{
12862306a36Sopenharmony_ci	unsigned long timeout = jiffies + HZ;
12962306a36Sopenharmony_ci	u16 cap = pdev->dpc_cap, status;
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	pci_read_config_word(pdev, cap + PCI_EXP_DPC_STATUS, &status);
13262306a36Sopenharmony_ci	while (status & PCI_EXP_DPC_RP_BUSY &&
13362306a36Sopenharmony_ci					!time_after(jiffies, timeout)) {
13462306a36Sopenharmony_ci		msleep(10);
13562306a36Sopenharmony_ci		pci_read_config_word(pdev, cap + PCI_EXP_DPC_STATUS, &status);
13662306a36Sopenharmony_ci	}
13762306a36Sopenharmony_ci	if (status & PCI_EXP_DPC_RP_BUSY) {
13862306a36Sopenharmony_ci		pci_warn(pdev, "root port still busy\n");
13962306a36Sopenharmony_ci		return -EBUSY;
14062306a36Sopenharmony_ci	}
14162306a36Sopenharmony_ci	return 0;
14262306a36Sopenharmony_ci}
14362306a36Sopenharmony_ci
14462306a36Sopenharmony_cipci_ers_result_t dpc_reset_link(struct pci_dev *pdev)
14562306a36Sopenharmony_ci{
14662306a36Sopenharmony_ci	pci_ers_result_t ret;
14762306a36Sopenharmony_ci	u16 cap;
14862306a36Sopenharmony_ci
14962306a36Sopenharmony_ci	set_bit(PCI_DPC_RECOVERING, &pdev->priv_flags);
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci	/*
15262306a36Sopenharmony_ci	 * DPC disables the Link automatically in hardware, so it has
15362306a36Sopenharmony_ci	 * already been reset by the time we get here.
15462306a36Sopenharmony_ci	 */
15562306a36Sopenharmony_ci	cap = pdev->dpc_cap;
15662306a36Sopenharmony_ci
15762306a36Sopenharmony_ci	/*
15862306a36Sopenharmony_ci	 * Wait until the Link is inactive, then clear DPC Trigger Status
15962306a36Sopenharmony_ci	 * to allow the Port to leave DPC.
16062306a36Sopenharmony_ci	 */
16162306a36Sopenharmony_ci	if (!pcie_wait_for_link(pdev, false))
16262306a36Sopenharmony_ci		pci_info(pdev, "Data Link Layer Link Active not cleared in 1000 msec\n");
16362306a36Sopenharmony_ci
16462306a36Sopenharmony_ci	if (pdev->dpc_rp_extensions && dpc_wait_rp_inactive(pdev)) {
16562306a36Sopenharmony_ci		clear_bit(PCI_DPC_RECOVERED, &pdev->priv_flags);
16662306a36Sopenharmony_ci		ret = PCI_ERS_RESULT_DISCONNECT;
16762306a36Sopenharmony_ci		goto out;
16862306a36Sopenharmony_ci	}
16962306a36Sopenharmony_ci
17062306a36Sopenharmony_ci	pci_write_config_word(pdev, cap + PCI_EXP_DPC_STATUS,
17162306a36Sopenharmony_ci			      PCI_EXP_DPC_STATUS_TRIGGER);
17262306a36Sopenharmony_ci
17362306a36Sopenharmony_ci	if (pci_bridge_wait_for_secondary_bus(pdev, "DPC")) {
17462306a36Sopenharmony_ci		clear_bit(PCI_DPC_RECOVERED, &pdev->priv_flags);
17562306a36Sopenharmony_ci		ret = PCI_ERS_RESULT_DISCONNECT;
17662306a36Sopenharmony_ci	} else {
17762306a36Sopenharmony_ci		set_bit(PCI_DPC_RECOVERED, &pdev->priv_flags);
17862306a36Sopenharmony_ci		ret = PCI_ERS_RESULT_RECOVERED;
17962306a36Sopenharmony_ci	}
18062306a36Sopenharmony_ciout:
18162306a36Sopenharmony_ci	clear_bit(PCI_DPC_RECOVERING, &pdev->priv_flags);
18262306a36Sopenharmony_ci	wake_up_all(&dpc_completed_waitqueue);
18362306a36Sopenharmony_ci	return ret;
18462306a36Sopenharmony_ci}
18562306a36Sopenharmony_ci
18662306a36Sopenharmony_cistatic void dpc_process_rp_pio_error(struct pci_dev *pdev)
18762306a36Sopenharmony_ci{
18862306a36Sopenharmony_ci	u16 cap = pdev->dpc_cap, dpc_status, first_error;
18962306a36Sopenharmony_ci	u32 status, mask, sev, syserr, exc, dw0, dw1, dw2, dw3, log, prefix;
19062306a36Sopenharmony_ci	int i;
19162306a36Sopenharmony_ci
19262306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_STATUS, &status);
19362306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_MASK, &mask);
19462306a36Sopenharmony_ci	pci_err(pdev, "rp_pio_status: %#010x, rp_pio_mask: %#010x\n",
19562306a36Sopenharmony_ci		status, mask);
19662306a36Sopenharmony_ci
19762306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_SEVERITY, &sev);
19862306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_SYSERROR, &syserr);
19962306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_EXCEPTION, &exc);
20062306a36Sopenharmony_ci	pci_err(pdev, "RP PIO severity=%#010x, syserror=%#010x, exception=%#010x\n",
20162306a36Sopenharmony_ci		sev, syserr, exc);
20262306a36Sopenharmony_ci
20362306a36Sopenharmony_ci	/* Get First Error Pointer */
20462306a36Sopenharmony_ci	pci_read_config_word(pdev, cap + PCI_EXP_DPC_STATUS, &dpc_status);
20562306a36Sopenharmony_ci	first_error = (dpc_status & 0x1f00) >> 8;
20662306a36Sopenharmony_ci
20762306a36Sopenharmony_ci	for (i = 0; i < ARRAY_SIZE(rp_pio_error_string); i++) {
20862306a36Sopenharmony_ci		if ((status & ~mask) & (1 << i))
20962306a36Sopenharmony_ci			pci_err(pdev, "[%2d] %s%s\n", i, rp_pio_error_string[i],
21062306a36Sopenharmony_ci				first_error == i ? " (First)" : "");
21162306a36Sopenharmony_ci	}
21262306a36Sopenharmony_ci
21362306a36Sopenharmony_ci	if (pdev->dpc_rp_log_size < 4)
21462306a36Sopenharmony_ci		goto clear_status;
21562306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_HEADER_LOG,
21662306a36Sopenharmony_ci			      &dw0);
21762306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_HEADER_LOG + 4,
21862306a36Sopenharmony_ci			      &dw1);
21962306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_HEADER_LOG + 8,
22062306a36Sopenharmony_ci			      &dw2);
22162306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_HEADER_LOG + 12,
22262306a36Sopenharmony_ci			      &dw3);
22362306a36Sopenharmony_ci	pci_err(pdev, "TLP Header: %#010x %#010x %#010x %#010x\n",
22462306a36Sopenharmony_ci		dw0, dw1, dw2, dw3);
22562306a36Sopenharmony_ci
22662306a36Sopenharmony_ci	if (pdev->dpc_rp_log_size < 5)
22762306a36Sopenharmony_ci		goto clear_status;
22862306a36Sopenharmony_ci	pci_read_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_IMPSPEC_LOG, &log);
22962306a36Sopenharmony_ci	pci_err(pdev, "RP PIO ImpSpec Log %#010x\n", log);
23062306a36Sopenharmony_ci
23162306a36Sopenharmony_ci	for (i = 0; i < pdev->dpc_rp_log_size - 5; i++) {
23262306a36Sopenharmony_ci		pci_read_config_dword(pdev,
23362306a36Sopenharmony_ci			cap + PCI_EXP_DPC_RP_PIO_TLPPREFIX_LOG + i * 4, &prefix);
23462306a36Sopenharmony_ci		pci_err(pdev, "TLP Prefix Header: dw%d, %#010x\n", i, prefix);
23562306a36Sopenharmony_ci	}
23662306a36Sopenharmony_ci clear_status:
23762306a36Sopenharmony_ci	pci_write_config_dword(pdev, cap + PCI_EXP_DPC_RP_PIO_STATUS, status);
23862306a36Sopenharmony_ci}
23962306a36Sopenharmony_ci
24062306a36Sopenharmony_cistatic int dpc_get_aer_uncorrect_severity(struct pci_dev *dev,
24162306a36Sopenharmony_ci					  struct aer_err_info *info)
24262306a36Sopenharmony_ci{
24362306a36Sopenharmony_ci	int pos = dev->aer_cap;
24462306a36Sopenharmony_ci	u32 status, mask, sev;
24562306a36Sopenharmony_ci
24662306a36Sopenharmony_ci	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_STATUS, &status);
24762306a36Sopenharmony_ci	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_MASK, &mask);
24862306a36Sopenharmony_ci	status &= ~mask;
24962306a36Sopenharmony_ci	if (!status)
25062306a36Sopenharmony_ci		return 0;
25162306a36Sopenharmony_ci
25262306a36Sopenharmony_ci	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_SEVER, &sev);
25362306a36Sopenharmony_ci	status &= sev;
25462306a36Sopenharmony_ci	if (status)
25562306a36Sopenharmony_ci		info->severity = AER_FATAL;
25662306a36Sopenharmony_ci	else
25762306a36Sopenharmony_ci		info->severity = AER_NONFATAL;
25862306a36Sopenharmony_ci
25962306a36Sopenharmony_ci	return 1;
26062306a36Sopenharmony_ci}
26162306a36Sopenharmony_ci
26262306a36Sopenharmony_civoid dpc_process_error(struct pci_dev *pdev)
26362306a36Sopenharmony_ci{
26462306a36Sopenharmony_ci	u16 cap = pdev->dpc_cap, status, source, reason, ext_reason;
26562306a36Sopenharmony_ci	struct aer_err_info info;
26662306a36Sopenharmony_ci
26762306a36Sopenharmony_ci	pci_read_config_word(pdev, cap + PCI_EXP_DPC_STATUS, &status);
26862306a36Sopenharmony_ci	pci_read_config_word(pdev, cap + PCI_EXP_DPC_SOURCE_ID, &source);
26962306a36Sopenharmony_ci
27062306a36Sopenharmony_ci	pci_info(pdev, "containment event, status:%#06x source:%#06x\n",
27162306a36Sopenharmony_ci		 status, source);
27262306a36Sopenharmony_ci
27362306a36Sopenharmony_ci	reason = (status & PCI_EXP_DPC_STATUS_TRIGGER_RSN) >> 1;
27462306a36Sopenharmony_ci	ext_reason = (status & PCI_EXP_DPC_STATUS_TRIGGER_RSN_EXT) >> 5;
27562306a36Sopenharmony_ci	pci_warn(pdev, "%s detected\n",
27662306a36Sopenharmony_ci		 (reason == 0) ? "unmasked uncorrectable error" :
27762306a36Sopenharmony_ci		 (reason == 1) ? "ERR_NONFATAL" :
27862306a36Sopenharmony_ci		 (reason == 2) ? "ERR_FATAL" :
27962306a36Sopenharmony_ci		 (ext_reason == 0) ? "RP PIO error" :
28062306a36Sopenharmony_ci		 (ext_reason == 1) ? "software trigger" :
28162306a36Sopenharmony_ci				     "reserved error");
28262306a36Sopenharmony_ci
28362306a36Sopenharmony_ci	/* show RP PIO error detail information */
28462306a36Sopenharmony_ci	if (pdev->dpc_rp_extensions && reason == 3 && ext_reason == 0)
28562306a36Sopenharmony_ci		dpc_process_rp_pio_error(pdev);
28662306a36Sopenharmony_ci	else if (reason == 0 &&
28762306a36Sopenharmony_ci		 dpc_get_aer_uncorrect_severity(pdev, &info) &&
28862306a36Sopenharmony_ci		 aer_get_device_error_info(pdev, &info)) {
28962306a36Sopenharmony_ci		aer_print_error(pdev, &info);
29062306a36Sopenharmony_ci		pci_aer_clear_nonfatal_status(pdev);
29162306a36Sopenharmony_ci		pci_aer_clear_fatal_status(pdev);
29262306a36Sopenharmony_ci	}
29362306a36Sopenharmony_ci}
29462306a36Sopenharmony_ci
29562306a36Sopenharmony_cistatic irqreturn_t dpc_handler(int irq, void *context)
29662306a36Sopenharmony_ci{
29762306a36Sopenharmony_ci	struct pci_dev *pdev = context;
29862306a36Sopenharmony_ci
29962306a36Sopenharmony_ci	dpc_process_error(pdev);
30062306a36Sopenharmony_ci
30162306a36Sopenharmony_ci	/* We configure DPC so it only triggers on ERR_FATAL */
30262306a36Sopenharmony_ci	pcie_do_recovery(pdev, pci_channel_io_frozen, dpc_reset_link);
30362306a36Sopenharmony_ci
30462306a36Sopenharmony_ci	return IRQ_HANDLED;
30562306a36Sopenharmony_ci}
30662306a36Sopenharmony_ci
30762306a36Sopenharmony_cistatic irqreturn_t dpc_irq(int irq, void *context)
30862306a36Sopenharmony_ci{
30962306a36Sopenharmony_ci	struct pci_dev *pdev = context;
31062306a36Sopenharmony_ci	u16 cap = pdev->dpc_cap, status;
31162306a36Sopenharmony_ci
31262306a36Sopenharmony_ci	pci_read_config_word(pdev, cap + PCI_EXP_DPC_STATUS, &status);
31362306a36Sopenharmony_ci
31462306a36Sopenharmony_ci	if (!(status & PCI_EXP_DPC_STATUS_INTERRUPT) || PCI_POSSIBLE_ERROR(status))
31562306a36Sopenharmony_ci		return IRQ_NONE;
31662306a36Sopenharmony_ci
31762306a36Sopenharmony_ci	pci_write_config_word(pdev, cap + PCI_EXP_DPC_STATUS,
31862306a36Sopenharmony_ci			      PCI_EXP_DPC_STATUS_INTERRUPT);
31962306a36Sopenharmony_ci	if (status & PCI_EXP_DPC_STATUS_TRIGGER)
32062306a36Sopenharmony_ci		return IRQ_WAKE_THREAD;
32162306a36Sopenharmony_ci	return IRQ_HANDLED;
32262306a36Sopenharmony_ci}
32362306a36Sopenharmony_ci
32462306a36Sopenharmony_civoid pci_dpc_init(struct pci_dev *pdev)
32562306a36Sopenharmony_ci{
32662306a36Sopenharmony_ci	u16 cap;
32762306a36Sopenharmony_ci
32862306a36Sopenharmony_ci	pdev->dpc_cap = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_DPC);
32962306a36Sopenharmony_ci	if (!pdev->dpc_cap)
33062306a36Sopenharmony_ci		return;
33162306a36Sopenharmony_ci
33262306a36Sopenharmony_ci	pci_read_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CAP, &cap);
33362306a36Sopenharmony_ci	if (!(cap & PCI_EXP_DPC_CAP_RP_EXT))
33462306a36Sopenharmony_ci		return;
33562306a36Sopenharmony_ci
33662306a36Sopenharmony_ci	pdev->dpc_rp_extensions = true;
33762306a36Sopenharmony_ci
33862306a36Sopenharmony_ci	/* Quirks may set dpc_rp_log_size if device or firmware is buggy */
33962306a36Sopenharmony_ci	if (!pdev->dpc_rp_log_size) {
34062306a36Sopenharmony_ci		pdev->dpc_rp_log_size =
34162306a36Sopenharmony_ci			(cap & PCI_EXP_DPC_RP_PIO_LOG_SIZE) >> 8;
34262306a36Sopenharmony_ci		if (pdev->dpc_rp_log_size < 4 || pdev->dpc_rp_log_size > 9) {
34362306a36Sopenharmony_ci			pci_err(pdev, "RP PIO log size %u is invalid\n",
34462306a36Sopenharmony_ci				pdev->dpc_rp_log_size);
34562306a36Sopenharmony_ci			pdev->dpc_rp_log_size = 0;
34662306a36Sopenharmony_ci		}
34762306a36Sopenharmony_ci	}
34862306a36Sopenharmony_ci}
34962306a36Sopenharmony_ci
35062306a36Sopenharmony_ci#define FLAG(x, y) (((x) & (y)) ? '+' : '-')
35162306a36Sopenharmony_cistatic int dpc_probe(struct pcie_device *dev)
35262306a36Sopenharmony_ci{
35362306a36Sopenharmony_ci	struct pci_dev *pdev = dev->port;
35462306a36Sopenharmony_ci	struct device *device = &dev->device;
35562306a36Sopenharmony_ci	int status;
35662306a36Sopenharmony_ci	u16 ctl, cap;
35762306a36Sopenharmony_ci
35862306a36Sopenharmony_ci	if (!pcie_aer_is_native(pdev) && !pcie_ports_dpc_native)
35962306a36Sopenharmony_ci		return -ENOTSUPP;
36062306a36Sopenharmony_ci
36162306a36Sopenharmony_ci	status = devm_request_threaded_irq(device, dev->irq, dpc_irq,
36262306a36Sopenharmony_ci					   dpc_handler, IRQF_SHARED,
36362306a36Sopenharmony_ci					   "pcie-dpc", pdev);
36462306a36Sopenharmony_ci	if (status) {
36562306a36Sopenharmony_ci		pci_warn(pdev, "request IRQ%d failed: %d\n", dev->irq,
36662306a36Sopenharmony_ci			 status);
36762306a36Sopenharmony_ci		return status;
36862306a36Sopenharmony_ci	}
36962306a36Sopenharmony_ci
37062306a36Sopenharmony_ci	pci_read_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CAP, &cap);
37162306a36Sopenharmony_ci	pci_read_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CTL, &ctl);
37262306a36Sopenharmony_ci
37362306a36Sopenharmony_ci	ctl = (ctl & 0xfff4) | PCI_EXP_DPC_CTL_EN_FATAL | PCI_EXP_DPC_CTL_INT_EN;
37462306a36Sopenharmony_ci	pci_write_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CTL, ctl);
37562306a36Sopenharmony_ci	pci_info(pdev, "enabled with IRQ %d\n", dev->irq);
37662306a36Sopenharmony_ci
37762306a36Sopenharmony_ci	pci_info(pdev, "error containment capabilities: Int Msg #%d, RPExt%c PoisonedTLP%c SwTrigger%c RP PIO Log %d, DL_ActiveErr%c\n",
37862306a36Sopenharmony_ci		 cap & PCI_EXP_DPC_IRQ, FLAG(cap, PCI_EXP_DPC_CAP_RP_EXT),
37962306a36Sopenharmony_ci		 FLAG(cap, PCI_EXP_DPC_CAP_POISONED_TLP),
38062306a36Sopenharmony_ci		 FLAG(cap, PCI_EXP_DPC_CAP_SW_TRIGGER), pdev->dpc_rp_log_size,
38162306a36Sopenharmony_ci		 FLAG(cap, PCI_EXP_DPC_CAP_DL_ACTIVE));
38262306a36Sopenharmony_ci
38362306a36Sopenharmony_ci	pci_add_ext_cap_save_buffer(pdev, PCI_EXT_CAP_ID_DPC, sizeof(u16));
38462306a36Sopenharmony_ci	return status;
38562306a36Sopenharmony_ci}
38662306a36Sopenharmony_ci
38762306a36Sopenharmony_cistatic void dpc_remove(struct pcie_device *dev)
38862306a36Sopenharmony_ci{
38962306a36Sopenharmony_ci	struct pci_dev *pdev = dev->port;
39062306a36Sopenharmony_ci	u16 ctl;
39162306a36Sopenharmony_ci
39262306a36Sopenharmony_ci	pci_read_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CTL, &ctl);
39362306a36Sopenharmony_ci	ctl &= ~(PCI_EXP_DPC_CTL_EN_FATAL | PCI_EXP_DPC_CTL_INT_EN);
39462306a36Sopenharmony_ci	pci_write_config_word(pdev, pdev->dpc_cap + PCI_EXP_DPC_CTL, ctl);
39562306a36Sopenharmony_ci}
39662306a36Sopenharmony_ci
39762306a36Sopenharmony_cistatic struct pcie_port_service_driver dpcdriver = {
39862306a36Sopenharmony_ci	.name		= "dpc",
39962306a36Sopenharmony_ci	.port_type	= PCIE_ANY_PORT,
40062306a36Sopenharmony_ci	.service	= PCIE_PORT_SERVICE_DPC,
40162306a36Sopenharmony_ci	.probe		= dpc_probe,
40262306a36Sopenharmony_ci	.remove		= dpc_remove,
40362306a36Sopenharmony_ci};
40462306a36Sopenharmony_ci
40562306a36Sopenharmony_ciint __init pcie_dpc_init(void)
40662306a36Sopenharmony_ci{
40762306a36Sopenharmony_ci	return pcie_port_service_register(&dpcdriver);
40862306a36Sopenharmony_ci}
409