blob: 9a8a906944adf99e1c602b7d3cd9aca6ee04152d [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
drh81d7fd12010-12-08 00:02:26 +000048#ifdef HAVE_EDITLINE
49# include <editline/editline.h>
50#endif
drh16e59552000-07-31 11:57:37 +000051#if defined(HAVE_READLINE) && HAVE_READLINE==1
drh8e7e7a22000-05-30 18:45:23 +000052# include <readline/readline.h>
53# include <readline/history.h>
drh81d7fd12010-12-08 00:02:26 +000054#endif
55#if !defined(HAVE_EDITLINE) && (!defined(HAVE_READLINE) || HAVE_READLINE!=1)
drh18f52e02012-01-16 16:56:31 +000056# define readline(p) local_getline(p,stdin,0)
persicom1d0b8722002-04-18 02:53:04 +000057# define add_history(X)
drh67505e72002-04-19 12:34:06 +000058# define read_history(X)
59# define write_history(X)
60# define stifle_history(X)
drh75897232000-05-29 14:26:00 +000061#endif
62
adamd2e8464a2006-09-06 21:39:40 +000063#if defined(_WIN32) || defined(WIN32)
64# include <io.h>
shane18e526c2008-12-10 22:30:24 +000065#define isatty(h) _isatty(h)
66#define access(f,m) _access((f),(m))
drh67ceaa62012-08-27 21:19:03 +000067#undef popen
drhe1da8fa2012-03-30 00:05:57 +000068#define popen(a,b) _popen((a),(b))
drh67ceaa62012-08-27 21:19:03 +000069#undef pclose
drhe1da8fa2012-03-30 00:05:57 +000070#define pclose(x) _pclose(x)
adamd2e8464a2006-09-06 21:39:40 +000071#else
drh4328c8b2003-04-26 02:50:11 +000072/* Make sure isatty() has a prototype.
73*/
drhb2acc3b2011-10-13 16:36:29 +000074extern int isatty(int);
adamd2e8464a2006-09-06 21:39:40 +000075#endif
drh4328c8b2003-04-26 02:50:11 +000076
chw65d3c132007-11-12 21:09:10 +000077#if defined(_WIN32_WCE)
78/* Windows CE (arm-wince-mingw32ce-gcc) does not provide isatty()
79 * thus we always assume that we have a console. That can be
80 * overridden with the -batch command line option.
81 */
82#define isatty(x) 1
83#endif
84
drhc6e41722011-04-11 15:36:26 +000085/* True if the timer is enabled */
86static int enableTimer = 0;
87
drhf0693c82011-10-11 20:41:54 +000088/* ctype macros that work with signed characters */
89#define IsSpace(X) isspace((unsigned char)X)
90#define IsDigit(X) isdigit((unsigned char)X)
91#define ToLower(X) (char)tolower((unsigned char)X)
92
drhd5d0f642013-02-20 00:54:21 +000093#if !defined(_WIN32) && !defined(WIN32) && !defined(_WRS_KERNEL) \
94 && !defined(__minux)
drh3b1a9882007-11-02 12:53:03 +000095#include <sys/time.h>
96#include <sys/resource.h>
97
drhda108222009-02-25 19:07:24 +000098/* Saved resource information for the beginning of an operation */
99static struct rusage sBegin;
100
drhda108222009-02-25 19:07:24 +0000101/*
102** Begin timing an operation
103*/
104static void beginTimer(void){
105 if( enableTimer ){
106 getrusage(RUSAGE_SELF, &sBegin);
107 }
108}
109
110/* Return the difference of two time_structs in seconds */
111static double timeDiff(struct timeval *pStart, struct timeval *pEnd){
112 return (pEnd->tv_usec - pStart->tv_usec)*0.000001 +
113 (double)(pEnd->tv_sec - pStart->tv_sec);
114}
115
116/*
117** Print the timing results.
118*/
119static void endTimer(void){
120 if( enableTimer ){
121 struct rusage sEnd;
122 getrusage(RUSAGE_SELF, &sEnd);
123 printf("CPU Time: user %f sys %f\n",
124 timeDiff(&sBegin.ru_utime, &sEnd.ru_utime),
125 timeDiff(&sBegin.ru_stime, &sEnd.ru_stime));
126 }
127}
shaneb320ccd2009-10-21 03:42:58 +0000128
drhda108222009-02-25 19:07:24 +0000129#define BEGIN_TIMER beginTimer()
130#define END_TIMER endTimer()
131#define HAS_TIMER 1
shaneb320ccd2009-10-21 03:42:58 +0000132
133#elif (defined(_WIN32) || defined(WIN32))
134
135#include <windows.h>
136
137/* Saved resource information for the beginning of an operation */
138static HANDLE hProcess;
139static FILETIME ftKernelBegin;
140static FILETIME ftUserBegin;
141typedef BOOL (WINAPI *GETPROCTIMES)(HANDLE, LPFILETIME, LPFILETIME, LPFILETIME, LPFILETIME);
142static GETPROCTIMES getProcessTimesAddr = NULL;
143
shaneb320ccd2009-10-21 03:42:58 +0000144/*
145** Check to see if we have timer support. Return 1 if necessary
146** support found (or found previously).
147*/
148static int hasTimer(void){
149 if( getProcessTimesAddr ){
150 return 1;
151 } else {
152 /* GetProcessTimes() isn't supported in WIN95 and some other Windows versions.
153 ** See if the version we are running on has it, and if it does, save off
154 ** a pointer to it and the current process handle.
155 */
156 hProcess = GetCurrentProcess();
157 if( hProcess ){
158 HINSTANCE hinstLib = LoadLibrary(TEXT("Kernel32.dll"));
159 if( NULL != hinstLib ){
160 getProcessTimesAddr = (GETPROCTIMES) GetProcAddress(hinstLib, "GetProcessTimes");
161 if( NULL != getProcessTimesAddr ){
162 return 1;
163 }
164 FreeLibrary(hinstLib);
165 }
166 }
167 }
168 return 0;
169}
170
171/*
172** Begin timing an operation
173*/
174static void beginTimer(void){
175 if( enableTimer && getProcessTimesAddr ){
176 FILETIME ftCreation, ftExit;
177 getProcessTimesAddr(hProcess, &ftCreation, &ftExit, &ftKernelBegin, &ftUserBegin);
178 }
179}
180
181/* Return the difference of two FILETIME structs in seconds */
182static double timeDiff(FILETIME *pStart, FILETIME *pEnd){
183 sqlite_int64 i64Start = *((sqlite_int64 *) pStart);
184 sqlite_int64 i64End = *((sqlite_int64 *) pEnd);
185 return (double) ((i64End - i64Start) / 10000000.0);
186}
187
188/*
189** Print the timing results.
190*/
191static void endTimer(void){
192 if( enableTimer && getProcessTimesAddr){
193 FILETIME ftCreation, ftExit, ftKernelEnd, ftUserEnd;
194 getProcessTimesAddr(hProcess, &ftCreation, &ftExit, &ftKernelEnd, &ftUserEnd);
195 printf("CPU Time: user %f sys %f\n",
196 timeDiff(&ftUserBegin, &ftUserEnd),
197 timeDiff(&ftKernelBegin, &ftKernelEnd));
198 }
199}
200
201#define BEGIN_TIMER beginTimer()
202#define END_TIMER endTimer()
203#define HAS_TIMER hasTimer()
204
drhda108222009-02-25 19:07:24 +0000205#else
206#define BEGIN_TIMER
207#define END_TIMER
208#define HAS_TIMER 0
209#endif
210
shanec0688ea2009-03-05 03:48:06 +0000211/*
212** Used to prevent warnings about unused parameters
213*/
214#define UNUSED_PARAMETER(x) (void)(x)
215
drhe91d16b2008-12-08 18:27:31 +0000216/*
drhc49f44e2006-10-26 18:15:42 +0000217** If the following flag is set, then command execution stops
218** at an error if we are not interactive.
219*/
220static int bail_on_error = 0;
221
222/*
drhc28490c2006-10-26 14:25:58 +0000223** Threat stdin as an interactive input if the following variable
224** is true. Otherwise, assume stdin is connected to a file or pipe.
225*/
226static int stdin_is_interactive = 1;
227
228/*
drh4c504392000-10-16 22:06:40 +0000229** The following is the open SQLite database. We make a pointer
230** to this database a static variable so that it can be accessed
231** by the SIGINT handler to interrupt database processing.
232*/
danielk197792f9a1b2004-06-19 09:08:16 +0000233static sqlite3 *db = 0;
drh4c504392000-10-16 22:06:40 +0000234
235/*
drh67505e72002-04-19 12:34:06 +0000236** True if an interrupt (Control-C) has been received.
237*/
drh43617e92006-03-06 20:55:46 +0000238static volatile int seenInterrupt = 0;
drh67505e72002-04-19 12:34:06 +0000239
240/*
persicom7e2dfdd2002-04-18 02:46:52 +0000241** This is the name of our program. It is set in main(), used
242** in a number of other places, mostly for error messages.
243*/
244static char *Argv0;
245
246/*
247** Prompt strings. Initialized in main. Settable with
248** .prompt main continue
249*/
250static char mainPrompt[20]; /* First line prompt. default: "sqlite> "*/
251static char continuePrompt[20]; /* Continuation prompt. default: " ...> " */
252
drhb0603412007-02-28 04:47:26 +0000253/*
254** Write I/O traces to the following stream.
255*/
rsebe0a9092007-07-30 18:24:38 +0000256#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +0000257static FILE *iotrace = 0;
rsebe0a9092007-07-30 18:24:38 +0000258#endif
drhb0603412007-02-28 04:47:26 +0000259
260/*
261** This routine works like printf in that its first argument is a
262** format string and subsequent arguments are values to be substituted
263** in place of % fields. The result of formatting this string
264** is written to iotrace.
265*/
rsebe0a9092007-07-30 18:24:38 +0000266#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +0000267static void iotracePrintf(const char *zFormat, ...){
268 va_list ap;
drhf075cd02007-02-28 06:14:25 +0000269 char *z;
drhb0603412007-02-28 04:47:26 +0000270 if( iotrace==0 ) return;
271 va_start(ap, zFormat);
drhf075cd02007-02-28 06:14:25 +0000272 z = sqlite3_vmprintf(zFormat, ap);
drhb0603412007-02-28 04:47:26 +0000273 va_end(ap);
drhf075cd02007-02-28 06:14:25 +0000274 fprintf(iotrace, "%s", z);
275 sqlite3_free(z);
drhb0603412007-02-28 04:47:26 +0000276}
rsebe0a9092007-07-30 18:24:38 +0000277#endif
drhb0603412007-02-28 04:47:26 +0000278
drh44c2eb12003-04-30 11:38:26 +0000279
persicom7e2dfdd2002-04-18 02:46:52 +0000280/*
drh83965662003-04-17 02:54:13 +0000281** Determines if a string is a number of not.
282*/
danielk19772e588c72005-12-09 14:25:08 +0000283static int isNumber(const char *z, int *realnum){
drhc8d74412004-08-31 23:41:26 +0000284 if( *z=='-' || *z=='+' ) z++;
drhf0693c82011-10-11 20:41:54 +0000285 if( !IsDigit(*z) ){
drhc8d74412004-08-31 23:41:26 +0000286 return 0;
287 }
288 z++;
289 if( realnum ) *realnum = 0;
drhf0693c82011-10-11 20:41:54 +0000290 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000291 if( *z=='.' ){
292 z++;
drhf0693c82011-10-11 20:41:54 +0000293 if( !IsDigit(*z) ) return 0;
294 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000295 if( realnum ) *realnum = 1;
296 }
297 if( *z=='e' || *z=='E' ){
298 z++;
299 if( *z=='+' || *z=='-' ) z++;
drhf0693c82011-10-11 20:41:54 +0000300 if( !IsDigit(*z) ) return 0;
301 while( IsDigit(*z) ){ z++; }
drhc8d74412004-08-31 23:41:26 +0000302 if( realnum ) *realnum = 1;
303 }
304 return *z==0;
305}
drh83965662003-04-17 02:54:13 +0000306
307/*
danielk1977bc6ada42004-06-30 08:20:16 +0000308** A global char* and an SQL function to access its current value
309** from within an SQL statement. This program used to use the
310** sqlite_exec_printf() API to substitue a string into an SQL statement.
311** The correct way to do this with sqlite3 is to use the bind API, but
312** since the shell is built around the callback paradigm it would be a lot
313** of work. Instead just use this hack, which is quite harmless.
314*/
315static const char *zShellStatic = 0;
316static void shellstaticFunc(
317 sqlite3_context *context,
318 int argc,
319 sqlite3_value **argv
320){
321 assert( 0==argc );
322 assert( zShellStatic );
shaned87897d2009-01-30 05:40:27 +0000323 UNUSED_PARAMETER(argc);
drh902b9ee2008-12-05 17:17:07 +0000324 UNUSED_PARAMETER(argv);
danielk1977bc6ada42004-06-30 08:20:16 +0000325 sqlite3_result_text(context, zShellStatic, -1, SQLITE_STATIC);
326}
327
328
329/*
drhfeac5f82004-08-01 00:10:45 +0000330** This routine reads a line of text from FILE in, stores
drh8e7e7a22000-05-30 18:45:23 +0000331** the text in memory obtained from malloc() and returns a pointer
332** to the text. NULL is returned at end of file, or if malloc()
333** fails.
334**
335** The interface is like "readline" but no command-line editing
336** is done.
337*/
drh18f52e02012-01-16 16:56:31 +0000338static char *local_getline(char *zPrompt, FILE *in, int csvFlag){
drh8e7e7a22000-05-30 18:45:23 +0000339 char *zLine;
340 int nLine;
drh8e7e7a22000-05-30 18:45:23 +0000341 int n;
drh18f52e02012-01-16 16:56:31 +0000342 int inQuote = 0;
drh8e7e7a22000-05-30 18:45:23 +0000343
344 if( zPrompt && *zPrompt ){
345 printf("%s",zPrompt);
346 fflush(stdout);
347 }
348 nLine = 100;
349 zLine = malloc( nLine );
350 if( zLine==0 ) return 0;
351 n = 0;
drhb07028f2011-10-14 21:49:18 +0000352 while( 1 ){
drh8e7e7a22000-05-30 18:45:23 +0000353 if( n+100>nLine ){
354 nLine = nLine*2 + 100;
355 zLine = realloc(zLine, nLine);
356 if( zLine==0 ) return 0;
357 }
drhdaffd0e2001-04-11 14:28:42 +0000358 if( fgets(&zLine[n], nLine - n, in)==0 ){
drh8e7e7a22000-05-30 18:45:23 +0000359 if( n==0 ){
360 free(zLine);
361 return 0;
362 }
363 zLine[n] = 0;
drh8e7e7a22000-05-30 18:45:23 +0000364 break;
365 }
drh18f52e02012-01-16 16:56:31 +0000366 while( zLine[n] ){
367 if( zLine[n]=='"' ) inQuote = !inQuote;
368 n++;
369 }
370 if( n>0 && zLine[n-1]=='\n' && (!inQuote || !csvFlag) ){
drh8e7e7a22000-05-30 18:45:23 +0000371 n--;
shaneh13b36022009-12-17 21:07:15 +0000372 if( n>0 && zLine[n-1]=='\r' ) n--;
drh8e7e7a22000-05-30 18:45:23 +0000373 zLine[n] = 0;
drhb07028f2011-10-14 21:49:18 +0000374 break;
drh8e7e7a22000-05-30 18:45:23 +0000375 }
376 }
377 zLine = realloc( zLine, n+1 );
378 return zLine;
379}
380
381/*
drhc28490c2006-10-26 14:25:58 +0000382** Retrieve a single line of input text.
drh8e7e7a22000-05-30 18:45:23 +0000383**
384** zPrior is a string of prior text retrieved. If not the empty
385** string, then issue a continuation prompt.
386*/
drhdaffd0e2001-04-11 14:28:42 +0000387static char *one_input_line(const char *zPrior, FILE *in){
drh8e7e7a22000-05-30 18:45:23 +0000388 char *zPrompt;
389 char *zResult;
drhdaffd0e2001-04-11 14:28:42 +0000390 if( in!=0 ){
drh18f52e02012-01-16 16:56:31 +0000391 return local_getline(0, in, 0);
drh8e7e7a22000-05-30 18:45:23 +0000392 }
393 if( zPrior && zPrior[0] ){
persicom7e2dfdd2002-04-18 02:46:52 +0000394 zPrompt = continuePrompt;
drh8e7e7a22000-05-30 18:45:23 +0000395 }else{
persicom7e2dfdd2002-04-18 02:46:52 +0000396 zPrompt = mainPrompt;
drh8e7e7a22000-05-30 18:45:23 +0000397 }
398 zResult = readline(zPrompt);
danielk19774af00c62005-01-23 23:43:21 +0000399#if defined(HAVE_READLINE) && HAVE_READLINE==1
drheb741d52006-06-03 17:37:25 +0000400 if( zResult && *zResult ) add_history(zResult);
danielk19774af00c62005-01-23 23:43:21 +0000401#endif
drh8e7e7a22000-05-30 18:45:23 +0000402 return zResult;
403}
404
persicom7e2dfdd2002-04-18 02:46:52 +0000405struct previous_mode_data {
406 int valid; /* Is there legit data in here? */
407 int mode;
408 int showHeader;
409 int colWidth[100];
410};
drh45e29d82006-11-20 16:21:10 +0000411
drh8e7e7a22000-05-30 18:45:23 +0000412/*
drh75897232000-05-29 14:26:00 +0000413** An pointer to an instance of this structure is passed from
414** the main program to the callback. This is used to communicate
415** state and mode information.
416*/
417struct callback_data {
shane626a6e42009-10-22 17:30:15 +0000418 sqlite3 *db; /* The database */
drhdaffd0e2001-04-11 14:28:42 +0000419 int echoOn; /* True to echo input commands */
shaneh642d8b82010-07-28 16:05:34 +0000420 int statsOn; /* True to display memory stats before each finalize */
drh28bd4bc2000-06-15 15:57:22 +0000421 int cnt; /* Number of records displayed so far */
422 FILE *out; /* Write results here */
drh42f64e52012-04-04 16:56:23 +0000423 FILE *traceOut; /* Output for sqlite3_trace() */
drh2f464a02011-10-13 00:41:49 +0000424 int nErr; /* Number of errors seen */
drh28bd4bc2000-06-15 15:57:22 +0000425 int mode; /* An output mode setting */
drh45e29d82006-11-20 16:21:10 +0000426 int writableSchema; /* True if PRAGMA writable_schema=ON */
drh28bd4bc2000-06-15 15:57:22 +0000427 int showHeader; /* True to show column names in List or Column mode */
drh33048c02001-10-01 14:29:22 +0000428 char *zDestTable; /* Name of destination table when MODE_Insert */
drh28bd4bc2000-06-15 15:57:22 +0000429 char separator[20]; /* Separator character for MODE_List */
drha0c66f52000-07-29 13:20:21 +0000430 int colWidth[100]; /* Requested width of each column when in column mode*/
431 int actualWidth[100]; /* Actual width of each column */
drh83965662003-04-17 02:54:13 +0000432 char nullvalue[20]; /* The text to print when a NULL comes back from
433 ** the database */
persicom7e2dfdd2002-04-18 02:46:52 +0000434 struct previous_mode_data explainPrev;
drh83965662003-04-17 02:54:13 +0000435 /* Holds the mode information just before
436 ** .explain ON */
drh44c2eb12003-04-30 11:38:26 +0000437 char outfile[FILENAME_MAX]; /* Filename for *out */
438 const char *zDbFilename; /* name of the database file */
drha7e61d82011-03-12 17:02:57 +0000439 const char *zVfs; /* Name of VFS to use */
shane626a6e42009-10-22 17:30:15 +0000440 sqlite3_stmt *pStmt; /* Current statement if any. */
drh127f9d72010-02-23 01:47:00 +0000441 FILE *pLog; /* Write log output here */
drh75897232000-05-29 14:26:00 +0000442};
443
444/*
445** These are the allowed modes.
446*/
drh967e8b72000-06-21 13:59:10 +0000447#define MODE_Line 0 /* One column per line. Blank line between records */
drh75897232000-05-29 14:26:00 +0000448#define MODE_Column 1 /* One record per line in neat columns */
449#define MODE_List 2 /* One record per line with a separator */
drhe3710332000-09-29 13:30:53 +0000450#define MODE_Semi 3 /* Same as MODE_List but append ";" to each line */
451#define MODE_Html 4 /* Generate an XHTML table */
452#define MODE_Insert 5 /* Generate SQL "insert" statements */
drhfeac5f82004-08-01 00:10:45 +0000453#define MODE_Tcl 6 /* Generate ANSI-C or TCL quoted elements */
drh8e64d1c2004-10-07 00:32:39 +0000454#define MODE_Csv 7 /* Quote strings, numbers are plain */
drh66ce4d02008-02-15 17:38:06 +0000455#define MODE_Explain 8 /* Like MODE_Column, but do not truncate data */
persicom7e2dfdd2002-04-18 02:46:52 +0000456
drh66ce4d02008-02-15 17:38:06 +0000457static const char *modeDescr[] = {
persicom7e2dfdd2002-04-18 02:46:52 +0000458 "line",
459 "column",
460 "list",
461 "semi",
462 "html",
drhfeac5f82004-08-01 00:10:45 +0000463 "insert",
464 "tcl",
drh8e64d1c2004-10-07 00:32:39 +0000465 "csv",
drh66ce4d02008-02-15 17:38:06 +0000466 "explain",
persicom7e2dfdd2002-04-18 02:46:52 +0000467};
drh75897232000-05-29 14:26:00 +0000468
469/*
470** Number of elements in an array
471*/
drh902b9ee2008-12-05 17:17:07 +0000472#define ArraySize(X) (int)(sizeof(X)/sizeof(X[0]))
drh75897232000-05-29 14:26:00 +0000473
474/*
drhea678832008-12-10 19:26:22 +0000475** Compute a string length that is limited to what can be stored in
476** lower 30 bits of a 32-bit signed integer.
477*/
drh4f21c4a2008-12-10 22:15:00 +0000478static int strlen30(const char *z){
drhea678832008-12-10 19:26:22 +0000479 const char *z2 = z;
480 while( *z2 ){ z2++; }
481 return 0x3fffffff & (int)(z2 - z);
482}
483
484/*
drh127f9d72010-02-23 01:47:00 +0000485** A callback for the sqlite3_log() interface.
486*/
487static void shellLog(void *pArg, int iErrCode, const char *zMsg){
488 struct callback_data *p = (struct callback_data*)pArg;
489 if( p->pLog==0 ) return;
490 fprintf(p->pLog, "(%d) %s\n", iErrCode, zMsg);
491 fflush(p->pLog);
492}
493
494/*
shane626a6e42009-10-22 17:30:15 +0000495** Output the given string as a hex-encoded blob (eg. X'1234' )
496*/
497static void output_hex_blob(FILE *out, const void *pBlob, int nBlob){
498 int i;
499 char *zBlob = (char *)pBlob;
500 fprintf(out,"X'");
drhb202d702012-04-24 12:12:57 +0000501 for(i=0; i<nBlob; i++){ fprintf(out,"%02x",zBlob[i]&0xff); }
shane626a6e42009-10-22 17:30:15 +0000502 fprintf(out,"'");
503}
504
505/*
drh28bd4bc2000-06-15 15:57:22 +0000506** Output the given string as a quoted string using SQL quoting conventions.
507*/
508static void output_quoted_string(FILE *out, const char *z){
509 int i;
510 int nSingle = 0;
drh28bd4bc2000-06-15 15:57:22 +0000511 for(i=0; z[i]; i++){
512 if( z[i]=='\'' ) nSingle++;
drh28bd4bc2000-06-15 15:57:22 +0000513 }
514 if( nSingle==0 ){
515 fprintf(out,"'%s'",z);
drh28bd4bc2000-06-15 15:57:22 +0000516 }else{
517 fprintf(out,"'");
518 while( *z ){
519 for(i=0; z[i] && z[i]!='\''; i++){}
520 if( i==0 ){
521 fprintf(out,"''");
522 z++;
523 }else if( z[i]=='\'' ){
524 fprintf(out,"%.*s''",i,z);
525 z += i+1;
526 }else{
drhcd7d2732002-02-26 23:24:26 +0000527 fprintf(out,"%s",z);
drh28bd4bc2000-06-15 15:57:22 +0000528 break;
529 }
530 }
drhcd7d2732002-02-26 23:24:26 +0000531 fprintf(out,"'");
drh28bd4bc2000-06-15 15:57:22 +0000532 }
533}
534
535/*
drhfeac5f82004-08-01 00:10:45 +0000536** Output the given string as a quoted according to C or TCL quoting rules.
537*/
538static void output_c_string(FILE *out, const char *z){
539 unsigned int c;
540 fputc('"', out);
541 while( (c = *(z++))!=0 ){
542 if( c=='\\' ){
543 fputc(c, out);
544 fputc(c, out);
mistachkin585dcb22012-12-04 00:23:43 +0000545 }else if( c=='"' ){
546 fputc('\\', out);
547 fputc('"', out);
drhfeac5f82004-08-01 00:10:45 +0000548 }else if( c=='\t' ){
549 fputc('\\', out);
550 fputc('t', out);
551 }else if( c=='\n' ){
552 fputc('\\', out);
553 fputc('n', out);
554 }else if( c=='\r' ){
555 fputc('\\', out);
556 fputc('r', out);
557 }else if( !isprint(c) ){
drh0a8640d2005-08-30 20:12:02 +0000558 fprintf(out, "\\%03o", c&0xff);
drhfeac5f82004-08-01 00:10:45 +0000559 }else{
560 fputc(c, out);
561 }
562 }
563 fputc('"', out);
564}
565
566/*
drhc08a4f12000-06-15 16:49:48 +0000567** Output the given string with characters that are special to
568** HTML escaped.
569*/
570static void output_html_string(FILE *out, const char *z){
571 int i;
572 while( *z ){
shane43d9cb22009-10-21 14:11:48 +0000573 for(i=0; z[i]
574 && z[i]!='<'
575 && z[i]!='&'
576 && z[i]!='>'
577 && z[i]!='\"'
578 && z[i]!='\'';
579 i++){}
drhc08a4f12000-06-15 16:49:48 +0000580 if( i>0 ){
581 fprintf(out,"%.*s",i,z);
582 }
583 if( z[i]=='<' ){
584 fprintf(out,"&lt;");
585 }else if( z[i]=='&' ){
586 fprintf(out,"&amp;");
shane43d9cb22009-10-21 14:11:48 +0000587 }else if( z[i]=='>' ){
588 fprintf(out,"&gt;");
589 }else if( z[i]=='\"' ){
590 fprintf(out,"&quot;");
591 }else if( z[i]=='\'' ){
592 fprintf(out,"&#39;");
drhc08a4f12000-06-15 16:49:48 +0000593 }else{
594 break;
595 }
596 z += i + 1;
597 }
598}
599
600/*
drhc49f44e2006-10-26 18:15:42 +0000601** If a field contains any character identified by a 1 in the following
602** array, then the string must be quoted for CSV.
603*/
604static const char needCsvQuote[] = {
605 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
606 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
607 1, 0, 1, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0,
608 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
609 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
610 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
611 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
612 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
613 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
614 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
615 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
616 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
617 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
618 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
619 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
620 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
621};
622
623/*
drh8e64d1c2004-10-07 00:32:39 +0000624** Output a single term of CSV. Actually, p->separator is used for
625** the separator, which may or may not be a comma. p->nullvalue is
drh18f52e02012-01-16 16:56:31 +0000626** the null value. Strings are quoted if necessary.
drh8e64d1c2004-10-07 00:32:39 +0000627*/
628static void output_csv(struct callback_data *p, const char *z, int bSep){
drhc49f44e2006-10-26 18:15:42 +0000629 FILE *out = p->out;
drh8e64d1c2004-10-07 00:32:39 +0000630 if( z==0 ){
drhc49f44e2006-10-26 18:15:42 +0000631 fprintf(out,"%s",p->nullvalue);
drh8e64d1c2004-10-07 00:32:39 +0000632 }else{
drhc49f44e2006-10-26 18:15:42 +0000633 int i;
drh4f21c4a2008-12-10 22:15:00 +0000634 int nSep = strlen30(p->separator);
drhc49f44e2006-10-26 18:15:42 +0000635 for(i=0; z[i]; i++){
drhc85375d2007-12-18 15:41:44 +0000636 if( needCsvQuote[((unsigned char*)z)[i]]
637 || (z[i]==p->separator[0] &&
638 (nSep==1 || memcmp(z, p->separator, nSep)==0)) ){
drhc49f44e2006-10-26 18:15:42 +0000639 i = 0;
640 break;
641 }
642 }
643 if( i==0 ){
644 putc('"', out);
645 for(i=0; z[i]; i++){
646 if( z[i]=='"' ) putc('"', out);
647 putc(z[i], out);
648 }
649 putc('"', out);
650 }else{
651 fprintf(out, "%s", z);
652 }
drh8e64d1c2004-10-07 00:32:39 +0000653 }
654 if( bSep ){
drhd0e77882008-01-14 15:20:08 +0000655 fprintf(p->out, "%s", p->separator);
drh8e64d1c2004-10-07 00:32:39 +0000656 }
657}
658
danielk19774af00c62005-01-23 23:43:21 +0000659#ifdef SIGINT
drh8e64d1c2004-10-07 00:32:39 +0000660/*
drh4c504392000-10-16 22:06:40 +0000661** This routine runs when the user presses Ctrl-C
662*/
663static void interrupt_handler(int NotUsed){
drh902b9ee2008-12-05 17:17:07 +0000664 UNUSED_PARAMETER(NotUsed);
drh67505e72002-04-19 12:34:06 +0000665 seenInterrupt = 1;
danielk19776f8a5032004-05-10 10:34:51 +0000666 if( db ) sqlite3_interrupt(db);
drh4c504392000-10-16 22:06:40 +0000667}
danielk19774af00c62005-01-23 23:43:21 +0000668#endif
drh4c504392000-10-16 22:06:40 +0000669
670/*
shane626a6e42009-10-22 17:30:15 +0000671** This is the callback routine that the shell
drh75897232000-05-29 14:26:00 +0000672** invokes for each row of a query result.
673*/
shane626a6e42009-10-22 17:30:15 +0000674static int shell_callback(void *pArg, int nArg, char **azArg, char **azCol, int *aiType){
drh75897232000-05-29 14:26:00 +0000675 int i;
676 struct callback_data *p = (struct callback_data*)pArg;
shaneb9fc17d2009-10-22 21:23:35 +0000677
drh75897232000-05-29 14:26:00 +0000678 switch( p->mode ){
679 case MODE_Line: {
drhe3710332000-09-29 13:30:53 +0000680 int w = 5;
drh6a535342001-10-19 16:44:56 +0000681 if( azArg==0 ) break;
drhe3710332000-09-29 13:30:53 +0000682 for(i=0; i<nArg; i++){
drh4f21c4a2008-12-10 22:15:00 +0000683 int len = strlen30(azCol[i] ? azCol[i] : "");
drhe3710332000-09-29 13:30:53 +0000684 if( len>w ) w = len;
685 }
drh75897232000-05-29 14:26:00 +0000686 if( p->cnt++>0 ) fprintf(p->out,"\n");
687 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000688 fprintf(p->out,"%*s = %s\n", w, azCol[i],
drha69d9162003-04-17 22:57:53 +0000689 azArg[i] ? azArg[i] : p->nullvalue);
drh75897232000-05-29 14:26:00 +0000690 }
691 break;
692 }
danielk19770d78bae2008-01-03 07:09:48 +0000693 case MODE_Explain:
drh75897232000-05-29 14:26:00 +0000694 case MODE_Column: {
drha0c66f52000-07-29 13:20:21 +0000695 if( p->cnt++==0 ){
drh75897232000-05-29 14:26:00 +0000696 for(i=0; i<nArg; i++){
drha0c66f52000-07-29 13:20:21 +0000697 int w, n;
698 if( i<ArraySize(p->colWidth) ){
danielk19770d78bae2008-01-03 07:09:48 +0000699 w = p->colWidth[i];
drh75897232000-05-29 14:26:00 +0000700 }else{
danielk19770d78bae2008-01-03 07:09:48 +0000701 w = 0;
drh75897232000-05-29 14:26:00 +0000702 }
drh078b1fd2012-09-21 13:40:02 +0000703 if( w==0 ){
drh4f21c4a2008-12-10 22:15:00 +0000704 w = strlen30(azCol[i] ? azCol[i] : "");
drha0c66f52000-07-29 13:20:21 +0000705 if( w<10 ) w = 10;
drh4f21c4a2008-12-10 22:15:00 +0000706 n = strlen30(azArg && azArg[i] ? azArg[i] : p->nullvalue);
drha0c66f52000-07-29 13:20:21 +0000707 if( w<n ) w = n;
708 }
709 if( i<ArraySize(p->actualWidth) ){
persicom1d0b8722002-04-18 02:53:04 +0000710 p->actualWidth[i] = w;
drha0c66f52000-07-29 13:20:21 +0000711 }
712 if( p->showHeader ){
drh078b1fd2012-09-21 13:40:02 +0000713 if( w<0 ){
714 fprintf(p->out,"%*.*s%s",-w,-w,azCol[i], i==nArg-1 ? "\n": " ");
715 }else{
716 fprintf(p->out,"%-*.*s%s",w,w,azCol[i], i==nArg-1 ? "\n": " ");
717 }
drha0c66f52000-07-29 13:20:21 +0000718 }
719 }
720 if( p->showHeader ){
721 for(i=0; i<nArg; i++){
722 int w;
723 if( i<ArraySize(p->actualWidth) ){
724 w = p->actualWidth[i];
drh078b1fd2012-09-21 13:40:02 +0000725 if( w<0 ) w = -w;
drha0c66f52000-07-29 13:20:21 +0000726 }else{
727 w = 10;
728 }
729 fprintf(p->out,"%-*.*s%s",w,w,"-----------------------------------"
730 "----------------------------------------------------------",
731 i==nArg-1 ? "\n": " ");
732 }
drh75897232000-05-29 14:26:00 +0000733 }
734 }
drh6a535342001-10-19 16:44:56 +0000735 if( azArg==0 ) break;
drh75897232000-05-29 14:26:00 +0000736 for(i=0; i<nArg; i++){
737 int w;
drha0c66f52000-07-29 13:20:21 +0000738 if( i<ArraySize(p->actualWidth) ){
739 w = p->actualWidth[i];
drh75897232000-05-29 14:26:00 +0000740 }else{
741 w = 10;
742 }
drhea678832008-12-10 19:26:22 +0000743 if( p->mode==MODE_Explain && azArg[i] &&
drh4f21c4a2008-12-10 22:15:00 +0000744 strlen30(azArg[i])>w ){
745 w = strlen30(azArg[i]);
danielk19770d78bae2008-01-03 07:09:48 +0000746 }
drh078b1fd2012-09-21 13:40:02 +0000747 if( w<0 ){
748 fprintf(p->out,"%*.*s%s",-w,-w,
749 azArg[i] ? azArg[i] : p->nullvalue, i==nArg-1 ? "\n": " ");
750 }else{
751 fprintf(p->out,"%-*.*s%s",w,w,
752 azArg[i] ? azArg[i] : p->nullvalue, i==nArg-1 ? "\n": " ");
753 }
drh75897232000-05-29 14:26:00 +0000754 }
755 break;
756 }
drhe3710332000-09-29 13:30:53 +0000757 case MODE_Semi:
drh75897232000-05-29 14:26:00 +0000758 case MODE_List: {
759 if( p->cnt++==0 && p->showHeader ){
760 for(i=0; i<nArg; i++){
761 fprintf(p->out,"%s%s",azCol[i], i==nArg-1 ? "\n" : p->separator);
762 }
763 }
drh6a535342001-10-19 16:44:56 +0000764 if( azArg==0 ) break;
drh75897232000-05-29 14:26:00 +0000765 for(i=0; i<nArg; i++){
drh4c653a02000-06-07 01:27:47 +0000766 char *z = azArg[i];
persicom7e2dfdd2002-04-18 02:46:52 +0000767 if( z==0 ) z = p->nullvalue;
drh71172c52002-01-24 00:00:21 +0000768 fprintf(p->out, "%s", z);
drhe3710332000-09-29 13:30:53 +0000769 if( i<nArg-1 ){
770 fprintf(p->out, "%s", p->separator);
771 }else if( p->mode==MODE_Semi ){
772 fprintf(p->out, ";\n");
773 }else{
774 fprintf(p->out, "\n");
775 }
drh75897232000-05-29 14:26:00 +0000776 }
777 break;
778 }
drh1e5d0e92000-05-31 23:33:17 +0000779 case MODE_Html: {
780 if( p->cnt++==0 && p->showHeader ){
mihailim57c591a2008-06-23 21:26:05 +0000781 fprintf(p->out,"<TR>");
drh1e5d0e92000-05-31 23:33:17 +0000782 for(i=0; i<nArg; i++){
shane43d9cb22009-10-21 14:11:48 +0000783 fprintf(p->out,"<TH>");
784 output_html_string(p->out, azCol[i]);
785 fprintf(p->out,"</TH>\n");
drh1e5d0e92000-05-31 23:33:17 +0000786 }
mihailim57c591a2008-06-23 21:26:05 +0000787 fprintf(p->out,"</TR>\n");
drh1e5d0e92000-05-31 23:33:17 +0000788 }
drh6a535342001-10-19 16:44:56 +0000789 if( azArg==0 ) break;
mihailim57c591a2008-06-23 21:26:05 +0000790 fprintf(p->out,"<TR>");
drh1e5d0e92000-05-31 23:33:17 +0000791 for(i=0; i<nArg; i++){
mihailim57c591a2008-06-23 21:26:05 +0000792 fprintf(p->out,"<TD>");
persicom7e2dfdd2002-04-18 02:46:52 +0000793 output_html_string(p->out, azArg[i] ? azArg[i] : p->nullvalue);
mihailim57c591a2008-06-23 21:26:05 +0000794 fprintf(p->out,"</TD>\n");
drh1e5d0e92000-05-31 23:33:17 +0000795 }
mihailim57c591a2008-06-23 21:26:05 +0000796 fprintf(p->out,"</TR>\n");
drh1e5d0e92000-05-31 23:33:17 +0000797 break;
798 }
drhfeac5f82004-08-01 00:10:45 +0000799 case MODE_Tcl: {
800 if( p->cnt++==0 && p->showHeader ){
801 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000802 output_c_string(p->out,azCol[i] ? azCol[i] : "");
mistachkin585dcb22012-12-04 00:23:43 +0000803 if(i<nArg-1) fprintf(p->out, "%s", p->separator);
drhfeac5f82004-08-01 00:10:45 +0000804 }
805 fprintf(p->out,"\n");
806 }
807 if( azArg==0 ) break;
808 for(i=0; i<nArg; i++){
809 output_c_string(p->out, azArg[i] ? azArg[i] : p->nullvalue);
mistachkin585dcb22012-12-04 00:23:43 +0000810 if(i<nArg-1) fprintf(p->out, "%s", p->separator);
drhfeac5f82004-08-01 00:10:45 +0000811 }
812 fprintf(p->out,"\n");
813 break;
814 }
drh8e64d1c2004-10-07 00:32:39 +0000815 case MODE_Csv: {
816 if( p->cnt++==0 && p->showHeader ){
817 for(i=0; i<nArg; i++){
drh2cc55692006-06-27 20:39:04 +0000818 output_csv(p, azCol[i] ? azCol[i] : "", i<nArg-1);
drh8e64d1c2004-10-07 00:32:39 +0000819 }
820 fprintf(p->out,"\n");
821 }
822 if( azArg==0 ) break;
823 for(i=0; i<nArg; i++){
824 output_csv(p, azArg[i], i<nArg-1);
825 }
826 fprintf(p->out,"\n");
827 break;
828 }
drh28bd4bc2000-06-15 15:57:22 +0000829 case MODE_Insert: {
shaneb9fc17d2009-10-22 21:23:35 +0000830 p->cnt++;
drh6a535342001-10-19 16:44:56 +0000831 if( azArg==0 ) break;
drh33048c02001-10-01 14:29:22 +0000832 fprintf(p->out,"INSERT INTO %s VALUES(",p->zDestTable);
drh28bd4bc2000-06-15 15:57:22 +0000833 for(i=0; i<nArg; i++){
834 char *zSep = i>0 ? ",": "";
shanead6b8d02009-10-22 18:12:58 +0000835 if( (azArg[i]==0) || (aiType && aiType[i]==SQLITE_NULL) ){
drh28bd4bc2000-06-15 15:57:22 +0000836 fprintf(p->out,"%sNULL",zSep);
shanead6b8d02009-10-22 18:12:58 +0000837 }else if( aiType && aiType[i]==SQLITE_TEXT ){
838 if( zSep[0] ) fprintf(p->out,"%s",zSep);
839 output_quoted_string(p->out, azArg[i]);
840 }else if( aiType && (aiType[i]==SQLITE_INTEGER || aiType[i]==SQLITE_FLOAT) ){
841 fprintf(p->out,"%s%s",zSep, azArg[i]);
shane626a6e42009-10-22 17:30:15 +0000842 }else if( aiType && aiType[i]==SQLITE_BLOB && p->pStmt ){
843 const void *pBlob = sqlite3_column_blob(p->pStmt, i);
844 int nBlob = sqlite3_column_bytes(p->pStmt, i);
845 if( zSep[0] ) fprintf(p->out,"%s",zSep);
846 output_hex_blob(p->out, pBlob, nBlob);
drhc8d74412004-08-31 23:41:26 +0000847 }else if( isNumber(azArg[i], 0) ){
drh28bd4bc2000-06-15 15:57:22 +0000848 fprintf(p->out,"%s%s",zSep, azArg[i]);
849 }else{
850 if( zSep[0] ) fprintf(p->out,"%s",zSep);
851 output_quoted_string(p->out, azArg[i]);
852 }
853 }
854 fprintf(p->out,");\n");
drh6a535342001-10-19 16:44:56 +0000855 break;
drh28bd4bc2000-06-15 15:57:22 +0000856 }
persicom1d0b8722002-04-18 02:53:04 +0000857 }
drh75897232000-05-29 14:26:00 +0000858 return 0;
859}
860
861/*
shane626a6e42009-10-22 17:30:15 +0000862** This is the callback routine that the SQLite library
863** invokes for each row of a query result.
864*/
865static int callback(void *pArg, int nArg, char **azArg, char **azCol){
866 /* since we don't have type info, call the shell_callback with a NULL value */
867 return shell_callback(pArg, nArg, azArg, azCol, NULL);
868}
869
870/*
drh33048c02001-10-01 14:29:22 +0000871** Set the destination table field of the callback_data structure to
872** the name of the table given. Escape any quote characters in the
873** table name.
874*/
875static void set_table_name(struct callback_data *p, const char *zName){
876 int i, n;
877 int needQuote;
878 char *z;
879
880 if( p->zDestTable ){
881 free(p->zDestTable);
882 p->zDestTable = 0;
883 }
884 if( zName==0 ) return;
drh4c755c02004-08-08 20:22:17 +0000885 needQuote = !isalpha((unsigned char)*zName) && *zName!='_';
drh33048c02001-10-01 14:29:22 +0000886 for(i=n=0; zName[i]; i++, n++){
drh4c755c02004-08-08 20:22:17 +0000887 if( !isalnum((unsigned char)zName[i]) && zName[i]!='_' ){
drh33048c02001-10-01 14:29:22 +0000888 needQuote = 1;
889 if( zName[i]=='\'' ) n++;
890 }
891 }
892 if( needQuote ) n += 2;
893 z = p->zDestTable = malloc( n+1 );
894 if( z==0 ){
shane86f5bdb2009-10-24 02:00:07 +0000895 fprintf(stderr,"Error: out of memory\n");
drh33048c02001-10-01 14:29:22 +0000896 exit(1);
897 }
898 n = 0;
899 if( needQuote ) z[n++] = '\'';
900 for(i=0; zName[i]; i++){
901 z[n++] = zName[i];
902 if( zName[i]=='\'' ) z[n++] = '\'';
903 }
904 if( needQuote ) z[n++] = '\'';
905 z[n] = 0;
906}
907
danielk19772a02e332004-06-05 08:04:36 +0000908/* zIn is either a pointer to a NULL-terminated string in memory obtained
909** from malloc(), or a NULL pointer. The string pointed to by zAppend is
910** added to zIn, and the result returned in memory obtained from malloc().
911** zIn, if it was not NULL, is freed.
912**
913** If the third argument, quote, is not '\0', then it is used as a
914** quote character for zAppend.
915*/
drhc28490c2006-10-26 14:25:58 +0000916static char *appendText(char *zIn, char const *zAppend, char quote){
danielk19772a02e332004-06-05 08:04:36 +0000917 int len;
918 int i;
drh4f21c4a2008-12-10 22:15:00 +0000919 int nAppend = strlen30(zAppend);
920 int nIn = (zIn?strlen30(zIn):0);
danielk19772a02e332004-06-05 08:04:36 +0000921
922 len = nAppend+nIn+1;
923 if( quote ){
924 len += 2;
925 for(i=0; i<nAppend; i++){
926 if( zAppend[i]==quote ) len++;
927 }
928 }
929
930 zIn = (char *)realloc(zIn, len);
931 if( !zIn ){
932 return 0;
933 }
934
935 if( quote ){
936 char *zCsr = &zIn[nIn];
937 *zCsr++ = quote;
938 for(i=0; i<nAppend; i++){
939 *zCsr++ = zAppend[i];
940 if( zAppend[i]==quote ) *zCsr++ = quote;
941 }
942 *zCsr++ = quote;
943 *zCsr++ = '\0';
944 assert( (zCsr-zIn)==len );
945 }else{
946 memcpy(&zIn[nIn], zAppend, nAppend);
947 zIn[len-1] = '\0';
948 }
949
950 return zIn;
951}
952
drhdd3d4592004-08-30 01:54:05 +0000953
954/*
drhb21a8e42012-01-28 21:08:51 +0000955** Execute a query statement that will generate SQL output. Print
956** the result columns, comma-separated, on a line and then add a
957** semicolon terminator to the end of that line.
drh45e29d82006-11-20 16:21:10 +0000958**
drhb21a8e42012-01-28 21:08:51 +0000959** If the number of columns is 1 and that column contains text "--"
960** then write the semicolon on a separate line. That way, if a
961** "--" comment occurs at the end of the statement, the comment
962** won't consume the semicolon terminator.
drhdd3d4592004-08-30 01:54:05 +0000963*/
drh157e29a2009-05-21 15:15:00 +0000964static int run_table_dump_query(
drh2f464a02011-10-13 00:41:49 +0000965 struct callback_data *p, /* Query context */
966 const char *zSelect, /* SELECT statement to extract content */
967 const char *zFirstRow /* Print before first row, if not NULL */
drh157e29a2009-05-21 15:15:00 +0000968){
drhdd3d4592004-08-30 01:54:05 +0000969 sqlite3_stmt *pSelect;
970 int rc;
drhb21a8e42012-01-28 21:08:51 +0000971 int nResult;
972 int i;
973 const char *z;
drh2f464a02011-10-13 00:41:49 +0000974 rc = sqlite3_prepare(p->db, zSelect, -1, &pSelect, 0);
drhdd3d4592004-08-30 01:54:05 +0000975 if( rc!=SQLITE_OK || !pSelect ){
drh2f464a02011-10-13 00:41:49 +0000976 fprintf(p->out, "/**** ERROR: (%d) %s *****/\n", rc, sqlite3_errmsg(p->db));
977 p->nErr++;
drhdd3d4592004-08-30 01:54:05 +0000978 return rc;
979 }
980 rc = sqlite3_step(pSelect);
drhb21a8e42012-01-28 21:08:51 +0000981 nResult = sqlite3_column_count(pSelect);
drhdd3d4592004-08-30 01:54:05 +0000982 while( rc==SQLITE_ROW ){
drh157e29a2009-05-21 15:15:00 +0000983 if( zFirstRow ){
drh2f464a02011-10-13 00:41:49 +0000984 fprintf(p->out, "%s", zFirstRow);
drh157e29a2009-05-21 15:15:00 +0000985 zFirstRow = 0;
986 }
drhb21a8e42012-01-28 21:08:51 +0000987 z = (const char*)sqlite3_column_text(pSelect, 0);
988 fprintf(p->out, "%s", z);
989 for(i=1; i<nResult; i++){
990 fprintf(p->out, ",%s", sqlite3_column_text(pSelect, i));
991 }
992 if( z==0 ) z = "";
993 while( z[0] && (z[0]!='-' || z[1]!='-') ) z++;
994 if( z[0] ){
995 fprintf(p->out, "\n;\n");
996 }else{
997 fprintf(p->out, ";\n");
998 }
drhdd3d4592004-08-30 01:54:05 +0000999 rc = sqlite3_step(pSelect);
1000 }
drh2f464a02011-10-13 00:41:49 +00001001 rc = sqlite3_finalize(pSelect);
1002 if( rc!=SQLITE_OK ){
1003 fprintf(p->out, "/**** ERROR: (%d) %s *****/\n", rc, sqlite3_errmsg(p->db));
1004 p->nErr++;
1005 }
1006 return rc;
drhdd3d4592004-08-30 01:54:05 +00001007}
1008
shane626a6e42009-10-22 17:30:15 +00001009/*
1010** Allocate space and save off current error string.
1011*/
1012static char *save_err_msg(
1013 sqlite3 *db /* Database to query */
1014){
1015 int nErrMsg = 1+strlen30(sqlite3_errmsg(db));
1016 char *zErrMsg = sqlite3_malloc(nErrMsg);
1017 if( zErrMsg ){
1018 memcpy(zErrMsg, sqlite3_errmsg(db), nErrMsg);
1019 }
1020 return zErrMsg;
1021}
1022
1023/*
shaneh642d8b82010-07-28 16:05:34 +00001024** Display memory stats.
1025*/
1026static int display_stats(
1027 sqlite3 *db, /* Database to query */
1028 struct callback_data *pArg, /* Pointer to struct callback_data */
1029 int bReset /* True to reset the stats */
1030){
1031 int iCur;
1032 int iHiwtr;
1033
1034 if( pArg && pArg->out ){
1035
1036 iHiwtr = iCur = -1;
1037 sqlite3_status(SQLITE_STATUS_MEMORY_USED, &iCur, &iHiwtr, bReset);
drh29dfbe32010-07-28 17:01:24 +00001038 fprintf(pArg->out, "Memory Used: %d (max %d) bytes\n", iCur, iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001039 iHiwtr = iCur = -1;
1040 sqlite3_status(SQLITE_STATUS_MALLOC_COUNT, &iCur, &iHiwtr, bReset);
drh2a58e9c2010-12-21 21:28:38 +00001041 fprintf(pArg->out, "Number of Outstanding Allocations: %d (max %d)\n", iCur, iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001042/*
1043** Not currently used by the CLI.
1044** iHiwtr = iCur = -1;
1045** sqlite3_status(SQLITE_STATUS_PAGECACHE_USED, &iCur, &iHiwtr, bReset);
1046** fprintf(pArg->out, "Number of Pcache Pages Used: %d (max %d) pages\n", iCur, iHiwtr);
1047*/
1048 iHiwtr = iCur = -1;
1049 sqlite3_status(SQLITE_STATUS_PAGECACHE_OVERFLOW, &iCur, &iHiwtr, bReset);
1050 fprintf(pArg->out, "Number of Pcache Overflow Bytes: %d (max %d) bytes\n", iCur, iHiwtr);
1051/*
1052** Not currently used by the CLI.
1053** iHiwtr = iCur = -1;
1054** sqlite3_status(SQLITE_STATUS_SCRATCH_USED, &iCur, &iHiwtr, bReset);
1055** fprintf(pArg->out, "Number of Scratch Allocations Used: %d (max %d)\n", iCur, iHiwtr);
1056*/
1057 iHiwtr = iCur = -1;
1058 sqlite3_status(SQLITE_STATUS_SCRATCH_OVERFLOW, &iCur, &iHiwtr, bReset);
1059 fprintf(pArg->out, "Number of Scratch Overflow Bytes: %d (max %d) bytes\n", iCur, iHiwtr);
1060 iHiwtr = iCur = -1;
1061 sqlite3_status(SQLITE_STATUS_MALLOC_SIZE, &iCur, &iHiwtr, bReset);
1062 fprintf(pArg->out, "Largest Allocation: %d bytes\n", iHiwtr);
1063 iHiwtr = iCur = -1;
1064 sqlite3_status(SQLITE_STATUS_PAGECACHE_SIZE, &iCur, &iHiwtr, bReset);
1065 fprintf(pArg->out, "Largest Pcache Allocation: %d bytes\n", iHiwtr);
1066 iHiwtr = iCur = -1;
1067 sqlite3_status(SQLITE_STATUS_SCRATCH_SIZE, &iCur, &iHiwtr, bReset);
1068 fprintf(pArg->out, "Largest Scratch Allocation: %d bytes\n", iHiwtr);
1069#ifdef YYTRACKMAXSTACKDEPTH
1070 iHiwtr = iCur = -1;
1071 sqlite3_status(SQLITE_STATUS_PARSER_STACK, &iCur, &iHiwtr, bReset);
1072 fprintf(pArg->out, "Deepest Parser Stack: %d (max %d)\n", iCur, iHiwtr);
1073#endif
1074 }
1075
1076 if( pArg && pArg->out && db ){
1077 iHiwtr = iCur = -1;
1078 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_USED, &iCur, &iHiwtr, bReset);
1079 fprintf(pArg->out, "Lookaside Slots Used: %d (max %d)\n", iCur, iHiwtr);
drh2a58e9c2010-12-21 21:28:38 +00001080 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_HIT, &iCur, &iHiwtr, bReset);
1081 fprintf(pArg->out, "Successful lookaside attempts: %d\n", iHiwtr);
1082 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_SIZE, &iCur, &iHiwtr, bReset);
1083 fprintf(pArg->out, "Lookaside failures due to size: %d\n", iHiwtr);
1084 sqlite3_db_status(db, SQLITE_DBSTATUS_LOOKASIDE_MISS_FULL, &iCur, &iHiwtr, bReset);
1085 fprintf(pArg->out, "Lookaside failures due to OOM: %d\n", iHiwtr);
shaneh642d8b82010-07-28 16:05:34 +00001086 iHiwtr = iCur = -1;
1087 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_USED, &iCur, &iHiwtr, bReset);
drhc78e6e42011-09-23 18:58:23 +00001088 fprintf(pArg->out, "Pager Heap Usage: %d bytes\n", iCur); iHiwtr = iCur = -1;
1089 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_HIT, &iCur, &iHiwtr, 1);
1090 fprintf(pArg->out, "Page cache hits: %d\n", iCur);
1091 iHiwtr = iCur = -1;
1092 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_MISS, &iCur, &iHiwtr, 1);
1093 fprintf(pArg->out, "Page cache misses: %d\n", iCur);
shaneh642d8b82010-07-28 16:05:34 +00001094 iHiwtr = iCur = -1;
drhfbbcd5d2012-03-24 20:09:33 +00001095 sqlite3_db_status(db, SQLITE_DBSTATUS_CACHE_WRITE, &iCur, &iHiwtr, 1);
1096 fprintf(pArg->out, "Page cache writes: %d\n", iCur);
1097 iHiwtr = iCur = -1;
shaneh642d8b82010-07-28 16:05:34 +00001098 sqlite3_db_status(db, SQLITE_DBSTATUS_SCHEMA_USED, &iCur, &iHiwtr, bReset);
1099 fprintf(pArg->out, "Schema Heap Usage: %d bytes\n", iCur);
1100 iHiwtr = iCur = -1;
1101 sqlite3_db_status(db, SQLITE_DBSTATUS_STMT_USED, &iCur, &iHiwtr, bReset);
1102 fprintf(pArg->out, "Statement Heap/Lookaside Usage: %d bytes\n", iCur);
1103 }
1104
1105 if( pArg && pArg->out && db && pArg->pStmt ){
1106 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_FULLSCAN_STEP, bReset);
1107 fprintf(pArg->out, "Fullscan Steps: %d\n", iCur);
1108 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_SORT, bReset);
1109 fprintf(pArg->out, "Sort Operations: %d\n", iCur);
1110 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_AUTOINDEX, bReset);
1111 fprintf(pArg->out, "Autoindex Inserts: %d\n", iCur);
drhbf159fa2013-06-25 22:01:22 +00001112 iCur = sqlite3_stmt_status(pArg->pStmt, SQLITE_STMTSTATUS_VM_STEP, bReset);
1113 fprintf(pArg->out, "Virtual Machine Steps: %d\n", iCur);
shaneh642d8b82010-07-28 16:05:34 +00001114 }
1115
1116 return 0;
1117}
1118
1119/*
shane626a6e42009-10-22 17:30:15 +00001120** Execute a statement or set of statements. Print
1121** any result rows/columns depending on the current mode
1122** set via the supplied callback.
1123**
1124** This is very similar to SQLite's built-in sqlite3_exec()
1125** function except it takes a slightly different callback
1126** and callback data argument.
1127*/
1128static int shell_exec(
1129 sqlite3 *db, /* An open database */
1130 const char *zSql, /* SQL to be evaluated */
1131 int (*xCallback)(void*,int,char**,char**,int*), /* Callback function */
1132 /* (not the same as sqlite3_exec) */
1133 struct callback_data *pArg, /* Pointer to struct callback_data */
1134 char **pzErrMsg /* Error msg written here */
1135){
dan4564ced2010-01-05 04:59:56 +00001136 sqlite3_stmt *pStmt = NULL; /* Statement to execute. */
1137 int rc = SQLITE_OK; /* Return Code */
drhb07028f2011-10-14 21:49:18 +00001138 int rc2;
dan4564ced2010-01-05 04:59:56 +00001139 const char *zLeftover; /* Tail of unprocessed SQL */
shane626a6e42009-10-22 17:30:15 +00001140
1141 if( pzErrMsg ){
1142 *pzErrMsg = NULL;
1143 }
1144
shaneb9fc17d2009-10-22 21:23:35 +00001145 while( zSql[0] && (SQLITE_OK == rc) ){
1146 rc = sqlite3_prepare_v2(db, zSql, -1, &pStmt, &zLeftover);
1147 if( SQLITE_OK != rc ){
shane626a6e42009-10-22 17:30:15 +00001148 if( pzErrMsg ){
1149 *pzErrMsg = save_err_msg(db);
1150 }
1151 }else{
shaneb9fc17d2009-10-22 21:23:35 +00001152 if( !pStmt ){
1153 /* this happens for a comment or white-space */
1154 zSql = zLeftover;
drhf0693c82011-10-11 20:41:54 +00001155 while( IsSpace(zSql[0]) ) zSql++;
shaneb9fc17d2009-10-22 21:23:35 +00001156 continue;
1157 }
shane626a6e42009-10-22 17:30:15 +00001158
shaneh642d8b82010-07-28 16:05:34 +00001159 /* save off the prepared statment handle and reset row count */
1160 if( pArg ){
1161 pArg->pStmt = pStmt;
1162 pArg->cnt = 0;
1163 }
1164
shanehb7977c52010-01-18 18:17:10 +00001165 /* echo the sql statement if echo on */
shaneh642d8b82010-07-28 16:05:34 +00001166 if( pArg && pArg->echoOn ){
drha8c62df2010-02-15 15:47:18 +00001167 const char *zStmtSql = sqlite3_sql(pStmt);
shaneh642d8b82010-07-28 16:05:34 +00001168 fprintf(pArg->out, "%s\n", zStmtSql ? zStmtSql : zSql);
drha8c62df2010-02-15 15:47:18 +00001169 }
shanehb7977c52010-01-18 18:17:10 +00001170
drh7e02e5e2011-12-06 19:44:51 +00001171 /* Output TESTCTRL_EXPLAIN text of requested */
1172 if( pArg && pArg->mode==MODE_Explain ){
1173 const char *zExplain = 0;
1174 sqlite3_test_control(SQLITE_TESTCTRL_EXPLAIN_STMT, pStmt, &zExplain);
1175 if( zExplain && zExplain[0] ){
1176 fprintf(pArg->out, "%s", zExplain);
1177 }
1178 }
1179
shaneb9fc17d2009-10-22 21:23:35 +00001180 /* perform the first step. this will tell us if we
1181 ** have a result set or not and how wide it is.
1182 */
1183 rc = sqlite3_step(pStmt);
1184 /* if we have a result set... */
1185 if( SQLITE_ROW == rc ){
1186 /* if we have a callback... */
1187 if( xCallback ){
1188 /* allocate space for col name ptr, value ptr, and type */
1189 int nCol = sqlite3_column_count(pStmt);
1190 void *pData = sqlite3_malloc(3*nCol*sizeof(const char*) + 1);
1191 if( !pData ){
1192 rc = SQLITE_NOMEM;
1193 }else{
1194 char **azCols = (char **)pData; /* Names of result columns */
1195 char **azVals = &azCols[nCol]; /* Results */
1196 int *aiTypes = (int *)&azVals[nCol]; /* Result types */
1197 int i;
1198 assert(sizeof(int) <= sizeof(char *));
1199 /* save off ptrs to column names */
1200 for(i=0; i<nCol; i++){
1201 azCols[i] = (char *)sqlite3_column_name(pStmt, i);
1202 }
shaneb9fc17d2009-10-22 21:23:35 +00001203 do{
1204 /* extract the data and data types */
1205 for(i=0; i<nCol; i++){
1206 azVals[i] = (char *)sqlite3_column_text(pStmt, i);
1207 aiTypes[i] = sqlite3_column_type(pStmt, i);
1208 if( !azVals[i] && (aiTypes[i]!=SQLITE_NULL) ){
1209 rc = SQLITE_NOMEM;
1210 break; /* from for */
1211 }
1212 } /* end for */
1213
1214 /* if data and types extracted successfully... */
1215 if( SQLITE_ROW == rc ){
1216 /* call the supplied callback with the result row data */
1217 if( xCallback(pArg, nCol, azVals, azCols, aiTypes) ){
1218 rc = SQLITE_ABORT;
1219 }else{
1220 rc = sqlite3_step(pStmt);
1221 }
1222 }
1223 } while( SQLITE_ROW == rc );
1224 sqlite3_free(pData);
shaneb9fc17d2009-10-22 21:23:35 +00001225 }
1226 }else{
1227 do{
1228 rc = sqlite3_step(pStmt);
1229 } while( rc == SQLITE_ROW );
1230 }
1231 }
1232
shaneh642d8b82010-07-28 16:05:34 +00001233 /* print usage stats if stats on */
1234 if( pArg && pArg->statsOn ){
1235 display_stats(db, pArg, 0);
1236 }
1237
dan4564ced2010-01-05 04:59:56 +00001238 /* Finalize the statement just executed. If this fails, save a
1239 ** copy of the error message. Otherwise, set zSql to point to the
1240 ** next statement to execute. */
drhb07028f2011-10-14 21:49:18 +00001241 rc2 = sqlite3_finalize(pStmt);
1242 if( rc!=SQLITE_NOMEM ) rc = rc2;
dan4564ced2010-01-05 04:59:56 +00001243 if( rc==SQLITE_OK ){
shaneb9fc17d2009-10-22 21:23:35 +00001244 zSql = zLeftover;
drhf0693c82011-10-11 20:41:54 +00001245 while( IsSpace(zSql[0]) ) zSql++;
dan4564ced2010-01-05 04:59:56 +00001246 }else if( pzErrMsg ){
1247 *pzErrMsg = save_err_msg(db);
shane626a6e42009-10-22 17:30:15 +00001248 }
shaneh642d8b82010-07-28 16:05:34 +00001249
1250 /* clear saved stmt handle */
1251 if( pArg ){
1252 pArg->pStmt = NULL;
1253 }
shane626a6e42009-10-22 17:30:15 +00001254 }
shaneb9fc17d2009-10-22 21:23:35 +00001255 } /* end while */
shane626a6e42009-10-22 17:30:15 +00001256
1257 return rc;
1258}
1259
drhdd3d4592004-08-30 01:54:05 +00001260
drh33048c02001-10-01 14:29:22 +00001261/*
drh4c653a02000-06-07 01:27:47 +00001262** This is a different callback routine used for dumping the database.
1263** Each row received by this callback consists of a table name,
1264** the table type ("index" or "table") and SQL to create the table.
1265** This routine should print text sufficient to recreate the table.
1266*/
1267static int dump_callback(void *pArg, int nArg, char **azArg, char **azCol){
danielk19772a02e332004-06-05 08:04:36 +00001268 int rc;
1269 const char *zTable;
1270 const char *zType;
1271 const char *zSql;
drh157e29a2009-05-21 15:15:00 +00001272 const char *zPrepStmt = 0;
drhdaffd0e2001-04-11 14:28:42 +00001273 struct callback_data *p = (struct callback_data *)pArg;
danielk19772a02e332004-06-05 08:04:36 +00001274
drh902b9ee2008-12-05 17:17:07 +00001275 UNUSED_PARAMETER(azCol);
drh4c653a02000-06-07 01:27:47 +00001276 if( nArg!=3 ) return 1;
danielk19772a02e332004-06-05 08:04:36 +00001277 zTable = azArg[0];
1278 zType = azArg[1];
1279 zSql = azArg[2];
1280
drh00b950d2005-09-11 02:03:03 +00001281 if( strcmp(zTable, "sqlite_sequence")==0 ){
drh157e29a2009-05-21 15:15:00 +00001282 zPrepStmt = "DELETE FROM sqlite_sequence;\n";
drh00b950d2005-09-11 02:03:03 +00001283 }else if( strcmp(zTable, "sqlite_stat1")==0 ){
1284 fprintf(p->out, "ANALYZE sqlite_master;\n");
1285 }else if( strncmp(zTable, "sqlite_", 7)==0 ){
1286 return 0;
drh45e29d82006-11-20 16:21:10 +00001287 }else if( strncmp(zSql, "CREATE VIRTUAL TABLE", 20)==0 ){
1288 char *zIns;
1289 if( !p->writableSchema ){
1290 fprintf(p->out, "PRAGMA writable_schema=ON;\n");
1291 p->writableSchema = 1;
1292 }
1293 zIns = sqlite3_mprintf(
1294 "INSERT INTO sqlite_master(type,name,tbl_name,rootpage,sql)"
1295 "VALUES('table','%q','%q',0,'%q');",
1296 zTable, zTable, zSql);
1297 fprintf(p->out, "%s\n", zIns);
1298 sqlite3_free(zIns);
1299 return 0;
drh00b950d2005-09-11 02:03:03 +00001300 }else{
1301 fprintf(p->out, "%s;\n", zSql);
drhf8eb96a2005-02-03 00:42:34 +00001302 }
danielk19772a02e332004-06-05 08:04:36 +00001303
1304 if( strcmp(zType, "table")==0 ){
1305 sqlite3_stmt *pTableInfo = 0;
danielk19772a02e332004-06-05 08:04:36 +00001306 char *zSelect = 0;
1307 char *zTableInfo = 0;
1308 char *zTmp = 0;
drh157e29a2009-05-21 15:15:00 +00001309 int nRow = 0;
danielk19772a02e332004-06-05 08:04:36 +00001310
1311 zTableInfo = appendText(zTableInfo, "PRAGMA table_info(", 0);
1312 zTableInfo = appendText(zTableInfo, zTable, '"');
1313 zTableInfo = appendText(zTableInfo, ");", 0);
1314
1315 rc = sqlite3_prepare(p->db, zTableInfo, -1, &pTableInfo, 0);
drh157e29a2009-05-21 15:15:00 +00001316 free(zTableInfo);
danielk19772a02e332004-06-05 08:04:36 +00001317 if( rc!=SQLITE_OK || !pTableInfo ){
1318 return 1;
1319 }
1320
1321 zSelect = appendText(zSelect, "SELECT 'INSERT INTO ' || ", 0);
drhbf92ec02012-03-22 12:50:34 +00001322 /* Always quote the table name, even if it appears to be pure ascii,
1323 ** in case it is a keyword. Ex: INSERT INTO "table" ... */
1324 zTmp = appendText(zTmp, zTable, '"');
danielk19772a02e332004-06-05 08:04:36 +00001325 if( zTmp ){
1326 zSelect = appendText(zSelect, zTmp, '\'');
drh85e72432012-04-11 11:38:53 +00001327 free(zTmp);
danielk19772a02e332004-06-05 08:04:36 +00001328 }
1329 zSelect = appendText(zSelect, " || ' VALUES(' || ", 0);
1330 rc = sqlite3_step(pTableInfo);
1331 while( rc==SQLITE_ROW ){
danielk19772e588c72005-12-09 14:25:08 +00001332 const char *zText = (const char *)sqlite3_column_text(pTableInfo, 1);
danielk19773f41e972004-06-08 00:39:01 +00001333 zSelect = appendText(zSelect, "quote(", 0);
danielk19772e588c72005-12-09 14:25:08 +00001334 zSelect = appendText(zSelect, zText, '"');
danielk19772a02e332004-06-05 08:04:36 +00001335 rc = sqlite3_step(pTableInfo);
1336 if( rc==SQLITE_ROW ){
drhb21a8e42012-01-28 21:08:51 +00001337 zSelect = appendText(zSelect, "), ", 0);
danielk19772a02e332004-06-05 08:04:36 +00001338 }else{
1339 zSelect = appendText(zSelect, ") ", 0);
1340 }
drh157e29a2009-05-21 15:15:00 +00001341 nRow++;
danielk19772a02e332004-06-05 08:04:36 +00001342 }
1343 rc = sqlite3_finalize(pTableInfo);
drh157e29a2009-05-21 15:15:00 +00001344 if( rc!=SQLITE_OK || nRow==0 ){
1345 free(zSelect);
danielk19772a02e332004-06-05 08:04:36 +00001346 return 1;
1347 }
1348 zSelect = appendText(zSelect, "|| ')' FROM ", 0);
1349 zSelect = appendText(zSelect, zTable, '"');
1350
drh2f464a02011-10-13 00:41:49 +00001351 rc = run_table_dump_query(p, zSelect, zPrepStmt);
drhdd3d4592004-08-30 01:54:05 +00001352 if( rc==SQLITE_CORRUPT ){
1353 zSelect = appendText(zSelect, " ORDER BY rowid DESC", 0);
drh2f464a02011-10-13 00:41:49 +00001354 run_table_dump_query(p, zSelect, 0);
drhdd3d4592004-08-30 01:54:05 +00001355 }
drh85e72432012-04-11 11:38:53 +00001356 free(zSelect);
drh4c653a02000-06-07 01:27:47 +00001357 }
drh4c653a02000-06-07 01:27:47 +00001358 return 0;
1359}
1360
1361/*
drh45e29d82006-11-20 16:21:10 +00001362** Run zQuery. Use dump_callback() as the callback routine so that
1363** the contents of the query are output as SQL statements.
1364**
drhdd3d4592004-08-30 01:54:05 +00001365** If we get a SQLITE_CORRUPT error, rerun the query after appending
1366** "ORDER BY rowid DESC" to the end.
1367*/
1368static int run_schema_dump_query(
1369 struct callback_data *p,
drh2f464a02011-10-13 00:41:49 +00001370 const char *zQuery
drhdd3d4592004-08-30 01:54:05 +00001371){
1372 int rc;
drh2f464a02011-10-13 00:41:49 +00001373 char *zErr = 0;
1374 rc = sqlite3_exec(p->db, zQuery, dump_callback, p, &zErr);
drhdd3d4592004-08-30 01:54:05 +00001375 if( rc==SQLITE_CORRUPT ){
1376 char *zQ2;
drh4f21c4a2008-12-10 22:15:00 +00001377 int len = strlen30(zQuery);
drh2f464a02011-10-13 00:41:49 +00001378 fprintf(p->out, "/****** CORRUPTION ERROR *******/\n");
1379 if( zErr ){
1380 fprintf(p->out, "/****** %s ******/\n", zErr);
1381 sqlite3_free(zErr);
1382 zErr = 0;
1383 }
drhdd3d4592004-08-30 01:54:05 +00001384 zQ2 = malloc( len+100 );
1385 if( zQ2==0 ) return rc;
drh8c5058b2012-04-16 17:22:30 +00001386 sqlite3_snprintf(len+100, zQ2, "%s ORDER BY rowid DESC", zQuery);
drh2f464a02011-10-13 00:41:49 +00001387 rc = sqlite3_exec(p->db, zQ2, dump_callback, p, &zErr);
1388 if( rc ){
1389 fprintf(p->out, "/****** ERROR: %s ******/\n", zErr);
1390 }else{
1391 rc = SQLITE_CORRUPT;
1392 }
1393 sqlite3_free(zErr);
drhdd3d4592004-08-30 01:54:05 +00001394 free(zQ2);
1395 }
1396 return rc;
1397}
1398
1399/*
drh75897232000-05-29 14:26:00 +00001400** Text of a help message
1401*/
persicom1d0b8722002-04-18 02:53:04 +00001402static char zHelp[] =
drh9ff849f2009-02-04 20:55:57 +00001403 ".backup ?DB? FILE Backup DB (default \"main\") to FILE\n"
drh20f99c42007-01-08 14:31:35 +00001404 ".bail ON|OFF Stop after hitting an error. Default OFF\n"
jplyon6a65bb32003-05-04 07:25:57 +00001405 ".databases List names and files of attached databases\n"
drhb860bc92004-08-04 15:16:55 +00001406 ".dump ?TABLE? ... Dump the database in an SQL text format\n"
shane86f5bdb2009-10-24 02:00:07 +00001407 " If TABLE specified, only dump tables matching\n"
1408 " LIKE pattern TABLE.\n"
drhdaffd0e2001-04-11 14:28:42 +00001409 ".echo ON|OFF Turn command echo on or off\n"
drh75897232000-05-29 14:26:00 +00001410 ".exit Exit this program\n"
shanehe2aa9d72009-11-06 17:20:17 +00001411 ".explain ?ON|OFF? Turn output mode suitable for EXPLAIN on or off.\n"
1412 " With no args, it turns EXPLAIN on.\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001413 ".header(s) ON|OFF Turn display of headers on or off\n"
drh75897232000-05-29 14:26:00 +00001414 ".help Show this message\n"
drhb860bc92004-08-04 15:16:55 +00001415 ".import FILE TABLE Import data from FILE into TABLE\n"
shane86f5bdb2009-10-24 02:00:07 +00001416 ".indices ?TABLE? Show names of all indices\n"
1417 " If TABLE specified, only show indices for tables\n"
1418 " matching LIKE pattern TABLE.\n"
drhae5e4452007-05-03 17:18:36 +00001419#ifdef SQLITE_ENABLE_IOTRACE
1420 ".iotrace FILE Enable I/O diagnostic logging to FILE\n"
1421#endif
drh70df4fe2006-06-13 15:12:21 +00001422#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh1e397f82006-06-08 15:28:43 +00001423 ".load FILE ?ENTRY? Load an extension library\n"
drh70df4fe2006-06-13 15:12:21 +00001424#endif
drh127f9d72010-02-23 01:47:00 +00001425 ".log FILE|off Turn logging on or off. FILE can be stderr/stdout\n"
danielk19776b77a362005-01-13 11:10:25 +00001426 ".mode MODE ?TABLE? Set output mode where MODE is one of:\n"
drh3b584fa2004-09-24 12:50:03 +00001427 " csv Comma-separated values\n"
drhb860bc92004-08-04 15:16:55 +00001428 " column Left-aligned columns. (See .width)\n"
1429 " html HTML <table> code\n"
1430 " insert SQL insert statements for TABLE\n"
1431 " line One value per line\n"
1432 " list Values delimited by .separator string\n"
1433 " tabs Tab-separated values\n"
1434 " tcl TCL list elements\n"
drh078b1fd2012-09-21 13:40:02 +00001435 ".nullvalue STRING Use STRING in place of NULL values\n"
drh75897232000-05-29 14:26:00 +00001436 ".output FILENAME Send output to FILENAME\n"
1437 ".output stdout Send output to the screen\n"
drh078b1fd2012-09-21 13:40:02 +00001438 ".print STRING... Print literal STRING\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001439 ".prompt MAIN CONTINUE Replace the standard prompts\n"
persicom7e2dfdd2002-04-18 02:46:52 +00001440 ".quit Exit this program\n"
drhdaffd0e2001-04-11 14:28:42 +00001441 ".read FILENAME Execute SQL in FILENAME\n"
drh9ff849f2009-02-04 20:55:57 +00001442 ".restore ?DB? FILE Restore content of DB (default \"main\") from FILE\n"
drh75897232000-05-29 14:26:00 +00001443 ".schema ?TABLE? Show the CREATE statements\n"
shane86f5bdb2009-10-24 02:00:07 +00001444 " If TABLE specified, only show tables matching\n"
1445 " LIKE pattern TABLE.\n"
drhb860bc92004-08-04 15:16:55 +00001446 ".separator STRING Change separator used by output mode and .import\n"
drhdd45df82002-04-18 12:39:03 +00001447 ".show Show the current values for various settings\n"
shaneh642d8b82010-07-28 16:05:34 +00001448 ".stats ON|OFF Turn stats on or off\n"
shane86f5bdb2009-10-24 02:00:07 +00001449 ".tables ?TABLE? List names of tables\n"
1450 " If TABLE specified, only list tables matching\n"
1451 " LIKE pattern TABLE.\n"
drh2dfbbca2000-07-28 14:32:48 +00001452 ".timeout MS Try opening locked tables for MS milliseconds\n"
drh42f64e52012-04-04 16:56:23 +00001453 ".trace FILE|off Output each SQL statement as it is run\n"
drhde60fc22011-12-14 17:53:36 +00001454 ".vfsname ?AUX? Print the name of the VFS stack\n"
shanehe2aa9d72009-11-06 17:20:17 +00001455 ".width NUM1 NUM2 ... Set column widths for \"column\" mode\n"
drh75897232000-05-29 14:26:00 +00001456;
1457
shaneb320ccd2009-10-21 03:42:58 +00001458static char zTimerHelp[] =
1459 ".timer ON|OFF Turn the CPU timer measurement on or off\n"
1460;
1461
drhdaffd0e2001-04-11 14:28:42 +00001462/* Forward reference */
drhc28490c2006-10-26 14:25:58 +00001463static int process_input(struct callback_data *p, FILE *in);
drhdaffd0e2001-04-11 14:28:42 +00001464
drh75897232000-05-29 14:26:00 +00001465/*
drh44c2eb12003-04-30 11:38:26 +00001466** Make sure the database is open. If it is not, then open it. If
1467** the database fails to open, print an error message and exit.
1468*/
1469static void open_db(struct callback_data *p){
1470 if( p->db==0 ){
drhbbb0be82012-06-27 16:12:27 +00001471 sqlite3_initialize();
danielk19774f057f92004-06-08 00:02:33 +00001472 sqlite3_open(p->zDbFilename, &p->db);
danielk197780290862004-05-22 09:21:21 +00001473 db = p->db;
drh4cea5ba2008-05-05 16:27:24 +00001474 if( db && sqlite3_errcode(db)==SQLITE_OK ){
1475 sqlite3_create_function(db, "shellstatic", 0, SQLITE_UTF8, 0,
1476 shellstaticFunc, 0, 0);
1477 }
1478 if( db==0 || SQLITE_OK!=sqlite3_errcode(db) ){
shane86f5bdb2009-10-24 02:00:07 +00001479 fprintf(stderr,"Error: unable to open database \"%s\": %s\n",
danielk197780290862004-05-22 09:21:21 +00001480 p->zDbFilename, sqlite3_errmsg(db));
drh22fbcb82004-02-01 01:22:50 +00001481 exit(1);
drh44c2eb12003-04-30 11:38:26 +00001482 }
drhc2e87a32006-06-27 15:16:14 +00001483#ifndef SQLITE_OMIT_LOAD_EXTENSION
1484 sqlite3_enable_load_extension(p->db, 1);
1485#endif
drh44c2eb12003-04-30 11:38:26 +00001486 }
1487}
1488
1489/*
drhfeac5f82004-08-01 00:10:45 +00001490** Do C-language style dequoting.
1491**
1492** \t -> tab
1493** \n -> newline
1494** \r -> carriage return
1495** \NNN -> ascii character NNN in octal
1496** \\ -> backslash
1497*/
1498static void resolve_backslashes(char *z){
shane7d3846a2008-12-11 02:58:26 +00001499 int i, j;
1500 char c;
drhfeac5f82004-08-01 00:10:45 +00001501 for(i=j=0; (c = z[i])!=0; i++, j++){
1502 if( c=='\\' ){
1503 c = z[++i];
1504 if( c=='n' ){
1505 c = '\n';
1506 }else if( c=='t' ){
1507 c = '\t';
1508 }else if( c=='r' ){
1509 c = '\r';
1510 }else if( c>='0' && c<='7' ){
drhaa816082005-12-29 12:53:09 +00001511 c -= '0';
drhfeac5f82004-08-01 00:10:45 +00001512 if( z[i+1]>='0' && z[i+1]<='7' ){
1513 i++;
1514 c = (c<<3) + z[i] - '0';
1515 if( z[i+1]>='0' && z[i+1]<='7' ){
1516 i++;
1517 c = (c<<3) + z[i] - '0';
1518 }
1519 }
1520 }
1521 }
1522 z[j] = c;
1523 }
1524 z[j] = 0;
1525}
1526
1527/*
drh348d19c2013-06-03 12:47:43 +00001528** Return the value of a hexadecimal digit. Return -1 if the input
1529** is not a hex digit.
drhc28490c2006-10-26 14:25:58 +00001530*/
drh348d19c2013-06-03 12:47:43 +00001531static int hexDigitValue(char c){
1532 if( c>='0' && c<='9' ) return c - '0';
1533 if( c>='a' && c<='f' ) return c - 'a' + 10;
1534 if( c>='A' && c<='F' ) return c - 'A' + 10;
1535 return -1;
drhc28490c2006-10-26 14:25:58 +00001536}
1537
1538/*
drh7d9f3942013-04-03 01:26:54 +00001539** Interpret zArg as an integer value, possibly with suffixes.
1540*/
1541static sqlite3_int64 integerValue(const char *zArg){
1542 sqlite3_int64 v = 0;
1543 static const struct { char *zSuffix; int iMult; } aMult[] = {
1544 { "KiB", 1024 },
1545 { "MiB", 1024*1024 },
1546 { "GiB", 1024*1024*1024 },
1547 { "KB", 1000 },
1548 { "MB", 1000000 },
1549 { "GB", 1000000000 },
1550 { "K", 1000 },
1551 { "M", 1000000 },
1552 { "G", 1000000000 },
1553 };
1554 int i;
1555 int isNeg = 0;
1556 if( zArg[0]=='-' ){
1557 isNeg = 1;
1558 zArg++;
1559 }else if( zArg[0]=='+' ){
1560 zArg++;
1561 }
drh348d19c2013-06-03 12:47:43 +00001562 if( zArg[0]=='0' && zArg[1]=='x' ){
1563 int x;
1564 zArg += 2;
1565 while( (x = hexDigitValue(zArg[0]))>=0 ){
1566 v = (v<<4) + x;
1567 zArg++;
1568 }
1569 }else{
1570 while( IsDigit(zArg[0]) ){
1571 v = v*10 + zArg[0] - '0';
1572 zArg++;
1573 }
drh7d9f3942013-04-03 01:26:54 +00001574 }
drhc2bed0a2013-05-24 11:57:50 +00001575 for(i=0; i<ArraySize(aMult); i++){
drh7d9f3942013-04-03 01:26:54 +00001576 if( sqlite3_stricmp(aMult[i].zSuffix, zArg)==0 ){
1577 v *= aMult[i].iMult;
1578 break;
1579 }
1580 }
1581 return isNeg? -v : v;
1582}
1583
1584/*
drh348d19c2013-06-03 12:47:43 +00001585** Interpret zArg as either an integer or a boolean value. Return 1 or 0
1586** for TRUE and FALSE. Return the integer value if appropriate.
1587*/
1588static int booleanValue(char *zArg){
1589 int i;
1590 if( zArg[0]=='0' && zArg[1]=='x' ){
1591 for(i=2; hexDigitValue(zArg[i])>=0; i++){}
1592 }else{
1593 for(i=0; zArg[i]>='0' && zArg[i]<='9'; i++){}
1594 }
1595 if( i>0 && zArg[i]==0 ) return (int)(integerValue(zArg) & 0xffffffff);
1596 if( sqlite3_stricmp(zArg, "on")==0 || sqlite3_stricmp(zArg,"yes")==0 ){
1597 return 1;
1598 }
1599 if( sqlite3_stricmp(zArg, "off")==0 || sqlite3_stricmp(zArg,"no")==0 ){
1600 return 0;
1601 }
1602 fprintf(stderr, "ERROR: Not a boolean value: \"%s\". Assuming \"no\".\n",
1603 zArg);
1604 return 0;
1605}
1606
1607/*
drh42f64e52012-04-04 16:56:23 +00001608** Close an output file, assuming it is not stderr or stdout
1609*/
1610static void output_file_close(FILE *f){
1611 if( f && f!=stdout && f!=stderr ) fclose(f);
1612}
1613
1614/*
1615** Try to open an output file. The names "stdout" and "stderr" are
1616** recognized and do the right thing. NULL is returned if the output
1617** filename is "off".
1618*/
1619static FILE *output_file_open(const char *zFile){
1620 FILE *f;
1621 if( strcmp(zFile,"stdout")==0 ){
1622 f = stdout;
1623 }else if( strcmp(zFile, "stderr")==0 ){
1624 f = stderr;
1625 }else if( strcmp(zFile, "off")==0 ){
1626 f = 0;
1627 }else{
1628 f = fopen(zFile, "wb");
1629 if( f==0 ){
1630 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
1631 }
1632 }
1633 return f;
1634}
1635
1636/*
1637** A routine for handling output from sqlite3_trace().
1638*/
1639static void sql_trace_callback(void *pArg, const char *z){
1640 FILE *f = (FILE*)pArg;
1641 if( f ) fprintf(f, "%s\n", z);
1642}
1643
1644/*
drhd8621b92012-04-17 09:09:33 +00001645** A no-op routine that runs with the ".breakpoint" doc-command. This is
1646** a useful spot to set a debugger breakpoint.
1647*/
1648static void test_breakpoint(void){
1649 static int nCall = 0;
1650 nCall++;
1651}
1652
1653/*
drh75897232000-05-29 14:26:00 +00001654** If an input line begins with "." then invoke this routine to
1655** process that line.
drh67505e72002-04-19 12:34:06 +00001656**
drh47ad6842006-11-08 12:25:42 +00001657** Return 1 on error, 2 to exit, and 0 otherwise.
drh75897232000-05-29 14:26:00 +00001658*/
drh44c2eb12003-04-30 11:38:26 +00001659static int do_meta_command(char *zLine, struct callback_data *p){
drh75897232000-05-29 14:26:00 +00001660 int i = 1;
1661 int nArg = 0;
1662 int n, c;
drh67505e72002-04-19 12:34:06 +00001663 int rc = 0;
drh75897232000-05-29 14:26:00 +00001664 char *azArg[50];
1665
1666 /* Parse the input line into tokens.
1667 */
1668 while( zLine[i] && nArg<ArraySize(azArg) ){
drhf0693c82011-10-11 20:41:54 +00001669 while( IsSpace(zLine[i]) ){ i++; }
drh06333682004-03-09 13:37:45 +00001670 if( zLine[i]==0 ) break;
drh75897232000-05-29 14:26:00 +00001671 if( zLine[i]=='\'' || zLine[i]=='"' ){
1672 int delim = zLine[i++];
1673 azArg[nArg++] = &zLine[i];
1674 while( zLine[i] && zLine[i]!=delim ){ i++; }
1675 if( zLine[i]==delim ){
1676 zLine[i++] = 0;
1677 }
drhfeac5f82004-08-01 00:10:45 +00001678 if( delim=='"' ) resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00001679 }else{
1680 azArg[nArg++] = &zLine[i];
drhf0693c82011-10-11 20:41:54 +00001681 while( zLine[i] && !IsSpace(zLine[i]) ){ i++; }
drh75897232000-05-29 14:26:00 +00001682 if( zLine[i] ) zLine[i++] = 0;
drhfeac5f82004-08-01 00:10:45 +00001683 resolve_backslashes(azArg[nArg-1]);
drh75897232000-05-29 14:26:00 +00001684 }
1685 }
1686
1687 /* Process the input line.
1688 */
shane9bd1b442009-10-23 01:27:39 +00001689 if( nArg==0 ) return 0; /* no tokens, no error */
drh4f21c4a2008-12-10 22:15:00 +00001690 n = strlen30(azArg[0]);
drh75897232000-05-29 14:26:00 +00001691 c = azArg[0][0];
drhbc46f022013-01-23 18:53:23 +00001692 if( c=='b' && n>=3 && strncmp(azArg[0], "backup", n)==0 ){
1693 const char *zDestFile = 0;
1694 const char *zDb = 0;
1695 const char *zKey = 0;
drh9ff849f2009-02-04 20:55:57 +00001696 sqlite3 *pDest;
1697 sqlite3_backup *pBackup;
drhbc46f022013-01-23 18:53:23 +00001698 int j;
1699 for(j=1; j<nArg; j++){
1700 const char *z = azArg[j];
1701 if( z[0]=='-' ){
1702 while( z[0]=='-' ) z++;
1703 if( strcmp(z,"key")==0 && j<nArg-1 ){
1704 zKey = azArg[++j];
1705 }else
1706 {
1707 fprintf(stderr, "unknown option: %s\n", azArg[j]);
1708 return 1;
1709 }
1710 }else if( zDestFile==0 ){
1711 zDestFile = azArg[j];
1712 }else if( zDb==0 ){
1713 zDb = zDestFile;
1714 zDestFile = azArg[j];
1715 }else{
1716 fprintf(stderr, "too many arguments to .backup\n");
1717 return 1;
1718 }
drh9ff849f2009-02-04 20:55:57 +00001719 }
drhbc46f022013-01-23 18:53:23 +00001720 if( zDestFile==0 ){
1721 fprintf(stderr, "missing FILENAME argument on .backup\n");
1722 return 1;
1723 }
1724 if( zDb==0 ) zDb = "main";
drh9ff849f2009-02-04 20:55:57 +00001725 rc = sqlite3_open(zDestFile, &pDest);
1726 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00001727 fprintf(stderr, "Error: cannot open \"%s\"\n", zDestFile);
drh9ff849f2009-02-04 20:55:57 +00001728 sqlite3_close(pDest);
1729 return 1;
1730 }
drhbc46f022013-01-23 18:53:23 +00001731#ifdef SQLITE_HAS_CODEC
1732 sqlite3_key(pDest, zKey, (int)strlen(zKey));
1733#else
1734 (void)zKey;
1735#endif
drhdc2c4912009-02-04 22:46:47 +00001736 open_db(p);
drh9ff849f2009-02-04 20:55:57 +00001737 pBackup = sqlite3_backup_init(pDest, "main", p->db, zDb);
1738 if( pBackup==0 ){
1739 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
1740 sqlite3_close(pDest);
1741 return 1;
1742 }
1743 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK ){}
1744 sqlite3_backup_finish(pBackup);
1745 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00001746 rc = 0;
drh9ff849f2009-02-04 20:55:57 +00001747 }else{
1748 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(pDest));
shane9bd1b442009-10-23 01:27:39 +00001749 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00001750 }
1751 sqlite3_close(pDest);
1752 }else
1753
shanehe2aa9d72009-11-06 17:20:17 +00001754 if( c=='b' && n>=3 && strncmp(azArg[0], "bail", n)==0 && nArg>1 && nArg<3 ){
drhc49f44e2006-10-26 18:15:42 +00001755 bail_on_error = booleanValue(azArg[1]);
1756 }else
1757
drhd8621b92012-04-17 09:09:33 +00001758 /* The undocumented ".breakpoint" command causes a call to the no-op
1759 ** routine named test_breakpoint().
1760 */
1761 if( c=='b' && n>=3 && strncmp(azArg[0], "breakpoint", n)==0 ){
1762 test_breakpoint();
1763 }else
1764
shanehe2aa9d72009-11-06 17:20:17 +00001765 if( c=='d' && n>1 && strncmp(azArg[0], "databases", n)==0 && nArg==1 ){
jplyon672a1ed2003-05-11 20:07:05 +00001766 struct callback_data data;
1767 char *zErrMsg = 0;
jplyon6a65bb32003-05-04 07:25:57 +00001768 open_db(p);
jplyon672a1ed2003-05-11 20:07:05 +00001769 memcpy(&data, p, sizeof(data));
drhd8885442004-03-17 23:42:12 +00001770 data.showHeader = 1;
jplyon672a1ed2003-05-11 20:07:05 +00001771 data.mode = MODE_Column;
drhd8885442004-03-17 23:42:12 +00001772 data.colWidth[0] = 3;
1773 data.colWidth[1] = 15;
1774 data.colWidth[2] = 58;
drh0b2110c2004-10-26 00:08:10 +00001775 data.cnt = 0;
danielk19776f8a5032004-05-10 10:34:51 +00001776 sqlite3_exec(p->db, "PRAGMA database_list; ", callback, &data, &zErrMsg);
jplyon672a1ed2003-05-11 20:07:05 +00001777 if( zErrMsg ){
1778 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00001779 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00001780 rc = 1;
jplyon6a65bb32003-05-04 07:25:57 +00001781 }
1782 }else
1783
shanehe2aa9d72009-11-06 17:20:17 +00001784 if( c=='d' && strncmp(azArg[0], "dump", n)==0 && nArg<3 ){
drh44c2eb12003-04-30 11:38:26 +00001785 open_db(p);
drhf1dfc4f2009-09-23 15:51:35 +00001786 /* When playing back a "dump", the content might appear in an order
1787 ** which causes immediate foreign key constraints to be violated.
1788 ** So disable foreign-key constraint enforcement to prevent problems. */
1789 fprintf(p->out, "PRAGMA foreign_keys=OFF;\n");
drh33048c02001-10-01 14:29:22 +00001790 fprintf(p->out, "BEGIN TRANSACTION;\n");
drh45e29d82006-11-20 16:21:10 +00001791 p->writableSchema = 0;
drh56197952011-10-13 16:30:13 +00001792 sqlite3_exec(p->db, "SAVEPOINT dump; PRAGMA writable_schema=ON", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00001793 p->nErr = 0;
drh4c653a02000-06-07 01:27:47 +00001794 if( nArg==1 ){
drhdd3d4592004-08-30 01:54:05 +00001795 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00001796 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00001797 "WHERE sql NOT NULL AND type=='table' AND name!='sqlite_sequence'"
drh4f324762009-05-21 14:51:03 +00001798 );
1799 run_schema_dump_query(p,
1800 "SELECT name, type, sql FROM sqlite_master "
drh2f464a02011-10-13 00:41:49 +00001801 "WHERE name=='sqlite_sequence'"
drh0b9a5942006-09-13 20:22:02 +00001802 );
drh2f464a02011-10-13 00:41:49 +00001803 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00001804 "SELECT sql FROM sqlite_master "
drh157e29a2009-05-21 15:15:00 +00001805 "WHERE sql NOT NULL AND type IN ('index','trigger','view')", 0
drha18c5682000-10-08 22:20:57 +00001806 );
drh4c653a02000-06-07 01:27:47 +00001807 }else{
1808 int i;
drhdd3d4592004-08-30 01:54:05 +00001809 for(i=1; i<nArg; i++){
danielk1977bc6ada42004-06-30 08:20:16 +00001810 zShellStatic = azArg[i];
drhdd3d4592004-08-30 01:54:05 +00001811 run_schema_dump_query(p,
drha18c5682000-10-08 22:20:57 +00001812 "SELECT name, type, sql FROM sqlite_master "
drhdd3d4592004-08-30 01:54:05 +00001813 "WHERE tbl_name LIKE shellstatic() AND type=='table'"
drh2f464a02011-10-13 00:41:49 +00001814 " AND sql NOT NULL");
1815 run_table_dump_query(p,
drh0b9a5942006-09-13 20:22:02 +00001816 "SELECT sql FROM sqlite_master "
drh45e29d82006-11-20 16:21:10 +00001817 "WHERE sql NOT NULL"
1818 " AND type IN ('index','trigger','view')"
drh157e29a2009-05-21 15:15:00 +00001819 " AND tbl_name LIKE shellstatic()", 0
drh0b9a5942006-09-13 20:22:02 +00001820 );
danielk1977bc6ada42004-06-30 08:20:16 +00001821 zShellStatic = 0;
drh4c653a02000-06-07 01:27:47 +00001822 }
1823 }
drh45e29d82006-11-20 16:21:10 +00001824 if( p->writableSchema ){
drh56197952011-10-13 16:30:13 +00001825 fprintf(p->out, "PRAGMA writable_schema=OFF;\n");
drh45e29d82006-11-20 16:21:10 +00001826 p->writableSchema = 0;
1827 }
drh56197952011-10-13 16:30:13 +00001828 sqlite3_exec(p->db, "PRAGMA writable_schema=OFF;", 0, 0, 0);
1829 sqlite3_exec(p->db, "RELEASE dump;", 0, 0, 0);
drh2f464a02011-10-13 00:41:49 +00001830 fprintf(p->out, p->nErr ? "ROLLBACK; -- due to errors\n" : "COMMIT;\n");
drh4c653a02000-06-07 01:27:47 +00001831 }else
drh75897232000-05-29 14:26:00 +00001832
shanehe2aa9d72009-11-06 17:20:17 +00001833 if( c=='e' && strncmp(azArg[0], "echo", n)==0 && nArg>1 && nArg<3 ){
drhc28490c2006-10-26 14:25:58 +00001834 p->echoOn = booleanValue(azArg[1]);
drhdaffd0e2001-04-11 14:28:42 +00001835 }else
1836
drhd3ac7d92013-01-25 18:33:43 +00001837 if( c=='e' && strncmp(azArg[0], "exit", n)==0 ){
drh348d19c2013-06-03 12:47:43 +00001838 if( nArg>1 && (rc = (int)integerValue(azArg[1]))!=0 ) exit(rc);
drh47ad6842006-11-08 12:25:42 +00001839 rc = 2;
drh75897232000-05-29 14:26:00 +00001840 }else
1841
shanehe2aa9d72009-11-06 17:20:17 +00001842 if( c=='e' && strncmp(azArg[0], "explain", n)==0 && nArg<3 ){
drhc28490c2006-10-26 14:25:58 +00001843 int val = nArg>=2 ? booleanValue(azArg[1]) : 1;
persicom7e2dfdd2002-04-18 02:46:52 +00001844 if(val == 1) {
1845 if(!p->explainPrev.valid) {
1846 p->explainPrev.valid = 1;
1847 p->explainPrev.mode = p->mode;
1848 p->explainPrev.showHeader = p->showHeader;
1849 memcpy(p->explainPrev.colWidth,p->colWidth,sizeof(p->colWidth));
1850 }
1851 /* We could put this code under the !p->explainValid
1852 ** condition so that it does not execute if we are already in
1853 ** explain mode. However, always executing it allows us an easy
1854 ** was to reset to explain mode in case the user previously
1855 ** did an .explain followed by a .width, .mode or .header
1856 ** command.
1857 */
danielk19770d78bae2008-01-03 07:09:48 +00001858 p->mode = MODE_Explain;
persicom7e2dfdd2002-04-18 02:46:52 +00001859 p->showHeader = 1;
1860 memset(p->colWidth,0,ArraySize(p->colWidth));
danielk19770d78bae2008-01-03 07:09:48 +00001861 p->colWidth[0] = 4; /* addr */
drh60a713c2008-01-21 16:22:45 +00001862 p->colWidth[1] = 13; /* opcode */
1863 p->colWidth[2] = 4; /* P1 */
1864 p->colWidth[3] = 4; /* P2 */
1865 p->colWidth[4] = 4; /* P3 */
1866 p->colWidth[5] = 13; /* P4 */
danielk19770d78bae2008-01-03 07:09:48 +00001867 p->colWidth[6] = 2; /* P5 */
drh60a713c2008-01-21 16:22:45 +00001868 p->colWidth[7] = 13; /* Comment */
persicom7e2dfdd2002-04-18 02:46:52 +00001869 }else if (p->explainPrev.valid) {
1870 p->explainPrev.valid = 0;
1871 p->mode = p->explainPrev.mode;
1872 p->showHeader = p->explainPrev.showHeader;
1873 memcpy(p->colWidth,p->explainPrev.colWidth,sizeof(p->colWidth));
1874 }
drh75897232000-05-29 14:26:00 +00001875 }else
1876
drhc28490c2006-10-26 14:25:58 +00001877 if( c=='h' && (strncmp(azArg[0], "header", n)==0 ||
shanehe2aa9d72009-11-06 17:20:17 +00001878 strncmp(azArg[0], "headers", n)==0) && nArg>1 && nArg<3 ){
drhc28490c2006-10-26 14:25:58 +00001879 p->showHeader = booleanValue(azArg[1]);
drh75897232000-05-29 14:26:00 +00001880 }else
1881
1882 if( c=='h' && strncmp(azArg[0], "help", n)==0 ){
drha81c64a2009-01-14 23:38:02 +00001883 fprintf(stderr,"%s",zHelp);
shaneb320ccd2009-10-21 03:42:58 +00001884 if( HAS_TIMER ){
1885 fprintf(stderr,"%s",zTimerHelp);
1886 }
drh75897232000-05-29 14:26:00 +00001887 }else
1888
shanehe2aa9d72009-11-06 17:20:17 +00001889 if( c=='i' && strncmp(azArg[0], "import", n)==0 && nArg==3 ){
drhfeac5f82004-08-01 00:10:45 +00001890 char *zTable = azArg[2]; /* Insert data into this table */
1891 char *zFile = azArg[1]; /* The file from which to extract data */
shane916f9612009-10-23 00:37:15 +00001892 sqlite3_stmt *pStmt = NULL; /* A statement */
drhfeac5f82004-08-01 00:10:45 +00001893 int nCol; /* Number of columns in the table */
1894 int nByte; /* Number of bytes in an SQL string */
1895 int i, j; /* Loop counters */
1896 int nSep; /* Number of bytes in p->separator[] */
1897 char *zSql; /* An SQL statement */
1898 char *zLine; /* A single line of input from the file */
1899 char **azCol; /* zLine[] broken up into columns */
1900 char *zCommit; /* How to commit changes */
drhb860bc92004-08-04 15:16:55 +00001901 FILE *in; /* The input file */
1902 int lineno = 0; /* Line number of input file */
drhfeac5f82004-08-01 00:10:45 +00001903
drha543c822006-06-08 16:10:14 +00001904 open_db(p);
drh4f21c4a2008-12-10 22:15:00 +00001905 nSep = strlen30(p->separator);
drhfeac5f82004-08-01 00:10:45 +00001906 if( nSep==0 ){
shane916f9612009-10-23 00:37:15 +00001907 fprintf(stderr, "Error: non-null separator required for import\n");
1908 return 1;
drhfeac5f82004-08-01 00:10:45 +00001909 }
drh7b075e32011-09-28 01:10:00 +00001910 zSql = sqlite3_mprintf("SELECT * FROM %s", zTable);
shane916f9612009-10-23 00:37:15 +00001911 if( zSql==0 ){
1912 fprintf(stderr, "Error: out of memory\n");
1913 return 1;
1914 }
drh4f21c4a2008-12-10 22:15:00 +00001915 nByte = strlen30(zSql);
drh5e6078b2006-01-31 19:07:22 +00001916 rc = sqlite3_prepare(p->db, zSql, -1, &pStmt, 0);
drhfeac5f82004-08-01 00:10:45 +00001917 sqlite3_free(zSql);
1918 if( rc ){
shane916f9612009-10-23 00:37:15 +00001919 if (pStmt) sqlite3_finalize(pStmt);
drhfeac5f82004-08-01 00:10:45 +00001920 fprintf(stderr,"Error: %s\n", sqlite3_errmsg(db));
shane916f9612009-10-23 00:37:15 +00001921 return 1;
drhfeac5f82004-08-01 00:10:45 +00001922 }
shane916f9612009-10-23 00:37:15 +00001923 nCol = sqlite3_column_count(pStmt);
drhfeac5f82004-08-01 00:10:45 +00001924 sqlite3_finalize(pStmt);
shane916f9612009-10-23 00:37:15 +00001925 pStmt = 0;
shane9bd1b442009-10-23 01:27:39 +00001926 if( nCol==0 ) return 0; /* no columns, no error */
drhfeac5f82004-08-01 00:10:45 +00001927 zSql = malloc( nByte + 20 + nCol*2 );
shane916f9612009-10-23 00:37:15 +00001928 if( zSql==0 ){
1929 fprintf(stderr, "Error: out of memory\n");
1930 return 1;
1931 }
drh7b075e32011-09-28 01:10:00 +00001932 sqlite3_snprintf(nByte+20, zSql, "INSERT INTO %s VALUES(?", zTable);
drh4f21c4a2008-12-10 22:15:00 +00001933 j = strlen30(zSql);
drhfeac5f82004-08-01 00:10:45 +00001934 for(i=1; i<nCol; i++){
1935 zSql[j++] = ',';
1936 zSql[j++] = '?';
1937 }
1938 zSql[j++] = ')';
1939 zSql[j] = 0;
drh5e6078b2006-01-31 19:07:22 +00001940 rc = sqlite3_prepare(p->db, zSql, -1, &pStmt, 0);
drhfeac5f82004-08-01 00:10:45 +00001941 free(zSql);
1942 if( rc ){
1943 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(db));
shane916f9612009-10-23 00:37:15 +00001944 if (pStmt) sqlite3_finalize(pStmt);
drh47ad6842006-11-08 12:25:42 +00001945 return 1;
drhfeac5f82004-08-01 00:10:45 +00001946 }
1947 in = fopen(zFile, "rb");
1948 if( in==0 ){
shane9bd1b442009-10-23 01:27:39 +00001949 fprintf(stderr, "Error: cannot open \"%s\"\n", zFile);
drhfeac5f82004-08-01 00:10:45 +00001950 sqlite3_finalize(pStmt);
shane916f9612009-10-23 00:37:15 +00001951 return 1;
drhfeac5f82004-08-01 00:10:45 +00001952 }
1953 azCol = malloc( sizeof(azCol[0])*(nCol+1) );
drh43617e92006-03-06 20:55:46 +00001954 if( azCol==0 ){
shane916f9612009-10-23 00:37:15 +00001955 fprintf(stderr, "Error: out of memory\n");
drh43617e92006-03-06 20:55:46 +00001956 fclose(in);
shane916f9612009-10-23 00:37:15 +00001957 sqlite3_finalize(pStmt);
1958 return 1;
drh43617e92006-03-06 20:55:46 +00001959 }
drhfeac5f82004-08-01 00:10:45 +00001960 sqlite3_exec(p->db, "BEGIN", 0, 0, 0);
1961 zCommit = "COMMIT";
drh18f52e02012-01-16 16:56:31 +00001962 while( (zLine = local_getline(0, in, 1))!=0 ){
1963 char *z, c;
1964 int inQuote = 0;
drhb860bc92004-08-04 15:16:55 +00001965 lineno++;
drhfeac5f82004-08-01 00:10:45 +00001966 azCol[0] = zLine;
drh18f52e02012-01-16 16:56:31 +00001967 for(i=0, z=zLine; (c = *z)!=0; z++){
1968 if( c=='"' ) inQuote = !inQuote;
1969 if( c=='\n' ) lineno++;
1970 if( !inQuote && c==p->separator[0] && strncmp(z,p->separator,nSep)==0 ){
drhfeac5f82004-08-01 00:10:45 +00001971 *z = 0;
1972 i++;
drhb860bc92004-08-04 15:16:55 +00001973 if( i<nCol ){
1974 azCol[i] = &z[nSep];
1975 z += nSep-1;
1976 }
drhfeac5f82004-08-01 00:10:45 +00001977 }
shane916f9612009-10-23 00:37:15 +00001978 } /* end for */
drh1cd7f832005-08-05 18:50:51 +00001979 *z = 0;
drhb860bc92004-08-04 15:16:55 +00001980 if( i+1!=nCol ){
shane916f9612009-10-23 00:37:15 +00001981 fprintf(stderr,
1982 "Error: %s line %d: expected %d columns of data but found %d\n",
1983 zFile, lineno, nCol, i+1);
drhb860bc92004-08-04 15:16:55 +00001984 zCommit = "ROLLBACK";
drh1822eee2008-12-04 12:26:00 +00001985 free(zLine);
shane916f9612009-10-23 00:37:15 +00001986 rc = 1;
1987 break; /* from while */
drhb860bc92004-08-04 15:16:55 +00001988 }
drhfeac5f82004-08-01 00:10:45 +00001989 for(i=0; i<nCol; i++){
drh18f52e02012-01-16 16:56:31 +00001990 if( azCol[i][0]=='"' ){
1991 int k;
1992 for(z=azCol[i], j=1, k=0; z[j]; j++){
1993 if( z[j]=='"' ){ j++; if( z[j]==0 ) break; }
1994 z[k++] = z[j];
1995 }
1996 z[k] = 0;
1997 }
drhfeac5f82004-08-01 00:10:45 +00001998 sqlite3_bind_text(pStmt, i+1, azCol[i], -1, SQLITE_STATIC);
1999 }
2000 sqlite3_step(pStmt);
2001 rc = sqlite3_reset(pStmt);
2002 free(zLine);
2003 if( rc!=SQLITE_OK ){
2004 fprintf(stderr,"Error: %s\n", sqlite3_errmsg(db));
2005 zCommit = "ROLLBACK";
drh47ad6842006-11-08 12:25:42 +00002006 rc = 1;
shane916f9612009-10-23 00:37:15 +00002007 break; /* from while */
drhfeac5f82004-08-01 00:10:45 +00002008 }
shane916f9612009-10-23 00:37:15 +00002009 } /* end while */
drhfeac5f82004-08-01 00:10:45 +00002010 free(azCol);
2011 fclose(in);
2012 sqlite3_finalize(pStmt);
drhb860bc92004-08-04 15:16:55 +00002013 sqlite3_exec(p->db, zCommit, 0, 0, 0);
drhfeac5f82004-08-01 00:10:45 +00002014 }else
2015
shanehe2aa9d72009-11-06 17:20:17 +00002016 if( c=='i' && strncmp(azArg[0], "indices", n)==0 && nArg<3 ){
drh75897232000-05-29 14:26:00 +00002017 struct callback_data data;
2018 char *zErrMsg = 0;
drh44c2eb12003-04-30 11:38:26 +00002019 open_db(p);
drh75897232000-05-29 14:26:00 +00002020 memcpy(&data, p, sizeof(data));
2021 data.showHeader = 0;
2022 data.mode = MODE_List;
shane86f5bdb2009-10-24 02:00:07 +00002023 if( nArg==1 ){
2024 rc = sqlite3_exec(p->db,
2025 "SELECT name FROM sqlite_master "
2026 "WHERE type='index' AND name NOT LIKE 'sqlite_%' "
2027 "UNION ALL "
2028 "SELECT name FROM sqlite_temp_master "
2029 "WHERE type='index' "
2030 "ORDER BY 1",
2031 callback, &data, &zErrMsg
2032 );
2033 }else{
2034 zShellStatic = azArg[1];
2035 rc = sqlite3_exec(p->db,
2036 "SELECT name FROM sqlite_master "
2037 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2038 "UNION ALL "
2039 "SELECT name FROM sqlite_temp_master "
2040 "WHERE type='index' AND tbl_name LIKE shellstatic() "
2041 "ORDER BY 1",
2042 callback, &data, &zErrMsg
2043 );
2044 zShellStatic = 0;
2045 }
drh75897232000-05-29 14:26:00 +00002046 if( zErrMsg ){
2047 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002048 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002049 rc = 1;
shane86f5bdb2009-10-24 02:00:07 +00002050 }else if( rc != SQLITE_OK ){
2051 fprintf(stderr,"Error: querying sqlite_master and sqlite_temp_master\n");
2052 rc = 1;
drh75897232000-05-29 14:26:00 +00002053 }
2054 }else
2055
drhae5e4452007-05-03 17:18:36 +00002056#ifdef SQLITE_ENABLE_IOTRACE
drhb0603412007-02-28 04:47:26 +00002057 if( c=='i' && strncmp(azArg[0], "iotrace", n)==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002058 extern void (*sqlite3IoTrace)(const char*, ...);
drhb0603412007-02-28 04:47:26 +00002059 if( iotrace && iotrace!=stdout ) fclose(iotrace);
2060 iotrace = 0;
2061 if( nArg<2 ){
mlcreech3a00f902008-03-04 17:45:01 +00002062 sqlite3IoTrace = 0;
drhb0603412007-02-28 04:47:26 +00002063 }else if( strcmp(azArg[1], "-")==0 ){
mlcreech3a00f902008-03-04 17:45:01 +00002064 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002065 iotrace = stdout;
2066 }else{
2067 iotrace = fopen(azArg[1], "w");
2068 if( iotrace==0 ){
shane9bd1b442009-10-23 01:27:39 +00002069 fprintf(stderr, "Error: cannot open \"%s\"\n", azArg[1]);
mlcreech3a00f902008-03-04 17:45:01 +00002070 sqlite3IoTrace = 0;
shane9bd1b442009-10-23 01:27:39 +00002071 rc = 1;
drhb0603412007-02-28 04:47:26 +00002072 }else{
mlcreech3a00f902008-03-04 17:45:01 +00002073 sqlite3IoTrace = iotracePrintf;
drhb0603412007-02-28 04:47:26 +00002074 }
2075 }
2076 }else
drhae5e4452007-05-03 17:18:36 +00002077#endif
drhb0603412007-02-28 04:47:26 +00002078
drh70df4fe2006-06-13 15:12:21 +00002079#ifndef SQLITE_OMIT_LOAD_EXTENSION
drh1e397f82006-06-08 15:28:43 +00002080 if( c=='l' && strncmp(azArg[0], "load", n)==0 && nArg>=2 ){
2081 const char *zFile, *zProc;
2082 char *zErrMsg = 0;
drh1e397f82006-06-08 15:28:43 +00002083 zFile = azArg[1];
2084 zProc = nArg>=3 ? azArg[2] : 0;
2085 open_db(p);
2086 rc = sqlite3_load_extension(p->db, zFile, zProc, &zErrMsg);
2087 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002088 fprintf(stderr, "Error: %s\n", zErrMsg);
drh1e397f82006-06-08 15:28:43 +00002089 sqlite3_free(zErrMsg);
drh47ad6842006-11-08 12:25:42 +00002090 rc = 1;
drh1e397f82006-06-08 15:28:43 +00002091 }
2092 }else
drh70df4fe2006-06-13 15:12:21 +00002093#endif
drh1e397f82006-06-08 15:28:43 +00002094
drhc8ba2122011-03-23 11:16:22 +00002095 if( c=='l' && strncmp(azArg[0], "log", n)==0 && nArg>=2 ){
drh127f9d72010-02-23 01:47:00 +00002096 const char *zFile = azArg[1];
drh42f64e52012-04-04 16:56:23 +00002097 output_file_close(p->pLog);
2098 p->pLog = output_file_open(zFile);
drh127f9d72010-02-23 01:47:00 +00002099 }else
2100
shanehe2aa9d72009-11-06 17:20:17 +00002101 if( c=='m' && strncmp(azArg[0], "mode", n)==0 && nArg==2 ){
drh4f21c4a2008-12-10 22:15:00 +00002102 int n2 = strlen30(azArg[1]);
shanehe2aa9d72009-11-06 17:20:17 +00002103 if( (n2==4 && strncmp(azArg[1],"line",n2)==0)
persicom7e2dfdd2002-04-18 02:46:52 +00002104 ||
shanehe2aa9d72009-11-06 17:20:17 +00002105 (n2==5 && strncmp(azArg[1],"lines",n2)==0) ){
drh75897232000-05-29 14:26:00 +00002106 p->mode = MODE_Line;
shanehe2aa9d72009-11-06 17:20:17 +00002107 }else if( (n2==6 && strncmp(azArg[1],"column",n2)==0)
persicom7e2dfdd2002-04-18 02:46:52 +00002108 ||
shanehe2aa9d72009-11-06 17:20:17 +00002109 (n2==7 && strncmp(azArg[1],"columns",n2)==0) ){
drh75897232000-05-29 14:26:00 +00002110 p->mode = MODE_Column;
shanehe2aa9d72009-11-06 17:20:17 +00002111 }else if( n2==4 && strncmp(azArg[1],"list",n2)==0 ){
drh75897232000-05-29 14:26:00 +00002112 p->mode = MODE_List;
shanehe2aa9d72009-11-06 17:20:17 +00002113 }else if( n2==4 && strncmp(azArg[1],"html",n2)==0 ){
drh1e5d0e92000-05-31 23:33:17 +00002114 p->mode = MODE_Html;
shanehe2aa9d72009-11-06 17:20:17 +00002115 }else if( n2==3 && strncmp(azArg[1],"tcl",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002116 p->mode = MODE_Tcl;
mistachkin585dcb22012-12-04 00:23:43 +00002117 sqlite3_snprintf(sizeof(p->separator), p->separator, " ");
shanehe2aa9d72009-11-06 17:20:17 +00002118 }else if( n2==3 && strncmp(azArg[1],"csv",n2)==0 ){
drh8e64d1c2004-10-07 00:32:39 +00002119 p->mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00002120 sqlite3_snprintf(sizeof(p->separator), p->separator, ",");
shanehe2aa9d72009-11-06 17:20:17 +00002121 }else if( n2==4 && strncmp(azArg[1],"tabs",n2)==0 ){
drhfeac5f82004-08-01 00:10:45 +00002122 p->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00002123 sqlite3_snprintf(sizeof(p->separator), p->separator, "\t");
shanehe2aa9d72009-11-06 17:20:17 +00002124 }else if( n2==6 && strncmp(azArg[1],"insert",n2)==0 ){
drh28bd4bc2000-06-15 15:57:22 +00002125 p->mode = MODE_Insert;
shanehe2aa9d72009-11-06 17:20:17 +00002126 set_table_name(p, "table");
drhdaffd0e2001-04-11 14:28:42 +00002127 }else {
shane9bd1b442009-10-23 01:27:39 +00002128 fprintf(stderr,"Error: mode should be one of: "
drhfeac5f82004-08-01 00:10:45 +00002129 "column csv html insert line list tabs tcl\n");
shane9bd1b442009-10-23 01:27:39 +00002130 rc = 1;
drh75897232000-05-29 14:26:00 +00002131 }
2132 }else
2133
shanehe2aa9d72009-11-06 17:20:17 +00002134 if( c=='m' && strncmp(azArg[0], "mode", n)==0 && nArg==3 ){
2135 int n2 = strlen30(azArg[1]);
2136 if( n2==6 && strncmp(azArg[1],"insert",n2)==0 ){
2137 p->mode = MODE_Insert;
2138 set_table_name(p, azArg[2]);
2139 }else {
2140 fprintf(stderr, "Error: invalid arguments: "
2141 " \"%s\". Enter \".help\" for help\n", azArg[2]);
2142 rc = 1;
2143 }
2144 }else
2145
persicom7e2dfdd2002-04-18 02:46:52 +00002146 if( c=='n' && strncmp(azArg[0], "nullvalue", n)==0 && nArg==2 ) {
drh5bb3eb92007-05-04 13:15:55 +00002147 sqlite3_snprintf(sizeof(p->nullvalue), p->nullvalue,
2148 "%.*s", (int)ArraySize(p->nullvalue)-1, azArg[1]);
persicom7e2dfdd2002-04-18 02:46:52 +00002149 }else
2150
drh75897232000-05-29 14:26:00 +00002151 if( c=='o' && strncmp(azArg[0], "output", n)==0 && nArg==2 ){
drh42f64e52012-04-04 16:56:23 +00002152 if( p->outfile[0]=='|' ){
2153 pclose(p->out);
2154 }else{
2155 output_file_close(p->out);
drh75897232000-05-29 14:26:00 +00002156 }
drh42f64e52012-04-04 16:56:23 +00002157 p->outfile[0] = 0;
2158 if( azArg[1][0]=='|' ){
drhe1da8fa2012-03-30 00:05:57 +00002159 p->out = popen(&azArg[1][1], "w");
2160 if( p->out==0 ){
2161 fprintf(stderr,"Error: cannot open pipe \"%s\"\n", &azArg[1][1]);
2162 p->out = stdout;
2163 rc = 1;
2164 }else{
2165 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", azArg[1]);
2166 }
drh75897232000-05-29 14:26:00 +00002167 }else{
drh42f64e52012-04-04 16:56:23 +00002168 p->out = output_file_open(azArg[1]);
drh75897232000-05-29 14:26:00 +00002169 if( p->out==0 ){
drh42f64e52012-04-04 16:56:23 +00002170 if( strcmp(azArg[1],"off")!=0 ){
2171 fprintf(stderr,"Error: cannot write to \"%s\"\n", azArg[1]);
2172 }
drh75897232000-05-29 14:26:00 +00002173 p->out = stdout;
shane9bd1b442009-10-23 01:27:39 +00002174 rc = 1;
persicom7e2dfdd2002-04-18 02:46:52 +00002175 } else {
drh42f64e52012-04-04 16:56:23 +00002176 sqlite3_snprintf(sizeof(p->outfile), p->outfile, "%s", azArg[1]);
drh75897232000-05-29 14:26:00 +00002177 }
2178 }
2179 }else
2180
drh078b1fd2012-09-21 13:40:02 +00002181 if( c=='p' && n>=3 && strncmp(azArg[0], "print", n)==0 ){
2182 int i;
2183 for(i=1; i<nArg; i++){
2184 if( i>1 ) fprintf(p->out, " ");
2185 fprintf(p->out, "%s", azArg[i]);
2186 }
2187 fprintf(p->out, "\n");
2188 }else
2189
drhdd45df82002-04-18 12:39:03 +00002190 if( c=='p' && strncmp(azArg[0], "prompt", n)==0 && (nArg==2 || nArg==3)){
persicom7e2dfdd2002-04-18 02:46:52 +00002191 if( nArg >= 2) {
2192 strncpy(mainPrompt,azArg[1],(int)ArraySize(mainPrompt)-1);
2193 }
2194 if( nArg >= 3) {
2195 strncpy(continuePrompt,azArg[2],(int)ArraySize(continuePrompt)-1);
2196 }
2197 }else
2198
shanehe2aa9d72009-11-06 17:20:17 +00002199 if( c=='q' && strncmp(azArg[0], "quit", n)==0 && nArg==1 ){
drh47ad6842006-11-08 12:25:42 +00002200 rc = 2;
persicom7e2dfdd2002-04-18 02:46:52 +00002201 }else
2202
drh9ff849f2009-02-04 20:55:57 +00002203 if( c=='r' && n>=3 && strncmp(azArg[0], "read", n)==0 && nArg==2 ){
drha1f9b5e2004-02-14 16:31:02 +00002204 FILE *alt = fopen(azArg[1], "rb");
drhdaffd0e2001-04-11 14:28:42 +00002205 if( alt==0 ){
shane9bd1b442009-10-23 01:27:39 +00002206 fprintf(stderr,"Error: cannot open \"%s\"\n", azArg[1]);
2207 rc = 1;
drhdaffd0e2001-04-11 14:28:42 +00002208 }else{
shane9bd1b442009-10-23 01:27:39 +00002209 rc = process_input(p, alt);
drhdaffd0e2001-04-11 14:28:42 +00002210 fclose(alt);
2211 }
2212 }else
2213
shanehe2aa9d72009-11-06 17:20:17 +00002214 if( c=='r' && n>=3 && strncmp(azArg[0], "restore", n)==0 && nArg>1 && nArg<4){
drh9ff849f2009-02-04 20:55:57 +00002215 const char *zSrcFile;
2216 const char *zDb;
2217 sqlite3 *pSrc;
2218 sqlite3_backup *pBackup;
drhdc2c4912009-02-04 22:46:47 +00002219 int nTimeout = 0;
2220
drh9ff849f2009-02-04 20:55:57 +00002221 if( nArg==2 ){
2222 zSrcFile = azArg[1];
2223 zDb = "main";
2224 }else{
2225 zSrcFile = azArg[2];
2226 zDb = azArg[1];
2227 }
2228 rc = sqlite3_open(zSrcFile, &pSrc);
2229 if( rc!=SQLITE_OK ){
shane9bd1b442009-10-23 01:27:39 +00002230 fprintf(stderr, "Error: cannot open \"%s\"\n", zSrcFile);
drh9ff849f2009-02-04 20:55:57 +00002231 sqlite3_close(pSrc);
2232 return 1;
2233 }
drhdc2c4912009-02-04 22:46:47 +00002234 open_db(p);
drh9ff849f2009-02-04 20:55:57 +00002235 pBackup = sqlite3_backup_init(p->db, zDb, pSrc, "main");
2236 if( pBackup==0 ){
2237 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
2238 sqlite3_close(pSrc);
2239 return 1;
2240 }
drhdc2c4912009-02-04 22:46:47 +00002241 while( (rc = sqlite3_backup_step(pBackup,100))==SQLITE_OK
2242 || rc==SQLITE_BUSY ){
2243 if( rc==SQLITE_BUSY ){
2244 if( nTimeout++ >= 3 ) break;
2245 sqlite3_sleep(100);
drh9ff849f2009-02-04 20:55:57 +00002246 }
2247 }
2248 sqlite3_backup_finish(pBackup);
2249 if( rc==SQLITE_DONE ){
shane9bd1b442009-10-23 01:27:39 +00002250 rc = 0;
drhdc2c4912009-02-04 22:46:47 +00002251 }else if( rc==SQLITE_BUSY || rc==SQLITE_LOCKED ){
shane9bd1b442009-10-23 01:27:39 +00002252 fprintf(stderr, "Error: source database is busy\n");
2253 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002254 }else{
2255 fprintf(stderr, "Error: %s\n", sqlite3_errmsg(p->db));
shane9bd1b442009-10-23 01:27:39 +00002256 rc = 1;
drh9ff849f2009-02-04 20:55:57 +00002257 }
2258 sqlite3_close(pSrc);
2259 }else
2260
shanehe2aa9d72009-11-06 17:20:17 +00002261 if( c=='s' && strncmp(azArg[0], "schema", n)==0 && nArg<3 ){
drh75897232000-05-29 14:26:00 +00002262 struct callback_data data;
2263 char *zErrMsg = 0;
drh44c2eb12003-04-30 11:38:26 +00002264 open_db(p);
drh75897232000-05-29 14:26:00 +00002265 memcpy(&data, p, sizeof(data));
2266 data.showHeader = 0;
drhe3710332000-09-29 13:30:53 +00002267 data.mode = MODE_Semi;
drh75897232000-05-29 14:26:00 +00002268 if( nArg>1 ){
drhc8d74412004-08-31 23:41:26 +00002269 int i;
drhf0693c82011-10-11 20:41:54 +00002270 for(i=0; azArg[1][i]; i++) azArg[1][i] = ToLower(azArg[1][i]);
drhc8d74412004-08-31 23:41:26 +00002271 if( strcmp(azArg[1],"sqlite_master")==0 ){
drha18c5682000-10-08 22:20:57 +00002272 char *new_argv[2], *new_colv[2];
2273 new_argv[0] = "CREATE TABLE sqlite_master (\n"
2274 " type text,\n"
2275 " name text,\n"
2276 " tbl_name text,\n"
drhadbca9c2001-09-27 15:11:53 +00002277 " rootpage integer,\n"
drha18c5682000-10-08 22:20:57 +00002278 " sql text\n"
2279 ")";
2280 new_argv[1] = 0;
2281 new_colv[0] = "sql";
2282 new_colv[1] = 0;
2283 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002284 rc = SQLITE_OK;
drhc8d74412004-08-31 23:41:26 +00002285 }else if( strcmp(azArg[1],"sqlite_temp_master")==0 ){
drhe0bc4042002-06-25 01:09:11 +00002286 char *new_argv[2], *new_colv[2];
2287 new_argv[0] = "CREATE TEMP TABLE sqlite_temp_master (\n"
2288 " type text,\n"
2289 " name text,\n"
2290 " tbl_name text,\n"
2291 " rootpage integer,\n"
2292 " sql text\n"
2293 ")";
2294 new_argv[1] = 0;
2295 new_colv[0] = "sql";
2296 new_colv[1] = 0;
2297 callback(&data, 1, new_argv, new_colv);
shane9bd1b442009-10-23 01:27:39 +00002298 rc = SQLITE_OK;
drha18c5682000-10-08 22:20:57 +00002299 }else{
danielk1977bc6ada42004-06-30 08:20:16 +00002300 zShellStatic = azArg[1];
shane9bd1b442009-10-23 01:27:39 +00002301 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002302 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002303 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002304 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002305 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh6ac7a582011-11-04 00:35:56 +00002306 "WHERE lower(tbl_name) LIKE shellstatic()"
2307 " AND type!='meta' AND sql NOTNULL "
drh1ba00292013-05-06 21:01:06 +00002308 "ORDER BY rowid",
danielk1977bc6ada42004-06-30 08:20:16 +00002309 callback, &data, &zErrMsg);
2310 zShellStatic = 0;
drha18c5682000-10-08 22:20:57 +00002311 }
drh75897232000-05-29 14:26:00 +00002312 }else{
shane9bd1b442009-10-23 01:27:39 +00002313 rc = sqlite3_exec(p->db,
drhe0bc4042002-06-25 01:09:11 +00002314 "SELECT sql FROM "
drhac43e982012-05-21 03:15:06 +00002315 " (SELECT sql sql, type type, tbl_name tbl_name, name name, rowid x"
drh8f800a72009-01-14 23:17:55 +00002316 " FROM sqlite_master UNION ALL"
drhac43e982012-05-21 03:15:06 +00002317 " SELECT sql, type, tbl_name, name, rowid FROM sqlite_temp_master) "
drh0c356672005-09-10 22:40:53 +00002318 "WHERE type!='meta' AND sql NOTNULL AND name NOT LIKE 'sqlite_%'"
drh1ba00292013-05-06 21:01:06 +00002319 "ORDER BY rowid",
drha18c5682000-10-08 22:20:57 +00002320 callback, &data, &zErrMsg
2321 );
drh75897232000-05-29 14:26:00 +00002322 }
drh75897232000-05-29 14:26:00 +00002323 if( zErrMsg ){
2324 fprintf(stderr,"Error: %s\n", zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002325 sqlite3_free(zErrMsg);
shane9bd1b442009-10-23 01:27:39 +00002326 rc = 1;
2327 }else if( rc != SQLITE_OK ){
2328 fprintf(stderr,"Error: querying schema information\n");
2329 rc = 1;
2330 }else{
2331 rc = 0;
drh75897232000-05-29 14:26:00 +00002332 }
2333 }else
2334
drh348d19c2013-06-03 12:47:43 +00002335 /* Undocumented commands for internal testing. Subject to change
2336 ** without notice. */
2337 if( c=='s' && n>=10 && strncmp(azArg[0], "selftest-", 9)==0 ){
2338 if( strncmp(azArg[0]+9, "boolean", n-9)==0 ){
2339 int i, v;
2340 for(i=1; i<nArg; i++){
2341 v = booleanValue(azArg[i]);
2342 fprintf(p->out, "%s: %d 0x%x\n", azArg[i], v, v);
2343 }
2344 }
2345 if( strncmp(azArg[0]+9, "integer", n-9)==0 ){
2346 int i; sqlite3_int64 v;
2347 for(i=1; i<nArg; i++){
2348 v = integerValue(azArg[i]);
2349 fprintf(p->out, "%s: %lld 0x%llx\n", azArg[i], v, v);
2350 }
2351 }
2352 }else
2353
drh75897232000-05-29 14:26:00 +00002354 if( c=='s' && strncmp(azArg[0], "separator", n)==0 && nArg==2 ){
drh5bb3eb92007-05-04 13:15:55 +00002355 sqlite3_snprintf(sizeof(p->separator), p->separator,
2356 "%.*s", (int)sizeof(p->separator)-1, azArg[1]);
drh75897232000-05-29 14:26:00 +00002357 }else
2358
shanehe2aa9d72009-11-06 17:20:17 +00002359 if( c=='s' && strncmp(azArg[0], "show", n)==0 && nArg==1 ){
persicom7e2dfdd2002-04-18 02:46:52 +00002360 int i;
2361 fprintf(p->out,"%9.9s: %s\n","echo", p->echoOn ? "on" : "off");
drh67505e72002-04-19 12:34:06 +00002362 fprintf(p->out,"%9.9s: %s\n","explain", p->explainPrev.valid ? "on" :"off");
drhdd45df82002-04-18 12:39:03 +00002363 fprintf(p->out,"%9.9s: %s\n","headers", p->showHeader ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00002364 fprintf(p->out,"%9.9s: %s\n","mode", modeDescr[p->mode]);
drhfeac5f82004-08-01 00:10:45 +00002365 fprintf(p->out,"%9.9s: ", "nullvalue");
2366 output_c_string(p->out, p->nullvalue);
2367 fprintf(p->out, "\n");
drh67505e72002-04-19 12:34:06 +00002368 fprintf(p->out,"%9.9s: %s\n","output",
drh4f21c4a2008-12-10 22:15:00 +00002369 strlen30(p->outfile) ? p->outfile : "stdout");
drhfeac5f82004-08-01 00:10:45 +00002370 fprintf(p->out,"%9.9s: ", "separator");
2371 output_c_string(p->out, p->separator);
2372 fprintf(p->out, "\n");
shaneh642d8b82010-07-28 16:05:34 +00002373 fprintf(p->out,"%9.9s: %s\n","stats", p->statsOn ? "on" : "off");
persicom7e2dfdd2002-04-18 02:46:52 +00002374 fprintf(p->out,"%9.9s: ","width");
2375 for (i=0;i<(int)ArraySize(p->colWidth) && p->colWidth[i] != 0;i++) {
drhfeac5f82004-08-01 00:10:45 +00002376 fprintf(p->out,"%d ",p->colWidth[i]);
persicom7e2dfdd2002-04-18 02:46:52 +00002377 }
drhfeac5f82004-08-01 00:10:45 +00002378 fprintf(p->out,"\n");
persicom7e2dfdd2002-04-18 02:46:52 +00002379 }else
2380
shaneh642d8b82010-07-28 16:05:34 +00002381 if( c=='s' && strncmp(azArg[0], "stats", n)==0 && nArg>1 && nArg<3 ){
2382 p->statsOn = booleanValue(azArg[1]);
2383 }else
2384
shanehe2aa9d72009-11-06 17:20:17 +00002385 if( c=='t' && n>1 && strncmp(azArg[0], "tables", n)==0 && nArg<3 ){
drh98781232012-04-23 12:38:05 +00002386 sqlite3_stmt *pStmt;
drhe3710332000-09-29 13:30:53 +00002387 char **azResult;
drh98781232012-04-23 12:38:05 +00002388 int nRow, nAlloc;
2389 char *zSql = 0;
2390 int ii;
drh44c2eb12003-04-30 11:38:26 +00002391 open_db(p);
drh98781232012-04-23 12:38:05 +00002392 rc = sqlite3_prepare_v2(p->db, "PRAGMA database_list", -1, &pStmt, 0);
2393 if( rc ) return rc;
2394 zSql = sqlite3_mprintf(
2395 "SELECT name FROM sqlite_master"
2396 " WHERE type IN ('table','view')"
2397 " AND name NOT LIKE 'sqlite_%%'"
2398 " AND name LIKE ?1");
2399 while( sqlite3_step(pStmt)==SQLITE_ROW ){
2400 const char *zDbName = (const char*)sqlite3_column_text(pStmt, 1);
2401 if( zDbName==0 || strcmp(zDbName,"main")==0 ) continue;
2402 if( strcmp(zDbName,"temp")==0 ){
2403 zSql = sqlite3_mprintf(
2404 "%z UNION ALL "
2405 "SELECT 'temp.' || name FROM sqlite_temp_master"
2406 " WHERE type IN ('table','view')"
2407 " AND name NOT LIKE 'sqlite_%%'"
2408 " AND name LIKE ?1", zSql);
2409 }else{
2410 zSql = sqlite3_mprintf(
2411 "%z UNION ALL "
2412 "SELECT '%q.' || name FROM \"%w\".sqlite_master"
2413 " WHERE type IN ('table','view')"
2414 " AND name NOT LIKE 'sqlite_%%'"
2415 " AND name LIKE ?1", zSql, zDbName, zDbName);
2416 }
drha50da102000-08-08 20:19:09 +00002417 }
drh98781232012-04-23 12:38:05 +00002418 sqlite3_finalize(pStmt);
2419 zSql = sqlite3_mprintf("%z ORDER BY 1", zSql);
2420 rc = sqlite3_prepare_v2(p->db, zSql, -1, &pStmt, 0);
2421 sqlite3_free(zSql);
2422 if( rc ) return rc;
2423 nRow = nAlloc = 0;
2424 azResult = 0;
2425 if( nArg>1 ){
2426 sqlite3_bind_text(pStmt, 1, azArg[1], -1, SQLITE_TRANSIENT);
shane9bd1b442009-10-23 01:27:39 +00002427 }else{
drh98781232012-04-23 12:38:05 +00002428 sqlite3_bind_text(pStmt, 1, "%", -1, SQLITE_STATIC);
2429 }
2430 while( sqlite3_step(pStmt)==SQLITE_ROW ){
2431 if( nRow>=nAlloc ){
2432 char **azNew;
2433 int n = nAlloc*2 + 10;
2434 azNew = sqlite3_realloc(azResult, sizeof(azResult[0])*n);
2435 if( azNew==0 ){
2436 fprintf(stderr, "Error: out of memory\n");
2437 break;
2438 }
2439 nAlloc = n;
2440 azResult = azNew;
2441 }
2442 azResult[nRow] = sqlite3_mprintf("%s", sqlite3_column_text(pStmt, 0));
2443 if( azResult[nRow] ) nRow++;
2444 }
2445 sqlite3_finalize(pStmt);
2446 if( nRow>0 ){
drhe3710332000-09-29 13:30:53 +00002447 int len, maxlen = 0;
2448 int i, j;
2449 int nPrintCol, nPrintRow;
drh98781232012-04-23 12:38:05 +00002450 for(i=0; i<nRow; i++){
drh4f21c4a2008-12-10 22:15:00 +00002451 len = strlen30(azResult[i]);
drhe3710332000-09-29 13:30:53 +00002452 if( len>maxlen ) maxlen = len;
2453 }
2454 nPrintCol = 80/(maxlen+2);
2455 if( nPrintCol<1 ) nPrintCol = 1;
2456 nPrintRow = (nRow + nPrintCol - 1)/nPrintCol;
2457 for(i=0; i<nPrintRow; i++){
drh98781232012-04-23 12:38:05 +00002458 for(j=i; j<nRow; j+=nPrintRow){
2459 char *zSp = j<nPrintRow ? "" : " ";
drh151b7d52013-05-06 20:28:54 +00002460 fprintf(p->out, "%s%-*s", zSp, maxlen, azResult[j] ? azResult[j] : "");
drhe3710332000-09-29 13:30:53 +00002461 }
drh151b7d52013-05-06 20:28:54 +00002462 fprintf(p->out, "\n");
drhe3710332000-09-29 13:30:53 +00002463 }
2464 }
drh98781232012-04-23 12:38:05 +00002465 for(ii=0; ii<nRow; ii++) sqlite3_free(azResult[ii]);
2466 sqlite3_free(azResult);
drh75897232000-05-29 14:26:00 +00002467 }else
2468
shaneh96887e12011-02-10 21:08:58 +00002469 if( c=='t' && n>=8 && strncmp(azArg[0], "testctrl", n)==0 && nArg>=2 ){
drhd416fe72011-03-17 16:45:50 +00002470 static const struct {
2471 const char *zCtrlName; /* Name of a test-control option */
2472 int ctrlCode; /* Integer code for that option */
2473 } aCtrl[] = {
2474 { "prng_save", SQLITE_TESTCTRL_PRNG_SAVE },
2475 { "prng_restore", SQLITE_TESTCTRL_PRNG_RESTORE },
2476 { "prng_reset", SQLITE_TESTCTRL_PRNG_RESET },
2477 { "bitvec_test", SQLITE_TESTCTRL_BITVEC_TEST },
2478 { "fault_install", SQLITE_TESTCTRL_FAULT_INSTALL },
2479 { "benign_malloc_hooks", SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS },
2480 { "pending_byte", SQLITE_TESTCTRL_PENDING_BYTE },
2481 { "assert", SQLITE_TESTCTRL_ASSERT },
2482 { "always", SQLITE_TESTCTRL_ALWAYS },
2483 { "reserve", SQLITE_TESTCTRL_RESERVE },
2484 { "optimizations", SQLITE_TESTCTRL_OPTIMIZATIONS },
2485 { "iskeyword", SQLITE_TESTCTRL_ISKEYWORD },
drhd416fe72011-03-17 16:45:50 +00002486 { "scratchmalloc", SQLITE_TESTCTRL_SCRATCHMALLOC },
2487 };
shaneh96887e12011-02-10 21:08:58 +00002488 int testctrl = -1;
2489 int rc = 0;
drhd416fe72011-03-17 16:45:50 +00002490 int i, n;
shaneh96887e12011-02-10 21:08:58 +00002491 open_db(p);
2492
drhd416fe72011-03-17 16:45:50 +00002493 /* convert testctrl text option to value. allow any unique prefix
2494 ** of the option name, or a numerical value. */
shanehcef83682011-04-07 03:41:01 +00002495 n = strlen30(azArg[1]);
drhfcd71b62011-04-05 22:08:24 +00002496 for(i=0; i<(int)(sizeof(aCtrl)/sizeof(aCtrl[0])); i++){
drhd416fe72011-03-17 16:45:50 +00002497 if( strncmp(azArg[1], aCtrl[i].zCtrlName, n)==0 ){
2498 if( testctrl<0 ){
2499 testctrl = aCtrl[i].ctrlCode;
2500 }else{
drhb07028f2011-10-14 21:49:18 +00002501 fprintf(stderr, "ambiguous option name: \"%s\"\n", azArg[1]);
drhd416fe72011-03-17 16:45:50 +00002502 testctrl = -1;
2503 break;
2504 }
2505 }
2506 }
drh348d19c2013-06-03 12:47:43 +00002507 if( testctrl<0 ) testctrl = (int)integerValue(azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002508 if( (testctrl<SQLITE_TESTCTRL_FIRST) || (testctrl>SQLITE_TESTCTRL_LAST) ){
2509 fprintf(stderr,"Error: invalid testctrl option: %s\n", azArg[1]);
2510 }else{
2511 switch(testctrl){
2512
2513 /* sqlite3_test_control(int, db, int) */
2514 case SQLITE_TESTCTRL_OPTIMIZATIONS:
2515 case SQLITE_TESTCTRL_RESERVE:
2516 if( nArg==3 ){
2517 int opt = (int)strtol(azArg[2], 0, 0);
2518 rc = sqlite3_test_control(testctrl, p->db, opt);
drh151b7d52013-05-06 20:28:54 +00002519 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00002520 } else {
drhd416fe72011-03-17 16:45:50 +00002521 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
2522 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002523 }
2524 break;
2525
2526 /* sqlite3_test_control(int) */
2527 case SQLITE_TESTCTRL_PRNG_SAVE:
2528 case SQLITE_TESTCTRL_PRNG_RESTORE:
2529 case SQLITE_TESTCTRL_PRNG_RESET:
shaneh96887e12011-02-10 21:08:58 +00002530 if( nArg==2 ){
2531 rc = sqlite3_test_control(testctrl);
drh151b7d52013-05-06 20:28:54 +00002532 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00002533 } else {
2534 fprintf(stderr,"Error: testctrl %s takes no options\n", azArg[1]);
2535 }
2536 break;
2537
2538 /* sqlite3_test_control(int, uint) */
2539 case SQLITE_TESTCTRL_PENDING_BYTE:
2540 if( nArg==3 ){
drh7d9f3942013-04-03 01:26:54 +00002541 unsigned int opt = (unsigned int)integerValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00002542 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00002543 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00002544 } else {
drhd416fe72011-03-17 16:45:50 +00002545 fprintf(stderr,"Error: testctrl %s takes a single unsigned"
2546 " int option\n", azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002547 }
2548 break;
2549
2550 /* sqlite3_test_control(int, int) */
2551 case SQLITE_TESTCTRL_ASSERT:
2552 case SQLITE_TESTCTRL_ALWAYS:
2553 if( nArg==3 ){
drh348d19c2013-06-03 12:47:43 +00002554 int opt = booleanValue(azArg[2]);
shaneh96887e12011-02-10 21:08:58 +00002555 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00002556 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00002557 } else {
drhd416fe72011-03-17 16:45:50 +00002558 fprintf(stderr,"Error: testctrl %s takes a single int option\n",
2559 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002560 }
2561 break;
2562
2563 /* sqlite3_test_control(int, char *) */
2564#ifdef SQLITE_N_KEYWORD
2565 case SQLITE_TESTCTRL_ISKEYWORD:
2566 if( nArg==3 ){
2567 const char *opt = azArg[2];
2568 rc = sqlite3_test_control(testctrl, opt);
drh151b7d52013-05-06 20:28:54 +00002569 fprintf(p->out, "%d (0x%08x)\n", rc, rc);
shaneh96887e12011-02-10 21:08:58 +00002570 } else {
drhd416fe72011-03-17 16:45:50 +00002571 fprintf(stderr,"Error: testctrl %s takes a single char * option\n",
2572 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002573 }
2574 break;
2575#endif
2576
2577 case SQLITE_TESTCTRL_BITVEC_TEST:
2578 case SQLITE_TESTCTRL_FAULT_INSTALL:
2579 case SQLITE_TESTCTRL_BENIGN_MALLOC_HOOKS:
2580 case SQLITE_TESTCTRL_SCRATCHMALLOC:
2581 default:
drhd416fe72011-03-17 16:45:50 +00002582 fprintf(stderr,"Error: CLI support for testctrl %s not implemented\n",
2583 azArg[1]);
shaneh96887e12011-02-10 21:08:58 +00002584 break;
2585 }
2586 }
2587 }else
2588
shanehe2aa9d72009-11-06 17:20:17 +00002589 if( c=='t' && n>4 && strncmp(azArg[0], "timeout", n)==0 && nArg==2 ){
drh44c2eb12003-04-30 11:38:26 +00002590 open_db(p);
drh348d19c2013-06-03 12:47:43 +00002591 sqlite3_busy_timeout(p->db, (int)integerValue(azArg[1]));
shanehe2aa9d72009-11-06 17:20:17 +00002592 }else
2593
drhd416fe72011-03-17 16:45:50 +00002594 if( HAS_TIMER && c=='t' && n>=5 && strncmp(azArg[0], "timer", n)==0
2595 && nArg==2
2596 ){
drh3b1a9882007-11-02 12:53:03 +00002597 enableTimer = booleanValue(azArg[1]);
shanehe2aa9d72009-11-06 17:20:17 +00002598 }else
2599
drh42f64e52012-04-04 16:56:23 +00002600 if( c=='t' && strncmp(azArg[0], "trace", n)==0 && nArg>1 ){
drh98781232012-04-23 12:38:05 +00002601 open_db(p);
drh42f64e52012-04-04 16:56:23 +00002602 output_file_close(p->traceOut);
2603 p->traceOut = output_file_open(azArg[1]);
drhbbb0be82012-06-27 16:12:27 +00002604#if !defined(SQLITE_OMIT_TRACE) && !defined(SQLITE_OMIT_FLOATING_POINT)
drh42f64e52012-04-04 16:56:23 +00002605 if( p->traceOut==0 ){
2606 sqlite3_trace(p->db, 0, 0);
2607 }else{
2608 sqlite3_trace(p->db, sql_trace_callback, p->traceOut);
2609 }
2610#endif
2611 }else
2612
drh9fd301b2011-06-03 13:28:22 +00002613 if( c=='v' && strncmp(azArg[0], "version", n)==0 ){
drh151b7d52013-05-06 20:28:54 +00002614 fprintf(p->out, "SQLite %s %s\n" /*extra-version-info*/,
drh9fd301b2011-06-03 13:28:22 +00002615 sqlite3_libversion(), sqlite3_sourceid());
2616 }else
2617
drhde60fc22011-12-14 17:53:36 +00002618 if( c=='v' && strncmp(azArg[0], "vfsname", n)==0 ){
2619 const char *zDbName = nArg==2 ? azArg[1] : "main";
2620 char *zVfsName = 0;
2621 if( p->db ){
2622 sqlite3_file_control(p->db, zDbName, SQLITE_FCNTL_VFSNAME, &zVfsName);
2623 if( zVfsName ){
drh151b7d52013-05-06 20:28:54 +00002624 fprintf(p->out, "%s\n", zVfsName);
drhde60fc22011-12-14 17:53:36 +00002625 sqlite3_free(zVfsName);
2626 }
2627 }
2628 }else
2629
drhcef4fc82012-09-21 22:50:45 +00002630#if defined(SQLITE_DEBUG) && defined(SQLITE_ENABLE_WHERETRACE)
2631 if( c=='w' && strncmp(azArg[0], "wheretrace", n)==0 ){
2632 extern int sqlite3WhereTrace;
drh7d9f3942013-04-03 01:26:54 +00002633 sqlite3WhereTrace = booleanValue(azArg[1]);
drhcef4fc82012-09-21 22:50:45 +00002634 }else
2635#endif
2636
shanehe2aa9d72009-11-06 17:20:17 +00002637 if( c=='w' && strncmp(azArg[0], "width", n)==0 && nArg>1 ){
drh75897232000-05-29 14:26:00 +00002638 int j;
drh43617e92006-03-06 20:55:46 +00002639 assert( nArg<=ArraySize(azArg) );
drh75897232000-05-29 14:26:00 +00002640 for(j=1; j<nArg && j<ArraySize(p->colWidth); j++){
drh348d19c2013-06-03 12:47:43 +00002641 p->colWidth[j-1] = (int)integerValue(azArg[j]);
drh75897232000-05-29 14:26:00 +00002642 }
2643 }else
2644
2645 {
shane9bd1b442009-10-23 01:27:39 +00002646 fprintf(stderr, "Error: unknown command or invalid arguments: "
drh67505e72002-04-19 12:34:06 +00002647 " \"%s\". Enter \".help\" for help\n", azArg[0]);
shane9bd1b442009-10-23 01:27:39 +00002648 rc = 1;
drh75897232000-05-29 14:26:00 +00002649 }
drh67505e72002-04-19 12:34:06 +00002650
2651 return rc;
drh75897232000-05-29 14:26:00 +00002652}
2653
drh67505e72002-04-19 12:34:06 +00002654/*
drh91a66392007-09-07 01:12:32 +00002655** Return TRUE if a semicolon occurs anywhere in the first N characters
2656** of string z[].
drh324ccef2003-02-05 14:06:20 +00002657*/
drh91a66392007-09-07 01:12:32 +00002658static int _contains_semicolon(const char *z, int N){
2659 int i;
2660 for(i=0; i<N; i++){ if( z[i]==';' ) return 1; }
2661 return 0;
drh324ccef2003-02-05 14:06:20 +00002662}
2663
2664/*
drh70c7a4b2003-04-26 03:03:06 +00002665** Test to see if a line consists entirely of whitespace.
2666*/
2667static int _all_whitespace(const char *z){
2668 for(; *z; z++){
drhf0693c82011-10-11 20:41:54 +00002669 if( IsSpace(z[0]) ) continue;
drh70c7a4b2003-04-26 03:03:06 +00002670 if( *z=='/' && z[1]=='*' ){
2671 z += 2;
2672 while( *z && (*z!='*' || z[1]!='/') ){ z++; }
2673 if( *z==0 ) return 0;
2674 z++;
2675 continue;
2676 }
2677 if( *z=='-' && z[1]=='-' ){
2678 z += 2;
2679 while( *z && *z!='\n' ){ z++; }
2680 if( *z==0 ) return 1;
2681 continue;
2682 }
2683 return 0;
2684 }
2685 return 1;
2686}
2687
2688/*
drha9b17162003-04-29 18:01:28 +00002689** Return TRUE if the line typed in is an SQL command terminator other
2690** than a semi-colon. The SQL Server style "go" command is understood
2691** as is the Oracle "/".
2692*/
2693static int _is_command_terminator(const char *zLine){
drhf0693c82011-10-11 20:41:54 +00002694 while( IsSpace(zLine[0]) ){ zLine++; };
drh233a5312008-12-18 22:25:13 +00002695 if( zLine[0]=='/' && _all_whitespace(&zLine[1]) ){
2696 return 1; /* Oracle */
2697 }
drhf0693c82011-10-11 20:41:54 +00002698 if( ToLower(zLine[0])=='g' && ToLower(zLine[1])=='o'
drhc8d74412004-08-31 23:41:26 +00002699 && _all_whitespace(&zLine[2]) ){
drha9b17162003-04-29 18:01:28 +00002700 return 1; /* SQL Server */
2701 }
2702 return 0;
2703}
2704
2705/*
drh233a5312008-12-18 22:25:13 +00002706** Return true if zSql is a complete SQL statement. Return false if it
2707** ends in the middle of a string literal or C-style comment.
2708*/
2709static int _is_complete(char *zSql, int nSql){
2710 int rc;
2711 if( zSql==0 ) return 1;
2712 zSql[nSql] = ';';
2713 zSql[nSql+1] = 0;
2714 rc = sqlite3_complete(zSql);
2715 zSql[nSql] = 0;
2716 return rc;
2717}
2718
2719/*
drh67505e72002-04-19 12:34:06 +00002720** Read input from *in and process it. If *in==0 then input
2721** is interactive - the user is typing it it. Otherwise, input
2722** is coming from a file or device. A prompt is issued and history
2723** is saved only if input is interactive. An interrupt signal will
2724** cause this routine to exit immediately, unless input is interactive.
drhc28490c2006-10-26 14:25:58 +00002725**
2726** Return the number of errors.
drh67505e72002-04-19 12:34:06 +00002727*/
drhc28490c2006-10-26 14:25:58 +00002728static int process_input(struct callback_data *p, FILE *in){
danielk19772ac27622007-07-03 05:31:16 +00002729 char *zLine = 0;
drhdaffd0e2001-04-11 14:28:42 +00002730 char *zSql = 0;
2731 int nSql = 0;
drh91a66392007-09-07 01:12:32 +00002732 int nSqlPrior = 0;
drhdaffd0e2001-04-11 14:28:42 +00002733 char *zErrMsg;
drhc49f44e2006-10-26 18:15:42 +00002734 int rc;
2735 int errCnt = 0;
drhc28490c2006-10-26 14:25:58 +00002736 int lineno = 0;
2737 int startline = 0;
drhc49f44e2006-10-26 18:15:42 +00002738
2739 while( errCnt==0 || !bail_on_error || (in==0 && stdin_is_interactive) ){
2740 fflush(p->out);
danielk19772ac27622007-07-03 05:31:16 +00002741 free(zLine);
drhc49f44e2006-10-26 18:15:42 +00002742 zLine = one_input_line(zSql, in);
2743 if( zLine==0 ){
drh9b8d3572012-04-21 11:33:39 +00002744 /* End of input */
2745 if( stdin_is_interactive ) printf("\n");
2746 break;
drhc49f44e2006-10-26 18:15:42 +00002747 }
drh67505e72002-04-19 12:34:06 +00002748 if( seenInterrupt ){
2749 if( in!=0 ) break;
2750 seenInterrupt = 0;
2751 }
drhc28490c2006-10-26 14:25:58 +00002752 lineno++;
drhf817b6b2003-06-16 00:16:41 +00002753 if( (zSql==0 || zSql[0]==0) && _all_whitespace(zLine) ) continue;
drh2af0b2d2002-02-21 02:25:02 +00002754 if( zLine && zLine[0]=='.' && nSql==0 ){
shaneb9fc17d2009-10-22 21:23:35 +00002755 if( p->echoOn ) printf("%s\n", zLine);
drhc49f44e2006-10-26 18:15:42 +00002756 rc = do_meta_command(zLine, p);
shane916f9612009-10-23 00:37:15 +00002757 if( rc==2 ){ /* exit requested */
drh47ad6842006-11-08 12:25:42 +00002758 break;
2759 }else if( rc ){
drhc49f44e2006-10-26 18:15:42 +00002760 errCnt++;
2761 }
drhdaffd0e2001-04-11 14:28:42 +00002762 continue;
2763 }
drh233a5312008-12-18 22:25:13 +00002764 if( _is_command_terminator(zLine) && _is_complete(zSql, nSql) ){
drh5bb3eb92007-05-04 13:15:55 +00002765 memcpy(zLine,";",2);
drha9b17162003-04-29 18:01:28 +00002766 }
drh91a66392007-09-07 01:12:32 +00002767 nSqlPrior = nSql;
drhdaffd0e2001-04-11 14:28:42 +00002768 if( zSql==0 ){
2769 int i;
drhf0693c82011-10-11 20:41:54 +00002770 for(i=0; zLine[i] && IsSpace(zLine[i]); i++){}
drhdaffd0e2001-04-11 14:28:42 +00002771 if( zLine[i]!=0 ){
drh4f21c4a2008-12-10 22:15:00 +00002772 nSql = strlen30(zLine);
drh233a5312008-12-18 22:25:13 +00002773 zSql = malloc( nSql+3 );
drhc1f44942006-05-10 14:39:13 +00002774 if( zSql==0 ){
shane9bd1b442009-10-23 01:27:39 +00002775 fprintf(stderr, "Error: out of memory\n");
drhc1f44942006-05-10 14:39:13 +00002776 exit(1);
2777 }
drh5bb3eb92007-05-04 13:15:55 +00002778 memcpy(zSql, zLine, nSql+1);
drhc28490c2006-10-26 14:25:58 +00002779 startline = lineno;
drhdaffd0e2001-04-11 14:28:42 +00002780 }
2781 }else{
drh4f21c4a2008-12-10 22:15:00 +00002782 int len = strlen30(zLine);
drh233a5312008-12-18 22:25:13 +00002783 zSql = realloc( zSql, nSql + len + 4 );
drhdaffd0e2001-04-11 14:28:42 +00002784 if( zSql==0 ){
shane9bd1b442009-10-23 01:27:39 +00002785 fprintf(stderr,"Error: out of memory\n");
drhdaffd0e2001-04-11 14:28:42 +00002786 exit(1);
2787 }
drh5bb3eb92007-05-04 13:15:55 +00002788 zSql[nSql++] = '\n';
2789 memcpy(&zSql[nSql], zLine, len+1);
drhdaffd0e2001-04-11 14:28:42 +00002790 nSql += len;
2791 }
drh91a66392007-09-07 01:12:32 +00002792 if( zSql && _contains_semicolon(&zSql[nSqlPrior], nSql-nSqlPrior)
2793 && sqlite3_complete(zSql) ){
drhdaffd0e2001-04-11 14:28:42 +00002794 p->cnt = 0;
drh44c2eb12003-04-30 11:38:26 +00002795 open_db(p);
drh3b1a9882007-11-02 12:53:03 +00002796 BEGIN_TIMER;
shane626a6e42009-10-22 17:30:15 +00002797 rc = shell_exec(p->db, zSql, shell_callback, p, &zErrMsg);
drh3b1a9882007-11-02 12:53:03 +00002798 END_TIMER;
drh7f953e22002-07-13 17:33:45 +00002799 if( rc || zErrMsg ){
drhc28490c2006-10-26 14:25:58 +00002800 char zPrefix[100];
2801 if( in!=0 || !stdin_is_interactive ){
drh5bb3eb92007-05-04 13:15:55 +00002802 sqlite3_snprintf(sizeof(zPrefix), zPrefix,
shane9bd1b442009-10-23 01:27:39 +00002803 "Error: near line %d:", startline);
drhc28490c2006-10-26 14:25:58 +00002804 }else{
shane9bd1b442009-10-23 01:27:39 +00002805 sqlite3_snprintf(sizeof(zPrefix), zPrefix, "Error:");
drhc28490c2006-10-26 14:25:58 +00002806 }
drh7f953e22002-07-13 17:33:45 +00002807 if( zErrMsg!=0 ){
shaned2bed1c2009-10-21 03:56:54 +00002808 fprintf(stderr, "%s %s\n", zPrefix, zErrMsg);
drh3f4fedb2004-05-31 19:34:33 +00002809 sqlite3_free(zErrMsg);
drh7f953e22002-07-13 17:33:45 +00002810 zErrMsg = 0;
2811 }else{
shaned2bed1c2009-10-21 03:56:54 +00002812 fprintf(stderr, "%s %s\n", zPrefix, sqlite3_errmsg(p->db));
drh7f953e22002-07-13 17:33:45 +00002813 }
drhc49f44e2006-10-26 18:15:42 +00002814 errCnt++;
drhdaffd0e2001-04-11 14:28:42 +00002815 }
2816 free(zSql);
2817 zSql = 0;
2818 nSql = 0;
drh7a411f42013-04-17 17:33:17 +00002819 }else if( zSql && _all_whitespace(zSql) ){
2820 free(zSql);
2821 zSql = 0;
2822 nSql = 0;
drhdaffd0e2001-04-11 14:28:42 +00002823 }
2824 }
2825 if( zSql ){
drhd416fe72011-03-17 16:45:50 +00002826 if( !_all_whitespace(zSql) ){
2827 fprintf(stderr, "Error: incomplete SQL: %s\n", zSql);
2828 }
drhdaffd0e2001-04-11 14:28:42 +00002829 free(zSql);
2830 }
danielk19772ac27622007-07-03 05:31:16 +00002831 free(zLine);
drh4d15a0d2012-12-01 20:21:22 +00002832 return errCnt>0;
drhdaffd0e2001-04-11 14:28:42 +00002833}
2834
drh67505e72002-04-19 12:34:06 +00002835/*
2836** Return a pathname which is the user's home directory. A
drh85e72432012-04-11 11:38:53 +00002837** 0 return indicates an error of some kind.
drh67505e72002-04-19 12:34:06 +00002838*/
2839static char *find_home_dir(void){
drh85e72432012-04-11 11:38:53 +00002840 static char *home_dir = NULL;
2841 if( home_dir ) return home_dir;
persicom7e2dfdd2002-04-18 02:46:52 +00002842
drh83905c92012-06-21 13:00:37 +00002843#if !defined(_WIN32) && !defined(WIN32) && !defined(_WIN32_WCE) && !defined(__RTP__) && !defined(_WRS_KERNEL)
mistachkinc8bde372012-06-18 08:00:56 +00002844 {
2845 struct passwd *pwent;
2846 uid_t uid = getuid();
2847 if( (pwent=getpwuid(uid)) != NULL) {
2848 home_dir = pwent->pw_dir;
2849 }
drh67505e72002-04-19 12:34:06 +00002850 }
2851#endif
2852
chw65d3c132007-11-12 21:09:10 +00002853#if defined(_WIN32_WCE)
2854 /* Windows CE (arm-wince-mingw32ce-gcc) does not provide getenv()
2855 */
drh85e72432012-04-11 11:38:53 +00002856 home_dir = "/";
chw65d3c132007-11-12 21:09:10 +00002857#else
2858
drh83905c92012-06-21 13:00:37 +00002859#if defined(_WIN32) || defined(WIN32)
drh164a1b62006-08-19 11:15:20 +00002860 if (!home_dir) {
2861 home_dir = getenv("USERPROFILE");
2862 }
2863#endif
2864
drh67505e72002-04-19 12:34:06 +00002865 if (!home_dir) {
2866 home_dir = getenv("HOME");
drh67505e72002-04-19 12:34:06 +00002867 }
2868
drh83905c92012-06-21 13:00:37 +00002869#if defined(_WIN32) || defined(WIN32)
drhe98d4fa2002-04-21 19:06:22 +00002870 if (!home_dir) {
drh164a1b62006-08-19 11:15:20 +00002871 char *zDrive, *zPath;
2872 int n;
2873 zDrive = getenv("HOMEDRIVE");
2874 zPath = getenv("HOMEPATH");
2875 if( zDrive && zPath ){
drh4f21c4a2008-12-10 22:15:00 +00002876 n = strlen30(zDrive) + strlen30(zPath) + 1;
drh164a1b62006-08-19 11:15:20 +00002877 home_dir = malloc( n );
2878 if( home_dir==0 ) return 0;
2879 sqlite3_snprintf(n, home_dir, "%s%s", zDrive, zPath);
2880 return home_dir;
2881 }
2882 home_dir = "c:\\";
drhe98d4fa2002-04-21 19:06:22 +00002883 }
2884#endif
2885
chw65d3c132007-11-12 21:09:10 +00002886#endif /* !_WIN32_WCE */
2887
drh67505e72002-04-19 12:34:06 +00002888 if( home_dir ){
drh4f21c4a2008-12-10 22:15:00 +00002889 int n = strlen30(home_dir) + 1;
drh5bb3eb92007-05-04 13:15:55 +00002890 char *z = malloc( n );
2891 if( z ) memcpy(z, home_dir, n);
drh67505e72002-04-19 12:34:06 +00002892 home_dir = z;
2893 }
drhe98d4fa2002-04-21 19:06:22 +00002894
drh67505e72002-04-19 12:34:06 +00002895 return home_dir;
2896}
2897
2898/*
2899** Read input from the file given by sqliterc_override. Or if that
2900** parameter is NULL, take input from ~/.sqliterc
shane9bd1b442009-10-23 01:27:39 +00002901**
2902** Returns the number of errors.
drh67505e72002-04-19 12:34:06 +00002903*/
shane9bd1b442009-10-23 01:27:39 +00002904static int process_sqliterc(
drh22fbcb82004-02-01 01:22:50 +00002905 struct callback_data *p, /* Configuration data */
2906 const char *sqliterc_override /* Name of config file. NULL to use default */
2907){
persicom7e2dfdd2002-04-18 02:46:52 +00002908 char *home_dir = NULL;
drh22fbcb82004-02-01 01:22:50 +00002909 const char *sqliterc = sqliterc_override;
drh43617e92006-03-06 20:55:46 +00002910 char *zBuf = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00002911 FILE *in = NULL;
shane9bd1b442009-10-23 01:27:39 +00002912 int rc = 0;
persicom7e2dfdd2002-04-18 02:46:52 +00002913
2914 if (sqliterc == NULL) {
drh67505e72002-04-19 12:34:06 +00002915 home_dir = find_home_dir();
drhe98d4fa2002-04-21 19:06:22 +00002916 if( home_dir==0 ){
chw97185482008-11-17 08:05:31 +00002917#if !defined(__RTP__) && !defined(_WRS_KERNEL)
shane86f5bdb2009-10-24 02:00:07 +00002918 fprintf(stderr,"%s: Error: cannot locate your home directory\n", Argv0);
chw97185482008-11-17 08:05:31 +00002919#endif
shane9bd1b442009-10-23 01:27:39 +00002920 return 1;
drhe98d4fa2002-04-21 19:06:22 +00002921 }
drh2f3de322012-06-27 16:41:31 +00002922 sqlite3_initialize();
drh85e72432012-04-11 11:38:53 +00002923 zBuf = sqlite3_mprintf("%s/.sqliterc",home_dir);
2924 sqliterc = zBuf;
persicom7e2dfdd2002-04-18 02:46:52 +00002925 }
drha1f9b5e2004-02-14 16:31:02 +00002926 in = fopen(sqliterc,"rb");
drh22fbcb82004-02-01 01:22:50 +00002927 if( in ){
drhc28490c2006-10-26 14:25:58 +00002928 if( stdin_is_interactive ){
shane86f5bdb2009-10-24 02:00:07 +00002929 fprintf(stderr,"-- Loading resources from %s\n",sqliterc);
drh22fbcb82004-02-01 01:22:50 +00002930 }
shane9bd1b442009-10-23 01:27:39 +00002931 rc = process_input(p,in);
drhdd45df82002-04-18 12:39:03 +00002932 fclose(in);
persicom7e2dfdd2002-04-18 02:46:52 +00002933 }
drh85e72432012-04-11 11:38:53 +00002934 sqlite3_free(zBuf);
shane9bd1b442009-10-23 01:27:39 +00002935 return rc;
persicom7e2dfdd2002-04-18 02:46:52 +00002936}
2937
drh67505e72002-04-19 12:34:06 +00002938/*
drhe1e38c42003-05-04 18:30:59 +00002939** Show available command line options
2940*/
2941static const char zOptions[] =
drhc49f44e2006-10-26 18:15:42 +00002942 " -bail stop after hitting an error\n"
drhc49f44e2006-10-26 18:15:42 +00002943 " -batch force batch I/O\n"
drhe1e38c42003-05-04 18:30:59 +00002944 " -column set output mode to 'column'\n"
mistachkin6d81d752012-10-25 15:43:28 +00002945 " -cmd COMMAND run \"COMMAND\" before reading stdin\n"
drhc49f44e2006-10-26 18:15:42 +00002946 " -csv set output mode to 'csv'\n"
drhcc3b4f82012-02-07 14:13:50 +00002947 " -echo print commands before execution\n"
mistachkin6d81d752012-10-25 15:43:28 +00002948 " -init FILENAME read/process named file\n"
drhcc3b4f82012-02-07 14:13:50 +00002949 " -[no]header turn headers on or off\n"
drh98d312f2012-10-25 15:23:14 +00002950#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
2951 " -heap SIZE Size of heap for memsys3 or memsys5\n"
2952#endif
drhcc3b4f82012-02-07 14:13:50 +00002953 " -help show this message\n"
drhe1e38c42003-05-04 18:30:59 +00002954 " -html set output mode to HTML\n"
drhcc3b4f82012-02-07 14:13:50 +00002955 " -interactive force interactive I/O\n"
drhe1e38c42003-05-04 18:30:59 +00002956 " -line set output mode to 'line'\n"
2957 " -list set output mode to 'list'\n"
drh7d9f3942013-04-03 01:26:54 +00002958 " -mmap N default mmap size set to N\n"
drhcc3b4f82012-02-07 14:13:50 +00002959#ifdef SQLITE_ENABLE_MULTIPLEX
2960 " -multiplex enable the multiplexor VFS\n"
2961#endif
drh98d312f2012-10-25 15:23:14 +00002962 " -nullvalue TEXT set text string for NULL values. Default ''\n"
2963 " -separator SEP set output field separator. Default: '|'\n"
shaneh642d8b82010-07-28 16:05:34 +00002964 " -stats print memory stats before each finalize\n"
drhe1e38c42003-05-04 18:30:59 +00002965 " -version show SQLite version\n"
drha7e61d82011-03-12 17:02:57 +00002966 " -vfs NAME use NAME as the default VFS\n"
drh2b625e22011-03-16 17:05:28 +00002967#ifdef SQLITE_ENABLE_VFSTRACE
2968 " -vfstrace enable tracing of all VFS calls\n"
2969#endif
drhe1e38c42003-05-04 18:30:59 +00002970;
2971static void usage(int showDetail){
drh80e8be92006-08-29 12:04:19 +00002972 fprintf(stderr,
2973 "Usage: %s [OPTIONS] FILENAME [SQL]\n"
2974 "FILENAME is the name of an SQLite database. A new database is created\n"
2975 "if the file does not previously exist.\n", Argv0);
drhe1e38c42003-05-04 18:30:59 +00002976 if( showDetail ){
drh80e8be92006-08-29 12:04:19 +00002977 fprintf(stderr, "OPTIONS include:\n%s", zOptions);
drhe1e38c42003-05-04 18:30:59 +00002978 }else{
2979 fprintf(stderr, "Use the -help option for additional information\n");
2980 }
2981 exit(1);
2982}
2983
2984/*
drh67505e72002-04-19 12:34:06 +00002985** Initialize the state information in data
2986*/
drh0850b532006-01-31 19:31:43 +00002987static void main_init(struct callback_data *data) {
persicom7e2dfdd2002-04-18 02:46:52 +00002988 memset(data, 0, sizeof(*data));
2989 data->mode = MODE_List;
drh5bb3eb92007-05-04 13:15:55 +00002990 memcpy(data->separator,"|", 2);
persicom7e2dfdd2002-04-18 02:46:52 +00002991 data->showHeader = 0;
drh52784bd2011-05-18 17:15:06 +00002992 sqlite3_config(SQLITE_CONFIG_URI, 1);
drh127f9d72010-02-23 01:47:00 +00002993 sqlite3_config(SQLITE_CONFIG_LOG, shellLog, data);
drh5bb3eb92007-05-04 13:15:55 +00002994 sqlite3_snprintf(sizeof(mainPrompt), mainPrompt,"sqlite> ");
2995 sqlite3_snprintf(sizeof(continuePrompt), continuePrompt," ...> ");
dan0f831772010-03-03 07:23:12 +00002996 sqlite3_config(SQLITE_CONFIG_SINGLETHREAD);
persicom7e2dfdd2002-04-18 02:46:52 +00002997}
2998
drh98d312f2012-10-25 15:23:14 +00002999/*
3000** Get the argument to an --option. Throw an error and die if no argument
3001** is available.
3002*/
3003static char *cmdline_option_value(int argc, char **argv, int i){
3004 if( i==argc ){
3005 fprintf(stderr, "%s: Error: missing argument to %s\n",
3006 argv[0], argv[argc-1]);
3007 exit(1);
3008 }
3009 return argv[i];
3010}
3011
drh75897232000-05-29 14:26:00 +00003012int main(int argc, char **argv){
drh75897232000-05-29 14:26:00 +00003013 char *zErrMsg = 0;
3014 struct callback_data data;
drh22fbcb82004-02-01 01:22:50 +00003015 const char *zInitFile = 0;
3016 char *zFirstCmd = 0;
drh44c2eb12003-04-30 11:38:26 +00003017 int i;
drhc28490c2006-10-26 14:25:58 +00003018 int rc = 0;
drh75897232000-05-29 14:26:00 +00003019
drh52784bd2011-05-18 17:15:06 +00003020 if( strcmp(sqlite3_sourceid(),SQLITE_SOURCE_ID)!=0 ){
3021 fprintf(stderr, "SQLite header and source version mismatch\n%s\n%s\n",
3022 sqlite3_sourceid(), SQLITE_SOURCE_ID);
3023 exit(1);
3024 }
drhdaffd0e2001-04-11 14:28:42 +00003025 Argv0 = argv[0];
persicom7e2dfdd2002-04-18 02:46:52 +00003026 main_init(&data);
drhc28490c2006-10-26 14:25:58 +00003027 stdin_is_interactive = isatty(0);
persicom7e2dfdd2002-04-18 02:46:52 +00003028
drh44c2eb12003-04-30 11:38:26 +00003029 /* Make sure we have a valid signal handler early, before anything
3030 ** else is done.
3031 */
drh4c504392000-10-16 22:06:40 +00003032#ifdef SIGINT
3033 signal(SIGINT, interrupt_handler);
3034#endif
drh44c2eb12003-04-30 11:38:26 +00003035
drh22fbcb82004-02-01 01:22:50 +00003036 /* Do an initial pass through the command-line argument to locate
3037 ** the name of the database file, the name of the initialization file,
drh9c88d682010-12-17 14:03:01 +00003038 ** the size of the alternative malloc heap,
drh22fbcb82004-02-01 01:22:50 +00003039 ** and the first command to execute.
drh44c2eb12003-04-30 11:38:26 +00003040 */
drh98d312f2012-10-25 15:23:14 +00003041 for(i=1; i<argc; i++){
drhc28490c2006-10-26 14:25:58 +00003042 char *z;
drhc28490c2006-10-26 14:25:58 +00003043 z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003044 if( z[0]!='-' ){
3045 if( data.zDbFilename==0 ){
3046 data.zDbFilename = z;
3047 continue;
3048 }
3049 if( zFirstCmd==0 ){
3050 zFirstCmd = z;
3051 continue;
3052 }
3053 fprintf(stderr,"%s: Error: too many options: \"%s\"\n", Argv0, argv[i]);
3054 fprintf(stderr,"Use -help for a list of options.\n");
3055 return 1;
3056 }
drhcc3b4f82012-02-07 14:13:50 +00003057 if( z[1]=='-' ) z++;
3058 if( strcmp(z,"-separator")==0
3059 || strcmp(z,"-nullvalue")==0
3060 || strcmp(z,"-cmd")==0
3061 ){
drh98d312f2012-10-25 15:23:14 +00003062 (void)cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003063 }else if( strcmp(z,"-init")==0 ){
drh98d312f2012-10-25 15:23:14 +00003064 zInitFile = cmdline_option_value(argc, argv, ++i);
drhcc3b4f82012-02-07 14:13:50 +00003065 }else if( strcmp(z,"-batch")==0 ){
drh98d312f2012-10-25 15:23:14 +00003066 /* Need to check for batch mode here to so we can avoid printing
3067 ** informational messages (like from process_sqliterc) before
3068 ** we do the actual processing of arguments later in a second pass.
3069 */
shanef69573d2009-10-24 02:06:14 +00003070 stdin_is_interactive = 0;
drhcc3b4f82012-02-07 14:13:50 +00003071 }else if( strcmp(z,"-heap")==0 ){
drhb07028f2011-10-14 21:49:18 +00003072#if defined(SQLITE_ENABLE_MEMSYS3) || defined(SQLITE_ENABLE_MEMSYS5)
drh9c88d682010-12-17 14:03:01 +00003073 int j, c;
3074 const char *zSize;
3075 sqlite3_int64 szHeap;
3076
drh98d312f2012-10-25 15:23:14 +00003077 zSize = cmdline_option_value(argc, argv, ++i);
drh7d9f3942013-04-03 01:26:54 +00003078 szHeap = integerValue(zSize);
drh9c88d682010-12-17 14:03:01 +00003079 if( szHeap>0x7fff0000 ) szHeap = 0x7fff0000;
drh9c88d682010-12-17 14:03:01 +00003080 sqlite3_config(SQLITE_CONFIG_HEAP, malloc((int)szHeap), (int)szHeap, 64);
3081#endif
drh97ae8ff2011-03-16 16:56:29 +00003082#ifdef SQLITE_ENABLE_VFSTRACE
drhcc3b4f82012-02-07 14:13:50 +00003083 }else if( strcmp(z,"-vfstrace")==0 ){
drh97ae8ff2011-03-16 16:56:29 +00003084 extern int vfstrace_register(
3085 const char *zTraceName,
3086 const char *zOldVfsName,
3087 int (*xOut)(const char*,void*),
3088 void *pOutArg,
3089 int makeDefault
3090 );
drh2b625e22011-03-16 17:05:28 +00003091 vfstrace_register("trace",0,(int(*)(const char*,void*))fputs,stderr,1);
drh97ae8ff2011-03-16 16:56:29 +00003092#endif
drh6f25e892011-07-08 17:02:57 +00003093#ifdef SQLITE_ENABLE_MULTIPLEX
drhcc3b4f82012-02-07 14:13:50 +00003094 }else if( strcmp(z,"-multiplex")==0 ){
drh6f25e892011-07-08 17:02:57 +00003095 extern int sqlite3_multiple_initialize(const char*,int);
3096 sqlite3_multiplex_initialize(0, 1);
3097#endif
drh7d9f3942013-04-03 01:26:54 +00003098 }else if( strcmp(z,"-mmap")==0 ){
drh9b4c59f2013-04-15 17:03:42 +00003099 sqlite3_int64 sz = integerValue(cmdline_option_value(argc,argv,++i));
3100 sqlite3_config(SQLITE_CONFIG_MMAP_SIZE, sz, sz);
drhcc3b4f82012-02-07 14:13:50 +00003101 }else if( strcmp(z,"-vfs")==0 ){
drh98d312f2012-10-25 15:23:14 +00003102 sqlite3_vfs *pVfs = sqlite3_vfs_find(cmdline_option_value(argc,argv,++i));
drha7e61d82011-03-12 17:02:57 +00003103 if( pVfs ){
3104 sqlite3_vfs_register(pVfs, 1);
3105 }else{
3106 fprintf(stderr, "no such VFS: \"%s\"\n", argv[i]);
3107 exit(1);
3108 }
drh44c2eb12003-04-30 11:38:26 +00003109 }
3110 }
drh98d312f2012-10-25 15:23:14 +00003111 if( data.zDbFilename==0 ){
danielk197703aded42004-11-22 05:26:27 +00003112#ifndef SQLITE_OMIT_MEMORYDB
drh22fbcb82004-02-01 01:22:50 +00003113 data.zDbFilename = ":memory:";
danielk197703aded42004-11-22 05:26:27 +00003114#else
shane86f5bdb2009-10-24 02:00:07 +00003115 fprintf(stderr,"%s: Error: no database filename specified\n", Argv0);
3116 return 1;
drh01b41712005-08-29 23:06:23 +00003117#endif
drh98d312f2012-10-25 15:23:14 +00003118 }
3119 data.out = stdout;
drh01b41712005-08-29 23:06:23 +00003120
drh44c2eb12003-04-30 11:38:26 +00003121 /* Go ahead and open the database file if it already exists. If the
3122 ** file does not exist, delay opening it. This prevents empty database
3123 ** files from being created if a user mistypes the database name argument
3124 ** to the sqlite command-line tool.
3125 */
drhc8d74412004-08-31 23:41:26 +00003126 if( access(data.zDbFilename, 0)==0 ){
drh44c2eb12003-04-30 11:38:26 +00003127 open_db(&data);
3128 }
3129
drh22fbcb82004-02-01 01:22:50 +00003130 /* Process the initialization file if there is one. If no -init option
3131 ** is given on the command line, look for a file named ~/.sqliterc and
3132 ** try to process it.
drh44c2eb12003-04-30 11:38:26 +00003133 */
shane86f5bdb2009-10-24 02:00:07 +00003134 rc = process_sqliterc(&data,zInitFile);
3135 if( rc>0 ){
3136 return rc;
3137 }
drh44c2eb12003-04-30 11:38:26 +00003138
drh22fbcb82004-02-01 01:22:50 +00003139 /* Make a second pass through the command-line argument and set
3140 ** options. This second pass is delayed until after the initialization
3141 ** file is processed so that the command-line arguments will override
3142 ** settings in the initialization file.
drh44c2eb12003-04-30 11:38:26 +00003143 */
drh98d312f2012-10-25 15:23:14 +00003144 for(i=1; i<argc; i++){
drh22fbcb82004-02-01 01:22:50 +00003145 char *z = argv[i];
drh98d312f2012-10-25 15:23:14 +00003146 if( z[0]!='-' ) continue;
drhc28490c2006-10-26 14:25:58 +00003147 if( z[1]=='-' ){ z++; }
drh2e584cd2006-09-25 13:09:22 +00003148 if( strcmp(z,"-init")==0 ){
drh22fbcb82004-02-01 01:22:50 +00003149 i++;
3150 }else if( strcmp(z,"-html")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003151 data.mode = MODE_Html;
drh22fbcb82004-02-01 01:22:50 +00003152 }else if( strcmp(z,"-list")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003153 data.mode = MODE_List;
drh22fbcb82004-02-01 01:22:50 +00003154 }else if( strcmp(z,"-line")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003155 data.mode = MODE_Line;
drh22fbcb82004-02-01 01:22:50 +00003156 }else if( strcmp(z,"-column")==0 ){
drh8b32e172002-04-08 02:42:57 +00003157 data.mode = MODE_Column;
drhc49f44e2006-10-26 18:15:42 +00003158 }else if( strcmp(z,"-csv")==0 ){
3159 data.mode = MODE_Csv;
drh5bb3eb92007-05-04 13:15:55 +00003160 memcpy(data.separator,",",2);
drh22fbcb82004-02-01 01:22:50 +00003161 }else if( strcmp(z,"-separator")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003162 sqlite3_snprintf(sizeof(data.separator), data.separator,
drh98d312f2012-10-25 15:23:14 +00003163 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003164 }else if( strcmp(z,"-nullvalue")==0 ){
drh5bb3eb92007-05-04 13:15:55 +00003165 sqlite3_snprintf(sizeof(data.nullvalue), data.nullvalue,
drh98d312f2012-10-25 15:23:14 +00003166 "%s",cmdline_option_value(argc,argv,++i));
drh22fbcb82004-02-01 01:22:50 +00003167 }else if( strcmp(z,"-header")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003168 data.showHeader = 1;
drh22fbcb82004-02-01 01:22:50 +00003169 }else if( strcmp(z,"-noheader")==0 ){
drh1e5d0e92000-05-31 23:33:17 +00003170 data.showHeader = 0;
drh22fbcb82004-02-01 01:22:50 +00003171 }else if( strcmp(z,"-echo")==0 ){
drhdaffd0e2001-04-11 14:28:42 +00003172 data.echoOn = 1;
shaneh642d8b82010-07-28 16:05:34 +00003173 }else if( strcmp(z,"-stats")==0 ){
3174 data.statsOn = 1;
drhc49f44e2006-10-26 18:15:42 +00003175 }else if( strcmp(z,"-bail")==0 ){
3176 bail_on_error = 1;
drh22fbcb82004-02-01 01:22:50 +00003177 }else if( strcmp(z,"-version")==0 ){
drh9fd301b2011-06-03 13:28:22 +00003178 printf("%s %s\n", sqlite3_libversion(), sqlite3_sourceid());
drh151e3e12006-06-06 12:32:21 +00003179 return 0;
drhc28490c2006-10-26 14:25:58 +00003180 }else if( strcmp(z,"-interactive")==0 ){
3181 stdin_is_interactive = 1;
3182 }else if( strcmp(z,"-batch")==0 ){
3183 stdin_is_interactive = 0;
drh9c88d682010-12-17 14:03:01 +00003184 }else if( strcmp(z,"-heap")==0 ){
3185 i++;
drh7d9f3942013-04-03 01:26:54 +00003186 }else if( strcmp(z,"-mmap")==0 ){
3187 i++;
drha7e61d82011-03-12 17:02:57 +00003188 }else if( strcmp(z,"-vfs")==0 ){
3189 i++;
drh6f25e892011-07-08 17:02:57 +00003190#ifdef SQLITE_ENABLE_VFSTRACE
drh97ae8ff2011-03-16 16:56:29 +00003191 }else if( strcmp(z,"-vfstrace")==0 ){
3192 i++;
drh6f25e892011-07-08 17:02:57 +00003193#endif
3194#ifdef SQLITE_ENABLE_MULTIPLEX
3195 }else if( strcmp(z,"-multiplex")==0 ){
3196 i++;
3197#endif
drhcc3b4f82012-02-07 14:13:50 +00003198 }else if( strcmp(z,"-help")==0 ){
drhe1e38c42003-05-04 18:30:59 +00003199 usage(1);
drhcc3b4f82012-02-07 14:13:50 +00003200 }else if( strcmp(z,"-cmd")==0 ){
3201 if( i==argc-1 ) break;
drh98d312f2012-10-25 15:23:14 +00003202 z = cmdline_option_value(argc,argv,++i);
drhcc3b4f82012-02-07 14:13:50 +00003203 if( z[0]=='.' ){
3204 rc = do_meta_command(z, &data);
drh99b39082013-04-17 12:19:48 +00003205 if( rc && bail_on_error ) return rc==2 ? 0 : rc;
drhcc3b4f82012-02-07 14:13:50 +00003206 }else{
3207 open_db(&data);
3208 rc = shell_exec(data.db, z, shell_callback, &data, &zErrMsg);
3209 if( zErrMsg!=0 ){
3210 fprintf(stderr,"Error: %s\n", zErrMsg);
3211 if( bail_on_error ) return rc!=0 ? rc : 1;
3212 }else if( rc!=0 ){
3213 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", z);
3214 if( bail_on_error ) return rc;
3215 }
3216 }
drh1e5d0e92000-05-31 23:33:17 +00003217 }else{
shane86f5bdb2009-10-24 02:00:07 +00003218 fprintf(stderr,"%s: Error: unknown option: %s\n", Argv0, z);
drhe1e38c42003-05-04 18:30:59 +00003219 fprintf(stderr,"Use -help for a list of options.\n");
drh1e5d0e92000-05-31 23:33:17 +00003220 return 1;
3221 }
3222 }
drh44c2eb12003-04-30 11:38:26 +00003223
drh22fbcb82004-02-01 01:22:50 +00003224 if( zFirstCmd ){
drh44c2eb12003-04-30 11:38:26 +00003225 /* Run just the command that follows the database name
3226 */
drh22fbcb82004-02-01 01:22:50 +00003227 if( zFirstCmd[0]=='.' ){
shane916f9612009-10-23 00:37:15 +00003228 rc = do_meta_command(zFirstCmd, &data);
drh99b39082013-04-17 12:19:48 +00003229 if( rc==2 ) rc = 0;
drh6ff13852001-11-25 13:18:23 +00003230 }else{
drh44c2eb12003-04-30 11:38:26 +00003231 open_db(&data);
shane626a6e42009-10-22 17:30:15 +00003232 rc = shell_exec(data.db, zFirstCmd, shell_callback, &data, &zErrMsg);
shane86f5bdb2009-10-24 02:00:07 +00003233 if( zErrMsg!=0 ){
3234 fprintf(stderr,"Error: %s\n", zErrMsg);
3235 return rc!=0 ? rc : 1;
3236 }else if( rc!=0 ){
3237 fprintf(stderr,"Error: unable to process SQL \"%s\"\n", zFirstCmd);
3238 return rc;
drh6ff13852001-11-25 13:18:23 +00003239 }
drh75897232000-05-29 14:26:00 +00003240 }
3241 }else{
drh44c2eb12003-04-30 11:38:26 +00003242 /* Run commands received from standard input
3243 */
drhc28490c2006-10-26 14:25:58 +00003244 if( stdin_is_interactive ){
drh67505e72002-04-19 12:34:06 +00003245 char *zHome;
3246 char *zHistory = 0;
drh5bb3eb92007-05-04 13:15:55 +00003247 int nHistory;
drh75897232000-05-29 14:26:00 +00003248 printf(
drh743e0032011-12-12 16:51:50 +00003249 "SQLite version %s %.19s\n" /*extra-version-info*/
mihailim65df9db2008-06-28 11:29:22 +00003250 "Enter \".help\" for instructions\n"
3251 "Enter SQL statements terminated with a \";\"\n",
drh9fd301b2011-06-03 13:28:22 +00003252 sqlite3_libversion(), sqlite3_sourceid()
drh75897232000-05-29 14:26:00 +00003253 );
drh67505e72002-04-19 12:34:06 +00003254 zHome = find_home_dir();
drhea678832008-12-10 19:26:22 +00003255 if( zHome ){
drh4f21c4a2008-12-10 22:15:00 +00003256 nHistory = strlen30(zHome) + 20;
drhea678832008-12-10 19:26:22 +00003257 if( (zHistory = malloc(nHistory))!=0 ){
3258 sqlite3_snprintf(nHistory, zHistory,"%s/.sqlite_history", zHome);
3259 }
drh67505e72002-04-19 12:34:06 +00003260 }
danielk19774af00c62005-01-23 23:43:21 +00003261#if defined(HAVE_READLINE) && HAVE_READLINE==1
drh67505e72002-04-19 12:34:06 +00003262 if( zHistory ) read_history(zHistory);
danielk19774af00c62005-01-23 23:43:21 +00003263#endif
drhc28490c2006-10-26 14:25:58 +00003264 rc = process_input(&data, 0);
drh67505e72002-04-19 12:34:06 +00003265 if( zHistory ){
3266 stifle_history(100);
3267 write_history(zHistory);
adamd0a3daa32006-07-28 20:16:14 +00003268 free(zHistory);
drh67505e72002-04-19 12:34:06 +00003269 }
drhdaffd0e2001-04-11 14:28:42 +00003270 }else{
drhc28490c2006-10-26 14:25:58 +00003271 rc = process_input(&data, stdin);
drh75897232000-05-29 14:26:00 +00003272 }
3273 }
drh33048c02001-10-01 14:29:22 +00003274 set_table_name(&data, 0);
drh72af0772010-05-06 20:19:55 +00003275 if( data.db ){
drhe14cd932010-12-08 03:28:17 +00003276 sqlite3_close(data.db);
adamd0a3daa32006-07-28 20:16:14 +00003277 }
drhc28490c2006-10-26 14:25:58 +00003278 return rc;
drh75897232000-05-29 14:26:00 +00003279}