121be3b31SYaroslav Tykhiy /* $NetBSD: pkill.c,v 1.16 2005/10/10 22:13:20 kleink Exp $ */ 221be3b31SYaroslav Tykhiy 321be3b31SYaroslav Tykhiy /*- 421be3b31SYaroslav Tykhiy * Copyright (c) 2002 The NetBSD Foundation, Inc. 521be3b31SYaroslav Tykhiy * Copyright (c) 2005 Pawel Jakub Dawidek <pjd@FreeBSD.org> 621be3b31SYaroslav Tykhiy * All rights reserved. 721be3b31SYaroslav Tykhiy * 821be3b31SYaroslav Tykhiy * This code is derived from software contributed to The NetBSD Foundation 921be3b31SYaroslav Tykhiy * by Andrew Doran. 1021be3b31SYaroslav Tykhiy * 1121be3b31SYaroslav Tykhiy * Redistribution and use in source and binary forms, with or without 1221be3b31SYaroslav Tykhiy * modification, are permitted provided that the following conditions 1321be3b31SYaroslav Tykhiy * are met: 1421be3b31SYaroslav Tykhiy * 1. Redistributions of source code must retain the above copyright 1521be3b31SYaroslav Tykhiy * notice, this list of conditions and the following disclaimer. 1621be3b31SYaroslav Tykhiy * 2. Redistributions in binary form must reproduce the above copyright 1721be3b31SYaroslav Tykhiy * notice, this list of conditions and the following disclaimer in the 1821be3b31SYaroslav Tykhiy * documentation and/or other materials provided with the distribution. 1921be3b31SYaroslav Tykhiy * 2021be3b31SYaroslav Tykhiy * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 2121be3b31SYaroslav Tykhiy * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 2221be3b31SYaroslav Tykhiy * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 2321be3b31SYaroslav Tykhiy * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 2421be3b31SYaroslav Tykhiy * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 2521be3b31SYaroslav Tykhiy * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 2621be3b31SYaroslav Tykhiy * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 2721be3b31SYaroslav Tykhiy * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 2821be3b31SYaroslav Tykhiy * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 2921be3b31SYaroslav Tykhiy * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 3021be3b31SYaroslav Tykhiy * POSSIBILITY OF SUCH DAMAGE. 3121be3b31SYaroslav Tykhiy */ 3221be3b31SYaroslav Tykhiy 3321be3b31SYaroslav Tykhiy #include <sys/cdefs.h> 3421be3b31SYaroslav Tykhiy __FBSDID("$FreeBSD$"); 3521be3b31SYaroslav Tykhiy 3621be3b31SYaroslav Tykhiy #include <sys/types.h> 3721be3b31SYaroslav Tykhiy #include <sys/param.h> 3821be3b31SYaroslav Tykhiy #include <sys/sysctl.h> 3921be3b31SYaroslav Tykhiy #include <sys/proc.h> 4021be3b31SYaroslav Tykhiy #include <sys/queue.h> 4121be3b31SYaroslav Tykhiy #include <sys/stat.h> 4221be3b31SYaroslav Tykhiy #include <sys/time.h> 4321be3b31SYaroslav Tykhiy #include <sys/user.h> 4421be3b31SYaroslav Tykhiy 457b0706f6SPawel Jakub Dawidek #include <assert.h> 4621be3b31SYaroslav Tykhiy #include <stdio.h> 4721be3b31SYaroslav Tykhiy #include <stdlib.h> 4821be3b31SYaroslav Tykhiy #include <limits.h> 4921be3b31SYaroslav Tykhiy #include <paths.h> 5021be3b31SYaroslav Tykhiy #include <string.h> 5121be3b31SYaroslav Tykhiy #include <unistd.h> 5221be3b31SYaroslav Tykhiy #include <signal.h> 5321be3b31SYaroslav Tykhiy #include <regex.h> 5421be3b31SYaroslav Tykhiy #include <ctype.h> 5521be3b31SYaroslav Tykhiy #include <fcntl.h> 5621be3b31SYaroslav Tykhiy #include <kvm.h> 5721be3b31SYaroslav Tykhiy #include <err.h> 5821be3b31SYaroslav Tykhiy #include <pwd.h> 5921be3b31SYaroslav Tykhiy #include <grp.h> 6021be3b31SYaroslav Tykhiy #include <errno.h> 6121be3b31SYaroslav Tykhiy #include <locale.h> 62c4f0631fSJamie Gritton #include <jail.h> 6321be3b31SYaroslav Tykhiy 6421be3b31SYaroslav Tykhiy #define STATUS_MATCH 0 6521be3b31SYaroslav Tykhiy #define STATUS_NOMATCH 1 6621be3b31SYaroslav Tykhiy #define STATUS_BADUSAGE 2 6721be3b31SYaroslav Tykhiy #define STATUS_ERROR 3 6821be3b31SYaroslav Tykhiy 6921be3b31SYaroslav Tykhiy #define MIN_PID 5 7021be3b31SYaroslav Tykhiy #define MAX_PID 99999 7121be3b31SYaroslav Tykhiy 7221be3b31SYaroslav Tykhiy /* Ignore system-processes (if '-S' flag is not specified) and myself. */ 7321be3b31SYaroslav Tykhiy #define PSKIP(kp) ((kp)->ki_pid == mypid || \ 74*db57c70aSKonstantin Belousov (!kthreads && ((kp)->ki_flag & P_KPROC) != 0)) 7521be3b31SYaroslav Tykhiy 7621be3b31SYaroslav Tykhiy enum listtype { 7721be3b31SYaroslav Tykhiy LT_GENERIC, 7821be3b31SYaroslav Tykhiy LT_USER, 7921be3b31SYaroslav Tykhiy LT_GROUP, 8021be3b31SYaroslav Tykhiy LT_TTY, 8121be3b31SYaroslav Tykhiy LT_PGRP, 82c4f0631fSJamie Gritton LT_JAIL, 83cc55ad3dSEdward Tomasz Napierala LT_SID, 84cc55ad3dSEdward Tomasz Napierala LT_CLASS 8521be3b31SYaroslav Tykhiy }; 8621be3b31SYaroslav Tykhiy 8721be3b31SYaroslav Tykhiy struct list { 8821be3b31SYaroslav Tykhiy SLIST_ENTRY(list) li_chain; 8921be3b31SYaroslav Tykhiy long li_number; 90cc55ad3dSEdward Tomasz Napierala char *li_name; 9121be3b31SYaroslav Tykhiy }; 9221be3b31SYaroslav Tykhiy 9321be3b31SYaroslav Tykhiy SLIST_HEAD(listhead, list); 9421be3b31SYaroslav Tykhiy 9521be3b31SYaroslav Tykhiy static struct kinfo_proc *plist; 9621be3b31SYaroslav Tykhiy static char *selected; 9721be3b31SYaroslav Tykhiy static const char *delim = "\n"; 9821be3b31SYaroslav Tykhiy static int nproc; 9921be3b31SYaroslav Tykhiy static int pgrep; 10021be3b31SYaroslav Tykhiy static int signum = SIGTERM; 10121be3b31SYaroslav Tykhiy static int newest; 10221be3b31SYaroslav Tykhiy static int oldest; 10321be3b31SYaroslav Tykhiy static int interactive; 10421be3b31SYaroslav Tykhiy static int inverse; 10521be3b31SYaroslav Tykhiy static int longfmt; 10621be3b31SYaroslav Tykhiy static int matchargs; 10721be3b31SYaroslav Tykhiy static int fullmatch; 10821be3b31SYaroslav Tykhiy static int kthreads; 10921be3b31SYaroslav Tykhiy static int cflags = REG_EXTENDED; 1107b0706f6SPawel Jakub Dawidek static int quiet; 11121be3b31SYaroslav Tykhiy static kvm_t *kd; 11221be3b31SYaroslav Tykhiy static pid_t mypid; 11321be3b31SYaroslav Tykhiy 11413e403fdSAntoine Brodin static struct listhead euidlist = SLIST_HEAD_INITIALIZER(euidlist); 11513e403fdSAntoine Brodin static struct listhead ruidlist = SLIST_HEAD_INITIALIZER(ruidlist); 11613e403fdSAntoine Brodin static struct listhead rgidlist = SLIST_HEAD_INITIALIZER(rgidlist); 11713e403fdSAntoine Brodin static struct listhead pgrplist = SLIST_HEAD_INITIALIZER(pgrplist); 11813e403fdSAntoine Brodin static struct listhead ppidlist = SLIST_HEAD_INITIALIZER(ppidlist); 11913e403fdSAntoine Brodin static struct listhead tdevlist = SLIST_HEAD_INITIALIZER(tdevlist); 12013e403fdSAntoine Brodin static struct listhead sidlist = SLIST_HEAD_INITIALIZER(sidlist); 12113e403fdSAntoine Brodin static struct listhead jidlist = SLIST_HEAD_INITIALIZER(jidlist); 122cc55ad3dSEdward Tomasz Napierala static struct listhead classlist = SLIST_HEAD_INITIALIZER(classlist); 12321be3b31SYaroslav Tykhiy 12421be3b31SYaroslav Tykhiy static void usage(void) __attribute__((__noreturn__)); 12521be3b31SYaroslav Tykhiy static int killact(const struct kinfo_proc *); 12621be3b31SYaroslav Tykhiy static int grepact(const struct kinfo_proc *); 12721be3b31SYaroslav Tykhiy static void makelist(struct listhead *, enum listtype, char *); 12821be3b31SYaroslav Tykhiy static int takepid(const char *, int); 12921be3b31SYaroslav Tykhiy 13021be3b31SYaroslav Tykhiy int 13121be3b31SYaroslav Tykhiy main(int argc, char **argv) 13221be3b31SYaroslav Tykhiy { 13321be3b31SYaroslav Tykhiy char buf[_POSIX2_LINE_MAX], *mstr, **pargv, *p, *q, *pidfile; 13421be3b31SYaroslav Tykhiy const char *execf, *coref; 1353ba0e470SBrian Somers int ancestors, debug_opt, did_action; 13621be3b31SYaroslav Tykhiy int i, ch, bestidx, rv, criteria, pidfromfile, pidfilelock; 13721be3b31SYaroslav Tykhiy size_t jsz; 13821be3b31SYaroslav Tykhiy int (*action)(const struct kinfo_proc *); 13921be3b31SYaroslav Tykhiy struct kinfo_proc *kp; 14021be3b31SYaroslav Tykhiy struct list *li; 14121be3b31SYaroslav Tykhiy struct timeval best_tval; 14221be3b31SYaroslav Tykhiy regex_t reg; 14321be3b31SYaroslav Tykhiy regmatch_t regmatch; 1440a091aebSBrian Somers pid_t pid; 14521be3b31SYaroslav Tykhiy 14621be3b31SYaroslav Tykhiy setlocale(LC_ALL, ""); 14721be3b31SYaroslav Tykhiy 14821be3b31SYaroslav Tykhiy if (strcmp(getprogname(), "pgrep") == 0) { 14921be3b31SYaroslav Tykhiy action = grepact; 15021be3b31SYaroslav Tykhiy pgrep = 1; 15121be3b31SYaroslav Tykhiy } else { 15221be3b31SYaroslav Tykhiy action = killact; 15321be3b31SYaroslav Tykhiy p = argv[1]; 15421be3b31SYaroslav Tykhiy 15521be3b31SYaroslav Tykhiy if (argc > 1 && p[0] == '-') { 15621be3b31SYaroslav Tykhiy p++; 15721be3b31SYaroslav Tykhiy i = (int)strtol(p, &q, 10); 15821be3b31SYaroslav Tykhiy if (*q == '\0') { 15921be3b31SYaroslav Tykhiy signum = i; 16021be3b31SYaroslav Tykhiy argv++; 16121be3b31SYaroslav Tykhiy argc--; 16221be3b31SYaroslav Tykhiy } else { 16312dacf62SJilles Tjoelker if (strncasecmp(p, "SIG", 3) == 0) 16421be3b31SYaroslav Tykhiy p += 3; 16521be3b31SYaroslav Tykhiy for (i = 1; i < NSIG; i++) 16621be3b31SYaroslav Tykhiy if (strcasecmp(sys_signame[i], p) == 0) 16721be3b31SYaroslav Tykhiy break; 16821be3b31SYaroslav Tykhiy if (i != NSIG) { 16921be3b31SYaroslav Tykhiy signum = i; 17021be3b31SYaroslav Tykhiy argv++; 17121be3b31SYaroslav Tykhiy argc--; 17221be3b31SYaroslav Tykhiy } 17321be3b31SYaroslav Tykhiy } 17421be3b31SYaroslav Tykhiy } 17521be3b31SYaroslav Tykhiy } 17621be3b31SYaroslav Tykhiy 1770a091aebSBrian Somers ancestors = 0; 17821be3b31SYaroslav Tykhiy criteria = 0; 17921be3b31SYaroslav Tykhiy debug_opt = 0; 18021be3b31SYaroslav Tykhiy pidfile = NULL; 18121be3b31SYaroslav Tykhiy pidfilelock = 0; 1827b0706f6SPawel Jakub Dawidek quiet = 0; 183c08dcaf1SRebecca Cran execf = NULL; 184c08dcaf1SRebecca Cran coref = _PATH_DEVNULL; 18521be3b31SYaroslav Tykhiy 186cc55ad3dSEdward Tomasz Napierala while ((ch = getopt(argc, argv, "DF:G:ILM:N:P:SU:ac:d:fg:ij:lnoqs:t:u:vx")) != -1) 18721be3b31SYaroslav Tykhiy switch (ch) { 18821be3b31SYaroslav Tykhiy case 'D': 18921be3b31SYaroslav Tykhiy debug_opt++; 19021be3b31SYaroslav Tykhiy break; 19121be3b31SYaroslav Tykhiy case 'F': 19221be3b31SYaroslav Tykhiy pidfile = optarg; 19321be3b31SYaroslav Tykhiy criteria = 1; 19421be3b31SYaroslav Tykhiy break; 19521be3b31SYaroslav Tykhiy case 'G': 19621be3b31SYaroslav Tykhiy makelist(&rgidlist, LT_GROUP, optarg); 19721be3b31SYaroslav Tykhiy criteria = 1; 19821be3b31SYaroslav Tykhiy break; 19921be3b31SYaroslav Tykhiy case 'I': 20021be3b31SYaroslav Tykhiy if (pgrep) 20121be3b31SYaroslav Tykhiy usage(); 20221be3b31SYaroslav Tykhiy interactive = 1; 20321be3b31SYaroslav Tykhiy break; 20421be3b31SYaroslav Tykhiy case 'L': 20521be3b31SYaroslav Tykhiy pidfilelock = 1; 20621be3b31SYaroslav Tykhiy break; 20721be3b31SYaroslav Tykhiy case 'M': 20821be3b31SYaroslav Tykhiy coref = optarg; 20921be3b31SYaroslav Tykhiy break; 21021be3b31SYaroslav Tykhiy case 'N': 21121be3b31SYaroslav Tykhiy execf = optarg; 21221be3b31SYaroslav Tykhiy break; 21321be3b31SYaroslav Tykhiy case 'P': 21421be3b31SYaroslav Tykhiy makelist(&ppidlist, LT_GENERIC, optarg); 21521be3b31SYaroslav Tykhiy criteria = 1; 21621be3b31SYaroslav Tykhiy break; 21721be3b31SYaroslav Tykhiy case 'S': 21821be3b31SYaroslav Tykhiy if (!pgrep) 21921be3b31SYaroslav Tykhiy usage(); 22021be3b31SYaroslav Tykhiy kthreads = 1; 22121be3b31SYaroslav Tykhiy break; 22221be3b31SYaroslav Tykhiy case 'U': 22321be3b31SYaroslav Tykhiy makelist(&ruidlist, LT_USER, optarg); 22421be3b31SYaroslav Tykhiy criteria = 1; 22521be3b31SYaroslav Tykhiy break; 2260a091aebSBrian Somers case 'a': 2270a091aebSBrian Somers ancestors++; 2280a091aebSBrian Somers break; 229cc55ad3dSEdward Tomasz Napierala case 'c': 230cc55ad3dSEdward Tomasz Napierala makelist(&classlist, LT_CLASS, optarg); 231cc55ad3dSEdward Tomasz Napierala criteria = 1; 232cc55ad3dSEdward Tomasz Napierala break; 23321be3b31SYaroslav Tykhiy case 'd': 23421be3b31SYaroslav Tykhiy if (!pgrep) 23521be3b31SYaroslav Tykhiy usage(); 23621be3b31SYaroslav Tykhiy delim = optarg; 23721be3b31SYaroslav Tykhiy break; 23821be3b31SYaroslav Tykhiy case 'f': 23921be3b31SYaroslav Tykhiy matchargs = 1; 24021be3b31SYaroslav Tykhiy break; 24121be3b31SYaroslav Tykhiy case 'g': 24221be3b31SYaroslav Tykhiy makelist(&pgrplist, LT_PGRP, optarg); 24321be3b31SYaroslav Tykhiy criteria = 1; 24421be3b31SYaroslav Tykhiy break; 24521be3b31SYaroslav Tykhiy case 'i': 24621be3b31SYaroslav Tykhiy cflags |= REG_ICASE; 24721be3b31SYaroslav Tykhiy break; 24821be3b31SYaroslav Tykhiy case 'j': 249c4f0631fSJamie Gritton makelist(&jidlist, LT_JAIL, optarg); 25021be3b31SYaroslav Tykhiy criteria = 1; 25121be3b31SYaroslav Tykhiy break; 25221be3b31SYaroslav Tykhiy case 'l': 25321be3b31SYaroslav Tykhiy longfmt = 1; 25421be3b31SYaroslav Tykhiy break; 25521be3b31SYaroslav Tykhiy case 'n': 25621be3b31SYaroslav Tykhiy newest = 1; 25721be3b31SYaroslav Tykhiy criteria = 1; 25821be3b31SYaroslav Tykhiy break; 25921be3b31SYaroslav Tykhiy case 'o': 26021be3b31SYaroslav Tykhiy oldest = 1; 26121be3b31SYaroslav Tykhiy criteria = 1; 26221be3b31SYaroslav Tykhiy break; 2637b0706f6SPawel Jakub Dawidek case 'q': 2647b0706f6SPawel Jakub Dawidek if (!pgrep) 2657b0706f6SPawel Jakub Dawidek usage(); 2667b0706f6SPawel Jakub Dawidek quiet = 1; 2677b0706f6SPawel Jakub Dawidek break; 26821be3b31SYaroslav Tykhiy case 's': 26921be3b31SYaroslav Tykhiy makelist(&sidlist, LT_SID, optarg); 27021be3b31SYaroslav Tykhiy criteria = 1; 27121be3b31SYaroslav Tykhiy break; 27221be3b31SYaroslav Tykhiy case 't': 27321be3b31SYaroslav Tykhiy makelist(&tdevlist, LT_TTY, optarg); 27421be3b31SYaroslav Tykhiy criteria = 1; 27521be3b31SYaroslav Tykhiy break; 27621be3b31SYaroslav Tykhiy case 'u': 27721be3b31SYaroslav Tykhiy makelist(&euidlist, LT_USER, optarg); 27821be3b31SYaroslav Tykhiy criteria = 1; 27921be3b31SYaroslav Tykhiy break; 28021be3b31SYaroslav Tykhiy case 'v': 28121be3b31SYaroslav Tykhiy inverse = 1; 28221be3b31SYaroslav Tykhiy break; 28321be3b31SYaroslav Tykhiy case 'x': 28421be3b31SYaroslav Tykhiy fullmatch = 1; 28521be3b31SYaroslav Tykhiy break; 28621be3b31SYaroslav Tykhiy default: 28721be3b31SYaroslav Tykhiy usage(); 28821be3b31SYaroslav Tykhiy /* NOTREACHED */ 28921be3b31SYaroslav Tykhiy } 29021be3b31SYaroslav Tykhiy 29121be3b31SYaroslav Tykhiy argc -= optind; 29221be3b31SYaroslav Tykhiy argv += optind; 29321be3b31SYaroslav Tykhiy if (argc != 0) 29421be3b31SYaroslav Tykhiy criteria = 1; 29521be3b31SYaroslav Tykhiy if (!criteria) 29621be3b31SYaroslav Tykhiy usage(); 29721be3b31SYaroslav Tykhiy if (newest && oldest) 29821be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Options -n and -o are mutually exclusive"); 29921be3b31SYaroslav Tykhiy if (pidfile != NULL) 30021be3b31SYaroslav Tykhiy pidfromfile = takepid(pidfile, pidfilelock); 30121be3b31SYaroslav Tykhiy else { 30221be3b31SYaroslav Tykhiy if (pidfilelock) { 30321be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 30421be3b31SYaroslav Tykhiy "Option -L doesn't make sense without -F"); 30521be3b31SYaroslav Tykhiy } 30621be3b31SYaroslav Tykhiy pidfromfile = -1; 30721be3b31SYaroslav Tykhiy } 30821be3b31SYaroslav Tykhiy 30921be3b31SYaroslav Tykhiy mypid = getpid(); 31021be3b31SYaroslav Tykhiy 31121be3b31SYaroslav Tykhiy /* 31221be3b31SYaroslav Tykhiy * Retrieve the list of running processes from the kernel. 31321be3b31SYaroslav Tykhiy */ 31421be3b31SYaroslav Tykhiy kd = kvm_openfiles(execf, coref, NULL, O_RDONLY, buf); 31521be3b31SYaroslav Tykhiy if (kd == NULL) 31621be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Cannot open kernel files (%s)", buf); 31721be3b31SYaroslav Tykhiy 31821be3b31SYaroslav Tykhiy /* 31921be3b31SYaroslav Tykhiy * Use KERN_PROC_PROC instead of KERN_PROC_ALL, since we 32021be3b31SYaroslav Tykhiy * just want processes and not individual kernel threads. 32121be3b31SYaroslav Tykhiy */ 32203eea902SEitan Adler if (pidfromfile >= 0) 32303eea902SEitan Adler plist = kvm_getprocs(kd, KERN_PROC_PID, pidfromfile, &nproc); 32403eea902SEitan Adler else 32521be3b31SYaroslav Tykhiy plist = kvm_getprocs(kd, KERN_PROC_PROC, 0, &nproc); 32621be3b31SYaroslav Tykhiy if (plist == NULL) { 32721be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Cannot get process list (%s)", 32821be3b31SYaroslav Tykhiy kvm_geterr(kd)); 32921be3b31SYaroslav Tykhiy } 33021be3b31SYaroslav Tykhiy 33121be3b31SYaroslav Tykhiy /* 33221be3b31SYaroslav Tykhiy * Allocate memory which will be used to keep track of the 33321be3b31SYaroslav Tykhiy * selection. 33421be3b31SYaroslav Tykhiy */ 33521be3b31SYaroslav Tykhiy if ((selected = malloc(nproc)) == NULL) { 33621be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot allocate memory for %d processes", 33721be3b31SYaroslav Tykhiy nproc); 33821be3b31SYaroslav Tykhiy } 33921be3b31SYaroslav Tykhiy memset(selected, 0, nproc); 34021be3b31SYaroslav Tykhiy 34121be3b31SYaroslav Tykhiy /* 34221be3b31SYaroslav Tykhiy * Refine the selection. 34321be3b31SYaroslav Tykhiy */ 34421be3b31SYaroslav Tykhiy for (; *argv != NULL; argv++) { 34521be3b31SYaroslav Tykhiy if ((rv = regcomp(®, *argv, cflags)) != 0) { 34621be3b31SYaroslav Tykhiy regerror(rv, ®, buf, sizeof(buf)); 34721be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 34821be3b31SYaroslav Tykhiy "Cannot compile regular expression `%s' (%s)", 34921be3b31SYaroslav Tykhiy *argv, buf); 35021be3b31SYaroslav Tykhiy } 35121be3b31SYaroslav Tykhiy 35221be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 35321be3b31SYaroslav Tykhiy if (PSKIP(kp)) { 35421be3b31SYaroslav Tykhiy if (debug_opt > 0) 35521be3b31SYaroslav Tykhiy fprintf(stderr, "* Skipped %5d %3d %s\n", 35621be3b31SYaroslav Tykhiy kp->ki_pid, kp->ki_uid, kp->ki_comm); 35721be3b31SYaroslav Tykhiy continue; 35821be3b31SYaroslav Tykhiy } 35921be3b31SYaroslav Tykhiy 36021be3b31SYaroslav Tykhiy if (matchargs && 36121be3b31SYaroslav Tykhiy (pargv = kvm_getargv(kd, kp, 0)) != NULL) { 36221be3b31SYaroslav Tykhiy jsz = 0; 36321be3b31SYaroslav Tykhiy while (jsz < sizeof(buf) && *pargv != NULL) { 36421be3b31SYaroslav Tykhiy jsz += snprintf(buf + jsz, 36521be3b31SYaroslav Tykhiy sizeof(buf) - jsz, 36621be3b31SYaroslav Tykhiy pargv[1] != NULL ? "%s " : "%s", 36721be3b31SYaroslav Tykhiy pargv[0]); 36821be3b31SYaroslav Tykhiy pargv++; 36921be3b31SYaroslav Tykhiy } 37021be3b31SYaroslav Tykhiy mstr = buf; 37121be3b31SYaroslav Tykhiy } else 37221be3b31SYaroslav Tykhiy mstr = kp->ki_comm; 37321be3b31SYaroslav Tykhiy 37421be3b31SYaroslav Tykhiy rv = regexec(®, mstr, 1, ®match, 0); 37521be3b31SYaroslav Tykhiy if (rv == 0) { 37621be3b31SYaroslav Tykhiy if (fullmatch) { 37721be3b31SYaroslav Tykhiy if (regmatch.rm_so == 0 && 37821be3b31SYaroslav Tykhiy regmatch.rm_eo == 37921be3b31SYaroslav Tykhiy (off_t)strlen(mstr)) 38021be3b31SYaroslav Tykhiy selected[i] = 1; 38121be3b31SYaroslav Tykhiy } else 38221be3b31SYaroslav Tykhiy selected[i] = 1; 38321be3b31SYaroslav Tykhiy } else if (rv != REG_NOMATCH) { 38421be3b31SYaroslav Tykhiy regerror(rv, ®, buf, sizeof(buf)); 38521be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 38621be3b31SYaroslav Tykhiy "Regular expression evaluation error (%s)", 38721be3b31SYaroslav Tykhiy buf); 38821be3b31SYaroslav Tykhiy } 38921be3b31SYaroslav Tykhiy if (debug_opt > 1) { 39021be3b31SYaroslav Tykhiy const char *rv_res = "NoMatch"; 39121be3b31SYaroslav Tykhiy if (selected[i]) 39221be3b31SYaroslav Tykhiy rv_res = "Matched"; 39321be3b31SYaroslav Tykhiy fprintf(stderr, "* %s %5d %3d %s\n", rv_res, 39421be3b31SYaroslav Tykhiy kp->ki_pid, kp->ki_uid, mstr); 39521be3b31SYaroslav Tykhiy } 39621be3b31SYaroslav Tykhiy } 39721be3b31SYaroslav Tykhiy 39821be3b31SYaroslav Tykhiy regfree(®); 39921be3b31SYaroslav Tykhiy } 40021be3b31SYaroslav Tykhiy 40121be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 40221be3b31SYaroslav Tykhiy if (PSKIP(kp)) 40321be3b31SYaroslav Tykhiy continue; 40421be3b31SYaroslav Tykhiy 40521be3b31SYaroslav Tykhiy if (pidfromfile >= 0 && kp->ki_pid != pidfromfile) { 40621be3b31SYaroslav Tykhiy selected[i] = 0; 40721be3b31SYaroslav Tykhiy continue; 40821be3b31SYaroslav Tykhiy } 40921be3b31SYaroslav Tykhiy 41021be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &ruidlist, li_chain) 41121be3b31SYaroslav Tykhiy if (kp->ki_ruid == (uid_t)li->li_number) 41221be3b31SYaroslav Tykhiy break; 41321be3b31SYaroslav Tykhiy if (SLIST_FIRST(&ruidlist) != NULL && li == NULL) { 41421be3b31SYaroslav Tykhiy selected[i] = 0; 41521be3b31SYaroslav Tykhiy continue; 41621be3b31SYaroslav Tykhiy } 41721be3b31SYaroslav Tykhiy 41821be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &rgidlist, li_chain) 41921be3b31SYaroslav Tykhiy if (kp->ki_rgid == (gid_t)li->li_number) 42021be3b31SYaroslav Tykhiy break; 42121be3b31SYaroslav Tykhiy if (SLIST_FIRST(&rgidlist) != NULL && li == NULL) { 42221be3b31SYaroslav Tykhiy selected[i] = 0; 42321be3b31SYaroslav Tykhiy continue; 42421be3b31SYaroslav Tykhiy } 42521be3b31SYaroslav Tykhiy 42621be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &euidlist, li_chain) 42721be3b31SYaroslav Tykhiy if (kp->ki_uid == (uid_t)li->li_number) 42821be3b31SYaroslav Tykhiy break; 42921be3b31SYaroslav Tykhiy if (SLIST_FIRST(&euidlist) != NULL && li == NULL) { 43021be3b31SYaroslav Tykhiy selected[i] = 0; 43121be3b31SYaroslav Tykhiy continue; 43221be3b31SYaroslav Tykhiy } 43321be3b31SYaroslav Tykhiy 43421be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &ppidlist, li_chain) 43521be3b31SYaroslav Tykhiy if (kp->ki_ppid == (pid_t)li->li_number) 43621be3b31SYaroslav Tykhiy break; 43721be3b31SYaroslav Tykhiy if (SLIST_FIRST(&ppidlist) != NULL && li == NULL) { 43821be3b31SYaroslav Tykhiy selected[i] = 0; 43921be3b31SYaroslav Tykhiy continue; 44021be3b31SYaroslav Tykhiy } 44121be3b31SYaroslav Tykhiy 44221be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &pgrplist, li_chain) 44321be3b31SYaroslav Tykhiy if (kp->ki_pgid == (pid_t)li->li_number) 44421be3b31SYaroslav Tykhiy break; 44521be3b31SYaroslav Tykhiy if (SLIST_FIRST(&pgrplist) != NULL && li == NULL) { 44621be3b31SYaroslav Tykhiy selected[i] = 0; 44721be3b31SYaroslav Tykhiy continue; 44821be3b31SYaroslav Tykhiy } 44921be3b31SYaroslav Tykhiy 45021be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &tdevlist, li_chain) { 45121be3b31SYaroslav Tykhiy if (li->li_number == -1 && 45221be3b31SYaroslav Tykhiy (kp->ki_flag & P_CONTROLT) == 0) 45321be3b31SYaroslav Tykhiy break; 45421be3b31SYaroslav Tykhiy if (kp->ki_tdev == (dev_t)li->li_number) 45521be3b31SYaroslav Tykhiy break; 45621be3b31SYaroslav Tykhiy } 45721be3b31SYaroslav Tykhiy if (SLIST_FIRST(&tdevlist) != NULL && li == NULL) { 45821be3b31SYaroslav Tykhiy selected[i] = 0; 45921be3b31SYaroslav Tykhiy continue; 46021be3b31SYaroslav Tykhiy } 46121be3b31SYaroslav Tykhiy 46221be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &sidlist, li_chain) 46321be3b31SYaroslav Tykhiy if (kp->ki_sid == (pid_t)li->li_number) 46421be3b31SYaroslav Tykhiy break; 46521be3b31SYaroslav Tykhiy if (SLIST_FIRST(&sidlist) != NULL && li == NULL) { 46621be3b31SYaroslav Tykhiy selected[i] = 0; 46721be3b31SYaroslav Tykhiy continue; 46821be3b31SYaroslav Tykhiy } 46921be3b31SYaroslav Tykhiy 47021be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &jidlist, li_chain) { 47121be3b31SYaroslav Tykhiy /* A particular jail ID, including 0 (not in jail) */ 47221be3b31SYaroslav Tykhiy if (kp->ki_jid == (int)li->li_number) 47321be3b31SYaroslav Tykhiy break; 47421be3b31SYaroslav Tykhiy /* Any jail */ 47521be3b31SYaroslav Tykhiy if (kp->ki_jid > 0 && li->li_number == -1) 47621be3b31SYaroslav Tykhiy break; 47721be3b31SYaroslav Tykhiy } 47821be3b31SYaroslav Tykhiy if (SLIST_FIRST(&jidlist) != NULL && li == NULL) { 47921be3b31SYaroslav Tykhiy selected[i] = 0; 48021be3b31SYaroslav Tykhiy continue; 48121be3b31SYaroslav Tykhiy } 48221be3b31SYaroslav Tykhiy 483cc55ad3dSEdward Tomasz Napierala SLIST_FOREACH(li, &classlist, li_chain) { 484cc55ad3dSEdward Tomasz Napierala /* 485cc55ad3dSEdward Tomasz Napierala * We skip P_SYSTEM processes to match ps(1) output. 486cc55ad3dSEdward Tomasz Napierala */ 487cc55ad3dSEdward Tomasz Napierala if ((kp->ki_flag & P_SYSTEM) == 0 && 488cc55ad3dSEdward Tomasz Napierala strcmp(kp->ki_loginclass, li->li_name) == 0) 489cc55ad3dSEdward Tomasz Napierala break; 490cc55ad3dSEdward Tomasz Napierala } 491cc55ad3dSEdward Tomasz Napierala if (SLIST_FIRST(&classlist) != NULL && li == NULL) { 492cc55ad3dSEdward Tomasz Napierala selected[i] = 0; 493cc55ad3dSEdward Tomasz Napierala continue; 494cc55ad3dSEdward Tomasz Napierala } 495cc55ad3dSEdward Tomasz Napierala 49621be3b31SYaroslav Tykhiy if (argc == 0) 49721be3b31SYaroslav Tykhiy selected[i] = 1; 49821be3b31SYaroslav Tykhiy } 49921be3b31SYaroslav Tykhiy 5000a091aebSBrian Somers if (!ancestors) { 5010a091aebSBrian Somers pid = mypid; 5020a091aebSBrian Somers while (pid) { 5030a091aebSBrian Somers for (i = 0, kp = plist; i < nproc; i++, kp++) { 5040a091aebSBrian Somers if (PSKIP(kp)) 5050a091aebSBrian Somers continue; 5060a091aebSBrian Somers if (kp->ki_pid == pid) { 5070a091aebSBrian Somers selected[i] = 0; 5080a091aebSBrian Somers pid = kp->ki_ppid; 5090a091aebSBrian Somers break; 5100a091aebSBrian Somers } 5110a091aebSBrian Somers } 5120a091aebSBrian Somers if (i == nproc) { 5130a091aebSBrian Somers if (pid == mypid) 5140a091aebSBrian Somers pid = getppid(); 5150a091aebSBrian Somers else 5160a091aebSBrian Somers break; /* Maybe we're in a jail ? */ 5170a091aebSBrian Somers } 5180a091aebSBrian Somers } 5190a091aebSBrian Somers } 5200a091aebSBrian Somers 52121be3b31SYaroslav Tykhiy if (newest || oldest) { 52221be3b31SYaroslav Tykhiy best_tval.tv_sec = 0; 52321be3b31SYaroslav Tykhiy best_tval.tv_usec = 0; 52421be3b31SYaroslav Tykhiy bestidx = -1; 52521be3b31SYaroslav Tykhiy 52621be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 52721be3b31SYaroslav Tykhiy if (!selected[i]) 52821be3b31SYaroslav Tykhiy continue; 52921be3b31SYaroslav Tykhiy if (bestidx == -1) { 53021be3b31SYaroslav Tykhiy /* The first entry of the list which matched. */ 53121be3b31SYaroslav Tykhiy ; 53221be3b31SYaroslav Tykhiy } else if (timercmp(&kp->ki_start, &best_tval, >)) { 53321be3b31SYaroslav Tykhiy /* This entry is newer than previous "best". */ 53421be3b31SYaroslav Tykhiy if (oldest) /* but we want the oldest */ 53521be3b31SYaroslav Tykhiy continue; 53621be3b31SYaroslav Tykhiy } else { 53721be3b31SYaroslav Tykhiy /* This entry is older than previous "best". */ 53821be3b31SYaroslav Tykhiy if (newest) /* but we want the newest */ 53921be3b31SYaroslav Tykhiy continue; 54021be3b31SYaroslav Tykhiy } 54121be3b31SYaroslav Tykhiy /* This entry is better than previous "best" entry. */ 54221be3b31SYaroslav Tykhiy best_tval.tv_sec = kp->ki_start.tv_sec; 54321be3b31SYaroslav Tykhiy best_tval.tv_usec = kp->ki_start.tv_usec; 54421be3b31SYaroslav Tykhiy bestidx = i; 54521be3b31SYaroslav Tykhiy } 54621be3b31SYaroslav Tykhiy 54721be3b31SYaroslav Tykhiy memset(selected, 0, nproc); 54821be3b31SYaroslav Tykhiy if (bestidx != -1) 54921be3b31SYaroslav Tykhiy selected[bestidx] = 1; 55021be3b31SYaroslav Tykhiy } 55121be3b31SYaroslav Tykhiy 55221be3b31SYaroslav Tykhiy /* 55321be3b31SYaroslav Tykhiy * Take the appropriate action for each matched process, if any. 55421be3b31SYaroslav Tykhiy */ 5553ba0e470SBrian Somers did_action = 0; 55621be3b31SYaroslav Tykhiy for (i = 0, rv = 0, kp = plist; i < nproc; i++, kp++) { 55721be3b31SYaroslav Tykhiy if (PSKIP(kp)) 55821be3b31SYaroslav Tykhiy continue; 55921be3b31SYaroslav Tykhiy if (selected[i]) { 5603ba0e470SBrian Somers if (longfmt && !pgrep) { 5613ba0e470SBrian Somers did_action = 1; 5623ba0e470SBrian Somers printf("kill -%d %d\n", signum, kp->ki_pid); 5633ba0e470SBrian Somers } 56421be3b31SYaroslav Tykhiy if (inverse) 56521be3b31SYaroslav Tykhiy continue; 56621be3b31SYaroslav Tykhiy } else if (!inverse) 56721be3b31SYaroslav Tykhiy continue; 56821be3b31SYaroslav Tykhiy rv |= (*action)(kp); 56921be3b31SYaroslav Tykhiy } 5703ba0e470SBrian Somers if (!did_action && !pgrep && longfmt) 5713ba0e470SBrian Somers fprintf(stderr, 5723ba0e470SBrian Somers "No matching processes belonging to you were found\n"); 57321be3b31SYaroslav Tykhiy 57421be3b31SYaroslav Tykhiy exit(rv ? STATUS_MATCH : STATUS_NOMATCH); 57521be3b31SYaroslav Tykhiy } 57621be3b31SYaroslav Tykhiy 57721be3b31SYaroslav Tykhiy static void 57821be3b31SYaroslav Tykhiy usage(void) 57921be3b31SYaroslav Tykhiy { 58021be3b31SYaroslav Tykhiy const char *ustr; 58121be3b31SYaroslav Tykhiy 58221be3b31SYaroslav Tykhiy if (pgrep) 5837b0706f6SPawel Jakub Dawidek ustr = "[-LSfilnoqvx] [-d delim]"; 58421be3b31SYaroslav Tykhiy else 5853ba0e470SBrian Somers ustr = "[-signal] [-ILfilnovx]"; 58621be3b31SYaroslav Tykhiy 58721be3b31SYaroslav Tykhiy fprintf(stderr, 58821be3b31SYaroslav Tykhiy "usage: %s %s [-F pidfile] [-G gid] [-M core] [-N system]\n" 589c4f0631fSJamie Gritton " [-P ppid] [-U uid] [-c class] [-g pgrp] [-j jail]\n" 590cc55ad3dSEdward Tomasz Napierala " [-s sid] [-t tty] [-u euid] pattern ...\n", 591cc55ad3dSEdward Tomasz Napierala getprogname(), ustr); 59221be3b31SYaroslav Tykhiy 59321be3b31SYaroslav Tykhiy exit(STATUS_BADUSAGE); 59421be3b31SYaroslav Tykhiy } 59521be3b31SYaroslav Tykhiy 59621be3b31SYaroslav Tykhiy static void 59721be3b31SYaroslav Tykhiy show_process(const struct kinfo_proc *kp) 59821be3b31SYaroslav Tykhiy { 59921be3b31SYaroslav Tykhiy char **argv; 60021be3b31SYaroslav Tykhiy 6017b0706f6SPawel Jakub Dawidek if (quiet) { 6027b0706f6SPawel Jakub Dawidek assert(pgrep); 6037b0706f6SPawel Jakub Dawidek return; 6047b0706f6SPawel Jakub Dawidek } 60521be3b31SYaroslav Tykhiy if ((longfmt || !pgrep) && matchargs && 60621be3b31SYaroslav Tykhiy (argv = kvm_getargv(kd, kp, 0)) != NULL) { 60721be3b31SYaroslav Tykhiy printf("%d ", (int)kp->ki_pid); 60821be3b31SYaroslav Tykhiy for (; *argv != NULL; argv++) { 60921be3b31SYaroslav Tykhiy printf("%s", *argv); 61021be3b31SYaroslav Tykhiy if (argv[1] != NULL) 61121be3b31SYaroslav Tykhiy putchar(' '); 61221be3b31SYaroslav Tykhiy } 61321be3b31SYaroslav Tykhiy } else if (longfmt || !pgrep) 61421be3b31SYaroslav Tykhiy printf("%d %s", (int)kp->ki_pid, kp->ki_comm); 61521be3b31SYaroslav Tykhiy else 61621be3b31SYaroslav Tykhiy printf("%d", (int)kp->ki_pid); 61721be3b31SYaroslav Tykhiy } 61821be3b31SYaroslav Tykhiy 61921be3b31SYaroslav Tykhiy static int 62021be3b31SYaroslav Tykhiy killact(const struct kinfo_proc *kp) 62121be3b31SYaroslav Tykhiy { 62221be3b31SYaroslav Tykhiy int ch, first; 62321be3b31SYaroslav Tykhiy 62421be3b31SYaroslav Tykhiy if (interactive) { 62521be3b31SYaroslav Tykhiy /* 62621be3b31SYaroslav Tykhiy * Be careful, ask before killing. 62721be3b31SYaroslav Tykhiy */ 62821be3b31SYaroslav Tykhiy printf("kill "); 62921be3b31SYaroslav Tykhiy show_process(kp); 63021be3b31SYaroslav Tykhiy printf("? "); 63121be3b31SYaroslav Tykhiy fflush(stdout); 63221be3b31SYaroslav Tykhiy first = ch = getchar(); 63321be3b31SYaroslav Tykhiy while (ch != '\n' && ch != EOF) 63421be3b31SYaroslav Tykhiy ch = getchar(); 63521be3b31SYaroslav Tykhiy if (first != 'y' && first != 'Y') 63621be3b31SYaroslav Tykhiy return (1); 63721be3b31SYaroslav Tykhiy } 63821be3b31SYaroslav Tykhiy if (kill(kp->ki_pid, signum) == -1) { 63921be3b31SYaroslav Tykhiy /* 64021be3b31SYaroslav Tykhiy * Check for ESRCH, which indicates that the process 64121be3b31SYaroslav Tykhiy * disappeared between us matching it and us 64221be3b31SYaroslav Tykhiy * signalling it; don't issue a warning about it. 64321be3b31SYaroslav Tykhiy */ 64421be3b31SYaroslav Tykhiy if (errno != ESRCH) 64521be3b31SYaroslav Tykhiy warn("signalling pid %d", (int)kp->ki_pid); 64621be3b31SYaroslav Tykhiy /* 64721be3b31SYaroslav Tykhiy * Return 0 to indicate that the process should not be 64821be3b31SYaroslav Tykhiy * considered a match, since we didn't actually get to 64921be3b31SYaroslav Tykhiy * signal it. 65021be3b31SYaroslav Tykhiy */ 65121be3b31SYaroslav Tykhiy return (0); 65221be3b31SYaroslav Tykhiy } 65321be3b31SYaroslav Tykhiy return (1); 65421be3b31SYaroslav Tykhiy } 65521be3b31SYaroslav Tykhiy 65621be3b31SYaroslav Tykhiy static int 65721be3b31SYaroslav Tykhiy grepact(const struct kinfo_proc *kp) 65821be3b31SYaroslav Tykhiy { 65921be3b31SYaroslav Tykhiy 66021be3b31SYaroslav Tykhiy show_process(kp); 6617b0706f6SPawel Jakub Dawidek if (!quiet) 66221be3b31SYaroslav Tykhiy printf("%s", delim); 66321be3b31SYaroslav Tykhiy return (1); 66421be3b31SYaroslav Tykhiy } 66521be3b31SYaroslav Tykhiy 66621be3b31SYaroslav Tykhiy static void 66721be3b31SYaroslav Tykhiy makelist(struct listhead *head, enum listtype type, char *src) 66821be3b31SYaroslav Tykhiy { 66921be3b31SYaroslav Tykhiy struct list *li; 67021be3b31SYaroslav Tykhiy struct passwd *pw; 67121be3b31SYaroslav Tykhiy struct group *gr; 67221be3b31SYaroslav Tykhiy struct stat st; 673eb948eacSEd Schouten const char *cp; 67421be3b31SYaroslav Tykhiy char *sp, *ep, buf[MAXPATHLEN]; 67521be3b31SYaroslav Tykhiy int empty; 67621be3b31SYaroslav Tykhiy 67721be3b31SYaroslav Tykhiy empty = 1; 67821be3b31SYaroslav Tykhiy 67921be3b31SYaroslav Tykhiy while ((sp = strsep(&src, ",")) != NULL) { 68021be3b31SYaroslav Tykhiy if (*sp == '\0') 68121be3b31SYaroslav Tykhiy usage(); 68221be3b31SYaroslav Tykhiy 68321be3b31SYaroslav Tykhiy if ((li = malloc(sizeof(*li))) == NULL) { 68421be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot allocate %zu bytes", 68521be3b31SYaroslav Tykhiy sizeof(*li)); 68621be3b31SYaroslav Tykhiy } 68721be3b31SYaroslav Tykhiy 68821be3b31SYaroslav Tykhiy SLIST_INSERT_HEAD(head, li, li_chain); 68921be3b31SYaroslav Tykhiy empty = 0; 69021be3b31SYaroslav Tykhiy 691cc55ad3dSEdward Tomasz Napierala if (type != LT_CLASS) 69221be3b31SYaroslav Tykhiy li->li_number = (uid_t)strtol(sp, &ep, 0); 693cc55ad3dSEdward Tomasz Napierala 694cc55ad3dSEdward Tomasz Napierala if (type != LT_CLASS && *ep == '\0') { 69521be3b31SYaroslav Tykhiy switch (type) { 69621be3b31SYaroslav Tykhiy case LT_PGRP: 69721be3b31SYaroslav Tykhiy if (li->li_number == 0) 69821be3b31SYaroslav Tykhiy li->li_number = getpgrp(); 69921be3b31SYaroslav Tykhiy break; 70021be3b31SYaroslav Tykhiy case LT_SID: 70121be3b31SYaroslav Tykhiy if (li->li_number == 0) 70221be3b31SYaroslav Tykhiy li->li_number = getsid(mypid); 70321be3b31SYaroslav Tykhiy break; 704c4f0631fSJamie Gritton case LT_JAIL: 70521be3b31SYaroslav Tykhiy if (li->li_number < 0) 70621be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 70721be3b31SYaroslav Tykhiy "Negative jail ID `%s'", sp); 70821be3b31SYaroslav Tykhiy /* For compatibility with old -j */ 70921be3b31SYaroslav Tykhiy if (li->li_number == 0) 71021be3b31SYaroslav Tykhiy li->li_number = -1; /* any jail */ 71121be3b31SYaroslav Tykhiy break; 7122a546db8SDavid E. O'Brien case LT_TTY: 7132a546db8SDavid E. O'Brien if (li->li_number < 0) 7142a546db8SDavid E. O'Brien errx(STATUS_BADUSAGE, 7152a546db8SDavid E. O'Brien "Negative /dev/pts tty `%s'", sp); 7162a546db8SDavid E. O'Brien snprintf(buf, sizeof(buf), _PATH_DEV "pts/%s", 7172a546db8SDavid E. O'Brien sp); 7182a546db8SDavid E. O'Brien if (stat(buf, &st) != -1) 7192a546db8SDavid E. O'Brien goto foundtty; 7202a546db8SDavid E. O'Brien if (errno == ENOENT) 7212a546db8SDavid E. O'Brien errx(STATUS_BADUSAGE, "No such tty: `" 7222a546db8SDavid E. O'Brien _PATH_DEV "pts/%s'", sp); 7232a546db8SDavid E. O'Brien err(STATUS_ERROR, "Cannot access `" 7242a546db8SDavid E. O'Brien _PATH_DEV "pts/%s'", sp); 7252a546db8SDavid E. O'Brien break; 72621be3b31SYaroslav Tykhiy default: 72721be3b31SYaroslav Tykhiy break; 72821be3b31SYaroslav Tykhiy } 72921be3b31SYaroslav Tykhiy continue; 73021be3b31SYaroslav Tykhiy } 73121be3b31SYaroslav Tykhiy 73221be3b31SYaroslav Tykhiy switch (type) { 73321be3b31SYaroslav Tykhiy case LT_USER: 73421be3b31SYaroslav Tykhiy if ((pw = getpwnam(sp)) == NULL) 73521be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Unknown user `%s'", sp); 73621be3b31SYaroslav Tykhiy li->li_number = pw->pw_uid; 73721be3b31SYaroslav Tykhiy break; 73821be3b31SYaroslav Tykhiy case LT_GROUP: 73921be3b31SYaroslav Tykhiy if ((gr = getgrnam(sp)) == NULL) 74021be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Unknown group `%s'", sp); 74121be3b31SYaroslav Tykhiy li->li_number = gr->gr_gid; 74221be3b31SYaroslav Tykhiy break; 74321be3b31SYaroslav Tykhiy case LT_TTY: 74421be3b31SYaroslav Tykhiy if (strcmp(sp, "-") == 0) { 74521be3b31SYaroslav Tykhiy li->li_number = -1; 74621be3b31SYaroslav Tykhiy break; 74721be3b31SYaroslav Tykhiy } else if (strcmp(sp, "co") == 0) { 74821be3b31SYaroslav Tykhiy cp = "console"; 74921be3b31SYaroslav Tykhiy } else { 75021be3b31SYaroslav Tykhiy cp = sp; 75121be3b31SYaroslav Tykhiy } 75221be3b31SYaroslav Tykhiy 753eb948eacSEd Schouten snprintf(buf, sizeof(buf), _PATH_DEV "%s", cp); 75418aa158fSEd Schouten if (stat(buf, &st) != -1) 75518aa158fSEd Schouten goto foundtty; 75621be3b31SYaroslav Tykhiy 75718aa158fSEd Schouten snprintf(buf, sizeof(buf), _PATH_DEV "tty%s", cp); 75818aa158fSEd Schouten if (stat(buf, &st) != -1) 75918aa158fSEd Schouten goto foundtty; 76018aa158fSEd Schouten 76118aa158fSEd Schouten if (errno == ENOENT) 76218aa158fSEd Schouten errx(STATUS_BADUSAGE, "No such tty: `%s'", sp); 76321be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot access `%s'", sp); 76421be3b31SYaroslav Tykhiy 76518aa158fSEd Schouten foundtty: if ((st.st_mode & S_IFCHR) == 0) 76621be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Not a tty: `%s'", sp); 76721be3b31SYaroslav Tykhiy 76821be3b31SYaroslav Tykhiy li->li_number = st.st_rdev; 76921be3b31SYaroslav Tykhiy break; 770c4f0631fSJamie Gritton case LT_JAIL: { 771c4f0631fSJamie Gritton int jid; 772c4f0631fSJamie Gritton 77321be3b31SYaroslav Tykhiy if (strcmp(sp, "none") == 0) 77421be3b31SYaroslav Tykhiy li->li_number = 0; 77521be3b31SYaroslav Tykhiy else if (strcmp(sp, "any") == 0) 77621be3b31SYaroslav Tykhiy li->li_number = -1; 777c4f0631fSJamie Gritton else if ((jid = jail_getid(sp)) != -1) 778c4f0631fSJamie Gritton li->li_number = jid; 77921be3b31SYaroslav Tykhiy else if (*ep != '\0') 78021be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 781c4f0631fSJamie Gritton "Invalid jail ID or name `%s'", sp); 78221be3b31SYaroslav Tykhiy break; 783c4f0631fSJamie Gritton } 784cc55ad3dSEdward Tomasz Napierala case LT_CLASS: 785cc55ad3dSEdward Tomasz Napierala li->li_number = -1; 786cc55ad3dSEdward Tomasz Napierala li->li_name = strdup(sp); 787cc55ad3dSEdward Tomasz Napierala if (li->li_name == NULL) 788cc55ad3dSEdward Tomasz Napierala err(STATUS_ERROR, "Cannot allocate memory"); 789cc55ad3dSEdward Tomasz Napierala break; 79021be3b31SYaroslav Tykhiy default: 79121be3b31SYaroslav Tykhiy usage(); 79221be3b31SYaroslav Tykhiy } 79321be3b31SYaroslav Tykhiy } 79421be3b31SYaroslav Tykhiy 79521be3b31SYaroslav Tykhiy if (empty) 79621be3b31SYaroslav Tykhiy usage(); 79721be3b31SYaroslav Tykhiy } 79821be3b31SYaroslav Tykhiy 79921be3b31SYaroslav Tykhiy static int 80021be3b31SYaroslav Tykhiy takepid(const char *pidfile, int pidfilelock) 80121be3b31SYaroslav Tykhiy { 80221be3b31SYaroslav Tykhiy char *endp, line[BUFSIZ]; 80321be3b31SYaroslav Tykhiy FILE *fh; 80421be3b31SYaroslav Tykhiy long rval; 80521be3b31SYaroslav Tykhiy 80621be3b31SYaroslav Tykhiy fh = fopen(pidfile, "r"); 80721be3b31SYaroslav Tykhiy if (fh == NULL) 80821be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot open pidfile `%s'", pidfile); 80921be3b31SYaroslav Tykhiy 81021be3b31SYaroslav Tykhiy if (pidfilelock) { 81121be3b31SYaroslav Tykhiy /* 81221be3b31SYaroslav Tykhiy * If we can lock pidfile, this means that daemon is not 81321be3b31SYaroslav Tykhiy * running, so would be better not to kill some random process. 81421be3b31SYaroslav Tykhiy */ 81521be3b31SYaroslav Tykhiy if (flock(fileno(fh), LOCK_EX | LOCK_NB) == 0) { 81621be3b31SYaroslav Tykhiy (void)fclose(fh); 81721be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "File '%s' can be locked", pidfile); 81821be3b31SYaroslav Tykhiy } else { 81921be3b31SYaroslav Tykhiy if (errno != EWOULDBLOCK) { 82021be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 82121be3b31SYaroslav Tykhiy "Error while locking file '%s'", pidfile); 82221be3b31SYaroslav Tykhiy } 82321be3b31SYaroslav Tykhiy } 82421be3b31SYaroslav Tykhiy } 82521be3b31SYaroslav Tykhiy 82621be3b31SYaroslav Tykhiy if (fgets(line, sizeof(line), fh) == NULL) { 82721be3b31SYaroslav Tykhiy if (feof(fh)) { 82821be3b31SYaroslav Tykhiy (void)fclose(fh); 82921be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Pidfile `%s' is empty", pidfile); 83021be3b31SYaroslav Tykhiy } 83121be3b31SYaroslav Tykhiy (void)fclose(fh); 83221be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot read from pid file `%s'", pidfile); 83321be3b31SYaroslav Tykhiy } 83421be3b31SYaroslav Tykhiy (void)fclose(fh); 83521be3b31SYaroslav Tykhiy 83621be3b31SYaroslav Tykhiy rval = strtol(line, &endp, 10); 83721be3b31SYaroslav Tykhiy if (*endp != '\0' && !isspace((unsigned char)*endp)) 83821be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile); 83921be3b31SYaroslav Tykhiy else if (rval < MIN_PID || rval > MAX_PID) 84021be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile); 84121be3b31SYaroslav Tykhiy return (rval); 84221be3b31SYaroslav Tykhiy } 843