isaac.c 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138
  1. /*Written by Timothy B. Terriberry (tterribe@xiph.org) 1999-2009 public domain.
  2. Based on the public domain implementation by Robert J. Jenkins Jr.*/
  3. #include <string.h>
  4. #include "isaac.h"
  5. #define ISAAC_MASK (0xFFFFFFFFU)
  6. static void isaac_update(isaac_ctx *_ctx){
  7. unsigned *m;
  8. unsigned *r;
  9. unsigned a;
  10. unsigned b;
  11. unsigned x;
  12. unsigned y;
  13. int i;
  14. m=_ctx->m;
  15. r=_ctx->r;
  16. a=_ctx->a;
  17. b=_ctx->b+(++_ctx->c)&ISAAC_MASK;
  18. for(i=0;i<ISAAC_SZ/2;i++){
  19. x=m[i];
  20. a=(a^a<<13)+m[i+ISAAC_SZ/2]&ISAAC_MASK;
  21. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  22. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  23. x=m[++i];
  24. a=(a^a>>6)+m[i+ISAAC_SZ/2]&ISAAC_MASK;
  25. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  26. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  27. x=m[++i];
  28. a=(a^a<<2)+m[i+ISAAC_SZ/2]&ISAAC_MASK;
  29. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  30. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  31. x=m[++i];
  32. a=(a^a>>16)+m[i+ISAAC_SZ/2]&ISAAC_MASK;
  33. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  34. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  35. }
  36. for(i=ISAAC_SZ/2;i<ISAAC_SZ;i++){
  37. x=m[i];
  38. a=(a^a<<13)+m[i-ISAAC_SZ/2]&ISAAC_MASK;
  39. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  40. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  41. x=m[++i];
  42. a=(a^a>>6)+m[i-ISAAC_SZ/2]&ISAAC_MASK;
  43. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  44. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  45. x=m[++i];
  46. a=(a^a<<2)+m[i-ISAAC_SZ/2]&ISAAC_MASK;
  47. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  48. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  49. x=m[++i];
  50. a=(a^a>>16)+m[i-ISAAC_SZ/2]&ISAAC_MASK;
  51. m[i]=y=m[(x&ISAAC_SZ-1<<2)>>2]+a+b&ISAAC_MASK;
  52. r[i]=b=m[y>>ISAAC_SZ_LOG+2&ISAAC_SZ-1]+x&ISAAC_MASK;
  53. }
  54. _ctx->b=b;
  55. _ctx->a=a;
  56. _ctx->n=ISAAC_SZ;
  57. }
  58. static void isaac_mix(unsigned _x[8]){
  59. static const unsigned char SHIFT[8]={11,2,8,16,10,4,8,9};
  60. int i;
  61. for(i=0;i<8;i++){
  62. _x[i]^=_x[i+1&7]<<SHIFT[i];
  63. _x[i+3&7]+=_x[i];
  64. _x[i+1&7]+=_x[i+2&7];
  65. i++;
  66. _x[i]^=_x[i+1&7]>>SHIFT[i];
  67. _x[i+3&7]+=_x[i];
  68. _x[i+1&7]+=_x[i+2&7];
  69. }
  70. }
  71. void isaac_init(isaac_ctx *_ctx,const void *_seed,int _nseed){
  72. const unsigned char *seed;
  73. unsigned *m;
  74. unsigned *r;
  75. unsigned x[8];
  76. int i;
  77. int j;
  78. _ctx->a=_ctx->b=_ctx->c=0;
  79. m=_ctx->m;
  80. r=_ctx->r;
  81. x[0]=x[1]=x[2]=x[3]=x[4]=x[5]=x[6]=x[7]=0x9E3779B9;
  82. for(i=0;i<4;i++)isaac_mix(x);
  83. if(_nseed>ISAAC_SEED_SZ_MAX)_nseed=ISAAC_SEED_SZ_MAX;
  84. seed=(const unsigned char *)_seed;
  85. for(i=0;i<_nseed>>2;i++){
  86. r[i]=seed[i<<2|3]<<24|seed[i<<2|2]<<16|seed[i<<2|1]<<8|seed[i<<2];
  87. }
  88. if(_nseed&3){
  89. r[i]=seed[i<<2];
  90. for(j=1;j<(_nseed&3);j++)r[i]+=seed[i<<2|j]<<(j<<3);
  91. i++;
  92. }
  93. memset(r+i,0,(ISAAC_SZ-i)*sizeof(*r));
  94. for(i=0;i<ISAAC_SZ;i+=8){
  95. for(j=0;j<8;j++)x[j]+=r[i+j];
  96. isaac_mix(x);
  97. memcpy(m+i,x,sizeof(x));
  98. }
  99. for(i=0;i<ISAAC_SZ;i+=8){
  100. for(j=0;j<8;j++)x[j]+=m[i+j];
  101. isaac_mix(x);
  102. memcpy(m+i,x,sizeof(x));
  103. }
  104. isaac_update(_ctx);
  105. }
  106. unsigned isaac_next_uint32(isaac_ctx *_ctx){
  107. if(!_ctx->n)isaac_update(_ctx);
  108. return _ctx->r[--_ctx->n];
  109. }
  110. /*Returns a uniform random integer less than the given maximum value.
  111. _n: The upper bound on the range of numbers returned (not inclusive).
  112. This must be strictly less than 2**32.
  113. Return: An integer uniformly distributed between 0 (inclusive) and _n
  114. (exclusive).*/
  115. unsigned isaac_next_uint(isaac_ctx *_ctx,unsigned _n){
  116. unsigned r;
  117. unsigned v;
  118. unsigned d;
  119. do{
  120. r=isaac_next_uint32(_ctx);
  121. v=r%_n;
  122. d=r-v;
  123. }
  124. while((d+_n-1&ISAAC_MASK)<d);
  125. return v;
  126. }