--- /dev/null
+/*
+ * BTS tracer
+ *
+ * Copyright (C) 2008 Markus Metzger <markus.t.metzger@gmail.com>
+ *
+ */
+
+#include <linux/module.h>
+#include <linux/fs.h>
+#include <linux/debugfs.h>
+#include <linux/ftrace.h>
+#include <linux/kallsyms.h>
+
+#include <asm/ds.h>
+
+#include "trace.h"
+
+
+#define SIZEOF_BTS (1 << 13)
+
+static DEFINE_PER_CPU(struct bts_tracer *, tracer);
+static DEFINE_PER_CPU(unsigned char[SIZEOF_BTS], buffer);
+
+#define this_tracer per_cpu(tracer, smp_processor_id())
+#define this_buffer per_cpu(buffer, smp_processor_id())
+
+
+/*
+ * Information to interpret a BTS record.
+ * This will go into an in-kernel BTS interface.
+ */
+static unsigned char sizeof_field;
+static unsigned long debugctl_mask;
+
+#define sizeof_bts (3 * sizeof_field)
+
+static void bts_trace_cpuinit(struct cpuinfo_x86 *c)
+{
+       switch (c->x86) {
+       case 0x6:
+               switch (c->x86_model) {
+               case 0x0 ... 0xC:
+                       break;
+               case 0xD:
+               case 0xE: /* Pentium M */
+                       sizeof_field = sizeof(long);
+                       debugctl_mask = (1<<6)|(1<<7);
+                       break;
+               default:
+                       sizeof_field = 8;
+                       debugctl_mask = (1<<6)|(1<<7);
+                       break;
+               }
+               break;
+       case 0xF:
+               switch (c->x86_model) {
+               case 0x0:
+               case 0x1:
+               case 0x2: /* Netburst */
+                       sizeof_field = sizeof(long);
+                       debugctl_mask = (1<<2)|(1<<3);
+                       break;
+               default:
+                       /* sorry, don't know about them */
+                       break;
+               }
+               break;
+       default:
+               /* sorry, don't know about them */
+               break;
+       }
+}
+
+static inline void bts_enable(void)
+{
+       unsigned long debugctl;
+
+       rdmsrl(MSR_IA32_DEBUGCTLMSR, debugctl);
+       wrmsrl(MSR_IA32_DEBUGCTLMSR, debugctl | debugctl_mask);
+}
+
+static inline void bts_disable(void)
+{
+       unsigned long debugctl;
+
+       rdmsrl(MSR_IA32_DEBUGCTLMSR, debugctl);
+       wrmsrl(MSR_IA32_DEBUGCTLMSR, debugctl & ~debugctl_mask);
+}
+
+static void bts_trace_reset(struct trace_array *tr)
+{
+       int cpu;
+
+       tr->time_start = ftrace_now(tr->cpu);
+
+       for_each_online_cpu(cpu)
+               tracing_reset(tr, cpu);
+}
+
+static void bts_trace_start_cpu(void *arg)
+{
+       this_tracer =
+               ds_request_bts(/* task = */ NULL, this_buffer, SIZEOF_BTS,
+                              /* ovfl = */ NULL, /* th = */ (size_t)-1);
+       if (IS_ERR(this_tracer)) {
+               this_tracer = NULL;
+               return;
+       }
+
+       bts_enable();
+}
+
+static void bts_trace_start(struct trace_array *tr)
+{
+       int cpu;
+
+       bts_trace_reset(tr);
+
+       for_each_cpu_mask(cpu, cpu_possible_map)
+               smp_call_function_single(cpu, bts_trace_start_cpu, NULL, 1);
+}
+
+static void bts_trace_stop_cpu(void *arg)
+{
+       if (this_tracer) {
+               bts_disable();
+
+               ds_release_bts(this_tracer);
+               this_tracer = NULL;
+       }
+}
+
+static void bts_trace_stop(struct trace_array *tr)
+{
+       int cpu;
+
+       for_each_cpu_mask(cpu, cpu_possible_map)
+               smp_call_function_single(cpu, bts_trace_stop_cpu, NULL, 1);
+}
+
+static int bts_trace_init(struct trace_array *tr)
+{
+       bts_trace_cpuinit(&boot_cpu_data);
+       bts_trace_reset(tr);
+       bts_trace_start(tr);
+
+       return 0;
+}
+
+static void bts_trace_print_header(struct seq_file *m)
+{
+#ifdef __i386__
+       seq_puts(m, "# CPU#    FROM           TO     FUNCTION\n");
+       seq_puts(m, "#  |       |             |         |\n");
+#else
+       seq_puts(m,
+                "# CPU#        FROM                   TO         FUNCTION\n");
+       seq_puts(m,
+                "#  |           |                     |             |\n");
+#endif
+}
+
+static enum print_line_t bts_trace_print_line(struct trace_iterator *iter)
+{
+       struct trace_entry *entry = iter->ent;
+       struct trace_seq *seq = &iter->seq;
+       struct bts_entry *it;
+
+       trace_assign_type(it, entry);
+
+       if (entry->type == TRACE_BTS) {
+               int ret;
+#ifdef CONFIG_KALLSYMS
+               char function[KSYM_SYMBOL_LEN];
+               sprint_symbol(function, it->from);
+#else
+               char *function = "<unknown>";
+#endif
+
+               ret = trace_seq_printf(seq, "%4d  0x%lx -> 0x%lx [%s]\n",
+                                      entry->cpu, it->from, it->to, function);
+               if (!ret)
+                       return TRACE_TYPE_PARTIAL_LINE;;
+               return TRACE_TYPE_HANDLED;
+       }
+       return TRACE_TYPE_UNHANDLED;
+}
+
+void trace_bts(struct trace_array *tr, unsigned long from, unsigned long to)
+{
+       struct ring_buffer_event *event;
+       struct bts_entry *entry;
+       unsigned long irq;
+
+       event = ring_buffer_lock_reserve(tr->buffer, sizeof(*entry), &irq);
+       if (!event)
+               return;
+       entry   = ring_buffer_event_data(event);
+       tracing_generic_entry_update(&entry->ent, 0, from);
+       entry->ent.type = TRACE_BTS;
+       entry->ent.cpu = smp_processor_id();
+       entry->from = from;
+       entry->to   = to;
+       ring_buffer_unlock_commit(tr->buffer, event, irq);
+}
+
+static void trace_bts_at(struct trace_array *tr, size_t index)
+{
+       const void *raw = NULL;
+       unsigned long from, to;
+       int err;
+
+       err = ds_access_bts(this_tracer, index, &raw);
+       if (err < 0)
+               return;
+
+       from = *(const unsigned long *)raw;
+       to = *(const unsigned long *)((const char *)raw + sizeof_field);
+
+       trace_bts(tr, from, to);
+}
+
+static void trace_bts_cpu(void *arg)
+{
+       struct trace_array *tr = (struct trace_array *) arg;
+       size_t index = 0, end = 0, i;
+       int err;
+
+       if (!this_tracer)
+               return;
+
+       bts_disable();
+
+       err = ds_get_bts_index(this_tracer, &index);
+       if (err < 0)
+               goto out;
+
+       err = ds_get_bts_end(this_tracer, &end);
+       if (err < 0)
+               goto out;
+
+       for (i = index; i < end; i++)
+               trace_bts_at(tr, i);
+
+       for (i = 0; i < index; i++)
+               trace_bts_at(tr, i);
+
+out:
+       bts_enable();
+}
+
+static void trace_bts_prepare(struct trace_iterator *iter)
+{
+       int cpu;
+
+       for_each_cpu_mask(cpu, cpu_possible_map)
+               smp_call_function_single(cpu, trace_bts_cpu, iter->tr, 1);
+}
+
+struct tracer bts_tracer __read_mostly =
+{
+       .name           = "bts",
+       .init           = bts_trace_init,
+       .reset          = bts_trace_stop,
+       .print_header   = bts_trace_print_header,
+       .print_line     = bts_trace_print_line,
+       .start          = bts_trace_start,
+       .stop           = bts_trace_stop,
+       .open           = trace_bts_prepare
+};
+
+__init static int init_bts_trace(void)
+{
+       return register_tracer(&bts_tracer);
+}
+device_initcall(init_bts_trace);