xref: /sqlite-3.40.0/src/prepare.c (revision 7cff0e34)
1 /*
2 ** 2005 May 25
3 **
4 ** The author disclaims copyright to this source code.  In place of
5 ** a legal notice, here is a blessing:
6 **
7 **    May you do good and not evil.
8 **    May you find forgiveness for yourself and forgive others.
9 **    May you share freely, never taking more than you give.
10 **
11 *************************************************************************
12 ** This file contains the implementation of the sqlite3_prepare()
13 ** interface, and routines that contribute to loading the database schema
14 ** from disk.
15 */
16 #include "sqliteInt.h"
17 
18 /*
19 ** Fill the InitData structure with an error message that indicates
20 ** that the database is corrupt.
21 */
22 static void corruptSchema(
23   InitData *pData,     /* Initialization context */
24   const char *zObj,    /* Object being parsed at the point of error */
25   const char *zExtra   /* Error information */
26 ){
27   sqlite3 *db = pData->db;
28   if( db->mallocFailed ){
29     pData->rc = SQLITE_NOMEM_BKPT;
30   }else if( pData->pzErrMsg[0]!=0 ){
31     /* A error message has already been generated.  Do not overwrite it */
32   }else if( pData->mInitFlags & INITFLAG_AlterTable ){
33     *pData->pzErrMsg = sqlite3DbStrDup(db, zExtra);
34     pData->rc = SQLITE_ERROR;
35   }else if( db->flags & SQLITE_WriteSchema ){
36     pData->rc = SQLITE_CORRUPT_BKPT;
37   }else{
38     char *z;
39     if( zObj==0 ) zObj = "?";
40     z = sqlite3MPrintf(db, "malformed database schema (%s)", zObj);
41     if( zExtra && zExtra[0] ) z = sqlite3MPrintf(db, "%z - %s", z, zExtra);
42     *pData->pzErrMsg = z;
43     pData->rc = SQLITE_CORRUPT_BKPT;
44   }
45 }
46 
47 /*
48 ** Check to see if any sibling index (another index on the same table)
49 ** of pIndex has the same root page number, and if it does, return true.
50 ** This would indicate a corrupt schema.
51 */
52 int sqlite3IndexHasDuplicateRootPage(Index *pIndex){
53   Index *p;
54   for(p=pIndex->pTable->pIndex; p; p=p->pNext){
55     if( p->tnum==pIndex->tnum && p!=pIndex ) return 1;
56   }
57   return 0;
58 }
59 
60 /* forward declaration */
61 static int sqlite3Prepare(
62   sqlite3 *db,              /* Database handle. */
63   const char *zSql,         /* UTF-8 encoded SQL statement. */
64   int nBytes,               /* Length of zSql in bytes. */
65   u32 prepFlags,            /* Zero or more SQLITE_PREPARE_* flags */
66   Vdbe *pReprepare,         /* VM being reprepared */
67   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
68   const char **pzTail       /* OUT: End of parsed string */
69 );
70 
71 
72 /*
73 ** This is the callback routine for the code that initializes the
74 ** database.  See sqlite3Init() below for additional information.
75 ** This routine is also called from the OP_ParseSchema opcode of the VDBE.
76 **
77 ** Each callback contains the following information:
78 **
79 **     argv[0] = type of object: "table", "index", "trigger", or "view".
80 **     argv[1] = name of thing being created
81 **     argv[2] = associated table if an index or trigger
82 **     argv[3] = root page number for table or index. 0 for trigger or view.
83 **     argv[4] = SQL text for the CREATE statement.
84 **
85 */
86 int sqlite3InitCallback(void *pInit, int argc, char **argv, char **NotUsed){
87   InitData *pData = (InitData*)pInit;
88   sqlite3 *db = pData->db;
89   int iDb = pData->iDb;
90 
91   assert( argc==5 );
92   UNUSED_PARAMETER2(NotUsed, argc);
93   assert( sqlite3_mutex_held(db->mutex) );
94   db->mDbFlags |= DBFLAG_EncodingFixed;
95   pData->nInitRow++;
96   if( db->mallocFailed ){
97     corruptSchema(pData, argv[1], 0);
98     return 1;
99   }
100 
101   assert( iDb>=0 && iDb<db->nDb );
102   if( argv==0 ) return 0;   /* Might happen if EMPTY_RESULT_CALLBACKS are on */
103   if( argv[3]==0 ){
104     corruptSchema(pData, argv[1], 0);
105   }else if( sqlite3_strnicmp(argv[4],"create ",7)==0 ){
106     /* Call the parser to process a CREATE TABLE, INDEX or VIEW.
107     ** But because db->init.busy is set to 1, no VDBE code is generated
108     ** or executed.  All the parser does is build the internal data
109     ** structures that describe the table, index, or view.
110     */
111     int rc;
112     u8 saved_iDb = db->init.iDb;
113     sqlite3_stmt *pStmt;
114     TESTONLY(int rcp);            /* Return code from sqlite3_prepare() */
115 
116     assert( db->init.busy );
117     db->init.iDb = iDb;
118     db->init.newTnum = sqlite3Atoi(argv[3]);
119     db->init.orphanTrigger = 0;
120     db->init.azInit = argv;
121     pStmt = 0;
122     TESTONLY(rcp = ) sqlite3Prepare(db, argv[4], -1, 0, 0, &pStmt, 0);
123     rc = db->errCode;
124     assert( (rc&0xFF)==(rcp&0xFF) );
125     db->init.iDb = saved_iDb;
126     /* assert( saved_iDb==0 || (db->mDbFlags & DBFLAG_Vacuum)!=0 ); */
127     if( SQLITE_OK!=rc ){
128       if( db->init.orphanTrigger ){
129         assert( iDb==1 );
130       }else{
131         if( rc > pData->rc ) pData->rc = rc;
132         if( rc==SQLITE_NOMEM ){
133           sqlite3OomFault(db);
134         }else if( rc!=SQLITE_INTERRUPT && (rc&0xFF)!=SQLITE_LOCKED ){
135           corruptSchema(pData, argv[1], sqlite3_errmsg(db));
136         }
137       }
138     }
139     sqlite3_finalize(pStmt);
140   }else if( argv[1]==0 || (argv[4]!=0 && argv[4][0]!=0) ){
141     corruptSchema(pData, argv[1], 0);
142   }else{
143     /* If the SQL column is blank it means this is an index that
144     ** was created to be the PRIMARY KEY or to fulfill a UNIQUE
145     ** constraint for a CREATE TABLE.  The index should have already
146     ** been created when we processed the CREATE TABLE.  All we have
147     ** to do here is record the root page number for that index.
148     */
149     Index *pIndex;
150     pIndex = sqlite3FindIndex(db, argv[1], db->aDb[iDb].zDbSName);
151     if( pIndex==0
152      || sqlite3GetInt32(argv[3],&pIndex->tnum)==0
153      || pIndex->tnum<2
154      || sqlite3IndexHasDuplicateRootPage(pIndex)
155     ){
156       corruptSchema(pData, argv[1], pIndex?"invalid rootpage":"orphan index");
157     }
158   }
159   return 0;
160 }
161 
162 /*
163 ** Attempt to read the database schema and initialize internal
164 ** data structures for a single database file.  The index of the
165 ** database file is given by iDb.  iDb==0 is used for the main
166 ** database.  iDb==1 should never be used.  iDb>=2 is used for
167 ** auxiliary databases.  Return one of the SQLITE_ error codes to
168 ** indicate success or failure.
169 */
170 int sqlite3InitOne(sqlite3 *db, int iDb, char **pzErrMsg, u32 mFlags){
171   int rc;
172   int i;
173 #ifndef SQLITE_OMIT_DEPRECATED
174   int size;
175 #endif
176   Db *pDb;
177   char const *azArg[6];
178   int meta[5];
179   InitData initData;
180   const char *zSchemaTabName;
181   int openedTransaction = 0;
182   int mask = ((db->mDbFlags & DBFLAG_EncodingFixed) | ~DBFLAG_EncodingFixed);
183 
184   assert( (db->mDbFlags & DBFLAG_SchemaKnownOk)==0 );
185   assert( iDb>=0 && iDb<db->nDb );
186   assert( db->aDb[iDb].pSchema );
187   assert( sqlite3_mutex_held(db->mutex) );
188   assert( iDb==1 || sqlite3BtreeHoldsMutex(db->aDb[iDb].pBt) );
189 
190   db->init.busy = 1;
191 
192   /* Construct the in-memory representation schema tables (sqlite_schema or
193   ** sqlite_temp_schema) by invoking the parser directly.  The appropriate
194   ** table name will be inserted automatically by the parser so we can just
195   ** use the abbreviation "x" here.  The parser will also automatically tag
196   ** the schema table as read-only. */
197   azArg[0] = "table";
198   azArg[1] = zSchemaTabName = SCHEMA_TABLE(iDb);
199   azArg[2] = azArg[1];
200   azArg[3] = "1";
201   azArg[4] = "CREATE TABLE x(type text,name text,tbl_name text,"
202                             "rootpage int,sql text)";
203   azArg[5] = 0;
204   initData.db = db;
205   initData.iDb = iDb;
206   initData.rc = SQLITE_OK;
207   initData.pzErrMsg = pzErrMsg;
208   initData.mInitFlags = mFlags;
209   initData.nInitRow = 0;
210   sqlite3InitCallback(&initData, 5, (char **)azArg, 0);
211   db->mDbFlags &= mask;
212   if( initData.rc ){
213     rc = initData.rc;
214     goto error_out;
215   }
216 
217   /* Create a cursor to hold the database open
218   */
219   pDb = &db->aDb[iDb];
220   if( pDb->pBt==0 ){
221     assert( iDb==1 );
222     DbSetProperty(db, 1, DB_SchemaLoaded);
223     rc = SQLITE_OK;
224     goto error_out;
225   }
226 
227   /* If there is not already a read-only (or read-write) transaction opened
228   ** on the b-tree database, open one now. If a transaction is opened, it
229   ** will be closed before this function returns.  */
230   sqlite3BtreeEnter(pDb->pBt);
231   if( !sqlite3BtreeIsInReadTrans(pDb->pBt) ){
232     rc = sqlite3BtreeBeginTrans(pDb->pBt, 0, 0);
233     if( rc!=SQLITE_OK ){
234       sqlite3SetString(pzErrMsg, db, sqlite3ErrStr(rc));
235       goto initone_error_out;
236     }
237     openedTransaction = 1;
238   }
239 
240   /* Get the database meta information.
241   **
242   ** Meta values are as follows:
243   **    meta[0]   Schema cookie.  Changes with each schema change.
244   **    meta[1]   File format of schema layer.
245   **    meta[2]   Size of the page cache.
246   **    meta[3]   Largest rootpage (auto/incr_vacuum mode)
247   **    meta[4]   Db text encoding. 1:UTF-8 2:UTF-16LE 3:UTF-16BE
248   **    meta[5]   User version
249   **    meta[6]   Incremental vacuum mode
250   **    meta[7]   unused
251   **    meta[8]   unused
252   **    meta[9]   unused
253   **
254   ** Note: The #defined SQLITE_UTF* symbols in sqliteInt.h correspond to
255   ** the possible values of meta[4].
256   */
257   for(i=0; i<ArraySize(meta); i++){
258     sqlite3BtreeGetMeta(pDb->pBt, i+1, (u32 *)&meta[i]);
259   }
260   if( (db->flags & SQLITE_ResetDatabase)!=0 ){
261     memset(meta, 0, sizeof(meta));
262   }
263   pDb->pSchema->schema_cookie = meta[BTREE_SCHEMA_VERSION-1];
264 
265   /* If opening a non-empty database, check the text encoding. For the
266   ** main database, set sqlite3.enc to the encoding of the main database.
267   ** For an attached db, it is an error if the encoding is not the same
268   ** as sqlite3.enc.
269   */
270   if( meta[BTREE_TEXT_ENCODING-1] ){  /* text encoding */
271     if( iDb==0 && (db->mDbFlags & DBFLAG_EncodingFixed)==0 ){
272       u8 encoding;
273 #ifndef SQLITE_OMIT_UTF16
274       /* If opening the main database, set ENC(db). */
275       encoding = (u8)meta[BTREE_TEXT_ENCODING-1] & 3;
276       if( encoding==0 ) encoding = SQLITE_UTF8;
277 #else
278       encoding = SQLITE_UTF8;
279 #endif
280       sqlite3SetTextEncoding(db, encoding);
281     }else{
282       /* If opening an attached database, the encoding much match ENC(db) */
283       if( (meta[BTREE_TEXT_ENCODING-1] & 3)!=ENC(db) ){
284         sqlite3SetString(pzErrMsg, db, "attached databases must use the same"
285             " text encoding as main database");
286         rc = SQLITE_ERROR;
287         goto initone_error_out;
288       }
289     }
290   }
291   pDb->pSchema->enc = ENC(db);
292 
293   if( pDb->pSchema->cache_size==0 ){
294 #ifndef SQLITE_OMIT_DEPRECATED
295     size = sqlite3AbsInt32(meta[BTREE_DEFAULT_CACHE_SIZE-1]);
296     if( size==0 ){ size = SQLITE_DEFAULT_CACHE_SIZE; }
297     pDb->pSchema->cache_size = size;
298 #else
299     pDb->pSchema->cache_size = SQLITE_DEFAULT_CACHE_SIZE;
300 #endif
301     sqlite3BtreeSetCacheSize(pDb->pBt, pDb->pSchema->cache_size);
302   }
303 
304   /*
305   ** file_format==1    Version 3.0.0.
306   ** file_format==2    Version 3.1.3.  // ALTER TABLE ADD COLUMN
307   ** file_format==3    Version 3.1.4.  // ditto but with non-NULL defaults
308   ** file_format==4    Version 3.3.0.  // DESC indices.  Boolean constants
309   */
310   pDb->pSchema->file_format = (u8)meta[BTREE_FILE_FORMAT-1];
311   if( pDb->pSchema->file_format==0 ){
312     pDb->pSchema->file_format = 1;
313   }
314   if( pDb->pSchema->file_format>SQLITE_MAX_FILE_FORMAT ){
315     sqlite3SetString(pzErrMsg, db, "unsupported file format");
316     rc = SQLITE_ERROR;
317     goto initone_error_out;
318   }
319 
320   /* Ticket #2804:  When we open a database in the newer file format,
321   ** clear the legacy_file_format pragma flag so that a VACUUM will
322   ** not downgrade the database and thus invalidate any descending
323   ** indices that the user might have created.
324   */
325   if( iDb==0 && meta[BTREE_FILE_FORMAT-1]>=4 ){
326     db->flags &= ~(u64)SQLITE_LegacyFileFmt;
327   }
328 
329   /* Read the schema information out of the schema tables
330   */
331   assert( db->init.busy );
332   {
333     char *zSql;
334     zSql = sqlite3MPrintf(db,
335         "SELECT*FROM\"%w\".%s ORDER BY rowid",
336         db->aDb[iDb].zDbSName, zSchemaTabName);
337 #ifndef SQLITE_OMIT_AUTHORIZATION
338     {
339       sqlite3_xauth xAuth;
340       xAuth = db->xAuth;
341       db->xAuth = 0;
342 #endif
343       rc = sqlite3_exec(db, zSql, sqlite3InitCallback, &initData, 0);
344 #ifndef SQLITE_OMIT_AUTHORIZATION
345       db->xAuth = xAuth;
346     }
347 #endif
348     if( rc==SQLITE_OK ) rc = initData.rc;
349     sqlite3DbFree(db, zSql);
350 #ifndef SQLITE_OMIT_ANALYZE
351     if( rc==SQLITE_OK ){
352       sqlite3AnalysisLoad(db, iDb);
353     }
354 #endif
355   }
356   if( db->mallocFailed ){
357     rc = SQLITE_NOMEM_BKPT;
358     sqlite3ResetAllSchemasOfConnection(db);
359   }
360   if( rc==SQLITE_OK || (db->flags&SQLITE_NoSchemaError)){
361     /* Black magic: If the SQLITE_NoSchemaError flag is set, then consider
362     ** the schema loaded, even if errors occurred. In this situation the
363     ** current sqlite3_prepare() operation will fail, but the following one
364     ** will attempt to compile the supplied statement against whatever subset
365     ** of the schema was loaded before the error occurred. The primary
366     ** purpose of this is to allow access to the sqlite_schema table
367     ** even when its contents have been corrupted.
368     */
369     DbSetProperty(db, iDb, DB_SchemaLoaded);
370     rc = SQLITE_OK;
371   }
372 
373   /* Jump here for an error that occurs after successfully allocating
374   ** curMain and calling sqlite3BtreeEnter(). For an error that occurs
375   ** before that point, jump to error_out.
376   */
377 initone_error_out:
378   if( openedTransaction ){
379     sqlite3BtreeCommit(pDb->pBt);
380   }
381   sqlite3BtreeLeave(pDb->pBt);
382 
383 error_out:
384   if( rc ){
385     if( rc==SQLITE_NOMEM || rc==SQLITE_IOERR_NOMEM ){
386       sqlite3OomFault(db);
387     }
388     sqlite3ResetOneSchema(db, iDb);
389   }
390   db->init.busy = 0;
391   return rc;
392 }
393 
394 /*
395 ** Initialize all database files - the main database file, the file
396 ** used to store temporary tables, and any additional database files
397 ** created using ATTACH statements.  Return a success code.  If an
398 ** error occurs, write an error message into *pzErrMsg.
399 **
400 ** After a database is initialized, the DB_SchemaLoaded bit is set
401 ** bit is set in the flags field of the Db structure.
402 */
403 int sqlite3Init(sqlite3 *db, char **pzErrMsg){
404   int i, rc;
405   int commit_internal = !(db->mDbFlags&DBFLAG_SchemaChange);
406 
407   assert( sqlite3_mutex_held(db->mutex) );
408   assert( sqlite3BtreeHoldsMutex(db->aDb[0].pBt) );
409   assert( db->init.busy==0 );
410   ENC(db) = SCHEMA_ENC(db);
411   assert( db->nDb>0 );
412   /* Do the main schema first */
413   if( !DbHasProperty(db, 0, DB_SchemaLoaded) ){
414     rc = sqlite3InitOne(db, 0, pzErrMsg, 0);
415     if( rc ) return rc;
416   }
417   /* All other schemas after the main schema. The "temp" schema must be last */
418   for(i=db->nDb-1; i>0; i--){
419     assert( i==1 || sqlite3BtreeHoldsMutex(db->aDb[i].pBt) );
420     if( !DbHasProperty(db, i, DB_SchemaLoaded) ){
421       rc = sqlite3InitOne(db, i, pzErrMsg, 0);
422       if( rc ) return rc;
423     }
424   }
425   if( commit_internal ){
426     sqlite3CommitInternalChanges(db);
427   }
428   return SQLITE_OK;
429 }
430 
431 /*
432 ** This routine is a no-op if the database schema is already initialized.
433 ** Otherwise, the schema is loaded. An error code is returned.
434 */
435 int sqlite3ReadSchema(Parse *pParse){
436   int rc = SQLITE_OK;
437   sqlite3 *db = pParse->db;
438   assert( sqlite3_mutex_held(db->mutex) );
439   if( !db->init.busy ){
440     rc = sqlite3Init(db, &pParse->zErrMsg);
441     if( rc!=SQLITE_OK ){
442       pParse->rc = rc;
443       pParse->nErr++;
444     }else if( db->noSharedCache ){
445       db->mDbFlags |= DBFLAG_SchemaKnownOk;
446     }
447   }
448   return rc;
449 }
450 
451 
452 /*
453 ** Check schema cookies in all databases.  If any cookie is out
454 ** of date set pParse->rc to SQLITE_SCHEMA.  If all schema cookies
455 ** make no changes to pParse->rc.
456 */
457 static void schemaIsValid(Parse *pParse){
458   sqlite3 *db = pParse->db;
459   int iDb;
460   int rc;
461   int cookie;
462 
463   assert( pParse->checkSchema );
464   assert( sqlite3_mutex_held(db->mutex) );
465   for(iDb=0; iDb<db->nDb; iDb++){
466     int openedTransaction = 0;         /* True if a transaction is opened */
467     Btree *pBt = db->aDb[iDb].pBt;     /* Btree database to read cookie from */
468     if( pBt==0 ) continue;
469 
470     /* If there is not already a read-only (or read-write) transaction opened
471     ** on the b-tree database, open one now. If a transaction is opened, it
472     ** will be closed immediately after reading the meta-value. */
473     if( !sqlite3BtreeIsInReadTrans(pBt) ){
474       rc = sqlite3BtreeBeginTrans(pBt, 0, 0);
475       if( rc==SQLITE_NOMEM || rc==SQLITE_IOERR_NOMEM ){
476         sqlite3OomFault(db);
477       }
478       if( rc!=SQLITE_OK ) return;
479       openedTransaction = 1;
480     }
481 
482     /* Read the schema cookie from the database. If it does not match the
483     ** value stored as part of the in-memory schema representation,
484     ** set Parse.rc to SQLITE_SCHEMA. */
485     sqlite3BtreeGetMeta(pBt, BTREE_SCHEMA_VERSION, (u32 *)&cookie);
486     assert( sqlite3SchemaMutexHeld(db, iDb, 0) );
487     if( cookie!=db->aDb[iDb].pSchema->schema_cookie ){
488       sqlite3ResetOneSchema(db, iDb);
489       pParse->rc = SQLITE_SCHEMA;
490     }
491 
492     /* Close the transaction, if one was opened. */
493     if( openedTransaction ){
494       sqlite3BtreeCommit(pBt);
495     }
496   }
497 }
498 
499 /*
500 ** Convert a schema pointer into the iDb index that indicates
501 ** which database file in db->aDb[] the schema refers to.
502 **
503 ** If the same database is attached more than once, the first
504 ** attached database is returned.
505 */
506 int sqlite3SchemaToIndex(sqlite3 *db, Schema *pSchema){
507   int i = -32768;
508 
509   /* If pSchema is NULL, then return -32768. This happens when code in
510   ** expr.c is trying to resolve a reference to a transient table (i.e. one
511   ** created by a sub-select). In this case the return value of this
512   ** function should never be used.
513   **
514   ** We return -32768 instead of the more usual -1 simply because using
515   ** -32768 as the incorrect index into db->aDb[] is much
516   ** more likely to cause a segfault than -1 (of course there are assert()
517   ** statements too, but it never hurts to play the odds) and
518   ** -32768 will still fit into a 16-bit signed integer.
519   */
520   assert( sqlite3_mutex_held(db->mutex) );
521   if( pSchema ){
522     for(i=0; 1; i++){
523       assert( i<db->nDb );
524       if( db->aDb[i].pSchema==pSchema ){
525         break;
526       }
527     }
528     assert( i>=0 && i<db->nDb );
529   }
530   return i;
531 }
532 
533 /*
534 ** Deallocate a single AggInfo object
535 */
536 static void agginfoFree(sqlite3 *db, AggInfo *p){
537   sqlite3DbFree(db, p->aCol);
538   sqlite3DbFree(db, p->aFunc);
539   sqlite3DbFree(db, p);
540 }
541 
542 /*
543 ** Free all memory allocations in the pParse object
544 */
545 void sqlite3ParserReset(Parse *pParse){
546   sqlite3 *db = pParse->db;
547   AggInfo *pThis = pParse->pAggList;
548   while( pThis ){
549     AggInfo *pNext = pThis->pNext;
550     agginfoFree(db, pThis);
551     pThis = pNext;
552   }
553   sqlite3DbFree(db, pParse->aLabel);
554   sqlite3ExprListDelete(db, pParse->pConstExpr);
555   if( db ){
556     assert( db->lookaside.bDisable >= pParse->disableLookaside );
557     db->lookaside.bDisable -= pParse->disableLookaside;
558     db->lookaside.sz = db->lookaside.bDisable ? 0 : db->lookaside.szTrue;
559   }
560   pParse->disableLookaside = 0;
561 }
562 
563 /*
564 ** Compile the UTF-8 encoded SQL statement zSql into a statement handle.
565 */
566 static int sqlite3Prepare(
567   sqlite3 *db,              /* Database handle. */
568   const char *zSql,         /* UTF-8 encoded SQL statement. */
569   int nBytes,               /* Length of zSql in bytes. */
570   u32 prepFlags,            /* Zero or more SQLITE_PREPARE_* flags */
571   Vdbe *pReprepare,         /* VM being reprepared */
572   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
573   const char **pzTail       /* OUT: End of parsed string */
574 ){
575   char *zErrMsg = 0;        /* Error message */
576   int rc = SQLITE_OK;       /* Result code */
577   int i;                    /* Loop counter */
578   Parse sParse;             /* Parsing context */
579 
580   memset(&sParse, 0, PARSE_HDR_SZ);
581   memset(PARSE_TAIL(&sParse), 0, PARSE_TAIL_SZ);
582   sParse.pReprepare = pReprepare;
583   assert( ppStmt && *ppStmt==0 );
584   /* assert( !db->mallocFailed ); // not true with SQLITE_USE_ALLOCA */
585   assert( sqlite3_mutex_held(db->mutex) );
586 
587   /* For a long-term use prepared statement avoid the use of
588   ** lookaside memory.
589   */
590   if( prepFlags & SQLITE_PREPARE_PERSISTENT ){
591     sParse.disableLookaside++;
592     DisableLookaside;
593   }
594   sParse.disableVtab = (prepFlags & SQLITE_PREPARE_NO_VTAB)!=0;
595 
596   /* Check to verify that it is possible to get a read lock on all
597   ** database schemas.  The inability to get a read lock indicates that
598   ** some other database connection is holding a write-lock, which in
599   ** turn means that the other connection has made uncommitted changes
600   ** to the schema.
601   **
602   ** Were we to proceed and prepare the statement against the uncommitted
603   ** schema changes and if those schema changes are subsequently rolled
604   ** back and different changes are made in their place, then when this
605   ** prepared statement goes to run the schema cookie would fail to detect
606   ** the schema change.  Disaster would follow.
607   **
608   ** This thread is currently holding mutexes on all Btrees (because
609   ** of the sqlite3BtreeEnterAll() in sqlite3LockAndPrepare()) so it
610   ** is not possible for another thread to start a new schema change
611   ** while this routine is running.  Hence, we do not need to hold
612   ** locks on the schema, we just need to make sure nobody else is
613   ** holding them.
614   **
615   ** Note that setting READ_UNCOMMITTED overrides most lock detection,
616   ** but it does *not* override schema lock detection, so this all still
617   ** works even if READ_UNCOMMITTED is set.
618   */
619   if( !db->noSharedCache ){
620     for(i=0; i<db->nDb; i++) {
621       Btree *pBt = db->aDb[i].pBt;
622       if( pBt ){
623         assert( sqlite3BtreeHoldsMutex(pBt) );
624         rc = sqlite3BtreeSchemaLocked(pBt);
625         if( rc ){
626           const char *zDb = db->aDb[i].zDbSName;
627           sqlite3ErrorWithMsg(db, rc, "database schema is locked: %s", zDb);
628           testcase( db->flags & SQLITE_ReadUncommit );
629           goto end_prepare;
630         }
631       }
632     }
633   }
634 
635   sqlite3VtabUnlockList(db);
636 
637   sParse.db = db;
638   if( nBytes>=0 && (nBytes==0 || zSql[nBytes-1]!=0) ){
639     char *zSqlCopy;
640     int mxLen = db->aLimit[SQLITE_LIMIT_SQL_LENGTH];
641     testcase( nBytes==mxLen );
642     testcase( nBytes==mxLen+1 );
643     if( nBytes>mxLen ){
644       sqlite3ErrorWithMsg(db, SQLITE_TOOBIG, "statement too long");
645       rc = sqlite3ApiExit(db, SQLITE_TOOBIG);
646       goto end_prepare;
647     }
648     zSqlCopy = sqlite3DbStrNDup(db, zSql, nBytes);
649     if( zSqlCopy ){
650       sqlite3RunParser(&sParse, zSqlCopy, &zErrMsg);
651       sParse.zTail = &zSql[sParse.zTail-zSqlCopy];
652       sqlite3DbFree(db, zSqlCopy);
653     }else{
654       sParse.zTail = &zSql[nBytes];
655     }
656   }else{
657     sqlite3RunParser(&sParse, zSql, &zErrMsg);
658   }
659   assert( 0==sParse.nQueryLoop );
660 
661   if( sParse.rc==SQLITE_DONE ){
662     sParse.rc = SQLITE_OK;
663   }
664   if( sParse.checkSchema ){
665     schemaIsValid(&sParse);
666   }
667   if( pzTail ){
668     *pzTail = sParse.zTail;
669   }
670 
671   if( db->init.busy==0 ){
672     sqlite3VdbeSetSql(sParse.pVdbe, zSql, (int)(sParse.zTail-zSql), prepFlags);
673   }
674   if( db->mallocFailed ){
675     sParse.rc = SQLITE_NOMEM_BKPT;
676   }
677   rc = sParse.rc;
678   if( rc!=SQLITE_OK ){
679     if( sParse.pVdbe ) sqlite3VdbeFinalize(sParse.pVdbe);
680     assert(!(*ppStmt));
681   }else{
682     *ppStmt = (sqlite3_stmt*)sParse.pVdbe;
683   }
684 
685   if( zErrMsg ){
686     sqlite3ErrorWithMsg(db, rc, "%s", zErrMsg);
687     sqlite3DbFree(db, zErrMsg);
688   }else{
689     sqlite3Error(db, rc);
690   }
691 
692   /* Delete any TriggerPrg structures allocated while parsing this statement. */
693   while( sParse.pTriggerPrg ){
694     TriggerPrg *pT = sParse.pTriggerPrg;
695     sParse.pTriggerPrg = pT->pNext;
696     sqlite3DbFree(db, pT);
697   }
698 
699 end_prepare:
700 
701   sqlite3ParserReset(&sParse);
702   return rc;
703 }
704 static int sqlite3LockAndPrepare(
705   sqlite3 *db,              /* Database handle. */
706   const char *zSql,         /* UTF-8 encoded SQL statement. */
707   int nBytes,               /* Length of zSql in bytes. */
708   u32 prepFlags,            /* Zero or more SQLITE_PREPARE_* flags */
709   Vdbe *pOld,               /* VM being reprepared */
710   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
711   const char **pzTail       /* OUT: End of parsed string */
712 ){
713   int rc;
714   int cnt = 0;
715 
716 #ifdef SQLITE_ENABLE_API_ARMOR
717   if( ppStmt==0 ) return SQLITE_MISUSE_BKPT;
718 #endif
719   *ppStmt = 0;
720   if( !sqlite3SafetyCheckOk(db)||zSql==0 ){
721     return SQLITE_MISUSE_BKPT;
722   }
723   sqlite3_mutex_enter(db->mutex);
724   sqlite3BtreeEnterAll(db);
725   do{
726     /* Make multiple attempts to compile the SQL, until it either succeeds
727     ** or encounters a permanent error.  A schema problem after one schema
728     ** reset is considered a permanent error. */
729     rc = sqlite3Prepare(db, zSql, nBytes, prepFlags, pOld, ppStmt, pzTail);
730     assert( rc==SQLITE_OK || *ppStmt==0 );
731   }while( rc==SQLITE_ERROR_RETRY
732        || (rc==SQLITE_SCHEMA && (sqlite3ResetOneSchema(db,-1), cnt++)==0) );
733   sqlite3BtreeLeaveAll(db);
734   rc = sqlite3ApiExit(db, rc);
735   assert( (rc&db->errMask)==rc );
736   sqlite3_mutex_leave(db->mutex);
737   return rc;
738 }
739 
740 
741 /*
742 ** Rerun the compilation of a statement after a schema change.
743 **
744 ** If the statement is successfully recompiled, return SQLITE_OK. Otherwise,
745 ** if the statement cannot be recompiled because another connection has
746 ** locked the sqlite3_schema table, return SQLITE_LOCKED. If any other error
747 ** occurs, return SQLITE_SCHEMA.
748 */
749 int sqlite3Reprepare(Vdbe *p){
750   int rc;
751   sqlite3_stmt *pNew;
752   const char *zSql;
753   sqlite3 *db;
754   u8 prepFlags;
755 
756   assert( sqlite3_mutex_held(sqlite3VdbeDb(p)->mutex) );
757   zSql = sqlite3_sql((sqlite3_stmt *)p);
758   assert( zSql!=0 );  /* Reprepare only called for prepare_v2() statements */
759   db = sqlite3VdbeDb(p);
760   assert( sqlite3_mutex_held(db->mutex) );
761   prepFlags = sqlite3VdbePrepareFlags(p);
762   rc = sqlite3LockAndPrepare(db, zSql, -1, prepFlags, p, &pNew, 0);
763   if( rc ){
764     if( rc==SQLITE_NOMEM ){
765       sqlite3OomFault(db);
766     }
767     assert( pNew==0 );
768     return rc;
769   }else{
770     assert( pNew!=0 );
771   }
772   sqlite3VdbeSwap((Vdbe*)pNew, p);
773   sqlite3TransferBindings(pNew, (sqlite3_stmt*)p);
774   sqlite3VdbeResetStepResult((Vdbe*)pNew);
775   sqlite3VdbeFinalize((Vdbe*)pNew);
776   return SQLITE_OK;
777 }
778 
779 
780 /*
781 ** Two versions of the official API.  Legacy and new use.  In the legacy
782 ** version, the original SQL text is not saved in the prepared statement
783 ** and so if a schema change occurs, SQLITE_SCHEMA is returned by
784 ** sqlite3_step().  In the new version, the original SQL text is retained
785 ** and the statement is automatically recompiled if an schema change
786 ** occurs.
787 */
788 int sqlite3_prepare(
789   sqlite3 *db,              /* Database handle. */
790   const char *zSql,         /* UTF-8 encoded SQL statement. */
791   int nBytes,               /* Length of zSql in bytes. */
792   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
793   const char **pzTail       /* OUT: End of parsed string */
794 ){
795   int rc;
796   rc = sqlite3LockAndPrepare(db,zSql,nBytes,0,0,ppStmt,pzTail);
797   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
798   return rc;
799 }
800 int sqlite3_prepare_v2(
801   sqlite3 *db,              /* Database handle. */
802   const char *zSql,         /* UTF-8 encoded SQL statement. */
803   int nBytes,               /* Length of zSql in bytes. */
804   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
805   const char **pzTail       /* OUT: End of parsed string */
806 ){
807   int rc;
808   /* EVIDENCE-OF: R-37923-12173 The sqlite3_prepare_v2() interface works
809   ** exactly the same as sqlite3_prepare_v3() with a zero prepFlags
810   ** parameter.
811   **
812   ** Proof in that the 5th parameter to sqlite3LockAndPrepare is 0 */
813   rc = sqlite3LockAndPrepare(db,zSql,nBytes,SQLITE_PREPARE_SAVESQL,0,
814                              ppStmt,pzTail);
815   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );
816   return rc;
817 }
818 int sqlite3_prepare_v3(
819   sqlite3 *db,              /* Database handle. */
820   const char *zSql,         /* UTF-8 encoded SQL statement. */
821   int nBytes,               /* Length of zSql in bytes. */
822   unsigned int prepFlags,   /* Zero or more SQLITE_PREPARE_* flags */
823   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
824   const char **pzTail       /* OUT: End of parsed string */
825 ){
826   int rc;
827   /* EVIDENCE-OF: R-56861-42673 sqlite3_prepare_v3() differs from
828   ** sqlite3_prepare_v2() only in having the extra prepFlags parameter,
829   ** which is a bit array consisting of zero or more of the
830   ** SQLITE_PREPARE_* flags.
831   **
832   ** Proof by comparison to the implementation of sqlite3_prepare_v2()
833   ** directly above. */
834   rc = sqlite3LockAndPrepare(db,zSql,nBytes,
835                  SQLITE_PREPARE_SAVESQL|(prepFlags&SQLITE_PREPARE_MASK),
836                  0,ppStmt,pzTail);
837   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );
838   return rc;
839 }
840 
841 
842 #ifndef SQLITE_OMIT_UTF16
843 /*
844 ** Compile the UTF-16 encoded SQL statement zSql into a statement handle.
845 */
846 static int sqlite3Prepare16(
847   sqlite3 *db,              /* Database handle. */
848   const void *zSql,         /* UTF-16 encoded SQL statement. */
849   int nBytes,               /* Length of zSql in bytes. */
850   u32 prepFlags,            /* Zero or more SQLITE_PREPARE_* flags */
851   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
852   const void **pzTail       /* OUT: End of parsed string */
853 ){
854   /* This function currently works by first transforming the UTF-16
855   ** encoded string to UTF-8, then invoking sqlite3_prepare(). The
856   ** tricky bit is figuring out the pointer to return in *pzTail.
857   */
858   char *zSql8;
859   const char *zTail8 = 0;
860   int rc = SQLITE_OK;
861 
862 #ifdef SQLITE_ENABLE_API_ARMOR
863   if( ppStmt==0 ) return SQLITE_MISUSE_BKPT;
864 #endif
865   *ppStmt = 0;
866   if( !sqlite3SafetyCheckOk(db)||zSql==0 ){
867     return SQLITE_MISUSE_BKPT;
868   }
869   if( nBytes>=0 ){
870     int sz;
871     const char *z = (const char*)zSql;
872     for(sz=0; sz<nBytes && (z[sz]!=0 || z[sz+1]!=0); sz += 2){}
873     nBytes = sz;
874   }
875   sqlite3_mutex_enter(db->mutex);
876   zSql8 = sqlite3Utf16to8(db, zSql, nBytes, SQLITE_UTF16NATIVE);
877   if( zSql8 ){
878     rc = sqlite3LockAndPrepare(db, zSql8, -1, prepFlags, 0, ppStmt, &zTail8);
879   }
880 
881   if( zTail8 && pzTail ){
882     /* If sqlite3_prepare returns a tail pointer, we calculate the
883     ** equivalent pointer into the UTF-16 string by counting the unicode
884     ** characters between zSql8 and zTail8, and then returning a pointer
885     ** the same number of characters into the UTF-16 string.
886     */
887     int chars_parsed = sqlite3Utf8CharLen(zSql8, (int)(zTail8-zSql8));
888     *pzTail = (u8 *)zSql + sqlite3Utf16ByteLen(zSql, chars_parsed);
889   }
890   sqlite3DbFree(db, zSql8);
891   rc = sqlite3ApiExit(db, rc);
892   sqlite3_mutex_leave(db->mutex);
893   return rc;
894 }
895 
896 /*
897 ** Two versions of the official API.  Legacy and new use.  In the legacy
898 ** version, the original SQL text is not saved in the prepared statement
899 ** and so if a schema change occurs, SQLITE_SCHEMA is returned by
900 ** sqlite3_step().  In the new version, the original SQL text is retained
901 ** and the statement is automatically recompiled if an schema change
902 ** occurs.
903 */
904 int sqlite3_prepare16(
905   sqlite3 *db,              /* Database handle. */
906   const void *zSql,         /* UTF-16 encoded SQL statement. */
907   int nBytes,               /* Length of zSql in bytes. */
908   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
909   const void **pzTail       /* OUT: End of parsed string */
910 ){
911   int rc;
912   rc = sqlite3Prepare16(db,zSql,nBytes,0,ppStmt,pzTail);
913   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
914   return rc;
915 }
916 int sqlite3_prepare16_v2(
917   sqlite3 *db,              /* Database handle. */
918   const void *zSql,         /* UTF-16 encoded SQL statement. */
919   int nBytes,               /* Length of zSql in bytes. */
920   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
921   const void **pzTail       /* OUT: End of parsed string */
922 ){
923   int rc;
924   rc = sqlite3Prepare16(db,zSql,nBytes,SQLITE_PREPARE_SAVESQL,ppStmt,pzTail);
925   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
926   return rc;
927 }
928 int sqlite3_prepare16_v3(
929   sqlite3 *db,              /* Database handle. */
930   const void *zSql,         /* UTF-16 encoded SQL statement. */
931   int nBytes,               /* Length of zSql in bytes. */
932   unsigned int prepFlags,   /* Zero or more SQLITE_PREPARE_* flags */
933   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
934   const void **pzTail       /* OUT: End of parsed string */
935 ){
936   int rc;
937   rc = sqlite3Prepare16(db,zSql,nBytes,
938          SQLITE_PREPARE_SAVESQL|(prepFlags&SQLITE_PREPARE_MASK),
939          ppStmt,pzTail);
940   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
941   return rc;
942 }
943 
944 #endif /* SQLITE_OMIT_UTF16 */
945