death to trailing whitespace
[openafs.git] / src / procmgmt / afskill.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  *
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 /* Kill command based on process management library (procmgmt.h).
11  * Intended for use on Windows NT, where it will only signal processes
12  * linked with the process management library (except SIGKILL).
13  */
14
15 #include <afsconfig.h>
16 #include <afs/param.h>
17
18
19 #include <afs/stds.h>
20
21 #include <stddef.h>
22 #include <stdlib.h>
23 #include <stdio.h>
24 #include <errno.h>
25 #include <string.h>
26
27 #include "procmgmt.h"
28
29
30 /* Define NULL-terminated array of signal name/number pairs */
31 typedef struct {
32     char *name;
33     int number;
34 } signal_map_t;
35
36 static signal_map_t signalMap[] = {
37     {"HUP", SIGHUP},
38     {"INT", SIGINT},
39     {"QUIT", SIGQUIT},
40     {"ILL", SIGILL},
41     {"ABRT", SIGABRT},
42     {"FPE", SIGFPE},
43     {"KILL", SIGKILL},
44     {"SEGV", SIGSEGV},
45     {"TERM", SIGTERM},
46     {"USR1", SIGUSR1},
47     {"USR2", SIGUSR2},
48     {"CLD", SIGCLD},
49     {"CHLD", SIGCHLD},
50     {"TSTP", SIGTSTP},
51     {NULL, 0}
52 };
53
54
55 static void
56 PrintSignalList(void)
57 {
58     int i;
59
60     for (i = 0; signalMap[i].name; i++) {
61         printf("%s ", signalMap[i].name);
62     }
63     printf("\n");
64 }
65
66 static int
67 SignalArgToNumber(const char *sigargp)
68 {
69     long signo;
70     char *endp;
71     int i;
72
73     /* check for numeric signal arg */
74     errno = 0;
75     signo = strtol(sigargp, &endp, 0);
76
77     if (errno == 0 && *endp == '\0') {
78         return (int)signo;
79     }
80
81     /* check for named signal arg */
82     for (i = 0; signalMap[i].name; i++) {
83         if (!strcasecmp(signalMap[i].name, sigargp)) {
84             return signalMap[i].number;
85         }
86     }
87
88     return -1;                  /* invalid signal argument */
89 }
90
91
92 static void
93 PrintUsage(const char *whoami)
94 {
95     printf("usage: %s [-signal] [pid] ...\n", whoami);
96     printf("       %s -l\n", whoami);
97 }
98
99
100
101 int
102 main(int argc, char *argv[])
103 {
104     int status = 0;
105     char *whoami;
106
107     /* set whoami to last component of argv[0] */
108     if ((whoami = strrchr(argv[0], '/')) != NULL) {
109         whoami++;
110     } else if ((whoami = strrchr(argv[0], '\\')) != NULL) {
111         whoami++;
112     } else {
113         whoami = argv[0];
114     }
115
116     /* parse command arguments */
117     if (argc <= 1) {
118         /* no arguments; print usage statement */
119         PrintUsage(whoami);
120
121     } else if (!strcmp(argv[1], "-l")) {
122         /* request to print a list of valid signals */
123         PrintSignalList();
124
125     } else {
126         /* request to send a signal */
127         int i;
128         int signo = SIGTERM;    /* SIGTERM is default if no signal specified */
129
130         for (i = 1; i < argc; i++) {
131             char *curargp = argv[i];
132
133             if (i == 1 && *curargp == '-') {
134                 /* signal argument specified; translate to a number */
135                 curargp++;      /* advance past signal flag indicator ('-') */
136
137                 if ((signo = SignalArgToNumber(curargp)) == -1) {
138                     printf("%s: unknown signal %s.\n", whoami, curargp);
139                     PrintUsage(whoami);
140                     status = 1;
141                     break;
142                 }
143             } else {
144                 /* pid specified; send signal */
145                 long pid;
146                 char *endp;
147
148                 errno = 0;
149                 pid = strtol(curargp, &endp, 0);
150
151                 if (errno != 0 || *endp != '\0' || pid <= 0) {
152                     printf("%s: invalid pid value %s.\n", whoami, curargp);
153                     status = 1;
154                 } else {
155                     if (kill((pid_t) pid, signo)) {
156                         if (errno == EINVAL) {
157                             printf("%s: invalid signal number %d.\n", whoami,
158                                    signo);
159                             status = 1;
160                             break;
161
162                         } else if (errno == ESRCH) {
163                             printf("%s: no such process %d.\n", whoami,
164                                    (int)pid);
165                             status = 1;
166
167                         } else if (errno == EPERM) {
168                             printf("%s: no privilege to signal %d.\n", whoami,
169                                    (int)pid);
170                             status = 1;
171
172                         } else {
173                             printf("%s: failed to signal %d (errno = %d).\n",
174                                    whoami, (int)pid, (int)errno);
175                             status = 1;
176                         }
177                     }
178                 }
179             }
180         }
181     }
182
183     return status;
184 }