#include <intrin.h>
#include <stdio.h>
#include <Windows.h>

#define SHIFT_NUMBER          0x0C
#define PAGE_SIZE             4096
#define BLOCK_SIZE            (1 << SHIFT_NUMBER)


void CacheLineFlush(LPBYTE lpArray, UINT index) {
  _mm_clflush(&(lpArray[index << SHIFT_NUMBER]));
}

void CacheLineFlush_all(LPBYTE lpArray) {
  for (UINT i = 0; i < 256; i++) CacheLineFlush(lpArray, i);
}

/************************************************************************/
/* 统计各个块的访问速度，并返回最快的那个块的索引                       */
/************************************************************************/
BYTE GetAccessByte(LPBYTE lpArray) {
  UINT64 speed[256];
  UINT64 start, min;
  UINT index, junk;
  BYTE result;

  //为min赋初始值
  min = 0;

  //测试访问速度
  for (int i = 0; i < 256; i++) {
    //获取array[index]的地址
    index = i << SHIFT_NUMBER;
    //mfence指令用于序列化内存访问，即让乱序执行无效化。
    //后面的指令必须在前面的内存读写完成后再开始发射执行。
    _mm_mfence();
    //记录开始周期
    start = __rdtscp(&junk);
    junk = *(LPDWORD)(&lpArray[index]);
    _mm_mfence();
    speed[i] = __rdtscp(&junk) - start;

    //如果是初始值，或者比当前值还小
    if ((min == 0) || (speed[i] < min)) {
      min = speed[i];
      result = (BYTE)i;
    }
  }

  return result;
}

/************************************************************************/
/* 用index作为数组索引，访问array的某个元素                             */
/************************************************************************/
BYTE AccessArray(LPBYTE lpArray, UINT index) {
  return lpArray[index << SHIFT_NUMBER];
}


int main()
{
  //假定的kernel内存
  BYTE kernel[4];
  //array是用户可控制的内存
  LPBYTE array;

  kernel[0] = 0x55;
  kernel[1] = 0xAA;
  kernel[2] = 0xF0;
  kernel[3] = 0x0F;

  array = (LPBYTE)_aligned_malloc(256 * BLOCK_SIZE, PAGE_SIZE);

  /*
  实现原理:
    kernel假定是受保护的内存数据 (本demo里可访问)。
    array为用户可控制的一个数组，一共分为256个块，每个块的大小为2的整数倍: (1 << SHIFT_NUMBER)。
    然后从kernel中读取一个byte，以这个byte为索引，去访问array所对应的块。
    之后立刻循环读取一遍array的各个块，如果之前访问成功了，那么对应的块应该还在缓存中，对应的访问时间要少很多。
    统计各个块的访问周期数，最快的块，他的索引就是受保护的那个byte。

    CacheLineFlush_all 函数用于把整个数组从缓存中清除出去，这样不至于污染访问速度。
    AccessArray 函数用于以一个索引去访问一个数组。
    GetAccessByte 函数用于测试数组的各个块的访问速度，返回最快的那个块的索引。
  */


  CacheLineFlush_all(array);
  AccessArray(array, kernel[0]);
  printf("Access fastest: 0x%02X\n", (DWORD)GetAccessByte(array));

  CacheLineFlush_all(array);
  AccessArray(array, kernel[1]);
  printf("Access fastest: 0x%02X\n", (DWORD)GetAccessByte(array));

  CacheLineFlush_all(array);
  AccessArray(array, kernel[2]);
  printf("Access fastest: 0x%02X\n", (DWORD)GetAccessByte(array));

  CacheLineFlush_all(array);
  AccessArray(array, kernel[3]);
  printf("Access fastest: 0x%02X\n", (DWORD)GetAccessByte(array));

  getchar();

  return 0;
}