2 * Copyright (C) 2012 Regents of the University of California
3 * Copyright (C) 2017 SiFive
5 * This program is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU General Public License
7 * as published by the Free Software Foundation, version 2.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
15 #include <linux/init.h>
16 #include <linux/linkage.h>
20 #include <asm/unistd.h>
21 #include <asm/thread_info.h>
22 #include <asm/asm-offsets.h>
28 * Prepares to enter a system call or exception by saving all registers to the
32 LOCAL _restore_kernel_tpsp
36 * If coming from userspace, preserve the user thread pointer and load
37 * the kernel thread pointer. If we came from the kernel, sscratch
38 * will contain 0, and we should continue on the current TP.
40 csrrw tp, sscratch, tp
41 bnez tp, _save_context
45 REG_S sp, TASK_TI_KERNEL_SP(tp)
47 REG_S sp, TASK_TI_USER_SP(tp)
48 REG_L sp, TASK_TI_KERNEL_SP(tp)
49 addi sp, sp, -(PT_SIZE_ON_STACK)
81 * Disable FPU to detect illegal usage of
82 * floating point in kernel space
86 REG_L s0, TASK_TI_USER_SP(tp)
93 REG_S s1, PT_SSTATUS(sp)
95 REG_S s3, PT_SBADADDR(sp)
96 REG_S s4, PT_SCAUSE(sp)
101 * Prepares to return from a system call or exception by restoring all
102 * registers from the stack.
105 REG_L a0, PT_SSTATUS(sp)
106 REG_L a2, PT_SEPC(sp)
134 REG_L x26, PT_S10(sp)
135 REG_L x27, PT_S11(sp)
144 ENTRY(handle_exception)
148 * Set sscratch register to 0, so that if a recursive exception
149 * occurs, the exception vector knows it came from the kernel
153 /* Load the global pointer */
156 la gp, __global_pointer$
159 la ra, ret_from_exception
161 * MSB of cause differentiates between
162 * interrupts and exceptions
166 /* Handle interrupts */
169 move a1, sp /* pt_regs */
172 /* Handle syscalls */
174 beq s4, t0, handle_syscall
176 /* Handle other exceptions */
177 slli t0, s4, RISCV_LGPTR
178 la t1, excp_vect_table
179 la t2, excp_vect_table_end
180 move a0, sp /* pt_regs */
182 /* Check if exception code lies within bounds */
190 /* save the initial A0 value (needed in signal handlers) */
191 REG_S a0, PT_ORIG_A0(sp)
193 * Advance SEPC to avoid executing the original
194 * scall instruction on sret
197 REG_S s2, PT_SEPC(sp)
198 /* System calls run with interrupts enabled */
200 /* Trace syscalls, but only if requested by the user. */
201 REG_L t0, TASK_TI_FLAGS(tp)
202 andi t0, t0, _TIF_SYSCALL_TRACE
203 bnez t0, handle_syscall_trace_enter
205 /* Check to make sure we don't jump to a bogus syscall number. */
207 la s0, sys_ni_syscall
208 /* Syscall number held in a7 */
210 la s0, sys_call_table
211 slli t0, a7, RISCV_LGPTR
218 /* Set user a0 to kernel a0 */
220 /* Trace syscalls, but only if requested by the user. */
221 REG_L t0, TASK_TI_FLAGS(tp)
222 andi t0, t0, _TIF_SYSCALL_TRACE
223 bnez t0, handle_syscall_trace_exit
226 REG_L s0, PT_SSTATUS(sp)
232 /* Interrupts must be disabled here so flags are checked atomically */
233 REG_L s0, TASK_TI_FLAGS(tp) /* current_thread_info->flags */
234 andi s1, s0, _TIF_WORK_MASK
235 bnez s1, work_pending
237 /* Save unwound kernel stack pointer in thread_info */
238 addi s0, sp, PT_SIZE_ON_STACK
239 REG_S s0, TASK_TI_KERNEL_SP(tp)
242 * Save TP into sscratch, so we can find the kernel data structures
252 /* Enter slow path for supplementary processing */
253 la ra, ret_from_exception
254 andi s1, s0, _TIF_NEED_RESCHED
255 bnez s1, work_resched
257 /* Handle pending signals and notify-resume requests */
258 csrs sstatus, SR_SIE /* Enable interrupts for do_notify_resume() */
259 move a0, sp /* pt_regs */
260 move a1, s0 /* current_thread_info->flags */
261 tail do_notify_resume
265 /* Slow paths for ptrace. */
266 handle_syscall_trace_enter:
268 call do_syscall_trace_enter
278 handle_syscall_trace_exit:
280 call do_syscall_trace_exit
283 END(handle_exception)
286 la ra, ret_from_exception
288 ENDPROC(ret_from_fork)
290 ENTRY(ret_from_kernel_thread)
293 la ra, ret_from_exception
296 ENDPROC(ret_from_kernel_thread)
300 * Integer register context switch
301 * The callee-saved registers must be saved and restored.
303 * a0: previous task_struct (must be preserved across the switch)
304 * a1: next task_struct
306 * The value of a0 and a1 must be preserved by this function, as that's how
307 * arguments are passed to schedule_tail.
310 /* Save context into prev->thread */
311 li a4, TASK_THREAD_RA
314 REG_S ra, TASK_THREAD_RA_RA(a3)
315 REG_S sp, TASK_THREAD_SP_RA(a3)
316 REG_S s0, TASK_THREAD_S0_RA(a3)
317 REG_S s1, TASK_THREAD_S1_RA(a3)
318 REG_S s2, TASK_THREAD_S2_RA(a3)
319 REG_S s3, TASK_THREAD_S3_RA(a3)
320 REG_S s4, TASK_THREAD_S4_RA(a3)
321 REG_S s5, TASK_THREAD_S5_RA(a3)
322 REG_S s6, TASK_THREAD_S6_RA(a3)
323 REG_S s7, TASK_THREAD_S7_RA(a3)
324 REG_S s8, TASK_THREAD_S8_RA(a3)
325 REG_S s9, TASK_THREAD_S9_RA(a3)
326 REG_S s10, TASK_THREAD_S10_RA(a3)
327 REG_S s11, TASK_THREAD_S11_RA(a3)
328 /* Restore context from next->thread */
329 REG_L ra, TASK_THREAD_RA_RA(a4)
330 REG_L sp, TASK_THREAD_SP_RA(a4)
331 REG_L s0, TASK_THREAD_S0_RA(a4)
332 REG_L s1, TASK_THREAD_S1_RA(a4)
333 REG_L s2, TASK_THREAD_S2_RA(a4)
334 REG_L s3, TASK_THREAD_S3_RA(a4)
335 REG_L s4, TASK_THREAD_S4_RA(a4)
336 REG_L s5, TASK_THREAD_S5_RA(a4)
337 REG_L s6, TASK_THREAD_S6_RA(a4)
338 REG_L s7, TASK_THREAD_S7_RA(a4)
339 REG_L s8, TASK_THREAD_S8_RA(a4)
340 REG_L s9, TASK_THREAD_S9_RA(a4)
341 REG_L s10, TASK_THREAD_S10_RA(a4)
342 REG_L s11, TASK_THREAD_S11_RA(a4)
343 /* Swap the CPU entry around. */
344 lw a3, TASK_TI_CPU(a0)
345 lw a4, TASK_TI_CPU(a1)
346 sw a3, TASK_TI_CPU(a1)
347 sw a4, TASK_TI_CPU(a0)
349 #error "TASK_TI != 0: tp will contain a 'struct thread_info', not a 'struct task_struct' so get_current() won't work."
358 li a2, TASK_THREAD_F0
363 fsd f0, TASK_THREAD_F0_F0(a0)
364 fsd f1, TASK_THREAD_F1_F0(a0)
365 fsd f2, TASK_THREAD_F2_F0(a0)
366 fsd f3, TASK_THREAD_F3_F0(a0)
367 fsd f4, TASK_THREAD_F4_F0(a0)
368 fsd f5, TASK_THREAD_F5_F0(a0)
369 fsd f6, TASK_THREAD_F6_F0(a0)
370 fsd f7, TASK_THREAD_F7_F0(a0)
371 fsd f8, TASK_THREAD_F8_F0(a0)
372 fsd f9, TASK_THREAD_F9_F0(a0)
373 fsd f10, TASK_THREAD_F10_F0(a0)
374 fsd f11, TASK_THREAD_F11_F0(a0)
375 fsd f12, TASK_THREAD_F12_F0(a0)
376 fsd f13, TASK_THREAD_F13_F0(a0)
377 fsd f14, TASK_THREAD_F14_F0(a0)
378 fsd f15, TASK_THREAD_F15_F0(a0)
379 fsd f16, TASK_THREAD_F16_F0(a0)
380 fsd f17, TASK_THREAD_F17_F0(a0)
381 fsd f18, TASK_THREAD_F18_F0(a0)
382 fsd f19, TASK_THREAD_F19_F0(a0)
383 fsd f20, TASK_THREAD_F20_F0(a0)
384 fsd f21, TASK_THREAD_F21_F0(a0)
385 fsd f22, TASK_THREAD_F22_F0(a0)
386 fsd f23, TASK_THREAD_F23_F0(a0)
387 fsd f24, TASK_THREAD_F24_F0(a0)
388 fsd f25, TASK_THREAD_F25_F0(a0)
389 fsd f26, TASK_THREAD_F26_F0(a0)
390 fsd f27, TASK_THREAD_F27_F0(a0)
391 fsd f28, TASK_THREAD_F28_F0(a0)
392 fsd f29, TASK_THREAD_F29_F0(a0)
393 fsd f30, TASK_THREAD_F30_F0(a0)
394 fsd f31, TASK_THREAD_F31_F0(a0)
395 sw t0, TASK_THREAD_FCSR_F0(a0)
398 ENDPROC(__fstate_save)
400 ENTRY(__fstate_restore)
401 li a2, TASK_THREAD_F0
404 lw t0, TASK_THREAD_FCSR_F0(a0)
406 fld f0, TASK_THREAD_F0_F0(a0)
407 fld f1, TASK_THREAD_F1_F0(a0)
408 fld f2, TASK_THREAD_F2_F0(a0)
409 fld f3, TASK_THREAD_F3_F0(a0)
410 fld f4, TASK_THREAD_F4_F0(a0)
411 fld f5, TASK_THREAD_F5_F0(a0)
412 fld f6, TASK_THREAD_F6_F0(a0)
413 fld f7, TASK_THREAD_F7_F0(a0)
414 fld f8, TASK_THREAD_F8_F0(a0)
415 fld f9, TASK_THREAD_F9_F0(a0)
416 fld f10, TASK_THREAD_F10_F0(a0)
417 fld f11, TASK_THREAD_F11_F0(a0)
418 fld f12, TASK_THREAD_F12_F0(a0)
419 fld f13, TASK_THREAD_F13_F0(a0)
420 fld f14, TASK_THREAD_F14_F0(a0)
421 fld f15, TASK_THREAD_F15_F0(a0)
422 fld f16, TASK_THREAD_F16_F0(a0)
423 fld f17, TASK_THREAD_F17_F0(a0)
424 fld f18, TASK_THREAD_F18_F0(a0)
425 fld f19, TASK_THREAD_F19_F0(a0)
426 fld f20, TASK_THREAD_F20_F0(a0)
427 fld f21, TASK_THREAD_F21_F0(a0)
428 fld f22, TASK_THREAD_F22_F0(a0)
429 fld f23, TASK_THREAD_F23_F0(a0)
430 fld f24, TASK_THREAD_F24_F0(a0)
431 fld f25, TASK_THREAD_F25_F0(a0)
432 fld f26, TASK_THREAD_F26_F0(a0)
433 fld f27, TASK_THREAD_F27_F0(a0)
434 fld f28, TASK_THREAD_F28_F0(a0)
435 fld f29, TASK_THREAD_F29_F0(a0)
436 fld f30, TASK_THREAD_F30_F0(a0)
437 fld f31, TASK_THREAD_F31_F0(a0)
441 ENDPROC(__fstate_restore)
445 /* Exception vector table */
446 ENTRY(excp_vect_table)
447 RISCV_PTR do_trap_insn_misaligned
448 RISCV_PTR do_trap_insn_fault
449 RISCV_PTR do_trap_insn_illegal
450 RISCV_PTR do_trap_break
451 RISCV_PTR do_trap_load_misaligned
452 RISCV_PTR do_trap_load_fault
453 RISCV_PTR do_trap_store_misaligned
454 RISCV_PTR do_trap_store_fault
455 RISCV_PTR do_trap_ecall_u /* system call, gets intercepted */
456 RISCV_PTR do_trap_ecall_s
457 RISCV_PTR do_trap_unknown
458 RISCV_PTR do_trap_ecall_m
459 RISCV_PTR do_page_fault /* instruction page fault */
460 RISCV_PTR do_page_fault /* load page fault */
461 RISCV_PTR do_trap_unknown
462 RISCV_PTR do_page_fault /* store page fault */