src/vserver.c

   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  **/
  28
  29 #include "collectd.h"
  30 #include "common.h"
  31 #include "plugin.h"
  32
  33 #include <dirent.h>
  34 #include <sys/types.h>
  35
  36 #define BUFSIZE 512
  37
  38 #define PROCDIR "/proc/virtual"
  39
  40 #if !KERNEL_LINUX
  41 # error "No applicable input method."
  42 #endif
  43
  44 static int pagesize = 0;
  45
  46 static int vserver_init (void)
  47 {
  48         /* XXX Should we check for getpagesize () in configure?
  49          * What's the right thing to do, if there is no getpagesize ()? */
  50         pagesize = getpagesize ();
  51
  52         return (0);
  53 } /* static void vserver_init(void) */
  54
  55 static void traffic_submit (const char *plugin_instance,
  56                 const char *type_instance, derive_t rx, derive_t tx)
  57 {
  58         value_t values[2];
  59         value_list_t vl = VALUE_LIST_INIT;
  60
  61         values[0].derive = rx;
  62         values[1].derive = tx;
  63
  64         vl.values = values;
  65         vl.values_len = STATIC_ARRAY_SIZE (values);
  66         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
  67         sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
  68         sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
  69         sstrncpy (vl.type, "if_octets", sizeof (vl.type));
  70         sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
  71
  72         plugin_dispatch_values (&vl);
  73 } /* void traffic_submit */
  74
  75 static void load_submit (const char *plugin_instance,
  76                 gauge_t snum, gauge_t mnum, gauge_t lnum)
  77 {
  78         value_t values[3];
  79         value_list_t vl = VALUE_LIST_INIT;
  80
  81         values[0].gauge = snum;
  82         values[1].gauge = mnum;
  83         values[2].gauge = lnum;
  84
  85         vl.values = values;
  86         vl.values_len = STATIC_ARRAY_SIZE (values);
  87         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
  88         sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
  89         sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
  90         sstrncpy (vl.type, "load", sizeof (vl.type));
  91
  92         plugin_dispatch_values (&vl);
  93 }
  94
  95 static void submit_gauge (const char *plugin_instance, const char *type,
  96                 const char *type_instance, gauge_t value)
  97
  98 {
  99         value_t values[1];
 100         value_list_t vl = VALUE_LIST_INIT;
 101
 102         values[0].gauge = value;
 103
 104         vl.values = values;
 105         vl.values_len = STATIC_ARRAY_SIZE (values);
 106         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
 107         sstrncpy (vl.plugin, "vserver", sizeof (vl.plugin));
 108         sstrncpy (vl.plugin_instance, plugin_instance, sizeof (vl.plugin_instance));
 109         sstrncpy (vl.type, type, sizeof (vl.type));
 110         sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
 111
 112         plugin_dispatch_values (&vl);
 113 } /* void submit_gauge */
 114
 115 static derive_t vserver_get_sock_bytes(const char *s)
 116 {
 117         value_t v;
 118         int status;
 119
 120         while (s[0] != '/')
 121                 ++s;
 122
 123         /* Remove '/' */
 124         ++s;
 125
 126         status = parse_value (s, &v, DS_TYPE_DERIVE);
 127         if (status != 0)
 128                 return (-1);
 129         return (v.derive);
 130 }
 131
 132 static int vserver_read (void)
 133 {
 134 #if NAME_MAX < 1024
 135 # define DIRENT_BUFFER_SIZE (sizeof (struct dirent) + 1024 + 1)
 136 #else
 137 # define DIRENT_BUFFER_SIZE (sizeof (struct dirent) + NAME_MAX + 1)
 138 #endif
 139
 140         DIR                     *proc;
 141         struct dirent   *dent; /* 42 */
 142         char dirent_buffer[DIRENT_BUFFER_SIZE];
 143
 144         errno = 0;
 145         proc = opendir (PROCDIR);
 146         if (proc == NULL)
 147         {
 148                 char errbuf[1024];
 149                 ERROR ("vserver plugin: fopen (%s): %s", PROCDIR,
 150                                 sstrerror (errno, errbuf, sizeof (errbuf)));
 151                 return (-1);
 152         }
 153
 154         while (42)
 155         {
 156                 int len;
 157                 char file[BUFSIZE];
 158
 159                 FILE *fh;
 160                 char buffer[BUFSIZE];
 161
 162                 struct stat statbuf;
 163                 char *cols[4];
 164
 165                 int status;
 166
 167                 status = readdir_r (proc, (struct dirent *) dirent_buffer, &dent);
 168                 if (status != 0)
 169                 {
 170                         char errbuf[4096];
 171                         ERROR ("vserver plugin: readdir_r failed: %s",
 172                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 173                         closedir (proc);
 174                         return (-1);
 175                 }
 176                 else if (dent == NULL)
 177                 {
 178                         /* end of directory */
 179                         break;
 180                 }
 181
 182                 if (dent->d_name[0] == '.')
 183                         continue;
 184
 185                 len = ssnprintf (file, sizeof (file), PROCDIR "/%s", dent->d_name);
 186                 if ((len < 0) || (len >= BUFSIZE))
 187                         continue;
 188
 189                 status = stat (file, &statbuf);
 190                 if (status != 0)
 191                 {
 192                         char errbuf[4096];
 193                         WARNING ("vserver plugin: stat (%s) failed: %s",
 194                                         file, sstrerror (errno, errbuf, sizeof (errbuf)));
 195                         continue;
 196                 }
 197
 198                 if (!S_ISDIR (statbuf.st_mode))
 199                         continue;
 200
 201                 /* socket message accounting */
 202                 len = ssnprintf (file, sizeof (file),
 203                                 PROCDIR "/%s/cacct", dent->d_name);
 204                 if ((len < 0) || ((size_t) len >= sizeof (file)))
 205                         continue;
 206
 207                 if (NULL == (fh = fopen (file, "r")))
 208                 {
 209                         char errbuf[1024];
 210                         ERROR ("Cannot open '%s': %s", file,
 211                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 212                 }
 213
 214                 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
 215                 {
 216                         derive_t rx;
 217                         derive_t tx;
 218                         char *type_instance;
 219
 220                         if (strsplit (buffer, cols, 4) < 4)
 221                                 continue;
 222
 223                         if (0 == strcmp (cols[0], "UNIX:"))
 224                                 type_instance = "unix";
 225                         else if (0 == strcmp (cols[0], "INET:"))
 226                                 type_instance = "inet";
 227                         else if (0 == strcmp (cols[0], "INET6:"))
 228                                 type_instance = "inet6";
 229                         else if (0 == strcmp (cols[0], "OTHER:"))
 230                                 type_instance = "other";
 231                         else if (0 == strcmp (cols[0], "UNSPEC:"))
 232                                 type_instance = "unspec";
 233                         else
 234                                 continue;
 235
 236                         rx = vserver_get_sock_bytes (cols[1]);
 237                         tx = vserver_get_sock_bytes (cols[2]);
 238                         /* cols[3] == errors */
 239
 240                         traffic_submit (dent->d_name, type_instance, rx, tx);
 241                 } /* while (fgets) */
 242
 243                 if (fh != NULL)
 244                 {
 245                         fclose (fh);
 246                         fh = NULL;
 247                 }
 248
 249                 /* thread information and load */
 250                 len = ssnprintf (file, sizeof (file),
 251                                 PROCDIR "/%s/cvirt", dent->d_name);
 252                 if ((len < 0) || ((size_t) len >= sizeof (file)))
 253                         continue;
 254
 255                 if (NULL == (fh = fopen (file, "r")))
 256                 {
 257                         char errbuf[1024];
 258                         ERROR ("Cannot open '%s': %s", file,
 259                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 260                 }
 261
 262                 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
 263                 {
 264                         int n = strsplit (buffer, cols, 4);
 265
 266                         if (2 == n)
 267                         {
 268                                 char   *type_instance;
 269                                 gauge_t value;
 270
 271                                 if (0 == strcmp (cols[0], "nr_threads:"))
 272                                         type_instance = "total";
 273                                 else if (0 == strcmp (cols[0], "nr_running:"))
 274                                         type_instance = "running";
 275                                 else if (0 == strcmp (cols[0], "nr_unintr:"))
 276                                         type_instance = "uninterruptable";
 277                                 else if (0 == strcmp (cols[0], "nr_onhold:"))
 278                                         type_instance = "onhold";
 279                                 else
 280                                         continue;
 281
 282                                 value = atof (cols[1]);
 283                                 submit_gauge (dent->d_name, "vs_threads", type_instance, value);
 284                         }
 285                         else if (4 == n) {
 286                                 if (0 == strcmp (cols[0], "loadavg:"))
 287                                 {
 288                                         gauge_t snum = atof (cols[1]);
 289                                         gauge_t mnum = atof (cols[2]);
 290                                         gauge_t lnum = atof (cols[3]);
 291                                         load_submit (dent->d_name, snum, mnum, lnum);
 292                                 }
 293                         }
 294                 } /* while (fgets) */
 295
 296                 if (fh != NULL)
 297                 {
 298                         fclose (fh);
 299                         fh = NULL;
 300                 }
 301
 302                 /* processes and memory usage */
 303                 len = ssnprintf (file, sizeof (file),
 304                                 PROCDIR "/%s/limit", dent->d_name);
 305                 if ((len < 0) || ((size_t) len >= sizeof (file)))
 306                         continue;
 307
 308                 if (NULL == (fh = fopen (file, "r")))
 309                 {
 310                         char errbuf[1024];
 311                         ERROR ("Cannot open '%s': %s", file,
 312                                         sstrerror (errno, errbuf, sizeof (errbuf)));
 313                 }
 314
 315                 while ((fh != NULL) && (NULL != fgets (buffer, BUFSIZE, fh)))
 316                 {
 317                         char *type = "vs_memory";
 318                         char *type_instance;
 319                         gauge_t value;
 320
 321                         if (strsplit (buffer, cols, 2) < 2)
 322                                 continue;
 323
 324                         if (0 == strcmp (cols[0], "PROC:"))
 325                         {
 326                                 type = "vs_processes";
 327                                 type_instance = "";
 328                                 value = atof (cols[1]);
 329                         }
 330                         else
 331                         {
 332                                 if (0 == strcmp (cols[0], "VM:"))
 333                                         type_instance = "vm";
 334                                 else if (0 == strcmp (cols[0], "VML:"))
 335                                         type_instance = "vml";
 336                                 else if (0 == strcmp (cols[0], "RSS:"))
 337                                         type_instance = "rss";
 338                                 else if (0 == strcmp (cols[0], "ANON:"))
 339                                         type_instance = "anon";
 340                                 else
 341                                         continue;
 342
 343                                 value = atof (cols[1]) * pagesize;
 344                         }
 345
 346                         submit_gauge (dent->d_name, type, type_instance, value);
 347                 } /* while (fgets) */
 348
 349                 if (fh != NULL)
 350                 {
 351                         fclose (fh);
 352                         fh = NULL;
 353                 }
 354         } /* while (readdir) */
 355
 356         closedir (proc);
 357
 358         return (0);
 359 } /* int vserver_read */
 360
 361 void module_register (void)
 362 {
 363         plugin_register_init ("vserver", vserver_init);
 364         plugin_register_read ("vserver", vserver_read);
 365 } /* void module_register(void) */
 366
 367 /* vim: set ts=4 sw=4 noexpandtab : */