Initial import from FreeBSD RELENG_4:
[dragonfly.git] / sys / kern / imgact_aout.c
1 /*
2  * Copyright (c) 1993, David Greenman
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  * $FreeBSD: src/sys/kern/imgact_aout.c,v 1.59.2.5 2001/11/03 01:41:08 ps Exp $
27  */
28
29 #include <sys/param.h>
30 #include <sys/resourcevar.h>
31 #include <sys/exec.h>
32 #include <sys/fcntl.h>
33 #include <sys/imgact.h>
34 #include <sys/imgact_aout.h>
35 #include <sys/kernel.h>
36 #include <sys/malloc.h>
37 #include <sys/namei.h>
38 #include <sys/pioctl.h>
39 #include <sys/proc.h>
40 #include <sys/signalvar.h>
41 #include <sys/stat.h>
42 #include <sys/sysent.h>
43 #include <sys/syscall.h>
44 #include <sys/vnode.h>
45 #include <sys/systm.h>
46 #include <machine/md_var.h>
47
48 #include <vm/vm.h>
49 #include <vm/vm_param.h>
50 #include <sys/lock.h>
51 #include <vm/pmap.h>
52 #include <vm/vm_map.h>
53 #include <vm/vm_object.h>
54 #include <sys/user.h>
55
56 static int      exec_aout_imgact __P((struct image_params *imgp));
57
58 struct sysentvec aout_sysvec = {
59         SYS_MAXSYSCALL,
60         sysent,
61         0,
62         0,
63         0,
64         0,
65         0,
66         0,
67         0,
68         sendsig,
69         sigcode,
70         &szsigcode,
71         0,
72         "FreeBSD a.out",
73         aout_coredump,
74         NULL,
75         MINSIGSTKSZ
76 };
77
78 static int
79 exec_aout_imgact(imgp)
80         struct image_params *imgp;
81 {
82         const struct exec *a_out = (const struct exec *) imgp->image_header;
83         struct vmspace *vmspace;
84         struct vnode *vp;
85         vm_map_t map;
86         vm_object_t object;
87         vm_offset_t text_end, data_end;
88         unsigned long virtual_offset;
89         unsigned long file_offset;
90         unsigned long bss_size;
91         int error;
92
93         /*
94          * Linux and *BSD binaries look very much alike,
95          * only the machine id is different:
96          * 0x64 for Linux, 0x86 for *BSD, 0x00 for BSDI.
97          * NetBSD is in network byte order.. ugh.
98          */
99         if (((a_out->a_magic >> 16) & 0xff) != 0x86 &&
100             ((a_out->a_magic >> 16) & 0xff) != 0 &&
101             ((((int)ntohl(a_out->a_magic)) >> 16) & 0xff) != 0x86)
102                 return -1;
103
104         /*
105          * Set file/virtual offset based on a.out variant.
106          *      We do two cases: host byte order and network byte order
107          *      (for NetBSD compatibility)
108          */
109         switch ((int)(a_out->a_magic & 0xffff)) {
110         case ZMAGIC:
111                 virtual_offset = 0;
112                 if (a_out->a_text) {
113                         file_offset = PAGE_SIZE;
114                 } else {
115                         /* Bill's "screwball mode" */
116                         file_offset = 0;
117                 }
118                 break;
119         case QMAGIC:
120                 virtual_offset = PAGE_SIZE;
121                 file_offset = 0;
122                 /* Pass PS_STRINGS for BSD/OS binaries only. */
123                 if (N_GETMID(*a_out) == MID_ZERO)
124                         imgp->ps_strings = PS_STRINGS;
125                 break;
126         default:
127                 /* NetBSD compatibility */
128                 switch ((int)(ntohl(a_out->a_magic) & 0xffff)) {
129                 case ZMAGIC:
130                 case QMAGIC:
131                         virtual_offset = PAGE_SIZE;
132                         file_offset = 0;
133                         break;
134                 default:
135                         return (-1);
136                 }
137         }
138
139         bss_size = roundup(a_out->a_bss, PAGE_SIZE);
140
141         /*
142          * Check various fields in header for validity/bounds.
143          */
144         if (/* entry point must lay with text region */
145             a_out->a_entry < virtual_offset ||
146             a_out->a_entry >= virtual_offset + a_out->a_text ||
147
148             /* text and data size must each be page rounded */
149             a_out->a_text & PAGE_MASK || a_out->a_data & PAGE_MASK)
150                 return (-1);
151
152         /* text + data can't exceed file size */
153         if (a_out->a_data + a_out->a_text > imgp->attr->va_size)
154                 return (EFAULT);
155
156         /*
157          * text/data/bss must not exceed limits
158          */
159         if (/* text can't exceed maximum text size */
160             a_out->a_text > maxtsiz ||
161
162             /* data + bss can't exceed rlimit */
163             a_out->a_data + bss_size >
164                 imgp->proc->p_rlimit[RLIMIT_DATA].rlim_cur)
165                         return (ENOMEM);
166
167         /* copy in arguments and/or environment from old process */
168         error = exec_extract_strings(imgp);
169         if (error)
170                 return (error);
171
172         /*
173          * Destroy old process VM and create a new one (with a new stack)
174          */
175         exec_new_vmspace(imgp);
176
177         /*
178          * The vm space can be changed by exec_new_vmspace
179          */
180         vmspace = imgp->proc->p_vmspace;
181
182         vp = imgp->vp;
183         map = &vmspace->vm_map;
184         vm_map_lock(map);
185         VOP_GETVOBJECT(vp, &object);
186         vm_object_reference(object);
187
188         text_end = virtual_offset + a_out->a_text;
189         error = vm_map_insert(map, object,
190                 file_offset,
191                 virtual_offset, text_end,
192                 VM_PROT_READ | VM_PROT_EXECUTE, VM_PROT_ALL,
193                 MAP_COPY_ON_WRITE | MAP_PREFAULT);
194         if (error) {
195                 vm_map_unlock(map);
196                 return (error);
197         }
198         data_end = text_end + a_out->a_data;
199         if (a_out->a_data) {
200                 vm_object_reference(object);
201                 error = vm_map_insert(map, object,
202                         file_offset + a_out->a_text,
203                         text_end, data_end,
204                         VM_PROT_ALL, VM_PROT_ALL,
205                         MAP_COPY_ON_WRITE | MAP_PREFAULT);
206                 if (error) {
207                         vm_map_unlock(map);
208                         return (error);
209                 }
210         }
211
212         if (bss_size) {
213                 error = vm_map_insert(map, NULL, 0,
214                         data_end, data_end + bss_size,
215                         VM_PROT_ALL, VM_PROT_ALL, 0);
216                 if (error) {
217                         vm_map_unlock(map);
218                         return (error);
219                 }
220         }
221         vm_map_unlock(map);
222
223         /* Fill in process VM information */
224         vmspace->vm_tsize = a_out->a_text >> PAGE_SHIFT;
225         vmspace->vm_dsize = (a_out->a_data + bss_size) >> PAGE_SHIFT;
226         vmspace->vm_taddr = (caddr_t) (uintptr_t) virtual_offset;
227         vmspace->vm_daddr = (caddr_t) (uintptr_t)
228                             (virtual_offset + a_out->a_text);
229
230         /* Fill in image_params */
231         imgp->interpreted = 0;
232         imgp->entry_addr = a_out->a_entry;
233
234         imgp->proc->p_sysent = &aout_sysvec;
235
236         /* Indicate that this file should not be modified */
237         imgp->vp->v_flag |= VTEXT;
238
239         return (0);
240 }
241
242 /*
243  * Dump core, into a file named as described in the comments for
244  * expand_name(), unless the process was setuid/setgid.
245  */
246 int
247 aout_coredump(p, vp, limit)
248         register struct proc *p;
249         register struct vnode *vp;
250         off_t limit;
251 {
252         register struct ucred *cred = p->p_ucred;
253         register struct vmspace *vm = p->p_vmspace;
254         int error;
255
256         if (ctob(UPAGES + vm->vm_dsize + vm->vm_ssize) >= limit)
257                 return (EFAULT);
258         bcopy(p, &p->p_addr->u_kproc.kp_proc, sizeof(struct proc));
259         fill_eproc(p, &p->p_addr->u_kproc.kp_eproc);
260         error = cpu_coredump(p, vp, cred);
261         if (error == 0)
262                 error = vn_rdwr_inchunks(UIO_WRITE, vp, vm->vm_daddr,
263                     (int)ctob(vm->vm_dsize), (off_t)ctob(UPAGES), UIO_USERSPACE,
264                     IO_UNIT | IO_DIRECT, cred, (int *) NULL, p);
265         if (error == 0)
266                 error = vn_rdwr_inchunks(UIO_WRITE, vp,
267                     (caddr_t) trunc_page(USRSTACK - ctob(vm->vm_ssize)),
268                     round_page(ctob(vm->vm_ssize)),
269                     (off_t)ctob(UPAGES) + ctob(vm->vm_dsize), UIO_USERSPACE,
270                     IO_UNIT | IO_DIRECT, cred, (int *) NULL, p);
271         return (error);
272 }
273
274 /*
275  * Tell kern_execve.c about it, with a little help from the linker.
276  */
277 static struct execsw aout_execsw = { exec_aout_imgact, "a.out" };
278 EXEC_SET(aout, aout_execsw);