src/load.c

   1 /**
   2  * collectd - src/load.c
   3  * Copyright (C) 2005-2008  Florian octo Forster
   4  * Copyright (C) 2009       Manuel Sanmartin
   5  * Copyright (C) 2013       Vedran Bartonicek
   6  *
   7  * This program is free software; you can redistribute it and/or modify it
   8  * under the terms of the GNU General Public License as published by the
   9  * Free Software Foundation; only version 2 of the License is applicable.
  10  *
  11  * This program is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  19  *
  20  * Authors:
  21  *   Florian octo Forster <octo at collectd.org>
  22  *   Manuel Sanmartin
  23  *   Vedran Bartonicek <vbartoni at gmail.com>
  24  **/
  25
  26 #define _DEFAULT_SOURCE
  27 #define _BSD_SOURCE
  28
  29 #include "collectd.h"
  30
  31 #include "common.h"
  32 #include "plugin.h"
  33
  34 #include <unistd.h>
  35
  36 #ifdef HAVE_SYS_LOADAVG_H
  37 #include <sys/loadavg.h>
  38 #endif
  39
  40 #if HAVE_STATGRAB_H
  41 #include <statgrab.h>
  42 #endif
  43
  44 #ifdef HAVE_GETLOADAVG
  45 #if !defined(LOADAVG_1MIN) || !defined(LOADAVG_5MIN) || !defined(LOADAVG_15MIN)
  46 #define LOADAVG_1MIN 0
  47 #define LOADAVG_5MIN 1
  48 #define LOADAVG_15MIN 2
  49 #endif
  50 #endif /* defined(HAVE_GETLOADAVG) */
  51
  52 #ifdef HAVE_PERFSTAT
  53 #include <libperfstat.h>
  54 #include <sys/proc.h> /* AIX 5 */
  55 #include <sys/protosw.h>
  56 #endif /* HAVE_PERFSTAT */
  57
  58 static _Bool report_relative_load = 0;
  59
  60 static const char *config_keys[] = {"ReportRelative"};
  61 static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
  62
  63 static int load_config(const char *key, const char *value) {
  64   if (strcasecmp(key, "ReportRelative") == 0)
  65 #ifdef _SC_NPROCESSORS_ONLN
  66     report_relative_load = IS_TRUE(value) ? 1 : 0;
  67 #else
  68     WARNING("load plugin: The \"ReportRelative\" configuration "
  69             "is not available, because I can't determine the "
  70             "number of CPUS on this system. Sorry.");
  71 #endif
  72   return (-1);
  73 }
  74 static void load_submit(gauge_t snum, gauge_t mnum, gauge_t lnum) {
  75   value_t values[3];
  76   value_list_t vl = VALUE_LIST_INIT;
  77   int cores = 0;
  78   char errbuf[1024];
  79
  80 #ifdef _SC_NPROCESSORS_ONLN
  81   if (report_relative_load) {
  82     if ((cores = sysconf(_SC_NPROCESSORS_ONLN)) < 1) {
  83       WARNING("load: sysconf failed : %s",
  84               sstrerror(errno, errbuf, sizeof(errbuf)));
  85     }
  86   }
  87 #endif
  88   if (cores > 0) {
  89     snum /= cores;
  90     mnum /= cores;
  91     lnum /= cores;
  92   }
  93
  94   values[0].gauge = snum;
  95   values[1].gauge = mnum;
  96   values[2].gauge = lnum;
  97
  98   vl.values = values;
  99   vl.values_len = STATIC_ARRAY_SIZE(values);
 100
 101   sstrncpy(vl.host, hostname_g, sizeof(vl.host));
 102   sstrncpy(vl.plugin, "load", sizeof(vl.plugin));
 103   sstrncpy(vl.type, "load", sizeof(vl.type));
 104
 105   if (cores > 0) {
 106     sstrncpy(vl.type_instance, "relative", sizeof(vl.type_instance));
 107   }
 108
 109   plugin_dispatch_values(&vl);
 110 }
 111
 112 static int load_read(void) {
 113 #if defined(HAVE_GETLOADAVG)
 114   double load[3];
 115
 116   if (getloadavg(load, 3) == 3)
 117     load_submit(load[LOADAVG_1MIN], load[LOADAVG_5MIN], load[LOADAVG_15MIN]);
 118   else {
 119     char errbuf[1024];
 120     WARNING("load: getloadavg failed: %s",
 121             sstrerror(errno, errbuf, sizeof(errbuf)));
 122   }
 123 /* #endif HAVE_GETLOADAVG */
 124
 125 #elif defined(KERNEL_LINUX)
 126   gauge_t snum, mnum, lnum;
 127   FILE *loadavg;
 128   char buffer[16];
 129
 130   char *fields[8];
 131   int numfields;
 132
 133   if ((loadavg = fopen("/proc/loadavg", "r")) == NULL) {
 134     char errbuf[1024];
 135     WARNING("load: fopen: %s", sstrerror(errno, errbuf, sizeof(errbuf)));
 136     return (-1);
 137   }
 138
 139   if (fgets(buffer, 16, loadavg) == NULL) {
 140     char errbuf[1024];
 141     WARNING("load: fgets: %s", sstrerror(errno, errbuf, sizeof(errbuf)));
 142     fclose(loadavg);
 143     return (-1);
 144   }
 145
 146   if (fclose(loadavg)) {
 147     char errbuf[1024];
 148     WARNING("load: fclose: %s", sstrerror(errno, errbuf, sizeof(errbuf)));
 149   }
 150
 151   numfields = strsplit(buffer, fields, 8);
 152
 153   if (numfields < 3)
 154     return (-1);
 155
 156   snum = atof(fields[0]);
 157   mnum = atof(fields[1]);
 158   lnum = atof(fields[2]);
 159
 160   load_submit(snum, mnum, lnum);
 161 /* #endif KERNEL_LINUX */
 162
 163 #elif HAVE_LIBSTATGRAB
 164   gauge_t snum, mnum, lnum;
 165   sg_load_stats *ls;
 166
 167   if ((ls = sg_get_load_stats()) == NULL)
 168     return;
 169
 170   snum = ls->min1;
 171   mnum = ls->min5;
 172   lnum = ls->min15;
 173   load_submit(snum, mnum, lnum);
 174 /* #endif HAVE_LIBSTATGRAB */
 175
 176 #elif HAVE_PERFSTAT
 177   gauge_t snum, mnum, lnum;
 178   perfstat_cpu_total_t cputotal;
 179
 180   if (perfstat_cpu_total(NULL, &cputotal, sizeof(perfstat_cpu_total_t), 1) <
 181       0) {
 182     char errbuf[1024];
 183     WARNING("load: perfstat_cpu : %s",
 184             sstrerror(errno, errbuf, sizeof(errbuf)));
 185     return (-1);
 186   }
 187
 188   snum = (float)cputotal.loadavg[0] / (float)(1 << SBITS);
 189   mnum = (float)cputotal.loadavg[1] / (float)(1 << SBITS);
 190   lnum = (float)cputotal.loadavg[2] / (float)(1 << SBITS);
 191   load_submit(snum, mnum, lnum);
 192 /* #endif HAVE_PERFSTAT */
 193
 194 #else
 195 #error "No applicable input method."
 196 #endif
 197
 198   return (0);
 199 }
 200
 201 void module_register(void) {
 202   plugin_register_config("load", load_config, config_keys, config_keys_num);
 203   plugin_register_read("load", load_read);
 204 } /* void module_register */