Merge branch 'collectd-5.7'
[collectd.git] / src / tape.c
1 /**
2  * collectd - src/tape.c
3  * Copyright (C) 2005,2006  Scott Garrett
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; either version 2 of the License, or (at your
8  * option) any later version.
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  *   Scott Garrett <sgarrett at technomancer.com>
21  **/
22
23 #include "collectd.h"
24
25 #include "common.h"
26 #include "plugin.h"
27
28 #if !HAVE_LIBKSTAT
29 #error "No applicable input method."
30 #endif
31
32 #define MAX_NUMTAPE 256
33 extern kstat_ctl_t *kc;
34 static kstat_t *ksp[MAX_NUMTAPE];
35 static int numtape = 0;
36
37 static int tape_init(void) {
38   kstat_t *ksp_chain;
39
40   numtape = 0;
41
42   if (kc == NULL)
43     return (-1);
44
45   for (numtape = 0, ksp_chain = kc->kc_chain;
46        (numtape < MAX_NUMTAPE) && (ksp_chain != NULL);
47        ksp_chain = ksp_chain->ks_next) {
48     if (strncmp(ksp_chain->ks_class, "tape", 4))
49       continue;
50     if (ksp_chain->ks_type != KSTAT_TYPE_IO)
51       continue;
52     ksp[numtape++] = ksp_chain;
53   }
54
55   return (0);
56 } /* int tape_init */
57
58 static void tape_submit(const char *plugin_instance, const char *type,
59                         derive_t read, derive_t write) {
60   value_list_t vl = VALUE_LIST_INIT;
61   value_t values[] = {
62       {.derive = read}, {.derive = write},
63   };
64
65   vl.values = values;
66   vl.values_len = STATIC_ARRAY_SIZE(values);
67   sstrncpy(vl.plugin, "tape", sizeof(vl.plugin));
68   sstrncpy(vl.plugin_instance, plugin_instance, sizeof(vl.plugin_instance));
69   sstrncpy(vl.type, type, sizeof(vl.type));
70
71   plugin_dispatch_values(&vl);
72 } /* void tape_submit */
73
74 static int tape_read(void) {
75
76 #if HAVE_KSTAT_IO_T_WRITES && HAVE_KSTAT_IO_T_NWRITES && HAVE_KSTAT_IO_T_WTIME
77 #define KIO_ROCTETS reads
78 #define KIO_WOCTETS writes
79 #define KIO_ROPS nreads
80 #define KIO_WOPS nwrites
81 #define KIO_RTIME rtime
82 #define KIO_WTIME wtime
83 #elif HAVE_KSTAT_IO_T_NWRITTEN && HAVE_KSTAT_IO_T_WRITES &&                    \
84     HAVE_KSTAT_IO_T_WTIME
85 #define KIO_ROCTETS nread
86 #define KIO_WOCTETS nwritten
87 #define KIO_ROPS reads
88 #define KIO_WOPS writes
89 #define KIO_RTIME rtime
90 #define KIO_WTIME wtime
91 #else
92 #error "kstat_io_t does not have the required members"
93 #endif
94   static kstat_io_t kio;
95
96   if (kc == NULL)
97     return (-1);
98
99   if (numtape <= 0)
100     return (-1);
101
102   for (int i = 0; i < numtape; i++) {
103     if (kstat_read(kc, ksp[i], &kio) == -1)
104       continue;
105
106     if (strncmp(ksp[i]->ks_class, "tape", 4) == 0) {
107       tape_submit(ksp[i]->ks_name, "tape_octets", kio.KIO_ROCTETS,
108                   kio.KIO_WOCTETS);
109       tape_submit(ksp[i]->ks_name, "tape_ops", kio.KIO_ROPS, kio.KIO_WOPS);
110       /* FIXME: Convert this to microseconds if necessary */
111       tape_submit(ksp[i]->ks_name, "tape_time", kio.KIO_RTIME, kio.KIO_WTIME);
112     }
113   }
114
115   return (0);
116 }
117
118 void module_register(void) {
119   plugin_register_init("tape", tape_init);
120   plugin_register_read("tape", tape_read);
121 }