summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/cgtop/cgtop.c25
1 files changed, 20 insertions, 5 deletions
diff --git a/src/cgtop/cgtop.c b/src/cgtop/cgtop.c
index f0ca83fd45..c439d09fd6 100644
--- a/src/cgtop/cgtop.c
+++ b/src/cgtop/cgtop.c
@@ -55,6 +55,7 @@ typedef struct Group {
} Group;
static unsigned arg_depth = 3;
+static unsigned arg_iterations = 0;
static usec_t arg_delay = 1*USEC_PER_SEC;
static enum {
@@ -505,6 +506,7 @@ static void help(void) {
" -m Order by memory load\n"
" -i Order by IO load\n"
" -d --delay=DELAY Specify delay\n"
+ " -n --iterations=N Run for N iterations before exiting\n"
" --depth=DEPTH Maximum traversal depth (default: 2)\n",
program_invocation_short_name);
}
@@ -516,10 +518,11 @@ static int parse_argv(int argc, char *argv[]) {
};
static const struct option options[] = {
- { "help", no_argument, NULL, 'h' },
- { "delay", required_argument, NULL, 'd' },
- { "depth", required_argument, NULL, ARG_DEPTH },
- { NULL, 0, NULL, 0 }
+ { "help", no_argument, NULL, 'h' },
+ { "delay", required_argument, NULL, 'd' },
+ { "iterations", required_argument, NULL, 'n' },
+ { "depth", required_argument, NULL, ARG_DEPTH },
+ { NULL, 0, NULL, 0 }
};
int c;
@@ -528,7 +531,7 @@ static int parse_argv(int argc, char *argv[]) {
assert(argc >= 1);
assert(argv);
- while ((c = getopt_long(argc, argv, "hptcmid:", options, NULL)) >= 0) {
+ while ((c = getopt_long(argc, argv, "hptcmin:d:", options, NULL)) >= 0) {
switch (c) {
@@ -554,6 +557,15 @@ static int parse_argv(int argc, char *argv[]) {
break;
+ case 'n':
+ r = safe_atou(optarg, &arg_iterations);
+ if (r < 0) {
+ log_error("Failed to parse iterations parameter.");
+ return -EINVAL;
+ }
+
+ break;
+
case 'p':
arg_order = ORDER_PATH;
break;
@@ -640,6 +652,9 @@ int main(int argc, char *argv[]) {
if (r < 0)
goto finish;
+ if (arg_iterations && iteration >= arg_iterations)
+ break;
+
r = read_one_char(stdin, &key, last_refresh + arg_delay - t, NULL);
if (r == -ETIMEDOUT)
continue;