sys/vm/vm_unix.c

   1 /*
   2  * (MPSAFE)
   3  *
   4  * Copyright (c) 1988 University of Utah.
   5  * Copyright (c) 1991, 1993
   6  *      The Regents of the University of California.  All rights reserved.
   7  *
   8  * This code is derived from software contributed to Berkeley by
   9  * the Systems Programming Group of the University of Utah Computer
  10  * Science Department.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer.
  17  * 2. Redistributions in binary form must reproduce the above copyright
  18  *    notice, this list of conditions and the following disclaimer in the
  19  *    documentation and/or other materials provided with the distribution.
  20  * 3. Neither the name of the University nor the names of its contributors
  21  *    may be used to endorse or promote products derived from this software
  22  *    without specific prior written permission.
  23  *
  24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  34  * SUCH DAMAGE.
  35  *
  36  * from: Utah $Hdr: vm_unix.c 1.1 89/11/07$
  37  *
  38  *      @(#)vm_unix.c   8.1 (Berkeley) 6/11/93
  39  * $FreeBSD: src/sys/vm/vm_unix.c,v 1.24.2.2 2002/07/02 20:06:19 dillon Exp $
  40  * $DragonFly: src/sys/vm/vm_unix.c,v 1.7 2006/11/07 17:51:24 dillon Exp $
  41  */
  42
  43 /*
  44  * Traditional sbrk/grow interface to VM
  45  */
  46 #include <sys/param.h>
  47 #include <sys/sysmsg.h>
  48 #include <sys/proc.h>
  49 #include <sys/resourcevar.h>
  50
  51 #include <vm/vm.h>
  52 #include <vm/vm_param.h>
  53 #include <sys/lock.h>
  54 #include <vm/pmap.h>
  55 #include <vm/vm_map.h>
  56
  57 #include <sys/thread.h>
  58
  59 /*
  60  * sys_sbrk backs the C library sbrk call
  61  *
  62  * void *sbrk(intptr_t incr)
  63  *
  64  * No requirements.
  65  */
  66 int
  67 sys_sbrk(struct sysmsg *sysmsg, const struct sbrk_args *uap)
  68 {
  69         struct proc *p = curproc;
  70         struct vmspace *vm = p->p_vmspace;
  71         vm_offset_t nbase;
  72         vm_offset_t base;
  73         vm_offset_t base_end;
  74         vm_offset_t incr;
  75         int rv;
  76         int error;
  77
  78         error = 0;
  79
  80         lwkt_gettoken(&vm->vm_map.token);
  81
  82         /*
  83          * Cannot assume that last data page for binary is mapped R/W.
  84          */
  85         base = round_page((vm_offset_t)vm->vm_daddr) + vm->vm_dsize;
  86         incr = uap->incr;
  87
  88         /*
  89          * Cannot allow space to be freed with sbrk() because it is not
  90          * thread-safe for userland and because unrelated mmap()s can reside
  91          * in the data space if the DATA rlimit is raised on the running
  92          * program.
  93          */
  94         if (incr < 0) {
  95                 error = EOPNOTSUPP;
  96                 goto done;
  97         }
  98
  99         /*
 100          * Userland requests current base
 101          */
 102         if (incr == 0) {
 103                 sysmsg->sysmsg_resultp = (void *)base;
 104                 goto done;
 105         }
 106
 107         /*
 108          * Calculate approximate area (vm_map_find() may change this).
 109          * Check for overflow, address space, and rlimit caps.
 110          */
 111         base_end = base + incr;
 112         if (base_end >= VM_MAX_USER_ADDRESS) {
 113                 error = ENOMEM;
 114                 goto done;
 115         }
 116         if (base_end < base ||
 117             base_end - (vm_offset_t)vm->vm_daddr >
 118             (vm_offset_t)p->p_rlimit[RLIMIT_DATA].rlim_cur) {
 119                 error = ENOMEM;
 120                 goto done;
 121         }
 122
 123         /*
 124          * Same-page optimization (protected by token)
 125          */
 126         if ((base & PAGE_MASK) != 0 &&
 127             ((base ^ (base_end - 1)) & ~(vm_offset_t)PAGE_MASK) == 0) {
 128                 sysmsg->sysmsg_resultp = (void *)base;
 129                 vm->vm_dsize += incr;
 130                 goto done;
 131         }
 132
 133         /*
 134          * Formally map more space
 135          */
 136         nbase = round_page(base);
 137         rv = vm_map_find(&vm->vm_map, NULL, NULL,
 138                          0, &nbase, round_page(incr),
 139                          PAGE_SIZE, FALSE,
 140                          VM_MAPTYPE_NORMAL, VM_SUBSYS_BRK,
 141                          VM_PROT_ALL, VM_PROT_ALL, 0);
 142         if (rv != KERN_SUCCESS) {
 143                 error = ENOMEM;
 144                 goto done;
 145         }
 146         base_end = nbase + round_page(incr);
 147         sysmsg->sysmsg_resultp = (void *)nbase;
 148
 149         /* wire the pages */
 150         if (vm->vm_map.flags & MAP_WIREFUTURE)
 151                 vm_map_kernel_wiring(&vm->vm_map, base, base_end, 0);
 152
 153         /*
 154          * Adjust dsize upwards only
 155          */
 156         incr = nbase + incr - round_page((vm_offset_t)vm->vm_daddr);
 157         if (vm->vm_dsize < incr)
 158                 vm->vm_dsize = incr;
 159
 160 done:
 161         lwkt_reltoken(&vm->vm_map.token);
 162
 163         return (error);
 164 }
 165
 166 /*
 167  * sys_obreak is used by the sbrk emulation code in libc when sbrk()
 168  * is not supported.
 169  *
 170  * obreak_args(char *nsize)
 171  *
 172  * No requirements.
 173  */
 174 int
 175 sys_obreak(struct sysmsg *sysmsg, const struct obreak_args *uap)
 176 {
 177         struct proc *p = curproc;
 178         struct vmspace *vm = p->p_vmspace;
 179         vm_offset_t new, old, base;
 180         int rv;
 181         int error;
 182
 183         error = 0;
 184
 185         lwkt_gettoken(&vm->vm_map.token);
 186
 187         base = round_page((vm_offset_t)vm->vm_daddr);
 188         new = round_page((vm_offset_t)uap->nsize);
 189         old = base + round_page(vm->vm_dsize);
 190
 191         if (new > base) {
 192                 /*
 193                  * We check resource limits here, but alow processes to
 194                  * reduce their usage, even if they remain over the limit.
 195                  */
 196                 if (new > old &&
 197                     (new - base) > (vm_offset_t) p->p_rlimit[RLIMIT_DATA].rlim_cur) {
 198                         error = ENOMEM;
 199                         goto done;
 200                 }
 201                 if (new >= VM_MAX_USER_ADDRESS) {
 202                         error = ENOMEM;
 203                         goto done;
 204                 }
 205         } else if (new < base) {
 206                 /*
 207                  * This is simply an invalid value.  If someone wants to
 208                  * do fancy address space manipulations, mmap and munmap
 209                  * can do most of what the user would want.
 210                  */
 211                 error = EINVAL;
 212                 goto done;
 213         }
 214
 215         if (new > old) {
 216                 vm_size_t diff;
 217
 218                 diff = new - old;
 219                 if (vm->vm_map.size + diff > p->p_rlimit[RLIMIT_VMEM].rlim_cur) {
 220                         error = ENOMEM;
 221                         goto done;
 222                 }
 223                 rv = vm_map_find(&vm->vm_map, NULL, NULL,
 224                                  0, &old, diff,
 225                                  PAGE_SIZE, FALSE,
 226                                  VM_MAPTYPE_NORMAL, VM_SUBSYS_BRK,
 227                                  VM_PROT_ALL, VM_PROT_ALL, 0);
 228                 if (rv != KERN_SUCCESS) {
 229                         error = ENOMEM;
 230                         goto done;
 231                 }
 232
 233                 /* wire the pages */
 234                 if (vm->vm_map.flags & MAP_WIREFUTURE)
 235                         vm_map_kernel_wiring(&vm->vm_map, old, new, 0);
 236
 237                 vm->vm_dsize += diff;
 238         } else if (new < old) {
 239                 error = EOPNOTSUPP;
 240 #if 0
 241                 rv = vm_map_remove(&vm->vm_map, new, old);
 242                 if (rv != KERN_SUCCESS) {
 243                         error = ENOMEM;
 244                         goto done;
 245                 }
 246                 vm->vm_dsize -= old - new;
 247 #endif
 248         }
 249 done:
 250         lwkt_reltoken(&vm->vm_map.token);
 251
 252         return (error);
 253 }