Return to rollback-test.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / sqlite3 / tool

sqlite3

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