markers-test/test-mark-speed-edit.c

   1 /* test-mark.c
   2  *
   3  */
   4
   5 #include <linux/module.h>
   6 #include <linux/proc_fs.h>
   7 #include <linux/sched.h>
   8 #include <linux/timex.h>
   9 #include <linux/marker.h>
  10 #include <asm/ptrace.h>
  11 #include <asm/system.h>
  12
  13 static void pmc_flush_cache(void)
  14   {
  15     register int i;
  16     /* write back and invalidate cache (a serializing instruction) */
  17
  18     __asm__ __volatile__ ( "wbinvd" : : : "memory" );
  19
  20     /* The wbinvd instruction does not wait for the external caches
  21      * to be flushed, but only requests that it be done.  The loop
  22      * is to be sure that enough time has elapsed, but the compiler
  23      * might simplify or even remove it.  The loop bound is for a
  24      * 512 KB L2 cache.  On a Pentium Pro/II/III, the loop uses
  25      * 2 cycles per iteration.
  26      *
  27      * Does wbinvd also cause the TLB to be flushed?
  28      * A comment in mtrr.c suggests that it does.
  29      */
  30
  31      for (i = 0; i < 512*1024; i++) {
  32        cpu_relax();
  33      }
  34   }
  35
  36
  37
  38 static void noinline test2(const struct marker *mdata,
  39         void *call_private, ...)
  40 {
  41         unsigned char *ins = __builtin_return_address(0) - 5;
  42 #if 0
  43         /* not called */
  44         printk("ip %p\n", __builtin_return_address(0));
  45         printk("prev_ins %hX %hX %hX %hX %hX\n",
  46                 ins[0], ins[1], ins[2], ins[3], ins[4]);
  47 #endif //0
  48         ins[0] = 0x90;
  49         ins[1] = 0x8d;
  50         ins[2] = 0x74;
  51         ins[3] = 0x26;
  52         ins[4] = 0x00;
  53 }
  54
  55 /*
  56  * Generic marker flavor always available.
  57  * Note : the empty asm volatile with read constraint is used here instead of a
  58  * "used" attribute to fix a gcc 4.1.x bug.
  59  * Make sure the alignment of the structure in the __markers section will
  60  * not add unwanted padding between the beginning of the section and the
  61  * structure. Force alignment to the same alignment as the section start.
  62  */
  63 #define __my_trace_mark(generic, name, call_private, format, args...)   \
  64         do {                                                            \
  65                 static const char __mstrtab_##name[]                    \
  66                 __attribute__((section("__markers_strings")))           \
  67                 = #name "\0" format;                                    \
  68                 static struct marker __mark_##name                      \
  69                 __attribute__((section("__markers"), aligned(8))) =     \
  70                 { __mstrtab_##name, &__mstrtab_##name[sizeof(#name)],   \
  71                 0, 0, marker_probe_cb,                                  \
  72                 { __mark_empty_function, NULL}, NULL };                 \
  73                 __mark_check_format(format, ## args);                   \
  74                 if (!generic) {                                         \
  75                         if (unlikely(imv_read(__mark_##name.state)))    \
  76                                 test2                   \
  77                                         (&__mark_##name, call_private,  \
  78                                         ## args);               \
  79                 } else {                                                \
  80                         if (unlikely(_imv_read(__mark_##name.state)))   \
  81                                 test2                   \
  82                                         (&__mark_##name, call_private,  \
  83                                         ## args);               \
  84                 }                                                       \
  85         } while (0)
  86
  87         //asm volatile ("");
  88 struct proc_dir_entry *pentry = NULL;
  89
  90 static inline void test(unsigned long arg, unsigned long arg2)
  91 {
  92         volatile int temp[5];
  93 #ifdef CACHEFLUSH
  94         pmc_flush_cache();
  95 #endif
  96         temp[2] = (temp[0] + 60) << 10;
  97         temp[3] = (temp[2] + 60) << 10;
  98         temp[4] = (temp[3] + 60) << 10;
  99         temp[0] = (temp[4] + 60) << 10;
 100         //asm volatile ("");
 101         //__my_trace_mark(1, kernel_debug_test, NULL, "%d %d %ld %ld", 2, current->pid, arg, arg2);
 102         barrier();
 103         test2(NULL, NULL, 2, current->pid, arg, arg2);
 104         barrier();
 105         //__my_trace_mark(0, kernel_debug_test, NULL, "%d %d %ld %ld", 2, current->pid, arg, arg2);
 106 }
 107
 108 static int my_open(struct inode *inode, struct file *file)
 109 {
 110         unsigned int i;
 111         cycles_t cycles1, cycles2;
 112         unsigned long flags;
 113
 114         local_irq_save(flags);
 115 #ifdef CACHEFLUSH
 116         pmc_flush_cache();      /* initial write back, without cycle count */
 117         msleep(20);     /* wait for L2 flush */
 118 #endif
 119         rdtsc_barrier();
 120         cycles1 = get_cycles();
 121         rdtsc_barrier();
 122         for(i=0; i<2000; i++) {
 123                 test(i, i);
 124                 test(i, i);
 125                 test(i, i);
 126                 test(i, i);
 127                 test(i, i);
 128                 test(i, i);
 129                 test(i, i);
 130                 test(i, i);
 131                 test(i, i);
 132                 test(i, i);
 133         }
 134         rdtsc_barrier();
 135         cycles2 = get_cycles();
 136         rdtsc_barrier();
 137         local_irq_restore(flags);
 138         printk("cycles : %llu\n", cycles2-cycles1);
 139         return -EPERM;
 140 }
 141
 142
 143 static struct file_operations my_operations = {
 144         .open = my_open,
 145 };
 146
 147 int init_module(void)
 148 {
 149         pentry = create_proc_entry("testmark", 0444, NULL);
 150         if (pentry)
 151                 pentry->proc_fops = &my_operations;
 152
 153         return 0;
 154 }
 155
 156 void cleanup_module(void)
 157 {
 158         remove_proc_entry("testmark", NULL);
 159 }
 160
 161 MODULE_LICENSE("GPL");
 162 MODULE_AUTHOR("Mathieu Desnoyers");
 163 MODULE_DESCRIPTION("Marker Test");
 164 MODULE_VERSION("1.0");
 165