1 /**
2 * collectd - src/vserver.c
3 * Copyright (C) 2006,2007 Sebastian Harl
4 * Copyright (C) 2007-2010 Florian octo Forster
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a
7 * copy of this software and associated documentation files (the "Software"),
8 * to deal in the Software without restriction, including without limitation
9 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10 * and/or sell copies of the Software, and to permit persons to whom the
11 * Software is furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in
14 * all copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22 * DEALINGS IN THE SOFTWARE.
23 *
24 * Authors:
25 * Sebastian Harl <sh at tokkee.org>
26 * Florian octo Forster <octo at collectd.org>
27 **/
29 #include "collectd.h"
31 #include "common.h"
32 #include "plugin.h"
34 #include <dirent.h>
35 #include <sys/types.h>
37 #define BUFSIZE 512
39 #define PROCDIR "/proc/virtual"
41 #if !KERNEL_LINUX
42 # error "No applicable input method."
43 #endif
45 static int pagesize = 0;
47 static int vserver_init (void)
48 {
49 /* XXX Should we check for getpagesize () in configure?
50 * What's the right thing to do, if there is no getpagesize ()? */
51 pagesize = getpagesize ();
53 return (0);
54 } /* static void vserver_init(void) */
56 static void traffic_submit (const char *plugin_instance,
57 const char *type_instance, derive_t rx, derive_t tx)
58 {
59 value_t values[2];
60 value_list_t vl = VALUE_LIST_INIT;
62 values[0].derive = rx;
63 values[1].derive = tx;
65 vl.values = values;
66 vl.values_len = STATIC_ARRAY_SIZE (values);
67 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
68 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
69 sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
70 sstrncpy (vl.type, "if_octets", sizeof (vl.type));
71 sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
73 plugin_dispatch_values (&vl);
74 } /* void traffic_submit */
76 static void load_submit (const char *plugin_instance,
77 gauge_t snum, gauge_t mnum, gauge_t lnum)
78 {
79 value_t values[3];
80 value_list_t vl = VALUE_LIST_INIT;
82 values[0].gauge = snum;
83 values[1].gauge = mnum;
84 values[2].gauge = lnum;
86 vl.values = values;
87 vl.values_len = STATIC_ARRAY_SIZE (values);
88 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
89 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
90 sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
91 sstrncpy (vl.type, "load", sizeof (vl.type));
93 plugin_dispatch_values (&vl);
94 }
96 static void submit_gauge (const char *plugin_instance, const char *type,
97 const char *type_instance, gauge_t value)
99 {
100 value_t values[1];
101 value_list_t vl = VALUE_LIST_INIT;
103 values[0].gauge = value;
105 vl.values = values;
106 vl.values_len = STATIC_ARRAY_SIZE (values);
107 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
108 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
109 sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
110 sstrncpy (vl.type, type, sizeof (vl.type));
111 sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
113 plugin_dispatch_values (&vl);
114 } /* void submit_gauge */
116 static derive_t vserver_get_sock_bytes(const char *s)
117 {
118 value_t v;
119 int status;
121 while (s[0] != '/')
122 ++s;
124 /* Remove '/' */
125 ++s;
127 status = parse_value (s, &v, DS_TYPE_DERIVE);
128 if (status != 0)
129 return (-1);
130 return (v.derive);
131 }
133 static int vserver_read (void)
134 {
135 DIR *proc;
137 errno = 0;
138 proc = opendir (PROCDIR);
139 if (proc == NULL)
140 {
141 char errbuf[1024];
142 ERROR ("vserver plugin: fopen (%s): %s", PROCDIR,
143 sstrerror (errno, errbuf, sizeof (errbuf)));
144 return (-1);
145 }
147 while (42)
148 {
149 struct dirent *dent;
150 int len;
151 char file[BUFSIZE];
153 FILE *fh;
154 char buffer[BUFSIZE];
156 struct stat statbuf;
157 char *cols[4];
159 int status;
161 errno = 0;
162 dent = readdir (proc);
163 if (dent == NULL)
164 {
165 char errbuf[4096];
167 if (errno == 0) /* end of directory */
168 break;
170 ERROR ("vserver plugin: failed to read directory %s: %s",
171 PROCDIR, sstrerror (errno, errbuf, sizeof (errbuf)));
172 closedir (proc);
173 return (-1);
174 }
176 if (dent->d_name[0] == '.')
177 continue;
179 len = ssnprintf (file, sizeof (file), PROCDIR "/%s", dent->d_name);
180 if ((len < 0) || (len >= BUFSIZE))
181 continue;
183 status = stat (file, &statbuf);
184 if (status != 0)
185 {
186 char errbuf[4096];
187 WARNING ("vserver plugin: stat (%s) failed: %s",
188 file, sstrerror (errno, errbuf, sizeof (errbuf)));
189 continue;
190 }
192 if (!S_ISDIR (statbuf.st_mode))
193 continue;
195 /* socket message accounting */
196 len = ssnprintf (file, sizeof (file),
197 PROCDIR "/%s/cacct", dent->d_name);
198 if ((len < 0) || ((size_t) len >= sizeof (file)))
199 continue;
201 if (NULL == (fh = fopen (file, "r")))
202 {
203 char errbuf[1024];
204 ERROR ("Cannot open '%s': %s", file,
205 sstrerror (errno, errbuf, sizeof (errbuf)));
206 }
208 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
209 {
210 derive_t rx;
211 derive_t tx;
212 const char *type_instance;
214 if (strsplit (buffer, cols, 4) < 4)
215 continue;
217 if (0 == strcmp (cols[0], "UNIX:"))
218 type_instance = "unix";
219 else if (0 == strcmp (cols[0], "INET:"))
220 type_instance = "inet";
221 else if (0 == strcmp (cols[0], "INET6:"))
222 type_instance = "inet6";
223 else if (0 == strcmp (cols[0], "OTHER:"))
224 type_instance = "other";
225 else if (0 == strcmp (cols[0], "UNSPEC:"))
226 type_instance = "unspec";
227 else
228 continue;
230 rx = vserver_get_sock_bytes (cols[1]);
231 tx = vserver_get_sock_bytes (cols[2]);
232 /* cols[3] == errors */
234 traffic_submit (dent->d_name, type_instance, rx, tx);
235 } /* while (fgets) */
237 if (fh != NULL)
238 {
239 fclose (fh);
240 fh = NULL;
241 }
243 /* thread information and load */
244 len = ssnprintf (file, sizeof (file),
245 PROCDIR "/%s/cvirt", dent->d_name);
246 if ((len < 0) || ((size_t) len >= sizeof (file)))
247 continue;
249 if (NULL == (fh = fopen (file, "r")))
250 {
251 char errbuf[1024];
252 ERROR ("Cannot open '%s': %s", file,
253 sstrerror (errno, errbuf, sizeof (errbuf)));
254 }
256 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
257 {
258 int n = strsplit (buffer, cols, 4);
260 if (2 == n)
261 {
262 const char *type_instance;
263 gauge_t value;
265 if (0 == strcmp (cols[0], "nr_threads:"))
266 type_instance = "total";
267 else if (0 == strcmp (cols[0], "nr_running:"))
268 type_instance = "running";
269 else if (0 == strcmp (cols[0], "nr_unintr:"))
270 type_instance = "uninterruptable";
271 else if (0 == strcmp (cols[0], "nr_onhold:"))
272 type_instance = "onhold";
273 else
274 continue;
276 value = atof (cols[1]);
277 submit_gauge (dent->d_name, "vs_threads", type_instance, value);
278 }
279 else if (4 == n) {
280 if (0 == strcmp (cols[0], "loadavg:"))
281 {
282 gauge_t snum = atof (cols[1]);
283 gauge_t mnum = atof (cols[2]);
284 gauge_t lnum = atof (cols[3]);
285 load_submit (dent->d_name, snum, mnum, lnum);
286 }
287 }
288 } /* while (fgets) */
290 if (fh != NULL)
291 {
292 fclose (fh);
293 fh = NULL;
294 }
296 /* processes and memory usage */
297 len = ssnprintf (file, sizeof (file),
298 PROCDIR "/%s/limit", dent->d_name);
299 if ((len < 0) || ((size_t) len >= sizeof (file)))
300 continue;
302 if (NULL == (fh = fopen (file, "r")))
303 {
304 char errbuf[1024];
305 ERROR ("Cannot open '%s': %s", file,
306 sstrerror (errno, errbuf, sizeof (errbuf)));
307 }
309 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
310 {
311 const char *type = "vs_memory";
312 const char *type_instance;
313 gauge_t value;
315 if (strsplit (buffer, cols, 2) < 2)
316 continue;
318 if (0 == strcmp (cols[0], "PROC:"))
319 {
320 type = "vs_processes";
321 type_instance = "";
322 value = atof (cols[1]);
323 }
324 else
325 {
326 if (0 == strcmp (cols[0], "VM:"))
327 type_instance = "vm";
328 else if (0 == strcmp (cols[0], "VML:"))
329 type_instance = "vml";
330 else if (0 == strcmp (cols[0], "RSS:"))
331 type_instance = "rss";
332 else if (0 == strcmp (cols[0], "ANON:"))
333 type_instance = "anon";
334 else
335 continue;
337 value = atof (cols[1]) * pagesize;
338 }
340 submit_gauge (dent->d_name, type, type_instance, value);
341 } /* while (fgets) */
343 if (fh != NULL)
344 {
345 fclose (fh);
346 fh = NULL;
347 }
348 } /* while (readdir) */
350 closedir (proc);
352 return (0);
353 } /* int vserver_read */
355 void module_register (void)
356 {
357 plugin_register_init ("vserver", vserver_init);
358 plugin_register_read ("vserver", vserver_read);
359 } /* void module_register(void) */
361 /* vim: set ts=4 sw=4 noexpandtab : */