sysdeps/unix/bsd/ulimit.c

   1 /* Copyright (C) 1991, 92, 94, 96, 97, 98, 2001 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, write to the Free
  16    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  17    02111-1307 USA.  */
  18
  19 #include <errno.h>
  20 #include <stdarg.h>
  21 #include <sysdep.h>
  22 #include <ulimit.h>
  23 #include <unistd.h>
  24 #include <sys/resource.h>
  25
  26 #ifndef  HAVE_GNU_LD
  27 #define  _etext etext
  28 #endif
  29
  30 extern int _etext;
  31
  32 /* Function depends on CMD:
  33    1 = Return the limit on the size of a file, in units of 512 bytes.
  34    2 = Set the limit on the size of a file to NEWLIMIT.  Only the
  35        super-user can increase the limit.
  36    3 = Return the maximum possible address of the data segment.
  37    4 = Return the maximum number of files that the calling process
  38        can open.
  39    Returns -1 on errors.  */
  40 long int
  41 ulimit (int cmd, ...)
  42 {
  43   struct rlimit limit;
  44   va_list va;
  45   long int result = -1;
  46
  47   va_start (va, cmd);
  48
  49   switch (cmd)
  50     {
  51     case UL_GETFSIZE:
  52       /* Get limit on file size.  */
  53       if (getrlimit (RLIMIT_FSIZE, &limit) == 0)
  54         /* Convert from bytes to 512 byte units.  */
  55         result = limit.rlim_cur / 512;
  56       break;
  57
  58     case UL_SETFSIZE:
  59       /* Set limit on file size.  */
  60       {
  61         long int newlimit = va_arg (va, long int);
  62
  63         if ((rlim_t) newlimit > RLIM_INFINITY / 512)
  64           {
  65             limit.rlim_cur = RLIM_INFINITY;
  66             limit.rlim_max = RLIM_INFINITY;
  67           }
  68         else
  69           {
  70             limit.rlim_cur = newlimit * 512;
  71             limit.rlim_max = newlimit * 512;
  72           }
  73
  74         result = setrlimit (RLIMIT_FSIZE, &limit);
  75       }
  76       break;
  77
  78     case __UL_GETMAXBRK:
  79       /* Get maximum address for `brk'.  */
  80       if (getrlimit (RLIMIT_DATA, &limit) == 0)
  81         result = ((long int) &_etext) + limit.rlim_cur;
  82       break;
  83
  84     case __UL_GETOPENMAX:
  85       result = sysconf (_SC_OPEN_MAX);
  86       break;
  87
  88     default:
  89       __set_errno (EINVAL);
  90     }
  91
  92   va_end (va);
  93
  94   return result;
  95 }