Add sys/resident.h for userland syscall prototypes, and give the unregister
[dragonfly.git] / sys / kern / imgact_resident.c
1 /*
2  * Copyright (c) 2003 Matthew Dillon <dillon@backplane.com>
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  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $DragonFly: src/sys/kern/imgact_resident.c,v 1.2 2004/01/20 21:03:23 dillon Exp $
27  */
28
29 #include <sys/param.h>
30 #include <sys/systm.h>
31 #include <sys/kernel.h>
32 #include <sys/sysproto.h>
33 #include <sys/exec.h>
34 #include <sys/imgact.h>
35 #include <sys/imgact_aout.h>
36 #include <sys/mman.h>
37 #include <sys/proc.h>
38 #include <sys/resourcevar.h>
39 #include <sys/sysent.h>
40 #include <sys/systm.h>
41 #include <sys/vnode.h>
42 #include <sys/inflate.h>
43
44 #include <vm/vm.h>
45 #include <vm/vm_param.h>
46 #include <sys/lock.h>
47 #include <vm/pmap.h>
48 #include <vm/vm_map.h>
49 #include <vm/vm_kern.h>
50 #include <vm/vm_extern.h>
51
52 int
53 exec_resident_imgact(struct image_params *imgp)
54 {
55         struct vmresident *vmres;
56
57         /*
58          * resident image activator
59          */
60         if ((vmres = imgp->vp->v_resident) == NULL)
61             return(-1);
62         exec_new_vmspace(imgp, vmres->vr_vmspace);
63         imgp->resident = 1;
64         imgp->interpreted = 0;
65         imgp->proc->p_sysent = vmres->vr_sysent;
66         imgp->entry_addr = vmres->vr_entry_addr;
67         return(0);
68 }
69
70 static int exec_res_id;
71 static TAILQ_HEAD(,vmresident) exec_res_list = TAILQ_HEAD_INITIALIZER(exec_res_list);
72
73 static MALLOC_DEFINE(M_EXEC_RES, "vmresident", "resident execs");
74
75 /*
76  * exec_sys_register(entry)
77  *
78  * Register ourselves for resident execution.  Only root can do this.  This
79  * will snapshot the vmspace and cause future exec's of the specified binary
80  * to use the snapshot directly rather then load & relocate a new copy.
81  */
82 int
83 exec_sys_register(struct exec_sys_register_args *uap)
84 {
85     struct vmresident *vmres;
86     struct vnode *vp;
87     struct proc *p;
88     int error;
89
90     p = curproc;
91     if ((error = suser(p->p_thread)) != 0)
92         return(error);
93     if ((vp = p->p_textvp) == NULL)
94         return(ENOENT);
95     if (vp->v_resident)
96         return(EEXIST);
97     vhold(vp);
98     vmres = malloc(sizeof(*vmres), M_EXEC_RES, M_WAITOK);
99     vp->v_resident = vmres;
100     vmres->vr_vnode = vp;
101     vmres->vr_sysent = p->p_sysent;
102     vmres->vr_id = ++exec_res_id;
103     vmres->vr_entry_addr = (intptr_t)uap->entry;
104     vmres->vr_vmspace = vmspace_fork(p->p_vmspace); /* XXX order */
105     TAILQ_INSERT_TAIL(&exec_res_list, vmres, vr_link);
106     return(0);
107 }
108
109 /*
110  * exec_sys_unregister(id)
111  *
112  *      Unregister the specified id.  If an id of -1 is used unregister
113  *      the registration associated with the current process.  An id of -2
114  *      unregisters everything.
115  */
116 int
117 exec_sys_unregister(struct exec_sys_unregister_args *uap)
118 {
119     struct vmresident *vmres;
120     struct proc *p;
121     int error;
122     int id;
123     int count;
124
125     p = curproc;
126     if ((error = suser(p->p_thread)) != 0)
127         return(error);
128
129     /*
130      * If id is -1, unregister ourselves
131      */
132     if ((id = uap->id) == -1 && p->p_textvp && p->p_textvp->v_resident)
133         id = p->p_textvp->v_resident->vr_id;
134
135     /*
136      * Look for the registration
137      */
138     error = ENOENT;
139     count = 0;
140 restart:
141     TAILQ_FOREACH(vmres, &exec_res_list, vr_link) {
142         if (id == -2 || vmres->vr_id == id) {
143             TAILQ_REMOVE(&exec_res_list, vmres, vr_link);
144             if (vmres->vr_vnode) {
145                 vmres->vr_vnode->v_resident = NULL;
146                 vdrop(vmres->vr_vnode);
147                 vmres->vr_vnode = NULL;
148             }
149             if (vmres->vr_vmspace) {
150                 vmspace_free(vmres->vr_vmspace);
151                 vmres->vr_vmspace = NULL;
152             }
153             free(vmres, M_EXEC_RES);
154             error = 0;
155             ++count;
156             goto restart;
157         }
158     }
159     if (error == 0)
160         uap->sysmsg_result = count;
161     return(error);
162 }
163