165a0e3d6SJohn Marino /* $NetBSD: pkill.c,v 1.16 2005/10/10 22:13:20 kleink Exp $ */
2df558cd9SChris Pressey
3df558cd9SChris Pressey /*-
4df558cd9SChris Pressey * Copyright (c) 2002 The NetBSD Foundation, Inc.
565a0e3d6SJohn Marino * Copyright (c) 2005 Pawel Jakub Dawidek <pjd@FreeBSD.org>
6df558cd9SChris Pressey * All rights reserved.
7df558cd9SChris Pressey *
8df558cd9SChris Pressey * This code is derived from software contributed to The NetBSD Foundation
9df558cd9SChris Pressey * by Andrew Doran.
10df558cd9SChris Pressey *
11df558cd9SChris Pressey * Redistribution and use in source and binary forms, with or without
12df558cd9SChris Pressey * modification, are permitted provided that the following conditions
13df558cd9SChris Pressey * are met:
14df558cd9SChris Pressey * 1. Redistributions of source code must retain the above copyright
15df558cd9SChris Pressey * notice, this list of conditions and the following disclaimer.
16df558cd9SChris Pressey * 2. Redistributions in binary form must reproduce the above copyright
17df558cd9SChris Pressey * notice, this list of conditions and the following disclaimer in the
18df558cd9SChris Pressey * documentation and/or other materials provided with the distribution.
19df558cd9SChris Pressey *
20df558cd9SChris Pressey * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
21df558cd9SChris Pressey * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
22df558cd9SChris Pressey * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
23df558cd9SChris Pressey * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
24df558cd9SChris Pressey * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
25df558cd9SChris Pressey * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
26df558cd9SChris Pressey * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
27df558cd9SChris Pressey * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
28df558cd9SChris Pressey * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
29df558cd9SChris Pressey * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30df558cd9SChris Pressey * POSSIBILITY OF SUCH DAMAGE.
3165a0e3d6SJohn Marino *
3265a0e3d6SJohn Marino * $FreeBSD: head/bin/pkill/pkill.c 256050 2013-10-04 16:08:44Z trasz $
33df558cd9SChris Pressey */
34df558cd9SChris Pressey
35e0ecab34SMatthew Dillon #include <sys/user.h>
36df558cd9SChris Pressey #include <sys/types.h>
37df558cd9SChris Pressey #include <sys/param.h>
38df558cd9SChris Pressey #include <sys/sysctl.h>
39df558cd9SChris Pressey #include <sys/queue.h>
40df558cd9SChris Pressey #include <sys/stat.h>
414e3518d8SChris Pressey #include <sys/fcntl.h>
4265a0e3d6SJohn Marino #include <sys/time.h>
43df558cd9SChris Pressey
4465a0e3d6SJohn Marino #include <assert.h>
45df558cd9SChris Pressey #include <stdio.h>
46df558cd9SChris Pressey #include <stdlib.h>
47df558cd9SChris Pressey #include <limits.h>
484e3518d8SChris Pressey #include <paths.h>
49df558cd9SChris Pressey #include <string.h>
50df558cd9SChris Pressey #include <unistd.h>
51df558cd9SChris Pressey #include <signal.h>
52df558cd9SChris Pressey #include <regex.h>
53df558cd9SChris Pressey #include <ctype.h>
54df558cd9SChris Pressey #include <kvm.h>
55df558cd9SChris Pressey #include <err.h>
56df558cd9SChris Pressey #include <pwd.h>
57df558cd9SChris Pressey #include <grp.h>
58df558cd9SChris Pressey #include <errno.h>
5965a0e3d6SJohn Marino #include <locale.h>
60df558cd9SChris Pressey
61df558cd9SChris Pressey #define STATUS_MATCH 0
62df558cd9SChris Pressey #define STATUS_NOMATCH 1
63df558cd9SChris Pressey #define STATUS_BADUSAGE 2
64df558cd9SChris Pressey #define STATUS_ERROR 3
65df558cd9SChris Pressey
6665a0e3d6SJohn Marino #define MIN_PID 5
67a841db25SSascha Wildner #define MAX_PID PID_MAX
6865a0e3d6SJohn Marino
6965a0e3d6SJohn Marino /* Ignore system-processes (if '-S' flag is not specified) and myself. */
7065a0e3d6SJohn Marino #define PSKIP(kp) ((kp)->kp_pid == mypid || \
719b5e1191SAntonio Huete Jimenez (!kthreads && (kp)->kp_pid <= 0))
7265a0e3d6SJohn Marino
73df558cd9SChris Pressey enum listtype {
7465a0e3d6SJohn Marino LT_GENERIC,
7565a0e3d6SJohn Marino LT_USER,
7665a0e3d6SJohn Marino LT_GROUP,
7765a0e3d6SJohn Marino LT_TTY,
7865a0e3d6SJohn Marino LT_PGRP,
7965a0e3d6SJohn Marino LT_JID,
8065a0e3d6SJohn Marino LT_SID,
8165a0e3d6SJohn Marino LT_CLASS
82df558cd9SChris Pressey };
83df558cd9SChris Pressey
84df558cd9SChris Pressey struct list {
85df558cd9SChris Pressey SLIST_ENTRY(list) li_chain;
8665a0e3d6SJohn Marino long li_number;
8765a0e3d6SJohn Marino char *li_name;
88df558cd9SChris Pressey };
89df558cd9SChris Pressey
90df558cd9SChris Pressey SLIST_HEAD(listhead, list);
91df558cd9SChris Pressey
9265a0e3d6SJohn Marino static struct kinfo_proc *plist;
9365a0e3d6SJohn Marino static char *selected;
9465a0e3d6SJohn Marino static const char *delim = "\n";
9565a0e3d6SJohn Marino static int nproc;
9665a0e3d6SJohn Marino static int pgrep;
9765a0e3d6SJohn Marino static int signum = SIGTERM;
9865a0e3d6SJohn Marino static int newest;
9965a0e3d6SJohn Marino static int oldest;
10065a0e3d6SJohn Marino static int interactive;
10165a0e3d6SJohn Marino static int inverse;
10265a0e3d6SJohn Marino static int longfmt;
10365a0e3d6SJohn Marino static int matchargs;
10465a0e3d6SJohn Marino static int fullmatch;
10565a0e3d6SJohn Marino static int kthreads;
10665a0e3d6SJohn Marino static int cflags = REG_EXTENDED;
10765a0e3d6SJohn Marino static int quiet;
10865a0e3d6SJohn Marino static kvm_t *kd;
10965a0e3d6SJohn Marino static pid_t mypid;
110df558cd9SChris Pressey
11165a0e3d6SJohn Marino static struct listhead euidlist = SLIST_HEAD_INITIALIZER(euidlist);
11265a0e3d6SJohn Marino static struct listhead ruidlist = SLIST_HEAD_INITIALIZER(ruidlist);
11365a0e3d6SJohn Marino static struct listhead rgidlist = SLIST_HEAD_INITIALIZER(rgidlist);
11465a0e3d6SJohn Marino static struct listhead pgrplist = SLIST_HEAD_INITIALIZER(pgrplist);
11565a0e3d6SJohn Marino static struct listhead ppidlist = SLIST_HEAD_INITIALIZER(ppidlist);
11665a0e3d6SJohn Marino static struct listhead tdevlist = SLIST_HEAD_INITIALIZER(tdevlist);
11765a0e3d6SJohn Marino static struct listhead sidlist = SLIST_HEAD_INITIALIZER(sidlist);
11865a0e3d6SJohn Marino static struct listhead jidlist = SLIST_HEAD_INITIALIZER(jidlist);
11965a0e3d6SJohn Marino static struct listhead classlist = SLIST_HEAD_INITIALIZER(classlist);
120df558cd9SChris Pressey
12165a0e3d6SJohn Marino static void usage(void) __attribute__((__noreturn__));
12265a0e3d6SJohn Marino static int killact(const struct kinfo_proc *);
12365a0e3d6SJohn Marino static int grepact(const struct kinfo_proc *);
12465a0e3d6SJohn Marino static void makelist(struct listhead *, enum listtype, char *);
12565a0e3d6SJohn Marino static int takepid(const char *, int);
126df558cd9SChris Pressey
127df558cd9SChris Pressey int
main(int argc,char ** argv)128df558cd9SChris Pressey main(int argc, char **argv)
129df558cd9SChris Pressey {
130*a2d71d01SLevente Kurusa char buf[_POSIX2_LINE_MAX], *mstr, **pargv, *p, *q, *pidfile, *tdev;
13165a0e3d6SJohn Marino const char *execf, *coref;
13265a0e3d6SJohn Marino int ancestors, debug_opt, did_action;
13365a0e3d6SJohn Marino int i, ch, bestidx, rv, criteria, pidfromfile, pidfilelock;
13465a0e3d6SJohn Marino size_t jsz;
13565a0e3d6SJohn Marino int (*action)(const struct kinfo_proc *);
1364e3518d8SChris Pressey struct kinfo_proc *kp;
137df558cd9SChris Pressey struct list *li;
13865a0e3d6SJohn Marino struct timeval best_tval;
139df558cd9SChris Pressey regex_t reg;
140df558cd9SChris Pressey regmatch_t regmatch;
14165a0e3d6SJohn Marino pid_t pid;
14265a0e3d6SJohn Marino
14365a0e3d6SJohn Marino setlocale(LC_ALL, "");
144df558cd9SChris Pressey
145df558cd9SChris Pressey if (strcmp(getprogname(), "pgrep") == 0) {
146df558cd9SChris Pressey action = grepact;
147df558cd9SChris Pressey pgrep = 1;
148df558cd9SChris Pressey } else {
149df558cd9SChris Pressey action = killact;
150df558cd9SChris Pressey p = argv[1];
151df558cd9SChris Pressey
152df558cd9SChris Pressey if (argc > 1 && p[0] == '-') {
153df558cd9SChris Pressey p++;
154df558cd9SChris Pressey i = (int)strtol(p, &q, 10);
155df558cd9SChris Pressey if (*q == '\0') {
156df558cd9SChris Pressey signum = i;
157df558cd9SChris Pressey argv++;
158df558cd9SChris Pressey argc--;
159df558cd9SChris Pressey } else {
16065a0e3d6SJohn Marino if (strncasecmp(p, "SIG", 3) == 0)
161df558cd9SChris Pressey p += 3;
16265a0e3d6SJohn Marino for (i = 1; i < NSIG; i++)
163df558cd9SChris Pressey if (strcasecmp(sys_signame[i], p) == 0)
164df558cd9SChris Pressey break;
165df558cd9SChris Pressey if (i != NSIG) {
166df558cd9SChris Pressey signum = i;
167df558cd9SChris Pressey argv++;
168df558cd9SChris Pressey argc--;
169df558cd9SChris Pressey }
170df558cd9SChris Pressey }
171df558cd9SChris Pressey }
172df558cd9SChris Pressey }
173df558cd9SChris Pressey
17465a0e3d6SJohn Marino ancestors = 0;
175df558cd9SChris Pressey criteria = 0;
17665a0e3d6SJohn Marino debug_opt = 0;
17765a0e3d6SJohn Marino pidfile = NULL;
17865a0e3d6SJohn Marino pidfilelock = 0;
17965a0e3d6SJohn Marino quiet = 0;
18065a0e3d6SJohn Marino execf = NULL;
18165a0e3d6SJohn Marino coref = _PATH_DEVNULL;
182df558cd9SChris Pressey
183*a2d71d01SLevente Kurusa while ((ch = getopt(argc, argv, "DF:G:ILM:N:P:STU:ac:d:fg:ij:lnoqs:t:u:vx")) != -1) {
184df558cd9SChris Pressey switch (ch) {
18565a0e3d6SJohn Marino case 'D':
18665a0e3d6SJohn Marino debug_opt++;
18765a0e3d6SJohn Marino break;
18865a0e3d6SJohn Marino case 'F':
18965a0e3d6SJohn Marino pidfile = optarg;
19065a0e3d6SJohn Marino criteria = 1;
19165a0e3d6SJohn Marino break;
192df558cd9SChris Pressey case 'G':
193df558cd9SChris Pressey makelist(&rgidlist, LT_GROUP, optarg);
194df558cd9SChris Pressey criteria = 1;
195df558cd9SChris Pressey break;
19665a0e3d6SJohn Marino case 'I':
19765a0e3d6SJohn Marino if (pgrep)
19865a0e3d6SJohn Marino usage();
19965a0e3d6SJohn Marino interactive = 1;
20065a0e3d6SJohn Marino break;
20165a0e3d6SJohn Marino case 'L':
20265a0e3d6SJohn Marino pidfilelock = 1;
20365a0e3d6SJohn Marino break;
20465a0e3d6SJohn Marino case 'M':
20565a0e3d6SJohn Marino coref = optarg;
20665a0e3d6SJohn Marino break;
20765a0e3d6SJohn Marino case 'N':
20865a0e3d6SJohn Marino execf = optarg;
20965a0e3d6SJohn Marino break;
210df558cd9SChris Pressey case 'P':
21165a0e3d6SJohn Marino makelist(&ppidlist, LT_GENERIC, optarg);
212df558cd9SChris Pressey criteria = 1;
213df558cd9SChris Pressey break;
21465a0e3d6SJohn Marino case 'S':
21565a0e3d6SJohn Marino if (!pgrep)
21665a0e3d6SJohn Marino usage();
21765a0e3d6SJohn Marino kthreads = 1;
21865a0e3d6SJohn Marino break;
219*a2d71d01SLevente Kurusa case 'T':
220*a2d71d01SLevente Kurusa tdev = ttyname(0);
221*a2d71d01SLevente Kurusa if (tdev == NULL)
222*a2d71d01SLevente Kurusa err(STATUS_ERROR, "Not a terminal");
223*a2d71d01SLevente Kurusa if ((tdev = strdup(tdev)) == NULL)
224*a2d71d01SLevente Kurusa err(STATUS_ERROR, "Cannot allocate memory");
225*a2d71d01SLevente Kurusa makelist(&tdevlist, LT_TTY, tdev);
226*a2d71d01SLevente Kurusa criteria = 1;
227*a2d71d01SLevente Kurusa break;
228df558cd9SChris Pressey case 'U':
229df558cd9SChris Pressey makelist(&ruidlist, LT_USER, optarg);
230df558cd9SChris Pressey criteria = 1;
231df558cd9SChris Pressey break;
23265a0e3d6SJohn Marino case 'a':
23365a0e3d6SJohn Marino ancestors++;
23465a0e3d6SJohn Marino break;
23565a0e3d6SJohn Marino case 'c':
23665a0e3d6SJohn Marino makelist(&classlist, LT_CLASS, optarg);
23765a0e3d6SJohn Marino criteria = 1;
23865a0e3d6SJohn Marino break;
239df558cd9SChris Pressey case 'd':
240df558cd9SChris Pressey if (!pgrep)
241df558cd9SChris Pressey usage();
242df558cd9SChris Pressey delim = optarg;
243df558cd9SChris Pressey break;
244df558cd9SChris Pressey case 'f':
245df558cd9SChris Pressey matchargs = 1;
246df558cd9SChris Pressey break;
247df558cd9SChris Pressey case 'g':
248df558cd9SChris Pressey makelist(&pgrplist, LT_PGRP, optarg);
249df558cd9SChris Pressey criteria = 1;
250df558cd9SChris Pressey break;
25165a0e3d6SJohn Marino case 'i':
25265a0e3d6SJohn Marino cflags |= REG_ICASE;
25365a0e3d6SJohn Marino break;
25454499c33SChris Turner case 'j':
25554499c33SChris Turner makelist(&jidlist, LT_JID, optarg);
25654499c33SChris Turner criteria = 1;
25754499c33SChris Turner break;
258df558cd9SChris Pressey case 'l':
259df558cd9SChris Pressey longfmt = 1;
260df558cd9SChris Pressey break;
261df558cd9SChris Pressey case 'n':
262df558cd9SChris Pressey newest = 1;
263df558cd9SChris Pressey criteria = 1;
264df558cd9SChris Pressey break;
26565a0e3d6SJohn Marino case 'o':
26665a0e3d6SJohn Marino oldest = 1;
26765a0e3d6SJohn Marino criteria = 1;
26865a0e3d6SJohn Marino break;
26965a0e3d6SJohn Marino case 'q':
27065a0e3d6SJohn Marino if (!pgrep)
27165a0e3d6SJohn Marino usage();
27265a0e3d6SJohn Marino quiet = 1;
27365a0e3d6SJohn Marino break;
274df558cd9SChris Pressey case 's':
275df558cd9SChris Pressey makelist(&sidlist, LT_SID, optarg);
276df558cd9SChris Pressey criteria = 1;
277df558cd9SChris Pressey break;
278df558cd9SChris Pressey case 't':
279df558cd9SChris Pressey makelist(&tdevlist, LT_TTY, optarg);
280df558cd9SChris Pressey criteria = 1;
281df558cd9SChris Pressey break;
282df558cd9SChris Pressey case 'u':
283df558cd9SChris Pressey makelist(&euidlist, LT_USER, optarg);
284df558cd9SChris Pressey criteria = 1;
285df558cd9SChris Pressey break;
286df558cd9SChris Pressey case 'v':
287df558cd9SChris Pressey inverse = 1;
288df558cd9SChris Pressey break;
289df558cd9SChris Pressey case 'x':
290df558cd9SChris Pressey fullmatch = 1;
291df558cd9SChris Pressey break;
292df558cd9SChris Pressey default:
293df558cd9SChris Pressey usage();
294df558cd9SChris Pressey /* NOTREACHED */
295df558cd9SChris Pressey }
296*a2d71d01SLevente Kurusa }
297df558cd9SChris Pressey
298df558cd9SChris Pressey argc -= optind;
299df558cd9SChris Pressey argv += optind;
300df558cd9SChris Pressey if (argc != 0)
301df558cd9SChris Pressey criteria = 1;
302df558cd9SChris Pressey if (!criteria)
303df558cd9SChris Pressey usage();
30465a0e3d6SJohn Marino if (newest && oldest)
30565a0e3d6SJohn Marino errx(STATUS_ERROR, "Options -n and -o are mutually exclusive");
30665a0e3d6SJohn Marino if (pidfile != NULL)
30765a0e3d6SJohn Marino pidfromfile = takepid(pidfile, pidfilelock);
30865a0e3d6SJohn Marino else {
30965a0e3d6SJohn Marino if (pidfilelock) {
31065a0e3d6SJohn Marino errx(STATUS_ERROR,
31165a0e3d6SJohn Marino "Option -L doesn't make sense without -F");
31265a0e3d6SJohn Marino }
31365a0e3d6SJohn Marino pidfromfile = -1;
31465a0e3d6SJohn Marino }
315df558cd9SChris Pressey
316df558cd9SChris Pressey mypid = getpid();
317df558cd9SChris Pressey
318df558cd9SChris Pressey /*
319df558cd9SChris Pressey * Retrieve the list of running processes from the kernel.
320df558cd9SChris Pressey */
32165a0e3d6SJohn Marino kd = kvm_openfiles(execf, coref, NULL, O_RDONLY, buf);
322df558cd9SChris Pressey if (kd == NULL)
32365a0e3d6SJohn Marino errx(STATUS_ERROR, "Cannot open kernel files (%s)", buf);
324df558cd9SChris Pressey
3259a8c8446SMatthew Dillon if (pidfromfile >= 0)
3269a8c8446SMatthew Dillon plist = kvm_getprocs(kd, KERN_PROC_PID, pidfromfile, &nproc);
3279a8c8446SMatthew Dillon else
3284e3518d8SChris Pressey plist = kvm_getprocs(kd, KERN_PROC_ALL, 0, &nproc);
32965a0e3d6SJohn Marino if (plist == NULL) {
33065a0e3d6SJohn Marino errx(STATUS_ERROR, "Cannot get process list (%s)",
33165a0e3d6SJohn Marino kvm_geterr(kd));
33265a0e3d6SJohn Marino }
333df558cd9SChris Pressey
334df558cd9SChris Pressey /*
335df558cd9SChris Pressey * Allocate memory which will be used to keep track of the
336df558cd9SChris Pressey * selection.
337df558cd9SChris Pressey */
33865a0e3d6SJohn Marino if ((selected = malloc(nproc)) == NULL) {
33965a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot allocate memory for %d processes",
34065a0e3d6SJohn Marino nproc);
34165a0e3d6SJohn Marino }
342df558cd9SChris Pressey memset(selected, 0, nproc);
343df558cd9SChris Pressey
344df558cd9SChris Pressey /*
345df558cd9SChris Pressey * Refine the selection.
346df558cd9SChris Pressey */
347df558cd9SChris Pressey for (; *argv != NULL; argv++) {
34865a0e3d6SJohn Marino if ((rv = regcomp(®, *argv, cflags)) != 0) {
349df558cd9SChris Pressey regerror(rv, ®, buf, sizeof(buf));
35065a0e3d6SJohn Marino errx(STATUS_BADUSAGE,
35165a0e3d6SJohn Marino "Cannot compile regular expression `%s' (%s)",
35265a0e3d6SJohn Marino *argv, buf);
353df558cd9SChris Pressey }
354df558cd9SChris Pressey
355df558cd9SChris Pressey for (i = 0, kp = plist; i < nproc; i++, kp++) {
35665a0e3d6SJohn Marino if (PSKIP(kp)) {
35765a0e3d6SJohn Marino if (debug_opt > 0)
35865a0e3d6SJohn Marino fprintf(stderr, "* Skipped %5d %3d %s\n",
35965a0e3d6SJohn Marino kp->kp_pid, kp->kp_uid, kp->kp_comm);
360df558cd9SChris Pressey continue;
36165a0e3d6SJohn Marino }
362df558cd9SChris Pressey
36365a0e3d6SJohn Marino if (matchargs &&
36465a0e3d6SJohn Marino (pargv = kvm_getargv(kd, kp, 0)) != NULL) {
36565a0e3d6SJohn Marino jsz = 0;
36665a0e3d6SJohn Marino while (jsz < sizeof(buf) && *pargv != NULL) {
36765a0e3d6SJohn Marino jsz += snprintf(buf + jsz,
36865a0e3d6SJohn Marino sizeof(buf) - jsz,
369df558cd9SChris Pressey pargv[1] != NULL ? "%s " : "%s",
370df558cd9SChris Pressey pargv[0]);
371df558cd9SChris Pressey pargv++;
372df558cd9SChris Pressey }
373df558cd9SChris Pressey mstr = buf;
374df558cd9SChris Pressey } else
3755dfd06acSSimon Schubert mstr = kp->kp_comm;
376df558cd9SChris Pressey
377df558cd9SChris Pressey rv = regexec(®, mstr, 1, ®match, 0);
378df558cd9SChris Pressey if (rv == 0) {
379df558cd9SChris Pressey if (fullmatch) {
380df558cd9SChris Pressey if (regmatch.rm_so == 0 &&
38165a0e3d6SJohn Marino regmatch.rm_eo ==
38265a0e3d6SJohn Marino (off_t)strlen(mstr))
383df558cd9SChris Pressey selected[i] = 1;
384df558cd9SChris Pressey } else
385df558cd9SChris Pressey selected[i] = 1;
386df558cd9SChris Pressey } else if (rv != REG_NOMATCH) {
387df558cd9SChris Pressey regerror(rv, ®, buf, sizeof(buf));
38865a0e3d6SJohn Marino errx(STATUS_ERROR,
38965a0e3d6SJohn Marino "Regular expression evaluation error (%s)",
39065a0e3d6SJohn Marino buf);
39165a0e3d6SJohn Marino }
39265a0e3d6SJohn Marino if (debug_opt > 1) {
39365a0e3d6SJohn Marino const char *rv_res = "NoMatch";
39465a0e3d6SJohn Marino if (selected[i])
39565a0e3d6SJohn Marino rv_res = "Matched";
39665a0e3d6SJohn Marino fprintf(stderr, "* %s %5d %3d %s\n", rv_res,
39765a0e3d6SJohn Marino kp->kp_pid, kp->kp_uid, mstr);
398df558cd9SChris Pressey }
399df558cd9SChris Pressey }
400df558cd9SChris Pressey
401df558cd9SChris Pressey regfree(®);
402df558cd9SChris Pressey }
403df558cd9SChris Pressey
404df558cd9SChris Pressey for (i = 0, kp = plist; i < nproc; i++, kp++) {
40565a0e3d6SJohn Marino if (PSKIP(kp))
406df558cd9SChris Pressey continue;
407df558cd9SChris Pressey
40865a0e3d6SJohn Marino if (pidfromfile >= 0 && kp->kp_pid != pidfromfile) {
40965a0e3d6SJohn Marino selected[i] = 0;
41065a0e3d6SJohn Marino continue;
41195a53fe4SChris Pressey }
41265a0e3d6SJohn Marino
41365a0e3d6SJohn Marino SLIST_FOREACH(li, &ruidlist, li_chain)
41465a0e3d6SJohn Marino if (kp->kp_ruid == (uid_t)li->li_number)
41565a0e3d6SJohn Marino break;
416df558cd9SChris Pressey if (SLIST_FIRST(&ruidlist) != NULL && li == NULL) {
417df558cd9SChris Pressey selected[i] = 0;
418df558cd9SChris Pressey continue;
419df558cd9SChris Pressey }
420df558cd9SChris Pressey
42165a0e3d6SJohn Marino SLIST_FOREACH(li, &rgidlist, li_chain)
42265a0e3d6SJohn Marino if (kp->kp_rgid == (gid_t)li->li_number)
423df558cd9SChris Pressey break;
424df558cd9SChris Pressey if (SLIST_FIRST(&rgidlist) != NULL && li == NULL) {
425df558cd9SChris Pressey selected[i] = 0;
426df558cd9SChris Pressey continue;
427df558cd9SChris Pressey }
428df558cd9SChris Pressey
42965a0e3d6SJohn Marino SLIST_FOREACH(li, &euidlist, li_chain)
43065a0e3d6SJohn Marino if (kp->kp_uid == (uid_t)li->li_number)
431df558cd9SChris Pressey break;
432df558cd9SChris Pressey if (SLIST_FIRST(&euidlist) != NULL && li == NULL) {
433df558cd9SChris Pressey selected[i] = 0;
434df558cd9SChris Pressey continue;
435df558cd9SChris Pressey }
436df558cd9SChris Pressey
43765a0e3d6SJohn Marino SLIST_FOREACH(li, &ppidlist, li_chain)
43865a0e3d6SJohn Marino if (kp->kp_ppid == (pid_t)li->li_number)
439df558cd9SChris Pressey break;
440df558cd9SChris Pressey if (SLIST_FIRST(&ppidlist) != NULL && li == NULL) {
441df558cd9SChris Pressey selected[i] = 0;
442df558cd9SChris Pressey continue;
443df558cd9SChris Pressey }
444df558cd9SChris Pressey
44565a0e3d6SJohn Marino SLIST_FOREACH(li, &pgrplist, li_chain)
44665a0e3d6SJohn Marino if (kp->kp_pgid == (pid_t)li->li_number)
447df558cd9SChris Pressey break;
448df558cd9SChris Pressey if (SLIST_FIRST(&pgrplist) != NULL && li == NULL) {
449df558cd9SChris Pressey selected[i] = 0;
450df558cd9SChris Pressey continue;
451df558cd9SChris Pressey }
452df558cd9SChris Pressey
453df558cd9SChris Pressey SLIST_FOREACH(li, &tdevlist, li_chain) {
45465a0e3d6SJohn Marino if (li->li_number == -1 &&
4555dfd06acSSimon Schubert (kp->kp_flags & P_CONTROLT) == 0)
456df558cd9SChris Pressey break;
45765a0e3d6SJohn Marino if (kp->kp_tdev == (dev_t)li->li_number)
458df558cd9SChris Pressey break;
459df558cd9SChris Pressey }
460df558cd9SChris Pressey if (SLIST_FIRST(&tdevlist) != NULL && li == NULL) {
461df558cd9SChris Pressey selected[i] = 0;
462df558cd9SChris Pressey continue;
463df558cd9SChris Pressey }
464df558cd9SChris Pressey
46565a0e3d6SJohn Marino SLIST_FOREACH(li, &sidlist, li_chain)
46665a0e3d6SJohn Marino if (kp->kp_sid == (pid_t)li->li_number)
467df558cd9SChris Pressey break;
468df558cd9SChris Pressey if (SLIST_FIRST(&sidlist) != NULL && li == NULL) {
469df558cd9SChris Pressey selected[i] = 0;
470df558cd9SChris Pressey continue;
471df558cd9SChris Pressey }
472df558cd9SChris Pressey
47354499c33SChris Turner SLIST_FOREACH(li, &jidlist, li_chain) {
47454499c33SChris Turner /* A particular jail ID, including 0 (not in jail) */
47565a0e3d6SJohn Marino if (kp->kp_jailid == (int)li->li_number)
47654499c33SChris Turner break;
47754499c33SChris Turner /* Any jail */
47865a0e3d6SJohn Marino if (kp->kp_jailid > 0 && li->li_number == -1)
47954499c33SChris Turner break;
48054499c33SChris Turner }
48154499c33SChris Turner if (SLIST_FIRST(&jidlist) != NULL && li == NULL) {
48254499c33SChris Turner selected[i] = 0;
48354499c33SChris Turner continue;
48454499c33SChris Turner }
48554499c33SChris Turner
48665a0e3d6SJohn Marino SLIST_FOREACH(li, &classlist, li_chain) {
48765a0e3d6SJohn Marino /*
48865a0e3d6SJohn Marino * We skip P_SYSTEM processes to match ps(1) output.
48965a0e3d6SJohn Marino */
49065a0e3d6SJohn Marino if ((kp->kp_flags & P_SYSTEM) == 0)
49165a0e3d6SJohn Marino break;
49265a0e3d6SJohn Marino }
49365a0e3d6SJohn Marino if (SLIST_FIRST(&classlist) != NULL && li == NULL) {
49465a0e3d6SJohn Marino selected[i] = 0;
49565a0e3d6SJohn Marino continue;
49665a0e3d6SJohn Marino }
49765a0e3d6SJohn Marino
498df558cd9SChris Pressey if (argc == 0)
499df558cd9SChris Pressey selected[i] = 1;
500df558cd9SChris Pressey }
501df558cd9SChris Pressey
50265a0e3d6SJohn Marino if (!ancestors) {
50365a0e3d6SJohn Marino pid = mypid;
50465a0e3d6SJohn Marino while (pid) {
50565a0e3d6SJohn Marino for (i = 0, kp = plist; i < nproc; i++, kp++) {
50665a0e3d6SJohn Marino if (PSKIP(kp))
50765a0e3d6SJohn Marino continue;
50865a0e3d6SJohn Marino if (kp->kp_pid == pid) {
50965a0e3d6SJohn Marino selected[i] = 0;
51065a0e3d6SJohn Marino pid = kp->kp_ppid;
51165a0e3d6SJohn Marino break;
51265a0e3d6SJohn Marino }
51365a0e3d6SJohn Marino }
51465a0e3d6SJohn Marino if (i == nproc) {
51565a0e3d6SJohn Marino if (pid == mypid)
51665a0e3d6SJohn Marino pid = getppid();
51765a0e3d6SJohn Marino else
51865a0e3d6SJohn Marino break; /* Maybe we're in a jail ? */
51965a0e3d6SJohn Marino }
52065a0e3d6SJohn Marino }
52165a0e3d6SJohn Marino }
52265a0e3d6SJohn Marino
52365a0e3d6SJohn Marino if (newest || oldest) {
52465a0e3d6SJohn Marino best_tval.tv_sec = 0;
52565a0e3d6SJohn Marino best_tval.tv_usec = 0;
526df558cd9SChris Pressey bestidx = -1;
527df558cd9SChris Pressey
528df558cd9SChris Pressey for (i = 0, kp = plist; i < nproc; i++, kp++) {
529df558cd9SChris Pressey if (!selected[i])
530df558cd9SChris Pressey continue;
53165a0e3d6SJohn Marino if (bestidx == -1) {
53265a0e3d6SJohn Marino /* The first entry of the list which matched. */
53365a0e3d6SJohn Marino ;
53465a0e3d6SJohn Marino } else if (timercmp(&kp->kp_start, &best_tval, >)) {
53565a0e3d6SJohn Marino /* This entry is newer than previous "best". */
53665a0e3d6SJohn Marino if (oldest) /* but we want the oldest */
53765a0e3d6SJohn Marino continue;
53865a0e3d6SJohn Marino } else {
53965a0e3d6SJohn Marino /* This entry is older than previous "best". */
54065a0e3d6SJohn Marino if (newest) /* but we want the newest */
54165a0e3d6SJohn Marino continue;
542df558cd9SChris Pressey }
54365a0e3d6SJohn Marino /* This entry is better than previous "best" entry. */
54465a0e3d6SJohn Marino best_tval.tv_sec = kp->kp_start.tv_sec;
54565a0e3d6SJohn Marino best_tval.tv_usec = kp->kp_start.tv_usec;
54665a0e3d6SJohn Marino bestidx = i;
547df558cd9SChris Pressey }
548df558cd9SChris Pressey
549df558cd9SChris Pressey memset(selected, 0, nproc);
550df558cd9SChris Pressey if (bestidx != -1)
551df558cd9SChris Pressey selected[bestidx] = 1;
552df558cd9SChris Pressey }
553df558cd9SChris Pressey
554df558cd9SChris Pressey /*
555df558cd9SChris Pressey * Take the appropriate action for each matched process, if any.
556df558cd9SChris Pressey */
55765a0e3d6SJohn Marino did_action = 0;
55865a0e3d6SJohn Marino for (i = 0, rv = 0, kp = plist; i < nproc; i++, kp++) {
55965a0e3d6SJohn Marino if (PSKIP(kp))
560df558cd9SChris Pressey continue;
561df558cd9SChris Pressey if (selected[i]) {
56265a0e3d6SJohn Marino if (longfmt && !pgrep) {
56365a0e3d6SJohn Marino did_action = 1;
56465a0e3d6SJohn Marino printf("kill -%d %d\n", signum, kp->kp_pid);
56565a0e3d6SJohn Marino }
566df558cd9SChris Pressey if (inverse)
567df558cd9SChris Pressey continue;
568df558cd9SChris Pressey } else if (!inverse)
569df558cd9SChris Pressey continue;
57065a0e3d6SJohn Marino rv |= (*action)(kp);
571df558cd9SChris Pressey }
57265a0e3d6SJohn Marino if (!did_action && !pgrep && longfmt)
57365a0e3d6SJohn Marino fprintf(stderr,
57465a0e3d6SJohn Marino "No matching processes belonging to you were found\n");
5754cfac1e6SChris Pressey
576df558cd9SChris Pressey exit(rv ? STATUS_MATCH : STATUS_NOMATCH);
577df558cd9SChris Pressey }
578df558cd9SChris Pressey
57965a0e3d6SJohn Marino static void
usage(void)580df558cd9SChris Pressey usage(void)
581df558cd9SChris Pressey {
582df558cd9SChris Pressey const char *ustr;
583df558cd9SChris Pressey
584df558cd9SChris Pressey if (pgrep)
585*a2d71d01SLevente Kurusa ustr = "[-LSTfilnoqvx] [-d delim]";
586df558cd9SChris Pressey else
587*a2d71d01SLevente Kurusa ustr = "[-signal] [-ILTfilnovx]";
588df558cd9SChris Pressey
589df558cd9SChris Pressey fprintf(stderr,
59065a0e3d6SJohn Marino "usage: %s %s [-F pidfile] [-G gid] [-M core] [-N system]\n"
59165a0e3d6SJohn Marino " [-P ppid] [-U uid] [-c class] [-g pgrp] [-j jid]\n"
59265a0e3d6SJohn Marino " [-s sid] [-t tty] [-u euid] pattern ...\n",
59354499c33SChris Turner getprogname(), ustr);
594df558cd9SChris Pressey
59565a0e3d6SJohn Marino exit(STATUS_BADUSAGE);
596df558cd9SChris Pressey }
597df558cd9SChris Pressey
59865a0e3d6SJohn Marino static void
show_process(const struct kinfo_proc * kp)59965a0e3d6SJohn Marino show_process(const struct kinfo_proc *kp)
600df558cd9SChris Pressey {
601df558cd9SChris Pressey char **argv;
602df558cd9SChris Pressey
60365a0e3d6SJohn Marino if (quiet) {
60465a0e3d6SJohn Marino assert(pgrep);
605df558cd9SChris Pressey return;
60665a0e3d6SJohn Marino }
60765a0e3d6SJohn Marino if ((longfmt || !pgrep) && matchargs &&
60865a0e3d6SJohn Marino (argv = kvm_getargv(kd, kp, 0)) != NULL) {
6095dfd06acSSimon Schubert printf("%d ", (int)kp->kp_pid);
610df558cd9SChris Pressey for (; *argv != NULL; argv++) {
611df558cd9SChris Pressey printf("%s", *argv);
612df558cd9SChris Pressey if (argv[1] != NULL)
613df558cd9SChris Pressey putchar(' ');
614df558cd9SChris Pressey }
61565a0e3d6SJohn Marino } else if (longfmt || !pgrep)
6165dfd06acSSimon Schubert printf("%d %s", (int)kp->kp_pid, kp->kp_comm);
617df558cd9SChris Pressey else
6185dfd06acSSimon Schubert printf("%d", (int)kp->kp_pid);
619df558cd9SChris Pressey }
620df558cd9SChris Pressey
62165a0e3d6SJohn Marino static int
killact(const struct kinfo_proc * kp)62265a0e3d6SJohn Marino killact(const struct kinfo_proc *kp)
623a59669e4SChris Pressey {
62465a0e3d6SJohn Marino int ch, first;
625a59669e4SChris Pressey
62665a0e3d6SJohn Marino if (interactive) {
62765a0e3d6SJohn Marino /*
62865a0e3d6SJohn Marino * Be careful, ask before killing.
62965a0e3d6SJohn Marino */
63065a0e3d6SJohn Marino printf("kill ");
63165a0e3d6SJohn Marino show_process(kp);
63265a0e3d6SJohn Marino printf("? ");
63365a0e3d6SJohn Marino fflush(stdout);
63465a0e3d6SJohn Marino first = ch = getchar();
63565a0e3d6SJohn Marino while (ch != '\n' && ch != EOF)
63665a0e3d6SJohn Marino ch = getchar();
63765a0e3d6SJohn Marino if (first != 'y' && first != 'Y')
63865a0e3d6SJohn Marino return (1);
63965a0e3d6SJohn Marino }
64065a0e3d6SJohn Marino if (kill(kp->kp_pid, signum) == -1) {
64165a0e3d6SJohn Marino /*
64265a0e3d6SJohn Marino * Check for ESRCH, which indicates that the process
64365a0e3d6SJohn Marino * disappeared between us matching it and us
64465a0e3d6SJohn Marino * signalling it; don't issue a warning about it.
64565a0e3d6SJohn Marino */
64665a0e3d6SJohn Marino if (errno != ESRCH)
64765a0e3d6SJohn Marino warn("signalling pid %d", (int)kp->kp_pid);
64865a0e3d6SJohn Marino /*
64965a0e3d6SJohn Marino * Return 0 to indicate that the process should not be
65065a0e3d6SJohn Marino * considered a match, since we didn't actually get to
65165a0e3d6SJohn Marino * signal it.
65265a0e3d6SJohn Marino */
65365a0e3d6SJohn Marino return (0);
65465a0e3d6SJohn Marino }
65565a0e3d6SJohn Marino return (1);
656a59669e4SChris Pressey }
657a59669e4SChris Pressey
65865a0e3d6SJohn Marino static int
grepact(const struct kinfo_proc * kp)65965a0e3d6SJohn Marino grepact(const struct kinfo_proc *kp)
66065a0e3d6SJohn Marino {
66165a0e3d6SJohn Marino
66265a0e3d6SJohn Marino show_process(kp);
66365a0e3d6SJohn Marino if (!quiet)
66465a0e3d6SJohn Marino printf("%s", delim);
66565a0e3d6SJohn Marino return (1);
66665a0e3d6SJohn Marino }
66765a0e3d6SJohn Marino
66865a0e3d6SJohn Marino static void
makelist(struct listhead * head,enum listtype type,char * src)669df558cd9SChris Pressey makelist(struct listhead *head, enum listtype type, char *src)
670df558cd9SChris Pressey {
671df558cd9SChris Pressey struct list *li;
672df558cd9SChris Pressey struct passwd *pw;
673df558cd9SChris Pressey struct group *gr;
674df558cd9SChris Pressey struct stat st;
67565a0e3d6SJohn Marino const char *cp;
67665a0e3d6SJohn Marino char *sp, *ep, buf[MAXPATHLEN];
677df558cd9SChris Pressey int empty;
678df558cd9SChris Pressey
679df558cd9SChris Pressey empty = 1;
680df558cd9SChris Pressey
681df558cd9SChris Pressey while ((sp = strsep(&src, ",")) != NULL) {
682df558cd9SChris Pressey if (*sp == '\0')
683df558cd9SChris Pressey usage();
684df558cd9SChris Pressey
68565a0e3d6SJohn Marino if ((li = malloc(sizeof(*li))) == NULL) {
68665a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot allocate %zu bytes",
68765a0e3d6SJohn Marino sizeof(*li));
68865a0e3d6SJohn Marino }
68965a0e3d6SJohn Marino
690df558cd9SChris Pressey SLIST_INSERT_HEAD(head, li, li_chain);
691df558cd9SChris Pressey empty = 0;
692df558cd9SChris Pressey
69365a0e3d6SJohn Marino if (type != LT_CLASS)
69465a0e3d6SJohn Marino li->li_number = (uid_t)strtol(sp, &ep, 0);
69565a0e3d6SJohn Marino
69665a0e3d6SJohn Marino if (type != LT_CLASS && *ep == '\0') {
697df558cd9SChris Pressey switch (type) {
698df558cd9SChris Pressey case LT_PGRP:
69965a0e3d6SJohn Marino if (li->li_number == 0)
70065a0e3d6SJohn Marino li->li_number = getpgrp();
701df558cd9SChris Pressey break;
702df558cd9SChris Pressey case LT_SID:
70365a0e3d6SJohn Marino if (li->li_number == 0)
70465a0e3d6SJohn Marino li->li_number = getsid(mypid);
705df558cd9SChris Pressey break;
70654499c33SChris Turner case LT_JID:
70765a0e3d6SJohn Marino if (li->li_number < 0)
70854499c33SChris Turner errx(STATUS_BADUSAGE,
70954499c33SChris Turner "Negative jail ID `%s'", sp);
71065a0e3d6SJohn Marino /* For compatibility with old -j */
71165a0e3d6SJohn Marino if (li->li_number == 0)
71265a0e3d6SJohn Marino li->li_number = -1; /* any jail */
71354499c33SChris Turner break;
71465a0e3d6SJohn Marino case LT_TTY:
71565a0e3d6SJohn Marino if (li->li_number < 0)
716a59669e4SChris Pressey errx(STATUS_BADUSAGE,
71765a0e3d6SJohn Marino "Negative /dev/pts tty `%s'", sp);
71865a0e3d6SJohn Marino snprintf(buf, sizeof(buf), _PATH_DEV "pts/%s",
71965a0e3d6SJohn Marino sp);
72065a0e3d6SJohn Marino if (stat(buf, &st) != -1)
72165a0e3d6SJohn Marino goto foundtty;
72265a0e3d6SJohn Marino if (errno == ENOENT)
72365a0e3d6SJohn Marino errx(STATUS_BADUSAGE, "No such tty: `"
72465a0e3d6SJohn Marino _PATH_DEV "pts/%s'", sp);
72565a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot access `"
72665a0e3d6SJohn Marino _PATH_DEV "pts/%s'", sp);
72765a0e3d6SJohn Marino break;
72865a0e3d6SJohn Marino default:
72965a0e3d6SJohn Marino break;
730a59669e4SChris Pressey }
73165a0e3d6SJohn Marino continue;
732a59669e4SChris Pressey }
73365a0e3d6SJohn Marino
73465a0e3d6SJohn Marino switch (type) {
73565a0e3d6SJohn Marino case LT_USER:
73665a0e3d6SJohn Marino if ((pw = getpwnam(sp)) == NULL)
73765a0e3d6SJohn Marino errx(STATUS_BADUSAGE, "Unknown user `%s'", sp);
73865a0e3d6SJohn Marino li->li_number = pw->pw_uid;
739df558cd9SChris Pressey break;
740df558cd9SChris Pressey case LT_GROUP:
74165a0e3d6SJohn Marino if ((gr = getgrnam(sp)) == NULL)
74265a0e3d6SJohn Marino errx(STATUS_BADUSAGE, "Unknown group `%s'", sp);
74365a0e3d6SJohn Marino li->li_number = gr->gr_gid;
744df558cd9SChris Pressey break;
745df558cd9SChris Pressey case LT_TTY:
746df558cd9SChris Pressey if (strcmp(sp, "-") == 0) {
74765a0e3d6SJohn Marino li->li_number = -1;
748df558cd9SChris Pressey break;
74965a0e3d6SJohn Marino } else if (strcmp(sp, "co") == 0) {
75065a0e3d6SJohn Marino cp = "console";
75165a0e3d6SJohn Marino } else {
75265a0e3d6SJohn Marino cp = sp;
753df558cd9SChris Pressey }
754df558cd9SChris Pressey
755*a2d71d01SLevente Kurusa if (strncmp(cp, _PATH_DEV, sizeof(_PATH_DEV)-1) == 0 &&
756*a2d71d01SLevente Kurusa stat(cp, &st) != -1) {
757*a2d71d01SLevente Kurusa goto foundtty;
758*a2d71d01SLevente Kurusa }
759*a2d71d01SLevente Kurusa
76065a0e3d6SJohn Marino snprintf(buf, sizeof(buf), _PATH_DEV "%s", cp);
76165a0e3d6SJohn Marino if (stat(buf, &st) != -1)
76265a0e3d6SJohn Marino goto foundtty;
763df558cd9SChris Pressey
76465a0e3d6SJohn Marino snprintf(buf, sizeof(buf), _PATH_DEV "tty%s", cp);
76565a0e3d6SJohn Marino if (stat(buf, &st) != -1)
76665a0e3d6SJohn Marino goto foundtty;
76765a0e3d6SJohn Marino
76865a0e3d6SJohn Marino if (errno == ENOENT)
76965a0e3d6SJohn Marino errx(STATUS_BADUSAGE, "No such tty: `%s'", sp);
77065a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot access `%s'", sp);
77165a0e3d6SJohn Marino
77265a0e3d6SJohn Marino foundtty: if ((st.st_mode & S_IFCHR) == 0)
77365a0e3d6SJohn Marino errx(STATUS_BADUSAGE, "Not a tty: `%s'", sp);
77465a0e3d6SJohn Marino
77565a0e3d6SJohn Marino li->li_number = st.st_rdev;
77665a0e3d6SJohn Marino break;
77765a0e3d6SJohn Marino case LT_JID:
77865a0e3d6SJohn Marino if (strcmp(sp, "none") == 0)
77965a0e3d6SJohn Marino li->li_number = 0;
78065a0e3d6SJohn Marino else if (strcmp(sp, "any") == 0)
78165a0e3d6SJohn Marino li->li_number = -1;
78265a0e3d6SJohn Marino else if (*ep != '\0')
78365a0e3d6SJohn Marino errx(STATUS_BADUSAGE,
78465a0e3d6SJohn Marino "Invalid jail ID `%s'", sp);
78565a0e3d6SJohn Marino break;
78665a0e3d6SJohn Marino case LT_CLASS:
78765a0e3d6SJohn Marino li->li_number = -1;
78865a0e3d6SJohn Marino li->li_name = strdup(sp);
78965a0e3d6SJohn Marino if (li->li_name == NULL)
79065a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot allocate memory");
791df558cd9SChris Pressey break;
792df558cd9SChris Pressey default:
793df558cd9SChris Pressey usage();
79495a53fe4SChris Pressey }
795df558cd9SChris Pressey }
796df558cd9SChris Pressey
797df558cd9SChris Pressey if (empty)
798df558cd9SChris Pressey usage();
799df558cd9SChris Pressey }
80065a0e3d6SJohn Marino
80165a0e3d6SJohn Marino static int
takepid(const char * pidfile,int pidfilelock)80265a0e3d6SJohn Marino takepid(const char *pidfile, int pidfilelock)
80365a0e3d6SJohn Marino {
80465a0e3d6SJohn Marino char *endp, line[BUFSIZ];
80565a0e3d6SJohn Marino FILE *fh;
80665a0e3d6SJohn Marino long rval;
80765a0e3d6SJohn Marino
80865a0e3d6SJohn Marino fh = fopen(pidfile, "r");
80965a0e3d6SJohn Marino if (fh == NULL)
81065a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot open pidfile `%s'", pidfile);
81165a0e3d6SJohn Marino
81265a0e3d6SJohn Marino if (pidfilelock) {
81365a0e3d6SJohn Marino /*
81465a0e3d6SJohn Marino * If we can lock pidfile, this means that daemon is not
81565a0e3d6SJohn Marino * running, so would be better not to kill some random process.
81665a0e3d6SJohn Marino */
81765a0e3d6SJohn Marino if (flock(fileno(fh), LOCK_EX | LOCK_NB) == 0) {
81865a0e3d6SJohn Marino (void)fclose(fh);
81965a0e3d6SJohn Marino errx(STATUS_ERROR, "File '%s' can be locked", pidfile);
82065a0e3d6SJohn Marino } else {
82165a0e3d6SJohn Marino if (errno != EWOULDBLOCK) {
82265a0e3d6SJohn Marino errx(STATUS_ERROR,
82365a0e3d6SJohn Marino "Error while locking file '%s'", pidfile);
82465a0e3d6SJohn Marino }
82565a0e3d6SJohn Marino }
82665a0e3d6SJohn Marino }
82765a0e3d6SJohn Marino
82865a0e3d6SJohn Marino if (fgets(line, sizeof(line), fh) == NULL) {
82965a0e3d6SJohn Marino if (feof(fh)) {
83065a0e3d6SJohn Marino (void)fclose(fh);
83165a0e3d6SJohn Marino errx(STATUS_ERROR, "Pidfile `%s' is empty", pidfile);
83265a0e3d6SJohn Marino }
83365a0e3d6SJohn Marino (void)fclose(fh);
83465a0e3d6SJohn Marino err(STATUS_ERROR, "Cannot read from pid file `%s'", pidfile);
83565a0e3d6SJohn Marino }
83665a0e3d6SJohn Marino (void)fclose(fh);
83765a0e3d6SJohn Marino
83865a0e3d6SJohn Marino rval = strtol(line, &endp, 10);
83965a0e3d6SJohn Marino if (*endp != '\0' && !isspace((unsigned char)*endp))
84065a0e3d6SJohn Marino errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile);
84165a0e3d6SJohn Marino else if (rval < MIN_PID || rval > MAX_PID)
84265a0e3d6SJohn Marino errx(STATUS_ERROR, "Invalid pid in file `%s'", pidfile);
84365a0e3d6SJohn Marino return (rval);
84465a0e3d6SJohn Marino }
845