LXR busybox/procps/pidof.c

   1/* vi: set sw=4 ts=4: */
   2/*
   3 * pidof implementation for busybox
   4 *
   5 * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
   6 *
   7 * Licensed under GPLv2, see file LICENSE in this source tree.
   8 */
   9
  10//usage:#if (ENABLE_FEATURE_PIDOF_SINGLE || ENABLE_FEATURE_PIDOF_OMIT)
  11//usage:#define pidof_trivial_usage
  12//usage:       "[OPTIONS] [NAME]..."
  13//usage:#define USAGE_PIDOF "\n"
  14//usage:#else
  15//usage:#define pidof_trivial_usage
  16//usage:       "[NAME]..."
  17//usage:#define USAGE_PIDOF /* none */
  18//usage:#endif
  19//usage:#define pidof_full_usage "\n\n"
  20//usage:       "List PIDs of all processes with names that match NAMEs"
  21//usage:        USAGE_PIDOF
  22//usage:        IF_FEATURE_PIDOF_SINGLE(
  23//usage:     "\n        -s      Show only one PID"
  24//usage:        )
  25//usage:        IF_FEATURE_PIDOF_OMIT(
  26//usage:     "\n        -o PID  Omit given pid"
  27//usage:     "\n                Use %PPID to omit pid of pidof's parent"
  28//usage:        )
  29//usage:
  30//usage:#define pidof_example_usage
  31//usage:       "$ pidof init\n"
  32//usage:       "1\n"
  33//usage:        IF_FEATURE_PIDOF_OMIT(
  34//usage:       "$ pidof /bin/sh\n20351 5973 5950\n")
  35//usage:        IF_FEATURE_PIDOF_OMIT(
  36//usage:       "$ pidof /bin/sh -o %PPID\n20351 5950")
  37
  38#include "libbb.h"
  39
  40enum {
  41        IF_FEATURE_PIDOF_SINGLE(OPTBIT_SINGLE,)
  42        IF_FEATURE_PIDOF_OMIT(  OPTBIT_OMIT  ,)
  43        OPT_SINGLE = IF_FEATURE_PIDOF_SINGLE((1<<OPTBIT_SINGLE)) + 0,
  44        OPT_OMIT   = IF_FEATURE_PIDOF_OMIT(  (1<<OPTBIT_OMIT  )) + 0,
  45};
  46
  47int pidof_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  48int pidof_main(int argc UNUSED_PARAM, char **argv)
  49{
  50        unsigned first = 1;
  51        unsigned opt;
  52#if ENABLE_FEATURE_PIDOF_OMIT
  53        llist_t *omits = NULL; /* list of pids to omit */
  54        opt_complementary = "o::";
  55#endif
  56
  57        /* do unconditional option parsing */
  58        opt = getopt32(argv, ""
  59                        IF_FEATURE_PIDOF_SINGLE ("s")
  60                        IF_FEATURE_PIDOF_OMIT("o:", &omits));
  61
  62#if ENABLE_FEATURE_PIDOF_OMIT
  63        /* fill omit list.  */
  64        {
  65                llist_t *omits_p = omits;
  66                while (1) {
  67                        omits_p = llist_find_str(omits_p, "%PPID");
  68                        if (!omits_p)
  69                                break;
  70                        /* are we asked to exclude the parent's process ID?  */
  71                        omits_p->data = utoa((unsigned)getppid());
  72                }
  73        }
  74#endif
  75        /* Looks like everything is set to go.  */
  76        argv += optind;
  77        while (*argv) {
  78                pid_t *pidList;
  79                pid_t *pl;
  80
  81                /* reverse the pidlist like GNU pidof does.  */
  82                pidList = pidlist_reverse(find_pid_by_name(*argv));
  83                for (pl = pidList; *pl; pl++) {
  84#if ENABLE_FEATURE_PIDOF_OMIT
  85                        if (opt & OPT_OMIT) {
  86                                llist_t *omits_p = omits;
  87                                while (omits_p) {
  88                                        if (xatoul(omits_p->data) == (unsigned long)(*pl)) {
  89                                                goto omitting;
  90                                        }
  91                                        omits_p = omits_p->link;
  92                                }
  93                        }
  94#endif
  95                        printf(" %u" + first, (unsigned)*pl);
  96                        first = 0;
  97                        if (ENABLE_FEATURE_PIDOF_SINGLE && (opt & OPT_SINGLE))
  98                                break;
  99#if ENABLE_FEATURE_PIDOF_OMIT
 100 omitting: ;
 101#endif
 102                }
 103                free(pidList);
 104                argv++;
 105        }
 106        if (!first)
 107                bb_putchar('\n');
 108
 109#if ENABLE_FEATURE_PIDOF_OMIT
 110        if (ENABLE_FEATURE_CLEAN_UP)
 111                llist_free(omits, NULL);
 112#endif
 113        return first; /* 1 (failure) - no processes found */
 114}
 115