pidof: size optimizations (-50 bytes)
[oweals/busybox.git] / libbb / find_pid_by_name.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Utility routines.
4  *
5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6  *
7  * Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
8  */
9
10 #include "libbb.h"
11
12 /*
13 In Linux we have three ways to determine "process name":
14 1. /proc/PID/stat has "...(name)...", among other things. It's so-called "comm" field.
15 2. /proc/PID/cmdline's first NUL-terminated string. It's argv[0] from exec syscall.
16 3. /proc/PID/exe symlink. Points to the running executable file.
17
18 kernel threads:
19  comm: thread name
20  cmdline: empty
21  exe: <readlink fails>
22
23 executable
24  comm: first 15 chars of base name
25  (if executable is a symlink, then first 15 chars of symlink name are used)
26  cmdline: argv[0] from exec syscall
27  exe: points to executable (resolves symlink, unlike comm)
28
29 script (an executable with #!/path/to/interpreter):
30  comm: first 15 chars of script's base name (symlinks are not resolved)
31  cmdline: /path/to/interpreter (symlinks are not resolved)
32  (script name is in argv[1], args are pushed into argv[2] etc)
33  exe: points to interpreter's executable (symlinks are resolved)
34
35 If FEATURE_PREFER_APPLETS=y (and more so if FEATURE_SH_STANDALONE=y),
36 some commands started from busybox shell, xargs or find are started by
37 execXXX("/proc/self/exe", applet_name, params....)
38 and therefore comm field contains "exe".
39 */
40
41 /* find_pid_by_name()
42  *
43  *  Modified by Vladimir Oleynik for use with libbb/procps.c
44  *  This finds the pid of the specified process.
45  *  Currently, it's implemented by rummaging through
46  *  the proc filesystem.
47  *
48  *  Returns a list of all matching PIDs
49  *  It is the caller's duty to free the returned pidlist.
50  */
51 pid_t* find_pid_by_name(const char* procName)
52 {
53         pid_t* pidList;
54         int i = 0;
55         procps_status_t* p = NULL;
56
57         pidList = xmalloc(sizeof(*pidList));
58         while ((p = procps_scan(p, PSSCAN_PID|PSSCAN_COMM))) {
59                 if (strncmp(p->comm, procName, sizeof(p->comm)-1) == 0) {
60                         pidList = xrealloc(pidList, sizeof(*pidList) * (i+2));
61                         pidList[i++] = p->pid;
62                 }
63         }
64
65         pidList[i] = 0;
66         return pidList;
67 }
68
69 pid_t *pidlist_reverse(pid_t *pidList)
70 {
71         int i = 0;
72         while (pidList[i])
73                 i++;
74         if (--i >= 0) {
75                 pid_t k;
76                 int j;
77                 for (j = 0; i > j; i--, j++) {
78                         k = pidList[i];
79                         pidList[i] = pidList[j];
80                         pidList[j] = k;
81                 }
82         }
83         return pidList;
84 }