pgrep/pkill: support -s and -P options

function                                             old     new   delta
pgrep_main                                           510     580     +70
packed_usage                                       26575   26616     +41
act                                                  234     236      +2
------------------------------------------------------------------------------
(add/remove: 0/0 grow/shrink: 3/0 up/down: 113/0)             Total: 113 bytes

Signed-off-by: Rob Landley <rob@landley.net>
Signed-off-by: Denys Vlasenko <vda.linux@googlemail.com>
This commit is contained in:
Denys Vlasenko 2009-07-06 00:00:12 +02:00
parent 666e1d3978
commit 57dc5345e2
2 changed files with 75 additions and 61 deletions

View File

@ -3229,7 +3229,7 @@
"$ patch -p0 -i example.diff" "$ patch -p0 -i example.diff"
#define pgrep_trivial_usage \ #define pgrep_trivial_usage \
"[-flnovx] PATTERN" "[-flnovx] [-s SID|-P PPID|PATTERN]"
#define pgrep_full_usage "\n\n" \ #define pgrep_full_usage "\n\n" \
"Display process(es) selected by regex PATTERN\n" \ "Display process(es) selected by regex PATTERN\n" \
"\nOptions:" \ "\nOptions:" \
@ -3237,8 +3237,10 @@
"\n -f Match against entire command line" \ "\n -f Match against entire command line" \
"\n -n Show the newest process only" \ "\n -n Show the newest process only" \
"\n -o Show the oldest process only" \ "\n -o Show the oldest process only" \
"\n -v Negate the matching" \ "\n -v Negate the match" \
"\n -x Match whole name (not substring)" \ "\n -x Match whole name (not substring)" \
"\n -s Match session ID (0 for current)" \
"\n -P Match parent process ID" \
#if (ENABLE_FEATURE_PIDOF_SINGLE || ENABLE_FEATURE_PIDOF_OMIT) #if (ENABLE_FEATURE_PIDOF_SINGLE || ENABLE_FEATURE_PIDOF_OMIT)
#define pidof_trivial_usage \ #define pidof_trivial_usage \
@ -3331,7 +3333,7 @@
"the new root file system" "the new root file system"
#define pkill_trivial_usage \ #define pkill_trivial_usage \
"[-l] | [-fnovx] [-signal] PATTERN" "[-l|-SIGNAL] [-fnovx] [-s SID|-P PPID|PATTERN]"
#define pkill_full_usage "\n\n" \ #define pkill_full_usage "\n\n" \
"Send a signal to process(es) selected by regex PATTERN\n" \ "Send a signal to process(es) selected by regex PATTERN\n" \
"\nOptions:" \ "\nOptions:" \
@ -3339,8 +3341,10 @@
"\n -f Match against entire command line" \ "\n -f Match against entire command line" \
"\n -n Signal the newest process only" \ "\n -n Signal the newest process only" \
"\n -o Signal the oldest process only" \ "\n -o Signal the oldest process only" \
"\n -v Negate the matching" \ "\n -v Negate the match" \
"\n -x Match whole name (not substring)" \ "\n -x Match whole name (not substring)" \
"\n -s Match session ID (0 for current)" \
"\n -P Match parent process ID" \
#define popmaildir_trivial_usage \ #define popmaildir_trivial_usage \
"[OPTIONS] Maildir [connection-helper ...]" "[OPTIONS] Maildir [connection-helper ...]"

View File

@ -6,7 +6,6 @@
* *
* Licensed under the GPL v2 or later, see the file LICENSE in this tarball. * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
*/ */
#include "libbb.h" #include "libbb.h"
#include "xregex.h" #include "xregex.h"
@ -15,26 +14,30 @@
#define pkill (ENABLE_PKILL && applet_name[1] == 'k') #define pkill (ENABLE_PKILL && applet_name[1] == 'k')
enum { enum {
/* "vlfxon" */ /* "vlfxons:P:" */
PGREPOPTBIT_V = 0, /* must be first, we need OPT_INVERT = 0/1 */ OPTBIT_V = 0, /* must be first, we need OPT_INVERT = 0/1 */
PGREPOPTBIT_L, OPTBIT_L,
PGREPOPTBIT_F, OPTBIT_F,
PGREPOPTBIT_X, OPTBIT_X,
PGREPOPTBIT_O, OPTBIT_O,
PGREPOPTBIT_N, OPTBIT_N,
OPTBIT_S,
OPTBIT_P,
}; };
#define OPT_INVERT (opt & (1 << PGREPOPTBIT_V)) #define OPT_INVERT (opt & (1 << OPTBIT_V))
#define OPT_LIST (opt & (1 << PGREPOPTBIT_L)) #define OPT_LIST (opt & (1 << OPTBIT_L))
#define OPT_FULL (opt & (1 << PGREPOPTBIT_F)) #define OPT_FULL (opt & (1 << OPTBIT_F))
#define OPT_ANCHOR (opt & (1 << PGREPOPTBIT_X)) #define OPT_ANCHOR (opt & (1 << OPTBIT_X))
#define OPT_FIRST (opt & (1 << PGREPOPTBIT_O)) #define OPT_FIRST (opt & (1 << OPTBIT_O))
#define OPT_LAST (opt & (1 << PGREPOPTBIT_N)) #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, unsigned opt) static void act(unsigned pid, char *cmd, int signo)
{ {
if (pgrep) { if (pgrep) {
if (OPT_LIST) if (option_mask32 & (1 << OPTBIT_L)) /* OPT_LIST */
printf("%d %s\n", pid, cmd); printf("%d %s\n", pid, cmd);
else else
printf("%d\n", pid); printf("%d\n", pid);
@ -45,13 +48,12 @@ static void act(unsigned pid, char *cmd, int signo, unsigned opt)
int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int pgrep_main(int argc UNUSED_PARAM, char **argv) int pgrep_main(int argc UNUSED_PARAM, char **argv)
{ {
unsigned pid = getpid(); unsigned pid;
int signo = SIGTERM; int signo;
unsigned opt; unsigned opt;
int scan_mask = PSSCAN_COMM; int scan_mask;
char *first_arg;
int first_arg_idx;
int matched_pid; int matched_pid;
int sid2match, ppid2match;
char *cmd_last; char *cmd_last;
procps_status_t *proc; procps_status_t *proc;
/* These are initialized to 0 */ /* These are initialized to 0 */
@ -64,52 +66,52 @@ int pgrep_main(int argc UNUSED_PARAM, char **argv)
memset(&Z, 0, sizeof(Z)); memset(&Z, 0, sizeof(Z));
/* We must avoid interpreting -NUM (signal num) as an option */ /* Parse -SIGNAL for pkill. Must be first option, if present */
first_arg_idx = 1; signo = SIGTERM;
while (1) { if (pkill && argv[1] && argv[1][0] == '-') {
first_arg = argv[first_arg_idx]; int temp = get_signum(argv[1]+1);
if (!first_arg) if (temp != -1) {
break; signo = temp;
/* not "-<small_letter>..."? */
if (first_arg[0] != '-' || first_arg[1] < 'a' || first_arg[1] > 'z') {
argv[first_arg_idx] = NULL; /* terminate argv here */
break;
}
first_arg_idx++;
}
opt = getopt32(argv, "vlfxon");
argv[first_arg_idx] = first_arg;
argv += optind;
//argc -= optind; - unused anyway
if (OPT_FULL)
scan_mask |= PSSCAN_ARGVN;
if (pkill) {
if (OPT_LIST) { /* -l: print the whole signal list */
print_signames();
return 0;
}
if (first_arg && first_arg[0] == '-') {
signo = get_signum(&first_arg[1]);
if (signo < 0) /* || signo > MAX_SIGNUM ? */
bb_error_msg_and_die("bad signal name '%s'", &first_arg[1]);
argv++; argv++;
} }
} }
/* One pattern is required */ /* Parse remaining options */
if (!argv[0] || argv[1]) ppid2match = -1;
sid2match = -1;
opt_complementary = "s+:P+"; /* numeric opts */
opt = getopt32(argv, "vlfxons: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;
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(); bb_show_usage();
xregcomp(&re_buffer, argv[0], 0); if (argv[0])
xregcomp(&re_buffer, argv[0], 0);
matched_pid = 0; matched_pid = 0;
cmd_last = NULL; cmd_last = NULL;
proc = NULL; proc = NULL;
while ((proc = procps_scan(proc, scan_mask)) != NULL) { while ((proc = procps_scan(proc, scan_mask)) != NULL) {
char *cmd; char *cmd;
if (proc->pid == pid) if (proc->pid == pid)
continue; continue;
cmd = proc->argv0; cmd = proc->argv0;
if (!cmd) { if (!cmd) {
cmd = proc->comm; cmd = proc->comm;
@ -120,8 +122,15 @@ int pgrep_main(int argc UNUSED_PARAM, char **argv)
i--; i--;
} }
} }
if (ppid2match >= 0 && ppid2match != proc->ppid)
continue;
if (sid2match >= 0 && sid2match != proc->sid)
continue;
/* NB: OPT_INVERT is always 0 or 1 */ /* NB: OPT_INVERT is always 0 or 1 */
if ((regexec(&re_buffer, cmd, 1, re_match, 0) == 0 /* match found */ if (!argv[0] ||
(regexec(&re_buffer, cmd, 1, re_match, 0) == 0 /* match found */
&& (!OPT_ANCHOR || (re_match[0].rm_so == 0 && re_match[0].rm_eo == (regoff_t)strlen(cmd)))) ^ OPT_INVERT && (!OPT_ANCHOR || (re_match[0].rm_so == 0 && re_match[0].rm_eo == (regoff_t)strlen(cmd)))) ^ OPT_INVERT
) { ) {
matched_pid = proc->pid; matched_pid = proc->pid;
@ -130,13 +139,14 @@ int pgrep_main(int argc UNUSED_PARAM, char **argv)
cmd_last = xstrdup(cmd); cmd_last = xstrdup(cmd);
continue; continue;
} }
act(proc->pid, cmd, signo, opt); act(proc->pid, cmd, signo);
if (OPT_FIRST) if (OPT_FIRST)
break; break;
} }
} }
if (cmd_last) { if (cmd_last) {
act(matched_pid, cmd_last, signo, opt); act(matched_pid, cmd_last, signo);
if (ENABLE_FEATURE_CLEAN_UP) if (ENABLE_FEATURE_CLEAN_UP)
free(cmd_last); free(cmd_last);
} }