xref: /linux/drivers/gpu/drm/amd/amdgpu/amdgpu_ras.c (revision 7255fcc80d4b525cc10cfaaf7f485830d4ed2000)
1 /*
2  * Copyright 2018 Advanced Micro Devices, Inc.
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  *
22  *
23  */
24 #include <linux/debugfs.h>
25 #include <linux/list.h>
26 #include <linux/module.h>
27 #include <linux/uaccess.h>
28 #include <linux/reboot.h>
29 #include <linux/syscalls.h>
30 #include <linux/pm_runtime.h>
31 #include <linux/list_sort.h>
32 
33 #include "amdgpu.h"
34 #include "amdgpu_ras.h"
35 #include "amdgpu_atomfirmware.h"
36 #include "amdgpu_xgmi.h"
37 #include "ivsrcid/nbio/irqsrcs_nbif_7_4.h"
38 #include "nbio_v4_3.h"
39 #include "nbio_v7_9.h"
40 #include "atom.h"
41 #include "amdgpu_reset.h"
42 #include "amdgpu_psp.h"
43 
44 #ifdef CONFIG_X86_MCE_AMD
45 #include <asm/mce.h>
46 
47 static bool notifier_registered;
48 #endif
49 static const char *RAS_FS_NAME = "ras";
50 
51 const char *ras_error_string[] = {
52 	"none",
53 	"parity",
54 	"single_correctable",
55 	"multi_uncorrectable",
56 	"poison",
57 };
58 
59 const char *ras_block_string[] = {
60 	"umc",
61 	"sdma",
62 	"gfx",
63 	"mmhub",
64 	"athub",
65 	"pcie_bif",
66 	"hdp",
67 	"xgmi_wafl",
68 	"df",
69 	"smn",
70 	"sem",
71 	"mp0",
72 	"mp1",
73 	"fuse",
74 	"mca",
75 	"vcn",
76 	"jpeg",
77 	"ih",
78 	"mpio",
79 };
80 
81 const char *ras_mca_block_string[] = {
82 	"mca_mp0",
83 	"mca_mp1",
84 	"mca_mpio",
85 	"mca_iohc",
86 };
87 
88 struct amdgpu_ras_block_list {
89 	/* ras block link */
90 	struct list_head node;
91 
92 	struct amdgpu_ras_block_object *ras_obj;
93 };
94 
95 const char *get_ras_block_str(struct ras_common_if *ras_block)
96 {
97 	if (!ras_block)
98 		return "NULL";
99 
100 	if (ras_block->block >= AMDGPU_RAS_BLOCK_COUNT ||
101 	    ras_block->block >= ARRAY_SIZE(ras_block_string))
102 		return "OUT OF RANGE";
103 
104 	if (ras_block->block == AMDGPU_RAS_BLOCK__MCA)
105 		return ras_mca_block_string[ras_block->sub_block_index];
106 
107 	return ras_block_string[ras_block->block];
108 }
109 
110 #define ras_block_str(_BLOCK_) \
111 	(((_BLOCK_) < ARRAY_SIZE(ras_block_string)) ? ras_block_string[_BLOCK_] : "Out Of Range")
112 
113 #define ras_err_str(i) (ras_error_string[ffs(i)])
114 
115 #define RAS_DEFAULT_FLAGS (AMDGPU_RAS_FLAG_INIT_BY_VBIOS)
116 
117 /* inject address is 52 bits */
118 #define	RAS_UMC_INJECT_ADDR_LIMIT	(0x1ULL << 52)
119 
120 /* typical ECC bad page rate is 1 bad page per 100MB VRAM */
121 #define RAS_BAD_PAGE_COVER              (100 * 1024 * 1024ULL)
122 
123 #define MAX_UMC_POISON_POLLING_TIME_ASYNC  100  //ms
124 
125 enum amdgpu_ras_retire_page_reservation {
126 	AMDGPU_RAS_RETIRE_PAGE_RESERVED,
127 	AMDGPU_RAS_RETIRE_PAGE_PENDING,
128 	AMDGPU_RAS_RETIRE_PAGE_FAULT,
129 };
130 
131 atomic_t amdgpu_ras_in_intr = ATOMIC_INIT(0);
132 
133 static bool amdgpu_ras_check_bad_page_unlock(struct amdgpu_ras *con,
134 				uint64_t addr);
135 static bool amdgpu_ras_check_bad_page(struct amdgpu_device *adev,
136 				uint64_t addr);
137 #ifdef CONFIG_X86_MCE_AMD
138 static void amdgpu_register_bad_pages_mca_notifier(struct amdgpu_device *adev);
139 struct mce_notifier_adev_list {
140 	struct amdgpu_device *devs[MAX_GPU_INSTANCE];
141 	int num_gpu;
142 };
143 static struct mce_notifier_adev_list mce_adev_list;
144 #endif
145 
146 void amdgpu_ras_set_error_query_ready(struct amdgpu_device *adev, bool ready)
147 {
148 	if (adev && amdgpu_ras_get_context(adev))
149 		amdgpu_ras_get_context(adev)->error_query_ready = ready;
150 }
151 
152 static bool amdgpu_ras_get_error_query_ready(struct amdgpu_device *adev)
153 {
154 	if (adev && amdgpu_ras_get_context(adev))
155 		return amdgpu_ras_get_context(adev)->error_query_ready;
156 
157 	return false;
158 }
159 
160 static int amdgpu_reserve_page_direct(struct amdgpu_device *adev, uint64_t address)
161 {
162 	struct ras_err_data err_data;
163 	struct eeprom_table_record err_rec;
164 	int ret;
165 
166 	if ((address >= adev->gmc.mc_vram_size) ||
167 	    (address >= RAS_UMC_INJECT_ADDR_LIMIT)) {
168 		dev_warn(adev->dev,
169 		         "RAS WARN: input address 0x%llx is invalid.\n",
170 		         address);
171 		return -EINVAL;
172 	}
173 
174 	if (amdgpu_ras_check_bad_page(adev, address)) {
175 		dev_warn(adev->dev,
176 			 "RAS WARN: 0x%llx has already been marked as bad page!\n",
177 			 address);
178 		return 0;
179 	}
180 
181 	ret = amdgpu_ras_error_data_init(&err_data);
182 	if (ret)
183 		return ret;
184 
185 	memset(&err_rec, 0x0, sizeof(struct eeprom_table_record));
186 	err_data.err_addr = &err_rec;
187 	amdgpu_umc_fill_error_record(&err_data, address, address, 0, 0);
188 
189 	if (amdgpu_bad_page_threshold != 0) {
190 		amdgpu_ras_add_bad_pages(adev, err_data.err_addr,
191 					 err_data.err_addr_cnt);
192 		amdgpu_ras_save_bad_pages(adev, NULL);
193 	}
194 
195 	amdgpu_ras_error_data_fini(&err_data);
196 
197 	dev_warn(adev->dev, "WARNING: THIS IS ONLY FOR TEST PURPOSES AND WILL CORRUPT RAS EEPROM\n");
198 	dev_warn(adev->dev, "Clear EEPROM:\n");
199 	dev_warn(adev->dev, "    echo 1 > /sys/kernel/debug/dri/0/ras/ras_eeprom_reset\n");
200 
201 	return 0;
202 }
203 
204 static ssize_t amdgpu_ras_debugfs_read(struct file *f, char __user *buf,
205 					size_t size, loff_t *pos)
206 {
207 	struct ras_manager *obj = (struct ras_manager *)file_inode(f)->i_private;
208 	struct ras_query_if info = {
209 		.head = obj->head,
210 	};
211 	ssize_t s;
212 	char val[128];
213 
214 	if (amdgpu_ras_query_error_status(obj->adev, &info))
215 		return -EINVAL;
216 
217 	/* Hardware counter will be reset automatically after the query on Vega20 and Arcturus */
218 	if (amdgpu_ip_version(obj->adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 2) &&
219 	    amdgpu_ip_version(obj->adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 4)) {
220 		if (amdgpu_ras_reset_error_status(obj->adev, info.head.block))
221 			dev_warn(obj->adev->dev, "Failed to reset error counter and error status");
222 	}
223 
224 	s = snprintf(val, sizeof(val), "%s: %lu\n%s: %lu\n",
225 			"ue", info.ue_count,
226 			"ce", info.ce_count);
227 	if (*pos >= s)
228 		return 0;
229 
230 	s -= *pos;
231 	s = min_t(u64, s, size);
232 
233 
234 	if (copy_to_user(buf, &val[*pos], s))
235 		return -EINVAL;
236 
237 	*pos += s;
238 
239 	return s;
240 }
241 
242 static const struct file_operations amdgpu_ras_debugfs_ops = {
243 	.owner = THIS_MODULE,
244 	.read = amdgpu_ras_debugfs_read,
245 	.write = NULL,
246 	.llseek = default_llseek
247 };
248 
249 static int amdgpu_ras_find_block_id_by_name(const char *name, int *block_id)
250 {
251 	int i;
252 
253 	for (i = 0; i < ARRAY_SIZE(ras_block_string); i++) {
254 		*block_id = i;
255 		if (strcmp(name, ras_block_string[i]) == 0)
256 			return 0;
257 	}
258 	return -EINVAL;
259 }
260 
261 static int amdgpu_ras_debugfs_ctrl_parse_data(struct file *f,
262 		const char __user *buf, size_t size,
263 		loff_t *pos, struct ras_debug_if *data)
264 {
265 	ssize_t s = min_t(u64, 64, size);
266 	char str[65];
267 	char block_name[33];
268 	char err[9] = "ue";
269 	int op = -1;
270 	int block_id;
271 	uint32_t sub_block;
272 	u64 address, value;
273 	/* default value is 0 if the mask is not set by user */
274 	u32 instance_mask = 0;
275 
276 	if (*pos)
277 		return -EINVAL;
278 	*pos = size;
279 
280 	memset(str, 0, sizeof(str));
281 	memset(data, 0, sizeof(*data));
282 
283 	if (copy_from_user(str, buf, s))
284 		return -EINVAL;
285 
286 	if (sscanf(str, "disable %32s", block_name) == 1)
287 		op = 0;
288 	else if (sscanf(str, "enable %32s %8s", block_name, err) == 2)
289 		op = 1;
290 	else if (sscanf(str, "inject %32s %8s", block_name, err) == 2)
291 		op = 2;
292 	else if (strstr(str, "retire_page") != NULL)
293 		op = 3;
294 	else if (str[0] && str[1] && str[2] && str[3])
295 		/* ascii string, but commands are not matched. */
296 		return -EINVAL;
297 
298 	if (op != -1) {
299 		if (op == 3) {
300 			if (sscanf(str, "%*s 0x%llx", &address) != 1 &&
301 			    sscanf(str, "%*s %llu", &address) != 1)
302 				return -EINVAL;
303 
304 			data->op = op;
305 			data->inject.address = address;
306 
307 			return 0;
308 		}
309 
310 		if (amdgpu_ras_find_block_id_by_name(block_name, &block_id))
311 			return -EINVAL;
312 
313 		data->head.block = block_id;
314 		/* only ue, ce and poison errors are supported */
315 		if (!memcmp("ue", err, 2))
316 			data->head.type = AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE;
317 		else if (!memcmp("ce", err, 2))
318 			data->head.type = AMDGPU_RAS_ERROR__SINGLE_CORRECTABLE;
319 		else if (!memcmp("poison", err, 6))
320 			data->head.type = AMDGPU_RAS_ERROR__POISON;
321 		else
322 			return -EINVAL;
323 
324 		data->op = op;
325 
326 		if (op == 2) {
327 			if (sscanf(str, "%*s %*s %*s 0x%x 0x%llx 0x%llx 0x%x",
328 				   &sub_block, &address, &value, &instance_mask) != 4 &&
329 			    sscanf(str, "%*s %*s %*s %u %llu %llu %u",
330 				   &sub_block, &address, &value, &instance_mask) != 4 &&
331 				sscanf(str, "%*s %*s %*s 0x%x 0x%llx 0x%llx",
332 				   &sub_block, &address, &value) != 3 &&
333 			    sscanf(str, "%*s %*s %*s %u %llu %llu",
334 				   &sub_block, &address, &value) != 3)
335 				return -EINVAL;
336 			data->head.sub_block_index = sub_block;
337 			data->inject.address = address;
338 			data->inject.value = value;
339 			data->inject.instance_mask = instance_mask;
340 		}
341 	} else {
342 		if (size < sizeof(*data))
343 			return -EINVAL;
344 
345 		if (copy_from_user(data, buf, sizeof(*data)))
346 			return -EINVAL;
347 	}
348 
349 	return 0;
350 }
351 
352 static void amdgpu_ras_instance_mask_check(struct amdgpu_device *adev,
353 				struct ras_debug_if *data)
354 {
355 	int num_xcc = adev->gfx.xcc_mask ? NUM_XCC(adev->gfx.xcc_mask) : 1;
356 	uint32_t mask, inst_mask = data->inject.instance_mask;
357 
358 	/* no need to set instance mask if there is only one instance */
359 	if (num_xcc <= 1 && inst_mask) {
360 		data->inject.instance_mask = 0;
361 		dev_dbg(adev->dev,
362 			"RAS inject mask(0x%x) isn't supported and force it to 0.\n",
363 			inst_mask);
364 
365 		return;
366 	}
367 
368 	switch (data->head.block) {
369 	case AMDGPU_RAS_BLOCK__GFX:
370 		mask = GENMASK(num_xcc - 1, 0);
371 		break;
372 	case AMDGPU_RAS_BLOCK__SDMA:
373 		mask = GENMASK(adev->sdma.num_instances - 1, 0);
374 		break;
375 	case AMDGPU_RAS_BLOCK__VCN:
376 	case AMDGPU_RAS_BLOCK__JPEG:
377 		mask = GENMASK(adev->vcn.num_vcn_inst - 1, 0);
378 		break;
379 	default:
380 		mask = inst_mask;
381 		break;
382 	}
383 
384 	/* remove invalid bits in instance mask */
385 	data->inject.instance_mask &= mask;
386 	if (inst_mask != data->inject.instance_mask)
387 		dev_dbg(adev->dev,
388 			"Adjust RAS inject mask 0x%x to 0x%x\n",
389 			inst_mask, data->inject.instance_mask);
390 }
391 
392 /**
393  * DOC: AMDGPU RAS debugfs control interface
394  *
395  * The control interface accepts struct ras_debug_if which has two members.
396  *
397  * First member: ras_debug_if::head or ras_debug_if::inject.
398  *
399  * head is used to indicate which IP block will be under control.
400  *
401  * head has four members, they are block, type, sub_block_index, name.
402  * block: which IP will be under control.
403  * type: what kind of error will be enabled/disabled/injected.
404  * sub_block_index: some IPs have subcomponets. say, GFX, sDMA.
405  * name: the name of IP.
406  *
407  * inject has three more members than head, they are address, value and mask.
408  * As their names indicate, inject operation will write the
409  * value to the address.
410  *
411  * The second member: struct ras_debug_if::op.
412  * It has three kinds of operations.
413  *
414  * - 0: disable RAS on the block. Take ::head as its data.
415  * - 1: enable RAS on the block. Take ::head as its data.
416  * - 2: inject errors on the block. Take ::inject as its data.
417  *
418  * How to use the interface?
419  *
420  * In a program
421  *
422  * Copy the struct ras_debug_if in your code and initialize it.
423  * Write the struct to the control interface.
424  *
425  * From shell
426  *
427  * .. code-block:: bash
428  *
429  *	echo "disable <block>" > /sys/kernel/debug/dri/<N>/ras/ras_ctrl
430  *	echo "enable  <block> <error>" > /sys/kernel/debug/dri/<N>/ras/ras_ctrl
431  *	echo "inject  <block> <error> <sub-block> <address> <value> <mask>" > /sys/kernel/debug/dri/<N>/ras/ras_ctrl
432  *
433  * Where N, is the card which you want to affect.
434  *
435  * "disable" requires only the block.
436  * "enable" requires the block and error type.
437  * "inject" requires the block, error type, address, and value.
438  *
439  * The block is one of: umc, sdma, gfx, etc.
440  *	see ras_block_string[] for details
441  *
442  * The error type is one of: ue, ce and poison where,
443  *	ue is multi-uncorrectable
444  *	ce is single-correctable
445  *	poison is poison
446  *
447  * The sub-block is a the sub-block index, pass 0 if there is no sub-block.
448  * The address and value are hexadecimal numbers, leading 0x is optional.
449  * The mask means instance mask, is optional, default value is 0x1.
450  *
451  * For instance,
452  *
453  * .. code-block:: bash
454  *
455  *	echo inject umc ue 0x0 0x0 0x0 > /sys/kernel/debug/dri/0/ras/ras_ctrl
456  *	echo inject umc ce 0 0 0 3 > /sys/kernel/debug/dri/0/ras/ras_ctrl
457  *	echo disable umc > /sys/kernel/debug/dri/0/ras/ras_ctrl
458  *
459  * How to check the result of the operation?
460  *
461  * To check disable/enable, see "ras" features at,
462  * /sys/class/drm/card[0/1/2...]/device/ras/features
463  *
464  * To check inject, see the corresponding error count at,
465  * /sys/class/drm/card[0/1/2...]/device/ras/[gfx|sdma|umc|...]_err_count
466  *
467  * .. note::
468  *	Operations are only allowed on blocks which are supported.
469  *	Check the "ras" mask at /sys/module/amdgpu/parameters/ras_mask
470  *	to see which blocks support RAS on a particular asic.
471  *
472  */
473 static ssize_t amdgpu_ras_debugfs_ctrl_write(struct file *f,
474 					     const char __user *buf,
475 					     size_t size, loff_t *pos)
476 {
477 	struct amdgpu_device *adev = (struct amdgpu_device *)file_inode(f)->i_private;
478 	struct ras_debug_if data;
479 	int ret = 0;
480 
481 	if (!amdgpu_ras_get_error_query_ready(adev)) {
482 		dev_warn(adev->dev, "RAS WARN: error injection "
483 				"currently inaccessible\n");
484 		return size;
485 	}
486 
487 	ret = amdgpu_ras_debugfs_ctrl_parse_data(f, buf, size, pos, &data);
488 	if (ret)
489 		return ret;
490 
491 	if (data.op == 3) {
492 		ret = amdgpu_reserve_page_direct(adev, data.inject.address);
493 		if (!ret)
494 			return size;
495 		else
496 			return ret;
497 	}
498 
499 	if (!amdgpu_ras_is_supported(adev, data.head.block))
500 		return -EINVAL;
501 
502 	switch (data.op) {
503 	case 0:
504 		ret = amdgpu_ras_feature_enable(adev, &data.head, 0);
505 		break;
506 	case 1:
507 		ret = amdgpu_ras_feature_enable(adev, &data.head, 1);
508 		break;
509 	case 2:
510 		if ((data.inject.address >= adev->gmc.mc_vram_size &&
511 		    adev->gmc.mc_vram_size) ||
512 		    (data.inject.address >= RAS_UMC_INJECT_ADDR_LIMIT)) {
513 			dev_warn(adev->dev, "RAS WARN: input address "
514 					"0x%llx is invalid.",
515 					data.inject.address);
516 			ret = -EINVAL;
517 			break;
518 		}
519 
520 		/* umc ce/ue error injection for a bad page is not allowed */
521 		if ((data.head.block == AMDGPU_RAS_BLOCK__UMC) &&
522 		    amdgpu_ras_check_bad_page(adev, data.inject.address)) {
523 			dev_warn(adev->dev, "RAS WARN: inject: 0x%llx has "
524 				 "already been marked as bad!\n",
525 				 data.inject.address);
526 			break;
527 		}
528 
529 		amdgpu_ras_instance_mask_check(adev, &data);
530 
531 		/* data.inject.address is offset instead of absolute gpu address */
532 		ret = amdgpu_ras_error_inject(adev, &data.inject);
533 		break;
534 	default:
535 		ret = -EINVAL;
536 		break;
537 	}
538 
539 	if (ret)
540 		return ret;
541 
542 	return size;
543 }
544 
545 /**
546  * DOC: AMDGPU RAS debugfs EEPROM table reset interface
547  *
548  * Some boards contain an EEPROM which is used to persistently store a list of
549  * bad pages which experiences ECC errors in vram.  This interface provides
550  * a way to reset the EEPROM, e.g., after testing error injection.
551  *
552  * Usage:
553  *
554  * .. code-block:: bash
555  *
556  *	echo 1 > ../ras/ras_eeprom_reset
557  *
558  * will reset EEPROM table to 0 entries.
559  *
560  */
561 static ssize_t amdgpu_ras_debugfs_eeprom_write(struct file *f,
562 					       const char __user *buf,
563 					       size_t size, loff_t *pos)
564 {
565 	struct amdgpu_device *adev =
566 		(struct amdgpu_device *)file_inode(f)->i_private;
567 	int ret;
568 
569 	ret = amdgpu_ras_eeprom_reset_table(
570 		&(amdgpu_ras_get_context(adev)->eeprom_control));
571 
572 	if (!ret) {
573 		/* Something was written to EEPROM.
574 		 */
575 		amdgpu_ras_get_context(adev)->flags = RAS_DEFAULT_FLAGS;
576 		return size;
577 	} else {
578 		return ret;
579 	}
580 }
581 
582 static const struct file_operations amdgpu_ras_debugfs_ctrl_ops = {
583 	.owner = THIS_MODULE,
584 	.read = NULL,
585 	.write = amdgpu_ras_debugfs_ctrl_write,
586 	.llseek = default_llseek
587 };
588 
589 static const struct file_operations amdgpu_ras_debugfs_eeprom_ops = {
590 	.owner = THIS_MODULE,
591 	.read = NULL,
592 	.write = amdgpu_ras_debugfs_eeprom_write,
593 	.llseek = default_llseek
594 };
595 
596 /**
597  * DOC: AMDGPU RAS sysfs Error Count Interface
598  *
599  * It allows the user to read the error count for each IP block on the gpu through
600  * /sys/class/drm/card[0/1/2...]/device/ras/[gfx/sdma/...]_err_count
601  *
602  * It outputs the multiple lines which report the uncorrected (ue) and corrected
603  * (ce) error counts.
604  *
605  * The format of one line is below,
606  *
607  * [ce|ue]: count
608  *
609  * Example:
610  *
611  * .. code-block:: bash
612  *
613  *	ue: 0
614  *	ce: 1
615  *
616  */
617 static ssize_t amdgpu_ras_sysfs_read(struct device *dev,
618 		struct device_attribute *attr, char *buf)
619 {
620 	struct ras_manager *obj = container_of(attr, struct ras_manager, sysfs_attr);
621 	struct ras_query_if info = {
622 		.head = obj->head,
623 	};
624 
625 	if (!amdgpu_ras_get_error_query_ready(obj->adev))
626 		return sysfs_emit(buf, "Query currently inaccessible\n");
627 
628 	if (amdgpu_ras_query_error_status(obj->adev, &info))
629 		return -EINVAL;
630 
631 	if (amdgpu_ip_version(obj->adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 2) &&
632 	    amdgpu_ip_version(obj->adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 4)) {
633 		if (amdgpu_ras_reset_error_status(obj->adev, info.head.block))
634 			dev_warn(obj->adev->dev, "Failed to reset error counter and error status");
635 	}
636 
637 	if (info.head.block == AMDGPU_RAS_BLOCK__UMC)
638 		return sysfs_emit(buf, "%s: %lu\n%s: %lu\n%s: %lu\n", "ue", info.ue_count,
639 				"ce", info.ce_count, "de", info.de_count);
640 	else
641 		return sysfs_emit(buf, "%s: %lu\n%s: %lu\n", "ue", info.ue_count,
642 				"ce", info.ce_count);
643 }
644 
645 /* obj begin */
646 
647 #define get_obj(obj) do { (obj)->use++; } while (0)
648 #define alive_obj(obj) ((obj)->use)
649 
650 static inline void put_obj(struct ras_manager *obj)
651 {
652 	if (obj && (--obj->use == 0)) {
653 		list_del(&obj->node);
654 		amdgpu_ras_error_data_fini(&obj->err_data);
655 	}
656 
657 	if (obj && (obj->use < 0))
658 		DRM_ERROR("RAS ERROR: Unbalance obj(%s) use\n", get_ras_block_str(&obj->head));
659 }
660 
661 /* make one obj and return it. */
662 static struct ras_manager *amdgpu_ras_create_obj(struct amdgpu_device *adev,
663 		struct ras_common_if *head)
664 {
665 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
666 	struct ras_manager *obj;
667 
668 	if (!adev->ras_enabled || !con)
669 		return NULL;
670 
671 	if (head->block >= AMDGPU_RAS_BLOCK_COUNT)
672 		return NULL;
673 
674 	if (head->block == AMDGPU_RAS_BLOCK__MCA) {
675 		if (head->sub_block_index >= AMDGPU_RAS_MCA_BLOCK__LAST)
676 			return NULL;
677 
678 		obj = &con->objs[AMDGPU_RAS_BLOCK__LAST + head->sub_block_index];
679 	} else
680 		obj = &con->objs[head->block];
681 
682 	/* already exist. return obj? */
683 	if (alive_obj(obj))
684 		return NULL;
685 
686 	if (amdgpu_ras_error_data_init(&obj->err_data))
687 		return NULL;
688 
689 	obj->head = *head;
690 	obj->adev = adev;
691 	list_add(&obj->node, &con->head);
692 	get_obj(obj);
693 
694 	return obj;
695 }
696 
697 /* return an obj equal to head, or the first when head is NULL */
698 struct ras_manager *amdgpu_ras_find_obj(struct amdgpu_device *adev,
699 		struct ras_common_if *head)
700 {
701 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
702 	struct ras_manager *obj;
703 	int i;
704 
705 	if (!adev->ras_enabled || !con)
706 		return NULL;
707 
708 	if (head) {
709 		if (head->block >= AMDGPU_RAS_BLOCK_COUNT)
710 			return NULL;
711 
712 		if (head->block == AMDGPU_RAS_BLOCK__MCA) {
713 			if (head->sub_block_index >= AMDGPU_RAS_MCA_BLOCK__LAST)
714 				return NULL;
715 
716 			obj = &con->objs[AMDGPU_RAS_BLOCK__LAST + head->sub_block_index];
717 		} else
718 			obj = &con->objs[head->block];
719 
720 		if (alive_obj(obj))
721 			return obj;
722 	} else {
723 		for (i = 0; i < AMDGPU_RAS_BLOCK_COUNT + AMDGPU_RAS_MCA_BLOCK_COUNT; i++) {
724 			obj = &con->objs[i];
725 			if (alive_obj(obj))
726 				return obj;
727 		}
728 	}
729 
730 	return NULL;
731 }
732 /* obj end */
733 
734 /* feature ctl begin */
735 static int amdgpu_ras_is_feature_allowed(struct amdgpu_device *adev,
736 					 struct ras_common_if *head)
737 {
738 	return adev->ras_hw_enabled & BIT(head->block);
739 }
740 
741 static int amdgpu_ras_is_feature_enabled(struct amdgpu_device *adev,
742 		struct ras_common_if *head)
743 {
744 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
745 
746 	return con->features & BIT(head->block);
747 }
748 
749 /*
750  * if obj is not created, then create one.
751  * set feature enable flag.
752  */
753 static int __amdgpu_ras_feature_enable(struct amdgpu_device *adev,
754 		struct ras_common_if *head, int enable)
755 {
756 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
757 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, head);
758 
759 	/* If hardware does not support ras, then do not create obj.
760 	 * But if hardware support ras, we can create the obj.
761 	 * Ras framework checks con->hw_supported to see if it need do
762 	 * corresponding initialization.
763 	 * IP checks con->support to see if it need disable ras.
764 	 */
765 	if (!amdgpu_ras_is_feature_allowed(adev, head))
766 		return 0;
767 
768 	if (enable) {
769 		if (!obj) {
770 			obj = amdgpu_ras_create_obj(adev, head);
771 			if (!obj)
772 				return -EINVAL;
773 		} else {
774 			/* In case we create obj somewhere else */
775 			get_obj(obj);
776 		}
777 		con->features |= BIT(head->block);
778 	} else {
779 		if (obj && amdgpu_ras_is_feature_enabled(adev, head)) {
780 			con->features &= ~BIT(head->block);
781 			put_obj(obj);
782 		}
783 	}
784 
785 	return 0;
786 }
787 
788 /* wrapper of psp_ras_enable_features */
789 int amdgpu_ras_feature_enable(struct amdgpu_device *adev,
790 		struct ras_common_if *head, bool enable)
791 {
792 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
793 	union ta_ras_cmd_input *info;
794 	int ret;
795 
796 	if (!con)
797 		return -EINVAL;
798 
799 	/* For non-gfx ip, do not enable ras feature if it is not allowed */
800 	/* For gfx ip, regardless of feature support status, */
801 	/* Force issue enable or disable ras feature commands */
802 	if (head->block != AMDGPU_RAS_BLOCK__GFX &&
803 	    !amdgpu_ras_is_feature_allowed(adev, head))
804 		return 0;
805 
806 	/* Only enable gfx ras feature from host side */
807 	if (head->block == AMDGPU_RAS_BLOCK__GFX &&
808 	    !amdgpu_sriov_vf(adev) &&
809 	    !amdgpu_ras_intr_triggered()) {
810 		info = kzalloc(sizeof(union ta_ras_cmd_input), GFP_KERNEL);
811 		if (!info)
812 			return -ENOMEM;
813 
814 		if (!enable) {
815 			info->disable_features = (struct ta_ras_disable_features_input) {
816 				.block_id =  amdgpu_ras_block_to_ta(head->block),
817 				.error_type = amdgpu_ras_error_to_ta(head->type),
818 			};
819 		} else {
820 			info->enable_features = (struct ta_ras_enable_features_input) {
821 				.block_id =  amdgpu_ras_block_to_ta(head->block),
822 				.error_type = amdgpu_ras_error_to_ta(head->type),
823 			};
824 		}
825 
826 		ret = psp_ras_enable_features(&adev->psp, info, enable);
827 		if (ret) {
828 			dev_err(adev->dev, "ras %s %s failed poison:%d ret:%d\n",
829 				enable ? "enable":"disable",
830 				get_ras_block_str(head),
831 				amdgpu_ras_is_poison_mode_supported(adev), ret);
832 			kfree(info);
833 			return ret;
834 		}
835 
836 		kfree(info);
837 	}
838 
839 	/* setup the obj */
840 	__amdgpu_ras_feature_enable(adev, head, enable);
841 
842 	return 0;
843 }
844 
845 /* Only used in device probe stage and called only once. */
846 int amdgpu_ras_feature_enable_on_boot(struct amdgpu_device *adev,
847 		struct ras_common_if *head, bool enable)
848 {
849 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
850 	int ret;
851 
852 	if (!con)
853 		return -EINVAL;
854 
855 	if (con->flags & AMDGPU_RAS_FLAG_INIT_BY_VBIOS) {
856 		if (enable) {
857 			/* There is no harm to issue a ras TA cmd regardless of
858 			 * the currecnt ras state.
859 			 * If current state == target state, it will do nothing
860 			 * But sometimes it requests driver to reset and repost
861 			 * with error code -EAGAIN.
862 			 */
863 			ret = amdgpu_ras_feature_enable(adev, head, 1);
864 			/* With old ras TA, we might fail to enable ras.
865 			 * Log it and just setup the object.
866 			 * TODO need remove this WA in the future.
867 			 */
868 			if (ret == -EINVAL) {
869 				ret = __amdgpu_ras_feature_enable(adev, head, 1);
870 				if (!ret)
871 					dev_info(adev->dev,
872 						"RAS INFO: %s setup object\n",
873 						get_ras_block_str(head));
874 			}
875 		} else {
876 			/* setup the object then issue a ras TA disable cmd.*/
877 			ret = __amdgpu_ras_feature_enable(adev, head, 1);
878 			if (ret)
879 				return ret;
880 
881 			/* gfx block ras dsiable cmd must send to ras-ta */
882 			if (head->block == AMDGPU_RAS_BLOCK__GFX)
883 				con->features |= BIT(head->block);
884 
885 			ret = amdgpu_ras_feature_enable(adev, head, 0);
886 
887 			/* clean gfx block ras features flag */
888 			if (adev->ras_enabled && head->block == AMDGPU_RAS_BLOCK__GFX)
889 				con->features &= ~BIT(head->block);
890 		}
891 	} else
892 		ret = amdgpu_ras_feature_enable(adev, head, enable);
893 
894 	return ret;
895 }
896 
897 static int amdgpu_ras_disable_all_features(struct amdgpu_device *adev,
898 		bool bypass)
899 {
900 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
901 	struct ras_manager *obj, *tmp;
902 
903 	list_for_each_entry_safe(obj, tmp, &con->head, node) {
904 		/* bypass psp.
905 		 * aka just release the obj and corresponding flags
906 		 */
907 		if (bypass) {
908 			if (__amdgpu_ras_feature_enable(adev, &obj->head, 0))
909 				break;
910 		} else {
911 			if (amdgpu_ras_feature_enable(adev, &obj->head, 0))
912 				break;
913 		}
914 	}
915 
916 	return con->features;
917 }
918 
919 static int amdgpu_ras_enable_all_features(struct amdgpu_device *adev,
920 		bool bypass)
921 {
922 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
923 	int i;
924 	const enum amdgpu_ras_error_type default_ras_type = AMDGPU_RAS_ERROR__NONE;
925 
926 	for (i = 0; i < AMDGPU_RAS_BLOCK_COUNT; i++) {
927 		struct ras_common_if head = {
928 			.block = i,
929 			.type = default_ras_type,
930 			.sub_block_index = 0,
931 		};
932 
933 		if (i == AMDGPU_RAS_BLOCK__MCA)
934 			continue;
935 
936 		if (bypass) {
937 			/*
938 			 * bypass psp. vbios enable ras for us.
939 			 * so just create the obj
940 			 */
941 			if (__amdgpu_ras_feature_enable(adev, &head, 1))
942 				break;
943 		} else {
944 			if (amdgpu_ras_feature_enable(adev, &head, 1))
945 				break;
946 		}
947 	}
948 
949 	for (i = 0; i < AMDGPU_RAS_MCA_BLOCK_COUNT; i++) {
950 		struct ras_common_if head = {
951 			.block = AMDGPU_RAS_BLOCK__MCA,
952 			.type = default_ras_type,
953 			.sub_block_index = i,
954 		};
955 
956 		if (bypass) {
957 			/*
958 			 * bypass psp. vbios enable ras for us.
959 			 * so just create the obj
960 			 */
961 			if (__amdgpu_ras_feature_enable(adev, &head, 1))
962 				break;
963 		} else {
964 			if (amdgpu_ras_feature_enable(adev, &head, 1))
965 				break;
966 		}
967 	}
968 
969 	return con->features;
970 }
971 /* feature ctl end */
972 
973 static int amdgpu_ras_block_match_default(struct amdgpu_ras_block_object *block_obj,
974 		enum amdgpu_ras_block block)
975 {
976 	if (!block_obj)
977 		return -EINVAL;
978 
979 	if (block_obj->ras_comm.block == block)
980 		return 0;
981 
982 	return -EINVAL;
983 }
984 
985 static struct amdgpu_ras_block_object *amdgpu_ras_get_ras_block(struct amdgpu_device *adev,
986 					enum amdgpu_ras_block block, uint32_t sub_block_index)
987 {
988 	struct amdgpu_ras_block_list *node, *tmp;
989 	struct amdgpu_ras_block_object *obj;
990 
991 	if (block >= AMDGPU_RAS_BLOCK__LAST)
992 		return NULL;
993 
994 	list_for_each_entry_safe(node, tmp, &adev->ras_list, node) {
995 		if (!node->ras_obj) {
996 			dev_warn(adev->dev, "Warning: abnormal ras list node.\n");
997 			continue;
998 		}
999 
1000 		obj = node->ras_obj;
1001 		if (obj->ras_block_match) {
1002 			if (obj->ras_block_match(obj, block, sub_block_index) == 0)
1003 				return obj;
1004 		} else {
1005 			if (amdgpu_ras_block_match_default(obj, block) == 0)
1006 				return obj;
1007 		}
1008 	}
1009 
1010 	return NULL;
1011 }
1012 
1013 static void amdgpu_ras_get_ecc_info(struct amdgpu_device *adev, struct ras_err_data *err_data)
1014 {
1015 	struct amdgpu_ras *ras = amdgpu_ras_get_context(adev);
1016 	int ret = 0;
1017 
1018 	/*
1019 	 * choosing right query method according to
1020 	 * whether smu support query error information
1021 	 */
1022 	ret = amdgpu_dpm_get_ecc_info(adev, (void *)&(ras->umc_ecc));
1023 	if (ret == -EOPNOTSUPP) {
1024 		if (adev->umc.ras && adev->umc.ras->ras_block.hw_ops &&
1025 			adev->umc.ras->ras_block.hw_ops->query_ras_error_count)
1026 			adev->umc.ras->ras_block.hw_ops->query_ras_error_count(adev, err_data);
1027 
1028 		/* umc query_ras_error_address is also responsible for clearing
1029 		 * error status
1030 		 */
1031 		if (adev->umc.ras && adev->umc.ras->ras_block.hw_ops &&
1032 		    adev->umc.ras->ras_block.hw_ops->query_ras_error_address)
1033 			adev->umc.ras->ras_block.hw_ops->query_ras_error_address(adev, err_data);
1034 	} else if (!ret) {
1035 		if (adev->umc.ras &&
1036 			adev->umc.ras->ecc_info_query_ras_error_count)
1037 			adev->umc.ras->ecc_info_query_ras_error_count(adev, err_data);
1038 
1039 		if (adev->umc.ras &&
1040 			adev->umc.ras->ecc_info_query_ras_error_address)
1041 			adev->umc.ras->ecc_info_query_ras_error_address(adev, err_data);
1042 	}
1043 }
1044 
1045 static void amdgpu_ras_error_print_error_data(struct amdgpu_device *adev,
1046 					      struct ras_manager *ras_mgr,
1047 					      struct ras_err_data *err_data,
1048 					      const char *blk_name,
1049 					      bool is_ue,
1050 					      bool is_de)
1051 {
1052 	struct amdgpu_smuio_mcm_config_info *mcm_info;
1053 	struct ras_err_node *err_node;
1054 	struct ras_err_info *err_info;
1055 
1056 	if (is_ue) {
1057 		for_each_ras_error(err_node, err_data) {
1058 			err_info = &err_node->err_info;
1059 			mcm_info = &err_info->mcm_info;
1060 			if (err_info->ue_count) {
1061 				dev_info(adev->dev, "socket: %d, die: %d, "
1062 					 "%lld new uncorrectable hardware errors detected in %s block\n",
1063 					 mcm_info->socket_id,
1064 					 mcm_info->die_id,
1065 					 err_info->ue_count,
1066 					 blk_name);
1067 			}
1068 		}
1069 
1070 		for_each_ras_error(err_node, &ras_mgr->err_data) {
1071 			err_info = &err_node->err_info;
1072 			mcm_info = &err_info->mcm_info;
1073 			dev_info(adev->dev, "socket: %d, die: %d, "
1074 				 "%lld uncorrectable hardware errors detected in total in %s block\n",
1075 				 mcm_info->socket_id, mcm_info->die_id, err_info->ue_count, blk_name);
1076 		}
1077 
1078 	} else {
1079 		if (is_de) {
1080 			for_each_ras_error(err_node, err_data) {
1081 				err_info = &err_node->err_info;
1082 				mcm_info = &err_info->mcm_info;
1083 				if (err_info->de_count) {
1084 					dev_info(adev->dev, "socket: %d, die: %d, "
1085 						"%lld new deferred hardware errors detected in %s block\n",
1086 						mcm_info->socket_id,
1087 						mcm_info->die_id,
1088 						err_info->de_count,
1089 						blk_name);
1090 				}
1091 			}
1092 
1093 			for_each_ras_error(err_node, &ras_mgr->err_data) {
1094 				err_info = &err_node->err_info;
1095 				mcm_info = &err_info->mcm_info;
1096 				dev_info(adev->dev, "socket: %d, die: %d, "
1097 					"%lld deferred hardware errors detected in total in %s block\n",
1098 					mcm_info->socket_id, mcm_info->die_id,
1099 					err_info->de_count, blk_name);
1100 			}
1101 		} else {
1102 			for_each_ras_error(err_node, err_data) {
1103 				err_info = &err_node->err_info;
1104 				mcm_info = &err_info->mcm_info;
1105 				if (err_info->ce_count) {
1106 					dev_info(adev->dev, "socket: %d, die: %d, "
1107 						"%lld new correctable hardware errors detected in %s block\n",
1108 						mcm_info->socket_id,
1109 						mcm_info->die_id,
1110 						err_info->ce_count,
1111 						blk_name);
1112 				}
1113 			}
1114 
1115 			for_each_ras_error(err_node, &ras_mgr->err_data) {
1116 				err_info = &err_node->err_info;
1117 				mcm_info = &err_info->mcm_info;
1118 				dev_info(adev->dev, "socket: %d, die: %d, "
1119 					"%lld correctable hardware errors detected in total in %s block\n",
1120 					mcm_info->socket_id, mcm_info->die_id,
1121 					err_info->ce_count, blk_name);
1122 			}
1123 		}
1124 	}
1125 }
1126 
1127 static inline bool err_data_has_source_info(struct ras_err_data *data)
1128 {
1129 	return !list_empty(&data->err_node_list);
1130 }
1131 
1132 static void amdgpu_ras_error_generate_report(struct amdgpu_device *adev,
1133 					     struct ras_query_if *query_if,
1134 					     struct ras_err_data *err_data)
1135 {
1136 	struct ras_manager *ras_mgr = amdgpu_ras_find_obj(adev, &query_if->head);
1137 	const char *blk_name = get_ras_block_str(&query_if->head);
1138 
1139 	if (err_data->ce_count) {
1140 		if (err_data_has_source_info(err_data)) {
1141 			amdgpu_ras_error_print_error_data(adev, ras_mgr, err_data,
1142 							  blk_name, false, false);
1143 		} else if (!adev->aid_mask &&
1144 			   adev->smuio.funcs &&
1145 			   adev->smuio.funcs->get_socket_id &&
1146 			   adev->smuio.funcs->get_die_id) {
1147 			dev_info(adev->dev, "socket: %d, die: %d "
1148 				 "%ld correctable hardware errors "
1149 				 "detected in %s block\n",
1150 				 adev->smuio.funcs->get_socket_id(adev),
1151 				 adev->smuio.funcs->get_die_id(adev),
1152 				 ras_mgr->err_data.ce_count,
1153 				 blk_name);
1154 		} else {
1155 			dev_info(adev->dev, "%ld correctable hardware errors "
1156 				 "detected in %s block\n",
1157 				 ras_mgr->err_data.ce_count,
1158 				 blk_name);
1159 		}
1160 	}
1161 
1162 	if (err_data->ue_count) {
1163 		if (err_data_has_source_info(err_data)) {
1164 			amdgpu_ras_error_print_error_data(adev, ras_mgr, err_data,
1165 							  blk_name, true, false);
1166 		} else if (!adev->aid_mask &&
1167 			   adev->smuio.funcs &&
1168 			   adev->smuio.funcs->get_socket_id &&
1169 			   adev->smuio.funcs->get_die_id) {
1170 			dev_info(adev->dev, "socket: %d, die: %d "
1171 				 "%ld uncorrectable hardware errors "
1172 				 "detected in %s block\n",
1173 				 adev->smuio.funcs->get_socket_id(adev),
1174 				 adev->smuio.funcs->get_die_id(adev),
1175 				 ras_mgr->err_data.ue_count,
1176 				 blk_name);
1177 		} else {
1178 			dev_info(adev->dev, "%ld uncorrectable hardware errors "
1179 				 "detected in %s block\n",
1180 				 ras_mgr->err_data.ue_count,
1181 				 blk_name);
1182 		}
1183 	}
1184 
1185 	if (err_data->de_count) {
1186 		if (err_data_has_source_info(err_data)) {
1187 			amdgpu_ras_error_print_error_data(adev, ras_mgr, err_data,
1188 							  blk_name, false, true);
1189 		} else if (!adev->aid_mask &&
1190 			   adev->smuio.funcs &&
1191 			   adev->smuio.funcs->get_socket_id &&
1192 			   adev->smuio.funcs->get_die_id) {
1193 			dev_info(adev->dev, "socket: %d, die: %d "
1194 				 "%ld deferred hardware errors "
1195 				 "detected in %s block\n",
1196 				 adev->smuio.funcs->get_socket_id(adev),
1197 				 adev->smuio.funcs->get_die_id(adev),
1198 				 ras_mgr->err_data.de_count,
1199 				 blk_name);
1200 		} else {
1201 			dev_info(adev->dev, "%ld deferred hardware errors "
1202 				 "detected in %s block\n",
1203 				 ras_mgr->err_data.de_count,
1204 				 blk_name);
1205 		}
1206 	}
1207 }
1208 
1209 static void amdgpu_rasmgr_error_data_statistic_update(struct ras_manager *obj, struct ras_err_data *err_data)
1210 {
1211 	struct ras_err_node *err_node;
1212 	struct ras_err_info *err_info;
1213 
1214 	if (err_data_has_source_info(err_data)) {
1215 		for_each_ras_error(err_node, err_data) {
1216 			err_info = &err_node->err_info;
1217 			amdgpu_ras_error_statistic_de_count(&obj->err_data,
1218 					&err_info->mcm_info, NULL, err_info->de_count);
1219 			amdgpu_ras_error_statistic_ce_count(&obj->err_data,
1220 					&err_info->mcm_info, NULL, err_info->ce_count);
1221 			amdgpu_ras_error_statistic_ue_count(&obj->err_data,
1222 					&err_info->mcm_info, NULL, err_info->ue_count);
1223 		}
1224 	} else {
1225 		/* for legacy asic path which doesn't has error source info */
1226 		obj->err_data.ue_count += err_data->ue_count;
1227 		obj->err_data.ce_count += err_data->ce_count;
1228 		obj->err_data.de_count += err_data->de_count;
1229 	}
1230 }
1231 
1232 static struct ras_manager *get_ras_manager(struct amdgpu_device *adev, enum amdgpu_ras_block blk)
1233 {
1234 	struct ras_common_if head;
1235 
1236 	memset(&head, 0, sizeof(head));
1237 	head.block = blk;
1238 
1239 	return amdgpu_ras_find_obj(adev, &head);
1240 }
1241 
1242 int amdgpu_ras_bind_aca(struct amdgpu_device *adev, enum amdgpu_ras_block blk,
1243 			const struct aca_info *aca_info, void *data)
1244 {
1245 	struct ras_manager *obj;
1246 
1247 	obj = get_ras_manager(adev, blk);
1248 	if (!obj)
1249 		return -EINVAL;
1250 
1251 	return amdgpu_aca_add_handle(adev, &obj->aca_handle, ras_block_str(blk), aca_info, data);
1252 }
1253 
1254 int amdgpu_ras_unbind_aca(struct amdgpu_device *adev, enum amdgpu_ras_block blk)
1255 {
1256 	struct ras_manager *obj;
1257 
1258 	obj = get_ras_manager(adev, blk);
1259 	if (!obj)
1260 		return -EINVAL;
1261 
1262 	amdgpu_aca_remove_handle(&obj->aca_handle);
1263 
1264 	return 0;
1265 }
1266 
1267 static int amdgpu_aca_log_ras_error_data(struct amdgpu_device *adev, enum amdgpu_ras_block blk,
1268 					 enum aca_error_type type, struct ras_err_data *err_data)
1269 {
1270 	struct ras_manager *obj;
1271 
1272 	obj = get_ras_manager(adev, blk);
1273 	if (!obj)
1274 		return -EINVAL;
1275 
1276 	return amdgpu_aca_get_error_data(adev, &obj->aca_handle, type, err_data);
1277 }
1278 
1279 ssize_t amdgpu_ras_aca_sysfs_read(struct device *dev, struct device_attribute *attr,
1280 				  struct aca_handle *handle, char *buf, void *data)
1281 {
1282 	struct ras_manager *obj = container_of(handle, struct ras_manager, aca_handle);
1283 	struct ras_query_if info = {
1284 		.head = obj->head,
1285 	};
1286 
1287 	if (amdgpu_ras_query_error_status(obj->adev, &info))
1288 		return -EINVAL;
1289 
1290 	return sysfs_emit(buf, "%s: %lu\n%s: %lu\n", "ue", info.ue_count,
1291 			  "ce", info.ce_count);
1292 }
1293 
1294 static int amdgpu_ras_query_error_status_helper(struct amdgpu_device *adev,
1295 						struct ras_query_if *info,
1296 						struct ras_err_data *err_data,
1297 						unsigned int error_query_mode)
1298 {
1299 	enum amdgpu_ras_block blk = info ? info->head.block : AMDGPU_RAS_BLOCK_COUNT;
1300 	struct amdgpu_ras_block_object *block_obj = NULL;
1301 	int ret;
1302 
1303 	if (blk == AMDGPU_RAS_BLOCK_COUNT)
1304 		return -EINVAL;
1305 
1306 	if (error_query_mode == AMDGPU_RAS_INVALID_ERROR_QUERY)
1307 		return -EINVAL;
1308 
1309 	if (error_query_mode == AMDGPU_RAS_DIRECT_ERROR_QUERY) {
1310 		if (info->head.block == AMDGPU_RAS_BLOCK__UMC) {
1311 			amdgpu_ras_get_ecc_info(adev, err_data);
1312 		} else {
1313 			block_obj = amdgpu_ras_get_ras_block(adev, info->head.block, 0);
1314 			if (!block_obj || !block_obj->hw_ops) {
1315 				dev_dbg_once(adev->dev, "%s doesn't config RAS function\n",
1316 					     get_ras_block_str(&info->head));
1317 				return -EINVAL;
1318 			}
1319 
1320 			if (block_obj->hw_ops->query_ras_error_count)
1321 				block_obj->hw_ops->query_ras_error_count(adev, err_data);
1322 
1323 			if ((info->head.block == AMDGPU_RAS_BLOCK__SDMA) ||
1324 			    (info->head.block == AMDGPU_RAS_BLOCK__GFX) ||
1325 			    (info->head.block == AMDGPU_RAS_BLOCK__MMHUB)) {
1326 				if (block_obj->hw_ops->query_ras_error_status)
1327 					block_obj->hw_ops->query_ras_error_status(adev);
1328 			}
1329 		}
1330 	} else {
1331 		if (amdgpu_aca_is_enabled(adev)) {
1332 			ret = amdgpu_aca_log_ras_error_data(adev, blk, ACA_ERROR_TYPE_UE, err_data);
1333 			if (ret)
1334 				return ret;
1335 
1336 			ret = amdgpu_aca_log_ras_error_data(adev, blk, ACA_ERROR_TYPE_CE, err_data);
1337 			if (ret)
1338 				return ret;
1339 		} else {
1340 			/* FIXME: add code to check return value later */
1341 			amdgpu_mca_smu_log_ras_error(adev, blk, AMDGPU_MCA_ERROR_TYPE_UE, err_data);
1342 			amdgpu_mca_smu_log_ras_error(adev, blk, AMDGPU_MCA_ERROR_TYPE_CE, err_data);
1343 		}
1344 	}
1345 
1346 	return 0;
1347 }
1348 
1349 /* query/inject/cure begin */
1350 int amdgpu_ras_query_error_status(struct amdgpu_device *adev, struct ras_query_if *info)
1351 {
1352 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, &info->head);
1353 	struct ras_err_data err_data;
1354 	unsigned int error_query_mode;
1355 	int ret;
1356 
1357 	if (!obj)
1358 		return -EINVAL;
1359 
1360 	ret = amdgpu_ras_error_data_init(&err_data);
1361 	if (ret)
1362 		return ret;
1363 
1364 	if (!amdgpu_ras_get_error_query_mode(adev, &error_query_mode))
1365 		return -EINVAL;
1366 
1367 	ret = amdgpu_ras_query_error_status_helper(adev, info,
1368 						   &err_data,
1369 						   error_query_mode);
1370 	if (ret)
1371 		goto out_fini_err_data;
1372 
1373 	amdgpu_rasmgr_error_data_statistic_update(obj, &err_data);
1374 
1375 	info->ue_count = obj->err_data.ue_count;
1376 	info->ce_count = obj->err_data.ce_count;
1377 	info->de_count = obj->err_data.de_count;
1378 
1379 	amdgpu_ras_error_generate_report(adev, info, &err_data);
1380 
1381 out_fini_err_data:
1382 	amdgpu_ras_error_data_fini(&err_data);
1383 
1384 	return ret;
1385 }
1386 
1387 int amdgpu_ras_reset_error_count(struct amdgpu_device *adev,
1388 		enum amdgpu_ras_block block)
1389 {
1390 	struct amdgpu_ras_block_object *block_obj = amdgpu_ras_get_ras_block(adev, block, 0);
1391 	struct amdgpu_ras *ras = amdgpu_ras_get_context(adev);
1392 	const struct amdgpu_mca_smu_funcs *mca_funcs = adev->mca.mca_funcs;
1393 	const struct aca_smu_funcs *smu_funcs = adev->aca.smu_funcs;
1394 	struct amdgpu_hive_info *hive;
1395 	int hive_ras_recovery = 0;
1396 
1397 	if (!block_obj || !block_obj->hw_ops) {
1398 		dev_dbg_once(adev->dev, "%s doesn't config RAS function\n",
1399 				ras_block_str(block));
1400 		return -EOPNOTSUPP;
1401 	}
1402 
1403 	if (!amdgpu_ras_is_supported(adev, block) ||
1404 	    !amdgpu_ras_get_aca_debug_mode(adev))
1405 		return -EOPNOTSUPP;
1406 
1407 	hive = amdgpu_get_xgmi_hive(adev);
1408 	if (hive) {
1409 		hive_ras_recovery = atomic_read(&hive->ras_recovery);
1410 		amdgpu_put_xgmi_hive(hive);
1411 	}
1412 
1413 	/* skip ras error reset in gpu reset */
1414 	if ((amdgpu_in_reset(adev) || atomic_read(&ras->in_recovery) ||
1415 	    hive_ras_recovery) &&
1416 	    ((smu_funcs && smu_funcs->set_debug_mode) ||
1417 	     (mca_funcs && mca_funcs->mca_set_debug_mode)))
1418 		return -EOPNOTSUPP;
1419 
1420 	if (block_obj->hw_ops->reset_ras_error_count)
1421 		block_obj->hw_ops->reset_ras_error_count(adev);
1422 
1423 	return 0;
1424 }
1425 
1426 int amdgpu_ras_reset_error_status(struct amdgpu_device *adev,
1427 		enum amdgpu_ras_block block)
1428 {
1429 	struct amdgpu_ras_block_object *block_obj = amdgpu_ras_get_ras_block(adev, block, 0);
1430 
1431 	if (amdgpu_ras_reset_error_count(adev, block) == -EOPNOTSUPP)
1432 		return 0;
1433 
1434 	if ((block == AMDGPU_RAS_BLOCK__GFX) ||
1435 	    (block == AMDGPU_RAS_BLOCK__MMHUB)) {
1436 		if (block_obj->hw_ops->reset_ras_error_status)
1437 			block_obj->hw_ops->reset_ras_error_status(adev);
1438 	}
1439 
1440 	return 0;
1441 }
1442 
1443 /* wrapper of psp_ras_trigger_error */
1444 int amdgpu_ras_error_inject(struct amdgpu_device *adev,
1445 		struct ras_inject_if *info)
1446 {
1447 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, &info->head);
1448 	struct ta_ras_trigger_error_input block_info = {
1449 		.block_id =  amdgpu_ras_block_to_ta(info->head.block),
1450 		.inject_error_type = amdgpu_ras_error_to_ta(info->head.type),
1451 		.sub_block_index = info->head.sub_block_index,
1452 		.address = info->address,
1453 		.value = info->value,
1454 	};
1455 	int ret = -EINVAL;
1456 	struct amdgpu_ras_block_object *block_obj = amdgpu_ras_get_ras_block(adev,
1457 							info->head.block,
1458 							info->head.sub_block_index);
1459 
1460 	/* inject on guest isn't allowed, return success directly */
1461 	if (amdgpu_sriov_vf(adev))
1462 		return 0;
1463 
1464 	if (!obj)
1465 		return -EINVAL;
1466 
1467 	if (!block_obj || !block_obj->hw_ops)	{
1468 		dev_dbg_once(adev->dev, "%s doesn't config RAS function\n",
1469 			     get_ras_block_str(&info->head));
1470 		return -EINVAL;
1471 	}
1472 
1473 	/* Calculate XGMI relative offset */
1474 	if (adev->gmc.xgmi.num_physical_nodes > 1 &&
1475 	    info->head.block != AMDGPU_RAS_BLOCK__GFX) {
1476 		block_info.address =
1477 			amdgpu_xgmi_get_relative_phy_addr(adev,
1478 							  block_info.address);
1479 	}
1480 
1481 	if (block_obj->hw_ops->ras_error_inject) {
1482 		if (info->head.block == AMDGPU_RAS_BLOCK__GFX)
1483 			ret = block_obj->hw_ops->ras_error_inject(adev, info, info->instance_mask);
1484 		else /* Special ras_error_inject is defined (e.g: xgmi) */
1485 			ret = block_obj->hw_ops->ras_error_inject(adev, &block_info,
1486 						info->instance_mask);
1487 	} else {
1488 		/* default path */
1489 		ret = psp_ras_trigger_error(&adev->psp, &block_info, info->instance_mask);
1490 	}
1491 
1492 	if (ret)
1493 		dev_err(adev->dev, "ras inject %s failed %d\n",
1494 			get_ras_block_str(&info->head), ret);
1495 
1496 	return ret;
1497 }
1498 
1499 /**
1500  * amdgpu_ras_query_error_count_helper -- Get error counter for specific IP
1501  * @adev: pointer to AMD GPU device
1502  * @ce_count: pointer to an integer to be set to the count of correctible errors.
1503  * @ue_count: pointer to an integer to be set to the count of uncorrectible errors.
1504  * @query_info: pointer to ras_query_if
1505  *
1506  * Return 0 for query success or do nothing, otherwise return an error
1507  * on failures
1508  */
1509 static int amdgpu_ras_query_error_count_helper(struct amdgpu_device *adev,
1510 					       unsigned long *ce_count,
1511 					       unsigned long *ue_count,
1512 					       struct ras_query_if *query_info)
1513 {
1514 	int ret;
1515 
1516 	if (!query_info)
1517 		/* do nothing if query_info is not specified */
1518 		return 0;
1519 
1520 	ret = amdgpu_ras_query_error_status(adev, query_info);
1521 	if (ret)
1522 		return ret;
1523 
1524 	*ce_count += query_info->ce_count;
1525 	*ue_count += query_info->ue_count;
1526 
1527 	/* some hardware/IP supports read to clear
1528 	 * no need to explictly reset the err status after the query call */
1529 	if (amdgpu_ip_version(adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 2) &&
1530 	    amdgpu_ip_version(adev, MP0_HWIP, 0) != IP_VERSION(11, 0, 4)) {
1531 		if (amdgpu_ras_reset_error_status(adev, query_info->head.block))
1532 			dev_warn(adev->dev,
1533 				 "Failed to reset error counter and error status\n");
1534 	}
1535 
1536 	return 0;
1537 }
1538 
1539 /**
1540  * amdgpu_ras_query_error_count -- Get error counts of all IPs or specific IP
1541  * @adev: pointer to AMD GPU device
1542  * @ce_count: pointer to an integer to be set to the count of correctible errors.
1543  * @ue_count: pointer to an integer to be set to the count of uncorrectible
1544  * errors.
1545  * @query_info: pointer to ras_query_if if the query request is only for
1546  * specific ip block; if info is NULL, then the qurey request is for
1547  * all the ip blocks that support query ras error counters/status
1548  *
1549  * If set, @ce_count or @ue_count, count and return the corresponding
1550  * error counts in those integer pointers. Return 0 if the device
1551  * supports RAS. Return -EOPNOTSUPP if the device doesn't support RAS.
1552  */
1553 int amdgpu_ras_query_error_count(struct amdgpu_device *adev,
1554 				 unsigned long *ce_count,
1555 				 unsigned long *ue_count,
1556 				 struct ras_query_if *query_info)
1557 {
1558 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1559 	struct ras_manager *obj;
1560 	unsigned long ce, ue;
1561 	int ret;
1562 
1563 	if (!adev->ras_enabled || !con)
1564 		return -EOPNOTSUPP;
1565 
1566 	/* Don't count since no reporting.
1567 	 */
1568 	if (!ce_count && !ue_count)
1569 		return 0;
1570 
1571 	ce = 0;
1572 	ue = 0;
1573 	if (!query_info) {
1574 		/* query all the ip blocks that support ras query interface */
1575 		list_for_each_entry(obj, &con->head, node) {
1576 			struct ras_query_if info = {
1577 				.head = obj->head,
1578 			};
1579 
1580 			ret = amdgpu_ras_query_error_count_helper(adev, &ce, &ue, &info);
1581 		}
1582 	} else {
1583 		/* query specific ip block */
1584 		ret = amdgpu_ras_query_error_count_helper(adev, &ce, &ue, query_info);
1585 	}
1586 
1587 	if (ret)
1588 		return ret;
1589 
1590 	if (ce_count)
1591 		*ce_count = ce;
1592 
1593 	if (ue_count)
1594 		*ue_count = ue;
1595 
1596 	return 0;
1597 }
1598 /* query/inject/cure end */
1599 
1600 
1601 /* sysfs begin */
1602 
1603 static int amdgpu_ras_badpages_read(struct amdgpu_device *adev,
1604 		struct ras_badpage **bps, unsigned int *count);
1605 
1606 static char *amdgpu_ras_badpage_flags_str(unsigned int flags)
1607 {
1608 	switch (flags) {
1609 	case AMDGPU_RAS_RETIRE_PAGE_RESERVED:
1610 		return "R";
1611 	case AMDGPU_RAS_RETIRE_PAGE_PENDING:
1612 		return "P";
1613 	case AMDGPU_RAS_RETIRE_PAGE_FAULT:
1614 	default:
1615 		return "F";
1616 	}
1617 }
1618 
1619 /**
1620  * DOC: AMDGPU RAS sysfs gpu_vram_bad_pages Interface
1621  *
1622  * It allows user to read the bad pages of vram on the gpu through
1623  * /sys/class/drm/card[0/1/2...]/device/ras/gpu_vram_bad_pages
1624  *
1625  * It outputs multiple lines, and each line stands for one gpu page.
1626  *
1627  * The format of one line is below,
1628  * gpu pfn : gpu page size : flags
1629  *
1630  * gpu pfn and gpu page size are printed in hex format.
1631  * flags can be one of below character,
1632  *
1633  * R: reserved, this gpu page is reserved and not able to use.
1634  *
1635  * P: pending for reserve, this gpu page is marked as bad, will be reserved
1636  * in next window of page_reserve.
1637  *
1638  * F: unable to reserve. this gpu page can't be reserved due to some reasons.
1639  *
1640  * Examples:
1641  *
1642  * .. code-block:: bash
1643  *
1644  *	0x00000001 : 0x00001000 : R
1645  *	0x00000002 : 0x00001000 : P
1646  *
1647  */
1648 
1649 static ssize_t amdgpu_ras_sysfs_badpages_read(struct file *f,
1650 		struct kobject *kobj, struct bin_attribute *attr,
1651 		char *buf, loff_t ppos, size_t count)
1652 {
1653 	struct amdgpu_ras *con =
1654 		container_of(attr, struct amdgpu_ras, badpages_attr);
1655 	struct amdgpu_device *adev = con->adev;
1656 	const unsigned int element_size =
1657 		sizeof("0xabcdabcd : 0x12345678 : R\n") - 1;
1658 	unsigned int start = div64_ul(ppos + element_size - 1, element_size);
1659 	unsigned int end = div64_ul(ppos + count - 1, element_size);
1660 	ssize_t s = 0;
1661 	struct ras_badpage *bps = NULL;
1662 	unsigned int bps_count = 0;
1663 
1664 	memset(buf, 0, count);
1665 
1666 	if (amdgpu_ras_badpages_read(adev, &bps, &bps_count))
1667 		return 0;
1668 
1669 	for (; start < end && start < bps_count; start++)
1670 		s += scnprintf(&buf[s], element_size + 1,
1671 				"0x%08x : 0x%08x : %1s\n",
1672 				bps[start].bp,
1673 				bps[start].size,
1674 				amdgpu_ras_badpage_flags_str(bps[start].flags));
1675 
1676 	kfree(bps);
1677 
1678 	return s;
1679 }
1680 
1681 static ssize_t amdgpu_ras_sysfs_features_read(struct device *dev,
1682 		struct device_attribute *attr, char *buf)
1683 {
1684 	struct amdgpu_ras *con =
1685 		container_of(attr, struct amdgpu_ras, features_attr);
1686 
1687 	return sysfs_emit(buf, "feature mask: 0x%x\n", con->features);
1688 }
1689 
1690 static ssize_t amdgpu_ras_sysfs_version_show(struct device *dev,
1691 		struct device_attribute *attr, char *buf)
1692 {
1693 	struct amdgpu_ras *con =
1694 		container_of(attr, struct amdgpu_ras, version_attr);
1695 	return sysfs_emit(buf, "table version: 0x%x\n", con->eeprom_control.tbl_hdr.version);
1696 }
1697 
1698 static ssize_t amdgpu_ras_sysfs_schema_show(struct device *dev,
1699 		struct device_attribute *attr, char *buf)
1700 {
1701 	struct amdgpu_ras *con =
1702 		container_of(attr, struct amdgpu_ras, schema_attr);
1703 	return sysfs_emit(buf, "schema: 0x%x\n", con->schema);
1704 }
1705 
1706 static void amdgpu_ras_sysfs_remove_bad_page_node(struct amdgpu_device *adev)
1707 {
1708 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1709 
1710 	if (adev->dev->kobj.sd)
1711 		sysfs_remove_file_from_group(&adev->dev->kobj,
1712 				&con->badpages_attr.attr,
1713 				RAS_FS_NAME);
1714 }
1715 
1716 static int amdgpu_ras_sysfs_remove_dev_attr_node(struct amdgpu_device *adev)
1717 {
1718 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1719 	struct attribute *attrs[] = {
1720 		&con->features_attr.attr,
1721 		&con->version_attr.attr,
1722 		&con->schema_attr.attr,
1723 		NULL
1724 	};
1725 	struct attribute_group group = {
1726 		.name = RAS_FS_NAME,
1727 		.attrs = attrs,
1728 	};
1729 
1730 	if (adev->dev->kobj.sd)
1731 		sysfs_remove_group(&adev->dev->kobj, &group);
1732 
1733 	return 0;
1734 }
1735 
1736 int amdgpu_ras_sysfs_create(struct amdgpu_device *adev,
1737 		struct ras_common_if *head)
1738 {
1739 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, head);
1740 
1741 	if (!obj || obj->attr_inuse)
1742 		return -EINVAL;
1743 
1744 	get_obj(obj);
1745 
1746 	snprintf(obj->fs_data.sysfs_name, sizeof(obj->fs_data.sysfs_name),
1747 		"%s_err_count", head->name);
1748 
1749 	obj->sysfs_attr = (struct device_attribute){
1750 		.attr = {
1751 			.name = obj->fs_data.sysfs_name,
1752 			.mode = S_IRUGO,
1753 		},
1754 			.show = amdgpu_ras_sysfs_read,
1755 	};
1756 	sysfs_attr_init(&obj->sysfs_attr.attr);
1757 
1758 	if (sysfs_add_file_to_group(&adev->dev->kobj,
1759 				&obj->sysfs_attr.attr,
1760 				RAS_FS_NAME)) {
1761 		put_obj(obj);
1762 		return -EINVAL;
1763 	}
1764 
1765 	obj->attr_inuse = 1;
1766 
1767 	return 0;
1768 }
1769 
1770 int amdgpu_ras_sysfs_remove(struct amdgpu_device *adev,
1771 		struct ras_common_if *head)
1772 {
1773 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, head);
1774 
1775 	if (!obj || !obj->attr_inuse)
1776 		return -EINVAL;
1777 
1778 	if (adev->dev->kobj.sd)
1779 		sysfs_remove_file_from_group(&adev->dev->kobj,
1780 				&obj->sysfs_attr.attr,
1781 				RAS_FS_NAME);
1782 	obj->attr_inuse = 0;
1783 	put_obj(obj);
1784 
1785 	return 0;
1786 }
1787 
1788 static int amdgpu_ras_sysfs_remove_all(struct amdgpu_device *adev)
1789 {
1790 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1791 	struct ras_manager *obj, *tmp;
1792 
1793 	list_for_each_entry_safe(obj, tmp, &con->head, node) {
1794 		amdgpu_ras_sysfs_remove(adev, &obj->head);
1795 	}
1796 
1797 	if (amdgpu_bad_page_threshold != 0)
1798 		amdgpu_ras_sysfs_remove_bad_page_node(adev);
1799 
1800 	amdgpu_ras_sysfs_remove_dev_attr_node(adev);
1801 
1802 	return 0;
1803 }
1804 /* sysfs end */
1805 
1806 /**
1807  * DOC: AMDGPU RAS Reboot Behavior for Unrecoverable Errors
1808  *
1809  * Normally when there is an uncorrectable error, the driver will reset
1810  * the GPU to recover.  However, in the event of an unrecoverable error,
1811  * the driver provides an interface to reboot the system automatically
1812  * in that event.
1813  *
1814  * The following file in debugfs provides that interface:
1815  * /sys/kernel/debug/dri/[0/1/2...]/ras/auto_reboot
1816  *
1817  * Usage:
1818  *
1819  * .. code-block:: bash
1820  *
1821  *	echo true > .../ras/auto_reboot
1822  *
1823  */
1824 /* debugfs begin */
1825 static struct dentry *amdgpu_ras_debugfs_create_ctrl_node(struct amdgpu_device *adev)
1826 {
1827 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1828 	struct amdgpu_ras_eeprom_control *eeprom = &con->eeprom_control;
1829 	struct drm_minor  *minor = adev_to_drm(adev)->primary;
1830 	struct dentry     *dir;
1831 
1832 	dir = debugfs_create_dir(RAS_FS_NAME, minor->debugfs_root);
1833 	debugfs_create_file("ras_ctrl", S_IWUGO | S_IRUGO, dir, adev,
1834 			    &amdgpu_ras_debugfs_ctrl_ops);
1835 	debugfs_create_file("ras_eeprom_reset", S_IWUGO | S_IRUGO, dir, adev,
1836 			    &amdgpu_ras_debugfs_eeprom_ops);
1837 	debugfs_create_u32("bad_page_cnt_threshold", 0444, dir,
1838 			   &con->bad_page_cnt_threshold);
1839 	debugfs_create_u32("ras_num_recs", 0444, dir, &eeprom->ras_num_recs);
1840 	debugfs_create_x32("ras_hw_enabled", 0444, dir, &adev->ras_hw_enabled);
1841 	debugfs_create_x32("ras_enabled", 0444, dir, &adev->ras_enabled);
1842 	debugfs_create_file("ras_eeprom_size", S_IRUGO, dir, adev,
1843 			    &amdgpu_ras_debugfs_eeprom_size_ops);
1844 	con->de_ras_eeprom_table = debugfs_create_file("ras_eeprom_table",
1845 						       S_IRUGO, dir, adev,
1846 						       &amdgpu_ras_debugfs_eeprom_table_ops);
1847 	amdgpu_ras_debugfs_set_ret_size(&con->eeprom_control);
1848 
1849 	/*
1850 	 * After one uncorrectable error happens, usually GPU recovery will
1851 	 * be scheduled. But due to the known problem in GPU recovery failing
1852 	 * to bring GPU back, below interface provides one direct way to
1853 	 * user to reboot system automatically in such case within
1854 	 * ERREVENT_ATHUB_INTERRUPT generated. Normal GPU recovery routine
1855 	 * will never be called.
1856 	 */
1857 	debugfs_create_bool("auto_reboot", S_IWUGO | S_IRUGO, dir, &con->reboot);
1858 
1859 	/*
1860 	 * User could set this not to clean up hardware's error count register
1861 	 * of RAS IPs during ras recovery.
1862 	 */
1863 	debugfs_create_bool("disable_ras_err_cnt_harvest", 0644, dir,
1864 			    &con->disable_ras_err_cnt_harvest);
1865 	return dir;
1866 }
1867 
1868 static void amdgpu_ras_debugfs_create(struct amdgpu_device *adev,
1869 				      struct ras_fs_if *head,
1870 				      struct dentry *dir)
1871 {
1872 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, &head->head);
1873 
1874 	if (!obj || !dir)
1875 		return;
1876 
1877 	get_obj(obj);
1878 
1879 	memcpy(obj->fs_data.debugfs_name,
1880 			head->debugfs_name,
1881 			sizeof(obj->fs_data.debugfs_name));
1882 
1883 	debugfs_create_file(obj->fs_data.debugfs_name, S_IWUGO | S_IRUGO, dir,
1884 			    obj, &amdgpu_ras_debugfs_ops);
1885 }
1886 
1887 void amdgpu_ras_debugfs_create_all(struct amdgpu_device *adev)
1888 {
1889 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1890 	struct dentry *dir;
1891 	struct ras_manager *obj;
1892 	struct ras_fs_if fs_info;
1893 
1894 	/*
1895 	 * it won't be called in resume path, no need to check
1896 	 * suspend and gpu reset status
1897 	 */
1898 	if (!IS_ENABLED(CONFIG_DEBUG_FS) || !con)
1899 		return;
1900 
1901 	dir = amdgpu_ras_debugfs_create_ctrl_node(adev);
1902 
1903 	list_for_each_entry(obj, &con->head, node) {
1904 		if (amdgpu_ras_is_supported(adev, obj->head.block) &&
1905 			(obj->attr_inuse == 1)) {
1906 			sprintf(fs_info.debugfs_name, "%s_err_inject",
1907 					get_ras_block_str(&obj->head));
1908 			fs_info.head = obj->head;
1909 			amdgpu_ras_debugfs_create(adev, &fs_info, dir);
1910 		}
1911 	}
1912 
1913 	if (amdgpu_aca_is_enabled(adev))
1914 		amdgpu_aca_smu_debugfs_init(adev, dir);
1915 	else
1916 		amdgpu_mca_smu_debugfs_init(adev, dir);
1917 }
1918 
1919 /* debugfs end */
1920 
1921 /* ras fs */
1922 static BIN_ATTR(gpu_vram_bad_pages, S_IRUGO,
1923 		amdgpu_ras_sysfs_badpages_read, NULL, 0);
1924 static DEVICE_ATTR(features, S_IRUGO,
1925 		amdgpu_ras_sysfs_features_read, NULL);
1926 static DEVICE_ATTR(version, 0444,
1927 		amdgpu_ras_sysfs_version_show, NULL);
1928 static DEVICE_ATTR(schema, 0444,
1929 		amdgpu_ras_sysfs_schema_show, NULL);
1930 static int amdgpu_ras_fs_init(struct amdgpu_device *adev)
1931 {
1932 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1933 	struct attribute_group group = {
1934 		.name = RAS_FS_NAME,
1935 	};
1936 	struct attribute *attrs[] = {
1937 		&con->features_attr.attr,
1938 		&con->version_attr.attr,
1939 		&con->schema_attr.attr,
1940 		NULL
1941 	};
1942 	struct bin_attribute *bin_attrs[] = {
1943 		NULL,
1944 		NULL,
1945 	};
1946 	int r;
1947 
1948 	group.attrs = attrs;
1949 
1950 	/* add features entry */
1951 	con->features_attr = dev_attr_features;
1952 	sysfs_attr_init(attrs[0]);
1953 
1954 	/* add version entry */
1955 	con->version_attr = dev_attr_version;
1956 	sysfs_attr_init(attrs[1]);
1957 
1958 	/* add schema entry */
1959 	con->schema_attr = dev_attr_schema;
1960 	sysfs_attr_init(attrs[2]);
1961 
1962 	if (amdgpu_bad_page_threshold != 0) {
1963 		/* add bad_page_features entry */
1964 		bin_attr_gpu_vram_bad_pages.private = NULL;
1965 		con->badpages_attr = bin_attr_gpu_vram_bad_pages;
1966 		bin_attrs[0] = &con->badpages_attr;
1967 		group.bin_attrs = bin_attrs;
1968 		sysfs_bin_attr_init(bin_attrs[0]);
1969 	}
1970 
1971 	r = sysfs_create_group(&adev->dev->kobj, &group);
1972 	if (r)
1973 		dev_err(adev->dev, "Failed to create RAS sysfs group!");
1974 
1975 	return 0;
1976 }
1977 
1978 static int amdgpu_ras_fs_fini(struct amdgpu_device *adev)
1979 {
1980 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
1981 	struct ras_manager *con_obj, *ip_obj, *tmp;
1982 
1983 	if (IS_ENABLED(CONFIG_DEBUG_FS)) {
1984 		list_for_each_entry_safe(con_obj, tmp, &con->head, node) {
1985 			ip_obj = amdgpu_ras_find_obj(adev, &con_obj->head);
1986 			if (ip_obj)
1987 				put_obj(ip_obj);
1988 		}
1989 	}
1990 
1991 	amdgpu_ras_sysfs_remove_all(adev);
1992 	return 0;
1993 }
1994 /* ras fs end */
1995 
1996 /* ih begin */
1997 
1998 /* For the hardware that cannot enable bif ring for both ras_controller_irq
1999  * and ras_err_evnet_athub_irq ih cookies, the driver has to poll status
2000  * register to check whether the interrupt is triggered or not, and properly
2001  * ack the interrupt if it is there
2002  */
2003 void amdgpu_ras_interrupt_fatal_error_handler(struct amdgpu_device *adev)
2004 {
2005 	/* Fatal error events are handled on host side */
2006 	if (amdgpu_sriov_vf(adev))
2007 		return;
2008 
2009 	if (adev->nbio.ras &&
2010 	    adev->nbio.ras->handle_ras_controller_intr_no_bifring)
2011 		adev->nbio.ras->handle_ras_controller_intr_no_bifring(adev);
2012 
2013 	if (adev->nbio.ras &&
2014 	    adev->nbio.ras->handle_ras_err_event_athub_intr_no_bifring)
2015 		adev->nbio.ras->handle_ras_err_event_athub_intr_no_bifring(adev);
2016 }
2017 
2018 static void amdgpu_ras_interrupt_poison_consumption_handler(struct ras_manager *obj,
2019 				struct amdgpu_iv_entry *entry)
2020 {
2021 	bool poison_stat = false;
2022 	struct amdgpu_device *adev = obj->adev;
2023 	struct amdgpu_ras_block_object *block_obj =
2024 		amdgpu_ras_get_ras_block(adev, obj->head.block, 0);
2025 
2026 	if (!block_obj)
2027 		return;
2028 
2029 	/* both query_poison_status and handle_poison_consumption are optional,
2030 	 * but at least one of them should be implemented if we need poison
2031 	 * consumption handler
2032 	 */
2033 	if (block_obj->hw_ops && block_obj->hw_ops->query_poison_status) {
2034 		poison_stat = block_obj->hw_ops->query_poison_status(adev);
2035 		if (!poison_stat) {
2036 			/* Not poison consumption interrupt, no need to handle it */
2037 			dev_info(adev->dev, "No RAS poison status in %s poison IH.\n",
2038 					block_obj->ras_comm.name);
2039 
2040 			return;
2041 		}
2042 	}
2043 
2044 	amdgpu_umc_poison_handler(adev, obj->head.block, false);
2045 
2046 	if (block_obj->hw_ops && block_obj->hw_ops->handle_poison_consumption)
2047 		poison_stat = block_obj->hw_ops->handle_poison_consumption(adev);
2048 
2049 	/* gpu reset is fallback for failed and default cases */
2050 	if (poison_stat) {
2051 		dev_info(adev->dev, "GPU reset for %s RAS poison consumption is issued!\n",
2052 				block_obj->ras_comm.name);
2053 		amdgpu_ras_reset_gpu(adev);
2054 	} else {
2055 		amdgpu_gfx_poison_consumption_handler(adev, entry);
2056 	}
2057 }
2058 
2059 static void amdgpu_ras_interrupt_poison_creation_handler(struct ras_manager *obj,
2060 				struct amdgpu_iv_entry *entry)
2061 {
2062 	dev_info(obj->adev->dev,
2063 		"Poison is created\n");
2064 }
2065 
2066 static void amdgpu_ras_interrupt_umc_handler(struct ras_manager *obj,
2067 				struct amdgpu_iv_entry *entry)
2068 {
2069 	struct ras_ih_data *data = &obj->ih_data;
2070 	struct ras_err_data err_data;
2071 	int ret;
2072 
2073 	if (!data->cb)
2074 		return;
2075 
2076 	ret = amdgpu_ras_error_data_init(&err_data);
2077 	if (ret)
2078 		return;
2079 
2080 	/* Let IP handle its data, maybe we need get the output
2081 	 * from the callback to update the error type/count, etc
2082 	 */
2083 	ret = data->cb(obj->adev, &err_data, entry);
2084 	/* ue will trigger an interrupt, and in that case
2085 	 * we need do a reset to recovery the whole system.
2086 	 * But leave IP do that recovery, here we just dispatch
2087 	 * the error.
2088 	 */
2089 	if (ret == AMDGPU_RAS_SUCCESS) {
2090 		/* these counts could be left as 0 if
2091 		 * some blocks do not count error number
2092 		 */
2093 		obj->err_data.ue_count += err_data.ue_count;
2094 		obj->err_data.ce_count += err_data.ce_count;
2095 		obj->err_data.de_count += err_data.de_count;
2096 	}
2097 
2098 	amdgpu_ras_error_data_fini(&err_data);
2099 }
2100 
2101 static void amdgpu_ras_interrupt_handler(struct ras_manager *obj)
2102 {
2103 	struct ras_ih_data *data = &obj->ih_data;
2104 	struct amdgpu_iv_entry entry;
2105 
2106 	while (data->rptr != data->wptr) {
2107 		rmb();
2108 		memcpy(&entry, &data->ring[data->rptr],
2109 				data->element_size);
2110 
2111 		wmb();
2112 		data->rptr = (data->aligned_element_size +
2113 				data->rptr) % data->ring_size;
2114 
2115 		if (amdgpu_ras_is_poison_mode_supported(obj->adev)) {
2116 			if (obj->head.block == AMDGPU_RAS_BLOCK__UMC)
2117 				amdgpu_ras_interrupt_poison_creation_handler(obj, &entry);
2118 			else
2119 				amdgpu_ras_interrupt_poison_consumption_handler(obj, &entry);
2120 		} else {
2121 			if (obj->head.block == AMDGPU_RAS_BLOCK__UMC)
2122 				amdgpu_ras_interrupt_umc_handler(obj, &entry);
2123 			else
2124 				dev_warn(obj->adev->dev,
2125 					"No RAS interrupt handler for non-UMC block with poison disabled.\n");
2126 		}
2127 	}
2128 }
2129 
2130 static void amdgpu_ras_interrupt_process_handler(struct work_struct *work)
2131 {
2132 	struct ras_ih_data *data =
2133 		container_of(work, struct ras_ih_data, ih_work);
2134 	struct ras_manager *obj =
2135 		container_of(data, struct ras_manager, ih_data);
2136 
2137 	amdgpu_ras_interrupt_handler(obj);
2138 }
2139 
2140 int amdgpu_ras_interrupt_dispatch(struct amdgpu_device *adev,
2141 		struct ras_dispatch_if *info)
2142 {
2143 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, &info->head);
2144 	struct ras_ih_data *data = &obj->ih_data;
2145 
2146 	if (!obj)
2147 		return -EINVAL;
2148 
2149 	if (data->inuse == 0)
2150 		return 0;
2151 
2152 	/* Might be overflow... */
2153 	memcpy(&data->ring[data->wptr], info->entry,
2154 			data->element_size);
2155 
2156 	wmb();
2157 	data->wptr = (data->aligned_element_size +
2158 			data->wptr) % data->ring_size;
2159 
2160 	schedule_work(&data->ih_work);
2161 
2162 	return 0;
2163 }
2164 
2165 int amdgpu_ras_interrupt_remove_handler(struct amdgpu_device *adev,
2166 		struct ras_common_if *head)
2167 {
2168 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, head);
2169 	struct ras_ih_data *data;
2170 
2171 	if (!obj)
2172 		return -EINVAL;
2173 
2174 	data = &obj->ih_data;
2175 	if (data->inuse == 0)
2176 		return 0;
2177 
2178 	cancel_work_sync(&data->ih_work);
2179 
2180 	kfree(data->ring);
2181 	memset(data, 0, sizeof(*data));
2182 	put_obj(obj);
2183 
2184 	return 0;
2185 }
2186 
2187 int amdgpu_ras_interrupt_add_handler(struct amdgpu_device *adev,
2188 		struct ras_common_if *head)
2189 {
2190 	struct ras_manager *obj = amdgpu_ras_find_obj(adev, head);
2191 	struct ras_ih_data *data;
2192 	struct amdgpu_ras_block_object *ras_obj;
2193 
2194 	if (!obj) {
2195 		/* in case we registe the IH before enable ras feature */
2196 		obj = amdgpu_ras_create_obj(adev, head);
2197 		if (!obj)
2198 			return -EINVAL;
2199 	} else
2200 		get_obj(obj);
2201 
2202 	ras_obj = container_of(head, struct amdgpu_ras_block_object, ras_comm);
2203 
2204 	data = &obj->ih_data;
2205 	/* add the callback.etc */
2206 	*data = (struct ras_ih_data) {
2207 		.inuse = 0,
2208 		.cb = ras_obj->ras_cb,
2209 		.element_size = sizeof(struct amdgpu_iv_entry),
2210 		.rptr = 0,
2211 		.wptr = 0,
2212 	};
2213 
2214 	INIT_WORK(&data->ih_work, amdgpu_ras_interrupt_process_handler);
2215 
2216 	data->aligned_element_size = ALIGN(data->element_size, 8);
2217 	/* the ring can store 64 iv entries. */
2218 	data->ring_size = 64 * data->aligned_element_size;
2219 	data->ring = kmalloc(data->ring_size, GFP_KERNEL);
2220 	if (!data->ring) {
2221 		put_obj(obj);
2222 		return -ENOMEM;
2223 	}
2224 
2225 	/* IH is ready */
2226 	data->inuse = 1;
2227 
2228 	return 0;
2229 }
2230 
2231 static int amdgpu_ras_interrupt_remove_all(struct amdgpu_device *adev)
2232 {
2233 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2234 	struct ras_manager *obj, *tmp;
2235 
2236 	list_for_each_entry_safe(obj, tmp, &con->head, node) {
2237 		amdgpu_ras_interrupt_remove_handler(adev, &obj->head);
2238 	}
2239 
2240 	return 0;
2241 }
2242 /* ih end */
2243 
2244 /* traversal all IPs except NBIO to query error counter */
2245 static void amdgpu_ras_log_on_err_counter(struct amdgpu_device *adev)
2246 {
2247 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2248 	struct ras_manager *obj;
2249 
2250 	if (!adev->ras_enabled || !con)
2251 		return;
2252 
2253 	list_for_each_entry(obj, &con->head, node) {
2254 		struct ras_query_if info = {
2255 			.head = obj->head,
2256 		};
2257 
2258 		/*
2259 		 * PCIE_BIF IP has one different isr by ras controller
2260 		 * interrupt, the specific ras counter query will be
2261 		 * done in that isr. So skip such block from common
2262 		 * sync flood interrupt isr calling.
2263 		 */
2264 		if (info.head.block == AMDGPU_RAS_BLOCK__PCIE_BIF)
2265 			continue;
2266 
2267 		/*
2268 		 * this is a workaround for aldebaran, skip send msg to
2269 		 * smu to get ecc_info table due to smu handle get ecc
2270 		 * info table failed temporarily.
2271 		 * should be removed until smu fix handle ecc_info table.
2272 		 */
2273 		if ((info.head.block == AMDGPU_RAS_BLOCK__UMC) &&
2274 		    (amdgpu_ip_version(adev, MP1_HWIP, 0) ==
2275 		     IP_VERSION(13, 0, 2)))
2276 			continue;
2277 
2278 		amdgpu_ras_query_error_status(adev, &info);
2279 
2280 		if (amdgpu_ip_version(adev, MP0_HWIP, 0) !=
2281 			    IP_VERSION(11, 0, 2) &&
2282 		    amdgpu_ip_version(adev, MP0_HWIP, 0) !=
2283 			    IP_VERSION(11, 0, 4) &&
2284 		    amdgpu_ip_version(adev, MP0_HWIP, 0) !=
2285 			    IP_VERSION(13, 0, 0)) {
2286 			if (amdgpu_ras_reset_error_status(adev, info.head.block))
2287 				dev_warn(adev->dev, "Failed to reset error counter and error status");
2288 		}
2289 	}
2290 }
2291 
2292 /* Parse RdRspStatus and WrRspStatus */
2293 static void amdgpu_ras_error_status_query(struct amdgpu_device *adev,
2294 					  struct ras_query_if *info)
2295 {
2296 	struct amdgpu_ras_block_object *block_obj;
2297 	/*
2298 	 * Only two block need to query read/write
2299 	 * RspStatus at current state
2300 	 */
2301 	if ((info->head.block != AMDGPU_RAS_BLOCK__GFX) &&
2302 		(info->head.block != AMDGPU_RAS_BLOCK__MMHUB))
2303 		return;
2304 
2305 	block_obj = amdgpu_ras_get_ras_block(adev,
2306 					info->head.block,
2307 					info->head.sub_block_index);
2308 
2309 	if (!block_obj || !block_obj->hw_ops) {
2310 		dev_dbg_once(adev->dev, "%s doesn't config RAS function\n",
2311 			     get_ras_block_str(&info->head));
2312 		return;
2313 	}
2314 
2315 	if (block_obj->hw_ops->query_ras_error_status)
2316 		block_obj->hw_ops->query_ras_error_status(adev);
2317 
2318 }
2319 
2320 static void amdgpu_ras_query_err_status(struct amdgpu_device *adev)
2321 {
2322 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2323 	struct ras_manager *obj;
2324 
2325 	if (!adev->ras_enabled || !con)
2326 		return;
2327 
2328 	list_for_each_entry(obj, &con->head, node) {
2329 		struct ras_query_if info = {
2330 			.head = obj->head,
2331 		};
2332 
2333 		amdgpu_ras_error_status_query(adev, &info);
2334 	}
2335 }
2336 
2337 /* recovery begin */
2338 
2339 /* return 0 on success.
2340  * caller need free bps.
2341  */
2342 static int amdgpu_ras_badpages_read(struct amdgpu_device *adev,
2343 		struct ras_badpage **bps, unsigned int *count)
2344 {
2345 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2346 	struct ras_err_handler_data *data;
2347 	int i = 0;
2348 	int ret = 0, status;
2349 
2350 	if (!con || !con->eh_data || !bps || !count)
2351 		return -EINVAL;
2352 
2353 	mutex_lock(&con->recovery_lock);
2354 	data = con->eh_data;
2355 	if (!data || data->count == 0) {
2356 		*bps = NULL;
2357 		ret = -EINVAL;
2358 		goto out;
2359 	}
2360 
2361 	*bps = kmalloc(sizeof(struct ras_badpage) * data->count, GFP_KERNEL);
2362 	if (!*bps) {
2363 		ret = -ENOMEM;
2364 		goto out;
2365 	}
2366 
2367 	for (; i < data->count; i++) {
2368 		(*bps)[i] = (struct ras_badpage){
2369 			.bp = data->bps[i].retired_page,
2370 			.size = AMDGPU_GPU_PAGE_SIZE,
2371 			.flags = AMDGPU_RAS_RETIRE_PAGE_RESERVED,
2372 		};
2373 		status = amdgpu_vram_mgr_query_page_status(&adev->mman.vram_mgr,
2374 				data->bps[i].retired_page);
2375 		if (status == -EBUSY)
2376 			(*bps)[i].flags = AMDGPU_RAS_RETIRE_PAGE_PENDING;
2377 		else if (status == -ENOENT)
2378 			(*bps)[i].flags = AMDGPU_RAS_RETIRE_PAGE_FAULT;
2379 	}
2380 
2381 	*count = data->count;
2382 out:
2383 	mutex_unlock(&con->recovery_lock);
2384 	return ret;
2385 }
2386 
2387 static void amdgpu_ras_do_recovery(struct work_struct *work)
2388 {
2389 	struct amdgpu_ras *ras =
2390 		container_of(work, struct amdgpu_ras, recovery_work);
2391 	struct amdgpu_device *remote_adev = NULL;
2392 	struct amdgpu_device *adev = ras->adev;
2393 	struct list_head device_list, *device_list_handle =  NULL;
2394 	struct amdgpu_hive_info *hive = amdgpu_get_xgmi_hive(adev);
2395 
2396 	if (hive)
2397 		atomic_set(&hive->ras_recovery, 1);
2398 	if (!ras->disable_ras_err_cnt_harvest) {
2399 
2400 		/* Build list of devices to query RAS related errors */
2401 		if  (hive && adev->gmc.xgmi.num_physical_nodes > 1) {
2402 			device_list_handle = &hive->device_list;
2403 		} else {
2404 			INIT_LIST_HEAD(&device_list);
2405 			list_add_tail(&adev->gmc.xgmi.head, &device_list);
2406 			device_list_handle = &device_list;
2407 		}
2408 
2409 		list_for_each_entry(remote_adev,
2410 				device_list_handle, gmc.xgmi.head) {
2411 			amdgpu_ras_query_err_status(remote_adev);
2412 			amdgpu_ras_log_on_err_counter(remote_adev);
2413 		}
2414 
2415 	}
2416 
2417 	if (amdgpu_device_should_recover_gpu(ras->adev)) {
2418 		struct amdgpu_reset_context reset_context;
2419 		memset(&reset_context, 0, sizeof(reset_context));
2420 
2421 		reset_context.method = AMD_RESET_METHOD_NONE;
2422 		reset_context.reset_req_dev = adev;
2423 
2424 		/* Perform full reset in fatal error mode */
2425 		if (!amdgpu_ras_is_poison_mode_supported(ras->adev))
2426 			set_bit(AMDGPU_NEED_FULL_RESET, &reset_context.flags);
2427 		else {
2428 			clear_bit(AMDGPU_NEED_FULL_RESET, &reset_context.flags);
2429 
2430 			if (ras->gpu_reset_flags & AMDGPU_RAS_GPU_RESET_MODE2_RESET) {
2431 				ras->gpu_reset_flags &= ~AMDGPU_RAS_GPU_RESET_MODE2_RESET;
2432 				reset_context.method = AMD_RESET_METHOD_MODE2;
2433 			}
2434 
2435 			/* Fatal error occurs in poison mode, mode1 reset is used to
2436 			 * recover gpu.
2437 			 */
2438 			if (ras->gpu_reset_flags & AMDGPU_RAS_GPU_RESET_MODE1_RESET) {
2439 				ras->gpu_reset_flags &= ~AMDGPU_RAS_GPU_RESET_MODE1_RESET;
2440 				set_bit(AMDGPU_NEED_FULL_RESET, &reset_context.flags);
2441 
2442 				/* For any RAS error that needs a full reset to
2443 				 * recover, set the fatal error status
2444 				 */
2445 				if (hive) {
2446 					list_for_each_entry(remote_adev,
2447 							    &hive->device_list,
2448 							    gmc.xgmi.head)
2449 						amdgpu_ras_set_fed(remote_adev,
2450 								   true);
2451 				} else {
2452 					amdgpu_ras_set_fed(adev, true);
2453 				}
2454 				psp_fatal_error_recovery_quirk(&adev->psp);
2455 			}
2456 		}
2457 
2458 		amdgpu_device_gpu_recover(ras->adev, NULL, &reset_context);
2459 	}
2460 	atomic_set(&ras->in_recovery, 0);
2461 	if (hive) {
2462 		atomic_set(&hive->ras_recovery, 0);
2463 		amdgpu_put_xgmi_hive(hive);
2464 	}
2465 }
2466 
2467 /* alloc/realloc bps array */
2468 static int amdgpu_ras_realloc_eh_data_space(struct amdgpu_device *adev,
2469 		struct ras_err_handler_data *data, int pages)
2470 {
2471 	unsigned int old_space = data->count + data->space_left;
2472 	unsigned int new_space = old_space + pages;
2473 	unsigned int align_space = ALIGN(new_space, 512);
2474 	void *bps = kmalloc(align_space * sizeof(*data->bps), GFP_KERNEL);
2475 
2476 	if (!bps) {
2477 		return -ENOMEM;
2478 	}
2479 
2480 	if (data->bps) {
2481 		memcpy(bps, data->bps,
2482 				data->count * sizeof(*data->bps));
2483 		kfree(data->bps);
2484 	}
2485 
2486 	data->bps = bps;
2487 	data->space_left += align_space - old_space;
2488 	return 0;
2489 }
2490 
2491 /* it deal with vram only. */
2492 int amdgpu_ras_add_bad_pages(struct amdgpu_device *adev,
2493 		struct eeprom_table_record *bps, int pages)
2494 {
2495 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2496 	struct ras_err_handler_data *data;
2497 	int ret = 0;
2498 	uint32_t i;
2499 
2500 	if (!con || !con->eh_data || !bps || pages <= 0)
2501 		return 0;
2502 
2503 	mutex_lock(&con->recovery_lock);
2504 	data = con->eh_data;
2505 	if (!data)
2506 		goto out;
2507 
2508 	for (i = 0; i < pages; i++) {
2509 		if (amdgpu_ras_check_bad_page_unlock(con,
2510 			bps[i].retired_page << AMDGPU_GPU_PAGE_SHIFT))
2511 			continue;
2512 
2513 		if (!data->space_left &&
2514 			amdgpu_ras_realloc_eh_data_space(adev, data, 256)) {
2515 			ret = -ENOMEM;
2516 			goto out;
2517 		}
2518 
2519 		amdgpu_vram_mgr_reserve_range(&adev->mman.vram_mgr,
2520 			bps[i].retired_page << AMDGPU_GPU_PAGE_SHIFT,
2521 			AMDGPU_GPU_PAGE_SIZE);
2522 
2523 		memcpy(&data->bps[data->count], &bps[i], sizeof(*data->bps));
2524 		data->count++;
2525 		data->space_left--;
2526 	}
2527 out:
2528 	mutex_unlock(&con->recovery_lock);
2529 
2530 	return ret;
2531 }
2532 
2533 /*
2534  * write error record array to eeprom, the function should be
2535  * protected by recovery_lock
2536  * new_cnt: new added UE count, excluding reserved bad pages, can be NULL
2537  */
2538 int amdgpu_ras_save_bad_pages(struct amdgpu_device *adev,
2539 		unsigned long *new_cnt)
2540 {
2541 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2542 	struct ras_err_handler_data *data;
2543 	struct amdgpu_ras_eeprom_control *control;
2544 	int save_count;
2545 
2546 	if (!con || !con->eh_data) {
2547 		if (new_cnt)
2548 			*new_cnt = 0;
2549 
2550 		return 0;
2551 	}
2552 
2553 	mutex_lock(&con->recovery_lock);
2554 	control = &con->eeprom_control;
2555 	data = con->eh_data;
2556 	save_count = data->count - control->ras_num_recs;
2557 	mutex_unlock(&con->recovery_lock);
2558 
2559 	if (new_cnt)
2560 		*new_cnt = save_count / adev->umc.retire_unit;
2561 
2562 	/* only new entries are saved */
2563 	if (save_count > 0) {
2564 		if (amdgpu_ras_eeprom_append(control,
2565 					     &data->bps[control->ras_num_recs],
2566 					     save_count)) {
2567 			dev_err(adev->dev, "Failed to save EEPROM table data!");
2568 			return -EIO;
2569 		}
2570 
2571 		dev_info(adev->dev, "Saved %d pages to EEPROM table.\n", save_count);
2572 	}
2573 
2574 	return 0;
2575 }
2576 
2577 /*
2578  * read error record array in eeprom and reserve enough space for
2579  * storing new bad pages
2580  */
2581 static int amdgpu_ras_load_bad_pages(struct amdgpu_device *adev)
2582 {
2583 	struct amdgpu_ras_eeprom_control *control =
2584 		&adev->psp.ras_context.ras->eeprom_control;
2585 	struct eeprom_table_record *bps;
2586 	int ret;
2587 
2588 	/* no bad page record, skip eeprom access */
2589 	if (control->ras_num_recs == 0 || amdgpu_bad_page_threshold == 0)
2590 		return 0;
2591 
2592 	bps = kcalloc(control->ras_num_recs, sizeof(*bps), GFP_KERNEL);
2593 	if (!bps)
2594 		return -ENOMEM;
2595 
2596 	ret = amdgpu_ras_eeprom_read(control, bps, control->ras_num_recs);
2597 	if (ret)
2598 		dev_err(adev->dev, "Failed to load EEPROM table records!");
2599 	else
2600 		ret = amdgpu_ras_add_bad_pages(adev, bps, control->ras_num_recs);
2601 
2602 	kfree(bps);
2603 	return ret;
2604 }
2605 
2606 static bool amdgpu_ras_check_bad_page_unlock(struct amdgpu_ras *con,
2607 				uint64_t addr)
2608 {
2609 	struct ras_err_handler_data *data = con->eh_data;
2610 	int i;
2611 
2612 	addr >>= AMDGPU_GPU_PAGE_SHIFT;
2613 	for (i = 0; i < data->count; i++)
2614 		if (addr == data->bps[i].retired_page)
2615 			return true;
2616 
2617 	return false;
2618 }
2619 
2620 /*
2621  * check if an address belongs to bad page
2622  *
2623  * Note: this check is only for umc block
2624  */
2625 static bool amdgpu_ras_check_bad_page(struct amdgpu_device *adev,
2626 				uint64_t addr)
2627 {
2628 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2629 	bool ret = false;
2630 
2631 	if (!con || !con->eh_data)
2632 		return ret;
2633 
2634 	mutex_lock(&con->recovery_lock);
2635 	ret = amdgpu_ras_check_bad_page_unlock(con, addr);
2636 	mutex_unlock(&con->recovery_lock);
2637 	return ret;
2638 }
2639 
2640 static void amdgpu_ras_validate_threshold(struct amdgpu_device *adev,
2641 					  uint32_t max_count)
2642 {
2643 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2644 
2645 	/*
2646 	 * Justification of value bad_page_cnt_threshold in ras structure
2647 	 *
2648 	 * Generally, 0 <= amdgpu_bad_page_threshold <= max record length
2649 	 * in eeprom or amdgpu_bad_page_threshold == -2, introduce two
2650 	 * scenarios accordingly.
2651 	 *
2652 	 * Bad page retirement enablement:
2653 	 *    - If amdgpu_bad_page_threshold = -2,
2654 	 *      bad_page_cnt_threshold = typical value by formula.
2655 	 *
2656 	 *    - When the value from user is 0 < amdgpu_bad_page_threshold <
2657 	 *      max record length in eeprom, use it directly.
2658 	 *
2659 	 * Bad page retirement disablement:
2660 	 *    - If amdgpu_bad_page_threshold = 0, bad page retirement
2661 	 *      functionality is disabled, and bad_page_cnt_threshold will
2662 	 *      take no effect.
2663 	 */
2664 
2665 	if (amdgpu_bad_page_threshold < 0) {
2666 		u64 val = adev->gmc.mc_vram_size;
2667 
2668 		do_div(val, RAS_BAD_PAGE_COVER);
2669 		con->bad_page_cnt_threshold = min(lower_32_bits(val),
2670 						  max_count);
2671 	} else {
2672 		con->bad_page_cnt_threshold = min_t(int, max_count,
2673 						    amdgpu_bad_page_threshold);
2674 	}
2675 }
2676 
2677 static int amdgpu_ras_page_retirement_thread(void *param)
2678 {
2679 	struct amdgpu_device *adev = (struct amdgpu_device *)param;
2680 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2681 
2682 	while (!kthread_should_stop()) {
2683 
2684 		wait_event_interruptible(con->page_retirement_wq,
2685 				kthread_should_stop() ||
2686 				atomic_read(&con->page_retirement_req_cnt));
2687 
2688 		if (kthread_should_stop())
2689 			break;
2690 
2691 		dev_info(adev->dev, "Start processing page retirement. request:%d\n",
2692 			atomic_read(&con->page_retirement_req_cnt));
2693 
2694 		atomic_dec(&con->page_retirement_req_cnt);
2695 
2696 		amdgpu_umc_bad_page_polling_timeout(adev,
2697 				false, MAX_UMC_POISON_POLLING_TIME_ASYNC);
2698 	}
2699 
2700 	return 0;
2701 }
2702 
2703 int amdgpu_ras_recovery_init(struct amdgpu_device *adev)
2704 {
2705 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2706 	struct ras_err_handler_data **data;
2707 	u32  max_eeprom_records_count = 0;
2708 	bool exc_err_limit = false;
2709 	int ret;
2710 
2711 	if (!con || amdgpu_sriov_vf(adev))
2712 		return 0;
2713 
2714 	/* Allow access to RAS EEPROM via debugfs, when the ASIC
2715 	 * supports RAS and debugfs is enabled, but when
2716 	 * adev->ras_enabled is unset, i.e. when "ras_enable"
2717 	 * module parameter is set to 0.
2718 	 */
2719 	con->adev = adev;
2720 
2721 	if (!adev->ras_enabled)
2722 		return 0;
2723 
2724 	data = &con->eh_data;
2725 	*data = kzalloc(sizeof(**data), GFP_KERNEL);
2726 	if (!*data) {
2727 		ret = -ENOMEM;
2728 		goto out;
2729 	}
2730 
2731 	mutex_init(&con->recovery_lock);
2732 	INIT_WORK(&con->recovery_work, amdgpu_ras_do_recovery);
2733 	atomic_set(&con->in_recovery, 0);
2734 	con->eeprom_control.bad_channel_bitmap = 0;
2735 
2736 	max_eeprom_records_count = amdgpu_ras_eeprom_max_record_count(&con->eeprom_control);
2737 	amdgpu_ras_validate_threshold(adev, max_eeprom_records_count);
2738 
2739 	/* Todo: During test the SMU might fail to read the eeprom through I2C
2740 	 * when the GPU is pending on XGMI reset during probe time
2741 	 * (Mostly after second bus reset), skip it now
2742 	 */
2743 	if (adev->gmc.xgmi.pending_reset)
2744 		return 0;
2745 	ret = amdgpu_ras_eeprom_init(&con->eeprom_control, &exc_err_limit);
2746 	/*
2747 	 * This calling fails when exc_err_limit is true or
2748 	 * ret != 0.
2749 	 */
2750 	if (exc_err_limit || ret)
2751 		goto free;
2752 
2753 	if (con->eeprom_control.ras_num_recs) {
2754 		ret = amdgpu_ras_load_bad_pages(adev);
2755 		if (ret)
2756 			goto free;
2757 
2758 		amdgpu_dpm_send_hbm_bad_pages_num(adev, con->eeprom_control.ras_num_recs);
2759 
2760 		if (con->update_channel_flag == true) {
2761 			amdgpu_dpm_send_hbm_bad_channel_flag(adev, con->eeprom_control.bad_channel_bitmap);
2762 			con->update_channel_flag = false;
2763 		}
2764 	}
2765 
2766 	mutex_init(&con->page_retirement_lock);
2767 	init_waitqueue_head(&con->page_retirement_wq);
2768 	atomic_set(&con->page_retirement_req_cnt, 0);
2769 	con->page_retirement_thread =
2770 		kthread_run(amdgpu_ras_page_retirement_thread, adev, "umc_page_retirement");
2771 	if (IS_ERR(con->page_retirement_thread)) {
2772 		con->page_retirement_thread = NULL;
2773 		dev_warn(adev->dev, "Failed to create umc_page_retirement thread!!!\n");
2774 	}
2775 
2776 #ifdef CONFIG_X86_MCE_AMD
2777 	if ((adev->asic_type == CHIP_ALDEBARAN) &&
2778 	    (adev->gmc.xgmi.connected_to_cpu))
2779 		amdgpu_register_bad_pages_mca_notifier(adev);
2780 #endif
2781 	return 0;
2782 
2783 free:
2784 	kfree((*data)->bps);
2785 	kfree(*data);
2786 	con->eh_data = NULL;
2787 out:
2788 	dev_warn(adev->dev, "Failed to initialize ras recovery! (%d)\n", ret);
2789 
2790 	/*
2791 	 * Except error threshold exceeding case, other failure cases in this
2792 	 * function would not fail amdgpu driver init.
2793 	 */
2794 	if (!exc_err_limit)
2795 		ret = 0;
2796 	else
2797 		ret = -EINVAL;
2798 
2799 	return ret;
2800 }
2801 
2802 static int amdgpu_ras_recovery_fini(struct amdgpu_device *adev)
2803 {
2804 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2805 	struct ras_err_handler_data *data = con->eh_data;
2806 
2807 	/* recovery_init failed to init it, fini is useless */
2808 	if (!data)
2809 		return 0;
2810 
2811 	if (con->page_retirement_thread)
2812 		kthread_stop(con->page_retirement_thread);
2813 
2814 	atomic_set(&con->page_retirement_req_cnt, 0);
2815 
2816 	cancel_work_sync(&con->recovery_work);
2817 
2818 	mutex_lock(&con->recovery_lock);
2819 	con->eh_data = NULL;
2820 	kfree(data->bps);
2821 	kfree(data);
2822 	mutex_unlock(&con->recovery_lock);
2823 
2824 	return 0;
2825 }
2826 /* recovery end */
2827 
2828 static bool amdgpu_ras_asic_supported(struct amdgpu_device *adev)
2829 {
2830 	if (amdgpu_sriov_vf(adev)) {
2831 		switch (amdgpu_ip_version(adev, MP0_HWIP, 0)) {
2832 		case IP_VERSION(13, 0, 2):
2833 		case IP_VERSION(13, 0, 6):
2834 			return true;
2835 		default:
2836 			return false;
2837 		}
2838 	}
2839 
2840 	if (adev->asic_type == CHIP_IP_DISCOVERY) {
2841 		switch (amdgpu_ip_version(adev, MP0_HWIP, 0)) {
2842 		case IP_VERSION(13, 0, 0):
2843 		case IP_VERSION(13, 0, 6):
2844 		case IP_VERSION(13, 0, 10):
2845 			return true;
2846 		default:
2847 			return false;
2848 		}
2849 	}
2850 
2851 	return adev->asic_type == CHIP_VEGA10 ||
2852 		adev->asic_type == CHIP_VEGA20 ||
2853 		adev->asic_type == CHIP_ARCTURUS ||
2854 		adev->asic_type == CHIP_ALDEBARAN ||
2855 		adev->asic_type == CHIP_SIENNA_CICHLID;
2856 }
2857 
2858 /*
2859  * this is workaround for vega20 workstation sku,
2860  * force enable gfx ras, ignore vbios gfx ras flag
2861  * due to GC EDC can not write
2862  */
2863 static void amdgpu_ras_get_quirks(struct amdgpu_device *adev)
2864 {
2865 	struct atom_context *ctx = adev->mode_info.atom_context;
2866 
2867 	if (!ctx)
2868 		return;
2869 
2870 	if (strnstr(ctx->vbios_pn, "D16406",
2871 		    sizeof(ctx->vbios_pn)) ||
2872 		strnstr(ctx->vbios_pn, "D36002",
2873 			sizeof(ctx->vbios_pn)))
2874 		adev->ras_hw_enabled |= (1 << AMDGPU_RAS_BLOCK__GFX);
2875 }
2876 
2877 /* Query ras capablity via atomfirmware interface */
2878 static void amdgpu_ras_query_ras_capablity_from_vbios(struct amdgpu_device *adev)
2879 {
2880 	/* mem_ecc cap */
2881 	if (amdgpu_atomfirmware_mem_ecc_supported(adev)) {
2882 		dev_info(adev->dev, "MEM ECC is active.\n");
2883 		adev->ras_hw_enabled |= (1 << AMDGPU_RAS_BLOCK__UMC |
2884 					 1 << AMDGPU_RAS_BLOCK__DF);
2885 	} else {
2886 		dev_info(adev->dev, "MEM ECC is not presented.\n");
2887 	}
2888 
2889 	/* sram_ecc cap */
2890 	if (amdgpu_atomfirmware_sram_ecc_supported(adev)) {
2891 		dev_info(adev->dev, "SRAM ECC is active.\n");
2892 		if (!amdgpu_sriov_vf(adev))
2893 			adev->ras_hw_enabled |= ~(1 << AMDGPU_RAS_BLOCK__UMC |
2894 						  1 << AMDGPU_RAS_BLOCK__DF);
2895 		else
2896 			adev->ras_hw_enabled |= (1 << AMDGPU_RAS_BLOCK__PCIE_BIF |
2897 						 1 << AMDGPU_RAS_BLOCK__SDMA |
2898 						 1 << AMDGPU_RAS_BLOCK__GFX);
2899 
2900 		/*
2901 		 * VCN/JPEG RAS can be supported on both bare metal and
2902 		 * SRIOV environment
2903 		 */
2904 		if (amdgpu_ip_version(adev, VCN_HWIP, 0) == IP_VERSION(2, 6, 0) ||
2905 		    amdgpu_ip_version(adev, VCN_HWIP, 0) == IP_VERSION(4, 0, 0) ||
2906 		    amdgpu_ip_version(adev, VCN_HWIP, 0) == IP_VERSION(4, 0, 3))
2907 			adev->ras_hw_enabled |= (1 << AMDGPU_RAS_BLOCK__VCN |
2908 						 1 << AMDGPU_RAS_BLOCK__JPEG);
2909 		else
2910 			adev->ras_hw_enabled &= ~(1 << AMDGPU_RAS_BLOCK__VCN |
2911 						  1 << AMDGPU_RAS_BLOCK__JPEG);
2912 
2913 		/*
2914 		 * XGMI RAS is not supported if xgmi num physical nodes
2915 		 * is zero
2916 		 */
2917 		if (!adev->gmc.xgmi.num_physical_nodes)
2918 			adev->ras_hw_enabled &= ~(1 << AMDGPU_RAS_BLOCK__XGMI_WAFL);
2919 	} else {
2920 		dev_info(adev->dev, "SRAM ECC is not presented.\n");
2921 	}
2922 }
2923 
2924 /* Query poison mode from umc/df IP callbacks */
2925 static void amdgpu_ras_query_poison_mode(struct amdgpu_device *adev)
2926 {
2927 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
2928 	bool df_poison, umc_poison;
2929 
2930 	/* poison setting is useless on SRIOV guest */
2931 	if (amdgpu_sriov_vf(adev) || !con)
2932 		return;
2933 
2934 	/* Init poison supported flag, the default value is false */
2935 	if (adev->gmc.xgmi.connected_to_cpu ||
2936 	    adev->gmc.is_app_apu) {
2937 		/* enabled by default when GPU is connected to CPU */
2938 		con->poison_supported = true;
2939 	} else if (adev->df.funcs &&
2940 	    adev->df.funcs->query_ras_poison_mode &&
2941 	    adev->umc.ras &&
2942 	    adev->umc.ras->query_ras_poison_mode) {
2943 		df_poison =
2944 			adev->df.funcs->query_ras_poison_mode(adev);
2945 		umc_poison =
2946 			adev->umc.ras->query_ras_poison_mode(adev);
2947 
2948 		/* Only poison is set in both DF and UMC, we can support it */
2949 		if (df_poison && umc_poison)
2950 			con->poison_supported = true;
2951 		else if (df_poison != umc_poison)
2952 			dev_warn(adev->dev,
2953 				"Poison setting is inconsistent in DF/UMC(%d:%d)!\n",
2954 				df_poison, umc_poison);
2955 	}
2956 }
2957 
2958 /*
2959  * check hardware's ras ability which will be saved in hw_supported.
2960  * if hardware does not support ras, we can skip some ras initializtion and
2961  * forbid some ras operations from IP.
2962  * if software itself, say boot parameter, limit the ras ability. We still
2963  * need allow IP do some limited operations, like disable. In such case,
2964  * we have to initialize ras as normal. but need check if operation is
2965  * allowed or not in each function.
2966  */
2967 static void amdgpu_ras_check_supported(struct amdgpu_device *adev)
2968 {
2969 	adev->ras_hw_enabled = adev->ras_enabled = 0;
2970 
2971 	if (!amdgpu_ras_asic_supported(adev))
2972 		return;
2973 
2974 	/* query ras capability from psp */
2975 	if (amdgpu_psp_get_ras_capability(&adev->psp))
2976 		goto init_ras_enabled_flag;
2977 
2978 	/* query ras capablity from bios */
2979 	if (!adev->gmc.xgmi.connected_to_cpu && !adev->gmc.is_app_apu) {
2980 		amdgpu_ras_query_ras_capablity_from_vbios(adev);
2981 	} else {
2982 		/* driver only manages a few IP blocks RAS feature
2983 		 * when GPU is connected cpu through XGMI */
2984 		adev->ras_hw_enabled |= (1 << AMDGPU_RAS_BLOCK__GFX |
2985 					   1 << AMDGPU_RAS_BLOCK__SDMA |
2986 					   1 << AMDGPU_RAS_BLOCK__MMHUB);
2987 	}
2988 
2989 	/* apply asic specific settings (vega20 only for now) */
2990 	amdgpu_ras_get_quirks(adev);
2991 
2992 	/* query poison mode from umc/df ip callback */
2993 	amdgpu_ras_query_poison_mode(adev);
2994 
2995 init_ras_enabled_flag:
2996 	/* hw_supported needs to be aligned with RAS block mask. */
2997 	adev->ras_hw_enabled &= AMDGPU_RAS_BLOCK_MASK;
2998 
2999 	adev->ras_enabled = amdgpu_ras_enable == 0 ? 0 :
3000 		adev->ras_hw_enabled & amdgpu_ras_mask;
3001 
3002 	/* aca is disabled by default */
3003 	adev->aca.is_enabled = false;
3004 }
3005 
3006 static void amdgpu_ras_counte_dw(struct work_struct *work)
3007 {
3008 	struct amdgpu_ras *con = container_of(work, struct amdgpu_ras,
3009 					      ras_counte_delay_work.work);
3010 	struct amdgpu_device *adev = con->adev;
3011 	struct drm_device *dev = adev_to_drm(adev);
3012 	unsigned long ce_count, ue_count;
3013 	int res;
3014 
3015 	res = pm_runtime_get_sync(dev->dev);
3016 	if (res < 0)
3017 		goto Out;
3018 
3019 	/* Cache new values.
3020 	 */
3021 	if (amdgpu_ras_query_error_count(adev, &ce_count, &ue_count, NULL) == 0) {
3022 		atomic_set(&con->ras_ce_count, ce_count);
3023 		atomic_set(&con->ras_ue_count, ue_count);
3024 	}
3025 
3026 	pm_runtime_mark_last_busy(dev->dev);
3027 Out:
3028 	pm_runtime_put_autosuspend(dev->dev);
3029 }
3030 
3031 static int amdgpu_get_ras_schema(struct amdgpu_device *adev)
3032 {
3033 	return  amdgpu_ras_is_poison_mode_supported(adev) ? AMDGPU_RAS_ERROR__POISON : 0 |
3034 			AMDGPU_RAS_ERROR__SINGLE_CORRECTABLE |
3035 			AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE |
3036 			AMDGPU_RAS_ERROR__PARITY;
3037 }
3038 
3039 int amdgpu_ras_init(struct amdgpu_device *adev)
3040 {
3041 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3042 	int r;
3043 
3044 	if (con)
3045 		return 0;
3046 
3047 	con = kzalloc(sizeof(*con) +
3048 			sizeof(struct ras_manager) * AMDGPU_RAS_BLOCK_COUNT +
3049 			sizeof(struct ras_manager) * AMDGPU_RAS_MCA_BLOCK_COUNT,
3050 			GFP_KERNEL);
3051 	if (!con)
3052 		return -ENOMEM;
3053 
3054 	con->adev = adev;
3055 	INIT_DELAYED_WORK(&con->ras_counte_delay_work, amdgpu_ras_counte_dw);
3056 	atomic_set(&con->ras_ce_count, 0);
3057 	atomic_set(&con->ras_ue_count, 0);
3058 
3059 	con->objs = (struct ras_manager *)(con + 1);
3060 
3061 	amdgpu_ras_set_context(adev, con);
3062 
3063 	amdgpu_ras_check_supported(adev);
3064 
3065 	if (!adev->ras_enabled || adev->asic_type == CHIP_VEGA10) {
3066 		/* set gfx block ras context feature for VEGA20 Gaming
3067 		 * send ras disable cmd to ras ta during ras late init.
3068 		 */
3069 		if (!adev->ras_enabled && adev->asic_type == CHIP_VEGA20) {
3070 			con->features |= BIT(AMDGPU_RAS_BLOCK__GFX);
3071 
3072 			return 0;
3073 		}
3074 
3075 		r = 0;
3076 		goto release_con;
3077 	}
3078 
3079 	con->update_channel_flag = false;
3080 	con->features = 0;
3081 	con->schema = 0;
3082 	INIT_LIST_HEAD(&con->head);
3083 	/* Might need get this flag from vbios. */
3084 	con->flags = RAS_DEFAULT_FLAGS;
3085 
3086 	/* initialize nbio ras function ahead of any other
3087 	 * ras functions so hardware fatal error interrupt
3088 	 * can be enabled as early as possible */
3089 	switch (amdgpu_ip_version(adev, NBIO_HWIP, 0)) {
3090 	case IP_VERSION(7, 4, 0):
3091 	case IP_VERSION(7, 4, 1):
3092 	case IP_VERSION(7, 4, 4):
3093 		if (!adev->gmc.xgmi.connected_to_cpu)
3094 			adev->nbio.ras = &nbio_v7_4_ras;
3095 		break;
3096 	case IP_VERSION(4, 3, 0):
3097 		if (adev->ras_hw_enabled & (1 << AMDGPU_RAS_BLOCK__DF))
3098 			/* unlike other generation of nbio ras,
3099 			 * nbio v4_3 only support fatal error interrupt
3100 			 * to inform software that DF is freezed due to
3101 			 * system fatal error event. driver should not
3102 			 * enable nbio ras in such case. Instead,
3103 			 * check DF RAS */
3104 			adev->nbio.ras = &nbio_v4_3_ras;
3105 		break;
3106 	case IP_VERSION(7, 9, 0):
3107 		if (!adev->gmc.is_app_apu)
3108 			adev->nbio.ras = &nbio_v7_9_ras;
3109 		break;
3110 	default:
3111 		/* nbio ras is not available */
3112 		break;
3113 	}
3114 
3115 	/* nbio ras block needs to be enabled ahead of other ras blocks
3116 	 * to handle fatal error */
3117 	r = amdgpu_nbio_ras_sw_init(adev);
3118 	if (r)
3119 		return r;
3120 
3121 	if (adev->nbio.ras &&
3122 	    adev->nbio.ras->init_ras_controller_interrupt) {
3123 		r = adev->nbio.ras->init_ras_controller_interrupt(adev);
3124 		if (r)
3125 			goto release_con;
3126 	}
3127 
3128 	if (adev->nbio.ras &&
3129 	    adev->nbio.ras->init_ras_err_event_athub_interrupt) {
3130 		r = adev->nbio.ras->init_ras_err_event_athub_interrupt(adev);
3131 		if (r)
3132 			goto release_con;
3133 	}
3134 
3135 	/* Packed socket_id to ras feature mask bits[31:29] */
3136 	if (adev->smuio.funcs &&
3137 	    adev->smuio.funcs->get_socket_id)
3138 		con->features |= ((adev->smuio.funcs->get_socket_id(adev)) <<
3139 					AMDGPU_RAS_FEATURES_SOCKETID_SHIFT);
3140 
3141 	/* Get RAS schema for particular SOC */
3142 	con->schema = amdgpu_get_ras_schema(adev);
3143 
3144 	if (amdgpu_ras_fs_init(adev)) {
3145 		r = -EINVAL;
3146 		goto release_con;
3147 	}
3148 
3149 	dev_info(adev->dev, "RAS INFO: ras initialized successfully, "
3150 		 "hardware ability[%x] ras_mask[%x]\n",
3151 		 adev->ras_hw_enabled, adev->ras_enabled);
3152 
3153 	return 0;
3154 release_con:
3155 	amdgpu_ras_set_context(adev, NULL);
3156 	kfree(con);
3157 
3158 	return r;
3159 }
3160 
3161 int amdgpu_persistent_edc_harvesting_supported(struct amdgpu_device *adev)
3162 {
3163 	if (adev->gmc.xgmi.connected_to_cpu ||
3164 	    adev->gmc.is_app_apu)
3165 		return 1;
3166 	return 0;
3167 }
3168 
3169 static int amdgpu_persistent_edc_harvesting(struct amdgpu_device *adev,
3170 					struct ras_common_if *ras_block)
3171 {
3172 	struct ras_query_if info = {
3173 		.head = *ras_block,
3174 	};
3175 
3176 	if (!amdgpu_persistent_edc_harvesting_supported(adev))
3177 		return 0;
3178 
3179 	if (amdgpu_ras_query_error_status(adev, &info) != 0)
3180 		DRM_WARN("RAS init harvest failure");
3181 
3182 	if (amdgpu_ras_reset_error_status(adev, ras_block->block) != 0)
3183 		DRM_WARN("RAS init harvest reset failure");
3184 
3185 	return 0;
3186 }
3187 
3188 bool amdgpu_ras_is_poison_mode_supported(struct amdgpu_device *adev)
3189 {
3190        struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3191 
3192        if (!con)
3193                return false;
3194 
3195        return con->poison_supported;
3196 }
3197 
3198 /* helper function to handle common stuff in ip late init phase */
3199 int amdgpu_ras_block_late_init(struct amdgpu_device *adev,
3200 			 struct ras_common_if *ras_block)
3201 {
3202 	struct amdgpu_ras_block_object *ras_obj = NULL;
3203 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3204 	struct ras_query_if *query_info;
3205 	unsigned long ue_count, ce_count;
3206 	int r;
3207 
3208 	/* disable RAS feature per IP block if it is not supported */
3209 	if (!amdgpu_ras_is_supported(adev, ras_block->block)) {
3210 		amdgpu_ras_feature_enable_on_boot(adev, ras_block, 0);
3211 		return 0;
3212 	}
3213 
3214 	r = amdgpu_ras_feature_enable_on_boot(adev, ras_block, 1);
3215 	if (r) {
3216 		if (adev->in_suspend || amdgpu_in_reset(adev)) {
3217 			/* in resume phase, if fail to enable ras,
3218 			 * clean up all ras fs nodes, and disable ras */
3219 			goto cleanup;
3220 		} else
3221 			return r;
3222 	}
3223 
3224 	/* check for errors on warm reset edc persisant supported ASIC */
3225 	amdgpu_persistent_edc_harvesting(adev, ras_block);
3226 
3227 	/* in resume phase, no need to create ras fs node */
3228 	if (adev->in_suspend || amdgpu_in_reset(adev))
3229 		return 0;
3230 
3231 	ras_obj = container_of(ras_block, struct amdgpu_ras_block_object, ras_comm);
3232 	if (ras_obj->ras_cb || (ras_obj->hw_ops &&
3233 	    (ras_obj->hw_ops->query_poison_status ||
3234 	    ras_obj->hw_ops->handle_poison_consumption))) {
3235 		r = amdgpu_ras_interrupt_add_handler(adev, ras_block);
3236 		if (r)
3237 			goto cleanup;
3238 	}
3239 
3240 	if (ras_obj->hw_ops &&
3241 	    (ras_obj->hw_ops->query_ras_error_count ||
3242 	     ras_obj->hw_ops->query_ras_error_status)) {
3243 		r = amdgpu_ras_sysfs_create(adev, ras_block);
3244 		if (r)
3245 			goto interrupt;
3246 
3247 		/* Those are the cached values at init.
3248 		 */
3249 		query_info = kzalloc(sizeof(*query_info), GFP_KERNEL);
3250 		if (!query_info)
3251 			return -ENOMEM;
3252 		memcpy(&query_info->head, ras_block, sizeof(struct ras_common_if));
3253 
3254 		if (amdgpu_ras_query_error_count(adev, &ce_count, &ue_count, query_info) == 0) {
3255 			atomic_set(&con->ras_ce_count, ce_count);
3256 			atomic_set(&con->ras_ue_count, ue_count);
3257 		}
3258 
3259 		kfree(query_info);
3260 	}
3261 
3262 	return 0;
3263 
3264 interrupt:
3265 	if (ras_obj->ras_cb)
3266 		amdgpu_ras_interrupt_remove_handler(adev, ras_block);
3267 cleanup:
3268 	amdgpu_ras_feature_enable(adev, ras_block, 0);
3269 	return r;
3270 }
3271 
3272 static int amdgpu_ras_block_late_init_default(struct amdgpu_device *adev,
3273 			 struct ras_common_if *ras_block)
3274 {
3275 	return amdgpu_ras_block_late_init(adev, ras_block);
3276 }
3277 
3278 /* helper function to remove ras fs node and interrupt handler */
3279 void amdgpu_ras_block_late_fini(struct amdgpu_device *adev,
3280 			  struct ras_common_if *ras_block)
3281 {
3282 	struct amdgpu_ras_block_object *ras_obj;
3283 	if (!ras_block)
3284 		return;
3285 
3286 	amdgpu_ras_sysfs_remove(adev, ras_block);
3287 
3288 	ras_obj = container_of(ras_block, struct amdgpu_ras_block_object, ras_comm);
3289 	if (ras_obj->ras_cb)
3290 		amdgpu_ras_interrupt_remove_handler(adev, ras_block);
3291 }
3292 
3293 static void amdgpu_ras_block_late_fini_default(struct amdgpu_device *adev,
3294 			  struct ras_common_if *ras_block)
3295 {
3296 	return amdgpu_ras_block_late_fini(adev, ras_block);
3297 }
3298 
3299 /* do some init work after IP late init as dependence.
3300  * and it runs in resume/gpu reset/booting up cases.
3301  */
3302 void amdgpu_ras_resume(struct amdgpu_device *adev)
3303 {
3304 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3305 	struct ras_manager *obj, *tmp;
3306 
3307 	if (!adev->ras_enabled || !con) {
3308 		/* clean ras context for VEGA20 Gaming after send ras disable cmd */
3309 		amdgpu_release_ras_context(adev);
3310 
3311 		return;
3312 	}
3313 
3314 	if (con->flags & AMDGPU_RAS_FLAG_INIT_BY_VBIOS) {
3315 		/* Set up all other IPs which are not implemented. There is a
3316 		 * tricky thing that IP's actual ras error type should be
3317 		 * MULTI_UNCORRECTABLE, but as driver does not handle it, so
3318 		 * ERROR_NONE make sense anyway.
3319 		 */
3320 		amdgpu_ras_enable_all_features(adev, 1);
3321 
3322 		/* We enable ras on all hw_supported block, but as boot
3323 		 * parameter might disable some of them and one or more IP has
3324 		 * not implemented yet. So we disable them on behalf.
3325 		 */
3326 		list_for_each_entry_safe(obj, tmp, &con->head, node) {
3327 			if (!amdgpu_ras_is_supported(adev, obj->head.block)) {
3328 				amdgpu_ras_feature_enable(adev, &obj->head, 0);
3329 				/* there should be no any reference. */
3330 				WARN_ON(alive_obj(obj));
3331 			}
3332 		}
3333 	}
3334 }
3335 
3336 void amdgpu_ras_suspend(struct amdgpu_device *adev)
3337 {
3338 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3339 
3340 	if (!adev->ras_enabled || !con)
3341 		return;
3342 
3343 	amdgpu_ras_disable_all_features(adev, 0);
3344 	/* Make sure all ras objects are disabled. */
3345 	if (AMDGPU_RAS_GET_FEATURES(con->features))
3346 		amdgpu_ras_disable_all_features(adev, 1);
3347 }
3348 
3349 int amdgpu_ras_late_init(struct amdgpu_device *adev)
3350 {
3351 	struct amdgpu_ras_block_list *node, *tmp;
3352 	struct amdgpu_ras_block_object *obj;
3353 	int r;
3354 
3355 	/* Guest side doesn't need init ras feature */
3356 	if (amdgpu_sriov_vf(adev))
3357 		return 0;
3358 
3359 	if (amdgpu_aca_is_enabled(adev)) {
3360 		if (amdgpu_in_reset(adev))
3361 			r = amdgpu_aca_reset(adev);
3362 		 else
3363 			r = amdgpu_aca_init(adev);
3364 		if (r)
3365 			return r;
3366 
3367 		amdgpu_ras_set_aca_debug_mode(adev, false);
3368 	} else {
3369 		amdgpu_ras_set_mca_debug_mode(adev, false);
3370 	}
3371 
3372 	list_for_each_entry_safe(node, tmp, &adev->ras_list, node) {
3373 		obj = node->ras_obj;
3374 		if (!obj) {
3375 			dev_warn(adev->dev, "Warning: abnormal ras list node.\n");
3376 			continue;
3377 		}
3378 
3379 		if (!amdgpu_ras_is_supported(adev, obj->ras_comm.block))
3380 			continue;
3381 
3382 		if (obj->ras_late_init) {
3383 			r = obj->ras_late_init(adev, &obj->ras_comm);
3384 			if (r) {
3385 				dev_err(adev->dev, "%s failed to execute ras_late_init! ret:%d\n",
3386 					obj->ras_comm.name, r);
3387 				return r;
3388 			}
3389 		} else
3390 			amdgpu_ras_block_late_init_default(adev, &obj->ras_comm);
3391 	}
3392 
3393 	return 0;
3394 }
3395 
3396 /* do some fini work before IP fini as dependence */
3397 int amdgpu_ras_pre_fini(struct amdgpu_device *adev)
3398 {
3399 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3400 
3401 	if (!adev->ras_enabled || !con)
3402 		return 0;
3403 
3404 
3405 	/* Need disable ras on all IPs here before ip [hw/sw]fini */
3406 	if (AMDGPU_RAS_GET_FEATURES(con->features))
3407 		amdgpu_ras_disable_all_features(adev, 0);
3408 	amdgpu_ras_recovery_fini(adev);
3409 	return 0;
3410 }
3411 
3412 int amdgpu_ras_fini(struct amdgpu_device *adev)
3413 {
3414 	struct amdgpu_ras_block_list *ras_node, *tmp;
3415 	struct amdgpu_ras_block_object *obj = NULL;
3416 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3417 
3418 	if (!adev->ras_enabled || !con)
3419 		return 0;
3420 
3421 	list_for_each_entry_safe(ras_node, tmp, &adev->ras_list, node) {
3422 		if (ras_node->ras_obj) {
3423 			obj = ras_node->ras_obj;
3424 			if (amdgpu_ras_is_supported(adev, obj->ras_comm.block) &&
3425 			    obj->ras_fini)
3426 				obj->ras_fini(adev, &obj->ras_comm);
3427 			else
3428 				amdgpu_ras_block_late_fini_default(adev, &obj->ras_comm);
3429 		}
3430 
3431 		/* Clear ras blocks from ras_list and free ras block list node */
3432 		list_del(&ras_node->node);
3433 		kfree(ras_node);
3434 	}
3435 
3436 	amdgpu_ras_fs_fini(adev);
3437 	amdgpu_ras_interrupt_remove_all(adev);
3438 
3439 	if (amdgpu_aca_is_enabled(adev))
3440 		amdgpu_aca_fini(adev);
3441 
3442 	WARN(AMDGPU_RAS_GET_FEATURES(con->features), "Feature mask is not cleared");
3443 
3444 	if (AMDGPU_RAS_GET_FEATURES(con->features))
3445 		amdgpu_ras_disable_all_features(adev, 0);
3446 
3447 	cancel_delayed_work_sync(&con->ras_counte_delay_work);
3448 
3449 	amdgpu_ras_set_context(adev, NULL);
3450 	kfree(con);
3451 
3452 	return 0;
3453 }
3454 
3455 bool amdgpu_ras_get_fed_status(struct amdgpu_device *adev)
3456 {
3457 	struct amdgpu_ras *ras;
3458 
3459 	ras = amdgpu_ras_get_context(adev);
3460 	if (!ras)
3461 		return false;
3462 
3463 	return atomic_read(&ras->fed);
3464 }
3465 
3466 void amdgpu_ras_set_fed(struct amdgpu_device *adev, bool status)
3467 {
3468 	struct amdgpu_ras *ras;
3469 
3470 	ras = amdgpu_ras_get_context(adev);
3471 	if (ras)
3472 		atomic_set(&ras->fed, !!status);
3473 }
3474 
3475 void amdgpu_ras_global_ras_isr(struct amdgpu_device *adev)
3476 {
3477 	if (atomic_cmpxchg(&amdgpu_ras_in_intr, 0, 1) == 0) {
3478 		struct amdgpu_ras *ras = amdgpu_ras_get_context(adev);
3479 
3480 		dev_info(adev->dev, "uncorrectable hardware error"
3481 			"(ERREVENT_ATHUB_INTERRUPT) detected!\n");
3482 
3483 		ras->gpu_reset_flags |= AMDGPU_RAS_GPU_RESET_MODE1_RESET;
3484 		amdgpu_ras_reset_gpu(adev);
3485 	}
3486 }
3487 
3488 bool amdgpu_ras_need_emergency_restart(struct amdgpu_device *adev)
3489 {
3490 	if (adev->asic_type == CHIP_VEGA20 &&
3491 	    adev->pm.fw_version <= 0x283400) {
3492 		return !(amdgpu_asic_reset_method(adev) == AMD_RESET_METHOD_BACO) &&
3493 				amdgpu_ras_intr_triggered();
3494 	}
3495 
3496 	return false;
3497 }
3498 
3499 void amdgpu_release_ras_context(struct amdgpu_device *adev)
3500 {
3501 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3502 
3503 	if (!con)
3504 		return;
3505 
3506 	if (!adev->ras_enabled && con->features & BIT(AMDGPU_RAS_BLOCK__GFX)) {
3507 		con->features &= ~BIT(AMDGPU_RAS_BLOCK__GFX);
3508 		amdgpu_ras_set_context(adev, NULL);
3509 		kfree(con);
3510 	}
3511 }
3512 
3513 #ifdef CONFIG_X86_MCE_AMD
3514 static struct amdgpu_device *find_adev(uint32_t node_id)
3515 {
3516 	int i;
3517 	struct amdgpu_device *adev = NULL;
3518 
3519 	for (i = 0; i < mce_adev_list.num_gpu; i++) {
3520 		adev = mce_adev_list.devs[i];
3521 
3522 		if (adev && adev->gmc.xgmi.connected_to_cpu &&
3523 		    adev->gmc.xgmi.physical_node_id == node_id)
3524 			break;
3525 		adev = NULL;
3526 	}
3527 
3528 	return adev;
3529 }
3530 
3531 #define GET_MCA_IPID_GPUID(m)	(((m) >> 44) & 0xF)
3532 #define GET_UMC_INST(m)		(((m) >> 21) & 0x7)
3533 #define GET_CHAN_INDEX(m)	((((m) >> 12) & 0x3) | (((m) >> 18) & 0x4))
3534 #define GPU_ID_OFFSET		8
3535 
3536 static int amdgpu_bad_page_notifier(struct notifier_block *nb,
3537 				    unsigned long val, void *data)
3538 {
3539 	struct mce *m = (struct mce *)data;
3540 	struct amdgpu_device *adev = NULL;
3541 	uint32_t gpu_id = 0;
3542 	uint32_t umc_inst = 0, ch_inst = 0;
3543 
3544 	/*
3545 	 * If the error was generated in UMC_V2, which belongs to GPU UMCs,
3546 	 * and error occurred in DramECC (Extended error code = 0) then only
3547 	 * process the error, else bail out.
3548 	 */
3549 	if (!m || !((smca_get_bank_type(m->extcpu, m->bank) == SMCA_UMC_V2) &&
3550 		    (XEC(m->status, 0x3f) == 0x0)))
3551 		return NOTIFY_DONE;
3552 
3553 	/*
3554 	 * If it is correctable error, return.
3555 	 */
3556 	if (mce_is_correctable(m))
3557 		return NOTIFY_OK;
3558 
3559 	/*
3560 	 * GPU Id is offset by GPU_ID_OFFSET in MCA_IPID_UMC register.
3561 	 */
3562 	gpu_id = GET_MCA_IPID_GPUID(m->ipid) - GPU_ID_OFFSET;
3563 
3564 	adev = find_adev(gpu_id);
3565 	if (!adev) {
3566 		DRM_WARN("%s: Unable to find adev for gpu_id: %d\n", __func__,
3567 								gpu_id);
3568 		return NOTIFY_DONE;
3569 	}
3570 
3571 	/*
3572 	 * If it is uncorrectable error, then find out UMC instance and
3573 	 * channel index.
3574 	 */
3575 	umc_inst = GET_UMC_INST(m->ipid);
3576 	ch_inst = GET_CHAN_INDEX(m->ipid);
3577 
3578 	dev_info(adev->dev, "Uncorrectable error detected in UMC inst: %d, chan_idx: %d",
3579 			     umc_inst, ch_inst);
3580 
3581 	if (!amdgpu_umc_page_retirement_mca(adev, m->addr, ch_inst, umc_inst))
3582 		return NOTIFY_OK;
3583 	else
3584 		return NOTIFY_DONE;
3585 }
3586 
3587 static struct notifier_block amdgpu_bad_page_nb = {
3588 	.notifier_call  = amdgpu_bad_page_notifier,
3589 	.priority       = MCE_PRIO_UC,
3590 };
3591 
3592 static void amdgpu_register_bad_pages_mca_notifier(struct amdgpu_device *adev)
3593 {
3594 	/*
3595 	 * Add the adev to the mce_adev_list.
3596 	 * During mode2 reset, amdgpu device is temporarily
3597 	 * removed from the mgpu_info list which can cause
3598 	 * page retirement to fail.
3599 	 * Use this list instead of mgpu_info to find the amdgpu
3600 	 * device on which the UMC error was reported.
3601 	 */
3602 	mce_adev_list.devs[mce_adev_list.num_gpu++] = adev;
3603 
3604 	/*
3605 	 * Register the x86 notifier only once
3606 	 * with MCE subsystem.
3607 	 */
3608 	if (notifier_registered == false) {
3609 		mce_register_decode_chain(&amdgpu_bad_page_nb);
3610 		notifier_registered = true;
3611 	}
3612 }
3613 #endif
3614 
3615 struct amdgpu_ras *amdgpu_ras_get_context(struct amdgpu_device *adev)
3616 {
3617 	if (!adev)
3618 		return NULL;
3619 
3620 	return adev->psp.ras_context.ras;
3621 }
3622 
3623 int amdgpu_ras_set_context(struct amdgpu_device *adev, struct amdgpu_ras *ras_con)
3624 {
3625 	if (!adev)
3626 		return -EINVAL;
3627 
3628 	adev->psp.ras_context.ras = ras_con;
3629 	return 0;
3630 }
3631 
3632 /* check if ras is supported on block, say, sdma, gfx */
3633 int amdgpu_ras_is_supported(struct amdgpu_device *adev,
3634 		unsigned int block)
3635 {
3636 	int ret = 0;
3637 	struct amdgpu_ras *ras = amdgpu_ras_get_context(adev);
3638 
3639 	if (block >= AMDGPU_RAS_BLOCK_COUNT)
3640 		return 0;
3641 
3642 	ret = ras && (adev->ras_enabled & (1 << block));
3643 
3644 	/* For the special asic with mem ecc enabled but sram ecc
3645 	 * not enabled, even if the ras block is not supported on
3646 	 * .ras_enabled, if the asic supports poison mode and the
3647 	 * ras block has ras configuration, it can be considered
3648 	 * that the ras block supports ras function.
3649 	 */
3650 	if (!ret &&
3651 	    (block == AMDGPU_RAS_BLOCK__GFX ||
3652 	     block == AMDGPU_RAS_BLOCK__SDMA ||
3653 	     block == AMDGPU_RAS_BLOCK__VCN ||
3654 	     block == AMDGPU_RAS_BLOCK__JPEG) &&
3655 		(amdgpu_ras_mask & (1 << block)) &&
3656 	    amdgpu_ras_is_poison_mode_supported(adev) &&
3657 	    amdgpu_ras_get_ras_block(adev, block, 0))
3658 		ret = 1;
3659 
3660 	return ret;
3661 }
3662 
3663 int amdgpu_ras_reset_gpu(struct amdgpu_device *adev)
3664 {
3665 	struct amdgpu_ras *ras = amdgpu_ras_get_context(adev);
3666 
3667 	if (atomic_cmpxchg(&ras->in_recovery, 0, 1) == 0)
3668 		amdgpu_reset_domain_schedule(ras->adev->reset_domain, &ras->recovery_work);
3669 	return 0;
3670 }
3671 
3672 int amdgpu_ras_set_mca_debug_mode(struct amdgpu_device *adev, bool enable)
3673 {
3674 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3675 	int ret = 0;
3676 
3677 	if (con) {
3678 		ret = amdgpu_mca_smu_set_debug_mode(adev, enable);
3679 		if (!ret)
3680 			con->is_aca_debug_mode = enable;
3681 	}
3682 
3683 	return ret;
3684 }
3685 
3686 int amdgpu_ras_set_aca_debug_mode(struct amdgpu_device *adev, bool enable)
3687 {
3688 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3689 	int ret = 0;
3690 
3691 	if (con) {
3692 		if (amdgpu_aca_is_enabled(adev))
3693 			ret = amdgpu_aca_smu_set_debug_mode(adev, enable);
3694 		else
3695 			ret = amdgpu_mca_smu_set_debug_mode(adev, enable);
3696 		if (!ret)
3697 			con->is_aca_debug_mode = enable;
3698 	}
3699 
3700 	return ret;
3701 }
3702 
3703 bool amdgpu_ras_get_aca_debug_mode(struct amdgpu_device *adev)
3704 {
3705 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3706 	const struct aca_smu_funcs *smu_funcs = adev->aca.smu_funcs;
3707 	const struct amdgpu_mca_smu_funcs *mca_funcs = adev->mca.mca_funcs;
3708 
3709 	if (!con)
3710 		return false;
3711 
3712 	if ((amdgpu_aca_is_enabled(adev) && smu_funcs && smu_funcs->set_debug_mode) ||
3713 	    (!amdgpu_aca_is_enabled(adev) && mca_funcs && mca_funcs->mca_set_debug_mode))
3714 		return con->is_aca_debug_mode;
3715 	else
3716 		return true;
3717 }
3718 
3719 bool amdgpu_ras_get_error_query_mode(struct amdgpu_device *adev,
3720 				     unsigned int *error_query_mode)
3721 {
3722 	struct amdgpu_ras *con = amdgpu_ras_get_context(adev);
3723 	const struct amdgpu_mca_smu_funcs *mca_funcs = adev->mca.mca_funcs;
3724 	const struct aca_smu_funcs *smu_funcs = adev->aca.smu_funcs;
3725 
3726 	if (!con) {
3727 		*error_query_mode = AMDGPU_RAS_INVALID_ERROR_QUERY;
3728 		return false;
3729 	}
3730 
3731 	if ((smu_funcs && smu_funcs->set_debug_mode) || (mca_funcs && mca_funcs->mca_set_debug_mode))
3732 		*error_query_mode =
3733 			(con->is_aca_debug_mode) ? AMDGPU_RAS_DIRECT_ERROR_QUERY : AMDGPU_RAS_FIRMWARE_ERROR_QUERY;
3734 	else
3735 		*error_query_mode = AMDGPU_RAS_DIRECT_ERROR_QUERY;
3736 
3737 	return true;
3738 }
3739 
3740 /* Register each ip ras block into amdgpu ras */
3741 int amdgpu_ras_register_ras_block(struct amdgpu_device *adev,
3742 		struct amdgpu_ras_block_object *ras_block_obj)
3743 {
3744 	struct amdgpu_ras_block_list *ras_node;
3745 	if (!adev || !ras_block_obj)
3746 		return -EINVAL;
3747 
3748 	ras_node = kzalloc(sizeof(*ras_node), GFP_KERNEL);
3749 	if (!ras_node)
3750 		return -ENOMEM;
3751 
3752 	INIT_LIST_HEAD(&ras_node->node);
3753 	ras_node->ras_obj = ras_block_obj;
3754 	list_add_tail(&ras_node->node, &adev->ras_list);
3755 
3756 	return 0;
3757 }
3758 
3759 void amdgpu_ras_get_error_type_name(uint32_t err_type, char *err_type_name)
3760 {
3761 	if (!err_type_name)
3762 		return;
3763 
3764 	switch (err_type) {
3765 	case AMDGPU_RAS_ERROR__SINGLE_CORRECTABLE:
3766 		sprintf(err_type_name, "correctable");
3767 		break;
3768 	case AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE:
3769 		sprintf(err_type_name, "uncorrectable");
3770 		break;
3771 	default:
3772 		sprintf(err_type_name, "unknown");
3773 		break;
3774 	}
3775 }
3776 
3777 bool amdgpu_ras_inst_get_memory_id_field(struct amdgpu_device *adev,
3778 					 const struct amdgpu_ras_err_status_reg_entry *reg_entry,
3779 					 uint32_t instance,
3780 					 uint32_t *memory_id)
3781 {
3782 	uint32_t err_status_lo_data, err_status_lo_offset;
3783 
3784 	if (!reg_entry)
3785 		return false;
3786 
3787 	err_status_lo_offset =
3788 		AMDGPU_RAS_REG_ENTRY_OFFSET(reg_entry->hwip, instance,
3789 					    reg_entry->seg_lo, reg_entry->reg_lo);
3790 	err_status_lo_data = RREG32(err_status_lo_offset);
3791 
3792 	if ((reg_entry->flags & AMDGPU_RAS_ERR_STATUS_VALID) &&
3793 	    !REG_GET_FIELD(err_status_lo_data, ERR_STATUS_LO, ERR_STATUS_VALID_FLAG))
3794 		return false;
3795 
3796 	*memory_id = REG_GET_FIELD(err_status_lo_data, ERR_STATUS_LO, MEMORY_ID);
3797 
3798 	return true;
3799 }
3800 
3801 bool amdgpu_ras_inst_get_err_cnt_field(struct amdgpu_device *adev,
3802 				       const struct amdgpu_ras_err_status_reg_entry *reg_entry,
3803 				       uint32_t instance,
3804 				       unsigned long *err_cnt)
3805 {
3806 	uint32_t err_status_hi_data, err_status_hi_offset;
3807 
3808 	if (!reg_entry)
3809 		return false;
3810 
3811 	err_status_hi_offset =
3812 		AMDGPU_RAS_REG_ENTRY_OFFSET(reg_entry->hwip, instance,
3813 					    reg_entry->seg_hi, reg_entry->reg_hi);
3814 	err_status_hi_data = RREG32(err_status_hi_offset);
3815 
3816 	if ((reg_entry->flags & AMDGPU_RAS_ERR_INFO_VALID) &&
3817 	    !REG_GET_FIELD(err_status_hi_data, ERR_STATUS_HI, ERR_INFO_VALID_FLAG))
3818 		/* keep the check here in case we need to refer to the result later */
3819 		dev_dbg(adev->dev, "Invalid err_info field\n");
3820 
3821 	/* read err count */
3822 	*err_cnt = REG_GET_FIELD(err_status_hi_data, ERR_STATUS, ERR_CNT);
3823 
3824 	return true;
3825 }
3826 
3827 void amdgpu_ras_inst_query_ras_error_count(struct amdgpu_device *adev,
3828 					   const struct amdgpu_ras_err_status_reg_entry *reg_list,
3829 					   uint32_t reg_list_size,
3830 					   const struct amdgpu_ras_memory_id_entry *mem_list,
3831 					   uint32_t mem_list_size,
3832 					   uint32_t instance,
3833 					   uint32_t err_type,
3834 					   unsigned long *err_count)
3835 {
3836 	uint32_t memory_id;
3837 	unsigned long err_cnt;
3838 	char err_type_name[16];
3839 	uint32_t i, j;
3840 
3841 	for (i = 0; i < reg_list_size; i++) {
3842 		/* query memory_id from err_status_lo */
3843 		if (!amdgpu_ras_inst_get_memory_id_field(adev, &reg_list[i],
3844 							 instance, &memory_id))
3845 			continue;
3846 
3847 		/* query err_cnt from err_status_hi */
3848 		if (!amdgpu_ras_inst_get_err_cnt_field(adev, &reg_list[i],
3849 						       instance, &err_cnt) ||
3850 		    !err_cnt)
3851 			continue;
3852 
3853 		*err_count += err_cnt;
3854 
3855 		/* log the errors */
3856 		amdgpu_ras_get_error_type_name(err_type, err_type_name);
3857 		if (!mem_list) {
3858 			/* memory_list is not supported */
3859 			dev_info(adev->dev,
3860 				 "%ld %s hardware errors detected in %s, instance: %d, memory_id: %d\n",
3861 				 err_cnt, err_type_name,
3862 				 reg_list[i].block_name,
3863 				 instance, memory_id);
3864 		} else {
3865 			for (j = 0; j < mem_list_size; j++) {
3866 				if (memory_id == mem_list[j].memory_id) {
3867 					dev_info(adev->dev,
3868 						 "%ld %s hardware errors detected in %s, instance: %d, memory block: %s\n",
3869 						 err_cnt, err_type_name,
3870 						 reg_list[i].block_name,
3871 						 instance, mem_list[j].name);
3872 					break;
3873 				}
3874 			}
3875 		}
3876 	}
3877 }
3878 
3879 void amdgpu_ras_inst_reset_ras_error_count(struct amdgpu_device *adev,
3880 					   const struct amdgpu_ras_err_status_reg_entry *reg_list,
3881 					   uint32_t reg_list_size,
3882 					   uint32_t instance)
3883 {
3884 	uint32_t err_status_lo_offset, err_status_hi_offset;
3885 	uint32_t i;
3886 
3887 	for (i = 0; i < reg_list_size; i++) {
3888 		err_status_lo_offset =
3889 			AMDGPU_RAS_REG_ENTRY_OFFSET(reg_list[i].hwip, instance,
3890 						    reg_list[i].seg_lo, reg_list[i].reg_lo);
3891 		err_status_hi_offset =
3892 			AMDGPU_RAS_REG_ENTRY_OFFSET(reg_list[i].hwip, instance,
3893 						    reg_list[i].seg_hi, reg_list[i].reg_hi);
3894 		WREG32(err_status_lo_offset, 0);
3895 		WREG32(err_status_hi_offset, 0);
3896 	}
3897 }
3898 
3899 int amdgpu_ras_error_data_init(struct ras_err_data *err_data)
3900 {
3901 	memset(err_data, 0, sizeof(*err_data));
3902 
3903 	INIT_LIST_HEAD(&err_data->err_node_list);
3904 
3905 	return 0;
3906 }
3907 
3908 static void amdgpu_ras_error_node_release(struct ras_err_node *err_node)
3909 {
3910 	if (!err_node)
3911 		return;
3912 
3913 	list_del(&err_node->node);
3914 	kvfree(err_node);
3915 }
3916 
3917 void amdgpu_ras_error_data_fini(struct ras_err_data *err_data)
3918 {
3919 	struct ras_err_node *err_node, *tmp;
3920 
3921 	list_for_each_entry_safe(err_node, tmp, &err_data->err_node_list, node)
3922 		amdgpu_ras_error_node_release(err_node);
3923 }
3924 
3925 static struct ras_err_node *amdgpu_ras_error_find_node_by_id(struct ras_err_data *err_data,
3926 							     struct amdgpu_smuio_mcm_config_info *mcm_info)
3927 {
3928 	struct ras_err_node *err_node;
3929 	struct amdgpu_smuio_mcm_config_info *ref_id;
3930 
3931 	if (!err_data || !mcm_info)
3932 		return NULL;
3933 
3934 	for_each_ras_error(err_node, err_data) {
3935 		ref_id = &err_node->err_info.mcm_info;
3936 
3937 		if (mcm_info->socket_id == ref_id->socket_id &&
3938 		    mcm_info->die_id == ref_id->die_id)
3939 			return err_node;
3940 	}
3941 
3942 	return NULL;
3943 }
3944 
3945 static struct ras_err_node *amdgpu_ras_error_node_new(void)
3946 {
3947 	struct ras_err_node *err_node;
3948 
3949 	err_node = kvzalloc(sizeof(*err_node), GFP_KERNEL);
3950 	if (!err_node)
3951 		return NULL;
3952 
3953 	INIT_LIST_HEAD(&err_node->node);
3954 
3955 	return err_node;
3956 }
3957 
3958 static int ras_err_info_cmp(void *priv, const struct list_head *a, const struct list_head *b)
3959 {
3960 	struct ras_err_node *nodea = container_of(a, struct ras_err_node, node);
3961 	struct ras_err_node *nodeb = container_of(b, struct ras_err_node, node);
3962 	struct amdgpu_smuio_mcm_config_info *infoa = &nodea->err_info.mcm_info;
3963 	struct amdgpu_smuio_mcm_config_info *infob = &nodeb->err_info.mcm_info;
3964 
3965 	if (unlikely(infoa->socket_id != infob->socket_id))
3966 		return infoa->socket_id - infob->socket_id;
3967 	else
3968 		return infoa->die_id - infob->die_id;
3969 
3970 	return 0;
3971 }
3972 
3973 static struct ras_err_info *amdgpu_ras_error_get_info(struct ras_err_data *err_data,
3974 				struct amdgpu_smuio_mcm_config_info *mcm_info)
3975 {
3976 	struct ras_err_node *err_node;
3977 
3978 	err_node = amdgpu_ras_error_find_node_by_id(err_data, mcm_info);
3979 	if (err_node)
3980 		return &err_node->err_info;
3981 
3982 	err_node = amdgpu_ras_error_node_new();
3983 	if (!err_node)
3984 		return NULL;
3985 
3986 	INIT_LIST_HEAD(&err_node->err_info.err_addr_list);
3987 
3988 	memcpy(&err_node->err_info.mcm_info, mcm_info, sizeof(*mcm_info));
3989 
3990 	err_data->err_list_count++;
3991 	list_add_tail(&err_node->node, &err_data->err_node_list);
3992 	list_sort(NULL, &err_data->err_node_list, ras_err_info_cmp);
3993 
3994 	return &err_node->err_info;
3995 }
3996 
3997 void amdgpu_ras_add_mca_err_addr(struct ras_err_info *err_info, struct ras_err_addr *err_addr)
3998 {
3999 	struct ras_err_addr *mca_err_addr;
4000 
4001 	mca_err_addr = kzalloc(sizeof(*mca_err_addr), GFP_KERNEL);
4002 	if (!mca_err_addr)
4003 		return;
4004 
4005 	INIT_LIST_HEAD(&mca_err_addr->node);
4006 
4007 	mca_err_addr->err_status = err_addr->err_status;
4008 	mca_err_addr->err_ipid = err_addr->err_ipid;
4009 	mca_err_addr->err_addr = err_addr->err_addr;
4010 
4011 	list_add_tail(&mca_err_addr->node, &err_info->err_addr_list);
4012 }
4013 
4014 void amdgpu_ras_del_mca_err_addr(struct ras_err_info *err_info, struct ras_err_addr *mca_err_addr)
4015 {
4016 	list_del(&mca_err_addr->node);
4017 	kfree(mca_err_addr);
4018 }
4019 
4020 int amdgpu_ras_error_statistic_ue_count(struct ras_err_data *err_data,
4021 		struct amdgpu_smuio_mcm_config_info *mcm_info,
4022 		struct ras_err_addr *err_addr, u64 count)
4023 {
4024 	struct ras_err_info *err_info;
4025 
4026 	if (!err_data || !mcm_info)
4027 		return -EINVAL;
4028 
4029 	if (!count)
4030 		return 0;
4031 
4032 	err_info = amdgpu_ras_error_get_info(err_data, mcm_info);
4033 	if (!err_info)
4034 		return -EINVAL;
4035 
4036 	if (err_addr && err_addr->err_status)
4037 		amdgpu_ras_add_mca_err_addr(err_info, err_addr);
4038 
4039 	err_info->ue_count += count;
4040 	err_data->ue_count += count;
4041 
4042 	return 0;
4043 }
4044 
4045 int amdgpu_ras_error_statistic_ce_count(struct ras_err_data *err_data,
4046 		struct amdgpu_smuio_mcm_config_info *mcm_info,
4047 		struct ras_err_addr *err_addr, u64 count)
4048 {
4049 	struct ras_err_info *err_info;
4050 
4051 	if (!err_data || !mcm_info)
4052 		return -EINVAL;
4053 
4054 	if (!count)
4055 		return 0;
4056 
4057 	err_info = amdgpu_ras_error_get_info(err_data, mcm_info);
4058 	if (!err_info)
4059 		return -EINVAL;
4060 
4061 	err_info->ce_count += count;
4062 	err_data->ce_count += count;
4063 
4064 	return 0;
4065 }
4066 
4067 int amdgpu_ras_error_statistic_de_count(struct ras_err_data *err_data,
4068 		struct amdgpu_smuio_mcm_config_info *mcm_info,
4069 		struct ras_err_addr *err_addr, u64 count)
4070 {
4071 	struct ras_err_info *err_info;
4072 
4073 	if (!err_data || !mcm_info)
4074 		return -EINVAL;
4075 
4076 	if (!count)
4077 		return 0;
4078 
4079 	err_info = amdgpu_ras_error_get_info(err_data, mcm_info);
4080 	if (!err_info)
4081 		return -EINVAL;
4082 
4083 	if (err_addr && err_addr->err_status)
4084 		amdgpu_ras_add_mca_err_addr(err_info, err_addr);
4085 
4086 	err_info->de_count += count;
4087 	err_data->de_count += count;
4088 
4089 	return 0;
4090 }
4091 
4092 #define mmMP0_SMN_C2PMSG_92	0x1609C
4093 #define mmMP0_SMN_C2PMSG_126	0x160BE
4094 static void amdgpu_ras_boot_time_error_reporting(struct amdgpu_device *adev,
4095 						 u32 instance, u32 boot_error)
4096 {
4097 	u32 socket_id, aid_id, hbm_id;
4098 	u32 reg_data;
4099 	u64 reg_addr;
4100 
4101 	socket_id = AMDGPU_RAS_GPU_ERR_SOCKET_ID(boot_error);
4102 	aid_id = AMDGPU_RAS_GPU_ERR_AID_ID(boot_error);
4103 	hbm_id = AMDGPU_RAS_GPU_ERR_HBM_ID(boot_error);
4104 
4105 	/* The pattern for smn addressing in other SOC could be different from
4106 	 * the one for aqua_vanjaram. We should revisit the code if the pattern
4107 	 * is changed. In such case, replace the aqua_vanjaram implementation
4108 	 * with more common helper */
4109 	reg_addr = (mmMP0_SMN_C2PMSG_92 << 2) +
4110 		   aqua_vanjaram_encode_ext_smn_addressing(instance);
4111 
4112 	reg_data = amdgpu_device_indirect_rreg_ext(adev, reg_addr);
4113 	dev_err(adev->dev, "socket: %d, aid: %d, firmware boot failed, fw status is 0x%x\n",
4114 		socket_id, aid_id, reg_data);
4115 
4116 	if (AMDGPU_RAS_GPU_ERR_MEM_TRAINING(boot_error))
4117 		dev_info(adev->dev, "socket: %d, aid: %d, hbm: %d, memory training failed\n",
4118 			 socket_id, aid_id, hbm_id);
4119 
4120 	if (AMDGPU_RAS_GPU_ERR_FW_LOAD(boot_error))
4121 		dev_info(adev->dev, "socket: %d, aid: %d, firmware load failed at boot time\n",
4122 			 socket_id, aid_id);
4123 
4124 	if (AMDGPU_RAS_GPU_ERR_WAFL_LINK_TRAINING(boot_error))
4125 		dev_info(adev->dev, "socket: %d, aid: %d, wafl link training failed\n",
4126 			 socket_id, aid_id);
4127 
4128 	if (AMDGPU_RAS_GPU_ERR_XGMI_LINK_TRAINING(boot_error))
4129 		dev_info(adev->dev, "socket: %d, aid: %d, xgmi link training failed\n",
4130 			 socket_id, aid_id);
4131 
4132 	if (AMDGPU_RAS_GPU_ERR_USR_CP_LINK_TRAINING(boot_error))
4133 		dev_info(adev->dev, "socket: %d, aid: %d, usr cp link training failed\n",
4134 			 socket_id, aid_id);
4135 
4136 	if (AMDGPU_RAS_GPU_ERR_USR_DP_LINK_TRAINING(boot_error))
4137 		dev_info(adev->dev, "socket: %d, aid: %d, usr dp link training failed\n",
4138 			 socket_id, aid_id);
4139 
4140 	if (AMDGPU_RAS_GPU_ERR_HBM_MEM_TEST(boot_error))
4141 		dev_info(adev->dev, "socket: %d, aid: %d, hbm: %d, hbm memory test failed\n",
4142 			 socket_id, aid_id, hbm_id);
4143 
4144 	if (AMDGPU_RAS_GPU_ERR_HBM_BIST_TEST(boot_error))
4145 		dev_info(adev->dev, "socket: %d, aid: %d, hbm: %d, hbm bist test failed\n",
4146 			 socket_id, aid_id, hbm_id);
4147 }
4148 
4149 static int amdgpu_ras_wait_for_boot_complete(struct amdgpu_device *adev,
4150 					     u32 instance, u32 *boot_error)
4151 {
4152 	u32 reg_addr;
4153 	u32 reg_data;
4154 	int retry_loop;
4155 
4156 	reg_addr = (mmMP0_SMN_C2PMSG_92 << 2) +
4157 		   aqua_vanjaram_encode_ext_smn_addressing(instance);
4158 
4159 	for (retry_loop = 0; retry_loop < AMDGPU_RAS_BOOT_STATUS_POLLING_LIMIT; retry_loop++) {
4160 		reg_data = amdgpu_device_indirect_rreg_ext(adev, reg_addr);
4161 		if ((reg_data & AMDGPU_RAS_BOOT_STATUS_MASK) == AMDGPU_RAS_BOOT_STEADY_STATUS) {
4162 			*boot_error = AMDGPU_RAS_BOOT_SUCEESS;
4163 			return 0;
4164 		}
4165 		msleep(1);
4166 	}
4167 
4168 	/* The pattern for smn addressing in other SOC could be different from
4169 	 * the one for aqua_vanjaram. We should revisit the code if the pattern
4170 	 * is changed. In such case, replace the aqua_vanjaram implementation
4171 	 * with more common helper */
4172 	reg_addr = (mmMP0_SMN_C2PMSG_126 << 2) +
4173 		   aqua_vanjaram_encode_ext_smn_addressing(instance);
4174 
4175 	for (retry_loop = 0; retry_loop < AMDGPU_RAS_BOOT_STATUS_POLLING_LIMIT; retry_loop++) {
4176 		reg_data = amdgpu_device_indirect_rreg_ext(adev, reg_addr);
4177 		if (AMDGPU_RAS_GPU_ERR_BOOT_STATUS(reg_data)) {
4178 			*boot_error = reg_data;
4179 			return 0;
4180 		}
4181 		msleep(1);
4182 	}
4183 
4184 	*boot_error = reg_data;
4185 	return -ETIME;
4186 }
4187 
4188 void amdgpu_ras_query_boot_status(struct amdgpu_device *adev, u32 num_instances)
4189 {
4190 	u32 boot_error = 0;
4191 	u32 i;
4192 
4193 	for (i = 0; i < num_instances; i++) {
4194 		if (amdgpu_ras_wait_for_boot_complete(adev, i, &boot_error))
4195 			amdgpu_ras_boot_time_error_reporting(adev, i, boot_error);
4196 	}
4197 }
4198