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> 6221be3b31SYaroslav Tykhiy 6321be3b31SYaroslav Tykhiy #define STATUS_MATCH 0 6421be3b31SYaroslav Tykhiy #define STATUS_NOMATCH 1 6521be3b31SYaroslav Tykhiy #define STATUS_BADUSAGE 2 6621be3b31SYaroslav Tykhiy #define STATUS_ERROR 3 6721be3b31SYaroslav Tykhiy 6821be3b31SYaroslav Tykhiy #define MIN_PID 5 6921be3b31SYaroslav Tykhiy #define MAX_PID 99999 7021be3b31SYaroslav Tykhiy 7121be3b31SYaroslav Tykhiy /* Ignore system-processes (if '-S' flag is not specified) and myself. */ 7221be3b31SYaroslav Tykhiy #define PSKIP(kp) ((kp)->ki_pid == mypid || \ 7321be3b31SYaroslav Tykhiy (!kthreads && ((kp)->ki_flag & P_KTHREAD) != 0)) 7421be3b31SYaroslav Tykhiy 7521be3b31SYaroslav Tykhiy enum listtype { 7621be3b31SYaroslav Tykhiy LT_GENERIC, 7721be3b31SYaroslav Tykhiy LT_USER, 7821be3b31SYaroslav Tykhiy LT_GROUP, 7921be3b31SYaroslav Tykhiy LT_TTY, 8021be3b31SYaroslav Tykhiy LT_PGRP, 8121be3b31SYaroslav Tykhiy LT_JID, 82*cc55ad3dSEdward Tomasz Napierala LT_SID, 83*cc55ad3dSEdward Tomasz Napierala LT_CLASS 8421be3b31SYaroslav Tykhiy }; 8521be3b31SYaroslav Tykhiy 8621be3b31SYaroslav Tykhiy struct list { 8721be3b31SYaroslav Tykhiy SLIST_ENTRY(list) li_chain; 8821be3b31SYaroslav Tykhiy long li_number; 89*cc55ad3dSEdward Tomasz Napierala char *li_name; 9021be3b31SYaroslav Tykhiy }; 9121be3b31SYaroslav Tykhiy 9221be3b31SYaroslav Tykhiy SLIST_HEAD(listhead, list); 9321be3b31SYaroslav Tykhiy 9421be3b31SYaroslav Tykhiy static struct kinfo_proc *plist; 9521be3b31SYaroslav Tykhiy static char *selected; 9621be3b31SYaroslav Tykhiy static const char *delim = "\n"; 9721be3b31SYaroslav Tykhiy static int nproc; 9821be3b31SYaroslav Tykhiy static int pgrep; 9921be3b31SYaroslav Tykhiy static int signum = SIGTERM; 10021be3b31SYaroslav Tykhiy static int newest; 10121be3b31SYaroslav Tykhiy static int oldest; 10221be3b31SYaroslav Tykhiy static int interactive; 10321be3b31SYaroslav Tykhiy static int inverse; 10421be3b31SYaroslav Tykhiy static int longfmt; 10521be3b31SYaroslav Tykhiy static int matchargs; 10621be3b31SYaroslav Tykhiy static int fullmatch; 10721be3b31SYaroslav Tykhiy static int kthreads; 10821be3b31SYaroslav Tykhiy static int cflags = REG_EXTENDED; 1097b0706f6SPawel Jakub Dawidek static int quiet; 11021be3b31SYaroslav Tykhiy static kvm_t *kd; 11121be3b31SYaroslav Tykhiy static pid_t mypid; 11221be3b31SYaroslav Tykhiy 11313e403fdSAntoine Brodin static struct listhead euidlist = SLIST_HEAD_INITIALIZER(euidlist); 11413e403fdSAntoine Brodin static struct listhead ruidlist = SLIST_HEAD_INITIALIZER(ruidlist); 11513e403fdSAntoine Brodin static struct listhead rgidlist = SLIST_HEAD_INITIALIZER(rgidlist); 11613e403fdSAntoine Brodin static struct listhead pgrplist = SLIST_HEAD_INITIALIZER(pgrplist); 11713e403fdSAntoine Brodin static struct listhead ppidlist = SLIST_HEAD_INITIALIZER(ppidlist); 11813e403fdSAntoine Brodin static struct listhead tdevlist = SLIST_HEAD_INITIALIZER(tdevlist); 11913e403fdSAntoine Brodin static struct listhead sidlist = SLIST_HEAD_INITIALIZER(sidlist); 12013e403fdSAntoine Brodin static struct listhead jidlist = SLIST_HEAD_INITIALIZER(jidlist); 121*cc55ad3dSEdward Tomasz Napierala static struct listhead classlist = SLIST_HEAD_INITIALIZER(classlist); 12221be3b31SYaroslav Tykhiy 12321be3b31SYaroslav Tykhiy static void usage(void) __attribute__((__noreturn__)); 12421be3b31SYaroslav Tykhiy static int killact(const struct kinfo_proc *); 12521be3b31SYaroslav Tykhiy static int grepact(const struct kinfo_proc *); 12621be3b31SYaroslav Tykhiy static void makelist(struct listhead *, enum listtype, char *); 12721be3b31SYaroslav Tykhiy static int takepid(const char *, int); 12821be3b31SYaroslav Tykhiy 12921be3b31SYaroslav Tykhiy int 13021be3b31SYaroslav Tykhiy main(int argc, char **argv) 13121be3b31SYaroslav Tykhiy { 13221be3b31SYaroslav Tykhiy char buf[_POSIX2_LINE_MAX], *mstr, **pargv, *p, *q, *pidfile; 13321be3b31SYaroslav Tykhiy const char *execf, *coref; 1343ba0e470SBrian Somers int ancestors, debug_opt, did_action; 13521be3b31SYaroslav Tykhiy int i, ch, bestidx, rv, criteria, pidfromfile, pidfilelock; 13621be3b31SYaroslav Tykhiy size_t jsz; 13721be3b31SYaroslav Tykhiy int (*action)(const struct kinfo_proc *); 13821be3b31SYaroslav Tykhiy struct kinfo_proc *kp; 13921be3b31SYaroslav Tykhiy struct list *li; 14021be3b31SYaroslav Tykhiy struct timeval best_tval; 14121be3b31SYaroslav Tykhiy regex_t reg; 14221be3b31SYaroslav Tykhiy regmatch_t regmatch; 1430a091aebSBrian Somers pid_t pid; 14421be3b31SYaroslav Tykhiy 14521be3b31SYaroslav Tykhiy setlocale(LC_ALL, ""); 14621be3b31SYaroslav Tykhiy 14721be3b31SYaroslav Tykhiy if (strcmp(getprogname(), "pgrep") == 0) { 14821be3b31SYaroslav Tykhiy action = grepact; 14921be3b31SYaroslav Tykhiy pgrep = 1; 15021be3b31SYaroslav Tykhiy } else { 15121be3b31SYaroslav Tykhiy action = killact; 15221be3b31SYaroslav Tykhiy p = argv[1]; 15321be3b31SYaroslav Tykhiy 15421be3b31SYaroslav Tykhiy if (argc > 1 && p[0] == '-') { 15521be3b31SYaroslav Tykhiy p++; 15621be3b31SYaroslav Tykhiy i = (int)strtol(p, &q, 10); 15721be3b31SYaroslav Tykhiy if (*q == '\0') { 15821be3b31SYaroslav Tykhiy signum = i; 15921be3b31SYaroslav Tykhiy argv++; 16021be3b31SYaroslav Tykhiy argc--; 16121be3b31SYaroslav Tykhiy } else { 16212dacf62SJilles Tjoelker if (strncasecmp(p, "SIG", 3) == 0) 16321be3b31SYaroslav Tykhiy p += 3; 16421be3b31SYaroslav Tykhiy for (i = 1; i < NSIG; i++) 16521be3b31SYaroslav Tykhiy if (strcasecmp(sys_signame[i], p) == 0) 16621be3b31SYaroslav Tykhiy break; 16721be3b31SYaroslav Tykhiy if (i != NSIG) { 16821be3b31SYaroslav Tykhiy signum = i; 16921be3b31SYaroslav Tykhiy argv++; 17021be3b31SYaroslav Tykhiy argc--; 17121be3b31SYaroslav Tykhiy } 17221be3b31SYaroslav Tykhiy } 17321be3b31SYaroslav Tykhiy } 17421be3b31SYaroslav Tykhiy } 17521be3b31SYaroslav Tykhiy 1760a091aebSBrian Somers ancestors = 0; 17721be3b31SYaroslav Tykhiy criteria = 0; 17821be3b31SYaroslav Tykhiy debug_opt = 0; 17921be3b31SYaroslav Tykhiy pidfile = NULL; 18021be3b31SYaroslav Tykhiy pidfilelock = 0; 1817b0706f6SPawel Jakub Dawidek quiet = 0; 182c08dcaf1SRebecca Cran execf = NULL; 183c08dcaf1SRebecca Cran coref = _PATH_DEVNULL; 18421be3b31SYaroslav Tykhiy 185*cc55ad3dSEdward Tomasz Napierala while ((ch = getopt(argc, argv, "DF:G:ILM:N:P:SU:ac:d:fg:ij:lnoqs:t:u:vx")) != -1) 18621be3b31SYaroslav Tykhiy switch (ch) { 18721be3b31SYaroslav Tykhiy case 'D': 18821be3b31SYaroslav Tykhiy debug_opt++; 18921be3b31SYaroslav Tykhiy break; 19021be3b31SYaroslav Tykhiy case 'F': 19121be3b31SYaroslav Tykhiy pidfile = optarg; 19221be3b31SYaroslav Tykhiy criteria = 1; 19321be3b31SYaroslav Tykhiy break; 19421be3b31SYaroslav Tykhiy case 'G': 19521be3b31SYaroslav Tykhiy makelist(&rgidlist, LT_GROUP, optarg); 19621be3b31SYaroslav Tykhiy criteria = 1; 19721be3b31SYaroslav Tykhiy break; 19821be3b31SYaroslav Tykhiy case 'I': 19921be3b31SYaroslav Tykhiy if (pgrep) 20021be3b31SYaroslav Tykhiy usage(); 20121be3b31SYaroslav Tykhiy interactive = 1; 20221be3b31SYaroslav Tykhiy break; 20321be3b31SYaroslav Tykhiy case 'L': 20421be3b31SYaroslav Tykhiy pidfilelock = 1; 20521be3b31SYaroslav Tykhiy break; 20621be3b31SYaroslav Tykhiy case 'M': 20721be3b31SYaroslav Tykhiy coref = optarg; 20821be3b31SYaroslav Tykhiy break; 20921be3b31SYaroslav Tykhiy case 'N': 21021be3b31SYaroslav Tykhiy execf = optarg; 21121be3b31SYaroslav Tykhiy break; 21221be3b31SYaroslav Tykhiy case 'P': 21321be3b31SYaroslav Tykhiy makelist(&ppidlist, LT_GENERIC, optarg); 21421be3b31SYaroslav Tykhiy criteria = 1; 21521be3b31SYaroslav Tykhiy break; 21621be3b31SYaroslav Tykhiy case 'S': 21721be3b31SYaroslav Tykhiy if (!pgrep) 21821be3b31SYaroslav Tykhiy usage(); 21921be3b31SYaroslav Tykhiy kthreads = 1; 22021be3b31SYaroslav Tykhiy break; 22121be3b31SYaroslav Tykhiy case 'U': 22221be3b31SYaroslav Tykhiy makelist(&ruidlist, LT_USER, optarg); 22321be3b31SYaroslav Tykhiy criteria = 1; 22421be3b31SYaroslav Tykhiy break; 2250a091aebSBrian Somers case 'a': 2260a091aebSBrian Somers ancestors++; 2270a091aebSBrian Somers break; 228*cc55ad3dSEdward Tomasz Napierala case 'c': 229*cc55ad3dSEdward Tomasz Napierala makelist(&classlist, LT_CLASS, optarg); 230*cc55ad3dSEdward Tomasz Napierala criteria = 1; 231*cc55ad3dSEdward Tomasz Napierala break; 23221be3b31SYaroslav Tykhiy case 'd': 23321be3b31SYaroslav Tykhiy if (!pgrep) 23421be3b31SYaroslav Tykhiy usage(); 23521be3b31SYaroslav Tykhiy delim = optarg; 23621be3b31SYaroslav Tykhiy break; 23721be3b31SYaroslav Tykhiy case 'f': 23821be3b31SYaroslav Tykhiy matchargs = 1; 23921be3b31SYaroslav Tykhiy break; 24021be3b31SYaroslav Tykhiy case 'g': 24121be3b31SYaroslav Tykhiy makelist(&pgrplist, LT_PGRP, optarg); 24221be3b31SYaroslav Tykhiy criteria = 1; 24321be3b31SYaroslav Tykhiy break; 24421be3b31SYaroslav Tykhiy case 'i': 24521be3b31SYaroslav Tykhiy cflags |= REG_ICASE; 24621be3b31SYaroslav Tykhiy break; 24721be3b31SYaroslav Tykhiy case 'j': 24821be3b31SYaroslav Tykhiy makelist(&jidlist, LT_JID, optarg); 24921be3b31SYaroslav Tykhiy criteria = 1; 25021be3b31SYaroslav Tykhiy break; 25121be3b31SYaroslav Tykhiy case 'l': 25221be3b31SYaroslav Tykhiy longfmt = 1; 25321be3b31SYaroslav Tykhiy break; 25421be3b31SYaroslav Tykhiy case 'n': 25521be3b31SYaroslav Tykhiy newest = 1; 25621be3b31SYaroslav Tykhiy criteria = 1; 25721be3b31SYaroslav Tykhiy break; 25821be3b31SYaroslav Tykhiy case 'o': 25921be3b31SYaroslav Tykhiy oldest = 1; 26021be3b31SYaroslav Tykhiy criteria = 1; 26121be3b31SYaroslav Tykhiy break; 2627b0706f6SPawel Jakub Dawidek case 'q': 2637b0706f6SPawel Jakub Dawidek if (!pgrep) 2647b0706f6SPawel Jakub Dawidek usage(); 2657b0706f6SPawel Jakub Dawidek quiet = 1; 2667b0706f6SPawel Jakub Dawidek break; 26721be3b31SYaroslav Tykhiy case 's': 26821be3b31SYaroslav Tykhiy makelist(&sidlist, LT_SID, optarg); 26921be3b31SYaroslav Tykhiy criteria = 1; 27021be3b31SYaroslav Tykhiy break; 27121be3b31SYaroslav Tykhiy case 't': 27221be3b31SYaroslav Tykhiy makelist(&tdevlist, LT_TTY, optarg); 27321be3b31SYaroslav Tykhiy criteria = 1; 27421be3b31SYaroslav Tykhiy break; 27521be3b31SYaroslav Tykhiy case 'u': 27621be3b31SYaroslav Tykhiy makelist(&euidlist, LT_USER, optarg); 27721be3b31SYaroslav Tykhiy criteria = 1; 27821be3b31SYaroslav Tykhiy break; 27921be3b31SYaroslav Tykhiy case 'v': 28021be3b31SYaroslav Tykhiy inverse = 1; 28121be3b31SYaroslav Tykhiy break; 28221be3b31SYaroslav Tykhiy case 'x': 28321be3b31SYaroslav Tykhiy fullmatch = 1; 28421be3b31SYaroslav Tykhiy break; 28521be3b31SYaroslav Tykhiy default: 28621be3b31SYaroslav Tykhiy usage(); 28721be3b31SYaroslav Tykhiy /* NOTREACHED */ 28821be3b31SYaroslav Tykhiy } 28921be3b31SYaroslav Tykhiy 29021be3b31SYaroslav Tykhiy argc -= optind; 29121be3b31SYaroslav Tykhiy argv += optind; 29221be3b31SYaroslav Tykhiy if (argc != 0) 29321be3b31SYaroslav Tykhiy criteria = 1; 29421be3b31SYaroslav Tykhiy if (!criteria) 29521be3b31SYaroslav Tykhiy usage(); 29621be3b31SYaroslav Tykhiy if (newest && oldest) 29721be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Options -n and -o are mutually exclusive"); 29821be3b31SYaroslav Tykhiy if (pidfile != NULL) 29921be3b31SYaroslav Tykhiy pidfromfile = takepid(pidfile, pidfilelock); 30021be3b31SYaroslav Tykhiy else { 30121be3b31SYaroslav Tykhiy if (pidfilelock) { 30221be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 30321be3b31SYaroslav Tykhiy "Option -L doesn't make sense without -F"); 30421be3b31SYaroslav Tykhiy } 30521be3b31SYaroslav Tykhiy pidfromfile = -1; 30621be3b31SYaroslav Tykhiy } 30721be3b31SYaroslav Tykhiy 30821be3b31SYaroslav Tykhiy mypid = getpid(); 30921be3b31SYaroslav Tykhiy 31021be3b31SYaroslav Tykhiy /* 31121be3b31SYaroslav Tykhiy * Retrieve the list of running processes from the kernel. 31221be3b31SYaroslav Tykhiy */ 31321be3b31SYaroslav Tykhiy kd = kvm_openfiles(execf, coref, NULL, O_RDONLY, buf); 31421be3b31SYaroslav Tykhiy if (kd == NULL) 31521be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Cannot open kernel files (%s)", buf); 31621be3b31SYaroslav Tykhiy 31721be3b31SYaroslav Tykhiy /* 31821be3b31SYaroslav Tykhiy * Use KERN_PROC_PROC instead of KERN_PROC_ALL, since we 31921be3b31SYaroslav Tykhiy * just want processes and not individual kernel threads. 32021be3b31SYaroslav Tykhiy */ 32121be3b31SYaroslav Tykhiy plist = kvm_getprocs(kd, KERN_PROC_PROC, 0, &nproc); 32221be3b31SYaroslav Tykhiy if (plist == NULL) { 32321be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Cannot get process list (%s)", 32421be3b31SYaroslav Tykhiy kvm_geterr(kd)); 32521be3b31SYaroslav Tykhiy } 32621be3b31SYaroslav Tykhiy 32721be3b31SYaroslav Tykhiy /* 32821be3b31SYaroslav Tykhiy * Allocate memory which will be used to keep track of the 32921be3b31SYaroslav Tykhiy * selection. 33021be3b31SYaroslav Tykhiy */ 33121be3b31SYaroslav Tykhiy if ((selected = malloc(nproc)) == NULL) { 33221be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot allocate memory for %d processes", 33321be3b31SYaroslav Tykhiy nproc); 33421be3b31SYaroslav Tykhiy } 33521be3b31SYaroslav Tykhiy memset(selected, 0, nproc); 33621be3b31SYaroslav Tykhiy 33721be3b31SYaroslav Tykhiy /* 33821be3b31SYaroslav Tykhiy * Refine the selection. 33921be3b31SYaroslav Tykhiy */ 34021be3b31SYaroslav Tykhiy for (; *argv != NULL; argv++) { 34121be3b31SYaroslav Tykhiy if ((rv = regcomp(®, *argv, cflags)) != 0) { 34221be3b31SYaroslav Tykhiy regerror(rv, ®, buf, sizeof(buf)); 34321be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 34421be3b31SYaroslav Tykhiy "Cannot compile regular expression `%s' (%s)", 34521be3b31SYaroslav Tykhiy *argv, buf); 34621be3b31SYaroslav Tykhiy } 34721be3b31SYaroslav Tykhiy 34821be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 34921be3b31SYaroslav Tykhiy if (PSKIP(kp)) { 35021be3b31SYaroslav Tykhiy if (debug_opt > 0) 35121be3b31SYaroslav Tykhiy fprintf(stderr, "* Skipped %5d %3d %s\n", 35221be3b31SYaroslav Tykhiy kp->ki_pid, kp->ki_uid, kp->ki_comm); 35321be3b31SYaroslav Tykhiy continue; 35421be3b31SYaroslav Tykhiy } 35521be3b31SYaroslav Tykhiy 35621be3b31SYaroslav Tykhiy if (matchargs && 35721be3b31SYaroslav Tykhiy (pargv = kvm_getargv(kd, kp, 0)) != NULL) { 35821be3b31SYaroslav Tykhiy jsz = 0; 35921be3b31SYaroslav Tykhiy while (jsz < sizeof(buf) && *pargv != NULL) { 36021be3b31SYaroslav Tykhiy jsz += snprintf(buf + jsz, 36121be3b31SYaroslav Tykhiy sizeof(buf) - jsz, 36221be3b31SYaroslav Tykhiy pargv[1] != NULL ? "%s " : "%s", 36321be3b31SYaroslav Tykhiy pargv[0]); 36421be3b31SYaroslav Tykhiy pargv++; 36521be3b31SYaroslav Tykhiy } 36621be3b31SYaroslav Tykhiy mstr = buf; 36721be3b31SYaroslav Tykhiy } else 36821be3b31SYaroslav Tykhiy mstr = kp->ki_comm; 36921be3b31SYaroslav Tykhiy 37021be3b31SYaroslav Tykhiy rv = regexec(®, mstr, 1, ®match, 0); 37121be3b31SYaroslav Tykhiy if (rv == 0) { 37221be3b31SYaroslav Tykhiy if (fullmatch) { 37321be3b31SYaroslav Tykhiy if (regmatch.rm_so == 0 && 37421be3b31SYaroslav Tykhiy regmatch.rm_eo == 37521be3b31SYaroslav Tykhiy (off_t)strlen(mstr)) 37621be3b31SYaroslav Tykhiy selected[i] = 1; 37721be3b31SYaroslav Tykhiy } else 37821be3b31SYaroslav Tykhiy selected[i] = 1; 37921be3b31SYaroslav Tykhiy } else if (rv != REG_NOMATCH) { 38021be3b31SYaroslav Tykhiy regerror(rv, ®, buf, sizeof(buf)); 38121be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 38221be3b31SYaroslav Tykhiy "Regular expression evaluation error (%s)", 38321be3b31SYaroslav Tykhiy buf); 38421be3b31SYaroslav Tykhiy } 38521be3b31SYaroslav Tykhiy if (debug_opt > 1) { 38621be3b31SYaroslav Tykhiy const char *rv_res = "NoMatch"; 38721be3b31SYaroslav Tykhiy if (selected[i]) 38821be3b31SYaroslav Tykhiy rv_res = "Matched"; 38921be3b31SYaroslav Tykhiy fprintf(stderr, "* %s %5d %3d %s\n", rv_res, 39021be3b31SYaroslav Tykhiy kp->ki_pid, kp->ki_uid, mstr); 39121be3b31SYaroslav Tykhiy } 39221be3b31SYaroslav Tykhiy } 39321be3b31SYaroslav Tykhiy 39421be3b31SYaroslav Tykhiy regfree(®); 39521be3b31SYaroslav Tykhiy } 39621be3b31SYaroslav Tykhiy 39721be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 39821be3b31SYaroslav Tykhiy if (PSKIP(kp)) 39921be3b31SYaroslav Tykhiy continue; 40021be3b31SYaroslav Tykhiy 40121be3b31SYaroslav Tykhiy if (pidfromfile >= 0 && kp->ki_pid != pidfromfile) { 40221be3b31SYaroslav Tykhiy selected[i] = 0; 40321be3b31SYaroslav Tykhiy continue; 40421be3b31SYaroslav Tykhiy } 40521be3b31SYaroslav Tykhiy 40621be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &ruidlist, li_chain) 40721be3b31SYaroslav Tykhiy if (kp->ki_ruid == (uid_t)li->li_number) 40821be3b31SYaroslav Tykhiy break; 40921be3b31SYaroslav Tykhiy if (SLIST_FIRST(&ruidlist) != NULL && li == NULL) { 41021be3b31SYaroslav Tykhiy selected[i] = 0; 41121be3b31SYaroslav Tykhiy continue; 41221be3b31SYaroslav Tykhiy } 41321be3b31SYaroslav Tykhiy 41421be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &rgidlist, li_chain) 41521be3b31SYaroslav Tykhiy if (kp->ki_rgid == (gid_t)li->li_number) 41621be3b31SYaroslav Tykhiy break; 41721be3b31SYaroslav Tykhiy if (SLIST_FIRST(&rgidlist) != NULL && li == NULL) { 41821be3b31SYaroslav Tykhiy selected[i] = 0; 41921be3b31SYaroslav Tykhiy continue; 42021be3b31SYaroslav Tykhiy } 42121be3b31SYaroslav Tykhiy 42221be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &euidlist, li_chain) 42321be3b31SYaroslav Tykhiy if (kp->ki_uid == (uid_t)li->li_number) 42421be3b31SYaroslav Tykhiy break; 42521be3b31SYaroslav Tykhiy if (SLIST_FIRST(&euidlist) != NULL && li == NULL) { 42621be3b31SYaroslav Tykhiy selected[i] = 0; 42721be3b31SYaroslav Tykhiy continue; 42821be3b31SYaroslav Tykhiy } 42921be3b31SYaroslav Tykhiy 43021be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &ppidlist, li_chain) 43121be3b31SYaroslav Tykhiy if (kp->ki_ppid == (pid_t)li->li_number) 43221be3b31SYaroslav Tykhiy break; 43321be3b31SYaroslav Tykhiy if (SLIST_FIRST(&ppidlist) != NULL && li == NULL) { 43421be3b31SYaroslav Tykhiy selected[i] = 0; 43521be3b31SYaroslav Tykhiy continue; 43621be3b31SYaroslav Tykhiy } 43721be3b31SYaroslav Tykhiy 43821be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &pgrplist, li_chain) 43921be3b31SYaroslav Tykhiy if (kp->ki_pgid == (pid_t)li->li_number) 44021be3b31SYaroslav Tykhiy break; 44121be3b31SYaroslav Tykhiy if (SLIST_FIRST(&pgrplist) != NULL && li == NULL) { 44221be3b31SYaroslav Tykhiy selected[i] = 0; 44321be3b31SYaroslav Tykhiy continue; 44421be3b31SYaroslav Tykhiy } 44521be3b31SYaroslav Tykhiy 44621be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &tdevlist, li_chain) { 44721be3b31SYaroslav Tykhiy if (li->li_number == -1 && 44821be3b31SYaroslav Tykhiy (kp->ki_flag & P_CONTROLT) == 0) 44921be3b31SYaroslav Tykhiy break; 45021be3b31SYaroslav Tykhiy if (kp->ki_tdev == (dev_t)li->li_number) 45121be3b31SYaroslav Tykhiy break; 45221be3b31SYaroslav Tykhiy } 45321be3b31SYaroslav Tykhiy if (SLIST_FIRST(&tdevlist) != NULL && li == NULL) { 45421be3b31SYaroslav Tykhiy selected[i] = 0; 45521be3b31SYaroslav Tykhiy continue; 45621be3b31SYaroslav Tykhiy } 45721be3b31SYaroslav Tykhiy 45821be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &sidlist, li_chain) 45921be3b31SYaroslav Tykhiy if (kp->ki_sid == (pid_t)li->li_number) 46021be3b31SYaroslav Tykhiy break; 46121be3b31SYaroslav Tykhiy if (SLIST_FIRST(&sidlist) != NULL && li == NULL) { 46221be3b31SYaroslav Tykhiy selected[i] = 0; 46321be3b31SYaroslav Tykhiy continue; 46421be3b31SYaroslav Tykhiy } 46521be3b31SYaroslav Tykhiy 46621be3b31SYaroslav Tykhiy SLIST_FOREACH(li, &jidlist, li_chain) { 46721be3b31SYaroslav Tykhiy /* A particular jail ID, including 0 (not in jail) */ 46821be3b31SYaroslav Tykhiy if (kp->ki_jid == (int)li->li_number) 46921be3b31SYaroslav Tykhiy break; 47021be3b31SYaroslav Tykhiy /* Any jail */ 47121be3b31SYaroslav Tykhiy if (kp->ki_jid > 0 && li->li_number == -1) 47221be3b31SYaroslav Tykhiy break; 47321be3b31SYaroslav Tykhiy } 47421be3b31SYaroslav Tykhiy if (SLIST_FIRST(&jidlist) != NULL && li == NULL) { 47521be3b31SYaroslav Tykhiy selected[i] = 0; 47621be3b31SYaroslav Tykhiy continue; 47721be3b31SYaroslav Tykhiy } 47821be3b31SYaroslav Tykhiy 479*cc55ad3dSEdward Tomasz Napierala SLIST_FOREACH(li, &classlist, li_chain) { 480*cc55ad3dSEdward Tomasz Napierala /* 481*cc55ad3dSEdward Tomasz Napierala * We skip P_SYSTEM processes to match ps(1) output. 482*cc55ad3dSEdward Tomasz Napierala */ 483*cc55ad3dSEdward Tomasz Napierala if ((kp->ki_flag & P_SYSTEM) == 0 && 484*cc55ad3dSEdward Tomasz Napierala kp->ki_loginclass != NULL && 485*cc55ad3dSEdward Tomasz Napierala strcmp(kp->ki_loginclass, li->li_name) == 0) 486*cc55ad3dSEdward Tomasz Napierala break; 487*cc55ad3dSEdward Tomasz Napierala } 488*cc55ad3dSEdward Tomasz Napierala if (SLIST_FIRST(&classlist) != NULL && li == NULL) { 489*cc55ad3dSEdward Tomasz Napierala selected[i] = 0; 490*cc55ad3dSEdward Tomasz Napierala continue; 491*cc55ad3dSEdward Tomasz Napierala } 492*cc55ad3dSEdward Tomasz Napierala 49321be3b31SYaroslav Tykhiy if (argc == 0) 49421be3b31SYaroslav Tykhiy selected[i] = 1; 49521be3b31SYaroslav Tykhiy } 49621be3b31SYaroslav Tykhiy 4970a091aebSBrian Somers if (!ancestors) { 4980a091aebSBrian Somers pid = mypid; 4990a091aebSBrian Somers while (pid) { 5000a091aebSBrian Somers for (i = 0, kp = plist; i < nproc; i++, kp++) { 5010a091aebSBrian Somers if (PSKIP(kp)) 5020a091aebSBrian Somers continue; 5030a091aebSBrian Somers if (kp->ki_pid == pid) { 5040a091aebSBrian Somers selected[i] = 0; 5050a091aebSBrian Somers pid = kp->ki_ppid; 5060a091aebSBrian Somers break; 5070a091aebSBrian Somers } 5080a091aebSBrian Somers } 5090a091aebSBrian Somers if (i == nproc) { 5100a091aebSBrian Somers if (pid == mypid) 5110a091aebSBrian Somers pid = getppid(); 5120a091aebSBrian Somers else 5130a091aebSBrian Somers break; /* Maybe we're in a jail ? */ 5140a091aebSBrian Somers } 5150a091aebSBrian Somers } 5160a091aebSBrian Somers } 5170a091aebSBrian Somers 51821be3b31SYaroslav Tykhiy if (newest || oldest) { 51921be3b31SYaroslav Tykhiy best_tval.tv_sec = 0; 52021be3b31SYaroslav Tykhiy best_tval.tv_usec = 0; 52121be3b31SYaroslav Tykhiy bestidx = -1; 52221be3b31SYaroslav Tykhiy 52321be3b31SYaroslav Tykhiy for (i = 0, kp = plist; i < nproc; i++, kp++) { 52421be3b31SYaroslav Tykhiy if (!selected[i]) 52521be3b31SYaroslav Tykhiy continue; 52621be3b31SYaroslav Tykhiy if (bestidx == -1) { 52721be3b31SYaroslav Tykhiy /* The first entry of the list which matched. */ 52821be3b31SYaroslav Tykhiy ; 52921be3b31SYaroslav Tykhiy } else if (timercmp(&kp->ki_start, &best_tval, >)) { 53021be3b31SYaroslav Tykhiy /* This entry is newer than previous "best". */ 53121be3b31SYaroslav Tykhiy if (oldest) /* but we want the oldest */ 53221be3b31SYaroslav Tykhiy continue; 53321be3b31SYaroslav Tykhiy } else { 53421be3b31SYaroslav Tykhiy /* This entry is older than previous "best". */ 53521be3b31SYaroslav Tykhiy if (newest) /* but we want the newest */ 53621be3b31SYaroslav Tykhiy continue; 53721be3b31SYaroslav Tykhiy } 53821be3b31SYaroslav Tykhiy /* This entry is better than previous "best" entry. */ 53921be3b31SYaroslav Tykhiy best_tval.tv_sec = kp->ki_start.tv_sec; 54021be3b31SYaroslav Tykhiy best_tval.tv_usec = kp->ki_start.tv_usec; 54121be3b31SYaroslav Tykhiy bestidx = i; 54221be3b31SYaroslav Tykhiy } 54321be3b31SYaroslav Tykhiy 54421be3b31SYaroslav Tykhiy memset(selected, 0, nproc); 54521be3b31SYaroslav Tykhiy if (bestidx != -1) 54621be3b31SYaroslav Tykhiy selected[bestidx] = 1; 54721be3b31SYaroslav Tykhiy } 54821be3b31SYaroslav Tykhiy 54921be3b31SYaroslav Tykhiy /* 55021be3b31SYaroslav Tykhiy * Take the appropriate action for each matched process, if any. 55121be3b31SYaroslav Tykhiy */ 5523ba0e470SBrian Somers did_action = 0; 55321be3b31SYaroslav Tykhiy for (i = 0, rv = 0, kp = plist; i < nproc; i++, kp++) { 55421be3b31SYaroslav Tykhiy if (PSKIP(kp)) 55521be3b31SYaroslav Tykhiy continue; 55621be3b31SYaroslav Tykhiy if (selected[i]) { 5573ba0e470SBrian Somers if (longfmt && !pgrep) { 5583ba0e470SBrian Somers did_action = 1; 5593ba0e470SBrian Somers printf("kill -%d %d\n", signum, kp->ki_pid); 5603ba0e470SBrian Somers } 56121be3b31SYaroslav Tykhiy if (inverse) 56221be3b31SYaroslav Tykhiy continue; 56321be3b31SYaroslav Tykhiy } else if (!inverse) 56421be3b31SYaroslav Tykhiy continue; 56521be3b31SYaroslav Tykhiy rv |= (*action)(kp); 56621be3b31SYaroslav Tykhiy } 5673ba0e470SBrian Somers if (!did_action && !pgrep && longfmt) 5683ba0e470SBrian Somers fprintf(stderr, 5693ba0e470SBrian Somers "No matching processes belonging to you were found\n"); 57021be3b31SYaroslav Tykhiy 57121be3b31SYaroslav Tykhiy exit(rv ? STATUS_MATCH : STATUS_NOMATCH); 57221be3b31SYaroslav Tykhiy } 57321be3b31SYaroslav Tykhiy 57421be3b31SYaroslav Tykhiy static void 57521be3b31SYaroslav Tykhiy usage(void) 57621be3b31SYaroslav Tykhiy { 57721be3b31SYaroslav Tykhiy const char *ustr; 57821be3b31SYaroslav Tykhiy 57921be3b31SYaroslav Tykhiy if (pgrep) 5807b0706f6SPawel Jakub Dawidek ustr = "[-LSfilnoqvx] [-d delim]"; 58121be3b31SYaroslav Tykhiy else 5823ba0e470SBrian Somers ustr = "[-signal] [-ILfilnovx]"; 58321be3b31SYaroslav Tykhiy 58421be3b31SYaroslav Tykhiy fprintf(stderr, 58521be3b31SYaroslav Tykhiy "usage: %s %s [-F pidfile] [-G gid] [-M core] [-N system]\n" 586*cc55ad3dSEdward Tomasz Napierala " [-P ppid] [-U uid] [-c class] [-g pgrp] [-j jid]\n" 587*cc55ad3dSEdward Tomasz Napierala " [-s sid] [-t tty] [-u euid] pattern ...\n", 588*cc55ad3dSEdward Tomasz Napierala getprogname(), ustr); 58921be3b31SYaroslav Tykhiy 59021be3b31SYaroslav Tykhiy exit(STATUS_BADUSAGE); 59121be3b31SYaroslav Tykhiy } 59221be3b31SYaroslav Tykhiy 59321be3b31SYaroslav Tykhiy static void 59421be3b31SYaroslav Tykhiy show_process(const struct kinfo_proc *kp) 59521be3b31SYaroslav Tykhiy { 59621be3b31SYaroslav Tykhiy char **argv; 59721be3b31SYaroslav Tykhiy 5987b0706f6SPawel Jakub Dawidek if (quiet) { 5997b0706f6SPawel Jakub Dawidek assert(pgrep); 6007b0706f6SPawel Jakub Dawidek return; 6017b0706f6SPawel Jakub Dawidek } 60221be3b31SYaroslav Tykhiy if ((longfmt || !pgrep) && matchargs && 60321be3b31SYaroslav Tykhiy (argv = kvm_getargv(kd, kp, 0)) != NULL) { 60421be3b31SYaroslav Tykhiy printf("%d ", (int)kp->ki_pid); 60521be3b31SYaroslav Tykhiy for (; *argv != NULL; argv++) { 60621be3b31SYaroslav Tykhiy printf("%s", *argv); 60721be3b31SYaroslav Tykhiy if (argv[1] != NULL) 60821be3b31SYaroslav Tykhiy putchar(' '); 60921be3b31SYaroslav Tykhiy } 61021be3b31SYaroslav Tykhiy } else if (longfmt || !pgrep) 61121be3b31SYaroslav Tykhiy printf("%d %s", (int)kp->ki_pid, kp->ki_comm); 61221be3b31SYaroslav Tykhiy else 61321be3b31SYaroslav Tykhiy printf("%d", (int)kp->ki_pid); 61421be3b31SYaroslav Tykhiy } 61521be3b31SYaroslav Tykhiy 61621be3b31SYaroslav Tykhiy static int 61721be3b31SYaroslav Tykhiy killact(const struct kinfo_proc *kp) 61821be3b31SYaroslav Tykhiy { 61921be3b31SYaroslav Tykhiy int ch, first; 62021be3b31SYaroslav Tykhiy 62121be3b31SYaroslav Tykhiy if (interactive) { 62221be3b31SYaroslav Tykhiy /* 62321be3b31SYaroslav Tykhiy * Be careful, ask before killing. 62421be3b31SYaroslav Tykhiy */ 62521be3b31SYaroslav Tykhiy printf("kill "); 62621be3b31SYaroslav Tykhiy show_process(kp); 62721be3b31SYaroslav Tykhiy printf("? "); 62821be3b31SYaroslav Tykhiy fflush(stdout); 62921be3b31SYaroslav Tykhiy first = ch = getchar(); 63021be3b31SYaroslav Tykhiy while (ch != '\n' && ch != EOF) 63121be3b31SYaroslav Tykhiy ch = getchar(); 63221be3b31SYaroslav Tykhiy if (first != 'y' && first != 'Y') 63321be3b31SYaroslav Tykhiy return (1); 63421be3b31SYaroslav Tykhiy } 63521be3b31SYaroslav Tykhiy if (kill(kp->ki_pid, signum) == -1) { 63621be3b31SYaroslav Tykhiy /* 63721be3b31SYaroslav Tykhiy * Check for ESRCH, which indicates that the process 63821be3b31SYaroslav Tykhiy * disappeared between us matching it and us 63921be3b31SYaroslav Tykhiy * signalling it; don't issue a warning about it. 64021be3b31SYaroslav Tykhiy */ 64121be3b31SYaroslav Tykhiy if (errno != ESRCH) 64221be3b31SYaroslav Tykhiy warn("signalling pid %d", (int)kp->ki_pid); 64321be3b31SYaroslav Tykhiy /* 64421be3b31SYaroslav Tykhiy * Return 0 to indicate that the process should not be 64521be3b31SYaroslav Tykhiy * considered a match, since we didn't actually get to 64621be3b31SYaroslav Tykhiy * signal it. 64721be3b31SYaroslav Tykhiy */ 64821be3b31SYaroslav Tykhiy return (0); 64921be3b31SYaroslav Tykhiy } 65021be3b31SYaroslav Tykhiy return (1); 65121be3b31SYaroslav Tykhiy } 65221be3b31SYaroslav Tykhiy 65321be3b31SYaroslav Tykhiy static int 65421be3b31SYaroslav Tykhiy grepact(const struct kinfo_proc *kp) 65521be3b31SYaroslav Tykhiy { 65621be3b31SYaroslav Tykhiy 65721be3b31SYaroslav Tykhiy show_process(kp); 6587b0706f6SPawel Jakub Dawidek if (!quiet) 65921be3b31SYaroslav Tykhiy printf("%s", delim); 66021be3b31SYaroslav Tykhiy return (1); 66121be3b31SYaroslav Tykhiy } 66221be3b31SYaroslav Tykhiy 66321be3b31SYaroslav Tykhiy static void 66421be3b31SYaroslav Tykhiy makelist(struct listhead *head, enum listtype type, char *src) 66521be3b31SYaroslav Tykhiy { 66621be3b31SYaroslav Tykhiy struct list *li; 66721be3b31SYaroslav Tykhiy struct passwd *pw; 66821be3b31SYaroslav Tykhiy struct group *gr; 66921be3b31SYaroslav Tykhiy struct stat st; 670eb948eacSEd Schouten const char *cp; 67121be3b31SYaroslav Tykhiy char *sp, *ep, buf[MAXPATHLEN]; 67221be3b31SYaroslav Tykhiy int empty; 67321be3b31SYaroslav Tykhiy 67421be3b31SYaroslav Tykhiy empty = 1; 67521be3b31SYaroslav Tykhiy 67621be3b31SYaroslav Tykhiy while ((sp = strsep(&src, ",")) != NULL) { 67721be3b31SYaroslav Tykhiy if (*sp == '\0') 67821be3b31SYaroslav Tykhiy usage(); 67921be3b31SYaroslav Tykhiy 68021be3b31SYaroslav Tykhiy if ((li = malloc(sizeof(*li))) == NULL) { 68121be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot allocate %zu bytes", 68221be3b31SYaroslav Tykhiy sizeof(*li)); 68321be3b31SYaroslav Tykhiy } 68421be3b31SYaroslav Tykhiy 68521be3b31SYaroslav Tykhiy SLIST_INSERT_HEAD(head, li, li_chain); 68621be3b31SYaroslav Tykhiy empty = 0; 68721be3b31SYaroslav Tykhiy 688*cc55ad3dSEdward Tomasz Napierala if (type != LT_CLASS) 68921be3b31SYaroslav Tykhiy li->li_number = (uid_t)strtol(sp, &ep, 0); 690*cc55ad3dSEdward Tomasz Napierala 691*cc55ad3dSEdward Tomasz Napierala if (type != LT_CLASS && *ep == '\0') { 69221be3b31SYaroslav Tykhiy switch (type) { 69321be3b31SYaroslav Tykhiy case LT_PGRP: 69421be3b31SYaroslav Tykhiy if (li->li_number == 0) 69521be3b31SYaroslav Tykhiy li->li_number = getpgrp(); 69621be3b31SYaroslav Tykhiy break; 69721be3b31SYaroslav Tykhiy case LT_SID: 69821be3b31SYaroslav Tykhiy if (li->li_number == 0) 69921be3b31SYaroslav Tykhiy li->li_number = getsid(mypid); 70021be3b31SYaroslav Tykhiy break; 70121be3b31SYaroslav Tykhiy case LT_JID: 70221be3b31SYaroslav Tykhiy if (li->li_number < 0) 70321be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 70421be3b31SYaroslav Tykhiy "Negative jail ID `%s'", sp); 70521be3b31SYaroslav Tykhiy /* For compatibility with old -j */ 70621be3b31SYaroslav Tykhiy if (li->li_number == 0) 70721be3b31SYaroslav Tykhiy li->li_number = -1; /* any jail */ 70821be3b31SYaroslav Tykhiy break; 7092a546db8SDavid E. O'Brien case LT_TTY: 7102a546db8SDavid E. O'Brien if (li->li_number < 0) 7112a546db8SDavid E. O'Brien errx(STATUS_BADUSAGE, 7122a546db8SDavid E. O'Brien "Negative /dev/pts tty `%s'", sp); 7132a546db8SDavid E. O'Brien snprintf(buf, sizeof(buf), _PATH_DEV "pts/%s", 7142a546db8SDavid E. O'Brien sp); 7152a546db8SDavid E. O'Brien if (stat(buf, &st) != -1) 7162a546db8SDavid E. O'Brien goto foundtty; 7172a546db8SDavid E. O'Brien if (errno == ENOENT) 7182a546db8SDavid E. O'Brien errx(STATUS_BADUSAGE, "No such tty: `" 7192a546db8SDavid E. O'Brien _PATH_DEV "pts/%s'", sp); 7202a546db8SDavid E. O'Brien err(STATUS_ERROR, "Cannot access `" 7212a546db8SDavid E. O'Brien _PATH_DEV "pts/%s'", sp); 7222a546db8SDavid E. O'Brien break; 72321be3b31SYaroslav Tykhiy default: 72421be3b31SYaroslav Tykhiy break; 72521be3b31SYaroslav Tykhiy } 72621be3b31SYaroslav Tykhiy continue; 72721be3b31SYaroslav Tykhiy } 72821be3b31SYaroslav Tykhiy 72921be3b31SYaroslav Tykhiy switch (type) { 73021be3b31SYaroslav Tykhiy case LT_USER: 73121be3b31SYaroslav Tykhiy if ((pw = getpwnam(sp)) == NULL) 73221be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Unknown user `%s'", sp); 73321be3b31SYaroslav Tykhiy li->li_number = pw->pw_uid; 73421be3b31SYaroslav Tykhiy break; 73521be3b31SYaroslav Tykhiy case LT_GROUP: 73621be3b31SYaroslav Tykhiy if ((gr = getgrnam(sp)) == NULL) 73721be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Unknown group `%s'", sp); 73821be3b31SYaroslav Tykhiy li->li_number = gr->gr_gid; 73921be3b31SYaroslav Tykhiy break; 74021be3b31SYaroslav Tykhiy case LT_TTY: 74121be3b31SYaroslav Tykhiy if (strcmp(sp, "-") == 0) { 74221be3b31SYaroslav Tykhiy li->li_number = -1; 74321be3b31SYaroslav Tykhiy break; 74421be3b31SYaroslav Tykhiy } else if (strcmp(sp, "co") == 0) { 74521be3b31SYaroslav Tykhiy cp = "console"; 74621be3b31SYaroslav Tykhiy } else { 74721be3b31SYaroslav Tykhiy cp = sp; 74821be3b31SYaroslav Tykhiy } 74921be3b31SYaroslav Tykhiy 750eb948eacSEd Schouten snprintf(buf, sizeof(buf), _PATH_DEV "%s", cp); 75118aa158fSEd Schouten if (stat(buf, &st) != -1) 75218aa158fSEd Schouten goto foundtty; 75321be3b31SYaroslav Tykhiy 75418aa158fSEd Schouten snprintf(buf, sizeof(buf), _PATH_DEV "tty%s", cp); 75518aa158fSEd Schouten if (stat(buf, &st) != -1) 75618aa158fSEd Schouten goto foundtty; 75718aa158fSEd Schouten 75818aa158fSEd Schouten if (errno == ENOENT) 75918aa158fSEd Schouten errx(STATUS_BADUSAGE, "No such tty: `%s'", sp); 76021be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot access `%s'", sp); 76121be3b31SYaroslav Tykhiy 76218aa158fSEd Schouten foundtty: if ((st.st_mode & S_IFCHR) == 0) 76321be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, "Not a tty: `%s'", sp); 76421be3b31SYaroslav Tykhiy 76521be3b31SYaroslav Tykhiy li->li_number = st.st_rdev; 76621be3b31SYaroslav Tykhiy break; 76721be3b31SYaroslav Tykhiy case LT_JID: 76821be3b31SYaroslav Tykhiy if (strcmp(sp, "none") == 0) 76921be3b31SYaroslav Tykhiy li->li_number = 0; 77021be3b31SYaroslav Tykhiy else if (strcmp(sp, "any") == 0) 77121be3b31SYaroslav Tykhiy li->li_number = -1; 77221be3b31SYaroslav Tykhiy else if (*ep != '\0') 77321be3b31SYaroslav Tykhiy errx(STATUS_BADUSAGE, 77421be3b31SYaroslav Tykhiy "Invalid jail ID `%s'", sp); 77521be3b31SYaroslav Tykhiy break; 776*cc55ad3dSEdward Tomasz Napierala case LT_CLASS: 777*cc55ad3dSEdward Tomasz Napierala li->li_number = -1; 778*cc55ad3dSEdward Tomasz Napierala li->li_name = strdup(sp); 779*cc55ad3dSEdward Tomasz Napierala if (li->li_name == NULL) 780*cc55ad3dSEdward Tomasz Napierala err(STATUS_ERROR, "Cannot allocate memory"); 781*cc55ad3dSEdward Tomasz Napierala break; 78221be3b31SYaroslav Tykhiy default: 78321be3b31SYaroslav Tykhiy usage(); 78421be3b31SYaroslav Tykhiy } 78521be3b31SYaroslav Tykhiy } 78621be3b31SYaroslav Tykhiy 78721be3b31SYaroslav Tykhiy if (empty) 78821be3b31SYaroslav Tykhiy usage(); 78921be3b31SYaroslav Tykhiy } 79021be3b31SYaroslav Tykhiy 79121be3b31SYaroslav Tykhiy static int 79221be3b31SYaroslav Tykhiy takepid(const char *pidfile, int pidfilelock) 79321be3b31SYaroslav Tykhiy { 79421be3b31SYaroslav Tykhiy char *endp, line[BUFSIZ]; 79521be3b31SYaroslav Tykhiy FILE *fh; 79621be3b31SYaroslav Tykhiy long rval; 79721be3b31SYaroslav Tykhiy 79821be3b31SYaroslav Tykhiy fh = fopen(pidfile, "r"); 79921be3b31SYaroslav Tykhiy if (fh == NULL) 80021be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot open pidfile `%s'", pidfile); 80121be3b31SYaroslav Tykhiy 80221be3b31SYaroslav Tykhiy if (pidfilelock) { 80321be3b31SYaroslav Tykhiy /* 80421be3b31SYaroslav Tykhiy * If we can lock pidfile, this means that daemon is not 80521be3b31SYaroslav Tykhiy * running, so would be better not to kill some random process. 80621be3b31SYaroslav Tykhiy */ 80721be3b31SYaroslav Tykhiy if (flock(fileno(fh), LOCK_EX | LOCK_NB) == 0) { 80821be3b31SYaroslav Tykhiy (void)fclose(fh); 80921be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "File '%s' can be locked", pidfile); 81021be3b31SYaroslav Tykhiy } else { 81121be3b31SYaroslav Tykhiy if (errno != EWOULDBLOCK) { 81221be3b31SYaroslav Tykhiy errx(STATUS_ERROR, 81321be3b31SYaroslav Tykhiy "Error while locking file '%s'", pidfile); 81421be3b31SYaroslav Tykhiy } 81521be3b31SYaroslav Tykhiy } 81621be3b31SYaroslav Tykhiy } 81721be3b31SYaroslav Tykhiy 81821be3b31SYaroslav Tykhiy if (fgets(line, sizeof(line), fh) == NULL) { 81921be3b31SYaroslav Tykhiy if (feof(fh)) { 82021be3b31SYaroslav Tykhiy (void)fclose(fh); 82121be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Pidfile `%s' is empty", pidfile); 82221be3b31SYaroslav Tykhiy } 82321be3b31SYaroslav Tykhiy (void)fclose(fh); 82421be3b31SYaroslav Tykhiy err(STATUS_ERROR, "Cannot read from pid file `%s'", pidfile); 82521be3b31SYaroslav Tykhiy } 82621be3b31SYaroslav Tykhiy (void)fclose(fh); 82721be3b31SYaroslav Tykhiy 82821be3b31SYaroslav Tykhiy rval = strtol(line, &endp, 10); 82921be3b31SYaroslav Tykhiy if (*endp != '\0' && !isspace((unsigned char)*endp)) 83021be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile); 83121be3b31SYaroslav Tykhiy else if (rval < MIN_PID || rval > MAX_PID) 83221be3b31SYaroslav Tykhiy errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile); 83321be3b31SYaroslav Tykhiy return (rval); 83421be3b31SYaroslav Tykhiy } 835