Magellan Linux

  • Back to homepage
    • /[magellan-source]/trunk/mkinitrd-magellan/busybox/procps/pgrep.c

    Contents of /trunk/mkinitrd-magellan/busybox/procps/pgrep.c

    Parent Directory Parent Directory | Revision Log Revision Log

    Revision 984 - (show annotations) (download)
    Sun May 30 11:32:42 2010 UTC (13 years, 11 months ago) by niro
    File MIME type: text/plain
    File size: 3498 byte(s) 
    -updated to busybox-1.16.1 and enabled blkid/uuid support in default config
    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 #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)))) ^ OPT_INVERT
    135 ) {
    136 matched_pid = proc->pid;
    137 if (OPT_LAST) {
    138 free(cmd_last);
    139 cmd_last = xstrdup(cmd);
    140 continue;
    141 }
    142 act(proc->pid, cmd, signo);
    143 if (OPT_FIRST)
    144 break;
    145 }
    146 }
    147
    148 if (cmd_last) {
    149 act(matched_pid, cmd_last, signo);
    150 if (ENABLE_FEATURE_CLEAN_UP)
    151 free(cmd_last);
    152 }
    153 return matched_pid == 0; /* return 1 if no processes listed/signaled */
    154 }