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  *      $Id$
  35  */
  36
  37 #include <sys/param.h>
  38 #include <sys/rtprio.h>
  39 #include <sys/errno.h>
  40
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <ctype.h>
  45 #include <unistd.h>
  46
  47 static void usage();
  48
  49 int
  50 main(argc, argv)
  51         int     argc;
  52         char  **argv;
  53 {
  54         char   *p;
  55         int     proc = 0;
  56         struct rtprio rtp;
  57
  58         /* find basename */
  59         if ((p = rindex(argv[0], '/')) == NULL)
  60                 p = argv[0];
  61         else
  62                 ++p;
  63
  64         if (!strcmp(p, "rtprio"))
  65                 rtp.type = RTP_PRIO_REALTIME;
  66         else if (!strcmp(p, "idprio"))
  67                 rtp.type = RTP_PRIO_IDLE;
  68
  69         switch (argc) {
  70         case 2:
  71                 proc = abs(atoi(argv[1]));      /* Should check if numeric
  72                                                  * arg! */
  73                 /* FALLTHROUGH */
  74         case 1:
  75                 if (rtprio(RTP_LOOKUP, proc, &rtp) != 0) {
  76                         perror(argv[0]);
  77                         exit (1);
  78                 }
  79                 printf("%s: ", p);
  80                 switch (rtp.type) {
  81                 case RTP_PRIO_REALTIME:
  82                         printf("realtime priority %d\n", rtp.prio);
  83                         break;
  84                 case RTP_PRIO_NORMAL:
  85                         printf("normal priority\n");
  86                         break;
  87                 case RTP_PRIO_IDLE:
  88                         printf("idle priority %d\n", rtp.prio);
  89                         break;
  90                 default:
  91                         printf("invalid priority type %d\n", rtp.type);
  92                         break;
  93                 }
  94                 exit(0);
  95         default:
  96                 if (argv[1][0] == '-' || isdigit(argv[1][0])) {
  97                         if (argv[1][0] == '-') {
  98                                 if (strcmp(argv[1], "-t") == 0) {
  99                                         rtp.type = RTP_PRIO_NORMAL;
 100                                         rtp.prio = 0;
 101                                 } else {
 102                                         usage(p);
 103                                         break;
 104                                 }
 105                         } else {
 106                                 rtp.prio = atoi(argv[1]);
 107                         }
 108                 } else {
 109                         usage(p);
 110                         break;
 111                 }
 112
 113                 if (argv[2][0] == '-')
 114                         proc = -atoi(argv[2]);
 115
 116                 if (rtprio(RTP_SET, proc, &rtp) != 0) {
 117                         perror(argv[0]);
 118                         exit (1);
 119                 }
 120
 121                 if (proc == 0) {
 122                         execvp(argv[2], &argv[2]);
 123                         perror(argv[0]);
 124                         exit (1);
 125                 }
 126         }
 127         exit (1);
 128 }
 129
 130 static void
 131 usage(basename)
 132         char   *basename;
 133 {
 134         (void) fprintf(stderr, "usage: %s\n", basename);
 135         (void) fprintf(stderr, "usage: %s [-]pid\n", basename);
 136         (void) fprintf(stderr, "usage: %s priority command [ args ] \n", basename);
 137         (void) fprintf(stderr, "usage: %s priority -pid \n", basename);
 138         (void) fprintf(stderr, "usage: %s -t command [ args ] \n", basename);
 139         (void) fprintf(stderr, "usage: %s -t -pid \n", basename);
 140
 141         exit(-1);
 142 }