Import SQLite 3.10.2.

third_party/sqlite/src/mptest/mptest.c

 /*
 ** 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.
 **
@@ skipping 28 matching lines @@
 # 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>
 
+#define ISSPACE(X) isspace((unsigned char)(X))
+#define ISDIGIT(X) isdigit((unsigned char)(X))
+
 /* The suffix to append to the child command lines, if any */
 #if defined(_WIN32)
 # define GETPID (int)GetCurrentProcessId
 #else
 # define GETPID getpid
 #endif
 
+/* The directory separator character(s) */
+#if defined(_WIN32)
+# define isDirSep(c) (((c) == '/') || ((c) == '\\'))
+#else
+# define isDirSep(c) ((c) == '/')
+#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 */
@@ skipping 107 matching lines @@
         }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;
+      if( (z[0]=='-' || z[0]=='+') && ISDIGIT(z[1]) ) z++;
+      if( !ISDIGIT(z[0]) ) return 0;
       z++;
-      while( isdigit(z[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
 */
@@ skipping 78 matching lines @@
     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--; }
+  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
@@ skipping 109 matching lines @@
 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;
+    char *zNew = sqlite3_realloc(p->z, nAlloc);
+    if( zNew==0 ) fatalError("out of memory");
+    p->z = zNew;
     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++){}
+  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);
@@ skipping 234 matching lines @@
   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]=='*') ){
+  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( 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++){}
+    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]; }
+  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])) ){
+  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]))
+    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]) ){
+    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;
 }
 
 /*
@@ skipping 32 matching lines @@
     }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;
+  for(i=j=0; z[i]; i++) if( isDirSep(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++){}
@@ skipping 36 matching lines @@
   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]=='*') ){
+    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++; }
+      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
@@ skipping 46 matching lines @@
     }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++){}
+      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++){}
+      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]!='/' ){
+      if( !isDirSep(zNewFile[0]) ){
         int k;
-        for(k=(int)strlen(zFilename)-1; k>=0 && zFilename[k]!='/'; k--){}
+        for(k=(int)strlen(zFilename)-1; k>=0 && !isDirSep(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++){}
+      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++){}
+      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
@@ skipping 155 matching lines @@
     }
   }
   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;
+    if( isDirSep(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){
+int SQLITE_CDECL 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;
+  const char *zJMode;
+  const char *zNRep;
+  int nRep = 1, iRep;
 
   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());
+  zJMode = findOption(argv+2, &n, "journalmode", 1);
+  zNRep = findOption(argv+2, &n, "repeat", 1);
+  if( zNRep ) nRep = atoi(zNRep);
+  if( nRep<1 ) nRep = 1;
   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("BEGIN: %s", argv[0]);
+      for(i=1; i<argc; i++) printf(" %s", argv[i]);
+      printf("\n");
       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);
+  if( zJMode ){
+#if defined(_WIN32)
+    if( sqlite3_stricmp(zJMode,"persist")==0
+     || sqlite3_stricmp(zJMode,"truncate")==0
+    ){
+      printf("Changing journal mode to DELETE from %s", zJMode);
+      zJMode = "DELETE";
+    }
+#endif
+    runSql("PRAGMA journal_mode=%Q;", zJMode);
+  }
+  if( !g.bSync ) trySql("PRAGMA synchronous=OFF");
   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(
+      "DROP TABLE IF EXISTS task;\n"
+      "DROP TABLE IF EXISTS counters;\n"
+      "DROP TABLE IF EXISTS client;\n"
       "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]);
+    for(iRep=1; iRep<=nRep; iRep++){
+      if( g.iTrace ) logMessage("begin script [%s] cycle %d\n", argv[2], iRep);
+      runScript(0, 0, zScript, argv[2]);
+      if( g.iTrace ) logMessage("end script [%s] cycle %d\n", argv[2], iRep);
+    }
     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);  
+  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);
+    printf("Summary: %d errors out of %d tests\n", g.nError, g.nTest);
+    printf("END: %s", argv[0]);
+    for(i=1; i<argc; i++) printf(" %s", argv[i]);
+    printf("\n");
   }
   return g.nError>0;
 }