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