When returning from idle, we rely on the fact that thread_info lives at
the end of the kernel stack, and restore this by masking the saved stack
pointer. Subsequent patches will sever the relationship between the
stack and thread_info, and to cater for this we must save/restore sp_el0
explicitly, storing it in cpu_suspend_ctx.

As cpu_suspend_ctx must be doubleword aligned, this leaves us with an
extra slot in cpu_suspend_ctx. We can use this to save/restore tpidr_el1
in the same way, which simplifies the code, avoiding pointer chasing on
the restore path (as we no longer need to load thread_info::cpu followed
by the relevant slot in __per_cpu_offset based on this).

This patch stashes both registers in cpu_suspend_ctx.

Signed-off-by: Mark Rutland <mark.rutl...@arm.com>
Cc: Catalin Marinas <catalin.mari...@arm.com>
Cc: James Morse <james.mo...@arm.com>
Cc: Lorenzo Pieralisi <lorenzo.pieral...@arm.com>
Cc: Will Deacon <will.dea...@arm.com>
---
 arch/arm64/include/asm/suspend.h | 2 +-
 arch/arm64/kernel/sleep.S        | 3 ---
 arch/arm64/kernel/suspend.c      | 6 ------
 arch/arm64/mm/proc.S             | 6 ++++++
 4 files changed, 7 insertions(+), 10 deletions(-)

diff --git a/arch/arm64/include/asm/suspend.h b/arch/arm64/include/asm/suspend.h
index 024d623..92d6a62 100644
--- a/arch/arm64/include/asm/suspend.h
+++ b/arch/arm64/include/asm/suspend.h
@@ -1,7 +1,7 @@
 #ifndef __ASM_SUSPEND_H
 #define __ASM_SUSPEND_H
 
-#define NR_CTX_REGS 10
+#define NR_CTX_REGS 12
 #define NR_CALLEE_SAVED_REGS 12
 
 /*
diff --git a/arch/arm64/kernel/sleep.S b/arch/arm64/kernel/sleep.S
index ccf79d8..74e6e19 100644
--- a/arch/arm64/kernel/sleep.S
+++ b/arch/arm64/kernel/sleep.S
@@ -132,9 +132,6 @@ ENTRY(_cpu_resume)
        /* load sp from context */
        ldr     x2, [x0, #CPU_CTX_SP]
        mov     sp, x2
-       /* save thread_info */
-       and     x2, x2, #~(THREAD_SIZE - 1)
-       msr     sp_el0, x2
        /*
         * cpu_do_resume expects x0 to contain context address pointer
         */
diff --git a/arch/arm64/kernel/suspend.c b/arch/arm64/kernel/suspend.c
index b616e365..9c33a49 100644
--- a/arch/arm64/kernel/suspend.c
+++ b/arch/arm64/kernel/suspend.c
@@ -42,12 +42,6 @@ void notrace __cpu_suspend_exit(void)
        cpu_uninstall_idmap();
 
        /*
-        * Restore per-cpu offset before any kernel
-        * subsystem relying on it has a chance to run.
-        */
-       set_my_cpu_offset(per_cpu_offset(smp_processor_id()));
-
-       /*
         * Restore HW breakpoint registers to sane values
         * before debug exceptions are possibly reenabled
         * through local_dbg_restore.
diff --git a/arch/arm64/mm/proc.S b/arch/arm64/mm/proc.S
index 5bb61de..0293db1 100644
--- a/arch/arm64/mm/proc.S
+++ b/arch/arm64/mm/proc.S
@@ -70,11 +70,14 @@ ENTRY(cpu_do_suspend)
        mrs     x8, mdscr_el1
        mrs     x9, oslsr_el1
        mrs     x10, sctlr_el1
+       mrs     x11, tpidr_el1
+       mrs     x12, sp_el0
        stp     x2, x3, [x0]
        stp     x4, xzr, [x0, #16]
        stp     x5, x6, [x0, #32]
        stp     x7, x8, [x0, #48]
        stp     x9, x10, [x0, #64]
+       stp     x11, x12, [x0, #80]
        ret
 ENDPROC(cpu_do_suspend)
 
@@ -89,6 +92,7 @@ ENTRY(cpu_do_resume)
        ldp     x6, x8, [x0, #32]
        ldp     x9, x10, [x0, #48]
        ldp     x11, x12, [x0, #64]
+       ldp     x13, x14, [x0, #80]
        msr     tpidr_el0, x2
        msr     tpidrro_el0, x3
        msr     contextidr_el1, x4
@@ -102,6 +106,8 @@ ENTRY(cpu_do_resume)
        msr     vbar_el1, x9
        msr     mdscr_el1, x10
        msr     sctlr_el1, x12
+       msr     tpidr_el1, x13
+       msr     sp_el0, x14
        /*
         * Restore oslsr_el1 by writing oslar_el1
         */
-- 
1.9.1

Reply via email to