blob: e4dfacb547f49babfe7ef9af14829a93efc32e11 [file] [log] [blame]
Denis Vlasenko72e1c892007-09-29 22:26:01 +00001/* vi: set sw=4 ts=4: */
2/*
3 * Mini pgrep/pkill implementation for busybox
4 *
5 * Copyright (C) 2007 Loic Grenie <loic.grenie@gmail.com>
6 *
7 * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
8 */
9
10#include <getopt.h>
11
12#include "libbb.h"
13#include "xregex.h"
14
15/* Idea taken from kill.c */
16#define pgrep (ENABLE_PGREP && applet_name[1] == 'g')
17#define pkill (ENABLE_PKILL && applet_name[1] == 'k')
18
19enum {
20 /* "vlfxon" */
21 PGREPOPTBIT_V = 0, /* must be first, we need OPT_INVERT = 0/1 */
22 PGREPOPTBIT_L,
23 PGREPOPTBIT_F,
24 PGREPOPTBIT_X,
25 PGREPOPTBIT_O,
26 PGREPOPTBIT_N,
27};
28
29#define OPT_INVERT (opt & (1 << PGREPOPTBIT_V))
30#define OPT_LIST (opt & (1 << PGREPOPTBIT_L))
31#define OPT_FULL (opt & (1 << PGREPOPTBIT_F))
32#define OPT_ANCHOR (opt & (1 << PGREPOPTBIT_X))
33#define OPT_FIRST (opt & (1 << PGREPOPTBIT_O))
34#define OPT_LAST (opt & (1 << PGREPOPTBIT_N))
35
36static void act(unsigned pid, char *cmd, int signo, unsigned opt)
37{
38 if (pgrep) {
39 if (OPT_LIST)
40 printf("%d %s\n", pid, cmd);
41 else
42 printf("%d\n", pid);
43 } else
44 kill(pid, signo);
45}
46
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +000047int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denis Vlasenko72e1c892007-09-29 22:26:01 +000048int pgrep_main(int argc, char **argv)
49{
50 unsigned pid = getpid();
51 int signo = SIGTERM;
52 unsigned opt;
53 int scan_mask = PSSCAN_COMM;
54 char *first_arg;
55 int first_arg_idx;
56 int matched_pid;
57 char *cmd_last;
58 procps_status_t *proc;
59 /* These are initialized to 0 */
60 struct {
61 regex_t re_buffer;
62 regmatch_t re_match[1];
63 } Z;
64#define re_buffer (Z.re_buffer)
65#define re_match (Z.re_match )
66
67 memset(&Z, 0, sizeof(Z));
68
69 /* We must avoid interpreting -NUM (signal num) as an option */
70 first_arg_idx = 1;
71 while (1) {
72 first_arg = argv[first_arg_idx];
73 if (!first_arg)
74 break;
Denis Vlasenkocdf62772008-03-17 08:42:43 +000075 /* not "-<small_letter>..."? */
Denis Vlasenko72e1c892007-09-29 22:26:01 +000076 if (first_arg[0] != '-' || first_arg[1] < 'a' || first_arg[1] > 'z') {
Denis Vlasenkocdf62772008-03-17 08:42:43 +000077 argv[first_arg_idx] = NULL; /* terminate argv here */
Denis Vlasenko72e1c892007-09-29 22:26:01 +000078 break;
79 }
80 first_arg_idx++;
81 }
82 opt = getopt32(argv, "vlfxon");
83 argv[first_arg_idx] = first_arg;
84
85 argv += optind;
86 //argc -= optind; - unused anyway
87 if (OPT_FULL)
88 scan_mask |= PSSCAN_ARGVN;
89
90 if (pkill) {
Denis Vlasenkoa4f4de92007-09-30 16:32:01 +000091 if (OPT_LIST) { /* -l: print the whole signal list */
92 print_signames();
93 return 0;
94 }
Denis Vlasenko72e1c892007-09-29 22:26:01 +000095 if (first_arg && first_arg[0] == '-') {
96 signo = get_signum(&first_arg[1]);
97 if (signo < 0) /* || signo > MAX_SIGNUM ? */
98 bb_error_msg_and_die("bad signal name '%s'", &first_arg[1]);
99 argv++;
100 }
101 }
102
103 /* One pattern is required */
104 if (!argv[0] || argv[1])
105 bb_show_usage();
106
107 xregcomp(&re_buffer, argv[0], 0);
108 matched_pid = 0;
109 cmd_last = NULL;
110 proc = NULL;
111 while ((proc = procps_scan(proc, scan_mask)) != NULL) {
112 char *cmd;
113 if (proc->pid == pid)
114 continue;
115 cmd = proc->argv0;
116 if (!cmd)
117 cmd = proc->comm;
118 /* NB: OPT_INVERT is always 0 or 1 */
119 if ((regexec(&re_buffer, cmd, 1, re_match, 0) == 0 /* match found */
Denis Vlasenko6b404432008-01-07 16:13:14 +0000120 && (!OPT_ANCHOR || (re_match[0].rm_so == 0 && re_match[0].rm_eo == strlen(cmd)))) ^ OPT_INVERT
Denis Vlasenko72e1c892007-09-29 22:26:01 +0000121 ) {
122 matched_pid = proc->pid;
123 if (OPT_LAST) {
124 free(cmd_last);
Denis Vlasenko5fb09652007-09-30 16:36:02 +0000125 cmd_last = xstrdup(cmd);
Denis Vlasenko72e1c892007-09-29 22:26:01 +0000126 continue;
127 }
128 act(proc->pid, cmd, signo, opt);
129 if (OPT_FIRST)
130 break;
131 }
132 }
133 if (cmd_last) {
134 act(matched_pid, cmd_last, signo, opt);
135 if (ENABLE_FEATURE_CLEAN_UP)
136 free(cmd_last);
137 }
138 return matched_pid == 0; /* return 1 if no processes listed/signaled */
139}