usr.bin/procstat/procstat_rlimit.c

   1 /*-
   2  * SPDX-License-Identifier: BSD-2-Clause
   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 #include <sys/param.h>
  32 #include <sys/time.h>
  33 #include <sys/resource.h>
  34 #include <sys/sysctl.h>
  35 #include <sys/user.h>
  36
  37 #include <err.h>
  38 #include <errno.h>
  39 #include <libprocstat.h>
  40 #include <libutil.h>
  41 #include <limits.h>
  42 #include <stdio.h>
  43 #include <stdlib.h>
  44 #include <string.h>
  45
  46 #include "procstat.h"
  47
  48 static struct {
  49         const char *name;
  50         const char *suffix;
  51 } rlimit_param[15] = {
  52         {"cputime",          "sec"},
  53         {"filesize",         "B  "},
  54         {"datasize",         "B  "},
  55         {"stacksize",        "B  "},
  56         {"coredumpsize",     "B  "},
  57         {"memoryuse",        "B  "},
  58         {"memorylocked",     "B  "},
  59         {"maxprocesses",     "   "},
  60         {"openfiles",        "   "},
  61         {"sbsize",           "B  "},
  62         {"vmemoryuse",       "B  "},
  63         {"pseudo-terminals", "   "},
  64         {"swapuse",          "B  "},
  65         {"kqueues",          "   "},
  66         {"umtxp",            "   "},
  67 };
  68
  69 #if RLIM_NLIMITS > 15
  70 #error "Resource limits have grown. Add new entries to rlimit_param[]."
  71 #endif
  72
  73 static const char *
  74 humanize_rlimit(int indx, rlim_t limit)
  75 {
  76         static char buf[14];
  77         int scale;
  78
  79         if (limit == RLIM_INFINITY)
  80                 return ("infinity     ");
  81
  82         scale = humanize_number(buf, sizeof(buf) - 1, (int64_t)limit,
  83             rlimit_param[indx].suffix, HN_AUTOSCALE | HN_GETSCALE, HN_DECIMAL);
  84         (void)humanize_number(buf, sizeof(buf) - 1, (int64_t)limit,
  85             rlimit_param[indx].suffix, HN_AUTOSCALE, HN_DECIMAL);
  86         /* Pad with one space if there is no suffix prefix. */
  87         if (scale == 0)
  88                 sprintf(buf + strlen(buf), " ");
  89         return (buf);
  90 }
  91
  92 void
  93 procstat_rlimit(struct procstat *prstat, struct kinfo_proc *kipp)
  94 {
  95         struct rlimit rlimit;
  96         int i;
  97
  98         if ((procstat_opts & PS_OPT_NOHEADER) == 0) {
  99                 xo_emit("{T:/%5s %-16s %-16s %16s %16s}\n",
 100                     "PID", "COMM", "RLIMIT", "SOFT     ", "HARD     ");
 101         }
 102         xo_emit("{ek:process_id/%5d}{e:command/%-16s/%s}", kipp->ki_pid,
 103             kipp->ki_comm);
 104         for (i = 0; i < RLIM_NLIMITS; i++) {
 105                 if (procstat_getrlimit(prstat, kipp, i, &rlimit) == -1)
 106                         return;
 107                 xo_emit("{dk:process_id/%5d} {d:command/%-16s} "
 108                     "{d:rlimit_param/%-16s} ", kipp->ki_pid, kipp->ki_comm,
 109                     rlimit_param[i].name);
 110
 111                 xo_open_container(rlimit_param[i].name);
 112                 if (rlimit.rlim_cur == RLIM_INFINITY)
 113                         xo_emit("{e:soft_limit/infinity}");
 114                 else
 115                         xo_emit("{e:soft_limit/%U}", rlimit.rlim_cur);
 116
 117                 if (rlimit.rlim_max == RLIM_INFINITY)
 118                         xo_emit("{e:hard_limit/infinity}");
 119                 else
 120                         xo_emit("{e:hard_limit/%U}", rlimit.rlim_max);
 121                 xo_close_container(rlimit_param[i].name);
 122
 123                 xo_emit("{d:rlim_cur/%16s} ",
 124                     humanize_rlimit(i, rlimit.rlim_cur));
 125                 xo_emit("{d:rlim_max/%16s}\n",
 126                     humanize_rlimit(i, rlimit.rlim_max));
 127         }
 128 }