random_seed.c 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238
  1. /*
  2. * random_seed.c
  3. *
  4. * Copyright (c) 2013 Metaparadigm Pte. Ltd.
  5. * Michael Clark <michael@metaparadigm.com>
  6. *
  7. * This library is free software; you can redistribute it and/or modify
  8. * it under the terms of the MIT license. See COPYING for details.
  9. *
  10. */
  11. #include "strerror_override.h"
  12. #include <stdio.h>
  13. #include "config.h"
  14. #include "random_seed.h"
  15. #define DEBUG_SEED(s)
  16. #if defined ENABLE_RDRAND
  17. /* cpuid */
  18. #if defined __GNUC__ && (defined __i386__ || defined __x86_64__)
  19. #define HAS_X86_CPUID 1
  20. static void do_cpuid(int regs[], int h)
  21. {
  22. __asm__ __volatile__(
  23. #if defined __x86_64__
  24. "pushq %%rbx;\n"
  25. #else
  26. "pushl %%ebx;\n"
  27. #endif
  28. "cpuid;\n"
  29. #if defined __x86_64__
  30. "popq %%rbx;\n"
  31. #else
  32. "popl %%ebx;\n"
  33. #endif
  34. : "=a"(regs[0]), [ebx] "=r"(regs[1]), "=c"(regs[2]), "=d"(regs[3])
  35. : "a"(h));
  36. }
  37. #elif defined _MSC_VER
  38. #define HAS_X86_CPUID 1
  39. #define do_cpuid __cpuid
  40. #endif
  41. /* has_rdrand */
  42. #if HAS_X86_CPUID
  43. static int has_rdrand()
  44. {
  45. // CPUID.01H:ECX.RDRAND[bit 30] == 1
  46. int regs[4];
  47. do_cpuid(regs, 1);
  48. return (regs[2] & (1 << 30)) != 0;
  49. }
  50. #endif
  51. /* get_rdrand_seed - GCC x86 and X64 */
  52. #if defined __GNUC__ && (defined __i386__ || defined __x86_64__)
  53. #define HAVE_RDRAND 1
  54. static int get_rdrand_seed()
  55. {
  56. DEBUG_SEED("get_rdrand_seed");
  57. int _eax;
  58. // rdrand eax
  59. __asm__ __volatile__("1: .byte 0x0F\n"
  60. " .byte 0xC7\n"
  61. " .byte 0xF0\n"
  62. " jnc 1b;\n"
  63. : "=a" (_eax));
  64. return _eax;
  65. }
  66. #endif
  67. #if defined _MSC_VER
  68. #if _MSC_VER >= 1700
  69. #define HAVE_RDRAND 1
  70. /* get_rdrand_seed - Visual Studio 2012 and above */
  71. static int get_rdrand_seed()
  72. {
  73. DEBUG_SEED("get_rdrand_seed");
  74. int r;
  75. while (_rdrand32_step(&r) == 0);
  76. return r;
  77. }
  78. #elif defined _M_IX86
  79. #define HAVE_RDRAND 1
  80. /* get_rdrand_seed - Visual Studio 2010 and below - x86 only */
  81. static int get_rdrand_seed()
  82. {
  83. DEBUG_SEED("get_rdrand_seed");
  84. int _eax;
  85. retry:
  86. // rdrand eax
  87. __asm _emit 0x0F __asm _emit 0xC7 __asm _emit 0xF0
  88. __asm jnc retry
  89. __asm mov _eax, eax
  90. return _eax;
  91. }
  92. #endif
  93. #endif
  94. #endif /* defined ENABLE_RDRAND */
  95. /* has_dev_urandom */
  96. #if defined (__APPLE__) || defined(__unix__) || defined(__linux__)
  97. #include <string.h>
  98. #include <fcntl.h>
  99. #include <unistd.h>
  100. #include <stdlib.h>
  101. #include <sys/stat.h>
  102. #define HAVE_DEV_RANDOM 1
  103. static const char *dev_random_file = "/dev/urandom";
  104. static int has_dev_urandom()
  105. {
  106. struct stat buf;
  107. if (stat(dev_random_file, &buf)) {
  108. return 0;
  109. }
  110. return ((buf.st_mode & S_IFCHR) != 0);
  111. }
  112. /* get_dev_random_seed */
  113. static int get_dev_random_seed()
  114. {
  115. DEBUG_SEED("get_dev_random_seed");
  116. int fd = open(dev_random_file, O_RDONLY);
  117. if (fd < 0) {
  118. fprintf(stderr, "error opening %s: %s", dev_random_file, strerror(errno));
  119. exit(1);
  120. }
  121. int r;
  122. ssize_t nread = read(fd, &r, sizeof(r));
  123. if (nread != sizeof(r)) {
  124. fprintf(stderr, "error short read %s: %s", dev_random_file, strerror(errno));
  125. exit(1);
  126. }
  127. close(fd);
  128. return r;
  129. }
  130. #endif
  131. /* get_cryptgenrandom_seed */
  132. #ifdef WIN32
  133. #define HAVE_CRYPTGENRANDOM 1
  134. #include <windows.h>
  135. #include <wincrypt.h>
  136. #ifndef __GNUC__
  137. #pragma comment(lib, "advapi32.lib")
  138. #endif
  139. static int get_cryptgenrandom_seed()
  140. {
  141. HCRYPTPROV hProvider = 0;
  142. int r;
  143. DEBUG_SEED("get_cryptgenrandom_seed");
  144. if (!CryptAcquireContextW(&hProvider, 0, 0, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT | CRYPT_SILENT)) {
  145. fprintf(stderr, "error CryptAcquireContextW");
  146. exit(1);
  147. }
  148. if (!CryptGenRandom(hProvider, sizeof(r), (BYTE*)&r)) {
  149. fprintf(stderr, "error CryptGenRandom");
  150. exit(1);
  151. }
  152. CryptReleaseContext(hProvider, 0);
  153. return r;
  154. }
  155. #endif
  156. /* get_time_seed */
  157. #include <time.h>
  158. static int get_time_seed()
  159. {
  160. DEBUG_SEED("get_time_seed");
  161. return (int)time(NULL) * 433494437;
  162. }
  163. /* json_c_get_random_seed */
  164. int json_c_get_random_seed()
  165. {
  166. #if HAVE_RDRAND
  167. if (has_rdrand()) return get_rdrand_seed();
  168. #endif
  169. #if HAVE_DEV_RANDOM
  170. if (has_dev_urandom()) return get_dev_random_seed();
  171. #endif
  172. #if HAVE_CRYPTGENRANDOM
  173. return get_cryptgenrandom_seed();
  174. #endif
  175. return get_time_seed();
  176. }