procps/pgrep.c - busybox - Git at Google

 /* vi: set sw=4 ts=4: */
 /*
  * Mini pgrep/pkill implementation for busybox
  *
  * Copyright (C) 2007 Loic Grenie <loic.grenie@gmail.com>
  *
  * 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 */
 }
	/* vi: set sw=4 ts=4: */
	/*
	* Mini pgrep/pkill implementation for busybox
	*
	* Copyright (C) 2007 Loic Grenie <loic.grenie@gmail.com>
	*
	* 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 */
	}