Add tests to bestindexC.test. No changes to code.
[sqlite.git] / src / shell.c.in
blob2984853442092b610485a4783124b56adc1cbcec
1 /*
2 ** 2001 September 15
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.
11 *************************************************************************
12 ** This file contains code to implement the "sqlite" command line
13 ** utility for accessing SQLite databases.
15 #if (defined(_WIN32) || defined(WIN32)) && !defined(_CRT_SECURE_NO_WARNINGS)
16 /* This needs to come before any includes for MSVC compiler */
17 #define _CRT_SECURE_NO_WARNINGS
18 #endif
19 typedef unsigned int u32;
20 typedef unsigned short int u16;
23 ** Optionally #include a user-defined header, whereby compilation options
24 ** may be set prior to where they take effect, but after platform setup.
25 ** If SQLITE_CUSTOM_INCLUDE=? is defined, its value names the #include
26 ** file. Note that this macro has a like effect on sqlite3.c compilation.
28 # define SHELL_STRINGIFY_(f) #f
29 # define SHELL_STRINGIFY(f) SHELL_STRINGIFY_(f)
30 #ifdef SQLITE_CUSTOM_INCLUDE
31 # include SHELL_STRINGIFY(SQLITE_CUSTOM_INCLUDE)
32 #endif
35 ** Determine if we are dealing with WinRT, which provides only a subset of
36 ** the full Win32 API.
38 #if !defined(SQLITE_OS_WINRT)
39 # define SQLITE_OS_WINRT 0
40 #endif
43 ** If SQLITE_SHELL_FIDDLE is defined then the shell is modified
44 ** somewhat for use as a WASM module in a web browser. This flag
45 ** should only be used when building the "fiddle" web application, as
46 ** the browser-mode build has much different user input requirements
47 ** and this build mode rewires the user input subsystem to account for
48 ** that.
52 ** Warning pragmas copied from msvc.h in the core.
54 #if defined(_MSC_VER)
55 #pragma warning(disable : 4054)
56 #pragma warning(disable : 4055)
57 #pragma warning(disable : 4100)
58 #pragma warning(disable : 4127)
59 #pragma warning(disable : 4130)
60 #pragma warning(disable : 4152)
61 #pragma warning(disable : 4189)
62 #pragma warning(disable : 4206)
63 #pragma warning(disable : 4210)
64 #pragma warning(disable : 4232)
65 #pragma warning(disable : 4244)
66 #pragma warning(disable : 4305)
67 #pragma warning(disable : 4306)
68 #pragma warning(disable : 4702)
69 #pragma warning(disable : 4706)
70 #endif /* defined(_MSC_VER) */
73 ** No support for loadable extensions in VxWorks.
75 #if (defined(__RTP__) || defined(_WRS_KERNEL)) && !SQLITE_OMIT_LOAD_EXTENSION
76 # define SQLITE_OMIT_LOAD_EXTENSION 1
77 #endif
80 ** Enable large-file support for fopen() and friends on unix.
82 #ifndef SQLITE_DISABLE_LFS
83 # define _LARGE_FILE 1
84 # ifndef _FILE_OFFSET_BITS
85 # define _FILE_OFFSET_BITS 64
86 # endif
87 # define _LARGEFILE_SOURCE 1
88 #endif
90 #if defined(SQLITE_SHELL_FIDDLE) && !defined(_POSIX_SOURCE)
92 ** emcc requires _POSIX_SOURCE (or one of several similar defines)
93 ** to expose strdup().
95 # define _POSIX_SOURCE
96 #endif
98 #include <stdlib.h>
99 #include <string.h>
100 #include <stdio.h>
101 #include <assert.h>
102 #include <math.h>
103 #include "sqlite3.h"
104 typedef sqlite3_int64 i64;
105 typedef sqlite3_uint64 u64;
106 typedef unsigned char u8;
107 #if SQLITE_USER_AUTHENTICATION
108 # include "sqlite3userauth.h"
109 #endif
110 #include <ctype.h>
111 #include <stdarg.h>
113 #if !defined(_WIN32) && !defined(WIN32)
114 # include <signal.h>
115 # if !defined(__RTP__) && !defined(_WRS_KERNEL) && !defined(SQLITE_WASI)
116 # include <pwd.h>
117 # endif
118 #endif
119 #if (!defined(_WIN32) && !defined(WIN32)) || defined(__MINGW32__)
120 # include <unistd.h>
121 # include <dirent.h>
122 # define GETPID getpid
123 # if defined(__MINGW32__)
124 # define DIRENT dirent
125 # ifndef S_ISLNK
126 # define S_ISLNK(mode) (0)
127 # endif
128 # endif
129 #else
130 # define GETPID (int)GetCurrentProcessId
131 #endif
132 #include <sys/types.h>
133 #include <sys/stat.h>
135 #if HAVE_READLINE
136 # include <readline/readline.h>
137 # include <readline/history.h>
138 #endif
140 #if HAVE_EDITLINE
141 # include <editline/readline.h>
142 #endif
144 #if HAVE_EDITLINE || HAVE_READLINE
146 # define shell_add_history(X) add_history(X)
147 # define shell_read_history(X) read_history(X)
148 # define shell_write_history(X) write_history(X)
149 # define shell_stifle_history(X) stifle_history(X)
150 # define shell_readline(X) readline(X)
152 #elif HAVE_LINENOISE
154 # include "linenoise.h"
155 # define shell_add_history(X) linenoiseHistoryAdd(X)
156 # define shell_read_history(X) linenoiseHistoryLoad(X)
157 # define shell_write_history(X) linenoiseHistorySave(X)
158 # define shell_stifle_history(X) linenoiseHistorySetMaxLen(X)
159 # define shell_readline(X) linenoise(X)
161 #else
163 # define shell_read_history(X)
164 # define shell_write_history(X)
165 # define shell_stifle_history(X)
167 # define SHELL_USE_LOCAL_GETLINE 1
168 #endif
170 #ifndef deliberate_fall_through
171 /* Quiet some compilers about some of our intentional code. */
172 # if defined(GCC_VERSION) && GCC_VERSION>=7000000
173 # define deliberate_fall_through __attribute__((fallthrough));
174 # else
175 # define deliberate_fall_through
176 # endif
177 #endif
179 #if defined(_WIN32) || defined(WIN32)
180 # if SQLITE_OS_WINRT
181 # define SQLITE_OMIT_POPEN 1
182 # else
183 # include <io.h>
184 # include <fcntl.h>
185 # define isatty(h) _isatty(h)
186 # ifndef access
187 # define access(f,m) _access((f),(m))
188 # endif
189 # ifndef unlink
190 # define unlink _unlink
191 # endif
192 # ifndef strdup
193 # define strdup _strdup
194 # endif
195 # undef popen
196 # define popen _popen
197 # undef pclose
198 # define pclose _pclose
199 # endif
200 #else
201 /* Make sure isatty() has a prototype. */
202 extern int isatty(int);
204 # if !defined(__RTP__) && !defined(_WRS_KERNEL) && !defined(SQLITE_WASI)
205 /* popen and pclose are not C89 functions and so are
206 ** sometimes omitted from the <stdio.h> header */
207 extern FILE *popen(const char*,const char*);
208 extern int pclose(FILE*);
209 # else
210 # define SQLITE_OMIT_POPEN 1
211 # endif
212 #endif
214 #if defined(_WIN32_WCE)
215 /* Windows CE (arm-wince-mingw32ce-gcc) does not provide isatty()
216 * thus we always assume that we have a console. That can be
217 * overridden with the -batch command line option.
219 #define isatty(x) 1
220 #endif
222 /* ctype macros that work with signed characters */
223 #define IsSpace(X) isspace((unsigned char)X)
224 #define IsDigit(X) isdigit((unsigned char)X)
225 #define ToLower(X) (char)tolower((unsigned char)X)
227 #if defined(_WIN32) || defined(WIN32)
228 #if SQLITE_OS_WINRT
229 #include <intrin.h>
230 #endif
231 #undef WIN32_LEAN_AND_MEAN
232 #define WIN32_LEAN_AND_MEAN
233 #include <windows.h>
235 /* string conversion routines only needed on Win32 */
236 extern char *sqlite3_win32_unicode_to_utf8(LPCWSTR);
237 extern LPWSTR sqlite3_win32_utf8_to_unicode(const char *zText);
238 #endif
240 /* Use console I/O package as a direct INCLUDE. */
241 #define SQLITE_INTERNAL_LINKAGE static
243 #ifdef SQLITE_SHELL_FIDDLE
244 /* Deselect most features from the console I/O package for Fiddle. */
245 # define SQLITE_CIO_NO_REDIRECT
246 # define SQLITE_CIO_NO_CLASSIFY
247 # define SQLITE_CIO_NO_TRANSLATE
248 # define SQLITE_CIO_NO_SETMODE
249 #endif
250 INCLUDE ../ext/consio/console_io.h
251 INCLUDE ../ext/consio/console_io.c
253 #ifndef SQLITE_SHELL_FIDDLE
255 /* From here onward, fgets() is redirected to the console_io library. */
256 # define fgets(b,n,f) fGetsUtf8(b,n,f)
258 * Define macros for emitting output text in various ways:
259 * sputz(s, z) => emit 0-terminated string z to given stream s
260 * sputf(s, f, ...) => emit varargs per format f to given stream s
261 * oputz(z) => emit 0-terminated string z to default stream
262 * oputf(f, ...) => emit varargs per format f to default stream
263 * eputz(z) => emit 0-terminated string z to error stream
264 * eputf(f, ...) => emit varargs per format f to error stream
265 * oputb(b, n) => emit char buffer b[0..n-1] to default stream
267 * Note that the default stream is whatever has been last set via:
268 * setOutputStream(FILE *pf)
269 * This is normally the stream that CLI normal output goes to.
270 * For the stand-alone CLI, it is stdout with no .output redirect.
272 * The ?putz(z) forms are required for the Fiddle builds for string literal
273 * output, in aid of enforcing format string to argument correspondence.
275 # define sputz(s,z) fPutsUtf8(z,s)
276 # define sputf fPrintfUtf8
277 # define oputz(z) oPutsUtf8(z)
278 # define oputf oPrintfUtf8
279 # define eputz(z) ePutsUtf8(z)
280 # define eputf ePrintfUtf8
281 # define oputb(buf,na) oPutbUtf8(buf,na)
283 #else
284 /* For Fiddle, all console handling and emit redirection is omitted. */
285 /* These next 3 macros are for emitting formatted output. When complaints
286 * from the WASM build are issued for non-formatted output, (when a mere
287 * string literal is to be emitted, the ?putz(z) forms should be used.
288 * (This permits compile-time checking of format string / argument mismatch.)
290 # define oputf(fmt, ...) printf(fmt,__VA_ARGS__)
291 # define eputf(fmt, ...) fprintf(stderr,fmt,__VA_ARGS__)
292 # define sputf(fp,fmt, ...) fprintf(fp,fmt,__VA_ARGS__)
293 /* These next 3 macros are for emitting simple string literals. */
294 # define oputz(z) fputs(z,stdout)
295 # define eputz(z) fputs(z,stderr)
296 # define sputz(fp,z) fputs(z,fp)
297 # define oputb(buf,na) fwrite(buf,1,na,stdout)
298 #endif
300 /* True if the timer is enabled */
301 static int enableTimer = 0;
303 /* A version of strcmp() that works with NULL values */
304 static int cli_strcmp(const char *a, const char *b){
305 if( a==0 ) a = "";
306 if( b==0 ) b = "";
307 return strcmp(a,b);
309 static int cli_strncmp(const char *a, const char *b, size_t n){
310 if( a==0 ) a = "";
311 if( b==0 ) b = "";
312 return strncmp(a,b,n);
315 /* Return the current wall-clock time */
316 static sqlite3_int64 timeOfDay(void){
317 static sqlite3_vfs *clockVfs = 0;
318 sqlite3_int64 t;
319 if( clockVfs==0 ) clockVfs = sqlite3_vfs_find(0);
320 if( clockVfs==0 ) return 0; /* Never actually happens */
321 if( clockVfs->iVersion>=2 && clockVfs->xCurrentTimeInt64!=0 ){
322 clockVfs->xCurrentTimeInt64(clockVfs, &t);
323 }else{
324 double r;
325 clockVfs->xCurrentTime(clockVfs, &r);
326 t = (sqlite3_int64)(r*86400000.0);
328 return t;
331 #if !defined(_WIN32) && !defined(WIN32) && !defined(__minux)
332 #include <sys/time.h>
333 #include <sys/resource.h>
335 /* VxWorks does not support getrusage() as far as we can determine */
336 #if defined(_WRS_KERNEL) || defined(__RTP__)
337 struct rusage {
338 struct timeval ru_utime; /* user CPU time used */
339 struct timeval ru_stime; /* system CPU time used */
341 #define getrusage(A,B) memset(B,0,sizeof(*B))
342 #endif
344 /* Saved resource information for the beginning of an operation */
345 static struct rusage sBegin; /* CPU time at start */
346 static sqlite3_int64 iBegin; /* Wall-clock time at start */
349 ** Begin timing an operation
351 static void beginTimer(void){
352 if( enableTimer ){
353 getrusage(RUSAGE_SELF, &sBegin);
354 iBegin = timeOfDay();
358 /* Return the difference of two time_structs in seconds */
359 static double timeDiff(struct timeval *pStart, struct timeval *pEnd){
360 return (pEnd->tv_usec - pStart->tv_usec)*0.000001 +
361 (double)(pEnd->tv_sec - pStart->tv_sec);
365 ** Print the timing results.
367 static void endTimer(void){
368 if( enableTimer ){
369 sqlite3_int64 iEnd = timeOfDay();
370 struct rusage sEnd;
371 getrusage(RUSAGE_SELF, &sEnd);
372 sputf(stdout, "Run Time: real %.3f user %f sys %f\n",
373 (iEnd - iBegin)*0.001,
374 timeDiff(&sBegin.ru_utime, &sEnd.ru_utime),
375 timeDiff(&sBegin.ru_stime, &sEnd.ru_stime));
379 #define BEGIN_TIMER beginTimer()
380 #define END_TIMER endTimer()
381 #define HAS_TIMER 1
383 #elif (defined(_WIN32) || defined(WIN32))
385 /* Saved resource information for the beginning of an operation */
386 static HANDLE hProcess;
387 static FILETIME ftKernelBegin;
388 static FILETIME ftUserBegin;
389 static sqlite3_int64 ftWallBegin;
390 typedef BOOL (WINAPI *GETPROCTIMES)(HANDLE, LPFILETIME, LPFILETIME,
391 LPFILETIME, LPFILETIME);
392 static GETPROCTIMES getProcessTimesAddr = NULL;
395 ** Check to see if we have timer support. Return 1 if necessary
396 ** support found (or found previously).
398 static int hasTimer(void){
399 if( getProcessTimesAddr ){
400 return 1;
401 } else {
402 #if !SQLITE_OS_WINRT
403 /* GetProcessTimes() isn't supported in WIN95 and some other Windows
404 ** versions. See if the version we are running on has it, and if it
405 ** does, save off a pointer to it and the current process handle.
407 hProcess = GetCurrentProcess();
408 if( hProcess ){
409 HINSTANCE hinstLib = LoadLibrary(TEXT("Kernel32.dll"));
410 if( NULL != hinstLib ){
411 getProcessTimesAddr =
412 (GETPROCTIMES) GetProcAddress(hinstLib, "GetProcessTimes");
413 if( NULL != getProcessTimesAddr ){
414 return 1;
416 FreeLibrary(hinstLib);
419 #endif
421 return 0;
425 ** Begin timing an operation
427 static void beginTimer(void){
428 if( enableTimer && getProcessTimesAddr ){
429 FILETIME ftCreation, ftExit;
430 getProcessTimesAddr(hProcess,&ftCreation,&ftExit,
431 &ftKernelBegin,&ftUserBegin);
432 ftWallBegin = timeOfDay();
436 /* Return the difference of two FILETIME structs in seconds */
437 static double timeDiff(FILETIME *pStart, FILETIME *pEnd){
438 sqlite_int64 i64Start = *((sqlite_int64 *) pStart);
439 sqlite_int64 i64End = *((sqlite_int64 *) pEnd);
440 return (double) ((i64End - i64Start) / 10000000.0);
444 ** Print the timing results.
446 static void endTimer(void){
447 if( enableTimer && getProcessTimesAddr){
448 FILETIME ftCreation, ftExit, ftKernelEnd, ftUserEnd;
449 sqlite3_int64 ftWallEnd = timeOfDay();
450 getProcessTimesAddr(hProcess,&ftCreation,&ftExit,&ftKernelEnd,&ftUserEnd);
451 sputf(stdout, "Run Time: real %.3f user %f sys %f\n",
452 (ftWallEnd - ftWallBegin)*0.001,
453 timeDiff(&ftUserBegin, &ftUserEnd),
454 timeDiff(&ftKernelBegin, &ftKernelEnd));
458 #define BEGIN_TIMER beginTimer()
459 #define END_TIMER endTimer()
460 #define HAS_TIMER hasTimer()
462 #else
463 #define BEGIN_TIMER
464 #define END_TIMER
465 #define HAS_TIMER 0
466 #endif
469 ** Used to prevent warnings about unused parameters
471 #define UNUSED_PARAMETER(x) (void)(x)
474 ** Number of elements in an array
476 #define ArraySize(X) (int)(sizeof(X)/sizeof(X[0]))
479 ** If the following flag is set, then command execution stops
480 ** at an error if we are not interactive.
482 static int bail_on_error = 0;
485 ** Treat stdin as an interactive input if the following variable
486 ** is true. Otherwise, assume stdin is connected to a file or pipe.
488 static int stdin_is_interactive = 1;
491 ** On Windows systems we need to know if standard output is a console
492 ** in order to show that UTF-16 translation is done in the sign-on
493 ** banner. The following variable is true if it is the console.
495 static int stdout_is_console = 1;
498 ** The following is the open SQLite database. We make a pointer
499 ** to this database a static variable so that it can be accessed
500 ** by the SIGINT handler to interrupt database processing.
502 static sqlite3 *globalDb = 0;
505 ** True if an interrupt (Control-C) has been received.
507 static volatile int seenInterrupt = 0;
510 ** This is the name of our program. It is set in main(), used
511 ** in a number of other places, mostly for error messages.
513 static char *Argv0;
516 ** Prompt strings. Initialized in main. Settable with
517 ** .prompt main continue
519 #define PROMPT_LEN_MAX 20
520 /* First line prompt. default: "sqlite> " */
521 static char mainPrompt[PROMPT_LEN_MAX];
522 /* Continuation prompt. default: " ...> " */
523 static char continuePrompt[PROMPT_LEN_MAX];
525 /* This is variant of the standard-library strncpy() routine with the
526 ** one change that the destination string is always zero-terminated, even
527 ** if there is no zero-terminator in the first n-1 characters of the source
528 ** string.
530 static char *shell_strncpy(char *dest, const char *src, size_t n){
531 size_t i;
532 for(i=0; i<n-1 && src[i]!=0; i++) dest[i] = src[i];
533 dest[i] = 0;
534 return dest;
538 ** Optionally disable dynamic continuation prompt.
539 ** Unless disabled, the continuation prompt shows open SQL lexemes if any,
540 ** or open parentheses level if non-zero, or continuation prompt as set.
541 ** This facility interacts with the scanner and process_input() where the
542 ** below 5 macros are used.
544 #ifdef SQLITE_OMIT_DYNAPROMPT
545 # define CONTINUATION_PROMPT continuePrompt
546 # define CONTINUE_PROMPT_RESET
547 # define CONTINUE_PROMPT_AWAITS(p,s)
548 # define CONTINUE_PROMPT_AWAITC(p,c)
549 # define CONTINUE_PAREN_INCR(p,n)
550 # define CONTINUE_PROMPT_PSTATE 0
551 typedef void *t_NoDynaPrompt;
552 # define SCAN_TRACKER_REFTYPE t_NoDynaPrompt
553 #else
554 # define CONTINUATION_PROMPT dynamicContinuePrompt()
555 # define CONTINUE_PROMPT_RESET \
556 do {setLexemeOpen(&dynPrompt,0,0); trackParenLevel(&dynPrompt,0);} while(0)
557 # define CONTINUE_PROMPT_AWAITS(p,s) \
558 if(p && stdin_is_interactive) setLexemeOpen(p, s, 0)
559 # define CONTINUE_PROMPT_AWAITC(p,c) \
560 if(p && stdin_is_interactive) setLexemeOpen(p, 0, c)
561 # define CONTINUE_PAREN_INCR(p,n) \
562 if(p && stdin_is_interactive) (trackParenLevel(p,n))
563 # define CONTINUE_PROMPT_PSTATE (&dynPrompt)
564 typedef struct DynaPrompt *t_DynaPromptRef;
565 # define SCAN_TRACKER_REFTYPE t_DynaPromptRef
567 static struct DynaPrompt {
568 char dynamicPrompt[PROMPT_LEN_MAX];
569 char acAwait[2];
570 int inParenLevel;
571 char *zScannerAwaits;
572 } dynPrompt = { {0}, {0}, 0, 0 };
574 /* Record parenthesis nesting level change, or force level to 0. */
575 static void trackParenLevel(struct DynaPrompt *p, int ni){
576 p->inParenLevel += ni;
577 if( ni==0 ) p->inParenLevel = 0;
578 p->zScannerAwaits = 0;
581 /* Record that a lexeme is opened, or closed with args==0. */
582 static void setLexemeOpen(struct DynaPrompt *p, char *s, char c){
583 if( s!=0 || c==0 ){
584 p->zScannerAwaits = s;
585 p->acAwait[0] = 0;
586 }else{
587 p->acAwait[0] = c;
588 p->zScannerAwaits = p->acAwait;
592 /* Upon demand, derive the continuation prompt to display. */
593 static char *dynamicContinuePrompt(void){
594 if( continuePrompt[0]==0
595 || (dynPrompt.zScannerAwaits==0 && dynPrompt.inParenLevel == 0) ){
596 return continuePrompt;
597 }else{
598 if( dynPrompt.zScannerAwaits ){
599 size_t ncp = strlen(continuePrompt);
600 size_t ndp = strlen(dynPrompt.zScannerAwaits);
601 if( ndp > ncp-3 ) return continuePrompt;
602 strcpy(dynPrompt.dynamicPrompt, dynPrompt.zScannerAwaits);
603 while( ndp<3 ) dynPrompt.dynamicPrompt[ndp++] = ' ';
604 shell_strncpy(dynPrompt.dynamicPrompt+3, continuePrompt+3,
605 PROMPT_LEN_MAX-4);
606 }else{
607 if( dynPrompt.inParenLevel>9 ){
608 shell_strncpy(dynPrompt.dynamicPrompt, "(..", 4);
609 }else if( dynPrompt.inParenLevel<0 ){
610 shell_strncpy(dynPrompt.dynamicPrompt, ")x!", 4);
611 }else{
612 shell_strncpy(dynPrompt.dynamicPrompt, "(x.", 4);
613 dynPrompt.dynamicPrompt[2] = (char)('0'+dynPrompt.inParenLevel);
615 shell_strncpy(dynPrompt.dynamicPrompt+3, continuePrompt+3,
616 PROMPT_LEN_MAX-4);
619 return dynPrompt.dynamicPrompt;
621 #endif /* !defined(SQLITE_OMIT_DYNAPROMPT) */
623 /* Indicate out-of-memory and exit. */
624 static void shell_out_of_memory(void){
625 eputz("Error: out of memory\n");
626 exit(1);
629 /* Check a pointer to see if it is NULL. If it is NULL, exit with an
630 ** out-of-memory error.
632 static void shell_check_oom(const void *p){
633 if( p==0 ) shell_out_of_memory();
637 ** Write I/O traces to the following stream.
639 #ifdef SQLITE_ENABLE_IOTRACE
640 static FILE *iotrace = 0;
641 #endif
644 ** This routine works like printf in that its first argument is a
645 ** format string and subsequent arguments are values to be substituted
646 ** in place of % fields. The result of formatting this string
647 ** is written to iotrace.
649 #ifdef SQLITE_ENABLE_IOTRACE
650 static void SQLITE_CDECL iotracePrintf(const char *zFormat, ...){
651 va_list ap;
652 char *z;
653 if( iotrace==0 ) return;
654 va_start(ap, zFormat);
655 z = sqlite3_vmprintf(zFormat, ap);
656 va_end(ap);
657 sputf(iotrace, "%s", z);
658 sqlite3_free(z);
660 #endif
663 ** Output string zUtf to Out stream as w characters. If w is negative,
664 ** then right-justify the text. W is the width in UTF-8 characters, not
665 ** in bytes. This is different from the %*.*s specification in printf
666 ** since with %*.*s the width is measured in bytes, not characters.
668 static void utf8_width_print(int w, const char *zUtf){
669 int i;
670 int n;
671 int aw = w<0 ? -w : w;
672 if( zUtf==0 ) zUtf = "";
673 for(i=n=0; zUtf[i]; i++){
674 if( (zUtf[i]&0xc0)!=0x80 ){
675 n++;
676 if( n==aw ){
677 do{ i++; }while( (zUtf[i]&0xc0)==0x80 );
678 break;
682 if( n>=aw ){
683 oputf("%.*s", i, zUtf);
684 }else if( w<0 ){
685 oputf("%*s%s", aw-n, "", zUtf);
686 }else{
687 oputf("%s%*s", zUtf, aw-n, "");
693 ** Determines if a string is a number of not.
695 static int isNumber(const char *z, int *realnum){
696 if( *z=='-' || *z=='+' ) z++;
697 if( !IsDigit(*z) ){
698 return 0;
700 z++;
701 if( realnum ) *realnum = 0;
702 while( IsDigit(*z) ){ z++; }
703 if( *z=='.' ){
704 z++;
705 if( !IsDigit(*z) ) return 0;
706 while( IsDigit(*z) ){ z++; }
707 if( realnum ) *realnum = 1;
709 if( *z=='e' || *z=='E' ){
710 z++;
711 if( *z=='+' || *z=='-' ) z++;
712 if( !IsDigit(*z) ) return 0;
713 while( IsDigit(*z) ){ z++; }
714 if( realnum ) *realnum = 1;
716 return *z==0;
720 ** Compute a string length that is limited to what can be stored in
721 ** lower 30 bits of a 32-bit signed integer.
723 static int strlen30(const char *z){
724 const char *z2 = z;
725 while( *z2 ){ z2++; }
726 return 0x3fffffff & (int)(z2 - z);
730 ** Return the length of a string in characters. Multibyte UTF8 characters
731 ** count as a single character.
733 static int strlenChar(const char *z){
734 int n = 0;
735 while( *z ){
736 if( (0xc0&*(z++))!=0x80 ) n++;
738 return n;
742 ** Return open FILE * if zFile exists, can be opened for read
743 ** and is an ordinary file or a character stream source.
744 ** Otherwise return 0.
746 static FILE * openChrSource(const char *zFile){
747 #if defined(_WIN32) || defined(WIN32)
748 struct __stat64 x = {0};
749 # define STAT_CHR_SRC(mode) ((mode & (_S_IFCHR|_S_IFIFO|_S_IFREG))!=0)
750 /* On Windows, open first, then check the stream nature. This order
751 ** is necessary because _stat() and sibs, when checking a named pipe,
752 ** effectively break the pipe as its supplier sees it. */
753 FILE *rv = fopen(zFile, "rb");
754 if( rv==0 ) return 0;
755 if( _fstat64(_fileno(rv), &x) != 0
756 || !STAT_CHR_SRC(x.st_mode)){
757 fclose(rv);
758 rv = 0;
760 return rv;
761 #else
762 struct stat x = {0};
763 int rc = stat(zFile, &x);
764 # define STAT_CHR_SRC(mode) (S_ISREG(mode)||S_ISFIFO(mode)||S_ISCHR(mode))
765 if( rc!=0 ) return 0;
766 if( STAT_CHR_SRC(x.st_mode) ){
767 return fopen(zFile, "rb");
768 }else{
769 return 0;
771 #endif
772 #undef STAT_CHR_SRC
776 ** This routine reads a line of text from FILE in, stores
777 ** the text in memory obtained from malloc() and returns a pointer
778 ** to the text. NULL is returned at end of file, or if malloc()
779 ** fails.
781 ** If zLine is not NULL then it is a malloced buffer returned from
782 ** a previous call to this routine that may be reused.
784 static char *local_getline(char *zLine, FILE *in){
785 int nLine = zLine==0 ? 0 : 100;
786 int n = 0;
788 while( 1 ){
789 if( n+100>nLine ){
790 nLine = nLine*2 + 100;
791 zLine = realloc(zLine, nLine);
792 shell_check_oom(zLine);
794 if( fgets(&zLine[n], nLine - n, in)==0 ){
795 if( n==0 ){
796 free(zLine);
797 return 0;
799 zLine[n] = 0;
800 break;
802 while( zLine[n] ) n++;
803 if( n>0 && zLine[n-1]=='\n' ){
804 n--;
805 if( n>0 && zLine[n-1]=='\r' ) n--;
806 zLine[n] = 0;
807 break;
810 return zLine;
814 ** Retrieve a single line of input text.
816 ** If in==0 then read from standard input and prompt before each line.
817 ** If isContinuation is true, then a continuation prompt is appropriate.
818 ** If isContinuation is zero, then the main prompt should be used.
820 ** If zPrior is not NULL then it is a buffer from a prior call to this
821 ** routine that can be reused.
823 ** The result is stored in space obtained from malloc() and must either
824 ** be freed by the caller or else passed back into this routine via the
825 ** zPrior argument for reuse.
827 #ifndef SQLITE_SHELL_FIDDLE
828 static char *one_input_line(FILE *in, char *zPrior, int isContinuation){
829 char *zPrompt;
830 char *zResult;
831 if( in!=0 ){
832 zResult = local_getline(zPrior, in);
833 }else{
834 zPrompt = isContinuation ? CONTINUATION_PROMPT : mainPrompt;
835 #if SHELL_USE_LOCAL_GETLINE
836 sputz(stdout, zPrompt);
837 fflush(stdout);
839 zResult = local_getline(zPrior, stdin);
840 zPrior = 0;
841 /* ^C trap creates a false EOF, so let "interrupt" thread catch up. */
842 if( zResult==0 ) sqlite3_sleep(50);
843 }while( zResult==0 && seenInterrupt>0 );
844 #else
845 free(zPrior);
846 zResult = shell_readline(zPrompt);
847 while( zResult==0 ){
848 /* ^C trap creates a false EOF, so let "interrupt" thread catch up. */
849 sqlite3_sleep(50);
850 if( seenInterrupt==0 ) break;
851 zResult = shell_readline("");
853 if( zResult && *zResult ) shell_add_history(zResult);
854 #endif
856 return zResult;
858 #endif /* !SQLITE_SHELL_FIDDLE */
861 ** Return the value of a hexadecimal digit. Return -1 if the input
862 ** is not a hex digit.
864 static int hexDigitValue(char c){
865 if( c>='0' && c<='9' ) return c - '0';
866 if( c>='a' && c<='f' ) return c - 'a' + 10;
867 if( c>='A' && c<='F' ) return c - 'A' + 10;
868 return -1;
872 ** Interpret zArg as an integer value, possibly with suffixes.
874 static sqlite3_int64 integerValue(const char *zArg){
875 sqlite3_int64 v = 0;
876 static const struct { char *zSuffix; int iMult; } aMult[] = {
877 { "KiB", 1024 },
878 { "MiB", 1024*1024 },
879 { "GiB", 1024*1024*1024 },
880 { "KB", 1000 },
881 { "MB", 1000000 },
882 { "GB", 1000000000 },
883 { "K", 1000 },
884 { "M", 1000000 },
885 { "G", 1000000000 },
887 int i;
888 int isNeg = 0;
889 if( zArg[0]=='-' ){
890 isNeg = 1;
891 zArg++;
892 }else if( zArg[0]=='+' ){
893 zArg++;
895 if( zArg[0]=='0' && zArg[1]=='x' ){
896 int x;
897 zArg += 2;
898 while( (x = hexDigitValue(zArg[0]))>=0 ){
899 v = (v<<4) + x;
900 zArg++;
902 }else{
903 while( IsDigit(zArg[0]) ){
904 v = v*10 + zArg[0] - '0';
905 zArg++;
908 for(i=0; i<ArraySize(aMult); i++){
909 if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){
910 v *= aMult[i].iMult;
911 break;
914 return isNeg? -v : v;
918 ** A variable length string to which one can append text.
920 typedef struct ShellText ShellText;
921 struct ShellText {
922 char *z;
923 int n;
924 int nAlloc;
928 ** Initialize and destroy a ShellText object
930 static void initText(ShellText *p){
931 memset(p, 0, sizeof(*p));
933 static void freeText(ShellText *p){
934 free(p->z);
935 initText(p);
938 /* zIn is either a pointer to a NULL-terminated string in memory obtained
939 ** from malloc(), or a NULL pointer. The string pointed to by zAppend is
940 ** added to zIn, and the result returned in memory obtained from malloc().
941 ** zIn, if it was not NULL, is freed.
943 ** If the third argument, quote, is not '\0', then it is used as a
944 ** quote character for zAppend.
946 static void appendText(ShellText *p, const char *zAppend, char quote){
947 i64 len;
948 i64 i;
949 i64 nAppend = strlen30(zAppend);
951 len = nAppend+p->n+1;
952 if( quote ){
953 len += 2;
954 for(i=0; i<nAppend; i++){
955 if( zAppend[i]==quote ) len++;
959 if( p->z==0 || p->n+len>=p->nAlloc ){
960 p->nAlloc = p->nAlloc*2 + len + 20;
961 p->z = realloc(p->z, p->nAlloc);
962 shell_check_oom(p->z);
965 if( quote ){
966 char *zCsr = p->z+p->n;
967 *zCsr++ = quote;
968 for(i=0; i<nAppend; i++){
969 *zCsr++ = zAppend[i];
970 if( zAppend[i]==quote ) *zCsr++ = quote;
972 *zCsr++ = quote;
973 p->n = (int)(zCsr - p->z);
974 *zCsr = '\0';
975 }else{
976 memcpy(p->z+p->n, zAppend, nAppend);
977 p->n += nAppend;
978 p->z[p->n] = '\0';
983 ** Attempt to determine if identifier zName needs to be quoted, either
984 ** because it contains non-alphanumeric characters, or because it is an
985 ** SQLite keyword. Be conservative in this estimate: When in doubt assume
986 ** that quoting is required.
988 ** Return '"' if quoting is required. Return 0 if no quoting is required.
990 static char quoteChar(const char *zName){
991 int i;
992 if( zName==0 ) return '"';
993 if( !isalpha((unsigned char)zName[0]) && zName[0]!='_' ) return '"';
994 for(i=0; zName[i]; i++){
995 if( !isalnum((unsigned char)zName[i]) && zName[i]!='_' ) return '"';
997 return sqlite3_keyword_check(zName, i) ? '"' : 0;
1001 ** Construct a fake object name and column list to describe the structure
1002 ** of the view, virtual table, or table valued function zSchema.zName.
1004 static char *shellFakeSchema(
1005 sqlite3 *db, /* The database connection containing the vtab */
1006 const char *zSchema, /* Schema of the database holding the vtab */
1007 const char *zName /* The name of the virtual table */
1009 sqlite3_stmt *pStmt = 0;
1010 char *zSql;
1011 ShellText s;
1012 char cQuote;
1013 char *zDiv = "(";
1014 int nRow = 0;
1016 zSql = sqlite3_mprintf("PRAGMA \"%w\".table_info=%Q;",
1017 zSchema ? zSchema : "main", zName);
1018 shell_check_oom(zSql);
1019 sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0);
1020 sqlite3_free(zSql);
1021 initText(&s);
1022 if( zSchema ){
1023 cQuote = quoteChar(zSchema);
1024 if( cQuote && sqlite3_stricmp(zSchema,"temp")==0 ) cQuote = 0;
1025 appendText(&s, zSchema, cQuote);
1026 appendText(&s, ".", 0);
1028 cQuote = quoteChar(zName);
1029 appendText(&s, zName, cQuote);
1030 while( sqlite3_step(pStmt)==SQLITE_ROW ){
1031 const char *zCol = (const char*)sqlite3_column_text(pStmt, 1);
1032 nRow++;
1033 appendText(&s, zDiv, 0);
1034 zDiv = ",";
1035 if( zCol==0 ) zCol = "";
1036 cQuote = quoteChar(zCol);
1037 appendText(&s, zCol, cQuote);
1039 appendText(&s, ")", 0);
1040 sqlite3_finalize(pStmt);
1041 if( nRow==0 ){
1042 freeText(&s);
1043 s.z = 0;
1045 return s.z;
1049 ** SQL function: strtod(X)
1051 ** Use the C-library strtod() function to convert string X into a double.
1052 ** Used for comparing the accuracy of SQLite's internal text-to-float conversion
1053 ** routines against the C-library.
1055 static void shellStrtod(
1056 sqlite3_context *pCtx,
1057 int nVal,
1058 sqlite3_value **apVal
1060 char *z = (char*)sqlite3_value_text(apVal[0]);
1061 UNUSED_PARAMETER(nVal);
1062 if( z==0 ) return;
1063 sqlite3_result_double(pCtx, strtod(z,0));
1067 ** SQL function: dtostr(X)
1069 ** Use the C-library printf() function to convert real value X into a string.
1070 ** Used for comparing the accuracy of SQLite's internal float-to-text conversion
1071 ** routines against the C-library.
1073 static void shellDtostr(
1074 sqlite3_context *pCtx,
1075 int nVal,
1076 sqlite3_value **apVal
1078 double r = sqlite3_value_double(apVal[0]);
1079 int n = nVal>=2 ? sqlite3_value_int(apVal[1]) : 26;
1080 char z[400];
1081 if( n<1 ) n = 1;
1082 if( n>350 ) n = 350;
1083 sqlite3_snprintf(sizeof(z), z, "%#+.*e", n, r);
1084 sqlite3_result_text(pCtx, z, -1, SQLITE_TRANSIENT);
1089 ** SQL function: shell_module_schema(X)
1091 ** Return a fake schema for the table-valued function or eponymous virtual
1092 ** table X.
1094 static void shellModuleSchema(
1095 sqlite3_context *pCtx,
1096 int nVal,
1097 sqlite3_value **apVal
1099 const char *zName;
1100 char *zFake;
1101 UNUSED_PARAMETER(nVal);
1102 zName = (const char*)sqlite3_value_text(apVal[0]);
1103 zFake = zName? shellFakeSchema(sqlite3_context_db_handle(pCtx), 0, zName) : 0;
1104 if( zFake ){
1105 sqlite3_result_text(pCtx, sqlite3_mprintf("/* %s */", zFake),
1106 -1, sqlite3_free);
1107 free(zFake);
1112 ** SQL function: shell_add_schema(S,X)
1114 ** Add the schema name X to the CREATE statement in S and return the result.
1115 ** Examples:
1117 ** CREATE TABLE t1(x) -> CREATE TABLE xyz.t1(x);
1119 ** Also works on
1121 ** CREATE INDEX
1122 ** CREATE UNIQUE INDEX
1123 ** CREATE VIEW
1124 ** CREATE TRIGGER
1125 ** CREATE VIRTUAL TABLE
1127 ** This UDF is used by the .schema command to insert the schema name of
1128 ** attached databases into the middle of the sqlite_schema.sql field.
1130 static void shellAddSchemaName(
1131 sqlite3_context *pCtx,
1132 int nVal,
1133 sqlite3_value **apVal
1135 static const char *aPrefix[] = {
1136 "TABLE",
1137 "INDEX",
1138 "UNIQUE INDEX",
1139 "VIEW",
1140 "TRIGGER",
1141 "VIRTUAL TABLE"
1143 int i = 0;
1144 const char *zIn = (const char*)sqlite3_value_text(apVal[0]);
1145 const char *zSchema = (const char*)sqlite3_value_text(apVal[1]);
1146 const char *zName = (const char*)sqlite3_value_text(apVal[2]);
1147 sqlite3 *db = sqlite3_context_db_handle(pCtx);
1148 UNUSED_PARAMETER(nVal);
1149 if( zIn!=0 && cli_strncmp(zIn, "CREATE ", 7)==0 ){
1150 for(i=0; i<ArraySize(aPrefix); i++){
1151 int n = strlen30(aPrefix[i]);
1152 if( cli_strncmp(zIn+7, aPrefix[i], n)==0 && zIn[n+7]==' ' ){
1153 char *z = 0;
1154 char *zFake = 0;
1155 if( zSchema ){
1156 char cQuote = quoteChar(zSchema);
1157 if( cQuote && sqlite3_stricmp(zSchema,"temp")!=0 ){
1158 z = sqlite3_mprintf("%.*s \"%w\".%s", n+7, zIn, zSchema, zIn+n+8);
1159 }else{
1160 z = sqlite3_mprintf("%.*s %s.%s", n+7, zIn, zSchema, zIn+n+8);
1163 if( zName
1164 && aPrefix[i][0]=='V'
1165 && (zFake = shellFakeSchema(db, zSchema, zName))!=0
1167 if( z==0 ){
1168 z = sqlite3_mprintf("%s\n/* %s */", zIn, zFake);
1169 }else{
1170 z = sqlite3_mprintf("%z\n/* %s */", z, zFake);
1172 free(zFake);
1174 if( z ){
1175 sqlite3_result_text(pCtx, z, -1, sqlite3_free);
1176 return;
1181 sqlite3_result_value(pCtx, apVal[0]);
1185 ** The source code for several run-time loadable extensions is inserted
1186 ** below by the ../tool/mkshellc.tcl script. Before processing that included
1187 ** code, we need to override some macros to make the included program code
1188 ** work here in the middle of this regular program.
1190 #define SQLITE_EXTENSION_INIT1
1191 #define SQLITE_EXTENSION_INIT2(X) (void)(X)
1193 #if defined(_WIN32) && defined(_MSC_VER)
1194 INCLUDE test_windirent.h
1195 INCLUDE test_windirent.c
1196 #define dirent DIRENT
1197 #endif
1198 INCLUDE ../ext/misc/memtrace.c
1199 INCLUDE ../ext/misc/pcachetrace.c
1200 INCLUDE ../ext/misc/shathree.c
1201 INCLUDE ../ext/misc/uint.c
1202 INCLUDE ../ext/misc/decimal.c
1203 #undef sqlite3_base_init
1204 #define sqlite3_base_init sqlite3_base64_init
1205 INCLUDE ../ext/misc/base64.c
1206 #undef sqlite3_base_init
1207 #define sqlite3_base_init sqlite3_base85_init
1208 #define OMIT_BASE85_CHECKER
1209 INCLUDE ../ext/misc/base85.c
1210 INCLUDE ../ext/misc/ieee754.c
1211 INCLUDE ../ext/misc/series.c
1212 INCLUDE ../ext/misc/regexp.c
1213 #ifndef SQLITE_SHELL_FIDDLE
1214 INCLUDE ../ext/misc/fileio.c
1215 INCLUDE ../ext/misc/completion.c
1216 INCLUDE ../ext/misc/appendvfs.c
1217 #endif
1218 #ifdef SQLITE_HAVE_ZLIB
1219 INCLUDE ../ext/misc/zipfile.c
1220 INCLUDE ../ext/misc/sqlar.c
1221 #endif
1222 INCLUDE ../ext/expert/sqlite3expert.h
1223 INCLUDE ../ext/expert/sqlite3expert.c
1225 INCLUDE ../ext/intck/sqlite3intck.h
1226 INCLUDE ../ext/intck/sqlite3intck.c
1228 #if !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_ENABLE_DBPAGE_VTAB)
1229 #define SQLITE_SHELL_HAVE_RECOVER 1
1230 #else
1231 #define SQLITE_SHELL_HAVE_RECOVER 0
1232 #endif
1233 #if SQLITE_SHELL_HAVE_RECOVER
1234 INCLUDE ../ext/recover/sqlite3recover.h
1235 # ifndef SQLITE_HAVE_SQLITE3R
1236 INCLUDE ../ext/recover/dbdata.c
1237 INCLUDE ../ext/recover/sqlite3recover.c
1238 # endif /* SQLITE_HAVE_SQLITE3R */
1239 #endif
1240 #ifdef SQLITE_SHELL_EXTSRC
1241 # include SHELL_STRINGIFY(SQLITE_SHELL_EXTSRC)
1242 #endif
1244 #if defined(SQLITE_ENABLE_SESSION)
1246 ** State information for a single open session
1248 typedef struct OpenSession OpenSession;
1249 struct OpenSession {
1250 char *zName; /* Symbolic name for this session */
1251 int nFilter; /* Number of xFilter rejection GLOB patterns */
1252 char **azFilter; /* Array of xFilter rejection GLOB patterns */
1253 sqlite3_session *p; /* The open session */
1255 #endif
1257 typedef struct ExpertInfo ExpertInfo;
1258 struct ExpertInfo {
1259 sqlite3expert *pExpert;
1260 int bVerbose;
1263 /* A single line in the EQP output */
1264 typedef struct EQPGraphRow EQPGraphRow;
1265 struct EQPGraphRow {
1266 int iEqpId; /* ID for this row */
1267 int iParentId; /* ID of the parent row */
1268 EQPGraphRow *pNext; /* Next row in sequence */
1269 char zText[1]; /* Text to display for this row */
1272 /* All EQP output is collected into an instance of the following */
1273 typedef struct EQPGraph EQPGraph;
1274 struct EQPGraph {
1275 EQPGraphRow *pRow; /* Linked list of all rows of the EQP output */
1276 EQPGraphRow *pLast; /* Last element of the pRow list */
1277 char zPrefix[100]; /* Graph prefix */
1280 /* Parameters affecting columnar mode result display (defaulting together) */
1281 typedef struct ColModeOpts {
1282 int iWrap; /* In columnar modes, wrap lines reaching this limit */
1283 u8 bQuote; /* Quote results for .mode box and table */
1284 u8 bWordWrap; /* In columnar modes, wrap at word boundaries */
1285 } ColModeOpts;
1286 #define ColModeOpts_default { 60, 0, 0 }
1287 #define ColModeOpts_default_qbox { 60, 1, 0 }
1290 ** State information about the database connection is contained in an
1291 ** instance of the following structure.
1293 typedef struct ShellState ShellState;
1294 struct ShellState {
1295 sqlite3 *db; /* The database */
1296 u8 autoExplain; /* Automatically turn on .explain mode */
1297 u8 autoEQP; /* Run EXPLAIN QUERY PLAN prior to each SQL stmt */
1298 u8 autoEQPtest; /* autoEQP is in test mode */
1299 u8 autoEQPtrace; /* autoEQP is in trace mode */
1300 u8 scanstatsOn; /* True to display scan stats before each finalize */
1301 u8 openMode; /* SHELL_OPEN_NORMAL, _APPENDVFS, or _ZIPFILE */
1302 u8 doXdgOpen; /* Invoke start/open/xdg-open in output_reset() */
1303 u8 nEqpLevel; /* Depth of the EQP output graph */
1304 u8 eTraceType; /* SHELL_TRACE_* value for type of trace */
1305 u8 bSafeMode; /* True to prohibit unsafe operations */
1306 u8 bSafeModePersist; /* The long-term value of bSafeMode */
1307 u8 eRestoreState; /* See comments above doAutoDetectRestore() */
1308 ColModeOpts cmOpts; /* Option values affecting columnar mode output */
1309 unsigned statsOn; /* True to display memory stats before each finalize */
1310 unsigned mEqpLines; /* Mask of vertical lines in the EQP output graph */
1311 int inputNesting; /* Track nesting level of .read and other redirects */
1312 int outCount; /* Revert to stdout when reaching zero */
1313 int cnt; /* Number of records displayed so far */
1314 int lineno; /* Line number of last line read from in */
1315 int openFlags; /* Additional flags to open. (SQLITE_OPEN_NOFOLLOW) */
1316 FILE *in; /* Read commands from this stream */
1317 FILE *out; /* Write results here */
1318 FILE *traceOut; /* Output for sqlite3_trace() */
1319 int nErr; /* Number of errors seen */
1320 int mode; /* An output mode setting */
1321 int modePrior; /* Saved mode */
1322 int cMode; /* temporary output mode for the current query */
1323 int normalMode; /* Output mode before ".explain on" */
1324 int writableSchema; /* True if PRAGMA writable_schema=ON */
1325 int showHeader; /* True to show column names in List or Column mode */
1326 int nCheck; /* Number of ".check" commands run */
1327 unsigned nProgress; /* Number of progress callbacks encountered */
1328 unsigned mxProgress; /* Maximum progress callbacks before failing */
1329 unsigned flgProgress; /* Flags for the progress callback */
1330 unsigned shellFlgs; /* Various flags */
1331 unsigned priorShFlgs; /* Saved copy of flags */
1332 sqlite3_int64 szMax; /* --maxsize argument to .open */
1333 char *zDestTable; /* Name of destination table when MODE_Insert */
1334 char *zTempFile; /* Temporary file that might need deleting */
1335 char zTestcase[30]; /* Name of current test case */
1336 char colSeparator[20]; /* Column separator character for several modes */
1337 char rowSeparator[20]; /* Row separator character for MODE_Ascii */
1338 char colSepPrior[20]; /* Saved column separator */
1339 char rowSepPrior[20]; /* Saved row separator */
1340 int *colWidth; /* Requested width of each column in columnar modes */
1341 int *actualWidth; /* Actual width of each column */
1342 int nWidth; /* Number of slots in colWidth[] and actualWidth[] */
1343 char nullValue[20]; /* The text to print when a NULL comes back from
1344 ** the database */
1345 char outfile[FILENAME_MAX]; /* Filename for *out */
1346 sqlite3_stmt *pStmt; /* Current statement if any. */
1347 FILE *pLog; /* Write log output here */
1348 struct AuxDb { /* Storage space for auxiliary database connections */
1349 sqlite3 *db; /* Connection pointer */
1350 const char *zDbFilename; /* Filename used to open the connection */
1351 char *zFreeOnClose; /* Free this memory allocation on close */
1352 #if defined(SQLITE_ENABLE_SESSION)
1353 int nSession; /* Number of active sessions */
1354 OpenSession aSession[4]; /* Array of sessions. [0] is in focus. */
1355 #endif
1356 } aAuxDb[5], /* Array of all database connections */
1357 *pAuxDb; /* Currently active database connection */
1358 int *aiIndent; /* Array of indents used in MODE_Explain */
1359 int nIndent; /* Size of array aiIndent[] */
1360 int iIndent; /* Index of current op in aiIndent[] */
1361 char *zNonce; /* Nonce for temporary safe-mode escapes */
1362 EQPGraph sGraph; /* Information for the graphical EXPLAIN QUERY PLAN */
1363 ExpertInfo expert; /* Valid if previous command was ".expert OPT..." */
1364 #ifdef SQLITE_SHELL_FIDDLE
1365 struct {
1366 const char * zInput; /* Input string from wasm/JS proxy */
1367 const char * zPos; /* Cursor pos into zInput */
1368 const char * zDefaultDbName; /* Default name for db file */
1369 } wasm;
1370 #endif
1373 #ifdef SQLITE_SHELL_FIDDLE
1374 static ShellState shellState;
1375 #endif
1378 /* Allowed values for ShellState.autoEQP
1380 #define AUTOEQP_off 0 /* Automatic EXPLAIN QUERY PLAN is off */
1381 #define AUTOEQP_on 1 /* Automatic EQP is on */
1382 #define AUTOEQP_trigger 2 /* On and also show plans for triggers */
1383 #define AUTOEQP_full 3 /* Show full EXPLAIN */
1385 /* Allowed values for ShellState.openMode
1387 #define SHELL_OPEN_UNSPEC 0 /* No open-mode specified */
1388 #define SHELL_OPEN_NORMAL 1 /* Normal database file */
1389 #define SHELL_OPEN_APPENDVFS 2 /* Use appendvfs */
1390 #define SHELL_OPEN_ZIPFILE 3 /* Use the zipfile virtual table */
1391 #define SHELL_OPEN_READONLY 4 /* Open a normal database read-only */
1392 #define SHELL_OPEN_DESERIALIZE 5 /* Open using sqlite3_deserialize() */
1393 #define SHELL_OPEN_HEXDB 6 /* Use "dbtotxt" output as data source */
1395 /* Allowed values for ShellState.eTraceType
1397 #define SHELL_TRACE_PLAIN 0 /* Show input SQL text */
1398 #define SHELL_TRACE_EXPANDED 1 /* Show expanded SQL text */
1399 #define SHELL_TRACE_NORMALIZED 2 /* Show normalized SQL text */
1401 /* Bits in the ShellState.flgProgress variable */
1402 #define SHELL_PROGRESS_QUIET 0x01 /* Omit announcing every progress callback */
1403 #define SHELL_PROGRESS_RESET 0x02 /* Reset the count when the progress
1404 ** callback limit is reached, and for each
1405 ** top-level SQL statement */
1406 #define SHELL_PROGRESS_ONCE 0x04 /* Cancel the --limit after firing once */
1409 ** These are the allowed shellFlgs values
1411 #define SHFLG_Pagecache 0x00000001 /* The --pagecache option is used */
1412 #define SHFLG_Lookaside 0x00000002 /* Lookaside memory is used */
1413 #define SHFLG_Backslash 0x00000004 /* The --backslash option is used */
1414 #define SHFLG_PreserveRowid 0x00000008 /* .dump preserves rowid values */
1415 #define SHFLG_Newlines 0x00000010 /* .dump --newline flag */
1416 #define SHFLG_CountChanges 0x00000020 /* .changes setting */
1417 #define SHFLG_Echo 0x00000040 /* .echo on/off, or --echo setting */
1418 #define SHFLG_HeaderSet 0x00000080 /* showHeader has been specified */
1419 #define SHFLG_DumpDataOnly 0x00000100 /* .dump show data only */
1420 #define SHFLG_DumpNoSys 0x00000200 /* .dump omits system tables */
1421 #define SHFLG_TestingMode 0x00000400 /* allow unsafe testing features */
1424 ** Macros for testing and setting shellFlgs
1426 #define ShellHasFlag(P,X) (((P)->shellFlgs & (X))!=0)
1427 #define ShellSetFlag(P,X) ((P)->shellFlgs|=(X))
1428 #define ShellClearFlag(P,X) ((P)->shellFlgs&=(~(X)))
1431 ** These are the allowed modes.
1433 #define MODE_Line 0 /* One column per line. Blank line between records */
1434 #define MODE_Column 1 /* One record per line in neat columns */
1435 #define MODE_List 2 /* One record per line with a separator */
1436 #define MODE_Semi 3 /* Same as MODE_List but append ";" to each line */
1437 #define MODE_Html 4 /* Generate an XHTML table */
1438 #define MODE_Insert 5 /* Generate SQL "insert" statements */
1439 #define MODE_Quote 6 /* Quote values as for SQL */
1440 #define MODE_Tcl 7 /* Generate ANSI-C or TCL quoted elements */
1441 #define MODE_Csv 8 /* Quote strings, numbers are plain */
1442 #define MODE_Explain 9 /* Like MODE_Column, but do not truncate data */
1443 #define MODE_Ascii 10 /* Use ASCII unit and record separators (0x1F/0x1E) */
1444 #define MODE_Pretty 11 /* Pretty-print schemas */
1445 #define MODE_EQP 12 /* Converts EXPLAIN QUERY PLAN output into a graph */
1446 #define MODE_Json 13 /* Output JSON */
1447 #define MODE_Markdown 14 /* Markdown formatting */
1448 #define MODE_Table 15 /* MySQL-style table formatting */
1449 #define MODE_Box 16 /* Unicode box-drawing characters */
1450 #define MODE_Count 17 /* Output only a count of the rows of output */
1451 #define MODE_Off 18 /* No query output shown */
1452 #define MODE_ScanExp 19 /* Like MODE_Explain, but for ".scanstats vm" */
1454 static const char *modeDescr[] = {
1455 "line",
1456 "column",
1457 "list",
1458 "semi",
1459 "html",
1460 "insert",
1461 "quote",
1462 "tcl",
1463 "csv",
1464 "explain",
1465 "ascii",
1466 "prettyprint",
1467 "eqp",
1468 "json",
1469 "markdown",
1470 "table",
1471 "box",
1472 "count",
1473 "off"
1477 ** These are the column/row/line separators used by the various
1478 ** import/export modes.
1480 #define SEP_Column "|"
1481 #define SEP_Row "\n"
1482 #define SEP_Tab "\t"
1483 #define SEP_Space " "
1484 #define SEP_Comma ","
1485 #define SEP_CrLf "\r\n"
1486 #define SEP_Unit "\x1F"
1487 #define SEP_Record "\x1E"
1490 ** Limit input nesting via .read or any other input redirect.
1491 ** It's not too expensive, so a generous allowance can be made.
1493 #define MAX_INPUT_NESTING 25
1496 ** A callback for the sqlite3_log() interface.
1498 static void shellLog(void *pArg, int iErrCode, const char *zMsg){
1499 ShellState *p = (ShellState*)pArg;
1500 if( p->pLog==0 ) return;
1501 sputf(p->pLog, "(%d) %s\n", iErrCode, zMsg);
1502 fflush(p->pLog);
1506 ** SQL function: shell_putsnl(X)
1508 ** Write the text X to the screen (or whatever output is being directed)
1509 ** adding a newline at the end, and then return X.
1511 static void shellPutsFunc(
1512 sqlite3_context *pCtx,
1513 int nVal,
1514 sqlite3_value **apVal
1516 /* Unused: (ShellState*)sqlite3_user_data(pCtx); */
1517 (void)nVal;
1518 oputf("%s\n", sqlite3_value_text(apVal[0]));
1519 sqlite3_result_value(pCtx, apVal[0]);
1523 ** If in safe mode, print an error message described by the arguments
1524 ** and exit immediately.
1526 static void failIfSafeMode(
1527 ShellState *p,
1528 const char *zErrMsg,
1531 if( p->bSafeMode ){
1532 va_list ap;
1533 char *zMsg;
1534 va_start(ap, zErrMsg);
1535 zMsg = sqlite3_vmprintf(zErrMsg, ap);
1536 va_end(ap);
1537 eputf("line %d: %s\n", p->lineno, zMsg);
1538 exit(1);
1543 ** SQL function: edit(VALUE)
1544 ** edit(VALUE,EDITOR)
1546 ** These steps:
1548 ** (1) Write VALUE into a temporary file.
1549 ** (2) Run program EDITOR on that temporary file.
1550 ** (3) Read the temporary file back and return its content as the result.
1551 ** (4) Delete the temporary file
1553 ** If the EDITOR argument is omitted, use the value in the VISUAL
1554 ** environment variable. If still there is no EDITOR, through an error.
1556 ** Also throw an error if the EDITOR program returns a non-zero exit code.
1558 #ifndef SQLITE_NOHAVE_SYSTEM
1559 static void editFunc(
1560 sqlite3_context *context,
1561 int argc,
1562 sqlite3_value **argv
1564 const char *zEditor;
1565 char *zTempFile = 0;
1566 sqlite3 *db;
1567 char *zCmd = 0;
1568 int bBin;
1569 int rc;
1570 int hasCRNL = 0;
1571 FILE *f = 0;
1572 sqlite3_int64 sz;
1573 sqlite3_int64 x;
1574 unsigned char *p = 0;
1576 if( argc==2 ){
1577 zEditor = (const char*)sqlite3_value_text(argv[1]);
1578 }else{
1579 zEditor = getenv("VISUAL");
1581 if( zEditor==0 ){
1582 sqlite3_result_error(context, "no editor for edit()", -1);
1583 return;
1585 if( sqlite3_value_type(argv[0])==SQLITE_NULL ){
1586 sqlite3_result_error(context, "NULL input to edit()", -1);
1587 return;
1589 db = sqlite3_context_db_handle(context);
1590 zTempFile = 0;
1591 sqlite3_file_control(db, 0, SQLITE_FCNTL_TEMPFILENAME, &zTempFile);
1592 if( zTempFile==0 ){
1593 sqlite3_uint64 r = 0;
1594 sqlite3_randomness(sizeof(r), &r);
1595 zTempFile = sqlite3_mprintf("temp%llx", r);
1596 if( zTempFile==0 ){
1597 sqlite3_result_error_nomem(context);
1598 return;
1601 bBin = sqlite3_value_type(argv[0])==SQLITE_BLOB;
1602 /* When writing the file to be edited, do \n to \r\n conversions on systems
1603 ** that want \r\n line endings */
1604 f = fopen(zTempFile, bBin ? "wb" : "w");
1605 if( f==0 ){
1606 sqlite3_result_error(context, "edit() cannot open temp file", -1);
1607 goto edit_func_end;
1609 sz = sqlite3_value_bytes(argv[0]);
1610 if( bBin ){
1611 x = fwrite(sqlite3_value_blob(argv[0]), 1, (size_t)sz, f);
1612 }else{
1613 const char *z = (const char*)sqlite3_value_text(argv[0]);
1614 /* Remember whether or not the value originally contained \r\n */
1615 if( z && strstr(z,"\r\n")!=0 ) hasCRNL = 1;
1616 x = fwrite(sqlite3_value_text(argv[0]), 1, (size_t)sz, f);
1618 fclose(f);
1619 f = 0;
1620 if( x!=sz ){
1621 sqlite3_result_error(context, "edit() could not write the whole file", -1);
1622 goto edit_func_end;
1624 zCmd = sqlite3_mprintf("%s \"%s\"", zEditor, zTempFile);
1625 if( zCmd==0 ){
1626 sqlite3_result_error_nomem(context);
1627 goto edit_func_end;
1629 rc = system(zCmd);
1630 sqlite3_free(zCmd);
1631 if( rc ){
1632 sqlite3_result_error(context, "EDITOR returned non-zero", -1);
1633 goto edit_func_end;
1635 f = fopen(zTempFile, "rb");
1636 if( f==0 ){
1637 sqlite3_result_error(context,
1638 "edit() cannot reopen temp file after edit", -1);
1639 goto edit_func_end;
1641 fseek(f, 0, SEEK_END);
1642 sz = ftell(f);
1643 rewind(f);
1644 p = sqlite3_malloc64( sz+1 );
1645 if( p==0 ){
1646 sqlite3_result_error_nomem(context);
1647 goto edit_func_end;
1649 x = fread(p, 1, (size_t)sz, f);
1650 fclose(f);
1651 f = 0;
1652 if( x!=sz ){
1653 sqlite3_result_error(context, "could not read back the whole file", -1);
1654 goto edit_func_end;
1656 if( bBin ){
1657 sqlite3_result_blob64(context, p, sz, sqlite3_free);
1658 }else{
1659 sqlite3_int64 i, j;
1660 if( hasCRNL ){
1661 /* If the original contains \r\n then do no conversions back to \n */
1662 }else{
1663 /* If the file did not originally contain \r\n then convert any new
1664 ** \r\n back into \n */
1665 p[sz] = 0;
1666 for(i=j=0; i<sz; i++){
1667 if( p[i]=='\r' && p[i+1]=='\n' ) i++;
1668 p[j++] = p[i];
1670 sz = j;
1671 p[sz] = 0;
1673 sqlite3_result_text64(context, (const char*)p, sz,
1674 sqlite3_free, SQLITE_UTF8);
1676 p = 0;
1678 edit_func_end:
1679 if( f ) fclose(f);
1680 unlink(zTempFile);
1681 sqlite3_free(zTempFile);
1682 sqlite3_free(p);
1684 #endif /* SQLITE_NOHAVE_SYSTEM */
1687 ** Save or restore the current output mode
1689 static void outputModePush(ShellState *p){
1690 p->modePrior = p->mode;
1691 p->priorShFlgs = p->shellFlgs;
1692 memcpy(p->colSepPrior, p->colSeparator, sizeof(p->colSeparator));
1693 memcpy(p->rowSepPrior, p->rowSeparator, sizeof(p->rowSeparator));
1695 static void outputModePop(ShellState *p){
1696 p->mode = p->modePrior;
1697 p->shellFlgs = p->priorShFlgs;
1698 memcpy(p->colSeparator, p->colSepPrior, sizeof(p->colSeparator));
1699 memcpy(p->rowSeparator, p->rowSepPrior, sizeof(p->rowSeparator));
1703 ** Output the given string as a hex-encoded blob (eg. X'1234' )
1705 static void output_hex_blob(const void *pBlob, int nBlob){
1706 int i;
1707 unsigned char *aBlob = (unsigned char*)pBlob;
1709 char *zStr = sqlite3_malloc(nBlob*2 + 1);
1710 shell_check_oom(zStr);
1712 for(i=0; i<nBlob; i++){
1713 static const char aHex[] = {
1714 '0', '1', '2', '3', '4', '5', '6', '7',
1715 '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'
1717 zStr[i*2] = aHex[ (aBlob[i] >> 4) ];
1718 zStr[i*2+1] = aHex[ (aBlob[i] & 0x0F) ];
1720 zStr[i*2] = '\0';
1722 oputf("X'%s'", zStr);
1723 sqlite3_free(zStr);
1727 ** Find a string that is not found anywhere in z[]. Return a pointer
1728 ** to that string.
1730 ** Try to use zA and zB first. If both of those are already found in z[]
1731 ** then make up some string and store it in the buffer zBuf.
1733 static const char *unused_string(
1734 const char *z, /* Result must not appear anywhere in z */
1735 const char *zA, const char *zB, /* Try these first */
1736 char *zBuf /* Space to store a generated string */
1738 unsigned i = 0;
1739 if( strstr(z, zA)==0 ) return zA;
1740 if( strstr(z, zB)==0 ) return zB;
1742 sqlite3_snprintf(20,zBuf,"(%s%u)", zA, i++);
1743 }while( strstr(z,zBuf)!=0 );
1744 return zBuf;
1748 ** Output the given string as a quoted string using SQL quoting conventions.
1750 ** See also: output_quoted_escaped_string()
1752 static void output_quoted_string(const char *z){
1753 int i;
1754 char c;
1755 #ifndef SQLITE_SHELL_FIDDLE
1756 FILE *pfO = setOutputStream(invalidFileStream);
1757 setBinaryMode(pfO, 1);
1758 #endif
1759 if( z==0 ) return;
1760 for(i=0; (c = z[i])!=0 && c!='\''; i++){}
1761 if( c==0 ){
1762 oputf("'%s'",z);
1763 }else{
1764 oputz("'");
1765 while( *z ){
1766 for(i=0; (c = z[i])!=0 && c!='\''; i++){}
1767 if( c=='\'' ) i++;
1768 if( i ){
1769 oputf("%.*s", i, z);
1770 z += i;
1772 if( c=='\'' ){
1773 oputz("'");
1774 continue;
1776 if( c==0 ){
1777 break;
1779 z++;
1781 oputz("'");
1783 #ifndef SQLITE_SHELL_FIDDLE
1784 setTextMode(pfO, 1);
1785 #else
1786 setTextMode(stdout, 1);
1787 #endif
1791 ** Output the given string as a quoted string using SQL quoting conventions.
1792 ** Additionallly , escape the "\n" and "\r" characters so that they do not
1793 ** get corrupted by end-of-line translation facilities in some operating
1794 ** systems.
1796 ** This is like output_quoted_string() but with the addition of the \r\n
1797 ** escape mechanism.
1799 static void output_quoted_escaped_string(const char *z){
1800 int i;
1801 char c;
1802 #ifndef SQLITE_SHELL_FIDDLE
1803 FILE *pfO = setOutputStream(invalidFileStream);
1804 setBinaryMode(pfO, 1);
1805 #endif
1806 for(i=0; (c = z[i])!=0 && c!='\'' && c!='\n' && c!='\r'; i++){}
1807 if( c==0 ){
1808 oputf("'%s'",z);
1809 }else{
1810 const char *zNL = 0;
1811 const char *zCR = 0;
1812 int nNL = 0;
1813 int nCR = 0;
1814 char zBuf1[20], zBuf2[20];
1815 for(i=0; z[i]; i++){
1816 if( z[i]=='\n' ) nNL++;
1817 if( z[i]=='\r' ) nCR++;
1819 if( nNL ){
1820 oputz("replace(");
1821 zNL = unused_string(z, "\\n", "\\012", zBuf1);
1823 if( nCR ){
1824 oputz("replace(");
1825 zCR = unused_string(z, "\\r", "\\015", zBuf2);
1827 oputz("'");
1828 while( *z ){
1829 for(i=0; (c = z[i])!=0 && c!='\n' && c!='\r' && c!='\''; i++){}
1830 if( c=='\'' ) i++;
1831 if( i ){
1832 oputf("%.*s", i, z);
1833 z += i;
1835 if( c=='\'' ){
1836 oputz("'");
1837 continue;
1839 if( c==0 ){
1840 break;
1842 z++;
1843 if( c=='\n' ){
1844 oputz(zNL);
1845 continue;
1847 oputz(zCR);
1849 oputz("'");
1850 if( nCR ){
1851 oputf(",'%s',char(13))", zCR);
1853 if( nNL ){
1854 oputf(",'%s',char(10))", zNL);
1857 #ifndef SQLITE_SHELL_FIDDLE
1858 setTextMode(pfO, 1);
1859 #else
1860 setTextMode(stdout, 1);
1861 #endif
1865 ** Find earliest of chars within s specified in zAny.
1866 ** With ns == ~0, is like strpbrk(s,zAny) and s must be 0-terminated.
1868 static const char *anyOfInStr(const char *s, const char *zAny, size_t ns){
1869 const char *pcFirst = 0;
1870 if( ns == ~(size_t)0 ) ns = strlen(s);
1871 while(*zAny){
1872 const char *pc = (const char*)memchr(s, *zAny&0xff, ns);
1873 if( pc ){
1874 pcFirst = pc;
1875 ns = pcFirst - s;
1877 ++zAny;
1879 return pcFirst;
1882 ** Output the given string as a quoted according to C or TCL quoting rules.
1884 static void output_c_string(const char *z){
1885 char c;
1886 static const char *zq = "\"";
1887 static long ctrlMask = ~0L;
1888 static const char *zDQBSRO = "\"\\\x7f"; /* double-quote, backslash, rubout */
1889 char ace[3] = "\\?";
1890 char cbsSay;
1891 oputz(zq);
1892 while( *z!=0 ){
1893 const char *pcDQBSRO = anyOfInStr(z, zDQBSRO, ~(size_t)0);
1894 const char *pcPast = zSkipValidUtf8(z, INT_MAX, ctrlMask);
1895 const char *pcEnd = (pcDQBSRO && pcDQBSRO < pcPast)? pcDQBSRO : pcPast;
1896 if( pcEnd > z ) oputb(z, (int)(pcEnd-z));
1897 if( (c = *pcEnd)==0 ) break;
1898 ++pcEnd;
1899 switch( c ){
1900 case '\\': case '"':
1901 cbsSay = (char)c;
1902 break;
1903 case '\t': cbsSay = 't'; break;
1904 case '\n': cbsSay = 'n'; break;
1905 case '\r': cbsSay = 'r'; break;
1906 case '\f': cbsSay = 'f'; break;
1907 default: cbsSay = 0; break;
1909 if( cbsSay ){
1910 ace[1] = cbsSay;
1911 oputz(ace);
1912 }else if( !isprint(c&0xff) ){
1913 oputf("\\%03o", c&0xff);
1914 }else{
1915 ace[1] = (char)c;
1916 oputz(ace+1);
1918 z = pcEnd;
1920 oputz(zq);
1924 ** Output the given string as a quoted according to JSON quoting rules.
1926 static void output_json_string(const char *z, i64 n){
1927 char c;
1928 static const char *zq = "\"";
1929 static long ctrlMask = ~0L;
1930 static const char *zDQBS = "\"\\";
1931 const char *pcLimit;
1932 char ace[3] = "\\?";
1933 char cbsSay;
1935 if( z==0 ) z = "";
1936 pcLimit = z + ((n<0)? strlen(z) : (size_t)n);
1937 oputz(zq);
1938 while( z < pcLimit ){
1939 const char *pcDQBS = anyOfInStr(z, zDQBS, pcLimit-z);
1940 const char *pcPast = zSkipValidUtf8(z, (int)(pcLimit-z), ctrlMask);
1941 const char *pcEnd = (pcDQBS && pcDQBS < pcPast)? pcDQBS : pcPast;
1942 if( pcEnd > z ){
1943 oputb(z, (int)(pcEnd-z));
1944 z = pcEnd;
1946 if( z >= pcLimit ) break;
1947 c = *(z++);
1948 switch( c ){
1949 case '"': case '\\':
1950 cbsSay = (char)c;
1951 break;
1952 case '\b': cbsSay = 'b'; break;
1953 case '\f': cbsSay = 'f'; break;
1954 case '\n': cbsSay = 'n'; break;
1955 case '\r': cbsSay = 'r'; break;
1956 case '\t': cbsSay = 't'; break;
1957 default: cbsSay = 0; break;
1959 if( cbsSay ){
1960 ace[1] = cbsSay;
1961 oputz(ace);
1962 }else if( c<=0x1f ){
1963 oputf("u%04x", c);
1964 }else{
1965 ace[1] = (char)c;
1966 oputz(ace+1);
1969 oputz(zq);
1973 ** Output the given string with characters that are special to
1974 ** HTML escaped.
1976 static void output_html_string(const char *z){
1977 int i;
1978 if( z==0 ) z = "";
1979 while( *z ){
1980 for(i=0; z[i]
1981 && z[i]!='<'
1982 && z[i]!='&'
1983 && z[i]!='>'
1984 && z[i]!='\"'
1985 && z[i]!='\'';
1986 i++){}
1987 if( i>0 ){
1988 oputf("%.*s",i,z);
1990 if( z[i]=='<' ){
1991 oputz("&lt;");
1992 }else if( z[i]=='&' ){
1993 oputz("&amp;");
1994 }else if( z[i]=='>' ){
1995 oputz("&gt;");
1996 }else if( z[i]=='\"' ){
1997 oputz("&quot;");
1998 }else if( z[i]=='\'' ){
1999 oputz("&#39;");
2000 }else{
2001 break;
2003 z += i + 1;
2008 ** If a field contains any character identified by a 1 in the following
2009 ** array, then the string must be quoted for CSV.
2011 static const char needCsvQuote[] = {
2012 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2013 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2014 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,
2015 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2016 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2017 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2018 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
2019 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
2020 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2021 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2022 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2023 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2024 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2025 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2026 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2027 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
2031 ** Output a single term of CSV. Actually, p->colSeparator is used for
2032 ** the separator, which may or may not be a comma. p->nullValue is
2033 ** the null value. Strings are quoted if necessary. The separator
2034 ** is only issued if bSep is true.
2036 static void output_csv(ShellState *p, const char *z, int bSep){
2037 if( z==0 ){
2038 oputf("%s",p->nullValue);
2039 }else{
2040 unsigned i;
2041 for(i=0; z[i]; i++){
2042 if( needCsvQuote[((unsigned char*)z)[i]] ){
2043 i = 0;
2044 break;
2047 if( i==0 || strstr(z, p->colSeparator)!=0 ){
2048 char *zQuoted = sqlite3_mprintf("\"%w\"", z);
2049 shell_check_oom(zQuoted);
2050 oputz(zQuoted);
2051 sqlite3_free(zQuoted);
2052 }else{
2053 oputz(z);
2056 if( bSep ){
2057 oputz(p->colSeparator);
2062 ** This routine runs when the user presses Ctrl-C
2064 static void interrupt_handler(int NotUsed){
2065 UNUSED_PARAMETER(NotUsed);
2066 if( ++seenInterrupt>1 ) exit(1);
2067 if( globalDb ) sqlite3_interrupt(globalDb);
2070 #if (defined(_WIN32) || defined(WIN32)) && !defined(_WIN32_WCE)
2072 ** This routine runs for console events (e.g. Ctrl-C) on Win32
2074 static BOOL WINAPI ConsoleCtrlHandler(
2075 DWORD dwCtrlType /* One of the CTRL_*_EVENT constants */
2077 if( dwCtrlType==CTRL_C_EVENT ){
2078 interrupt_handler(0);
2079 return TRUE;
2081 return FALSE;
2083 #endif
2085 #ifndef SQLITE_OMIT_AUTHORIZATION
2087 ** This authorizer runs in safe mode.
2089 static int safeModeAuth(
2090 void *pClientData,
2091 int op,
2092 const char *zA1,
2093 const char *zA2,
2094 const char *zA3,
2095 const char *zA4
2097 ShellState *p = (ShellState*)pClientData;
2098 static const char *azProhibitedFunctions[] = {
2099 "edit",
2100 "fts3_tokenizer",
2101 "load_extension",
2102 "readfile",
2103 "writefile",
2104 "zipfile",
2105 "zipfile_cds",
2107 UNUSED_PARAMETER(zA1);
2108 UNUSED_PARAMETER(zA3);
2109 UNUSED_PARAMETER(zA4);
2110 switch( op ){
2111 case SQLITE_ATTACH: {
2112 #ifndef SQLITE_SHELL_FIDDLE
2113 /* In WASM builds the filesystem is a virtual sandbox, so
2114 ** there's no harm in using ATTACH. */
2115 failIfSafeMode(p, "cannot run ATTACH in safe mode");
2116 #endif
2117 break;
2119 case SQLITE_FUNCTION: {
2120 int i;
2121 for(i=0; i<ArraySize(azProhibitedFunctions); i++){
2122 if( sqlite3_stricmp(zA2, azProhibitedFunctions[i])==0 ){
2123 failIfSafeMode(p, "cannot use the %s() function in safe mode",
2124 azProhibitedFunctions[i]);
2127 break;
2130 return SQLITE_OK;
2134 ** When the ".auth ON" is set, the following authorizer callback is
2135 ** invoked. It always returns SQLITE_OK.
2137 static int shellAuth(
2138 void *pClientData,
2139 int op,
2140 const char *zA1,
2141 const char *zA2,
2142 const char *zA3,
2143 const char *zA4
2145 ShellState *p = (ShellState*)pClientData;
2146 static const char *azAction[] = { 0,
2147 "CREATE_INDEX", "CREATE_TABLE", "CREATE_TEMP_INDEX",
2148 "CREATE_TEMP_TABLE", "CREATE_TEMP_TRIGGER", "CREATE_TEMP_VIEW",
2149 "CREATE_TRIGGER", "CREATE_VIEW", "DELETE",
2150 "DROP_INDEX", "DROP_TABLE", "DROP_TEMP_INDEX",
2151 "DROP_TEMP_TABLE", "DROP_TEMP_TRIGGER", "DROP_TEMP_VIEW",
2152 "DROP_TRIGGER", "DROP_VIEW", "INSERT",
2153 "PRAGMA", "READ", "SELECT",
2154 "TRANSACTION", "UPDATE", "ATTACH",
2155 "DETACH", "ALTER_TABLE", "REINDEX",
2156 "ANALYZE", "CREATE_VTABLE", "DROP_VTABLE",
2157 "FUNCTION", "SAVEPOINT", "RECURSIVE"
2159 int i;
2160 const char *az[4];
2161 az[0] = zA1;
2162 az[1] = zA2;
2163 az[2] = zA3;
2164 az[3] = zA4;
2165 oputf("authorizer: %s", azAction[op]);
2166 for(i=0; i<4; i++){
2167 oputz(" ");
2168 if( az[i] ){
2169 output_c_string(az[i]);
2170 }else{
2171 oputz("NULL");
2174 oputz("\n");
2175 if( p->bSafeMode ) (void)safeModeAuth(pClientData, op, zA1, zA2, zA3, zA4);
2176 return SQLITE_OK;
2178 #endif
2181 ** Print a schema statement. Part of MODE_Semi and MODE_Pretty output.
2183 ** This routine converts some CREATE TABLE statements for shadow tables
2184 ** in FTS3/4/5 into CREATE TABLE IF NOT EXISTS statements.
2186 ** If the schema statement in z[] contains a start-of-comment and if
2187 ** sqlite3_complete() returns false, try to terminate the comment before
2188 ** printing the result. https://sqlite.org/forum/forumpost/d7be961c5c
2190 static void printSchemaLine(const char *z, const char *zTail){
2191 char *zToFree = 0;
2192 if( z==0 ) return;
2193 if( zTail==0 ) return;
2194 if( zTail[0]==';' && (strstr(z, "/*")!=0 || strstr(z,"--")!=0) ){
2195 const char *zOrig = z;
2196 static const char *azTerm[] = { "", "*/", "\n" };
2197 int i;
2198 for(i=0; i<ArraySize(azTerm); i++){
2199 char *zNew = sqlite3_mprintf("%s%s;", zOrig, azTerm[i]);
2200 shell_check_oom(zNew);
2201 if( sqlite3_complete(zNew) ){
2202 size_t n = strlen(zNew);
2203 zNew[n-1] = 0;
2204 zToFree = zNew;
2205 z = zNew;
2206 break;
2208 sqlite3_free(zNew);
2211 if( sqlite3_strglob("CREATE TABLE ['\"]*", z)==0 ){
2212 oputf("CREATE TABLE IF NOT EXISTS %s%s", z+13, zTail);
2213 }else{
2214 oputf("%s%s", z, zTail);
2216 sqlite3_free(zToFree);
2218 static void printSchemaLineN(char *z, int n, const char *zTail){
2219 char c = z[n];
2220 z[n] = 0;
2221 printSchemaLine(z, zTail);
2222 z[n] = c;
2226 ** Return true if string z[] has nothing but whitespace and comments to the
2227 ** end of the first line.
2229 static int wsToEol(const char *z){
2230 int i;
2231 for(i=0; z[i]; i++){
2232 if( z[i]=='\n' ) return 1;
2233 if( IsSpace(z[i]) ) continue;
2234 if( z[i]=='-' && z[i+1]=='-' ) return 1;
2235 return 0;
2237 return 1;
2241 ** Add a new entry to the EXPLAIN QUERY PLAN data
2243 static void eqp_append(ShellState *p, int iEqpId, int p2, const char *zText){
2244 EQPGraphRow *pNew;
2245 i64 nText;
2246 if( zText==0 ) return;
2247 nText = strlen(zText);
2248 if( p->autoEQPtest ){
2249 oputf("%d,%d,%s\n", iEqpId, p2, zText);
2251 pNew = sqlite3_malloc64( sizeof(*pNew) + nText );
2252 shell_check_oom(pNew);
2253 pNew->iEqpId = iEqpId;
2254 pNew->iParentId = p2;
2255 memcpy(pNew->zText, zText, nText+1);
2256 pNew->pNext = 0;
2257 if( p->sGraph.pLast ){
2258 p->sGraph.pLast->pNext = pNew;
2259 }else{
2260 p->sGraph.pRow = pNew;
2262 p->sGraph.pLast = pNew;
2266 ** Free and reset the EXPLAIN QUERY PLAN data that has been collected
2267 ** in p->sGraph.
2269 static void eqp_reset(ShellState *p){
2270 EQPGraphRow *pRow, *pNext;
2271 for(pRow = p->sGraph.pRow; pRow; pRow = pNext){
2272 pNext = pRow->pNext;
2273 sqlite3_free(pRow);
2275 memset(&p->sGraph, 0, sizeof(p->sGraph));
2278 /* Return the next EXPLAIN QUERY PLAN line with iEqpId that occurs after
2279 ** pOld, or return the first such line if pOld is NULL
2281 static EQPGraphRow *eqp_next_row(ShellState *p, int iEqpId, EQPGraphRow *pOld){
2282 EQPGraphRow *pRow = pOld ? pOld->pNext : p->sGraph.pRow;
2283 while( pRow && pRow->iParentId!=iEqpId ) pRow = pRow->pNext;
2284 return pRow;
2287 /* Render a single level of the graph that has iEqpId as its parent. Called
2288 ** recursively to render sublevels.
2290 static void eqp_render_level(ShellState *p, int iEqpId){
2291 EQPGraphRow *pRow, *pNext;
2292 i64 n = strlen(p->sGraph.zPrefix);
2293 char *z;
2294 for(pRow = eqp_next_row(p, iEqpId, 0); pRow; pRow = pNext){
2295 pNext = eqp_next_row(p, iEqpId, pRow);
2296 z = pRow->zText;
2297 oputf("%s%s%s\n", p->sGraph.zPrefix, pNext ? "|--" : "`--", z);
2298 if( n<(i64)sizeof(p->sGraph.zPrefix)-7 ){
2299 memcpy(&p->sGraph.zPrefix[n], pNext ? "| " : " ", 4);
2300 eqp_render_level(p, pRow->iEqpId);
2301 p->sGraph.zPrefix[n] = 0;
2307 ** Display and reset the EXPLAIN QUERY PLAN data
2309 static void eqp_render(ShellState *p, i64 nCycle){
2310 EQPGraphRow *pRow = p->sGraph.pRow;
2311 if( pRow ){
2312 if( pRow->zText[0]=='-' ){
2313 if( pRow->pNext==0 ){
2314 eqp_reset(p);
2315 return;
2317 oputf("%s\n", pRow->zText+3);
2318 p->sGraph.pRow = pRow->pNext;
2319 sqlite3_free(pRow);
2320 }else if( nCycle>0 ){
2321 oputf("QUERY PLAN (cycles=%lld [100%%])\n", nCycle);
2322 }else{
2323 oputz("QUERY PLAN\n");
2325 p->sGraph.zPrefix[0] = 0;
2326 eqp_render_level(p, 0);
2327 eqp_reset(p);
2331 #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
2333 ** Progress handler callback.
2335 static int progress_handler(void *pClientData) {
2336 ShellState *p = (ShellState*)pClientData;
2337 p->nProgress++;
2338 if( p->nProgress>=p->mxProgress && p->mxProgress>0 ){
2339 oputf("Progress limit reached (%u)\n", p->nProgress);
2340 if( p->flgProgress & SHELL_PROGRESS_RESET ) p->nProgress = 0;
2341 if( p->flgProgress & SHELL_PROGRESS_ONCE ) p->mxProgress = 0;
2342 return 1;
2344 if( (p->flgProgress & SHELL_PROGRESS_QUIET)==0 ){
2345 oputf("Progress %u\n", p->nProgress);
2347 return 0;
2349 #endif /* SQLITE_OMIT_PROGRESS_CALLBACK */
2352 ** Print N dashes
2354 static void print_dashes(int N){
2355 const char zDash[] = "--------------------------------------------------";
2356 const int nDash = sizeof(zDash) - 1;
2357 while( N>nDash ){
2358 oputz(zDash);
2359 N -= nDash;
2361 oputf("%.*s", N, zDash);
2365 ** Print a markdown or table-style row separator using ascii-art
2367 static void print_row_separator(
2368 ShellState *p,
2369 int nArg,
2370 const char *zSep
2372 int i;
2373 if( nArg>0 ){
2374 oputz(zSep);
2375 print_dashes(p->actualWidth[0]+2);
2376 for(i=1; i<nArg; i++){
2377 oputz(zSep);
2378 print_dashes(p->actualWidth[i]+2);
2380 oputz(zSep);
2382 oputz("\n");
2386 ** This is the callback routine that the shell
2387 ** invokes for each row of a query result.
2389 static int shell_callback(
2390 void *pArg,
2391 int nArg, /* Number of result columns */
2392 char **azArg, /* Text of each result column */
2393 char **azCol, /* Column names */
2394 int *aiType /* Column types. Might be NULL */
2396 int i;
2397 ShellState *p = (ShellState*)pArg;
2399 if( azArg==0 ) return 0;
2400 switch( p->cMode ){
2401 case MODE_Count:
2402 case MODE_Off: {
2403 break;
2405 case MODE_Line: {
2406 int w = 5;
2407 if( azArg==0 ) break;
2408 for(i=0; i<nArg; i++){
2409 int len = strlen30(azCol[i] ? azCol[i] : "");
2410 if( len>w ) w = len;
2412 if( p->cnt++>0 ) oputz(p->rowSeparator);
2413 for(i=0; i<nArg; i++){
2414 oputf("%*s = %s%s", w, azCol[i],
2415 azArg[i] ? azArg[i] : p->nullValue, p->rowSeparator);
2417 break;
2419 case MODE_ScanExp:
2420 case MODE_Explain: {
2421 static const int aExplainWidth[] = {4, 13, 4, 4, 4, 13, 2, 13};
2422 static const int aExplainMap[] = {0, 1, 2, 3, 4, 5, 6, 7 };
2423 static const int aScanExpWidth[] = {4, 6, 6, 13, 4, 4, 4, 13, 2, 13};
2424 static const int aScanExpMap[] = {0, 9, 8, 1, 2, 3, 4, 5, 6, 7 };
2426 const int *aWidth = aExplainWidth;
2427 const int *aMap = aExplainMap;
2428 int nWidth = ArraySize(aExplainWidth);
2429 int iIndent = 1;
2431 if( p->cMode==MODE_ScanExp ){
2432 aWidth = aScanExpWidth;
2433 aMap = aScanExpMap;
2434 nWidth = ArraySize(aScanExpWidth);
2435 iIndent = 3;
2437 if( nArg>nWidth ) nArg = nWidth;
2439 /* If this is the first row seen, print out the headers */
2440 if( p->cnt++==0 ){
2441 for(i=0; i<nArg; i++){
2442 utf8_width_print(aWidth[i], azCol[ aMap[i] ]);
2443 oputz(i==nArg-1 ? "\n" : " ");
2445 for(i=0; i<nArg; i++){
2446 print_dashes(aWidth[i]);
2447 oputz(i==nArg-1 ? "\n" : " ");
2451 /* If there is no data, exit early. */
2452 if( azArg==0 ) break;
2454 for(i=0; i<nArg; i++){
2455 const char *zSep = " ";
2456 int w = aWidth[i];
2457 const char *zVal = azArg[ aMap[i] ];
2458 if( i==nArg-1 ) w = 0;
2459 if( zVal && strlenChar(zVal)>w ){
2460 w = strlenChar(zVal);
2461 zSep = " ";
2463 if( i==iIndent && p->aiIndent && p->pStmt ){
2464 if( p->iIndent<p->nIndent ){
2465 oputf("%*.s", p->aiIndent[p->iIndent], "");
2467 p->iIndent++;
2469 utf8_width_print(w, zVal ? zVal : p->nullValue);
2470 oputz(i==nArg-1 ? "\n" : zSep);
2472 break;
2474 case MODE_Semi: { /* .schema and .fullschema output */
2475 printSchemaLine(azArg[0], ";\n");
2476 break;
2478 case MODE_Pretty: { /* .schema and .fullschema with --indent */
2479 char *z;
2480 int j;
2481 int nParen = 0;
2482 char cEnd = 0;
2483 char c;
2484 int nLine = 0;
2485 assert( nArg==1 );
2486 if( azArg[0]==0 ) break;
2487 if( sqlite3_strlike("CREATE VIEW%", azArg[0], 0)==0
2488 || sqlite3_strlike("CREATE TRIG%", azArg[0], 0)==0
2490 oputf("%s;\n", azArg[0]);
2491 break;
2493 z = sqlite3_mprintf("%s", azArg[0]);
2494 shell_check_oom(z);
2495 j = 0;
2496 for(i=0; IsSpace(z[i]); i++){}
2497 for(; (c = z[i])!=0; i++){
2498 if( IsSpace(c) ){
2499 if( z[j-1]=='\r' ) z[j-1] = '\n';
2500 if( IsSpace(z[j-1]) || z[j-1]=='(' ) continue;
2501 }else if( (c=='(' || c==')') && j>0 && IsSpace(z[j-1]) ){
2502 j--;
2504 z[j++] = c;
2506 while( j>0 && IsSpace(z[j-1]) ){ j--; }
2507 z[j] = 0;
2508 if( strlen30(z)>=79 ){
2509 for(i=j=0; (c = z[i])!=0; i++){ /* Copy from z[i] back to z[j] */
2510 if( c==cEnd ){
2511 cEnd = 0;
2512 }else if( c=='"' || c=='\'' || c=='`' ){
2513 cEnd = c;
2514 }else if( c=='[' ){
2515 cEnd = ']';
2516 }else if( c=='-' && z[i+1]=='-' ){
2517 cEnd = '\n';
2518 }else if( c=='(' ){
2519 nParen++;
2520 }else if( c==')' ){
2521 nParen--;
2522 if( nLine>0 && nParen==0 && j>0 ){
2523 printSchemaLineN(z, j, "\n");
2524 j = 0;
2527 z[j++] = c;
2528 if( nParen==1 && cEnd==0
2529 && (c=='(' || c=='\n' || (c==',' && !wsToEol(z+i+1)))
2531 if( c=='\n' ) j--;
2532 printSchemaLineN(z, j, "\n ");
2533 j = 0;
2534 nLine++;
2535 while( IsSpace(z[i+1]) ){ i++; }
2538 z[j] = 0;
2540 printSchemaLine(z, ";\n");
2541 sqlite3_free(z);
2542 break;
2544 case MODE_List: {
2545 if( p->cnt++==0 && p->showHeader ){
2546 for(i=0; i<nArg; i++){
2547 oputf("%s%s",azCol[i], i==nArg-1 ? p->rowSeparator : p->colSeparator);
2550 if( azArg==0 ) break;
2551 for(i=0; i<nArg; i++){
2552 char *z = azArg[i];
2553 if( z==0 ) z = p->nullValue;
2554 oputz(z);
2555 oputz((i<nArg-1)? p->colSeparator : p->rowSeparator);
2557 break;
2559 case MODE_Html: {
2560 if( p->cnt++==0 && p->showHeader ){
2561 oputz("<TR>");
2562 for(i=0; i<nArg; i++){
2563 oputz("<TH>");
2564 output_html_string(azCol[i]);
2565 oputz("</TH>\n");
2567 oputz("</TR>\n");
2569 if( azArg==0 ) break;
2570 oputz("<TR>");
2571 for(i=0; i<nArg; i++){
2572 oputz("<TD>");
2573 output_html_string(azArg[i] ? azArg[i] : p->nullValue);
2574 oputz("</TD>\n");
2576 oputz("</TR>\n");
2577 break;
2579 case MODE_Tcl: {
2580 if( p->cnt++==0 && p->showHeader ){
2581 for(i=0; i<nArg; i++){
2582 output_c_string(azCol[i] ? azCol[i] : "");
2583 if(i<nArg-1) oputz(p->colSeparator);
2585 oputz(p->rowSeparator);
2587 if( azArg==0 ) break;
2588 for(i=0; i<nArg; i++){
2589 output_c_string(azArg[i] ? azArg[i] : p->nullValue);
2590 if(i<nArg-1) oputz(p->colSeparator);
2592 oputz(p->rowSeparator);
2593 break;
2595 case MODE_Csv: {
2596 setBinaryMode(p->out, 1);
2597 if( p->cnt++==0 && p->showHeader ){
2598 for(i=0; i<nArg; i++){
2599 output_csv(p, azCol[i] ? azCol[i] : "", i<nArg-1);
2601 oputz(p->rowSeparator);
2603 if( nArg>0 ){
2604 for(i=0; i<nArg; i++){
2605 output_csv(p, azArg[i], i<nArg-1);
2607 oputz(p->rowSeparator);
2609 setTextMode(p->out, 1);
2610 break;
2612 case MODE_Insert: {
2613 if( azArg==0 ) break;
2614 oputf("INSERT INTO %s",p->zDestTable);
2615 if( p->showHeader ){
2616 oputz("(");
2617 for(i=0; i<nArg; i++){
2618 if( i>0 ) oputz(",");
2619 if( quoteChar(azCol[i]) ){
2620 char *z = sqlite3_mprintf("\"%w\"", azCol[i]);
2621 shell_check_oom(z);
2622 oputz(z);
2623 sqlite3_free(z);
2624 }else{
2625 oputf("%s", azCol[i]);
2628 oputz(")");
2630 p->cnt++;
2631 for(i=0; i<nArg; i++){
2632 oputz(i>0 ? "," : " VALUES(");
2633 if( (azArg[i]==0) || (aiType && aiType[i]==SQLITE_NULL) ){
2634 oputz("NULL");
2635 }else if( aiType && aiType[i]==SQLITE_TEXT ){
2636 if( ShellHasFlag(p, SHFLG_Newlines) ){
2637 output_quoted_string(azArg[i]);
2638 }else{
2639 output_quoted_escaped_string(azArg[i]);
2641 }else if( aiType && aiType[i]==SQLITE_INTEGER ){
2642 oputz(azArg[i]);
2643 }else if( aiType && aiType[i]==SQLITE_FLOAT ){
2644 char z[50];
2645 double r = sqlite3_column_double(p->pStmt, i);
2646 sqlite3_uint64 ur;
2647 memcpy(&ur,&r,sizeof(r));
2648 if( ur==0x7ff0000000000000LL ){
2649 oputz("9.0e+999");
2650 }else if( ur==0xfff0000000000000LL ){
2651 oputz("-9.0e+999");
2652 }else{
2653 sqlite3_int64 ir = (sqlite3_int64)r;
2654 if( r==(double)ir ){
2655 sqlite3_snprintf(50,z,"%lld.0", ir);
2656 }else{
2657 sqlite3_snprintf(50,z,"%!.20g", r);
2659 oputz(z);
2661 }else if( aiType && aiType[i]==SQLITE_BLOB && p->pStmt ){
2662 const void *pBlob = sqlite3_column_blob(p->pStmt, i);
2663 int nBlob = sqlite3_column_bytes(p->pStmt, i);
2664 output_hex_blob(pBlob, nBlob);
2665 }else if( isNumber(azArg[i], 0) ){
2666 oputz(azArg[i]);
2667 }else if( ShellHasFlag(p, SHFLG_Newlines) ){
2668 output_quoted_string(azArg[i]);
2669 }else{
2670 output_quoted_escaped_string(azArg[i]);
2673 oputz(");\n");
2674 break;
2676 case MODE_Json: {
2677 if( azArg==0 ) break;
2678 if( p->cnt==0 ){
2679 fputs("[{", p->out);
2680 }else{
2681 fputs(",\n{", p->out);
2683 p->cnt++;
2684 for(i=0; i<nArg; i++){
2685 output_json_string(azCol[i], -1);
2686 oputz(":");
2687 if( (azArg[i]==0) || (aiType && aiType[i]==SQLITE_NULL) ){
2688 oputz("null");
2689 }else if( aiType && aiType[i]==SQLITE_FLOAT ){
2690 char z[50];
2691 double r = sqlite3_column_double(p->pStmt, i);
2692 sqlite3_uint64 ur;
2693 memcpy(&ur,&r,sizeof(r));
2694 if( ur==0x7ff0000000000000LL ){
2695 oputz("9.0e+999");
2696 }else if( ur==0xfff0000000000000LL ){
2697 oputz("-9.0e+999");
2698 }else{
2699 sqlite3_snprintf(50,z,"%!.20g", r);
2700 oputz(z);
2702 }else if( aiType && aiType[i]==SQLITE_BLOB && p->pStmt ){
2703 const void *pBlob = sqlite3_column_blob(p->pStmt, i);
2704 int nBlob = sqlite3_column_bytes(p->pStmt, i);
2705 output_json_string(pBlob, nBlob);
2706 }else if( aiType && aiType[i]==SQLITE_TEXT ){
2707 output_json_string(azArg[i], -1);
2708 }else{
2709 oputz(azArg[i]);
2711 if( i<nArg-1 ){
2712 oputz(",");
2715 oputz("}");
2716 break;
2718 case MODE_Quote: {
2719 if( azArg==0 ) break;
2720 if( p->cnt==0 && p->showHeader ){
2721 for(i=0; i<nArg; i++){
2722 if( i>0 ) fputs(p->colSeparator, p->out);
2723 output_quoted_string(azCol[i]);
2725 fputs(p->rowSeparator, p->out);
2727 p->cnt++;
2728 for(i=0; i<nArg; i++){
2729 if( i>0 ) fputs(p->colSeparator, p->out);
2730 if( (azArg[i]==0) || (aiType && aiType[i]==SQLITE_NULL) ){
2731 oputz("NULL");
2732 }else if( aiType && aiType[i]==SQLITE_TEXT ){
2733 output_quoted_string(azArg[i]);
2734 }else if( aiType && aiType[i]==SQLITE_INTEGER ){
2735 oputz(azArg[i]);
2736 }else if( aiType && aiType[i]==SQLITE_FLOAT ){
2737 char z[50];
2738 double r = sqlite3_column_double(p->pStmt, i);
2739 sqlite3_snprintf(50,z,"%!.20g", r);
2740 oputz(z);
2741 }else if( aiType && aiType[i]==SQLITE_BLOB && p->pStmt ){
2742 const void *pBlob = sqlite3_column_blob(p->pStmt, i);
2743 int nBlob = sqlite3_column_bytes(p->pStmt, i);
2744 output_hex_blob(pBlob, nBlob);
2745 }else if( isNumber(azArg[i], 0) ){
2746 oputz(azArg[i]);
2747 }else{
2748 output_quoted_string(azArg[i]);
2751 fputs(p->rowSeparator, p->out);
2752 break;
2754 case MODE_Ascii: {
2755 if( p->cnt++==0 && p->showHeader ){
2756 for(i=0; i<nArg; i++){
2757 if( i>0 ) oputz(p->colSeparator);
2758 oputz(azCol[i] ? azCol[i] : "");
2760 oputz(p->rowSeparator);
2762 if( azArg==0 ) break;
2763 for(i=0; i<nArg; i++){
2764 if( i>0 ) oputz(p->colSeparator);
2765 oputz(azArg[i] ? azArg[i] : p->nullValue);
2767 oputz(p->rowSeparator);
2768 break;
2770 case MODE_EQP: {
2771 eqp_append(p, atoi(azArg[0]), atoi(azArg[1]), azArg[3]);
2772 break;
2775 return 0;
2779 ** This is the callback routine that the SQLite library
2780 ** invokes for each row of a query result.
2782 static int callback(void *pArg, int nArg, char **azArg, char **azCol){
2783 /* since we don't have type info, call the shell_callback with a NULL value */
2784 return shell_callback(pArg, nArg, azArg, azCol, NULL);
2788 ** This is the callback routine from sqlite3_exec() that appends all
2789 ** output onto the end of a ShellText object.
2791 static int captureOutputCallback(void *pArg, int nArg, char **azArg, char **az){
2792 ShellText *p = (ShellText*)pArg;
2793 int i;
2794 UNUSED_PARAMETER(az);
2795 if( azArg==0 ) return 0;
2796 if( p->n ) appendText(p, "|", 0);
2797 for(i=0; i<nArg; i++){
2798 if( i ) appendText(p, ",", 0);
2799 if( azArg[i] ) appendText(p, azArg[i], 0);
2801 return 0;
2805 ** Generate an appropriate SELFTEST table in the main database.
2807 static void createSelftestTable(ShellState *p){
2808 char *zErrMsg = 0;
2809 sqlite3_exec(p->db,
2810 "SAVEPOINT selftest_init;\n"
2811 "CREATE TABLE IF NOT EXISTS selftest(\n"
2812 " tno INTEGER PRIMARY KEY,\n" /* Test number */
2813 " op TEXT,\n" /* Operator: memo run */
2814 " cmd TEXT,\n" /* Command text */
2815 " ans TEXT\n" /* Desired answer */
2816 ");"
2817 "CREATE TEMP TABLE [_shell$self](op,cmd,ans);\n"
2818 "INSERT INTO [_shell$self](rowid,op,cmd)\n"
2819 " VALUES(coalesce((SELECT (max(tno)+100)/10 FROM selftest),10),\n"
2820 " 'memo','Tests generated by --init');\n"
2821 "INSERT INTO [_shell$self]\n"
2822 " SELECT 'run',\n"
2823 " 'SELECT hex(sha3_query(''SELECT type,name,tbl_name,sql "
2824 "FROM sqlite_schema ORDER BY 2'',224))',\n"
2825 " hex(sha3_query('SELECT type,name,tbl_name,sql "
2826 "FROM sqlite_schema ORDER BY 2',224));\n"
2827 "INSERT INTO [_shell$self]\n"
2828 " SELECT 'run',"
2829 " 'SELECT hex(sha3_query(''SELECT * FROM \"' ||"
2830 " printf('%w',name) || '\" NOT INDEXED'',224))',\n"
2831 " hex(sha3_query(printf('SELECT * FROM \"%w\" NOT INDEXED',name),224))\n"
2832 " FROM (\n"
2833 " SELECT name FROM sqlite_schema\n"
2834 " WHERE type='table'\n"
2835 " AND name<>'selftest'\n"
2836 " AND coalesce(rootpage,0)>0\n"
2837 " )\n"
2838 " ORDER BY name;\n"
2839 "INSERT INTO [_shell$self]\n"
2840 " VALUES('run','PRAGMA integrity_check','ok');\n"
2841 "INSERT INTO selftest(tno,op,cmd,ans)"
2842 " SELECT rowid*10,op,cmd,ans FROM [_shell$self];\n"
2843 "DROP TABLE [_shell$self];"
2844 ,0,0,&zErrMsg);
2845 if( zErrMsg ){
2846 eputf("SELFTEST initialization failure: %s\n", zErrMsg);
2847 sqlite3_free(zErrMsg);
2849 sqlite3_exec(p->db, "RELEASE selftest_init",0,0,0);
2854 ** Set the destination table field of the ShellState structure to
2855 ** the name of the table given. Escape any quote characters in the
2856 ** table name.
2858 static void set_table_name(ShellState *p, const char *zName){
2859 int i, n;
2860 char cQuote;
2861 char *z;
2863 if( p->zDestTable ){
2864 free(p->zDestTable);
2865 p->zDestTable = 0;
2867 if( zName==0 ) return;
2868 cQuote = quoteChar(zName);
2869 n = strlen30(zName);
2870 if( cQuote ) n += n+2;
2871 z = p->zDestTable = malloc( n+1 );
2872 shell_check_oom(z);
2873 n = 0;
2874 if( cQuote ) z[n++] = cQuote;
2875 for(i=0; zName[i]; i++){
2876 z[n++] = zName[i];
2877 if( zName[i]==cQuote ) z[n++] = cQuote;
2879 if( cQuote ) z[n++] = cQuote;
2880 z[n] = 0;
2884 ** Maybe construct two lines of text that point out the position of a
2885 ** syntax error. Return a pointer to the text, in memory obtained from
2886 ** sqlite3_malloc(). Or, if the most recent error does not involve a
2887 ** specific token that we can point to, return an empty string.
2889 ** In all cases, the memory returned is obtained from sqlite3_malloc64()
2890 ** and should be released by the caller invoking sqlite3_free().
2892 static char *shell_error_context(const char *zSql, sqlite3 *db){
2893 int iOffset;
2894 size_t len;
2895 char *zCode;
2896 char *zMsg;
2897 int i;
2898 if( db==0
2899 || zSql==0
2900 || (iOffset = sqlite3_error_offset(db))<0
2901 || iOffset>=(int)strlen(zSql)
2903 return sqlite3_mprintf("");
2905 while( iOffset>50 ){
2906 iOffset--;
2907 zSql++;
2908 while( (zSql[0]&0xc0)==0x80 ){ zSql++; iOffset--; }
2910 len = strlen(zSql);
2911 if( len>78 ){
2912 len = 78;
2913 while( len>0 && (zSql[len]&0xc0)==0x80 ) len--;
2915 zCode = sqlite3_mprintf("%.*s", len, zSql);
2916 shell_check_oom(zCode);
2917 for(i=0; zCode[i]; i++){ if( IsSpace(zSql[i]) ) zCode[i] = ' '; }
2918 if( iOffset<25 ){
2919 zMsg = sqlite3_mprintf("\n %z\n %*s^--- error here", zCode,iOffset,"");
2920 }else{
2921 zMsg = sqlite3_mprintf("\n %z\n %*serror here ---^", zCode,iOffset-14,"");
2923 return zMsg;
2928 ** Execute a query statement that will generate SQL output. Print
2929 ** the result columns, comma-separated, on a line and then add a
2930 ** semicolon terminator to the end of that line.
2932 ** If the number of columns is 1 and that column contains text "--"
2933 ** then write the semicolon on a separate line. That way, if a
2934 ** "--" comment occurs at the end of the statement, the comment
2935 ** won't consume the semicolon terminator.
2937 static int run_table_dump_query(
2938 ShellState *p, /* Query context */
2939 const char *zSelect /* SELECT statement to extract content */
2941 sqlite3_stmt *pSelect;
2942 int rc;
2943 int nResult;
2944 int i;
2945 const char *z;
2946 rc = sqlite3_prepare_v2(p->db, zSelect, -1, &pSelect, 0);
2947 if( rc!=SQLITE_OK || !pSelect ){
2948 char *zContext = shell_error_context(zSelect, p->db);
2949 oputf("/**** ERROR: (%d) %s *****/\n%s",
2950 rc, sqlite3_errmsg(p->db), zContext);
2951 sqlite3_free(zContext);
2952 if( (rc&0xff)!=SQLITE_CORRUPT ) p->nErr++;
2953 return rc;
2955 rc = sqlite3_step(pSelect);
2956 nResult = sqlite3_column_count(pSelect);
2957 while( rc==SQLITE_ROW ){
2958 z = (const char*)sqlite3_column_text(pSelect, 0);
2959 oputf("%s", z);
2960 for(i=1; i<nResult; i++){
2961 oputf(",%s", sqlite3_column_text(pSelect, i));
2963 if( z==0 ) z = "";
2964 while( z[0] && (z[0]!='-' || z[1]!='-') ) z++;
2965 if( z[0] ){
2966 oputz("\n;\n");
2967 }else{
2968 oputz(";\n");
2970 rc = sqlite3_step(pSelect);
2972 rc = sqlite3_finalize(pSelect);
2973 if( rc!=SQLITE_OK ){
2974 oputf("/**** ERROR: (%d) %s *****/\n", rc, sqlite3_errmsg(p->db));
2975 if( (rc&0xff)!=SQLITE_CORRUPT ) p->nErr++;
2977 return rc;
2981 ** Allocate space and save off string indicating current error.
2983 static char *save_err_msg(
2984 sqlite3 *db, /* Database to query */
2985 const char *zPhase, /* When the error occurs */
2986 int rc, /* Error code returned from API */
2987 const char *zSql /* SQL string, or NULL */
2989 char *zErr;
2990 char *zContext;
2991 sqlite3_str *pStr = sqlite3_str_new(0);
2992 sqlite3_str_appendf(pStr, "%s, %s", zPhase, sqlite3_errmsg(db));
2993 if( rc>1 ){
2994 sqlite3_str_appendf(pStr, " (%d)", rc);
2996 zContext = shell_error_context(zSql, db);
2997 if( zContext ){
2998 sqlite3_str_appendall(pStr, zContext);
2999 sqlite3_free(zContext);
3001 zErr = sqlite3_str_finish(pStr);
3002 shell_check_oom(zErr);
3003 return zErr;
3006 #ifdef __linux__
3008 ** Attempt to display I/O stats on Linux using /proc/PID/io
3010 static void displayLinuxIoStats(void){
3011 FILE *in;
3012 char z[200];
3013 sqlite3_snprintf(sizeof(z), z, "/proc/%d/io", getpid());
3014 in = fopen(z, "rb");
3015 if( in==0 ) return;
3016 while( fgets(z, sizeof(z), in)!=0 ){
3017 static const struct {
3018 const char *zPattern;
3019 const char *zDesc;
3020 } aTrans[] = {
3021 { "rchar: ", "Bytes received by read():" },
3022 { "wchar: ", "Bytes sent to write():" },
3023 { "syscr: ", "Read() system calls:" },
3024 { "syscw: ", "Write() system calls:" },
3025 { "read_bytes: ", "Bytes read from storage:" },
3026 { "write_bytes: ", "Bytes written to storage:" },
3027 { "cancelled_write_bytes: ", "Cancelled write bytes:" },
3029 int i;
3030 for(i=0; i<ArraySize(aTrans); i++){
3031 int n = strlen30(aTrans[i].zPattern);
3032 if( cli_strncmp(aTrans[i].zPattern, z, n)==0 ){
3033 oputf("%-36s %s", aTrans[i].zDesc, &z[n]);
3034 break;
3038 fclose(in);
3040 #endif
3043 ** Display a single line of status using 64-bit values.
3045 static void displayStatLine(
3046 char *zLabel, /* Label for this one line */
3047 char *zFormat, /* Format for the result */
3048 int iStatusCtrl, /* Which status to display */
3049 int bReset /* True to reset the stats */
3051 sqlite3_int64 iCur = -1;
3052 sqlite3_int64 iHiwtr = -1;
3053 int i, nPercent;
3054 char zLine[200];
3055 sqlite3_status64(iStatusCtrl, &iCur, &iHiwtr, bReset);
3056 for(i=0, nPercent=0; zFormat[i]; i++){
3057 if( zFormat[i]=='%' ) nPercent++;
3059 if( nPercent>1 ){
3060 sqlite3_snprintf(sizeof(zLine), zLine, zFormat, iCur, iHiwtr);
3061 }else{
3062 sqlite3_snprintf(sizeof(zLine), zLine, zFormat, iHiwtr);
3064 oputf("%-36s %s\n", zLabel, zLine);
3068 ** Display memory stats.
3070 static int display_stats(
3071 sqlite3 *db, /* Database to query */
3072 ShellState *pArg, /* Pointer to ShellState */
3073 int bReset /* True to reset the stats */
3075 int iCur;
3076 int iHiwtr;
3077 if( pArg==0 || pArg->out==0 ) return 0;
3079 if( pArg->pStmt && pArg->statsOn==2 ){
3080 int nCol, i, x;
3081 sqlite3_stmt *pStmt = pArg->pStmt;
3082 char z[100];
3083 nCol = sqlite3_column_count(pStmt);
3084 oputf("%-36s %d\n", "Number of output columns:", nCol);
3085 for(i=0; i<nCol; i++){
3086 sqlite3_snprintf(sizeof(z),z,"Column %d %nname:", i, &x);
3087 oputf("%-36s %s\n", z, sqlite3_column_name(pStmt,i));
3088 #ifndef SQLITE_OMIT_DECLTYPE
3089 sqlite3_snprintf(30, z+x, "declared type:");
3090 oputf("%-36s %s\n", z, sqlite3_column_decltype(pStmt, i));
3091 #endif
3092 #ifdef SQLITE_ENABLE_COLUMN_METADATA
3093 sqlite3_snprintf(30, z+x, "database name:");
3094 oputf("%-36s %s\n", z, sqlite3_column_database_name(pStmt,i));
3095 sqlite3_snprintf(30, z+x, "table name:");
3096 oputf("%-36s %s\n", z, sqlite3_column_table_name(pStmt,i));
3097 sqlite3_snprintf(30, z+x, "origin name:");
3098 oputf("%-36s %s\n", z, sqlite3_column_origin_name(pStmt,i));
3099 #endif
3103 if( pArg->statsOn==3 ){
3104 if( pArg->pStmt ){
3105 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_VM_STEP,bReset);
3106 oputf("VM-steps: %d\n", iCur);
3108 return 0;
3111 displayStatLine("Memory Used:",
3112 "%lld (max %lld) bytes", SQLITE_STATUS_MEMORY_USED, bReset);
3113 displayStatLine("Number of Outstanding Allocations:",
3114 "%lld (max %lld)", SQLITE_STATUS_MALLOC_COUNT, bReset);
3115 if( pArg->shellFlgs & SHFLG_Pagecache ){
3116 displayStatLine("Number of Pcache Pages Used:",
3117 "%lld (max %lld) pages", SQLITE_STATUS_PAGECACHE_USED, bReset);
3119 displayStatLine("Number of Pcache Overflow Bytes:",
3120 "%lld (max %lld) bytes", SQLITE_STATUS_PAGECACHE_OVERFLOW, bReset);
3121 displayStatLine("Largest Allocation:",
3122 "%lld bytes", SQLITE_STATUS_MALLOC_SIZE, bReset);
3123 displayStatLine("Largest Pcache Allocation:",
3124 "%lld bytes", SQLITE_STATUS_PAGECACHE_SIZE, bReset);
3125 #ifdef YYTRACKMAXSTACKDEPTH
3126 displayStatLine("Deepest Parser Stack:",
3127 "%lld (max %lld)", SQLITE_STATUS_PARSER_STACK, bReset);
3128 #endif
3130 if( db ){
3131 if( pArg->shellFlgs & SHFLG_Lookaside ){
3132 iHiwtr = iCur = -1;
3133 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_USED,
3134 &iCur, &iHiwtr, bReset);
3135 oputf("Lookaside Slots Used: %d (max %d)\n", iCur, iHiwtr);
3136 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_HIT,
3137 &iCur, &iHiwtr, bReset);
3138 oputf("Successful lookaside attempts: %d\n", iHiwtr);
3139 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_SIZE,
3140 &iCur, &iHiwtr, bReset);
3141 oputf("Lookaside failures due to size: %d\n", iHiwtr);
3142 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_FULL,
3143 &iCur, &iHiwtr, bReset);
3144 oputf("Lookaside failures due to OOM: %d\n", iHiwtr);
3146 iHiwtr = iCur = -1;
3147 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_USED, &iCur, &iHiwtr, bReset);
3148 oputf("Pager Heap Usage: %d bytes\n", iCur);
3149 iHiwtr = iCur = -1;
3150 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_HIT, &iCur, &iHiwtr, 1);
3151 oputf("Page cache hits: %d\n", iCur);
3152 iHiwtr = iCur = -1;
3153 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_MISS, &iCur, &iHiwtr, 1);
3154 oputf("Page cache misses: %d\n", iCur);
3155 iHiwtr = iCur = -1;
3156 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_WRITE, &iCur, &iHiwtr, 1);
3157 oputf("Page cache writes: %d\n", iCur);
3158 iHiwtr = iCur = -1;
3159 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_SPILL, &iCur, &iHiwtr, 1);
3160 oputf("Page cache spills: %d\n", iCur);
3161 iHiwtr = iCur = -1;
3162 sqlite3_db_status(db, SQLITE_DBSTATUS_SCHEMA_USED, &iCur, &iHiwtr, bReset);
3163 oputf("Schema Heap Usage: %d bytes\n", iCur);
3164 iHiwtr = iCur = -1;
3165 sqlite3_db_status(db, SQLITE_DBSTATUS_STMT_USED, &iCur, &iHiwtr, bReset);
3166 oputf("Statement Heap/Lookaside Usage: %d bytes\n", iCur);
3169 if( pArg->pStmt ){
3170 int iHit, iMiss;
3171 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_FULLSCAN_STEP,
3172 bReset);
3173 oputf("Fullscan Steps: %d\n", iCur);
3174 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_SORT, bReset);
3175 oputf("Sort Operations: %d\n", iCur);
3176 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_AUTOINDEX,bReset);
3177 oputf("Autoindex Inserts: %d\n", iCur);
3178 iHit = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_FILTER_HIT,
3179 bReset);
3180 iMiss = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_FILTER_MISS,
3181 bReset);
3182 if( iHit || iMiss ){
3183 oputf("Bloom filter bypass taken: %d/%d\n", iHit, iHit+iMiss);
3185 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_VM_STEP, bReset);
3186 oputf("Virtual Machine Steps: %d\n", iCur);
3187 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_REPREPARE,bReset);
3188 oputf("Reprepare operations: %d\n", iCur);
3189 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_RUN, bReset);
3190 oputf("Number of times run: %d\n", iCur);
3191 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_MEMUSED, bReset);
3192 oputf("Memory used by prepared stmt: %d\n", iCur);
3195 #ifdef __linux__
3196 displayLinuxIoStats();
3197 #endif
3199 /* Do not remove this machine readable comment: extra-stats-output-here */
3201 return 0;
3205 #ifdef SQLITE_ENABLE_STMT_SCANSTATUS
3206 static int scanStatsHeight(sqlite3_stmt *p, int iEntry){
3207 int iPid = 0;
3208 int ret = 1;
3209 sqlite3_stmt_scanstatus_v2(p, iEntry,
3210 SQLITE_SCANSTAT_SELECTID, SQLITE_SCANSTAT_COMPLEX, (void*)&iPid
3212 while( iPid!=0 ){
3213 int ii;
3214 for(ii=0; 1; ii++){
3215 int iId;
3216 int res;
3217 res = sqlite3_stmt_scanstatus_v2(p, ii,
3218 SQLITE_SCANSTAT_SELECTID, SQLITE_SCANSTAT_COMPLEX, (void*)&iId
3220 if( res ) break;
3221 if( iId==iPid ){
3222 sqlite3_stmt_scanstatus_v2(p, ii,
3223 SQLITE_SCANSTAT_PARENTID, SQLITE_SCANSTAT_COMPLEX, (void*)&iPid
3227 ret++;
3229 return ret;
3231 #endif
3233 #ifdef SQLITE_ENABLE_STMT_SCANSTATUS
3234 static void display_explain_scanstats(
3235 sqlite3 *db, /* Database to query */
3236 ShellState *pArg /* Pointer to ShellState */
3238 static const int f = SQLITE_SCANSTAT_COMPLEX;
3239 sqlite3_stmt *p = pArg->pStmt;
3240 int ii = 0;
3241 i64 nTotal = 0;
3242 int nWidth = 0;
3243 eqp_reset(pArg);
3245 for(ii=0; 1; ii++){
3246 const char *z = 0;
3247 int n = 0;
3248 if( sqlite3_stmt_scanstatus_v2(p,ii,SQLITE_SCANSTAT_EXPLAIN,f,(void*)&z) ){
3249 break;
3251 n = (int)strlen(z) + scanStatsHeight(p, ii)*3;
3252 if( n>nWidth ) nWidth = n;
3254 nWidth += 4;
3256 sqlite3_stmt_scanstatus_v2(p, -1, SQLITE_SCANSTAT_NCYCLE, f, (void*)&nTotal);
3257 for(ii=0; 1; ii++){
3258 i64 nLoop = 0;
3259 i64 nRow = 0;
3260 i64 nCycle = 0;
3261 int iId = 0;
3262 int iPid = 0;
3263 const char *zo = 0;
3264 const char *zName = 0;
3265 char *zText = 0;
3266 double rEst = 0.0;
3268 if( sqlite3_stmt_scanstatus_v2(p,ii,SQLITE_SCANSTAT_EXPLAIN,f,(void*)&zo) ){
3269 break;
3271 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_EST,f,(void*)&rEst);
3272 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_NLOOP,f,(void*)&nLoop);
3273 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_NVISIT,f,(void*)&nRow);
3274 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_NCYCLE,f,(void*)&nCycle);
3275 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_SELECTID,f,(void*)&iId);
3276 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_PARENTID,f,(void*)&iPid);
3277 sqlite3_stmt_scanstatus_v2(p, ii, SQLITE_SCANSTAT_NAME,f,(void*)&zName);
3279 zText = sqlite3_mprintf("%s", zo);
3280 if( nCycle>=0 || nLoop>=0 || nRow>=0 ){
3281 char *z = 0;
3282 if( nCycle>=0 && nTotal>0 ){
3283 z = sqlite3_mprintf("%zcycles=%lld [%d%%]", z,
3284 nCycle, ((nCycle*100)+nTotal/2) / nTotal
3287 if( nLoop>=0 ){
3288 z = sqlite3_mprintf("%z%sloops=%lld", z, z ? " " : "", nLoop);
3290 if( nRow>=0 ){
3291 z = sqlite3_mprintf("%z%srows=%lld", z, z ? " " : "", nRow);
3294 if( zName && pArg->scanstatsOn>1 ){
3295 double rpl = (double)nRow / (double)nLoop;
3296 z = sqlite3_mprintf("%z rpl=%.1f est=%.1f", z, rpl, rEst);
3299 zText = sqlite3_mprintf(
3300 "% *z (%z)", -1*(nWidth-scanStatsHeight(p, ii)*3), zText, z
3304 eqp_append(pArg, iId, iPid, zText);
3305 sqlite3_free(zText);
3308 eqp_render(pArg, nTotal);
3310 #endif
3314 ** Parameter azArray points to a zero-terminated array of strings. zStr
3315 ** points to a single nul-terminated string. Return non-zero if zStr
3316 ** is equal, according to strcmp(), to any of the strings in the array.
3317 ** Otherwise, return zero.
3319 static int str_in_array(const char *zStr, const char **azArray){
3320 int i;
3321 for(i=0; azArray[i]; i++){
3322 if( 0==cli_strcmp(zStr, azArray[i]) ) return 1;
3324 return 0;
3328 ** If compiled statement pSql appears to be an EXPLAIN statement, allocate
3329 ** and populate the ShellState.aiIndent[] array with the number of
3330 ** spaces each opcode should be indented before it is output.
3332 ** The indenting rules are:
3334 ** * For each "Next", "Prev", "VNext" or "VPrev" instruction, indent
3335 ** all opcodes that occur between the p2 jump destination and the opcode
3336 ** itself by 2 spaces.
3338 ** * Do the previous for "Return" instructions for when P2 is positive.
3339 ** See tag-20220407a in wherecode.c and vdbe.c.
3341 ** * For each "Goto", if the jump destination is earlier in the program
3342 ** and ends on one of:
3343 ** Yield SeekGt SeekLt RowSetRead Rewind
3344 ** or if the P1 parameter is one instead of zero,
3345 ** then indent all opcodes between the earlier instruction
3346 ** and "Goto" by 2 spaces.
3348 static void explain_data_prepare(ShellState *p, sqlite3_stmt *pSql){
3349 int *abYield = 0; /* True if op is an OP_Yield */
3350 int nAlloc = 0; /* Allocated size of p->aiIndent[], abYield */
3351 int iOp; /* Index of operation in p->aiIndent[] */
3353 const char *azNext[] = { "Next", "Prev", "VPrev", "VNext", "SorterNext",
3354 "Return", 0 };
3355 const char *azYield[] = { "Yield", "SeekLT", "SeekGT", "RowSetRead",
3356 "Rewind", 0 };
3357 const char *azGoto[] = { "Goto", 0 };
3359 /* The caller guarantees that the leftmost 4 columns of the statement
3360 ** passed to this function are equivalent to the leftmost 4 columns
3361 ** of EXPLAIN statement output. In practice the statement may be
3362 ** an EXPLAIN, or it may be a query on the bytecode() virtual table. */
3363 assert( sqlite3_column_count(pSql)>=4 );
3364 assert( 0==sqlite3_stricmp( sqlite3_column_name(pSql, 0), "addr" ) );
3365 assert( 0==sqlite3_stricmp( sqlite3_column_name(pSql, 1), "opcode" ) );
3366 assert( 0==sqlite3_stricmp( sqlite3_column_name(pSql, 2), "p1" ) );
3367 assert( 0==sqlite3_stricmp( sqlite3_column_name(pSql, 3), "p2" ) );
3369 for(iOp=0; SQLITE_ROW==sqlite3_step(pSql); iOp++){
3370 int i;
3371 int iAddr = sqlite3_column_int(pSql, 0);
3372 const char *zOp = (const char*)sqlite3_column_text(pSql, 1);
3373 int p1 = sqlite3_column_int(pSql, 2);
3374 int p2 = sqlite3_column_int(pSql, 3);
3376 /* Assuming that p2 is an instruction address, set variable p2op to the
3377 ** index of that instruction in the aiIndent[] array. p2 and p2op may be
3378 ** different if the current instruction is part of a sub-program generated
3379 ** by an SQL trigger or foreign key. */
3380 int p2op = (p2 + (iOp-iAddr));
3382 /* Grow the p->aiIndent array as required */
3383 if( iOp>=nAlloc ){
3384 nAlloc += 100;
3385 p->aiIndent = (int*)sqlite3_realloc64(p->aiIndent, nAlloc*sizeof(int));
3386 shell_check_oom(p->aiIndent);
3387 abYield = (int*)sqlite3_realloc64(abYield, nAlloc*sizeof(int));
3388 shell_check_oom(abYield);
3391 abYield[iOp] = str_in_array(zOp, azYield);
3392 p->aiIndent[iOp] = 0;
3393 p->nIndent = iOp+1;
3394 if( str_in_array(zOp, azNext) && p2op>0 ){
3395 for(i=p2op; i<iOp; i++) p->aiIndent[i] += 2;
3397 if( str_in_array(zOp, azGoto) && p2op<iOp && (abYield[p2op] || p1) ){
3398 for(i=p2op; i<iOp; i++) p->aiIndent[i] += 2;
3402 p->iIndent = 0;
3403 sqlite3_free(abYield);
3404 sqlite3_reset(pSql);
3408 ** Free the array allocated by explain_data_prepare().
3410 static void explain_data_delete(ShellState *p){
3411 sqlite3_free(p->aiIndent);
3412 p->aiIndent = 0;
3413 p->nIndent = 0;
3414 p->iIndent = 0;
3417 static void exec_prepared_stmt(ShellState*, sqlite3_stmt*);
3420 ** Display scan stats.
3422 static void display_scanstats(
3423 sqlite3 *db, /* Database to query */
3424 ShellState *pArg /* Pointer to ShellState */
3426 #ifndef SQLITE_ENABLE_STMT_SCANSTATUS
3427 UNUSED_PARAMETER(db);
3428 UNUSED_PARAMETER(pArg);
3429 #else
3430 if( pArg->scanstatsOn==3 ){
3431 const char *zSql =
3432 " SELECT addr, opcode, p1, p2, p3, p4, p5, comment, nexec,"
3433 " round(ncycle*100.0 / (sum(ncycle) OVER ()), 2)||'%' AS cycles"
3434 " FROM bytecode(?)";
3436 int rc = SQLITE_OK;
3437 sqlite3_stmt *pStmt = 0;
3438 rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0);
3439 if( rc==SQLITE_OK ){
3440 sqlite3_stmt *pSave = pArg->pStmt;
3441 pArg->pStmt = pStmt;
3442 sqlite3_bind_pointer(pStmt, 1, pSave, "stmt-pointer", 0);
3444 pArg->cnt = 0;
3445 pArg->cMode = MODE_ScanExp;
3446 explain_data_prepare(pArg, pStmt);
3447 exec_prepared_stmt(pArg, pStmt);
3448 explain_data_delete(pArg);
3450 sqlite3_finalize(pStmt);
3451 pArg->pStmt = pSave;
3453 }else{
3454 display_explain_scanstats(db, pArg);
3456 #endif
3460 ** Disable and restore .wheretrace and .treetrace/.selecttrace settings.
3462 static unsigned int savedSelectTrace;
3463 static unsigned int savedWhereTrace;
3464 static void disable_debug_trace_modes(void){
3465 unsigned int zero = 0;
3466 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 0, &savedSelectTrace);
3467 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 1, &zero);
3468 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 2, &savedWhereTrace);
3469 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 3, &zero);
3471 static void restore_debug_trace_modes(void){
3472 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 1, &savedSelectTrace);
3473 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 3, &savedWhereTrace);
3476 /* Create the TEMP table used to store parameter bindings */
3477 static void bind_table_init(ShellState *p){
3478 int wrSchema = 0;
3479 int defensiveMode = 0;
3480 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, -1, &defensiveMode);
3481 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, 0, 0);
3482 sqlite3_db_config(p->db, SQLITE_DBCONFIG_WRITABLE_SCHEMA, -1, &wrSchema);
3483 sqlite3_db_config(p->db, SQLITE_DBCONFIG_WRITABLE_SCHEMA, 1, 0);
3484 sqlite3_exec(p->db,
3485 "CREATE TABLE IF NOT EXISTS temp.sqlite_parameters(\n"
3486 " key TEXT PRIMARY KEY,\n"
3487 " value\n"
3488 ") WITHOUT ROWID;",
3489 0, 0, 0);
3490 sqlite3_db_config(p->db, SQLITE_DBCONFIG_WRITABLE_SCHEMA, wrSchema, 0);
3491 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, defensiveMode, 0);
3495 ** Bind parameters on a prepared statement.
3497 ** Parameter bindings are taken from a TEMP table of the form:
3499 ** CREATE TEMP TABLE sqlite_parameters(key TEXT PRIMARY KEY, value)
3500 ** WITHOUT ROWID;
3502 ** No bindings occur if this table does not exist. The name of the table
3503 ** begins with "sqlite_" so that it will not collide with ordinary application
3504 ** tables. The table must be in the TEMP schema.
3506 static void bind_prepared_stmt(ShellState *pArg, sqlite3_stmt *pStmt){
3507 int nVar;
3508 int i;
3509 int rc;
3510 sqlite3_stmt *pQ = 0;
3512 nVar = sqlite3_bind_parameter_count(pStmt);
3513 if( nVar==0 ) return; /* Nothing to do */
3514 if( sqlite3_table_column_metadata(pArg->db, "TEMP", "sqlite_parameters",
3515 "key", 0, 0, 0, 0, 0)!=SQLITE_OK ){
3516 rc = SQLITE_NOTFOUND;
3517 pQ = 0;
3518 }else{
3519 rc = sqlite3_prepare_v2(pArg->db,
3520 "SELECT value FROM temp.sqlite_parameters"
3521 " WHERE key=?1", -1, &pQ, 0);
3523 for(i=1; i<=nVar; i++){
3524 char zNum[30];
3525 const char *zVar = sqlite3_bind_parameter_name(pStmt, i);
3526 if( zVar==0 ){
3527 sqlite3_snprintf(sizeof(zNum),zNum,"?%d",i);
3528 zVar = zNum;
3530 sqlite3_bind_text(pQ, 1, zVar, -1, SQLITE_STATIC);
3531 if( rc==SQLITE_OK && pQ && sqlite3_step(pQ)==SQLITE_ROW ){
3532 sqlite3_bind_value(pStmt, i, sqlite3_column_value(pQ, 0));
3533 #ifdef NAN
3534 }else if( sqlite3_strlike("_NAN", zVar, 0)==0 ){
3535 sqlite3_bind_double(pStmt, i, NAN);
3536 #endif
3537 #ifdef INFINITY
3538 }else if( sqlite3_strlike("_INF", zVar, 0)==0 ){
3539 sqlite3_bind_double(pStmt, i, INFINITY);
3540 #endif
3541 }else{
3542 sqlite3_bind_null(pStmt, i);
3544 sqlite3_reset(pQ);
3546 sqlite3_finalize(pQ);
3550 ** UTF8 box-drawing characters. Imagine box lines like this:
3552 ** 1
3553 ** |
3554 ** 4 --+-- 2
3555 ** |
3556 ** 3
3558 ** Each box characters has between 2 and 4 of the lines leading from
3559 ** the center. The characters are here identified by the numbers of
3560 ** their corresponding lines.
3562 #define BOX_24 "\342\224\200" /* U+2500 --- */
3563 #define BOX_13 "\342\224\202" /* U+2502 | */
3564 #define BOX_23 "\342\224\214" /* U+250c ,- */
3565 #define BOX_34 "\342\224\220" /* U+2510 -, */
3566 #define BOX_12 "\342\224\224" /* U+2514 '- */
3567 #define BOX_14 "\342\224\230" /* U+2518 -' */
3568 #define BOX_123 "\342\224\234" /* U+251c |- */
3569 #define BOX_134 "\342\224\244" /* U+2524 -| */
3570 #define BOX_234 "\342\224\254" /* U+252c -,- */
3571 #define BOX_124 "\342\224\264" /* U+2534 -'- */
3572 #define BOX_1234 "\342\224\274" /* U+253c -|- */
3574 /* Draw horizontal line N characters long using unicode box
3575 ** characters
3577 static void print_box_line(int N){
3578 const char zDash[] =
3579 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24
3580 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24 BOX_24;
3581 const int nDash = sizeof(zDash) - 1;
3582 N *= 3;
3583 while( N>nDash ){
3584 oputz(zDash);
3585 N -= nDash;
3587 oputf("%.*s", N, zDash);
3591 ** Draw a horizontal separator for a MODE_Box table.
3593 static void print_box_row_separator(
3594 ShellState *p,
3595 int nArg,
3596 const char *zSep1,
3597 const char *zSep2,
3598 const char *zSep3
3600 int i;
3601 if( nArg>0 ){
3602 oputz(zSep1);
3603 print_box_line(p->actualWidth[0]+2);
3604 for(i=1; i<nArg; i++){
3605 oputz(zSep2);
3606 print_box_line(p->actualWidth[i]+2);
3608 oputz(zSep3);
3610 oputz("\n");
3614 ** z[] is a line of text that is to be displayed the .mode box or table or
3615 ** similar tabular formats. z[] might contain control characters such
3616 ** as \n, \t, \f, or \r.
3618 ** Compute characters to display on the first line of z[]. Stop at the
3619 ** first \r, \n, or \f. Expand \t into spaces. Return a copy (obtained
3620 ** from malloc()) of that first line, which caller should free sometime.
3621 ** Write anything to display on the next line into *pzTail. If this is
3622 ** the last line, write a NULL into *pzTail. (*pzTail is not allocated.)
3624 static char *translateForDisplayAndDup(
3625 const unsigned char *z, /* Input text to be transformed */
3626 const unsigned char **pzTail, /* OUT: Tail of the input for next line */
3627 int mxWidth, /* Max width. 0 means no limit */
3628 u8 bWordWrap /* If true, avoid breaking mid-word */
3630 int i; /* Input bytes consumed */
3631 int j; /* Output bytes generated */
3632 int k; /* Input bytes to be displayed */
3633 int n; /* Output column number */
3634 unsigned char *zOut; /* Output text */
3636 if( z==0 ){
3637 *pzTail = 0;
3638 return 0;
3640 if( mxWidth<0 ) mxWidth = -mxWidth;
3641 if( mxWidth==0 ) mxWidth = 1000000;
3642 i = j = n = 0;
3643 while( n<mxWidth ){
3644 if( z[i]>=' ' ){
3645 n++;
3646 do{ i++; j++; }while( (z[i]&0xc0)==0x80 );
3647 continue;
3649 if( z[i]=='\t' ){
3651 n++;
3652 j++;
3653 }while( (n&7)!=0 && n<mxWidth );
3654 i++;
3655 continue;
3657 break;
3659 if( n>=mxWidth && bWordWrap ){
3660 /* Perhaps try to back up to a better place to break the line */
3661 for(k=i; k>i/2; k--){
3662 if( isspace(z[k-1]) ) break;
3664 if( k<=i/2 ){
3665 for(k=i; k>i/2; k--){
3666 if( isalnum(z[k-1])!=isalnum(z[k]) && (z[k]&0xc0)!=0x80 ) break;
3669 if( k<=i/2 ){
3670 k = i;
3671 }else{
3672 i = k;
3673 while( z[i]==' ' ) i++;
3675 }else{
3676 k = i;
3678 if( n>=mxWidth && z[i]>=' ' ){
3679 *pzTail = &z[i];
3680 }else if( z[i]=='\r' && z[i+1]=='\n' ){
3681 *pzTail = z[i+2] ? &z[i+2] : 0;
3682 }else if( z[i]==0 || z[i+1]==0 ){
3683 *pzTail = 0;
3684 }else{
3685 *pzTail = &z[i+1];
3687 zOut = malloc( j+1 );
3688 shell_check_oom(zOut);
3689 i = j = n = 0;
3690 while( i<k ){
3691 if( z[i]>=' ' ){
3692 n++;
3693 do{ zOut[j++] = z[i++]; }while( (z[i]&0xc0)==0x80 );
3694 continue;
3696 if( z[i]=='\t' ){
3698 n++;
3699 zOut[j++] = ' ';
3700 }while( (n&7)!=0 && n<mxWidth );
3701 i++;
3702 continue;
3704 break;
3706 zOut[j] = 0;
3707 return (char*)zOut;
3710 /* Extract the value of the i-th current column for pStmt as an SQL literal
3711 ** value. Memory is obtained from sqlite3_malloc64() and must be freed by
3712 ** the caller.
3714 static char *quoted_column(sqlite3_stmt *pStmt, int i){
3715 switch( sqlite3_column_type(pStmt, i) ){
3716 case SQLITE_NULL: {
3717 return sqlite3_mprintf("NULL");
3719 case SQLITE_INTEGER:
3720 case SQLITE_FLOAT: {
3721 return sqlite3_mprintf("%s",sqlite3_column_text(pStmt,i));
3723 case SQLITE_TEXT: {
3724 return sqlite3_mprintf("%Q",sqlite3_column_text(pStmt,i));
3726 case SQLITE_BLOB: {
3727 int j;
3728 sqlite3_str *pStr = sqlite3_str_new(0);
3729 const unsigned char *a = sqlite3_column_blob(pStmt,i);
3730 int n = sqlite3_column_bytes(pStmt,i);
3731 sqlite3_str_append(pStr, "x'", 2);
3732 for(j=0; j<n; j++){
3733 sqlite3_str_appendf(pStr, "%02x", a[j]);
3735 sqlite3_str_append(pStr, "'", 1);
3736 return sqlite3_str_finish(pStr);
3739 return 0; /* Not reached */
3743 ** Run a prepared statement and output the result in one of the
3744 ** table-oriented formats: MODE_Column, MODE_Markdown, MODE_Table,
3745 ** or MODE_Box.
3747 ** This is different from ordinary exec_prepared_stmt() in that
3748 ** it has to run the entire query and gather the results into memory
3749 ** first, in order to determine column widths, before providing
3750 ** any output.
3752 static void exec_prepared_stmt_columnar(
3753 ShellState *p, /* Pointer to ShellState */
3754 sqlite3_stmt *pStmt /* Statement to run */
3756 sqlite3_int64 nRow = 0;
3757 int nColumn = 0;
3758 char **azData = 0;
3759 sqlite3_int64 nAlloc = 0;
3760 char *abRowDiv = 0;
3761 const unsigned char *uz;
3762 const char *z;
3763 char **azQuoted = 0;
3764 int rc;
3765 sqlite3_int64 i, nData;
3766 int j, nTotal, w, n;
3767 const char *colSep = 0;
3768 const char *rowSep = 0;
3769 const unsigned char **azNextLine = 0;
3770 int bNextLine = 0;
3771 int bMultiLineRowExists = 0;
3772 int bw = p->cmOpts.bWordWrap;
3773 const char *zEmpty = "";
3774 const char *zShowNull = p->nullValue;
3776 rc = sqlite3_step(pStmt);
3777 if( rc!=SQLITE_ROW ) return;
3778 nColumn = sqlite3_column_count(pStmt);
3779 if( nColumn==0 ) goto columnar_end;
3780 nAlloc = nColumn*4;
3781 if( nAlloc<=0 ) nAlloc = 1;
3782 azData = sqlite3_malloc64( nAlloc*sizeof(char*) );
3783 shell_check_oom(azData);
3784 azNextLine = sqlite3_malloc64( nColumn*sizeof(char*) );
3785 shell_check_oom(azNextLine);
3786 memset((void*)azNextLine, 0, nColumn*sizeof(char*) );
3787 if( p->cmOpts.bQuote ){
3788 azQuoted = sqlite3_malloc64( nColumn*sizeof(char*) );
3789 shell_check_oom(azQuoted);
3790 memset(azQuoted, 0, nColumn*sizeof(char*) );
3792 abRowDiv = sqlite3_malloc64( nAlloc/nColumn );
3793 shell_check_oom(abRowDiv);
3794 if( nColumn>p->nWidth ){
3795 p->colWidth = realloc(p->colWidth, (nColumn+1)*2*sizeof(int));
3796 shell_check_oom(p->colWidth);
3797 for(i=p->nWidth; i<nColumn; i++) p->colWidth[i] = 0;
3798 p->nWidth = nColumn;
3799 p->actualWidth = &p->colWidth[nColumn];
3801 memset(p->actualWidth, 0, nColumn*sizeof(int));
3802 for(i=0; i<nColumn; i++){
3803 w = p->colWidth[i];
3804 if( w<0 ) w = -w;
3805 p->actualWidth[i] = w;
3807 for(i=0; i<nColumn; i++){
3808 const unsigned char *zNotUsed;
3809 int wx = p->colWidth[i];
3810 if( wx==0 ){
3811 wx = p->cmOpts.iWrap;
3813 if( wx<0 ) wx = -wx;
3814 uz = (const unsigned char*)sqlite3_column_name(pStmt,i);
3815 if( uz==0 ) uz = (u8*)"";
3816 azData[i] = translateForDisplayAndDup(uz, &zNotUsed, wx, bw);
3819 int useNextLine = bNextLine;
3820 bNextLine = 0;
3821 if( (nRow+2)*nColumn >= nAlloc ){
3822 nAlloc *= 2;
3823 azData = sqlite3_realloc64(azData, nAlloc*sizeof(char*));
3824 shell_check_oom(azData);
3825 abRowDiv = sqlite3_realloc64(abRowDiv, nAlloc/nColumn);
3826 shell_check_oom(abRowDiv);
3828 abRowDiv[nRow] = 1;
3829 nRow++;
3830 for(i=0; i<nColumn; i++){
3831 int wx = p->colWidth[i];
3832 if( wx==0 ){
3833 wx = p->cmOpts.iWrap;
3835 if( wx<0 ) wx = -wx;
3836 if( useNextLine ){
3837 uz = azNextLine[i];
3838 if( uz==0 ) uz = (u8*)zEmpty;
3839 }else if( p->cmOpts.bQuote ){
3840 sqlite3_free(azQuoted[i]);
3841 azQuoted[i] = quoted_column(pStmt,i);
3842 uz = (const unsigned char*)azQuoted[i];
3843 }else{
3844 uz = (const unsigned char*)sqlite3_column_text(pStmt,i);
3845 if( uz==0 ) uz = (u8*)zShowNull;
3847 azData[nRow*nColumn + i]
3848 = translateForDisplayAndDup(uz, &azNextLine[i], wx, bw);
3849 if( azNextLine[i] ){
3850 bNextLine = 1;
3851 abRowDiv[nRow-1] = 0;
3852 bMultiLineRowExists = 1;
3855 }while( bNextLine || sqlite3_step(pStmt)==SQLITE_ROW );
3856 nTotal = nColumn*(nRow+1);
3857 for(i=0; i<nTotal; i++){
3858 z = azData[i];
3859 if( z==0 ) z = (char*)zEmpty;
3860 n = strlenChar(z);
3861 j = i%nColumn;
3862 if( n>p->actualWidth[j] ) p->actualWidth[j] = n;
3864 if( seenInterrupt ) goto columnar_end;
3865 switch( p->cMode ){
3866 case MODE_Column: {
3867 colSep = " ";
3868 rowSep = "\n";
3869 if( p->showHeader ){
3870 for(i=0; i<nColumn; i++){
3871 w = p->actualWidth[i];
3872 if( p->colWidth[i]<0 ) w = -w;
3873 utf8_width_print(w, azData[i]);
3874 fputs(i==nColumn-1?"\n":" ", p->out);
3876 for(i=0; i<nColumn; i++){
3877 print_dashes(p->actualWidth[i]);
3878 fputs(i==nColumn-1?"\n":" ", p->out);
3881 break;
3883 case MODE_Table: {
3884 colSep = " | ";
3885 rowSep = " |\n";
3886 print_row_separator(p, nColumn, "+");
3887 fputs("| ", p->out);
3888 for(i=0; i<nColumn; i++){
3889 w = p->actualWidth[i];
3890 n = strlenChar(azData[i]);
3891 oputf("%*s%s%*s", (w-n)/2, "", azData[i], (w-n+1)/2, "");
3892 oputz(i==nColumn-1?" |\n":" | ");
3894 print_row_separator(p, nColumn, "+");
3895 break;
3897 case MODE_Markdown: {
3898 colSep = " | ";
3899 rowSep = " |\n";
3900 fputs("| ", p->out);
3901 for(i=0; i<nColumn; i++){
3902 w = p->actualWidth[i];
3903 n = strlenChar(azData[i]);
3904 oputf("%*s%s%*s", (w-n)/2, "", azData[i], (w-n+1)/2, "");
3905 oputz(i==nColumn-1?" |\n":" | ");
3907 print_row_separator(p, nColumn, "|");
3908 break;
3910 case MODE_Box: {
3911 colSep = " " BOX_13 " ";
3912 rowSep = " " BOX_13 "\n";
3913 print_box_row_separator(p, nColumn, BOX_23, BOX_234, BOX_34);
3914 oputz(BOX_13 " ");
3915 for(i=0; i<nColumn; i++){
3916 w = p->actualWidth[i];
3917 n = strlenChar(azData[i]);
3918 oputf("%*s%s%*s%s",
3919 (w-n)/2, "", azData[i], (w-n+1)/2, "",
3920 i==nColumn-1?" "BOX_13"\n":" "BOX_13" ");
3922 print_box_row_separator(p, nColumn, BOX_123, BOX_1234, BOX_134);
3923 break;
3926 for(i=nColumn, j=0; i<nTotal; i++, j++){
3927 if( j==0 && p->cMode!=MODE_Column ){
3928 oputz(p->cMode==MODE_Box?BOX_13" ":"| ");
3930 z = azData[i];
3931 if( z==0 ) z = p->nullValue;
3932 w = p->actualWidth[j];
3933 if( p->colWidth[j]<0 ) w = -w;
3934 utf8_width_print(w, z);
3935 if( j==nColumn-1 ){
3936 oputz(rowSep);
3937 if( bMultiLineRowExists && abRowDiv[i/nColumn-1] && i+1<nTotal ){
3938 if( p->cMode==MODE_Table ){
3939 print_row_separator(p, nColumn, "+");
3940 }else if( p->cMode==MODE_Box ){
3941 print_box_row_separator(p, nColumn, BOX_123, BOX_1234, BOX_134);
3942 }else if( p->cMode==MODE_Column ){
3943 oputz("\n");
3946 j = -1;
3947 if( seenInterrupt ) goto columnar_end;
3948 }else{
3949 oputz(colSep);
3952 if( p->cMode==MODE_Table ){
3953 print_row_separator(p, nColumn, "+");
3954 }else if( p->cMode==MODE_Box ){
3955 print_box_row_separator(p, nColumn, BOX_12, BOX_124, BOX_14);
3957 columnar_end:
3958 if( seenInterrupt ){
3959 oputz("Interrupt\n");
3961 nData = (nRow+1)*nColumn;
3962 for(i=0; i<nData; i++){
3963 z = azData[i];
3964 if( z!=zEmpty && z!=zShowNull ) free(azData[i]);
3966 sqlite3_free(azData);
3967 sqlite3_free((void*)azNextLine);
3968 sqlite3_free(abRowDiv);
3969 if( azQuoted ){
3970 for(i=0; i<nColumn; i++) sqlite3_free(azQuoted[i]);
3971 sqlite3_free(azQuoted);
3976 ** Run a prepared statement
3978 static void exec_prepared_stmt(
3979 ShellState *pArg, /* Pointer to ShellState */
3980 sqlite3_stmt *pStmt /* Statement to run */
3982 int rc;
3983 sqlite3_uint64 nRow = 0;
3985 if( pArg->cMode==MODE_Column
3986 || pArg->cMode==MODE_Table
3987 || pArg->cMode==MODE_Box
3988 || pArg->cMode==MODE_Markdown
3990 exec_prepared_stmt_columnar(pArg, pStmt);
3991 return;
3994 /* perform the first step. this will tell us if we
3995 ** have a result set or not and how wide it is.
3997 rc = sqlite3_step(pStmt);
3998 /* if we have a result set... */
3999 if( SQLITE_ROW == rc ){
4000 /* allocate space for col name ptr, value ptr, and type */
4001 int nCol = sqlite3_column_count(pStmt);
4002 void *pData = sqlite3_malloc64(3*nCol*sizeof(const char*) + 1);
4003 if( !pData ){
4004 shell_out_of_memory();
4005 }else{
4006 char **azCols = (char **)pData; /* Names of result columns */
4007 char **azVals = &azCols[nCol]; /* Results */
4008 int *aiTypes = (int *)&azVals[nCol]; /* Result types */
4009 int i, x;
4010 assert(sizeof(int) <= sizeof(char *));
4011 /* save off ptrs to column names */
4012 for(i=0; i<nCol; i++){
4013 azCols[i] = (char *)sqlite3_column_name(pStmt, i);
4016 nRow++;
4017 /* extract the data and data types */
4018 for(i=0; i<nCol; i++){
4019 aiTypes[i] = x = sqlite3_column_type(pStmt, i);
4020 if( x==SQLITE_BLOB
4021 && pArg
4022 && (pArg->cMode==MODE_Insert || pArg->cMode==MODE_Quote)
4024 azVals[i] = "";
4025 }else{
4026 azVals[i] = (char*)sqlite3_column_text(pStmt, i);
4028 if( !azVals[i] && (aiTypes[i]!=SQLITE_NULL) ){
4029 rc = SQLITE_NOMEM;
4030 break; /* from for */
4032 } /* end for */
4034 /* if data and types extracted successfully... */
4035 if( SQLITE_ROW == rc ){
4036 /* call the supplied callback with the result row data */
4037 if( shell_callback(pArg, nCol, azVals, azCols, aiTypes) ){
4038 rc = SQLITE_ABORT;
4039 }else{
4040 rc = sqlite3_step(pStmt);
4043 } while( SQLITE_ROW == rc );
4044 sqlite3_free(pData);
4045 if( pArg->cMode==MODE_Json ){
4046 fputs("]\n", pArg->out);
4047 }else if( pArg->cMode==MODE_Count ){
4048 char zBuf[200];
4049 sqlite3_snprintf(sizeof(zBuf), zBuf, "%llu row%s\n",
4050 nRow, nRow!=1 ? "s" : "");
4051 printf("%s", zBuf);
4057 #ifndef SQLITE_OMIT_VIRTUALTABLE
4059 ** This function is called to process SQL if the previous shell command
4060 ** was ".expert". It passes the SQL in the second argument directly to
4061 ** the sqlite3expert object.
4063 ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error
4064 ** code. In this case, (*pzErr) may be set to point to a buffer containing
4065 ** an English language error message. It is the responsibility of the
4066 ** caller to eventually free this buffer using sqlite3_free().
4068 static int expertHandleSQL(
4069 ShellState *pState,
4070 const char *zSql,
4071 char **pzErr
4073 assert( pState->expert.pExpert );
4074 assert( pzErr==0 || *pzErr==0 );
4075 return sqlite3_expert_sql(pState->expert.pExpert, zSql, pzErr);
4079 ** This function is called either to silently clean up the object
4080 ** created by the ".expert" command (if bCancel==1), or to generate a
4081 ** report from it and then clean it up (if bCancel==0).
4083 ** If successful, SQLITE_OK is returned. Otherwise, an SQLite error
4084 ** code. In this case, (*pzErr) may be set to point to a buffer containing
4085 ** an English language error message. It is the responsibility of the
4086 ** caller to eventually free this buffer using sqlite3_free().
4088 static int expertFinish(
4089 ShellState *pState,
4090 int bCancel,
4091 char **pzErr
4093 int rc = SQLITE_OK;
4094 sqlite3expert *p = pState->expert.pExpert;
4095 assert( p );
4096 assert( bCancel || pzErr==0 || *pzErr==0 );
4097 if( bCancel==0 ){
4098 int bVerbose = pState->expert.bVerbose;
4100 rc = sqlite3_expert_analyze(p, pzErr);
4101 if( rc==SQLITE_OK ){
4102 int nQuery = sqlite3_expert_count(p);
4103 int i;
4105 if( bVerbose ){
4106 const char *zCand = sqlite3_expert_report(p,0,EXPERT_REPORT_CANDIDATES);
4107 oputz("-- Candidates -----------------------------\n");
4108 oputf("%s\n", zCand);
4110 for(i=0; i<nQuery; i++){
4111 const char *zSql = sqlite3_expert_report(p, i, EXPERT_REPORT_SQL);
4112 const char *zIdx = sqlite3_expert_report(p, i, EXPERT_REPORT_INDEXES);
4113 const char *zEQP = sqlite3_expert_report(p, i, EXPERT_REPORT_PLAN);
4114 if( zIdx==0 ) zIdx = "(no new indexes)\n";
4115 if( bVerbose ){
4116 oputf("-- Query %d --------------------------------\n",i+1);
4117 oputf("%s\n\n", zSql);
4119 oputf("%s\n", zIdx);
4120 oputf("%s\n", zEQP);
4124 sqlite3_expert_destroy(p);
4125 pState->expert.pExpert = 0;
4126 return rc;
4130 ** Implementation of ".expert" dot command.
4132 static int expertDotCommand(
4133 ShellState *pState, /* Current shell tool state */
4134 char **azArg, /* Array of arguments passed to dot command */
4135 int nArg /* Number of entries in azArg[] */
4137 int rc = SQLITE_OK;
4138 char *zErr = 0;
4139 int i;
4140 int iSample = 0;
4142 assert( pState->expert.pExpert==0 );
4143 memset(&pState->expert, 0, sizeof(ExpertInfo));
4145 for(i=1; rc==SQLITE_OK && i<nArg; i++){
4146 char *z = azArg[i];
4147 int n;
4148 if( z[0]=='-' && z[1]=='-' ) z++;
4149 n = strlen30(z);
4150 if( n>=2 && 0==cli_strncmp(z, "-verbose", n) ){
4151 pState->expert.bVerbose = 1;
4153 else if( n>=2 && 0==cli_strncmp(z, "-sample", n) ){
4154 if( i==(nArg-1) ){
4155 eputf("option requires an argument: %s\n", z);
4156 rc = SQLITE_ERROR;
4157 }else{
4158 iSample = (int)integerValue(azArg[++i]);
4159 if( iSample<0 || iSample>100 ){
4160 eputf("value out of range: %s\n", azArg[i]);
4161 rc = SQLITE_ERROR;
4165 else{
4166 eputf("unknown option: %s\n", z);
4167 rc = SQLITE_ERROR;
4171 if( rc==SQLITE_OK ){
4172 pState->expert.pExpert = sqlite3_expert_new(pState->db, &zErr);
4173 if( pState->expert.pExpert==0 ){
4174 eputf("sqlite3_expert_new: %s\n", zErr ? zErr : "out of memory");
4175 rc = SQLITE_ERROR;
4176 }else{
4177 sqlite3_expert_config(
4178 pState->expert.pExpert, EXPERT_CONFIG_SAMPLE, iSample
4182 sqlite3_free(zErr);
4184 return rc;
4186 #endif /* ifndef SQLITE_OMIT_VIRTUALTABLE */
4189 ** Execute a statement or set of statements. Print
4190 ** any result rows/columns depending on the current mode
4191 ** set via the supplied callback.
4193 ** This is very similar to SQLite's built-in sqlite3_exec()
4194 ** function except it takes a slightly different callback
4195 ** and callback data argument.
4197 static int shell_exec(
4198 ShellState *pArg, /* Pointer to ShellState */
4199 const char *zSql, /* SQL to be evaluated */
4200 char **pzErrMsg /* Error msg written here */
4202 sqlite3_stmt *pStmt = NULL; /* Statement to execute. */
4203 int rc = SQLITE_OK; /* Return Code */
4204 int rc2;
4205 const char *zLeftover; /* Tail of unprocessed SQL */
4206 sqlite3 *db = pArg->db;
4208 if( pzErrMsg ){
4209 *pzErrMsg = NULL;
4212 #ifndef SQLITE_OMIT_VIRTUALTABLE
4213 if( pArg->expert.pExpert ){
4214 rc = expertHandleSQL(pArg, zSql, pzErrMsg);
4215 return expertFinish(pArg, (rc!=SQLITE_OK), pzErrMsg);
4217 #endif
4219 while( zSql[0] && (SQLITE_OK == rc) ){
4220 static const char *zStmtSql;
4221 rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, &zLeftover);
4222 if( SQLITE_OK != rc ){
4223 if( pzErrMsg ){
4224 *pzErrMsg = save_err_msg(db, "in prepare", rc, zSql);
4226 }else{
4227 if( !pStmt ){
4228 /* this happens for a comment or white-space */
4229 zSql = zLeftover;
4230 while( IsSpace(zSql[0]) ) zSql++;
4231 continue;
4233 zStmtSql = sqlite3_sql(pStmt);
4234 if( zStmtSql==0 ) zStmtSql = "";
4235 while( IsSpace(zStmtSql[0]) ) zStmtSql++;
4237 /* save off the prepared statement handle and reset row count */
4238 if( pArg ){
4239 pArg->pStmt = pStmt;
4240 pArg->cnt = 0;
4243 /* Show the EXPLAIN QUERY PLAN if .eqp is on */
4244 if( pArg && pArg->autoEQP && sqlite3_stmt_isexplain(pStmt)==0 ){
4245 sqlite3_stmt *pExplain;
4246 int triggerEQP = 0;
4247 disable_debug_trace_modes();
4248 sqlite3_db_config(db, SQLITE_DBCONFIG_TRIGGER_EQP, -1, &triggerEQP);
4249 if( pArg->autoEQP>=AUTOEQP_trigger ){
4250 sqlite3_db_config(db, SQLITE_DBCONFIG_TRIGGER_EQP, 1, 0);
4252 pExplain = pStmt;
4253 sqlite3_reset(pExplain);
4254 rc = sqlite3_stmt_explain(pExplain, 2);
4255 if( rc==SQLITE_OK ){
4256 while( sqlite3_step(pExplain)==SQLITE_ROW ){
4257 const char *zEQPLine = (const char*)sqlite3_column_text(pExplain,3);
4258 int iEqpId = sqlite3_column_int(pExplain, 0);
4259 int iParentId = sqlite3_column_int(pExplain, 1);
4260 if( zEQPLine==0 ) zEQPLine = "";
4261 if( zEQPLine[0]=='-' ) eqp_render(pArg, 0);
4262 eqp_append(pArg, iEqpId, iParentId, zEQPLine);
4264 eqp_render(pArg, 0);
4266 if( pArg->autoEQP>=AUTOEQP_full ){
4267 /* Also do an EXPLAIN for ".eqp full" mode */
4268 sqlite3_reset(pExplain);
4269 rc = sqlite3_stmt_explain(pExplain, 1);
4270 if( rc==SQLITE_OK ){
4271 pArg->cMode = MODE_Explain;
4272 assert( sqlite3_stmt_isexplain(pExplain)==1 );
4273 explain_data_prepare(pArg, pExplain);
4274 exec_prepared_stmt(pArg, pExplain);
4275 explain_data_delete(pArg);
4278 if( pArg->autoEQP>=AUTOEQP_trigger && triggerEQP==0 ){
4279 sqlite3_db_config(db, SQLITE_DBCONFIG_TRIGGER_EQP, 0, 0);
4281 sqlite3_reset(pStmt);
4282 sqlite3_stmt_explain(pStmt, 0);
4283 restore_debug_trace_modes();
4286 if( pArg ){
4287 int bIsExplain = (sqlite3_stmt_isexplain(pStmt)==1);
4288 pArg->cMode = pArg->mode;
4289 if( pArg->autoExplain ){
4290 if( bIsExplain ){
4291 pArg->cMode = MODE_Explain;
4293 if( sqlite3_stmt_isexplain(pStmt)==2 ){
4294 pArg->cMode = MODE_EQP;
4298 /* If the shell is currently in ".explain" mode, gather the extra
4299 ** data required to add indents to the output.*/
4300 if( pArg->cMode==MODE_Explain && bIsExplain ){
4301 explain_data_prepare(pArg, pStmt);
4305 bind_prepared_stmt(pArg, pStmt);
4306 exec_prepared_stmt(pArg, pStmt);
4307 explain_data_delete(pArg);
4308 eqp_render(pArg, 0);
4310 /* print usage stats if stats on */
4311 if( pArg && pArg->statsOn ){
4312 display_stats(db, pArg, 0);
4315 /* print loop-counters if required */
4316 if( pArg && pArg->scanstatsOn ){
4317 display_scanstats(db, pArg);
4320 /* Finalize the statement just executed. If this fails, save a
4321 ** copy of the error message. Otherwise, set zSql to point to the
4322 ** next statement to execute. */
4323 rc2 = sqlite3_finalize(pStmt);
4324 if( rc!=SQLITE_NOMEM ) rc = rc2;
4325 if( rc==SQLITE_OK ){
4326 zSql = zLeftover;
4327 while( IsSpace(zSql[0]) ) zSql++;
4328 }else if( pzErrMsg ){
4329 *pzErrMsg = save_err_msg(db, "stepping", rc, 0);
4332 /* clear saved stmt handle */
4333 if( pArg ){
4334 pArg->pStmt = NULL;
4337 } /* end while */
4339 return rc;
4343 ** Release memory previously allocated by tableColumnList().
4345 static void freeColumnList(char **azCol){
4346 int i;
4347 for(i=1; azCol[i]; i++){
4348 sqlite3_free(azCol[i]);
4350 /* azCol[0] is a static string */
4351 sqlite3_free(azCol);
4355 ** Return a list of pointers to strings which are the names of all
4356 ** columns in table zTab. The memory to hold the names is dynamically
4357 ** allocated and must be released by the caller using a subsequent call
4358 ** to freeColumnList().
4360 ** The azCol[0] entry is usually NULL. However, if zTab contains a rowid
4361 ** value that needs to be preserved, then azCol[0] is filled in with the
4362 ** name of the rowid column.
4364 ** The first regular column in the table is azCol[1]. The list is terminated
4365 ** by an entry with azCol[i]==0.
4367 static char **tableColumnList(ShellState *p, const char *zTab){
4368 char **azCol = 0;
4369 sqlite3_stmt *pStmt;
4370 char *zSql;
4371 int nCol = 0;
4372 int nAlloc = 0;
4373 int nPK = 0; /* Number of PRIMARY KEY columns seen */
4374 int isIPK = 0; /* True if one PRIMARY KEY column of type INTEGER */
4375 int preserveRowid = ShellHasFlag(p, SHFLG_PreserveRowid);
4376 int rc;
4378 zSql = sqlite3_mprintf("PRAGMA table_info=%Q", zTab);
4379 shell_check_oom(zSql);
4380 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
4381 sqlite3_free(zSql);
4382 if( rc ) return 0;
4383 while( sqlite3_step(pStmt)==SQLITE_ROW ){
4384 if( nCol>=nAlloc-2 ){
4385 nAlloc = nAlloc*2 + nCol + 10;
4386 azCol = sqlite3_realloc(azCol, nAlloc*sizeof(azCol[0]));
4387 shell_check_oom(azCol);
4389 azCol[++nCol] = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 1));
4390 shell_check_oom(azCol[nCol]);
4391 if( sqlite3_column_int(pStmt, 5) ){
4392 nPK++;
4393 if( nPK==1
4394 && sqlite3_stricmp((const char*)sqlite3_column_text(pStmt,2),
4395 "INTEGER")==0
4397 isIPK = 1;
4398 }else{
4399 isIPK = 0;
4403 sqlite3_finalize(pStmt);
4404 if( azCol==0 ) return 0;
4405 azCol[0] = 0;
4406 azCol[nCol+1] = 0;
4408 /* The decision of whether or not a rowid really needs to be preserved
4409 ** is tricky. We never need to preserve a rowid for a WITHOUT ROWID table
4410 ** or a table with an INTEGER PRIMARY KEY. We are unable to preserve
4411 ** rowids on tables where the rowid is inaccessible because there are other
4412 ** columns in the table named "rowid", "_rowid_", and "oid".
4414 if( preserveRowid && isIPK ){
4415 /* If a single PRIMARY KEY column with type INTEGER was seen, then it
4416 ** might be an alias for the ROWID. But it might also be a WITHOUT ROWID
4417 ** table or a INTEGER PRIMARY KEY DESC column, neither of which are
4418 ** ROWID aliases. To distinguish these cases, check to see if
4419 ** there is a "pk" entry in "PRAGMA index_list". There will be
4420 ** no "pk" index if the PRIMARY KEY really is an alias for the ROWID.
4422 zSql = sqlite3_mprintf("SELECT 1 FROM pragma_index_list(%Q)"
4423 " WHERE origin='pk'", zTab);
4424 shell_check_oom(zSql);
4425 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
4426 sqlite3_free(zSql);
4427 if( rc ){
4428 freeColumnList(azCol);
4429 return 0;
4431 rc = sqlite3_step(pStmt);
4432 sqlite3_finalize(pStmt);
4433 preserveRowid = rc==SQLITE_ROW;
4435 if( preserveRowid ){
4436 /* Only preserve the rowid if we can find a name to use for the
4437 ** rowid */
4438 static char *azRowid[] = { "rowid", "_rowid_", "oid" };
4439 int i, j;
4440 for(j=0; j<3; j++){
4441 for(i=1; i<=nCol; i++){
4442 if( sqlite3_stricmp(azRowid[j],azCol[i])==0 ) break;
4444 if( i>nCol ){
4445 /* At this point, we know that azRowid[j] is not the name of any
4446 ** ordinary column in the table. Verify that azRowid[j] is a valid
4447 ** name for the rowid before adding it to azCol[0]. WITHOUT ROWID
4448 ** tables will fail this last check */
4449 rc = sqlite3_table_column_metadata(p->db,0,zTab,azRowid[j],0,0,0,0,0);
4450 if( rc==SQLITE_OK ) azCol[0] = azRowid[j];
4451 break;
4455 return azCol;
4459 ** Toggle the reverse_unordered_selects setting.
4461 static void toggleSelectOrder(sqlite3 *db){
4462 sqlite3_stmt *pStmt = 0;
4463 int iSetting = 0;
4464 char zStmt[100];
4465 sqlite3_prepare_v2(db, "PRAGMA reverse_unordered_selects", -1, &pStmt, 0);
4466 if( sqlite3_step(pStmt)==SQLITE_ROW ){
4467 iSetting = sqlite3_column_int(pStmt, 0);
4469 sqlite3_finalize(pStmt);
4470 sqlite3_snprintf(sizeof(zStmt), zStmt,
4471 "PRAGMA reverse_unordered_selects(%d)", !iSetting);
4472 sqlite3_exec(db, zStmt, 0, 0, 0);
4476 ** This is a different callback routine used for dumping the database.
4477 ** Each row received by this callback consists of a table name,
4478 ** the table type ("index" or "table") and SQL to create the table.
4479 ** This routine should print text sufficient to recreate the table.
4481 static int dump_callback(void *pArg, int nArg, char **azArg, char **azNotUsed){
4482 int rc;
4483 const char *zTable;
4484 const char *zType;
4485 const char *zSql;
4486 ShellState *p = (ShellState *)pArg;
4487 int dataOnly;
4488 int noSys;
4490 UNUSED_PARAMETER(azNotUsed);
4491 if( nArg!=3 || azArg==0 ) return 0;
4492 zTable = azArg[0];
4493 zType = azArg[1];
4494 zSql = azArg[2];
4495 if( zTable==0 ) return 0;
4496 if( zType==0 ) return 0;
4497 dataOnly = (p->shellFlgs & SHFLG_DumpDataOnly)!=0;
4498 noSys = (p->shellFlgs & SHFLG_DumpNoSys)!=0;
4500 if( cli_strcmp(zTable, "sqlite_sequence")==0 && !noSys ){
4501 if( !dataOnly ) oputz("DELETE FROM sqlite_sequence;\n");
4502 }else if( sqlite3_strglob("sqlite_stat?", zTable)==0 && !noSys ){
4503 if( !dataOnly ) oputz("ANALYZE sqlite_schema;\n");
4504 }else if( cli_strncmp(zTable, "sqlite_", 7)==0 ){
4505 return 0;
4506 }else if( dataOnly ){
4507 /* no-op */
4508 }else if( cli_strncmp(zSql, "CREATE VIRTUAL TABLE", 20)==0 ){
4509 char *zIns;
4510 if( !p->writableSchema ){
4511 oputz("PRAGMA writable_schema=ON;\n");
4512 p->writableSchema = 1;
4514 zIns = sqlite3_mprintf(
4515 "INSERT INTO sqlite_schema(type,name,tbl_name,rootpage,sql)"
4516 "VALUES('table','%q','%q',0,'%q');",
4517 zTable, zTable, zSql);
4518 shell_check_oom(zIns);
4519 oputf("%s\n", zIns);
4520 sqlite3_free(zIns);
4521 return 0;
4522 }else{
4523 printSchemaLine(zSql, ";\n");
4526 if( cli_strcmp(zType, "table")==0 ){
4527 ShellText sSelect;
4528 ShellText sTable;
4529 char **azCol;
4530 int i;
4531 char *savedDestTable;
4532 int savedMode;
4534 azCol = tableColumnList(p, zTable);
4535 if( azCol==0 ){
4536 p->nErr++;
4537 return 0;
4540 /* Always quote the table name, even if it appears to be pure ascii,
4541 ** in case it is a keyword. Ex: INSERT INTO "table" ... */
4542 initText(&sTable);
4543 appendText(&sTable, zTable, quoteChar(zTable));
4544 /* If preserving the rowid, add a column list after the table name.
4545 ** In other words: "INSERT INTO tab(rowid,a,b,c,...) VALUES(...)"
4546 ** instead of the usual "INSERT INTO tab VALUES(...)".
4548 if( azCol[0] ){
4549 appendText(&sTable, "(", 0);
4550 appendText(&sTable, azCol[0], 0);
4551 for(i=1; azCol[i]; i++){
4552 appendText(&sTable, ",", 0);
4553 appendText(&sTable, azCol[i], quoteChar(azCol[i]));
4555 appendText(&sTable, ")", 0);
4558 /* Build an appropriate SELECT statement */
4559 initText(&sSelect);
4560 appendText(&sSelect, "SELECT ", 0);
4561 if( azCol[0] ){
4562 appendText(&sSelect, azCol[0], 0);
4563 appendText(&sSelect, ",", 0);
4565 for(i=1; azCol[i]; i++){
4566 appendText(&sSelect, azCol[i], quoteChar(azCol[i]));
4567 if( azCol[i+1] ){
4568 appendText(&sSelect, ",", 0);
4571 freeColumnList(azCol);
4572 appendText(&sSelect, " FROM ", 0);
4573 appendText(&sSelect, zTable, quoteChar(zTable));
4575 savedDestTable = p->zDestTable;
4576 savedMode = p->mode;
4577 p->zDestTable = sTable.z;
4578 p->mode = p->cMode = MODE_Insert;
4579 rc = shell_exec(p, sSelect.z, 0);
4580 if( (rc&0xff)==SQLITE_CORRUPT ){
4581 oputz("/****** CORRUPTION ERROR *******/\n");
4582 toggleSelectOrder(p->db);
4583 shell_exec(p, sSelect.z, 0);
4584 toggleSelectOrder(p->db);
4586 p->zDestTable = savedDestTable;
4587 p->mode = savedMode;
4588 freeText(&sTable);
4589 freeText(&sSelect);
4590 if( rc ) p->nErr++;
4592 return 0;
4596 ** Run zQuery. Use dump_callback() as the callback routine so that
4597 ** the contents of the query are output as SQL statements.
4599 ** If we get a SQLITE_CORRUPT error, rerun the query after appending
4600 ** "ORDER BY rowid DESC" to the end.
4602 static int run_schema_dump_query(
4603 ShellState *p,
4604 const char *zQuery
4606 int rc;
4607 char *zErr = 0;
4608 rc = sqlite3_exec(p->db, zQuery, dump_callback, p, &zErr);
4609 if( rc==SQLITE_CORRUPT ){
4610 char *zQ2;
4611 int len = strlen30(zQuery);
4612 oputz("/****** CORRUPTION ERROR *******/\n");
4613 if( zErr ){
4614 oputf("/****** %s ******/\n", zErr);
4615 sqlite3_free(zErr);
4616 zErr = 0;
4618 zQ2 = malloc( len+100 );
4619 if( zQ2==0 ) return rc;
4620 sqlite3_snprintf(len+100, zQ2, "%s ORDER BY rowid DESC", zQuery);
4621 rc = sqlite3_exec(p->db, zQ2, dump_callback, p, &zErr);
4622 if( rc ){
4623 oputf("/****** ERROR: %s ******/\n", zErr);
4624 }else{
4625 rc = SQLITE_CORRUPT;
4627 sqlite3_free(zErr);
4628 free(zQ2);
4630 return rc;
4634 ** Text of help messages.
4636 ** The help text for each individual command begins with a line that starts
4637 ** with ".". Subsequent lines are supplemental information.
4639 ** There must be two or more spaces between the end of the command and the
4640 ** start of the description of what that command does.
4642 static const char *(azHelp[]) = {
4643 #if defined(SQLITE_HAVE_ZLIB) && !defined(SQLITE_OMIT_VIRTUALTABLE) \
4644 && !defined(SQLITE_SHELL_FIDDLE)
4645 ".archive ... Manage SQL archives",
4646 " Each command must have exactly one of the following options:",
4647 " -c, --create Create a new archive",
4648 " -u, --update Add or update files with changed mtime",
4649 " -i, --insert Like -u but always add even if unchanged",
4650 " -r, --remove Remove files from archive",
4651 " -t, --list List contents of archive",
4652 " -x, --extract Extract files from archive",
4653 " Optional arguments:",
4654 " -v, --verbose Print each filename as it is processed",
4655 " -f FILE, --file FILE Use archive FILE (default is current db)",
4656 " -a FILE, --append FILE Open FILE using the apndvfs VFS",
4657 " -C DIR, --directory DIR Read/extract files from directory DIR",
4658 " -g, --glob Use glob matching for names in archive",
4659 " -n, --dryrun Show the SQL that would have occurred",
4660 " Examples:",
4661 " .ar -cf ARCHIVE foo bar # Create ARCHIVE from files foo and bar",
4662 " .ar -tf ARCHIVE # List members of ARCHIVE",
4663 " .ar -xvf ARCHIVE # Verbosely extract files from ARCHIVE",
4664 " See also:",
4665 " http://sqlite.org/cli.html#sqlite_archive_support",
4666 #endif
4667 #ifndef SQLITE_OMIT_AUTHORIZATION
4668 ".auth ON|OFF Show authorizer callbacks",
4669 #endif
4670 #ifndef SQLITE_SHELL_FIDDLE
4671 ".backup ?DB? FILE Backup DB (default \"main\") to FILE",
4672 " Options:",
4673 " --append Use the appendvfs",
4674 " --async Write to FILE without journal and fsync()",
4675 #endif
4676 ".bail on|off Stop after hitting an error. Default OFF",
4677 #ifndef SQLITE_SHELL_FIDDLE
4678 ".cd DIRECTORY Change the working directory to DIRECTORY",
4679 #endif
4680 ".changes on|off Show number of rows changed by SQL",
4681 #ifndef SQLITE_SHELL_FIDDLE
4682 ".check GLOB Fail if output since .testcase does not match",
4683 ".clone NEWDB Clone data into NEWDB from the existing database",
4684 #endif
4685 ".connection [close] [#] Open or close an auxiliary database connection",
4686 #if defined(_WIN32) || defined(WIN32)
4687 ".crnl on|off Translate \\n to \\r\\n. Default ON",
4688 #endif
4689 ".databases List names and files of attached databases",
4690 ".dbconfig ?op? ?val? List or change sqlite3_db_config() options",
4691 #if SQLITE_SHELL_HAVE_RECOVER
4692 ".dbinfo ?DB? Show status information about the database",
4693 #endif
4694 ".dump ?OBJECTS? Render database content as SQL",
4695 " Options:",
4696 " --data-only Output only INSERT statements",
4697 " --newlines Allow unescaped newline characters in output",
4698 " --nosys Omit system tables (ex: \"sqlite_stat1\")",
4699 " --preserve-rowids Include ROWID values in the output",
4700 " OBJECTS is a LIKE pattern for tables, indexes, triggers or views to dump",
4701 " Additional LIKE patterns can be given in subsequent arguments",
4702 ".echo on|off Turn command echo on or off",
4703 ".eqp on|off|full|... Enable or disable automatic EXPLAIN QUERY PLAN",
4704 " Other Modes:",
4705 #ifdef SQLITE_DEBUG
4706 " test Show raw EXPLAIN QUERY PLAN output",
4707 " trace Like \"full\" but enable \"PRAGMA vdbe_trace\"",
4708 #endif
4709 " trigger Like \"full\" but also show trigger bytecode",
4710 #ifndef SQLITE_SHELL_FIDDLE
4711 ".excel Display the output of next command in spreadsheet",
4712 " --bom Put a UTF8 byte-order mark on intermediate file",
4713 #endif
4714 #ifndef SQLITE_SHELL_FIDDLE
4715 ".exit ?CODE? Exit this program with return-code CODE",
4716 #endif
4717 ".expert EXPERIMENTAL. Suggest indexes for queries",
4718 ".explain ?on|off|auto? Change the EXPLAIN formatting mode. Default: auto",
4719 ".filectrl CMD ... Run various sqlite3_file_control() operations",
4720 " --schema SCHEMA Use SCHEMA instead of \"main\"",
4721 " --help Show CMD details",
4722 ".fullschema ?--indent? Show schema and the content of sqlite_stat tables",
4723 ".headers on|off Turn display of headers on or off",
4724 ".help ?-all? ?PATTERN? Show help text for PATTERN",
4725 #ifndef SQLITE_SHELL_FIDDLE
4726 ".import FILE TABLE Import data from FILE into TABLE",
4727 " Options:",
4728 " --ascii Use \\037 and \\036 as column and row separators",
4729 " --csv Use , and \\n as column and row separators",
4730 " --skip N Skip the first N rows of input",
4731 " --schema S Target table to be S.TABLE",
4732 " -v \"Verbose\" - increase auxiliary output",
4733 " Notes:",
4734 " * If TABLE does not exist, it is created. The first row of input",
4735 " determines the column names.",
4736 " * If neither --csv or --ascii are used, the input mode is derived",
4737 " from the \".mode\" output mode",
4738 " * If FILE begins with \"|\" then it is a command that generates the",
4739 " input text.",
4740 #endif
4741 #ifndef SQLITE_OMIT_TEST_CONTROL
4742 ",imposter INDEX TABLE Create imposter table TABLE on index INDEX",
4743 #endif
4744 ".indexes ?TABLE? Show names of indexes",
4745 " If TABLE is specified, only show indexes for",
4746 " tables matching TABLE using the LIKE operator.",
4747 ".intck ?STEPS_PER_UNLOCK? Run an incremental integrity check on the db",
4748 #ifdef SQLITE_ENABLE_IOTRACE
4749 ",iotrace FILE Enable I/O diagnostic logging to FILE",
4750 #endif
4751 ".limit ?LIMIT? ?VAL? Display or change the value of an SQLITE_LIMIT",
4752 ".lint OPTIONS Report potential schema issues.",
4753 " Options:",
4754 " fkey-indexes Find missing foreign key indexes",
4755 #if !defined(SQLITE_OMIT_LOAD_EXTENSION) && !defined(SQLITE_SHELL_FIDDLE)
4756 ".load FILE ?ENTRY? Load an extension library",
4757 #endif
4758 #if !defined(SQLITE_SHELL_FIDDLE)
4759 ".log FILE|on|off Turn logging on or off. FILE can be stderr/stdout",
4760 #else
4761 ".log on|off Turn logging on or off.",
4762 #endif
4763 ".mode MODE ?OPTIONS? Set output mode",
4764 " MODE is one of:",
4765 " ascii Columns/rows delimited by 0x1F and 0x1E",
4766 " box Tables using unicode box-drawing characters",
4767 " csv Comma-separated values",
4768 " column Output in columns. (See .width)",
4769 " html HTML <table> code",
4770 " insert SQL insert statements for TABLE",
4771 " json Results in a JSON array",
4772 " line One value per line",
4773 " list Values delimited by \"|\"",
4774 " markdown Markdown table format",
4775 " qbox Shorthand for \"box --wrap 60 --quote\"",
4776 " quote Escape answers as for SQL",
4777 " table ASCII-art table",
4778 " tabs Tab-separated values",
4779 " tcl TCL list elements",
4780 " OPTIONS: (for columnar modes or insert mode):",
4781 " --wrap N Wrap output lines to no longer than N characters",
4782 " --wordwrap B Wrap or not at word boundaries per B (on/off)",
4783 " --ww Shorthand for \"--wordwrap 1\"",
4784 " --quote Quote output text as SQL literals",
4785 " --noquote Do not quote output text",
4786 " TABLE The name of SQL table used for \"insert\" mode",
4787 #ifndef SQLITE_SHELL_FIDDLE
4788 ".nonce STRING Suspend safe mode for one command if nonce matches",
4789 #endif
4790 ".nullvalue STRING Use STRING in place of NULL values",
4791 #ifndef SQLITE_SHELL_FIDDLE
4792 ".once ?OPTIONS? ?FILE? Output for the next SQL command only to FILE",
4793 " If FILE begins with '|' then open as a pipe",
4794 " --bom Put a UTF8 byte-order mark at the beginning",
4795 " -e Send output to the system text editor",
4796 " -x Send output as CSV to a spreadsheet (same as \".excel\")",
4797 /* Note that .open is (partially) available in WASM builds but is
4798 ** currently only intended to be used by the fiddle tool, not
4799 ** end users, so is "undocumented." */
4800 ".open ?OPTIONS? ?FILE? Close existing database and reopen FILE",
4801 " Options:",
4802 " --append Use appendvfs to append database to the end of FILE",
4803 #endif
4804 #ifndef SQLITE_OMIT_DESERIALIZE
4805 " --deserialize Load into memory using sqlite3_deserialize()",
4806 " --hexdb Load the output of \"dbtotxt\" as an in-memory db",
4807 " --maxsize N Maximum size for --hexdb or --deserialized database",
4808 #endif
4809 " --new Initialize FILE to an empty database",
4810 " --nofollow Do not follow symbolic links",
4811 " --readonly Open FILE readonly",
4812 " --zip FILE is a ZIP archive",
4813 #ifndef SQLITE_SHELL_FIDDLE
4814 ".output ?FILE? Send output to FILE or stdout if FILE is omitted",
4815 " If FILE begins with '|' then open it as a pipe.",
4816 " Options:",
4817 " --bom Prefix output with a UTF8 byte-order mark",
4818 " -e Send output to the system text editor",
4819 " -x Send output as CSV to a spreadsheet",
4820 #endif
4821 ".parameter CMD ... Manage SQL parameter bindings",
4822 " clear Erase all bindings",
4823 " init Initialize the TEMP table that holds bindings",
4824 " list List the current parameter bindings",
4825 " set PARAMETER VALUE Given SQL parameter PARAMETER a value of VALUE",
4826 " PARAMETER should start with one of: $ : @ ?",
4827 " unset PARAMETER Remove PARAMETER from the binding table",
4828 ".print STRING... Print literal STRING",
4829 #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
4830 ".progress N Invoke progress handler after every N opcodes",
4831 " --limit N Interrupt after N progress callbacks",
4832 " --once Do no more than one progress interrupt",
4833 " --quiet|-q No output except at interrupts",
4834 " --reset Reset the count for each input and interrupt",
4835 #endif
4836 ".prompt MAIN CONTINUE Replace the standard prompts",
4837 #ifndef SQLITE_SHELL_FIDDLE
4838 ".quit Stop interpreting input stream, exit if primary.",
4839 ".read FILE Read input from FILE or command output",
4840 " If FILE begins with \"|\", it is a command that generates the input.",
4841 #endif
4842 #if SQLITE_SHELL_HAVE_RECOVER
4843 ".recover Recover as much data as possible from corrupt db.",
4844 " --ignore-freelist Ignore pages that appear to be on db freelist",
4845 " --lost-and-found TABLE Alternative name for the lost-and-found table",
4846 " --no-rowids Do not attempt to recover rowid values",
4847 " that are not also INTEGER PRIMARY KEYs",
4848 #endif
4849 #ifndef SQLITE_SHELL_FIDDLE
4850 ".restore ?DB? FILE Restore content of DB (default \"main\") from FILE",
4851 ".save ?OPTIONS? FILE Write database to FILE (an alias for .backup ...)",
4852 #endif
4853 ".scanstats on|off|est Turn sqlite3_stmt_scanstatus() metrics on or off",
4854 ".schema ?PATTERN? Show the CREATE statements matching PATTERN",
4855 " Options:",
4856 " --indent Try to pretty-print the schema",
4857 " --nosys Omit objects whose names start with \"sqlite_\"",
4858 ",selftest ?OPTIONS? Run tests defined in the SELFTEST table",
4859 " Options:",
4860 " --init Create a new SELFTEST table",
4861 " -v Verbose output",
4862 ".separator COL ?ROW? Change the column and row separators",
4863 #if defined(SQLITE_ENABLE_SESSION)
4864 ".session ?NAME? CMD ... Create or control sessions",
4865 " Subcommands:",
4866 " attach TABLE Attach TABLE",
4867 " changeset FILE Write a changeset into FILE",
4868 " close Close one session",
4869 " enable ?BOOLEAN? Set or query the enable bit",
4870 " filter GLOB... Reject tables matching GLOBs",
4871 " indirect ?BOOLEAN? Mark or query the indirect status",
4872 " isempty Query whether the session is empty",
4873 " list List currently open session names",
4874 " open DB NAME Open a new session on DB",
4875 " patchset FILE Write a patchset into FILE",
4876 " If ?NAME? is omitted, the first defined session is used.",
4877 #endif
4878 ".sha3sum ... Compute a SHA3 hash of database content",
4879 " Options:",
4880 " --schema Also hash the sqlite_schema table",
4881 " --sha3-224 Use the sha3-224 algorithm",
4882 " --sha3-256 Use the sha3-256 algorithm (default)",
4883 " --sha3-384 Use the sha3-384 algorithm",
4884 " --sha3-512 Use the sha3-512 algorithm",
4885 " Any other argument is a LIKE pattern for tables to hash",
4886 #if !defined(SQLITE_NOHAVE_SYSTEM) && !defined(SQLITE_SHELL_FIDDLE)
4887 ".shell CMD ARGS... Run CMD ARGS... in a system shell",
4888 #endif
4889 ".show Show the current values for various settings",
4890 ".stats ?ARG? Show stats or turn stats on or off",
4891 " off Turn off automatic stat display",
4892 " on Turn on automatic stat display",
4893 " stmt Show statement stats",
4894 " vmstep Show the virtual machine step count only",
4895 #if !defined(SQLITE_NOHAVE_SYSTEM) && !defined(SQLITE_SHELL_FIDDLE)
4896 ".system CMD ARGS... Run CMD ARGS... in a system shell",
4897 #endif
4898 ".tables ?TABLE? List names of tables matching LIKE pattern TABLE",
4899 #ifndef SQLITE_SHELL_FIDDLE
4900 ",testcase NAME Begin redirecting output to 'testcase-out.txt'",
4901 #endif
4902 ",testctrl CMD ... Run various sqlite3_test_control() operations",
4903 " Run \".testctrl\" with no arguments for details",
4904 ".timeout MS Try opening locked tables for MS milliseconds",
4905 ".timer on|off Turn SQL timer on or off",
4906 #ifndef SQLITE_OMIT_TRACE
4907 ".trace ?OPTIONS? Output each SQL statement as it is run",
4908 " FILE Send output to FILE",
4909 " stdout Send output to stdout",
4910 " stderr Send output to stderr",
4911 " off Disable tracing",
4912 " --expanded Expand query parameters",
4913 #ifdef SQLITE_ENABLE_NORMALIZE
4914 " --normalized Normal the SQL statements",
4915 #endif
4916 " --plain Show SQL as it is input",
4917 " --stmt Trace statement execution (SQLITE_TRACE_STMT)",
4918 " --profile Profile statements (SQLITE_TRACE_PROFILE)",
4919 " --row Trace each row (SQLITE_TRACE_ROW)",
4920 " --close Trace connection close (SQLITE_TRACE_CLOSE)",
4921 #endif /* SQLITE_OMIT_TRACE */
4922 #ifdef SQLITE_DEBUG
4923 ".unmodule NAME ... Unregister virtual table modules",
4924 " --allexcept Unregister everything except those named",
4925 #endif
4926 ".version Show source, library and compiler versions",
4927 ".vfsinfo ?AUX? Information about the top-level VFS",
4928 ".vfslist List all available VFSes",
4929 ".vfsname ?AUX? Print the name of the VFS stack",
4930 ".width NUM1 NUM2 ... Set minimum column widths for columnar output",
4931 " Negative values right-justify",
4935 ** Output help text.
4937 ** zPattern describes the set of commands for which help text is provided.
4938 ** If zPattern is NULL, then show all commands, but only give a one-line
4939 ** description of each.
4941 ** Return the number of matches.
4943 static int showHelp(FILE *out, const char *zPattern){
4944 int i = 0;
4945 int j = 0;
4946 int n = 0;
4947 char *zPat;
4948 if( zPattern==0
4949 || zPattern[0]=='0'
4950 || cli_strcmp(zPattern,"-a")==0
4951 || cli_strcmp(zPattern,"-all")==0
4952 || cli_strcmp(zPattern,"--all")==0
4954 enum HelpWanted { HW_NoCull = 0, HW_SummaryOnly = 1, HW_Undoc = 2 };
4955 enum HelpHave { HH_Undoc = 2, HH_Summary = 1, HH_More = 0 };
4956 /* Show all or most commands
4957 ** *zPattern==0 => summary of documented commands only
4958 ** *zPattern=='0' => whole help for undocumented commands
4959 ** Otherwise => whole help for documented commands
4961 enum HelpWanted hw = HW_SummaryOnly;
4962 enum HelpHave hh = HH_More;
4963 if( zPattern!=0 ){
4964 hw = (*zPattern=='0')? HW_NoCull|HW_Undoc : HW_NoCull;
4966 for(i=0; i<ArraySize(azHelp); i++){
4967 switch( azHelp[i][0] ){
4968 case ',':
4969 hh = HH_Summary|HH_Undoc;
4970 break;
4971 case '.':
4972 hh = HH_Summary;
4973 break;
4974 default:
4975 hh &= ~HH_Summary;
4976 break;
4978 if( ((hw^hh)&HH_Undoc)==0 ){
4979 if( (hh&HH_Summary)!=0 ){
4980 sputf(out, ".%s\n", azHelp[i]+1);
4981 ++n;
4982 }else if( (hw&HW_SummaryOnly)==0 ){
4983 sputf(out, "%s\n", azHelp[i]);
4987 }else{
4988 /* Seek documented commands for which zPattern is an exact prefix */
4989 zPat = sqlite3_mprintf(".%s*", zPattern);
4990 shell_check_oom(zPat);
4991 for(i=0; i<ArraySize(azHelp); i++){
4992 if( sqlite3_strglob(zPat, azHelp[i])==0 ){
4993 sputf(out, "%s\n", azHelp[i]);
4994 j = i+1;
4995 n++;
4998 sqlite3_free(zPat);
4999 if( n ){
5000 if( n==1 ){
5001 /* when zPattern is a prefix of exactly one command, then include
5002 ** the details of that command, which should begin at offset j */
5003 while( j<ArraySize(azHelp)-1 && azHelp[j][0]==' ' ){
5004 sputf(out, "%s\n", azHelp[j]);
5005 j++;
5008 return n;
5010 /* Look for documented commands that contain zPattern anywhere.
5011 ** Show complete text of all documented commands that match. */
5012 zPat = sqlite3_mprintf("%%%s%%", zPattern);
5013 shell_check_oom(zPat);
5014 for(i=0; i<ArraySize(azHelp); i++){
5015 if( azHelp[i][0]==',' ){
5016 while( i<ArraySize(azHelp)-1 && azHelp[i+1][0]==' ' ) ++i;
5017 continue;
5019 if( azHelp[i][0]=='.' ) j = i;
5020 if( sqlite3_strlike(zPat, azHelp[i], 0)==0 ){
5021 sputf(out, "%s\n", azHelp[j]);
5022 while( j<ArraySize(azHelp)-1 && azHelp[j+1][0]==' ' ){
5023 j++;
5024 sputf(out, "%s\n", azHelp[j]);
5026 i = j;
5027 n++;
5030 sqlite3_free(zPat);
5032 return n;
5035 /* Forward reference */
5036 static int process_input(ShellState *p);
5039 ** Read the content of file zName into memory obtained from sqlite3_malloc64()
5040 ** and return a pointer to the buffer. The caller is responsible for freeing
5041 ** the memory.
5043 ** If parameter pnByte is not NULL, (*pnByte) is set to the number of bytes
5044 ** read.
5046 ** For convenience, a nul-terminator byte is always appended to the data read
5047 ** from the file before the buffer is returned. This byte is not included in
5048 ** the final value of (*pnByte), if applicable.
5050 ** NULL is returned if any error is encountered. The final value of *pnByte
5051 ** is undefined in this case.
5053 static char *readFile(const char *zName, int *pnByte){
5054 FILE *in = fopen(zName, "rb");
5055 long nIn;
5056 size_t nRead;
5057 char *pBuf;
5058 int rc;
5059 if( in==0 ) return 0;
5060 rc = fseek(in, 0, SEEK_END);
5061 if( rc!=0 ){
5062 eputf("Error: '%s' not seekable\n", zName);
5063 fclose(in);
5064 return 0;
5066 nIn = ftell(in);
5067 rewind(in);
5068 pBuf = sqlite3_malloc64( nIn+1 );
5069 if( pBuf==0 ){
5070 eputz("Error: out of memory\n");
5071 fclose(in);
5072 return 0;
5074 nRead = fread(pBuf, nIn, 1, in);
5075 fclose(in);
5076 if( nRead!=1 ){
5077 sqlite3_free(pBuf);
5078 eputf("Error: cannot read '%s'\n", zName);
5079 return 0;
5081 pBuf[nIn] = 0;
5082 if( pnByte ) *pnByte = nIn;
5083 return pBuf;
5086 #if defined(SQLITE_ENABLE_SESSION)
5088 ** Close a single OpenSession object and release all of its associated
5089 ** resources.
5091 static void session_close(OpenSession *pSession){
5092 int i;
5093 sqlite3session_delete(pSession->p);
5094 sqlite3_free(pSession->zName);
5095 for(i=0; i<pSession->nFilter; i++){
5096 sqlite3_free(pSession->azFilter[i]);
5098 sqlite3_free(pSession->azFilter);
5099 memset(pSession, 0, sizeof(OpenSession));
5101 #endif
5104 ** Close all OpenSession objects and release all associated resources.
5106 #if defined(SQLITE_ENABLE_SESSION)
5107 static void session_close_all(ShellState *p, int i){
5108 int j;
5109 struct AuxDb *pAuxDb = i<0 ? p->pAuxDb : &p->aAuxDb[i];
5110 for(j=0; j<pAuxDb->nSession; j++){
5111 session_close(&pAuxDb->aSession[j]);
5113 pAuxDb->nSession = 0;
5115 #else
5116 # define session_close_all(X,Y)
5117 #endif
5120 ** Implementation of the xFilter function for an open session. Omit
5121 ** any tables named by ".session filter" but let all other table through.
5123 #if defined(SQLITE_ENABLE_SESSION)
5124 static int session_filter(void *pCtx, const char *zTab){
5125 OpenSession *pSession = (OpenSession*)pCtx;
5126 int i;
5127 for(i=0; i<pSession->nFilter; i++){
5128 if( sqlite3_strglob(pSession->azFilter[i], zTab)==0 ) return 0;
5130 return 1;
5132 #endif
5135 ** Try to deduce the type of file for zName based on its content. Return
5136 ** one of the SHELL_OPEN_* constants.
5138 ** If the file does not exist or is empty but its name looks like a ZIP
5139 ** archive and the dfltZip flag is true, then assume it is a ZIP archive.
5140 ** Otherwise, assume an ordinary database regardless of the filename if
5141 ** the type cannot be determined from content.
5143 int deduceDatabaseType(const char *zName, int dfltZip){
5144 FILE *f = fopen(zName, "rb");
5145 size_t n;
5146 int rc = SHELL_OPEN_UNSPEC;
5147 char zBuf[100];
5148 if( f==0 ){
5149 if( dfltZip && sqlite3_strlike("%.zip",zName,0)==0 ){
5150 return SHELL_OPEN_ZIPFILE;
5151 }else{
5152 return SHELL_OPEN_NORMAL;
5155 n = fread(zBuf, 16, 1, f);
5156 if( n==1 && memcmp(zBuf, "SQLite format 3", 16)==0 ){
5157 fclose(f);
5158 return SHELL_OPEN_NORMAL;
5160 fseek(f, -25, SEEK_END);
5161 n = fread(zBuf, 25, 1, f);
5162 if( n==1 && memcmp(zBuf, "Start-Of-SQLite3-", 17)==0 ){
5163 rc = SHELL_OPEN_APPENDVFS;
5164 }else{
5165 fseek(f, -22, SEEK_END);
5166 n = fread(zBuf, 22, 1, f);
5167 if( n==1 && zBuf[0]==0x50 && zBuf[1]==0x4b && zBuf[2]==0x05
5168 && zBuf[3]==0x06 ){
5169 rc = SHELL_OPEN_ZIPFILE;
5170 }else if( n==0 && dfltZip && sqlite3_strlike("%.zip",zName,0)==0 ){
5171 rc = SHELL_OPEN_ZIPFILE;
5174 fclose(f);
5175 return rc;
5178 #ifndef SQLITE_OMIT_DESERIALIZE
5180 ** Reconstruct an in-memory database using the output from the "dbtotxt"
5181 ** program. Read content from the file in p->aAuxDb[].zDbFilename.
5182 ** If p->aAuxDb[].zDbFilename is 0, then read from standard input.
5184 static unsigned char *readHexDb(ShellState *p, int *pnData){
5185 unsigned char *a = 0;
5186 int nLine;
5187 int n = 0;
5188 int pgsz = 0;
5189 int iOffset = 0;
5190 int j, k;
5191 int rc;
5192 FILE *in;
5193 const char *zDbFilename = p->pAuxDb->zDbFilename;
5194 unsigned int x[16];
5195 char zLine[1000];
5196 if( zDbFilename ){
5197 in = fopen(zDbFilename, "r");
5198 if( in==0 ){
5199 eputf("cannot open \"%s\" for reading\n", zDbFilename);
5200 return 0;
5202 nLine = 0;
5203 }else{
5204 in = p->in;
5205 nLine = p->lineno;
5206 if( in==0 ) in = stdin;
5208 *pnData = 0;
5209 nLine++;
5210 if( fgets(zLine, sizeof(zLine), in)==0 ) goto readHexDb_error;
5211 rc = sscanf(zLine, "| size %d pagesize %d", &n, &pgsz);
5212 if( rc!=2 ) goto readHexDb_error;
5213 if( n<0 ) goto readHexDb_error;
5214 if( pgsz<512 || pgsz>65536 || (pgsz&(pgsz-1))!=0 ) goto readHexDb_error;
5215 n = (n+pgsz-1)&~(pgsz-1); /* Round n up to the next multiple of pgsz */
5216 a = sqlite3_malloc( n ? n : 1 );
5217 shell_check_oom(a);
5218 memset(a, 0, n);
5219 if( pgsz<512 || pgsz>65536 || (pgsz & (pgsz-1))!=0 ){
5220 eputz("invalid pagesize\n");
5221 goto readHexDb_error;
5223 for(nLine++; fgets(zLine, sizeof(zLine), in)!=0; nLine++){
5224 rc = sscanf(zLine, "| page %d offset %d", &j, &k);
5225 if( rc==2 ){
5226 iOffset = k;
5227 continue;
5229 if( cli_strncmp(zLine, "| end ", 6)==0 ){
5230 break;
5232 rc = sscanf(zLine,"| %d: %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x",
5233 &j, &x[0], &x[1], &x[2], &x[3], &x[4], &x[5], &x[6], &x[7],
5234 &x[8], &x[9], &x[10], &x[11], &x[12], &x[13], &x[14], &x[15]);
5235 if( rc==17 ){
5236 k = iOffset+j;
5237 if( k+16<=n && k>=0 ){
5238 int ii;
5239 for(ii=0; ii<16; ii++) a[k+ii] = x[ii]&0xff;
5243 *pnData = n;
5244 if( in!=p->in ){
5245 fclose(in);
5246 }else{
5247 p->lineno = nLine;
5249 return a;
5251 readHexDb_error:
5252 if( in!=p->in ){
5253 fclose(in);
5254 }else{
5255 while( fgets(zLine, sizeof(zLine), p->in)!=0 ){
5256 nLine++;
5257 if(cli_strncmp(zLine, "| end ", 6)==0 ) break;
5259 p->lineno = nLine;
5261 sqlite3_free(a);
5262 eputf("Error on line %d of --hexdb input\n", nLine);
5263 return 0;
5265 #endif /* SQLITE_OMIT_DESERIALIZE */
5268 ** Scalar function "usleep(X)" invokes sqlite3_sleep(X) and returns X.
5270 static void shellUSleepFunc(
5271 sqlite3_context *context,
5272 int argcUnused,
5273 sqlite3_value **argv
5275 int sleep = sqlite3_value_int(argv[0]);
5276 (void)argcUnused;
5277 sqlite3_sleep(sleep/1000);
5278 sqlite3_result_int(context, sleep);
5281 /* Flags for open_db().
5283 ** The default behavior of open_db() is to exit(1) if the database fails to
5284 ** open. The OPEN_DB_KEEPALIVE flag changes that so that it prints an error
5285 ** but still returns without calling exit.
5287 ** The OPEN_DB_ZIPFILE flag causes open_db() to prefer to open files as a
5288 ** ZIP archive if the file does not exist or is empty and its name matches
5289 ** the *.zip pattern.
5291 #define OPEN_DB_KEEPALIVE 0x001 /* Return after error if true */
5292 #define OPEN_DB_ZIPFILE 0x002 /* Open as ZIP if name matches *.zip */
5295 ** Make sure the database is open. If it is not, then open it. If
5296 ** the database fails to open, print an error message and exit.
5298 static void open_db(ShellState *p, int openFlags){
5299 if( p->db==0 ){
5300 const char *zDbFilename = p->pAuxDb->zDbFilename;
5301 if( p->openMode==SHELL_OPEN_UNSPEC ){
5302 if( zDbFilename==0 || zDbFilename[0]==0 ){
5303 p->openMode = SHELL_OPEN_NORMAL;
5304 }else{
5305 p->openMode = (u8)deduceDatabaseType(zDbFilename,
5306 (openFlags & OPEN_DB_ZIPFILE)!=0);
5309 switch( p->openMode ){
5310 case SHELL_OPEN_APPENDVFS: {
5311 sqlite3_open_v2(zDbFilename, &p->db,
5312 SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE|p->openFlags, "apndvfs");
5313 break;
5315 case SHELL_OPEN_HEXDB:
5316 case SHELL_OPEN_DESERIALIZE: {
5317 sqlite3_open(0, &p->db);
5318 break;
5320 case SHELL_OPEN_ZIPFILE: {
5321 sqlite3_open(":memory:", &p->db);
5322 break;
5324 case SHELL_OPEN_READONLY: {
5325 sqlite3_open_v2(zDbFilename, &p->db,
5326 SQLITE_OPEN_READONLY|p->openFlags, 0);
5327 break;
5329 case SHELL_OPEN_UNSPEC:
5330 case SHELL_OPEN_NORMAL: {
5331 sqlite3_open_v2(zDbFilename, &p->db,
5332 SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE|p->openFlags, 0);
5333 break;
5336 if( p->db==0 || SQLITE_OK!=sqlite3_errcode(p->db) ){
5337 eputf("Error: unable to open database \"%s\": %s\n",
5338 zDbFilename, sqlite3_errmsg(p->db));
5339 if( (openFlags & OPEN_DB_KEEPALIVE)==0 ){
5340 exit(1);
5342 sqlite3_close(p->db);
5343 sqlite3_open(":memory:", &p->db);
5344 if( p->db==0 || SQLITE_OK!=sqlite3_errcode(p->db) ){
5345 eputz("Also: unable to open substitute in-memory database.\n");
5346 exit(1);
5347 }else{
5348 eputf("Notice: using substitute in-memory database instead of \"%s\"\n",
5349 zDbFilename);
5352 globalDb = p->db;
5353 sqlite3_db_config(p->db, SQLITE_DBCONFIG_STMT_SCANSTATUS, (int)0, (int*)0);
5355 /* Reflect the use or absence of --unsafe-testing invocation. */
5357 int testmode_on = ShellHasFlag(p,SHFLG_TestingMode);
5358 sqlite3_db_config(p->db, SQLITE_DBCONFIG_TRUSTED_SCHEMA, testmode_on,0);
5359 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, !testmode_on,0);
5362 #ifndef SQLITE_OMIT_LOAD_EXTENSION
5363 sqlite3_enable_load_extension(p->db, 1);
5364 #endif
5365 sqlite3_shathree_init(p->db, 0, 0);
5366 sqlite3_uint_init(p->db, 0, 0);
5367 sqlite3_decimal_init(p->db, 0, 0);
5368 sqlite3_base64_init(p->db, 0, 0);
5369 sqlite3_base85_init(p->db, 0, 0);
5370 sqlite3_regexp_init(p->db, 0, 0);
5371 sqlite3_ieee_init(p->db, 0, 0);
5372 sqlite3_series_init(p->db, 0, 0);
5373 #ifndef SQLITE_SHELL_FIDDLE
5374 sqlite3_fileio_init(p->db, 0, 0);
5375 sqlite3_completion_init(p->db, 0, 0);
5376 #endif
5377 #ifdef SQLITE_HAVE_ZLIB
5378 if( !p->bSafeModePersist ){
5379 sqlite3_zipfile_init(p->db, 0, 0);
5380 sqlite3_sqlar_init(p->db, 0, 0);
5382 #endif
5383 #ifdef SQLITE_SHELL_EXTFUNCS
5384 /* Create a preprocessing mechanism for extensions to make
5385 * their own provisions for being built into the shell.
5386 * This is a short-span macro. See further below for usage.
5388 #define SHELL_SUB_MACRO(base, variant) base ## _ ## variant
5389 #define SHELL_SUBMACRO(base, variant) SHELL_SUB_MACRO(base, variant)
5390 /* Let custom-included extensions get their ..._init() called.
5391 * The WHATEVER_INIT( db, pzErrorMsg, pApi ) macro should cause
5392 * the extension's sqlite3_*_init( db, pzErrorMsg, pApi )
5393 * initialization routine to be called.
5396 int irc = SHELL_SUBMACRO(SQLITE_SHELL_EXTFUNCS, INIT)(p->db);
5397 /* Let custom-included extensions expose their functionality.
5398 * The WHATEVER_EXPOSE( db, pzErrorMsg ) macro should cause
5399 * the SQL functions, virtual tables, collating sequences or
5400 * VFS's implemented by the extension to be registered.
5402 if( irc==SQLITE_OK
5403 || irc==SQLITE_OK_LOAD_PERMANENTLY ){
5404 SHELL_SUBMACRO(SQLITE_SHELL_EXTFUNCS, EXPOSE)(p->db, 0);
5406 #undef SHELL_SUB_MACRO
5407 #undef SHELL_SUBMACRO
5409 #endif
5411 sqlite3_create_function(p->db, "strtod", 1, SQLITE_UTF8, 0,
5412 shellStrtod, 0, 0);
5413 sqlite3_create_function(p->db, "dtostr", 1, SQLITE_UTF8, 0,
5414 shellDtostr, 0, 0);
5415 sqlite3_create_function(p->db, "dtostr", 2, SQLITE_UTF8, 0,
5416 shellDtostr, 0, 0);
5417 sqlite3_create_function(p->db, "shell_add_schema", 3, SQLITE_UTF8, 0,
5418 shellAddSchemaName, 0, 0);
5419 sqlite3_create_function(p->db, "shell_module_schema", 1, SQLITE_UTF8, 0,
5420 shellModuleSchema, 0, 0);
5421 sqlite3_create_function(p->db, "shell_putsnl", 1, SQLITE_UTF8, p,
5422 shellPutsFunc, 0, 0);
5423 sqlite3_create_function(p->db, "usleep",1,SQLITE_UTF8,0,
5424 shellUSleepFunc, 0, 0);
5425 #ifndef SQLITE_NOHAVE_SYSTEM
5426 sqlite3_create_function(p->db, "edit", 1, SQLITE_UTF8, 0,
5427 editFunc, 0, 0);
5428 sqlite3_create_function(p->db, "edit", 2, SQLITE_UTF8, 0,
5429 editFunc, 0, 0);
5430 #endif
5432 if( p->openMode==SHELL_OPEN_ZIPFILE ){
5433 char *zSql = sqlite3_mprintf(
5434 "CREATE VIRTUAL TABLE zip USING zipfile(%Q);", zDbFilename);
5435 shell_check_oom(zSql);
5436 sqlite3_exec(p->db, zSql, 0, 0, 0);
5437 sqlite3_free(zSql);
5439 #ifndef SQLITE_OMIT_DESERIALIZE
5440 else
5441 if( p->openMode==SHELL_OPEN_DESERIALIZE || p->openMode==SHELL_OPEN_HEXDB ){
5442 int rc;
5443 int nData = 0;
5444 unsigned char *aData;
5445 if( p->openMode==SHELL_OPEN_DESERIALIZE ){
5446 aData = (unsigned char*)readFile(zDbFilename, &nData);
5447 }else{
5448 aData = readHexDb(p, &nData);
5450 if( aData==0 ){
5451 return;
5453 rc = sqlite3_deserialize(p->db, "main", aData, nData, nData,
5454 SQLITE_DESERIALIZE_RESIZEABLE |
5455 SQLITE_DESERIALIZE_FREEONCLOSE);
5456 if( rc ){
5457 eputf("Error: sqlite3_deserialize() returns %d\n", rc);
5459 if( p->szMax>0 ){
5460 sqlite3_file_control(p->db, "main", SQLITE_FCNTL_SIZE_LIMIT, &p->szMax);
5463 #endif
5465 if( p->db!=0 ){
5466 if( p->bSafeModePersist ){
5467 sqlite3_set_authorizer(p->db, safeModeAuth, p);
5469 sqlite3_db_config(
5470 p->db, SQLITE_DBCONFIG_STMT_SCANSTATUS, p->scanstatsOn, (int*)0
5476 ** Attempt to close the database connection. Report errors.
5478 void close_db(sqlite3 *db){
5479 int rc = sqlite3_close(db);
5480 if( rc ){
5481 eputf("Error: sqlite3_close() returns %d: %s\n", rc, sqlite3_errmsg(db));
5485 #if HAVE_READLINE || HAVE_EDITLINE
5487 ** Readline completion callbacks
5489 static char *readline_completion_generator(const char *text, int state){
5490 static sqlite3_stmt *pStmt = 0;
5491 char *zRet;
5492 if( state==0 ){
5493 char *zSql;
5494 sqlite3_finalize(pStmt);
5495 zSql = sqlite3_mprintf("SELECT DISTINCT candidate COLLATE nocase"
5496 " FROM completion(%Q) ORDER BY 1", text);
5497 shell_check_oom(zSql);
5498 sqlite3_prepare_v2(globalDb, zSql, -1, &pStmt, 0);
5499 sqlite3_free(zSql);
5501 if( sqlite3_step(pStmt)==SQLITE_ROW ){
5502 const char *z = (const char*)sqlite3_column_text(pStmt,0);
5503 zRet = z ? strdup(z) : 0;
5504 }else{
5505 sqlite3_finalize(pStmt);
5506 pStmt = 0;
5507 zRet = 0;
5509 return zRet;
5511 static char **readline_completion(const char *zText, int iStart, int iEnd){
5512 (void)iStart;
5513 (void)iEnd;
5514 rl_attempted_completion_over = 1;
5515 return rl_completion_matches(zText, readline_completion_generator);
5518 #elif HAVE_LINENOISE
5520 ** Linenoise completion callback
5522 static void linenoise_completion(const char *zLine, linenoiseCompletions *lc){
5523 i64 nLine = strlen(zLine);
5524 i64 i, iStart;
5525 sqlite3_stmt *pStmt = 0;
5526 char *zSql;
5527 char zBuf[1000];
5529 if( nLine>(i64)sizeof(zBuf)-30 ) return;
5530 if( zLine[0]=='.' || zLine[0]=='#') return;
5531 for(i=nLine-1; i>=0 && (isalnum(zLine[i]) || zLine[i]=='_'); i--){}
5532 if( i==nLine-1 ) return;
5533 iStart = i+1;
5534 memcpy(zBuf, zLine, iStart);
5535 zSql = sqlite3_mprintf("SELECT DISTINCT candidate COLLATE nocase"
5536 " FROM completion(%Q,%Q) ORDER BY 1",
5537 &zLine[iStart], zLine);
5538 shell_check_oom(zSql);
5539 sqlite3_prepare_v2(globalDb, zSql, -1, &pStmt, 0);
5540 sqlite3_free(zSql);
5541 sqlite3_exec(globalDb, "PRAGMA page_count", 0, 0, 0); /* Load the schema */
5542 while( sqlite3_step(pStmt)==SQLITE_ROW ){
5543 const char *zCompletion = (const char*)sqlite3_column_text(pStmt, 0);
5544 int nCompletion = sqlite3_column_bytes(pStmt, 0);
5545 if( iStart+nCompletion < (i64)sizeof(zBuf)-1 && zCompletion ){
5546 memcpy(zBuf+iStart, zCompletion, nCompletion+1);
5547 linenoiseAddCompletion(lc, zBuf);
5550 sqlite3_finalize(pStmt);
5552 #endif
5555 ** Do C-language style dequoting.
5557 ** \a -> alarm
5558 ** \b -> backspace
5559 ** \t -> tab
5560 ** \n -> newline
5561 ** \v -> vertical tab
5562 ** \f -> form feed
5563 ** \r -> carriage return
5564 ** \s -> space
5565 ** \" -> "
5566 ** \' -> '
5567 ** \\ -> backslash
5568 ** \NNN -> ascii character NNN in octal
5569 ** \xHH -> ascii character HH in hexadecimal
5571 static void resolve_backslashes(char *z){
5572 int i, j;
5573 char c;
5574 while( *z && *z!='\\' ) z++;
5575 for(i=j=0; (c = z[i])!=0; i++, j++){
5576 if( c=='\\' && z[i+1]!=0 ){
5577 c = z[++i];
5578 if( c=='a' ){
5579 c = '\a';
5580 }else if( c=='b' ){
5581 c = '\b';
5582 }else if( c=='t' ){
5583 c = '\t';
5584 }else if( c=='n' ){
5585 c = '\n';
5586 }else if( c=='v' ){
5587 c = '\v';
5588 }else if( c=='f' ){
5589 c = '\f';
5590 }else if( c=='r' ){
5591 c = '\r';
5592 }else if( c=='"' ){
5593 c = '"';
5594 }else if( c=='\'' ){
5595 c = '\'';
5596 }else if( c=='\\' ){
5597 c = '\\';
5598 }else if( c=='x' ){
5599 int nhd = 0, hdv;
5600 u8 hv = 0;
5601 while( nhd<2 && (c=z[i+1+nhd])!=0 && (hdv=hexDigitValue(c))>=0 ){
5602 hv = (u8)((hv<<4)|hdv);
5603 ++nhd;
5605 i += nhd;
5606 c = (u8)hv;
5607 }else if( c>='0' && c<='7' ){
5608 c -= '0';
5609 if( z[i+1]>='0' && z[i+1]<='7' ){
5610 i++;
5611 c = (c<<3) + z[i] - '0';
5612 if( z[i+1]>='0' && z[i+1]<='7' ){
5613 i++;
5614 c = (c<<3) + z[i] - '0';
5619 z[j] = c;
5621 if( j<i ) z[j] = 0;
5625 ** Interpret zArg as either an integer or a boolean value. Return 1 or 0
5626 ** for TRUE and FALSE. Return the integer value if appropriate.
5628 static int booleanValue(const char *zArg){
5629 int i;
5630 if( zArg[0]=='0' && zArg[1]=='x' ){
5631 for(i=2; hexDigitValue(zArg[i])>=0; i++){}
5632 }else{
5633 for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){}
5635 if( i>0 && zArg[i]==0 ) return (int)(integerValue(zArg) & 0xffffffff);
5636 if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){
5637 return 1;
5639 if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){
5640 return 0;
5642 eputf("ERROR: Not a boolean value: \"%s\". Assuming \"no\".\n", zArg);
5643 return 0;
5647 ** Set or clear a shell flag according to a boolean value.
5649 static void setOrClearFlag(ShellState *p, unsigned mFlag, const char *zArg){
5650 if( booleanValue(zArg) ){
5651 ShellSetFlag(p, mFlag);
5652 }else{
5653 ShellClearFlag(p, mFlag);
5658 ** Close an output file, assuming it is not stderr or stdout
5660 static void output_file_close(FILE *f){
5661 if( f && f!=stdout && f!=stderr ) fclose(f);
5665 ** Try to open an output file. The names "stdout" and "stderr" are
5666 ** recognized and do the right thing. NULL is returned if the output
5667 ** filename is "off".
5669 static FILE *output_file_open(const char *zFile, int bTextMode){
5670 FILE *f;
5671 if( cli_strcmp(zFile,"stdout")==0 ){
5672 f = stdout;
5673 }else if( cli_strcmp(zFile, "stderr")==0 ){
5674 f = stderr;
5675 }else if( cli_strcmp(zFile, "off")==0 ){
5676 f = 0;
5677 }else{
5678 f = fopen(zFile, bTextMode ? "w" : "wb");
5679 if( f==0 ){
5680 eputf("Error: cannot open \"%s\"\n", zFile);
5683 return f;
5686 #ifndef SQLITE_OMIT_TRACE
5688 ** A routine for handling output from sqlite3_trace().
5690 static int sql_trace_callback(
5691 unsigned mType, /* The trace type */
5692 void *pArg, /* The ShellState pointer */
5693 void *pP, /* Usually a pointer to sqlite_stmt */
5694 void *pX /* Auxiliary output */
5696 ShellState *p = (ShellState*)pArg;
5697 sqlite3_stmt *pStmt;
5698 const char *zSql;
5699 i64 nSql;
5700 if( p->traceOut==0 ) return 0;
5701 if( mType==SQLITE_TRACE_CLOSE ){
5702 sputz(p->traceOut, "-- closing database connection\n");
5703 return 0;
5705 if( mType!=SQLITE_TRACE_ROW && pX!=0 && ((const char*)pX)[0]=='-' ){
5706 zSql = (const char*)pX;
5707 }else{
5708 pStmt = (sqlite3_stmt*)pP;
5709 switch( p->eTraceType ){
5710 case SHELL_TRACE_EXPANDED: {
5711 zSql = sqlite3_expanded_sql(pStmt);
5712 break;
5714 #ifdef SQLITE_ENABLE_NORMALIZE
5715 case SHELL_TRACE_NORMALIZED: {
5716 zSql = sqlite3_normalized_sql(pStmt);
5717 break;
5719 #endif
5720 default: {
5721 zSql = sqlite3_sql(pStmt);
5722 break;
5726 if( zSql==0 ) return 0;
5727 nSql = strlen(zSql);
5728 if( nSql>1000000000 ) nSql = 1000000000;
5729 while( nSql>0 && zSql[nSql-1]==';' ){ nSql--; }
5730 switch( mType ){
5731 case SQLITE_TRACE_ROW:
5732 case SQLITE_TRACE_STMT: {
5733 sputf(p->traceOut, "%.*s;\n", (int)nSql, zSql);
5734 break;
5736 case SQLITE_TRACE_PROFILE: {
5737 sqlite3_int64 nNanosec = pX ? *(sqlite3_int64*)pX : 0;
5738 sputf(p->traceOut, "%.*s; -- %lld ns\n", (int)nSql, zSql, nNanosec);
5739 break;
5742 return 0;
5744 #endif
5747 ** A no-op routine that runs with the ".breakpoint" doc-command. This is
5748 ** a useful spot to set a debugger breakpoint.
5750 ** This routine does not do anything practical. The code are there simply
5751 ** to prevent the compiler from optimizing this routine out.
5753 static void test_breakpoint(void){
5754 static unsigned int nCall = 0;
5755 if( (nCall++)==0xffffffff ) printf("Many .breakpoints have run\n");
5759 ** An object used to read a CSV and other files for import.
5761 typedef struct ImportCtx ImportCtx;
5762 struct ImportCtx {
5763 const char *zFile; /* Name of the input file */
5764 FILE *in; /* Read the CSV text from this input stream */
5765 int (SQLITE_CDECL *xCloser)(FILE*); /* Func to close in */
5766 char *z; /* Accumulated text for a field */
5767 int n; /* Number of bytes in z */
5768 int nAlloc; /* Space allocated for z[] */
5769 int nLine; /* Current line number */
5770 int nRow; /* Number of rows imported */
5771 int nErr; /* Number of errors encountered */
5772 int bNotFirst; /* True if one or more bytes already read */
5773 int cTerm; /* Character that terminated the most recent field */
5774 int cColSep; /* The column separator character. (Usually ",") */
5775 int cRowSep; /* The row separator character. (Usually "\n") */
5778 /* Clean up resourced used by an ImportCtx */
5779 static void import_cleanup(ImportCtx *p){
5780 if( p->in!=0 && p->xCloser!=0 ){
5781 p->xCloser(p->in);
5782 p->in = 0;
5784 sqlite3_free(p->z);
5785 p->z = 0;
5788 /* Append a single byte to z[] */
5789 static void import_append_char(ImportCtx *p, int c){
5790 if( p->n+1>=p->nAlloc ){
5791 p->nAlloc += p->nAlloc + 100;
5792 p->z = sqlite3_realloc64(p->z, p->nAlloc);
5793 shell_check_oom(p->z);
5795 p->z[p->n++] = (char)c;
5798 /* Read a single field of CSV text. Compatible with rfc4180 and extended
5799 ** with the option of having a separator other than ",".
5801 ** + Input comes from p->in.
5802 ** + Store results in p->z of length p->n. Space to hold p->z comes
5803 ** from sqlite3_malloc64().
5804 ** + Use p->cSep as the column separator. The default is ",".
5805 ** + Use p->rSep as the row separator. The default is "\n".
5806 ** + Keep track of the line number in p->nLine.
5807 ** + Store the character that terminates the field in p->cTerm. Store
5808 ** EOF on end-of-file.
5809 ** + Report syntax errors on stderr
5811 static char *SQLITE_CDECL csv_read_one_field(ImportCtx *p){
5812 int c;
5813 int cSep = (u8)p->cColSep;
5814 int rSep = (u8)p->cRowSep;
5815 p->n = 0;
5816 c = fgetc(p->in);
5817 if( c==EOF || seenInterrupt ){
5818 p->cTerm = EOF;
5819 return 0;
5821 if( c=='"' ){
5822 int pc, ppc;
5823 int startLine = p->nLine;
5824 int cQuote = c;
5825 pc = ppc = 0;
5826 while( 1 ){
5827 c = fgetc(p->in);
5828 if( c==rSep ) p->nLine++;
5829 if( c==cQuote ){
5830 if( pc==cQuote ){
5831 pc = 0;
5832 continue;
5835 if( (c==cSep && pc==cQuote)
5836 || (c==rSep && pc==cQuote)
5837 || (c==rSep && pc=='\r' && ppc==cQuote)
5838 || (c==EOF && pc==cQuote)
5840 do{ p->n--; }while( p->z[p->n]!=cQuote );
5841 p->cTerm = c;
5842 break;
5844 if( pc==cQuote && c!='\r' ){
5845 eputf("%s:%d: unescaped %c character\n", p->zFile, p->nLine, cQuote);
5847 if( c==EOF ){
5848 eputf("%s:%d: unterminated %c-quoted field\n",
5849 p->zFile, startLine, cQuote);
5850 p->cTerm = c;
5851 break;
5853 import_append_char(p, c);
5854 ppc = pc;
5855 pc = c;
5857 }else{
5858 /* If this is the first field being parsed and it begins with the
5859 ** UTF-8 BOM (0xEF BB BF) then skip the BOM */
5860 if( (c&0xff)==0xef && p->bNotFirst==0 ){
5861 import_append_char(p, c);
5862 c = fgetc(p->in);
5863 if( (c&0xff)==0xbb ){
5864 import_append_char(p, c);
5865 c = fgetc(p->in);
5866 if( (c&0xff)==0xbf ){
5867 p->bNotFirst = 1;
5868 p->n = 0;
5869 return csv_read_one_field(p);
5873 while( c!=EOF && c!=cSep && c!=rSep ){
5874 import_append_char(p, c);
5875 c = fgetc(p->in);
5877 if( c==rSep ){
5878 p->nLine++;
5879 if( p->n>0 && p->z[p->n-1]=='\r' ) p->n--;
5881 p->cTerm = c;
5883 if( p->z ) p->z[p->n] = 0;
5884 p->bNotFirst = 1;
5885 return p->z;
5888 /* Read a single field of ASCII delimited text.
5890 ** + Input comes from p->in.
5891 ** + Store results in p->z of length p->n. Space to hold p->z comes
5892 ** from sqlite3_malloc64().
5893 ** + Use p->cSep as the column separator. The default is "\x1F".
5894 ** + Use p->rSep as the row separator. The default is "\x1E".
5895 ** + Keep track of the row number in p->nLine.
5896 ** + Store the character that terminates the field in p->cTerm. Store
5897 ** EOF on end-of-file.
5898 ** + Report syntax errors on stderr
5900 static char *SQLITE_CDECL ascii_read_one_field(ImportCtx *p){
5901 int c;
5902 int cSep = (u8)p->cColSep;
5903 int rSep = (u8)p->cRowSep;
5904 p->n = 0;
5905 c = fgetc(p->in);
5906 if( c==EOF || seenInterrupt ){
5907 p->cTerm = EOF;
5908 return 0;
5910 while( c!=EOF && c!=cSep && c!=rSep ){
5911 import_append_char(p, c);
5912 c = fgetc(p->in);
5914 if( c==rSep ){
5915 p->nLine++;
5917 p->cTerm = c;
5918 if( p->z ) p->z[p->n] = 0;
5919 return p->z;
5923 ** Try to transfer data for table zTable. If an error is seen while
5924 ** moving forward, try to go backwards. The backwards movement won't
5925 ** work for WITHOUT ROWID tables.
5927 static void tryToCloneData(
5928 ShellState *p,
5929 sqlite3 *newDb,
5930 const char *zTable
5932 sqlite3_stmt *pQuery = 0;
5933 sqlite3_stmt *pInsert = 0;
5934 char *zQuery = 0;
5935 char *zInsert = 0;
5936 int rc;
5937 int i, j, n;
5938 int nTable = strlen30(zTable);
5939 int k = 0;
5940 int cnt = 0;
5941 const int spinRate = 10000;
5943 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\"", zTable);
5944 shell_check_oom(zQuery);
5945 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
5946 if( rc ){
5947 eputf("Error %d: %s on [%s]\n",
5948 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db), zQuery);
5949 goto end_data_xfer;
5951 n = sqlite3_column_count(pQuery);
5952 zInsert = sqlite3_malloc64(200 + nTable + n*3);
5953 shell_check_oom(zInsert);
5954 sqlite3_snprintf(200+nTable,zInsert,
5955 "INSERT OR IGNORE INTO \"%s\" VALUES(?", zTable);
5956 i = strlen30(zInsert);
5957 for(j=1; j<n; j++){
5958 memcpy(zInsert+i, ",?", 2);
5959 i += 2;
5961 memcpy(zInsert+i, ");", 3);
5962 rc = sqlite3_prepare_v2(newDb, zInsert, -1, &pInsert, 0);
5963 if( rc ){
5964 eputf("Error %d: %s on [%s]\n",
5965 sqlite3_extended_errcode(newDb), sqlite3_errmsg(newDb), zInsert);
5966 goto end_data_xfer;
5968 for(k=0; k<2; k++){
5969 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
5970 for(i=0; i<n; i++){
5971 switch( sqlite3_column_type(pQuery, i) ){
5972 case SQLITE_NULL: {
5973 sqlite3_bind_null(pInsert, i+1);
5974 break;
5976 case SQLITE_INTEGER: {
5977 sqlite3_bind_int64(pInsert, i+1, sqlite3_column_int64(pQuery,i));
5978 break;
5980 case SQLITE_FLOAT: {
5981 sqlite3_bind_double(pInsert, i+1, sqlite3_column_double(pQuery,i));
5982 break;
5984 case SQLITE_TEXT: {
5985 sqlite3_bind_text(pInsert, i+1,
5986 (const char*)sqlite3_column_text(pQuery,i),
5987 -1, SQLITE_STATIC);
5988 break;
5990 case SQLITE_BLOB: {
5991 sqlite3_bind_blob(pInsert, i+1, sqlite3_column_blob(pQuery,i),
5992 sqlite3_column_bytes(pQuery,i),
5993 SQLITE_STATIC);
5994 break;
5997 } /* End for */
5998 rc = sqlite3_step(pInsert);
5999 if( rc!=SQLITE_OK && rc!=SQLITE_ROW && rc!=SQLITE_DONE ){
6000 eputf("Error %d: %s\n",
6001 sqlite3_extended_errcode(newDb), sqlite3_errmsg(newDb));
6003 sqlite3_reset(pInsert);
6004 cnt++;
6005 if( (cnt%spinRate)==0 ){
6006 printf("%c\b", "|/-\\"[(cnt/spinRate)%4]);
6007 fflush(stdout);
6009 } /* End while */
6010 if( rc==SQLITE_DONE ) break;
6011 sqlite3_finalize(pQuery);
6012 sqlite3_free(zQuery);
6013 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\" ORDER BY rowid DESC;",
6014 zTable);
6015 shell_check_oom(zQuery);
6016 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
6017 if( rc ){
6018 eputf("Warning: cannot step \"%s\" backwards", zTable);
6019 break;
6021 } /* End for(k=0...) */
6023 end_data_xfer:
6024 sqlite3_finalize(pQuery);
6025 sqlite3_finalize(pInsert);
6026 sqlite3_free(zQuery);
6027 sqlite3_free(zInsert);
6032 ** Try to transfer all rows of the schema that match zWhere. For
6033 ** each row, invoke xForEach() on the object defined by that row.
6034 ** If an error is encountered while moving forward through the
6035 ** sqlite_schema table, try again moving backwards.
6037 static void tryToCloneSchema(
6038 ShellState *p,
6039 sqlite3 *newDb,
6040 const char *zWhere,
6041 void (*xForEach)(ShellState*,sqlite3*,const char*)
6043 sqlite3_stmt *pQuery = 0;
6044 char *zQuery = 0;
6045 int rc;
6046 const unsigned char *zName;
6047 const unsigned char *zSql;
6048 char *zErrMsg = 0;
6050 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_schema"
6051 " WHERE %s ORDER BY rowid ASC", zWhere);
6052 shell_check_oom(zQuery);
6053 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
6054 if( rc ){
6055 eputf("Error: (%d) %s on [%s]\n", sqlite3_extended_errcode(p->db),
6056 sqlite3_errmsg(p->db), zQuery);
6057 goto end_schema_xfer;
6059 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
6060 zName = sqlite3_column_text(pQuery, 0);
6061 zSql = sqlite3_column_text(pQuery, 1);
6062 if( zName==0 || zSql==0 ) continue;
6063 if( sqlite3_stricmp((char*)zName, "sqlite_sequence")!=0 ){
6064 sputf(stdout, "%s... ", zName); fflush(stdout);
6065 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
6066 if( zErrMsg ){
6067 eputf("Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
6068 sqlite3_free(zErrMsg);
6069 zErrMsg = 0;
6072 if( xForEach ){
6073 xForEach(p, newDb, (const char*)zName);
6075 sputz(stdout, "done\n");
6077 if( rc!=SQLITE_DONE ){
6078 sqlite3_finalize(pQuery);
6079 sqlite3_free(zQuery);
6080 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_schema"
6081 " WHERE %s ORDER BY rowid DESC", zWhere);
6082 shell_check_oom(zQuery);
6083 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
6084 if( rc ){
6085 eputf("Error: (%d) %s on [%s]\n",
6086 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db), zQuery);
6087 goto end_schema_xfer;
6089 while( sqlite3_step(pQuery)==SQLITE_ROW ){
6090 zName = sqlite3_column_text(pQuery, 0);
6091 zSql = sqlite3_column_text(pQuery, 1);
6092 if( zName==0 || zSql==0 ) continue;
6093 if( sqlite3_stricmp((char*)zName, "sqlite_sequence")==0 ) continue;
6094 sputf(stdout, "%s... ", zName); fflush(stdout);
6095 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
6096 if( zErrMsg ){
6097 eputf("Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
6098 sqlite3_free(zErrMsg);
6099 zErrMsg = 0;
6101 if( xForEach ){
6102 xForEach(p, newDb, (const char*)zName);
6104 sputz(stdout, "done\n");
6107 end_schema_xfer:
6108 sqlite3_finalize(pQuery);
6109 sqlite3_free(zQuery);
6113 ** Open a new database file named "zNewDb". Try to recover as much information
6114 ** as possible out of the main database (which might be corrupt) and write it
6115 ** into zNewDb.
6117 static void tryToClone(ShellState *p, const char *zNewDb){
6118 int rc;
6119 sqlite3 *newDb = 0;
6120 if( access(zNewDb,0)==0 ){
6121 eputf("File \"%s\" already exists.\n", zNewDb);
6122 return;
6124 rc = sqlite3_open(zNewDb, &newDb);
6125 if( rc ){
6126 eputf("Cannot create output database: %s\n", sqlite3_errmsg(newDb));
6127 }else{
6128 sqlite3_exec(p->db, "PRAGMA writable_schema=ON;", 0, 0, 0);
6129 sqlite3_exec(newDb, "BEGIN EXCLUSIVE;", 0, 0, 0);
6130 tryToCloneSchema(p, newDb, "type='table'", tryToCloneData);
6131 tryToCloneSchema(p, newDb, "type!='table'", 0);
6132 sqlite3_exec(newDb, "COMMIT;", 0, 0, 0);
6133 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
6135 close_db(newDb);
6138 #ifndef SQLITE_SHELL_FIDDLE
6140 ** Change the output stream (file or pipe or console) to something else.
6142 static void output_redir(ShellState *p, FILE *pfNew){
6143 if( p->out != stdout ) eputz("Output already redirected.\n");
6144 else{
6145 p->out = pfNew;
6146 setOutputStream(pfNew);
6151 ** Change the output file back to stdout.
6153 ** If the p->doXdgOpen flag is set, that means the output was being
6154 ** redirected to a temporary file named by p->zTempFile. In that case,
6155 ** launch start/open/xdg-open on that temporary file.
6157 static void output_reset(ShellState *p){
6158 if( p->outfile[0]=='|' ){
6159 #ifndef SQLITE_OMIT_POPEN
6160 pclose(p->out);
6161 #endif
6162 }else{
6163 output_file_close(p->out);
6164 #ifndef SQLITE_NOHAVE_SYSTEM
6165 if( p->doXdgOpen ){
6166 const char *zXdgOpenCmd =
6167 #if defined(_WIN32)
6168 "start";
6169 #elif defined(__APPLE__)
6170 "open";
6171 #else
6172 "xdg-open";
6173 #endif
6174 char *zCmd;
6175 zCmd = sqlite3_mprintf("%s %s", zXdgOpenCmd, p->zTempFile);
6176 if( system(zCmd) ){
6177 eputf("Failed: [%s]\n", zCmd);
6178 }else{
6179 /* Give the start/open/xdg-open command some time to get
6180 ** going before we continue, and potential delete the
6181 ** p->zTempFile data file out from under it */
6182 sqlite3_sleep(2000);
6184 sqlite3_free(zCmd);
6185 outputModePop(p);
6186 p->doXdgOpen = 0;
6188 #endif /* !defined(SQLITE_NOHAVE_SYSTEM) */
6190 p->outfile[0] = 0;
6191 p->out = stdout;
6192 setOutputStream(stdout);
6194 #else
6195 # define output_redir(SS,pfO)
6196 # define output_reset(SS)
6197 #endif
6200 ** Run an SQL command and return the single integer result.
6202 static int db_int(sqlite3 *db, const char *zSql){
6203 sqlite3_stmt *pStmt;
6204 int res = 0;
6205 sqlite3_prepare_v2(db, zSql, -1, &pStmt, 0);
6206 if( pStmt && sqlite3_step(pStmt)==SQLITE_ROW ){
6207 res = sqlite3_column_int(pStmt,0);
6209 sqlite3_finalize(pStmt);
6210 return res;
6213 #if SQLITE_SHELL_HAVE_RECOVER
6215 ** Convert a 2-byte or 4-byte big-endian integer into a native integer
6217 static unsigned int get2byteInt(unsigned char *a){
6218 return (a[0]<<8) + a[1];
6220 static unsigned int get4byteInt(unsigned char *a){
6221 return (a[0]<<24) + (a[1]<<16) + (a[2]<<8) + a[3];
6225 ** Implementation of the ".dbinfo" command.
6227 ** Return 1 on error, 2 to exit, and 0 otherwise.
6229 static int shell_dbinfo_command(ShellState *p, int nArg, char **azArg){
6230 static const struct { const char *zName; int ofst; } aField[] = {
6231 { "file change counter:", 24 },
6232 { "database page count:", 28 },
6233 { "freelist page count:", 36 },
6234 { "schema cookie:", 40 },
6235 { "schema format:", 44 },
6236 { "default cache size:", 48 },
6237 { "autovacuum top root:", 52 },
6238 { "incremental vacuum:", 64 },
6239 { "text encoding:", 56 },
6240 { "user version:", 60 },
6241 { "application id:", 68 },
6242 { "software version:", 96 },
6244 static const struct { const char *zName; const char *zSql; } aQuery[] = {
6245 { "number of tables:",
6246 "SELECT count(*) FROM %s WHERE type='table'" },
6247 { "number of indexes:",
6248 "SELECT count(*) FROM %s WHERE type='index'" },
6249 { "number of triggers:",
6250 "SELECT count(*) FROM %s WHERE type='trigger'" },
6251 { "number of views:",
6252 "SELECT count(*) FROM %s WHERE type='view'" },
6253 { "schema size:",
6254 "SELECT total(length(sql)) FROM %s" },
6256 int i, rc;
6257 unsigned iDataVersion;
6258 char *zSchemaTab;
6259 char *zDb = nArg>=2 ? azArg[1] : "main";
6260 sqlite3_stmt *pStmt = 0;
6261 unsigned char aHdr[100];
6262 open_db(p, 0);
6263 if( p->db==0 ) return 1;
6264 rc = sqlite3_prepare_v2(p->db,
6265 "SELECT data FROM sqlite_dbpage(?1) WHERE pgno=1",
6266 -1, &pStmt, 0);
6267 if( rc ){
6268 eputf("error: %s\n", sqlite3_errmsg(p->db));
6269 sqlite3_finalize(pStmt);
6270 return 1;
6272 sqlite3_bind_text(pStmt, 1, zDb, -1, SQLITE_STATIC);
6273 if( sqlite3_step(pStmt)==SQLITE_ROW
6274 && sqlite3_column_bytes(pStmt,0)>100
6276 const u8 *pb = sqlite3_column_blob(pStmt,0);
6277 shell_check_oom(pb);
6278 memcpy(aHdr, pb, 100);
6279 sqlite3_finalize(pStmt);
6280 }else{
6281 eputz("unable to read database header\n");
6282 sqlite3_finalize(pStmt);
6283 return 1;
6285 i = get2byteInt(aHdr+16);
6286 if( i==1 ) i = 65536;
6287 oputf("%-20s %d\n", "database page size:", i);
6288 oputf("%-20s %d\n", "write format:", aHdr[18]);
6289 oputf("%-20s %d\n", "read format:", aHdr[19]);
6290 oputf("%-20s %d\n", "reserved bytes:", aHdr[20]);
6291 for(i=0; i<ArraySize(aField); i++){
6292 int ofst = aField[i].ofst;
6293 unsigned int val = get4byteInt(aHdr + ofst);
6294 oputf("%-20s %u", aField[i].zName, val);
6295 switch( ofst ){
6296 case 56: {
6297 if( val==1 ) oputz(" (utf8)");
6298 if( val==2 ) oputz(" (utf16le)");
6299 if( val==3 ) oputz(" (utf16be)");
6302 oputz("\n");
6304 if( zDb==0 ){
6305 zSchemaTab = sqlite3_mprintf("main.sqlite_schema");
6306 }else if( cli_strcmp(zDb,"temp")==0 ){
6307 zSchemaTab = sqlite3_mprintf("%s", "sqlite_temp_schema");
6308 }else{
6309 zSchemaTab = sqlite3_mprintf("\"%w\".sqlite_schema", zDb);
6311 for(i=0; i<ArraySize(aQuery); i++){
6312 char *zSql = sqlite3_mprintf(aQuery[i].zSql, zSchemaTab);
6313 int val = db_int(p->db, zSql);
6314 sqlite3_free(zSql);
6315 oputf("%-20s %d\n", aQuery[i].zName, val);
6317 sqlite3_free(zSchemaTab);
6318 sqlite3_file_control(p->db, zDb, SQLITE_FCNTL_DATA_VERSION, &iDataVersion);
6319 oputf("%-20s %u\n", "data version", iDataVersion);
6320 return 0;
6322 #endif /* SQLITE_SHELL_HAVE_RECOVER */
6325 ** Print the current sqlite3_errmsg() value to stderr and return 1.
6327 static int shellDatabaseError(sqlite3 *db){
6328 const char *zErr = sqlite3_errmsg(db);
6329 eputf("Error: %s\n", zErr);
6330 return 1;
6334 ** Compare the pattern in zGlob[] against the text in z[]. Return TRUE
6335 ** if they match and FALSE (0) if they do not match.
6337 ** Globbing rules:
6339 ** '*' Matches any sequence of zero or more characters.
6341 ** '?' Matches exactly one character.
6343 ** [...] Matches one character from the enclosed list of
6344 ** characters.
6346 ** [^...] Matches one character not in the enclosed list.
6348 ** '#' Matches any sequence of one or more digits with an
6349 ** optional + or - sign in front
6351 ** ' ' Any span of whitespace matches any other span of
6352 ** whitespace.
6354 ** Extra whitespace at the end of z[] is ignored.
6356 static int testcase_glob(const char *zGlob, const char *z){
6357 int c, c2;
6358 int invert;
6359 int seen;
6361 while( (c = (*(zGlob++)))!=0 ){
6362 if( IsSpace(c) ){
6363 if( !IsSpace(*z) ) return 0;
6364 while( IsSpace(*zGlob) ) zGlob++;
6365 while( IsSpace(*z) ) z++;
6366 }else if( c=='*' ){
6367 while( (c=(*(zGlob++))) == '*' || c=='?' ){
6368 if( c=='?' && (*(z++))==0 ) return 0;
6370 if( c==0 ){
6371 return 1;
6372 }else if( c=='[' ){
6373 while( *z && testcase_glob(zGlob-1,z)==0 ){
6374 z++;
6376 return (*z)!=0;
6378 while( (c2 = (*(z++)))!=0 ){
6379 while( c2!=c ){
6380 c2 = *(z++);
6381 if( c2==0 ) return 0;
6383 if( testcase_glob(zGlob,z) ) return 1;
6385 return 0;
6386 }else if( c=='?' ){
6387 if( (*(z++))==0 ) return 0;
6388 }else if( c=='[' ){
6389 int prior_c = 0;
6390 seen = 0;
6391 invert = 0;
6392 c = *(z++);
6393 if( c==0 ) return 0;
6394 c2 = *(zGlob++);
6395 if( c2=='^' ){
6396 invert = 1;
6397 c2 = *(zGlob++);
6399 if( c2==']' ){
6400 if( c==']' ) seen = 1;
6401 c2 = *(zGlob++);
6403 while( c2 && c2!=']' ){
6404 if( c2=='-' && zGlob[0]!=']' && zGlob[0]!=0 && prior_c>0 ){
6405 c2 = *(zGlob++);
6406 if( c>=prior_c && c<=c2 ) seen = 1;
6407 prior_c = 0;
6408 }else{
6409 if( c==c2 ){
6410 seen = 1;
6412 prior_c = c2;
6414 c2 = *(zGlob++);
6416 if( c2==0 || (seen ^ invert)==0 ) return 0;
6417 }else if( c=='#' ){
6418 if( (z[0]=='-' || z[0]=='+') && IsDigit(z[1]) ) z++;
6419 if( !IsDigit(z[0]) ) return 0;
6420 z++;
6421 while( IsDigit(z[0]) ){ z++; }
6422 }else{
6423 if( c!=(*(z++)) ) return 0;
6426 while( IsSpace(*z) ){ z++; }
6427 return *z==0;
6432 ** Compare the string as a command-line option with either one or two
6433 ** initial "-" characters.
6435 static int optionMatch(const char *zStr, const char *zOpt){
6436 if( zStr[0]!='-' ) return 0;
6437 zStr++;
6438 if( zStr[0]=='-' ) zStr++;
6439 return cli_strcmp(zStr, zOpt)==0;
6443 ** Delete a file.
6445 int shellDeleteFile(const char *zFilename){
6446 int rc;
6447 #ifdef _WIN32
6448 wchar_t *z = sqlite3_win32_utf8_to_unicode(zFilename);
6449 rc = _wunlink(z);
6450 sqlite3_free(z);
6451 #else
6452 rc = unlink(zFilename);
6453 #endif
6454 return rc;
6458 ** Try to delete the temporary file (if there is one) and free the
6459 ** memory used to hold the name of the temp file.
6461 static void clearTempFile(ShellState *p){
6462 if( p->zTempFile==0 ) return;
6463 if( p->doXdgOpen ) return;
6464 if( shellDeleteFile(p->zTempFile) ) return;
6465 sqlite3_free(p->zTempFile);
6466 p->zTempFile = 0;
6470 ** Create a new temp file name with the given suffix.
6472 static void newTempFile(ShellState *p, const char *zSuffix){
6473 clearTempFile(p);
6474 sqlite3_free(p->zTempFile);
6475 p->zTempFile = 0;
6476 if( p->db ){
6477 sqlite3_file_control(p->db, 0, SQLITE_FCNTL_TEMPFILENAME, &p->zTempFile);
6479 if( p->zTempFile==0 ){
6480 /* If p->db is an in-memory database then the TEMPFILENAME file-control
6481 ** will not work and we will need to fallback to guessing */
6482 char *zTemp;
6483 sqlite3_uint64 r;
6484 sqlite3_randomness(sizeof(r), &r);
6485 zTemp = getenv("TEMP");
6486 if( zTemp==0 ) zTemp = getenv("TMP");
6487 if( zTemp==0 ){
6488 #ifdef _WIN32
6489 zTemp = "\\tmp";
6490 #else
6491 zTemp = "/tmp";
6492 #endif
6494 p->zTempFile = sqlite3_mprintf("%s/temp%llx.%s", zTemp, r, zSuffix);
6495 }else{
6496 p->zTempFile = sqlite3_mprintf("%z.%s", p->zTempFile, zSuffix);
6498 shell_check_oom(p->zTempFile);
6503 ** The implementation of SQL scalar function fkey_collate_clause(), used
6504 ** by the ".lint fkey-indexes" command. This scalar function is always
6505 ** called with four arguments - the parent table name, the parent column name,
6506 ** the child table name and the child column name.
6508 ** fkey_collate_clause('parent-tab', 'parent-col', 'child-tab', 'child-col')
6510 ** If either of the named tables or columns do not exist, this function
6511 ** returns an empty string. An empty string is also returned if both tables
6512 ** and columns exist but have the same default collation sequence. Or,
6513 ** if both exist but the default collation sequences are different, this
6514 ** function returns the string " COLLATE <parent-collation>", where
6515 ** <parent-collation> is the default collation sequence of the parent column.
6517 static void shellFkeyCollateClause(
6518 sqlite3_context *pCtx,
6519 int nVal,
6520 sqlite3_value **apVal
6522 sqlite3 *db = sqlite3_context_db_handle(pCtx);
6523 const char *zParent;
6524 const char *zParentCol;
6525 const char *zParentSeq;
6526 const char *zChild;
6527 const char *zChildCol;
6528 const char *zChildSeq = 0; /* Initialize to avoid false-positive warning */
6529 int rc;
6531 assert( nVal==4 );
6532 zParent = (const char*)sqlite3_value_text(apVal[0]);
6533 zParentCol = (const char*)sqlite3_value_text(apVal[1]);
6534 zChild = (const char*)sqlite3_value_text(apVal[2]);
6535 zChildCol = (const char*)sqlite3_value_text(apVal[3]);
6537 sqlite3_result_text(pCtx, "", -1, SQLITE_STATIC);
6538 rc = sqlite3_table_column_metadata(
6539 db, "main", zParent, zParentCol, 0, &zParentSeq, 0, 0, 0
6541 if( rc==SQLITE_OK ){
6542 rc = sqlite3_table_column_metadata(
6543 db, "main", zChild, zChildCol, 0, &zChildSeq, 0, 0, 0
6547 if( rc==SQLITE_OK && sqlite3_stricmp(zParentSeq, zChildSeq) ){
6548 char *z = sqlite3_mprintf(" COLLATE %s", zParentSeq);
6549 sqlite3_result_text(pCtx, z, -1, SQLITE_TRANSIENT);
6550 sqlite3_free(z);
6556 ** The implementation of dot-command ".lint fkey-indexes".
6558 static int lintFkeyIndexes(
6559 ShellState *pState, /* Current shell tool state */
6560 char **azArg, /* Array of arguments passed to dot command */
6561 int nArg /* Number of entries in azArg[] */
6563 sqlite3 *db = pState->db; /* Database handle to query "main" db of */
6564 int bVerbose = 0; /* If -verbose is present */
6565 int bGroupByParent = 0; /* If -groupbyparent is present */
6566 int i; /* To iterate through azArg[] */
6567 const char *zIndent = ""; /* How much to indent CREATE INDEX by */
6568 int rc; /* Return code */
6569 sqlite3_stmt *pSql = 0; /* Compiled version of SQL statement below */
6572 ** This SELECT statement returns one row for each foreign key constraint
6573 ** in the schema of the main database. The column values are:
6575 ** 0. The text of an SQL statement similar to:
6577 ** "EXPLAIN QUERY PLAN SELECT 1 FROM child_table WHERE child_key=?"
6579 ** This SELECT is similar to the one that the foreign keys implementation
6580 ** needs to run internally on child tables. If there is an index that can
6581 ** be used to optimize this query, then it can also be used by the FK
6582 ** implementation to optimize DELETE or UPDATE statements on the parent
6583 ** table.
6585 ** 1. A GLOB pattern suitable for sqlite3_strglob(). If the plan output by
6586 ** the EXPLAIN QUERY PLAN command matches this pattern, then the schema
6587 ** contains an index that can be used to optimize the query.
6589 ** 2. Human readable text that describes the child table and columns. e.g.
6591 ** "child_table(child_key1, child_key2)"
6593 ** 3. Human readable text that describes the parent table and columns. e.g.
6595 ** "parent_table(parent_key1, parent_key2)"
6597 ** 4. A full CREATE INDEX statement for an index that could be used to
6598 ** optimize DELETE or UPDATE statements on the parent table. e.g.
6600 ** "CREATE INDEX child_table_child_key ON child_table(child_key)"
6602 ** 5. The name of the parent table.
6604 ** These six values are used by the C logic below to generate the report.
6606 const char *zSql =
6607 "SELECT "
6608 " 'EXPLAIN QUERY PLAN SELECT 1 FROM ' || quote(s.name) || ' WHERE '"
6609 " || group_concat(quote(s.name) || '.' || quote(f.[from]) || '=?' "
6610 " || fkey_collate_clause("
6611 " f.[table], COALESCE(f.[to], p.[name]), s.name, f.[from]),' AND ')"
6612 ", "
6613 " 'SEARCH ' || s.name || ' USING COVERING INDEX*('"
6614 " || group_concat('*=?', ' AND ') || ')'"
6615 ", "
6616 " s.name || '(' || group_concat(f.[from], ', ') || ')'"
6617 ", "
6618 " f.[table] || '(' || group_concat(COALESCE(f.[to], p.[name])) || ')'"
6619 ", "
6620 " 'CREATE INDEX ' || quote(s.name ||'_'|| group_concat(f.[from], '_'))"
6621 " || ' ON ' || quote(s.name) || '('"
6622 " || group_concat(quote(f.[from]) ||"
6623 " fkey_collate_clause("
6624 " f.[table], COALESCE(f.[to], p.[name]), s.name, f.[from]), ', ')"
6625 " || ');'"
6626 ", "
6627 " f.[table] "
6628 "FROM sqlite_schema AS s, pragma_foreign_key_list(s.name) AS f "
6629 "LEFT JOIN pragma_table_info AS p ON (pk-1=seq AND p.arg=f.[table]) "
6630 "GROUP BY s.name, f.id "
6631 "ORDER BY (CASE WHEN ? THEN f.[table] ELSE s.name END)"
6633 const char *zGlobIPK = "SEARCH * USING INTEGER PRIMARY KEY (rowid=?)";
6635 for(i=2; i<nArg; i++){
6636 int n = strlen30(azArg[i]);
6637 if( n>1 && sqlite3_strnicmp("-verbose", azArg[i], n)==0 ){
6638 bVerbose = 1;
6640 else if( n>1 && sqlite3_strnicmp("-groupbyparent", azArg[i], n)==0 ){
6641 bGroupByParent = 1;
6642 zIndent = " ";
6644 else{
6645 eputf("Usage: %s %s ?-verbose? ?-groupbyparent?\n", azArg[0], azArg[1]);
6646 return SQLITE_ERROR;
6650 /* Register the fkey_collate_clause() SQL function */
6651 rc = sqlite3_create_function(db, "fkey_collate_clause", 4, SQLITE_UTF8,
6652 0, shellFkeyCollateClause, 0, 0
6656 if( rc==SQLITE_OK ){
6657 rc = sqlite3_prepare_v2(db, zSql, -1, &pSql, 0);
6659 if( rc==SQLITE_OK ){
6660 sqlite3_bind_int(pSql, 1, bGroupByParent);
6663 if( rc==SQLITE_OK ){
6664 int rc2;
6665 char *zPrev = 0;
6666 while( SQLITE_ROW==sqlite3_step(pSql) ){
6667 int res = -1;
6668 sqlite3_stmt *pExplain = 0;
6669 const char *zEQP = (const char*)sqlite3_column_text(pSql, 0);
6670 const char *zGlob = (const char*)sqlite3_column_text(pSql, 1);
6671 const char *zFrom = (const char*)sqlite3_column_text(pSql, 2);
6672 const char *zTarget = (const char*)sqlite3_column_text(pSql, 3);
6673 const char *zCI = (const char*)sqlite3_column_text(pSql, 4);
6674 const char *zParent = (const char*)sqlite3_column_text(pSql, 5);
6676 if( zEQP==0 ) continue;
6677 if( zGlob==0 ) continue;
6678 rc = sqlite3_prepare_v2(db, zEQP, -1, &pExplain, 0);
6679 if( rc!=SQLITE_OK ) break;
6680 if( SQLITE_ROW==sqlite3_step(pExplain) ){
6681 const char *zPlan = (const char*)sqlite3_column_text(pExplain, 3);
6682 res = zPlan!=0 && ( 0==sqlite3_strglob(zGlob, zPlan)
6683 || 0==sqlite3_strglob(zGlobIPK, zPlan));
6685 rc = sqlite3_finalize(pExplain);
6686 if( rc!=SQLITE_OK ) break;
6688 if( res<0 ){
6689 eputz("Error: internal error");
6690 break;
6691 }else{
6692 if( bGroupByParent
6693 && (bVerbose || res==0)
6694 && (zPrev==0 || sqlite3_stricmp(zParent, zPrev))
6696 oputf("-- Parent table %s\n", zParent);
6697 sqlite3_free(zPrev);
6698 zPrev = sqlite3_mprintf("%s", zParent);
6701 if( res==0 ){
6702 oputf("%s%s --> %s\n", zIndent, zCI, zTarget);
6703 }else if( bVerbose ){
6704 oputf("%s/* no extra indexes required for %s -> %s */\n",
6705 zIndent, zFrom, zTarget
6710 sqlite3_free(zPrev);
6712 if( rc!=SQLITE_OK ){
6713 eputf("%s\n", sqlite3_errmsg(db));
6716 rc2 = sqlite3_finalize(pSql);
6717 if( rc==SQLITE_OK && rc2!=SQLITE_OK ){
6718 rc = rc2;
6719 eputf("%s\n", sqlite3_errmsg(db));
6721 }else{
6722 eputf("%s\n", sqlite3_errmsg(db));
6725 return rc;
6729 ** Implementation of ".lint" dot command.
6731 static int lintDotCommand(
6732 ShellState *pState, /* Current shell tool state */
6733 char **azArg, /* Array of arguments passed to dot command */
6734 int nArg /* Number of entries in azArg[] */
6736 int n;
6737 n = (nArg>=2 ? strlen30(azArg[1]) : 0);
6738 if( n<1 || sqlite3_strnicmp(azArg[1], "fkey-indexes", n) ) goto usage;
6739 return lintFkeyIndexes(pState, azArg, nArg);
6741 usage:
6742 eputf("Usage %s sub-command ?switches...?\n", azArg[0]);
6743 eputz("Where sub-commands are:\n");
6744 eputz(" fkey-indexes\n");
6745 return SQLITE_ERROR;
6748 static void shellPrepare(
6749 sqlite3 *db,
6750 int *pRc,
6751 const char *zSql,
6752 sqlite3_stmt **ppStmt
6754 *ppStmt = 0;
6755 if( *pRc==SQLITE_OK ){
6756 int rc = sqlite3_prepare_v2(db, zSql, -1, ppStmt, 0);
6757 if( rc!=SQLITE_OK ){
6758 eputf("sql error: %s (%d)\n", sqlite3_errmsg(db), sqlite3_errcode(db));
6759 *pRc = rc;
6765 ** Create a prepared statement using printf-style arguments for the SQL.
6767 static void shellPreparePrintf(
6768 sqlite3 *db,
6769 int *pRc,
6770 sqlite3_stmt **ppStmt,
6771 const char *zFmt,
6774 *ppStmt = 0;
6775 if( *pRc==SQLITE_OK ){
6776 va_list ap;
6777 char *z;
6778 va_start(ap, zFmt);
6779 z = sqlite3_vmprintf(zFmt, ap);
6780 va_end(ap);
6781 if( z==0 ){
6782 *pRc = SQLITE_NOMEM;
6783 }else{
6784 shellPrepare(db, pRc, z, ppStmt);
6785 sqlite3_free(z);
6791 ** Finalize the prepared statement created using shellPreparePrintf().
6793 static void shellFinalize(
6794 int *pRc,
6795 sqlite3_stmt *pStmt
6797 if( pStmt ){
6798 sqlite3 *db = sqlite3_db_handle(pStmt);
6799 int rc = sqlite3_finalize(pStmt);
6800 if( *pRc==SQLITE_OK ){
6801 if( rc!=SQLITE_OK ){
6802 eputf("SQL error: %s\n", sqlite3_errmsg(db));
6804 *pRc = rc;
6809 #if !defined SQLITE_OMIT_VIRTUALTABLE
6810 /* Reset the prepared statement created using shellPreparePrintf().
6812 ** This routine is could be marked "static". But it is not always used,
6813 ** depending on compile-time options. By omitting the "static", we avoid
6814 ** nuisance compiler warnings about "defined but not used".
6816 void shellReset(
6817 int *pRc,
6818 sqlite3_stmt *pStmt
6820 int rc = sqlite3_reset(pStmt);
6821 if( *pRc==SQLITE_OK ){
6822 if( rc!=SQLITE_OK ){
6823 sqlite3 *db = sqlite3_db_handle(pStmt);
6824 eputf("SQL error: %s\n", sqlite3_errmsg(db));
6826 *pRc = rc;
6829 #endif /* !defined SQLITE_OMIT_VIRTUALTABLE */
6831 #if !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_HAVE_ZLIB)
6832 /******************************************************************************
6833 ** The ".archive" or ".ar" command.
6836 ** Structure representing a single ".ar" command.
6838 typedef struct ArCommand ArCommand;
6839 struct ArCommand {
6840 u8 eCmd; /* An AR_CMD_* value */
6841 u8 bVerbose; /* True if --verbose */
6842 u8 bZip; /* True if the archive is a ZIP */
6843 u8 bDryRun; /* True if --dry-run */
6844 u8 bAppend; /* True if --append */
6845 u8 bGlob; /* True if --glob */
6846 u8 fromCmdLine; /* Run from -A instead of .archive */
6847 int nArg; /* Number of command arguments */
6848 char *zSrcTable; /* "sqlar", "zipfile($file)" or "zip" */
6849 const char *zFile; /* --file argument, or NULL */
6850 const char *zDir; /* --directory argument, or NULL */
6851 char **azArg; /* Array of command arguments */
6852 ShellState *p; /* Shell state */
6853 sqlite3 *db; /* Database containing the archive */
6857 ** Print a usage message for the .ar command to stderr and return SQLITE_ERROR.
6859 static int arUsage(FILE *f){
6860 showHelp(f,"archive");
6861 return SQLITE_ERROR;
6865 ** Print an error message for the .ar command to stderr and return
6866 ** SQLITE_ERROR.
6868 static int arErrorMsg(ArCommand *pAr, const char *zFmt, ...){
6869 va_list ap;
6870 char *z;
6871 va_start(ap, zFmt);
6872 z = sqlite3_vmprintf(zFmt, ap);
6873 va_end(ap);
6874 eputf("Error: %s\n", z);
6875 if( pAr->fromCmdLine ){
6876 eputz("Use \"-A\" for more help\n");
6877 }else{
6878 eputz("Use \".archive --help\" for more help\n");
6880 sqlite3_free(z);
6881 return SQLITE_ERROR;
6885 ** Values for ArCommand.eCmd.
6887 #define AR_CMD_CREATE 1
6888 #define AR_CMD_UPDATE 2
6889 #define AR_CMD_INSERT 3
6890 #define AR_CMD_EXTRACT 4
6891 #define AR_CMD_LIST 5
6892 #define AR_CMD_HELP 6
6893 #define AR_CMD_REMOVE 7
6896 ** Other (non-command) switches.
6898 #define AR_SWITCH_VERBOSE 8
6899 #define AR_SWITCH_FILE 9
6900 #define AR_SWITCH_DIRECTORY 10
6901 #define AR_SWITCH_APPEND 11
6902 #define AR_SWITCH_DRYRUN 12
6903 #define AR_SWITCH_GLOB 13
6905 static int arProcessSwitch(ArCommand *pAr, int eSwitch, const char *zArg){
6906 switch( eSwitch ){
6907 case AR_CMD_CREATE:
6908 case AR_CMD_EXTRACT:
6909 case AR_CMD_LIST:
6910 case AR_CMD_REMOVE:
6911 case AR_CMD_UPDATE:
6912 case AR_CMD_INSERT:
6913 case AR_CMD_HELP:
6914 if( pAr->eCmd ){
6915 return arErrorMsg(pAr, "multiple command options");
6917 pAr->eCmd = eSwitch;
6918 break;
6920 case AR_SWITCH_DRYRUN:
6921 pAr->bDryRun = 1;
6922 break;
6923 case AR_SWITCH_GLOB:
6924 pAr->bGlob = 1;
6925 break;
6926 case AR_SWITCH_VERBOSE:
6927 pAr->bVerbose = 1;
6928 break;
6929 case AR_SWITCH_APPEND:
6930 pAr->bAppend = 1;
6931 deliberate_fall_through;
6932 case AR_SWITCH_FILE:
6933 pAr->zFile = zArg;
6934 break;
6935 case AR_SWITCH_DIRECTORY:
6936 pAr->zDir = zArg;
6937 break;
6940 return SQLITE_OK;
6944 ** Parse the command line for an ".ar" command. The results are written into
6945 ** structure (*pAr). SQLITE_OK is returned if the command line is parsed
6946 ** successfully, otherwise an error message is written to stderr and
6947 ** SQLITE_ERROR returned.
6949 static int arParseCommand(
6950 char **azArg, /* Array of arguments passed to dot command */
6951 int nArg, /* Number of entries in azArg[] */
6952 ArCommand *pAr /* Populate this object */
6954 struct ArSwitch {
6955 const char *zLong;
6956 char cShort;
6957 u8 eSwitch;
6958 u8 bArg;
6959 } aSwitch[] = {
6960 { "create", 'c', AR_CMD_CREATE, 0 },
6961 { "extract", 'x', AR_CMD_EXTRACT, 0 },
6962 { "insert", 'i', AR_CMD_INSERT, 0 },
6963 { "list", 't', AR_CMD_LIST, 0 },
6964 { "remove", 'r', AR_CMD_REMOVE, 0 },
6965 { "update", 'u', AR_CMD_UPDATE, 0 },
6966 { "help", 'h', AR_CMD_HELP, 0 },
6967 { "verbose", 'v', AR_SWITCH_VERBOSE, 0 },
6968 { "file", 'f', AR_SWITCH_FILE, 1 },
6969 { "append", 'a', AR_SWITCH_APPEND, 1 },
6970 { "directory", 'C', AR_SWITCH_DIRECTORY, 1 },
6971 { "dryrun", 'n', AR_SWITCH_DRYRUN, 0 },
6972 { "glob", 'g', AR_SWITCH_GLOB, 0 },
6974 int nSwitch = sizeof(aSwitch) / sizeof(struct ArSwitch);
6975 struct ArSwitch *pEnd = &aSwitch[nSwitch];
6977 if( nArg<=1 ){
6978 eputz("Wrong number of arguments. Usage:\n");
6979 return arUsage(stderr);
6980 }else{
6981 char *z = azArg[1];
6982 if( z[0]!='-' ){
6983 /* Traditional style [tar] invocation */
6984 int i;
6985 int iArg = 2;
6986 for(i=0; z[i]; i++){
6987 const char *zArg = 0;
6988 struct ArSwitch *pOpt;
6989 for(pOpt=&aSwitch[0]; pOpt<pEnd; pOpt++){
6990 if( z[i]==pOpt->cShort ) break;
6992 if( pOpt==pEnd ){
6993 return arErrorMsg(pAr, "unrecognized option: %c", z[i]);
6995 if( pOpt->bArg ){
6996 if( iArg>=nArg ){
6997 return arErrorMsg(pAr, "option requires an argument: %c",z[i]);
6999 zArg = azArg[iArg++];
7001 if( arProcessSwitch(pAr, pOpt->eSwitch, zArg) ) return SQLITE_ERROR;
7003 pAr->nArg = nArg-iArg;
7004 if( pAr->nArg>0 ){
7005 pAr->azArg = &azArg[iArg];
7007 }else{
7008 /* Non-traditional invocation */
7009 int iArg;
7010 for(iArg=1; iArg<nArg; iArg++){
7011 int n;
7012 z = azArg[iArg];
7013 if( z[0]!='-' ){
7014 /* All remaining command line words are command arguments. */
7015 pAr->azArg = &azArg[iArg];
7016 pAr->nArg = nArg-iArg;
7017 break;
7019 n = strlen30(z);
7021 if( z[1]!='-' ){
7022 int i;
7023 /* One or more short options */
7024 for(i=1; i<n; i++){
7025 const char *zArg = 0;
7026 struct ArSwitch *pOpt;
7027 for(pOpt=&aSwitch[0]; pOpt<pEnd; pOpt++){
7028 if( z[i]==pOpt->cShort ) break;
7030 if( pOpt==pEnd ){
7031 return arErrorMsg(pAr, "unrecognized option: %c", z[i]);
7033 if( pOpt->bArg ){
7034 if( i<(n-1) ){
7035 zArg = &z[i+1];
7036 i = n;
7037 }else{
7038 if( iArg>=(nArg-1) ){
7039 return arErrorMsg(pAr, "option requires an argument: %c",
7040 z[i]);
7042 zArg = azArg[++iArg];
7045 if( arProcessSwitch(pAr, pOpt->eSwitch, zArg) ) return SQLITE_ERROR;
7047 }else if( z[2]=='\0' ){
7048 /* A -- option, indicating that all remaining command line words
7049 ** are command arguments. */
7050 pAr->azArg = &azArg[iArg+1];
7051 pAr->nArg = nArg-iArg-1;
7052 break;
7053 }else{
7054 /* A long option */
7055 const char *zArg = 0; /* Argument for option, if any */
7056 struct ArSwitch *pMatch = 0; /* Matching option */
7057 struct ArSwitch *pOpt; /* Iterator */
7058 for(pOpt=&aSwitch[0]; pOpt<pEnd; pOpt++){
7059 const char *zLong = pOpt->zLong;
7060 if( (n-2)<=strlen30(zLong) && 0==memcmp(&z[2], zLong, n-2) ){
7061 if( pMatch ){
7062 return arErrorMsg(pAr, "ambiguous option: %s",z);
7063 }else{
7064 pMatch = pOpt;
7069 if( pMatch==0 ){
7070 return arErrorMsg(pAr, "unrecognized option: %s", z);
7072 if( pMatch->bArg ){
7073 if( iArg>=(nArg-1) ){
7074 return arErrorMsg(pAr, "option requires an argument: %s", z);
7076 zArg = azArg[++iArg];
7078 if( arProcessSwitch(pAr, pMatch->eSwitch, zArg) ) return SQLITE_ERROR;
7083 if( pAr->eCmd==0 ){
7084 eputz("Required argument missing. Usage:\n");
7085 return arUsage(stderr);
7087 return SQLITE_OK;
7091 ** This function assumes that all arguments within the ArCommand.azArg[]
7092 ** array refer to archive members, as for the --extract, --list or --remove
7093 ** commands. It checks that each of them are "present". If any specified
7094 ** file is not present in the archive, an error is printed to stderr and an
7095 ** error code returned. Otherwise, if all specified arguments are present
7096 ** in the archive, SQLITE_OK is returned. Here, "present" means either an
7097 ** exact equality when pAr->bGlob is false or a "name GLOB pattern" match
7098 ** when pAr->bGlob is true.
7100 ** This function strips any trailing '/' characters from each argument.
7101 ** This is consistent with the way the [tar] command seems to work on
7102 ** Linux.
7104 static int arCheckEntries(ArCommand *pAr){
7105 int rc = SQLITE_OK;
7106 if( pAr->nArg ){
7107 int i, j;
7108 sqlite3_stmt *pTest = 0;
7109 const char *zSel = (pAr->bGlob)
7110 ? "SELECT name FROM %s WHERE glob($name,name)"
7111 : "SELECT name FROM %s WHERE name=$name";
7113 shellPreparePrintf(pAr->db, &rc, &pTest, zSel, pAr->zSrcTable);
7114 j = sqlite3_bind_parameter_index(pTest, "$name");
7115 for(i=0; i<pAr->nArg && rc==SQLITE_OK; i++){
7116 char *z = pAr->azArg[i];
7117 int n = strlen30(z);
7118 int bOk = 0;
7119 while( n>0 && z[n-1]=='/' ) n--;
7120 z[n] = '\0';
7121 sqlite3_bind_text(pTest, j, z, -1, SQLITE_STATIC);
7122 if( SQLITE_ROW==sqlite3_step(pTest) ){
7123 bOk = 1;
7125 shellReset(&rc, pTest);
7126 if( rc==SQLITE_OK && bOk==0 ){
7127 eputf("not found in archive: %s\n", z);
7128 rc = SQLITE_ERROR;
7131 shellFinalize(&rc, pTest);
7133 return rc;
7137 ** Format a WHERE clause that can be used against the "sqlar" table to
7138 ** identify all archive members that match the command arguments held
7139 ** in (*pAr). Leave this WHERE clause in (*pzWhere) before returning.
7140 ** The caller is responsible for eventually calling sqlite3_free() on
7141 ** any non-NULL (*pzWhere) value. Here, "match" means strict equality
7142 ** when pAr->bGlob is false and GLOB match when pAr->bGlob is true.
7144 static void arWhereClause(
7145 int *pRc,
7146 ArCommand *pAr,
7147 char **pzWhere /* OUT: New WHERE clause */
7149 char *zWhere = 0;
7150 const char *zSameOp = (pAr->bGlob)? "GLOB" : "=";
7151 if( *pRc==SQLITE_OK ){
7152 if( pAr->nArg==0 ){
7153 zWhere = sqlite3_mprintf("1");
7154 }else{
7155 int i;
7156 const char *zSep = "";
7157 for(i=0; i<pAr->nArg; i++){
7158 const char *z = pAr->azArg[i];
7159 zWhere = sqlite3_mprintf(
7160 "%z%s name %s '%q' OR substr(name,1,%d) %s '%q/'",
7161 zWhere, zSep, zSameOp, z, strlen30(z)+1, zSameOp, z
7163 if( zWhere==0 ){
7164 *pRc = SQLITE_NOMEM;
7165 break;
7167 zSep = " OR ";
7171 *pzWhere = zWhere;
7175 ** Implementation of .ar "lisT" command.
7177 static int arListCommand(ArCommand *pAr){
7178 const char *zSql = "SELECT %s FROM %s WHERE %s";
7179 const char *azCols[] = {
7180 "name",
7181 "lsmode(mode), sz, datetime(mtime, 'unixepoch'), name"
7184 char *zWhere = 0;
7185 sqlite3_stmt *pSql = 0;
7186 int rc;
7188 rc = arCheckEntries(pAr);
7189 arWhereClause(&rc, pAr, &zWhere);
7191 shellPreparePrintf(pAr->db, &rc, &pSql, zSql, azCols[pAr->bVerbose],
7192 pAr->zSrcTable, zWhere);
7193 if( pAr->bDryRun ){
7194 oputf("%s\n", sqlite3_sql(pSql));
7195 }else{
7196 while( rc==SQLITE_OK && SQLITE_ROW==sqlite3_step(pSql) ){
7197 if( pAr->bVerbose ){
7198 oputf("%s % 10d %s %s\n",
7199 sqlite3_column_text(pSql, 0), sqlite3_column_int(pSql, 1),
7200 sqlite3_column_text(pSql, 2),sqlite3_column_text(pSql, 3));
7201 }else{
7202 oputf("%s\n", sqlite3_column_text(pSql, 0));
7206 shellFinalize(&rc, pSql);
7207 sqlite3_free(zWhere);
7208 return rc;
7212 ** Implementation of .ar "Remove" command.
7214 static int arRemoveCommand(ArCommand *pAr){
7215 int rc = 0;
7216 char *zSql = 0;
7217 char *zWhere = 0;
7219 if( pAr->nArg ){
7220 /* Verify that args actually exist within the archive before proceeding.
7221 ** And formulate a WHERE clause to match them. */
7222 rc = arCheckEntries(pAr);
7223 arWhereClause(&rc, pAr, &zWhere);
7225 if( rc==SQLITE_OK ){
7226 zSql = sqlite3_mprintf("DELETE FROM %s WHERE %s;",
7227 pAr->zSrcTable, zWhere);
7228 if( pAr->bDryRun ){
7229 oputf("%s\n", zSql);
7230 }else{
7231 char *zErr = 0;
7232 rc = sqlite3_exec(pAr->db, "SAVEPOINT ar;", 0, 0, 0);
7233 if( rc==SQLITE_OK ){
7234 rc = sqlite3_exec(pAr->db, zSql, 0, 0, &zErr);
7235 if( rc!=SQLITE_OK ){
7236 sqlite3_exec(pAr->db, "ROLLBACK TO ar; RELEASE ar;", 0, 0, 0);
7237 }else{
7238 rc = sqlite3_exec(pAr->db, "RELEASE ar;", 0, 0, 0);
7241 if( zErr ){
7242 sputf(stdout, "ERROR: %s\n", zErr); /* stdout? */
7243 sqlite3_free(zErr);
7247 sqlite3_free(zWhere);
7248 sqlite3_free(zSql);
7249 return rc;
7253 ** Implementation of .ar "eXtract" command.
7255 static int arExtractCommand(ArCommand *pAr){
7256 const char *zSql1 =
7257 "SELECT "
7258 " ($dir || name),"
7259 " writefile(($dir || name), %s, mode, mtime) "
7260 "FROM %s WHERE (%s) AND (data IS NULL OR $dirOnly = 0)"
7261 " AND name NOT GLOB '*..[/\\]*'";
7263 const char *azExtraArg[] = {
7264 "sqlar_uncompress(data, sz)",
7265 "data"
7268 sqlite3_stmt *pSql = 0;
7269 int rc = SQLITE_OK;
7270 char *zDir = 0;
7271 char *zWhere = 0;
7272 int i, j;
7274 /* If arguments are specified, check that they actually exist within
7275 ** the archive before proceeding. And formulate a WHERE clause to
7276 ** match them. */
7277 rc = arCheckEntries(pAr);
7278 arWhereClause(&rc, pAr, &zWhere);
7280 if( rc==SQLITE_OK ){
7281 if( pAr->zDir ){
7282 zDir = sqlite3_mprintf("%s/", pAr->zDir);
7283 }else{
7284 zDir = sqlite3_mprintf("");
7286 if( zDir==0 ) rc = SQLITE_NOMEM;
7289 shellPreparePrintf(pAr->db, &rc, &pSql, zSql1,
7290 azExtraArg[pAr->bZip], pAr->zSrcTable, zWhere
7293 if( rc==SQLITE_OK ){
7294 j = sqlite3_bind_parameter_index(pSql, "$dir");
7295 sqlite3_bind_text(pSql, j, zDir, -1, SQLITE_STATIC);
7297 /* Run the SELECT statement twice. The first time, writefile() is called
7298 ** for all archive members that should be extracted. The second time,
7299 ** only for the directories. This is because the timestamps for
7300 ** extracted directories must be reset after they are populated (as
7301 ** populating them changes the timestamp). */
7302 for(i=0; i<2; i++){
7303 j = sqlite3_bind_parameter_index(pSql, "$dirOnly");
7304 sqlite3_bind_int(pSql, j, i);
7305 if( pAr->bDryRun ){
7306 oputf("%s\n", sqlite3_sql(pSql));
7307 }else{
7308 while( rc==SQLITE_OK && SQLITE_ROW==sqlite3_step(pSql) ){
7309 if( i==0 && pAr->bVerbose ){
7310 oputf("%s\n", sqlite3_column_text(pSql, 0));
7314 shellReset(&rc, pSql);
7316 shellFinalize(&rc, pSql);
7319 sqlite3_free(zDir);
7320 sqlite3_free(zWhere);
7321 return rc;
7325 ** Run the SQL statement in zSql. Or if doing a --dryrun, merely print it out.
7327 static int arExecSql(ArCommand *pAr, const char *zSql){
7328 int rc;
7329 if( pAr->bDryRun ){
7330 oputf("%s\n", zSql);
7331 rc = SQLITE_OK;
7332 }else{
7333 char *zErr = 0;
7334 rc = sqlite3_exec(pAr->db, zSql, 0, 0, &zErr);
7335 if( zErr ){
7336 sputf(stdout, "ERROR: %s\n", zErr);
7337 sqlite3_free(zErr);
7340 return rc;
7345 ** Implementation of .ar "create", "insert", and "update" commands.
7347 ** create -> Create a new SQL archive
7348 ** insert -> Insert or reinsert all files listed
7349 ** update -> Insert files that have changed or that were not
7350 ** previously in the archive
7352 ** Create the "sqlar" table in the database if it does not already exist.
7353 ** Then add each file in the azFile[] array to the archive. Directories
7354 ** are added recursively. If argument bVerbose is non-zero, a message is
7355 ** printed on stdout for each file archived.
7357 ** The create command is the same as update, except that it drops
7358 ** any existing "sqlar" table before beginning. The "insert" command
7359 ** always overwrites every file named on the command-line, where as
7360 ** "update" only overwrites if the size or mtime or mode has changed.
7362 static int arCreateOrUpdateCommand(
7363 ArCommand *pAr, /* Command arguments and options */
7364 int bUpdate, /* true for a --create. */
7365 int bOnlyIfChanged /* Only update if file has changed */
7367 const char *zCreate =
7368 "CREATE TABLE IF NOT EXISTS sqlar(\n"
7369 " name TEXT PRIMARY KEY, -- name of the file\n"
7370 " mode INT, -- access permissions\n"
7371 " mtime INT, -- last modification time\n"
7372 " sz INT, -- original file size\n"
7373 " data BLOB -- compressed content\n"
7374 ")";
7375 const char *zDrop = "DROP TABLE IF EXISTS sqlar";
7376 const char *zInsertFmt[2] = {
7377 "REPLACE INTO %s(name,mode,mtime,sz,data)\n"
7378 " SELECT\n"
7379 " %s,\n"
7380 " mode,\n"
7381 " mtime,\n"
7382 " CASE substr(lsmode(mode),1,1)\n"
7383 " WHEN '-' THEN length(data)\n"
7384 " WHEN 'd' THEN 0\n"
7385 " ELSE -1 END,\n"
7386 " sqlar_compress(data)\n"
7387 " FROM fsdir(%Q,%Q) AS disk\n"
7388 " WHERE lsmode(mode) NOT LIKE '?%%'%s;"
7390 "REPLACE INTO %s(name,mode,mtime,data)\n"
7391 " SELECT\n"
7392 " %s,\n"
7393 " mode,\n"
7394 " mtime,\n"
7395 " data\n"
7396 " FROM fsdir(%Q,%Q) AS disk\n"
7397 " WHERE lsmode(mode) NOT LIKE '?%%'%s;"
7399 int i; /* For iterating through azFile[] */
7400 int rc; /* Return code */
7401 const char *zTab = 0; /* SQL table into which to insert */
7402 char *zSql;
7403 char zTemp[50];
7404 char *zExists = 0;
7406 arExecSql(pAr, "PRAGMA page_size=512");
7407 rc = arExecSql(pAr, "SAVEPOINT ar;");
7408 if( rc!=SQLITE_OK ) return rc;
7409 zTemp[0] = 0;
7410 if( pAr->bZip ){
7411 /* Initialize the zipfile virtual table, if necessary */
7412 if( pAr->zFile ){
7413 sqlite3_uint64 r;
7414 sqlite3_randomness(sizeof(r),&r);
7415 sqlite3_snprintf(sizeof(zTemp),zTemp,"zip%016llx",r);
7416 zTab = zTemp;
7417 zSql = sqlite3_mprintf(
7418 "CREATE VIRTUAL TABLE temp.%s USING zipfile(%Q)",
7419 zTab, pAr->zFile
7421 rc = arExecSql(pAr, zSql);
7422 sqlite3_free(zSql);
7423 }else{
7424 zTab = "zip";
7426 }else{
7427 /* Initialize the table for an SQLAR */
7428 zTab = "sqlar";
7429 if( bUpdate==0 ){
7430 rc = arExecSql(pAr, zDrop);
7431 if( rc!=SQLITE_OK ) goto end_ar_transaction;
7433 rc = arExecSql(pAr, zCreate);
7435 if( bOnlyIfChanged ){
7436 zExists = sqlite3_mprintf(
7437 " AND NOT EXISTS("
7438 "SELECT 1 FROM %s AS mem"
7439 " WHERE mem.name=disk.name"
7440 " AND mem.mtime=disk.mtime"
7441 " AND mem.mode=disk.mode)", zTab);
7442 }else{
7443 zExists = sqlite3_mprintf("");
7445 if( zExists==0 ) rc = SQLITE_NOMEM;
7446 for(i=0; i<pAr->nArg && rc==SQLITE_OK; i++){
7447 char *zSql2 = sqlite3_mprintf(zInsertFmt[pAr->bZip], zTab,
7448 pAr->bVerbose ? "shell_putsnl(name)" : "name",
7449 pAr->azArg[i], pAr->zDir, zExists);
7450 rc = arExecSql(pAr, zSql2);
7451 sqlite3_free(zSql2);
7453 end_ar_transaction:
7454 if( rc!=SQLITE_OK ){
7455 sqlite3_exec(pAr->db, "ROLLBACK TO ar; RELEASE ar;", 0, 0, 0);
7456 }else{
7457 rc = arExecSql(pAr, "RELEASE ar;");
7458 if( pAr->bZip && pAr->zFile ){
7459 zSql = sqlite3_mprintf("DROP TABLE %s", zTemp);
7460 arExecSql(pAr, zSql);
7461 sqlite3_free(zSql);
7464 sqlite3_free(zExists);
7465 return rc;
7469 ** Implementation of ".ar" dot command.
7471 static int arDotCommand(
7472 ShellState *pState, /* Current shell tool state */
7473 int fromCmdLine, /* True if -A command-line option, not .ar cmd */
7474 char **azArg, /* Array of arguments passed to dot command */
7475 int nArg /* Number of entries in azArg[] */
7477 ArCommand cmd;
7478 int rc;
7479 memset(&cmd, 0, sizeof(cmd));
7480 cmd.fromCmdLine = fromCmdLine;
7481 rc = arParseCommand(azArg, nArg, &cmd);
7482 if( rc==SQLITE_OK ){
7483 int eDbType = SHELL_OPEN_UNSPEC;
7484 cmd.p = pState;
7485 cmd.db = pState->db;
7486 if( cmd.zFile ){
7487 eDbType = deduceDatabaseType(cmd.zFile, 1);
7488 }else{
7489 eDbType = pState->openMode;
7491 if( eDbType==SHELL_OPEN_ZIPFILE ){
7492 if( cmd.eCmd==AR_CMD_EXTRACT || cmd.eCmd==AR_CMD_LIST ){
7493 if( cmd.zFile==0 ){
7494 cmd.zSrcTable = sqlite3_mprintf("zip");
7495 }else{
7496 cmd.zSrcTable = sqlite3_mprintf("zipfile(%Q)", cmd.zFile);
7499 cmd.bZip = 1;
7500 }else if( cmd.zFile ){
7501 int flags;
7502 if( cmd.bAppend ) eDbType = SHELL_OPEN_APPENDVFS;
7503 if( cmd.eCmd==AR_CMD_CREATE || cmd.eCmd==AR_CMD_INSERT
7504 || cmd.eCmd==AR_CMD_REMOVE || cmd.eCmd==AR_CMD_UPDATE ){
7505 flags = SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE;
7506 }else{
7507 flags = SQLITE_OPEN_READONLY;
7509 cmd.db = 0;
7510 if( cmd.bDryRun ){
7511 oputf("-- open database '%s'%s\n", cmd.zFile,
7512 eDbType==SHELL_OPEN_APPENDVFS ? " using 'apndvfs'" : "");
7514 rc = sqlite3_open_v2(cmd.zFile, &cmd.db, flags,
7515 eDbType==SHELL_OPEN_APPENDVFS ? "apndvfs" : 0);
7516 if( rc!=SQLITE_OK ){
7517 eputf("cannot open file: %s (%s)\n", cmd.zFile, sqlite3_errmsg(cmd.db));
7518 goto end_ar_command;
7520 sqlite3_fileio_init(cmd.db, 0, 0);
7521 sqlite3_sqlar_init(cmd.db, 0, 0);
7522 sqlite3_create_function(cmd.db, "shell_putsnl", 1, SQLITE_UTF8, cmd.p,
7523 shellPutsFunc, 0, 0);
7526 if( cmd.zSrcTable==0 && cmd.bZip==0 && cmd.eCmd!=AR_CMD_HELP ){
7527 if( cmd.eCmd!=AR_CMD_CREATE
7528 && sqlite3_table_column_metadata(cmd.db,0,"sqlar","name",0,0,0,0,0)
7530 eputz("database does not contain an 'sqlar' table\n");
7531 rc = SQLITE_ERROR;
7532 goto end_ar_command;
7534 cmd.zSrcTable = sqlite3_mprintf("sqlar");
7537 switch( cmd.eCmd ){
7538 case AR_CMD_CREATE:
7539 rc = arCreateOrUpdateCommand(&cmd, 0, 0);
7540 break;
7542 case AR_CMD_EXTRACT:
7543 rc = arExtractCommand(&cmd);
7544 break;
7546 case AR_CMD_LIST:
7547 rc = arListCommand(&cmd);
7548 break;
7550 case AR_CMD_HELP:
7551 arUsage(pState->out);
7552 break;
7554 case AR_CMD_INSERT:
7555 rc = arCreateOrUpdateCommand(&cmd, 1, 0);
7556 break;
7558 case AR_CMD_REMOVE:
7559 rc = arRemoveCommand(&cmd);
7560 break;
7562 default:
7563 assert( cmd.eCmd==AR_CMD_UPDATE );
7564 rc = arCreateOrUpdateCommand(&cmd, 1, 1);
7565 break;
7568 end_ar_command:
7569 if( cmd.db!=pState->db ){
7570 close_db(cmd.db);
7572 sqlite3_free(cmd.zSrcTable);
7574 return rc;
7576 /* End of the ".archive" or ".ar" command logic
7577 *******************************************************************************/
7578 #endif /* !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_HAVE_ZLIB) */
7580 #if SQLITE_SHELL_HAVE_RECOVER
7583 ** This function is used as a callback by the recover extension. Simply
7584 ** print the supplied SQL statement to stdout.
7586 static int recoverSqlCb(void *pCtx, const char *zSql){
7587 ShellState *pState = (ShellState*)pCtx;
7588 sputf(pState->out, "%s;\n", zSql);
7589 return SQLITE_OK;
7593 ** This function is called to recover data from the database. A script
7594 ** to construct a new database containing all recovered data is output
7595 ** on stream pState->out.
7597 static int recoverDatabaseCmd(ShellState *pState, int nArg, char **azArg){
7598 int rc = SQLITE_OK;
7599 const char *zRecoveryDb = ""; /* Name of "recovery" database. Debug only */
7600 const char *zLAF = "lost_and_found";
7601 int bFreelist = 1; /* 0 if --ignore-freelist is specified */
7602 int bRowids = 1; /* 0 if --no-rowids */
7603 sqlite3_recover *p = 0;
7604 int i = 0;
7606 for(i=1; i<nArg; i++){
7607 char *z = azArg[i];
7608 int n;
7609 if( z[0]=='-' && z[1]=='-' ) z++;
7610 n = strlen30(z);
7611 if( n<=17 && memcmp("-ignore-freelist", z, n)==0 ){
7612 bFreelist = 0;
7613 }else
7614 if( n<=12 && memcmp("-recovery-db", z, n)==0 && i<(nArg-1) ){
7615 /* This option determines the name of the ATTACH-ed database used
7616 ** internally by the recovery extension. The default is "" which
7617 ** means to use a temporary database that is automatically deleted
7618 ** when closed. This option is undocumented and might disappear at
7619 ** any moment. */
7620 i++;
7621 zRecoveryDb = azArg[i];
7622 }else
7623 if( n<=15 && memcmp("-lost-and-found", z, n)==0 && i<(nArg-1) ){
7624 i++;
7625 zLAF = azArg[i];
7626 }else
7627 if( n<=10 && memcmp("-no-rowids", z, n)==0 ){
7628 bRowids = 0;
7630 else{
7631 eputf("unexpected option: %s\n", azArg[i]);
7632 showHelp(pState->out, azArg[0]);
7633 return 1;
7637 p = sqlite3_recover_init_sql(
7638 pState->db, "main", recoverSqlCb, (void*)pState
7641 sqlite3_recover_config(p, 789, (void*)zRecoveryDb); /* Debug use only */
7642 sqlite3_recover_config(p, SQLITE_RECOVER_LOST_AND_FOUND, (void*)zLAF);
7643 sqlite3_recover_config(p, SQLITE_RECOVER_ROWIDS, (void*)&bRowids);
7644 sqlite3_recover_config(p, SQLITE_RECOVER_FREELIST_CORRUPT,(void*)&bFreelist);
7646 sqlite3_recover_run(p);
7647 if( sqlite3_recover_errcode(p)!=SQLITE_OK ){
7648 const char *zErr = sqlite3_recover_errmsg(p);
7649 int errCode = sqlite3_recover_errcode(p);
7650 eputf("sql error: %s (%d)\n", zErr, errCode);
7652 rc = sqlite3_recover_finish(p);
7653 return rc;
7655 #endif /* SQLITE_SHELL_HAVE_RECOVER */
7658 ** Implementation of ".intck STEPS_PER_UNLOCK" command.
7660 static int intckDatabaseCmd(ShellState *pState, i64 nStepPerUnlock){
7661 sqlite3_intck *p = 0;
7662 int rc = SQLITE_OK;
7664 rc = sqlite3_intck_open(pState->db, "main", &p);
7665 if( rc==SQLITE_OK ){
7666 i64 nStep = 0;
7667 i64 nError = 0;
7668 const char *zErr = 0;
7669 while( SQLITE_OK==sqlite3_intck_step(p) ){
7670 const char *zMsg = sqlite3_intck_message(p);
7671 if( zMsg ){
7672 oputf("%s\n", zMsg);
7673 nError++;
7675 nStep++;
7676 if( nStepPerUnlock && (nStep % nStepPerUnlock)==0 ){
7677 sqlite3_intck_unlock(p);
7680 rc = sqlite3_intck_error(p, &zErr);
7681 if( zErr ){
7682 eputf("%s\n", zErr);
7684 sqlite3_intck_close(p);
7686 oputf("%lld steps, %lld errors\n", nStep, nError);
7689 return rc;
7693 * zAutoColumn(zCol, &db, ?) => Maybe init db, add column zCol to it.
7694 * zAutoColumn(0, &db, ?) => (db!=0) Form columns spec for CREATE TABLE,
7695 * close db and set it to 0, and return the columns spec, to later
7696 * be sqlite3_free()'ed by the caller.
7697 * The return is 0 when either:
7698 * (a) The db was not initialized and zCol==0 (There are no columns.)
7699 * (b) zCol!=0 (Column was added, db initialized as needed.)
7700 * The 3rd argument, pRenamed, references an out parameter. If the
7701 * pointer is non-zero, its referent will be set to a summary of renames
7702 * done if renaming was necessary, or set to 0 if none was done. The out
7703 * string (if any) must be sqlite3_free()'ed by the caller.
7705 #ifdef SHELL_DEBUG
7706 #define rc_err_oom_die(rc) \
7707 if( rc==SQLITE_NOMEM ) shell_check_oom(0); \
7708 else if(!(rc==SQLITE_OK||rc==SQLITE_DONE)) \
7709 eputf("E:%d\n",rc), assert(0)
7710 #else
7711 static void rc_err_oom_die(int rc){
7712 if( rc==SQLITE_NOMEM ) shell_check_oom(0);
7713 assert(rc==SQLITE_OK||rc==SQLITE_DONE);
7715 #endif
7717 #ifdef SHELL_COLFIX_DB /* If this is set, the DB can be in a file. */
7718 static char zCOL_DB[] = SHELL_STRINGIFY(SHELL_COLFIX_DB);
7719 #else /* Otherwise, memory is faster/better for the transient DB. */
7720 static const char *zCOL_DB = ":memory:";
7721 #endif
7723 /* Define character (as C string) to separate generated column ordinal
7724 * from protected part of incoming column names. This defaults to "_"
7725 * so that incoming column identifiers that did not need not be quoted
7726 * remain usable without being quoted. It must be one character.
7728 #ifndef SHELL_AUTOCOLUMN_SEP
7729 # define AUTOCOLUMN_SEP "_"
7730 #else
7731 # define AUTOCOLUMN_SEP SHELL_STRINGIFY(SHELL_AUTOCOLUMN_SEP)
7732 #endif
7734 static char *zAutoColumn(const char *zColNew, sqlite3 **pDb, char **pzRenamed){
7735 /* Queries and D{D,M}L used here */
7736 static const char * const zTabMake = "\
7737 CREATE TABLE ColNames(\
7738 cpos INTEGER PRIMARY KEY,\
7739 name TEXT, nlen INT, chop INT, reps INT, suff TEXT);\
7740 CREATE VIEW RepeatedNames AS \
7741 SELECT DISTINCT t.name FROM ColNames t \
7742 WHERE t.name COLLATE NOCASE IN (\
7743 SELECT o.name FROM ColNames o WHERE o.cpos<>t.cpos\
7746 static const char * const zTabFill = "\
7747 INSERT INTO ColNames(name,nlen,chop,reps,suff)\
7748 VALUES(iif(length(?1)>0,?1,'?'),max(length(?1),1),0,0,'')\
7750 static const char * const zHasDupes = "\
7751 SELECT count(DISTINCT (substring(name,1,nlen-chop)||suff) COLLATE NOCASE)\
7752 <count(name) FROM ColNames\
7754 #ifdef SHELL_COLUMN_RENAME_CLEAN
7755 static const char * const zDedoctor = "\
7756 UPDATE ColNames SET chop=iif(\
7757 (substring(name,nlen,1) BETWEEN '0' AND '9')\
7758 AND (rtrim(name,'0123456790') glob '*"AUTOCOLUMN_SEP"'),\
7759 nlen-length(rtrim(name, '"AUTOCOLUMN_SEP"0123456789')),\
7763 #endif
7764 static const char * const zSetReps = "\
7765 UPDATE ColNames AS t SET reps=\
7766 (SELECT count(*) FROM ColNames d \
7767 WHERE substring(t.name,1,t.nlen-t.chop)=substring(d.name,1,d.nlen-d.chop)\
7768 COLLATE NOCASE\
7771 #ifdef SQLITE_ENABLE_MATH_FUNCTIONS
7772 static const char * const zColDigits = "\
7773 SELECT CAST(ceil(log(count(*)+0.5)) AS INT) FROM ColNames \
7775 #else
7776 /* Counting on SQLITE_MAX_COLUMN < 100,000 here. (32767 is the hard limit.) */
7777 static const char * const zColDigits = "\
7778 SELECT CASE WHEN (nc < 10) THEN 1 WHEN (nc < 100) THEN 2 \
7779 WHEN (nc < 1000) THEN 3 WHEN (nc < 10000) THEN 4 \
7780 ELSE 5 FROM (SELECT count(*) AS nc FROM ColNames) \
7782 #endif
7783 static const char * const zRenameRank =
7784 #ifdef SHELL_COLUMN_RENAME_CLEAN
7785 "UPDATE ColNames AS t SET suff="
7786 "iif(reps>1, printf('%c%0*d', '"AUTOCOLUMN_SEP"', $1, cpos), '')"
7787 #else /* ...RENAME_MINIMAL_ONE_PASS */
7788 "WITH Lzn(nlz) AS (" /* Find minimum extraneous leading 0's for uniqueness */
7789 " SELECT 0 AS nlz"
7790 " UNION"
7791 " SELECT nlz+1 AS nlz FROM Lzn"
7792 " WHERE EXISTS("
7793 " SELECT 1"
7794 " FROM ColNames t, ColNames o"
7795 " WHERE"
7796 " iif(t.name IN (SELECT * FROM RepeatedNames),"
7797 " printf('%s"AUTOCOLUMN_SEP"%s',"
7798 " t.name, substring(printf('%.*c%0.*d',nlz+1,'0',$1,t.cpos),2)),"
7799 " t.name"
7800 " )"
7801 " ="
7802 " iif(o.name IN (SELECT * FROM RepeatedNames),"
7803 " printf('%s"AUTOCOLUMN_SEP"%s',"
7804 " o.name, substring(printf('%.*c%0.*d',nlz+1,'0',$1,o.cpos),2)),"
7805 " o.name"
7806 " )"
7807 " COLLATE NOCASE"
7808 " AND o.cpos<>t.cpos"
7809 " GROUP BY t.cpos"
7810 " )"
7811 ") UPDATE Colnames AS t SET"
7812 " chop = 0," /* No chopping, never touch incoming names. */
7813 " suff = iif(name IN (SELECT * FROM RepeatedNames),"
7814 " printf('"AUTOCOLUMN_SEP"%s', substring("
7815 " printf('%.*c%0.*d',(SELECT max(nlz) FROM Lzn)+1,'0',1,t.cpos),2)),"
7816 " ''"
7817 " )"
7818 #endif
7820 static const char * const zCollectVar = "\
7821 SELECT\
7822 '('||x'0a'\
7823 || group_concat(\
7824 cname||' TEXT',\
7825 ','||iif((cpos-1)%4>0, ' ', x'0a'||' '))\
7826 ||')' AS ColsSpec \
7827 FROM (\
7828 SELECT cpos, printf('\"%w\"',printf('%!.*s%s', nlen-chop,name,suff)) AS cname \
7829 FROM ColNames ORDER BY cpos\
7831 static const char * const zRenamesDone =
7832 "SELECT group_concat("
7833 " printf('\"%w\" to \"%w\"',name,printf('%!.*s%s', nlen-chop, name, suff)),"
7834 " ','||x'0a')"
7835 "FROM ColNames WHERE suff<>'' OR chop!=0"
7837 int rc;
7838 sqlite3_stmt *pStmt = 0;
7839 assert(pDb!=0);
7840 if( zColNew ){
7841 /* Add initial or additional column. Init db if necessary. */
7842 if( *pDb==0 ){
7843 if( SQLITE_OK!=sqlite3_open(zCOL_DB, pDb) ) return 0;
7844 #ifdef SHELL_COLFIX_DB
7845 if(*zCOL_DB!=':')
7846 sqlite3_exec(*pDb,"drop table if exists ColNames;"
7847 "drop view if exists RepeatedNames;",0,0,0);
7848 #endif
7849 #undef SHELL_COLFIX_DB
7850 rc = sqlite3_exec(*pDb, zTabMake, 0, 0, 0);
7851 rc_err_oom_die(rc);
7853 assert(*pDb!=0);
7854 rc = sqlite3_prepare_v2(*pDb, zTabFill, -1, &pStmt, 0);
7855 rc_err_oom_die(rc);
7856 rc = sqlite3_bind_text(pStmt, 1, zColNew, -1, 0);
7857 rc_err_oom_die(rc);
7858 rc = sqlite3_step(pStmt);
7859 rc_err_oom_die(rc);
7860 sqlite3_finalize(pStmt);
7861 return 0;
7862 }else if( *pDb==0 ){
7863 return 0;
7864 }else{
7865 /* Formulate the columns spec, close the DB, zero *pDb. */
7866 char *zColsSpec = 0;
7867 int hasDupes = db_int(*pDb, zHasDupes);
7868 int nDigits = (hasDupes)? db_int(*pDb, zColDigits) : 0;
7869 if( hasDupes ){
7870 #ifdef SHELL_COLUMN_RENAME_CLEAN
7871 rc = sqlite3_exec(*pDb, zDedoctor, 0, 0, 0);
7872 rc_err_oom_die(rc);
7873 #endif
7874 rc = sqlite3_exec(*pDb, zSetReps, 0, 0, 0);
7875 rc_err_oom_die(rc);
7876 rc = sqlite3_prepare_v2(*pDb, zRenameRank, -1, &pStmt, 0);
7877 rc_err_oom_die(rc);
7878 sqlite3_bind_int(pStmt, 1, nDigits);
7879 rc = sqlite3_step(pStmt);
7880 sqlite3_finalize(pStmt);
7881 if( rc!=SQLITE_DONE ) rc_err_oom_die(SQLITE_NOMEM);
7883 assert(db_int(*pDb, zHasDupes)==0); /* Consider: remove this */
7884 rc = sqlite3_prepare_v2(*pDb, zCollectVar, -1, &pStmt, 0);
7885 rc_err_oom_die(rc);
7886 rc = sqlite3_step(pStmt);
7887 if( rc==SQLITE_ROW ){
7888 zColsSpec = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
7889 }else{
7890 zColsSpec = 0;
7892 if( pzRenamed!=0 ){
7893 if( !hasDupes ) *pzRenamed = 0;
7894 else{
7895 sqlite3_finalize(pStmt);
7896 if( SQLITE_OK==sqlite3_prepare_v2(*pDb, zRenamesDone, -1, &pStmt, 0)
7897 && SQLITE_ROW==sqlite3_step(pStmt) ){
7898 *pzRenamed = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
7899 }else
7900 *pzRenamed = 0;
7903 sqlite3_finalize(pStmt);
7904 sqlite3_close(*pDb);
7905 *pDb = 0;
7906 return zColsSpec;
7911 ** Check if the sqlite_schema table contains one or more virtual tables. If
7912 ** parameter zLike is not NULL, then it is an SQL expression that the
7913 ** sqlite_schema row must also match. If one or more such rows are found,
7914 ** print the following warning to the output:
7916 ** WARNING: Script requires that SQLITE_DBCONFIG_DEFENSIVE be disabled
7918 static int outputDumpWarning(ShellState *p, const char *zLike){
7919 int rc = SQLITE_OK;
7920 sqlite3_stmt *pStmt = 0;
7921 shellPreparePrintf(p->db, &rc, &pStmt,
7922 "SELECT 1 FROM sqlite_schema o WHERE "
7923 "sql LIKE 'CREATE VIRTUAL TABLE%%' AND %s", zLike ? zLike : "true"
7925 if( rc==SQLITE_OK && sqlite3_step(pStmt)==SQLITE_ROW ){
7926 oputz("/* WARNING: "
7927 "Script requires that SQLITE_DBCONFIG_DEFENSIVE be disabled */\n"
7930 shellFinalize(&rc, pStmt);
7931 return rc;
7935 ** Fault-Simulator state and logic.
7937 static struct {
7938 int iId; /* ID that triggers a simulated fault. -1 means "any" */
7939 int iErr; /* The error code to return on a fault */
7940 int iCnt; /* Trigger the fault only if iCnt is already zero */
7941 int iInterval; /* Reset iCnt to this value after each fault */
7942 int eVerbose; /* When to print output */
7943 int nHit; /* Number of hits seen so far */
7944 int nRepeat; /* Turn off after this many hits. 0 for never */
7945 int nSkip; /* Skip this many before first fault */
7946 } faultsim_state = {-1, 0, 0, 0, 0, 0, 0};
7949 ** This is the fault-sim callback
7951 static int faultsim_callback(int iArg){
7952 if( faultsim_state.iId>0 && faultsim_state.iId!=iArg ){
7953 return SQLITE_OK;
7955 if( faultsim_state.iCnt ){
7956 if( faultsim_state.iCnt>0 ) faultsim_state.iCnt--;
7957 if( faultsim_state.eVerbose>=2 ){
7958 oputf("FAULT-SIM id=%d no-fault (cnt=%d)\n", iArg, faultsim_state.iCnt);
7960 return SQLITE_OK;
7962 if( faultsim_state.eVerbose>=1 ){
7963 oputf("FAULT-SIM id=%d returns %d\n", iArg, faultsim_state.iErr);
7965 faultsim_state.iCnt = faultsim_state.iInterval;
7966 faultsim_state.nHit++;
7967 if( faultsim_state.nRepeat>0 && faultsim_state.nRepeat<=faultsim_state.nHit ){
7968 faultsim_state.iCnt = -1;
7970 return faultsim_state.iErr;
7974 ** If an input line begins with "." then invoke this routine to
7975 ** process that line.
7977 ** Return 1 on error, 2 to exit, and 0 otherwise.
7979 static int do_meta_command(char *zLine, ShellState *p){
7980 int h = 1;
7981 int nArg = 0;
7982 int n, c;
7983 int rc = 0;
7984 char *azArg[52];
7986 #ifndef SQLITE_OMIT_VIRTUALTABLE
7987 if( p->expert.pExpert ){
7988 expertFinish(p, 1, 0);
7990 #endif
7992 /* Parse the input line into tokens.
7994 while( zLine[h] && nArg<ArraySize(azArg)-1 ){
7995 while( IsSpace(zLine[h]) ){ h++; }
7996 if( zLine[h]==0 ) break;
7997 if( zLine[h]=='\'' || zLine[h]=='"' ){
7998 int delim = zLine[h++];
7999 azArg[nArg++] = &zLine[h];
8000 while( zLine[h] && zLine[h]!=delim ){
8001 if( zLine[h]=='\\' && delim=='"' && zLine[h+1]!=0 ) h++;
8002 h++;
8004 if( zLine[h]==delim ){
8005 zLine[h++] = 0;
8007 if( delim=='"' ) resolve_backslashes(azArg[nArg-1]);
8008 }else{
8009 azArg[nArg++] = &zLine[h];
8010 while( zLine[h] && !IsSpace(zLine[h]) ){ h++; }
8011 if( zLine[h] ) zLine[h++] = 0;
8014 azArg[nArg] = 0;
8016 /* Process the input line.
8018 if( nArg==0 ) return 0; /* no tokens, no error */
8019 n = strlen30(azArg[0]);
8020 c = azArg[0][0];
8021 clearTempFile(p);
8023 #ifndef SQLITE_OMIT_AUTHORIZATION
8024 if( c=='a' && cli_strncmp(azArg[0], "auth", n)==0 ){
8025 if( nArg!=2 ){
8026 eputz("Usage: .auth ON|OFF\n");
8027 rc = 1;
8028 goto meta_command_exit;
8030 open_db(p, 0);
8031 if( booleanValue(azArg[1]) ){
8032 sqlite3_set_authorizer(p->db, shellAuth, p);
8033 }else if( p->bSafeModePersist ){
8034 sqlite3_set_authorizer(p->db, safeModeAuth, p);
8035 }else{
8036 sqlite3_set_authorizer(p->db, 0, 0);
8038 }else
8039 #endif
8041 #if !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_HAVE_ZLIB) \
8042 && !defined(SQLITE_SHELL_FIDDLE)
8043 if( c=='a' && cli_strncmp(azArg[0], "archive", n)==0 ){
8044 open_db(p, 0);
8045 failIfSafeMode(p, "cannot run .archive in safe mode");
8046 rc = arDotCommand(p, 0, azArg, nArg);
8047 }else
8048 #endif
8050 #ifndef SQLITE_SHELL_FIDDLE
8051 if( (c=='b' && n>=3 && cli_strncmp(azArg[0], "backup", n)==0)
8052 || (c=='s' && n>=3 && cli_strncmp(azArg[0], "save", n)==0)
8054 const char *zDestFile = 0;
8055 const char *zDb = 0;
8056 sqlite3 *pDest;
8057 sqlite3_backup *pBackup;
8058 int j;
8059 int bAsync = 0;
8060 const char *zVfs = 0;
8061 failIfSafeMode(p, "cannot run .%s in safe mode", azArg[0]);
8062 for(j=1; j<nArg; j++){
8063 const char *z = azArg[j];
8064 if( z[0]=='-' ){
8065 if( z[1]=='-' ) z++;
8066 if( cli_strcmp(z, "-append")==0 ){
8067 zVfs = "apndvfs";
8068 }else
8069 if( cli_strcmp(z, "-async")==0 ){
8070 bAsync = 1;
8071 }else
8073 eputf("unknown option: %s\n", azArg[j]);
8074 return 1;
8076 }else if( zDestFile==0 ){
8077 zDestFile = azArg[j];
8078 }else if( zDb==0 ){
8079 zDb = zDestFile;
8080 zDestFile = azArg[j];
8081 }else{
8082 eputz("Usage: .backup ?DB? ?OPTIONS? FILENAME\n");
8083 return 1;
8086 if( zDestFile==0 ){
8087 eputz("missing FILENAME argument on .backup\n");
8088 return 1;
8090 if( zDb==0 ) zDb = "main";
8091 rc = sqlite3_open_v2(zDestFile, &pDest,
8092 SQLITE_OPEN_READWRITE|SQLITE_OPEN_CREATE, zVfs);
8093 if( rc!=SQLITE_OK ){
8094 eputf("Error: cannot open \"%s\"\n", zDestFile);
8095 close_db(pDest);
8096 return 1;
8098 if( bAsync ){
8099 sqlite3_exec(pDest, "PRAGMA synchronous=OFF; PRAGMA journal_mode=OFF;",
8100 0, 0, 0);
8102 open_db(p, 0);
8103 pBackup = sqlite3_backup_init(pDest, "main", p->db, zDb);
8104 if( pBackup==0 ){
8105 eputf("Error: %s\n", sqlite3_errmsg(pDest));
8106 close_db(pDest);
8107 return 1;
8109 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK ){}
8110 sqlite3_backup_finish(pBackup);
8111 if( rc==SQLITE_DONE ){
8112 rc = 0;
8113 }else{
8114 eputf("Error: %s\n", sqlite3_errmsg(pDest));
8115 rc = 1;
8117 close_db(pDest);
8118 }else
8119 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
8121 if( c=='b' && n>=3 && cli_strncmp(azArg[0], "bail", n)==0 ){
8122 if( nArg==2 ){
8123 bail_on_error = booleanValue(azArg[1]);
8124 }else{
8125 eputz("Usage: .bail on|off\n");
8126 rc = 1;
8128 }else
8130 /* Undocumented. Legacy only. See "crnl" below */
8131 if( c=='b' && n>=3 && cli_strncmp(azArg[0], "binary", n)==0 ){
8132 if( nArg==2 ){
8133 if( booleanValue(azArg[1]) ){
8134 setBinaryMode(p->out, 1);
8135 }else{
8136 setTextMode(p->out, 1);
8138 }else{
8139 eputz("The \".binary\" command is deprecated. Use \".crnl\" instead.\n"
8140 "Usage: .binary on|off\n");
8141 rc = 1;
8143 }else
8145 /* The undocumented ".breakpoint" command causes a call to the no-op
8146 ** routine named test_breakpoint().
8148 if( c=='b' && n>=3 && cli_strncmp(azArg[0], "breakpoint", n)==0 ){
8149 test_breakpoint();
8150 }else
8152 #ifndef SQLITE_SHELL_FIDDLE
8153 if( c=='c' && cli_strcmp(azArg[0],"cd")==0 ){
8154 failIfSafeMode(p, "cannot run .cd in safe mode");
8155 if( nArg==2 ){
8156 #if defined(_WIN32) || defined(WIN32)
8157 wchar_t *z = sqlite3_win32_utf8_to_unicode(azArg[1]);
8158 rc = !SetCurrentDirectoryW(z);
8159 sqlite3_free(z);
8160 #else
8161 rc = chdir(azArg[1]);
8162 #endif
8163 if( rc ){
8164 eputf("Cannot change to directory \"%s\"\n", azArg[1]);
8165 rc = 1;
8167 }else{
8168 eputz("Usage: .cd DIRECTORY\n");
8169 rc = 1;
8171 }else
8172 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
8174 if( c=='c' && n>=3 && cli_strncmp(azArg[0], "changes", n)==0 ){
8175 if( nArg==2 ){
8176 setOrClearFlag(p, SHFLG_CountChanges, azArg[1]);
8177 }else{
8178 eputz("Usage: .changes on|off\n");
8179 rc = 1;
8181 }else
8183 #ifndef SQLITE_SHELL_FIDDLE
8184 /* Cancel output redirection, if it is currently set (by .testcase)
8185 ** Then read the content of the testcase-out.txt file and compare against
8186 ** azArg[1]. If there are differences, report an error and exit.
8188 if( c=='c' && n>=3 && cli_strncmp(azArg[0], "check", n)==0 ){
8189 char *zRes = 0;
8190 output_reset(p);
8191 if( nArg!=2 ){
8192 eputz("Usage: .check GLOB-PATTERN\n");
8193 rc = 2;
8194 }else if( (zRes = readFile("testcase-out.txt", 0))==0 ){
8195 rc = 2;
8196 }else if( testcase_glob(azArg[1],zRes)==0 ){
8197 eputf("testcase-%s FAILED\n Expected: [%s]\n Got: [%s]\n",
8198 p->zTestcase, azArg[1], zRes);
8199 rc = 1;
8200 }else{
8201 oputf("testcase-%s ok\n", p->zTestcase);
8202 p->nCheck++;
8204 sqlite3_free(zRes);
8205 }else
8206 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
8208 #ifndef SQLITE_SHELL_FIDDLE
8209 if( c=='c' && cli_strncmp(azArg[0], "clone", n)==0 ){
8210 failIfSafeMode(p, "cannot run .clone in safe mode");
8211 if( nArg==2 ){
8212 tryToClone(p, azArg[1]);
8213 }else{
8214 eputz("Usage: .clone FILENAME\n");
8215 rc = 1;
8217 }else
8218 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
8220 if( c=='c' && cli_strncmp(azArg[0], "connection", n)==0 ){
8221 if( nArg==1 ){
8222 /* List available connections */
8223 int i;
8224 for(i=0; i<ArraySize(p->aAuxDb); i++){
8225 const char *zFile = p->aAuxDb[i].zDbFilename;
8226 if( p->aAuxDb[i].db==0 && p->pAuxDb!=&p->aAuxDb[i] ){
8227 zFile = "(not open)";
8228 }else if( zFile==0 ){
8229 zFile = "(memory)";
8230 }else if( zFile[0]==0 ){
8231 zFile = "(temporary-file)";
8233 if( p->pAuxDb == &p->aAuxDb[i] ){
8234 sputf(stdout, "ACTIVE %d: %s\n", i, zFile);
8235 }else if( p->aAuxDb[i].db!=0 ){
8236 sputf(stdout, " %d: %s\n", i, zFile);
8239 }else if( nArg==2 && IsDigit(azArg[1][0]) && azArg[1][1]==0 ){
8240 int i = azArg[1][0] - '0';
8241 if( p->pAuxDb != &p->aAuxDb[i] && i>=0 && i<ArraySize(p->aAuxDb) ){
8242 p->pAuxDb->db = p->db;
8243 p->pAuxDb = &p->aAuxDb[i];
8244 globalDb = p->db = p->pAuxDb->db;
8245 p->pAuxDb->db = 0;
8247 }else if( nArg==3 && cli_strcmp(azArg[1], "close")==0
8248 && IsDigit(azArg[2][0]) && azArg[2][1]==0 ){
8249 int i = azArg[2][0] - '0';
8250 if( i<0 || i>=ArraySize(p->aAuxDb) ){
8251 /* No-op */
8252 }else if( p->pAuxDb == &p->aAuxDb[i] ){
8253 eputz("cannot close the active database connection\n");
8254 rc = 1;
8255 }else if( p->aAuxDb[i].db ){
8256 session_close_all(p, i);
8257 close_db(p->aAuxDb[i].db);
8258 p->aAuxDb[i].db = 0;
8260 }else{
8261 eputz("Usage: .connection [close] [CONNECTION-NUMBER]\n");
8262 rc = 1;
8264 }else
8266 if( c=='c' && n==4 && cli_strncmp(azArg[0], "crnl", n)==0 ){
8267 if( nArg==2 ){
8268 if( booleanValue(azArg[1]) ){
8269 setTextMode(p->out, 1);
8270 }else{
8271 setBinaryMode(p->out, 1);
8273 }else{
8274 #if !defined(_WIN32) && !defined(WIN32)
8275 eputz("The \".crnl\" is a no-op on non-Windows machines.\n");
8276 #endif
8277 eputz("Usage: .crnl on|off\n");
8278 rc = 1;
8280 }else
8282 if( c=='d' && n>1 && cli_strncmp(azArg[0], "databases", n)==0 ){
8283 char **azName = 0;
8284 int nName = 0;
8285 sqlite3_stmt *pStmt;
8286 int i;
8287 open_db(p, 0);
8288 rc = sqlite3_prepare_v2(p->db, "PRAGMA database_list", -1, &pStmt, 0);
8289 if( rc ){
8290 eputf("Error: %s\n", sqlite3_errmsg(p->db));
8291 rc = 1;
8292 }else{
8293 while( sqlite3_step(pStmt)==SQLITE_ROW ){
8294 const char *zSchema = (const char *)sqlite3_column_text(pStmt,1);
8295 const char *zFile = (const char*)sqlite3_column_text(pStmt,2);
8296 if( zSchema==0 || zFile==0 ) continue;
8297 azName = sqlite3_realloc(azName, (nName+1)*2*sizeof(char*));
8298 shell_check_oom(azName);
8299 azName[nName*2] = strdup(zSchema);
8300 azName[nName*2+1] = strdup(zFile);
8301 nName++;
8304 sqlite3_finalize(pStmt);
8305 for(i=0; i<nName; i++){
8306 int eTxn = sqlite3_txn_state(p->db, azName[i*2]);
8307 int bRdonly = sqlite3_db_readonly(p->db, azName[i*2]);
8308 const char *z = azName[i*2+1];
8309 oputf("%s: %s %s%s\n",
8310 azName[i*2], z && z[0] ? z : "\"\"", bRdonly ? "r/o" : "r/w",
8311 eTxn==SQLITE_TXN_NONE ? "" :
8312 eTxn==SQLITE_TXN_READ ? " read-txn" : " write-txn");
8313 free(azName[i*2]);
8314 free(azName[i*2+1]);
8316 sqlite3_free(azName);
8317 }else
8319 if( c=='d' && n>=3 && cli_strncmp(azArg[0], "dbconfig", n)==0 ){
8320 static const struct DbConfigChoices {
8321 const char *zName;
8322 int op;
8323 } aDbConfig[] = {
8324 { "defensive", SQLITE_DBCONFIG_DEFENSIVE },
8325 { "dqs_ddl", SQLITE_DBCONFIG_DQS_DDL },
8326 { "dqs_dml", SQLITE_DBCONFIG_DQS_DML },
8327 { "enable_fkey", SQLITE_DBCONFIG_ENABLE_FKEY },
8328 { "enable_qpsg", SQLITE_DBCONFIG_ENABLE_QPSG },
8329 { "enable_trigger", SQLITE_DBCONFIG_ENABLE_TRIGGER },
8330 { "enable_view", SQLITE_DBCONFIG_ENABLE_VIEW },
8331 { "fts3_tokenizer", SQLITE_DBCONFIG_ENABLE_FTS3_TOKENIZER },
8332 { "legacy_alter_table", SQLITE_DBCONFIG_LEGACY_ALTER_TABLE },
8333 { "legacy_file_format", SQLITE_DBCONFIG_LEGACY_FILE_FORMAT },
8334 { "load_extension", SQLITE_DBCONFIG_ENABLE_LOAD_EXTENSION },
8335 { "no_ckpt_on_close", SQLITE_DBCONFIG_NO_CKPT_ON_CLOSE },
8336 { "reset_database", SQLITE_DBCONFIG_RESET_DATABASE },
8337 { "reverse_scanorder", SQLITE_DBCONFIG_REVERSE_SCANORDER },
8338 { "stmt_scanstatus", SQLITE_DBCONFIG_STMT_SCANSTATUS },
8339 { "trigger_eqp", SQLITE_DBCONFIG_TRIGGER_EQP },
8340 { "trusted_schema", SQLITE_DBCONFIG_TRUSTED_SCHEMA },
8341 { "writable_schema", SQLITE_DBCONFIG_WRITABLE_SCHEMA },
8343 int ii, v;
8344 open_db(p, 0);
8345 for(ii=0; ii<ArraySize(aDbConfig); ii++){
8346 if( nArg>1 && cli_strcmp(azArg[1], aDbConfig[ii].zName)!=0 ) continue;
8347 if( nArg>=3 ){
8348 sqlite3_db_config(p->db, aDbConfig[ii].op, booleanValue(azArg[2]), 0);
8350 sqlite3_db_config(p->db, aDbConfig[ii].op, -1, &v);
8351 oputf("%19s %s\n", aDbConfig[ii].zName, v ? "on" : "off");
8352 if( nArg>1 ) break;
8354 if( nArg>1 && ii==ArraySize(aDbConfig) ){
8355 eputf("Error: unknown dbconfig \"%s\"\n", azArg[1]);
8356 eputz("Enter \".dbconfig\" with no arguments for a list\n");
8358 }else
8360 #if SQLITE_SHELL_HAVE_RECOVER
8361 if( c=='d' && n>=3 && cli_strncmp(azArg[0], "dbinfo", n)==0 ){
8362 rc = shell_dbinfo_command(p, nArg, azArg);
8363 }else
8365 if( c=='r' && cli_strncmp(azArg[0], "recover", n)==0 ){
8366 open_db(p, 0);
8367 rc = recoverDatabaseCmd(p, nArg, azArg);
8368 }else
8369 #endif /* SQLITE_SHELL_HAVE_RECOVER */
8371 if( c=='d' && cli_strncmp(azArg[0], "dump", n)==0 ){
8372 char *zLike = 0;
8373 char *zSql;
8374 int i;
8375 int savedShowHeader = p->showHeader;
8376 int savedShellFlags = p->shellFlgs;
8377 ShellClearFlag(p,
8378 SHFLG_PreserveRowid|SHFLG_Newlines|SHFLG_Echo
8379 |SHFLG_DumpDataOnly|SHFLG_DumpNoSys);
8380 for(i=1; i<nArg; i++){
8381 if( azArg[i][0]=='-' ){
8382 const char *z = azArg[i]+1;
8383 if( z[0]=='-' ) z++;
8384 if( cli_strcmp(z,"preserve-rowids")==0 ){
8385 #ifdef SQLITE_OMIT_VIRTUALTABLE
8386 eputz("The --preserve-rowids option is not compatible"
8387 " with SQLITE_OMIT_VIRTUALTABLE\n");
8388 rc = 1;
8389 sqlite3_free(zLike);
8390 goto meta_command_exit;
8391 #else
8392 ShellSetFlag(p, SHFLG_PreserveRowid);
8393 #endif
8394 }else
8395 if( cli_strcmp(z,"newlines")==0 ){
8396 ShellSetFlag(p, SHFLG_Newlines);
8397 }else
8398 if( cli_strcmp(z,"data-only")==0 ){
8399 ShellSetFlag(p, SHFLG_DumpDataOnly);
8400 }else
8401 if( cli_strcmp(z,"nosys")==0 ){
8402 ShellSetFlag(p, SHFLG_DumpNoSys);
8403 }else
8405 eputf("Unknown option \"%s\" on \".dump\"\n", azArg[i]);
8406 rc = 1;
8407 sqlite3_free(zLike);
8408 goto meta_command_exit;
8410 }else{
8411 /* azArg[i] contains a LIKE pattern. This ".dump" request should
8412 ** only dump data for tables for which either the table name matches
8413 ** the LIKE pattern, or the table appears to be a shadow table of
8414 ** a virtual table for which the name matches the LIKE pattern.
8416 char *zExpr = sqlite3_mprintf(
8417 "name LIKE %Q ESCAPE '\\' OR EXISTS ("
8418 " SELECT 1 FROM sqlite_schema WHERE "
8419 " name LIKE %Q ESCAPE '\\' AND"
8420 " sql LIKE 'CREATE VIRTUAL TABLE%%' AND"
8421 " substr(o.name, 1, length(name)+1) == (name||'_')"
8422 ")", azArg[i], azArg[i]
8425 if( zLike ){
8426 zLike = sqlite3_mprintf("%z OR %z", zLike, zExpr);
8427 }else{
8428 zLike = zExpr;
8433 open_db(p, 0);
8435 outputDumpWarning(p, zLike);
8436 if( (p->shellFlgs & SHFLG_DumpDataOnly)==0 ){
8437 /* When playing back a "dump", the content might appear in an order
8438 ** which causes immediate foreign key constraints to be violated.
8439 ** So disable foreign-key constraint enforcement to prevent problems. */
8440 oputz("PRAGMA foreign_keys=OFF;\n");
8441 oputz("BEGIN TRANSACTION;\n");
8443 p->writableSchema = 0;
8444 p->showHeader = 0;
8445 /* Set writable_schema=ON since doing so forces SQLite to initialize
8446 ** as much of the schema as it can even if the sqlite_schema table is
8447 ** corrupt. */
8448 sqlite3_exec(p->db, "SAVEPOINT dump; PRAGMA writable_schema=ON", 0, 0, 0);
8449 p->nErr = 0;
8450 if( zLike==0 ) zLike = sqlite3_mprintf("true");
8451 zSql = sqlite3_mprintf(
8452 "SELECT name, type, sql FROM sqlite_schema AS o "
8453 "WHERE (%s) AND type=='table'"
8454 " AND sql NOT NULL"
8455 " ORDER BY tbl_name='sqlite_sequence', rowid",
8456 zLike
8458 run_schema_dump_query(p,zSql);
8459 sqlite3_free(zSql);
8460 if( (p->shellFlgs & SHFLG_DumpDataOnly)==0 ){
8461 zSql = sqlite3_mprintf(
8462 "SELECT sql FROM sqlite_schema AS o "
8463 "WHERE (%s) AND sql NOT NULL"
8464 " AND type IN ('index','trigger','view') "
8465 "ORDER BY type COLLATE NOCASE DESC",
8466 zLike
8468 run_table_dump_query(p, zSql);
8469 sqlite3_free(zSql);
8471 sqlite3_free(zLike);
8472 if( p->writableSchema ){
8473 oputz("PRAGMA writable_schema=OFF;\n");
8474 p->writableSchema = 0;
8476 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
8477 sqlite3_exec(p->db, "RELEASE dump;", 0, 0, 0);
8478 if( (p->shellFlgs & SHFLG_DumpDataOnly)==0 ){
8479 oputz(p->nErr?"ROLLBACK; -- due to errors\n":"COMMIT;\n");
8481 p->showHeader = savedShowHeader;
8482 p->shellFlgs = savedShellFlags;
8483 }else
8485 if( c=='e' && cli_strncmp(azArg[0], "echo", n)==0 ){
8486 if( nArg==2 ){
8487 setOrClearFlag(p, SHFLG_Echo, azArg[1]);
8488 }else{
8489 eputz("Usage: .echo on|off\n");
8490 rc = 1;
8492 }else
8494 if( c=='e' && cli_strncmp(azArg[0], "eqp", n)==0 ){
8495 if( nArg==2 ){
8496 p->autoEQPtest = 0;
8497 if( p->autoEQPtrace ){
8498 if( p->db ) sqlite3_exec(p->db, "PRAGMA vdbe_trace=OFF;", 0, 0, 0);
8499 p->autoEQPtrace = 0;
8501 if( cli_strcmp(azArg[1],"full")==0 ){
8502 p->autoEQP = AUTOEQP_full;
8503 }else if( cli_strcmp(azArg[1],"trigger")==0 ){
8504 p->autoEQP = AUTOEQP_trigger;
8505 #ifdef SQLITE_DEBUG
8506 }else if( cli_strcmp(azArg[1],"test")==0 ){
8507 p->autoEQP = AUTOEQP_on;
8508 p->autoEQPtest = 1;
8509 }else if( cli_strcmp(azArg[1],"trace")==0 ){
8510 p->autoEQP = AUTOEQP_full;
8511 p->autoEQPtrace = 1;
8512 open_db(p, 0);
8513 sqlite3_exec(p->db, "SELECT name FROM sqlite_schema LIMIT 1", 0, 0, 0);
8514 sqlite3_exec(p->db, "PRAGMA vdbe_trace=ON;", 0, 0, 0);
8515 #endif
8516 }else{
8517 p->autoEQP = (u8)booleanValue(azArg[1]);
8519 }else{
8520 eputz("Usage: .eqp off|on|trace|trigger|full\n");
8521 rc = 1;
8523 }else
8525 #ifndef SQLITE_SHELL_FIDDLE
8526 if( c=='e' && cli_strncmp(azArg[0], "exit", n)==0 ){
8527 if( nArg>1 && (rc = (int)integerValue(azArg[1]))!=0 ) exit(rc);
8528 rc = 2;
8529 }else
8530 #endif
8532 /* The ".explain" command is automatic now. It is largely pointless. It
8533 ** retained purely for backwards compatibility */
8534 if( c=='e' && cli_strncmp(azArg[0], "explain", n)==0 ){
8535 int val = 1;
8536 if( nArg>=2 ){
8537 if( cli_strcmp(azArg[1],"auto")==0 ){
8538 val = 99;
8539 }else{
8540 val = booleanValue(azArg[1]);
8543 if( val==1 && p->mode!=MODE_Explain ){
8544 p->normalMode = p->mode;
8545 p->mode = MODE_Explain;
8546 p->autoExplain = 0;
8547 }else if( val==0 ){
8548 if( p->mode==MODE_Explain ) p->mode = p->normalMode;
8549 p->autoExplain = 0;
8550 }else if( val==99 ){
8551 if( p->mode==MODE_Explain ) p->mode = p->normalMode;
8552 p->autoExplain = 1;
8554 }else
8556 #ifndef SQLITE_OMIT_VIRTUALTABLE
8557 if( c=='e' && cli_strncmp(azArg[0], "expert", n)==0 ){
8558 if( p->bSafeMode ){
8559 eputf("Cannot run experimental commands such as \"%s\" in safe mode\n",
8560 azArg[0]);
8561 rc = 1;
8562 }else{
8563 open_db(p, 0);
8564 expertDotCommand(p, azArg, nArg);
8566 }else
8567 #endif
8569 if( c=='f' && cli_strncmp(azArg[0], "filectrl", n)==0 ){
8570 static const struct {
8571 const char *zCtrlName; /* Name of a test-control option */
8572 int ctrlCode; /* Integer code for that option */
8573 const char *zUsage; /* Usage notes */
8574 } aCtrl[] = {
8575 { "chunk_size", SQLITE_FCNTL_CHUNK_SIZE, "SIZE" },
8576 { "data_version", SQLITE_FCNTL_DATA_VERSION, "" },
8577 { "has_moved", SQLITE_FCNTL_HAS_MOVED, "" },
8578 { "lock_timeout", SQLITE_FCNTL_LOCK_TIMEOUT, "MILLISEC" },
8579 { "persist_wal", SQLITE_FCNTL_PERSIST_WAL, "[BOOLEAN]" },
8580 /* { "pragma", SQLITE_FCNTL_PRAGMA, "NAME ARG" },*/
8581 { "psow", SQLITE_FCNTL_POWERSAFE_OVERWRITE, "[BOOLEAN]" },
8582 { "reserve_bytes", SQLITE_FCNTL_RESERVE_BYTES, "[N]" },
8583 { "size_limit", SQLITE_FCNTL_SIZE_LIMIT, "[LIMIT]" },
8584 { "tempfilename", SQLITE_FCNTL_TEMPFILENAME, "" },
8585 /* { "win32_av_retry", SQLITE_FCNTL_WIN32_AV_RETRY, "COUNT DELAY" },*/
8587 int filectrl = -1;
8588 int iCtrl = -1;
8589 sqlite3_int64 iRes = 0; /* Integer result to display if rc2==1 */
8590 int isOk = 0; /* 0: usage 1: %lld 2: no-result */
8591 int n2, i;
8592 const char *zCmd = 0;
8593 const char *zSchema = 0;
8595 open_db(p, 0);
8596 zCmd = nArg>=2 ? azArg[1] : "help";
8598 if( zCmd[0]=='-'
8599 && (cli_strcmp(zCmd,"--schema")==0 || cli_strcmp(zCmd,"-schema")==0)
8600 && nArg>=4
8602 zSchema = azArg[2];
8603 for(i=3; i<nArg; i++) azArg[i-2] = azArg[i];
8604 nArg -= 2;
8605 zCmd = azArg[1];
8608 /* The argument can optionally begin with "-" or "--" */
8609 if( zCmd[0]=='-' && zCmd[1] ){
8610 zCmd++;
8611 if( zCmd[0]=='-' && zCmd[1] ) zCmd++;
8614 /* --help lists all file-controls */
8615 if( cli_strcmp(zCmd,"help")==0 ){
8616 oputz("Available file-controls:\n");
8617 for(i=0; i<ArraySize(aCtrl); i++){
8618 oputf(" .filectrl %s %s\n", aCtrl[i].zCtrlName, aCtrl[i].zUsage);
8620 rc = 1;
8621 goto meta_command_exit;
8624 /* convert filectrl text option to value. allow any unique prefix
8625 ** of the option name, or a numerical value. */
8626 n2 = strlen30(zCmd);
8627 for(i=0; i<ArraySize(aCtrl); i++){
8628 if( cli_strncmp(zCmd, aCtrl[i].zCtrlName, n2)==0 ){
8629 if( filectrl<0 ){
8630 filectrl = aCtrl[i].ctrlCode;
8631 iCtrl = i;
8632 }else{
8633 eputf("Error: ambiguous file-control: \"%s\"\n"
8634 "Use \".filectrl --help\" for help\n", zCmd);
8635 rc = 1;
8636 goto meta_command_exit;
8640 if( filectrl<0 ){
8641 eputf("Error: unknown file-control: %s\n"
8642 "Use \".filectrl --help\" for help\n", zCmd);
8643 }else{
8644 switch(filectrl){
8645 case SQLITE_FCNTL_SIZE_LIMIT: {
8646 if( nArg!=2 && nArg!=3 ) break;
8647 iRes = nArg==3 ? integerValue(azArg[2]) : -1;
8648 sqlite3_file_control(p->db, zSchema, SQLITE_FCNTL_SIZE_LIMIT, &iRes);
8649 isOk = 1;
8650 break;
8652 case SQLITE_FCNTL_LOCK_TIMEOUT:
8653 case SQLITE_FCNTL_CHUNK_SIZE: {
8654 int x;
8655 if( nArg!=3 ) break;
8656 x = (int)integerValue(azArg[2]);
8657 sqlite3_file_control(p->db, zSchema, filectrl, &x);
8658 isOk = 2;
8659 break;
8661 case SQLITE_FCNTL_PERSIST_WAL:
8662 case SQLITE_FCNTL_POWERSAFE_OVERWRITE: {
8663 int x;
8664 if( nArg!=2 && nArg!=3 ) break;
8665 x = nArg==3 ? booleanValue(azArg[2]) : -1;
8666 sqlite3_file_control(p->db, zSchema, filectrl, &x);
8667 iRes = x;
8668 isOk = 1;
8669 break;
8671 case SQLITE_FCNTL_DATA_VERSION:
8672 case SQLITE_FCNTL_HAS_MOVED: {
8673 int x;
8674 if( nArg!=2 ) break;
8675 sqlite3_file_control(p->db, zSchema, filectrl, &x);
8676 iRes = x;
8677 isOk = 1;
8678 break;
8680 case SQLITE_FCNTL_TEMPFILENAME: {
8681 char *z = 0;
8682 if( nArg!=2 ) break;
8683 sqlite3_file_control(p->db, zSchema, filectrl, &z);
8684 if( z ){
8685 oputf("%s\n", z);
8686 sqlite3_free(z);
8688 isOk = 2;
8689 break;
8691 case SQLITE_FCNTL_RESERVE_BYTES: {
8692 int x;
8693 if( nArg>=3 ){
8694 x = atoi(azArg[2]);
8695 sqlite3_file_control(p->db, zSchema, filectrl, &x);
8697 x = -1;
8698 sqlite3_file_control(p->db, zSchema, filectrl, &x);
8699 oputf("%d\n", x);
8700 isOk = 2;
8701 break;
8705 if( isOk==0 && iCtrl>=0 ){
8706 oputf("Usage: .filectrl %s %s\n", zCmd,aCtrl[iCtrl].zUsage);
8707 rc = 1;
8708 }else if( isOk==1 ){
8709 char zBuf[100];
8710 sqlite3_snprintf(sizeof(zBuf), zBuf, "%lld", iRes);
8711 oputf("%s\n", zBuf);
8713 }else
8715 if( c=='f' && cli_strncmp(azArg[0], "fullschema", n)==0 ){
8716 ShellState data;
8717 int doStats = 0;
8718 memcpy(&data, p, sizeof(data));
8719 data.showHeader = 0;
8720 data.cMode = data.mode = MODE_Semi;
8721 if( nArg==2 && optionMatch(azArg[1], "indent") ){
8722 data.cMode = data.mode = MODE_Pretty;
8723 nArg = 1;
8725 if( nArg!=1 ){
8726 eputz("Usage: .fullschema ?--indent?\n");
8727 rc = 1;
8728 goto meta_command_exit;
8730 open_db(p, 0);
8731 rc = sqlite3_exec(p->db,
8732 "SELECT sql FROM"
8733 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
8734 " FROM sqlite_schema UNION ALL"
8735 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_schema) "
8736 "WHERE type!='meta' AND sql NOTNULL AND name NOT LIKE 'sqlite_%' "
8737 "ORDER BY x",
8738 callback, &data, 0
8740 if( rc==SQLITE_OK ){
8741 sqlite3_stmt *pStmt;
8742 rc = sqlite3_prepare_v2(p->db,
8743 "SELECT rowid FROM sqlite_schema"
8744 " WHERE name GLOB 'sqlite_stat[134]'",
8745 -1, &pStmt, 0);
8746 if( rc==SQLITE_OK ){
8747 doStats = sqlite3_step(pStmt)==SQLITE_ROW;
8748 sqlite3_finalize(pStmt);
8751 if( doStats==0 ){
8752 oputz("/* No STAT tables available */\n");
8753 }else{
8754 oputz("ANALYZE sqlite_schema;\n");
8755 data.cMode = data.mode = MODE_Insert;
8756 data.zDestTable = "sqlite_stat1";
8757 shell_exec(&data, "SELECT * FROM sqlite_stat1", 0);
8758 data.zDestTable = "sqlite_stat4";
8759 shell_exec(&data, "SELECT * FROM sqlite_stat4", 0);
8760 oputz("ANALYZE sqlite_schema;\n");
8762 }else
8764 if( c=='h' && cli_strncmp(azArg[0], "headers", n)==0 ){
8765 if( nArg==2 ){
8766 p->showHeader = booleanValue(azArg[1]);
8767 p->shellFlgs |= SHFLG_HeaderSet;
8768 }else{
8769 eputz("Usage: .headers on|off\n");
8770 rc = 1;
8772 }else
8774 if( c=='h' && cli_strncmp(azArg[0], "help", n)==0 ){
8775 if( nArg>=2 ){
8776 n = showHelp(p->out, azArg[1]);
8777 if( n==0 ){
8778 oputf("Nothing matches '%s'\n", azArg[1]);
8780 }else{
8781 showHelp(p->out, 0);
8783 }else
8785 #ifndef SQLITE_SHELL_FIDDLE
8786 if( c=='i' && cli_strncmp(azArg[0], "import", n)==0 ){
8787 char *zTable = 0; /* Insert data into this table */
8788 char *zSchema = 0; /* Schema of zTable */
8789 char *zFile = 0; /* Name of file to extra content from */
8790 sqlite3_stmt *pStmt = NULL; /* A statement */
8791 int nCol; /* Number of columns in the table */
8792 i64 nByte; /* Number of bytes in an SQL string */
8793 int i, j; /* Loop counters */
8794 int needCommit; /* True to COMMIT or ROLLBACK at end */
8795 int nSep; /* Number of bytes in p->colSeparator[] */
8796 char *zSql = 0; /* An SQL statement */
8797 ImportCtx sCtx; /* Reader context */
8798 char *(SQLITE_CDECL *xRead)(ImportCtx*); /* Func to read one value */
8799 int eVerbose = 0; /* Larger for more console output */
8800 int nSkip = 0; /* Initial lines to skip */
8801 int useOutputMode = 1; /* Use output mode to determine separators */
8802 char *zCreate = 0; /* CREATE TABLE statement text */
8804 failIfSafeMode(p, "cannot run .import in safe mode");
8805 memset(&sCtx, 0, sizeof(sCtx));
8806 if( p->mode==MODE_Ascii ){
8807 xRead = ascii_read_one_field;
8808 }else{
8809 xRead = csv_read_one_field;
8811 rc = 1;
8812 for(i=1; i<nArg; i++){
8813 char *z = azArg[i];
8814 if( z[0]=='-' && z[1]=='-' ) z++;
8815 if( z[0]!='-' ){
8816 if( zFile==0 ){
8817 zFile = z;
8818 }else if( zTable==0 ){
8819 zTable = z;
8820 }else{
8821 oputf("ERROR: extra argument: \"%s\". Usage:\n", z);
8822 showHelp(p->out, "import");
8823 goto meta_command_exit;
8825 }else if( cli_strcmp(z,"-v")==0 ){
8826 eVerbose++;
8827 }else if( cli_strcmp(z,"-schema")==0 && i<nArg-1 ){
8828 zSchema = azArg[++i];
8829 }else if( cli_strcmp(z,"-skip")==0 && i<nArg-1 ){
8830 nSkip = integerValue(azArg[++i]);
8831 }else if( cli_strcmp(z,"-ascii")==0 ){
8832 sCtx.cColSep = SEP_Unit[0];
8833 sCtx.cRowSep = SEP_Record[0];
8834 xRead = ascii_read_one_field;
8835 useOutputMode = 0;
8836 }else if( cli_strcmp(z,"-csv")==0 ){
8837 sCtx.cColSep = ',';
8838 sCtx.cRowSep = '\n';
8839 xRead = csv_read_one_field;
8840 useOutputMode = 0;
8841 }else{
8842 oputf("ERROR: unknown option: \"%s\". Usage:\n", z);
8843 showHelp(p->out, "import");
8844 goto meta_command_exit;
8847 if( zTable==0 ){
8848 oputf("ERROR: missing %s argument. Usage:\n",
8849 zFile==0 ? "FILE" : "TABLE");
8850 showHelp(p->out, "import");
8851 goto meta_command_exit;
8853 seenInterrupt = 0;
8854 open_db(p, 0);
8855 if( useOutputMode ){
8856 /* If neither the --csv or --ascii options are specified, then set
8857 ** the column and row separator characters from the output mode. */
8858 nSep = strlen30(p->colSeparator);
8859 if( nSep==0 ){
8860 eputz("Error: non-null column separator required for import\n");
8861 goto meta_command_exit;
8863 if( nSep>1 ){
8864 eputz("Error: multi-character column separators not allowed"
8865 " for import\n");
8866 goto meta_command_exit;
8868 nSep = strlen30(p->rowSeparator);
8869 if( nSep==0 ){
8870 eputz("Error: non-null row separator required for import\n");
8871 goto meta_command_exit;
8873 if( nSep==2 && p->mode==MODE_Csv
8874 && cli_strcmp(p->rowSeparator,SEP_CrLf)==0
8876 /* When importing CSV (only), if the row separator is set to the
8877 ** default output row separator, change it to the default input
8878 ** row separator. This avoids having to maintain different input
8879 ** and output row separators. */
8880 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
8881 nSep = strlen30(p->rowSeparator);
8883 if( nSep>1 ){
8884 eputz("Error: multi-character row separators not allowed"
8885 " for import\n");
8886 goto meta_command_exit;
8888 sCtx.cColSep = (u8)p->colSeparator[0];
8889 sCtx.cRowSep = (u8)p->rowSeparator[0];
8891 sCtx.zFile = zFile;
8892 sCtx.nLine = 1;
8893 if( sCtx.zFile[0]=='|' ){
8894 #ifdef SQLITE_OMIT_POPEN
8895 eputz("Error: pipes are not supported in this OS\n");
8896 goto meta_command_exit;
8897 #else
8898 sCtx.in = popen(sCtx.zFile+1, "r");
8899 sCtx.zFile = "<pipe>";
8900 sCtx.xCloser = pclose;
8901 #endif
8902 }else{
8903 sCtx.in = fopen(sCtx.zFile, "rb");
8904 sCtx.xCloser = fclose;
8906 if( sCtx.in==0 ){
8907 eputf("Error: cannot open \"%s\"\n", zFile);
8908 goto meta_command_exit;
8910 if( eVerbose>=2 || (eVerbose>=1 && useOutputMode) ){
8911 char zSep[2];
8912 zSep[1] = 0;
8913 zSep[0] = sCtx.cColSep;
8914 oputz("Column separator ");
8915 output_c_string(zSep);
8916 oputz(", row separator ");
8917 zSep[0] = sCtx.cRowSep;
8918 output_c_string(zSep);
8919 oputz("\n");
8921 sCtx.z = sqlite3_malloc64(120);
8922 if( sCtx.z==0 ){
8923 import_cleanup(&sCtx);
8924 shell_out_of_memory();
8926 /* Below, resources must be freed before exit. */
8927 while( (nSkip--)>0 ){
8928 while( xRead(&sCtx) && sCtx.cTerm==sCtx.cColSep ){}
8930 import_append_char(&sCtx, 0); /* To ensure sCtx.z is allocated */
8931 if( sqlite3_table_column_metadata(p->db, zSchema, zTable,0,0,0,0,0,0) ){
8932 /* Table does not exist. Create it. */
8933 sqlite3 *dbCols = 0;
8934 char *zRenames = 0;
8935 char *zColDefs;
8936 zCreate = sqlite3_mprintf("CREATE TABLE \"%w\".\"%w\"",
8937 zSchema ? zSchema : "main", zTable);
8938 while( xRead(&sCtx) ){
8939 zAutoColumn(sCtx.z, &dbCols, 0);
8940 if( sCtx.cTerm!=sCtx.cColSep ) break;
8942 zColDefs = zAutoColumn(0, &dbCols, &zRenames);
8943 if( zRenames!=0 ){
8944 sputf((stdin_is_interactive && p->in==stdin)? p->out : stderr,
8945 "Columns renamed during .import %s due to duplicates:\n"
8946 "%s\n", sCtx.zFile, zRenames);
8947 sqlite3_free(zRenames);
8949 assert(dbCols==0);
8950 if( zColDefs==0 ){
8951 eputf("%s: empty file\n", sCtx.zFile);
8952 import_cleanup(&sCtx);
8953 rc = 1;
8954 goto meta_command_exit;
8956 zCreate = sqlite3_mprintf("%z%z\n", zCreate, zColDefs);
8957 if( zCreate==0 ){
8958 import_cleanup(&sCtx);
8959 shell_out_of_memory();
8961 if( eVerbose>=1 ){
8962 oputf("%s\n", zCreate);
8964 rc = sqlite3_exec(p->db, zCreate, 0, 0, 0);
8965 sqlite3_free(zCreate);
8966 zCreate = 0;
8967 if( rc ){
8968 eputf("%s failed:\n%s\n", zCreate, sqlite3_errmsg(p->db));
8969 import_cleanup(&sCtx);
8970 rc = 1;
8971 goto meta_command_exit;
8974 zSql = sqlite3_mprintf("SELECT count(*) FROM pragma_table_info(%Q,%Q);",
8975 zTable, zSchema);
8976 if( zSql==0 ){
8977 import_cleanup(&sCtx);
8978 shell_out_of_memory();
8980 nByte = strlen(zSql);
8981 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
8982 sqlite3_free(zSql);
8983 zSql = 0;
8984 if( rc ){
8985 if (pStmt) sqlite3_finalize(pStmt);
8986 eputf("Error: %s\n", sqlite3_errmsg(p->db));
8987 import_cleanup(&sCtx);
8988 rc = 1;
8989 goto meta_command_exit;
8991 if( sqlite3_step(pStmt)==SQLITE_ROW ){
8992 nCol = sqlite3_column_int(pStmt, 0);
8993 }else{
8994 nCol = 0;
8996 sqlite3_finalize(pStmt);
8997 pStmt = 0;
8998 if( nCol==0 ) return 0; /* no columns, no error */
8999 zSql = sqlite3_malloc64( nByte*2 + 20 + nCol*2 );
9000 if( zSql==0 ){
9001 import_cleanup(&sCtx);
9002 shell_out_of_memory();
9004 if( zSchema ){
9005 sqlite3_snprintf(nByte+20, zSql, "INSERT INTO \"%w\".\"%w\" VALUES(?",
9006 zSchema, zTable);
9007 }else{
9008 sqlite3_snprintf(nByte+20, zSql, "INSERT INTO \"%w\" VALUES(?", zTable);
9010 j = strlen30(zSql);
9011 for(i=1; i<nCol; i++){
9012 zSql[j++] = ',';
9013 zSql[j++] = '?';
9015 zSql[j++] = ')';
9016 zSql[j] = 0;
9017 if( eVerbose>=2 ){
9018 oputf("Insert using: %s\n", zSql);
9020 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
9021 sqlite3_free(zSql);
9022 zSql = 0;
9023 if( rc ){
9024 eputf("Error: %s\n", sqlite3_errmsg(p->db));
9025 if (pStmt) sqlite3_finalize(pStmt);
9026 import_cleanup(&sCtx);
9027 rc = 1;
9028 goto meta_command_exit;
9030 needCommit = sqlite3_get_autocommit(p->db);
9031 if( needCommit ) sqlite3_exec(p->db, "BEGIN", 0, 0, 0);
9033 int startLine = sCtx.nLine;
9034 for(i=0; i<nCol; i++){
9035 char *z = xRead(&sCtx);
9037 ** Did we reach end-of-file before finding any columns?
9038 ** If so, stop instead of NULL filling the remaining columns.
9040 if( z==0 && i==0 ) break;
9042 ** Did we reach end-of-file OR end-of-line before finding any
9043 ** columns in ASCII mode? If so, stop instead of NULL filling
9044 ** the remaining columns.
9046 if( p->mode==MODE_Ascii && (z==0 || z[0]==0) && i==0 ) break;
9048 ** For CSV mode, per RFC 4180, accept EOF in lieu of final
9049 ** record terminator but only for last field of multi-field row.
9050 ** (If there are too few fields, it's not valid CSV anyway.)
9052 if( z==0 && (xRead==csv_read_one_field) && i==nCol-1 && i>0 ){
9053 z = "";
9055 sqlite3_bind_text(pStmt, i+1, z, -1, SQLITE_TRANSIENT);
9056 if( i<nCol-1 && sCtx.cTerm!=sCtx.cColSep ){
9057 eputf("%s:%d: expected %d columns but found %d"
9058 " - filling the rest with NULL\n",
9059 sCtx.zFile, startLine, nCol, i+1);
9060 i += 2;
9061 while( i<=nCol ){ sqlite3_bind_null(pStmt, i); i++; }
9064 if( sCtx.cTerm==sCtx.cColSep ){
9066 xRead(&sCtx);
9067 i++;
9068 }while( sCtx.cTerm==sCtx.cColSep );
9069 eputf("%s:%d: expected %d columns but found %d - extras ignored\n",
9070 sCtx.zFile, startLine, nCol, i);
9072 if( i>=nCol ){
9073 sqlite3_step(pStmt);
9074 rc = sqlite3_reset(pStmt);
9075 if( rc!=SQLITE_OK ){
9076 eputf("%s:%d: INSERT failed: %s\n",
9077 sCtx.zFile, startLine, sqlite3_errmsg(p->db));
9078 sCtx.nErr++;
9079 }else{
9080 sCtx.nRow++;
9083 }while( sCtx.cTerm!=EOF );
9085 import_cleanup(&sCtx);
9086 sqlite3_finalize(pStmt);
9087 if( needCommit ) sqlite3_exec(p->db, "COMMIT", 0, 0, 0);
9088 if( eVerbose>0 ){
9089 oputf("Added %d rows with %d errors using %d lines of input\n",
9090 sCtx.nRow, sCtx.nErr, sCtx.nLine-1);
9092 }else
9093 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
9095 #ifndef SQLITE_UNTESTABLE
9096 if( c=='i' && cli_strncmp(azArg[0], "imposter", n)==0 ){
9097 char *zSql;
9098 char *zCollist = 0;
9099 sqlite3_stmt *pStmt;
9100 int tnum = 0;
9101 int isWO = 0; /* True if making an imposter of a WITHOUT ROWID table */
9102 int lenPK = 0; /* Length of the PRIMARY KEY string for isWO tables */
9103 int i;
9104 if( !ShellHasFlag(p,SHFLG_TestingMode) ){
9105 eputf(".%s unavailable without --unsafe-testing\n",
9106 "imposter");
9107 rc = 1;
9108 goto meta_command_exit;
9110 if( !(nArg==3 || (nArg==2 && sqlite3_stricmp(azArg[1],"off")==0)) ){
9111 eputz("Usage: .imposter INDEX IMPOSTER\n"
9112 " .imposter off\n");
9113 /* Also allowed, but not documented:
9115 ** .imposter TABLE IMPOSTER
9117 ** where TABLE is a WITHOUT ROWID table. In that case, the
9118 ** imposter is another WITHOUT ROWID table with the columns in
9119 ** storage order. */
9120 rc = 1;
9121 goto meta_command_exit;
9123 open_db(p, 0);
9124 if( nArg==2 ){
9125 sqlite3_test_control(SQLITE_TESTCTRL_IMPOSTER, p->db, "main", 0, 1);
9126 goto meta_command_exit;
9128 zSql = sqlite3_mprintf(
9129 "SELECT rootpage, 0 FROM sqlite_schema"
9130 " WHERE name='%q' AND type='index'"
9131 "UNION ALL "
9132 "SELECT rootpage, 1 FROM sqlite_schema"
9133 " WHERE name='%q' AND type='table'"
9134 " AND sql LIKE '%%without%%rowid%%'",
9135 azArg[1], azArg[1]
9137 sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
9138 sqlite3_free(zSql);
9139 if( sqlite3_step(pStmt)==SQLITE_ROW ){
9140 tnum = sqlite3_column_int(pStmt, 0);
9141 isWO = sqlite3_column_int(pStmt, 1);
9143 sqlite3_finalize(pStmt);
9144 zSql = sqlite3_mprintf("PRAGMA index_xinfo='%q'", azArg[1]);
9145 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
9146 sqlite3_free(zSql);
9147 i = 0;
9148 while( rc==SQLITE_OK && sqlite3_step(pStmt)==SQLITE_ROW ){
9149 char zLabel[20];
9150 const char *zCol = (const char*)sqlite3_column_text(pStmt,2);
9151 i++;
9152 if( zCol==0 ){
9153 if( sqlite3_column_int(pStmt,1)==-1 ){
9154 zCol = "_ROWID_";
9155 }else{
9156 sqlite3_snprintf(sizeof(zLabel),zLabel,"expr%d",i);
9157 zCol = zLabel;
9160 if( isWO && lenPK==0 && sqlite3_column_int(pStmt,5)==0 && zCollist ){
9161 lenPK = (int)strlen(zCollist);
9163 if( zCollist==0 ){
9164 zCollist = sqlite3_mprintf("\"%w\"", zCol);
9165 }else{
9166 zCollist = sqlite3_mprintf("%z,\"%w\"", zCollist, zCol);
9169 sqlite3_finalize(pStmt);
9170 if( i==0 || tnum==0 ){
9171 eputf("no such index: \"%s\"\n", azArg[1]);
9172 rc = 1;
9173 sqlite3_free(zCollist);
9174 goto meta_command_exit;
9176 if( lenPK==0 ) lenPK = 100000;
9177 zSql = sqlite3_mprintf(
9178 "CREATE TABLE \"%w\"(%s,PRIMARY KEY(%.*s))WITHOUT ROWID",
9179 azArg[2], zCollist, lenPK, zCollist);
9180 sqlite3_free(zCollist);
9181 rc = sqlite3_test_control(SQLITE_TESTCTRL_IMPOSTER, p->db, "main", 1, tnum);
9182 if( rc==SQLITE_OK ){
9183 rc = sqlite3_exec(p->db, zSql, 0, 0, 0);
9184 sqlite3_test_control(SQLITE_TESTCTRL_IMPOSTER, p->db, "main", 0, 0);
9185 if( rc ){
9186 eputf("Error in [%s]: %s\n", zSql, sqlite3_errmsg(p->db));
9187 }else{
9188 sputf(stdout, "%s;\n", zSql);
9189 sputf(stdout, "WARNING: writing to an imposter table will corrupt"
9190 " the \"%s\" %s!\n", azArg[1], isWO ? "table" : "index");
9192 }else{
9193 eputf("SQLITE_TESTCTRL_IMPOSTER returns %d\n", rc);
9194 rc = 1;
9196 sqlite3_free(zSql);
9197 }else
9198 #endif /* !defined(SQLITE_OMIT_TEST_CONTROL) */
9200 if( c=='i' && cli_strncmp(azArg[0], "intck", n)==0 ){
9201 i64 iArg = 0;
9202 if( nArg==2 ){
9203 iArg = integerValue(azArg[1]);
9204 if( iArg==0 ) iArg = -1;
9206 if( (nArg!=1 && nArg!=2) || iArg<0 ){
9207 eputf("%s","Usage: .intck STEPS_PER_UNLOCK\n");
9208 rc = 1;
9209 goto meta_command_exit;
9211 open_db(p, 0);
9212 rc = intckDatabaseCmd(p, iArg);
9213 }else
9215 #ifdef SQLITE_ENABLE_IOTRACE
9216 if( c=='i' && cli_strncmp(azArg[0], "iotrace", n)==0 ){
9217 SQLITE_API extern void (SQLITE_CDECL *sqlite3IoTrace)(const char*, ...);
9218 if( iotrace && iotrace!=stdout ) fclose(iotrace);
9219 iotrace = 0;
9220 if( nArg<2 ){
9221 sqlite3IoTrace = 0;
9222 }else if( cli_strcmp(azArg[1], "-")==0 ){
9223 sqlite3IoTrace = iotracePrintf;
9224 iotrace = stdout;
9225 }else{
9226 iotrace = fopen(azArg[1], "w");
9227 if( iotrace==0 ){
9228 eputf("Error: cannot open \"%s\"\n", azArg[1]);
9229 sqlite3IoTrace = 0;
9230 rc = 1;
9231 }else{
9232 sqlite3IoTrace = iotracePrintf;
9235 }else
9236 #endif
9238 if( c=='l' && n>=5 && cli_strncmp(azArg[0], "limits", n)==0 ){
9239 static const struct {
9240 const char *zLimitName; /* Name of a limit */
9241 int limitCode; /* Integer code for that limit */
9242 } aLimit[] = {
9243 { "length", SQLITE_LIMIT_LENGTH },
9244 { "sql_length", SQLITE_LIMIT_SQL_LENGTH },
9245 { "column", SQLITE_LIMIT_COLUMN },
9246 { "expr_depth", SQLITE_LIMIT_EXPR_DEPTH },
9247 { "compound_select", SQLITE_LIMIT_COMPOUND_SELECT },
9248 { "vdbe_op", SQLITE_LIMIT_VDBE_OP },
9249 { "function_arg", SQLITE_LIMIT_FUNCTION_ARG },
9250 { "attached", SQLITE_LIMIT_ATTACHED },
9251 { "like_pattern_length", SQLITE_LIMIT_LIKE_PATTERN_LENGTH },
9252 { "variable_number", SQLITE_LIMIT_VARIABLE_NUMBER },
9253 { "trigger_depth", SQLITE_LIMIT_TRIGGER_DEPTH },
9254 { "worker_threads", SQLITE_LIMIT_WORKER_THREADS },
9256 int i, n2;
9257 open_db(p, 0);
9258 if( nArg==1 ){
9259 for(i=0; i<ArraySize(aLimit); i++){
9260 sputf(stdout, "%20s %d\n", aLimit[i].zLimitName,
9261 sqlite3_limit(p->db, aLimit[i].limitCode, -1));
9263 }else if( nArg>3 ){
9264 eputz("Usage: .limit NAME ?NEW-VALUE?\n");
9265 rc = 1;
9266 goto meta_command_exit;
9267 }else{
9268 int iLimit = -1;
9269 n2 = strlen30(azArg[1]);
9270 for(i=0; i<ArraySize(aLimit); i++){
9271 if( sqlite3_strnicmp(aLimit[i].zLimitName, azArg[1], n2)==0 ){
9272 if( iLimit<0 ){
9273 iLimit = i;
9274 }else{
9275 eputf("ambiguous limit: \"%s\"\n", azArg[1]);
9276 rc = 1;
9277 goto meta_command_exit;
9281 if( iLimit<0 ){
9282 eputf("unknown limit: \"%s\"\n"
9283 "enter \".limits\" with no arguments for a list.\n",
9284 azArg[1]);
9285 rc = 1;
9286 goto meta_command_exit;
9288 if( nArg==3 ){
9289 sqlite3_limit(p->db, aLimit[iLimit].limitCode,
9290 (int)integerValue(azArg[2]));
9292 sputf(stdout, "%20s %d\n", aLimit[iLimit].zLimitName,
9293 sqlite3_limit(p->db, aLimit[iLimit].limitCode, -1));
9295 }else
9297 if( c=='l' && n>2 && cli_strncmp(azArg[0], "lint", n)==0 ){
9298 open_db(p, 0);
9299 lintDotCommand(p, azArg, nArg);
9300 }else
9302 #if !defined(SQLITE_OMIT_LOAD_EXTENSION) && !defined(SQLITE_SHELL_FIDDLE)
9303 if( c=='l' && cli_strncmp(azArg[0], "load", n)==0 ){
9304 const char *zFile, *zProc;
9305 char *zErrMsg = 0;
9306 failIfSafeMode(p, "cannot run .load in safe mode");
9307 if( nArg<2 || azArg[1][0]==0 ){
9308 /* Must have a non-empty FILE. (Will not load self.) */
9309 eputz("Usage: .load FILE ?ENTRYPOINT?\n");
9310 rc = 1;
9311 goto meta_command_exit;
9313 zFile = azArg[1];
9314 zProc = nArg>=3 ? azArg[2] : 0;
9315 open_db(p, 0);
9316 rc = sqlite3_load_extension(p->db, zFile, zProc, &zErrMsg);
9317 if( rc!=SQLITE_OK ){
9318 eputf("Error: %s\n", zErrMsg);
9319 sqlite3_free(zErrMsg);
9320 rc = 1;
9322 }else
9323 #endif
9325 if( c=='l' && cli_strncmp(azArg[0], "log", n)==0 ){
9326 if( nArg!=2 ){
9327 eputz("Usage: .log FILENAME\n");
9328 rc = 1;
9329 }else{
9330 const char *zFile = azArg[1];
9331 if( p->bSafeMode
9332 && cli_strcmp(zFile,"on")!=0
9333 && cli_strcmp(zFile,"off")!=0
9335 sputz(stdout, "cannot set .log to anything other"
9336 " than \"on\" or \"off\"\n");
9337 zFile = "off";
9339 output_file_close(p->pLog);
9340 if( cli_strcmp(zFile,"on")==0 ) zFile = "stdout";
9341 p->pLog = output_file_open(zFile, 0);
9343 }else
9345 if( c=='m' && cli_strncmp(azArg[0], "mode", n)==0 ){
9346 const char *zMode = 0;
9347 const char *zTabname = 0;
9348 int i, n2;
9349 ColModeOpts cmOpts = ColModeOpts_default;
9350 for(i=1; i<nArg; i++){
9351 const char *z = azArg[i];
9352 if( optionMatch(z,"wrap") && i+1<nArg ){
9353 cmOpts.iWrap = integerValue(azArg[++i]);
9354 }else if( optionMatch(z,"ww") ){
9355 cmOpts.bWordWrap = 1;
9356 }else if( optionMatch(z,"wordwrap") && i+1<nArg ){
9357 cmOpts.bWordWrap = (u8)booleanValue(azArg[++i]);
9358 }else if( optionMatch(z,"quote") ){
9359 cmOpts.bQuote = 1;
9360 }else if( optionMatch(z,"noquote") ){
9361 cmOpts.bQuote = 0;
9362 }else if( zMode==0 ){
9363 zMode = z;
9364 /* Apply defaults for qbox pseudo-mode. If that
9365 * overwrites already-set values, user was informed of this.
9367 if( cli_strcmp(z, "qbox")==0 ){
9368 ColModeOpts cmo = ColModeOpts_default_qbox;
9369 zMode = "box";
9370 cmOpts = cmo;
9372 }else if( zTabname==0 ){
9373 zTabname = z;
9374 }else if( z[0]=='-' ){
9375 eputf("unknown option: %s\n", z);
9376 eputz("options:\n"
9377 " --noquote\n"
9378 " --quote\n"
9379 " --wordwrap on/off\n"
9380 " --wrap N\n"
9381 " --ww\n");
9382 rc = 1;
9383 goto meta_command_exit;
9384 }else{
9385 eputf("extra argument: \"%s\"\n", z);
9386 rc = 1;
9387 goto meta_command_exit;
9390 if( zMode==0 ){
9391 if( p->mode==MODE_Column
9392 || (p->mode>=MODE_Markdown && p->mode<=MODE_Box)
9394 oputf("current output mode: %s --wrap %d --wordwrap %s --%squote\n",
9395 modeDescr[p->mode], p->cmOpts.iWrap,
9396 p->cmOpts.bWordWrap ? "on" : "off",
9397 p->cmOpts.bQuote ? "" : "no");
9398 }else{
9399 oputf("current output mode: %s\n", modeDescr[p->mode]);
9401 zMode = modeDescr[p->mode];
9403 n2 = strlen30(zMode);
9404 if( cli_strncmp(zMode,"lines",n2)==0 ){
9405 p->mode = MODE_Line;
9406 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
9407 }else if( cli_strncmp(zMode,"columns",n2)==0 ){
9408 p->mode = MODE_Column;
9409 if( (p->shellFlgs & SHFLG_HeaderSet)==0 ){
9410 p->showHeader = 1;
9412 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
9413 p->cmOpts = cmOpts;
9414 }else if( cli_strncmp(zMode,"list",n2)==0 ){
9415 p->mode = MODE_List;
9416 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Column);
9417 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
9418 }else if( cli_strncmp(zMode,"html",n2)==0 ){
9419 p->mode = MODE_Html;
9420 }else if( cli_strncmp(zMode,"tcl",n2)==0 ){
9421 p->mode = MODE_Tcl;
9422 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Space);
9423 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
9424 }else if( cli_strncmp(zMode,"csv",n2)==0 ){
9425 p->mode = MODE_Csv;
9426 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Comma);
9427 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_CrLf);
9428 }else if( cli_strncmp(zMode,"tabs",n2)==0 ){
9429 p->mode = MODE_List;
9430 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Tab);
9431 }else if( cli_strncmp(zMode,"insert",n2)==0 ){
9432 p->mode = MODE_Insert;
9433 set_table_name(p, zTabname ? zTabname : "table");
9434 }else if( cli_strncmp(zMode,"quote",n2)==0 ){
9435 p->mode = MODE_Quote;
9436 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Comma);
9437 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Row);
9438 }else if( cli_strncmp(zMode,"ascii",n2)==0 ){
9439 p->mode = MODE_Ascii;
9440 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Unit);
9441 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_Record);
9442 }else if( cli_strncmp(zMode,"markdown",n2)==0 ){
9443 p->mode = MODE_Markdown;
9444 p->cmOpts = cmOpts;
9445 }else if( cli_strncmp(zMode,"table",n2)==0 ){
9446 p->mode = MODE_Table;
9447 p->cmOpts = cmOpts;
9448 }else if( cli_strncmp(zMode,"box",n2)==0 ){
9449 p->mode = MODE_Box;
9450 p->cmOpts = cmOpts;
9451 }else if( cli_strncmp(zMode,"count",n2)==0 ){
9452 p->mode = MODE_Count;
9453 }else if( cli_strncmp(zMode,"off",n2)==0 ){
9454 p->mode = MODE_Off;
9455 }else if( cli_strncmp(zMode,"json",n2)==0 ){
9456 p->mode = MODE_Json;
9457 }else{
9458 eputz("Error: mode should be one of: "
9459 "ascii box column csv html insert json line list markdown "
9460 "qbox quote table tabs tcl\n");
9461 rc = 1;
9463 p->cMode = p->mode;
9464 }else
9466 #ifndef SQLITE_SHELL_FIDDLE
9467 if( c=='n' && cli_strcmp(azArg[0], "nonce")==0 ){
9468 if( nArg!=2 ){
9469 eputz("Usage: .nonce NONCE\n");
9470 rc = 1;
9471 }else if( p->zNonce==0 || cli_strcmp(azArg[1],p->zNonce)!=0 ){
9472 eputf("line %d: incorrect nonce: \"%s\"\n",
9473 p->lineno, azArg[1]);
9474 exit(1);
9475 }else{
9476 p->bSafeMode = 0;
9477 return 0; /* Return immediately to bypass the safe mode reset
9478 ** at the end of this procedure */
9480 }else
9481 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
9483 if( c=='n' && cli_strncmp(azArg[0], "nullvalue", n)==0 ){
9484 if( nArg==2 ){
9485 sqlite3_snprintf(sizeof(p->nullValue), p->nullValue,
9486 "%.*s", (int)ArraySize(p->nullValue)-1, azArg[1]);
9487 }else{
9488 eputz("Usage: .nullvalue STRING\n");
9489 rc = 1;
9491 }else
9493 if( c=='o' && cli_strncmp(azArg[0], "open", n)==0 && n>=2 ){
9494 const char *zFN = 0; /* Pointer to constant filename */
9495 char *zNewFilename = 0; /* Name of the database file to open */
9496 int iName = 1; /* Index in azArg[] of the filename */
9497 int newFlag = 0; /* True to delete file before opening */
9498 int openMode = SHELL_OPEN_UNSPEC;
9500 /* Check for command-line arguments */
9501 for(iName=1; iName<nArg; iName++){
9502 const char *z = azArg[iName];
9503 #ifndef SQLITE_SHELL_FIDDLE
9504 if( optionMatch(z,"new") ){
9505 newFlag = 1;
9506 #ifdef SQLITE_HAVE_ZLIB
9507 }else if( optionMatch(z, "zip") ){
9508 openMode = SHELL_OPEN_ZIPFILE;
9509 #endif
9510 }else if( optionMatch(z, "append") ){
9511 openMode = SHELL_OPEN_APPENDVFS;
9512 }else if( optionMatch(z, "readonly") ){
9513 openMode = SHELL_OPEN_READONLY;
9514 }else if( optionMatch(z, "nofollow") ){
9515 p->openFlags |= SQLITE_OPEN_NOFOLLOW;
9516 #ifndef SQLITE_OMIT_DESERIALIZE
9517 }else if( optionMatch(z, "deserialize") ){
9518 openMode = SHELL_OPEN_DESERIALIZE;
9519 }else if( optionMatch(z, "hexdb") ){
9520 openMode = SHELL_OPEN_HEXDB;
9521 }else if( optionMatch(z, "maxsize") && iName+1<nArg ){
9522 p->szMax = integerValue(azArg[++iName]);
9523 #endif /* SQLITE_OMIT_DESERIALIZE */
9524 }else
9525 #endif /* !SQLITE_SHELL_FIDDLE */
9526 if( z[0]=='-' ){
9527 eputf("unknown option: %s\n", z);
9528 rc = 1;
9529 goto meta_command_exit;
9530 }else if( zFN ){
9531 eputf("extra argument: \"%s\"\n", z);
9532 rc = 1;
9533 goto meta_command_exit;
9534 }else{
9535 zFN = z;
9539 /* Close the existing database */
9540 session_close_all(p, -1);
9541 close_db(p->db);
9542 p->db = 0;
9543 p->pAuxDb->zDbFilename = 0;
9544 sqlite3_free(p->pAuxDb->zFreeOnClose);
9545 p->pAuxDb->zFreeOnClose = 0;
9546 p->openMode = openMode;
9547 p->openFlags = 0;
9548 p->szMax = 0;
9550 /* If a filename is specified, try to open it first */
9551 if( zFN || p->openMode==SHELL_OPEN_HEXDB ){
9552 if( newFlag && zFN && !p->bSafeMode ) shellDeleteFile(zFN);
9553 #ifndef SQLITE_SHELL_FIDDLE
9554 if( p->bSafeMode
9555 && p->openMode!=SHELL_OPEN_HEXDB
9556 && zFN
9557 && cli_strcmp(zFN,":memory:")!=0
9559 failIfSafeMode(p, "cannot open disk-based database files in safe mode");
9561 #else
9562 /* WASM mode has its own sandboxed pseudo-filesystem. */
9563 #endif
9564 if( zFN ){
9565 zNewFilename = sqlite3_mprintf("%s", zFN);
9566 shell_check_oom(zNewFilename);
9567 }else{
9568 zNewFilename = 0;
9570 p->pAuxDb->zDbFilename = zNewFilename;
9571 open_db(p, OPEN_DB_KEEPALIVE);
9572 if( p->db==0 ){
9573 eputf("Error: cannot open '%s'\n", zNewFilename);
9574 sqlite3_free(zNewFilename);
9575 }else{
9576 p->pAuxDb->zFreeOnClose = zNewFilename;
9579 if( p->db==0 ){
9580 /* As a fall-back open a TEMP database */
9581 p->pAuxDb->zDbFilename = 0;
9582 open_db(p, 0);
9584 }else
9586 #ifndef SQLITE_SHELL_FIDDLE
9587 if( (c=='o'
9588 && (cli_strncmp(azArg[0], "output", n)==0
9589 || cli_strncmp(azArg[0], "once", n)==0))
9590 || (c=='e' && n==5 && cli_strcmp(azArg[0],"excel")==0)
9592 char *zFile = 0;
9593 int bTxtMode = 0;
9594 int i;
9595 int eMode = 0;
9596 int bOnce = 0; /* 0: .output, 1: .once, 2: .excel */
9597 static const char *zBomUtf8 = "\xef\xbb\xbf";
9598 const char *zBom = 0;
9600 failIfSafeMode(p, "cannot run .%s in safe mode", azArg[0]);
9601 if( c=='e' ){
9602 eMode = 'x';
9603 bOnce = 2;
9604 }else if( cli_strncmp(azArg[0],"once",n)==0 ){
9605 bOnce = 1;
9607 for(i=1; i<nArg; i++){
9608 char *z = azArg[i];
9609 if( z[0]=='-' ){
9610 if( z[1]=='-' ) z++;
9611 if( cli_strcmp(z,"-bom")==0 ){
9612 zBom = zBomUtf8;
9613 }else if( c!='e' && cli_strcmp(z,"-x")==0 ){
9614 eMode = 'x'; /* spreadsheet */
9615 }else if( c!='e' && cli_strcmp(z,"-e")==0 ){
9616 eMode = 'e'; /* text editor */
9617 }else{
9618 oputf("ERROR: unknown option: \"%s\". Usage:\n", azArg[i]);
9619 showHelp(p->out, azArg[0]);
9620 rc = 1;
9621 goto meta_command_exit;
9623 }else if( zFile==0 && eMode!='e' && eMode!='x' ){
9624 zFile = sqlite3_mprintf("%s", z);
9625 if( zFile && zFile[0]=='|' ){
9626 while( i+1<nArg ) zFile = sqlite3_mprintf("%z %s", zFile, azArg[++i]);
9627 break;
9629 }else{
9630 oputf("ERROR: extra parameter: \"%s\". Usage:\n", azArg[i]);
9631 showHelp(p->out, azArg[0]);
9632 rc = 1;
9633 sqlite3_free(zFile);
9634 goto meta_command_exit;
9637 if( zFile==0 ){
9638 zFile = sqlite3_mprintf("stdout");
9640 if( bOnce ){
9641 p->outCount = 2;
9642 }else{
9643 p->outCount = 0;
9645 output_reset(p);
9646 #ifndef SQLITE_NOHAVE_SYSTEM
9647 if( eMode=='e' || eMode=='x' ){
9648 p->doXdgOpen = 1;
9649 outputModePush(p);
9650 if( eMode=='x' ){
9651 /* spreadsheet mode. Output as CSV. */
9652 newTempFile(p, "csv");
9653 ShellClearFlag(p, SHFLG_Echo);
9654 p->mode = MODE_Csv;
9655 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator, SEP_Comma);
9656 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator, SEP_CrLf);
9657 }else{
9658 /* text editor mode */
9659 newTempFile(p, "txt");
9660 bTxtMode = 1;
9662 sqlite3_free(zFile);
9663 zFile = sqlite3_mprintf("%s", p->zTempFile);
9665 #endif /* SQLITE_NOHAVE_SYSTEM */
9666 shell_check_oom(zFile);
9667 if( zFile[0]=='|' ){
9668 #ifdef SQLITE_OMIT_POPEN
9669 eputz("Error: pipes are not supported in this OS\n");
9670 rc = 1;
9671 output_redir(p, stdout);
9672 #else
9673 FILE *pfPipe = popen(zFile + 1, "w");
9674 if( pfPipe==0 ){
9675 eputf("Error: cannot open pipe \"%s\"\n", zFile + 1);
9676 rc = 1;
9677 }else{
9678 output_redir(p, pfPipe);
9679 if( zBom ) oputz(zBom);
9680 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
9682 #endif
9683 }else{
9684 FILE *pfFile = output_file_open(zFile, bTxtMode);
9685 if( pfFile==0 ){
9686 if( cli_strcmp(zFile,"off")!=0 ){
9687 eputf("Error: cannot write to \"%s\"\n", zFile);
9689 rc = 1;
9690 } else {
9691 output_redir(p, pfFile);
9692 if( zBom ) oputz(zBom);
9693 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
9696 sqlite3_free(zFile);
9697 }else
9698 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
9700 if( c=='p' && n>=3 && cli_strncmp(azArg[0], "parameter", n)==0 ){
9701 open_db(p,0);
9702 if( nArg<=1 ) goto parameter_syntax_error;
9704 /* .parameter clear
9705 ** Clear all bind parameters by dropping the TEMP table that holds them.
9707 if( nArg==2 && cli_strcmp(azArg[1],"clear")==0 ){
9708 sqlite3_exec(p->db, "DROP TABLE IF EXISTS temp.sqlite_parameters;",
9709 0, 0, 0);
9710 }else
9712 /* .parameter list
9713 ** List all bind parameters.
9715 if( nArg==2 && cli_strcmp(azArg[1],"list")==0 ){
9716 sqlite3_stmt *pStmt = 0;
9717 int rx;
9718 int len = 0;
9719 rx = sqlite3_prepare_v2(p->db,
9720 "SELECT max(length(key)) "
9721 "FROM temp.sqlite_parameters;", -1, &pStmt, 0);
9722 if( rx==SQLITE_OK && sqlite3_step(pStmt)==SQLITE_ROW ){
9723 len = sqlite3_column_int(pStmt, 0);
9724 if( len>40 ) len = 40;
9726 sqlite3_finalize(pStmt);
9727 pStmt = 0;
9728 if( len ){
9729 rx = sqlite3_prepare_v2(p->db,
9730 "SELECT key, quote(value) "
9731 "FROM temp.sqlite_parameters;", -1, &pStmt, 0);
9732 while( rx==SQLITE_OK && sqlite3_step(pStmt)==SQLITE_ROW ){
9733 oputf("%-*s %s\n", len, sqlite3_column_text(pStmt,0),
9734 sqlite3_column_text(pStmt,1));
9736 sqlite3_finalize(pStmt);
9738 }else
9740 /* .parameter init
9741 ** Make sure the TEMP table used to hold bind parameters exists.
9742 ** Create it if necessary.
9744 if( nArg==2 && cli_strcmp(azArg[1],"init")==0 ){
9745 bind_table_init(p);
9746 }else
9748 /* .parameter set NAME VALUE
9749 ** Set or reset a bind parameter. NAME should be the full parameter
9750 ** name exactly as it appears in the query. (ex: $abc, @def). The
9751 ** VALUE can be in either SQL literal notation, or if not it will be
9752 ** understood to be a text string.
9754 if( nArg==4 && cli_strcmp(azArg[1],"set")==0 ){
9755 int rx;
9756 char *zSql;
9757 sqlite3_stmt *pStmt;
9758 const char *zKey = azArg[2];
9759 const char *zValue = azArg[3];
9760 bind_table_init(p);
9761 zSql = sqlite3_mprintf(
9762 "REPLACE INTO temp.sqlite_parameters(key,value)"
9763 "VALUES(%Q,%s);", zKey, zValue);
9764 shell_check_oom(zSql);
9765 pStmt = 0;
9766 rx = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
9767 sqlite3_free(zSql);
9768 if( rx!=SQLITE_OK ){
9769 sqlite3_finalize(pStmt);
9770 pStmt = 0;
9771 zSql = sqlite3_mprintf(
9772 "REPLACE INTO temp.sqlite_parameters(key,value)"
9773 "VALUES(%Q,%Q);", zKey, zValue);
9774 shell_check_oom(zSql);
9775 rx = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
9776 sqlite3_free(zSql);
9777 if( rx!=SQLITE_OK ){
9778 oputf("Error: %s\n", sqlite3_errmsg(p->db));
9779 sqlite3_finalize(pStmt);
9780 pStmt = 0;
9781 rc = 1;
9784 sqlite3_step(pStmt);
9785 sqlite3_finalize(pStmt);
9786 }else
9788 /* .parameter unset NAME
9789 ** Remove the NAME binding from the parameter binding table, if it
9790 ** exists.
9792 if( nArg==3 && cli_strcmp(azArg[1],"unset")==0 ){
9793 char *zSql = sqlite3_mprintf(
9794 "DELETE FROM temp.sqlite_parameters WHERE key=%Q", azArg[2]);
9795 shell_check_oom(zSql);
9796 sqlite3_exec(p->db, zSql, 0, 0, 0);
9797 sqlite3_free(zSql);
9798 }else
9799 /* If no command name matches, show a syntax error */
9800 parameter_syntax_error:
9801 showHelp(p->out, "parameter");
9802 }else
9804 if( c=='p' && n>=3 && cli_strncmp(azArg[0], "print", n)==0 ){
9805 int i;
9806 for(i=1; i<nArg; i++){
9807 if( i>1 ) oputz(" ");
9808 oputz(azArg[i]);
9810 oputz("\n");
9811 }else
9813 #ifndef SQLITE_OMIT_PROGRESS_CALLBACK
9814 if( c=='p' && n>=3 && cli_strncmp(azArg[0], "progress", n)==0 ){
9815 int i;
9816 int nn = 0;
9817 p->flgProgress = 0;
9818 p->mxProgress = 0;
9819 p->nProgress = 0;
9820 for(i=1; i<nArg; i++){
9821 const char *z = azArg[i];
9822 if( z[0]=='-' ){
9823 z++;
9824 if( z[0]=='-' ) z++;
9825 if( cli_strcmp(z,"quiet")==0 || cli_strcmp(z,"q")==0 ){
9826 p->flgProgress |= SHELL_PROGRESS_QUIET;
9827 continue;
9829 if( cli_strcmp(z,"reset")==0 ){
9830 p->flgProgress |= SHELL_PROGRESS_RESET;
9831 continue;
9833 if( cli_strcmp(z,"once")==0 ){
9834 p->flgProgress |= SHELL_PROGRESS_ONCE;
9835 continue;
9837 if( cli_strcmp(z,"limit")==0 ){
9838 if( i+1>=nArg ){
9839 eputz("Error: missing argument on --limit\n");
9840 rc = 1;
9841 goto meta_command_exit;
9842 }else{
9843 p->mxProgress = (int)integerValue(azArg[++i]);
9845 continue;
9847 eputf("Error: unknown option: \"%s\"\n", azArg[i]);
9848 rc = 1;
9849 goto meta_command_exit;
9850 }else{
9851 nn = (int)integerValue(z);
9854 open_db(p, 0);
9855 sqlite3_progress_handler(p->db, nn, progress_handler, p);
9856 }else
9857 #endif /* SQLITE_OMIT_PROGRESS_CALLBACK */
9859 if( c=='p' && cli_strncmp(azArg[0], "prompt", n)==0 ){
9860 if( nArg >= 2) {
9861 shell_strncpy(mainPrompt,azArg[1],(int)ArraySize(mainPrompt)-1);
9863 if( nArg >= 3) {
9864 shell_strncpy(continuePrompt,azArg[2],(int)ArraySize(continuePrompt)-1);
9866 }else
9868 #ifndef SQLITE_SHELL_FIDDLE
9869 if( c=='q' && cli_strncmp(azArg[0], "quit", n)==0 ){
9870 rc = 2;
9871 }else
9872 #endif
9874 #ifndef SQLITE_SHELL_FIDDLE
9875 if( c=='r' && n>=3 && cli_strncmp(azArg[0], "read", n)==0 ){
9876 FILE *inSaved = p->in;
9877 int savedLineno = p->lineno;
9878 failIfSafeMode(p, "cannot run .read in safe mode");
9879 if( nArg!=2 ){
9880 eputz("Usage: .read FILE\n");
9881 rc = 1;
9882 goto meta_command_exit;
9884 if( azArg[1][0]=='|' ){
9885 #ifdef SQLITE_OMIT_POPEN
9886 eputz("Error: pipes are not supported in this OS\n");
9887 rc = 1;
9888 p->out = stdout;
9889 #else
9890 p->in = popen(azArg[1]+1, "r");
9891 if( p->in==0 ){
9892 eputf("Error: cannot open \"%s\"\n", azArg[1]);
9893 rc = 1;
9894 }else{
9895 rc = process_input(p);
9896 pclose(p->in);
9898 #endif
9899 }else if( (p->in = openChrSource(azArg[1]))==0 ){
9900 eputf("Error: cannot open \"%s\"\n", azArg[1]);
9901 rc = 1;
9902 }else{
9903 rc = process_input(p);
9904 fclose(p->in);
9906 p->in = inSaved;
9907 p->lineno = savedLineno;
9908 }else
9909 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
9911 #ifndef SQLITE_SHELL_FIDDLE
9912 if( c=='r' && n>=3 && cli_strncmp(azArg[0], "restore", n)==0 ){
9913 const char *zSrcFile;
9914 const char *zDb;
9915 sqlite3 *pSrc;
9916 sqlite3_backup *pBackup;
9917 int nTimeout = 0;
9919 failIfSafeMode(p, "cannot run .restore in safe mode");
9920 if( nArg==2 ){
9921 zSrcFile = azArg[1];
9922 zDb = "main";
9923 }else if( nArg==3 ){
9924 zSrcFile = azArg[2];
9925 zDb = azArg[1];
9926 }else{
9927 eputz("Usage: .restore ?DB? FILE\n");
9928 rc = 1;
9929 goto meta_command_exit;
9931 rc = sqlite3_open(zSrcFile, &pSrc);
9932 if( rc!=SQLITE_OK ){
9933 eputf("Error: cannot open \"%s\"\n", zSrcFile);
9934 close_db(pSrc);
9935 return 1;
9937 open_db(p, 0);
9938 pBackup = sqlite3_backup_init(p->db, zDb, pSrc, "main");
9939 if( pBackup==0 ){
9940 eputf("Error: %s\n", sqlite3_errmsg(p->db));
9941 close_db(pSrc);
9942 return 1;
9944 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK
9945 || rc==SQLITE_BUSY ){
9946 if( rc==SQLITE_BUSY ){
9947 if( nTimeout++ >= 3 ) break;
9948 sqlite3_sleep(100);
9951 sqlite3_backup_finish(pBackup);
9952 if( rc==SQLITE_DONE ){
9953 rc = 0;
9954 }else if( rc==SQLITE_BUSY || rc==SQLITE_LOCKED ){
9955 eputz("Error: source database is busy\n");
9956 rc = 1;
9957 }else{
9958 eputf("Error: %s\n", sqlite3_errmsg(p->db));
9959 rc = 1;
9961 close_db(pSrc);
9962 }else
9963 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
9965 if( c=='s' && cli_strncmp(azArg[0], "scanstats", n)==0 ){
9966 if( nArg==2 ){
9967 if( cli_strcmp(azArg[1], "vm")==0 ){
9968 p->scanstatsOn = 3;
9969 }else
9970 if( cli_strcmp(azArg[1], "est")==0 ){
9971 p->scanstatsOn = 2;
9972 }else{
9973 p->scanstatsOn = (u8)booleanValue(azArg[1]);
9975 open_db(p, 0);
9976 sqlite3_db_config(
9977 p->db, SQLITE_DBCONFIG_STMT_SCANSTATUS, p->scanstatsOn, (int*)0
9979 #if !defined(SQLITE_ENABLE_STMT_SCANSTATUS)
9980 eputz("Warning: .scanstats not available in this build.\n");
9981 #elif !defined(SQLITE_ENABLE_BYTECODE_VTAB)
9982 if( p->scanstatsOn==3 ){
9983 eputz("Warning: \".scanstats vm\" not available in this build.\n");
9985 #endif
9986 }else{
9987 eputz("Usage: .scanstats on|off|est\n");
9988 rc = 1;
9990 }else
9992 if( c=='s' && cli_strncmp(azArg[0], "schema", n)==0 ){
9993 ShellText sSelect;
9994 ShellState data;
9995 char *zErrMsg = 0;
9996 const char *zDiv = "(";
9997 const char *zName = 0;
9998 int iSchema = 0;
9999 int bDebug = 0;
10000 int bNoSystemTabs = 0;
10001 int ii;
10003 open_db(p, 0);
10004 memcpy(&data, p, sizeof(data));
10005 data.showHeader = 0;
10006 data.cMode = data.mode = MODE_Semi;
10007 initText(&sSelect);
10008 for(ii=1; ii<nArg; ii++){
10009 if( optionMatch(azArg[ii],"indent") ){
10010 data.cMode = data.mode = MODE_Pretty;
10011 }else if( optionMatch(azArg[ii],"debug") ){
10012 bDebug = 1;
10013 }else if( optionMatch(azArg[ii],"nosys") ){
10014 bNoSystemTabs = 1;
10015 }else if( azArg[ii][0]=='-' ){
10016 eputf("Unknown option: \"%s\"\n", azArg[ii]);
10017 rc = 1;
10018 goto meta_command_exit;
10019 }else if( zName==0 ){
10020 zName = azArg[ii];
10021 }else{
10022 eputz("Usage: .schema ?--indent? ?--nosys? ?LIKE-PATTERN?\n");
10023 rc = 1;
10024 goto meta_command_exit;
10027 if( zName!=0 ){
10028 int isSchema = sqlite3_strlike(zName, "sqlite_master", '\\')==0
10029 || sqlite3_strlike(zName, "sqlite_schema", '\\')==0
10030 || sqlite3_strlike(zName,"sqlite_temp_master", '\\')==0
10031 || sqlite3_strlike(zName,"sqlite_temp_schema", '\\')==0;
10032 if( isSchema ){
10033 char *new_argv[2], *new_colv[2];
10034 new_argv[0] = sqlite3_mprintf(
10035 "CREATE TABLE %s (\n"
10036 " type text,\n"
10037 " name text,\n"
10038 " tbl_name text,\n"
10039 " rootpage integer,\n"
10040 " sql text\n"
10041 ")", zName);
10042 shell_check_oom(new_argv[0]);
10043 new_argv[1] = 0;
10044 new_colv[0] = "sql";
10045 new_colv[1] = 0;
10046 callback(&data, 1, new_argv, new_colv);
10047 sqlite3_free(new_argv[0]);
10050 if( zDiv ){
10051 sqlite3_stmt *pStmt = 0;
10052 rc = sqlite3_prepare_v2(p->db, "SELECT name FROM pragma_database_list",
10053 -1, &pStmt, 0);
10054 if( rc ){
10055 eputf("Error: %s\n", sqlite3_errmsg(p->db));
10056 sqlite3_finalize(pStmt);
10057 rc = 1;
10058 goto meta_command_exit;
10060 appendText(&sSelect, "SELECT sql FROM", 0);
10061 iSchema = 0;
10062 while( sqlite3_step(pStmt)==SQLITE_ROW ){
10063 const char *zDb = (const char*)sqlite3_column_text(pStmt, 0);
10064 char zScNum[30];
10065 sqlite3_snprintf(sizeof(zScNum), zScNum, "%d", ++iSchema);
10066 appendText(&sSelect, zDiv, 0);
10067 zDiv = " UNION ALL ";
10068 appendText(&sSelect, "SELECT shell_add_schema(sql,", 0);
10069 if( sqlite3_stricmp(zDb, "main")!=0 ){
10070 appendText(&sSelect, zDb, '\'');
10071 }else{
10072 appendText(&sSelect, "NULL", 0);
10074 appendText(&sSelect, ",name) AS sql, type, tbl_name, name, rowid,", 0);
10075 appendText(&sSelect, zScNum, 0);
10076 appendText(&sSelect, " AS snum, ", 0);
10077 appendText(&sSelect, zDb, '\'');
10078 appendText(&sSelect, " AS sname FROM ", 0);
10079 appendText(&sSelect, zDb, quoteChar(zDb));
10080 appendText(&sSelect, ".sqlite_schema", 0);
10082 sqlite3_finalize(pStmt);
10083 #ifndef SQLITE_OMIT_INTROSPECTION_PRAGMAS
10084 if( zName ){
10085 appendText(&sSelect,
10086 " UNION ALL SELECT shell_module_schema(name),"
10087 " 'table', name, name, name, 9e+99, 'main' FROM pragma_module_list",
10090 #endif
10091 appendText(&sSelect, ") WHERE ", 0);
10092 if( zName ){
10093 char *zQarg = sqlite3_mprintf("%Q", zName);
10094 int bGlob;
10095 shell_check_oom(zQarg);
10096 bGlob = strchr(zName, '*') != 0 || strchr(zName, '?') != 0 ||
10097 strchr(zName, '[') != 0;
10098 if( strchr(zName, '.') ){
10099 appendText(&sSelect, "lower(printf('%s.%s',sname,tbl_name))", 0);
10100 }else{
10101 appendText(&sSelect, "lower(tbl_name)", 0);
10103 appendText(&sSelect, bGlob ? " GLOB " : " LIKE ", 0);
10104 appendText(&sSelect, zQarg, 0);
10105 if( !bGlob ){
10106 appendText(&sSelect, " ESCAPE '\\' ", 0);
10108 appendText(&sSelect, " AND ", 0);
10109 sqlite3_free(zQarg);
10111 if( bNoSystemTabs ){
10112 appendText(&sSelect, "name NOT LIKE 'sqlite_%%' AND ", 0);
10114 appendText(&sSelect, "sql IS NOT NULL"
10115 " ORDER BY snum, rowid", 0);
10116 if( bDebug ){
10117 oputf("SQL: %s;\n", sSelect.z);
10118 }else{
10119 rc = sqlite3_exec(p->db, sSelect.z, callback, &data, &zErrMsg);
10121 freeText(&sSelect);
10123 if( zErrMsg ){
10124 eputf("Error: %s\n", zErrMsg);
10125 sqlite3_free(zErrMsg);
10126 rc = 1;
10127 }else if( rc != SQLITE_OK ){
10128 eputz("Error: querying schema information\n");
10129 rc = 1;
10130 }else{
10131 rc = 0;
10133 }else
10135 if( (c=='s' && n==11 && cli_strncmp(azArg[0], "selecttrace", n)==0)
10136 || (c=='t' && n==9 && cli_strncmp(azArg[0], "treetrace", n)==0)
10138 unsigned int x = nArg>=2? (unsigned int)integerValue(azArg[1]) : 0xffffffff;
10139 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 1, &x);
10140 }else
10142 #if defined(SQLITE_ENABLE_SESSION)
10143 if( c=='s' && cli_strncmp(azArg[0],"session",n)==0 && n>=3 ){
10144 struct AuxDb *pAuxDb = p->pAuxDb;
10145 OpenSession *pSession = &pAuxDb->aSession[0];
10146 char **azCmd = &azArg[1];
10147 int iSes = 0;
10148 int nCmd = nArg - 1;
10149 int i;
10150 if( nArg<=1 ) goto session_syntax_error;
10151 open_db(p, 0);
10152 if( nArg>=3 ){
10153 for(iSes=0; iSes<pAuxDb->nSession; iSes++){
10154 if( cli_strcmp(pAuxDb->aSession[iSes].zName, azArg[1])==0 ) break;
10156 if( iSes<pAuxDb->nSession ){
10157 pSession = &pAuxDb->aSession[iSes];
10158 azCmd++;
10159 nCmd--;
10160 }else{
10161 pSession = &pAuxDb->aSession[0];
10162 iSes = 0;
10166 /* .session attach TABLE
10167 ** Invoke the sqlite3session_attach() interface to attach a particular
10168 ** table so that it is never filtered.
10170 if( cli_strcmp(azCmd[0],"attach")==0 ){
10171 if( nCmd!=2 ) goto session_syntax_error;
10172 if( pSession->p==0 ){
10173 session_not_open:
10174 eputz("ERROR: No sessions are open\n");
10175 }else{
10176 rc = sqlite3session_attach(pSession->p, azCmd[1]);
10177 if( rc ){
10178 eputf("ERROR: sqlite3session_attach() returns %d\n",rc);
10179 rc = 0;
10182 }else
10184 /* .session changeset FILE
10185 ** .session patchset FILE
10186 ** Write a changeset or patchset into a file. The file is overwritten.
10188 if( cli_strcmp(azCmd[0],"changeset")==0
10189 || cli_strcmp(azCmd[0],"patchset")==0
10191 FILE *out = 0;
10192 failIfSafeMode(p, "cannot run \".session %s\" in safe mode", azCmd[0]);
10193 if( nCmd!=2 ) goto session_syntax_error;
10194 if( pSession->p==0 ) goto session_not_open;
10195 out = fopen(azCmd[1], "wb");
10196 if( out==0 ){
10197 eputf("ERROR: cannot open \"%s\" for writing\n",
10198 azCmd[1]);
10199 }else{
10200 int szChng;
10201 void *pChng;
10202 if( azCmd[0][0]=='c' ){
10203 rc = sqlite3session_changeset(pSession->p, &szChng, &pChng);
10204 }else{
10205 rc = sqlite3session_patchset(pSession->p, &szChng, &pChng);
10207 if( rc ){
10208 sputf(stdout, "Error: error code %d\n", rc);
10209 rc = 0;
10211 if( pChng
10212 && fwrite(pChng, szChng, 1, out)!=1 ){
10213 eputf("ERROR: Failed to write entire %d-byte output\n", szChng);
10215 sqlite3_free(pChng);
10216 fclose(out);
10218 }else
10220 /* .session close
10221 ** Close the identified session
10223 if( cli_strcmp(azCmd[0], "close")==0 ){
10224 if( nCmd!=1 ) goto session_syntax_error;
10225 if( pAuxDb->nSession ){
10226 session_close(pSession);
10227 pAuxDb->aSession[iSes] = pAuxDb->aSession[--pAuxDb->nSession];
10229 }else
10231 /* .session enable ?BOOLEAN?
10232 ** Query or set the enable flag
10234 if( cli_strcmp(azCmd[0], "enable")==0 ){
10235 int ii;
10236 if( nCmd>2 ) goto session_syntax_error;
10237 ii = nCmd==1 ? -1 : booleanValue(azCmd[1]);
10238 if( pAuxDb->nSession ){
10239 ii = sqlite3session_enable(pSession->p, ii);
10240 oputf("session %s enable flag = %d\n", pSession->zName, ii);
10242 }else
10244 /* .session filter GLOB ....
10245 ** Set a list of GLOB patterns of table names to be excluded.
10247 if( cli_strcmp(azCmd[0], "filter")==0 ){
10248 int ii, nByte;
10249 if( nCmd<2 ) goto session_syntax_error;
10250 if( pAuxDb->nSession ){
10251 for(ii=0; ii<pSession->nFilter; ii++){
10252 sqlite3_free(pSession->azFilter[ii]);
10254 sqlite3_free(pSession->azFilter);
10255 nByte = sizeof(pSession->azFilter[0])*(nCmd-1);
10256 pSession->azFilter = sqlite3_malloc( nByte );
10257 shell_check_oom( pSession->azFilter );
10258 for(ii=1; ii<nCmd; ii++){
10259 char *x = pSession->azFilter[ii-1] = sqlite3_mprintf("%s", azCmd[ii]);
10260 shell_check_oom(x);
10262 pSession->nFilter = ii-1;
10264 }else
10266 /* .session indirect ?BOOLEAN?
10267 ** Query or set the indirect flag
10269 if( cli_strcmp(azCmd[0], "indirect")==0 ){
10270 int ii;
10271 if( nCmd>2 ) goto session_syntax_error;
10272 ii = nCmd==1 ? -1 : booleanValue(azCmd[1]);
10273 if( pAuxDb->nSession ){
10274 ii = sqlite3session_indirect(pSession->p, ii);
10275 oputf("session %s indirect flag = %d\n", pSession->zName, ii);
10277 }else
10279 /* .session isempty
10280 ** Determine if the session is empty
10282 if( cli_strcmp(azCmd[0], "isempty")==0 ){
10283 int ii;
10284 if( nCmd!=1 ) goto session_syntax_error;
10285 if( pAuxDb->nSession ){
10286 ii = sqlite3session_isempty(pSession->p);
10287 oputf("session %s isempty flag = %d\n", pSession->zName, ii);
10289 }else
10291 /* .session list
10292 ** List all currently open sessions
10294 if( cli_strcmp(azCmd[0],"list")==0 ){
10295 for(i=0; i<pAuxDb->nSession; i++){
10296 oputf("%d %s\n", i, pAuxDb->aSession[i].zName);
10298 }else
10300 /* .session open DB NAME
10301 ** Open a new session called NAME on the attached database DB.
10302 ** DB is normally "main".
10304 if( cli_strcmp(azCmd[0],"open")==0 ){
10305 char *zName;
10306 if( nCmd!=3 ) goto session_syntax_error;
10307 zName = azCmd[2];
10308 if( zName[0]==0 ) goto session_syntax_error;
10309 for(i=0; i<pAuxDb->nSession; i++){
10310 if( cli_strcmp(pAuxDb->aSession[i].zName,zName)==0 ){
10311 eputf("Session \"%s\" already exists\n", zName);
10312 goto meta_command_exit;
10315 if( pAuxDb->nSession>=ArraySize(pAuxDb->aSession) ){
10316 eputf("Maximum of %d sessions\n", ArraySize(pAuxDb->aSession));
10317 goto meta_command_exit;
10319 pSession = &pAuxDb->aSession[pAuxDb->nSession];
10320 rc = sqlite3session_create(p->db, azCmd[1], &pSession->p);
10321 if( rc ){
10322 eputf("Cannot open session: error code=%d\n", rc);
10323 rc = 0;
10324 goto meta_command_exit;
10326 pSession->nFilter = 0;
10327 sqlite3session_table_filter(pSession->p, session_filter, pSession);
10328 pAuxDb->nSession++;
10329 pSession->zName = sqlite3_mprintf("%s", zName);
10330 shell_check_oom(pSession->zName);
10331 }else
10332 /* If no command name matches, show a syntax error */
10333 session_syntax_error:
10334 showHelp(p->out, "session");
10335 }else
10336 #endif
10338 #ifdef SQLITE_DEBUG
10339 /* Undocumented commands for internal testing. Subject to change
10340 ** without notice. */
10341 if( c=='s' && n>=10 && cli_strncmp(azArg[0], "selftest-", 9)==0 ){
10342 if( cli_strncmp(azArg[0]+9, "boolean", n-9)==0 ){
10343 int i, v;
10344 for(i=1; i<nArg; i++){
10345 v = booleanValue(azArg[i]);
10346 oputf("%s: %d 0x%x\n", azArg[i], v, v);
10349 if( cli_strncmp(azArg[0]+9, "integer", n-9)==0 ){
10350 int i; sqlite3_int64 v;
10351 for(i=1; i<nArg; i++){
10352 char zBuf[200];
10353 v = integerValue(azArg[i]);
10354 sqlite3_snprintf(sizeof(zBuf),zBuf,"%s: %lld 0x%llx\n", azArg[i],v,v);
10355 oputz(zBuf);
10358 }else
10359 #endif
10361 if( c=='s' && n>=4 && cli_strncmp(azArg[0],"selftest",n)==0 ){
10362 int bIsInit = 0; /* True to initialize the SELFTEST table */
10363 int bVerbose = 0; /* Verbose output */
10364 int bSelftestExists; /* True if SELFTEST already exists */
10365 int i, k; /* Loop counters */
10366 int nTest = 0; /* Number of tests runs */
10367 int nErr = 0; /* Number of errors seen */
10368 ShellText str; /* Answer for a query */
10369 sqlite3_stmt *pStmt = 0; /* Query against the SELFTEST table */
10371 open_db(p,0);
10372 for(i=1; i<nArg; i++){
10373 const char *z = azArg[i];
10374 if( z[0]=='-' && z[1]=='-' ) z++;
10375 if( cli_strcmp(z,"-init")==0 ){
10376 bIsInit = 1;
10377 }else
10378 if( cli_strcmp(z,"-v")==0 ){
10379 bVerbose++;
10380 }else
10382 eputf("Unknown option \"%s\" on \"%s\"\n", azArg[i], azArg[0]);
10383 eputz("Should be one of: --init -v\n");
10384 rc = 1;
10385 goto meta_command_exit;
10388 if( sqlite3_table_column_metadata(p->db,"main","selftest",0,0,0,0,0,0)
10389 != SQLITE_OK ){
10390 bSelftestExists = 0;
10391 }else{
10392 bSelftestExists = 1;
10394 if( bIsInit ){
10395 createSelftestTable(p);
10396 bSelftestExists = 1;
10398 initText(&str);
10399 appendText(&str, "x", 0);
10400 for(k=bSelftestExists; k>=0; k--){
10401 if( k==1 ){
10402 rc = sqlite3_prepare_v2(p->db,
10403 "SELECT tno,op,cmd,ans FROM selftest ORDER BY tno",
10404 -1, &pStmt, 0);
10405 }else{
10406 rc = sqlite3_prepare_v2(p->db,
10407 "VALUES(0,'memo','Missing SELFTEST table - default checks only',''),"
10408 " (1,'run','PRAGMA integrity_check','ok')",
10409 -1, &pStmt, 0);
10411 if( rc ){
10412 eputz("Error querying the selftest table\n");
10413 rc = 1;
10414 sqlite3_finalize(pStmt);
10415 goto meta_command_exit;
10417 for(i=1; sqlite3_step(pStmt)==SQLITE_ROW; i++){
10418 int tno = sqlite3_column_int(pStmt, 0);
10419 const char *zOp = (const char*)sqlite3_column_text(pStmt, 1);
10420 const char *zSql = (const char*)sqlite3_column_text(pStmt, 2);
10421 const char *zAns = (const char*)sqlite3_column_text(pStmt, 3);
10423 if( zOp==0 ) continue;
10424 if( zSql==0 ) continue;
10425 if( zAns==0 ) continue;
10426 k = 0;
10427 if( bVerbose>0 ){
10428 sputf(stdout, "%d: %s %s\n", tno, zOp, zSql);
10430 if( cli_strcmp(zOp,"memo")==0 ){
10431 oputf("%s\n", zSql);
10432 }else
10433 if( cli_strcmp(zOp,"run")==0 ){
10434 char *zErrMsg = 0;
10435 str.n = 0;
10436 str.z[0] = 0;
10437 rc = sqlite3_exec(p->db, zSql, captureOutputCallback, &str, &zErrMsg);
10438 nTest++;
10439 if( bVerbose ){
10440 oputf("Result: %s\n", str.z);
10442 if( rc || zErrMsg ){
10443 nErr++;
10444 rc = 1;
10445 oputf("%d: error-code-%d: %s\n", tno, rc, zErrMsg);
10446 sqlite3_free(zErrMsg);
10447 }else if( cli_strcmp(zAns,str.z)!=0 ){
10448 nErr++;
10449 rc = 1;
10450 oputf("%d: Expected: [%s]\n", tno, zAns);
10451 oputf("%d: Got: [%s]\n", tno, str.z);
10454 else{
10455 eputf("Unknown operation \"%s\" on selftest line %d\n", zOp, tno);
10456 rc = 1;
10457 break;
10459 } /* End loop over rows of content from SELFTEST */
10460 sqlite3_finalize(pStmt);
10461 } /* End loop over k */
10462 freeText(&str);
10463 oputf("%d errors out of %d tests\n", nErr, nTest);
10464 }else
10466 if( c=='s' && cli_strncmp(azArg[0], "separator", n)==0 ){
10467 if( nArg<2 || nArg>3 ){
10468 eputz("Usage: .separator COL ?ROW?\n");
10469 rc = 1;
10471 if( nArg>=2 ){
10472 sqlite3_snprintf(sizeof(p->colSeparator), p->colSeparator,
10473 "%.*s", (int)ArraySize(p->colSeparator)-1, azArg[1]);
10475 if( nArg>=3 ){
10476 sqlite3_snprintf(sizeof(p->rowSeparator), p->rowSeparator,
10477 "%.*s", (int)ArraySize(p->rowSeparator)-1, azArg[2]);
10479 }else
10481 if( c=='s' && n>=4 && cli_strncmp(azArg[0],"sha3sum",n)==0 ){
10482 const char *zLike = 0; /* Which table to checksum. 0 means everything */
10483 int i; /* Loop counter */
10484 int bSchema = 0; /* Also hash the schema */
10485 int bSeparate = 0; /* Hash each table separately */
10486 int iSize = 224; /* Hash algorithm to use */
10487 int bDebug = 0; /* Only show the query that would have run */
10488 sqlite3_stmt *pStmt; /* For querying tables names */
10489 char *zSql; /* SQL to be run */
10490 char *zSep; /* Separator */
10491 ShellText sSql; /* Complete SQL for the query to run the hash */
10492 ShellText sQuery; /* Set of queries used to read all content */
10493 open_db(p, 0);
10494 for(i=1; i<nArg; i++){
10495 const char *z = azArg[i];
10496 if( z[0]=='-' ){
10497 z++;
10498 if( z[0]=='-' ) z++;
10499 if( cli_strcmp(z,"schema")==0 ){
10500 bSchema = 1;
10501 }else
10502 if( cli_strcmp(z,"sha3-224")==0 || cli_strcmp(z,"sha3-256")==0
10503 || cli_strcmp(z,"sha3-384")==0 || cli_strcmp(z,"sha3-512")==0
10505 iSize = atoi(&z[5]);
10506 }else
10507 if( cli_strcmp(z,"debug")==0 ){
10508 bDebug = 1;
10509 }else
10511 eputf("Unknown option \"%s\" on \"%s\"\n", azArg[i], azArg[0]);
10512 showHelp(p->out, azArg[0]);
10513 rc = 1;
10514 goto meta_command_exit;
10516 }else if( zLike ){
10517 eputz("Usage: .sha3sum ?OPTIONS? ?LIKE-PATTERN?\n");
10518 rc = 1;
10519 goto meta_command_exit;
10520 }else{
10521 zLike = z;
10522 bSeparate = 1;
10523 if( sqlite3_strlike("sqlite\\_%", zLike, '\\')==0 ) bSchema = 1;
10526 if( bSchema ){
10527 zSql = "SELECT lower(name) as tname FROM sqlite_schema"
10528 " WHERE type='table' AND coalesce(rootpage,0)>1"
10529 " UNION ALL SELECT 'sqlite_schema'"
10530 " ORDER BY 1 collate nocase";
10531 }else{
10532 zSql = "SELECT lower(name) as tname FROM sqlite_schema"
10533 " WHERE type='table' AND coalesce(rootpage,0)>1"
10534 " AND name NOT LIKE 'sqlite_%'"
10535 " ORDER BY 1 collate nocase";
10537 sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
10538 initText(&sQuery);
10539 initText(&sSql);
10540 appendText(&sSql, "WITH [sha3sum$query](a,b) AS(",0);
10541 zSep = "VALUES(";
10542 while( SQLITE_ROW==sqlite3_step(pStmt) ){
10543 const char *zTab = (const char*)sqlite3_column_text(pStmt,0);
10544 if( zTab==0 ) continue;
10545 if( zLike && sqlite3_strlike(zLike, zTab, 0)!=0 ) continue;
10546 if( cli_strncmp(zTab, "sqlite_",7)!=0 ){
10547 appendText(&sQuery,"SELECT * FROM ", 0);
10548 appendText(&sQuery,zTab,'"');
10549 appendText(&sQuery," NOT INDEXED;", 0);
10550 }else if( cli_strcmp(zTab, "sqlite_schema")==0 ){
10551 appendText(&sQuery,"SELECT type,name,tbl_name,sql FROM sqlite_schema"
10552 " ORDER BY name;", 0);
10553 }else if( cli_strcmp(zTab, "sqlite_sequence")==0 ){
10554 appendText(&sQuery,"SELECT name,seq FROM sqlite_sequence"
10555 " ORDER BY name;", 0);
10556 }else if( cli_strcmp(zTab, "sqlite_stat1")==0 ){
10557 appendText(&sQuery,"SELECT tbl,idx,stat FROM sqlite_stat1"
10558 " ORDER BY tbl,idx;", 0);
10559 }else if( cli_strcmp(zTab, "sqlite_stat4")==0 ){
10560 appendText(&sQuery, "SELECT * FROM ", 0);
10561 appendText(&sQuery, zTab, 0);
10562 appendText(&sQuery, " ORDER BY tbl, idx, rowid;\n", 0);
10564 appendText(&sSql, zSep, 0);
10565 appendText(&sSql, sQuery.z, '\'');
10566 sQuery.n = 0;
10567 appendText(&sSql, ",", 0);
10568 appendText(&sSql, zTab, '\'');
10569 zSep = "),(";
10571 sqlite3_finalize(pStmt);
10572 if( bSeparate ){
10573 zSql = sqlite3_mprintf(
10574 "%s))"
10575 " SELECT lower(hex(sha3_query(a,%d))) AS hash, b AS label"
10576 " FROM [sha3sum$query]",
10577 sSql.z, iSize);
10578 }else{
10579 zSql = sqlite3_mprintf(
10580 "%s))"
10581 " SELECT lower(hex(sha3_query(group_concat(a,''),%d))) AS hash"
10582 " FROM [sha3sum$query]",
10583 sSql.z, iSize);
10585 shell_check_oom(zSql);
10586 freeText(&sQuery);
10587 freeText(&sSql);
10588 if( bDebug ){
10589 oputf("%s\n", zSql);
10590 }else{
10591 shell_exec(p, zSql, 0);
10593 #if !defined(SQLITE_OMIT_SCHEMA_PRAGMAS) && !defined(SQLITE_OMIT_VIRTUALTABLE)
10595 int lrc;
10596 char *zRevText = /* Query for reversible to-blob-to-text check */
10597 "SELECT lower(name) as tname FROM sqlite_schema\n"
10598 "WHERE type='table' AND coalesce(rootpage,0)>1\n"
10599 "AND name NOT LIKE 'sqlite_%%'%s\n"
10600 "ORDER BY 1 collate nocase";
10601 zRevText = sqlite3_mprintf(zRevText, zLike? " AND name LIKE $tspec" : "");
10602 zRevText = sqlite3_mprintf(
10603 /* lower-case query is first run, producing upper-case query. */
10604 "with tabcols as materialized(\n"
10605 "select tname, cname\n"
10606 "from ("
10607 " select printf('\"%%w\"',ss.tname) as tname,"
10608 " printf('\"%%w\"',ti.name) as cname\n"
10609 " from (%z) ss\n inner join pragma_table_info(tname) ti))\n"
10610 "select 'SELECT total(bad_text_count) AS bad_text_count\n"
10611 "FROM ('||group_concat(query, ' UNION ALL ')||')' as btc_query\n"
10612 " from (select 'SELECT COUNT(*) AS bad_text_count\n"
10613 "FROM '||tname||' WHERE '\n"
10614 "||group_concat('CAST(CAST('||cname||' AS BLOB) AS TEXT)<>'||cname\n"
10615 "|| ' AND typeof('||cname||')=''text'' ',\n"
10616 "' OR ') as query, tname from tabcols group by tname)"
10617 , zRevText);
10618 shell_check_oom(zRevText);
10619 if( bDebug ) oputf("%s\n", zRevText);
10620 lrc = sqlite3_prepare_v2(p->db, zRevText, -1, &pStmt, 0);
10621 if( lrc!=SQLITE_OK ){
10622 /* assert(lrc==SQLITE_NOMEM); // might also be SQLITE_ERROR if the
10623 ** user does cruel and unnatural things like ".limit expr_depth 0". */
10624 rc = 1;
10625 }else{
10626 if( zLike ) sqlite3_bind_text(pStmt,1,zLike,-1,SQLITE_STATIC);
10627 lrc = SQLITE_ROW==sqlite3_step(pStmt);
10628 if( lrc ){
10629 const char *zGenQuery = (char*)sqlite3_column_text(pStmt,0);
10630 sqlite3_stmt *pCheckStmt;
10631 lrc = sqlite3_prepare_v2(p->db, zGenQuery, -1, &pCheckStmt, 0);
10632 if( bDebug ) oputf("%s\n", zGenQuery);
10633 if( lrc!=SQLITE_OK ){
10634 rc = 1;
10635 }else{
10636 if( SQLITE_ROW==sqlite3_step(pCheckStmt) ){
10637 double countIrreversible = sqlite3_column_double(pCheckStmt, 0);
10638 if( countIrreversible>0 ){
10639 int sz = (int)(countIrreversible + 0.5);
10640 eputf("Digest includes %d invalidly encoded text field%s.\n",
10641 sz, (sz>1)? "s": "");
10644 sqlite3_finalize(pCheckStmt);
10646 sqlite3_finalize(pStmt);
10649 if( rc ) eputz(".sha3sum failed.\n");
10650 sqlite3_free(zRevText);
10652 #endif /* !defined(*_OMIT_SCHEMA_PRAGMAS) && !defined(*_OMIT_VIRTUALTABLE) */
10653 sqlite3_free(zSql);
10654 }else
10656 #if !defined(SQLITE_NOHAVE_SYSTEM) && !defined(SQLITE_SHELL_FIDDLE)
10657 if( c=='s'
10658 && (cli_strncmp(azArg[0], "shell", n)==0
10659 || cli_strncmp(azArg[0],"system",n)==0)
10661 char *zCmd;
10662 int i, x;
10663 failIfSafeMode(p, "cannot run .%s in safe mode", azArg[0]);
10664 if( nArg<2 ){
10665 eputz("Usage: .system COMMAND\n");
10666 rc = 1;
10667 goto meta_command_exit;
10669 zCmd = sqlite3_mprintf(strchr(azArg[1],' ')==0?"%s":"\"%s\"", azArg[1]);
10670 for(i=2; i<nArg && zCmd!=0; i++){
10671 zCmd = sqlite3_mprintf(strchr(azArg[i],' ')==0?"%z %s":"%z \"%s\"",
10672 zCmd, azArg[i]);
10674 consoleRestore();
10675 x = zCmd!=0 ? system(zCmd) : 1;
10676 consoleRenewSetup();
10677 sqlite3_free(zCmd);
10678 if( x ) eputf("System command returns %d\n", x);
10679 }else
10680 #endif /* !defined(SQLITE_NOHAVE_SYSTEM) && !defined(SQLITE_SHELL_FIDDLE) */
10682 if( c=='s' && cli_strncmp(azArg[0], "show", n)==0 ){
10683 static const char *azBool[] = { "off", "on", "trigger", "full"};
10684 const char *zOut;
10685 int i;
10686 if( nArg!=1 ){
10687 eputz("Usage: .show\n");
10688 rc = 1;
10689 goto meta_command_exit;
10691 oputf("%12.12s: %s\n","echo",
10692 azBool[ShellHasFlag(p, SHFLG_Echo)]);
10693 oputf("%12.12s: %s\n","eqp", azBool[p->autoEQP&3]);
10694 oputf("%12.12s: %s\n","explain",
10695 p->mode==MODE_Explain ? "on" : p->autoExplain ? "auto" : "off");
10696 oputf("%12.12s: %s\n","headers", azBool[p->showHeader!=0]);
10697 if( p->mode==MODE_Column
10698 || (p->mode>=MODE_Markdown && p->mode<=MODE_Box)
10700 oputf("%12.12s: %s --wrap %d --wordwrap %s --%squote\n", "mode",
10701 modeDescr[p->mode], p->cmOpts.iWrap,
10702 p->cmOpts.bWordWrap ? "on" : "off",
10703 p->cmOpts.bQuote ? "" : "no");
10704 }else{
10705 oputf("%12.12s: %s\n","mode", modeDescr[p->mode]);
10707 oputf("%12.12s: ", "nullvalue");
10708 output_c_string(p->nullValue);
10709 oputz("\n");
10710 oputf("%12.12s: %s\n","output",
10711 strlen30(p->outfile) ? p->outfile : "stdout");
10712 oputf("%12.12s: ", "colseparator");
10713 output_c_string(p->colSeparator);
10714 oputz("\n");
10715 oputf("%12.12s: ", "rowseparator");
10716 output_c_string(p->rowSeparator);
10717 oputz("\n");
10718 switch( p->statsOn ){
10719 case 0: zOut = "off"; break;
10720 default: zOut = "on"; break;
10721 case 2: zOut = "stmt"; break;
10722 case 3: zOut = "vmstep"; break;
10724 oputf("%12.12s: %s\n","stats", zOut);
10725 oputf("%12.12s: ", "width");
10726 for (i=0;i<p->nWidth;i++) {
10727 oputf("%d ", p->colWidth[i]);
10729 oputz("\n");
10730 oputf("%12.12s: %s\n", "filename",
10731 p->pAuxDb->zDbFilename ? p->pAuxDb->zDbFilename : "");
10732 }else
10734 if( c=='s' && cli_strncmp(azArg[0], "stats", n)==0 ){
10735 if( nArg==2 ){
10736 if( cli_strcmp(azArg[1],"stmt")==0 ){
10737 p->statsOn = 2;
10738 }else if( cli_strcmp(azArg[1],"vmstep")==0 ){
10739 p->statsOn = 3;
10740 }else{
10741 p->statsOn = (u8)booleanValue(azArg[1]);
10743 }else if( nArg==1 ){
10744 display_stats(p->db, p, 0);
10745 }else{
10746 eputz("Usage: .stats ?on|off|stmt|vmstep?\n");
10747 rc = 1;
10749 }else
10751 if( (c=='t' && n>1 && cli_strncmp(azArg[0], "tables", n)==0)
10752 || (c=='i' && (cli_strncmp(azArg[0], "indices", n)==0
10753 || cli_strncmp(azArg[0], "indexes", n)==0) )
10755 sqlite3_stmt *pStmt;
10756 char **azResult;
10757 int nRow, nAlloc;
10758 int ii;
10759 ShellText s;
10760 initText(&s);
10761 open_db(p, 0);
10762 rc = sqlite3_prepare_v2(p->db, "PRAGMA database_list", -1, &pStmt, 0);
10763 if( rc ){
10764 sqlite3_finalize(pStmt);
10765 return shellDatabaseError(p->db);
10768 if( nArg>2 && c=='i' ){
10769 /* It is an historical accident that the .indexes command shows an error
10770 ** when called with the wrong number of arguments whereas the .tables
10771 ** command does not. */
10772 eputz("Usage: .indexes ?LIKE-PATTERN?\n");
10773 rc = 1;
10774 sqlite3_finalize(pStmt);
10775 goto meta_command_exit;
10777 for(ii=0; sqlite3_step(pStmt)==SQLITE_ROW; ii++){
10778 const char *zDbName = (const char*)sqlite3_column_text(pStmt, 1);
10779 if( zDbName==0 ) continue;
10780 if( s.z && s.z[0] ) appendText(&s, " UNION ALL ", 0);
10781 if( sqlite3_stricmp(zDbName, "main")==0 ){
10782 appendText(&s, "SELECT name FROM ", 0);
10783 }else{
10784 appendText(&s, "SELECT ", 0);
10785 appendText(&s, zDbName, '\'');
10786 appendText(&s, "||'.'||name FROM ", 0);
10788 appendText(&s, zDbName, '"');
10789 appendText(&s, ".sqlite_schema ", 0);
10790 if( c=='t' ){
10791 appendText(&s," WHERE type IN ('table','view')"
10792 " AND name NOT LIKE 'sqlite_%'"
10793 " AND name LIKE ?1", 0);
10794 }else{
10795 appendText(&s," WHERE type='index'"
10796 " AND tbl_name LIKE ?1", 0);
10799 rc = sqlite3_finalize(pStmt);
10800 if( rc==SQLITE_OK ){
10801 appendText(&s, " ORDER BY 1", 0);
10802 rc = sqlite3_prepare_v2(p->db, s.z, -1, &pStmt, 0);
10804 freeText(&s);
10805 if( rc ) return shellDatabaseError(p->db);
10807 /* Run the SQL statement prepared by the above block. Store the results
10808 ** as an array of nul-terminated strings in azResult[]. */
10809 nRow = nAlloc = 0;
10810 azResult = 0;
10811 if( nArg>1 ){
10812 sqlite3_bind_text(pStmt, 1, azArg[1], -1, SQLITE_TRANSIENT);
10813 }else{
10814 sqlite3_bind_text(pStmt, 1, "%", -1, SQLITE_STATIC);
10816 while( sqlite3_step(pStmt)==SQLITE_ROW ){
10817 if( nRow>=nAlloc ){
10818 char **azNew;
10819 int n2 = nAlloc*2 + 10;
10820 azNew = sqlite3_realloc64(azResult, sizeof(azResult[0])*n2);
10821 shell_check_oom(azNew);
10822 nAlloc = n2;
10823 azResult = azNew;
10825 azResult[nRow] = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
10826 shell_check_oom(azResult[nRow]);
10827 nRow++;
10829 if( sqlite3_finalize(pStmt)!=SQLITE_OK ){
10830 rc = shellDatabaseError(p->db);
10833 /* Pretty-print the contents of array azResult[] to the output */
10834 if( rc==0 && nRow>0 ){
10835 int len, maxlen = 0;
10836 int i, j;
10837 int nPrintCol, nPrintRow;
10838 for(i=0; i<nRow; i++){
10839 len = strlen30(azResult[i]);
10840 if( len>maxlen ) maxlen = len;
10842 nPrintCol = 80/(maxlen+2);
10843 if( nPrintCol<1 ) nPrintCol = 1;
10844 nPrintRow = (nRow + nPrintCol - 1)/nPrintCol;
10845 for(i=0; i<nPrintRow; i++){
10846 for(j=i; j<nRow; j+=nPrintRow){
10847 char *zSp = j<nPrintRow ? "" : " ";
10848 oputf("%s%-*s", zSp, maxlen, azResult[j] ? azResult[j]:"");
10850 oputz("\n");
10854 for(ii=0; ii<nRow; ii++) sqlite3_free(azResult[ii]);
10855 sqlite3_free(azResult);
10856 }else
10858 #ifndef SQLITE_SHELL_FIDDLE
10859 /* Begin redirecting output to the file "testcase-out.txt" */
10860 if( c=='t' && cli_strcmp(azArg[0],"testcase")==0 ){
10861 output_reset(p);
10862 p->out = output_file_open("testcase-out.txt", 0);
10863 if( p->out==0 ){
10864 eputz("Error: cannot open 'testcase-out.txt'\n");
10866 if( nArg>=2 ){
10867 sqlite3_snprintf(sizeof(p->zTestcase), p->zTestcase, "%s", azArg[1]);
10868 }else{
10869 sqlite3_snprintf(sizeof(p->zTestcase), p->zTestcase, "?");
10871 }else
10872 #endif /* !defined(SQLITE_SHELL_FIDDLE) */
10874 #ifndef SQLITE_UNTESTABLE
10875 if( c=='t' && n>=8 && cli_strncmp(azArg[0], "testctrl", n)==0 ){
10876 static const struct {
10877 const char *zCtrlName; /* Name of a test-control option */
10878 int ctrlCode; /* Integer code for that option */
10879 int unSafe; /* Not valid unless --unsafe-testing */
10880 const char *zUsage; /* Usage notes */
10881 } aCtrl[] = {
10882 {"always", SQLITE_TESTCTRL_ALWAYS, 1, "BOOLEAN" },
10883 {"assert", SQLITE_TESTCTRL_ASSERT, 1, "BOOLEAN" },
10884 /*{"benign_malloc_hooks",SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS,1, "" },*/
10885 /*{"bitvec_test", SQLITE_TESTCTRL_BITVEC_TEST, 1, "" },*/
10886 {"byteorder", SQLITE_TESTCTRL_BYTEORDER, 0, "" },
10887 {"extra_schema_checks",SQLITE_TESTCTRL_EXTRA_SCHEMA_CHECKS,0,"BOOLEAN" },
10888 {"fault_install", SQLITE_TESTCTRL_FAULT_INSTALL, 1,"args..." },
10889 {"fk_no_action", SQLITE_TESTCTRL_FK_NO_ACTION, 0, "BOOLEAN" },
10890 {"imposter", SQLITE_TESTCTRL_IMPOSTER,1,"SCHEMA ON/OFF ROOTPAGE"},
10891 {"internal_functions", SQLITE_TESTCTRL_INTERNAL_FUNCTIONS,0,"" },
10892 {"json_selfcheck", SQLITE_TESTCTRL_JSON_SELFCHECK ,0,"BOOLEAN" },
10893 {"localtime_fault", SQLITE_TESTCTRL_LOCALTIME_FAULT,0,"BOOLEAN" },
10894 {"never_corrupt", SQLITE_TESTCTRL_NEVER_CORRUPT,1, "BOOLEAN" },
10895 {"optimizations", SQLITE_TESTCTRL_OPTIMIZATIONS,0,"DISABLE-MASK" },
10896 #ifdef YYCOVERAGE
10897 {"parser_coverage", SQLITE_TESTCTRL_PARSER_COVERAGE,0,"" },
10898 #endif
10899 {"pending_byte", SQLITE_TESTCTRL_PENDING_BYTE,0, "OFFSET " },
10900 {"prng_restore", SQLITE_TESTCTRL_PRNG_RESTORE,0, "" },
10901 {"prng_save", SQLITE_TESTCTRL_PRNG_SAVE, 0, "" },
10902 {"prng_seed", SQLITE_TESTCTRL_PRNG_SEED, 0, "SEED ?db?" },
10903 {"seek_count", SQLITE_TESTCTRL_SEEK_COUNT, 0, "" },
10904 {"sorter_mmap", SQLITE_TESTCTRL_SORTER_MMAP, 0, "NMAX" },
10905 {"tune", SQLITE_TESTCTRL_TUNE, 1, "ID VALUE" },
10906 {"uselongdouble", SQLITE_TESTCTRL_USELONGDOUBLE,0,"?BOOLEAN|\"default\"?"},
10908 int testctrl = -1;
10909 int iCtrl = -1;
10910 int rc2 = 0; /* 0: usage. 1: %d 2: %x 3: no-output */
10911 int isOk = 0;
10912 int i, n2;
10913 const char *zCmd = 0;
10915 open_db(p, 0);
10916 zCmd = nArg>=2 ? azArg[1] : "help";
10918 /* The argument can optionally begin with "-" or "--" */
10919 if( zCmd[0]=='-' && zCmd[1] ){
10920 zCmd++;
10921 if( zCmd[0]=='-' && zCmd[1] ) zCmd++;
10924 /* --help lists all test-controls */
10925 if( cli_strcmp(zCmd,"help")==0 ){
10926 oputz("Available test-controls:\n");
10927 for(i=0; i<ArraySize(aCtrl); i++){
10928 if( aCtrl[i].unSafe && !ShellHasFlag(p,SHFLG_TestingMode) ) continue;
10929 oputf(" .testctrl %s %s\n",
10930 aCtrl[i].zCtrlName, aCtrl[i].zUsage);
10932 rc = 1;
10933 goto meta_command_exit;
10936 /* convert testctrl text option to value. allow any unique prefix
10937 ** of the option name, or a numerical value. */
10938 n2 = strlen30(zCmd);
10939 for(i=0; i<ArraySize(aCtrl); i++){
10940 if( aCtrl[i].unSafe && !ShellHasFlag(p,SHFLG_TestingMode) ) continue;
10941 if( cli_strncmp(zCmd, aCtrl[i].zCtrlName, n2)==0 ){
10942 if( testctrl<0 ){
10943 testctrl = aCtrl[i].ctrlCode;
10944 iCtrl = i;
10945 }else{
10946 eputf("Error: ambiguous test-control: \"%s\"\n"
10947 "Use \".testctrl --help\" for help\n", zCmd);
10948 rc = 1;
10949 goto meta_command_exit;
10953 if( testctrl<0 ){
10954 eputf("Error: unknown test-control: %s\n"
10955 "Use \".testctrl --help\" for help\n", zCmd);
10956 }else{
10957 switch(testctrl){
10959 /* sqlite3_test_control(int, db, int) */
10960 case SQLITE_TESTCTRL_OPTIMIZATIONS:
10961 case SQLITE_TESTCTRL_FK_NO_ACTION:
10962 if( nArg==3 ){
10963 unsigned int opt = (unsigned int)strtol(azArg[2], 0, 0);
10964 rc2 = sqlite3_test_control(testctrl, p->db, opt);
10965 isOk = 3;
10967 break;
10969 /* sqlite3_test_control(int) */
10970 case SQLITE_TESTCTRL_PRNG_SAVE:
10971 case SQLITE_TESTCTRL_PRNG_RESTORE:
10972 case SQLITE_TESTCTRL_BYTEORDER:
10973 if( nArg==2 ){
10974 rc2 = sqlite3_test_control(testctrl);
10975 isOk = testctrl==SQLITE_TESTCTRL_BYTEORDER ? 1 : 3;
10977 break;
10979 /* sqlite3_test_control(int, uint) */
10980 case SQLITE_TESTCTRL_PENDING_BYTE:
10981 if( nArg==3 ){
10982 unsigned int opt = (unsigned int)integerValue(azArg[2]);
10983 rc2 = sqlite3_test_control(testctrl, opt);
10984 isOk = 3;
10986 break;
10988 /* sqlite3_test_control(int, int, sqlite3*) */
10989 case SQLITE_TESTCTRL_PRNG_SEED:
10990 if( nArg==3 || nArg==4 ){
10991 int ii = (int)integerValue(azArg[2]);
10992 sqlite3 *db;
10993 if( ii==0 && cli_strcmp(azArg[2],"random")==0 ){
10994 sqlite3_randomness(sizeof(ii),&ii);
10995 sputf(stdout, "-- random seed: %d\n", ii);
10997 if( nArg==3 ){
10998 db = 0;
10999 }else{
11000 db = p->db;
11001 /* Make sure the schema has been loaded */
11002 sqlite3_table_column_metadata(db, 0, "x", 0, 0, 0, 0, 0, 0);
11004 rc2 = sqlite3_test_control(testctrl, ii, db);
11005 isOk = 3;
11007 break;
11009 /* sqlite3_test_control(int, int) */
11010 case SQLITE_TESTCTRL_ASSERT:
11011 case SQLITE_TESTCTRL_ALWAYS:
11012 if( nArg==3 ){
11013 int opt = booleanValue(azArg[2]);
11014 rc2 = sqlite3_test_control(testctrl, opt);
11015 isOk = 1;
11017 break;
11019 /* sqlite3_test_control(int, int) */
11020 case SQLITE_TESTCTRL_LOCALTIME_FAULT:
11021 case SQLITE_TESTCTRL_NEVER_CORRUPT:
11022 if( nArg==3 ){
11023 int opt = booleanValue(azArg[2]);
11024 rc2 = sqlite3_test_control(testctrl, opt);
11025 isOk = 3;
11027 break;
11029 /* sqlite3_test_control(int, int) */
11030 case SQLITE_TESTCTRL_USELONGDOUBLE: {
11031 int opt = -1;
11032 if( nArg==3 ){
11033 if( cli_strcmp(azArg[2],"default")==0 ){
11034 opt = 2;
11035 }else{
11036 opt = booleanValue(azArg[2]);
11039 rc2 = sqlite3_test_control(testctrl, opt);
11040 isOk = 1;
11041 break;
11044 /* sqlite3_test_control(sqlite3*) */
11045 case SQLITE_TESTCTRL_INTERNAL_FUNCTIONS:
11046 rc2 = sqlite3_test_control(testctrl, p->db);
11047 isOk = 3;
11048 break;
11050 case SQLITE_TESTCTRL_IMPOSTER:
11051 if( nArg==5 ){
11052 rc2 = sqlite3_test_control(testctrl, p->db,
11053 azArg[2],
11054 integerValue(azArg[3]),
11055 integerValue(azArg[4]));
11056 isOk = 3;
11058 break;
11060 case SQLITE_TESTCTRL_SEEK_COUNT: {
11061 u64 x = 0;
11062 rc2 = sqlite3_test_control(testctrl, p->db, &x);
11063 oputf("%llu\n", x);
11064 isOk = 3;
11065 break;
11068 #ifdef YYCOVERAGE
11069 case SQLITE_TESTCTRL_PARSER_COVERAGE: {
11070 if( nArg==2 ){
11071 sqlite3_test_control(testctrl, p->out);
11072 isOk = 3;
11074 break;
11076 #endif
11077 #ifdef SQLITE_DEBUG
11078 case SQLITE_TESTCTRL_TUNE: {
11079 if( nArg==4 ){
11080 int id = (int)integerValue(azArg[2]);
11081 int val = (int)integerValue(azArg[3]);
11082 sqlite3_test_control(testctrl, id, &val);
11083 isOk = 3;
11084 }else if( nArg==3 ){
11085 int id = (int)integerValue(azArg[2]);
11086 sqlite3_test_control(testctrl, -id, &rc2);
11087 isOk = 1;
11088 }else if( nArg==2 ){
11089 int id = 1;
11090 while(1){
11091 int val = 0;
11092 rc2 = sqlite3_test_control(testctrl, -id, &val);
11093 if( rc2!=SQLITE_OK ) break;
11094 if( id>1 ) oputz(" ");
11095 oputf("%d: %d", id, val);
11096 id++;
11098 if( id>1 ) oputz("\n");
11099 isOk = 3;
11101 break;
11103 #endif
11104 case SQLITE_TESTCTRL_SORTER_MMAP:
11105 if( nArg==3 ){
11106 int opt = (unsigned int)integerValue(azArg[2]);
11107 rc2 = sqlite3_test_control(testctrl, p->db, opt);
11108 isOk = 3;
11110 break;
11111 case SQLITE_TESTCTRL_JSON_SELFCHECK:
11112 if( nArg==2 ){
11113 rc2 = -1;
11114 isOk = 1;
11115 }else{
11116 rc2 = booleanValue(azArg[2]);
11117 isOk = 3;
11119 sqlite3_test_control(testctrl, &rc2);
11120 break;
11121 case SQLITE_TESTCTRL_FAULT_INSTALL: {
11122 int kk;
11123 int bShowHelp = nArg<=2;
11124 isOk = 3;
11125 for(kk=2; kk<nArg; kk++){
11126 const char *z = azArg[kk];
11127 if( z[0]=='-' && z[1]=='-' ) z++;
11128 if( cli_strcmp(z,"off")==0 ){
11129 sqlite3_test_control(testctrl, 0);
11130 }else if( cli_strcmp(z,"on")==0 ){
11131 faultsim_state.iCnt = faultsim_state.nSkip;
11132 if( faultsim_state.iErr==0 ) faultsim_state.iErr = 1;
11133 faultsim_state.nHit = 0;
11134 sqlite3_test_control(testctrl, faultsim_callback);
11135 }else if( cli_strcmp(z,"reset")==0 ){
11136 faultsim_state.iCnt = faultsim_state.nSkip;
11137 faultsim_state.nHit = 0;
11138 sqlite3_test_control(testctrl, faultsim_callback);
11139 }else if( cli_strcmp(z,"status")==0 ){
11140 oputf("faultsim.iId: %d\n", faultsim_state.iId);
11141 oputf("faultsim.iErr: %d\n", faultsim_state.iErr);
11142 oputf("faultsim.iCnt: %d\n", faultsim_state.iCnt);
11143 oputf("faultsim.nHit: %d\n", faultsim_state.nHit);
11144 oputf("faultsim.iInterval: %d\n", faultsim_state.iInterval);
11145 oputf("faultsim.eVerbose: %d\n", faultsim_state.eVerbose);
11146 oputf("faultsim.nRepeat: %d\n", faultsim_state.nRepeat);
11147 oputf("faultsim.nSkip: %d\n", faultsim_state.nSkip);
11148 }else if( cli_strcmp(z,"-v")==0 ){
11149 if( faultsim_state.eVerbose<2 ) faultsim_state.eVerbose++;
11150 }else if( cli_strcmp(z,"-q")==0 ){
11151 if( faultsim_state.eVerbose>0 ) faultsim_state.eVerbose--;
11152 }else if( cli_strcmp(z,"-id")==0 && kk+1<nArg ){
11153 faultsim_state.iId = atoi(azArg[++kk]);
11154 }else if( cli_strcmp(z,"-errcode")==0 && kk+1<nArg ){
11155 faultsim_state.iErr = atoi(azArg[++kk]);
11156 }else if( cli_strcmp(z,"-interval")==0 && kk+1<nArg ){
11157 faultsim_state.iInterval = atoi(azArg[++kk]);
11158 }else if( cli_strcmp(z,"-repeat")==0 && kk+1<nArg ){
11159 faultsim_state.nRepeat = atoi(azArg[++kk]);
11160 }else if( cli_strcmp(z,"-skip")==0 && kk+1<nArg ){
11161 faultsim_state.nSkip = atoi(azArg[++kk]);
11162 }else if( cli_strcmp(z,"-?")==0 || sqlite3_strglob("*help*",z)==0){
11163 bShowHelp = 1;
11164 }else{
11165 eputf("Unrecognized fault_install argument: \"%s\"\n",
11166 azArg[kk]);
11167 rc = 1;
11168 bShowHelp = 1;
11169 break;
11172 if( bShowHelp ){
11173 oputz(
11174 "Usage: .testctrl fault_install ARGS\n"
11175 "Possible arguments:\n"
11176 " off Disable faultsim\n"
11177 " on Activate faultsim\n"
11178 " reset Reset the trigger counter\n"
11179 " status Show current status\n"
11180 " -v Increase verbosity\n"
11181 " -q Decrease verbosity\n"
11182 " --errcode N When triggered, return N as error code\n"
11183 " --id ID Trigger only for the ID specified\n"
11184 " --interval N Trigger only after every N-th call\n"
11185 " --repeat N Turn off after N hits. 0 means never\n"
11186 " --skip N Skip the first N encounters\n"
11189 break;
11193 if( isOk==0 && iCtrl>=0 ){
11194 oputf("Usage: .testctrl %s %s\n", zCmd,aCtrl[iCtrl].zUsage);
11195 rc = 1;
11196 }else if( isOk==1 ){
11197 oputf("%d\n", rc2);
11198 }else if( isOk==2 ){
11199 oputf("0x%08x\n", rc2);
11201 }else
11202 #endif /* !defined(SQLITE_UNTESTABLE) */
11204 if( c=='t' && n>4 && cli_strncmp(azArg[0], "timeout", n)==0 ){
11205 open_db(p, 0);
11206 sqlite3_busy_timeout(p->db, nArg>=2 ? (int)integerValue(azArg[1]) : 0);
11207 }else
11209 if( c=='t' && n>=5 && cli_strncmp(azArg[0], "timer", n)==0 ){
11210 if( nArg==2 ){
11211 enableTimer = booleanValue(azArg[1]);
11212 if( enableTimer && !HAS_TIMER ){
11213 eputz("Error: timer not available on this system.\n");
11214 enableTimer = 0;
11216 }else{
11217 eputz("Usage: .timer on|off\n");
11218 rc = 1;
11220 }else
11222 #ifndef SQLITE_OMIT_TRACE
11223 if( c=='t' && cli_strncmp(azArg[0], "trace", n)==0 ){
11224 int mType = 0;
11225 int jj;
11226 open_db(p, 0);
11227 for(jj=1; jj<nArg; jj++){
11228 const char *z = azArg[jj];
11229 if( z[0]=='-' ){
11230 if( optionMatch(z, "expanded") ){
11231 p->eTraceType = SHELL_TRACE_EXPANDED;
11233 #ifdef SQLITE_ENABLE_NORMALIZE
11234 else if( optionMatch(z, "normalized") ){
11235 p->eTraceType = SHELL_TRACE_NORMALIZED;
11237 #endif
11238 else if( optionMatch(z, "plain") ){
11239 p->eTraceType = SHELL_TRACE_PLAIN;
11241 else if( optionMatch(z, "profile") ){
11242 mType |= SQLITE_TRACE_PROFILE;
11244 else if( optionMatch(z, "row") ){
11245 mType |= SQLITE_TRACE_ROW;
11247 else if( optionMatch(z, "stmt") ){
11248 mType |= SQLITE_TRACE_STMT;
11250 else if( optionMatch(z, "close") ){
11251 mType |= SQLITE_TRACE_CLOSE;
11253 else {
11254 eputf("Unknown option \"%s\" on \".trace\"\n", z);
11255 rc = 1;
11256 goto meta_command_exit;
11258 }else{
11259 output_file_close(p->traceOut);
11260 p->traceOut = output_file_open(z, 0);
11263 if( p->traceOut==0 ){
11264 sqlite3_trace_v2(p->db, 0, 0, 0);
11265 }else{
11266 if( mType==0 ) mType = SQLITE_TRACE_STMT;
11267 sqlite3_trace_v2(p->db, mType, sql_trace_callback, p);
11269 }else
11270 #endif /* !defined(SQLITE_OMIT_TRACE) */
11272 #if defined(SQLITE_DEBUG) && !defined(SQLITE_OMIT_VIRTUALTABLE)
11273 if( c=='u' && cli_strncmp(azArg[0], "unmodule", n)==0 ){
11274 int ii;
11275 int lenOpt;
11276 char *zOpt;
11277 if( nArg<2 ){
11278 eputz("Usage: .unmodule [--allexcept] NAME ...\n");
11279 rc = 1;
11280 goto meta_command_exit;
11282 open_db(p, 0);
11283 zOpt = azArg[1];
11284 if( zOpt[0]=='-' && zOpt[1]=='-' && zOpt[2]!=0 ) zOpt++;
11285 lenOpt = (int)strlen(zOpt);
11286 if( lenOpt>=3 && cli_strncmp(zOpt, "-allexcept",lenOpt)==0 ){
11287 assert( azArg[nArg]==0 );
11288 sqlite3_drop_modules(p->db, nArg>2 ? (const char**)(azArg+2) : 0);
11289 }else{
11290 for(ii=1; ii<nArg; ii++){
11291 sqlite3_create_module(p->db, azArg[ii], 0, 0);
11294 }else
11295 #endif
11297 #if SQLITE_USER_AUTHENTICATION
11298 if( c=='u' && cli_strncmp(azArg[0], "user", n)==0 ){
11299 if( nArg<2 ){
11300 eputz("Usage: .user SUBCOMMAND ...\n");
11301 rc = 1;
11302 goto meta_command_exit;
11304 open_db(p, 0);
11305 if( cli_strcmp(azArg[1],"login")==0 ){
11306 if( nArg!=4 ){
11307 eputz("Usage: .user login USER PASSWORD\n");
11308 rc = 1;
11309 goto meta_command_exit;
11311 rc = sqlite3_user_authenticate(p->db, azArg[2], azArg[3],
11312 strlen30(azArg[3]));
11313 if( rc ){
11314 eputf("Authentication failed for user %s\n", azArg[2]);
11315 rc = 1;
11317 }else if( cli_strcmp(azArg[1],"add")==0 ){
11318 if( nArg!=5 ){
11319 eputz("Usage: .user add USER PASSWORD ISADMIN\n");
11320 rc = 1;
11321 goto meta_command_exit;
11323 rc = sqlite3_user_add(p->db, azArg[2], azArg[3], strlen30(azArg[3]),
11324 booleanValue(azArg[4]));
11325 if( rc ){
11326 eputf("User-Add failed: %d\n", rc);
11327 rc = 1;
11329 }else if( cli_strcmp(azArg[1],"edit")==0 ){
11330 if( nArg!=5 ){
11331 eputz("Usage: .user edit USER PASSWORD ISADMIN\n");
11332 rc = 1;
11333 goto meta_command_exit;
11335 rc = sqlite3_user_change(p->db, azArg[2], azArg[3], strlen30(azArg[3]),
11336 booleanValue(azArg[4]));
11337 if( rc ){
11338 eputf("User-Edit failed: %d\n", rc);
11339 rc = 1;
11341 }else if( cli_strcmp(azArg[1],"delete")==0 ){
11342 if( nArg!=3 ){
11343 eputz("Usage: .user delete USER\n");
11344 rc = 1;
11345 goto meta_command_exit;
11347 rc = sqlite3_user_delete(p->db, azArg[2]);
11348 if( rc ){
11349 eputf("User-Delete failed: %d\n", rc);
11350 rc = 1;
11352 }else{
11353 eputz("Usage: .user login|add|edit|delete ...\n");
11354 rc = 1;
11355 goto meta_command_exit;
11357 }else
11358 #endif /* SQLITE_USER_AUTHENTICATION */
11360 if( c=='v' && cli_strncmp(azArg[0], "version", n)==0 ){
11361 char *zPtrSz = sizeof(void*)==8 ? "64-bit" : "32-bit";
11362 oputf("SQLite %s %s\n" /*extra-version-info*/,
11363 sqlite3_libversion(), sqlite3_sourceid());
11364 #if SQLITE_HAVE_ZLIB
11365 oputf("zlib version %s\n", zlibVersion());
11366 #endif
11367 #define CTIMEOPT_VAL_(opt) #opt
11368 #define CTIMEOPT_VAL(opt) CTIMEOPT_VAL_(opt)
11369 #if defined(__clang__) && defined(__clang_major__)
11370 oputf("clang-" CTIMEOPT_VAL(__clang_major__) "."
11371 CTIMEOPT_VAL(__clang_minor__) "."
11372 CTIMEOPT_VAL(__clang_patchlevel__) " (%s)\n", zPtrSz);
11373 #elif defined(_MSC_VER)
11374 oputf("msvc-" CTIMEOPT_VAL(_MSC_VER) " (%s)\n", zPtrSz);
11375 #elif defined(__GNUC__) && defined(__VERSION__)
11376 oputf("gcc-" __VERSION__ " (%s)\n", zPtrSz);
11377 #endif
11378 }else
11380 if( c=='v' && cli_strncmp(azArg[0], "vfsinfo", n)==0 ){
11381 const char *zDbName = nArg==2 ? azArg[1] : "main";
11382 sqlite3_vfs *pVfs = 0;
11383 if( p->db ){
11384 sqlite3_file_control(p->db, zDbName, SQLITE_FCNTL_VFS_POINTER, &pVfs);
11385 if( pVfs ){
11386 oputf("vfs.zName = \"%s\"\n", pVfs->zName);
11387 oputf("vfs.iVersion = %d\n", pVfs->iVersion);
11388 oputf("vfs.szOsFile = %d\n", pVfs->szOsFile);
11389 oputf("vfs.mxPathname = %d\n", pVfs->mxPathname);
11392 }else
11394 if( c=='v' && cli_strncmp(azArg[0], "vfslist", n)==0 ){
11395 sqlite3_vfs *pVfs;
11396 sqlite3_vfs *pCurrent = 0;
11397 if( p->db ){
11398 sqlite3_file_control(p->db, "main", SQLITE_FCNTL_VFS_POINTER, &pCurrent);
11400 for(pVfs=sqlite3_vfs_find(0); pVfs; pVfs=pVfs->pNext){
11401 oputf("vfs.zName = \"%s\"%s\n", pVfs->zName,
11402 pVfs==pCurrent ? " <--- CURRENT" : "");
11403 oputf("vfs.iVersion = %d\n", pVfs->iVersion);
11404 oputf("vfs.szOsFile = %d\n", pVfs->szOsFile);
11405 oputf("vfs.mxPathname = %d\n", pVfs->mxPathname);
11406 if( pVfs->pNext ){
11407 oputz("-----------------------------------\n");
11410 }else
11412 if( c=='v' && cli_strncmp(azArg[0], "vfsname", n)==0 ){
11413 const char *zDbName = nArg==2 ? azArg[1] : "main";
11414 char *zVfsName = 0;
11415 if( p->db ){
11416 sqlite3_file_control(p->db, zDbName, SQLITE_FCNTL_VFSNAME, &zVfsName);
11417 if( zVfsName ){
11418 oputf("%s\n", zVfsName);
11419 sqlite3_free(zVfsName);
11422 }else
11424 if( c=='w' && cli_strncmp(azArg[0], "wheretrace", n)==0 ){
11425 unsigned int x = nArg>=2? (unsigned int)integerValue(azArg[1]) : 0xffffffff;
11426 sqlite3_test_control(SQLITE_TESTCTRL_TRACEFLAGS, 3, &x);
11427 }else
11429 if( c=='w' && cli_strncmp(azArg[0], "width", n)==0 ){
11430 int j;
11431 assert( nArg<=ArraySize(azArg) );
11432 p->nWidth = nArg-1;
11433 p->colWidth = realloc(p->colWidth, (p->nWidth+1)*sizeof(int)*2);
11434 if( p->colWidth==0 && p->nWidth>0 ) shell_out_of_memory();
11435 if( p->nWidth ) p->actualWidth = &p->colWidth[p->nWidth];
11436 for(j=1; j<nArg; j++){
11437 p->colWidth[j-1] = (int)integerValue(azArg[j]);
11439 }else
11442 eputf("Error: unknown command or invalid arguments: "
11443 " \"%s\". Enter \".help\" for help\n", azArg[0]);
11444 rc = 1;
11447 meta_command_exit:
11448 if( p->outCount ){
11449 p->outCount--;
11450 if( p->outCount==0 ) output_reset(p);
11452 p->bSafeMode = p->bSafeModePersist;
11453 return rc;
11456 /* Line scan result and intermediate states (supporting scan resumption)
11458 #ifndef CHAR_BIT
11459 # define CHAR_BIT 8
11460 #endif
11461 typedef enum {
11462 QSS_HasDark = 1<<CHAR_BIT, QSS_EndingSemi = 2<<CHAR_BIT,
11463 QSS_CharMask = (1<<CHAR_BIT)-1, QSS_ScanMask = 3<<CHAR_BIT,
11464 QSS_Start = 0
11465 } QuickScanState;
11466 #define QSS_SETV(qss, newst) ((newst) | ((qss) & QSS_ScanMask))
11467 #define QSS_INPLAIN(qss) (((qss)&QSS_CharMask)==QSS_Start)
11468 #define QSS_PLAINWHITE(qss) (((qss)&~QSS_EndingSemi)==QSS_Start)
11469 #define QSS_PLAINDARK(qss) (((qss)&~QSS_EndingSemi)==QSS_HasDark)
11470 #define QSS_SEMITERM(qss) (((qss)&~QSS_HasDark)==QSS_EndingSemi)
11473 ** Scan line for classification to guide shell's handling.
11474 ** The scan is resumable for subsequent lines when prior
11475 ** return values are passed as the 2nd argument.
11477 static QuickScanState quickscan(char *zLine, QuickScanState qss,
11478 SCAN_TRACKER_REFTYPE pst){
11479 char cin;
11480 char cWait = (char)qss; /* intentional narrowing loss */
11481 if( cWait==0 ){
11482 PlainScan:
11483 assert( cWait==0 );
11484 while( (cin = *zLine++)!=0 ){
11485 if( IsSpace(cin) )
11486 continue;
11487 switch (cin){
11488 case '-':
11489 if( *zLine!='-' )
11490 break;
11491 while((cin = *++zLine)!=0 )
11492 if( cin=='\n')
11493 goto PlainScan;
11494 return qss;
11495 case ';':
11496 qss |= QSS_EndingSemi;
11497 continue;
11498 case '/':
11499 if( *zLine=='*' ){
11500 ++zLine;
11501 cWait = '*';
11502 CONTINUE_PROMPT_AWAITS(pst, "/*");
11503 qss = QSS_SETV(qss, cWait);
11504 goto TermScan;
11506 break;
11507 case '[':
11508 cin = ']';
11509 deliberate_fall_through;
11510 case '`': case '\'': case '"':
11511 cWait = cin;
11512 qss = QSS_HasDark | cWait;
11513 CONTINUE_PROMPT_AWAITC(pst, cin);
11514 goto TermScan;
11515 case '(':
11516 CONTINUE_PAREN_INCR(pst, 1);
11517 break;
11518 case ')':
11519 CONTINUE_PAREN_INCR(pst, -1);
11520 break;
11521 default:
11522 break;
11524 qss = (qss & ~QSS_EndingSemi) | QSS_HasDark;
11526 }else{
11527 TermScan:
11528 while( (cin = *zLine++)!=0 ){
11529 if( cin==cWait ){
11530 switch( cWait ){
11531 case '*':
11532 if( *zLine != '/' )
11533 continue;
11534 ++zLine;
11535 cWait = 0;
11536 CONTINUE_PROMPT_AWAITC(pst, 0);
11537 qss = QSS_SETV(qss, 0);
11538 goto PlainScan;
11539 case '`': case '\'': case '"':
11540 if(*zLine==cWait){
11541 /* Swallow doubled end-delimiter.*/
11542 ++zLine;
11543 continue;
11545 deliberate_fall_through;
11546 case ']':
11547 cWait = 0;
11548 CONTINUE_PROMPT_AWAITC(pst, 0);
11549 qss = QSS_SETV(qss, 0);
11550 goto PlainScan;
11551 default: assert(0);
11556 return qss;
11560 ** Return TRUE if the line typed in is an SQL command terminator other
11561 ** than a semi-colon. The SQL Server style "go" command is understood
11562 ** as is the Oracle "/".
11564 static int line_is_command_terminator(char *zLine){
11565 while( IsSpace(zLine[0]) ){ zLine++; };
11566 if( zLine[0]=='/' )
11567 zLine += 1; /* Oracle */
11568 else if ( ToLower(zLine[0])=='g' && ToLower(zLine[1])=='o' )
11569 zLine += 2; /* SQL Server */
11570 else
11571 return 0;
11572 return quickscan(zLine, QSS_Start, 0)==QSS_Start;
11576 ** The CLI needs a working sqlite3_complete() to work properly. So error
11577 ** out of the build if compiling with SQLITE_OMIT_COMPLETE.
11579 #ifdef SQLITE_OMIT_COMPLETE
11580 # error the CLI application is imcompatable with SQLITE_OMIT_COMPLETE.
11581 #endif
11584 ** Return true if zSql is a complete SQL statement. Return false if it
11585 ** ends in the middle of a string literal or C-style comment.
11587 static int line_is_complete(char *zSql, int nSql){
11588 int rc;
11589 if( zSql==0 ) return 1;
11590 zSql[nSql] = ';';
11591 zSql[nSql+1] = 0;
11592 rc = sqlite3_complete(zSql);
11593 zSql[nSql] = 0;
11594 return rc;
11598 ** This function is called after processing each line of SQL in the
11599 ** runOneSqlLine() function. Its purpose is to detect scenarios where
11600 ** defensive mode should be automatically turned off. Specifically, when
11602 ** 1. The first line of input is "PRAGMA foreign_keys=OFF;",
11603 ** 2. The second line of input is "BEGIN TRANSACTION;",
11604 ** 3. The database is empty, and
11605 ** 4. The shell is not running in --safe mode.
11607 ** The implementation uses the ShellState.eRestoreState to maintain state:
11609 ** 0: Have not seen any SQL.
11610 ** 1: Have seen "PRAGMA foreign_keys=OFF;".
11611 ** 2-6: Currently running .dump transaction. If the "2" bit is set,
11612 ** disable DEFENSIVE when done. If "4" is set, disable DQS_DDL.
11613 ** 7: Nothing left to do. This function becomes a no-op.
11615 static int doAutoDetectRestore(ShellState *p, const char *zSql){
11616 int rc = SQLITE_OK;
11618 if( p->eRestoreState<7 ){
11619 switch( p->eRestoreState ){
11620 case 0: {
11621 const char *zExpect = "PRAGMA foreign_keys=OFF;";
11622 assert( strlen(zExpect)==24 );
11623 if( p->bSafeMode==0 && memcmp(zSql, zExpect, 25)==0 ){
11624 p->eRestoreState = 1;
11625 }else{
11626 p->eRestoreState = 7;
11628 break;
11631 case 1: {
11632 int bIsDump = 0;
11633 const char *zExpect = "BEGIN TRANSACTION;";
11634 assert( strlen(zExpect)==18 );
11635 if( memcmp(zSql, zExpect, 19)==0 ){
11636 /* Now check if the database is empty. */
11637 const char *zQuery = "SELECT 1 FROM sqlite_schema LIMIT 1";
11638 sqlite3_stmt *pStmt = 0;
11640 bIsDump = 1;
11641 shellPrepare(p->db, &rc, zQuery, &pStmt);
11642 if( rc==SQLITE_OK && sqlite3_step(pStmt)==SQLITE_ROW ){
11643 bIsDump = 0;
11645 shellFinalize(&rc, pStmt);
11647 if( bIsDump && rc==SQLITE_OK ){
11648 int bDefense = 0;
11649 int bDqsDdl = 0;
11650 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, -1, &bDefense);
11651 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DQS_DDL, -1, &bDqsDdl);
11652 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, 0, 0);
11653 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DQS_DDL, 1, 0);
11654 p->eRestoreState = (bDefense ? 2 : 0) + (bDqsDdl ? 4 : 0);
11655 }else{
11656 p->eRestoreState = 7;
11658 break;
11661 default: {
11662 if( sqlite3_get_autocommit(p->db) ){
11663 if( (p->eRestoreState & 2) ){
11664 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DEFENSIVE, 1, 0);
11666 if( (p->eRestoreState & 4) ){
11667 sqlite3_db_config(p->db, SQLITE_DBCONFIG_DQS_DDL, 0, 0);
11669 p->eRestoreState = 7;
11671 break;
11676 return rc;
11680 ** Run a single line of SQL. Return the number of errors.
11682 static int runOneSqlLine(ShellState *p, char *zSql, FILE *in, int startline){
11683 int rc;
11684 char *zErrMsg = 0;
11686 open_db(p, 0);
11687 if( ShellHasFlag(p,SHFLG_Backslash) ) resolve_backslashes(zSql);
11688 if( p->flgProgress & SHELL_PROGRESS_RESET ) p->nProgress = 0;
11689 BEGIN_TIMER;
11690 rc = shell_exec(p, zSql, &zErrMsg);
11691 END_TIMER;
11692 if( rc || zErrMsg ){
11693 char zPrefix[100];
11694 const char *zErrorTail;
11695 const char *zErrorType;
11696 if( zErrMsg==0 ){
11697 zErrorType = "Error";
11698 zErrorTail = sqlite3_errmsg(p->db);
11699 }else if( cli_strncmp(zErrMsg, "in prepare, ",12)==0 ){
11700 zErrorType = "Parse error";
11701 zErrorTail = &zErrMsg[12];
11702 }else if( cli_strncmp(zErrMsg, "stepping, ", 10)==0 ){
11703 zErrorType = "Runtime error";
11704 zErrorTail = &zErrMsg[10];
11705 }else{
11706 zErrorType = "Error";
11707 zErrorTail = zErrMsg;
11709 if( in!=0 || !stdin_is_interactive ){
11710 sqlite3_snprintf(sizeof(zPrefix), zPrefix,
11711 "%s near line %d:", zErrorType, startline);
11712 }else{
11713 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "%s:", zErrorType);
11715 eputf("%s %s\n", zPrefix, zErrorTail);
11716 sqlite3_free(zErrMsg);
11717 zErrMsg = 0;
11718 return 1;
11719 }else if( ShellHasFlag(p, SHFLG_CountChanges) ){
11720 char zLineBuf[2000];
11721 sqlite3_snprintf(sizeof(zLineBuf), zLineBuf,
11722 "changes: %lld total_changes: %lld",
11723 sqlite3_changes64(p->db), sqlite3_total_changes64(p->db));
11724 oputf("%s\n", zLineBuf);
11727 if( doAutoDetectRestore(p, zSql) ) return 1;
11728 return 0;
11731 static void echo_group_input(ShellState *p, const char *zDo){
11732 if( ShellHasFlag(p, SHFLG_Echo) ) oputf("%s\n", zDo);
11735 #ifdef SQLITE_SHELL_FIDDLE
11737 ** Alternate one_input_line() impl for wasm mode. This is not in the primary
11738 ** impl because we need the global shellState and cannot access it from that
11739 ** function without moving lots of code around (creating a larger/messier diff).
11741 static char *one_input_line(FILE *in, char *zPrior, int isContinuation){
11742 /* Parse the next line from shellState.wasm.zInput. */
11743 const char *zBegin = shellState.wasm.zPos;
11744 const char *z = zBegin;
11745 char *zLine = 0;
11746 i64 nZ = 0;
11748 UNUSED_PARAMETER(in);
11749 UNUSED_PARAMETER(isContinuation);
11750 if(!z || !*z){
11751 return 0;
11753 while(*z && isspace(*z)) ++z;
11754 zBegin = z;
11755 for(; *z && '\n'!=*z; ++nZ, ++z){}
11756 if(nZ>0 && '\r'==zBegin[nZ-1]){
11757 --nZ;
11759 shellState.wasm.zPos = z;
11760 zLine = realloc(zPrior, nZ+1);
11761 shell_check_oom(zLine);
11762 memcpy(zLine, zBegin, nZ);
11763 zLine[nZ] = 0;
11764 return zLine;
11766 #endif /* SQLITE_SHELL_FIDDLE */
11769 ** Read input from *in and process it. If *in==0 then input
11770 ** is interactive - the user is typing it it. Otherwise, input
11771 ** is coming from a file or device. A prompt is issued and history
11772 ** is saved only if input is interactive. An interrupt signal will
11773 ** cause this routine to exit immediately, unless input is interactive.
11775 ** Return the number of errors.
11777 static int process_input(ShellState *p){
11778 char *zLine = 0; /* A single input line */
11779 char *zSql = 0; /* Accumulated SQL text */
11780 i64 nLine; /* Length of current line */
11781 i64 nSql = 0; /* Bytes of zSql[] used */
11782 i64 nAlloc = 0; /* Allocated zSql[] space */
11783 int rc; /* Error code */
11784 int errCnt = 0; /* Number of errors seen */
11785 i64 startline = 0; /* Line number for start of current input */
11786 QuickScanState qss = QSS_Start; /* Accumulated line status (so far) */
11788 if( p->inputNesting==MAX_INPUT_NESTING ){
11789 /* This will be more informative in a later version. */
11790 eputf("Input nesting limit (%d) reached at line %d."
11791 " Check recursion.\n", MAX_INPUT_NESTING, p->lineno);
11792 return 1;
11794 ++p->inputNesting;
11795 p->lineno = 0;
11796 CONTINUE_PROMPT_RESET;
11797 while( errCnt==0 || !bail_on_error || (p->in==0 && stdin_is_interactive) ){
11798 fflush(p->out);
11799 zLine = one_input_line(p->in, zLine, nSql>0);
11800 if( zLine==0 ){
11801 /* End of input */
11802 if( p->in==0 && stdin_is_interactive ) oputz("\n");
11803 break;
11805 if( seenInterrupt ){
11806 if( p->in!=0 ) break;
11807 seenInterrupt = 0;
11809 p->lineno++;
11810 if( QSS_INPLAIN(qss)
11811 && line_is_command_terminator(zLine)
11812 && line_is_complete(zSql, nSql) ){
11813 memcpy(zLine,";",2);
11815 qss = quickscan(zLine, qss, CONTINUE_PROMPT_PSTATE);
11816 if( QSS_PLAINWHITE(qss) && nSql==0 ){
11817 /* Just swallow single-line whitespace */
11818 echo_group_input(p, zLine);
11819 qss = QSS_Start;
11820 continue;
11822 if( zLine && (zLine[0]=='.' || zLine[0]=='#') && nSql==0 ){
11823 CONTINUE_PROMPT_RESET;
11824 echo_group_input(p, zLine);
11825 if( zLine[0]=='.' ){
11826 rc = do_meta_command(zLine, p);
11827 if( rc==2 ){ /* exit requested */
11828 break;
11829 }else if( rc ){
11830 errCnt++;
11833 qss = QSS_Start;
11834 continue;
11836 /* No single-line dispositions remain; accumulate line(s). */
11837 nLine = strlen(zLine);
11838 if( nSql+nLine+2>=nAlloc ){
11839 /* Grow buffer by half-again increments when big. */
11840 nAlloc = nSql+(nSql>>1)+nLine+100;
11841 zSql = realloc(zSql, nAlloc);
11842 shell_check_oom(zSql);
11844 if( nSql==0 ){
11845 i64 i;
11846 for(i=0; zLine[i] && IsSpace(zLine[i]); i++){}
11847 assert( nAlloc>0 && zSql!=0 );
11848 memcpy(zSql, zLine+i, nLine+1-i);
11849 startline = p->lineno;
11850 nSql = nLine-i;
11851 }else{
11852 zSql[nSql++] = '\n';
11853 memcpy(zSql+nSql, zLine, nLine+1);
11854 nSql += nLine;
11856 if( nSql && QSS_SEMITERM(qss) && sqlite3_complete(zSql) ){
11857 echo_group_input(p, zSql);
11858 errCnt += runOneSqlLine(p, zSql, p->in, startline);
11859 CONTINUE_PROMPT_RESET;
11860 nSql = 0;
11861 if( p->outCount ){
11862 output_reset(p);
11863 p->outCount = 0;
11864 }else{
11865 clearTempFile(p);
11867 p->bSafeMode = p->bSafeModePersist;
11868 qss = QSS_Start;
11869 }else if( nSql && QSS_PLAINWHITE(qss) ){
11870 echo_group_input(p, zSql);
11871 nSql = 0;
11872 qss = QSS_Start;
11875 if( nSql ){
11876 /* This may be incomplete. Let the SQL parser deal with that. */
11877 echo_group_input(p, zSql);
11878 errCnt += runOneSqlLine(p, zSql, p->in, startline);
11879 CONTINUE_PROMPT_RESET;
11881 free(zSql);
11882 free(zLine);
11883 --p->inputNesting;
11884 return errCnt>0;
11888 ** Return a pathname which is the user's home directory. A
11889 ** 0 return indicates an error of some kind.
11891 static char *find_home_dir(int clearFlag){
11892 static char *home_dir = NULL;
11893 if( clearFlag ){
11894 free(home_dir);
11895 home_dir = 0;
11896 return 0;
11898 if( home_dir ) return home_dir;
11900 #if !defined(_WIN32) && !defined(WIN32) && !defined(_WIN32_WCE) \
11901 && !defined(__RTP__) && !defined(_WRS_KERNEL) && !defined(SQLITE_WASI)
11903 struct passwd *pwent;
11904 uid_t uid = getuid();
11905 if( (pwent=getpwuid(uid)) != NULL) {
11906 home_dir = pwent->pw_dir;
11909 #endif
11911 #if defined(_WIN32_WCE)
11912 /* Windows CE (arm-wince-mingw32ce-gcc) does not provide getenv()
11914 home_dir = "/";
11915 #else
11917 #if defined(_WIN32) || defined(WIN32)
11918 if (!home_dir) {
11919 home_dir = getenv("USERPROFILE");
11921 #endif
11923 if (!home_dir) {
11924 home_dir = getenv("HOME");
11927 #if defined(_WIN32) || defined(WIN32)
11928 if (!home_dir) {
11929 char *zDrive, *zPath;
11930 int n;
11931 zDrive = getenv("HOMEDRIVE");
11932 zPath = getenv("HOMEPATH");
11933 if( zDrive && zPath ){
11934 n = strlen30(zDrive) + strlen30(zPath) + 1;
11935 home_dir = malloc( n );
11936 if( home_dir==0 ) return 0;
11937 sqlite3_snprintf(n, home_dir, "%s%s", zDrive, zPath);
11938 return home_dir;
11940 home_dir = "c:\\";
11942 #endif
11944 #endif /* !_WIN32_WCE */
11946 if( home_dir ){
11947 i64 n = strlen(home_dir) + 1;
11948 char *z = malloc( n );
11949 if( z ) memcpy(z, home_dir, n);
11950 home_dir = z;
11953 return home_dir;
11957 ** On non-Windows platforms, look for $XDG_CONFIG_HOME.
11958 ** If ${XDG_CONFIG_HOME}/sqlite3/sqliterc is found, return
11959 ** the path to it, else return 0. The result is cached for
11960 ** subsequent calls.
11962 static const char *find_xdg_config(void){
11963 #if defined(_WIN32) || defined(WIN32) || defined(_WIN32_WCE) \
11964 || defined(__RTP__) || defined(_WRS_KERNEL)
11965 return 0;
11966 #else
11967 static int alreadyTried = 0;
11968 static char *zConfig = 0;
11969 const char *zXdgHome;
11971 if( alreadyTried!=0 ){
11972 return zConfig;
11974 alreadyTried = 1;
11975 zXdgHome = getenv("XDG_CONFIG_HOME");
11976 if( zXdgHome==0 ){
11977 return 0;
11979 zConfig = sqlite3_mprintf("%s/sqlite3/sqliterc", zXdgHome);
11980 shell_check_oom(zConfig);
11981 if( access(zConfig,0)!=0 ){
11982 sqlite3_free(zConfig);
11983 zConfig = 0;
11985 return zConfig;
11986 #endif
11990 ** Read input from the file given by sqliterc_override. Or if that
11991 ** parameter is NULL, take input from the first of find_xdg_config()
11992 ** or ~/.sqliterc which is found.
11994 ** Returns the number of errors.
11996 static void process_sqliterc(
11997 ShellState *p, /* Configuration data */
11998 const char *sqliterc_override /* Name of config file. NULL to use default */
12000 char *home_dir = NULL;
12001 const char *sqliterc = sqliterc_override;
12002 char *zBuf = 0;
12003 FILE *inSaved = p->in;
12004 int savedLineno = p->lineno;
12006 if( sqliterc == NULL ){
12007 sqliterc = find_xdg_config();
12009 if( sqliterc == NULL ){
12010 home_dir = find_home_dir(0);
12011 if( home_dir==0 ){
12012 eputz("-- warning: cannot find home directory;"
12013 " cannot read ~/.sqliterc\n");
12014 return;
12016 zBuf = sqlite3_mprintf("%s/.sqliterc",home_dir);
12017 shell_check_oom(zBuf);
12018 sqliterc = zBuf;
12020 p->in = fopen(sqliterc,"rb");
12021 if( p->in ){
12022 if( stdin_is_interactive ){
12023 eputf("-- Loading resources from %s\n", sqliterc);
12025 if( process_input(p) && bail_on_error ) exit(1);
12026 fclose(p->in);
12027 }else if( sqliterc_override!=0 ){
12028 eputf("cannot open: \"%s\"\n", sqliterc);
12029 if( bail_on_error ) exit(1);
12031 p->in = inSaved;
12032 p->lineno = savedLineno;
12033 sqlite3_free(zBuf);
12037 ** Show available command line options
12039 static const char zOptions[] =
12040 " -- treat no subsequent arguments as options\n"
12041 #if defined(SQLITE_HAVE_ZLIB) && !defined(SQLITE_OMIT_VIRTUALTABLE)
12042 " -A ARGS... run \".archive ARGS\" and exit\n"
12043 #endif
12044 " -append append the database to the end of the file\n"
12045 " -ascii set output mode to 'ascii'\n"
12046 " -bail stop after hitting an error\n"
12047 " -batch force batch I/O\n"
12048 " -box set output mode to 'box'\n"
12049 " -column set output mode to 'column'\n"
12050 " -cmd COMMAND run \"COMMAND\" before reading stdin\n"
12051 " -csv set output mode to 'csv'\n"
12052 #if !defined(SQLITE_OMIT_DESERIALIZE)
12053 " -deserialize open the database using sqlite3_deserialize()\n"
12054 #endif
12055 " -echo print inputs before execution\n"
12056 " -init FILENAME read/process named file\n"
12057 " -[no]header turn headers on or off\n"
12058 #if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
12059 " -heap SIZE Size of heap for memsys3 or memsys5\n"
12060 #endif
12061 " -help show this message\n"
12062 " -html set output mode to HTML\n"
12063 " -interactive force interactive I/O\n"
12064 " -json set output mode to 'json'\n"
12065 " -line set output mode to 'line'\n"
12066 " -list set output mode to 'list'\n"
12067 " -lookaside SIZE N use N entries of SZ bytes for lookaside memory\n"
12068 " -markdown set output mode to 'markdown'\n"
12069 #if !defined(SQLITE_OMIT_DESERIALIZE)
12070 " -maxsize N maximum size for a --deserialize database\n"
12071 #endif
12072 " -memtrace trace all memory allocations and deallocations\n"
12073 " -mmap N default mmap size set to N\n"
12074 #ifdef SQLITE_ENABLE_MULTIPLEX
12075 " -multiplex enable the multiplexor VFS\n"
12076 #endif
12077 " -newline SEP set output row separator. Default: '\\n'\n"
12078 " -nofollow refuse to open symbolic links to database files\n"
12079 " -nonce STRING set the safe-mode escape nonce\n"
12080 " -no-rowid-in-view Disable rowid-in-view using sqlite3_config()\n"
12081 " -nullvalue TEXT set text string for NULL values. Default ''\n"
12082 " -pagecache SIZE N use N slots of SZ bytes each for page cache memory\n"
12083 " -pcachetrace trace all page cache operations\n"
12084 " -quote set output mode to 'quote'\n"
12085 " -readonly open the database read-only\n"
12086 " -safe enable safe-mode\n"
12087 " -separator SEP set output column separator. Default: '|'\n"
12088 #ifdef SQLITE_ENABLE_SORTER_REFERENCES
12089 " -sorterref SIZE sorter references threshold size\n"
12090 #endif
12091 " -stats print memory stats before each finalize\n"
12092 " -table set output mode to 'table'\n"
12093 " -tabs set output mode to 'tabs'\n"
12094 " -unsafe-testing allow unsafe commands and modes for testing\n"
12095 " -version show SQLite version\n"
12096 " -vfs NAME use NAME as the default VFS\n"
12097 #ifdef SQLITE_ENABLE_VFSTRACE
12098 " -vfstrace enable tracing of all VFS calls\n"
12099 #endif
12100 #ifdef SQLITE_HAVE_ZLIB
12101 " -zip open the file as a ZIP Archive\n"
12102 #endif
12104 static void usage(int showDetail){
12105 eputf("Usage: %s [OPTIONS] [FILENAME [SQL]]\n"
12106 "FILENAME is the name of an SQLite database. A new database is created\n"
12107 "if the file does not previously exist. Defaults to :memory:.\n", Argv0);
12108 if( showDetail ){
12109 eputf("OPTIONS include:\n%s", zOptions);
12110 }else{
12111 eputz("Use the -help option for additional information\n");
12113 exit(0);
12117 ** Internal check: Verify that the SQLite is uninitialized. Print a
12118 ** error message if it is initialized.
12120 static void verify_uninitialized(void){
12121 if( sqlite3_config(-1)==SQLITE_MISUSE ){
12122 sputz(stdout, "WARNING: attempt to configure SQLite after"
12123 " initialization.\n");
12128 ** Initialize the state information in data
12130 static void main_init(ShellState *data) {
12131 memset(data, 0, sizeof(*data));
12132 data->normalMode = data->cMode = data->mode = MODE_List;
12133 data->autoExplain = 1;
12134 data->pAuxDb = &data->aAuxDb[0];
12135 memcpy(data->colSeparator,SEP_Column, 2);
12136 memcpy(data->rowSeparator,SEP_Row, 2);
12137 data->showHeader = 0;
12138 data->shellFlgs = SHFLG_Lookaside;
12139 sqlite3_config(SQLITE_CONFIG_LOG, shellLog, data);
12140 #if !defined(SQLITE_SHELL_FIDDLE)
12141 verify_uninitialized();
12142 #endif
12143 sqlite3_config(SQLITE_CONFIG_URI, 1);
12144 sqlite3_config(SQLITE_CONFIG_MULTITHREAD);
12145 sqlite3_snprintf(sizeof(mainPrompt), mainPrompt,"sqlite> ");
12146 sqlite3_snprintf(sizeof(continuePrompt), continuePrompt," ...> ");
12150 ** Output text to the console in a font that attracts extra attention.
12152 #if defined(_WIN32) || defined(WIN32)
12153 static void printBold(const char *zText){
12154 #if !SQLITE_OS_WINRT
12155 HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
12156 CONSOLE_SCREEN_BUFFER_INFO defaultScreenInfo;
12157 GetConsoleScreenBufferInfo(out, &defaultScreenInfo);
12158 SetConsoleTextAttribute(out,
12159 FOREGROUND_RED|FOREGROUND_INTENSITY
12161 #endif
12162 sputz(stdout, zText);
12163 #if !SQLITE_OS_WINRT
12164 SetConsoleTextAttribute(out, defaultScreenInfo.wAttributes);
12165 #endif
12167 #else
12168 static void printBold(const char *zText){
12169 sputf(stdout, "\033[1m%s\033[0m", zText);
12171 #endif
12174 ** Get the argument to an --option. Throw an error and die if no argument
12175 ** is available.
12177 static char *cmdline_option_value(int argc, char **argv, int i){
12178 if( i==argc ){
12179 eputf("%s: Error: missing argument to %s\n", argv[0], argv[argc-1]);
12180 exit(1);
12182 return argv[i];
12185 static void sayAbnormalExit(void){
12186 if( seenInterrupt ) eputz("Program interrupted.\n");
12189 #ifndef SQLITE_SHELL_IS_UTF8
12190 # if (defined(_WIN32) || defined(WIN32)) \
12191 && (defined(_MSC_VER) || (defined(UNICODE) && defined(__GNUC__)))
12192 # define SQLITE_SHELL_IS_UTF8 (0)
12193 # else
12194 # define SQLITE_SHELL_IS_UTF8 (1)
12195 # endif
12196 #endif
12198 #ifdef SQLITE_SHELL_FIDDLE
12199 # define main fiddle_main
12200 #endif
12202 #if SQLITE_SHELL_IS_UTF8
12203 int SQLITE_CDECL main(int argc, char **argv){
12204 #else
12205 int SQLITE_CDECL wmain(int argc, wchar_t **wargv){
12206 char **argv;
12207 #endif
12208 #ifdef SQLITE_DEBUG
12209 sqlite3_int64 mem_main_enter = 0;
12210 #endif
12211 char *zErrMsg = 0;
12212 #ifdef SQLITE_SHELL_FIDDLE
12213 # define data shellState
12214 #else
12215 ShellState data;
12216 StreamsAreConsole consStreams = SAC_NoConsole;
12217 #endif
12218 const char *zInitFile = 0;
12219 int i;
12220 int rc = 0;
12221 int warnInmemoryDb = 0;
12222 int readStdin = 1;
12223 int nCmd = 0;
12224 int nOptsEnd = argc;
12225 char **azCmd = 0;
12226 const char *zVfs = 0; /* Value of -vfs command-line option */
12227 #if !SQLITE_SHELL_IS_UTF8
12228 char **argvToFree = 0;
12229 int argcToFree = 0;
12230 #endif
12231 setvbuf(stderr, 0, _IONBF, 0); /* Make sure stderr is unbuffered */
12233 #ifdef SQLITE_SHELL_FIDDLE
12234 stdin_is_interactive = 0;
12235 stdout_is_console = 1;
12236 data.wasm.zDefaultDbName = "/fiddle.sqlite3";
12237 #else
12238 consStreams = consoleClassifySetup(stdin, stdout, stderr);
12239 stdin_is_interactive = (consStreams & SAC_InConsole)!=0;
12240 stdout_is_console = (consStreams & SAC_OutConsole)!=0;
12241 atexit(consoleRestore);
12242 #endif
12243 atexit(sayAbnormalExit);
12244 #ifdef SQLITE_DEBUG
12245 mem_main_enter = sqlite3_memory_used();
12246 #endif
12247 #if !defined(_WIN32_WCE)
12248 if( getenv("SQLITE_DEBUG_BREAK") ){
12249 if( isatty(0) && isatty(2) ){
12250 eputf("attach debugger to process %d and press any key to continue.\n",
12251 GETPID());
12252 fgetc(stdin);
12253 }else{
12254 #if defined(_WIN32) || defined(WIN32)
12255 #if SQLITE_OS_WINRT
12256 __debugbreak();
12257 #else
12258 DebugBreak();
12259 #endif
12260 #elif defined(SIGTRAP)
12261 raise(SIGTRAP);
12262 #endif
12265 #endif
12266 /* Register a valid signal handler early, before much else is done. */
12267 #ifdef SIGINT
12268 signal(SIGINT, interrupt_handler);
12269 #elif (defined(_WIN32) || defined(WIN32)) && !defined(_WIN32_WCE)
12270 if( !SetConsoleCtrlHandler(ConsoleCtrlHandler, TRUE) ){
12271 eputz("No ^C handler.\n");
12273 #endif
12275 #if USE_SYSTEM_SQLITE+0!=1
12276 if( cli_strncmp(sqlite3_sourceid(),SQLITE_SOURCE_ID,60)!=0 ){
12277 eputf("SQLite header and source version mismatch\n%s\n%s\n",
12278 sqlite3_sourceid(), SQLITE_SOURCE_ID);
12279 exit(1);
12281 #endif
12282 main_init(&data);
12284 /* On Windows, we must translate command-line arguments into UTF-8.
12285 ** The SQLite memory allocator subsystem has to be enabled in order to
12286 ** do this. But we want to run an sqlite3_shutdown() afterwards so that
12287 ** subsequent sqlite3_config() calls will work. So copy all results into
12288 ** memory that does not come from the SQLite memory allocator.
12290 #if !SQLITE_SHELL_IS_UTF8
12291 sqlite3_initialize();
12292 argvToFree = malloc(sizeof(argv[0])*argc*2);
12293 shell_check_oom(argvToFree);
12294 argcToFree = argc;
12295 argv = argvToFree + argc;
12296 for(i=0; i<argc; i++){
12297 char *z = sqlite3_win32_unicode_to_utf8(wargv[i]);
12298 i64 n;
12299 shell_check_oom(z);
12300 n = strlen(z);
12301 argv[i] = malloc( n+1 );
12302 shell_check_oom(argv[i]);
12303 memcpy(argv[i], z, n+1);
12304 argvToFree[i] = argv[i];
12305 sqlite3_free(z);
12307 sqlite3_shutdown();
12308 #endif
12310 assert( argc>=1 && argv && argv[0] );
12311 Argv0 = argv[0];
12313 #ifdef SQLITE_SHELL_DBNAME_PROC
12315 /* If the SQLITE_SHELL_DBNAME_PROC macro is defined, then it is the name
12316 ** of a C-function that will provide the name of the database file. Use
12317 ** this compile-time option to embed this shell program in larger
12318 ** applications. */
12319 extern void SQLITE_SHELL_DBNAME_PROC(const char**);
12320 SQLITE_SHELL_DBNAME_PROC(&data.pAuxDb->zDbFilename);
12321 warnInmemoryDb = 0;
12323 #endif
12325 /* Do an initial pass through the command-line argument to locate
12326 ** the name of the database file, the name of the initialization file,
12327 ** the size of the alternative malloc heap, options affecting commands
12328 ** or SQL run from the command line, and the first command to execute.
12330 #ifndef SQLITE_SHELL_FIDDLE
12331 verify_uninitialized();
12332 #endif
12333 for(i=1; i<argc; i++){
12334 char *z;
12335 z = argv[i];
12336 if( z[0]!='-' || i>nOptsEnd ){
12337 if( data.aAuxDb->zDbFilename==0 ){
12338 data.aAuxDb->zDbFilename = z;
12339 }else{
12340 /* Excess arguments are interpreted as SQL (or dot-commands) and
12341 ** mean that nothing is read from stdin */
12342 readStdin = 0;
12343 nCmd++;
12344 azCmd = realloc(azCmd, sizeof(azCmd[0])*nCmd);
12345 shell_check_oom(azCmd);
12346 azCmd[nCmd-1] = z;
12348 continue;
12350 if( z[1]=='-' ) z++;
12351 if( cli_strcmp(z, "-")==0 ){
12352 nOptsEnd = i;
12353 continue;
12354 }else if( cli_strcmp(z,"-separator")==0
12355 || cli_strcmp(z,"-nullvalue")==0
12356 || cli_strcmp(z,"-newline")==0
12357 || cli_strcmp(z,"-cmd")==0
12359 (void)cmdline_option_value(argc, argv, ++i);
12360 }else if( cli_strcmp(z,"-init")==0 ){
12361 zInitFile = cmdline_option_value(argc, argv, ++i);
12362 }else if( cli_strcmp(z,"-interactive")==0 ){
12363 }else if( cli_strcmp(z,"-batch")==0 ){
12364 /* Need to check for batch mode here to so we can avoid printing
12365 ** informational messages (like from process_sqliterc) before
12366 ** we do the actual processing of arguments later in a second pass.
12368 stdin_is_interactive = 0;
12369 }else if( cli_strcmp(z,"-utf8")==0 ){
12370 }else if( cli_strcmp(z,"-no-utf8")==0 ){
12371 }else if( cli_strcmp(z,"-no-rowid-in-view")==0 ){
12372 int val = 0;
12373 sqlite3_config(SQLITE_CONFIG_ROWID_IN_VIEW, &val);
12374 assert( val==0 );
12375 }else if( cli_strcmp(z,"-heap")==0 ){
12376 #if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
12377 const char *zSize;
12378 sqlite3_int64 szHeap;
12380 zSize = cmdline_option_value(argc, argv, ++i);
12381 szHeap = integerValue(zSize);
12382 if( szHeap>0x7fff0000 ) szHeap = 0x7fff0000;
12383 verify_uninitialized();
12384 sqlite3_config(SQLITE_CONFIG_HEAP, malloc((int)szHeap), (int)szHeap, 64);
12385 #else
12386 (void)cmdline_option_value(argc, argv, ++i);
12387 #endif
12388 }else if( cli_strcmp(z,"-pagecache")==0 ){
12389 sqlite3_int64 n, sz;
12390 sz = integerValue(cmdline_option_value(argc,argv,++i));
12391 if( sz>70000 ) sz = 70000;
12392 if( sz<0 ) sz = 0;
12393 n = integerValue(cmdline_option_value(argc,argv,++i));
12394 if( sz>0 && n>0 && 0xffffffffffffLL/sz<n ){
12395 n = 0xffffffffffffLL/sz;
12397 verify_uninitialized();
12398 sqlite3_config(SQLITE_CONFIG_PAGECACHE,
12399 (n>0 && sz>0) ? malloc(n*sz) : 0, sz, n);
12400 data.shellFlgs |= SHFLG_Pagecache;
12401 }else if( cli_strcmp(z,"-lookaside")==0 ){
12402 int n, sz;
12403 sz = (int)integerValue(cmdline_option_value(argc,argv,++i));
12404 if( sz<0 ) sz = 0;
12405 n = (int)integerValue(cmdline_option_value(argc,argv,++i));
12406 if( n<0 ) n = 0;
12407 verify_uninitialized();
12408 sqlite3_config(SQLITE_CONFIG_LOOKASIDE, sz, n);
12409 if( sz*n==0 ) data.shellFlgs &= ~SHFLG_Lookaside;
12410 }else if( cli_strcmp(z,"-threadsafe")==0 ){
12411 int n;
12412 n = (int)integerValue(cmdline_option_value(argc,argv,++i));
12413 verify_uninitialized();
12414 switch( n ){
12415 case 0: sqlite3_config(SQLITE_CONFIG_SINGLETHREAD); break;
12416 case 2: sqlite3_config(SQLITE_CONFIG_MULTITHREAD); break;
12417 default: sqlite3_config(SQLITE_CONFIG_SERIALIZED); break;
12419 #ifdef SQLITE_ENABLE_VFSTRACE
12420 }else if( cli_strcmp(z,"-vfstrace")==0 ){
12421 extern int vfstrace_register(
12422 const char *zTraceName,
12423 const char *zOldVfsName,
12424 int (*xOut)(const char*,void*),
12425 void *pOutArg,
12426 int makeDefault
12428 vfstrace_register("trace",0,(int(*)(const char*,void*))fputs,stderr,1);
12429 #endif
12430 #ifdef SQLITE_ENABLE_MULTIPLEX
12431 }else if( cli_strcmp(z,"-multiplex")==0 ){
12432 extern int sqlite3_multiplex_initialize(const char*,int);
12433 sqlite3_multiplex_initialize(0, 1);
12434 #endif
12435 }else if( cli_strcmp(z,"-mmap")==0 ){
12436 sqlite3_int64 sz = integerValue(cmdline_option_value(argc,argv,++i));
12437 verify_uninitialized();
12438 sqlite3_config(SQLITE_CONFIG_MMAP_SIZE, sz, sz);
12439 #if defined(SQLITE_ENABLE_SORTER_REFERENCES)
12440 }else if( cli_strcmp(z,"-sorterref")==0 ){
12441 sqlite3_int64 sz = integerValue(cmdline_option_value(argc,argv,++i));
12442 verify_uninitialized();
12443 sqlite3_config(SQLITE_CONFIG_SORTERREF_SIZE, (int)sz);
12444 #endif
12445 }else if( cli_strcmp(z,"-vfs")==0 ){
12446 zVfs = cmdline_option_value(argc, argv, ++i);
12447 #ifdef SQLITE_HAVE_ZLIB
12448 }else if( cli_strcmp(z,"-zip")==0 ){
12449 data.openMode = SHELL_OPEN_ZIPFILE;
12450 #endif
12451 }else if( cli_strcmp(z,"-append")==0 ){
12452 data.openMode = SHELL_OPEN_APPENDVFS;
12453 #ifndef SQLITE_OMIT_DESERIALIZE
12454 }else if( cli_strcmp(z,"-deserialize")==0 ){
12455 data.openMode = SHELL_OPEN_DESERIALIZE;
12456 }else if( cli_strcmp(z,"-maxsize")==0 && i+1<argc ){
12457 data.szMax = integerValue(argv[++i]);
12458 #endif
12459 }else if( cli_strcmp(z,"-readonly")==0 ){
12460 data.openMode = SHELL_OPEN_READONLY;
12461 }else if( cli_strcmp(z,"-nofollow")==0 ){
12462 data.openFlags = SQLITE_OPEN_NOFOLLOW;
12463 #if !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_HAVE_ZLIB)
12464 }else if( cli_strncmp(z, "-A",2)==0 ){
12465 /* All remaining command-line arguments are passed to the ".archive"
12466 ** command, so ignore them */
12467 break;
12468 #endif
12469 }else if( cli_strcmp(z, "-memtrace")==0 ){
12470 sqlite3MemTraceActivate(stderr);
12471 }else if( cli_strcmp(z, "-pcachetrace")==0 ){
12472 sqlite3PcacheTraceActivate(stderr);
12473 }else if( cli_strcmp(z,"-bail")==0 ){
12474 bail_on_error = 1;
12475 }else if( cli_strcmp(z,"-nonce")==0 ){
12476 free(data.zNonce);
12477 data.zNonce = strdup(cmdline_option_value(argc, argv, ++i));
12478 }else if( cli_strcmp(z,"-unsafe-testing")==0 ){
12479 ShellSetFlag(&data,SHFLG_TestingMode);
12480 }else if( cli_strcmp(z,"-safe")==0 ){
12481 /* no-op - catch this on the second pass */
12484 #ifndef SQLITE_SHELL_FIDDLE
12485 verify_uninitialized();
12486 #endif
12489 #ifdef SQLITE_SHELL_INIT_PROC
12491 /* If the SQLITE_SHELL_INIT_PROC macro is defined, then it is the name
12492 ** of a C-function that will perform initialization actions on SQLite that
12493 ** occur just before or after sqlite3_initialize(). Use this compile-time
12494 ** option to embed this shell program in larger applications. */
12495 extern void SQLITE_SHELL_INIT_PROC(void);
12496 SQLITE_SHELL_INIT_PROC();
12498 #else
12499 /* All the sqlite3_config() calls have now been made. So it is safe
12500 ** to call sqlite3_initialize() and process any command line -vfs option. */
12501 sqlite3_initialize();
12502 #endif
12504 if( zVfs ){
12505 sqlite3_vfs *pVfs = sqlite3_vfs_find(zVfs);
12506 if( pVfs ){
12507 sqlite3_vfs_register(pVfs, 1);
12508 }else{
12509 eputf("no such VFS: \"%s\"\n", zVfs);
12510 exit(1);
12514 if( data.pAuxDb->zDbFilename==0 ){
12515 #ifndef SQLITE_OMIT_MEMORYDB
12516 data.pAuxDb->zDbFilename = ":memory:";
12517 warnInmemoryDb = argc==1;
12518 #else
12519 eputf("%s: Error: no database filename specified\n", Argv0);
12520 return 1;
12521 #endif
12523 data.out = stdout;
12524 #ifndef SQLITE_SHELL_FIDDLE
12525 sqlite3_appendvfs_init(0,0,0);
12526 #endif
12528 /* Go ahead and open the database file if it already exists. If the
12529 ** file does not exist, delay opening it. This prevents empty database
12530 ** files from being created if a user mistypes the database name argument
12531 ** to the sqlite command-line tool.
12533 if( access(data.pAuxDb->zDbFilename, 0)==0 ){
12534 open_db(&data, 0);
12537 /* Process the initialization file if there is one. If no -init option
12538 ** is given on the command line, look for a file named ~/.sqliterc and
12539 ** try to process it.
12541 process_sqliterc(&data,zInitFile);
12543 /* Make a second pass through the command-line argument and set
12544 ** options. This second pass is delayed until after the initialization
12545 ** file is processed so that the command-line arguments will override
12546 ** settings in the initialization file.
12548 for(i=1; i<argc; i++){
12549 char *z = argv[i];
12550 if( z[0]!='-' || i>=nOptsEnd ) continue;
12551 if( z[1]=='-' ){ z++; }
12552 if( cli_strcmp(z,"-init")==0 ){
12553 i++;
12554 }else if( cli_strcmp(z,"-html")==0 ){
12555 data.mode = MODE_Html;
12556 }else if( cli_strcmp(z,"-list")==0 ){
12557 data.mode = MODE_List;
12558 }else if( cli_strcmp(z,"-quote")==0 ){
12559 data.mode = MODE_Quote;
12560 sqlite3_snprintf(sizeof(data.colSeparator), data.colSeparator, SEP_Comma);
12561 sqlite3_snprintf(sizeof(data.rowSeparator), data.rowSeparator, SEP_Row);
12562 }else if( cli_strcmp(z,"-line")==0 ){
12563 data.mode = MODE_Line;
12564 }else if( cli_strcmp(z,"-column")==0 ){
12565 data.mode = MODE_Column;
12566 }else if( cli_strcmp(z,"-json")==0 ){
12567 data.mode = MODE_Json;
12568 }else if( cli_strcmp(z,"-markdown")==0 ){
12569 data.mode = MODE_Markdown;
12570 }else if( cli_strcmp(z,"-table")==0 ){
12571 data.mode = MODE_Table;
12572 }else if( cli_strcmp(z,"-box")==0 ){
12573 data.mode = MODE_Box;
12574 }else if( cli_strcmp(z,"-csv")==0 ){
12575 data.mode = MODE_Csv;
12576 memcpy(data.colSeparator,",",2);
12577 #ifdef SQLITE_HAVE_ZLIB
12578 }else if( cli_strcmp(z,"-zip")==0 ){
12579 data.openMode = SHELL_OPEN_ZIPFILE;
12580 #endif
12581 }else if( cli_strcmp(z,"-append")==0 ){
12582 data.openMode = SHELL_OPEN_APPENDVFS;
12583 #ifndef SQLITE_OMIT_DESERIALIZE
12584 }else if( cli_strcmp(z,"-deserialize")==0 ){
12585 data.openMode = SHELL_OPEN_DESERIALIZE;
12586 }else if( cli_strcmp(z,"-maxsize")==0 && i+1<argc ){
12587 data.szMax = integerValue(argv[++i]);
12588 #endif
12589 }else if( cli_strcmp(z,"-readonly")==0 ){
12590 data.openMode = SHELL_OPEN_READONLY;
12591 }else if( cli_strcmp(z,"-nofollow")==0 ){
12592 data.openFlags |= SQLITE_OPEN_NOFOLLOW;
12593 }else if( cli_strcmp(z,"-ascii")==0 ){
12594 data.mode = MODE_Ascii;
12595 sqlite3_snprintf(sizeof(data.colSeparator), data.colSeparator,SEP_Unit);
12596 sqlite3_snprintf(sizeof(data.rowSeparator), data.rowSeparator,SEP_Record);
12597 }else if( cli_strcmp(z,"-tabs")==0 ){
12598 data.mode = MODE_List;
12599 sqlite3_snprintf(sizeof(data.colSeparator), data.colSeparator,SEP_Tab);
12600 sqlite3_snprintf(sizeof(data.rowSeparator), data.rowSeparator,SEP_Row);
12601 }else if( cli_strcmp(z,"-separator")==0 ){
12602 sqlite3_snprintf(sizeof(data.colSeparator), data.colSeparator,
12603 "%s",cmdline_option_value(argc,argv,++i));
12604 }else if( cli_strcmp(z,"-newline")==0 ){
12605 sqlite3_snprintf(sizeof(data.rowSeparator), data.rowSeparator,
12606 "%s",cmdline_option_value(argc,argv,++i));
12607 }else if( cli_strcmp(z,"-nullvalue")==0 ){
12608 sqlite3_snprintf(sizeof(data.nullValue), data.nullValue,
12609 "%s",cmdline_option_value(argc,argv,++i));
12610 }else if( cli_strcmp(z,"-header")==0 ){
12611 data.showHeader = 1;
12612 ShellSetFlag(&data, SHFLG_HeaderSet);
12613 }else if( cli_strcmp(z,"-noheader")==0 ){
12614 data.showHeader = 0;
12615 ShellSetFlag(&data, SHFLG_HeaderSet);
12616 }else if( cli_strcmp(z,"-echo")==0 ){
12617 ShellSetFlag(&data, SHFLG_Echo);
12618 }else if( cli_strcmp(z,"-eqp")==0 ){
12619 data.autoEQP = AUTOEQP_on;
12620 }else if( cli_strcmp(z,"-eqpfull")==0 ){
12621 data.autoEQP = AUTOEQP_full;
12622 }else if( cli_strcmp(z,"-stats")==0 ){
12623 data.statsOn = 1;
12624 }else if( cli_strcmp(z,"-scanstats")==0 ){
12625 data.scanstatsOn = 1;
12626 }else if( cli_strcmp(z,"-backslash")==0 ){
12627 /* Undocumented command-line option: -backslash
12628 ** Causes C-style backslash escapes to be evaluated in SQL statements
12629 ** prior to sending the SQL into SQLite. Useful for injecting
12630 ** crazy bytes in the middle of SQL statements for testing and debugging.
12632 ShellSetFlag(&data, SHFLG_Backslash);
12633 }else if( cli_strcmp(z,"-bail")==0 ){
12634 /* No-op. The bail_on_error flag should already be set. */
12635 }else if( cli_strcmp(z,"-version")==0 ){
12636 sputf(stdout, "%s %s (%d-bit)\n",
12637 sqlite3_libversion(), sqlite3_sourceid(), 8*(int)sizeof(char*));
12638 return 0;
12639 }else if( cli_strcmp(z,"-interactive")==0 ){
12640 /* Need to check for interactive override here to so that it can
12641 ** affect console setup (for Windows only) and testing thereof.
12643 stdin_is_interactive = 1;
12644 }else if( cli_strcmp(z,"-batch")==0 ){
12645 /* already handled */
12646 }else if( cli_strcmp(z,"-utf8")==0 ){
12647 /* already handled */
12648 }else if( cli_strcmp(z,"-no-utf8")==0 ){
12649 /* already handled */
12650 }else if( cli_strcmp(z,"-no-rowid-in-view")==0 ){
12651 /* already handled */
12652 }else if( cli_strcmp(z,"-heap")==0 ){
12653 i++;
12654 }else if( cli_strcmp(z,"-pagecache")==0 ){
12655 i+=2;
12656 }else if( cli_strcmp(z,"-lookaside")==0 ){
12657 i+=2;
12658 }else if( cli_strcmp(z,"-threadsafe")==0 ){
12659 i+=2;
12660 }else if( cli_strcmp(z,"-nonce")==0 ){
12661 i += 2;
12662 }else if( cli_strcmp(z,"-mmap")==0 ){
12663 i++;
12664 }else if( cli_strcmp(z,"-memtrace")==0 ){
12665 i++;
12666 }else if( cli_strcmp(z,"-pcachetrace")==0 ){
12667 i++;
12668 #ifdef SQLITE_ENABLE_SORTER_REFERENCES
12669 }else if( cli_strcmp(z,"-sorterref")==0 ){
12670 i++;
12671 #endif
12672 }else if( cli_strcmp(z,"-vfs")==0 ){
12673 i++;
12674 #ifdef SQLITE_ENABLE_VFSTRACE
12675 }else if( cli_strcmp(z,"-vfstrace")==0 ){
12676 i++;
12677 #endif
12678 #ifdef SQLITE_ENABLE_MULTIPLEX
12679 }else if( cli_strcmp(z,"-multiplex")==0 ){
12680 i++;
12681 #endif
12682 }else if( cli_strcmp(z,"-help")==0 ){
12683 usage(1);
12684 }else if( cli_strcmp(z,"-cmd")==0 ){
12685 /* Run commands that follow -cmd first and separately from commands
12686 ** that simply appear on the command-line. This seems goofy. It would
12687 ** be better if all commands ran in the order that they appear. But
12688 ** we retain the goofy behavior for historical compatibility. */
12689 if( i==argc-1 ) break;
12690 z = cmdline_option_value(argc,argv,++i);
12691 if( z[0]=='.' ){
12692 rc = do_meta_command(z, &data);
12693 if( rc && bail_on_error ) return rc==2 ? 0 : rc;
12694 }else{
12695 open_db(&data, 0);
12696 rc = shell_exec(&data, z, &zErrMsg);
12697 if( zErrMsg!=0 ){
12698 eputf("Error: %s\n", zErrMsg);
12699 if( bail_on_error ) return rc!=0 ? rc : 1;
12700 }else if( rc!=0 ){
12701 eputf("Error: unable to process SQL \"%s\"\n", z);
12702 if( bail_on_error ) return rc;
12705 #if !defined(SQLITE_OMIT_VIRTUALTABLE) && defined(SQLITE_HAVE_ZLIB)
12706 }else if( cli_strncmp(z, "-A", 2)==0 ){
12707 if( nCmd>0 ){
12708 eputf("Error: cannot mix regular SQL or dot-commands"
12709 " with \"%s\"\n", z);
12710 return 1;
12712 open_db(&data, OPEN_DB_ZIPFILE);
12713 if( z[2] ){
12714 argv[i] = &z[2];
12715 arDotCommand(&data, 1, argv+(i-1), argc-(i-1));
12716 }else{
12717 arDotCommand(&data, 1, argv+i, argc-i);
12719 readStdin = 0;
12720 break;
12721 #endif
12722 }else if( cli_strcmp(z,"-safe")==0 ){
12723 data.bSafeMode = data.bSafeModePersist = 1;
12724 }else if( cli_strcmp(z,"-unsafe-testing")==0 ){
12725 /* Acted upon in first pass. */
12726 }else{
12727 eputf("%s: Error: unknown option: %s\n", Argv0, z);
12728 eputz("Use -help for a list of options.\n");
12729 return 1;
12731 data.cMode = data.mode;
12734 if( !readStdin ){
12735 /* Run all arguments that do not begin with '-' as if they were separate
12736 ** command-line inputs, except for the argToSkip argument which contains
12737 ** the database filename.
12739 for(i=0; i<nCmd; i++){
12740 if( azCmd[i][0]=='.' ){
12741 rc = do_meta_command(azCmd[i], &data);
12742 if( rc ){
12743 free(azCmd);
12744 return rc==2 ? 0 : rc;
12746 }else{
12747 open_db(&data, 0);
12748 echo_group_input(&data, azCmd[i]);
12749 rc = shell_exec(&data, azCmd[i], &zErrMsg);
12750 if( zErrMsg || rc ){
12751 if( zErrMsg!=0 ){
12752 eputf("Error: %s\n", zErrMsg);
12753 }else{
12754 eputf("Error: unable to process SQL: %s\n", azCmd[i]);
12756 sqlite3_free(zErrMsg);
12757 free(azCmd);
12758 return rc!=0 ? rc : 1;
12762 }else{
12763 /* Run commands received from standard input
12765 if( stdin_is_interactive ){
12766 char *zHome;
12767 char *zHistory;
12768 int nHistory;
12769 #if CIO_WIN_WC_XLATE
12770 # define SHELL_CIO_CHAR_SET (stdout_is_console? " (UTF-16 console I/O)" : "")
12771 #else
12772 # define SHELL_CIO_CHAR_SET ""
12773 #endif
12774 sputf(stdout, "SQLite version %s %.19s%s\n" /*extra-version-info*/
12775 "Enter \".help\" for usage hints.\n",
12776 sqlite3_libversion(), sqlite3_sourceid(), SHELL_CIO_CHAR_SET);
12777 if( warnInmemoryDb ){
12778 sputz(stdout, "Connected to a ");
12779 printBold("transient in-memory database");
12780 sputz(stdout, ".\nUse \".open FILENAME\" to reopen on a"
12781 " persistent database.\n");
12783 zHistory = getenv("SQLITE_HISTORY");
12784 if( zHistory ){
12785 zHistory = strdup(zHistory);
12786 }else if( (zHome = find_home_dir(0))!=0 ){
12787 nHistory = strlen30(zHome) + 20;
12788 if( (zHistory = malloc(nHistory))!=0 ){
12789 sqlite3_snprintf(nHistory, zHistory,"%s/.sqlite_history", zHome);
12792 if( zHistory ){ shell_read_history(zHistory); }
12793 #if HAVE_READLINE || HAVE_EDITLINE
12794 rl_attempted_completion_function = readline_completion;
12795 #elif HAVE_LINENOISE
12796 linenoiseSetCompletionCallback(linenoise_completion);
12797 #endif
12798 data.in = 0;
12799 rc = process_input(&data);
12800 if( zHistory ){
12801 shell_stifle_history(2000);
12802 shell_write_history(zHistory);
12803 free(zHistory);
12805 }else{
12806 data.in = stdin;
12807 rc = process_input(&data);
12810 #ifndef SQLITE_SHELL_FIDDLE
12811 /* In WASM mode we have to leave the db state in place so that
12812 ** client code can "push" SQL into it after this call returns. */
12813 #ifndef SQLITE_OMIT_VIRTUALTABLE
12814 if( data.expert.pExpert ){
12815 expertFinish(&data, 1, 0);
12817 #endif
12818 free(azCmd);
12819 set_table_name(&data, 0);
12820 if( data.db ){
12821 session_close_all(&data, -1);
12822 close_db(data.db);
12824 for(i=0; i<ArraySize(data.aAuxDb); i++){
12825 sqlite3_free(data.aAuxDb[i].zFreeOnClose);
12826 if( data.aAuxDb[i].db ){
12827 session_close_all(&data, i);
12828 close_db(data.aAuxDb[i].db);
12831 find_home_dir(1);
12832 output_reset(&data);
12833 data.doXdgOpen = 0;
12834 clearTempFile(&data);
12835 #if !SQLITE_SHELL_IS_UTF8
12836 for(i=0; i<argcToFree; i++) free(argvToFree[i]);
12837 free(argvToFree);
12838 #endif
12839 free(data.colWidth);
12840 free(data.zNonce);
12841 /* Clear the global data structure so that valgrind will detect memory
12842 ** leaks */
12843 memset(&data, 0, sizeof(data));
12844 #ifdef SQLITE_DEBUG
12845 if( sqlite3_memory_used()>mem_main_enter ){
12846 eputf("Memory leaked: %u bytes\n",
12847 (unsigned int)(sqlite3_memory_used()-mem_main_enter));
12849 #endif
12850 #endif /* !SQLITE_SHELL_FIDDLE */
12851 return rc;
12855 #ifdef SQLITE_SHELL_FIDDLE
12856 /* Only for emcc experimentation purposes. */
12857 int fiddle_experiment(int a,int b){
12858 return a + b;
12862 ** Returns a pointer to the current DB handle.
12864 sqlite3 * fiddle_db_handle(){
12865 return globalDb;
12869 ** Returns a pointer to the given DB name's VFS. If zDbName is 0 then
12870 ** "main" is assumed. Returns 0 if no db with the given name is
12871 ** open.
12873 sqlite3_vfs * fiddle_db_vfs(const char *zDbName){
12874 sqlite3_vfs * pVfs = 0;
12875 if(globalDb){
12876 sqlite3_file_control(globalDb, zDbName ? zDbName : "main",
12877 SQLITE_FCNTL_VFS_POINTER, &pVfs);
12879 return pVfs;
12882 /* Only for emcc experimentation purposes. */
12883 sqlite3 * fiddle_db_arg(sqlite3 *arg){
12884 oputf("fiddle_db_arg(%p)\n", (const void*)arg);
12885 return arg;
12889 ** Intended to be called via a SharedWorker() while a separate
12890 ** SharedWorker() (which manages the wasm module) is performing work
12891 ** which should be interrupted. Unfortunately, SharedWorker is not
12892 ** portable enough to make real use of.
12894 void fiddle_interrupt(void){
12895 if( globalDb ) sqlite3_interrupt(globalDb);
12899 ** Returns the filename of the given db name, assuming "main" if
12900 ** zDbName is NULL. Returns NULL if globalDb is not opened.
12902 const char * fiddle_db_filename(const char * zDbName){
12903 return globalDb
12904 ? sqlite3_db_filename(globalDb, zDbName ? zDbName : "main")
12905 : NULL;
12909 ** Completely wipes out the contents of the currently-opened database
12910 ** but leaves its storage intact for reuse. If any transactions are
12911 ** active, they are forcibly rolled back.
12913 void fiddle_reset_db(void){
12914 if( globalDb ){
12915 int rc;
12916 while( sqlite3_txn_state(globalDb,0)>0 ){
12918 ** Resolve problem reported in
12919 ** https://sqlite.org/forum/forumpost/0b41a25d65
12921 oputz("Rolling back in-progress transaction.\n");
12922 sqlite3_exec(globalDb,"ROLLBACK", 0, 0, 0);
12924 rc = sqlite3_db_config(globalDb, SQLITE_DBCONFIG_RESET_DATABASE, 1, 0);
12925 if( 0==rc ) sqlite3_exec(globalDb, "VACUUM", 0, 0, 0);
12926 sqlite3_db_config(globalDb, SQLITE_DBCONFIG_RESET_DATABASE, 0, 0);
12931 ** Uses the current database's VFS xRead to stream the db file's
12932 ** contents out to the given callback. The callback gets a single
12933 ** chunk of size n (its 2nd argument) on each call and must return 0
12934 ** on success, non-0 on error. This function returns 0 on success,
12935 ** SQLITE_NOTFOUND if no db is open, or propagates any other non-0
12936 ** code from the callback. Note that this is not thread-friendly: it
12937 ** expects that it will be the only thread reading the db file and
12938 ** takes no measures to ensure that is the case.
12940 int fiddle_export_db( int (*xCallback)(unsigned const char *zOut, int n) ){
12941 sqlite3_int64 nSize = 0;
12942 sqlite3_int64 nPos = 0;
12943 sqlite3_file * pFile = 0;
12944 unsigned char buf[1024 * 8];
12945 int nBuf = (int)sizeof(buf);
12946 int rc = shellState.db
12947 ? sqlite3_file_control(shellState.db, "main",
12948 SQLITE_FCNTL_FILE_POINTER, &pFile)
12949 : SQLITE_NOTFOUND;
12950 if( rc ) return rc;
12951 rc = pFile->pMethods->xFileSize(pFile, &nSize);
12952 if( rc ) return rc;
12953 if(nSize % nBuf){
12954 /* DB size is not an even multiple of the buffer size. Reduce
12955 ** buffer size so that we do not unduly inflate the db size when
12956 ** exporting. */
12957 if(0 == nSize % 4096) nBuf = 4096;
12958 else if(0 == nSize % 2048) nBuf = 2048;
12959 else if(0 == nSize % 1024) nBuf = 1024;
12960 else nBuf = 512;
12962 for( ; 0==rc && nPos<nSize; nPos += nBuf ){
12963 rc = pFile->pMethods->xRead(pFile, buf, nBuf, nPos);
12964 if(SQLITE_IOERR_SHORT_READ == rc){
12965 rc = (nPos + nBuf) < nSize ? rc : 0/*assume EOF*/;
12967 if( 0==rc ) rc = xCallback(buf, nBuf);
12969 return rc;
12973 ** Trivial exportable function for emscripten. It processes zSql as if
12974 ** it were input to the sqlite3 shell and redirects all output to the
12975 ** wasm binding. fiddle_main() must have been called before this
12976 ** is called, or results are undefined.
12978 void fiddle_exec(const char * zSql){
12979 if(zSql && *zSql){
12980 if('.'==*zSql) puts(zSql);
12981 shellState.wasm.zInput = zSql;
12982 shellState.wasm.zPos = zSql;
12983 process_input(&shellState);
12984 shellState.wasm.zInput = shellState.wasm.zPos = 0;
12987 #endif /* SQLITE_SHELL_FIDDLE */