blkparse: Avoid segfault for wrong cpu number.
[blktrace.git] / blkparse_fmt.c
CommitLineData
71d5d4c9
JA
1/*
2 * This file contains format parsing code for blkparse, allowing you to
3 * customize the individual action format and generel output format.
4 */
5#include <stdio.h>
6#include <string.h>
7#include <stdlib.h>
8#include <unistd.h>
9#include <ctype.h>
7bd4fd0a 10#include <time.h>
71d5d4c9
JA
11
12#include "blktrace.h"
13
4c446dc0 14#define VALID_SPECS "ABCDFGIMPQRSTUWX"
71d5d4c9
JA
15
16#define HEADER "%D %2c %8s %5T.%9t %5p %2a %3d "
17
18static char *override_format[256];
19
20static inline int valid_spec(int spec)
21{
22 return strchr(VALID_SPECS, spec) != NULL;
23}
24
e820abd7 25void set_all_format_specs(char *option)
71d5d4c9
JA
26{
27 char *p;
28
29 for (p = VALID_SPECS; *p; p++)
30 if (override_format[(int)(*p)] == NULL)
e820abd7 31 override_format[(int)(*p)] = strdup(option);
71d5d4c9
JA
32}
33
e820abd7 34int add_format_spec(char *option)
71d5d4c9
JA
35{
36 int spec = optarg[0];
37
38 if (!valid_spec(spec)) {
39 fprintf(stderr,"Bad format specifier %c\n", spec);
40 return 1;
41 }
42 if (optarg[1] != ',') {
e820abd7 43 fprintf(stderr,"Bad format specifier - need ',' %s\n", option);
71d5d4c9
JA
44 return 1;
45 }
e820abd7 46 option += 2;
71d5d4c9 47
0ac4c20e 48 override_format[spec] = strdup(option);
71d5d4c9
JA
49
50 return 0;
51}
52
768277cc
JA
53static inline void fill_rwbs(char *rwbs, struct blk_io_trace *t)
54{
55 int w = t->action & BLK_TC_ACT(BLK_TC_WRITE);
f86990ed 56 int a = t->action & BLK_TC_ACT(BLK_TC_AHEAD);
768277cc
JA
57 int b = t->action & BLK_TC_ACT(BLK_TC_BARRIER);
58 int s = t->action & BLK_TC_ACT(BLK_TC_SYNC);
711e4d25 59 int m = t->action & BLK_TC_ACT(BLK_TC_META);
37081bf5 60 int d = t->action & BLK_TC_ACT(BLK_TC_DISCARD);
768277cc
JA
61 int i = 0;
62
37081bf5
DW
63 if (d)
64 rwbs[i++] = 'D';
65 else if (w)
768277cc 66 rwbs[i++] = 'W';
d57450af 67 else if (t->bytes)
768277cc 68 rwbs[i++] = 'R';
d57450af
JA
69 else
70 rwbs[i++] = 'N';
f86990ed
NS
71 if (a)
72 rwbs[i++] = 'A';
768277cc
JA
73 if (b)
74 rwbs[i++] = 'B';
75 if (s)
76 rwbs[i++] = 'S';
711e4d25
JA
77 if (m)
78 rwbs[i++] = 'M';
768277cc
JA
79
80 rwbs[i] = '\0';
81}
82
7bd4fd0a
OK
83static const char *
84print_time(unsigned long long timestamp)
85{
86 static char timebuf[128];
87 struct tm *tm;
88 time_t sec;
89 unsigned long nsec;
90
91 sec = abs_start_time.tv_sec + SECONDS(timestamp);
92 nsec = abs_start_time.tv_nsec + NANO_SECONDS(timestamp);
93 if (nsec >= 1000000000) {
94 nsec -= 1000000000;
95 sec += 1;
96 }
97
98 tm = localtime(&sec);
99 snprintf(timebuf, sizeof(timebuf),
100 "%02u:%02u:%02u.%06lu",
101 tm->tm_hour,
102 tm->tm_min,
103 tm->tm_sec,
104 nsec / 1000);
105 return timebuf;
106}
107
63756b50 108static inline int pdu_rest_is_zero(unsigned char *pdu, int len)
1d6376a5 109{
63756b50 110 static char zero[4096];
1d6376a5 111
63756b50 112 return !memcmp(pdu, zero, len);
1d6376a5
JA
113}
114
768277cc
JA
115static char *dump_pdu(unsigned char *pdu_buf, int pdu_len)
116{
117 static char p[4096];
118 int i, len;
119
120 if (!pdu_buf || !pdu_len)
121 return NULL;
122
123 for (len = 0, i = 0; i < pdu_len; i++) {
124 if (i)
125 len += sprintf(p + len, " ");
126
127 len += sprintf(p + len, "%02x", pdu_buf[i]);
1d6376a5
JA
128
129 /*
130 * usually dump for cdb dumps where we can see lots of
131 * zeroes, stop when the rest is just zeroes and indicate
132 * so with a .. appended
133 */
134 if (!pdu_buf[i] && pdu_rest_is_zero(pdu_buf + i, pdu_len - i)) {
135 sprintf(p + len, " ..");
136 break;
137 }
768277cc
JA
138 }
139
140 return p;
141}
142
a8f30e64
JA
143#define pdu_start(t) (((void *) (t) + sizeof(struct blk_io_trace)))
144
768277cc
JA
145static unsigned int get_pdu_int(struct blk_io_trace *t)
146{
a8f30e64 147 __u64 *val = pdu_start(t);
768277cc
JA
148
149 return be64_to_cpu(*val);
150}
151
a8f30e64
JA
152static void get_pdu_remap(struct blk_io_trace *t, struct blk_io_trace_remap *r)
153{
154 struct blk_io_trace_remap *__r = pdu_start(t);
580f7440 155 __u64 sector_from = __r->sector_from;
a8f30e64 156
97d13fb0 157 r->device_from = be32_to_cpu(__r->device_from);
580f7440
AB
158 r->device_to = be32_to_cpu(__r->device_to);
159 r->sector_from = be64_to_cpu(sector_from);
a8f30e64
JA
160}
161
71d5d4c9
JA
162static void print_field(char *act, struct per_cpu_info *pci,
163 struct blk_io_trace *t, unsigned long long elapsed,
164 int pdu_len, unsigned char *pdu_buf, char field,
165 int minus, int has_w, int width)
166{
167 char format[64];
168
169 if (has_w) {
170 if (minus)
171 sprintf(format, "%%-%d", width);
172 else
173 sprintf(format, "%%%d", width);
174 } else
175 sprintf(format, "%%");
176
177 switch (field) {
178 case 'a':
179 fprintf(ofp, strcat(format, "s"), act);
180 break;
181 case 'c':
182 fprintf(ofp, strcat(format, "d"), pci->cpu);
183 break;
bfc70ad5
JA
184 case 'C': {
185 char *name = find_process_name(t->pid);
186
187 fprintf(ofp, strcat(format, "s"), name);
71d5d4c9 188 break;
bfc70ad5 189 }
71d5d4c9 190 case 'd': {
711e4d25 191 char rwbs[6];
768277cc
JA
192
193 fill_rwbs(rwbs, t);
71d5d4c9
JA
194 fprintf(ofp, strcat(format, "s"), rwbs);
195 break;
196 }
197 case 'D': /* format width ignored */
198 fprintf(ofp,"%3d,%-3d", MAJOR(t->device), MINOR(t->device));
199 break;
200 case 'e':
201 fprintf(ofp, strcat(format, "d"), t->error);
202 break;
203 case 'M':
204 fprintf(ofp, strcat(format, "d"), MAJOR(t->device));
205 break;
206 case 'm':
207 fprintf(ofp, strcat(format, "d"), MINOR(t->device));
208 break;
209 case 'n':
ae957cbc 210 fprintf(ofp, strcat(format, "u"), t_sec(t));
71d5d4c9 211 break;
1c8ca7b5
JA
212 case 'N':
213 fprintf(ofp, strcat(format, "u"), t->bytes);
214 break;
71d5d4c9
JA
215 case 'p':
216 fprintf(ofp, strcat(format, "u"), t->pid);
217 break;
768277cc
JA
218 case 'P': { /* format width ignored */
219 char *p = dump_pdu(pdu_buf, pdu_len);
220 if (p)
221 fprintf(ofp, "%s", p);
71d5d4c9 222 break;
768277cc 223 }
71d5d4c9
JA
224 case 's':
225 fprintf(ofp, strcat(format, "ld"), t->sequence);
226 break;
227 case 'S':
228 fprintf(ofp, strcat(format, "lu"), t->sector);
229 break;
230 case 't':
231 sprintf(format, "%%0%dlu", has_w ? width : 9);
232 fprintf(ofp, format, NANO_SECONDS(t->time));
233 break;
234 case 'T':
235 fprintf(ofp, strcat(format, "d"), SECONDS(t->time));
236 break;
237 case 'u':
238 if (elapsed == -1ULL) {
239 fprintf(stderr, "Expecting elapsed value\n");
240 exit(1);
241 }
242 fprintf(ofp, strcat(format, "llu"), elapsed / 1000);
243 break;
7bd4fd0a 244 case 'U':
768277cc 245 fprintf(ofp, strcat(format, "u"), get_pdu_int(t));
71d5d4c9 246 break;
7bd4fd0a
OK
247 case 'z':
248 fprintf(ofp, strcat(format, "s"), print_time(t->time));
249 break;
71d5d4c9
JA
250 default:
251 fprintf(ofp,strcat(format, "c"), field);
252 break;
253 }
254}
255
768277cc
JA
256static char *parse_field(char *act, struct per_cpu_info *pci,
257 struct blk_io_trace *t, unsigned long long elapsed,
258 int pdu_len, unsigned char *pdu_buf,
71d5d4c9
JA
259 char *master_format)
260{
261 int minus = 0;
262 int has_w = 0;
263 int width = 0;
264 char *p = master_format;
265
266 if (*p == '-') {
267 minus = 1;
268 p++;
269 }
270 if (isdigit(*p)) {
271 has_w = 1;
272 do {
273 width = (width * 10) + (*p++ - '0');
274 } while ((*p) && (isdigit(*p)));
275 }
276 if (*p) {
277 print_field(act, pci, t, elapsed, pdu_len, pdu_buf, *p++,
278 minus, has_w, width);
279 }
280 return p;
281}
282
768277cc
JA
283static void process_default(char *act, struct per_cpu_info *pci,
284 struct blk_io_trace *t, unsigned long long elapsed,
285 int pdu_len, unsigned char *pdu_buf)
71d5d4c9 286{
580f7440 287 struct blk_io_trace_remap r = { .device_from = 0, };
711e4d25 288 char rwbs[6];
bfc70ad5 289 char *name;
71d5d4c9 290
768277cc
JA
291 fill_rwbs(rwbs, t);
292
580f7440
AB
293 /*
294 * For remaps we have to modify the device using the remap structure
295 * passed up.
296 */
297 if (act[0] == 'A') {
298 get_pdu_remap(t, &r);
299 t->device = r.device_to;
300 }
301
768277cc
JA
302 /*
303 * The header is always the same
304 */
305 fprintf(ofp, "%3d,%-3d %2d %8d %5d.%09lu %5u %2s %3s ",
306 MAJOR(t->device), MINOR(t->device), pci->cpu, t->sequence,
307 (int) SECONDS(t->time), (unsigned long) NANO_SECONDS(t->time),
308 t->pid, act, rwbs);
71d5d4c9 309
bfc70ad5
JA
310 name = find_process_name(t->pid);
311
768277cc 312 switch (act[0]) {
80c53608 313 case 'R': /* Requeue */
71d5d4c9 314 case 'C': /* Complete */
768277cc
JA
315 if (t->action & BLK_TC_ACT(BLK_TC_PC)) {
316 char *p = dump_pdu(pdu_buf, pdu_len);
317 if (p)
318 fprintf(ofp, "(%s) ", p);
319 fprintf(ofp, "[%d]\n", t->error);
320 } else {
71d5d4c9 321 if (elapsed != -1ULL) {
d57450af
JA
322 if (t_sec(t))
323 fprintf(ofp, "%llu + %u (%8llu) [%d]\n",
324 (unsigned long long) t->sector,
325 t_sec(t), elapsed, t->error);
326 else
327 fprintf(ofp, "%llu (%8llu) [%d]\n",
328 (unsigned long long) t->sector,
329 elapsed, t->error);
768277cc 330 } else {
d57450af
JA
331 if (t_sec(t))
332 fprintf(ofp, "%llu + %u [%d]\n",
333 (unsigned long long) t->sector,
334 t_sec(t), t->error);
335 else
336 fprintf(ofp, "%llu [%d]\n",
337 (unsigned long long) t->sector,
338 t->error);
768277cc 339 }
71d5d4c9 340 }
71d5d4c9
JA
341 break;
342
343 case 'D': /* Issue */
71d5d4c9 344 case 'I': /* Insert */
71d5d4c9 345 case 'Q': /* Queue */
bd5e71cf 346 case 'B': /* Bounce */
768277cc
JA
347 if (t->action & BLK_TC_ACT(BLK_TC_PC)) {
348 char *p;
1c8ca7b5 349 fprintf(ofp, "%u ", t->bytes);
768277cc
JA
350 p = dump_pdu(pdu_buf, pdu_len);
351 if (p)
352 fprintf(ofp, "(%s) ", p);
bfc70ad5 353 fprintf(ofp, "[%s]\n", name);
768277cc
JA
354 } else {
355 if (elapsed != -1ULL) {
d57450af
JA
356 if (t_sec(t))
357 fprintf(ofp, "%llu + %u (%8llu) [%s]\n",
358 (unsigned long long) t->sector,
359 t_sec(t), elapsed, name);
360 else
361 fprintf(ofp, "(%8llu) [%s]\n", elapsed,
362 name);
768277cc 363 } else {
d57450af
JA
364 if (t_sec(t))
365 fprintf(ofp, "%llu + %u [%s]\n",
366 (unsigned long long) t->sector,
367 t_sec(t), name);
368 else
369 fprintf(ofp, "[%s]\n", name);
768277cc
JA
370 }
371 }
71d5d4c9
JA
372 break;
373
bd5e71cf 374 case 'M': /* Back merge */
71d5d4c9 375 case 'F': /* Front merge */
768277cc
JA
376 case 'G': /* Get request */
377 case 'S': /* Sleep request */
d57450af
JA
378 if (t_sec(t))
379 fprintf(ofp, "%llu + %u [%s]\n",
380 (unsigned long long) t->sector, t_sec(t), name);
381 else
382 fprintf(ofp, "[%s]\n", name);
71d5d4c9
JA
383 break;
384
385 case 'P': /* Plug */
bfc70ad5 386 fprintf(ofp, "[%s]\n", name);
71d5d4c9
JA
387 break;
388
389 case 'U': /* Unplug IO */
a8f30e64 390 case 'T': /* Unplug timer */
bfc70ad5 391 fprintf(ofp, "[%s] %u\n", name, get_pdu_int(t));
71d5d4c9
JA
392 break;
393
97d13fb0 394 case 'A': /* remap */
580f7440 395 get_pdu_remap(t, &r);
a8f30e64 396 fprintf(ofp, "%llu + %u <- (%d,%d) %llu\n",
e9f883ca 397 (unsigned long long) t->sector, t_sec(t),
580f7440
AB
398 MAJOR(r.device_from), MINOR(r.device_from),
399 (unsigned long long) r.sector_from);
a8f30e64 400 break;
1a15f6a8 401
71d5d4c9 402 case 'X': /* Split */
768277cc 403 fprintf(ofp, "%llu / %u [%s]\n", (unsigned long long) t->sector,
bfc70ad5 404 get_pdu_int(t), name);
71d5d4c9
JA
405 break;
406
1a15f6a8
AB
407 case 'm': /* Message */
408 fprintf(ofp, "%*s\n", pdu_len, pdu_buf);
409 break;
410
71d5d4c9 411 default:
768277cc
JA
412 fprintf(stderr, "Unknown action %c\n", act[0]);
413 break;
71d5d4c9
JA
414 }
415
71d5d4c9
JA
416}
417
418void process_fmt(char *act, struct per_cpu_info *pci, struct blk_io_trace *t,
419 unsigned long long elapsed, int pdu_len,
420 unsigned char *pdu_buf)
421{
768277cc
JA
422 char *p = override_format[(int) *act];
423
424 if (!p) {
425 process_default(act, pci, t, elapsed, pdu_len, pdu_buf);
426 return;
427 }
71d5d4c9
JA
428
429 while (*p) {
430 switch (*p) {
431 case '%': /* Field specifier */
432 p++;
433 if (*p == '%')
434 fprintf(ofp, "%c", *p++);
435 else if (!*p)
436 fprintf(ofp, "%c", '%');
437 else
438 p = parse_field(act, pci, t, elapsed,
439 pdu_len, pdu_buf, p);
440 break;
441 case '\\': { /* escape */
442 switch (p[1]) {
443 case 'b': fprintf(ofp, "\b"); break;
444 case 'n': fprintf(ofp, "\n"); break;
445 case 'r': fprintf(ofp, "\r"); break;
446 case 't': fprintf(ofp, "\t"); break;
447 default:
448 fprintf(stderr,
449 "Invalid escape char in format %c\n",
450 p[1]);
451 exit(1);
452 /*NOTREACHED*/
453 }
454 p += 2;
455 break;
456 }
457 default:
458 fprintf(ofp, "%c", *p++);
459 break;
460 }
461 }
462}
463
464