Return to rollback-test.c CVS log

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

Annotation of embedaddon/sqlite3/tool/rollback-test.c, revision 1.1

1.1     ! misho       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: }