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 v2, see the file LICENSE in this tarball.
|
---|
8 | */
|
---|
9 |
|
---|
10 | #include "busybox.h"
|
---|
11 | #include <stdio.h>
|
---|
12 | #include <stdlib.h>
|
---|
13 | #include <errno.h>
|
---|
14 | #include <unistd.h>
|
---|
15 | #include <signal.h>
|
---|
16 | #include <ctype.h>
|
---|
17 | #include <string.h>
|
---|
18 | #include <sys/types.h>
|
---|
19 | #include <unistd.h>
|
---|
20 |
|
---|
21 | #if ENABLE_FEATURE_PIDOF_SINGLE
|
---|
22 | #define _SINGLE_COMPL(a) a
|
---|
23 | #define SINGLE (1<<0)
|
---|
24 | #else
|
---|
25 | #define _SINGLE_COMPL(a)
|
---|
26 | #define SINGLE (0)
|
---|
27 | #endif
|
---|
28 |
|
---|
29 | #if ENABLE_FEATURE_PIDOF_OMIT
|
---|
30 | #define _OMIT_COMPL(a) a
|
---|
31 | #define _OMIT(a) ,a
|
---|
32 | #if ENABLE_FEATURE_PIDOF_SINGLE
|
---|
33 | #define OMIT (1<<1)
|
---|
34 | #else
|
---|
35 | #define OMIT (1<<0)
|
---|
36 | #endif
|
---|
37 | #else
|
---|
38 | #define _OMIT_COMPL(a) ""
|
---|
39 | #define _OMIT(a)
|
---|
40 | #define OMIT (0)
|
---|
41 | #define omitted (0)
|
---|
42 | #endif
|
---|
43 |
|
---|
44 | int pidof_main(int argc, char **argv)
|
---|
45 | {
|
---|
46 | unsigned n = 0;
|
---|
47 | unsigned fail = 1;
|
---|
48 | unsigned long int opt;
|
---|
49 | #if ENABLE_FEATURE_PIDOF_OMIT
|
---|
50 | llist_t *omits = NULL; /* list of pids to omit */
|
---|
51 | bb_opt_complementally = _OMIT_COMPL("o::");
|
---|
52 | #endif
|
---|
53 |
|
---|
54 | /* do unconditional option parsing */
|
---|
55 | opt = bb_getopt_ulflags(argc, argv,
|
---|
56 | _SINGLE_COMPL("s") _OMIT_COMPL("o:")
|
---|
57 | _OMIT(&omits));
|
---|
58 |
|
---|
59 | #if ENABLE_FEATURE_PIDOF_OMIT
|
---|
60 | /* fill omit list. */
|
---|
61 | {
|
---|
62 | char getppid_str[32];
|
---|
63 | llist_t * omits_p = omits;
|
---|
64 | while (omits_p) {
|
---|
65 | /* are we asked to exclude the parent's process ID? */
|
---|
66 | if (!strncmp(omits_p->data, "%PPID", 5)) {
|
---|
67 | llist_pop(&omits_p);
|
---|
68 | snprintf(getppid_str, sizeof(getppid_str), "%d", getppid());
|
---|
69 | llist_add_to(&omits_p, getppid_str);
|
---|
70 | }
|
---|
71 | omits_p = omits_p->link;
|
---|
72 | }
|
---|
73 | }
|
---|
74 | #endif
|
---|
75 | /* Looks like everything is set to go. */
|
---|
76 | while(optind < argc) {
|
---|
77 | long *pidList;
|
---|
78 | long *pl;
|
---|
79 |
|
---|
80 | /* reverse the pidlist like GNU pidof does. */
|
---|
81 | pidList = pidlist_reverse(find_pid_by_name(argv[optind]));
|
---|
82 | for(pl = pidList; *pl > 0; pl++) {
|
---|
83 | #if ENABLE_FEATURE_PIDOF_OMIT
|
---|
84 | unsigned omitted = 0;
|
---|
85 | if (opt & OMIT) {
|
---|
86 | llist_t *omits_p = omits;
|
---|
87 | while (omits_p)
|
---|
88 | if (strtol(omits_p->data, NULL, 10) == *pl) {
|
---|
89 | omitted = 1; break;
|
---|
90 | } else
|
---|
91 | omits_p = omits_p->link;
|
---|
92 | }
|
---|
93 | #endif
|
---|
94 | if (!omitted) {
|
---|
95 | if (n) {
|
---|
96 | putchar(' ');
|
---|
97 | } else {
|
---|
98 | n = 1;
|
---|
99 | }
|
---|
100 | printf("%ld", *pl);
|
---|
101 | }
|
---|
102 | fail = (!ENABLE_FEATURE_PIDOF_OMIT && omitted);
|
---|
103 |
|
---|
104 | if (ENABLE_FEATURE_PIDOF_SINGLE && (opt & SINGLE))
|
---|
105 | break;
|
---|
106 | }
|
---|
107 | free(pidList);
|
---|
108 | optind++;
|
---|
109 | }
|
---|
110 | putchar('\n');
|
---|
111 |
|
---|
112 | #if ENABLE_FEATURE_PIDOF_OMIT
|
---|
113 | if (ENABLE_FEATURE_CLEAN_UP)
|
---|
114 | llist_free(omits, NULL);
|
---|
115 | #endif
|
---|
116 | return fail ? EXIT_FAILURE : EXIT_SUCCESS;
|
---|
117 | }
|
---|