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