| Index: third_party/sqlite/sqlite-src-3080704/mptest/mptest.c
|
| diff --git a/third_party/sqlite/sqlite-src-3080704/mptest/mptest.c b/third_party/sqlite/sqlite-src-3080704/mptest/mptest.c
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..059ae102fafb8bc398606d4265788821b5489da9
|
| --- /dev/null
|
| +++ b/third_party/sqlite/sqlite-src-3080704/mptest/mptest.c
|
| @@ -0,0 +1,1401 @@
|
| +/*
|
| +** 2013-04-05
|
| +**
|
| +** 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 is a program used for testing SQLite, and specifically for testing
|
| +** the ability of independent processes to access the same SQLite database
|
| +** concurrently.
|
| +**
|
| +** Compile this program as follows:
|
| +**
|
| +** gcc -g -c -Wall sqlite3.c $(OPTS)
|
| +** gcc -g -o mptest mptest.c sqlite3.o $(LIBS)
|
| +**
|
| +** Recommended options:
|
| +**
|
| +** -DHAVE_USLEEP
|
| +** -DSQLITE_NO_SYNC
|
| +** -DSQLITE_THREADSAFE=0
|
| +** -DSQLITE_OMIT_LOAD_EXTENSION
|
| +**
|
| +** Run like this:
|
| +**
|
| +** ./mptest $database $script
|
| +**
|
| +** where $database is the database to use for testing and $script is a
|
| +** test script.
|
| +*/
|
| +#include "sqlite3.h"
|
| +#include <stdio.h>
|
| +#if defined(_WIN32)
|
| +# define WIN32_LEAN_AND_MEAN
|
| +# include <windows.h>
|
| +#else
|
| +# include <unistd.h>
|
| +#endif
|
| +#include <stdlib.h>
|
| +#include <string.h>
|
| +#include <assert.h>
|
| +#include <ctype.h>
|
| +
|
| +/* The suffix to append to the child command lines, if any */
|
| +#if defined(_WIN32)
|
| +# define GETPID (int)GetCurrentProcessId
|
| +#else
|
| +# define GETPID getpid
|
| +#endif
|
| +
|
| +/* Mark a parameter as unused to suppress compiler warnings */
|
| +#define UNUSED_PARAMETER(x) (void)x
|
| +
|
| +/* Global data
|
| +*/
|
| +static struct Global {
|
| + char *argv0; /* Name of the executable */
|
| + const char *zVfs; /* Name of VFS to use. Often NULL meaning "default" */
|
| + char *zDbFile; /* Name of the database */
|
| + sqlite3 *db; /* Open connection to database */
|
| + char *zErrLog; /* Filename for error log */
|
| + FILE *pErrLog; /* Where to write errors */
|
| + char *zLog; /* Name of output log file */
|
| + FILE *pLog; /* Where to write log messages */
|
| + char zName[32]; /* Symbolic name of this process */
|
| + int taskId; /* Task ID. 0 means supervisor. */
|
| + int iTrace; /* Tracing level */
|
| + int bSqlTrace; /* True to trace SQL commands */
|
| + int bIgnoreSqlErrors; /* Ignore errors in SQL statements */
|
| + int nError; /* Number of errors */
|
| + int nTest; /* Number of --match operators */
|
| + int iTimeout; /* Milliseconds until a busy timeout */
|
| + int bSync; /* Call fsync() */
|
| +} g;
|
| +
|
| +/* Default timeout */
|
| +#define DEFAULT_TIMEOUT 10000
|
| +
|
| +/*
|
| +** Print a message adding zPrefix[] to the beginning of every line.
|
| +*/
|
| +static void printWithPrefix(FILE *pOut, const char *zPrefix, const char *zMsg){
|
| + while( zMsg && zMsg[0] ){
|
| + int i;
|
| + for(i=0; zMsg[i] && zMsg[i]!='\n' && zMsg[i]!='\r'; i++){}
|
| + fprintf(pOut, "%s%.*s\n", zPrefix, i, zMsg);
|
| + zMsg += i;
|
| + while( zMsg[0]=='\n' || zMsg[0]=='\r' ) zMsg++;
|
| + }
|
| +}
|
| +
|
| +/*
|
| +** Compare two pointers to strings, where the pointers might be NULL.
|
| +*/
|
| +static int safe_strcmp(const char *a, const char *b){
|
| + if( a==b ) return 0;
|
| + if( a==0 ) return -1;
|
| + if( b==0 ) return 1;
|
| + return strcmp(a,b);
|
| +}
|
| +
|
| +/*
|
| +** Return TRUE if string z[] matches glob pattern zGlob[].
|
| +** Return FALSE if the pattern does not match.
|
| +**
|
| +** Globbing rules:
|
| +**
|
| +** '*' Matches any sequence of zero or more characters.
|
| +**
|
| +** '?' Matches exactly one character.
|
| +**
|
| +** [...] Matches one character from the enclosed list of
|
| +** characters.
|
| +**
|
| +** [^...] Matches one character not in the enclosed list.
|
| +**
|
| +** '#' Matches any sequence of one or more digits with an
|
| +** optional + or - sign in front
|
| +*/
|
| +int strglob(const char *zGlob, const char *z){
|
| + int c, c2;
|
| + int invert;
|
| + int seen;
|
| +
|
| + while( (c = (*(zGlob++)))!=0 ){
|
| + if( c=='*' ){
|
| + while( (c=(*(zGlob++))) == '*' || c=='?' ){
|
| + if( c=='?' && (*(z++))==0 ) return 0;
|
| + }
|
| + if( c==0 ){
|
| + return 1;
|
| + }else if( c=='[' ){
|
| + while( *z && strglob(zGlob-1,z) ){
|
| + z++;
|
| + }
|
| + return (*z)!=0;
|
| + }
|
| + while( (c2 = (*(z++)))!=0 ){
|
| + while( c2!=c ){
|
| + c2 = *(z++);
|
| + if( c2==0 ) return 0;
|
| + }
|
| + if( strglob(zGlob,z) ) return 1;
|
| + }
|
| + return 0;
|
| + }else if( c=='?' ){
|
| + if( (*(z++))==0 ) return 0;
|
| + }else if( c=='[' ){
|
| + int prior_c = 0;
|
| + seen = 0;
|
| + invert = 0;
|
| + c = *(z++);
|
| + if( c==0 ) return 0;
|
| + c2 = *(zGlob++);
|
| + if( c2=='^' ){
|
| + invert = 1;
|
| + c2 = *(zGlob++);
|
| + }
|
| + if( c2==']' ){
|
| + if( c==']' ) seen = 1;
|
| + c2 = *(zGlob++);
|
| + }
|
| + while( c2 && c2!=']' ){
|
| + if( c2=='-' && zGlob[0]!=']' && zGlob[0]!=0 && prior_c>0 ){
|
| + c2 = *(zGlob++);
|
| + if( c>=prior_c && c<=c2 ) seen = 1;
|
| + prior_c = 0;
|
| + }else{
|
| + if( c==c2 ){
|
| + seen = 1;
|
| + }
|
| + prior_c = c2;
|
| + }
|
| + c2 = *(zGlob++);
|
| + }
|
| + if( c2==0 || (seen ^ invert)==0 ) return 0;
|
| + }else if( c=='#' ){
|
| + if( (z[0]=='-' || z[0]=='+') && isdigit(z[1]) ) z++;
|
| + if( !isdigit(z[0]) ) return 0;
|
| + z++;
|
| + while( isdigit(z[0]) ){ z++; }
|
| + }else{
|
| + if( c!=(*(z++)) ) return 0;
|
| + }
|
| + }
|
| + return *z==0;
|
| +}
|
| +
|
| +/*
|
| +** Close output stream pOut if it is not stdout or stderr
|
| +*/
|
| +static void maybeClose(FILE *pOut){
|
| + if( pOut!=stdout && pOut!=stderr ) fclose(pOut);
|
| +}
|
| +
|
| +/*
|
| +** Print an error message
|
| +*/
|
| +static void errorMessage(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zMsg;
|
| + char zPrefix[30];
|
| + va_start(ap, zFormat);
|
| + zMsg = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s:ERROR: ", g.zName);
|
| + if( g.pLog ){
|
| + printWithPrefix(g.pLog, zPrefix, zMsg);
|
| + fflush(g.pLog);
|
| + }
|
| + if( g.pErrLog && safe_strcmp(g.zErrLog,g.zLog) ){
|
| + printWithPrefix(g.pErrLog, zPrefix, zMsg);
|
| + fflush(g.pErrLog);
|
| + }
|
| + sqlite3_free(zMsg);
|
| + g.nError++;
|
| +}
|
| +
|
| +/* Forward declaration */
|
| +static int trySql(const char*, ...);
|
| +
|
| +/*
|
| +** Print an error message and then quit.
|
| +*/
|
| +static void fatalError(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zMsg;
|
| + char zPrefix[30];
|
| + va_start(ap, zFormat);
|
| + zMsg = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s:FATAL: ", g.zName);
|
| + if( g.pLog ){
|
| + printWithPrefix(g.pLog, zPrefix, zMsg);
|
| + fflush(g.pLog);
|
| + maybeClose(g.pLog);
|
| + }
|
| + if( g.pErrLog && safe_strcmp(g.zErrLog,g.zLog) ){
|
| + printWithPrefix(g.pErrLog, zPrefix, zMsg);
|
| + fflush(g.pErrLog);
|
| + maybeClose(g.pErrLog);
|
| + }
|
| + sqlite3_free(zMsg);
|
| + if( g.db ){
|
| + int nTry = 0;
|
| + g.iTimeout = 0;
|
| + while( trySql("UPDATE client SET wantHalt=1;")==SQLITE_BUSY
|
| + && (nTry++)<100 ){
|
| + sqlite3_sleep(10);
|
| + }
|
| + }
|
| + sqlite3_close(g.db);
|
| + exit(1);
|
| +}
|
| +
|
| +
|
| +/*
|
| +** Print a log message
|
| +*/
|
| +static void logMessage(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zMsg;
|
| + char zPrefix[30];
|
| + va_start(ap, zFormat);
|
| + zMsg = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s: ", g.zName);
|
| + if( g.pLog ){
|
| + printWithPrefix(g.pLog, zPrefix, zMsg);
|
| + fflush(g.pLog);
|
| + }
|
| + sqlite3_free(zMsg);
|
| +}
|
| +
|
| +/*
|
| +** Return the length of a string omitting trailing whitespace
|
| +*/
|
| +static int clipLength(const char *z){
|
| + int n = (int)strlen(z);
|
| + while( n>0 && isspace(z[n-1]) ){ n--; }
|
| + return n;
|
| +}
|
| +
|
| +/*
|
| +** Auxiliary SQL function to return the name of the VFS
|
| +*/
|
| +static void vfsNameFunc(
|
| + sqlite3_context *context,
|
| + int argc,
|
| + sqlite3_value **argv
|
| +){
|
| + sqlite3 *db = sqlite3_context_db_handle(context);
|
| + char *zVfs = 0;
|
| + UNUSED_PARAMETER(argc);
|
| + UNUSED_PARAMETER(argv);
|
| + sqlite3_file_control(db, "main", SQLITE_FCNTL_VFSNAME, &zVfs);
|
| + if( zVfs ){
|
| + sqlite3_result_text(context, zVfs, -1, sqlite3_free);
|
| + }
|
| +}
|
| +
|
| +/*
|
| +** Busy handler with a g.iTimeout-millisecond timeout
|
| +*/
|
| +static int busyHandler(void *pCD, int count){
|
| + UNUSED_PARAMETER(pCD);
|
| + if( count*10>g.iTimeout ){
|
| + if( g.iTimeout>0 ) errorMessage("timeout after %dms", g.iTimeout);
|
| + return 0;
|
| + }
|
| + sqlite3_sleep(10);
|
| + return 1;
|
| +}
|
| +
|
| +/*
|
| +** SQL Trace callback
|
| +*/
|
| +static void sqlTraceCallback(void *NotUsed1, const char *zSql){
|
| + UNUSED_PARAMETER(NotUsed1);
|
| + logMessage("[%.*s]", clipLength(zSql), zSql);
|
| +}
|
| +
|
| +/*
|
| +** SQL error log callback
|
| +*/
|
| +static void sqlErrorCallback(void *pArg, int iErrCode, const char *zMsg){
|
| + UNUSED_PARAMETER(pArg);
|
| + if( iErrCode==SQLITE_ERROR && g.bIgnoreSqlErrors ) return;
|
| + if( (iErrCode&0xff)==SQLITE_SCHEMA && g.iTrace<3 ) return;
|
| + if( g.iTimeout==0 && (iErrCode&0xff)==SQLITE_BUSY && g.iTrace<3 ) return;
|
| + if( (iErrCode&0xff)==SQLITE_NOTICE ){
|
| + logMessage("(info) %s", zMsg);
|
| + }else{
|
| + errorMessage("(errcode=%d) %s", iErrCode, zMsg);
|
| + }
|
| +}
|
| +
|
| +/*
|
| +** Prepare an SQL statement. Issue a fatal error if unable.
|
| +*/
|
| +static sqlite3_stmt *prepareSql(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zSql;
|
| + int rc;
|
| + sqlite3_stmt *pStmt = 0;
|
| + va_start(ap, zFormat);
|
| + zSql = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + rc = sqlite3_prepare_v2(g.db, zSql, -1, &pStmt, 0);
|
| + if( rc!=SQLITE_OK ){
|
| + sqlite3_finalize(pStmt);
|
| + fatalError("%s\n%s\n", sqlite3_errmsg(g.db), zSql);
|
| + }
|
| + sqlite3_free(zSql);
|
| + return pStmt;
|
| +}
|
| +
|
| +/*
|
| +** Run arbitrary SQL. Issue a fatal error on failure.
|
| +*/
|
| +static void runSql(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zSql;
|
| + int rc;
|
| + va_start(ap, zFormat);
|
| + zSql = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + rc = sqlite3_exec(g.db, zSql, 0, 0, 0);
|
| + if( rc!=SQLITE_OK ){
|
| + fatalError("%s\n%s\n", sqlite3_errmsg(g.db), zSql);
|
| + }
|
| + sqlite3_free(zSql);
|
| +}
|
| +
|
| +/*
|
| +** Try to run arbitrary SQL. Return success code.
|
| +*/
|
| +static int trySql(const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zSql;
|
| + int rc;
|
| + va_start(ap, zFormat);
|
| + zSql = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + rc = sqlite3_exec(g.db, zSql, 0, 0, 0);
|
| + sqlite3_free(zSql);
|
| + return rc;
|
| +}
|
| +
|
| +/* Structure for holding an arbitrary length string
|
| +*/
|
| +typedef struct String String;
|
| +struct String {
|
| + char *z; /* the string */
|
| + int n; /* Slots of z[] used */
|
| + int nAlloc; /* Slots of z[] allocated */
|
| +};
|
| +
|
| +/* Free a string */
|
| +static void stringFree(String *p){
|
| + if( p->z ) sqlite3_free(p->z);
|
| + memset(p, 0, sizeof(*p));
|
| +}
|
| +
|
| +/* Append n bytes of text to a string. If n<0 append the entire string. */
|
| +static void stringAppend(String *p, const char *z, int n){
|
| + if( n<0 ) n = (int)strlen(z);
|
| + if( p->n+n>=p->nAlloc ){
|
| + int nAlloc = p->nAlloc*2 + n + 100;
|
| + char *z = sqlite3_realloc(p->z, nAlloc);
|
| + if( z==0 ) fatalError("out of memory");
|
| + p->z = z;
|
| + p->nAlloc = nAlloc;
|
| + }
|
| + memcpy(p->z+p->n, z, n);
|
| + p->n += n;
|
| + p->z[p->n] = 0;
|
| +}
|
| +
|
| +/* Reset a string to an empty string */
|
| +static void stringReset(String *p){
|
| + if( p->z==0 ) stringAppend(p, " ", 1);
|
| + p->n = 0;
|
| + p->z[0] = 0;
|
| +}
|
| +
|
| +/* Append a new token onto the end of the string */
|
| +static void stringAppendTerm(String *p, const char *z){
|
| + int i;
|
| + if( p->n ) stringAppend(p, " ", 1);
|
| + if( z==0 ){
|
| + stringAppend(p, "nil", 3);
|
| + return;
|
| + }
|
| + for(i=0; z[i] && !isspace(z[i]); i++){}
|
| + if( i>0 && z[i]==0 ){
|
| + stringAppend(p, z, i);
|
| + return;
|
| + }
|
| + stringAppend(p, "'", 1);
|
| + while( z[0] ){
|
| + for(i=0; z[i] && z[i]!='\''; i++){}
|
| + if( z[i] ){
|
| + stringAppend(p, z, i+1);
|
| + stringAppend(p, "'", 1);
|
| + z += i+1;
|
| + }else{
|
| + stringAppend(p, z, i);
|
| + break;
|
| + }
|
| + }
|
| + stringAppend(p, "'", 1);
|
| +}
|
| +
|
| +/*
|
| +** Callback function for evalSql()
|
| +*/
|
| +static int evalCallback(void *pCData, int argc, char **argv, char **azCol){
|
| + String *p = (String*)pCData;
|
| + int i;
|
| + UNUSED_PARAMETER(azCol);
|
| + for(i=0; i<argc; i++) stringAppendTerm(p, argv[i]);
|
| + return 0;
|
| +}
|
| +
|
| +/*
|
| +** Run arbitrary SQL and record the results in an output string
|
| +** given by the first parameter.
|
| +*/
|
| +static int evalSql(String *p, const char *zFormat, ...){
|
| + va_list ap;
|
| + char *zSql;
|
| + int rc;
|
| + char *zErrMsg = 0;
|
| + va_start(ap, zFormat);
|
| + zSql = sqlite3_vmprintf(zFormat, ap);
|
| + va_end(ap);
|
| + assert( g.iTimeout>0 );
|
| + rc = sqlite3_exec(g.db, zSql, evalCallback, p, &zErrMsg);
|
| + sqlite3_free(zSql);
|
| + if( rc ){
|
| + char zErr[30];
|
| + sqlite3_snprintf(sizeof(zErr), zErr, "error(%d)", rc);
|
| + stringAppendTerm(p, zErr);
|
| + if( zErrMsg ){
|
| + stringAppendTerm(p, zErrMsg);
|
| + sqlite3_free(zErrMsg);
|
| + }
|
| + }
|
| + return rc;
|
| +}
|
| +
|
| +/*
|
| +** Auxiliary SQL function to recursively evaluate SQL.
|
| +*/
|
| +static void evalFunc(
|
| + sqlite3_context *context,
|
| + int argc,
|
| + sqlite3_value **argv
|
| +){
|
| + sqlite3 *db = sqlite3_context_db_handle(context);
|
| + const char *zSql = (const char*)sqlite3_value_text(argv[0]);
|
| + String res;
|
| + char *zErrMsg = 0;
|
| + int rc;
|
| + UNUSED_PARAMETER(argc);
|
| + memset(&res, 0, sizeof(res));
|
| + rc = sqlite3_exec(db, zSql, evalCallback, &res, &zErrMsg);
|
| + if( zErrMsg ){
|
| + sqlite3_result_error(context, zErrMsg, -1);
|
| + sqlite3_free(zErrMsg);
|
| + }else if( rc ){
|
| + sqlite3_result_error_code(context, rc);
|
| + }else{
|
| + sqlite3_result_text(context, res.z, -1, SQLITE_TRANSIENT);
|
| + }
|
| + stringFree(&res);
|
| +}
|
| +
|
| +/*
|
| +** Look up the next task for client iClient in the database.
|
| +** Return the task script and the task number and mark that
|
| +** task as being under way.
|
| +*/
|
| +static int startScript(
|
| + int iClient, /* The client number */
|
| + char **pzScript, /* Write task script here */
|
| + int *pTaskId, /* Write task number here */
|
| + char **pzTaskName /* Name of the task */
|
| +){
|
| + sqlite3_stmt *pStmt = 0;
|
| + int taskId;
|
| + int rc;
|
| + int totalTime = 0;
|
| +
|
| + *pzScript = 0;
|
| + g.iTimeout = 0;
|
| + while(1){
|
| + rc = trySql("BEGIN IMMEDIATE");
|
| + if( rc==SQLITE_BUSY ){
|
| + sqlite3_sleep(10);
|
| + totalTime += 10;
|
| + continue;
|
| + }
|
| + if( rc!=SQLITE_OK ){
|
| + fatalError("in startScript: %s", sqlite3_errmsg(g.db));
|
| + }
|
| + if( g.nError || g.nTest ){
|
| + runSql("UPDATE counters SET nError=nError+%d, nTest=nTest+%d",
|
| + g.nError, g.nTest);
|
| + g.nError = 0;
|
| + g.nTest = 0;
|
| + }
|
| + pStmt = prepareSql("SELECT 1 FROM client WHERE id=%d AND wantHalt",iClient);
|
| + rc = sqlite3_step(pStmt);
|
| + sqlite3_finalize(pStmt);
|
| + if( rc==SQLITE_ROW ){
|
| + runSql("DELETE FROM client WHERE id=%d", iClient);
|
| + g.iTimeout = DEFAULT_TIMEOUT;
|
| + runSql("COMMIT TRANSACTION;");
|
| + return SQLITE_DONE;
|
| + }
|
| + pStmt = prepareSql(
|
| + "SELECT script, id, name FROM task"
|
| + " WHERE client=%d AND starttime IS NULL"
|
| + " ORDER BY id LIMIT 1", iClient);
|
| + rc = sqlite3_step(pStmt);
|
| + if( rc==SQLITE_ROW ){
|
| + int n = sqlite3_column_bytes(pStmt, 0);
|
| + *pzScript = sqlite3_malloc(n+1);
|
| + strcpy(*pzScript, (const char*)sqlite3_column_text(pStmt, 0));
|
| + *pTaskId = taskId = sqlite3_column_int(pStmt, 1);
|
| + *pzTaskName = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 2));
|
| + sqlite3_finalize(pStmt);
|
| + runSql("UPDATE task"
|
| + " SET starttime=strftime('%%Y-%%m-%%d %%H:%%M:%%f','now')"
|
| + " WHERE id=%d;", taskId);
|
| + g.iTimeout = DEFAULT_TIMEOUT;
|
| + runSql("COMMIT TRANSACTION;");
|
| + return SQLITE_OK;
|
| + }
|
| + sqlite3_finalize(pStmt);
|
| + if( rc==SQLITE_DONE ){
|
| + if( totalTime>30000 ){
|
| + errorMessage("Waited over 30 seconds with no work. Giving up.");
|
| + runSql("DELETE FROM client WHERE id=%d; COMMIT;", iClient);
|
| + sqlite3_close(g.db);
|
| + exit(1);
|
| + }
|
| + while( trySql("COMMIT")==SQLITE_BUSY ){
|
| + sqlite3_sleep(10);
|
| + totalTime += 10;
|
| + }
|
| + sqlite3_sleep(100);
|
| + totalTime += 100;
|
| + continue;
|
| + }
|
| + fatalError("%s", sqlite3_errmsg(g.db));
|
| + }
|
| + g.iTimeout = DEFAULT_TIMEOUT;
|
| +}
|
| +
|
| +/*
|
| +** Mark a script as having finished. Remove the CLIENT table entry
|
| +** if bShutdown is true.
|
| +*/
|
| +static int finishScript(int iClient, int taskId, int bShutdown){
|
| + runSql("UPDATE task"
|
| + " SET endtime=strftime('%%Y-%%m-%%d %%H:%%M:%%f','now')"
|
| + " WHERE id=%d;", taskId);
|
| + if( bShutdown ){
|
| + runSql("DELETE FROM client WHERE id=%d", iClient);
|
| + }
|
| + return SQLITE_OK;
|
| +}
|
| +
|
| +/*
|
| +** Start up a client process for iClient, if it is not already
|
| +** running. If the client is already running, then this routine
|
| +** is a no-op.
|
| +*/
|
| +static void startClient(int iClient){
|
| + runSql("INSERT OR IGNORE INTO client VALUES(%d,0)", iClient);
|
| + if( sqlite3_changes(g.db) ){
|
| + char *zSys;
|
| + int rc;
|
| + zSys = sqlite3_mprintf("%s \"%s\" --client %d --trace %d",
|
| + g.argv0, g.zDbFile, iClient, g.iTrace);
|
| + if( g.bSqlTrace ){
|
| + zSys = sqlite3_mprintf("%z --sqltrace", zSys);
|
| + }
|
| + if( g.bSync ){
|
| + zSys = sqlite3_mprintf("%z --sync", zSys);
|
| + }
|
| + if( g.zVfs ){
|
| + zSys = sqlite3_mprintf("%z --vfs \"%s\"", zSys, g.zVfs);
|
| + }
|
| + if( g.iTrace>=2 ) logMessage("system('%q')", zSys);
|
| +#if !defined(_WIN32)
|
| + zSys = sqlite3_mprintf("%z &", zSys);
|
| + rc = system(zSys);
|
| + if( rc ) errorMessage("system() fails with error code %d", rc);
|
| +#else
|
| + {
|
| + STARTUPINFOA startupInfo;
|
| + PROCESS_INFORMATION processInfo;
|
| + memset(&startupInfo, 0, sizeof(startupInfo));
|
| + startupInfo.cb = sizeof(startupInfo);
|
| + memset(&processInfo, 0, sizeof(processInfo));
|
| + rc = CreateProcessA(NULL, zSys, NULL, NULL, FALSE, 0, NULL, NULL,
|
| + &startupInfo, &processInfo);
|
| + if( rc ){
|
| + CloseHandle(processInfo.hThread);
|
| + CloseHandle(processInfo.hProcess);
|
| + }else{
|
| + errorMessage("CreateProcessA() fails with error code %lu",
|
| + GetLastError());
|
| + }
|
| + }
|
| +#endif
|
| + sqlite3_free(zSys);
|
| + }
|
| +}
|
| +
|
| +/*
|
| +** Read the entire content of a file into memory
|
| +*/
|
| +static char *readFile(const char *zFilename){
|
| + FILE *in = fopen(zFilename, "rb");
|
| + long sz;
|
| + char *z;
|
| + if( in==0 ){
|
| + fatalError("cannot open \"%s\" for reading", zFilename);
|
| + }
|
| + fseek(in, 0, SEEK_END);
|
| + sz = ftell(in);
|
| + rewind(in);
|
| + z = sqlite3_malloc( sz+1 );
|
| + sz = (long)fread(z, 1, sz, in);
|
| + z[sz] = 0;
|
| + fclose(in);
|
| + return z;
|
| +}
|
| +
|
| +/*
|
| +** Return the length of the next token.
|
| +*/
|
| +static int tokenLength(const char *z, int *pnLine){
|
| + int n = 0;
|
| + if( isspace(z[0]) || (z[0]=='/' && z[1]=='*') ){
|
| + int inC = 0;
|
| + int c;
|
| + if( z[0]=='/' ){
|
| + inC = 1;
|
| + n = 2;
|
| + }
|
| + while( (c = z[n++])!=0 ){
|
| + if( c=='\n' ) (*pnLine)++;
|
| + if( isspace(c) ) continue;
|
| + if( inC && c=='*' && z[n]=='/' ){
|
| + n++;
|
| + inC = 0;
|
| + }else if( !inC && c=='/' && z[n]=='*' ){
|
| + n++;
|
| + inC = 1;
|
| + }else if( !inC ){
|
| + break;
|
| + }
|
| + }
|
| + n--;
|
| + }else if( z[0]=='-' && z[1]=='-' ){
|
| + for(n=2; z[n] && z[n]!='\n'; n++){}
|
| + if( z[n] ){ (*pnLine)++; n++; }
|
| + }else if( z[0]=='"' || z[0]=='\'' ){
|
| + int delim = z[0];
|
| + for(n=1; z[n]; n++){
|
| + if( z[n]=='\n' ) (*pnLine)++;
|
| + if( z[n]==delim ){
|
| + n++;
|
| + if( z[n+1]!=delim ) break;
|
| + }
|
| + }
|
| + }else{
|
| + int c;
|
| + for(n=1; (c = z[n])!=0 && !isspace(c) && c!='"' && c!='\'' && c!=';'; n++){}
|
| + }
|
| + return n;
|
| +}
|
| +
|
| +/*
|
| +** Copy a single token into a string buffer.
|
| +*/
|
| +static int extractToken(const char *zIn, int nIn, char *zOut, int nOut){
|
| + int i;
|
| + if( nIn<=0 ){
|
| + zOut[0] = 0;
|
| + return 0;
|
| + }
|
| + for(i=0; i<nIn && i<nOut-1 && !isspace(zIn[i]); i++){ zOut[i] = zIn[i]; }
|
| + zOut[i] = 0;
|
| + return i;
|
| +}
|
| +
|
| +/*
|
| +** Find the number of characters up to the start of the next "--end" token.
|
| +*/
|
| +static int findEnd(const char *z, int *pnLine){
|
| + int n = 0;
|
| + while( z[n] && (strncmp(z+n,"--end",5) || !isspace(z[n+5])) ){
|
| + n += tokenLength(z+n, pnLine);
|
| + }
|
| + return n;
|
| +}
|
| +
|
| +/*
|
| +** Find the number of characters up to the first character past the
|
| +** of the next "--endif" or "--else" token. Nested --if commands are
|
| +** also skipped.
|
| +*/
|
| +static int findEndif(const char *z, int stopAtElse, int *pnLine){
|
| + int n = 0;
|
| + while( z[n] ){
|
| + int len = tokenLength(z+n, pnLine);
|
| + if( (strncmp(z+n,"--endif",7)==0 && isspace(z[n+7]))
|
| + || (stopAtElse && strncmp(z+n,"--else",6)==0 && isspace(z[n+6]))
|
| + ){
|
| + return n+len;
|
| + }
|
| + if( strncmp(z+n,"--if",4)==0 && isspace(z[n+4]) ){
|
| + int skip = findEndif(z+n+len, 0, pnLine);
|
| + n += skip + len;
|
| + }else{
|
| + n += len;
|
| + }
|
| + }
|
| + return n;
|
| +}
|
| +
|
| +/*
|
| +** Wait for a client process to complete all its tasks
|
| +*/
|
| +static void waitForClient(int iClient, int iTimeout, char *zErrPrefix){
|
| + sqlite3_stmt *pStmt;
|
| + int rc;
|
| + if( iClient>0 ){
|
| + pStmt = prepareSql(
|
| + "SELECT 1 FROM task"
|
| + " WHERE client=%d"
|
| + " AND client IN (SELECT id FROM client)"
|
| + " AND endtime IS NULL",
|
| + iClient);
|
| + }else{
|
| + pStmt = prepareSql(
|
| + "SELECT 1 FROM task"
|
| + " WHERE client IN (SELECT id FROM client)"
|
| + " AND endtime IS NULL");
|
| + }
|
| + g.iTimeout = 0;
|
| + while( ((rc = sqlite3_step(pStmt))==SQLITE_BUSY || rc==SQLITE_ROW)
|
| + && iTimeout>0
|
| + ){
|
| + sqlite3_reset(pStmt);
|
| + sqlite3_sleep(50);
|
| + iTimeout -= 50;
|
| + }
|
| + sqlite3_finalize(pStmt);
|
| + g.iTimeout = DEFAULT_TIMEOUT;
|
| + if( rc!=SQLITE_DONE ){
|
| + if( zErrPrefix==0 ) zErrPrefix = "";
|
| + if( iClient>0 ){
|
| + errorMessage("%stimeout waiting for client %d", zErrPrefix, iClient);
|
| + }else{
|
| + errorMessage("%stimeout waiting for all clients", zErrPrefix);
|
| + }
|
| + }
|
| +}
|
| +
|
| +/* Return a pointer to the tail of a filename
|
| +*/
|
| +static char *filenameTail(char *z){
|
| + int i, j;
|
| + for(i=j=0; z[i]; i++) if( z[i]=='/' ) j = i+1;
|
| + return z+j;
|
| +}
|
| +
|
| +/*
|
| +** Interpret zArg as a boolean value. Return either 0 or 1.
|
| +*/
|
| +static int booleanValue(char *zArg){
|
| + int i;
|
| + if( zArg==0 ) return 0;
|
| + for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){}
|
| + if( i>0 && zArg[i]==0 ) return atoi(zArg);
|
| + if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){
|
| + return 1;
|
| + }
|
| + if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){
|
| + return 0;
|
| + }
|
| + errorMessage("unknown boolean: [%s]", zArg);
|
| + return 0;
|
| +}
|
| +
|
| +
|
| +/* This routine exists as a convenient place to set a debugger
|
| +** breakpoint.
|
| +*/
|
| +static void test_breakpoint(void){ static volatile int cnt = 0; cnt++; }
|
| +
|
| +/* Maximum number of arguments to a --command */
|
| +#define MX_ARG 2
|
| +
|
| +/*
|
| +** Run a script.
|
| +*/
|
| +static void runScript(
|
| + int iClient, /* The client number, or 0 for the master */
|
| + int taskId, /* The task ID for clients. 0 for master */
|
| + char *zScript, /* Text of the script */
|
| + char *zFilename /* File from which script was read. */
|
| +){
|
| + int lineno = 1;
|
| + int prevLine = 1;
|
| + int ii = 0;
|
| + int iBegin = 0;
|
| + int n, c, j;
|
| + int len;
|
| + int nArg;
|
| + String sResult;
|
| + char zCmd[30];
|
| + char zError[1000];
|
| + char azArg[MX_ARG][100];
|
| +
|
| + memset(&sResult, 0, sizeof(sResult));
|
| + stringReset(&sResult);
|
| + while( (c = zScript[ii])!=0 ){
|
| + prevLine = lineno;
|
| + len = tokenLength(zScript+ii, &lineno);
|
| + if( isspace(c) || (c=='/' && zScript[ii+1]=='*') ){
|
| + ii += len;
|
| + continue;
|
| + }
|
| + if( c!='-' || zScript[ii+1]!='-' || !isalpha(zScript[ii+2]) ){
|
| + ii += len;
|
| + continue;
|
| + }
|
| +
|
| + /* Run any prior SQL before processing the new --command */
|
| + if( ii>iBegin ){
|
| + char *zSql = sqlite3_mprintf("%.*s", ii-iBegin, zScript+iBegin);
|
| + evalSql(&sResult, zSql);
|
| + sqlite3_free(zSql);
|
| + iBegin = ii + len;
|
| + }
|
| +
|
| + /* Parse the --command */
|
| + if( g.iTrace>=2 ) logMessage("%.*s", len, zScript+ii);
|
| + n = extractToken(zScript+ii+2, len-2, zCmd, sizeof(zCmd));
|
| + for(nArg=0; n<len-2 && nArg<MX_ARG; nArg++){
|
| + while( n<len-2 && isspace(zScript[ii+2+n]) ){ n++; }
|
| + if( n>=len-2 ) break;
|
| + n += extractToken(zScript+ii+2+n, len-2-n,
|
| + azArg[nArg], sizeof(azArg[nArg]));
|
| + }
|
| + for(j=nArg; j<MX_ARG; j++) azArg[j++][0] = 0;
|
| +
|
| + /*
|
| + ** --sleep N
|
| + **
|
| + ** Pause for N milliseconds
|
| + */
|
| + if( strcmp(zCmd, "sleep")==0 ){
|
| + sqlite3_sleep(atoi(azArg[0]));
|
| + }else
|
| +
|
| + /*
|
| + ** --exit N
|
| + **
|
| + ** Exit this process. If N>0 then exit without shutting down
|
| + ** SQLite. (In other words, simulate a crash.)
|
| + */
|
| + if( strcmp(zCmd, "exit")==0 ){
|
| + int rc = atoi(azArg[0]);
|
| + finishScript(iClient, taskId, 1);
|
| + if( rc==0 ) sqlite3_close(g.db);
|
| + exit(rc);
|
| + }else
|
| +
|
| + /*
|
| + ** --testcase NAME
|
| + **
|
| + ** Begin a new test case. Announce in the log that the test case
|
| + ** has begun.
|
| + */
|
| + if( strcmp(zCmd, "testcase")==0 ){
|
| + if( g.iTrace==1 ) logMessage("%.*s", len - 1, zScript+ii);
|
| + stringReset(&sResult);
|
| + }else
|
| +
|
| + /*
|
| + ** --finish
|
| + **
|
| + ** Mark the current task as having finished, even if it is not.
|
| + ** This can be used in conjunction with --exit to simulate a crash.
|
| + */
|
| + if( strcmp(zCmd, "finish")==0 && iClient>0 ){
|
| + finishScript(iClient, taskId, 1);
|
| + }else
|
| +
|
| + /*
|
| + ** --reset
|
| + **
|
| + ** Reset accumulated results back to an empty string
|
| + */
|
| + if( strcmp(zCmd, "reset")==0 ){
|
| + stringReset(&sResult);
|
| + }else
|
| +
|
| + /*
|
| + ** --match ANSWER...
|
| + **
|
| + ** Check to see if output matches ANSWER. Report an error if not.
|
| + */
|
| + if( strcmp(zCmd, "match")==0 ){
|
| + int jj;
|
| + char *zAns = zScript+ii;
|
| + for(jj=7; jj<len-1 && isspace(zAns[jj]); jj++){}
|
| + zAns += jj;
|
| + if( len-jj-1!=sResult.n || strncmp(sResult.z, zAns, len-jj-1) ){
|
| + errorMessage("line %d of %s:\nExpected [%.*s]\n Got [%s]",
|
| + prevLine, zFilename, len-jj-1, zAns, sResult.z);
|
| + }
|
| + g.nTest++;
|
| + stringReset(&sResult);
|
| + }else
|
| +
|
| + /*
|
| + ** --glob ANSWER...
|
| + ** --notglob ANSWER....
|
| + **
|
| + ** Check to see if output does or does not match the glob pattern
|
| + ** ANSWER.
|
| + */
|
| + if( strcmp(zCmd, "glob")==0 || strcmp(zCmd, "notglob")==0 ){
|
| + int jj;
|
| + char *zAns = zScript+ii;
|
| + char *zCopy;
|
| + int isGlob = (zCmd[0]=='g');
|
| + for(jj=9-3*isGlob; jj<len-1 && isspace(zAns[jj]); jj++){}
|
| + zAns += jj;
|
| + zCopy = sqlite3_mprintf("%.*s", len-jj-1, zAns);
|
| + if( (sqlite3_strglob(zCopy, sResult.z)==0)^isGlob ){
|
| + errorMessage("line %d of %s:\nExpected [%s]\n Got [%s]",
|
| + prevLine, zFilename, zCopy, sResult.z);
|
| + }
|
| + sqlite3_free(zCopy);
|
| + g.nTest++;
|
| + stringReset(&sResult);
|
| + }else
|
| +
|
| + /*
|
| + ** --output
|
| + **
|
| + ** Output the result of the previous SQL.
|
| + */
|
| + if( strcmp(zCmd, "output")==0 ){
|
| + logMessage("%s", sResult.z);
|
| + }else
|
| +
|
| + /*
|
| + ** --source FILENAME
|
| + **
|
| + ** Run a subscript from a separate file.
|
| + */
|
| + if( strcmp(zCmd, "source")==0 ){
|
| + char *zNewFile, *zNewScript;
|
| + char *zToDel = 0;
|
| + zNewFile = azArg[0];
|
| + if( zNewFile[0]!='/' ){
|
| + int k;
|
| + for(k=(int)strlen(zFilename)-1; k>=0 && zFilename[k]!='/'; k--){}
|
| + if( k>0 ){
|
| + zNewFile = zToDel = sqlite3_mprintf("%.*s/%s", k,zFilename,zNewFile);
|
| + }
|
| + }
|
| + zNewScript = readFile(zNewFile);
|
| + if( g.iTrace ) logMessage("begin script [%s]\n", zNewFile);
|
| + runScript(0, 0, zNewScript, zNewFile);
|
| + sqlite3_free(zNewScript);
|
| + if( g.iTrace ) logMessage("end script [%s]\n", zNewFile);
|
| + sqlite3_free(zToDel);
|
| + }else
|
| +
|
| + /*
|
| + ** --print MESSAGE....
|
| + **
|
| + ** Output the remainder of the line to the log file
|
| + */
|
| + if( strcmp(zCmd, "print")==0 ){
|
| + int jj;
|
| + for(jj=7; jj<len && isspace(zScript[ii+jj]); jj++){}
|
| + logMessage("%.*s", len-jj, zScript+ii+jj);
|
| + }else
|
| +
|
| + /*
|
| + ** --if EXPR
|
| + **
|
| + ** Skip forward to the next matching --endif or --else if EXPR is false.
|
| + */
|
| + if( strcmp(zCmd, "if")==0 ){
|
| + int jj, rc;
|
| + sqlite3_stmt *pStmt;
|
| + for(jj=4; jj<len && isspace(zScript[ii+jj]); jj++){}
|
| + pStmt = prepareSql("SELECT %.*s", len-jj, zScript+ii+jj);
|
| + rc = sqlite3_step(pStmt);
|
| + if( rc!=SQLITE_ROW || sqlite3_column_int(pStmt, 0)==0 ){
|
| + ii += findEndif(zScript+ii+len, 1, &lineno);
|
| + }
|
| + sqlite3_finalize(pStmt);
|
| + }else
|
| +
|
| + /*
|
| + ** --else
|
| + **
|
| + ** This command can only be encountered if currently inside an --if that
|
| + ** is true. Skip forward to the next matching --endif.
|
| + */
|
| + if( strcmp(zCmd, "else")==0 ){
|
| + ii += findEndif(zScript+ii+len, 0, &lineno);
|
| + }else
|
| +
|
| + /*
|
| + ** --endif
|
| + **
|
| + ** This command can only be encountered if currently inside an --if that
|
| + ** is true or an --else of a false if. This is a no-op.
|
| + */
|
| + if( strcmp(zCmd, "endif")==0 ){
|
| + /* no-op */
|
| + }else
|
| +
|
| + /*
|
| + ** --start CLIENT
|
| + **
|
| + ** Start up the given client.
|
| + */
|
| + if( strcmp(zCmd, "start")==0 && iClient==0 ){
|
| + int iNewClient = atoi(azArg[0]);
|
| + if( iNewClient>0 ){
|
| + startClient(iNewClient);
|
| + }
|
| + }else
|
| +
|
| + /*
|
| + ** --wait CLIENT TIMEOUT
|
| + **
|
| + ** Wait until all tasks complete for the given client. If CLIENT is
|
| + ** "all" then wait for all clients to complete. Wait no longer than
|
| + ** TIMEOUT milliseconds (default 10,000)
|
| + */
|
| + if( strcmp(zCmd, "wait")==0 && iClient==0 ){
|
| + int iTimeout = nArg>=2 ? atoi(azArg[1]) : 10000;
|
| + sqlite3_snprintf(sizeof(zError),zError,"line %d of %s\n",
|
| + prevLine, zFilename);
|
| + waitForClient(atoi(azArg[0]), iTimeout, zError);
|
| + }else
|
| +
|
| + /*
|
| + ** --task CLIENT
|
| + ** <task-content-here>
|
| + ** --end
|
| + **
|
| + ** Assign work to a client. Start the client if it is not running
|
| + ** already.
|
| + */
|
| + if( strcmp(zCmd, "task")==0 && iClient==0 ){
|
| + int iTarget = atoi(azArg[0]);
|
| + int iEnd;
|
| + char *zTask;
|
| + char *zTName;
|
| + iEnd = findEnd(zScript+ii+len, &lineno);
|
| + if( iTarget<0 ){
|
| + errorMessage("line %d of %s: bad client number: %d",
|
| + prevLine, zFilename, iTarget);
|
| + }else{
|
| + zTask = sqlite3_mprintf("%.*s", iEnd, zScript+ii+len);
|
| + if( nArg>1 ){
|
| + zTName = sqlite3_mprintf("%s", azArg[1]);
|
| + }else{
|
| + zTName = sqlite3_mprintf("%s:%d", filenameTail(zFilename), prevLine);
|
| + }
|
| + startClient(iTarget);
|
| + runSql("INSERT INTO task(client,script,name)"
|
| + " VALUES(%d,'%q',%Q)", iTarget, zTask, zTName);
|
| + sqlite3_free(zTask);
|
| + sqlite3_free(zTName);
|
| + }
|
| + iEnd += tokenLength(zScript+ii+len+iEnd, &lineno);
|
| + len += iEnd;
|
| + iBegin = ii+len;
|
| + }else
|
| +
|
| + /*
|
| + ** --breakpoint
|
| + **
|
| + ** This command calls "test_breakpoint()" which is a routine provided
|
| + ** as a convenient place to set a debugger breakpoint.
|
| + */
|
| + if( strcmp(zCmd, "breakpoint")==0 ){
|
| + test_breakpoint();
|
| + }else
|
| +
|
| + /*
|
| + ** --show-sql-errors BOOLEAN
|
| + **
|
| + ** Turn display of SQL errors on and off.
|
| + */
|
| + if( strcmp(zCmd, "show-sql-errors")==0 ){
|
| + g.bIgnoreSqlErrors = nArg>=1 ? !booleanValue(azArg[0]) : 1;
|
| + }else
|
| +
|
| +
|
| + /* error */{
|
| + errorMessage("line %d of %s: unknown command --%s",
|
| + prevLine, zFilename, zCmd);
|
| + }
|
| + ii += len;
|
| + }
|
| + if( iBegin<ii ){
|
| + char *zSql = sqlite3_mprintf("%.*s", ii-iBegin, zScript+iBegin);
|
| + runSql(zSql);
|
| + sqlite3_free(zSql);
|
| + }
|
| + stringFree(&sResult);
|
| +}
|
| +
|
| +/*
|
| +** Look for a command-line option. If present, return a pointer.
|
| +** Return NULL if missing.
|
| +**
|
| +** hasArg==0 means the option is a flag. It is either present or not.
|
| +** hasArg==1 means the option has an argument. Return a pointer to the
|
| +** argument.
|
| +*/
|
| +static char *findOption(
|
| + char **azArg,
|
| + int *pnArg,
|
| + const char *zOption,
|
| + int hasArg
|
| +){
|
| + int i, j;
|
| + char *zReturn = 0;
|
| + int nArg = *pnArg;
|
| +
|
| + assert( hasArg==0 || hasArg==1 );
|
| + for(i=0; i<nArg; i++){
|
| + const char *z;
|
| + if( i+hasArg >= nArg ) break;
|
| + z = azArg[i];
|
| + if( z[0]!='-' ) continue;
|
| + z++;
|
| + if( z[0]=='-' ){
|
| + if( z[1]==0 ) break;
|
| + z++;
|
| + }
|
| + if( strcmp(z,zOption)==0 ){
|
| + if( hasArg && i==nArg-1 ){
|
| + fatalError("command-line option \"--%s\" requires an argument", z);
|
| + }
|
| + if( hasArg ){
|
| + zReturn = azArg[i+1];
|
| + }else{
|
| + zReturn = azArg[i];
|
| + }
|
| + j = i+1+(hasArg!=0);
|
| + while( j<nArg ) azArg[i++] = azArg[j++];
|
| + *pnArg = i;
|
| + return zReturn;
|
| + }
|
| + }
|
| + return zReturn;
|
| +}
|
| +
|
| +/* Print a usage message for the program and exit */
|
| +static void usage(const char *argv0){
|
| + int i;
|
| + const char *zTail = argv0;
|
| + for(i=0; argv0[i]; i++){
|
| + if( argv0[i]=='/' ) zTail = argv0+i+1;
|
| + }
|
| + fprintf(stderr,"Usage: %s DATABASE ?OPTIONS? ?SCRIPT?\n", zTail);
|
| + exit(1);
|
| +}
|
| +
|
| +/* Report on unrecognized arguments */
|
| +static void unrecognizedArguments(
|
| + const char *argv0,
|
| + int nArg,
|
| + char **azArg
|
| +){
|
| + int i;
|
| + fprintf(stderr,"%s: unrecognized arguments:", argv0);
|
| + for(i=0; i<nArg; i++){
|
| + fprintf(stderr," %s", azArg[i]);
|
| + }
|
| + fprintf(stderr,"\n");
|
| + exit(1);
|
| +}
|
| +
|
| +int main(int argc, char **argv){
|
| + const char *zClient;
|
| + int iClient;
|
| + int n, i;
|
| + int openFlags = SQLITE_OPEN_READWRITE;
|
| + int rc;
|
| + char *zScript;
|
| + int taskId;
|
| + const char *zTrace;
|
| + const char *zCOption;
|
| +
|
| + g.argv0 = argv[0];
|
| + g.iTrace = 1;
|
| + if( argc<2 ) usage(argv[0]);
|
| + g.zDbFile = argv[1];
|
| + if( strglob("*.test", g.zDbFile) ) usage(argv[0]);
|
| + if( strcmp(sqlite3_sourceid(), SQLITE_SOURCE_ID)!=0 ){
|
| + fprintf(stderr, "SQLite library and header mismatch\n"
|
| + "Library: %s\n"
|
| + "Header: %s\n",
|
| + sqlite3_sourceid(), SQLITE_SOURCE_ID);
|
| + exit(1);
|
| + }
|
| + n = argc-2;
|
| + sqlite3_snprintf(sizeof(g.zName), g.zName, "%05d.mptest", GETPID());
|
| + g.zVfs = findOption(argv+2, &n, "vfs", 1);
|
| + zClient = findOption(argv+2, &n, "client", 1);
|
| + g.zErrLog = findOption(argv+2, &n, "errlog", 1);
|
| + g.zLog = findOption(argv+2, &n, "log", 1);
|
| + zTrace = findOption(argv+2, &n, "trace", 1);
|
| + if( zTrace ) g.iTrace = atoi(zTrace);
|
| + if( findOption(argv+2, &n, "quiet", 0)!=0 ) g.iTrace = 0;
|
| + g.bSqlTrace = findOption(argv+2, &n, "sqltrace", 0)!=0;
|
| + g.bSync = findOption(argv+2, &n, "sync", 0)!=0;
|
| + if( g.zErrLog ){
|
| + g.pErrLog = fopen(g.zErrLog, "a");
|
| + }else{
|
| + g.pErrLog = stderr;
|
| + }
|
| + if( g.zLog ){
|
| + g.pLog = fopen(g.zLog, "a");
|
| + }else{
|
| + g.pLog = stdout;
|
| + }
|
| +
|
| + sqlite3_config(SQLITE_CONFIG_LOG, sqlErrorCallback, 0);
|
| + if( zClient ){
|
| + iClient = atoi(zClient);
|
| + if( iClient<1 ) fatalError("illegal client number: %d\n", iClient);
|
| + sqlite3_snprintf(sizeof(g.zName), g.zName, "%05d.client%02d",
|
| + GETPID(), iClient);
|
| + }else{
|
| + if( g.iTrace>0 ){
|
| + printf("With SQLite " SQLITE_VERSION " " SQLITE_SOURCE_ID "\n" );
|
| + for(i=0; (zCOption = sqlite3_compileoption_get(i))!=0; i++){
|
| + printf("-DSQLITE_%s\n", zCOption);
|
| + }
|
| + fflush(stdout);
|
| + }
|
| + iClient = 0;
|
| + unlink(g.zDbFile);
|
| + openFlags |= SQLITE_OPEN_CREATE;
|
| + }
|
| + rc = sqlite3_open_v2(g.zDbFile, &g.db, openFlags, g.zVfs);
|
| + if( rc ) fatalError("cannot open [%s]", g.zDbFile);
|
| + sqlite3_enable_load_extension(g.db, 1);
|
| + sqlite3_busy_handler(g.db, busyHandler, 0);
|
| + sqlite3_create_function(g.db, "vfsname", 0, SQLITE_UTF8, 0,
|
| + vfsNameFunc, 0, 0);
|
| + sqlite3_create_function(g.db, "eval", 1, SQLITE_UTF8, 0,
|
| + evalFunc, 0, 0);
|
| + g.iTimeout = DEFAULT_TIMEOUT;
|
| + if( g.bSqlTrace ) sqlite3_trace(g.db, sqlTraceCallback, 0);
|
| + if( !g.bSync ) trySql("PRAGMA synchronous=OFF");
|
| + if( iClient>0 ){
|
| + if( n>0 ) unrecognizedArguments(argv[0], n, argv+2);
|
| + if( g.iTrace ) logMessage("start-client");
|
| + while(1){
|
| + char *zTaskName = 0;
|
| + rc = startScript(iClient, &zScript, &taskId, &zTaskName);
|
| + if( rc==SQLITE_DONE ) break;
|
| + if( g.iTrace ) logMessage("begin %s (%d)", zTaskName, taskId);
|
| + runScript(iClient, taskId, zScript, zTaskName);
|
| + if( g.iTrace ) logMessage("end %s (%d)", zTaskName, taskId);
|
| + finishScript(iClient, taskId, 0);
|
| + sqlite3_free(zTaskName);
|
| + sqlite3_sleep(10);
|
| + }
|
| + if( g.iTrace ) logMessage("end-client");
|
| + }else{
|
| + sqlite3_stmt *pStmt;
|
| + int iTimeout;
|
| + if( n==0 ){
|
| + fatalError("missing script filename");
|
| + }
|
| + if( n>1 ) unrecognizedArguments(argv[0], n, argv+2);
|
| + runSql(
|
| + "CREATE TABLE task(\n"
|
| + " id INTEGER PRIMARY KEY,\n"
|
| + " name TEXT,\n"
|
| + " client INTEGER,\n"
|
| + " starttime DATE,\n"
|
| + " endtime DATE,\n"
|
| + " script TEXT\n"
|
| + ");"
|
| + "CREATE INDEX task_i1 ON task(client, starttime);\n"
|
| + "CREATE INDEX task_i2 ON task(client, endtime);\n"
|
| + "CREATE TABLE counters(nError,nTest);\n"
|
| + "INSERT INTO counters VALUES(0,0);\n"
|
| + "CREATE TABLE client(id INTEGER PRIMARY KEY, wantHalt);\n"
|
| + );
|
| + zScript = readFile(argv[2]);
|
| + if( g.iTrace ) logMessage("begin script [%s]\n", argv[2]);
|
| + runScript(0, 0, zScript, argv[2]);
|
| + sqlite3_free(zScript);
|
| + if( g.iTrace ) logMessage("end script [%s]\n", argv[2]);
|
| + waitForClient(0, 2000, "during shutdown...\n");
|
| + trySql("UPDATE client SET wantHalt=1");
|
| + sqlite3_sleep(10);
|
| + g.iTimeout = 0;
|
| + iTimeout = 1000;
|
| + while( ((rc = trySql("SELECT 1 FROM client"))==SQLITE_BUSY
|
| + || rc==SQLITE_ROW) && iTimeout>0 ){
|
| + sqlite3_sleep(10);
|
| + iTimeout -= 10;
|
| + }
|
| + sqlite3_sleep(100);
|
| + pStmt = prepareSql("SELECT nError, nTest FROM counters");
|
| + iTimeout = 1000;
|
| + while( (rc = sqlite3_step(pStmt))==SQLITE_BUSY && iTimeout>0 ){
|
| + sqlite3_sleep(10);
|
| + iTimeout -= 10;
|
| + }
|
| + if( rc==SQLITE_ROW ){
|
| + g.nError += sqlite3_column_int(pStmt, 0);
|
| + g.nTest += sqlite3_column_int(pStmt, 1);
|
| + }
|
| + sqlite3_finalize(pStmt);
|
| + }
|
| + sqlite3_close(g.db);
|
| + maybeClose(g.pLog);
|
| + maybeClose(g.pErrLog);
|
| + if( iClient==0 ){
|
| + printf("Summary: %d errors in %d tests\n", g.nError, g.nTest);
|
| + }
|
| + return g.nError>0;
|
| +}
|
|
|