2018-01-30 03:41:02 +00:00
|
|
|
/*
|
2020-03-27 09:23:49 +00:00
|
|
|
* Copyright (c) 2018 Ariadne Conill <ariadne@dereferenced.org>
|
2018-01-30 03:41:02 +00:00
|
|
|
*
|
|
|
|
* 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 <stddef.h>
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
|
2020-03-29 14:27:40 +00:00
|
|
|
#include "defs.h"
|
2021-01-03 05:08:54 +00:00
|
|
|
#include <libucontext/libucontext.h>
|
2018-01-30 03:41:02 +00:00
|
|
|
|
2020-12-06 08:59:59 +00:00
|
|
|
extern void libucontext_trampoline(void);
|
2018-01-30 03:41:02 +00:00
|
|
|
|
2021-01-08 09:56:45 +00:00
|
|
|
_Static_assert(offsetof(libucontext_ucontext_t, uc_mcontext.gregs) == MCONTEXT_GREGS, "MCONTEXT_GREGS is invalid");
|
2018-01-30 03:41:02 +00:00
|
|
|
|
|
|
|
void
|
2020-12-06 10:22:27 +00:00
|
|
|
libucontext_makecontext(libucontext_ucontext_t *ucp, void (*func)(void), int argc, ...)
|
2018-01-30 03:41:02 +00:00
|
|
|
{
|
2020-12-06 09:44:45 +00:00
|
|
|
libucontext_greg_t *sp;
|
2018-01-30 03:41:02 +00:00
|
|
|
va_list va;
|
|
|
|
int i;
|
|
|
|
unsigned int uc_link;
|
|
|
|
|
|
|
|
uc_link = (argc > 6 ? argc - 6 : 0) + 1;
|
|
|
|
|
2020-12-06 09:44:45 +00:00
|
|
|
sp = (libucontext_greg_t *) ((uintptr_t) ucp->uc_stack.ss_sp + ucp->uc_stack.ss_size);
|
2018-01-30 03:41:02 +00:00
|
|
|
sp -= uc_link;
|
2020-12-06 09:44:45 +00:00
|
|
|
sp = (libucontext_greg_t *) (((uintptr_t) sp & -16L) - 8);
|
2018-01-30 03:41:02 +00:00
|
|
|
|
|
|
|
ucp->uc_mcontext.gregs[REG_RIP] = (uintptr_t) func;
|
|
|
|
ucp->uc_mcontext.gregs[REG_RBX] = (uintptr_t) &sp[uc_link];
|
|
|
|
ucp->uc_mcontext.gregs[REG_RSP] = (uintptr_t) sp;
|
|
|
|
|
2020-12-06 08:59:59 +00:00
|
|
|
sp[0] = (uintptr_t) &libucontext_trampoline;
|
2018-01-30 03:41:02 +00:00
|
|
|
sp[uc_link] = (uintptr_t) ucp->uc_link;
|
|
|
|
|
|
|
|
va_start(va, argc);
|
|
|
|
|
|
|
|
for (i = 0; i < argc; i++)
|
|
|
|
switch (i)
|
|
|
|
{
|
|
|
|
case 0:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_RDI] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
case 1:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_RSI] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
case 2:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_RDX] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
case 3:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_RCX] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
case 4:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_R8] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
case 5:
|
2020-12-06 09:44:45 +00:00
|
|
|
ucp->uc_mcontext.gregs[REG_R9] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
default:
|
2020-12-06 09:44:45 +00:00
|
|
|
sp[i - 5] = va_arg (va, libucontext_greg_t);
|
2018-01-30 03:41:02 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
va_end(va);
|
|
|
|
}
|
|
|
|
|
2021-01-03 02:33:27 +00:00
|
|
|
#ifdef EXPORT_UNPREFIXED
|
2020-12-06 08:56:59 +00:00
|
|
|
extern __typeof(libucontext_makecontext) makecontext __attribute__((weak, __alias__("libucontext_makecontext")));
|
2021-01-08 10:37:27 +00:00
|
|
|
extern __typeof(libucontext_makecontext) __makecontext __attribute__((weak, __alias__("libucontext_makecontext")));
|
2021-01-03 02:33:27 +00:00
|
|
|
#endif
|