1 /**
2 * collectd - src/users.c
3 * Copyright (C) 2005-2007 Sebastian Harl
4 * Copyright (C) 2005 Niki W. Waibel
5 * Copyright (C) 2005-2007 Florian octo Forster
6 * Copyright (C) 2008 Oleg King
7 *
8 * This program is free software; you can redistribute it and/or modify it
9 * under the terms of the GNU General Public License as published by the
10 * Free Software Foundation; only version 2 of the license is applicable.
11 *
12 * This program is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License for more details.
16 *
17 * You should have received a copy of the GNU General Public License along
18 * with this program; if not, write to the Free Software Foundation, Inc.,
19 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
20 *
21 * Authors:
22 * Sebastian Harl <sh at tokkee.org>
23 * Niki W. Waibel <niki.waibel at newlogic.com>
24 * Florian octo Forster <octo at collectd.org>
25 * Oleg King <king2 at kaluga.ru>
26 **/
28 #include "collectd.h"
30 #include "common.h"
31 #include "plugin.h"
33 #if HAVE_STATGRAB_H
34 #include <statgrab.h>
35 #endif /* HAVE_STATGRAB_H */
37 #if HAVE_UTMPX_H
38 #include <utmpx.h>
39 /* #endif HAVE_UTMPX_H */
41 #elif HAVE_UTMP_H
42 #include <utmp.h>
43 /* #endif HAVE_UTMP_H */
44 #endif
46 static void users_submit(gauge_t value) {
47 value_t values[1];
48 value_list_t vl = VALUE_LIST_INIT;
50 values[0].gauge = value;
52 vl.values = values;
53 vl.values_len = 1;
54 sstrncpy(vl.host, hostname_g, sizeof(vl.host));
55 sstrncpy(vl.plugin, "users", sizeof(vl.plugin));
56 sstrncpy(vl.type, "users", sizeof(vl.plugin));
58 plugin_dispatch_values(&vl);
59 } /* void users_submit */
61 static int users_read(void) {
62 #if HAVE_GETUTXENT
63 unsigned int users = 0;
64 struct utmpx *entry = NULL;
66 /* according to the *utent(3) man page none of the functions sets errno
67 in case of an error, so we cannot do any error-checking here */
68 setutxent();
70 while (NULL != (entry = getutxent())) {
71 if (USER_PROCESS == entry->ut_type) {
72 ++users;
73 }
74 }
75 endutxent();
77 users_submit(users);
78 /* #endif HAVE_GETUTXENT */
80 #elif HAVE_GETUTENT
81 unsigned int users = 0;
82 struct utmp *entry = NULL;
84 /* according to the *utent(3) man page none of the functions sets errno
85 in case of an error, so we cannot do any error-checking here */
86 setutent();
88 while (NULL != (entry = getutent())) {
89 if (USER_PROCESS == entry->ut_type) {
90 ++users;
91 }
92 }
93 endutent();
95 users_submit(users);
96 /* #endif HAVE_GETUTENT */
98 #elif HAVE_LIBSTATGRAB
99 sg_user_stats *us;
101 #if HAVE_LIBSTATGRAB_0_90
102 size_t num_entries;
103 us = sg_get_user_stats(&num_entries);
104 #else
105 us = sg_get_user_stats();
106 #endif
107 if (us == NULL)
108 return (-1);
110 users_submit((gauge_t)
111 #if HAVE_LIBSTATGRAB_0_90
112 num_entries);
113 #else
114 us->num_entries);
115 #endif
116 /* #endif HAVE_LIBSTATGRAB */
118 #else
119 #error "No applicable input method."
120 #endif
122 return (0);
123 } /* int users_read */
125 void module_register(void) {
126 plugin_register_read("users", users_read);
127 } /* void module_register(void) */