third_party/sqlite/src/tool/rollback-test.c

   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
  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
  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
  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
  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
  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 }