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 the GPL version 2, see the file LICENSE in this tarball.
8 */
9
10#include "libbb.h"
11
12enum {
13	IF_FEATURE_PIDOF_SINGLE(OPTBIT_SINGLE,)
14	IF_FEATURE_PIDOF_OMIT(  OPTBIT_OMIT  ,)
15	OPT_SINGLE = IF_FEATURE_PIDOF_SINGLE((1<<OPTBIT_SINGLE)) + 0,
16	OPT_OMIT   = IF_FEATURE_PIDOF_OMIT(  (1<<OPTBIT_OMIT  )) + 0,
17};
18
19int pidof_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
20int pidof_main(int argc UNUSED_PARAM, char **argv)
21{
22	unsigned first = 1;
23	unsigned opt;
24#if ENABLE_FEATURE_PIDOF_OMIT
25	llist_t *omits = NULL; /* list of pids to omit */
26	opt_complementary = "o::";
27#endif
28
29	/* do unconditional option parsing */
30	opt = getopt32(argv, ""
31			IF_FEATURE_PIDOF_SINGLE ("s")
32			IF_FEATURE_PIDOF_OMIT("o:", &omits));
33
34#if ENABLE_FEATURE_PIDOF_OMIT
35	/* fill omit list.  */
36	{
37		llist_t *omits_p = omits;
38		while (1) {
39			omits_p = llist_find_str(omits_p, "%PPID");
40			if (!omits_p)
41				break;
42			/* are we asked to exclude the parent's process ID?  */
43			omits_p->data = utoa((unsigned)getppid());
44		}
45	}
46#endif
47	/* Looks like everything is set to go.  */
48	argv += optind;
49	while (*argv) {
50		pid_t *pidList;
51		pid_t *pl;
52
53		/* reverse the pidlist like GNU pidof does.  */
54		pidList = pidlist_reverse(find_pid_by_name(*argv));
55		for (pl = pidList; *pl; pl++) {
56#if ENABLE_FEATURE_PIDOF_OMIT
57			if (opt & OPT_OMIT) {
58				llist_t *omits_p = omits;
59				while (omits_p) {
60					if (xatoul(omits_p->data) == (unsigned long)(*pl)) {
61						goto omitting;
62					}
63					omits_p = omits_p->link;
64				}
65			}
66#endif
67			printf(" %u" + first, (unsigned)*pl);
68			first = 0;
69			if (ENABLE_FEATURE_PIDOF_SINGLE && (opt & OPT_SINGLE))
70				break;
71#if ENABLE_FEATURE_PIDOF_OMIT
72 omitting: ;
73#endif
74		}
75		free(pidList);
76		argv++;
77	}
78	if (!first)
79		bb_putchar('\n');
80
81#if ENABLE_FEATURE_PIDOF_OMIT
82	if (ENABLE_FEATURE_CLEAN_UP)
83		llist_free(omits, NULL);
84#endif
85	return first; /* 1 (failure) - no processes found */
86}
87