switch.S 983 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. /*
  3. * Copyright (C) 2020-2022 Loongson Technology Corporation Limited
  4. */
  5. #include <asm/asm.h>
  6. #include <asm/asmmacro.h>
  7. #include <asm/asm-offsets.h>
  8. #include <asm/loongarch.h>
  9. #include <asm/regdef.h>
  10. #include <asm/stackframe.h>
  11. #include <asm/thread_info.h>
  12. /*
  13. * task_struct *__switch_to(task_struct *prev, task_struct *next,
  14. * struct thread_info *next_ti)
  15. */
  16. .align 5
  17. SYM_FUNC_START(__switch_to)
  18. csrrd t1, LOONGARCH_CSR_PRMD
  19. stptr.d t1, a0, THREAD_CSRPRMD
  20. cpu_save_nonscratch a0
  21. stptr.d ra, a0, THREAD_REG01
  22. stptr.d a3, a0, THREAD_SCHED_RA
  23. stptr.d a4, a0, THREAD_SCHED_CFA
  24. #if defined(CONFIG_STACKPROTECTOR) && !defined(CONFIG_SMP)
  25. la t7, __stack_chk_guard
  26. LONG_L t8, a1, TASK_STACK_CANARY
  27. LONG_S t8, t7, 0
  28. #endif
  29. move tp, a2
  30. cpu_restore_nonscratch a1
  31. li.w t0, _THREAD_SIZE
  32. PTR_ADD t0, t0, tp
  33. set_saved_sp t0, t1, t2
  34. ldptr.d t1, a1, THREAD_CSRPRMD
  35. csrwr t1, LOONGARCH_CSR_PRMD
  36. jr ra
  37. SYM_FUNC_END(__switch_to)