Merge branch 'pr/1649'
[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 {
39         kstat_t *ksp_chain;
40
41         numtape = 0;
42
43         if (kc == NULL)
44                 return (-1);
45
46         for (numtape = 0, ksp_chain = kc->kc_chain;
47                         (numtape < MAX_NUMTAPE) && (ksp_chain != NULL);
48                         ksp_chain = ksp_chain->ks_next)
49         {
50                 if (strncmp (ksp_chain->ks_class, "tape", 4) )
51                         continue;
52                 if (ksp_chain->ks_type != KSTAT_TYPE_IO)
53                         continue;
54                 ksp[numtape++] = ksp_chain;
55         }
56
57         return (0);
58 } /* int tape_init */
59
60 static void tape_submit (const char *plugin_instance,
61                 const char *type,
62                 derive_t read, derive_t write)
63 {
64         value_list_t vl = VALUE_LIST_INIT;
65         value_t values[] = {
66                 { .derive = read },
67                 { .derive = write },
68         };
69
70         vl.values = values;
71         vl.values_len = STATIC_ARRAY_SIZE (values);
72         sstrncpy (vl.host, hostname_g, sizeof (vl.host));
73         sstrncpy (vl.plugin, "tape", sizeof (vl.plugin));
74         sstrncpy (vl.plugin_instance, plugin_instance,
75                         sizeof (vl.plugin_instance));
76         sstrncpy (vl.type, type, sizeof (vl.type));
77
78         plugin_dispatch_values (&vl);
79 } /* void tape_submit */
80
81 static int tape_read (void)
82 {
83
84 #if HAVE_KSTAT_IO_T_WRITES && HAVE_KSTAT_IO_T_NWRITES && HAVE_KSTAT_IO_T_WTIME
85 # define KIO_ROCTETS reads
86 # define KIO_WOCTETS writes
87 # define KIO_ROPS    nreads
88 # define KIO_WOPS    nwrites
89 # define KIO_RTIME   rtime
90 # define KIO_WTIME   wtime
91 #elif HAVE_KSTAT_IO_T_NWRITTEN && HAVE_KSTAT_IO_T_WRITES && HAVE_KSTAT_IO_T_WTIME
92 # define KIO_ROCTETS nread
93 # define KIO_WOCTETS nwritten
94 # define KIO_ROPS    reads
95 # define KIO_WOPS    writes
96 # define KIO_RTIME   rtime
97 # define KIO_WTIME   wtime
98 #else
99 # error "kstat_io_t does not have the required members"
100 #endif
101         static kstat_io_t kio;
102
103         if (kc == NULL)
104                 return (-1);
105
106         if (numtape <= 0)
107                 return (-1);
108
109         for (int i = 0; i < numtape; i++)
110         {
111                 if (kstat_read (kc, ksp[i], &kio) == -1)
112                         continue;
113
114                 if (strncmp (ksp[i]->ks_class, "tape", 4) == 0)
115                 {
116                         tape_submit (ksp[i]->ks_name, "tape_octets",
117                                         kio.KIO_ROCTETS, kio.KIO_WOCTETS);
118                         tape_submit (ksp[i]->ks_name, "tape_ops",
119                                         kio.KIO_ROPS, kio.KIO_WOPS);
120                         /* FIXME: Convert this to microseconds if necessary */
121                         tape_submit (ksp[i]->ks_name, "tape_time",
122                                         kio.KIO_RTIME, kio.KIO_WTIME);
123                 }
124         }
125
126         return (0);
127 }
128
129 void module_register (void)
130 {
131         plugin_register_init ("tape", tape_init);
132         plugin_register_read ("tape", tape_read);
133 }