random.c
上传用户:sunhongbo
上传日期:2022-01-25
资源大小:3010k
文件大小:4k
源码类别:

数据库系统

开发平台:

C/C++

  1. /*
  2. ** 2001 September 15
  3. **
  4. ** The author disclaims copyright to this source code.  In place of
  5. ** a legal notice, here is a blessing:
  6. **
  7. **    May you do good and not evil.
  8. **    May you find forgiveness for yourself and forgive others.
  9. **    May you share freely, never taking more than you give.
  10. **
  11. *************************************************************************
  12. ** This file contains code to implement a pseudo-random number
  13. ** generator (PRNG) for SQLite.
  14. **
  15. ** Random numbers are used by some of the database backends in order
  16. ** to generate random integer keys for tables or random filenames.
  17. **
  18. ** $Id: random.c,v 1.23 2008/03/21 16:45:47 drh Exp $
  19. */
  20. #include "sqliteInt.h"
  21. /* All threads share a single random number generator.
  22. ** This structure is the current state of the generator.
  23. */
  24. static struct sqlite3PrngType {
  25.   unsigned char isInit;          /* True if initialized */
  26.   unsigned char i, j;            /* State variables */
  27.   unsigned char s[256];          /* State variables */
  28. } sqlite3Prng;
  29. /*
  30. ** Get a single 8-bit random value from the RC4 PRNG.  The Mutex
  31. ** must be held while executing this routine.
  32. **
  33. ** Why not just use a library random generator like lrand48() for this?
  34. ** Because the OP_NewRowid opcode in the VDBE depends on having a very
  35. ** good source of random numbers.  The lrand48() library function may
  36. ** well be good enough.  But maybe not.  Or maybe lrand48() has some
  37. ** subtle problems on some systems that could cause problems.  It is hard
  38. ** to know.  To minimize the risk of problems due to bad lrand48()
  39. ** implementations, SQLite uses this random number generator based
  40. ** on RC4, which we know works very well.
  41. **
  42. ** (Later):  Actually, OP_NewRowid does not depend on a good source of
  43. ** randomness any more.  But we will leave this code in all the same.
  44. */
  45. static int randomByte(void){
  46.   unsigned char t;
  47.   /* Initialize the state of the random number generator once,
  48.   ** the first time this routine is called.  The seed value does
  49.   ** not need to contain a lot of randomness since we are not
  50.   ** trying to do secure encryption or anything like that...
  51.   **
  52.   ** Nothing in this file or anywhere else in SQLite does any kind of
  53.   ** encryption.  The RC4 algorithm is being used as a PRNG (pseudo-random
  54.   ** number generator) not as an encryption device.
  55.   */
  56.   if( !sqlite3Prng.isInit ){
  57.     int i;
  58.     char k[256];
  59.     sqlite3Prng.j = 0;
  60.     sqlite3Prng.i = 0;
  61.     sqlite3OsRandomness(sqlite3_vfs_find(0), 256, k);
  62.     for(i=0; i<256; i++){
  63.       sqlite3Prng.s[i] = i;
  64.     }
  65.     for(i=0; i<256; i++){
  66.       sqlite3Prng.j += sqlite3Prng.s[i] + k[i];
  67.       t = sqlite3Prng.s[sqlite3Prng.j];
  68.       sqlite3Prng.s[sqlite3Prng.j] = sqlite3Prng.s[i];
  69.       sqlite3Prng.s[i] = t;
  70.     }
  71.     sqlite3Prng.isInit = 1;
  72.   }
  73.   /* Generate and return single random byte
  74.   */
  75.   sqlite3Prng.i++;
  76.   t = sqlite3Prng.s[sqlite3Prng.i];
  77.   sqlite3Prng.j += t;
  78.   sqlite3Prng.s[sqlite3Prng.i] = sqlite3Prng.s[sqlite3Prng.j];
  79.   sqlite3Prng.s[sqlite3Prng.j] = t;
  80.   t += sqlite3Prng.s[sqlite3Prng.i];
  81.   return sqlite3Prng.s[t];
  82. }
  83. /*
  84. ** Return N random bytes.
  85. */
  86. void sqlite3_randomness(int N, void *pBuf){
  87.   unsigned char *zBuf = pBuf;
  88.   static sqlite3_mutex *mutex = 0;
  89.   if( mutex==0 ){
  90.     mutex = sqlite3_mutex_alloc(SQLITE_MUTEX_STATIC_PRNG);
  91.   }
  92.   sqlite3_mutex_enter(mutex);
  93.   while( N-- ){
  94.     *(zBuf++) = randomByte();
  95.   }
  96.   sqlite3_mutex_leave(mutex);
  97. }
  98. #ifndef SQLITE_OMIT_BUILTIN_TEST
  99. /*
  100. ** For testing purposes, we sometimes want to preserve the state of
  101. ** PRNG and restore the PRNG to its saved state at a later time.
  102. ** The sqlite3_test_control() interface calls these routines to
  103. ** control the PRNG.
  104. */
  105. static struct sqlite3PrngType sqlite3SavedPrng;
  106. void sqlite3PrngSaveState(void){
  107.   memcpy(&sqlite3SavedPrng, &sqlite3Prng, sizeof(sqlite3Prng));
  108. }
  109. void sqlite3PrngRestoreState(void){
  110.   memcpy(&sqlite3Prng, &sqlite3SavedPrng, sizeof(sqlite3Prng));
  111. }
  112. void sqlite3PrngResetState(void){
  113.   sqlite3Prng.isInit = 0;
  114. }
  115. #endif /* SQLITE_OMIT_BUILTIN_TEST */