    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25

    use crate::trap::trap_return;

    #[repr(C)]
    pub struct TaskContext {
        ra: usize,
        sp: usize,
        s: [usize; 12],
    }

    impl TaskContext {
        pub fn zero_init() -> Self {
            Self {
                ra: 0,
                sp: 0,
                s: [0; 12],
            }
        }
        pub fn goto_trap_return(kstack_ptr: usize) -> Self {
            Self {
                ra: trap_return as usize,
                sp: kstack_ptr,
                s: [0; 12],
            }
        }
    }