btt: fix missing cleanup call
[blktrace.git] / btt / bt_timeline.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 <string.h>
23 #include <unistd.h>
24 #include <sys/time.h>
25 #include <time.h>
26 #include "globals.h"
27
28 char bt_timeline_version[] = "2.02";
29
30 char *devices, *exes, *input_name, *output_name, *seek_name, *bno_dump_name;
31 char *d2c_name, *q2c_name, *per_io_name, *unplug_hist_name;
32 FILE *ranges_ofp, *avgs_ofp, *per_io_ofp;
33 int verbose, done, time_bounded, output_all_data, seek_absolute;
34 double t_astart, t_aend;
35 unsigned long n_traces;
36 struct avgs_info all_avgs;
37 unsigned int n_devs;
38 time_t genesis, last_vtrace;
39 LIST_HEAD(all_devs);
40 LIST_HEAD(all_procs);
41 LIST_HEAD(free_ios);
42 LIST_HEAD(free_bilinks);
43 __u64 q_histo[N_HIST_BKTS], d_histo[N_HIST_BKTS];
44
45 double range_delta = 0.1;
46 __u64 last_q = (__u64)-1;
47
48 struct region_info all_regions = {
49         .qranges = LIST_HEAD_INIT(all_regions.qranges),
50         .cranges = LIST_HEAD_INIT(all_regions.cranges),
51 };
52
53 #if defined(DEBUG)
54         int rb_tree_size;
55 #endif
56
57 #if defined(COUNT_IOS)
58 unsigned long nios_reused, nios_alloced, nios_freed;
59 LIST_HEAD(cios);
60 #endif
61
62 int process(void);
63
64 int main(int argc, char *argv[])
65 {
66         handle_args(argc, argv);
67
68         init_dev_heads();
69         iostat_init();
70         if (process() || output_avgs(avgs_ofp) || output_ranges(ranges_ofp))
71                 return 1;
72
73         if (iostat_ofp) {
74                 fprintf(iostat_ofp, "\n");
75                 iostat_dump_stats(iostat_last_stamp, 1);
76         }
77
78         if (ranges_ofp != stdout) 
79                 fclose(ranges_ofp);
80         if (avgs_ofp != stdout) 
81                 fclose(avgs_ofp);
82
83         seek_clean();
84         latency_clean();
85         bno_dump_clean();
86         dev_map_exit();
87         dip_exit();
88         pip_exit();
89         io_free_all();
90         region_exit(&all_regions);
91
92         free(input_name);
93         if (output_name) free(output_name);
94
95         clean_args();
96         clean_bufs();
97
98         return 0;
99 }
100
101 static inline double tv2dbl(struct timeval *tv)
102 {
103         return (double)tv->tv_sec + (((double)tv->tv_usec) / (1000.0 * 1000.0));
104 }
105
106 int process(void)
107 {
108         int ret = 0;
109         struct io *iop = io_alloc();
110         struct timeval tvs, tve;
111
112         genesis = last_vtrace = time(NULL);
113         gettimeofday(&tvs, NULL);
114         while (!done && next_trace(&iop->t, &iop->pdu)) {
115                 add_trace(iop);
116                 iop = io_alloc();
117         }
118
119         io_release(iop);
120         gettimeofday(&tve, NULL);
121
122         if (verbose) {
123                 double tps, dt_input = tv2dbl(&tve) - tv2dbl(&tvs);
124                 
125                 tps = (double)n_traces / dt_input;
126                 printf("\r                                        "
127                        "                                        \r");
128                 printf("%10lu traces @ %.1lf Ktps in %.6lf seconds\n",
129                         n_traces, tps/1000.0,
130                         dt_input);
131
132 #               if defined(DEBUG)
133                         printf("\ttree = |%d|\n", rb_tree_size);
134                         if (rb_tree_size > 0)
135                                 dump_rb_trees();
136 #               endif
137
138 #               if defined(COUNT_IOS)
139                         {
140                                 struct io *_iop;
141                                 struct list_head *_p;
142                                 FILE *_ofp = fopen("cios.txt", "w");
143                                 printf("(%ld + %ld) = %ld - %ld = %ld\n", 
144                                         nios_alloced, nios_reused, 
145                                         nios_alloced + nios_reused, 
146                                         nios_freed, 
147                                         (nios_alloced + nios_reused) 
148                                                                 - nios_freed);
149
150                                 __list_for_each(_p, &cios) {
151                                         _iop = list_entry(_p, struct io, 
152                                                                 cio_head);
153                                         __dump_iop(_ofp, _iop, 0);
154                                 }
155                                 fclose(_ofp);
156                         }
157 #               endif
158         }
159
160         return ret;
161 }