using System; using i64 = System.Int64; using u32 = System.UInt32; using u8 = System.Byte; namespace Community.CsharpSqlite { public partial class Sqlite3 { /* ** 2001 September 15 ** ** The author disclaims copyright to this source code. In place of ** a legal notice, here is a blessing: ** ** May you do good and not evil. ** May you find forgiveness for yourself and forgive others. ** May you share freely, never taking more than you give. ** ************************************************************************* ** This file contains code to implement a pseudo-random number ** generator (PRNG) for SQLite. ** ** Random numbers are used by some of the database backends in order ** to generate random integer keys for tables or random filenames. ************************************************************************* ** Included in SQLite3 port to C#-SQLite; 2008 Noah B Hart ** C#-SQLite is an independent reimplementation of the SQLite software library ** ** SQLITE_SOURCE_ID: 2010-08-23 18:52:01 42537b60566f288167f1b5864a5435986838e3a3 ** ************************************************************************* */ //#include "sqliteInt.h" /* All threads share a single random number generator. ** This structure is the current state of the generator. */ public class sqlite3PrngType { public bool isInit; /* True if initialized */ public int i; public int j; /* State variables */ public u8[] s = new u8[256]; /* State variables */ public sqlite3PrngType Copy() { sqlite3PrngType cp = (sqlite3PrngType)MemberwiseClone(); cp.s = new u8[s.Length]; Array.Copy(s, cp.s, s.Length); return cp; } } public static sqlite3PrngType sqlite3Prng = new sqlite3PrngType(); /* ** Get a single 8-bit random value from the RC4 PRNG. The Mutex ** must be held while executing this routine. ** ** Why not just use a library random generator like lrand48() for this? ** Because the OP_NewRowid opcode in the VDBE depends on having a very ** good source of random numbers. The lrand48() library function may ** well be good enough. But maybe not. Or maybe lrand48() has some ** subtle problems on some systems that could cause problems. It is hard ** to know. To minimize the risk of problems due to bad lrand48() ** implementations, SQLite uses this random number generator based ** on RC4, which we know works very well. ** ** (Later): Actually, OP_NewRowid does not depend on a good source of ** randomness any more. But we will leave this code in all the same. */ private static u8 randomu8() { u8 t; /* The "wsdPrng" macro will resolve to the pseudo-random number generator ** state vector. If writable static data is unsupported on the target, ** we have to locate the state vector at run-time. In the more common ** case where writable static data is supported, wsdPrng can refer directly ** to the "sqlite3Prng" state vector declared above. */ #if SQLITE_OMIT_WSD struct sqlite3PrngType *p = &GLOBAL(struct sqlite3PrngType, sqlite3Prng); //# define wsdPrng p[0] #else //# define wsdPrng sqlite3Prng sqlite3PrngType wsdPrng = sqlite3Prng; #endif /* Initialize the state of the random number generator once, ** the first time this routine is called. The seed value does ** not need to contain a lot of randomness since we are not ** trying to do secure encryption or anything like that... ** ** Nothing in this file or anywhere else in SQLite does any kind of ** encryption. The RC4 algorithm is being used as a PRNG (pseudo-random ** number generator) not as an encryption device. */ if (!wsdPrng.isInit) { int i; u8[] k = new u8[256]; wsdPrng.j = 0; wsdPrng.i = 0; sqlite3OsRandomness(sqlite3_vfs_find(""), 256, k); for (i = 0; i < 255; i++) { wsdPrng.s[i] = (u8)i; } for (i = 0; i < 255; i++) { wsdPrng.j = (u8)(wsdPrng.j + wsdPrng.s[i] + k[i]); t = wsdPrng.s[wsdPrng.j]; wsdPrng.s[wsdPrng.j] = wsdPrng.s[i]; wsdPrng.s[i] = t; } wsdPrng.isInit = true; } /* Generate and return single random u8 */ wsdPrng.i++; t = wsdPrng.s[(u8)wsdPrng.i]; wsdPrng.j = (u8)(wsdPrng.j + t); wsdPrng.s[(u8)wsdPrng.i] = wsdPrng.s[wsdPrng.j]; wsdPrng.s[wsdPrng.j] = t; t += wsdPrng.s[(u8)wsdPrng.i]; return wsdPrng.s[t]; } /* ** Return N random u8s. */ private static void sqlite3_randomness(int N, ref i64 pBuf) { //u8[] zBuf = new u8[N]; pBuf = 0; #if SQLITE_THREADSAFE sqlite3_mutex mutex = sqlite3MutexAlloc(SQLITE_MUTEX_STATIC_PRNG); #endif sqlite3_mutex_enter(mutex); while (N-- > 0) { pBuf = (u32)((pBuf << 8) + randomu8());// zBuf[N] = randomu8(); } sqlite3_mutex_leave(mutex); } private static void sqlite3_randomness(byte[] pBuf, int Offset, int N) { i64 iBuf = System.DateTime.Now.Ticks; #if SQLITE_THREADSAFE sqlite3_mutex mutex = sqlite3MutexAlloc(SQLITE_MUTEX_STATIC_PRNG); #endif sqlite3_mutex_enter(mutex); while (N-- > 0) { iBuf = (u32)((iBuf << 8) + randomu8());// zBuf[N] = randomu8(); pBuf[Offset++] = (byte)iBuf; } sqlite3_mutex_leave(mutex); } #if !SQLITE_OMIT_BUILTIN_TEST /* ** For testing purposes, we sometimes want to preserve the state of ** PRNG and restore the PRNG to its saved state at a later time, or ** to reset the PRNG to its initial state. These routines accomplish ** those tasks. ** ** The sqlite3_test_control() interface calls these routines to ** control the PRNG. */ private static sqlite3PrngType sqlite3SavedPrng = null; private static void sqlite3PrngSaveState() { sqlite3SavedPrng = sqlite3Prng.Copy(); // memcpy( // &GLOBAL(struct sqlite3PrngType, sqlite3SavedPrng), // &GLOBAL(struct sqlite3PrngType, sqlite3Prng), // sizeof(sqlite3Prng) //); } private static void sqlite3PrngRestoreState() { sqlite3Prng = sqlite3SavedPrng.Copy(); //memcpy( // &GLOBAL(struct sqlite3PrngType, sqlite3Prng), // &GLOBAL(struct sqlite3PrngType, sqlite3SavedPrng), // sizeof(sqlite3Prng) //); } private static void sqlite3PrngResetState() { sqlite3Prng.isInit = false;// GLOBAL(struct sqlite3PrngType, sqlite3Prng).isInit = 0; } #endif //* SQLITE_OMIT_BUILTIN_TEST */ } }