75b8281a9af5e0fa5599555196c309fdb1e46180
[dragonfly.git] / lib / libc / amd64 / gen / makecontext.c
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  * $FreeBSD: src/lib/libc/amd64/gen/makecontext.c,v 1.2 2003/12/05 01:36:44 peter Exp $
27  * $DragonFly: src/lib/libc/amd64/gen/Attic/makecontext.c,v 1.1 2004/02/02 05:43:14 dillon Exp $
28  */
29
30
31 #include <sys/types.h>
32 #include <sys/ucontext.h>
33 #include <stdarg.h>
34 #include <stdlib.h>
35
36 typedef void (*func_t)(uint64_t, uint64_t, uint64_t, uint64_t, uint64_t,
37     uint64_t);
38
39 /* Prototypes */
40 static void ctx_wrapper(ucontext_t *ucp, func_t func, uint64_t *args);
41
42 __weak_reference(__makecontext, makecontext);
43
44 void
45 __makecontext(ucontext_t *ucp, void (*start)(void), int argc, ...)
46 {
47         uint64_t *args;
48         uint64_t *sp;
49         va_list ap;
50         int i;
51
52         /* A valid context is required. */
53         if ((ucp == NULL) || (ucp->uc_mcontext.mc_len != sizeof(mcontext_t)))
54                 return;
55         else if ((argc < 0) || (argc > 6) || (ucp->uc_stack.ss_sp == NULL) ||
56             (ucp->uc_stack.ss_size < MINSIGSTKSZ)) {
57                 /*
58                  * This should really return -1 with errno set to ENOMEM
59                  * or something, but the spec says that makecontext is
60                  * a void function.   At least make sure that the context
61                  * isn't valid so it can't be used without an error.
62                  */
63                 ucp->uc_mcontext.mc_len = 0;
64                 return;
65         }
66
67         /* Align the stack to 16 bytes. */
68         sp = (uint64_t *)(ucp->uc_stack.ss_sp + ucp->uc_stack.ss_size);
69         sp = (uint64_t *)((uint64_t)sp & -15UL);
70
71         /* Allocate space for a maximum of 6 arguments on the stack. */
72         args = sp - 6;
73
74         /*
75          * Account for arguments on stack and do the funky C entry alignment.
76          * This means that we need an 8-byte-odd alignment since the ABI expects
77          * the return address to be pushed, thus breaking the 16 byte alignment.
78          */
79         sp -= 7;
80
81         /* Add the arguments: */
82         va_start(ap, argc);
83         for (i = 0; i < argc; i++)
84                 args[i] = va_arg(ap, uint64_t);
85         va_end(ap);
86         for (i = argc; i < 6; i++)
87                 args[i] = 0;
88
89         ucp->uc_mcontext.mc_rdi = (register_t)ucp;
90         ucp->uc_mcontext.mc_rsi = (register_t)start;
91         ucp->uc_mcontext.mc_rdx = (register_t)args;
92         ucp->uc_mcontext.mc_rbp = (register_t)sp;
93         ucp->uc_mcontext.mc_rbx = (register_t)sp;
94         ucp->uc_mcontext.mc_rsp = (register_t)sp;
95         ucp->uc_mcontext.mc_rip = (register_t)ctx_wrapper;
96 }
97
98 static void
99 ctx_wrapper(ucontext_t *ucp, func_t func, uint64_t *args)
100 {
101         (*func)(args[0], args[1], args[2], args[3], args[4], args[5]);
102         if (ucp->uc_link == NULL)
103                 exit(0);
104         setcontext((const ucontext_t *)ucp->uc_link);
105         /* should never get here */
106         abort();
107         /* NOTREACHED */
108 }