tools/lib/traceevent/event-parse.c

   1 // SPDX-License-Identifier: LGPL-2.1
   2 /*
   3  * Copyright (C) 2009, 2010 Red Hat Inc, Steven Rostedt <[email protected]>
   4  *
   5  *
   6  *  The parts for function graph printing was taken and modified from the
   7  *  Linux Kernel that were written by
   8  *    - Copyright (C) 2009  Frederic Weisbecker,
   9  *  Frederic Weisbecker gave his permission to relicense the code to
  10  *  the Lesser General Public License.
  11  */
  12 #include <inttypes.h>
  13 #include <stdio.h>
  14 #include <stdlib.h>
  15 #include <string.h>
  16 #include <stdarg.h>
  17 #include <ctype.h>
  18 #include <errno.h>
  19 #include <stdint.h>
  20 #include <limits.h>
  21 #include <linux/time64.h>
  22
  23 #include <netinet/in.h>
  24 #include "event-parse.h"
  25
  26 #include "event-parse-local.h"
  27 #include "event-utils.h"
  28 #include "trace-seq.h"
  29
  30 static const char *input_buf;
  31 static unsigned long long input_buf_ptr;
  32 static unsigned long long input_buf_siz;
  33
  34 static int is_flag_field;
  35 static int is_symbolic_field;
  36
  37 static int show_warning = 1;
  38
  39 #define do_warning(fmt, ...)                            \
  40         do {                                            \
  41                 if (show_warning)                       \
  42                         warning(fmt, ##__VA_ARGS__);    \
  43         } while (0)
  44
  45 #define do_warning_event(event, fmt, ...)                       \
  46         do {                                                    \
  47                 if (!show_warning)                              \
  48                         continue;                               \
  49                                                                 \
  50                 if (event)                                      \
  51                         warning("[%s:%s] " fmt, event->system,  \
  52                                 event->name, ##__VA_ARGS__);    \
  53                 else                                            \
  54                         warning(fmt, ##__VA_ARGS__);            \
  55         } while (0)
  56
  57 static void init_input_buf(const char *buf, unsigned long long size)
  58 {
  59         input_buf = buf;
  60         input_buf_siz = size;
  61         input_buf_ptr = 0;
  62 }
  63
  64 const char *tep_get_input_buf(void)
  65 {
  66         return input_buf;
  67 }
  68
  69 unsigned long long tep_get_input_buf_ptr(void)
  70 {
  71         return input_buf_ptr;
  72 }
  73
  74 struct event_handler {
  75         struct event_handler            *next;
  76         int                             id;
  77         const char                      *sys_name;
  78         const char                      *event_name;
  79         tep_event_handler_func          func;
  80         void                            *context;
  81 };
  82
  83 struct func_params {
  84         struct func_params      *next;
  85         enum tep_func_arg_type  type;
  86 };
  87
  88 struct tep_function_handler {
  89         struct tep_function_handler     *next;
  90         enum tep_func_arg_type          ret_type;
  91         char                            *name;
  92         tep_func_handler                func;
  93         struct func_params              *params;
  94         int                             nr_args;
  95 };
  96
  97 static unsigned long long
  98 process_defined_func(struct trace_seq *s, void *data, int size,
  99                      struct tep_event *event, struct tep_print_arg *arg);
 100
 101 static void free_func_handle(struct tep_function_handler *func);
 102
 103 /**
 104  * tep_buffer_init - init buffer for parsing
 105  * @buf: buffer to parse
 106  * @size: the size of the buffer
 107  *
 108  * For use with tep_read_token(), this initializes the internal
 109  * buffer that tep_read_token() will parse.
 110  */
 111 void tep_buffer_init(const char *buf, unsigned long long size)
 112 {
 113         init_input_buf(buf, size);
 114 }
 115
 116 void breakpoint(void)
 117 {
 118         static int x;
 119         x++;
 120 }
 121
 122 struct tep_print_arg *alloc_arg(void)
 123 {
 124         return calloc(1, sizeof(struct tep_print_arg));
 125 }
 126
 127 struct tep_cmdline {
 128         char *comm;
 129         int pid;
 130 };
 131
 132 static int cmdline_cmp(const void *a, const void *b)
 133 {
 134         const struct tep_cmdline *ca = a;
 135         const struct tep_cmdline *cb = b;
 136
 137         if (ca->pid < cb->pid)
 138                 return -1;
 139         if (ca->pid > cb->pid)
 140                 return 1;
 141
 142         return 0;
 143 }
 144
 145 struct cmdline_list {
 146         struct cmdline_list     *next;
 147         char                    *comm;
 148         int                     pid;
 149 };
 150
 151 static int cmdline_init(struct tep_handle *tep)
 152 {
 153         struct cmdline_list *cmdlist = tep->cmdlist;
 154         struct cmdline_list *item;
 155         struct tep_cmdline *cmdlines;
 156         int i;
 157
 158         cmdlines = malloc(sizeof(*cmdlines) * tep->cmdline_count);
 159         if (!cmdlines)
 160                 return -1;
 161
 162         i = 0;
 163         while (cmdlist) {
 164                 cmdlines[i].pid = cmdlist->pid;
 165                 cmdlines[i].comm = cmdlist->comm;
 166                 i++;
 167                 item = cmdlist;
 168                 cmdlist = cmdlist->next;
 169                 free(item);
 170         }
 171
 172         qsort(cmdlines, tep->cmdline_count, sizeof(*cmdlines), cmdline_cmp);
 173
 174         tep->cmdlines = cmdlines;
 175         tep->cmdlist = NULL;
 176
 177         return 0;
 178 }
 179
 180 static const char *find_cmdline(struct tep_handle *tep, int pid)
 181 {
 182         const struct tep_cmdline *comm;
 183         struct tep_cmdline key;
 184
 185         if (!pid)
 186                 return "<idle>";
 187
 188         if (!tep->cmdlines && cmdline_init(tep))
 189                 return "<not enough memory for cmdlines!>";
 190
 191         key.pid = pid;
 192
 193         comm = bsearch(&key, tep->cmdlines, tep->cmdline_count,
 194                        sizeof(*tep->cmdlines), cmdline_cmp);
 195
 196         if (comm)
 197                 return comm->comm;
 198         return "<...>";
 199 }
 200
 201 /**
 202  * tep_is_pid_registered - return if a pid has a cmdline registered
 203  * @tep: a handle to the trace event parser context
 204  * @pid: The pid to check if it has a cmdline registered with.
 205  *
 206  * Returns true if the pid has a cmdline mapped to it
 207  * false otherwise.
 208  */
 209 bool tep_is_pid_registered(struct tep_handle *tep, int pid)
 210 {
 211         const struct tep_cmdline *comm;
 212         struct tep_cmdline key;
 213
 214         if (!pid)
 215                 return true;
 216
 217         if (!tep->cmdlines && cmdline_init(tep))
 218                 return false;
 219
 220         key.pid = pid;
 221
 222         comm = bsearch(&key, tep->cmdlines, tep->cmdline_count,
 223                        sizeof(*tep->cmdlines), cmdline_cmp);
 224
 225         if (comm)
 226                 return true;
 227         return false;
 228 }
 229
 230 /*
 231  * If the command lines have been converted to an array, then
 232  * we must add this pid. This is much slower than when cmdlines
 233  * are added before the array is initialized.
 234  */
 235 static int add_new_comm(struct tep_handle *tep,
 236                         const char *comm, int pid, bool override)
 237 {
 238         struct tep_cmdline *cmdlines = tep->cmdlines;
 239         struct tep_cmdline *cmdline;
 240         struct tep_cmdline key;
 241         char *new_comm;
 242
 243         if (!pid)
 244                 return 0;
 245
 246         /* avoid duplicates */
 247         key.pid = pid;
 248
 249         cmdline = bsearch(&key, tep->cmdlines, tep->cmdline_count,
 250                           sizeof(*tep->cmdlines), cmdline_cmp);
 251         if (cmdline) {
 252                 if (!override) {
 253                         errno = EEXIST;
 254                         return -1;
 255                 }
 256                 new_comm = strdup(comm);
 257                 if (!new_comm) {
 258                         errno = ENOMEM;
 259                         return -1;
 260                 }
 261                 free(cmdline->comm);
 262                 cmdline->comm = new_comm;
 263
 264                 return 0;
 265         }
 266
 267         cmdlines = realloc(cmdlines, sizeof(*cmdlines) * (tep->cmdline_count + 1));
 268         if (!cmdlines) {
 269                 errno = ENOMEM;
 270                 return -1;
 271         }
 272
 273         cmdlines[tep->cmdline_count].comm = strdup(comm);
 274         if (!cmdlines[tep->cmdline_count].comm) {
 275                 free(cmdlines);
 276                 errno = ENOMEM;
 277                 return -1;
 278         }
 279
 280         cmdlines[tep->cmdline_count].pid = pid;
 281
 282         if (cmdlines[tep->cmdline_count].comm)
 283                 tep->cmdline_count++;
 284
 285         qsort(cmdlines, tep->cmdline_count, sizeof(*cmdlines), cmdline_cmp);
 286         tep->cmdlines = cmdlines;
 287
 288         return 0;
 289 }
 290
 291 static int _tep_register_comm(struct tep_handle *tep,
 292                               const char *comm, int pid, bool override)
 293 {
 294         struct cmdline_list *item;
 295
 296         if (tep->cmdlines)
 297                 return add_new_comm(tep, comm, pid, override);
 298
 299         item = malloc(sizeof(*item));
 300         if (!item)
 301                 return -1;
 302
 303         if (comm)
 304                 item->comm = strdup(comm);
 305         else
 306                 item->comm = strdup("<...>");
 307         if (!item->comm) {
 308                 free(item);
 309                 return -1;
 310         }
 311         item->pid = pid;
 312         item->next = tep->cmdlist;
 313
 314         tep->cmdlist = item;
 315         tep->cmdline_count++;
 316
 317         return 0;
 318 }
 319
 320 /**
 321  * tep_register_comm - register a pid / comm mapping
 322  * @tep: a handle to the trace event parser context
 323  * @comm: the command line to register
 324  * @pid: the pid to map the command line to
 325  *
 326  * This adds a mapping to search for command line names with
 327  * a given pid. The comm is duplicated. If a command with the same pid
 328  * already exist, -1 is returned and errno is set to EEXIST
 329  */
 330 int tep_register_comm(struct tep_handle *tep, const char *comm, int pid)
 331 {
 332         return _tep_register_comm(tep, comm, pid, false);
 333 }
 334
 335 /**
 336  * tep_override_comm - register a pid / comm mapping
 337  * @tep: a handle to the trace event parser context
 338  * @comm: the command line to register
 339  * @pid: the pid to map the command line to
 340  *
 341  * This adds a mapping to search for command line names with
 342  * a given pid. The comm is duplicated. If a command with the same pid
 343  * already exist, the command string is udapted with the new one
 344  */
 345 int tep_override_comm(struct tep_handle *tep, const char *comm, int pid)
 346 {
 347         if (!tep->cmdlines && cmdline_init(tep)) {
 348                 errno = ENOMEM;
 349                 return -1;
 350         }
 351         return _tep_register_comm(tep, comm, pid, true);
 352 }
 353
 354 int tep_register_trace_clock(struct tep_handle *tep, const char *trace_clock)
 355 {
 356         tep->trace_clock = strdup(trace_clock);
 357         if (!tep->trace_clock) {
 358                 errno = ENOMEM;
 359                 return -1;
 360         }
 361         return 0;
 362 }
 363
 364 struct func_map {
 365         unsigned long long              addr;
 366         char                            *func;
 367         char                            *mod;
 368 };
 369
 370 struct func_list {
 371         struct func_list        *next;
 372         unsigned long long      addr;
 373         char                    *func;
 374         char                    *mod;
 375 };
 376
 377 static int func_cmp(const void *a, const void *b)
 378 {
 379         const struct func_map *fa = a;
 380         const struct func_map *fb = b;
 381
 382         if (fa->addr < fb->addr)
 383                 return -1;
 384         if (fa->addr > fb->addr)
 385                 return 1;
 386
 387         return 0;
 388 }
 389
 390 /*
 391  * We are searching for a record in between, not an exact
 392  * match.
 393  */
 394 static int func_bcmp(const void *a, const void *b)
 395 {
 396         const struct func_map *fa = a;
 397         const struct func_map *fb = b;
 398
 399         if ((fa->addr == fb->addr) ||
 400
 401             (fa->addr > fb->addr &&
 402              fa->addr < (fb+1)->addr))
 403                 return 0;
 404
 405         if (fa->addr < fb->addr)
 406                 return -1;
 407
 408         return 1;
 409 }
 410
 411 static int func_map_init(struct tep_handle *tep)
 412 {
 413         struct func_list *funclist;
 414         struct func_list *item;
 415         struct func_map *func_map;
 416         int i;
 417
 418         func_map = malloc(sizeof(*func_map) * (tep->func_count + 1));
 419         if (!func_map)
 420                 return -1;
 421
 422         funclist = tep->funclist;
 423
 424         i = 0;
 425         while (funclist) {
 426                 func_map[i].func = funclist->func;
 427                 func_map[i].addr = funclist->addr;
 428                 func_map[i].mod = funclist->mod;
 429                 i++;
 430                 item = funclist;
 431                 funclist = funclist->next;
 432                 free(item);
 433         }
 434
 435         qsort(func_map, tep->func_count, sizeof(*func_map), func_cmp);
 436
 437         /*
 438          * Add a special record at the end.
 439          */
 440         func_map[tep->func_count].func = NULL;
 441         func_map[tep->func_count].addr = 0;
 442         func_map[tep->func_count].mod = NULL;
 443
 444         tep->func_map = func_map;
 445         tep->funclist = NULL;
 446
 447         return 0;
 448 }
 449
 450 static struct func_map *
 451 __find_func(struct tep_handle *tep, unsigned long long addr)
 452 {
 453         struct func_map *func;
 454         struct func_map key;
 455
 456         if (!tep->func_map)
 457                 func_map_init(tep);
 458
 459         key.addr = addr;
 460
 461         func = bsearch(&key, tep->func_map, tep->func_count,
 462                        sizeof(*tep->func_map), func_bcmp);
 463
 464         return func;
 465 }
 466
 467 struct func_resolver {
 468         tep_func_resolver_t     *func;
 469         void                    *priv;
 470         struct func_map         map;
 471 };
 472
 473 /**
 474  * tep_set_function_resolver - set an alternative function resolver
 475  * @tep: a handle to the trace event parser context
 476  * @resolver: function to be used
 477  * @priv: resolver function private state.
 478  *
 479  * Some tools may have already a way to resolve kernel functions, allow them to
 480  * keep using it instead of duplicating all the entries inside tep->funclist.
 481  */
 482 int tep_set_function_resolver(struct tep_handle *tep,
 483                               tep_func_resolver_t *func, void *priv)
 484 {
 485         struct func_resolver *resolver = malloc(sizeof(*resolver));
 486
 487         if (resolver == NULL)
 488                 return -1;
 489
 490         resolver->func = func;
 491         resolver->priv = priv;
 492
 493         free(tep->func_resolver);
 494         tep->func_resolver = resolver;
 495
 496         return 0;
 497 }
 498
 499 /**
 500  * tep_reset_function_resolver - reset alternative function resolver
 501  * @tep: a handle to the trace event parser context
 502  *
 503  * Stop using whatever alternative resolver was set, use the default
 504  * one instead.
 505  */
 506 void tep_reset_function_resolver(struct tep_handle *tep)
 507 {
 508         free(tep->func_resolver);
 509         tep->func_resolver = NULL;
 510 }
 511
 512 static struct func_map *
 513 find_func(struct tep_handle *tep, unsigned long long addr)
 514 {
 515         struct func_map *map;
 516
 517         if (!tep->func_resolver)
 518                 return __find_func(tep, addr);
 519
 520         map = &tep->func_resolver->map;
 521         map->mod  = NULL;
 522         map->addr = addr;
 523         map->func = tep->func_resolver->func(tep->func_resolver->priv,
 524                                              &map->addr, &map->mod);
 525         if (map->func == NULL)
 526                 return NULL;
 527
 528         return map;
 529 }
 530
 531 /**
 532  * tep_find_function - find a function by a given address
 533  * @tep: a handle to the trace event parser context
 534  * @addr: the address to find the function with
 535  *
 536  * Returns a pointer to the function stored that has the given
 537  * address. Note, the address does not have to be exact, it
 538  * will select the function that would contain the address.
 539  */
 540 const char *tep_find_function(struct tep_handle *tep, unsigned long long addr)
 541 {
 542         struct func_map *map;
 543
 544         map = find_func(tep, addr);
 545         if (!map)
 546                 return NULL;
 547
 548         return map->func;
 549 }
 550
 551 /**
 552  * tep_find_function_address - find a function address by a given address
 553  * @tep: a handle to the trace event parser context
 554  * @addr: the address to find the function with
 555  *
 556  * Returns the address the function starts at. This can be used in
 557  * conjunction with tep_find_function to print both the function
 558  * name and the function offset.
 559  */
 560 unsigned long long
 561 tep_find_function_address(struct tep_handle *tep, unsigned long long addr)
 562 {
 563         struct func_map *map;
 564
 565         map = find_func(tep, addr);
 566         if (!map)
 567                 return 0;
 568
 569         return map->addr;
 570 }
 571
 572 /**
 573  * tep_register_function - register a function with a given address
 574  * @tep: a handle to the trace event parser context
 575  * @function: the function name to register
 576  * @addr: the address the function starts at
 577  * @mod: the kernel module the function may be in (NULL for none)
 578  *
 579  * This registers a function name with an address and module.
 580  * The @func passed in is duplicated.
 581  */
 582 int tep_register_function(struct tep_handle *tep, char *func,
 583                           unsigned long long addr, char *mod)
 584 {
 585         struct func_list *item = malloc(sizeof(*item));
 586
 587         if (!item)
 588                 return -1;
 589
 590         item->next = tep->funclist;
 591         item->func = strdup(func);
 592         if (!item->func)
 593                 goto out_free;
 594
 595         if (mod) {
 596                 item->mod = strdup(mod);
 597                 if (!item->mod)
 598                         goto out_free_func;
 599         } else
 600                 item->mod = NULL;
 601         item->addr = addr;
 602
 603         tep->funclist = item;
 604         tep->func_count++;
 605
 606         return 0;
 607
 608 out_free_func:
 609         free(item->func);
 610         item->func = NULL;
 611 out_free:
 612         free(item);
 613         errno = ENOMEM;
 614         return -1;
 615 }
 616
 617 /**
 618  * tep_print_funcs - print out the stored functions
 619  * @tep: a handle to the trace event parser context
 620  *
 621  * This prints out the stored functions.
 622  */
 623 void tep_print_funcs(struct tep_handle *tep)
 624 {
 625         int i;
 626
 627         if (!tep->func_map)
 628                 func_map_init(tep);
 629
 630         for (i = 0; i < (int)tep->func_count; i++) {
 631                 printf("%016llx %s",
 632                        tep->func_map[i].addr,
 633                        tep->func_map[i].func);
 634                 if (tep->func_map[i].mod)
 635                         printf(" [%s]\n", tep->func_map[i].mod);
 636                 else
 637                         printf("\n");
 638         }
 639 }
 640
 641 struct printk_map {
 642         unsigned long long              addr;
 643         char                            *printk;
 644 };
 645
 646 struct printk_list {
 647         struct printk_list      *next;
 648         unsigned long long      addr;
 649         char                    *printk;
 650 };
 651
 652 static int printk_cmp(const void *a, const void *b)
 653 {
 654         const struct printk_map *pa = a;
 655         const struct printk_map *pb = b;
 656
 657         if (pa->addr < pb->addr)
 658                 return -1;
 659         if (pa->addr > pb->addr)
 660                 return 1;
 661
 662         return 0;
 663 }
 664
 665 static int printk_map_init(struct tep_handle *tep)
 666 {
 667         struct printk_list *printklist;
 668         struct printk_list *item;
 669         struct printk_map *printk_map;
 670         int i;
 671
 672         printk_map = malloc(sizeof(*printk_map) * (tep->printk_count + 1));
 673         if (!printk_map)
 674                 return -1;
 675
 676         printklist = tep->printklist;
 677
 678         i = 0;
 679         while (printklist) {
 680                 printk_map[i].printk = printklist->printk;
 681                 printk_map[i].addr = printklist->addr;
 682                 i++;
 683                 item = printklist;
 684                 printklist = printklist->next;
 685                 free(item);
 686         }
 687
 688         qsort(printk_map, tep->printk_count, sizeof(*printk_map), printk_cmp);
 689
 690         tep->printk_map = printk_map;
 691         tep->printklist = NULL;
 692
 693         return 0;
 694 }
 695
 696 static struct printk_map *
 697 find_printk(struct tep_handle *tep, unsigned long long addr)
 698 {
 699         struct printk_map *printk;
 700         struct printk_map key;
 701
 702         if (!tep->printk_map && printk_map_init(tep))
 703                 return NULL;
 704
 705         key.addr = addr;
 706
 707         printk = bsearch(&key, tep->printk_map, tep->printk_count,
 708                          sizeof(*tep->printk_map), printk_cmp);
 709
 710         return printk;
 711 }
 712
 713 /**
 714  * tep_register_print_string - register a string by its address
 715  * @tep: a handle to the trace event parser context
 716  * @fmt: the string format to register
 717  * @addr: the address the string was located at
 718  *
 719  * This registers a string by the address it was stored in the kernel.
 720  * The @fmt passed in is duplicated.
 721  */
 722 int tep_register_print_string(struct tep_handle *tep, const char *fmt,
 723                               unsigned long long addr)
 724 {
 725         struct printk_list *item = malloc(sizeof(*item));
 726         char *p;
 727
 728         if (!item)
 729                 return -1;
 730
 731         item->next = tep->printklist;
 732         item->addr = addr;
 733
 734         /* Strip off quotes and '\n' from the end */
 735         if (fmt[0] == '"')
 736                 fmt++;
 737         item->printk = strdup(fmt);
 738         if (!item->printk)
 739                 goto out_free;
 740
 741         p = item->printk + strlen(item->printk) - 1;
 742         if (*p == '"')
 743                 *p = 0;
 744
 745         p -= 2;
 746         if (strcmp(p, "\\n") == 0)
 747                 *p = 0;
 748
 749         tep->printklist = item;
 750         tep->printk_count++;
 751
 752         return 0;
 753
 754 out_free:
 755         free(item);
 756         errno = ENOMEM;
 757         return -1;
 758 }
 759
 760 /**
 761  * tep_print_printk - print out the stored strings
 762  * @tep: a handle to the trace event parser context
 763  *
 764  * This prints the string formats that were stored.
 765  */
 766 void tep_print_printk(struct tep_handle *tep)
 767 {
 768         int i;
 769
 770         if (!tep->printk_map)
 771                 printk_map_init(tep);
 772
 773         for (i = 0; i < (int)tep->printk_count; i++) {
 774                 printf("%016llx %s\n",
 775                        tep->printk_map[i].addr,
 776                        tep->printk_map[i].printk);
 777         }
 778 }
 779
 780 static struct tep_event *alloc_event(void)
 781 {
 782         return calloc(1, sizeof(struct tep_event));
 783 }
 784
 785 static int add_event(struct tep_handle *tep, struct tep_event *event)
 786 {
 787         int i;
 788         struct tep_event **events = realloc(tep->events, sizeof(event) *
 789                                             (tep->nr_events + 1));
 790         if (!events)
 791                 return -1;
 792
 793         tep->events = events;
 794
 795         for (i = 0; i < tep->nr_events; i++) {
 796                 if (tep->events[i]->id > event->id)
 797                         break;
 798         }
 799         if (i < tep->nr_events)
 800                 memmove(&tep->events[i + 1],
 801                         &tep->events[i],
 802                         sizeof(event) * (tep->nr_events - i));
 803
 804         tep->events[i] = event;
 805         tep->nr_events++;
 806
 807         event->tep = tep;
 808
 809         return 0;
 810 }
 811
 812 static int event_item_type(enum tep_event_type type)
 813 {
 814         switch (type) {
 815         case TEP_EVENT_ITEM ... TEP_EVENT_SQUOTE:
 816                 return 1;
 817         case TEP_EVENT_ERROR ... TEP_EVENT_DELIM:
 818         default:
 819                 return 0;
 820         }
 821 }
 822
 823 static void free_flag_sym(struct tep_print_flag_sym *fsym)
 824 {
 825         struct tep_print_flag_sym *next;
 826
 827         while (fsym) {
 828                 next = fsym->next;
 829                 free(fsym->value);
 830                 free(fsym->str);
 831                 free(fsym);
 832                 fsym = next;
 833         }
 834 }
 835
 836 static void free_arg(struct tep_print_arg *arg)
 837 {
 838         struct tep_print_arg *farg;
 839
 840         if (!arg)
 841                 return;
 842
 843         switch (arg->type) {
 844         case TEP_PRINT_ATOM:
 845                 free(arg->atom.atom);
 846                 break;
 847         case TEP_PRINT_FIELD:
 848                 free(arg->field.name);
 849                 break;
 850         case TEP_PRINT_FLAGS:
 851                 free_arg(arg->flags.field);
 852                 free(arg->flags.delim);
 853                 free_flag_sym(arg->flags.flags);
 854                 break;
 855         case TEP_PRINT_SYMBOL:
 856                 free_arg(arg->symbol.field);
 857                 free_flag_sym(arg->symbol.symbols);
 858                 break;
 859         case TEP_PRINT_HEX:
 860         case TEP_PRINT_HEX_STR:
 861                 free_arg(arg->hex.field);
 862                 free_arg(arg->hex.size);
 863                 break;
 864         case TEP_PRINT_INT_ARRAY:
 865                 free_arg(arg->int_array.field);
 866                 free_arg(arg->int_array.count);
 867                 free_arg(arg->int_array.el_size);
 868                 break;
 869         case TEP_PRINT_TYPE:
 870                 free(arg->typecast.type);
 871                 free_arg(arg->typecast.item);
 872                 break;
 873         case TEP_PRINT_STRING:
 874         case TEP_PRINT_BSTRING:
 875                 free(arg->string.string);
 876                 break;
 877         case TEP_PRINT_BITMASK:
 878                 free(arg->bitmask.bitmask);
 879                 break;
 880         case TEP_PRINT_DYNAMIC_ARRAY:
 881         case TEP_PRINT_DYNAMIC_ARRAY_LEN:
 882                 free(arg->dynarray.index);
 883                 break;
 884         case TEP_PRINT_OP:
 885                 free(arg->op.op);
 886                 free_arg(arg->op.left);
 887                 free_arg(arg->op.right);
 888                 break;
 889         case TEP_PRINT_FUNC:
 890                 while (arg->func.args) {
 891                         farg = arg->func.args;
 892                         arg->func.args = farg->next;
 893                         free_arg(farg);
 894                 }
 895                 break;
 896
 897         case TEP_PRINT_NULL:
 898         default:
 899                 break;
 900         }
 901
 902         free(arg);
 903 }
 904
 905 static enum tep_event_type get_type(int ch)
 906 {
 907         if (ch == '\n')
 908                 return TEP_EVENT_NEWLINE;
 909         if (isspace(ch))
 910                 return TEP_EVENT_SPACE;
 911         if (isalnum(ch) || ch == '_')
 912                 return TEP_EVENT_ITEM;
 913         if (ch == '\'')
 914                 return TEP_EVENT_SQUOTE;
 915         if (ch == '"')
 916                 return TEP_EVENT_DQUOTE;
 917         if (!isprint(ch))
 918                 return TEP_EVENT_NONE;
 919         if (ch == '(' || ch == ')' || ch == ',')
 920                 return TEP_EVENT_DELIM;
 921
 922         return TEP_EVENT_OP;
 923 }
 924
 925 static int __read_char(void)
 926 {
 927         if (input_buf_ptr >= input_buf_siz)
 928                 return -1;
 929
 930         return input_buf[input_buf_ptr++];
 931 }
 932
 933 static int __peek_char(void)
 934 {
 935         if (input_buf_ptr >= input_buf_siz)
 936                 return -1;
 937
 938         return input_buf[input_buf_ptr];
 939 }
 940
 941 /**
 942  * tep_peek_char - peek at the next character that will be read
 943  *
 944  * Returns the next character read, or -1 if end of buffer.
 945  */
 946 int tep_peek_char(void)
 947 {
 948         return __peek_char();
 949 }
 950
 951 static int extend_token(char **tok, char *buf, int size)
 952 {
 953         char *newtok = realloc(*tok, size);
 954
 955         if (!newtok) {
 956                 free(*tok);
 957                 *tok = NULL;
 958                 return -1;
 959         }
 960
 961         if (!*tok)
 962                 strcpy(newtok, buf);
 963         else
 964                 strcat(newtok, buf);
 965         *tok = newtok;
 966
 967         return 0;
 968 }
 969
 970 static enum tep_event_type force_token(const char *str, char **tok);
 971
 972 static enum tep_event_type __read_token(char **tok)
 973 {
 974         char buf[BUFSIZ];
 975         int ch, last_ch, quote_ch, next_ch;
 976         int i = 0;
 977         int tok_size = 0;
 978         enum tep_event_type type;
 979
 980         *tok = NULL;
 981
 982
 983         ch = __read_char();
 984         if (ch < 0)
 985                 return TEP_EVENT_NONE;
 986
 987         type = get_type(ch);
 988         if (type == TEP_EVENT_NONE)
 989                 return type;
 990
 991         buf[i++] = ch;
 992
 993         switch (type) {
 994         case TEP_EVENT_NEWLINE:
 995         case TEP_EVENT_DELIM:
 996                 if (asprintf(tok, "%c", ch) < 0)
 997                         return TEP_EVENT_ERROR;
 998
 999                 return type;
1000
1001         case TEP_EVENT_OP:
1002                 switch (ch) {
1003                 case '-':
1004                         next_ch = __peek_char();
1005                         if (next_ch == '>') {
1006                                 buf[i++] = __read_char();
1007                                 break;
1008                         }
1009                         /* fall through */
1010                 case '+':
1011                 case '|':
1012                 case '&':
1013                 case '>':
1014                 case '<':
1015                         last_ch = ch;
1016                         ch = __peek_char();
1017                         if (ch != last_ch)
1018                                 goto test_equal;
1019                         buf[i++] = __read_char();
1020                         switch (last_ch) {
1021                         case '>':
1022                         case '<':
1023                                 goto test_equal;
1024                         default:
1025                                 break;
1026                         }
1027                         break;
1028                 case '!':
1029                 case '=':
1030                         goto test_equal;
1031                 default: /* what should we do instead? */
1032                         break;
1033                 }
1034                 buf[i] = 0;
1035                 *tok = strdup(buf);
1036                 return type;
1037
1038  test_equal:
1039                 ch = __peek_char();
1040                 if (ch == '=')
1041                         buf[i++] = __read_char();
1042                 goto out;
1043
1044         case TEP_EVENT_DQUOTE:
1045         case TEP_EVENT_SQUOTE:
1046                 /* don't keep quotes */
1047                 i--;
1048                 quote_ch = ch;
1049                 last_ch = 0;
1050  concat:
1051                 do {
1052                         if (i == (BUFSIZ - 1)) {
1053                                 buf[i] = 0;
1054                                 tok_size += BUFSIZ;
1055
1056                                 if (extend_token(tok, buf, tok_size) < 0)
1057                                         return TEP_EVENT_NONE;
1058                                 i = 0;
1059                         }
1060                         last_ch = ch;
1061                         ch = __read_char();
1062                         buf[i++] = ch;
1063                         /* the '\' '\' will cancel itself */
1064                         if (ch == '\\' && last_ch == '\\')
1065                                 last_ch = 0;
1066                 } while (ch != quote_ch || last_ch == '\\');
1067                 /* remove the last quote */
1068                 i--;
1069
1070                 /*
1071                  * For strings (double quotes) check the next token.
1072                  * If it is another string, concatinate the two.
1073                  */
1074                 if (type == TEP_EVENT_DQUOTE) {
1075                         unsigned long long save_input_buf_ptr = input_buf_ptr;
1076
1077                         do {
1078                                 ch = __read_char();
1079                         } while (isspace(ch));
1080                         if (ch == '"')
1081                                 goto concat;
1082                         input_buf_ptr = save_input_buf_ptr;
1083                 }
1084
1085                 goto out;
1086
1087         case TEP_EVENT_ERROR ... TEP_EVENT_SPACE:
1088         case TEP_EVENT_ITEM:
1089         default:
1090                 break;
1091         }
1092
1093         while (get_type(__peek_char()) == type) {
1094                 if (i == (BUFSIZ - 1)) {
1095                         buf[i] = 0;
1096                         tok_size += BUFSIZ;
1097
1098                         if (extend_token(tok, buf, tok_size) < 0)
1099                                 return TEP_EVENT_NONE;
1100                         i = 0;
1101                 }
1102                 ch = __read_char();
1103                 buf[i++] = ch;
1104         }
1105
1106  out:
1107         buf[i] = 0;
1108         if (extend_token(tok, buf, tok_size + i + 1) < 0)
1109                 return TEP_EVENT_NONE;
1110
1111         if (type == TEP_EVENT_ITEM) {
1112                 /*
1113                  * Older versions of the kernel has a bug that
1114                  * creates invalid symbols and will break the mac80211
1115                  * parsing. This is a work around to that bug.
1116                  *
1117                  * See Linux kernel commit:
1118                  *  811cb50baf63461ce0bdb234927046131fc7fa8b
1119                  */
1120                 if (strcmp(*tok, "LOCAL_PR_FMT") == 0) {
1121                         free(*tok);
1122                         *tok = NULL;
1123                         return force_token("\"%s\" ", tok);
1124                 } else if (strcmp(*tok, "STA_PR_FMT") == 0) {
1125                         free(*tok);
1126                         *tok = NULL;
1127                         return force_token("\" sta:%pM\" ", tok);
1128                 } else if (strcmp(*tok, "VIF_PR_FMT") == 0) {
1129                         free(*tok);
1130                         *tok = NULL;
1131                         return force_token("\" vif:%p(%d)\" ", tok);
1132                 }
1133         }
1134
1135         return type;
1136 }
1137
1138 static enum tep_event_type force_token(const char *str, char **tok)
1139 {
1140         const char *save_input_buf;
1141         unsigned long long save_input_buf_ptr;
1142         unsigned long long save_input_buf_siz;
1143         enum tep_event_type type;
1144
1145         /* save off the current input pointers */
1146         save_input_buf = input_buf;
1147         save_input_buf_ptr = input_buf_ptr;
1148         save_input_buf_siz = input_buf_siz;
1149
1150         init_input_buf(str, strlen(str));
1151
1152         type = __read_token(tok);
1153
1154         /* reset back to original token */
1155         input_buf = save_input_buf;
1156         input_buf_ptr = save_input_buf_ptr;
1157         input_buf_siz = save_input_buf_siz;
1158
1159         return type;
1160 }
1161
1162 static void free_token(char *tok)
1163 {
1164         if (tok)
1165                 free(tok);
1166 }
1167
1168 static enum tep_event_type read_token(char **tok)
1169 {
1170         enum tep_event_type type;
1171
1172         for (;;) {
1173                 type = __read_token(tok);
1174                 if (type != TEP_EVENT_SPACE)
1175                         return type;
1176
1177                 free_token(*tok);
1178         }
1179
1180         /* not reached */
1181         *tok = NULL;
1182         return TEP_EVENT_NONE;
1183 }
1184
1185 /**
1186  * tep_read_token - access to utilities to use the tep parser
1187  * @tok: The token to return
1188  *
1189  * This will parse tokens from the string given by
1190  * tep_init_data().
1191  *
1192  * Returns the token type.
1193  */
1194 enum tep_event_type tep_read_token(char **tok)
1195 {
1196         return read_token(tok);
1197 }
1198
1199 /**
1200  * tep_free_token - free a token returned by tep_read_token
1201  * @token: the token to free
1202  */
1203 void tep_free_token(char *token)
1204 {
1205         free_token(token);
1206 }
1207
1208 /* no newline */
1209 static enum tep_event_type read_token_item(char **tok)
1210 {
1211         enum tep_event_type type;
1212
1213         for (;;) {
1214                 type = __read_token(tok);
1215                 if (type != TEP_EVENT_SPACE && type != TEP_EVENT_NEWLINE)
1216                         return type;
1217                 free_token(*tok);
1218                 *tok = NULL;
1219         }
1220
1221         /* not reached */
1222         *tok = NULL;
1223         return TEP_EVENT_NONE;
1224 }
1225
1226 static int test_type(enum tep_event_type type, enum tep_event_type expect)
1227 {
1228         if (type != expect) {
1229                 do_warning("Error: expected type %d but read %d",
1230                     expect, type);
1231                 return -1;
1232         }
1233         return 0;
1234 }
1235
1236 static int test_type_token(enum tep_event_type type, const char *token,
1237                     enum tep_event_type expect, const char *expect_tok)
1238 {
1239         if (type != expect) {
1240                 do_warning("Error: expected type %d but read %d",
1241                     expect, type);
1242                 return -1;
1243         }
1244
1245         if (strcmp(token, expect_tok) != 0) {
1246                 do_warning("Error: expected '%s' but read '%s'",
1247                     expect_tok, token);
1248                 return -1;
1249         }
1250         return 0;
1251 }
1252
1253 static int __read_expect_type(enum tep_event_type expect, char **tok, int newline_ok)
1254 {
1255         enum tep_event_type type;
1256
1257         if (newline_ok)
1258                 type = read_token(tok);
1259         else
1260                 type = read_token_item(tok);
1261         return test_type(type, expect);
1262 }
1263
1264 static int read_expect_type(enum tep_event_type expect, char **tok)
1265 {
1266         return __read_expect_type(expect, tok, 1);
1267 }
1268
1269 static int __read_expected(enum tep_event_type expect, const char *str,
1270                            int newline_ok)
1271 {
1272         enum tep_event_type type;
1273         char *token;
1274         int ret;
1275
1276         if (newline_ok)
1277                 type = read_token(&token);
1278         else
1279                 type = read_token_item(&token);
1280
1281         ret = test_type_token(type, token, expect, str);
1282
1283         free_token(token);
1284
1285         return ret;
1286 }
1287
1288 static int read_expected(enum tep_event_type expect, const char *str)
1289 {
1290         return __read_expected(expect, str, 1);
1291 }
1292
1293 static int read_expected_item(enum tep_event_type expect, const char *str)
1294 {
1295         return __read_expected(expect, str, 0);
1296 }
1297
1298 static char *event_read_name(void)
1299 {
1300         char *token;
1301
1302         if (read_expected(TEP_EVENT_ITEM, "name") < 0)
1303                 return NULL;
1304
1305         if (read_expected(TEP_EVENT_OP, ":") < 0)
1306                 return NULL;
1307
1308         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
1309                 goto fail;
1310
1311         return token;
1312
1313  fail:
1314         free_token(token);
1315         return NULL;
1316 }
1317
1318 static int event_read_id(void)
1319 {
1320         char *token;
1321         int id;
1322
1323         if (read_expected_item(TEP_EVENT_ITEM, "ID") < 0)
1324                 return -1;
1325
1326         if (read_expected(TEP_EVENT_OP, ":") < 0)
1327                 return -1;
1328
1329         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
1330                 goto fail;
1331
1332         id = strtoul(token, NULL, 0);
1333         free_token(token);
1334         return id;
1335
1336  fail:
1337         free_token(token);
1338         return -1;
1339 }
1340
1341 static int field_is_string(struct tep_format_field *field)
1342 {
1343         if ((field->flags & TEP_FIELD_IS_ARRAY) &&
1344             (strstr(field->type, "char") || strstr(field->type, "u8") ||
1345              strstr(field->type, "s8")))
1346                 return 1;
1347
1348         return 0;
1349 }
1350
1351 static int field_is_dynamic(struct tep_format_field *field)
1352 {
1353         if (strncmp(field->type, "__data_loc", 10) == 0)
1354                 return 1;
1355
1356         return 0;
1357 }
1358
1359 static int field_is_long(struct tep_format_field *field)
1360 {
1361         /* includes long long */
1362         if (strstr(field->type, "long"))
1363                 return 1;
1364
1365         return 0;
1366 }
1367
1368 static unsigned int type_size(const char *name)
1369 {
1370         /* This covers all TEP_FIELD_IS_STRING types. */
1371         static struct {
1372                 const char *type;
1373                 unsigned int size;
1374         } table[] = {
1375                 { "u8",   1 },
1376                 { "u16",  2 },
1377                 { "u32",  4 },
1378                 { "u64",  8 },
1379                 { "s8",   1 },
1380                 { "s16",  2 },
1381                 { "s32",  4 },
1382                 { "s64",  8 },
1383                 { "char", 1 },
1384                 { },
1385         };
1386         int i;
1387
1388         for (i = 0; table[i].type; i++) {
1389                 if (!strcmp(table[i].type, name))
1390                         return table[i].size;
1391         }
1392
1393         return 0;
1394 }
1395
1396 static int event_read_fields(struct tep_event *event, struct tep_format_field **fields)
1397 {
1398         struct tep_format_field *field = NULL;
1399         enum tep_event_type type;
1400         char *token;
1401         char *last_token;
1402         int count = 0;
1403
1404         do {
1405                 unsigned int size_dynamic = 0;
1406
1407                 type = read_token(&token);
1408                 if (type == TEP_EVENT_NEWLINE) {
1409                         free_token(token);
1410                         return count;
1411                 }
1412
1413                 count++;
1414
1415                 if (test_type_token(type, token, TEP_EVENT_ITEM, "field"))
1416                         goto fail;
1417                 free_token(token);
1418
1419                 type = read_token(&token);
1420                 /*
1421                  * The ftrace fields may still use the "special" name.
1422                  * Just ignore it.
1423                  */
1424                 if (event->flags & TEP_EVENT_FL_ISFTRACE &&
1425                     type == TEP_EVENT_ITEM && strcmp(token, "special") == 0) {
1426                         free_token(token);
1427                         type = read_token(&token);
1428                 }
1429
1430                 if (test_type_token(type, token, TEP_EVENT_OP, ":") < 0)
1431                         goto fail;
1432
1433                 free_token(token);
1434                 if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
1435                         goto fail;
1436
1437                 last_token = token;
1438
1439                 field = calloc(1, sizeof(*field));
1440                 if (!field)
1441                         goto fail;
1442
1443                 field->event = event;
1444
1445                 /* read the rest of the type */
1446                 for (;;) {
1447                         type = read_token(&token);
1448                         if (type == TEP_EVENT_ITEM ||
1449                             (type == TEP_EVENT_OP && strcmp(token, "*") == 0) ||
1450                             /*
1451                              * Some of the ftrace fields are broken and have
1452                              * an illegal "." in them.
1453                              */
1454                             (event->flags & TEP_EVENT_FL_ISFTRACE &&
1455                              type == TEP_EVENT_OP && strcmp(token, ".") == 0)) {
1456
1457                                 if (strcmp(token, "*") == 0)
1458                                         field->flags |= TEP_FIELD_IS_POINTER;
1459
1460                                 if (field->type) {
1461                                         char *new_type;
1462                                         new_type = realloc(field->type,
1463                                                            strlen(field->type) +
1464                                                            strlen(last_token) + 2);
1465                                         if (!new_type) {
1466                                                 free(last_token);
1467                                                 goto fail;
1468                                         }
1469                                         field->type = new_type;
1470                                         strcat(field->type, " ");
1471                                         strcat(field->type, last_token);
1472                                         free(last_token);
1473                                 } else
1474                                         field->type = last_token;
1475                                 last_token = token;
1476                                 continue;
1477                         }
1478
1479                         break;
1480                 }
1481
1482                 if (!field->type) {
1483                         do_warning_event(event, "%s: no type found", __func__);
1484                         goto fail;
1485                 }
1486                 field->name = field->alias = last_token;
1487
1488                 if (test_type(type, TEP_EVENT_OP))
1489                         goto fail;
1490
1491                 if (strcmp(token, "[") == 0) {
1492                         enum tep_event_type last_type = type;
1493                         char *brackets = token;
1494                         char *new_brackets;
1495                         int len;
1496
1497                         field->flags |= TEP_FIELD_IS_ARRAY;
1498
1499                         type = read_token(&token);
1500
1501                         if (type == TEP_EVENT_ITEM)
1502                                 field->arraylen = strtoul(token, NULL, 0);
1503                         else
1504                                 field->arraylen = 0;
1505
1506                         while (strcmp(token, "]") != 0) {
1507                                 if (last_type == TEP_EVENT_ITEM &&
1508                                     type == TEP_EVENT_ITEM)
1509                                         len = 2;
1510                                 else
1511                                         len = 1;
1512                                 last_type = type;
1513
1514                                 new_brackets = realloc(brackets,
1515                                                        strlen(brackets) +
1516                                                        strlen(token) + len);
1517                                 if (!new_brackets) {
1518                                         free(brackets);
1519                                         goto fail;
1520                                 }
1521                                 brackets = new_brackets;
1522                                 if (len == 2)
1523                                         strcat(brackets, " ");
1524                                 strcat(brackets, token);
1525                                 /* We only care about the last token */
1526                                 field->arraylen = strtoul(token, NULL, 0);
1527                                 free_token(token);
1528                                 type = read_token(&token);
1529                                 if (type == TEP_EVENT_NONE) {
1530                                         do_warning_event(event, "failed to find token");
1531                                         goto fail;
1532                                 }
1533                         }
1534
1535                         free_token(token);
1536
1537                         new_brackets = realloc(brackets, strlen(brackets) + 2);
1538                         if (!new_brackets) {
1539                                 free(brackets);
1540                                 goto fail;
1541                         }
1542                         brackets = new_brackets;
1543                         strcat(brackets, "]");
1544
1545                         /* add brackets to type */
1546
1547                         type = read_token(&token);
1548                         /*
1549                          * If the next token is not an OP, then it is of
1550                          * the format: type [] item;
1551                          */
1552                         if (type == TEP_EVENT_ITEM) {
1553                                 char *new_type;
1554                                 new_type = realloc(field->type,
1555                                                    strlen(field->type) +
1556                                                    strlen(field->name) +
1557                                                    strlen(brackets) + 2);
1558                                 if (!new_type) {
1559                                         free(brackets);
1560                                         goto fail;
1561                                 }
1562                                 field->type = new_type;
1563                                 strcat(field->type, " ");
1564                                 strcat(field->type, field->name);
1565                                 size_dynamic = type_size(field->name);
1566                                 free_token(field->name);
1567                                 strcat(field->type, brackets);
1568                                 field->name = field->alias = token;
1569                                 type = read_token(&token);
1570                         } else {
1571                                 char *new_type;
1572                                 new_type = realloc(field->type,
1573                                                    strlen(field->type) +
1574                                                    strlen(brackets) + 1);
1575                                 if (!new_type) {
1576                                         free(brackets);
1577                                         goto fail;
1578                                 }
1579                                 field->type = new_type;
1580                                 strcat(field->type, brackets);
1581                         }
1582                         free(brackets);
1583                 }
1584
1585                 if (field_is_string(field))
1586                         field->flags |= TEP_FIELD_IS_STRING;
1587                 if (field_is_dynamic(field))
1588                         field->flags |= TEP_FIELD_IS_DYNAMIC;
1589                 if (field_is_long(field))
1590                         field->flags |= TEP_FIELD_IS_LONG;
1591
1592                 if (test_type_token(type, token,  TEP_EVENT_OP, ";"))
1593                         goto fail;
1594                 free_token(token);
1595
1596                 if (read_expected(TEP_EVENT_ITEM, "offset") < 0)
1597                         goto fail_expect;
1598
1599                 if (read_expected(TEP_EVENT_OP, ":") < 0)
1600                         goto fail_expect;
1601
1602                 if (read_expect_type(TEP_EVENT_ITEM, &token))
1603                         goto fail;
1604                 field->offset = strtoul(token, NULL, 0);
1605                 free_token(token);
1606
1607                 if (read_expected(TEP_EVENT_OP, ";") < 0)
1608                         goto fail_expect;
1609
1610                 if (read_expected(TEP_EVENT_ITEM, "size") < 0)
1611                         goto fail_expect;
1612
1613                 if (read_expected(TEP_EVENT_OP, ":") < 0)
1614                         goto fail_expect;
1615
1616                 if (read_expect_type(TEP_EVENT_ITEM, &token))
1617                         goto fail;
1618                 field->size = strtoul(token, NULL, 0);
1619                 free_token(token);
1620
1621                 if (read_expected(TEP_EVENT_OP, ";") < 0)
1622                         goto fail_expect;
1623
1624                 type = read_token(&token);
1625                 if (type != TEP_EVENT_NEWLINE) {
1626                         /* newer versions of the kernel have a "signed" type */
1627                         if (test_type_token(type, token, TEP_EVENT_ITEM, "signed"))
1628                                 goto fail;
1629
1630                         free_token(token);
1631
1632                         if (read_expected(TEP_EVENT_OP, ":") < 0)
1633                                 goto fail_expect;
1634
1635                         if (read_expect_type(TEP_EVENT_ITEM, &token))
1636                                 goto fail;
1637
1638                         if (strtoul(token, NULL, 0))
1639                                 field->flags |= TEP_FIELD_IS_SIGNED;
1640
1641                         free_token(token);
1642                         if (read_expected(TEP_EVENT_OP, ";") < 0)
1643                                 goto fail_expect;
1644
1645                         if (read_expect_type(TEP_EVENT_NEWLINE, &token))
1646                                 goto fail;
1647                 }
1648
1649                 free_token(token);
1650
1651                 if (field->flags & TEP_FIELD_IS_ARRAY) {
1652                         if (field->arraylen)
1653                                 field->elementsize = field->size / field->arraylen;
1654                         else if (field->flags & TEP_FIELD_IS_DYNAMIC)
1655                                 field->elementsize = size_dynamic;
1656                         else if (field->flags & TEP_FIELD_IS_STRING)
1657                                 field->elementsize = 1;
1658                         else if (field->flags & TEP_FIELD_IS_LONG)
1659                                 field->elementsize = event->tep ?
1660                                                      event->tep->long_size :
1661                                                      sizeof(long);
1662                 } else
1663                         field->elementsize = field->size;
1664
1665                 *fields = field;
1666                 fields = &field->next;
1667
1668         } while (1);
1669
1670         return 0;
1671
1672 fail:
1673         free_token(token);
1674 fail_expect:
1675         if (field) {
1676                 free(field->type);
1677                 free(field->name);
1678                 free(field);
1679         }
1680         return -1;
1681 }
1682
1683 static int event_read_format(struct tep_event *event)
1684 {
1685         char *token;
1686         int ret;
1687
1688         if (read_expected_item(TEP_EVENT_ITEM, "format") < 0)
1689                 return -1;
1690
1691         if (read_expected(TEP_EVENT_OP, ":") < 0)
1692                 return -1;
1693
1694         if (read_expect_type(TEP_EVENT_NEWLINE, &token))
1695                 goto fail;
1696         free_token(token);
1697
1698         ret = event_read_fields(event, &event->format.common_fields);
1699         if (ret < 0)
1700                 return ret;
1701         event->format.nr_common = ret;
1702
1703         ret = event_read_fields(event, &event->format.fields);
1704         if (ret < 0)
1705                 return ret;
1706         event->format.nr_fields = ret;
1707
1708         return 0;
1709
1710  fail:
1711         free_token(token);
1712         return -1;
1713 }
1714
1715 static enum tep_event_type
1716 process_arg_token(struct tep_event *event, struct tep_print_arg *arg,
1717                   char **tok, enum tep_event_type type);
1718
1719 static enum tep_event_type
1720 process_arg(struct tep_event *event, struct tep_print_arg *arg, char **tok)
1721 {
1722         enum tep_event_type type;
1723         char *token;
1724
1725         type = read_token(&token);
1726         *tok = token;
1727
1728         return process_arg_token(event, arg, tok, type);
1729 }
1730
1731 static enum tep_event_type
1732 process_op(struct tep_event *event, struct tep_print_arg *arg, char **tok);
1733
1734 /*
1735  * For __print_symbolic() and __print_flags, we need to completely
1736  * evaluate the first argument, which defines what to print next.
1737  */
1738 static enum tep_event_type
1739 process_field_arg(struct tep_event *event, struct tep_print_arg *arg, char **tok)
1740 {
1741         enum tep_event_type type;
1742
1743         type = process_arg(event, arg, tok);
1744
1745         while (type == TEP_EVENT_OP) {
1746                 type = process_op(event, arg, tok);
1747         }
1748
1749         return type;
1750 }
1751
1752 static enum tep_event_type
1753 process_cond(struct tep_event *event, struct tep_print_arg *top, char **tok)
1754 {
1755         struct tep_print_arg *arg, *left, *right;
1756         enum tep_event_type type;
1757         char *token = NULL;
1758
1759         arg = alloc_arg();
1760         left = alloc_arg();
1761         right = alloc_arg();
1762
1763         if (!arg || !left || !right) {
1764                 do_warning_event(event, "%s: not enough memory!", __func__);
1765                 /* arg will be freed at out_free */
1766                 free_arg(left);
1767                 free_arg(right);
1768                 goto out_free;
1769         }
1770
1771         arg->type = TEP_PRINT_OP;
1772         arg->op.left = left;
1773         arg->op.right = right;
1774
1775         *tok = NULL;
1776         type = process_arg(event, left, &token);
1777
1778  again:
1779         if (type == TEP_EVENT_ERROR)
1780                 goto out_free;
1781
1782         /* Handle other operations in the arguments */
1783         if (type == TEP_EVENT_OP && strcmp(token, ":") != 0) {
1784                 type = process_op(event, left, &token);
1785                 goto again;
1786         }
1787
1788         if (test_type_token(type, token, TEP_EVENT_OP, ":"))
1789                 goto out_free;
1790
1791         arg->op.op = token;
1792
1793         type = process_arg(event, right, &token);
1794
1795         top->op.right = arg;
1796
1797         *tok = token;
1798         return type;
1799
1800 out_free:
1801         /* Top may point to itself */
1802         top->op.right = NULL;
1803         free_token(token);
1804         free_arg(arg);
1805         return TEP_EVENT_ERROR;
1806 }
1807
1808 static enum tep_event_type
1809 process_array(struct tep_event *event, struct tep_print_arg *top, char **tok)
1810 {
1811         struct tep_print_arg *arg;
1812         enum tep_event_type type;
1813         char *token = NULL;
1814
1815         arg = alloc_arg();
1816         if (!arg) {
1817                 do_warning_event(event, "%s: not enough memory!", __func__);
1818                 /* '*tok' is set to top->op.op.  No need to free. */
1819                 *tok = NULL;
1820                 return TEP_EVENT_ERROR;
1821         }
1822
1823         *tok = NULL;
1824         type = process_arg(event, arg, &token);
1825         if (test_type_token(type, token, TEP_EVENT_OP, "]"))
1826                 goto out_free;
1827
1828         top->op.right = arg;
1829
1830         free_token(token);
1831         type = read_token_item(&token);
1832         *tok = token;
1833
1834         return type;
1835
1836 out_free:
1837         free_token(token);
1838         free_arg(arg);
1839         return TEP_EVENT_ERROR;
1840 }
1841
1842 static int get_op_prio(char *op)
1843 {
1844         if (!op[1]) {
1845                 switch (op[0]) {
1846                 case '~':
1847                 case '!':
1848                         return 4;
1849                 case '*':
1850                 case '/':
1851                 case '%':
1852                         return 6;
1853                 case '+':
1854                 case '-':
1855                         return 7;
1856                         /* '>>' and '<<' are 8 */
1857                 case '<':
1858                 case '>':
1859                         return 9;
1860                         /* '==' and '!=' are 10 */
1861                 case '&':
1862                         return 11;
1863                 case '^':
1864                         return 12;
1865                 case '|':
1866                         return 13;
1867                 case '?':
1868                         return 16;
1869                 default:
1870                         do_warning("unknown op '%c'", op[0]);
1871                         return -1;
1872                 }
1873         } else {
1874                 if (strcmp(op, "++") == 0 ||
1875                     strcmp(op, "--") == 0) {
1876                         return 3;
1877                 } else if (strcmp(op, ">>") == 0 ||
1878                            strcmp(op, "<<") == 0) {
1879                         return 8;
1880                 } else if (strcmp(op, ">=") == 0 ||
1881                            strcmp(op, "<=") == 0) {
1882                         return 9;
1883                 } else if (strcmp(op, "==") == 0 ||
1884                            strcmp(op, "!=") == 0) {
1885                         return 10;
1886                 } else if (strcmp(op, "&&") == 0) {
1887                         return 14;
1888                 } else if (strcmp(op, "||") == 0) {
1889                         return 15;
1890                 } else {
1891                         do_warning("unknown op '%s'", op);
1892                         return -1;
1893                 }
1894         }
1895 }
1896
1897 static int set_op_prio(struct tep_print_arg *arg)
1898 {
1899
1900         /* single ops are the greatest */
1901         if (!arg->op.left || arg->op.left->type == TEP_PRINT_NULL)
1902                 arg->op.prio = 0;
1903         else
1904                 arg->op.prio = get_op_prio(arg->op.op);
1905
1906         return arg->op.prio;
1907 }
1908
1909 /* Note, *tok does not get freed, but will most likely be saved */
1910 static enum tep_event_type
1911 process_op(struct tep_event *event, struct tep_print_arg *arg, char **tok)
1912 {
1913         struct tep_print_arg *left, *right = NULL;
1914         enum tep_event_type type;
1915         char *token;
1916
1917         /* the op is passed in via tok */
1918         token = *tok;
1919
1920         if (arg->type == TEP_PRINT_OP && !arg->op.left) {
1921                 /* handle single op */
1922                 if (token[1]) {
1923                         do_warning_event(event, "bad op token %s", token);
1924                         goto out_free;
1925                 }
1926                 switch (token[0]) {
1927                 case '~':
1928                 case '!':
1929                 case '+':
1930                 case '-':
1931                         break;
1932                 default:
1933                         do_warning_event(event, "bad op token %s", token);
1934                         goto out_free;
1935
1936                 }
1937
1938                 /* make an empty left */
1939                 left = alloc_arg();
1940                 if (!left)
1941                         goto out_warn_free;
1942
1943                 left->type = TEP_PRINT_NULL;
1944                 arg->op.left = left;
1945
1946                 right = alloc_arg();
1947                 if (!right)
1948                         goto out_warn_free;
1949
1950                 arg->op.right = right;
1951
1952                 /* do not free the token, it belongs to an op */
1953                 *tok = NULL;
1954                 type = process_arg(event, right, tok);
1955
1956         } else if (strcmp(token, "?") == 0) {
1957
1958                 left = alloc_arg();
1959                 if (!left)
1960                         goto out_warn_free;
1961
1962                 /* copy the top arg to the left */
1963                 *left = *arg;
1964
1965                 arg->type = TEP_PRINT_OP;
1966                 arg->op.op = token;
1967                 arg->op.left = left;
1968                 arg->op.prio = 0;
1969
1970                 /* it will set arg->op.right */
1971                 type = process_cond(event, arg, tok);
1972
1973         } else if (strcmp(token, ">>") == 0 ||
1974                    strcmp(token, "<<") == 0 ||
1975                    strcmp(token, "&") == 0 ||
1976                    strcmp(token, "|") == 0 ||
1977                    strcmp(token, "&&") == 0 ||
1978                    strcmp(token, "||") == 0 ||
1979                    strcmp(token, "-") == 0 ||
1980                    strcmp(token, "+") == 0 ||
1981                    strcmp(token, "*") == 0 ||
1982                    strcmp(token, "^") == 0 ||
1983                    strcmp(token, "/") == 0 ||
1984                    strcmp(token, "%") == 0 ||
1985                    strcmp(token, "<") == 0 ||
1986                    strcmp(token, ">") == 0 ||
1987                    strcmp(token, "<=") == 0 ||
1988                    strcmp(token, ">=") == 0 ||
1989                    strcmp(token, "==") == 0 ||
1990                    strcmp(token, "!=") == 0) {
1991
1992                 left = alloc_arg();
1993                 if (!left)
1994                         goto out_warn_free;
1995
1996                 /* copy the top arg to the left */
1997                 *left = *arg;
1998
1999                 arg->type = TEP_PRINT_OP;
2000                 arg->op.op = token;
2001                 arg->op.left = left;
2002                 arg->op.right = NULL;
2003
2004                 if (set_op_prio(arg) == -1) {
2005                         event->flags |= TEP_EVENT_FL_FAILED;
2006                         /* arg->op.op (= token) will be freed at out_free */
2007                         arg->op.op = NULL;
2008                         goto out_free;
2009                 }
2010
2011                 type = read_token_item(&token);
2012                 *tok = token;
2013
2014                 /* could just be a type pointer */
2015                 if ((strcmp(arg->op.op, "*") == 0) &&
2016                     type == TEP_EVENT_DELIM && (strcmp(token, ")") == 0)) {
2017                         char *new_atom;
2018
2019                         if (left->type != TEP_PRINT_ATOM) {
2020                                 do_warning_event(event, "bad pointer type");
2021                                 goto out_free;
2022                         }
2023                         new_atom = realloc(left->atom.atom,
2024                                             strlen(left->atom.atom) + 3);
2025                         if (!new_atom)
2026                                 goto out_warn_free;
2027
2028                         left->atom.atom = new_atom;
2029                         strcat(left->atom.atom, " *");
2030                         free(arg->op.op);
2031                         *arg = *left;
2032                         free(left);
2033
2034                         return type;
2035                 }
2036
2037                 right = alloc_arg();
2038                 if (!right)
2039                         goto out_warn_free;
2040
2041                 type = process_arg_token(event, right, tok, type);
2042                 if (type == TEP_EVENT_ERROR) {
2043                         free_arg(right);
2044                         /* token was freed in process_arg_token() via *tok */
2045                         token = NULL;
2046                         goto out_free;
2047                 }
2048
2049                 if (right->type == TEP_PRINT_OP &&
2050                     get_op_prio(arg->op.op) < get_op_prio(right->op.op)) {
2051                         struct tep_print_arg tmp;
2052
2053                         /* rotate ops according to the priority */
2054                         arg->op.right = right->op.left;
2055
2056                         tmp = *arg;
2057                         *arg = *right;
2058                         *right = tmp;
2059
2060                         arg->op.left = right;
2061                 } else {
2062                         arg->op.right = right;
2063                 }
2064
2065         } else if (strcmp(token, "[") == 0) {
2066
2067                 left = alloc_arg();
2068                 if (!left)
2069                         goto out_warn_free;
2070
2071                 *left = *arg;
2072
2073                 arg->type = TEP_PRINT_OP;
2074                 arg->op.op = token;
2075                 arg->op.left = left;
2076
2077                 arg->op.prio = 0;
2078
2079                 /* it will set arg->op.right */
2080                 type = process_array(event, arg, tok);
2081
2082         } else {
2083                 do_warning_event(event, "unknown op '%s'", token);
2084                 event->flags |= TEP_EVENT_FL_FAILED;
2085                 /* the arg is now the left side */
2086                 goto out_free;
2087         }
2088
2089         if (type == TEP_EVENT_OP && strcmp(*tok, ":") != 0) {
2090                 int prio;
2091
2092                 /* higher prios need to be closer to the root */
2093                 prio = get_op_prio(*tok);
2094
2095                 if (prio > arg->op.prio)
2096                         return process_op(event, arg, tok);
2097
2098                 return process_op(event, right, tok);
2099         }
2100
2101         return type;
2102
2103 out_warn_free:
2104         do_warning_event(event, "%s: not enough memory!", __func__);
2105 out_free:
2106         free_token(token);
2107         *tok = NULL;
2108         return TEP_EVENT_ERROR;
2109 }
2110
2111 static enum tep_event_type
2112 process_entry(struct tep_event *event __maybe_unused, struct tep_print_arg *arg,
2113               char **tok)
2114 {
2115         enum tep_event_type type;
2116         char *field;
2117         char *token;
2118
2119         if (read_expected(TEP_EVENT_OP, "->") < 0)
2120                 goto out_err;
2121
2122         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
2123                 goto out_free;
2124         field = token;
2125
2126         arg->type = TEP_PRINT_FIELD;
2127         arg->field.name = field;
2128
2129         if (is_flag_field) {
2130                 arg->field.field = tep_find_any_field(event, arg->field.name);
2131                 arg->field.field->flags |= TEP_FIELD_IS_FLAG;
2132                 is_flag_field = 0;
2133         } else if (is_symbolic_field) {
2134                 arg->field.field = tep_find_any_field(event, arg->field.name);
2135                 arg->field.field->flags |= TEP_FIELD_IS_SYMBOLIC;
2136                 is_symbolic_field = 0;
2137         }
2138
2139         type = read_token(&token);
2140         *tok = token;
2141
2142         return type;
2143
2144  out_free:
2145         free_token(token);
2146  out_err:
2147         *tok = NULL;
2148         return TEP_EVENT_ERROR;
2149 }
2150
2151 static int alloc_and_process_delim(struct tep_event *event, char *next_token,
2152                                    struct tep_print_arg **print_arg)
2153 {
2154         struct tep_print_arg *field;
2155         enum tep_event_type type;
2156         char *token;
2157         int ret = 0;
2158
2159         field = alloc_arg();
2160         if (!field) {
2161                 do_warning_event(event, "%s: not enough memory!", __func__);
2162                 errno = ENOMEM;
2163                 return -1;
2164         }
2165
2166         type = process_arg(event, field, &token);
2167
2168         if (test_type_token(type, token, TEP_EVENT_DELIM, next_token)) {
2169                 errno = EINVAL;
2170                 ret = -1;
2171                 free_arg(field);
2172                 goto out_free_token;
2173         }
2174
2175         *print_arg = field;
2176
2177 out_free_token:
2178         free_token(token);
2179
2180         return ret;
2181 }
2182
2183 static char *arg_eval (struct tep_print_arg *arg);
2184
2185 static unsigned long long
2186 eval_type_str(unsigned long long val, const char *type, int pointer)
2187 {
2188         int sign = 0;
2189         char *ref;
2190         int len;
2191
2192         len = strlen(type);
2193
2194         if (pointer) {
2195
2196                 if (type[len-1] != '*') {
2197                         do_warning("pointer expected with non pointer type");
2198                         return val;
2199                 }
2200
2201                 ref = malloc(len);
2202                 if (!ref) {
2203                         do_warning("%s: not enough memory!", __func__);
2204                         return val;
2205                 }
2206                 memcpy(ref, type, len);
2207
2208                 /* chop off the " *" */
2209                 ref[len - 2] = 0;
2210
2211                 val = eval_type_str(val, ref, 0);
2212                 free(ref);
2213                 return val;
2214         }
2215
2216         /* check if this is a pointer */
2217         if (type[len - 1] == '*')
2218                 return val;
2219
2220         /* Try to figure out the arg size*/
2221         if (strncmp(type, "struct", 6) == 0)
2222                 /* all bets off */
2223                 return val;
2224
2225         if (strcmp(type, "u8") == 0)
2226                 return val & 0xff;
2227
2228         if (strcmp(type, "u16") == 0)
2229                 return val & 0xffff;
2230
2231         if (strcmp(type, "u32") == 0)
2232                 return val & 0xffffffff;
2233
2234         if (strcmp(type, "u64") == 0 ||
2235             strcmp(type, "s64") == 0)
2236                 return val;
2237
2238         if (strcmp(type, "s8") == 0)
2239                 return (unsigned long long)(char)val & 0xff;
2240
2241         if (strcmp(type, "s16") == 0)
2242                 return (unsigned long long)(short)val & 0xffff;
2243
2244         if (strcmp(type, "s32") == 0)
2245                 return (unsigned long long)(int)val & 0xffffffff;
2246
2247         if (strncmp(type, "unsigned ", 9) == 0) {
2248                 sign = 0;
2249                 type += 9;
2250         }
2251
2252         if (strcmp(type, "char") == 0) {
2253                 if (sign)
2254                         return (unsigned long long)(char)val & 0xff;
2255                 else
2256                         return val & 0xff;
2257         }
2258
2259         if (strcmp(type, "short") == 0) {
2260                 if (sign)
2261                         return (unsigned long long)(short)val & 0xffff;
2262                 else
2263                         return val & 0xffff;
2264         }
2265
2266         if (strcmp(type, "int") == 0) {
2267                 if (sign)
2268                         return (unsigned long long)(int)val & 0xffffffff;
2269                 else
2270                         return val & 0xffffffff;
2271         }
2272
2273         return val;
2274 }
2275
2276 /*
2277  * Try to figure out the type.
2278  */
2279 static unsigned long long
2280 eval_type(unsigned long long val, struct tep_print_arg *arg, int pointer)
2281 {
2282         if (arg->type != TEP_PRINT_TYPE) {
2283                 do_warning("expected type argument");
2284                 return 0;
2285         }
2286
2287         return eval_type_str(val, arg->typecast.type, pointer);
2288 }
2289
2290 static int arg_num_eval(struct tep_print_arg *arg, long long *val)
2291 {
2292         long long left, right;
2293         int ret = 1;
2294
2295         switch (arg->type) {
2296         case TEP_PRINT_ATOM:
2297                 *val = strtoll(arg->atom.atom, NULL, 0);
2298                 break;
2299         case TEP_PRINT_TYPE:
2300                 ret = arg_num_eval(arg->typecast.item, val);
2301                 if (!ret)
2302                         break;
2303                 *val = eval_type(*val, arg, 0);
2304                 break;
2305         case TEP_PRINT_OP:
2306                 switch (arg->op.op[0]) {
2307                 case '|':
2308                         ret = arg_num_eval(arg->op.left, &left);
2309                         if (!ret)
2310                                 break;
2311                         ret = arg_num_eval(arg->op.right, &right);
2312                         if (!ret)
2313                                 break;
2314                         if (arg->op.op[1])
2315                                 *val = left || right;
2316                         else
2317                                 *val = left | right;
2318                         break;
2319                 case '&':
2320                         ret = arg_num_eval(arg->op.left, &left);
2321                         if (!ret)
2322                                 break;
2323                         ret = arg_num_eval(arg->op.right, &right);
2324                         if (!ret)
2325                                 break;
2326                         if (arg->op.op[1])
2327                                 *val = left && right;
2328                         else
2329                                 *val = left & right;
2330                         break;
2331                 case '<':
2332                         ret = arg_num_eval(arg->op.left, &left);
2333                         if (!ret)
2334                                 break;
2335                         ret = arg_num_eval(arg->op.right, &right);
2336                         if (!ret)
2337                                 break;
2338                         switch (arg->op.op[1]) {
2339                         case 0:
2340                                 *val = left < right;
2341                                 break;
2342                         case '<':
2343                                 *val = left << right;
2344                                 break;
2345                         case '=':
2346                                 *val = left <= right;
2347                                 break;
2348                         default:
2349                                 do_warning("unknown op '%s'", arg->op.op);
2350                                 ret = 0;
2351                         }
2352                         break;
2353                 case '>':
2354                         ret = arg_num_eval(arg->op.left, &left);
2355                         if (!ret)
2356                                 break;
2357                         ret = arg_num_eval(arg->op.right, &right);
2358                         if (!ret)
2359                                 break;
2360                         switch (arg->op.op[1]) {
2361                         case 0:
2362                                 *val = left > right;
2363                                 break;
2364                         case '>':
2365                                 *val = left >> right;
2366                                 break;
2367                         case '=':
2368                                 *val = left >= right;
2369                                 break;
2370                         default:
2371                                 do_warning("unknown op '%s'", arg->op.op);
2372                                 ret = 0;
2373                         }
2374                         break;
2375                 case '=':
2376                         ret = arg_num_eval(arg->op.left, &left);
2377                         if (!ret)
2378                                 break;
2379                         ret = arg_num_eval(arg->op.right, &right);
2380                         if (!ret)
2381                                 break;
2382
2383                         if (arg->op.op[1] != '=') {
2384                                 do_warning("unknown op '%s'", arg->op.op);
2385                                 ret = 0;
2386                         } else
2387                                 *val = left == right;
2388                         break;
2389                 case '!':
2390                         ret = arg_num_eval(arg->op.left, &left);
2391                         if (!ret)
2392                                 break;
2393                         ret = arg_num_eval(arg->op.right, &right);
2394                         if (!ret)
2395                                 break;
2396
2397                         switch (arg->op.op[1]) {
2398                         case '=':
2399                                 *val = left != right;
2400                                 break;
2401                         default:
2402                                 do_warning("unknown op '%s'", arg->op.op);
2403                                 ret = 0;
2404                         }
2405                         break;
2406                 case '-':
2407                         /* check for negative */
2408                         if (arg->op.left->type == TEP_PRINT_NULL)
2409                                 left = 0;
2410                         else
2411                                 ret = arg_num_eval(arg->op.left, &left);
2412                         if (!ret)
2413                                 break;
2414                         ret = arg_num_eval(arg->op.right, &right);
2415                         if (!ret)
2416                                 break;
2417                         *val = left - right;
2418                         break;
2419                 case '+':
2420                         if (arg->op.left->type == TEP_PRINT_NULL)
2421                                 left = 0;
2422                         else
2423                                 ret = arg_num_eval(arg->op.left, &left);
2424                         if (!ret)
2425                                 break;
2426                         ret = arg_num_eval(arg->op.right, &right);
2427                         if (!ret)
2428                                 break;
2429                         *val = left + right;
2430                         break;
2431                 case '~':
2432                         ret = arg_num_eval(arg->op.right, &right);
2433                         if (!ret)
2434                                 break;
2435                         *val = ~right;
2436                         break;
2437                 default:
2438                         do_warning("unknown op '%s'", arg->op.op);
2439                         ret = 0;
2440                 }
2441                 break;
2442
2443         case TEP_PRINT_NULL:
2444         case TEP_PRINT_FIELD ... TEP_PRINT_SYMBOL:
2445         case TEP_PRINT_STRING:
2446         case TEP_PRINT_BSTRING:
2447         case TEP_PRINT_BITMASK:
2448         default:
2449                 do_warning("invalid eval type %d", arg->type);
2450                 ret = 0;
2451
2452         }
2453         return ret;
2454 }
2455
2456 static char *arg_eval (struct tep_print_arg *arg)
2457 {
2458         long long val;
2459         static char buf[24];
2460
2461         switch (arg->type) {
2462         case TEP_PRINT_ATOM:
2463                 return arg->atom.atom;
2464         case TEP_PRINT_TYPE:
2465                 return arg_eval(arg->typecast.item);
2466         case TEP_PRINT_OP:
2467                 if (!arg_num_eval(arg, &val))
2468                         break;
2469                 sprintf(buf, "%lld", val);
2470                 return buf;
2471
2472         case TEP_PRINT_NULL:
2473         case TEP_PRINT_FIELD ... TEP_PRINT_SYMBOL:
2474         case TEP_PRINT_STRING:
2475         case TEP_PRINT_BSTRING:
2476         case TEP_PRINT_BITMASK:
2477         default:
2478                 do_warning("invalid eval type %d", arg->type);
2479                 break;
2480         }
2481
2482         return NULL;
2483 }
2484
2485 static enum tep_event_type
2486 process_fields(struct tep_event *event, struct tep_print_flag_sym **list, char **tok)
2487 {
2488         enum tep_event_type type;
2489         struct tep_print_arg *arg = NULL;
2490         struct tep_print_flag_sym *field;
2491         char *token = *tok;
2492         char *value;
2493
2494         do {
2495                 free_token(token);
2496                 type = read_token_item(&token);
2497                 if (test_type_token(type, token, TEP_EVENT_OP, "{"))
2498                         break;
2499
2500                 arg = alloc_arg();
2501                 if (!arg)
2502                         goto out_free;
2503
2504                 free_token(token);
2505                 type = process_arg(event, arg, &token);
2506
2507                 if (type == TEP_EVENT_OP)
2508                         type = process_op(event, arg, &token);
2509
2510                 if (type == TEP_EVENT_ERROR)
2511                         goto out_free;
2512
2513                 if (test_type_token(type, token, TEP_EVENT_DELIM, ","))
2514                         goto out_free;
2515
2516                 field = calloc(1, sizeof(*field));
2517                 if (!field)
2518                         goto out_free;
2519
2520                 value = arg_eval(arg);
2521                 if (value == NULL)
2522                         goto out_free_field;
2523                 field->value = strdup(value);
2524                 if (field->value == NULL)
2525                         goto out_free_field;
2526
2527                 free_arg(arg);
2528                 arg = alloc_arg();
2529                 if (!arg)
2530                         goto out_free;
2531
2532                 free_token(token);
2533                 type = process_arg(event, arg, &token);
2534                 if (test_type_token(type, token, TEP_EVENT_OP, "}"))
2535                         goto out_free_field;
2536
2537                 value = arg_eval(arg);
2538                 if (value == NULL)
2539                         goto out_free_field;
2540                 field->str = strdup(value);
2541                 if (field->str == NULL)
2542                         goto out_free_field;
2543                 free_arg(arg);
2544                 arg = NULL;
2545
2546                 *list = field;
2547                 list = &field->next;
2548
2549                 free_token(token);
2550                 type = read_token_item(&token);
2551         } while (type == TEP_EVENT_DELIM && strcmp(token, ",") == 0);
2552
2553         *tok = token;
2554         return type;
2555
2556 out_free_field:
2557         free_flag_sym(field);
2558 out_free:
2559         free_arg(arg);
2560         free_token(token);
2561         *tok = NULL;
2562
2563         return TEP_EVENT_ERROR;
2564 }
2565
2566 static enum tep_event_type
2567 process_flags(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2568 {
2569         struct tep_print_arg *field;
2570         enum tep_event_type type;
2571         char *token = NULL;
2572
2573         memset(arg, 0, sizeof(*arg));
2574         arg->type = TEP_PRINT_FLAGS;
2575
2576         field = alloc_arg();
2577         if (!field) {
2578                 do_warning_event(event, "%s: not enough memory!", __func__);
2579                 goto out_free;
2580         }
2581
2582         type = process_field_arg(event, field, &token);
2583
2584         /* Handle operations in the first argument */
2585         while (type == TEP_EVENT_OP)
2586                 type = process_op(event, field, &token);
2587
2588         if (test_type_token(type, token, TEP_EVENT_DELIM, ","))
2589                 goto out_free_field;
2590         free_token(token);
2591
2592         arg->flags.field = field;
2593
2594         type = read_token_item(&token);
2595         if (event_item_type(type)) {
2596                 arg->flags.delim = token;
2597                 type = read_token_item(&token);
2598         }
2599
2600         if (test_type_token(type, token, TEP_EVENT_DELIM, ","))
2601                 goto out_free;
2602
2603         type = process_fields(event, &arg->flags.flags, &token);
2604         if (test_type_token(type, token, TEP_EVENT_DELIM, ")"))
2605                 goto out_free;
2606
2607         free_token(token);
2608         type = read_token_item(tok);
2609         return type;
2610
2611 out_free_field:
2612         free_arg(field);
2613 out_free:
2614         free_token(token);
2615         *tok = NULL;
2616         return TEP_EVENT_ERROR;
2617 }
2618
2619 static enum tep_event_type
2620 process_symbols(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2621 {
2622         struct tep_print_arg *field;
2623         enum tep_event_type type;
2624         char *token = NULL;
2625
2626         memset(arg, 0, sizeof(*arg));
2627         arg->type = TEP_PRINT_SYMBOL;
2628
2629         field = alloc_arg();
2630         if (!field) {
2631                 do_warning_event(event, "%s: not enough memory!", __func__);
2632                 goto out_free;
2633         }
2634
2635         type = process_field_arg(event, field, &token);
2636
2637         if (test_type_token(type, token, TEP_EVENT_DELIM, ","))
2638                 goto out_free_field;
2639
2640         arg->symbol.field = field;
2641
2642         type = process_fields(event, &arg->symbol.symbols, &token);
2643         if (test_type_token(type, token, TEP_EVENT_DELIM, ")"))
2644                 goto out_free;
2645
2646         free_token(token);
2647         type = read_token_item(tok);
2648         return type;
2649
2650 out_free_field:
2651         free_arg(field);
2652 out_free:
2653         free_token(token);
2654         *tok = NULL;
2655         return TEP_EVENT_ERROR;
2656 }
2657
2658 static enum tep_event_type
2659 process_hex_common(struct tep_event *event, struct tep_print_arg *arg,
2660                    char **tok, enum tep_print_arg_type type)
2661 {
2662         memset(arg, 0, sizeof(*arg));
2663         arg->type = type;
2664
2665         if (alloc_and_process_delim(event, ",", &arg->hex.field))
2666                 goto out;
2667
2668         if (alloc_and_process_delim(event, ")", &arg->hex.size))
2669                 goto free_field;
2670
2671         return read_token_item(tok);
2672
2673 free_field:
2674         free_arg(arg->hex.field);
2675         arg->hex.field = NULL;
2676 out:
2677         *tok = NULL;
2678         return TEP_EVENT_ERROR;
2679 }
2680
2681 static enum tep_event_type
2682 process_hex(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2683 {
2684         return process_hex_common(event, arg, tok, TEP_PRINT_HEX);
2685 }
2686
2687 static enum tep_event_type
2688 process_hex_str(struct tep_event *event, struct tep_print_arg *arg,
2689                 char **tok)
2690 {
2691         return process_hex_common(event, arg, tok, TEP_PRINT_HEX_STR);
2692 }
2693
2694 static enum tep_event_type
2695 process_int_array(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2696 {
2697         memset(arg, 0, sizeof(*arg));
2698         arg->type = TEP_PRINT_INT_ARRAY;
2699
2700         if (alloc_and_process_delim(event, ",", &arg->int_array.field))
2701                 goto out;
2702
2703         if (alloc_and_process_delim(event, ",", &arg->int_array.count))
2704                 goto free_field;
2705
2706         if (alloc_and_process_delim(event, ")", &arg->int_array.el_size))
2707                 goto free_size;
2708
2709         return read_token_item(tok);
2710
2711 free_size:
2712         free_arg(arg->int_array.count);
2713         arg->int_array.count = NULL;
2714 free_field:
2715         free_arg(arg->int_array.field);
2716         arg->int_array.field = NULL;
2717 out:
2718         *tok = NULL;
2719         return TEP_EVENT_ERROR;
2720 }
2721
2722 static enum tep_event_type
2723 process_dynamic_array(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2724 {
2725         struct tep_format_field *field;
2726         enum tep_event_type type;
2727         char *token;
2728
2729         memset(arg, 0, sizeof(*arg));
2730         arg->type = TEP_PRINT_DYNAMIC_ARRAY;
2731
2732         /*
2733          * The item within the parenthesis is another field that holds
2734          * the index into where the array starts.
2735          */
2736         type = read_token(&token);
2737         *tok = token;
2738         if (type != TEP_EVENT_ITEM)
2739                 goto out_free;
2740
2741         /* Find the field */
2742
2743         field = tep_find_field(event, token);
2744         if (!field)
2745                 goto out_free;
2746
2747         arg->dynarray.field = field;
2748         arg->dynarray.index = 0;
2749
2750         if (read_expected(TEP_EVENT_DELIM, ")") < 0)
2751                 goto out_free;
2752
2753         free_token(token);
2754         type = read_token_item(&token);
2755         *tok = token;
2756         if (type != TEP_EVENT_OP || strcmp(token, "[") != 0)
2757                 return type;
2758
2759         free_token(token);
2760         arg = alloc_arg();
2761         if (!arg) {
2762                 do_warning_event(event, "%s: not enough memory!", __func__);
2763                 *tok = NULL;
2764                 return TEP_EVENT_ERROR;
2765         }
2766
2767         type = process_arg(event, arg, &token);
2768         if (type == TEP_EVENT_ERROR)
2769                 goto out_free_arg;
2770
2771         if (!test_type_token(type, token, TEP_EVENT_OP, "]"))
2772                 goto out_free_arg;
2773
2774         free_token(token);
2775         type = read_token_item(tok);
2776         return type;
2777
2778  out_free_arg:
2779         free_arg(arg);
2780  out_free:
2781         free_token(token);
2782         *tok = NULL;
2783         return TEP_EVENT_ERROR;
2784 }
2785
2786 static enum tep_event_type
2787 process_dynamic_array_len(struct tep_event *event, struct tep_print_arg *arg,
2788                           char **tok)
2789 {
2790         struct tep_format_field *field;
2791         enum tep_event_type type;
2792         char *token;
2793
2794         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
2795                 goto out_free;
2796
2797         arg->type = TEP_PRINT_DYNAMIC_ARRAY_LEN;
2798
2799         /* Find the field */
2800         field = tep_find_field(event, token);
2801         if (!field)
2802                 goto out_free;
2803
2804         arg->dynarray.field = field;
2805         arg->dynarray.index = 0;
2806
2807         if (read_expected(TEP_EVENT_DELIM, ")") < 0)
2808                 goto out_err;
2809
2810         type = read_token(&token);
2811         *tok = token;
2812
2813         return type;
2814
2815  out_free:
2816         free_token(token);
2817  out_err:
2818         *tok = NULL;
2819         return TEP_EVENT_ERROR;
2820 }
2821
2822 static enum tep_event_type
2823 process_paren(struct tep_event *event, struct tep_print_arg *arg, char **tok)
2824 {
2825         struct tep_print_arg *item_arg;
2826         enum tep_event_type type;
2827         char *token;
2828
2829         type = process_arg(event, arg, &token);
2830
2831         if (type == TEP_EVENT_ERROR)
2832                 goto out_free;
2833
2834         if (type == TEP_EVENT_OP)
2835                 type = process_op(event, arg, &token);
2836
2837         if (type == TEP_EVENT_ERROR)
2838                 goto out_free;
2839
2840         if (test_type_token(type, token, TEP_EVENT_DELIM, ")"))
2841                 goto out_free;
2842
2843         free_token(token);
2844         type = read_token_item(&token);
2845
2846         /*
2847          * If the next token is an item or another open paren, then
2848          * this was a typecast.
2849          */
2850         if (event_item_type(type) ||
2851             (type == TEP_EVENT_DELIM && strcmp(token, "(") == 0)) {
2852
2853                 /* make this a typecast and contine */
2854
2855                 /* prevous must be an atom */
2856                 if (arg->type != TEP_PRINT_ATOM) {
2857                         do_warning_event(event, "previous needed to be TEP_PRINT_ATOM");
2858                         goto out_free;
2859                 }
2860
2861                 item_arg = alloc_arg();
2862                 if (!item_arg) {
2863                         do_warning_event(event, "%s: not enough memory!",
2864                                          __func__);
2865                         goto out_free;
2866                 }
2867
2868                 arg->type = TEP_PRINT_TYPE;
2869                 arg->typecast.type = arg->atom.atom;
2870                 arg->typecast.item = item_arg;
2871                 type = process_arg_token(event, item_arg, &token, type);
2872
2873         }
2874
2875         *tok = token;
2876         return type;
2877
2878  out_free:
2879         free_token(token);
2880         *tok = NULL;
2881         return TEP_EVENT_ERROR;
2882 }
2883
2884
2885 static enum tep_event_type
2886 process_str(struct tep_event *event __maybe_unused, struct tep_print_arg *arg,
2887             char **tok)
2888 {
2889         enum tep_event_type type;
2890         char *token;
2891
2892         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
2893                 goto out_free;
2894
2895         arg->type = TEP_PRINT_STRING;
2896         arg->string.string = token;
2897         arg->string.offset = -1;
2898
2899         if (read_expected(TEP_EVENT_DELIM, ")") < 0)
2900                 goto out_err;
2901
2902         type = read_token(&token);
2903         *tok = token;
2904
2905         return type;
2906
2907  out_free:
2908         free_token(token);
2909  out_err:
2910         *tok = NULL;
2911         return TEP_EVENT_ERROR;
2912 }
2913
2914 static enum tep_event_type
2915 process_bitmask(struct tep_event *event __maybe_unused, struct tep_print_arg *arg,
2916                 char **tok)
2917 {
2918         enum tep_event_type type;
2919         char *token;
2920
2921         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
2922                 goto out_free;
2923
2924         arg->type = TEP_PRINT_BITMASK;
2925         arg->bitmask.bitmask = token;
2926         arg->bitmask.offset = -1;
2927
2928         if (read_expected(TEP_EVENT_DELIM, ")") < 0)
2929                 goto out_err;
2930
2931         type = read_token(&token);
2932         *tok = token;
2933
2934         return type;
2935
2936  out_free:
2937         free_token(token);
2938  out_err:
2939         *tok = NULL;
2940         return TEP_EVENT_ERROR;
2941 }
2942
2943 static struct tep_function_handler *
2944 find_func_handler(struct tep_handle *tep, char *func_name)
2945 {
2946         struct tep_function_handler *func;
2947
2948         if (!tep)
2949                 return NULL;
2950
2951         for (func = tep->func_handlers; func; func = func->next) {
2952                 if (strcmp(func->name, func_name) == 0)
2953                         break;
2954         }
2955
2956         return func;
2957 }
2958
2959 static void remove_func_handler(struct tep_handle *tep, char *func_name)
2960 {
2961         struct tep_function_handler *func;
2962         struct tep_function_handler **next;
2963
2964         next = &tep->func_handlers;
2965         while ((func = *next)) {
2966                 if (strcmp(func->name, func_name) == 0) {
2967                         *next = func->next;
2968                         free_func_handle(func);
2969                         break;
2970                 }
2971                 next = &func->next;
2972         }
2973 }
2974
2975 static enum tep_event_type
2976 process_func_handler(struct tep_event *event, struct tep_function_handler *func,
2977                      struct tep_print_arg *arg, char **tok)
2978 {
2979         struct tep_print_arg **next_arg;
2980         struct tep_print_arg *farg;
2981         enum tep_event_type type;
2982         char *token;
2983         int i;
2984
2985         arg->type = TEP_PRINT_FUNC;
2986         arg->func.func = func;
2987
2988         *tok = NULL;
2989
2990         next_arg = &(arg->func.args);
2991         for (i = 0; i < func->nr_args; i++) {
2992                 farg = alloc_arg();
2993                 if (!farg) {
2994                         do_warning_event(event, "%s: not enough memory!",
2995                                          __func__);
2996                         return TEP_EVENT_ERROR;
2997                 }
2998
2999                 type = process_arg(event, farg, &token);
3000                 if (i < (func->nr_args - 1)) {
3001                         if (type != TEP_EVENT_DELIM || strcmp(token, ",") != 0) {
3002                                 do_warning_event(event,
3003                                         "Error: function '%s()' expects %d arguments but event %s only uses %d",
3004                                         func->name, func->nr_args,
3005                                         event->name, i + 1);
3006                                 goto err;
3007                         }
3008                 } else {
3009                         if (type != TEP_EVENT_DELIM || strcmp(token, ")") != 0) {
3010                                 do_warning_event(event,
3011                                         "Error: function '%s()' only expects %d arguments but event %s has more",
3012                                         func->name, func->nr_args, event->name);
3013                                 goto err;
3014                         }
3015                 }
3016
3017                 *next_arg = farg;
3018                 next_arg = &(farg->next);
3019                 free_token(token);
3020         }
3021
3022         type = read_token(&token);
3023         *tok = token;
3024
3025         return type;
3026
3027 err:
3028         free_arg(farg);
3029         free_token(token);
3030         return TEP_EVENT_ERROR;
3031 }
3032
3033 static enum tep_event_type
3034 process_function(struct tep_event *event, struct tep_print_arg *arg,
3035                  char *token, char **tok)
3036 {
3037         struct tep_function_handler *func;
3038
3039         if (strcmp(token, "__print_flags") == 0) {
3040                 free_token(token);
3041                 is_flag_field = 1;
3042                 return process_flags(event, arg, tok);
3043         }
3044         if (strcmp(token, "__print_symbolic") == 0) {
3045                 free_token(token);
3046                 is_symbolic_field = 1;
3047                 return process_symbols(event, arg, tok);
3048         }
3049         if (strcmp(token, "__print_hex") == 0) {
3050                 free_token(token);
3051                 return process_hex(event, arg, tok);
3052         }
3053         if (strcmp(token, "__print_hex_str") == 0) {
3054                 free_token(token);
3055                 return process_hex_str(event, arg, tok);
3056         }
3057         if (strcmp(token, "__print_array") == 0) {
3058                 free_token(token);
3059                 return process_int_array(event, arg, tok);
3060         }
3061         if (strcmp(token, "__get_str") == 0) {
3062                 free_token(token);
3063                 return process_str(event, arg, tok);
3064         }
3065         if (strcmp(token, "__get_bitmask") == 0) {
3066                 free_token(token);
3067                 return process_bitmask(event, arg, tok);
3068         }
3069         if (strcmp(token, "__get_dynamic_array") == 0) {
3070                 free_token(token);
3071                 return process_dynamic_array(event, arg, tok);
3072         }
3073         if (strcmp(token, "__get_dynamic_array_len") == 0) {
3074                 free_token(token);
3075                 return process_dynamic_array_len(event, arg, tok);
3076         }
3077
3078         func = find_func_handler(event->tep, token);
3079         if (func) {
3080                 free_token(token);
3081                 return process_func_handler(event, func, arg, tok);
3082         }
3083
3084         do_warning_event(event, "function %s not defined", token);
3085         free_token(token);
3086         return TEP_EVENT_ERROR;
3087 }
3088
3089 static enum tep_event_type
3090 process_arg_token(struct tep_event *event, struct tep_print_arg *arg,
3091                   char **tok, enum tep_event_type type)
3092 {
3093         char *token;
3094         char *atom;
3095
3096         token = *tok;
3097
3098         switch (type) {
3099         case TEP_EVENT_ITEM:
3100                 if (strcmp(token, "REC") == 0) {
3101                         free_token(token);
3102                         type = process_entry(event, arg, &token);
3103                         break;
3104                 }
3105                 atom = token;
3106                 /* test the next token */
3107                 type = read_token_item(&token);
3108
3109                 /*
3110                  * If the next token is a parenthesis, then this
3111                  * is a function.
3112                  */
3113                 if (type == TEP_EVENT_DELIM && strcmp(token, "(") == 0) {
3114                         free_token(token);
3115                         token = NULL;
3116                         /* this will free atom. */
3117                         type = process_function(event, arg, atom, &token);
3118                         break;
3119                 }
3120                 /* atoms can be more than one token long */
3121                 while (type == TEP_EVENT_ITEM) {
3122                         char *new_atom;
3123                         new_atom = realloc(atom,
3124                                            strlen(atom) + strlen(token) + 2);
3125                         if (!new_atom) {
3126                                 free(atom);
3127                                 *tok = NULL;
3128                                 free_token(token);
3129                                 return TEP_EVENT_ERROR;
3130                         }
3131                         atom = new_atom;
3132                         strcat(atom, " ");
3133                         strcat(atom, token);
3134                         free_token(token);
3135                         type = read_token_item(&token);
3136                 }
3137
3138                 arg->type = TEP_PRINT_ATOM;
3139                 arg->atom.atom = atom;
3140                 break;
3141
3142         case TEP_EVENT_DQUOTE:
3143         case TEP_EVENT_SQUOTE:
3144                 arg->type = TEP_PRINT_ATOM;
3145                 arg->atom.atom = token;
3146                 type = read_token_item(&token);
3147                 break;
3148         case TEP_EVENT_DELIM:
3149                 if (strcmp(token, "(") == 0) {
3150                         free_token(token);
3151                         type = process_paren(event, arg, &token);
3152                         break;
3153                 }
3154         case TEP_EVENT_OP:
3155                 /* handle single ops */
3156                 arg->type = TEP_PRINT_OP;
3157                 arg->op.op = token;
3158                 arg->op.left = NULL;
3159                 type = process_op(event, arg, &token);
3160
3161                 /* On error, the op is freed */
3162                 if (type == TEP_EVENT_ERROR)
3163                         arg->op.op = NULL;
3164
3165                 /* return error type if errored */
3166                 break;
3167
3168         case TEP_EVENT_ERROR ... TEP_EVENT_NEWLINE:
3169         default:
3170                 do_warning_event(event, "unexpected type %d", type);
3171                 return TEP_EVENT_ERROR;
3172         }
3173         *tok = token;
3174
3175         return type;
3176 }
3177
3178 static int event_read_print_args(struct tep_event *event, struct tep_print_arg **list)
3179 {
3180         enum tep_event_type type = TEP_EVENT_ERROR;
3181         struct tep_print_arg *arg;
3182         char *token;
3183         int args = 0;
3184
3185         do {
3186                 if (type == TEP_EVENT_NEWLINE) {
3187                         type = read_token_item(&token);
3188                         continue;
3189                 }
3190
3191                 arg = alloc_arg();
3192                 if (!arg) {
3193                         do_warning_event(event, "%s: not enough memory!",
3194                                          __func__);
3195                         return -1;
3196                 }
3197
3198                 type = process_arg(event, arg, &token);
3199
3200                 if (type == TEP_EVENT_ERROR) {
3201                         free_token(token);
3202                         free_arg(arg);
3203                         return -1;
3204                 }
3205
3206                 *list = arg;
3207                 args++;
3208
3209                 if (type == TEP_EVENT_OP) {
3210                         type = process_op(event, arg, &token);
3211                         free_token(token);
3212                         if (type == TEP_EVENT_ERROR) {
3213                                 *list = NULL;
3214                                 free_arg(arg);
3215                                 return -1;
3216                         }
3217                         list = &arg->next;
3218                         continue;
3219                 }
3220
3221                 if (type == TEP_EVENT_DELIM && strcmp(token, ",") == 0) {
3222                         free_token(token);
3223                         *list = arg;
3224                         list = &arg->next;
3225                         continue;
3226                 }
3227                 break;
3228         } while (type != TEP_EVENT_NONE);
3229
3230         if (type != TEP_EVENT_NONE && type != TEP_EVENT_ERROR)
3231                 free_token(token);
3232
3233         return args;
3234 }
3235
3236 static int event_read_print(struct tep_event *event)
3237 {
3238         enum tep_event_type type;
3239         char *token;
3240         int ret;
3241
3242         if (read_expected_item(TEP_EVENT_ITEM, "print") < 0)
3243                 return -1;
3244
3245         if (read_expected(TEP_EVENT_ITEM, "fmt") < 0)
3246                 return -1;
3247
3248         if (read_expected(TEP_EVENT_OP, ":") < 0)
3249                 return -1;
3250
3251         if (read_expect_type(TEP_EVENT_DQUOTE, &token) < 0)
3252                 goto fail;
3253
3254  concat:
3255         event->print_fmt.format = token;
3256         event->print_fmt.args = NULL;
3257
3258         /* ok to have no arg */
3259         type = read_token_item(&token);
3260
3261         if (type == TEP_EVENT_NONE)
3262                 return 0;
3263
3264         /* Handle concatenation of print lines */
3265         if (type == TEP_EVENT_DQUOTE) {
3266                 char *cat;
3267
3268                 if (asprintf(&cat, "%s%s", event->print_fmt.format, token) < 0)
3269                         goto fail;
3270                 free_token(token);
3271                 free_token(event->print_fmt.format);
3272                 event->print_fmt.format = NULL;
3273                 token = cat;
3274                 goto concat;
3275         }
3276
3277         if (test_type_token(type, token, TEP_EVENT_DELIM, ","))
3278                 goto fail;
3279
3280         free_token(token);
3281
3282         ret = event_read_print_args(event, &event->print_fmt.args);
3283         if (ret < 0)
3284                 return -1;
3285
3286         return ret;
3287
3288  fail:
3289         free_token(token);
3290         return -1;
3291 }
3292
3293 /**
3294  * tep_find_common_field - return a common field by event
3295  * @event: handle for the event
3296  * @name: the name of the common field to return
3297  *
3298  * Returns a common field from the event by the given @name.
3299  * This only searches the common fields and not all field.
3300  */
3301 struct tep_format_field *
3302 tep_find_common_field(struct tep_event *event, const char *name)
3303 {
3304         struct tep_format_field *format;
3305
3306         for (format = event->format.common_fields;
3307              format; format = format->next) {
3308                 if (strcmp(format->name, name) == 0)
3309                         break;
3310         }
3311
3312         return format;
3313 }
3314
3315 /**
3316  * tep_find_field - find a non-common field
3317  * @event: handle for the event
3318  * @name: the name of the non-common field
3319  *
3320  * Returns a non-common field by the given @name.
3321  * This does not search common fields.
3322  */
3323 struct tep_format_field *
3324 tep_find_field(struct tep_event *event, const char *name)
3325 {
3326         struct tep_format_field *format;
3327
3328         for (format = event->format.fields;
3329              format; format = format->next) {
3330                 if (strcmp(format->name, name) == 0)
3331                         break;
3332         }
3333
3334         return format;
3335 }
3336
3337 /**
3338  * tep_find_any_field - find any field by name
3339  * @event: handle for the event
3340  * @name: the name of the field
3341  *
3342  * Returns a field by the given @name.
3343  * This searches the common field names first, then
3344  * the non-common ones if a common one was not found.
3345  */
3346 struct tep_format_field *
3347 tep_find_any_field(struct tep_event *event, const char *name)
3348 {
3349         struct tep_format_field *format;
3350
3351         format = tep_find_common_field(event, name);
3352         if (format)
3353                 return format;
3354         return tep_find_field(event, name);
3355 }
3356
3357 /**
3358  * tep_read_number - read a number from data
3359  * @tep: a handle to the trace event parser context
3360  * @ptr: the raw data
3361  * @size: the size of the data that holds the number
3362  *
3363  * Returns the number (converted to host) from the
3364  * raw data.
3365  */
3366 unsigned long long tep_read_number(struct tep_handle *tep,
3367                                    const void *ptr, int size)
3368 {
3369         unsigned long long val;
3370
3371         switch (size) {
3372         case 1:
3373                 return *(unsigned char *)ptr;
3374         case 2:
3375                 return tep_data2host2(tep, *(unsigned short *)ptr);
3376         case 4:
3377                 return tep_data2host4(tep, *(unsigned int *)ptr);
3378         case 8:
3379                 memcpy(&val, (ptr), sizeof(unsigned long long));
3380                 return tep_data2host8(tep, val);
3381         default:
3382                 /* BUG! */
3383                 return 0;
3384         }
3385 }
3386
3387 /**
3388  * tep_read_number_field - read a number from data
3389  * @field: a handle to the field
3390  * @data: the raw data to read
3391  * @value: the value to place the number in
3392  *
3393  * Reads raw data according to a field offset and size,
3394  * and translates it into @value.
3395  *
3396  * Returns 0 on success, -1 otherwise.
3397  */
3398 int tep_read_number_field(struct tep_format_field *field, const void *data,
3399                           unsigned long long *value)
3400 {
3401         if (!field)
3402                 return -1;
3403         switch (field->size) {
3404         case 1:
3405         case 2:
3406         case 4:
3407         case 8:
3408                 *value = tep_read_number(field->event->tep,
3409                                          data + field->offset, field->size);
3410                 return 0;
3411         default:
3412                 return -1;
3413         }
3414 }
3415
3416 static int get_common_info(struct tep_handle *tep,
3417                            const char *type, int *offset, int *size)
3418 {
3419         struct tep_event *event;
3420         struct tep_format_field *field;
3421
3422         /*
3423          * All events should have the same common elements.
3424          * Pick any event to find where the type is;
3425          */
3426         if (!tep->events) {
3427                 do_warning("no event_list!");
3428                 return -1;
3429         }
3430
3431         event = tep->events[0];
3432         field = tep_find_common_field(event, type);
3433         if (!field)
3434                 return -1;
3435
3436         *offset = field->offset;
3437         *size = field->size;
3438
3439         return 0;
3440 }
3441
3442 static int __parse_common(struct tep_handle *tep, void *data,
3443                           int *size, int *offset, const char *name)
3444 {
3445         int ret;
3446
3447         if (!*size) {
3448                 ret = get_common_info(tep, name, offset, size);
3449                 if (ret < 0)
3450                         return ret;
3451         }
3452         return tep_read_number(tep, data + *offset, *size);
3453 }
3454
3455 static int trace_parse_common_type(struct tep_handle *tep, void *data)
3456 {
3457         return __parse_common(tep, data,
3458                               &tep->type_size, &tep->type_offset,
3459                               "common_type");
3460 }
3461
3462 static int parse_common_pid(struct tep_handle *tep, void *data)
3463 {
3464         return __parse_common(tep, data,
3465                               &tep->pid_size, &tep->pid_offset,
3466                               "common_pid");
3467 }
3468
3469 static int parse_common_pc(struct tep_handle *tep, void *data)
3470 {
3471         return __parse_common(tep, data,
3472                               &tep->pc_size, &tep->pc_offset,
3473                               "common_preempt_count");
3474 }
3475
3476 static int parse_common_flags(struct tep_handle *tep, void *data)
3477 {
3478         return __parse_common(tep, data,
3479                               &tep->flags_size, &tep->flags_offset,
3480                               "common_flags");
3481 }
3482
3483 static int parse_common_lock_depth(struct tep_handle *tep, void *data)
3484 {
3485         return __parse_common(tep, data,
3486                               &tep->ld_size, &tep->ld_offset,
3487                               "common_lock_depth");
3488 }
3489
3490 static int parse_common_migrate_disable(struct tep_handle *tep, void *data)
3491 {
3492         return __parse_common(tep, data,
3493                               &tep->ld_size, &tep->ld_offset,
3494                               "common_migrate_disable");
3495 }
3496
3497 static int events_id_cmp(const void *a, const void *b);
3498
3499 /**
3500  * tep_find_event - find an event by given id
3501  * @tep: a handle to the trace event parser context
3502  * @id: the id of the event
3503  *
3504  * Returns an event that has a given @id.
3505  */
3506 struct tep_event *tep_find_event(struct tep_handle *tep, int id)
3507 {
3508         struct tep_event **eventptr;
3509         struct tep_event key;
3510         struct tep_event *pkey = &key;
3511
3512         /* Check cache first */
3513         if (tep->last_event && tep->last_event->id == id)
3514                 return tep->last_event;
3515
3516         key.id = id;
3517
3518         eventptr = bsearch(&pkey, tep->events, tep->nr_events,
3519                            sizeof(*tep->events), events_id_cmp);
3520
3521         if (eventptr) {
3522                 tep->last_event = *eventptr;
3523                 return *eventptr;
3524         }
3525
3526         return NULL;
3527 }
3528
3529 /**
3530  * tep_find_event_by_name - find an event by given name
3531  * @tep: a handle to the trace event parser context
3532  * @sys: the system name to search for
3533  * @name: the name of the event to search for
3534  *
3535  * This returns an event with a given @name and under the system
3536  * @sys. If @sys is NULL the first event with @name is returned.
3537  */
3538 struct tep_event *
3539 tep_find_event_by_name(struct tep_handle *tep,
3540                        const char *sys, const char *name)
3541 {
3542         struct tep_event *event = NULL;
3543         int i;
3544
3545         if (tep->last_event &&
3546             strcmp(tep->last_event->name, name) == 0 &&
3547             (!sys || strcmp(tep->last_event->system, sys) == 0))
3548                 return tep->last_event;
3549
3550         for (i = 0; i < tep->nr_events; i++) {
3551                 event = tep->events[i];
3552                 if (strcmp(event->name, name) == 0) {
3553                         if (!sys)
3554                                 break;
3555                         if (strcmp(event->system, sys) == 0)
3556                                 break;
3557                 }
3558         }
3559         if (i == tep->nr_events)
3560                 event = NULL;
3561
3562         tep->last_event = event;
3563         return event;
3564 }
3565
3566 static unsigned long long
3567 eval_num_arg(void *data, int size, struct tep_event *event, struct tep_print_arg *arg)
3568 {
3569         struct tep_handle *tep = event->tep;
3570         unsigned long long val = 0;
3571         unsigned long long left, right;
3572         struct tep_print_arg *typearg = NULL;
3573         struct tep_print_arg *larg;
3574         unsigned long offset;
3575         unsigned int field_size;
3576
3577         switch (arg->type) {
3578         case TEP_PRINT_NULL:
3579                 /* ?? */
3580                 return 0;
3581         case TEP_PRINT_ATOM:
3582                 return strtoull(arg->atom.atom, NULL, 0);
3583         case TEP_PRINT_FIELD:
3584                 if (!arg->field.field) {
3585                         arg->field.field = tep_find_any_field(event, arg->field.name);
3586                         if (!arg->field.field)
3587                                 goto out_warning_field;
3588
3589                 }
3590                 /* must be a number */
3591                 val = tep_read_number(tep, data + arg->field.field->offset,
3592                                       arg->field.field->size);
3593                 break;
3594         case TEP_PRINT_FLAGS:
3595         case TEP_PRINT_SYMBOL:
3596         case TEP_PRINT_INT_ARRAY:
3597         case TEP_PRINT_HEX:
3598         case TEP_PRINT_HEX_STR:
3599                 break;
3600         case TEP_PRINT_TYPE:
3601                 val = eval_num_arg(data, size, event, arg->typecast.item);
3602                 return eval_type(val, arg, 0);
3603         case TEP_PRINT_STRING:
3604         case TEP_PRINT_BSTRING:
3605         case TEP_PRINT_BITMASK:
3606                 return 0;
3607         case TEP_PRINT_FUNC: {
3608                 struct trace_seq s;
3609                 trace_seq_init(&s);
3610                 val = process_defined_func(&s, data, size, event, arg);
3611                 trace_seq_destroy(&s);
3612                 return val;
3613         }
3614         case TEP_PRINT_OP:
3615                 if (strcmp(arg->op.op, "[") == 0) {
3616                         /*
3617                          * Arrays are special, since we don't want
3618                          * to read the arg as is.
3619                          */
3620                         right = eval_num_arg(data, size, event, arg->op.right);
3621
3622                         /* handle typecasts */
3623                         larg = arg->op.left;
3624                         while (larg->type == TEP_PRINT_TYPE) {
3625                                 if (!typearg)
3626                                         typearg = larg;
3627                                 larg = larg->typecast.item;
3628                         }
3629
3630                         /* Default to long size */
3631                         field_size = tep->long_size;
3632
3633                         switch (larg->type) {
3634                         case TEP_PRINT_DYNAMIC_ARRAY:
3635                                 offset = tep_read_number(tep,
3636                                                    data + larg->dynarray.field->offset,
3637                                                    larg->dynarray.field->size);
3638                                 if (larg->dynarray.field->elementsize)
3639                                         field_size = larg->dynarray.field->elementsize;
3640                                 /*
3641                                  * The actual length of the dynamic array is stored
3642                                  * in the top half of the field, and the offset
3643                                  * is in the bottom half of the 32 bit field.
3644                                  */
3645                                 offset &= 0xffff;
3646                                 offset += right;
3647                                 break;
3648                         case TEP_PRINT_FIELD:
3649                                 if (!larg->field.field) {
3650                                         larg->field.field =
3651                                                 tep_find_any_field(event, larg->field.name);
3652                                         if (!larg->field.field) {
3653                                                 arg = larg;
3654                                                 goto out_warning_field;
3655                                         }
3656                                 }
3657                                 field_size = larg->field.field->elementsize;
3658                                 offset = larg->field.field->offset +
3659                                         right * larg->field.field->elementsize;
3660                                 break;
3661                         default:
3662                                 goto default_op; /* oops, all bets off */
3663                         }
3664                         val = tep_read_number(tep,
3665                                               data + offset, field_size);
3666                         if (typearg)
3667                                 val = eval_type(val, typearg, 1);
3668                         break;
3669                 } else if (strcmp(arg->op.op, "?") == 0) {
3670                         left = eval_num_arg(data, size, event, arg->op.left);
3671                         arg = arg->op.right;
3672                         if (left)
3673                                 val = eval_num_arg(data, size, event, arg->op.left);
3674                         else
3675                                 val = eval_num_arg(data, size, event, arg->op.right);
3676                         break;
3677                 }
3678  default_op:
3679                 left = eval_num_arg(data, size, event, arg->op.left);
3680                 right = eval_num_arg(data, size, event, arg->op.right);
3681                 switch (arg->op.op[0]) {
3682                 case '!':
3683                         switch (arg->op.op[1]) {
3684                         case 0:
3685                                 val = !right;
3686                                 break;
3687                         case '=':
3688                                 val = left != right;
3689                                 break;
3690                         default:
3691                                 goto out_warning_op;
3692                         }
3693                         break;
3694                 case '~':
3695                         val = ~right;
3696                         break;
3697                 case '|':
3698                         if (arg->op.op[1])
3699                                 val = left || right;
3700                         else
3701                                 val = left | right;
3702                         break;
3703                 case '&':
3704                         if (arg->op.op[1])
3705                                 val = left && right;
3706                         else
3707                                 val = left & right;
3708                         break;
3709                 case '<':
3710                         switch (arg->op.op[1]) {
3711                         case 0:
3712                                 val = left < right;
3713                                 break;
3714                         case '<':
3715                                 val = left << right;
3716                                 break;
3717                         case '=':
3718                                 val = left <= right;
3719                                 break;
3720                         default:
3721                                 goto out_warning_op;
3722                         }
3723                         break;
3724                 case '>':
3725                         switch (arg->op.op[1]) {
3726                         case 0:
3727                                 val = left > right;
3728                                 break;
3729                         case '>':
3730                                 val = left >> right;
3731                                 break;
3732                         case '=':
3733                                 val = left >= right;
3734                                 break;
3735                         default:
3736                                 goto out_warning_op;
3737                         }
3738                         break;
3739                 case '=':
3740                         if (arg->op.op[1] != '=')
3741                                 goto out_warning_op;
3742
3743                         val = left == right;
3744                         break;
3745                 case '-':
3746                         val = left - right;
3747                         break;
3748                 case '+':
3749                         val = left + right;
3750                         break;
3751                 case '/':
3752                         val = left / right;
3753                         break;
3754                 case '%':
3755                         val = left % right;
3756                         break;
3757                 case '*':
3758                         val = left * right;
3759                         break;
3760                 default:
3761                         goto out_warning_op;
3762                 }
3763                 break;
3764         case TEP_PRINT_DYNAMIC_ARRAY_LEN:
3765                 offset = tep_read_number(tep,
3766                                          data + arg->dynarray.field->offset,
3767                                          arg->dynarray.field->size);
3768                 /*
3769                  * The total allocated length of the dynamic array is
3770                  * stored in the top half of the field, and the offset
3771                  * is in the bottom half of the 32 bit field.
3772                  */
3773                 val = (unsigned long long)(offset >> 16);
3774                 break;
3775         case TEP_PRINT_DYNAMIC_ARRAY:
3776                 /* Without [], we pass the address to the dynamic data */
3777                 offset = tep_read_number(tep,
3778                                          data + arg->dynarray.field->offset,
3779                                          arg->dynarray.field->size);
3780                 /*
3781                  * The total allocated length of the dynamic array is
3782                  * stored in the top half of the field, and the offset
3783                  * is in the bottom half of the 32 bit field.
3784                  */
3785                 offset &= 0xffff;
3786                 val = (unsigned long long)((unsigned long)data + offset);
3787                 break;
3788         default: /* not sure what to do there */
3789                 return 0;
3790         }
3791         return val;
3792
3793 out_warning_op:
3794         do_warning_event(event, "%s: unknown op '%s'", __func__, arg->op.op);
3795         return 0;
3796
3797 out_warning_field:
3798         do_warning_event(event, "%s: field %s not found",
3799                          __func__, arg->field.name);
3800         return 0;
3801 }
3802
3803 struct flag {
3804         const char *name;
3805         unsigned long long value;
3806 };
3807
3808 static const struct flag flags[] = {
3809         { "HI_SOFTIRQ", 0 },
3810         { "TIMER_SOFTIRQ", 1 },
3811         { "NET_TX_SOFTIRQ", 2 },
3812         { "NET_RX_SOFTIRQ", 3 },
3813         { "BLOCK_SOFTIRQ", 4 },
3814         { "IRQ_POLL_SOFTIRQ", 5 },
3815         { "TASKLET_SOFTIRQ", 6 },
3816         { "SCHED_SOFTIRQ", 7 },
3817         { "HRTIMER_SOFTIRQ", 8 },
3818         { "RCU_SOFTIRQ", 9 },
3819
3820         { "HRTIMER_NORESTART", 0 },
3821         { "HRTIMER_RESTART", 1 },
3822 };
3823
3824 static long long eval_flag(const char *flag)
3825 {
3826         int i;
3827
3828         /*
3829          * Some flags in the format files do not get converted.
3830          * If the flag is not numeric, see if it is something that
3831          * we already know about.
3832          */
3833         if (isdigit(flag[0]))
3834                 return strtoull(flag, NULL, 0);
3835
3836         for (i = 0; i < (int)(sizeof(flags)/sizeof(flags[0])); i++)
3837                 if (strcmp(flags[i].name, flag) == 0)
3838                         return flags[i].value;
3839
3840         return -1LL;
3841 }
3842
3843 static void print_str_to_seq(struct trace_seq *s, const char *format,
3844                              int len_arg, const char *str)
3845 {
3846         if (len_arg >= 0)
3847                 trace_seq_printf(s, format, len_arg, str);
3848         else
3849                 trace_seq_printf(s, format, str);
3850 }
3851
3852 static void print_bitmask_to_seq(struct tep_handle *tep,
3853                                  struct trace_seq *s, const char *format,
3854                                  int len_arg, const void *data, int size)
3855 {
3856         int nr_bits = size * 8;
3857         int str_size = (nr_bits + 3) / 4;
3858         int len = 0;
3859         char buf[3];
3860         char *str;
3861         int index;
3862         int i;
3863
3864         /*
3865          * The kernel likes to put in commas every 32 bits, we
3866          * can do the same.
3867          */
3868         str_size += (nr_bits - 1) / 32;
3869
3870         str = malloc(str_size + 1);
3871         if (!str) {
3872                 do_warning("%s: not enough memory!", __func__);
3873                 return;
3874         }
3875         str[str_size] = 0;
3876
3877         /* Start out with -2 for the two chars per byte */
3878         for (i = str_size - 2; i >= 0; i -= 2) {
3879                 /*
3880                  * data points to a bit mask of size bytes.
3881                  * In the kernel, this is an array of long words, thus
3882                  * endianness is very important.
3883                  */
3884                 if (tep->file_bigendian)
3885                         index = size - (len + 1);
3886                 else
3887                         index = len;
3888
3889                 snprintf(buf, 3, "%02x", *((unsigned char *)data + index));
3890                 memcpy(str + i, buf, 2);
3891                 len++;
3892                 if (!(len & 3) && i > 0) {
3893                         i--;
3894                         str[i] = ',';
3895                 }
3896         }
3897
3898         if (len_arg >= 0)
3899                 trace_seq_printf(s, format, len_arg, str);
3900         else
3901                 trace_seq_printf(s, format, str);
3902
3903         free(str);
3904 }
3905
3906 static void print_str_arg(struct trace_seq *s, void *data, int size,
3907                           struct tep_event *event, const char *format,
3908                           int len_arg, struct tep_print_arg *arg)
3909 {
3910         struct tep_handle *tep = event->tep;
3911         struct tep_print_flag_sym *flag;
3912         struct tep_format_field *field;
3913         struct printk_map *printk;
3914         long long val, fval;
3915         unsigned long long addr;
3916         char *str;
3917         unsigned char *hex;
3918         int print;
3919         int i, len;
3920
3921         switch (arg->type) {
3922         case TEP_PRINT_NULL:
3923                 /* ?? */
3924                 return;
3925         case TEP_PRINT_ATOM:
3926                 print_str_to_seq(s, format, len_arg, arg->atom.atom);
3927                 return;
3928         case TEP_PRINT_FIELD:
3929                 field = arg->field.field;
3930                 if (!field) {
3931                         field = tep_find_any_field(event, arg->field.name);
3932                         if (!field) {
3933                                 str = arg->field.name;
3934                                 goto out_warning_field;
3935                         }
3936                         arg->field.field = field;
3937                 }
3938                 /* Zero sized fields, mean the rest of the data */
3939                 len = field->size ? : size - field->offset;
3940
3941                 /*
3942                  * Some events pass in pointers. If this is not an array
3943                  * and the size is the same as long_size, assume that it
3944                  * is a pointer.
3945                  */
3946                 if (!(field->flags & TEP_FIELD_IS_ARRAY) &&
3947                     field->size == tep->long_size) {
3948
3949                         /* Handle heterogeneous recording and processing
3950                          * architectures
3951                          *
3952                          * CASE I:
3953                          * Traces recorded on 32-bit devices (32-bit
3954                          * addressing) and processed on 64-bit devices:
3955                          * In this case, only 32 bits should be read.
3956                          *
3957                          * CASE II:
3958                          * Traces recorded on 64 bit devices and processed
3959                          * on 32-bit devices:
3960                          * In this case, 64 bits must be read.
3961                          */
3962                         addr = (tep->long_size == 8) ?
3963                                 *(unsigned long long *)(data + field->offset) :
3964                                 (unsigned long long)*(unsigned int *)(data + field->offset);
3965
3966                         /* Check if it matches a print format */
3967                         printk = find_printk(tep, addr);
3968                         if (printk)
3969                                 trace_seq_puts(s, printk->printk);
3970                         else
3971                                 trace_seq_printf(s, "%llx", addr);
3972                         break;
3973                 }
3974                 str = malloc(len + 1);
3975                 if (!str) {
3976                         do_warning_event(event, "%s: not enough memory!",
3977                                          __func__);
3978                         return;
3979                 }
3980                 memcpy(str, data + field->offset, len);
3981                 str[len] = 0;
3982                 print_str_to_seq(s, format, len_arg, str);
3983                 free(str);
3984                 break;
3985         case TEP_PRINT_FLAGS:
3986                 val = eval_num_arg(data, size, event, arg->flags.field);
3987                 print = 0;
3988                 for (flag = arg->flags.flags; flag; flag = flag->next) {
3989                         fval = eval_flag(flag->value);
3990                         if (!val && fval < 0) {
3991                                 print_str_to_seq(s, format, len_arg, flag->str);
3992                                 break;
3993                         }
3994                         if (fval > 0 && (val & fval) == fval) {
3995                                 if (print && arg->flags.delim)
3996                                         trace_seq_puts(s, arg->flags.delim);
3997                                 print_str_to_seq(s, format, len_arg, flag->str);
3998                                 print = 1;
3999                                 val &= ~fval;
4000                         }
4001                 }
4002                 if (val) {
4003                         if (print && arg->flags.delim)
4004                                 trace_seq_puts(s, arg->flags.delim);
4005                         trace_seq_printf(s, "0x%llx", val);
4006                 }
4007                 break;
4008         case TEP_PRINT_SYMBOL:
4009                 val = eval_num_arg(data, size, event, arg->symbol.field);
4010                 for (flag = arg->symbol.symbols; flag; flag = flag->next) {
4011                         fval = eval_flag(flag->value);
4012                         if (val == fval) {
4013                                 print_str_to_seq(s, format, len_arg, flag->str);
4014                                 break;
4015                         }
4016                 }
4017                 if (!flag)
4018                         trace_seq_printf(s, "0x%llx", val);
4019                 break;
4020         case TEP_PRINT_HEX:
4021         case TEP_PRINT_HEX_STR:
4022                 if (arg->hex.field->type == TEP_PRINT_DYNAMIC_ARRAY) {
4023                         unsigned long offset;
4024                         offset = tep_read_number(tep,
4025                                 data + arg->hex.field->dynarray.field->offset,
4026                                 arg->hex.field->dynarray.field->size);
4027                         hex = data + (offset & 0xffff);
4028                 } else {
4029                         field = arg->hex.field->field.field;
4030                         if (!field) {
4031                                 str = arg->hex.field->field.name;
4032                                 field = tep_find_any_field(event, str);
4033                                 if (!field)
4034                                         goto out_warning_field;
4035                                 arg->hex.field->field.field = field;
4036                         }
4037                         hex = data + field->offset;
4038                 }
4039                 len = eval_num_arg(data, size, event, arg->hex.size);
4040                 for (i = 0; i < len; i++) {
4041                         if (i && arg->type == TEP_PRINT_HEX)
4042                                 trace_seq_putc(s, ' ');
4043                         trace_seq_printf(s, "%02x", hex[i]);
4044                 }
4045                 break;
4046
4047         case TEP_PRINT_INT_ARRAY: {
4048                 void *num;
4049                 int el_size;
4050
4051                 if (arg->int_array.field->type == TEP_PRINT_DYNAMIC_ARRAY) {
4052                         unsigned long offset;
4053                         struct tep_format_field *field =
4054                                 arg->int_array.field->dynarray.field;
4055                         offset = tep_read_number(tep,
4056                                                  data + field->offset,
4057                                                  field->size);
4058                         num = data + (offset & 0xffff);
4059                 } else {
4060                         field = arg->int_array.field->field.field;
4061                         if (!field) {
4062                                 str = arg->int_array.field->field.name;
4063                                 field = tep_find_any_field(event, str);
4064                                 if (!field)
4065                                         goto out_warning_field;
4066                                 arg->int_array.field->field.field = field;
4067                         }
4068                         num = data + field->offset;
4069                 }
4070                 len = eval_num_arg(data, size, event, arg->int_array.count);
4071                 el_size = eval_num_arg(data, size, event,
4072                                        arg->int_array.el_size);
4073                 for (i = 0; i < len; i++) {
4074                         if (i)
4075                                 trace_seq_putc(s, ' ');
4076
4077                         if (el_size == 1) {
4078                                 trace_seq_printf(s, "%u", *(uint8_t *)num);
4079                         } else if (el_size == 2) {
4080                                 trace_seq_printf(s, "%u", *(uint16_t *)num);
4081                         } else if (el_size == 4) {
4082                                 trace_seq_printf(s, "%u", *(uint32_t *)num);
4083                         } else if (el_size == 8) {
4084                                 trace_seq_printf(s, "%"PRIu64, *(uint64_t *)num);
4085                         } else {
4086                                 trace_seq_printf(s, "BAD SIZE:%d 0x%x",
4087                                                  el_size, *(uint8_t *)num);
4088                                 el_size = 1;
4089                         }
4090
4091                         num += el_size;
4092                 }
4093                 break;
4094         }
4095         case TEP_PRINT_TYPE:
4096                 break;
4097         case TEP_PRINT_STRING: {
4098                 int str_offset;
4099
4100                 if (arg->string.offset == -1) {
4101                         struct tep_format_field *f;
4102
4103                         f = tep_find_any_field(event, arg->string.string);
4104                         arg->string.offset = f->offset;
4105                 }
4106                 str_offset = tep_data2host4(tep, *(unsigned int *)(data + arg->string.offset));
4107                 str_offset &= 0xffff;
4108                 print_str_to_seq(s, format, len_arg, ((char *)data) + str_offset);
4109                 break;
4110         }
4111         case TEP_PRINT_BSTRING:
4112                 print_str_to_seq(s, format, len_arg, arg->string.string);
4113                 break;
4114         case TEP_PRINT_BITMASK: {
4115                 int bitmask_offset;
4116                 int bitmask_size;
4117
4118                 if (arg->bitmask.offset == -1) {
4119                         struct tep_format_field *f;
4120
4121                         f = tep_find_any_field(event, arg->bitmask.bitmask);
4122                         arg->bitmask.offset = f->offset;
4123                 }
4124                 bitmask_offset = tep_data2host4(tep, *(unsigned int *)(data + arg->bitmask.offset));
4125                 bitmask_size = bitmask_offset >> 16;
4126                 bitmask_offset &= 0xffff;
4127                 print_bitmask_to_seq(tep, s, format, len_arg,
4128                                      data + bitmask_offset, bitmask_size);
4129                 break;
4130         }
4131         case TEP_PRINT_OP:
4132                 /*
4133                  * The only op for string should be ? :
4134                  */
4135                 if (arg->op.op[0] != '?')
4136                         return;
4137                 val = eval_num_arg(data, size, event, arg->op.left);
4138                 if (val)
4139                         print_str_arg(s, data, size, event,
4140                                       format, len_arg, arg->op.right->op.left);
4141                 else
4142                         print_str_arg(s, data, size, event,
4143                                       format, len_arg, arg->op.right->op.right);
4144                 break;
4145         case TEP_PRINT_FUNC:
4146                 process_defined_func(s, data, size, event, arg);
4147                 break;
4148         default:
4149                 /* well... */
4150                 break;
4151         }
4152
4153         return;
4154
4155 out_warning_field:
4156         do_warning_event(event, "%s: field %s not found",
4157                          __func__, arg->field.name);
4158 }
4159
4160 static unsigned long long
4161 process_defined_func(struct trace_seq *s, void *data, int size,
4162                      struct tep_event *event, struct tep_print_arg *arg)
4163 {
4164         struct tep_function_handler *func_handle = arg->func.func;
4165         struct func_params *param;
4166         unsigned long long *args;
4167         unsigned long long ret;
4168         struct tep_print_arg *farg;
4169         struct trace_seq str;
4170         struct save_str {
4171                 struct save_str *next;
4172                 char *str;
4173         } *strings = NULL, *string;
4174         int i;
4175
4176         if (!func_handle->nr_args) {
4177                 ret = (*func_handle->func)(s, NULL);
4178                 goto out;
4179         }
4180
4181         farg = arg->func.args;
4182         param = func_handle->params;
4183
4184         ret = ULLONG_MAX;
4185         args = malloc(sizeof(*args) * func_handle->nr_args);
4186         if (!args)
4187                 goto out;
4188
4189         for (i = 0; i < func_handle->nr_args; i++) {
4190                 switch (param->type) {
4191                 case TEP_FUNC_ARG_INT:
4192                 case TEP_FUNC_ARG_LONG:
4193                 case TEP_FUNC_ARG_PTR:
4194                         args[i] = eval_num_arg(data, size, event, farg);
4195                         break;
4196                 case TEP_FUNC_ARG_STRING:
4197                         trace_seq_init(&str);
4198                         print_str_arg(&str, data, size, event, "%s", -1, farg);
4199                         trace_seq_terminate(&str);
4200                         string = malloc(sizeof(*string));
4201                         if (!string) {
4202                                 do_warning_event(event, "%s(%d): malloc str",
4203                                                  __func__, __LINE__);
4204                                 goto out_free;
4205                         }
4206                         string->next = strings;
4207                         string->str = strdup(str.buffer);
4208                         if (!string->str) {
4209                                 free(string);
4210                                 do_warning_event(event, "%s(%d): malloc str",
4211                                                  __func__, __LINE__);
4212                                 goto out_free;
4213                         }
4214                         args[i] = (uintptr_t)string->str;
4215                         strings = string;
4216                         trace_seq_destroy(&str);
4217                         break;
4218                 default:
4219                         /*
4220                          * Something went totally wrong, this is not
4221                          * an input error, something in this code broke.
4222                          */
4223                         do_warning_event(event, "Unexpected end of arguments\n");
4224                         goto out_free;
4225                 }
4226                 farg = farg->next;
4227                 param = param->next;
4228         }
4229
4230         ret = (*func_handle->func)(s, args);
4231 out_free:
4232         free(args);
4233         while (strings) {
4234                 string = strings;
4235                 strings = string->next;
4236                 free(string->str);
4237                 free(string);
4238         }
4239
4240  out:
4241         /* TBD : handle return type here */
4242         return ret;
4243 }
4244
4245 static void free_args(struct tep_print_arg *args)
4246 {
4247         struct tep_print_arg *next;
4248
4249         while (args) {
4250                 next = args->next;
4251
4252                 free_arg(args);
4253                 args = next;
4254         }
4255 }
4256
4257 static struct tep_print_arg *make_bprint_args(char *fmt, void *data, int size, struct tep_event *event)
4258 {
4259         struct tep_handle *tep = event->tep;
4260         struct tep_format_field *field, *ip_field;
4261         struct tep_print_arg *args, *arg, **next;
4262         unsigned long long ip, val;
4263         char *ptr;
4264         void *bptr;
4265         int vsize = 0;
4266
4267         field = tep->bprint_buf_field;
4268         ip_field = tep->bprint_ip_field;
4269
4270         if (!field) {
4271                 field = tep_find_field(event, "buf");
4272                 if (!field) {
4273                         do_warning_event(event, "can't find buffer field for binary printk");
4274                         return NULL;
4275                 }
4276                 ip_field = tep_find_field(event, "ip");
4277                 if (!ip_field) {
4278                         do_warning_event(event, "can't find ip field for binary printk");
4279                         return NULL;
4280                 }
4281                 tep->bprint_buf_field = field;
4282                 tep->bprint_ip_field = ip_field;
4283         }
4284
4285         ip = tep_read_number(tep, data + ip_field->offset, ip_field->size);
4286
4287         /*
4288          * The first arg is the IP pointer.
4289          */
4290         args = alloc_arg();
4291         if (!args) {
4292                 do_warning_event(event, "%s(%d): not enough memory!",
4293                                  __func__, __LINE__);
4294                 return NULL;
4295         }
4296         arg = args;
4297         arg->next = NULL;
4298         next = &arg->next;
4299
4300         arg->type = TEP_PRINT_ATOM;
4301
4302         if (asprintf(&arg->atom.atom, "%lld", ip) < 0)
4303                 goto out_free;
4304
4305         /* skip the first "%ps: " */
4306         for (ptr = fmt + 5, bptr = data + field->offset;
4307              bptr < data + size && *ptr; ptr++) {
4308                 int ls = 0;
4309
4310                 if (*ptr == '%') {
4311  process_again:
4312                         ptr++;
4313                         switch (*ptr) {
4314                         case '%':
4315                                 break;
4316                         case 'l':
4317                                 ls++;
4318                                 goto process_again;
4319                         case 'L':
4320                                 ls = 2;
4321                                 goto process_again;
4322                         case '0' ... '9':
4323                                 goto process_again;
4324                         case '.':
4325                                 goto process_again;
4326                         case 'z':
4327                         case 'Z':
4328                                 ls = 1;
4329                                 goto process_again;
4330                         case 'p':
4331                                 ls = 1;
4332                                 if (isalnum(ptr[1])) {
4333                                         ptr++;
4334                                         /* Check for special pointers */
4335                                         switch (*ptr) {
4336                                         case 's':
4337                                         case 'S':
4338                                         case 'f':
4339                                         case 'F':
4340                                         case 'x':
4341                                                 break;
4342                                         default:
4343                                                 /*
4344                                                  * Older kernels do not process
4345                                                  * dereferenced pointers.
4346                                                  * Only process if the pointer
4347                                                  * value is a printable.
4348                                                  */
4349                                                 if (isprint(*(char *)bptr))
4350                                                         goto process_string;
4351                                         }
4352                                 }
4353                                 /* fall through */
4354                         case 'd':
4355                         case 'u':
4356                         case 'x':
4357                         case 'i':
4358                                 switch (ls) {
4359                                 case 0:
4360                                         vsize = 4;
4361                                         break;
4362                                 case 1:
4363                                         vsize = tep->long_size;
4364                                         break;
4365                                 case 2:
4366                                         vsize = 8;
4367                                         break;
4368                                 default:
4369                                         vsize = ls; /* ? */
4370                                         break;
4371                                 }
4372                         /* fall through */
4373                         case '*':
4374                                 if (*ptr == '*')
4375                                         vsize = 4;
4376
4377                                 /* the pointers are always 4 bytes aligned */
4378                                 bptr = (void *)(((unsigned long)bptr + 3) &
4379                                                 ~3);
4380                                 val = tep_read_number(tep, bptr, vsize);
4381                                 bptr += vsize;
4382                                 arg = alloc_arg();
4383                                 if (!arg) {
4384                                         do_warning_event(event, "%s(%d): not enough memory!",
4385                                                    __func__, __LINE__);
4386                                         goto out_free;
4387                                 }
4388                                 arg->next = NULL;
4389                                 arg->type = TEP_PRINT_ATOM;
4390                                 if (asprintf(&arg->atom.atom, "%lld", val) < 0) {
4391                                         free(arg);
4392                                         goto out_free;
4393                                 }
4394                                 *next = arg;
4395                                 next = &arg->next;
4396                                 /*
4397                                  * The '*' case means that an arg is used as the length.
4398                                  * We need to continue to figure out for what.
4399                                  */
4400                                 if (*ptr == '*')
4401                                         goto process_again;
4402
4403                                 break;
4404                         case 's':
4405  process_string:
4406                                 arg = alloc_arg();
4407                                 if (!arg) {
4408                                         do_warning_event(event, "%s(%d): not enough memory!",
4409                                                    __func__, __LINE__);
4410                                         goto out_free;
4411                                 }
4412                                 arg->next = NULL;
4413                                 arg->type = TEP_PRINT_BSTRING;
4414                                 arg->string.string = strdup(bptr);
4415                                 if (!arg->string.string)
4416                                         goto out_free;
4417                                 bptr += strlen(bptr) + 1;
4418                                 *next = arg;
4419                                 next = &arg->next;
4420                         default:
4421                                 break;
4422                         }
4423                 }
4424         }
4425
4426         return args;
4427
4428 out_free:
4429         free_args(args);
4430         return NULL;
4431 }
4432
4433 static char *
4434 get_bprint_format(void *data, int size __maybe_unused,
4435                   struct tep_event *event)
4436 {
4437         struct tep_handle *tep = event->tep;
4438         unsigned long long addr;
4439         struct tep_format_field *field;
4440         struct printk_map *printk;
4441         char *format;
4442
4443         field = tep->bprint_fmt_field;
4444
4445         if (!field) {
4446                 field = tep_find_field(event, "fmt");
4447                 if (!field) {
4448                         do_warning_event(event, "can't find format field for binary printk");
4449                         return NULL;
4450                 }
4451                 tep->bprint_fmt_field = field;
4452         }
4453
4454         addr = tep_read_number(tep, data + field->offset, field->size);
4455
4456         printk = find_printk(tep, addr);
4457         if (!printk) {
4458                 if (asprintf(&format, "%%pf: (NO FORMAT FOUND at %llx)\n", addr) < 0)
4459                         return NULL;
4460                 return format;
4461         }
4462
4463         if (asprintf(&format, "%s: %s", "%pf", printk->printk) < 0)
4464                 return NULL;
4465
4466         return format;
4467 }
4468
4469 static void print_mac_arg(struct trace_seq *s, int mac, void *data, int size,
4470                           struct tep_event *event, struct tep_print_arg *arg)
4471 {
4472         unsigned char *buf;
4473         const char *fmt = "%.2x:%.2x:%.2x:%.2x:%.2x:%.2x";
4474
4475         if (arg->type == TEP_PRINT_FUNC) {
4476                 process_defined_func(s, data, size, event, arg);
4477                 return;
4478         }
4479
4480         if (arg->type != TEP_PRINT_FIELD) {
4481                 trace_seq_printf(s, "ARG TYPE NOT FIELD BUT %d",
4482                                  arg->type);
4483                 return;
4484         }
4485
4486         if (mac == 'm')
4487                 fmt = "%.2x%.2x%.2x%.2x%.2x%.2x";
4488         if (!arg->field.field) {
4489                 arg->field.field =
4490                         tep_find_any_field(event, arg->field.name);
4491                 if (!arg->field.field) {
4492                         do_warning_event(event, "%s: field %s not found",
4493                                          __func__, arg->field.name);
4494                         return;
4495                 }
4496         }
4497         if (arg->field.field->size != 6) {
4498                 trace_seq_printf(s, "INVALIDMAC");
4499                 return;
4500         }
4501         buf = data + arg->field.field->offset;
4502         trace_seq_printf(s, fmt, buf[0], buf[1], buf[2], buf[3], buf[4], buf[5]);
4503 }
4504
4505 static void print_ip4_addr(struct trace_seq *s, char i, unsigned char *buf)
4506 {
4507         const char *fmt;
4508
4509         if (i == 'i')
4510                 fmt = "%03d.%03d.%03d.%03d";
4511         else
4512                 fmt = "%d.%d.%d.%d";
4513
4514         trace_seq_printf(s, fmt, buf[0], buf[1], buf[2], buf[3]);
4515 }
4516
4517 static inline bool ipv6_addr_v4mapped(const struct in6_addr *a)
4518 {
4519         return ((unsigned long)(a->s6_addr32[0] | a->s6_addr32[1]) |
4520                 (unsigned long)(a->s6_addr32[2] ^ htonl(0x0000ffff))) == 0UL;
4521 }
4522
4523 static inline bool ipv6_addr_is_isatap(const struct in6_addr *addr)
4524 {
4525         return (addr->s6_addr32[2] | htonl(0x02000000)) == htonl(0x02005EFE);
4526 }
4527
4528 static void print_ip6c_addr(struct trace_seq *s, unsigned char *addr)
4529 {
4530         int i, j, range;
4531         unsigned char zerolength[8];
4532         int longest = 1;
4533         int colonpos = -1;
4534         uint16_t word;
4535         uint8_t hi, lo;
4536         bool needcolon = false;
4537         bool useIPv4;
4538         struct in6_addr in6;
4539
4540         memcpy(&in6, addr, sizeof(struct in6_addr));
4541
4542         useIPv4 = ipv6_addr_v4mapped(&in6) || ipv6_addr_is_isatap(&in6);
4543
4544         memset(zerolength, 0, sizeof(zerolength));
4545
4546         if (useIPv4)
4547                 range = 6;
4548         else
4549                 range = 8;
4550
4551         /* find position of longest 0 run */
4552         for (i = 0; i < range; i++) {
4553                 for (j = i; j < range; j++) {
4554                         if (in6.s6_addr16[j] != 0)
4555                                 break;
4556                         zerolength[i]++;
4557                 }
4558         }
4559         for (i = 0; i < range; i++) {
4560                 if (zerolength[i] > longest) {
4561                         longest = zerolength[i];
4562                         colonpos = i;
4563                 }
4564         }
4565         if (longest == 1)               /* don't compress a single 0 */
4566                 colonpos = -1;
4567
4568         /* emit address */
4569         for (i = 0; i < range; i++) {
4570                 if (i == colonpos) {
4571                         if (needcolon || i == 0)
4572                                 trace_seq_printf(s, ":");
4573                         trace_seq_printf(s, ":");
4574                         needcolon = false;
4575                         i += longest - 1;
4576                         continue;
4577                 }
4578                 if (needcolon) {
4579                         trace_seq_printf(s, ":");
4580                         needcolon = false;
4581                 }
4582                 /* hex u16 without leading 0s */
4583                 word = ntohs(in6.s6_addr16[i]);
4584                 hi = word >> 8;
4585                 lo = word & 0xff;
4586                 if (hi)
4587                         trace_seq_printf(s, "%x%02x", hi, lo);
4588                 else
4589                         trace_seq_printf(s, "%x", lo);
4590
4591                 needcolon = true;
4592         }
4593
4594         if (useIPv4) {
4595                 if (needcolon)
4596                         trace_seq_printf(s, ":");
4597                 print_ip4_addr(s, 'I', &in6.s6_addr[12]);
4598         }
4599
4600         return;
4601 }
4602
4603 static void print_ip6_addr(struct trace_seq *s, char i, unsigned char *buf)
4604 {
4605         int j;
4606
4607         for (j = 0; j < 16; j += 2) {
4608                 trace_seq_printf(s, "%02x%02x", buf[j], buf[j+1]);
4609                 if (i == 'I' && j < 14)
4610                         trace_seq_printf(s, ":");
4611         }
4612 }
4613
4614 /*
4615  * %pi4   print an IPv4 address with leading zeros
4616  * %pI4   print an IPv4 address without leading zeros
4617  * %pi6   print an IPv6 address without colons
4618  * %pI6   print an IPv6 address with colons
4619  * %pI6c  print an IPv6 address in compressed form with colons
4620  * %pISpc print an IP address based on sockaddr; p adds port.
4621  */
4622 static int print_ipv4_arg(struct trace_seq *s, const char *ptr, char i,
4623                           void *data, int size, struct tep_event *event,
4624                           struct tep_print_arg *arg)
4625 {
4626         unsigned char *buf;
4627
4628         if (arg->type == TEP_PRINT_FUNC) {
4629                 process_defined_func(s, data, size, event, arg);
4630                 return 0;
4631         }
4632
4633         if (arg->type != TEP_PRINT_FIELD) {
4634                 trace_seq_printf(s, "ARG TYPE NOT FIELD BUT %d", arg->type);
4635                 return 0;
4636         }
4637
4638         if (!arg->field.field) {
4639                 arg->field.field =
4640                         tep_find_any_field(event, arg->field.name);
4641                 if (!arg->field.field) {
4642                         do_warning("%s: field %s not found",
4643                                    __func__, arg->field.name);
4644                         return 0;
4645                 }
4646         }
4647
4648         buf = data + arg->field.field->offset;
4649
4650         if (arg->field.field->size != 4) {
4651                 trace_seq_printf(s, "INVALIDIPv4");
4652                 return 0;
4653         }
4654         print_ip4_addr(s, i, buf);
4655
4656         return 0;
4657 }
4658
4659 static int print_ipv6_arg(struct trace_seq *s, const char *ptr, char i,
4660                           void *data, int size, struct tep_event *event,
4661                           struct tep_print_arg *arg)
4662 {
4663         char have_c = 0;
4664         unsigned char *buf;
4665         int rc = 0;
4666
4667         /* pI6c */
4668         if (i == 'I' && *ptr == 'c') {
4669                 have_c = 1;
4670                 ptr++;
4671                 rc++;
4672         }
4673
4674         if (arg->type == TEP_PRINT_FUNC) {
4675                 process_defined_func(s, data, size, event, arg);
4676                 return rc;
4677         }
4678
4679         if (arg->type != TEP_PRINT_FIELD) {
4680                 trace_seq_printf(s, "ARG TYPE NOT FIELD BUT %d", arg->type);
4681                 return rc;
4682         }
4683
4684         if (!arg->field.field) {
4685                 arg->field.field =
4686                         tep_find_any_field(event, arg->field.name);
4687                 if (!arg->field.field) {
4688                         do_warning("%s: field %s not found",
4689                                    __func__, arg->field.name);
4690                         return rc;
4691                 }
4692         }
4693
4694         buf = data + arg->field.field->offset;
4695
4696         if (arg->field.field->size != 16) {
4697                 trace_seq_printf(s, "INVALIDIPv6");
4698                 return rc;
4699         }
4700
4701         if (have_c)
4702                 print_ip6c_addr(s, buf);
4703         else
4704                 print_ip6_addr(s, i, buf);
4705
4706         return rc;
4707 }
4708
4709 static int print_ipsa_arg(struct trace_seq *s, const char *ptr, char i,
4710                           void *data, int size, struct tep_event *event,
4711                           struct tep_print_arg *arg)
4712 {
4713         char have_c = 0, have_p = 0;
4714         unsigned char *buf;
4715         struct sockaddr_storage *sa;
4716         int rc = 0;
4717
4718         /* pISpc */
4719         if (i == 'I') {
4720                 if (*ptr == 'p') {
4721                         have_p = 1;
4722                         ptr++;
4723                         rc++;
4724                 }
4725                 if (*ptr == 'c') {
4726                         have_c = 1;
4727                         ptr++;
4728                         rc++;
4729                 }
4730         }
4731
4732         if (arg->type == TEP_PRINT_FUNC) {
4733                 process_defined_func(s, data, size, event, arg);
4734                 return rc;
4735         }
4736
4737         if (arg->type != TEP_PRINT_FIELD) {
4738                 trace_seq_printf(s, "ARG TYPE NOT FIELD BUT %d", arg->type);
4739                 return rc;
4740         }
4741
4742         if (!arg->field.field) {
4743                 arg->field.field =
4744                         tep_find_any_field(event, arg->field.name);
4745                 if (!arg->field.field) {
4746                         do_warning("%s: field %s not found",
4747                                    __func__, arg->field.name);
4748                         return rc;
4749                 }
4750         }
4751
4752         sa = (struct sockaddr_storage *) (data + arg->field.field->offset);
4753
4754         if (sa->ss_family == AF_INET) {
4755                 struct sockaddr_in *sa4 = (struct sockaddr_in *) sa;
4756
4757                 if (arg->field.field->size < sizeof(struct sockaddr_in)) {
4758                         trace_seq_printf(s, "INVALIDIPv4");
4759                         return rc;
4760                 }
4761
4762                 print_ip4_addr(s, i, (unsigned char *) &sa4->sin_addr);
4763                 if (have_p)
4764                         trace_seq_printf(s, ":%d", ntohs(sa4->sin_port));
4765
4766
4767         } else if (sa->ss_family == AF_INET6) {
4768                 struct sockaddr_in6 *sa6 = (struct sockaddr_in6 *) sa;
4769
4770                 if (arg->field.field->size < sizeof(struct sockaddr_in6)) {
4771                         trace_seq_printf(s, "INVALIDIPv6");
4772                         return rc;
4773                 }
4774
4775                 if (have_p)
4776                         trace_seq_printf(s, "[");
4777
4778                 buf = (unsigned char *) &sa6->sin6_addr;
4779                 if (have_c)
4780                         print_ip6c_addr(s, buf);
4781                 else
4782                         print_ip6_addr(s, i, buf);
4783
4784                 if (have_p)
4785                         trace_seq_printf(s, "]:%d", ntohs(sa6->sin6_port));
4786         }
4787
4788         return rc;
4789 }
4790
4791 static int print_ip_arg(struct trace_seq *s, const char *ptr,
4792                         void *data, int size, struct tep_event *event,
4793                         struct tep_print_arg *arg)
4794 {
4795         char i = *ptr;  /* 'i' or 'I' */
4796         char ver;
4797         int rc = 0;
4798
4799         ptr++;
4800         rc++;
4801
4802         ver = *ptr;
4803         ptr++;
4804         rc++;
4805
4806         switch (ver) {
4807         case '4':
4808                 rc += print_ipv4_arg(s, ptr, i, data, size, event, arg);
4809                 break;
4810         case '6':
4811                 rc += print_ipv6_arg(s, ptr, i, data, size, event, arg);
4812                 break;
4813         case 'S':
4814                 rc += print_ipsa_arg(s, ptr, i, data, size, event, arg);
4815                 break;
4816         default:
4817                 return 0;
4818         }
4819
4820         return rc;
4821 }
4822
4823 static int is_printable_array(char *p, unsigned int len)
4824 {
4825         unsigned int i;
4826
4827         for (i = 0; i < len && p[i]; i++)
4828                 if (!isprint(p[i]) && !isspace(p[i]))
4829                     return 0;
4830         return 1;
4831 }
4832
4833 void tep_print_field(struct trace_seq *s, void *data,
4834                      struct tep_format_field *field)
4835 {
4836         unsigned long long val;
4837         unsigned int offset, len, i;
4838         struct tep_handle *tep = field->event->tep;
4839
4840         if (field->flags & TEP_FIELD_IS_ARRAY) {
4841                 offset = field->offset;
4842                 len = field->size;
4843                 if (field->flags & TEP_FIELD_IS_DYNAMIC) {
4844                         val = tep_read_number(tep, data + offset, len);
4845                         offset = val;
4846                         len = offset >> 16;
4847                         offset &= 0xffff;
4848                 }
4849                 if (field->flags & TEP_FIELD_IS_STRING &&
4850                     is_printable_array(data + offset, len)) {
4851                         trace_seq_printf(s, "%s", (char *)data + offset);
4852                 } else {
4853                         trace_seq_puts(s, "ARRAY[");
4854                         for (i = 0; i < len; i++) {
4855                                 if (i)
4856                                         trace_seq_puts(s, ", ");
4857                                 trace_seq_printf(s, "%02x",
4858                                                  *((unsigned char *)data + offset + i));
4859                         }
4860                         trace_seq_putc(s, ']');
4861                         field->flags &= ~TEP_FIELD_IS_STRING;
4862                 }
4863         } else {
4864                 val = tep_read_number(tep, data + field->offset,
4865                                       field->size);
4866                 if (field->flags & TEP_FIELD_IS_POINTER) {
4867                         trace_seq_printf(s, "0x%llx", val);
4868                 } else if (field->flags & TEP_FIELD_IS_SIGNED) {
4869                         switch (field->size) {
4870                         case 4:
4871                                 /*
4872                                  * If field is long then print it in hex.
4873                                  * A long usually stores pointers.
4874                                  */
4875                                 if (field->flags & TEP_FIELD_IS_LONG)
4876                                         trace_seq_printf(s, "0x%x", (int)val);
4877                                 else
4878                                         trace_seq_printf(s, "%d", (int)val);
4879                                 break;
4880                         case 2:
4881                                 trace_seq_printf(s, "%2d", (short)val);
4882                                 break;
4883                         case 1:
4884                                 trace_seq_printf(s, "%1d", (char)val);
4885                                 break;
4886                         default:
4887                                 trace_seq_printf(s, "%lld", val);
4888                         }
4889                 } else {
4890                         if (field->flags & TEP_FIELD_IS_LONG)
4891                                 trace_seq_printf(s, "0x%llx", val);
4892                         else
4893                                 trace_seq_printf(s, "%llu", val);
4894                 }
4895         }
4896 }
4897
4898 void tep_print_fields(struct trace_seq *s, void *data,
4899                       int size __maybe_unused, struct tep_event *event)
4900 {
4901         struct tep_format_field *field;
4902
4903         field = event->format.fields;
4904         while (field) {
4905                 trace_seq_printf(s, " %s=", field->name);
4906                 tep_print_field(s, data, field);
4907                 field = field->next;
4908         }
4909 }
4910
4911 static void pretty_print(struct trace_seq *s, void *data, int size, struct tep_event *event)
4912 {
4913         struct tep_handle *tep = event->tep;
4914         struct tep_print_fmt *print_fmt = &event->print_fmt;
4915         struct tep_print_arg *arg = print_fmt->args;
4916         struct tep_print_arg *args = NULL;
4917         const char *ptr = print_fmt->format;
4918         unsigned long long val;
4919         struct func_map *func;
4920         const char *saveptr;
4921         struct trace_seq p;
4922         char *bprint_fmt = NULL;
4923         char format[32];
4924         int show_func;
4925         int len_as_arg;
4926         int len_arg = 0;
4927         int len;
4928         int ls;
4929
4930         if (event->flags & TEP_EVENT_FL_FAILED) {
4931                 trace_seq_printf(s, "[FAILED TO PARSE]");
4932                 tep_print_fields(s, data, size, event);
4933                 return;
4934         }
4935
4936         if (event->flags & TEP_EVENT_FL_ISBPRINT) {
4937                 bprint_fmt = get_bprint_format(data, size, event);
4938                 args = make_bprint_args(bprint_fmt, data, size, event);
4939                 arg = args;
4940                 ptr = bprint_fmt;
4941         }
4942
4943         for (; *ptr; ptr++) {
4944                 ls = 0;
4945                 if (*ptr == '\\') {
4946                         ptr++;
4947                         switch (*ptr) {
4948                         case 'n':
4949                                 trace_seq_putc(s, '\n');
4950                                 break;
4951                         case 't':
4952                                 trace_seq_putc(s, '\t');
4953                                 break;
4954                         case 'r':
4955                                 trace_seq_putc(s, '\r');
4956                                 break;
4957                         case '\\':
4958                                 trace_seq_putc(s, '\\');
4959                                 break;
4960                         default:
4961                                 trace_seq_putc(s, *ptr);
4962                                 break;
4963                         }
4964
4965                 } else if (*ptr == '%') {
4966                         saveptr = ptr;
4967                         show_func = 0;
4968                         len_as_arg = 0;
4969  cont_process:
4970                         ptr++;
4971                         switch (*ptr) {
4972                         case '%':
4973                                 trace_seq_putc(s, '%');
4974                                 break;
4975                         case '#':
4976                                 /* FIXME: need to handle properly */
4977                                 goto cont_process;
4978                         case 'h':
4979                                 ls--;
4980                                 goto cont_process;
4981                         case 'l':
4982                                 ls++;
4983                                 goto cont_process;
4984                         case 'L':
4985                                 ls = 2;
4986                                 goto cont_process;
4987                         case '*':
4988                                 /* The argument is the length. */
4989                                 if (!arg) {
4990                                         do_warning_event(event, "no argument match");
4991                                         event->flags |= TEP_EVENT_FL_FAILED;
4992                                         goto out_failed;
4993                                 }
4994                                 len_arg = eval_num_arg(data, size, event, arg);
4995                                 len_as_arg = 1;
4996                                 arg = arg->next;
4997                                 goto cont_process;
4998                         case '.':
4999                         case 'z':
5000                         case 'Z':
5001                         case '0' ... '9':
5002                         case '-':
5003                                 goto cont_process;
5004                         case 'p':
5005                                 if (tep->long_size == 4)
5006                                         ls = 1;
5007                                 else
5008                                         ls = 2;
5009
5010                                 if (isalnum(ptr[1]))
5011                                         ptr++;
5012
5013                                 if (arg->type == TEP_PRINT_BSTRING) {
5014                                         trace_seq_puts(s, arg->string.string);
5015                                         arg = arg->next;
5016                                         break;
5017                                 }
5018
5019                                 if (*ptr == 'F' || *ptr == 'f' ||
5020                                     *ptr == 'S' || *ptr == 's') {
5021                                         show_func = *ptr;
5022                                 } else if (*ptr == 'M' || *ptr == 'm') {
5023                                         print_mac_arg(s, *ptr, data, size, event, arg);
5024                                         arg = arg->next;
5025                                         break;
5026                                 } else if (*ptr == 'I' || *ptr == 'i') {
5027                                         int n;
5028
5029                                         n = print_ip_arg(s, ptr, data, size, event, arg);
5030                                         if (n > 0) {
5031                                                 ptr += n - 1;
5032                                                 arg = arg->next;
5033                                                 break;
5034                                         }
5035                                 }
5036
5037                                 /* fall through */
5038                         case 'd':
5039                         case 'i':
5040                         case 'x':
5041                         case 'X':
5042                         case 'u':
5043                                 if (!arg) {
5044                                         do_warning_event(event, "no argument match");
5045                                         event->flags |= TEP_EVENT_FL_FAILED;
5046                                         goto out_failed;
5047                                 }
5048
5049                                 len = ((unsigned long)ptr + 1) -
5050                                         (unsigned long)saveptr;
5051
5052                                 /* should never happen */
5053                                 if (len > 31) {
5054                                         do_warning_event(event, "bad format!");
5055                                         event->flags |= TEP_EVENT_FL_FAILED;
5056                                         len = 31;
5057                                 }
5058
5059                                 memcpy(format, saveptr, len);
5060                                 format[len] = 0;
5061
5062                                 val = eval_num_arg(data, size, event, arg);
5063                                 arg = arg->next;
5064
5065                                 if (show_func) {
5066                                         func = find_func(tep, val);
5067                                         if (func) {
5068                                                 trace_seq_puts(s, func->func);
5069                                                 if (show_func == 'F')
5070                                                         trace_seq_printf(s,
5071                                                                "+0x%llx",
5072                                                                val - func->addr);
5073                                                 break;
5074                                         }
5075                                 }
5076                                 if (tep->long_size == 8 && ls == 1 &&
5077                                     sizeof(long) != 8) {
5078                                         char *p;
5079
5080                                         /* make %l into %ll */
5081                                         if (ls == 1 && (p = strchr(format, 'l')))
5082                                                 memmove(p+1, p, strlen(p)+1);
5083                                         else if (strcmp(format, "%p") == 0)
5084                                                 strcpy(format, "0x%llx");
5085                                         ls = 2;
5086                                 }
5087                                 switch (ls) {
5088                                 case -2:
5089                                         if (len_as_arg)
5090                                                 trace_seq_printf(s, format, len_arg, (char)val);
5091                                         else
5092                                                 trace_seq_printf(s, format, (char)val);
5093                                         break;
5094                                 case -1:
5095                                         if (len_as_arg)
5096                                                 trace_seq_printf(s, format, len_arg, (short)val);
5097                                         else
5098                                                 trace_seq_printf(s, format, (short)val);
5099                                         break;
5100                                 case 0:
5101                                         if (len_as_arg)
5102                                                 trace_seq_printf(s, format, len_arg, (int)val);
5103                                         else
5104                                                 trace_seq_printf(s, format, (int)val);
5105                                         break;
5106                                 case 1:
5107                                         if (len_as_arg)
5108                                                 trace_seq_printf(s, format, len_arg, (long)val);
5109                                         else
5110                                                 trace_seq_printf(s, format, (long)val);
5111                                         break;
5112                                 case 2:
5113                                         if (len_as_arg)
5114                                                 trace_seq_printf(s, format, len_arg,
5115                                                                  (long long)val);
5116                                         else
5117                                                 trace_seq_printf(s, format, (long long)val);
5118                                         break;
5119                                 default:
5120                                         do_warning_event(event, "bad count (%d)", ls);
5121                                         event->flags |= TEP_EVENT_FL_FAILED;
5122                                 }
5123                                 break;
5124                         case 's':
5125                                 if (!arg) {
5126                                         do_warning_event(event, "no matching argument");
5127                                         event->flags |= TEP_EVENT_FL_FAILED;
5128                                         goto out_failed;
5129                                 }
5130
5131                                 len = ((unsigned long)ptr + 1) -
5132                                         (unsigned long)saveptr;
5133
5134                                 /* should never happen */
5135                                 if (len > 31) {
5136                                         do_warning_event(event, "bad format!");
5137                                         event->flags |= TEP_EVENT_FL_FAILED;
5138                                         len = 31;
5139                                 }
5140
5141                                 memcpy(format, saveptr, len);
5142                                 format[len] = 0;
5143                                 if (!len_as_arg)
5144                                         len_arg = -1;
5145                                 /* Use helper trace_seq */
5146                                 trace_seq_init(&p);
5147                                 print_str_arg(&p, data, size, event,
5148                                               format, len_arg, arg);
5149                                 trace_seq_terminate(&p);
5150                                 trace_seq_puts(s, p.buffer);
5151                                 trace_seq_destroy(&p);
5152                                 arg = arg->next;
5153                                 break;
5154                         default:
5155                                 trace_seq_printf(s, ">%c<", *ptr);
5156
5157                         }
5158                 } else
5159                         trace_seq_putc(s, *ptr);
5160         }
5161
5162         if (event->flags & TEP_EVENT_FL_FAILED) {
5163 out_failed:
5164                 trace_seq_printf(s, "[FAILED TO PARSE]");
5165         }
5166
5167         if (args) {
5168                 free_args(args);
5169                 free(bprint_fmt);
5170         }
5171 }
5172
5173 /**
5174  * tep_data_latency_format - parse the data for the latency format
5175  * @tep: a handle to the trace event parser context
5176  * @s: the trace_seq to write to
5177  * @record: the record to read from
5178  *
5179  * This parses out the Latency format (interrupts disabled,
5180  * need rescheduling, in hard/soft interrupt, preempt count
5181  * and lock depth) and places it into the trace_seq.
5182  */
5183 void tep_data_latency_format(struct tep_handle *tep,
5184                              struct trace_seq *s, struct tep_record *record)
5185 {
5186         static int check_lock_depth = 1;
5187         static int check_migrate_disable = 1;
5188         static int lock_depth_exists;
5189         static int migrate_disable_exists;
5190         unsigned int lat_flags;
5191         unsigned int pc;
5192         int lock_depth = 0;
5193         int migrate_disable = 0;
5194         int hardirq;
5195         int softirq;
5196         void *data = record->data;
5197
5198         lat_flags = parse_common_flags(tep, data);
5199         pc = parse_common_pc(tep, data);
5200         /* lock_depth may not always exist */
5201         if (lock_depth_exists)
5202                 lock_depth = parse_common_lock_depth(tep, data);
5203         else if (check_lock_depth) {
5204                 lock_depth = parse_common_lock_depth(tep, data);
5205                 if (lock_depth < 0)
5206                         check_lock_depth = 0;
5207                 else
5208                         lock_depth_exists = 1;
5209         }
5210
5211         /* migrate_disable may not always exist */
5212         if (migrate_disable_exists)
5213                 migrate_disable = parse_common_migrate_disable(tep, data);
5214         else if (check_migrate_disable) {
5215                 migrate_disable = parse_common_migrate_disable(tep, data);
5216                 if (migrate_disable < 0)
5217                         check_migrate_disable = 0;
5218                 else
5219                         migrate_disable_exists = 1;
5220         }
5221
5222         hardirq = lat_flags & TRACE_FLAG_HARDIRQ;
5223         softirq = lat_flags & TRACE_FLAG_SOFTIRQ;
5224
5225         trace_seq_printf(s, "%c%c%c",
5226                (lat_flags & TRACE_FLAG_IRQS_OFF) ? 'd' :
5227                (lat_flags & TRACE_FLAG_IRQS_NOSUPPORT) ?
5228                'X' : '.',
5229                (lat_flags & TRACE_FLAG_NEED_RESCHED) ?
5230                'N' : '.',
5231                (hardirq && softirq) ? 'H' :
5232                hardirq ? 'h' : softirq ? 's' : '.');
5233
5234         if (pc)
5235                 trace_seq_printf(s, "%x", pc);
5236         else
5237                 trace_seq_putc(s, '.');
5238
5239         if (migrate_disable_exists) {
5240                 if (migrate_disable < 0)
5241                         trace_seq_putc(s, '.');
5242                 else
5243                         trace_seq_printf(s, "%d", migrate_disable);
5244         }
5245
5246         if (lock_depth_exists) {
5247                 if (lock_depth < 0)
5248                         trace_seq_putc(s, '.');
5249                 else
5250                         trace_seq_printf(s, "%d", lock_depth);
5251         }
5252
5253         trace_seq_terminate(s);
5254 }
5255
5256 /**
5257  * tep_data_type - parse out the given event type
5258  * @tep: a handle to the trace event parser context
5259  * @rec: the record to read from
5260  *
5261  * This returns the event id from the @rec.
5262  */
5263 int tep_data_type(struct tep_handle *tep, struct tep_record *rec)
5264 {
5265         return trace_parse_common_type(tep, rec->data);
5266 }
5267
5268 /**
5269  * tep_data_pid - parse the PID from record
5270  * @tep: a handle to the trace event parser context
5271  * @rec: the record to parse
5272  *
5273  * This returns the PID from a record.
5274  */
5275 int tep_data_pid(struct tep_handle *tep, struct tep_record *rec)
5276 {
5277         return parse_common_pid(tep, rec->data);
5278 }
5279
5280 /**
5281  * tep_data_preempt_count - parse the preempt count from the record
5282  * @tep: a handle to the trace event parser context
5283  * @rec: the record to parse
5284  *
5285  * This returns the preempt count from a record.
5286  */
5287 int tep_data_preempt_count(struct tep_handle *tep, struct tep_record *rec)
5288 {
5289         return parse_common_pc(tep, rec->data);
5290 }
5291
5292 /**
5293  * tep_data_flags - parse the latency flags from the record
5294  * @tep: a handle to the trace event parser context
5295  * @rec: the record to parse
5296  *
5297  * This returns the latency flags from a record.
5298  *
5299  *  Use trace_flag_type enum for the flags (see event-parse.h).
5300  */
5301 int tep_data_flags(struct tep_handle *tep, struct tep_record *rec)
5302 {
5303         return parse_common_flags(tep, rec->data);
5304 }
5305
5306 /**
5307  * tep_data_comm_from_pid - return the command line from PID
5308  * @tep: a handle to the trace event parser context
5309  * @pid: the PID of the task to search for
5310  *
5311  * This returns a pointer to the command line that has the given
5312  * @pid.
5313  */
5314 const char *tep_data_comm_from_pid(struct tep_handle *tep, int pid)
5315 {
5316         const char *comm;
5317
5318         comm = find_cmdline(tep, pid);
5319         return comm;
5320 }
5321
5322 static struct tep_cmdline *
5323 pid_from_cmdlist(struct tep_handle *tep, const char *comm, struct tep_cmdline *next)
5324 {
5325         struct cmdline_list *cmdlist = (struct cmdline_list *)next;
5326
5327         if (cmdlist)
5328                 cmdlist = cmdlist->next;
5329         else
5330                 cmdlist = tep->cmdlist;
5331
5332         while (cmdlist && strcmp(cmdlist->comm, comm) != 0)
5333                 cmdlist = cmdlist->next;
5334
5335         return (struct tep_cmdline *)cmdlist;
5336 }
5337
5338 /**
5339  * tep_data_pid_from_comm - return the pid from a given comm
5340  * @tep: a handle to the trace event parser context
5341  * @comm: the cmdline to find the pid from
5342  * @next: the cmdline structure to find the next comm
5343  *
5344  * This returns the cmdline structure that holds a pid for a given
5345  * comm, or NULL if none found. As there may be more than one pid for
5346  * a given comm, the result of this call can be passed back into
5347  * a recurring call in the @next parameter, and then it will find the
5348  * next pid.
5349  * Also, it does a linear search, so it may be slow.
5350  */
5351 struct tep_cmdline *tep_data_pid_from_comm(struct tep_handle *tep, const char *comm,
5352                                            struct tep_cmdline *next)
5353 {
5354         struct tep_cmdline *cmdline;
5355
5356         /*
5357          * If the cmdlines have not been converted yet, then use
5358          * the list.
5359          */
5360         if (!tep->cmdlines)
5361                 return pid_from_cmdlist(tep, comm, next);
5362
5363         if (next) {
5364                 /*
5365                  * The next pointer could have been still from
5366                  * a previous call before cmdlines were created
5367                  */
5368                 if (next < tep->cmdlines ||
5369                     next >= tep->cmdlines + tep->cmdline_count)
5370                         next = NULL;
5371                 else
5372                         cmdline  = next++;
5373         }
5374
5375         if (!next)
5376                 cmdline = tep->cmdlines;
5377
5378         while (cmdline < tep->cmdlines + tep->cmdline_count) {
5379                 if (strcmp(cmdline->comm, comm) == 0)
5380                         return cmdline;
5381                 cmdline++;
5382         }
5383         return NULL;
5384 }
5385
5386 /**
5387  * tep_cmdline_pid - return the pid associated to a given cmdline
5388  * @tep: a handle to the trace event parser context
5389  * @cmdline: The cmdline structure to get the pid from
5390  *
5391  * Returns the pid for a give cmdline. If @cmdline is NULL, then
5392  * -1 is returned.
5393  */
5394 int tep_cmdline_pid(struct tep_handle *tep, struct tep_cmdline *cmdline)
5395 {
5396         struct cmdline_list *cmdlist = (struct cmdline_list *)cmdline;
5397
5398         if (!cmdline)
5399                 return -1;
5400
5401         /*
5402          * If cmdlines have not been created yet, or cmdline is
5403          * not part of the array, then treat it as a cmdlist instead.
5404          */
5405         if (!tep->cmdlines ||
5406             cmdline < tep->cmdlines ||
5407             cmdline >= tep->cmdlines + tep->cmdline_count)
5408                 return cmdlist->pid;
5409
5410         return cmdline->pid;
5411 }
5412
5413 /**
5414  * tep_event_info - parse the data into the print format
5415  * @s: the trace_seq to write to
5416  * @event: the handle to the event
5417  * @record: the record to read from
5418  *
5419  * This parses the raw @data using the given @event information and
5420  * writes the print format into the trace_seq.
5421  */
5422 void tep_event_info(struct trace_seq *s, struct tep_event *event,
5423                     struct tep_record *record)
5424 {
5425         int print_pretty = 1;
5426
5427         if (event->tep->print_raw || (event->flags & TEP_EVENT_FL_PRINTRAW))
5428                 tep_print_fields(s, record->data, record->size, event);
5429         else {
5430
5431                 if (event->handler && !(event->flags & TEP_EVENT_FL_NOHANDLE))
5432                         print_pretty = event->handler(s, record, event,
5433                                                       event->context);
5434
5435                 if (print_pretty)
5436                         pretty_print(s, record->data, record->size, event);
5437         }
5438
5439         trace_seq_terminate(s);
5440 }
5441
5442 static bool is_timestamp_in_us(char *trace_clock, bool use_trace_clock)
5443 {
5444         if (!trace_clock || !use_trace_clock)
5445                 return true;
5446
5447         if (!strcmp(trace_clock, "local") || !strcmp(trace_clock, "global")
5448             || !strcmp(trace_clock, "uptime") || !strcmp(trace_clock, "perf")
5449             || !strncmp(trace_clock, "mono", 4))
5450                 return true;
5451
5452         /* trace_clock is setting in tsc or counter mode */
5453         return false;
5454 }
5455
5456 /**
5457  * tep_find_event_by_record - return the event from a given record
5458  * @tep: a handle to the trace event parser context
5459  * @record: The record to get the event from
5460  *
5461  * Returns the associated event for a given record, or NULL if non is
5462  * is found.
5463  */
5464 struct tep_event *
5465 tep_find_event_by_record(struct tep_handle *tep, struct tep_record *record)
5466 {
5467         int type;
5468
5469         if (record->size < 0) {
5470                 do_warning("ug! negative record size %d", record->size);
5471                 return NULL;
5472         }
5473
5474         type = trace_parse_common_type(tep, record->data);
5475
5476         return tep_find_event(tep, type);
5477 }
5478
5479 /**
5480  * tep_print_event_task - Write the event task comm, pid and CPU
5481  * @tep: a handle to the trace event parser context
5482  * @s: the trace_seq to write to
5483  * @event: the handle to the record's event
5484  * @record: The record to get the event from
5485  *
5486  * Writes the tasks comm, pid and CPU to @s.
5487  */
5488 void tep_print_event_task(struct tep_handle *tep, struct trace_seq *s,
5489                           struct tep_event *event,
5490                           struct tep_record *record)
5491 {
5492         void *data = record->data;
5493         const char *comm;
5494         int pid;
5495
5496         pid = parse_common_pid(tep, data);
5497         comm = find_cmdline(tep, pid);
5498
5499         if (tep->latency_format)
5500                 trace_seq_printf(s, "%8.8s-%-5d %3d", comm, pid, record->cpu);
5501         else
5502                 trace_seq_printf(s, "%16s-%-5d [%03d]", comm, pid, record->cpu);
5503 }
5504
5505 /**
5506  * tep_print_event_time - Write the event timestamp
5507  * @tep: a handle to the trace event parser context
5508  * @s: the trace_seq to write to
5509  * @event: the handle to the record's event
5510  * @record: The record to get the event from
5511  * @use_trace_clock: Set to parse according to the @tep->trace_clock
5512  *
5513  * Writes the timestamp of the record into @s.
5514  */
5515 void tep_print_event_time(struct tep_handle *tep, struct trace_seq *s,
5516                           struct tep_event *event,
5517                           struct tep_record *record,
5518                           bool use_trace_clock)
5519 {
5520         unsigned long secs;
5521         unsigned long usecs;
5522         unsigned long nsecs;
5523         int p;
5524         bool use_usec_format;
5525
5526         use_usec_format = is_timestamp_in_us(tep->trace_clock, use_trace_clock);
5527         if (use_usec_format) {
5528                 secs = record->ts / NSEC_PER_SEC;
5529                 nsecs = record->ts - secs * NSEC_PER_SEC;
5530         }
5531
5532         if (tep->latency_format) {
5533                 tep_data_latency_format(tep, s, record);
5534         }
5535
5536         if (use_usec_format) {
5537                 if (tep->flags & TEP_NSEC_OUTPUT) {
5538                         usecs = nsecs;
5539                         p = 9;
5540                 } else {
5541                         usecs = (nsecs + 500) / NSEC_PER_USEC;
5542                         /* To avoid usecs larger than 1 sec */
5543                         if (usecs >= USEC_PER_SEC) {
5544                                 usecs -= USEC_PER_SEC;
5545                                 secs++;
5546                         }
5547                         p = 6;
5548                 }
5549
5550                 trace_seq_printf(s, " %5lu.%0*lu:", secs, p, usecs);
5551         } else
5552                 trace_seq_printf(s, " %12llu:", record->ts);
5553 }
5554
5555 /**
5556  * tep_print_event_data - Write the event data section
5557  * @tep: a handle to the trace event parser context
5558  * @s: the trace_seq to write to
5559  * @event: the handle to the record's event
5560  * @record: The record to get the event from
5561  *
5562  * Writes the parsing of the record's data to @s.
5563  */
5564 void tep_print_event_data(struct tep_handle *tep, struct trace_seq *s,
5565                           struct tep_event *event,
5566                           struct tep_record *record)
5567 {
5568         static const char *spaces = "                    "; /* 20 spaces */
5569         int len;
5570
5571         trace_seq_printf(s, " %s: ", event->name);
5572
5573         /* Space out the event names evenly. */
5574         len = strlen(event->name);
5575         if (len < 20)
5576                 trace_seq_printf(s, "%.*s", 20 - len, spaces);
5577
5578         tep_event_info(s, event, record);
5579 }
5580
5581 void tep_print_event(struct tep_handle *tep, struct trace_seq *s,
5582                      struct tep_record *record, bool use_trace_clock)
5583 {
5584         struct tep_event *event;
5585
5586         event = tep_find_event_by_record(tep, record);
5587         if (!event) {
5588                 int i;
5589                 int type = trace_parse_common_type(tep, record->data);
5590
5591                 do_warning("ug! no event found for type %d", type);
5592                 trace_seq_printf(s, "[UNKNOWN TYPE %d]", type);
5593                 for (i = 0; i < record->size; i++)
5594                         trace_seq_printf(s, " %02x",
5595                                          ((unsigned char *)record->data)[i]);
5596                 return;
5597         }
5598
5599         tep_print_event_task(tep, s, event, record);
5600         tep_print_event_time(tep, s, event, record, use_trace_clock);
5601         tep_print_event_data(tep, s, event, record);
5602 }
5603
5604 static int events_id_cmp(const void *a, const void *b)
5605 {
5606         struct tep_event * const * ea = a;
5607         struct tep_event * const * eb = b;
5608
5609         if ((*ea)->id < (*eb)->id)
5610                 return -1;
5611
5612         if ((*ea)->id > (*eb)->id)
5613                 return 1;
5614
5615         return 0;
5616 }
5617
5618 static int events_name_cmp(const void *a, const void *b)
5619 {
5620         struct tep_event * const * ea = a;
5621         struct tep_event * const * eb = b;
5622         int res;
5623
5624         res = strcmp((*ea)->name, (*eb)->name);
5625         if (res)
5626                 return res;
5627
5628         res = strcmp((*ea)->system, (*eb)->system);
5629         if (res)
5630                 return res;
5631
5632         return events_id_cmp(a, b);
5633 }
5634
5635 static int events_system_cmp(const void *a, const void *b)
5636 {
5637         struct tep_event * const * ea = a;
5638         struct tep_event * const * eb = b;
5639         int res;
5640
5641         res = strcmp((*ea)->system, (*eb)->system);
5642         if (res)
5643                 return res;
5644
5645         res = strcmp((*ea)->name, (*eb)->name);
5646         if (res)
5647                 return res;
5648
5649         return events_id_cmp(a, b);
5650 }
5651
5652 static struct tep_event **list_events_copy(struct tep_handle *tep)
5653 {
5654         struct tep_event **events;
5655
5656         if (!tep)
5657                 return NULL;
5658
5659         events = malloc(sizeof(*events) * (tep->nr_events + 1));
5660         if (!events)
5661                 return NULL;
5662
5663         memcpy(events, tep->events, sizeof(*events) * tep->nr_events);
5664         events[tep->nr_events] = NULL;
5665         return events;
5666 }
5667
5668 static void list_events_sort(struct tep_event **events, int nr_events,
5669                              enum tep_event_sort_type sort_type)
5670 {
5671         int (*sort)(const void *a, const void *b);
5672
5673         switch (sort_type) {
5674         case TEP_EVENT_SORT_ID:
5675                 sort = events_id_cmp;
5676                 break;
5677         case TEP_EVENT_SORT_NAME:
5678                 sort = events_name_cmp;
5679                 break;
5680         case TEP_EVENT_SORT_SYSTEM:
5681                 sort = events_system_cmp;
5682                 break;
5683         default:
5684                 sort = NULL;
5685         }
5686
5687         if (sort)
5688                 qsort(events, nr_events, sizeof(*events), sort);
5689 }
5690
5691 /**
5692  * tep_list_events - Get events, sorted by given criteria.
5693  * @tep: a handle to the tep context
5694  * @sort_type: desired sort order of the events in the array
5695  *
5696  * Returns an array of pointers to all events, sorted by the given
5697  * @sort_type criteria. The last element of the array is NULL. The returned
5698  * memory must not be freed, it is managed by the library.
5699  * The function is not thread safe.
5700  */
5701 struct tep_event **tep_list_events(struct tep_handle *tep,
5702                                    enum tep_event_sort_type sort_type)
5703 {
5704         struct tep_event **events;
5705
5706         if (!tep)
5707                 return NULL;
5708
5709         events = tep->sort_events;
5710         if (events && tep->last_type == sort_type)
5711                 return events;
5712
5713         if (!events) {
5714                 events = list_events_copy(tep);
5715                 if (!events)
5716                         return NULL;
5717
5718                 tep->sort_events = events;
5719
5720                 /* the internal events are sorted by id */
5721                 if (sort_type == TEP_EVENT_SORT_ID) {
5722                         tep->last_type = sort_type;
5723                         return events;
5724                 }
5725         }
5726
5727         list_events_sort(events, tep->nr_events, sort_type);
5728         tep->last_type = sort_type;
5729
5730         return events;
5731 }
5732
5733
5734 /**
5735  * tep_list_events_copy - Thread safe version of tep_list_events()
5736  * @tep: a handle to the tep context
5737  * @sort_type: desired sort order of the events in the array
5738  *
5739  * Returns an array of pointers to all events, sorted by the given
5740  * @sort_type criteria. The last element of the array is NULL. The returned
5741  * array is newly allocated inside the function and must be freed by the caller
5742  */
5743 struct tep_event **tep_list_events_copy(struct tep_handle *tep,
5744                                         enum tep_event_sort_type sort_type)
5745 {
5746         struct tep_event **events;
5747
5748         if (!tep)
5749                 return NULL;
5750
5751         events = list_events_copy(tep);
5752         if (!events)
5753                 return NULL;
5754
5755         /* the internal events are sorted by id */
5756         if (sort_type == TEP_EVENT_SORT_ID)
5757                 return events;
5758
5759         list_events_sort(events, tep->nr_events, sort_type);
5760
5761         return events;
5762 }
5763
5764 static struct tep_format_field **
5765 get_event_fields(const char *type, const char *name,
5766                  int count, struct tep_format_field *list)
5767 {
5768         struct tep_format_field **fields;
5769         struct tep_format_field *field;
5770         int i = 0;
5771
5772         fields = malloc(sizeof(*fields) * (count + 1));
5773         if (!fields)
5774                 return NULL;
5775
5776         for (field = list; field; field = field->next) {
5777                 fields[i++] = field;
5778                 if (i == count + 1) {
5779                         do_warning("event %s has more %s fields than specified",
5780                                 name, type);
5781                         i--;
5782                         break;
5783                 }
5784         }
5785
5786         if (i != count)
5787                 do_warning("event %s has less %s fields than specified",
5788                         name, type);
5789
5790         fields[i] = NULL;
5791
5792         return fields;
5793 }
5794
5795 /**
5796  * tep_event_common_fields - return a list of common fields for an event
5797  * @event: the event to return the common fields of.
5798  *
5799  * Returns an allocated array of fields. The last item in the array is NULL.
5800  * The array must be freed with free().
5801  */
5802 struct tep_format_field **tep_event_common_fields(struct tep_event *event)
5803 {
5804         return get_event_fields("common", event->name,
5805                                 event->format.nr_common,
5806                                 event->format.common_fields);
5807 }
5808
5809 /**
5810  * tep_event_fields - return a list of event specific fields for an event
5811  * @event: the event to return the fields of.
5812  *
5813  * Returns an allocated array of fields. The last item in the array is NULL.
5814  * The array must be freed with free().
5815  */
5816 struct tep_format_field **tep_event_fields(struct tep_event *event)
5817 {
5818         return get_event_fields("event", event->name,
5819                                 event->format.nr_fields,
5820                                 event->format.fields);
5821 }
5822
5823 static void print_fields(struct trace_seq *s, struct tep_print_flag_sym *field)
5824 {
5825         trace_seq_printf(s, "{ %s, %s }", field->value, field->str);
5826         if (field->next) {
5827                 trace_seq_puts(s, ", ");
5828                 print_fields(s, field->next);
5829         }
5830 }
5831
5832 /* for debugging */
5833 static void print_args(struct tep_print_arg *args)
5834 {
5835         int print_paren = 1;
5836         struct trace_seq s;
5837
5838         switch (args->type) {
5839         case TEP_PRINT_NULL:
5840                 printf("null");
5841                 break;
5842         case TEP_PRINT_ATOM:
5843                 printf("%s", args->atom.atom);
5844                 break;
5845         case TEP_PRINT_FIELD:
5846                 printf("REC->%s", args->field.name);
5847                 break;
5848         case TEP_PRINT_FLAGS:
5849                 printf("__print_flags(");
5850                 print_args(args->flags.field);
5851                 printf(", %s, ", args->flags.delim);
5852                 trace_seq_init(&s);
5853                 print_fields(&s, args->flags.flags);
5854                 trace_seq_do_printf(&s);
5855                 trace_seq_destroy(&s);
5856                 printf(")");
5857                 break;
5858         case TEP_PRINT_SYMBOL:
5859                 printf("__print_symbolic(");
5860                 print_args(args->symbol.field);
5861                 printf(", ");
5862                 trace_seq_init(&s);
5863                 print_fields(&s, args->symbol.symbols);
5864                 trace_seq_do_printf(&s);
5865                 trace_seq_destroy(&s);
5866                 printf(")");
5867                 break;
5868         case TEP_PRINT_HEX:
5869                 printf("__print_hex(");
5870                 print_args(args->hex.field);
5871                 printf(", ");
5872                 print_args(args->hex.size);
5873                 printf(")");
5874                 break;
5875         case TEP_PRINT_HEX_STR:
5876                 printf("__print_hex_str(");
5877                 print_args(args->hex.field);
5878                 printf(", ");
5879                 print_args(args->hex.size);
5880                 printf(")");
5881                 break;
5882         case TEP_PRINT_INT_ARRAY:
5883                 printf("__print_array(");
5884                 print_args(args->int_array.field);
5885                 printf(", ");
5886                 print_args(args->int_array.count);
5887                 printf(", ");
5888                 print_args(args->int_array.el_size);
5889                 printf(")");
5890                 break;
5891         case TEP_PRINT_STRING:
5892         case TEP_PRINT_BSTRING:
5893                 printf("__get_str(%s)", args->string.string);
5894                 break;
5895         case TEP_PRINT_BITMASK:
5896                 printf("__get_bitmask(%s)", args->bitmask.bitmask);
5897                 break;
5898         case TEP_PRINT_TYPE:
5899                 printf("(%s)", args->typecast.type);
5900                 print_args(args->typecast.item);
5901                 break;
5902         case TEP_PRINT_OP:
5903                 if (strcmp(args->op.op, ":") == 0)
5904                         print_paren = 0;
5905                 if (print_paren)
5906                         printf("(");
5907                 print_args(args->op.left);
5908                 printf(" %s ", args->op.op);
5909                 print_args(args->op.right);
5910                 if (print_paren)
5911                         printf(")");
5912                 break;
5913         default:
5914                 /* we should warn... */
5915                 return;
5916         }
5917         if (args->next) {
5918                 printf("\n");
5919                 print_args(args->next);
5920         }
5921 }
5922
5923 static void parse_header_field(const char *field,
5924                                int *offset, int *size, int mandatory)
5925 {
5926         unsigned long long save_input_buf_ptr;
5927         unsigned long long save_input_buf_siz;
5928         char *token;
5929         int type;
5930
5931         save_input_buf_ptr = input_buf_ptr;
5932         save_input_buf_siz = input_buf_siz;
5933
5934         if (read_expected(TEP_EVENT_ITEM, "field") < 0)
5935                 return;
5936         if (read_expected(TEP_EVENT_OP, ":") < 0)
5937                 return;
5938
5939         /* type */
5940         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
5941                 goto fail;
5942         free_token(token);
5943
5944         /*
5945          * If this is not a mandatory field, then test it first.
5946          */
5947         if (mandatory) {
5948                 if (read_expected(TEP_EVENT_ITEM, field) < 0)
5949                         return;
5950         } else {
5951                 if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
5952                         goto fail;
5953                 if (strcmp(token, field) != 0)
5954                         goto discard;
5955                 free_token(token);
5956         }
5957
5958         if (read_expected(TEP_EVENT_OP, ";") < 0)
5959                 return;
5960         if (read_expected(TEP_EVENT_ITEM, "offset") < 0)
5961                 return;
5962         if (read_expected(TEP_EVENT_OP, ":") < 0)
5963                 return;
5964         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
5965                 goto fail;
5966         *offset = atoi(token);
5967         free_token(token);
5968         if (read_expected(TEP_EVENT_OP, ";") < 0)
5969                 return;
5970         if (read_expected(TEP_EVENT_ITEM, "size") < 0)
5971                 return;
5972         if (read_expected(TEP_EVENT_OP, ":") < 0)
5973                 return;
5974         if (read_expect_type(TEP_EVENT_ITEM, &token) < 0)
5975                 goto fail;
5976         *size = atoi(token);
5977         free_token(token);
5978         if (read_expected(TEP_EVENT_OP, ";") < 0)
5979                 return;
5980         type = read_token(&token);
5981         if (type != TEP_EVENT_NEWLINE) {
5982                 /* newer versions of the kernel have a "signed" type */
5983                 if (type != TEP_EVENT_ITEM)
5984                         goto fail;
5985
5986                 if (strcmp(token, "signed") != 0)
5987                         goto fail;
5988
5989                 free_token(token);
5990
5991                 if (read_expected(TEP_EVENT_OP, ":") < 0)
5992                         return;
5993
5994                 if (read_expect_type(TEP_EVENT_ITEM, &token))
5995                         goto fail;
5996
5997                 free_token(token);
5998                 if (read_expected(TEP_EVENT_OP, ";") < 0)
5999                         return;
6000
6001                 if (read_expect_type(TEP_EVENT_NEWLINE, &token))
6002                         goto fail;
6003         }
6004  fail:
6005         free_token(token);
6006         return;
6007
6008  discard:
6009         input_buf_ptr = save_input_buf_ptr;
6010         input_buf_siz = save_input_buf_siz;
6011         *offset = 0;
6012         *size = 0;
6013         free_token(token);
6014 }
6015
6016 /**
6017  * tep_parse_header_page - parse the data stored in the header page
6018  * @tep: a handle to the trace event parser context
6019  * @buf: the buffer storing the header page format string
6020  * @size: the size of @buf
6021  * @long_size: the long size to use if there is no header
6022  *
6023  * This parses the header page format for information on the
6024  * ring buffer used. The @buf should be copied from
6025  *
6026  * /sys/kernel/debug/tracing/events/header_page
6027  */
6028 int tep_parse_header_page(struct tep_handle *tep, char *buf, unsigned long size,
6029                           int long_size)
6030 {
6031         int ignore;
6032
6033         if (!size) {
6034                 /*
6035                  * Old kernels did not have header page info.
6036                  * Sorry but we just use what we find here in user space.
6037                  */
6038                 tep->header_page_ts_size = sizeof(long long);
6039                 tep->header_page_size_size = long_size;
6040                 tep->header_page_data_offset = sizeof(long long) + long_size;
6041                 tep->old_format = 1;
6042                 return -1;
6043         }
6044         init_input_buf(buf, size);
6045
6046         parse_header_field("timestamp", &tep->header_page_ts_offset,
6047                            &tep->header_page_ts_size, 1);
6048         parse_header_field("commit", &tep->header_page_size_offset,
6049                            &tep->header_page_size_size, 1);
6050         parse_header_field("overwrite", &tep->header_page_overwrite,
6051                            &ignore, 0);
6052         parse_header_field("data", &tep->header_page_data_offset,
6053                            &tep->header_page_data_size, 1);
6054
6055         return 0;
6056 }
6057
6058 static int event_matches(struct tep_event *event,
6059                          int id, const char *sys_name,
6060                          const char *event_name)
6061 {
6062         if (id >= 0 && id != event->id)
6063                 return 0;
6064
6065         if (event_name && (strcmp(event_name, event->name) != 0))
6066                 return 0;
6067
6068         if (sys_name && (strcmp(sys_name, event->system) != 0))
6069                 return 0;
6070
6071         return 1;
6072 }
6073
6074 static void free_handler(struct event_handler *handle)
6075 {
6076         free((void *)handle->sys_name);
6077         free((void *)handle->event_name);
6078         free(handle);
6079 }
6080
6081 static int find_event_handle(struct tep_handle *tep, struct tep_event *event)
6082 {
6083         struct event_handler *handle, **next;
6084
6085         for (next = &tep->handlers; *next;
6086              next = &(*next)->next) {
6087                 handle = *next;
6088                 if (event_matches(event, handle->id,
6089                                   handle->sys_name,
6090                                   handle->event_name))
6091                         break;
6092         }
6093
6094         if (!(*next))
6095                 return 0;
6096
6097         pr_stat("overriding event (%d) %s:%s with new print handler",
6098                 event->id, event->system, event->name);
6099
6100         event->handler = handle->func;
6101         event->context = handle->context;
6102
6103         *next = handle->next;
6104         free_handler(handle);
6105
6106         return 1;
6107 }
6108
6109 /**
6110  * __tep_parse_format - parse the event format
6111  * @buf: the buffer storing the event format string
6112  * @size: the size of @buf
6113  * @sys: the system the event belongs to
6114  *
6115  * This parses the event format and creates an event structure
6116  * to quickly parse raw data for a given event.
6117  *
6118  * These files currently come from:
6119  *
6120  * /sys/kernel/debug/tracing/events/.../.../format
6121  */
6122 enum tep_errno __tep_parse_format(struct tep_event **eventp,
6123                                   struct tep_handle *tep, const char *buf,
6124                                   unsigned long size, const char *sys)
6125 {
6126         struct tep_event *event;
6127         int ret;
6128
6129         init_input_buf(buf, size);
6130
6131         *eventp = event = alloc_event();
6132         if (!event)
6133                 return TEP_ERRNO__MEM_ALLOC_FAILED;
6134
6135         event->name = event_read_name();
6136         if (!event->name) {
6137                 /* Bad event? */
6138                 ret = TEP_ERRNO__MEM_ALLOC_FAILED;
6139                 goto event_alloc_failed;
6140         }
6141
6142         if (strcmp(sys, "ftrace") == 0) {
6143                 event->flags |= TEP_EVENT_FL_ISFTRACE;
6144
6145                 if (strcmp(event->name, "bprint") == 0)
6146                         event->flags |= TEP_EVENT_FL_ISBPRINT;
6147         }
6148
6149         event->id = event_read_id();
6150         if (event->id < 0) {
6151                 ret = TEP_ERRNO__READ_ID_FAILED;
6152                 /*
6153                  * This isn't an allocation error actually.
6154                  * But as the ID is critical, just bail out.
6155                  */
6156                 goto event_alloc_failed;
6157         }
6158
6159         event->system = strdup(sys);
6160         if (!event->system) {
6161                 ret = TEP_ERRNO__MEM_ALLOC_FAILED;
6162                 goto event_alloc_failed;
6163         }
6164
6165         /* Add tep to event so that it can be referenced */
6166         event->tep = tep;
6167
6168         ret = event_read_format(event);
6169         if (ret < 0) {
6170                 ret = TEP_ERRNO__READ_FORMAT_FAILED;
6171                 goto event_parse_failed;
6172         }
6173
6174         /*
6175          * If the event has an override, don't print warnings if the event
6176          * print format fails to parse.
6177          */
6178         if (tep && find_event_handle(tep, event))
6179                 show_warning = 0;
6180
6181         ret = event_read_print(event);
6182         show_warning = 1;
6183
6184         if (ret < 0) {
6185                 ret = TEP_ERRNO__READ_PRINT_FAILED;
6186                 goto event_parse_failed;
6187         }
6188
6189         if (!ret && (event->flags & TEP_EVENT_FL_ISFTRACE)) {
6190                 struct tep_format_field *field;
6191                 struct tep_print_arg *arg, **list;
6192
6193                 /* old ftrace had no args */
6194                 list = &event->print_fmt.args;
6195                 for (field = event->format.fields; field; field = field->next) {
6196                         arg = alloc_arg();
6197                         if (!arg) {
6198                                 event->flags |= TEP_EVENT_FL_FAILED;
6199                                 return TEP_ERRNO__OLD_FTRACE_ARG_FAILED;
6200                         }
6201                         arg->type = TEP_PRINT_FIELD;
6202                         arg->field.name = strdup(field->name);
6203                         if (!arg->field.name) {
6204                                 event->flags |= TEP_EVENT_FL_FAILED;
6205                                 free_arg(arg);
6206                                 return TEP_ERRNO__OLD_FTRACE_ARG_FAILED;
6207                         }
6208                         arg->field.field = field;
6209                         *list = arg;
6210                         list = &arg->next;
6211                 }
6212                 return 0;
6213         }
6214
6215         return 0;
6216
6217  event_parse_failed:
6218         event->flags |= TEP_EVENT_FL_FAILED;
6219         return ret;
6220
6221  event_alloc_failed:
6222         free(event->system);
6223         free(event->name);
6224         free(event);
6225         *eventp = NULL;
6226         return ret;
6227 }
6228
6229 static enum tep_errno
6230 __parse_event(struct tep_handle *tep,
6231               struct tep_event **eventp,
6232               const char *buf, unsigned long size,
6233               const char *sys)
6234 {
6235         int ret = __tep_parse_format(eventp, tep, buf, size, sys);
6236         struct tep_event *event = *eventp;
6237
6238         if (event == NULL)
6239                 return ret;
6240
6241         if (tep && add_event(tep, event)) {
6242                 ret = TEP_ERRNO__MEM_ALLOC_FAILED;
6243                 goto event_add_failed;
6244         }
6245
6246 #define PRINT_ARGS 0
6247         if (PRINT_ARGS && event->print_fmt.args)
6248                 print_args(event->print_fmt.args);
6249
6250         return 0;
6251
6252 event_add_failed:
6253         tep_free_event(event);
6254         return ret;
6255 }
6256
6257 /**
6258  * tep_parse_format - parse the event format
6259  * @tep: a handle to the trace event parser context
6260  * @eventp: returned format
6261  * @buf: the buffer storing the event format string
6262  * @size: the size of @buf
6263  * @sys: the system the event belongs to
6264  *
6265  * This parses the event format and creates an event structure
6266  * to quickly parse raw data for a given event.
6267  *
6268  * These files currently come from:
6269  *
6270  * /sys/kernel/debug/tracing/events/.../.../format
6271  */
6272 enum tep_errno tep_parse_format(struct tep_handle *tep,
6273                                 struct tep_event **eventp,
6274                                 const char *buf,
6275                                 unsigned long size, const char *sys)
6276 {
6277         return __parse_event(tep, eventp, buf, size, sys);
6278 }
6279
6280 /**
6281  * tep_parse_event - parse the event format
6282  * @tep: a handle to the trace event parser context
6283  * @buf: the buffer storing the event format string
6284  * @size: the size of @buf
6285  * @sys: the system the event belongs to
6286  *
6287  * This parses the event format and creates an event structure
6288  * to quickly parse raw data for a given event.
6289  *
6290  * These files currently come from:
6291  *
6292  * /sys/kernel/debug/tracing/events/.../.../format
6293  */
6294 enum tep_errno tep_parse_event(struct tep_handle *tep, const char *buf,
6295                                unsigned long size, const char *sys)
6296 {
6297         struct tep_event *event = NULL;
6298         return __parse_event(tep, &event, buf, size, sys);
6299 }
6300
6301 int get_field_val(struct trace_seq *s, struct tep_format_field *field,
6302                   const char *name, struct tep_record *record,
6303                   unsigned long long *val, int err)
6304 {
6305         if (!field) {
6306                 if (err)
6307                         trace_seq_printf(s, "<CANT FIND FIELD %s>", name);
6308                 return -1;
6309         }
6310
6311         if (tep_read_number_field(field, record->data, val)) {
6312                 if (err)
6313                         trace_seq_printf(s, " %s=INVALID", name);
6314                 return -1;
6315         }
6316
6317         return 0;
6318 }
6319
6320 /**
6321  * tep_get_field_raw - return the raw pointer into the data field
6322  * @s: The seq to print to on error
6323  * @event: the event that the field is for
6324  * @name: The name of the field
6325  * @record: The record with the field name.
6326  * @len: place to store the field length.
6327  * @err: print default error if failed.
6328  *
6329  * Returns a pointer into record->data of the field and places
6330  * the length of the field in @len.
6331  *
6332  * On failure, it returns NULL.
6333  */
6334 void *tep_get_field_raw(struct trace_seq *s, struct tep_event *event,
6335                         const char *name, struct tep_record *record,
6336                         int *len, int err)
6337 {
6338         struct tep_format_field *field;
6339         void *data = record->data;
6340         unsigned offset;
6341         int dummy;
6342
6343         if (!event)
6344                 return NULL;
6345
6346         field = tep_find_field(event, name);
6347
6348         if (!field) {
6349                 if (err)
6350                         trace_seq_printf(s, "<CANT FIND FIELD %s>", name);
6351                 return NULL;
6352         }
6353
6354         /* Allow @len to be NULL */
6355         if (!len)
6356                 len = &dummy;
6357
6358         offset = field->offset;
6359         if (field->flags & TEP_FIELD_IS_DYNAMIC) {
6360                 offset = tep_read_number(event->tep,
6361                                          data + offset, field->size);
6362                 *len = offset >> 16;
6363                 offset &= 0xffff;
6364         } else
6365                 *len = field->size;
6366
6367         return data + offset;
6368 }
6369
6370 /**
6371  * tep_get_field_val - find a field and return its value
6372  * @s: The seq to print to on error
6373  * @event: the event that the field is for
6374  * @name: The name of the field
6375  * @record: The record with the field name.
6376  * @val: place to store the value of the field.
6377  * @err: print default error if failed.
6378  *
6379  * Returns 0 on success -1 on field not found.
6380  */
6381 int tep_get_field_val(struct trace_seq *s, struct tep_event *event,
6382                       const char *name, struct tep_record *record,
6383                       unsigned long long *val, int err)
6384 {
6385         struct tep_format_field *field;
6386
6387         if (!event)
6388                 return -1;
6389
6390         field = tep_find_field(event, name);
6391
6392         return get_field_val(s, field, name, record, val, err);
6393 }
6394
6395 /**
6396  * tep_get_common_field_val - find a common field and return its value
6397  * @s: The seq to print to on error
6398  * @event: the event that the field is for
6399  * @name: The name of the field
6400  * @record: The record with the field name.
6401  * @val: place to store the value of the field.
6402  * @err: print default error if failed.
6403  *
6404  * Returns 0 on success -1 on field not found.
6405  */
6406 int tep_get_common_field_val(struct trace_seq *s, struct tep_event *event,
6407                              const char *name, struct tep_record *record,
6408                              unsigned long long *val, int err)
6409 {
6410         struct tep_format_field *field;
6411
6412         if (!event)
6413                 return -1;
6414
6415         field = tep_find_common_field(event, name);
6416
6417         return get_field_val(s, field, name, record, val, err);
6418 }
6419
6420 /**
6421  * tep_get_any_field_val - find a any field and return its value
6422  * @s: The seq to print to on error
6423  * @event: the event that the field is for
6424  * @name: The name of the field
6425  * @record: The record with the field name.
6426  * @val: place to store the value of the field.
6427  * @err: print default error if failed.
6428  *
6429  * Returns 0 on success -1 on field not found.
6430  */
6431 int tep_get_any_field_val(struct trace_seq *s, struct tep_event *event,
6432                           const char *name, struct tep_record *record,
6433                           unsigned long long *val, int err)
6434 {
6435         struct tep_format_field *field;
6436
6437         if (!event)
6438                 return -1;
6439
6440         field = tep_find_any_field(event, name);
6441
6442         return get_field_val(s, field, name, record, val, err);
6443 }
6444
6445 /**
6446  * tep_print_num_field - print a field and a format
6447  * @s: The seq to print to
6448  * @fmt: The printf format to print the field with.
6449  * @event: the event that the field is for
6450  * @name: The name of the field
6451  * @record: The record with the field name.
6452  * @err: print default error if failed.
6453  *
6454  * Returns positive value on success, negative in case of an error,
6455  * or 0 if buffer is full.
6456  */
6457 int tep_print_num_field(struct trace_seq *s, const char *fmt,
6458                         struct tep_event *event, const char *name,
6459                         struct tep_record *record, int err)
6460 {
6461         struct tep_format_field *field = tep_find_field(event, name);
6462         unsigned long long val;
6463
6464         if (!field)
6465                 goto failed;
6466
6467         if (tep_read_number_field(field, record->data, &val))
6468                 goto failed;
6469
6470         return trace_seq_printf(s, fmt, val);
6471
6472  failed:
6473         if (err)
6474                 trace_seq_printf(s, "CAN'T FIND FIELD \"%s\"", name);
6475         return -1;
6476 }
6477
6478 /**
6479  * tep_print_func_field - print a field and a format for function pointers
6480  * @s: The seq to print to
6481  * @fmt: The printf format to print the field with.
6482  * @event: the event that the field is for
6483  * @name: The name of the field
6484  * @record: The record with the field name.
6485  * @err: print default error if failed.
6486  *
6487  * Returns positive value on success, negative in case of an error,
6488  * or 0 if buffer is full.
6489  */
6490 int tep_print_func_field(struct trace_seq *s, const char *fmt,
6491                          struct tep_event *event, const char *name,
6492                          struct tep_record *record, int err)
6493 {
6494         struct tep_format_field *field = tep_find_field(event, name);
6495         struct tep_handle *tep = event->tep;
6496         unsigned long long val;
6497         struct func_map *func;
6498         char tmp[128];
6499
6500         if (!field)
6501                 goto failed;
6502
6503         if (tep_read_number_field(field, record->data, &val))
6504                 goto failed;
6505
6506         func = find_func(tep, val);
6507
6508         if (func)
6509                 snprintf(tmp, 128, "%s/0x%llx", func->func, func->addr - val);
6510         else
6511                 sprintf(tmp, "0x%08llx", val);
6512
6513         return trace_seq_printf(s, fmt, tmp);
6514
6515  failed:
6516         if (err)
6517                 trace_seq_printf(s, "CAN'T FIND FIELD \"%s\"", name);
6518         return -1;
6519 }
6520
6521 static void free_func_handle(struct tep_function_handler *func)
6522 {
6523         struct func_params *params;
6524
6525         free(func->name);
6526
6527         while (func->params) {
6528                 params = func->params;
6529                 func->params = params->next;
6530                 free(params);
6531         }
6532
6533         free(func);
6534 }
6535
6536 /**
6537  * tep_register_print_function - register a helper function
6538  * @tep: a handle to the trace event parser context
6539  * @func: the function to process the helper function
6540  * @ret_type: the return type of the helper function
6541  * @name: the name of the helper function
6542  * @parameters: A list of enum tep_func_arg_type
6543  *
6544  * Some events may have helper functions in the print format arguments.
6545  * This allows a plugin to dynamically create a way to process one
6546  * of these functions.
6547  *
6548  * The @parameters is a variable list of tep_func_arg_type enums that
6549  * must end with TEP_FUNC_ARG_VOID.
6550  */
6551 int tep_register_print_function(struct tep_handle *tep,
6552                                 tep_func_handler func,
6553                                 enum tep_func_arg_type ret_type,
6554                                 char *name, ...)
6555 {
6556         struct tep_function_handler *func_handle;
6557         struct func_params **next_param;
6558         struct func_params *param;
6559         enum tep_func_arg_type type;
6560         va_list ap;
6561         int ret;
6562
6563         func_handle = find_func_handler(tep, name);
6564         if (func_handle) {
6565                 /*
6566                  * This is most like caused by the users own
6567                  * plugins updating the function. This overrides the
6568                  * system defaults.
6569                  */
6570                 pr_stat("override of function helper '%s'", name);
6571                 remove_func_handler(tep, name);
6572         }
6573
6574         func_handle = calloc(1, sizeof(*func_handle));
6575         if (!func_handle) {
6576                 do_warning("Failed to allocate function handler");
6577                 return TEP_ERRNO__MEM_ALLOC_FAILED;
6578         }
6579
6580         func_handle->ret_type = ret_type;
6581         func_handle->name = strdup(name);
6582         func_handle->func = func;
6583         if (!func_handle->name) {
6584                 do_warning("Failed to allocate function name");
6585                 free(func_handle);
6586                 return TEP_ERRNO__MEM_ALLOC_FAILED;
6587         }
6588
6589         next_param = &(func_handle->params);
6590         va_start(ap, name);
6591         for (;;) {
6592                 type = va_arg(ap, enum tep_func_arg_type);
6593                 if (type == TEP_FUNC_ARG_VOID)
6594                         break;
6595
6596                 if (type >= TEP_FUNC_ARG_MAX_TYPES) {
6597                         do_warning("Invalid argument type %d", type);
6598                         ret = TEP_ERRNO__INVALID_ARG_TYPE;
6599                         goto out_free;
6600                 }
6601
6602                 param = malloc(sizeof(*param));
6603                 if (!param) {
6604                         do_warning("Failed to allocate function param");
6605                         ret = TEP_ERRNO__MEM_ALLOC_FAILED;
6606                         goto out_free;
6607                 }
6608                 param->type = type;
6609                 param->next = NULL;
6610
6611                 *next_param = param;
6612                 next_param = &(param->next);
6613
6614                 func_handle->nr_args++;
6615         }
6616         va_end(ap);
6617
6618         func_handle->next = tep->func_handlers;
6619         tep->func_handlers = func_handle;
6620
6621         return 0;
6622  out_free:
6623         va_end(ap);
6624         free_func_handle(func_handle);
6625         return ret;
6626 }
6627
6628 /**
6629  * tep_unregister_print_function - unregister a helper function
6630  * @tep: a handle to the trace event parser context
6631  * @func: the function to process the helper function
6632  * @name: the name of the helper function
6633  *
6634  * This function removes existing print handler for function @name.
6635  *
6636  * Returns 0 if the handler was removed successully, -1 otherwise.
6637  */
6638 int tep_unregister_print_function(struct tep_handle *tep,
6639                                   tep_func_handler func, char *name)
6640 {
6641         struct tep_function_handler *func_handle;
6642
6643         func_handle = find_func_handler(tep, name);
6644         if (func_handle && func_handle->func == func) {
6645                 remove_func_handler(tep, name);
6646                 return 0;
6647         }
6648         return -1;
6649 }
6650
6651 static struct tep_event *search_event(struct tep_handle *tep, int id,
6652                                       const char *sys_name,
6653                                       const char *event_name)
6654 {
6655         struct tep_event *event;
6656
6657         if (id >= 0) {
6658                 /* search by id */
6659                 event = tep_find_event(tep, id);
6660                 if (!event)
6661                         return NULL;
6662                 if (event_name && (strcmp(event_name, event->name) != 0))
6663                         return NULL;
6664                 if (sys_name && (strcmp(sys_name, event->system) != 0))
6665                         return NULL;
6666         } else {
6667                 event = tep_find_event_by_name(tep, sys_name, event_name);
6668                 if (!event)
6669                         return NULL;
6670         }
6671         return event;
6672 }
6673
6674 /**
6675  * tep_register_event_handler - register a way to parse an event
6676  * @tep: a handle to the trace event parser context
6677  * @id: the id of the event to register
6678  * @sys_name: the system name the event belongs to
6679  * @event_name: the name of the event
6680  * @func: the function to call to parse the event information
6681  * @context: the data to be passed to @func
6682  *
6683  * This function allows a developer to override the parsing of
6684  * a given event. If for some reason the default print format
6685  * is not sufficient, this function will register a function
6686  * for an event to be used to parse the data instead.
6687  *
6688  * If @id is >= 0, then it is used to find the event.
6689  * else @sys_name and @event_name are used.
6690  *
6691  * Returns:
6692  *  TEP_REGISTER_SUCCESS_OVERWRITE if an existing handler is overwritten
6693  *  TEP_REGISTER_SUCCESS if a new handler is registered successfully
6694  *  negative TEP_ERRNO_... in case of an error
6695  *
6696  */
6697 int tep_register_event_handler(struct tep_handle *tep, int id,
6698                                const char *sys_name, const char *event_name,
6699                                tep_event_handler_func func, void *context)
6700 {
6701         struct tep_event *event;
6702         struct event_handler *handle;
6703
6704         event = search_event(tep, id, sys_name, event_name);
6705         if (event == NULL)
6706                 goto not_found;
6707
6708         pr_stat("overriding event (%d) %s:%s with new print handler",
6709                 event->id, event->system, event->name);
6710
6711         event->handler = func;
6712         event->context = context;
6713         return TEP_REGISTER_SUCCESS_OVERWRITE;
6714
6715  not_found:
6716         /* Save for later use. */
6717         handle = calloc(1, sizeof(*handle));
6718         if (!handle) {
6719                 do_warning("Failed to allocate event handler");
6720                 return TEP_ERRNO__MEM_ALLOC_FAILED;
6721         }
6722
6723         handle->id = id;
6724         if (event_name)
6725                 handle->event_name = strdup(event_name);
6726         if (sys_name)
6727                 handle->sys_name = strdup(sys_name);
6728
6729         if ((event_name && !handle->event_name) ||
6730             (sys_name && !handle->sys_name)) {
6731                 do_warning("Failed to allocate event/sys name");
6732                 free((void *)handle->event_name);
6733                 free((void *)handle->sys_name);
6734                 free(handle);
6735                 return TEP_ERRNO__MEM_ALLOC_FAILED;
6736         }
6737
6738         handle->func = func;
6739         handle->next = tep->handlers;
6740         tep->handlers = handle;
6741         handle->context = context;
6742
6743         return TEP_REGISTER_SUCCESS;
6744 }
6745
6746 static int handle_matches(struct event_handler *handler, int id,
6747                           const char *sys_name, const char *event_name,
6748                           tep_event_handler_func func, void *context)
6749 {
6750         if (id >= 0 && id != handler->id)
6751                 return 0;
6752
6753         if (event_name && (strcmp(event_name, handler->event_name) != 0))
6754                 return 0;
6755
6756         if (sys_name && (strcmp(sys_name, handler->sys_name) != 0))
6757                 return 0;
6758
6759         if (func != handler->func || context != handler->context)
6760                 return 0;
6761
6762         return 1;
6763 }
6764
6765 /**
6766  * tep_unregister_event_handler - unregister an existing event handler
6767  * @tep: a handle to the trace event parser context
6768  * @id: the id of the event to unregister
6769  * @sys_name: the system name the handler belongs to
6770  * @event_name: the name of the event handler
6771  * @func: the function to call to parse the event information
6772  * @context: the data to be passed to @func
6773  *
6774  * This function removes existing event handler (parser).
6775  *
6776  * If @id is >= 0, then it is used to find the event.
6777  * else @sys_name and @event_name are used.
6778  *
6779  * Returns 0 if handler was removed successfully, -1 if event was not found.
6780  */
6781 int tep_unregister_event_handler(struct tep_handle *tep, int id,
6782                                  const char *sys_name, const char *event_name,
6783                                  tep_event_handler_func func, void *context)
6784 {
6785         struct tep_event *event;
6786         struct event_handler *handle;
6787         struct event_handler **next;
6788
6789         event = search_event(tep, id, sys_name, event_name);
6790         if (event == NULL)
6791                 goto not_found;
6792
6793         if (event->handler == func && event->context == context) {
6794                 pr_stat("removing override handler for event (%d) %s:%s. Going back to default handler.",
6795                         event->id, event->system, event->name);
6796
6797                 event->handler = NULL;
6798                 event->context = NULL;
6799                 return 0;
6800         }
6801
6802 not_found:
6803         for (next = &tep->handlers; *next; next = &(*next)->next) {
6804                 handle = *next;
6805                 if (handle_matches(handle, id, sys_name, event_name,
6806                                    func, context))
6807                         break;
6808         }
6809
6810         if (!(*next))
6811                 return -1;
6812
6813         *next = handle->next;
6814         free_handler(handle);
6815
6816         return 0;
6817 }
6818
6819 /**
6820  * tep_alloc - create a tep handle
6821  */
6822 struct tep_handle *tep_alloc(void)
6823 {
6824         struct tep_handle *tep = calloc(1, sizeof(*tep));
6825
6826         if (tep) {
6827                 tep->ref_count = 1;
6828                 tep->host_bigendian = tep_is_bigendian();
6829         }
6830
6831         return tep;
6832 }
6833
6834 void tep_ref(struct tep_handle *tep)
6835 {
6836         tep->ref_count++;
6837 }
6838
6839 int tep_get_ref(struct tep_handle *tep)
6840 {
6841         if (tep)
6842                 return tep->ref_count;
6843         return 0;
6844 }
6845
6846 void tep_free_format_field(struct tep_format_field *field)
6847 {
6848         free(field->type);
6849         if (field->alias != field->name)
6850                 free(field->alias);
6851         free(field->name);
6852         free(field);
6853 }
6854
6855 static void free_format_fields(struct tep_format_field *field)
6856 {
6857         struct tep_format_field *next;
6858
6859         while (field) {
6860                 next = field->next;
6861                 tep_free_format_field(field);
6862                 field = next;
6863         }
6864 }
6865
6866 static void free_formats(struct tep_format *format)
6867 {
6868         free_format_fields(format->common_fields);
6869         free_format_fields(format->fields);
6870 }
6871
6872 void tep_free_event(struct tep_event *event)
6873 {
6874         free(event->name);
6875         free(event->system);
6876
6877         free_formats(&event->format);
6878
6879         free(event->print_fmt.format);
6880         free_args(event->print_fmt.args);
6881
6882         free(event);
6883 }
6884
6885 /**
6886  * tep_free - free a tep handle
6887  * @tep: the tep handle to free
6888  */
6889 void tep_free(struct tep_handle *tep)
6890 {
6891         struct cmdline_list *cmdlist, *cmdnext;
6892         struct func_list *funclist, *funcnext;
6893         struct printk_list *printklist, *printknext;
6894         struct tep_function_handler *func_handler;
6895         struct event_handler *handle;
6896         int i;
6897
6898         if (!tep)
6899                 return;
6900
6901         cmdlist = tep->cmdlist;
6902         funclist = tep->funclist;
6903         printklist = tep->printklist;
6904
6905         tep->ref_count--;
6906         if (tep->ref_count)
6907                 return;
6908
6909         if (tep->cmdlines) {
6910                 for (i = 0; i < tep->cmdline_count; i++)
6911                         free(tep->cmdlines[i].comm);
6912                 free(tep->cmdlines);
6913         }
6914
6915         while (cmdlist) {
6916                 cmdnext = cmdlist->next;
6917                 free(cmdlist->comm);
6918                 free(cmdlist);
6919                 cmdlist = cmdnext;
6920         }
6921
6922         if (tep->func_map) {
6923                 for (i = 0; i < (int)tep->func_count; i++) {
6924                         free(tep->func_map[i].func);
6925                         free(tep->func_map[i].mod);
6926                 }
6927                 free(tep->func_map);
6928         }
6929
6930         while (funclist) {
6931                 funcnext = funclist->next;
6932                 free(funclist->func);
6933                 free(funclist->mod);
6934                 free(funclist);
6935                 funclist = funcnext;
6936         }
6937
6938         while (tep->func_handlers) {
6939                 func_handler = tep->func_handlers;
6940                 tep->func_handlers = func_handler->next;
6941                 free_func_handle(func_handler);
6942         }
6943
6944         if (tep->printk_map) {
6945                 for (i = 0; i < (int)tep->printk_count; i++)
6946                         free(tep->printk_map[i].printk);
6947                 free(tep->printk_map);
6948         }
6949
6950         while (printklist) {
6951                 printknext = printklist->next;
6952                 free(printklist->printk);
6953                 free(printklist);
6954                 printklist = printknext;
6955         }
6956
6957         for (i = 0; i < tep->nr_events; i++)
6958                 tep_free_event(tep->events[i]);
6959
6960         while (tep->handlers) {
6961                 handle = tep->handlers;
6962                 tep->handlers = handle->next;
6963                 free_handler(handle);
6964         }
6965
6966         free(tep->trace_clock);
6967         free(tep->events);
6968         free(tep->sort_events);
6969         free(tep->func_resolver);
6970
6971         free(tep);
6972 }
6973
6974 void tep_unref(struct tep_handle *tep)
6975 {
6976         tep_free(tep);
6977 }