* __cpu_soft_restart(el2_switch, entry, arg0, arg1, arg2) - Helper for
  * cpu_soft_restart.
  *
- * @el2_switch: Flag to indicate a swich to EL2 is needed.
+ * @el2_switch: Flag to indicate a switch to EL2 is needed.
  * @entry: Location to jump to for soft reset.
- * arg0: First argument passed to @entry.
- * arg1: Second argument passed to @entry.
- * arg2: Third argument passed to @entry.
+ * arg0: First argument passed to @entry. (relocation list)
+ * arg1: Second argument passed to @entry.(physical kernel entry)
+ * arg2: Third argument passed to @entry. (physical dtb address)
  *
  * Put the CPU into the same state as it would be if it had been reset, and
  * branch to what would be the reset vector. It must be executed with the
 
         * uses physical addressing to relocate the new image to its final
         * position and transfers control to the image entry point when the
         * relocation is complete.
+        * In kexec case, kimage->start points to purgatory assuming that
+        * kernel entry and dtb address are embedded in purgatory by
+        * userspace (kexec-tools).
+        * In kexec_file case, the kernel starts directly without purgatory.
         */
-
-       cpu_soft_restart(reboot_code_buffer_phys, kimage->head, kimage->start, 0);
+       cpu_soft_restart(reboot_code_buffer_phys, kimage->head, kimage->start,
+#ifdef CONFIG_KEXEC_FILE
+                                               kimage->arch.dtb_mem);
+#else
+                                               0);
+#endif
 
        BUG(); /* Should never get here. */
 }
 
 ENTRY(arm64_relocate_new_kernel)
 
        /* Setup the list loop variables. */
+       mov     x18, x2                         /* x18 = dtb address */
        mov     x17, x1                         /* x17 = kimage_start */
        mov     x16, x0                         /* x16 = kimage_head */
        raw_dcache_line_size x15, x0            /* x15 = dcache line size */
        isb
 
        /* Start new image. */
-       mov     x0, xzr
+       mov     x0, x18
        mov     x1, xzr
        mov     x2, xzr
        mov     x3, xzr