src/users.c

   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  */
  24
  25 #include "users.h"
  26
  27 #if COLLECT_USERS
  28 #define MODULE_NAME "users"
  29
  30 #include "plugin.h"
  31 #include "common.h"
  32
  33 #ifdef HAVE_UTMPX_H
  34 #include <utmpx.h>
  35 #elif defined(HAVE_UTMP_H)
  36 #include <utmp.h>
  37 #endif
  38
  39 static char *rrd_file = "users.rrd";
  40
  41 static char *ds_def[] = {
  42     "DS:users:GAUGE:25:0:65535",
  43     NULL
  44 };
  45 static int ds_num = 1;
  46
  47 extern time_t curtime;
  48
  49 void users_init(void)
  50 {
  51     /* we have nothing to do here :-) */
  52     return;
  53 }
  54
  55 void users_read(void)
  56 {
  57 #ifdef HAVE_GETUTXENT
  58     unsigned int users = 0;
  59     struct utmpx *entry = NULL;
  60
  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();
  64
  65     while (NULL != (entry = getutxent()))
  66         if (USER_PROCESS == entry->ut_type)
  67             ++users;
  68     endutxent();
  69
  70     users_submit(users);
  71 /* #endif HAVE_GETUTXENT */
  72
  73 #elif defined(HAVE_GETUTENT)
  74     unsigned int users = 0;
  75     struct utmp *entry = NULL;
  76
  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();
  80
  81     while (NULL != (entry = getutent()))
  82         if (USER_PROCESS == entry->ut_type)
  83             ++users;
  84     endutent();
  85
  86     users_submit(users);
  87 #endif
  88
  89         return;
  90 }
  91
  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] = "";
  99
 100     if (snprintf(buf, BUFSIZE, "%u:%u",
 101                 (unsigned int)curtime,
 102                 users) >= BUFSIZE)
 103         return;
 104
 105     plugin_submit(MODULE_NAME, NULL, buf);
 106     return;
 107 }
 108 #undef BUFSIZE
 109
 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 }
 118
 119 void module_register(void)
 120 {
 121     plugin_register(MODULE_NAME, users_init, users_read, users_write);
 122     return;
 123 }
 124
 125 #undef MODULE_NAME
 126 #endif /* COLLECT_USERS */
 127