1 | // Copyright 2022 The Marl Authors. |
2 | // |
3 | // Licensed under the Apache License. Version 2.0 (the "License"); |
4 | // you may not use this file except in compliance with the License. |
5 | // You may obtain a copy of the License at |
6 | // |
7 | // https://www.apache.org/licenses/LICENSE-2.0 |
8 | // |
9 | // Unless required by applicable law or agreed to in writing, software |
10 | // distributed under the License is distributed on an "AS IS" BASIS, |
11 | // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
12 | // See the License for the specific language governing permissions and |
13 | // limitations under the License. |
14 | |
15 | #if defined(__loongarch_lp64) |
16 | |
17 | #include "osfiber_asm_loongarch64.h" |
18 | |
19 | #include "marl/export.h" |
20 | |
21 | MARL_EXPORT |
22 | void marl_fiber_trampoline(void (*target)(void*), void* arg) { |
23 | target(arg); |
24 | } |
25 | |
26 | MARL_EXPORT |
27 | void marl_fiber_set_target(struct marl_fiber_context* ctx, |
28 | void* stack, |
29 | uint32_t stack_size, |
30 | void (*target)(void*), |
31 | void* arg) { |
32 | uintptr_t* stack_top = (uintptr_t*)((uint8_t*)(stack) + stack_size); |
33 | ctx->ra = (uintptr_t)&marl_fiber_trampoline; |
34 | ctx->a0 = (uintptr_t)target; |
35 | ctx->a1 = (uintptr_t)arg; |
36 | ctx->sp = ((uintptr_t)stack_top) & ~(uintptr_t)15; |
37 | } |
38 | |
39 | #endif // defined(__loongarch_lp64) |
40 | |