1 /*
2  * Copyright 2014 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/firmware.h>
25 #include <linux/slab.h>
26 #include <linux/module.h>
27 
28 #include "amdgpu.h"
29 #include "amdgpu_ucode.h"
30 
31 static void amdgpu_ucode_print_common_hdr(const struct common_firmware_header *hdr)
32 {
33 	DRM_DEBUG("size_bytes: %u\n", le32_to_cpu(hdr->size_bytes));
34 	DRM_DEBUG("header_size_bytes: %u\n", le32_to_cpu(hdr->header_size_bytes));
35 	DRM_DEBUG("header_version_major: %u\n", le16_to_cpu(hdr->header_version_major));
36 	DRM_DEBUG("header_version_minor: %u\n", le16_to_cpu(hdr->header_version_minor));
37 	DRM_DEBUG("ip_version_major: %u\n", le16_to_cpu(hdr->ip_version_major));
38 	DRM_DEBUG("ip_version_minor: %u\n", le16_to_cpu(hdr->ip_version_minor));
39 	DRM_DEBUG("ucode_version: 0x%08x\n", le32_to_cpu(hdr->ucode_version));
40 	DRM_DEBUG("ucode_size_bytes: %u\n", le32_to_cpu(hdr->ucode_size_bytes));
41 	DRM_DEBUG("ucode_array_offset_bytes: %u\n",
42 		  le32_to_cpu(hdr->ucode_array_offset_bytes));
43 	DRM_DEBUG("crc32: 0x%08x\n", le32_to_cpu(hdr->crc32));
44 }
45 
46 void amdgpu_ucode_print_mc_hdr(const struct common_firmware_header *hdr)
47 {
48 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
49 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
50 
51 	DRM_DEBUG("MC\n");
52 	amdgpu_ucode_print_common_hdr(hdr);
53 
54 	if (version_major == 1) {
55 		const struct mc_firmware_header_v1_0 *mc_hdr =
56 			container_of(hdr, struct mc_firmware_header_v1_0, header);
57 
58 		DRM_DEBUG("io_debug_size_bytes: %u\n",
59 			  le32_to_cpu(mc_hdr->io_debug_size_bytes));
60 		DRM_DEBUG("io_debug_array_offset_bytes: %u\n",
61 			  le32_to_cpu(mc_hdr->io_debug_array_offset_bytes));
62 	} else {
63 		DRM_ERROR("Unknown MC ucode version: %u.%u\n", version_major, version_minor);
64 	}
65 }
66 
67 void amdgpu_ucode_print_smc_hdr(const struct common_firmware_header *hdr)
68 {
69 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
70 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
71 
72 	DRM_DEBUG("SMC\n");
73 	amdgpu_ucode_print_common_hdr(hdr);
74 
75 	if (version_major == 1) {
76 		const struct smc_firmware_header_v1_0 *smc_hdr =
77 			container_of(hdr, struct smc_firmware_header_v1_0, header);
78 
79 		DRM_DEBUG("ucode_start_addr: %u\n", le32_to_cpu(smc_hdr->ucode_start_addr));
80 	} else if (version_major == 2) {
81 		const struct smc_firmware_header_v1_0 *v1_hdr =
82 			container_of(hdr, struct smc_firmware_header_v1_0, header);
83 		const struct smc_firmware_header_v2_0 *v2_hdr =
84 			container_of(v1_hdr, struct smc_firmware_header_v2_0, v1_0);
85 
86 		DRM_INFO("ppt_offset_bytes: %u\n", le32_to_cpu(v2_hdr->ppt_offset_bytes));
87 		DRM_INFO("ppt_size_bytes: %u\n", le32_to_cpu(v2_hdr->ppt_size_bytes));
88 	} else {
89 		DRM_ERROR("Unknown SMC ucode version: %u.%u\n", version_major, version_minor);
90 	}
91 }
92 
93 void amdgpu_ucode_print_gfx_hdr(const struct common_firmware_header *hdr)
94 {
95 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
96 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
97 
98 	DRM_DEBUG("GFX\n");
99 	amdgpu_ucode_print_common_hdr(hdr);
100 
101 	if (version_major == 1) {
102 		const struct gfx_firmware_header_v1_0 *gfx_hdr =
103 			container_of(hdr, struct gfx_firmware_header_v1_0, header);
104 
105 		DRM_DEBUG("ucode_feature_version: %u\n",
106 			  le32_to_cpu(gfx_hdr->ucode_feature_version));
107 		DRM_DEBUG("jt_offset: %u\n", le32_to_cpu(gfx_hdr->jt_offset));
108 		DRM_DEBUG("jt_size: %u\n", le32_to_cpu(gfx_hdr->jt_size));
109 	} else {
110 		DRM_ERROR("Unknown GFX ucode version: %u.%u\n", version_major, version_minor);
111 	}
112 }
113 
114 void amdgpu_ucode_print_rlc_hdr(const struct common_firmware_header *hdr)
115 {
116 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
117 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
118 
119 	DRM_DEBUG("RLC\n");
120 	amdgpu_ucode_print_common_hdr(hdr);
121 
122 	if (version_major == 1) {
123 		const struct rlc_firmware_header_v1_0 *rlc_hdr =
124 			container_of(hdr, struct rlc_firmware_header_v1_0, header);
125 
126 		DRM_DEBUG("ucode_feature_version: %u\n",
127 			  le32_to_cpu(rlc_hdr->ucode_feature_version));
128 		DRM_DEBUG("save_and_restore_offset: %u\n",
129 			  le32_to_cpu(rlc_hdr->save_and_restore_offset));
130 		DRM_DEBUG("clear_state_descriptor_offset: %u\n",
131 			  le32_to_cpu(rlc_hdr->clear_state_descriptor_offset));
132 		DRM_DEBUG("avail_scratch_ram_locations: %u\n",
133 			  le32_to_cpu(rlc_hdr->avail_scratch_ram_locations));
134 		DRM_DEBUG("master_pkt_description_offset: %u\n",
135 			  le32_to_cpu(rlc_hdr->master_pkt_description_offset));
136 	} else if (version_major == 2) {
137 		const struct rlc_firmware_header_v2_0 *rlc_hdr =
138 			container_of(hdr, struct rlc_firmware_header_v2_0, header);
139 
140 		DRM_DEBUG("ucode_feature_version: %u\n",
141 			  le32_to_cpu(rlc_hdr->ucode_feature_version));
142 		DRM_DEBUG("jt_offset: %u\n", le32_to_cpu(rlc_hdr->jt_offset));
143 		DRM_DEBUG("jt_size: %u\n", le32_to_cpu(rlc_hdr->jt_size));
144 		DRM_DEBUG("save_and_restore_offset: %u\n",
145 			  le32_to_cpu(rlc_hdr->save_and_restore_offset));
146 		DRM_DEBUG("clear_state_descriptor_offset: %u\n",
147 			  le32_to_cpu(rlc_hdr->clear_state_descriptor_offset));
148 		DRM_DEBUG("avail_scratch_ram_locations: %u\n",
149 			  le32_to_cpu(rlc_hdr->avail_scratch_ram_locations));
150 		DRM_DEBUG("reg_restore_list_size: %u\n",
151 			  le32_to_cpu(rlc_hdr->reg_restore_list_size));
152 		DRM_DEBUG("reg_list_format_start: %u\n",
153 			  le32_to_cpu(rlc_hdr->reg_list_format_start));
154 		DRM_DEBUG("reg_list_format_separate_start: %u\n",
155 			  le32_to_cpu(rlc_hdr->reg_list_format_separate_start));
156 		DRM_DEBUG("starting_offsets_start: %u\n",
157 			  le32_to_cpu(rlc_hdr->starting_offsets_start));
158 		DRM_DEBUG("reg_list_format_size_bytes: %u\n",
159 			  le32_to_cpu(rlc_hdr->reg_list_format_size_bytes));
160 		DRM_DEBUG("reg_list_format_array_offset_bytes: %u\n",
161 			  le32_to_cpu(rlc_hdr->reg_list_format_array_offset_bytes));
162 		DRM_DEBUG("reg_list_size_bytes: %u\n",
163 			  le32_to_cpu(rlc_hdr->reg_list_size_bytes));
164 		DRM_DEBUG("reg_list_array_offset_bytes: %u\n",
165 			  le32_to_cpu(rlc_hdr->reg_list_array_offset_bytes));
166 		DRM_DEBUG("reg_list_format_separate_size_bytes: %u\n",
167 			  le32_to_cpu(rlc_hdr->reg_list_format_separate_size_bytes));
168 		DRM_DEBUG("reg_list_format_separate_array_offset_bytes: %u\n",
169 			  le32_to_cpu(rlc_hdr->reg_list_format_separate_array_offset_bytes));
170 		DRM_DEBUG("reg_list_separate_size_bytes: %u\n",
171 			  le32_to_cpu(rlc_hdr->reg_list_separate_size_bytes));
172 		DRM_DEBUG("reg_list_separate_array_offset_bytes: %u\n",
173 			  le32_to_cpu(rlc_hdr->reg_list_separate_array_offset_bytes));
174 		if (version_minor == 1) {
175 			const struct rlc_firmware_header_v2_1 *v2_1 =
176 				container_of(rlc_hdr, struct rlc_firmware_header_v2_1, v2_0);
177 			DRM_DEBUG("reg_list_format_direct_reg_list_length: %u\n",
178 				  le32_to_cpu(v2_1->reg_list_format_direct_reg_list_length));
179 			DRM_DEBUG("save_restore_list_cntl_ucode_ver: %u\n",
180 				  le32_to_cpu(v2_1->save_restore_list_cntl_ucode_ver));
181 			DRM_DEBUG("save_restore_list_cntl_feature_ver: %u\n",
182 				  le32_to_cpu(v2_1->save_restore_list_cntl_feature_ver));
183 			DRM_DEBUG("save_restore_list_cntl_size_bytes %u\n",
184 				  le32_to_cpu(v2_1->save_restore_list_cntl_size_bytes));
185 			DRM_DEBUG("save_restore_list_cntl_offset_bytes: %u\n",
186 				  le32_to_cpu(v2_1->save_restore_list_cntl_offset_bytes));
187 			DRM_DEBUG("save_restore_list_gpm_ucode_ver: %u\n",
188 				  le32_to_cpu(v2_1->save_restore_list_gpm_ucode_ver));
189 			DRM_DEBUG("save_restore_list_gpm_feature_ver: %u\n",
190 				  le32_to_cpu(v2_1->save_restore_list_gpm_feature_ver));
191 			DRM_DEBUG("save_restore_list_gpm_size_bytes %u\n",
192 				  le32_to_cpu(v2_1->save_restore_list_gpm_size_bytes));
193 			DRM_DEBUG("save_restore_list_gpm_offset_bytes: %u\n",
194 				  le32_to_cpu(v2_1->save_restore_list_gpm_offset_bytes));
195 			DRM_DEBUG("save_restore_list_srm_ucode_ver: %u\n",
196 				  le32_to_cpu(v2_1->save_restore_list_srm_ucode_ver));
197 			DRM_DEBUG("save_restore_list_srm_feature_ver: %u\n",
198 				  le32_to_cpu(v2_1->save_restore_list_srm_feature_ver));
199 			DRM_DEBUG("save_restore_list_srm_size_bytes %u\n",
200 				  le32_to_cpu(v2_1->save_restore_list_srm_size_bytes));
201 			DRM_DEBUG("save_restore_list_srm_offset_bytes: %u\n",
202 				  le32_to_cpu(v2_1->save_restore_list_srm_offset_bytes));
203 		}
204 	} else {
205 		DRM_ERROR("Unknown RLC ucode version: %u.%u\n", version_major, version_minor);
206 	}
207 }
208 
209 void amdgpu_ucode_print_sdma_hdr(const struct common_firmware_header *hdr)
210 {
211 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
212 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
213 
214 	DRM_DEBUG("SDMA\n");
215 	amdgpu_ucode_print_common_hdr(hdr);
216 
217 	if (version_major == 1) {
218 		const struct sdma_firmware_header_v1_0 *sdma_hdr =
219 			container_of(hdr, struct sdma_firmware_header_v1_0, header);
220 
221 		DRM_DEBUG("ucode_feature_version: %u\n",
222 			  le32_to_cpu(sdma_hdr->ucode_feature_version));
223 		DRM_DEBUG("ucode_change_version: %u\n",
224 			  le32_to_cpu(sdma_hdr->ucode_change_version));
225 		DRM_DEBUG("jt_offset: %u\n", le32_to_cpu(sdma_hdr->jt_offset));
226 		DRM_DEBUG("jt_size: %u\n", le32_to_cpu(sdma_hdr->jt_size));
227 		if (version_minor >= 1) {
228 			const struct sdma_firmware_header_v1_1 *sdma_v1_1_hdr =
229 				container_of(sdma_hdr, struct sdma_firmware_header_v1_1, v1_0);
230 			DRM_DEBUG("digest_size: %u\n", le32_to_cpu(sdma_v1_1_hdr->digest_size));
231 		}
232 	} else {
233 		DRM_ERROR("Unknown SDMA ucode version: %u.%u\n",
234 			  version_major, version_minor);
235 	}
236 }
237 
238 void amdgpu_ucode_print_psp_hdr(const struct common_firmware_header *hdr)
239 {
240 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
241 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
242 
243 	DRM_DEBUG("PSP\n");
244 	amdgpu_ucode_print_common_hdr(hdr);
245 
246 	if (version_major == 1) {
247 		const struct psp_firmware_header_v1_0 *psp_hdr =
248 			container_of(hdr, struct psp_firmware_header_v1_0, header);
249 
250 		DRM_DEBUG("ucode_feature_version: %u\n",
251 			  le32_to_cpu(psp_hdr->ucode_feature_version));
252 		DRM_DEBUG("sos_offset_bytes: %u\n",
253 			  le32_to_cpu(psp_hdr->sos_offset_bytes));
254 		DRM_DEBUG("sos_size_bytes: %u\n",
255 			  le32_to_cpu(psp_hdr->sos_size_bytes));
256 		if (version_minor == 1) {
257 			const struct psp_firmware_header_v1_1 *psp_hdr_v1_1 =
258 				container_of(psp_hdr, struct psp_firmware_header_v1_1, v1_0);
259 			DRM_DEBUG("toc_header_version: %u\n",
260 				  le32_to_cpu(psp_hdr_v1_1->toc_header_version));
261 			DRM_DEBUG("toc_offset_bytes: %u\n",
262 				  le32_to_cpu(psp_hdr_v1_1->toc_offset_bytes));
263 			DRM_DEBUG("toc_size_bytes: %u\n",
264 				  le32_to_cpu(psp_hdr_v1_1->toc_size_bytes));
265 			DRM_DEBUG("kdb_header_version: %u\n",
266 				  le32_to_cpu(psp_hdr_v1_1->kdb_header_version));
267 			DRM_DEBUG("kdb_offset_bytes: %u\n",
268 				  le32_to_cpu(psp_hdr_v1_1->kdb_offset_bytes));
269 			DRM_DEBUG("kdb_size_bytes: %u\n",
270 				  le32_to_cpu(psp_hdr_v1_1->kdb_size_bytes));
271 		}
272 	} else {
273 		DRM_ERROR("Unknown PSP ucode version: %u.%u\n",
274 			  version_major, version_minor);
275 	}
276 }
277 
278 void amdgpu_ucode_print_gpu_info_hdr(const struct common_firmware_header *hdr)
279 {
280 	uint16_t version_major = le16_to_cpu(hdr->header_version_major);
281 	uint16_t version_minor = le16_to_cpu(hdr->header_version_minor);
282 
283 	DRM_DEBUG("GPU_INFO\n");
284 	amdgpu_ucode_print_common_hdr(hdr);
285 
286 	if (version_major == 1) {
287 		const struct gpu_info_firmware_header_v1_0 *gpu_info_hdr =
288 			container_of(hdr, struct gpu_info_firmware_header_v1_0, header);
289 
290 		DRM_DEBUG("version_major: %u\n",
291 			  le16_to_cpu(gpu_info_hdr->version_major));
292 		DRM_DEBUG("version_minor: %u\n",
293 			  le16_to_cpu(gpu_info_hdr->version_minor));
294 	} else {
295 		DRM_ERROR("Unknown gpu_info ucode version: %u.%u\n", version_major, version_minor);
296 	}
297 }
298 
299 int amdgpu_ucode_validate(const struct firmware *fw)
300 {
301 	const struct common_firmware_header *hdr =
302 		(const struct common_firmware_header *)fw->data;
303 
304 	if (fw->size == le32_to_cpu(hdr->size_bytes))
305 		return 0;
306 
307 	return -EINVAL;
308 }
309 
310 bool amdgpu_ucode_hdr_version(union amdgpu_firmware_header *hdr,
311 				uint16_t hdr_major, uint16_t hdr_minor)
312 {
313 	if ((hdr->common.header_version_major == hdr_major) &&
314 		(hdr->common.header_version_minor == hdr_minor))
315 		return false;
316 	return true;
317 }
318 
319 enum amdgpu_firmware_load_type
320 amdgpu_ucode_get_load_type(struct amdgpu_device *adev, int load_type)
321 {
322 	switch (adev->asic_type) {
323 #ifdef CONFIG_DRM_AMDGPU_SI
324 	case CHIP_TAHITI:
325 	case CHIP_PITCAIRN:
326 	case CHIP_VERDE:
327 	case CHIP_OLAND:
328 	case CHIP_HAINAN:
329 		return AMDGPU_FW_LOAD_DIRECT;
330 #endif
331 #ifdef CONFIG_DRM_AMDGPU_CIK
332 	case CHIP_BONAIRE:
333 	case CHIP_KAVERI:
334 	case CHIP_KABINI:
335 	case CHIP_HAWAII:
336 	case CHIP_MULLINS:
337 		return AMDGPU_FW_LOAD_DIRECT;
338 #endif
339 	case CHIP_TOPAZ:
340 	case CHIP_TONGA:
341 	case CHIP_FIJI:
342 	case CHIP_CARRIZO:
343 	case CHIP_STONEY:
344 	case CHIP_POLARIS10:
345 	case CHIP_POLARIS11:
346 	case CHIP_POLARIS12:
347 	case CHIP_VEGAM:
348 		return AMDGPU_FW_LOAD_SMU;
349 	case CHIP_VEGA10:
350 	case CHIP_RAVEN:
351 	case CHIP_VEGA12:
352 	case CHIP_VEGA20:
353 	case CHIP_NAVI10:
354 		if (!load_type)
355 			return AMDGPU_FW_LOAD_DIRECT;
356 		else
357 			return AMDGPU_FW_LOAD_PSP;
358 	default:
359 		DRM_ERROR("Unknown firmware load type\n");
360 	}
361 
362 	return AMDGPU_FW_LOAD_DIRECT;
363 }
364 
365 #define FW_VERSION_ATTR(name, mode, field)				\
366 static ssize_t show_##name(struct device *dev,				\
367 			  struct device_attribute *attr,		\
368 			  char *buf)					\
369 {									\
370 	struct drm_device *ddev = dev_get_drvdata(dev);			\
371 	struct amdgpu_device *adev = ddev->dev_private;			\
372 									\
373 	return snprintf(buf, PAGE_SIZE, "0x%08x\n", adev->field);	\
374 }									\
375 static DEVICE_ATTR(name, mode, show_##name, NULL)
376 
377 FW_VERSION_ATTR(vce_fw_version, 0444, vce.fw_version);
378 FW_VERSION_ATTR(uvd_fw_version, 0444, uvd.fw_version);
379 FW_VERSION_ATTR(mc_fw_version, 0444, gmc.fw_version);
380 FW_VERSION_ATTR(me_fw_version, 0444, gfx.me_fw_version);
381 FW_VERSION_ATTR(pfp_fw_version, 0444, gfx.pfp_fw_version);
382 FW_VERSION_ATTR(ce_fw_version, 0444, gfx.ce_fw_version);
383 FW_VERSION_ATTR(rlc_fw_version, 0444, gfx.rlc_fw_version);
384 FW_VERSION_ATTR(rlc_srlc_fw_version, 0444, gfx.rlc_srlc_fw_version);
385 FW_VERSION_ATTR(rlc_srlg_fw_version, 0444, gfx.rlc_srlg_fw_version);
386 FW_VERSION_ATTR(rlc_srls_fw_version, 0444, gfx.rlc_srls_fw_version);
387 FW_VERSION_ATTR(mec_fw_version, 0444, gfx.mec_fw_version);
388 FW_VERSION_ATTR(mec2_fw_version, 0444, gfx.mec2_fw_version);
389 FW_VERSION_ATTR(sos_fw_version, 0444, psp.sos_fw_version);
390 FW_VERSION_ATTR(asd_fw_version, 0444, psp.asd_fw_version);
391 FW_VERSION_ATTR(ta_ras_fw_version, 0444, psp.ta_fw_version);
392 FW_VERSION_ATTR(ta_xgmi_fw_version, 0444, psp.ta_fw_version);
393 FW_VERSION_ATTR(smc_fw_version, 0444, pm.fw_version);
394 FW_VERSION_ATTR(sdma_fw_version, 0444, sdma.instance[0].fw_version);
395 FW_VERSION_ATTR(sdma2_fw_version, 0444, sdma.instance[1].fw_version);
396 FW_VERSION_ATTR(vcn_fw_version, 0444, vcn.fw_version);
397 FW_VERSION_ATTR(dmcu_fw_version, 0444, dm.dmcu_fw_version);
398 
399 static struct attribute *fw_attrs[] = {
400 	&dev_attr_vce_fw_version.attr, &dev_attr_uvd_fw_version.attr,
401 	&dev_attr_mc_fw_version.attr, &dev_attr_me_fw_version.attr,
402 	&dev_attr_pfp_fw_version.attr, &dev_attr_ce_fw_version.attr,
403 	&dev_attr_rlc_fw_version.attr, &dev_attr_rlc_srlc_fw_version.attr,
404 	&dev_attr_rlc_srlg_fw_version.attr, &dev_attr_rlc_srls_fw_version.attr,
405 	&dev_attr_mec_fw_version.attr, &dev_attr_mec2_fw_version.attr,
406 	&dev_attr_sos_fw_version.attr, &dev_attr_asd_fw_version.attr,
407 	&dev_attr_ta_ras_fw_version.attr, &dev_attr_ta_xgmi_fw_version.attr,
408 	&dev_attr_smc_fw_version.attr, &dev_attr_sdma_fw_version.attr,
409 	&dev_attr_sdma2_fw_version.attr, &dev_attr_vcn_fw_version.attr,
410 	&dev_attr_dmcu_fw_version.attr, NULL
411 };
412 
413 static const struct attribute_group fw_attr_group = {
414 	.name = "fw_version",
415 	.attrs = fw_attrs
416 };
417 
418 int amdgpu_ucode_sysfs_init(struct amdgpu_device *adev)
419 {
420 	return sysfs_create_group(&adev->dev->kobj, &fw_attr_group);
421 }
422 
423 void amdgpu_ucode_sysfs_fini(struct amdgpu_device *adev)
424 {
425 	sysfs_remove_group(&adev->dev->kobj, &fw_attr_group);
426 }
427 
428 static int amdgpu_ucode_init_single_fw(struct amdgpu_device *adev,
429 				       struct amdgpu_firmware_info *ucode,
430 				       uint64_t mc_addr, void *kptr)
431 {
432 	const struct common_firmware_header *header = NULL;
433 	const struct gfx_firmware_header_v1_0 *cp_hdr = NULL;
434 	const struct dmcu_firmware_header_v1_0 *dmcu_hdr = NULL;
435 
436 	if (NULL == ucode->fw)
437 		return 0;
438 
439 	ucode->mc_addr = mc_addr;
440 	ucode->kaddr = kptr;
441 
442 	if (ucode->ucode_id == AMDGPU_UCODE_ID_STORAGE)
443 		return 0;
444 
445 	header = (const struct common_firmware_header *)ucode->fw->data;
446 	cp_hdr = (const struct gfx_firmware_header_v1_0 *)ucode->fw->data;
447 	dmcu_hdr = (const struct dmcu_firmware_header_v1_0 *)ucode->fw->data;
448 
449 	if (adev->firmware.load_type != AMDGPU_FW_LOAD_PSP ||
450 	    (ucode->ucode_id != AMDGPU_UCODE_ID_CP_MEC1 &&
451 	     ucode->ucode_id != AMDGPU_UCODE_ID_CP_MEC2 &&
452 	     ucode->ucode_id != AMDGPU_UCODE_ID_CP_MEC1_JT &&
453 	     ucode->ucode_id != AMDGPU_UCODE_ID_CP_MEC2_JT &&
454 	     ucode->ucode_id != AMDGPU_UCODE_ID_RLC_RESTORE_LIST_CNTL &&
455 	     ucode->ucode_id != AMDGPU_UCODE_ID_RLC_RESTORE_LIST_GPM_MEM &&
456 	     ucode->ucode_id != AMDGPU_UCODE_ID_RLC_RESTORE_LIST_SRM_MEM &&
457 		 ucode->ucode_id != AMDGPU_UCODE_ID_DMCU_ERAM &&
458 		 ucode->ucode_id != AMDGPU_UCODE_ID_DMCU_INTV)) {
459 		ucode->ucode_size = le32_to_cpu(header->ucode_size_bytes);
460 
461 		memcpy(ucode->kaddr, (void *)((uint8_t *)ucode->fw->data +
462 					      le32_to_cpu(header->ucode_array_offset_bytes)),
463 		       ucode->ucode_size);
464 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_CP_MEC1 ||
465 		   ucode->ucode_id == AMDGPU_UCODE_ID_CP_MEC2) {
466 		ucode->ucode_size = le32_to_cpu(header->ucode_size_bytes) -
467 			le32_to_cpu(cp_hdr->jt_size) * 4;
468 
469 		memcpy(ucode->kaddr, (void *)((uint8_t *)ucode->fw->data +
470 					      le32_to_cpu(header->ucode_array_offset_bytes)),
471 		       ucode->ucode_size);
472 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_CP_MEC1_JT ||
473 		   ucode->ucode_id == AMDGPU_UCODE_ID_CP_MEC2_JT) {
474 		ucode->ucode_size = le32_to_cpu(cp_hdr->jt_size) * 4;
475 
476 		memcpy(ucode->kaddr, (void *)((uint8_t *)ucode->fw->data +
477 					      le32_to_cpu(header->ucode_array_offset_bytes) +
478 					      le32_to_cpu(cp_hdr->jt_offset) * 4),
479 		       ucode->ucode_size);
480 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_DMCU_ERAM) {
481 		ucode->ucode_size = le32_to_cpu(header->ucode_size_bytes) -
482 				le32_to_cpu(dmcu_hdr->intv_size_bytes);
483 
484 		memcpy(ucode->kaddr, (void *)((uint8_t *)ucode->fw->data +
485 					      le32_to_cpu(header->ucode_array_offset_bytes)),
486 		       ucode->ucode_size);
487 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_DMCU_INTV) {
488 		ucode->ucode_size = le32_to_cpu(dmcu_hdr->intv_size_bytes);
489 
490 		memcpy(ucode->kaddr, (void *)((uint8_t *)ucode->fw->data +
491 					      le32_to_cpu(header->ucode_array_offset_bytes) +
492 					      le32_to_cpu(dmcu_hdr->intv_offset_bytes)),
493 		       ucode->ucode_size);
494 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_RLC_RESTORE_LIST_CNTL) {
495 		ucode->ucode_size = adev->gfx.rlc.save_restore_list_cntl_size_bytes;
496 		memcpy(ucode->kaddr, adev->gfx.rlc.save_restore_list_cntl,
497 		       ucode->ucode_size);
498 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_RLC_RESTORE_LIST_GPM_MEM) {
499 		ucode->ucode_size = adev->gfx.rlc.save_restore_list_gpm_size_bytes;
500 		memcpy(ucode->kaddr, adev->gfx.rlc.save_restore_list_gpm,
501 		       ucode->ucode_size);
502 	} else if (ucode->ucode_id == AMDGPU_UCODE_ID_RLC_RESTORE_LIST_SRM_MEM) {
503 		ucode->ucode_size = adev->gfx.rlc.save_restore_list_srm_size_bytes;
504 		memcpy(ucode->kaddr, adev->gfx.rlc.save_restore_list_srm,
505 		       ucode->ucode_size);
506 	}
507 
508 	return 0;
509 }
510 
511 static int amdgpu_ucode_patch_jt(struct amdgpu_firmware_info *ucode,
512 				uint64_t mc_addr, void *kptr)
513 {
514 	const struct gfx_firmware_header_v1_0 *header = NULL;
515 	const struct common_firmware_header *comm_hdr = NULL;
516 	uint8_t* src_addr = NULL;
517 	uint8_t* dst_addr = NULL;
518 
519 	if (NULL == ucode->fw)
520 		return 0;
521 
522 	comm_hdr = (const struct common_firmware_header *)ucode->fw->data;
523 	header = (const struct gfx_firmware_header_v1_0 *)ucode->fw->data;
524 	dst_addr = ucode->kaddr +
525 			   ALIGN(le32_to_cpu(comm_hdr->ucode_size_bytes),
526 			   PAGE_SIZE);
527 	src_addr = (uint8_t *)ucode->fw->data +
528 			   le32_to_cpu(comm_hdr->ucode_array_offset_bytes) +
529 			   (le32_to_cpu(header->jt_offset) * 4);
530 	memcpy(dst_addr, src_addr, le32_to_cpu(header->jt_size) * 4);
531 
532 	return 0;
533 }
534 
535 int amdgpu_ucode_create_bo(struct amdgpu_device *adev)
536 {
537 	if (adev->firmware.load_type != AMDGPU_FW_LOAD_DIRECT) {
538 		amdgpu_bo_create_kernel(adev, adev->firmware.fw_size, PAGE_SIZE,
539 			amdgpu_sriov_vf(adev) ? AMDGPU_GEM_DOMAIN_VRAM : AMDGPU_GEM_DOMAIN_GTT,
540 			&adev->firmware.fw_buf,
541 			&adev->firmware.fw_buf_mc,
542 			&adev->firmware.fw_buf_ptr);
543 		if (!adev->firmware.fw_buf) {
544 			dev_err(adev->dev, "failed to create kernel buffer for firmware.fw_buf\n");
545 			return -ENOMEM;
546 		} else if (amdgpu_sriov_vf(adev)) {
547 			memset(adev->firmware.fw_buf_ptr, 0, adev->firmware.fw_size);
548 		}
549 	}
550 	return 0;
551 }
552 
553 void amdgpu_ucode_free_bo(struct amdgpu_device *adev)
554 {
555 	if (adev->firmware.load_type != AMDGPU_FW_LOAD_DIRECT)
556 		amdgpu_bo_free_kernel(&adev->firmware.fw_buf,
557 		&adev->firmware.fw_buf_mc,
558 		&adev->firmware.fw_buf_ptr);
559 }
560 
561 int amdgpu_ucode_init_bo(struct amdgpu_device *adev)
562 {
563 	uint64_t fw_offset = 0;
564 	int i;
565 	struct amdgpu_firmware_info *ucode = NULL;
566 
567  /* for baremetal, the ucode is allocated in gtt, so don't need to fill the bo when reset/suspend */
568 	if (!amdgpu_sriov_vf(adev) && (adev->in_gpu_reset || adev->in_suspend))
569 		return 0;
570 	/*
571 	 * if SMU loaded firmware, it needn't add SMC, UVD, and VCE
572 	 * ucode info here
573 	 */
574 	if (adev->firmware.load_type != AMDGPU_FW_LOAD_PSP) {
575 		if (amdgpu_sriov_vf(adev))
576 			adev->firmware.max_ucodes = AMDGPU_UCODE_ID_MAXIMUM - 3;
577 		else
578 			adev->firmware.max_ucodes = AMDGPU_UCODE_ID_MAXIMUM - 4;
579 	} else {
580 		adev->firmware.max_ucodes = AMDGPU_UCODE_ID_MAXIMUM;
581 	}
582 
583 	for (i = 0; i < adev->firmware.max_ucodes; i++) {
584 		ucode = &adev->firmware.ucode[i];
585 		if (ucode->fw) {
586 			amdgpu_ucode_init_single_fw(adev, ucode, adev->firmware.fw_buf_mc + fw_offset,
587 						    adev->firmware.fw_buf_ptr + fw_offset);
588 			if (i == AMDGPU_UCODE_ID_CP_MEC1 &&
589 			    adev->firmware.load_type != AMDGPU_FW_LOAD_PSP) {
590 				const struct gfx_firmware_header_v1_0 *cp_hdr;
591 				cp_hdr = (const struct gfx_firmware_header_v1_0 *)ucode->fw->data;
592 				amdgpu_ucode_patch_jt(ucode,  adev->firmware.fw_buf_mc + fw_offset,
593 						    adev->firmware.fw_buf_ptr + fw_offset);
594 				fw_offset += ALIGN(le32_to_cpu(cp_hdr->jt_size) << 2, PAGE_SIZE);
595 			}
596 			fw_offset += ALIGN(ucode->ucode_size, PAGE_SIZE);
597 		}
598 	}
599 	return 0;
600 }
601