blob: 19f09dc8d8296ab2380f6383043c53c697d14bbb [file] [log] [blame]
drh75897232000-05-29 14:26:00 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drh75897232000-05-29 14:26:00 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drh75897232000-05-29 14:26:00 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drh75897232000-05-29 14:26:00 +000010**
11*************************************************************************
drhb19a2bc2001-09-16 00:13:26 +000012** This file contains C code routines that are called by the SQLite parser
13** when syntax rules are reduced. The routines in this file handle the
14** following kinds of SQL syntax:
drh75897232000-05-29 14:26:00 +000015**
drhbed86902000-06-02 13:27:59 +000016** CREATE TABLE
17** DROP TABLE
18** CREATE INDEX
19** DROP INDEX
drh832508b2002-03-02 17:04:07 +000020** creating ID lists
drhb19a2bc2001-09-16 00:13:26 +000021** BEGIN TRANSACTION
22** COMMIT
23** ROLLBACK
drhbed86902000-06-02 13:27:59 +000024**
drh633e6d52008-07-28 19:34:53 +000025** $Id: build.c,v 1.491 2008/07/28 19:34:53 drh Exp $
drh75897232000-05-29 14:26:00 +000026*/
27#include "sqliteInt.h"
drhf57b14a2001-09-14 18:54:08 +000028#include <ctype.h>
drh75897232000-05-29 14:26:00 +000029
30/*
drhe0bc4042002-06-25 01:09:11 +000031** This routine is called when a new SQL statement is beginning to
drh23bf66d2004-12-14 03:34:34 +000032** be parsed. Initialize the pParse structure as needed.
drhe0bc4042002-06-25 01:09:11 +000033*/
danielk19774adee202004-05-08 08:23:19 +000034void sqlite3BeginParse(Parse *pParse, int explainFlag){
drhe0bc4042002-06-25 01:09:11 +000035 pParse->explain = explainFlag;
drh7c972de2003-09-06 22:18:07 +000036 pParse->nVar = 0;
drhe0bc4042002-06-25 01:09:11 +000037}
38
danielk1977c00da102006-01-07 13:21:04 +000039#ifndef SQLITE_OMIT_SHARED_CACHE
40/*
41** The TableLock structure is only used by the sqlite3TableLock() and
42** codeTableLocks() functions.
43*/
44struct TableLock {
drhd698bc12006-03-23 23:33:26 +000045 int iDb; /* The database containing the table to be locked */
46 int iTab; /* The root page of the table to be locked */
47 u8 isWriteLock; /* True for write lock. False for a read lock */
48 const char *zName; /* Name of the table */
danielk1977c00da102006-01-07 13:21:04 +000049};
50
51/*
drhd698bc12006-03-23 23:33:26 +000052** Record the fact that we want to lock a table at run-time.
danielk1977c00da102006-01-07 13:21:04 +000053**
drhd698bc12006-03-23 23:33:26 +000054** The table to be locked has root page iTab and is found in database iDb.
55** A read or a write lock can be taken depending on isWritelock.
56**
57** This routine just records the fact that the lock is desired. The
58** code to make the lock occur is generated by a later call to
59** codeTableLocks() which occurs during sqlite3FinishCoding().
danielk1977c00da102006-01-07 13:21:04 +000060*/
61void sqlite3TableLock(
drhd698bc12006-03-23 23:33:26 +000062 Parse *pParse, /* Parsing context */
63 int iDb, /* Index of the database containing the table to lock */
64 int iTab, /* Root page number of the table to be locked */
65 u8 isWriteLock, /* True for a write lock */
66 const char *zName /* Name of the table to be locked */
danielk1977c00da102006-01-07 13:21:04 +000067){
68 int i;
69 int nBytes;
70 TableLock *p;
danielk1977c00da102006-01-07 13:21:04 +000071
drhe53831d2007-08-17 01:14:38 +000072 if( iDb<0 ){
danielk1977c00da102006-01-07 13:21:04 +000073 return;
74 }
75
76 for(i=0; i<pParse->nTableLock; i++){
77 p = &pParse->aTableLock[i];
78 if( p->iDb==iDb && p->iTab==iTab ){
79 p->isWriteLock = (p->isWriteLock || isWriteLock);
80 return;
81 }
82 }
83
84 nBytes = sizeof(TableLock) * (pParse->nTableLock+1);
drh17435752007-08-16 04:30:38 +000085 pParse->aTableLock =
86 sqlite3DbReallocOrFree(pParse->db, pParse->aTableLock, nBytes);
danielk1977c00da102006-01-07 13:21:04 +000087 if( pParse->aTableLock ){
88 p = &pParse->aTableLock[pParse->nTableLock++];
89 p->iDb = iDb;
90 p->iTab = iTab;
91 p->isWriteLock = isWriteLock;
92 p->zName = zName;
drhf3a65f72007-08-22 20:18:21 +000093 }else{
94 pParse->nTableLock = 0;
95 pParse->db->mallocFailed = 1;
danielk1977c00da102006-01-07 13:21:04 +000096 }
97}
98
99/*
100** Code an OP_TableLock instruction for each table locked by the
101** statement (configured by calls to sqlite3TableLock()).
102*/
103static void codeTableLocks(Parse *pParse){
104 int i;
105 Vdbe *pVdbe;
danielk1977c00da102006-01-07 13:21:04 +0000106
107 if( 0==(pVdbe = sqlite3GetVdbe(pParse)) ){
108 return;
109 }
110
111 for(i=0; i<pParse->nTableLock; i++){
112 TableLock *p = &pParse->aTableLock[i];
113 int p1 = p->iDb;
drh6a9ad3d2008-04-02 16:29:30 +0000114 sqlite3VdbeAddOp4(pVdbe, OP_TableLock, p1, p->iTab, p->isWriteLock,
115 p->zName, P4_STATIC);
danielk1977c00da102006-01-07 13:21:04 +0000116 }
117}
118#else
119 #define codeTableLocks(x)
120#endif
121
drhe0bc4042002-06-25 01:09:11 +0000122/*
drh75897232000-05-29 14:26:00 +0000123** This routine is called after a single SQL statement has been
drh80242052004-06-09 00:48:12 +0000124** parsed and a VDBE program to execute that statement has been
125** prepared. This routine puts the finishing touches on the
126** VDBE program and resets the pParse structure for the next
127** parse.
drh75897232000-05-29 14:26:00 +0000128**
129** Note that if an error occurred, it might be the case that
130** no VDBE code was generated.
131*/
drh80242052004-06-09 00:48:12 +0000132void sqlite3FinishCoding(Parse *pParse){
drh9bb575f2004-09-06 17:24:11 +0000133 sqlite3 *db;
drh80242052004-06-09 00:48:12 +0000134 Vdbe *v;
drhb86ccfb2003-01-28 23:13:10 +0000135
drh17435752007-08-16 04:30:38 +0000136 db = pParse->db;
137 if( db->mallocFailed ) return;
drh205f48e2004-11-05 00:43:11 +0000138 if( pParse->nested ) return;
drhc4dd3fd2008-01-22 01:48:05 +0000139 if( pParse->nErr ) return;
danielk1977441daf62005-02-01 03:46:43 +0000140 if( !pParse->pVdbe ){
danielk1977c30f9e72005-02-09 07:05:46 +0000141 if( pParse->rc==SQLITE_OK && pParse->nErr ){
142 pParse->rc = SQLITE_ERROR;
drhe134ff12006-02-18 16:36:45 +0000143 return;
danielk1977c30f9e72005-02-09 07:05:46 +0000144 }
danielk1977441daf62005-02-01 03:46:43 +0000145 }
danielk197748d0d862005-02-01 03:09:52 +0000146
drh80242052004-06-09 00:48:12 +0000147 /* Begin by generating some termination code at the end of the
148 ** vdbe program
149 */
drh80242052004-06-09 00:48:12 +0000150 v = sqlite3GetVdbe(pParse);
151 if( v ){
drh66a51672008-01-03 00:01:23 +0000152 sqlite3VdbeAddOp0(v, OP_Halt);
drh0e3d7472004-06-19 17:33:07 +0000153
154 /* The cookie mask contains one bit for each database file open.
155 ** (Bit 0 is for main, bit 1 is for temp, and so forth.) Bits are
156 ** set for each database that is used. Generate code to start a
157 ** transaction on each used database and to verify the schema cookie
158 ** on each used database.
159 */
drhc275b4e2004-07-19 17:25:24 +0000160 if( pParse->cookieGoto>0 ){
drh80242052004-06-09 00:48:12 +0000161 u32 mask;
drh26e4a8b2008-05-01 17:16:52 +0000162 int iDb;
drhd654be82005-09-20 17:42:23 +0000163 sqlite3VdbeJumpHere(v, pParse->cookieGoto-1);
drh80242052004-06-09 00:48:12 +0000164 for(iDb=0, mask=1; iDb<db->nDb; mask<<=1, iDb++){
165 if( (mask & pParse->cookieMask)==0 ) continue;
drhfb982642007-08-30 01:19:59 +0000166 sqlite3VdbeUsesBtree(v, iDb);
drh66a51672008-01-03 00:01:23 +0000167 sqlite3VdbeAddOp2(v,OP_Transaction, iDb, (mask & pParse->writeMask)!=0);
168 sqlite3VdbeAddOp2(v,OP_VerifyCookie, iDb, pParse->cookieValue[iDb]);
drh80242052004-06-09 00:48:12 +0000169 }
danielk1977f9e7dda2006-06-16 16:08:53 +0000170#ifndef SQLITE_OMIT_VIRTUALTABLE
drh26e4a8b2008-05-01 17:16:52 +0000171 {
172 int i;
173 for(i=0; i<pParse->nVtabLock; i++){
174 char *vtab = (char *)pParse->apVtabLock[i]->pVtab;
175 sqlite3VdbeAddOp4(v, OP_VBegin, 0, 0, 0, vtab, P4_VTAB);
176 }
177 pParse->nVtabLock = 0;
danielk1977f9e7dda2006-06-16 16:08:53 +0000178 }
179#endif
danielk1977c00da102006-01-07 13:21:04 +0000180
181 /* Once all the cookies have been verified and transactions opened,
182 ** obtain the required table-locks. This is a no-op unless the
183 ** shared-cache feature is enabled.
184 */
185 codeTableLocks(pParse);
drh66a51672008-01-03 00:01:23 +0000186 sqlite3VdbeAddOp2(v, OP_Goto, 0, pParse->cookieGoto);
drh80242052004-06-09 00:48:12 +0000187 }
drh80242052004-06-09 00:48:12 +0000188
drh19e2d372005-08-29 23:00:03 +0000189#ifndef SQLITE_OMIT_TRACE
drh949f9cd2008-01-12 21:35:57 +0000190 if( !db->init.busy ){
191 /* Change the P4 argument of the first opcode (which will always be
192 ** an OP_Trace) to be the complete text of the current SQL statement.
193 */
194 VdbeOp *pOp = sqlite3VdbeGetOp(v, 0);
195 if( pOp && pOp->opcode==OP_Trace ){
196 sqlite3VdbeChangeP4(v, 0, pParse->zSql, pParse->zTail-pParse->zSql);
197 }
198 }
drh19e2d372005-08-29 23:00:03 +0000199#endif /* SQLITE_OMIT_TRACE */
drh71c697e2004-08-08 23:39:19 +0000200 }
201
drh3f7d4e42004-07-24 14:35:58 +0000202
drh80242052004-06-09 00:48:12 +0000203 /* Get the VDBE program ready for execution
204 */
drh17435752007-08-16 04:30:38 +0000205 if( v && pParse->nErr==0 && !db->mallocFailed ){
drhcf1023c2007-05-08 20:59:49 +0000206#ifdef SQLITE_DEBUG
drhb86ccfb2003-01-28 23:13:10 +0000207 FILE *trace = (db->flags & SQLITE_VdbeTrace)!=0 ? stdout : 0;
danielk19774adee202004-05-08 08:23:19 +0000208 sqlite3VdbeTrace(v, trace);
drhcf1023c2007-05-08 20:59:49 +0000209#endif
drh2f7794c2008-04-01 03:27:39 +0000210 assert( pParse->disableColCache==0 ); /* Disables and re-enables match */
drh290c1942004-08-21 17:54:45 +0000211 sqlite3VdbeMakeReady(v, pParse->nVar, pParse->nMem+3,
drh13449892005-09-07 21:22:45 +0000212 pParse->nTab+3, pParse->explain);
danielk1977441daf62005-02-01 03:46:43 +0000213 pParse->rc = SQLITE_DONE;
drhd8bc7082000-06-07 23:51:50 +0000214 pParse->colNamesSet = 0;
drh826fb5a2004-02-14 23:59:57 +0000215 }else if( pParse->rc==SQLITE_OK ){
drh483750b2003-01-29 18:46:51 +0000216 pParse->rc = SQLITE_ERROR;
drh75897232000-05-29 14:26:00 +0000217 }
drha226d052002-09-25 19:04:07 +0000218 pParse->nTab = 0;
219 pParse->nMem = 0;
220 pParse->nSet = 0;
drh7c972de2003-09-06 22:18:07 +0000221 pParse->nVar = 0;
drh80242052004-06-09 00:48:12 +0000222 pParse->cookieMask = 0;
drhc275b4e2004-07-19 17:25:24 +0000223 pParse->cookieGoto = 0;
drh75897232000-05-29 14:26:00 +0000224}
225
226/*
drh205f48e2004-11-05 00:43:11 +0000227** Run the parser and code generator recursively in order to generate
228** code for the SQL statement given onto the end of the pParse context
229** currently under construction. When the parser is run recursively
230** this way, the final OP_Halt is not appended and other initialization
231** and finalization steps are omitted because those are handling by the
232** outermost parser.
233**
234** Not everything is nestable. This facility is designed to permit
235** INSERT, UPDATE, and DELETE operations against SQLITE_MASTER. Use
drhf1974842004-11-05 03:56:00 +0000236** care if you decide to try to use this routine for some other purposes.
drh205f48e2004-11-05 00:43:11 +0000237*/
238void sqlite3NestedParse(Parse *pParse, const char *zFormat, ...){
239 va_list ap;
240 char *zSql;
drhfb45d8c2008-07-08 00:06:49 +0000241 char *zErrMsg = 0;
drh633e6d52008-07-28 19:34:53 +0000242 sqlite3 *db = pParse->db;
drhf1974842004-11-05 03:56:00 +0000243# define SAVE_SZ (sizeof(Parse) - offsetof(Parse,nVar))
244 char saveBuf[SAVE_SZ];
245
drh205f48e2004-11-05 00:43:11 +0000246 if( pParse->nErr ) return;
247 assert( pParse->nested<10 ); /* Nesting should only be of limited depth */
248 va_start(ap, zFormat);
drh633e6d52008-07-28 19:34:53 +0000249 zSql = sqlite3VMPrintf(db, zFormat, ap);
drh205f48e2004-11-05 00:43:11 +0000250 va_end(ap);
drh73c42a12004-11-20 18:13:10 +0000251 if( zSql==0 ){
252 return; /* A malloc must have failed */
253 }
drh205f48e2004-11-05 00:43:11 +0000254 pParse->nested++;
drhf1974842004-11-05 03:56:00 +0000255 memcpy(saveBuf, &pParse->nVar, SAVE_SZ);
256 memset(&pParse->nVar, 0, SAVE_SZ);
drhfb45d8c2008-07-08 00:06:49 +0000257 sqlite3RunParser(pParse, zSql, &zErrMsg);
drh633e6d52008-07-28 19:34:53 +0000258 sqlite3DbFree(db, zErrMsg);
259 sqlite3DbFree(db, zSql);
drhf1974842004-11-05 03:56:00 +0000260 memcpy(&pParse->nVar, saveBuf, SAVE_SZ);
drh205f48e2004-11-05 00:43:11 +0000261 pParse->nested--;
262}
263
264/*
danielk19778a414492004-06-29 08:59:35 +0000265** Locate the in-memory structure that describes a particular database
266** table given the name of that table and (optionally) the name of the
267** database containing the table. Return NULL if not found.
drha69d9162003-04-17 22:57:53 +0000268**
danielk19778a414492004-06-29 08:59:35 +0000269** If zDatabase is 0, all databases are searched for the table and the
270** first matching table is returned. (No checking for duplicate table
271** names is done.) The search order is TEMP first, then MAIN, then any
272** auxiliary databases added using the ATTACH command.
drhf26e09c2003-05-31 16:21:12 +0000273**
danielk19774adee202004-05-08 08:23:19 +0000274** See also sqlite3LocateTable().
drh75897232000-05-29 14:26:00 +0000275*/
drh9bb575f2004-09-06 17:24:11 +0000276Table *sqlite3FindTable(sqlite3 *db, const char *zName, const char *zDatabase){
drhd24cc422003-03-27 12:51:24 +0000277 Table *p = 0;
278 int i;
drh0a687d12008-07-08 14:52:07 +0000279 int nName;
drh645f63e2004-06-22 13:22:40 +0000280 assert( zName!=0 );
drh0a687d12008-07-08 14:52:07 +0000281 nName = sqlite3Strlen(db, zName) + 1;
danielk197753c0f742005-03-29 03:10:59 +0000282 for(i=OMIT_TEMPDB; i<db->nDb; i++){
drh812d7a22003-03-27 13:50:00 +0000283 int j = (i<2) ? i^1 : i; /* Search TEMP before MAIN */
danielk19774adee202004-05-08 08:23:19 +0000284 if( zDatabase!=0 && sqlite3StrICmp(zDatabase, db->aDb[j].zName) ) continue;
drh0a687d12008-07-08 14:52:07 +0000285 p = sqlite3HashFind(&db->aDb[j].pSchema->tblHash, zName, nName);
drhd24cc422003-03-27 12:51:24 +0000286 if( p ) break;
287 }
drh74e24cd2002-01-09 03:19:59 +0000288 return p;
drh75897232000-05-29 14:26:00 +0000289}
290
291/*
danielk19778a414492004-06-29 08:59:35 +0000292** Locate the in-memory structure that describes a particular database
293** table given the name of that table and (optionally) the name of the
294** database containing the table. Return NULL if not found. Also leave an
295** error message in pParse->zErrMsg.
drha69d9162003-04-17 22:57:53 +0000296**
danielk19778a414492004-06-29 08:59:35 +0000297** The difference between this routine and sqlite3FindTable() is that this
298** routine leaves an error message in pParse->zErrMsg where
299** sqlite3FindTable() does not.
drha69d9162003-04-17 22:57:53 +0000300*/
drhca424112008-01-25 15:04:48 +0000301Table *sqlite3LocateTable(
302 Parse *pParse, /* context in which to report errors */
303 int isView, /* True if looking for a VIEW rather than a TABLE */
304 const char *zName, /* Name of the table we are looking for */
305 const char *zDbase /* Name of the database. Might be NULL */
306){
drha69d9162003-04-17 22:57:53 +0000307 Table *p;
drhf26e09c2003-05-31 16:21:12 +0000308
danielk19778a414492004-06-29 08:59:35 +0000309 /* Read the database schema. If an error occurs, leave an error message
310 ** and code in pParse and return NULL. */
311 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){
312 return 0;
313 }
314
danielk19774adee202004-05-08 08:23:19 +0000315 p = sqlite3FindTable(pParse->db, zName, zDbase);
drha69d9162003-04-17 22:57:53 +0000316 if( p==0 ){
drhca424112008-01-25 15:04:48 +0000317 const char *zMsg = isView ? "no such view" : "no such table";
danielk19778a414492004-06-29 08:59:35 +0000318 if( zDbase ){
drhca424112008-01-25 15:04:48 +0000319 sqlite3ErrorMsg(pParse, "%s: %s.%s", zMsg, zDbase, zName);
drha69d9162003-04-17 22:57:53 +0000320 }else{
drhca424112008-01-25 15:04:48 +0000321 sqlite3ErrorMsg(pParse, "%s: %s", zMsg, zName);
drha69d9162003-04-17 22:57:53 +0000322 }
drha6ecd332004-06-10 00:29:09 +0000323 pParse->checkSchema = 1;
drha69d9162003-04-17 22:57:53 +0000324 }
325 return p;
326}
327
328/*
329** Locate the in-memory structure that describes
330** a particular index given the name of that index
331** and the name of the database that contains the index.
drhf57b3392001-10-08 13:22:32 +0000332** Return NULL if not found.
drhf26e09c2003-05-31 16:21:12 +0000333**
334** If zDatabase is 0, all databases are searched for the
335** table and the first matching index is returned. (No checking
336** for duplicate index names is done.) The search order is
337** TEMP first, then MAIN, then any auxiliary databases added
338** using the ATTACH command.
drh75897232000-05-29 14:26:00 +0000339*/
drh9bb575f2004-09-06 17:24:11 +0000340Index *sqlite3FindIndex(sqlite3 *db, const char *zName, const char *zDb){
drhd24cc422003-03-27 12:51:24 +0000341 Index *p = 0;
342 int i;
drh0a687d12008-07-08 14:52:07 +0000343 int nName = sqlite3Strlen(db, zName)+1;
danielk197753c0f742005-03-29 03:10:59 +0000344 for(i=OMIT_TEMPDB; i<db->nDb; i++){
drh812d7a22003-03-27 13:50:00 +0000345 int j = (i<2) ? i^1 : i; /* Search TEMP before MAIN */
danielk1977e501b892006-01-09 06:29:47 +0000346 Schema *pSchema = db->aDb[j].pSchema;
danielk19774adee202004-05-08 08:23:19 +0000347 if( zDb && sqlite3StrICmp(zDb, db->aDb[j].zName) ) continue;
danielk1977da184232006-01-05 11:34:32 +0000348 assert( pSchema || (j==1 && !db->aDb[1].pBt) );
349 if( pSchema ){
drh0a687d12008-07-08 14:52:07 +0000350 p = sqlite3HashFind(&pSchema->idxHash, zName, nName);
danielk1977da184232006-01-05 11:34:32 +0000351 }
drhd24cc422003-03-27 12:51:24 +0000352 if( p ) break;
353 }
drh74e24cd2002-01-09 03:19:59 +0000354 return p;
drh75897232000-05-29 14:26:00 +0000355}
356
357/*
drh956bc922004-07-24 17:38:29 +0000358** Reclaim the memory used by an index
359*/
360static void freeIndex(Index *p){
drh633e6d52008-07-28 19:34:53 +0000361 sqlite3 *db = p->pTable->db;
362 sqlite3DbFree(db, p->zColAff);
363 sqlite3DbFree(db, p);
drh956bc922004-07-24 17:38:29 +0000364}
365
366/*
drh75897232000-05-29 14:26:00 +0000367** Remove the given index from the index hash table, and free
368** its memory structures.
369**
drhd229ca92002-01-09 13:30:41 +0000370** The index is removed from the database hash tables but
371** it is not unlinked from the Table that it indexes.
drhdaffd0e2001-04-11 14:28:42 +0000372** Unlinking from the Table must be done by the calling function.
drh75897232000-05-29 14:26:00 +0000373*/
drh74161702006-02-24 02:53:49 +0000374static void sqliteDeleteIndex(Index *p){
drhd229ca92002-01-09 13:30:41 +0000375 Index *pOld;
danielk1977da184232006-01-05 11:34:32 +0000376 const char *zName = p->zName;
drhd24cc422003-03-27 12:51:24 +0000377
drh0a687d12008-07-08 14:52:07 +0000378 pOld = sqlite3HashInsert(&p->pSchema->idxHash, zName, strlen(zName)+1, 0);
drh85c23c62005-08-20 03:03:04 +0000379 assert( pOld==0 || pOld==p );
drh956bc922004-07-24 17:38:29 +0000380 freeIndex(p);
drh75897232000-05-29 14:26:00 +0000381}
382
383/*
drhc96d8532005-05-03 12:30:33 +0000384** For the index called zIdxName which is found in the database iDb,
385** unlike that index from its Table then remove the index from
386** the index hash table and free all memory structures associated
387** with the index.
drh5e00f6c2001-09-13 13:46:56 +0000388*/
drh9bb575f2004-09-06 17:24:11 +0000389void sqlite3UnlinkAndDeleteIndex(sqlite3 *db, int iDb, const char *zIdxName){
drh956bc922004-07-24 17:38:29 +0000390 Index *pIndex;
391 int len;
danielk1977da184232006-01-05 11:34:32 +0000392 Hash *pHash = &db->aDb[iDb].pSchema->idxHash;
drh956bc922004-07-24 17:38:29 +0000393
drh0a687d12008-07-08 14:52:07 +0000394 len = sqlite3Strlen(db, zIdxName);
danielk1977da184232006-01-05 11:34:32 +0000395 pIndex = sqlite3HashInsert(pHash, zIdxName, len+1, 0);
drh956bc922004-07-24 17:38:29 +0000396 if( pIndex ){
397 if( pIndex->pTable->pIndex==pIndex ){
398 pIndex->pTable->pIndex = pIndex->pNext;
399 }else{
400 Index *p;
401 for(p=pIndex->pTable->pIndex; p && p->pNext!=pIndex; p=p->pNext){}
402 if( p && p->pNext==pIndex ){
403 p->pNext = pIndex->pNext;
404 }
drh5e00f6c2001-09-13 13:46:56 +0000405 }
drh956bc922004-07-24 17:38:29 +0000406 freeIndex(pIndex);
drh5e00f6c2001-09-13 13:46:56 +0000407 }
drh956bc922004-07-24 17:38:29 +0000408 db->flags |= SQLITE_InternChanges;
drh5e00f6c2001-09-13 13:46:56 +0000409}
410
411/*
drhe0bc4042002-06-25 01:09:11 +0000412** Erase all schema information from the in-memory hash tables of
drh234c39d2004-07-24 03:30:47 +0000413** a single database. This routine is called to reclaim memory
414** before the database closes. It is also called during a rollback
danielk1977e0d4b062004-06-28 01:11:46 +0000415** if there were schema changes during the transaction or if a
416** schema-cookie mismatch occurs.
drh1c2d8412003-03-31 00:30:47 +0000417**
418** If iDb<=0 then reset the internal schema tables for all database
419** files. If iDb>=2 then reset the internal schema for only the
jplyoncfa56842004-01-19 04:55:56 +0000420** single file indicated.
drh74e24cd2002-01-09 03:19:59 +0000421*/
drh9bb575f2004-09-06 17:24:11 +0000422void sqlite3ResetInternalSchema(sqlite3 *db, int iDb){
drh1c2d8412003-03-31 00:30:47 +0000423 int i, j;
drh1c2d8412003-03-31 00:30:47 +0000424 assert( iDb>=0 && iDb<db->nDb );
danielk19774eab8b72007-12-27 15:12:16 +0000425
426 if( iDb==0 ){
427 sqlite3BtreeEnterAll(db);
428 }
drh1c2d8412003-03-31 00:30:47 +0000429 for(i=iDb; i<db->nDb; i++){
drhd24cc422003-03-27 12:51:24 +0000430 Db *pDb = &db->aDb[i];
danielk1977da184232006-01-05 11:34:32 +0000431 if( pDb->pSchema ){
danielk19774eab8b72007-12-27 15:12:16 +0000432 assert(i==1 || (pDb->pBt && sqlite3BtreeHoldsMutex(pDb->pBt)));
danielk1977de0fe3e2006-01-06 06:33:12 +0000433 sqlite3SchemaFree(pDb->pSchema);
drhd24cc422003-03-27 12:51:24 +0000434 }
drh1c2d8412003-03-31 00:30:47 +0000435 if( iDb>0 ) return;
drh74e24cd2002-01-09 03:19:59 +0000436 }
drh1c2d8412003-03-31 00:30:47 +0000437 assert( iDb==0 );
438 db->flags &= ~SQLITE_InternChanges;
danielk19774eab8b72007-12-27 15:12:16 +0000439 sqlite3BtreeLeaveAll(db);
drh1c2d8412003-03-31 00:30:47 +0000440
441 /* If one or more of the auxiliary database files has been closed,
danielk1977311019b2006-01-10 07:14:23 +0000442 ** then remove them from the auxiliary database list. We take the
drh1c2d8412003-03-31 00:30:47 +0000443 ** opportunity to do this here since we have just deleted all of the
444 ** schema hash tables and therefore do not have to make any changes
445 ** to any of those tables.
446 */
drh4d189ca2004-02-12 18:46:38 +0000447 for(i=0; i<db->nDb; i++){
448 struct Db *pDb = &db->aDb[i];
449 if( pDb->pBt==0 ){
450 if( pDb->pAux && pDb->xFreeAux ) pDb->xFreeAux(pDb->pAux);
451 pDb->pAux = 0;
452 }
453 }
drh1c2d8412003-03-31 00:30:47 +0000454 for(i=j=2; i<db->nDb; i++){
drh4d189ca2004-02-12 18:46:38 +0000455 struct Db *pDb = &db->aDb[i];
456 if( pDb->pBt==0 ){
drh633e6d52008-07-28 19:34:53 +0000457 sqlite3DbFree(db, pDb->zName);
drh4d189ca2004-02-12 18:46:38 +0000458 pDb->zName = 0;
drh1c2d8412003-03-31 00:30:47 +0000459 continue;
460 }
461 if( j<i ){
drh8bf8dc92003-05-17 17:35:10 +0000462 db->aDb[j] = db->aDb[i];
drh1c2d8412003-03-31 00:30:47 +0000463 }
drh8bf8dc92003-05-17 17:35:10 +0000464 j++;
drh1c2d8412003-03-31 00:30:47 +0000465 }
466 memset(&db->aDb[j], 0, (db->nDb-j)*sizeof(db->aDb[j]));
467 db->nDb = j;
468 if( db->nDb<=2 && db->aDb!=db->aDbStatic ){
469 memcpy(db->aDbStatic, db->aDb, 2*sizeof(db->aDb[0]));
drh633e6d52008-07-28 19:34:53 +0000470 sqlite3DbFree(db, db->aDb);
drh1c2d8412003-03-31 00:30:47 +0000471 db->aDb = db->aDbStatic;
472 }
drhe0bc4042002-06-25 01:09:11 +0000473}
474
475/*
drhe0bc4042002-06-25 01:09:11 +0000476** This routine is called when a commit occurs.
477*/
drh9bb575f2004-09-06 17:24:11 +0000478void sqlite3CommitInternalChanges(sqlite3 *db){
drhe0bc4042002-06-25 01:09:11 +0000479 db->flags &= ~SQLITE_InternChanges;
drh74e24cd2002-01-09 03:19:59 +0000480}
481
482/*
drh956bc922004-07-24 17:38:29 +0000483** Clear the column names from a table or view.
484*/
485static void sqliteResetColumnNames(Table *pTable){
486 int i;
487 Column *pCol;
drh633e6d52008-07-28 19:34:53 +0000488 sqlite3 *db = pTable->db;
drh956bc922004-07-24 17:38:29 +0000489 assert( pTable!=0 );
drhdd5b2fa2005-03-28 03:39:55 +0000490 if( (pCol = pTable->aCol)!=0 ){
491 for(i=0; i<pTable->nCol; i++, pCol++){
drh633e6d52008-07-28 19:34:53 +0000492 sqlite3DbFree(db, pCol->zName);
493 sqlite3ExprDelete(db, pCol->pDflt);
494 sqlite3DbFree(db, pCol->zType);
495 sqlite3DbFree(db, pCol->zColl);
drhdd5b2fa2005-03-28 03:39:55 +0000496 }
drh633e6d52008-07-28 19:34:53 +0000497 sqlite3DbFree(db, pTable->aCol);
drh956bc922004-07-24 17:38:29 +0000498 }
drh956bc922004-07-24 17:38:29 +0000499 pTable->aCol = 0;
500 pTable->nCol = 0;
501}
502
503/*
drh75897232000-05-29 14:26:00 +0000504** Remove the memory data structures associated with the given
drh967e8b72000-06-21 13:59:10 +0000505** Table. No changes are made to disk by this routine.
drh75897232000-05-29 14:26:00 +0000506**
507** This routine just deletes the data structure. It does not unlink
drhc2eef3b2002-08-31 18:53:06 +0000508** the table data structure from the hash table. Nor does it remove
509** foreign keys from the sqlite.aFKey hash table. But it does destroy
510** memory structures of the indices and foreign keys associated with
511** the table.
drh75897232000-05-29 14:26:00 +0000512*/
danielk1977a04a34f2007-04-16 15:06:25 +0000513void sqlite3DeleteTable(Table *pTable){
drh75897232000-05-29 14:26:00 +0000514 Index *pIndex, *pNext;
drhc2eef3b2002-08-31 18:53:06 +0000515 FKey *pFKey, *pNextFKey;
drh633e6d52008-07-28 19:34:53 +0000516 sqlite3 *db;
drhc2eef3b2002-08-31 18:53:06 +0000517
drh75897232000-05-29 14:26:00 +0000518 if( pTable==0 ) return;
drh633e6d52008-07-28 19:34:53 +0000519 db = pTable->db;
drhc2eef3b2002-08-31 18:53:06 +0000520
drhed8a3bb2005-06-06 21:19:56 +0000521 /* Do not delete the table until the reference count reaches zero. */
522 pTable->nRef--;
523 if( pTable->nRef>0 ){
524 return;
525 }
526 assert( pTable->nRef==0 );
527
drhc2eef3b2002-08-31 18:53:06 +0000528 /* Delete all indices associated with this table
529 */
530 for(pIndex = pTable->pIndex; pIndex; pIndex=pNext){
531 pNext = pIndex->pNext;
danielk1977da184232006-01-05 11:34:32 +0000532 assert( pIndex->pSchema==pTable->pSchema );
drh74161702006-02-24 02:53:49 +0000533 sqliteDeleteIndex(pIndex);
drhc2eef3b2002-08-31 18:53:06 +0000534 }
535
danielk1977576ec6b2005-01-21 11:55:25 +0000536#ifndef SQLITE_OMIT_FOREIGN_KEY
drhc2eef3b2002-08-31 18:53:06 +0000537 /* Delete all foreign keys associated with this table. The keys
danielk1977a04a34f2007-04-16 15:06:25 +0000538 ** should have already been unlinked from the pSchema->aFKey hash table
drhc2eef3b2002-08-31 18:53:06 +0000539 */
540 for(pFKey=pTable->pFKey; pFKey; pFKey=pNextFKey){
541 pNextFKey = pFKey->pNextFrom;
danielk1977da184232006-01-05 11:34:32 +0000542 assert( sqlite3HashFind(&pTable->pSchema->aFKey,
drhd24cc422003-03-27 12:51:24 +0000543 pFKey->zTo, strlen(pFKey->zTo)+1)!=pFKey );
drh633e6d52008-07-28 19:34:53 +0000544 sqlite3DbFree(db, pFKey);
drhc2eef3b2002-08-31 18:53:06 +0000545 }
danielk1977576ec6b2005-01-21 11:55:25 +0000546#endif
drhc2eef3b2002-08-31 18:53:06 +0000547
548 /* Delete the Table structure itself.
549 */
drh956bc922004-07-24 17:38:29 +0000550 sqliteResetColumnNames(pTable);
drh633e6d52008-07-28 19:34:53 +0000551 sqlite3DbFree(db, pTable->zName);
552 sqlite3DbFree(db, pTable->zColAff);
553 sqlite3SelectDelete(db, pTable->pSelect);
drhffe07b22005-11-03 00:41:17 +0000554#ifndef SQLITE_OMIT_CHECK
drh633e6d52008-07-28 19:34:53 +0000555 sqlite3ExprDelete(db, pTable->pCheck);
drhffe07b22005-11-03 00:41:17 +0000556#endif
drhb9bb7c12006-06-11 23:41:55 +0000557 sqlite3VtabClear(pTable);
drh633e6d52008-07-28 19:34:53 +0000558 sqlite3DbFree(db, pTable);
drh75897232000-05-29 14:26:00 +0000559}
560
561/*
drh5edc3122001-09-13 21:53:09 +0000562** Unlink the given table from the hash tables and the delete the
drhc2eef3b2002-08-31 18:53:06 +0000563** table structure with all its indices and foreign keys.
drh5edc3122001-09-13 21:53:09 +0000564*/
drh9bb575f2004-09-06 17:24:11 +0000565void sqlite3UnlinkAndDeleteTable(sqlite3 *db, int iDb, const char *zTabName){
drh956bc922004-07-24 17:38:29 +0000566 Table *p;
drhc2eef3b2002-08-31 18:53:06 +0000567 FKey *pF1, *pF2;
drh956bc922004-07-24 17:38:29 +0000568 Db *pDb;
569
drhd229ca92002-01-09 13:30:41 +0000570 assert( db!=0 );
drh956bc922004-07-24 17:38:29 +0000571 assert( iDb>=0 && iDb<db->nDb );
572 assert( zTabName && zTabName[0] );
573 pDb = &db->aDb[iDb];
danielk1977da184232006-01-05 11:34:32 +0000574 p = sqlite3HashInsert(&pDb->pSchema->tblHash, zTabName, strlen(zTabName)+1,0);
drh956bc922004-07-24 17:38:29 +0000575 if( p ){
danielk1977576ec6b2005-01-21 11:55:25 +0000576#ifndef SQLITE_OMIT_FOREIGN_KEY
drh956bc922004-07-24 17:38:29 +0000577 for(pF1=p->pFKey; pF1; pF1=pF1->pNextFrom){
578 int nTo = strlen(pF1->zTo) + 1;
danielk1977da184232006-01-05 11:34:32 +0000579 pF2 = sqlite3HashFind(&pDb->pSchema->aFKey, pF1->zTo, nTo);
drh956bc922004-07-24 17:38:29 +0000580 if( pF2==pF1 ){
danielk1977da184232006-01-05 11:34:32 +0000581 sqlite3HashInsert(&pDb->pSchema->aFKey, pF1->zTo, nTo, pF1->pNextTo);
drh956bc922004-07-24 17:38:29 +0000582 }else{
583 while( pF2 && pF2->pNextTo!=pF1 ){ pF2=pF2->pNextTo; }
584 if( pF2 ){
585 pF2->pNextTo = pF1->pNextTo;
586 }
drhc2eef3b2002-08-31 18:53:06 +0000587 }
588 }
danielk1977576ec6b2005-01-21 11:55:25 +0000589#endif
danielk1977a04a34f2007-04-16 15:06:25 +0000590 sqlite3DeleteTable(p);
drhc2eef3b2002-08-31 18:53:06 +0000591 }
drh956bc922004-07-24 17:38:29 +0000592 db->flags |= SQLITE_InternChanges;
drh74e24cd2002-01-09 03:19:59 +0000593}
594
595/*
drha99db3b2004-06-19 14:49:12 +0000596** Given a token, return a string that consists of the text of that
597** token with any quotations removed. Space to hold the returned string
598** is obtained from sqliteMalloc() and must be freed by the calling
599** function.
drh75897232000-05-29 14:26:00 +0000600**
drhc96d8532005-05-03 12:30:33 +0000601** Tokens are often just pointers into the original SQL text and so
drha99db3b2004-06-19 14:49:12 +0000602** are not \000 terminated and are not persistent. The returned string
603** is \000 terminated and is persistent.
drh75897232000-05-29 14:26:00 +0000604*/
drh17435752007-08-16 04:30:38 +0000605char *sqlite3NameFromToken(sqlite3 *db, Token *pName){
drha99db3b2004-06-19 14:49:12 +0000606 char *zName;
607 if( pName ){
drh17435752007-08-16 04:30:38 +0000608 zName = sqlite3DbStrNDup(db, (char*)pName->z, pName->n);
drha99db3b2004-06-19 14:49:12 +0000609 sqlite3Dequote(zName);
610 }else{
611 zName = 0;
612 }
drh75897232000-05-29 14:26:00 +0000613 return zName;
614}
615
616/*
danielk1977cbb18d22004-05-28 11:37:27 +0000617** Open the sqlite_master table stored in database number iDb for
618** writing. The table is opened using cursor 0.
drhe0bc4042002-06-25 01:09:11 +0000619*/
danielk1977c00da102006-01-07 13:21:04 +0000620void sqlite3OpenMasterTable(Parse *p, int iDb){
621 Vdbe *v = sqlite3GetVdbe(p);
622 sqlite3TableLock(p, iDb, MASTER_ROOT, 1, SCHEMA_TABLE(iDb));
danielk1977cd3e8f72008-03-25 09:47:35 +0000623 sqlite3VdbeAddOp2(v, OP_SetNumColumns, 0, 5);/* sqlite_master has 5 columns */
danielk1977207872a2008-01-03 07:54:23 +0000624 sqlite3VdbeAddOp3(v, OP_OpenWrite, 0, MASTER_ROOT, iDb);
drhe0bc4042002-06-25 01:09:11 +0000625}
626
627/*
danielk1977cbb18d22004-05-28 11:37:27 +0000628** The token *pName contains the name of a database (either "main" or
629** "temp" or the name of an attached db). This routine returns the
630** index of the named database in db->aDb[], or -1 if the named db
631** does not exist.
632*/
drhff2d5ea2005-07-23 00:41:48 +0000633int sqlite3FindDb(sqlite3 *db, Token *pName){
danielk1977576ec6b2005-01-21 11:55:25 +0000634 int i = -1; /* Database number */
drh73c42a12004-11-20 18:13:10 +0000635 int n; /* Number of characters in the name */
636 Db *pDb; /* A database whose name space is being searched */
637 char *zName; /* Name we are searching for */
638
drh17435752007-08-16 04:30:38 +0000639 zName = sqlite3NameFromToken(db, pName);
drh73c42a12004-11-20 18:13:10 +0000640 if( zName ){
641 n = strlen(zName);
danielk1977576ec6b2005-01-21 11:55:25 +0000642 for(i=(db->nDb-1), pDb=&db->aDb[i]; i>=0; i--, pDb--){
danielk197753c0f742005-03-29 03:10:59 +0000643 if( (!OMIT_TEMPDB || i!=1 ) && n==strlen(pDb->zName) &&
644 0==sqlite3StrICmp(pDb->zName, zName) ){
danielk1977576ec6b2005-01-21 11:55:25 +0000645 break;
drh73c42a12004-11-20 18:13:10 +0000646 }
danielk1977cbb18d22004-05-28 11:37:27 +0000647 }
drh633e6d52008-07-28 19:34:53 +0000648 sqlite3DbFree(db, zName);
danielk1977cbb18d22004-05-28 11:37:27 +0000649 }
danielk1977576ec6b2005-01-21 11:55:25 +0000650 return i;
danielk1977cbb18d22004-05-28 11:37:27 +0000651}
652
drh0e3d7472004-06-19 17:33:07 +0000653/* The table or view or trigger name is passed to this routine via tokens
654** pName1 and pName2. If the table name was fully qualified, for example:
655**
656** CREATE TABLE xxx.yyy (...);
657**
658** Then pName1 is set to "xxx" and pName2 "yyy". On the other hand if
659** the table name is not fully qualified, i.e.:
660**
661** CREATE TABLE yyy(...);
662**
663** Then pName1 is set to "yyy" and pName2 is "".
664**
665** This routine sets the *ppUnqual pointer to point at the token (pName1 or
666** pName2) that stores the unqualified table name. The index of the
667** database "xxx" is returned.
668*/
danielk1977ef2cb632004-05-29 02:37:19 +0000669int sqlite3TwoPartName(
drh0e3d7472004-06-19 17:33:07 +0000670 Parse *pParse, /* Parsing and code generating context */
drh90f5ecb2004-07-22 01:19:35 +0000671 Token *pName1, /* The "xxx" in the name "xxx.yyy" or "xxx" */
drh0e3d7472004-06-19 17:33:07 +0000672 Token *pName2, /* The "yyy" in the name "xxx.yyy" */
673 Token **pUnqual /* Write the unqualified object name here */
danielk1977cbb18d22004-05-28 11:37:27 +0000674){
drh0e3d7472004-06-19 17:33:07 +0000675 int iDb; /* Database holding the object */
danielk1977cbb18d22004-05-28 11:37:27 +0000676 sqlite3 *db = pParse->db;
677
678 if( pName2 && pName2->n>0 ){
679 assert( !db->init.busy );
680 *pUnqual = pName2;
drhff2d5ea2005-07-23 00:41:48 +0000681 iDb = sqlite3FindDb(db, pName1);
danielk1977cbb18d22004-05-28 11:37:27 +0000682 if( iDb<0 ){
683 sqlite3ErrorMsg(pParse, "unknown database %T", pName1);
684 pParse->nErr++;
685 return -1;
686 }
687 }else{
688 assert( db->init.iDb==0 || db->init.busy );
689 iDb = db->init.iDb;
690 *pUnqual = pName1;
691 }
692 return iDb;
693}
694
695/*
danielk1977d8123362004-06-12 09:25:12 +0000696** This routine is used to check if the UTF-8 string zName is a legal
697** unqualified name for a new schema object (table, index, view or
698** trigger). All names are legal except those that begin with the string
699** "sqlite_" (in upper, lower or mixed case). This portion of the namespace
700** is reserved for internal use.
701*/
702int sqlite3CheckObjectName(Parse *pParse, const char *zName){
drhf1974842004-11-05 03:56:00 +0000703 if( !pParse->db->init.busy && pParse->nested==0
danielk19773a3f38e2005-05-22 06:49:56 +0000704 && (pParse->db->flags & SQLITE_WriteSchema)==0
drhf1974842004-11-05 03:56:00 +0000705 && 0==sqlite3StrNICmp(zName, "sqlite_", 7) ){
danielk1977d8123362004-06-12 09:25:12 +0000706 sqlite3ErrorMsg(pParse, "object name reserved for internal use: %s", zName);
707 return SQLITE_ERROR;
708 }
709 return SQLITE_OK;
710}
711
712/*
drh75897232000-05-29 14:26:00 +0000713** Begin constructing a new table representation in memory. This is
714** the first of several action routines that get called in response
drhd9b02572001-04-15 00:37:09 +0000715** to a CREATE TABLE statement. In particular, this routine is called
drh74161702006-02-24 02:53:49 +0000716** after seeing tokens "CREATE" and "TABLE" and the table name. The isTemp
drhe0bc4042002-06-25 01:09:11 +0000717** flag is true if the table should be stored in the auxiliary database
718** file instead of in the main database file. This is normally the case
719** when the "TEMP" or "TEMPORARY" keyword occurs in between
drhf57b3392001-10-08 13:22:32 +0000720** CREATE and TABLE.
drhd9b02572001-04-15 00:37:09 +0000721**
drhf57b3392001-10-08 13:22:32 +0000722** The new table record is initialized and put in pParse->pNewTable.
723** As more of the CREATE TABLE statement is parsed, additional action
724** routines will be called to add more information to this record.
danielk19774adee202004-05-08 08:23:19 +0000725** At the end of the CREATE TABLE statement, the sqlite3EndTable() routine
drhf57b3392001-10-08 13:22:32 +0000726** is called to complete the construction of the new table record.
drh75897232000-05-29 14:26:00 +0000727*/
danielk19774adee202004-05-08 08:23:19 +0000728void sqlite3StartTable(
drhe5f9c642003-01-13 23:27:31 +0000729 Parse *pParse, /* Parser context */
danielk1977cbb18d22004-05-28 11:37:27 +0000730 Token *pName1, /* First part of the name of the table or view */
731 Token *pName2, /* Second part of the name of the table or view */
drhe5f9c642003-01-13 23:27:31 +0000732 int isTemp, /* True if this is a TEMP table */
drhfaa59552005-12-29 23:33:54 +0000733 int isView, /* True if this is a VIEW */
danielk1977f1a381e2006-06-16 08:01:02 +0000734 int isVirtual, /* True if this is a VIRTUAL table */
drhfaa59552005-12-29 23:33:54 +0000735 int noErr /* Do nothing if table already exists */
drhe5f9c642003-01-13 23:27:31 +0000736){
drh75897232000-05-29 14:26:00 +0000737 Table *pTable;
drh23bf66d2004-12-14 03:34:34 +0000738 char *zName = 0; /* The name of the new table */
drh9bb575f2004-09-06 17:24:11 +0000739 sqlite3 *db = pParse->db;
drhadbca9c2001-09-27 15:11:53 +0000740 Vdbe *v;
danielk1977cbb18d22004-05-28 11:37:27 +0000741 int iDb; /* Database number to create the table in */
742 Token *pName; /* Unqualified name of the table to create */
drh75897232000-05-29 14:26:00 +0000743
danielk1977cbb18d22004-05-28 11:37:27 +0000744 /* The table or view name to create is passed to this routine via tokens
745 ** pName1 and pName2. If the table name was fully qualified, for example:
746 **
747 ** CREATE TABLE xxx.yyy (...);
748 **
749 ** Then pName1 is set to "xxx" and pName2 "yyy". On the other hand if
750 ** the table name is not fully qualified, i.e.:
751 **
752 ** CREATE TABLE yyy(...);
753 **
754 ** Then pName1 is set to "yyy" and pName2 is "".
755 **
756 ** The call below sets the pName pointer to point at the token (pName1 or
757 ** pName2) that stores the unqualified table name. The variable iDb is
758 ** set to the index of the database that the table or view is to be
759 ** created in.
760 */
danielk1977ef2cb632004-05-29 02:37:19 +0000761 iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pName);
danielk1977cbb18d22004-05-28 11:37:27 +0000762 if( iDb<0 ) return;
danielk197753c0f742005-03-29 03:10:59 +0000763 if( !OMIT_TEMPDB && isTemp && iDb>1 ){
danielk1977cbb18d22004-05-28 11:37:27 +0000764 /* If creating a temp table, the name may not be qualified */
765 sqlite3ErrorMsg(pParse, "temporary table name must be unqualified");
danielk1977cbb18d22004-05-28 11:37:27 +0000766 return;
767 }
danielk197753c0f742005-03-29 03:10:59 +0000768 if( !OMIT_TEMPDB && isTemp ) iDb = 1;
danielk1977cbb18d22004-05-28 11:37:27 +0000769
770 pParse->sNameToken = *pName;
drh17435752007-08-16 04:30:38 +0000771 zName = sqlite3NameFromToken(db, pName);
danielk1977e0048402004-06-15 16:51:01 +0000772 if( zName==0 ) return;
danielk1977d8123362004-06-12 09:25:12 +0000773 if( SQLITE_OK!=sqlite3CheckObjectName(pParse, zName) ){
drh23bf66d2004-12-14 03:34:34 +0000774 goto begin_table_error;
danielk1977d8123362004-06-12 09:25:12 +0000775 }
drh1d85d932004-02-14 23:05:52 +0000776 if( db->init.iDb==1 ) isTemp = 1;
drhe5f9c642003-01-13 23:27:31 +0000777#ifndef SQLITE_OMIT_AUTHORIZATION
drhd24cc422003-03-27 12:51:24 +0000778 assert( (isTemp & 1)==isTemp );
drhe5f9c642003-01-13 23:27:31 +0000779 {
780 int code;
danielk1977cbb18d22004-05-28 11:37:27 +0000781 char *zDb = db->aDb[iDb].zName;
danielk19774adee202004-05-08 08:23:19 +0000782 if( sqlite3AuthCheck(pParse, SQLITE_INSERT, SCHEMA_TABLE(isTemp), 0, zDb) ){
drh23bf66d2004-12-14 03:34:34 +0000783 goto begin_table_error;
drhe22a3342003-04-22 20:30:37 +0000784 }
drhe5f9c642003-01-13 23:27:31 +0000785 if( isView ){
danielk197753c0f742005-03-29 03:10:59 +0000786 if( !OMIT_TEMPDB && isTemp ){
drhe5f9c642003-01-13 23:27:31 +0000787 code = SQLITE_CREATE_TEMP_VIEW;
788 }else{
789 code = SQLITE_CREATE_VIEW;
790 }
791 }else{
danielk197753c0f742005-03-29 03:10:59 +0000792 if( !OMIT_TEMPDB && isTemp ){
drhe5f9c642003-01-13 23:27:31 +0000793 code = SQLITE_CREATE_TEMP_TABLE;
794 }else{
795 code = SQLITE_CREATE_TABLE;
796 }
797 }
danielk1977f1a381e2006-06-16 08:01:02 +0000798 if( !isVirtual && sqlite3AuthCheck(pParse, code, zName, 0, zDb) ){
drh23bf66d2004-12-14 03:34:34 +0000799 goto begin_table_error;
drhe5f9c642003-01-13 23:27:31 +0000800 }
801 }
802#endif
drhf57b3392001-10-08 13:22:32 +0000803
drhf57b3392001-10-08 13:22:32 +0000804 /* Make sure the new table name does not collide with an existing
danielk19773df6b252004-05-29 10:23:19 +0000805 ** index or table name in the same database. Issue an error message if
danielk19777e6ebfb2006-06-12 11:24:37 +0000806 ** it does. The exception is if the statement being parsed was passed
807 ** to an sqlite3_declare_vtab() call. In that case only the column names
808 ** and types will be used, so there is no need to test for namespace
809 ** collisions.
drhf57b3392001-10-08 13:22:32 +0000810 */
danielk19777e6ebfb2006-06-12 11:24:37 +0000811 if( !IN_DECLARE_VTAB ){
812 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){
813 goto begin_table_error;
drhfaa59552005-12-29 23:33:54 +0000814 }
danielk19777e6ebfb2006-06-12 11:24:37 +0000815 pTable = sqlite3FindTable(db, zName, db->aDb[iDb].zName);
816 if( pTable ){
817 if( !noErr ){
818 sqlite3ErrorMsg(pParse, "table %T already exists", pName);
819 }
820 goto begin_table_error;
821 }
822 if( sqlite3FindIndex(db, zName, 0)!=0 && (iDb==0 || !db->init.busy) ){
823 sqlite3ErrorMsg(pParse, "there is already an index named %s", zName);
824 goto begin_table_error;
825 }
drh75897232000-05-29 14:26:00 +0000826 }
danielk19777e6ebfb2006-06-12 11:24:37 +0000827
danielk197726783a52007-08-29 14:06:22 +0000828 pTable = sqlite3DbMallocZero(db, sizeof(Table));
drh6d4abfb2001-10-22 02:58:08 +0000829 if( pTable==0 ){
drh17435752007-08-16 04:30:38 +0000830 db->mallocFailed = 1;
danielk1977e0048402004-06-15 16:51:01 +0000831 pParse->rc = SQLITE_NOMEM;
832 pParse->nErr++;
drh23bf66d2004-12-14 03:34:34 +0000833 goto begin_table_error;
drh6d4abfb2001-10-22 02:58:08 +0000834 }
drh75897232000-05-29 14:26:00 +0000835 pTable->zName = zName;
drh4a324312001-12-21 14:30:42 +0000836 pTable->iPKey = -1;
danielk1977da184232006-01-05 11:34:32 +0000837 pTable->pSchema = db->aDb[iDb].pSchema;
drhed8a3bb2005-06-06 21:19:56 +0000838 pTable->nRef = 1;
drh633e6d52008-07-28 19:34:53 +0000839 pTable->db = db;
danielk1977a04a34f2007-04-16 15:06:25 +0000840 if( pParse->pNewTable ) sqlite3DeleteTable(pParse->pNewTable);
drh75897232000-05-29 14:26:00 +0000841 pParse->pNewTable = pTable;
drh17f71932002-02-21 12:01:27 +0000842
drh4794f732004-11-05 17:17:50 +0000843 /* If this is the magic sqlite_sequence table used by autoincrement,
844 ** then record a pointer to this table in the main database structure
845 ** so that INSERT can find the table easily.
846 */
847#ifndef SQLITE_OMIT_AUTOINCREMENT
drh78776ec2005-06-14 02:12:46 +0000848 if( !pParse->nested && strcmp(zName, "sqlite_sequence")==0 ){
danielk1977da184232006-01-05 11:34:32 +0000849 pTable->pSchema->pSeqTab = pTable;
drh4794f732004-11-05 17:17:50 +0000850 }
851#endif
852
drh17f71932002-02-21 12:01:27 +0000853 /* Begin generating the code that will insert the table record into
854 ** the SQLITE_MASTER table. Note in particular that we must go ahead
855 ** and allocate the record number for the table entry now. Before any
856 ** PRIMARY KEY or UNIQUE keywords are parsed. Those keywords will cause
857 ** indices to be created and the table record must come before the
858 ** indices. Hence, the record number for the table must be allocated
859 ** now.
860 */
danielk19774adee202004-05-08 08:23:19 +0000861 if( !db->init.busy && (v = sqlite3GetVdbe(pParse))!=0 ){
drhb7654112008-01-12 12:48:07 +0000862 int j1;
drhe321c292006-01-12 01:56:43 +0000863 int fileFormat;
drhb7654112008-01-12 12:48:07 +0000864 int reg1, reg2, reg3;
danielk1977cbb18d22004-05-28 11:37:27 +0000865 sqlite3BeginWriteOperation(pParse, 0, iDb);
drhb17131a2004-11-05 22:18:49 +0000866
danielk197720b1eaf2006-07-26 16:22:14 +0000867#ifndef SQLITE_OMIT_VIRTUALTABLE
868 if( isVirtual ){
drh66a51672008-01-03 00:01:23 +0000869 sqlite3VdbeAddOp0(v, OP_VBegin);
danielk197720b1eaf2006-07-26 16:22:14 +0000870 }
871#endif
872
danielk197736963fd2005-02-19 08:18:05 +0000873 /* If the file format and encoding in the database have not been set,
874 ** set them now.
danielk1977d008cfe2004-06-19 02:22:10 +0000875 */
drhb7654112008-01-12 12:48:07 +0000876 reg1 = pParse->regRowid = ++pParse->nMem;
877 reg2 = pParse->regRoot = ++pParse->nMem;
878 reg3 = ++pParse->nMem;
879 sqlite3VdbeAddOp3(v, OP_ReadCookie, iDb, reg3, 1); /* file_format */
drhfb982642007-08-30 01:19:59 +0000880 sqlite3VdbeUsesBtree(v, iDb);
drhb7654112008-01-12 12:48:07 +0000881 j1 = sqlite3VdbeAddOp1(v, OP_If, reg3);
drhe321c292006-01-12 01:56:43 +0000882 fileFormat = (db->flags & SQLITE_LegacyFileFmt)!=0 ?
drh76fe8032006-07-11 14:17:51 +0000883 1 : SQLITE_MAX_FILE_FORMAT;
drhb7654112008-01-12 12:48:07 +0000884 sqlite3VdbeAddOp2(v, OP_Integer, fileFormat, reg3);
885 sqlite3VdbeAddOp3(v, OP_SetCookie, iDb, 1, reg3);
886 sqlite3VdbeAddOp2(v, OP_Integer, ENC(db), reg3);
887 sqlite3VdbeAddOp3(v, OP_SetCookie, iDb, 4, reg3);
888 sqlite3VdbeJumpHere(v, j1);
danielk1977d008cfe2004-06-19 02:22:10 +0000889
drh4794f732004-11-05 17:17:50 +0000890 /* This just creates a place-holder record in the sqlite_master table.
891 ** The record created does not contain anything yet. It will be replaced
892 ** by the real entry in code generated at sqlite3EndTable().
drhb17131a2004-11-05 22:18:49 +0000893 **
894 ** The rowid for the new entry is left on the top of the stack.
895 ** The rowid value is needed by the code that sqlite3EndTable will
896 ** generate.
drh4794f732004-11-05 17:17:50 +0000897 */
danielk1977f1a381e2006-06-16 08:01:02 +0000898#if !defined(SQLITE_OMIT_VIEW) || !defined(SQLITE_OMIT_VIRTUALTABLE)
899 if( isView || isVirtual ){
drhb7654112008-01-12 12:48:07 +0000900 sqlite3VdbeAddOp2(v, OP_Integer, 0, reg2);
danielk1977a21c6b62005-01-24 10:25:59 +0000901 }else
902#endif
903 {
drhb7654112008-01-12 12:48:07 +0000904 sqlite3VdbeAddOp2(v, OP_CreateTable, iDb, reg2);
danielk1977a21c6b62005-01-24 10:25:59 +0000905 }
danielk1977c00da102006-01-07 13:21:04 +0000906 sqlite3OpenMasterTable(pParse, iDb);
drhb7654112008-01-12 12:48:07 +0000907 sqlite3VdbeAddOp2(v, OP_NewRowid, 0, reg1);
908 sqlite3VdbeAddOp2(v, OP_Null, 0, reg3);
909 sqlite3VdbeAddOp3(v, OP_Insert, 0, reg3, reg1);
910 sqlite3VdbeChangeP5(v, OPFLAG_APPEND);
drh66a51672008-01-03 00:01:23 +0000911 sqlite3VdbeAddOp0(v, OP_Close);
drh5e00f6c2001-09-13 13:46:56 +0000912 }
drh23bf66d2004-12-14 03:34:34 +0000913
914 /* Normal (non-error) return. */
915 return;
916
917 /* If an error occurs, we jump here */
918begin_table_error:
drh633e6d52008-07-28 19:34:53 +0000919 sqlite3DbFree(db, zName);
drh23bf66d2004-12-14 03:34:34 +0000920 return;
drh75897232000-05-29 14:26:00 +0000921}
922
923/*
danielk1977c60e9b82005-01-31 12:42:29 +0000924** This macro is used to compare two strings in a case-insensitive manner.
925** It is slightly faster than calling sqlite3StrICmp() directly, but
926** produces larger code.
927**
928** WARNING: This macro is not compatible with the strcmp() family. It
929** returns true if the two strings are equal, otherwise false.
930*/
931#define STRICMP(x, y) (\
932sqlite3UpperToLower[*(unsigned char *)(x)]== \
933sqlite3UpperToLower[*(unsigned char *)(y)] \
934&& sqlite3StrICmp((x)+1,(y)+1)==0 )
935
936/*
drh75897232000-05-29 14:26:00 +0000937** Add a new column to the table currently being constructed.
drhd9b02572001-04-15 00:37:09 +0000938**
939** The parser calls this routine once for each column declaration
danielk19774adee202004-05-08 08:23:19 +0000940** in a CREATE TABLE statement. sqlite3StartTable() gets called
drhd9b02572001-04-15 00:37:09 +0000941** first to get things going. Then this routine is called for each
942** column.
drh75897232000-05-29 14:26:00 +0000943*/
danielk19774adee202004-05-08 08:23:19 +0000944void sqlite3AddColumn(Parse *pParse, Token *pName){
drh75897232000-05-29 14:26:00 +0000945 Table *p;
drh97fc3d02002-05-22 21:27:03 +0000946 int i;
drha99db3b2004-06-19 14:49:12 +0000947 char *z;
drhc9b84a12002-06-20 11:36:48 +0000948 Column *pCol;
drhbb4957f2008-03-20 14:03:29 +0000949 sqlite3 *db = pParse->db;
drh75897232000-05-29 14:26:00 +0000950 if( (p = pParse->pNewTable)==0 ) return;
drhbb4957f2008-03-20 14:03:29 +0000951#if SQLITE_MAX_COLUMN
952 if( p->nCol+1>db->aLimit[SQLITE_LIMIT_COLUMN] ){
drhe5c941b2007-05-08 13:58:26 +0000953 sqlite3ErrorMsg(pParse, "too many columns on %s", p->zName);
954 return;
955 }
drhbb4957f2008-03-20 14:03:29 +0000956#endif
drh17435752007-08-16 04:30:38 +0000957 z = sqlite3NameFromToken(pParse->db, pName);
drh97fc3d02002-05-22 21:27:03 +0000958 if( z==0 ) return;
drh97fc3d02002-05-22 21:27:03 +0000959 for(i=0; i<p->nCol; i++){
danielk1977c60e9b82005-01-31 12:42:29 +0000960 if( STRICMP(z, p->aCol[i].zName) ){
danielk19774adee202004-05-08 08:23:19 +0000961 sqlite3ErrorMsg(pParse, "duplicate column name: %s", z);
drh633e6d52008-07-28 19:34:53 +0000962 sqlite3DbFree(db, z);
drh97fc3d02002-05-22 21:27:03 +0000963 return;
964 }
965 }
drh75897232000-05-29 14:26:00 +0000966 if( (p->nCol & 0x7)==0 ){
drh6d4abfb2001-10-22 02:58:08 +0000967 Column *aNew;
danielk197726783a52007-08-29 14:06:22 +0000968 aNew = sqlite3DbRealloc(pParse->db,p->aCol,(p->nCol+8)*sizeof(p->aCol[0]));
danielk1977d5d56522005-03-16 12:15:20 +0000969 if( aNew==0 ){
drh633e6d52008-07-28 19:34:53 +0000970 sqlite3DbFree(db, z);
danielk1977d5d56522005-03-16 12:15:20 +0000971 return;
972 }
drh6d4abfb2001-10-22 02:58:08 +0000973 p->aCol = aNew;
drh75897232000-05-29 14:26:00 +0000974 }
drhc9b84a12002-06-20 11:36:48 +0000975 pCol = &p->aCol[p->nCol];
976 memset(pCol, 0, sizeof(p->aCol[0]));
977 pCol->zName = z;
danielk1977a37cdde2004-05-16 11:15:36 +0000978
979 /* If there is no type specified, columns have the default affinity
danielk19774f057f92004-06-08 00:02:33 +0000980 ** 'NONE'. If there is a type specified, then sqlite3AddColumnType() will
981 ** be called next to set pCol->affinity correctly.
danielk1977a37cdde2004-05-16 11:15:36 +0000982 */
danielk19774f057f92004-06-08 00:02:33 +0000983 pCol->affinity = SQLITE_AFF_NONE;
drhc9b84a12002-06-20 11:36:48 +0000984 p->nCol++;
drh75897232000-05-29 14:26:00 +0000985}
986
987/*
drh382c0242001-10-06 16:33:02 +0000988** This routine is called by the parser while in the middle of
989** parsing a CREATE TABLE statement. A "NOT NULL" constraint has
990** been seen on a column. This routine sets the notNull flag on
991** the column currently under construction.
992*/
danielk19774adee202004-05-08 08:23:19 +0000993void sqlite3AddNotNull(Parse *pParse, int onError){
drh382c0242001-10-06 16:33:02 +0000994 Table *p;
995 int i;
996 if( (p = pParse->pNewTable)==0 ) return;
997 i = p->nCol-1;
drh9cfcf5d2002-01-29 18:41:24 +0000998 if( i>=0 ) p->aCol[i].notNull = onError;
drh382c0242001-10-06 16:33:02 +0000999}
1000
1001/*
danielk197752a83fb2005-01-31 12:56:44 +00001002** Scan the column type name zType (length nType) and return the
1003** associated affinity type.
danielk1977b3dff962005-02-01 01:21:55 +00001004**
1005** This routine does a case-independent search of zType for the
1006** substrings in the following table. If one of the substrings is
1007** found, the corresponding affinity is returned. If zType contains
1008** more than one of the substrings, entries toward the top of
1009** the table take priority. For example, if zType is 'BLOBINT',
drh8a512562005-11-14 22:29:05 +00001010** SQLITE_AFF_INTEGER is returned.
danielk1977b3dff962005-02-01 01:21:55 +00001011**
1012** Substring | Affinity
1013** --------------------------------
1014** 'INT' | SQLITE_AFF_INTEGER
1015** 'CHAR' | SQLITE_AFF_TEXT
1016** 'CLOB' | SQLITE_AFF_TEXT
1017** 'TEXT' | SQLITE_AFF_TEXT
1018** 'BLOB' | SQLITE_AFF_NONE
drh8a512562005-11-14 22:29:05 +00001019** 'REAL' | SQLITE_AFF_REAL
1020** 'FLOA' | SQLITE_AFF_REAL
1021** 'DOUB' | SQLITE_AFF_REAL
danielk1977b3dff962005-02-01 01:21:55 +00001022**
1023** If none of the substrings in the above table are found,
1024** SQLITE_AFF_NUMERIC is returned.
danielk197752a83fb2005-01-31 12:56:44 +00001025*/
drh8a512562005-11-14 22:29:05 +00001026char sqlite3AffinityType(const Token *pType){
danielk1977b3dff962005-02-01 01:21:55 +00001027 u32 h = 0;
1028 char aff = SQLITE_AFF_NUMERIC;
drh487e2622005-06-25 18:42:14 +00001029 const unsigned char *zIn = pType->z;
1030 const unsigned char *zEnd = &pType->z[pType->n];
danielk197752a83fb2005-01-31 12:56:44 +00001031
danielk1977b3dff962005-02-01 01:21:55 +00001032 while( zIn!=zEnd ){
1033 h = (h<<8) + sqlite3UpperToLower[*zIn];
1034 zIn++;
danielk1977201f7162005-02-01 02:13:29 +00001035 if( h==(('c'<<24)+('h'<<16)+('a'<<8)+'r') ){ /* CHAR */
1036 aff = SQLITE_AFF_TEXT;
1037 }else if( h==(('c'<<24)+('l'<<16)+('o'<<8)+'b') ){ /* CLOB */
1038 aff = SQLITE_AFF_TEXT;
1039 }else if( h==(('t'<<24)+('e'<<16)+('x'<<8)+'t') ){ /* TEXT */
1040 aff = SQLITE_AFF_TEXT;
1041 }else if( h==(('b'<<24)+('l'<<16)+('o'<<8)+'b') /* BLOB */
drh8a512562005-11-14 22:29:05 +00001042 && (aff==SQLITE_AFF_NUMERIC || aff==SQLITE_AFF_REAL) ){
danielk1977b3dff962005-02-01 01:21:55 +00001043 aff = SQLITE_AFF_NONE;
drh8a512562005-11-14 22:29:05 +00001044#ifndef SQLITE_OMIT_FLOATING_POINT
1045 }else if( h==(('r'<<24)+('e'<<16)+('a'<<8)+'l') /* REAL */
1046 && aff==SQLITE_AFF_NUMERIC ){
1047 aff = SQLITE_AFF_REAL;
1048 }else if( h==(('f'<<24)+('l'<<16)+('o'<<8)+'a') /* FLOA */
1049 && aff==SQLITE_AFF_NUMERIC ){
1050 aff = SQLITE_AFF_REAL;
1051 }else if( h==(('d'<<24)+('o'<<16)+('u'<<8)+'b') /* DOUB */
1052 && aff==SQLITE_AFF_NUMERIC ){
1053 aff = SQLITE_AFF_REAL;
1054#endif
danielk1977201f7162005-02-01 02:13:29 +00001055 }else if( (h&0x00FFFFFF)==(('i'<<16)+('n'<<8)+'t') ){ /* INT */
drh8a512562005-11-14 22:29:05 +00001056 aff = SQLITE_AFF_INTEGER;
danielk1977b3dff962005-02-01 01:21:55 +00001057 break;
danielk197752a83fb2005-01-31 12:56:44 +00001058 }
1059 }
danielk1977b3dff962005-02-01 01:21:55 +00001060
1061 return aff;
danielk197752a83fb2005-01-31 12:56:44 +00001062}
1063
1064/*
drh382c0242001-10-06 16:33:02 +00001065** This routine is called by the parser while in the middle of
1066** parsing a CREATE TABLE statement. The pFirst token is the first
1067** token in the sequence of tokens that describe the type of the
1068** column currently under construction. pLast is the last token
1069** in the sequence. Use this information to construct a string
1070** that contains the typename of the column and store that string
1071** in zType.
1072*/
drh487e2622005-06-25 18:42:14 +00001073void sqlite3AddColumnType(Parse *pParse, Token *pType){
drh382c0242001-10-06 16:33:02 +00001074 Table *p;
drh487e2622005-06-25 18:42:14 +00001075 int i;
drhc9b84a12002-06-20 11:36:48 +00001076 Column *pCol;
drh633e6d52008-07-28 19:34:53 +00001077 sqlite3 *db;
drh487e2622005-06-25 18:42:14 +00001078
drh382c0242001-10-06 16:33:02 +00001079 if( (p = pParse->pNewTable)==0 ) return;
1080 i = p->nCol-1;
drhf57b3392001-10-08 13:22:32 +00001081 if( i<0 ) return;
drhc9b84a12002-06-20 11:36:48 +00001082 pCol = &p->aCol[i];
drh633e6d52008-07-28 19:34:53 +00001083 db = pParse->db;
1084 sqlite3DbFree(db, pCol->zType);
1085 pCol->zType = sqlite3NameFromToken(db, pType);
drh8a512562005-11-14 22:29:05 +00001086 pCol->affinity = sqlite3AffinityType(pType);
drh382c0242001-10-06 16:33:02 +00001087}
1088
1089/*
danielk19777977a172004-11-09 12:44:37 +00001090** The expression is the default value for the most recently added column
1091** of the table currently under construction.
1092**
1093** Default value expressions must be constant. Raise an exception if this
1094** is not the case.
drhd9b02572001-04-15 00:37:09 +00001095**
1096** This routine is called by the parser while in the middle of
1097** parsing a CREATE TABLE statement.
drh7020f652000-06-03 18:06:52 +00001098*/
danielk19777977a172004-11-09 12:44:37 +00001099void sqlite3AddDefaultValue(Parse *pParse, Expr *pExpr){
drh7020f652000-06-03 18:06:52 +00001100 Table *p;
danielk19777977a172004-11-09 12:44:37 +00001101 Column *pCol;
drh633e6d52008-07-28 19:34:53 +00001102 sqlite3 *db = pParse->db;
drh42b9d7c2005-08-13 00:56:27 +00001103 if( (p = pParse->pNewTable)!=0 ){
1104 pCol = &(p->aCol[p->nCol-1]);
1105 if( !sqlite3ExprIsConstantOrFunction(pExpr) ){
1106 sqlite3ErrorMsg(pParse, "default value of column [%s] is not constant",
1107 pCol->zName);
1108 }else{
drh417ec632006-08-14 14:23:41 +00001109 Expr *pCopy;
drh633e6d52008-07-28 19:34:53 +00001110 sqlite3ExprDelete(db, pCol->pDflt);
drh17435752007-08-16 04:30:38 +00001111 pCol->pDflt = pCopy = sqlite3ExprDup(db, pExpr);
drh417ec632006-08-14 14:23:41 +00001112 if( pCopy ){
drh17435752007-08-16 04:30:38 +00001113 sqlite3TokenCopy(db, &pCopy->span, &pExpr->span);
drh417ec632006-08-14 14:23:41 +00001114 }
drh42b9d7c2005-08-13 00:56:27 +00001115 }
danielk19777977a172004-11-09 12:44:37 +00001116 }
drh633e6d52008-07-28 19:34:53 +00001117 sqlite3ExprDelete(db, pExpr);
drh7020f652000-06-03 18:06:52 +00001118}
1119
1120/*
drh4a324312001-12-21 14:30:42 +00001121** Designate the PRIMARY KEY for the table. pList is a list of names
1122** of columns that form the primary key. If pList is NULL, then the
1123** most recently added column of the table is the primary key.
1124**
1125** A table can have at most one primary key. If the table already has
1126** a primary key (and this is the second primary key) then create an
1127** error.
1128**
1129** If the PRIMARY KEY is on a single column whose datatype is INTEGER,
drh23bf66d2004-12-14 03:34:34 +00001130** then we will try to use that column as the rowid. Set the Table.iPKey
drh4a324312001-12-21 14:30:42 +00001131** field of the table under construction to be the index of the
1132** INTEGER PRIMARY KEY column. Table.iPKey is set to -1 if there is
1133** no INTEGER PRIMARY KEY.
1134**
1135** If the key is not an INTEGER PRIMARY KEY, then create a unique
1136** index for the key. No index is created for INTEGER PRIMARY KEYs.
1137*/
drh205f48e2004-11-05 00:43:11 +00001138void sqlite3AddPrimaryKey(
1139 Parse *pParse, /* Parsing context */
1140 ExprList *pList, /* List of field names to be indexed */
1141 int onError, /* What to do with a uniqueness conflict */
drhfdd6e852005-12-16 01:06:16 +00001142 int autoInc, /* True if the AUTOINCREMENT keyword is present */
1143 int sortOrder /* SQLITE_SO_ASC or SQLITE_SO_DESC */
drh205f48e2004-11-05 00:43:11 +00001144){
drh4a324312001-12-21 14:30:42 +00001145 Table *pTab = pParse->pNewTable;
1146 char *zType = 0;
drh78100cc2003-08-23 22:40:53 +00001147 int iCol = -1, i;
danielk1977c7d54102006-06-15 07:29:00 +00001148 if( pTab==0 || IN_DECLARE_VTAB ) goto primary_key_exit;
drh4a324312001-12-21 14:30:42 +00001149 if( pTab->hasPrimKey ){
danielk19774adee202004-05-08 08:23:19 +00001150 sqlite3ErrorMsg(pParse,
drhf7a9e1a2004-02-22 18:40:56 +00001151 "table \"%s\" has more than one primary key", pTab->zName);
drhe0194f22003-02-26 13:52:51 +00001152 goto primary_key_exit;
drh4a324312001-12-21 14:30:42 +00001153 }
1154 pTab->hasPrimKey = 1;
1155 if( pList==0 ){
1156 iCol = pTab->nCol - 1;
drh78100cc2003-08-23 22:40:53 +00001157 pTab->aCol[iCol].isPrimKey = 1;
1158 }else{
danielk19770202b292004-06-09 09:55:16 +00001159 for(i=0; i<pList->nExpr; i++){
drh78100cc2003-08-23 22:40:53 +00001160 for(iCol=0; iCol<pTab->nCol; iCol++){
drhd3d39e92004-05-20 22:16:29 +00001161 if( sqlite3StrICmp(pList->a[i].zName, pTab->aCol[iCol].zName)==0 ){
1162 break;
1163 }
drh78100cc2003-08-23 22:40:53 +00001164 }
drh6e4fc2c2005-09-15 21:24:51 +00001165 if( iCol<pTab->nCol ){
drh688c9f02005-09-16 02:48:01 +00001166 pTab->aCol[iCol].isPrimKey = 1;
drh6e4fc2c2005-09-15 21:24:51 +00001167 }
drh4a324312001-12-21 14:30:42 +00001168 }
danielk19770202b292004-06-09 09:55:16 +00001169 if( pList->nExpr>1 ) iCol = -1;
drh4a324312001-12-21 14:30:42 +00001170 }
1171 if( iCol>=0 && iCol<pTab->nCol ){
1172 zType = pTab->aCol[iCol].zType;
1173 }
drhfdd6e852005-12-16 01:06:16 +00001174 if( zType && sqlite3StrICmp(zType, "INTEGER")==0
1175 && sortOrder==SQLITE_SO_ASC ){
drh4a324312001-12-21 14:30:42 +00001176 pTab->iPKey = iCol;
drh9cfcf5d2002-01-29 18:41:24 +00001177 pTab->keyConf = onError;
drh205f48e2004-11-05 00:43:11 +00001178 pTab->autoInc = autoInc;
1179 }else if( autoInc ){
drh4794f732004-11-05 17:17:50 +00001180#ifndef SQLITE_OMIT_AUTOINCREMENT
drh205f48e2004-11-05 00:43:11 +00001181 sqlite3ErrorMsg(pParse, "AUTOINCREMENT is only allowed on an "
1182 "INTEGER PRIMARY KEY");
drh4794f732004-11-05 17:17:50 +00001183#endif
drh4a324312001-12-21 14:30:42 +00001184 }else{
drh4d91a702006-01-04 15:54:36 +00001185 sqlite3CreateIndex(pParse, 0, 0, 0, pList, onError, 0, 0, sortOrder, 0);
drhe0194f22003-02-26 13:52:51 +00001186 pList = 0;
drh4a324312001-12-21 14:30:42 +00001187 }
drhe0194f22003-02-26 13:52:51 +00001188
1189primary_key_exit:
drh633e6d52008-07-28 19:34:53 +00001190 sqlite3ExprListDelete(pParse->db, pList);
drhe0194f22003-02-26 13:52:51 +00001191 return;
drh4a324312001-12-21 14:30:42 +00001192}
1193
1194/*
drhffe07b22005-11-03 00:41:17 +00001195** Add a new CHECK constraint to the table currently under construction.
1196*/
1197void sqlite3AddCheckConstraint(
1198 Parse *pParse, /* Parsing context */
1199 Expr *pCheckExpr /* The check expression */
1200){
drh633e6d52008-07-28 19:34:53 +00001201 sqlite3 *db = pParse->db;
drhffe07b22005-11-03 00:41:17 +00001202#ifndef SQLITE_OMIT_CHECK
1203 Table *pTab = pParse->pNewTable;
danielk1977c7d54102006-06-15 07:29:00 +00001204 if( pTab && !IN_DECLARE_VTAB ){
drhffe07b22005-11-03 00:41:17 +00001205 /* The CHECK expression must be duplicated so that tokens refer
1206 ** to malloced space and not the (ephemeral) text of the CREATE TABLE
1207 ** statement */
drh17435752007-08-16 04:30:38 +00001208 pTab->pCheck = sqlite3ExprAnd(db, pTab->pCheck,
1209 sqlite3ExprDup(db, pCheckExpr));
drhffe07b22005-11-03 00:41:17 +00001210 }
1211#endif
drh633e6d52008-07-28 19:34:53 +00001212 sqlite3ExprDelete(db, pCheckExpr);
drhffe07b22005-11-03 00:41:17 +00001213}
1214
1215/*
drhd3d39e92004-05-20 22:16:29 +00001216** Set the collation function of the most recently parsed table column
1217** to the CollSeq given.
drh8e2ca022002-06-17 17:07:19 +00001218*/
danielk197739002502007-11-12 09:50:26 +00001219void sqlite3AddCollateType(Parse *pParse, Token *pToken){
drh8e2ca022002-06-17 17:07:19 +00001220 Table *p;
danielk19770202b292004-06-09 09:55:16 +00001221 int i;
danielk197739002502007-11-12 09:50:26 +00001222 char *zColl; /* Dequoted name of collation sequence */
drh633e6d52008-07-28 19:34:53 +00001223 sqlite3 *db;
danielk1977a37cdde2004-05-16 11:15:36 +00001224
danielk1977b8cbb872006-06-19 05:33:45 +00001225 if( (p = pParse->pNewTable)==0 ) return;
danielk19770202b292004-06-09 09:55:16 +00001226 i = p->nCol-1;
drh633e6d52008-07-28 19:34:53 +00001227 db = pParse->db;
1228 zColl = sqlite3NameFromToken(db, pToken);
danielk197739002502007-11-12 09:50:26 +00001229 if( !zColl ) return;
1230
1231 if( sqlite3LocateCollSeq(pParse, zColl, -1) ){
danielk1977b3bf5562006-01-10 17:58:23 +00001232 Index *pIdx;
danielk197739002502007-11-12 09:50:26 +00001233 p->aCol[i].zColl = zColl;
danielk1977b3bf5562006-01-10 17:58:23 +00001234
1235 /* If the column is declared as "<name> PRIMARY KEY COLLATE <type>",
1236 ** then an index may have been created on this column before the
1237 ** collation type was added. Correct this if it is the case.
1238 */
1239 for(pIdx=p->pIndex; pIdx; pIdx=pIdx->pNext){
1240 assert( pIdx->nColumn==1 );
1241 if( pIdx->aiColumn[0]==i ){
1242 pIdx->azColl[0] = p->aCol[i].zColl;
danielk19777cedc8d2004-06-10 10:50:08 +00001243 }
1244 }
danielk197739002502007-11-12 09:50:26 +00001245 }else{
drh633e6d52008-07-28 19:34:53 +00001246 sqlite3DbFree(db, zColl);
danielk19777cedc8d2004-06-10 10:50:08 +00001247 }
danielk19777cedc8d2004-06-10 10:50:08 +00001248}
1249
danielk1977466be562004-06-10 02:16:01 +00001250/*
1251** This function returns the collation sequence for database native text
1252** encoding identified by the string zName, length nName.
1253**
1254** If the requested collation sequence is not available, or not available
1255** in the database native encoding, the collation factory is invoked to
1256** request it. If the collation factory does not supply such a sequence,
1257** and the sequence is available in another text encoding, then that is
1258** returned instead.
1259**
1260** If no versions of the requested collations sequence are available, or
1261** another error occurs, NULL is returned and an error message written into
1262** pParse.
drha34001c2007-02-02 12:44:37 +00001263**
1264** This routine is a wrapper around sqlite3FindCollSeq(). This routine
1265** invokes the collation factory if the named collation cannot be found
1266** and generates an error message.
danielk1977466be562004-06-10 02:16:01 +00001267*/
danielk19770202b292004-06-09 09:55:16 +00001268CollSeq *sqlite3LocateCollSeq(Parse *pParse, const char *zName, int nName){
danielk19774dade032005-05-25 10:45:10 +00001269 sqlite3 *db = pParse->db;
danielk197714db2662006-01-09 16:12:04 +00001270 u8 enc = ENC(db);
danielk19774dade032005-05-25 10:45:10 +00001271 u8 initbusy = db->init.busy;
danielk1977b3bf5562006-01-10 17:58:23 +00001272 CollSeq *pColl;
danielk19774dade032005-05-25 10:45:10 +00001273
danielk1977b3bf5562006-01-10 17:58:23 +00001274 pColl = sqlite3FindCollSeq(db, enc, zName, nName, initbusy);
danielk19777cedc8d2004-06-10 10:50:08 +00001275 if( !initbusy && (!pColl || !pColl->xCmp) ){
danielk19774dade032005-05-25 10:45:10 +00001276 pColl = sqlite3GetCollSeq(db, pColl, zName, nName);
1277 if( !pColl ){
1278 if( nName<0 ){
drh0a687d12008-07-08 14:52:07 +00001279 nName = sqlite3Strlen(db, zName);
danielk1977466be562004-06-10 02:16:01 +00001280 }
danielk19774dade032005-05-25 10:45:10 +00001281 sqlite3ErrorMsg(pParse, "no such collation sequence: %.*s", nName, zName);
1282 pColl = 0;
danielk1977466be562004-06-10 02:16:01 +00001283 }
1284 }
1285
danielk19770202b292004-06-09 09:55:16 +00001286 return pColl;
1287}
1288
1289
drh8e2ca022002-06-17 17:07:19 +00001290/*
drh3f7d4e42004-07-24 14:35:58 +00001291** Generate code that will increment the schema cookie.
drh50e5dad2001-09-15 00:57:28 +00001292**
1293** The schema cookie is used to determine when the schema for the
1294** database changes. After each schema change, the cookie value
1295** changes. When a process first reads the schema it records the
1296** cookie. Thereafter, whenever it goes to access the database,
1297** it checks the cookie to make sure the schema has not changed
1298** since it was last read.
1299**
1300** This plan is not completely bullet-proof. It is possible for
1301** the schema to change multiple times and for the cookie to be
1302** set back to prior value. But schema changes are infrequent
1303** and the probability of hitting the same cookie value is only
1304** 1 chance in 2^32. So we're safe enough.
1305*/
drh9cbf3422008-01-17 16:22:13 +00001306void sqlite3ChangeCookie(Parse *pParse, int iDb){
1307 int r1 = sqlite3GetTempReg(pParse);
1308 sqlite3 *db = pParse->db;
1309 Vdbe *v = pParse->pVdbe;
1310 sqlite3VdbeAddOp2(v, OP_Integer, db->aDb[iDb].pSchema->schema_cookie+1, r1);
1311 sqlite3VdbeAddOp3(v, OP_SetCookie, iDb, 0, r1);
1312 sqlite3ReleaseTempReg(pParse, r1);
drh50e5dad2001-09-15 00:57:28 +00001313}
1314
1315/*
drh969fa7c2002-02-18 18:30:32 +00001316** Measure the number of characters needed to output the given
1317** identifier. The number returned includes any quotes used
1318** but does not include the null terminator.
drh234c39d2004-07-24 03:30:47 +00001319**
1320** The estimate is conservative. It might be larger that what is
1321** really needed.
drh969fa7c2002-02-18 18:30:32 +00001322*/
1323static int identLength(const char *z){
1324 int n;
drh17f71932002-02-21 12:01:27 +00001325 for(n=0; *z; n++, z++){
drh234c39d2004-07-24 03:30:47 +00001326 if( *z=='"' ){ n++; }
drh969fa7c2002-02-18 18:30:32 +00001327 }
drh234c39d2004-07-24 03:30:47 +00001328 return n + 2;
drh969fa7c2002-02-18 18:30:32 +00001329}
1330
1331/*
1332** Write an identifier onto the end of the given string. Add
1333** quote characters as needed.
1334*/
drh4c755c02004-08-08 20:22:17 +00001335static void identPut(char *z, int *pIdx, char *zSignedIdent){
1336 unsigned char *zIdent = (unsigned char*)zSignedIdent;
drh17f71932002-02-21 12:01:27 +00001337 int i, j, needQuote;
drh969fa7c2002-02-18 18:30:32 +00001338 i = *pIdx;
drh17f71932002-02-21 12:01:27 +00001339 for(j=0; zIdent[j]; j++){
1340 if( !isalnum(zIdent[j]) && zIdent[j]!='_' ) break;
1341 }
1342 needQuote = zIdent[j]!=0 || isdigit(zIdent[0])
danielk19774adee202004-05-08 08:23:19 +00001343 || sqlite3KeywordCode(zIdent, j)!=TK_ID;
drh234c39d2004-07-24 03:30:47 +00001344 if( needQuote ) z[i++] = '"';
drh969fa7c2002-02-18 18:30:32 +00001345 for(j=0; zIdent[j]; j++){
1346 z[i++] = zIdent[j];
drh234c39d2004-07-24 03:30:47 +00001347 if( zIdent[j]=='"' ) z[i++] = '"';
drh969fa7c2002-02-18 18:30:32 +00001348 }
drh234c39d2004-07-24 03:30:47 +00001349 if( needQuote ) z[i++] = '"';
drh969fa7c2002-02-18 18:30:32 +00001350 z[i] = 0;
1351 *pIdx = i;
1352}
1353
1354/*
1355** Generate a CREATE TABLE statement appropriate for the given
1356** table. Memory to hold the text of the statement is obtained
1357** from sqliteMalloc() and must be freed by the calling function.
1358*/
drh820a9062008-01-31 13:35:48 +00001359static char *createTableStmt(sqlite3 *db, Table *p, int isTemp){
drh969fa7c2002-02-18 18:30:32 +00001360 int i, k, n;
1361 char *zStmt;
drh234c39d2004-07-24 03:30:47 +00001362 char *zSep, *zSep2, *zEnd, *z;
1363 Column *pCol;
drh969fa7c2002-02-18 18:30:32 +00001364 n = 0;
drh234c39d2004-07-24 03:30:47 +00001365 for(pCol = p->aCol, i=0; i<p->nCol; i++, pCol++){
1366 n += identLength(pCol->zName);
1367 z = pCol->zType;
1368 if( z ){
1369 n += (strlen(z) + 1);
danielk1977517eb642004-06-07 10:00:31 +00001370 }
drh969fa7c2002-02-18 18:30:32 +00001371 }
1372 n += identLength(p->zName);
drh234c39d2004-07-24 03:30:47 +00001373 if( n<50 ){
drh969fa7c2002-02-18 18:30:32 +00001374 zSep = "";
1375 zSep2 = ",";
1376 zEnd = ")";
1377 }else{
1378 zSep = "\n ";
1379 zSep2 = ",\n ";
1380 zEnd = "\n)";
1381 }
drhe0bc4042002-06-25 01:09:11 +00001382 n += 35 + 6*p->nCol;
drhe5ae5732008-06-15 02:51:47 +00001383 zStmt = sqlite3Malloc( n );
drh820a9062008-01-31 13:35:48 +00001384 if( zStmt==0 ){
1385 db->mallocFailed = 1;
1386 return 0;
1387 }
drh5bb3eb92007-05-04 13:15:55 +00001388 sqlite3_snprintf(n, zStmt,
1389 !OMIT_TEMPDB&&isTemp ? "CREATE TEMP TABLE ":"CREATE TABLE ");
drh969fa7c2002-02-18 18:30:32 +00001390 k = strlen(zStmt);
1391 identPut(zStmt, &k, p->zName);
1392 zStmt[k++] = '(';
drh234c39d2004-07-24 03:30:47 +00001393 for(pCol=p->aCol, i=0; i<p->nCol; i++, pCol++){
drh5bb3eb92007-05-04 13:15:55 +00001394 sqlite3_snprintf(n-k, &zStmt[k], zSep);
drh969fa7c2002-02-18 18:30:32 +00001395 k += strlen(&zStmt[k]);
1396 zSep = zSep2;
drh234c39d2004-07-24 03:30:47 +00001397 identPut(zStmt, &k, pCol->zName);
1398 if( (z = pCol->zType)!=0 ){
danielk1977517eb642004-06-07 10:00:31 +00001399 zStmt[k++] = ' ';
drh5bb3eb92007-05-04 13:15:55 +00001400 assert( strlen(z)+k+1<=n );
1401 sqlite3_snprintf(n-k, &zStmt[k], "%s", z);
drh234c39d2004-07-24 03:30:47 +00001402 k += strlen(z);
danielk1977517eb642004-06-07 10:00:31 +00001403 }
drh969fa7c2002-02-18 18:30:32 +00001404 }
drh5bb3eb92007-05-04 13:15:55 +00001405 sqlite3_snprintf(n-k, &zStmt[k], "%s", zEnd);
drh969fa7c2002-02-18 18:30:32 +00001406 return zStmt;
1407}
1408
1409/*
drh75897232000-05-29 14:26:00 +00001410** This routine is called to report the final ")" that terminates
1411** a CREATE TABLE statement.
1412**
drhf57b3392001-10-08 13:22:32 +00001413** The table structure that other action routines have been building
1414** is added to the internal hash tables, assuming no errors have
1415** occurred.
drh75897232000-05-29 14:26:00 +00001416**
drh1d85d932004-02-14 23:05:52 +00001417** An entry for the table is made in the master table on disk, unless
1418** this is a temporary table or db->init.busy==1. When db->init.busy==1
drhf57b3392001-10-08 13:22:32 +00001419** it means we are reading the sqlite_master table because we just
1420** connected to the database or because the sqlite_master table has
drhddba9e52005-03-19 01:41:21 +00001421** recently changed, so the entry for this table already exists in
drhf57b3392001-10-08 13:22:32 +00001422** the sqlite_master table. We do not want to create it again.
drh969fa7c2002-02-18 18:30:32 +00001423**
1424** If the pSelect argument is not NULL, it means that this routine
1425** was called to create a table generated from a
1426** "CREATE TABLE ... AS SELECT ..." statement. The column names of
1427** the new table will match the result set of the SELECT.
drh75897232000-05-29 14:26:00 +00001428*/
danielk197719a8e7e2005-03-17 05:03:38 +00001429void sqlite3EndTable(
1430 Parse *pParse, /* Parse context */
1431 Token *pCons, /* The ',' token after the last column defn. */
1432 Token *pEnd, /* The final ')' token in the CREATE TABLE */
1433 Select *pSelect /* Select from a "CREATE ... AS SELECT" */
1434){
drh75897232000-05-29 14:26:00 +00001435 Table *p;
drh9bb575f2004-09-06 17:24:11 +00001436 sqlite3 *db = pParse->db;
danielk1977da184232006-01-05 11:34:32 +00001437 int iDb;
drh75897232000-05-29 14:26:00 +00001438
drh17435752007-08-16 04:30:38 +00001439 if( (pEnd==0 && pSelect==0) || pParse->nErr || db->mallocFailed ) {
danielk1977261919c2005-12-06 12:52:59 +00001440 return;
1441 }
drh28037572000-08-02 13:47:41 +00001442 p = pParse->pNewTable;
drhdaffd0e2001-04-11 14:28:42 +00001443 if( p==0 ) return;
drh75897232000-05-29 14:26:00 +00001444
danielk1977517eb642004-06-07 10:00:31 +00001445 assert( !db->init.busy || !pSelect );
1446
drhb9bb7c12006-06-11 23:41:55 +00001447 iDb = sqlite3SchemaToIndex(db, p->pSchema);
danielk1977da184232006-01-05 11:34:32 +00001448
drhffe07b22005-11-03 00:41:17 +00001449#ifndef SQLITE_OMIT_CHECK
1450 /* Resolve names in all CHECK constraint expressions.
1451 */
1452 if( p->pCheck ){
1453 SrcList sSrc; /* Fake SrcList for pParse->pNewTable */
1454 NameContext sNC; /* Name context for pParse->pNewTable */
1455
1456 memset(&sNC, 0, sizeof(sNC));
1457 memset(&sSrc, 0, sizeof(sSrc));
1458 sSrc.nSrc = 1;
1459 sSrc.a[0].zName = p->zName;
1460 sSrc.a[0].pTab = p;
1461 sSrc.a[0].iCursor = -1;
1462 sNC.pParse = pParse;
1463 sNC.pSrcList = &sSrc;
drh06f65412005-11-03 02:03:13 +00001464 sNC.isCheck = 1;
drhffe07b22005-11-03 00:41:17 +00001465 if( sqlite3ExprResolveNames(&sNC, p->pCheck) ){
1466 return;
1467 }
1468 }
1469#endif /* !defined(SQLITE_OMIT_CHECK) */
1470
drh1d85d932004-02-14 23:05:52 +00001471 /* If the db->init.busy is 1 it means we are reading the SQL off the
drhe0bc4042002-06-25 01:09:11 +00001472 ** "sqlite_master" or "sqlite_temp_master" table on the disk.
1473 ** So do not write to the disk again. Extract the root page number
drh1d85d932004-02-14 23:05:52 +00001474 ** for the table from the db->init.newTnum field. (The page number
drhe0bc4042002-06-25 01:09:11 +00001475 ** should have been put there by the sqliteOpenCb routine.)
drhd78eeee2001-09-13 16:18:53 +00001476 */
drh1d85d932004-02-14 23:05:52 +00001477 if( db->init.busy ){
1478 p->tnum = db->init.newTnum;
drhd78eeee2001-09-13 16:18:53 +00001479 }
1480
drhe3c41372001-09-17 20:25:58 +00001481 /* If not initializing, then create a record for the new table
drh17f71932002-02-21 12:01:27 +00001482 ** in the SQLITE_MASTER table of the database. The record number
1483 ** for the new table entry should already be on the stack.
drhf57b3392001-10-08 13:22:32 +00001484 **
drhe0bc4042002-06-25 01:09:11 +00001485 ** If this is a TEMPORARY table, write the entry into the auxiliary
1486 ** file instead of into the main database file.
drh75897232000-05-29 14:26:00 +00001487 */
drh1d85d932004-02-14 23:05:52 +00001488 if( !db->init.busy ){
drh4ff6dfa2002-03-03 23:06:00 +00001489 int n;
drhd8bc7082000-06-07 23:51:50 +00001490 Vdbe *v;
drh4794f732004-11-05 17:17:50 +00001491 char *zType; /* "view" or "table" */
1492 char *zType2; /* "VIEW" or "TABLE" */
1493 char *zStmt; /* Text of the CREATE TABLE or CREATE VIEW statement */
drh75897232000-05-29 14:26:00 +00001494
danielk19774adee202004-05-08 08:23:19 +00001495 v = sqlite3GetVdbe(pParse);
drh75897232000-05-29 14:26:00 +00001496 if( v==0 ) return;
danielk1977517eb642004-06-07 10:00:31 +00001497
drh66a51672008-01-03 00:01:23 +00001498 sqlite3VdbeAddOp1(v, OP_Close, 0);
danielk1977e6efa742004-11-10 11:55:10 +00001499
drh4794f732004-11-05 17:17:50 +00001500 /* Create the rootpage for the new table and push it onto the stack.
1501 ** A view has no rootpage, so just push a zero onto the stack for
1502 ** views. Initialize zType at the same time.
1503 */
drh4ff6dfa2002-03-03 23:06:00 +00001504 if( p->pSelect==0 ){
1505 /* A regular table */
drh4794f732004-11-05 17:17:50 +00001506 zType = "table";
1507 zType2 = "TABLE";
danielk1977576ec6b2005-01-21 11:55:25 +00001508#ifndef SQLITE_OMIT_VIEW
drh4ff6dfa2002-03-03 23:06:00 +00001509 }else{
1510 /* A view */
drh4794f732004-11-05 17:17:50 +00001511 zType = "view";
1512 zType2 = "VIEW";
danielk1977576ec6b2005-01-21 11:55:25 +00001513#endif
drh4ff6dfa2002-03-03 23:06:00 +00001514 }
danielk1977517eb642004-06-07 10:00:31 +00001515
danielk1977517eb642004-06-07 10:00:31 +00001516 /* If this is a CREATE TABLE xx AS SELECT ..., execute the SELECT
1517 ** statement to populate the new table. The root-page number for the
1518 ** new table is on the top of the vdbe stack.
1519 **
1520 ** Once the SELECT has been coded by sqlite3Select(), it is in a
1521 ** suitable state to query for the column names and types to be used
1522 ** by the new table.
danielk1977c00da102006-01-07 13:21:04 +00001523 **
1524 ** A shared-cache write-lock is not required to write to the new table,
1525 ** as a schema-lock must have already been obtained to create it. Since
1526 ** a schema-lock excludes all other database users, the write-lock would
1527 ** be redundant.
danielk1977517eb642004-06-07 10:00:31 +00001528 */
1529 if( pSelect ){
drh1013c932008-01-06 00:25:21 +00001530 SelectDest dest;
danielk1977517eb642004-06-07 10:00:31 +00001531 Table *pSelTab;
drh1013c932008-01-06 00:25:21 +00001532
drhb7654112008-01-12 12:48:07 +00001533 sqlite3VdbeAddOp3(v, OP_OpenWrite, 1, pParse->regRoot, iDb);
drh98757152008-01-09 23:04:12 +00001534 sqlite3VdbeChangeP5(v, 1);
danielk1977517eb642004-06-07 10:00:31 +00001535 pParse->nTab = 2;
drh1013c932008-01-06 00:25:21 +00001536 sqlite3SelectDestInit(&dest, SRT_Table, 1);
drha9671a22008-07-08 23:40:20 +00001537 sqlite3Select(pParse, pSelect, &dest, 0, 0, 0);
drh66a51672008-01-03 00:01:23 +00001538 sqlite3VdbeAddOp1(v, OP_Close, 1);
danielk1977517eb642004-06-07 10:00:31 +00001539 if( pParse->nErr==0 ){
1540 pSelTab = sqlite3ResultSetOfSelect(pParse, 0, pSelect);
1541 if( pSelTab==0 ) return;
1542 assert( p->aCol==0 );
1543 p->nCol = pSelTab->nCol;
1544 p->aCol = pSelTab->aCol;
1545 pSelTab->nCol = 0;
1546 pSelTab->aCol = 0;
danielk1977a04a34f2007-04-16 15:06:25 +00001547 sqlite3DeleteTable(pSelTab);
danielk1977517eb642004-06-07 10:00:31 +00001548 }
1549 }
drh4794f732004-11-05 17:17:50 +00001550
drh4794f732004-11-05 17:17:50 +00001551 /* Compute the complete text of the CREATE statement */
1552 if( pSelect ){
drh820a9062008-01-31 13:35:48 +00001553 zStmt = createTableStmt(db, p, p->pSchema==db->aDb[1].pSchema);
drh4794f732004-11-05 17:17:50 +00001554 }else{
drh97903fe2005-05-24 20:19:57 +00001555 n = pEnd->z - pParse->sNameToken.z + 1;
danielk19771e536952007-08-16 10:09:01 +00001556 zStmt = sqlite3MPrintf(db,
1557 "CREATE %s %.*s", zType2, n, pParse->sNameToken.z
1558 );
drh4794f732004-11-05 17:17:50 +00001559 }
1560
1561 /* A slot for the record has already been allocated in the
1562 ** SQLITE_MASTER table. We just need to update that slot with all
1563 ** the information we've collected. The rowid for the preallocated
1564 ** slot is the 2nd item on the stack. The top of the stack is the
1565 ** root page for the new table (or a 0 if this is a view).
1566 */
1567 sqlite3NestedParse(pParse,
1568 "UPDATE %Q.%s "
drhb7654112008-01-12 12:48:07 +00001569 "SET type='%s', name=%Q, tbl_name=%Q, rootpage=#%d, sql=%Q "
1570 "WHERE rowid=#%d",
danielk1977da184232006-01-05 11:34:32 +00001571 db->aDb[iDb].zName, SCHEMA_TABLE(iDb),
drh4794f732004-11-05 17:17:50 +00001572 zType,
1573 p->zName,
1574 p->zName,
drhb7654112008-01-12 12:48:07 +00001575 pParse->regRoot,
1576 zStmt,
1577 pParse->regRowid
drh4794f732004-11-05 17:17:50 +00001578 );
drh633e6d52008-07-28 19:34:53 +00001579 sqlite3DbFree(db, zStmt);
drh9cbf3422008-01-17 16:22:13 +00001580 sqlite3ChangeCookie(pParse, iDb);
drh2958a4e2004-11-12 03:56:15 +00001581
1582#ifndef SQLITE_OMIT_AUTOINCREMENT
1583 /* Check to see if we need to create an sqlite_sequence table for
1584 ** keeping track of autoincrement keys.
1585 */
1586 if( p->autoInc ){
danielk1977da184232006-01-05 11:34:32 +00001587 Db *pDb = &db->aDb[iDb];
1588 if( pDb->pSchema->pSeqTab==0 ){
drh2958a4e2004-11-12 03:56:15 +00001589 sqlite3NestedParse(pParse,
drhf3388142004-11-13 03:48:06 +00001590 "CREATE TABLE %Q.sqlite_sequence(name,seq)",
1591 pDb->zName
drh2958a4e2004-11-12 03:56:15 +00001592 );
1593 }
1594 }
1595#endif
drh4794f732004-11-05 17:17:50 +00001596
1597 /* Reparse everything to update our internal data structures */
drh66a51672008-01-03 00:01:23 +00001598 sqlite3VdbeAddOp4(v, OP_ParseSchema, iDb, 0, 0,
1599 sqlite3MPrintf(db, "tbl_name='%q'",p->zName), P4_DYNAMIC);
drh75897232000-05-29 14:26:00 +00001600 }
drh17e9e292003-02-01 13:53:28 +00001601
drh2958a4e2004-11-12 03:56:15 +00001602
drh17e9e292003-02-01 13:53:28 +00001603 /* Add the table to the in-memory representation of the database.
1604 */
drh234c39d2004-07-24 03:30:47 +00001605 if( db->init.busy && pParse->nErr==0 ){
drh17e9e292003-02-01 13:53:28 +00001606 Table *pOld;
drhbe5c89a2004-07-26 00:31:09 +00001607 FKey *pFKey;
danielk1977e501b892006-01-09 06:29:47 +00001608 Schema *pSchema = p->pSchema;
danielk1977da184232006-01-05 11:34:32 +00001609 pOld = sqlite3HashInsert(&pSchema->tblHash, p->zName, strlen(p->zName)+1,p);
drh17e9e292003-02-01 13:53:28 +00001610 if( pOld ){
1611 assert( p==pOld ); /* Malloc must have failed inside HashInsert() */
drh17435752007-08-16 04:30:38 +00001612 db->mallocFailed = 1;
drh17e9e292003-02-01 13:53:28 +00001613 return;
1614 }
danielk1977576ec6b2005-01-21 11:55:25 +00001615#ifndef SQLITE_OMIT_FOREIGN_KEY
drh17e9e292003-02-01 13:53:28 +00001616 for(pFKey=p->pFKey; pFKey; pFKey=pFKey->pNextFrom){
danielk1977a1644fd2007-08-29 12:31:25 +00001617 void *data;
drh17e9e292003-02-01 13:53:28 +00001618 int nTo = strlen(pFKey->zTo) + 1;
danielk1977da184232006-01-05 11:34:32 +00001619 pFKey->pNextTo = sqlite3HashFind(&pSchema->aFKey, pFKey->zTo, nTo);
danielk1977a1644fd2007-08-29 12:31:25 +00001620 data = sqlite3HashInsert(&pSchema->aFKey, pFKey->zTo, nTo, pFKey);
1621 if( data==(void *)pFKey ){
1622 db->mallocFailed = 1;
1623 }
drh17e9e292003-02-01 13:53:28 +00001624 }
danielk1977576ec6b2005-01-21 11:55:25 +00001625#endif
drh17e9e292003-02-01 13:53:28 +00001626 pParse->pNewTable = 0;
1627 db->nTable++;
1628 db->flags |= SQLITE_InternChanges;
danielk197719a8e7e2005-03-17 05:03:38 +00001629
1630#ifndef SQLITE_OMIT_ALTERTABLE
1631 if( !p->pSelect ){
danielk1977bab45c62006-01-16 15:14:27 +00001632 const char *zName = (const char *)pParse->sNameToken.z;
drh9a087a92007-05-15 14:34:32 +00001633 int nName;
danielk197719a8e7e2005-03-17 05:03:38 +00001634 assert( !pSelect && pCons && pEnd );
danielk1977bab45c62006-01-16 15:14:27 +00001635 if( pCons->z==0 ){
1636 pCons = pEnd;
1637 }
1638 nName = (const char *)pCons->z - zName;
drh9a087a92007-05-15 14:34:32 +00001639 p->addColOffset = 13 + sqlite3Utf8CharLen(zName, nName);
danielk197719a8e7e2005-03-17 05:03:38 +00001640 }
1641#endif
drh17e9e292003-02-01 13:53:28 +00001642 }
drh75897232000-05-29 14:26:00 +00001643}
1644
drhb7f91642004-10-31 02:22:47 +00001645#ifndef SQLITE_OMIT_VIEW
drh75897232000-05-29 14:26:00 +00001646/*
drha76b5df2002-02-23 02:32:10 +00001647** The parser calls this routine in order to create a new VIEW
1648*/
danielk19774adee202004-05-08 08:23:19 +00001649void sqlite3CreateView(
drha76b5df2002-02-23 02:32:10 +00001650 Parse *pParse, /* The parsing context */
1651 Token *pBegin, /* The CREATE token that begins the statement */
danielk197748dec7e2004-05-28 12:33:30 +00001652 Token *pName1, /* The token that holds the name of the view */
1653 Token *pName2, /* The token that holds the name of the view */
drh6276c1c2002-07-08 22:03:32 +00001654 Select *pSelect, /* A SELECT statement that will become the new view */
drhfdd48a72006-09-11 23:45:48 +00001655 int isTemp, /* TRUE for a TEMPORARY view */
1656 int noErr /* Suppress error messages if VIEW already exists */
drha76b5df2002-02-23 02:32:10 +00001657){
drha76b5df2002-02-23 02:32:10 +00001658 Table *p;
drh4b59ab52002-08-24 18:24:51 +00001659 int n;
drh4c755c02004-08-08 20:22:17 +00001660 const unsigned char *z;
drh4b59ab52002-08-24 18:24:51 +00001661 Token sEnd;
drhf26e09c2003-05-31 16:21:12 +00001662 DbFixer sFix;
danielk197748dec7e2004-05-28 12:33:30 +00001663 Token *pName;
danielk1977da184232006-01-05 11:34:32 +00001664 int iDb;
drh17435752007-08-16 04:30:38 +00001665 sqlite3 *db = pParse->db;
drha76b5df2002-02-23 02:32:10 +00001666
drh7c3d64f2005-06-06 15:32:08 +00001667 if( pParse->nVar>0 ){
1668 sqlite3ErrorMsg(pParse, "parameters are not allowed in views");
drh633e6d52008-07-28 19:34:53 +00001669 sqlite3SelectDelete(db, pSelect);
drh7c3d64f2005-06-06 15:32:08 +00001670 return;
1671 }
drhfdd48a72006-09-11 23:45:48 +00001672 sqlite3StartTable(pParse, pName1, pName2, isTemp, 1, 0, noErr);
drha76b5df2002-02-23 02:32:10 +00001673 p = pParse->pNewTable;
drhed6c8672003-01-12 18:02:16 +00001674 if( p==0 || pParse->nErr ){
drh633e6d52008-07-28 19:34:53 +00001675 sqlite3SelectDelete(db, pSelect);
drh417be792002-03-03 18:59:40 +00001676 return;
1677 }
danielk1977ef2cb632004-05-29 02:37:19 +00001678 sqlite3TwoPartName(pParse, pName1, pName2, &pName);
drh17435752007-08-16 04:30:38 +00001679 iDb = sqlite3SchemaToIndex(db, p->pSchema);
danielk1977da184232006-01-05 11:34:32 +00001680 if( sqlite3FixInit(&sFix, pParse, iDb, "view", pName)
danielk19774adee202004-05-08 08:23:19 +00001681 && sqlite3FixSelect(&sFix, pSelect)
drhf26e09c2003-05-31 16:21:12 +00001682 ){
drh633e6d52008-07-28 19:34:53 +00001683 sqlite3SelectDelete(db, pSelect);
drhf26e09c2003-05-31 16:21:12 +00001684 return;
1685 }
drh174b6192002-12-03 02:22:52 +00001686
drh4b59ab52002-08-24 18:24:51 +00001687 /* Make a copy of the entire SELECT statement that defines the view.
1688 ** This will force all the Expr.token.z values to be dynamically
1689 ** allocated rather than point to the input string - which means that
danielk197724b03fd2004-05-10 10:34:34 +00001690 ** they will persist after the current sqlite3_exec() call returns.
drh4b59ab52002-08-24 18:24:51 +00001691 */
drh17435752007-08-16 04:30:38 +00001692 p->pSelect = sqlite3SelectDup(db, pSelect);
drh633e6d52008-07-28 19:34:53 +00001693 sqlite3SelectDelete(db, pSelect);
drh17435752007-08-16 04:30:38 +00001694 if( db->mallocFailed ){
danielk1977261919c2005-12-06 12:52:59 +00001695 return;
1696 }
drh17435752007-08-16 04:30:38 +00001697 if( !db->init.busy ){
danielk19774adee202004-05-08 08:23:19 +00001698 sqlite3ViewGetColumnNames(pParse, p);
drh417be792002-03-03 18:59:40 +00001699 }
drh4b59ab52002-08-24 18:24:51 +00001700
1701 /* Locate the end of the CREATE VIEW statement. Make sEnd point to
1702 ** the end.
1703 */
drha76b5df2002-02-23 02:32:10 +00001704 sEnd = pParse->sLastToken;
1705 if( sEnd.z[0]!=0 && sEnd.z[0]!=';' ){
1706 sEnd.z += sEnd.n;
1707 }
1708 sEnd.n = 0;
drhb089c0b2004-06-26 14:46:39 +00001709 n = sEnd.z - pBegin->z;
drh4c755c02004-08-08 20:22:17 +00001710 z = (const unsigned char*)pBegin->z;
drh4ff6dfa2002-03-03 23:06:00 +00001711 while( n>0 && (z[n-1]==';' || isspace(z[n-1])) ){ n--; }
1712 sEnd.z = &z[n-1];
1713 sEnd.n = 1;
drh4b59ab52002-08-24 18:24:51 +00001714
danielk19774adee202004-05-08 08:23:19 +00001715 /* Use sqlite3EndTable() to add the view to the SQLITE_MASTER table */
danielk197719a8e7e2005-03-17 05:03:38 +00001716 sqlite3EndTable(pParse, 0, &sEnd, 0);
drha76b5df2002-02-23 02:32:10 +00001717 return;
drh417be792002-03-03 18:59:40 +00001718}
drhb7f91642004-10-31 02:22:47 +00001719#endif /* SQLITE_OMIT_VIEW */
drha76b5df2002-02-23 02:32:10 +00001720
danielk1977fe3fcbe22006-06-12 12:08:45 +00001721#if !defined(SQLITE_OMIT_VIEW) || !defined(SQLITE_OMIT_VIRTUALTABLE)
drh417be792002-03-03 18:59:40 +00001722/*
1723** The Table structure pTable is really a VIEW. Fill in the names of
1724** the columns of the view in the pTable structure. Return the number
jplyoncfa56842004-01-19 04:55:56 +00001725** of errors. If an error is seen leave an error message in pParse->zErrMsg.
drh417be792002-03-03 18:59:40 +00001726*/
danielk19774adee202004-05-08 08:23:19 +00001727int sqlite3ViewGetColumnNames(Parse *pParse, Table *pTable){
drh9b3187e2005-01-18 14:45:47 +00001728 Table *pSelTab; /* A fake table from which we get the result set */
1729 Select *pSel; /* Copy of the SELECT that implements the view */
1730 int nErr = 0; /* Number of errors encountered */
1731 int n; /* Temporarily holds the number of cursors assigned */
drh17435752007-08-16 04:30:38 +00001732 sqlite3 *db = pParse->db; /* Database connection for malloc errors */
drha6d0ffc2007-10-12 20:42:28 +00001733 int (*xAuth)(void*,int,const char*,const char*,const char*,const char*);
drh417be792002-03-03 18:59:40 +00001734
1735 assert( pTable );
1736
danielk1977fe3fcbe22006-06-12 12:08:45 +00001737#ifndef SQLITE_OMIT_VIRTUALTABLE
1738 if( sqlite3VtabCallConnect(pParse, pTable) ){
1739 return SQLITE_ERROR;
1740 }
drh4cbdda92006-06-14 19:00:20 +00001741 if( IsVirtual(pTable) ) return 0;
danielk1977fe3fcbe22006-06-12 12:08:45 +00001742#endif
1743
1744#ifndef SQLITE_OMIT_VIEW
drh417be792002-03-03 18:59:40 +00001745 /* A positive nCol means the columns names for this view are
1746 ** already known.
1747 */
1748 if( pTable->nCol>0 ) return 0;
1749
1750 /* A negative nCol is a special marker meaning that we are currently
1751 ** trying to compute the column names. If we enter this routine with
1752 ** a negative nCol, it means two or more views form a loop, like this:
1753 **
1754 ** CREATE VIEW one AS SELECT * FROM two;
1755 ** CREATE VIEW two AS SELECT * FROM one;
drh3b167c72002-06-28 12:18:47 +00001756 **
1757 ** Actually, this error is caught previously and so the following test
1758 ** should always fail. But we will leave it in place just to be safe.
drh417be792002-03-03 18:59:40 +00001759 */
1760 if( pTable->nCol<0 ){
danielk19774adee202004-05-08 08:23:19 +00001761 sqlite3ErrorMsg(pParse, "view %s is circularly defined", pTable->zName);
drh417be792002-03-03 18:59:40 +00001762 return 1;
1763 }
drh85c23c62005-08-20 03:03:04 +00001764 assert( pTable->nCol>=0 );
drh417be792002-03-03 18:59:40 +00001765
1766 /* If we get this far, it means we need to compute the table names.
drh9b3187e2005-01-18 14:45:47 +00001767 ** Note that the call to sqlite3ResultSetOfSelect() will expand any
1768 ** "*" elements in the results set of the view and will assign cursors
1769 ** to the elements of the FROM clause. But we do not want these changes
1770 ** to be permanent. So the computation is done on a copy of the SELECT
1771 ** statement that defines the view.
drh417be792002-03-03 18:59:40 +00001772 */
drh9b3187e2005-01-18 14:45:47 +00001773 assert( pTable->pSelect );
drh17435752007-08-16 04:30:38 +00001774 pSel = sqlite3SelectDup(db, pTable->pSelect);
danielk1977261919c2005-12-06 12:52:59 +00001775 if( pSel ){
1776 n = pParse->nTab;
1777 sqlite3SrcListAssignCursors(pParse, pSel->pSrc);
1778 pTable->nCol = -1;
danielk1977db2d2862007-10-15 07:08:44 +00001779#ifndef SQLITE_OMIT_AUTHORIZATION
drha6d0ffc2007-10-12 20:42:28 +00001780 xAuth = db->xAuth;
1781 db->xAuth = 0;
danielk1977261919c2005-12-06 12:52:59 +00001782 pSelTab = sqlite3ResultSetOfSelect(pParse, 0, pSel);
drha6d0ffc2007-10-12 20:42:28 +00001783 db->xAuth = xAuth;
danielk1977db2d2862007-10-15 07:08:44 +00001784#else
1785 pSelTab = sqlite3ResultSetOfSelect(pParse, 0, pSel);
1786#endif
danielk1977261919c2005-12-06 12:52:59 +00001787 pParse->nTab = n;
1788 if( pSelTab ){
1789 assert( pTable->aCol==0 );
1790 pTable->nCol = pSelTab->nCol;
1791 pTable->aCol = pSelTab->aCol;
1792 pSelTab->nCol = 0;
1793 pSelTab->aCol = 0;
danielk1977a04a34f2007-04-16 15:06:25 +00001794 sqlite3DeleteTable(pSelTab);
danielk1977da184232006-01-05 11:34:32 +00001795 pTable->pSchema->flags |= DB_UnresetViews;
danielk1977261919c2005-12-06 12:52:59 +00001796 }else{
1797 pTable->nCol = 0;
1798 nErr++;
1799 }
drh633e6d52008-07-28 19:34:53 +00001800 sqlite3SelectDelete(db, pSel);
danielk1977261919c2005-12-06 12:52:59 +00001801 } else {
drh417be792002-03-03 18:59:40 +00001802 nErr++;
1803 }
drhb7f91642004-10-31 02:22:47 +00001804#endif /* SQLITE_OMIT_VIEW */
danielk19774b2688a2006-06-20 11:01:07 +00001805 return nErr;
danielk1977fe3fcbe22006-06-12 12:08:45 +00001806}
1807#endif /* !defined(SQLITE_OMIT_VIEW) || !defined(SQLITE_OMIT_VIRTUALTABLE) */
drh417be792002-03-03 18:59:40 +00001808
drhb7f91642004-10-31 02:22:47 +00001809#ifndef SQLITE_OMIT_VIEW
drh417be792002-03-03 18:59:40 +00001810/*
drh8bf8dc92003-05-17 17:35:10 +00001811** Clear the column names from every VIEW in database idx.
drh417be792002-03-03 18:59:40 +00001812*/
drh9bb575f2004-09-06 17:24:11 +00001813static void sqliteViewResetAll(sqlite3 *db, int idx){
drh417be792002-03-03 18:59:40 +00001814 HashElem *i;
drh8bf8dc92003-05-17 17:35:10 +00001815 if( !DbHasProperty(db, idx, DB_UnresetViews) ) return;
danielk1977da184232006-01-05 11:34:32 +00001816 for(i=sqliteHashFirst(&db->aDb[idx].pSchema->tblHash); i;i=sqliteHashNext(i)){
drh417be792002-03-03 18:59:40 +00001817 Table *pTab = sqliteHashData(i);
1818 if( pTab->pSelect ){
drh956bc922004-07-24 17:38:29 +00001819 sqliteResetColumnNames(pTab);
drh417be792002-03-03 18:59:40 +00001820 }
1821 }
drh8bf8dc92003-05-17 17:35:10 +00001822 DbClearProperty(db, idx, DB_UnresetViews);
drha76b5df2002-02-23 02:32:10 +00001823}
drhb7f91642004-10-31 02:22:47 +00001824#else
1825# define sqliteViewResetAll(A,B)
1826#endif /* SQLITE_OMIT_VIEW */
drha76b5df2002-02-23 02:32:10 +00001827
drh75897232000-05-29 14:26:00 +00001828/*
danielk1977a0bf2652004-11-04 14:30:04 +00001829** This function is called by the VDBE to adjust the internal schema
1830** used by SQLite when the btree layer moves a table root page. The
1831** root-page of a table or index in database iDb has changed from iFrom
1832** to iTo.
drh6205d4a2006-03-24 03:36:26 +00001833**
1834** Ticket #1728: The symbol table might still contain information
1835** on tables and/or indices that are the process of being deleted.
1836** If you are unlucky, one of those deleted indices or tables might
1837** have the same rootpage number as the real table or index that is
1838** being moved. So we cannot stop searching after the first match
1839** because the first match might be for one of the deleted indices
1840** or tables and not the table/index that is actually being moved.
1841** We must continue looping until all tables and indices with
1842** rootpage==iFrom have been converted to have a rootpage of iTo
1843** in order to be certain that we got the right one.
danielk1977a0bf2652004-11-04 14:30:04 +00001844*/
1845#ifndef SQLITE_OMIT_AUTOVACUUM
1846void sqlite3RootPageMoved(Db *pDb, int iFrom, int iTo){
1847 HashElem *pElem;
danielk1977da184232006-01-05 11:34:32 +00001848 Hash *pHash;
1849
1850 pHash = &pDb->pSchema->tblHash;
1851 for(pElem=sqliteHashFirst(pHash); pElem; pElem=sqliteHashNext(pElem)){
danielk1977a0bf2652004-11-04 14:30:04 +00001852 Table *pTab = sqliteHashData(pElem);
1853 if( pTab->tnum==iFrom ){
1854 pTab->tnum = iTo;
danielk1977a0bf2652004-11-04 14:30:04 +00001855 }
1856 }
danielk1977da184232006-01-05 11:34:32 +00001857 pHash = &pDb->pSchema->idxHash;
1858 for(pElem=sqliteHashFirst(pHash); pElem; pElem=sqliteHashNext(pElem)){
danielk1977a0bf2652004-11-04 14:30:04 +00001859 Index *pIdx = sqliteHashData(pElem);
1860 if( pIdx->tnum==iFrom ){
1861 pIdx->tnum = iTo;
danielk1977a0bf2652004-11-04 14:30:04 +00001862 }
1863 }
danielk1977a0bf2652004-11-04 14:30:04 +00001864}
1865#endif
1866
1867/*
1868** Write code to erase the table with root-page iTable from database iDb.
1869** Also write code to modify the sqlite_master table and internal schema
1870** if a root-page of another table is moved by the btree-layer whilst
1871** erasing iTable (this can happen with an auto-vacuum database).
1872*/
drh4e0cff62004-11-05 05:10:28 +00001873static void destroyRootPage(Parse *pParse, int iTable, int iDb){
1874 Vdbe *v = sqlite3GetVdbe(pParse);
drhb7654112008-01-12 12:48:07 +00001875 int r1 = sqlite3GetTempReg(pParse);
1876 sqlite3VdbeAddOp3(v, OP_Destroy, iTable, r1, iDb);
drh40e016e2004-11-04 14:47:11 +00001877#ifndef SQLITE_OMIT_AUTOVACUUM
drhb7654112008-01-12 12:48:07 +00001878 /* OP_Destroy stores an in integer r1. If this integer
drh4e0cff62004-11-05 05:10:28 +00001879 ** is non-zero, then it is the root page number of a table moved to
drh81db88e2004-12-07 12:29:17 +00001880 ** location iTable. The following code modifies the sqlite_master table to
drh4e0cff62004-11-05 05:10:28 +00001881 ** reflect this.
1882 **
drhb7654112008-01-12 12:48:07 +00001883 ** The "#%d" in the SQL is a special constant that means whatever value
drh4e0cff62004-11-05 05:10:28 +00001884 ** is on the top of the stack. See sqlite3RegisterExpr().
1885 */
danielk197763e3e9f2004-11-05 09:19:27 +00001886 sqlite3NestedParse(pParse,
drhb7654112008-01-12 12:48:07 +00001887 "UPDATE %Q.%s SET rootpage=%d WHERE #%d AND rootpage=#%d",
1888 pParse->db->aDb[iDb].zName, SCHEMA_TABLE(iDb), iTable, r1, r1);
danielk1977a0bf2652004-11-04 14:30:04 +00001889#endif
drhb7654112008-01-12 12:48:07 +00001890 sqlite3ReleaseTempReg(pParse, r1);
danielk1977a0bf2652004-11-04 14:30:04 +00001891}
1892
1893/*
1894** Write VDBE code to erase table pTab and all associated indices on disk.
1895** Code to update the sqlite_master tables and internal schema definitions
1896** in case a root-page belonging to another table is moved by the btree layer
1897** is also added (this can happen with an auto-vacuum database).
1898*/
drh4e0cff62004-11-05 05:10:28 +00001899static void destroyTable(Parse *pParse, Table *pTab){
danielk1977a0bf2652004-11-04 14:30:04 +00001900#ifdef SQLITE_OMIT_AUTOVACUUM
drheee46cf2004-11-06 00:02:48 +00001901 Index *pIdx;
drh29c636b2006-01-09 23:40:25 +00001902 int iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema);
1903 destroyRootPage(pParse, pTab->tnum, iDb);
danielk1977a0bf2652004-11-04 14:30:04 +00001904 for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){
drh29c636b2006-01-09 23:40:25 +00001905 destroyRootPage(pParse, pIdx->tnum, iDb);
danielk1977a0bf2652004-11-04 14:30:04 +00001906 }
1907#else
1908 /* If the database may be auto-vacuum capable (if SQLITE_OMIT_AUTOVACUUM
1909 ** is not defined), then it is important to call OP_Destroy on the
1910 ** table and index root-pages in order, starting with the numerically
1911 ** largest root-page number. This guarantees that none of the root-pages
1912 ** to be destroyed is relocated by an earlier OP_Destroy. i.e. if the
1913 ** following were coded:
1914 **
1915 ** OP_Destroy 4 0
1916 ** ...
1917 ** OP_Destroy 5 0
1918 **
1919 ** and root page 5 happened to be the largest root-page number in the
1920 ** database, then root page 5 would be moved to page 4 by the
1921 ** "OP_Destroy 4 0" opcode. The subsequent "OP_Destroy 5 0" would hit
1922 ** a free-list page.
1923 */
1924 int iTab = pTab->tnum;
1925 int iDestroyed = 0;
1926
1927 while( 1 ){
1928 Index *pIdx;
1929 int iLargest = 0;
1930
1931 if( iDestroyed==0 || iTab<iDestroyed ){
1932 iLargest = iTab;
1933 }
1934 for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){
1935 int iIdx = pIdx->tnum;
danielk1977da184232006-01-05 11:34:32 +00001936 assert( pIdx->pSchema==pTab->pSchema );
danielk1977a0bf2652004-11-04 14:30:04 +00001937 if( (iDestroyed==0 || (iIdx<iDestroyed)) && iIdx>iLargest ){
1938 iLargest = iIdx;
1939 }
1940 }
danielk1977da184232006-01-05 11:34:32 +00001941 if( iLargest==0 ){
1942 return;
1943 }else{
1944 int iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema);
1945 destroyRootPage(pParse, iLargest, iDb);
1946 iDestroyed = iLargest;
1947 }
danielk1977a0bf2652004-11-04 14:30:04 +00001948 }
1949#endif
1950}
1951
1952/*
drh75897232000-05-29 14:26:00 +00001953** This routine is called to do the work of a DROP TABLE statement.
drhd9b02572001-04-15 00:37:09 +00001954** pName is the name of the table to be dropped.
drh75897232000-05-29 14:26:00 +00001955*/
drha0733842005-12-29 01:11:36 +00001956void sqlite3DropTable(Parse *pParse, SrcList *pName, int isView, int noErr){
danielk1977a8858102004-05-28 12:11:21 +00001957 Table *pTab;
drh75897232000-05-29 14:26:00 +00001958 Vdbe *v;
drh9bb575f2004-09-06 17:24:11 +00001959 sqlite3 *db = pParse->db;
drhd24cc422003-03-27 12:51:24 +00001960 int iDb;
drh75897232000-05-29 14:26:00 +00001961
drh17435752007-08-16 04:30:38 +00001962 if( pParse->nErr || db->mallocFailed ){
drh6f7adc82006-01-11 21:41:20 +00001963 goto exit_drop_table;
1964 }
danielk1977a8858102004-05-28 12:11:21 +00001965 assert( pName->nSrc==1 );
drhca424112008-01-25 15:04:48 +00001966 pTab = sqlite3LocateTable(pParse, isView,
1967 pName->a[0].zName, pName->a[0].zDatabase);
danielk1977a8858102004-05-28 12:11:21 +00001968
drha0733842005-12-29 01:11:36 +00001969 if( pTab==0 ){
1970 if( noErr ){
1971 sqlite3ErrorClear(pParse);
1972 }
1973 goto exit_drop_table;
1974 }
danielk1977da184232006-01-05 11:34:32 +00001975 iDb = sqlite3SchemaToIndex(db, pTab->pSchema);
drhe22a3342003-04-22 20:30:37 +00001976 assert( iDb>=0 && iDb<db->nDb );
danielk1977b5258c32007-10-04 18:11:15 +00001977
1978 /* If pTab is a virtual table, call ViewGetColumnNames() to ensure
1979 ** it is initialized.
1980 */
1981 if( IsVirtual(pTab) && sqlite3ViewGetColumnNames(pParse, pTab) ){
1982 goto exit_drop_table;
1983 }
drhe5f9c642003-01-13 23:27:31 +00001984#ifndef SQLITE_OMIT_AUTHORIZATION
drhe5f9c642003-01-13 23:27:31 +00001985 {
1986 int code;
danielk1977da184232006-01-05 11:34:32 +00001987 const char *zTab = SCHEMA_TABLE(iDb);
1988 const char *zDb = db->aDb[iDb].zName;
danielk1977f1a381e2006-06-16 08:01:02 +00001989 const char *zArg2 = 0;
danielk19774adee202004-05-08 08:23:19 +00001990 if( sqlite3AuthCheck(pParse, SQLITE_DELETE, zTab, 0, zDb)){
danielk1977a8858102004-05-28 12:11:21 +00001991 goto exit_drop_table;
drhe22a3342003-04-22 20:30:37 +00001992 }
drhe5f9c642003-01-13 23:27:31 +00001993 if( isView ){
danielk197753c0f742005-03-29 03:10:59 +00001994 if( !OMIT_TEMPDB && iDb==1 ){
drhe5f9c642003-01-13 23:27:31 +00001995 code = SQLITE_DROP_TEMP_VIEW;
1996 }else{
1997 code = SQLITE_DROP_VIEW;
1998 }
danielk19774b2688a2006-06-20 11:01:07 +00001999#ifndef SQLITE_OMIT_VIRTUALTABLE
danielk1977f1a381e2006-06-16 08:01:02 +00002000 }else if( IsVirtual(pTab) ){
2001 code = SQLITE_DROP_VTABLE;
2002 zArg2 = pTab->pMod->zName;
danielk19774b2688a2006-06-20 11:01:07 +00002003#endif
drhe5f9c642003-01-13 23:27:31 +00002004 }else{
danielk197753c0f742005-03-29 03:10:59 +00002005 if( !OMIT_TEMPDB && iDb==1 ){
drhe5f9c642003-01-13 23:27:31 +00002006 code = SQLITE_DROP_TEMP_TABLE;
2007 }else{
2008 code = SQLITE_DROP_TABLE;
2009 }
2010 }
danielk1977f1a381e2006-06-16 08:01:02 +00002011 if( sqlite3AuthCheck(pParse, code, pTab->zName, zArg2, zDb) ){
danielk1977a8858102004-05-28 12:11:21 +00002012 goto exit_drop_table;
drhe5f9c642003-01-13 23:27:31 +00002013 }
danielk1977a8858102004-05-28 12:11:21 +00002014 if( sqlite3AuthCheck(pParse, SQLITE_DELETE, pTab->zName, 0, zDb) ){
2015 goto exit_drop_table;
drh77ad4e42003-01-14 02:49:27 +00002016 }
drhe5f9c642003-01-13 23:27:31 +00002017 }
2018#endif
danielk1977da184232006-01-05 11:34:32 +00002019 if( pTab->readOnly || pTab==db->aDb[iDb].pSchema->pSeqTab ){
danielk1977a8858102004-05-28 12:11:21 +00002020 sqlite3ErrorMsg(pParse, "table %s may not be dropped", pTab->zName);
danielk1977a8858102004-05-28 12:11:21 +00002021 goto exit_drop_table;
drh75897232000-05-29 14:26:00 +00002022 }
danielk1977576ec6b2005-01-21 11:55:25 +00002023
2024#ifndef SQLITE_OMIT_VIEW
2025 /* Ensure DROP TABLE is not used on a view, and DROP VIEW is not used
2026 ** on a table.
2027 */
danielk1977a8858102004-05-28 12:11:21 +00002028 if( isView && pTab->pSelect==0 ){
2029 sqlite3ErrorMsg(pParse, "use DROP TABLE to delete table %s", pTab->zName);
2030 goto exit_drop_table;
drh4ff6dfa2002-03-03 23:06:00 +00002031 }
danielk1977a8858102004-05-28 12:11:21 +00002032 if( !isView && pTab->pSelect ){
2033 sqlite3ErrorMsg(pParse, "use DROP VIEW to delete view %s", pTab->zName);
2034 goto exit_drop_table;
drh4ff6dfa2002-03-03 23:06:00 +00002035 }
danielk1977576ec6b2005-01-21 11:55:25 +00002036#endif
drh75897232000-05-29 14:26:00 +00002037
drh1ccde152000-06-17 13:12:39 +00002038 /* Generate code to remove the table from the master table
2039 ** on disk.
2040 */
danielk19774adee202004-05-08 08:23:19 +00002041 v = sqlite3GetVdbe(pParse);
drh75897232000-05-29 14:26:00 +00002042 if( v ){
drhe0bc4042002-06-25 01:09:11 +00002043 Trigger *pTrigger;
drh2958a4e2004-11-12 03:56:15 +00002044 Db *pDb = &db->aDb[iDb];
drh77658e22007-12-04 16:54:52 +00002045 sqlite3BeginWriteOperation(pParse, 1, iDb);
drh8bf8dc92003-05-17 17:35:10 +00002046
danielk197720b1eaf2006-07-26 16:22:14 +00002047#ifndef SQLITE_OMIT_VIRTUALTABLE
2048 if( IsVirtual(pTab) ){
2049 Vdbe *v = sqlite3GetVdbe(pParse);
2050 if( v ){
drh66a51672008-01-03 00:01:23 +00002051 sqlite3VdbeAddOp0(v, OP_VBegin);
danielk197720b1eaf2006-07-26 16:22:14 +00002052 }
2053 }
2054#endif
2055
danielk19778e227872004-06-07 07:52:17 +00002056 /* Drop all triggers associated with the table being dropped. Code
2057 ** is generated to remove entries from sqlite_master and/or
2058 ** sqlite_temp_master if required.
2059 */
danielk1977a8858102004-05-28 12:11:21 +00002060 pTrigger = pTab->pTrigger;
drhe0bc4042002-06-25 01:09:11 +00002061 while( pTrigger ){
danielk1977da184232006-01-05 11:34:32 +00002062 assert( pTrigger->pSchema==pTab->pSchema ||
2063 pTrigger->pSchema==db->aDb[1].pSchema );
drh74161702006-02-24 02:53:49 +00002064 sqlite3DropTriggerPtr(pParse, pTrigger);
drh956bc922004-07-24 17:38:29 +00002065 pTrigger = pTrigger->pNext;
danielk1977c3f9bad2002-05-15 08:30:12 +00002066 }
drh8bf8dc92003-05-17 17:35:10 +00002067
danielk19774d36b812004-11-19 07:07:30 +00002068#ifndef SQLITE_OMIT_AUTOINCREMENT
2069 /* Remove any entries of the sqlite_sequence table associated with
2070 ** the table being dropped. This is done before the table is dropped
2071 ** at the btree level, in case the sqlite_sequence table needs to
2072 ** move as a result of the drop (can happen in auto-vacuum mode).
2073 */
2074 if( pTab->autoInc ){
2075 sqlite3NestedParse(pParse,
2076 "DELETE FROM %s.sqlite_sequence WHERE name=%Q",
2077 pDb->zName, pTab->zName
2078 );
2079 }
2080#endif
2081
danielk19778e227872004-06-07 07:52:17 +00002082 /* Drop all SQLITE_MASTER table and index entries that refer to the
2083 ** table. The program name loops through the master table and deletes
2084 ** every row that refers to a table of the same name as the one being
2085 ** dropped. Triggers are handled seperately because a trigger can be
2086 ** created in the temp database that refers to a table in another
2087 ** database.
2088 */
drhf1974842004-11-05 03:56:00 +00002089 sqlite3NestedParse(pParse,
drh4794f732004-11-05 17:17:50 +00002090 "DELETE FROM %Q.%s WHERE tbl_name=%Q and type!='trigger'",
drh2958a4e2004-11-12 03:56:15 +00002091 pDb->zName, SCHEMA_TABLE(iDb), pTab->zName);
danielk1977006015d2008-04-11 17:11:26 +00002092
2093 /* Drop any statistics from the sqlite_stat1 table, if it exists */
2094 if( sqlite3FindTable(db, "sqlite_stat1", db->aDb[iDb].zName) ){
2095 sqlite3NestedParse(pParse,
2096 "DELETE FROM %Q.sqlite_stat1 WHERE tbl=%Q", pDb->zName, pTab->zName
2097 );
2098 }
2099
danielk19774b2688a2006-06-20 11:01:07 +00002100 if( !isView && !IsVirtual(pTab) ){
drh4e0cff62004-11-05 05:10:28 +00002101 destroyTable(pParse, pTab);
drh5e00f6c2001-09-13 13:46:56 +00002102 }
drh2958a4e2004-11-12 03:56:15 +00002103
danielk1977a21c6b62005-01-24 10:25:59 +00002104 /* Remove the table entry from SQLite's internal schema and modify
2105 ** the schema cookie.
drh2958a4e2004-11-12 03:56:15 +00002106 */
drh4cbdda92006-06-14 19:00:20 +00002107 if( IsVirtual(pTab) ){
drh66a51672008-01-03 00:01:23 +00002108 sqlite3VdbeAddOp4(v, OP_VDestroy, iDb, 0, 0, pTab->zName, 0);
drhb9bb7c12006-06-11 23:41:55 +00002109 }
drh66a51672008-01-03 00:01:23 +00002110 sqlite3VdbeAddOp4(v, OP_DropTable, iDb, 0, 0, pTab->zName, 0);
drh9cbf3422008-01-17 16:22:13 +00002111 sqlite3ChangeCookie(pParse, iDb);
drh75897232000-05-29 14:26:00 +00002112 }
drhd24cc422003-03-27 12:51:24 +00002113 sqliteViewResetAll(db, iDb);
danielk1977a8858102004-05-28 12:11:21 +00002114
2115exit_drop_table:
drh633e6d52008-07-28 19:34:53 +00002116 sqlite3SrcListDelete(db, pName);
drh75897232000-05-29 14:26:00 +00002117}
2118
2119/*
drhc2eef3b2002-08-31 18:53:06 +00002120** This routine is called to create a new foreign key on the table
2121** currently under construction. pFromCol determines which columns
2122** in the current table point to the foreign key. If pFromCol==0 then
2123** connect the key to the last column inserted. pTo is the name of
2124** the table referred to. pToCol is a list of tables in the other
2125** pTo table that the foreign key points to. flags contains all
2126** information about the conflict resolution algorithms specified
2127** in the ON DELETE, ON UPDATE and ON INSERT clauses.
2128**
2129** An FKey structure is created and added to the table currently
2130** under construction in the pParse->pNewTable field. The new FKey
2131** is not linked into db->aFKey at this point - that does not happen
danielk19774adee202004-05-08 08:23:19 +00002132** until sqlite3EndTable().
drhc2eef3b2002-08-31 18:53:06 +00002133**
2134** The foreign key is set for IMMEDIATE processing. A subsequent call
danielk19774adee202004-05-08 08:23:19 +00002135** to sqlite3DeferForeignKey() might change this to DEFERRED.
drhc2eef3b2002-08-31 18:53:06 +00002136*/
danielk19774adee202004-05-08 08:23:19 +00002137void sqlite3CreateForeignKey(
drhc2eef3b2002-08-31 18:53:06 +00002138 Parse *pParse, /* Parsing context */
danielk19770202b292004-06-09 09:55:16 +00002139 ExprList *pFromCol, /* Columns in this table that point to other table */
drhc2eef3b2002-08-31 18:53:06 +00002140 Token *pTo, /* Name of the other table */
danielk19770202b292004-06-09 09:55:16 +00002141 ExprList *pToCol, /* Columns in the other table */
drhc2eef3b2002-08-31 18:53:06 +00002142 int flags /* Conflict resolution algorithms. */
2143){
drhb7f91642004-10-31 02:22:47 +00002144#ifndef SQLITE_OMIT_FOREIGN_KEY
drh40e016e2004-11-04 14:47:11 +00002145 FKey *pFKey = 0;
drhc2eef3b2002-08-31 18:53:06 +00002146 Table *p = pParse->pNewTable;
2147 int nByte;
2148 int i;
2149 int nCol;
2150 char *z;
drh633e6d52008-07-28 19:34:53 +00002151 sqlite3 *db;
drhc2eef3b2002-08-31 18:53:06 +00002152
2153 assert( pTo!=0 );
drh633e6d52008-07-28 19:34:53 +00002154 db = pParse->db;
danielk1977c7d54102006-06-15 07:29:00 +00002155 if( p==0 || pParse->nErr || IN_DECLARE_VTAB ) goto fk_end;
drhc2eef3b2002-08-31 18:53:06 +00002156 if( pFromCol==0 ){
2157 int iCol = p->nCol-1;
2158 if( iCol<0 ) goto fk_end;
danielk19770202b292004-06-09 09:55:16 +00002159 if( pToCol && pToCol->nExpr!=1 ){
danielk19774adee202004-05-08 08:23:19 +00002160 sqlite3ErrorMsg(pParse, "foreign key on %s"
drhf7a9e1a2004-02-22 18:40:56 +00002161 " should reference only one column of table %T",
2162 p->aCol[iCol].zName, pTo);
drhc2eef3b2002-08-31 18:53:06 +00002163 goto fk_end;
2164 }
2165 nCol = 1;
danielk19770202b292004-06-09 09:55:16 +00002166 }else if( pToCol && pToCol->nExpr!=pFromCol->nExpr ){
danielk19774adee202004-05-08 08:23:19 +00002167 sqlite3ErrorMsg(pParse,
drhc2eef3b2002-08-31 18:53:06 +00002168 "number of columns in foreign key does not match the number of "
drhf7a9e1a2004-02-22 18:40:56 +00002169 "columns in the referenced table");
drhc2eef3b2002-08-31 18:53:06 +00002170 goto fk_end;
2171 }else{
danielk19770202b292004-06-09 09:55:16 +00002172 nCol = pFromCol->nExpr;
drhc2eef3b2002-08-31 18:53:06 +00002173 }
2174 nByte = sizeof(*pFKey) + nCol*sizeof(pFKey->aCol[0]) + pTo->n + 1;
2175 if( pToCol ){
danielk19770202b292004-06-09 09:55:16 +00002176 for(i=0; i<pToCol->nExpr; i++){
drhc2eef3b2002-08-31 18:53:06 +00002177 nByte += strlen(pToCol->a[i].zName) + 1;
2178 }
2179 }
drh633e6d52008-07-28 19:34:53 +00002180 pFKey = sqlite3DbMallocZero(db, nByte );
drh17435752007-08-16 04:30:38 +00002181 if( pFKey==0 ){
2182 goto fk_end;
2183 }
drhc2eef3b2002-08-31 18:53:06 +00002184 pFKey->pFrom = p;
2185 pFKey->pNextFrom = p->pFKey;
drhdf68f6b2002-09-21 15:57:57 +00002186 z = (char*)&pFKey[1];
2187 pFKey->aCol = (struct sColMap*)z;
2188 z += sizeof(struct sColMap)*nCol;
2189 pFKey->zTo = z;
drhc2eef3b2002-08-31 18:53:06 +00002190 memcpy(z, pTo->z, pTo->n);
2191 z[pTo->n] = 0;
2192 z += pTo->n+1;
2193 pFKey->pNextTo = 0;
2194 pFKey->nCol = nCol;
drhc2eef3b2002-08-31 18:53:06 +00002195 if( pFromCol==0 ){
2196 pFKey->aCol[0].iFrom = p->nCol-1;
2197 }else{
2198 for(i=0; i<nCol; i++){
2199 int j;
2200 for(j=0; j<p->nCol; j++){
danielk19774adee202004-05-08 08:23:19 +00002201 if( sqlite3StrICmp(p->aCol[j].zName, pFromCol->a[i].zName)==0 ){
drhc2eef3b2002-08-31 18:53:06 +00002202 pFKey->aCol[i].iFrom = j;
2203 break;
2204 }
2205 }
2206 if( j>=p->nCol ){
danielk19774adee202004-05-08 08:23:19 +00002207 sqlite3ErrorMsg(pParse,
drhf7a9e1a2004-02-22 18:40:56 +00002208 "unknown column \"%s\" in foreign key definition",
2209 pFromCol->a[i].zName);
drhc2eef3b2002-08-31 18:53:06 +00002210 goto fk_end;
2211 }
2212 }
2213 }
2214 if( pToCol ){
2215 for(i=0; i<nCol; i++){
2216 int n = strlen(pToCol->a[i].zName);
2217 pFKey->aCol[i].zCol = z;
2218 memcpy(z, pToCol->a[i].zName, n);
2219 z[n] = 0;
2220 z += n+1;
2221 }
2222 }
2223 pFKey->isDeferred = 0;
2224 pFKey->deleteConf = flags & 0xff;
2225 pFKey->updateConf = (flags >> 8 ) & 0xff;
2226 pFKey->insertConf = (flags >> 16 ) & 0xff;
2227
2228 /* Link the foreign key to the table as the last step.
2229 */
2230 p->pFKey = pFKey;
2231 pFKey = 0;
2232
2233fk_end:
drh633e6d52008-07-28 19:34:53 +00002234 sqlite3DbFree(db, pFKey);
drhb7f91642004-10-31 02:22:47 +00002235#endif /* !defined(SQLITE_OMIT_FOREIGN_KEY) */
drh633e6d52008-07-28 19:34:53 +00002236 sqlite3ExprListDelete(db, pFromCol);
2237 sqlite3ExprListDelete(db, pToCol);
drhc2eef3b2002-08-31 18:53:06 +00002238}
2239
2240/*
2241** This routine is called when an INITIALLY IMMEDIATE or INITIALLY DEFERRED
2242** clause is seen as part of a foreign key definition. The isDeferred
2243** parameter is 1 for INITIALLY DEFERRED and 0 for INITIALLY IMMEDIATE.
2244** The behavior of the most recently created foreign key is adjusted
2245** accordingly.
2246*/
danielk19774adee202004-05-08 08:23:19 +00002247void sqlite3DeferForeignKey(Parse *pParse, int isDeferred){
drhb7f91642004-10-31 02:22:47 +00002248#ifndef SQLITE_OMIT_FOREIGN_KEY
drhc2eef3b2002-08-31 18:53:06 +00002249 Table *pTab;
2250 FKey *pFKey;
2251 if( (pTab = pParse->pNewTable)==0 || (pFKey = pTab->pFKey)==0 ) return;
2252 pFKey->isDeferred = isDeferred;
drhb7f91642004-10-31 02:22:47 +00002253#endif
drhc2eef3b2002-08-31 18:53:06 +00002254}
2255
2256/*
drh063336a2004-11-05 20:58:39 +00002257** Generate code that will erase and refill index *pIdx. This is
2258** used to initialize a newly created index or to recompute the
2259** content of an index in response to a REINDEX command.
2260**
2261** if memRootPage is not negative, it means that the index is newly
drh1db639c2008-01-17 02:36:28 +00002262** created. The register specified by memRootPage contains the
drh063336a2004-11-05 20:58:39 +00002263** root page number of the index. If memRootPage is negative, then
2264** the index already exists and must be cleared before being refilled and
2265** the root page number of the index is taken from pIndex->tnum.
2266*/
2267static void sqlite3RefillIndex(Parse *pParse, Index *pIndex, int memRootPage){
2268 Table *pTab = pIndex->pTable; /* The table that is indexed */
2269 int iTab = pParse->nTab; /* Btree cursor used for pTab */
2270 int iIdx = pParse->nTab+1; /* Btree cursor used for pIndex */
2271 int addr1; /* Address of top of loop */
2272 int tnum; /* Root page of index */
2273 Vdbe *v; /* Generate code into this virtual machine */
danielk1977b3bf5562006-01-10 17:58:23 +00002274 KeyInfo *pKey; /* KeyInfo for index */
drh2d401ab2008-01-10 23:50:11 +00002275 int regIdxKey; /* Registers containing the index key */
2276 int regRecord; /* Register holding assemblied index record */
drh17435752007-08-16 04:30:38 +00002277 sqlite3 *db = pParse->db; /* The database connection */
2278 int iDb = sqlite3SchemaToIndex(db, pIndex->pSchema);
drh063336a2004-11-05 20:58:39 +00002279
danielk19771d54df82004-11-23 15:41:16 +00002280#ifndef SQLITE_OMIT_AUTHORIZATION
2281 if( sqlite3AuthCheck(pParse, SQLITE_REINDEX, pIndex->zName, 0,
drh17435752007-08-16 04:30:38 +00002282 db->aDb[iDb].zName ) ){
danielk19771d54df82004-11-23 15:41:16 +00002283 return;
2284 }
2285#endif
2286
danielk1977c00da102006-01-07 13:21:04 +00002287 /* Require a write-lock on the table to perform this operation */
2288 sqlite3TableLock(pParse, iDb, pTab->tnum, 1, pTab->zName);
2289
drh063336a2004-11-05 20:58:39 +00002290 v = sqlite3GetVdbe(pParse);
2291 if( v==0 ) return;
2292 if( memRootPage>=0 ){
drh1db639c2008-01-17 02:36:28 +00002293 tnum = memRootPage;
drh063336a2004-11-05 20:58:39 +00002294 }else{
2295 tnum = pIndex->tnum;
drh66a51672008-01-03 00:01:23 +00002296 sqlite3VdbeAddOp2(v, OP_Clear, tnum, iDb);
drh063336a2004-11-05 20:58:39 +00002297 }
danielk1977b3bf5562006-01-10 17:58:23 +00002298 pKey = sqlite3IndexKeyinfo(pParse, pIndex);
danielk1977207872a2008-01-03 07:54:23 +00002299 sqlite3VdbeAddOp4(v, OP_OpenWrite, iIdx, tnum, iDb,
drh66a51672008-01-03 00:01:23 +00002300 (char *)pKey, P4_KEYINFO_HANDOFF);
drh98757152008-01-09 23:04:12 +00002301 if( memRootPage>=0 ){
2302 sqlite3VdbeChangeP5(v, 1);
2303 }
danielk1977c00da102006-01-07 13:21:04 +00002304 sqlite3OpenTable(pParse, iTab, iDb, pTab, OP_OpenRead);
drh66a51672008-01-03 00:01:23 +00002305 addr1 = sqlite3VdbeAddOp2(v, OP_Rewind, iTab, 0);
drh2d401ab2008-01-10 23:50:11 +00002306 regRecord = sqlite3GetTempReg(pParse);
drhe14006d2008-03-25 17:23:32 +00002307 regIdxKey = sqlite3GenerateIndexKey(pParse, pIndex, iTab, regRecord, 1);
drh7f057c92005-06-24 03:53:06 +00002308 if( pIndex->onError!=OE_None ){
drh2d401ab2008-01-10 23:50:11 +00002309 int j1, j2;
2310 int regRowid;
2311
2312 regRowid = regIdxKey + pIndex->nColumn;
2313 j1 = sqlite3VdbeAddOp3(v, OP_IsNull, regIdxKey, 0, pIndex->nColumn);
2314 j2 = sqlite3VdbeAddOp4(v, OP_IsUnique, iIdx,
shane1fc41292008-07-08 22:28:48 +00002315 0, regRowid, SQLITE_INT_TO_PTR(regRecord), P4_INT32);
drh66a51672008-01-03 00:01:23 +00002316 sqlite3VdbeAddOp4(v, OP_Halt, SQLITE_CONSTRAINT, OE_Abort, 0,
2317 "indexed columns are not unique", P4_STATIC);
drh2d401ab2008-01-10 23:50:11 +00002318 sqlite3VdbeJumpHere(v, j1);
2319 sqlite3VdbeJumpHere(v, j2);
drh4343fea2004-11-05 23:46:15 +00002320 }
drh2d401ab2008-01-10 23:50:11 +00002321 sqlite3VdbeAddOp2(v, OP_IdxInsert, iIdx, regRecord);
2322 sqlite3ReleaseTempReg(pParse, regRecord);
drh66a51672008-01-03 00:01:23 +00002323 sqlite3VdbeAddOp2(v, OP_Next, iTab, addr1+1);
drhd654be82005-09-20 17:42:23 +00002324 sqlite3VdbeJumpHere(v, addr1);
drh66a51672008-01-03 00:01:23 +00002325 sqlite3VdbeAddOp1(v, OP_Close, iTab);
2326 sqlite3VdbeAddOp1(v, OP_Close, iIdx);
drh063336a2004-11-05 20:58:39 +00002327}
2328
2329/*
drh23bf66d2004-12-14 03:34:34 +00002330** Create a new index for an SQL table. pName1.pName2 is the name of the index
2331** and pTblList is the name of the table that is to be indexed. Both will
drhadbca9c2001-09-27 15:11:53 +00002332** be NULL for a primary key or an index that is created to satisfy a
2333** UNIQUE constraint. If pTable and pIndex are NULL, use pParse->pNewTable
drh382c0242001-10-06 16:33:02 +00002334** as the table to be indexed. pParse->pNewTable is a table that is
2335** currently being constructed by a CREATE TABLE statement.
drh75897232000-05-29 14:26:00 +00002336**
drh382c0242001-10-06 16:33:02 +00002337** pList is a list of columns to be indexed. pList will be NULL if this
2338** is a primary key or unique-constraint on the most recent column added
2339** to the table currently under construction.
drh75897232000-05-29 14:26:00 +00002340*/
danielk19774adee202004-05-08 08:23:19 +00002341void sqlite3CreateIndex(
drh23bf66d2004-12-14 03:34:34 +00002342 Parse *pParse, /* All information about this parse */
2343 Token *pName1, /* First part of index name. May be NULL */
2344 Token *pName2, /* Second part of index name. May be NULL */
2345 SrcList *pTblName, /* Table to index. Use pParse->pNewTable if 0 */
danielk19770202b292004-06-09 09:55:16 +00002346 ExprList *pList, /* A list of columns to be indexed */
drh23bf66d2004-12-14 03:34:34 +00002347 int onError, /* OE_Abort, OE_Ignore, OE_Replace, or OE_None */
drh1c55ba02007-07-02 19:31:27 +00002348 Token *pStart, /* The CREATE token that begins this statement */
drhfdd6e852005-12-16 01:06:16 +00002349 Token *pEnd, /* The ")" that closes the CREATE INDEX statement */
drh4d91a702006-01-04 15:54:36 +00002350 int sortOrder, /* Sort order of primary key when pList==NULL */
2351 int ifNotExist /* Omit error if index already exists */
drh75897232000-05-29 14:26:00 +00002352){
drhfdd6e852005-12-16 01:06:16 +00002353 Table *pTab = 0; /* Table to be indexed */
2354 Index *pIndex = 0; /* The index to be created */
2355 char *zName = 0; /* Name of the index */
2356 int nName; /* Number of characters in zName */
drhbeae3192001-09-22 18:12:08 +00002357 int i, j;
drhfdd6e852005-12-16 01:06:16 +00002358 Token nullId; /* Fake token for an empty ID list */
2359 DbFixer sFix; /* For assigning database names to pTable */
2360 int sortOrderMask; /* 1 to honor DESC in index. 0 to ignore. */
drh9bb575f2004-09-06 17:24:11 +00002361 sqlite3 *db = pParse->db;
drhfdd6e852005-12-16 01:06:16 +00002362 Db *pDb; /* The specific table containing the indexed database */
2363 int iDb; /* Index of the database that is being written */
2364 Token *pName = 0; /* Unqualified name of the index to create */
2365 struct ExprList_item *pListItem; /* For looping over pList */
danielk1977b3bf5562006-01-10 17:58:23 +00002366 int nCol;
2367 int nExtra = 0;
2368 char *zExtra;
danielk1977cbb18d22004-05-28 11:37:27 +00002369
drh17435752007-08-16 04:30:38 +00002370 if( pParse->nErr || db->mallocFailed || IN_DECLARE_VTAB ){
danielk1977e501b892006-01-09 06:29:47 +00002371 goto exit_create_index;
2372 }
drhdaffd0e2001-04-11 14:28:42 +00002373
drh75897232000-05-29 14:26:00 +00002374 /*
2375 ** Find the table that is to be indexed. Return early if not found.
2376 */
danielk1977cbb18d22004-05-28 11:37:27 +00002377 if( pTblName!=0 ){
danielk1977cbb18d22004-05-28 11:37:27 +00002378
2379 /* Use the two-part index name to determine the database
danielk1977ef2cb632004-05-29 02:37:19 +00002380 ** to search for the table. 'Fix' the table name to this db
2381 ** before looking up the table.
danielk1977cbb18d22004-05-28 11:37:27 +00002382 */
2383 assert( pName1 && pName2 );
danielk1977ef2cb632004-05-29 02:37:19 +00002384 iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pName);
danielk1977cbb18d22004-05-28 11:37:27 +00002385 if( iDb<0 ) goto exit_create_index;
2386
danielk197753c0f742005-03-29 03:10:59 +00002387#ifndef SQLITE_OMIT_TEMPDB
danielk1977ef2cb632004-05-29 02:37:19 +00002388 /* If the index name was unqualified, check if the the table
danielk1977fe910332007-12-02 11:46:34 +00002389 ** is a temp table. If so, set the database to 1. Do not do this
2390 ** if initialising a database schema.
danielk1977cbb18d22004-05-28 11:37:27 +00002391 */
danielk1977fe910332007-12-02 11:46:34 +00002392 if( !db->init.busy ){
2393 pTab = sqlite3SrcListLookup(pParse, pTblName);
2394 if( pName2 && pName2->n==0 && pTab && pTab->pSchema==db->aDb[1].pSchema ){
2395 iDb = 1;
2396 }
danielk1977ef2cb632004-05-29 02:37:19 +00002397 }
danielk197753c0f742005-03-29 03:10:59 +00002398#endif
danielk1977ef2cb632004-05-29 02:37:19 +00002399
2400 if( sqlite3FixInit(&sFix, pParse, iDb, "index", pName) &&
2401 sqlite3FixSrcList(&sFix, pTblName)
2402 ){
drh85c23c62005-08-20 03:03:04 +00002403 /* Because the parser constructs pTblName from a single identifier,
2404 ** sqlite3FixSrcList can never fail. */
2405 assert(0);
danielk1977cbb18d22004-05-28 11:37:27 +00002406 }
drhca424112008-01-25 15:04:48 +00002407 pTab = sqlite3LocateTable(pParse, 0, pTblName->a[0].zName,
danielk1977ef2cb632004-05-29 02:37:19 +00002408 pTblName->a[0].zDatabase);
danielk1977cbb18d22004-05-28 11:37:27 +00002409 if( !pTab ) goto exit_create_index;
danielk1977da184232006-01-05 11:34:32 +00002410 assert( db->aDb[iDb].pSchema==pTab->pSchema );
drh75897232000-05-29 14:26:00 +00002411 }else{
drhe3c41372001-09-17 20:25:58 +00002412 assert( pName==0 );
danielk1977da184232006-01-05 11:34:32 +00002413 pTab = pParse->pNewTable;
drha6370df2006-01-04 21:40:06 +00002414 if( !pTab ) goto exit_create_index;
danielk1977da184232006-01-05 11:34:32 +00002415 iDb = sqlite3SchemaToIndex(db, pTab->pSchema);
drh75897232000-05-29 14:26:00 +00002416 }
drhfdd6e852005-12-16 01:06:16 +00002417 pDb = &db->aDb[iDb];
danielk1977cbb18d22004-05-28 11:37:27 +00002418
drh75897232000-05-29 14:26:00 +00002419 if( pTab==0 || pParse->nErr ) goto exit_create_index;
drh0be9df02003-03-30 00:19:49 +00002420 if( pTab->readOnly ){
danielk19774adee202004-05-08 08:23:19 +00002421 sqlite3ErrorMsg(pParse, "table %s may not be indexed", pTab->zName);
drh0be9df02003-03-30 00:19:49 +00002422 goto exit_create_index;
2423 }
danielk1977576ec6b2005-01-21 11:55:25 +00002424#ifndef SQLITE_OMIT_VIEW
drha76b5df2002-02-23 02:32:10 +00002425 if( pTab->pSelect ){
danielk19774adee202004-05-08 08:23:19 +00002426 sqlite3ErrorMsg(pParse, "views may not be indexed");
drha76b5df2002-02-23 02:32:10 +00002427 goto exit_create_index;
2428 }
danielk1977576ec6b2005-01-21 11:55:25 +00002429#endif
danielk19775ee9d692006-06-21 12:36:25 +00002430#ifndef SQLITE_OMIT_VIRTUALTABLE
2431 if( IsVirtual(pTab) ){
2432 sqlite3ErrorMsg(pParse, "virtual tables may not be indexed");
2433 goto exit_create_index;
2434 }
2435#endif
drh75897232000-05-29 14:26:00 +00002436
2437 /*
2438 ** Find the name of the index. Make sure there is not already another
drhf57b3392001-10-08 13:22:32 +00002439 ** index or table with the same name.
2440 **
2441 ** Exception: If we are reading the names of permanent indices from the
2442 ** sqlite_master table (because some other process changed the schema) and
2443 ** one of the index names collides with the name of a temporary table or
drhd24cc422003-03-27 12:51:24 +00002444 ** index, then we will continue to process this index.
drhf57b3392001-10-08 13:22:32 +00002445 **
2446 ** If pName==0 it means that we are
drhadbca9c2001-09-27 15:11:53 +00002447 ** dealing with a primary key or UNIQUE constraint. We have to invent our
2448 ** own name.
drh75897232000-05-29 14:26:00 +00002449 */
danielk1977d8123362004-06-12 09:25:12 +00002450 if( pName ){
drh17435752007-08-16 04:30:38 +00002451 zName = sqlite3NameFromToken(db, pName);
danielk19778a414492004-06-29 08:59:35 +00002452 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ) goto exit_create_index;
drhe3c41372001-09-17 20:25:58 +00002453 if( zName==0 ) goto exit_create_index;
danielk1977d8123362004-06-12 09:25:12 +00002454 if( SQLITE_OK!=sqlite3CheckObjectName(pParse, zName) ){
drhd24cc422003-03-27 12:51:24 +00002455 goto exit_create_index;
drhe3c41372001-09-17 20:25:58 +00002456 }
danielk1977d8123362004-06-12 09:25:12 +00002457 if( !db->init.busy ){
danielk19778a414492004-06-29 08:59:35 +00002458 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ) goto exit_create_index;
danielk1977d45a0312007-03-13 16:32:25 +00002459 if( sqlite3FindTable(db, zName, 0)!=0 ){
2460 sqlite3ErrorMsg(pParse, "there is already a table named %s", zName);
2461 goto exit_create_index;
2462 }
2463 }
danielk197759a33f92007-03-17 10:26:59 +00002464 if( sqlite3FindIndex(db, zName, pDb->zName)!=0 ){
2465 if( !ifNotExist ){
2466 sqlite3ErrorMsg(pParse, "index %s already exists", zName);
danielk1977d8123362004-06-12 09:25:12 +00002467 }
danielk197759a33f92007-03-17 10:26:59 +00002468 goto exit_create_index;
2469 }
danielk1977a21c6b62005-01-24 10:25:59 +00002470 }else{
drhadbca9c2001-09-27 15:11:53 +00002471 int n;
2472 Index *pLoop;
2473 for(pLoop=pTab->pIndex, n=1; pLoop; pLoop=pLoop->pNext, n++){}
drhf089aa42008-07-08 19:34:06 +00002474 zName = sqlite3MPrintf(db, "sqlite_autoindex_%s_%d", pTab->zName, n);
danielk1977a1644fd2007-08-29 12:31:25 +00002475 if( zName==0 ){
danielk1977a1644fd2007-08-29 12:31:25 +00002476 goto exit_create_index;
2477 }
drh75897232000-05-29 14:26:00 +00002478 }
2479
drhe5f9c642003-01-13 23:27:31 +00002480 /* Check for authorization to create an index.
2481 */
2482#ifndef SQLITE_OMIT_AUTHORIZATION
drhe22a3342003-04-22 20:30:37 +00002483 {
drhfdd6e852005-12-16 01:06:16 +00002484 const char *zDb = pDb->zName;
danielk197753c0f742005-03-29 03:10:59 +00002485 if( sqlite3AuthCheck(pParse, SQLITE_INSERT, SCHEMA_TABLE(iDb), 0, zDb) ){
drhe22a3342003-04-22 20:30:37 +00002486 goto exit_create_index;
2487 }
2488 i = SQLITE_CREATE_INDEX;
danielk197753c0f742005-03-29 03:10:59 +00002489 if( !OMIT_TEMPDB && iDb==1 ) i = SQLITE_CREATE_TEMP_INDEX;
danielk19774adee202004-05-08 08:23:19 +00002490 if( sqlite3AuthCheck(pParse, i, zName, pTab->zName, zDb) ){
drhe22a3342003-04-22 20:30:37 +00002491 goto exit_create_index;
2492 }
drhe5f9c642003-01-13 23:27:31 +00002493 }
2494#endif
2495
drh75897232000-05-29 14:26:00 +00002496 /* If pList==0, it means this routine was called to make a primary
drh1ccde152000-06-17 13:12:39 +00002497 ** key out of the last column added to the table under construction.
drh75897232000-05-29 14:26:00 +00002498 ** So create a fake list to simulate this.
2499 */
2500 if( pList==0 ){
drh2646da72005-12-09 20:02:05 +00002501 nullId.z = (u8*)pTab->aCol[pTab->nCol-1].zName;
2502 nullId.n = strlen((char*)nullId.z);
danielk19771e536952007-08-16 10:09:01 +00002503 pList = sqlite3ExprListAppend(pParse, 0, 0, &nullId);
drh75897232000-05-29 14:26:00 +00002504 if( pList==0 ) goto exit_create_index;
drhfdd6e852005-12-16 01:06:16 +00002505 pList->a[0].sortOrder = sortOrder;
drh75897232000-05-29 14:26:00 +00002506 }
2507
danielk1977b3bf5562006-01-10 17:58:23 +00002508 /* Figure out how many bytes of space are required to store explicitly
2509 ** specified collation sequence names.
2510 */
2511 for(i=0; i<pList->nExpr; i++){
2512 Expr *pExpr = pList->a[i].pExpr;
2513 if( pExpr ){
2514 nExtra += (1 + strlen(pExpr->pColl->zName));
2515 }
2516 }
2517
drh75897232000-05-29 14:26:00 +00002518 /*
2519 ** Allocate the index structure.
2520 */
drhfdd6e852005-12-16 01:06:16 +00002521 nName = strlen(zName);
danielk1977b3bf5562006-01-10 17:58:23 +00002522 nCol = pList->nExpr;
drh17435752007-08-16 04:30:38 +00002523 pIndex = sqlite3DbMallocZero(db,
danielk1977b3bf5562006-01-10 17:58:23 +00002524 sizeof(Index) + /* Index structure */
2525 sizeof(int)*nCol + /* Index.aiColumn */
2526 sizeof(int)*(nCol+1) + /* Index.aiRowEst */
2527 sizeof(char *)*nCol + /* Index.azColl */
2528 sizeof(u8)*nCol + /* Index.aSortOrder */
2529 nName + 1 + /* Index.zName */
2530 nExtra /* Collation sequence names */
2531 );
drh17435752007-08-16 04:30:38 +00002532 if( db->mallocFailed ){
2533 goto exit_create_index;
2534 }
drh78aecb72006-02-10 18:08:09 +00002535 pIndex->azColl = (char**)(&pIndex[1]);
2536 pIndex->aiColumn = (int *)(&pIndex->azColl[nCol]);
danielk1977bab45c62006-01-16 15:14:27 +00002537 pIndex->aiRowEst = (unsigned *)(&pIndex->aiColumn[nCol]);
drh78aecb72006-02-10 18:08:09 +00002538 pIndex->aSortOrder = (u8 *)(&pIndex->aiRowEst[nCol+1]);
danielk1977b3bf5562006-01-10 17:58:23 +00002539 pIndex->zName = (char *)(&pIndex->aSortOrder[nCol]);
2540 zExtra = (char *)(&pIndex->zName[nName+1]);
drh5bb3eb92007-05-04 13:15:55 +00002541 memcpy(pIndex->zName, zName, nName+1);
drh75897232000-05-29 14:26:00 +00002542 pIndex->pTable = pTab;
danielk19770202b292004-06-09 09:55:16 +00002543 pIndex->nColumn = pList->nExpr;
drhea1ba172003-04-20 00:00:23 +00002544 pIndex->onError = onError;
drh485b39b2002-07-13 03:11:52 +00002545 pIndex->autoIndex = pName==0;
danielk1977da184232006-01-05 11:34:32 +00002546 pIndex->pSchema = db->aDb[iDb].pSchema;
drh75897232000-05-29 14:26:00 +00002547
drhfdd6e852005-12-16 01:06:16 +00002548 /* Check to see if we should honor DESC requests on index columns
2549 */
danielk1977da184232006-01-05 11:34:32 +00002550 if( pDb->pSchema->file_format>=4 ){
drhfdd6e852005-12-16 01:06:16 +00002551 sortOrderMask = -1; /* Honor DESC */
drhfdd6e852005-12-16 01:06:16 +00002552 }else{
2553 sortOrderMask = 0; /* Ignore DESC */
2554 }
2555
drh1ccde152000-06-17 13:12:39 +00002556 /* Scan the names of the columns of the table to be indexed and
2557 ** load the column indices into the Index structure. Report an error
2558 ** if any column is not found.
drh75897232000-05-29 14:26:00 +00002559 */
drhfdd6e852005-12-16 01:06:16 +00002560 for(i=0, pListItem=pList->a; i<pList->nExpr; i++, pListItem++){
2561 const char *zColName = pListItem->zName;
2562 Column *pTabCol;
drh85eeb692005-12-21 03:16:42 +00002563 int requestedSortOrder;
drha34001c2007-02-02 12:44:37 +00002564 char *zColl; /* Collation sequence name */
danielk1977b3bf5562006-01-10 17:58:23 +00002565
drhfdd6e852005-12-16 01:06:16 +00002566 for(j=0, pTabCol=pTab->aCol; j<pTab->nCol; j++, pTabCol++){
2567 if( sqlite3StrICmp(zColName, pTabCol->zName)==0 ) break;
drh75897232000-05-29 14:26:00 +00002568 }
2569 if( j>=pTab->nCol ){
danielk19774adee202004-05-08 08:23:19 +00002570 sqlite3ErrorMsg(pParse, "table %s has no column named %s",
drhfdd6e852005-12-16 01:06:16 +00002571 pTab->zName, zColName);
drh75897232000-05-29 14:26:00 +00002572 goto exit_create_index;
2573 }
drha34001c2007-02-02 12:44:37 +00002574 /* TODO: Add a test to make sure that the same column is not named
2575 ** more than once within the same index. Only the first instance of
2576 ** the column will ever be used by the optimizer. Note that using the
2577 ** same column more than once cannot be an error because that would
2578 ** break backwards compatibility - it needs to be a warning.
2579 */
drh967e8b72000-06-21 13:59:10 +00002580 pIndex->aiColumn[i] = j;
drhfdd6e852005-12-16 01:06:16 +00002581 if( pListItem->pExpr ){
2582 assert( pListItem->pExpr->pColl );
danielk1977b3bf5562006-01-10 17:58:23 +00002583 zColl = zExtra;
drh5bb3eb92007-05-04 13:15:55 +00002584 sqlite3_snprintf(nExtra, zExtra, "%s", pListItem->pExpr->pColl->zName);
danielk1977b3bf5562006-01-10 17:58:23 +00002585 zExtra += (strlen(zColl) + 1);
danielk19770202b292004-06-09 09:55:16 +00002586 }else{
danielk1977b3bf5562006-01-10 17:58:23 +00002587 zColl = pTab->aCol[j].zColl;
2588 if( !zColl ){
2589 zColl = db->pDfltColl->zName;
2590 }
danielk19770202b292004-06-09 09:55:16 +00002591 }
danielk1977b3bf5562006-01-10 17:58:23 +00002592 if( !db->init.busy && !sqlite3LocateCollSeq(pParse, zColl, -1) ){
danielk19777cedc8d2004-06-10 10:50:08 +00002593 goto exit_create_index;
2594 }
danielk1977b3bf5562006-01-10 17:58:23 +00002595 pIndex->azColl[i] = zColl;
drhd946db02005-12-29 19:23:06 +00002596 requestedSortOrder = pListItem->sortOrder & sortOrderMask;
danielk1977b3bf5562006-01-10 17:58:23 +00002597 pIndex->aSortOrder[i] = requestedSortOrder;
drh75897232000-05-29 14:26:00 +00002598 }
drh51147ba2005-07-23 22:59:55 +00002599 sqlite3DefaultRowEst(pIndex);
drh75897232000-05-29 14:26:00 +00002600
danielk1977d8123362004-06-12 09:25:12 +00002601 if( pTab==pParse->pNewTable ){
2602 /* This routine has been called to create an automatic index as a
2603 ** result of a PRIMARY KEY or UNIQUE clause on a column definition, or
2604 ** a PRIMARY KEY or UNIQUE clause following the column definitions.
2605 ** i.e. one of:
2606 **
2607 ** CREATE TABLE t(x PRIMARY KEY, y);
2608 ** CREATE TABLE t(x, y, UNIQUE(x, y));
2609 **
2610 ** Either way, check to see if the table already has such an index. If
2611 ** so, don't bother creating this one. This only applies to
2612 ** automatically created indices. Users can do as they wish with
2613 ** explicit indices.
2614 */
2615 Index *pIdx;
2616 for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){
2617 int k;
2618 assert( pIdx->onError!=OE_None );
2619 assert( pIdx->autoIndex );
2620 assert( pIndex->onError!=OE_None );
2621
2622 if( pIdx->nColumn!=pIndex->nColumn ) continue;
2623 for(k=0; k<pIdx->nColumn; k++){
danielk1977b3bf5562006-01-10 17:58:23 +00002624 const char *z1 = pIdx->azColl[k];
2625 const char *z2 = pIndex->azColl[k];
danielk1977d8123362004-06-12 09:25:12 +00002626 if( pIdx->aiColumn[k]!=pIndex->aiColumn[k] ) break;
danielk1977b3bf5562006-01-10 17:58:23 +00002627 if( pIdx->aSortOrder[k]!=pIndex->aSortOrder[k] ) break;
2628 if( z1!=z2 && sqlite3StrICmp(z1, z2) ) break;
danielk1977d8123362004-06-12 09:25:12 +00002629 }
2630 if( k==pIdx->nColumn ){
danielk1977f736b772004-06-17 06:13:34 +00002631 if( pIdx->onError!=pIndex->onError ){
2632 /* This constraint creates the same index as a previous
2633 ** constraint specified somewhere in the CREATE TABLE statement.
2634 ** However the ON CONFLICT clauses are different. If both this
2635 ** constraint and the previous equivalent constraint have explicit
2636 ** ON CONFLICT clauses this is an error. Otherwise, use the
2637 ** explicitly specified behaviour for the index.
2638 */
2639 if( !(pIdx->onError==OE_Default || pIndex->onError==OE_Default) ){
2640 sqlite3ErrorMsg(pParse,
2641 "conflicting ON CONFLICT clauses specified", 0);
2642 }
2643 if( pIdx->onError==OE_Default ){
2644 pIdx->onError = pIndex->onError;
2645 }
2646 }
danielk1977d8123362004-06-12 09:25:12 +00002647 goto exit_create_index;
2648 }
2649 }
2650 }
2651
drh75897232000-05-29 14:26:00 +00002652 /* Link the new Index structure to its table and to the other
drhadbca9c2001-09-27 15:11:53 +00002653 ** in-memory database structures.
drh75897232000-05-29 14:26:00 +00002654 */
drh234c39d2004-07-24 03:30:47 +00002655 if( db->init.busy ){
drh6d4abfb2001-10-22 02:58:08 +00002656 Index *p;
danielk1977da184232006-01-05 11:34:32 +00002657 p = sqlite3HashInsert(&pIndex->pSchema->idxHash,
drh3c8bf552003-07-01 18:13:14 +00002658 pIndex->zName, strlen(pIndex->zName)+1, pIndex);
drh6d4abfb2001-10-22 02:58:08 +00002659 if( p ){
2660 assert( p==pIndex ); /* Malloc must have failed */
drh17435752007-08-16 04:30:38 +00002661 db->mallocFailed = 1;
drh6d4abfb2001-10-22 02:58:08 +00002662 goto exit_create_index;
2663 }
drh5e00f6c2001-09-13 13:46:56 +00002664 db->flags |= SQLITE_InternChanges;
drh234c39d2004-07-24 03:30:47 +00002665 if( pTblName!=0 ){
2666 pIndex->tnum = db->init.newTnum;
2667 }
drhd78eeee2001-09-13 16:18:53 +00002668 }
2669
drh1d85d932004-02-14 23:05:52 +00002670 /* If the db->init.busy is 0 then create the index on disk. This
drh75897232000-05-29 14:26:00 +00002671 ** involves writing the index into the master table and filling in the
2672 ** index with the current table contents.
2673 **
drh1d85d932004-02-14 23:05:52 +00002674 ** The db->init.busy is 0 when the user first enters a CREATE INDEX
2675 ** command. db->init.busy is 1 when a database is opened and
drh75897232000-05-29 14:26:00 +00002676 ** CREATE INDEX statements are read out of the master table. In
2677 ** the latter case the index already exists on disk, which is why
2678 ** we don't want to recreate it.
drh5edc3122001-09-13 21:53:09 +00002679 **
danielk1977cbb18d22004-05-28 11:37:27 +00002680 ** If pTblName==0 it means this index is generated as a primary key
drh382c0242001-10-06 16:33:02 +00002681 ** or UNIQUE constraint of a CREATE TABLE statement. Since the table
2682 ** has just been created, it contains no data and the index initialization
2683 ** step can be skipped.
drh75897232000-05-29 14:26:00 +00002684 */
drh1d85d932004-02-14 23:05:52 +00002685 else if( db->init.busy==0 ){
drhadbca9c2001-09-27 15:11:53 +00002686 Vdbe *v;
drh063336a2004-11-05 20:58:39 +00002687 char *zStmt;
drh0a07c102008-01-03 18:03:08 +00002688 int iMem = ++pParse->nMem;
drh75897232000-05-29 14:26:00 +00002689
danielk19774adee202004-05-08 08:23:19 +00002690 v = sqlite3GetVdbe(pParse);
drh75897232000-05-29 14:26:00 +00002691 if( v==0 ) goto exit_create_index;
drh063336a2004-11-05 20:58:39 +00002692
drhfdd6e852005-12-16 01:06:16 +00002693
drh063336a2004-11-05 20:58:39 +00002694 /* Create the rootpage for the index
2695 */
drhaee128d2005-02-14 20:48:18 +00002696 sqlite3BeginWriteOperation(pParse, 1, iDb);
drhb7654112008-01-12 12:48:07 +00002697 sqlite3VdbeAddOp2(v, OP_CreateIndex, iDb, iMem);
drh063336a2004-11-05 20:58:39 +00002698
2699 /* Gather the complete text of the CREATE INDEX statement into
2700 ** the zStmt variable
2701 */
drhe0bc4042002-06-25 01:09:11 +00002702 if( pStart && pEnd ){
drh063336a2004-11-05 20:58:39 +00002703 /* A named index with an explicit CREATE INDEX statement */
danielk19771e536952007-08-16 10:09:01 +00002704 zStmt = sqlite3MPrintf(db, "CREATE%s INDEX %.*s",
drh063336a2004-11-05 20:58:39 +00002705 onError==OE_None ? "" : " UNIQUE",
drh97903fe2005-05-24 20:19:57 +00002706 pEnd->z - pName->z + 1,
drh063336a2004-11-05 20:58:39 +00002707 pName->z);
2708 }else{
2709 /* An automatic index created by a PRIMARY KEY or UNIQUE constraint */
drhe497f002004-11-07 13:01:49 +00002710 /* zStmt = sqlite3MPrintf(""); */
2711 zStmt = 0;
drh75897232000-05-29 14:26:00 +00002712 }
drh063336a2004-11-05 20:58:39 +00002713
2714 /* Add an entry in sqlite_master for this index
2715 */
2716 sqlite3NestedParse(pParse,
drhb7654112008-01-12 12:48:07 +00002717 "INSERT INTO %Q.%s VALUES('index',%Q,%Q,#%d,%Q);",
drh063336a2004-11-05 20:58:39 +00002718 db->aDb[iDb].zName, SCHEMA_TABLE(iDb),
2719 pIndex->zName,
2720 pTab->zName,
drhb7654112008-01-12 12:48:07 +00002721 iMem,
drh063336a2004-11-05 20:58:39 +00002722 zStmt
2723 );
drh633e6d52008-07-28 19:34:53 +00002724 sqlite3DbFree(db, zStmt);
drh063336a2004-11-05 20:58:39 +00002725
danielk1977a21c6b62005-01-24 10:25:59 +00002726 /* Fill the index with data and reparse the schema. Code an OP_Expire
2727 ** to invalidate all pre-compiled statements.
drh063336a2004-11-05 20:58:39 +00002728 */
danielk1977cbb18d22004-05-28 11:37:27 +00002729 if( pTblName ){
drh063336a2004-11-05 20:58:39 +00002730 sqlite3RefillIndex(pParse, pIndex, iMem);
drh9cbf3422008-01-17 16:22:13 +00002731 sqlite3ChangeCookie(pParse, iDb);
drh66a51672008-01-03 00:01:23 +00002732 sqlite3VdbeAddOp4(v, OP_ParseSchema, iDb, 0, 0,
2733 sqlite3MPrintf(db, "name='%q'", pIndex->zName), P4_DYNAMIC);
2734 sqlite3VdbeAddOp1(v, OP_Expire, 0);
drh5e00f6c2001-09-13 13:46:56 +00002735 }
drh75897232000-05-29 14:26:00 +00002736 }
2737
danielk1977d8123362004-06-12 09:25:12 +00002738 /* When adding an index to the list of indices for a table, make
2739 ** sure all indices labeled OE_Replace come after all those labeled
2740 ** OE_Ignore. This is necessary for the correct operation of UPDATE
2741 ** and INSERT.
2742 */
drh234c39d2004-07-24 03:30:47 +00002743 if( db->init.busy || pTblName==0 ){
2744 if( onError!=OE_Replace || pTab->pIndex==0
2745 || pTab->pIndex->onError==OE_Replace){
2746 pIndex->pNext = pTab->pIndex;
2747 pTab->pIndex = pIndex;
2748 }else{
2749 Index *pOther = pTab->pIndex;
2750 while( pOther->pNext && pOther->pNext->onError!=OE_Replace ){
2751 pOther = pOther->pNext;
2752 }
2753 pIndex->pNext = pOther->pNext;
2754 pOther->pNext = pIndex;
danielk1977d8123362004-06-12 09:25:12 +00002755 }
drh234c39d2004-07-24 03:30:47 +00002756 pIndex = 0;
danielk1977d8123362004-06-12 09:25:12 +00002757 }
danielk1977d8123362004-06-12 09:25:12 +00002758
drh75897232000-05-29 14:26:00 +00002759 /* Clean up before exiting */
2760exit_create_index:
drh956bc922004-07-24 17:38:29 +00002761 if( pIndex ){
2762 freeIndex(pIndex);
2763 }
drh633e6d52008-07-28 19:34:53 +00002764 sqlite3ExprListDelete(db, pList);
2765 sqlite3SrcListDelete(db, pTblName);
2766 sqlite3DbFree(db, zName);
drh75897232000-05-29 14:26:00 +00002767 return;
2768}
2769
2770/*
drhd28bcb32005-12-21 14:43:11 +00002771** Generate code to make sure the file format number is at least minFormat.
2772** The generated code will increase the file format number if necessary.
2773*/
2774void sqlite3MinimumFileFormat(Parse *pParse, int iDb, int minFormat){
2775 Vdbe *v;
2776 v = sqlite3GetVdbe(pParse);
2777 if( v ){
drh1db639c2008-01-17 02:36:28 +00002778 int r1 = sqlite3GetTempReg(pParse);
2779 int r2 = sqlite3GetTempReg(pParse);
2780 int j1;
2781 sqlite3VdbeAddOp3(v, OP_ReadCookie, iDb, r1, 1);
drhfb982642007-08-30 01:19:59 +00002782 sqlite3VdbeUsesBtree(v, iDb);
drh1db639c2008-01-17 02:36:28 +00002783 sqlite3VdbeAddOp2(v, OP_Integer, minFormat, r2);
2784 j1 = sqlite3VdbeAddOp3(v, OP_Ge, r2, 0, r1);
2785 sqlite3VdbeAddOp3(v, OP_SetCookie, iDb, 1, r2);
2786 sqlite3VdbeJumpHere(v, j1);
2787 sqlite3ReleaseTempReg(pParse, r1);
2788 sqlite3ReleaseTempReg(pParse, r2);
drhd28bcb32005-12-21 14:43:11 +00002789 }
2790}
2791
2792/*
drh51147ba2005-07-23 22:59:55 +00002793** Fill the Index.aiRowEst[] array with default information - information
drh91124b32005-08-18 18:15:05 +00002794** to be used when we have not run the ANALYZE command.
drh28c4cf42005-07-27 20:41:43 +00002795**
2796** aiRowEst[0] is suppose to contain the number of elements in the index.
2797** Since we do not know, guess 1 million. aiRowEst[1] is an estimate of the
2798** number of rows in the table that match any particular value of the
2799** first column of the index. aiRowEst[2] is an estimate of the number
2800** of rows that match any particular combiniation of the first 2 columns
2801** of the index. And so forth. It must always be the case that
2802*
2803** aiRowEst[N]<=aiRowEst[N-1]
2804** aiRowEst[N]>=1
2805**
2806** Apart from that, we have little to go on besides intuition as to
2807** how aiRowEst[] should be initialized. The numbers generated here
2808** are based on typical values found in actual indices.
drh51147ba2005-07-23 22:59:55 +00002809*/
2810void sqlite3DefaultRowEst(Index *pIdx){
drh37108e12005-08-31 13:13:31 +00002811 unsigned *a = pIdx->aiRowEst;
drh51147ba2005-07-23 22:59:55 +00002812 int i;
drh28c4cf42005-07-27 20:41:43 +00002813 assert( a!=0 );
2814 a[0] = 1000000;
drhdb513882006-05-11 23:14:59 +00002815 for(i=pIdx->nColumn; i>=5; i--){
2816 a[i] = 5;
2817 }
2818 while( i>=1 ){
2819 a[i] = 11 - i;
2820 i--;
drh28c4cf42005-07-27 20:41:43 +00002821 }
2822 if( pIdx->onError!=OE_None ){
2823 a[pIdx->nColumn] = 1;
drh51147ba2005-07-23 22:59:55 +00002824 }
2825}
2826
2827/*
drh74e24cd2002-01-09 03:19:59 +00002828** This routine will drop an existing named index. This routine
2829** implements the DROP INDEX statement.
drh75897232000-05-29 14:26:00 +00002830*/
drh4d91a702006-01-04 15:54:36 +00002831void sqlite3DropIndex(Parse *pParse, SrcList *pName, int ifExists){
drh75897232000-05-29 14:26:00 +00002832 Index *pIndex;
drh75897232000-05-29 14:26:00 +00002833 Vdbe *v;
drh9bb575f2004-09-06 17:24:11 +00002834 sqlite3 *db = pParse->db;
danielk1977da184232006-01-05 11:34:32 +00002835 int iDb;
drh75897232000-05-29 14:26:00 +00002836
drh17435752007-08-16 04:30:38 +00002837 if( pParse->nErr || db->mallocFailed ){
danielk1977d5d56522005-03-16 12:15:20 +00002838 goto exit_drop_index;
2839 }
drhd24cc422003-03-27 12:51:24 +00002840 assert( pName->nSrc==1 );
danielk1977d5d56522005-03-16 12:15:20 +00002841 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){
2842 goto exit_drop_index;
2843 }
danielk19774adee202004-05-08 08:23:19 +00002844 pIndex = sqlite3FindIndex(db, pName->a[0].zName, pName->a[0].zDatabase);
drh75897232000-05-29 14:26:00 +00002845 if( pIndex==0 ){
drh4d91a702006-01-04 15:54:36 +00002846 if( !ifExists ){
2847 sqlite3ErrorMsg(pParse, "no such index: %S", pName, 0);
2848 }
drha6ecd332004-06-10 00:29:09 +00002849 pParse->checkSchema = 1;
drhd24cc422003-03-27 12:51:24 +00002850 goto exit_drop_index;
drh75897232000-05-29 14:26:00 +00002851 }
drh485b39b2002-07-13 03:11:52 +00002852 if( pIndex->autoIndex ){
danielk19774adee202004-05-08 08:23:19 +00002853 sqlite3ErrorMsg(pParse, "index associated with UNIQUE "
drh485b39b2002-07-13 03:11:52 +00002854 "or PRIMARY KEY constraint cannot be dropped", 0);
drhd24cc422003-03-27 12:51:24 +00002855 goto exit_drop_index;
2856 }
danielk1977da184232006-01-05 11:34:32 +00002857 iDb = sqlite3SchemaToIndex(db, pIndex->pSchema);
drhe5f9c642003-01-13 23:27:31 +00002858#ifndef SQLITE_OMIT_AUTHORIZATION
2859 {
2860 int code = SQLITE_DROP_INDEX;
2861 Table *pTab = pIndex->pTable;
danielk1977da184232006-01-05 11:34:32 +00002862 const char *zDb = db->aDb[iDb].zName;
2863 const char *zTab = SCHEMA_TABLE(iDb);
danielk19774adee202004-05-08 08:23:19 +00002864 if( sqlite3AuthCheck(pParse, SQLITE_DELETE, zTab, 0, zDb) ){
drhd24cc422003-03-27 12:51:24 +00002865 goto exit_drop_index;
drhe5f9c642003-01-13 23:27:31 +00002866 }
danielk1977da184232006-01-05 11:34:32 +00002867 if( !OMIT_TEMPDB && iDb ) code = SQLITE_DROP_TEMP_INDEX;
danielk19774adee202004-05-08 08:23:19 +00002868 if( sqlite3AuthCheck(pParse, code, pIndex->zName, pTab->zName, zDb) ){
drhd24cc422003-03-27 12:51:24 +00002869 goto exit_drop_index;
drhe5f9c642003-01-13 23:27:31 +00002870 }
drhed6c8672003-01-12 18:02:16 +00002871 }
drhe5f9c642003-01-13 23:27:31 +00002872#endif
drh75897232000-05-29 14:26:00 +00002873
2874 /* Generate code to remove the index and from the master table */
danielk19774adee202004-05-08 08:23:19 +00002875 v = sqlite3GetVdbe(pParse);
drh75897232000-05-29 14:26:00 +00002876 if( v ){
drh77658e22007-12-04 16:54:52 +00002877 sqlite3BeginWriteOperation(pParse, 1, iDb);
drhb17131a2004-11-05 22:18:49 +00002878 sqlite3NestedParse(pParse,
2879 "DELETE FROM %Q.%s WHERE name=%Q",
2880 db->aDb[iDb].zName, SCHEMA_TABLE(iDb),
2881 pIndex->zName
2882 );
danielk1977006015d2008-04-11 17:11:26 +00002883 if( sqlite3FindTable(db, "sqlite_stat1", db->aDb[iDb].zName) ){
2884 sqlite3NestedParse(pParse,
2885 "DELETE FROM %Q.sqlite_stat1 WHERE idx=%Q",
2886 db->aDb[iDb].zName, pIndex->zName
2887 );
2888 }
drh9cbf3422008-01-17 16:22:13 +00002889 sqlite3ChangeCookie(pParse, iDb);
drhb17131a2004-11-05 22:18:49 +00002890 destroyRootPage(pParse, pIndex->tnum, iDb);
drh66a51672008-01-03 00:01:23 +00002891 sqlite3VdbeAddOp4(v, OP_DropIndex, iDb, 0, 0, pIndex->zName, 0);
drh75897232000-05-29 14:26:00 +00002892 }
2893
drhd24cc422003-03-27 12:51:24 +00002894exit_drop_index:
drh633e6d52008-07-28 19:34:53 +00002895 sqlite3SrcListDelete(db, pName);
drh75897232000-05-29 14:26:00 +00002896}
2897
2898/*
drhcf643722007-03-27 13:36:37 +00002899** pArray is a pointer to an array of objects. Each object in the
2900** array is szEntry bytes in size. This routine allocates a new
2901** object on the end of the array.
drh13449892005-09-07 21:22:45 +00002902**
drhcf643722007-03-27 13:36:37 +00002903** *pnEntry is the number of entries already in use. *pnAlloc is
2904** the previously allocated size of the array. initSize is the
2905** suggested initial array size allocation.
drh13449892005-09-07 21:22:45 +00002906**
drhcf643722007-03-27 13:36:37 +00002907** The index of the new entry is returned in *pIdx.
drh13449892005-09-07 21:22:45 +00002908**
drhcf643722007-03-27 13:36:37 +00002909** This routine returns a pointer to the array of objects. This
2910** might be the same as the pArray parameter or it might be a different
2911** pointer if the array was resized.
drh13449892005-09-07 21:22:45 +00002912*/
drhcf643722007-03-27 13:36:37 +00002913void *sqlite3ArrayAllocate(
drh17435752007-08-16 04:30:38 +00002914 sqlite3 *db, /* Connection to notify of malloc failures */
drhcf643722007-03-27 13:36:37 +00002915 void *pArray, /* Array of objects. Might be reallocated */
2916 int szEntry, /* Size of each object in the array */
2917 int initSize, /* Suggested initial allocation, in elements */
2918 int *pnEntry, /* Number of objects currently in use */
2919 int *pnAlloc, /* Current size of the allocation, in elements */
2920 int *pIdx /* Write the index of a new slot here */
2921){
2922 char *z;
2923 if( *pnEntry >= *pnAlloc ){
drh13449892005-09-07 21:22:45 +00002924 void *pNew;
drh5360ad32005-09-08 00:13:27 +00002925 int newSize;
drhcf643722007-03-27 13:36:37 +00002926 newSize = (*pnAlloc)*2 + initSize;
danielk197726783a52007-08-29 14:06:22 +00002927 pNew = sqlite3DbRealloc(db, pArray, newSize*szEntry);
drh13449892005-09-07 21:22:45 +00002928 if( pNew==0 ){
drhcf643722007-03-27 13:36:37 +00002929 *pIdx = -1;
2930 return pArray;
drh13449892005-09-07 21:22:45 +00002931 }
drhcf643722007-03-27 13:36:37 +00002932 *pnAlloc = newSize;
2933 pArray = pNew;
drh13449892005-09-07 21:22:45 +00002934 }
drhcf643722007-03-27 13:36:37 +00002935 z = (char*)pArray;
2936 memset(&z[*pnEntry * szEntry], 0, szEntry);
2937 *pIdx = *pnEntry;
2938 ++*pnEntry;
2939 return pArray;
drh13449892005-09-07 21:22:45 +00002940}
2941
2942/*
drh75897232000-05-29 14:26:00 +00002943** Append a new element to the given IdList. Create a new IdList if
2944** need be.
drhdaffd0e2001-04-11 14:28:42 +00002945**
2946** A new IdList is returned, or NULL if malloc() fails.
drh75897232000-05-29 14:26:00 +00002947*/
drh17435752007-08-16 04:30:38 +00002948IdList *sqlite3IdListAppend(sqlite3 *db, IdList *pList, Token *pToken){
drh13449892005-09-07 21:22:45 +00002949 int i;
drh75897232000-05-29 14:26:00 +00002950 if( pList==0 ){
drh17435752007-08-16 04:30:38 +00002951 pList = sqlite3DbMallocZero(db, sizeof(IdList) );
drh75897232000-05-29 14:26:00 +00002952 if( pList==0 ) return 0;
drh4305d102003-07-30 12:34:12 +00002953 pList->nAlloc = 0;
drh75897232000-05-29 14:26:00 +00002954 }
drhcf643722007-03-27 13:36:37 +00002955 pList->a = sqlite3ArrayAllocate(
drh17435752007-08-16 04:30:38 +00002956 db,
drhcf643722007-03-27 13:36:37 +00002957 pList->a,
2958 sizeof(pList->a[0]),
2959 5,
2960 &pList->nId,
2961 &pList->nAlloc,
2962 &i
2963 );
drh13449892005-09-07 21:22:45 +00002964 if( i<0 ){
drh633e6d52008-07-28 19:34:53 +00002965 sqlite3IdListDelete(db, pList);
drh13449892005-09-07 21:22:45 +00002966 return 0;
drh75897232000-05-29 14:26:00 +00002967 }
drh17435752007-08-16 04:30:38 +00002968 pList->a[i].zName = sqlite3NameFromToken(db, pToken);
drh75897232000-05-29 14:26:00 +00002969 return pList;
2970}
2971
2972/*
drhfe05af82005-07-21 03:14:59 +00002973** Delete an IdList.
2974*/
drh633e6d52008-07-28 19:34:53 +00002975void sqlite3IdListDelete(sqlite3 *db, IdList *pList){
drhfe05af82005-07-21 03:14:59 +00002976 int i;
2977 if( pList==0 ) return;
2978 for(i=0; i<pList->nId; i++){
drh633e6d52008-07-28 19:34:53 +00002979 sqlite3DbFree(db, pList->a[i].zName);
drhfe05af82005-07-21 03:14:59 +00002980 }
drh633e6d52008-07-28 19:34:53 +00002981 sqlite3DbFree(db, pList->a);
2982 sqlite3DbFree(db, pList);
drhfe05af82005-07-21 03:14:59 +00002983}
2984
2985/*
2986** Return the index in pList of the identifier named zId. Return -1
2987** if not found.
2988*/
2989int sqlite3IdListIndex(IdList *pList, const char *zName){
2990 int i;
2991 if( pList==0 ) return -1;
2992 for(i=0; i<pList->nId; i++){
2993 if( sqlite3StrICmp(pList->a[i].zName, zName)==0 ) return i;
2994 }
2995 return -1;
2996}
2997
2998/*
drhad3cab52002-05-24 02:04:32 +00002999** Append a new table name to the given SrcList. Create a new SrcList if
3000** need be. A new entry is created in the SrcList even if pToken is NULL.
3001**
3002** A new SrcList is returned, or NULL if malloc() fails.
drh113088e2003-03-20 01:16:58 +00003003**
3004** If pDatabase is not null, it means that the table has an optional
3005** database name prefix. Like this: "database.table". The pDatabase
3006** points to the table name and the pTable points to the database name.
3007** The SrcList.a[].zName field is filled with the table name which might
3008** come from pTable (if pDatabase is NULL) or from pDatabase.
3009** SrcList.a[].zDatabase is filled with the database name from pTable,
3010** or with NULL if no database is specified.
3011**
3012** In other words, if call like this:
3013**
drh17435752007-08-16 04:30:38 +00003014** sqlite3SrcListAppend(D,A,B,0);
drh113088e2003-03-20 01:16:58 +00003015**
3016** Then B is a table name and the database name is unspecified. If called
3017** like this:
3018**
drh17435752007-08-16 04:30:38 +00003019** sqlite3SrcListAppend(D,A,B,C);
drh113088e2003-03-20 01:16:58 +00003020**
3021** Then C is the table name and B is the database name.
drhad3cab52002-05-24 02:04:32 +00003022*/
drh17435752007-08-16 04:30:38 +00003023SrcList *sqlite3SrcListAppend(
3024 sqlite3 *db, /* Connection to notify of malloc failures */
3025 SrcList *pList, /* Append to this SrcList. NULL creates a new SrcList */
3026 Token *pTable, /* Table to append */
3027 Token *pDatabase /* Database of the table */
3028){
drha99db3b2004-06-19 14:49:12 +00003029 struct SrcList_item *pItem;
drhad3cab52002-05-24 02:04:32 +00003030 if( pList==0 ){
drh17435752007-08-16 04:30:38 +00003031 pList = sqlite3DbMallocZero(db, sizeof(SrcList) );
drhad3cab52002-05-24 02:04:32 +00003032 if( pList==0 ) return 0;
drh4305d102003-07-30 12:34:12 +00003033 pList->nAlloc = 1;
drhad3cab52002-05-24 02:04:32 +00003034 }
drh4305d102003-07-30 12:34:12 +00003035 if( pList->nSrc>=pList->nAlloc ){
drh113088e2003-03-20 01:16:58 +00003036 SrcList *pNew;
drh4305d102003-07-30 12:34:12 +00003037 pList->nAlloc *= 2;
danielk197726783a52007-08-29 14:06:22 +00003038 pNew = sqlite3DbRealloc(db, pList,
drh4305d102003-07-30 12:34:12 +00003039 sizeof(*pList) + (pList->nAlloc-1)*sizeof(pList->a[0]) );
drh113088e2003-03-20 01:16:58 +00003040 if( pNew==0 ){
drh633e6d52008-07-28 19:34:53 +00003041 sqlite3SrcListDelete(db, pList);
drhad3cab52002-05-24 02:04:32 +00003042 return 0;
3043 }
drh113088e2003-03-20 01:16:58 +00003044 pList = pNew;
drhad3cab52002-05-24 02:04:32 +00003045 }
drha99db3b2004-06-19 14:49:12 +00003046 pItem = &pList->a[pList->nSrc];
3047 memset(pItem, 0, sizeof(pList->a[0]));
drh113088e2003-03-20 01:16:58 +00003048 if( pDatabase && pDatabase->z==0 ){
3049 pDatabase = 0;
3050 }
3051 if( pDatabase && pTable ){
3052 Token *pTemp = pDatabase;
3053 pDatabase = pTable;
3054 pTable = pTemp;
3055 }
drh17435752007-08-16 04:30:38 +00003056 pItem->zName = sqlite3NameFromToken(db, pTable);
3057 pItem->zDatabase = sqlite3NameFromToken(db, pDatabase);
drha99db3b2004-06-19 14:49:12 +00003058 pItem->iCursor = -1;
danielk19771787cca2006-02-10 07:07:14 +00003059 pItem->isPopulated = 0;
drhad3cab52002-05-24 02:04:32 +00003060 pList->nSrc++;
3061 return pList;
3062}
3063
3064/*
drh63eb5f22003-04-29 16:20:44 +00003065** Assign cursors to all tables in a SrcList
3066*/
danielk19774adee202004-05-08 08:23:19 +00003067void sqlite3SrcListAssignCursors(Parse *pParse, SrcList *pList){
drh63eb5f22003-04-29 16:20:44 +00003068 int i;
drh9b3187e2005-01-18 14:45:47 +00003069 struct SrcList_item *pItem;
drh17435752007-08-16 04:30:38 +00003070 assert(pList || pParse->db->mallocFailed );
danielk1977261919c2005-12-06 12:52:59 +00003071 if( pList ){
3072 for(i=0, pItem=pList->a; i<pList->nSrc; i++, pItem++){
3073 if( pItem->iCursor>=0 ) break;
3074 pItem->iCursor = pParse->nTab++;
3075 if( pItem->pSelect ){
3076 sqlite3SrcListAssignCursors(pParse, pItem->pSelect->pSrc);
3077 }
drh63eb5f22003-04-29 16:20:44 +00003078 }
3079 }
3080}
3081
3082/*
drhad3cab52002-05-24 02:04:32 +00003083** Delete an entire SrcList including all its substructure.
3084*/
drh633e6d52008-07-28 19:34:53 +00003085void sqlite3SrcListDelete(sqlite3 *db, SrcList *pList){
drhad3cab52002-05-24 02:04:32 +00003086 int i;
drhbe5c89a2004-07-26 00:31:09 +00003087 struct SrcList_item *pItem;
drhad3cab52002-05-24 02:04:32 +00003088 if( pList==0 ) return;
drhbe5c89a2004-07-26 00:31:09 +00003089 for(pItem=pList->a, i=0; i<pList->nSrc; i++, pItem++){
drh633e6d52008-07-28 19:34:53 +00003090 sqlite3DbFree(db, pItem->zDatabase);
3091 sqlite3DbFree(db, pItem->zName);
3092 sqlite3DbFree(db, pItem->zAlias);
danielk1977a04a34f2007-04-16 15:06:25 +00003093 sqlite3DeleteTable(pItem->pTab);
drh633e6d52008-07-28 19:34:53 +00003094 sqlite3SelectDelete(db, pItem->pSelect);
3095 sqlite3ExprDelete(db, pItem->pOn);
3096 sqlite3IdListDelete(db, pItem->pUsing);
drh75897232000-05-29 14:26:00 +00003097 }
drh633e6d52008-07-28 19:34:53 +00003098 sqlite3DbFree(db, pList);
drh75897232000-05-29 14:26:00 +00003099}
3100
drh982cef72000-05-30 16:27:03 +00003101/*
drh61dfc312006-12-16 16:25:15 +00003102** This routine is called by the parser to add a new term to the
3103** end of a growing FROM clause. The "p" parameter is the part of
3104** the FROM clause that has already been constructed. "p" is NULL
3105** if this is the first term of the FROM clause. pTable and pDatabase
3106** are the name of the table and database named in the FROM clause term.
3107** pDatabase is NULL if the database name qualifier is missing - the
3108** usual case. If the term has a alias, then pAlias points to the
3109** alias token. If the term is a subquery, then pSubquery is the
3110** SELECT statement that the subquery encodes. The pTable and
3111** pDatabase parameters are NULL for subqueries. The pOn and pUsing
3112** parameters are the content of the ON and USING clauses.
3113**
3114** Return a new SrcList which encodes is the FROM with the new
3115** term added.
3116*/
3117SrcList *sqlite3SrcListAppendFromTerm(
drh17435752007-08-16 04:30:38 +00003118 Parse *pParse, /* Parsing context */
drh61dfc312006-12-16 16:25:15 +00003119 SrcList *p, /* The left part of the FROM clause already seen */
3120 Token *pTable, /* Name of the table to add to the FROM clause */
3121 Token *pDatabase, /* Name of the database containing pTable */
3122 Token *pAlias, /* The right-hand side of the AS subexpression */
3123 Select *pSubquery, /* A subquery used in place of a table name */
3124 Expr *pOn, /* The ON clause of a join */
3125 IdList *pUsing /* The USING clause of a join */
3126){
3127 struct SrcList_item *pItem;
drh17435752007-08-16 04:30:38 +00003128 sqlite3 *db = pParse->db;
3129 p = sqlite3SrcListAppend(db, p, pTable, pDatabase);
drh61dfc312006-12-16 16:25:15 +00003130 if( p==0 || p->nSrc==0 ){
drh633e6d52008-07-28 19:34:53 +00003131 sqlite3ExprDelete(db, pOn);
3132 sqlite3IdListDelete(db, pUsing);
3133 sqlite3SelectDelete(db, pSubquery);
drh61dfc312006-12-16 16:25:15 +00003134 return p;
3135 }
3136 pItem = &p->a[p->nSrc-1];
3137 if( pAlias && pAlias->n ){
drh17435752007-08-16 04:30:38 +00003138 pItem->zAlias = sqlite3NameFromToken(db, pAlias);
drh61dfc312006-12-16 16:25:15 +00003139 }
3140 pItem->pSelect = pSubquery;
3141 pItem->pOn = pOn;
3142 pItem->pUsing = pUsing;
3143 return p;
3144}
3145
3146/*
3147** When building up a FROM clause in the parser, the join operator
3148** is initially attached to the left operand. But the code generator
3149** expects the join operator to be on the right operand. This routine
3150** Shifts all join operators from left to right for an entire FROM
3151** clause.
3152**
3153** Example: Suppose the join is like this:
3154**
3155** A natural cross join B
3156**
3157** The operator is "natural cross join". The A and B operands are stored
3158** in p->a[0] and p->a[1], respectively. The parser initially stores the
3159** operator with A. This routine shifts that operator over to B.
3160*/
3161void sqlite3SrcListShiftJoinType(SrcList *p){
3162 if( p && p->a ){
3163 int i;
3164 for(i=p->nSrc-1; i>0; i--){
3165 p->a[i].jointype = p->a[i-1].jointype;
3166 }
3167 p->a[0].jointype = 0;
3168 }
3169}
3170
3171/*
drhc4a3c772001-04-04 11:48:57 +00003172** Begin a transaction
3173*/
drh684917c2004-10-05 02:41:42 +00003174void sqlite3BeginTransaction(Parse *pParse, int type){
drh9bb575f2004-09-06 17:24:11 +00003175 sqlite3 *db;
danielk19771d850a72004-05-31 08:26:49 +00003176 Vdbe *v;
drh684917c2004-10-05 02:41:42 +00003177 int i;
drh5e00f6c2001-09-13 13:46:56 +00003178
drh001bbcb2003-03-19 03:14:00 +00003179 if( pParse==0 || (db=pParse->db)==0 || db->aDb[0].pBt==0 ) return;
drh17435752007-08-16 04:30:38 +00003180 if( pParse->nErr || db->mallocFailed ) return;
danielk19774adee202004-05-08 08:23:19 +00003181 if( sqlite3AuthCheck(pParse, SQLITE_TRANSACTION, "BEGIN", 0, 0) ) return;
danielk19771d850a72004-05-31 08:26:49 +00003182
3183 v = sqlite3GetVdbe(pParse);
3184 if( !v ) return;
drh684917c2004-10-05 02:41:42 +00003185 if( type!=TK_DEFERRED ){
3186 for(i=0; i<db->nDb; i++){
drh66a51672008-01-03 00:01:23 +00003187 sqlite3VdbeAddOp2(v, OP_Transaction, i, (type==TK_EXCLUSIVE)+1);
drhfb982642007-08-30 01:19:59 +00003188 sqlite3VdbeUsesBtree(v, i);
drh684917c2004-10-05 02:41:42 +00003189 }
3190 }
drh66a51672008-01-03 00:01:23 +00003191 sqlite3VdbeAddOp2(v, OP_AutoCommit, 0, 0);
drhc4a3c772001-04-04 11:48:57 +00003192}
3193
3194/*
3195** Commit a transaction
3196*/
danielk19774adee202004-05-08 08:23:19 +00003197void sqlite3CommitTransaction(Parse *pParse){
drh9bb575f2004-09-06 17:24:11 +00003198 sqlite3 *db;
danielk19771d850a72004-05-31 08:26:49 +00003199 Vdbe *v;
drh5e00f6c2001-09-13 13:46:56 +00003200
drh001bbcb2003-03-19 03:14:00 +00003201 if( pParse==0 || (db=pParse->db)==0 || db->aDb[0].pBt==0 ) return;
drh17435752007-08-16 04:30:38 +00003202 if( pParse->nErr || db->mallocFailed ) return;
danielk19774adee202004-05-08 08:23:19 +00003203 if( sqlite3AuthCheck(pParse, SQLITE_TRANSACTION, "COMMIT", 0, 0) ) return;
danielk19771d850a72004-05-31 08:26:49 +00003204
3205 v = sqlite3GetVdbe(pParse);
3206 if( v ){
drh66a51672008-01-03 00:01:23 +00003207 sqlite3VdbeAddOp2(v, OP_AutoCommit, 1, 0);
drh02f75f12004-02-24 01:04:11 +00003208 }
drhc4a3c772001-04-04 11:48:57 +00003209}
3210
3211/*
3212** Rollback a transaction
3213*/
danielk19774adee202004-05-08 08:23:19 +00003214void sqlite3RollbackTransaction(Parse *pParse){
drh9bb575f2004-09-06 17:24:11 +00003215 sqlite3 *db;
drh5e00f6c2001-09-13 13:46:56 +00003216 Vdbe *v;
3217
drh001bbcb2003-03-19 03:14:00 +00003218 if( pParse==0 || (db=pParse->db)==0 || db->aDb[0].pBt==0 ) return;
drh17435752007-08-16 04:30:38 +00003219 if( pParse->nErr || db->mallocFailed ) return;
danielk19774adee202004-05-08 08:23:19 +00003220 if( sqlite3AuthCheck(pParse, SQLITE_TRANSACTION, "ROLLBACK", 0, 0) ) return;
danielk19771d850a72004-05-31 08:26:49 +00003221
danielk19774adee202004-05-08 08:23:19 +00003222 v = sqlite3GetVdbe(pParse);
drh5e00f6c2001-09-13 13:46:56 +00003223 if( v ){
drh66a51672008-01-03 00:01:23 +00003224 sqlite3VdbeAddOp2(v, OP_AutoCommit, 1, 1);
drh02f75f12004-02-24 01:04:11 +00003225 }
drhc4a3c772001-04-04 11:48:57 +00003226}
drhf57b14a2001-09-14 18:54:08 +00003227
3228/*
drhdc3ff9c2004-08-18 02:10:15 +00003229** Make sure the TEMP database is open and available for use. Return
3230** the number of errors. Leave any error messages in the pParse structure.
3231*/
danielk1977ddfb2f02006-02-17 12:25:14 +00003232int sqlite3OpenTempDatabase(Parse *pParse){
drhdc3ff9c2004-08-18 02:10:15 +00003233 sqlite3 *db = pParse->db;
3234 if( db->aDb[1].pBt==0 && !pParse->explain ){
drh33f4e022007-09-03 15:19:34 +00003235 int rc;
3236 static const int flags =
3237 SQLITE_OPEN_READWRITE |
3238 SQLITE_OPEN_CREATE |
3239 SQLITE_OPEN_EXCLUSIVE |
3240 SQLITE_OPEN_DELETEONCLOSE |
3241 SQLITE_OPEN_TEMP_DB;
3242
3243 rc = sqlite3BtreeFactory(db, 0, 0, SQLITE_DEFAULT_CACHE_SIZE, flags,
drhc797d4d2007-05-08 01:08:49 +00003244 &db->aDb[1].pBt);
drhdc3ff9c2004-08-18 02:10:15 +00003245 if( rc!=SQLITE_OK ){
3246 sqlite3ErrorMsg(pParse, "unable to open a temporary database "
3247 "file for storing temporary tables");
3248 pParse->rc = rc;
3249 return 1;
3250 }
drh60a713c2008-01-21 16:22:45 +00003251 assert( (db->flags & SQLITE_InTrans)==0 || db->autoCommit );
danielk197714db2662006-01-09 16:12:04 +00003252 assert( db->aDb[1].pSchema );
drhfdc40e92008-04-17 20:59:37 +00003253 sqlite3PagerJournalMode(sqlite3BtreePager(db->aDb[1].pBt),
3254 db->dfltJournalMode);
drhdc3ff9c2004-08-18 02:10:15 +00003255 }
3256 return 0;
3257}
3258
3259/*
drh80242052004-06-09 00:48:12 +00003260** Generate VDBE code that will verify the schema cookie and start
3261** a read-transaction for all named database files.
3262**
3263** It is important that all schema cookies be verified and all
3264** read transactions be started before anything else happens in
3265** the VDBE program. But this routine can be called after much other
3266** code has been generated. So here is what we do:
3267**
drhc275b4e2004-07-19 17:25:24 +00003268** The first time this routine is called, we code an OP_Goto that
drh80242052004-06-09 00:48:12 +00003269** will jump to a subroutine at the end of the program. Then we
3270** record every database that needs its schema verified in the
3271** pParse->cookieMask field. Later, after all other code has been
3272** generated, the subroutine that does the cookie verifications and
drhc275b4e2004-07-19 17:25:24 +00003273** starts the transactions will be coded and the OP_Goto P2 value
drh80242052004-06-09 00:48:12 +00003274** will be made to point to that subroutine. The generation of the
3275** cookie verification subroutine code happens in sqlite3FinishCoding().
drhc275b4e2004-07-19 17:25:24 +00003276**
3277** If iDb<0 then code the OP_Goto only - don't set flag to verify the
3278** schema on any databases. This can be used to position the OP_Goto
3279** early in the code, before we know if any database tables will be used.
drh001bbcb2003-03-19 03:14:00 +00003280*/
danielk19774adee202004-05-08 08:23:19 +00003281void sqlite3CodeVerifySchema(Parse *pParse, int iDb){
drh9bb575f2004-09-06 17:24:11 +00003282 sqlite3 *db;
drh80242052004-06-09 00:48:12 +00003283 Vdbe *v;
3284 int mask;
3285
3286 v = sqlite3GetVdbe(pParse);
3287 if( v==0 ) return; /* This only happens if there was a prior error */
3288 db = pParse->db;
drhc275b4e2004-07-19 17:25:24 +00003289 if( pParse->cookieGoto==0 ){
drh66a51672008-01-03 00:01:23 +00003290 pParse->cookieGoto = sqlite3VdbeAddOp2(v, OP_Goto, 0, 0)+1;
drh80242052004-06-09 00:48:12 +00003291 }
drhc275b4e2004-07-19 17:25:24 +00003292 if( iDb>=0 ){
3293 assert( iDb<db->nDb );
3294 assert( db->aDb[iDb].pBt!=0 || iDb==1 );
drhc797d4d2007-05-08 01:08:49 +00003295 assert( iDb<SQLITE_MAX_ATTACHED+2 );
drhc275b4e2004-07-19 17:25:24 +00003296 mask = 1<<iDb;
3297 if( (pParse->cookieMask & mask)==0 ){
3298 pParse->cookieMask |= mask;
danielk1977da184232006-01-05 11:34:32 +00003299 pParse->cookieValue[iDb] = db->aDb[iDb].pSchema->schema_cookie;
danielk197753c0f742005-03-29 03:10:59 +00003300 if( !OMIT_TEMPDB && iDb==1 ){
drhdc3ff9c2004-08-18 02:10:15 +00003301 sqlite3OpenTempDatabase(pParse);
3302 }
drhc275b4e2004-07-19 17:25:24 +00003303 }
drh001bbcb2003-03-19 03:14:00 +00003304 }
drh001bbcb2003-03-19 03:14:00 +00003305}
3306
3307/*
drh1c928532002-01-31 15:54:21 +00003308** Generate VDBE code that prepares for doing an operation that
drhc977f7f2002-05-21 11:38:11 +00003309** might change the database.
3310**
3311** This routine starts a new transaction if we are not already within
3312** a transaction. If we are already within a transaction, then a checkpoint
drh7f0f12e2004-05-21 13:39:50 +00003313** is set if the setStatement parameter is true. A checkpoint should
drhc977f7f2002-05-21 11:38:11 +00003314** be set for operations that might fail (due to a constraint) part of
3315** the way through and which will need to undo some writes without having to
3316** rollback the whole transaction. For operations where all constraints
3317** can be checked before any changes are made to the database, it is never
3318** necessary to undo a write and the checkpoint should not be set.
drhcabb0812002-09-14 13:47:32 +00003319**
drh8bf8dc92003-05-17 17:35:10 +00003320** Only database iDb and the temp database are made writable by this call.
3321** If iDb==0, then the main and temp databases are made writable. If
3322** iDb==1 then only the temp database is made writable. If iDb>1 then the
3323** specified auxiliary database and the temp database are made writable.
drh1c928532002-01-31 15:54:21 +00003324*/
drh7f0f12e2004-05-21 13:39:50 +00003325void sqlite3BeginWriteOperation(Parse *pParse, int setStatement, int iDb){
danielk19771d850a72004-05-31 08:26:49 +00003326 Vdbe *v = sqlite3GetVdbe(pParse);
drh663fc632002-02-02 18:49:19 +00003327 if( v==0 ) return;
drh80242052004-06-09 00:48:12 +00003328 sqlite3CodeVerifySchema(pParse, iDb);
3329 pParse->writeMask |= 1<<iDb;
danielk197763e3e9f2004-11-05 09:19:27 +00003330 if( setStatement && pParse->nested==0 ){
drh66a51672008-01-03 00:01:23 +00003331 sqlite3VdbeAddOp1(v, OP_Statement, iDb);
danielk19771d850a72004-05-31 08:26:49 +00003332 }
danielk197753c0f742005-03-29 03:10:59 +00003333 if( (OMIT_TEMPDB || iDb!=1) && pParse->db->aDb[1].pBt!=0 ){
danielk19771d850a72004-05-31 08:26:49 +00003334 sqlite3BeginWriteOperation(pParse, setStatement, 1);
drh663fc632002-02-02 18:49:19 +00003335 }
3336}
3337
drh4343fea2004-11-05 23:46:15 +00003338/*
3339** Check to see if pIndex uses the collating sequence pColl. Return
3340** true if it does and false if it does not.
3341*/
3342#ifndef SQLITE_OMIT_REINDEX
danielk1977b3bf5562006-01-10 17:58:23 +00003343static int collationMatch(const char *zColl, Index *pIndex){
3344 int i;
3345 for(i=0; i<pIndex->nColumn; i++){
3346 const char *z = pIndex->azColl[i];
3347 if( z==zColl || (z && zColl && 0==sqlite3StrICmp(z, zColl)) ){
3348 return 1;
3349 }
drh4343fea2004-11-05 23:46:15 +00003350 }
3351 return 0;
3352}
3353#endif
3354
3355/*
3356** Recompute all indices of pTab that use the collating sequence pColl.
3357** If pColl==0 then recompute all indices of pTab.
3358*/
3359#ifndef SQLITE_OMIT_REINDEX
danielk1977b3bf5562006-01-10 17:58:23 +00003360static void reindexTable(Parse *pParse, Table *pTab, char const *zColl){
drh4343fea2004-11-05 23:46:15 +00003361 Index *pIndex; /* An index associated with pTab */
3362
3363 for(pIndex=pTab->pIndex; pIndex; pIndex=pIndex->pNext){
danielk1977b3bf5562006-01-10 17:58:23 +00003364 if( zColl==0 || collationMatch(zColl, pIndex) ){
danielk1977da184232006-01-05 11:34:32 +00003365 int iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema);
3366 sqlite3BeginWriteOperation(pParse, 0, iDb);
drh4343fea2004-11-05 23:46:15 +00003367 sqlite3RefillIndex(pParse, pIndex, -1);
3368 }
3369 }
3370}
3371#endif
3372
3373/*
3374** Recompute all indices of all tables in all databases where the
3375** indices use the collating sequence pColl. If pColl==0 then recompute
3376** all indices everywhere.
3377*/
3378#ifndef SQLITE_OMIT_REINDEX
danielk1977b3bf5562006-01-10 17:58:23 +00003379static void reindexDatabases(Parse *pParse, char const *zColl){
drh4343fea2004-11-05 23:46:15 +00003380 Db *pDb; /* A single database */
3381 int iDb; /* The database index number */
3382 sqlite3 *db = pParse->db; /* The database connection */
3383 HashElem *k; /* For looping over tables in pDb */
3384 Table *pTab; /* A table in the database */
3385
3386 for(iDb=0, pDb=db->aDb; iDb<db->nDb; iDb++, pDb++){
drh43617e92006-03-06 20:55:46 +00003387 assert( pDb!=0 );
danielk1977da184232006-01-05 11:34:32 +00003388 for(k=sqliteHashFirst(&pDb->pSchema->tblHash); k; k=sqliteHashNext(k)){
drh4343fea2004-11-05 23:46:15 +00003389 pTab = (Table*)sqliteHashData(k);
danielk1977b3bf5562006-01-10 17:58:23 +00003390 reindexTable(pParse, pTab, zColl);
drh4343fea2004-11-05 23:46:15 +00003391 }
3392 }
3393}
3394#endif
3395
3396/*
drheee46cf2004-11-06 00:02:48 +00003397** Generate code for the REINDEX command.
3398**
3399** REINDEX -- 1
3400** REINDEX <collation> -- 2
3401** REINDEX ?<database>.?<tablename> -- 3
3402** REINDEX ?<database>.?<indexname> -- 4
3403**
3404** Form 1 causes all indices in all attached databases to be rebuilt.
3405** Form 2 rebuilds all indices in all databases that use the named
3406** collating function. Forms 3 and 4 rebuild the named index or all
3407** indices associated with the named table.
drh4343fea2004-11-05 23:46:15 +00003408*/
3409#ifndef SQLITE_OMIT_REINDEX
3410void sqlite3Reindex(Parse *pParse, Token *pName1, Token *pName2){
3411 CollSeq *pColl; /* Collating sequence to be reindexed, or NULL */
3412 char *z; /* Name of a table or index */
3413 const char *zDb; /* Name of the database */
3414 Table *pTab; /* A table in the database */
3415 Index *pIndex; /* An index associated with pTab */
3416 int iDb; /* The database index number */
3417 sqlite3 *db = pParse->db; /* The database connection */
3418 Token *pObjName; /* Name of the table or index to be reindexed */
3419
danielk197733a5edc2005-01-27 00:22:02 +00003420 /* Read the database schema. If an error occurs, leave an error message
3421 ** and code in pParse and return NULL. */
3422 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){
danielk1977e63739a2005-01-27 00:33:37 +00003423 return;
danielk197733a5edc2005-01-27 00:22:02 +00003424 }
3425
drhe497f002004-11-07 13:01:49 +00003426 if( pName1==0 || pName1->z==0 ){
drh4343fea2004-11-05 23:46:15 +00003427 reindexDatabases(pParse, 0);
3428 return;
drhe497f002004-11-07 13:01:49 +00003429 }else if( pName2==0 || pName2->z==0 ){
danielk197739002502007-11-12 09:50:26 +00003430 char *zColl;
danielk1977b3bf5562006-01-10 17:58:23 +00003431 assert( pName1->z );
danielk197739002502007-11-12 09:50:26 +00003432 zColl = sqlite3NameFromToken(pParse->db, pName1);
3433 if( !zColl ) return;
3434 pColl = sqlite3FindCollSeq(db, ENC(db), zColl, -1, 0);
drh4343fea2004-11-05 23:46:15 +00003435 if( pColl ){
danielk1977f0113002006-01-24 12:09:17 +00003436 if( zColl ){
3437 reindexDatabases(pParse, zColl);
drh633e6d52008-07-28 19:34:53 +00003438 sqlite3DbFree(db, zColl);
danielk1977b3bf5562006-01-10 17:58:23 +00003439 }
drh4343fea2004-11-05 23:46:15 +00003440 return;
3441 }
drh633e6d52008-07-28 19:34:53 +00003442 sqlite3DbFree(db, zColl);
drh4343fea2004-11-05 23:46:15 +00003443 }
3444 iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pObjName);
3445 if( iDb<0 ) return;
drh17435752007-08-16 04:30:38 +00003446 z = sqlite3NameFromToken(db, pObjName);
drh84f31122007-05-12 15:00:14 +00003447 if( z==0 ) return;
drh4343fea2004-11-05 23:46:15 +00003448 zDb = db->aDb[iDb].zName;
3449 pTab = sqlite3FindTable(db, z, zDb);
3450 if( pTab ){
3451 reindexTable(pParse, pTab, 0);
drh633e6d52008-07-28 19:34:53 +00003452 sqlite3DbFree(db, z);
drh4343fea2004-11-05 23:46:15 +00003453 return;
3454 }
3455 pIndex = sqlite3FindIndex(db, z, zDb);
drh633e6d52008-07-28 19:34:53 +00003456 sqlite3DbFree(db, z);
drh4343fea2004-11-05 23:46:15 +00003457 if( pIndex ){
3458 sqlite3BeginWriteOperation(pParse, 0, iDb);
3459 sqlite3RefillIndex(pParse, pIndex, -1);
3460 return;
3461 }
3462 sqlite3ErrorMsg(pParse, "unable to identify the object to be reindexed");
3463}
3464#endif
danielk1977b3bf5562006-01-10 17:58:23 +00003465
3466/*
3467** Return a dynamicly allocated KeyInfo structure that can be used
3468** with OP_OpenRead or OP_OpenWrite to access database index pIdx.
3469**
3470** If successful, a pointer to the new structure is returned. In this case
drh633e6d52008-07-28 19:34:53 +00003471** the caller is responsible for calling sqlite3DbFree(db, ) on the returned
danielk1977b3bf5562006-01-10 17:58:23 +00003472** pointer. If an error occurs (out of memory or missing collation
3473** sequence), NULL is returned and the state of pParse updated to reflect
3474** the error.
3475*/
3476KeyInfo *sqlite3IndexKeyinfo(Parse *pParse, Index *pIdx){
3477 int i;
3478 int nCol = pIdx->nColumn;
3479 int nBytes = sizeof(KeyInfo) + (nCol-1)*sizeof(CollSeq*) + nCol;
drh633e6d52008-07-28 19:34:53 +00003480 sqlite3 *db = pParse->db;
3481 KeyInfo *pKey = (KeyInfo *)sqlite3DbMallocZero(db, nBytes);
danielk1977b3bf5562006-01-10 17:58:23 +00003482
3483 if( pKey ){
drhb21c8cd2007-08-21 19:33:56 +00003484 pKey->db = pParse->db;
danielk1977b3bf5562006-01-10 17:58:23 +00003485 pKey->aSortOrder = (u8 *)&(pKey->aColl[nCol]);
3486 assert( &pKey->aSortOrder[nCol]==&(((u8 *)pKey)[nBytes]) );
3487 for(i=0; i<nCol; i++){
3488 char *zColl = pIdx->azColl[i];
3489 assert( zColl );
3490 pKey->aColl[i] = sqlite3LocateCollSeq(pParse, zColl, -1);
3491 pKey->aSortOrder[i] = pIdx->aSortOrder[i];
3492 }
3493 pKey->nField = nCol;
3494 }
3495
3496 if( pParse->nErr ){
drh633e6d52008-07-28 19:34:53 +00003497 sqlite3DbFree(db, pKey);
danielk1977b3bf5562006-01-10 17:58:23 +00003498 pKey = 0;
3499 }
3500 return pKey;
3501}