1 /*
2 ** This program is used to generate and verify databases with hot journals.
3 ** Use this program to generate a hot journal on one machine and verify
4 ** that it rolls back correctly on another machine with a different
5 ** architecture.
6 **
7 ** Usage:
8 **
9 ** rollback-test new [-utf8] [-utf16le] [-utf16be] [-pagesize=N] DATABASE
10 ** rollback-test check DATABASE
11 ** rollback-test crash [-wal] [-rollback] DATABASE
12 */
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include "sqlite3.h"
17
usage(char * argv0)18 static void usage(char *argv0){
19 fprintf(stderr,
20 "Usage: %s new [-utf8] [-utf16le] [-utf16be] [-pagesize=N] DATABASE\n"
21 " %s check DATABASE\n"
22 " %s crash [-wal] DATABASE\n",
23 argv0, argv0, argv0
24 );
25 exit(1);
26 }
27
openDb(const char * zFilename)28 static sqlite3 *openDb(const char *zFilename){
29 int rc;
30 sqlite3 *db;
31 rc = sqlite3_open(zFilename, &db);
32 if( rc ){
33 fprintf(stderr, "Cannot open \"%s\": %s\n",
34 zFilename, sqlite3_errmsg(db));
35 sqlite3_close(db);
36 exit(1);
37 }
38 return db;
39 }
40
41 static int nReply = 0;
42 static char zReply[1000];
43
execCallback(void * NotUsed,int nArg,char ** azArg,char ** azCol)44 static int execCallback(void *NotUsed, int nArg, char **azArg, char **azCol){
45 int i, n;
46 char *z;
47 for(i=0; i<nArg; i++){
48 z = azArg[i];
49 if( z==0 ) z = "NULL";
50 if( nReply>0 && nReply<sizeof(zReply)-1 ) zReply[nReply++] = ' ';
51 n = strlen(z);
52 if( nReply+n>=sizeof(zReply)-1 ) n = sizeof(zReply) - nReply - 1;
53 memcpy(&zReply[nReply], z, n);
54 nReply += n;
55 zReply[nReply] = 0;
56 }
57 return 0;
58 }
59
runSql(sqlite3 * db,const char * zSql)60 static void runSql(sqlite3 *db, const char *zSql){
61 char *zErr = 0;
62 int rc;
63 nReply = 0;
64 rc = sqlite3_exec(db, zSql, execCallback, 0, &zErr);
65 if( zErr ){
66 fprintf(stderr, "SQL error: %s\n", zErr);
67 exit(1);
68 }
69 if( rc ){
70 fprintf(stderr, "SQL error: %s\n", sqlite3_errmsg(db));
71 exit(1);
72 }
73 }
74
main(int argc,char ** argv)75 int main(int argc, char **argv){
76 sqlite3 *db;
77 int i;
78
79 if( argc<3 ) usage(argv[0]);
80 if( strcmp(argv[1], "new")==0 ){
81 db = openDb(argv[argc-1]);
82 for(i=2; i<argc-1; i++){
83 if( strcmp(argv[i],"-utf8")==0 ){
84 runSql(db, "PRAGMA encoding=UTF8");
85 }else if( strcmp(argv[i], "-utf16le")==0 ){
86 runSql(db, "PRAGMA encoding=UTF16LE");
87 }else if( strcmp(argv[i], "-utf16be")==0 ){
88 runSql(db, "PRAGMA encoding=UTF16BE");
89 }else if( strncmp(argv[i], "-pagesize=", 10)==0 ){
90 int szPg = atoi(&argv[i][10]);
91 char zBuf[100];
92 sprintf(zBuf, "PRAGMA pagesize=%d", szPg);
93 runSql(db, zBuf);
94 }else{
95 fprintf(stderr, "unknown option %s\n", argv[i]);
96 usage(argv[0]);
97 }
98 }
99 runSql(db,
100 "BEGIN;"
101 "CREATE TABLE t1(x INTEGER PRIMARY KEY, y);"
102 "INSERT INTO t1(y) VALUES('abcdefghijklmnopqrstuvwxyz');"
103 "INSERT INTO t1(y) VALUES('abcdefghijklmnopqrstuvwxyz');"
104 "INSERT INTO t1(y) SELECT y FROM t1;" /* 4 */
105 "INSERT INTO t1(y) SELECT y FROM t1;" /* 8 */
106 "INSERT INTO t1(y) SELECT y FROM t1;" /* 16 */
107 "INSERT INTO t1(y) SELECT y FROM t1;" /* 32 */
108 "INSERT INTO t1(y) SELECT y FROM t1;" /* 64 */
109 "INSERT INTO t1(y) SELECT y FROM t1;" /* 128 */
110 "INSERT INTO t1(y) SELECT y FROM t1;" /* 256 */
111 "INSERT INTO t1(y) SELECT y FROM t1;" /* 512 */
112 "INSERT INTO t1(y) SELECT y FROM t1;" /* 1024 */
113 "UPDATE t1 SET y=(y || x);"
114 "CREATE INDEX t1y ON t1(y);"
115 "COMMIT;"
116 );
117 sqlite3_close(db);
118 }else if( strcmp(argv[1], "check")==0 ){
119 db = openDb(argv[argc-1]);
120 runSql(db, "PRAGMA integrity_check");
121 if( strcmp(zReply, "ok")!=0 ){
122 fprintf(stderr, "Integrity check: %s\n", zReply);
123 exit(1);
124 }
125 runSql(db,
126 "SELECT count(*) FROM t1 WHERE y<>('abcdefghijklmnopqrstuvwxyz' || x)"
127 );
128 if( strcmp(zReply, "0")!=0 ){
129 fprintf(stderr, "Wrong content\n");
130 exit(1);
131 }
132 printf("Ok\n");
133 }else if( strcmp(argv[1], "crash")==0 ){
134 db = openDb(argv[argc-1]);
135 for(i=2; i<argc-1; i++){
136 if( strcmp(argv[i],"-wal")==0 ){
137 runSql(db, "PRAGMA journal_mode=WAL");
138 }else if( strcmp(argv[i], "-rollback")==0 ){
139 runSql(db, "PRAGMA journal_mode=DELETE");
140 }else{
141 fprintf(stderr, "unknown option %s\n", argv[i]);
142 usage(argv[0]);
143 }
144 }
145 runSql(db,
146 "PRAGMA cache_size=10;"
147 "BEGIN;"
148 "UPDATE t1 SET y=(y || -x)"
149 );
150 exit(0);
151 }else{
152 usage(argv[0]);
153 }
154 return 0;
155 }
156