[deliverable/linux.git] / drivers / pci / pcie / aer / aerdrv.c

/*
 * drivers/pci/pcie/aer/aerdrv.c
 *
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive
 * for more details.
 *
 * This file implements the AER root port service driver. The driver will
 * register an irq handler. When root port triggers an AER interrupt, the irq
 * handler will collect root port status and schedule a work.
 *
 * Copyright (C) 2006 Intel Corp.
 *	Tom Long Nguyen (tom.l.nguyen@intel.com)
 *	Zhang Yanmin (yanmin.zhang@intel.com)
 *
 */

#include <linux/module.h>
#include <linux/pci.h>
#include <linux/pci-acpi.h>
#include <linux/sched.h>
#include <linux/kernel.h>
#include <linux/errno.h>
#include <linux/pm.h>
#include <linux/init.h>
#include <linux/interrupt.h>
#include <linux/delay.h>
#include <linux/pcieport_if.h>
#include <linux/slab.h>

#include "aerdrv.h"
#include "../../pci.h"

/*
 * Version Information
 */
#define DRIVER_VERSION "v1.0"
#define DRIVER_AUTHOR "tom.l.nguyen@intel.com"
#define DRIVER_DESC "Root Port Advanced Error Reporting Driver"
MODULE_AUTHOR(DRIVER_AUTHOR);
MODULE_DESCRIPTION(DRIVER_DESC);
MODULE_LICENSE("GPL");

static int aer_probe(struct pcie_device *dev);
static void aer_remove(struct pcie_device *dev);
static pci_ers_result_t aer_error_detected(struct pci_dev *dev,
	enum pci_channel_state error);
static void aer_error_resume(struct pci_dev *dev);
static pci_ers_result_t aer_root_reset(struct pci_dev *dev);

static const struct pci_error_handlers aer_error_handlers = {
	.error_detected = aer_error_detected,
	.resume		= aer_error_resume,
};

static struct pcie_port_service_driver aerdriver = {
	.name		= "aer",
	.port_type	= PCI_EXP_TYPE_ROOT_PORT,
	.service	= PCIE_PORT_SERVICE_AER,

	.probe		= aer_probe,
	.remove		= aer_remove,

	.err_handler	= &aer_error_handlers,

	.reset_link	= aer_root_reset,
};

static int pcie_aer_disable;

void pci_no_aer(void)
{
	pcie_aer_disable = 1;	/* has priority over 'forceload' */
}

bool pci_aer_available(void)
{
	return !pcie_aer_disable && pci_msi_enabled();
}

static int set_device_error_reporting(struct pci_dev *dev, void *data)
{
	bool enable = *((bool *)data);
	int type = pci_pcie_type(dev);

	if ((type == PCI_EXP_TYPE_ROOT_PORT) ||
	    (type == PCI_EXP_TYPE_UPSTREAM) ||
	    (type == PCI_EXP_TYPE_DOWNSTREAM)) {
		if (enable)
			pci_enable_pcie_error_reporting(dev);
		else
			pci_disable_pcie_error_reporting(dev);
	}

	if (enable)
		pcie_set_ecrc_checking(dev);

	return 0;
}

/**
 * set_downstream_devices_error_reporting - enable/disable the error reporting  bits on the root port and its downstream ports.
 * @dev: pointer to root port's pci_dev data structure
 * @enable: true = enable error reporting, false = disable error reporting.
 */
static void set_downstream_devices_error_reporting(struct pci_dev *dev,
						   bool enable)
{
	set_device_error_reporting(dev, &enable);

	if (!dev->subordinate)
		return;
	pci_walk_bus(dev->subordinate, set_device_error_reporting, &enable);
}

/**
 * aer_enable_rootport - enable Root Port's interrupts when receiving messages
 * @rpc: pointer to a Root Port data structure
 *
 * Invoked when PCIe bus loads AER service driver.
 */
static void aer_enable_rootport(struct aer_rpc *rpc)
{
	struct pci_dev *pdev = rpc->rpd->port;
	int aer_pos;
	u16 reg16;
	u32 reg32;

	/* Clear PCIe Capability's Device Status */
	pcie_capability_read_word(pdev, PCI_EXP_DEVSTA, &reg16);
	pcie_capability_write_word(pdev, PCI_EXP_DEVSTA, reg16);

	/* Disable system error generation in response to error messages */
	pcie_capability_clear_word(pdev, PCI_EXP_RTCTL,
				   SYSTEM_ERROR_INTR_ON_MESG_MASK);

	aer_pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ERR);
	/* Clear error status */
	pci_read_config_dword(pdev, aer_pos + PCI_ERR_ROOT_STATUS, &reg32);
	pci_write_config_dword(pdev, aer_pos + PCI_ERR_ROOT_STATUS, reg32);
	pci_read_config_dword(pdev, aer_pos + PCI_ERR_COR_STATUS, &reg32);
	pci_write_config_dword(pdev, aer_pos + PCI_ERR_COR_STATUS, reg32);
	pci_read_config_dword(pdev, aer_pos + PCI_ERR_UNCOR_STATUS, &reg32);
	pci_write_config_dword(pdev, aer_pos + PCI_ERR_UNCOR_STATUS, reg32);

	/*
	 * Enable error reporting for the root port device and downstream port
	 * devices.
	 */
	set_downstream_devices_error_reporting(pdev, true);

	/* Enable Root Port's interrupt in response to error messages */
	pci_read_config_dword(pdev, aer_pos + PCI_ERR_ROOT_COMMAND, &reg32);
	reg32 |= ROOT_PORT_INTR_ON_MESG_MASK;
	pci_write_config_dword(pdev, aer_pos + PCI_ERR_ROOT_COMMAND, reg32);
}

/**
 * aer_disable_rootport - disable Root Port's interrupts when receiving messages
 * @rpc: pointer to a Root Port data structure
 *
 * Invoked when PCIe bus unloads AER service driver.
 */
static void aer_disable_rootport(struct aer_rpc *rpc)
{
	struct pci_dev *pdev = rpc->rpd->port;
	u32 reg32;
	int pos;

	/*
	 * Disable error reporting for the root port device and downstream port
	 * devices.
	 */
	set_downstream_devices_error_reporting(pdev, false);

	pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ERR);
	/* Disable Root's interrupt in response to error messages */
	pci_read_config_dword(pdev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
	pci_write_config_dword(pdev, pos + PCI_ERR_ROOT_COMMAND, reg32);

	/* Clear Root's error status reg */
	pci_read_config_dword(pdev, pos + PCI_ERR_ROOT_STATUS, &reg32);
	pci_write_config_dword(pdev, pos + PCI_ERR_ROOT_STATUS, reg32);
}

/**
 * aer_irq - Root Port's ISR
 * @irq: IRQ assigned to Root Port
 * @context: pointer to Root Port data structure
 *
 * Invoked when Root Port detects AER messages.
 */
irqreturn_t aer_irq(int irq, void *context)
{
	unsigned int status, id;
	struct pcie_device *pdev = (struct pcie_device *)context;
	struct aer_rpc *rpc = get_service_data(pdev);
	int next_prod_idx;
	unsigned long flags;
	int pos;

	pos = pci_find_ext_capability(pdev->port, PCI_EXT_CAP_ID_ERR);
	/*
	 * Must lock access to Root Error Status Reg, Root Error ID Reg,
	 * and Root error producer/consumer index
	 */
	spin_lock_irqsave(&rpc->e_lock, flags);

	/* Read error status */
	pci_read_config_dword(pdev->port, pos + PCI_ERR_ROOT_STATUS, &status);
	if (!(status & (PCI_ERR_ROOT_UNCOR_RCV|PCI_ERR_ROOT_COR_RCV))) {
		spin_unlock_irqrestore(&rpc->e_lock, flags);
		return IRQ_NONE;
	}

	/* Read error source and clear error status */
	pci_read_config_dword(pdev->port, pos + PCI_ERR_ROOT_ERR_SRC, &id);
	pci_write_config_dword(pdev->port, pos + PCI_ERR_ROOT_STATUS, status);

	/* Store error source for later DPC handler */
	next_prod_idx = rpc->prod_idx + 1;
	if (next_prod_idx == AER_ERROR_SOURCES_MAX)
		next_prod_idx = 0;
	if (next_prod_idx == rpc->cons_idx) {
		/*
		 * Error Storm Condition - possibly the same error occurred.
		 * Drop the error.
		 */
		spin_unlock_irqrestore(&rpc->e_lock, flags);
		return IRQ_HANDLED;
	}
	rpc->e_sources[rpc->prod_idx].status =  status;
	rpc->e_sources[rpc->prod_idx].id = id;
	rpc->prod_idx = next_prod_idx;
	spin_unlock_irqrestore(&rpc->e_lock, flags);

	/*  Invoke DPC handler */
	schedule_work(&rpc->dpc_handler);

	return IRQ_HANDLED;
}
EXPORT_SYMBOL_GPL(aer_irq);

/**
 * aer_alloc_rpc - allocate Root Port data structure
 * @dev: pointer to the pcie_dev data structure
 *
 * Invoked when Root Port's AER service is loaded.
 */
static struct aer_rpc *aer_alloc_rpc(struct pcie_device *dev)
{
	struct aer_rpc *rpc;

	rpc = kzalloc(sizeof(struct aer_rpc), GFP_KERNEL);
	if (!rpc)
		return NULL;

	/* Initialize Root lock access, e_lock, to Root Error Status Reg */
	spin_lock_init(&rpc->e_lock);

	rpc->rpd = dev;
	INIT_WORK(&rpc->dpc_handler, aer_isr);
	mutex_init(&rpc->rpc_mutex);
	init_waitqueue_head(&rpc->wait_release);

	/* Use PCIe bus function to store rpc into PCIe device */
	set_service_data(dev, rpc);

	return rpc;
}

/**
 * aer_remove - clean up resources
 * @dev: pointer to the pcie_dev data structure
 *
 * Invoked when PCI Express bus unloads or AER probe fails.
 */
static void aer_remove(struct pcie_device *dev)
{
	struct aer_rpc *rpc = get_service_data(dev);

	if (rpc) {
		/* If register interrupt service, it must be free. */
		if (rpc->isr)
			free_irq(dev->irq, dev);

		wait_event(rpc->wait_release, rpc->prod_idx == rpc->cons_idx);

		aer_disable_rootport(rpc);
		kfree(rpc);
		set_service_data(dev, NULL);
	}
}

/**
 * aer_probe - initialize resources
 * @dev: pointer to the pcie_dev data structure
 * @id: pointer to the service id data structure
 *
 * Invoked when PCI Express bus loads AER service driver.
 */
static int aer_probe(struct pcie_device *dev)
{
	int status;
	struct aer_rpc *rpc;
	struct device *device = &dev->device;

	/* Init */
	status = aer_init(dev);
	if (status)
		return status;

	/* Alloc rpc data structure */
	rpc = aer_alloc_rpc(dev);
	if (!rpc) {
		dev_printk(KERN_DEBUG, device, "alloc rpc failed\n");
		aer_remove(dev);
		return -ENOMEM;
	}

	/* Request IRQ ISR */
	status = request_irq(dev->irq, aer_irq, IRQF_SHARED, "aerdrv", dev);
	if (status) {
		dev_printk(KERN_DEBUG, device, "request IRQ failed\n");
		aer_remove(dev);
		return status;
	}

	rpc->isr = 1;

	aer_enable_rootport(rpc);

	return status;
}

/**
 * aer_root_reset - reset link on Root Port
 * @dev: pointer to Root Port's pci_dev data structure
 *
 * Invoked by Port Bus driver when performing link reset at Root Port.
 */
static pci_ers_result_t aer_root_reset(struct pci_dev *dev)
{
	u32 reg32;
	int pos;

	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ERR);

	/* Disable Root's interrupt in response to error messages */
	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, reg32);

	aer_do_secondary_bus_reset(dev);
	dev_printk(KERN_DEBUG, &dev->dev, "Root Port link has been reset\n");

	/* Clear Root Error Status */
	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_STATUS, &reg32);
	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_STATUS, reg32);

	/* Enable Root Port's interrupt in response to error messages */
	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	reg32 |= ROOT_PORT_INTR_ON_MESG_MASK;
	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, reg32);

	return PCI_ERS_RESULT_RECOVERED;
}

/**
 * aer_error_detected - update severity status
 * @dev: pointer to Root Port's pci_dev data structure
 * @error: error severity being notified by port bus
 *
 * Invoked by Port Bus driver during error recovery.
 */
static pci_ers_result_t aer_error_detected(struct pci_dev *dev,
			enum pci_channel_state error)
{
	/* Root Port has no impact. Always recovers. */
	return PCI_ERS_RESULT_CAN_RECOVER;
}

/**
 * aer_error_resume - clean up corresponding error status bits
 * @dev: pointer to Root Port's pci_dev data structure
 *
 * Invoked by Port Bus driver during nonfatal recovery.
 */
static void aer_error_resume(struct pci_dev *dev)
{
	int pos;
	u32 status, mask;
	u16 reg16;

	/* Clean up Root device status */
	pcie_capability_read_word(dev, PCI_EXP_DEVSTA, &reg16);
	pcie_capability_write_word(dev, PCI_EXP_DEVSTA, reg16);

	/* Clean AER Root Error Status */
	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ERR);
	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_STATUS, &status);
	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_SEVER, &mask);
	if (dev->error_state == pci_channel_io_normal)
		status &= ~mask; /* Clear corresponding nonfatal bits */
	else
		status &= mask; /* Clear corresponding fatal bits */
	pci_write_config_dword(dev, pos + PCI_ERR_UNCOR_STATUS, status);
}

/**
 * aer_service_init - register AER root service driver
 *
 * Invoked when AER root service driver is loaded.
 */
static int __init aer_service_init(void)
{
	if (!pci_aer_available() || aer_acpi_firmware_first())
		return -ENXIO;
	return pcie_port_service_register(&aerdriver);
}

/**
 * aer_service_exit - unregister AER root service driver
 *
 * Invoked when AER root service driver is unloaded.
 */
static void __exit aer_service_exit(void)
{
	pcie_port_service_unregister(&aerdriver);
}

module_init(aer_service_init);
module_exit(aer_service_exit);
Commit	Line	Data
6c2b374d ZY	1	/*
	2	* drivers/pci/pcie/aer/aerdrv.c
	3	*
	4	* This file is subject to the terms and conditions of the GNU General Public
	5	* License. See the file "COPYING" in the main directory of this archive
	6	* for more details.
	7	*
	8	* This file implements the AER root port service driver. The driver will
	9	* register an irq handler. When root port triggers an AER interrupt, the irq
	10	* handler will collect root port status and schedule a work.
	11	*
	12	* Copyright (C) 2006 Intel Corp.
	13	* Tom Long Nguyen (tom.l.nguyen@intel.com)
	14	* Zhang Yanmin (yanmin.zhang@intel.com)
	15	*
	16	*/
	17
	18	#include <linux/module.h>
	19	#include <linux/pci.h>
415e12b2	20	#include <linux/pci-acpi.h>
d43c36dc	21	#include <linux/sched.h>
6c2b374d ZY	22	#include <linux/kernel.h>
	23	#include <linux/errno.h>
	24	#include <linux/pm.h>
	25	#include <linux/init.h>
	26	#include <linux/interrupt.h>
	27	#include <linux/delay.h>
	28	#include <linux/pcieport_if.h>
5a0e3ad6	29	#include <linux/slab.h>
6c2b374d ZY	30
6c2b374d ZY	31	#include "aerdrv.h"
5d9526d0	32	#include "../../pci.h"
6c2b374d ZY	33
	34	/*
	35	* Version Information
	36	*/
	37	#define DRIVER_VERSION "v1.0"
	38	#define DRIVER_AUTHOR "tom.l.nguyen@intel.com"
	39	#define DRIVER_DESC "Root Port Advanced Error Reporting Driver"
	40	MODULE_AUTHOR(DRIVER_AUTHOR);
	41	MODULE_DESCRIPTION(DRIVER_DESC);
	42	MODULE_LICENSE("GPL");
	43
15856ad5	44	static int aer_probe(struct pcie_device *dev);
6c2b374d	45	static void aer_remove(struct pcie_device *dev);
6c2b374d ZY	46	static pci_ers_result_t aer_error_detected(struct pci_dev *dev,
	47	enum pci_channel_state error);
	48	static void aer_error_resume(struct pci_dev *dev);
	49	static pci_ers_result_t aer_root_reset(struct pci_dev *dev);
	50
49453028	51	static const struct pci_error_handlers aer_error_handlers = {
6c2b374d	52	.error_detected = aer_error_detected,
c9a91883	53	.resume = aer_error_resume,
6c2b374d ZY	54	};
6c2b374d ZY	55
c1996c27	56	static struct pcie_port_service_driver aerdriver = {
6c2b374d	57	.name = "aer",
694f88ef	58	.port_type = PCI_EXP_TYPE_ROOT_PORT,
22106368	59	.service = PCIE_PORT_SERVICE_AER,
6c2b374d ZY	60
	61	.probe = aer_probe,
	62	.remove = aer_remove,
	63
6c2b374d ZY	64	.err_handler = &aer_error_handlers,
	65
	66	.reset_link = aer_root_reset,
	67	};
	68
7f785763 RD	69	static int pcie_aer_disable;
	70
	71	void pci_no_aer(void)
	72	{
	73	pcie_aer_disable = 1; /* has priority over 'forceload' */
	74	}
	75
f1a7bfaf RW	76	bool pci_aer_available(void)
	77	{
	78	return !pcie_aer_disable && pci_msi_enabled();
	79	}
	80
843f4697 HS	81	static int set_device_error_reporting(struct pci_dev dev, void data)
	82	{
	83	bool enable = ((bool )data);
62f87c0e	84	int type = pci_pcie_type(dev);
843f4697	85
62f87c0e YW	86	if ((type == PCI_EXP_TYPE_ROOT_PORT) \|\|
	87	(type == PCI_EXP_TYPE_UPSTREAM) \|\|
	88	(type == PCI_EXP_TYPE_DOWNSTREAM)) {
843f4697 HS	89	if (enable)
	90	pci_enable_pcie_error_reporting(dev);
	91	else
	92	pci_disable_pcie_error_reporting(dev);
	93	}
	94
	95	if (enable)
	96	pcie_set_ecrc_checking(dev);
	97
	98	return 0;
	99	}
	100
	101	/**
	102	* set_downstream_devices_error_reporting - enable/disable the error reporting bits on the root port and its downstream ports.
	103	* @dev: pointer to root port's pci_dev data structure
	104	* @enable: true = enable error reporting, false = disable error reporting.
	105	*/
	106	static void set_downstream_devices_error_reporting(struct pci_dev *dev,
	107	bool enable)
	108	{
	109	set_device_error_reporting(dev, &enable);
	110
	111	if (!dev->subordinate)
	112	return;
	113	pci_walk_bus(dev->subordinate, set_device_error_reporting, &enable);
	114	}
	115
	116	/**
	117	* aer_enable_rootport - enable Root Port's interrupts when receiving messages
	118	* @rpc: pointer to a Root Port data structure
	119	*
	120	* Invoked when PCIe bus loads AER service driver.
	121	*/
	122	static void aer_enable_rootport(struct aer_rpc *rpc)
	123	{
	124	struct pci_dev *pdev = rpc->rpd->port;
43bd4ee8	125	int aer_pos;
843f4697 HS	126	u16 reg16;
	127	u32 reg32;
	128
843f4697	129	/* Clear PCIe Capability's Device Status */
43bd4ee8 JL	130	pcie_capability_read_word(pdev, PCI_EXP_DEVSTA, &reg16);
43bd4ee8 JL	131	pcie_capability_write_word(pdev, PCI_EXP_DEVSTA, reg16);
843f4697 HS	132
843f4697 HS	133	/* Disable system error generation in response to error messages */
43bd4ee8 JL	134	pcie_capability_clear_word(pdev, PCI_EXP_RTCTL,
43bd4ee8 JL	135	SYSTEM_ERROR_INTR_ON_MESG_MASK);
843f4697 HS	136
	137	aer_pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ERR);
	138	/* Clear error status */
	139	pci_read_config_dword(pdev, aer_pos + PCI_ERR_ROOT_STATUS, &reg32);
	140	pci_write_config_dword(pdev, aer_pos + PCI_ERR_ROOT_STATUS, reg32);
	141	pci_read_config_dword(pdev, aer_pos + PCI_ERR_COR_STATUS, &reg32);
	142	pci_write_config_dword(pdev, aer_pos + PCI_ERR_COR_STATUS, reg32);
	143	pci_read_config_dword(pdev, aer_pos + PCI_ERR_UNCOR_STATUS, &reg32);
	144	pci_write_config_dword(pdev, aer_pos + PCI_ERR_UNCOR_STATUS, reg32);
	145
	146	/*
	147	* Enable error reporting for the root port device and downstream port
	148	* devices.
	149	*/
	150	set_downstream_devices_error_reporting(pdev, true);
	151
	152	/* Enable Root Port's interrupt in response to error messages */
	153	pci_read_config_dword(pdev, aer_pos + PCI_ERR_ROOT_COMMAND, &reg32);
	154	reg32 \|= ROOT_PORT_INTR_ON_MESG_MASK;
	155	pci_write_config_dword(pdev, aer_pos + PCI_ERR_ROOT_COMMAND, reg32);
	156	}
	157
	158	/**
	159	* aer_disable_rootport - disable Root Port's interrupts when receiving messages
	160	* @rpc: pointer to a Root Port data structure
	161	*
	162	* Invoked when PCIe bus unloads AER service driver.
	163	*/
	164	static void aer_disable_rootport(struct aer_rpc *rpc)
	165	{
	166	struct pci_dev *pdev = rpc->rpd->port;
	167	u32 reg32;
	168	int pos;
	169
	170	/*
	171	* Disable error reporting for the root port device and downstream port
	172	* devices.
	173	*/
	174	set_downstream_devices_error_reporting(pdev, false);
	175
	176	pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_ERR);
	177	/* Disable Root's interrupt in response to error messages */
	178	pci_read_config_dword(pdev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	179	reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
	180	pci_write_config_dword(pdev, pos + PCI_ERR_ROOT_COMMAND, reg32);
	181
	182	/* Clear Root's error status reg */
	183	pci_read_config_dword(pdev, pos + PCI_ERR_ROOT_STATUS, &reg32);
	184	pci_write_config_dword(pdev, pos + PCI_ERR_ROOT_STATUS, reg32);
	185	}
	186
6c2b374d ZY	187	/**
	188	* aer_irq - Root Port's ISR
	189	* @irq: IRQ assigned to Root Port
	190	* @context: pointer to Root Port data structure
6c2b374d ZY	191	*
6c2b374d ZY	192	* Invoked when Root Port detects AER messages.
f6d37800	193	*/
634deb02	194	irqreturn_t aer_irq(int irq, void *context)
6c2b374d ZY	195	{
	196	unsigned int status, id;
	197	struct pcie_device pdev = (struct pcie_device )context;
	198	struct aer_rpc *rpc = get_service_data(pdev);
	199	int next_prod_idx;
	200	unsigned long flags;
	201	int pos;
	202
0927678f	203	pos = pci_find_ext_capability(pdev->port, PCI_EXT_CAP_ID_ERR);
6c2b374d ZY	204	/*
	205	* Must lock access to Root Error Status Reg, Root Error ID Reg,
	206	* and Root error producer/consumer index
	207	*/
	208	spin_lock_irqsave(&rpc->e_lock, flags);
	209
	210	/* Read error status */
	211	pci_read_config_dword(pdev->port, pos + PCI_ERR_ROOT_STATUS, &status);
e167bfca	212	if (!(status & (PCI_ERR_ROOT_UNCOR_RCV\|PCI_ERR_ROOT_COR_RCV))) {
6c2b374d ZY	213	spin_unlock_irqrestore(&rpc->e_lock, flags);
	214	return IRQ_NONE;
	215	}
	216
	217	/* Read error source and clear error status */
f647a44f	218	pci_read_config_dword(pdev->port, pos + PCI_ERR_ROOT_ERR_SRC, &id);
6c2b374d ZY	219	pci_write_config_dword(pdev->port, pos + PCI_ERR_ROOT_STATUS, status);
	220
	221	/* Store error source for later DPC handler */
	222	next_prod_idx = rpc->prod_idx + 1;
	223	if (next_prod_idx == AER_ERROR_SOURCES_MAX)
	224	next_prod_idx = 0;
	225	if (next_prod_idx == rpc->cons_idx) {
	226	/*
	227	* Error Storm Condition - possibly the same error occurred.
	228	* Drop the error.
	229	*/
	230	spin_unlock_irqrestore(&rpc->e_lock, flags);
	231	return IRQ_HANDLED;
	232	}
	233	rpc->e_sources[rpc->prod_idx].status = status;
	234	rpc->e_sources[rpc->prod_idx].id = id;
	235	rpc->prod_idx = next_prod_idx;
	236	spin_unlock_irqrestore(&rpc->e_lock, flags);
	237
	238	/* Invoke DPC handler */
	239	schedule_work(&rpc->dpc_handler);
	240
	241	return IRQ_HANDLED;
	242	}
634deb02	243	EXPORT_SYMBOL_GPL(aer_irq);
6c2b374d ZY	244
	245	/**
	246	* aer_alloc_rpc - allocate Root Port data structure
	247	* @dev: pointer to the pcie_dev data structure
	248	*
	249	* Invoked when Root Port's AER service is loaded.
f6d37800	250	*/
c9a91883	251	static struct aer_rpc aer_alloc_rpc(struct pcie_device dev)
6c2b374d ZY	252	{
	253	struct aer_rpc *rpc;
	254
c9a91883 HS	255	rpc = kzalloc(sizeof(struct aer_rpc), GFP_KERNEL);
c9a91883 HS	256	if (!rpc)
6c2b374d ZY	257	return NULL;
6c2b374d ZY	258
f6d37800	259	/* Initialize Root lock access, e_lock, to Root Error Status Reg */
f5609d7e	260	spin_lock_init(&rpc->e_lock);
6c2b374d ZY	261
6c2b374d ZY	262	rpc->rpd = dev;
65f27f38	263	INIT_WORK(&rpc->dpc_handler, aer_isr);
6c2b374d ZY	264	mutex_init(&rpc->rpc_mutex);
	265	init_waitqueue_head(&rpc->wait_release);
	266
45e829ea	267	/* Use PCIe bus function to store rpc into PCIe device */
6c2b374d ZY	268	set_service_data(dev, rpc);
	269
	270	return rpc;
	271	}
	272
	273	/**
	274	* aer_remove - clean up resources
	275	* @dev: pointer to the pcie_dev data structure
	276	*
	277	* Invoked when PCI Express bus unloads or AER probe fails.
f6d37800	278	*/
6c2b374d ZY	279	static void aer_remove(struct pcie_device *dev)
	280	{
	281	struct aer_rpc *rpc = get_service_data(dev);
	282
	283	if (rpc) {
	284	/* If register interrupt service, it must be free. */
	285	if (rpc->isr)
	286	free_irq(dev->irq, dev);
	287
	288	wait_event(rpc->wait_release, rpc->prod_idx == rpc->cons_idx);
	289
460d298d HS	290	aer_disable_rootport(rpc);
460d298d HS	291	kfree(rpc);
6c2b374d ZY	292	set_service_data(dev, NULL);
	293	}
	294	}
	295
	296	/**
	297	* aer_probe - initialize resources
	298	* @dev: pointer to the pcie_dev data structure
	299	* @id: pointer to the service id data structure
	300	*
	301	* Invoked when PCI Express bus loads AER service driver.
f6d37800	302	*/
15856ad5	303	static int aer_probe(struct pcie_device *dev)
6c2b374d ZY	304	{
	305	int status;
	306	struct aer_rpc *rpc;
	307	struct device *device = &dev->device;
	308
	309	/* Init */
c9a91883 HS	310	status = aer_init(dev);
c9a91883 HS	311	if (status)
6c2b374d ZY	312	return status;
	313
	314	/* Alloc rpc data structure */
c9a91883 HS	315	rpc = aer_alloc_rpc(dev);
c9a91883 HS	316	if (!rpc) {
531f254e	317	dev_printk(KERN_DEBUG, device, "alloc rpc failed\n");
6c2b374d ZY	318	aer_remove(dev);
	319	return -ENOMEM;
	320	}
	321
	322	/* Request IRQ ISR */
c9a91883 HS	323	status = request_irq(dev->irq, aer_irq, IRQF_SHARED, "aerdrv", dev);
c9a91883 HS	324	if (status) {
531f254e	325	dev_printk(KERN_DEBUG, device, "request IRQ failed\n");
6c2b374d ZY	326	aer_remove(dev);
	327	return status;
	328	}
	329
	330	rpc->isr = 1;
	331
	332	aer_enable_rootport(rpc);
	333
	334	return status;
	335	}
	336
	337	/**
	338	* aer_root_reset - reset link on Root Port
	339	* @dev: pointer to Root Port's pci_dev data structure
	340	*
	341	* Invoked by Port Bus driver when performing link reset at Root Port.
f6d37800	342	*/
6c2b374d ZY	343	static pci_ers_result_t aer_root_reset(struct pci_dev *dev)
6c2b374d ZY	344	{
c6d34edd	345	u32 reg32;
6c2b374d ZY	346	int pos;
6c2b374d ZY	347
0927678f	348	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ERR);
6c2b374d ZY	349
6c2b374d ZY	350	/* Disable Root's interrupt in response to error messages */
c6d34edd HS	351	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	352	reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
	353	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, reg32);
6c2b374d	354
89713422	355	aer_do_secondary_bus_reset(dev);
531f254e	356	dev_printk(KERN_DEBUG, &dev->dev, "Root Port link has been reset\n");
6c2b374d	357
c6d34edd HS	358	/* Clear Root Error Status */
	359	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_STATUS, &reg32);
	360	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_STATUS, reg32);
	361
6c2b374d	362	/* Enable Root Port's interrupt in response to error messages */
c6d34edd HS	363	pci_read_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, &reg32);
	364	reg32 \|= ROOT_PORT_INTR_ON_MESG_MASK;
	365	pci_write_config_dword(dev, pos + PCI_ERR_ROOT_COMMAND, reg32);
6c2b374d ZY	366
	367	return PCI_ERS_RESULT_RECOVERED;
	368	}
	369
	370	/**
	371	* aer_error_detected - update severity status
	372	* @dev: pointer to Root Port's pci_dev data structure
	373	* @error: error severity being notified by port bus
	374	*
	375	* Invoked by Port Bus driver during error recovery.
f6d37800	376	*/
6c2b374d ZY	377	static pci_ers_result_t aer_error_detected(struct pci_dev *dev,
	378	enum pci_channel_state error)
	379	{
	380	/* Root Port has no impact. Always recovers. */
	381	return PCI_ERS_RESULT_CAN_RECOVER;
	382	}
	383
	384	/**
	385	* aer_error_resume - clean up corresponding error status bits
	386	* @dev: pointer to Root Port's pci_dev data structure
	387	*
	388	* Invoked by Port Bus driver during nonfatal recovery.
f6d37800	389	*/
6c2b374d ZY	390	static void aer_error_resume(struct pci_dev *dev)
	391	{
	392	int pos;
	393	u32 status, mask;
	394	u16 reg16;
	395
	396	/* Clean up Root device status */
43bd4ee8 JL	397	pcie_capability_read_word(dev, PCI_EXP_DEVSTA, &reg16);
43bd4ee8 JL	398	pcie_capability_write_word(dev, PCI_EXP_DEVSTA, reg16);
6c2b374d ZY	399
6c2b374d ZY	400	/* Clean AER Root Error Status */
0927678f	401	pos = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ERR);
6c2b374d ZY	402	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_STATUS, &status);
	403	pci_read_config_dword(dev, pos + PCI_ERR_UNCOR_SEVER, &mask);
	404	if (dev->error_state == pci_channel_io_normal)
	405	status &= ~mask; /* Clear corresponding nonfatal bits */
	406	else
	407	status &= mask; /* Clear corresponding fatal bits */
	408	pci_write_config_dword(dev, pos + PCI_ERR_UNCOR_STATUS, status);
	409	}
	410
	411	/**
	412	* aer_service_init - register AER root service driver
	413	*
	414	* Invoked when AER root service driver is loaded.
f6d37800	415	*/
6c2b374d ZY	416	static int __init aer_service_init(void)
6c2b374d ZY	417	{
b22c3d82	418	if (!pci_aer_available() \|\| aer_acpi_firmware_first())
3e77a3f7	419	return -ENXIO;
c1996c27	420	return pcie_port_service_register(&aerdriver);
6c2b374d ZY	421	}
	422
	423	/**
	424	* aer_service_exit - unregister AER root service driver
	425	*
	426	* Invoked when AER root service driver is unloaded.
f6d37800	427	*/
6c2b374d ZY	428	static void __exit aer_service_exit(void)
6c2b374d ZY	429	{
c1996c27	430	pcie_port_service_unregister(&aerdriver);
6c2b374d ZY	431	}
	432
	433	module_init(aer_service_init);
	434	module_exit(aer_service_exit);