persistentstorage/sql/SQLite364/prepare.c
changeset 0 08ec8eefde2f
equal deleted inserted replaced
-1:000000000000 0:08ec8eefde2f
       
     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 ** $Id: prepare.c,v 1.97 2008/09/08 09:06:19 danielk1977 Exp $
       
    17 */
       
    18 #include "sqliteInt.h"
       
    19 #include <ctype.h>
       
    20 
       
    21 /*
       
    22 ** Fill the InitData structure with an error message that indicates
       
    23 ** that the database is corrupt.
       
    24 */
       
    25 static void corruptSchema(
       
    26   InitData *pData,     /* Initialization context */
       
    27   const char *zObj,    /* Object being parsed at the point of error */
       
    28   const char *zExtra   /* Error information */
       
    29 ){
       
    30   sqlite3 *db = pData->db;
       
    31   if( !db->mallocFailed && (db->flags & SQLITE_RecoveryMode)==0 ){
       
    32     if( zObj==0 ) zObj = "?";
       
    33     sqlite3SetString(pData->pzErrMsg, pData->db,
       
    34        "malformed database schema (%s)", zObj);
       
    35     if( zExtra && zExtra[0] ){
       
    36       *pData->pzErrMsg = sqlite3MAppendf(pData->db, *pData->pzErrMsg, "%s - %s",
       
    37                                   *pData->pzErrMsg, zExtra);
       
    38     }
       
    39   }
       
    40   pData->rc = SQLITE_CORRUPT;
       
    41 }
       
    42 
       
    43 /*
       
    44 ** This is the callback routine for the code that initializes the
       
    45 ** database.  See sqlite3Init() below for additional information.
       
    46 ** This routine is also called from the OP_ParseSchema opcode of the VDBE.
       
    47 **
       
    48 ** Each callback contains the following information:
       
    49 **
       
    50 **     argv[0] = name of thing being created
       
    51 **     argv[1] = root page number for table or index. 0 for trigger or view.
       
    52 **     argv[2] = SQL text for the CREATE statement.
       
    53 **
       
    54 */
       
    55 int sqlite3InitCallback(void *pInit, int argc, char **argv, char **azColName){
       
    56   InitData *pData = (InitData*)pInit;
       
    57   sqlite3 *db = pData->db;
       
    58   int iDb = pData->iDb;
       
    59 
       
    60   assert( sqlite3_mutex_held(db->mutex) );
       
    61   DbClearProperty(db, iDb, DB_Empty);
       
    62   if( db->mallocFailed ){
       
    63     corruptSchema(pData, argv[0], 0);
       
    64     return SQLITE_NOMEM;
       
    65   }
       
    66 
       
    67   assert( argc==3 );
       
    68   assert( iDb>=0 && iDb<db->nDb );
       
    69   if( argv==0 ) return 0;   /* Might happen if EMPTY_RESULT_CALLBACKS are on */
       
    70   if( argv[1]==0 ){
       
    71     corruptSchema(pData, argv[0], 0);
       
    72   }else if( argv[2] && argv[2][0] ){
       
    73     /* Call the parser to process a CREATE TABLE, INDEX or VIEW.
       
    74     ** But because db->init.busy is set to 1, no VDBE code is generated
       
    75     ** or executed.  All the parser does is build the internal data
       
    76     ** structures that describe the table, index, or view.
       
    77     */
       
    78     char *zErr;
       
    79     int rc;
       
    80     u8 lookasideEnabled;
       
    81     assert( db->init.busy );
       
    82     db->init.iDb = iDb;
       
    83     db->init.newTnum = atoi(argv[1]);
       
    84     lookasideEnabled = db->lookaside.bEnabled;
       
    85     db->lookaside.bEnabled = 0;
       
    86     rc = sqlite3_exec(db, argv[2], 0, 0, &zErr);
       
    87     db->init.iDb = 0;
       
    88     db->lookaside.bEnabled = lookasideEnabled;
       
    89     assert( rc!=SQLITE_OK || zErr==0 );
       
    90     if( SQLITE_OK!=rc ){
       
    91       pData->rc = rc;
       
    92       if( rc==SQLITE_NOMEM ){
       
    93         db->mallocFailed = 1;
       
    94       }else if( rc!=SQLITE_INTERRUPT ){
       
    95         corruptSchema(pData, argv[0], zErr);
       
    96       }
       
    97       sqlite3DbFree(db, zErr);
       
    98     }
       
    99   }else if( argv[0]==0 ){
       
   100     corruptSchema(pData, 0, 0);
       
   101   }else{
       
   102     /* If the SQL column is blank it means this is an index that
       
   103     ** was created to be the PRIMARY KEY or to fulfill a UNIQUE
       
   104     ** constraint for a CREATE TABLE.  The index should have already
       
   105     ** been created when we processed the CREATE TABLE.  All we have
       
   106     ** to do here is record the root page number for that index.
       
   107     */
       
   108     Index *pIndex;
       
   109     pIndex = sqlite3FindIndex(db, argv[0], db->aDb[iDb].zName);
       
   110     if( pIndex==0 || pIndex->tnum!=0 ){
       
   111       /* This can occur if there exists an index on a TEMP table which
       
   112       ** has the same name as another index on a permanent index.  Since
       
   113       ** the permanent table is hidden by the TEMP table, we can also
       
   114       ** safely ignore the index on the permanent table.
       
   115       */
       
   116       /* Do Nothing */;
       
   117     }else{
       
   118       pIndex->tnum = atoi(argv[1]);
       
   119     }
       
   120   }
       
   121   return 0;
       
   122 }
       
   123 
       
   124 /*
       
   125 ** Attempt to read the database schema and initialize internal
       
   126 ** data structures for a single database file.  The index of the
       
   127 ** database file is given by iDb.  iDb==0 is used for the main
       
   128 ** database.  iDb==1 should never be used.  iDb>=2 is used for
       
   129 ** auxiliary databases.  Return one of the SQLITE_ error codes to
       
   130 ** indicate success or failure.
       
   131 */
       
   132 static int sqlite3InitOne(sqlite3 *db, int iDb, char **pzErrMsg){
       
   133   int rc;
       
   134   BtCursor *curMain;
       
   135   int size;
       
   136   Table *pTab;
       
   137   Db *pDb;
       
   138   char const *azArg[4];
       
   139   int meta[10];
       
   140   InitData initData;
       
   141   char const *zMasterSchema;
       
   142   char const *zMasterName = SCHEMA_TABLE(iDb);
       
   143 
       
   144   /*
       
   145   ** The master database table has a structure like this
       
   146   */
       
   147   static const char master_schema[] = 
       
   148      "CREATE TABLE sqlite_master(\n"
       
   149      "  type text,\n"
       
   150      "  name text,\n"
       
   151      "  tbl_name text,\n"
       
   152      "  rootpage integer,\n"
       
   153      "  sql text\n"
       
   154      ")"
       
   155   ;
       
   156 #ifndef SQLITE_OMIT_TEMPDB
       
   157   static const char temp_master_schema[] = 
       
   158      "CREATE TEMP TABLE sqlite_temp_master(\n"
       
   159      "  type text,\n"
       
   160      "  name text,\n"
       
   161      "  tbl_name text,\n"
       
   162      "  rootpage integer,\n"
       
   163      "  sql text\n"
       
   164      ")"
       
   165   ;
       
   166 #else
       
   167   #define temp_master_schema 0
       
   168 #endif
       
   169 
       
   170   assert( iDb>=0 && iDb<db->nDb );
       
   171   assert( db->aDb[iDb].pSchema );
       
   172   assert( sqlite3_mutex_held(db->mutex) );
       
   173   assert( iDb==1 || sqlite3BtreeHoldsMutex(db->aDb[iDb].pBt) );
       
   174 
       
   175   /* zMasterSchema and zInitScript are set to point at the master schema
       
   176   ** and initialisation script appropriate for the database being
       
   177   ** initialised. zMasterName is the name of the master table.
       
   178   */
       
   179   if( !OMIT_TEMPDB && iDb==1 ){
       
   180     zMasterSchema = temp_master_schema;
       
   181   }else{
       
   182     zMasterSchema = master_schema;
       
   183   }
       
   184   zMasterName = SCHEMA_TABLE(iDb);
       
   185 
       
   186   /* Construct the schema tables.  */
       
   187   azArg[0] = zMasterName;
       
   188   azArg[1] = "1";
       
   189   azArg[2] = zMasterSchema;
       
   190   azArg[3] = 0;
       
   191   initData.db = db;
       
   192   initData.iDb = iDb;
       
   193   initData.rc = SQLITE_OK;
       
   194   initData.pzErrMsg = pzErrMsg;
       
   195   (void)sqlite3SafetyOff(db);
       
   196   sqlite3InitCallback(&initData, 3, (char **)azArg, 0);
       
   197   (void)sqlite3SafetyOn(db);
       
   198   if( initData.rc ){
       
   199     rc = initData.rc;
       
   200     goto error_out;
       
   201   }
       
   202   pTab = sqlite3FindTable(db, zMasterName, db->aDb[iDb].zName);
       
   203   if( pTab ){
       
   204     pTab->tabFlags |= TF_Readonly;
       
   205   }
       
   206 
       
   207   /* Create a cursor to hold the database open
       
   208   */
       
   209   pDb = &db->aDb[iDb];
       
   210   if( pDb->pBt==0 ){
       
   211     if( !OMIT_TEMPDB && iDb==1 ){
       
   212       DbSetProperty(db, 1, DB_SchemaLoaded);
       
   213     }
       
   214     return SQLITE_OK;
       
   215   }
       
   216   curMain = sqlite3MallocZero(sqlite3BtreeCursorSize());
       
   217   if( !curMain ){
       
   218     rc = SQLITE_NOMEM;
       
   219     goto error_out;
       
   220   }
       
   221   sqlite3BtreeEnter(pDb->pBt);
       
   222   rc = sqlite3BtreeCursor(pDb->pBt, MASTER_ROOT, 0, 0, curMain);
       
   223   if( rc!=SQLITE_OK && rc!=SQLITE_EMPTY ){
       
   224     sqlite3SetString(pzErrMsg, db, "%s", sqlite3ErrStr(rc));
       
   225     goto initone_error_out;
       
   226   }
       
   227 
       
   228   /* Get the database meta information.
       
   229   **
       
   230   ** Meta values are as follows:
       
   231   **    meta[0]   Schema cookie.  Changes with each schema change.
       
   232   **    meta[1]   File format of schema layer.
       
   233   **    meta[2]   Size of the page cache.
       
   234   **    meta[3]   Use freelist if 0.  Autovacuum if greater than zero.
       
   235   **    meta[4]   Db text encoding. 1:UTF-8 2:UTF-16LE 3:UTF-16BE
       
   236   **    meta[5]   The user cookie. Used by the application.
       
   237   **    meta[6]   Incremental-vacuum flag.
       
   238   **    meta[7]
       
   239   **    meta[8]
       
   240   **    meta[9]
       
   241   **
       
   242   ** Note: The #defined SQLITE_UTF* symbols in sqliteInt.h correspond to
       
   243   ** the possible values of meta[4].
       
   244   */
       
   245   if( rc==SQLITE_OK ){
       
   246     int i;
       
   247     for(i=0; i<sizeof(meta)/sizeof(meta[0]); i++){
       
   248       rc = sqlite3BtreeGetMeta(pDb->pBt, i+1, (u32 *)&meta[i]);
       
   249       if( rc ){
       
   250         sqlite3SetString(pzErrMsg, db, "%s", sqlite3ErrStr(rc));
       
   251         goto initone_error_out;
       
   252       }
       
   253     }
       
   254   }else{
       
   255     memset(meta, 0, sizeof(meta));
       
   256   }
       
   257   pDb->pSchema->schema_cookie = meta[0];
       
   258 
       
   259   /* If opening a non-empty database, check the text encoding. For the
       
   260   ** main database, set sqlite3.enc to the encoding of the main database.
       
   261   ** For an attached db, it is an error if the encoding is not the same
       
   262   ** as sqlite3.enc.
       
   263   */
       
   264   if( meta[4] ){  /* text encoding */
       
   265     if( iDb==0 ){
       
   266       /* If opening the main database, set ENC(db). */
       
   267       ENC(db) = (u8)meta[4];
       
   268       db->pDfltColl = sqlite3FindCollSeq(db, SQLITE_UTF8, "BINARY", 6, 0);
       
   269     }else{
       
   270       /* If opening an attached database, the encoding much match ENC(db) */
       
   271       if( meta[4]!=ENC(db) ){
       
   272         sqlite3SetString(pzErrMsg, db, "attached databases must use the same"
       
   273             " text encoding as main database");
       
   274         rc = SQLITE_ERROR;
       
   275         goto initone_error_out;
       
   276       }
       
   277     }
       
   278   }else{
       
   279     DbSetProperty(db, iDb, DB_Empty);
       
   280   }
       
   281   pDb->pSchema->enc = ENC(db);
       
   282 
       
   283   if( pDb->pSchema->cache_size==0 ){
       
   284     size = meta[2];
       
   285     if( size==0 ){ size = SQLITE_DEFAULT_CACHE_SIZE; }
       
   286     if( size<0 ) size = -size;
       
   287     pDb->pSchema->cache_size = size;
       
   288     sqlite3BtreeSetCacheSize(pDb->pBt, pDb->pSchema->cache_size);
       
   289   }
       
   290 
       
   291   /*
       
   292   ** file_format==1    Version 3.0.0.
       
   293   ** file_format==2    Version 3.1.3.  // ALTER TABLE ADD COLUMN
       
   294   ** file_format==3    Version 3.1.4.  // ditto but with non-NULL defaults
       
   295   ** file_format==4    Version 3.3.0.  // DESC indices.  Boolean constants
       
   296   */
       
   297   pDb->pSchema->file_format = meta[1];
       
   298   if( pDb->pSchema->file_format==0 ){
       
   299     pDb->pSchema->file_format = 1;
       
   300   }
       
   301   if( pDb->pSchema->file_format>SQLITE_MAX_FILE_FORMAT ){
       
   302     sqlite3SetString(pzErrMsg, db, "unsupported file format");
       
   303     rc = SQLITE_ERROR;
       
   304     goto initone_error_out;
       
   305   }
       
   306 
       
   307   /* Ticket #2804:  When we open a database in the newer file format,
       
   308   ** clear the legacy_file_format pragma flag so that a VACUUM will
       
   309   ** not downgrade the database and thus invalidate any descending
       
   310   ** indices that the user might have created.
       
   311   */
       
   312   if( iDb==0 && meta[1]>=4 ){
       
   313     db->flags &= ~SQLITE_LegacyFileFmt;
       
   314   }
       
   315 
       
   316   /* Read the schema information out of the schema tables
       
   317   */
       
   318   assert( db->init.busy );
       
   319   if( rc==SQLITE_EMPTY ){
       
   320     /* For an empty database, there is nothing to read */
       
   321     rc = SQLITE_OK;
       
   322   }else{
       
   323     char *zSql;
       
   324     zSql = sqlite3MPrintf(db, 
       
   325         "SELECT name, rootpage, sql FROM '%q'.%s",
       
   326         db->aDb[iDb].zName, zMasterName);
       
   327     (void)sqlite3SafetyOff(db);
       
   328 #ifndef SQLITE_OMIT_AUTHORIZATION
       
   329     {
       
   330       int (*xAuth)(void*,int,const char*,const char*,const char*,const char*);
       
   331       xAuth = db->xAuth;
       
   332       db->xAuth = 0;
       
   333 #endif
       
   334       rc = sqlite3_exec(db, zSql, sqlite3InitCallback, &initData, 0);
       
   335 #ifndef SQLITE_OMIT_AUTHORIZATION
       
   336       db->xAuth = xAuth;
       
   337     }
       
   338 #endif
       
   339     if( rc==SQLITE_OK ) rc = initData.rc;
       
   340     (void)sqlite3SafetyOn(db);
       
   341     sqlite3DbFree(db, zSql);
       
   342 #ifndef SQLITE_OMIT_ANALYZE
       
   343     if( rc==SQLITE_OK ){
       
   344       sqlite3AnalysisLoad(db, iDb);
       
   345     }
       
   346 #endif
       
   347   }
       
   348   if( db->mallocFailed ){
       
   349     rc = SQLITE_NOMEM;
       
   350     sqlite3ResetInternalSchema(db, 0);
       
   351   }
       
   352   if( rc==SQLITE_OK || (db->flags&SQLITE_RecoveryMode)){
       
   353     /* Black magic: If the SQLITE_RecoveryMode flag is set, then consider
       
   354     ** the schema loaded, even if errors occured. In this situation the 
       
   355     ** current sqlite3_prepare() operation will fail, but the following one
       
   356     ** will attempt to compile the supplied statement against whatever subset
       
   357     ** of the schema was loaded before the error occured. The primary
       
   358     ** purpose of this is to allow access to the sqlite_master table
       
   359     ** even when its contents have been corrupted.
       
   360     */
       
   361     DbSetProperty(db, iDb, DB_SchemaLoaded);
       
   362     rc = SQLITE_OK;
       
   363   }
       
   364 
       
   365   /* Jump here for an error that occurs after successfully allocating
       
   366   ** curMain and calling sqlite3BtreeEnter(). For an error that occurs
       
   367   ** before that point, jump to error_out.
       
   368   */
       
   369 initone_error_out:
       
   370   sqlite3BtreeCloseCursor(curMain);
       
   371   sqlite3_free(curMain);
       
   372   sqlite3BtreeLeave(pDb->pBt);
       
   373 
       
   374 error_out:
       
   375   if( rc==SQLITE_NOMEM || rc==SQLITE_IOERR_NOMEM ){
       
   376     db->mallocFailed = 1;
       
   377   }
       
   378   return rc;
       
   379 }
       
   380 
       
   381 /*
       
   382 ** Initialize all database files - the main database file, the file
       
   383 ** used to store temporary tables, and any additional database files
       
   384 ** created using ATTACH statements.  Return a success code.  If an
       
   385 ** error occurs, write an error message into *pzErrMsg.
       
   386 **
       
   387 ** After a database is initialized, the DB_SchemaLoaded bit is set
       
   388 ** bit is set in the flags field of the Db structure. If the database
       
   389 ** file was of zero-length, then the DB_Empty flag is also set.
       
   390 */
       
   391 int sqlite3Init(sqlite3 *db, char **pzErrMsg){
       
   392   int i, rc;
       
   393   int commit_internal = !(db->flags&SQLITE_InternChanges);
       
   394   
       
   395   assert( sqlite3_mutex_held(db->mutex) );
       
   396   if( db->init.busy ) return SQLITE_OK;
       
   397   rc = SQLITE_OK;
       
   398   db->init.busy = 1;
       
   399   for(i=0; rc==SQLITE_OK && i<db->nDb; i++){
       
   400     if( DbHasProperty(db, i, DB_SchemaLoaded) || i==1 ) continue;
       
   401     rc = sqlite3InitOne(db, i, pzErrMsg);
       
   402     if( rc ){
       
   403       sqlite3ResetInternalSchema(db, i);
       
   404     }
       
   405   }
       
   406 
       
   407   /* Once all the other databases have been initialised, load the schema
       
   408   ** for the TEMP database. This is loaded last, as the TEMP database
       
   409   ** schema may contain references to objects in other databases.
       
   410   */
       
   411 #ifndef SQLITE_OMIT_TEMPDB
       
   412   if( rc==SQLITE_OK && db->nDb>1 && !DbHasProperty(db, 1, DB_SchemaLoaded) ){
       
   413     rc = sqlite3InitOne(db, 1, pzErrMsg);
       
   414     if( rc ){
       
   415       sqlite3ResetInternalSchema(db, 1);
       
   416     }
       
   417   }
       
   418 #endif
       
   419 
       
   420   db->init.busy = 0;
       
   421   if( rc==SQLITE_OK && commit_internal ){
       
   422     sqlite3CommitInternalChanges(db);
       
   423   }
       
   424 
       
   425   return rc; 
       
   426 }
       
   427 
       
   428 /*
       
   429 ** This routine is a no-op if the database schema is already initialised.
       
   430 ** Otherwise, the schema is loaded. An error code is returned.
       
   431 */
       
   432 int sqlite3ReadSchema(Parse *pParse){
       
   433   int rc = SQLITE_OK;
       
   434   sqlite3 *db = pParse->db;
       
   435   assert( sqlite3_mutex_held(db->mutex) );
       
   436   if( !db->init.busy ){
       
   437     rc = sqlite3Init(db, &pParse->zErrMsg);
       
   438   }
       
   439   if( rc!=SQLITE_OK ){
       
   440     pParse->rc = rc;
       
   441     pParse->nErr++;
       
   442   }
       
   443   return rc;
       
   444 }
       
   445 
       
   446 
       
   447 /*
       
   448 ** Check schema cookies in all databases.  If any cookie is out
       
   449 ** of date, return 0.  If all schema cookies are current, return 1.
       
   450 */
       
   451 static int schemaIsValid(sqlite3 *db){
       
   452   int iDb;
       
   453   int rc;
       
   454   BtCursor *curTemp;
       
   455   int cookie;
       
   456   int allOk = 1;
       
   457 
       
   458   curTemp = (BtCursor *)sqlite3Malloc(sqlite3BtreeCursorSize());
       
   459   if( curTemp ){
       
   460     assert( sqlite3_mutex_held(db->mutex) );
       
   461     for(iDb=0; allOk && iDb<db->nDb; iDb++){
       
   462       Btree *pBt;
       
   463       pBt = db->aDb[iDb].pBt;
       
   464       if( pBt==0 ) continue;
       
   465       memset(curTemp, 0, sqlite3BtreeCursorSize());
       
   466       rc = sqlite3BtreeCursor(pBt, MASTER_ROOT, 0, 0, curTemp);
       
   467       if( rc==SQLITE_OK ){
       
   468         rc = sqlite3BtreeGetMeta(pBt, 1, (u32 *)&cookie);
       
   469         if( rc==SQLITE_OK && cookie!=db->aDb[iDb].pSchema->schema_cookie ){
       
   470           allOk = 0;
       
   471         }
       
   472         sqlite3BtreeCloseCursor(curTemp);
       
   473       }
       
   474       if( rc==SQLITE_NOMEM || rc==SQLITE_IOERR_NOMEM ){
       
   475         db->mallocFailed = 1;
       
   476       }
       
   477     }
       
   478     sqlite3_free(curTemp);
       
   479   }else{
       
   480     allOk = 0;
       
   481     db->mallocFailed = 1;
       
   482   }
       
   483 
       
   484   return allOk;
       
   485 }
       
   486 
       
   487 /*
       
   488 ** Convert a schema pointer into the iDb index that indicates
       
   489 ** which database file in db->aDb[] the schema refers to.
       
   490 **
       
   491 ** If the same database is attached more than once, the first
       
   492 ** attached database is returned.
       
   493 */
       
   494 int sqlite3SchemaToIndex(sqlite3 *db, Schema *pSchema){
       
   495   int i = -1000000;
       
   496 
       
   497   /* If pSchema is NULL, then return -1000000. This happens when code in 
       
   498   ** expr.c is trying to resolve a reference to a transient table (i.e. one
       
   499   ** created by a sub-select). In this case the return value of this 
       
   500   ** function should never be used.
       
   501   **
       
   502   ** We return -1000000 instead of the more usual -1 simply because using
       
   503   ** -1000000 as incorrectly using -1000000 index into db->aDb[] is much 
       
   504   ** more likely to cause a segfault than -1 (of course there are assert()
       
   505   ** statements too, but it never hurts to play the odds).
       
   506   */
       
   507   assert( sqlite3_mutex_held(db->mutex) );
       
   508   if( pSchema ){
       
   509     for(i=0; i<db->nDb; i++){
       
   510       if( db->aDb[i].pSchema==pSchema ){
       
   511         break;
       
   512       }
       
   513     }
       
   514     assert( i>=0 &&i>=0 &&  i<db->nDb );
       
   515   }
       
   516   return i;
       
   517 }
       
   518 
       
   519 /*
       
   520 ** Compile the UTF-8 encoded SQL statement zSql into a statement handle.
       
   521 */
       
   522 static int sqlite3Prepare(
       
   523   sqlite3 *db,              /* Database handle. */
       
   524   const char *zSql,         /* UTF-8 encoded SQL statement. */
       
   525   int nBytes,               /* Length of zSql in bytes. */
       
   526   int saveSqlFlag,          /* True to copy SQL text into the sqlite3_stmt */
       
   527   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   528   const char **pzTail       /* OUT: End of parsed string */
       
   529 ){
       
   530   Parse sParse;
       
   531   char *zErrMsg = 0;
       
   532   int rc = SQLITE_OK;
       
   533   int i;
       
   534 
       
   535   assert( ppStmt );
       
   536   *ppStmt = 0;
       
   537   if( sqlite3SafetyOn(db) ){
       
   538     return SQLITE_MISUSE;
       
   539   }
       
   540   assert( !db->mallocFailed );
       
   541   assert( sqlite3_mutex_held(db->mutex) );
       
   542 
       
   543   /* If any attached database schemas are locked, do not proceed with
       
   544   ** compilation. Instead return SQLITE_LOCKED immediately.
       
   545   */
       
   546   for(i=0; i<db->nDb; i++) {
       
   547     Btree *pBt = db->aDb[i].pBt;
       
   548     if( pBt ){
       
   549       int rc;
       
   550       rc = sqlite3BtreeSchemaLocked(pBt);
       
   551       if( rc ){
       
   552         const char *zDb = db->aDb[i].zName;
       
   553         sqlite3Error(db, SQLITE_LOCKED, "database schema is locked: %s", zDb);
       
   554         (void)sqlite3SafetyOff(db);
       
   555         return sqlite3ApiExit(db, SQLITE_LOCKED);
       
   556       }
       
   557     }
       
   558   }
       
   559   
       
   560   memset(&sParse, 0, sizeof(sParse));
       
   561   sParse.db = db;
       
   562   if( nBytes>=0 && (nBytes==0 || zSql[nBytes-1]!=0) ){
       
   563     char *zSqlCopy;
       
   564     int mxLen = db->aLimit[SQLITE_LIMIT_SQL_LENGTH];
       
   565     if( nBytes>mxLen ){
       
   566       sqlite3Error(db, SQLITE_TOOBIG, "statement too long");
       
   567       (void)sqlite3SafetyOff(db);
       
   568       return sqlite3ApiExit(db, SQLITE_TOOBIG);
       
   569     }
       
   570     zSqlCopy = sqlite3DbStrNDup(db, zSql, nBytes);
       
   571     if( zSqlCopy ){
       
   572       sqlite3RunParser(&sParse, zSqlCopy, &zErrMsg);
       
   573       sqlite3DbFree(db, zSqlCopy);
       
   574       sParse.zTail = &zSql[sParse.zTail-zSqlCopy];
       
   575     }else{
       
   576       sParse.zTail = &zSql[nBytes];
       
   577     }
       
   578   }else{
       
   579     sqlite3RunParser(&sParse, zSql, &zErrMsg);
       
   580   }
       
   581 
       
   582   if( db->mallocFailed ){
       
   583     sParse.rc = SQLITE_NOMEM;
       
   584   }
       
   585   if( sParse.rc==SQLITE_DONE ) sParse.rc = SQLITE_OK;
       
   586   if( sParse.checkSchema && !schemaIsValid(db) ){
       
   587     sParse.rc = SQLITE_SCHEMA;
       
   588   }
       
   589   if( sParse.rc==SQLITE_SCHEMA ){
       
   590     sqlite3ResetInternalSchema(db, 0);
       
   591   }
       
   592   if( db->mallocFailed ){
       
   593     sParse.rc = SQLITE_NOMEM;
       
   594   }
       
   595   if( pzTail ){
       
   596     *pzTail = sParse.zTail;
       
   597   }
       
   598   rc = sParse.rc;
       
   599 
       
   600 #ifndef SQLITE_OMIT_EXPLAIN
       
   601   if( rc==SQLITE_OK && sParse.pVdbe && sParse.explain ){
       
   602     if( sParse.explain==2 ){
       
   603       sqlite3VdbeSetNumCols(sParse.pVdbe, 3);
       
   604       sqlite3VdbeSetColName(sParse.pVdbe, 0, COLNAME_NAME, "order", P4_STATIC);
       
   605       sqlite3VdbeSetColName(sParse.pVdbe, 1, COLNAME_NAME, "from", P4_STATIC);
       
   606       sqlite3VdbeSetColName(sParse.pVdbe, 2, COLNAME_NAME, "detail", P4_STATIC);
       
   607     }else{
       
   608       sqlite3VdbeSetNumCols(sParse.pVdbe, 8);
       
   609       sqlite3VdbeSetColName(sParse.pVdbe, 0, COLNAME_NAME, "addr", P4_STATIC);
       
   610       sqlite3VdbeSetColName(sParse.pVdbe, 1, COLNAME_NAME, "opcode", P4_STATIC);
       
   611       sqlite3VdbeSetColName(sParse.pVdbe, 2, COLNAME_NAME, "p1", P4_STATIC);
       
   612       sqlite3VdbeSetColName(sParse.pVdbe, 3, COLNAME_NAME, "p2", P4_STATIC);
       
   613       sqlite3VdbeSetColName(sParse.pVdbe, 4, COLNAME_NAME, "p3", P4_STATIC);
       
   614       sqlite3VdbeSetColName(sParse.pVdbe, 5, COLNAME_NAME, "p4", P4_STATIC);
       
   615       sqlite3VdbeSetColName(sParse.pVdbe, 6, COLNAME_NAME, "p5", P4_STATIC);
       
   616       sqlite3VdbeSetColName(sParse.pVdbe, 7, COLNAME_NAME, "comment",P4_STATIC);
       
   617     }
       
   618   }
       
   619 #endif
       
   620 
       
   621   if( sqlite3SafetyOff(db) ){
       
   622     rc = SQLITE_MISUSE;
       
   623   }
       
   624 
       
   625   if( saveSqlFlag ){
       
   626     sqlite3VdbeSetSql(sParse.pVdbe, zSql, sParse.zTail - zSql);
       
   627   }
       
   628   if( rc!=SQLITE_OK || db->mallocFailed ){
       
   629     sqlite3_finalize((sqlite3_stmt*)sParse.pVdbe);
       
   630     assert(!(*ppStmt));
       
   631   }else{
       
   632     *ppStmt = (sqlite3_stmt*)sParse.pVdbe;
       
   633   }
       
   634 
       
   635   if( zErrMsg ){
       
   636     sqlite3Error(db, rc, "%s", zErrMsg);
       
   637     sqlite3DbFree(db, zErrMsg);
       
   638   }else{
       
   639     sqlite3Error(db, rc, 0);
       
   640   }
       
   641 
       
   642   rc = sqlite3ApiExit(db, rc);
       
   643   assert( (rc&db->errMask)==rc );
       
   644   return rc;
       
   645 }
       
   646 static int sqlite3LockAndPrepare(
       
   647   sqlite3 *db,              /* Database handle. */
       
   648   const char *zSql,         /* UTF-8 encoded SQL statement. */
       
   649   int nBytes,               /* Length of zSql in bytes. */
       
   650   int saveSqlFlag,          /* True to copy SQL text into the sqlite3_stmt */
       
   651   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   652   const char **pzTail       /* OUT: End of parsed string */
       
   653 ){
       
   654   int rc;
       
   655   if( !sqlite3SafetyCheckOk(db) ){
       
   656     return SQLITE_MISUSE;
       
   657   }
       
   658   sqlite3_mutex_enter(db->mutex);
       
   659   sqlite3BtreeEnterAll(db);
       
   660   rc = sqlite3Prepare(db, zSql, nBytes, saveSqlFlag, ppStmt, pzTail);
       
   661   sqlite3BtreeLeaveAll(db);
       
   662   sqlite3_mutex_leave(db->mutex);
       
   663   return rc;
       
   664 }
       
   665 
       
   666 /*
       
   667 ** Rerun the compilation of a statement after a schema change.
       
   668 ** Return true if the statement was recompiled successfully.
       
   669 ** Return false if there is an error of some kind.
       
   670 */
       
   671 int sqlite3Reprepare(Vdbe *p){
       
   672   int rc;
       
   673   sqlite3_stmt *pNew;
       
   674   const char *zSql;
       
   675   sqlite3 *db;
       
   676 
       
   677   assert( sqlite3_mutex_held(sqlite3VdbeDb(p)->mutex) );
       
   678   zSql = sqlite3_sql((sqlite3_stmt *)p);
       
   679   assert( zSql!=0 );  /* Reprepare only called for prepare_v2() statements */
       
   680   db = sqlite3VdbeDb(p);
       
   681   assert( sqlite3_mutex_held(db->mutex) );
       
   682   rc = sqlite3LockAndPrepare(db, zSql, -1, 0, &pNew, 0);
       
   683   if( rc ){
       
   684     if( rc==SQLITE_NOMEM ){
       
   685       db->mallocFailed = 1;
       
   686     }
       
   687     assert( pNew==0 );
       
   688     return 0;
       
   689   }else{
       
   690     assert( pNew!=0 );
       
   691   }
       
   692   sqlite3VdbeSwap((Vdbe*)pNew, p);
       
   693   sqlite3TransferBindings(pNew, (sqlite3_stmt*)p);
       
   694   sqlite3VdbeResetStepResult((Vdbe*)pNew);
       
   695   sqlite3VdbeFinalize((Vdbe*)pNew);
       
   696   return 1;
       
   697 }
       
   698 
       
   699 
       
   700 /*
       
   701 ** Two versions of the official API.  Legacy and new use.  In the legacy
       
   702 ** version, the original SQL text is not saved in the prepared statement
       
   703 ** and so if a schema change occurs, SQLITE_SCHEMA is returned by
       
   704 ** sqlite3_step().  In the new version, the original SQL text is retained
       
   705 ** and the statement is automatically recompiled if an schema change
       
   706 ** occurs.
       
   707 */
       
   708 int sqlite3_prepare(
       
   709   sqlite3 *db,              /* Database handle. */
       
   710   const char *zSql,         /* UTF-8 encoded SQL statement. */
       
   711   int nBytes,               /* Length of zSql in bytes. */
       
   712   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   713   const char **pzTail       /* OUT: End of parsed string */
       
   714 ){
       
   715   int rc;
       
   716   rc = sqlite3LockAndPrepare(db,zSql,nBytes,0,ppStmt,pzTail);
       
   717   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
       
   718   return rc;
       
   719 }
       
   720 int sqlite3_prepare_v2(
       
   721   sqlite3 *db,              /* Database handle. */
       
   722   const char *zSql,         /* UTF-8 encoded SQL statement. */
       
   723   int nBytes,               /* Length of zSql in bytes. */
       
   724   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   725   const char **pzTail       /* OUT: End of parsed string */
       
   726 ){
       
   727   int rc;
       
   728   rc = sqlite3LockAndPrepare(db,zSql,nBytes,1,ppStmt,pzTail);
       
   729   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
       
   730   return rc;
       
   731 }
       
   732 
       
   733 
       
   734 #ifndef SQLITE_OMIT_UTF16
       
   735 /*
       
   736 ** Compile the UTF-16 encoded SQL statement zSql into a statement handle.
       
   737 */
       
   738 static int sqlite3Prepare16(
       
   739   sqlite3 *db,              /* Database handle. */ 
       
   740   const void *zSql,         /* UTF-8 encoded SQL statement. */
       
   741   int nBytes,               /* Length of zSql in bytes. */
       
   742   int saveSqlFlag,          /* True to save SQL text into the sqlite3_stmt */
       
   743   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   744   const void **pzTail       /* OUT: End of parsed string */
       
   745 ){
       
   746   /* This function currently works by first transforming the UTF-16
       
   747   ** encoded string to UTF-8, then invoking sqlite3_prepare(). The
       
   748   ** tricky bit is figuring out the pointer to return in *pzTail.
       
   749   */
       
   750   char *zSql8;
       
   751   const char *zTail8 = 0;
       
   752   int rc = SQLITE_OK;
       
   753 
       
   754   if( !sqlite3SafetyCheckOk(db) ){
       
   755     return SQLITE_MISUSE;
       
   756   }
       
   757   sqlite3_mutex_enter(db->mutex);
       
   758   zSql8 = sqlite3Utf16to8(db, zSql, nBytes);
       
   759   if( zSql8 ){
       
   760     rc = sqlite3LockAndPrepare(db, zSql8, -1, saveSqlFlag, ppStmt, &zTail8);
       
   761   }
       
   762 
       
   763   if( zTail8 && pzTail ){
       
   764     /* If sqlite3_prepare returns a tail pointer, we calculate the
       
   765     ** equivalent pointer into the UTF-16 string by counting the unicode
       
   766     ** characters between zSql8 and zTail8, and then returning a pointer
       
   767     ** the same number of characters into the UTF-16 string.
       
   768     */
       
   769     int chars_parsed = sqlite3Utf8CharLen(zSql8, zTail8-zSql8);
       
   770     *pzTail = (u8 *)zSql + sqlite3Utf16ByteLen(zSql, chars_parsed);
       
   771   }
       
   772   sqlite3DbFree(db, zSql8); 
       
   773   rc = sqlite3ApiExit(db, rc);
       
   774   sqlite3_mutex_leave(db->mutex);
       
   775   return rc;
       
   776 }
       
   777 
       
   778 /*
       
   779 ** Two versions of the official API.  Legacy and new use.  In the legacy
       
   780 ** version, the original SQL text is not saved in the prepared statement
       
   781 ** and so if a schema change occurs, SQLITE_SCHEMA is returned by
       
   782 ** sqlite3_step().  In the new version, the original SQL text is retained
       
   783 ** and the statement is automatically recompiled if an schema change
       
   784 ** occurs.
       
   785 */
       
   786 int sqlite3_prepare16(
       
   787   sqlite3 *db,              /* Database handle. */ 
       
   788   const void *zSql,         /* UTF-8 encoded SQL statement. */
       
   789   int nBytes,               /* Length of zSql in bytes. */
       
   790   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   791   const void **pzTail       /* OUT: End of parsed string */
       
   792 ){
       
   793   int rc;
       
   794   rc = sqlite3Prepare16(db,zSql,nBytes,0,ppStmt,pzTail);
       
   795   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
       
   796   return rc;
       
   797 }
       
   798 int sqlite3_prepare16_v2(
       
   799   sqlite3 *db,              /* Database handle. */ 
       
   800   const void *zSql,         /* UTF-8 encoded SQL statement. */
       
   801   int nBytes,               /* Length of zSql in bytes. */
       
   802   sqlite3_stmt **ppStmt,    /* OUT: A pointer to the prepared statement */
       
   803   const void **pzTail       /* OUT: End of parsed string */
       
   804 ){
       
   805   int rc;
       
   806   rc = sqlite3Prepare16(db,zSql,nBytes,1,ppStmt,pzTail);
       
   807   assert( rc==SQLITE_OK || ppStmt==0 || *ppStmt==0 );  /* VERIFY: F13021 */
       
   808   return rc;
       
   809 }
       
   810 
       
   811 #endif /* SQLITE_OMIT_UTF16 */