blkparse: Avoid segfault for wrong cpu number.
[blktrace.git] / blktrace.h
... / ...
CommitLineData
1#ifndef BLKTRACE_H
2#define BLKTRACE_H
3
4#include <stdio.h>
5#include <byteswap.h>
6#include <endian.h>
7
8#include "blktrace_api.h"
9#include "rbtree.h"
10
11#define MINORBITS 20
12#define MINORMASK ((1U << MINORBITS) - 1)
13#define MAJOR(dev) ((unsigned int) ((dev) >> MINORBITS))
14#define MINOR(dev) ((unsigned int) ((dev) & MINORMASK))
15
16#define SECONDS(x) ((unsigned long long)(x) / 1000000000)
17#define NANO_SECONDS(x) ((unsigned long long)(x) % 1000000000)
18#define DOUBLE_TO_NANO_ULL(d) ((unsigned long long)((d) * 1000000000))
19
20#define min(a, b) ((a) < (b) ? (a) : (b))
21#define max(a, b) ((a) > (b) ? (a) : (b))
22
23#define t_sec(t) ((t)->bytes >> 9)
24#define t_kb(t) ((t)->bytes >> 10)
25
26typedef __u32 u32;
27typedef __u8 u8;
28
29struct io_stats {
30 unsigned long qreads, qwrites, creads, cwrites, mreads, mwrites;
31 unsigned long ireads, iwrites, rrqueue, wrqueue;
32 unsigned long long qread_kb, qwrite_kb, cread_kb, cwrite_kb;
33 unsigned long long iread_kb, iwrite_kb;
34 unsigned long long mread_kb, mwrite_kb;
35 unsigned long qreads_pc, qwrites_pc, ireads_pc, iwrites_pc;
36 unsigned long rrqueue_pc, wrqueue_pc, creads_pc, cwrites_pc;
37 unsigned long long qread_kb_pc, qwrite_kb_pc, iread_kb_pc, iwrite_kb_pc;
38 unsigned long io_unplugs, timer_unplugs;
39};
40
41struct per_cpu_info {
42 unsigned int cpu;
43 unsigned int nelems;
44
45 int fd;
46 int fdblock;
47 char fname[128];
48
49 struct io_stats io_stats;
50
51 struct rb_root rb_last;
52 unsigned long rb_last_entries;
53 unsigned long last_sequence;
54 unsigned long smallest_seq_read;
55
56 struct skip_info *skips_head;
57 struct skip_info *skips_tail;
58};
59
60extern FILE *ofp;
61extern int data_is_native;
62extern struct timespec abs_start_time;
63
64#define CHECK_MAGIC(t) (((t)->magic & 0xffffff00) == BLK_IO_TRACE_MAGIC)
65#define SUPPORTED_VERSION (0x07)
66
67#if __BYTE_ORDER == __LITTLE_ENDIAN
68#define be16_to_cpu(x) __bswap_16(x)
69#define be32_to_cpu(x) __bswap_32(x)
70#define be64_to_cpu(x) __bswap_64(x)
71#define cpu_to_be16(x) __bswap_16(x)
72#define cpu_to_be32(x) __bswap_32(x)
73#define cpu_to_be64(x) __bswap_64(x)
74#elif __BYTE_ORDER == __BIG_ENDIAN
75#define be16_to_cpu(x) (x)
76#define be32_to_cpu(x) (x)
77#define be64_to_cpu(x) (x)
78#define cpu_to_be16(x) (x)
79#define cpu_to_be32(x) (x)
80#define cpu_to_be64(x) (x)
81#else
82#error "Bad arch"
83#endif
84
85static inline int verify_trace(struct blk_io_trace *t)
86{
87 if (!CHECK_MAGIC(t)) {
88 fprintf(stderr, "bad trace magic %x\n", t->magic);
89 return 1;
90 }
91 if ((t->magic & 0xff) != SUPPORTED_VERSION) {
92 fprintf(stderr, "unsupported trace version %x\n",
93 t->magic & 0xff);
94 return 1;
95 }
96
97 return 0;
98}
99
100static inline void trace_to_cpu(struct blk_io_trace *t)
101{
102 if (data_is_native)
103 return;
104
105 t->magic = be32_to_cpu(t->magic);
106 t->sequence = be32_to_cpu(t->sequence);
107 t->time = be64_to_cpu(t->time);
108 t->sector = be64_to_cpu(t->sector);
109 t->bytes = be32_to_cpu(t->bytes);
110 t->action = be32_to_cpu(t->action);
111 t->pid = be32_to_cpu(t->pid);
112 t->device = be32_to_cpu(t->device);
113 t->cpu = be16_to_cpu(t->cpu);
114 t->error = be16_to_cpu(t->error);
115 t->pdu_len = be16_to_cpu(t->pdu_len);
116}
117
118/*
119 * check whether data is native or not
120 */
121static inline int check_data_endianness(u32 magic)
122{
123 if ((magic & 0xffffff00) == BLK_IO_TRACE_MAGIC) {
124 data_is_native = 1;
125 return 0;
126 }
127
128 magic = __bswap_32(magic);
129 if ((magic & 0xffffff00) == BLK_IO_TRACE_MAGIC) {
130 data_is_native = 0;
131 return 0;
132 }
133
134 return 1;
135}
136
137extern void set_all_format_specs(char *);
138extern int add_format_spec(char *);
139extern void process_fmt(char *, struct per_cpu_info *, struct blk_io_trace *,
140 unsigned long long, int, unsigned char *);
141extern int valid_act_opt(int);
142extern int find_mask_map(char *);
143extern char *find_process_name(pid_t);
144
145#endif