memcheck/tests/freebsd/getrlimitusage.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause
   3  *
   4  * Copyright (c) 2024 The FreeBSD Foundation
   5  *
   6  * This software was developed by Konstantin Belousov <kib@FreeBSD.org>
   7  * under sponsorship from the FreeBSD Foundation.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  *
  18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  21  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  28  * SUCH DAMAGE.
  29  */
  30
  31 #include <sys/param.h>
  32 #define _RLIMIT_IDENT
  33 #include <err.h>
  34 #include <errno.h>
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <sys/resource.h>
  38 #include <unistd.h>
  39
  40 static void usage(void)
  41 {
  42     fprintf(stderr, "usage: getrlimitusage [-e] [-p pid]\n");
  43     exit(2);
  44 }
  45
  46 int main(int argc, char *argv[])
  47 {
  48     rlim_t res;
  49     int c, flags;
  50
  51     flags = 0;
  52     while ((c = getopt(argc, argv, "e")) != -1)
  53     {
  54         switch (c)
  55         {
  56         case 'e':
  57             flags |= GETRLIMITUSAGE_EUID;
  58             break;
  59         case '?':
  60         default:
  61             usage();
  62         }
  63     }
  64     argc -= optind;
  65     argv += optind;
  66     if (argc != 0)
  67         usage();
  68
  69     for (unsigned i = 0;; i++)
  70     {
  71         if (getrlimitusage(i, flags, &res) == -1)
  72         {
  73             if (errno == ENXIO)
  74             {
  75                 res = -1;
  76             }
  77             else
  78             {
  79                 if (errno != EINVAL)
  80                     err(1, "getrlimitusage(%d)", errno);
  81                 break;
  82             }
  83         }
  84         // add some rounding to try to make regtest stable
  85         switch (i)
  86         {
  87         case 7:
  88         case 9:
  89         case 12:
  90         case 13:
  91         case 14:
  92         case 15:
  93             res = 0U;
  94             break;
  95         case 5:
  96             res = res/10000000 * 10000000;
  97             break;
  98         case 10:
  99             res = res/100000000 * 100000000;
 100             break;
 101         default:
 102              break;
 103         }
 104         fprintf(stderr, "%s (%d):\t%jd\n", i < nitems(rlimit_ident) ? rlimit_ident[i] : "unknown", i, (uintmax_t)res
 105 );
 106     }
 107     exit(0);
 108 }