blob: 4edbaec17819a9b26a01d9cf3e01705aea3c85e9 [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"
drh6d36ffe2014-06-16 15:01:37 +00001582 ".eqp on|off Enable or disable automatic EXPLAIN QUERY PLAN\n"
drh75897232000-05-29 14:26:00 +00001583 ".exit Exit this program\n"
drhc2ce0be2014-05-29 12:36:14 +00001584 ".explain ?on|off? Turn output mode suitable for EXPLAIN on or off.\n"
shanehe2aa9d72009-11-06 17:20:17 +00001585 " With no args, it turns EXPLAIN on.\n"
drhc2ce0be2014-05-29 12:36:14 +00001586 ".headers on|off Turn display of headers on or off\n"
drh75897232000-05-29 14:26:00 +00001587 ".help Show this message\n"
drhb860bc92004-08-04 15:16:55 +00001588 ".import FILE TABLE Import data from FILE into TABLE\n"
shane86f5bdb2009-10-24 02:00:07 +00001589 ".indices ?TABLE? Show names of all indices\n"
1590 " If TABLE specified, only show indices for tables\n"
1591 " matching LIKE pattern TABLE.\n"
drhae5e4452007-05-03 17:18:36 +00001592#ifdef SQLITE_ENABLE_IOTRACE
1593 ".iotrace FILE Enable I/O diagnostic logging to FILE\n"
1594#endif
drh70df4fe2006-06-13 15:12:21 +00001595#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh1e397f82006-06-08 15:28:43 +00001596 ".load FILE ?ENTRY? Load an extension library\n"
drh70df4fe2006-06-13 15:12:21 +00001597#endif
drh127f9d72010-02-23 01:47:00 +00001598 ".log FILE|off Turn logging on or off. FILE can be stderr/stdout\n"
danielk19776b77a362005-01-13 11:10:25 +00001599 ".mode MODE ?TABLE? Set output mode where MODE is one of:\n"
drh3b584fa2004-09-24 12:50:03 +00001600 " csv Comma-separated values\n"
drhb860bc92004-08-04 15:16:55 +00001601 " column Left-aligned columns. (See .width)\n"
1602 " html HTML <table> code\n"
1603 " insert SQL insert statements for TABLE\n"
1604 " line One value per line\n"
1605 " list Values delimited by .separator string\n"
1606 " tabs Tab-separated values\n"
1607 " tcl TCL list elements\n"
drh078b1fd2012-09-21 13:40:02 +00001608 ".nullvalue STRING Use STRING in place of NULL values\n"
drhc2ce0be2014-05-29 12:36:14 +00001609 ".once FILENAME Output for the next SQL command only to FILENAME\n"
drh05782482013-10-24 15:20:20 +00001610 ".open ?FILENAME? Close existing database and reopen FILENAME\n"
drhc2ce0be2014-05-29 12:36:14 +00001611 ".output ?FILENAME? Send output to FILENAME or stdout\n"
drh078b1fd2012-09-21 13:40:02 +00001612 ".print STRING... Print literal STRING\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001613 ".prompt MAIN CONTINUE Replace the standard prompts\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001614 ".quit Exit this program\n"
drhdaffd0e2001-04-11 14:28:42 +00001615 ".read FILENAME Execute SQL in FILENAME\n"
drh9ff849f2009-02-04 20:55:57 +00001616 ".restore ?DB? FILE Restore content of DB (default \"main\") from FILE\n"
drh5c7976f2014-02-10 19:59:27 +00001617 ".save FILE Write in-memory database into FILE\n"
drh75897232000-05-29 14:26:00 +00001618 ".schema ?TABLE? Show the CREATE statements\n"
shane86f5bdb2009-10-24 02:00:07 +00001619 " If TABLE specified, only show tables matching\n"
1620 " LIKE pattern TABLE.\n"
drhb860bc92004-08-04 15:16:55 +00001621 ".separator STRING Change separator used by output mode and .import\n"
drh62cdde52014-05-28 20:22:28 +00001622 ".shell CMD ARGS... Run CMD ARGS... in a system shell\n"
drhdd45df82002-04-18 12:39:03 +00001623 ".show Show the current values for various settings\n"
drhc2ce0be2014-05-29 12:36:14 +00001624 ".stats on|off Turn stats on or off\n"
drh62cdde52014-05-28 20:22:28 +00001625 ".system CMD ARGS... Run CMD ARGS... in a system shell\n"
shane86f5bdb2009-10-24 02:00:07 +00001626 ".tables ?TABLE? List names of tables\n"
1627 " If TABLE specified, only list tables matching\n"
1628 " LIKE pattern TABLE.\n"
drh2dfbbca2000-07-28 14:32:48 +00001629 ".timeout MS Try opening locked tables for MS milliseconds\n"
drhc2ce0be2014-05-29 12:36:14 +00001630 ".timer on|off Turn SQL timer on or off\n"
drh42f64e52012-04-04 16:56:23 +00001631 ".trace FILE|off Output each SQL statement as it is run\n"
drhde60fc22011-12-14 17:53:36 +00001632 ".vfsname ?AUX? Print the name of the VFS stack\n"
shanehe2aa9d72009-11-06 17:20:17 +00001633 ".width NUM1 NUM2 ... Set column widths for \"column\" mode\n"
drh62cdde52014-05-28 20:22:28 +00001634 " Negative values right-justify\n"
drh75897232000-05-29 14:26:00 +00001635;
1636
drhdaffd0e2001-04-11 14:28:42 +00001637/* Forward reference */
drhc28490c2006-10-26 14:25:58 +00001638static int process_input(struct callback_data *p, FILE *in);
drhdaffd0e2001-04-11 14:28:42 +00001639
drh75897232000-05-29 14:26:00 +00001640/*
drh44c2eb12003-04-30 11:38:26 +00001641** Make sure the database is open. If it is not, then open it. If
1642** the database fails to open, print an error message and exit.
1643*/
drh05782482013-10-24 15:20:20 +00001644static void open_db(struct callback_data *p, int keepAlive){
drh44c2eb12003-04-30 11:38:26 +00001645 if( p->db==0 ){
drhbbb0be82012-06-27 16:12:27 +00001646 sqlite3_initialize();
danielk19774f057f92004-06-08 00:02:33 +00001647 sqlite3_open(p->zDbFilename, &p->db);
danielk197780290862004-05-22 09:21:21 +00001648 db = p->db;
drh4cea5ba2008-05-05 16:27:24 +00001649 if( db && sqlite3_errcode(db)==SQLITE_OK ){
1650 sqlite3_create_function(db, "shellstatic", 0, SQLITE_UTF8, 0,
1651 shellstaticFunc, 0, 0);
1652 }
1653 if( db==0 || SQLITE_OK!=sqlite3_errcode(db) ){
shane86f5bdb2009-10-24 02:00:07 +00001654 fprintf(stderr,"Error: unable to open database \"%s\": %s\n",
danielk197780290862004-05-22 09:21:21 +00001655 p->zDbFilename, sqlite3_errmsg(db));
drh05782482013-10-24 15:20:20 +00001656 if( keepAlive ) return;
drh22fbcb82004-02-01 01:22:50 +00001657 exit(1);
drh44c2eb12003-04-30 11:38:26 +00001658 }
drhc2e87a32006-06-27 15:16:14 +00001659#ifndef SQLITE_OMIT_LOAD_EXTENSION
1660 sqlite3_enable_load_extension(p->db, 1);
1661#endif
drh44c2eb12003-04-30 11:38:26 +00001662 }
1663}
1664
1665/*
drhfeac5f82004-08-01 00:10:45 +00001666** Do C-language style dequoting.
1667**
1668** \t -> tab
1669** \n -> newline
1670** \r -> carriage return
drh4c56b992013-06-27 13:26:55 +00001671** \" -> "
drhfeac5f82004-08-01 00:10:45 +00001672** \NNN -> ascii character NNN in octal
1673** \\ -> backslash
1674*/
1675static void resolve_backslashes(char *z){
shane7d3846a2008-12-11 02:58:26 +00001676 int i, j;
1677 char c;
drhc2ce0be2014-05-29 12:36:14 +00001678 while( *z && *z!='\\' ) z++;
drhfeac5f82004-08-01 00:10:45 +00001679 for(i=j=0; (c = z[i])!=0; i++, j++){
1680 if( c=='\\' ){
1681 c = z[++i];
1682 if( c=='n' ){
1683 c = '\n';
1684 }else if( c=='t' ){
1685 c = '\t';
1686 }else if( c=='r' ){
1687 c = '\r';
drh4c56b992013-06-27 13:26:55 +00001688 }else if( c=='\\' ){
1689 c = '\\';
drhfeac5f82004-08-01 00:10:45 +00001690 }else if( c>='0' && c<='7' ){
drhaa816082005-12-29 12:53:09 +00001691 c -= '0';
drhfeac5f82004-08-01 00:10:45 +00001692 if( z[i+1]>='0' && z[i+1]<='7' ){
1693 i++;
1694 c = (c<<3) + z[i] - '0';
1695 if( z[i+1]>='0' && z[i+1]<='7' ){
1696 i++;
1697 c = (c<<3) + z[i] - '0';
1698 }
1699 }
1700 }
1701 }
1702 z[j] = c;
1703 }
drhc2ce0be2014-05-29 12:36:14 +00001704 if( j<i ) z[j] = 0;
drhfeac5f82004-08-01 00:10:45 +00001705}
1706
1707/*
drh348d19c2013-06-03 12:47:43 +00001708** Return the value of a hexadecimal digit. Return -1 if the input
1709** is not a hex digit.
drhc28490c2006-10-26 14:25:58 +00001710*/
drh348d19c2013-06-03 12:47:43 +00001711static int hexDigitValue(char c){
1712 if( c>='0' && c<='9' ) return c - '0';
1713 if( c>='a' && c<='f' ) return c - 'a' + 10;
1714 if( c>='A' && c<='F' ) return c - 'A' + 10;
1715 return -1;
drhc28490c2006-10-26 14:25:58 +00001716}
1717
1718/*
drh7d9f3942013-04-03 01:26:54 +00001719** Interpret zArg as an integer value, possibly with suffixes.
1720*/
1721static sqlite3_int64 integerValue(const char *zArg){
1722 sqlite3_int64 v = 0;
1723 static const struct { char *zSuffix; int iMult; } aMult[] = {
1724 { "KiB", 1024 },
1725 { "MiB", 1024*1024 },
1726 { "GiB", 1024*1024*1024 },
1727 { "KB", 1000 },
1728 { "MB", 1000000 },
1729 { "GB", 1000000000 },
1730 { "K", 1000 },
1731 { "M", 1000000 },
1732 { "G", 1000000000 },
1733 };
1734 int i;
1735 int isNeg = 0;
1736 if( zArg[0]=='-' ){
1737 isNeg = 1;
1738 zArg++;
1739 }else if( zArg[0]=='+' ){
1740 zArg++;
1741 }
drh348d19c2013-06-03 12:47:43 +00001742 if( zArg[0]=='0' && zArg[1]=='x' ){
1743 int x;
1744 zArg += 2;
1745 while( (x = hexDigitValue(zArg[0]))>=0 ){
1746 v = (v<<4) + x;
1747 zArg++;
1748 }
1749 }else{
1750 while( IsDigit(zArg[0]) ){
1751 v = v*10 + zArg[0] - '0';
1752 zArg++;
1753 }
drh7d9f3942013-04-03 01:26:54 +00001754 }
drhc2bed0a2013-05-24 11:57:50 +00001755 for(i=0; i<ArraySize(aMult); i++){
drh7d9f3942013-04-03 01:26:54 +00001756 if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){
1757 v *= aMult[i].iMult;
1758 break;
1759 }
1760 }
1761 return isNeg? -v : v;
1762}
1763
1764/*
drh348d19c2013-06-03 12:47:43 +00001765** Interpret zArg as either an integer or a boolean value. Return 1 or 0
1766** for TRUE and FALSE. Return the integer value if appropriate.
1767*/
1768static int booleanValue(char *zArg){
1769 int i;
1770 if( zArg[0]=='0' && zArg[1]=='x' ){
1771 for(i=2; hexDigitValue(zArg[i])>=0; i++){}
1772 }else{
1773 for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){}
1774 }
1775 if( i>0 && zArg[i]==0 ) return (int)(integerValue(zArg) & 0xffffffff);
1776 if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){
1777 return 1;
1778 }
1779 if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){
1780 return 0;
1781 }
1782 fprintf(stderr, "ERROR: Not a boolean value: \"%s\". Assuming \"no\".\n",
1783 zArg);
1784 return 0;
1785}
1786
1787/*
drh42f64e52012-04-04 16:56:23 +00001788** Close an output file, assuming it is not stderr or stdout
1789*/
1790static void output_file_close(FILE *f){
1791 if( f && f!=stdout && f!=stderr ) fclose(f);
1792}
1793
1794/*
1795** Try to open an output file. The names "stdout" and "stderr" are
1796** recognized and do the right thing. NULL is returned if the output
1797** filename is "off".
1798*/
1799static FILE *output_file_open(const char *zFile){
1800 FILE *f;
1801 if( strcmp(zFile,"stdout")==0 ){
1802 f = stdout;
1803 }else if( strcmp(zFile, "stderr")==0 ){
1804 f = stderr;
1805 }else if( strcmp(zFile, "off")==0 ){
1806 f = 0;
1807 }else{
1808 f = fopen(zFile, "wb");
1809 if( f==0 ){
1810 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
1811 }
1812 }
1813 return f;
1814}
1815
1816/*
1817** A routine for handling output from sqlite3_trace().
1818*/
1819static void sql_trace_callback(void *pArg, const char *z){
1820 FILE *f = (FILE*)pArg;
1821 if( f ) fprintf(f, "%s\n", z);
1822}
1823
1824/*
drhd8621b92012-04-17 09:09:33 +00001825** A no-op routine that runs with the ".breakpoint" doc-command. This is
1826** a useful spot to set a debugger breakpoint.
1827*/
1828static void test_breakpoint(void){
1829 static int nCall = 0;
1830 nCall++;
1831}
1832
1833/*
drhdb95f682013-06-26 22:46:00 +00001834** An object used to read a CSV file
1835*/
1836typedef struct CSVReader CSVReader;
1837struct CSVReader {
1838 const char *zFile; /* Name of the input file */
1839 FILE *in; /* Read the CSV text from this input stream */
1840 char *z; /* Accumulated text for a field */
1841 int n; /* Number of bytes in z */
1842 int nAlloc; /* Space allocated for z[] */
1843 int nLine; /* Current line number */
1844 int cTerm; /* Character that terminated the most recent field */
1845 int cSeparator; /* The separator character. (Usually ",") */
1846};
1847
1848/* Append a single byte to z[] */
1849static void csv_append_char(CSVReader *p, int c){
1850 if( p->n+1>=p->nAlloc ){
1851 p->nAlloc += p->nAlloc + 100;
1852 p->z = sqlite3_realloc(p->z, p->nAlloc);
1853 if( p->z==0 ){
1854 fprintf(stderr, "out of memory\n");
1855 exit(1);
1856 }
1857 }
1858 p->z[p->n++] = (char)c;
1859}
1860
1861/* Read a single field of CSV text. Compatible with rfc4180 and extended
1862** with the option of having a separator other than ",".
1863**
1864** + Input comes from p->in.
1865** + Store results in p->z of length p->n. Space to hold p->z comes
1866** from sqlite3_malloc().
1867** + Use p->cSep as the separator. The default is ",".
1868** + Keep track of the line number in p->nLine.
1869** + Store the character that terminates the field in p->cTerm. Store
1870** EOF on end-of-file.
1871** + Report syntax errors on stderr
1872*/
1873static char *csv_read_one_field(CSVReader *p){
drha81ad172013-12-11 14:00:04 +00001874 int c, pc, ppc;
drhdb95f682013-06-26 22:46:00 +00001875 int cSep = p->cSeparator;
1876 p->n = 0;
1877 c = fgetc(p->in);
1878 if( c==EOF || seenInterrupt ){
1879 p->cTerm = EOF;
1880 return 0;
1881 }
1882 if( c=='"' ){
1883 int startLine = p->nLine;
1884 int cQuote = c;
drha81ad172013-12-11 14:00:04 +00001885 pc = ppc = 0;
drhdb95f682013-06-26 22:46:00 +00001886 while( 1 ){
1887 c = fgetc(p->in);
1888 if( c=='\n' ) p->nLine++;
1889 if( c==cQuote ){
1890 if( pc==cQuote ){
1891 pc = 0;
1892 continue;
1893 }
1894 }
1895 if( (c==cSep && pc==cQuote)
1896 || (c=='\n' && pc==cQuote)
drha81ad172013-12-11 14:00:04 +00001897 || (c=='\n' && pc=='\r' && ppc==cQuote)
drhdb95f682013-06-26 22:46:00 +00001898 || (c==EOF && pc==cQuote)
1899 ){
1900 do{ p->n--; }while( p->z[p->n]!=cQuote );
drhdb95f682013-06-26 22:46:00 +00001901 p->cTerm = c;
1902 break;
1903 }
1904 if( pc==cQuote && c!='\r' ){
1905 fprintf(stderr, "%s:%d: unescaped %c character\n",
1906 p->zFile, p->nLine, cQuote);
1907 }
1908 if( c==EOF ){
1909 fprintf(stderr, "%s:%d: unterminated %c-quoted field\n",
1910 p->zFile, startLine, cQuote);
drhdb95f682013-06-26 22:46:00 +00001911 p->cTerm = EOF;
1912 break;
1913 }
1914 csv_append_char(p, c);
drha81ad172013-12-11 14:00:04 +00001915 ppc = pc;
drhdb95f682013-06-26 22:46:00 +00001916 pc = c;
drhd0a64dc2013-06-30 20:24:26 +00001917 }
drhdb95f682013-06-26 22:46:00 +00001918 }else{
drhd0a64dc2013-06-30 20:24:26 +00001919 while( c!=EOF && c!=cSep && c!='\n' ){
drhdb95f682013-06-26 22:46:00 +00001920 csv_append_char(p, c);
drhd0a64dc2013-06-30 20:24:26 +00001921 c = fgetc(p->in);
drhdb95f682013-06-26 22:46:00 +00001922 }
1923 if( c=='\n' ){
1924 p->nLine++;
drh3852b682014-02-26 13:53:34 +00001925 if( p->n>0 && p->z[p->n-1]=='\r' ) p->n--;
drhdb95f682013-06-26 22:46:00 +00001926 }
drhdb95f682013-06-26 22:46:00 +00001927 p->cTerm = c;
1928 }
drh8dd675e2013-07-12 21:09:24 +00001929 if( p->z ) p->z[p->n] = 0;
drhdb95f682013-06-26 22:46:00 +00001930 return p->z;
1931}
1932
1933/*
drh4bbcf102014-02-06 02:46:08 +00001934** Try to transfer data for table zTable. If an error is seen while
1935** moving forward, try to go backwards. The backwards movement won't
1936** work for WITHOUT ROWID tables.
drh3350ce92014-02-06 00:49:12 +00001937*/
mistachkine31ae902014-02-06 01:15:29 +00001938static void tryToCloneData(
drh3350ce92014-02-06 00:49:12 +00001939 struct callback_data *p,
1940 sqlite3 *newDb,
1941 const char *zTable
1942){
1943 sqlite3_stmt *pQuery = 0;
1944 sqlite3_stmt *pInsert = 0;
1945 char *zQuery = 0;
1946 char *zInsert = 0;
1947 int rc;
1948 int i, j, n;
1949 int nTable = (int)strlen(zTable);
1950 int k = 0;
drh4bbcf102014-02-06 02:46:08 +00001951 int cnt = 0;
1952 const int spinRate = 10000;
drh3350ce92014-02-06 00:49:12 +00001953
1954 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\"", zTable);
1955 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
1956 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00001957 fprintf(stderr, "Error %d: %s on [%s]\n",
drh3350ce92014-02-06 00:49:12 +00001958 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
1959 zQuery);
1960 goto end_data_xfer;
1961 }
1962 n = sqlite3_column_count(pQuery);
1963 zInsert = sqlite3_malloc(200 + nTable + n*3);
1964 if( zInsert==0 ){
1965 fprintf(stderr, "out of memory\n");
1966 goto end_data_xfer;
1967 }
1968 sqlite3_snprintf(200+nTable,zInsert,
1969 "INSERT OR IGNORE INTO \"%s\" VALUES(?", zTable);
1970 i = (int)strlen(zInsert);
1971 for(j=1; j<n; j++){
1972 memcpy(zInsert+i, ",?", 2);
1973 i += 2;
1974 }
1975 memcpy(zInsert+i, ");", 3);
1976 rc = sqlite3_prepare_v2(newDb, zInsert, -1, &pInsert, 0);
1977 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00001978 fprintf(stderr, "Error %d: %s on [%s]\n",
drh3350ce92014-02-06 00:49:12 +00001979 sqlite3_extended_errcode(newDb), sqlite3_errmsg(newDb),
1980 zQuery);
1981 goto end_data_xfer;
1982 }
1983 for(k=0; k<2; k++){
1984 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
1985 for(i=0; i<n; i++){
1986 switch( sqlite3_column_type(pQuery, i) ){
1987 case SQLITE_NULL: {
1988 sqlite3_bind_null(pInsert, i+1);
1989 break;
1990 }
1991 case SQLITE_INTEGER: {
1992 sqlite3_bind_int64(pInsert, i+1, sqlite3_column_int64(pQuery,i));
1993 break;
1994 }
1995 case SQLITE_FLOAT: {
1996 sqlite3_bind_double(pInsert, i+1, sqlite3_column_double(pQuery,i));
1997 break;
1998 }
1999 case SQLITE_TEXT: {
2000 sqlite3_bind_text(pInsert, i+1,
2001 (const char*)sqlite3_column_text(pQuery,i),
2002 -1, SQLITE_STATIC);
2003 break;
2004 }
2005 case SQLITE_BLOB: {
2006 sqlite3_bind_blob(pInsert, i+1, sqlite3_column_blob(pQuery,i),
2007 sqlite3_column_bytes(pQuery,i),
2008 SQLITE_STATIC);
2009 break;
2010 }
2011 }
2012 } /* End for */
drh4bbcf102014-02-06 02:46:08 +00002013 rc = sqlite3_step(pInsert);
2014 if( rc!=SQLITE_OK && rc!=SQLITE_ROW && rc!=SQLITE_DONE ){
2015 fprintf(stderr, "Error %d: %s\n", sqlite3_extended_errcode(newDb),
2016 sqlite3_errmsg(newDb));
2017 }
drh3350ce92014-02-06 00:49:12 +00002018 sqlite3_reset(pInsert);
drh4bbcf102014-02-06 02:46:08 +00002019 cnt++;
2020 if( (cnt%spinRate)==0 ){
2021 printf("%c\b", "|/-\\"[(cnt/spinRate)%4]);
2022 fflush(stdout);
2023 }
drh3350ce92014-02-06 00:49:12 +00002024 } /* End while */
2025 if( rc==SQLITE_DONE ) break;
2026 sqlite3_finalize(pQuery);
2027 sqlite3_free(zQuery);
2028 zQuery = sqlite3_mprintf("SELECT * FROM \"%w\" ORDER BY rowid DESC;",
2029 zTable);
2030 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2031 if( rc ){
drh4bbcf102014-02-06 02:46:08 +00002032 fprintf(stderr, "Warning: cannot step \"%s\" backwards", zTable);
2033 break;
drh3350ce92014-02-06 00:49:12 +00002034 }
2035 } /* End for(k=0...) */
2036
2037end_data_xfer:
2038 sqlite3_finalize(pQuery);
2039 sqlite3_finalize(pInsert);
2040 sqlite3_free(zQuery);
2041 sqlite3_free(zInsert);
2042}
2043
2044
2045/*
2046** Try to transfer all rows of the schema that match zWhere. For
2047** each row, invoke xForEach() on the object defined by that row.
drh4bbcf102014-02-06 02:46:08 +00002048** If an error is encountered while moving forward through the
2049** sqlite_master table, try again moving backwards.
drh3350ce92014-02-06 00:49:12 +00002050*/
mistachkine31ae902014-02-06 01:15:29 +00002051static void tryToCloneSchema(
drh3350ce92014-02-06 00:49:12 +00002052 struct callback_data *p,
2053 sqlite3 *newDb,
2054 const char *zWhere,
2055 void (*xForEach)(struct callback_data*,sqlite3*,const char*)
2056){
2057 sqlite3_stmt *pQuery = 0;
2058 char *zQuery = 0;
2059 int rc;
2060 const unsigned char *zName;
2061 const unsigned char *zSql;
drh4bbcf102014-02-06 02:46:08 +00002062 char *zErrMsg = 0;
drh3350ce92014-02-06 00:49:12 +00002063
2064 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_master"
2065 " WHERE %s", zWhere);
2066 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2067 if( rc ){
2068 fprintf(stderr, "Error: (%d) %s on [%s]\n",
2069 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
2070 zQuery);
2071 goto end_schema_xfer;
2072 }
2073 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
2074 zName = sqlite3_column_text(pQuery, 0);
2075 zSql = sqlite3_column_text(pQuery, 1);
2076 printf("%s... ", zName); fflush(stdout);
drh4bbcf102014-02-06 02:46:08 +00002077 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
2078 if( zErrMsg ){
2079 fprintf(stderr, "Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
2080 sqlite3_free(zErrMsg);
2081 zErrMsg = 0;
2082 }
drh3350ce92014-02-06 00:49:12 +00002083 if( xForEach ){
2084 xForEach(p, newDb, (const char*)zName);
2085 }
2086 printf("done\n");
2087 }
2088 if( rc!=SQLITE_DONE ){
2089 sqlite3_finalize(pQuery);
2090 sqlite3_free(zQuery);
2091 zQuery = sqlite3_mprintf("SELECT name, sql FROM sqlite_master"
2092 " WHERE %s ORDER BY rowid DESC", zWhere);
2093 rc = sqlite3_prepare_v2(p->db, zQuery, -1, &pQuery, 0);
2094 if( rc ){
2095 fprintf(stderr, "Error: (%d) %s on [%s]\n",
2096 sqlite3_extended_errcode(p->db), sqlite3_errmsg(p->db),
2097 zQuery);
2098 goto end_schema_xfer;
2099 }
2100 while( (rc = sqlite3_step(pQuery))==SQLITE_ROW ){
2101 zName = sqlite3_column_text(pQuery, 0);
2102 zSql = sqlite3_column_text(pQuery, 1);
2103 printf("%s... ", zName); fflush(stdout);
drh4bbcf102014-02-06 02:46:08 +00002104 sqlite3_exec(newDb, (const char*)zSql, 0, 0, &zErrMsg);
2105 if( zErrMsg ){
2106 fprintf(stderr, "Error: %s\nSQL: [%s]\n", zErrMsg, zSql);
2107 sqlite3_free(zErrMsg);
2108 zErrMsg = 0;
2109 }
drh3350ce92014-02-06 00:49:12 +00002110 if( xForEach ){
2111 xForEach(p, newDb, (const char*)zName);
2112 }
2113 printf("done\n");
2114 }
2115 }
2116end_schema_xfer:
2117 sqlite3_finalize(pQuery);
2118 sqlite3_free(zQuery);
2119}
2120
2121/*
2122** Open a new database file named "zNewDb". Try to recover as much information
2123** as possible out of the main database (which might be corrupt) and write it
2124** into zNewDb.
2125*/
mistachkine31ae902014-02-06 01:15:29 +00002126static void tryToClone(struct callback_data *p, const char *zNewDb){
drh3350ce92014-02-06 00:49:12 +00002127 int rc;
2128 sqlite3 *newDb = 0;
2129 if( access(zNewDb,0)==0 ){
2130 fprintf(stderr, "File \"%s\" already exists.\n", zNewDb);
2131 return;
2132 }
2133 rc = sqlite3_open(zNewDb, &newDb);
2134 if( rc ){
2135 fprintf(stderr, "Cannot create output database: %s\n",
2136 sqlite3_errmsg(newDb));
2137 }else{
drh54d0d2d2014-04-03 00:32:13 +00002138 sqlite3_exec(p->db, "PRAGMA writable_schema=ON;", 0, 0, 0);
drh3350ce92014-02-06 00:49:12 +00002139 sqlite3_exec(newDb, "BEGIN EXCLUSIVE;", 0, 0, 0);
mistachkine31ae902014-02-06 01:15:29 +00002140 tryToCloneSchema(p, newDb, "type='table'", tryToCloneData);
2141 tryToCloneSchema(p, newDb, "type!='table'", 0);
drh3350ce92014-02-06 00:49:12 +00002142 sqlite3_exec(newDb, "COMMIT;", 0, 0, 0);
drh54d0d2d2014-04-03 00:32:13 +00002143 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
drh3350ce92014-02-06 00:49:12 +00002144 }
2145 sqlite3_close(newDb);
2146}
2147
2148/*
drhc2ce0be2014-05-29 12:36:14 +00002149** Change the output file back to stdout
2150*/
2151static void output_reset(struct callback_data *p){
2152 if( p->outfile[0]=='|' ){
2153 pclose(p->out);
2154 }else{
2155 output_file_close(p->out);
2156 }
2157 p->outfile[0] = 0;
2158 p->out = stdout;
2159}
2160
2161/*
drh75897232000-05-29 14:26:00 +00002162** If an input line begins with "." then invoke this routine to
2163** process that line.
drh67505e72002-04-19 12:34:06 +00002164**
drh47ad6842006-11-08 12:25:42 +00002165** Return 1 on error, 2 to exit, and 0 otherwise.
drh75897232000-05-29 14:26:00 +00002166*/
drh44c2eb12003-04-30 11:38:26 +00002167static int do_meta_command(char *zLine, struct callback_data *p){
drh75897232000-05-29 14:26:00 +00002168 int i = 1;
2169 int nArg = 0;
2170 int n, c;
drh67505e72002-04-19 12:34:06 +00002171 int rc = 0;
drh75897232000-05-29 14:26:00 +00002172 char *azArg[50];
2173
2174 /* Parse the input line into tokens.
2175 */
2176 while( zLine[i] && nArg<ArraySize(azArg) ){
drhf0693c82011-10-11 20:41:54 +00002177 while( IsSpace(zLine[i]) ){ i++; }
drh06333682004-03-09 13:37:45 +00002178 if( zLine[i]==0 ) break;
drh75897232000-05-29 14:26:00 +00002179 if( zLine[i]=='\'' || zLine[i]=='"' ){
2180 int delim = zLine[i++];
2181 azArg[nArg++] = &zLine[i];
drh4c56b992013-06-27 13:26:55 +00002182 while( zLine[i] && zLine[i]!=delim ){
2183 if( zLine[i]=='\\' && delim=='"' && zLine[i+1]!=0 ) i++;
2184 i++;
2185 }
drh75897232000-05-29 14:26:00 +00002186 if( zLine[i]==delim ){
2187 zLine[i++] = 0;
2188 }
drhfeac5f82004-08-01 00:10:45 +00002189 if( delim=='"' ) resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00002190 }else{
2191 azArg[nArg++] = &zLine[i];
drhf0693c82011-10-11 20:41:54 +00002192 while( zLine[i] && !IsSpace(zLine[i]) ){ i++; }
drh75897232000-05-29 14:26:00 +00002193 if( zLine[i] ) zLine[i++] = 0;
drhfeac5f82004-08-01 00:10:45 +00002194 resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00002195 }
2196 }
2197
2198 /* Process the input line.
2199 */
shane9bd1b442009-10-23 01:27:39 +00002200 if( nArg==0 ) return 0; /* no tokens, no error */
drh4f21c4a2008-12-10 22:15:00 +00002201 n = strlen30(azArg[0]);
drh75897232000-05-29 14:26:00 +00002202 c = azArg[0][0];
drh5c7976f2014-02-10 19:59:27 +00002203 if( (c=='b' && n>=3 && strncmp(azArg[0], "backup", n)==0)
2204 || (c=='s' && n>=3 && strncmp(azArg[0], "save", n)==0)
2205 ){
drhbc46f022013-01-23 18:53:23 +00002206 const char *zDestFile = 0;
2207 const char *zDb = 0;
drh9ff849f2009-02-04 20:55:57 +00002208 sqlite3 *pDest;
2209 sqlite3_backup *pBackup;
drhbc46f022013-01-23 18:53:23 +00002210 int j;
2211 for(j=1; j<nArg; j++){
2212 const char *z = azArg[j];
2213 if( z[0]=='-' ){
2214 while( z[0]=='-' ) z++;
drhaf664332013-07-18 20:28:29 +00002215 /* No options to process at this time */
drhbc46f022013-01-23 18:53:23 +00002216 {
2217 fprintf(stderr, "unknown option: %s\n", azArg[j]);
2218 return 1;
2219 }
2220 }else if( zDestFile==0 ){
2221 zDestFile = azArg[j];
2222 }else if( zDb==0 ){
2223 zDb = zDestFile;
2224 zDestFile = azArg[j];
2225 }else{
2226 fprintf(stderr, "too many arguments to .backup\n");
2227 return 1;
2228 }
drh9ff849f2009-02-04 20:55:57 +00002229 }
drhbc46f022013-01-23 18:53:23 +00002230 if( zDestFile==0 ){
2231 fprintf(stderr, "missing FILENAME argument on .backup\n");
2232 return 1;
2233 }
2234 if( zDb==0 ) zDb = "main";
drh9ff849f2009-02-04 20:55:57 +00002235 rc = sqlite3_open(zDestFile, &pDest);
2236 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002237 fprintf(stderr, "Error: cannot open \"%s\"\n", zDestFile);
drh9ff849f2009-02-04 20:55:57 +00002238 sqlite3_close(pDest);
2239 return 1;
2240 }
drh05782482013-10-24 15:20:20 +00002241 open_db(p, 0);
drh9ff849f2009-02-04 20:55:57 +00002242 pBackup = sqlite3_backup_init(pDest, "main", p->db, zDb);
2243 if( pBackup==0 ){
2244 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
2245 sqlite3_close(pDest);
2246 return 1;
2247 }
2248 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK ){}
2249 sqlite3_backup_finish(pBackup);
2250 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00002251 rc = 0;
drh9ff849f2009-02-04 20:55:57 +00002252 }else{
2253 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
shane9bd1b442009-10-23 01:27:39 +00002254 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002255 }
2256 sqlite3_close(pDest);
2257 }else
2258
drhc2ce0be2014-05-29 12:36:14 +00002259 if( c=='b' && n>=3 && strncmp(azArg[0], "bail", n)==0 ){
2260 if( nArg==2 ){
2261 bail_on_error = booleanValue(azArg[1]);
2262 }else{
2263 fprintf(stderr, "Usage: .bail on|off\n");
2264 rc = 1;
2265 }
drhc49f44e2006-10-26 18:15:42 +00002266 }else
2267
drhd8621b92012-04-17 09:09:33 +00002268 /* The undocumented ".breakpoint" command causes a call to the no-op
2269 ** routine named test_breakpoint().
2270 */
2271 if( c=='b' && n>=3 && strncmp(azArg[0], "breakpoint", n)==0 ){
2272 test_breakpoint();
2273 }else
2274
drhc2ce0be2014-05-29 12:36:14 +00002275 if( c=='c' && strncmp(azArg[0], "clone", n)==0 ){
2276 if( nArg==2 ){
2277 tryToClone(p, azArg[1]);
2278 }else{
2279 fprintf(stderr, "Usage: .clone FILENAME\n");
2280 rc = 1;
2281 }
mistachkine31ae902014-02-06 01:15:29 +00002282 }else
2283
drhc2ce0be2014-05-29 12:36:14 +00002284 if( c=='d' && n>1 && strncmp(azArg[0], "databases", n)==0 ){
jplyon672a1ed2003-05-11 20:07:05 +00002285 struct callback_data data;
2286 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002287 open_db(p, 0);
jplyon672a1ed2003-05-11 20:07:05 +00002288 memcpy(&data, p, sizeof(data));
drhd8885442004-03-17 23:42:12 +00002289 data.showHeader = 1;
jplyon672a1ed2003-05-11 20:07:05 +00002290 data.mode = MODE_Column;
drhd8885442004-03-17 23:42:12 +00002291 data.colWidth[0] = 3;
2292 data.colWidth[1] = 15;
2293 data.colWidth[2] = 58;
drh0b2110c2004-10-26 00:08:10 +00002294 data.cnt = 0;
danielk19776f8a5032004-05-10 10:34:51 +00002295 sqlite3_exec(p->db, "PRAGMA database_list; ", callback, &data, &zErrMsg);
jplyon672a1ed2003-05-11 20:07:05 +00002296 if( zErrMsg ){
2297 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002298 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002299 rc = 1;
jplyon6a65bb32003-05-04 07:25:57 +00002300 }
2301 }else
2302
drhc2ce0be2014-05-29 12:36:14 +00002303 if( c=='d' && strncmp(azArg[0], "dump", n)==0 ){
drh05782482013-10-24 15:20:20 +00002304 open_db(p, 0);
drhf1dfc4f2009-09-23 15:51:35 +00002305 /* When playing back a "dump", the content might appear in an order
2306 ** which causes immediate foreign key constraints to be violated.
2307 ** So disable foreign-key constraint enforcement to prevent problems. */
drhc2ce0be2014-05-29 12:36:14 +00002308 if( nArg!=1 && nArg!=2 ){
2309 fprintf(stderr, "Usage: .dump ?LIKE-PATTERN?\n");
2310 rc = 1;
2311 goto meta_command_exit;
2312 }
drhf1dfc4f2009-09-23 15:51:35 +00002313 fprintf(p->out, "PRAGMA foreign_keys=OFF;\n");
drh33048c02001-10-01 14:29:22 +00002314 fprintf(p->out, "BEGIN TRANSACTION;\n");
drh45e29d82006-11-20 16:21:10 +00002315 p->writableSchema = 0;
drh56197952011-10-13 16:30:13 +00002316 sqlite3_exec(p->db, "SAVEPOINT dump; PRAGMA writable_schema=ON", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00002317 p->nErr = 0;
drh4c653a02000-06-07 01:27:47 +00002318 if( nArg==1 ){
drhdd3d4592004-08-30 01:54:05 +00002319 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00002320 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00002321 "WHERE sql NOT NULL AND type=='table' AND name!='sqlite_sequence'"
drh4f324762009-05-21 14:51:03 +00002322 );
2323 run_schema_dump_query(p,
2324 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00002325 "WHERE name=='sqlite_sequence'"
drh0b9a5942006-09-13 20:22:02 +00002326 );
drh2f464a02011-10-13 00:41:49 +00002327 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00002328 "SELECT sql FROM sqlite_master "
drh157e29a2009-05-21 15:15:00 +00002329 "WHERE sql NOT NULL AND type IN ('index','trigger','view')", 0
drha18c5682000-10-08 22:20:57 +00002330 );
drh4c653a02000-06-07 01:27:47 +00002331 }else{
2332 int i;
drhdd3d4592004-08-30 01:54:05 +00002333 for(i=1; i<nArg; i++){
danielk1977bc6ada42004-06-30 08:20:16 +00002334 zShellStatic = azArg[i];
drhdd3d4592004-08-30 01:54:05 +00002335 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00002336 "SELECT name, type, sql FROM sqlite_master "
drhdd3d4592004-08-30 01:54:05 +00002337 "WHERE tbl_name LIKE shellstatic() AND type=='table'"
drh2f464a02011-10-13 00:41:49 +00002338 " AND sql NOT NULL");
2339 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00002340 "SELECT sql FROM sqlite_master "
drh45e29d82006-11-20 16:21:10 +00002341 "WHERE sql NOT NULL"
2342 " AND type IN ('index','trigger','view')"
drh157e29a2009-05-21 15:15:00 +00002343 " AND tbl_name LIKE shellstatic()", 0
drh0b9a5942006-09-13 20:22:02 +00002344 );
danielk1977bc6ada42004-06-30 08:20:16 +00002345 zShellStatic = 0;
drh4c653a02000-06-07 01:27:47 +00002346 }
2347 }
drh45e29d82006-11-20 16:21:10 +00002348 if( p->writableSchema ){
drh56197952011-10-13 16:30:13 +00002349 fprintf(p->out, "PRAGMA writable_schema=OFF;\n");
drh45e29d82006-11-20 16:21:10 +00002350 p->writableSchema = 0;
2351 }
drh56197952011-10-13 16:30:13 +00002352 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
2353 sqlite3_exec(p->db, "RELEASE dump;", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00002354 fprintf(p->out, p->nErr ? "ROLLBACK; -- due to errors\n" : "COMMIT;\n");
drh4c653a02000-06-07 01:27:47 +00002355 }else
drh75897232000-05-29 14:26:00 +00002356
drhc2ce0be2014-05-29 12:36:14 +00002357 if( c=='e' && strncmp(azArg[0], "echo", n)==0 ){
2358 if( nArg==2 ){
2359 p->echoOn = booleanValue(azArg[1]);
2360 }else{
2361 fprintf(stderr, "Usage: .echo on|off\n");
2362 rc = 1;
2363 }
drhdaffd0e2001-04-11 14:28:42 +00002364 }else
2365
drhc2ce0be2014-05-29 12:36:14 +00002366 if( c=='e' && strncmp(azArg[0], "eqp", n)==0 ){
2367 if( nArg==2 ){
2368 p->autoEQP = booleanValue(azArg[1]);
2369 }else{
2370 fprintf(stderr, "Usage: .eqp on|off\n");
2371 rc = 1;
2372 }
drhefbf3b12014-02-28 20:47:24 +00002373 }else
2374
drhd3ac7d92013-01-25 18:33:43 +00002375 if( c=='e' && strncmp(azArg[0], "exit", n)==0 ){
drh348d19c2013-06-03 12:47:43 +00002376 if( nArg>1 && (rc = (int)integerValue(azArg[1]))!=0 ) exit(rc);
drh47ad6842006-11-08 12:25:42 +00002377 rc = 2;
drh75897232000-05-29 14:26:00 +00002378 }else
2379
drhc2ce0be2014-05-29 12:36:14 +00002380 if( c=='e' && strncmp(azArg[0], "explain", n)==0 ){
drhc28490c2006-10-26 14:25:58 +00002381 int val = nArg>=2 ? booleanValue(azArg[1]) : 1;
persicom7e2dfdd2002-04-18 02:46:52 +00002382 if(val == 1) {
2383 if(!p->explainPrev.valid) {
2384 p->explainPrev.valid = 1;
2385 p->explainPrev.mode = p->mode;
2386 p->explainPrev.showHeader = p->showHeader;
2387 memcpy(p->explainPrev.colWidth,p->colWidth,sizeof(p->colWidth));
2388 }
2389 /* We could put this code under the !p->explainValid
2390 ** condition so that it does not execute if we are already in
2391 ** explain mode. However, always executing it allows us an easy
2392 ** was to reset to explain mode in case the user previously
2393 ** did an .explain followed by a .width, .mode or .header
2394 ** command.
2395 */
danielk19770d78bae2008-01-03 07:09:48 +00002396 p->mode = MODE_Explain;
persicom7e2dfdd2002-04-18 02:46:52 +00002397 p->showHeader = 1;
drhac68ced2013-11-27 13:24:18 +00002398 memset(p->colWidth,0,sizeof(p->colWidth));
danielk19770d78bae2008-01-03 07:09:48 +00002399 p->colWidth[0] = 4; /* addr */
drh60a713c2008-01-21 16:22:45 +00002400 p->colWidth[1] = 13; /* opcode */
2401 p->colWidth[2] = 4; /* P1 */
2402 p->colWidth[3] = 4; /* P2 */
2403 p->colWidth[4] = 4; /* P3 */
2404 p->colWidth[5] = 13; /* P4 */
danielk19770d78bae2008-01-03 07:09:48 +00002405 p->colWidth[6] = 2; /* P5 */
drh60a713c2008-01-21 16:22:45 +00002406 p->colWidth[7] = 13; /* Comment */
persicom7e2dfdd2002-04-18 02:46:52 +00002407 }else if (p->explainPrev.valid) {
2408 p->explainPrev.valid = 0;
2409 p->mode = p->explainPrev.mode;
2410 p->showHeader = p->explainPrev.showHeader;
2411 memcpy(p->colWidth,p->explainPrev.colWidth,sizeof(p->colWidth));
2412 }
drh75897232000-05-29 14:26:00 +00002413 }else
2414
drhc2ce0be2014-05-29 12:36:14 +00002415 if( c=='h' && strncmp(azArg[0], "headers", n)==0 ){
2416 if( nArg==2 ){
2417 p->showHeader = booleanValue(azArg[1]);
2418 }else{
2419 fprintf(stderr, "Usage: .headers on|off\n");
2420 rc = 1;
shaneb320ccd2009-10-21 03:42:58 +00002421 }
drh75897232000-05-29 14:26:00 +00002422 }else
2423
drhc2ce0be2014-05-29 12:36:14 +00002424 if( c=='h' && strncmp(azArg[0], "help", n)==0 ){
2425 fprintf(p->out, "%s", zHelp);
2426 }else
2427
2428 if( c=='i' && strncmp(azArg[0], "import", n)==0 ){
drh01f37542014-05-31 15:43:33 +00002429 char *zTable; /* Insert data into this table */
2430 char *zFile; /* Name of file to extra content from */
shane916f9612009-10-23 00:37:15 +00002431 sqlite3_stmt *pStmt = NULL; /* A statement */
drhfeac5f82004-08-01 00:10:45 +00002432 int nCol; /* Number of columns in the table */
2433 int nByte; /* Number of bytes in an SQL string */
2434 int i, j; /* Loop counters */
drh2d463112013-08-06 14:36:36 +00002435 int needCommit; /* True to COMMIT or ROLLBACK at end */
drhfeac5f82004-08-01 00:10:45 +00002436 int nSep; /* Number of bytes in p->separator[] */
2437 char *zSql; /* An SQL statement */
drhdb95f682013-06-26 22:46:00 +00002438 CSVReader sCsv; /* Reader context */
drh5bde8162013-06-27 14:07:53 +00002439 int (*xCloser)(FILE*); /* Procedure to close th3 connection */
drhfeac5f82004-08-01 00:10:45 +00002440
drhc2ce0be2014-05-29 12:36:14 +00002441 if( nArg!=3 ){
2442 fprintf(stderr, "Usage: .import FILE TABLE\n");
2443 goto meta_command_exit;
2444 }
drh01f37542014-05-31 15:43:33 +00002445 zFile = azArg[1];
2446 zTable = azArg[2];
drhdb95f682013-06-26 22:46:00 +00002447 seenInterrupt = 0;
2448 memset(&sCsv, 0, sizeof(sCsv));
drh05782482013-10-24 15:20:20 +00002449 open_db(p, 0);
drh4f21c4a2008-12-10 22:15:00 +00002450 nSep = strlen30(p->separator);
drhfeac5f82004-08-01 00:10:45 +00002451 if( nSep==0 ){
shane916f9612009-10-23 00:37:15 +00002452 fprintf(stderr, "Error: non-null separator required for import\n");
2453 return 1;
drhfeac5f82004-08-01 00:10:45 +00002454 }
drhdb95f682013-06-26 22:46:00 +00002455 if( nSep>1 ){
2456 fprintf(stderr, "Error: multi-character separators not allowed"
2457 " for import\n");
2458 return 1;
2459 }
drh5bde8162013-06-27 14:07:53 +00002460 sCsv.zFile = zFile;
2461 sCsv.nLine = 1;
2462 if( sCsv.zFile[0]=='|' ){
2463 sCsv.in = popen(sCsv.zFile+1, "r");
2464 sCsv.zFile = "<pipe>";
2465 xCloser = pclose;
2466 }else{
2467 sCsv.in = fopen(sCsv.zFile, "rb");
2468 xCloser = fclose;
2469 }
drhdb95f682013-06-26 22:46:00 +00002470 if( sCsv.in==0 ){
drh5bde8162013-06-27 14:07:53 +00002471 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
drhdb95f682013-06-26 22:46:00 +00002472 return 1;
2473 }
2474 sCsv.cSeparator = p->separator[0];
drh7b075e32011-09-28 01:10:00 +00002475 zSql = sqlite3_mprintf("SELECT * FROM %s", zTable);
shane916f9612009-10-23 00:37:15 +00002476 if( zSql==0 ){
2477 fprintf(stderr, "Error: out of memory\n");
drh5bde8162013-06-27 14:07:53 +00002478 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002479 return 1;
2480 }
drh4f21c4a2008-12-10 22:15:00 +00002481 nByte = strlen30(zSql);
drhc7181902014-02-27 15:04:13 +00002482 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
dan6a8ac852014-05-26 18:27:12 +00002483 csv_append_char(&sCsv, 0); /* To ensure sCsv.z is allocated */
drhdb95f682013-06-26 22:46:00 +00002484 if( rc && sqlite3_strglob("no such table: *", sqlite3_errmsg(db))==0 ){
2485 char *zCreate = sqlite3_mprintf("CREATE TABLE %s", zTable);
2486 char cSep = '(';
2487 while( csv_read_one_field(&sCsv) ){
2488 zCreate = sqlite3_mprintf("%z%c\n \"%s\" TEXT", zCreate, cSep, sCsv.z);
2489 cSep = ',';
2490 if( sCsv.cTerm!=sCsv.cSeparator ) break;
2491 }
drh5bde8162013-06-27 14:07:53 +00002492 if( cSep=='(' ){
2493 sqlite3_free(zCreate);
2494 sqlite3_free(sCsv.z);
2495 xCloser(sCsv.in);
2496 fprintf(stderr,"%s: empty file\n", sCsv.zFile);
2497 return 1;
2498 }
drhdb95f682013-06-26 22:46:00 +00002499 zCreate = sqlite3_mprintf("%z\n)", zCreate);
2500 rc = sqlite3_exec(p->db, zCreate, 0, 0, 0);
2501 sqlite3_free(zCreate);
2502 if( rc ){
2503 fprintf(stderr, "CREATE TABLE %s(...) failed: %s\n", zTable,
2504 sqlite3_errmsg(db));
2505 sqlite3_free(sCsv.z);
drh5bde8162013-06-27 14:07:53 +00002506 xCloser(sCsv.in);
drhdb95f682013-06-26 22:46:00 +00002507 return 1;
2508 }
drhc7181902014-02-27 15:04:13 +00002509 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
drhdb95f682013-06-26 22:46:00 +00002510 }
drhfeac5f82004-08-01 00:10:45 +00002511 sqlite3_free(zSql);
2512 if( rc ){
shane916f9612009-10-23 00:37:15 +00002513 if (pStmt) sqlite3_finalize(pStmt);
drhfeac5f82004-08-01 00:10:45 +00002514 fprintf(stderr,"Error: %s\n", sqlite3_errmsg(db));
drh5bde8162013-06-27 14:07:53 +00002515 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002516 return 1;
drhfeac5f82004-08-01 00:10:45 +00002517 }
shane916f9612009-10-23 00:37:15 +00002518 nCol = sqlite3_column_count(pStmt);
drhfeac5f82004-08-01 00:10:45 +00002519 sqlite3_finalize(pStmt);
shane916f9612009-10-23 00:37:15 +00002520 pStmt = 0;
shane9bd1b442009-10-23 01:27:39 +00002521 if( nCol==0 ) return 0; /* no columns, no error */
drhdb95f682013-06-26 22:46:00 +00002522 zSql = sqlite3_malloc( nByte*2 + 20 + nCol*2 );
shane916f9612009-10-23 00:37:15 +00002523 if( zSql==0 ){
2524 fprintf(stderr, "Error: out of memory\n");
drh5bde8162013-06-27 14:07:53 +00002525 xCloser(sCsv.in);
shane916f9612009-10-23 00:37:15 +00002526 return 1;
2527 }
drhdb95f682013-06-26 22:46:00 +00002528 sqlite3_snprintf(nByte+20, zSql, "INSERT INTO \"%w\" VALUES(?", zTable);
drh4f21c4a2008-12-10 22:15:00 +00002529 j = strlen30(zSql);
drhfeac5f82004-08-01 00:10:45 +00002530 for(i=1; i<nCol; i++){
2531 zSql[j++] = ',';
2532 zSql[j++] = '?';
2533 }
2534 zSql[j++] = ')';
2535 zSql[j] = 0;
drhc7181902014-02-27 15:04:13 +00002536 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
drhdb95f682013-06-26 22:46:00 +00002537 sqlite3_free(zSql);
drhfeac5f82004-08-01 00:10:45 +00002538 if( rc ){
2539 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(db));
shane916f9612009-10-23 00:37:15 +00002540 if (pStmt) sqlite3_finalize(pStmt);
drh5bde8162013-06-27 14:07:53 +00002541 xCloser(sCsv.in);
drh47ad6842006-11-08 12:25:42 +00002542 return 1;
drhfeac5f82004-08-01 00:10:45 +00002543 }
drh2d463112013-08-06 14:36:36 +00002544 needCommit = sqlite3_get_autocommit(db);
2545 if( needCommit ) sqlite3_exec(db, "BEGIN", 0, 0, 0);
drhdb95f682013-06-26 22:46:00 +00002546 do{
2547 int startLine = sCsv.nLine;
drhfeac5f82004-08-01 00:10:45 +00002548 for(i=0; i<nCol; i++){
drhdb95f682013-06-26 22:46:00 +00002549 char *z = csv_read_one_field(&sCsv);
2550 if( z==0 && i==0 ) break;
2551 sqlite3_bind_text(pStmt, i+1, z, -1, SQLITE_TRANSIENT);
2552 if( i<nCol-1 && sCsv.cTerm!=sCsv.cSeparator ){
2553 fprintf(stderr, "%s:%d: expected %d columns but found %d - "
2554 "filling the rest with NULL\n",
2555 sCsv.zFile, startLine, nCol, i+1);
2556 i++;
2557 while( i<nCol ){ sqlite3_bind_null(pStmt, i); i++; }
drh18f52e02012-01-16 16:56:31 +00002558 }
drhfeac5f82004-08-01 00:10:45 +00002559 }
drhdb95f682013-06-26 22:46:00 +00002560 if( sCsv.cTerm==sCsv.cSeparator ){
2561 do{
2562 csv_read_one_field(&sCsv);
2563 i++;
2564 }while( sCsv.cTerm==sCsv.cSeparator );
2565 fprintf(stderr, "%s:%d: expected %d columns but found %d - "
2566 "extras ignored\n",
2567 sCsv.zFile, startLine, nCol, i);
drhfeac5f82004-08-01 00:10:45 +00002568 }
drhdb95f682013-06-26 22:46:00 +00002569 if( i>=nCol ){
2570 sqlite3_step(pStmt);
2571 rc = sqlite3_reset(pStmt);
2572 if( rc!=SQLITE_OK ){
2573 fprintf(stderr, "%s:%d: INSERT failed: %s\n", sCsv.zFile, startLine,
2574 sqlite3_errmsg(db));
2575 }
2576 }
2577 }while( sCsv.cTerm!=EOF );
2578
drh5bde8162013-06-27 14:07:53 +00002579 xCloser(sCsv.in);
drhdb95f682013-06-26 22:46:00 +00002580 sqlite3_free(sCsv.z);
drhfeac5f82004-08-01 00:10:45 +00002581 sqlite3_finalize(pStmt);
drh2d463112013-08-06 14:36:36 +00002582 if( needCommit ) sqlite3_exec(db, "COMMIT", 0, 0, 0);
drhfeac5f82004-08-01 00:10:45 +00002583 }else
2584
drhc2ce0be2014-05-29 12:36:14 +00002585 if( c=='i' && strncmp(azArg[0], "indices", n)==0 ){
drh75897232000-05-29 14:26:00 +00002586 struct callback_data data;
2587 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002588 open_db(p, 0);
drh75897232000-05-29 14:26:00 +00002589 memcpy(&data, p, sizeof(data));
2590 data.showHeader = 0;
2591 data.mode = MODE_List;
shane86f5bdb2009-10-24 02:00:07 +00002592 if( nArg==1 ){
2593 rc = sqlite3_exec(p->db,
2594 "SELECT name FROM sqlite_master "
2595 "WHERE type='index' AND name NOT LIKE 'sqlite_%' "
2596 "UNION ALL "
2597 "SELECT name FROM sqlite_temp_master "
2598 "WHERE type='index' "
2599 "ORDER BY 1",
2600 callback, &data, &zErrMsg
2601 );
drhc2ce0be2014-05-29 12:36:14 +00002602 }else if( nArg==2 ){
shane86f5bdb2009-10-24 02:00:07 +00002603 zShellStatic = azArg[1];
2604 rc = sqlite3_exec(p->db,
2605 "SELECT name FROM sqlite_master "
2606 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2607 "UNION ALL "
2608 "SELECT name FROM sqlite_temp_master "
2609 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2610 "ORDER BY 1",
2611 callback, &data, &zErrMsg
2612 );
2613 zShellStatic = 0;
drhc2ce0be2014-05-29 12:36:14 +00002614 }else{
2615 fprintf(stderr, "Usage: .indices ?LIKE-PATTERN?\n");
2616 rc = 1;
2617 goto meta_command_exit;
shane86f5bdb2009-10-24 02:00:07 +00002618 }
drh75897232000-05-29 14:26:00 +00002619 if( zErrMsg ){
2620 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002621 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002622 rc = 1;
shane86f5bdb2009-10-24 02:00:07 +00002623 }else if( rc != SQLITE_OK ){
2624 fprintf(stderr,"Error: querying sqlite_master and sqlite_temp_master\n");
2625 rc = 1;
drh75897232000-05-29 14:26:00 +00002626 }
2627 }else
2628
drhae5e4452007-05-03 17:18:36 +00002629#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +00002630 if( c=='i' && strncmp(azArg[0], "iotrace", n)==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002631 extern void (*sqlite3IoTrace)(const char*, ...);
drhb0603412007-02-28 04:47:26 +00002632 if( iotrace && iotrace!=stdout ) fclose(iotrace);
2633 iotrace = 0;
2634 if( nArg<2 ){
mlcreech3a00f902008-03-04 17:45:01 +00002635 sqlite3IoTrace = 0;
drhb0603412007-02-28 04:47:26 +00002636 }else if( strcmp(azArg[1], "-")==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002637 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002638 iotrace = stdout;
2639 }else{
2640 iotrace = fopen(azArg[1], "w");
2641 if( iotrace==0 ){
shane9bd1b442009-10-23 01:27:39 +00002642 fprintf(stderr, "Error: cannot open \"%s\"\n", azArg[1]);
mlcreech3a00f902008-03-04 17:45:01 +00002643 sqlite3IoTrace = 0;
shane9bd1b442009-10-23 01:27:39 +00002644 rc = 1;
drhb0603412007-02-28 04:47:26 +00002645 }else{
mlcreech3a00f902008-03-04 17:45:01 +00002646 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002647 }
2648 }
2649 }else
drhae5e4452007-05-03 17:18:36 +00002650#endif
drhb0603412007-02-28 04:47:26 +00002651
drh70df4fe2006-06-13 15:12:21 +00002652#ifndef SQLITE_OMIT_LOAD_EXTENSION
drhc2ce0be2014-05-29 12:36:14 +00002653 if( c=='l' && strncmp(azArg[0], "load", n)==0 ){
drh1e397f82006-06-08 15:28:43 +00002654 const char *zFile, *zProc;
2655 char *zErrMsg = 0;
drhc2ce0be2014-05-29 12:36:14 +00002656 if( nArg<2 ){
2657 fprintf(stderr, "Usage: .load FILE ?ENTRYPOINT?\n");
2658 rc = 1;
2659 goto meta_command_exit;
2660 }
drh1e397f82006-06-08 15:28:43 +00002661 zFile = azArg[1];
2662 zProc = nArg>=3 ? azArg[2] : 0;
drh05782482013-10-24 15:20:20 +00002663 open_db(p, 0);
drh1e397f82006-06-08 15:28:43 +00002664 rc = sqlite3_load_extension(p->db, zFile, zProc, &zErrMsg);
2665 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002666 fprintf(stderr, "Error: %s\n", zErrMsg);
drh1e397f82006-06-08 15:28:43 +00002667 sqlite3_free(zErrMsg);
drh47ad6842006-11-08 12:25:42 +00002668 rc = 1;
drh1e397f82006-06-08 15:28:43 +00002669 }
2670 }else
drh70df4fe2006-06-13 15:12:21 +00002671#endif
drh1e397f82006-06-08 15:28:43 +00002672
drhc2ce0be2014-05-29 12:36:14 +00002673 if( c=='l' && strncmp(azArg[0], "log", n)==0 ){
2674 if( nArg!=2 ){
2675 fprintf(stderr, "Usage: .log FILENAME\n");
2676 rc = 1;
2677 }else{
2678 const char *zFile = azArg[1];
2679 output_file_close(p->pLog);
2680 p->pLog = output_file_open(zFile);
2681 }
drh127f9d72010-02-23 01:47:00 +00002682 }else
2683
drhc2ce0be2014-05-29 12:36:14 +00002684 if( c=='m' && strncmp(azArg[0], "mode", n)==0 ){
2685 const char *zMode = nArg>=2 ? azArg[1] : "";
2686 int n2 = (int)strlen(zMode);
2687 int c2 = zMode[0];
2688 if( c2=='l' && n2>2 && strncmp(azArg[1],"lines",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002689 p->mode = MODE_Line;
drhc2ce0be2014-05-29 12:36:14 +00002690 }else if( c2=='c' && strncmp(azArg[1],"columns",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002691 p->mode = MODE_Column;
drhc2ce0be2014-05-29 12:36:14 +00002692 }else if( c2=='l' && n2>2 && strncmp(azArg[1],"list",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002693 p->mode = MODE_List;
drhc2ce0be2014-05-29 12:36:14 +00002694 }else if( c2=='h' && strncmp(azArg[1],"html",n2)==0 ){
drh1e5d0e92000-05-31 23:33:17 +00002695 p->mode = MODE_Html;
drhc2ce0be2014-05-29 12:36:14 +00002696 }else if( c2=='t' && strncmp(azArg[1],"tcl",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002697 p->mode = MODE_Tcl;
mistachkin585dcb22012-12-04 00:23:43 +00002698 sqlite3_snprintf(sizeof(p->separator), p->separator, " ");
drhc2ce0be2014-05-29 12:36:14 +00002699 }else if( c2=='c' && strncmp(azArg[1],"csv",n2)==0 ){
drh8e64d1c2004-10-07 00:32:39 +00002700 p->mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00002701 sqlite3_snprintf(sizeof(p->separator), p->separator, ",");
drhc2ce0be2014-05-29 12:36:14 +00002702 }else if( c2=='t' && strncmp(azArg[1],"tabs",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002703 p->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00002704 sqlite3_snprintf(sizeof(p->separator), p->separator, "\t");
drhc2ce0be2014-05-29 12:36:14 +00002705 }else if( c2=='i' && strncmp(azArg[1],"insert",n2)==0 ){
drh28bd4bc2000-06-15 15:57:22 +00002706 p->mode = MODE_Insert;
drhc2ce0be2014-05-29 12:36:14 +00002707 set_table_name(p, nArg>=3 ? azArg[2] : "table");
drhdaffd0e2001-04-11 14:28:42 +00002708 }else {
shane9bd1b442009-10-23 01:27:39 +00002709 fprintf(stderr,"Error: mode should be one of: "
drhfeac5f82004-08-01 00:10:45 +00002710 "column csv html insert line list tabs tcl\n");
shane9bd1b442009-10-23 01:27:39 +00002711 rc = 1;
drh75897232000-05-29 14:26:00 +00002712 }
2713 }else
2714
drhc2ce0be2014-05-29 12:36:14 +00002715 if( c=='n' && strncmp(azArg[0], "nullvalue", n)==0 ){
2716 if( nArg==2 ){
2717 sqlite3_snprintf(sizeof(p->nullvalue), p->nullvalue,
2718 "%.*s", (int)ArraySize(p->nullvalue)-1, azArg[1]);
2719 }else{
2720 fprintf(stderr, "Usage: .nullvalue STRING\n");
shanehe2aa9d72009-11-06 17:20:17 +00002721 rc = 1;
2722 }
2723 }else
2724
drh05782482013-10-24 15:20:20 +00002725 if( c=='o' && strncmp(azArg[0], "open", n)==0 && n>=2 ){
2726 sqlite3 *savedDb = p->db;
2727 const char *zSavedFilename = p->zDbFilename;
2728 char *zNewFilename = 0;
2729 p->db = 0;
2730 if( nArg>=2 ){
2731 p->zDbFilename = zNewFilename = sqlite3_mprintf("%s", azArg[1]);
2732 }
2733 open_db(p, 1);
2734 if( p->db!=0 ){
2735 sqlite3_close(savedDb);
2736 sqlite3_free(p->zFreeOnClose);
2737 p->zFreeOnClose = zNewFilename;
2738 }else{
2739 sqlite3_free(zNewFilename);
2740 p->db = savedDb;
2741 p->zDbFilename = zSavedFilename;
2742 }
2743 }else
2744
drhc2ce0be2014-05-29 12:36:14 +00002745 if( c=='o'
2746 && (strncmp(azArg[0], "output", n)==0 || strncmp(azArg[0], "once", n)==0)
2747 ){
2748 const char *zFile = nArg>=2 ? azArg[1] : "stdout";
2749 if( nArg>2 ){
2750 fprintf(stderr, "Usage: .%s FILE\n", azArg[0]);
2751 rc = 1;
2752 goto meta_command_exit;
drh75897232000-05-29 14:26:00 +00002753 }
drhc2ce0be2014-05-29 12:36:14 +00002754 if( n>1 && strncmp(azArg[0], "once", n)==0 ){
2755 if( nArg<2 ){
2756 fprintf(stderr, "Usage: .once FILE\n");
2757 rc = 1;
2758 goto meta_command_exit;
2759 }
2760 p->outCount = 2;
2761 }else{
2762 p->outCount = 0;
2763 }
2764 output_reset(p);
2765 if( zFile[0]=='|' ){
2766 p->out = popen(zFile + 1, "w");
drhe1da8fa2012-03-30 00:05:57 +00002767 if( p->out==0 ){
drhc2ce0be2014-05-29 12:36:14 +00002768 fprintf(stderr,"Error: cannot open pipe \"%s\"\n", zFile + 1);
drhe1da8fa2012-03-30 00:05:57 +00002769 p->out = stdout;
2770 rc = 1;
2771 }else{
drhc2ce0be2014-05-29 12:36:14 +00002772 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
drhe1da8fa2012-03-30 00:05:57 +00002773 }
drh75897232000-05-29 14:26:00 +00002774 }else{
drhc2ce0be2014-05-29 12:36:14 +00002775 p->out = output_file_open(zFile);
drh75897232000-05-29 14:26:00 +00002776 if( p->out==0 ){
drhc2ce0be2014-05-29 12:36:14 +00002777 if( strcmp(zFile,"off")!=0 ){
2778 fprintf(stderr,"Error: cannot write to \"%s\"\n", zFile);
drh42f64e52012-04-04 16:56:23 +00002779 }
drh75897232000-05-29 14:26:00 +00002780 p->out = stdout;
shane9bd1b442009-10-23 01:27:39 +00002781 rc = 1;
persicom7e2dfdd2002-04-18 02:46:52 +00002782 } else {
drhc2ce0be2014-05-29 12:36:14 +00002783 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", zFile);
drh75897232000-05-29 14:26:00 +00002784 }
2785 }
2786 }else
2787
drh078b1fd2012-09-21 13:40:02 +00002788 if( c=='p' && n>=3 && strncmp(azArg[0], "print", n)==0 ){
2789 int i;
2790 for(i=1; i<nArg; i++){
2791 if( i>1 ) fprintf(p->out, " ");
2792 fprintf(p->out, "%s", azArg[i]);
2793 }
2794 fprintf(p->out, "\n");
2795 }else
2796
drhc2ce0be2014-05-29 12:36:14 +00002797 if( c=='p' && strncmp(azArg[0], "prompt", n)==0 ){
persicom7e2dfdd2002-04-18 02:46:52 +00002798 if( nArg >= 2) {
2799 strncpy(mainPrompt,azArg[1],(int)ArraySize(mainPrompt)-1);
2800 }
2801 if( nArg >= 3) {
2802 strncpy(continuePrompt,azArg[2],(int)ArraySize(continuePrompt)-1);
2803 }
2804 }else
2805
drhc2ce0be2014-05-29 12:36:14 +00002806 if( c=='q' && strncmp(azArg[0], "quit", n)==0 ){
drh47ad6842006-11-08 12:25:42 +00002807 rc = 2;
persicom7e2dfdd2002-04-18 02:46:52 +00002808 }else
2809
drhc2ce0be2014-05-29 12:36:14 +00002810 if( c=='r' && n>=3 && strncmp(azArg[0], "read", n)==0 ){
2811 FILE *alt;
2812 if( nArg!=2 ){
2813 fprintf(stderr, "Usage: .read FILE\n");
2814 rc = 1;
2815 goto meta_command_exit;
2816 }
2817 alt = fopen(azArg[1], "rb");
drhdaffd0e2001-04-11 14:28:42 +00002818 if( alt==0 ){
shane9bd1b442009-10-23 01:27:39 +00002819 fprintf(stderr,"Error: cannot open \"%s\"\n", azArg[1]);
2820 rc = 1;
drhdaffd0e2001-04-11 14:28:42 +00002821 }else{
shane9bd1b442009-10-23 01:27:39 +00002822 rc = process_input(p, alt);
drhdaffd0e2001-04-11 14:28:42 +00002823 fclose(alt);
2824 }
2825 }else
2826
drhc2ce0be2014-05-29 12:36:14 +00002827 if( c=='r' && n>=3 && strncmp(azArg[0], "restore", n)==0 ){
drh9ff849f2009-02-04 20:55:57 +00002828 const char *zSrcFile;
2829 const char *zDb;
2830 sqlite3 *pSrc;
2831 sqlite3_backup *pBackup;
drhdc2c4912009-02-04 22:46:47 +00002832 int nTimeout = 0;
2833
drh9ff849f2009-02-04 20:55:57 +00002834 if( nArg==2 ){
2835 zSrcFile = azArg[1];
2836 zDb = "main";
drhc2ce0be2014-05-29 12:36:14 +00002837 }else if( nArg==3 ){
drh9ff849f2009-02-04 20:55:57 +00002838 zSrcFile = azArg[2];
2839 zDb = azArg[1];
drhc2ce0be2014-05-29 12:36:14 +00002840 }else{
2841 fprintf(stderr, "Usage: .restore ?DB? FILE\n");
2842 rc = 1;
2843 goto meta_command_exit;
drh9ff849f2009-02-04 20:55:57 +00002844 }
2845 rc = sqlite3_open(zSrcFile, &pSrc);
2846 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002847 fprintf(stderr, "Error: cannot open \"%s\"\n", zSrcFile);
drh9ff849f2009-02-04 20:55:57 +00002848 sqlite3_close(pSrc);
2849 return 1;
2850 }
drh05782482013-10-24 15:20:20 +00002851 open_db(p, 0);
drh9ff849f2009-02-04 20:55:57 +00002852 pBackup = sqlite3_backup_init(p->db, zDb, pSrc, "main");
2853 if( pBackup==0 ){
2854 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
2855 sqlite3_close(pSrc);
2856 return 1;
2857 }
drhdc2c4912009-02-04 22:46:47 +00002858 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK
2859 || rc==SQLITE_BUSY ){
2860 if( rc==SQLITE_BUSY ){
2861 if( nTimeout++ >= 3 ) break;
2862 sqlite3_sleep(100);
drh9ff849f2009-02-04 20:55:57 +00002863 }
2864 }
2865 sqlite3_backup_finish(pBackup);
2866 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00002867 rc = 0;
drhdc2c4912009-02-04 22:46:47 +00002868 }else if( rc==SQLITE_BUSY || rc==SQLITE_LOCKED ){
shane9bd1b442009-10-23 01:27:39 +00002869 fprintf(stderr, "Error: source database is busy\n");
2870 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002871 }else{
2872 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
shane9bd1b442009-10-23 01:27:39 +00002873 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002874 }
2875 sqlite3_close(pSrc);
2876 }else
2877
drhc2ce0be2014-05-29 12:36:14 +00002878 if( c=='s' && strncmp(azArg[0], "schema", n)==0 ){
drh75897232000-05-29 14:26:00 +00002879 struct callback_data data;
2880 char *zErrMsg = 0;
drh05782482013-10-24 15:20:20 +00002881 open_db(p, 0);
drh75897232000-05-29 14:26:00 +00002882 memcpy(&data, p, sizeof(data));
2883 data.showHeader = 0;
drhe3710332000-09-29 13:30:53 +00002884 data.mode = MODE_Semi;
drhc2ce0be2014-05-29 12:36:14 +00002885 if( nArg==2 ){
drhc8d74412004-08-31 23:41:26 +00002886 int i;
drhf0693c82011-10-11 20:41:54 +00002887 for(i=0; azArg[1][i]; i++) azArg[1][i] = ToLower(azArg[1][i]);
drhc8d74412004-08-31 23:41:26 +00002888 if( strcmp(azArg[1],"sqlite_master")==0 ){
drha18c5682000-10-08 22:20:57 +00002889 char *new_argv[2], *new_colv[2];
2890 new_argv[0] = "CREATE TABLE sqlite_master (\n"
2891 " type text,\n"
2892 " name text,\n"
2893 " tbl_name text,\n"
drhadbca9c2001-09-27 15:11:53 +00002894 " rootpage integer,\n"
drha18c5682000-10-08 22:20:57 +00002895 " sql text\n"
2896 ")";
2897 new_argv[1] = 0;
2898 new_colv[0] = "sql";
2899 new_colv[1] = 0;
2900 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002901 rc = SQLITE_OK;
drhc8d74412004-08-31 23:41:26 +00002902 }else if( strcmp(azArg[1],"sqlite_temp_master")==0 ){
drhe0bc4042002-06-25 01:09:11 +00002903 char *new_argv[2], *new_colv[2];
2904 new_argv[0] = "CREATE TEMP TABLE sqlite_temp_master (\n"
2905 " type text,\n"
2906 " name text,\n"
2907 " tbl_name text,\n"
2908 " rootpage integer,\n"
2909 " sql text\n"
2910 ")";
2911 new_argv[1] = 0;
2912 new_colv[0] = "sql";
2913 new_colv[1] = 0;
2914 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002915 rc = SQLITE_OK;
drha18c5682000-10-08 22:20:57 +00002916 }else{
danielk1977bc6ada42004-06-30 08:20:16 +00002917 zShellStatic = azArg[1];
shane9bd1b442009-10-23 01:27:39 +00002918 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002919 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002920 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002921 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002922 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh6ac7a582011-11-04 00:35:56 +00002923 "WHERE lower(tbl_name) LIKE shellstatic()"
2924 " AND type!='meta' AND sql NOTNULL "
drh1ba00292013-05-06 21:01:06 +00002925 "ORDER BY rowid",
danielk1977bc6ada42004-06-30 08:20:16 +00002926 callback, &data, &zErrMsg);
2927 zShellStatic = 0;
drha18c5682000-10-08 22:20:57 +00002928 }
drhc2ce0be2014-05-29 12:36:14 +00002929 }else if( nArg==1 ){
shane9bd1b442009-10-23 01:27:39 +00002930 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002931 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002932 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002933 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002934 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh0c356672005-09-10 22:40:53 +00002935 "WHERE type!='meta' AND sql NOTNULL AND name NOT LIKE 'sqlite_%'"
drh1ba00292013-05-06 21:01:06 +00002936 "ORDER BY rowid",
drha18c5682000-10-08 22:20:57 +00002937 callback, &data, &zErrMsg
2938 );
drhc2ce0be2014-05-29 12:36:14 +00002939 }else{
2940 fprintf(stderr, "Usage: .schema ?LIKE-PATTERN?\n");
2941 rc = 1;
2942 goto meta_command_exit;
drh75897232000-05-29 14:26:00 +00002943 }
drh75897232000-05-29 14:26:00 +00002944 if( zErrMsg ){
2945 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002946 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002947 rc = 1;
2948 }else if( rc != SQLITE_OK ){
2949 fprintf(stderr,"Error: querying schema information\n");
2950 rc = 1;
2951 }else{
2952 rc = 0;
drh75897232000-05-29 14:26:00 +00002953 }
2954 }else
2955
drh340f5822013-06-27 13:01:21 +00002956#ifdef SQLITE_DEBUG
drh348d19c2013-06-03 12:47:43 +00002957 /* Undocumented commands for internal testing. Subject to change
2958 ** without notice. */
2959 if( c=='s' && n>=10 && strncmp(azArg[0], "selftest-", 9)==0 ){
2960 if( strncmp(azArg[0]+9, "boolean", n-9)==0 ){
2961 int i, v;
2962 for(i=1; i<nArg; i++){
2963 v = booleanValue(azArg[i]);
2964 fprintf(p->out, "%s: %d 0x%x\n", azArg[i], v, v);
2965 }
2966 }
2967 if( strncmp(azArg[0]+9, "integer", n-9)==0 ){
2968 int i; sqlite3_int64 v;
2969 for(i=1; i<nArg; i++){
drh340f5822013-06-27 13:01:21 +00002970 char zBuf[200];
drh348d19c2013-06-03 12:47:43 +00002971 v = integerValue(azArg[i]);
drhc2ce0be2014-05-29 12:36:14 +00002972 sqlite3_snprintf(sizeof(zBuf),zBuf,"%s: %lld 0x%llx\n", azArg[i],v,v);
drh340f5822013-06-27 13:01:21 +00002973 fprintf(p->out, "%s", zBuf);
drh348d19c2013-06-03 12:47:43 +00002974 }
2975 }
2976 }else
drh340f5822013-06-27 13:01:21 +00002977#endif
drh348d19c2013-06-03 12:47:43 +00002978
drhc2ce0be2014-05-29 12:36:14 +00002979 if( c=='s' && strncmp(azArg[0], "separator", n)==0 ){
2980 if( nArg==2 ){
2981 sqlite3_snprintf(sizeof(p->separator), p->separator,
2982 "%.*s", (int)sizeof(p->separator)-1, azArg[1]);
2983 }else{
2984 fprintf(stderr, "Usage: .separator STRING\n");
2985 rc = 1;
2986 }
drh75897232000-05-29 14:26:00 +00002987 }else
2988
drh62cdde52014-05-28 20:22:28 +00002989 if( c=='s'
2990 && (strncmp(azArg[0], "shell", n)==0 || strncmp(azArg[0],"system",n)==0)
drh62cdde52014-05-28 20:22:28 +00002991 ){
2992 char *zCmd;
2993 int i;
drhc2ce0be2014-05-29 12:36:14 +00002994 if( nArg<2 ){
2995 fprintf(stderr, "Usage: .system COMMAND\n");
2996 rc = 1;
2997 goto meta_command_exit;
2998 }
drhdcb3e3d2014-05-29 03:17:29 +00002999 zCmd = sqlite3_mprintf(strchr(azArg[1],' ')==0?"%s":"\"%s\"", azArg[1]);
drh62cdde52014-05-28 20:22:28 +00003000 for(i=2; i<nArg; i++){
drhdcb3e3d2014-05-29 03:17:29 +00003001 zCmd = sqlite3_mprintf(strchr(azArg[i],' ')==0?"%z %s":"%z \"%s\"",
3002 zCmd, azArg[i]);
drh62cdde52014-05-28 20:22:28 +00003003 }
drh3c4461f2014-05-29 20:39:59 +00003004 (void)system(zCmd);
drh62cdde52014-05-28 20:22:28 +00003005 sqlite3_free(zCmd);
3006 }else
3007
drhc2ce0be2014-05-29 12:36:14 +00003008 if( c=='s' && strncmp(azArg[0], "show", n)==0 ){
persicom7e2dfdd2002-04-18 02:46:52 +00003009 int i;
drhc2ce0be2014-05-29 12:36:14 +00003010 if( nArg!=1 ){
3011 fprintf(stderr, "Usage: .show\n");
3012 rc = 1;
3013 goto meta_command_exit;
3014 }
persicom7e2dfdd2002-04-18 02:46:52 +00003015 fprintf(p->out,"%9.9s: %s\n","echo", p->echoOn ? "on" : "off");
drhefbf3b12014-02-28 20:47:24 +00003016 fprintf(p->out,"%9.9s: %s\n","eqp", p->autoEQP ? "on" : "off");
drh67505e72002-04-19 12:34:06 +00003017 fprintf(p->out,"%9.9s: %s\n","explain", p->explainPrev.valid ? "on" :"off");
drhdd45df82002-04-18 12:39:03 +00003018 fprintf(p->out,"%9.9s: %s\n","headers", p->showHeader ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00003019 fprintf(p->out,"%9.9s: %s\n","mode", modeDescr[p->mode]);
drhfeac5f82004-08-01 00:10:45 +00003020 fprintf(p->out,"%9.9s: ", "nullvalue");
3021 output_c_string(p->out, p->nullvalue);
3022 fprintf(p->out, "\n");
drh67505e72002-04-19 12:34:06 +00003023 fprintf(p->out,"%9.9s: %s\n","output",
drh4f21c4a2008-12-10 22:15:00 +00003024 strlen30(p->outfile) ? p->outfile : "stdout");
drhfeac5f82004-08-01 00:10:45 +00003025 fprintf(p->out,"%9.9s: ", "separator");
3026 output_c_string(p->out, p->separator);
3027 fprintf(p->out, "\n");
shaneh642d8b82010-07-28 16:05:34 +00003028 fprintf(p->out,"%9.9s: %s\n","stats", p->statsOn ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00003029 fprintf(p->out,"%9.9s: ","width");
3030 for (i=0;i<(int)ArraySize(p->colWidth) && p->colWidth[i] != 0;i++) {
drhfeac5f82004-08-01 00:10:45 +00003031 fprintf(p->out,"%d ",p->colWidth[i]);
persicom7e2dfdd2002-04-18 02:46:52 +00003032 }
drhfeac5f82004-08-01 00:10:45 +00003033 fprintf(p->out,"\n");
persicom7e2dfdd2002-04-18 02:46:52 +00003034 }else
3035
drhc2ce0be2014-05-29 12:36:14 +00003036 if( c=='s' && strncmp(azArg[0], "stats", n)==0 ){
3037 if( nArg==2 ){
3038 p->statsOn = booleanValue(azArg[1]);
3039 }else{
3040 fprintf(stderr, "Usage: .stats on|off\n");
3041 rc = 1;
3042 }
shaneh642d8b82010-07-28 16:05:34 +00003043 }else
3044
drhc2ce0be2014-05-29 12:36:14 +00003045 if( c=='t' && n>1 && strncmp(azArg[0], "tables", n)==0 ){
drh98781232012-04-23 12:38:05 +00003046 sqlite3_stmt *pStmt;
drhe3710332000-09-29 13:30:53 +00003047 char **azResult;
drh98781232012-04-23 12:38:05 +00003048 int nRow, nAlloc;
3049 char *zSql = 0;
3050 int ii;
drh05782482013-10-24 15:20:20 +00003051 open_db(p, 0);
drh98781232012-04-23 12:38:05 +00003052 rc = sqlite3_prepare_v2(p->db, "PRAGMA database_list", -1, &pStmt, 0);
3053 if( rc ) return rc;
3054 zSql = sqlite3_mprintf(
3055 "SELECT name FROM sqlite_master"
3056 " WHERE type IN ('table','view')"
3057 " AND name NOT LIKE 'sqlite_%%'"
3058 " AND name LIKE ?1");
3059 while( sqlite3_step(pStmt)==SQLITE_ROW ){
3060 const char *zDbName = (const char*)sqlite3_column_text(pStmt, 1);
3061 if( zDbName==0 || strcmp(zDbName,"main")==0 ) continue;
3062 if( strcmp(zDbName,"temp")==0 ){
3063 zSql = sqlite3_mprintf(
3064 "%z UNION ALL "
3065 "SELECT 'temp.' || name FROM sqlite_temp_master"
3066 " WHERE type IN ('table','view')"
3067 " AND name NOT LIKE 'sqlite_%%'"
3068 " AND name LIKE ?1", zSql);
3069 }else{
3070 zSql = sqlite3_mprintf(
3071 "%z UNION ALL "
3072 "SELECT '%q.' || name FROM \"%w\".sqlite_master"
3073 " WHERE type IN ('table','view')"
3074 " AND name NOT LIKE 'sqlite_%%'"
3075 " AND name LIKE ?1", zSql, zDbName, zDbName);
3076 }
drha50da102000-08-08 20:19:09 +00003077 }
drh98781232012-04-23 12:38:05 +00003078 sqlite3_finalize(pStmt);
3079 zSql = sqlite3_mprintf("%z ORDER BY 1", zSql);
3080 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
3081 sqlite3_free(zSql);
3082 if( rc ) return rc;
3083 nRow = nAlloc = 0;
3084 azResult = 0;
3085 if( nArg>1 ){
3086 sqlite3_bind_text(pStmt, 1, azArg[1], -1, SQLITE_TRANSIENT);
shane9bd1b442009-10-23 01:27:39 +00003087 }else{
drh98781232012-04-23 12:38:05 +00003088 sqlite3_bind_text(pStmt, 1, "%", -1, SQLITE_STATIC);
3089 }
3090 while( sqlite3_step(pStmt)==SQLITE_ROW ){
3091 if( nRow>=nAlloc ){
3092 char **azNew;
3093 int n = nAlloc*2 + 10;
3094 azNew = sqlite3_realloc(azResult, sizeof(azResult[0])*n);
3095 if( azNew==0 ){
3096 fprintf(stderr, "Error: out of memory\n");
3097 break;
3098 }
3099 nAlloc = n;
3100 azResult = azNew;
3101 }
3102 azResult[nRow] = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
3103 if( azResult[nRow] ) nRow++;
3104 }
3105 sqlite3_finalize(pStmt);
3106 if( nRow>0 ){
drhe3710332000-09-29 13:30:53 +00003107 int len, maxlen = 0;
3108 int i, j;
3109 int nPrintCol, nPrintRow;
drh98781232012-04-23 12:38:05 +00003110 for(i=0; i<nRow; i++){
drh4f21c4a2008-12-10 22:15:00 +00003111 len = strlen30(azResult[i]);
drhe3710332000-09-29 13:30:53 +00003112 if( len>maxlen ) maxlen = len;
3113 }
3114 nPrintCol = 80/(maxlen+2);
3115 if( nPrintCol<1 ) nPrintCol = 1;
3116 nPrintRow = (nRow + nPrintCol - 1)/nPrintCol;
3117 for(i=0; i<nPrintRow; i++){
drh98781232012-04-23 12:38:05 +00003118 for(j=i; j<nRow; j+=nPrintRow){
3119 char *zSp = j<nPrintRow ? "" : " ";
drh151b7d52013-05-06 20:28:54 +00003120 fprintf(p->out, "%s%-*s", zSp, maxlen, azResult[j] ? azResult[j] : "");
drhe3710332000-09-29 13:30:53 +00003121 }
drh151b7d52013-05-06 20:28:54 +00003122 fprintf(p->out, "\n");
drhe3710332000-09-29 13:30:53 +00003123 }
3124 }
drh98781232012-04-23 12:38:05 +00003125 for(ii=0; ii<nRow; ii++) sqlite3_free(azResult[ii]);
3126 sqlite3_free(azResult);
drh75897232000-05-29 14:26:00 +00003127 }else
3128
shaneh96887e12011-02-10 21:08:58 +00003129 if( c=='t' && n>=8 && strncmp(azArg[0], "testctrl", n)==0 && nArg>=2 ){
drhd416fe72011-03-17 16:45:50 +00003130 static const struct {
3131 const char *zCtrlName; /* Name of a test-control option */
3132 int ctrlCode; /* Integer code for that option */
3133 } aCtrl[] = {
3134 { "prng_save", SQLITE_TESTCTRL_PRNG_SAVE },
3135 { "prng_restore", SQLITE_TESTCTRL_PRNG_RESTORE },
3136 { "prng_reset", SQLITE_TESTCTRL_PRNG_RESET },
3137 { "bitvec_test", SQLITE_TESTCTRL_BITVEC_TEST },
3138 { "fault_install", SQLITE_TESTCTRL_FAULT_INSTALL },
3139 { "benign_malloc_hooks", SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS },
3140 { "pending_byte", SQLITE_TESTCTRL_PENDING_BYTE },
3141 { "assert", SQLITE_TESTCTRL_ASSERT },
3142 { "always", SQLITE_TESTCTRL_ALWAYS },
3143 { "reserve", SQLITE_TESTCTRL_RESERVE },
3144 { "optimizations", SQLITE_TESTCTRL_OPTIMIZATIONS },
3145 { "iskeyword", SQLITE_TESTCTRL_ISKEYWORD },
drhd416fe72011-03-17 16:45:50 +00003146 { "scratchmalloc", SQLITE_TESTCTRL_SCRATCHMALLOC },
drh2cf4acb2014-04-18 00:06:02 +00003147 { "byteorder", SQLITE_TESTCTRL_BYTEORDER },
drhd416fe72011-03-17 16:45:50 +00003148 };
shaneh96887e12011-02-10 21:08:58 +00003149 int testctrl = -1;
3150 int rc = 0;
drhd416fe72011-03-17 16:45:50 +00003151 int i, n;
drh05782482013-10-24 15:20:20 +00003152 open_db(p, 0);
shaneh96887e12011-02-10 21:08:58 +00003153
drhd416fe72011-03-17 16:45:50 +00003154 /* convert testctrl text option to value. allow any unique prefix
3155 ** of the option name, or a numerical value. */
shanehcef83682011-04-07 03:41:01 +00003156 n = strlen30(azArg[1]);
drhfcd71b62011-04-05 22:08:24 +00003157 for(i=0; i<(int)(sizeof(aCtrl)/sizeof(aCtrl[0])); i++){
drhd416fe72011-03-17 16:45:50 +00003158 if( strncmp(azArg[1], aCtrl[i].zCtrlName, n)==0 ){
3159 if( testctrl<0 ){
3160 testctrl = aCtrl[i].ctrlCode;
3161 }else{
drhb07028f2011-10-14 21:49:18 +00003162 fprintf(stderr, "ambiguous option name: \"%s\"\n", azArg[1]);
drhd416fe72011-03-17 16:45:50 +00003163 testctrl = -1;
3164 break;
3165 }
3166 }
3167 }
drh348d19c2013-06-03 12:47:43 +00003168 if( testctrl<0 ) testctrl = (int)integerValue(azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003169 if( (testctrl<SQLITE_TESTCTRL_FIRST) || (testctrl>SQLITE_TESTCTRL_LAST) ){
3170 fprintf(stderr,"Error: invalid testctrl option: %s\n", azArg[1]);
3171 }else{
3172 switch(testctrl){
3173
3174 /* sqlite3_test_control(int, db, int) */
3175 case SQLITE_TESTCTRL_OPTIMIZATIONS:
3176 case SQLITE_TESTCTRL_RESERVE:
3177 if( nArg==3 ){
3178 int opt = (int)strtol(azArg[2], 0, 0);
3179 rc = sqlite3_test_control(testctrl, p->db, opt);
drh151b7d52013-05-06 20:28:54 +00003180 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003181 } else {
drhd416fe72011-03-17 16:45:50 +00003182 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
3183 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003184 }
3185 break;
3186
3187 /* sqlite3_test_control(int) */
drh2cf4acb2014-04-18 00:06:02 +00003188 case SQLITE_TESTCTRL_PRNG_SAVE:
3189 case SQLITE_TESTCTRL_PRNG_RESTORE:
shaneh96887e12011-02-10 21:08:58 +00003190 case SQLITE_TESTCTRL_PRNG_RESET:
drh2cf4acb2014-04-18 00:06:02 +00003191 case SQLITE_TESTCTRL_BYTEORDER:
shaneh96887e12011-02-10 21:08:58 +00003192 if( nArg==2 ){
3193 rc = sqlite3_test_control(testctrl);
drh151b7d52013-05-06 20:28:54 +00003194 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003195 } else {
3196 fprintf(stderr,"Error: testctrl %s takes no options\n", azArg[1]);
3197 }
3198 break;
3199
3200 /* sqlite3_test_control(int, uint) */
3201 case SQLITE_TESTCTRL_PENDING_BYTE:
3202 if( nArg==3 ){
drhaf664332013-07-18 20:28:29 +00003203 unsigned int opt = (unsigned int)integerValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00003204 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003205 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003206 } else {
drhd416fe72011-03-17 16:45:50 +00003207 fprintf(stderr,"Error: testctrl %s takes a single unsigned"
3208 " int option\n", azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003209 }
3210 break;
3211
3212 /* sqlite3_test_control(int, int) */
3213 case SQLITE_TESTCTRL_ASSERT:
3214 case SQLITE_TESTCTRL_ALWAYS:
3215 if( nArg==3 ){
drh348d19c2013-06-03 12:47:43 +00003216 int opt = booleanValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00003217 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003218 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003219 } else {
drhd416fe72011-03-17 16:45:50 +00003220 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
3221 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003222 }
3223 break;
3224
3225 /* sqlite3_test_control(int, char *) */
3226#ifdef SQLITE_N_KEYWORD
3227 case SQLITE_TESTCTRL_ISKEYWORD:
3228 if( nArg==3 ){
3229 const char *opt = azArg[2];
3230 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00003231 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00003232 } else {
drhd416fe72011-03-17 16:45:50 +00003233 fprintf(stderr,"Error: testctrl %s takes a single char * option\n",
3234 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003235 }
3236 break;
3237#endif
3238
3239 case SQLITE_TESTCTRL_BITVEC_TEST:
3240 case SQLITE_TESTCTRL_FAULT_INSTALL:
3241 case SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS:
3242 case SQLITE_TESTCTRL_SCRATCHMALLOC:
3243 default:
drhd416fe72011-03-17 16:45:50 +00003244 fprintf(stderr,"Error: CLI support for testctrl %s not implemented\n",
3245 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00003246 break;
3247 }
3248 }
3249 }else
3250
drhc2ce0be2014-05-29 12:36:14 +00003251 if( c=='t' && n>4 && strncmp(azArg[0], "timeout", n)==0 ){
drh05782482013-10-24 15:20:20 +00003252 open_db(p, 0);
drhc2ce0be2014-05-29 12:36:14 +00003253 sqlite3_busy_timeout(p->db, nArg>=2 ? (int)integerValue(azArg[1]) : 0);
shanehe2aa9d72009-11-06 17:20:17 +00003254 }else
3255
drhc2ce0be2014-05-29 12:36:14 +00003256 if( c=='t' && n>=5 && strncmp(azArg[0], "timer", n)==0 ){
3257 if( nArg==2 ){
3258 enableTimer = booleanValue(azArg[1]);
3259 if( enableTimer && !HAS_TIMER ){
3260 fprintf(stderr, "Error: timer not available on this system.\n");
3261 enableTimer = 0;
3262 }
3263 }else{
3264 fprintf(stderr, "Usage: .timer on|off\n");
3265 rc = 1;
3266 }
shanehe2aa9d72009-11-06 17:20:17 +00003267 }else
3268
drhc2ce0be2014-05-29 12:36:14 +00003269 if( c=='t' && strncmp(azArg[0], "trace", n)==0 ){
drh05782482013-10-24 15:20:20 +00003270 open_db(p, 0);
drh42f64e52012-04-04 16:56:23 +00003271 output_file_close(p->traceOut);
drhc2ce0be2014-05-29 12:36:14 +00003272 if( nArg!=2 ){
3273 fprintf(stderr, "Usage: .trace FILE|off\n");
3274 rc = 1;
3275 goto meta_command_exit;
3276 }
drh42f64e52012-04-04 16:56:23 +00003277 p->traceOut = output_file_open(azArg[1]);
drhbbb0be82012-06-27 16:12:27 +00003278#if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT)
drh42f64e52012-04-04 16:56:23 +00003279 if( p->traceOut==0 ){
3280 sqlite3_trace(p->db, 0, 0);
3281 }else{
3282 sqlite3_trace(p->db, sql_trace_callback, p->traceOut);
3283 }
3284#endif
3285 }else
3286
drh9fd301b2011-06-03 13:28:22 +00003287 if( c=='v' && strncmp(azArg[0], "version", n)==0 ){
drh151b7d52013-05-06 20:28:54 +00003288 fprintf(p->out, "SQLite %s %s\n" /*extra-version-info*/,
drh9fd301b2011-06-03 13:28:22 +00003289 sqlite3_libversion(), sqlite3_sourceid());
3290 }else
3291
drhde60fc22011-12-14 17:53:36 +00003292 if( c=='v' && strncmp(azArg[0], "vfsname", n)==0 ){
3293 const char *zDbName = nArg==2 ? azArg[1] : "main";
3294 char *zVfsName = 0;
3295 if( p->db ){
3296 sqlite3_file_control(p->db, zDbName, SQLITE_FCNTL_VFSNAME, &zVfsName);
3297 if( zVfsName ){
drh151b7d52013-05-06 20:28:54 +00003298 fprintf(p->out, "%s\n", zVfsName);
drhde60fc22011-12-14 17:53:36 +00003299 sqlite3_free(zVfsName);
3300 }
3301 }
3302 }else
3303
drhcef4fc82012-09-21 22:50:45 +00003304#if defined(SQLITE_DEBUG) && defined(SQLITE_ENABLE_WHERETRACE)
3305 if( c=='w' && strncmp(azArg[0], "wheretrace", n)==0 ){
3306 extern int sqlite3WhereTrace;
drhc2ce0be2014-05-29 12:36:14 +00003307 sqlite3WhereTrace = nArg>=2 ? booleanValue(azArg[1]) : 0xff;
drhcef4fc82012-09-21 22:50:45 +00003308 }else
3309#endif
3310
drhc2ce0be2014-05-29 12:36:14 +00003311 if( c=='w' && strncmp(azArg[0], "width", n)==0 ){
drh75897232000-05-29 14:26:00 +00003312 int j;
drh43617e92006-03-06 20:55:46 +00003313 assert( nArg<=ArraySize(azArg) );
drh75897232000-05-29 14:26:00 +00003314 for(j=1; j<nArg && j<ArraySize(p->colWidth); j++){
drh348d19c2013-06-03 12:47:43 +00003315 p->colWidth[j-1] = (int)integerValue(azArg[j]);
drh75897232000-05-29 14:26:00 +00003316 }
3317 }else
3318
3319 {
shane9bd1b442009-10-23 01:27:39 +00003320 fprintf(stderr, "Error: unknown command or invalid arguments: "
drh67505e72002-04-19 12:34:06 +00003321 " \"%s\". Enter \".help\" for help\n", azArg[0]);
shane9bd1b442009-10-23 01:27:39 +00003322 rc = 1;
drh75897232000-05-29 14:26:00 +00003323 }
drh67505e72002-04-19 12:34:06 +00003324
drhc2ce0be2014-05-29 12:36:14 +00003325meta_command_exit:
3326 if( p->outCount ){
3327 p->outCount--;
3328 if( p->outCount==0 ) output_reset(p);
3329 }
drh67505e72002-04-19 12:34:06 +00003330 return rc;
drh75897232000-05-29 14:26:00 +00003331}
3332
drh67505e72002-04-19 12:34:06 +00003333/*
drh91a66392007-09-07 01:12:32 +00003334** Return TRUE if a semicolon occurs anywhere in the first N characters
3335** of string z[].
drh324ccef2003-02-05 14:06:20 +00003336*/
drh9f099fd2013-08-06 14:01:46 +00003337static int line_contains_semicolon(const char *z, int N){
drh91a66392007-09-07 01:12:32 +00003338 int i;
3339 for(i=0; i<N; i++){ if( z[i]==';' ) return 1; }
3340 return 0;
drh324ccef2003-02-05 14:06:20 +00003341}
3342
3343/*
drh70c7a4b2003-04-26 03:03:06 +00003344** Test to see if a line consists entirely of whitespace.
3345*/
3346static int _all_whitespace(const char *z){
3347 for(; *z; z++){
drhf0693c82011-10-11 20:41:54 +00003348 if( IsSpace(z[0]) ) continue;
drh70c7a4b2003-04-26 03:03:06 +00003349 if( *z=='/' && z[1]=='*' ){
3350 z += 2;
3351 while( *z && (*z!='*' || z[1]!='/') ){ z++; }
3352 if( *z==0 ) return 0;
3353 z++;
3354 continue;
3355 }
3356 if( *z=='-' && z[1]=='-' ){
3357 z += 2;
3358 while( *z && *z!='\n' ){ z++; }
3359 if( *z==0 ) return 1;
3360 continue;
3361 }
3362 return 0;
3363 }
3364 return 1;
3365}
3366
3367/*
drha9b17162003-04-29 18:01:28 +00003368** Return TRUE if the line typed in is an SQL command terminator other
3369** than a semi-colon. The SQL Server style "go" command is understood
3370** as is the Oracle "/".
3371*/
drh9f099fd2013-08-06 14:01:46 +00003372static int line_is_command_terminator(const char *zLine){
drhf0693c82011-10-11 20:41:54 +00003373 while( IsSpace(zLine[0]) ){ zLine++; };
drh233a5312008-12-18 22:25:13 +00003374 if( zLine[0]=='/' && _all_whitespace(&zLine[1]) ){
3375 return 1; /* Oracle */
3376 }
drhf0693c82011-10-11 20:41:54 +00003377 if( ToLower(zLine[0])=='g' && ToLower(zLine[1])=='o'
drhc8d74412004-08-31 23:41:26 +00003378 && _all_whitespace(&zLine[2]) ){
drha9b17162003-04-29 18:01:28 +00003379 return 1; /* SQL Server */
3380 }
3381 return 0;
3382}
3383
3384/*
drh233a5312008-12-18 22:25:13 +00003385** Return true if zSql is a complete SQL statement. Return false if it
3386** ends in the middle of a string literal or C-style comment.
3387*/
drh9f099fd2013-08-06 14:01:46 +00003388static int line_is_complete(char *zSql, int nSql){
drh233a5312008-12-18 22:25:13 +00003389 int rc;
3390 if( zSql==0 ) return 1;
3391 zSql[nSql] = ';';
3392 zSql[nSql+1] = 0;
3393 rc = sqlite3_complete(zSql);
3394 zSql[nSql] = 0;
3395 return rc;
3396}
3397
3398/*
drh67505e72002-04-19 12:34:06 +00003399** Read input from *in and process it. If *in==0 then input
3400** is interactive - the user is typing it it. Otherwise, input
3401** is coming from a file or device. A prompt is issued and history
3402** is saved only if input is interactive. An interrupt signal will
3403** cause this routine to exit immediately, unless input is interactive.
drhc28490c2006-10-26 14:25:58 +00003404**
3405** Return the number of errors.
drh67505e72002-04-19 12:34:06 +00003406*/
drhc28490c2006-10-26 14:25:58 +00003407static int process_input(struct callback_data *p, FILE *in){
drh9f099fd2013-08-06 14:01:46 +00003408 char *zLine = 0; /* A single input line */
3409 char *zSql = 0; /* Accumulated SQL text */
3410 int nLine; /* Length of current line */
3411 int nSql = 0; /* Bytes of zSql[] used */
3412 int nAlloc = 0; /* Allocated zSql[] space */
3413 int nSqlPrior = 0; /* Bytes of zSql[] used by prior line */
3414 char *zErrMsg; /* Error message returned */
3415 int rc; /* Error code */
3416 int errCnt = 0; /* Number of errors seen */
3417 int lineno = 0; /* Current line number */
3418 int startline = 0; /* Line number for start of current input */
drhc49f44e2006-10-26 18:15:42 +00003419
3420 while( errCnt==0 || !bail_on_error || (in==0 && stdin_is_interactive) ){
3421 fflush(p->out);
drh9f099fd2013-08-06 14:01:46 +00003422 zLine = one_input_line(in, zLine, nSql>0);
drhc49f44e2006-10-26 18:15:42 +00003423 if( zLine==0 ){
drh9b8d3572012-04-21 11:33:39 +00003424 /* End of input */
3425 if( stdin_is_interactive ) printf("\n");
3426 break;
drhc49f44e2006-10-26 18:15:42 +00003427 }
drh67505e72002-04-19 12:34:06 +00003428 if( seenInterrupt ){
3429 if( in!=0 ) break;
3430 seenInterrupt = 0;
3431 }
drhc28490c2006-10-26 14:25:58 +00003432 lineno++;
drh849a9d92013-12-21 15:46:06 +00003433 if( nSql==0 && _all_whitespace(zLine) ){
3434 if( p->echoOn ) printf("%s\n", zLine);
3435 continue;
3436 }
drh2af0b2d2002-02-21 02:25:02 +00003437 if( zLine && zLine[0]=='.' && nSql==0 ){
shaneb9fc17d2009-10-22 21:23:35 +00003438 if( p->echoOn ) printf("%s\n", zLine);
drhc49f44e2006-10-26 18:15:42 +00003439 rc = do_meta_command(zLine, p);
shane916f9612009-10-23 00:37:15 +00003440 if( rc==2 ){ /* exit requested */
drh47ad6842006-11-08 12:25:42 +00003441 break;
3442 }else if( rc ){
drhc49f44e2006-10-26 18:15:42 +00003443 errCnt++;
3444 }
drhdaffd0e2001-04-11 14:28:42 +00003445 continue;
3446 }
drh9f099fd2013-08-06 14:01:46 +00003447 if( line_is_command_terminator(zLine) && line_is_complete(zSql, nSql) ){
drh5bb3eb92007-05-04 13:15:55 +00003448 memcpy(zLine,";",2);
drha9b17162003-04-29 18:01:28 +00003449 }
drh9f099fd2013-08-06 14:01:46 +00003450 nLine = strlen30(zLine);
3451 if( nSql+nLine+2>=nAlloc ){
3452 nAlloc = nSql+nLine+100;
3453 zSql = realloc(zSql, nAlloc);
drhdaffd0e2001-04-11 14:28:42 +00003454 if( zSql==0 ){
drh9f099fd2013-08-06 14:01:46 +00003455 fprintf(stderr, "Error: out of memory\n");
drhdaffd0e2001-04-11 14:28:42 +00003456 exit(1);
3457 }
drhdaffd0e2001-04-11 14:28:42 +00003458 }
drh9f099fd2013-08-06 14:01:46 +00003459 nSqlPrior = nSql;
3460 if( nSql==0 ){
3461 int i;
3462 for(i=0; zLine[i] && IsSpace(zLine[i]); i++){}
drh77dfd5b2013-08-19 11:15:48 +00003463 assert( nAlloc>0 && zSql!=0 );
drh9f099fd2013-08-06 14:01:46 +00003464 memcpy(zSql, zLine+i, nLine+1-i);
3465 startline = lineno;
3466 nSql = nLine-i;
3467 }else{
3468 zSql[nSql++] = '\n';
3469 memcpy(zSql+nSql, zLine, nLine+1);
3470 nSql += nLine;
3471 }
3472 if( nSql && line_contains_semicolon(&zSql[nSqlPrior], nSql-nSqlPrior)
drh91a66392007-09-07 01:12:32 +00003473 && sqlite3_complete(zSql) ){
drhdaffd0e2001-04-11 14:28:42 +00003474 p->cnt = 0;
drh05782482013-10-24 15:20:20 +00003475 open_db(p, 0);
drh3b1a9882007-11-02 12:53:03 +00003476 BEGIN_TIMER;
shane626a6e42009-10-22 17:30:15 +00003477 rc = shell_exec(p->db, zSql, shell_callback, p, &zErrMsg);
drh3b1a9882007-11-02 12:53:03 +00003478 END_TIMER;
drh7f953e22002-07-13 17:33:45 +00003479 if( rc || zErrMsg ){
drhc28490c2006-10-26 14:25:58 +00003480 char zPrefix[100];
3481 if( in!=0 || !stdin_is_interactive ){
drh5bb3eb92007-05-04 13:15:55 +00003482 sqlite3_snprintf(sizeof(zPrefix), zPrefix,
shane9bd1b442009-10-23 01:27:39 +00003483 "Error: near line %d:", startline);
drhc28490c2006-10-26 14:25:58 +00003484 }else{
shane9bd1b442009-10-23 01:27:39 +00003485 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "Error:");
drhc28490c2006-10-26 14:25:58 +00003486 }
drh7f953e22002-07-13 17:33:45 +00003487 if( zErrMsg!=0 ){
shaned2bed1c2009-10-21 03:56:54 +00003488 fprintf(stderr, "%s %s\n", zPrefix, zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00003489 sqlite3_free(zErrMsg);
drh7f953e22002-07-13 17:33:45 +00003490 zErrMsg = 0;
3491 }else{
shaned2bed1c2009-10-21 03:56:54 +00003492 fprintf(stderr, "%s %s\n", zPrefix, sqlite3_errmsg(p->db));
drh7f953e22002-07-13 17:33:45 +00003493 }
drhc49f44e2006-10-26 18:15:42 +00003494 errCnt++;
drhdaffd0e2001-04-11 14:28:42 +00003495 }
drhdaffd0e2001-04-11 14:28:42 +00003496 nSql = 0;
drhc2ce0be2014-05-29 12:36:14 +00003497 if( p->outCount ){
3498 output_reset(p);
3499 p->outCount = 0;
3500 }
drh9f099fd2013-08-06 14:01:46 +00003501 }else if( nSql && _all_whitespace(zSql) ){
drh849a9d92013-12-21 15:46:06 +00003502 if( p->echoOn ) printf("%s\n", zSql);
drh7a411f42013-04-17 17:33:17 +00003503 nSql = 0;
drhdaffd0e2001-04-11 14:28:42 +00003504 }
3505 }
drh9f099fd2013-08-06 14:01:46 +00003506 if( nSql ){
drhd416fe72011-03-17 16:45:50 +00003507 if( !_all_whitespace(zSql) ){
3508 fprintf(stderr, "Error: incomplete SQL: %s\n", zSql);
3509 }
drhdaffd0e2001-04-11 14:28:42 +00003510 free(zSql);
3511 }
danielk19772ac27622007-07-03 05:31:16 +00003512 free(zLine);
drh4d15a0d2012-12-01 20:21:22 +00003513 return errCnt>0;
drhdaffd0e2001-04-11 14:28:42 +00003514}
3515
drh67505e72002-04-19 12:34:06 +00003516/*
3517** Return a pathname which is the user's home directory. A
drh85e72432012-04-11 11:38:53 +00003518** 0 return indicates an error of some kind.
drh67505e72002-04-19 12:34:06 +00003519*/
3520static char *find_home_dir(void){
drh85e72432012-04-11 11:38:53 +00003521 static char *home_dir = NULL;
3522 if( home_dir ) return home_dir;
persicom7e2dfdd2002-04-18 02:46:52 +00003523
drh83905c92012-06-21 13:00:37 +00003524#if !defined(_WIN32) && !defined(WIN32) && !defined(_WIN32_WCE) && !defined(__RTP__) && !defined(_WRS_KERNEL)
mistachkinc8bde372012-06-18 08:00:56 +00003525 {
3526 struct passwd *pwent;
3527 uid_t uid = getuid();
3528 if( (pwent=getpwuid(uid)) != NULL) {
3529 home_dir = pwent->pw_dir;
3530 }
drh67505e72002-04-19 12:34:06 +00003531 }
3532#endif
3533
chw65d3c132007-11-12 21:09:10 +00003534#if defined(_WIN32_WCE)
3535 /* Windows CE (arm-wince-mingw32ce-gcc) does not provide getenv()
3536 */
drh85e72432012-04-11 11:38:53 +00003537 home_dir = "/";
chw65d3c132007-11-12 21:09:10 +00003538#else
3539
drh83905c92012-06-21 13:00:37 +00003540#if defined(_WIN32) || defined(WIN32)
drh164a1b62006-08-19 11:15:20 +00003541 if (!home_dir) {
3542 home_dir = getenv("USERPROFILE");
3543 }
3544#endif
3545
drh67505e72002-04-19 12:34:06 +00003546 if (!home_dir) {
3547 home_dir = getenv("HOME");
drh67505e72002-04-19 12:34:06 +00003548 }
3549
drh83905c92012-06-21 13:00:37 +00003550#if defined(_WIN32) || defined(WIN32)
drhe98d4fa2002-04-21 19:06:22 +00003551 if (!home_dir) {
drh164a1b62006-08-19 11:15:20 +00003552 char *zDrive, *zPath;
3553 int n;
3554 zDrive = getenv("HOMEDRIVE");
3555 zPath = getenv("HOMEPATH");
3556 if( zDrive && zPath ){
drh4f21c4a2008-12-10 22:15:00 +00003557 n = strlen30(zDrive) + strlen30(zPath) + 1;
drh164a1b62006-08-19 11:15:20 +00003558 home_dir = malloc( n );
3559 if( home_dir==0 ) return 0;
3560 sqlite3_snprintf(n, home_dir, "%s%s", zDrive, zPath);
3561 return home_dir;
3562 }
3563 home_dir = "c:\\";
drhe98d4fa2002-04-21 19:06:22 +00003564 }
3565#endif
3566
chw65d3c132007-11-12 21:09:10 +00003567#endif /* !_WIN32_WCE */
3568
drh67505e72002-04-19 12:34:06 +00003569 if( home_dir ){
drh4f21c4a2008-12-10 22:15:00 +00003570 int n = strlen30(home_dir) + 1;
drh5bb3eb92007-05-04 13:15:55 +00003571 char *z = malloc( n );
3572 if( z ) memcpy(z, home_dir, n);
drh67505e72002-04-19 12:34:06 +00003573 home_dir = z;
3574 }
drhe98d4fa2002-04-21 19:06:22 +00003575
drh67505e72002-04-19 12:34:06 +00003576 return home_dir;
3577}
3578
3579/*
3580** Read input from the file given by sqliterc_override. Or if that
3581** parameter is NULL, take input from ~/.sqliterc
shane9bd1b442009-10-23 01:27:39 +00003582**
3583** Returns the number of errors.
drh67505e72002-04-19 12:34:06 +00003584*/
shane9bd1b442009-10-23 01:27:39 +00003585static int process_sqliterc(
drh22fbcb82004-02-01 01:22:50 +00003586 struct callback_data *p, /* Configuration data */
3587 const char *sqliterc_override /* Name of config file. NULL to use default */
3588){
persicom7e2dfdd2002-04-18 02:46:52 +00003589 char *home_dir = NULL;
drh22fbcb82004-02-01 01:22:50 +00003590 const char *sqliterc = sqliterc_override;
drh43617e92006-03-06 20:55:46 +00003591 char *zBuf = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00003592 FILE *in = NULL;
shane9bd1b442009-10-23 01:27:39 +00003593 int rc = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00003594
3595 if (sqliterc == NULL) {
drh67505e72002-04-19 12:34:06 +00003596 home_dir = find_home_dir();
drhe98d4fa2002-04-21 19:06:22 +00003597 if( home_dir==0 ){
chw97185482008-11-17 08:05:31 +00003598#if !defined(__RTP__) && !defined(_WRS_KERNEL)
shane86f5bdb2009-10-24 02:00:07 +00003599 fprintf(stderr,"%s: Error: cannot locate your home directory\n", Argv0);
chw97185482008-11-17 08:05:31 +00003600#endif
shane9bd1b442009-10-23 01:27:39 +00003601 return 1;
drhe98d4fa2002-04-21 19:06:22 +00003602 }
drh2f3de322012-06-27 16:41:31 +00003603 sqlite3_initialize();
drh85e72432012-04-11 11:38:53 +00003604 zBuf = sqlite3_mprintf("%s/.sqliterc",home_dir);
3605 sqliterc = zBuf;
persicom7e2dfdd2002-04-18 02:46:52 +00003606 }
drha1f9b5e2004-02-14 16:31:02 +00003607 in = fopen(sqliterc,"rb");
drh22fbcb82004-02-01 01:22:50 +00003608 if( in ){
drhc28490c2006-10-26 14:25:58 +00003609 if( stdin_is_interactive ){
shane86f5bdb2009-10-24 02:00:07 +00003610 fprintf(stderr,"-- Loading resources from %s\n",sqliterc);
drh22fbcb82004-02-01 01:22:50 +00003611 }
shane9bd1b442009-10-23 01:27:39 +00003612 rc = process_input(p,in);
drhdd45df82002-04-18 12:39:03 +00003613 fclose(in);
persicom7e2dfdd2002-04-18 02:46:52 +00003614 }
drh85e72432012-04-11 11:38:53 +00003615 sqlite3_free(zBuf);
shane9bd1b442009-10-23 01:27:39 +00003616 return rc;
persicom7e2dfdd2002-04-18 02:46:52 +00003617}
3618
drh67505e72002-04-19 12:34:06 +00003619/*
drhe1e38c42003-05-04 18:30:59 +00003620** Show available command line options
3621*/
3622static const char zOptions[] =
drhc49f44e2006-10-26 18:15:42 +00003623 " -bail stop after hitting an error\n"
drhc49f44e2006-10-26 18:15:42 +00003624 " -batch force batch I/O\n"
drhe1e38c42003-05-04 18:30:59 +00003625 " -column set output mode to 'column'\n"
mistachkin6d81d752012-10-25 15:43:28 +00003626 " -cmd COMMAND run \"COMMAND\" before reading stdin\n"
drhc49f44e2006-10-26 18:15:42 +00003627 " -csv set output mode to 'csv'\n"
drhcc3b4f82012-02-07 14:13:50 +00003628 " -echo print commands before execution\n"
mistachkin6d81d752012-10-25 15:43:28 +00003629 " -init FILENAME read/process named file\n"
drhcc3b4f82012-02-07 14:13:50 +00003630 " -[no]header turn headers on or off\n"
drh98d312f2012-10-25 15:23:14 +00003631#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
3632 " -heap SIZE Size of heap for memsys3 or memsys5\n"
3633#endif
drhcc3b4f82012-02-07 14:13:50 +00003634 " -help show this message\n"
drhe1e38c42003-05-04 18:30:59 +00003635 " -html set output mode to HTML\n"
drhcc3b4f82012-02-07 14:13:50 +00003636 " -interactive force interactive I/O\n"
drhe1e38c42003-05-04 18:30:59 +00003637 " -line set output mode to 'line'\n"
3638 " -list set output mode to 'list'\n"
drh7d9f3942013-04-03 01:26:54 +00003639 " -mmap N default mmap size set to N\n"
drhcc3b4f82012-02-07 14:13:50 +00003640#ifdef SQLITE_ENABLE_MULTIPLEX
3641 " -multiplex enable the multiplexor VFS\n"
3642#endif
drh98d312f2012-10-25 15:23:14 +00003643 " -nullvalue TEXT set text string for NULL values. Default ''\n"
3644 " -separator SEP set output field separator. Default: '|'\n"
shaneh642d8b82010-07-28 16:05:34 +00003645 " -stats print memory stats before each finalize\n"
drhe1e38c42003-05-04 18:30:59 +00003646 " -version show SQLite version\n"
drha7e61d82011-03-12 17:02:57 +00003647 " -vfs NAME use NAME as the default VFS\n"
drh2b625e22011-03-16 17:05:28 +00003648#ifdef SQLITE_ENABLE_VFSTRACE
3649 " -vfstrace enable tracing of all VFS calls\n"
3650#endif
drhe1e38c42003-05-04 18:30:59 +00003651;
3652static void usage(int showDetail){
drh80e8be92006-08-29 12:04:19 +00003653 fprintf(stderr,
3654 "Usage: %s [OPTIONS] FILENAME [SQL]\n"
3655 "FILENAME is the name of an SQLite database. A new database is created\n"
3656 "if the file does not previously exist.\n", Argv0);
drhe1e38c42003-05-04 18:30:59 +00003657 if( showDetail ){
drh80e8be92006-08-29 12:04:19 +00003658 fprintf(stderr, "OPTIONS include:\n%s", zOptions);
drhe1e38c42003-05-04 18:30:59 +00003659 }else{
3660 fprintf(stderr, "Use the -help option for additional information\n");
3661 }
3662 exit(1);
3663}
3664
3665/*
drh67505e72002-04-19 12:34:06 +00003666** Initialize the state information in data
3667*/
drh0850b532006-01-31 19:31:43 +00003668static void main_init(struct callback_data *data) {
persicom7e2dfdd2002-04-18 02:46:52 +00003669 memset(data, 0, sizeof(*data));
3670 data->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00003671 memcpy(data->separator,"|", 2);
persicom7e2dfdd2002-04-18 02:46:52 +00003672 data->showHeader = 0;
drh52784bd2011-05-18 17:15:06 +00003673 sqlite3_config(SQLITE_CONFIG_URI, 1);
drh127f9d72010-02-23 01:47:00 +00003674 sqlite3_config(SQLITE_CONFIG_LOG, shellLog, data);
drh5bb3eb92007-05-04 13:15:55 +00003675 sqlite3_snprintf(sizeof(mainPrompt), mainPrompt,"sqlite> ");
3676 sqlite3_snprintf(sizeof(continuePrompt), continuePrompt," ...> ");
dan0f831772010-03-03 07:23:12 +00003677 sqlite3_config(SQLITE_CONFIG_SINGLETHREAD);
persicom7e2dfdd2002-04-18 02:46:52 +00003678}
3679
drh98d312f2012-10-25 15:23:14 +00003680/*
drh5c7976f2014-02-10 19:59:27 +00003681** Output text to the console in a font that attracts extra attention.
drh1247aa42014-02-10 19:27:05 +00003682*/
3683#ifdef _WIN32
drh5c7976f2014-02-10 19:59:27 +00003684static void printBold(const char *zText){
3685 HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
3686 CONSOLE_SCREEN_BUFFER_INFO defaultScreenInfo;
3687 GetConsoleScreenBufferInfo(out, &defaultScreenInfo);
3688 SetConsoleTextAttribute(out,
3689 FOREGROUND_RED|FOREGROUND_INTENSITY
3690 );
3691 printf("%s", zText);
3692 SetConsoleTextAttribute(out, defaultScreenInfo.wAttributes);
drh1247aa42014-02-10 19:27:05 +00003693}
3694#else
drh5c7976f2014-02-10 19:59:27 +00003695static void printBold(const char *zText){
3696 printf("\033[1m%s\033[0m", zText);
drh1247aa42014-02-10 19:27:05 +00003697}
3698#endif
3699
3700/*
drh98d312f2012-10-25 15:23:14 +00003701** Get the argument to an --option. Throw an error and die if no argument
3702** is available.
3703*/
3704static char *cmdline_option_value(int argc, char **argv, int i){
3705 if( i==argc ){
3706 fprintf(stderr, "%s: Error: missing argument to %s\n",
3707 argv[0], argv[argc-1]);
3708 exit(1);
3709 }
3710 return argv[i];
3711}
3712
drh75897232000-05-29 14:26:00 +00003713int main(int argc, char **argv){
drh75897232000-05-29 14:26:00 +00003714 char *zErrMsg = 0;
3715 struct callback_data data;
drh22fbcb82004-02-01 01:22:50 +00003716 const char *zInitFile = 0;
3717 char *zFirstCmd = 0;
drh44c2eb12003-04-30 11:38:26 +00003718 int i;
drhc28490c2006-10-26 14:25:58 +00003719 int rc = 0;
drhb3735912014-02-10 16:13:42 +00003720 int warnInmemoryDb = 0;
drh75897232000-05-29 14:26:00 +00003721
drh69b30ab2014-02-27 15:11:52 +00003722#if USE_SYSTEM_SQLITE+0!=1
drh52784bd2011-05-18 17:15:06 +00003723 if( strcmp(sqlite3_sourceid(),SQLITE_SOURCE_ID)!=0 ){
3724 fprintf(stderr, "SQLite header and source version mismatch\n%s\n%s\n",
3725 sqlite3_sourceid(), SQLITE_SOURCE_ID);
3726 exit(1);
3727 }
drhc7181902014-02-27 15:04:13 +00003728#endif
drhdaffd0e2001-04-11 14:28:42 +00003729 Argv0 = argv[0];
persicom7e2dfdd2002-04-18 02:46:52 +00003730 main_init(&data);
drhc28490c2006-10-26 14:25:58 +00003731 stdin_is_interactive = isatty(0);
persicom7e2dfdd2002-04-18 02:46:52 +00003732
drh44c2eb12003-04-30 11:38:26 +00003733 /* Make sure we have a valid signal handler early, before anything
3734 ** else is done.
3735 */
drh4c504392000-10-16 22:06:40 +00003736#ifdef SIGINT
3737 signal(SIGINT, interrupt_handler);
3738#endif
drh44c2eb12003-04-30 11:38:26 +00003739
drh22fbcb82004-02-01 01:22:50 +00003740 /* Do an initial pass through the command-line argument to locate
3741 ** the name of the database file, the name of the initialization file,
drh9c88d682010-12-17 14:03:01 +00003742 ** the size of the alternative malloc heap,
drh22fbcb82004-02-01 01:22:50 +00003743 ** and the first command to execute.
drh44c2eb12003-04-30 11:38:26 +00003744 */
drh98d312f2012-10-25 15:23:14 +00003745 for(i=1; i<argc; i++){
drhc28490c2006-10-26 14:25:58 +00003746 char *z;
drhc28490c2006-10-26 14:25:58 +00003747 z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003748 if( z[0]!='-' ){
3749 if( data.zDbFilename==0 ){
3750 data.zDbFilename = z;
3751 continue;
3752 }
3753 if( zFirstCmd==0 ){
3754 zFirstCmd = z;
3755 continue;
3756 }
3757 fprintf(stderr,"%s: Error: too many options: \"%s\"\n", Argv0, argv[i]);
3758 fprintf(stderr,"Use -help for a list of options.\n");
3759 return 1;
3760 }
drhcc3b4f82012-02-07 14:13:50 +00003761 if( z[1]=='-' ) z++;
3762 if( strcmp(z,"-separator")==0
3763 || strcmp(z,"-nullvalue")==0
3764 || strcmp(z,"-cmd")==0
3765 ){
drh98d312f2012-10-25 15:23:14 +00003766 (void)cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003767 }else if( strcmp(z,"-init")==0 ){
drh98d312f2012-10-25 15:23:14 +00003768 zInitFile = cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003769 }else if( strcmp(z,"-batch")==0 ){
drh98d312f2012-10-25 15:23:14 +00003770 /* Need to check for batch mode here to so we can avoid printing
3771 ** informational messages (like from process_sqliterc) before
3772 ** we do the actual processing of arguments later in a second pass.
3773 */
shanef69573d2009-10-24 02:06:14 +00003774 stdin_is_interactive = 0;
drhcc3b4f82012-02-07 14:13:50 +00003775 }else if( strcmp(z,"-heap")==0 ){
drhb07028f2011-10-14 21:49:18 +00003776#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
drh9c88d682010-12-17 14:03:01 +00003777 const char *zSize;
3778 sqlite3_int64 szHeap;
3779
drh98d312f2012-10-25 15:23:14 +00003780 zSize = cmdline_option_value(argc, argv, ++i);
drh7d9f3942013-04-03 01:26:54 +00003781 szHeap = integerValue(zSize);
drh9c88d682010-12-17 14:03:01 +00003782 if( szHeap>0x7fff0000 ) szHeap = 0x7fff0000;
drh9c88d682010-12-17 14:03:01 +00003783 sqlite3_config(SQLITE_CONFIG_HEAP, malloc((int)szHeap), (int)szHeap, 64);
3784#endif
drh97ae8ff2011-03-16 16:56:29 +00003785#ifdef SQLITE_ENABLE_VFSTRACE
drhcc3b4f82012-02-07 14:13:50 +00003786 }else if( strcmp(z,"-vfstrace")==0 ){
drh97ae8ff2011-03-16 16:56:29 +00003787 extern int vfstrace_register(
3788 const char *zTraceName,
3789 const char *zOldVfsName,
3790 int (*xOut)(const char*,void*),
3791 void *pOutArg,
3792 int makeDefault
3793 );
drh2b625e22011-03-16 17:05:28 +00003794 vfstrace_register("trace",0,(int(*)(const char*,void*))fputs,stderr,1);
drh97ae8ff2011-03-16 16:56:29 +00003795#endif
drh6f25e892011-07-08 17:02:57 +00003796#ifdef SQLITE_ENABLE_MULTIPLEX
drhcc3b4f82012-02-07 14:13:50 +00003797 }else if( strcmp(z,"-multiplex")==0 ){
drh6f25e892011-07-08 17:02:57 +00003798 extern int sqlite3_multiple_initialize(const char*,int);
3799 sqlite3_multiplex_initialize(0, 1);
3800#endif
drh7d9f3942013-04-03 01:26:54 +00003801 }else if( strcmp(z,"-mmap")==0 ){
drh9b4c59f2013-04-15 17:03:42 +00003802 sqlite3_int64 sz = integerValue(cmdline_option_value(argc,argv,++i));
3803 sqlite3_config(SQLITE_CONFIG_MMAP_SIZE, sz, sz);
drhcc3b4f82012-02-07 14:13:50 +00003804 }else if( strcmp(z,"-vfs")==0 ){
drh98d312f2012-10-25 15:23:14 +00003805 sqlite3_vfs *pVfs = sqlite3_vfs_find(cmdline_option_value(argc,argv,++i));
drha7e61d82011-03-12 17:02:57 +00003806 if( pVfs ){
3807 sqlite3_vfs_register(pVfs, 1);
3808 }else{
3809 fprintf(stderr, "no such VFS: \"%s\"\n", argv[i]);
3810 exit(1);
3811 }
drh44c2eb12003-04-30 11:38:26 +00003812 }
3813 }
drh98d312f2012-10-25 15:23:14 +00003814 if( data.zDbFilename==0 ){
danielk197703aded42004-11-22 05:26:27 +00003815#ifndef SQLITE_OMIT_MEMORYDB
drh22fbcb82004-02-01 01:22:50 +00003816 data.zDbFilename = ":memory:";
drh1247aa42014-02-10 19:27:05 +00003817 warnInmemoryDb = argc==1;
danielk197703aded42004-11-22 05:26:27 +00003818#else
shane86f5bdb2009-10-24 02:00:07 +00003819 fprintf(stderr,"%s: Error: no database filename specified\n", Argv0);
3820 return 1;
drh01b41712005-08-29 23:06:23 +00003821#endif
drhc7181902014-02-27 15:04:13 +00003822#ifdef SQLITE_SHELL_DBNAME_PROC
3823 { extern void SQLITE_SHELL_DBNAME_PROC(const char**);
3824 SQLITE_SHELL_DBNAME_PROC(&data.zDbFilename);
3825 warnInmemoryDb = 0; }
3826#endif
drh98d312f2012-10-25 15:23:14 +00003827 }
3828 data.out = stdout;
drh01b41712005-08-29 23:06:23 +00003829
drh44c2eb12003-04-30 11:38:26 +00003830 /* Go ahead and open the database file if it already exists. If the
3831 ** file does not exist, delay opening it. This prevents empty database
3832 ** files from being created if a user mistypes the database name argument
3833 ** to the sqlite command-line tool.
3834 */
drhc8d74412004-08-31 23:41:26 +00003835 if( access(data.zDbFilename, 0)==0 ){
drh05782482013-10-24 15:20:20 +00003836 open_db(&data, 0);
drh44c2eb12003-04-30 11:38:26 +00003837 }
3838
drh22fbcb82004-02-01 01:22:50 +00003839 /* Process the initialization file if there is one. If no -init option
3840 ** is given on the command line, look for a file named ~/.sqliterc and
3841 ** try to process it.
drh44c2eb12003-04-30 11:38:26 +00003842 */
shane86f5bdb2009-10-24 02:00:07 +00003843 rc = process_sqliterc(&data,zInitFile);
3844 if( rc>0 ){
3845 return rc;
3846 }
drh44c2eb12003-04-30 11:38:26 +00003847
drh22fbcb82004-02-01 01:22:50 +00003848 /* Make a second pass through the command-line argument and set
3849 ** options. This second pass is delayed until after the initialization
3850 ** file is processed so that the command-line arguments will override
3851 ** settings in the initialization file.
drh44c2eb12003-04-30 11:38:26 +00003852 */
drh98d312f2012-10-25 15:23:14 +00003853 for(i=1; i<argc; i++){
drh22fbcb82004-02-01 01:22:50 +00003854 char *z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003855 if( z[0]!='-' ) continue;
drhc28490c2006-10-26 14:25:58 +00003856 if( z[1]=='-' ){ z++; }
drh2e584cd2006-09-25 13:09:22 +00003857 if( strcmp(z,"-init")==0 ){
drh22fbcb82004-02-01 01:22:50 +00003858 i++;
3859 }else if( strcmp(z,"-html")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003860 data.mode = MODE_Html;
drh22fbcb82004-02-01 01:22:50 +00003861 }else if( strcmp(z,"-list")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003862 data.mode = MODE_List;
drh22fbcb82004-02-01 01:22:50 +00003863 }else if( strcmp(z,"-line")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003864 data.mode = MODE_Line;
drh22fbcb82004-02-01 01:22:50 +00003865 }else if( strcmp(z,"-column")==0 ){
drh8b32e172002-04-08 02:42:57 +00003866 data.mode = MODE_Column;
drhc49f44e2006-10-26 18:15:42 +00003867 }else if( strcmp(z,"-csv")==0 ){
3868 data.mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00003869 memcpy(data.separator,",",2);
drh22fbcb82004-02-01 01:22:50 +00003870 }else if( strcmp(z,"-separator")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003871 sqlite3_snprintf(sizeof(data.separator), data.separator,
drh98d312f2012-10-25 15:23:14 +00003872 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003873 }else if( strcmp(z,"-nullvalue")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003874 sqlite3_snprintf(sizeof(data.nullvalue), data.nullvalue,
drh98d312f2012-10-25 15:23:14 +00003875 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003876 }else if( strcmp(z,"-header")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003877 data.showHeader = 1;
drh22fbcb82004-02-01 01:22:50 +00003878 }else if( strcmp(z,"-noheader")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003879 data.showHeader = 0;
drh22fbcb82004-02-01 01:22:50 +00003880 }else if( strcmp(z,"-echo")==0 ){
drhdaffd0e2001-04-11 14:28:42 +00003881 data.echoOn = 1;
drhefbf3b12014-02-28 20:47:24 +00003882 }else if( strcmp(z,"-eqp")==0 ){
3883 data.autoEQP = 1;
shaneh642d8b82010-07-28 16:05:34 +00003884 }else if( strcmp(z,"-stats")==0 ){
3885 data.statsOn = 1;
drhc49f44e2006-10-26 18:15:42 +00003886 }else if( strcmp(z,"-bail")==0 ){
3887 bail_on_error = 1;
drh22fbcb82004-02-01 01:22:50 +00003888 }else if( strcmp(z,"-version")==0 ){
drh9fd301b2011-06-03 13:28:22 +00003889 printf("%s %s\n", sqlite3_libversion(), sqlite3_sourceid());
drh151e3e12006-06-06 12:32:21 +00003890 return 0;
drhc28490c2006-10-26 14:25:58 +00003891 }else if( strcmp(z,"-interactive")==0 ){
3892 stdin_is_interactive = 1;
3893 }else if( strcmp(z,"-batch")==0 ){
3894 stdin_is_interactive = 0;
drh9c88d682010-12-17 14:03:01 +00003895 }else if( strcmp(z,"-heap")==0 ){
3896 i++;
drh7d9f3942013-04-03 01:26:54 +00003897 }else if( strcmp(z,"-mmap")==0 ){
3898 i++;
drha7e61d82011-03-12 17:02:57 +00003899 }else if( strcmp(z,"-vfs")==0 ){
3900 i++;
drh6f25e892011-07-08 17:02:57 +00003901#ifdef SQLITE_ENABLE_VFSTRACE
drh97ae8ff2011-03-16 16:56:29 +00003902 }else if( strcmp(z,"-vfstrace")==0 ){
3903 i++;
drh6f25e892011-07-08 17:02:57 +00003904#endif
3905#ifdef SQLITE_ENABLE_MULTIPLEX
3906 }else if( strcmp(z,"-multiplex")==0 ){
3907 i++;
3908#endif
drhcc3b4f82012-02-07 14:13:50 +00003909 }else if( strcmp(z,"-help")==0 ){
drhe1e38c42003-05-04 18:30:59 +00003910 usage(1);
drhcc3b4f82012-02-07 14:13:50 +00003911 }else if( strcmp(z,"-cmd")==0 ){
3912 if( i==argc-1 ) break;
drh98d312f2012-10-25 15:23:14 +00003913 z = cmdline_option_value(argc,argv,++i);
drhcc3b4f82012-02-07 14:13:50 +00003914 if( z[0]=='.' ){
3915 rc = do_meta_command(z, &data);
drh99b39082013-04-17 12:19:48 +00003916 if( rc && bail_on_error ) return rc==2 ? 0 : rc;
drhcc3b4f82012-02-07 14:13:50 +00003917 }else{
drh05782482013-10-24 15:20:20 +00003918 open_db(&data, 0);
drhcc3b4f82012-02-07 14:13:50 +00003919 rc = shell_exec(data.db, z, shell_callback, &data, &zErrMsg);
3920 if( zErrMsg!=0 ){
3921 fprintf(stderr,"Error: %s\n", zErrMsg);
3922 if( bail_on_error ) return rc!=0 ? rc : 1;
3923 }else if( rc!=0 ){
3924 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", z);
3925 if( bail_on_error ) return rc;
3926 }
3927 }
drh1e5d0e92000-05-31 23:33:17 +00003928 }else{
shane86f5bdb2009-10-24 02:00:07 +00003929 fprintf(stderr,"%s: Error: unknown option: %s\n", Argv0, z);
drhe1e38c42003-05-04 18:30:59 +00003930 fprintf(stderr,"Use -help for a list of options.\n");
drh1e5d0e92000-05-31 23:33:17 +00003931 return 1;
3932 }
3933 }
drh44c2eb12003-04-30 11:38:26 +00003934
drh22fbcb82004-02-01 01:22:50 +00003935 if( zFirstCmd ){
drh44c2eb12003-04-30 11:38:26 +00003936 /* Run just the command that follows the database name
3937 */
drh22fbcb82004-02-01 01:22:50 +00003938 if( zFirstCmd[0]=='.' ){
shane916f9612009-10-23 00:37:15 +00003939 rc = do_meta_command(zFirstCmd, &data);
drh99b39082013-04-17 12:19:48 +00003940 if( rc==2 ) rc = 0;
drh6ff13852001-11-25 13:18:23 +00003941 }else{
drh05782482013-10-24 15:20:20 +00003942 open_db(&data, 0);
shane626a6e42009-10-22 17:30:15 +00003943 rc = shell_exec(data.db, zFirstCmd, shell_callback, &data, &zErrMsg);
shane86f5bdb2009-10-24 02:00:07 +00003944 if( zErrMsg!=0 ){
3945 fprintf(stderr,"Error: %s\n", zErrMsg);
3946 return rc!=0 ? rc : 1;
3947 }else if( rc!=0 ){
3948 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", zFirstCmd);
3949 return rc;
drh6ff13852001-11-25 13:18:23 +00003950 }
drh75897232000-05-29 14:26:00 +00003951 }
3952 }else{
drh44c2eb12003-04-30 11:38:26 +00003953 /* Run commands received from standard input
3954 */
drhc28490c2006-10-26 14:25:58 +00003955 if( stdin_is_interactive ){
drh67505e72002-04-19 12:34:06 +00003956 char *zHome;
3957 char *zHistory = 0;
drh5bb3eb92007-05-04 13:15:55 +00003958 int nHistory;
drh75897232000-05-29 14:26:00 +00003959 printf(
drh743e0032011-12-12 16:51:50 +00003960 "SQLite version %s %.19s\n" /*extra-version-info*/
drh1247aa42014-02-10 19:27:05 +00003961 "Enter \".help\" for usage hints.\n",
drh9fd301b2011-06-03 13:28:22 +00003962 sqlite3_libversion(), sqlite3_sourceid()
drh75897232000-05-29 14:26:00 +00003963 );
drhb3735912014-02-10 16:13:42 +00003964 if( warnInmemoryDb ){
drh1247aa42014-02-10 19:27:05 +00003965 printf("Connected to a ");
mistachkin378d01a2014-03-06 02:15:42 +00003966 printBold("transient in-memory database");
3967 printf(".\nUse \".open FILENAME\" to reopen on a "
drh1247aa42014-02-10 19:27:05 +00003968 "persistent database.\n");
drhb3735912014-02-10 16:13:42 +00003969 }
drh67505e72002-04-19 12:34:06 +00003970 zHome = find_home_dir();
drhea678832008-12-10 19:26:22 +00003971 if( zHome ){
drh4f21c4a2008-12-10 22:15:00 +00003972 nHistory = strlen30(zHome) + 20;
drhea678832008-12-10 19:26:22 +00003973 if( (zHistory = malloc(nHistory))!=0 ){
3974 sqlite3_snprintf(nHistory, zHistory,"%s/.sqlite_history", zHome);
3975 }
drh67505e72002-04-19 12:34:06 +00003976 }
drhaaa21b42014-02-11 14:37:51 +00003977#if defined(HAVE_READLINE)
drh67505e72002-04-19 12:34:06 +00003978 if( zHistory ) read_history(zHistory);
danielk19774af00c62005-01-23 23:43:21 +00003979#endif
drhc28490c2006-10-26 14:25:58 +00003980 rc = process_input(&data, 0);
drh67505e72002-04-19 12:34:06 +00003981 if( zHistory ){
3982 stifle_history(100);
3983 write_history(zHistory);
adamd0a3daa32006-07-28 20:16:14 +00003984 free(zHistory);
drh67505e72002-04-19 12:34:06 +00003985 }
drhdaffd0e2001-04-11 14:28:42 +00003986 }else{
drhc28490c2006-10-26 14:25:58 +00003987 rc = process_input(&data, stdin);
drh75897232000-05-29 14:26:00 +00003988 }
3989 }
drh33048c02001-10-01 14:29:22 +00003990 set_table_name(&data, 0);
drh72af0772010-05-06 20:19:55 +00003991 if( data.db ){
drhe14cd932010-12-08 03:28:17 +00003992 sqlite3_close(data.db);
adamd0a3daa32006-07-28 20:16:14 +00003993 }
drh05782482013-10-24 15:20:20 +00003994 sqlite3_free(data.zFreeOnClose);
drhc28490c2006-10-26 14:25:58 +00003995 return rc;
drh75897232000-05-29 14:26:00 +00003996}