usr.bin/procstat/procstat_rlimit.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
   3  *
   4  * Copyright (c) 2011 Mikolaj Golub
   5  * Copyright (c) 2015 Allan Jude <allanjude@freebsd.org>
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  */
  29
  30 #include <sys/cdefs.h>
  31 __FBSDID("$FreeBSD$");
  32
  33 #include <sys/param.h>
  34 #include <sys/time.h>
  35 #include <sys/resource.h>
  36 #include <sys/sysctl.h>
  37 #include <sys/user.h>
  38
  39 #include <err.h>
  40 #include <errno.h>
  41 #include <libprocstat.h>
  42 #include <libutil.h>
  43 #include <limits.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <string.h>
  47
  48 #include "procstat.h"
  49
  50 static struct {
  51         const char *name;
  52         const char *suffix;
  53 } rlimit_param[15] = {
  54         {"cputime",          "sec"},
  55         {"filesize",         "B  "},
  56         {"datasize",         "B  "},
  57         {"stacksize",        "B  "},
  58         {"coredumpsize",     "B  "},
  59         {"memoryuse",        "B  "},
  60         {"memorylocked",     "B  "},
  61         {"maxprocesses",     "   "},
  62         {"openfiles",        "   "},
  63         {"sbsize",           "B  "},
  64         {"vmemoryuse",       "B  "},
  65         {"pseudo-terminals", "   "},
  66         {"swapuse",          "B  "},
  67         {"kqueues",          "   "},
  68         {"umtxp",            "   "},
  69 };
  70
  71 #if RLIM_NLIMITS > 15
  72 #error "Resource limits have grown. Add new entries to rlimit_param[]."
  73 #endif
  74
  75 static const char *
  76 humanize_rlimit(int indx, rlim_t limit)
  77 {
  78         static char buf[14];
  79         int scale;
  80
  81         if (limit == RLIM_INFINITY)
  82                 return ("infinity     ");
  83
  84         scale = humanize_number(buf, sizeof(buf) - 1, (int64_t)limit,
  85             rlimit_param[indx].suffix, HN_AUTOSCALE | HN_GETSCALE, HN_DECIMAL);
  86         (void)humanize_number(buf, sizeof(buf) - 1, (int64_t)limit,
  87             rlimit_param[indx].suffix, HN_AUTOSCALE, HN_DECIMAL);
  88         /* Pad with one space if there is no suffix prefix. */
  89         if (scale == 0)
  90                 sprintf(buf + strlen(buf), " ");
  91         return (buf);
  92 }
  93
  94 void
  95 procstat_rlimit(struct procstat *prstat, struct kinfo_proc *kipp)
  96 {
  97         struct rlimit rlimit;
  98         int i;
  99
 100         if ((procstat_opts & PS_OPT_NOHEADER) == 0) {
 101                 xo_emit("{T:/%5s %-16s %-16s %16s %16s}\n",
 102                     "PID", "COMM", "RLIMIT", "SOFT     ", "HARD     ");
 103         }
 104         xo_emit("{ek:process_id/%5d}{e:command/%-16s/%s}", kipp->ki_pid,
 105             kipp->ki_comm);
 106         for (i = 0; i < RLIM_NLIMITS; i++) {
 107                 if (procstat_getrlimit(prstat, kipp, i, &rlimit) == -1)
 108                         return;
 109                 xo_emit("{dk:process_id/%5d} {d:command/%-16s} "
 110                     "{d:rlimit_param/%-16s} ", kipp->ki_pid, kipp->ki_comm,
 111                     rlimit_param[i].name);
 112
 113                 xo_open_container(rlimit_param[i].name);
 114                 if (rlimit.rlim_cur == RLIM_INFINITY)
 115                         xo_emit("{e:soft_limit/infinity}");
 116                 else
 117                         xo_emit("{e:soft_limit/%U}", rlimit.rlim_cur);
 118
 119                 if (rlimit.rlim_max == RLIM_INFINITY)
 120                         xo_emit("{e:hard_limit/infinity}");
 121                 else
 122                         xo_emit("{e:hard_limit/%U}", rlimit.rlim_max);
 123                 xo_close_container(rlimit_param[i].name);
 124
 125                 xo_emit("{d:rlim_cur/%16s} ",
 126                     humanize_rlimit(i, rlimit.rlim_cur));
 127                 xo_emit("{d:rlim_max/%16s}\n",
 128                     humanize_rlimit(i, rlimit.rlim_max));
 129         }
 130 }