1 /* 2 * Copyright (c) 2003 Marcel Moolenaar 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 #include <sys/types.h> 31 #include <sys/ucontext.h> 32 #include <stdarg.h> 33 #include <stdlib.h> 34 35 typedef void (*func_t)(uint64_t, uint64_t, uint64_t, uint64_t, uint64_t, 36 uint64_t); 37 38 /* Prototypes */ 39 static void makectx_wrapper(ucontext_t *ucp, func_t func, uint64_t *args); 40 41 __weak_reference(__makecontext, makecontext); 42 43 void 44 __makecontext(ucontext_t *ucp, void (*start)(void), int argc, ...) 45 { 46 uint64_t *args; 47 uint64_t *sp; 48 va_list ap; 49 int i; 50 51 /* A valid context is required. */ 52 if ((ucp == NULL) || (ucp->uc_mcontext.mc_len != sizeof(mcontext_t))) 53 return; 54 else if ((argc < 0) || (argc > 6) || (ucp->uc_stack.ss_sp == NULL) || 55 (ucp->uc_stack.ss_size < MINSIGSTKSZ)) { 56 /* 57 * This should really return -1 with errno set to ENOMEM 58 * or something, but the spec says that makecontext is 59 * a void function. At least make sure that the context 60 * isn't valid so it can't be used without an error. 61 */ 62 ucp->uc_mcontext.mc_len = 0; 63 return; 64 } 65 66 /* Align the stack to 16 bytes. */ 67 sp = (uint64_t *)(ucp->uc_stack.ss_sp + ucp->uc_stack.ss_size); 68 sp = (uint64_t *)((uint64_t)sp & ~15UL); 69 70 /* Allocate space for a maximum of 6 arguments on the stack. */ 71 args = sp - 6; 72 73 /* 74 * Account for arguments on stack and do the funky C entry alignment. 75 * This means that we need an 8-byte-odd alignment since the ABI expects 76 * the return address to be pushed, thus breaking the 16 byte alignment. 77 */ 78 sp -= 7; 79 80 /* Add the arguments: */ 81 va_start(ap, argc); 82 for (i = 0; i < argc; i++) 83 args[i] = va_arg(ap, uint64_t); 84 va_end(ap); 85 for (i = argc; i < 6; i++) 86 args[i] = 0; 87 88 ucp->uc_mcontext.mc_rdi = (register_t)ucp; 89 ucp->uc_mcontext.mc_rsi = (register_t)start; 90 ucp->uc_mcontext.mc_rdx = (register_t)args; 91 ucp->uc_mcontext.mc_rbp = 0; 92 ucp->uc_mcontext.mc_rbx = (register_t)sp; 93 ucp->uc_mcontext.mc_rsp = (register_t)sp; 94 ucp->uc_mcontext.mc_rip = (register_t)makectx_wrapper; 95 } 96 97 static void 98 makectx_wrapper(ucontext_t *ucp, func_t func, uint64_t *args) 99 { 100 (*func)(args[0], args[1], args[2], args[3], args[4], args[5]); 101 if (ucp->uc_link == NULL) 102 exit(0); 103 setcontext((const ucontext_t *)ucp->uc_link); 104 /* should never get here */ 105 abort(); 106 /* NOTREACHED */ 107 } 108