1cce7d176Sdrh /* 2b19a2bc6Sdrh ** 2001 September 15 3cce7d176Sdrh ** 4b19a2bc6Sdrh ** The author disclaims copyright to this source code. In place of 5b19a2bc6Sdrh ** a legal notice, here is a blessing: 6cce7d176Sdrh ** 7b19a2bc6Sdrh ** May you do good and not evil. 8b19a2bc6Sdrh ** May you find forgiveness for yourself and forgive others. 9b19a2bc6Sdrh ** May you share freely, never taking more than you give. 10cce7d176Sdrh ** 11cce7d176Sdrh ************************************************************************* 12cce7d176Sdrh ** This file contains C code routines that are called by the parser 13b19a2bc6Sdrh ** to handle INSERT statements in SQLite. 14cce7d176Sdrh ** 15*ededfd5eSdanielk1977 ** $Id: insert.c,v 1.112 2004/06/17 07:53:03 danielk1977 Exp $ 16cce7d176Sdrh */ 17cce7d176Sdrh #include "sqliteInt.h" 18cce7d176Sdrh 19cce7d176Sdrh /* 203d1bfeaaSdanielk1977 ** Set P3 of the most recently inserted opcode to a column affinity 21a37cdde0Sdanielk1977 ** string for index pIdx. A column affinity string has one character 223d1bfeaaSdanielk1977 ** for each column in the table, according to the affinity of the column: 233d1bfeaaSdanielk1977 ** 243d1bfeaaSdanielk1977 ** Character Column affinity 253d1bfeaaSdanielk1977 ** ------------------------------ 263d1bfeaaSdanielk1977 ** 'n' NUMERIC 273d1bfeaaSdanielk1977 ** 'i' INTEGER 283d1bfeaaSdanielk1977 ** 't' TEXT 293d1bfeaaSdanielk1977 ** 'o' NONE 303d1bfeaaSdanielk1977 */ 31a37cdde0Sdanielk1977 void sqlite3IndexAffinityStr(Vdbe *v, Index *pIdx){ 32a37cdde0Sdanielk1977 if( !pIdx->zColAff ){ 33e014a838Sdanielk1977 /* The first time a column affinity string for a particular index is 34a37cdde0Sdanielk1977 ** required, it is allocated and populated here. It is then stored as 35e014a838Sdanielk1977 ** a member of the Index structure for subsequent use. 36a37cdde0Sdanielk1977 ** 37a37cdde0Sdanielk1977 ** The column affinity string will eventually be deleted by 38e014a838Sdanielk1977 ** sqliteDeleteIndex() when the Index structure itself is cleaned 39a37cdde0Sdanielk1977 ** up. 40a37cdde0Sdanielk1977 */ 41a37cdde0Sdanielk1977 int n; 42a37cdde0Sdanielk1977 Table *pTab = pIdx->pTable; 43a37cdde0Sdanielk1977 pIdx->zColAff = (char *)sqliteMalloc(pIdx->nColumn+1); 44a37cdde0Sdanielk1977 if( !pIdx->zColAff ){ 45a37cdde0Sdanielk1977 return; 46a37cdde0Sdanielk1977 } 47a37cdde0Sdanielk1977 for(n=0; n<pIdx->nColumn; n++){ 48a37cdde0Sdanielk1977 pIdx->zColAff[n] = pTab->aCol[pIdx->aiColumn[n]].affinity; 49a37cdde0Sdanielk1977 } 50a37cdde0Sdanielk1977 pIdx->zColAff[pIdx->nColumn] = '\0'; 51a37cdde0Sdanielk1977 } 523d1bfeaaSdanielk1977 53a37cdde0Sdanielk1977 sqlite3VdbeChangeP3(v, -1, pIdx->zColAff, P3_STATIC); 54a37cdde0Sdanielk1977 } 55a37cdde0Sdanielk1977 56a37cdde0Sdanielk1977 /* 57a37cdde0Sdanielk1977 ** Set P3 of the most recently inserted opcode to a column affinity 58a37cdde0Sdanielk1977 ** string for table pTab. A column affinity string has one character 59a37cdde0Sdanielk1977 ** for each column indexed by the index, according to the affinity of the 60a37cdde0Sdanielk1977 ** column: 61a37cdde0Sdanielk1977 ** 62a37cdde0Sdanielk1977 ** Character Column affinity 63a37cdde0Sdanielk1977 ** ------------------------------ 64a37cdde0Sdanielk1977 ** 'n' NUMERIC 65a37cdde0Sdanielk1977 ** 'i' INTEGER 66a37cdde0Sdanielk1977 ** 't' TEXT 67a37cdde0Sdanielk1977 ** 'o' NONE 68a37cdde0Sdanielk1977 */ 69a37cdde0Sdanielk1977 void sqlite3TableAffinityStr(Vdbe *v, Table *pTab){ 703d1bfeaaSdanielk1977 /* The first time a column affinity string for a particular table 713d1bfeaaSdanielk1977 ** is required, it is allocated and populated here. It is then 723d1bfeaaSdanielk1977 ** stored as a member of the Table structure for subsequent use. 733d1bfeaaSdanielk1977 ** 743d1bfeaaSdanielk1977 ** The column affinity string will eventually be deleted by 753d1bfeaaSdanielk1977 ** sqlite3DeleteTable() when the Table structure itself is cleaned up. 763d1bfeaaSdanielk1977 */ 773d1bfeaaSdanielk1977 if( !pTab->zColAff ){ 783d1bfeaaSdanielk1977 char *zColAff; 793d1bfeaaSdanielk1977 int i; 803d1bfeaaSdanielk1977 81a37cdde0Sdanielk1977 zColAff = (char *)sqliteMalloc(pTab->nCol+1); 823d1bfeaaSdanielk1977 if( !zColAff ){ 83a37cdde0Sdanielk1977 return; 843d1bfeaaSdanielk1977 } 853d1bfeaaSdanielk1977 863d1bfeaaSdanielk1977 for(i=0; i<pTab->nCol; i++){ 87a37cdde0Sdanielk1977 zColAff[i] = pTab->aCol[i].affinity; 883d1bfeaaSdanielk1977 } 893d1bfeaaSdanielk1977 zColAff[pTab->nCol] = '\0'; 903d1bfeaaSdanielk1977 913d1bfeaaSdanielk1977 pTab->zColAff = zColAff; 923d1bfeaaSdanielk1977 } 933d1bfeaaSdanielk1977 943d1bfeaaSdanielk1977 sqlite3VdbeChangeP3(v, -1, pTab->zColAff, P3_STATIC); 953d1bfeaaSdanielk1977 } 963d1bfeaaSdanielk1977 973d1bfeaaSdanielk1977 983d1bfeaaSdanielk1977 /* 991ccde15dSdrh ** This routine is call to handle SQL of the following forms: 100cce7d176Sdrh ** 101cce7d176Sdrh ** insert into TABLE (IDLIST) values(EXPRLIST) 1021ccde15dSdrh ** insert into TABLE (IDLIST) select 103cce7d176Sdrh ** 1041ccde15dSdrh ** The IDLIST following the table name is always optional. If omitted, 1051ccde15dSdrh ** then a list of all columns for the table is substituted. The IDLIST 106967e8b73Sdrh ** appears in the pColumn parameter. pColumn is NULL if IDLIST is omitted. 1071ccde15dSdrh ** 1081ccde15dSdrh ** The pList parameter holds EXPRLIST in the first form of the INSERT 1091ccde15dSdrh ** statement above, and pSelect is NULL. For the second form, pList is 1101ccde15dSdrh ** NULL and pSelect is a pointer to the select statement used to generate 1111ccde15dSdrh ** data for the insert. 112142e30dfSdrh ** 113142e30dfSdrh ** The code generated follows one of three templates. For a simple 114142e30dfSdrh ** select with data coming from a VALUES clause, the code executes 115142e30dfSdrh ** once straight down through. The template looks like this: 116142e30dfSdrh ** 117142e30dfSdrh ** open write cursor to <table> and its indices 118142e30dfSdrh ** puts VALUES clause expressions onto the stack 119142e30dfSdrh ** write the resulting record into <table> 120142e30dfSdrh ** cleanup 121142e30dfSdrh ** 122142e30dfSdrh ** If the statement is of the form 123142e30dfSdrh ** 124142e30dfSdrh ** INSERT INTO <table> SELECT ... 125142e30dfSdrh ** 126142e30dfSdrh ** And the SELECT clause does not read from <table> at any time, then 127142e30dfSdrh ** the generated code follows this template: 128142e30dfSdrh ** 129142e30dfSdrh ** goto B 130142e30dfSdrh ** A: setup for the SELECT 131142e30dfSdrh ** loop over the tables in the SELECT 132142e30dfSdrh ** gosub C 133142e30dfSdrh ** end loop 134142e30dfSdrh ** cleanup after the SELECT 135142e30dfSdrh ** goto D 136142e30dfSdrh ** B: open write cursor to <table> and its indices 137142e30dfSdrh ** goto A 138142e30dfSdrh ** C: insert the select result into <table> 139142e30dfSdrh ** return 140142e30dfSdrh ** D: cleanup 141142e30dfSdrh ** 142142e30dfSdrh ** The third template is used if the insert statement takes its 143142e30dfSdrh ** values from a SELECT but the data is being inserted into a table 144142e30dfSdrh ** that is also read as part of the SELECT. In the third form, 145142e30dfSdrh ** we have to use a intermediate table to store the results of 146142e30dfSdrh ** the select. The template is like this: 147142e30dfSdrh ** 148142e30dfSdrh ** goto B 149142e30dfSdrh ** A: setup for the SELECT 150142e30dfSdrh ** loop over the tables in the SELECT 151142e30dfSdrh ** gosub C 152142e30dfSdrh ** end loop 153142e30dfSdrh ** cleanup after the SELECT 154142e30dfSdrh ** goto D 155142e30dfSdrh ** C: insert the select result into the intermediate table 156142e30dfSdrh ** return 157142e30dfSdrh ** B: open a cursor to an intermediate table 158142e30dfSdrh ** goto A 159142e30dfSdrh ** D: open write cursor to <table> and its indices 160142e30dfSdrh ** loop over the intermediate table 161142e30dfSdrh ** transfer values form intermediate table into <table> 162142e30dfSdrh ** end the loop 163142e30dfSdrh ** cleanup 164cce7d176Sdrh */ 1654adee20fSdanielk1977 void sqlite3Insert( 166cce7d176Sdrh Parse *pParse, /* Parser context */ 167113088ecSdrh SrcList *pTabList, /* Name of table into which we are inserting */ 168cce7d176Sdrh ExprList *pList, /* List of values to be inserted */ 1695974a30fSdrh Select *pSelect, /* A SELECT statement to use as the data source */ 1709cfcf5d4Sdrh IdList *pColumn, /* Column names corresponding to IDLIST. */ 1719cfcf5d4Sdrh int onError /* How to handle constraint errors */ 172cce7d176Sdrh ){ 1735974a30fSdrh Table *pTab; /* The table to insert into */ 174113088ecSdrh char *zTab; /* Name of the table into which we are inserting */ 175e22a334bSdrh const char *zDb; /* Name of the database holding this table */ 1765974a30fSdrh int i, j, idx; /* Loop counters */ 1775974a30fSdrh Vdbe *v; /* Generate code into this virtual machine */ 1785974a30fSdrh Index *pIdx; /* For looping over indices of the table */ 179967e8b73Sdrh int nColumn; /* Number of columns in the data */ 180cfe9a69fSdanielk1977 int base = 0; /* VDBE Cursor number for pTab */ 181cfe9a69fSdanielk1977 int iCont=0,iBreak=0; /* Beginning and end of the loop over srcTab */ 182ecdc7530Sdrh sqlite *db; /* The main database structure */ 1834a32431cSdrh int keyColumn = -1; /* Column that is the INTEGER PRIMARY KEY */ 1840ca3e24bSdrh int endOfLoop; /* Label for the end of the insertion loop */ 185142e30dfSdrh int useTempTable; /* Store SELECT results in intermediate table */ 186cfe9a69fSdanielk1977 int srcTab = 0; /* Data comes from this temporary cursor if >=0 */ 187cfe9a69fSdanielk1977 int iSelectLoop = 0; /* Address of code that implements the SELECT */ 188cfe9a69fSdanielk1977 int iCleanup = 0; /* Address of the cleanup code */ 189cfe9a69fSdanielk1977 int iInsertBlock = 0; /* Address of the subroutine used to insert data */ 190cfe9a69fSdanielk1977 int iCntMem = 0; /* Memory cell used for the row counter */ 1915cf590c1Sdrh int isView; /* True if attempting to insert into a view */ 192cce7d176Sdrh 193c3f9bad2Sdanielk1977 int row_triggers_exist = 0; /* True if there are FOR EACH ROW triggers */ 19470ce3f0cSdrh int before_triggers; /* True if there are BEFORE triggers */ 19570ce3f0cSdrh int after_triggers; /* True if there are AFTER triggers */ 19670ce3f0cSdrh int newIdx = -1; /* Cursor for the NEW table */ 197c3f9bad2Sdanielk1977 19824b03fd0Sdanielk1977 if( pParse->nErr || sqlite3_malloc_failed ) goto insert_cleanup; 199ecdc7530Sdrh db = pParse->db; 200daffd0e5Sdrh 2011ccde15dSdrh /* Locate the table into which we will be inserting new information. 2021ccde15dSdrh */ 203113088ecSdrh assert( pTabList->nSrc==1 ); 204113088ecSdrh zTab = pTabList->a[0].zName; 205daffd0e5Sdrh if( zTab==0 ) goto insert_cleanup; 2064adee20fSdanielk1977 pTab = sqlite3SrcListLookup(pParse, pTabList); 207c3f9bad2Sdanielk1977 if( pTab==0 ){ 208c3f9bad2Sdanielk1977 goto insert_cleanup; 209c3f9bad2Sdanielk1977 } 210e22a334bSdrh assert( pTab->iDb<db->nDb ); 211e22a334bSdrh zDb = db->aDb[pTab->iDb].zName; 2124adee20fSdanielk1977 if( sqlite3AuthCheck(pParse, SQLITE_INSERT, pTab->zName, 0, zDb) ){ 2131962bda7Sdrh goto insert_cleanup; 2141962bda7Sdrh } 215c3f9bad2Sdanielk1977 216c3f9bad2Sdanielk1977 /* Ensure that: 217c3f9bad2Sdanielk1977 * (a) the table is not read-only, 218c3f9bad2Sdanielk1977 * (b) that if it is a view then ON INSERT triggers exist 219c3f9bad2Sdanielk1977 */ 2204adee20fSdanielk1977 before_triggers = sqlite3TriggersExist(pParse, pTab->pTrigger, TK_INSERT, 22170ce3f0cSdrh TK_BEFORE, TK_ROW, 0); 2224adee20fSdanielk1977 after_triggers = sqlite3TriggersExist(pParse, pTab->pTrigger, TK_INSERT, 22370ce3f0cSdrh TK_AFTER, TK_ROW, 0); 22470ce3f0cSdrh row_triggers_exist = before_triggers || after_triggers; 2255cf590c1Sdrh isView = pTab->pSelect!=0; 2264adee20fSdanielk1977 if( sqlite3IsReadOnly(pParse, pTab, before_triggers) ){ 227c3f9bad2Sdanielk1977 goto insert_cleanup; 228c3f9bad2Sdanielk1977 } 229a76b5dfcSdrh if( pTab==0 ) goto insert_cleanup; 2301ccde15dSdrh 231f573c99bSdrh /* If pTab is really a view, make sure it has been initialized. 232f573c99bSdrh */ 2334adee20fSdanielk1977 if( isView && sqlite3ViewGetColumnNames(pParse, pTab) ){ 234f573c99bSdrh goto insert_cleanup; 235f573c99bSdrh } 236f573c99bSdrh 2377cedc8d4Sdanielk1977 /* Ensure all required collation sequences are available. */ 2387cedc8d4Sdanielk1977 for(pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext){ 2397cedc8d4Sdanielk1977 if( sqlite3CheckIndexCollSeq(pParse, pIdx) ){ 2407cedc8d4Sdanielk1977 goto insert_cleanup; 2417cedc8d4Sdanielk1977 } 2427cedc8d4Sdanielk1977 } 2437cedc8d4Sdanielk1977 2441ccde15dSdrh /* Allocate a VDBE 2451ccde15dSdrh */ 2464adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 2475974a30fSdrh if( v==0 ) goto insert_cleanup; 2484adee20fSdanielk1977 sqlite3BeginWriteOperation(pParse, pSelect || row_triggers_exist, pTab->iDb); 2491ccde15dSdrh 250c3f9bad2Sdanielk1977 /* if there are row triggers, allocate a temp table for new.* references. */ 251f29ce559Sdanielk1977 if( row_triggers_exist ){ 252c3f9bad2Sdanielk1977 newIdx = pParse->nTab++; 253f29ce559Sdanielk1977 } 254c3f9bad2Sdanielk1977 2551ccde15dSdrh /* Figure out how many columns of data are supplied. If the data 256142e30dfSdrh ** is coming from a SELECT statement, then this step also generates 257142e30dfSdrh ** all the code to implement the SELECT statement and invoke a subroutine 258142e30dfSdrh ** to process each row of the result. (Template 2.) If the SELECT 259142e30dfSdrh ** statement uses the the table that is being inserted into, then the 260142e30dfSdrh ** subroutine is also coded here. That subroutine stores the SELECT 261142e30dfSdrh ** results in a temporary table. (Template 3.) 2621ccde15dSdrh */ 2635974a30fSdrh if( pSelect ){ 264142e30dfSdrh /* Data is coming from a SELECT. Generate code to implement that SELECT 265142e30dfSdrh */ 266142e30dfSdrh int rc, iInitCode; 2674adee20fSdanielk1977 iInitCode = sqlite3VdbeAddOp(v, OP_Goto, 0, 0); 2684adee20fSdanielk1977 iSelectLoop = sqlite3VdbeCurrentAddr(v); 2694adee20fSdanielk1977 iInsertBlock = sqlite3VdbeMakeLabel(v); 27084ac9d02Sdanielk1977 rc = sqlite3Select(pParse, pSelect, SRT_Subroutine, iInsertBlock, 0,0,0,0); 27124b03fd0Sdanielk1977 if( rc || pParse->nErr || sqlite3_malloc_failed ) goto insert_cleanup; 2724adee20fSdanielk1977 iCleanup = sqlite3VdbeMakeLabel(v); 2734adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iCleanup); 2745974a30fSdrh assert( pSelect->pEList ); 275967e8b73Sdrh nColumn = pSelect->pEList->nExpr; 276142e30dfSdrh 277142e30dfSdrh /* Set useTempTable to TRUE if the result of the SELECT statement 278142e30dfSdrh ** should be written into a temporary table. Set to FALSE if each 279142e30dfSdrh ** row of the SELECT can be written directly into the result table. 280048c530cSdrh ** 281048c530cSdrh ** A temp table must be used if the table being updated is also one 282048c530cSdrh ** of the tables being read by the SELECT statement. Also use a 283048c530cSdrh ** temp table in the case of row triggers. 284142e30dfSdrh */ 285048c530cSdrh if( row_triggers_exist ){ 286048c530cSdrh useTempTable = 1; 287048c530cSdrh }else{ 28884ac9d02Sdanielk1977 int addr = sqlite3VdbeFindOp(v, 0, OP_OpenRead, pTab->tnum); 289048c530cSdrh useTempTable = 0; 290048c530cSdrh if( addr>0 ){ 2914adee20fSdanielk1977 VdbeOp *pOp = sqlite3VdbeGetOp(v, addr-2); 292048c530cSdrh if( pOp->opcode==OP_Integer && pOp->p1==pTab->iDb ){ 293048c530cSdrh useTempTable = 1; 294048c530cSdrh } 295048c530cSdrh } 296048c530cSdrh } 297142e30dfSdrh 298142e30dfSdrh if( useTempTable ){ 299142e30dfSdrh /* Generate the subroutine that SELECT calls to process each row of 300142e30dfSdrh ** the result. Store the result in a temporary table 301142e30dfSdrh */ 302142e30dfSdrh srcTab = pParse->nTab++; 3034adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iInsertBlock); 3044adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, nColumn, 0); 305a37cdde0Sdanielk1977 sqlite3TableAffinityStr(v, pTab); 3064adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NewRecno, srcTab, 0); 3074adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pull, 1, 0); 3084adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, srcTab, 0); 3094adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Return, 0, 0); 310142e30dfSdrh 311142e30dfSdrh /* The following code runs first because the GOTO at the very top 312142e30dfSdrh ** of the program jumps to it. Create the temporary table, then jump 313142e30dfSdrh ** back up and execute the SELECT code above. 314142e30dfSdrh */ 3154adee20fSdanielk1977 sqlite3VdbeChangeP2(v, iInitCode, sqlite3VdbeCurrentAddr(v)); 3164adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_OpenTemp, srcTab, 0); 317b6f5452fSdrh sqlite3VdbeAddOp(v, OP_SetNumColumns, srcTab, nColumn); 3184adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iSelectLoop); 3194adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iCleanup); 3205974a30fSdrh }else{ 3214adee20fSdanielk1977 sqlite3VdbeChangeP2(v, iInitCode, sqlite3VdbeCurrentAddr(v)); 322142e30dfSdrh } 323142e30dfSdrh }else{ 324142e30dfSdrh /* This is the case if the data for the INSERT is coming from a VALUES 325142e30dfSdrh ** clause 326142e30dfSdrh */ 327ad3cab52Sdrh SrcList dummy; 328daffd0e5Sdrh assert( pList!=0 ); 3295974a30fSdrh srcTab = -1; 330142e30dfSdrh useTempTable = 0; 3315974a30fSdrh assert( pList ); 332967e8b73Sdrh nColumn = pList->nExpr; 333ad3cab52Sdrh dummy.nSrc = 0; 334e64e7b20Sdrh for(i=0; i<nColumn; i++){ 3354adee20fSdanielk1977 if( sqlite3ExprResolveIds(pParse, &dummy, 0, pList->a[i].pExpr) ){ 336e64e7b20Sdrh goto insert_cleanup; 337e64e7b20Sdrh } 3384adee20fSdanielk1977 if( sqlite3ExprCheck(pParse, pList->a[i].pExpr, 0, 0) ){ 339b04a5d87Sdrh goto insert_cleanup; 340b04a5d87Sdrh } 341e64e7b20Sdrh } 3425974a30fSdrh } 3431ccde15dSdrh 3441ccde15dSdrh /* Make sure the number of columns in the source data matches the number 3451ccde15dSdrh ** of columns to be inserted into the table. 3461ccde15dSdrh */ 347967e8b73Sdrh if( pColumn==0 && nColumn!=pTab->nCol ){ 3484adee20fSdanielk1977 sqlite3ErrorMsg(pParse, 349da93d238Sdrh "table %S has %d columns but %d values were supplied", 350da93d238Sdrh pTabList, 0, pTab->nCol, nColumn); 351cce7d176Sdrh goto insert_cleanup; 352cce7d176Sdrh } 353967e8b73Sdrh if( pColumn!=0 && nColumn!=pColumn->nId ){ 3544adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "%d values for %d columns", nColumn, pColumn->nId); 355cce7d176Sdrh goto insert_cleanup; 356cce7d176Sdrh } 3571ccde15dSdrh 3581ccde15dSdrh /* If the INSERT statement included an IDLIST term, then make sure 3591ccde15dSdrh ** all elements of the IDLIST really are columns of the table and 3601ccde15dSdrh ** remember the column indices. 361c8392586Sdrh ** 362c8392586Sdrh ** If the table has an INTEGER PRIMARY KEY column and that column 363c8392586Sdrh ** is named in the IDLIST, then record in the keyColumn variable 364c8392586Sdrh ** the index into IDLIST of the primary key column. keyColumn is 365c8392586Sdrh ** the index of the primary key as it appears in IDLIST, not as 366c8392586Sdrh ** is appears in the original table. (The index of the primary 367c8392586Sdrh ** key in the original table is pTab->iPKey.) 3681ccde15dSdrh */ 369967e8b73Sdrh if( pColumn ){ 370967e8b73Sdrh for(i=0; i<pColumn->nId; i++){ 371967e8b73Sdrh pColumn->a[i].idx = -1; 372cce7d176Sdrh } 373967e8b73Sdrh for(i=0; i<pColumn->nId; i++){ 374cce7d176Sdrh for(j=0; j<pTab->nCol; j++){ 3754adee20fSdanielk1977 if( sqlite3StrICmp(pColumn->a[i].zName, pTab->aCol[j].zName)==0 ){ 376967e8b73Sdrh pColumn->a[i].idx = j; 3774a32431cSdrh if( j==pTab->iPKey ){ 3789aa028daSdrh keyColumn = i; 3794a32431cSdrh } 380cce7d176Sdrh break; 381cce7d176Sdrh } 382cce7d176Sdrh } 383cce7d176Sdrh if( j>=pTab->nCol ){ 3844adee20fSdanielk1977 if( sqlite3IsRowid(pColumn->a[i].zName) ){ 385a0217ba7Sdrh keyColumn = i; 386a0217ba7Sdrh }else{ 3874adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "table %S has no column named %s", 388da93d238Sdrh pTabList, 0, pColumn->a[i].zName); 389cce7d176Sdrh pParse->nErr++; 390cce7d176Sdrh goto insert_cleanup; 391cce7d176Sdrh } 392cce7d176Sdrh } 393cce7d176Sdrh } 394a0217ba7Sdrh } 3951ccde15dSdrh 396aacc543eSdrh /* If there is no IDLIST term but the table has an integer primary 397c8392586Sdrh ** key, the set the keyColumn variable to the primary key column index 398c8392586Sdrh ** in the original table definition. 3994a32431cSdrh */ 4004a32431cSdrh if( pColumn==0 ){ 4014a32431cSdrh keyColumn = pTab->iPKey; 4024a32431cSdrh } 4034a32431cSdrh 404142e30dfSdrh /* Open the temp table for FOR EACH ROW triggers 405142e30dfSdrh */ 406f29ce559Sdanielk1977 if( row_triggers_exist ){ 4074adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_OpenPseudo, newIdx, 0); 40884ac9d02Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, newIdx, pTab->nCol); 409f29ce559Sdanielk1977 } 410c3f9bad2Sdanielk1977 411c3f9bad2Sdanielk1977 /* Initialize the count of rows to be inserted 4121ccde15dSdrh */ 413142e30dfSdrh if( db->flags & SQLITE_CountRows ){ 414142e30dfSdrh iCntMem = pParse->nMem++; 4154adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, 0, 0); 4164adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemStore, iCntMem, 1); 417c3f9bad2Sdanielk1977 } 418c3f9bad2Sdanielk1977 419c3f9bad2Sdanielk1977 /* Open tables and indices if there are no row triggers */ 420c3f9bad2Sdanielk1977 if( !row_triggers_exist ){ 4215974a30fSdrh base = pParse->nTab; 4224adee20fSdanielk1977 idx = sqlite3OpenTableAndIndices(pParse, pTab, base); 423832508b7Sdrh pParse->nTab += idx; 424feeb1394Sdrh } 425feeb1394Sdrh 426142e30dfSdrh /* If the data source is a temporary table, then we have to create 4271ccde15dSdrh ** a loop because there might be multiple rows of data. If the data 428142e30dfSdrh ** source is a subroutine call from the SELECT statement, then we need 429142e30dfSdrh ** to launch the SELECT statement processing. 4301ccde15dSdrh */ 431142e30dfSdrh if( useTempTable ){ 4324adee20fSdanielk1977 iBreak = sqlite3VdbeMakeLabel(v); 4334adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Rewind, srcTab, iBreak); 4344adee20fSdanielk1977 iCont = sqlite3VdbeCurrentAddr(v); 435142e30dfSdrh }else if( pSelect ){ 4364adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, iSelectLoop); 4374adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iInsertBlock); 438bed8690fSdrh } 4391ccde15dSdrh 4405cf590c1Sdrh /* Run the BEFORE and INSTEAD OF triggers, if there are any 44170ce3f0cSdrh */ 4424adee20fSdanielk1977 endOfLoop = sqlite3VdbeMakeLabel(v); 44370ce3f0cSdrh if( before_triggers ){ 444c3f9bad2Sdanielk1977 44570ce3f0cSdrh /* build the NEW.* reference row. Note that if there is an INTEGER 44670ce3f0cSdrh ** PRIMARY KEY into which a NULL is being inserted, that NULL will be 44770ce3f0cSdrh ** translated into a unique ID for the row. But on a BEFORE trigger, 44870ce3f0cSdrh ** we do not know what the unique ID will be (because the insert has 44970ce3f0cSdrh ** not happened yet) so we substitute a rowid of -1 45070ce3f0cSdrh */ 45170ce3f0cSdrh if( keyColumn<0 ){ 4524adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, -1, 0); 45370ce3f0cSdrh }else if( useTempTable ){ 4544adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, srcTab, keyColumn); 45570ce3f0cSdrh }else if( pSelect ){ 4564adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nColumn - keyColumn - 1, 1); 45770ce3f0cSdrh }else{ 4584adee20fSdanielk1977 sqlite3ExprCode(pParse, pList->a[keyColumn].pExpr); 4594adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotNull, -1, sqlite3VdbeCurrentAddr(v)+3); 4604adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 4614adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, -1, 0); 4624adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MustBeInt, 0, 0); 46370ce3f0cSdrh } 46470ce3f0cSdrh 46570ce3f0cSdrh /* Create the new column data 46670ce3f0cSdrh */ 467c3f9bad2Sdanielk1977 for(i=0; i<pTab->nCol; i++){ 468c3f9bad2Sdanielk1977 if( pColumn==0 ){ 469c3f9bad2Sdanielk1977 j = i; 470c3f9bad2Sdanielk1977 }else{ 471c3f9bad2Sdanielk1977 for(j=0; j<pColumn->nId; j++){ 472c3f9bad2Sdanielk1977 if( pColumn->a[j].idx==i ) break; 473c3f9bad2Sdanielk1977 } 474c3f9bad2Sdanielk1977 } 475c3f9bad2Sdanielk1977 if( pColumn && j>=pColumn->nId ){ 4760f69c1e3Sdanielk1977 sqlite3VdbeOp3(v, OP_String8, 0, 0, pTab->aCol[i].zDflt, P3_STATIC); 477142e30dfSdrh }else if( useTempTable ){ 4784adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, srcTab, j); 479142e30dfSdrh }else if( pSelect ){ 4804adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nColumn-j-1, 1); 481c3f9bad2Sdanielk1977 }else{ 4824adee20fSdanielk1977 sqlite3ExprCode(pParse, pList->a[j].pExpr); 483c3f9bad2Sdanielk1977 } 484c3f9bad2Sdanielk1977 } 4854adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, pTab->nCol, 0); 486a37cdde0Sdanielk1977 487a37cdde0Sdanielk1977 /* If this is an INSERT on a view with an INSTEAD OF INSERT trigger, 488a37cdde0Sdanielk1977 ** do not attempt any conversions before assembling the record. 489a37cdde0Sdanielk1977 ** If this is a real table, attempt conversions as required by the 490a37cdde0Sdanielk1977 ** table column affinities. 491a37cdde0Sdanielk1977 */ 492a37cdde0Sdanielk1977 if( !isView ){ 493a37cdde0Sdanielk1977 sqlite3TableAffinityStr(v, pTab); 494a37cdde0Sdanielk1977 } 4954adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, newIdx, 0); 496c3f9bad2Sdanielk1977 4975cf590c1Sdrh /* Fire BEFORE or INSTEAD OF triggers */ 4984adee20fSdanielk1977 if( sqlite3CodeRowTrigger(pParse, TK_INSERT, 0, TK_BEFORE, pTab, 499b2fe7d8cSdrh newIdx, -1, onError, endOfLoop) ){ 500f29ce559Sdanielk1977 goto insert_cleanup; 501f29ce559Sdanielk1977 } 50270ce3f0cSdrh } 503c3f9bad2Sdanielk1977 50470ce3f0cSdrh /* If any triggers exists, the opening of tables and indices is deferred 50570ce3f0cSdrh ** until now. 50670ce3f0cSdrh */ 5075cf590c1Sdrh if( row_triggers_exist && !isView ){ 508c3f9bad2Sdanielk1977 base = pParse->nTab; 5094adee20fSdanielk1977 idx = sqlite3OpenTableAndIndices(pParse, pTab, base); 510c3f9bad2Sdanielk1977 pParse->nTab += idx; 511c3f9bad2Sdanielk1977 } 512c3f9bad2Sdanielk1977 5134a32431cSdrh /* Push the record number for the new entry onto the stack. The 5144a32431cSdrh ** record number is a randomly generate integer created by NewRecno 5154a32431cSdrh ** except when the table has an INTEGER PRIMARY KEY column, in which 516b419a926Sdrh ** case the record number is the same as that column. 5171ccde15dSdrh */ 5185cf590c1Sdrh if( !isView ){ 5194a32431cSdrh if( keyColumn>=0 ){ 520142e30dfSdrh if( useTempTable ){ 5214adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, srcTab, keyColumn); 522142e30dfSdrh }else if( pSelect ){ 5234adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nColumn - keyColumn - 1, 1); 5244a32431cSdrh }else{ 5254adee20fSdanielk1977 sqlite3ExprCode(pParse, pList->a[keyColumn].pExpr); 52627a32783Sdrh } 527e1e68f49Sdrh /* If the PRIMARY KEY expression is NULL, then use OP_NewRecno 528e1e68f49Sdrh ** to generate a unique primary key value. 529e1e68f49Sdrh */ 5304adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NotNull, -1, sqlite3VdbeCurrentAddr(v)+3); 5314adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 5324adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NewRecno, base, 0); 5334adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MustBeInt, 0, 0); 5344a32431cSdrh }else{ 5354adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_NewRecno, base, 0); 5364a32431cSdrh } 5374a32431cSdrh 538aacc543eSdrh /* Push onto the stack, data for all columns of the new entry, beginning 5394a32431cSdrh ** with the first column. 5404a32431cSdrh */ 541cce7d176Sdrh for(i=0; i<pTab->nCol; i++){ 5424a32431cSdrh if( i==pTab->iPKey ){ 5434a32431cSdrh /* The value of the INTEGER PRIMARY KEY column is always a NULL. 544aacc543eSdrh ** Whenever this column is read, the record number will be substituted 545aacc543eSdrh ** in its place. So will fill this column with a NULL to avoid 546aacc543eSdrh ** taking up data space with information that will never be used. */ 5470f69c1e3Sdanielk1977 sqlite3VdbeAddOp(v, OP_String8, 0, 0); 5484a32431cSdrh continue; 5494a32431cSdrh } 550967e8b73Sdrh if( pColumn==0 ){ 551cce7d176Sdrh j = i; 552cce7d176Sdrh }else{ 553967e8b73Sdrh for(j=0; j<pColumn->nId; j++){ 554967e8b73Sdrh if( pColumn->a[j].idx==i ) break; 555cce7d176Sdrh } 556cce7d176Sdrh } 557967e8b73Sdrh if( pColumn && j>=pColumn->nId ){ 5580f69c1e3Sdanielk1977 sqlite3VdbeOp3(v, OP_String8, 0, 0, pTab->aCol[i].zDflt, P3_STATIC); 559142e30dfSdrh }else if( useTempTable ){ 5604adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Column, srcTab, j); 561142e30dfSdrh }else if( pSelect ){ 5624adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, i+nColumn-j, 1); 563cce7d176Sdrh }else{ 5644adee20fSdanielk1977 sqlite3ExprCode(pParse, pList->a[j].pExpr); 565cce7d176Sdrh } 566cce7d176Sdrh } 5671ccde15dSdrh 5680ca3e24bSdrh /* Generate code to check constraints and generate index keys and 5690ca3e24bSdrh ** do the insertion. 5704a32431cSdrh */ 5714adee20fSdanielk1977 sqlite3GenerateConstraintChecks(pParse, pTab, base, 0, keyColumn>=0, 572a0217ba7Sdrh 0, onError, endOfLoop); 5734adee20fSdanielk1977 sqlite3CompleteInsertion(pParse, pTab, base, 0,0,0, 57470ce3f0cSdrh after_triggers ? newIdx : -1); 5755cf590c1Sdrh } 5761bee3d7bSdrh 577feeb1394Sdrh /* Update the count of rows that are inserted 5781bee3d7bSdrh */ 579142e30dfSdrh if( (db->flags & SQLITE_CountRows)!=0 ){ 5804adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemIncr, iCntMem, 0); 5811bee3d7bSdrh } 582c3f9bad2Sdanielk1977 583c3f9bad2Sdanielk1977 if( row_triggers_exist ){ 584c3f9bad2Sdanielk1977 /* Close all tables opened */ 5855cf590c1Sdrh if( !isView ){ 5864adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, base, 0); 587c3f9bad2Sdanielk1977 for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){ 5884adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, idx+base, 0); 589c3f9bad2Sdanielk1977 } 590c3f9bad2Sdanielk1977 } 591c3f9bad2Sdanielk1977 592c3f9bad2Sdanielk1977 /* Code AFTER triggers */ 5934adee20fSdanielk1977 if( sqlite3CodeRowTrigger(pParse, TK_INSERT, 0, TK_AFTER, pTab, newIdx, -1, 5946f34903eSdanielk1977 onError, endOfLoop) ){ 595f29ce559Sdanielk1977 goto insert_cleanup; 596f29ce559Sdanielk1977 } 597c3f9bad2Sdanielk1977 } 5981bee3d7bSdrh 5991ccde15dSdrh /* The bottom of the loop, if the data source is a SELECT statement 6001ccde15dSdrh */ 6014adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, endOfLoop); 602142e30dfSdrh if( useTempTable ){ 6034adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Next, srcTab, iCont); 6044adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iBreak); 6054adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, srcTab, 0); 606142e30dfSdrh }else if( pSelect ){ 6074adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, nColumn, 0); 6084adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Return, 0, 0); 6094adee20fSdanielk1977 sqlite3VdbeResolveLabel(v, iCleanup); 6106b56344dSdrh } 611c3f9bad2Sdanielk1977 612c3f9bad2Sdanielk1977 if( !row_triggers_exist ){ 613c3f9bad2Sdanielk1977 /* Close all tables opened */ 6144adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, base, 0); 6156b56344dSdrh for(idx=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, idx++){ 6164adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, idx+base, 0); 617cce7d176Sdrh } 618c3f9bad2Sdanielk1977 } 619c3f9bad2Sdanielk1977 6204adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_SetCounts, 0, 0); 6214adee20fSdanielk1977 sqlite3EndWriteOperation(pParse); 6225e00f6c7Sdrh 6231bee3d7bSdrh /* 6241bee3d7bSdrh ** Return the number of rows inserted. 6251bee3d7bSdrh */ 626142e30dfSdrh if( db->flags & SQLITE_CountRows ){ 6274adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MemLoad, iCntMem, 0); 6284adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Callback, 1, 0); 62922322fd4Sdanielk1977 sqlite3VdbeSetNumCols(v, 1); 6303cf86063Sdanielk1977 sqlite3VdbeSetColName(v, 0, "rows inserted", P3_STATIC); 6311bee3d7bSdrh } 632cce7d176Sdrh 633cce7d176Sdrh insert_cleanup: 6344adee20fSdanielk1977 sqlite3SrcListDelete(pTabList); 6354adee20fSdanielk1977 if( pList ) sqlite3ExprListDelete(pList); 6364adee20fSdanielk1977 if( pSelect ) sqlite3SelectDelete(pSelect); 6374adee20fSdanielk1977 sqlite3IdListDelete(pColumn); 638cce7d176Sdrh } 6399cfcf5d4Sdrh 6409cfcf5d4Sdrh /* 6419cfcf5d4Sdrh ** Generate code to do a constraint check prior to an INSERT or an UPDATE. 6429cfcf5d4Sdrh ** 6439cfcf5d4Sdrh ** When this routine is called, the stack contains (from bottom to top) 6440ca3e24bSdrh ** the following values: 6450ca3e24bSdrh ** 646b2fe7d8cSdrh ** 1. The recno of the row to be updated before the update. This 647b419a926Sdrh ** value is omitted unless we are doing an UPDATE that involves a 648b419a926Sdrh ** change to the record number. 6490ca3e24bSdrh ** 650b419a926Sdrh ** 2. The recno of the row after the update. 6510ca3e24bSdrh ** 6520ca3e24bSdrh ** 3. The data in the first column of the entry after the update. 6530ca3e24bSdrh ** 6540ca3e24bSdrh ** i. Data from middle columns... 6550ca3e24bSdrh ** 6560ca3e24bSdrh ** N. The data in the last column of the entry after the update. 6570ca3e24bSdrh ** 658b419a926Sdrh ** The old recno shown as entry (1) above is omitted unless both isUpdate 6591c92853dSdrh ** and recnoChng are 1. isUpdate is true for UPDATEs and false for 6601c92853dSdrh ** INSERTs and recnoChng is true if the record number is being changed. 6610ca3e24bSdrh ** 6620ca3e24bSdrh ** The code generated by this routine pushes additional entries onto 6630ca3e24bSdrh ** the stack which are the keys for new index entries for the new record. 6640ca3e24bSdrh ** The order of index keys is the same as the order of the indices on 6650ca3e24bSdrh ** the pTable->pIndex list. A key is only created for index i if 6660ca3e24bSdrh ** aIdxUsed!=0 and aIdxUsed[i]!=0. 6679cfcf5d4Sdrh ** 6689cfcf5d4Sdrh ** This routine also generates code to check constraints. NOT NULL, 6699cfcf5d4Sdrh ** CHECK, and UNIQUE constraints are all checked. If a constraint fails, 6701c92853dSdrh ** then the appropriate action is performed. There are five possible 6711c92853dSdrh ** actions: ROLLBACK, ABORT, FAIL, REPLACE, and IGNORE. 6729cfcf5d4Sdrh ** 6739cfcf5d4Sdrh ** Constraint type Action What Happens 6749cfcf5d4Sdrh ** --------------- ---------- ---------------------------------------- 6751c92853dSdrh ** any ROLLBACK The current transaction is rolled back and 67624b03fd0Sdanielk1977 ** sqlite3_exec() returns immediately with a 6779cfcf5d4Sdrh ** return code of SQLITE_CONSTRAINT. 6789cfcf5d4Sdrh ** 6791c92853dSdrh ** any ABORT Back out changes from the current command 6801c92853dSdrh ** only (do not do a complete rollback) then 68124b03fd0Sdanielk1977 ** cause sqlite3_exec() to return immediately 6821c92853dSdrh ** with SQLITE_CONSTRAINT. 6831c92853dSdrh ** 6841c92853dSdrh ** any FAIL Sqlite_exec() returns immediately with a 6851c92853dSdrh ** return code of SQLITE_CONSTRAINT. The 6861c92853dSdrh ** transaction is not rolled back and any 6871c92853dSdrh ** prior changes are retained. 6881c92853dSdrh ** 6899cfcf5d4Sdrh ** any IGNORE The record number and data is popped from 6909cfcf5d4Sdrh ** the stack and there is an immediate jump 6919cfcf5d4Sdrh ** to label ignoreDest. 6929cfcf5d4Sdrh ** 6939cfcf5d4Sdrh ** NOT NULL REPLACE The NULL value is replace by the default 6949cfcf5d4Sdrh ** value for that column. If the default value 6959cfcf5d4Sdrh ** is NULL, the action is the same as ABORT. 6969cfcf5d4Sdrh ** 6979cfcf5d4Sdrh ** UNIQUE REPLACE The other row that conflicts with the row 6989cfcf5d4Sdrh ** being inserted is removed. 6999cfcf5d4Sdrh ** 7009cfcf5d4Sdrh ** CHECK REPLACE Illegal. The results in an exception. 7019cfcf5d4Sdrh ** 7021c92853dSdrh ** Which action to take is determined by the overrideError parameter. 7031c92853dSdrh ** Or if overrideError==OE_Default, then the pParse->onError parameter 7041c92853dSdrh ** is used. Or if pParse->onError==OE_Default then the onError value 7051c92853dSdrh ** for the constraint is used. 7069cfcf5d4Sdrh ** 707aaab5725Sdrh ** The calling routine must open a read/write cursor for pTab with 7089cfcf5d4Sdrh ** cursor number "base". All indices of pTab must also have open 7099cfcf5d4Sdrh ** read/write cursors with cursor number base+i for the i-th cursor. 7109cfcf5d4Sdrh ** Except, if there is no possibility of a REPLACE action then 7119cfcf5d4Sdrh ** cursors do not need to be open for indices where aIdxUsed[i]==0. 7129cfcf5d4Sdrh ** 7139cfcf5d4Sdrh ** If the isUpdate flag is true, it means that the "base" cursor is 7149cfcf5d4Sdrh ** initially pointing to an entry that is being updated. The isUpdate 7159cfcf5d4Sdrh ** flag causes extra code to be generated so that the "base" cursor 7169cfcf5d4Sdrh ** is still pointing at the same entry after the routine returns. 7179cfcf5d4Sdrh ** Without the isUpdate flag, the "base" cursor might be moved. 7189cfcf5d4Sdrh */ 7194adee20fSdanielk1977 void sqlite3GenerateConstraintChecks( 7209cfcf5d4Sdrh Parse *pParse, /* The parser context */ 7219cfcf5d4Sdrh Table *pTab, /* the table into which we are inserting */ 7229cfcf5d4Sdrh int base, /* Index of a read/write cursor pointing at pTab */ 7239cfcf5d4Sdrh char *aIdxUsed, /* Which indices are used. NULL means all are used */ 7240ca3e24bSdrh int recnoChng, /* True if the record number will change */ 725b419a926Sdrh int isUpdate, /* True for UPDATE, False for INSERT */ 7269cfcf5d4Sdrh int overrideError, /* Override onError to this if not OE_Default */ 727b419a926Sdrh int ignoreDest /* Jump to this label on an OE_Ignore resolution */ 7289cfcf5d4Sdrh ){ 7299cfcf5d4Sdrh int i; 7309cfcf5d4Sdrh Vdbe *v; 7319cfcf5d4Sdrh int nCol; 7329cfcf5d4Sdrh int onError; 7339cfcf5d4Sdrh int addr; 7349cfcf5d4Sdrh int extra; 7350ca3e24bSdrh int iCur; 7360ca3e24bSdrh Index *pIdx; 7370ca3e24bSdrh int seenReplace = 0; 738cfe9a69fSdanielk1977 int jumpInst1=0, jumpInst2; 7390ca3e24bSdrh int contAddr; 740b419a926Sdrh int hasTwoRecnos = (isUpdate && recnoChng); 7419cfcf5d4Sdrh 7424adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 7439cfcf5d4Sdrh assert( v!=0 ); 744417be79cSdrh assert( pTab->pSelect==0 ); /* This table is not a VIEW */ 7459cfcf5d4Sdrh nCol = pTab->nCol; 7469cfcf5d4Sdrh 7479cfcf5d4Sdrh /* Test all NOT NULL constraints. 7489cfcf5d4Sdrh */ 7499cfcf5d4Sdrh for(i=0; i<nCol; i++){ 7500ca3e24bSdrh if( i==pTab->iPKey ){ 7510ca3e24bSdrh continue; 7520ca3e24bSdrh } 7539cfcf5d4Sdrh onError = pTab->aCol[i].notNull; 7540ca3e24bSdrh if( onError==OE_None ) continue; 7559cfcf5d4Sdrh if( overrideError!=OE_Default ){ 7569cfcf5d4Sdrh onError = overrideError; 757a996e477Sdrh }else if( onError==OE_Default ){ 758a996e477Sdrh onError = OE_Abort; 7599cfcf5d4Sdrh } 7609cfcf5d4Sdrh if( onError==OE_Replace && pTab->aCol[i].zDflt==0 ){ 7619cfcf5d4Sdrh onError = OE_Abort; 7629cfcf5d4Sdrh } 7634adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol-1-i, 1); 7644adee20fSdanielk1977 addr = sqlite3VdbeAddOp(v, OP_NotNull, 1, 0); 7659cfcf5d4Sdrh switch( onError ){ 7661c92853dSdrh case OE_Rollback: 7671c92853dSdrh case OE_Abort: 7681c92853dSdrh case OE_Fail: { 769483750baSdrh char *zMsg = 0; 7704adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Halt, SQLITE_CONSTRAINT, onError); 7714adee20fSdanielk1977 sqlite3SetString(&zMsg, pTab->zName, ".", pTab->aCol[i].zName, 77241743984Sdrh " may not be NULL", (char*)0); 7734adee20fSdanielk1977 sqlite3VdbeChangeP3(v, -1, zMsg, P3_DYNAMIC); 7749cfcf5d4Sdrh break; 7759cfcf5d4Sdrh } 7769cfcf5d4Sdrh case OE_Ignore: { 7774adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, nCol+1+hasTwoRecnos, 0); 7784adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, ignoreDest); 7799cfcf5d4Sdrh break; 7809cfcf5d4Sdrh } 7819cfcf5d4Sdrh case OE_Replace: { 7820f69c1e3Sdanielk1977 sqlite3VdbeOp3(v, OP_String8, 0, 0, pTab->aCol[i].zDflt, P3_STATIC); 7834adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Push, nCol-i, 0); 7849cfcf5d4Sdrh break; 7859cfcf5d4Sdrh } 7860ca3e24bSdrh default: assert(0); 7879cfcf5d4Sdrh } 7884adee20fSdanielk1977 sqlite3VdbeChangeP2(v, addr, sqlite3VdbeCurrentAddr(v)); 7899cfcf5d4Sdrh } 7909cfcf5d4Sdrh 7919cfcf5d4Sdrh /* Test all CHECK constraints 7929cfcf5d4Sdrh */ 7930bd1f4eaSdrh /**** TBD ****/ 7949cfcf5d4Sdrh 7950bd1f4eaSdrh /* If we have an INTEGER PRIMARY KEY, make sure the primary key 7960bd1f4eaSdrh ** of the new record does not previously exist. Except, if this 7970bd1f4eaSdrh ** is an UPDATE and the primary key is not changing, that is OK. 7989cfcf5d4Sdrh */ 7995383ae5cSdrh if( recnoChng ){ 8000ca3e24bSdrh onError = pTab->keyConf; 8010ca3e24bSdrh if( overrideError!=OE_Default ){ 8020ca3e24bSdrh onError = overrideError; 803a996e477Sdrh }else if( onError==OE_Default ){ 804a996e477Sdrh onError = OE_Abort; 8050ca3e24bSdrh } 806a0217ba7Sdrh 80779b0c956Sdrh if( isUpdate ){ 8084adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+1, 1); 8094adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+1, 1); 8104adee20fSdanielk1977 jumpInst1 = sqlite3VdbeAddOp(v, OP_Eq, 0, 0); 81179b0c956Sdrh } 8124adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol, 1); 8134adee20fSdanielk1977 jumpInst2 = sqlite3VdbeAddOp(v, OP_NotExists, base, 0); 8140ca3e24bSdrh switch( onError ){ 815a0217ba7Sdrh default: { 816a0217ba7Sdrh onError = OE_Abort; 817a0217ba7Sdrh /* Fall thru into the next case */ 818a0217ba7Sdrh } 8191c92853dSdrh case OE_Rollback: 8201c92853dSdrh case OE_Abort: 8211c92853dSdrh case OE_Fail: { 8224adee20fSdanielk1977 sqlite3VdbeOp3(v, OP_Halt, SQLITE_CONSTRAINT, onError, 823701a0aebSdrh "PRIMARY KEY must be unique", P3_STATIC); 8240ca3e24bSdrh break; 8250ca3e24bSdrh } 8265383ae5cSdrh case OE_Replace: { 8274adee20fSdanielk1977 sqlite3GenerateRowIndexDelete(pParse->db, v, pTab, base, 0); 8285383ae5cSdrh if( isUpdate ){ 8294adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+hasTwoRecnos, 1); 8307cf6e4deSdrh sqlite3VdbeAddOp(v, OP_MoveGe, base, 0); 8315383ae5cSdrh } 8325383ae5cSdrh seenReplace = 1; 8335383ae5cSdrh break; 8345383ae5cSdrh } 8350ca3e24bSdrh case OE_Ignore: { 8365383ae5cSdrh assert( seenReplace==0 ); 8374adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, nCol+1+hasTwoRecnos, 0); 8384adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, ignoreDest); 8390ca3e24bSdrh break; 8400ca3e24bSdrh } 8410ca3e24bSdrh } 8424adee20fSdanielk1977 contAddr = sqlite3VdbeCurrentAddr(v); 8434adee20fSdanielk1977 sqlite3VdbeChangeP2(v, jumpInst2, contAddr); 844f5905aa7Sdrh if( isUpdate ){ 8454adee20fSdanielk1977 sqlite3VdbeChangeP2(v, jumpInst1, contAddr); 8464adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+1, 1); 8477cf6e4deSdrh sqlite3VdbeAddOp(v, OP_MoveGe, base, 0); 8480ca3e24bSdrh } 8490ca3e24bSdrh } 8500bd1f4eaSdrh 8510bd1f4eaSdrh /* Test all UNIQUE constraints by creating entries for each UNIQUE 8520bd1f4eaSdrh ** index and making sure that duplicate entries do not already exist. 8530bd1f4eaSdrh ** Add the new records to the indices as we go. 8540bd1f4eaSdrh */ 855b2fe7d8cSdrh extra = -1; 856b2fe7d8cSdrh for(iCur=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, iCur++){ 857b2fe7d8cSdrh if( aIdxUsed && aIdxUsed[iCur]==0 ) continue; /* Skip unused indices */ 8589cfcf5d4Sdrh extra++; 859b2fe7d8cSdrh 860b2fe7d8cSdrh /* Create a key for accessing the index entry */ 8614adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+extra, 1); 8629cfcf5d4Sdrh for(i=0; i<pIdx->nColumn; i++){ 8639cfcf5d4Sdrh int idx = pIdx->aiColumn[i]; 8649cfcf5d4Sdrh if( idx==pTab->iPKey ){ 8654adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, i+extra+nCol+1, 1); 8669cfcf5d4Sdrh }else{ 8674adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, i+extra+nCol-idx, 1); 8689cfcf5d4Sdrh } 8699cfcf5d4Sdrh } 870*ededfd5eSdanielk1977 jumpInst1 = sqlite3VdbeAddOp(v, OP_MakeRecord, pIdx->nColumn, (1<<24)); 871a37cdde0Sdanielk1977 sqlite3IndexAffinityStr(v, pIdx); 872b2fe7d8cSdrh 873b2fe7d8cSdrh /* Find out what action to take in case there is an indexing conflict */ 8749cfcf5d4Sdrh onError = pIdx->onError; 875b2fe7d8cSdrh if( onError==OE_None ) continue; /* pIdx is not a UNIQUE index */ 8769cfcf5d4Sdrh if( overrideError!=OE_Default ){ 8779cfcf5d4Sdrh onError = overrideError; 878a996e477Sdrh }else if( onError==OE_Default ){ 879a996e477Sdrh onError = OE_Abort; 8809cfcf5d4Sdrh } 8815383ae5cSdrh if( seenReplace ){ 8825383ae5cSdrh if( onError==OE_Ignore ) onError = OE_Replace; 8835383ae5cSdrh else if( onError==OE_Fail ) onError = OE_Abort; 8845383ae5cSdrh } 8855383ae5cSdrh 886b2fe7d8cSdrh 887b2fe7d8cSdrh /* Check to see if the new index entry will be unique */ 8884adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, extra+nCol+1+hasTwoRecnos, 1); 8894adee20fSdanielk1977 jumpInst2 = sqlite3VdbeAddOp(v, OP_IsUnique, base+iCur+1, 0); 890b2fe7d8cSdrh 891b2fe7d8cSdrh /* Generate code that executes if the new index entry is not unique */ 8929cfcf5d4Sdrh switch( onError ){ 8931c92853dSdrh case OE_Rollback: 8941c92853dSdrh case OE_Abort: 8951c92853dSdrh case OE_Fail: { 89637ed48edSdrh int j, n1, n2; 89737ed48edSdrh char zErrMsg[200]; 89837ed48edSdrh strcpy(zErrMsg, pIdx->nColumn>1 ? "columns " : "column "); 89937ed48edSdrh n1 = strlen(zErrMsg); 90037ed48edSdrh for(j=0; j<pIdx->nColumn && n1<sizeof(zErrMsg)-30; j++){ 90137ed48edSdrh char *zCol = pTab->aCol[pIdx->aiColumn[j]].zName; 90237ed48edSdrh n2 = strlen(zCol); 90337ed48edSdrh if( j>0 ){ 90437ed48edSdrh strcpy(&zErrMsg[n1], ", "); 90537ed48edSdrh n1 += 2; 90637ed48edSdrh } 90737ed48edSdrh if( n1+n2>sizeof(zErrMsg)-30 ){ 90837ed48edSdrh strcpy(&zErrMsg[n1], "..."); 90937ed48edSdrh n1 += 3; 91037ed48edSdrh break; 91137ed48edSdrh }else{ 91237ed48edSdrh strcpy(&zErrMsg[n1], zCol); 91337ed48edSdrh n1 += n2; 91437ed48edSdrh } 91537ed48edSdrh } 91637ed48edSdrh strcpy(&zErrMsg[n1], 91737ed48edSdrh pIdx->nColumn>1 ? " are not unique" : " is not unique"); 9184adee20fSdanielk1977 sqlite3VdbeOp3(v, OP_Halt, SQLITE_CONSTRAINT, onError, zErrMsg, 0); 9199cfcf5d4Sdrh break; 9209cfcf5d4Sdrh } 9219cfcf5d4Sdrh case OE_Ignore: { 9220ca3e24bSdrh assert( seenReplace==0 ); 9234adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, nCol+extra+3+hasTwoRecnos, 0); 9244adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Goto, 0, ignoreDest); 9259cfcf5d4Sdrh break; 9269cfcf5d4Sdrh } 9279cfcf5d4Sdrh case OE_Replace: { 9284adee20fSdanielk1977 sqlite3GenerateRowDelete(pParse->db, v, pTab, base, 0); 9299cfcf5d4Sdrh if( isUpdate ){ 9304adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, nCol+extra+1+hasTwoRecnos, 1); 9317cf6e4deSdrh sqlite3VdbeAddOp(v, OP_MoveGe, base, 0); 9329cfcf5d4Sdrh } 9330ca3e24bSdrh seenReplace = 1; 9349cfcf5d4Sdrh break; 9359cfcf5d4Sdrh } 9360ca3e24bSdrh default: assert(0); 9379cfcf5d4Sdrh } 9384adee20fSdanielk1977 contAddr = sqlite3VdbeCurrentAddr(v); 939*ededfd5eSdanielk1977 assert( contAddr<(1<<24) ); 9400bd1f4eaSdrh #if NULL_DISTINCT_FOR_UNIQUE 941*ededfd5eSdanielk1977 sqlite3VdbeChangeP2(v, jumpInst1, contAddr | (1<<24)); 9420bd1f4eaSdrh #endif 9434adee20fSdanielk1977 sqlite3VdbeChangeP2(v, jumpInst2, contAddr); 9449cfcf5d4Sdrh } 9459cfcf5d4Sdrh } 9460ca3e24bSdrh 9470ca3e24bSdrh /* 9480ca3e24bSdrh ** This routine generates code to finish the INSERT or UPDATE operation 9494adee20fSdanielk1977 ** that was started by a prior call to sqlite3GenerateConstraintChecks. 9500ca3e24bSdrh ** The stack must contain keys for all active indices followed by data 9510ca3e24bSdrh ** and the recno for the new entry. This routine creates the new 9520ca3e24bSdrh ** entries in all indices and in the main table. 9530ca3e24bSdrh ** 954b419a926Sdrh ** The arguments to this routine should be the same as the first six 9554adee20fSdanielk1977 ** arguments to sqlite3GenerateConstraintChecks. 9560ca3e24bSdrh */ 9574adee20fSdanielk1977 void sqlite3CompleteInsertion( 9580ca3e24bSdrh Parse *pParse, /* The parser context */ 9590ca3e24bSdrh Table *pTab, /* the table into which we are inserting */ 9600ca3e24bSdrh int base, /* Index of a read/write cursor pointing at pTab */ 9610ca3e24bSdrh char *aIdxUsed, /* Which indices are used. NULL means all are used */ 962b419a926Sdrh int recnoChng, /* True if the record number will change */ 96370ce3f0cSdrh int isUpdate, /* True for UPDATE, False for INSERT */ 96470ce3f0cSdrh int newIdx /* Index of NEW table for triggers. -1 if none */ 9650ca3e24bSdrh ){ 9660ca3e24bSdrh int i; 9670ca3e24bSdrh Vdbe *v; 9680ca3e24bSdrh int nIdx; 9690ca3e24bSdrh Index *pIdx; 9700ca3e24bSdrh 9714adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 9720ca3e24bSdrh assert( v!=0 ); 973417be79cSdrh assert( pTab->pSelect==0 ); /* This table is not a VIEW */ 9740ca3e24bSdrh for(nIdx=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, nIdx++){} 9750ca3e24bSdrh for(i=nIdx-1; i>=0; i--){ 9760ca3e24bSdrh if( aIdxUsed && aIdxUsed[i]==0 ) continue; 9774adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_IdxPut, base+i+1, 0); 9780ca3e24bSdrh } 9794adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_MakeRecord, pTab->nCol, 0); 980a37cdde0Sdanielk1977 sqlite3TableAffinityStr(v, pTab); 98170ce3f0cSdrh if( newIdx>=0 ){ 9824adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, 1, 0); 9834adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Dup, 1, 0); 9844adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, newIdx, 0); 98570ce3f0cSdrh } 9864adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_PutIntKey, base, 987b0c374ffSrdc (pParse->trigStack?0:OPFLAG_NCHANGE) | 988b0c374ffSrdc (isUpdate?0:OPFLAG_LASTROWID) | OPFLAG_CSCHANGE); 989b419a926Sdrh if( isUpdate && recnoChng ){ 9904adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Pop, 1, 0); 9910ca3e24bSdrh } 9920ca3e24bSdrh } 993cd44690aSdrh 994cd44690aSdrh /* 995cd44690aSdrh ** Generate code that will open write cursors for a table and for all 996cd44690aSdrh ** indices of that table. The "base" parameter is the cursor number used 997cd44690aSdrh ** for the table. Indices are opened on subsequent cursors. 998cd44690aSdrh ** 999cd44690aSdrh ** Return the total number of cursors opened. This is always at least 1000cd44690aSdrh ** 1 (for the main table) plus more for each cursor. 1001cd44690aSdrh */ 10024adee20fSdanielk1977 int sqlite3OpenTableAndIndices(Parse *pParse, Table *pTab, int base){ 1003cd44690aSdrh int i; 1004cd44690aSdrh Index *pIdx; 10054adee20fSdanielk1977 Vdbe *v = sqlite3GetVdbe(pParse); 1006cd44690aSdrh assert( v!=0 ); 10074adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, pTab->iDb, 0); 1008d3d39e93Sdrh sqlite3VdbeAddOp(v, OP_OpenWrite, base, pTab->tnum); 1009b4964b72Sdanielk1977 sqlite3VdbeAddOp(v, OP_SetNumColumns, base, pTab->nCol); 1010cd44690aSdrh for(i=1, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, i++){ 10114adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Integer, pIdx->iDb, 0); 1012d3d39e93Sdrh sqlite3VdbeOp3(v, OP_OpenWrite, i+base, pIdx->tnum, 1013d3d39e93Sdrh (char*)&pIdx->keyInfo, P3_KEYINFO); 1014cd44690aSdrh } 1015cd44690aSdrh return i; 1016cd44690aSdrh } 1017