procps/pgrep.c

   1 /* 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 GPLv2 or later, see file LICENSE in this source tree.
   8  */
   9 #include "libbb.h"
  10 #include "xregex.h"
  11
  12 /* Idea taken from kill.c */
  13 #define pgrep (ENABLE_PGREP && applet_name[1] == 'g')
  14 #define pkill (ENABLE_PKILL && applet_name[1] == 'k')
  15
  16 enum {
  17         /* "vlfxons:P:" */
  18         OPTBIT_V = 0, /* must be first, we need OPT_INVERT = 0/1 */
  19         OPTBIT_L,
  20         OPTBIT_F,
  21         OPTBIT_X,
  22         OPTBIT_O,
  23         OPTBIT_N,
  24         OPTBIT_S,
  25         OPTBIT_P,
  26 };
  27
  28 #define OPT_INVERT      (opt & (1 << OPTBIT_V))
  29 #define OPT_LIST        (opt & (1 << OPTBIT_L))
  30 #define OPT_FULL        (opt & (1 << OPTBIT_F))
  31 #define OPT_ANCHOR      (opt & (1 << OPTBIT_X))
  32 #define OPT_FIRST       (opt & (1 << OPTBIT_O))
  33 #define OPT_LAST        (opt & (1 << OPTBIT_N))
  34 #define OPT_SID         (opt & (1 << OPTBIT_S))
  35 #define OPT_PPID        (opt & (1 << OPTBIT_P))
  36
  37 static void act(unsigned pid, char *cmd, int signo)
  38 {
  39         if (pgrep) {
  40                 if (option_mask32 & (1 << OPTBIT_L)) /* OPT_LIST */
  41                         printf("%d %s\n", pid, cmd);
  42                 else
  43                         printf("%d\n", pid);
  44         } else
  45                 kill(pid, signo);
  46 }
  47
  48 int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  49 int pgrep_main(int argc UNUSED_PARAM, char **argv)
  50 {
  51         unsigned pid;
  52         int signo;
  53         unsigned opt;
  54         int scan_mask;
  55         int matched_pid;
  56         int sid2match, ppid2match;
  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         /* Parse -SIGNAL for pkill. Must be first option, if present */
  70         signo = SIGTERM;
  71         if (pkill && argv[1] && argv[1][0] == '-') {
  72                 int temp = get_signum(argv[1]+1);
  73                 if (temp != -1) {
  74                         signo = temp;
  75                         argv++;
  76                 }
  77         }
  78
  79         /* Parse remaining options */
  80         ppid2match = -1;
  81         sid2match = -1;
  82         opt_complementary = "s+:P+"; /* numeric opts */
  83         opt = getopt32(argv, "vlfxons:P:", &sid2match, &ppid2match);
  84         argv += optind;
  85
  86         if (pkill && OPT_LIST) { /* -l: print the whole signal list */
  87                 print_signames();
  88                 return 0;
  89         }
  90
  91         pid = getpid();
  92         if (sid2match == 0)
  93                 sid2match = getsid(pid);
  94
  95         scan_mask = PSSCAN_COMM | PSSCAN_ARGV0;
  96         if (OPT_FULL)
  97                 scan_mask |= PSSCAN_ARGVN;
  98
  99         /* One pattern is required, if no -s and no -P */
 100         if ((sid2match & ppid2match) < 0 && (!argv[0] || argv[1]))
 101                 bb_show_usage();
 102
 103         if (argv[0])
 104                 xregcomp(&re_buffer, argv[0], 0);
 105
 106         matched_pid = 0;
 107         cmd_last = NULL;
 108         proc = NULL;
 109         while ((proc = procps_scan(proc, scan_mask)) != NULL) {
 110                 char *cmd;
 111
 112                 if (proc->pid == pid)
 113                         continue;
 114
 115                 cmd = proc->argv0;
 116                 if (!cmd) {
 117                         cmd = proc->comm;
 118                 } else {
 119                         int i = proc->argv_len;
 120                         while (--i >= 0) {
 121                                 if ((unsigned char)cmd[i] < ' ')
 122                                         cmd[i] = ' ';
 123                         }
 124                 }
 125
 126                 if (ppid2match >= 0 && ppid2match != proc->ppid)
 127                         continue;
 128                 if (sid2match >= 0  && sid2match != proc->sid)
 129                         continue;
 130
 131                 /* NB: OPT_INVERT is always 0 or 1 */
 132                 if (!argv[0]
 133                  || (regexec(&re_buffer, cmd, 1, re_match, 0) == 0 /* match found */
 134                     && (!OPT_ANCHOR || (re_match[0].rm_so == 0 && re_match[0].rm_eo == (regoff_t)strlen(cmd)))
 135                     ) ^ OPT_INVERT
 136                 ) {
 137                         matched_pid = proc->pid;
 138                         if (OPT_LAST) {
 139                                 free(cmd_last);
 140                                 cmd_last = xstrdup(cmd);
 141                                 continue;
 142                         }
 143                         act(proc->pid, cmd, signo);
 144                         if (OPT_FIRST)
 145                                 break;
 146                 }
 147         }
 148
 149         if (cmd_last) {
 150                 act(matched_pid, cmd_last, signo);
 151                 if (ENABLE_FEATURE_CLEAN_UP)
 152                         free(cmd_last);
 153         }
 154         return matched_pid == 0; /* return 1 if no processes listed/signaled */
 155 }