96cd6214622b94019aacb8e42f0cdd9f6344743a
1 /*
2 * users.c
3 *
4 * users plugin for collectd
5 *
6 * This plugin collects the number of users currently logged into the system.
7 *
8 * Written by Sebastian Harl <sh@tokkee.org>
9 *
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 */
25 #include "users.h"
27 #if COLLECT_USERS
28 #define MODULE_NAME "users"
30 #include "plugin.h"
31 #include "common.h"
33 #ifdef HAVE_UTMPX_H
34 #include <utmpx.h>
35 #elif defined(HAVE_UTMP_H)
36 #include <utmp.h>
37 #endif
39 static char *rrd_file = "users.rrd";
41 static char *ds_def[] = {
42 "DS:users:GAUGE:25:0:65535",
43 NULL
44 };
45 static int ds_num = 1;
47 extern time_t curtime;
49 void users_init(void)
50 {
51 /* we have nothing to do here :-) */
52 return;
53 }
55 void users_read(void)
56 {
57 #ifdef HAVE_GETUTXENT
58 unsigned int users = 0;
59 struct utmpx *entry = NULL;
61 /* according to the *utent(3) man page none of the functions sets errno in
62 * case of an error, so we cannot do any error-checking here */
63 setutxent();
65 while (NULL != (entry = getutxent()))
66 if (USER_PROCESS == entry->ut_type)
67 ++users;
68 endutxent();
70 users_submit(users);
71 /* #endif HAVE_GETUTXENT */
73 #elif defined(HAVE_GETUTENT)
74 unsigned int users = 0;
75 struct utmp *entry = NULL;
77 /* according to the *utent(3) man page none of the functions sets errno in
78 * case of an error, so we cannot do any error-checking here */
79 setutent();
81 while (NULL != (entry = getutent()))
82 if (USER_PROCESS == entry->ut_type)
83 ++users;
84 endutent();
86 users_submit(users);
87 #endif
89 return;
90 }
92 /* I don't like this temporary macro definition - well it's used everywhere
93 * else in the collectd-sources, so I will just stick with it... */
94 #define BUFSIZE 256
95 void users_submit(users)
96 unsigned int users;
97 {
98 char buf[BUFSIZE] = "";
100 if (snprintf(buf, BUFSIZE, "%u:%u",
101 (unsigned int)curtime,
102 users) >= BUFSIZE)
103 return;
105 plugin_submit(MODULE_NAME, NULL, buf);
106 return;
107 }
108 #undef BUFSIZE
110 void users_write(host, inst, val)
111 char *host;
112 char *inst;
113 char *val;
114 {
115 rrd_update_file(host, rrd_file, val, ds_def, ds_num);
116 return;
117 }
119 void module_register(void)
120 {
121 plugin_register(MODULE_NAME, users_init, users_read, users_write);
122 return;
123 }
125 #undef MODULE_NAME
126 #endif /* COLLECT_USERS */