diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 14:07:11 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-13 14:07:11 +0000 |
commit | 63847496f14c813a5d80efd5b7de0f1294ffe1e3 (patch) | |
tree | 01c7571c7c762ceee70638549a99834fdd7c411b /tool/rollback-test.c | |
parent | Initial commit. (diff) | |
download | sqlite3-63847496f14c813a5d80efd5b7de0f1294ffe1e3.tar.xz sqlite3-63847496f14c813a5d80efd5b7de0f1294ffe1e3.zip |
Adding upstream version 3.45.1.upstream/3.45.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tool/rollback-test.c')
-rw-r--r-- | tool/rollback-test.c | 155 |
1 files changed, 155 insertions, 0 deletions
diff --git a/tool/rollback-test.c b/tool/rollback-test.c new file mode 100644 index 0000000..915d9d2 --- /dev/null +++ b/tool/rollback-test.c @@ -0,0 +1,155 @@ +/* +** This program is used to generate and verify databases with hot journals. +** Use this program to generate a hot journal on one machine and verify +** that it rolls back correctly on another machine with a different +** architecture. +** +** Usage: +** +** rollback-test new [-utf8] [-utf16le] [-utf16be] [-pagesize=N] DATABASE +** rollback-test check DATABASE +** rollback-test crash [-wal] [-rollback] DATABASE +*/ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include "sqlite3.h" + +static void usage(char *argv0){ + fprintf(stderr, + "Usage: %s new [-utf8] [-utf16le] [-utf16be] [-pagesize=N] DATABASE\n" + " %s check DATABASE\n" + " %s crash [-wal] DATABASE\n", + argv0, argv0, argv0 + ); + exit(1); +} + +static sqlite3 *openDb(const char *zFilename){ + int rc; + sqlite3 *db; + rc = sqlite3_open(zFilename, &db); + if( rc ){ + fprintf(stderr, "Cannot open \"%s\": %s\n", + zFilename, sqlite3_errmsg(db)); + sqlite3_close(db); + exit(1); + } + return db; +} + +static int nReply = 0; +static char zReply[1000]; + +static int execCallback(void *NotUsed, int nArg, char **azArg, char **azCol){ + int i, n; + char *z; + for(i=0; i<nArg; i++){ + z = azArg[i]; + if( z==0 ) z = "NULL"; + if( nReply>0 && nReply<sizeof(zReply)-1 ) zReply[nReply++] = ' '; + n = strlen(z); + if( nReply+n>=sizeof(zReply)-1 ) n = sizeof(zReply) - nReply - 1; + memcpy(&zReply[nReply], z, n); + nReply += n; + zReply[nReply] = 0; + } + return 0; +} + +static void runSql(sqlite3 *db, const char *zSql){ + char *zErr = 0; + int rc; + nReply = 0; + rc = sqlite3_exec(db, zSql, execCallback, 0, &zErr); + if( zErr ){ + fprintf(stderr, "SQL error: %s\n", zErr); + exit(1); + } + if( rc ){ + fprintf(stderr, "SQL error: %s\n", sqlite3_errmsg(db)); + exit(1); + } +} + +int main(int argc, char **argv){ + sqlite3 *db; + int i; + + if( argc<3 ) usage(argv[0]); + if( strcmp(argv[1], "new")==0 ){ + db = openDb(argv[argc-1]); + for(i=2; i<argc-1; i++){ + if( strcmp(argv[i],"-utf8")==0 ){ + runSql(db, "PRAGMA encoding=UTF8"); + }else if( strcmp(argv[i], "-utf16le")==0 ){ + runSql(db, "PRAGMA encoding=UTF16LE"); + }else if( strcmp(argv[i], "-utf16be")==0 ){ + runSql(db, "PRAGMA encoding=UTF16BE"); + }else if( strncmp(argv[i], "-pagesize=", 10)==0 ){ + int szPg = atoi(&argv[i][10]); + char zBuf[100]; + sprintf(zBuf, "PRAGMA pagesize=%d", szPg); + runSql(db, zBuf); + }else{ + fprintf(stderr, "unknown option %s\n", argv[i]); + usage(argv[0]); + } + } + runSql(db, + "BEGIN;" + "CREATE TABLE t1(x INTEGER PRIMARY KEY, y);" + "INSERT INTO t1(y) VALUES('abcdefghijklmnopqrstuvwxyz');" + "INSERT INTO t1(y) VALUES('abcdefghijklmnopqrstuvwxyz');" + "INSERT INTO t1(y) SELECT y FROM t1;" /* 4 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 8 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 16 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 32 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 64 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 128 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 256 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 512 */ + "INSERT INTO t1(y) SELECT y FROM t1;" /* 1024 */ + "UPDATE t1 SET y=(y || x);" + "CREATE INDEX t1y ON t1(y);" + "COMMIT;" + ); + sqlite3_close(db); + }else if( strcmp(argv[1], "check")==0 ){ + db = openDb(argv[argc-1]); + runSql(db, "PRAGMA integrity_check"); + if( strcmp(zReply, "ok")!=0 ){ + fprintf(stderr, "Integrity check: %s\n", zReply); + exit(1); + } + runSql(db, + "SELECT count(*) FROM t1 WHERE y<>('abcdefghijklmnopqrstuvwxyz' || x)" + ); + if( strcmp(zReply, "0")!=0 ){ + fprintf(stderr, "Wrong content\n"); + exit(1); + } + printf("Ok\n"); + }else if( strcmp(argv[1], "crash")==0 ){ + db = openDb(argv[argc-1]); + for(i=2; i<argc-1; i++){ + if( strcmp(argv[i],"-wal")==0 ){ + runSql(db, "PRAGMA journal_mode=WAL"); + }else if( strcmp(argv[i], "-rollback")==0 ){ + runSql(db, "PRAGMA journal_mode=DELETE"); + }else{ + fprintf(stderr, "unknown option %s\n", argv[i]); + usage(argv[0]); + } + } + runSql(db, + "PRAGMA cache_size=10;" + "BEGIN;" + "UPDATE t1 SET y=(y || -x)" + ); + exit(0); + }else{ + usage(argv[0]); + } + return 0; +} |