usr.sbin/rtprio/rtprio.c

   1 /*
   2  * Copyright (c) 1994 David Greenman
   3  * Copyright (c) 1994 Henrik Vestergaard Draboel (hvd@terry.ping.dk)
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  * 3. All advertising materials mentioning features or use of this software
  15  *    must display the following acknowledgement:
  16  *      This product includes software developed by Henrik Vestergaard Draboel.
  17  *      This product includes software developed by David Greenman.
  18  * 4. Neither the names of the authors nor the names of contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE AUTHORS AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $FreeBSD: src/usr.sbin/rtprio/rtprio.c,v 1.8.2.1 2001/01/29 08:07:26 kuriyama Exp $
  35  * $DragonFly: src/usr.sbin/rtprio/rtprio.c,v 1.2 2003/06/17 04:30:03 dillon Exp $
  36  */
  37
  38 #include <sys/param.h>
  39 #include <sys/rtprio.h>
  40 #include <sys/errno.h>
  41
  42 #include <ctype.h>
  43 #include <err.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <string.h>
  47 #include <unistd.h>
  48
  49 static void usage();
  50
  51 int
  52 main(argc, argv)
  53         int     argc;
  54         char  **argv;
  55 {
  56         char   *p;
  57         int     proc = 0;
  58         struct rtprio rtp;
  59
  60         /* find basename */
  61         if ((p = rindex(argv[0], '/')) == NULL)
  62                 p = argv[0];
  63         else
  64                 ++p;
  65
  66         if (!strcmp(p, "rtprio"))
  67                 rtp.type = RTP_PRIO_REALTIME;
  68         else if (!strcmp(p, "idprio"))
  69                 rtp.type = RTP_PRIO_IDLE;
  70
  71         switch (argc) {
  72         case 2:
  73                 proc = abs(atoi(argv[1]));      /* Should check if numeric
  74                                                  * arg! */
  75                 /* FALLTHROUGH */
  76         case 1:
  77                 if (rtprio(RTP_LOOKUP, proc, &rtp) != 0)
  78                         err(1, "%s", argv[0]);
  79                 printf("%s: ", p);
  80                 switch (rtp.type) {
  81                 case RTP_PRIO_REALTIME:
  82                 case RTP_PRIO_FIFO:
  83                         printf("realtime priority %d\n", rtp.prio);
  84                         break;
  85                 case RTP_PRIO_NORMAL:
  86                         printf("normal priority\n");
  87                         break;
  88                 case RTP_PRIO_IDLE:
  89                         printf("idle priority %d\n", rtp.prio);
  90                         break;
  91                 default:
  92                         printf("invalid priority type %d\n", rtp.type);
  93                         break;
  94                 }
  95                 exit(0);
  96         default:
  97                 if (argv[1][0] == '-' || isdigit(argv[1][0])) {
  98                         if (argv[1][0] == '-') {
  99                                 if (strcmp(argv[1], "-t") == 0) {
 100                                         rtp.type = RTP_PRIO_NORMAL;
 101                                         rtp.prio = 0;
 102                                 } else {
 103                                         usage();
 104                                         break;
 105                                 }
 106                         } else {
 107                                 rtp.prio = atoi(argv[1]);
 108                         }
 109                 } else {
 110                         usage();
 111                         break;
 112                 }
 113
 114                 if (argv[2][0] == '-')
 115                         proc = -atoi(argv[2]);
 116
 117                 if (rtprio(RTP_SET, proc, &rtp) != 0)
 118                         err(1, "%s", argv[0]);
 119
 120                 if (proc == 0) {
 121                         execvp(argv[2], &argv[2]);
 122                         err(1, "%s", argv[2]);
 123                 }
 124                 exit(0);
 125         }
 126         exit (1);
 127 }
 128
 129 static void
 130 usage()
 131 {
 132         (void) fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n",
 133                 "usage: [id|rt]prio",
 134                 "       [id|rt]prio [-]pid",
 135                 "       [id|rt]prio priority command [args]",
 136                 "       [id|rt]prio priority -pid",
 137                 "       [id|rt]prio -t command [args]",
 138                 "       [id|rt]prio -t -pid");
 139         exit(1);
 140 }