xref: /sqlite-3.40.0/src/insert.c (revision ededfd5e)
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