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 **/
26 #define _DEFAULT_SOURCE
27 #define _BSD_SOURCE
29 #include "collectd.h"
31 #include "common.h"
32 #include "plugin.h"
34 #include <unistd.h>
36 #ifdef HAVE_SYS_LOADAVG_H
37 #include <sys/loadavg.h>
38 #endif
40 #if HAVE_STATGRAB_H
41 #include <statgrab.h>
42 #endif
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) */
52 #ifdef HAVE_PERFSTAT
53 #include <libperfstat.h>
54 #include <sys/proc.h> /* AIX 5 */
55 #include <sys/protosw.h>
56 #endif /* HAVE_PERFSTAT */
58 static _Bool report_relative_load = 0;
60 static const char *config_keys[] = {"ReportRelative"};
61 static int config_keys_num = STATIC_ARRAY_SIZE(config_keys);
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];
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 }
94 values[0].gauge = snum;
95 values[1].gauge = mnum;
96 values[2].gauge = lnum;
98 vl.values = values;
99 vl.values_len = STATIC_ARRAY_SIZE(values);
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));
105 if (cores > 0) {
106 sstrncpy(vl.type_instance, "relative", sizeof(vl.type_instance));
107 }
109 plugin_dispatch_values(&vl);
110 }
112 static int load_read(void) {
113 #if defined(HAVE_GETLOADAVG)
114 double load[3];
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 */
125 #elif defined(KERNEL_LINUX)
126 gauge_t snum, mnum, lnum;
127 FILE *loadavg;
128 char buffer[16];
130 char *fields[8];
131 int numfields;
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 }
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 }
146 if (fclose(loadavg)) {
147 char errbuf[1024];
148 WARNING("load: fclose: %s", sstrerror(errno, errbuf, sizeof(errbuf)));
149 }
151 numfields = strsplit(buffer, fields, 8);
153 if (numfields < 3)
154 return (-1);
156 snum = atof(fields[0]);
157 mnum = atof(fields[1]);
158 lnum = atof(fields[2]);
160 load_submit(snum, mnum, lnum);
161 /* #endif KERNEL_LINUX */
163 #elif HAVE_LIBSTATGRAB
164 gauge_t snum, mnum, lnum;
165 sg_load_stats *ls;
167 if ((ls = sg_get_load_stats()) == NULL)
168 return;
170 snum = ls->min1;
171 mnum = ls->min5;
172 lnum = ls->min15;
173 load_submit(snum, mnum, lnum);
174 /* #endif HAVE_LIBSTATGRAB */
176 #elif HAVE_PERFSTAT
177 gauge_t snum, mnum, lnum;
178 perfstat_cpu_total_t cputotal;
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 }
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 */
194 #else
195 #error "No applicable input method."
196 #endif
198 return (0);
199 }
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 */