all plugins: remove pthread.h include
[collectd.git] / src / md.c
1 /**
2  * collectd - src/md.c
3  * Copyright (C) 2010,2011  Michael Hanselmann
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  * Author:
19  *   Michael Hanselmann
20  **/
21
22 #include "collectd.h"
23 #include "common.h"
24 #include "plugin.h"
25 #include "utils_ignorelist.h"
26
27 #include <sys/ioctl.h>
28
29 #include <linux/major.h>
30 #include <linux/raid/md_u.h>
31
32 #define PROC_DISKSTATS "/proc/diskstats"
33 #define DEV_DIR "/dev"
34
35 static const char *config_keys[] =
36 {
37   "Device",
38   "IgnoreSelected"
39 };
40 static int config_keys_num = STATIC_ARRAY_SIZE (config_keys);
41
42 static ignorelist_t *ignorelist = NULL;
43
44 static int md_config (const char *key, const char *value)
45 {
46   if (ignorelist == NULL)
47     ignorelist = ignorelist_create (/* invert = */ 1);
48   if (ignorelist == NULL)
49     return (1);
50
51   if (strcasecmp (key, "Device") == 0)
52   {
53     ignorelist_add (ignorelist, value);
54   }
55   else if (strcasecmp (key, "IgnoreSelected") == 0)
56   {
57     ignorelist_set_invert (ignorelist, IS_TRUE (value) ? 0 : 1);
58   }
59   else
60   {
61     return (-1);
62   }
63
64   return (0);
65 }
66
67 static void md_submit (const int minor, const char *type_instance,
68     gauge_t value)
69 {
70   value_t values[1];
71   value_list_t vl = VALUE_LIST_INIT;
72
73   values[0].gauge = value;
74
75   vl.values = values;
76   vl.values_len = 1;
77   sstrncpy (vl.host, hostname_g, sizeof (vl.host));
78   sstrncpy (vl.plugin, "md", sizeof (vl.plugin));
79   ssnprintf (vl.plugin_instance, sizeof (vl.plugin_instance),
80       "%i", minor);
81   sstrncpy (vl.type, "md_disks", sizeof (vl.type));
82   sstrncpy (vl.type_instance, type_instance,
83       sizeof (vl.type_instance));
84
85   plugin_dispatch_values (&vl);
86 } /* void md_submit */
87
88 static void md_process (const int minor, const char *path)
89 {
90   char errbuf[1024];
91   int fd;
92   struct stat st;
93   mdu_array_info_t array;
94   gauge_t disks_missing;
95
96   fd = open (path, O_RDONLY);
97   if (fd < 0)
98   {
99     WARNING ("md: open(%s): %s", path,
100         sstrerror (errno, errbuf, sizeof (errbuf)));
101     return;
102   }
103
104   if (fstat (fd, &st) < 0)
105   {
106     WARNING ("md: Unable to fstat file descriptor for %s: %s", path,
107         sstrerror (errno, errbuf, sizeof (errbuf)));
108     close (fd);
109     return;
110   }
111
112   if (! S_ISBLK (st.st_mode))
113   {
114     WARNING ("md: %s is no block device", path);
115     close (fd);
116     return;
117   }
118
119   if (st.st_rdev != makedev (MD_MAJOR, minor))
120   {
121     WARNING ("md: Major/minor of %s are %i:%i, should be %i:%i",
122         path, (int)major(st.st_rdev), (int)minor(st.st_rdev),
123         (int)MD_MAJOR, minor);
124     close (fd);
125     return;
126   }
127
128   /* Retrieve md information */
129   if (ioctl (fd, GET_ARRAY_INFO, &array) < 0) {
130     WARNING ("md: Unable to retrieve array info from %s: %s", path,
131         sstrerror (errno, errbuf, sizeof (errbuf)));
132     close (fd);
133     return;
134   }
135
136   close (fd);
137
138   /*
139    * The mdu_array_info_t structure contains numbers of disks in the array.
140    * However, disks are accounted for more than once:
141    *
142    * active:  Number of active (in sync) disks.
143    * spare:   Number of stand-by disks.
144    * working: Number of working disks. (active + sync)
145    * failed:  Number of failed disks.
146    * nr:      Number of physically present disks. (working + failed)
147    * raid:    Number of disks in the RAID. This may be larger than "nr" if
148    *          disks are missing and smaller than "nr" when spare disks are
149    *          around.
150    */
151   md_submit (minor, "active",  (gauge_t) array.active_disks);
152   md_submit (minor, "failed",  (gauge_t) array.failed_disks);
153   md_submit (minor, "spare",   (gauge_t) array.spare_disks);
154
155   disks_missing = 0.0;
156   if (array.raid_disks > array.nr_disks)
157     disks_missing = (gauge_t) (array.raid_disks - array.nr_disks);
158   md_submit (minor, "missing", disks_missing);
159 } /* void md_process */
160
161 static int md_read (void)
162 {
163   FILE *fh;
164   char buffer[1024];
165
166   fh = fopen (PROC_DISKSTATS, "r");
167   if (fh == NULL) {
168     char errbuf[1024];
169     WARNING ("md: Unable to open %s: %s",
170         PROC_DISKSTATS ,
171         sstrerror (errno, errbuf, sizeof (errbuf)));
172     return (-1);
173   }
174
175   /* Iterate md devices */
176   while (fgets (buffer, sizeof (buffer), fh) != NULL)
177   {
178     char path[PATH_MAX];
179     char *fields[4];
180     char *name;
181     int major, minor;
182
183     /* Extract interesting fields */
184     if (strsplit (buffer, fields, STATIC_ARRAY_SIZE(fields)) < 3)
185       continue;
186
187     major = atoi (fields[0]);
188
189     if (major != MD_MAJOR)
190       continue;
191
192     minor = atoi (fields[1]);
193     name = fields[2];
194
195     if (ignorelist_match (ignorelist, name))
196       continue;
197
198     /* FIXME: Don't hardcode path. Walk /dev collecting major,
199      * minor and name, then use lookup table to find device.
200      * Alternatively create a temporary device file with correct
201      * major/minor, but that again can be tricky if the filesystem
202      * with the device file is mounted using the "nodev" option.
203      */
204     ssnprintf (path, sizeof (path), "%s/%s", DEV_DIR, name);
205
206     md_process (minor, path);
207   }
208
209   fclose (fh);
210
211   return (0);
212 } /* int md_read */
213
214 void module_register (void)
215 {
216   plugin_register_config ("md", md_config, config_keys, config_keys_num);
217   plugin_register_read ("md", md_read);
218 } /* void module_register */