xref: /openbmc/linux/drivers/pci/pcie/aer.c (revision 42641042)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Implement the AER root port service driver. The driver registers an IRQ
4  * handler. When a root port triggers an AER interrupt, the IRQ handler
5  * collects root port status and schedules work.
6  *
7  * Copyright (C) 2006 Intel Corp.
8  *	Tom Long Nguyen (tom.l.nguyen@intel.com)
9  *	Zhang Yanmin (yanmin.zhang@intel.com)
10  *
11  * (C) Copyright 2009 Hewlett-Packard Development Company, L.P.
12  *    Andrew Patterson <andrew.patterson@hp.com>
13  */
14 
15 #define pr_fmt(fmt) "AER: " fmt
16 #define dev_fmt pr_fmt
17 
18 #include <linux/bitops.h>
19 #include <linux/cper.h>
20 #include <linux/pci.h>
21 #include <linux/pci-acpi.h>
22 #include <linux/sched.h>
23 #include <linux/kernel.h>
24 #include <linux/errno.h>
25 #include <linux/pm.h>
26 #include <linux/init.h>
27 #include <linux/interrupt.h>
28 #include <linux/delay.h>
29 #include <linux/kfifo.h>
30 #include <linux/slab.h>
31 #include <acpi/apei.h>
32 #include <ras/ras_event.h>
33 
34 #include "../pci.h"
35 #include "portdrv.h"
36 
37 #define AER_ERROR_SOURCES_MAX		128
38 
39 #define AER_MAX_TYPEOF_COR_ERRS		16	/* as per PCI_ERR_COR_STATUS */
40 #define AER_MAX_TYPEOF_UNCOR_ERRS	27	/* as per PCI_ERR_UNCOR_STATUS*/
41 
42 struct aer_err_source {
43 	unsigned int status;
44 	unsigned int id;
45 };
46 
47 struct aer_rpc {
48 	struct pci_dev *rpd;		/* Root Port device */
49 	DECLARE_KFIFO(aer_fifo, struct aer_err_source, AER_ERROR_SOURCES_MAX);
50 };
51 
52 /* AER stats for the device */
53 struct aer_stats {
54 
55 	/*
56 	 * Fields for all AER capable devices. They indicate the errors
57 	 * "as seen by this device". Note that this may mean that if an
58 	 * end point is causing problems, the AER counters may increment
59 	 * at its link partner (e.g. root port) because the errors will be
60 	 * "seen" by the link partner and not the the problematic end point
61 	 * itself (which may report all counters as 0 as it never saw any
62 	 * problems).
63 	 */
64 	/* Counters for different type of correctable errors */
65 	u64 dev_cor_errs[AER_MAX_TYPEOF_COR_ERRS];
66 	/* Counters for different type of fatal uncorrectable errors */
67 	u64 dev_fatal_errs[AER_MAX_TYPEOF_UNCOR_ERRS];
68 	/* Counters for different type of nonfatal uncorrectable errors */
69 	u64 dev_nonfatal_errs[AER_MAX_TYPEOF_UNCOR_ERRS];
70 	/* Total number of ERR_COR sent by this device */
71 	u64 dev_total_cor_errs;
72 	/* Total number of ERR_FATAL sent by this device */
73 	u64 dev_total_fatal_errs;
74 	/* Total number of ERR_NONFATAL sent by this device */
75 	u64 dev_total_nonfatal_errs;
76 
77 	/*
78 	 * Fields for Root ports & root complex event collectors only, these
79 	 * indicate the total number of ERR_COR, ERR_FATAL, and ERR_NONFATAL
80 	 * messages received by the root port / event collector, INCLUDING the
81 	 * ones that are generated internally (by the rootport itself)
82 	 */
83 	u64 rootport_total_cor_errs;
84 	u64 rootport_total_fatal_errs;
85 	u64 rootport_total_nonfatal_errs;
86 };
87 
88 #define AER_LOG_TLP_MASKS		(PCI_ERR_UNC_POISON_TLP|	\
89 					PCI_ERR_UNC_ECRC|		\
90 					PCI_ERR_UNC_UNSUP|		\
91 					PCI_ERR_UNC_COMP_ABORT|		\
92 					PCI_ERR_UNC_UNX_COMP|		\
93 					PCI_ERR_UNC_MALF_TLP)
94 
95 #define SYSTEM_ERROR_INTR_ON_MESG_MASK	(PCI_EXP_RTCTL_SECEE|	\
96 					PCI_EXP_RTCTL_SENFEE|	\
97 					PCI_EXP_RTCTL_SEFEE)
98 #define ROOT_PORT_INTR_ON_MESG_MASK	(PCI_ERR_ROOT_CMD_COR_EN|	\
99 					PCI_ERR_ROOT_CMD_NONFATAL_EN|	\
100 					PCI_ERR_ROOT_CMD_FATAL_EN)
101 #define ERR_COR_ID(d)			(d & 0xffff)
102 #define ERR_UNCOR_ID(d)			(d >> 16)
103 
104 static int pcie_aer_disable;
105 static pci_ers_result_t aer_root_reset(struct pci_dev *dev);
106 
107 void pci_no_aer(void)
108 {
109 	pcie_aer_disable = 1;
110 }
111 
112 bool pci_aer_available(void)
113 {
114 	return !pcie_aer_disable && pci_msi_enabled();
115 }
116 
117 #ifdef CONFIG_PCIE_ECRC
118 
119 #define ECRC_POLICY_DEFAULT 0		/* ECRC set by BIOS */
120 #define ECRC_POLICY_OFF     1		/* ECRC off for performance */
121 #define ECRC_POLICY_ON      2		/* ECRC on for data integrity */
122 
123 static int ecrc_policy = ECRC_POLICY_DEFAULT;
124 
125 static const char * const ecrc_policy_str[] = {
126 	[ECRC_POLICY_DEFAULT] = "bios",
127 	[ECRC_POLICY_OFF] = "off",
128 	[ECRC_POLICY_ON] = "on"
129 };
130 
131 /**
132  * enable_ecrc_checking - enable PCIe ECRC checking for a device
133  * @dev: the PCI device
134  *
135  * Returns 0 on success, or negative on failure.
136  */
137 static int enable_ecrc_checking(struct pci_dev *dev)
138 {
139 	int aer = dev->aer_cap;
140 	u32 reg32;
141 
142 	if (!aer)
143 		return -ENODEV;
144 
145 	pci_read_config_dword(dev, aer + PCI_ERR_CAP, &reg32);
146 	if (reg32 & PCI_ERR_CAP_ECRC_GENC)
147 		reg32 |= PCI_ERR_CAP_ECRC_GENE;
148 	if (reg32 & PCI_ERR_CAP_ECRC_CHKC)
149 		reg32 |= PCI_ERR_CAP_ECRC_CHKE;
150 	pci_write_config_dword(dev, aer + PCI_ERR_CAP, reg32);
151 
152 	return 0;
153 }
154 
155 /**
156  * disable_ecrc_checking - disables PCIe ECRC checking for a device
157  * @dev: the PCI device
158  *
159  * Returns 0 on success, or negative on failure.
160  */
161 static int disable_ecrc_checking(struct pci_dev *dev)
162 {
163 	int aer = dev->aer_cap;
164 	u32 reg32;
165 
166 	if (!aer)
167 		return -ENODEV;
168 
169 	pci_read_config_dword(dev, aer + PCI_ERR_CAP, &reg32);
170 	reg32 &= ~(PCI_ERR_CAP_ECRC_GENE | PCI_ERR_CAP_ECRC_CHKE);
171 	pci_write_config_dword(dev, aer + PCI_ERR_CAP, reg32);
172 
173 	return 0;
174 }
175 
176 /**
177  * pcie_set_ecrc_checking - set/unset PCIe ECRC checking for a device based on global policy
178  * @dev: the PCI device
179  */
180 void pcie_set_ecrc_checking(struct pci_dev *dev)
181 {
182 	switch (ecrc_policy) {
183 	case ECRC_POLICY_DEFAULT:
184 		return;
185 	case ECRC_POLICY_OFF:
186 		disable_ecrc_checking(dev);
187 		break;
188 	case ECRC_POLICY_ON:
189 		enable_ecrc_checking(dev);
190 		break;
191 	default:
192 		return;
193 	}
194 }
195 
196 /**
197  * pcie_ecrc_get_policy - parse kernel command-line ecrc option
198  * @str: ECRC policy from kernel command line to use
199  */
200 void pcie_ecrc_get_policy(char *str)
201 {
202 	int i;
203 
204 	i = match_string(ecrc_policy_str, ARRAY_SIZE(ecrc_policy_str), str);
205 	if (i < 0)
206 		return;
207 
208 	ecrc_policy = i;
209 }
210 #endif	/* CONFIG_PCIE_ECRC */
211 
212 #define	PCI_EXP_AER_FLAGS	(PCI_EXP_DEVCTL_CERE | PCI_EXP_DEVCTL_NFERE | \
213 				 PCI_EXP_DEVCTL_FERE | PCI_EXP_DEVCTL_URRE)
214 
215 int pcie_aer_is_native(struct pci_dev *dev)
216 {
217 	struct pci_host_bridge *host = pci_find_host_bridge(dev->bus);
218 
219 	if (!dev->aer_cap)
220 		return 0;
221 
222 	return pcie_ports_native || host->native_aer;
223 }
224 
225 int pci_enable_pcie_error_reporting(struct pci_dev *dev)
226 {
227 	int rc;
228 
229 	if (!pcie_aer_is_native(dev))
230 		return -EIO;
231 
232 	rc = pcie_capability_set_word(dev, PCI_EXP_DEVCTL, PCI_EXP_AER_FLAGS);
233 	return pcibios_err_to_errno(rc);
234 }
235 EXPORT_SYMBOL_GPL(pci_enable_pcie_error_reporting);
236 
237 int pci_disable_pcie_error_reporting(struct pci_dev *dev)
238 {
239 	int rc;
240 
241 	if (!pcie_aer_is_native(dev))
242 		return -EIO;
243 
244 	rc = pcie_capability_clear_word(dev, PCI_EXP_DEVCTL, PCI_EXP_AER_FLAGS);
245 	return pcibios_err_to_errno(rc);
246 }
247 EXPORT_SYMBOL_GPL(pci_disable_pcie_error_reporting);
248 
249 int pci_aer_clear_nonfatal_status(struct pci_dev *dev)
250 {
251 	int aer = dev->aer_cap;
252 	u32 status, sev;
253 
254 	if (!pcie_aer_is_native(dev))
255 		return -EIO;
256 
257 	/* Clear status bits for ERR_NONFATAL errors only */
258 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, &status);
259 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_SEVER, &sev);
260 	status &= ~sev;
261 	if (status)
262 		pci_write_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, status);
263 
264 	return 0;
265 }
266 EXPORT_SYMBOL_GPL(pci_aer_clear_nonfatal_status);
267 
268 void pci_aer_clear_fatal_status(struct pci_dev *dev)
269 {
270 	int aer = dev->aer_cap;
271 	u32 status, sev;
272 
273 	if (!pcie_aer_is_native(dev))
274 		return;
275 
276 	/* Clear status bits for ERR_FATAL errors only */
277 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, &status);
278 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_SEVER, &sev);
279 	status &= sev;
280 	if (status)
281 		pci_write_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, status);
282 }
283 
284 /**
285  * pci_aer_raw_clear_status - Clear AER error registers.
286  * @dev: the PCI device
287  *
288  * Clearing AER error status registers unconditionally, regardless of
289  * whether they're owned by firmware or the OS.
290  *
291  * Returns 0 on success, or negative on failure.
292  */
293 int pci_aer_raw_clear_status(struct pci_dev *dev)
294 {
295 	int aer = dev->aer_cap;
296 	u32 status;
297 	int port_type;
298 
299 	if (!aer)
300 		return -EIO;
301 
302 	port_type = pci_pcie_type(dev);
303 	if (port_type == PCI_EXP_TYPE_ROOT_PORT ||
304 	    port_type == PCI_EXP_TYPE_RC_EC) {
305 		pci_read_config_dword(dev, aer + PCI_ERR_ROOT_STATUS, &status);
306 		pci_write_config_dword(dev, aer + PCI_ERR_ROOT_STATUS, status);
307 	}
308 
309 	pci_read_config_dword(dev, aer + PCI_ERR_COR_STATUS, &status);
310 	pci_write_config_dword(dev, aer + PCI_ERR_COR_STATUS, status);
311 
312 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, &status);
313 	pci_write_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, status);
314 
315 	return 0;
316 }
317 
318 int pci_aer_clear_status(struct pci_dev *dev)
319 {
320 	if (!pcie_aer_is_native(dev))
321 		return -EIO;
322 
323 	return pci_aer_raw_clear_status(dev);
324 }
325 
326 void pci_save_aer_state(struct pci_dev *dev)
327 {
328 	int aer = dev->aer_cap;
329 	struct pci_cap_saved_state *save_state;
330 	u32 *cap;
331 
332 	if (!aer)
333 		return;
334 
335 	save_state = pci_find_saved_ext_cap(dev, PCI_EXT_CAP_ID_ERR);
336 	if (!save_state)
337 		return;
338 
339 	cap = &save_state->cap.data[0];
340 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_MASK, cap++);
341 	pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_SEVER, cap++);
342 	pci_read_config_dword(dev, aer + PCI_ERR_COR_MASK, cap++);
343 	pci_read_config_dword(dev, aer + PCI_ERR_CAP, cap++);
344 	if (pcie_cap_has_rtctl(dev))
345 		pci_read_config_dword(dev, aer + PCI_ERR_ROOT_COMMAND, cap++);
346 }
347 
348 void pci_restore_aer_state(struct pci_dev *dev)
349 {
350 	int aer = dev->aer_cap;
351 	struct pci_cap_saved_state *save_state;
352 	u32 *cap;
353 
354 	if (!aer)
355 		return;
356 
357 	save_state = pci_find_saved_ext_cap(dev, PCI_EXT_CAP_ID_ERR);
358 	if (!save_state)
359 		return;
360 
361 	cap = &save_state->cap.data[0];
362 	pci_write_config_dword(dev, aer + PCI_ERR_UNCOR_MASK, *cap++);
363 	pci_write_config_dword(dev, aer + PCI_ERR_UNCOR_SEVER, *cap++);
364 	pci_write_config_dword(dev, aer + PCI_ERR_COR_MASK, *cap++);
365 	pci_write_config_dword(dev, aer + PCI_ERR_CAP, *cap++);
366 	if (pcie_cap_has_rtctl(dev))
367 		pci_write_config_dword(dev, aer + PCI_ERR_ROOT_COMMAND, *cap++);
368 }
369 
370 void pci_aer_init(struct pci_dev *dev)
371 {
372 	int n;
373 
374 	dev->aer_cap = pci_find_ext_capability(dev, PCI_EXT_CAP_ID_ERR);
375 	if (!dev->aer_cap)
376 		return;
377 
378 	dev->aer_stats = kzalloc(sizeof(struct aer_stats), GFP_KERNEL);
379 
380 	/*
381 	 * We save/restore PCI_ERR_UNCOR_MASK, PCI_ERR_UNCOR_SEVER,
382 	 * PCI_ERR_COR_MASK, and PCI_ERR_CAP.  Root and Root Complex Event
383 	 * Collectors also implement PCI_ERR_ROOT_COMMAND (PCIe r5.0, sec
384 	 * 7.8.4).
385 	 */
386 	n = pcie_cap_has_rtctl(dev) ? 5 : 4;
387 	pci_add_ext_cap_save_buffer(dev, PCI_EXT_CAP_ID_ERR, sizeof(u32) * n);
388 
389 	pci_aer_clear_status(dev);
390 }
391 
392 void pci_aer_exit(struct pci_dev *dev)
393 {
394 	kfree(dev->aer_stats);
395 	dev->aer_stats = NULL;
396 }
397 
398 #define AER_AGENT_RECEIVER		0
399 #define AER_AGENT_REQUESTER		1
400 #define AER_AGENT_COMPLETER		2
401 #define AER_AGENT_TRANSMITTER		3
402 
403 #define AER_AGENT_REQUESTER_MASK(t)	((t == AER_CORRECTABLE) ?	\
404 	0 : (PCI_ERR_UNC_COMP_TIME|PCI_ERR_UNC_UNSUP))
405 #define AER_AGENT_COMPLETER_MASK(t)	((t == AER_CORRECTABLE) ?	\
406 	0 : PCI_ERR_UNC_COMP_ABORT)
407 #define AER_AGENT_TRANSMITTER_MASK(t)	((t == AER_CORRECTABLE) ?	\
408 	(PCI_ERR_COR_REP_ROLL|PCI_ERR_COR_REP_TIMER) : 0)
409 
410 #define AER_GET_AGENT(t, e)						\
411 	((e & AER_AGENT_COMPLETER_MASK(t)) ? AER_AGENT_COMPLETER :	\
412 	(e & AER_AGENT_REQUESTER_MASK(t)) ? AER_AGENT_REQUESTER :	\
413 	(e & AER_AGENT_TRANSMITTER_MASK(t)) ? AER_AGENT_TRANSMITTER :	\
414 	AER_AGENT_RECEIVER)
415 
416 #define AER_PHYSICAL_LAYER_ERROR	0
417 #define AER_DATA_LINK_LAYER_ERROR	1
418 #define AER_TRANSACTION_LAYER_ERROR	2
419 
420 #define AER_PHYSICAL_LAYER_ERROR_MASK(t) ((t == AER_CORRECTABLE) ?	\
421 	PCI_ERR_COR_RCVR : 0)
422 #define AER_DATA_LINK_LAYER_ERROR_MASK(t) ((t == AER_CORRECTABLE) ?	\
423 	(PCI_ERR_COR_BAD_TLP|						\
424 	PCI_ERR_COR_BAD_DLLP|						\
425 	PCI_ERR_COR_REP_ROLL|						\
426 	PCI_ERR_COR_REP_TIMER) : PCI_ERR_UNC_DLP)
427 
428 #define AER_GET_LAYER_ERROR(t, e)					\
429 	((e & AER_PHYSICAL_LAYER_ERROR_MASK(t)) ? AER_PHYSICAL_LAYER_ERROR : \
430 	(e & AER_DATA_LINK_LAYER_ERROR_MASK(t)) ? AER_DATA_LINK_LAYER_ERROR : \
431 	AER_TRANSACTION_LAYER_ERROR)
432 
433 /*
434  * AER error strings
435  */
436 static const char *aer_error_severity_string[] = {
437 	"Uncorrected (Non-Fatal)",
438 	"Uncorrected (Fatal)",
439 	"Corrected"
440 };
441 
442 static const char *aer_error_layer[] = {
443 	"Physical Layer",
444 	"Data Link Layer",
445 	"Transaction Layer"
446 };
447 
448 static const char *aer_correctable_error_string[] = {
449 	"RxErr",			/* Bit Position 0	*/
450 	NULL,
451 	NULL,
452 	NULL,
453 	NULL,
454 	NULL,
455 	"BadTLP",			/* Bit Position 6	*/
456 	"BadDLLP",			/* Bit Position 7	*/
457 	"Rollover",			/* Bit Position 8	*/
458 	NULL,
459 	NULL,
460 	NULL,
461 	"Timeout",			/* Bit Position 12	*/
462 	"NonFatalErr",			/* Bit Position 13	*/
463 	"CorrIntErr",			/* Bit Position 14	*/
464 	"HeaderOF",			/* Bit Position 15	*/
465 	NULL,				/* Bit Position 16	*/
466 	NULL,				/* Bit Position 17	*/
467 	NULL,				/* Bit Position 18	*/
468 	NULL,				/* Bit Position 19	*/
469 	NULL,				/* Bit Position 20	*/
470 	NULL,				/* Bit Position 21	*/
471 	NULL,				/* Bit Position 22	*/
472 	NULL,				/* Bit Position 23	*/
473 	NULL,				/* Bit Position 24	*/
474 	NULL,				/* Bit Position 25	*/
475 	NULL,				/* Bit Position 26	*/
476 	NULL,				/* Bit Position 27	*/
477 	NULL,				/* Bit Position 28	*/
478 	NULL,				/* Bit Position 29	*/
479 	NULL,				/* Bit Position 30	*/
480 	NULL,				/* Bit Position 31	*/
481 };
482 
483 static const char *aer_uncorrectable_error_string[] = {
484 	"Undefined",			/* Bit Position 0	*/
485 	NULL,
486 	NULL,
487 	NULL,
488 	"DLP",				/* Bit Position 4	*/
489 	"SDES",				/* Bit Position 5	*/
490 	NULL,
491 	NULL,
492 	NULL,
493 	NULL,
494 	NULL,
495 	NULL,
496 	"TLP",				/* Bit Position 12	*/
497 	"FCP",				/* Bit Position 13	*/
498 	"CmpltTO",			/* Bit Position 14	*/
499 	"CmpltAbrt",			/* Bit Position 15	*/
500 	"UnxCmplt",			/* Bit Position 16	*/
501 	"RxOF",				/* Bit Position 17	*/
502 	"MalfTLP",			/* Bit Position 18	*/
503 	"ECRC",				/* Bit Position 19	*/
504 	"UnsupReq",			/* Bit Position 20	*/
505 	"ACSViol",			/* Bit Position 21	*/
506 	"UncorrIntErr",			/* Bit Position 22	*/
507 	"BlockedTLP",			/* Bit Position 23	*/
508 	"AtomicOpBlocked",		/* Bit Position 24	*/
509 	"TLPBlockedErr",		/* Bit Position 25	*/
510 	"PoisonTLPBlocked",		/* Bit Position 26	*/
511 	NULL,				/* Bit Position 27	*/
512 	NULL,				/* Bit Position 28	*/
513 	NULL,				/* Bit Position 29	*/
514 	NULL,				/* Bit Position 30	*/
515 	NULL,				/* Bit Position 31	*/
516 };
517 
518 static const char *aer_agent_string[] = {
519 	"Receiver ID",
520 	"Requester ID",
521 	"Completer ID",
522 	"Transmitter ID"
523 };
524 
525 #define aer_stats_dev_attr(name, stats_array, strings_array,		\
526 			   total_string, total_field)			\
527 	static ssize_t							\
528 	name##_show(struct device *dev, struct device_attribute *attr,	\
529 		     char *buf)						\
530 {									\
531 	unsigned int i;							\
532 	struct pci_dev *pdev = to_pci_dev(dev);				\
533 	u64 *stats = pdev->aer_stats->stats_array;			\
534 	size_t len = 0;							\
535 									\
536 	for (i = 0; i < ARRAY_SIZE(strings_array); i++) {		\
537 		if (strings_array[i])					\
538 			len += sysfs_emit_at(buf, len, "%s %llu\n",	\
539 					     strings_array[i],		\
540 					     stats[i]);			\
541 		else if (stats[i])					\
542 			len += sysfs_emit_at(buf, len,			\
543 					     #stats_array "_bit[%d] %llu\n",\
544 					     i, stats[i]);		\
545 	}								\
546 	len += sysfs_emit_at(buf, len, "TOTAL_%s %llu\n", total_string,	\
547 			     pdev->aer_stats->total_field);		\
548 	return len;							\
549 }									\
550 static DEVICE_ATTR_RO(name)
551 
552 aer_stats_dev_attr(aer_dev_correctable, dev_cor_errs,
553 		   aer_correctable_error_string, "ERR_COR",
554 		   dev_total_cor_errs);
555 aer_stats_dev_attr(aer_dev_fatal, dev_fatal_errs,
556 		   aer_uncorrectable_error_string, "ERR_FATAL",
557 		   dev_total_fatal_errs);
558 aer_stats_dev_attr(aer_dev_nonfatal, dev_nonfatal_errs,
559 		   aer_uncorrectable_error_string, "ERR_NONFATAL",
560 		   dev_total_nonfatal_errs);
561 
562 #define aer_stats_rootport_attr(name, field)				\
563 	static ssize_t							\
564 	name##_show(struct device *dev, struct device_attribute *attr,	\
565 		     char *buf)						\
566 {									\
567 	struct pci_dev *pdev = to_pci_dev(dev);				\
568 	return sysfs_emit(buf, "%llu\n", pdev->aer_stats->field);	\
569 }									\
570 static DEVICE_ATTR_RO(name)
571 
572 aer_stats_rootport_attr(aer_rootport_total_err_cor,
573 			 rootport_total_cor_errs);
574 aer_stats_rootport_attr(aer_rootport_total_err_fatal,
575 			 rootport_total_fatal_errs);
576 aer_stats_rootport_attr(aer_rootport_total_err_nonfatal,
577 			 rootport_total_nonfatal_errs);
578 
579 static struct attribute *aer_stats_attrs[] __ro_after_init = {
580 	&dev_attr_aer_dev_correctable.attr,
581 	&dev_attr_aer_dev_fatal.attr,
582 	&dev_attr_aer_dev_nonfatal.attr,
583 	&dev_attr_aer_rootport_total_err_cor.attr,
584 	&dev_attr_aer_rootport_total_err_fatal.attr,
585 	&dev_attr_aer_rootport_total_err_nonfatal.attr,
586 	NULL
587 };
588 
589 static umode_t aer_stats_attrs_are_visible(struct kobject *kobj,
590 					   struct attribute *a, int n)
591 {
592 	struct device *dev = kobj_to_dev(kobj);
593 	struct pci_dev *pdev = to_pci_dev(dev);
594 
595 	if (!pdev->aer_stats)
596 		return 0;
597 
598 	if ((a == &dev_attr_aer_rootport_total_err_cor.attr ||
599 	     a == &dev_attr_aer_rootport_total_err_fatal.attr ||
600 	     a == &dev_attr_aer_rootport_total_err_nonfatal.attr) &&
601 	    ((pci_pcie_type(pdev) != PCI_EXP_TYPE_ROOT_PORT) &&
602 	     (pci_pcie_type(pdev) != PCI_EXP_TYPE_RC_EC)))
603 		return 0;
604 
605 	return a->mode;
606 }
607 
608 const struct attribute_group aer_stats_attr_group = {
609 	.attrs  = aer_stats_attrs,
610 	.is_visible = aer_stats_attrs_are_visible,
611 };
612 
613 static void pci_dev_aer_stats_incr(struct pci_dev *pdev,
614 				   struct aer_err_info *info)
615 {
616 	unsigned long status = info->status & ~info->mask;
617 	int i, max = -1;
618 	u64 *counter = NULL;
619 	struct aer_stats *aer_stats = pdev->aer_stats;
620 
621 	if (!aer_stats)
622 		return;
623 
624 	switch (info->severity) {
625 	case AER_CORRECTABLE:
626 		aer_stats->dev_total_cor_errs++;
627 		counter = &aer_stats->dev_cor_errs[0];
628 		max = AER_MAX_TYPEOF_COR_ERRS;
629 		break;
630 	case AER_NONFATAL:
631 		aer_stats->dev_total_nonfatal_errs++;
632 		counter = &aer_stats->dev_nonfatal_errs[0];
633 		max = AER_MAX_TYPEOF_UNCOR_ERRS;
634 		break;
635 	case AER_FATAL:
636 		aer_stats->dev_total_fatal_errs++;
637 		counter = &aer_stats->dev_fatal_errs[0];
638 		max = AER_MAX_TYPEOF_UNCOR_ERRS;
639 		break;
640 	}
641 
642 	for_each_set_bit(i, &status, max)
643 		counter[i]++;
644 }
645 
646 static void pci_rootport_aer_stats_incr(struct pci_dev *pdev,
647 				 struct aer_err_source *e_src)
648 {
649 	struct aer_stats *aer_stats = pdev->aer_stats;
650 
651 	if (!aer_stats)
652 		return;
653 
654 	if (e_src->status & PCI_ERR_ROOT_COR_RCV)
655 		aer_stats->rootport_total_cor_errs++;
656 
657 	if (e_src->status & PCI_ERR_ROOT_UNCOR_RCV) {
658 		if (e_src->status & PCI_ERR_ROOT_FATAL_RCV)
659 			aer_stats->rootport_total_fatal_errs++;
660 		else
661 			aer_stats->rootport_total_nonfatal_errs++;
662 	}
663 }
664 
665 static void __print_tlp_header(struct pci_dev *dev,
666 			       struct aer_header_log_regs *t)
667 {
668 	pci_err(dev, "  TLP Header: %08x %08x %08x %08x\n",
669 		t->dw0, t->dw1, t->dw2, t->dw3);
670 }
671 
672 static void __aer_print_error(struct pci_dev *dev,
673 			      struct aer_err_info *info)
674 {
675 	const char **strings;
676 	unsigned long status = info->status & ~info->mask;
677 	const char *level, *errmsg;
678 	int i;
679 
680 	if (info->severity == AER_CORRECTABLE) {
681 		strings = aer_correctable_error_string;
682 		level = KERN_WARNING;
683 	} else {
684 		strings = aer_uncorrectable_error_string;
685 		level = KERN_ERR;
686 	}
687 
688 	for_each_set_bit(i, &status, 32) {
689 		errmsg = strings[i];
690 		if (!errmsg)
691 			errmsg = "Unknown Error Bit";
692 
693 		pci_printk(level, dev, "   [%2d] %-22s%s\n", i, errmsg,
694 				info->first_error == i ? " (First)" : "");
695 	}
696 	pci_dev_aer_stats_incr(dev, info);
697 }
698 
699 void aer_print_error(struct pci_dev *dev, struct aer_err_info *info)
700 {
701 	int layer, agent;
702 	int id = ((dev->bus->number << 8) | dev->devfn);
703 	const char *level;
704 
705 	if (!info->status) {
706 		pci_err(dev, "PCIe Bus Error: severity=%s, type=Inaccessible, (Unregistered Agent ID)\n",
707 			aer_error_severity_string[info->severity]);
708 		goto out;
709 	}
710 
711 	layer = AER_GET_LAYER_ERROR(info->severity, info->status);
712 	agent = AER_GET_AGENT(info->severity, info->status);
713 
714 	level = (info->severity == AER_CORRECTABLE) ? KERN_WARNING : KERN_ERR;
715 
716 	pci_printk(level, dev, "PCIe Bus Error: severity=%s, type=%s, (%s)\n",
717 		   aer_error_severity_string[info->severity],
718 		   aer_error_layer[layer], aer_agent_string[agent]);
719 
720 	pci_printk(level, dev, "  device [%04x:%04x] error status/mask=%08x/%08x\n",
721 		   dev->vendor, dev->device, info->status, info->mask);
722 
723 	__aer_print_error(dev, info);
724 
725 	if (info->tlp_header_valid)
726 		__print_tlp_header(dev, &info->tlp);
727 
728 out:
729 	if (info->id && info->error_dev_num > 1 && info->id == id)
730 		pci_err(dev, "  Error of this Agent is reported first\n");
731 
732 	trace_aer_event(dev_name(&dev->dev), (info->status & ~info->mask),
733 			info->severity, info->tlp_header_valid, &info->tlp);
734 }
735 
736 static void aer_print_port_info(struct pci_dev *dev, struct aer_err_info *info)
737 {
738 	u8 bus = info->id >> 8;
739 	u8 devfn = info->id & 0xff;
740 
741 	pci_info(dev, "%s%s error received: %04x:%02x:%02x.%d\n",
742 		 info->multi_error_valid ? "Multiple " : "",
743 		 aer_error_severity_string[info->severity],
744 		 pci_domain_nr(dev->bus), bus, PCI_SLOT(devfn),
745 		 PCI_FUNC(devfn));
746 }
747 
748 #ifdef CONFIG_ACPI_APEI_PCIEAER
749 int cper_severity_to_aer(int cper_severity)
750 {
751 	switch (cper_severity) {
752 	case CPER_SEV_RECOVERABLE:
753 		return AER_NONFATAL;
754 	case CPER_SEV_FATAL:
755 		return AER_FATAL;
756 	default:
757 		return AER_CORRECTABLE;
758 	}
759 }
760 EXPORT_SYMBOL_GPL(cper_severity_to_aer);
761 
762 void cper_print_aer(struct pci_dev *dev, int aer_severity,
763 		    struct aer_capability_regs *aer)
764 {
765 	int layer, agent, tlp_header_valid = 0;
766 	u32 status, mask;
767 	struct aer_err_info info;
768 
769 	if (aer_severity == AER_CORRECTABLE) {
770 		status = aer->cor_status;
771 		mask = aer->cor_mask;
772 	} else {
773 		status = aer->uncor_status;
774 		mask = aer->uncor_mask;
775 		tlp_header_valid = status & AER_LOG_TLP_MASKS;
776 	}
777 
778 	layer = AER_GET_LAYER_ERROR(aer_severity, status);
779 	agent = AER_GET_AGENT(aer_severity, status);
780 
781 	memset(&info, 0, sizeof(info));
782 	info.severity = aer_severity;
783 	info.status = status;
784 	info.mask = mask;
785 	info.first_error = PCI_ERR_CAP_FEP(aer->cap_control);
786 
787 	pci_err(dev, "aer_status: 0x%08x, aer_mask: 0x%08x\n", status, mask);
788 	__aer_print_error(dev, &info);
789 	pci_err(dev, "aer_layer=%s, aer_agent=%s\n",
790 		aer_error_layer[layer], aer_agent_string[agent]);
791 
792 	if (aer_severity != AER_CORRECTABLE)
793 		pci_err(dev, "aer_uncor_severity: 0x%08x\n",
794 			aer->uncor_severity);
795 
796 	if (tlp_header_valid)
797 		__print_tlp_header(dev, &aer->header_log);
798 
799 	trace_aer_event(dev_name(&dev->dev), (status & ~mask),
800 			aer_severity, tlp_header_valid, &aer->header_log);
801 }
802 #endif
803 
804 /**
805  * add_error_device - list device to be handled
806  * @e_info: pointer to error info
807  * @dev: pointer to pci_dev to be added
808  */
809 static int add_error_device(struct aer_err_info *e_info, struct pci_dev *dev)
810 {
811 	if (e_info->error_dev_num < AER_MAX_MULTI_ERR_DEVICES) {
812 		e_info->dev[e_info->error_dev_num] = pci_dev_get(dev);
813 		e_info->error_dev_num++;
814 		return 0;
815 	}
816 	return -ENOSPC;
817 }
818 
819 /**
820  * is_error_source - check whether the device is source of reported error
821  * @dev: pointer to pci_dev to be checked
822  * @e_info: pointer to reported error info
823  */
824 static bool is_error_source(struct pci_dev *dev, struct aer_err_info *e_info)
825 {
826 	int aer = dev->aer_cap;
827 	u32 status, mask;
828 	u16 reg16;
829 
830 	/*
831 	 * When bus id is equal to 0, it might be a bad id
832 	 * reported by root port.
833 	 */
834 	if ((PCI_BUS_NUM(e_info->id) != 0) &&
835 	    !(dev->bus->bus_flags & PCI_BUS_FLAGS_NO_AERSID)) {
836 		/* Device ID match? */
837 		if (e_info->id == ((dev->bus->number << 8) | dev->devfn))
838 			return true;
839 
840 		/* Continue id comparing if there is no multiple error */
841 		if (!e_info->multi_error_valid)
842 			return false;
843 	}
844 
845 	/*
846 	 * When either
847 	 *      1) bus id is equal to 0. Some ports might lose the bus
848 	 *              id of error source id;
849 	 *      2) bus flag PCI_BUS_FLAGS_NO_AERSID is set
850 	 *      3) There are multiple errors and prior ID comparing fails;
851 	 * We check AER status registers to find possible reporter.
852 	 */
853 	if (atomic_read(&dev->enable_cnt) == 0)
854 		return false;
855 
856 	/* Check if AER is enabled */
857 	pcie_capability_read_word(dev, PCI_EXP_DEVCTL, &reg16);
858 	if (!(reg16 & PCI_EXP_AER_FLAGS))
859 		return false;
860 
861 	if (!aer)
862 		return false;
863 
864 	/* Check if error is recorded */
865 	if (e_info->severity == AER_CORRECTABLE) {
866 		pci_read_config_dword(dev, aer + PCI_ERR_COR_STATUS, &status);
867 		pci_read_config_dword(dev, aer + PCI_ERR_COR_MASK, &mask);
868 	} else {
869 		pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS, &status);
870 		pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_MASK, &mask);
871 	}
872 	if (status & ~mask)
873 		return true;
874 
875 	return false;
876 }
877 
878 static int find_device_iter(struct pci_dev *dev, void *data)
879 {
880 	struct aer_err_info *e_info = (struct aer_err_info *)data;
881 
882 	if (is_error_source(dev, e_info)) {
883 		/* List this device */
884 		if (add_error_device(e_info, dev)) {
885 			/* We cannot handle more... Stop iteration */
886 			/* TODO: Should print error message here? */
887 			return 1;
888 		}
889 
890 		/* If there is only a single error, stop iteration */
891 		if (!e_info->multi_error_valid)
892 			return 1;
893 	}
894 	return 0;
895 }
896 
897 /**
898  * find_source_device - search through device hierarchy for source device
899  * @parent: pointer to Root Port pci_dev data structure
900  * @e_info: including detailed error information such like id
901  *
902  * Return true if found.
903  *
904  * Invoked by DPC when error is detected at the Root Port.
905  * Caller of this function must set id, severity, and multi_error_valid of
906  * struct aer_err_info pointed by @e_info properly.  This function must fill
907  * e_info->error_dev_num and e_info->dev[], based on the given information.
908  */
909 static bool find_source_device(struct pci_dev *parent,
910 		struct aer_err_info *e_info)
911 {
912 	struct pci_dev *dev = parent;
913 	int result;
914 
915 	/* Must reset in this function */
916 	e_info->error_dev_num = 0;
917 
918 	/* Is Root Port an agent that sends error message? */
919 	result = find_device_iter(dev, e_info);
920 	if (result)
921 		return true;
922 
923 	if (pci_pcie_type(parent) == PCI_EXP_TYPE_RC_EC)
924 		pcie_walk_rcec(parent, find_device_iter, e_info);
925 	else
926 		pci_walk_bus(parent->subordinate, find_device_iter, e_info);
927 
928 	if (!e_info->error_dev_num) {
929 		pci_info(parent, "can't find device of ID%04x\n", e_info->id);
930 		return false;
931 	}
932 	return true;
933 }
934 
935 /**
936  * handle_error_source - handle logging error into an event log
937  * @dev: pointer to pci_dev data structure of error source device
938  * @info: comprehensive error information
939  *
940  * Invoked when an error being detected by Root Port.
941  */
942 static void handle_error_source(struct pci_dev *dev, struct aer_err_info *info)
943 {
944 	int aer = dev->aer_cap;
945 
946 	if (info->severity == AER_CORRECTABLE) {
947 		/*
948 		 * Correctable error does not need software intervention.
949 		 * No need to go through error recovery process.
950 		 */
951 		if (aer)
952 			pci_write_config_dword(dev, aer + PCI_ERR_COR_STATUS,
953 					info->status);
954 		if (pcie_aer_is_native(dev))
955 			pcie_clear_device_status(dev);
956 	} else if (info->severity == AER_NONFATAL)
957 		pcie_do_recovery(dev, pci_channel_io_normal, aer_root_reset);
958 	else if (info->severity == AER_FATAL)
959 		pcie_do_recovery(dev, pci_channel_io_frozen, aer_root_reset);
960 	pci_dev_put(dev);
961 }
962 
963 #ifdef CONFIG_ACPI_APEI_PCIEAER
964 
965 #define AER_RECOVER_RING_ORDER		4
966 #define AER_RECOVER_RING_SIZE		(1 << AER_RECOVER_RING_ORDER)
967 
968 struct aer_recover_entry {
969 	u8	bus;
970 	u8	devfn;
971 	u16	domain;
972 	int	severity;
973 	struct aer_capability_regs *regs;
974 };
975 
976 static DEFINE_KFIFO(aer_recover_ring, struct aer_recover_entry,
977 		    AER_RECOVER_RING_SIZE);
978 
979 static void aer_recover_work_func(struct work_struct *work)
980 {
981 	struct aer_recover_entry entry;
982 	struct pci_dev *pdev;
983 
984 	while (kfifo_get(&aer_recover_ring, &entry)) {
985 		pdev = pci_get_domain_bus_and_slot(entry.domain, entry.bus,
986 						   entry.devfn);
987 		if (!pdev) {
988 			pr_err("no pci_dev for %04x:%02x:%02x.%x\n",
989 			       entry.domain, entry.bus,
990 			       PCI_SLOT(entry.devfn), PCI_FUNC(entry.devfn));
991 			continue;
992 		}
993 		cper_print_aer(pdev, entry.severity, entry.regs);
994 		if (entry.severity == AER_NONFATAL)
995 			pcie_do_recovery(pdev, pci_channel_io_normal,
996 					 aer_root_reset);
997 		else if (entry.severity == AER_FATAL)
998 			pcie_do_recovery(pdev, pci_channel_io_frozen,
999 					 aer_root_reset);
1000 		pci_dev_put(pdev);
1001 	}
1002 }
1003 
1004 /*
1005  * Mutual exclusion for writers of aer_recover_ring, reader side don't
1006  * need lock, because there is only one reader and lock is not needed
1007  * between reader and writer.
1008  */
1009 static DEFINE_SPINLOCK(aer_recover_ring_lock);
1010 static DECLARE_WORK(aer_recover_work, aer_recover_work_func);
1011 
1012 void aer_recover_queue(int domain, unsigned int bus, unsigned int devfn,
1013 		       int severity, struct aer_capability_regs *aer_regs)
1014 {
1015 	struct aer_recover_entry entry = {
1016 		.bus		= bus,
1017 		.devfn		= devfn,
1018 		.domain		= domain,
1019 		.severity	= severity,
1020 		.regs		= aer_regs,
1021 	};
1022 
1023 	if (kfifo_in_spinlocked(&aer_recover_ring, &entry, 1,
1024 				 &aer_recover_ring_lock))
1025 		schedule_work(&aer_recover_work);
1026 	else
1027 		pr_err("buffer overflow in recovery for %04x:%02x:%02x.%x\n",
1028 		       domain, bus, PCI_SLOT(devfn), PCI_FUNC(devfn));
1029 }
1030 EXPORT_SYMBOL_GPL(aer_recover_queue);
1031 #endif
1032 
1033 /**
1034  * aer_get_device_error_info - read error status from dev and store it to info
1035  * @dev: pointer to the device expected to have a error record
1036  * @info: pointer to structure to store the error record
1037  *
1038  * Return 1 on success, 0 on error.
1039  *
1040  * Note that @info is reused among all error devices. Clear fields properly.
1041  */
1042 int aer_get_device_error_info(struct pci_dev *dev, struct aer_err_info *info)
1043 {
1044 	int type = pci_pcie_type(dev);
1045 	int aer = dev->aer_cap;
1046 	int temp;
1047 
1048 	/* Must reset in this function */
1049 	info->status = 0;
1050 	info->tlp_header_valid = 0;
1051 
1052 	/* The device might not support AER */
1053 	if (!aer)
1054 		return 0;
1055 
1056 	if (info->severity == AER_CORRECTABLE) {
1057 		pci_read_config_dword(dev, aer + PCI_ERR_COR_STATUS,
1058 			&info->status);
1059 		pci_read_config_dword(dev, aer + PCI_ERR_COR_MASK,
1060 			&info->mask);
1061 		if (!(info->status & ~info->mask))
1062 			return 0;
1063 	} else if (type == PCI_EXP_TYPE_ROOT_PORT ||
1064 		   type == PCI_EXP_TYPE_RC_EC ||
1065 		   type == PCI_EXP_TYPE_DOWNSTREAM ||
1066 		   info->severity == AER_NONFATAL) {
1067 
1068 		/* Link is still healthy for IO reads */
1069 		pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_STATUS,
1070 			&info->status);
1071 		pci_read_config_dword(dev, aer + PCI_ERR_UNCOR_MASK,
1072 			&info->mask);
1073 		if (!(info->status & ~info->mask))
1074 			return 0;
1075 
1076 		/* Get First Error Pointer */
1077 		pci_read_config_dword(dev, aer + PCI_ERR_CAP, &temp);
1078 		info->first_error = PCI_ERR_CAP_FEP(temp);
1079 
1080 		if (info->status & AER_LOG_TLP_MASKS) {
1081 			info->tlp_header_valid = 1;
1082 			pci_read_config_dword(dev,
1083 				aer + PCI_ERR_HEADER_LOG, &info->tlp.dw0);
1084 			pci_read_config_dword(dev,
1085 				aer + PCI_ERR_HEADER_LOG + 4, &info->tlp.dw1);
1086 			pci_read_config_dword(dev,
1087 				aer + PCI_ERR_HEADER_LOG + 8, &info->tlp.dw2);
1088 			pci_read_config_dword(dev,
1089 				aer + PCI_ERR_HEADER_LOG + 12, &info->tlp.dw3);
1090 		}
1091 	}
1092 
1093 	return 1;
1094 }
1095 
1096 static inline void aer_process_err_devices(struct aer_err_info *e_info)
1097 {
1098 	int i;
1099 
1100 	/* Report all before handle them, not to lost records by reset etc. */
1101 	for (i = 0; i < e_info->error_dev_num && e_info->dev[i]; i++) {
1102 		if (aer_get_device_error_info(e_info->dev[i], e_info))
1103 			aer_print_error(e_info->dev[i], e_info);
1104 	}
1105 	for (i = 0; i < e_info->error_dev_num && e_info->dev[i]; i++) {
1106 		if (aer_get_device_error_info(e_info->dev[i], e_info))
1107 			handle_error_source(e_info->dev[i], e_info);
1108 	}
1109 }
1110 
1111 /**
1112  * aer_isr_one_error - consume an error detected by root port
1113  * @rpc: pointer to the root port which holds an error
1114  * @e_src: pointer to an error source
1115  */
1116 static void aer_isr_one_error(struct aer_rpc *rpc,
1117 		struct aer_err_source *e_src)
1118 {
1119 	struct pci_dev *pdev = rpc->rpd;
1120 	struct aer_err_info e_info;
1121 
1122 	pci_rootport_aer_stats_incr(pdev, e_src);
1123 
1124 	/*
1125 	 * There is a possibility that both correctable error and
1126 	 * uncorrectable error being logged. Report correctable error first.
1127 	 */
1128 	if (e_src->status & PCI_ERR_ROOT_COR_RCV) {
1129 		e_info.id = ERR_COR_ID(e_src->id);
1130 		e_info.severity = AER_CORRECTABLE;
1131 
1132 		if (e_src->status & PCI_ERR_ROOT_MULTI_COR_RCV)
1133 			e_info.multi_error_valid = 1;
1134 		else
1135 			e_info.multi_error_valid = 0;
1136 		aer_print_port_info(pdev, &e_info);
1137 
1138 		if (find_source_device(pdev, &e_info))
1139 			aer_process_err_devices(&e_info);
1140 	}
1141 
1142 	if (e_src->status & PCI_ERR_ROOT_UNCOR_RCV) {
1143 		e_info.id = ERR_UNCOR_ID(e_src->id);
1144 
1145 		if (e_src->status & PCI_ERR_ROOT_FATAL_RCV)
1146 			e_info.severity = AER_FATAL;
1147 		else
1148 			e_info.severity = AER_NONFATAL;
1149 
1150 		if (e_src->status & PCI_ERR_ROOT_MULTI_UNCOR_RCV)
1151 			e_info.multi_error_valid = 1;
1152 		else
1153 			e_info.multi_error_valid = 0;
1154 
1155 		aer_print_port_info(pdev, &e_info);
1156 
1157 		if (find_source_device(pdev, &e_info))
1158 			aer_process_err_devices(&e_info);
1159 	}
1160 }
1161 
1162 /**
1163  * aer_isr - consume errors detected by root port
1164  * @irq: IRQ assigned to Root Port
1165  * @context: pointer to Root Port data structure
1166  *
1167  * Invoked, as DPC, when root port records new detected error
1168  */
1169 static irqreturn_t aer_isr(int irq, void *context)
1170 {
1171 	struct pcie_device *dev = (struct pcie_device *)context;
1172 	struct aer_rpc *rpc = get_service_data(dev);
1173 	struct aer_err_source e_src;
1174 
1175 	if (kfifo_is_empty(&rpc->aer_fifo))
1176 		return IRQ_NONE;
1177 
1178 	while (kfifo_get(&rpc->aer_fifo, &e_src))
1179 		aer_isr_one_error(rpc, &e_src);
1180 	return IRQ_HANDLED;
1181 }
1182 
1183 /**
1184  * aer_irq - Root Port's ISR
1185  * @irq: IRQ assigned to Root Port
1186  * @context: pointer to Root Port data structure
1187  *
1188  * Invoked when Root Port detects AER messages.
1189  */
1190 static irqreturn_t aer_irq(int irq, void *context)
1191 {
1192 	struct pcie_device *pdev = (struct pcie_device *)context;
1193 	struct aer_rpc *rpc = get_service_data(pdev);
1194 	struct pci_dev *rp = rpc->rpd;
1195 	int aer = rp->aer_cap;
1196 	struct aer_err_source e_src = {};
1197 
1198 	pci_read_config_dword(rp, aer + PCI_ERR_ROOT_STATUS, &e_src.status);
1199 	if (!(e_src.status & (PCI_ERR_ROOT_UNCOR_RCV|PCI_ERR_ROOT_COR_RCV)))
1200 		return IRQ_NONE;
1201 
1202 	pci_read_config_dword(rp, aer + PCI_ERR_ROOT_ERR_SRC, &e_src.id);
1203 	pci_write_config_dword(rp, aer + PCI_ERR_ROOT_STATUS, e_src.status);
1204 
1205 	if (!kfifo_put(&rpc->aer_fifo, e_src))
1206 		return IRQ_HANDLED;
1207 
1208 	return IRQ_WAKE_THREAD;
1209 }
1210 
1211 static int set_device_error_reporting(struct pci_dev *dev, void *data)
1212 {
1213 	bool enable = *((bool *)data);
1214 	int type = pci_pcie_type(dev);
1215 
1216 	if ((type == PCI_EXP_TYPE_ROOT_PORT) ||
1217 	    (type == PCI_EXP_TYPE_RC_EC) ||
1218 	    (type == PCI_EXP_TYPE_UPSTREAM) ||
1219 	    (type == PCI_EXP_TYPE_DOWNSTREAM)) {
1220 		if (enable)
1221 			pci_enable_pcie_error_reporting(dev);
1222 		else
1223 			pci_disable_pcie_error_reporting(dev);
1224 	}
1225 
1226 	if (enable)
1227 		pcie_set_ecrc_checking(dev);
1228 
1229 	return 0;
1230 }
1231 
1232 /**
1233  * set_downstream_devices_error_reporting - enable/disable the error reporting  bits on the root port and its downstream ports.
1234  * @dev: pointer to root port's pci_dev data structure
1235  * @enable: true = enable error reporting, false = disable error reporting.
1236  */
1237 static void set_downstream_devices_error_reporting(struct pci_dev *dev,
1238 						   bool enable)
1239 {
1240 	set_device_error_reporting(dev, &enable);
1241 
1242 	if (pci_pcie_type(dev) == PCI_EXP_TYPE_RC_EC)
1243 		pcie_walk_rcec(dev, set_device_error_reporting, &enable);
1244 	else if (dev->subordinate)
1245 		pci_walk_bus(dev->subordinate, set_device_error_reporting,
1246 			     &enable);
1247 
1248 }
1249 
1250 /**
1251  * aer_enable_rootport - enable Root Port's interrupts when receiving messages
1252  * @rpc: pointer to a Root Port data structure
1253  *
1254  * Invoked when PCIe bus loads AER service driver.
1255  */
1256 static void aer_enable_rootport(struct aer_rpc *rpc)
1257 {
1258 	struct pci_dev *pdev = rpc->rpd;
1259 	int aer = pdev->aer_cap;
1260 	u16 reg16;
1261 	u32 reg32;
1262 
1263 	/* Clear PCIe Capability's Device Status */
1264 	pcie_capability_read_word(pdev, PCI_EXP_DEVSTA, &reg16);
1265 	pcie_capability_write_word(pdev, PCI_EXP_DEVSTA, reg16);
1266 
1267 	/* Disable system error generation in response to error messages */
1268 	pcie_capability_clear_word(pdev, PCI_EXP_RTCTL,
1269 				   SYSTEM_ERROR_INTR_ON_MESG_MASK);
1270 
1271 	/* Clear error status */
1272 	pci_read_config_dword(pdev, aer + PCI_ERR_ROOT_STATUS, &reg32);
1273 	pci_write_config_dword(pdev, aer + PCI_ERR_ROOT_STATUS, reg32);
1274 	pci_read_config_dword(pdev, aer + PCI_ERR_COR_STATUS, &reg32);
1275 	pci_write_config_dword(pdev, aer + PCI_ERR_COR_STATUS, reg32);
1276 	pci_read_config_dword(pdev, aer + PCI_ERR_UNCOR_STATUS, &reg32);
1277 	pci_write_config_dword(pdev, aer + PCI_ERR_UNCOR_STATUS, reg32);
1278 
1279 	/*
1280 	 * Enable error reporting for the root port device and downstream port
1281 	 * devices.
1282 	 */
1283 	set_downstream_devices_error_reporting(pdev, true);
1284 
1285 	/* Enable Root Port's interrupt in response to error messages */
1286 	pci_read_config_dword(pdev, aer + PCI_ERR_ROOT_COMMAND, &reg32);
1287 	reg32 |= ROOT_PORT_INTR_ON_MESG_MASK;
1288 	pci_write_config_dword(pdev, aer + PCI_ERR_ROOT_COMMAND, reg32);
1289 }
1290 
1291 /**
1292  * aer_disable_rootport - disable Root Port's interrupts when receiving messages
1293  * @rpc: pointer to a Root Port data structure
1294  *
1295  * Invoked when PCIe bus unloads AER service driver.
1296  */
1297 static void aer_disable_rootport(struct aer_rpc *rpc)
1298 {
1299 	struct pci_dev *pdev = rpc->rpd;
1300 	int aer = pdev->aer_cap;
1301 	u32 reg32;
1302 
1303 	/*
1304 	 * Disable error reporting for the root port device and downstream port
1305 	 * devices.
1306 	 */
1307 	set_downstream_devices_error_reporting(pdev, false);
1308 
1309 	/* Disable Root's interrupt in response to error messages */
1310 	pci_read_config_dword(pdev, aer + PCI_ERR_ROOT_COMMAND, &reg32);
1311 	reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
1312 	pci_write_config_dword(pdev, aer + PCI_ERR_ROOT_COMMAND, reg32);
1313 
1314 	/* Clear Root's error status reg */
1315 	pci_read_config_dword(pdev, aer + PCI_ERR_ROOT_STATUS, &reg32);
1316 	pci_write_config_dword(pdev, aer + PCI_ERR_ROOT_STATUS, reg32);
1317 }
1318 
1319 /**
1320  * aer_remove - clean up resources
1321  * @dev: pointer to the pcie_dev data structure
1322  *
1323  * Invoked when PCI Express bus unloads or AER probe fails.
1324  */
1325 static void aer_remove(struct pcie_device *dev)
1326 {
1327 	struct aer_rpc *rpc = get_service_data(dev);
1328 
1329 	aer_disable_rootport(rpc);
1330 }
1331 
1332 /**
1333  * aer_probe - initialize resources
1334  * @dev: pointer to the pcie_dev data structure
1335  *
1336  * Invoked when PCI Express bus loads AER service driver.
1337  */
1338 static int aer_probe(struct pcie_device *dev)
1339 {
1340 	int status;
1341 	struct aer_rpc *rpc;
1342 	struct device *device = &dev->device;
1343 	struct pci_dev *port = dev->port;
1344 
1345 	/* Limit to Root Ports or Root Complex Event Collectors */
1346 	if ((pci_pcie_type(port) != PCI_EXP_TYPE_RC_EC) &&
1347 	    (pci_pcie_type(port) != PCI_EXP_TYPE_ROOT_PORT))
1348 		return -ENODEV;
1349 
1350 	rpc = devm_kzalloc(device, sizeof(struct aer_rpc), GFP_KERNEL);
1351 	if (!rpc)
1352 		return -ENOMEM;
1353 
1354 	rpc->rpd = port;
1355 	INIT_KFIFO(rpc->aer_fifo);
1356 	set_service_data(dev, rpc);
1357 
1358 	status = devm_request_threaded_irq(device, dev->irq, aer_irq, aer_isr,
1359 					   IRQF_SHARED, "aerdrv", dev);
1360 	if (status) {
1361 		pci_err(port, "request AER IRQ %d failed\n", dev->irq);
1362 		return status;
1363 	}
1364 
1365 	aer_enable_rootport(rpc);
1366 	pci_info(port, "enabled with IRQ %d\n", dev->irq);
1367 	return 0;
1368 }
1369 
1370 /**
1371  * aer_root_reset - reset Root Port hierarchy, RCEC, or RCiEP
1372  * @dev: pointer to Root Port, RCEC, or RCiEP
1373  *
1374  * Invoked by Port Bus driver when performing reset.
1375  */
1376 static pci_ers_result_t aer_root_reset(struct pci_dev *dev)
1377 {
1378 	int type = pci_pcie_type(dev);
1379 	struct pci_dev *root;
1380 	int aer;
1381 	struct pci_host_bridge *host = pci_find_host_bridge(dev->bus);
1382 	u32 reg32;
1383 	int rc;
1384 
1385 	/*
1386 	 * Only Root Ports and RCECs have AER Root Command and Root Status
1387 	 * registers.  If "dev" is an RCiEP, the relevant registers are in
1388 	 * the RCEC.
1389 	 */
1390 	if (type == PCI_EXP_TYPE_RC_END)
1391 		root = dev->rcec;
1392 	else
1393 		root = pcie_find_root_port(dev);
1394 
1395 	/*
1396 	 * If the platform retained control of AER, an RCiEP may not have
1397 	 * an RCEC visible to us, so dev->rcec ("root") may be NULL.  In
1398 	 * that case, firmware is responsible for these registers.
1399 	 */
1400 	aer = root ? root->aer_cap : 0;
1401 
1402 	if ((host->native_aer || pcie_ports_native) && aer) {
1403 		/* Disable Root's interrupt in response to error messages */
1404 		pci_read_config_dword(root, aer + PCI_ERR_ROOT_COMMAND, &reg32);
1405 		reg32 &= ~ROOT_PORT_INTR_ON_MESG_MASK;
1406 		pci_write_config_dword(root, aer + PCI_ERR_ROOT_COMMAND, reg32);
1407 	}
1408 
1409 	if (type == PCI_EXP_TYPE_RC_EC || type == PCI_EXP_TYPE_RC_END) {
1410 		rc = pcie_reset_flr(dev, PCI_RESET_DO_RESET);
1411 		if (!rc)
1412 			pci_info(dev, "has been reset\n");
1413 		else
1414 			pci_info(dev, "not reset (no FLR support: %d)\n", rc);
1415 	} else {
1416 		rc = pci_bus_error_reset(dev);
1417 		pci_info(dev, "%s Port link has been reset (%d)\n",
1418 			pci_is_root_bus(dev->bus) ? "Root" : "Downstream", rc);
1419 	}
1420 
1421 	if ((host->native_aer || pcie_ports_native) && aer) {
1422 		/* Clear Root Error Status */
1423 		pci_read_config_dword(root, aer + PCI_ERR_ROOT_STATUS, &reg32);
1424 		pci_write_config_dword(root, aer + PCI_ERR_ROOT_STATUS, reg32);
1425 
1426 		/* Enable Root Port's interrupt in response to error messages */
1427 		pci_read_config_dword(root, aer + PCI_ERR_ROOT_COMMAND, &reg32);
1428 		reg32 |= ROOT_PORT_INTR_ON_MESG_MASK;
1429 		pci_write_config_dword(root, aer + PCI_ERR_ROOT_COMMAND, reg32);
1430 	}
1431 
1432 	return rc ? PCI_ERS_RESULT_DISCONNECT : PCI_ERS_RESULT_RECOVERED;
1433 }
1434 
1435 static struct pcie_port_service_driver aerdriver = {
1436 	.name		= "aer",
1437 	.port_type	= PCIE_ANY_PORT,
1438 	.service	= PCIE_PORT_SERVICE_AER,
1439 
1440 	.probe		= aer_probe,
1441 	.remove		= aer_remove,
1442 };
1443 
1444 /**
1445  * pcie_aer_init - register AER root service driver
1446  *
1447  * Invoked when AER root service driver is loaded.
1448  */
1449 int __init pcie_aer_init(void)
1450 {
1451 	if (!pci_aer_available())
1452 		return -ENXIO;
1453 	return pcie_port_service_register(&aerdriver);
1454 }
1455