From 3b8a0d4f4f61931e616005151679110a605fb3a3 Mon Sep 17 00:00:00 2001 From: ad Date: Fri, 1 Mar 2002 11:21:58 +0000 Subject: [PATCH] pgrep and pkill: search the process table for processes with a particular set of characteristics, and either print or signal the matching set. Mostly compatible with the Solaris commands of the same name. --- usr.bin/pkill/Makefile | 13 + usr.bin/pkill/pkill.1 | 182 ++++++++++++++ usr.bin/pkill/pkill.c | 534 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 729 insertions(+) create mode 100644 usr.bin/pkill/Makefile create mode 100644 usr.bin/pkill/pkill.1 create mode 100644 usr.bin/pkill/pkill.c diff --git a/usr.bin/pkill/Makefile b/usr.bin/pkill/Makefile new file mode 100644 index 000000000000..1b4fc23fa503 --- /dev/null +++ b/usr.bin/pkill/Makefile @@ -0,0 +1,13 @@ +# $NetBSD: Makefile,v 1.1 2002/03/01 11:21:58 ad Exp $ + +PROG= pkill +SRCS= pkill.c +MAN= pkill.1 + +LDADD+= -lkvm +DPADD+= ${LIBKVM} + +LINKS+= ${BINDIR}/pkill ${BINDIR}/pgrep +MLINKS+=pkill.1 pgrep.1 + +.include diff --git a/usr.bin/pkill/pkill.1 b/usr.bin/pkill/pkill.1 new file mode 100644 index 000000000000..0a73796bebbd --- /dev/null +++ b/usr.bin/pkill/pkill.1 @@ -0,0 +1,182 @@ +.\" $NetBSD: pkill.1,v 1.1 2002/03/01 11:21:58 ad Exp $ +.\" +.\" Copyright (c) 2002 The NetBSD Foundation, Inc. +.\" All rights reserved. +.\" +.\" This code is derived from software contributed to The NetBSD Foundation +.\" by Andrew Doran. +.\" +.\" Redistribution and use in source and binary forms, with or without +.\" modification, are permitted provided that the following conditions +.\" are met: +.\" 1. Redistributions of source code must retain the above copyright +.\" notice, this list of conditions and the following disclaimer. +.\" 2. Redistributions in binary form must reproduce the above copyright +.\" notice, this list of conditions and the following disclaimer in the +.\" documentation and/or other materials provided with the distribution. +.\" 3. All advertising materials mentioning features or use of this software +.\" must display the following acknowledgement: +.\" This product includes software developed by the NetBSD +.\" Foundation, Inc. and its contributors. +.\" 4. Neither the name of The NetBSD Foundation nor the names of its +.\" contributors may be used to endorse or promote products derived +.\" from this software without specific prior written permission. +.\" +.\" THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS +.\" ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED +.\" TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR +.\" PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS +.\" BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +.\" CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF +.\" SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS +.\" INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN +.\" CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) +.\" ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE +.\" POSSIBILITY OF SUCH DAMAGE. +.\" +.Dd March 1, 2002 +.Dt PKILL 1 +.Os +.Sh NAME +.Nm pkill +.Nd find or signal processes by name +.Sh SYNOPSIS +.Nm pgrep +.Op Fl flnvx +.Op Fl G Ar gid +.Op Fl P Ar ppid +.Op Fl U Ar uid +.Op Fl d Ar delim +.Op Fl g Ar pgrp +.Op Fl s Ar sid +.Op Fl t Ar tty +.Op Fl u Ar euid +.Op Ar pattern Op ... +.Nm pkill +.Op Fl signal +.Op Fl fnvx +.Op Fl G Ar gid +.Op Fl P Ar ppid +.Op Fl U Ar uid +.Op Fl g Ar pgrp +.Op Fl s Ar sid +.Op Fl t Ar tty +.Op Fl u Ar euid +.Op Ar pattern Op ... +.Sh DESCRIPTION +The +.Nm pgrep +command searches the process table on the running system and prints the +process IDs of all processes that match the criteria given on the command +line. +.Pp +The +.Nm pkill +command searches the process table on the running system and signals all +processes that match the criteria given on the command line. +.Pp +The following options are available: +.Bl -tag -width binladen +.It Fl G Ar gid +Restrict matches to processes with a real group ID in the comma-seperated +list +.Ar gid . +.It Fl P Ar ppid +Restrict matches to processes with a parent process ID in the +comma-seperated list +.Ar ppid . +.It Fl U Ar uid +Restrict matches to processes with a real user ID in the comma-seperated +list +.Ar uid . +.It Fl d Ar delim +Specify a delimiter to be printed between each process ID. +The default is a newline. +This option can only be used with the +.Nm pgrep +command. +.It Fl f +Match against full argument lists. +The default is to match against process names. +.It Fl g Ar pgrp +Restrict matches to processes with a process group ID in the comma-seperated +list +.Ar pgrp . +The value zero is taken to mean the process group ID of the running +.Nm pgrep +or +.Nm pkill +command. +.It Fl l +Long output. +Print the process name in addition to the process ID for each matching +process. +If used in conjunction with +.Fl f , +print the process ID and the full argument list for each matching process. +This option can only be used with the +.Nm pgrep +command. +.It Fl n +Match only the most recently created process, if any. +.It Fl s Ar sid +Restrict matches to processes with a session ID in the comma-seperated +list +.Ar sid . +The value zero is taken to mean the session ID of the running +.Nm pgrep +or +.Nm pkill +command. +.It Fl t Ar tty +Restrict matches to processes associated with a terminal in the +comma-seperated list +.Ar tty . +Terminal names may be of the form `ttyxx' or the shortened form `xx'. +.It Fl u Ar euid +Restrict matches to processes with an effective user ID in the +comma-seperated list +.Ar euid . +.It Fl v +Reverse the sense of the matching; display processes that do not match the +given criteria. +.It Fl x +Require an exact match of the process name, or argument list if -f is given. +The default is to match any substring. +.It Fl signal +A non-negative decimal number or symbolic signal name specifying the signal +to be sent instead of the default TERM. +This option is valid only when given as the first argument to +.Nm pkill . +.El +.Pp +Note that a running +.Nm pgrep +or +.Nm pkill +process will never consider itself a potential match. +.Sh EXIT STATUS +.Nm pgrep +and +.Nm pkill +return one of the following values upon exit: +.Bl -tag -width foo +.It 0 +One or more processes were matched. +.It 1 +No processes were matched. +.It 2 +Invalid options were specified on the command line. +.It 3 +An internal error occurred. +.El +.Sh SEE ALSO +.Xr kill 1 , +.Xr ps 1 , +.Xr kill 2 , +.Xr sigaction 2 , +.Xr re_format 7 , +.Xr signal 7 +.Sh AUTHORS +.An Andrew Doran +.Aq ad@netbsd.org . diff --git a/usr.bin/pkill/pkill.c b/usr.bin/pkill/pkill.c new file mode 100644 index 000000000000..1d1ddcd8ae37 --- /dev/null +++ b/usr.bin/pkill/pkill.c @@ -0,0 +1,534 @@ +/* $NetBSD: pkill.c,v 1.1 2002/03/01 11:21:58 ad Exp $ */ + +/*- + * Copyright (c) 2002 The NetBSD Foundation, Inc. + * All rights reserved. + * + * This code is derived from software contributed to The NetBSD Foundation + * by Andrew Doran. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. All advertising materials mentioning features or use of this software + * must display the following acknowledgement: + * This product includes software developed by the NetBSD + * Foundation, Inc. and its contributors. + * 4. Neither the name of The NetBSD Foundation nor the names of its + * contributors may be used to endorse or promote products derived + * from this software without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS + * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED + * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +#include +#ifndef LINT +__RCSID("$NetBSD: pkill.c,v 1.1 2002/03/01 11:21:58 ad Exp $"); +#endif /* !LINT */ + +#include +#include +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define STATUS_MATCH 0 +#define STATUS_NOMATCH 1 +#define STATUS_BADUSAGE 2 +#define STATUS_ERROR 3 + +enum listtype { + LT_GENERIC, + LT_USER, + LT_GROUP, + LT_TTY, + LT_PGRP, + LT_SID +}; + +struct list { + SLIST_ENTRY(list) li_chain; + long li_number; +}; + +SLIST_HEAD(listhead, list); + +struct kinfo_proc2 *plist; +char *selected; +char *delim = "\n"; +int nproc; +int pgrep; +int signum = SIGTERM; +int newest; +int inverse; +int longfmt; +int matchargs; +int fullmatch; +kvm_t *kd; +pid_t mypid; + +struct listhead euidlist = SLIST_HEAD_INITIALIZER(list); +struct listhead ruidlist = SLIST_HEAD_INITIALIZER(list); +struct listhead rgidlist = SLIST_HEAD_INITIALIZER(list); +struct listhead pgrplist = SLIST_HEAD_INITIALIZER(list); +struct listhead ppidlist = SLIST_HEAD_INITIALIZER(list); +struct listhead tdevlist = SLIST_HEAD_INITIALIZER(list); +struct listhead sidlist = SLIST_HEAD_INITIALIZER(list); + +int main(int, char **); +void usage(void); +void killact(struct kinfo_proc2 *); +void grepact(struct kinfo_proc2 *); +void makelist(struct listhead *, enum listtype, char *); + +int +main(int argc, char **argv) +{ + extern char *optarg; + extern int optind; + char buf[_POSIX2_LINE_MAX], *mstr, **pargv, *p; + int i, j, ch, bestidx, rv, criteria; + void (*action)(struct kinfo_proc2 *); + struct kinfo_proc2 *kp; + struct list *li; + u_int32_t bestsec, bestusec; + regex_t reg; + regmatch_t regmatch; + + if (strcmp(getprogname(), "pgrep") == 0) { + action = grepact; + pgrep = 1; + } else { + action = killact; + p = argv[1]; + + if (p[0] == '-') { + p++; + i = (int)strtol(p, &p, 10); + if (*p == '\0') { + signum = i; + argv++; + } else { + if (strncasecmp(p, "sig", 3) == 0) + p += 3; + for (i = 1; i < NSIG; i++) + if (strcasecmp(sys_signame[i], p) == 0) + break; + if (i != NSIG) { + signum = i; + argv++; + } + } + } + } + + criteria = 0; + + while ((ch = getopt(argc, argv, "G:P:U:d:fg:lns:t:vx")) != -1) + switch (ch) { + case 'G': + makelist(&rgidlist, LT_GROUP, optarg); + criteria = 1; + break; + case 'P': + makelist(&ppidlist, LT_GENERIC, optarg); + criteria = 1; + break; + case 'U': + makelist(&ruidlist, LT_USER, optarg); + criteria = 1; + break; + case 'd': + if (!pgrep) + usage(); + delim = optarg; + break; + case 'f': + matchargs = 1; + break; + case 'g': + makelist(&pgrplist, LT_PGRP, optarg); + criteria = 1; + break; + case 'l': + if (!pgrep) + usage(); + longfmt = 1; + break; + case 'n': + newest = 1; + criteria = 1; + break; + case 's': + makelist(&sidlist, LT_SID, optarg); + criteria = 1; + break; + case 't': + makelist(&tdevlist, LT_TTY, optarg); + criteria = 1; + break; + case 'u': + makelist(&euidlist, LT_USER, optarg); + criteria = 1; + break; + case 'v': + inverse = 1; + break; + case 'x': + fullmatch = 1; + break; + default: + usage(); + /* NOTREACHED */ + } + + argc -= optind; + argv += optind; + if (argc != 0) + criteria = 1; + if (!criteria) + usage(); + + mypid = getpid(); + + /* + * Retrieve the list of running processes from the kernel. + */ + kd = kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, buf); + if (kd == NULL) + errx(STATUS_ERROR, "kvm_openfiles(): %s", buf); + + plist = kvm_getproc2(kd, KERN_PROC_ALL, 0, sizeof(*plist), &nproc); + if (plist == NULL) + errx(STATUS_ERROR, "kvm_getprocs2() failed"); + + /* + * Allocate memory which will be used to keep track of the + * selection. + */ + if ((selected = malloc(nproc)) == NULL) + errx(STATUS_ERROR, "memory allocation failure"); + memset(selected, 0, nproc); + + /* + * Refine the selection. + */ + for (; *argv != NULL; argv++) { + if ((rv = regcomp(®, *argv, REG_EXTENDED)) != 0) { + regerror(rv, ®, buf, sizeof(buf)); + errx(STATUS_BADUSAGE, "bad expression: %s", buf); + } + + for (i = 0, kp = plist; i < nproc; i++, kp++) { + if ((kp->p_flag & P_SYSTEM) != 0) + continue; + + if (matchargs) { + if ((pargv = kvm_getargv2(kd, kp, 0)) == NULL) + continue; + + j = 0; + while (j < sizeof(buf) && *pargv != NULL) { + j += snprintf(buf + j, sizeof(buf) - j, + pargv[1] != NULL ? "%s " : "%s", + pargv[0]); + pargv++; + } + + mstr = buf; + } else + mstr = kp->p_comm; + + rv = regexec(®, mstr, 1, ®match, 0); + if (rv == 0) { + if (fullmatch) { + if (regmatch.rm_so == 0 && + regmatch.rm_eo == strlen(mstr)) + selected[i] = 1; + } else + selected[i] = 1; + } else if (rv != REG_NOMATCH) { + regerror(rv, ®, buf, sizeof(buf)); + errx(STATUS_ERROR, "regexec(): %s", buf); + } + } + + regfree(®); + } + + for (i = 0, kp = plist; i < nproc; i++, kp++) { + if ((kp->p_flag & P_SYSTEM) != 0) + continue; + + SLIST_FOREACH(li, &ruidlist, li_chain) + if (kp->p_ruid == (uid_t)li->li_number) + break; + if (SLIST_FIRST(&ruidlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &rgidlist, li_chain) + if (kp->p_rgid == (gid_t)li->li_number) + break; + if (SLIST_FIRST(&rgidlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &euidlist, li_chain) + if (kp->p_uid == (uid_t)li->li_number) + break; + if (SLIST_FIRST(&euidlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &ppidlist, li_chain) + if (kp->p_ppid == (uid_t)li->li_number) + break; + if (SLIST_FIRST(&ppidlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &pgrplist, li_chain) + if (kp->p__pgid == (uid_t)li->li_number) + break; + if (SLIST_FIRST(&pgrplist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &tdevlist, li_chain) { + if (li->li_number == -1 && + (kp->p_flag & P_CONTROLT) == 0) + break; + if (kp->p_tdev == (uid_t)li->li_number) + break; + } + if (SLIST_FIRST(&tdevlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + SLIST_FOREACH(li, &sidlist, li_chain) + if (kp->p_sid == (uid_t)li->li_number) + break; + if (SLIST_FIRST(&sidlist) != NULL && li == NULL) { + selected[i] = 0; + continue; + } + + if (argc == 0) + selected[i] = 1; + } + + if (newest) { + bestsec = 0; + bestusec = 0; + bestidx = -1; + + for (i = 0, kp = plist; i < nproc; i++, kp++) { + if (!selected[i]) + continue; + + if (kp->p_ustart_sec > bestsec || + (kp->p_ustart_sec == bestsec + && kp->p_ustart_usec > bestusec)) { + bestsec = kp->p_ustart_sec; + bestusec = kp->p_ustart_usec; + bestidx = i; + } + } + + memset(selected, 0, nproc); + if (bestidx != -1) + selected[bestidx] = 1; + } + + /* + * Take the appropriate action for each matched process, if any. + */ + for (i = 0, rv = 0, kp = plist; i < nproc; i++, kp++) { + if (kp->p_pid == mypid) + continue; + if (selected[i]) { + if (inverse) + continue; + } else if (!inverse) + continue; + + rv = 1; + (*action)(kp); + } + + exit(rv ? STATUS_MATCH : STATUS_NOMATCH); +} + +void +usage(void) +{ + const char *ustr; + + if (pgrep) + ustr = "[-flnvx] [-d delim]"; + else + ustr = "[-signal] [-fnvx]"; + + fprintf(stderr, "usage: %s %s [-G gid] [-P ppid] [-U uid] [-g pgrp] " + "[-s sid] [-t tty] [-u euid] pattern ...\n", getprogname(), ustr); + + exit(STATUS_ERROR); +} + +void +killact(struct kinfo_proc2 *kp) +{ + + if (kill(kp->p_pid, signum) == -1) + err(STATUS_ERROR, "signalling pid %d", (int)kp->p_pid); +} + +void +grepact(struct kinfo_proc2 *kp) +{ + char **argv; + + if (longfmt && matchargs) { + if ((argv = kvm_getargv2(kd, kp, 0)) == NULL) + return; + + printf("%d ", (int)kp->p_pid); + for (; *argv != NULL; argv++) { + printf("%s", *argv); + if (argv[1] != NULL) + putchar(' '); + } + } else if (longfmt) + printf("%d %s", (int)kp->p_pid, kp->p_comm); + else + printf("%d", (int)kp->p_pid); + + printf("%s", delim); +} + +void +makelist(struct listhead *head, enum listtype type, char *src) +{ + struct list *li; + struct passwd *pw; + struct group *gr; + struct stat st; + char *sp, *p, buf[MAXPATHLEN]; + int empty; + + empty = 1; + + while ((sp = strsep(&src, ",")) != NULL) { + if (*sp == '\0') + usage(); + + if ((li = malloc(sizeof(*li))) == NULL) + errx(STATUS_ERROR, "memory allocation failure"); + SLIST_INSERT_HEAD(head, li, li_chain); + empty = 0; + + li->li_number = (uid_t)strtol(sp, &p, 0); + if (*p == '\0') { + switch (type) { + case LT_PGRP: + if (li->li_number == 0) + li->li_number = getpgrp(); + break; + case LT_SID: + if (li->li_number == 0) + li->li_number = getsid(mypid); + break; + case LT_TTY: + usage(); + default: + break; + } + continue; + } + + switch (type) { + case LT_USER: + if ((pw = getpwnam(sp)) == NULL) + errx(STATUS_BADUSAGE, "unknown user `%s'", + optarg); + li->li_number = pw->pw_uid; + break; + case LT_GROUP: + if ((gr = getgrnam(sp)) == NULL) + errx(STATUS_BADUSAGE, "unknown group `%s'", + optarg); + li->li_number = gr->gr_gid; + break; + case LT_TTY: + if (strcmp(sp, "-") == 0) { + li->li_number = -1; + break; + } else if (strcmp(sp, "co") == 0) + p = "console"; + else if (strncmp(sp, "tty", 3) == 0) + p = sp; + else + p = NULL; + + if (p == NULL) + snprintf(buf, sizeof(buf), "/dev/tty%s", sp); + else + snprintf(buf, sizeof(buf), "/dev/%s", p); + + if (stat(buf, &st) < 0) { + if (errno == ENOENT) + errx(STATUS_BADUSAGE, + "no such tty: `%s'", sp); + err(STATUS_ERROR, "stat(%s)", sp); + } + + if ((st.st_mode & S_IFCHR) == 0) + errx(STATUS_BADUSAGE, "not a tty: `%s'", sp); + + li->li_number = st.st_rdev; + break; + default: + usage(); + }; + } + + if (empty) + usage(); +}