arm64: zynqmp: Fix m-a2197-01 DT based on latest schematics

[oweals/u-boot.git] / arch / arm / mach-k3 / security.c

diff --git a/arch/arm/mach-k3/security.c b/arch/arm/mach-k3/security.c
index 52f49bf01fe2a1c360f920a885ec933e64c54b62..4e011ee10ef7038839c006eba92eaa915ac9d5fa 100644 (file)
--- a/arch/arm/mach-k3/security.c
+++ b/arch/arm/mach-k3/security.c
@@ -11,25 +11,16 @@
 #include <linux/soc/ti/ti_sci_protocol.h>
 #include <mach/spl.h>
 #include <spl.h>
+#include <asm/arch/sys_proto.h>
 
 void board_fit_image_post_process(void **p_image, size_t *p_size)
 {
-       struct udevice *dev;
-       struct ti_sci_handle *ti_sci;
-       struct ti_sci_proc_ops *proc_ops;
+       struct ti_sci_handle *ti_sci = get_ti_sci_handle();
+       struct ti_sci_proc_ops *proc_ops = &ti_sci->ops.proc_ops;
        u64 image_addr;
        u32 image_size;
        int ret;
 
-       /* Get handle to Device Management and Security Controller (SYSFW) */
-       ret = uclass_get_device_by_name(UCLASS_FIRMWARE, "dmsc", &dev);
-       if (ret) {
-               printf("Failed to get handle to SYSFW (%d)\n", ret);
-               hang();
-       }
-       ti_sci = (struct ti_sci_handle *)(ti_sci_get_handle_from_sysfw(dev));
-       proc_ops = &ti_sci->ops.proc_ops;
-
        image_addr = (uintptr_t)*p_image;
 
        debug("Authenticating image at address 0x%016llx\n", image_addr);