7188b13e366b3b4b0f40bd4b8dbe5e8418b1bcd7
1 /**
2 * collectd - src/vserver.c
3 * Copyright (C) 2006,2007 Sebastian Harl
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; only version 2 of the license is applicable.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
17 *
18 * Authors:
19 * Sebastian Harl <sh at tokkee.org>
20 **/
22 #include "collectd.h"
23 #include "common.h"
24 #include "plugin.h"
26 #include <dirent.h>
27 #include <sys/types.h>
29 #define BUFSIZE 512
31 #define PROCDIR "/proc/virtual"
33 #if !KERNEL_LINUX
34 # error "No applicable input method."
35 #endif
37 static int pagesize = 0;
39 static int vserver_init (void)
40 {
41 /* XXX Should we check for getpagesize () in configure?
42 * What's the right thing to do, if there is no getpagesize ()? */
43 pagesize = getpagesize ();
45 return (0);
46 } /* static void vserver_init(void) */
48 static void traffic_submit (const char *plugin_instance,
49 const char *type_instance, counter_t rx, counter_t tx)
50 {
51 value_t values[2];
52 value_list_t vl = VALUE_LIST_INIT;
54 values[0].counter = rx;
55 values[1].counter = tx;
57 vl.values = values;
58 vl.values_len = STATIC_ARRAY_SIZE (values);
59 vl.time = time (NULL);
60 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
61 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
62 strncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
63 strncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
65 plugin_dispatch_values ("if_octets", &vl);
66 } /* void traffic_submit */
68 static void load_submit (const char *plugin_instance,
69 gauge_t snum, gauge_t mnum, gauge_t lnum)
70 {
71 value_t values[3];
72 value_list_t vl = VALUE_LIST_INIT;
74 values[0].gauge = snum;
75 values[1].gauge = mnum;
76 values[2].gauge = lnum;
78 vl.values = values;
79 vl.values_len = STATIC_ARRAY_SIZE (values);
80 vl.time = time (NULL);
81 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
82 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
83 strncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
85 plugin_dispatch_values ("load", &vl);
86 }
88 static void submit_gauge (const char *plugin_instance, const char *type,
89 const char *type_instance, gauge_t value)
91 {
92 value_t values[1];
93 value_list_t vl = VALUE_LIST_INIT;
95 values[0].gauge = value;
97 vl.values = values;
98 vl.values_len = STATIC_ARRAY_SIZE (values);
99 vl.time = time (NULL);
100 sstrncpy (vl.host, hostname_g, sizeof (vl.host));
101 sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
102 strncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
103 strncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
105 plugin_dispatch_values (type, &vl);
106 } /* void submit_gauge */
108 static inline long long __get_sock_bytes(const char *s)
109 {
110 while (s[0] != '/')
111 ++s;
113 /* Remove '/' */
114 ++s;
115 return atoll(s);
116 }
118 static int vserver_read (void)
119 {
120 #if NAME_MAX < 1024
121 # define DIRENT_BUFFER_SIZE (sizeof (struct dirent) + 1024 + 1)
122 #else
123 # define DIRENT_BUFFER_SIZE (sizeof (struct dirent) + NAME_MAX + 1)
124 #endif
126 DIR *proc;
127 struct dirent *dent; /* 42 */
128 char dirent_buffer[DIRENT_BUFFER_SIZE];
130 errno = 0;
131 proc = opendir (PROCDIR);
132 if (proc == NULL)
133 {
134 char errbuf[1024];
135 ERROR ("vserver plugin: fopen (%s): %s", PROCDIR,
136 sstrerror (errno, errbuf, sizeof (errbuf)));
137 return (-1);
138 }
140 while (42)
141 {
142 size_t len;
143 char file[BUFSIZE];
145 FILE *fh;
146 char buffer[BUFSIZE];
148 struct stat statbuf;
149 char *cols[4];
151 int status;
153 status = readdir_r (proc, (struct dirent *) dirent_buffer, &dent);
154 if (status != 0)
155 {
156 char errbuf[4096];
157 ERROR ("vserver plugin: readdir_r failed: %s",
158 sstrerror (errno, errbuf, sizeof (errbuf)));
159 closedir (proc);
160 return (-1);
161 }
162 else if (dent == NULL)
163 {
164 /* end of directory */
165 break;
166 }
168 if (dent->d_name[0] == '.')
169 continue;
171 len = snprintf (file, sizeof (file), PROCDIR "/%s", dent->d_name);
172 if ((len < 0) || (len >= BUFSIZE))
173 continue;
175 status = stat (file, &statbuf);
176 if (status != 0)
177 {
178 char errbuf[4096];
179 WARNING ("vserver plugin: stat (%s) failed: %s",
180 file, sstrerror (errno, errbuf, sizeof (errbuf)));
181 continue;
182 }
184 if (!S_ISDIR (statbuf.st_mode))
185 continue;
187 /* socket message accounting */
188 len = snprintf (file, BUFSIZE, PROCDIR "/%s/cacct", dent->d_name);
189 if ((len < 0) || (len >= BUFSIZE))
190 continue;
192 if (NULL == (fh = fopen (file, "r")))
193 {
194 char errbuf[1024];
195 ERROR ("Cannot open '%s': %s", file,
196 sstrerror (errno, errbuf, sizeof (errbuf)));
197 }
199 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
200 {
201 counter_t rx;
202 counter_t tx;
203 char *type_instance;
205 if (strsplit (buffer, cols, 4) < 4)
206 continue;
208 if (0 == strcmp (cols[0], "UNIX:"))
209 type_instance = "unix";
210 else if (0 == strcmp (cols[0], "INET:"))
211 type_instance = "inet";
212 else if (0 == strcmp (cols[0], "INET6:"))
213 type_instance = "inet6";
214 else if (0 == strcmp (cols[0], "OTHER:"))
215 type_instance = "other";
216 else if (0 == strcmp (cols[0], "UNSPEC:"))
217 type_instance = "unspec";
218 else
219 continue;
221 rx = __get_sock_bytes (cols[1]);
222 tx = __get_sock_bytes (cols[2]);
223 /* cols[3] == errors */
225 traffic_submit (dent->d_name, type_instance, rx, tx);
226 } /* while (fgets) */
228 if (fh != NULL)
229 {
230 fclose (fh);
231 fh = NULL;
232 }
234 /* thread information and load */
235 len = snprintf (file, BUFSIZE, PROCDIR "/%s/cvirt", dent->d_name);
236 if ((len < 0) || (len >= BUFSIZE))
237 continue;
239 if (NULL == (fh = fopen (file, "r")))
240 {
241 char errbuf[1024];
242 ERROR ("Cannot open '%s': %s", file,
243 sstrerror (errno, errbuf, sizeof (errbuf)));
244 }
246 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
247 {
248 int n = strsplit (buffer, cols, 4);
250 if (2 == n)
251 {
252 char *type_instance;
253 gauge_t value;
255 if (0 == strcmp (cols[0], "nr_threads:"))
256 type_instance = "total";
257 else if (0 == strcmp (cols[0], "nr_running:"))
258 type_instance = "running";
259 else if (0 == strcmp (cols[0], "nr_unintr:"))
260 type_instance = "uninterruptable";
261 else if (0 == strcmp (cols[0], "nr_onhold:"))
262 type_instance = "onhold";
263 else
264 continue;
266 value = atof (cols[1]);
267 submit_gauge (dent->d_name, "vs_threads", type_instance, value);
268 }
269 else if (4 == n) {
270 if (0 == strcmp (cols[0], "loadavg:"))
271 {
272 gauge_t snum = atof (cols[1]);
273 gauge_t mnum = atof (cols[2]);
274 gauge_t lnum = atof (cols[3]);
275 load_submit (dent->d_name, snum, mnum, lnum);
276 }
277 }
278 } /* while (fgets) */
280 if (fh != NULL)
281 {
282 fclose (fh);
283 fh = NULL;
284 }
286 /* processes and memory usage */
287 len = snprintf (file, BUFSIZE, PROCDIR "/%s/limit", dent->d_name);
288 if ((len < 0) || (len >= BUFSIZE))
289 continue;
291 if (NULL == (fh = fopen (file, "r")))
292 {
293 char errbuf[1024];
294 ERROR ("Cannot open '%s': %s", file,
295 sstrerror (errno, errbuf, sizeof (errbuf)));
296 }
298 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
299 {
300 char *type = "vs_memory";
301 char *type_instance;
302 gauge_t value;
304 if (strsplit (buffer, cols, 2) < 2)
305 continue;
307 if (0 == strcmp (cols[0], "PROC:"))
308 {
309 type = "vs_processes";
310 type_instance = "";
311 value = atof (cols[1]);
312 }
313 else
314 {
315 if (0 == strcmp (cols[0], "VM:"))
316 type_instance = "vm";
317 else if (0 == strcmp (cols[0], "VML:"))
318 type_instance = "vml";
319 else if (0 == strcmp (cols[0], "RSS:"))
320 type_instance = "rss";
321 else if (0 == strcmp (cols[0], "ANON:"))
322 type_instance = "anon";
323 else
324 continue;
326 value = atof (cols[1]) * pagesize;
327 }
329 submit_gauge (dent->d_name, type, type_instance, value);
330 } /* while (fgets) */
332 if (fh != NULL)
333 {
334 fclose (fh);
335 fh = NULL;
336 }
337 } /* while (readdir) */
339 closedir (proc);
341 return (0);
342 } /* int vserver_read */
344 void module_register (void)
345 {
346 plugin_register_init ("vserver", vserver_init);
347 plugin_register_read ("vserver", vserver_read);
348 } /* void module_register(void) */
350 /* vim: set ts=4 sw=4 noexpandtab : */