summaryrefslogtreecommitdiff
path: root/sysdeps/freebsd/proclist.c
blob: bb4e0de81bb93ea3b045f0beeecf7fa72a084d8b (plain)
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
105
106
107
/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 4 -*- */

/* $Id$ */

/* Copyright (C) 1998 Joshua Sled
   This file is part of LibGTop 1.0.

   Contributed by Joshua Sled <jsled@xcf.berkeley.edu>, July 1998.

   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 <glibtop.h>
#include <glibtop/error.h>
#include <glibtop/xmalloc.h>
#include <glibtop/proclist.h>

#include <glibtop_suid.h>

static const unsigned long _glibtop_sysdeps_proclist =
(1L << GLIBTOP_PROCLIST_TOTAL) + (1L << GLIBTOP_PROCLIST_NUMBER) +
(1L << GLIBTOP_PROCLIST_SIZE);

/* Fetch list of currently running processes.
 * The interface of this function is a little bit different from the others:
 * buf->flags is only set if the call succeeded, in this case pids_chain,
 * a list of the pids of all currently running processes is returned,
 * buf->number is the number of elements of this list and buf->size is
 * the size of one single element (sizeof (unsigned)). The total size is
 * stored in buf->total.
 *
 * The calling function has to free the memory to which a pointer is returned.
 *
 * IMPORTANT NOTE:
 *   On error, this function MUST return NULL and set buf->flags to zero !
 *   On success, it returnes a pointer to a list of buf->number elements
 *   each buf->size big. The total size is stored in buf->total.
 * The calling function has to free the memory to which a pointer is returned.
 *
 * On error, NULL is returned and buf->flags is zero. */

/* Init function. */

int
glibtop_init_proclist_p (glibtop *server)
{
    server->sysdeps.proclist = _glibtop_sysdeps_proclist;

    return 0;
}

unsigned *
glibtop_get_proclist_p (glibtop *server, glibtop_proclist *buf,
			int64_t real_which, int64_t arg)
{
    struct kinfo_proc *pinfo;
    unsigned *pids = NULL;
    int which, count;
    int i,j;

    glibtop_init_p (server, (1L << GLIBTOP_SYSDEPS_PROCLIST), 0);
	
    memset (buf, 0, sizeof (glibtop_proclist));

    which = (int)(real_which & GLIBTOP_KERN_PROC_MASK);

    /* Get the process data */
    pinfo = kvm_getprocs (server->_priv->machine.kd, which, arg, &count);
    if ((pinfo == NULL) || (count < 1)) {
	glibtop_warn_io_r (server, "kvm_getprocs (proclist)");
	return NULL;
    }
    count--;

    /* Allocate count objects in the pids_chain array
     * Same as malloc is pids is NULL, which it is. */
    pids = glibtop_realloc_r (server, pids, count * sizeof (unsigned));
    /* Copy the pids over to this chain */
    for (i=j=0; i < count; i++) {
	if ((real_which & GLIBTOP_EXCLUDE_IDLE) &&
	    (pinfo[i].kp_proc.p_stat != SRUN))
	    continue;
	else if ((real_which & GLIBTOP_EXCLUDE_SYSTEM) &&
		 (pinfo[i].kp_eproc.e_pcred.p_ruid == 0))
	    continue;
	pids [j++] = (unsigned) pinfo[i].kp_proc.p_pid;
    } /* end for */
    /* Set the fields in buf */
    buf->number = j;
    buf->size = sizeof (unsigned);
    buf->total = j * sizeof (unsigned);
    buf->flags = _glibtop_sysdeps_proclist;
    return pids;
}