Merge branch 'stable/for-linus-5.15' of git://git.kernel.org/pub/scm/linux/kernel...
[linux-2.6-microblaze.git] / drivers / gpu / drm / amd / amdgpu / psp_v11_0.c
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 #include <linux/firmware.h>
24 #include <linux/module.h>
25 #include <linux/vmalloc.h>
26 #include <drm/drm_drv.h>
27
28 #include "amdgpu.h"
29 #include "amdgpu_psp.h"
30 #include "amdgpu_ras.h"
31 #include "amdgpu_ucode.h"
32 #include "soc15_common.h"
33 #include "psp_v11_0.h"
34
35 #include "mp/mp_11_0_offset.h"
36 #include "mp/mp_11_0_sh_mask.h"
37 #include "gc/gc_9_0_offset.h"
38 #include "sdma0/sdma0_4_0_offset.h"
39 #include "nbio/nbio_7_4_offset.h"
40
41 #include "oss/osssys_4_0_offset.h"
42 #include "oss/osssys_4_0_sh_mask.h"
43
44 MODULE_FIRMWARE("amdgpu/vega20_sos.bin");
45 MODULE_FIRMWARE("amdgpu/vega20_asd.bin");
46 MODULE_FIRMWARE("amdgpu/vega20_ta.bin");
47 MODULE_FIRMWARE("amdgpu/navi10_sos.bin");
48 MODULE_FIRMWARE("amdgpu/navi10_asd.bin");
49 MODULE_FIRMWARE("amdgpu/navi10_ta.bin");
50 MODULE_FIRMWARE("amdgpu/navi14_sos.bin");
51 MODULE_FIRMWARE("amdgpu/navi14_asd.bin");
52 MODULE_FIRMWARE("amdgpu/navi14_ta.bin");
53 MODULE_FIRMWARE("amdgpu/navi12_sos.bin");
54 MODULE_FIRMWARE("amdgpu/navi12_asd.bin");
55 MODULE_FIRMWARE("amdgpu/navi12_ta.bin");
56 MODULE_FIRMWARE("amdgpu/arcturus_sos.bin");
57 MODULE_FIRMWARE("amdgpu/arcturus_asd.bin");
58 MODULE_FIRMWARE("amdgpu/arcturus_ta.bin");
59 MODULE_FIRMWARE("amdgpu/sienna_cichlid_sos.bin");
60 MODULE_FIRMWARE("amdgpu/sienna_cichlid_ta.bin");
61 MODULE_FIRMWARE("amdgpu/navy_flounder_sos.bin");
62 MODULE_FIRMWARE("amdgpu/navy_flounder_ta.bin");
63 MODULE_FIRMWARE("amdgpu/vangogh_asd.bin");
64 MODULE_FIRMWARE("amdgpu/vangogh_toc.bin");
65 MODULE_FIRMWARE("amdgpu/dimgrey_cavefish_sos.bin");
66 MODULE_FIRMWARE("amdgpu/dimgrey_cavefish_ta.bin");
67 MODULE_FIRMWARE("amdgpu/beige_goby_sos.bin");
68 MODULE_FIRMWARE("amdgpu/beige_goby_ta.bin");
69
70 /* address block */
71 #define smnMP1_FIRMWARE_FLAGS           0x3010024
72 /* navi10 reg offset define */
73 #define mmRLC_GPM_UCODE_ADDR_NV10       0x5b61
74 #define mmRLC_GPM_UCODE_DATA_NV10       0x5b62
75 #define mmSDMA0_UCODE_ADDR_NV10         0x5880
76 #define mmSDMA0_UCODE_DATA_NV10         0x5881
77 /* memory training timeout define */
78 #define MEM_TRAIN_SEND_MSG_TIMEOUT_US   3000000
79
80 /* For large FW files the time to complete can be very long */
81 #define USBC_PD_POLLING_LIMIT_S 240
82
83 static int psp_v11_0_init_microcode(struct psp_context *psp)
84 {
85         struct amdgpu_device *adev = psp->adev;
86         const char *chip_name;
87         char fw_name[PSP_FW_NAME_LEN];
88         int err = 0;
89         const struct ta_firmware_header_v1_0 *ta_hdr;
90
91         DRM_DEBUG("\n");
92
93         switch (adev->asic_type) {
94         case CHIP_VEGA20:
95                 chip_name = "vega20";
96                 break;
97         case CHIP_NAVI10:
98                 chip_name = "navi10";
99                 break;
100         case CHIP_NAVI14:
101                 chip_name = "navi14";
102                 break;
103         case CHIP_NAVI12:
104                 chip_name = "navi12";
105                 break;
106         case CHIP_ARCTURUS:
107                 chip_name = "arcturus";
108                 break;
109         case CHIP_SIENNA_CICHLID:
110                 chip_name = "sienna_cichlid";
111                 break;
112         case CHIP_NAVY_FLOUNDER:
113                 chip_name = "navy_flounder";
114                 break;
115         case CHIP_VANGOGH:
116                 chip_name = "vangogh";
117                 break;
118         case CHIP_DIMGREY_CAVEFISH:
119                 chip_name = "dimgrey_cavefish";
120                 break;
121         case CHIP_BEIGE_GOBY:
122                 chip_name = "beige_goby";
123                 break;
124         default:
125                 BUG();
126         }
127
128
129         switch (adev->asic_type) {
130         case CHIP_VEGA20:
131         case CHIP_ARCTURUS:
132                 err = psp_init_sos_microcode(psp, chip_name);
133                 if (err)
134                         return err;
135                 err = psp_init_asd_microcode(psp, chip_name);
136                 if (err)
137                         return err;
138                 snprintf(fw_name, sizeof(fw_name), "amdgpu/%s_ta.bin", chip_name);
139                 err = request_firmware(&adev->psp.ta_fw, fw_name, adev->dev);
140                 if (err) {
141                         release_firmware(adev->psp.ta_fw);
142                         adev->psp.ta_fw = NULL;
143                         dev_info(adev->dev,
144                                  "psp v11.0: Failed to load firmware \"%s\"\n", fw_name);
145                 } else {
146                         err = amdgpu_ucode_validate(adev->psp.ta_fw);
147                         if (err)
148                                 goto out2;
149
150                         ta_hdr = (const struct ta_firmware_header_v1_0 *)adev->psp.ta_fw->data;
151                         adev->psp.ta_xgmi_ucode_version = le32_to_cpu(ta_hdr->ta_xgmi_ucode_version);
152                         adev->psp.ta_xgmi_ucode_size = le32_to_cpu(ta_hdr->ta_xgmi_size_bytes);
153                         adev->psp.ta_xgmi_start_addr = (uint8_t *)ta_hdr +
154                                 le32_to_cpu(ta_hdr->header.ucode_array_offset_bytes);
155                         adev->psp.ta_fw_version = le32_to_cpu(ta_hdr->header.ucode_version);
156                         adev->psp.ta_ras_ucode_version = le32_to_cpu(ta_hdr->ta_ras_ucode_version);
157                         adev->psp.ta_ras_ucode_size = le32_to_cpu(ta_hdr->ta_ras_size_bytes);
158                         adev->psp.ta_ras_start_addr = (uint8_t *)adev->psp.ta_xgmi_start_addr +
159                                 le32_to_cpu(ta_hdr->ta_ras_offset_bytes);
160                 }
161                 break;
162         case CHIP_NAVI10:
163         case CHIP_NAVI14:
164         case CHIP_NAVI12:
165                 err = psp_init_sos_microcode(psp, chip_name);
166                 if (err)
167                         return err;
168                 err = psp_init_asd_microcode(psp, chip_name);
169                 if (err)
170                         return err;
171                 if (amdgpu_sriov_vf(adev))
172                         break;
173                 snprintf(fw_name, sizeof(fw_name), "amdgpu/%s_ta.bin", chip_name);
174                 err = request_firmware(&adev->psp.ta_fw, fw_name, adev->dev);
175                 if (err) {
176                         release_firmware(adev->psp.ta_fw);
177                         adev->psp.ta_fw = NULL;
178                         dev_info(adev->dev,
179                                  "psp v11.0: Failed to load firmware \"%s\"\n", fw_name);
180                 } else {
181                         err = amdgpu_ucode_validate(adev->psp.ta_fw);
182                         if (err)
183                                 goto out2;
184
185                         ta_hdr = (const struct ta_firmware_header_v1_0 *)adev->psp.ta_fw->data;
186                         adev->psp.ta_hdcp_ucode_version = le32_to_cpu(ta_hdr->ta_hdcp_ucode_version);
187                         adev->psp.ta_hdcp_ucode_size = le32_to_cpu(ta_hdr->ta_hdcp_size_bytes);
188                         adev->psp.ta_hdcp_start_addr = (uint8_t *)ta_hdr +
189                                 le32_to_cpu(ta_hdr->header.ucode_array_offset_bytes);
190
191                         adev->psp.ta_fw_version = le32_to_cpu(ta_hdr->header.ucode_version);
192
193                         adev->psp.ta_dtm_ucode_version = le32_to_cpu(ta_hdr->ta_dtm_ucode_version);
194                         adev->psp.ta_dtm_ucode_size = le32_to_cpu(ta_hdr->ta_dtm_size_bytes);
195                         adev->psp.ta_dtm_start_addr = (uint8_t *)adev->psp.ta_hdcp_start_addr +
196                                 le32_to_cpu(ta_hdr->ta_dtm_offset_bytes);
197                 }
198                 break;
199         case CHIP_SIENNA_CICHLID:
200         case CHIP_NAVY_FLOUNDER:
201         case CHIP_DIMGREY_CAVEFISH:
202                 err = psp_init_sos_microcode(psp, chip_name);
203                 if (err)
204                         return err;
205                 err = psp_init_ta_microcode(psp, chip_name);
206                 if (err)
207                         return err;
208                 break;
209         case CHIP_BEIGE_GOBY:
210                 err = psp_init_sos_microcode(psp, chip_name);
211                 if (err)
212                         return err;
213                 err = psp_init_ta_microcode(psp, chip_name);
214                 if (err)
215                         return err;
216                 break;
217         case CHIP_VANGOGH:
218                 err = psp_init_asd_microcode(psp, chip_name);
219                 if (err)
220                         return err;
221                 err = psp_init_toc_microcode(psp, chip_name);
222                 if (err)
223                         return err;
224                 break;
225         default:
226                 BUG();
227         }
228
229         return 0;
230
231 out2:
232         release_firmware(adev->psp.ta_fw);
233         adev->psp.ta_fw = NULL;
234         return err;
235 }
236
237 static int psp_v11_0_wait_for_bootloader(struct psp_context *psp)
238 {
239         struct amdgpu_device *adev = psp->adev;
240
241         int ret;
242         int retry_loop;
243
244         for (retry_loop = 0; retry_loop < 10; retry_loop++) {
245                 /* Wait for bootloader to signify that is
246                     ready having bit 31 of C2PMSG_35 set to 1 */
247                 ret = psp_wait_for(psp,
248                                    SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
249                                    0x80000000,
250                                    0x80000000,
251                                    false);
252
253                 if (ret == 0)
254                         return 0;
255         }
256
257         return ret;
258 }
259
260 static bool psp_v11_0_is_sos_alive(struct psp_context *psp)
261 {
262         struct amdgpu_device *adev = psp->adev;
263         uint32_t sol_reg;
264
265         sol_reg = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_81);
266
267         return sol_reg != 0x0;
268 }
269
270 static int psp_v11_0_bootloader_load_kdb(struct psp_context *psp)
271 {
272         int ret;
273         uint32_t psp_gfxdrv_command_reg = 0;
274         struct amdgpu_device *adev = psp->adev;
275
276         /* Check tOS sign of life register to confirm sys driver and sOS
277          * are already been loaded.
278          */
279         if (psp_v11_0_is_sos_alive(psp))
280                 return 0;
281
282         ret = psp_v11_0_wait_for_bootloader(psp);
283         if (ret)
284                 return ret;
285
286         /* Copy PSP KDB binary to memory */
287         psp_copy_fw(psp, psp->kdb_start_addr, psp->kdb_bin_size);
288
289         /* Provide the PSP KDB to bootloader */
290         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
291                (uint32_t)(psp->fw_pri_mc_addr >> 20));
292         psp_gfxdrv_command_reg = PSP_BL__LOAD_KEY_DATABASE;
293         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
294                psp_gfxdrv_command_reg);
295
296         ret = psp_v11_0_wait_for_bootloader(psp);
297
298         return ret;
299 }
300
301 static int psp_v11_0_bootloader_load_spl(struct psp_context *psp)
302 {
303         int ret;
304         uint32_t psp_gfxdrv_command_reg = 0;
305         struct amdgpu_device *adev = psp->adev;
306
307         /* Check tOS sign of life register to confirm sys driver and sOS
308          * are already been loaded.
309          */
310         if (psp_v11_0_is_sos_alive(psp))
311                 return 0;
312
313         ret = psp_v11_0_wait_for_bootloader(psp);
314         if (ret)
315                 return ret;
316
317         /* Copy PSP SPL binary to memory */
318         psp_copy_fw(psp, psp->spl_start_addr, psp->spl_bin_size);
319
320         /* Provide the PSP SPL to bootloader */
321         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
322                (uint32_t)(psp->fw_pri_mc_addr >> 20));
323         psp_gfxdrv_command_reg = PSP_BL__LOAD_TOS_SPL_TABLE;
324         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
325                psp_gfxdrv_command_reg);
326
327         ret = psp_v11_0_wait_for_bootloader(psp);
328
329         return ret;
330 }
331
332 static int psp_v11_0_bootloader_load_sysdrv(struct psp_context *psp)
333 {
334         int ret;
335         uint32_t psp_gfxdrv_command_reg = 0;
336         struct amdgpu_device *adev = psp->adev;
337
338         /* Check sOS sign of life register to confirm sys driver and sOS
339          * are already been loaded.
340          */
341         if (psp_v11_0_is_sos_alive(psp))
342                 return 0;
343
344         ret = psp_v11_0_wait_for_bootloader(psp);
345         if (ret)
346                 return ret;
347
348         /* Copy PSP System Driver binary to memory */
349         psp_copy_fw(psp, psp->sys_start_addr, psp->sys_bin_size);
350
351         /* Provide the sys driver to bootloader */
352         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
353                (uint32_t)(psp->fw_pri_mc_addr >> 20));
354         psp_gfxdrv_command_reg = PSP_BL__LOAD_SYSDRV;
355         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
356                psp_gfxdrv_command_reg);
357
358         /* there might be handshake issue with hardware which needs delay */
359         mdelay(20);
360
361         ret = psp_v11_0_wait_for_bootloader(psp);
362
363         return ret;
364 }
365
366 static int psp_v11_0_bootloader_load_sos(struct psp_context *psp)
367 {
368         int ret;
369         unsigned int psp_gfxdrv_command_reg = 0;
370         struct amdgpu_device *adev = psp->adev;
371
372         /* Check sOS sign of life register to confirm sys driver and sOS
373          * are already been loaded.
374          */
375         if (psp_v11_0_is_sos_alive(psp))
376                 return 0;
377
378         ret = psp_v11_0_wait_for_bootloader(psp);
379         if (ret)
380                 return ret;
381
382         /* Copy Secure OS binary to PSP memory */
383         psp_copy_fw(psp, psp->sos_start_addr, psp->sos_bin_size);
384
385         /* Provide the PSP secure OS to bootloader */
386         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36,
387                (uint32_t)(psp->fw_pri_mc_addr >> 20));
388         psp_gfxdrv_command_reg = PSP_BL__LOAD_SOSDRV;
389         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35,
390                psp_gfxdrv_command_reg);
391
392         /* there might be handshake issue with hardware which needs delay */
393         mdelay(20);
394         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_81),
395                            RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_81),
396                            0, true);
397
398         return ret;
399 }
400
401 static int psp_v11_0_ring_init(struct psp_context *psp,
402                               enum psp_ring_type ring_type)
403 {
404         int ret = 0;
405         struct psp_ring *ring;
406         struct amdgpu_device *adev = psp->adev;
407
408         ring = &psp->km_ring;
409
410         ring->ring_type = ring_type;
411
412         /* allocate 4k Page of Local Frame Buffer memory for ring */
413         ring->ring_size = 0x1000;
414         ret = amdgpu_bo_create_kernel(adev, ring->ring_size, PAGE_SIZE,
415                                       AMDGPU_GEM_DOMAIN_VRAM,
416                                       &adev->firmware.rbuf,
417                                       &ring->ring_mem_mc_addr,
418                                       (void **)&ring->ring_mem);
419         if (ret) {
420                 ring->ring_size = 0;
421                 return ret;
422         }
423
424         return 0;
425 }
426
427 static int psp_v11_0_ring_stop(struct psp_context *psp,
428                               enum psp_ring_type ring_type)
429 {
430         int ret = 0;
431         struct amdgpu_device *adev = psp->adev;
432
433         /* Write the ring destroy command*/
434         if (amdgpu_sriov_vf(adev))
435                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101,
436                                      GFX_CTRL_CMD_ID_DESTROY_GPCOM_RING);
437         else
438                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64,
439                                      GFX_CTRL_CMD_ID_DESTROY_RINGS);
440
441         /* there might be handshake issue with hardware which needs delay */
442         mdelay(20);
443
444         /* Wait for response flag (bit 31) */
445         if (amdgpu_sriov_vf(adev))
446                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_101),
447                                    0x80000000, 0x80000000, false);
448         else
449                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
450                                    0x80000000, 0x80000000, false);
451
452         return ret;
453 }
454
455 static int psp_v11_0_ring_create(struct psp_context *psp,
456                                 enum psp_ring_type ring_type)
457 {
458         int ret = 0;
459         unsigned int psp_ring_reg = 0;
460         struct psp_ring *ring = &psp->km_ring;
461         struct amdgpu_device *adev = psp->adev;
462
463         if (amdgpu_sriov_vf(adev)) {
464                 ring->ring_wptr = 0;
465                 ret = psp_v11_0_ring_stop(psp, ring_type);
466                 if (ret) {
467                         DRM_ERROR("psp_v11_0_ring_stop_sriov failed!\n");
468                         return ret;
469                 }
470
471                 /* Write low address of the ring to C2PMSG_102 */
472                 psp_ring_reg = lower_32_bits(ring->ring_mem_mc_addr);
473                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_102, psp_ring_reg);
474                 /* Write high address of the ring to C2PMSG_103 */
475                 psp_ring_reg = upper_32_bits(ring->ring_mem_mc_addr);
476                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_103, psp_ring_reg);
477
478                 /* Write the ring initialization command to C2PMSG_101 */
479                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101,
480                                              GFX_CTRL_CMD_ID_INIT_GPCOM_RING);
481
482                 /* there might be handshake issue with hardware which needs delay */
483                 mdelay(20);
484
485                 /* Wait for response flag (bit 31) in C2PMSG_101 */
486                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_101),
487                                    0x80000000, 0x8000FFFF, false);
488
489         } else {
490                 /* Wait for sOS ready for ring creation */
491                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
492                                    0x80000000, 0x80000000, false);
493                 if (ret) {
494                         DRM_ERROR("Failed to wait for sOS ready for ring creation\n");
495                         return ret;
496                 }
497
498                 /* Write low address of the ring to C2PMSG_69 */
499                 psp_ring_reg = lower_32_bits(ring->ring_mem_mc_addr);
500                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_69, psp_ring_reg);
501                 /* Write high address of the ring to C2PMSG_70 */
502                 psp_ring_reg = upper_32_bits(ring->ring_mem_mc_addr);
503                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_70, psp_ring_reg);
504                 /* Write size of ring to C2PMSG_71 */
505                 psp_ring_reg = ring->ring_size;
506                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_71, psp_ring_reg);
507                 /* Write the ring initialization command to C2PMSG_64 */
508                 psp_ring_reg = ring_type;
509                 psp_ring_reg = psp_ring_reg << 16;
510                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_64, psp_ring_reg);
511
512                 /* there might be handshake issue with hardware which needs delay */
513                 mdelay(20);
514
515                 /* Wait for response flag (bit 31) in C2PMSG_64 */
516                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64),
517                                    0x80000000, 0x8000FFFF, false);
518         }
519
520         return ret;
521 }
522
523
524 static int psp_v11_0_ring_destroy(struct psp_context *psp,
525                                  enum psp_ring_type ring_type)
526 {
527         int ret = 0;
528         struct psp_ring *ring = &psp->km_ring;
529         struct amdgpu_device *adev = psp->adev;
530
531         ret = psp_v11_0_ring_stop(psp, ring_type);
532         if (ret)
533                 DRM_ERROR("Fail to stop psp ring\n");
534
535         amdgpu_bo_free_kernel(&adev->firmware.rbuf,
536                               &ring->ring_mem_mc_addr,
537                               (void **)&ring->ring_mem);
538
539         return ret;
540 }
541
542 static int psp_v11_0_mode1_reset(struct psp_context *psp)
543 {
544         int ret;
545         uint32_t offset;
546         struct amdgpu_device *adev = psp->adev;
547
548         offset = SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_64);
549
550         ret = psp_wait_for(psp, offset, 0x80000000, 0x8000FFFF, false);
551
552         if (ret) {
553                 DRM_INFO("psp is not working correctly before mode1 reset!\n");
554                 return -EINVAL;
555         }
556
557         /*send the mode 1 reset command*/
558         WREG32(offset, GFX_CTRL_CMD_ID_MODE1_RST);
559
560         msleep(500);
561
562         offset = SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_33);
563
564         ret = psp_wait_for(psp, offset, 0x80000000, 0x80000000, false);
565
566         if (ret) {
567                 DRM_INFO("psp mode 1 reset failed!\n");
568                 return -EINVAL;
569         }
570
571         DRM_INFO("psp mode1 reset succeed \n");
572
573         return 0;
574 }
575
576 static int psp_v11_0_memory_training_send_msg(struct psp_context *psp, int msg)
577 {
578         int ret;
579         int i;
580         uint32_t data_32;
581         int max_wait;
582         struct amdgpu_device *adev = psp->adev;
583
584         data_32 = (psp->mem_train_ctx.c2p_train_data_offset >> 20);
585         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36, data_32);
586         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35, msg);
587
588         max_wait = MEM_TRAIN_SEND_MSG_TIMEOUT_US / adev->usec_timeout;
589         for (i = 0; i < max_wait; i++) {
590                 ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
591                                    0x80000000, 0x80000000, false);
592                 if (ret == 0)
593                         break;
594         }
595         if (i < max_wait)
596                 ret = 0;
597         else
598                 ret = -ETIME;
599
600         DRM_DEBUG("training %s %s, cost %d @ %d ms\n",
601                   (msg == PSP_BL__DRAM_SHORT_TRAIN) ? "short" : "long",
602                   (ret == 0) ? "succeed" : "failed",
603                   i, adev->usec_timeout/1000);
604         return ret;
605 }
606
607 /*
608  * save and restore process
609  */
610 static int psp_v11_0_memory_training(struct psp_context *psp, uint32_t ops)
611 {
612         struct psp_memory_training_context *ctx = &psp->mem_train_ctx;
613         uint32_t *pcache = (uint32_t *)ctx->sys_cache;
614         struct amdgpu_device *adev = psp->adev;
615         uint32_t p2c_header[4];
616         uint32_t sz;
617         void *buf;
618         int ret, idx;
619
620         if (ctx->init == PSP_MEM_TRAIN_NOT_SUPPORT) {
621                 DRM_DEBUG("Memory training is not supported.\n");
622                 return 0;
623         } else if (ctx->init != PSP_MEM_TRAIN_INIT_SUCCESS) {
624                 DRM_ERROR("Memory training initialization failure.\n");
625                 return -EINVAL;
626         }
627
628         if (psp_v11_0_is_sos_alive(psp)) {
629                 DRM_DEBUG("SOS is alive, skip memory training.\n");
630                 return 0;
631         }
632
633         amdgpu_device_vram_access(adev, ctx->p2c_train_data_offset, p2c_header, sizeof(p2c_header), false);
634         DRM_DEBUG("sys_cache[%08x,%08x,%08x,%08x] p2c_header[%08x,%08x,%08x,%08x]\n",
635                   pcache[0], pcache[1], pcache[2], pcache[3],
636                   p2c_header[0], p2c_header[1], p2c_header[2], p2c_header[3]);
637
638         if (ops & PSP_MEM_TRAIN_SEND_SHORT_MSG) {
639                 DRM_DEBUG("Short training depends on restore.\n");
640                 ops |= PSP_MEM_TRAIN_RESTORE;
641         }
642
643         if ((ops & PSP_MEM_TRAIN_RESTORE) &&
644             pcache[0] != MEM_TRAIN_SYSTEM_SIGNATURE) {
645                 DRM_DEBUG("sys_cache[0] is invalid, restore depends on save.\n");
646                 ops |= PSP_MEM_TRAIN_SAVE;
647         }
648
649         if (p2c_header[0] == MEM_TRAIN_SYSTEM_SIGNATURE &&
650             !(pcache[0] == MEM_TRAIN_SYSTEM_SIGNATURE &&
651               pcache[3] == p2c_header[3])) {
652                 DRM_DEBUG("sys_cache is invalid or out-of-date, need save training data to sys_cache.\n");
653                 ops |= PSP_MEM_TRAIN_SAVE;
654         }
655
656         if ((ops & PSP_MEM_TRAIN_SAVE) &&
657             p2c_header[0] != MEM_TRAIN_SYSTEM_SIGNATURE) {
658                 DRM_DEBUG("p2c_header[0] is invalid, save depends on long training.\n");
659                 ops |= PSP_MEM_TRAIN_SEND_LONG_MSG;
660         }
661
662         if (ops & PSP_MEM_TRAIN_SEND_LONG_MSG) {
663                 ops &= ~PSP_MEM_TRAIN_SEND_SHORT_MSG;
664                 ops |= PSP_MEM_TRAIN_SAVE;
665         }
666
667         DRM_DEBUG("Memory training ops:%x.\n", ops);
668
669         if (ops & PSP_MEM_TRAIN_SEND_LONG_MSG) {
670                 /*
671                  * Long training will encroach a certain amount on the bottom of VRAM;
672                  * save the content from the bottom of VRAM to system memory
673                  * before training, and restore it after training to avoid
674                  * VRAM corruption.
675                  */
676                 sz = GDDR6_MEM_TRAINING_ENCROACHED_SIZE;
677
678                 if (adev->gmc.visible_vram_size < sz || !adev->mman.aper_base_kaddr) {
679                         DRM_ERROR("visible_vram_size %llx or aper_base_kaddr %p is not initialized.\n",
680                                   adev->gmc.visible_vram_size,
681                                   adev->mman.aper_base_kaddr);
682                         return -EINVAL;
683                 }
684
685                 buf = vmalloc(sz);
686                 if (!buf) {
687                         DRM_ERROR("failed to allocate system memory.\n");
688                         return -ENOMEM;
689                 }
690
691                 if (drm_dev_enter(&adev->ddev, &idx)) {
692                         memcpy_fromio(buf, adev->mman.aper_base_kaddr, sz);
693                         ret = psp_v11_0_memory_training_send_msg(psp, PSP_BL__DRAM_LONG_TRAIN);
694                         if (ret) {
695                                 DRM_ERROR("Send long training msg failed.\n");
696                                 vfree(buf);
697                                 drm_dev_exit(idx);
698                                 return ret;
699                         }
700
701                         memcpy_toio(adev->mman.aper_base_kaddr, buf, sz);
702                         adev->hdp.funcs->flush_hdp(adev, NULL);
703                         vfree(buf);
704                         drm_dev_exit(idx);
705                 } else {
706                         vfree(buf);
707                         return -ENODEV;
708                 }
709         }
710
711         if (ops & PSP_MEM_TRAIN_SAVE) {
712                 amdgpu_device_vram_access(psp->adev, ctx->p2c_train_data_offset, ctx->sys_cache, ctx->train_data_size, false);
713         }
714
715         if (ops & PSP_MEM_TRAIN_RESTORE) {
716                 amdgpu_device_vram_access(psp->adev, ctx->c2p_train_data_offset, ctx->sys_cache, ctx->train_data_size, true);
717         }
718
719         if (ops & PSP_MEM_TRAIN_SEND_SHORT_MSG) {
720                 ret = psp_v11_0_memory_training_send_msg(psp, (amdgpu_force_long_training > 0) ?
721                                                          PSP_BL__DRAM_LONG_TRAIN : PSP_BL__DRAM_SHORT_TRAIN);
722                 if (ret) {
723                         DRM_ERROR("send training msg failed.\n");
724                         return ret;
725                 }
726         }
727         ctx->training_cnt++;
728         return 0;
729 }
730
731 static uint32_t psp_v11_0_ring_get_wptr(struct psp_context *psp)
732 {
733         uint32_t data;
734         struct amdgpu_device *adev = psp->adev;
735
736         if (amdgpu_sriov_vf(adev))
737                 data = psp->km_ring.ring_wptr;
738         else
739                 data = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_67);
740
741         return data;
742 }
743
744 static void psp_v11_0_ring_set_wptr(struct psp_context *psp, uint32_t value)
745 {
746         struct amdgpu_device *adev = psp->adev;
747
748         if (amdgpu_sriov_vf(adev)) {
749                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_102, value);
750                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_101, GFX_CTRL_CMD_ID_CONSUME_CMD);
751                 psp->km_ring.ring_wptr = value;
752         } else
753                 WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_67, value);
754 }
755
756 static int psp_v11_0_load_usbc_pd_fw(struct psp_context *psp, dma_addr_t dma_addr)
757 {
758         struct amdgpu_device *adev = psp->adev;
759         uint32_t reg_status;
760         int ret, i = 0;
761
762         /* Write lower 32-bit address of the PD Controller FW */
763         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36, lower_32_bits(dma_addr));
764         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
765                              0x80000000, 0x80000000, false);
766         if (ret)
767                 return ret;
768
769         /* Fireup interrupt so PSP can pick up the lower address */
770         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35, 0x800000);
771         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
772                              0x80000000, 0x80000000, false);
773         if (ret)
774                 return ret;
775
776         reg_status = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35);
777
778         if ((reg_status & 0xFFFF) != 0) {
779                 DRM_ERROR("Lower address load failed - MP0_SMN_C2PMSG_35.Bits [15:0] = %02x...\n",
780                                 reg_status & 0xFFFF);
781                 return -EIO;
782         }
783
784         /* Write upper 32-bit address of the PD Controller FW */
785         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36, upper_32_bits(dma_addr));
786
787         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
788                              0x80000000, 0x80000000, false);
789         if (ret)
790                 return ret;
791
792         /* Fireup interrupt so PSP can pick up the upper address */
793         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35, 0x4000000);
794
795         /* FW load takes very long time */
796         do {
797                 msleep(1000);
798                 reg_status = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35);
799
800                 if (reg_status & 0x80000000)
801                         goto done;
802
803         } while (++i < USBC_PD_POLLING_LIMIT_S);
804
805         return -ETIME;
806 done:
807
808         if ((reg_status & 0xFFFF) != 0) {
809                 DRM_ERROR("Upper address load failed - MP0_SMN_C2PMSG_35.Bits [15:0] = x%04x\n",
810                                 reg_status & 0xFFFF);
811                 return -EIO;
812         }
813
814         return 0;
815 }
816
817 static int psp_v11_0_read_usbc_pd_fw(struct psp_context *psp, uint32_t *fw_ver)
818 {
819         struct amdgpu_device *adev = psp->adev;
820         int ret;
821
822         WREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_35, C2PMSG_CMD_GFX_USB_PD_FW_VER);
823
824         ret = psp_wait_for(psp, SOC15_REG_OFFSET(MP0, 0, mmMP0_SMN_C2PMSG_35),
825                                      0x80000000, 0x80000000, false);
826         if (!ret)
827                 *fw_ver = RREG32_SOC15(MP0, 0, mmMP0_SMN_C2PMSG_36);
828
829         return ret;
830 }
831
832 static const struct psp_funcs psp_v11_0_funcs = {
833         .init_microcode = psp_v11_0_init_microcode,
834         .bootloader_load_kdb = psp_v11_0_bootloader_load_kdb,
835         .bootloader_load_spl = psp_v11_0_bootloader_load_spl,
836         .bootloader_load_sysdrv = psp_v11_0_bootloader_load_sysdrv,
837         .bootloader_load_sos = psp_v11_0_bootloader_load_sos,
838         .ring_init = psp_v11_0_ring_init,
839         .ring_create = psp_v11_0_ring_create,
840         .ring_stop = psp_v11_0_ring_stop,
841         .ring_destroy = psp_v11_0_ring_destroy,
842         .mode1_reset = psp_v11_0_mode1_reset,
843         .mem_training = psp_v11_0_memory_training,
844         .ring_get_wptr = psp_v11_0_ring_get_wptr,
845         .ring_set_wptr = psp_v11_0_ring_set_wptr,
846         .load_usbc_pd_fw = psp_v11_0_load_usbc_pd_fw,
847         .read_usbc_pd_fw = psp_v11_0_read_usbc_pd_fw
848 };
849
850 void psp_v11_0_set_psp_funcs(struct psp_context *psp)
851 {
852         psp->funcs = &psp_v11_0_funcs;
853 }