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