util-linux/chrt.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * chrt - manipulate real-time attributes of a process
   4  * Copyright (c) 2006-2007 Bernhard Reutner-Fischer
   5  *
   6  * Licensed under GPLv2 or later, see file LICENSE in this source tree.
   7  */
   8 //config:config CHRT
   9 //config:       bool "chrt (4.7 kb)"
  10 //config:       default y
  11 //config:       help
  12 //config:       Manipulate real-time attributes of a process.
  13 //config:       This requires sched_{g,s}etparam support in your libc.
  14
  15 //applet:IF_CHRT(APPLET_NOEXEC(chrt, chrt, BB_DIR_USR_BIN, BB_SUID_DROP, chrt))
  16
  17 //kbuild:lib-$(CONFIG_CHRT) += chrt.o
  18
  19 //usage:#define chrt_trivial_usage
  20 //usage:       "-m | -p [PRIO] PID | [-rfobi] PRIO PROG [ARGS]"
  21 //usage:#define chrt_full_usage "\n\n"
  22 //usage:       "Change scheduling priority and class for a process\n"
  23 //usage:     "\n        -m      Show min/max priorities"
  24 //usage:     "\n        -p      Operate on PID"
  25 //usage:     "\n        -r      Set SCHED_RR class"
  26 //usage:     "\n        -f      Set SCHED_FIFO class"
  27 //usage:     "\n        -o      Set SCHED_OTHER class"
  28 //usage:     "\n        -b      Set SCHED_BATCH class"
  29 //usage:     "\n        -i      Set SCHED_IDLE class"
  30 //usage:
  31 //usage:#define chrt_example_usage
  32 //usage:       "$ chrt -r 4 sleep 900; x=$!\n"
  33 //usage:       "$ chrt -f -p 3 $x\n"
  34 //usage:       "You need CAP_SYS_NICE privileges to set scheduling attributes of a process"
  35
  36 #include <sched.h>
  37 #include "libbb.h"
  38 #ifndef SCHED_IDLE
  39 # define SCHED_IDLE 5
  40 #endif
  41
  42 static const char *policy_name(int pol)
  43 {
  44         if (pol > 6)
  45                 return utoa(pol);
  46         return nth_string(
  47                 "OTHER"   "\0" /* 0:SCHED_OTHER */
  48                 "FIFO"    "\0" /* 1:SCHED_FIFO */
  49                 "RR"      "\0" /* 2:SCHED_RR */
  50                 "BATCH"   "\0" /* 3:SCHED_BATCH */
  51                 "ISO"     "\0" /* 4:SCHED_ISO */
  52                 "IDLE"    "\0" /* 5:SCHED_IDLE */
  53                 "DEADLINE",    /* 6:SCHED_DEADLINE */
  54                 pol
  55         );
  56 }
  57
  58 static void show_min_max(int pol)
  59 {
  60         const char *fmt = "SCHED_%s min/max priority\t: %u/%u\n";
  61         int max, min;
  62
  63         max = sched_get_priority_max(pol);
  64         min = sched_get_priority_min(pol);
  65         if ((max|min) < 0)
  66                 fmt = "SCHED_%s not supported\n";
  67         printf(fmt, policy_name(pol), min, max);
  68 }
  69
  70 #define OPT_m (1<<0)
  71 #define OPT_p (1<<1)
  72 #define OPT_r (1<<2)
  73 #define OPT_f (1<<3)
  74 #define OPT_o (1<<4)
  75 #define OPT_b (1<<5)
  76 #define OPT_i (1<<6)
  77
  78 int chrt_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  79 int chrt_main(int argc UNUSED_PARAM, char **argv)
  80 {
  81         pid_t pid = 0;
  82         unsigned opt;
  83         struct sched_param sp;
  84         char *pid_str;
  85         char *priority = priority; /* for compiler */
  86         const char *current_new;
  87         int policy = SCHED_RR;
  88
  89         opt = getopt32(argv, "^"
  90                         "+" "mprfobi"
  91                         "\0"
  92                         /* only one policy accepted: */
  93                         "r--fobi:f--robi:o--rfbi:b--rfoi:i--rfob"
  94         );
  95         if (opt & OPT_m) { /* print min/max and exit */
  96                 show_min_max(SCHED_OTHER);
  97                 show_min_max(SCHED_FIFO);
  98                 show_min_max(SCHED_RR);
  99                 show_min_max(SCHED_BATCH);
 100                 show_min_max(SCHED_IDLE);
 101                 fflush_stdout_and_exit(EXIT_SUCCESS);
 102         }
 103         //if (opt & OPT_r)
 104         //      policy = SCHED_RR; - default, already set
 105         if (opt & OPT_f)
 106                 policy = SCHED_FIFO;
 107         if (opt & OPT_o)
 108                 policy = SCHED_OTHER;
 109         if (opt & OPT_b)
 110                 policy = SCHED_BATCH;
 111         if (opt & OPT_i)
 112                 policy = SCHED_IDLE;
 113
 114         argv += optind;
 115         if (!argv[0])
 116                 bb_show_usage();
 117         if (opt & OPT_p) {
 118                 pid_str = *argv++;
 119                 if (*argv) { /* "-p PRIO PID [...]" */
 120                         priority = pid_str;
 121                         pid_str = *argv;
 122                 }
 123                 /* else "-p PID", and *argv == NULL */
 124                 pid = xatoul_range(pid_str, 1, ((unsigned)(pid_t)ULONG_MAX) >> 1);
 125         } else {
 126                 priority = *argv++;
 127                 if (!*argv)
 128                         bb_show_usage();
 129         }
 130
 131         current_new = "current\0new";
 132         if (opt & OPT_p) {
 133                 int pol;
 134  print_rt_info:
 135                 pol = sched_getscheduler(pid);
 136                 if (pol < 0)
 137                         bb_perror_msg_and_die("can't %cet pid %u's policy", 'g', (int)pid);
 138 #ifdef SCHED_RESET_ON_FORK
 139                 /* "Since Linux 2.6.32, the SCHED_RESET_ON_FORK flag
 140                  * can be ORed in policy when calling sched_setscheduler().
 141                  * As a result of including this flag, children created by
 142                  * fork(2) do not inherit privileged scheduling policies"
 143                  *
 144                  * This bit is also returned by sched_getscheduler()!
 145                  * (TODO: do we want to show it?)
 146                  */
 147                 pol &= ~SCHED_RESET_ON_FORK;
 148 #endif
 149                 printf("pid %u's %s scheduling policy: SCHED_%s\n",
 150                         pid, current_new, policy_name(pol)
 151                 );
 152                 if (sched_getparam(pid, &sp))
 153                         bb_perror_msg_and_die("can't get pid %u's attributes", (int)pid);
 154                 printf("pid %u's %s scheduling priority: %d\n",
 155                         (int)pid, current_new, sp.sched_priority
 156                 );
 157                 if (!*argv) {
 158                         /* Either it was just "-p PID",
 159                          * or it was "-p PRIO PID" and we came here
 160                          * for the second time (see goto below) */
 161                         return EXIT_SUCCESS;
 162                 }
 163                 *argv = NULL;
 164                 current_new += 8;
 165         }
 166
 167         sp.sched_priority = xstrtou_range(priority, 0,
 168                 sched_get_priority_min(policy), sched_get_priority_max(policy)
 169         );
 170
 171         if (sched_setscheduler(pid, policy, &sp) < 0)
 172                 bb_perror_msg_and_die("can't %cet pid %u's policy", 's', (int)pid);
 173
 174         if (!argv[0]) /* "-p PRIO PID [...]" */
 175                 goto print_rt_info;
 176
 177         BB_EXECVP_or_die(argv);
 178 }