Tree wide: Reformat with clang-format.
[collectd.git] / src / lvm.c
1 /**
2  * collectd - src/lvm.c
3  * Copyright (C) 2013       Chad Malfait
4  * Copyright (C) 2014       Carnegie Mellon University
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU General Public License as published by the
8  * Free Software Foundation; only version 2 of the License is applicable.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License along
16  * with this program; if not, write to the Free Software Foundation, Inc.,
17  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
18  *
19  * Authors:
20  *   Chad Malfait <malfaitc at yahoo.com>
21  *   Benjamin Gilbert <bgilbert at cs.cmu.edu>
22  **/
23
24 #include <lvm2app.h>
25
26 #include "collectd.h"
27
28 #include "common.h"
29 #include "plugin.h"
30
31 #define NO_VALUE UINT64_MAX
32 #define PERCENT_SCALE_FACTOR 1e-8
33
34 static uint64_t get_lv_property_int(lv_t lv, char const *property) {
35   lvm_property_value_t v;
36
37   v = lvm_lv_get_property(lv, property);
38   if (!v.is_valid || !v.is_integer)
39     return NO_VALUE;
40   /* May be NO_VALUE if @property does not apply to this LV */
41   return v.value.integer;
42 }
43
44 static char const *get_lv_property_string(lv_t lv, char const *property) {
45   lvm_property_value_t v;
46
47   v = lvm_lv_get_property(lv, property);
48   if (!v.is_valid || !v.is_string)
49     return NULL;
50   return v.value.string;
51 }
52
53 static void lvm_submit(char const *plugin_instance, char const *type_instance,
54                        uint64_t ivalue) {
55   value_t v;
56   value_list_t vl = VALUE_LIST_INIT;
57
58   v.gauge = (gauge_t)ivalue;
59
60   vl.values = &v;
61   vl.values_len = 1;
62
63   sstrncpy(vl.host, hostname_g, sizeof(vl.host));
64   sstrncpy(vl.plugin, "lvm", sizeof(vl.plugin));
65   sstrncpy(vl.plugin_instance, plugin_instance, sizeof(vl.plugin_instance));
66   sstrncpy(vl.type, "df_complex", sizeof(vl.type));
67   sstrncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
68
69   plugin_dispatch_values(&vl);
70 }
71
72 static void report_lv_utilization(lv_t lv, char const *vg_name,
73                                   char const *lv_name, uint64_t lv_size,
74                                   char const *used_percent_property) {
75   uint64_t used_percent_unscaled;
76   uint64_t used_bytes;
77   char plugin_instance[DATA_MAX_NAME_LEN];
78
79   used_percent_unscaled = get_lv_property_int(lv, used_percent_property);
80   if (used_percent_unscaled == NO_VALUE)
81     return;
82   used_bytes = lv_size * (used_percent_unscaled * PERCENT_SCALE_FACTOR);
83
84   ssnprintf(plugin_instance, sizeof(plugin_instance), "%s-%s", vg_name,
85             lv_name);
86   lvm_submit(plugin_instance, "used", used_bytes);
87   lvm_submit(plugin_instance, "free", lv_size - used_bytes);
88 }
89
90 static void report_thin_pool_utilization(lv_t lv, char const *vg_name,
91                                          uint64_t lv_size) {
92   char const *data_lv;
93   char const *metadata_lv;
94   uint64_t metadata_size;
95
96   data_lv = get_lv_property_string(lv, "data_lv");
97   metadata_lv = get_lv_property_string(lv, "metadata_lv");
98   metadata_size = get_lv_property_int(lv, "lv_metadata_size");
99   if (data_lv == NULL || metadata_lv == NULL || metadata_size == NO_VALUE)
100     return;
101
102   report_lv_utilization(lv, vg_name, data_lv, lv_size, "data_percent");
103   report_lv_utilization(lv, vg_name, metadata_lv, metadata_size,
104                         "metadata_percent");
105 }
106
107 static void vg_read(vg_t vg, char const *vg_name) {
108   struct dm_list *lvs;
109   struct lvm_lv_list *lvl;
110   char const *name;
111   char const *attrs;
112   uint64_t size;
113
114   lvm_submit(vg_name, "free", lvm_vg_get_free_size(vg));
115
116   lvs = lvm_vg_list_lvs(vg);
117   if (!lvs) {
118     /* no VGs are defined, which is not an error per se */
119     return;
120   }
121
122   dm_list_iterate_items(lvl, lvs) {
123     name = lvm_lv_get_name(lvl->lv);
124     attrs = get_lv_property_string(lvl->lv, "lv_attr");
125     size = lvm_lv_get_size(lvl->lv);
126     if (name == NULL || attrs == NULL || size == NO_VALUE)
127       continue;
128
129     /* Condition on volume type.  We want the reported sizes in the
130        volume group to sum to the size of the volume group, so we ignore
131        virtual volumes.  */
132     switch (attrs[0]) {
133     case 's':
134     case 'S':
135       /* Snapshot.  Also report used/free space. */
136       report_lv_utilization(lvl->lv, vg_name, name, size, "data_percent");
137       break;
138     case 't':
139       /* Thin pool virtual volume.  We report the underlying data
140          and metadata volumes, not this one.  Report used/free
141          space, then ignore. */
142       report_thin_pool_utilization(lvl->lv, vg_name, size);
143       continue;
144     case 'v':
145       /* Virtual volume.  Ignore. */
146       continue;
147     case 'V':
148       /* Thin volume or thin snapshot.  Ignore. */
149       continue;
150     }
151     lvm_submit(vg_name, name, size);
152   }
153 }
154
155 static int lvm_read(void) {
156   lvm_t lvm;
157   struct dm_list *vg_names;
158   struct lvm_str_list *name_list;
159
160   lvm = lvm_init(NULL);
161   if (!lvm) {
162     ERROR("lvm plugin: lvm_init failed.");
163     return (-1);
164   }
165
166   vg_names = lvm_list_vg_names(lvm);
167   if (!vg_names) {
168     ERROR("lvm plugin lvm_list_vg_name failed %s", lvm_errmsg(lvm));
169     lvm_quit(lvm);
170     return (-1);
171   }
172
173   dm_list_iterate_items(name_list, vg_names) {
174     vg_t vg;
175
176     vg = lvm_vg_open(lvm, name_list->str, "r", 0);
177     if (!vg) {
178       ERROR("lvm plugin: lvm_vg_open (%s) failed: %s", name_list->str,
179             lvm_errmsg(lvm));
180       continue;
181     }
182
183     vg_read(vg, name_list->str);
184     lvm_vg_close(vg);
185   }
186
187   lvm_quit(lvm);
188   return (0);
189 } /*lvm_read */
190
191 void module_register(void) {
192   plugin_register_read("lvm", lvm_read);
193 } /* void module_register */