| OLD | NEW |
|---|
| (Empty) |
| 1 /* | |
| 2 ** 2013-04-05 | |
| 3 ** | |
| 4 ** The author disclaims copyright to this source code. In place of | |
| 5 ** a legal notice, here is a blessing: | |
| 6 ** | |
| 7 ** May you do good and not evil. | |
| 8 ** May you find forgiveness for yourself and forgive others. | |
| 9 ** May you share freely, never taking more than you give. | |
| 10 ** | |
| 11 ************************************************************************* | |
| 12 ** | |
| 13 ** This is a program used for testing SQLite, and specifically for testing | |
| 14 ** the ability of independent processes to access the same SQLite database | |
| 15 ** concurrently. | |
| 16 ** | |
| 17 ** Compile this program as follows: | |
| 18 ** | |
| 19 ** gcc -g -c -Wall sqlite3.c $(OPTS) | |
| 20 ** gcc -g -o mptest mptest.c sqlite3.o $(LIBS) | |
| 21 ** | |
| 22 ** Recommended options: | |
| 23 ** | |
| 24 ** -DHAVE_USLEEP | |
| 25 ** -DSQLITE_NO_SYNC | |
| 26 ** -DSQLITE_THREADSAFE=0 | |
| 27 ** -DSQLITE_OMIT_LOAD_EXTENSION | |
| 28 ** | |
| 29 ** Run like this: | |
| 30 ** | |
| 31 ** ./mptest $database $script | |
| 32 ** | |
| 33 ** where $database is the database to use for testing and $script is a | |
| 34 ** test script. | |
| 35 */ | |
| 36 #include "sqlite3.h" | |
| 37 #include <stdio.h> | |
| 38 #if defined(_WIN32) | |
| 39 # define WIN32_LEAN_AND_MEAN | |
| 40 # include <windows.h> | |
| 41 #else | |
| 42 # include <unistd.h> | |
| 43 #endif | |
| 44 #include <stdlib.h> | |
| 45 #include <string.h> | |
| 46 #include <assert.h> | |
| 47 #include <ctype.h> | |
| 48 | |
| 49 /* The suffix to append to the child command lines, if any */ | |
| 50 #if defined(_WIN32) | |
| 51 # define GETPID (int)GetCurrentProcessId | |
| 52 #else | |
| 53 # define GETPID getpid | |
| 54 #endif | |
| 55 | |
| 56 /* Mark a parameter as unused to suppress compiler warnings */ | |
| 57 #define UNUSED_PARAMETER(x) (void)x | |
| 58 | |
| 59 /* Global data | |
| 60 */ | |
| 61 static struct Global { | |
| 62 char *argv0; /* Name of the executable */ | |
| 63 const char *zVfs; /* Name of VFS to use. Often NULL meaning "default" */ | |
| 64 char *zDbFile; /* Name of the database */ | |
| 65 sqlite3 *db; /* Open connection to database */ | |
| 66 char *zErrLog; /* Filename for error log */ | |
| 67 FILE *pErrLog; /* Where to write errors */ | |
| 68 char *zLog; /* Name of output log file */ | |
| 69 FILE *pLog; /* Where to write log messages */ | |
| 70 char zName[32]; /* Symbolic name of this process */ | |
| 71 int taskId; /* Task ID. 0 means supervisor. */ | |
| 72 int iTrace; /* Tracing level */ | |
| 73 int bSqlTrace; /* True to trace SQL commands */ | |
| 74 int bIgnoreSqlErrors; /* Ignore errors in SQL statements */ | |
| 75 int nError; /* Number of errors */ | |
| 76 int nTest; /* Number of --match operators */ | |
| 77 int iTimeout; /* Milliseconds until a busy timeout */ | |
| 78 int bSync; /* Call fsync() */ | |
| 79 } g; | |
| 80 | |
| 81 /* Default timeout */ | |
| 82 #define DEFAULT_TIMEOUT 10000 | |
| 83 | |
| 84 /* | |
| 85 ** Print a message adding zPrefix[] to the beginning of every line. | |
| 86 */ | |
| 87 static void printWithPrefix(FILE *pOut, const char *zPrefix, const char *zMsg){ | |
| 88 while( zMsg && zMsg[0] ){ | |
| 89 int i; | |
| 90 for(i=0; zMsg[i] && zMsg[i]!='\n' && zMsg[i]!='\r'; i++){} | |
| 91 fprintf(pOut, "%s%.*s\n", zPrefix, i, zMsg); | |
| 92 zMsg += i; | |
| 93 while( zMsg[0]=='\n' || zMsg[0]=='\r' ) zMsg++; | |
| 94 } | |
| 95 } | |
| 96 | |
| 97 /* | |
| 98 ** Compare two pointers to strings, where the pointers might be NULL. | |
| 99 */ | |
| 100 static int safe_strcmp(const char *a, const char *b){ | |
| 101 if( a==b ) return 0; | |
| 102 if( a==0 ) return -1; | |
| 103 if( b==0 ) return 1; | |
| 104 return strcmp(a,b); | |
| 105 } | |
| 106 | |
| 107 /* | |
| 108 ** Return TRUE if string z[] matches glob pattern zGlob[]. | |
| 109 ** Return FALSE if the pattern does not match. | |
| 110 ** | |
| 111 ** Globbing rules: | |
| 112 ** | |
| 113 ** '*' Matches any sequence of zero or more characters. | |
| 114 ** | |
| 115 ** '?' Matches exactly one character. | |
| 116 ** | |
| 117 ** [...] Matches one character from the enclosed list of | |
| 118 ** characters. | |
| 119 ** | |
| 120 ** [^...] Matches one character not in the enclosed list. | |
| 121 ** | |
| 122 ** '#' Matches any sequence of one or more digits with an | |
| 123 ** optional + or - sign in front | |
| 124 */ | |
| 125 int strglob(const char *zGlob, const char *z){ | |
| 126 int c, c2; | |
| 127 int invert; | |
| 128 int seen; | |
| 129 | |
| 130 while( (c = (*(zGlob++)))!=0 ){ | |
| 131 if( c=='*' ){ | |
| 132 while( (c=(*(zGlob++))) == '*' || c=='?' ){ | |
| 133 if( c=='?' && (*(z++))==0 ) return 0; | |
| 134 } | |
| 135 if( c==0 ){ | |
| 136 return 1; | |
| 137 }else if( c=='[' ){ | |
| 138 while( *z && strglob(zGlob-1,z) ){ | |
| 139 z++; | |
| 140 } | |
| 141 return (*z)!=0; | |
| 142 } | |
| 143 while( (c2 = (*(z++)))!=0 ){ | |
| 144 while( c2!=c ){ | |
| 145 c2 = *(z++); | |
| 146 if( c2==0 ) return 0; | |
| 147 } | |
| 148 if( strglob(zGlob,z) ) return 1; | |
| 149 } | |
| 150 return 0; | |
| 151 }else if( c=='?' ){ | |
| 152 if( (*(z++))==0 ) return 0; | |
| 153 }else if( c=='[' ){ | |
| 154 int prior_c = 0; | |
| 155 seen = 0; | |
| 156 invert = 0; | |
| 157 c = *(z++); | |
| 158 if( c==0 ) return 0; | |
| 159 c2 = *(zGlob++); | |
| 160 if( c2=='^' ){ | |
| 161 invert = 1; | |
| 162 c2 = *(zGlob++); | |
| 163 } | |
| 164 if( c2==']' ){ | |
| 165 if( c==']' ) seen = 1; | |
| 166 c2 = *(zGlob++); | |
| 167 } | |
| 168 while( c2 && c2!=']' ){ | |
| 169 if( c2=='-' && zGlob[0]!=']' && zGlob[0]!=0 && prior_c>0 ){ | |
| 170 c2 = *(zGlob++); | |
| 171 if( c>=prior_c && c<=c2 ) seen = 1; | |
| 172 prior_c = 0; | |
| 173 }else{ | |
| 174 if( c==c2 ){ | |
| 175 seen = 1; | |
| 176 } | |
| 177 prior_c = c2; | |
| 178 } | |
| 179 c2 = *(zGlob++); | |
| 180 } | |
| 181 if( c2==0 || (seen ^ invert)==0 ) return 0; | |
| 182 }else if( c=='#' ){ | |
| 183 if( (z[0]=='-' || z[0]=='+') && isdigit(z[1]) ) z++; | |
| 184 if( !isdigit(z[0]) ) return 0; | |
| 185 z++; | |
| 186 while( isdigit(z[0]) ){ z++; } | |
| 187 }else{ | |
| 188 if( c!=(*(z++)) ) return 0; | |
| 189 } | |
| 190 } | |
| 191 return *z==0; | |
| 192 } | |
| 193 | |
| 194 /* | |
| 195 ** Close output stream pOut if it is not stdout or stderr | |
| 196 */ | |
| 197 static void maybeClose(FILE *pOut){ | |
| 198 if( pOut!=stdout && pOut!=stderr ) fclose(pOut); | |
| 199 } | |
| 200 | |
| 201 /* | |
| 202 ** Print an error message | |
| 203 */ | |
| 204 static void errorMessage(const char *zFormat, ...){ | |
| 205 va_list ap; | |
| 206 char *zMsg; | |
| 207 char zPrefix[30]; | |
| 208 va_start(ap, zFormat); | |
| 209 zMsg = sqlite3_vmprintf(zFormat, ap); | |
| 210 va_end(ap); | |
| 211 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s:ERROR: ", g.zName); | |
| 212 if( g.pLog ){ | |
| 213 printWithPrefix(g.pLog, zPrefix, zMsg); | |
| 214 fflush(g.pLog); | |
| 215 } | |
| 216 if( g.pErrLog && safe_strcmp(g.zErrLog,g.zLog) ){ | |
| 217 printWithPrefix(g.pErrLog, zPrefix, zMsg); | |
| 218 fflush(g.pErrLog); | |
| 219 } | |
| 220 sqlite3_free(zMsg); | |
| 221 g.nError++; | |
| 222 } | |
| 223 | |
| 224 /* Forward declaration */ | |
| 225 static int trySql(const char*, ...); | |
| 226 | |
| 227 /* | |
| 228 ** Print an error message and then quit. | |
| 229 */ | |
| 230 static void fatalError(const char *zFormat, ...){ | |
| 231 va_list ap; | |
| 232 char *zMsg; | |
| 233 char zPrefix[30]; | |
| 234 va_start(ap, zFormat); | |
| 235 zMsg = sqlite3_vmprintf(zFormat, ap); | |
| 236 va_end(ap); | |
| 237 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s:FATAL: ", g.zName); | |
| 238 if( g.pLog ){ | |
| 239 printWithPrefix(g.pLog, zPrefix, zMsg); | |
| 240 fflush(g.pLog); | |
| 241 maybeClose(g.pLog); | |
| 242 } | |
| 243 if( g.pErrLog && safe_strcmp(g.zErrLog,g.zLog) ){ | |
| 244 printWithPrefix(g.pErrLog, zPrefix, zMsg); | |
| 245 fflush(g.pErrLog); | |
| 246 maybeClose(g.pErrLog); | |
| 247 } | |
| 248 sqlite3_free(zMsg); | |
| 249 if( g.db ){ | |
| 250 int nTry = 0; | |
| 251 g.iTimeout = 0; | |
| 252 while( trySql("UPDATE client SET wantHalt=1;")==SQLITE_BUSY | |
| 253 && (nTry++)<100 ){ | |
| 254 sqlite3_sleep(10); | |
| 255 } | |
| 256 } | |
| 257 sqlite3_close(g.db); | |
| 258 exit(1); | |
| 259 } | |
| 260 | |
| 261 | |
| 262 /* | |
| 263 ** Print a log message | |
| 264 */ | |
| 265 static void logMessage(const char *zFormat, ...){ | |
| 266 va_list ap; | |
| 267 char *zMsg; | |
| 268 char zPrefix[30]; | |
| 269 va_start(ap, zFormat); | |
| 270 zMsg = sqlite3_vmprintf(zFormat, ap); | |
| 271 va_end(ap); | |
| 272 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s: ", g.zName); | |
| 273 if( g.pLog ){ | |
| 274 printWithPrefix(g.pLog, zPrefix, zMsg); | |
| 275 fflush(g.pLog); | |
| 276 } | |
| 277 sqlite3_free(zMsg); | |
| 278 } | |
| 279 | |
| 280 /* | |
| 281 ** Return the length of a string omitting trailing whitespace | |
| 282 */ | |
| 283 static int clipLength(const char *z){ | |
| 284 int n = (int)strlen(z); | |
| 285 while( n>0 && isspace(z[n-1]) ){ n--; } | |
| 286 return n; | |
| 287 } | |
| 288 | |
| 289 /* | |
| 290 ** Auxiliary SQL function to return the name of the VFS | |
| 291 */ | |
| 292 static void vfsNameFunc( | |
| 293 sqlite3_context *context, | |
| 294 int argc, | |
| 295 sqlite3_value **argv | |
| 296 ){ | |
| 297 sqlite3 *db = sqlite3_context_db_handle(context); | |
| 298 char *zVfs = 0; | |
| 299 UNUSED_PARAMETER(argc); | |
| 300 UNUSED_PARAMETER(argv); | |
| 301 sqlite3_file_control(db, "main", SQLITE_FCNTL_VFSNAME, &zVfs); | |
| 302 if( zVfs ){ | |
| 303 sqlite3_result_text(context, zVfs, -1, sqlite3_free); | |
| 304 } | |
| 305 } | |
| 306 | |
| 307 /* | |
| 308 ** Busy handler with a g.iTimeout-millisecond timeout | |
| 309 */ | |
| 310 static int busyHandler(void *pCD, int count){ | |
| 311 UNUSED_PARAMETER(pCD); | |
| 312 if( count*10>g.iTimeout ){ | |
| 313 if( g.iTimeout>0 ) errorMessage("timeout after %dms", g.iTimeout); | |
| 314 return 0; | |
| 315 } | |
| 316 sqlite3_sleep(10); | |
| 317 return 1; | |
| 318 } | |
| 319 | |
| 320 /* | |
| 321 ** SQL Trace callback | |
| 322 */ | |
| 323 static void sqlTraceCallback(void *NotUsed1, const char *zSql){ | |
| 324 UNUSED_PARAMETER(NotUsed1); | |
| 325 logMessage("[%.*s]", clipLength(zSql), zSql); | |
| 326 } | |
| 327 | |
| 328 /* | |
| 329 ** SQL error log callback | |
| 330 */ | |
| 331 static void sqlErrorCallback(void *pArg, int iErrCode, const char *zMsg){ | |
| 332 UNUSED_PARAMETER(pArg); | |
| 333 if( iErrCode==SQLITE_ERROR && g.bIgnoreSqlErrors ) return; | |
| 334 if( (iErrCode&0xff)==SQLITE_SCHEMA && g.iTrace<3 ) return; | |
| 335 if( g.iTimeout==0 && (iErrCode&0xff)==SQLITE_BUSY && g.iTrace<3 ) return; | |
| 336 if( (iErrCode&0xff)==SQLITE_NOTICE ){ | |
| 337 logMessage("(info) %s", zMsg); | |
| 338 }else{ | |
| 339 errorMessage("(errcode=%d) %s", iErrCode, zMsg); | |
| 340 } | |
| 341 } | |
| 342 | |
| 343 /* | |
| 344 ** Prepare an SQL statement. Issue a fatal error if unable. | |
| 345 */ | |
| 346 static sqlite3_stmt *prepareSql(const char *zFormat, ...){ | |
| 347 va_list ap; | |
| 348 char *zSql; | |
| 349 int rc; | |
| 350 sqlite3_stmt *pStmt = 0; | |
| 351 va_start(ap, zFormat); | |
| 352 zSql = sqlite3_vmprintf(zFormat, ap); | |
| 353 va_end(ap); | |
| 354 rc = sqlite3_prepare_v2(g.db, zSql, -1, &pStmt, 0); | |
| 355 if( rc!=SQLITE_OK ){ | |
| 356 sqlite3_finalize(pStmt); | |
| 357 fatalError("%s\n%s\n", sqlite3_errmsg(g.db), zSql); | |
| 358 } | |
| 359 sqlite3_free(zSql); | |
| 360 return pStmt; | |
| 361 } | |
| 362 | |
| 363 /* | |
| 364 ** Run arbitrary SQL. Issue a fatal error on failure. | |
| 365 */ | |
| 366 static void runSql(const char *zFormat, ...){ | |
| 367 va_list ap; | |
| 368 char *zSql; | |
| 369 int rc; | |
| 370 va_start(ap, zFormat); | |
| 371 zSql = sqlite3_vmprintf(zFormat, ap); | |
| 372 va_end(ap); | |
| 373 rc = sqlite3_exec(g.db, zSql, 0, 0, 0); | |
| 374 if( rc!=SQLITE_OK ){ | |
| 375 fatalError("%s\n%s\n", sqlite3_errmsg(g.db), zSql); | |
| 376 } | |
| 377 sqlite3_free(zSql); | |
| 378 } | |
| 379 | |
| 380 /* | |
| 381 ** Try to run arbitrary SQL. Return success code. | |
| 382 */ | |
| 383 static int trySql(const char *zFormat, ...){ | |
| 384 va_list ap; | |
| 385 char *zSql; | |
| 386 int rc; | |
| 387 va_start(ap, zFormat); | |
| 388 zSql = sqlite3_vmprintf(zFormat, ap); | |
| 389 va_end(ap); | |
| 390 rc = sqlite3_exec(g.db, zSql, 0, 0, 0); | |
| 391 sqlite3_free(zSql); | |
| 392 return rc; | |
| 393 } | |
| 394 | |
| 395 /* Structure for holding an arbitrary length string | |
| 396 */ | |
| 397 typedef struct String String; | |
| 398 struct String { | |
| 399 char *z; /* the string */ | |
| 400 int n; /* Slots of z[] used */ | |
| 401 int nAlloc; /* Slots of z[] allocated */ | |
| 402 }; | |
| 403 | |
| 404 /* Free a string */ | |
| 405 static void stringFree(String *p){ | |
| 406 if( p->z ) sqlite3_free(p->z); | |
| 407 memset(p, 0, sizeof(*p)); | |
| 408 } | |
| 409 | |
| 410 /* Append n bytes of text to a string. If n<0 append the entire string. */ | |
| 411 static void stringAppend(String *p, const char *z, int n){ | |
| 412 if( n<0 ) n = (int)strlen(z); | |
| 413 if( p->n+n>=p->nAlloc ){ | |
| 414 int nAlloc = p->nAlloc*2 + n + 100; | |
| 415 char *z = sqlite3_realloc(p->z, nAlloc); | |
| 416 if( z==0 ) fatalError("out of memory"); | |
| 417 p->z = z; | |
| 418 p->nAlloc = nAlloc; | |
| 419 } | |
| 420 memcpy(p->z+p->n, z, n); | |
| 421 p->n += n; | |
| 422 p->z[p->n] = 0; | |
| 423 } | |
| 424 | |
| 425 /* Reset a string to an empty string */ | |
| 426 static void stringReset(String *p){ | |
| 427 if( p->z==0 ) stringAppend(p, " ", 1); | |
| 428 p->n = 0; | |
| 429 p->z[0] = 0; | |
| 430 } | |
| 431 | |
| 432 /* Append a new token onto the end of the string */ | |
| 433 static void stringAppendTerm(String *p, const char *z){ | |
| 434 int i; | |
| 435 if( p->n ) stringAppend(p, " ", 1); | |
| 436 if( z==0 ){ | |
| 437 stringAppend(p, "nil", 3); | |
| 438 return; | |
| 439 } | |
| 440 for(i=0; z[i] && !isspace(z[i]); i++){} | |
| 441 if( i>0 && z[i]==0 ){ | |
| 442 stringAppend(p, z, i); | |
| 443 return; | |
| 444 } | |
| 445 stringAppend(p, "'", 1); | |
| 446 while( z[0] ){ | |
| 447 for(i=0; z[i] && z[i]!='\''; i++){} | |
| 448 if( z[i] ){ | |
| 449 stringAppend(p, z, i+1); | |
| 450 stringAppend(p, "'", 1); | |
| 451 z += i+1; | |
| 452 }else{ | |
| 453 stringAppend(p, z, i); | |
| 454 break; | |
| 455 } | |
| 456 } | |
| 457 stringAppend(p, "'", 1); | |
| 458 } | |
| 459 | |
| 460 /* | |
| 461 ** Callback function for evalSql() | |
| 462 */ | |
| 463 static int evalCallback(void *pCData, int argc, char **argv, char **azCol){ | |
| 464 String *p = (String*)pCData; | |
| 465 int i; | |
| 466 UNUSED_PARAMETER(azCol); | |
| 467 for(i=0; i<argc; i++) stringAppendTerm(p, argv[i]); | |
| 468 return 0; | |
| 469 } | |
| 470 | |
| 471 /* | |
| 472 ** Run arbitrary SQL and record the results in an output string | |
| 473 ** given by the first parameter. | |
| 474 */ | |
| 475 static int evalSql(String *p, const char *zFormat, ...){ | |
| 476 va_list ap; | |
| 477 char *zSql; | |
| 478 int rc; | |
| 479 char *zErrMsg = 0; | |
| 480 va_start(ap, zFormat); | |
| 481 zSql = sqlite3_vmprintf(zFormat, ap); | |
| 482 va_end(ap); | |
| 483 assert( g.iTimeout>0 ); | |
| 484 rc = sqlite3_exec(g.db, zSql, evalCallback, p, &zErrMsg); | |
| 485 sqlite3_free(zSql); | |
| 486 if( rc ){ | |
| 487 char zErr[30]; | |
| 488 sqlite3_snprintf(sizeof(zErr), zErr, "error(%d)", rc); | |
| 489 stringAppendTerm(p, zErr); | |
| 490 if( zErrMsg ){ | |
| 491 stringAppendTerm(p, zErrMsg); | |
| 492 sqlite3_free(zErrMsg); | |
| 493 } | |
| 494 } | |
| 495 return rc; | |
| 496 } | |
| 497 | |
| 498 /* | |
| 499 ** Auxiliary SQL function to recursively evaluate SQL. | |
| 500 */ | |
| 501 static void evalFunc( | |
| 502 sqlite3_context *context, | |
| 503 int argc, | |
| 504 sqlite3_value **argv | |
| 505 ){ | |
| 506 sqlite3 *db = sqlite3_context_db_handle(context); | |
| 507 const char *zSql = (const char*)sqlite3_value_text(argv[0]); | |
| 508 String res; | |
| 509 char *zErrMsg = 0; | |
| 510 int rc; | |
| 511 UNUSED_PARAMETER(argc); | |
| 512 memset(&res, 0, sizeof(res)); | |
| 513 rc = sqlite3_exec(db, zSql, evalCallback, &res, &zErrMsg); | |
| 514 if( zErrMsg ){ | |
| 515 sqlite3_result_error(context, zErrMsg, -1); | |
| 516 sqlite3_free(zErrMsg); | |
| 517 }else if( rc ){ | |
| 518 sqlite3_result_error_code(context, rc); | |
| 519 }else{ | |
| 520 sqlite3_result_text(context, res.z, -1, SQLITE_TRANSIENT); | |
| 521 } | |
| 522 stringFree(&res); | |
| 523 } | |
| 524 | |
| 525 /* | |
| 526 ** Look up the next task for client iClient in the database. | |
| 527 ** Return the task script and the task number and mark that | |
| 528 ** task as being under way. | |
| 529 */ | |
| 530 static int startScript( | |
| 531 int iClient, /* The client number */ | |
| 532 char **pzScript, /* Write task script here */ | |
| 533 int *pTaskId, /* Write task number here */ | |
| 534 char **pzTaskName /* Name of the task */ | |
| 535 ){ | |
| 536 sqlite3_stmt *pStmt = 0; | |
| 537 int taskId; | |
| 538 int rc; | |
| 539 int totalTime = 0; | |
| 540 | |
| 541 *pzScript = 0; | |
| 542 g.iTimeout = 0; | |
| 543 while(1){ | |
| 544 rc = trySql("BEGIN IMMEDIATE"); | |
| 545 if( rc==SQLITE_BUSY ){ | |
| 546 sqlite3_sleep(10); | |
| 547 totalTime += 10; | |
| 548 continue; | |
| 549 } | |
| 550 if( rc!=SQLITE_OK ){ | |
| 551 fatalError("in startScript: %s", sqlite3_errmsg(g.db)); | |
| 552 } | |
| 553 if( g.nError || g.nTest ){ | |
| 554 runSql("UPDATE counters SET nError=nError+%d, nTest=nTest+%d", | |
| 555 g.nError, g.nTest); | |
| 556 g.nError = 0; | |
| 557 g.nTest = 0; | |
| 558 } | |
| 559 pStmt = prepareSql("SELECT 1 FROM client WHERE id=%d AND wantHalt",iClient); | |
| 560 rc = sqlite3_step(pStmt); | |
| 561 sqlite3_finalize(pStmt); | |
| 562 if( rc==SQLITE_ROW ){ | |
| 563 runSql("DELETE FROM client WHERE id=%d", iClient); | |
| 564 g.iTimeout = DEFAULT_TIMEOUT; | |
| 565 runSql("COMMIT TRANSACTION;"); | |
| 566 return SQLITE_DONE; | |
| 567 } | |
| 568 pStmt = prepareSql( | |
| 569 "SELECT script, id, name FROM task" | |
| 570 " WHERE client=%d AND starttime IS NULL" | |
| 571 " ORDER BY id LIMIT 1", iClient); | |
| 572 rc = sqlite3_step(pStmt); | |
| 573 if( rc==SQLITE_ROW ){ | |
| 574 int n = sqlite3_column_bytes(pStmt, 0); | |
| 575 *pzScript = sqlite3_malloc(n+1); | |
| 576 strcpy(*pzScript, (const char*)sqlite3_column_text(pStmt, 0)); | |
| 577 *pTaskId = taskId = sqlite3_column_int(pStmt, 1); | |
| 578 *pzTaskName = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 2)); | |
| 579 sqlite3_finalize(pStmt); | |
| 580 runSql("UPDATE task" | |
| 581 " SET starttime=strftime('%%Y-%%m-%%d %%H:%%M:%%f','now')" | |
| 582 " WHERE id=%d;", taskId); | |
| 583 g.iTimeout = DEFAULT_TIMEOUT; | |
| 584 runSql("COMMIT TRANSACTION;"); | |
| 585 return SQLITE_OK; | |
| 586 } | |
| 587 sqlite3_finalize(pStmt); | |
| 588 if( rc==SQLITE_DONE ){ | |
| 589 if( totalTime>30000 ){ | |
| 590 errorMessage("Waited over 30 seconds with no work. Giving up."); | |
| 591 runSql("DELETE FROM client WHERE id=%d; COMMIT;", iClient); | |
| 592 sqlite3_close(g.db); | |
| 593 exit(1); | |
| 594 } | |
| 595 while( trySql("COMMIT")==SQLITE_BUSY ){ | |
| 596 sqlite3_sleep(10); | |
| 597 totalTime += 10; | |
| 598 } | |
| 599 sqlite3_sleep(100); | |
| 600 totalTime += 100; | |
| 601 continue; | |
| 602 } | |
| 603 fatalError("%s", sqlite3_errmsg(g.db)); | |
| 604 } | |
| 605 g.iTimeout = DEFAULT_TIMEOUT; | |
| 606 } | |
| 607 | |
| 608 /* | |
| 609 ** Mark a script as having finished. Remove the CLIENT table entry | |
| 610 ** if bShutdown is true. | |
| 611 */ | |
| 612 static int finishScript(int iClient, int taskId, int bShutdown){ | |
| 613 runSql("UPDATE task" | |
| 614 " SET endtime=strftime('%%Y-%%m-%%d %%H:%%M:%%f','now')" | |
| 615 " WHERE id=%d;", taskId); | |
| 616 if( bShutdown ){ | |
| 617 runSql("DELETE FROM client WHERE id=%d", iClient); | |
| 618 } | |
| 619 return SQLITE_OK; | |
| 620 } | |
| 621 | |
| 622 /* | |
| 623 ** Start up a client process for iClient, if it is not already | |
| 624 ** running. If the client is already running, then this routine | |
| 625 ** is a no-op. | |
| 626 */ | |
| 627 static void startClient(int iClient){ | |
| 628 runSql("INSERT OR IGNORE INTO client VALUES(%d,0)", iClient); | |
| 629 if( sqlite3_changes(g.db) ){ | |
| 630 char *zSys; | |
| 631 int rc; | |
| 632 zSys = sqlite3_mprintf("%s \"%s\" --client %d --trace %d", | |
| 633 g.argv0, g.zDbFile, iClient, g.iTrace); | |
| 634 if( g.bSqlTrace ){ | |
| 635 zSys = sqlite3_mprintf("%z --sqltrace", zSys); | |
| 636 } | |
| 637 if( g.bSync ){ | |
| 638 zSys = sqlite3_mprintf("%z --sync", zSys); | |
| 639 } | |
| 640 if( g.zVfs ){ | |
| 641 zSys = sqlite3_mprintf("%z --vfs \"%s\"", zSys, g.zVfs); | |
| 642 } | |
| 643 if( g.iTrace>=2 ) logMessage("system('%q')", zSys); | |
| 644 #if !defined(_WIN32) | |
| 645 zSys = sqlite3_mprintf("%z &", zSys); | |
| 646 rc = system(zSys); | |
| 647 if( rc ) errorMessage("system() fails with error code %d", rc); | |
| 648 #else | |
| 649 { | |
| 650 STARTUPINFOA startupInfo; | |
| 651 PROCESS_INFORMATION processInfo; | |
| 652 memset(&startupInfo, 0, sizeof(startupInfo)); | |
| 653 startupInfo.cb = sizeof(startupInfo); | |
| 654 memset(&processInfo, 0, sizeof(processInfo)); | |
| 655 rc = CreateProcessA(NULL, zSys, NULL, NULL, FALSE, 0, NULL, NULL, | |
| 656 &startupInfo, &processInfo); | |
| 657 if( rc ){ | |
| 658 CloseHandle(processInfo.hThread); | |
| 659 CloseHandle(processInfo.hProcess); | |
| 660 }else{ | |
| 661 errorMessage("CreateProcessA() fails with error code %lu", | |
| 662 GetLastError()); | |
| 663 } | |
| 664 } | |
| 665 #endif | |
| 666 sqlite3_free(zSys); | |
| 667 } | |
| 668 } | |
| 669 | |
| 670 /* | |
| 671 ** Read the entire content of a file into memory | |
| 672 */ | |
| 673 static char *readFile(const char *zFilename){ | |
| 674 FILE *in = fopen(zFilename, "rb"); | |
| 675 long sz; | |
| 676 char *z; | |
| 677 if( in==0 ){ | |
| 678 fatalError("cannot open \"%s\" for reading", zFilename); | |
| 679 } | |
| 680 fseek(in, 0, SEEK_END); | |
| 681 sz = ftell(in); | |
| 682 rewind(in); | |
| 683 z = sqlite3_malloc( sz+1 ); | |
| 684 sz = (long)fread(z, 1, sz, in); | |
| 685 z[sz] = 0; | |
| 686 fclose(in); | |
| 687 return z; | |
| 688 } | |
| 689 | |
| 690 /* | |
| 691 ** Return the length of the next token. | |
| 692 */ | |
| 693 static int tokenLength(const char *z, int *pnLine){ | |
| 694 int n = 0; | |
| 695 if( isspace(z[0]) || (z[0]=='/' && z[1]=='*') ){ | |
| 696 int inC = 0; | |
| 697 int c; | |
| 698 if( z[0]=='/' ){ | |
| 699 inC = 1; | |
| 700 n = 2; | |
| 701 } | |
| 702 while( (c = z[n++])!=0 ){ | |
| 703 if( c=='\n' ) (*pnLine)++; | |
| 704 if( isspace(c) ) continue; | |
| 705 if( inC && c=='*' && z[n]=='/' ){ | |
| 706 n++; | |
| 707 inC = 0; | |
| 708 }else if( !inC && c=='/' && z[n]=='*' ){ | |
| 709 n++; | |
| 710 inC = 1; | |
| 711 }else if( !inC ){ | |
| 712 break; | |
| 713 } | |
| 714 } | |
| 715 n--; | |
| 716 }else if( z[0]=='-' && z[1]=='-' ){ | |
| 717 for(n=2; z[n] && z[n]!='\n'; n++){} | |
| 718 if( z[n] ){ (*pnLine)++; n++; } | |
| 719 }else if( z[0]=='"' || z[0]=='\'' ){ | |
| 720 int delim = z[0]; | |
| 721 for(n=1; z[n]; n++){ | |
| 722 if( z[n]=='\n' ) (*pnLine)++; | |
| 723 if( z[n]==delim ){ | |
| 724 n++; | |
| 725 if( z[n+1]!=delim ) break; | |
| 726 } | |
| 727 } | |
| 728 }else{ | |
| 729 int c; | |
| 730 for(n=1; (c = z[n])!=0 && !isspace(c) && c!='"' && c!='\'' && c!=';'; n++){} | |
| 731 } | |
| 732 return n; | |
| 733 } | |
| 734 | |
| 735 /* | |
| 736 ** Copy a single token into a string buffer. | |
| 737 */ | |
| 738 static int extractToken(const char *zIn, int nIn, char *zOut, int nOut){ | |
| 739 int i; | |
| 740 if( nIn<=0 ){ | |
| 741 zOut[0] = 0; | |
| 742 return 0; | |
| 743 } | |
| 744 for(i=0; i<nIn && i<nOut-1 && !isspace(zIn[i]); i++){ zOut[i] = zIn[i]; } | |
| 745 zOut[i] = 0; | |
| 746 return i; | |
| 747 } | |
| 748 | |
| 749 /* | |
| 750 ** Find the number of characters up to the start of the next "--end" token. | |
| 751 */ | |
| 752 static int findEnd(const char *z, int *pnLine){ | |
| 753 int n = 0; | |
| 754 while( z[n] && (strncmp(z+n,"--end",5) || !isspace(z[n+5])) ){ | |
| 755 n += tokenLength(z+n, pnLine); | |
| 756 } | |
| 757 return n; | |
| 758 } | |
| 759 | |
| 760 /* | |
| 761 ** Find the number of characters up to the first character past the | |
| 762 ** of the next "--endif" or "--else" token. Nested --if commands are | |
| 763 ** also skipped. | |
| 764 */ | |
| 765 static int findEndif(const char *z, int stopAtElse, int *pnLine){ | |
| 766 int n = 0; | |
| 767 while( z[n] ){ | |
| 768 int len = tokenLength(z+n, pnLine); | |
| 769 if( (strncmp(z+n,"--endif",7)==0 && isspace(z[n+7])) | |
| 770 || (stopAtElse && strncmp(z+n,"--else",6)==0 && isspace(z[n+6])) | |
| 771 ){ | |
| 772 return n+len; | |
| 773 } | |
| 774 if( strncmp(z+n,"--if",4)==0 && isspace(z[n+4]) ){ | |
| 775 int skip = findEndif(z+n+len, 0, pnLine); | |
| 776 n += skip + len; | |
| 777 }else{ | |
| 778 n += len; | |
| 779 } | |
| 780 } | |
| 781 return n; | |
| 782 } | |
| 783 | |
| 784 /* | |
| 785 ** Wait for a client process to complete all its tasks | |
| 786 */ | |
| 787 static void waitForClient(int iClient, int iTimeout, char *zErrPrefix){ | |
| 788 sqlite3_stmt *pStmt; | |
| 789 int rc; | |
| 790 if( iClient>0 ){ | |
| 791 pStmt = prepareSql( | |
| 792 "SELECT 1 FROM task" | |
| 793 " WHERE client=%d" | |
| 794 " AND client IN (SELECT id FROM client)" | |
| 795 " AND endtime IS NULL", | |
| 796 iClient); | |
| 797 }else{ | |
| 798 pStmt = prepareSql( | |
| 799 "SELECT 1 FROM task" | |
| 800 " WHERE client IN (SELECT id FROM client)" | |
| 801 " AND endtime IS NULL"); | |
| 802 } | |
| 803 g.iTimeout = 0; | |
| 804 while( ((rc = sqlite3_step(pStmt))==SQLITE_BUSY || rc==SQLITE_ROW) | |
| 805 && iTimeout>0 | |
| 806 ){ | |
| 807 sqlite3_reset(pStmt); | |
| 808 sqlite3_sleep(50); | |
| 809 iTimeout -= 50; | |
| 810 } | |
| 811 sqlite3_finalize(pStmt); | |
| 812 g.iTimeout = DEFAULT_TIMEOUT; | |
| 813 if( rc!=SQLITE_DONE ){ | |
| 814 if( zErrPrefix==0 ) zErrPrefix = ""; | |
| 815 if( iClient>0 ){ | |
| 816 errorMessage("%stimeout waiting for client %d", zErrPrefix, iClient); | |
| 817 }else{ | |
| 818 errorMessage("%stimeout waiting for all clients", zErrPrefix); | |
| 819 } | |
| 820 } | |
| 821 } | |
| 822 | |
| 823 /* Return a pointer to the tail of a filename | |
| 824 */ | |
| 825 static char *filenameTail(char *z){ | |
| 826 int i, j; | |
| 827 for(i=j=0; z[i]; i++) if( z[i]=='/' ) j = i+1; | |
| 828 return z+j; | |
| 829 } | |
| 830 | |
| 831 /* | |
| 832 ** Interpret zArg as a boolean value. Return either 0 or 1. | |
| 833 */ | |
| 834 static int booleanValue(char *zArg){ | |
| 835 int i; | |
| 836 if( zArg==0 ) return 0; | |
| 837 for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){} | |
| 838 if( i>0 && zArg[i]==0 ) return atoi(zArg); | |
| 839 if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){ | |
| 840 return 1; | |
| 841 } | |
| 842 if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){ | |
| 843 return 0; | |
| 844 } | |
| 845 errorMessage("unknown boolean: [%s]", zArg); | |
| 846 return 0; | |
| 847 } | |
| 848 | |
| 849 | |
| 850 /* This routine exists as a convenient place to set a debugger | |
| 851 ** breakpoint. | |
| 852 */ | |
| 853 static void test_breakpoint(void){ static volatile int cnt = 0; cnt++; } | |
| 854 | |
| 855 /* Maximum number of arguments to a --command */ | |
| 856 #define MX_ARG 2 | |
| 857 | |
| 858 /* | |
| 859 ** Run a script. | |
| 860 */ | |
| 861 static void runScript( | |
| 862 int iClient, /* The client number, or 0 for the master */ | |
| 863 int taskId, /* The task ID for clients. 0 for master */ | |
| 864 char *zScript, /* Text of the script */ | |
| 865 char *zFilename /* File from which script was read. */ | |
| 866 ){ | |
| 867 int lineno = 1; | |
| 868 int prevLine = 1; | |
| 869 int ii = 0; | |
| 870 int iBegin = 0; | |
| 871 int n, c, j; | |
| 872 int len; | |
| 873 int nArg; | |
| 874 String sResult; | |
| 875 char zCmd[30]; | |
| 876 char zError[1000]; | |
| 877 char azArg[MX_ARG][100]; | |
| 878 | |
| 879 memset(&sResult, 0, sizeof(sResult)); | |
| 880 stringReset(&sResult); | |
| 881 while( (c = zScript[ii])!=0 ){ | |
| 882 prevLine = lineno; | |
| 883 len = tokenLength(zScript+ii, &lineno); | |
| 884 if( isspace(c) || (c=='/' && zScript[ii+1]=='*') ){ | |
| 885 ii += len; | |
| 886 continue; | |
| 887 } | |
| 888 if( c!='-' || zScript[ii+1]!='-' || !isalpha(zScript[ii+2]) ){ | |
| 889 ii += len; | |
| 890 continue; | |
| 891 } | |
| 892 | |
| 893 /* Run any prior SQL before processing the new --command */ | |
| 894 if( ii>iBegin ){ | |
| 895 char *zSql = sqlite3_mprintf("%.*s", ii-iBegin, zScript+iBegin); | |
| 896 evalSql(&sResult, zSql); | |
| 897 sqlite3_free(zSql); | |
| 898 iBegin = ii + len; | |
| 899 } | |
| 900 | |
| 901 /* Parse the --command */ | |
| 902 if( g.iTrace>=2 ) logMessage("%.*s", len, zScript+ii); | |
| 903 n = extractToken(zScript+ii+2, len-2, zCmd, sizeof(zCmd)); | |
| 904 for(nArg=0; n<len-2 && nArg<MX_ARG; nArg++){ | |
| 905 while( n<len-2 && isspace(zScript[ii+2+n]) ){ n++; } | |
| 906 if( n>=len-2 ) break; | |
| 907 n += extractToken(zScript+ii+2+n, len-2-n, | |
| 908 azArg[nArg], sizeof(azArg[nArg])); | |
| 909 } | |
| 910 for(j=nArg; j<MX_ARG; j++) azArg[j++][0] = 0; | |
| 911 | |
| 912 /* | |
| 913 ** --sleep N | |
| 914 ** | |
| 915 ** Pause for N milliseconds | |
| 916 */ | |
| 917 if( strcmp(zCmd, "sleep")==0 ){ | |
| 918 sqlite3_sleep(atoi(azArg[0])); | |
| 919 }else | |
| 920 | |
| 921 /* | |
| 922 ** --exit N | |
| 923 ** | |
| 924 ** Exit this process. If N>0 then exit without shutting down | |
| 925 ** SQLite. (In other words, simulate a crash.) | |
| 926 */ | |
| 927 if( strcmp(zCmd, "exit")==0 ){ | |
| 928 int rc = atoi(azArg[0]); | |
| 929 finishScript(iClient, taskId, 1); | |
| 930 if( rc==0 ) sqlite3_close(g.db); | |
| 931 exit(rc); | |
| 932 }else | |
| 933 | |
| 934 /* | |
| 935 ** --testcase NAME | |
| 936 ** | |
| 937 ** Begin a new test case. Announce in the log that the test case | |
| 938 ** has begun. | |
| 939 */ | |
| 940 if( strcmp(zCmd, "testcase")==0 ){ | |
| 941 if( g.iTrace==1 ) logMessage("%.*s", len - 1, zScript+ii); | |
| 942 stringReset(&sResult); | |
| 943 }else | |
| 944 | |
| 945 /* | |
| 946 ** --finish | |
| 947 ** | |
| 948 ** Mark the current task as having finished, even if it is not. | |
| 949 ** This can be used in conjunction with --exit to simulate a crash. | |
| 950 */ | |
| 951 if( strcmp(zCmd, "finish")==0 && iClient>0 ){ | |
| 952 finishScript(iClient, taskId, 1); | |
| 953 }else | |
| 954 | |
| 955 /* | |
| 956 ** --reset | |
| 957 ** | |
| 958 ** Reset accumulated results back to an empty string | |
| 959 */ | |
| 960 if( strcmp(zCmd, "reset")==0 ){ | |
| 961 stringReset(&sResult); | |
| 962 }else | |
| 963 | |
| 964 /* | |
| 965 ** --match ANSWER... | |
| 966 ** | |
| 967 ** Check to see if output matches ANSWER. Report an error if not. | |
| 968 */ | |
| 969 if( strcmp(zCmd, "match")==0 ){ | |
| 970 int jj; | |
| 971 char *zAns = zScript+ii; | |
| 972 for(jj=7; jj<len-1 && isspace(zAns[jj]); jj++){} | |
| 973 zAns += jj; | |
| 974 if( len-jj-1!=sResult.n || strncmp(sResult.z, zAns, len-jj-1) ){ | |
| 975 errorMessage("line %d of %s:\nExpected [%.*s]\n Got [%s]", | |
| 976 prevLine, zFilename, len-jj-1, zAns, sResult.z); | |
| 977 } | |
| 978 g.nTest++; | |
| 979 stringReset(&sResult); | |
| 980 }else | |
| 981 | |
| 982 /* | |
| 983 ** --glob ANSWER... | |
| 984 ** --notglob ANSWER.... | |
| 985 ** | |
| 986 ** Check to see if output does or does not match the glob pattern | |
| 987 ** ANSWER. | |
| 988 */ | |
| 989 if( strcmp(zCmd, "glob")==0 || strcmp(zCmd, "notglob")==0 ){ | |
| 990 int jj; | |
| 991 char *zAns = zScript+ii; | |
| 992 char *zCopy; | |
| 993 int isGlob = (zCmd[0]=='g'); | |
| 994 for(jj=9-3*isGlob; jj<len-1 && isspace(zAns[jj]); jj++){} | |
| 995 zAns += jj; | |
| 996 zCopy = sqlite3_mprintf("%.*s", len-jj-1, zAns); | |
| 997 if( (sqlite3_strglob(zCopy, sResult.z)==0)^isGlob ){ | |
| 998 errorMessage("line %d of %s:\nExpected [%s]\n Got [%s]", | |
| 999 prevLine, zFilename, zCopy, sResult.z); | |
| 1000 } | |
| 1001 sqlite3_free(zCopy); | |
| 1002 g.nTest++; | |
| 1003 stringReset(&sResult); | |
| 1004 }else | |
| 1005 | |
| 1006 /* | |
| 1007 ** --output | |
| 1008 ** | |
| 1009 ** Output the result of the previous SQL. | |
| 1010 */ | |
| 1011 if( strcmp(zCmd, "output")==0 ){ | |
| 1012 logMessage("%s", sResult.z); | |
| 1013 }else | |
| 1014 | |
| 1015 /* | |
| 1016 ** --source FILENAME | |
| 1017 ** | |
| 1018 ** Run a subscript from a separate file. | |
| 1019 */ | |
| 1020 if( strcmp(zCmd, "source")==0 ){ | |
| 1021 char *zNewFile, *zNewScript; | |
| 1022 char *zToDel = 0; | |
| 1023 zNewFile = azArg[0]; | |
| 1024 if( zNewFile[0]!='/' ){ | |
| 1025 int k; | |
| 1026 for(k=(int)strlen(zFilename)-1; k>=0 && zFilename[k]!='/'; k--){} | |
| 1027 if( k>0 ){ | |
| 1028 zNewFile = zToDel = sqlite3_mprintf("%.*s/%s", k,zFilename,zNewFile); | |
| 1029 } | |
| 1030 } | |
| 1031 zNewScript = readFile(zNewFile); | |
| 1032 if( g.iTrace ) logMessage("begin script [%s]\n", zNewFile); | |
| 1033 runScript(0, 0, zNewScript, zNewFile); | |
| 1034 sqlite3_free(zNewScript); | |
| 1035 if( g.iTrace ) logMessage("end script [%s]\n", zNewFile); | |
| 1036 sqlite3_free(zToDel); | |
| 1037 }else | |
| 1038 | |
| 1039 /* | |
| 1040 ** --print MESSAGE.... | |
| 1041 ** | |
| 1042 ** Output the remainder of the line to the log file | |
| 1043 */ | |
| 1044 if( strcmp(zCmd, "print")==0 ){ | |
| 1045 int jj; | |
| 1046 for(jj=7; jj<len && isspace(zScript[ii+jj]); jj++){} | |
| 1047 logMessage("%.*s", len-jj, zScript+ii+jj); | |
| 1048 }else | |
| 1049 | |
| 1050 /* | |
| 1051 ** --if EXPR | |
| 1052 ** | |
| 1053 ** Skip forward to the next matching --endif or --else if EXPR is false. | |
| 1054 */ | |
| 1055 if( strcmp(zCmd, "if")==0 ){ | |
| 1056 int jj, rc; | |
| 1057 sqlite3_stmt *pStmt; | |
| 1058 for(jj=4; jj<len && isspace(zScript[ii+jj]); jj++){} | |
| 1059 pStmt = prepareSql("SELECT %.*s", len-jj, zScript+ii+jj); | |
| 1060 rc = sqlite3_step(pStmt); | |
| 1061 if( rc!=SQLITE_ROW || sqlite3_column_int(pStmt, 0)==0 ){ | |
| 1062 ii += findEndif(zScript+ii+len, 1, &lineno); | |
| 1063 } | |
| 1064 sqlite3_finalize(pStmt); | |
| 1065 }else | |
| 1066 | |
| 1067 /* | |
| 1068 ** --else | |
| 1069 ** | |
| 1070 ** This command can only be encountered if currently inside an --if that | |
| 1071 ** is true. Skip forward to the next matching --endif. | |
| 1072 */ | |
| 1073 if( strcmp(zCmd, "else")==0 ){ | |
| 1074 ii += findEndif(zScript+ii+len, 0, &lineno); | |
| 1075 }else | |
| 1076 | |
| 1077 /* | |
| 1078 ** --endif | |
| 1079 ** | |
| 1080 ** This command can only be encountered if currently inside an --if that | |
| 1081 ** is true or an --else of a false if. This is a no-op. | |
| 1082 */ | |
| 1083 if( strcmp(zCmd, "endif")==0 ){ | |
| 1084 /* no-op */ | |
| 1085 }else | |
| 1086 | |
| 1087 /* | |
| 1088 ** --start CLIENT | |
| 1089 ** | |
| 1090 ** Start up the given client. | |
| 1091 */ | |
| 1092 if( strcmp(zCmd, "start")==0 && iClient==0 ){ | |
| 1093 int iNewClient = atoi(azArg[0]); | |
| 1094 if( iNewClient>0 ){ | |
| 1095 startClient(iNewClient); | |
| 1096 } | |
| 1097 }else | |
| 1098 | |
| 1099 /* | |
| 1100 ** --wait CLIENT TIMEOUT | |
| 1101 ** | |
| 1102 ** Wait until all tasks complete for the given client. If CLIENT is | |
| 1103 ** "all" then wait for all clients to complete. Wait no longer than | |
| 1104 ** TIMEOUT milliseconds (default 10,000) | |
| 1105 */ | |
| 1106 if( strcmp(zCmd, "wait")==0 && iClient==0 ){ | |
| 1107 int iTimeout = nArg>=2 ? atoi(azArg[1]) : 10000; | |
| 1108 sqlite3_snprintf(sizeof(zError),zError,"line %d of %s\n", | |
| 1109 prevLine, zFilename); | |
| 1110 waitForClient(atoi(azArg[0]), iTimeout, zError); | |
| 1111 }else | |
| 1112 | |
| 1113 /* | |
| 1114 ** --task CLIENT | |
| 1115 ** <task-content-here> | |
| 1116 ** --end | |
| 1117 ** | |
| 1118 ** Assign work to a client. Start the client if it is not running | |
| 1119 ** already. | |
| 1120 */ | |
| 1121 if( strcmp(zCmd, "task")==0 && iClient==0 ){ | |
| 1122 int iTarget = atoi(azArg[0]); | |
| 1123 int iEnd; | |
| 1124 char *zTask; | |
| 1125 char *zTName; | |
| 1126 iEnd = findEnd(zScript+ii+len, &lineno); | |
| 1127 if( iTarget<0 ){ | |
| 1128 errorMessage("line %d of %s: bad client number: %d", | |
| 1129 prevLine, zFilename, iTarget); | |
| 1130 }else{ | |
| 1131 zTask = sqlite3_mprintf("%.*s", iEnd, zScript+ii+len); | |
| 1132 if( nArg>1 ){ | |
| 1133 zTName = sqlite3_mprintf("%s", azArg[1]); | |
| 1134 }else{ | |
| 1135 zTName = sqlite3_mprintf("%s:%d", filenameTail(zFilename), prevLine); | |
| 1136 } | |
| 1137 startClient(iTarget); | |
| 1138 runSql("INSERT INTO task(client,script,name)" | |
| 1139 " VALUES(%d,'%q',%Q)", iTarget, zTask, zTName); | |
| 1140 sqlite3_free(zTask); | |
| 1141 sqlite3_free(zTName); | |
| 1142 } | |
| 1143 iEnd += tokenLength(zScript+ii+len+iEnd, &lineno); | |
| 1144 len += iEnd; | |
| 1145 iBegin = ii+len; | |
| 1146 }else | |
| 1147 | |
| 1148 /* | |
| 1149 ** --breakpoint | |
| 1150 ** | |
| 1151 ** This command calls "test_breakpoint()" which is a routine provided | |
| 1152 ** as a convenient place to set a debugger breakpoint. | |
| 1153 */ | |
| 1154 if( strcmp(zCmd, "breakpoint")==0 ){ | |
| 1155 test_breakpoint(); | |
| 1156 }else | |
| 1157 | |
| 1158 /* | |
| 1159 ** --show-sql-errors BOOLEAN | |
| 1160 ** | |
| 1161 ** Turn display of SQL errors on and off. | |
| 1162 */ | |
| 1163 if( strcmp(zCmd, "show-sql-errors")==0 ){ | |
| 1164 g.bIgnoreSqlErrors = nArg>=1 ? !booleanValue(azArg[0]) : 1; | |
| 1165 }else | |
| 1166 | |
| 1167 | |
| 1168 /* error */{ | |
| 1169 errorMessage("line %d of %s: unknown command --%s", | |
| 1170 prevLine, zFilename, zCmd); | |
| 1171 } | |
| 1172 ii += len; | |
| 1173 } | |
| 1174 if( iBegin<ii ){ | |
| 1175 char *zSql = sqlite3_mprintf("%.*s", ii-iBegin, zScript+iBegin); | |
| 1176 runSql(zSql); | |
| 1177 sqlite3_free(zSql); | |
| 1178 } | |
| 1179 stringFree(&sResult); | |
| 1180 } | |
| 1181 | |
| 1182 /* | |
| 1183 ** Look for a command-line option. If present, return a pointer. | |
| 1184 ** Return NULL if missing. | |
| 1185 ** | |
| 1186 ** hasArg==0 means the option is a flag. It is either present or not. | |
| 1187 ** hasArg==1 means the option has an argument. Return a pointer to the | |
| 1188 ** argument. | |
| 1189 */ | |
| 1190 static char *findOption( | |
| 1191 char **azArg, | |
| 1192 int *pnArg, | |
| 1193 const char *zOption, | |
| 1194 int hasArg | |
| 1195 ){ | |
| 1196 int i, j; | |
| 1197 char *zReturn = 0; | |
| 1198 int nArg = *pnArg; | |
| 1199 | |
| 1200 assert( hasArg==0 || hasArg==1 ); | |
| 1201 for(i=0; i<nArg; i++){ | |
| 1202 const char *z; | |
| 1203 if( i+hasArg >= nArg ) break; | |
| 1204 z = azArg[i]; | |
| 1205 if( z[0]!='-' ) continue; | |
| 1206 z++; | |
| 1207 if( z[0]=='-' ){ | |
| 1208 if( z[1]==0 ) break; | |
| 1209 z++; | |
| 1210 } | |
| 1211 if( strcmp(z,zOption)==0 ){ | |
| 1212 if( hasArg && i==nArg-1 ){ | |
| 1213 fatalError("command-line option \"--%s\" requires an argument", z); | |
| 1214 } | |
| 1215 if( hasArg ){ | |
| 1216 zReturn = azArg[i+1]; | |
| 1217 }else{ | |
| 1218 zReturn = azArg[i]; | |
| 1219 } | |
| 1220 j = i+1+(hasArg!=0); | |
| 1221 while( j<nArg ) azArg[i++] = azArg[j++]; | |
| 1222 *pnArg = i; | |
| 1223 return zReturn; | |
| 1224 } | |
| 1225 } | |
| 1226 return zReturn; | |
| 1227 } | |
| 1228 | |
| 1229 /* Print a usage message for the program and exit */ | |
| 1230 static void usage(const char *argv0){ | |
| 1231 int i; | |
| 1232 const char *zTail = argv0; | |
| 1233 for(i=0; argv0[i]; i++){ | |
| 1234 if( argv0[i]=='/' ) zTail = argv0+i+1; | |
| 1235 } | |
| 1236 fprintf(stderr,"Usage: %s DATABASE ?OPTIONS? ?SCRIPT?\n", zTail); | |
| 1237 exit(1); | |
| 1238 } | |
| 1239 | |
| 1240 /* Report on unrecognized arguments */ | |
| 1241 static void unrecognizedArguments( | |
| 1242 const char *argv0, | |
| 1243 int nArg, | |
| 1244 char **azArg | |
| 1245 ){ | |
| 1246 int i; | |
| 1247 fprintf(stderr,"%s: unrecognized arguments:", argv0); | |
| 1248 for(i=0; i<nArg; i++){ | |
| 1249 fprintf(stderr," %s", azArg[i]); | |
| 1250 } | |
| 1251 fprintf(stderr,"\n"); | |
| 1252 exit(1); | |
| 1253 } | |
| 1254 | |
| 1255 int main(int argc, char **argv){ | |
| 1256 const char *zClient; | |
| 1257 int iClient; | |
| 1258 int n, i; | |
| 1259 int openFlags = SQLITE_OPEN_READWRITE; | |
| 1260 int rc; | |
| 1261 char *zScript; | |
| 1262 int taskId; | |
| 1263 const char *zTrace; | |
| 1264 const char *zCOption; | |
| 1265 | |
| 1266 g.argv0 = argv[0]; | |
| 1267 g.iTrace = 1; | |
| 1268 if( argc<2 ) usage(argv[0]); | |
| 1269 g.zDbFile = argv[1]; | |
| 1270 if( strglob("*.test", g.zDbFile) ) usage(argv[0]); | |
| 1271 if( strcmp(sqlite3_sourceid(), SQLITE_SOURCE_ID)!=0 ){ | |
| 1272 fprintf(stderr, "SQLite library and header mismatch\n" | |
| 1273 "Library: %s\n" | |
| 1274 "Header: %s\n", | |
| 1275 sqlite3_sourceid(), SQLITE_SOURCE_ID); | |
| 1276 exit(1); | |
| 1277 } | |
| 1278 n = argc-2; | |
| 1279 sqlite3_snprintf(sizeof(g.zName), g.zName, "%05d.mptest", GETPID()); | |
| 1280 g.zVfs = findOption(argv+2, &n, "vfs", 1); | |
| 1281 zClient = findOption(argv+2, &n, "client", 1); | |
| 1282 g.zErrLog = findOption(argv+2, &n, "errlog", 1); | |
| 1283 g.zLog = findOption(argv+2, &n, "log", 1); | |
| 1284 zTrace = findOption(argv+2, &n, "trace", 1); | |
| 1285 if( zTrace ) g.iTrace = atoi(zTrace); | |
| 1286 if( findOption(argv+2, &n, "quiet", 0)!=0 ) g.iTrace = 0; | |
| 1287 g.bSqlTrace = findOption(argv+2, &n, "sqltrace", 0)!=0; | |
| 1288 g.bSync = findOption(argv+2, &n, "sync", 0)!=0; | |
| 1289 if( g.zErrLog ){ | |
| 1290 g.pErrLog = fopen(g.zErrLog, "a"); | |
| 1291 }else{ | |
| 1292 g.pErrLog = stderr; | |
| 1293 } | |
| 1294 if( g.zLog ){ | |
| 1295 g.pLog = fopen(g.zLog, "a"); | |
| 1296 }else{ | |
| 1297 g.pLog = stdout; | |
| 1298 } | |
| 1299 | |
| 1300 sqlite3_config(SQLITE_CONFIG_LOG, sqlErrorCallback, 0); | |
| 1301 if( zClient ){ | |
| 1302 iClient = atoi(zClient); | |
| 1303 if( iClient<1 ) fatalError("illegal client number: %d\n", iClient); | |
| 1304 sqlite3_snprintf(sizeof(g.zName), g.zName, "%05d.client%02d", | |
| 1305 GETPID(), iClient); | |
| 1306 }else{ | |
| 1307 if( g.iTrace>0 ){ | |
| 1308 printf("With SQLite " SQLITE_VERSION " " SQLITE_SOURCE_ID "\n" ); | |
| 1309 for(i=0; (zCOption = sqlite3_compileoption_get(i))!=0; i++){ | |
| 1310 printf("-DSQLITE_%s\n", zCOption); | |
| 1311 } | |
| 1312 fflush(stdout); | |
| 1313 } | |
| 1314 iClient = 0; | |
| 1315 unlink(g.zDbFile); | |
| 1316 openFlags |= SQLITE_OPEN_CREATE; | |
| 1317 } | |
| 1318 rc = sqlite3_open_v2(g.zDbFile, &g.db, openFlags, g.zVfs); | |
| 1319 if( rc ) fatalError("cannot open [%s]", g.zDbFile); | |
| 1320 sqlite3_enable_load_extension(g.db, 1); | |
| 1321 sqlite3_busy_handler(g.db, busyHandler, 0); | |
| 1322 sqlite3_create_function(g.db, "vfsname", 0, SQLITE_UTF8, 0, | |
| 1323 vfsNameFunc, 0, 0); | |
| 1324 sqlite3_create_function(g.db, "eval", 1, SQLITE_UTF8, 0, | |
| 1325 evalFunc, 0, 0); | |
| 1326 g.iTimeout = DEFAULT_TIMEOUT; | |
| 1327 if( g.bSqlTrace ) sqlite3_trace(g.db, sqlTraceCallback, 0); | |
| 1328 if( !g.bSync ) trySql("PRAGMA synchronous=OFF"); | |
| 1329 if( iClient>0 ){ | |
| 1330 if( n>0 ) unrecognizedArguments(argv[0], n, argv+2); | |
| 1331 if( g.iTrace ) logMessage("start-client"); | |
| 1332 while(1){ | |
| 1333 char *zTaskName = 0; | |
| 1334 rc = startScript(iClient, &zScript, &taskId, &zTaskName); | |
| 1335 if( rc==SQLITE_DONE ) break; | |
| 1336 if( g.iTrace ) logMessage("begin %s (%d)", zTaskName, taskId); | |
| 1337 runScript(iClient, taskId, zScript, zTaskName); | |
| 1338 if( g.iTrace ) logMessage("end %s (%d)", zTaskName, taskId); | |
| 1339 finishScript(iClient, taskId, 0); | |
| 1340 sqlite3_free(zTaskName); | |
| 1341 sqlite3_sleep(10); | |
| 1342 } | |
| 1343 if( g.iTrace ) logMessage("end-client"); | |
| 1344 }else{ | |
| 1345 sqlite3_stmt *pStmt; | |
| 1346 int iTimeout; | |
| 1347 if( n==0 ){ | |
| 1348 fatalError("missing script filename"); | |
| 1349 } | |
| 1350 if( n>1 ) unrecognizedArguments(argv[0], n, argv+2); | |
| 1351 runSql( | |
| 1352 "CREATE TABLE task(\n" | |
| 1353 " id INTEGER PRIMARY KEY,\n" | |
| 1354 " name TEXT,\n" | |
| 1355 " client INTEGER,\n" | |
| 1356 " starttime DATE,\n" | |
| 1357 " endtime DATE,\n" | |
| 1358 " script TEXT\n" | |
| 1359 ");" | |
| 1360 "CREATE INDEX task_i1 ON task(client, starttime);\n" | |
| 1361 "CREATE INDEX task_i2 ON task(client, endtime);\n" | |
| 1362 "CREATE TABLE counters(nError,nTest);\n" | |
| 1363 "INSERT INTO counters VALUES(0,0);\n" | |
| 1364 "CREATE TABLE client(id INTEGER PRIMARY KEY, wantHalt);\n" | |
| 1365 ); | |
| 1366 zScript = readFile(argv[2]); | |
| 1367 if( g.iTrace ) logMessage("begin script [%s]\n", argv[2]); | |
| 1368 runScript(0, 0, zScript, argv[2]); | |
| 1369 sqlite3_free(zScript); | |
| 1370 if( g.iTrace ) logMessage("end script [%s]\n", argv[2]); | |
| 1371 waitForClient(0, 2000, "during shutdown...\n"); | |
| 1372 trySql("UPDATE client SET wantHalt=1"); | |
| 1373 sqlite3_sleep(10); | |
| 1374 g.iTimeout = 0; | |
| 1375 iTimeout = 1000; | |
| 1376 while( ((rc = trySql("SELECT 1 FROM client"))==SQLITE_BUSY | |
| 1377 || rc==SQLITE_ROW) && iTimeout>0 ){ | |
| 1378 sqlite3_sleep(10); | |
| 1379 iTimeout -= 10; | |
| 1380 } | |
| 1381 sqlite3_sleep(100); | |
| 1382 pStmt = prepareSql("SELECT nError, nTest FROM counters"); | |
| 1383 iTimeout = 1000; | |
| 1384 while( (rc = sqlite3_step(pStmt))==SQLITE_BUSY && iTimeout>0 ){ | |
| 1385 sqlite3_sleep(10); | |
| 1386 iTimeout -= 10; | |
| 1387 } | |
| 1388 if( rc==SQLITE_ROW ){ | |
| 1389 g.nError += sqlite3_column_int(pStmt, 0); | |
| 1390 g.nTest += sqlite3_column_int(pStmt, 1); | |
| 1391 } | |
| 1392 sqlite3_finalize(pStmt); | |
| 1393 } | |
| 1394 sqlite3_close(g.db); | |
| 1395 maybeClose(g.pLog); | |
| 1396 maybeClose(g.pErrLog); | |
| 1397 if( iClient==0 ){ | |
| 1398 printf("Summary: %d errors in %d tests\n", g.nError, g.nTest); | |
| 1399 } | |
| 1400 return g.nError>0; | |
| 1401 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2363173002:
[sqlite] Remove obsolete reference version 3.8.7.4. (Closed)
