xref: /sqlite-3.40.0/test/threadtest3.c (revision 4be02b90)
1*4be02b90Sdan 
2*4be02b90Sdan /*
3*4be02b90Sdan ** The code in this file runs a few multi-threaded test cases using the
4*4be02b90Sdan ** SQLite library. It can be compiled to an executable on unix using the
5*4be02b90Sdan ** following command:
6*4be02b90Sdan **
7*4be02b90Sdan **   gcc -O2 threadtest3.c sqlite3.c -ldl -lpthread -lm
8*4be02b90Sdan **
9*4be02b90Sdan ** Then run the compiled program. The exit status is non-zero if any tests
10*4be02b90Sdan ** failed (hopefully there is also some output to stdout to clarify what went
11*4be02b90Sdan ** wrong).
12*4be02b90Sdan **
13*4be02b90Sdan ** There are three parts to the code in this file, in the following order:
14*4be02b90Sdan **
15*4be02b90Sdan **   1. Code for the SQL aggregate function md5sum() copied from
16*4be02b90Sdan **      tclsqlite.c in the SQLite distribution. The names of all the
17*4be02b90Sdan **      types and functions in this section begin with "MD5" or "md5".
18*4be02b90Sdan **
19*4be02b90Sdan **   2. A set of utility functions that may be used to implement
20*4be02b90Sdan **      multi-threaded test cases. These are all called by test code
21*4be02b90Sdan **      via macros that help with error reporting. The macros are defined
22*4be02b90Sdan **      immediately below this comment.
23*4be02b90Sdan **
24*4be02b90Sdan **   3. The test code itself. And a main() routine to drive the test
25*4be02b90Sdan **      code.
26*4be02b90Sdan */
27*4be02b90Sdan 
28*4be02b90Sdan /*************************************************************************
29*4be02b90Sdan ** Start of test code/infrastructure interface macros.
30*4be02b90Sdan **
31*4be02b90Sdan ** The following macros constitute the interface between the test
32*4be02b90Sdan ** programs and the test infrastructure. Test infrastructure code
33*4be02b90Sdan ** does not itself use any of these macros. Test code should not
34*4be02b90Sdan ** call any of the macroname_x() functions directly.
35*4be02b90Sdan **
36*4be02b90Sdan ** See the header comments above the corresponding macroname_x()
37*4be02b90Sdan ** function for a description of each interface.
38*4be02b90Sdan */
39*4be02b90Sdan 
40*4be02b90Sdan /* Database functions */
41*4be02b90Sdan #define opendb(w,x,y,z)         (SEL(w), opendb_x(w,x,y,z))
42*4be02b90Sdan #define closedb(y,z)            (SEL(y), closedb_x(y,z))
43*4be02b90Sdan 
44*4be02b90Sdan /* Functions to execute SQL */
45*4be02b90Sdan #define sql_script(x,y,z)       (SEL(x), sql_script_x(x,y,z))
46*4be02b90Sdan #define integrity_check(x,y)    (SEL(x), integrity_check_x(x,y))
47*4be02b90Sdan #define execsql_i64(x,y,...)    (SEL(x), execsql_i64_x(x,y,__VA_ARGS__))
48*4be02b90Sdan #define execsql_text(x,y,z,...) (SEL(x), execsql_text_x(x,y,z,__VA_ARGS__))
49*4be02b90Sdan #define execsql(x,y,...)        (SEL(x), (void)execsql_i64_x(x,y,__VA_ARGS__))
50*4be02b90Sdan 
51*4be02b90Sdan /* Thread functions */
52*4be02b90Sdan #define launch_thread(w,x,y,z)  (SEL(w), launch_thread_x(w,x,y,z))
53*4be02b90Sdan #define join_all_threads(y,z)   (SEL(y), join_all_threads_x(y,z))
54*4be02b90Sdan 
55*4be02b90Sdan /* Timer functions */
56*4be02b90Sdan #define setstoptime(y,z)        (SEL(y), setstoptime_x(y,z))
57*4be02b90Sdan #define timetostop(z)           (SEL(z), timetostop_x(z))
58*4be02b90Sdan 
59*4be02b90Sdan /* Report/clear errors. */
60*4be02b90Sdan #define test_error(z, ...)      test_error_x(z, sqlite3_mprintf(__VA_ARGS__))
61*4be02b90Sdan #define clear_error(y,z)        clear_error_x(y, z)
62*4be02b90Sdan 
63*4be02b90Sdan /* File-system operations */
64*4be02b90Sdan #define filesize(y,z)           (SEL(y), filesize_x(y,z))
65*4be02b90Sdan #define filecopy(x,y,z)         (SEL(x), filecopy_x(x,y,z))
66*4be02b90Sdan 
67*4be02b90Sdan /*
68*4be02b90Sdan ** End of test code/infrastructure interface macros.
69*4be02b90Sdan *************************************************************************/
70*4be02b90Sdan 
71*4be02b90Sdan 
72*4be02b90Sdan 
73*4be02b90Sdan 
74*4be02b90Sdan #include <sqlite3.h>
75*4be02b90Sdan #include <unistd.h>
76*4be02b90Sdan #include <stdio.h>
77*4be02b90Sdan #include <pthread.h>
78*4be02b90Sdan #include <assert.h>
79*4be02b90Sdan #include <sys/types.h>
80*4be02b90Sdan #include <sys/stat.h>
81*4be02b90Sdan #include <string.h>
82*4be02b90Sdan #include <fcntl.h>
83*4be02b90Sdan #include <errno.h>
84*4be02b90Sdan 
85*4be02b90Sdan /*
86*4be02b90Sdan  * This code implements the MD5 message-digest algorithm.
87*4be02b90Sdan  * The algorithm is due to Ron Rivest.  This code was
88*4be02b90Sdan  * written by Colin Plumb in 1993, no copyright is claimed.
89*4be02b90Sdan  * This code is in the public domain; do with it what you wish.
90*4be02b90Sdan  *
91*4be02b90Sdan  * Equivalent code is available from RSA Data Security, Inc.
92*4be02b90Sdan  * This code has been tested against that, and is equivalent,
93*4be02b90Sdan  * except that you don't need to include two pages of legalese
94*4be02b90Sdan  * with every copy.
95*4be02b90Sdan  *
96*4be02b90Sdan  * To compute the message digest of a chunk of bytes, declare an
97*4be02b90Sdan  * MD5Context structure, pass it to MD5Init, call MD5Update as
98*4be02b90Sdan  * needed on buffers full of bytes, and then call MD5Final, which
99*4be02b90Sdan  * will fill a supplied 16-byte array with the digest.
100*4be02b90Sdan  */
101*4be02b90Sdan 
102*4be02b90Sdan /*
103*4be02b90Sdan  * If compiled on a machine that doesn't have a 32-bit integer,
104*4be02b90Sdan  * you just set "uint32" to the appropriate datatype for an
105*4be02b90Sdan  * unsigned 32-bit integer.  For example:
106*4be02b90Sdan  *
107*4be02b90Sdan  *       cc -Duint32='unsigned long' md5.c
108*4be02b90Sdan  *
109*4be02b90Sdan  */
110*4be02b90Sdan #ifndef uint32
111*4be02b90Sdan #  define uint32 unsigned int
112*4be02b90Sdan #endif
113*4be02b90Sdan 
114*4be02b90Sdan struct MD5Context {
115*4be02b90Sdan   int isInit;
116*4be02b90Sdan   uint32 buf[4];
117*4be02b90Sdan   uint32 bits[2];
118*4be02b90Sdan   unsigned char in[64];
119*4be02b90Sdan };
120*4be02b90Sdan typedef struct MD5Context MD5Context;
121*4be02b90Sdan 
122*4be02b90Sdan /*
123*4be02b90Sdan  * Note: this code is harmless on little-endian machines.
124*4be02b90Sdan  */
125*4be02b90Sdan static void byteReverse (unsigned char *buf, unsigned longs){
126*4be02b90Sdan   uint32 t;
127*4be02b90Sdan   do {
128*4be02b90Sdan     t = (uint32)((unsigned)buf[3]<<8 | buf[2]) << 16 |
129*4be02b90Sdan           ((unsigned)buf[1]<<8 | buf[0]);
130*4be02b90Sdan     *(uint32 *)buf = t;
131*4be02b90Sdan     buf += 4;
132*4be02b90Sdan   } while (--longs);
133*4be02b90Sdan }
134*4be02b90Sdan /* The four core functions - F1 is optimized somewhat */
135*4be02b90Sdan 
136*4be02b90Sdan /* #define F1(x, y, z) (x & y | ~x & z) */
137*4be02b90Sdan #define F1(x, y, z) (z ^ (x & (y ^ z)))
138*4be02b90Sdan #define F2(x, y, z) F1(z, x, y)
139*4be02b90Sdan #define F3(x, y, z) (x ^ y ^ z)
140*4be02b90Sdan #define F4(x, y, z) (y ^ (x | ~z))
141*4be02b90Sdan 
142*4be02b90Sdan /* This is the central step in the MD5 algorithm. */
143*4be02b90Sdan #define MD5STEP(f, w, x, y, z, data, s) \
144*4be02b90Sdan   ( w += f(x, y, z) + data,  w = w<<s | w>>(32-s),  w += x )
145*4be02b90Sdan 
146*4be02b90Sdan /*
147*4be02b90Sdan  * The core of the MD5 algorithm, this alters an existing MD5 hash to
148*4be02b90Sdan  * reflect the addition of 16 longwords of new data.  MD5Update blocks
149*4be02b90Sdan  * the data and converts bytes into longwords for this routine.
150*4be02b90Sdan  */
151*4be02b90Sdan static void MD5Transform(uint32 buf[4], const uint32 in[16]){
152*4be02b90Sdan   register uint32 a, b, c, d;
153*4be02b90Sdan 
154*4be02b90Sdan   a = buf[0];
155*4be02b90Sdan   b = buf[1];
156*4be02b90Sdan   c = buf[2];
157*4be02b90Sdan   d = buf[3];
158*4be02b90Sdan 
159*4be02b90Sdan   MD5STEP(F1, a, b, c, d, in[ 0]+0xd76aa478,  7);
160*4be02b90Sdan   MD5STEP(F1, d, a, b, c, in[ 1]+0xe8c7b756, 12);
161*4be02b90Sdan   MD5STEP(F1, c, d, a, b, in[ 2]+0x242070db, 17);
162*4be02b90Sdan   MD5STEP(F1, b, c, d, a, in[ 3]+0xc1bdceee, 22);
163*4be02b90Sdan   MD5STEP(F1, a, b, c, d, in[ 4]+0xf57c0faf,  7);
164*4be02b90Sdan   MD5STEP(F1, d, a, b, c, in[ 5]+0x4787c62a, 12);
165*4be02b90Sdan   MD5STEP(F1, c, d, a, b, in[ 6]+0xa8304613, 17);
166*4be02b90Sdan   MD5STEP(F1, b, c, d, a, in[ 7]+0xfd469501, 22);
167*4be02b90Sdan   MD5STEP(F1, a, b, c, d, in[ 8]+0x698098d8,  7);
168*4be02b90Sdan   MD5STEP(F1, d, a, b, c, in[ 9]+0x8b44f7af, 12);
169*4be02b90Sdan   MD5STEP(F1, c, d, a, b, in[10]+0xffff5bb1, 17);
170*4be02b90Sdan   MD5STEP(F1, b, c, d, a, in[11]+0x895cd7be, 22);
171*4be02b90Sdan   MD5STEP(F1, a, b, c, d, in[12]+0x6b901122,  7);
172*4be02b90Sdan   MD5STEP(F1, d, a, b, c, in[13]+0xfd987193, 12);
173*4be02b90Sdan   MD5STEP(F1, c, d, a, b, in[14]+0xa679438e, 17);
174*4be02b90Sdan   MD5STEP(F1, b, c, d, a, in[15]+0x49b40821, 22);
175*4be02b90Sdan 
176*4be02b90Sdan   MD5STEP(F2, a, b, c, d, in[ 1]+0xf61e2562,  5);
177*4be02b90Sdan   MD5STEP(F2, d, a, b, c, in[ 6]+0xc040b340,  9);
178*4be02b90Sdan   MD5STEP(F2, c, d, a, b, in[11]+0x265e5a51, 14);
179*4be02b90Sdan   MD5STEP(F2, b, c, d, a, in[ 0]+0xe9b6c7aa, 20);
180*4be02b90Sdan   MD5STEP(F2, a, b, c, d, in[ 5]+0xd62f105d,  5);
181*4be02b90Sdan   MD5STEP(F2, d, a, b, c, in[10]+0x02441453,  9);
182*4be02b90Sdan   MD5STEP(F2, c, d, a, b, in[15]+0xd8a1e681, 14);
183*4be02b90Sdan   MD5STEP(F2, b, c, d, a, in[ 4]+0xe7d3fbc8, 20);
184*4be02b90Sdan   MD5STEP(F2, a, b, c, d, in[ 9]+0x21e1cde6,  5);
185*4be02b90Sdan   MD5STEP(F2, d, a, b, c, in[14]+0xc33707d6,  9);
186*4be02b90Sdan   MD5STEP(F2, c, d, a, b, in[ 3]+0xf4d50d87, 14);
187*4be02b90Sdan   MD5STEP(F2, b, c, d, a, in[ 8]+0x455a14ed, 20);
188*4be02b90Sdan   MD5STEP(F2, a, b, c, d, in[13]+0xa9e3e905,  5);
189*4be02b90Sdan   MD5STEP(F2, d, a, b, c, in[ 2]+0xfcefa3f8,  9);
190*4be02b90Sdan   MD5STEP(F2, c, d, a, b, in[ 7]+0x676f02d9, 14);
191*4be02b90Sdan   MD5STEP(F2, b, c, d, a, in[12]+0x8d2a4c8a, 20);
192*4be02b90Sdan 
193*4be02b90Sdan   MD5STEP(F3, a, b, c, d, in[ 5]+0xfffa3942,  4);
194*4be02b90Sdan   MD5STEP(F3, d, a, b, c, in[ 8]+0x8771f681, 11);
195*4be02b90Sdan   MD5STEP(F3, c, d, a, b, in[11]+0x6d9d6122, 16);
196*4be02b90Sdan   MD5STEP(F3, b, c, d, a, in[14]+0xfde5380c, 23);
197*4be02b90Sdan   MD5STEP(F3, a, b, c, d, in[ 1]+0xa4beea44,  4);
198*4be02b90Sdan   MD5STEP(F3, d, a, b, c, in[ 4]+0x4bdecfa9, 11);
199*4be02b90Sdan   MD5STEP(F3, c, d, a, b, in[ 7]+0xf6bb4b60, 16);
200*4be02b90Sdan   MD5STEP(F3, b, c, d, a, in[10]+0xbebfbc70, 23);
201*4be02b90Sdan   MD5STEP(F3, a, b, c, d, in[13]+0x289b7ec6,  4);
202*4be02b90Sdan   MD5STEP(F3, d, a, b, c, in[ 0]+0xeaa127fa, 11);
203*4be02b90Sdan   MD5STEP(F3, c, d, a, b, in[ 3]+0xd4ef3085, 16);
204*4be02b90Sdan   MD5STEP(F3, b, c, d, a, in[ 6]+0x04881d05, 23);
205*4be02b90Sdan   MD5STEP(F3, a, b, c, d, in[ 9]+0xd9d4d039,  4);
206*4be02b90Sdan   MD5STEP(F3, d, a, b, c, in[12]+0xe6db99e5, 11);
207*4be02b90Sdan   MD5STEP(F3, c, d, a, b, in[15]+0x1fa27cf8, 16);
208*4be02b90Sdan   MD5STEP(F3, b, c, d, a, in[ 2]+0xc4ac5665, 23);
209*4be02b90Sdan 
210*4be02b90Sdan   MD5STEP(F4, a, b, c, d, in[ 0]+0xf4292244,  6);
211*4be02b90Sdan   MD5STEP(F4, d, a, b, c, in[ 7]+0x432aff97, 10);
212*4be02b90Sdan   MD5STEP(F4, c, d, a, b, in[14]+0xab9423a7, 15);
213*4be02b90Sdan   MD5STEP(F4, b, c, d, a, in[ 5]+0xfc93a039, 21);
214*4be02b90Sdan   MD5STEP(F4, a, b, c, d, in[12]+0x655b59c3,  6);
215*4be02b90Sdan   MD5STEP(F4, d, a, b, c, in[ 3]+0x8f0ccc92, 10);
216*4be02b90Sdan   MD5STEP(F4, c, d, a, b, in[10]+0xffeff47d, 15);
217*4be02b90Sdan   MD5STEP(F4, b, c, d, a, in[ 1]+0x85845dd1, 21);
218*4be02b90Sdan   MD5STEP(F4, a, b, c, d, in[ 8]+0x6fa87e4f,  6);
219*4be02b90Sdan   MD5STEP(F4, d, a, b, c, in[15]+0xfe2ce6e0, 10);
220*4be02b90Sdan   MD5STEP(F4, c, d, a, b, in[ 6]+0xa3014314, 15);
221*4be02b90Sdan   MD5STEP(F4, b, c, d, a, in[13]+0x4e0811a1, 21);
222*4be02b90Sdan   MD5STEP(F4, a, b, c, d, in[ 4]+0xf7537e82,  6);
223*4be02b90Sdan   MD5STEP(F4, d, a, b, c, in[11]+0xbd3af235, 10);
224*4be02b90Sdan   MD5STEP(F4, c, d, a, b, in[ 2]+0x2ad7d2bb, 15);
225*4be02b90Sdan   MD5STEP(F4, b, c, d, a, in[ 9]+0xeb86d391, 21);
226*4be02b90Sdan 
227*4be02b90Sdan   buf[0] += a;
228*4be02b90Sdan   buf[1] += b;
229*4be02b90Sdan   buf[2] += c;
230*4be02b90Sdan   buf[3] += d;
231*4be02b90Sdan }
232*4be02b90Sdan 
233*4be02b90Sdan /*
234*4be02b90Sdan  * Start MD5 accumulation.  Set bit count to 0 and buffer to mysterious
235*4be02b90Sdan  * initialization constants.
236*4be02b90Sdan  */
237*4be02b90Sdan static void MD5Init(MD5Context *ctx){
238*4be02b90Sdan   ctx->isInit = 1;
239*4be02b90Sdan   ctx->buf[0] = 0x67452301;
240*4be02b90Sdan   ctx->buf[1] = 0xefcdab89;
241*4be02b90Sdan   ctx->buf[2] = 0x98badcfe;
242*4be02b90Sdan   ctx->buf[3] = 0x10325476;
243*4be02b90Sdan   ctx->bits[0] = 0;
244*4be02b90Sdan   ctx->bits[1] = 0;
245*4be02b90Sdan }
246*4be02b90Sdan 
247*4be02b90Sdan /*
248*4be02b90Sdan  * Update context to reflect the concatenation of another buffer full
249*4be02b90Sdan  * of bytes.
250*4be02b90Sdan  */
251*4be02b90Sdan static
252*4be02b90Sdan void MD5Update(MD5Context *ctx, const unsigned char *buf, unsigned int len){
253*4be02b90Sdan   uint32 t;
254*4be02b90Sdan 
255*4be02b90Sdan   /* Update bitcount */
256*4be02b90Sdan 
257*4be02b90Sdan   t = ctx->bits[0];
258*4be02b90Sdan   if ((ctx->bits[0] = t + ((uint32)len << 3)) < t)
259*4be02b90Sdan     ctx->bits[1]++; /* Carry from low to high */
260*4be02b90Sdan   ctx->bits[1] += len >> 29;
261*4be02b90Sdan 
262*4be02b90Sdan   t = (t >> 3) & 0x3f;    /* Bytes already in shsInfo->data */
263*4be02b90Sdan 
264*4be02b90Sdan   /* Handle any leading odd-sized chunks */
265*4be02b90Sdan 
266*4be02b90Sdan   if ( t ) {
267*4be02b90Sdan     unsigned char *p = (unsigned char *)ctx->in + t;
268*4be02b90Sdan 
269*4be02b90Sdan     t = 64-t;
270*4be02b90Sdan     if (len < t) {
271*4be02b90Sdan       memcpy(p, buf, len);
272*4be02b90Sdan       return;
273*4be02b90Sdan     }
274*4be02b90Sdan     memcpy(p, buf, t);
275*4be02b90Sdan     byteReverse(ctx->in, 16);
276*4be02b90Sdan     MD5Transform(ctx->buf, (uint32 *)ctx->in);
277*4be02b90Sdan     buf += t;
278*4be02b90Sdan     len -= t;
279*4be02b90Sdan   }
280*4be02b90Sdan 
281*4be02b90Sdan   /* Process data in 64-byte chunks */
282*4be02b90Sdan 
283*4be02b90Sdan   while (len >= 64) {
284*4be02b90Sdan     memcpy(ctx->in, buf, 64);
285*4be02b90Sdan     byteReverse(ctx->in, 16);
286*4be02b90Sdan     MD5Transform(ctx->buf, (uint32 *)ctx->in);
287*4be02b90Sdan     buf += 64;
288*4be02b90Sdan     len -= 64;
289*4be02b90Sdan   }
290*4be02b90Sdan 
291*4be02b90Sdan   /* Handle any remaining bytes of data. */
292*4be02b90Sdan 
293*4be02b90Sdan   memcpy(ctx->in, buf, len);
294*4be02b90Sdan }
295*4be02b90Sdan 
296*4be02b90Sdan /*
297*4be02b90Sdan  * Final wrapup - pad to 64-byte boundary with the bit pattern
298*4be02b90Sdan  * 1 0* (64-bit count of bits processed, MSB-first)
299*4be02b90Sdan  */
300*4be02b90Sdan static void MD5Final(unsigned char digest[16], MD5Context *ctx){
301*4be02b90Sdan   unsigned count;
302*4be02b90Sdan   unsigned char *p;
303*4be02b90Sdan 
304*4be02b90Sdan   /* Compute number of bytes mod 64 */
305*4be02b90Sdan   count = (ctx->bits[0] >> 3) & 0x3F;
306*4be02b90Sdan 
307*4be02b90Sdan   /* Set the first char of padding to 0x80.  This is safe since there is
308*4be02b90Sdan      always at least one byte free */
309*4be02b90Sdan   p = ctx->in + count;
310*4be02b90Sdan   *p++ = 0x80;
311*4be02b90Sdan 
312*4be02b90Sdan   /* Bytes of padding needed to make 64 bytes */
313*4be02b90Sdan   count = 64 - 1 - count;
314*4be02b90Sdan 
315*4be02b90Sdan   /* Pad out to 56 mod 64 */
316*4be02b90Sdan   if (count < 8) {
317*4be02b90Sdan     /* Two lots of padding:  Pad the first block to 64 bytes */
318*4be02b90Sdan     memset(p, 0, count);
319*4be02b90Sdan     byteReverse(ctx->in, 16);
320*4be02b90Sdan     MD5Transform(ctx->buf, (uint32 *)ctx->in);
321*4be02b90Sdan 
322*4be02b90Sdan     /* Now fill the next block with 56 bytes */
323*4be02b90Sdan     memset(ctx->in, 0, 56);
324*4be02b90Sdan   } else {
325*4be02b90Sdan     /* Pad block to 56 bytes */
326*4be02b90Sdan     memset(p, 0, count-8);
327*4be02b90Sdan   }
328*4be02b90Sdan   byteReverse(ctx->in, 14);
329*4be02b90Sdan 
330*4be02b90Sdan   /* Append length in bits and transform */
331*4be02b90Sdan   ((uint32 *)ctx->in)[ 14 ] = ctx->bits[0];
332*4be02b90Sdan   ((uint32 *)ctx->in)[ 15 ] = ctx->bits[1];
333*4be02b90Sdan 
334*4be02b90Sdan   MD5Transform(ctx->buf, (uint32 *)ctx->in);
335*4be02b90Sdan   byteReverse((unsigned char *)ctx->buf, 4);
336*4be02b90Sdan   memcpy(digest, ctx->buf, 16);
337*4be02b90Sdan   memset(ctx, 0, sizeof(ctx));    /* In case it is sensitive */
338*4be02b90Sdan }
339*4be02b90Sdan 
340*4be02b90Sdan /*
341*4be02b90Sdan ** Convert a 128-bit MD5 digest into a 32-digit base-16 number.
342*4be02b90Sdan */
343*4be02b90Sdan static void MD5DigestToBase16(unsigned char *digest, char *zBuf){
344*4be02b90Sdan   static char const zEncode[] = "0123456789abcdef";
345*4be02b90Sdan   int i, j;
346*4be02b90Sdan 
347*4be02b90Sdan   for(j=i=0; i<16; i++){
348*4be02b90Sdan     int a = digest[i];
349*4be02b90Sdan     zBuf[j++] = zEncode[(a>>4)&0xf];
350*4be02b90Sdan     zBuf[j++] = zEncode[a & 0xf];
351*4be02b90Sdan   }
352*4be02b90Sdan   zBuf[j] = 0;
353*4be02b90Sdan }
354*4be02b90Sdan 
355*4be02b90Sdan /*
356*4be02b90Sdan ** During testing, the special md5sum() aggregate function is available.
357*4be02b90Sdan ** inside SQLite.  The following routines implement that function.
358*4be02b90Sdan */
359*4be02b90Sdan static void md5step(sqlite3_context *context, int argc, sqlite3_value **argv){
360*4be02b90Sdan   MD5Context *p;
361*4be02b90Sdan   int i;
362*4be02b90Sdan   if( argc<1 ) return;
363*4be02b90Sdan   p = sqlite3_aggregate_context(context, sizeof(*p));
364*4be02b90Sdan   if( p==0 ) return;
365*4be02b90Sdan   if( !p->isInit ){
366*4be02b90Sdan     MD5Init(p);
367*4be02b90Sdan   }
368*4be02b90Sdan   for(i=0; i<argc; i++){
369*4be02b90Sdan     const char *zData = (char*)sqlite3_value_text(argv[i]);
370*4be02b90Sdan     if( zData ){
371*4be02b90Sdan       MD5Update(p, (unsigned char*)zData, strlen(zData));
372*4be02b90Sdan     }
373*4be02b90Sdan   }
374*4be02b90Sdan }
375*4be02b90Sdan static void md5finalize(sqlite3_context *context){
376*4be02b90Sdan   MD5Context *p;
377*4be02b90Sdan   unsigned char digest[16];
378*4be02b90Sdan   char zBuf[33];
379*4be02b90Sdan   p = sqlite3_aggregate_context(context, sizeof(*p));
380*4be02b90Sdan   MD5Final(digest,p);
381*4be02b90Sdan   MD5DigestToBase16(digest, zBuf);
382*4be02b90Sdan   sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT);
383*4be02b90Sdan }
384*4be02b90Sdan 
385*4be02b90Sdan /*************************************************************************
386*4be02b90Sdan ** End of copied md5sum() code.
387*4be02b90Sdan */
388*4be02b90Sdan 
389*4be02b90Sdan typedef sqlite3_int64 i64;
390*4be02b90Sdan 
391*4be02b90Sdan typedef struct Error Error;
392*4be02b90Sdan typedef struct Sqlite Sqlite;
393*4be02b90Sdan typedef struct Statement Statement;
394*4be02b90Sdan 
395*4be02b90Sdan typedef struct Threadset Threadset;
396*4be02b90Sdan typedef struct Thread Thread;
397*4be02b90Sdan 
398*4be02b90Sdan /* Total number of errors in this process so far. */
399*4be02b90Sdan static int nGlobalErr = 0;
400*4be02b90Sdan 
401*4be02b90Sdan /* Set to true to run in "process" instead of "thread" mode. */
402*4be02b90Sdan static int bProcessMode = 0;
403*4be02b90Sdan 
404*4be02b90Sdan struct Error {
405*4be02b90Sdan   int rc;
406*4be02b90Sdan   int iLine;
407*4be02b90Sdan   char *zErr;
408*4be02b90Sdan };
409*4be02b90Sdan 
410*4be02b90Sdan struct Sqlite {
411*4be02b90Sdan   sqlite3 *db;                    /* Database handle */
412*4be02b90Sdan   Statement *pCache;              /* Linked list of cached statements */
413*4be02b90Sdan   int nText;                      /* Size of array at aText[] */
414*4be02b90Sdan   char **aText;                   /* Stored text results */
415*4be02b90Sdan };
416*4be02b90Sdan 
417*4be02b90Sdan struct Statement {
418*4be02b90Sdan   sqlite3_stmt *pStmt;            /* Pre-compiled statement handle */
419*4be02b90Sdan   Statement *pNext;               /* Next statement in linked-list */
420*4be02b90Sdan };
421*4be02b90Sdan 
422*4be02b90Sdan struct Thread {
423*4be02b90Sdan   int iTid;                       /* Thread number within test */
424*4be02b90Sdan   int iArg;                       /* Integer argument passed by caller */
425*4be02b90Sdan 
426*4be02b90Sdan   pthread_t tid;                  /* Thread id */
427*4be02b90Sdan   char *(*xProc)(int, int);       /* Thread main proc */
428*4be02b90Sdan   Thread *pNext;                  /* Next in this list of threads */
429*4be02b90Sdan };
430*4be02b90Sdan 
431*4be02b90Sdan struct Threadset {
432*4be02b90Sdan   int iMaxTid;                    /* Largest iTid value allocated so far */
433*4be02b90Sdan   Thread *pThread;                /* Linked list of threads */
434*4be02b90Sdan };
435*4be02b90Sdan 
436*4be02b90Sdan static void free_err(Error *p){
437*4be02b90Sdan   sqlite3_free(p->zErr);
438*4be02b90Sdan   p->zErr = 0;
439*4be02b90Sdan   p->rc = 0;
440*4be02b90Sdan }
441*4be02b90Sdan 
442*4be02b90Sdan static void print_err(Error *p){
443*4be02b90Sdan   if( p->rc!=SQLITE_OK ){
444*4be02b90Sdan     printf("Error: (%d) \"%s\" at line %d\n", p->rc, p->zErr, p->iLine);
445*4be02b90Sdan     nGlobalErr++;
446*4be02b90Sdan   }
447*4be02b90Sdan }
448*4be02b90Sdan 
449*4be02b90Sdan static void print_and_free_err(Error *p){
450*4be02b90Sdan   print_err(p);
451*4be02b90Sdan   free_err(p);
452*4be02b90Sdan }
453*4be02b90Sdan 
454*4be02b90Sdan static void system_error(Error *pErr, int iSys){
455*4be02b90Sdan   pErr->rc = iSys;
456*4be02b90Sdan   pErr->zErr = (char *)sqlite3_malloc(512);
457*4be02b90Sdan   strerror_r(iSys, pErr->zErr, 512);
458*4be02b90Sdan   pErr->zErr[511] = '\0';
459*4be02b90Sdan }
460*4be02b90Sdan 
461*4be02b90Sdan static void sqlite_error(
462*4be02b90Sdan   Error *pErr,
463*4be02b90Sdan   Sqlite *pDb,
464*4be02b90Sdan   const char *zFunc
465*4be02b90Sdan ){
466*4be02b90Sdan   pErr->rc = sqlite3_errcode(pDb->db);
467*4be02b90Sdan   pErr->zErr = sqlite3_mprintf(
468*4be02b90Sdan       "sqlite3_%s() - %s (%d)", zFunc, sqlite3_errmsg(pDb->db),
469*4be02b90Sdan       sqlite3_extended_errcode(pDb->db)
470*4be02b90Sdan   );
471*4be02b90Sdan }
472*4be02b90Sdan 
473*4be02b90Sdan static void test_error_x(
474*4be02b90Sdan   Error *pErr,
475*4be02b90Sdan   char *zErr
476*4be02b90Sdan ){
477*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
478*4be02b90Sdan     pErr->rc = 1;
479*4be02b90Sdan     pErr->zErr = zErr;
480*4be02b90Sdan   }else{
481*4be02b90Sdan     sqlite3_free(zErr);
482*4be02b90Sdan   }
483*4be02b90Sdan }
484*4be02b90Sdan 
485*4be02b90Sdan static void clear_error_x(
486*4be02b90Sdan   Error *pErr,
487*4be02b90Sdan   int rc
488*4be02b90Sdan ){
489*4be02b90Sdan   if( pErr->rc==rc ){
490*4be02b90Sdan     pErr->rc = SQLITE_OK;
491*4be02b90Sdan     sqlite3_free(pErr->zErr);
492*4be02b90Sdan     pErr->zErr = 0;
493*4be02b90Sdan   }
494*4be02b90Sdan }
495*4be02b90Sdan 
496*4be02b90Sdan static int busyhandler(void *pArg, int n){
497*4be02b90Sdan   usleep(10*1000);
498*4be02b90Sdan   return 1;
499*4be02b90Sdan }
500*4be02b90Sdan 
501*4be02b90Sdan static void opendb_x(
502*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
503*4be02b90Sdan   Sqlite *pDb,                    /* OUT: Database handle */
504*4be02b90Sdan   const char *zFile,              /* Database file name */
505*4be02b90Sdan   int bDelete                     /* True to delete db file before opening */
506*4be02b90Sdan ){
507*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
508*4be02b90Sdan     int rc;
509*4be02b90Sdan     if( bDelete ) unlink(zFile);
510*4be02b90Sdan     rc = sqlite3_open(zFile, &pDb->db);
511*4be02b90Sdan     if( rc ){
512*4be02b90Sdan       sqlite_error(pErr, pDb, "open");
513*4be02b90Sdan       sqlite3_close(pDb->db);
514*4be02b90Sdan       pDb->db = 0;
515*4be02b90Sdan     }else{
516*4be02b90Sdan       sqlite3_create_function(
517*4be02b90Sdan           pDb->db, "md5sum", -1, SQLITE_UTF8, 0, 0, md5step, md5finalize
518*4be02b90Sdan       );
519*4be02b90Sdan       sqlite3_busy_handler(pDb->db, busyhandler, 0);
520*4be02b90Sdan     }
521*4be02b90Sdan   }
522*4be02b90Sdan }
523*4be02b90Sdan 
524*4be02b90Sdan static void closedb_x(
525*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
526*4be02b90Sdan   Sqlite *pDb                     /* OUT: Database handle */
527*4be02b90Sdan ){
528*4be02b90Sdan   int rc;
529*4be02b90Sdan   int i;
530*4be02b90Sdan   Statement *pIter;
531*4be02b90Sdan   Statement *pNext;
532*4be02b90Sdan   for(pIter=pDb->pCache; pIter; pIter=pNext){
533*4be02b90Sdan     pNext = pIter->pNext;
534*4be02b90Sdan     sqlite3_finalize(pIter->pStmt);
535*4be02b90Sdan     sqlite3_free(pIter);
536*4be02b90Sdan   }
537*4be02b90Sdan   for(i=0; i<pDb->nText; i++){
538*4be02b90Sdan     sqlite3_free(pDb->aText[i]);
539*4be02b90Sdan   }
540*4be02b90Sdan   sqlite3_free(pDb->aText);
541*4be02b90Sdan   rc = sqlite3_close(pDb->db);
542*4be02b90Sdan   if( rc && pErr->rc==SQLITE_OK ){
543*4be02b90Sdan     pErr->zErr = sqlite3_mprintf("%s", sqlite3_errmsg(pDb->db));
544*4be02b90Sdan   }
545*4be02b90Sdan   memset(pDb, 0, sizeof(Sqlite));
546*4be02b90Sdan }
547*4be02b90Sdan 
548*4be02b90Sdan static void sql_script_x(
549*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
550*4be02b90Sdan   Sqlite *pDb,                    /* Database handle */
551*4be02b90Sdan   const char *zSql                /* SQL script to execute */
552*4be02b90Sdan ){
553*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
554*4be02b90Sdan     pErr->rc = sqlite3_exec(pDb->db, zSql, 0, 0, &pErr->zErr);
555*4be02b90Sdan   }
556*4be02b90Sdan }
557*4be02b90Sdan 
558*4be02b90Sdan static Statement *getSqlStatement(
559*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
560*4be02b90Sdan   Sqlite *pDb,                    /* Database handle */
561*4be02b90Sdan   const char *zSql                /* SQL statement */
562*4be02b90Sdan ){
563*4be02b90Sdan   Statement *pRet;
564*4be02b90Sdan   int rc;
565*4be02b90Sdan 
566*4be02b90Sdan   for(pRet=pDb->pCache; pRet; pRet=pRet->pNext){
567*4be02b90Sdan     if( 0==strcmp(sqlite3_sql(pRet->pStmt), zSql) ){
568*4be02b90Sdan       return pRet;
569*4be02b90Sdan     }
570*4be02b90Sdan   }
571*4be02b90Sdan 
572*4be02b90Sdan   pRet = sqlite3_malloc(sizeof(Statement));
573*4be02b90Sdan   rc = sqlite3_prepare_v2(pDb->db, zSql, -1, &pRet->pStmt, 0);
574*4be02b90Sdan   if( rc!=SQLITE_OK ){
575*4be02b90Sdan     sqlite_error(pErr, pDb, "prepare_v2");
576*4be02b90Sdan     return 0;
577*4be02b90Sdan   }
578*4be02b90Sdan   assert( 0==strcmp(sqlite3_sql(pRet->pStmt), zSql) );
579*4be02b90Sdan 
580*4be02b90Sdan   pRet->pNext = pDb->pCache;
581*4be02b90Sdan   pDb->pCache = pRet;
582*4be02b90Sdan   return pRet;
583*4be02b90Sdan }
584*4be02b90Sdan 
585*4be02b90Sdan static sqlite3_stmt *getAndBindSqlStatement(
586*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
587*4be02b90Sdan   Sqlite *pDb,                    /* Database handle */
588*4be02b90Sdan   va_list ap                      /* SQL followed by parameters */
589*4be02b90Sdan ){
590*4be02b90Sdan   Statement *pStatement;          /* The SQLite statement wrapper */
591*4be02b90Sdan   sqlite3_stmt *pStmt;            /* The SQLite statement to return */
592*4be02b90Sdan   int i;                          /* Used to iterate through parameters */
593*4be02b90Sdan 
594*4be02b90Sdan   pStatement = getSqlStatement(pErr, pDb, va_arg(ap, const char *));
595*4be02b90Sdan   if( !pStatement ) return 0;
596*4be02b90Sdan   pStmt = pStatement->pStmt;
597*4be02b90Sdan   for(i=1; i<=sqlite3_bind_parameter_count(pStmt); i++){
598*4be02b90Sdan     const char *zName = sqlite3_bind_parameter_name(pStmt, i);
599*4be02b90Sdan     void * pArg = va_arg(ap, void*);
600*4be02b90Sdan 
601*4be02b90Sdan     switch( zName[1] ){
602*4be02b90Sdan       case 'i':
603*4be02b90Sdan         sqlite3_bind_int64(pStmt, i, *(i64 *)pArg);
604*4be02b90Sdan         break;
605*4be02b90Sdan 
606*4be02b90Sdan       default:
607*4be02b90Sdan         pErr->rc = 1;
608*4be02b90Sdan         pErr->zErr = sqlite3_mprintf("Cannot discern type: \"%s\"", zName);
609*4be02b90Sdan         pStmt = 0;
610*4be02b90Sdan         break;
611*4be02b90Sdan     }
612*4be02b90Sdan   }
613*4be02b90Sdan 
614*4be02b90Sdan   return pStmt;
615*4be02b90Sdan }
616*4be02b90Sdan 
617*4be02b90Sdan static i64 execsql_i64_x(
618*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
619*4be02b90Sdan   Sqlite *pDb,                    /* Database handle */
620*4be02b90Sdan   ...                             /* SQL and pointers to parameter values */
621*4be02b90Sdan ){
622*4be02b90Sdan   i64 iRet = 0;
623*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
624*4be02b90Sdan     sqlite3_stmt *pStmt;          /* SQL statement to execute */
625*4be02b90Sdan     va_list ap;                   /* ... arguments */
626*4be02b90Sdan     int i;                        /* Used to iterate through parameters */
627*4be02b90Sdan     va_start(ap, pDb);
628*4be02b90Sdan     pStmt = getAndBindSqlStatement(pErr, pDb, ap);
629*4be02b90Sdan     if( pStmt ){
630*4be02b90Sdan       int rc;
631*4be02b90Sdan       int first = 1;
632*4be02b90Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
633*4be02b90Sdan         if( first && sqlite3_column_count(pStmt)>0 ){
634*4be02b90Sdan           iRet = sqlite3_column_int64(pStmt, 0);
635*4be02b90Sdan         }
636*4be02b90Sdan         first = 0;
637*4be02b90Sdan       }
638*4be02b90Sdan       if( SQLITE_OK!=sqlite3_reset(pStmt) ){
639*4be02b90Sdan         sqlite_error(pErr, pDb, "reset");
640*4be02b90Sdan       }
641*4be02b90Sdan     }
642*4be02b90Sdan     va_end(ap);
643*4be02b90Sdan   }
644*4be02b90Sdan   return iRet;
645*4be02b90Sdan }
646*4be02b90Sdan 
647*4be02b90Sdan static char * execsql_text_x(
648*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
649*4be02b90Sdan   Sqlite *pDb,                    /* Database handle */
650*4be02b90Sdan   int iSlot,                      /* Db handle slot to store text in */
651*4be02b90Sdan   ...                             /* SQL and pointers to parameter values */
652*4be02b90Sdan ){
653*4be02b90Sdan   char *zRet = 0;
654*4be02b90Sdan 
655*4be02b90Sdan   if( iSlot>=pDb->nText ){
656*4be02b90Sdan     int nByte = sizeof(char *)*(iSlot+1);
657*4be02b90Sdan     pDb->aText = (char **)sqlite3_realloc(pDb->aText, nByte);
658*4be02b90Sdan     memset(&pDb->aText[pDb->nText], 0, sizeof(char*)*(iSlot+1-pDb->nText));
659*4be02b90Sdan     pDb->nText = iSlot+1;
660*4be02b90Sdan   }
661*4be02b90Sdan 
662*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
663*4be02b90Sdan     sqlite3_stmt *pStmt;          /* SQL statement to execute */
664*4be02b90Sdan     va_list ap;                   /* ... arguments */
665*4be02b90Sdan     int i;                        /* Used to iterate through parameters */
666*4be02b90Sdan     va_start(ap, iSlot);
667*4be02b90Sdan     pStmt = getAndBindSqlStatement(pErr, pDb, ap);
668*4be02b90Sdan     if( pStmt ){
669*4be02b90Sdan       int rc;
670*4be02b90Sdan       int first = 1;
671*4be02b90Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
672*4be02b90Sdan         if( first && sqlite3_column_count(pStmt)>0 ){
673*4be02b90Sdan           zRet = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
674*4be02b90Sdan           sqlite3_free(pDb->aText[iSlot]);
675*4be02b90Sdan           pDb->aText[iSlot] = zRet;
676*4be02b90Sdan         }
677*4be02b90Sdan         first = 0;
678*4be02b90Sdan       }
679*4be02b90Sdan       if( SQLITE_OK!=sqlite3_reset(pStmt) ){
680*4be02b90Sdan         sqlite_error(pErr, pDb, "reset");
681*4be02b90Sdan       }
682*4be02b90Sdan     }
683*4be02b90Sdan     va_end(ap);
684*4be02b90Sdan   }
685*4be02b90Sdan 
686*4be02b90Sdan   return zRet;
687*4be02b90Sdan }
688*4be02b90Sdan 
689*4be02b90Sdan static void integrity_check_x(
690*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
691*4be02b90Sdan   Sqlite *pDb                     /* Database handle */
692*4be02b90Sdan ){
693*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
694*4be02b90Sdan     Statement *pStatement;        /* Statement to execute */
695*4be02b90Sdan     int rc;                       /* Return code */
696*4be02b90Sdan     char *zErr = 0;               /* Integrity check error */
697*4be02b90Sdan 
698*4be02b90Sdan     pStatement = getSqlStatement(pErr, pDb, "PRAGMA integrity_check");
699*4be02b90Sdan     if( pStatement ){
700*4be02b90Sdan       sqlite3_stmt *pStmt = pStatement->pStmt;
701*4be02b90Sdan       while( SQLITE_ROW==sqlite3_step(pStmt) ){
702*4be02b90Sdan         const char *z = sqlite3_column_text(pStmt, 0);
703*4be02b90Sdan         if( strcmp(z, "ok") ){
704*4be02b90Sdan           if( zErr==0 ){
705*4be02b90Sdan             zErr = sqlite3_mprintf("%s", z);
706*4be02b90Sdan           }else{
707*4be02b90Sdan             zErr = sqlite3_mprintf("%z\n%s", zErr, z);
708*4be02b90Sdan           }
709*4be02b90Sdan         }
710*4be02b90Sdan       }
711*4be02b90Sdan       sqlite3_reset(pStmt);
712*4be02b90Sdan 
713*4be02b90Sdan       if( zErr ){
714*4be02b90Sdan         pErr->zErr = zErr;
715*4be02b90Sdan         pErr->rc = 1;
716*4be02b90Sdan       }
717*4be02b90Sdan     }
718*4be02b90Sdan   }
719*4be02b90Sdan }
720*4be02b90Sdan 
721*4be02b90Sdan static void *launch_thread_main(void *pArg){
722*4be02b90Sdan   Thread *p = (Thread *)pArg;
723*4be02b90Sdan   return (void *)p->xProc(p->iTid, p->iArg);
724*4be02b90Sdan }
725*4be02b90Sdan 
726*4be02b90Sdan static void launch_thread_x(
727*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
728*4be02b90Sdan   Threadset *pThreads,            /* Thread set */
729*4be02b90Sdan   char *(*xProc)(int, int),       /* Proc to run */
730*4be02b90Sdan   int iArg                        /* Argument passed to thread proc */
731*4be02b90Sdan ){
732*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
733*4be02b90Sdan     int iTid = ++pThreads->iMaxTid;
734*4be02b90Sdan     Thread *p;
735*4be02b90Sdan     int rc;
736*4be02b90Sdan 
737*4be02b90Sdan     p = (Thread *)sqlite3_malloc(sizeof(Thread));
738*4be02b90Sdan     memset(p, 0, sizeof(Thread));
739*4be02b90Sdan     p->iTid = iTid;
740*4be02b90Sdan     p->iArg = iArg;
741*4be02b90Sdan     p->xProc = xProc;
742*4be02b90Sdan 
743*4be02b90Sdan     rc = pthread_create(&p->tid, NULL, launch_thread_main, (void *)p);
744*4be02b90Sdan     if( rc!=0 ){
745*4be02b90Sdan       system_error(pErr, rc);
746*4be02b90Sdan       sqlite3_free(p);
747*4be02b90Sdan     }else{
748*4be02b90Sdan       p->pNext = pThreads->pThread;
749*4be02b90Sdan       pThreads->pThread = p;
750*4be02b90Sdan     }
751*4be02b90Sdan   }
752*4be02b90Sdan }
753*4be02b90Sdan 
754*4be02b90Sdan static void join_all_threads_x(
755*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
756*4be02b90Sdan   Threadset *pThreads             /* Thread set */
757*4be02b90Sdan ){
758*4be02b90Sdan   Thread *p;
759*4be02b90Sdan   Thread *pNext;
760*4be02b90Sdan   for(p=pThreads->pThread; p; p=pNext){
761*4be02b90Sdan     void *ret;
762*4be02b90Sdan     pNext = p->pNext;
763*4be02b90Sdan     int rc;
764*4be02b90Sdan     rc = pthread_join(p->tid, &ret);
765*4be02b90Sdan     if( rc!=0 ){
766*4be02b90Sdan       if( pErr->rc==SQLITE_OK ) system_error(pErr, rc);
767*4be02b90Sdan     }else{
768*4be02b90Sdan       printf("Thread %d says: %s\n", p->iTid, (ret==0 ? "..." : (char *)ret));
769*4be02b90Sdan     }
770*4be02b90Sdan     sqlite3_free(p);
771*4be02b90Sdan   }
772*4be02b90Sdan   pThreads->pThread = 0;
773*4be02b90Sdan }
774*4be02b90Sdan 
775*4be02b90Sdan static i64 filesize_x(
776*4be02b90Sdan   Error *pErr,
777*4be02b90Sdan   const char *zFile
778*4be02b90Sdan ){
779*4be02b90Sdan   i64 iRet = 0;
780*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
781*4be02b90Sdan     struct stat sStat;
782*4be02b90Sdan     if( stat(zFile, &sStat) ){
783*4be02b90Sdan       iRet = -1;
784*4be02b90Sdan     }else{
785*4be02b90Sdan       iRet = sStat.st_size;
786*4be02b90Sdan     }
787*4be02b90Sdan   }
788*4be02b90Sdan   return iRet;
789*4be02b90Sdan }
790*4be02b90Sdan 
791*4be02b90Sdan static void filecopy_x(
792*4be02b90Sdan   Error *pErr,
793*4be02b90Sdan   const char *zFrom,
794*4be02b90Sdan   const char *zTo
795*4be02b90Sdan ){
796*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
797*4be02b90Sdan     i64 nByte = filesize_x(pErr, zFrom);
798*4be02b90Sdan     if( nByte<0 ){
799*4be02b90Sdan       test_error_x(pErr, sqlite3_mprintf("no such file: %s", zFrom));
800*4be02b90Sdan     }else{
801*4be02b90Sdan       i64 iOff;
802*4be02b90Sdan       char aBuf[1024];
803*4be02b90Sdan       int fd1;
804*4be02b90Sdan       int fd2;
805*4be02b90Sdan       unlink(zTo);
806*4be02b90Sdan 
807*4be02b90Sdan       fd1 = open(zFrom, O_RDONLY);
808*4be02b90Sdan       if( fd1<0 ){
809*4be02b90Sdan         system_error(pErr, errno);
810*4be02b90Sdan         return;
811*4be02b90Sdan       }
812*4be02b90Sdan       fd2 = open(zTo, O_RDWR|O_CREAT|O_EXCL, 0644);
813*4be02b90Sdan       if( fd2<0 ){
814*4be02b90Sdan         system_error(pErr, errno);
815*4be02b90Sdan         close(fd1);
816*4be02b90Sdan         return;
817*4be02b90Sdan       }
818*4be02b90Sdan 
819*4be02b90Sdan       iOff = 0;
820*4be02b90Sdan       while( iOff<nByte ){
821*4be02b90Sdan         int nCopy = sizeof(aBuf);
822*4be02b90Sdan         if( nCopy+iOff>nByte ){
823*4be02b90Sdan           nCopy = nByte - iOff;
824*4be02b90Sdan         }
825*4be02b90Sdan         if( nCopy!=read(fd1, aBuf, nCopy) ){
826*4be02b90Sdan           system_error(pErr, errno);
827*4be02b90Sdan           break;
828*4be02b90Sdan         }
829*4be02b90Sdan         if( nCopy!=write(fd2, aBuf, nCopy) ){
830*4be02b90Sdan           system_error(pErr, errno);
831*4be02b90Sdan           break;
832*4be02b90Sdan         }
833*4be02b90Sdan         iOff += nCopy;
834*4be02b90Sdan       }
835*4be02b90Sdan 
836*4be02b90Sdan       close(fd1);
837*4be02b90Sdan       close(fd2);
838*4be02b90Sdan     }
839*4be02b90Sdan   }
840*4be02b90Sdan }
841*4be02b90Sdan 
842*4be02b90Sdan /*
843*4be02b90Sdan ** Used by setstoptime() and timetostop().
844*4be02b90Sdan */
845*4be02b90Sdan static double timelimit = 0.0;
846*4be02b90Sdan static sqlite3_vfs *pTimelimitVfs = 0;
847*4be02b90Sdan 
848*4be02b90Sdan static void setstoptime_x(
849*4be02b90Sdan   Error *pErr,                    /* IN/OUT: Error code */
850*4be02b90Sdan   int nMs                         /* Milliseconds until "stop time" */
851*4be02b90Sdan ){
852*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
853*4be02b90Sdan     double t;
854*4be02b90Sdan     int rc;
855*4be02b90Sdan     pTimelimitVfs = sqlite3_vfs_find(0);
856*4be02b90Sdan     rc = pTimelimitVfs->xCurrentTime(pTimelimitVfs, &t);
857*4be02b90Sdan     if( rc!=SQLITE_OK ){
858*4be02b90Sdan       pErr->rc = rc;
859*4be02b90Sdan     }else{
860*4be02b90Sdan       timelimit = t + ((double)nMs)/(1000.0*60.0*60.0*24.0);
861*4be02b90Sdan     }
862*4be02b90Sdan   }
863*4be02b90Sdan }
864*4be02b90Sdan 
865*4be02b90Sdan static int timetostop_x(
866*4be02b90Sdan   Error *pErr                     /* IN/OUT: Error code */
867*4be02b90Sdan ){
868*4be02b90Sdan   int ret = 1;
869*4be02b90Sdan   if( pErr->rc==SQLITE_OK ){
870*4be02b90Sdan     double t;
871*4be02b90Sdan     int rc;
872*4be02b90Sdan     rc = pTimelimitVfs->xCurrentTime(pTimelimitVfs, &t);
873*4be02b90Sdan     if( rc!=SQLITE_OK ){
874*4be02b90Sdan       pErr->rc = rc;
875*4be02b90Sdan     }else{
876*4be02b90Sdan       ret = (t >= timelimit);
877*4be02b90Sdan     }
878*4be02b90Sdan   }
879*4be02b90Sdan   return ret;
880*4be02b90Sdan }
881*4be02b90Sdan 
882*4be02b90Sdan /*
883*4be02b90Sdan ** The "Set Error Line" macro.
884*4be02b90Sdan */
885*4be02b90Sdan #define SEL(e) ((e)->iLine = ((e)->rc ? (e)->iLine : __LINE__))
886*4be02b90Sdan 
887*4be02b90Sdan 
888*4be02b90Sdan /*************************************************************************
889*4be02b90Sdan **************************************************************************
890*4be02b90Sdan **************************************************************************
891*4be02b90Sdan ** End infrastructure. Begin tests.
892*4be02b90Sdan */
893*4be02b90Sdan 
894*4be02b90Sdan #define WALTHREAD1_NTHREAD  10
895*4be02b90Sdan #define WALTHREAD3_NTHREAD  6
896*4be02b90Sdan 
897*4be02b90Sdan static char *walthread1_thread(int iTid, int iArg){
898*4be02b90Sdan   Error err = {0};                /* Error code and message */
899*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
900*4be02b90Sdan   int nIter = 0;                  /* Iterations so far */
901*4be02b90Sdan 
902*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
903*4be02b90Sdan   while( !timetostop(&err) ){
904*4be02b90Sdan     const char *azSql[] = {
905*4be02b90Sdan       "SELECT md5sum(x) FROM t1 WHERE rowid != (SELECT max(rowid) FROM t1)",
906*4be02b90Sdan       "SELECT x FROM t1 WHERE rowid = (SELECT max(rowid) FROM t1)",
907*4be02b90Sdan     };
908*4be02b90Sdan     char *z1, *z2, *z3;
909*4be02b90Sdan 
910*4be02b90Sdan     execsql(&err, &db, "BEGIN");
911*4be02b90Sdan     integrity_check(&err, &db);
912*4be02b90Sdan     z1 = execsql_text(&err, &db, 1, azSql[0]);
913*4be02b90Sdan     z2 = execsql_text(&err, &db, 2, azSql[1]);
914*4be02b90Sdan     z3 = execsql_text(&err, &db, 3, azSql[0]);
915*4be02b90Sdan     execsql(&err, &db, "COMMIT");
916*4be02b90Sdan 
917*4be02b90Sdan     if( strcmp(z1, z2) || strcmp(z1, z3) ){
918*4be02b90Sdan       test_error(&err, "Failed read: %s %s %s", z1, z2, z3);
919*4be02b90Sdan     }
920*4be02b90Sdan 
921*4be02b90Sdan     sql_script(&err, &db,
922*4be02b90Sdan         "BEGIN;"
923*4be02b90Sdan           "INSERT INTO t1 VALUES(randomblob(100));"
924*4be02b90Sdan           "INSERT INTO t1 VALUES(randomblob(100));"
925*4be02b90Sdan           "INSERT INTO t1 SELECT md5sum(x) FROM t1;"
926*4be02b90Sdan         "COMMIT;"
927*4be02b90Sdan     );
928*4be02b90Sdan     nIter++;
929*4be02b90Sdan   }
930*4be02b90Sdan   closedb(&err, &db);
931*4be02b90Sdan 
932*4be02b90Sdan   print_and_free_err(&err);
933*4be02b90Sdan   return sqlite3_mprintf("%d iterations", nIter);
934*4be02b90Sdan }
935*4be02b90Sdan 
936*4be02b90Sdan static char *walthread1_ckpt_thread(int iTid, int iArg){
937*4be02b90Sdan   Error err = {0};                /* Error code and message */
938*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
939*4be02b90Sdan   int nCkpt = 0;                  /* Checkpoints so far */
940*4be02b90Sdan 
941*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
942*4be02b90Sdan   while( !timetostop(&err) ){
943*4be02b90Sdan     usleep(500*1000);
944*4be02b90Sdan     execsql(&err, &db, "PRAGMA wal_checkpoint");
945*4be02b90Sdan     if( err.rc==SQLITE_OK ) nCkpt++;
946*4be02b90Sdan     clear_error(&err, SQLITE_BUSY);
947*4be02b90Sdan   }
948*4be02b90Sdan   closedb(&err, &db);
949*4be02b90Sdan 
950*4be02b90Sdan   print_and_free_err(&err);
951*4be02b90Sdan   return sqlite3_mprintf("%d checkpoints", nCkpt);
952*4be02b90Sdan }
953*4be02b90Sdan 
954*4be02b90Sdan static void walthread1(int nMs){
955*4be02b90Sdan   Error err = {0};                /* Error code and message */
956*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
957*4be02b90Sdan   Threadset threads = {0};        /* Test threads */
958*4be02b90Sdan   int i;                          /* Iterator variable */
959*4be02b90Sdan 
960*4be02b90Sdan   opendb(&err, &db, "test.db", 1);
961*4be02b90Sdan   sql_script(&err, &db,
962*4be02b90Sdan       "PRAGMA journal_mode = WAL;"
963*4be02b90Sdan       "CREATE TABLE t1(x PRIMARY KEY);"
964*4be02b90Sdan       "INSERT INTO t1 VALUES(randomblob(100));"
965*4be02b90Sdan       "INSERT INTO t1 VALUES(randomblob(100));"
966*4be02b90Sdan       "INSERT INTO t1 SELECT md5sum(x) FROM t1;"
967*4be02b90Sdan   );
968*4be02b90Sdan 
969*4be02b90Sdan   setstoptime(&err, nMs);
970*4be02b90Sdan   for(i=0; i<WALTHREAD1_NTHREAD; i++){
971*4be02b90Sdan     launch_thread(&err, &threads, walthread1_thread, 0);
972*4be02b90Sdan   }
973*4be02b90Sdan   launch_thread(&err, &threads, walthread1_ckpt_thread, 0);
974*4be02b90Sdan   join_all_threads(&err, &threads);
975*4be02b90Sdan 
976*4be02b90Sdan   print_and_free_err(&err);
977*4be02b90Sdan }
978*4be02b90Sdan 
979*4be02b90Sdan static char *walthread2_thread(int iTid, int iArg){
980*4be02b90Sdan   Error err = {0};                /* Error code and message */
981*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
982*4be02b90Sdan   int anTrans[2] = {0, 0};        /* Number of WAL and Rollback transactions */
983*4be02b90Sdan 
984*4be02b90Sdan   const char *zJournal = "PRAGMA journal_mode = WAL";
985*4be02b90Sdan   if( iArg ){ zJournal = "PRAGMA journal_mode = DELETE"; }
986*4be02b90Sdan 
987*4be02b90Sdan   while( !timetostop(&err) ){
988*4be02b90Sdan     int journal_exists = 0;
989*4be02b90Sdan     int wal_exists = 0;
990*4be02b90Sdan 
991*4be02b90Sdan     opendb(&err, &db, "test.db", 0);
992*4be02b90Sdan 
993*4be02b90Sdan     sql_script(&err, &db, zJournal);
994*4be02b90Sdan     clear_error(&err, SQLITE_BUSY);
995*4be02b90Sdan     sql_script(&err, &db, "BEGIN");
996*4be02b90Sdan     sql_script(&err, &db, "INSERT INTO t1 VALUES(NULL, randomblob(100))");
997*4be02b90Sdan 
998*4be02b90Sdan     journal_exists = (filesize(&err, "test.db-journal") >= 0);
999*4be02b90Sdan     wal_exists = (filesize(&err, "test.db-wal") >= 0);
1000*4be02b90Sdan     if( (journal_exists+wal_exists)!=1 ){
1001*4be02b90Sdan       test_error(&err, "File system looks incorrect (%d, %d)",
1002*4be02b90Sdan           journal_exists, wal_exists
1003*4be02b90Sdan       );
1004*4be02b90Sdan     }
1005*4be02b90Sdan     anTrans[journal_exists]++;
1006*4be02b90Sdan 
1007*4be02b90Sdan     sql_script(&err, &db, "COMMIT");
1008*4be02b90Sdan     integrity_check(&err, &db);
1009*4be02b90Sdan     closedb(&err, &db);
1010*4be02b90Sdan   }
1011*4be02b90Sdan 
1012*4be02b90Sdan   print_and_free_err(&err);
1013*4be02b90Sdan   return sqlite3_mprintf("W %d R %d", anTrans[0], anTrans[1]);
1014*4be02b90Sdan }
1015*4be02b90Sdan 
1016*4be02b90Sdan static void walthread2(int nMs){
1017*4be02b90Sdan   Error err = {0};
1018*4be02b90Sdan   Sqlite db = {0};
1019*4be02b90Sdan   Threadset threads = {0};
1020*4be02b90Sdan 
1021*4be02b90Sdan   opendb(&err, &db, "test.db", 1);
1022*4be02b90Sdan   sql_script(&err, &db, "CREATE TABLE t1(x INTEGER PRIMARY KEY, y UNIQUE)");
1023*4be02b90Sdan   closedb(&err, &db);
1024*4be02b90Sdan 
1025*4be02b90Sdan   setstoptime(&err, nMs);
1026*4be02b90Sdan   launch_thread(&err, &threads, walthread2_thread, 0);
1027*4be02b90Sdan   launch_thread(&err, &threads, walthread2_thread, 0);
1028*4be02b90Sdan   launch_thread(&err, &threads, walthread2_thread, 1);
1029*4be02b90Sdan   launch_thread(&err, &threads, walthread2_thread, 1);
1030*4be02b90Sdan   join_all_threads(&err, &threads);
1031*4be02b90Sdan 
1032*4be02b90Sdan   print_and_free_err(&err);
1033*4be02b90Sdan }
1034*4be02b90Sdan 
1035*4be02b90Sdan static char *walthread3_thread(int iTid, int iArg){
1036*4be02b90Sdan   Error err = {0};                /* Error code and message */
1037*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
1038*4be02b90Sdan   i64 iNextWrite;                 /* Next value this thread will write */
1039*4be02b90Sdan 
1040*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
1041*4be02b90Sdan   sql_script(&err, &db, "PRAGMA wal_autocheckpoint = 10");
1042*4be02b90Sdan 
1043*4be02b90Sdan   iNextWrite = iArg+1;
1044*4be02b90Sdan   while( 1 ){
1045*4be02b90Sdan     i64 sum1;
1046*4be02b90Sdan     i64 sum2;
1047*4be02b90Sdan     int stop = 0;                 /* True to stop executing (test timed out) */
1048*4be02b90Sdan 
1049*4be02b90Sdan     while( 0==(stop = timetostop(&err)) ){
1050*4be02b90Sdan       i64 iMax = execsql_i64(&err, &db, "SELECT max(cnt) FROM t1");
1051*4be02b90Sdan       if( iMax+1==iNextWrite ) break;
1052*4be02b90Sdan     }
1053*4be02b90Sdan     if( stop ) break;
1054*4be02b90Sdan 
1055*4be02b90Sdan     sum1 = execsql_i64(&err, &db, "SELECT sum(cnt) FROM t1");
1056*4be02b90Sdan     sum2 = execsql_i64(&err, &db, "SELECT sum(sum1) FROM t1");
1057*4be02b90Sdan     execsql_i64(&err, &db,
1058*4be02b90Sdan         "INSERT INTO t1 VALUES(:iNextWrite, :iSum1, :iSum2)",
1059*4be02b90Sdan         &iNextWrite, &sum1, &sum2
1060*4be02b90Sdan     );
1061*4be02b90Sdan     integrity_check(&err, &db);
1062*4be02b90Sdan 
1063*4be02b90Sdan     iNextWrite += WALTHREAD3_NTHREAD;
1064*4be02b90Sdan   }
1065*4be02b90Sdan 
1066*4be02b90Sdan   closedb(&err, &db);
1067*4be02b90Sdan   print_and_free_err(&err);
1068*4be02b90Sdan   return 0;
1069*4be02b90Sdan }
1070*4be02b90Sdan 
1071*4be02b90Sdan static void walthread3(int nMs){
1072*4be02b90Sdan   Error err = {0};
1073*4be02b90Sdan   Sqlite db = {0};
1074*4be02b90Sdan   Threadset threads = {0};
1075*4be02b90Sdan   int i;
1076*4be02b90Sdan 
1077*4be02b90Sdan   opendb(&err, &db, "test.db", 1);
1078*4be02b90Sdan   sql_script(&err, &db,
1079*4be02b90Sdan       "PRAGMA journal_mode = WAL;"
1080*4be02b90Sdan       "CREATE TABLE t1(cnt PRIMARY KEY, sum1, sum2);"
1081*4be02b90Sdan       "CREATE INDEX i1 ON t1(sum1);"
1082*4be02b90Sdan       "CREATE INDEX i2 ON t1(sum2);"
1083*4be02b90Sdan       "INSERT INTO t1 VALUES(0, 0, 0);"
1084*4be02b90Sdan   );
1085*4be02b90Sdan   closedb(&err, &db);
1086*4be02b90Sdan 
1087*4be02b90Sdan   setstoptime(&err, nMs);
1088*4be02b90Sdan   for(i=0; i<WALTHREAD3_NTHREAD; i++){
1089*4be02b90Sdan     launch_thread(&err, &threads, walthread3_thread, i);
1090*4be02b90Sdan   }
1091*4be02b90Sdan   join_all_threads(&err, &threads);
1092*4be02b90Sdan 
1093*4be02b90Sdan   print_and_free_err(&err);
1094*4be02b90Sdan }
1095*4be02b90Sdan 
1096*4be02b90Sdan static char *walthread4_reader_thread(int iTid, int iArg){
1097*4be02b90Sdan   Error err = {0};                /* Error code and message */
1098*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
1099*4be02b90Sdan 
1100*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
1101*4be02b90Sdan   while( !timetostop(&err) ){
1102*4be02b90Sdan     integrity_check(&err, &db);
1103*4be02b90Sdan   }
1104*4be02b90Sdan   closedb(&err, &db);
1105*4be02b90Sdan 
1106*4be02b90Sdan   print_and_free_err(&err);
1107*4be02b90Sdan   return 0;
1108*4be02b90Sdan }
1109*4be02b90Sdan 
1110*4be02b90Sdan static char *walthread4_writer_thread(int iTid, int iArg){
1111*4be02b90Sdan   Error err = {0};                /* Error code and message */
1112*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
1113*4be02b90Sdan   i64 iRow = 1;
1114*4be02b90Sdan 
1115*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
1116*4be02b90Sdan   sql_script(&err, &db, "PRAGMA wal_autocheckpoint = 15;");
1117*4be02b90Sdan   while( !timetostop(&err) ){
1118*4be02b90Sdan     execsql_i64(
1119*4be02b90Sdan         &err, &db, "REPLACE INTO t1 VALUES(:iRow, randomblob(300))", &iRow
1120*4be02b90Sdan     );
1121*4be02b90Sdan     iRow++;
1122*4be02b90Sdan     if( iRow==10 ) iRow = 0;
1123*4be02b90Sdan   }
1124*4be02b90Sdan   closedb(&err, &db);
1125*4be02b90Sdan 
1126*4be02b90Sdan   print_and_free_err(&err);
1127*4be02b90Sdan   return 0;
1128*4be02b90Sdan }
1129*4be02b90Sdan 
1130*4be02b90Sdan static void walthread4(int nMs){
1131*4be02b90Sdan   Error err = {0};
1132*4be02b90Sdan   Sqlite db = {0};
1133*4be02b90Sdan   Threadset threads = {0};
1134*4be02b90Sdan 
1135*4be02b90Sdan   opendb(&err, &db, "test.db", 1);
1136*4be02b90Sdan   sql_script(&err, &db,
1137*4be02b90Sdan       "PRAGMA journal_mode = WAL;"
1138*4be02b90Sdan       "CREATE TABLE t1(a INTEGER PRIMARY KEY, b UNIQUE);"
1139*4be02b90Sdan   );
1140*4be02b90Sdan   closedb(&err, &db);
1141*4be02b90Sdan 
1142*4be02b90Sdan   setstoptime(&err, nMs);
1143*4be02b90Sdan   launch_thread(&err, &threads, walthread4_reader_thread, 0);
1144*4be02b90Sdan   launch_thread(&err, &threads, walthread4_writer_thread, 0);
1145*4be02b90Sdan   join_all_threads(&err, &threads);
1146*4be02b90Sdan 
1147*4be02b90Sdan   print_and_free_err(&err);
1148*4be02b90Sdan }
1149*4be02b90Sdan 
1150*4be02b90Sdan static char *walthread5_thread(int iTid, int iArg){
1151*4be02b90Sdan   Error err = {0};                /* Error code and message */
1152*4be02b90Sdan   Sqlite db = {0};                /* SQLite database connection */
1153*4be02b90Sdan   i64 nRow;
1154*4be02b90Sdan 
1155*4be02b90Sdan   opendb(&err, &db, "test.db", 0);
1156*4be02b90Sdan   nRow = execsql_i64(&err, &db, "SELECT count(*) FROM t1");
1157*4be02b90Sdan   closedb(&err, &db);
1158*4be02b90Sdan 
1159*4be02b90Sdan   if( nRow!=65536 ) test_error(&err, "Bad row count: %d", (int)nRow);
1160*4be02b90Sdan   print_and_free_err(&err);
1161*4be02b90Sdan   return 0;
1162*4be02b90Sdan }
1163*4be02b90Sdan static void walthread5(int nMs){
1164*4be02b90Sdan   Error err = {0};
1165*4be02b90Sdan   Sqlite db = {0};
1166*4be02b90Sdan   Threadset threads = {0};
1167*4be02b90Sdan 
1168*4be02b90Sdan   opendb(&err, &db, "test.db", 1);
1169*4be02b90Sdan   sql_script(&err, &db,
1170*4be02b90Sdan       "PRAGMA wal_autocheckpoint = 0;"
1171*4be02b90Sdan       "PRAGMA page_size = 1024;"
1172*4be02b90Sdan       "PRAGMA journal_mode = WAL;"
1173*4be02b90Sdan       "CREATE TABLE t1(x);"
1174*4be02b90Sdan       "BEGIN;"
1175*4be02b90Sdan       "INSERT INTO t1 VALUES(randomblob(900));"
1176*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*     2 */"
1177*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*     4 */"
1178*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*     8 */"
1179*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*    16 */"
1180*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*    32 */"
1181*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*    64 */"
1182*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*   128 */"
1183*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*   256 */"
1184*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*   512 */"
1185*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*  1024 */"
1186*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*  2048 */"
1187*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*  4096 */"
1188*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /*  8192 */"
1189*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /* 16384 */"
1190*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /* 32768 */"
1191*4be02b90Sdan       "INSERT INTO t1 SELECT randomblob(900) FROM t1;      /* 65536 */"
1192*4be02b90Sdan       "COMMIT;"
1193*4be02b90Sdan   );
1194*4be02b90Sdan   filecopy(&err, "test.db", "test_sv.db");
1195*4be02b90Sdan   filecopy(&err, "test.db-wal", "test_sv.db-wal");
1196*4be02b90Sdan   closedb(&err, &db);
1197*4be02b90Sdan 
1198*4be02b90Sdan   filecopy(&err, "test_sv.db", "test.db");
1199*4be02b90Sdan   filecopy(&err, "test_sv.db-wal", "test.db-wal");
1200*4be02b90Sdan 
1201*4be02b90Sdan   if( err.rc==SQLITE_OK ){
1202*4be02b90Sdan     printf("  WAL file is %d bytes,", (int)filesize(&err,"test.db-wal"));
1203*4be02b90Sdan     printf(" DB file is %d.\n", (int)filesize(&err,"test.db"));
1204*4be02b90Sdan   }
1205*4be02b90Sdan 
1206*4be02b90Sdan   setstoptime(&err, nMs);
1207*4be02b90Sdan   launch_thread(&err, &threads, walthread5_thread, 0);
1208*4be02b90Sdan   launch_thread(&err, &threads, walthread5_thread, 0);
1209*4be02b90Sdan   launch_thread(&err, &threads, walthread5_thread, 0);
1210*4be02b90Sdan   launch_thread(&err, &threads, walthread5_thread, 0);
1211*4be02b90Sdan   launch_thread(&err, &threads, walthread5_thread, 0);
1212*4be02b90Sdan   join_all_threads(&err, &threads);
1213*4be02b90Sdan 
1214*4be02b90Sdan   if( err.rc==SQLITE_OK ){
1215*4be02b90Sdan     printf("  WAL file is %d bytes,", (int)filesize(&err,"test.db-wal"));
1216*4be02b90Sdan     printf(" DB file is %d.\n", (int)filesize(&err,"test.db"));
1217*4be02b90Sdan   }
1218*4be02b90Sdan 
1219*4be02b90Sdan   print_and_free_err(&err);
1220*4be02b90Sdan }
1221*4be02b90Sdan 
1222*4be02b90Sdan int main(int argc, char **argv){
1223*4be02b90Sdan   struct ThreadTest {
1224*4be02b90Sdan     void (*xTest)(int);
1225*4be02b90Sdan     const char *zTest;
1226*4be02b90Sdan     int nMs;
1227*4be02b90Sdan   } aTest[] = {
1228*4be02b90Sdan     { walthread1, "walthread1", 20000 },
1229*4be02b90Sdan     { walthread2, "walthread2", 20000 },
1230*4be02b90Sdan     { walthread3, "walthread3", 20000 },
1231*4be02b90Sdan     { walthread4, "walthread4", 20000 },
1232*4be02b90Sdan     { walthread5, "walthread5",  1000 },
1233*4be02b90Sdan   };
1234*4be02b90Sdan 
1235*4be02b90Sdan   int i;
1236*4be02b90Sdan   char *zTest = 0;
1237*4be02b90Sdan   int nTest = 0;
1238*4be02b90Sdan   int bTestfound = 0;
1239*4be02b90Sdan   int bPrefix = 0;
1240*4be02b90Sdan 
1241*4be02b90Sdan   if( argc>2 ) goto usage;
1242*4be02b90Sdan   if( argc==2 ){
1243*4be02b90Sdan     zTest = argv[1];
1244*4be02b90Sdan     nTest = strlen(zTest);
1245*4be02b90Sdan     if( zTest[nTest-1]=='*' ){
1246*4be02b90Sdan       nTest--;
1247*4be02b90Sdan       bPrefix = 1;
1248*4be02b90Sdan     }
1249*4be02b90Sdan   }
1250*4be02b90Sdan 
1251*4be02b90Sdan   sqlite3_config(SQLITE_CONFIG_MULTITHREAD);
1252*4be02b90Sdan 
1253*4be02b90Sdan   for(i=0; i<sizeof(aTest)/sizeof(aTest[0]); i++){
1254*4be02b90Sdan     char const *z = aTest[i].zTest;
1255*4be02b90Sdan     int n = strlen(z);
1256*4be02b90Sdan     if( !zTest || ((bPrefix || n==nTest) && 0==strncmp(zTest, z, nTest)) ){
1257*4be02b90Sdan       printf("Running %s for %d seconds...\n", z, aTest[i].nMs/1000);
1258*4be02b90Sdan       aTest[i].xTest(aTest[i].nMs);
1259*4be02b90Sdan       bTestfound++;
1260*4be02b90Sdan     }
1261*4be02b90Sdan   }
1262*4be02b90Sdan   if( bTestfound==0 ) goto usage;
1263*4be02b90Sdan 
1264*4be02b90Sdan   printf("Total of %d errors across all tests\n", nGlobalErr);
1265*4be02b90Sdan   return (nGlobalErr>0 ? 255 : 0);
1266*4be02b90Sdan 
1267*4be02b90Sdan  usage:
1268*4be02b90Sdan   printf("Usage: %s [testname|testprefix*]\n", argv[0]);
1269*4be02b90Sdan   printf("Available tests are:\n");
1270*4be02b90Sdan   for(i=0; i<sizeof(aTest)/sizeof(aTest[0]); i++){
1271*4be02b90Sdan     printf("   %s\n", aTest[i].zTest);
1272*4be02b90Sdan   }
1273*4be02b90Sdan 
1274*4be02b90Sdan   return 254;
1275*4be02b90Sdan }
1276*4be02b90Sdan 
1277*4be02b90Sdan 
1278