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