Fix btt to handle large numbers of output files
[blktrace.git] / btt / devs.c
1 /*
2  * blktrace output analysis: generate a timeline & gather statistics
3  *
4  * Copyright (C) 2006 Alan D. Brunelle <Alan.Brunelle@hp.com>
5  *
6  *  This program is free software; you can redistribute it and/or modify
7  *  it under the terms of the GNU General Public License as published by
8  *  the Free Software Foundation; either version 2 of the License, or
9  *  (at your option) any later version.
10  *
11  *  This program is distributed in the hope that it will be useful,
12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  *  GNU General Public License for more details.
15  *
16  *  You should have received a copy of the GNU General Public License
17  *  along with this program; if not, write to the Free Software
18  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  */
21 #include <stdio.h>
22 #include "globals.h"
23
24 #define N_DEV_HASH      128
25 #define DEV_HASH(dev)   ((MAJOR(dev) ^ MINOR(dev)) & (N_DEV_HASH - 1))
26 struct list_head        dev_heads[N_DEV_HASH];
27
28 static inline void *dip_rb_mkhds(void)
29 {
30         size_t len = N_IOP_TYPES * sizeof(struct rb_root);
31         return memset(malloc(len), 0, len);
32 }
33
34 static void __destroy(struct rb_node *n)
35 {
36         if (n) {
37                 struct io *iop = rb_entry(n, struct io, rb_node);
38
39                 __destroy(n->rb_left);
40                 __destroy(n->rb_right);
41                 io_release(iop);
42         }
43 }
44
45 static void __destroy_heads(struct rb_root *roots)
46 {
47         int i;
48
49         for (i = 0; i < N_IOP_TYPES; i++)
50                 __destroy(roots[i].rb_node);
51
52         free(roots);
53 }
54
55 void init_dev_heads(void)
56 {
57         int i;
58         for (i = 0; i < N_DEV_HASH; i++)
59                 INIT_LIST_HEAD(&dev_heads[i]);
60 }
61
62 struct d_info *__dip_find(__u32 device)
63 {
64         struct d_info *dip;
65         struct list_head *p;
66
67         __list_for_each(p, &dev_heads[DEV_HASH(device)]) {
68                 dip = list_entry(p, struct d_info, hash_head);
69                 if (device == dip->device)
70                         return dip;
71         }
72
73         return NULL;
74 }
75
76 void dip_exit(void)
77 {
78         struct d_info *dip;
79         struct list_head *p, *q;
80
81         list_for_each_safe(p, q, &all_devs) {
82                 dip = list_entry(p, struct d_info, all_head);
83
84                 __destroy_heads(dip->heads);
85                 region_exit(&dip->regions);
86                 seeki_exit(dip->seek_handle);
87                 seeki_exit(dip->q2q_handle);
88                 aqd_exit(dip->aqd_handle);
89                 plat_exit(dip->q2d_plat_handle);
90                 plat_exit(dip->q2c_plat_handle);
91                 plat_exit(dip->d2c_plat_handle);
92                 bno_dump_exit(dip->bno_dump_handle);
93                 unplug_hist_exit(dip->unplug_hist_handle);
94                 if (output_all_data)
95                         q2d_release(dip->q2d_priv);
96                 if (dip->pit_fp)
97                         fclose(dip->pit_fp);
98                 free(dip);
99         }
100 }
101
102 static inline char *mkhandle(char *str, __u32 device, char *post)
103 {
104         int mjr = device >> MINORBITS;
105         int mnr = device & ((1 << MINORBITS) - 1);
106
107         sprintf(str, "%03d,%03d%s", mjr, mnr, post);
108         return str;
109 }
110
111 static inline FILE *open_pit(char *str)
112 {
113         FILE *fp = my_fopen(str, "w");
114
115         if (fp == NULL)
116                 perror(str);
117
118         return fp;
119 }
120
121 struct d_info *dip_add(__u32 device, struct io *iop)
122 {
123         struct d_info *dip = __dip_find(device);
124
125         if (dip == NULL) {
126                 char str[256];
127
128                 dip = malloc(sizeof(struct d_info));
129                 memset(dip, 0, sizeof(*dip));
130                 dip->heads = dip_rb_mkhds();
131                 region_init(&dip->regions);
132                 dip->device = device;
133                 dip->last_q = (__u64)-1;
134                 dip->map = dev_map_find(device);
135                 dip->bno_dump_handle = bno_dump_init(device);
136                 dip->unplug_hist_handle = unplug_hist_init(device);
137                 dip->seek_handle = seeki_init(mkhandle(str, device, "_d2d"));
138                 dip->q2q_handle = seeki_init(mkhandle(str, device, "_q2q"));
139                 dip->aqd_handle = aqd_init(mkhandle(str, device, "_aqd"));
140                 dip->q2d_plat_handle =
141                                 plat_init(mkhandle(str, device, "_q2d_plat"));
142                 dip->q2c_plat_handle =
143                                 plat_init(mkhandle(str, device, "_q2c_plat"));
144                 dip->d2c_plat_handle =
145                                 plat_init(mkhandle(str, device, "_d2c_plat"));
146                 latency_init(dip);
147                 list_add_tail(&dip->hash_head, &dev_heads[DEV_HASH(device)]);
148                 list_add_tail(&dip->all_head, &all_devs);
149                 dip->start_time = BIT_TIME(iop->t.time);
150                 dip->pre_culling = 1;
151                 if (output_all_data)
152                         dip->q2d_priv = q2d_init();
153                 n_devs++;
154                 if (per_io_trees)
155                         dip->pit_fp = open_pit(mkhandle(per_io_trees,
156                                                           device, "_pit.dat"));
157         }
158
159         if (dip->pre_culling) {
160                 if (iop->type == IOP_Q || iop->type == IOP_A)
161                         dip->pre_culling = 0;
162                 else
163                         return NULL;
164         }
165
166         iop->linked = dip_rb_ins(dip, iop);
167         dip->end_time = BIT_TIME(iop->t.time);
168
169         return dip;
170 }
171
172 void dip_rem(struct io *iop)
173 {
174         if (iop->linked) {
175                 dip_rb_rem(iop);
176                 iop->linked = 0;
177         }
178 }
179
180 void dip_foreach(struct io *iop, enum iop_type type,
181                  void (*fnc)(struct io *iop, struct io *this), int rm_after)
182 {
183         if (rm_after) {
184                 LIST_HEAD(head);
185                 struct io *this;
186                 struct list_head *p, *q;
187
188                 dip_rb_fe(iop->dip, type, iop, fnc, &head);
189                 list_for_each_safe(p, q, &head) {
190                         this = list_entry(p, struct io, f_head);
191                         list_del(&this->f_head);
192                         io_release(this);
193                 }
194         }
195         else
196                 dip_rb_fe(iop->dip, type, iop, fnc, NULL);
197 }
198
199 void dip_foreach_list(struct io *iop, enum iop_type type, struct list_head *hd)
200 {
201         dip_rb_fe(iop->dip, type, iop, NULL, hd);
202 }
203
204 struct io *dip_find_sec(struct d_info *dip, enum iop_type type, __u64 sec)
205 {
206         return dip_rb_find_sec(dip, type, sec);
207 }
208
209 void dip_foreach_out(void (*func)(struct d_info *, void *), void *arg)
210 {
211         if (devices == NULL) {
212                 struct list_head *p;
213                 __list_for_each(p, &all_devs)
214                         func(list_entry(p, struct d_info, all_head), arg);
215         }
216         else {
217                 int i;
218                 struct d_info *dip;
219                 unsigned int mjr, mnr;
220                 char *p = devices;
221
222                 while (p && ((i = sscanf(p, "%u,%u", &mjr, &mnr)) == 2)) {
223                         dip = __dip_find((__u32)((mjr << MINORBITS) | mnr));
224                         func(dip, arg);
225                         p = strchr(p, ';');
226                         if (p) p++;
227                 }
228         }
229 }
230
231 void dip_plug(__u32 dev, double cur_time)
232 {
233         struct d_info *dip = __dip_find(dev);
234
235         if (!dip || dip->is_plugged) return;
236
237         dip->is_plugged = 1;
238         dip->last_plug = cur_time;
239 }
240
241 void dip_unplug(__u32 dev, double cur_time, __u64 nios_up)
242 {
243         struct d_info *dip = __dip_find(dev);
244
245         if (dip && dip->is_plugged) {
246                 dip->nplugs++;
247                 dip->plugged_time += (cur_time - dip->last_plug);
248                 dip->is_plugged = 0;
249                 dip->nios_up += nios_up;
250         }
251 }
252
253 void dip_unplug_tm(__u32 dev, __u64 nios_up)
254 {
255         struct d_info *dip = __dip_find(dev);
256
257         if (dip && dip->is_plugged) {
258                 dip->n_timer_unplugs++;
259                 dip->nios_upt += nios_up;
260                 dip->nplugs_t++;
261         }
262 }