lunaix-os/arch/x86/hal/rngx86.c

   1 #include <lunaix/device.h>
   2 #include <lunaix/mm/pagetable.h>
   3
   4 #include <klibc/string.h>
   5
   6 static inline void
   7 rng_fill(void* data, size_t len)
   8 {
   9 #ifdef CONFIG_ARCH_X86_64
  10     asm volatile("1:\n"
  11                  "subq   $8, %1\n"
  12                  "rdrand %%rax\n"
  13                  "movq   %%rax, (%0, %1, 1)\n"
  14                  "addq   $8, %%rax\n"
  15                  "testq  %1, %1\n"
  16                  "jnz    1b"
  17                  ::
  18                  "r"((ptr_t)data),
  19                  "r"((len & ~0x7))
  20                  :
  21                  "rax");
  22 #else
  23     asm volatile("1:\n"
  24                  "subl $4, %1\n"
  25                  "rdrand %%eax\n"
  26                  "movl %%eax, (%0, %1, 1)\n"
  27                  "addl $4, %%eax\n"
  28                  "testl  %1, %1\n"
  29                  "jnz 1b"
  30                  ::
  31                  "r"((ptr_t)data),
  32                  "r"((len & ~0x3))
  33                  :
  34                  "%eax");
  35 #endif
  36 }
  37
  38 static int
  39 __rand_rd_pg(struct device* dev, void* buf, size_t offset)
  40 {
  41     rng_fill(buf, PAGE_SIZE);
  42     return PAGE_SIZE;
  43 }
  44
  45 static int
  46 __rand_rd(struct device* dev, void* buf, size_t offset, size_t len)
  47 {
  48     if (unlikely(len < sizeof(ptr_t))) {
  49         int tmp_buf = 0;
  50         rng_fill(&tmp_buf, sizeof(ptr_t));
  51         memcpy(buf, &tmp_buf, len);
  52     } else {
  53         rng_fill(buf, len);
  54     }
  55     return len;
  56 }
  57
  58 int
  59 pdev_randdev_init(struct device_def* devdef)
  60 {
  61     // FIXME add check on cpuid for presence of rdrand
  62     struct device* devrand = device_allocseq(NULL, NULL);
  63     devrand->ops.read = __rand_rd;
  64     devrand->ops.read_page = __rand_rd_pg;
  65
  66     register_device(devrand, &devdef->class, "rand");
  67
  68     return 0;
  69 }
  70
  71 static struct device_def devrandx86_def = {
  72     .name = "x86 On-Chip RNG",
  73     .class = DEVCLASS(DEVIF_SOC, DEVFN_CHAR, DEV_RNG),
  74     .init = pdev_randdev_init};
  75 EXPORT_DEVICE(randdev, &devrandx86_def, load_onboot);