| Index: third_party/sqlite/src/test/threadtest2.c
|
| diff --git a/third_party/sqlite/src/test/threadtest2.c b/third_party/sqlite/src/test/threadtest2.c
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..25b1d908d284610f7fd97410c9dfce5fdb0d0d9d
|
| --- /dev/null
|
| +++ b/third_party/sqlite/src/test/threadtest2.c
|
| @@ -0,0 +1,133 @@
|
| +/*
|
| +** 2004 January 13
|
| +**
|
| +** The author disclaims copyright to this source code. In place of
|
| +** a legal notice, here is a blessing:
|
| +**
|
| +** May you do good and not evil.
|
| +** May you find forgiveness for yourself and forgive others.
|
| +** May you share freely, never taking more than you give.
|
| +**
|
| +*************************************************************************
|
| +** This file implements a simple standalone program used to test whether
|
| +** or not the SQLite library is threadsafe.
|
| +**
|
| +** This file is NOT part of the standard SQLite library. It is used for
|
| +** testing only.
|
| +*/
|
| +#include <stdio.h>
|
| +#include <unistd.h>
|
| +#include <pthread.h>
|
| +#include <string.h>
|
| +#include <stdlib.h>
|
| +#include "sqlite.h"
|
| +
|
| +/*
|
| +** Name of the database
|
| +*/
|
| +#define DB_FILE "test.db"
|
| +
|
| +/*
|
| +** When this variable becomes non-zero, all threads stop
|
| +** what they are doing.
|
| +*/
|
| +volatile int all_stop = 0;
|
| +
|
| +/*
|
| +** Callback from the integrity check. If the result is anything other
|
| +** than "ok" it means the integrity check has failed. Set the "all_stop"
|
| +** global variable to stop all other activity. Print the error message
|
| +** or print OK if the string "ok" is seen.
|
| +*/
|
| +int check_callback(void *pid, int argc, char **argv, char **notUsed2){
|
| + int id = (int)pid;
|
| + if( strcmp(argv[0],"ok") ){
|
| + all_stop = 1;
|
| + fprintf(stderr,"id: %s\n", id, argv[0]);
|
| + }else{
|
| + /* fprintf(stderr,"%d: OK\n", id); */
|
| + }
|
| + return 0;
|
| +}
|
| +
|
| +/*
|
| +** Do an integrity check on the database. If the first integrity check
|
| +** fails, try it a second time.
|
| +*/
|
| +int integrity_check(sqlite *db, int id){
|
| + int rc;
|
| + if( all_stop ) return 0;
|
| + /* fprintf(stderr,"%d: CHECK\n", id); */
|
| + rc = sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
|
| + if( rc!=SQLITE_OK && rc!=SQLITE_BUSY ){
|
| + fprintf(stderr,"%d, Integrity check returns %d\n", id, rc);
|
| + }
|
| + if( all_stop ){
|
| + sqlite3_exec(db, "pragma integrity_check", check_callback, 0, 0);
|
| + }
|
| + return 0;
|
| +}
|
| +
|
| +/*
|
| +** This is the worker thread
|
| +*/
|
| +void *worker(void *workerArg){
|
| + sqlite *db;
|
| + int id = (int)workerArg;
|
| + int rc;
|
| + int cnt = 0;
|
| + fprintf(stderr, "Starting worker %d\n", id);
|
| + while( !all_stop && cnt++<10000 ){
|
| + if( cnt%100==0 ) printf("%d: %d\n", id, cnt);
|
| + while( (sqlite3_open(DB_FILE, &db))!=SQLITE_OK ) sched_yield();
|
| + sqlite3_exec(db, "PRAGMA synchronous=OFF", 0, 0, 0);
|
| + /* integrity_check(db, id); */
|
| + if( all_stop ){ sqlite3_close(db); break; }
|
| + /* fprintf(stderr, "%d: BEGIN\n", id); */
|
| + rc = sqlite3_exec(db, "INSERT INTO t1 VALUES('bogus data')", 0, 0, 0);
|
| + /* fprintf(stderr, "%d: END rc=%d\n", id, rc); */
|
| + sqlite3_close(db);
|
| + }
|
| + fprintf(stderr, "Worker %d finished\n", id);
|
| + return 0;
|
| +}
|
| +
|
| +/*
|
| +** Initialize the database and start the threads
|
| +*/
|
| +int main(int argc, char **argv){
|
| + sqlite *db;
|
| + int i, rc;
|
| + pthread_t aThread[5];
|
| +
|
| + if( strcmp(DB_FILE,":memory:") ){
|
| + char *zJournal = sqlite3_mprintf("%s-journal", DB_FILE);
|
| + unlink(DB_FILE);
|
| + unlink(zJournal);
|
| + sqlite3_free(zJournal);
|
| + }
|
| + sqlite3_open(DB_FILE, &db);
|
| + if( db==0 ){
|
| + fprintf(stderr,"unable to initialize database\n");
|
| + exit(1);
|
| + }
|
| + rc = sqlite3_exec(db, "CREATE TABLE t1(x);", 0,0,0);
|
| + if( rc ){
|
| + fprintf(stderr,"cannot create table t1: %d\n", rc);
|
| + exit(1);
|
| + }
|
| + sqlite3_close(db);
|
| + for(i=0; i<sizeof(aThread)/sizeof(aThread[0]); i++){
|
| + pthread_create(&aThread[i], 0, worker, (void*)i);
|
| + }
|
| + for(i=0; i<sizeof(aThread)/sizeof(aThread[i]); i++){
|
| + pthread_join(aThread[i], 0);
|
| + }
|
| + if( !all_stop ){
|
| + printf("Everything seems ok.\n");
|
| + return 0;
|
| + }else{
|
| + printf("We hit an error.\n");
|
| + return 1;
|
| + }
|
| +}
|
|
|