1 /*- 2 * Copyright (c) 2000 Peter Wemm <peter@FreeBSD.org> 3 * Copyright (c) 2000 Paul Saab <ps@FreeBSD.org> 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 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 * 27 * $FreeBSD: src/usr.bin/killall/killall.c,v 1.5.2.4 2001/05/19 19:22:49 phk Exp $ 28 * $DragonFly: src/usr.bin/killall/killall.c,v 1.9 2007/02/01 10:33:26 corecode Exp $ 29 */ 30 31 #include <sys/user.h> 32 #include <sys/param.h> 33 #include <sys/stat.h> 34 #include <sys/sysctl.h> 35 #include <fcntl.h> 36 #include <dirent.h> 37 #include <stdio.h> 38 #include <stdlib.h> 39 #include <string.h> 40 #include <pwd.h> 41 #include <signal.h> 42 #include <regex.h> 43 #include <ctype.h> 44 #include <err.h> 45 #include <errno.h> 46 #include <unistd.h> 47 48 static char *prog; 49 50 static void __dead2 51 usage(void) 52 { 53 54 fprintf(stderr, "usage: %s [-l] [-v] [-m] [-sig] [-u user] [-j jail] [-t tty] [-c cmd] [cmd]...\n", prog); 55 fprintf(stderr, "At least one option or argument to specify processes must be given.\n"); 56 exit(1); 57 } 58 59 static char * 60 upper(const char *str) 61 { 62 static char buf[80]; 63 char *s; 64 65 strlcpy(buf, str, sizeof(buf)); 66 for (s = buf; *s; s++) 67 *s = toupper((unsigned char)*s); 68 return buf; 69 } 70 71 72 static void 73 printsig(FILE *fp) 74 { 75 const char *const * p; 76 int cnt; 77 int offset = 0; 78 79 for (cnt = NSIG, p = sys_signame + 1; --cnt; ++p) { 80 offset += fprintf(fp, "%s ", upper(*p)); 81 if (offset >= 75 && cnt > 1) { 82 offset = 0; 83 fprintf(fp, "\n"); 84 } 85 } 86 fprintf(fp, "\n"); 87 } 88 89 static void 90 nosig(char *name) 91 { 92 93 warnx("unknown signal %s; valid signals:", name); 94 printsig(stderr); 95 exit(1); 96 } 97 98 int 99 main(int ac, char **av) 100 { 101 struct kinfo_proc *procs = NULL, *newprocs; 102 struct stat sb; 103 struct passwd *pw; 104 regex_t rgx; 105 regmatch_t pmatch; 106 int i, j; 107 char buf[256]; 108 char *user = NULL; 109 char *tty = NULL; 110 char *cmd = NULL; 111 int qflag = 0; 112 int vflag = 0; 113 int sflag = 0; 114 int jflag = 0, jailid = 0; 115 int dflag = 0; 116 int mflag = 0; 117 uid_t uid = 0; 118 dev_t tdev = 0; 119 pid_t mypid; 120 char thiscmd[MAXCOMLEN + 1]; 121 pid_t thispid; 122 uid_t thisuid; 123 dev_t thistdev; 124 int sig = SIGTERM; 125 const char *const *p; 126 char *ep; 127 int errors = 0; 128 int mib[4]; 129 size_t miblen; 130 int st, nprocs; 131 size_t size; 132 int matched; 133 int killed = 0; 134 135 prog = av[0]; 136 av++; 137 ac--; 138 139 while (ac > 0) { 140 if (strcmp(*av, "-l") == 0) { 141 printsig(stdout); 142 exit(0); 143 } 144 if (strcmp(*av, "-help") == 0) 145 usage(); 146 if (**av == '-') { 147 ++*av; 148 switch (**av) { 149 case 'u': 150 ++*av; 151 if (**av == '\0') 152 ++av; 153 --ac; 154 user = *av; 155 break; 156 case 't': 157 ++*av; 158 if (**av == '\0') 159 ++av; 160 --ac; 161 tty = *av; 162 break; 163 case 'c': 164 ++*av; 165 if (**av == '\0') 166 ++av; 167 --ac; 168 cmd = *av; 169 break; 170 case 'j': 171 { 172 const char *errstr; 173 ++*av; 174 if (**av == '\0') 175 ++av; 176 --ac; 177 jailid = strtonum(*av, 1, INT_MAX, &errstr); 178 179 if (errstr) 180 errx(1, "jail id is %s: %s", errstr, *av); 181 jflag++; 182 break; 183 } 184 case 'q': 185 qflag++; 186 break; 187 case 'v': 188 vflag++; 189 break; 190 case 's': 191 sflag++; 192 break; 193 case 'd': 194 dflag++; 195 break; 196 case 'm': 197 mflag++; 198 break; 199 default: 200 if (isalpha((unsigned char)**av)) { 201 if (strncasecmp(*av, "sig", 3) == 0) 202 *av += 3; 203 for (sig = NSIG, p = sys_signame + 1; 204 --sig; ++p) 205 if (strcasecmp(*p, *av) == 0) { 206 sig = p - sys_signame; 207 break; 208 } 209 if (!sig) 210 nosig(*av); 211 } else if (isdigit((unsigned char)**av)) { 212 sig = strtol(*av, &ep, 10); 213 if (!*av || *ep) 214 errx(1, "illegal signal number: %s", *av); 215 if (sig < 0 || sig >= NSIG) 216 nosig(*av); 217 } else 218 nosig(*av); 219 } 220 ++av; 221 --ac; 222 } else { 223 break; 224 } 225 } 226 227 if (user == NULL && tty == NULL && cmd == NULL && jflag == 0 && ac == 0) 228 usage(); 229 230 if (tty) { 231 if (strncmp(tty, "/dev/", 5) == 0) 232 snprintf(buf, sizeof(buf), "%s", tty); 233 else if (strncmp(tty, "tty", 3) == 0) 234 snprintf(buf, sizeof(buf), "/dev/%s", tty); 235 else 236 snprintf(buf, sizeof(buf), "/dev/tty%s", tty); 237 if (stat(buf, &sb) < 0) 238 err(1, "stat(%s)", buf); 239 if (!S_ISCHR(sb.st_mode)) 240 errx(1, "%s: not a character device", buf); 241 tdev = sb.st_rdev; 242 if (dflag) 243 printf("ttydev:0x%x\n", tdev); 244 } 245 if (user) { 246 pw = getpwnam(user); 247 if (pw == NULL) 248 errx(1, "user %s does not exist", user); 249 uid = pw->pw_uid; 250 if (dflag) 251 printf("uid:%d\n", uid); 252 } else { 253 uid = getuid(); 254 if (uid != 0) { 255 pw = getpwuid(uid); 256 if (pw) 257 user = pw->pw_name; 258 if (dflag) 259 printf("uid:%d\n", uid); 260 } 261 } 262 size = 0; 263 mib[0] = CTL_KERN; 264 mib[1] = KERN_PROC; 265 mib[2] = KERN_PROC_ALL; 266 mib[3] = 0; 267 miblen = 3; 268 269 if (user && mib[2] == KERN_PROC_ALL) { 270 mib[2] = KERN_PROC_RUID; 271 mib[3] = uid; 272 miblen = 4; 273 } 274 if (tty && mib[2] == KERN_PROC_ALL) { 275 mib[2] = KERN_PROC_TTY; 276 mib[3] = tdev; 277 miblen = 4; 278 } 279 280 st = sysctl(mib, miblen, NULL, &size, NULL, 0); 281 do { 282 size += size / 10; 283 newprocs = realloc(procs, size); 284 if (newprocs == 0) { 285 if (procs) 286 free(procs); 287 errx(1, "could not reallocate memory"); 288 } 289 procs = newprocs; 290 st = sysctl(mib, miblen, procs, &size, NULL, 0); 291 } while (st == -1 && errno == ENOMEM); 292 if (st == -1) 293 err(1, "could not sysctl(KERN_PROC)"); 294 if (size % sizeof(struct kinfo_proc) != 0) { 295 fprintf(stderr, "proc size mismatch (%zu total, %zu chunks)\n", 296 size, sizeof(struct kinfo_proc)); 297 fprintf(stderr, "userland out of sync with kernel, recompile libkvm etc\n"); 298 exit(1); 299 } 300 nprocs = size / sizeof(struct kinfo_proc); 301 if (dflag) 302 printf("nprocs %d\n", nprocs); 303 mypid = getpid(); 304 305 for (i = 0; i < nprocs; i++) { 306 thispid = procs[i].kp_pid; 307 strncpy(thiscmd, procs[i].kp_comm, MAXCOMLEN); 308 thiscmd[MAXCOMLEN] = '\0'; 309 thistdev = procs[i].kp_tdev; 310 thisuid = procs[i].kp_ruid; /* real uid */ 311 312 if (thispid == mypid) 313 continue; 314 matched = 1; 315 if ((int)procs[i].kp_pid < 0) 316 matched = 0; 317 if (user) { 318 if (thisuid != uid) 319 matched = 0; 320 } 321 if (tty) { 322 if (thistdev != tdev) 323 matched = 0; 324 } 325 if (jflag) { 326 if (procs[i].kp_jailid != jailid) 327 matched = 0; 328 } 329 if (cmd) { 330 if (mflag) { 331 if (regcomp(&rgx, cmd, 332 REG_EXTENDED|REG_NOSUB) != 0) { 333 mflag = 0; 334 warnx("%s: illegal regexp", cmd); 335 } 336 } 337 if (mflag) { 338 pmatch.rm_so = 0; 339 pmatch.rm_eo = strlen(thiscmd); 340 if (regexec(&rgx, thiscmd, 0, &pmatch, 341 REG_STARTEND) != 0) 342 matched = 0; 343 regfree(&rgx); 344 } else { 345 if (strcmp(thiscmd, cmd) != 0) 346 matched = 0; 347 } 348 } 349 if (matched == 0) 350 continue; 351 if (ac > 0) { 352 matched = 0; 353 for (j = 0; j < ac; j++) { 354 if (mflag) { 355 if (regcomp(&rgx, av[j], 356 REG_EXTENDED|REG_NOSUB) != 0) { 357 mflag = 0; 358 warnx("%s: illegal regexp", av[j]); 359 } 360 } 361 if (mflag) { 362 pmatch.rm_so = 0; 363 pmatch.rm_eo = strlen(thiscmd); 364 if (regexec(&rgx, thiscmd, 0, &pmatch, 365 REG_STARTEND) == 0) 366 matched = 1; 367 regfree(&rgx); 368 } else { 369 if (strcmp(thiscmd, av[j]) == 0) 370 matched = 1; 371 } 372 if (matched) 373 break; 374 } 375 if (matched == 0) 376 continue; 377 } 378 if (dflag) 379 printf("sig:%d, cmd:%s, pid:%d, dev:0x%x uid:%d\n", sig, 380 thiscmd, thispid, thistdev, thisuid); 381 382 if (vflag || sflag) 383 printf("kill -%s %d\n", upper(sys_signame[sig]), 384 thispid); 385 386 killed++; 387 if (!dflag && !sflag) { 388 if (kill(thispid, sig) < 0 /* && errno != ESRCH */ ) { 389 warn("kill -%s %d", upper(sys_signame[sig]), 390 thispid); 391 errors = 1; 392 } 393 } 394 } 395 if (!qflag && killed == 0) { 396 fprintf(stderr, "No matching processes %swere found\n", 397 getuid() != 0 ? "belonging to you " : ""); 398 errors = 1; 399 } 400 exit(errors); 401 } 402