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