blob: 1f8681f63950d2e0141c0f894293aadc0fd96f07 [file] [log] [blame]
drh75897232000-05-29 14:26:00 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drh75897232000-05-29 14:26:00 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drh75897232000-05-29 14:26:00 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** 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.
drh75897232000-05-29 14:26:00 +000010**
11*************************************************************************
12** This file contains code to implement the "sqlite" command line
13** utility for accessing SQLite databases.
drh75897232000-05-29 14:26:00 +000014*/
mistachkina3b2ff52011-09-16 20:16:36 +000015#if (defined(_WIN32) || defined(WIN32)) && !defined(_CRT_SECURE_NO_WARNINGS)
shane18e526c2008-12-10 22:30:24 +000016/* This needs to come before any includes for MSVC compiler */
17#define _CRT_SECURE_NO_WARNINGS
18#endif
19
drh36f7dd32011-10-13 16:02:17 +000020/*
21** Enable large-file support for fopen() and friends on unix.
22*/
23#ifndef SQLITE_DISABLE_LFS
24# define _LARGE_FILE 1
25# ifndef _FILE_OFFSET_BITS
26# define _FILE_OFFSET_BITS 64
27# endif
28# define _LARGEFILE_SOURCE 1
29#endif
30
drh75897232000-05-29 14:26:00 +000031#include <stdlib.h>
32#include <string.h>
33#include <stdio.h>
danielk19772a02e332004-06-05 08:04:36 +000034#include <assert.h>
drh1d482dd2004-05-31 18:23:07 +000035#include "sqlite3.h"
drh75897232000-05-29 14:26:00 +000036#include <ctype.h>
drhb0603412007-02-28 04:47:26 +000037#include <stdarg.h>
persicom7e2dfdd2002-04-18 02:46:52 +000038
drh83905c92012-06-21 13:00:37 +000039#if !defined(_WIN32) && !defined(WIN32)
drh4c504392000-10-16 22:06:40 +000040# include <signal.h>
chw97185482008-11-17 08:05:31 +000041# if !defined(__RTP__) && !defined(_WRS_KERNEL)
42# include <pwd.h>
43# endif
drhdd45df82002-04-18 12:39:03 +000044# include <unistd.h>
45# include <sys/types.h>
drh4c504392000-10-16 22:06:40 +000046#endif
drh75897232000-05-29 14:26:00 +000047
drhaaa21b42014-02-11 14:37:51 +000048#if defined(HAVE_READLINE) && HAVE_READLINE!=0
drh8e7e7a22000-05-30 18:45:23 +000049# include <readline/readline.h>
50# include <readline/history.h>
drhaaa21b42014-02-11 14:37:51 +000051#else
52# undef HAVE_READLINE
drh81d7fd12010-12-08 00:02:26 +000053#endif
drhaaa21b42014-02-11 14:37:51 +000054#if defined(HAVE_EDITLINE) && !defined(HAVE_READLINE)
55# define HAVE_READLINE 1
56# include <editline/readline.h>
57#endif
58#if !defined(HAVE_READLINE)
persicom1d0b8722002-04-18 02:53:04 +000059# define add_history(X)
drh67505e72002-04-19 12:34:06 +000060# define read_history(X)
61# define write_history(X)
62# define stifle_history(X)
drh75897232000-05-29 14:26:00 +000063#endif
64
adamd2e8464a2006-09-06 21:39:40 +000065#if defined(_WIN32) || defined(WIN32)
66# include <io.h>
shane18e526c2008-12-10 22:30:24 +000067#define isatty(h) _isatty(h)
drh07901eb2014-02-28 19:37:45 +000068#ifndef access
69# define access(f,m) _access((f),(m))
70#endif
drh67ceaa62012-08-27 21:19:03 +000071#undef popen
drh53371f92013-07-25 17:07:03 +000072#define popen _popen
drh67ceaa62012-08-27 21:19:03 +000073#undef pclose
drh12cd6cf2013-06-29 15:40:22 +000074#define pclose _pclose
adamd2e8464a2006-09-06 21:39:40 +000075#else
drh4328c8b2003-04-26 02:50:11 +000076/* Make sure isatty() has a prototype.
77*/
drhb2acc3b2011-10-13 16:36:29 +000078extern int isatty(int);
drh4328c8b2003-04-26 02:50:11 +000079
drh53371f92013-07-25 17:07:03 +000080/* popen and pclose are not C89 functions and so are sometimes omitted from
81** the <stdio.h> header */
mistachkinf6418892013-08-28 01:54:12 +000082extern FILE *popen(const char*,const char*);
83extern int pclose(FILE*);
84#endif
drh53371f92013-07-25 17:07:03 +000085
chw65d3c132007-11-12 21:09:10 +000086#if defined(_WIN32_WCE)
87/* Windows CE (arm-wince-mingw32ce-gcc) does not provide isatty()
88 * thus we always assume that we have a console. That can be
89 * overridden with the -batch command line option.
90 */
91#define isatty(x) 1
92#endif
93
drhf0693c82011-10-11 20:41:54 +000094/* ctype macros that work with signed characters */
95#define IsSpace(X) isspace((unsigned char)X)
96#define IsDigit(X) isdigit((unsigned char)X)
97#define ToLower(X) (char)tolower((unsigned char)X)
98
drh43408312013-10-30 12:43:36 +000099
100/* True if the timer is enabled */
101static int enableTimer = 0;
102
103/* Return the current wall-clock time */
104static sqlite3_int64 timeOfDay(void){
105 static sqlite3_vfs *clockVfs = 0;
106 sqlite3_int64 t;
107 if( clockVfs==0 ) clockVfs = sqlite3_vfs_find(0);
108 if( clockVfs->iVersion>=1 && clockVfs->xCurrentTimeInt64!=0 ){
109 clockVfs->xCurrentTimeInt64(clockVfs, &t);
110 }else{
111 double r;
112 clockVfs->xCurrentTime(clockVfs, &r);
113 t = (sqlite3_int64)(r*86400000.0);
114 }
115 return t;
116}
117
drhd5d0f642013-02-20 00:54:21 +0000118#if !defined(_WIN32) && !defined(WIN32) && !defined(_WRS_KERNEL) \
119 && !defined(__minux)
drh3b1a9882007-11-02 12:53:03 +0000120#include <sys/time.h>
121#include <sys/resource.h>
122
drhda108222009-02-25 19:07:24 +0000123/* Saved resource information for the beginning of an operation */
drh43408312013-10-30 12:43:36 +0000124static struct rusage sBegin; /* CPU time at start */
125static sqlite3_int64 iBegin; /* Wall-clock time at start */
drhda108222009-02-25 19:07:24 +0000126
drhda108222009-02-25 19:07:24 +0000127/*
128** Begin timing an operation
129*/
130static void beginTimer(void){
131 if( enableTimer ){
132 getrusage(RUSAGE_SELF, &sBegin);
drh43408312013-10-30 12:43:36 +0000133 iBegin = timeOfDay();
drhda108222009-02-25 19:07:24 +0000134 }
135}
136
137/* Return the difference of two time_structs in seconds */
138static double timeDiff(struct timeval *pStart, struct timeval *pEnd){
139 return (pEnd->tv_usec - pStart->tv_usec)*0.000001 +
140 (double)(pEnd->tv_sec - pStart->tv_sec);
141}
142
143/*
144** Print the timing results.
145*/
146static void endTimer(void){
147 if( enableTimer ){
148 struct rusage sEnd;
drh43408312013-10-30 12:43:36 +0000149 sqlite3_int64 iEnd = timeOfDay();
drhda108222009-02-25 19:07:24 +0000150 getrusage(RUSAGE_SELF, &sEnd);
drh43408312013-10-30 12:43:36 +0000151 printf("Run Time: real %.3f user %f sys %f\n",
152 (iEnd - iBegin)*0.001,
drhda108222009-02-25 19:07:24 +0000153 timeDiff(&sBegin.ru_utime, &sEnd.ru_utime),
154 timeDiff(&sBegin.ru_stime, &sEnd.ru_stime));
155 }
156}
shaneb320ccd2009-10-21 03:42:58 +0000157
drhda108222009-02-25 19:07:24 +0000158#define BEGIN_TIMER beginTimer()
159#define END_TIMER endTimer()
160#define HAS_TIMER 1
shaneb320ccd2009-10-21 03:42:58 +0000161
162#elif (defined(_WIN32) || defined(WIN32))
163
164#include <windows.h>
165
166/* Saved resource information for the beginning of an operation */
167static HANDLE hProcess;
168static FILETIME ftKernelBegin;
169static FILETIME ftUserBegin;
drh43408312013-10-30 12:43:36 +0000170static sqlite3_int64 ftWallBegin;
shaneb320ccd2009-10-21 03:42:58 +0000171typedef BOOL (WINAPI *GETPROCTIMES)(HANDLE, LPFILETIME, LPFILETIME, LPFILETIME, LPFILETIME);
172static GETPROCTIMES getProcessTimesAddr = NULL;
173
shaneb320ccd2009-10-21 03:42:58 +0000174/*
175** Check to see if we have timer support. Return 1 if necessary
176** support found (or found previously).
177*/
178static int hasTimer(void){
179 if( getProcessTimesAddr ){
180 return 1;
181 } else {
182 /* GetProcessTimes() isn't supported in WIN95 and some other Windows versions.
183 ** See if the version we are running on has it, and if it does, save off
184 ** a pointer to it and the current process handle.
185 */
186 hProcess = GetCurrentProcess();
187 if( hProcess ){
188 HINSTANCE hinstLib = LoadLibrary(TEXT("Kernel32.dll"));
189 if( NULL != hinstLib ){
190 getProcessTimesAddr = (GETPROCTIMES) GetProcAddress(hinstLib, "GetProcessTimes");
191 if( NULL != getProcessTimesAddr ){
192 return 1;
193 }
194 FreeLibrary(hinstLib);
195 }
196 }
197 }
198 return 0;
199}
200
201/*
202** Begin timing an operation
203*/
204static void beginTimer(void){
205 if( enableTimer && getProcessTimesAddr ){
206 FILETIME ftCreation, ftExit;
207 getProcessTimesAddr(hProcess, &ftCreation, &ftExit, &ftKernelBegin, &ftUserBegin);
drh43408312013-10-30 12:43:36 +0000208 ftWallBegin = timeOfDay();
shaneb320ccd2009-10-21 03:42:58 +0000209 }
210}
211
212/* Return the difference of two FILETIME structs in seconds */
213static double timeDiff(FILETIME *pStart, FILETIME *pEnd){
214 sqlite_int64 i64Start = *((sqlite_int64 *) pStart);
215 sqlite_int64 i64End = *((sqlite_int64 *) pEnd);
216 return (double) ((i64End - i64Start) / 10000000.0);
217}
218
219/*
220** Print the timing results.
221*/
222static void endTimer(void){
223 if( enableTimer && getProcessTimesAddr){
224 FILETIME ftCreation, ftExit, ftKernelEnd, ftUserEnd;
drh43408312013-10-30 12:43:36 +0000225 sqlite3_int64 ftWallEnd = timeOfDay();
shaneb320ccd2009-10-21 03:42:58 +0000226 getProcessTimesAddr(hProcess, &ftCreation, &ftExit, &ftKernelEnd, &ftUserEnd);
drh43408312013-10-30 12:43:36 +0000227 printf("Run Time: real %.3f user %f sys %f\n",
228 (ftWallEnd - ftWallBegin)*0.001,
shaneb320ccd2009-10-21 03:42:58 +0000229 timeDiff(&ftUserBegin, &ftUserEnd),
230 timeDiff(&ftKernelBegin, &ftKernelEnd));
231 }
232}
233
234#define BEGIN_TIMER beginTimer()
235#define END_TIMER endTimer()
236#define HAS_TIMER hasTimer()
237
drhda108222009-02-25 19:07:24 +0000238#else
239#define BEGIN_TIMER
240#define END_TIMER
241#define HAS_TIMER 0
242#endif
243
shanec0688ea2009-03-05 03:48:06 +0000244/*
245** Used to prevent warnings about unused parameters
246*/
247#define UNUSED_PARAMETER(x) (void)(x)
248
drhe91d16b2008-12-08 18:27:31 +0000249/*
drhc49f44e2006-10-26 18:15:42 +0000250** If the following flag is set, then command execution stops
251** at an error if we are not interactive.
252*/
253static int bail_on_error = 0;
254
255/*
drhc28490c2006-10-26 14:25:58 +0000256** Threat stdin as an interactive input if the following variable
257** is true. Otherwise, assume stdin is connected to a file or pipe.
258*/
259static int stdin_is_interactive = 1;
260
261/*
drh4c504392000-10-16 22:06:40 +0000262** The following is the open SQLite database. We make a pointer
263** to this database a static variable so that it can be accessed
264** by the SIGINT handler to interrupt database processing.
265*/
danielk197792f9a1b2004-06-19 09:08:16 +0000266static sqlite3 *db = 0;
drh4c504392000-10-16 22:06:40 +0000267
268/*
drh67505e72002-04-19 12:34:06 +0000269** True if an interrupt (Control-C) has been received.
270*/
drh43617e92006-03-06 20:55:46 +0000271static volatile int seenInterrupt = 0;
drh67505e72002-04-19 12:34:06 +0000272
273/*
persicom7e2dfdd2002-04-18 02:46:52 +0000274** This is the name of our program. It is set in main(), used
275** in a number of other places, mostly for error messages.
276*/
277static char *Argv0;
278
279/*
280** Prompt strings. Initialized in main. Settable with
281** .prompt main continue
282*/
283static char mainPrompt[20]; /* First line prompt. default: "sqlite> "*/
284static char continuePrompt[20]; /* Continuation prompt. default: " ...> " */
285
drhb0603412007-02-28 04:47:26 +0000286/*
287** Write I/O traces to the following stream.
288*/
rsebe0a9092007-07-30 18:24:38 +0000289#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +0000290static FILE *iotrace = 0;
rsebe0a9092007-07-30 18:24:38 +0000291#endif
drhb0603412007-02-28 04:47:26 +0000292
293/*
294** This routine works like printf in that its first argument is a
295** format string and subsequent arguments are values to be substituted
296** in place of % fields. The result of formatting this string
297** is written to iotrace.
298*/
rsebe0a9092007-07-30 18:24:38 +0000299#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +0000300static void iotracePrintf(const char *zFormat, ...){
301 va_list ap;
drhf075cd02007-02-28 06:14:25 +0000302 char *z;
drhb0603412007-02-28 04:47:26 +0000303 if( iotrace==0 ) return;
304 va_start(ap, zFormat);
drhf075cd02007-02-28 06:14:25 +0000305 z = sqlite3_vmprintf(zFormat, ap);
drhb0603412007-02-28 04:47:26 +0000306 va_end(ap);
drhf075cd02007-02-28 06:14:25 +0000307 fprintf(iotrace, "%s", z);
308 sqlite3_free(z);
drhb0603412007-02-28 04:47:26 +0000309}
rsebe0a9092007-07-30 18:24:38 +0000310#endif
drhb0603412007-02-28 04:47:26 +0000311
drh44c2eb12003-04-30 11:38:26 +0000312
persicom7e2dfdd2002-04-18 02:46:52 +0000313/*
drh83965662003-04-17 02:54:13 +0000314** Determines if a string is a number of not.
315*/
danielk19772e588c72005-12-09 14:25:08 +0000316static int isNumber(const char *z, int *realnum){
drhc8d74412004-08-31 23:41:26 +0000317 if( *z=='-' || *z=='+' ) z++;
drhf0693c82011-10-11 20:41:54 +0000318 if( !IsDigit(*z) ){
drhc8d74412004-08-31 23:41:26 +0000319 return 0;
320 }
321 z++;
322 if( realnum ) *realnum = 0;
drhf0693c82011-10-11 20:41:54 +0000323 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000324 if( *z=='.' ){
325 z++;
drhf0693c82011-10-11 20:41:54 +0000326 if( !IsDigit(*z) ) return 0;
327 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000328 if( realnum ) *realnum = 1;
329 }
330 if( *z=='e' || *z=='E' ){
331 z++;
332 if( *z=='+' || *z=='-' ) z++;
drhf0693c82011-10-11 20:41:54 +0000333 if( !IsDigit(*z) ) return 0;
334 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000335 if( realnum ) *realnum = 1;
336 }
337 return *z==0;
338}
drh83965662003-04-17 02:54:13 +0000339
340/*
danielk1977bc6ada42004-06-30 08:20:16 +0000341** A global char* and an SQL function to access its current value
342** from within an SQL statement. This program used to use the
343** sqlite_exec_printf() API to substitue a string into an SQL statement.
344** The correct way to do this with sqlite3 is to use the bind API, but
345** since the shell is built around the callback paradigm it would be a lot
346** of work. Instead just use this hack, which is quite harmless.
347*/
348static const char *zShellStatic = 0;
349static void shellstaticFunc(
350 sqlite3_context *context,
351 int argc,
352 sqlite3_value **argv
353){
354 assert( 0==argc );
355 assert( zShellStatic );
shaned87897d2009-01-30 05:40:27 +0000356 UNUSED_PARAMETER(argc);
drh902b9ee2008-12-05 17:17:07 +0000357 UNUSED_PARAMETER(argv);
danielk1977bc6ada42004-06-30 08:20:16 +0000358 sqlite3_result_text(context, zShellStatic, -1, SQLITE_STATIC);
359}
360
361
362/*
drhfeac5f82004-08-01 00:10:45 +0000363** This routine reads a line of text from FILE in, stores
drh8e7e7a22000-05-30 18:45:23 +0000364** the text in memory obtained from malloc() and returns a pointer
365** to the text. NULL is returned at end of file, or if malloc()
366** fails.
367**
drh9f099fd2013-08-06 14:01:46 +0000368** If zLine is not NULL then it is a malloced buffer returned from
369** a previous call to this routine that may be reused.
drh8e7e7a22000-05-30 18:45:23 +0000370*/
drh9f099fd2013-08-06 14:01:46 +0000371static char *local_getline(char *zLine, FILE *in){
372 int nLine = zLine==0 ? 0 : 100;
373 int n = 0;
drh8e7e7a22000-05-30 18:45:23 +0000374
drhb07028f2011-10-14 21:49:18 +0000375 while( 1 ){
drh8e7e7a22000-05-30 18:45:23 +0000376 if( n+100>nLine ){
377 nLine = nLine*2 + 100;
378 zLine = realloc(zLine, nLine);
379 if( zLine==0 ) return 0;
380 }
drhdaffd0e2001-04-11 14:28:42 +0000381 if( fgets(&zLine[n], nLine - n, in)==0 ){
drh8e7e7a22000-05-30 18:45:23 +0000382 if( n==0 ){
383 free(zLine);
384 return 0;
385 }
386 zLine[n] = 0;
drh8e7e7a22000-05-30 18:45:23 +0000387 break;
388 }
drh9f099fd2013-08-06 14:01:46 +0000389 while( zLine[n] ) n++;
390 if( n>0 && zLine[n-1]=='\n' ){
drh8e7e7a22000-05-30 18:45:23 +0000391 n--;
shaneh13b36022009-12-17 21:07:15 +0000392 if( n>0 && zLine[n-1]=='\r' ) n--;
drh8e7e7a22000-05-30 18:45:23 +0000393 zLine[n] = 0;
drhb07028f2011-10-14 21:49:18 +0000394 break;
drh8e7e7a22000-05-30 18:45:23 +0000395 }
396 }
drh8e7e7a22000-05-30 18:45:23 +0000397 return zLine;
398}
399
400/*
drhc28490c2006-10-26 14:25:58 +0000401** Retrieve a single line of input text.
drh8e7e7a22000-05-30 18:45:23 +0000402**
drh9f099fd2013-08-06 14:01:46 +0000403** If in==0 then read from standard input and prompt before each line.
404** If isContinuation is true, then a continuation prompt is appropriate.
405** If isContinuation is zero, then the main prompt should be used.
406**
407** If zPrior is not NULL then it is a buffer from a prior call to this
408** routine that can be reused.
409**
410** The result is stored in space obtained from malloc() and must either
411** be freed by the caller or else passed back into this routine via the
412** zPrior argument for reuse.
drh8e7e7a22000-05-30 18:45:23 +0000413*/
drh9f099fd2013-08-06 14:01:46 +0000414static char *one_input_line(FILE *in, char *zPrior, int isContinuation){
drh8e7e7a22000-05-30 18:45:23 +0000415 char *zPrompt;
416 char *zResult;
drhdaffd0e2001-04-11 14:28:42 +0000417 if( in!=0 ){
drh9f099fd2013-08-06 14:01:46 +0000418 zResult = local_getline(zPrior, in);
drh8e7e7a22000-05-30 18:45:23 +0000419 }else{
drh9f099fd2013-08-06 14:01:46 +0000420 zPrompt = isContinuation ? continuePrompt : mainPrompt;
drhaaa21b42014-02-11 14:37:51 +0000421#if defined(HAVE_READLINE)
drh9f099fd2013-08-06 14:01:46 +0000422 free(zPrior);
423 zResult = readline(zPrompt);
424 if( zResult && *zResult ) add_history(zResult);
425#else
426 printf("%s", zPrompt);
427 fflush(stdout);
428 zResult = local_getline(zPrior, stdin);
danielk19774af00c62005-01-23 23:43:21 +0000429#endif
drh9f099fd2013-08-06 14:01:46 +0000430 }
drh8e7e7a22000-05-30 18:45:23 +0000431 return zResult;
432}
433
persicom7e2dfdd2002-04-18 02:46:52 +0000434struct previous_mode_data {
435 int valid; /* Is there legit data in here? */
436 int mode;
437 int showHeader;
438 int colWidth[100];
439};
drh45e29d82006-11-20 16:21:10 +0000440
drh8e7e7a22000-05-30 18:45:23 +0000441/*
drh75897232000-05-29 14:26:00 +0000442** An pointer to an instance of this structure is passed from
443** the main program to the callback. This is used to communicate
444** state and mode information.
445*/
446struct callback_data {
shane626a6e42009-10-22 17:30:15 +0000447 sqlite3 *db; /* The database */
drhdaffd0e2001-04-11 14:28:42 +0000448 int echoOn; /* True to echo input commands */
drhc2ce0be2014-05-29 12:36:14 +0000449 int autoEQP; /* Run EXPLAIN QUERY PLAN prior to seach SQL stmt */
shaneh642d8b82010-07-28 16:05:34 +0000450 int statsOn; /* True to display memory stats before each finalize */
drhc2ce0be2014-05-29 12:36:14 +0000451 int outCount; /* Revert to stdout when reaching zero */
drh28bd4bc2000-06-15 15:57:22 +0000452 int cnt; /* Number of records displayed so far */
453 FILE *out; /* Write results here */
drh42f64e52012-04-04 16:56:23 +0000454 FILE *traceOut; /* Output for sqlite3_trace() */
drh2f464a02011-10-13 00:41:49 +0000455 int nErr; /* Number of errors seen */
drh28bd4bc2000-06-15 15:57:22 +0000456 int mode; /* An output mode setting */
drh45e29d82006-11-20 16:21:10 +0000457 int writableSchema; /* True if PRAGMA writable_schema=ON */
drh28bd4bc2000-06-15 15:57:22 +0000458 int showHeader; /* True to show column names in List or Column mode */
drh33048c02001-10-01 14:29:22 +0000459 char *zDestTable; /* Name of destination table when MODE_Insert */
drh28bd4bc2000-06-15 15:57:22 +0000460 char separator[20]; /* Separator character for MODE_List */
drha0c66f52000-07-29 13:20:21 +0000461 int colWidth[100]; /* Requested width of each column when in column mode*/
462 int actualWidth[100]; /* Actual width of each column */
drh83965662003-04-17 02:54:13 +0000463 char nullvalue[20]; /* The text to print when a NULL comes back from
464 ** the database */
persicom7e2dfdd2002-04-18 02:46:52 +0000465 struct previous_mode_data explainPrev;
drh83965662003-04-17 02:54:13 +0000466 /* Holds the mode information just before
467 ** .explain ON */
drh44c2eb12003-04-30 11:38:26 +0000468 char outfile[FILENAME_MAX]; /* Filename for *out */
469 const char *zDbFilename; /* name of the database file */
drh05782482013-10-24 15:20:20 +0000470 char *zFreeOnClose; /* Filename to free when closing */
drha7e61d82011-03-12 17:02:57 +0000471 const char *zVfs; /* Name of VFS to use */
shane626a6e42009-10-22 17:30:15 +0000472 sqlite3_stmt *pStmt; /* Current statement if any. */
drh127f9d72010-02-23 01:47:00 +0000473 FILE *pLog; /* Write log output here */
dana98bf362013-11-13 18:35:01 +0000474 int *aiIndent; /* Array of indents used in MODE_Explain */
475 int nIndent; /* Size of array aiIndent[] */
danc4650bb2013-11-18 08:41:06 +0000476 int iIndent; /* Index of current op in aiIndent[] */
drh75897232000-05-29 14:26:00 +0000477};
478
479/*
480** These are the allowed modes.
481*/
drh967e8b72000-06-21 13:59:10 +0000482#define MODE_Line 0 /* One column per line. Blank line between records */
drh75897232000-05-29 14:26:00 +0000483#define MODE_Column 1 /* One record per line in neat columns */
484#define MODE_List 2 /* One record per line with a separator */
drhe3710332000-09-29 13:30:53 +0000485#define MODE_Semi 3 /* Same as MODE_List but append ";" to each line */
486#define MODE_Html 4 /* Generate an XHTML table */
487#define MODE_Insert 5 /* Generate SQL "insert" statements */
drhfeac5f82004-08-01 00:10:45 +0000488#define MODE_Tcl 6 /* Generate ANSI-C or TCL quoted elements */
drh8e64d1c2004-10-07 00:32:39 +0000489#define MODE_Csv 7 /* Quote strings, numbers are plain */
drh66ce4d02008-02-15 17:38:06 +0000490#define MODE_Explain 8 /* Like MODE_Column, but do not truncate data */
persicom7e2dfdd2002-04-18 02:46:52 +0000491
drh66ce4d02008-02-15 17:38:06 +0000492static const char *modeDescr[] = {
persicom7e2dfdd2002-04-18 02:46:52 +0000493 "line",
494 "column",
495 "list",
496 "semi",
497 "html",
drhfeac5f82004-08-01 00:10:45 +0000498 "insert",
499 "tcl",
drh8e64d1c2004-10-07 00:32:39 +0000500 "csv",
drh66ce4d02008-02-15 17:38:06 +0000501 "explain",
persicom7e2dfdd2002-04-18 02:46:52 +0000502};
drh75897232000-05-29 14:26:00 +0000503
504/*
505** Number of elements in an array
506*/
drh902b9ee2008-12-05 17:17:07 +0000507#define ArraySize(X) (int)(sizeof(X)/sizeof(X[0]))
drh75897232000-05-29 14:26:00 +0000508
509/*
drhea678832008-12-10 19:26:22 +0000510** Compute a string length that is limited to what can be stored in
511** lower 30 bits of a 32-bit signed integer.
512*/
drh4f21c4a2008-12-10 22:15:00 +0000513static int strlen30(const char *z){
drhea678832008-12-10 19:26:22 +0000514 const char *z2 = z;
515 while( *z2 ){ z2++; }
516 return 0x3fffffff & (int)(z2 - z);
517}
518
519/*
drh127f9d72010-02-23 01:47:00 +0000520** A callback for the sqlite3_log() interface.
521*/
522static void shellLog(void *pArg, int iErrCode, const char *zMsg){
523 struct callback_data *p = (struct callback_data*)pArg;
524 if( p->pLog==0 ) return;
525 fprintf(p->pLog, "(%d) %s\n", iErrCode, zMsg);
526 fflush(p->pLog);
527}
528
529/*
shane626a6e42009-10-22 17:30:15 +0000530** Output the given string as a hex-encoded blob (eg. X'1234' )
531*/
532static void output_hex_blob(FILE *out, const void *pBlob, int nBlob){
533 int i;
534 char *zBlob = (char *)pBlob;
535 fprintf(out,"X'");
drhb202d702012-04-24 12:12:57 +0000536 for(i=0; i<nBlob; i++){ fprintf(out,"%02x",zBlob[i]&0xff); }
shane626a6e42009-10-22 17:30:15 +0000537 fprintf(out,"'");
538}
539
540/*
drh28bd4bc2000-06-15 15:57:22 +0000541** Output the given string as a quoted string using SQL quoting conventions.
542*/
543static void output_quoted_string(FILE *out, const char *z){
544 int i;
545 int nSingle = 0;
drh28bd4bc2000-06-15 15:57:22 +0000546 for(i=0; z[i]; i++){
547 if( z[i]=='\'' ) nSingle++;
drh28bd4bc2000-06-15 15:57:22 +0000548 }
549 if( nSingle==0 ){
550 fprintf(out,"'%s'",z);
drh28bd4bc2000-06-15 15:57:22 +0000551 }else{
552 fprintf(out,"'");
553 while( *z ){
554 for(i=0; z[i] && z[i]!='\''; i++){}
555 if( i==0 ){
556 fprintf(out,"''");
557 z++;
558 }else if( z[i]=='\'' ){
559 fprintf(out,"%.*s''",i,z);
560 z += i+1;
561 }else{
drhcd7d2732002-02-26 23:24:26 +0000562 fprintf(out,"%s",z);
drh28bd4bc2000-06-15 15:57:22 +0000563 break;
564 }
565 }
drhcd7d2732002-02-26 23:24:26 +0000566 fprintf(out,"'");
drh28bd4bc2000-06-15 15:57:22 +0000567 }
568}
569
570/*
drhfeac5f82004-08-01 00:10:45 +0000571** Output the given string as a quoted according to C or TCL quoting rules.
572*/
573static void output_c_string(FILE *out, const char *z){
574 unsigned int c;
575 fputc('"', out);
576 while( (c = *(z++))!=0 ){
577 if( c=='\\' ){
578 fputc(c, out);
579 fputc(c, out);
mistachkin585dcb22012-12-04 00:23:43 +0000580 }else if( c=='"' ){
581 fputc('\\', out);
582 fputc('"', out);
drhfeac5f82004-08-01 00:10:45 +0000583 }else if( c=='\t' ){
584 fputc('\\', out);
585 fputc('t', out);
586 }else if( c=='\n' ){
587 fputc('\\', out);
588 fputc('n', out);
589 }else if( c=='\r' ){
590 fputc('\\', out);
591 fputc('r', out);
mistachkinf6418892013-08-28 01:54:12 +0000592 }else if( !isprint(c&0xff) ){
drh0a8640d2005-08-30 20:12:02 +0000593 fprintf(out, "\\%03o", c&0xff);
drhfeac5f82004-08-01 00:10:45 +0000594 }else{
595 fputc(c, out);
596 }
597 }
598 fputc('"', out);
599}
600
601/*
drhc08a4f12000-06-15 16:49:48 +0000602** Output the given string with characters that are special to
603** HTML escaped.
604*/
605static void output_html_string(FILE *out, const char *z){
606 int i;
drhc3d6ba42014-01-13 20:38:35 +0000607 if( z==0 ) z = "";
drhc08a4f12000-06-15 16:49:48 +0000608 while( *z ){
shane43d9cb22009-10-21 14:11:48 +0000609 for(i=0; z[i]
610 && z[i]!='<'
611 && z[i]!='&'
612 && z[i]!='>'
613 && z[i]!='\"'
614 && z[i]!='\'';
615 i++){}
drhc08a4f12000-06-15 16:49:48 +0000616 if( i>0 ){
617 fprintf(out,"%.*s",i,z);
618 }
619 if( z[i]=='<' ){
620 fprintf(out,"&lt;");
621 }else if( z[i]=='&' ){
622 fprintf(out,"&amp;");
shane43d9cb22009-10-21 14:11:48 +0000623 }else if( z[i]=='>' ){
624 fprintf(out,"&gt;");
625 }else if( z[i]=='\"' ){
626 fprintf(out,"&quot;");
627 }else if( z[i]=='\'' ){
628 fprintf(out,"&#39;");
drhc08a4f12000-06-15 16:49:48 +0000629 }else{
630 break;
631 }
632 z += i + 1;
633 }
634}
635
636/*
drhc49f44e2006-10-26 18:15:42 +0000637** If a field contains any character identified by a 1 in the following
638** array, then the string must be quoted for CSV.
639*/
640static const char needCsvQuote[] = {
641 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
642 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
643 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,
644 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
645 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
646 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
647 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
648 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
649 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
650 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
651 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
652 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
653 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
654 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
655 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
656 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
657};
658
659/*
drh8e64d1c2004-10-07 00:32:39 +0000660** Output a single term of CSV. Actually, p->separator is used for
661** the separator, which may or may not be a comma. p->nullvalue is
drh18f52e02012-01-16 16:56:31 +0000662** the null value. Strings are quoted if necessary.
drh8e64d1c2004-10-07 00:32:39 +0000663*/
664static void output_csv(struct callback_data *p, const char *z, int bSep){
drhc49f44e2006-10-26 18:15:42 +0000665 FILE *out = p->out;
drh8e64d1c2004-10-07 00:32:39 +0000666 if( z==0 ){
drhc49f44e2006-10-26 18:15:42 +0000667 fprintf(out,"%s",p->nullvalue);
drh8e64d1c2004-10-07 00:32:39 +0000668 }else{
drhc49f44e2006-10-26 18:15:42 +0000669 int i;
drh4f21c4a2008-12-10 22:15:00 +0000670 int nSep = strlen30(p->separator);
drhc49f44e2006-10-26 18:15:42 +0000671 for(i=0; z[i]; i++){
drhc85375d2007-12-18 15:41:44 +0000672 if( needCsvQuote[((unsigned char*)z)[i]]
673 || (z[i]==p->separator[0] &&
674 (nSep==1 || memcmp(z, p->separator, nSep)==0)) ){
drhc49f44e2006-10-26 18:15:42 +0000675 i = 0;
676 break;
677 }
678 }
679 if( i==0 ){
680 putc('"', out);
681 for(i=0; z[i]; i++){
682 if( z[i]=='"' ) putc('"', out);
683 putc(z[i], out);
684 }
685 putc('"', out);
686 }else{
687 fprintf(out, "%s", z);
688 }
drh8e64d1c2004-10-07 00:32:39 +0000689 }
690 if( bSep ){
drhd0e77882008-01-14 15:20:08 +0000691 fprintf(p->out, "%s", p->separator);
drh8e64d1c2004-10-07 00:32:39 +0000692 }
693}
694
danielk19774af00c62005-01-23 23:43:21 +0000695#ifdef SIGINT
drh8e64d1c2004-10-07 00:32:39 +0000696/*
drh4c504392000-10-16 22:06:40 +0000697** This routine runs when the user presses Ctrl-C
698*/
699static void interrupt_handler(int NotUsed){
drh902b9ee2008-12-05 17:17:07 +0000700 UNUSED_PARAMETER(NotUsed);
drh43ae8f62014-05-23 12:03:47 +0000701 seenInterrupt++;
702 if( seenInterrupt>2 ) exit(1);
danielk19776f8a5032004-05-10 10:34:51 +0000703 if( db ) sqlite3_interrupt(db);
drh4c504392000-10-16 22:06:40 +0000704}
danielk19774af00c62005-01-23 23:43:21 +0000705#endif
drh4c504392000-10-16 22:06:40 +0000706
707/*
shane626a6e42009-10-22 17:30:15 +0000708** This is the callback routine that the shell
drh75897232000-05-29 14:26:00 +0000709** invokes for each row of a query result.
710*/
shane626a6e42009-10-22 17:30:15 +0000711static int shell_callback(void *pArg, int nArg, char **azArg, char **azCol, int *aiType){
drh75897232000-05-29 14:26:00 +0000712 int i;
713 struct callback_data *p = (struct callback_data*)pArg;
shaneb9fc17d2009-10-22 21:23:35 +0000714
drh75897232000-05-29 14:26:00 +0000715 switch( p->mode ){
716 case MODE_Line: {
drhe3710332000-09-29 13:30:53 +0000717 int w = 5;
drh6a535342001-10-19 16:44:56 +0000718 if( azArg==0 ) break;
drhe3710332000-09-29 13:30:53 +0000719 for(i=0; i<nArg; i++){
drh4f21c4a2008-12-10 22:15:00 +0000720 int len = strlen30(azCol[i] ? azCol[i] : "");
drhe3710332000-09-29 13:30:53 +0000721 if( len>w ) w = len;
722 }
drh75897232000-05-29 14:26:00 +0000723 if( p->cnt++>0 ) fprintf(p->out,"\n");
724 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000725 fprintf(p->out,"%*s = %s\n", w, azCol[i],
drha69d9162003-04-17 22:57:53 +0000726 azArg[i] ? azArg[i] : p->nullvalue);
drh75897232000-05-29 14:26:00 +0000727 }
728 break;
729 }
danielk19770d78bae2008-01-03 07:09:48 +0000730 case MODE_Explain:
drh75897232000-05-29 14:26:00 +0000731 case MODE_Column: {
drha0c66f52000-07-29 13:20:21 +0000732 if( p->cnt++==0 ){
drh75897232000-05-29 14:26:00 +0000733 for(i=0; i<nArg; i++){
drha0c66f52000-07-29 13:20:21 +0000734 int w, n;
735 if( i<ArraySize(p->colWidth) ){
danielk19770d78bae2008-01-03 07:09:48 +0000736 w = p->colWidth[i];
drh75897232000-05-29 14:26:00 +0000737 }else{
danielk19770d78bae2008-01-03 07:09:48 +0000738 w = 0;
drh75897232000-05-29 14:26:00 +0000739 }
drh078b1fd2012-09-21 13:40:02 +0000740 if( w==0 ){
drh4f21c4a2008-12-10 22:15:00 +0000741 w = strlen30(azCol[i] ? azCol[i] : "");
drha0c66f52000-07-29 13:20:21 +0000742 if( w<10 ) w = 10;
drh4f21c4a2008-12-10 22:15:00 +0000743 n = strlen30(azArg && azArg[i] ? azArg[i] : p->nullvalue);
drha0c66f52000-07-29 13:20:21 +0000744 if( w<n ) w = n;
745 }
746 if( i<ArraySize(p->actualWidth) ){
persicom1d0b8722002-04-18 02:53:04 +0000747 p->actualWidth[i] = w;
drha0c66f52000-07-29 13:20:21 +0000748 }
749 if( p->showHeader ){
drh078b1fd2012-09-21 13:40:02 +0000750 if( w<0 ){
751 fprintf(p->out,"%*.*s%s",-w,-w,azCol[i], i==nArg-1 ? "\n": " ");
752 }else{
753 fprintf(p->out,"%-*.*s%s",w,w,azCol[i], i==nArg-1 ? "\n": " ");
754 }
drha0c66f52000-07-29 13:20:21 +0000755 }
756 }
757 if( p->showHeader ){
758 for(i=0; i<nArg; i++){
759 int w;
760 if( i<ArraySize(p->actualWidth) ){
761 w = p->actualWidth[i];
drh078b1fd2012-09-21 13:40:02 +0000762 if( w<0 ) w = -w;
drha0c66f52000-07-29 13:20:21 +0000763 }else{
764 w = 10;
765 }
766 fprintf(p->out,"%-*.*s%s",w,w,"-----------------------------------"
767 "----------------------------------------------------------",
768 i==nArg-1 ? "\n": " ");
769 }
drh75897232000-05-29 14:26:00 +0000770 }
771 }
drh6a535342001-10-19 16:44:56 +0000772 if( azArg==0 ) break;
drh75897232000-05-29 14:26:00 +0000773 for(i=0; i<nArg; i++){
774 int w;
drha0c66f52000-07-29 13:20:21 +0000775 if( i<ArraySize(p->actualWidth) ){
776 w = p->actualWidth[i];
drh75897232000-05-29 14:26:00 +0000777 }else{
778 w = 10;
779 }
dana98bf362013-11-13 18:35:01 +0000780 if( p->mode==MODE_Explain && azArg[i] && strlen30(azArg[i])>w ){
drh4f21c4a2008-12-10 22:15:00 +0000781 w = strlen30(azArg[i]);
danielk19770d78bae2008-01-03 07:09:48 +0000782 }
dana98bf362013-11-13 18:35:01 +0000783 if( i==1 && p->aiIndent && p->pStmt ){
danc4650bb2013-11-18 08:41:06 +0000784 if( p->iIndent<p->nIndent ){
785 fprintf(p->out, "%*.s", p->aiIndent[p->iIndent], "");
dana98bf362013-11-13 18:35:01 +0000786 }
danc4650bb2013-11-18 08:41:06 +0000787 p->iIndent++;
dana98bf362013-11-13 18:35:01 +0000788 }
drh078b1fd2012-09-21 13:40:02 +0000789 if( w<0 ){
790 fprintf(p->out,"%*.*s%s",-w,-w,
791 azArg[i] ? azArg[i] : p->nullvalue, i==nArg-1 ? "\n": " ");
792 }else{
793 fprintf(p->out,"%-*.*s%s",w,w,
794 azArg[i] ? azArg[i] : p->nullvalue, i==nArg-1 ? "\n": " ");
795 }
drh75897232000-05-29 14:26:00 +0000796 }
797 break;
798 }
drhe3710332000-09-29 13:30:53 +0000799 case MODE_Semi:
drh75897232000-05-29 14:26:00 +0000800 case MODE_List: {
801 if( p->cnt++==0 && p->showHeader ){
802 for(i=0; i<nArg; i++){
803 fprintf(p->out,"%s%s",azCol[i], i==nArg-1 ? "\n" : p->separator);
804 }
805 }
drh6a535342001-10-19 16:44:56 +0000806 if( azArg==0 ) break;
drh75897232000-05-29 14:26:00 +0000807 for(i=0; i<nArg; i++){
drh4c653a02000-06-07 01:27:47 +0000808 char *z = azArg[i];
persicom7e2dfdd2002-04-18 02:46:52 +0000809 if( z==0 ) z = p->nullvalue;
drh71172c52002-01-24 00:00:21 +0000810 fprintf(p->out, "%s", z);
drhe3710332000-09-29 13:30:53 +0000811 if( i<nArg-1 ){
812 fprintf(p->out, "%s", p->separator);
813 }else if( p->mode==MODE_Semi ){
814 fprintf(p->out, ";\n");
815 }else{
816 fprintf(p->out, "\n");
817 }
drh75897232000-05-29 14:26:00 +0000818 }
819 break;
820 }
drh1e5d0e92000-05-31 23:33:17 +0000821 case MODE_Html: {
822 if( p->cnt++==0 && p->showHeader ){
mihailim57c591a2008-06-23 21:26:05 +0000823 fprintf(p->out,"<TR>");
drh1e5d0e92000-05-31 23:33:17 +0000824 for(i=0; i<nArg; i++){
shane43d9cb22009-10-21 14:11:48 +0000825 fprintf(p->out,"<TH>");
826 output_html_string(p->out, azCol[i]);
827 fprintf(p->out,"</TH>\n");
drh1e5d0e92000-05-31 23:33:17 +0000828 }
mihailim57c591a2008-06-23 21:26:05 +0000829 fprintf(p->out,"</TR>\n");
drh1e5d0e92000-05-31 23:33:17 +0000830 }
drh6a535342001-10-19 16:44:56 +0000831 if( azArg==0 ) break;
mihailim57c591a2008-06-23 21:26:05 +0000832 fprintf(p->out,"<TR>");
drh1e5d0e92000-05-31 23:33:17 +0000833 for(i=0; i<nArg; i++){
mihailim57c591a2008-06-23 21:26:05 +0000834 fprintf(p->out,"<TD>");
persicom7e2dfdd2002-04-18 02:46:52 +0000835 output_html_string(p->out, azArg[i] ? azArg[i] : p->nullvalue);
mihailim57c591a2008-06-23 21:26:05 +0000836 fprintf(p->out,"</TD>\n");
drh1e5d0e92000-05-31 23:33:17 +0000837 }
mihailim57c591a2008-06-23 21:26:05 +0000838 fprintf(p->out,"</TR>\n");
drh1e5d0e92000-05-31 23:33:17 +0000839 break;
840 }
drhfeac5f82004-08-01 00:10:45 +0000841 case MODE_Tcl: {
842 if( p->cnt++==0 && p->showHeader ){
843 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000844 output_c_string(p->out,azCol[i] ? azCol[i] : "");
mistachkin585dcb22012-12-04 00:23:43 +0000845 if(i<nArg-1) fprintf(p->out, "%s", p->separator);
drhfeac5f82004-08-01 00:10:45 +0000846 }
847 fprintf(p->out,"\n");
848 }
849 if( azArg==0 ) break;
850 for(i=0; i<nArg; i++){
851 output_c_string(p->out, azArg[i] ? azArg[i] : p->nullvalue);
mistachkin585dcb22012-12-04 00:23:43 +0000852 if(i<nArg-1) fprintf(p->out, "%s", p->separator);
drhfeac5f82004-08-01 00:10:45 +0000853 }
854 fprintf(p->out,"\n");
855 break;
856 }
drh8e64d1c2004-10-07 00:32:39 +0000857 case MODE_Csv: {
858 if( p->cnt++==0 && p->showHeader ){
859 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000860 output_csv(p, azCol[i] ? azCol[i] : "", i<nArg-1);
drh8e64d1c2004-10-07 00:32:39 +0000861 }
862 fprintf(p->out,"\n");
863 }
864 if( azArg==0 ) break;
865 for(i=0; i<nArg; i++){
866 output_csv(p, azArg[i], i<nArg-1);
867 }
868 fprintf(p->out,"\n");
869 break;
870 }
drh28bd4bc2000-06-15 15:57:22 +0000871 case MODE_Insert: {
shaneb9fc17d2009-10-22 21:23:35 +0000872 p->cnt++;
drh6a535342001-10-19 16:44:56 +0000873 if( azArg==0 ) break;
drh33048c02001-10-01 14:29:22 +0000874 fprintf(p->out,"INSERT INTO %s VALUES(",p->zDestTable);
drh28bd4bc2000-06-15 15:57:22 +0000875 for(i=0; i<nArg; i++){
876 char *zSep = i>0 ? ",": "";
shanead6b8d02009-10-22 18:12:58 +0000877 if( (azArg[i]==0) || (aiType && aiType[i]==SQLITE_NULL) ){
drh28bd4bc2000-06-15 15:57:22 +0000878 fprintf(p->out,"%sNULL",zSep);
shanead6b8d02009-10-22 18:12:58 +0000879 }else if( aiType && aiType[i]==SQLITE_TEXT ){
880 if( zSep[0] ) fprintf(p->out,"%s",zSep);
881 output_quoted_string(p->out, azArg[i]);
drhc2ce0be2014-05-29 12:36:14 +0000882 }else if( aiType && (aiType[i]==SQLITE_INTEGER
883 || aiType[i]==SQLITE_FLOAT) ){
shanead6b8d02009-10-22 18:12:58 +0000884 fprintf(p->out,"%s%s",zSep, azArg[i]);
shane626a6e42009-10-22 17:30:15 +0000885 }else if( aiType && aiType[i]==SQLITE_BLOB && p->pStmt ){
886 const void *pBlob = sqlite3_column_blob(p->pStmt, i);
887 int nBlob = sqlite3_column_bytes(p->pStmt, i);
888 if( zSep[0] ) fprintf(p->out,"%s",zSep);
889 output_hex_blob(p->out, pBlob, nBlob);
drhc8d74412004-08-31 23:41:26 +0000890 }else if( isNumber(azArg[i], 0) ){
drh28bd4bc2000-06-15 15:57:22 +0000891 fprintf(p->out,"%s%s",zSep, azArg[i]);
892 }else{
893 if( zSep[0] ) fprintf(p->out,"%s",zSep);
894 output_quoted_string(p->out, azArg[i]);
895 }
896 }
897 fprintf(p->out,");\n");
drh6a535342001-10-19 16:44:56 +0000898 break;
drh28bd4bc2000-06-15 15:57:22 +0000899 }
persicom1d0b8722002-04-18 02:53:04 +0000900 }
drh75897232000-05-29 14:26:00 +0000901 return 0;
902}
903
904/*
shane626a6e42009-10-22 17:30:15 +0000905** This is the callback routine that the SQLite library
906** invokes for each row of a query result.
907*/
908static int callback(void *pArg, int nArg, char **azArg, char **azCol){
909 /* since we don't have type info, call the shell_callback with a NULL value */
910 return shell_callback(pArg, nArg, azArg, azCol, NULL);
911}
912
913/*
drh33048c02001-10-01 14:29:22 +0000914** Set the destination table field of the callback_data structure to
915** the name of the table given. Escape any quote characters in the
916** table name.
917*/
918static void set_table_name(struct callback_data *p, const char *zName){
919 int i, n;
920 int needQuote;
921 char *z;
922
923 if( p->zDestTable ){
924 free(p->zDestTable);
925 p->zDestTable = 0;
926 }
927 if( zName==0 ) return;
drh4c755c02004-08-08 20:22:17 +0000928 needQuote = !isalpha((unsigned char)*zName) && *zName!='_';
drh33048c02001-10-01 14:29:22 +0000929 for(i=n=0; zName[i]; i++, n++){
drh4c755c02004-08-08 20:22:17 +0000930 if( !isalnum((unsigned char)zName[i]) && zName[i]!='_' ){
drh33048c02001-10-01 14:29:22 +0000931 needQuote = 1;
932 if( zName[i]=='\'' ) n++;
933 }
934 }
935 if( needQuote ) n += 2;
936 z = p->zDestTable = malloc( n+1 );
937 if( z==0 ){
shane86f5bdb2009-10-24 02:00:07 +0000938 fprintf(stderr,"Error: out of memory\n");
drh33048c02001-10-01 14:29:22 +0000939 exit(1);
940 }
941 n = 0;
942 if( needQuote ) z[n++] = '\'';
943 for(i=0; zName[i]; i++){
944 z[n++] = zName[i];
945 if( zName[i]=='\'' ) z[n++] = '\'';
946 }
947 if( needQuote ) z[n++] = '\'';
948 z[n] = 0;
949}
950
danielk19772a02e332004-06-05 08:04:36 +0000951/* zIn is either a pointer to a NULL-terminated string in memory obtained
952** from malloc(), or a NULL pointer. The string pointed to by zAppend is
953** added to zIn, and the result returned in memory obtained from malloc().
954** zIn, if it was not NULL, is freed.
955**
956** If the third argument, quote, is not '\0', then it is used as a
957** quote character for zAppend.
958*/
drhc28490c2006-10-26 14:25:58 +0000959static char *appendText(char *zIn, char const *zAppend, char quote){
danielk19772a02e332004-06-05 08:04:36 +0000960 int len;
961 int i;
drh4f21c4a2008-12-10 22:15:00 +0000962 int nAppend = strlen30(zAppend);
963 int nIn = (zIn?strlen30(zIn):0);
danielk19772a02e332004-06-05 08:04:36 +0000964
965 len = nAppend+nIn+1;
966 if( quote ){
967 len += 2;
968 for(i=0; i<nAppend; i++){
969 if( zAppend[i]==quote ) len++;
970 }
971 }
972
973 zIn = (char *)realloc(zIn, len);
974 if( !zIn ){
975 return 0;
976 }
977
978 if( quote ){
979 char *zCsr = &zIn[nIn];
980 *zCsr++ = quote;
981 for(i=0; i<nAppend; i++){
982 *zCsr++ = zAppend[i];
983 if( zAppend[i]==quote ) *zCsr++ = quote;
984 }
985 *zCsr++ = quote;
986 *zCsr++ = '\0';
987 assert( (zCsr-zIn)==len );
988 }else{
989 memcpy(&zIn[nIn], zAppend, nAppend);
990 zIn[len-1] = '\0';
991 }
992
993 return zIn;
994}
995
drhdd3d4592004-08-30 01:54:05 +0000996
997/*
drhb21a8e42012-01-28 21:08:51 +0000998** Execute a query statement that will generate SQL output. Print
999** the result columns, comma-separated, on a line and then add a
1000** semicolon terminator to the end of that line.
drh45e29d82006-11-20 16:21:10 +00001001**
drhb21a8e42012-01-28 21:08:51 +00001002** If the number of columns is 1 and that column contains text "--"
1003** then write the semicolon on a separate line. That way, if a
1004** "--" comment occurs at the end of the statement, the comment
1005** won't consume the semicolon terminator.
drhdd3d4592004-08-30 01:54:05 +00001006*/
drh157e29a2009-05-21 15:15:00 +00001007static int run_table_dump_query(
drh2f464a02011-10-13 00:41:49 +00001008 struct callback_data *p, /* Query context */
1009 const char *zSelect, /* SELECT statement to extract content */
1010 const char *zFirstRow /* Print before first row, if not NULL */
drh157e29a2009-05-21 15:15:00 +00001011){
drhdd3d4592004-08-30 01:54:05 +00001012 sqlite3_stmt *pSelect;
1013 int rc;
drhb21a8e42012-01-28 21:08:51 +00001014 int nResult;
1015 int i;
1016 const char *z;
drhc7181902014-02-27 15:04:13 +00001017 rc = sqlite3_prepare_v2(p->db, zSelect, -1, &pSelect, 0);
drhdd3d4592004-08-30 01:54:05 +00001018 if( rc!=SQLITE_OK || !pSelect ){
drh2f464a02011-10-13 00:41:49 +00001019 fprintf(p->out, "/**** ERROR: (%d) %s *****/\n", rc, sqlite3_errmsg(p->db));
drh4384e982013-10-01 15:30:05 +00001020 if( (rc&0xff)!=SQLITE_CORRUPT ) p->nErr++;
drhdd3d4592004-08-30 01:54:05 +00001021 return rc;
1022 }
1023 rc = sqlite3_step(pSelect);
drhb21a8e42012-01-28 21:08:51 +00001024 nResult = sqlite3_column_count(pSelect);
drhdd3d4592004-08-30 01:54:05 +00001025 while( rc==SQLITE_ROW ){
drh157e29a2009-05-21 15:15:00 +00001026 if( zFirstRow ){
drh2f464a02011-10-13 00:41:49 +00001027 fprintf(p->out, "%s", zFirstRow);
drh157e29a2009-05-21 15:15:00 +00001028 zFirstRow = 0;
1029 }
drhb21a8e42012-01-28 21:08:51 +00001030 z = (const char*)sqlite3_column_text(pSelect, 0);
1031 fprintf(p->out, "%s", z);
1032 for(i=1; i<nResult; i++){
1033 fprintf(p->out, ",%s", sqlite3_column_text(pSelect, i));
1034 }
1035 if( z==0 ) z = "";
1036 while( z[0] && (z[0]!='-' || z[1]!='-') ) z++;
1037 if( z[0] ){
1038 fprintf(p->out, "\n;\n");
1039 }else{
1040 fprintf(p->out, ";\n");
1041 }
drhdd3d4592004-08-30 01:54:05 +00001042 rc = sqlite3_step(pSelect);
1043 }
drh2f464a02011-10-13 00:41:49 +00001044 rc = sqlite3_finalize(pSelect);
1045 if( rc!=SQLITE_OK ){
1046 fprintf(p->out, "/**** ERROR: (%d) %s *****/\n", rc, sqlite3_errmsg(p->db));
drh4384e982013-10-01 15:30:05 +00001047 if( (rc&0xff)!=SQLITE_CORRUPT ) p->nErr++;
drh2f464a02011-10-13 00:41:49 +00001048 }
1049 return rc;
drhdd3d4592004-08-30 01:54:05 +00001050}
1051
shane626a6e42009-10-22 17:30:15 +00001052/*
1053** Allocate space and save off current error string.
1054*/
1055static char *save_err_msg(
1056 sqlite3 *db /* Database to query */
1057){
1058 int nErrMsg = 1+strlen30(sqlite3_errmsg(db));
1059 char *zErrMsg = sqlite3_malloc(nErrMsg);
1060 if( zErrMsg ){
1061 memcpy(zErrMsg, sqlite3_errmsg(db), nErrMsg);
1062 }
1063 return zErrMsg;
1064}
1065
1066/*
shaneh642d8b82010-07-28 16:05:34 +00001067** Display memory stats.
1068*/
1069static int display_stats(
1070 sqlite3 *db, /* Database to query */
1071 struct callback_data *pArg, /* Pointer to struct callback_data */
1072 int bReset /* True to reset the stats */
1073){
1074 int iCur;
1075 int iHiwtr;
1076
1077 if( pArg && pArg->out ){
1078
1079 iHiwtr = iCur = -1;
1080 sqlite3_status(SQLITE_STATUS_MEMORY_USED, &iCur, &iHiwtr, bReset);
drh29dfbe32010-07-28 17:01:24 +00001081 fprintf(pArg->out, "Memory Used: %d (max %d) bytes\n", iCur, iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001082 iHiwtr = iCur = -1;
1083 sqlite3_status(SQLITE_STATUS_MALLOC_COUNT, &iCur, &iHiwtr, bReset);
drh2a58e9c2010-12-21 21:28:38 +00001084 fprintf(pArg->out, "Number of Outstanding Allocations: %d (max %d)\n", iCur, iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001085/*
1086** Not currently used by the CLI.
1087** iHiwtr = iCur = -1;
1088** sqlite3_status(SQLITE_STATUS_PAGECACHE_USED, &iCur, &iHiwtr, bReset);
1089** fprintf(pArg->out, "Number of Pcache Pages Used: %d (max %d) pages\n", iCur, iHiwtr);
1090*/
1091 iHiwtr = iCur = -1;
1092 sqlite3_status(SQLITE_STATUS_PAGECACHE_OVERFLOW, &iCur, &iHiwtr, bReset);
1093 fprintf(pArg->out, "Number of Pcache Overflow Bytes: %d (max %d) bytes\n", iCur, iHiwtr);
1094/*
1095** Not currently used by the CLI.
1096** iHiwtr = iCur = -1;
1097** sqlite3_status(SQLITE_STATUS_SCRATCH_USED, &iCur, &iHiwtr, bReset);
1098** fprintf(pArg->out, "Number of Scratch Allocations Used: %d (max %d)\n", iCur, iHiwtr);
1099*/
1100 iHiwtr = iCur = -1;
1101 sqlite3_status(SQLITE_STATUS_SCRATCH_OVERFLOW, &iCur, &iHiwtr, bReset);
1102 fprintf(pArg->out, "Number of Scratch Overflow Bytes: %d (max %d) bytes\n", iCur, iHiwtr);
1103 iHiwtr = iCur = -1;
1104 sqlite3_status(SQLITE_STATUS_MALLOC_SIZE, &iCur, &iHiwtr, bReset);
1105 fprintf(pArg->out, "Largest Allocation: %d bytes\n", iHiwtr);
1106 iHiwtr = iCur = -1;
1107 sqlite3_status(SQLITE_STATUS_PAGECACHE_SIZE, &iCur, &iHiwtr, bReset);
1108 fprintf(pArg->out, "Largest Pcache Allocation: %d bytes\n", iHiwtr);
1109 iHiwtr = iCur = -1;
1110 sqlite3_status(SQLITE_STATUS_SCRATCH_SIZE, &iCur, &iHiwtr, bReset);
1111 fprintf(pArg->out, "Largest Scratch Allocation: %d bytes\n", iHiwtr);
1112#ifdef YYTRACKMAXSTACKDEPTH
1113 iHiwtr = iCur = -1;
1114 sqlite3_status(SQLITE_STATUS_PARSER_STACK, &iCur, &iHiwtr, bReset);
1115 fprintf(pArg->out, "Deepest Parser Stack: %d (max %d)\n", iCur, iHiwtr);
1116#endif
1117 }
1118
1119 if( pArg && pArg->out && db ){
1120 iHiwtr = iCur = -1;
1121 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_USED, &iCur, &iHiwtr, bReset);
1122 fprintf(pArg->out, "Lookaside Slots Used: %d (max %d)\n", iCur, iHiwtr);
drh2a58e9c2010-12-21 21:28:38 +00001123 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_HIT, &iCur, &iHiwtr, bReset);
1124 fprintf(pArg->out, "Successful lookaside attempts: %d\n", iHiwtr);
1125 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_SIZE, &iCur, &iHiwtr, bReset);
1126 fprintf(pArg->out, "Lookaside failures due to size: %d\n", iHiwtr);
1127 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_FULL, &iCur, &iHiwtr, bReset);
1128 fprintf(pArg->out, "Lookaside failures due to OOM: %d\n", iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001129 iHiwtr = iCur = -1;
1130 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_USED, &iCur, &iHiwtr, bReset);
drhc78e6e42011-09-23 18:58:23 +00001131 fprintf(pArg->out, "Pager Heap Usage: %d bytes\n", iCur); iHiwtr = iCur = -1;
1132 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_HIT, &iCur, &iHiwtr, 1);
1133 fprintf(pArg->out, "Page cache hits: %d\n", iCur);
1134 iHiwtr = iCur = -1;
1135 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_MISS, &iCur, &iHiwtr, 1);
1136 fprintf(pArg->out, "Page cache misses: %d\n", iCur);
shaneh642d8b82010-07-28 16:05:34 +00001137 iHiwtr = iCur = -1;
drhfbbcd5d2012-03-24 20:09:33 +00001138 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_WRITE, &iCur, &iHiwtr, 1);
1139 fprintf(pArg->out, "Page cache writes: %d\n", iCur);
1140 iHiwtr = iCur = -1;
shaneh642d8b82010-07-28 16:05:34 +00001141 sqlite3_db_status(db, SQLITE_DBSTATUS_SCHEMA_USED, &iCur, &iHiwtr, bReset);
1142 fprintf(pArg->out, "Schema Heap Usage: %d bytes\n", iCur);
1143 iHiwtr = iCur = -1;
1144 sqlite3_db_status(db, SQLITE_DBSTATUS_STMT_USED, &iCur, &iHiwtr, bReset);
1145 fprintf(pArg->out, "Statement Heap/Lookaside Usage: %d bytes\n", iCur);
1146 }
1147
1148 if( pArg && pArg->out && db && pArg->pStmt ){
1149 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_FULLSCAN_STEP, bReset);
1150 fprintf(pArg->out, "Fullscan Steps: %d\n", iCur);
1151 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_SORT, bReset);
1152 fprintf(pArg->out, "Sort Operations: %d\n", iCur);
1153 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_AUTOINDEX, bReset);
1154 fprintf(pArg->out, "Autoindex Inserts: %d\n", iCur);
drhbf159fa2013-06-25 22:01:22 +00001155 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_VM_STEP, bReset);
1156 fprintf(pArg->out, "Virtual Machine Steps: %d\n", iCur);
shaneh642d8b82010-07-28 16:05:34 +00001157 }
1158
1159 return 0;
1160}
1161
1162/*
dana98bf362013-11-13 18:35:01 +00001163** Parameter azArray points to a zero-terminated array of strings. zStr
1164** points to a single nul-terminated string. Return non-zero if zStr
1165** is equal, according to strcmp(), to any of the strings in the array.
1166** Otherwise, return zero.
1167*/
1168static int str_in_array(const char *zStr, const char **azArray){
1169 int i;
1170 for(i=0; azArray[i]; i++){
1171 if( 0==strcmp(zStr, azArray[i]) ) return 1;
1172 }
1173 return 0;
1174}
1175
1176/*
1177** If compiled statement pSql appears to be an EXPLAIN statement, allocate
1178** and populate the callback_data.aiIndent[] array with the number of
1179** spaces each opcode should be indented before it is output.
1180**
1181** The indenting rules are:
1182**
1183** * For each "Next", "Prev", "VNext" or "VPrev" instruction, indent
1184** all opcodes that occur between the p2 jump destination and the opcode
1185** itself by 2 spaces.
1186**
drh01752bc2013-11-14 23:59:33 +00001187** * For each "Goto", if the jump destination is earlier in the program
1188** and ends on one of:
drhe73f0592014-01-21 22:25:45 +00001189** Yield SeekGt SeekLt RowSetRead Rewind
drhfe705102014-03-06 13:38:37 +00001190** or if the P1 parameter is one instead of zero,
drh01752bc2013-11-14 23:59:33 +00001191** then indent all opcodes between the earlier instruction
drhd2447442013-11-13 19:01:41 +00001192** and "Goto" by 2 spaces.
dana98bf362013-11-13 18:35:01 +00001193*/
1194static void explain_data_prepare(struct callback_data *p, sqlite3_stmt *pSql){
1195 const char *zSql; /* The text of the SQL statement */
1196 const char *z; /* Used to check if this is an EXPLAIN */
1197 int *abYield = 0; /* True if op is an OP_Yield */
1198 int nAlloc = 0; /* Allocated size of p->aiIndent[], abYield */
danc4650bb2013-11-18 08:41:06 +00001199 int iOp; /* Index of operation in p->aiIndent[] */
dana98bf362013-11-13 18:35:01 +00001200
drh8ad0de32014-03-20 18:45:27 +00001201 const char *azNext[] = { "Next", "Prev", "VPrev", "VNext", "SorterNext",
1202 "NextIfOpen", "PrevIfOpen", 0 };
drhb463fef2014-05-29 20:17:57 +00001203 const char *azYield[] = { "Yield", "SeekLT", "SeekGT", "RowSetRead", "Rewind", 0 };
dana98bf362013-11-13 18:35:01 +00001204 const char *azGoto[] = { "Goto", 0 };
1205
1206 /* Try to figure out if this is really an EXPLAIN statement. If this
1207 ** cannot be verified, return early. */
1208 zSql = sqlite3_sql(pSql);
1209 if( zSql==0 ) return;
1210 for(z=zSql; *z==' ' || *z=='\t' || *z=='\n' || *z=='\f' || *z=='\r'; z++);
1211 if( sqlite3_strnicmp(z, "explain", 7) ) return;
1212
1213 for(iOp=0; SQLITE_ROW==sqlite3_step(pSql); iOp++){
1214 int i;
danc4650bb2013-11-18 08:41:06 +00001215 int iAddr = sqlite3_column_int(pSql, 0);
dana98bf362013-11-13 18:35:01 +00001216 const char *zOp = (const char*)sqlite3_column_text(pSql, 1);
danc4650bb2013-11-18 08:41:06 +00001217
1218 /* Set p2 to the P2 field of the current opcode. Then, assuming that
1219 ** p2 is an instruction address, set variable p2op to the index of that
1220 ** instruction in the aiIndent[] array. p2 and p2op may be different if
1221 ** the current instruction is part of a sub-program generated by an
1222 ** SQL trigger or foreign key. */
dana98bf362013-11-13 18:35:01 +00001223 int p2 = sqlite3_column_int(pSql, 3);
danc4650bb2013-11-18 08:41:06 +00001224 int p2op = (p2 + (iOp-iAddr));
dana98bf362013-11-13 18:35:01 +00001225
1226 /* Grow the p->aiIndent array as required */
1227 if( iOp>=nAlloc ){
1228 nAlloc += 100;
1229 p->aiIndent = (int*)sqlite3_realloc(p->aiIndent, nAlloc*sizeof(int));
1230 abYield = (int*)sqlite3_realloc(abYield, nAlloc*sizeof(int));
1231 }
1232 abYield[iOp] = str_in_array(zOp, azYield);
1233 p->aiIndent[iOp] = 0;
1234 p->nIndent = iOp+1;
1235
1236 if( str_in_array(zOp, azNext) ){
danc4650bb2013-11-18 08:41:06 +00001237 for(i=p2op; i<iOp; i++) p->aiIndent[i] += 2;
dana98bf362013-11-13 18:35:01 +00001238 }
drhfe705102014-03-06 13:38:37 +00001239 if( str_in_array(zOp, azGoto) && p2op<p->nIndent
1240 && (abYield[p2op] || sqlite3_column_int(pSql, 2))
1241 ){
drhe73f0592014-01-21 22:25:45 +00001242 for(i=p2op+1; i<iOp; i++) p->aiIndent[i] += 2;
dana98bf362013-11-13 18:35:01 +00001243 }
1244 }
1245
danc4650bb2013-11-18 08:41:06 +00001246 p->iIndent = 0;
dana98bf362013-11-13 18:35:01 +00001247 sqlite3_free(abYield);
1248 sqlite3_reset(pSql);
1249}
1250
1251/*
1252** Free the array allocated by explain_data_prepare().
1253*/
1254static void explain_data_delete(struct callback_data *p){
1255 sqlite3_free(p->aiIndent);
1256 p->aiIndent = 0;
1257 p->nIndent = 0;
danc4650bb2013-11-18 08:41:06 +00001258 p->iIndent = 0;
dana98bf362013-11-13 18:35:01 +00001259}
1260
1261/*
shane626a6e42009-10-22 17:30:15 +00001262** Execute a statement or set of statements. Print
1263** any result rows/columns depending on the current mode
1264** set via the supplied callback.
1265**
1266** This is very similar to SQLite's built-in sqlite3_exec()
1267** function except it takes a slightly different callback
1268** and callback data argument.
1269*/
1270static int shell_exec(
1271 sqlite3 *db, /* An open database */
1272 const char *zSql, /* SQL to be evaluated */
1273 int (*xCallback)(void*,int,char**,char**,int*), /* Callback function */
1274 /* (not the same as sqlite3_exec) */
1275 struct callback_data *pArg, /* Pointer to struct callback_data */
1276 char **pzErrMsg /* Error msg written here */
1277){
dan4564ced2010-01-05 04:59:56 +00001278 sqlite3_stmt *pStmt = NULL; /* Statement to execute. */
1279 int rc = SQLITE_OK; /* Return Code */
drhb07028f2011-10-14 21:49:18 +00001280 int rc2;
dan4564ced2010-01-05 04:59:56 +00001281 const char *zLeftover; /* Tail of unprocessed SQL */
shane626a6e42009-10-22 17:30:15 +00001282
1283 if( pzErrMsg ){
1284 *pzErrMsg = NULL;
1285 }
1286
shaneb9fc17d2009-10-22 21:23:35 +00001287 while( zSql[0] && (SQLITE_OK == rc) ){
1288 rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, &zLeftover);
1289 if( SQLITE_OK != rc ){
shane626a6e42009-10-22 17:30:15 +00001290 if( pzErrMsg ){
1291 *pzErrMsg = save_err_msg(db);
1292 }
1293 }else{
shaneb9fc17d2009-10-22 21:23:35 +00001294 if( !pStmt ){
1295 /* this happens for a comment or white-space */
1296 zSql = zLeftover;
drhf0693c82011-10-11 20:41:54 +00001297 while( IsSpace(zSql[0]) ) zSql++;
shaneb9fc17d2009-10-22 21:23:35 +00001298 continue;
1299 }
shane626a6e42009-10-22 17:30:15 +00001300
shaneh642d8b82010-07-28 16:05:34 +00001301 /* save off the prepared statment handle and reset row count */
1302 if( pArg ){
1303 pArg->pStmt = pStmt;
1304 pArg->cnt = 0;
1305 }
1306
shanehb7977c52010-01-18 18:17:10 +00001307 /* echo the sql statement if echo on */
shaneh642d8b82010-07-28 16:05:34 +00001308 if( pArg && pArg->echoOn ){
drha8c62df2010-02-15 15:47:18 +00001309 const char *zStmtSql = sqlite3_sql(pStmt);
shaneh642d8b82010-07-28 16:05:34 +00001310 fprintf(pArg->out, "%s\n", zStmtSql ? zStmtSql : zSql);
drha8c62df2010-02-15 15:47:18 +00001311 }
shanehb7977c52010-01-18 18:17:10 +00001312
drhefbf3b12014-02-28 20:47:24 +00001313 /* Show the EXPLAIN QUERY PLAN if .eqp is on */
1314 if( pArg && pArg->autoEQP ){
1315 sqlite3_stmt *pExplain;
1316 char *zEQP = sqlite3_mprintf("EXPLAIN QUERY PLAN %s", sqlite3_sql(pStmt));
1317 rc = sqlite3_prepare_v2(db, zEQP, -1, &pExplain, 0);
1318 if( rc==SQLITE_OK ){
1319 while( sqlite3_step(pExplain)==SQLITE_ROW ){
1320 fprintf(pArg->out,"--EQP-- %d,", sqlite3_column_int(pExplain, 0));
1321 fprintf(pArg->out,"%d,", sqlite3_column_int(pExplain, 1));
1322 fprintf(pArg->out,"%d,", sqlite3_column_int(pExplain, 2));
1323 fprintf(pArg->out,"%s\n", sqlite3_column_text(pExplain, 3));
1324 }
1325 }
1326 sqlite3_finalize(pExplain);
1327 sqlite3_free(zEQP);
1328 }
1329
drh7e02e5e2011-12-06 19:44:51 +00001330 /* Output TESTCTRL_EXPLAIN text of requested */
1331 if( pArg && pArg->mode==MODE_Explain ){
1332 const char *zExplain = 0;
1333 sqlite3_test_control(SQLITE_TESTCTRL_EXPLAIN_STMT, pStmt, &zExplain);
1334 if( zExplain && zExplain[0] ){
1335 fprintf(pArg->out, "%s", zExplain);
1336 }
1337 }
1338
dana98bf362013-11-13 18:35:01 +00001339 /* If the shell is currently in ".explain" mode, gather the extra
1340 ** data required to add indents to the output.*/
drh0a305922013-11-21 23:37:02 +00001341 if( pArg && pArg->mode==MODE_Explain ){
dana98bf362013-11-13 18:35:01 +00001342 explain_data_prepare(pArg, pStmt);
1343 }
1344
shaneb9fc17d2009-10-22 21:23:35 +00001345 /* perform the first step. this will tell us if we
1346 ** have a result set or not and how wide it is.
1347 */
1348 rc = sqlite3_step(pStmt);
1349 /* if we have a result set... */
1350 if( SQLITE_ROW == rc ){
1351 /* if we have a callback... */
1352 if( xCallback ){
1353 /* allocate space for col name ptr, value ptr, and type */
1354 int nCol = sqlite3_column_count(pStmt);
1355 void *pData = sqlite3_malloc(3*nCol*sizeof(const char*) + 1);
1356 if( !pData ){
1357 rc = SQLITE_NOMEM;
1358 }else{
1359 char **azCols = (char **)pData; /* Names of result columns */
1360 char **azVals = &azCols[nCol]; /* Results */
1361 int *aiTypes = (int *)&azVals[nCol]; /* Result types */
drh55a1b302013-09-04 16:08:50 +00001362 int i, x;
shaneb9fc17d2009-10-22 21:23:35 +00001363 assert(sizeof(int) <= sizeof(char *));
1364 /* save off ptrs to column names */
1365 for(i=0; i<nCol; i++){
1366 azCols[i] = (char *)sqlite3_column_name(pStmt, i);
1367 }
shaneb9fc17d2009-10-22 21:23:35 +00001368 do{
1369 /* extract the data and data types */
1370 for(i=0; i<nCol; i++){
drh55a1b302013-09-04 16:08:50 +00001371 aiTypes[i] = x = sqlite3_column_type(pStmt, i);
drh3432daa2013-10-11 16:35:49 +00001372 if( x==SQLITE_BLOB && pArg && pArg->mode==MODE_Insert ){
drh55a1b302013-09-04 16:08:50 +00001373 azVals[i] = "";
1374 }else{
1375 azVals[i] = (char*)sqlite3_column_text(pStmt, i);
1376 }
shaneb9fc17d2009-10-22 21:23:35 +00001377 if( !azVals[i] && (aiTypes[i]!=SQLITE_NULL) ){
1378 rc = SQLITE_NOMEM;
1379 break; /* from for */
1380 }
1381 } /* end for */
1382
1383 /* if data and types extracted successfully... */
1384 if( SQLITE_ROW == rc ){
1385 /* call the supplied callback with the result row data */
1386 if( xCallback(pArg, nCol, azVals, azCols, aiTypes) ){
1387 rc = SQLITE_ABORT;
1388 }else{
1389 rc = sqlite3_step(pStmt);
1390 }
1391 }
1392 } while( SQLITE_ROW == rc );
1393 sqlite3_free(pData);
shaneb9fc17d2009-10-22 21:23:35 +00001394 }
1395 }else{
1396 do{
1397 rc = sqlite3_step(pStmt);
1398 } while( rc == SQLITE_ROW );
1399 }
1400 }
1401
dana98bf362013-11-13 18:35:01 +00001402 explain_data_delete(pArg);
1403
shaneh642d8b82010-07-28 16:05:34 +00001404 /* print usage stats if stats on */
1405 if( pArg && pArg->statsOn ){
1406 display_stats(db, pArg, 0);
1407 }
1408
dan4564ced2010-01-05 04:59:56 +00001409 /* Finalize the statement just executed. If this fails, save a
1410 ** copy of the error message. Otherwise, set zSql to point to the
1411 ** next statement to execute. */
drhb07028f2011-10-14 21:49:18 +00001412 rc2 = sqlite3_finalize(pStmt);
1413 if( rc!=SQLITE_NOMEM ) rc = rc2;
dan4564ced2010-01-05 04:59:56 +00001414 if( rc==SQLITE_OK ){
shaneb9fc17d2009-10-22 21:23:35 +00001415 zSql = zLeftover;
drhf0693c82011-10-11 20:41:54 +00001416 while( IsSpace(zSql[0]) ) zSql++;
dan4564ced2010-01-05 04:59:56 +00001417 }else if( pzErrMsg ){
1418 *pzErrMsg = save_err_msg(db);
shane626a6e42009-10-22 17:30:15 +00001419 }
shaneh642d8b82010-07-28 16:05:34 +00001420
1421 /* clear saved stmt handle */
1422 if( pArg ){
1423 pArg->pStmt = NULL;
1424 }
shane626a6e42009-10-22 17:30:15 +00001425 }
shaneb9fc17d2009-10-22 21:23:35 +00001426 } /* end while */
shane626a6e42009-10-22 17:30:15 +00001427
1428 return rc;
1429}
1430
drhdd3d4592004-08-30 01:54:05 +00001431
drh33048c02001-10-01 14:29:22 +00001432/*
drh4c653a02000-06-07 01:27:47 +00001433** This is a different callback routine used for dumping the database.
1434** Each row received by this callback consists of a table name,
1435** the table type ("index" or "table") and SQL to create the table.
1436** This routine should print text sufficient to recreate the table.
1437*/
1438static int dump_callback(void *pArg, int nArg, char **azArg, char **azCol){
danielk19772a02e332004-06-05 08:04:36 +00001439 int rc;
1440 const char *zTable;
1441 const char *zType;
1442 const char *zSql;
drh157e29a2009-05-21 15:15:00 +00001443 const char *zPrepStmt = 0;
drhdaffd0e2001-04-11 14:28:42 +00001444 struct callback_data *p = (struct callback_data *)pArg;
danielk19772a02e332004-06-05 08:04:36 +00001445
drh902b9ee2008-12-05 17:17:07 +00001446 UNUSED_PARAMETER(azCol);
drh4c653a02000-06-07 01:27:47 +00001447 if( nArg!=3 ) return 1;
danielk19772a02e332004-06-05 08:04:36 +00001448 zTable = azArg[0];
1449 zType = azArg[1];
1450 zSql = azArg[2];
1451
drh00b950d2005-09-11 02:03:03 +00001452 if( strcmp(zTable, "sqlite_sequence")==0 ){
drh157e29a2009-05-21 15:15:00 +00001453 zPrepStmt = "DELETE FROM sqlite_sequence;\n";
drh7ed10322013-08-07 16:04:27 +00001454 }else if( sqlite3_strglob("sqlite_stat?", zTable)==0 ){
drh00b950d2005-09-11 02:03:03 +00001455 fprintf(p->out, "ANALYZE sqlite_master;\n");
1456 }else if( strncmp(zTable, "sqlite_", 7)==0 ){
1457 return 0;
drh45e29d82006-11-20 16:21:10 +00001458 }else if( strncmp(zSql, "CREATE VIRTUAL TABLE", 20)==0 ){
1459 char *zIns;
1460 if( !p->writableSchema ){
1461 fprintf(p->out, "PRAGMA writable_schema=ON;\n");
1462 p->writableSchema = 1;
1463 }
1464 zIns = sqlite3_mprintf(
1465 "INSERT INTO sqlite_master(type,name,tbl_name,rootpage,sql)"
1466 "VALUES('table','%q','%q',0,'%q');",
1467 zTable, zTable, zSql);
1468 fprintf(p->out, "%s\n", zIns);
1469 sqlite3_free(zIns);
1470 return 0;
drh00b950d2005-09-11 02:03:03 +00001471 }else{
1472 fprintf(p->out, "%s;\n", zSql);
drhf8eb96a2005-02-03 00:42:34 +00001473 }
danielk19772a02e332004-06-05 08:04:36 +00001474
1475 if( strcmp(zType, "table")==0 ){
1476 sqlite3_stmt *pTableInfo = 0;
danielk19772a02e332004-06-05 08:04:36 +00001477 char *zSelect = 0;
1478 char *zTableInfo = 0;
1479 char *zTmp = 0;
drh157e29a2009-05-21 15:15:00 +00001480 int nRow = 0;
danielk19772a02e332004-06-05 08:04:36 +00001481
1482 zTableInfo = appendText(zTableInfo, "PRAGMA table_info(", 0);
1483 zTableInfo = appendText(zTableInfo, zTable, '"');
1484 zTableInfo = appendText(zTableInfo, ");", 0);
1485
drhc7181902014-02-27 15:04:13 +00001486 rc = sqlite3_prepare_v2(p->db, zTableInfo, -1, &pTableInfo, 0);
drh157e29a2009-05-21 15:15:00 +00001487 free(zTableInfo);
danielk19772a02e332004-06-05 08:04:36 +00001488 if( rc!=SQLITE_OK || !pTableInfo ){
1489 return 1;
1490 }
1491
1492 zSelect = appendText(zSelect, "SELECT 'INSERT INTO ' || ", 0);
drhbf92ec02012-03-22 12:50:34 +00001493 /* Always quote the table name, even if it appears to be pure ascii,
1494 ** in case it is a keyword. Ex: INSERT INTO "table" ... */
1495 zTmp = appendText(zTmp, zTable, '"');
danielk19772a02e332004-06-05 08:04:36 +00001496 if( zTmp ){
1497 zSelect = appendText(zSelect, zTmp, '\'');
drh85e72432012-04-11 11:38:53 +00001498 free(zTmp);
danielk19772a02e332004-06-05 08:04:36 +00001499 }
1500 zSelect = appendText(zSelect, " || ' VALUES(' || ", 0);
1501 rc = sqlite3_step(pTableInfo);
1502 while( rc==SQLITE_ROW ){
danielk19772e588c72005-12-09 14:25:08 +00001503 const char *zText = (const char *)sqlite3_column_text(pTableInfo, 1);
danielk19773f41e972004-06-08 00:39:01 +00001504 zSelect = appendText(zSelect, "quote(", 0);
danielk19772e588c72005-12-09 14:25:08 +00001505 zSelect = appendText(zSelect, zText, '"');
danielk19772a02e332004-06-05 08:04:36 +00001506 rc = sqlite3_step(pTableInfo);
1507 if( rc==SQLITE_ROW ){
drhb21a8e42012-01-28 21:08:51 +00001508 zSelect = appendText(zSelect, "), ", 0);
danielk19772a02e332004-06-05 08:04:36 +00001509 }else{
1510 zSelect = appendText(zSelect, ") ", 0);
1511 }
drh157e29a2009-05-21 15:15:00 +00001512 nRow++;
danielk19772a02e332004-06-05 08:04:36 +00001513 }
1514 rc = sqlite3_finalize(pTableInfo);
drh157e29a2009-05-21 15:15:00 +00001515 if( rc!=SQLITE_OK || nRow==0 ){
1516 free(zSelect);
danielk19772a02e332004-06-05 08:04:36 +00001517 return 1;
1518 }
1519 zSelect = appendText(zSelect, "|| ')' FROM ", 0);
1520 zSelect = appendText(zSelect, zTable, '"');
1521
drh2f464a02011-10-13 00:41:49 +00001522 rc = run_table_dump_query(p, zSelect, zPrepStmt);
drhdd3d4592004-08-30 01:54:05 +00001523 if( rc==SQLITE_CORRUPT ){
1524 zSelect = appendText(zSelect, " ORDER BY rowid DESC", 0);
drh2f464a02011-10-13 00:41:49 +00001525 run_table_dump_query(p, zSelect, 0);
drhdd3d4592004-08-30 01:54:05 +00001526 }
drh85e72432012-04-11 11:38:53 +00001527 free(zSelect);
drh4c653a02000-06-07 01:27:47 +00001528 }
drh4c653a02000-06-07 01:27:47 +00001529 return 0;
1530}
1531
1532/*
drh45e29d82006-11-20 16:21:10 +00001533** Run zQuery. Use dump_callback() as the callback routine so that
1534** the contents of the query are output as SQL statements.
1535**
drhdd3d4592004-08-30 01:54:05 +00001536** If we get a SQLITE_CORRUPT error, rerun the query after appending
1537** "ORDER BY rowid DESC" to the end.
1538*/
1539static int run_schema_dump_query(
1540 struct callback_data *p,
drh2f464a02011-10-13 00:41:49 +00001541 const char *zQuery
drhdd3d4592004-08-30 01:54:05 +00001542){
1543 int rc;
drh2f464a02011-10-13 00:41:49 +00001544 char *zErr = 0;
1545 rc = sqlite3_exec(p->db, zQuery, dump_callback, p, &zErr);
drhdd3d4592004-08-30 01:54:05 +00001546 if( rc==SQLITE_CORRUPT ){
1547 char *zQ2;
drh4f21c4a2008-12-10 22:15:00 +00001548 int len = strlen30(zQuery);
drh2f464a02011-10-13 00:41:49 +00001549 fprintf(p->out, "/****** CORRUPTION ERROR *******/\n");
1550 if( zErr ){
1551 fprintf(p->out, "/****** %s ******/\n", zErr);
1552 sqlite3_free(zErr);
1553 zErr = 0;
1554 }
drhdd3d4592004-08-30 01:54:05 +00001555 zQ2 = malloc( len+100 );
1556 if( zQ2==0 ) return rc;
drh8c5058b2012-04-16 17:22:30 +00001557 sqlite3_snprintf(len+100, zQ2, "%s ORDER BY rowid DESC", zQuery);
drh2f464a02011-10-13 00:41:49 +00001558 rc = sqlite3_exec(p->db, zQ2, dump_callback, p, &zErr);
1559 if( rc ){
1560 fprintf(p->out, "/****** ERROR: %s ******/\n", zErr);
1561 }else{
1562 rc = SQLITE_CORRUPT;
1563 }
1564 sqlite3_free(zErr);
drhdd3d4592004-08-30 01:54:05 +00001565 free(zQ2);
1566 }
1567 return rc;
1568}
1569
1570/*
drh75897232000-05-29 14:26:00 +00001571** Text of a help message
1572*/
persicom1d0b8722002-04-18 02:53:04 +00001573static char zHelp[] =
drh9ff849f2009-02-04 20:55:57 +00001574 ".backup ?DB? FILE Backup DB (default \"main\") to FILE\n"
drhc2ce0be2014-05-29 12:36:14 +00001575 ".bail on|off Stop after hitting an error. Default OFF\n"
drh4bbcf102014-02-06 02:46:08 +00001576 ".clone NEWDB Clone data into NEWDB from the existing database\n"
jplyon6a65bb32003-05-04 07:25:57 +00001577 ".databases List names and files of attached databases\n"
drhb860bc92004-08-04 15:16:55 +00001578 ".dump ?TABLE? ... Dump the database in an SQL text format\n"
shane86f5bdb2009-10-24 02:00:07 +00001579 " If TABLE specified, only dump tables matching\n"
1580 " LIKE pattern TABLE.\n"
drhc2ce0be2014-05-29 12:36:14 +00001581 ".echo on|off Turn command echo on or off\n"
drh75897232000-05-29 14:26:00 +00001582 ".exit Exit this program\n"
drhc2ce0be2014-05-29 12:36:14 +00001583 ".explain ?on|off? Turn output mode suitable for EXPLAIN on or off.\n"
shanehe2aa9d72009-11-06 17:20:17 +00001584 " With no args, it turns EXPLAIN on.\n"
drhc2ce0be2014-05-29 12:36:14 +00001585 ".headers on|off Turn display of headers on or off\n"
drh75897232000-05-29 14:26:00 +00001586 ".help Show this message\n"
drhb860bc92004-08-04 15:16:55 +00001587 ".import FILE TABLE Import data from FILE into TABLE\n"
shane86f5bdb2009-10-24 02:00:07 +00001588 ".indices ?TABLE? Show names of all indices\n"
1589 " If TABLE specified, only show indices for tables\n"
1590 " matching LIKE pattern TABLE.\n"
drhae5e4452007-05-03 17:18:36 +00001591#ifdef SQLITE_ENABLE_IOTRACE
1592 ".iotrace FILE Enable I/O diagnostic logging to FILE\n"
1593#endif
drh70df4fe2006-06-13 15:12:21 +00001594#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh1e397f82006-06-08 15:28:43 +00001595 ".load FILE ?ENTRY? Load an extension library\n"
drh70df4fe2006-06-13 15:12:21 +00001596#endif
drh127f9d72010-02-23 01:47:00 +00001597 ".log FILE|off Turn logging on or off. FILE can be stderr/stdout\n"
danielk19776b77a362005-01-13 11:10:25 +00001598 ".mode MODE ?TABLE? Set output mode where MODE is one of:\n"
drh3b584fa2004-09-24 12:50:03 +00001599 " csv Comma-separated values\n"
drhb860bc92004-08-04 15:16:55 +00001600 " column Left-aligned columns. (See .width)\n"
1601 " html HTML <table> code\n"
1602 " insert SQL insert statements for TABLE\n"
1603 " line One value per line\n"
1604 " list Values delimited by .separator string\n"
1605 " tabs Tab-separated values\n"
1606 " tcl TCL list elements\n"
drh078b1fd2012-09-21 13:40:02 +00001607 ".nullvalue STRING Use STRING in place of NULL values\n"
drhc2ce0be2014-05-29 12:36:14 +00001608 ".once FILENAME Output for the next SQL command only to FILENAME\n"
drh05782482013-10-24 15:20:20 +00001609 ".open ?FILENAME? Close existing database and reopen FILENAME\n"
drhc2ce0be2014-05-29 12:36:14 +00001610 ".output ?FILENAME? Send output to FILENAME or stdout\n"
drh078b1fd2012-09-21 13:40:02 +00001611 ".print STRING... Print literal STRING\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001612 ".prompt MAIN CONTINUE Replace the standard prompts\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001613 ".quit Exit this program\n"
drhdaffd0e2001-04-11 14:28:42 +00001614 ".read FILENAME Execute SQL in FILENAME\n"
drh9ff849f2009-02-04 20:55:57 +00001615 ".restore ?DB? FILE Restore content of DB (default \"main\") from FILE\n"
drh5c7976f2014-02-10 19:59:27 +00001616 ".save FILE Write in-memory database into FILE\n"
drh75897232000-05-29 14:26:00 +00001617 ".schema ?TABLE? Show the CREATE statements\n"
shane86f5bdb2009-10-24 02:00:07 +00001618 " If TABLE specified, only show tables matching\n"
1619 " LIKE pattern TABLE.\n"
drhb860bc92004-08-04 15:16:55 +00001620 ".separator STRING Change separator used by output mode and .import\n"
drh62cdde52014-05-28 20:22:28 +00001621 ".shell CMD ARGS... Run CMD ARGS... in a system shell\n"
drhdd45df82002-04-18 12:39:03 +00001622 ".show Show the current values for various settings\n"
drhc2ce0be2014-05-29 12:36:14 +00001623 ".stats on|off Turn stats on or off\n"
drh62cdde52014-05-28 20:22:28 +00001624 ".system CMD ARGS... Run CMD ARGS... in a system shell\n"
shane86f5bdb2009-10-24 02:00:07 +00001625 ".tables ?TABLE? List names of tables\n"
1626 " If TABLE specified, only list tables matching\n"
1627 " LIKE pattern TABLE.\n"
drh2dfbbca2000-07-28 14:32:48 +00001628 ".timeout MS Try opening locked tables for MS milliseconds\n"
drhc2ce0be2014-05-29 12:36:14 +00001629 ".timer on|off Turn SQL timer on or off\n"
drh42f64e52012-04-04 16:56:23 +00001630 ".trace FILE|off Output each SQL statement as it is run\n"
drhde60fc22011-12-14 17:53:36 +00001631 ".vfsname ?AUX? Print the name of the VFS stack\n"
shanehe2aa9d72009-11-06 17:20:17 +00001632 ".width NUM1 NUM2 ... Set column widths for \"column\" mode\n"
drh62cdde52014-05-28 20:22:28 +00001633 " Negative values right-justify\n"
drh75897232000-05-29 14:26:00 +00001634;
1635
drhdaffd0e2001-04-11 14:28:42 +00001636/* Forward reference */
drhc28490c2006-10-26 14:25:58 +00001637static int process_input(struct callback_data *p, FILE *in);
drhdaffd0e2001-04-11 14:28:42 +00001638
drh75897232000-05-29 14:26:00 +00001639/*
drh44c2eb12003-04-30 11:38:26 +00001640** Make sure the database is open. If it is not, then open it. If
1641** the database fails to open, print an error message and exit.
1642*/
drh05782482013-10-24 15:20:20 +00001643static void open_db(struct callback_data *p, int keepAlive){
drh44c2eb12003-04-30 11:38:26 +00001644 if( p->db==0 ){
drhbbb0be82012-06-27 16:12:27 +00001645 sqlite3_initialize();
danielk19774f057f92004-06-08 00:02:33 +00001646 sqlite3_open(p->zDbFilename, &p->db);
danielk197780290862004-05-22 09:21:21 +00001647 db = p->db;
drh4cea5ba2008-05-05 16:27:24 +00001648 if( db && sqlite3_errcode(db)==SQLITE_OK ){
1649 sqlite3_create_function(db, "shellstatic", 0, SQLITE_UTF8, 0,
1650 shellstaticFunc, 0, 0);
1651 }
1652 if( db==0 || SQLITE_OK!=sqlite3_errcode(db) ){
shane86f5bdb2009-10-24 02:00:07 +00001653 fprintf(stderr,"Error: unable to open database \"%s\": %s\n",
danielk197780290862004-05-22 09:21:21 +00001654 p->zDbFilename, sqlite3_errmsg(db));
drh05782482013-10-24 15:20:20 +00001655 if( keepAlive ) return;
drh22fbcb82004-02-01 01:22:50 +00001656 exit(1);
drh44c2eb12003-04-30 11:38:26 +00001657 }
drhc2e87a32006-06-27 15:16:14 +00001658#ifndef SQLITE_OMIT_LOAD_EXTENSION
1659 sqlite3_enable_load_extension(p->db, 1);
1660#endif
drh44c2eb12003-04-30 11:38:26 +00001661 }
1662}
1663
1664/*
drhfeac5f82004-08-01 00:10:45 +00001665** Do C-language style dequoting.
1666**
1667** \t -> tab
1668** \n -> newline
1669** \r -> carriage return
drh4c56b992013-06-27 13:26:55 +00001670** \" -> "
drhfeac5f82004-08-01 00:10:45 +00001671** \NNN -> ascii character NNN in octal
1672** \\ -> backslash
1673*/
1674static void resolve_backslashes(char *z){
shane7d3846a2008-12-11 02:58:26 +00001675 int i, j;
1676 char c;
drhc2ce0be2014-05-29 12:36:14 +00001677 while( *z && *z!='\\' ) z++;
drhfeac5f82004-08-01 00:10:45 +00001678 for(i=j=0; (c = z[i])!=0; i++, j++){
1679 if( c=='\\' ){
1680 c = z[++i];
1681 if( c=='n' ){
1682 c = '\n';
1683 }else if( c=='t' ){
1684 c = '\t';
1685 }else if( c=='r' ){
1686 c = '\r';
drh4c56b992013-06-27 13:26:55 +00001687 }else if( c=='\\' ){
1688 c = '\\';
drhfeac5f82004-08-01 00:10:45 +00001689 }else if( c>='0' && c<='7' ){
drhaa816082005-12-29 12:53:09 +00001690 c -= '0';
drhfeac5f82004-08-01 00:10:45 +00001691 if( z[i+1]>='0' && z[i+1]<='7' ){
1692 i++;
1693 c = (c<<3) + z[i] - '0';
1694 if( z[i+1]>='0' && z[i+1]<='7' ){
1695 i++;
1696 c = (c<<3) + z[i] - '0';
1697 }
1698 }
1699 }
1700 }
1701 z[j] = c;
1702 }
drhc2ce0be2014-05-29 12:36:14 +00001703 if( j<i ) z[j] = 0;
drhfeac5f82004-08-01 00:10:45 +00001704}
1705
1706/*
drh348d19c2013-06-03 12:47:43 +00001707** Return the value of a hexadecimal digit. Return -1 if the input
1708** is not a hex digit.
drhc28490c2006-10-26 14:25:58 +00001709*/
drh348d19c2013-06-03 12:47:43 +00001710static int hexDigitValue(char c){
1711 if( c>='0' && c<='9' ) return c - '0';
1712 if( c>='a' && c<='f' ) return c - 'a' + 10;
1713 if( c>='A' && c<='F' ) return c - 'A' + 10;
1714 return -1;
drhc28490c2006-10-26 14:25:58 +00001715}
1716
1717/*
drh7d9f3942013-04-03 01:26:54 +00001718** Interpret zArg as an integer value, possibly with suffixes.
1719*/
1720static sqlite3_int64 integerValue(const char *zArg){
1721 sqlite3_int64 v = 0;
1722 static const struct { char *zSuffix; int iMult; } aMult[] = {
1723 { "KiB", 1024 },
1724 { "MiB", 1024*1024 },
1725 { "GiB", 1024*1024*1024 },
1726 { "KB", 1000 },
1727 { "MB", 1000000 },
1728 { "GB", 1000000000 },
1729 { "K", 1000 },
1730 { "M", 1000000 },
1731 { "G", 1000000000 },
1732 };
1733 int i;
1734 int isNeg = 0;
1735 if( zArg[0]=='-' ){
1736 isNeg = 1;
1737 zArg++;
1738 }else if( zArg[0]=='+' ){
1739 zArg++;
1740 }
drh348d19c2013-06-03 12:47:43 +00001741 if( zArg[0]=='0' && zArg[1]=='x' ){
1742 int x;
1743 zArg += 2;
1744 while( (x = hexDigitValue(zArg[0]))>=0 ){
1745 v = (v<<4) + x;
1746 zArg++;
1747 }
1748 }else{
1749 while( IsDigit(zArg[0]) ){
1750 v = v*10 + zArg[0] - '0';
1751 zArg++;
1752 }
drh7d9f3942013-04-03 01:26:54 +00001753 }
drhc2bed0a2013-05-24 11:57:50 +00001754 for(i=0; i<ArraySize(aMult); i++){
drh7d9f3942013-04-03 01:26:54 +00001755 if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){
1756 v *= aMult[i].iMult;
1757 break;
1758 }
1759 }
1760 return isNeg? -v : v;
1761}
1762
1763/*
drh348d19c2013-06-03 12:47:43 +00001764** Interpret zArg as either an integer or a boolean value. Return 1 or 0
1765** for TRUE and FALSE. Return the integer value if appropriate.
1766*/
1767static int booleanValue(char *zArg){
1768 int i;
1769 if( zArg[0]=='0' && zArg[1]=='x' ){
1770 for(i=2; hexDigitValue(zArg[i])>=0; i++){}
1771 }else{
1772 for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){}
1773 }
1774 if( i>0 && zArg[i]==0 ) return (int)(integerValue(zArg) & 0xffffffff);
1775 if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){
1776 return 1;
1777 }
1778 if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){
1779 return 0;
1780 }
1781 fprintf(stderr, "ERROR: Not a boolean value: \"%s\". Assuming \"no\".\n",
1782 zArg);
1783 return 0;
1784}
1785
1786/*
drh42f64e52012-04-04 16:56:23 +00001787** Close an output file, assuming it is not stderr or stdout
1788*/
1789static void output_file_close(FILE *f){
1790 if( f && f!=stdout && f!=stderr ) fclose(f);
1791}
1792
1793/*
1794** Try to open an output file. The names "stdout" and "stderr" are
1795** recognized and do the right thing. NULL is returned if the output
1796** filename is "off".
1797*/
1798static FILE *output_file_open(const char *zFile){
1799 FILE *f;
1800 if( strcmp(zFile,"stdout")==0 ){
1801 f = stdout;
1802 }else if( strcmp(zFile, "stderr")==0 ){
1803 f = stderr;
1804 }else if( strcmp(zFile, "off")==0 ){
1805 f = 0;
1806 }else{
1807 f = fopen(zFile, "wb");
1808 if( f==0 ){
1809 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
1810 }
1811 }
1812 return f;
1813}
1814
1815/*
1816** A routine for handling output from sqlite3_trace().
1817*/
1818static void sql_trace_callback(void *pArg, const char *z){
1819 FILE *f = (FILE*)pArg;
1820 if( f ) fprintf(f, "%s\n", z);
1821}
1822
1823/*
drhd8621b92012-04-17 09:09:33 +00001824** A no-op routine that runs with the ".breakpoint" doc-command. This is
1825** a useful spot to set a debugger breakpoint.
1826*/
1827static void test_breakpoint(void){
1828 static int nCall = 0;
1829 nCall++;
1830}
1831
1832/*
drhdb95f682013-06-26 22:46:00 +00001833** An object used to read a CSV file
1834*/
1835typedef struct CSVReader CSVReader;
1836struct CSVReader {
1837 const char *zFile; /* Name of the input file */
1838 FILE *in; /* Read the CSV text from this input stream */
1839 char *z; /* Accumulated text for a field */
1840 int n; /* Number of bytes in z */
1841 int nAlloc; /* Space allocated for z[] */
1842 int nLine; /* Current line number */
1843 int cTerm; /* Character that terminated the most recent field */
1844 int cSeparator; /* The separator character. (Usually ",") */
1845};
1846
1847/* Append a single byte to z[] */
1848static void csv_append_char(CSVReader *p, int c){
1849 if( p->n+1>=p->nAlloc ){
1850 p->nAlloc += p->nAlloc + 100;
1851 p->z = sqlite3_realloc(p->z, p->nAlloc);
1852 if( p->z==0 ){
1853 fprintf(stderr, "out of memory\n");
1854 exit(1);
1855 }
1856 }
1857 p->z[p->n++] = (char)c;
1858}
1859
1860/* Read a single field of CSV text. Compatible with rfc4180 and extended
1861** with the option of having a separator other than ",".
1862**
1863** + Input comes from p->in.
1864** + Store results in p->z of length p->n. Space to hold p->z comes
1865** from sqlite3_malloc().
1866** + Use p->cSep as the separator. The default is ",".
1867** + Keep track of the line number in p->nLine.
1868** + Store the character that terminates the field in p->cTerm. Store
1869** EOF on end-of-file.
1870** + Report syntax errors on stderr
1871*/
1872static char *csv_read_one_field(CSVReader *p){
drha81ad172013-12-11 14:00:04 +00001873 int c, pc, ppc;
drhdb95f682013-06-26 22:46:00 +00001874 int cSep = p->cSeparator;
1875 p->n = 0;
1876 c = fgetc(p->in);
1877 if( c==EOF || seenInterrupt ){
1878 p->cTerm = EOF;
1879 return 0;
1880 }
1881 if( c=='"' ){
1882 int startLine = p->nLine;
1883 int cQuote = c;
drha81ad172013-12-11 14:00:04 +00001884 pc = ppc = 0;
drhdb95f682013-06-26 22:46:00 +00001885 while( 1 ){
1886 c = fgetc(p->in);
1887 if( c=='\n' ) p->nLine++;
1888 if( c==cQuote ){
1889 if( pc==cQuote ){
1890 pc = 0;
1891 continue;
1892 }
1893 }
1894 if( (c==cSep && pc==cQuote)
1895 || (c=='\n' && pc==cQuote)
drha81ad172013-12-11 14:00:04 +00001896 || (c=='\n' && pc=='\r' && ppc==cQuote)
drhdb95f682013-06-26 22:46:00 +00001897 || (c==EOF && pc==cQuote)
1898 ){
1899 do{ p->n--; }while( p->z[p->n]!=cQuote );
drhdb95f682013-06-26 22:46:00 +00001900 p->cTerm = c;
1901 break;
1902 }
1903 if( pc==cQuote && c!='\r' ){
1904 fprintf(stderr, "%s:%d: unescaped %c character\n",
1905 p->zFile, p->nLine, cQuote);
1906 }
1907 if( c==EOF ){
1908 fprintf(stderr, "%s:%d: unterminated %c-quoted field\n",
1909 p->zFile, startLine, cQuote);
drhdb95f682013-06-26 22:46:00 +00001910 p->cTerm = EOF;
1911 break;
1912 }
1913 csv_append_char(p, c);
drha81ad172013-12-11 14:00:04 +00001914 ppc = pc;
drhdb95f682013-06-26 22:46:00 +00001915 pc = c;
drhd0a64dc2013-06-30 20:24:26 +00001916 }
drhdb95f682013-06-26 22:46:00 +00001917 }else{
drhd0a64dc2013-06-30 20:24:26 +00001918 while( c!=EOF && c!=cSep && c!='\n' ){
drhdb95f682013-06-26 22:46:00 +00001919 csv_append_char(p, c);
drhd0a64dc2013-06-30 20:24:26 +00001920 c = fgetc(p->in);
drhdb95f682013-06-26 22:46:00 +00001921 }
1922 if( c=='\n' ){
1923 p->nLine++;
drh3852b682014-02-26 13:53:34 +00001924 if( p->n>0 && p->z[p->n-1]=='\r' ) p->n--;
drhdb95f682013-06-26 22:46:00 +00001925 }
drhdb95f682013-06-26 22:46:00 +00001926 p->cTerm = c;
1927 }
drh8dd675e2013-07-12 21:09:24 +00001928 if( p->z ) p->z[p->n] = 0;
drhdb95f682013-06-26 22:46:00 +00001929 return p->z;
1930}
1931
1932/*
drh4bbcf102014-02-06 02:46:08 +00001933** Try to transfer data for table zTable. If an error is seen while
1934** moving forward, try to go backwards. The backwards movement won't
1935** work for WITHOUT ROWID tables.
drh3350ce92014-02-06 00:49:12 +00001936*/
mistachkine31ae902014-02-06 01:15:29 +00001937static void tryToCloneData(
drh3350ce92014-02-06 00:49:12 +00001938 struct callback_data *p,
1939 sqlite3 *newDb,
1940 const char *zTable
1941){
1942 sqlite3_stmt *pQuery = 0;
1943 sqlite3_stmt *pInsert = 0;
1944 char *zQuery = 0;
1945 char *zInsert = 0;
1946 int rc;
1947 int i, j, n;
1948 int nTable = (int)strlen(zTable);
1949 int k = 0;
drh4bbcf102014-02-06 02:46:08 +00001950 int cnt = 0;
1951 const int spinRate = 10000;
drh3350ce92014-02-06 00:49:12 +00001952
1953 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\"", zTable);
1954 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
1955 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00001956 fprintf(stderr, "Error %d: %s on [%s]\n",
drh3350ce92014-02-06 00:49:12 +00001957 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
1958 zQuery);
1959 goto end_data_xfer;
1960 }
1961 n = sqlite3_column_count(pQuery);
1962 zInsert = sqlite3_malloc(200 + nTable + n*3);
1963 if( zInsert==0 ){
1964 fprintf(stderr, "out of memory\n");
1965 goto end_data_xfer;
1966 }
1967 sqlite3_snprintf(200+nTable,zInsert,
1968 "INSERT OR IGNORE INTO \"%s\" VALUES(?", zTable);
1969 i = (int)strlen(zInsert);
1970 for(j=1; j<n; j++){
1971 memcpy(zInsert+i, ",?", 2);
1972 i += 2;
1973 }
1974 memcpy(zInsert+i, ");", 3);
1975 rc = sqlite3_prepare_v2(newDb, zInsert, -1, &pInsert, 0);
1976 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00001977 fprintf(stderr, "Error %d: %s on [%s]\n",
drh3350ce92014-02-06 00:49:12 +00001978 sqlite3_extended_errcode(newDb), sqlite3_errmsg(newDb),
1979 zQuery);
1980 goto end_data_xfer;
1981 }
1982 for(k=0; k<2; k++){
1983 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
1984 for(i=0; i<n; i++){
1985 switch( sqlite3_column_type(pQuery, i) ){
1986 case SQLITE_NULL: {
1987 sqlite3_bind_null(pInsert, i+1);
1988 break;
1989 }
1990 case SQLITE_INTEGER: {
1991 sqlite3_bind_int64(pInsert, i+1, sqlite3_column_int64(pQuery,i));
1992 break;
1993 }
1994 case SQLITE_FLOAT: {
1995 sqlite3_bind_double(pInsert, i+1, sqlite3_column_double(pQuery,i));
1996 break;
1997 }
1998 case SQLITE_TEXT: {
1999 sqlite3_bind_text(pInsert, i+1,
2000 (const char*)sqlite3_column_text(pQuery,i),
2001 -1, SQLITE_STATIC);
2002 break;
2003 }
2004 case SQLITE_BLOB: {
2005 sqlite3_bind_blob(pInsert, i+1, sqlite3_column_blob(pQuery,i),
2006 sqlite3_column_bytes(pQuery,i),
2007 SQLITE_STATIC);
2008 break;
2009 }
2010 }
2011 } /* End for */
drh4bbcf102014-02-06 02:46:08 +00002012 rc = sqlite3_step(pInsert);
2013 if( rc!=SQLITE_OK && rc!=SQLITE_ROW && rc!=SQLITE_DONE ){
2014 fprintf(stderr, "Error %d: %s\n", sqlite3_extended_errcode(newDb),
2015 sqlite3_errmsg(newDb));
2016 }
drh3350ce92014-02-06 00:49:12 +00002017 sqlite3_reset(pInsert);
drh4bbcf102014-02-06 02:46:08 +00002018 cnt++;
2019 if( (cnt%spinRate)==0 ){
2020 printf("%c\b", "|/-\\"[(cnt/spinRate)%4]);
2021 fflush(stdout);
2022 }
drh3350ce92014-02-06 00:49:12 +00002023 } /* End while */
2024 if( rc==SQLITE_DONE ) break;
2025 sqlite3_finalize(pQuery);
2026 sqlite3_free(zQuery);
2027 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\" ORDER BY rowid DESC;",
2028 zTable);
2029 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2030 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00002031 fprintf(stderr, "Warning: cannot step \"%s\" backwards", zTable);
2032 break;
drh3350ce92014-02-06 00:49:12 +00002033 }
2034 } /* End for(k=0...) */
2035
2036end_data_xfer:
2037 sqlite3_finalize(pQuery);
2038 sqlite3_finalize(pInsert);
2039 sqlite3_free(zQuery);
2040 sqlite3_free(zInsert);
2041}
2042
2043
2044/*
2045** Try to transfer all rows of the schema that match zWhere. For
2046** each row, invoke xForEach() on the object defined by that row.
drh4bbcf102014-02-06 02:46:08 +00002047** If an error is encountered while moving forward through the
2048** sqlite_master table, try again moving backwards.
drh3350ce92014-02-06 00:49:12 +00002049*/
mistachkine31ae902014-02-06 01:15:29 +00002050static void tryToCloneSchema(
drh3350ce92014-02-06 00:49:12 +00002051 struct callback_data *p,
2052 sqlite3 *newDb,
2053 const char *zWhere,
2054 void (*xForEach)(struct callback_data*,sqlite3*,const char*)
2055){
2056 sqlite3_stmt *pQuery = 0;
2057 char *zQuery = 0;
2058 int rc;
2059 const unsigned char *zName;
2060 const unsigned char *zSql;
drh4bbcf102014-02-06 02:46:08 +00002061 char *zErrMsg = 0;
drh3350ce92014-02-06 00:49:12 +00002062
2063 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_master"
2064 " WHERE %s", zWhere);
2065 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2066 if( rc ){
2067 fprintf(stderr, "Error: (%d) %s on [%s]\n",
2068 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
2069 zQuery);
2070 goto end_schema_xfer;
2071 }
2072 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
2073 zName = sqlite3_column_text(pQuery, 0);
2074 zSql = sqlite3_column_text(pQuery, 1);
2075 printf("%s... ", zName); fflush(stdout);
drh4bbcf102014-02-06 02:46:08 +00002076 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
2077 if( zErrMsg ){
2078 fprintf(stderr, "Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
2079 sqlite3_free(zErrMsg);
2080 zErrMsg = 0;
2081 }
drh3350ce92014-02-06 00:49:12 +00002082 if( xForEach ){
2083 xForEach(p, newDb, (const char*)zName);
2084 }
2085 printf("done\n");
2086 }
2087 if( rc!=SQLITE_DONE ){
2088 sqlite3_finalize(pQuery);
2089 sqlite3_free(zQuery);
2090 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_master"
2091 " WHERE %s ORDER BY rowid DESC", zWhere);
2092 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2093 if( rc ){
2094 fprintf(stderr, "Error: (%d) %s on [%s]\n",
2095 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
2096 zQuery);
2097 goto end_schema_xfer;
2098 }
2099 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
2100 zName = sqlite3_column_text(pQuery, 0);
2101 zSql = sqlite3_column_text(pQuery, 1);
2102 printf("%s... ", zName); fflush(stdout);
drh4bbcf102014-02-06 02:46:08 +00002103 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
2104 if( zErrMsg ){
2105 fprintf(stderr, "Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
2106 sqlite3_free(zErrMsg);
2107 zErrMsg = 0;
2108 }
drh3350ce92014-02-06 00:49:12 +00002109 if( xForEach ){
2110 xForEach(p, newDb, (const char*)zName);
2111 }
2112 printf("done\n");
2113 }
2114 }
2115end_schema_xfer:
2116 sqlite3_finalize(pQuery);
2117 sqlite3_free(zQuery);
2118}
2119
2120/*
2121** Open a new database file named "zNewDb". Try to recover as much information
2122** as possible out of the main database (which might be corrupt) and write it
2123** into zNewDb.
2124*/
mistachkine31ae902014-02-06 01:15:29 +00002125static void tryToClone(struct callback_data *p, const char *zNewDb){
drh3350ce92014-02-06 00:49:12 +00002126 int rc;
2127 sqlite3 *newDb = 0;
2128 if( access(zNewDb,0)==0 ){
2129 fprintf(stderr, "File \"%s\" already exists.\n", zNewDb);
2130 return;
2131 }
2132 rc = sqlite3_open(zNewDb, &newDb);
2133 if( rc ){
2134 fprintf(stderr, "Cannot create output database: %s\n",
2135 sqlite3_errmsg(newDb));
2136 }else{
drh54d0d2d2014-04-03 00:32:13 +00002137 sqlite3_exec(p->db, "PRAGMA writable_schema=ON;", 0, 0, 0);
drh3350ce92014-02-06 00:49:12 +00002138 sqlite3_exec(newDb, "BEGIN EXCLUSIVE;", 0, 0, 0);
mistachkine31ae902014-02-06 01:15:29 +00002139 tryToCloneSchema(p, newDb, "type='table'", tryToCloneData);
2140 tryToCloneSchema(p, newDb, "type!='table'", 0);
drh3350ce92014-02-06 00:49:12 +00002141 sqlite3_exec(newDb, "COMMIT;", 0, 0, 0);
drh54d0d2d2014-04-03 00:32:13 +00002142 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
drh3350ce92014-02-06 00:49:12 +00002143 }
2144 sqlite3_close(newDb);
2145}
2146
2147/*
drhc2ce0be2014-05-29 12:36:14 +00002148** Change the output file back to stdout
2149*/
2150static void output_reset(struct callback_data *p){
2151 if( p->outfile[0]=='|' ){
2152 pclose(p->out);
2153 }else{
2154 output_file_close(p->out);
2155 }
2156 p->outfile[0] = 0;
2157 p->out = stdout;
2158}
2159
2160/*
drh75897232000-05-29 14:26:00 +00002161** If an input line begins with "." then invoke this routine to
2162** process that line.
drh67505e72002-04-19 12:34:06 +00002163**
drh47ad6842006-11-08 12:25:42 +00002164** Return 1 on error, 2 to exit, and 0 otherwise.
drh75897232000-05-29 14:26:00 +00002165*/
drh44c2eb12003-04-30 11:38:26 +00002166static int do_meta_command(char *zLine, struct callback_data *p){
drh75897232000-05-29 14:26:00 +00002167 int i = 1;
2168 int nArg = 0;
2169 int n, c;
drh67505e72002-04-19 12:34:06 +00002170 int rc = 0;
drh75897232000-05-29 14:26:00 +00002171 char *azArg[50];
2172
2173 /* Parse the input line into tokens.
2174 */
2175 while( zLine[i] && nArg<ArraySize(azArg) ){
drhf0693c82011-10-11 20:41:54 +00002176 while( IsSpace(zLine[i]) ){ i++; }
drh06333682004-03-09 13:37:45 +00002177 if( zLine[i]==0 ) break;
drh75897232000-05-29 14:26:00 +00002178 if( zLine[i]=='\'' || zLine[i]=='"' ){
2179 int delim = zLine[i++];
2180 azArg[nArg++] = &zLine[i];
drh4c56b992013-06-27 13:26:55 +00002181 while( zLine[i] && zLine[i]!=delim ){
2182 if( zLine[i]=='\\' && delim=='"' && zLine[i+1]!=0 ) i++;
2183 i++;
2184 }
drh75897232000-05-29 14:26:00 +00002185 if( zLine[i]==delim ){
2186 zLine[i++] = 0;
2187 }
drhfeac5f82004-08-01 00:10:45 +00002188 if( delim=='"' ) resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00002189 }else{
2190 azArg[nArg++] = &zLine[i];
drhf0693c82011-10-11 20:41:54 +00002191 while( zLine[i] && !IsSpace(zLine[i]) ){ i++; }
drh75897232000-05-29 14:26:00 +00002192 if( zLine[i] ) zLine[i++] = 0;
drhfeac5f82004-08-01 00:10:45 +00002193 resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00002194 }
2195 }
2196
2197 /* Process the input line.
2198 */
shane9bd1b442009-10-23 01:27:39 +00002199 if( nArg==0 ) return 0; /* no tokens, no error */
drh4f21c4a2008-12-10 22:15:00 +00002200 n = strlen30(azArg[0]);
drh75897232000-05-29 14:26:00 +00002201 c = azArg[0][0];
drh5c7976f2014-02-10 19:59:27 +00002202 if( (c=='b' && n>=3 && strncmp(azArg[0], "backup", n)==0)
2203 || (c=='s' && n>=3 && strncmp(azArg[0], "save", n)==0)
2204 ){
drhbc46f022013-01-23 18:53:23 +00002205 const char *zDestFile = 0;
2206 const char *zDb = 0;
drh9ff849f2009-02-04 20:55:57 +00002207 sqlite3 *pDest;
2208 sqlite3_backup *pBackup;
drhbc46f022013-01-23 18:53:23 +00002209 int j;
2210 for(j=1; j<nArg; j++){
2211 const char *z = azArg[j];
2212 if( z[0]=='-' ){
2213 while( z[0]=='-' ) z++;
drhaf664332013-07-18 20:28:29 +00002214 /* No options to process at this time */
drhbc46f022013-01-23 18:53:23 +00002215 {
2216 fprintf(stderr, "unknown option: %s\n", azArg[j]);
2217 return 1;
2218 }
2219 }else if( zDestFile==0 ){
2220 zDestFile = azArg[j];
2221 }else if( zDb==0 ){
2222 zDb = zDestFile;
2223 zDestFile = azArg[j];
2224 }else{
2225 fprintf(stderr, "too many arguments to .backup\n");
2226 return 1;
2227 }
drh9ff849f2009-02-04 20:55:57 +00002228 }
drhbc46f022013-01-23 18:53:23 +00002229 if( zDestFile==0 ){
2230 fprintf(stderr, "missing FILENAME argument on .backup\n");
2231 return 1;
2232 }
2233 if( zDb==0 ) zDb = "main";
drh9ff849f2009-02-04 20:55:57 +00002234 rc = sqlite3_open(zDestFile, &pDest);
2235 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002236 fprintf(stderr, "Error: cannot open \"%s\"\n", zDestFile);
drh9ff849f2009-02-04 20:55:57 +00002237 sqlite3_close(pDest);
2238 return 1;
2239 }
drh05782482013-10-24 15:20:20 +00002240 open_db(p, 0);
drh9ff849f2009-02-04 20:55:57 +00002241 pBackup = sqlite3_backup_init(pDest, "main", p->db, zDb);
2242 if( pBackup==0 ){
2243 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
2244 sqlite3_close(pDest);
2245 return 1;
2246 }
2247 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK ){}
2248 sqlite3_backup_finish(pBackup);
2249 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00002250 rc = 0;
drh9ff849f2009-02-04 20:55:57 +00002251 }else{
2252 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
shane9bd1b442009-10-23 01:27:39 +00002253 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002254 }
2255 sqlite3_close(pDest);
2256 }else
2257
drhc2ce0be2014-05-29 12:36:14 +00002258 if( c=='b' && n>=3 && strncmp(azArg[0], "bail", n)==0 ){
2259 if( nArg==2 ){
2260 bail_on_error = booleanValue(azArg[1]);
2261 }else{
2262 fprintf(stderr, "Usage: .bail on|off\n");
2263 rc = 1;
2264 }
drhc49f44e2006-10-26 18:15:42 +00002265 }else
2266
drhd8621b92012-04-17 09:09:33 +00002267 /* The undocumented ".breakpoint" command causes a call to the no-op
2268 ** routine named test_breakpoint().
2269 */
2270 if( c=='b' && n>=3 && strncmp(azArg[0], "breakpoint", n)==0 ){
2271 test_breakpoint();
2272 }else
2273
drhc2ce0be2014-05-29 12:36:14 +00002274 if( c=='c' && strncmp(azArg[0], "clone", n)==0 ){
2275 if( nArg==2 ){
2276 tryToClone(p, azArg[1]);
2277 }else{
2278 fprintf(stderr, "Usage: .clone FILENAME\n");
2279 rc = 1;
2280 }
mistachkine31ae902014-02-06 01:15:29 +00002281 }else
2282
drhc2ce0be2014-05-29 12:36:14 +00002283 if( c=='d' && n>1 && strncmp(azArg[0], "databases", n)==0 ){
jplyon672a1ed2003-05-11 20:07:05 +00002284 struct callback_data data;
2285 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002286 open_db(p, 0);
jplyon672a1ed2003-05-11 20:07:05 +00002287 memcpy(&data, p, sizeof(data));
drhd8885442004-03-17 23:42:12 +00002288 data.showHeader = 1;
jplyon672a1ed2003-05-11 20:07:05 +00002289 data.mode = MODE_Column;
drhd8885442004-03-17 23:42:12 +00002290 data.colWidth[0] = 3;
2291 data.colWidth[1] = 15;
2292 data.colWidth[2] = 58;
drh0b2110c2004-10-26 00:08:10 +00002293 data.cnt = 0;
danielk19776f8a5032004-05-10 10:34:51 +00002294 sqlite3_exec(p->db, "PRAGMA database_list; ", callback, &data, &zErrMsg);
jplyon672a1ed2003-05-11 20:07:05 +00002295 if( zErrMsg ){
2296 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002297 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002298 rc = 1;
jplyon6a65bb32003-05-04 07:25:57 +00002299 }
2300 }else
2301
drhc2ce0be2014-05-29 12:36:14 +00002302 if( c=='d' && strncmp(azArg[0], "dump", n)==0 ){
drh05782482013-10-24 15:20:20 +00002303 open_db(p, 0);
drhf1dfc4f2009-09-23 15:51:35 +00002304 /* When playing back a "dump", the content might appear in an order
2305 ** which causes immediate foreign key constraints to be violated.
2306 ** So disable foreign-key constraint enforcement to prevent problems. */
drhc2ce0be2014-05-29 12:36:14 +00002307 if( nArg!=1 && nArg!=2 ){
2308 fprintf(stderr, "Usage: .dump ?LIKE-PATTERN?\n");
2309 rc = 1;
2310 goto meta_command_exit;
2311 }
drhf1dfc4f2009-09-23 15:51:35 +00002312 fprintf(p->out, "PRAGMA foreign_keys=OFF;\n");
drh33048c02001-10-01 14:29:22 +00002313 fprintf(p->out, "BEGIN TRANSACTION;\n");
drh45e29d82006-11-20 16:21:10 +00002314 p->writableSchema = 0;
drh56197952011-10-13 16:30:13 +00002315 sqlite3_exec(p->db, "SAVEPOINT dump; PRAGMA writable_schema=ON", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00002316 p->nErr = 0;
drh4c653a02000-06-07 01:27:47 +00002317 if( nArg==1 ){
drhdd3d4592004-08-30 01:54:05 +00002318 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00002319 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00002320 "WHERE sql NOT NULL AND type=='table' AND name!='sqlite_sequence'"
drh4f324762009-05-21 14:51:03 +00002321 );
2322 run_schema_dump_query(p,
2323 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00002324 "WHERE name=='sqlite_sequence'"
drh0b9a5942006-09-13 20:22:02 +00002325 );
drh2f464a02011-10-13 00:41:49 +00002326 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00002327 "SELECT sql FROM sqlite_master "
drh157e29a2009-05-21 15:15:00 +00002328 "WHERE sql NOT NULL AND type IN ('index','trigger','view')", 0
drha18c5682000-10-08 22:20:57 +00002329 );
drh4c653a02000-06-07 01:27:47 +00002330 }else{
2331 int i;
drhdd3d4592004-08-30 01:54:05 +00002332 for(i=1; i<nArg; i++){
danielk1977bc6ada42004-06-30 08:20:16 +00002333 zShellStatic = azArg[i];
drhdd3d4592004-08-30 01:54:05 +00002334 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00002335 "SELECT name, type, sql FROM sqlite_master "
drhdd3d4592004-08-30 01:54:05 +00002336 "WHERE tbl_name LIKE shellstatic() AND type=='table'"
drh2f464a02011-10-13 00:41:49 +00002337 " AND sql NOT NULL");
2338 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00002339 "SELECT sql FROM sqlite_master "
drh45e29d82006-11-20 16:21:10 +00002340 "WHERE sql NOT NULL"
2341 " AND type IN ('index','trigger','view')"
drh157e29a2009-05-21 15:15:00 +00002342 " AND tbl_name LIKE shellstatic()", 0
drh0b9a5942006-09-13 20:22:02 +00002343 );
danielk1977bc6ada42004-06-30 08:20:16 +00002344 zShellStatic = 0;
drh4c653a02000-06-07 01:27:47 +00002345 }
2346 }
drh45e29d82006-11-20 16:21:10 +00002347 if( p->writableSchema ){
drh56197952011-10-13 16:30:13 +00002348 fprintf(p->out, "PRAGMA writable_schema=OFF;\n");
drh45e29d82006-11-20 16:21:10 +00002349 p->writableSchema = 0;
2350 }
drh56197952011-10-13 16:30:13 +00002351 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
2352 sqlite3_exec(p->db, "RELEASE dump;", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00002353 fprintf(p->out, p->nErr ? "ROLLBACK; -- due to errors\n" : "COMMIT;\n");
drh4c653a02000-06-07 01:27:47 +00002354 }else
drh75897232000-05-29 14:26:00 +00002355
drhc2ce0be2014-05-29 12:36:14 +00002356 if( c=='e' && strncmp(azArg[0], "echo", n)==0 ){
2357 if( nArg==2 ){
2358 p->echoOn = booleanValue(azArg[1]);
2359 }else{
2360 fprintf(stderr, "Usage: .echo on|off\n");
2361 rc = 1;
2362 }
drhdaffd0e2001-04-11 14:28:42 +00002363 }else
2364
drhc2ce0be2014-05-29 12:36:14 +00002365 if( c=='e' && strncmp(azArg[0], "eqp", n)==0 ){
2366 if( nArg==2 ){
2367 p->autoEQP = booleanValue(azArg[1]);
2368 }else{
2369 fprintf(stderr, "Usage: .eqp on|off\n");
2370 rc = 1;
2371 }
drhefbf3b12014-02-28 20:47:24 +00002372 }else
2373
drhd3ac7d92013-01-25 18:33:43 +00002374 if( c=='e' && strncmp(azArg[0], "exit", n)==0 ){
drh348d19c2013-06-03 12:47:43 +00002375 if( nArg>1 && (rc = (int)integerValue(azArg[1]))!=0 ) exit(rc);
drh47ad6842006-11-08 12:25:42 +00002376 rc = 2;
drh75897232000-05-29 14:26:00 +00002377 }else
2378
drhc2ce0be2014-05-29 12:36:14 +00002379 if( c=='e' && strncmp(azArg[0], "explain", n)==0 ){
drhc28490c2006-10-26 14:25:58 +00002380 int val = nArg>=2 ? booleanValue(azArg[1]) : 1;
persicom7e2dfdd2002-04-18 02:46:52 +00002381 if(val == 1) {
2382 if(!p->explainPrev.valid) {
2383 p->explainPrev.valid = 1;
2384 p->explainPrev.mode = p->mode;
2385 p->explainPrev.showHeader = p->showHeader;
2386 memcpy(p->explainPrev.colWidth,p->colWidth,sizeof(p->colWidth));
2387 }
2388 /* We could put this code under the !p->explainValid
2389 ** condition so that it does not execute if we are already in
2390 ** explain mode. However, always executing it allows us an easy
2391 ** was to reset to explain mode in case the user previously
2392 ** did an .explain followed by a .width, .mode or .header
2393 ** command.
2394 */
danielk19770d78bae2008-01-03 07:09:48 +00002395 p->mode = MODE_Explain;
persicom7e2dfdd2002-04-18 02:46:52 +00002396 p->showHeader = 1;
drhac68ced2013-11-27 13:24:18 +00002397 memset(p->colWidth,0,sizeof(p->colWidth));
danielk19770d78bae2008-01-03 07:09:48 +00002398 p->colWidth[0] = 4; /* addr */
drh60a713c2008-01-21 16:22:45 +00002399 p->colWidth[1] = 13; /* opcode */
2400 p->colWidth[2] = 4; /* P1 */
2401 p->colWidth[3] = 4; /* P2 */
2402 p->colWidth[4] = 4; /* P3 */
2403 p->colWidth[5] = 13; /* P4 */
danielk19770d78bae2008-01-03 07:09:48 +00002404 p->colWidth[6] = 2; /* P5 */
drh60a713c2008-01-21 16:22:45 +00002405 p->colWidth[7] = 13; /* Comment */
persicom7e2dfdd2002-04-18 02:46:52 +00002406 }else if (p->explainPrev.valid) {
2407 p->explainPrev.valid = 0;
2408 p->mode = p->explainPrev.mode;
2409 p->showHeader = p->explainPrev.showHeader;
2410 memcpy(p->colWidth,p->explainPrev.colWidth,sizeof(p->colWidth));
2411 }
drh75897232000-05-29 14:26:00 +00002412 }else
2413
drhc2ce0be2014-05-29 12:36:14 +00002414 if( c=='h' && strncmp(azArg[0], "headers", n)==0 ){
2415 if( nArg==2 ){
2416 p->showHeader = booleanValue(azArg[1]);
2417 }else{
2418 fprintf(stderr, "Usage: .headers on|off\n");
2419 rc = 1;
shaneb320ccd2009-10-21 03:42:58 +00002420 }
drh75897232000-05-29 14:26:00 +00002421 }else
2422
drhc2ce0be2014-05-29 12:36:14 +00002423 if( c=='h' && strncmp(azArg[0], "help", n)==0 ){
2424 fprintf(p->out, "%s", zHelp);
2425 }else
2426
2427 if( c=='i' && strncmp(azArg[0], "import", n)==0 ){
drh01f37542014-05-31 15:43:33 +00002428 char *zTable; /* Insert data into this table */
2429 char *zFile; /* Name of file to extra content from */
shane916f9612009-10-23 00:37:15 +00002430 sqlite3_stmt *pStmt = NULL; /* A statement */
drhfeac5f82004-08-01 00:10:45 +00002431 int nCol; /* Number of columns in the table */
2432 int nByte; /* Number of bytes in an SQL string */
2433 int i, j; /* Loop counters */
drh2d463112013-08-06 14:36:36 +00002434 int needCommit; /* True to COMMIT or ROLLBACK at end */
drhfeac5f82004-08-01 00:10:45 +00002435 int nSep; /* Number of bytes in p->separator[] */
2436 char *zSql; /* An SQL statement */
drhdb95f682013-06-26 22:46:00 +00002437 CSVReader sCsv; /* Reader context */
drh5bde8162013-06-27 14:07:53 +00002438 int (*xCloser)(FILE*); /* Procedure to close th3 connection */
drhfeac5f82004-08-01 00:10:45 +00002439
drhc2ce0be2014-05-29 12:36:14 +00002440 if( nArg!=3 ){
2441 fprintf(stderr, "Usage: .import FILE TABLE\n");
2442 goto meta_command_exit;
2443 }
drh01f37542014-05-31 15:43:33 +00002444 zFile = azArg[1];
2445 zTable = azArg[2];
drhdb95f682013-06-26 22:46:00 +00002446 seenInterrupt = 0;
2447 memset(&sCsv, 0, sizeof(sCsv));
drh05782482013-10-24 15:20:20 +00002448 open_db(p, 0);
drh4f21c4a2008-12-10 22:15:00 +00002449 nSep = strlen30(p->separator);
drhfeac5f82004-08-01 00:10:45 +00002450 if( nSep==0 ){
shane916f9612009-10-23 00:37:15 +00002451 fprintf(stderr, "Error: non-null separator required for import\n");
2452 return 1;
drhfeac5f82004-08-01 00:10:45 +00002453 }
drhdb95f682013-06-26 22:46:00 +00002454 if( nSep>1 ){
2455 fprintf(stderr, "Error: multi-character separators not allowed"
2456 " for import\n");
2457 return 1;
2458 }
drh5bde8162013-06-27 14:07:53 +00002459 sCsv.zFile = zFile;
2460 sCsv.nLine = 1;
2461 if( sCsv.zFile[0]=='|' ){
2462 sCsv.in = popen(sCsv.zFile+1, "r");
2463 sCsv.zFile = "<pipe>";
2464 xCloser = pclose;
2465 }else{
2466 sCsv.in = fopen(sCsv.zFile, "rb");
2467 xCloser = fclose;
2468 }
drhdb95f682013-06-26 22:46:00 +00002469 if( sCsv.in==0 ){
drh5bde8162013-06-27 14:07:53 +00002470 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
drhdb95f682013-06-26 22:46:00 +00002471 return 1;
2472 }
2473 sCsv.cSeparator = p->separator[0];
drh7b075e32011-09-28 01:10:00 +00002474 zSql = sqlite3_mprintf("SELECT * FROM %s", zTable);
shane916f9612009-10-23 00:37:15 +00002475 if( zSql==0 ){
2476 fprintf(stderr, "Error: out of memory\n");
drh5bde8162013-06-27 14:07:53 +00002477 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002478 return 1;
2479 }
drh4f21c4a2008-12-10 22:15:00 +00002480 nByte = strlen30(zSql);
drhc7181902014-02-27 15:04:13 +00002481 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
dan6a8ac852014-05-26 18:27:12 +00002482 csv_append_char(&sCsv, 0); /* To ensure sCsv.z is allocated */
drhdb95f682013-06-26 22:46:00 +00002483 if( rc && sqlite3_strglob("no such table: *", sqlite3_errmsg(db))==0 ){
2484 char *zCreate = sqlite3_mprintf("CREATE TABLE %s", zTable);
2485 char cSep = '(';
2486 while( csv_read_one_field(&sCsv) ){
2487 zCreate = sqlite3_mprintf("%z%c\n \"%s\" TEXT", zCreate, cSep, sCsv.z);
2488 cSep = ',';
2489 if( sCsv.cTerm!=sCsv.cSeparator ) break;
2490 }
drh5bde8162013-06-27 14:07:53 +00002491 if( cSep=='(' ){
2492 sqlite3_free(zCreate);
2493 sqlite3_free(sCsv.z);
2494 xCloser(sCsv.in);
2495 fprintf(stderr,"%s: empty file\n", sCsv.zFile);
2496 return 1;
2497 }
drhdb95f682013-06-26 22:46:00 +00002498 zCreate = sqlite3_mprintf("%z\n)", zCreate);
2499 rc = sqlite3_exec(p->db, zCreate, 0, 0, 0);
2500 sqlite3_free(zCreate);
2501 if( rc ){
2502 fprintf(stderr, "CREATE TABLE %s(...) failed: %s\n", zTable,
2503 sqlite3_errmsg(db));
2504 sqlite3_free(sCsv.z);
drh5bde8162013-06-27 14:07:53 +00002505 xCloser(sCsv.in);
drhdb95f682013-06-26 22:46:00 +00002506 return 1;
2507 }
drhc7181902014-02-27 15:04:13 +00002508 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
drhdb95f682013-06-26 22:46:00 +00002509 }
drhfeac5f82004-08-01 00:10:45 +00002510 sqlite3_free(zSql);
2511 if( rc ){
shane916f9612009-10-23 00:37:15 +00002512 if (pStmt) sqlite3_finalize(pStmt);
drhfeac5f82004-08-01 00:10:45 +00002513 fprintf(stderr,"Error: %s\n", sqlite3_errmsg(db));
drh5bde8162013-06-27 14:07:53 +00002514 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002515 return 1;
drhfeac5f82004-08-01 00:10:45 +00002516 }
shane916f9612009-10-23 00:37:15 +00002517 nCol = sqlite3_column_count(pStmt);
drhfeac5f82004-08-01 00:10:45 +00002518 sqlite3_finalize(pStmt);
shane916f9612009-10-23 00:37:15 +00002519 pStmt = 0;
shane9bd1b442009-10-23 01:27:39 +00002520 if( nCol==0 ) return 0; /* no columns, no error */
drhdb95f682013-06-26 22:46:00 +00002521 zSql = sqlite3_malloc( nByte*2 + 20 + nCol*2 );
shane916f9612009-10-23 00:37:15 +00002522 if( zSql==0 ){
2523 fprintf(stderr, "Error: out of memory\n");
drh5bde8162013-06-27 14:07:53 +00002524 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002525 return 1;
2526 }
drhdb95f682013-06-26 22:46:00 +00002527 sqlite3_snprintf(nByte+20, zSql, "INSERT INTO \"%w\" VALUES(?", zTable);
drh4f21c4a2008-12-10 22:15:00 +00002528 j = strlen30(zSql);
drhfeac5f82004-08-01 00:10:45 +00002529 for(i=1; i<nCol; i++){
2530 zSql[j++] = ',';
2531 zSql[j++] = '?';
2532 }
2533 zSql[j++] = ')';
2534 zSql[j] = 0;
drhc7181902014-02-27 15:04:13 +00002535 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
drhdb95f682013-06-26 22:46:00 +00002536 sqlite3_free(zSql);
drhfeac5f82004-08-01 00:10:45 +00002537 if( rc ){
2538 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(db));
shane916f9612009-10-23 00:37:15 +00002539 if (pStmt) sqlite3_finalize(pStmt);
drh5bde8162013-06-27 14:07:53 +00002540 xCloser(sCsv.in);
drh47ad6842006-11-08 12:25:42 +00002541 return 1;
drhfeac5f82004-08-01 00:10:45 +00002542 }
drh2d463112013-08-06 14:36:36 +00002543 needCommit = sqlite3_get_autocommit(db);
2544 if( needCommit ) sqlite3_exec(db, "BEGIN", 0, 0, 0);
drhdb95f682013-06-26 22:46:00 +00002545 do{
2546 int startLine = sCsv.nLine;
drhfeac5f82004-08-01 00:10:45 +00002547 for(i=0; i<nCol; i++){
drhdb95f682013-06-26 22:46:00 +00002548 char *z = csv_read_one_field(&sCsv);
2549 if( z==0 && i==0 ) break;
2550 sqlite3_bind_text(pStmt, i+1, z, -1, SQLITE_TRANSIENT);
2551 if( i<nCol-1 && sCsv.cTerm!=sCsv.cSeparator ){
2552 fprintf(stderr, "%s:%d: expected %d columns but found %d - "
2553 "filling the rest with NULL\n",
2554 sCsv.zFile, startLine, nCol, i+1);
2555 i++;
2556 while( i<nCol ){ sqlite3_bind_null(pStmt, i); i++; }
drh18f52e02012-01-16 16:56:31 +00002557 }
drhfeac5f82004-08-01 00:10:45 +00002558 }
drhdb95f682013-06-26 22:46:00 +00002559 if( sCsv.cTerm==sCsv.cSeparator ){
2560 do{
2561 csv_read_one_field(&sCsv);
2562 i++;
2563 }while( sCsv.cTerm==sCsv.cSeparator );
2564 fprintf(stderr, "%s:%d: expected %d columns but found %d - "
2565 "extras ignored\n",
2566 sCsv.zFile, startLine, nCol, i);
drhfeac5f82004-08-01 00:10:45 +00002567 }
drhdb95f682013-06-26 22:46:00 +00002568 if( i>=nCol ){
2569 sqlite3_step(pStmt);
2570 rc = sqlite3_reset(pStmt);
2571 if( rc!=SQLITE_OK ){
2572 fprintf(stderr, "%s:%d: INSERT failed: %s\n", sCsv.zFile, startLine,
2573 sqlite3_errmsg(db));
2574 }
2575 }
2576 }while( sCsv.cTerm!=EOF );
2577
drh5bde8162013-06-27 14:07:53 +00002578 xCloser(sCsv.in);
drhdb95f682013-06-26 22:46:00 +00002579 sqlite3_free(sCsv.z);
drhfeac5f82004-08-01 00:10:45 +00002580 sqlite3_finalize(pStmt);
drh2d463112013-08-06 14:36:36 +00002581 if( needCommit ) sqlite3_exec(db, "COMMIT", 0, 0, 0);
drhfeac5f82004-08-01 00:10:45 +00002582 }else
2583
drhc2ce0be2014-05-29 12:36:14 +00002584 if( c=='i' && strncmp(azArg[0], "indices", n)==0 ){
drh75897232000-05-29 14:26:00 +00002585 struct callback_data data;
2586 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002587 open_db(p, 0);
drh75897232000-05-29 14:26:00 +00002588 memcpy(&data, p, sizeof(data));
2589 data.showHeader = 0;
2590 data.mode = MODE_List;
shane86f5bdb2009-10-24 02:00:07 +00002591 if( nArg==1 ){
2592 rc = sqlite3_exec(p->db,
2593 "SELECT name FROM sqlite_master "
2594 "WHERE type='index' AND name NOT LIKE 'sqlite_%' "
2595 "UNION ALL "
2596 "SELECT name FROM sqlite_temp_master "
2597 "WHERE type='index' "
2598 "ORDER BY 1",
2599 callback, &data, &zErrMsg
2600 );
drhc2ce0be2014-05-29 12:36:14 +00002601 }else if( nArg==2 ){
shane86f5bdb2009-10-24 02:00:07 +00002602 zShellStatic = azArg[1];
2603 rc = sqlite3_exec(p->db,
2604 "SELECT name FROM sqlite_master "
2605 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2606 "UNION ALL "
2607 "SELECT name FROM sqlite_temp_master "
2608 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2609 "ORDER BY 1",
2610 callback, &data, &zErrMsg
2611 );
2612 zShellStatic = 0;
drhc2ce0be2014-05-29 12:36:14 +00002613 }else{
2614 fprintf(stderr, "Usage: .indices ?LIKE-PATTERN?\n");
2615 rc = 1;
2616 goto meta_command_exit;
shane86f5bdb2009-10-24 02:00:07 +00002617 }
drh75897232000-05-29 14:26:00 +00002618 if( zErrMsg ){
2619 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002620 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002621 rc = 1;
shane86f5bdb2009-10-24 02:00:07 +00002622 }else if( rc != SQLITE_OK ){
2623 fprintf(stderr,"Error: querying sqlite_master and sqlite_temp_master\n");
2624 rc = 1;
drh75897232000-05-29 14:26:00 +00002625 }
2626 }else
2627
drhae5e4452007-05-03 17:18:36 +00002628#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +00002629 if( c=='i' && strncmp(azArg[0], "iotrace", n)==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002630 extern void (*sqlite3IoTrace)(const char*, ...);
drhb0603412007-02-28 04:47:26 +00002631 if( iotrace && iotrace!=stdout ) fclose(iotrace);
2632 iotrace = 0;
2633 if( nArg<2 ){
mlcreech3a00f902008-03-04 17:45:01 +00002634 sqlite3IoTrace = 0;
drhb0603412007-02-28 04:47:26 +00002635 }else if( strcmp(azArg[1], "-")==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002636 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002637 iotrace = stdout;
2638 }else{
2639 iotrace = fopen(azArg[1], "w");
2640 if( iotrace==0 ){
shane9bd1b442009-10-23 01:27:39 +00002641 fprintf(stderr, "Error: cannot open \"%s\"\n", azArg[1]);
mlcreech3a00f902008-03-04 17:45:01 +00002642 sqlite3IoTrace = 0;
shane9bd1b442009-10-23 01:27:39 +00002643 rc = 1;
drhb0603412007-02-28 04:47:26 +00002644 }else{
mlcreech3a00f902008-03-04 17:45:01 +00002645 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002646 }
2647 }
2648 }else
drhae5e4452007-05-03 17:18:36 +00002649#endif
drhb0603412007-02-28 04:47:26 +00002650
drh70df4fe2006-06-13 15:12:21 +00002651#ifndef SQLITE_OMIT_LOAD_EXTENSION
drhc2ce0be2014-05-29 12:36:14 +00002652 if( c=='l' && strncmp(azArg[0], "load", n)==0 ){
drh1e397f82006-06-08 15:28:43 +00002653 const char *zFile, *zProc;
2654 char *zErrMsg = 0;
drhc2ce0be2014-05-29 12:36:14 +00002655 if( nArg<2 ){
2656 fprintf(stderr, "Usage: .load FILE ?ENTRYPOINT?\n");
2657 rc = 1;
2658 goto meta_command_exit;
2659 }
drh1e397f82006-06-08 15:28:43 +00002660 zFile = azArg[1];
2661 zProc = nArg>=3 ? azArg[2] : 0;
drh05782482013-10-24 15:20:20 +00002662 open_db(p, 0);
drh1e397f82006-06-08 15:28:43 +00002663 rc = sqlite3_load_extension(p->db, zFile, zProc, &zErrMsg);
2664 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002665 fprintf(stderr, "Error: %s\n", zErrMsg);
drh1e397f82006-06-08 15:28:43 +00002666 sqlite3_free(zErrMsg);
drh47ad6842006-11-08 12:25:42 +00002667 rc = 1;
drh1e397f82006-06-08 15:28:43 +00002668 }
2669 }else
drh70df4fe2006-06-13 15:12:21 +00002670#endif
drh1e397f82006-06-08 15:28:43 +00002671
drhc2ce0be2014-05-29 12:36:14 +00002672 if( c=='l' && strncmp(azArg[0], "log", n)==0 ){
2673 if( nArg!=2 ){
2674 fprintf(stderr, "Usage: .log FILENAME\n");
2675 rc = 1;
2676 }else{
2677 const char *zFile = azArg[1];
2678 output_file_close(p->pLog);
2679 p->pLog = output_file_open(zFile);
2680 }
drh127f9d72010-02-23 01:47:00 +00002681 }else
2682
drhc2ce0be2014-05-29 12:36:14 +00002683 if( c=='m' && strncmp(azArg[0], "mode", n)==0 ){
2684 const char *zMode = nArg>=2 ? azArg[1] : "";
2685 int n2 = (int)strlen(zMode);
2686 int c2 = zMode[0];
2687 if( c2=='l' && n2>2 && strncmp(azArg[1],"lines",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002688 p->mode = MODE_Line;
drhc2ce0be2014-05-29 12:36:14 +00002689 }else if( c2=='c' && strncmp(azArg[1],"columns",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002690 p->mode = MODE_Column;
drhc2ce0be2014-05-29 12:36:14 +00002691 }else if( c2=='l' && n2>2 && strncmp(azArg[1],"list",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002692 p->mode = MODE_List;
drhc2ce0be2014-05-29 12:36:14 +00002693 }else if( c2=='h' && strncmp(azArg[1],"html",n2)==0 ){
drh1e5d0e92000-05-31 23:33:17 +00002694 p->mode = MODE_Html;
drhc2ce0be2014-05-29 12:36:14 +00002695 }else if( c2=='t' && strncmp(azArg[1],"tcl",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002696 p->mode = MODE_Tcl;
mistachkin585dcb22012-12-04 00:23:43 +00002697 sqlite3_snprintf(sizeof(p->separator), p->separator, " ");
drhc2ce0be2014-05-29 12:36:14 +00002698 }else if( c2=='c' && strncmp(azArg[1],"csv",n2)==0 ){
drh8e64d1c2004-10-07 00:32:39 +00002699 p->mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00002700 sqlite3_snprintf(sizeof(p->separator), p->separator, ",");
drhc2ce0be2014-05-29 12:36:14 +00002701 }else if( c2=='t' && strncmp(azArg[1],"tabs",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002702 p->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00002703 sqlite3_snprintf(sizeof(p->separator), p->separator, "\t");
drhc2ce0be2014-05-29 12:36:14 +00002704 }else if( c2=='i' && strncmp(azArg[1],"insert",n2)==0 ){
drh28bd4bc2000-06-15 15:57:22 +00002705 p->mode = MODE_Insert;
drhc2ce0be2014-05-29 12:36:14 +00002706 set_table_name(p, nArg>=3 ? azArg[2] : "table");
drhdaffd0e2001-04-11 14:28:42 +00002707 }else {
shane9bd1b442009-10-23 01:27:39 +00002708 fprintf(stderr,"Error: mode should be one of: "
drhfeac5f82004-08-01 00:10:45 +00002709 "column csv html insert line list tabs tcl\n");
shane9bd1b442009-10-23 01:27:39 +00002710 rc = 1;
drh75897232000-05-29 14:26:00 +00002711 }
2712 }else
2713
drhc2ce0be2014-05-29 12:36:14 +00002714 if( c=='n' && strncmp(azArg[0], "nullvalue", n)==0 ){
2715 if( nArg==2 ){
2716 sqlite3_snprintf(sizeof(p->nullvalue), p->nullvalue,
2717 "%.*s", (int)ArraySize(p->nullvalue)-1, azArg[1]);
2718 }else{
2719 fprintf(stderr, "Usage: .nullvalue STRING\n");
shanehe2aa9d72009-11-06 17:20:17 +00002720 rc = 1;
2721 }
2722 }else
2723
drh05782482013-10-24 15:20:20 +00002724 if( c=='o' && strncmp(azArg[0], "open", n)==0 && n>=2 ){
2725 sqlite3 *savedDb = p->db;
2726 const char *zSavedFilename = p->zDbFilename;
2727 char *zNewFilename = 0;
2728 p->db = 0;
2729 if( nArg>=2 ){
2730 p->zDbFilename = zNewFilename = sqlite3_mprintf("%s", azArg[1]);
2731 }
2732 open_db(p, 1);
2733 if( p->db!=0 ){
2734 sqlite3_close(savedDb);
2735 sqlite3_free(p->zFreeOnClose);
2736 p->zFreeOnClose = zNewFilename;
2737 }else{
2738 sqlite3_free(zNewFilename);
2739 p->db = savedDb;
2740 p->zDbFilename = zSavedFilename;
2741 }
2742 }else
2743
drhc2ce0be2014-05-29 12:36:14 +00002744 if( c=='o'
2745 && (strncmp(azArg[0], "output", n)==0 || strncmp(azArg[0], "once", n)==0)
2746 ){
2747 const char *zFile = nArg>=2 ? azArg[1] : "stdout";
2748 if( nArg>2 ){
2749 fprintf(stderr, "Usage: .%s FILE\n", azArg[0]);
2750 rc = 1;
2751 goto meta_command_exit;
drh75897232000-05-29 14:26:00 +00002752 }
drhc2ce0be2014-05-29 12:36:14 +00002753 if( n>1 && strncmp(azArg[0], "once", n)==0 ){
2754 if( nArg<2 ){
2755 fprintf(stderr, "Usage: .once FILE\n");
2756 rc = 1;
2757 goto meta_command_exit;
2758 }
2759 p->outCount = 2;
2760 }else{
2761 p->outCount = 0;
2762 }
2763 output_reset(p);
2764 if( zFile[0]=='|' ){
2765 p->out = popen(zFile + 1, "w");
drhe1da8fa2012-03-30 00:05:57 +00002766 if( p->out==0 ){
drhc2ce0be2014-05-29 12:36:14 +00002767 fprintf(stderr,"Error: cannot open pipe \"%s\"\n", zFile + 1);
drhe1da8fa2012-03-30 00:05:57 +00002768 p->out = stdout;
2769 rc = 1;
2770 }else{
drhc2ce0be2014-05-29 12:36:14 +00002771 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
drhe1da8fa2012-03-30 00:05:57 +00002772 }
drh75897232000-05-29 14:26:00 +00002773 }else{
drhc2ce0be2014-05-29 12:36:14 +00002774 p->out = output_file_open(zFile);
drh75897232000-05-29 14:26:00 +00002775 if( p->out==0 ){
drhc2ce0be2014-05-29 12:36:14 +00002776 if( strcmp(zFile,"off")!=0 ){
2777 fprintf(stderr,"Error: cannot write to \"%s\"\n", zFile);
drh42f64e52012-04-04 16:56:23 +00002778 }
drh75897232000-05-29 14:26:00 +00002779 p->out = stdout;
shane9bd1b442009-10-23 01:27:39 +00002780 rc = 1;
persicom7e2dfdd2002-04-18 02:46:52 +00002781 } else {
drhc2ce0be2014-05-29 12:36:14 +00002782 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
drh75897232000-05-29 14:26:00 +00002783 }
2784 }
2785 }else
2786
drh078b1fd2012-09-21 13:40:02 +00002787 if( c=='p' && n>=3 && strncmp(azArg[0], "print", n)==0 ){
2788 int i;
2789 for(i=1; i<nArg; i++){
2790 if( i>1 ) fprintf(p->out, " ");
2791 fprintf(p->out, "%s", azArg[i]);
2792 }
2793 fprintf(p->out, "\n");
2794 }else
2795
drhc2ce0be2014-05-29 12:36:14 +00002796 if( c=='p' && strncmp(azArg[0], "prompt", n)==0 ){
persicom7e2dfdd2002-04-18 02:46:52 +00002797 if( nArg >= 2) {
2798 strncpy(mainPrompt,azArg[1],(int)ArraySize(mainPrompt)-1);
2799 }
2800 if( nArg >= 3) {
2801 strncpy(continuePrompt,azArg[2],(int)ArraySize(continuePrompt)-1);
2802 }
2803 }else
2804
drhc2ce0be2014-05-29 12:36:14 +00002805 if( c=='q' && strncmp(azArg[0], "quit", n)==0 ){
drh47ad6842006-11-08 12:25:42 +00002806 rc = 2;
persicom7e2dfdd2002-04-18 02:46:52 +00002807 }else
2808
drhc2ce0be2014-05-29 12:36:14 +00002809 if( c=='r' && n>=3 && strncmp(azArg[0], "read", n)==0 ){
2810 FILE *alt;
2811 if( nArg!=2 ){
2812 fprintf(stderr, "Usage: .read FILE\n");
2813 rc = 1;
2814 goto meta_command_exit;
2815 }
2816 alt = fopen(azArg[1], "rb");
drhdaffd0e2001-04-11 14:28:42 +00002817 if( alt==0 ){
shane9bd1b442009-10-23 01:27:39 +00002818 fprintf(stderr,"Error: cannot open \"%s\"\n", azArg[1]);
2819 rc = 1;
drhdaffd0e2001-04-11 14:28:42 +00002820 }else{
shane9bd1b442009-10-23 01:27:39 +00002821 rc = process_input(p, alt);
drhdaffd0e2001-04-11 14:28:42 +00002822 fclose(alt);
2823 }
2824 }else
2825
drhc2ce0be2014-05-29 12:36:14 +00002826 if( c=='r' && n>=3 && strncmp(azArg[0], "restore", n)==0 ){
drh9ff849f2009-02-04 20:55:57 +00002827 const char *zSrcFile;
2828 const char *zDb;
2829 sqlite3 *pSrc;
2830 sqlite3_backup *pBackup;
drhdc2c4912009-02-04 22:46:47 +00002831 int nTimeout = 0;
2832
drh9ff849f2009-02-04 20:55:57 +00002833 if( nArg==2 ){
2834 zSrcFile = azArg[1];
2835 zDb = "main";
drhc2ce0be2014-05-29 12:36:14 +00002836 }else if( nArg==3 ){
drh9ff849f2009-02-04 20:55:57 +00002837 zSrcFile = azArg[2];
2838 zDb = azArg[1];
drhc2ce0be2014-05-29 12:36:14 +00002839 }else{
2840 fprintf(stderr, "Usage: .restore ?DB? FILE\n");
2841 rc = 1;
2842 goto meta_command_exit;
drh9ff849f2009-02-04 20:55:57 +00002843 }
2844 rc = sqlite3_open(zSrcFile, &pSrc);
2845 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002846 fprintf(stderr, "Error: cannot open \"%s\"\n", zSrcFile);
drh9ff849f2009-02-04 20:55:57 +00002847 sqlite3_close(pSrc);
2848 return 1;
2849 }
drh05782482013-10-24 15:20:20 +00002850 open_db(p, 0);
drh9ff849f2009-02-04 20:55:57 +00002851 pBackup = sqlite3_backup_init(p->db, zDb, pSrc, "main");
2852 if( pBackup==0 ){
2853 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
2854 sqlite3_close(pSrc);
2855 return 1;
2856 }
drhdc2c4912009-02-04 22:46:47 +00002857 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK
2858 || rc==SQLITE_BUSY ){
2859 if( rc==SQLITE_BUSY ){
2860 if( nTimeout++ >= 3 ) break;
2861 sqlite3_sleep(100);
drh9ff849f2009-02-04 20:55:57 +00002862 }
2863 }
2864 sqlite3_backup_finish(pBackup);
2865 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00002866 rc = 0;
drhdc2c4912009-02-04 22:46:47 +00002867 }else if( rc==SQLITE_BUSY || rc==SQLITE_LOCKED ){
shane9bd1b442009-10-23 01:27:39 +00002868 fprintf(stderr, "Error: source database is busy\n");
2869 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002870 }else{
2871 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
shane9bd1b442009-10-23 01:27:39 +00002872 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002873 }
2874 sqlite3_close(pSrc);
2875 }else
2876
drhc2ce0be2014-05-29 12:36:14 +00002877 if( c=='s' && strncmp(azArg[0], "schema", n)==0 ){
drh75897232000-05-29 14:26:00 +00002878 struct callback_data data;
2879 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002880 open_db(p, 0);
drh75897232000-05-29 14:26:00 +00002881 memcpy(&data, p, sizeof(data));
2882 data.showHeader = 0;
drhe3710332000-09-29 13:30:53 +00002883 data.mode = MODE_Semi;
drhc2ce0be2014-05-29 12:36:14 +00002884 if( nArg==2 ){
drhc8d74412004-08-31 23:41:26 +00002885 int i;
drhf0693c82011-10-11 20:41:54 +00002886 for(i=0; azArg[1][i]; i++) azArg[1][i] = ToLower(azArg[1][i]);
drhc8d74412004-08-31 23:41:26 +00002887 if( strcmp(azArg[1],"sqlite_master")==0 ){
drha18c5682000-10-08 22:20:57 +00002888 char *new_argv[2], *new_colv[2];
2889 new_argv[0] = "CREATE TABLE sqlite_master (\n"
2890 " type text,\n"
2891 " name text,\n"
2892 " tbl_name text,\n"
drhadbca9c2001-09-27 15:11:53 +00002893 " rootpage integer,\n"
drha18c5682000-10-08 22:20:57 +00002894 " sql text\n"
2895 ")";
2896 new_argv[1] = 0;
2897 new_colv[0] = "sql";
2898 new_colv[1] = 0;
2899 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002900 rc = SQLITE_OK;
drhc8d74412004-08-31 23:41:26 +00002901 }else if( strcmp(azArg[1],"sqlite_temp_master")==0 ){
drhe0bc4042002-06-25 01:09:11 +00002902 char *new_argv[2], *new_colv[2];
2903 new_argv[0] = "CREATE TEMP TABLE sqlite_temp_master (\n"
2904 " type text,\n"
2905 " name text,\n"
2906 " tbl_name text,\n"
2907 " rootpage integer,\n"
2908 " sql text\n"
2909 ")";
2910 new_argv[1] = 0;
2911 new_colv[0] = "sql";
2912 new_colv[1] = 0;
2913 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002914 rc = SQLITE_OK;
drha18c5682000-10-08 22:20:57 +00002915 }else{
danielk1977bc6ada42004-06-30 08:20:16 +00002916 zShellStatic = azArg[1];
shane9bd1b442009-10-23 01:27:39 +00002917 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002918 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002919 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002920 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002921 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh6ac7a582011-11-04 00:35:56 +00002922 "WHERE lower(tbl_name) LIKE shellstatic()"
2923 " AND type!='meta' AND sql NOTNULL "
drh1ba00292013-05-06 21:01:06 +00002924 "ORDER BY rowid",
danielk1977bc6ada42004-06-30 08:20:16 +00002925 callback, &data, &zErrMsg);
2926 zShellStatic = 0;
drha18c5682000-10-08 22:20:57 +00002927 }
drhc2ce0be2014-05-29 12:36:14 +00002928 }else if( nArg==1 ){
shane9bd1b442009-10-23 01:27:39 +00002929 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002930 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002931 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002932 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002933 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh0c356672005-09-10 22:40:53 +00002934 "WHERE type!='meta' AND sql NOTNULL AND name NOT LIKE 'sqlite_%'"
drh1ba00292013-05-06 21:01:06 +00002935 "ORDER BY rowid",
drha18c5682000-10-08 22:20:57 +00002936 callback, &data, &zErrMsg
2937 );
drhc2ce0be2014-05-29 12:36:14 +00002938 }else{
2939 fprintf(stderr, "Usage: .schema ?LIKE-PATTERN?\n");
2940 rc = 1;
2941 goto meta_command_exit;
drh75897232000-05-29 14:26:00 +00002942 }
drh75897232000-05-29 14:26:00 +00002943 if( zErrMsg ){
2944 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002945 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002946 rc = 1;
2947 }else if( rc != SQLITE_OK ){
2948 fprintf(stderr,"Error: querying schema information\n");
2949 rc = 1;
2950 }else{
2951 rc = 0;
drh75897232000-05-29 14:26:00 +00002952 }
2953 }else
2954
drh340f5822013-06-27 13:01:21 +00002955#ifdef SQLITE_DEBUG
drh348d19c2013-06-03 12:47:43 +00002956 /* Undocumented commands for internal testing. Subject to change
2957 ** without notice. */
2958 if( c=='s' && n>=10 && strncmp(azArg[0], "selftest-", 9)==0 ){
2959 if( strncmp(azArg[0]+9, "boolean", n-9)==0 ){
2960 int i, v;
2961 for(i=1; i<nArg; i++){
2962 v = booleanValue(azArg[i]);
2963 fprintf(p->out, "%s: %d 0x%x\n", azArg[i], v, v);
2964 }
2965 }
2966 if( strncmp(azArg[0]+9, "integer", n-9)==0 ){
2967 int i; sqlite3_int64 v;
2968 for(i=1; i<nArg; i++){
drh340f5822013-06-27 13:01:21 +00002969 char zBuf[200];
drh348d19c2013-06-03 12:47:43 +00002970 v = integerValue(azArg[i]);
drhc2ce0be2014-05-29 12:36:14 +00002971 sqlite3_snprintf(sizeof(zBuf),zBuf,"%s: %lld 0x%llx\n", azArg[i],v,v);
drh340f5822013-06-27 13:01:21 +00002972 fprintf(p->out, "%s", zBuf);
drh348d19c2013-06-03 12:47:43 +00002973 }
2974 }
2975 }else
drh340f5822013-06-27 13:01:21 +00002976#endif
drh348d19c2013-06-03 12:47:43 +00002977
drhc2ce0be2014-05-29 12:36:14 +00002978 if( c=='s' && strncmp(azArg[0], "separator", n)==0 ){
2979 if( nArg==2 ){
2980 sqlite3_snprintf(sizeof(p->separator), p->separator,
2981 "%.*s", (int)sizeof(p->separator)-1, azArg[1]);
2982 }else{
2983 fprintf(stderr, "Usage: .separator STRING\n");
2984 rc = 1;
2985 }
drh75897232000-05-29 14:26:00 +00002986 }else
2987
drh62cdde52014-05-28 20:22:28 +00002988 if( c=='s'
2989 && (strncmp(azArg[0], "shell", n)==0 || strncmp(azArg[0],"system",n)==0)
drh62cdde52014-05-28 20:22:28 +00002990 ){
2991 char *zCmd;
2992 int i;
drhc2ce0be2014-05-29 12:36:14 +00002993 if( nArg<2 ){
2994 fprintf(stderr, "Usage: .system COMMAND\n");
2995 rc = 1;
2996 goto meta_command_exit;
2997 }
drhdcb3e3d2014-05-29 03:17:29 +00002998 zCmd = sqlite3_mprintf(strchr(azArg[1],' ')==0?"%s":"\"%s\"", azArg[1]);
drh62cdde52014-05-28 20:22:28 +00002999 for(i=2; i<nArg; i++){
drhdcb3e3d2014-05-29 03:17:29 +00003000 zCmd = sqlite3_mprintf(strchr(azArg[i],' ')==0?"%z %s":"%z \"%s\"",
3001 zCmd, azArg[i]);
drh62cdde52014-05-28 20:22:28 +00003002 }
drh3c4461f2014-05-29 20:39:59 +00003003 (void)system(zCmd);
drh62cdde52014-05-28 20:22:28 +00003004 sqlite3_free(zCmd);
3005 }else
3006
drhc2ce0be2014-05-29 12:36:14 +00003007 if( c=='s' && strncmp(azArg[0], "show", n)==0 ){
persicom7e2dfdd2002-04-18 02:46:52 +00003008 int i;
drhc2ce0be2014-05-29 12:36:14 +00003009 if( nArg!=1 ){
3010 fprintf(stderr, "Usage: .show\n");
3011 rc = 1;
3012 goto meta_command_exit;
3013 }
persicom7e2dfdd2002-04-18 02:46:52 +00003014 fprintf(p->out,"%9.9s: %s\n","echo", p->echoOn ? "on" : "off");
drhefbf3b12014-02-28 20:47:24 +00003015 fprintf(p->out,"%9.9s: %s\n","eqp", p->autoEQP ? "on" : "off");
drh67505e72002-04-19 12:34:06 +00003016 fprintf(p->out,"%9.9s: %s\n","explain", p->explainPrev.valid ? "on" :"off");
drhdd45df82002-04-18 12:39:03 +00003017 fprintf(p->out,"%9.9s: %s\n","headers", p->showHeader ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00003018 fprintf(p->out,"%9.9s: %s\n","mode", modeDescr[p->mode]);
drhfeac5f82004-08-01 00:10:45 +00003019 fprintf(p->out,"%9.9s: ", "nullvalue");
3020 output_c_string(p->out, p->nullvalue);
3021 fprintf(p->out, "\n");
drh67505e72002-04-19 12:34:06 +00003022 fprintf(p->out,"%9.9s: %s\n","output",
drh4f21c4a2008-12-10 22:15:00 +00003023 strlen30(p->outfile) ? p->outfile : "stdout");
drhfeac5f82004-08-01 00:10:45 +00003024 fprintf(p->out,"%9.9s: ", "separator");
3025 output_c_string(p->out, p->separator);
3026 fprintf(p->out, "\n");
shaneh642d8b82010-07-28 16:05:34 +00003027 fprintf(p->out,"%9.9s: %s\n","stats", p->statsOn ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00003028 fprintf(p->out,"%9.9s: ","width");
3029 for (i=0;i<(int)ArraySize(p->colWidth) && p->colWidth[i] != 0;i++) {
drhfeac5f82004-08-01 00:10:45 +00003030 fprintf(p->out,"%d ",p->colWidth[i]);
persicom7e2dfdd2002-04-18 02:46:52 +00003031 }
drhfeac5f82004-08-01 00:10:45 +00003032 fprintf(p->out,"\n");
persicom7e2dfdd2002-04-18 02:46:52 +00003033 }else
3034
drhc2ce0be2014-05-29 12:36:14 +00003035 if( c=='s' && strncmp(azArg[0], "stats", n)==0 ){
3036 if( nArg==2 ){
3037 p->statsOn = booleanValue(azArg[1]);
3038 }else{
3039 fprintf(stderr, "Usage: .stats on|off\n");
3040 rc = 1;
3041 }
shaneh642d8b82010-07-28 16:05:34 +00003042 }else
3043
drhc2ce0be2014-05-29 12:36:14 +00003044 if( c=='t' && n>1 && strncmp(azArg[0], "tables", n)==0 ){
drh98781232012-04-23 12:38:05 +00003045 sqlite3_stmt *pStmt;
drhe3710332000-09-29 13:30:53 +00003046 char **azResult;
drh98781232012-04-23 12:38:05 +00003047 int nRow, nAlloc;
3048 char *zSql = 0;
3049 int ii;
drh05782482013-10-24 15:20:20 +00003050 open_db(p, 0);
drh98781232012-04-23 12:38:05 +00003051 rc = sqlite3_prepare_v2(p->db, "PRAGMA database_list", -1, &pStmt, 0);
3052 if( rc ) return rc;
3053 zSql = sqlite3_mprintf(
3054 "SELECT name FROM sqlite_master"
3055 " WHERE type IN ('table','view')"
3056 " AND name NOT LIKE 'sqlite_%%'"
3057 " AND name LIKE ?1");
3058 while( sqlite3_step(pStmt)==SQLITE_ROW ){
3059 const char *zDbName = (const char*)sqlite3_column_text(pStmt, 1);
3060 if( zDbName==0 || strcmp(zDbName,"main")==0 ) continue;
3061 if( strcmp(zDbName,"temp")==0 ){
3062 zSql = sqlite3_mprintf(
3063 "%z UNION ALL "
3064 "SELECT 'temp.' || name FROM sqlite_temp_master"
3065 " WHERE type IN ('table','view')"
3066 " AND name NOT LIKE 'sqlite_%%'"
3067 " AND name LIKE ?1", zSql);
3068 }else{
3069 zSql = sqlite3_mprintf(
3070 "%z UNION ALL "
3071 "SELECT '%q.' || name FROM \"%w\".sqlite_master"
3072 " WHERE type IN ('table','view')"
3073 " AND name NOT LIKE 'sqlite_%%'"
3074 " AND name LIKE ?1", zSql, zDbName, zDbName);
3075 }
drha50da102000-08-08 20:19:09 +00003076 }
drh98781232012-04-23 12:38:05 +00003077 sqlite3_finalize(pStmt);
3078 zSql = sqlite3_mprintf("%z ORDER BY 1", zSql);
3079 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
3080 sqlite3_free(zSql);
3081 if( rc ) return rc;
3082 nRow = nAlloc = 0;
3083 azResult = 0;
3084 if( nArg>1 ){
3085 sqlite3_bind_text(pStmt, 1, azArg[1], -1, SQLITE_TRANSIENT);
shane9bd1b442009-10-23 01:27:39 +00003086 }else{
drh98781232012-04-23 12:38:05 +00003087 sqlite3_bind_text(pStmt, 1, "%", -1, SQLITE_STATIC);
3088 }
3089 while( sqlite3_step(pStmt)==SQLITE_ROW ){
3090 if( nRow>=nAlloc ){
3091 char **azNew;
3092 int n = nAlloc*2 + 10;
3093 azNew = sqlite3_realloc(azResult, sizeof(azResult[0])*n);
3094 if( azNew==0 ){
3095 fprintf(stderr, "Error: out of memory\n");
3096 break;
3097 }
3098 nAlloc = n;
3099 azResult = azNew;
3100 }
3101 azResult[nRow] = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
3102 if( azResult[nRow] ) nRow++;
3103 }
3104 sqlite3_finalize(pStmt);
3105 if( nRow>0 ){
drhe3710332000-09-29 13:30:53 +00003106 int len, maxlen = 0;
3107 int i, j;
3108 int nPrintCol, nPrintRow;
drh98781232012-04-23 12:38:05 +00003109 for(i=0; i<nRow; i++){
drh4f21c4a2008-12-10 22:15:00 +00003110 len = strlen30(azResult[i]);
drhe3710332000-09-29 13:30:53 +00003111 if( len>maxlen ) maxlen = len;
3112 }
3113 nPrintCol = 80/(maxlen+2);
3114 if( nPrintCol<1 ) nPrintCol = 1;
3115 nPrintRow = (nRow + nPrintCol - 1)/nPrintCol;
3116 for(i=0; i<nPrintRow; i++){
drh98781232012-04-23 12:38:05 +00003117 for(j=i; j<nRow; j+=nPrintRow){
3118 char *zSp = j<nPrintRow ? "" : " ";
drh151b7d52013-05-06 20:28:54 +00003119 fprintf(p->out, "%s%-*s", zSp, maxlen, azResult[j] ? azResult[j] : "");
drhe3710332000-09-29 13:30:53 +00003120 }
drh151b7d52013-05-06 20:28:54 +00003121 fprintf(p->out, "\n");
drhe3710332000-09-29 13:30:53 +00003122 }
3123 }
drh98781232012-04-23 12:38:05 +00003124 for(ii=0; ii<nRow; ii++) sqlite3_free(azResult[ii]);
3125 sqlite3_free(azResult);
drh75897232000-05-29 14:26:00 +00003126 }else
3127
shaneh96887e12011-02-10 21:08:58 +00003128 if( c=='t' && n>=8 && strncmp(azArg[0], "testctrl", n)==0 && nArg>=2 ){
drhd416fe72011-03-17 16:45:50 +00003129 static const struct {
3130 const char *zCtrlName; /* Name of a test-control option */
3131 int ctrlCode; /* Integer code for that option */
3132 } aCtrl[] = {
3133 { "prng_save", SQLITE_TESTCTRL_PRNG_SAVE },
3134 { "prng_restore", SQLITE_TESTCTRL_PRNG_RESTORE },
3135 { "prng_reset", SQLITE_TESTCTRL_PRNG_RESET },
3136 { "bitvec_test", SQLITE_TESTCTRL_BITVEC_TEST },
3137 { "fault_install", SQLITE_TESTCTRL_FAULT_INSTALL },
3138 { "benign_malloc_hooks", SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS },
3139 { "pending_byte", SQLITE_TESTCTRL_PENDING_BYTE },
3140 { "assert", SQLITE_TESTCTRL_ASSERT },
3141 { "always", SQLITE_TESTCTRL_ALWAYS },
3142 { "reserve", SQLITE_TESTCTRL_RESERVE },
3143 { "optimizations", SQLITE_TESTCTRL_OPTIMIZATIONS },
3144 { "iskeyword", SQLITE_TESTCTRL_ISKEYWORD },
drhd416fe72011-03-17 16:45:50 +00003145 { "scratchmalloc", SQLITE_TESTCTRL_SCRATCHMALLOC },
drh2cf4acb2014-04-18 00:06:02 +00003146 { "byteorder", SQLITE_TESTCTRL_BYTEORDER },
drhd416fe72011-03-17 16:45:50 +00003147 };
shaneh96887e12011-02-10 21:08:58 +00003148 int testctrl = -1;
3149 int rc = 0;
drhd416fe72011-03-17 16:45:50 +00003150 int i, n;
drh05782482013-10-24 15:20:20 +00003151 open_db(p, 0);
shaneh96887e12011-02-10 21:08:58 +00003152
drhd416fe72011-03-17 16:45:50 +00003153 /* convert testctrl text option to value. allow any unique prefix
3154 ** of the option name, or a numerical value. */
shanehcef83682011-04-07 03:41:01 +00003155 n = strlen30(azArg[1]);
drhfcd71b62011-04-05 22:08:24 +00003156 for(i=0; i<(int)(sizeof(aCtrl)/sizeof(aCtrl[0])); i++){
drhd416fe72011-03-17 16:45:50 +00003157 if( strncmp(azArg[1], aCtrl[i].zCtrlName, n)==0 ){
3158 if( testctrl<0 ){
3159 testctrl = aCtrl[i].ctrlCode;
3160 }else{
drhb07028f2011-10-14 21:49:18 +00003161 fprintf(stderr, "ambiguous option name: \"%s\"\n", azArg[1]);
drhd416fe72011-03-17 16:45:50 +00003162 testctrl = -1;
3163 break;
3164 }
3165 }
3166 }
drh348d19c2013-06-03 12:47:43 +00003167 if( testctrl<0 ) testctrl = (int)integerValue(azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003168 if( (testctrl<SQLITE_TESTCTRL_FIRST) || (testctrl>SQLITE_TESTCTRL_LAST) ){
3169 fprintf(stderr,"Error: invalid testctrl option: %s\n", azArg[1]);
3170 }else{
3171 switch(testctrl){
3172
3173 /* sqlite3_test_control(int, db, int) */
3174 case SQLITE_TESTCTRL_OPTIMIZATIONS:
3175 case SQLITE_TESTCTRL_RESERVE:
3176 if( nArg==3 ){
3177 int opt = (int)strtol(azArg[2], 0, 0);
3178 rc = sqlite3_test_control(testctrl, p->db, opt);
drh151b7d52013-05-06 20:28:54 +00003179 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003180 } else {
drhd416fe72011-03-17 16:45:50 +00003181 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
3182 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003183 }
3184 break;
3185
3186 /* sqlite3_test_control(int) */
drh2cf4acb2014-04-18 00:06:02 +00003187 case SQLITE_TESTCTRL_PRNG_SAVE:
3188 case SQLITE_TESTCTRL_PRNG_RESTORE:
shaneh96887e12011-02-10 21:08:58 +00003189 case SQLITE_TESTCTRL_PRNG_RESET:
drh2cf4acb2014-04-18 00:06:02 +00003190 case SQLITE_TESTCTRL_BYTEORDER:
shaneh96887e12011-02-10 21:08:58 +00003191 if( nArg==2 ){
3192 rc = sqlite3_test_control(testctrl);
drh151b7d52013-05-06 20:28:54 +00003193 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003194 } else {
3195 fprintf(stderr,"Error: testctrl %s takes no options\n", azArg[1]);
3196 }
3197 break;
3198
3199 /* sqlite3_test_control(int, uint) */
3200 case SQLITE_TESTCTRL_PENDING_BYTE:
3201 if( nArg==3 ){
drhaf664332013-07-18 20:28:29 +00003202 unsigned int opt = (unsigned int)integerValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00003203 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003204 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003205 } else {
drhd416fe72011-03-17 16:45:50 +00003206 fprintf(stderr,"Error: testctrl %s takes a single unsigned"
3207 " int option\n", azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003208 }
3209 break;
3210
3211 /* sqlite3_test_control(int, int) */
3212 case SQLITE_TESTCTRL_ASSERT:
3213 case SQLITE_TESTCTRL_ALWAYS:
3214 if( nArg==3 ){
drh348d19c2013-06-03 12:47:43 +00003215 int opt = booleanValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00003216 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003217 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003218 } else {
drhd416fe72011-03-17 16:45:50 +00003219 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
3220 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003221 }
3222 break;
3223
3224 /* sqlite3_test_control(int, char *) */
3225#ifdef SQLITE_N_KEYWORD
3226 case SQLITE_TESTCTRL_ISKEYWORD:
3227 if( nArg==3 ){
3228 const char *opt = azArg[2];
3229 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003230 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003231 } else {
drhd416fe72011-03-17 16:45:50 +00003232 fprintf(stderr,"Error: testctrl %s takes a single char * option\n",
3233 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003234 }
3235 break;
3236#endif
3237
3238 case SQLITE_TESTCTRL_BITVEC_TEST:
3239 case SQLITE_TESTCTRL_FAULT_INSTALL:
3240 case SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS:
3241 case SQLITE_TESTCTRL_SCRATCHMALLOC:
3242 default:
drhd416fe72011-03-17 16:45:50 +00003243 fprintf(stderr,"Error: CLI support for testctrl %s not implemented\n",
3244 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003245 break;
3246 }
3247 }
3248 }else
3249
drhc2ce0be2014-05-29 12:36:14 +00003250 if( c=='t' && n>4 && strncmp(azArg[0], "timeout", n)==0 ){
drh05782482013-10-24 15:20:20 +00003251 open_db(p, 0);
drhc2ce0be2014-05-29 12:36:14 +00003252 sqlite3_busy_timeout(p->db, nArg>=2 ? (int)integerValue(azArg[1]) : 0);
shanehe2aa9d72009-11-06 17:20:17 +00003253 }else
3254
drhc2ce0be2014-05-29 12:36:14 +00003255 if( c=='t' && n>=5 && strncmp(azArg[0], "timer", n)==0 ){
3256 if( nArg==2 ){
3257 enableTimer = booleanValue(azArg[1]);
3258 if( enableTimer && !HAS_TIMER ){
3259 fprintf(stderr, "Error: timer not available on this system.\n");
3260 enableTimer = 0;
3261 }
3262 }else{
3263 fprintf(stderr, "Usage: .timer on|off\n");
3264 rc = 1;
3265 }
shanehe2aa9d72009-11-06 17:20:17 +00003266 }else
3267
drhc2ce0be2014-05-29 12:36:14 +00003268 if( c=='t' && strncmp(azArg[0], "trace", n)==0 ){
drh05782482013-10-24 15:20:20 +00003269 open_db(p, 0);
drh42f64e52012-04-04 16:56:23 +00003270 output_file_close(p->traceOut);
drhc2ce0be2014-05-29 12:36:14 +00003271 if( nArg!=2 ){
3272 fprintf(stderr, "Usage: .trace FILE|off\n");
3273 rc = 1;
3274 goto meta_command_exit;
3275 }
drh42f64e52012-04-04 16:56:23 +00003276 p->traceOut = output_file_open(azArg[1]);
drhbbb0be82012-06-27 16:12:27 +00003277#if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT)
drh42f64e52012-04-04 16:56:23 +00003278 if( p->traceOut==0 ){
3279 sqlite3_trace(p->db, 0, 0);
3280 }else{
3281 sqlite3_trace(p->db, sql_trace_callback, p->traceOut);
3282 }
3283#endif
3284 }else
3285
drh9fd301b2011-06-03 13:28:22 +00003286 if( c=='v' && strncmp(azArg[0], "version", n)==0 ){
drh151b7d52013-05-06 20:28:54 +00003287 fprintf(p->out, "SQLite %s %s\n" /*extra-version-info*/,
drh9fd301b2011-06-03 13:28:22 +00003288 sqlite3_libversion(), sqlite3_sourceid());
3289 }else
3290
drhde60fc22011-12-14 17:53:36 +00003291 if( c=='v' && strncmp(azArg[0], "vfsname", n)==0 ){
3292 const char *zDbName = nArg==2 ? azArg[1] : "main";
3293 char *zVfsName = 0;
3294 if( p->db ){
3295 sqlite3_file_control(p->db, zDbName, SQLITE_FCNTL_VFSNAME, &zVfsName);
3296 if( zVfsName ){
drh151b7d52013-05-06 20:28:54 +00003297 fprintf(p->out, "%s\n", zVfsName);
drhde60fc22011-12-14 17:53:36 +00003298 sqlite3_free(zVfsName);
3299 }
3300 }
3301 }else
3302
drhcef4fc82012-09-21 22:50:45 +00003303#if defined(SQLITE_DEBUG) && defined(SQLITE_ENABLE_WHERETRACE)
3304 if( c=='w' && strncmp(azArg[0], "wheretrace", n)==0 ){
3305 extern int sqlite3WhereTrace;
drhc2ce0be2014-05-29 12:36:14 +00003306 sqlite3WhereTrace = nArg>=2 ? booleanValue(azArg[1]) : 0xff;
drhcef4fc82012-09-21 22:50:45 +00003307 }else
3308#endif
3309
drhc2ce0be2014-05-29 12:36:14 +00003310 if( c=='w' && strncmp(azArg[0], "width", n)==0 ){
drh75897232000-05-29 14:26:00 +00003311 int j;
drh43617e92006-03-06 20:55:46 +00003312 assert( nArg<=ArraySize(azArg) );
drh75897232000-05-29 14:26:00 +00003313 for(j=1; j<nArg && j<ArraySize(p->colWidth); j++){
drh348d19c2013-06-03 12:47:43 +00003314 p->colWidth[j-1] = (int)integerValue(azArg[j]);
drh75897232000-05-29 14:26:00 +00003315 }
3316 }else
3317
3318 {
shane9bd1b442009-10-23 01:27:39 +00003319 fprintf(stderr, "Error: unknown command or invalid arguments: "
drh67505e72002-04-19 12:34:06 +00003320 " \"%s\". Enter \".help\" for help\n", azArg[0]);
shane9bd1b442009-10-23 01:27:39 +00003321 rc = 1;
drh75897232000-05-29 14:26:00 +00003322 }
drh67505e72002-04-19 12:34:06 +00003323
drhc2ce0be2014-05-29 12:36:14 +00003324meta_command_exit:
3325 if( p->outCount ){
3326 p->outCount--;
3327 if( p->outCount==0 ) output_reset(p);
3328 }
drh67505e72002-04-19 12:34:06 +00003329 return rc;
drh75897232000-05-29 14:26:00 +00003330}
3331
drh67505e72002-04-19 12:34:06 +00003332/*
drh91a66392007-09-07 01:12:32 +00003333** Return TRUE if a semicolon occurs anywhere in the first N characters
3334** of string z[].
drh324ccef2003-02-05 14:06:20 +00003335*/
drh9f099fd2013-08-06 14:01:46 +00003336static int line_contains_semicolon(const char *z, int N){
drh91a66392007-09-07 01:12:32 +00003337 int i;
3338 for(i=0; i<N; i++){ if( z[i]==';' ) return 1; }
3339 return 0;
drh324ccef2003-02-05 14:06:20 +00003340}
3341
3342/*
drh70c7a4b2003-04-26 03:03:06 +00003343** Test to see if a line consists entirely of whitespace.
3344*/
3345static int _all_whitespace(const char *z){
3346 for(; *z; z++){
drhf0693c82011-10-11 20:41:54 +00003347 if( IsSpace(z[0]) ) continue;
drh70c7a4b2003-04-26 03:03:06 +00003348 if( *z=='/' && z[1]=='*' ){
3349 z += 2;
3350 while( *z && (*z!='*' || z[1]!='/') ){ z++; }
3351 if( *z==0 ) return 0;
3352 z++;
3353 continue;
3354 }
3355 if( *z=='-' && z[1]=='-' ){
3356 z += 2;
3357 while( *z && *z!='\n' ){ z++; }
3358 if( *z==0 ) return 1;
3359 continue;
3360 }
3361 return 0;
3362 }
3363 return 1;
3364}
3365
3366/*
drha9b17162003-04-29 18:01:28 +00003367** Return TRUE if the line typed in is an SQL command terminator other
3368** than a semi-colon. The SQL Server style "go" command is understood
3369** as is the Oracle "/".
3370*/
drh9f099fd2013-08-06 14:01:46 +00003371static int line_is_command_terminator(const char *zLine){
drhf0693c82011-10-11 20:41:54 +00003372 while( IsSpace(zLine[0]) ){ zLine++; };
drh233a5312008-12-18 22:25:13 +00003373 if( zLine[0]=='/' && _all_whitespace(&zLine[1]) ){
3374 return 1; /* Oracle */
3375 }
drhf0693c82011-10-11 20:41:54 +00003376 if( ToLower(zLine[0])=='g' && ToLower(zLine[1])=='o'
drhc8d74412004-08-31 23:41:26 +00003377 && _all_whitespace(&zLine[2]) ){
drha9b17162003-04-29 18:01:28 +00003378 return 1; /* SQL Server */
3379 }
3380 return 0;
3381}
3382
3383/*
drh233a5312008-12-18 22:25:13 +00003384** Return true if zSql is a complete SQL statement. Return false if it
3385** ends in the middle of a string literal or C-style comment.
3386*/
drh9f099fd2013-08-06 14:01:46 +00003387static int line_is_complete(char *zSql, int nSql){
drh233a5312008-12-18 22:25:13 +00003388 int rc;
3389 if( zSql==0 ) return 1;
3390 zSql[nSql] = ';';
3391 zSql[nSql+1] = 0;
3392 rc = sqlite3_complete(zSql);
3393 zSql[nSql] = 0;
3394 return rc;
3395}
3396
3397/*
drh67505e72002-04-19 12:34:06 +00003398** Read input from *in and process it. If *in==0 then input
3399** is interactive - the user is typing it it. Otherwise, input
3400** is coming from a file or device. A prompt is issued and history
3401** is saved only if input is interactive. An interrupt signal will
3402** cause this routine to exit immediately, unless input is interactive.
drhc28490c2006-10-26 14:25:58 +00003403**
3404** Return the number of errors.
drh67505e72002-04-19 12:34:06 +00003405*/
drhc28490c2006-10-26 14:25:58 +00003406static int process_input(struct callback_data *p, FILE *in){
drh9f099fd2013-08-06 14:01:46 +00003407 char *zLine = 0; /* A single input line */
3408 char *zSql = 0; /* Accumulated SQL text */
3409 int nLine; /* Length of current line */
3410 int nSql = 0; /* Bytes of zSql[] used */
3411 int nAlloc = 0; /* Allocated zSql[] space */
3412 int nSqlPrior = 0; /* Bytes of zSql[] used by prior line */
3413 char *zErrMsg; /* Error message returned */
3414 int rc; /* Error code */
3415 int errCnt = 0; /* Number of errors seen */
3416 int lineno = 0; /* Current line number */
3417 int startline = 0; /* Line number for start of current input */
drhc49f44e2006-10-26 18:15:42 +00003418
3419 while( errCnt==0 || !bail_on_error || (in==0 && stdin_is_interactive) ){
3420 fflush(p->out);
drh9f099fd2013-08-06 14:01:46 +00003421 zLine = one_input_line(in, zLine, nSql>0);
drhc49f44e2006-10-26 18:15:42 +00003422 if( zLine==0 ){
drh9b8d3572012-04-21 11:33:39 +00003423 /* End of input */
3424 if( stdin_is_interactive ) printf("\n");
3425 break;
drhc49f44e2006-10-26 18:15:42 +00003426 }
drh67505e72002-04-19 12:34:06 +00003427 if( seenInterrupt ){
3428 if( in!=0 ) break;
3429 seenInterrupt = 0;
3430 }
drhc28490c2006-10-26 14:25:58 +00003431 lineno++;
drh849a9d92013-12-21 15:46:06 +00003432 if( nSql==0 && _all_whitespace(zLine) ){
3433 if( p->echoOn ) printf("%s\n", zLine);
3434 continue;
3435 }
drh2af0b2d2002-02-21 02:25:02 +00003436 if( zLine && zLine[0]=='.' && nSql==0 ){
shaneb9fc17d2009-10-22 21:23:35 +00003437 if( p->echoOn ) printf("%s\n", zLine);
drhc49f44e2006-10-26 18:15:42 +00003438 rc = do_meta_command(zLine, p);
shane916f9612009-10-23 00:37:15 +00003439 if( rc==2 ){ /* exit requested */
drh47ad6842006-11-08 12:25:42 +00003440 break;
3441 }else if( rc ){
drhc49f44e2006-10-26 18:15:42 +00003442 errCnt++;
3443 }
drhdaffd0e2001-04-11 14:28:42 +00003444 continue;
3445 }
drh9f099fd2013-08-06 14:01:46 +00003446 if( line_is_command_terminator(zLine) && line_is_complete(zSql, nSql) ){
drh5bb3eb92007-05-04 13:15:55 +00003447 memcpy(zLine,";",2);
drha9b17162003-04-29 18:01:28 +00003448 }
drh9f099fd2013-08-06 14:01:46 +00003449 nLine = strlen30(zLine);
3450 if( nSql+nLine+2>=nAlloc ){
3451 nAlloc = nSql+nLine+100;
3452 zSql = realloc(zSql, nAlloc);
drhdaffd0e2001-04-11 14:28:42 +00003453 if( zSql==0 ){
drh9f099fd2013-08-06 14:01:46 +00003454 fprintf(stderr, "Error: out of memory\n");
drhdaffd0e2001-04-11 14:28:42 +00003455 exit(1);
3456 }
drhdaffd0e2001-04-11 14:28:42 +00003457 }
drh9f099fd2013-08-06 14:01:46 +00003458 nSqlPrior = nSql;
3459 if( nSql==0 ){
3460 int i;
3461 for(i=0; zLine[i] && IsSpace(zLine[i]); i++){}
drh77dfd5b2013-08-19 11:15:48 +00003462 assert( nAlloc>0 && zSql!=0 );
drh9f099fd2013-08-06 14:01:46 +00003463 memcpy(zSql, zLine+i, nLine+1-i);
3464 startline = lineno;
3465 nSql = nLine-i;
3466 }else{
3467 zSql[nSql++] = '\n';
3468 memcpy(zSql+nSql, zLine, nLine+1);
3469 nSql += nLine;
3470 }
3471 if( nSql && line_contains_semicolon(&zSql[nSqlPrior], nSql-nSqlPrior)
drh91a66392007-09-07 01:12:32 +00003472 && sqlite3_complete(zSql) ){
drhdaffd0e2001-04-11 14:28:42 +00003473 p->cnt = 0;
drh05782482013-10-24 15:20:20 +00003474 open_db(p, 0);
drh3b1a9882007-11-02 12:53:03 +00003475 BEGIN_TIMER;
shane626a6e42009-10-22 17:30:15 +00003476 rc = shell_exec(p->db, zSql, shell_callback, p, &zErrMsg);
drh3b1a9882007-11-02 12:53:03 +00003477 END_TIMER;
drh7f953e22002-07-13 17:33:45 +00003478 if( rc || zErrMsg ){
drhc28490c2006-10-26 14:25:58 +00003479 char zPrefix[100];
3480 if( in!=0 || !stdin_is_interactive ){
drh5bb3eb92007-05-04 13:15:55 +00003481 sqlite3_snprintf(sizeof(zPrefix), zPrefix,
shane9bd1b442009-10-23 01:27:39 +00003482 "Error: near line %d:", startline);
drhc28490c2006-10-26 14:25:58 +00003483 }else{
shane9bd1b442009-10-23 01:27:39 +00003484 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "Error:");
drhc28490c2006-10-26 14:25:58 +00003485 }
drh7f953e22002-07-13 17:33:45 +00003486 if( zErrMsg!=0 ){
shaned2bed1c2009-10-21 03:56:54 +00003487 fprintf(stderr, "%s %s\n", zPrefix, zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00003488 sqlite3_free(zErrMsg);
drh7f953e22002-07-13 17:33:45 +00003489 zErrMsg = 0;
3490 }else{
shaned2bed1c2009-10-21 03:56:54 +00003491 fprintf(stderr, "%s %s\n", zPrefix, sqlite3_errmsg(p->db));
drh7f953e22002-07-13 17:33:45 +00003492 }
drhc49f44e2006-10-26 18:15:42 +00003493 errCnt++;
drhdaffd0e2001-04-11 14:28:42 +00003494 }
drhdaffd0e2001-04-11 14:28:42 +00003495 nSql = 0;
drhc2ce0be2014-05-29 12:36:14 +00003496 if( p->outCount ){
3497 output_reset(p);
3498 p->outCount = 0;
3499 }
drh9f099fd2013-08-06 14:01:46 +00003500 }else if( nSql && _all_whitespace(zSql) ){
drh849a9d92013-12-21 15:46:06 +00003501 if( p->echoOn ) printf("%s\n", zSql);
drh7a411f42013-04-17 17:33:17 +00003502 nSql = 0;
drhdaffd0e2001-04-11 14:28:42 +00003503 }
3504 }
drh9f099fd2013-08-06 14:01:46 +00003505 if( nSql ){
drhd416fe72011-03-17 16:45:50 +00003506 if( !_all_whitespace(zSql) ){
3507 fprintf(stderr, "Error: incomplete SQL: %s\n", zSql);
3508 }
drhdaffd0e2001-04-11 14:28:42 +00003509 free(zSql);
3510 }
danielk19772ac27622007-07-03 05:31:16 +00003511 free(zLine);
drh4d15a0d2012-12-01 20:21:22 +00003512 return errCnt>0;
drhdaffd0e2001-04-11 14:28:42 +00003513}
3514
drh67505e72002-04-19 12:34:06 +00003515/*
3516** Return a pathname which is the user's home directory. A
drh85e72432012-04-11 11:38:53 +00003517** 0 return indicates an error of some kind.
drh67505e72002-04-19 12:34:06 +00003518*/
3519static char *find_home_dir(void){
drh85e72432012-04-11 11:38:53 +00003520 static char *home_dir = NULL;
3521 if( home_dir ) return home_dir;
persicom7e2dfdd2002-04-18 02:46:52 +00003522
drh83905c92012-06-21 13:00:37 +00003523#if !defined(_WIN32) && !defined(WIN32) && !defined(_WIN32_WCE) && !defined(__RTP__) && !defined(_WRS_KERNEL)
mistachkinc8bde372012-06-18 08:00:56 +00003524 {
3525 struct passwd *pwent;
3526 uid_t uid = getuid();
3527 if( (pwent=getpwuid(uid)) != NULL) {
3528 home_dir = pwent->pw_dir;
3529 }
drh67505e72002-04-19 12:34:06 +00003530 }
3531#endif
3532
chw65d3c132007-11-12 21:09:10 +00003533#if defined(_WIN32_WCE)
3534 /* Windows CE (arm-wince-mingw32ce-gcc) does not provide getenv()
3535 */
drh85e72432012-04-11 11:38:53 +00003536 home_dir = "/";
chw65d3c132007-11-12 21:09:10 +00003537#else
3538
drh83905c92012-06-21 13:00:37 +00003539#if defined(_WIN32) || defined(WIN32)
drh164a1b62006-08-19 11:15:20 +00003540 if (!home_dir) {
3541 home_dir = getenv("USERPROFILE");
3542 }
3543#endif
3544
drh67505e72002-04-19 12:34:06 +00003545 if (!home_dir) {
3546 home_dir = getenv("HOME");
drh67505e72002-04-19 12:34:06 +00003547 }
3548
drh83905c92012-06-21 13:00:37 +00003549#if defined(_WIN32) || defined(WIN32)
drhe98d4fa2002-04-21 19:06:22 +00003550 if (!home_dir) {
drh164a1b62006-08-19 11:15:20 +00003551 char *zDrive, *zPath;
3552 int n;
3553 zDrive = getenv("HOMEDRIVE");
3554 zPath = getenv("HOMEPATH");
3555 if( zDrive && zPath ){
drh4f21c4a2008-12-10 22:15:00 +00003556 n = strlen30(zDrive) + strlen30(zPath) + 1;
drh164a1b62006-08-19 11:15:20 +00003557 home_dir = malloc( n );
3558 if( home_dir==0 ) return 0;
3559 sqlite3_snprintf(n, home_dir, "%s%s", zDrive, zPath);
3560 return home_dir;
3561 }
3562 home_dir = "c:\\";
drhe98d4fa2002-04-21 19:06:22 +00003563 }
3564#endif
3565
chw65d3c132007-11-12 21:09:10 +00003566#endif /* !_WIN32_WCE */
3567
drh67505e72002-04-19 12:34:06 +00003568 if( home_dir ){
drh4f21c4a2008-12-10 22:15:00 +00003569 int n = strlen30(home_dir) + 1;
drh5bb3eb92007-05-04 13:15:55 +00003570 char *z = malloc( n );
3571 if( z ) memcpy(z, home_dir, n);
drh67505e72002-04-19 12:34:06 +00003572 home_dir = z;
3573 }
drhe98d4fa2002-04-21 19:06:22 +00003574
drh67505e72002-04-19 12:34:06 +00003575 return home_dir;
3576}
3577
3578/*
3579** Read input from the file given by sqliterc_override. Or if that
3580** parameter is NULL, take input from ~/.sqliterc
shane9bd1b442009-10-23 01:27:39 +00003581**
3582** Returns the number of errors.
drh67505e72002-04-19 12:34:06 +00003583*/
shane9bd1b442009-10-23 01:27:39 +00003584static int process_sqliterc(
drh22fbcb82004-02-01 01:22:50 +00003585 struct callback_data *p, /* Configuration data */
3586 const char *sqliterc_override /* Name of config file. NULL to use default */
3587){
persicom7e2dfdd2002-04-18 02:46:52 +00003588 char *home_dir = NULL;
drh22fbcb82004-02-01 01:22:50 +00003589 const char *sqliterc = sqliterc_override;
drh43617e92006-03-06 20:55:46 +00003590 char *zBuf = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00003591 FILE *in = NULL;
shane9bd1b442009-10-23 01:27:39 +00003592 int rc = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00003593
3594 if (sqliterc == NULL) {
drh67505e72002-04-19 12:34:06 +00003595 home_dir = find_home_dir();
drhe98d4fa2002-04-21 19:06:22 +00003596 if( home_dir==0 ){
chw97185482008-11-17 08:05:31 +00003597#if !defined(__RTP__) && !defined(_WRS_KERNEL)
shane86f5bdb2009-10-24 02:00:07 +00003598 fprintf(stderr,"%s: Error: cannot locate your home directory\n", Argv0);
chw97185482008-11-17 08:05:31 +00003599#endif
shane9bd1b442009-10-23 01:27:39 +00003600 return 1;
drhe98d4fa2002-04-21 19:06:22 +00003601 }
drh2f3de322012-06-27 16:41:31 +00003602 sqlite3_initialize();
drh85e72432012-04-11 11:38:53 +00003603 zBuf = sqlite3_mprintf("%s/.sqliterc",home_dir);
3604 sqliterc = zBuf;
persicom7e2dfdd2002-04-18 02:46:52 +00003605 }
drha1f9b5e2004-02-14 16:31:02 +00003606 in = fopen(sqliterc,"rb");
drh22fbcb82004-02-01 01:22:50 +00003607 if( in ){
drhc28490c2006-10-26 14:25:58 +00003608 if( stdin_is_interactive ){
shane86f5bdb2009-10-24 02:00:07 +00003609 fprintf(stderr,"-- Loading resources from %s\n",sqliterc);
drh22fbcb82004-02-01 01:22:50 +00003610 }
shane9bd1b442009-10-23 01:27:39 +00003611 rc = process_input(p,in);
drhdd45df82002-04-18 12:39:03 +00003612 fclose(in);
persicom7e2dfdd2002-04-18 02:46:52 +00003613 }
drh85e72432012-04-11 11:38:53 +00003614 sqlite3_free(zBuf);
shane9bd1b442009-10-23 01:27:39 +00003615 return rc;
persicom7e2dfdd2002-04-18 02:46:52 +00003616}
3617
drh67505e72002-04-19 12:34:06 +00003618/*
drhe1e38c42003-05-04 18:30:59 +00003619** Show available command line options
3620*/
3621static const char zOptions[] =
drhc49f44e2006-10-26 18:15:42 +00003622 " -bail stop after hitting an error\n"
drhc49f44e2006-10-26 18:15:42 +00003623 " -batch force batch I/O\n"
drhe1e38c42003-05-04 18:30:59 +00003624 " -column set output mode to 'column'\n"
mistachkin6d81d752012-10-25 15:43:28 +00003625 " -cmd COMMAND run \"COMMAND\" before reading stdin\n"
drhc49f44e2006-10-26 18:15:42 +00003626 " -csv set output mode to 'csv'\n"
drhcc3b4f82012-02-07 14:13:50 +00003627 " -echo print commands before execution\n"
mistachkin6d81d752012-10-25 15:43:28 +00003628 " -init FILENAME read/process named file\n"
drhcc3b4f82012-02-07 14:13:50 +00003629 " -[no]header turn headers on or off\n"
drh98d312f2012-10-25 15:23:14 +00003630#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
3631 " -heap SIZE Size of heap for memsys3 or memsys5\n"
3632#endif
drhcc3b4f82012-02-07 14:13:50 +00003633 " -help show this message\n"
drhe1e38c42003-05-04 18:30:59 +00003634 " -html set output mode to HTML\n"
drhcc3b4f82012-02-07 14:13:50 +00003635 " -interactive force interactive I/O\n"
drhe1e38c42003-05-04 18:30:59 +00003636 " -line set output mode to 'line'\n"
3637 " -list set output mode to 'list'\n"
drh7d9f3942013-04-03 01:26:54 +00003638 " -mmap N default mmap size set to N\n"
drhcc3b4f82012-02-07 14:13:50 +00003639#ifdef SQLITE_ENABLE_MULTIPLEX
3640 " -multiplex enable the multiplexor VFS\n"
3641#endif
drh98d312f2012-10-25 15:23:14 +00003642 " -nullvalue TEXT set text string for NULL values. Default ''\n"
3643 " -separator SEP set output field separator. Default: '|'\n"
shaneh642d8b82010-07-28 16:05:34 +00003644 " -stats print memory stats before each finalize\n"
drhe1e38c42003-05-04 18:30:59 +00003645 " -version show SQLite version\n"
drha7e61d82011-03-12 17:02:57 +00003646 " -vfs NAME use NAME as the default VFS\n"
drh2b625e22011-03-16 17:05:28 +00003647#ifdef SQLITE_ENABLE_VFSTRACE
3648 " -vfstrace enable tracing of all VFS calls\n"
3649#endif
drhe1e38c42003-05-04 18:30:59 +00003650;
3651static void usage(int showDetail){
drh80e8be92006-08-29 12:04:19 +00003652 fprintf(stderr,
3653 "Usage: %s [OPTIONS] FILENAME [SQL]\n"
3654 "FILENAME is the name of an SQLite database. A new database is created\n"
3655 "if the file does not previously exist.\n", Argv0);
drhe1e38c42003-05-04 18:30:59 +00003656 if( showDetail ){
drh80e8be92006-08-29 12:04:19 +00003657 fprintf(stderr, "OPTIONS include:\n%s", zOptions);
drhe1e38c42003-05-04 18:30:59 +00003658 }else{
3659 fprintf(stderr, "Use the -help option for additional information\n");
3660 }
3661 exit(1);
3662}
3663
3664/*
drh67505e72002-04-19 12:34:06 +00003665** Initialize the state information in data
3666*/
drh0850b532006-01-31 19:31:43 +00003667static void main_init(struct callback_data *data) {
persicom7e2dfdd2002-04-18 02:46:52 +00003668 memset(data, 0, sizeof(*data));
3669 data->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00003670 memcpy(data->separator,"|", 2);
persicom7e2dfdd2002-04-18 02:46:52 +00003671 data->showHeader = 0;
drh52784bd2011-05-18 17:15:06 +00003672 sqlite3_config(SQLITE_CONFIG_URI, 1);
drh127f9d72010-02-23 01:47:00 +00003673 sqlite3_config(SQLITE_CONFIG_LOG, shellLog, data);
drh5bb3eb92007-05-04 13:15:55 +00003674 sqlite3_snprintf(sizeof(mainPrompt), mainPrompt,"sqlite> ");
3675 sqlite3_snprintf(sizeof(continuePrompt), continuePrompt," ...> ");
dan0f831772010-03-03 07:23:12 +00003676 sqlite3_config(SQLITE_CONFIG_SINGLETHREAD);
persicom7e2dfdd2002-04-18 02:46:52 +00003677}
3678
drh98d312f2012-10-25 15:23:14 +00003679/*
drh5c7976f2014-02-10 19:59:27 +00003680** Output text to the console in a font that attracts extra attention.
drh1247aa42014-02-10 19:27:05 +00003681*/
3682#ifdef _WIN32
drh5c7976f2014-02-10 19:59:27 +00003683static void printBold(const char *zText){
3684 HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
3685 CONSOLE_SCREEN_BUFFER_INFO defaultScreenInfo;
3686 GetConsoleScreenBufferInfo(out, &defaultScreenInfo);
3687 SetConsoleTextAttribute(out,
3688 FOREGROUND_RED|FOREGROUND_INTENSITY
3689 );
3690 printf("%s", zText);
3691 SetConsoleTextAttribute(out, defaultScreenInfo.wAttributes);
drh1247aa42014-02-10 19:27:05 +00003692}
3693#else
drh5c7976f2014-02-10 19:59:27 +00003694static void printBold(const char *zText){
3695 printf("\033[1m%s\033[0m", zText);
drh1247aa42014-02-10 19:27:05 +00003696}
3697#endif
3698
3699/*
drh98d312f2012-10-25 15:23:14 +00003700** Get the argument to an --option. Throw an error and die if no argument
3701** is available.
3702*/
3703static char *cmdline_option_value(int argc, char **argv, int i){
3704 if( i==argc ){
3705 fprintf(stderr, "%s: Error: missing argument to %s\n",
3706 argv[0], argv[argc-1]);
3707 exit(1);
3708 }
3709 return argv[i];
3710}
3711
drh75897232000-05-29 14:26:00 +00003712int main(int argc, char **argv){
drh75897232000-05-29 14:26:00 +00003713 char *zErrMsg = 0;
3714 struct callback_data data;
drh22fbcb82004-02-01 01:22:50 +00003715 const char *zInitFile = 0;
3716 char *zFirstCmd = 0;
drh44c2eb12003-04-30 11:38:26 +00003717 int i;
drhc28490c2006-10-26 14:25:58 +00003718 int rc = 0;
drhb3735912014-02-10 16:13:42 +00003719 int warnInmemoryDb = 0;
drh75897232000-05-29 14:26:00 +00003720
drh69b30ab2014-02-27 15:11:52 +00003721#if USE_SYSTEM_SQLITE+0!=1
drh52784bd2011-05-18 17:15:06 +00003722 if( strcmp(sqlite3_sourceid(),SQLITE_SOURCE_ID)!=0 ){
3723 fprintf(stderr, "SQLite header and source version mismatch\n%s\n%s\n",
3724 sqlite3_sourceid(), SQLITE_SOURCE_ID);
3725 exit(1);
3726 }
drhc7181902014-02-27 15:04:13 +00003727#endif
drhdaffd0e2001-04-11 14:28:42 +00003728 Argv0 = argv[0];
persicom7e2dfdd2002-04-18 02:46:52 +00003729 main_init(&data);
drhc28490c2006-10-26 14:25:58 +00003730 stdin_is_interactive = isatty(0);
persicom7e2dfdd2002-04-18 02:46:52 +00003731
drh44c2eb12003-04-30 11:38:26 +00003732 /* Make sure we have a valid signal handler early, before anything
3733 ** else is done.
3734 */
drh4c504392000-10-16 22:06:40 +00003735#ifdef SIGINT
3736 signal(SIGINT, interrupt_handler);
3737#endif
drh44c2eb12003-04-30 11:38:26 +00003738
drh22fbcb82004-02-01 01:22:50 +00003739 /* Do an initial pass through the command-line argument to locate
3740 ** the name of the database file, the name of the initialization file,
drh9c88d682010-12-17 14:03:01 +00003741 ** the size of the alternative malloc heap,
drh22fbcb82004-02-01 01:22:50 +00003742 ** and the first command to execute.
drh44c2eb12003-04-30 11:38:26 +00003743 */
drh98d312f2012-10-25 15:23:14 +00003744 for(i=1; i<argc; i++){
drhc28490c2006-10-26 14:25:58 +00003745 char *z;
drhc28490c2006-10-26 14:25:58 +00003746 z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003747 if( z[0]!='-' ){
3748 if( data.zDbFilename==0 ){
3749 data.zDbFilename = z;
3750 continue;
3751 }
3752 if( zFirstCmd==0 ){
3753 zFirstCmd = z;
3754 continue;
3755 }
3756 fprintf(stderr,"%s: Error: too many options: \"%s\"\n", Argv0, argv[i]);
3757 fprintf(stderr,"Use -help for a list of options.\n");
3758 return 1;
3759 }
drhcc3b4f82012-02-07 14:13:50 +00003760 if( z[1]=='-' ) z++;
3761 if( strcmp(z,"-separator")==0
3762 || strcmp(z,"-nullvalue")==0
3763 || strcmp(z,"-cmd")==0
3764 ){
drh98d312f2012-10-25 15:23:14 +00003765 (void)cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003766 }else if( strcmp(z,"-init")==0 ){
drh98d312f2012-10-25 15:23:14 +00003767 zInitFile = cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003768 }else if( strcmp(z,"-batch")==0 ){
drh98d312f2012-10-25 15:23:14 +00003769 /* Need to check for batch mode here to so we can avoid printing
3770 ** informational messages (like from process_sqliterc) before
3771 ** we do the actual processing of arguments later in a second pass.
3772 */
shanef69573d2009-10-24 02:06:14 +00003773 stdin_is_interactive = 0;
drhcc3b4f82012-02-07 14:13:50 +00003774 }else if( strcmp(z,"-heap")==0 ){
drhb07028f2011-10-14 21:49:18 +00003775#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
drh9c88d682010-12-17 14:03:01 +00003776 const char *zSize;
3777 sqlite3_int64 szHeap;
3778
drh98d312f2012-10-25 15:23:14 +00003779 zSize = cmdline_option_value(argc, argv, ++i);
drh7d9f3942013-04-03 01:26:54 +00003780 szHeap = integerValue(zSize);
drh9c88d682010-12-17 14:03:01 +00003781 if( szHeap>0x7fff0000 ) szHeap = 0x7fff0000;
drh9c88d682010-12-17 14:03:01 +00003782 sqlite3_config(SQLITE_CONFIG_HEAP, malloc((int)szHeap), (int)szHeap, 64);
3783#endif
drh97ae8ff2011-03-16 16:56:29 +00003784#ifdef SQLITE_ENABLE_VFSTRACE
drhcc3b4f82012-02-07 14:13:50 +00003785 }else if( strcmp(z,"-vfstrace")==0 ){
drh97ae8ff2011-03-16 16:56:29 +00003786 extern int vfstrace_register(
3787 const char *zTraceName,
3788 const char *zOldVfsName,
3789 int (*xOut)(const char*,void*),
3790 void *pOutArg,
3791 int makeDefault
3792 );
drh2b625e22011-03-16 17:05:28 +00003793 vfstrace_register("trace",0,(int(*)(const char*,void*))fputs,stderr,1);
drh97ae8ff2011-03-16 16:56:29 +00003794#endif
drh6f25e892011-07-08 17:02:57 +00003795#ifdef SQLITE_ENABLE_MULTIPLEX
drhcc3b4f82012-02-07 14:13:50 +00003796 }else if( strcmp(z,"-multiplex")==0 ){
drh6f25e892011-07-08 17:02:57 +00003797 extern int sqlite3_multiple_initialize(const char*,int);
3798 sqlite3_multiplex_initialize(0, 1);
3799#endif
drh7d9f3942013-04-03 01:26:54 +00003800 }else if( strcmp(z,"-mmap")==0 ){
drh9b4c59f2013-04-15 17:03:42 +00003801 sqlite3_int64 sz = integerValue(cmdline_option_value(argc,argv,++i));
3802 sqlite3_config(SQLITE_CONFIG_MMAP_SIZE, sz, sz);
drhcc3b4f82012-02-07 14:13:50 +00003803 }else if( strcmp(z,"-vfs")==0 ){
drh98d312f2012-10-25 15:23:14 +00003804 sqlite3_vfs *pVfs = sqlite3_vfs_find(cmdline_option_value(argc,argv,++i));
drha7e61d82011-03-12 17:02:57 +00003805 if( pVfs ){
3806 sqlite3_vfs_register(pVfs, 1);
3807 }else{
3808 fprintf(stderr, "no such VFS: \"%s\"\n", argv[i]);
3809 exit(1);
3810 }
drh44c2eb12003-04-30 11:38:26 +00003811 }
3812 }
drh98d312f2012-10-25 15:23:14 +00003813 if( data.zDbFilename==0 ){
danielk197703aded42004-11-22 05:26:27 +00003814#ifndef SQLITE_OMIT_MEMORYDB
drh22fbcb82004-02-01 01:22:50 +00003815 data.zDbFilename = ":memory:";
drh1247aa42014-02-10 19:27:05 +00003816 warnInmemoryDb = argc==1;
danielk197703aded42004-11-22 05:26:27 +00003817#else
shane86f5bdb2009-10-24 02:00:07 +00003818 fprintf(stderr,"%s: Error: no database filename specified\n", Argv0);
3819 return 1;
drh01b41712005-08-29 23:06:23 +00003820#endif
drhc7181902014-02-27 15:04:13 +00003821#ifdef SQLITE_SHELL_DBNAME_PROC
3822 { extern void SQLITE_SHELL_DBNAME_PROC(const char**);
3823 SQLITE_SHELL_DBNAME_PROC(&data.zDbFilename);
3824 warnInmemoryDb = 0; }
3825#endif
drh98d312f2012-10-25 15:23:14 +00003826 }
3827 data.out = stdout;
drh01b41712005-08-29 23:06:23 +00003828
drh44c2eb12003-04-30 11:38:26 +00003829 /* Go ahead and open the database file if it already exists. If the
3830 ** file does not exist, delay opening it. This prevents empty database
3831 ** files from being created if a user mistypes the database name argument
3832 ** to the sqlite command-line tool.
3833 */
drhc8d74412004-08-31 23:41:26 +00003834 if( access(data.zDbFilename, 0)==0 ){
drh05782482013-10-24 15:20:20 +00003835 open_db(&data, 0);
drh44c2eb12003-04-30 11:38:26 +00003836 }
3837
drh22fbcb82004-02-01 01:22:50 +00003838 /* Process the initialization file if there is one. If no -init option
3839 ** is given on the command line, look for a file named ~/.sqliterc and
3840 ** try to process it.
drh44c2eb12003-04-30 11:38:26 +00003841 */
shane86f5bdb2009-10-24 02:00:07 +00003842 rc = process_sqliterc(&data,zInitFile);
3843 if( rc>0 ){
3844 return rc;
3845 }
drh44c2eb12003-04-30 11:38:26 +00003846
drh22fbcb82004-02-01 01:22:50 +00003847 /* Make a second pass through the command-line argument and set
3848 ** options. This second pass is delayed until after the initialization
3849 ** file is processed so that the command-line arguments will override
3850 ** settings in the initialization file.
drh44c2eb12003-04-30 11:38:26 +00003851 */
drh98d312f2012-10-25 15:23:14 +00003852 for(i=1; i<argc; i++){
drh22fbcb82004-02-01 01:22:50 +00003853 char *z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003854 if( z[0]!='-' ) continue;
drhc28490c2006-10-26 14:25:58 +00003855 if( z[1]=='-' ){ z++; }
drh2e584cd2006-09-25 13:09:22 +00003856 if( strcmp(z,"-init")==0 ){
drh22fbcb82004-02-01 01:22:50 +00003857 i++;
3858 }else if( strcmp(z,"-html")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003859 data.mode = MODE_Html;
drh22fbcb82004-02-01 01:22:50 +00003860 }else if( strcmp(z,"-list")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003861 data.mode = MODE_List;
drh22fbcb82004-02-01 01:22:50 +00003862 }else if( strcmp(z,"-line")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003863 data.mode = MODE_Line;
drh22fbcb82004-02-01 01:22:50 +00003864 }else if( strcmp(z,"-column")==0 ){
drh8b32e172002-04-08 02:42:57 +00003865 data.mode = MODE_Column;
drhc49f44e2006-10-26 18:15:42 +00003866 }else if( strcmp(z,"-csv")==0 ){
3867 data.mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00003868 memcpy(data.separator,",",2);
drh22fbcb82004-02-01 01:22:50 +00003869 }else if( strcmp(z,"-separator")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003870 sqlite3_snprintf(sizeof(data.separator), data.separator,
drh98d312f2012-10-25 15:23:14 +00003871 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003872 }else if( strcmp(z,"-nullvalue")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003873 sqlite3_snprintf(sizeof(data.nullvalue), data.nullvalue,
drh98d312f2012-10-25 15:23:14 +00003874 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003875 }else if( strcmp(z,"-header")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003876 data.showHeader = 1;
drh22fbcb82004-02-01 01:22:50 +00003877 }else if( strcmp(z,"-noheader")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003878 data.showHeader = 0;
drh22fbcb82004-02-01 01:22:50 +00003879 }else if( strcmp(z,"-echo")==0 ){
drhdaffd0e2001-04-11 14:28:42 +00003880 data.echoOn = 1;
drhefbf3b12014-02-28 20:47:24 +00003881 }else if( strcmp(z,"-eqp")==0 ){
3882 data.autoEQP = 1;
shaneh642d8b82010-07-28 16:05:34 +00003883 }else if( strcmp(z,"-stats")==0 ){
3884 data.statsOn = 1;
drhc49f44e2006-10-26 18:15:42 +00003885 }else if( strcmp(z,"-bail")==0 ){
3886 bail_on_error = 1;
drh22fbcb82004-02-01 01:22:50 +00003887 }else if( strcmp(z,"-version")==0 ){
drh9fd301b2011-06-03 13:28:22 +00003888 printf("%s %s\n", sqlite3_libversion(), sqlite3_sourceid());
drh151e3e12006-06-06 12:32:21 +00003889 return 0;
drhc28490c2006-10-26 14:25:58 +00003890 }else if( strcmp(z,"-interactive")==0 ){
3891 stdin_is_interactive = 1;
3892 }else if( strcmp(z,"-batch")==0 ){
3893 stdin_is_interactive = 0;
drh9c88d682010-12-17 14:03:01 +00003894 }else if( strcmp(z,"-heap")==0 ){
3895 i++;
drh7d9f3942013-04-03 01:26:54 +00003896 }else if( strcmp(z,"-mmap")==0 ){
3897 i++;
drha7e61d82011-03-12 17:02:57 +00003898 }else if( strcmp(z,"-vfs")==0 ){
3899 i++;
drh6f25e892011-07-08 17:02:57 +00003900#ifdef SQLITE_ENABLE_VFSTRACE
drh97ae8ff2011-03-16 16:56:29 +00003901 }else if( strcmp(z,"-vfstrace")==0 ){
3902 i++;
drh6f25e892011-07-08 17:02:57 +00003903#endif
3904#ifdef SQLITE_ENABLE_MULTIPLEX
3905 }else if( strcmp(z,"-multiplex")==0 ){
3906 i++;
3907#endif
drhcc3b4f82012-02-07 14:13:50 +00003908 }else if( strcmp(z,"-help")==0 ){
drhe1e38c42003-05-04 18:30:59 +00003909 usage(1);
drhcc3b4f82012-02-07 14:13:50 +00003910 }else if( strcmp(z,"-cmd")==0 ){
3911 if( i==argc-1 ) break;
drh98d312f2012-10-25 15:23:14 +00003912 z = cmdline_option_value(argc,argv,++i);
drhcc3b4f82012-02-07 14:13:50 +00003913 if( z[0]=='.' ){
3914 rc = do_meta_command(z, &data);
drh99b39082013-04-17 12:19:48 +00003915 if( rc && bail_on_error ) return rc==2 ? 0 : rc;
drhcc3b4f82012-02-07 14:13:50 +00003916 }else{
drh05782482013-10-24 15:20:20 +00003917 open_db(&data, 0);
drhcc3b4f82012-02-07 14:13:50 +00003918 rc = shell_exec(data.db, z, shell_callback, &data, &zErrMsg);
3919 if( zErrMsg!=0 ){
3920 fprintf(stderr,"Error: %s\n", zErrMsg);
3921 if( bail_on_error ) return rc!=0 ? rc : 1;
3922 }else if( rc!=0 ){
3923 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", z);
3924 if( bail_on_error ) return rc;
3925 }
3926 }
drh1e5d0e92000-05-31 23:33:17 +00003927 }else{
shane86f5bdb2009-10-24 02:00:07 +00003928 fprintf(stderr,"%s: Error: unknown option: %s\n", Argv0, z);
drhe1e38c42003-05-04 18:30:59 +00003929 fprintf(stderr,"Use -help for a list of options.\n");
drh1e5d0e92000-05-31 23:33:17 +00003930 return 1;
3931 }
3932 }
drh44c2eb12003-04-30 11:38:26 +00003933
drh22fbcb82004-02-01 01:22:50 +00003934 if( zFirstCmd ){
drh44c2eb12003-04-30 11:38:26 +00003935 /* Run just the command that follows the database name
3936 */
drh22fbcb82004-02-01 01:22:50 +00003937 if( zFirstCmd[0]=='.' ){
shane916f9612009-10-23 00:37:15 +00003938 rc = do_meta_command(zFirstCmd, &data);
drh99b39082013-04-17 12:19:48 +00003939 if( rc==2 ) rc = 0;
drh6ff13852001-11-25 13:18:23 +00003940 }else{
drh05782482013-10-24 15:20:20 +00003941 open_db(&data, 0);
shane626a6e42009-10-22 17:30:15 +00003942 rc = shell_exec(data.db, zFirstCmd, shell_callback, &data, &zErrMsg);
shane86f5bdb2009-10-24 02:00:07 +00003943 if( zErrMsg!=0 ){
3944 fprintf(stderr,"Error: %s\n", zErrMsg);
3945 return rc!=0 ? rc : 1;
3946 }else if( rc!=0 ){
3947 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", zFirstCmd);
3948 return rc;
drh6ff13852001-11-25 13:18:23 +00003949 }
drh75897232000-05-29 14:26:00 +00003950 }
3951 }else{
drh44c2eb12003-04-30 11:38:26 +00003952 /* Run commands received from standard input
3953 */
drhc28490c2006-10-26 14:25:58 +00003954 if( stdin_is_interactive ){
drh67505e72002-04-19 12:34:06 +00003955 char *zHome;
3956 char *zHistory = 0;
drh5bb3eb92007-05-04 13:15:55 +00003957 int nHistory;
drh75897232000-05-29 14:26:00 +00003958 printf(
drh743e0032011-12-12 16:51:50 +00003959 "SQLite version %s %.19s\n" /*extra-version-info*/
drh1247aa42014-02-10 19:27:05 +00003960 "Enter \".help\" for usage hints.\n",
drh9fd301b2011-06-03 13:28:22 +00003961 sqlite3_libversion(), sqlite3_sourceid()
drh75897232000-05-29 14:26:00 +00003962 );
drhb3735912014-02-10 16:13:42 +00003963 if( warnInmemoryDb ){
drh1247aa42014-02-10 19:27:05 +00003964 printf("Connected to a ");
mistachkin378d01a2014-03-06 02:15:42 +00003965 printBold("transient in-memory database");
3966 printf(".\nUse \".open FILENAME\" to reopen on a "
drh1247aa42014-02-10 19:27:05 +00003967 "persistent database.\n");
drhb3735912014-02-10 16:13:42 +00003968 }
drh67505e72002-04-19 12:34:06 +00003969 zHome = find_home_dir();
drhea678832008-12-10 19:26:22 +00003970 if( zHome ){
drh4f21c4a2008-12-10 22:15:00 +00003971 nHistory = strlen30(zHome) + 20;
drhea678832008-12-10 19:26:22 +00003972 if( (zHistory = malloc(nHistory))!=0 ){
3973 sqlite3_snprintf(nHistory, zHistory,"%s/.sqlite_history", zHome);
3974 }
drh67505e72002-04-19 12:34:06 +00003975 }
drhaaa21b42014-02-11 14:37:51 +00003976#if defined(HAVE_READLINE)
drh67505e72002-04-19 12:34:06 +00003977 if( zHistory ) read_history(zHistory);
danielk19774af00c62005-01-23 23:43:21 +00003978#endif
drhc28490c2006-10-26 14:25:58 +00003979 rc = process_input(&data, 0);
drh67505e72002-04-19 12:34:06 +00003980 if( zHistory ){
3981 stifle_history(100);
3982 write_history(zHistory);
adamd0a3daa32006-07-28 20:16:14 +00003983 free(zHistory);
drh67505e72002-04-19 12:34:06 +00003984 }
drhdaffd0e2001-04-11 14:28:42 +00003985 }else{
drhc28490c2006-10-26 14:25:58 +00003986 rc = process_input(&data, stdin);
drh75897232000-05-29 14:26:00 +00003987 }
3988 }
drh33048c02001-10-01 14:29:22 +00003989 set_table_name(&data, 0);
drh72af0772010-05-06 20:19:55 +00003990 if( data.db ){
drhe14cd932010-12-08 03:28:17 +00003991 sqlite3_close(data.db);
adamd0a3daa32006-07-28 20:16:14 +00003992 }
drh05782482013-10-24 15:20:20 +00003993 sqlite3_free(data.zFreeOnClose);
drhc28490c2006-10-26 14:25:58 +00003994 return rc;
drh75897232000-05-29 14:26:00 +00003995}