Don't like btrecord against libaio and librt, as it doesn't use any of their symbols
[blktrace.git] / blktrace.h
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
26 typedef __u32 u32;
27 typedef __u8 u8;
28
29 struct 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 io_unplugs, timer_unplugs;
36 };
37
38 struct per_cpu_info {
39         unsigned int cpu;
40         unsigned int nelems;
41
42         int fd;
43         int fdblock;
44         char fname[128];
45
46         struct io_stats io_stats;
47
48         struct rb_root rb_last;
49         unsigned long rb_last_entries;
50         unsigned long last_sequence;
51         unsigned long smallest_seq_read;
52
53         struct skip_info *skips_head;
54         struct skip_info *skips_tail;
55 };
56
57 extern FILE *ofp;
58 extern int data_is_native;
59 extern struct timespec abs_start_time;
60
61 #define CHECK_MAGIC(t)          (((t)->magic & 0xffffff00) == BLK_IO_TRACE_MAGIC)
62 #define SUPPORTED_VERSION       (0x07)
63
64 #if __BYTE_ORDER == __LITTLE_ENDIAN
65 #define be16_to_cpu(x)          __bswap_16(x)
66 #define be32_to_cpu(x)          __bswap_32(x)
67 #define be64_to_cpu(x)          __bswap_64(x)
68 #define cpu_to_be16(x)          __bswap_16(x)
69 #define cpu_to_be32(x)          __bswap_32(x)
70 #define cpu_to_be64(x)          __bswap_64(x)
71 #elif __BYTE_ORDER == __BIG_ENDIAN
72 #define be16_to_cpu(x)          (x)
73 #define be32_to_cpu(x)          (x)
74 #define be64_to_cpu(x)          (x)
75 #define cpu_to_be16(x)          (x)
76 #define cpu_to_be32(x)          (x)
77 #define cpu_to_be64(x)          (x)
78 #else
79 #error "Bad arch"
80 #endif
81
82 static inline int verify_trace(struct blk_io_trace *t)
83 {
84         if (!CHECK_MAGIC(t)) {
85                 fprintf(stderr, "bad trace magic %x\n", t->magic);
86                 return 1;
87         }
88         if ((t->magic & 0xff) != SUPPORTED_VERSION) {
89                 fprintf(stderr, "unsupported trace version %x\n", 
90                         t->magic & 0xff);
91                 return 1;
92         }
93
94         return 0;
95 }
96
97 static inline void trace_to_cpu(struct blk_io_trace *t)
98 {
99         if (data_is_native)
100                 return;
101
102         t->magic        = be32_to_cpu(t->magic);
103         t->sequence     = be32_to_cpu(t->sequence);
104         t->time         = be64_to_cpu(t->time);
105         t->sector       = be64_to_cpu(t->sector);
106         t->bytes        = be32_to_cpu(t->bytes);
107         t->action       = be32_to_cpu(t->action);
108         t->pid          = be32_to_cpu(t->pid);
109         t->device       = be32_to_cpu(t->device);
110         t->cpu          = be16_to_cpu(t->cpu);
111         t->error        = be16_to_cpu(t->error);
112         t->pdu_len      = be16_to_cpu(t->pdu_len);
113 }
114
115 /*
116  * check whether data is native or not
117  */
118 static inline int check_data_endianness(u32 magic)
119 {
120         if ((magic & 0xffffff00) == BLK_IO_TRACE_MAGIC) {
121                 data_is_native = 1;
122                 return 0;
123         }
124
125         magic = __bswap_32(magic);
126         if ((magic & 0xffffff00) == BLK_IO_TRACE_MAGIC) {
127                 data_is_native = 0;
128                 return 0;
129         }
130
131         return 1;
132 }
133
134 extern void set_all_format_specs(char *);
135 extern int add_format_spec(char *);
136 extern void process_fmt(char *, struct per_cpu_info *, struct blk_io_trace *,
137                         unsigned long long, int, unsigned char *);
138 extern int valid_act_opt(int);
139 extern int find_mask_map(char *);
140 extern char *find_process_name(pid_t);
141
142 #endif