2 * prlimit - get/set process resource limits.
4 * Copyright (C) 2011 Davidlohr Bueso <dave@gnu.org>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
28 #include <sys/resource.h>
30 #include <libsmartcols.h>
37 #include "closestream.h"
40 # define RLIMIT_RTTIME 15
62 /* basic output flags */
63 static int no_headings
;
73 static struct prlimit_desc prlimit_desc
[] =
75 [AS
] = { "AS", N_("address space limit"), N_("bytes"), RLIMIT_AS
},
76 [CORE
] = { "CORE", N_("max core file size"), N_("bytes"), RLIMIT_CORE
},
77 [CPU
] = { "CPU", N_("CPU time"), N_("seconds"), RLIMIT_CPU
},
78 [DATA
] = { "DATA", N_("max data size"), N_("bytes"), RLIMIT_DATA
},
79 [FSIZE
] = { "FSIZE", N_("max file size"), N_("bytes"), RLIMIT_FSIZE
},
80 [LOCKS
] = { "LOCKS", N_("max number of file locks held"), N_("locks"), RLIMIT_LOCKS
},
81 [MEMLOCK
] = { "MEMLOCK", N_("max locked-in-memory address space"), N_("bytes"), RLIMIT_MEMLOCK
},
82 [MSGQUEUE
] = { "MSGQUEUE", N_("max bytes in POSIX mqueues"), N_("bytes"), RLIMIT_MSGQUEUE
},
83 [NICE
] = { "NICE", N_("max nice prio allowed to raise"), NULL
, RLIMIT_NICE
},
84 [NOFILE
] = { "NOFILE", N_("max number of open files"), N_("files"), RLIMIT_NOFILE
},
85 [NPROC
] = { "NPROC", N_("max number of processes"), N_("processes"), RLIMIT_NPROC
},
86 [RSS
] = { "RSS", N_("max resident set size"), N_("bytes"), RLIMIT_RSS
},
87 [RTPRIO
] = { "RTPRIO", N_("max real-time priority"), NULL
, RLIMIT_RTPRIO
},
88 [RTTIME
] = { "RTTIME", N_("timeout for real-time tasks"), N_("microsecs"), RLIMIT_RTTIME
},
89 [SIGPENDING
] = { "SIGPENDING", N_("max number of pending signals"), N_("signals"), RLIMIT_SIGPENDING
},
90 [STACK
] = { "STACK", N_("max stack size"), N_("bytes"), RLIMIT_STACK
}
93 #define MAX_RESOURCES ARRAY_SIZE(prlimit_desc)
96 struct list_head lims
;
99 struct prlimit_desc
*desc
;
100 int modify
; /* PRLIMIT_{SOFT,HARD} mask */
103 #define PRLIMIT_EMPTY_LIMIT {{ 0, 0, }, NULL, 0 }
115 const char *name
; /* header */
116 double whint
; /* width hint (N < 1 is in percent of termwidth) */
117 int flags
; /* SCOLS_FL_* */
121 /* columns descriptions */
122 static struct colinfo infos
[] = {
123 [COL_RES
] = { "RESOURCE", 0.25, SCOLS_FL_TRUNC
, N_("resource name") },
124 [COL_HELP
] = { "DESCRIPTION", 0.1, SCOLS_FL_TRUNC
, N_("resource description")},
125 [COL_SOFT
] = { "SOFT", 0.1, SCOLS_FL_RIGHT
, N_("soft limit")},
126 [COL_HARD
] = { "HARD", 1, SCOLS_FL_RIGHT
, N_("hard limit (ceiling)")},
127 [COL_UNITS
] = { "UNITS", 0.1, SCOLS_FL_TRUNC
, N_("units")},
130 static int columns
[ARRAY_SIZE(infos
) * 2];
135 #define INFINITY_STR "unlimited"
136 #define INFINITY_STRLEN (sizeof(INFINITY_STR) - 1)
138 #define PRLIMIT_SOFT (1 << 1)
139 #define PRLIMIT_HARD (1 << 2)
141 static pid_t pid
; /* calling process (default) */
145 # include <sys/syscall.h>
146 static int prlimit(pid_t p
, int resource
,
147 const struct rlimit
*new_limit
,
148 struct rlimit
*old_limit
)
150 return syscall(SYS_prlimit64
, p
, resource
, new_limit
, old_limit
);
154 static void __attribute__((__noreturn__
)) usage(void)
159 fputs(USAGE_HEADER
, out
);
162 _(" %s [options] [-p PID]\n"), program_invocation_short_name
);
164 _(" %s [options] COMMAND\n"), program_invocation_short_name
);
166 fputs(USAGE_SEPARATOR
, out
);
167 fputs(_("Show or change the resource limits of a process.\n"), out
);
169 fputs(_("\nGeneral Options:\n"), out
);
170 fputs(_(" -p, --pid <pid> process id\n"
171 " -o, --output <list> define which output columns to use\n"
172 " --noheadings don't print headings\n"
173 " --raw use the raw output format\n"
174 " --verbose verbose output\n"
176 printf(USAGE_HELP_OPTIONS(24));
178 fputs(_("\nResources Options:\n"), out
);
179 fputs(_(" -c, --core maximum size of core files created\n"
180 " -d, --data maximum size of a process's data segment\n"
181 " -e, --nice maximum nice priority allowed to raise\n"
182 " -f, --fsize maximum size of files written by the process\n"
183 " -i, --sigpending maximum number of pending signals\n"
184 " -l, --memlock maximum size a process may lock into memory\n"
185 " -m, --rss maximum resident set size\n"
186 " -n, --nofile maximum number of open files\n"
187 " -q, --msgqueue maximum bytes in POSIX message queues\n"
188 " -r, --rtprio maximum real-time scheduling priority\n"
189 " -s, --stack maximum stack size\n"
190 " -t, --cpu maximum amount of CPU time in seconds\n"
191 " -u, --nproc maximum number of user processes\n"
192 " -v, --as size of virtual memory\n"
193 " -x, --locks maximum number of file locks\n"
194 " -y, --rttime CPU time in microseconds a process scheduled\n"
195 " under real-time scheduling\n"), out
);
197 fputs(USAGE_COLUMNS
, out
);
198 for (i
= 0; i
< ARRAY_SIZE(infos
); i
++)
199 fprintf(out
, " %11s %s\n", infos
[i
].name
, _(infos
[i
].help
));
201 printf(USAGE_MAN_TAIL("prlimit(1)"));
206 static inline int get_column_id(int num
)
208 assert(num
< ncolumns
);
209 assert(columns
[num
] < (int) ARRAY_SIZE(infos
));
214 static inline struct colinfo
*get_column_info(unsigned num
)
216 return &infos
[ get_column_id(num
) ];
219 static void add_scols_line(struct libscols_table
*table
, struct prlimit
*l
)
222 struct libscols_line
*line
;
227 line
= scols_table_new_line(table
, NULL
);
229 err(EXIT_FAILURE
, _("failed to allocate output line"));
231 for (i
= 0; i
< ncolumns
; i
++) {
234 switch (get_column_id(i
)) {
236 str
= xstrdup(l
->desc
->name
);
239 str
= xstrdup(l
->desc
->help
);
242 if (l
->rlim
.rlim_cur
== RLIM_INFINITY
)
243 str
= xstrdup(_("unlimited"));
245 xasprintf(&str
, "%llu", (unsigned long long) l
->rlim
.rlim_cur
);
248 if (l
->rlim
.rlim_max
== RLIM_INFINITY
)
249 str
= xstrdup(_("unlimited"));
251 xasprintf(&str
, "%llu", (unsigned long long) l
->rlim
.rlim_max
);
254 str
= l
->desc
->unit
? xstrdup(_(l
->desc
->unit
)) : NULL
;
260 if (str
&& scols_line_refer_data(line
, i
, str
))
261 err(EXIT_FAILURE
, _("failed to add output data"));
265 static int column_name_to_id(const char *name
, size_t namesz
)
271 for (i
= 0; i
< ARRAY_SIZE(infos
); i
++) {
272 const char *cn
= infos
[i
].name
;
274 if (!strncasecmp(name
, cn
, namesz
) && !*(cn
+ namesz
))
277 warnx(_("unknown column: %s"), name
);
281 static void rem_prlim(struct prlimit
*lim
)
285 list_del(&lim
->lims
);
289 static int show_limits(struct list_head
*lims
)
292 struct list_head
*p
, *pnext
;
293 struct libscols_table
*table
;
295 table
= scols_new_table();
297 err(EXIT_FAILURE
, _("failed to allocate output table"));
299 scols_table_enable_raw(table
, raw
);
300 scols_table_enable_noheadings(table
, no_headings
);
302 for (i
= 0; i
< ncolumns
; i
++) {
303 struct colinfo
*col
= get_column_info(i
);
305 if (!scols_table_new_column(table
, col
->name
, col
->whint
, col
->flags
))
306 err(EXIT_FAILURE
, _("failed to allocate output column"));
309 list_for_each_safe(p
, pnext
, lims
) {
310 struct prlimit
*lim
= list_entry(p
, struct prlimit
, lims
);
312 add_scols_line(table
, lim
);
316 scols_print_table(table
);
317 scols_unref_table(table
);
322 * If one of the limits is unknown (default value for not being passed), we
323 * need to get the current limit and use it. I see no other way other than
326 static void get_unknown_hardsoft(struct prlimit
*lim
)
330 if (prlimit(pid
, lim
->desc
->resource
, NULL
, &old
) == -1)
331 err(EXIT_FAILURE
, _("failed to get old %s limit"),
334 if (!(lim
->modify
& PRLIMIT_SOFT
))
335 lim
->rlim
.rlim_cur
= old
.rlim_cur
;
336 else if (!(lim
->modify
& PRLIMIT_HARD
))
337 lim
->rlim
.rlim_max
= old
.rlim_max
;
340 static void do_prlimit(struct list_head
*lims
)
342 struct list_head
*p
, *pnext
;
344 list_for_each_safe(p
, pnext
, lims
) {
345 struct rlimit
*new = NULL
, *old
= NULL
;
346 struct prlimit
*lim
= list_entry(p
, struct prlimit
, lims
);
349 if (lim
->modify
!= (PRLIMIT_HARD
| PRLIMIT_SOFT
))
350 get_unknown_hardsoft(lim
);
352 if ((lim
->rlim
.rlim_cur
> lim
->rlim
.rlim_max
) &&
353 (lim
->rlim
.rlim_cur
!= RLIM_INFINITY
||
354 lim
->rlim
.rlim_max
!= RLIM_INFINITY
))
355 errx(EXIT_FAILURE
, _("the soft limit %s cannot exceed the hard limit"),
361 if (verbose
&& new) {
362 printf(_("New %s limit for pid %d: "), lim
->desc
->name
,
363 pid
? pid
: getpid());
364 if (new->rlim_cur
== RLIM_INFINITY
)
365 printf("<%s", _("unlimited"));
367 printf("<%ju", (uintmax_t)new->rlim_cur
);
369 if (new->rlim_max
== RLIM_INFINITY
)
370 printf(":%s>\n", _("unlimited"));
372 printf(":%ju>\n", (uintmax_t)new->rlim_max
);
375 if (prlimit(pid
, lim
->desc
->resource
, new, old
) == -1)
376 err(EXIT_FAILURE
, lim
->modify
?
377 _("failed to set the %s resource limit") :
378 _("failed to get the %s resource limit"),
382 rem_prlim(lim
); /* modify only; don't show */
386 static int get_range(char *str
, rlim_t
*soft
, rlim_t
*hard
, int *found
)
394 *soft
= *hard
= RLIM_INFINITY
;
396 if (!strcmp(str
, INFINITY_STR
)) { /* <unlimited> */
397 *found
|= PRLIMIT_SOFT
| PRLIMIT_HARD
;
400 } else if (*str
== ':') { /* <:hard> */
403 if (strcmp(str
, INFINITY_STR
) != 0) {
404 *hard
= strtoull(str
, &end
, 10);
406 if (errno
|| !end
|| *end
|| end
== str
)
409 *found
|= PRLIMIT_HARD
;
414 if (strncmp(str
, INFINITY_STR
, INFINITY_STRLEN
) == 0) {
415 /* <unlimited> or <unlimited:> */
416 end
= str
+ INFINITY_STRLEN
;
418 /* <value> or <soft:> */
419 *hard
= *soft
= strtoull(str
, &end
, 10);
420 if (errno
|| !end
|| end
== str
)
424 if (*end
== ':' && !*(end
+ 1)) /* <soft:> */
425 *found
|= PRLIMIT_SOFT
;
427 else if (*end
== ':') { /* <soft:hard> */
430 if (!strcmp(str
, INFINITY_STR
))
431 *hard
= RLIM_INFINITY
;
435 *hard
= strtoull(str
, &end
, 10);
437 if (errno
|| !end
|| *end
|| end
== str
)
440 *found
|= PRLIMIT_SOFT
| PRLIMIT_HARD
;
443 *found
|= PRLIMIT_SOFT
| PRLIMIT_HARD
;
449 static int parse_prlim(struct rlimit
*lim
, char *ops
, size_t id
)
454 if (get_range(ops
, &soft
, &hard
, &found
))
455 errx(EXIT_FAILURE
, _("failed to parse %s limit"),
456 prlimit_desc
[id
].name
);
458 lim
->rlim_cur
= soft
;
459 lim
->rlim_max
= hard
;
464 static int add_prlim(char *ops
, struct list_head
*lims
, size_t id
)
466 struct prlimit
*lim
= xcalloc(1, sizeof(*lim
));
468 INIT_LIST_HEAD(&lim
->lims
);
469 lim
->desc
= &prlimit_desc
[id
];
472 lim
->modify
= parse_prlim(&lim
->rlim
, ops
, id
);
474 list_add_tail(&lim
->lims
, lims
);
478 int main(int argc
, char **argv
)
481 struct list_head lims
;
484 VERBOSE_OPTION
= CHAR_MAX
+ 1,
489 static const struct option longopts
[] = {
490 { "pid", required_argument
, NULL
, 'p' },
491 { "output", required_argument
, NULL
, 'o' },
492 { "as", optional_argument
, NULL
, 'v' },
493 { "core", optional_argument
, NULL
, 'c' },
494 { "cpu", optional_argument
, NULL
, 't' },
495 { "data", optional_argument
, NULL
, 'd' },
496 { "fsize", optional_argument
, NULL
, 'f' },
497 { "locks", optional_argument
, NULL
, 'x' },
498 { "memlock", optional_argument
, NULL
, 'l' },
499 { "msgqueue", optional_argument
, NULL
, 'q' },
500 { "nice", optional_argument
, NULL
, 'e' },
501 { "nofile", optional_argument
, NULL
, 'n' },
502 { "nproc", optional_argument
, NULL
, 'u' },
503 { "rss", optional_argument
, NULL
, 'm' },
504 { "rtprio", optional_argument
, NULL
, 'r' },
505 { "rttime", optional_argument
, NULL
, 'y' },
506 { "sigpending", optional_argument
, NULL
, 'i' },
507 { "stack", optional_argument
, NULL
, 's' },
508 { "version", no_argument
, NULL
, 'V' },
509 { "help", no_argument
, NULL
, 'h' },
510 { "noheadings", no_argument
, NULL
, NOHEADINGS_OPTION
},
511 { "raw", no_argument
, NULL
, RAW_OPTION
},
512 { "verbose", no_argument
, NULL
, VERBOSE_OPTION
},
516 setlocale(LC_ALL
, "");
517 bindtextdomain(PACKAGE
, LOCALEDIR
);
519 atexit(close_stdout
);
521 INIT_LIST_HEAD(&lims
);
524 * Something is very wrong if this doesn't succeed,
525 * assuming STACK is the last resource, of course.
527 assert(MAX_RESOURCES
== STACK
+ 1);
529 while((opt
= getopt_long(argc
, argv
,
530 "+c::d::e::f::i::l::m::n::q::r::s::t::u::v::x::y::p:o:vVh",
531 longopts
, NULL
)) != -1) {
534 add_prlim(optarg
, &lims
, CORE
);
537 add_prlim(optarg
, &lims
, DATA
);
540 add_prlim(optarg
, &lims
, NICE
);
543 add_prlim(optarg
, &lims
, FSIZE
);
546 add_prlim(optarg
, &lims
, SIGPENDING
);
549 add_prlim(optarg
, &lims
, MEMLOCK
);
552 add_prlim(optarg
, &lims
, RSS
);
555 add_prlim(optarg
, &lims
, NOFILE
);
558 add_prlim(optarg
, &lims
, MSGQUEUE
);
561 add_prlim(optarg
, &lims
, RTPRIO
);
564 add_prlim(optarg
, &lims
, STACK
);
567 add_prlim(optarg
, &lims
, CPU
);
570 add_prlim(optarg
, &lims
, NPROC
);
573 add_prlim(optarg
, &lims
, AS
);
576 add_prlim(optarg
, &lims
, LOCKS
);
579 add_prlim(optarg
, &lims
, RTTIME
);
584 errx(EXIT_FAILURE
, _("option --pid may be specified only once"));
585 pid
= strtos32_or_err(optarg
, _("invalid PID argument"));
590 ncolumns
= string_to_idarray(optarg
,
591 columns
, ARRAY_SIZE(columns
),
597 printf(UTIL_LINUX_VERSION
);
600 case NOHEADINGS_OPTION
:
610 errtryhelp(EXIT_FAILURE
);
613 if (argc
> optind
&& pid
)
614 errx(EXIT_FAILURE
, _("options --pid and COMMAND are mutually exclusive"));
616 /* default columns */
617 columns
[ncolumns
++] = COL_RES
;
618 columns
[ncolumns
++] = COL_HELP
;
619 columns
[ncolumns
++] = COL_SOFT
;
620 columns
[ncolumns
++] = COL_HARD
;
621 columns
[ncolumns
++] = COL_UNITS
;
626 if (list_empty(&lims
)) {
627 /* default is to print all resources */
630 for (n
= 0; n
< MAX_RESOURCES
; n
++)
631 add_prlim(NULL
, &lims
, n
);
636 if (!list_empty(&lims
))
640 /* prlimit [options] COMMAND */
641 execvp(argv
[optind
], &argv
[optind
]);
642 errexec(argv
[optind
]);