renice/renice.c

   1 /*      $NetBSD: renice.c,v 1.5 1997/10/19 14:01:38 lukem Exp $ */
   2
   3 /*
   4  * Copyright (c) 1983, 1989, 1993
   5  *      The Regents of the University of California.  All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  * 3. All advertising materials mentioning features or use of this software
  16  *    must display the following acknowledgement:
  17  *      This product includes software developed by the University of
  18  *      California, Berkeley and its contributors.
  19  * 4. Neither the name of the University nor the names of its contributors
  20  *    may be used to endorse or promote products derived from this software
  21  *    without specific prior written permission.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  */
  35
  36 #include <sys/cdefs.h>
  37 #ifndef lint
  38 __COPYRIGHT("@(#) Copyright (c) 1983, 1989, 1993\n\
  39         The Regents of the University of California.  All rights reserved.\n");
  40 #endif /* not lint */
  41
  42 #ifndef lint
  43 /*static char sccsid[] = "from: @(#)renice.c    8.1 (Berkeley) 6/9/93";*/
  44 __RCSID("$NetBSD: renice.c,v 1.5 1997/10/19 14:01:38 lukem Exp $");
  45 #endif /* not lint */
  46
  47 #include <sys/types.h>
  48 #include <sys/time.h>
  49 #include <sys/resource.h>
  50
  51 #include <err.h>
  52 #include <pwd.h>
  53 #include <stdio.h>
  54 #include <stdlib.h>
  55
  56 int     donice __P((int, int, int));
  57 int     main __P((int, char **));
  58
  59 /*
  60  * Change the priority (nice) of processes
  61  * or groups of processes which are already
  62  * running.
  63  */
  64 int
  65 main(argc, argv)
  66         int argc;
  67         char **argv;
  68 {
  69         int which = PRIO_PROCESS;
  70         int who = 0, prio, errs = 0;
  71
  72         argc--, argv++;
  73         if (argc < 2) {
  74                 fprintf(stderr, "usage: renice priority [ [ -p ] pids ] ");
  75                 fprintf(stderr, "[ [ -g ] pgrps ] [ [ -u ] users ]\n");
  76                 exit(1);
  77         }
  78         prio = atoi(*argv);
  79         argc--, argv++;
  80         if (prio > PRIO_MAX)
  81                 prio = PRIO_MAX;
  82         if (prio < PRIO_MIN)
  83                 prio = PRIO_MIN;
  84         for (; argc > 0; argc--, argv++) {
  85                 if (strcmp(*argv, "-g") == 0) {
  86                         which = PRIO_PGRP;
  87                         continue;
  88                 }
  89                 if (strcmp(*argv, "-u") == 0) {
  90                         which = PRIO_USER;
  91                         continue;
  92                 }
  93                 if (strcmp(*argv, "-p") == 0) {
  94                         which = PRIO_PROCESS;
  95                         continue;
  96                 }
  97                 if (which == PRIO_USER) {
  98                         register struct passwd *pwd = getpwnam(*argv);
  99
 100                         if (pwd == NULL) {
 101                                 warnx("%s: unknown user", *argv);
 102                                 continue;
 103                         }
 104                         who = pwd->pw_uid;
 105                 } else {
 106                         who = atoi(*argv);
 107                         if (who < 0) {
 108                                 warnx("%s: bad value", *argv);
 109                                 continue;
 110                         }
 111                 }
 112                 errs += donice(which, who, prio);
 113         }
 114         exit(errs != 0);
 115 }
 116
 117 int
 118 donice(which, who, prio)
 119         int which, who, prio;
 120 {
 121         int oldprio;
 122         extern int errno;
 123
 124         errno = 0, oldprio = getpriority(which, who);
 125         if (oldprio == -1 && errno) {
 126                 warn("%d: getpriority", who);
 127                 return (1);
 128         }
 129         if (setpriority(which, who, prio) < 0) {
 130                 warn("%d: setpriority", who);
 131                 return (1);
 132         }
 133         printf("%d: old priority %d, new priority %d\n", who, oldprio, prio);
 134         return (0);
 135 }