/*
   * trace_output.c
   *
   * Copyright (C) 2008 Red Hat Inc, Steven Rostedt <srostedt@redhat.com>
   *
   */
  
  #include <linux/module.h>
  #include <linux/mutex.h>
  #include <linux/ftrace.h>
  
  #include "trace_output.h"
  
  /* must be a power of 2 */
  #define EVENT_HASHSIZE	128

  DECLARE_RWSEM(trace_event_sem);

  static struct hlist_head event_hash[EVENT_HASHSIZE] __read_mostly;
  
  static int next_event_type = __TRACE_LAST_TYPE + 1;

  enum print_line_t trace_print_bputs_msg_only(struct trace_iterator *iter)
  {
  	struct trace_seq *s = &iter->seq;
  	struct trace_entry *entry = iter->ent;
  	struct bputs_entry *field;

  
  	trace_assign_type(field, entry);

  	trace_seq_puts(s, field->str);

  	return trace_handle_return(s);

  }

  enum print_line_t trace_print_bprintk_msg_only(struct trace_iterator *iter)
  {
  	struct trace_seq *s = &iter->seq;
  	struct trace_entry *entry = iter->ent;
  	struct bprint_entry *field;

  
  	trace_assign_type(field, entry);

  	trace_seq_bprintf(s, field->fmt, field->buf);

  	return trace_handle_return(s);

  }
  
  enum print_line_t trace_print_printk_msg_only(struct trace_iterator *iter)
  {
  	struct trace_seq *s = &iter->seq;
  	struct trace_entry *entry = iter->ent;
  	struct print_entry *field;

  
  	trace_assign_type(field, entry);

  	trace_seq_puts(s, field->buf);

  	return trace_handle_return(s);

  }

  const char *

  trace_print_flags_seq(struct trace_seq *p, const char *delim,
  		      unsigned long flags,
  		      const struct trace_print_flags *flag_array)

  {
  	unsigned long mask;
  	const char *str;

  	const char *ret = trace_seq_buffer_ptr(p);

  	int i, first = 1;

  	for (i = 0;  flag_array[i].name && flags; i++) {
  
  		mask = flag_array[i].mask;
  		if ((flags & mask) != mask)
  			continue;
  
  		str = flag_array[i].name;
  		flags &= ~mask;

  		if (!first && delim)

  			trace_seq_puts(p, delim);

  		else
  			first = 0;

  		trace_seq_puts(p, str);
  	}
  
  	/* check for left over flags */
  	if (flags) {

  		if (!first && delim)

  			trace_seq_puts(p, delim);
  		trace_seq_printf(p, "0x%lx", flags);
  	}
  
  	trace_seq_putc(p, 0);

  	return ret;

  }

  EXPORT_SYMBOL(trace_print_flags_seq);

  const char *

  trace_print_symbols_seq(struct trace_seq *p, unsigned long val,
  			const struct trace_print_flags *symbol_array)

  {
  	int i;

  	const char *ret = trace_seq_buffer_ptr(p);

  
  	for (i = 0;  symbol_array[i].name; i++) {
  
  		if (val != symbol_array[i].mask)
  			continue;
  
  		trace_seq_puts(p, symbol_array[i].name);
  		break;
  	}

  	if (ret == (const char *)(trace_seq_buffer_ptr(p)))

  		trace_seq_printf(p, "0x%lx", val);

  	trace_seq_putc(p, 0);

  	return ret;

  }

  EXPORT_SYMBOL(trace_print_symbols_seq);

  #if BITS_PER_LONG == 32
  const char *

  trace_print_symbols_seq_u64(struct trace_seq *p, unsigned long long val,

  			 const struct trace_print_flags_u64 *symbol_array)
  {
  	int i;

  	const char *ret = trace_seq_buffer_ptr(p);

  
  	for (i = 0;  symbol_array[i].name; i++) {
  
  		if (val != symbol_array[i].mask)
  			continue;
  
  		trace_seq_puts(p, symbol_array[i].name);
  		break;
  	}

  	if (ret == (const char *)(trace_seq_buffer_ptr(p)))

  		trace_seq_printf(p, "0x%llx", val);
  
  	trace_seq_putc(p, 0);
  
  	return ret;
  }

  EXPORT_SYMBOL(trace_print_symbols_seq_u64);

  #endif

  const char *

  trace_print_bitmask_seq(struct trace_seq *p, void *bitmask_ptr,
  			unsigned int bitmask_size)

  {

  	const char *ret = trace_seq_buffer_ptr(p);

  
  	trace_seq_bitmask(p, bitmask_ptr, bitmask_size * 8);
  	trace_seq_putc(p, 0);
  
  	return ret;
  }

  EXPORT_SYMBOL_GPL(trace_print_bitmask_seq);

  
  const char *

  trace_print_hex_seq(struct trace_seq *p, const unsigned char *buf, int buf_len)

  {
  	int i;

  	const char *ret = trace_seq_buffer_ptr(p);

  
  	for (i = 0; i < buf_len; i++)
  		trace_seq_printf(p, "%s%2.2x", i == 0 ? "" : " ", buf[i]);
  
  	trace_seq_putc(p, 0);
  
  	return ret;
  }

  EXPORT_SYMBOL(trace_print_hex_seq);

  const char *

  trace_print_array_seq(struct trace_seq *p, const void *buf, int count,
  		      size_t el_size)

  {
  	const char *ret = trace_seq_buffer_ptr(p);
  	const char *prefix = "";
  	void *ptr = (void *)buf;

  	size_t buf_len = count * el_size;

  
  	trace_seq_putc(p, '{');
  
  	while (ptr < buf + buf_len) {
  		switch (el_size) {
  		case 1:
  			trace_seq_printf(p, "%s0x%x", prefix,
  					 *(u8 *)ptr);
  			break;
  		case 2:
  			trace_seq_printf(p, "%s0x%x", prefix,
  					 *(u16 *)ptr);
  			break;
  		case 4:
  			trace_seq_printf(p, "%s0x%x", prefix,
  					 *(u32 *)ptr);
  			break;
  		case 8:
  			trace_seq_printf(p, "%s0x%llx", prefix,
  					 *(u64 *)ptr);
  			break;
  		default:
  			trace_seq_printf(p, "BAD SIZE:%zu 0x%x", el_size,
  					 *(u8 *)ptr);
  			el_size = 1;
  		}
  		prefix = ",";
  		ptr += el_size;
  	}
  
  	trace_seq_putc(p, '}');
  	trace_seq_putc(p, 0);
  
  	return ret;
  }

  EXPORT_SYMBOL(trace_print_array_seq);

  int trace_raw_output_prep(struct trace_iterator *iter,
  			  struct trace_event *trace_event)

  {

  	struct trace_event_call *event;

  	struct trace_seq *s = &iter->seq;
  	struct trace_seq *p = &iter->tmp_seq;
  	struct trace_entry *entry;

  	event = container_of(trace_event, struct trace_event_call, event);

  	entry = iter->ent;
  
  	if (entry->type != event->event.type) {
  		WARN_ON_ONCE(1);
  		return TRACE_TYPE_UNHANDLED;
  	}
  
  	trace_seq_init(p);

  	trace_seq_printf(s, "%s: ", trace_event_name(event));

  	return trace_handle_return(s);

  }

  EXPORT_SYMBOL(trace_raw_output_prep);

  static int trace_output_raw(struct trace_iterator *iter, char *name,
  			    char *fmt, va_list ap)

  {
  	struct trace_seq *s = &iter->seq;

  	trace_seq_printf(s, "%s: ", name);
  	trace_seq_vprintf(s, fmt, ap);

  	return trace_handle_return(s);

  }

  int trace_output_call(struct trace_iterator *iter, char *name, char *fmt, ...)

  {
  	va_list ap;
  	int ret;
  
  	va_start(ap, fmt);

  	ret = trace_output_raw(iter, name, fmt, ap);

  	va_end(ap);
  
  	return ret;
  }

  EXPORT_SYMBOL_GPL(trace_output_call);

  #ifdef CONFIG_KRETPROBES
  static inline const char *kretprobed(const char *name)
  {
  	static const char tramp_name[] = "kretprobe_trampoline";
  	int size = sizeof(tramp_name);
  
  	if (strncmp(tramp_name, name, size) == 0)
  		return "[unknown/kretprobe'd]";
  	return name;
  }
  #else
  static inline const char *kretprobed(const char *name)
  {
  	return name;
  }
  #endif /* CONFIG_KRETPROBES */

  static void

  seq_print_sym_short(struct trace_seq *s, const char *fmt, unsigned long address)
  {
  #ifdef CONFIG_KALLSYMS
  	char str[KSYM_SYMBOL_LEN];
  	const char *name;
  
  	kallsyms_lookup(address, NULL, NULL, NULL, str);
  
  	name = kretprobed(str);

  	trace_seq_printf(s, fmt, name);

  #endif

  }

  static void

  seq_print_sym_offset(struct trace_seq *s, const char *fmt,
  		     unsigned long address)
  {
  #ifdef CONFIG_KALLSYMS
  	char str[KSYM_SYMBOL_LEN];
  	const char *name;
  
  	sprint_symbol(str, address);
  	name = kretprobed(str);

  	trace_seq_printf(s, fmt, name);

  #endif

  }
  
  #ifndef CONFIG_64BIT
  # define IP_FMT "%08lx"
  #else
  # define IP_FMT "%016lx"
  #endif

  static int seq_print_user_ip(struct trace_seq *s, struct mm_struct *mm,
  			     unsigned long ip, unsigned long sym_flags)

  {
  	struct file *file = NULL;
  	unsigned long vmstart = 0;
  	int ret = 1;

  	if (s->full)
  		return 0;

  	if (mm) {
  		const struct vm_area_struct *vma;
  
  		down_read(&mm->mmap_sem);
  		vma = find_vma(mm, ip);
  		if (vma) {
  			file = vma->vm_file;
  			vmstart = vma->vm_start;
  		}
  		if (file) {
  			ret = trace_seq_path(s, &file->f_path);
  			if (ret)

  				trace_seq_printf(s, "[+0x%lx]",
  						 ip - vmstart);

  		}
  		up_read(&mm->mmap_sem);
  	}
  	if (ret && ((sym_flags & TRACE_ITER_SYM_ADDR) || !file))

  		trace_seq_printf(s, " <" IP_FMT ">", ip);
  	return !trace_seq_has_overflowed(s);

  }
  
  int

  seq_print_ip_sym(struct trace_seq *s, unsigned long ip, unsigned long sym_flags)
  {

  	if (!ip) {
  		trace_seq_putc(s, '0');
  		goto out;
  	}

  
  	if (sym_flags & TRACE_ITER_SYM_OFFSET)

  		seq_print_sym_offset(s, "%s", ip);

  	else

  		seq_print_sym_short(s, "%s", ip);

  
  	if (sym_flags & TRACE_ITER_SYM_ADDR)

  		trace_seq_printf(s, " <" IP_FMT ">", ip);
  
   out:
  	return !trace_seq_has_overflowed(s);

  }

  /**
   * trace_print_lat_fmt - print the irq, preempt and lockdep fields
   * @s: trace seq struct to write to
   * @entry: The trace entry field from the ring buffer
   *
   * Prints the generic fields of irqs off, in hard or softirq, preempt

   * count.

   */
  int trace_print_lat_fmt(struct trace_seq *s, struct trace_entry *entry)

  {

  	char hardsoft_irq;
  	char need_resched;
  	char irqs_off;
  	int hardirq;
  	int softirq;

  	int nmi;

  	nmi = entry->flags & TRACE_FLAG_NMI;

  	hardirq = entry->flags & TRACE_FLAG_HARDIRQ;
  	softirq = entry->flags & TRACE_FLAG_SOFTIRQ;

  	irqs_off =
  		(entry->flags & TRACE_FLAG_IRQS_OFF) ? 'd' :
  		(entry->flags & TRACE_FLAG_IRQS_NOSUPPORT) ? 'X' :
  		'.';

  
  	switch (entry->flags & (TRACE_FLAG_NEED_RESCHED |
  				TRACE_FLAG_PREEMPT_RESCHED)) {
  	case TRACE_FLAG_NEED_RESCHED | TRACE_FLAG_PREEMPT_RESCHED:
  		need_resched = 'N';
  		break;
  	case TRACE_FLAG_NEED_RESCHED:
  		need_resched = 'n';
  		break;
  	case TRACE_FLAG_PREEMPT_RESCHED:
  		need_resched = 'p';
  		break;
  	default:
  		need_resched = '.';
  		break;
  	}

  	hardsoft_irq =

  		(nmi && hardirq)     ? 'Z' :
  		nmi                  ? 'z' :

  		(hardirq && softirq) ? 'H' :

  		hardirq              ? 'h' :
  		softirq              ? 's' :
  		                       '.' ;

  	trace_seq_printf(s, "%c%c%c",
  			 irqs_off, need_resched, hardsoft_irq);

  	if (entry->preempt_count)

  		trace_seq_printf(s, "%x", entry->preempt_count);

  	else

  		trace_seq_putc(s, '.');

  	return !trace_seq_has_overflowed(s);

  }

  static int
  lat_print_generic(struct trace_seq *s, struct trace_entry *entry, int cpu)
  {
  	char comm[TASK_COMM_LEN];
  
  	trace_find_cmdline(entry->pid, comm);

  	trace_seq_printf(s, "%8.8s-%-5d %3d",
  			 comm, entry->pid, cpu);

  
  	return trace_print_lat_fmt(s, entry);
  }

  #undef MARK
  #define MARK(v, s) {.val = v, .sym = s}
  /* trace overhead mark */
  static const struct trace_mark {
  	unsigned long long	val; /* unit: nsec */
  	char			sym;
  } mark[] = {
  	MARK(1000000000ULL	, '$'), /* 1 sec */

  	MARK(100000000ULL	, '@'), /* 100 msec */
  	MARK(10000000ULL	, '*'), /* 10 msec */

  	MARK(1000000ULL		, '#'), /* 1000 usecs */
  	MARK(100000ULL		, '!'), /* 100 usecs */
  	MARK(10000ULL		, '+'), /* 10 usecs */
  };
  #undef MARK
  
  char trace_find_mark(unsigned long long d)
  {
  	int i;
  	int size = ARRAY_SIZE(mark);
  
  	for (i = 0; i < size; i++) {

  		if (d > mark[i].val)

  			break;
  	}
  
  	return (i == size) ? ' ' : mark[i].sym;
  }

  static int

  lat_print_timestamp(struct trace_iterator *iter, u64 next_ts)

  {

  	struct trace_array *tr = iter->tr;
  	unsigned long verbose = tr->trace_flags & TRACE_ITER_VERBOSE;

  	unsigned long in_ns = iter->iter_flags & TRACE_FILE_TIME_IN_NS;

  	unsigned long long abs_ts = iter->ts - iter->trace_buffer->time_start;

  	unsigned long long rel_ts = next_ts - iter->ts;
  	struct trace_seq *s = &iter->seq;
  
  	if (in_ns) {
  		abs_ts = ns2usecs(abs_ts);
  		rel_ts = ns2usecs(rel_ts);
  	}
  
  	if (verbose && in_ns) {
  		unsigned long abs_usec = do_div(abs_ts, USEC_PER_MSEC);
  		unsigned long abs_msec = (unsigned long)abs_ts;
  		unsigned long rel_usec = do_div(rel_ts, USEC_PER_MSEC);
  		unsigned long rel_msec = (unsigned long)rel_ts;

  		trace_seq_printf(
  			s, "[%08llx] %ld.%03ldms (+%ld.%03ldms): ",
  			ns2usecs(iter->ts),
  			abs_msec, abs_usec,
  			rel_msec, rel_usec);

  	} else if (verbose && !in_ns) {

  		trace_seq_printf(
  			s, "[%016llx] %lld (+%lld): ",
  			iter->ts, abs_ts, rel_ts);

  	} else if (!verbose && in_ns) {

  		trace_seq_printf(
  			s, " %4lldus%c: ",
  			abs_ts,

  			trace_find_mark(rel_ts * NSEC_PER_USEC));

  	} else { /* !verbose && !in_ns */

  		trace_seq_printf(s, " %4lld: ", abs_ts);

  	}

  
  	return !trace_seq_has_overflowed(s);

  }
  
  int trace_print_context(struct trace_iterator *iter)
  {

  	struct trace_array *tr = iter->tr;

  	struct trace_seq *s = &iter->seq;
  	struct trace_entry *entry = iter->ent;

  	unsigned long long t;
  	unsigned long secs, usec_rem;

  	char comm[TASK_COMM_LEN];
  
  	trace_find_cmdline(entry->pid, comm);

  	trace_seq_printf(s, "%16s-%-5d [%03d] ",

  			       comm, entry->pid, iter->cpu);

  	if (tr->trace_flags & TRACE_ITER_IRQ_INFO)

  		trace_print_lat_fmt(s, entry);

  	if (iter->iter_flags & TRACE_FILE_TIME_IN_NS) {
  		t = ns2usecs(iter->ts);
  		usec_rem = do_div(t, USEC_PER_SEC);
  		secs = (unsigned long)t;

  		trace_seq_printf(s, " %5lu.%06lu: ", secs, usec_rem);

  	} else

  		trace_seq_printf(s, " %12llu: ", iter->ts);
  
  	return !trace_seq_has_overflowed(s);

  }
  
  int trace_print_lat_context(struct trace_iterator *iter)
  {

  	struct trace_array *tr = iter->tr;

  	/* trace_find_next_entry will reset ent_size */
  	int ent_size = iter->ent_size;

  	struct trace_seq *s = &iter->seq;

  	u64 next_ts;

  	struct trace_entry *entry = iter->ent,
  			   *next_entry = trace_find_next_entry(iter, NULL,
  							       &next_ts);

  	unsigned long verbose = (tr->trace_flags & TRACE_ITER_VERBOSE);

  	/* Restore the original ent_size */
  	iter->ent_size = ent_size;

  	if (!next_entry)
  		next_ts = iter->ts;

  
  	if (verbose) {

  		char comm[TASK_COMM_LEN];
  
  		trace_find_cmdline(entry->pid, comm);

  		trace_seq_printf(
  			s, "%16s %5d %3d %d %08x %08lx ",
  			comm, entry->pid, iter->cpu, entry->flags,
  			entry->preempt_count, iter->idx);

  	} else {

  		lat_print_generic(s, entry, iter->cpu);

  	}

  	lat_print_timestamp(iter, next_ts);

  	return !trace_seq_has_overflowed(s);

  }

  static const char state_to_char[] = TASK_STATE_TO_CHAR_STR;
  
  static int task_state_char(unsigned long state)
  {
  	int bit = state ? __ffs(state) + 1 : 0;
  
  	return bit < sizeof(state_to_char) - 1 ? state_to_char[bit] : '?';
  }

  /**
   * ftrace_find_event - find a registered event
   * @type: the type of event to look for
   *
   * Returns an event of type @type otherwise NULL

   * Called with trace_event_read_lock() held.

   */
  struct trace_event *ftrace_find_event(int type)
  {
  	struct trace_event *event;

  	unsigned key;
  
  	key = type & (EVENT_HASHSIZE - 1);

  	hlist_for_each_entry(event, &event_hash[key], node) {

  		if (event->type == type)
  			return event;
  	}
  
  	return NULL;
  }

  static LIST_HEAD(ftrace_event_list);
  
  static int trace_search_list(struct list_head **list)
  {
  	struct trace_event *e;
  	int last = __TRACE_LAST_TYPE;
  
  	if (list_empty(&ftrace_event_list)) {
  		*list = &ftrace_event_list;
  		return last + 1;
  	}
  
  	/*
  	 * We used up all possible max events,
  	 * lets see if somebody freed one.
  	 */
  	list_for_each_entry(e, &ftrace_event_list, list) {
  		if (e->type != last + 1)
  			break;
  		last++;
  	}
  
  	/* Did we used up all 65 thousand events??? */

  	if ((last + 1) > TRACE_EVENT_TYPE_MAX)

  		return 0;
  
  	*list = &e->list;
  	return last + 1;
  }

  void trace_event_read_lock(void)
  {

  	down_read(&trace_event_sem);

  }
  
  void trace_event_read_unlock(void)
  {

  	up_read(&trace_event_sem);

  }

  /**

   * register_trace_event - register output for an event type

   * @event: the event type to register
   *
   * Event types are stored in a hash and this hash is used to
   * find a way to print an event. If the @event->type is set
   * then it will use that type, otherwise it will assign a
   * type to use.
   *
   * If you assign your own type, please make sure it is added
   * to the trace_type enum in trace.h, to avoid collisions
   * with the dynamic types.
   *
   * Returns the event type number or zero on error.
   */

  int register_trace_event(struct trace_event *event)

  {
  	unsigned key;
  	int ret = 0;

  	down_write(&trace_event_sem);

  	if (WARN_ON(!event))

  		goto out;

  	if (WARN_ON(!event->funcs))
  		goto out;

  	INIT_LIST_HEAD(&event->list);
  
  	if (!event->type) {

  		struct list_head *list = NULL;

  		if (next_event_type > TRACE_EVENT_TYPE_MAX) {

  
  			event->type = trace_search_list(&list);
  			if (!event->type)
  				goto out;
  
  		} else {

  			event->type = next_event_type++;
  			list = &ftrace_event_list;
  		}
  
  		if (WARN_ON(ftrace_find_event(event->type)))
  			goto out;
  
  		list_add_tail(&event->list, list);
  
  	} else if (event->type > __TRACE_LAST_TYPE) {

  		printk(KERN_WARNING "Need to add type to trace.h
  ");
  		WARN_ON(1);

  		goto out;

  	} else {
  		/* Is this event already used */
  		if (ftrace_find_event(event->type))
  			goto out;
  	}

  	if (event->funcs->trace == NULL)
  		event->funcs->trace = trace_nop_print;
  	if (event->funcs->raw == NULL)
  		event->funcs->raw = trace_nop_print;
  	if (event->funcs->hex == NULL)
  		event->funcs->hex = trace_nop_print;
  	if (event->funcs->binary == NULL)
  		event->funcs->binary = trace_nop_print;

  	key = event->type & (EVENT_HASHSIZE - 1);

  	hlist_add_head(&event->node, &event_hash[key]);

  
  	ret = event->type;
   out:

  	up_write(&trace_event_sem);

  
  	return ret;
  }

  EXPORT_SYMBOL_GPL(register_trace_event);

  /*

   * Used by module code with the trace_event_sem held for write.

   */

  int __unregister_trace_event(struct trace_event *event)

  {
  	hlist_del(&event->node);
  	list_del(&event->list);
  	return 0;
  }

  /**

   * unregister_trace_event - remove a no longer used event

   * @event: the event to remove
   */

  int unregister_trace_event(struct trace_event *event)

  {

  	down_write(&trace_event_sem);

  	__unregister_trace_event(event);

  	up_write(&trace_event_sem);

  
  	return 0;
  }

  EXPORT_SYMBOL_GPL(unregister_trace_event);

  
  /*
   * Standard events
   */

  enum print_line_t trace_nop_print(struct trace_iterator *iter, int flags,
  				  struct trace_event *event)

  {

  	trace_seq_printf(&iter->seq, "type: %d
  ", iter->ent->type);

  	return trace_handle_return(&iter->seq);

  }
  
  /* TRACE_FN */

  static enum print_line_t trace_fn_trace(struct trace_iterator *iter, int flags,
  					struct trace_event *event)

  {
  	struct ftrace_entry *field;

  	struct trace_seq *s = &iter->seq;

  	trace_assign_type(field, iter->ent);

  	seq_print_ip_sym(s, field->ip, flags);

  
  	if ((flags & TRACE_ITER_PRINT_PARENT) && field->parent_ip) {

  		trace_seq_puts(s, " <-");
  		seq_print_ip_sym(s, field->parent_ip, flags);

  	}

  	trace_seq_putc(s, '
  ');

  	return trace_handle_return(s);

  }

  static enum print_line_t trace_fn_raw(struct trace_iterator *iter, int flags,
  				      struct trace_event *event)

  {
  	struct ftrace_entry *field;

  	trace_assign_type(field, iter->ent);

  	trace_seq_printf(&iter->seq, "%lx %lx
  ",
  			 field->ip,
  			 field->parent_ip);

  	return trace_handle_return(&iter->seq);

  }

  static enum print_line_t trace_fn_hex(struct trace_iterator *iter, int flags,
  				      struct trace_event *event)

  {
  	struct ftrace_entry *field;

  	struct trace_seq *s = &iter->seq;

  	trace_assign_type(field, iter->ent);

  	SEQ_PUT_HEX_FIELD(s, field->ip);
  	SEQ_PUT_HEX_FIELD(s, field->parent_ip);

  	return trace_handle_return(s);

  }

  static enum print_line_t trace_fn_bin(struct trace_iterator *iter, int flags,
  				      struct trace_event *event)

  {
  	struct ftrace_entry *field;

  	struct trace_seq *s = &iter->seq;

  	trace_assign_type(field, iter->ent);

  	SEQ_PUT_FIELD(s, field->ip);
  	SEQ_PUT_FIELD(s, field->parent_ip);

  	return trace_handle_return(s);

  }

  static struct trace_event_functions trace_fn_funcs = {

  	.trace		= trace_fn_trace,

  	.raw		= trace_fn_raw,
  	.hex		= trace_fn_hex,
  	.binary		= trace_fn_bin,
  };

  static struct trace_event trace_fn_event = {
  	.type		= TRACE_FN,
  	.funcs		= &trace_fn_funcs,
  };

  /* TRACE_CTX an TRACE_WAKE */

  static enum print_line_t trace_ctxwake_print(struct trace_iterator *iter,
  					     char *delim)

  {
  	struct ctx_switch_entry *field;

  	char comm[TASK_COMM_LEN];

  	int S, T;

  	trace_assign_type(field, iter->ent);

  
  	T = task_state_char(field->next_state);
  	S = task_state_char(field->prev_state);

  	trace_find_cmdline(field->next_pid, comm);

  	trace_seq_printf(&iter->seq,
  			 " %5d:%3d:%c %s [%03d] %5d:%3d:%c %s
  ",
  			 field->prev_pid,
  			 field->prev_prio,
  			 S, delim,
  			 field->next_cpu,
  			 field->next_pid,
  			 field->next_prio,
  			 T, comm);
  
  	return trace_handle_return(&iter->seq);

  }

  static enum print_line_t trace_ctx_print(struct trace_iterator *iter, int flags,
  					 struct trace_event *event)

  {

  	return trace_ctxwake_print(iter, "==>");

  }

  static enum print_line_t trace_wake_print(struct trace_iterator *iter,

  					  int flags, struct trace_event *event)

  {

  	return trace_ctxwake_print(iter, "  +");

  }

  static int trace_ctxwake_raw(struct trace_iterator *iter, char S)

  {
  	struct ctx_switch_entry *field;
  	int T;

  	trace_assign_type(field, iter->ent);

  
  	if (!S)

  		S = task_state_char(field->prev_state);

  	T = task_state_char(field->next_state);

  	trace_seq_printf(&iter->seq, "%d %d %c %d %d %d %c
  ",
  			 field->prev_pid,
  			 field->prev_prio,
  			 S,
  			 field->next_cpu,
  			 field->next_pid,
  			 field->next_prio,
  			 T);
  
  	return trace_handle_return(&iter->seq);

  }

  static enum print_line_t trace_ctx_raw(struct trace_iterator *iter, int flags,
  				       struct trace_event *event)

  {

  	return trace_ctxwake_raw(iter, 0);

  }

  static enum print_line_t trace_wake_raw(struct trace_iterator *iter, int flags,
  					struct trace_event *event)

  {

  	return trace_ctxwake_raw(iter, '+');

  }

  static int trace_ctxwake_hex(struct trace_iterator *iter, char S)

  {
  	struct ctx_switch_entry *field;

  	struct trace_seq *s = &iter->seq;

  	int T;

  	trace_assign_type(field, iter->ent);

  
  	if (!S)

  		S = task_state_char(field->prev_state);

  	T = task_state_char(field->next_state);

  	SEQ_PUT_HEX_FIELD(s, field->prev_pid);
  	SEQ_PUT_HEX_FIELD(s, field->prev_prio);
  	SEQ_PUT_HEX_FIELD(s, S);
  	SEQ_PUT_HEX_FIELD(s, field->next_cpu);
  	SEQ_PUT_HEX_FIELD(s, field->next_pid);
  	SEQ_PUT_HEX_FIELD(s, field->next_prio);
  	SEQ_PUT_HEX_FIELD(s, T);

  	return trace_handle_return(s);

  }

  static enum print_line_t trace_ctx_hex(struct trace_iterator *iter, int flags,
  				       struct trace_event *event)

  {

  	return trace_ctxwake_hex(iter, 0);

  }

  static enum print_line_t trace_wake_hex(struct trace_iterator *iter, int flags,
  					struct trace_event *event)

  {

  	return trace_ctxwake_hex(iter, '+');

  }

  static enum print_line_t trace_ctxwake_bin(struct trace_iterator *iter,

  					   int flags, struct trace_event *event)

  {
  	struct ctx_switch_entry *field;

  	struct trace_seq *s = &iter->seq;

  	trace_assign_type(field, iter->ent);

  	SEQ_PUT_FIELD(s, field->prev_pid);
  	SEQ_PUT_FIELD(s, field->prev_prio);
  	SEQ_PUT_FIELD(s, field->prev_state);
  	SEQ_PUT_FIELD(s, field->next_cpu);
  	SEQ_PUT_FIELD(s, field->next_pid);
  	SEQ_PUT_FIELD(s, field->next_prio);
  	SEQ_PUT_FIELD(s, field->next_state);

  	return trace_handle_return(s);

  }

  static struct trace_event_functions trace_ctx_funcs = {

  	.trace		= trace_ctx_print,

  	.raw		= trace_ctx_raw,
  	.hex		= trace_ctx_hex,
  	.binary		= trace_ctxwake_bin,
  };

  static struct trace_event trace_ctx_event = {
  	.type		= TRACE_CTX,
  	.funcs		= &trace_ctx_funcs,
  };
  
  static struct trace_event_functions trace_wake_funcs = {

  	.trace		= trace_wake_print,

  	.raw		= trace_wake_raw,
  	.hex		= trace_wake_hex,
  	.binary		= trace_ctxwake_bin,
  };

  static struct trace_event trace_wake_event = {
  	.type		= TRACE_WAKE,
  	.funcs		= &trace_wake_funcs,
  };

  /* TRACE_STACK */

  static enum print_line_t trace_stack_print(struct trace_iterator *iter,

  					   int flags, struct trace_event *event)

  {
  	struct stack_entry *field;

  	struct trace_seq *s = &iter->seq;

  	unsigned long *p;
  	unsigned long *end;

  	trace_assign_type(field, iter->ent);

  	end = (unsigned long *)((long)iter->ent + iter->ent_size);

  	trace_seq_puts(s, "<stack trace>
  ");

  
  	for (p = field->caller; p && *p != ULONG_MAX && p < end; p++) {

  		if (trace_seq_has_overflowed(s))
  			break;

  		trace_seq_puts(s, " => ");
  		seq_print_ip_sym(s, *p, flags);
  		trace_seq_putc(s, '
  ');
  	}

  	return trace_handle_return(s);

  }

  static struct trace_event_functions trace_stack_funcs = {

  	.trace		= trace_stack_print,

  };

  static struct trace_event trace_stack_event = {
  	.type		= TRACE_STACK,
  	.funcs		= &trace_stack_funcs,
  };

  /* TRACE_USER_STACK */

  static enum print_line_t trace_user_stack_print(struct trace_iterator *iter,

  						int flags, struct trace_event *event)

  {

  	struct trace_array *tr = iter->tr;

  	struct userstack_entry *field;

  	struct trace_seq *s = &iter->seq;

  	struct mm_struct *mm = NULL;
  	unsigned int i;

  	trace_assign_type(field, iter->ent);

  	trace_seq_puts(s, "<user stack trace>
  ");

  	if (tr->trace_flags & TRACE_ITER_SYM_USEROBJ) {

  		struct task_struct *task;
  		/*
  		 * we do the lookup on the thread group leader,
  		 * since individual threads might have already quit!
  		 */
  		rcu_read_lock();
  		task = find_task_by_vpid(field->tgid);
  		if (task)
  			mm = get_task_mm(task);
  		rcu_read_unlock();
  	}
  
  	for (i = 0; i < FTRACE_STACK_ENTRIES; i++) {
  		unsigned long ip = field->caller[i];
  
  		if (ip == ULONG_MAX || trace_seq_has_overflowed(s))
  			break;
  
  		trace_seq_puts(s, " => ");
  
  		if (!ip) {
  			trace_seq_puts(s, "??");
  			trace_seq_putc(s, '
  ');
  			continue;
  		}
  
  		seq_print_user_ip(s, mm, ip, flags);
  		trace_seq_putc(s, '
  ');
  	}
  
  	if (mm)
  		mmput(mm);

  	return trace_handle_return(s);

  }

  static struct trace_event_functions trace_user_stack_funcs = {

  	.trace		= trace_user_stack_print,

  };

  static struct trace_event trace_user_stack_event = {
  	.type		= TRACE_USER_STACK,
  	.funcs		= &trace_user_stack_funcs,
  };

  /* TRACE_BPUTS */
  static enum print_line_t
  trace_bputs_print(struct trace_iterator *iter, int flags,
  		   struct trace_event *event)
  {
  	struct trace_entry *entry = iter->ent;
  	struct trace_seq *s = &iter->seq;
  	struct bputs_entry *field;
  
  	trace_assign_type(field, entry);

  	seq_print_ip_sym(s, field->ip, flags);
  	trace_seq_puts(s, ": ");
  	trace_seq_puts(s, field->str);

  	return trace_handle_return(s);

  }
  
  
  static enum print_line_t
  trace_bputs_raw(struct trace_iterator *iter, int flags,
  		struct trace_event *event)
  {
  	struct bputs_entry *field;
  	struct trace_seq *s = &iter->seq;
  
  	trace_assign_type(field, iter->ent);

  	trace_seq_printf(s, ": %lx : ", field->ip);
  	trace_seq_puts(s, field->str);

  	return trace_handle_return(s);

  }
  
  static struct trace_event_functions trace_bputs_funcs = {
  	.trace		= trace_bputs_print,
  	.raw		= trace_bputs_raw,
  };
  
  static struct trace_event trace_bputs_event = {
  	.type		= TRACE_BPUTS,
  	.funcs		= &trace_bputs_funcs,
  };

  /* TRACE_BPRINT */

  static enum print_line_t

  trace_bprint_print(struct trace_iterator *iter, int flags,
  		   struct trace_event *event)

  {
  	struct trace_entry *entry = iter->ent;
  	struct trace_seq *s = &iter->seq;

  	struct bprint_entry *field;

  
  	trace_assign_type(field, entry);

  	seq_print_ip_sym(s, field->ip, flags);
  	trace_seq_puts(s, ": ");
  	trace_seq_bprintf(s, field->fmt, field->buf);

  	return trace_handle_return(s);

  }

  static enum print_line_t

  trace_bprint_raw(struct trace_iterator *iter, int flags,
  		 struct trace_event *event)

  {

  	struct bprint_entry *field;

  	struct trace_seq *s = &iter->seq;

  	trace_assign_type(field, iter->ent);

  	trace_seq_printf(s, ": %lx : ", field->ip);
  	trace_seq_bprintf(s, field->fmt, field->buf);

  	return trace_handle_return(s);

  }

  static struct trace_event_functions trace_bprint_funcs = {
  	.trace		= trace_bprint_print,
  	.raw		= trace_bprint_raw,
  };

  static struct trace_event trace_bprint_event = {
  	.type		= TRACE_BPRINT,

  	.funcs		= &trace_bprint_funcs,

  };
  
  /* TRACE_PRINT */
  static enum print_line_t trace_print_print(struct trace_iterator *iter,

  					   int flags, struct trace_event *event)

  {
  	struct print_entry *field;
  	struct trace_seq *s = &iter->seq;
  
  	trace_assign_type(field, iter->ent);

  	seq_print_ip_sym(s, field->ip, flags);
  	trace_seq_printf(s, ": %s", field->buf);

  	return trace_handle_return(s);

  }

  static enum print_line_t trace_print_raw(struct trace_iterator *iter, int flags,
  					 struct trace_event *event)

  {
  	struct print_entry *field;
  
  	trace_assign_type(field, iter->ent);

  	trace_seq_printf(&iter->seq, "# %lx %s", field->ip, field->buf);

  	return trace_handle_return(&iter->seq);

  }

  static struct trace_event_functions trace_print_funcs = {

  	.trace		= trace_print_print,
  	.raw		= trace_print_raw,

  };

  static struct trace_event trace_print_event = {
  	.type	 	= TRACE_PRINT,
  	.funcs		= &trace_print_funcs,
  };

  static struct trace_event *events[] __initdata = {
  	&trace_fn_event,
  	&trace_ctx_event,
  	&trace_wake_event,

  	&trace_stack_event,
  	&trace_user_stack_event,

  	&trace_bputs_event,

  	&trace_bprint_event,

  	&trace_print_event,
  	NULL
  };
  
  __init static int init_events(void)
  {
  	struct trace_event *event;
  	int i, ret;
  
  	for (i = 0; events[i]; i++) {
  		event = events[i];

  		ret = register_trace_event(event);

  		if (!ret) {
  			printk(KERN_WARNING "event %d failed to register
  ",
  			       event->type);
  			WARN_ON_ONCE(1);
  		}
  	}
  
  	return 0;
  }

  early_initcall(init_events);