Home | History | Annotate | Line # | Download | only in nice
nice.c revision 1.14 
      1 /*	$NetBSD: nice.c,v 1.14 2007/12/15 19:44:52 perry Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1989 The Regents of the University of California.
      5  * 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. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 #ifndef lint
     34 __COPYRIGHT(
     35     "@(#) Copyright (c) 1989 The Regents of the University of California.\n\
     36  All rights reserved.\n");
     37 #endif /* not lint */
     38 
     39 #ifndef lint
     40 #if 0
     41 static char sccsid[] = "@(#)nice.c	5.4 (Berkeley) 6/1/90";
     42 #endif
     43 __RCSID("$NetBSD: nice.c,v 1.14 2007/12/15 19:44:52 perry Exp $");
     44 #endif /* not lint */
     45 
     46 #include <sys/time.h>
     47 #include <sys/resource.h>
     48 
     49 #include <stdio.h>
     50 #include <stdlib.h>
     51 #include <string.h>
     52 #include <limits.h>
     53 #include <locale.h>
     54 #include <ctype.h>
     55 #include <errno.h>
     56 #include <err.h>
     57 #include <unistd.h>
     58 
     59 #define	DEFNICE	10
     60 
     61 static void usage(void) __dead;
     62 
     63 int
     64 main(int argc, char **argv)
     65 {
     66 	char *ep;
     67 	int niceness = DEFNICE;
     68 	int c;
     69 	long tmp;
     70 
     71 	setprogname(argv[0]);
     72 	(void)setlocale(LC_ALL, "");
     73 
     74         /* handle obsolete -number syntax */
     75         if (argc > 1 && argv[1][0] == '-' &&
     76 	    isdigit((unsigned char)argv[1][1])) {
     77 		niceness = atoi (argv[1] + 1);
     78                 argc--; argv++;
     79         }
     80 
     81 	while ((c = getopt (argc, argv, "n:")) != -1) {
     82 		switch (c) {
     83 		case 'n':
     84 			errno = 0;
     85 			tmp = strtol(optarg, &ep, 10);
     86 			if (*ep != '\0' || tmp < INT_MIN || tmp > INT_MAX)
     87 				errx(EXIT_FAILURE, "invalid argument: `%s'",
     88 				    optarg);
     89 			niceness = (int)tmp;
     90 			break;
     91 		default:
     92 			usage();
     93 			break;
     94 		}
     95 	}
     96 	argc -= optind;
     97 	argv += optind;
     98 
     99 	if (argc == 0)
    100 		usage();
    101 
    102 	errno = 0;
    103 	niceness += getpriority(PRIO_PROCESS, 0);
    104 	if (errno) {
    105 		err(EXIT_FAILURE, "getpriority");
    106 		/* NOTREACHED */
    107 	}
    108 	if (setpriority(PRIO_PROCESS, 0, niceness) == -1) {
    109 		warn("setpriority");
    110 	}
    111 
    112 	(void)execvp(argv[0], &argv[0]);
    113 	err((errno == ENOENT || errno == ENOTDIR) ? 127 : 126, "%s", argv[0]);
    114 	/* NOTREACHED */
    115 }
    116 
    117 static void
    118 usage(void)
    119 {
    120 	(void)fprintf(stderr,
    121 	    "Usage: %s [ -n increment ] utility [ argument ...]\n",
    122 	    getprogname());
    123 	exit(EXIT_FAILURE);
    124 }
    125