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 <sys/proc.h> /* AIX 5 */
54 # include <sys/protosw.h>
55 # include <libperfstat.h>
56 #endif /* HAVE_PERFSTAT */
58 static _Bool report_relative_load = 0;
60 static const char *config_keys[] =
61 {
62 "ReportRelative"
63 };
64 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
66 static int load_config (const char *key, const char *value)
67 {
68 if (strcasecmp (key, "ReportRelative") == 0)
69 #ifdef _SC_NPROCESSORS_ONLN
70 report_relative_load = IS_TRUE (value) ? 1 : 0;
71 #else
72 WARNING ("load plugin: The \"ReportRelative\" configuration "
73 "is not available, because I can't determine the "
74 "number of CPUS on this system. Sorry.");
75 #endif
76 return (-1);
78 }
79 static void load_submit (gauge_t snum, gauge_t mnum, gauge_t lnum)
80 {
81 int cores = 0;
82 char errbuf[1024];
84 #ifdef _SC_NPROCESSORS_ONLN
85 if (report_relative_load) {
86 if ((cores = sysconf(_SC_NPROCESSORS_ONLN)) < 1) {
87 WARNING ("load: sysconf failed : %s",
88 sstrerror (errno, errbuf, sizeof (errbuf)));
89 }
90 }
91 #endif
92 if (cores > 0) {
93 snum /= cores;
94 mnum /= cores;
95 lnum /= cores;
96 }
98 value_list_t vl = VALUE_LIST_INIT;
99 value_t values[] = {
100 { .gauge = snum },
101 { .gauge = mnum },
102 { .gauge = lnum },
103 };
105 vl.values = values;
106 vl.values_len = STATIC_ARRAY_SIZE (values);
108 sstrncpy (vl.plugin, "load", sizeof (vl.plugin));
109 sstrncpy (vl.type, "load", sizeof (vl.type));
111 if (cores > 0) {
112 sstrncpy(vl.type_instance, "relative",
113 sizeof (vl.type_instance));
114 }
116 plugin_dispatch_values (&vl);
117 }
119 static int load_read (void)
120 {
121 #if defined(HAVE_GETLOADAVG)
122 double load[3];
124 if (getloadavg (load, 3) == 3)
125 load_submit (load[LOADAVG_1MIN], load[LOADAVG_5MIN], load[LOADAVG_15MIN]);
126 else
127 {
128 char errbuf[1024];
129 WARNING ("load: getloadavg failed: %s",
130 sstrerror (errno, errbuf, sizeof (errbuf)));
131 }
132 /* #endif HAVE_GETLOADAVG */
134 #elif defined(KERNEL_LINUX)
135 gauge_t snum, mnum, lnum;
136 FILE *loadavg;
137 char buffer[16];
139 char *fields[8];
140 int numfields;
142 if ((loadavg = fopen ("/proc/loadavg", "r")) == NULL)
143 {
144 char errbuf[1024];
145 WARNING ("load: fopen: %s",
146 sstrerror (errno, errbuf, sizeof (errbuf)));
147 return (-1);
148 }
150 if (fgets (buffer, 16, loadavg) == NULL)
151 {
152 char errbuf[1024];
153 WARNING ("load: fgets: %s",
154 sstrerror (errno, errbuf, sizeof (errbuf)));
155 fclose (loadavg);
156 return (-1);
157 }
159 if (fclose (loadavg))
160 {
161 char errbuf[1024];
162 WARNING ("load: fclose: %s",
163 sstrerror (errno, errbuf, sizeof (errbuf)));
164 }
166 numfields = strsplit (buffer, fields, 8);
168 if (numfields < 3)
169 return (-1);
171 snum = atof (fields[0]);
172 mnum = atof (fields[1]);
173 lnum = atof (fields[2]);
175 load_submit(snum, mnum, lnum);
176 /* #endif KERNEL_LINUX */
178 #elif HAVE_LIBSTATGRAB
179 gauge_t snum, mnum, lnum;
180 sg_load_stats *ls;
182 if ((ls = sg_get_load_stats ()) == NULL)
183 return;
185 snum = ls->min1;
186 mnum = ls->min5;
187 lnum = ls->min15;
188 load_submit(snum, mnum, lnum);
189 /* #endif HAVE_LIBSTATGRAB */
191 #elif HAVE_PERFSTAT
192 gauge_t snum, mnum, lnum;
193 perfstat_cpu_total_t cputotal;
195 if (perfstat_cpu_total(NULL, &cputotal, sizeof(perfstat_cpu_total_t), 1) < 0)
196 {
197 char errbuf[1024];
198 WARNING ("load: perfstat_cpu : %s",
199 sstrerror (errno, errbuf, sizeof (errbuf)));
200 return (-1);
201 }
203 snum = (float)cputotal.loadavg[0]/(float)(1<<SBITS);
204 mnum = (float)cputotal.loadavg[1]/(float)(1<<SBITS);
205 lnum = (float)cputotal.loadavg[2]/(float)(1<<SBITS);
206 load_submit(snum, mnum, lnum);
207 /* #endif HAVE_PERFSTAT */
209 #else
210 # error "No applicable input method."
211 #endif
213 return (0);
214 }
216 void module_register (void)
217 {
218 plugin_register_config ("load", load_config, config_keys, config_keys_num);
219 plugin_register_read ("load", load_read);
220 } /* void module_register */