src/users.c

   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  **/
  27
  28 #include "collectd.h"
  29
  30 #include "common.h"
  31 #include "plugin.h"
  32
  33 #if HAVE_STATGRAB_H
  34 #include <statgrab.h>
  35 #endif /* HAVE_STATGRAB_H */
  36
  37 #if HAVE_UTMPX_H
  38 #include <utmpx.h>
  39 /* #endif HAVE_UTMPX_H */
  40
  41 #elif HAVE_UTMP_H
  42 #include <utmp.h>
  43 /* #endif HAVE_UTMP_H */
  44 #endif
  45
  46 static void users_submit(gauge_t value) {
  47   value_t values[1];
  48   value_list_t vl = VALUE_LIST_INIT;
  49
  50   values[0].gauge = value;
  51
  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));
  57
  58   plugin_dispatch_values(&vl);
  59 } /* void users_submit */
  60
  61 static int users_read(void) {
  62 #if HAVE_GETUTXENT
  63   unsigned int users = 0;
  64   struct utmpx *entry = NULL;
  65
  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();
  69
  70   while (NULL != (entry = getutxent())) {
  71     if (USER_PROCESS == entry->ut_type) {
  72       ++users;
  73     }
  74   }
  75   endutxent();
  76
  77   users_submit(users);
  78 /* #endif HAVE_GETUTXENT */
  79
  80 #elif HAVE_GETUTENT
  81   unsigned int users = 0;
  82   struct utmp *entry = NULL;
  83
  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();
  87
  88   while (NULL != (entry = getutent())) {
  89     if (USER_PROCESS == entry->ut_type) {
  90       ++users;
  91     }
  92   }
  93   endutent();
  94
  95   users_submit(users);
  96 /* #endif HAVE_GETUTENT */
  97
  98 #elif HAVE_LIBSTATGRAB
  99   sg_user_stats *us;
 100
 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);
 109
 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 */
 117
 118 #else
 119 #error "No applicable input method."
 120 #endif
 121
 122   return (0);
 123 } /* int users_read */
 124
 125 void module_register(void) {
 126   plugin_register_read("users", users_read);
 127 } /* void module_register(void) */