forked from ariadne/libucontext
75 lines
2.5 KiB
ArmAsm
75 lines
2.5 KiB
ArmAsm
/*
|
|
* Copyright (c) 2018, 2020 Ariadne Conill <ariadne@dereferenced.org>
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* This software is provided 'as is' and without any warranty, express or
|
|
* implied. In no event shall the authors be liable for any damages arising
|
|
* from the use of this software.
|
|
*/
|
|
|
|
#include "defs.h"
|
|
|
|
ALIAS(swapcontext, __swapcontext)
|
|
|
|
FUNC(__swapcontext)
|
|
/* copy all of the current registers into the ucontext structure pointed by
|
|
the first argument */
|
|
movq %r8, REG_OFFSET(REG_R8)(%rdi)
|
|
movq %r9, REG_OFFSET(REG_R9)(%rdi)
|
|
movq %r10, REG_OFFSET(REG_R10)(%rdi)
|
|
movq %r11, REG_OFFSET(REG_R11)(%rdi)
|
|
movq %r12, REG_OFFSET(REG_R12)(%rdi)
|
|
movq %r13, REG_OFFSET(REG_R13)(%rdi)
|
|
movq %r14, REG_OFFSET(REG_R14)(%rdi)
|
|
movq %r15, REG_OFFSET(REG_R15)(%rdi)
|
|
movq %rdi, REG_OFFSET(REG_RDI)(%rdi)
|
|
movq %rsi, REG_OFFSET(REG_RSI)(%rdi)
|
|
movq %rbp, REG_OFFSET(REG_RBP)(%rdi)
|
|
movq %rbx, REG_OFFSET(REG_RBX)(%rdi)
|
|
movq %rdx, REG_OFFSET(REG_RDX)(%rdi)
|
|
movq %rax, REG_OFFSET(REG_RAX)(%rdi)
|
|
movq %rcx, REG_OFFSET(REG_RCX)(%rdi)
|
|
|
|
/* the first argument on the stack is the jump target (%rip), so we store it in the RIP
|
|
register in the ucontext structure. */
|
|
movq (%rsp), %rcx
|
|
movq %rcx, REG_OFFSET(REG_RIP)(%rdi)
|
|
|
|
/* finally take the stack pointer address (%rsp) offsetting by 8 to skip over the jump
|
|
target. */
|
|
leaq 8(%rsp), %rcx
|
|
movq %rcx, REG_OFFSET(REG_RSP)(%rdi)
|
|
|
|
/* set all of the registers to their new states, stored in the second
|
|
ucontext structure */
|
|
movq REG_OFFSET(REG_R8)(%rsi), %r8
|
|
movq REG_OFFSET(REG_R9)(%rsi), %r9
|
|
movq REG_OFFSET(REG_R10)(%rsi), %r10
|
|
movq REG_OFFSET(REG_R11)(%rsi), %r11
|
|
movq REG_OFFSET(REG_R12)(%rsi), %r12
|
|
movq REG_OFFSET(REG_R13)(%rsi), %r13
|
|
movq REG_OFFSET(REG_R14)(%rsi), %r14
|
|
movq REG_OFFSET(REG_R15)(%rsi), %r15
|
|
movq REG_OFFSET(REG_RDI)(%rsi), %rdi
|
|
movq REG_OFFSET(REG_RBP)(%rsi), %rbp
|
|
movq REG_OFFSET(REG_RBX)(%rsi), %rbx
|
|
movq REG_OFFSET(REG_RDX)(%rsi), %rdx
|
|
movq REG_OFFSET(REG_RAX)(%rsi), %rax
|
|
movq REG_OFFSET(REG_RCX)(%rsi), %rcx
|
|
movq REG_OFFSET(REG_RSP)(%rsi), %rsp
|
|
|
|
/* set the jump target by pushing it to the stack.
|
|
ret will pop the new %rip from the stack, causing us to jump there. */
|
|
pushq REG_OFFSET(REG_RIP)(%rsi)
|
|
|
|
/* finally, set %rsi correctly since we do not need it anymore. */
|
|
movq REG_OFFSET(REG_RSI)(%rsi), %rsi
|
|
|
|
/* we're all done here, return 0 */
|
|
xorl %eax, %eax
|
|
ret
|
|
END(__swapcontext)
|