src/nice.c

   1 /* nice -- run a program with modified scheduling priority
   2    Copyright (C) 1990, 1991 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software
  16    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17
  18 /* David MacKenzie <djm@ai.mit.edu> */
  19
  20 #include <stdio.h>
  21 #include <getopt.h>
  22 #include <sys/types.h>
  23 #ifndef NICE_PRIORITY
  24 #include <sys/time.h>
  25 #include <sys/resource.h>
  26 #endif
  27 #include "system.h"
  28
  29 void error ();
  30
  31 static int isinteger ();
  32 static void usage ();
  33
  34 /* The name this program was run with. */
  35 char *program_name;
  36
  37 static struct option const longopts[] =
  38 {
  39   {"adjustment", 1, NULL, 'n'},
  40   {NULL, 0, NULL, 0}
  41 };
  42
  43 void
  44 main (argc, argv)
  45      int argc;
  46      char **argv;
  47 {
  48   int current_priority;
  49   int adjustment = 0;
  50   int minusflag = 0;
  51   int adjustment_given = 0;
  52   int optc;
  53
  54   program_name = argv[0];
  55
  56   while ((optc = getopt_long (argc, argv, "+0123456789-n:", longopts,
  57                               (int *) 0)) != EOF)
  58     {
  59       switch (optc)
  60         {
  61         case '?':
  62           usage ();
  63
  64         case 'n':
  65           if (!isinteger (optarg))
  66             error (1, 0, "invalid priority `%s'", optarg);
  67           adjustment = atoi (optarg);
  68           adjustment_given = 1;
  69           break;
  70
  71         case '-':
  72           minusflag = 1;
  73           break;
  74
  75         default:
  76           adjustment = adjustment * 10 + optc - '0';
  77           adjustment_given = 1;
  78         }
  79     }
  80
  81   if (minusflag)
  82     adjustment = -adjustment;
  83   if (!adjustment_given)
  84     adjustment = 10;
  85
  86   if (optind == argc)
  87     {
  88       if (adjustment_given)
  89         usage ();
  90       /* No command given; print the priority. */
  91       errno = 0;
  92 #ifndef NICE_PRIORITY
  93       current_priority = getpriority (PRIO_PROCESS, 0);
  94 #else
  95       current_priority = nice (0);
  96 #endif
  97       if (current_priority == -1 && errno != 0)
  98         error (1, errno, "cannot get priority");
  99       printf ("%d\n", current_priority);
 100       exit (0);
 101     }
 102
 103   errno = 0;
 104 #ifndef NICE_PRIORITY
 105   current_priority = getpriority (PRIO_PROCESS, 0);
 106 #else
 107   current_priority = nice (0);
 108 #endif
 109   if (current_priority == -1 && errno != 0)
 110     error (1, errno, "cannot get priority");
 111
 112 #ifndef NICE_PRIORITY
 113   if (setpriority (PRIO_PROCESS, 0, current_priority + adjustment))
 114 #else
 115   if (nice (adjustment) == -1)
 116 #endif
 117     error (1, errno, "cannot set priority");
 118
 119   execvp (argv[optind], &argv[optind]);
 120   error (errno == ENOENT ? 127 : 126, errno, "%s", argv[optind]);
 121 }
 122
 123 /* Return nonzero if S represents a (possibly signed) decimal integer,
 124    zero if not. */
 125
 126 static int
 127 isinteger (s)
 128      char *s;
 129 {
 130   if (*s == '-')
 131     ++s;
 132   if (*s == 0)
 133     return 0;
 134   while (*s)
 135     {
 136       if (*s < '0' || *s > '9')
 137         return 0;
 138       ++s;
 139     }
 140   return 1;
 141 }
 142
 143 static void
 144 usage ()
 145 {
 146   fprintf (stderr, "\
 147 Usage: %s [-n adjustment] [-adjustment] [--adjustment=adjustment]\n\
 148        [command [arg...]]\n",
 149            program_name);
 150   exit (1);
 151 }