/* vi: set sw=4 ts=4: */ /* * Mini pgrep/pkill implementation for busybox * * Copyright (C) 2007 Loic Grenie * * Licensed under GPLv2 or later, see file LICENSE in this source tree. */ //config:config PGREP //config: bool "pgrep (6.5 kb)" //config: default y //config: help //config: Look for processes by name. //config: //config:config PKILL //config: bool "pkill (7.5 kb)" //config: default y //config: help //config: Send signals to processes by name. //applet:IF_PGREP(APPLET_ODDNAME(pgrep, pgrep, BB_DIR_USR_BIN, BB_SUID_DROP, pgrep)) // APPLET_ODDNAME:name main location suid_type help //applet:IF_PKILL(APPLET_ODDNAME(pkill, pgrep, BB_DIR_USR_BIN, BB_SUID_DROP, pkill)) /* can't be noexec: can find _itself_ under wrong name, since after fork only, * /proc/PID/cmdline and comm are wrong! Can fix comm (prctl(PR_SET_NAME)), * but cmdline? */ //kbuild:lib-$(CONFIG_PGREP) += pgrep.o //kbuild:lib-$(CONFIG_PKILL) += pgrep.o //usage:#define pgrep_trivial_usage //usage: "[-flanovx] [-s SID|-P PPID|PATTERN]" //usage:#define pgrep_full_usage "\n\n" //usage: "Display process(es) selected by regex PATTERN\n" //usage: "\n -l Show command name too" //usage: "\n -a Show command line too" //usage: "\n -f Match against entire command line" //usage: "\n -n Show the newest process only" //usage: "\n -o Show the oldest process only" //usage: "\n -v Negate the match" //usage: "\n -x Match whole name (not substring)" //usage: "\n -s Match session ID (0 for current)" //usage: "\n -P Match parent process ID" //usage: //usage:#define pkill_trivial_usage //usage: "[-l|-SIGNAL] [-fnovx] [-s SID|-P PPID|PATTERN]" //usage:#define pkill_full_usage "\n\n" //usage: "Send a signal to process(es) selected by regex PATTERN\n" //usage: "\n -l List all signals" //usage: "\n -f Match against entire command line" //usage: "\n -n Signal the newest process only" //usage: "\n -o Signal the oldest process only" //usage: "\n -v Negate the match" //usage: "\n -x Match whole name (not substring)" //usage: "\n -s Match session ID (0 for current)" //usage: "\n -P Match parent process ID" #include "libbb.h" #include "xregex.h" /* Idea taken from kill.c */ #define pgrep (ENABLE_PGREP && (!ENABLE_PKILL || applet_name[1] == 'g')) #define pkill (ENABLE_PKILL && (!ENABLE_PGREP || applet_name[1] == 'k')) enum { /* "vlafxons:+P:+" */ OPTBIT_V = 0, /* must be first, we need OPT_INVERT = 0/1 */ OPTBIT_L, OPTBIT_A, OPTBIT_F, OPTBIT_X, OPTBIT_O, OPTBIT_N, OPTBIT_S, OPTBIT_P, }; #define OPT_INVERT (opt & (1 << OPTBIT_V)) #define OPT_LIST (opt & (1 << OPTBIT_L)) #define OPT_LISTFULL (opt & (1 << OPTBIT_A)) #define OPT_FULL (opt & (1 << OPTBIT_F)) #define OPT_ANCHOR (opt & (1 << OPTBIT_X)) #define OPT_FIRST (opt & (1 << OPTBIT_O)) #define OPT_LAST (opt & (1 << OPTBIT_N)) #define OPT_SID (opt & (1 << OPTBIT_S)) #define OPT_PPID (opt & (1 << OPTBIT_P)) static void act(unsigned pid, char *cmd, int signo) { if (pgrep) { if (option_mask32 & ((1 << OPTBIT_L)|(1 << OPTBIT_A))) /* -l or -a */ printf("%u %s\n", pid, cmd); else printf("%u\n", pid); } else kill(pid, signo); } int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int pgrep_main(int argc UNUSED_PARAM, char **argv) { unsigned pid; int signo; unsigned opt; int scan_mask; int matched_pid; int sid2match, ppid2match; char *cmd_last; procps_status_t *proc; /* These are initialized to 0 */ struct { regex_t re_buffer; regmatch_t re_match[1]; } Z; #define re_buffer (Z.re_buffer) #define re_match (Z.re_match ) memset(&Z, 0, sizeof(Z)); /* Parse -SIGNAL for pkill. Must be first option, if present */ signo = SIGTERM; if (pkill && argv[1] && argv[1][0] == '-') { int temp = get_signum(argv[1]+1); if (temp != -1) { signo = temp; argv++; } } /* Parse remaining options */ ppid2match = -1; sid2match = -1; opt = getopt32(argv, "vlafxons:+P:+", &sid2match, &ppid2match); argv += optind; if (pkill && OPT_LIST) { /* -l: print the whole signal list */ print_signames(); return 0; } pid = getpid(); if (sid2match == 0) sid2match = getsid(pid); scan_mask = PSSCAN_COMM | PSSCAN_ARGV0; if (OPT_FULL) scan_mask |= PSSCAN_ARGVN; /* One pattern is required, if no -s and no -P */ if ((sid2match & ppid2match) < 0 && (!argv[0] || argv[1])) bb_show_usage(); if (argv[0]) xregcomp(&re_buffer, argv[0], OPT_ANCHOR ? REG_EXTENDED : (REG_EXTENDED|REG_NOSUB)); matched_pid = 0; cmd_last = NULL; proc = NULL; while ((proc = procps_scan(proc, scan_mask)) != NULL) { char *cmd; int cmdlen, match; if (proc->pid == pid) continue; if (!OPT_INVERT) { /* Quickly reject -sN -PN mismatches... unless -v */ if (ppid2match >= 0 && ppid2match != proc->ppid) continue; if (sid2match >= 0 && sid2match != proc->sid) continue; } cmdlen = -1; cmd = proc->argv0; if (!cmd) { cmd = proc->comm; } else { int i = proc->argv_len; if (!OPT_LISTFULL) cmdlen = strlen(cmd); /* not -a: find first NUL */ /* * "sleep 11" looks like "sleep""\0""11""\0" in argv0. * Make sure last "\0" does not get converted to " ": */ if (i && cmd[i-1] == '\0') i--; while (--i >= 0) { if ((unsigned char)cmd[i] < ' ') cmd[i] = ' '; } } if (OPT_INVERT) { /* "pgrep -v -P1 firefox" means "not (ppid=1 AND name=firefox)" * or equivalently "ppid!=1 OR name!=firefox". * Check the first condition and if true, skip matching. */ if (ppid2match >= 0 && ppid2match != proc->ppid) goto got_it; if (sid2match >= 0 && sid2match != proc->sid) goto got_it; } match = !argv[0]; /* if no PATTERN, then it's a match, else... */ if (!match) { again: match = (regexec(&re_buffer, cmd, 1, re_match, 0) == 0); if (!match && cmd != proc->comm) { /* if argv[] did not match, try comm */ cmdlen = -1; cmd = proc->comm; goto again; } if (match && OPT_ANCHOR) { /* -x requires full string match */ match = (re_match[0].rm_so == 0 && cmd[re_match[0].rm_eo] == '\0'); } } /* NB: OPT_INVERT is always 0 or 1 */ if (match ^ OPT_INVERT) { got_it: matched_pid = proc->pid; if (OPT_LAST) { free(cmd_last); cmd_last = xstrdup(cmd); continue; } if (cmdlen >= 0) cmd[cmdlen] = '\0'; act(proc->pid, cmd, signo); if (OPT_FIRST) break; } } if (cmd_last) { act(matched_pid, cmd_last, signo); if (ENABLE_FEATURE_CLEAN_UP) free(cmd_last); } return matched_pid == 0; /* return 1 if no processes listed/signaled */ }