Home | History | Annotate | Download | only in tool
      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 }
    156