git.maemo.org Git - busybox4maemo/blob - 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 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
  19 enum {
  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
  36 static 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
  47 int pgrep_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  48 int pgrep_main(int argc ATTRIBUTE_UNUSED, 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;
  75                 /* not "-<small_letter>..."? */
  76                 if (first_arg[0] != '-' || first_arg[1] < 'a' || first_arg[1] > 'z') {
  77                         argv[first_arg_idx] = NULL; /* terminate argv here */
  78                         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) {
  91                 if (OPT_LIST) { /* -l: print the whole signal list */
  92                         print_signames();
  93                         return 0;
  94                 }
  95                 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 */
 120                      && (!OPT_ANCHOR || (re_match[0].rm_so == 0 && re_match[0].rm_eo == strlen(cmd)))) ^ OPT_INVERT
 121                 ) {
 122                         matched_pid = proc->pid;
 123                         if (OPT_LAST) {
 124                                 free(cmd_last);
 125                                 cmd_last = xstrdup(cmd);
 126                                 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 }