1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
/*
This file is part of LibGTop 2.0.
LibGTop is free software; you can redistribute it and/or modify it
under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License,
or (at your option) any later version.
LibGTop is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with LibGTop; see the file COPYING. If not, write to the
Free Software Foundation, Inc., 59 Temple Place - Suite 330,
Boston, MA 02111-1307, USA.
*/
#include <config.h>
#include <unistd.h>
#include <glibtop.h>
#include <glibtop/error.h>
#include <glibtop/proclist.h>
#include <glibtop_suid.h>
#include <sys/param.h>
#include <sys/sysctl.h>
static const unsigned long _glibtop_sysdeps_proclist =
(1L << GLIBTOP_PROCLIST_TOTAL) + (1L << GLIBTOP_PROCLIST_NUMBER) +
(1L << GLIBTOP_PROCLIST_SIZE);
/* Init function. */
void
_glibtop_init_proclist_p (glibtop *server)
{
server->sysdeps.proclist = _glibtop_sysdeps_proclist;
}
pid_t *
glibtop_get_proclist_p (glibtop *server, glibtop_proclist *buf,
gint64 which, gint64 arg)
{
unsigned count, total, i;
pid_t *pids_chain;
int mib[4];
struct kinfo_proc *kp;
size_t length;
glibtop_init_p (server, (1 << GLIBTOP_SYSDEPS_PROCLIST), 0);
memset (buf, 0, sizeof (glibtop_proclist));
mib [0] = CTL_KERN;
mib [1] = KERN_PROC;
mib [2] = (int)(which & GLIBTOP_KERN_PROC_MASK);
mib [3] = (int)arg;
if (sysctl (mib, 4, NULL, &length, NULL, 0) < 0) {
glibtop_warn_io_r (server, "sysctl (proclist)");
return NULL;
}
if ((kp = (struct kinfo_proc *) g_malloc(length)) == NULL) {
glibtop_warn_io_r (server, "malloc (proclist)");
return NULL;
}
if (sysctl (mib, 4, kp, &length, NULL, 0) < 0) {
glibtop_warn_io_r (server, "sysctl (proclist)");
g_free (kp);
return NULL;
}
count = length / sizeof (struct kinfo_proc);
pids_chain = g_malloc(count * sizeof (unsigned));
if (pids_chain == NULL) {
glibtop_warn_io_r (server, "g_realloc (proclist)");
g_free (kp);
return NULL;
}
for (total = 0, i = 0; i < count; i++) {
if ((which & GLIBTOP_EXCLUDE_IDLE) &&
(kp [i].kp_proc.p_stat != SRUN))
continue;
if ((which & GLIBTOP_EXCLUDE_SYSTEM) &&
(kp [i].kp_eproc.e_pcred.p_ruid == 0))
continue;
pids_chain [total++] = (unsigned) kp [i].kp_proc.p_pid;
}
g_free (kp);
buf->number = total;
buf->size = sizeof (unsigned);
buf->total = total * sizeof (unsigned);
buf->flags = _glibtop_sysdeps_proclist;
return pids_chain;
}
|