r23780: Find and fix more GPL2 -> GPL3.
[bbaumbach/samba-autobuild/.git] / examples / perfcounter / perf_writer_disk.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  Performance Counter Daemon
4  *
5  *  Copyright (C) Marcin Krzysztof Porwit    2005
6  *  
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 3 of the License, or
10  *  (at your option) any later version.
11  *  
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *  
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, write to the Free Software
19  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20  */
21
22 #include "perf.h"
23
24 void init_diskdata_desc(PERF_DATA_BLOCK *data)
25 {
26         init_perf_counter(&(data->diskInfo.diskObjDesc),
27                           &(data->diskInfo.diskObjDesc),
28                           get_counter_id(data),
29                           "Logical Disk",
30                           "The Logical Disk object consists of counters that show information about disks.",
31                           0,
32                           PERF_OBJECT);
33         init_perf_counter(&(data->diskInfo.freeMegs),
34                           &(data->diskInfo.diskObjDesc),
35                           get_counter_id(data),
36                           "Megabytes Free",
37                           "The amount of available disk space, in megabytes.",
38                           PERF_SIZE_LARGE | PERF_TYPE_NUMBER | PERF_NUMBER_DECIMAL | PERF_DISPLAY_NO_SUFFIX,
39                           PERF_COUNTER);
40         init_perf_counter(&(data->diskInfo.writesPerSec),
41                           &(data->diskInfo.diskObjDesc),
42                           get_counter_id(data),
43                           "Writes/sec",
44                           "The number of writes per second to that disk.",
45                           PERF_SIZE_DWORD | PERF_TYPE_COUNTER | PERF_COUNTER_RATE | PERF_DELTA_COUNTER | PERF_DISPLAY_PER_SEC,
46                           PERF_COUNTER);
47         init_perf_counter(&(data->diskInfo.readsPerSec),
48                           &(data->diskInfo.diskObjDesc),
49                           get_counter_id(data),
50                           "Reads/sec",
51                           "The number of reads of that disk per second.",
52                           PERF_SIZE_DWORD | PERF_TYPE_COUNTER | PERF_COUNTER_RATE | PERF_DELTA_COUNTER | PERF_DISPLAY_PER_SEC,
53                           PERF_COUNTER);
54
55         return;
56 }
57 void init_num_disks(PERF_DATA_BLOCK *data)
58 {
59         FILE *mtab;
60         char buf[PROC_BUF];
61         char *start, *stop;
62         int i = 0, num;
63
64         if(!(mtab = fopen("/etc/mtab", "r")))
65         {
66                 perror("init_disk_names: fopen");
67                 exit(1);
68         }
69
70         rewind(mtab);
71         fflush(mtab);
72
73         while(fgets(buf, sizeof(buf), mtab))
74         {
75                 if(start = strstr(buf, "/dev/"))
76                 {
77                         if(start = strstr(start, "da"))
78                         {
79                                 i++;
80                         }
81                 }
82         }
83
84         data->diskInfo.numDisks = i;
85         fclose(mtab);
86
87         return;
88 }
89         
90 void init_disk_names(PERF_DATA_BLOCK *data)
91 {
92         FILE *mtab;
93         char buf[PROC_BUF];
94         char *start, *stop;
95         int i = 0, num;
96
97         if(!(mtab = fopen("/etc/mtab", "r")))
98         {
99                 perror("init_disk_names: fopen");
100                 exit(1);
101         }
102
103         rewind(mtab);
104         fflush(mtab);
105
106         while(fgets(buf, sizeof(buf), mtab))
107         {
108                 if(start = strstr(buf, "/dev/"))
109                 {
110                         if(start = strstr(start, "da"))
111                         {
112                                 start -=1;
113                                 stop = strstr(start, " ");
114                                 memcpy(data->diskInfo.mdata[i].name, start, stop - start);
115                                 start = stop +1;
116                                 stop = strstr(start, " ");
117                                 memcpy(data->diskInfo.mdata[i].mountpoint, start, stop - start);
118                                 i++;
119                         }
120                 }
121         }
122
123         fclose(mtab);
124
125         return;
126 }
127
128 void get_diskinfo(PERF_DATA_BLOCK *data)
129 {
130         int i;
131         DiskData *p;
132         struct statfs statfsbuf;
133         int status, num;
134         char buf[LARGE_BUF], *start;
135         FILE *diskstats;
136         long reads, writes, discard;
137
138         diskstats = fopen("/proc/diskstats", "r");
139         rewind(diskstats);
140         fflush(diskstats);
141         status = fread(buf, sizeof(char), LARGE_BUF, diskstats);
142         fclose(diskstats);
143
144         for(i = 0; i < data->diskInfo.numDisks; i++)
145         {
146                 p = &(data->diskInfo.data[i]);
147                 status = statfs(data->diskInfo.mdata[i].mountpoint, &statfsbuf);
148                 p->freeMegs = (statfsbuf.f_bfree*statfsbuf.f_bsize)/1048576;
149                 start = strstr(buf, data->diskInfo.mdata[i].name);
150                 start += strlen(data->diskInfo.mdata[i].name) + 1;
151                 num = sscanf(start, "%u %u %u %u",
152                              &reads,
153                              &discard, 
154                              &writes, 
155                              &discard);
156                 p->writesPerSec = writes;
157                 p->readsPerSec = reads;
158                 fprintf(stderr, "%s:\t%u\t%u\n",
159                         data->diskInfo.mdata[i].mountpoint,
160                         reads, writes);
161         }
162         return;
163 }
164 void init_disk_data(PERF_DATA_BLOCK *data)
165 {
166         init_diskdata_desc(data);
167         
168         init_num_disks(data);
169
170         data->diskInfo.mdata = calloc(data->diskInfo.numDisks, sizeof(DiskMetaData));
171         if(!data->diskInfo.mdata)
172         {
173                 fatal("init_disk_data: out of memory");
174         }
175
176         init_disk_names(data);
177
178         data->diskInfo.data = calloc(data->diskInfo.numDisks, sizeof(DiskData));
179         if(!data->diskInfo.data)
180         {
181                 fatal("init_disk_data: out of memory");
182         }
183
184         get_diskinfo(data);
185
186         return;
187 }
188
189 void output_disk_desc(PERF_DATA_BLOCK *data, RuntimeSettings rt)
190 {
191         output_perf_desc(data->diskInfo.diskObjDesc, rt);
192         output_perf_desc(data->diskInfo.freeMegs, rt);
193         output_perf_desc(data->diskInfo.writesPerSec, rt);
194         output_perf_desc(data->diskInfo.readsPerSec, rt);
195         output_num_instances(data->diskInfo.diskObjDesc, data->diskInfo.numDisks, rt);
196
197         return;
198 }
199
200 void output_diskinfo(PERF_DATA_BLOCK *data, RuntimeSettings rt, int tdb_flags)
201 {
202         int i;
203         
204         output_perf_counter(data->diskInfo.freeMegs,
205                             data->diskInfo.data[0].freeMegs,
206                             rt, tdb_flags);
207         output_perf_counter(data->diskInfo.writesPerSec,
208                             (unsigned long long)data->diskInfo.data[0].writesPerSec,
209                             rt, tdb_flags);
210         output_perf_counter(data->diskInfo.readsPerSec,
211                             (unsigned long long)data->diskInfo.data[0].readsPerSec,
212                             rt, tdb_flags);
213
214         for(i = 0; i < data->diskInfo.numDisks; i++)
215         {
216                 output_perf_instance(data->diskInfo.diskObjDesc.index,
217                                      i,
218                                      (void *)&(data->diskInfo.data[i]),
219                                      sizeof(DiskData),
220                                      data->diskInfo.mdata[i].mountpoint,
221                                      rt, tdb_flags);
222         }
223
224         return;
225 }