[PATCH] kernel: Update to 2.6.14-rc1-git
[blktrace.git] / blkparse.c
... / ...
CommitLineData
1/*
2 * block queue tracing parse application
3 *
4 * Copyright (C) 2005 Jens Axboe <axboe@suse.de>
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 <sys/types.h>
22#include <sys/stat.h>
23#include <unistd.h>
24#include <stdio.h>
25#include <fcntl.h>
26#include <stdlib.h>
27#include <string.h>
28#include <getopt.h>
29#include <errno.h>
30#include <signal.h>
31#include <locale.h>
32#include <limits.h>
33
34#include "blktrace.h"
35#include "rbtree.h"
36
37#define SECONDS(x) ((unsigned long long)(x) / 1000000000)
38#define NANO_SECONDS(x) ((unsigned long long)(x) % 1000000000)
39#define DOUBLE_TO_NANO_ULL(d) ((unsigned long long)((d) * 1000000000))
40
41#define MINORBITS 20
42#define MINORMASK ((1U << MINORBITS) - 1)
43#define MAJOR(dev) ((unsigned int) ((dev) >> MINORBITS))
44#define MINOR(dev) ((unsigned int) ((dev) & MINORMASK))
45
46#define min(a, b) ((a) < (b) ? (a) : (b))
47
48struct io_stats {
49 unsigned long qreads, qwrites, creads, cwrites, mreads, mwrites;
50 unsigned long ireads, iwrites;
51 unsigned long long qread_kb, qwrite_kb, cread_kb, cwrite_kb;
52 unsigned long long iread_kb, iwrite_kb;
53};
54
55struct per_cpu_info {
56 int cpu;
57 int nelems;
58
59 int fd;
60 char fname[128];
61
62 struct io_stats io_stats;
63};
64
65struct per_dev_info {
66 dev_t id;
67 char *name;
68
69 int backwards;
70 unsigned long long events;
71 unsigned long long last_reported_time;
72 struct io_stats io_stats;
73
74 int ncpus;
75 struct per_cpu_info *cpus;
76};
77
78struct per_process_info {
79 char name[16];
80 __u32 pid;
81 struct io_stats io_stats;
82 struct per_process_info *hash_next, *list_next;
83
84 /*
85 * individual io stats
86 */
87 unsigned long long longest_allocation_wait[2];
88 unsigned long long longest_dispatch_wait[2];
89 unsigned long long longest_completion_wait[2];
90};
91
92#define PPI_HASH_SHIFT (8)
93static struct per_process_info *ppi_hash[1 << PPI_HASH_SHIFT];
94static struct per_process_info *ppi_list;
95
96#define S_OPTS "i:o:b:stqw:"
97static struct option l_opts[] = {
98 {
99 .name = "input",
100 .has_arg = 1,
101 .flag = NULL,
102 .val = 'i'
103 },
104 {
105 .name = "output",
106 .has_arg = 1,
107 .flag = NULL,
108 .val = 'o'
109 },
110 {
111 .name = "batch",
112 .has_arg = 1,
113 .flag = NULL,
114 .val = 'b'
115 },
116 {
117 .name = "per program stats",
118 .has_arg = 0,
119 .flag = NULL,
120 .val = 's'
121 },
122 {
123 .name = "track ios",
124 .has_arg = 0,
125 .flag = NULL,
126 .val = 't'
127 },
128 {
129 .name = "quiet",
130 .has_arg = 0,
131 .flag = NULL,
132 .val = 'q'
133 },
134 {
135 .name = "stopwatch",
136 .has_arg = 1,
137 .flag = NULL,
138 .val = 'w'
139 },
140 {
141 .name = NULL,
142 .has_arg = 0,
143 .flag = NULL,
144 .val = 0
145 }
146};
147
148static struct rb_root rb_sort_root;
149static struct rb_root rb_track_root;
150
151/*
152 * for sorting the displayed output
153 */
154struct trace {
155 struct blk_io_trace *bit;
156 struct rb_node rb_node;
157};
158
159/*
160 * for tracking individual ios
161 */
162struct io_track {
163 struct rb_node rb_node;
164
165 dev_t device;
166 __u64 sector;
167 __u32 pid;
168 unsigned long long allocation_time;
169 unsigned long long queue_time;
170 unsigned long long dispatch_time;
171 unsigned long long completion_time;
172};
173
174static int ndevices;
175static struct per_dev_info *devices;
176static char *get_dev_name(struct per_dev_info *, char *, int);
177
178static FILE *ofp;
179static char *output_name;
180
181static unsigned long long genesis_time;
182static unsigned long long stopwatch_start; /* start from zero by default */
183static unsigned long long stopwatch_end = ULONG_LONG_MAX; /* "infinity" */
184
185static int per_process_stats;
186static int track_ios;
187
188#define RB_BATCH_DEFAULT (1024)
189static int rb_batch = RB_BATCH_DEFAULT;
190
191static int pipeline;
192
193#define is_done() (*(volatile int *)(&done))
194static volatile int done;
195
196static inline unsigned long hash_long(unsigned long val)
197{
198#if __WORDSIZE == 32
199 val *= 0x9e370001UL;
200#elif __WORDSIZE == 64
201 val *= 0x9e37fffffffc0001UL;
202#else
203#error unknown word size
204#endif
205
206 return val >> (__WORDSIZE - PPI_HASH_SHIFT);
207}
208
209static inline void add_process_to_hash(struct per_process_info *ppi)
210{
211 const int hash_idx = hash_long(ppi->pid);
212
213 ppi->hash_next = ppi_hash[hash_idx];
214 ppi_hash[hash_idx] = ppi;
215}
216
217static inline void add_process_to_list(struct per_process_info *ppi)
218{
219 ppi->list_next = ppi_list;
220 ppi_list = ppi;
221}
222
223static struct per_process_info *find_process_by_pid(__u32 pid)
224{
225 const int hash_idx = hash_long(pid);
226 struct per_process_info *ppi;
227
228 ppi = ppi_hash[hash_idx];
229 while (ppi) {
230 if (ppi->pid == pid)
231 return ppi;
232
233 ppi = ppi->hash_next;
234 }
235
236 return NULL;
237}
238
239static inline int trace_rb_insert(struct trace *t)
240{
241 struct rb_node **p = &rb_sort_root.rb_node;
242 struct rb_node *parent = NULL;
243 struct trace *__t;
244
245 if (genesis_time == 0 || t->bit->time < genesis_time)
246 genesis_time = t->bit->time;
247
248 while (*p) {
249 parent = *p;
250 __t = rb_entry(parent, struct trace, rb_node);
251
252 if (t->bit->time < __t->bit->time)
253 p = &(*p)->rb_left;
254 else if (t->bit->time > __t->bit->time)
255 p = &(*p)->rb_right;
256 else if (t->bit->device < __t->bit->device)
257 p = &(*p)->rb_left;
258 else if (t->bit->device > __t->bit->device)
259 p = &(*p)->rb_right;
260 else if (t->bit->sequence < __t->bit->sequence)
261 p = &(*p)->rb_left;
262 else if (t->bit->sequence > __t->bit->sequence)
263 p = &(*p)->rb_right;
264 else if (t->bit->device == __t->bit->device) {
265 fprintf(stderr,
266 "sequence alias (%d) on device %d,%d!\n",
267 t->bit->sequence,
268 MAJOR(t->bit->device), MINOR(t->bit->device));
269 return 1;
270 }
271 }
272
273 rb_link_node(&t->rb_node, parent, p);
274 rb_insert_color(&t->rb_node, &rb_sort_root);
275 return 0;
276}
277
278static inline int track_rb_insert(struct io_track *iot)
279{
280 struct rb_node **p = &rb_track_root.rb_node;
281 struct rb_node *parent = NULL;
282 struct io_track *__iot;
283
284 while (*p) {
285 parent = *p;
286
287 __iot = rb_entry(parent, struct io_track, rb_node);
288
289 if (iot->device < __iot->device)
290 p = &(*p)->rb_left;
291 else if (iot->device > __iot->device)
292 p = &(*p)->rb_right;
293 else if (iot->sector < __iot->sector)
294 p = &(*p)->rb_left;
295 else if (iot->sector > __iot->sector)
296 p = &(*p)->rb_right;
297 else {
298 fprintf(stderr,
299 "sector alias (%llu) on device %d,%d!\n",
300 iot->sector,
301 MAJOR(iot->device), MINOR(iot->device));
302 return 1;
303 }
304 }
305
306 rb_link_node(&iot->rb_node, parent, p);
307 rb_insert_color(&iot->rb_node, &rb_track_root);
308 return 0;
309}
310
311static struct io_track *__find_track(dev_t device, __u64 sector)
312{
313 struct rb_node **p = &rb_track_root.rb_node;
314 struct rb_node *parent = NULL;
315 struct io_track *__iot;
316
317 while (*p) {
318 parent = *p;
319
320 __iot = rb_entry(parent, struct io_track, rb_node);
321
322 if (device < __iot->device)
323 p = &(*p)->rb_left;
324 else if (device > __iot->device)
325 p = &(*p)->rb_right;
326 else if (sector < __iot->sector)
327 p = &(*p)->rb_left;
328 else if (sector > __iot->sector)
329 p = &(*p)->rb_right;
330 else
331 return __iot;
332 }
333
334 return NULL;
335}
336
337static struct io_track *find_track(__u32 pid, dev_t device, __u64 sector)
338{
339 struct io_track *iot;
340
341 iot = __find_track(device, sector);
342 if (!iot) {
343 iot = malloc(sizeof(*iot));
344 iot->pid = pid;
345 iot->device = device;
346 iot->sector = sector;
347 track_rb_insert(iot);
348 }
349
350 return iot;
351}
352
353static void log_track_merge(struct blk_io_trace *t)
354{
355 struct io_track *iot;
356
357 if (!track_ios)
358 return;
359 if ((t->action & BLK_TC_ACT(BLK_TC_FS)) == 0)
360 return;
361
362 iot = __find_track(t->device, t->sector - (t->bytes >> 10));
363 if (!iot) {
364 fprintf(stderr, "Trying to merge on non-existing request\n");
365 return;
366 }
367
368 rb_erase(&iot->rb_node, &rb_track_root);
369 iot->sector -= t->bytes >> 10;
370 track_rb_insert(iot);
371}
372
373static void log_track_getrq(struct blk_io_trace *t)
374{
375 struct io_track *iot;
376
377 if (!track_ios)
378 return;
379
380 iot = find_track(t->pid, t->device, t->sector);
381 iot->allocation_time = t->time;
382}
383
384
385/*
386 * return time between rq allocation and queue
387 */
388static unsigned long long log_track_queue(struct blk_io_trace *t)
389{
390 unsigned long long elapsed;
391 struct io_track *iot;
392
393 if (!track_ios)
394 return -1;
395
396 iot = find_track(t->pid, t->device, t->sector);
397 iot->queue_time = t->time;
398 elapsed = iot->queue_time - iot->allocation_time;
399
400 if (per_process_stats) {
401 struct per_process_info *ppi = find_process_by_pid(iot->pid);
402 int w = (t->action & BLK_TC_ACT(BLK_TC_WRITE)) != 0;
403
404 if (ppi && elapsed > ppi->longest_allocation_wait[w])
405 ppi->longest_allocation_wait[w] = elapsed;
406 }
407
408 return elapsed;
409}
410
411/*
412 * return time between queue and issue
413 */
414static unsigned long long log_track_issue(struct blk_io_trace *t)
415{
416 unsigned long long elapsed;
417 struct io_track *iot;
418
419 if (!track_ios)
420 return -1;
421 if ((t->action & BLK_TC_ACT(BLK_TC_FS)) == 0)
422 return -1;
423
424 iot = __find_track(t->device, t->sector);
425 if (!iot) {
426 fprintf(stderr, "Trying to issue on non-existing request\n");
427 return -1;
428 }
429
430 iot->dispatch_time = t->time;
431 elapsed = iot->dispatch_time - iot->queue_time;
432
433 if (per_process_stats) {
434 struct per_process_info *ppi = find_process_by_pid(iot->pid);
435 int w = (t->action & BLK_TC_ACT(BLK_TC_WRITE)) != 0;
436
437 if (ppi && elapsed > ppi->longest_dispatch_wait[w])
438 ppi->longest_dispatch_wait[w] = elapsed;
439 }
440
441 return elapsed;
442}
443
444/*
445 * return time between dispatch and complete
446 */
447static unsigned long long log_track_complete(struct blk_io_trace *t)
448{
449 unsigned long long elapsed;
450 struct io_track *iot;
451
452 if (!track_ios)
453 return -1;
454 if ((t->action & BLK_TC_ACT(BLK_TC_FS)) == 0)
455 return -1;
456
457 iot = __find_track(t->device, t->sector);
458 if (!iot) {
459 fprintf(stderr, "Trying to dispatch on non-existing request\n");
460 return -1;
461 }
462
463 iot->completion_time = t->time;
464 elapsed = iot->completion_time - iot->dispatch_time;
465
466 if (per_process_stats) {
467 struct per_process_info *ppi = find_process_by_pid(iot->pid);
468 int w = (t->action & BLK_TC_ACT(BLK_TC_WRITE)) != 0;
469
470 if (ppi && elapsed > ppi->longest_completion_wait[w])
471 ppi->longest_completion_wait[w] = elapsed;
472 }
473
474 /*
475 * kill the trace, we don't need it after completion
476 */
477 rb_erase(&iot->rb_node, &rb_track_root);
478 free(iot);
479
480 return elapsed;
481}
482
483
484static struct io_stats *find_process_io_stats(__u32 pid, char *name)
485{
486 struct per_process_info *ppi = find_process_by_pid(pid);
487
488 if (!ppi) {
489 ppi = malloc(sizeof(*ppi));
490 memset(ppi, 0, sizeof(*ppi));
491 strncpy(ppi->name, name, sizeof(ppi->name));
492 ppi->pid = pid;
493 add_process_to_hash(ppi);
494 add_process_to_list(ppi);
495 }
496
497 return &ppi->io_stats;
498}
499
500
501static void resize_cpu_info(struct per_dev_info *pdi, int cpu)
502{
503 struct per_cpu_info *cpus = pdi->cpus;
504 int ncpus = pdi->ncpus;
505 int new_count = cpu + 1;
506 int new_space, size;
507 char *new_start;
508
509 size = new_count * sizeof(struct per_cpu_info);
510 cpus = realloc(cpus, size);
511 if (!cpus) {
512 char name[20];
513 fprintf(stderr, "Out of memory, CPU info for device %s (%d)\n",
514 get_dev_name(pdi, name, sizeof(name)), size);
515 exit(1);
516 }
517
518 new_start = (char *)cpus + (ncpus * sizeof(struct per_cpu_info));
519 new_space = (new_count - ncpus) * sizeof(struct per_cpu_info);
520 memset(new_start, 0, new_space);
521
522 pdi->ncpus = new_count;
523 pdi->cpus = cpus;
524}
525
526static struct per_cpu_info *get_cpu_info(struct per_dev_info *pdi, int cpu)
527{
528 if (cpu >= pdi->ncpus)
529 resize_cpu_info(pdi, cpu);
530 return &pdi->cpus[cpu];
531}
532
533
534static int resize_devices(char *name)
535{
536 int size = (ndevices + 1) * sizeof(struct per_dev_info);
537
538 devices = realloc(devices, size);
539 if (!devices) {
540 fprintf(stderr, "Out of memory, device %s (%d)\n", name, size);
541 return 1;
542 }
543 memset(&devices[ndevices], 0, sizeof(struct per_dev_info));
544 devices[ndevices].name = name;
545 ndevices++;
546 return 0;
547}
548
549static struct per_dev_info *get_dev_info(dev_t id, int create)
550{
551 int i;
552
553 for (i = 0; i < ndevices; i++)
554 if (devices[i].id == id)
555 return &devices[i];
556 if (!create)
557 return NULL;
558 if (resize_devices(NULL) != 0)
559 return NULL;
560 return &devices[ndevices-1];
561}
562
563static char *get_dev_name(struct per_dev_info *pdi, char *buffer, int size)
564{
565 if (pdi->name)
566 snprintf(buffer, size, "%s", pdi->name);
567 else
568 snprintf(buffer, size, "%d,%d", MAJOR(pdi->id), MINOR(pdi->id));
569 return buffer;
570}
571
572
573static void check_time(struct per_dev_info *pdi, struct blk_io_trace *bit)
574{
575 unsigned long long this = bit->time;
576 unsigned long long last = pdi->last_reported_time;
577
578 pdi->backwards = (this < last) ? 'B' : ' ';
579 pdi->last_reported_time = this;
580}
581
582
583static inline void __account_m(struct io_stats *ios, struct blk_io_trace *t,
584 int rw)
585{
586 if (rw) {
587 ios->mwrites++;
588 ios->qwrite_kb += t->bytes >> 10;
589 } else {
590 ios->mreads++;
591 ios->qread_kb += t->bytes >> 10;
592 }
593}
594
595static inline void account_m(struct blk_io_trace *t, struct per_cpu_info *pci,
596 int rw)
597{
598 __account_m(&pci->io_stats, t, rw);
599
600 if (per_process_stats) {
601 struct io_stats *ios = find_process_io_stats(t->pid, t->comm);
602
603 __account_m(ios, t, rw);
604 }
605}
606
607static inline void __account_q(struct io_stats *ios, struct blk_io_trace *t,
608 int rw)
609{
610 if (rw) {
611 ios->qwrites++;
612 ios->qwrite_kb += t->bytes >> 10;
613 } else {
614 ios->qreads++;
615 ios->qread_kb += t->bytes >> 10;
616 }
617}
618
619static inline void account_q(struct blk_io_trace *t, struct per_cpu_info *pci,
620 int rw)
621{
622 __account_q(&pci->io_stats, t, rw);
623
624 if (per_process_stats) {
625 struct io_stats *ios = find_process_io_stats(t->pid, t->comm);
626
627 __account_q(ios, t, rw);
628 }
629}
630
631static inline void __account_c(struct io_stats *ios, int rw, unsigned int bytes)
632{
633 if (rw) {
634 ios->cwrites++;
635 ios->cwrite_kb += bytes >> 10;
636 } else {
637 ios->creads++;
638 ios->cread_kb += bytes >> 10;
639 }
640}
641
642static inline void account_c(struct blk_io_trace *t, struct per_cpu_info *pci,
643 int rw, int bytes)
644{
645 __account_c(&pci->io_stats, rw, bytes);
646
647 if (per_process_stats) {
648 struct io_stats *ios = find_process_io_stats(t->pid, t->comm);
649
650 __account_c(ios, rw, bytes);
651 }
652}
653
654static inline void __account_i(struct io_stats *ios, int rw, unsigned int bytes)
655{
656 if (rw) {
657 ios->iwrites++;
658 ios->iwrite_kb += bytes >> 10;
659 } else {
660 ios->ireads++;
661 ios->iread_kb += bytes >> 10;
662 }
663}
664
665static inline void account_i(struct blk_io_trace *t, struct per_cpu_info *pci,
666 int rw)
667{
668 __account_i(&pci->io_stats, rw, t->bytes);
669
670 if (per_process_stats) {
671 struct io_stats *ios = find_process_io_stats(t->pid, t->comm);
672
673 __account_i(ios, rw, t->bytes);
674 }
675}
676
677static void output(struct per_cpu_info *pci, char *s)
678{
679 fprintf(ofp, "%s", s);
680}
681
682static char hstring[256];
683static char tstring[256];
684
685static inline char *setup_header(struct per_cpu_info *pci,
686 struct blk_io_trace *t, char act)
687{
688 int w = t->action & BLK_TC_ACT(BLK_TC_WRITE);
689 int b = t->action & BLK_TC_ACT(BLK_TC_BARRIER);
690 int s = t->action & BLK_TC_ACT(BLK_TC_SYNC);
691 char rwbs[4];
692 int i = 0;
693
694 if (w)
695 rwbs[i++] = 'W';
696 else
697 rwbs[i++] = 'R';
698 if (b)
699 rwbs[i++] = 'B';
700 if (s)
701 rwbs[i++] = 'S';
702
703 rwbs[i] = '\0';
704
705 sprintf(hstring, "%3d,%-3d %2d %8ld %5Lu.%09Lu %5u %c %3s",
706 MAJOR(t->device), MINOR(t->device), pci->cpu,
707 (unsigned long)t->sequence, SECONDS(t->time),
708 NANO_SECONDS(t->time), t->pid, act, rwbs);
709
710 return hstring;
711}
712
713static void log_complete(struct per_cpu_info *pci, struct blk_io_trace *t,
714 char act)
715{
716 unsigned long long elapsed = log_track_complete(t);
717
718 if (elapsed != -1ULL) {
719 unsigned long usec = elapsed / 1000;
720
721 sprintf(tstring,"%s %Lu + %u (%8lu) [%d]\n",
722 setup_header(pci, t, act),
723 (unsigned long long)t->sector, t->bytes >> 9,
724 usec, t->error);
725 } else {
726 sprintf(tstring,"%s %Lu + %u [%d]\n", setup_header(pci, t, act),
727 (unsigned long long)t->sector, t->bytes >> 9, t->error);
728 }
729
730 output(pci, tstring);
731}
732
733static void log_queue(struct per_cpu_info *pci, struct blk_io_trace *t,
734 char act)
735{
736 unsigned long long elapsed = log_track_queue(t);
737
738 if (elapsed != -1ULL) {
739 unsigned long usec = elapsed / 1000;
740
741 sprintf(tstring,"%s %Lu + %u (%8lu) [%s]\n",
742 setup_header(pci, t, act),
743 (unsigned long long)t->sector, t->bytes >> 9,
744 usec, t->comm);
745 } else {
746 sprintf(tstring,"%s %Lu + %u [%s]\n", setup_header(pci, t, act),
747 (unsigned long long)t->sector, t->bytes >> 9, t->comm);
748 }
749 output(pci, tstring);
750}
751
752static void log_issue(struct per_cpu_info *pci, struct blk_io_trace *t,
753 char act)
754{
755 unsigned long long elapsed = log_track_issue(t);
756
757 if (elapsed != -1ULL) {
758 double usec = (double) elapsed / 1000;
759
760 sprintf(tstring,"%s %Lu + %u (%8.2f) [%s]\n",
761 setup_header(pci, t, act),
762 (unsigned long long)t->sector, t->bytes >> 9,
763 usec, t->comm);
764 } else {
765 sprintf(tstring,"%s %Lu + %u [%s]\n", setup_header(pci, t, act),
766 (unsigned long long)t->sector, t->bytes >> 9, t->comm);
767 }
768
769 output(pci, tstring);
770}
771
772static void log_merge(struct per_cpu_info *pci, struct blk_io_trace *t,
773 char act)
774{
775 log_track_merge(t);
776
777 sprintf(tstring,"%s %Lu + %u [%s]\n", setup_header(pci, t, act),
778 (unsigned long long)t->sector, t->bytes >> 9, t->comm);
779 output(pci, tstring);
780}
781
782static void log_generic(struct per_cpu_info *pci, struct blk_io_trace *t,
783 char act)
784{
785 sprintf(tstring,"%s %Lu + %u [%s]\n", setup_header(pci, t, act),
786 (unsigned long long)t->sector, t->bytes >> 9, t->comm);
787 output(pci, tstring);
788}
789
790static int log_pc(struct per_cpu_info *pci, struct blk_io_trace *t, char act)
791{
792 unsigned char *buf;
793 int i;
794
795 sprintf(tstring,"%s ", setup_header(pci, t, act));
796 output(pci, tstring);
797
798 buf = (unsigned char *) t + sizeof(*t);
799 for (i = 0; i < t->pdu_len; i++) {
800 sprintf(tstring,"%02x ", buf[i]);
801 output(pci, tstring);
802 }
803
804 if (act == 'C') {
805 sprintf(tstring,"[%d]\n", t->error);
806 output(pci, tstring);
807 } else {
808 sprintf(tstring,"[%s]\n", t->comm);
809 output(pci, tstring);
810 }
811 return 0;
812}
813
814static int dump_trace_pc(struct blk_io_trace *t, struct per_cpu_info *pci)
815{
816 int ret = 0;
817
818 switch (t->action & 0xffff) {
819 case __BLK_TA_QUEUE:
820 log_generic(pci, t, 'Q');
821 break;
822 case __BLK_TA_GETRQ:
823 log_generic(pci, t, 'G');
824 break;
825 case __BLK_TA_SLEEPRQ:
826 log_generic(pci, t, 'S');
827 break;
828 case __BLK_TA_REQUEUE:
829 log_generic(pci, t, 'R');
830 break;
831 case __BLK_TA_ISSUE:
832 ret = log_pc(pci, t, 'D');
833 break;
834 case __BLK_TA_COMPLETE:
835 log_pc(pci, t, 'C');
836 break;
837 default:
838 fprintf(stderr, "Bad pc action %x\n", t->action);
839 ret = 1;
840 break;
841 }
842
843 return ret;
844}
845
846static void dump_trace_fs(struct blk_io_trace *t, struct per_cpu_info *pci)
847{
848 int w = t->action & BLK_TC_ACT(BLK_TC_WRITE);
849 int act = t->action & 0xffff;
850
851 switch (act) {
852 case __BLK_TA_QUEUE:
853 account_q(t, pci, w);
854 log_queue(pci, t, 'Q');
855 break;
856 case __BLK_TA_BACKMERGE:
857 account_m(t, pci, w);
858 log_merge(pci, t, 'M');
859 break;
860 case __BLK_TA_FRONTMERGE:
861 account_m(t, pci, w);
862 log_merge(pci, t, 'F');
863 break;
864 case __BLK_TA_GETRQ:
865 log_track_getrq(t);
866 log_generic(pci, t, 'G');
867 break;
868 case __BLK_TA_SLEEPRQ:
869 log_generic(pci, t, 'S');
870 break;
871 case __BLK_TA_REQUEUE:
872 account_c(t, pci, w, -t->bytes);
873 log_queue(pci, t, 'R');
874 break;
875 case __BLK_TA_ISSUE:
876 account_i(t, pci, w);
877 log_issue(pci, t, 'D');
878 break;
879 case __BLK_TA_COMPLETE:
880 account_c(t, pci, w, t->bytes);
881 log_complete(pci, t, 'C');
882 break;
883 default:
884 fprintf(stderr, "Bad fs action %x\n", t->action);
885 break;
886 }
887}
888
889static int dump_trace(struct blk_io_trace *t, struct per_cpu_info *pci,
890 struct per_dev_info *pdi)
891{
892 int ret = 0;
893
894 if (t->action & BLK_TC_ACT(BLK_TC_PC))
895 ret = dump_trace_pc(t, pci);
896 else
897 dump_trace_fs(t, pci);
898
899 pdi->events++;
900 return ret;
901}
902
903static void dump_io_stats(struct io_stats *ios, char *msg)
904{
905 fprintf(ofp, "%s\n", msg);
906
907 fprintf(ofp, " Reads Queued: %'8lu, %'8LuKiB\t", ios->qreads, ios->qread_kb);
908 fprintf(ofp, " Writes Queued: %'8lu, %'8LuKiB\n", ios->qwrites,ios->qwrite_kb);
909
910 fprintf(ofp, " Read Dispatches: %'8lu, %'8LuKiB\t", ios->ireads, ios->iread_kb);
911 fprintf(ofp, " Write Dispatches: %'8lu, %'8LuKiB\n", ios->iwrites,ios->iwrite_kb);
912 fprintf(ofp, " Reads Completed: %'8lu, %'8LuKiB\t", ios->creads, ios->cread_kb);
913 fprintf(ofp, " Writes Completed: %'8lu, %'8LuKiB\n", ios->cwrites,ios->cwrite_kb);
914 fprintf(ofp, " Read Merges: %'8lu%8c\t", ios->mreads, ' ');
915
916 fprintf(ofp, " Write Merges: %'8lu\n", ios->mwrites);
917}
918
919static void dump_wait_stats(struct per_process_info *ppi)
920{
921 unsigned long rawait = ppi->longest_allocation_wait[0] / 1000;
922 unsigned long rdwait = ppi->longest_dispatch_wait[0] / 1000;
923 unsigned long rcwait = ppi->longest_completion_wait[0] / 1000;
924 unsigned long wawait = ppi->longest_allocation_wait[1] / 1000;
925 unsigned long wdwait = ppi->longest_dispatch_wait[1] / 1000;
926 unsigned long wcwait = ppi->longest_completion_wait[1] / 1000;
927
928 fprintf(ofp, " Allocation wait: %'8lu%8c\t", rawait, ' ');
929 fprintf(ofp, " Allocation wait: %'8lu\n", wawait);
930 fprintf(ofp, " Dispatch wait: %'8lu%8c\t", rdwait, ' ');
931 fprintf(ofp, " Dispatch wait: %'8lu\n", wdwait);
932 fprintf(ofp, " Completion wait: %'8lu%8c\t", rcwait, ' ');
933 fprintf(ofp, " Completion wait: %'8lu\n", wcwait);
934}
935
936static void show_process_stats(void)
937{
938 struct per_process_info *ppi;
939
940 ppi = ppi_list;
941 while (ppi) {
942 dump_io_stats(&ppi->io_stats, ppi->name);
943 dump_wait_stats(ppi);
944 ppi = ppi->list_next;
945 }
946
947 fprintf(ofp, "\n");
948}
949
950static void show_device_and_cpu_stats(void)
951{
952 struct per_dev_info *pdi;
953 struct per_cpu_info *pci;
954 struct io_stats total, *ios;
955 int i, j, pci_events;
956 char line[3 + 8/*cpu*/ + 2 + 32/*dev*/ + 3];
957 char name[32];
958
959 for (pdi = devices, i = 0; i < ndevices; i++, pdi++) {
960
961 memset(&total, 0, sizeof(total));
962 pci_events = 0;
963
964 if (i > 0)
965 fprintf(ofp, "\n");
966
967 for (pci = pdi->cpus, j = 0; j < pdi->ncpus; j++, pci++) {
968 if (!pci->nelems)
969 continue;
970
971 ios = &pci->io_stats;
972 total.qreads += ios->qreads;
973 total.qwrites += ios->qwrites;
974 total.creads += ios->creads;
975 total.cwrites += ios->cwrites;
976 total.mreads += ios->mreads;
977 total.mwrites += ios->mwrites;
978 total.ireads += ios->ireads;
979 total.iwrites += ios->iwrites;
980 total.qread_kb += ios->qread_kb;
981 total.qwrite_kb += ios->qwrite_kb;
982 total.cread_kb += ios->cread_kb;
983 total.cwrite_kb += ios->cwrite_kb;
984 total.iread_kb += ios->iread_kb;
985 total.iwrite_kb += ios->iwrite_kb;
986
987 snprintf(line, sizeof(line) - 1, "CPU%d (%s):",
988 j, get_dev_name(pdi, name, sizeof(name)));
989 dump_io_stats(ios, line);
990 pci_events++;
991 }
992
993 if (pci_events > 1) {
994 fprintf(ofp, "\n");
995 snprintf(line, sizeof(line) - 1, "Total (%s):",
996 get_dev_name(pdi, name, sizeof(name)));
997 dump_io_stats(&total, line);
998 }
999
1000 fprintf(ofp, "Events (%s): %'Lu\n",
1001 get_dev_name(pdi, line, sizeof(line)), pdi->events);
1002 }
1003}
1004
1005static struct blk_io_trace *find_trace(void *p, unsigned long offset, int nr)
1006{
1007 unsigned long max_offset = min(offset,nr * sizeof(struct blk_io_trace));
1008 unsigned long off;
1009 struct blk_io_trace *bit;
1010 __u32 magic;
1011
1012 for (off = 0; off < max_offset; off++) {
1013 bit = p + off;
1014
1015 magic = be32_to_cpu(bit->magic);
1016 if ((magic & 0xffffff00) == BLK_IO_TRACE_MAGIC)
1017 return bit;
1018 }
1019
1020 return NULL;
1021}
1022
1023static int sort_entries(void *traces, unsigned long offset, int nr,
1024 struct per_dev_info *fpdi, struct per_cpu_info *fpci)
1025{
1026 struct per_dev_info *pdi;
1027 struct per_cpu_info *pci;
1028 struct blk_io_trace *bit;
1029 struct trace *t;
1030 void *start = traces;
1031
1032 while (traces - start <= offset - sizeof(*bit)) {
1033 if (!nr)
1034 break;
1035
1036 bit = find_trace(traces, offset - (traces - start), nr);
1037 if (!bit)
1038 break;
1039
1040 t = malloc(sizeof(*t));
1041 if (!t) {
1042 fprintf(stderr, "Out of memory, seq %d on dev %d,%d\n",
1043 bit->sequence,
1044 MAJOR(bit->device), MINOR(bit->device));
1045 return -1;
1046 }
1047 t->bit = bit;
1048 memset(&t->rb_node, 0, sizeof(t->rb_node));
1049
1050 trace_to_cpu(bit);
1051
1052 if (verify_trace(bit)) {
1053 free(t);
1054 break;
1055 }
1056
1057 pdi = fpdi ? fpdi : get_dev_info(bit->device, 1);
1058 pdi->id = bit->device;
1059 pci = fpci ? fpci : get_cpu_info(pdi, bit->cpu);
1060 pci->cpu = bit->cpu;
1061 pci->nelems++;
1062
1063 if (trace_rb_insert(t)) {
1064 free(t);
1065 return -1;
1066 }
1067
1068 traces += sizeof(*bit) + bit->pdu_len;
1069 nr--;
1070 }
1071
1072 return 0;
1073}
1074
1075static void free_entries_rb(void)
1076{
1077 struct rb_node *n;
1078
1079 while ((n = rb_first(&rb_sort_root)) != NULL) {
1080 struct trace *t = rb_entry(n, struct trace, rb_node);
1081
1082 rb_erase(&t->rb_node, &rb_sort_root);
1083 free(t);
1084 }
1085}
1086
1087static void show_entries_rb(void)
1088{
1089 struct per_dev_info *pdi;
1090 struct blk_io_trace *bit;
1091 struct rb_node *n;
1092 struct trace *t;
1093 int cpu;
1094
1095 n = rb_first(&rb_sort_root);
1096 if (!n)
1097 return;
1098
1099 do {
1100 t = rb_entry(n, struct trace, rb_node);
1101 bit = t->bit;
1102
1103 pdi = get_dev_info(bit->device, 0);
1104 if (!pdi) {
1105 fprintf(stderr, "Unknown device ID? (%d,%d)\n",
1106 MAJOR(bit->device), MINOR(bit->device));
1107 break;
1108 }
1109 cpu = bit->cpu;
1110 if (cpu > pdi->ncpus) {
1111 fprintf(stderr, "Unknown CPU ID? (%d, device %d,%d)\n",
1112 cpu, MAJOR(bit->device), MINOR(bit->device));
1113 break;
1114 }
1115
1116 bit->time -= genesis_time;
1117 if (bit->time < stopwatch_start)
1118 continue;
1119 if (bit->time >= stopwatch_end)
1120 break;
1121
1122 check_time(pdi, bit);
1123
1124 if (dump_trace(bit, &pdi->cpus[cpu], pdi))
1125 break;
1126
1127 } while ((n = rb_next(n)) != NULL);
1128}
1129
1130static int read_data(int fd, void *buffer, int bytes, int block)
1131{
1132 int ret, bytes_left, fl;
1133 void *p;
1134
1135 fl = fcntl(fd, F_GETFL);
1136
1137 if (!block)
1138 fcntl(fd, F_SETFL, fl | O_NONBLOCK);
1139 else
1140 fcntl(fd, F_SETFL, fl & ~O_NONBLOCK);
1141
1142 bytes_left = bytes;
1143 p = buffer;
1144 while (bytes_left > 0) {
1145 ret = read(fd, p, bytes_left);
1146 if (!ret)
1147 return 1;
1148 else if (ret < 0) {
1149 if (errno != EAGAIN)
1150 perror("read");
1151 return -1;
1152 } else {
1153 p += ret;
1154 bytes_left -= ret;
1155 }
1156 }
1157
1158 return 0;
1159}
1160
1161static int do_file(void)
1162{
1163 struct per_dev_info *pdi;
1164 int i, j, nfiles = 0;
1165
1166 for (pdi = devices, i = 0; i < ndevices; i++, pdi++) {
1167 for (j = 0;; j++, nfiles++) {
1168 struct per_cpu_info *pci;
1169 struct stat st;
1170 void *tb;
1171
1172 pci = get_cpu_info(pdi, j);
1173 pci->cpu = j;
1174
1175 snprintf(pci->fname, sizeof(pci->fname)-1,
1176 "%s_out.%d", pdi->name, j);
1177 if (stat(pci->fname, &st) < 0)
1178 break;
1179 if (!st.st_size)
1180 continue;
1181
1182 printf("Processing %s\n", pci->fname);
1183
1184 tb = malloc(st.st_size);
1185 if (!tb) {
1186 fprintf(stderr, "Out of memory, skip file %s\n",
1187 pci->fname);
1188 continue;
1189 }
1190
1191 pci->fd = open(pci->fname, O_RDONLY);
1192 if (pci->fd < 0) {
1193 perror(pci->fname);
1194 free(tb);
1195 continue;
1196 }
1197
1198 if (read_data(pci->fd, tb, st.st_size, 1)) {
1199 close(pci->fd);
1200 free(tb);
1201 continue;
1202 }
1203
1204 if (sort_entries(tb, st.st_size, ~0U, pdi, pci) == -1) {
1205 close(pci->fd);
1206 free(tb);
1207 continue;
1208 }
1209
1210 printf("Completed %s (CPU%d %d, entries)\n",
1211 pci->fname, j, pci->nelems);
1212 close(pci->fd);
1213 }
1214 }
1215
1216 if (!nfiles) {
1217 fprintf(stderr, "No files found\n");
1218 return 1;
1219 }
1220
1221 show_entries_rb();
1222 return 0;
1223}
1224
1225static void resize_buffer(void **buffer, long *size, long offset)
1226{
1227 long old_size = *size;
1228
1229 if (*size == 0)
1230 *size = 64 * sizeof(struct blk_io_trace);
1231
1232 *size *= 2;
1233 *buffer = realloc(*buffer, *size);
1234
1235 if (old_size)
1236 memset(*buffer + offset, 0, *size - old_size);
1237}
1238
1239static int read_sort_events(int fd, void **buffer, long *max_offset)
1240{
1241 long offset;
1242 int events;
1243
1244 events = offset = 0;
1245 do {
1246 struct blk_io_trace *t;
1247 int pdu_len;
1248 __u32 magic;
1249
1250 if (*max_offset - offset < sizeof(*t))
1251 resize_buffer(buffer, max_offset, offset);
1252
1253 if (read_data(fd, *buffer + offset, sizeof(*t), !events))
1254 break;
1255
1256 t = *buffer + offset;
1257 offset += sizeof(*t);
1258
1259 magic = be32_to_cpu(t->magic);
1260 if ((magic & 0xffffff00) != BLK_IO_TRACE_MAGIC) {
1261 fprintf(stderr, "Bad magic %x\n", magic);
1262 break;
1263 }
1264
1265 pdu_len = be16_to_cpu(t->pdu_len);
1266 if (pdu_len) {
1267 if (*max_offset - offset <= pdu_len)
1268 resize_buffer(buffer, max_offset, offset);
1269
1270 if (read_data(fd, *buffer + offset, pdu_len, 1))
1271 break;
1272
1273 offset += pdu_len;
1274 }
1275
1276 events++;
1277 } while (!is_done() && events < rb_batch);
1278
1279 return events;
1280}
1281
1282static int do_stdin(void)
1283{
1284 int fd;
1285 void *ptr = NULL;
1286 long max_offset;
1287
1288 fd = dup(STDIN_FILENO);
1289 max_offset = 0;
1290 do {
1291 int events;
1292
1293 events = read_sort_events(fd, &ptr, &max_offset);
1294 if (!events)
1295 break;
1296
1297 if (sort_entries(ptr, ~0UL, events, NULL, NULL) == -1)
1298 break;
1299
1300 show_entries_rb();
1301 free_entries_rb();
1302 } while (1);
1303
1304 if (ptr)
1305 free(ptr);
1306
1307 close(fd);
1308 return 0;
1309}
1310
1311static void flush_output(void)
1312{
1313 fflush(ofp);
1314}
1315
1316static void handle_sigint(int sig)
1317{
1318 done = 1;
1319 flush_output();
1320}
1321
1322/*
1323 * Extract start and duration times from a string, allowing
1324 * us to specify a time interval of interest within a trace.
1325 * Format: "duration" (start is zero) or "start:duration".
1326 */
1327static int find_stopwatch_interval(char *string)
1328{
1329 double value;
1330 char *sp;
1331
1332 value = strtod(string, &sp);
1333 if (sp == string) {
1334 fprintf(stderr,"Invalid stopwatch timer: %s\n", string);
1335 return 1;
1336 }
1337 if (*sp == ':') {
1338 stopwatch_start = DOUBLE_TO_NANO_ULL(value);
1339 string = sp + 1;
1340 value = strtod(string, &sp);
1341 if (sp == string || *sp != '\0') {
1342 fprintf(stderr,"Invalid stopwatch duration time: %s\n",
1343 string);
1344 return 1;
1345 }
1346 } else if (*sp != '\0') {
1347 fprintf(stderr,"Invalid stopwatch start timer: %s\n", string);
1348 return 1;
1349 }
1350 stopwatch_end = stopwatch_start + DOUBLE_TO_NANO_ULL(value);
1351 return 0;
1352}
1353
1354static void usage(char *prog)
1355{
1356 fprintf(stderr, "Usage: %s "
1357 "[-i <name>] [-o <output>] [-s] [-w N[:n]] <name>...\n",
1358 prog);
1359}
1360
1361int main(int argc, char *argv[])
1362{
1363 char *ofp_buffer;
1364 int c, ret, mode;
1365 int per_device_and_cpu_stats = 1;
1366
1367 while ((c = getopt_long(argc, argv, S_OPTS, l_opts, NULL)) != -1) {
1368 switch (c) {
1369 case 'i':
1370 if (!strcmp(optarg, "-") && !pipeline)
1371 pipeline = 1;
1372 else if (resize_devices(optarg) != 0)
1373 return 1;
1374 break;
1375 case 'o':
1376 output_name = optarg;
1377 break;
1378 case 'b':
1379 rb_batch = atoi(optarg);
1380 if (rb_batch <= 0)
1381 rb_batch = RB_BATCH_DEFAULT;
1382 break;
1383 case 's':
1384 per_process_stats = 1;
1385 break;
1386 case 't':
1387 track_ios = 1;
1388 break;
1389 case 'q':
1390 per_device_and_cpu_stats = 0;
1391 break;
1392 case 'w':
1393 if (find_stopwatch_interval(optarg) != 0)
1394 return 1;
1395 break;
1396 default:
1397 usage(argv[0]);
1398 return 1;
1399 }
1400 }
1401
1402 while (optind < argc) {
1403 if (!strcmp(argv[optind], "-") && !pipeline)
1404 pipeline = 1;
1405 else if (resize_devices(argv[optind]) != 0)
1406 return 1;
1407 optind++;
1408 }
1409
1410 if (!pipeline && !ndevices) {
1411 usage(argv[0]);
1412 return 1;
1413 }
1414
1415 memset(&rb_sort_root, 0, sizeof(rb_sort_root));
1416 memset(&rb_track_root, 0, sizeof(rb_track_root));
1417
1418 signal(SIGINT, handle_sigint);
1419 signal(SIGHUP, handle_sigint);
1420 signal(SIGTERM, handle_sigint);
1421
1422 setlocale(LC_NUMERIC, "en_US");
1423
1424 if (!output_name) {
1425 ofp = fdopen(STDOUT_FILENO, "w");
1426 mode = _IOLBF;
1427 } else {
1428 char ofname[128];
1429
1430 snprintf(ofname, sizeof(ofname) - 1, "%s.log", output_name);
1431 ofp = fopen(ofname, "w");
1432 mode = _IOFBF;
1433 }
1434
1435 if (!ofp) {
1436 perror("fopen");
1437 return 1;
1438 }
1439
1440 ofp_buffer = malloc(4096);
1441 if (setvbuf(ofp, ofp_buffer, mode, 4096)) {
1442 perror("setvbuf");
1443 return 1;
1444 }
1445
1446 if (pipeline)
1447 ret = do_stdin();
1448 else
1449 ret = do_file();
1450
1451 if (per_process_stats)
1452 show_process_stats();
1453
1454 if (per_device_and_cpu_stats)
1455 show_device_and_cpu_stats();
1456
1457 flush_output();
1458 return ret;
1459}