blob: 6547041472c381c61169cd327258138680bf26cf [file] [log] [blame]
drhcce7d172000-05-31 15:34:51 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drhcce7d172000-05-31 15:34:51 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drhcce7d172000-05-31 15:34:51 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drhcce7d172000-05-31 15:34:51 +000010**
drha3388cc2010-07-30 16:54:25 +000011*************************************************************************
drhcce7d172000-05-31 15:34:51 +000012** This file contains C code routines that are called by the parser
13** to handle UPDATE statements.
drhcce7d172000-05-31 15:34:51 +000014*/
15#include "sqliteInt.h"
16
drhedb193b2006-06-27 13:20:21 +000017#ifndef SQLITE_OMIT_VIRTUALTABLE
drh9c419382006-06-16 21:13:21 +000018/* Forward declaration */
19static void updateVirtualTable(
20 Parse *pParse, /* The parsing context */
21 SrcList *pSrc, /* The virtual table to be modified */
22 Table *pTab, /* The virtual table */
23 ExprList *pChanges, /* The columns to change in the UPDATE statement */
24 Expr *pRowidExpr, /* Expression used to recompute the rowid */
25 int *aXRef, /* Mapping from columns of pTab to entries in pChanges */
danb061d052011-04-25 18:49:57 +000026 Expr *pWhere, /* WHERE clause of the UPDATE statement */
27 int onError /* ON CONFLICT strategy */
drh9c419382006-06-16 21:13:21 +000028);
drhedb193b2006-06-27 13:20:21 +000029#endif /* SQLITE_OMIT_VIRTUALTABLE */
drh9c419382006-06-16 21:13:21 +000030
drhcce7d172000-05-31 15:34:51 +000031/*
drh8a512562005-11-14 22:29:05 +000032** The most recently coded instruction was an OP_Column to retrieve the
drh66a51672008-01-03 00:01:23 +000033** i-th column of table pTab. This routine sets the P4 parameter of the
danielk1977aee18ef2005-03-09 12:26:50 +000034** OP_Column to the default value, if any.
35**
36** The default value of a column is specified by a DEFAULT clause in the
37** column definition. This was either supplied by the user when the table
38** was created, or added later to the table definition by an ALTER TABLE
39** command. If the latter, then the row-records in the table btree on disk
40** may not contain a value for the column and the default value, taken
drh66a51672008-01-03 00:01:23 +000041** from the P4 parameter of the OP_Column instruction, is returned instead.
danielk1977aee18ef2005-03-09 12:26:50 +000042** If the former, then all row-records are guaranteed to include a value
drh66a51672008-01-03 00:01:23 +000043** for the column and the P4 value is not required.
danielk1977aee18ef2005-03-09 12:26:50 +000044**
45** Column definitions created by an ALTER TABLE command may only have
46** literal default values specified: a number, null or a string. (If a more
47** complicated default expression value was provided, it is evaluated
48** when the ALTER TABLE is executed and one of the literal values written
drh1e32bed2020-06-19 13:33:53 +000049** into the sqlite_schema table.)
danielk1977aee18ef2005-03-09 12:26:50 +000050**
drh66a51672008-01-03 00:01:23 +000051** Therefore, the P4 parameter is only required if the default value for
danielk1977aee18ef2005-03-09 12:26:50 +000052** the column is a literal number, string or null. The sqlite3ValueFromExpr()
53** function is capable of transforming these types of expressions into
54** sqlite3_value objects.
danielk1977c7538b52009-07-27 10:05:04 +000055**
drh8c1febb2020-05-16 18:01:02 +000056** If column as REAL affinity and the table is an ordinary b-tree table
57** (not a virtual table) then the value might have been stored as an
58** integer. In that case, add an OP_RealAffinity opcode to make sure
59** it has been converted into REAL.
danielk1977aee18ef2005-03-09 12:26:50 +000060*/
danielk1977c7538b52009-07-27 10:05:04 +000061void sqlite3ColumnDefault(Vdbe *v, Table *pTab, int i, int iReg){
drhe289d602009-05-05 15:46:09 +000062 assert( pTab!=0 );
drhf38524d2021-08-02 16:41:57 +000063 if( !IsView(pTab) ){
dan7a419232013-08-06 20:01:43 +000064 sqlite3_value *pValue = 0;
danielk197714db2662006-01-09 16:12:04 +000065 u8 enc = ENC(sqlite3VdbeDb(v));
danielk1977aee18ef2005-03-09 12:26:50 +000066 Column *pCol = &pTab->aCol[i];
drhcf9d36d2021-08-02 18:03:43 +000067 VdbeComment((v, "%s.%s", pTab->zName, pCol->zCnName));
danielk1977c9cf6e32007-06-25 16:29:33 +000068 assert( i<pTab->nCol );
drh79cf2b72021-07-31 20:30:41 +000069 sqlite3ValueFromExpr(sqlite3VdbeDb(v),
70 sqlite3ColumnExpr(pTab,pCol), enc,
drhd4e70eb2008-01-02 00:34:36 +000071 pCol->affinity, &pValue);
drh29dda4a2005-07-21 18:23:20 +000072 if( pValue ){
drhf14b7fb2016-12-07 21:35:55 +000073 sqlite3VdbeAppendP4(v, pValue, P4_MEM);
drh29dda4a2005-07-21 18:23:20 +000074 }
danielk1977aee18ef2005-03-09 12:26:50 +000075 }
drhda060052017-01-16 16:43:02 +000076#ifndef SQLITE_OMIT_FLOATING_POINT
drh8c1febb2020-05-16 18:01:02 +000077 if( pTab->aCol[i].affinity==SQLITE_AFF_REAL && !IsVirtual(pTab) ){
drhda060052017-01-16 16:43:02 +000078 sqlite3VdbeAddOp1(v, OP_RealAffinity, iReg);
79 }
80#endif
danielk1977aee18ef2005-03-09 12:26:50 +000081}
82
83/*
drhe9816d82018-09-15 21:38:48 +000084** Check to see if column iCol of index pIdx references any of the
85** columns defined by aXRef and chngRowid. Return true if it does
drh40b4e7a2018-09-17 20:47:38 +000086** and false if not. This is an optimization. False-positives are a
87** performance degradation, but false-negatives can result in a corrupt
88** index and incorrect answers.
drhe9816d82018-09-15 21:38:48 +000089**
drhe9816d82018-09-15 21:38:48 +000090** aXRef[j] will be non-negative if column j of the original table is
91** being updated. chngRowid will be true if the rowid of the table is
92** being updated.
93*/
drh2b7cdf62018-09-16 15:01:25 +000094static int indexColumnIsBeingUpdated(
95 Index *pIdx, /* The index to check */
96 int iCol, /* Which column of the index to check */
drhe9816d82018-09-15 21:38:48 +000097 int *aXRef, /* aXRef[j]>=0 if column j is being updated */
98 int chngRowid /* true if the rowid is being updated */
99){
drh2b7cdf62018-09-16 15:01:25 +0000100 i16 iIdxCol = pIdx->aiColumn[iCol];
drh86f34922018-09-16 18:23:48 +0000101 assert( iIdxCol!=XN_ROWID ); /* Cannot index rowid */
drh2b7cdf62018-09-16 15:01:25 +0000102 if( iIdxCol>=0 ){
103 return aXRef[iIdxCol]>=0;
104 }
drh2b7cdf62018-09-16 15:01:25 +0000105 assert( iIdxCol==XN_EXPR );
drhe9816d82018-09-15 21:38:48 +0000106 assert( pIdx->aColExpr!=0 );
107 assert( pIdx->aColExpr->a[iCol].pExpr!=0 );
108 return sqlite3ExprReferencesUpdatedColumn(pIdx->aColExpr->a[iCol].pExpr,
109 aXRef,chngRowid);
110}
111
112/*
drh40b4e7a2018-09-17 20:47:38 +0000113** Check to see if index pIdx is a partial index whose conditional
114** expression might change values due to an UPDATE. Return true if
115** the index is subject to change and false if the index is guaranteed
116** to be unchanged. This is an optimization. False-positives are a
117** performance degradation, but false-negatives can result in a corrupt
118** index and incorrect answers.
119**
120** aXRef[j] will be non-negative if column j of the original table is
121** being updated. chngRowid will be true if the rowid of the table is
122** being updated.
123*/
124static int indexWhereClauseMightChange(
125 Index *pIdx, /* The index to check */
126 int *aXRef, /* aXRef[j]>=0 if column j is being updated */
127 int chngRowid /* true if the rowid is being updated */
128){
129 if( pIdx->pPartIdxWhere==0 ) return 0;
130 return sqlite3ExprReferencesUpdatedColumn(pIdx->pPartIdxWhere,
131 aXRef, chngRowid);
132}
133
134/*
dan576d5a82020-07-15 18:30:01 +0000135** Allocate and return a pointer to an expression of type TK_ROW with
136** Expr.iColumn set to value (iCol+1). The resolver will modify the
137** expression to be a TK_COLUMN reading column iCol of the first
138** table in the source-list (pSrc->a[0]).
139*/
140static Expr *exprRowColumn(Parse *pParse, int iCol){
141 Expr *pRet = sqlite3PExpr(pParse, TK_ROW, 0, 0);
142 if( pRet ) pRet->iColumn = iCol+1;
143 return pRet;
144}
145
146/*
147** Assuming both the pLimit and pOrderBy parameters are NULL, this function
148** generates VM code to run the query:
dan69887c92020-04-27 20:55:33 +0000149**
dan576d5a82020-07-15 18:30:01 +0000150** SELECT <other-columns>, pChanges FROM pTabList WHERE pWhere
dan69887c92020-04-27 20:55:33 +0000151**
152** and write the results to the ephemeral table already opened as cursor
153** iEph. None of pChanges, pTabList or pWhere are modified or consumed by
154** this function, they must be deleted by the caller.
155**
dan576d5a82020-07-15 18:30:01 +0000156** Or, if pLimit and pOrderBy are not NULL, and pTab is not a view:
157**
158** SELECT <other-columns>, pChanges FROM pTabList
159** WHERE pWhere
160** GROUP BY <other-columns>
161** ORDER BY pOrderBy LIMIT pLimit
162**
163** If pTab is a view, the GROUP BY clause is omitted.
164**
dan69887c92020-04-27 20:55:33 +0000165** Exactly how results are written to table iEph, and exactly what
166** the <other-columns> in the query above are is determined by the type
167** of table pTabList->a[0].pTab.
168**
169** If the table is a WITHOUT ROWID table, then argument pPk must be its
170** PRIMARY KEY. In this case <other-columns> are the primary key columns
171** of the table, in order. The results of the query are written to ephemeral
172** table iEph as index keys, using OP_IdxInsert.
173**
174** If the table is actually a view, then <other-columns> are all columns of
175** the view. The results are written to the ephemeral table iEph as records
176** with automatically assigned integer keys.
177**
178** If the table is a virtual or ordinary intkey table, then <other-columns>
179** is its rowid. For a virtual table, the results are written to iEph as
180** records with automatically assigned integer keys For intkey tables, the
181** rowid value in <other-columns> is used as the integer key, and the
182** remaining fields make up the table record.
183*/
dan576d5a82020-07-15 18:30:01 +0000184static void updateFromSelect(
dan69887c92020-04-27 20:55:33 +0000185 Parse *pParse, /* Parse context */
186 int iEph, /* Cursor for open eph. table */
187 Index *pPk, /* PK if table 0 is WITHOUT ROWID */
188 ExprList *pChanges, /* List of expressions to return */
189 SrcList *pTabList, /* List of tables to select from */
danf2972b62020-04-29 20:11:01 +0000190 Expr *pWhere, /* WHERE clause for query */
drha7c74002020-07-18 18:44:59 +0000191 ExprList *pOrderBy, /* ORDER BY clause */
192 Expr *pLimit /* LIMIT clause */
dan69887c92020-04-27 20:55:33 +0000193){
194 int i;
dan69887c92020-04-27 20:55:33 +0000195 SelectDest dest;
196 Select *pSelect = 0;
197 ExprList *pList = 0;
dan576d5a82020-07-15 18:30:01 +0000198 ExprList *pGrp = 0;
dan7e1d9512020-07-17 15:34:54 +0000199 Expr *pLimit2 = 0;
200 ExprList *pOrderBy2 = 0;
dan1e113842020-04-30 15:49:56 +0000201 sqlite3 *db = pParse->db;
202 Table *pTab = pTabList->a[0].pTab;
203 SrcList *pSrc;
204 Expr *pWhere2;
dan69887c92020-04-27 20:55:33 +0000205 int eDest;
206
dan7e1d9512020-07-17 15:34:54 +0000207#ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT
dan1e113842020-04-30 15:49:56 +0000208 if( pOrderBy && pLimit==0 ) {
209 sqlite3ErrorMsg(pParse, "ORDER BY without LIMIT on UPDATE");
210 return;
211 }
dan7e1d9512020-07-17 15:34:54 +0000212 pOrderBy2 = sqlite3ExprListDup(db, pOrderBy, 0);
213 pLimit2 = sqlite3ExprDup(db, pLimit, 0);
drha7c74002020-07-18 18:44:59 +0000214#else
215 UNUSED_PARAMETER(pOrderBy);
216 UNUSED_PARAMETER(pLimit);
dan7e1d9512020-07-17 15:34:54 +0000217#endif
dan1e113842020-04-30 15:49:56 +0000218
219 pSrc = sqlite3SrcListDup(db, pTabList, 0);
220 pWhere2 = sqlite3ExprDup(db, pWhere, 0);
dan1e113842020-04-30 15:49:56 +0000221
dan69887c92020-04-27 20:55:33 +0000222 assert( pTabList->nSrc>1 );
223 if( pSrc ){
drhcd1499f2021-05-20 00:44:04 +0000224 pSrc->a[0].fg.notCte = 1;
dan69887c92020-04-27 20:55:33 +0000225 pSrc->a[0].iCursor = -1;
226 pSrc->a[0].pTab->nTabRef--;
227 pSrc->a[0].pTab = 0;
228 }
229 if( pPk ){
230 for(i=0; i<pPk->nKeyCol; i++){
dan576d5a82020-07-15 18:30:01 +0000231 Expr *pNew = exprRowColumn(pParse, pPk->aiColumn[i]);
dan7e1d9512020-07-17 15:34:54 +0000232#ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT
danf2972b62020-04-29 20:11:01 +0000233 if( pLimit ){
dan576d5a82020-07-15 18:30:01 +0000234 pGrp = sqlite3ExprListAppend(pParse, pGrp, sqlite3ExprDup(db, pNew, 0));
danf2972b62020-04-29 20:11:01 +0000235 }
dan7e1d9512020-07-17 15:34:54 +0000236#endif
danf2972b62020-04-29 20:11:01 +0000237 pList = sqlite3ExprListAppend(pParse, pList, pNew);
dan69887c92020-04-27 20:55:33 +0000238 }
dan4d906f12020-10-12 14:29:11 +0000239 eDest = IsVirtual(pTab) ? SRT_Table : SRT_Upfrom;
drhf38524d2021-08-02 16:41:57 +0000240 }else if( IsView(pTab) ){
dan576d5a82020-07-15 18:30:01 +0000241 for(i=0; i<pTab->nCol; i++){
242 pList = sqlite3ExprListAppend(pParse, pList, exprRowColumn(pParse, i));
243 }
dan69887c92020-04-27 20:55:33 +0000244 eDest = SRT_Table;
245 }else{
dan9ed322d2020-04-29 17:41:29 +0000246 eDest = IsVirtual(pTab) ? SRT_Table : SRT_Upfrom;
dan576d5a82020-07-15 18:30:01 +0000247 pList = sqlite3ExprListAppend(pParse, 0, sqlite3PExpr(pParse,TK_ROW,0,0));
dan7e1d9512020-07-17 15:34:54 +0000248#ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT
danf2972b62020-04-29 20:11:01 +0000249 if( pLimit ){
dan576d5a82020-07-15 18:30:01 +0000250 pGrp = sqlite3ExprListAppend(pParse, 0, sqlite3PExpr(pParse,TK_ROW,0,0));
danf2972b62020-04-29 20:11:01 +0000251 }
dan7e1d9512020-07-17 15:34:54 +0000252#endif
dan69887c92020-04-27 20:55:33 +0000253 }
drh2be1f2a2021-04-09 22:20:20 +0000254 assert( pChanges!=0 || pParse->db->mallocFailed );
255 if( pChanges ){
dan8b023cf2020-04-30 18:28:40 +0000256 for(i=0; i<pChanges->nExpr; i++){
257 pList = sqlite3ExprListAppend(pParse, pList,
258 sqlite3ExprDup(db, pChanges->a[i].pExpr, 0)
259 );
260 }
dan69887c92020-04-27 20:55:33 +0000261 }
dan243210b2020-07-15 15:32:59 +0000262 pSelect = sqlite3SelectNew(pParse, pList,
dan5daf69e2021-07-05 11:27:13 +0000263 pSrc, pWhere2, pGrp, 0, pOrderBy2, SF_UFSrcCheck|SF_IncludeHidden, pLimit2
danf2972b62020-04-29 20:11:01 +0000264 );
drh0fb78f02021-07-16 01:19:19 +0000265 if( pSelect ) pSelect->selFlags |= SF_OrderByReqd;
dan69887c92020-04-27 20:55:33 +0000266 sqlite3SelectDestInit(&dest, eDest, iEph);
dan9ed322d2020-04-29 17:41:29 +0000267 dest.iSDParm2 = (pPk ? pPk->nKeyCol : -1);
dan69887c92020-04-27 20:55:33 +0000268 sqlite3Select(pParse, pSelect, &dest);
269 sqlite3SelectDelete(db, pSelect);
270}
271
272/*
drhcce7d172000-05-31 15:34:51 +0000273** Process an UPDATE statement.
drhb2fe7d82003-04-20 17:29:23 +0000274**
dan243210b2020-07-15 15:32:59 +0000275** UPDATE OR IGNORE tbl SET a=b, c=d FROM tbl2... WHERE e<5 AND f NOT NULL;
276** \_______/ \_/ \______/ \_____/ \________________/
277** onError | pChanges | pWhere
278** \_______________________/
279** pTabList
drhcce7d172000-05-31 15:34:51 +0000280*/
dan3b61ebb2017-11-10 20:13:14 +0000281void sqlite3Update(
drhcce7d172000-05-31 15:34:51 +0000282 Parse *pParse, /* The parser context */
drh113088e2003-03-20 01:16:58 +0000283 SrcList *pTabList, /* The table in which we should change things */
drhcce7d172000-05-31 15:34:51 +0000284 ExprList *pChanges, /* Things to be changed */
drh9cfcf5d2002-01-29 18:41:24 +0000285 Expr *pWhere, /* The WHERE clause. May be null */
danb3c16b82017-11-09 19:53:06 +0000286 int onError, /* How to handle constraint errors */
dan3b61ebb2017-11-10 20:13:14 +0000287 ExprList *pOrderBy, /* ORDER BY clause. May be null */
drheac9fab2018-04-16 13:00:50 +0000288 Expr *pLimit, /* LIMIT clause. May be null */
289 Upsert *pUpsert /* ON CONFLICT clause, or null */
drhcce7d172000-05-31 15:34:51 +0000290){
drh8a53ce22019-10-17 15:59:03 +0000291 int i, j, k; /* Loop counters */
drhcce7d172000-05-31 15:34:51 +0000292 Table *pTab; /* The table to be updated */
drhc3e356f2013-11-04 18:34:46 +0000293 int addrTop = 0; /* VDBE instruction address of the start of the loop */
drhaec7dc62020-07-18 18:03:26 +0000294 WhereInfo *pWInfo = 0; /* Information about the WHERE clause */
drhcce7d172000-05-31 15:34:51 +0000295 Vdbe *v; /* The virtual database engine */
296 Index *pIdx; /* For looping over indices */
drhea22abe2013-10-25 19:17:17 +0000297 Index *pPk; /* The PRIMARY KEY index for WITHOUT ROWID tables */
drhcce7d172000-05-31 15:34:51 +0000298 int nIdx; /* Number of indices that need updating */
drh9e9374b2019-05-08 19:06:59 +0000299 int nAllIdx; /* Total number of indexes */
drh6a534992013-11-16 20:13:39 +0000300 int iBaseCur; /* Base cursor number */
drh77f64bb2013-10-31 12:13:37 +0000301 int iDataCur; /* Cursor for the canonical data btree */
302 int iIdxCur; /* Cursor for the first index */
drh9bb575f2004-09-06 17:24:11 +0000303 sqlite3 *db; /* The database structure */
drha7c3b932019-05-07 19:13:42 +0000304 int *aRegIdx = 0; /* Registers for to each index and the main table */
drhcce7d172000-05-31 15:34:51 +0000305 int *aXRef = 0; /* aXRef[i] is the index in pChanges->a[] of the
drh967e8b72000-06-21 13:59:10 +0000306 ** an expression for the i-th column of the table.
307 ** aXRef[i]==-1 if the i-th column is not changed. */
drh6a534992013-11-16 20:13:39 +0000308 u8 *aToOpen; /* 1 for tables and indices to be opened */
drhf8ffb272013-11-01 17:08:56 +0000309 u8 chngPk; /* PRIMARY KEY changed in a WITHOUT ROWID table */
310 u8 chngRowid; /* Rowid changed in a normal table */
311 u8 chngKey; /* Either chngPk or chngRowid */
drhf0863fe2005-06-12 21:35:51 +0000312 Expr *pRowidExpr = 0; /* Expression defining the new record number */
dan243210b2020-07-15 15:32:59 +0000313 int iRowidExpr = -1; /* Index of "rowid=" (or IPK) assignment in pChanges */
drh85e20962003-04-25 17:52:11 +0000314 AuthContext sContext; /* The authorization context */
danielk1977b3bce662005-01-29 08:32:43 +0000315 NameContext sNC; /* The name-context to resolve expressions in */
danielk1977da184232006-01-05 11:34:32 +0000316 int iDb; /* Database containing the table being updated */
danf91c1312017-01-10 20:04:38 +0000317 int eOnePass; /* ONEPASS_XXX value from where.c */
dan1da40a32009-09-19 17:00:31 +0000318 int hasFK; /* True if foreign key processing is required */
drhea22abe2013-10-25 19:17:17 +0000319 int labelBreak; /* Jump here to break out of UPDATE loop */
320 int labelContinue; /* Jump here to continue next step of UPDATE loop */
danf91c1312017-01-10 20:04:38 +0000321 int flags; /* Flags for sqlite3WhereBegin() */
drhcce7d172000-05-31 15:34:51 +0000322
drh798da522004-11-04 04:42:28 +0000323#ifndef SQLITE_OMIT_TRIGGER
danbb5f1682009-11-27 12:12:34 +0000324 int isView; /* True when updating a view (INSTEAD OF trigger) */
325 Trigger *pTrigger; /* List of triggers on pTab, if required */
326 int tmask; /* Mask of TRIGGER_BEFORE|TRIGGER_AFTER */
drh798da522004-11-04 04:42:28 +0000327#endif
drhfd7c9912009-12-01 22:09:25 +0000328 int newmask; /* Mask of NEW.* columns accessed by BEFORE triggers */
drhea22abe2013-10-25 19:17:17 +0000329 int iEph = 0; /* Ephemeral table holding all primary key values */
drh00f91cf2013-11-07 21:32:16 +0000330 int nKey = 0; /* Number of elements in regKey for WITHOUT ROWID */
drhfc8d4f92013-11-08 15:19:46 +0000331 int aiCurOnePass[2]; /* The write cursors opened by WHERE_ONEPASS */
drhae051a82017-01-12 16:21:54 +0000332 int addrOpen = 0; /* Address of OP_OpenEphemeral */
333 int iPk = 0; /* First of nPk cells holding PRIMARY KEY value */
334 i16 nPk = 0; /* Number of components of the PRIMARY KEY */
dan2c6fec22017-01-11 19:03:08 +0000335 int bReplace = 0; /* True if REPLACE conflict resolution might happen */
drhbe3da242019-12-29 00:52:41 +0000336 int bFinishSeek = 1; /* The OP_FinishSeek opcode is needed */
dan243210b2020-07-15 15:32:59 +0000337 int nChangeFrom = 0; /* If there is a FROM, pChanges->nExpr, else 0 */
danielk1977c3f9bad2002-05-15 08:30:12 +0000338
drh04adf412008-01-08 18:57:50 +0000339 /* Register Allocations */
340 int regRowCount = 0; /* A count of rows changed */
drh94f26a12015-09-29 12:32:56 +0000341 int regOldRowid = 0; /* The old rowid */
342 int regNewRowid = 0; /* The new rowid */
343 int regNew = 0; /* Content of the NEW.* table in triggers */
drhb8475df2011-12-09 16:21:19 +0000344 int regOld = 0; /* Content of OLD.* table in triggers */
drh4f21c4a2008-12-10 22:15:00 +0000345 int regRowSet = 0; /* Rowset of rows to be updated */
drhea22abe2013-10-25 19:17:17 +0000346 int regKey = 0; /* composite PRIMARY KEY value */
drh04adf412008-01-08 18:57:50 +0000347
drheeb844a2009-08-08 18:01:07 +0000348 memset(&sContext, 0, sizeof(sContext));
drh17435752007-08-16 04:30:38 +0000349 db = pParse->db;
drh0c7d3d32022-01-24 16:47:12 +0000350 assert( db->pParse==pParse );
351 if( pParse->nErr ){
drh6f7adc82006-01-11 21:41:20 +0000352 goto update_cleanup;
353 }
drh0c7d3d32022-01-24 16:47:12 +0000354 assert( db->mallocFailed==0 );
drhdaffd0e2001-04-11 14:28:42 +0000355
drhb2fe7d82003-04-20 17:29:23 +0000356 /* Locate the table which we want to update.
drhcce7d172000-05-31 15:34:51 +0000357 */
danielk19774adee202004-05-08 08:23:19 +0000358 pTab = sqlite3SrcListLookup(pParse, pTabList);
drha69d9162003-04-17 22:57:53 +0000359 if( pTab==0 ) goto update_cleanup;
danielk1977da184232006-01-05 11:34:32 +0000360 iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema);
drhb7f91642004-10-31 02:22:47 +0000361
362 /* Figure out if we have any triggers and if the table being
dan76d462e2009-08-30 11:42:51 +0000363 ** updated is a view.
drhb7f91642004-10-31 02:22:47 +0000364 */
365#ifndef SQLITE_OMIT_TRIGGER
danbb5f1682009-11-27 12:12:34 +0000366 pTrigger = sqlite3TriggersExist(pParse, pTab, TK_UPDATE, pChanges, &tmask);
drhf38524d2021-08-02 16:41:57 +0000367 isView = IsView(pTab);
danbb5f1682009-11-27 12:12:34 +0000368 assert( pTrigger || tmask==0 );
drhb7f91642004-10-31 02:22:47 +0000369#else
danielk19772f886d12009-02-28 10:47:41 +0000370# define pTrigger 0
drhb7f91642004-10-31 02:22:47 +0000371# define isView 0
danbb5f1682009-11-27 12:12:34 +0000372# define tmask 0
drhb7f91642004-10-31 02:22:47 +0000373#endif
374#ifdef SQLITE_OMIT_VIEW
375# undef isView
376# define isView 0
377#endif
dan243210b2020-07-15 15:32:59 +0000378
drh2a7dcbf2022-04-06 15:41:53 +0000379#if TREETRACE_ENABLED
380 if( sqlite3TreeTrace & 0x10000 ){
381 sqlite3TreeViewLine(0, "In sqlite3Update() at %s:%d", __FILE__, __LINE__);
382 sqlite3TreeViewUpdate(pParse->pWith, pTabList, pChanges, pWhere,
383 onError, pOrderBy, pLimit, pUpsert, pTrigger);
384 }
385#endif
386
dan243210b2020-07-15 15:32:59 +0000387 /* If there was a FROM clause, set nChangeFrom to the number of expressions
388 ** in the change-list. Otherwise, set it to 0. There cannot be a FROM
389 ** clause if this function is being called to generate code for part of
390 ** an UPSERT statement. */
dan69887c92020-04-27 20:55:33 +0000391 nChangeFrom = (pTabList->nSrc>1) ? pChanges->nExpr : 0;
392 assert( nChangeFrom==0 || pUpsert==0 );
drhb7f91642004-10-31 02:22:47 +0000393
danb3c16b82017-11-09 19:53:06 +0000394#ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT
danf2972b62020-04-29 20:11:01 +0000395 if( !isView && nChangeFrom==0 ){
danb3c16b82017-11-09 19:53:06 +0000396 pWhere = sqlite3LimitWhere(
drh8c0833f2017-11-14 23:48:23 +0000397 pParse, pTabList, pWhere, pOrderBy, pLimit, "UPDATE"
danb3c16b82017-11-09 19:53:06 +0000398 );
399 pOrderBy = 0;
drh8c0833f2017-11-14 23:48:23 +0000400 pLimit = 0;
danb3c16b82017-11-09 19:53:06 +0000401 }
402#endif
403
danielk1977595a5232009-07-24 17:58:53 +0000404 if( sqlite3ViewGetColumnNames(pParse, pTab) ){
drh5cf590c2003-04-24 01:45:04 +0000405 goto update_cleanup;
drha69d9162003-04-17 22:57:53 +0000406 }
danbb5f1682009-11-27 12:12:34 +0000407 if( sqlite3IsReadOnly(pParse, pTab, tmask) ){
danielk1977b3d24bf2006-06-19 03:05:10 +0000408 goto update_cleanup;
drh5cf590c2003-04-24 01:45:04 +0000409 }
drhcce7d172000-05-31 15:34:51 +0000410
drh53e3fc72002-07-16 17:22:50 +0000411 /* Allocate a cursors for the main database table and for all indices.
412 ** The index cursors might not be used, but if they are used they
413 ** need to occur right after the database cursor. So go ahead and
414 ** allocate enough space, just in case.
415 */
drh2ac4e5c2018-04-20 00:40:19 +0000416 iBaseCur = iDataCur = pParse->nTab++;
drh313619f2013-10-31 20:34:06 +0000417 iIdxCur = iDataCur+1;
drh77f64bb2013-10-31 12:13:37 +0000418 pPk = HasRowid(pTab) ? 0 : sqlite3PrimaryKeyIndex(pTab);
drh2ac4e5c2018-04-20 00:40:19 +0000419 testcase( pPk!=0 && pPk!=pTab->pIndex );
drh77f64bb2013-10-31 12:13:37 +0000420 for(nIdx=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, nIdx++){
drh2ac4e5c2018-04-20 00:40:19 +0000421 if( pPk==pIdx ){
drh313619f2013-10-31 20:34:06 +0000422 iDataCur = pParse->nTab;
drh313619f2013-10-31 20:34:06 +0000423 }
drh53e3fc72002-07-16 17:22:50 +0000424 pParse->nTab++;
425 }
drh7fc3aba2018-04-20 13:18:51 +0000426 if( pUpsert ){
drhfb2213e2018-04-20 15:56:24 +0000427 /* On an UPSERT, reuse the same cursors already opened by INSERT */
drh7fc3aba2018-04-20 13:18:51 +0000428 iDataCur = pUpsert->iDataCur;
429 iIdxCur = pUpsert->iIdxCur;
430 pParse->nTab = iBaseCur;
431 }
drh2ac4e5c2018-04-20 00:40:19 +0000432 pTabList->a[0].iCursor = iDataCur;
drh53e3fc72002-07-16 17:22:50 +0000433
drh6a534992013-11-16 20:13:39 +0000434 /* Allocate space for aXRef[], aRegIdx[], and aToOpen[].
435 ** Initialize aXRef[] and aToOpen[] to their default values.
436 */
drha7c3b932019-05-07 19:13:42 +0000437 aXRef = sqlite3DbMallocRawNN(db, sizeof(int) * (pTab->nCol+nIdx+1) + nIdx+2 );
drh6a534992013-11-16 20:13:39 +0000438 if( aXRef==0 ) goto update_cleanup;
439 aRegIdx = aXRef+pTab->nCol;
drha7c3b932019-05-07 19:13:42 +0000440 aToOpen = (u8*)(aRegIdx+nIdx+1);
drh6a534992013-11-16 20:13:39 +0000441 memset(aToOpen, 1, nIdx+1);
442 aToOpen[nIdx+1] = 0;
443 for(i=0; i<pTab->nCol; i++) aXRef[i] = -1;
444
danielk1977b3bce662005-01-29 08:32:43 +0000445 /* Initialize the name-context */
446 memset(&sNC, 0, sizeof(sNC));
447 sNC.pParse = pParse;
448 sNC.pSrcList = pTabList;
drheac9fab2018-04-16 13:00:50 +0000449 sNC.uNC.pUpsert = pUpsert;
450 sNC.ncFlags = NC_UUpsert;
danielk1977b3bce662005-01-29 08:32:43 +0000451
mistachkin16fd04c2019-10-16 17:46:22 +0000452 /* Begin generating code. */
453 v = sqlite3GetVdbe(pParse);
454 if( v==0 ) goto update_cleanup;
455
drh85e20962003-04-25 17:52:11 +0000456 /* Resolve the column names in all the expressions of the
457 ** of the UPDATE statement. Also find the column index
drhed6c8672003-01-12 18:02:16 +0000458 ** for each column to be updated in the pChanges array. For each
459 ** column to be updated, make sure we have authorization to change
460 ** that column.
drhcce7d172000-05-31 15:34:51 +0000461 */
drhf8ffb272013-11-01 17:08:56 +0000462 chngRowid = chngPk = 0;
drhcce7d172000-05-31 15:34:51 +0000463 for(i=0; i<pChanges->nExpr; i++){
drhcf9d36d2021-08-02 18:03:43 +0000464 u8 hCol = sqlite3StrIHash(pChanges->a[i].zEName);
dan69887c92020-04-27 20:55:33 +0000465 /* If this is an UPDATE with a FROM clause, do not resolve expressions
466 ** here. The call to sqlite3Select() below will do that. */
467 if( nChangeFrom==0 && sqlite3ResolveExprNames(&sNC, pChanges->a[i].pExpr) ){
drhcce7d172000-05-31 15:34:51 +0000468 goto update_cleanup;
469 }
470 for(j=0; j<pTab->nCol; j++){
drhcf9d36d2021-08-02 18:03:43 +0000471 if( pTab->aCol[j].hName==hCol
472 && sqlite3StrICmp(pTab->aCol[j].zCnName, pChanges->a[i].zEName)==0
473 ){
drh9647ff82002-01-14 02:56:24 +0000474 if( j==pTab->iPKey ){
drhf0863fe2005-06-12 21:35:51 +0000475 chngRowid = 1;
476 pRowidExpr = pChanges->a[i].pExpr;
dan74657872020-05-01 18:43:49 +0000477 iRowidExpr = i;
drhf8ffb272013-11-01 17:08:56 +0000478 }else if( pPk && (pTab->aCol[j].colFlags & COLFLAG_PRIMKEY)!=0 ){
479 chngPk = 1;
drh4a324312001-12-21 14:30:42 +0000480 }
drh8a53ce22019-10-17 15:59:03 +0000481#ifndef SQLITE_OMIT_GENERATED_COLUMNS
drh427b96a2019-10-22 13:01:24 +0000482 else if( pTab->aCol[j].colFlags & COLFLAG_GENERATED ){
drh6ab61d72019-10-23 15:47:33 +0000483 testcase( pTab->aCol[j].colFlags & COLFLAG_VIRTUAL );
484 testcase( pTab->aCol[j].colFlags & COLFLAG_STORED );
drh8a53ce22019-10-17 15:59:03 +0000485 sqlite3ErrorMsg(pParse,
486 "cannot UPDATE generated column \"%s\"",
drhcf9d36d2021-08-02 18:03:43 +0000487 pTab->aCol[j].zCnName);
drh8a53ce22019-10-17 15:59:03 +0000488 goto update_cleanup;
489 }
490#endif
drhcce7d172000-05-31 15:34:51 +0000491 aXRef[j] = i;
492 break;
493 }
494 }
495 if( j>=pTab->nCol ){
drh41cee662019-12-12 20:22:34 +0000496 if( pPk==0 && sqlite3IsRowid(pChanges->a[i].zEName) ){
drh27228982013-05-06 13:22:50 +0000497 j = -1;
drhf0863fe2005-06-12 21:35:51 +0000498 chngRowid = 1;
499 pRowidExpr = pChanges->a[i].pExpr;
dan74657872020-05-01 18:43:49 +0000500 iRowidExpr = i;
drha0217ba2003-06-01 01:10:33 +0000501 }else{
drh41cee662019-12-12 20:22:34 +0000502 sqlite3ErrorMsg(pParse, "no such column: %s", pChanges->a[i].zEName);
dan1db95102010-06-28 10:15:19 +0000503 pParse->checkSchema = 1;
drha0217ba2003-06-01 01:10:33 +0000504 goto update_cleanup;
505 }
drhcce7d172000-05-31 15:34:51 +0000506 }
drhed6c8672003-01-12 18:02:16 +0000507#ifndef SQLITE_OMIT_AUTHORIZATION
drhe5f9c642003-01-13 23:27:31 +0000508 {
509 int rc;
danielk19774adee202004-05-08 08:23:19 +0000510 rc = sqlite3AuthCheck(pParse, SQLITE_UPDATE, pTab->zName,
drhcf9d36d2021-08-02 18:03:43 +0000511 j<0 ? "ROWID" : pTab->aCol[j].zCnName,
drh69c33822016-08-18 14:33:11 +0000512 db->aDb[iDb].zDbSName);
drhe5f9c642003-01-13 23:27:31 +0000513 if( rc==SQLITE_DENY ){
514 goto update_cleanup;
515 }else if( rc==SQLITE_IGNORE ){
516 aXRef[j] = -1;
517 }
drhed6c8672003-01-12 18:02:16 +0000518 }
519#endif
drhcce7d172000-05-31 15:34:51 +0000520 }
drhf8ffb272013-11-01 17:08:56 +0000521 assert( (chngRowid & chngPk)==0 );
522 assert( chngRowid==0 || chngRowid==1 );
523 assert( chngPk==0 || chngPk==1 );
524 chngKey = chngRowid + chngPk;
drhcce7d172000-05-31 15:34:51 +0000525
drh7e7fd732019-10-22 13:59:23 +0000526#ifndef SQLITE_OMIT_GENERATED_COLUMNS
527 /* Mark generated columns as changing if their generator expressions
528 ** reference any changing column. The actual aXRef[] value for
529 ** generated expressions is not used, other than to check to see that it
530 ** is non-negative, so the value of aXRef[] for generated columns can be
531 ** set to any non-negative number. We use 99999 so that the value is
532 ** obvious when looking at aXRef[] in a symbolic debugger.
533 */
534 if( pTab->tabFlags & TF_HasGenerated ){
535 int bProgress;
536 testcase( pTab->tabFlags & TF_HasVirtual );
537 testcase( pTab->tabFlags & TF_HasStored );
538 do{
539 bProgress = 0;
540 for(i=0; i<pTab->nCol; i++){
541 if( aXRef[i]>=0 ) continue;
542 if( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)==0 ) continue;
drh79cf2b72021-07-31 20:30:41 +0000543 if( sqlite3ExprReferencesUpdatedColumn(
544 sqlite3ColumnExpr(pTab, &pTab->aCol[i]),
545 aXRef, chngRowid)
546 ){
drh7e7fd732019-10-22 13:59:23 +0000547 aXRef[i] = 99999;
548 bProgress = 1;
549 }
550 }
551 }while( bProgress );
552 }
553#endif
554
dan1acb5392015-11-26 19:33:41 +0000555 /* The SET expressions are not actually used inside the WHERE loop.
556 ** So reset the colUsed mask. Unless this is a virtual table. In that
557 ** case, set all bits of the colUsed mask (to ensure that the virtual
558 ** table implementation makes all columns available).
drh5c82f4d2013-11-19 02:34:11 +0000559 */
drh8426e362016-03-08 01:32:30 +0000560 pTabList->a[0].colUsed = IsVirtual(pTab) ? ALLBITS : 0;
drh5c82f4d2013-11-19 02:34:11 +0000561
drhf8ffb272013-11-01 17:08:56 +0000562 hasFK = sqlite3FkRequired(pParse, pTab, aXRef, chngKey);
dane7a94d82009-10-01 16:09:04 +0000563
drh6a534992013-11-16 20:13:39 +0000564 /* There is one entry in the aRegIdx[] array for each index on the table
565 ** being updated. Fill in aRegIdx[] with a register number that will hold
drh1f9ca2c2015-08-25 16:57:52 +0000566 ** the key for accessing each index.
drhcce7d172000-05-31 15:34:51 +0000567 */
drh247c1b42019-01-22 13:45:48 +0000568 if( onError==OE_Replace ) bReplace = 1;
drh9e9374b2019-05-08 19:06:59 +0000569 for(nAllIdx=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, nAllIdx++){
drhaa9b8962008-01-08 02:57:55 +0000570 int reg;
drh40b4e7a2018-09-17 20:47:38 +0000571 if( chngKey || hasFK>1 || pIdx==pPk
572 || indexWhereClauseMightChange(pIdx,aXRef,chngRowid)
573 ){
drhaa9b8962008-01-08 02:57:55 +0000574 reg = ++pParse->nMem;
drh2c4dfc32016-11-10 14:24:04 +0000575 pParse->nMem += pIdx->nColumn;
drhaa9b8962008-01-08 02:57:55 +0000576 }else{
577 reg = 0;
drhf8ffb272013-11-01 17:08:56 +0000578 for(i=0; i<pIdx->nKeyCol; i++){
drh2b7cdf62018-09-16 15:01:25 +0000579 if( indexColumnIsBeingUpdated(pIdx, i, aXRef, chngRowid) ){
drhaa9b8962008-01-08 02:57:55 +0000580 reg = ++pParse->nMem;
drh2c4dfc32016-11-10 14:24:04 +0000581 pParse->nMem += pIdx->nColumn;
drh247c1b42019-01-22 13:45:48 +0000582 if( onError==OE_Default && pIdx->onError==OE_Replace ){
dan2c6fec22017-01-11 19:03:08 +0000583 bReplace = 1;
584 }
drhaa9b8962008-01-08 02:57:55 +0000585 break;
586 }
drh4a324312001-12-21 14:30:42 +0000587 }
drhcce7d172000-05-31 15:34:51 +0000588 }
drh9e9374b2019-05-08 19:06:59 +0000589 if( reg==0 ) aToOpen[nAllIdx+1] = 0;
590 aRegIdx[nAllIdx] = reg;
drhcce7d172000-05-31 15:34:51 +0000591 }
drh9e9374b2019-05-08 19:06:59 +0000592 aRegIdx[nAllIdx] = ++pParse->nMem; /* Register storing the table record */
dan2c6fec22017-01-11 19:03:08 +0000593 if( bReplace ){
594 /* If REPLACE conflict resolution might be invoked, open cursors on all
595 ** indexes in case they are needed to delete records. */
596 memset(aToOpen, 1, nIdx+1);
597 }
drhcce7d172000-05-31 15:34:51 +0000598
drh4794f732004-11-05 17:17:50 +0000599 if( pParse->nested==0 ) sqlite3VdbeCountChanges(v);
drhac9151d2018-04-20 19:32:35 +0000600 sqlite3BeginWriteOperation(pParse, pTrigger || hasFK, iDb);
drhcce7d172000-05-31 15:34:51 +0000601
dan165921a2009-08-28 18:53:45 +0000602 /* Allocate required registers. */
dan0f400372015-09-28 20:03:49 +0000603 if( !IsVirtual(pTab) ){
drh9e9374b2019-05-08 19:06:59 +0000604 /* For now, regRowSet and aRegIdx[nAllIdx] share the same register.
605 ** If regRowSet turns out to be needed, then aRegIdx[nAllIdx] will be
606 ** reallocated. aRegIdx[nAllIdx] is the register in which the main
607 ** table record is written. regRowSet holds the RowSet for the
608 ** two-pass update algorithm. */
609 assert( aRegIdx[nAllIdx]==pParse->nMem );
610 regRowSet = aRegIdx[nAllIdx];
dan0f400372015-09-28 20:03:49 +0000611 regOldRowid = regNewRowid = ++pParse->nMem;
612 if( chngPk || pTrigger || hasFK ){
613 regOld = pParse->nMem + 1;
614 pParse->nMem += pTab->nCol;
615 }
616 if( chngKey || pTrigger || hasFK ){
617 regNewRowid = ++pParse->nMem;
618 }
619 regNew = pParse->nMem + 1;
dan165921a2009-08-28 18:53:45 +0000620 pParse->nMem += pTab->nCol;
621 }
dan165921a2009-08-28 18:53:45 +0000622
623 /* Start the view context. */
danielk19774273dea2006-06-17 06:31:18 +0000624 if( isView ){
625 sqlite3AuthContextPush(pParse, &sContext, pTab->zName);
626 }
627
drh9d2985c2005-09-08 01:58:42 +0000628 /* If we are trying to update a view, realize that view into
peter.d.reid60ec9142014-09-06 16:39:46 +0000629 ** an ephemeral table.
drh5cf590c2003-04-24 01:45:04 +0000630 */
shanefa4e62f2008-09-01 21:59:42 +0000631#if !defined(SQLITE_OMIT_VIEW) && !defined(SQLITE_OMIT_TRIGGER)
dan69887c92020-04-27 20:55:33 +0000632 if( nChangeFrom==0 && isView ){
danb3c16b82017-11-09 19:53:06 +0000633 sqlite3MaterializeView(pParse, pTab,
drh8c0833f2017-11-14 23:48:23 +0000634 pWhere, pOrderBy, pLimit, iDataCur
danb3c16b82017-11-09 19:53:06 +0000635 );
636 pOrderBy = 0;
drh8c0833f2017-11-14 23:48:23 +0000637 pLimit = 0;
drh0f35a6b2008-02-12 16:52:14 +0000638 }
shanefa4e62f2008-09-01 21:59:42 +0000639#endif
drh1013c932008-01-06 00:25:21 +0000640
drh0f35a6b2008-02-12 16:52:14 +0000641 /* Resolve the column names in all the expressions in the
642 ** WHERE clause.
643 */
dan69887c92020-04-27 20:55:33 +0000644 if( nChangeFrom==0 && sqlite3ResolveExprNames(&sNC, pWhere) ){
drh0f35a6b2008-02-12 16:52:14 +0000645 goto update_cleanup;
drh5cf590c2003-04-24 01:45:04 +0000646 }
647
dan0f400372015-09-28 20:03:49 +0000648#ifndef SQLITE_OMIT_VIRTUALTABLE
649 /* Virtual tables must be handled separately */
650 if( IsVirtual(pTab) ){
651 updateVirtualTable(pParse, pTabList, pTab, pChanges, pRowidExpr, aXRef,
652 pWhere, onError);
653 goto update_cleanup;
654 }
655#endif
656
drhfb2213e2018-04-20 15:56:24 +0000657 /* Jump to labelBreak to abandon further processing of this UPDATE */
drhec4ccdb2018-12-29 02:26:59 +0000658 labelContinue = labelBreak = sqlite3VdbeMakeLabel(pParse);
danf91c1312017-01-10 20:04:38 +0000659
drh5deb1812018-04-20 16:27:57 +0000660 /* Not an UPSERT. Normal processing. Begin by
661 ** initialize the count of updated rows */
662 if( (db->flags&SQLITE_CountRows)!=0
663 && !pParse->pTriggerTab
664 && !pParse->nested
drhd086aa02021-01-29 21:31:59 +0000665 && !pParse->bReturning
drh5deb1812018-04-20 16:27:57 +0000666 && pUpsert==0
667 ){
668 regRowCount = ++pParse->nMem;
669 sqlite3VdbeAddOp2(v, OP_Integer, 0, regRowCount);
670 }
671
dan69887c92020-04-27 20:55:33 +0000672 if( nChangeFrom==0 && HasRowid(pTab) ){
drh5deb1812018-04-20 16:27:57 +0000673 sqlite3VdbeAddOp3(v, OP_Null, 0, regRowSet, regOldRowid);
drhea248722020-11-14 20:03:34 +0000674 iEph = pParse->nTab++;
drh32881be2020-11-17 21:26:13 +0000675 addrOpen = sqlite3VdbeAddOp3(v, OP_OpenEphemeral, iEph, 0, regRowSet);
drh5deb1812018-04-20 16:27:57 +0000676 }else{
dan69887c92020-04-27 20:55:33 +0000677 assert( pPk!=0 || HasRowid(pTab) );
678 nPk = pPk ? pPk->nKeyCol : 0;
drh5deb1812018-04-20 16:27:57 +0000679 iPk = pParse->nMem+1;
680 pParse->nMem += nPk;
dan69887c92020-04-27 20:55:33 +0000681 pParse->nMem += nChangeFrom;
drh5deb1812018-04-20 16:27:57 +0000682 regKey = ++pParse->nMem;
683 if( pUpsert==0 ){
dan69887c92020-04-27 20:55:33 +0000684 int nEphCol = nPk + nChangeFrom + (isView ? pTab->nCol : 0);
drh5deb1812018-04-20 16:27:57 +0000685 iEph = pParse->nTab++;
dan69887c92020-04-27 20:55:33 +0000686 if( pPk ) sqlite3VdbeAddOp3(v, OP_Null, 0, iPk, iPk+nPk-1);
687 addrOpen = sqlite3VdbeAddOp2(v, OP_OpenEphemeral, iEph, nEphCol);
danbe952c12020-07-13 20:10:29 +0000688 if( pPk ){
689 KeyInfo *pKeyInfo = sqlite3KeyInfoOfIndex(pParse, pPk);
690 if( pKeyInfo ){
691 pKeyInfo->nAllField = nEphCol;
692 sqlite3VdbeAppendP4(v, pKeyInfo, P4_KEYINFO);
693 }
694 }
dan69887c92020-04-27 20:55:33 +0000695 if( nChangeFrom ){
dan576d5a82020-07-15 18:30:01 +0000696 updateFromSelect(
danf2972b62020-04-29 20:11:01 +0000697 pParse, iEph, pPk, pChanges, pTabList, pWhere, pOrderBy, pLimit
698 );
dan69887c92020-04-27 20:55:33 +0000699#ifndef SQLITE_OMIT_SUBQUERY
700 if( isView ) iDataCur = iEph;
701#endif
702 }
drh5deb1812018-04-20 16:27:57 +0000703 }
704 }
705
dan69887c92020-04-27 20:55:33 +0000706 if( nChangeFrom ){
707 sqlite3MultiWrite(pParse);
708 eOnePass = ONEPASS_OFF;
dan74657872020-05-01 18:43:49 +0000709 nKey = nPk;
710 regKey = iPk;
drhfb2213e2018-04-20 15:56:24 +0000711 }else{
dan69887c92020-04-27 20:55:33 +0000712 if( pUpsert ){
713 /* If this is an UPSERT, then all cursors have already been opened by
714 ** the outer INSERT and the data cursor should be pointing at the row
715 ** that is to be updated. So bypass the code that searches for the
716 ** row(s) to be updated.
717 */
718 pWInfo = 0;
719 eOnePass = ONEPASS_SINGLE;
720 sqlite3ExprIfFalse(pParse, pWhere, labelBreak, SQLITE_JUMPIFNULL);
721 bFinishSeek = 0;
722 }else{
723 /* Begin the database scan.
724 **
725 ** Do not consider a single-pass strategy for a multi-row update if
726 ** there are any triggers or foreign keys to process, or rows may
727 ** be deleted as a result of REPLACE conflict handling. Any of these
728 ** things might disturb a cursor being used to scan through the table
729 ** or index, causing a single-pass approach to malfunction. */
drh68c0c712020-08-14 20:04:26 +0000730 flags = WHERE_ONEPASS_DESIRED;
dan69887c92020-04-27 20:55:33 +0000731 if( !pParse->nested && !pTrigger && !hasFK && !chngKey && !bReplace ){
732 flags |= WHERE_ONEPASS_MULTIROW;
733 }
drh895bab32022-01-27 16:14:50 +0000734 pWInfo = sqlite3WhereBegin(pParse, pTabList, pWhere,0,0,0,flags,iIdxCur);
dan69887c92020-04-27 20:55:33 +0000735 if( pWInfo==0 ) goto update_cleanup;
736
737 /* A one-pass strategy that might update more than one row may not
738 ** be used if any column of the index used for the scan is being
739 ** updated. Otherwise, if there is an index on "b", statements like
740 ** the following could create an infinite loop:
741 **
742 ** UPDATE t1 SET b=b+1 WHERE b>?
743 **
744 ** Fall back to ONEPASS_OFF if where.c has selected a ONEPASS_MULTI
745 ** strategy that uses an index for which one or more columns are being
746 ** updated. */
747 eOnePass = sqlite3WhereOkOnePass(pWInfo, aiCurOnePass);
748 bFinishSeek = sqlite3WhereUsesDeferredSeek(pWInfo);
749 if( eOnePass!=ONEPASS_SINGLE ){
750 sqlite3MultiWrite(pParse);
751 if( eOnePass==ONEPASS_MULTI ){
752 int iCur = aiCurOnePass[1];
753 if( iCur>=0 && iCur!=iDataCur && aToOpen[iCur-iBaseCur] ){
754 eOnePass = ONEPASS_OFF;
755 }
756 assert( iCur!=iDataCur || !HasRowid(pTab) );
drhac9151d2018-04-20 19:32:35 +0000757 }
dan69887c92020-04-27 20:55:33 +0000758 }
759 }
760
761 if( HasRowid(pTab) ){
762 /* Read the rowid of the current row of the WHERE scan. In ONEPASS_OFF
763 ** mode, write the rowid into the FIFO. In either of the one-pass modes,
764 ** leave it in register regOldRowid. */
765 sqlite3VdbeAddOp2(v, OP_Rowid, iDataCur, regOldRowid);
766 if( eOnePass==ONEPASS_OFF ){
dan69887c92020-04-27 20:55:33 +0000767 aRegIdx[nAllIdx] = ++pParse->nMem;
drhea248722020-11-14 20:03:34 +0000768 sqlite3VdbeAddOp3(v, OP_Insert, iEph, regRowSet, regOldRowid);
769 }else{
drh32881be2020-11-17 21:26:13 +0000770 if( ALWAYS(addrOpen) ) sqlite3VdbeChangeToNoop(v, addrOpen);
dan69887c92020-04-27 20:55:33 +0000771 }
772 }else{
773 /* Read the PK of the current row into an array of registers. In
774 ** ONEPASS_OFF mode, serialize the array into a record and store it in
775 ** the ephemeral table. Or, in ONEPASS_SINGLE or MULTI mode, change
776 ** the OP_OpenEphemeral instruction to a Noop (the ephemeral table
777 ** is not required) and leave the PK fields in the array of registers. */
778 for(i=0; i<nPk; i++){
779 assert( pPk->aiColumn[i]>=0 );
780 sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur,
781 pPk->aiColumn[i], iPk+i);
782 }
783 if( eOnePass ){
784 if( addrOpen ) sqlite3VdbeChangeToNoop(v, addrOpen);
785 nKey = nPk;
786 regKey = iPk;
787 }else{
788 sqlite3VdbeAddOp4(v, OP_MakeRecord, iPk, nPk, regKey,
789 sqlite3IndexAffinityStr(db, pPk), nPk);
790 sqlite3VdbeAddOp4Int(v, OP_IdxInsert, iEph, regKey, iPk, nPk);
drhfb2213e2018-04-20 15:56:24 +0000791 }
drhfb2213e2018-04-20 15:56:24 +0000792 }
793 }
794
drhfb2213e2018-04-20 15:56:24 +0000795 if( pUpsert==0 ){
dan69887c92020-04-27 20:55:33 +0000796 if( nChangeFrom==0 && eOnePass!=ONEPASS_MULTI ){
drhfb2213e2018-04-20 15:56:24 +0000797 sqlite3WhereEnd(pWInfo);
798 }
799
800 if( !isView ){
801 int addrOnce = 0;
802
803 /* Open every index that needs updating. */
804 if( eOnePass!=ONEPASS_OFF ){
805 if( aiCurOnePass[0]>=0 ) aToOpen[aiCurOnePass[0]-iBaseCur] = 0;
806 if( aiCurOnePass[1]>=0 ) aToOpen[aiCurOnePass[1]-iBaseCur] = 0;
807 }
808
809 if( eOnePass==ONEPASS_MULTI && (nIdx-(aiCurOnePass[1]>=0))>0 ){
810 addrOnce = sqlite3VdbeAddOp0(v, OP_Once); VdbeCoverage(v);
811 }
812 sqlite3OpenTableAndIndices(pParse, pTab, OP_OpenWrite, 0, iBaseCur,
813 aToOpen, 0, 0);
drhdc4f6fc2020-02-07 19:44:13 +0000814 if( addrOnce ){
815 sqlite3VdbeJumpHereOrPopInst(v, addrOnce);
816 }
drhfb2213e2018-04-20 15:56:24 +0000817 }
818
819 /* Top of the update loop */
danf91c1312017-01-10 20:04:38 +0000820 if( eOnePass!=ONEPASS_OFF ){
drh6e5020e2021-04-07 15:45:01 +0000821 if( aiCurOnePass[0]!=iDataCur
822 && aiCurOnePass[1]!=iDataCur
823#ifdef SQLITE_ALLOW_ROWID_IN_VIEW
824 && !isView
825#endif
826 ){
drhfb2213e2018-04-20 15:56:24 +0000827 assert( pPk );
828 sqlite3VdbeAddOp4Int(v, OP_NotFound, iDataCur, labelBreak, regKey,nKey);
drh6ccbd272018-07-10 17:10:44 +0000829 VdbeCoverage(v);
drhfb2213e2018-04-20 15:56:24 +0000830 }
drh4b8bd842018-04-20 20:09:22 +0000831 if( eOnePass!=ONEPASS_SINGLE ){
drhec4ccdb2018-12-29 02:26:59 +0000832 labelContinue = sqlite3VdbeMakeLabel(pParse);
drhfb2213e2018-04-20 15:56:24 +0000833 }
834 sqlite3VdbeAddOp2(v, OP_IsNull, pPk ? regKey : regOldRowid, labelBreak);
835 VdbeCoverageIf(v, pPk==0);
836 VdbeCoverageIf(v, pPk!=0);
dan69887c92020-04-27 20:55:33 +0000837 }else if( pPk || nChangeFrom ){
drhec4ccdb2018-12-29 02:26:59 +0000838 labelContinue = sqlite3VdbeMakeLabel(pParse);
drhfb2213e2018-04-20 15:56:24 +0000839 sqlite3VdbeAddOp2(v, OP_Rewind, iEph, labelBreak); VdbeCoverage(v);
dan69887c92020-04-27 20:55:33 +0000840 addrTop = sqlite3VdbeCurrentAddr(v);
841 if( nChangeFrom ){
842 if( !isView ){
843 if( pPk ){
844 for(i=0; i<nPk; i++){
845 sqlite3VdbeAddOp3(v, OP_Column, iEph, i, iPk+i);
846 }
847 sqlite3VdbeAddOp4Int(
848 v, OP_NotFound, iDataCur, labelContinue, iPk, nPk
drh6265c472020-07-16 00:45:25 +0000849 ); VdbeCoverage(v);
dan69887c92020-04-27 20:55:33 +0000850 }else{
851 sqlite3VdbeAddOp2(v, OP_Rowid, iEph, regOldRowid);
852 sqlite3VdbeAddOp3(
853 v, OP_NotExists, iDataCur, labelContinue, regOldRowid
drh6265c472020-07-16 00:45:25 +0000854 ); VdbeCoverage(v);
dan69887c92020-04-27 20:55:33 +0000855 }
856 }
dan69887c92020-04-27 20:55:33 +0000857 }else{
858 sqlite3VdbeAddOp2(v, OP_RowData, iEph, regKey);
859 sqlite3VdbeAddOp4Int(v, OP_NotFound, iDataCur, labelContinue, regKey,0);
860 VdbeCoverage(v);
861 }
drhfb2213e2018-04-20 15:56:24 +0000862 }else{
drhea248722020-11-14 20:03:34 +0000863 sqlite3VdbeAddOp2(v, OP_Rewind, iEph, labelBreak); VdbeCoverage(v);
864 labelContinue = sqlite3VdbeMakeLabel(pParse);
865 addrTop = sqlite3VdbeAddOp2(v, OP_Rowid, iEph, regOldRowid);
drhfb2213e2018-04-20 15:56:24 +0000866 VdbeCoverage(v);
867 sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue, regOldRowid);
868 VdbeCoverage(v);
danf91c1312017-01-10 20:04:38 +0000869 }
drh08c88eb2008-04-10 13:33:18 +0000870 }
danielk1977e448dc42008-01-02 11:50:51 +0000871
drh70076402018-04-12 19:51:22 +0000872 /* If the rowid value will change, set register regNewRowid to
873 ** contain the new value. If the rowid is not being modified,
dan1da40a32009-09-19 17:00:31 +0000874 ** then regNewRowid is the same register as regOldRowid, which is
875 ** already populated. */
drhf8ffb272013-11-01 17:08:56 +0000876 assert( chngKey || pTrigger || hasFK || regOldRowid==regNewRowid );
dan1da40a32009-09-19 17:00:31 +0000877 if( chngRowid ){
dan74657872020-05-01 18:43:49 +0000878 assert( iRowidExpr>=0 );
879 if( nChangeFrom==0 ){
880 sqlite3ExprCode(pParse, pRowidExpr, regNewRowid);
881 }else{
882 sqlite3VdbeAddOp3(v, OP_Column, iEph, iRowidExpr, regNewRowid);
883 }
drh688852a2014-02-17 22:40:43 +0000884 sqlite3VdbeAddOp1(v, OP_MustBeInt, regNewRowid); VdbeCoverage(v);
dan1da40a32009-09-19 17:00:31 +0000885 }
886
drhf8ffb272013-11-01 17:08:56 +0000887 /* Compute the old pre-UPDATE content of the row being changed, if that
888 ** information is needed */
889 if( chngPk || hasFK || pTrigger ){
dane7a94d82009-10-01 16:09:04 +0000890 u32 oldmask = (hasFK ? sqlite3FkOldmask(pParse, pTab) : 0);
danbb5f1682009-11-27 12:12:34 +0000891 oldmask |= sqlite3TriggerColmask(pParse,
892 pTrigger, pChanges, 0, TRIGGER_BEFORE|TRIGGER_AFTER, pTab, onError
893 );
drhff37c402019-10-31 20:54:20 +0000894 for(i=0; i<pTab->nCol; i++){
drh8a53ce22019-10-17 15:59:03 +0000895 u32 colFlags = pTab->aCol[i].colFlags;
drhff37c402019-10-31 20:54:20 +0000896 k = sqlite3TableColumnToStorage(pTab, i) + regOld;
drhf8ffb272013-11-01 17:08:56 +0000897 if( oldmask==0xffffffff
drh693e6712014-01-24 22:58:00 +0000898 || (i<32 && (oldmask & MASKBIT32(i))!=0)
drh8a53ce22019-10-17 15:59:03 +0000899 || (colFlags & COLFLAG_PRIMKEY)!=0
drhf8ffb272013-11-01 17:08:56 +0000900 ){
drh693e6712014-01-24 22:58:00 +0000901 testcase( oldmask!=0xffffffff && i==31 );
drh6df9c4b2019-10-18 12:52:08 +0000902 sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k);
danielk1977c3f9bad2002-05-15 08:30:12 +0000903 }else{
drh8a53ce22019-10-17 15:59:03 +0000904 sqlite3VdbeAddOp2(v, OP_Null, 0, k);
danielk1977c3f9bad2002-05-15 08:30:12 +0000905 }
906 }
drhf8ffb272013-11-01 17:08:56 +0000907 if( chngRowid==0 && pPk==0 ){
dan1da40a32009-09-19 17:00:31 +0000908 sqlite3VdbeAddOp2(v, OP_Copy, regOldRowid, regNewRowid);
909 }
dan165921a2009-08-28 18:53:45 +0000910 }
drh5cf590c2003-04-24 01:45:04 +0000911
dan165921a2009-08-28 18:53:45 +0000912 /* Populate the array of registers beginning at regNew with the new
peter.d.reid60ec9142014-09-06 16:39:46 +0000913 ** row data. This array is used to check constants, create the new
dan165921a2009-08-28 18:53:45 +0000914 ** table and index records, and as the values for any new.* references
danbb5f1682009-11-27 12:12:34 +0000915 ** made by triggers.
916 **
917 ** If there are one or more BEFORE triggers, then do not populate the
918 ** registers associated with columns that are (a) not modified by
919 ** this UPDATE statement and (b) not accessed by new.* references. The
920 ** values for registers not modified by the UPDATE must be reloaded from
921 ** the database after the BEFORE triggers are fired anyway (as the trigger
922 ** may have modified them). So not loading those that are not going to
923 ** be used eliminates some redundant opcodes.
924 */
925 newmask = sqlite3TriggerColmask(
926 pParse, pTrigger, pChanges, 1, TRIGGER_BEFORE, pTab, onError
927 );
drh8a53ce22019-10-17 15:59:03 +0000928 for(i=0, k=regNew; i<pTab->nCol; i++, k++){
dan165921a2009-08-28 18:53:45 +0000929 if( i==pTab->iPKey ){
drh8a53ce22019-10-17 15:59:03 +0000930 sqlite3VdbeAddOp2(v, OP_Null, 0, k);
drhc1431142019-10-17 17:54:05 +0000931 }else if( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)!=0 ){
932 if( pTab->aCol[i].colFlags & COLFLAG_VIRTUAL ) k--;
dan165921a2009-08-28 18:53:45 +0000933 }else{
drh5cf590c2003-04-24 01:45:04 +0000934 j = aXRef[i];
danbb5f1682009-11-27 12:12:34 +0000935 if( j>=0 ){
dan69887c92020-04-27 20:55:33 +0000936 if( nChangeFrom ){
dan69887c92020-04-27 20:55:33 +0000937 int nOff = (isView ? pTab->nCol : nPk);
drha7c74002020-07-18 18:44:59 +0000938 assert( eOnePass==ONEPASS_OFF );
dan69887c92020-04-27 20:55:33 +0000939 sqlite3VdbeAddOp3(v, OP_Column, iEph, nOff+j, k);
940 }else{
941 sqlite3ExprCode(pParse, pChanges->a[j].pExpr, k);
942 }
drh693e6712014-01-24 22:58:00 +0000943 }else if( 0==(tmask&TRIGGER_BEFORE) || i>31 || (newmask & MASKBIT32(i)) ){
danbb5f1682009-11-27 12:12:34 +0000944 /* This branch loads the value of a column that will not be changed
945 ** into a register. This is done if there are no BEFORE triggers, or
946 ** if there are one or more BEFORE triggers that use this value via
947 ** a new.* reference in a trigger program.
948 */
drhfd7c9912009-12-01 22:09:25 +0000949 testcase( i==31 );
950 testcase( i==32 );
drh6df9c4b2019-10-18 12:52:08 +0000951 sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k);
drhbe3da242019-12-29 00:52:41 +0000952 bFinishSeek = 0;
drhedfac342013-12-19 02:56:01 +0000953 }else{
drh8a53ce22019-10-17 15:59:03 +0000954 sqlite3VdbeAddOp2(v, OP_Null, 0, k);
drh5cf590c2003-04-24 01:45:04 +0000955 }
956 }
dan165921a2009-08-28 18:53:45 +0000957 }
drhc1431142019-10-17 17:54:05 +0000958#ifndef SQLITE_OMIT_GENERATED_COLUMNS
drh427b96a2019-10-22 13:01:24 +0000959 if( pTab->tabFlags & TF_HasGenerated ){
960 testcase( pTab->tabFlags & TF_HasVirtual );
961 testcase( pTab->tabFlags & TF_HasStored );
drhdd6cc9b2019-10-19 18:47:27 +0000962 sqlite3ComputeGeneratedColumns(pParse, regNew, pTab);
drhc1431142019-10-17 17:54:05 +0000963 }
964#endif
drh5cf590c2003-04-24 01:45:04 +0000965
dan165921a2009-08-28 18:53:45 +0000966 /* Fire any BEFORE UPDATE triggers. This happens before constraints are
danbb5f1682009-11-27 12:12:34 +0000967 ** verified. One could argue that this is wrong.
968 */
969 if( tmask&TRIGGER_BEFORE ){
drh57bf4a82014-02-17 14:59:22 +0000970 sqlite3TableAffinity(v, pTab, regNew);
dan2832ad42009-08-31 15:27:27 +0000971 sqlite3CodeRowTrigger(pParse, pTrigger, TK_UPDATE, pChanges,
drhc3e356f2013-11-04 18:34:46 +0000972 TRIGGER_BEFORE, pTab, regOldRowid, onError, labelContinue);
danae0931e2009-09-09 15:06:10 +0000973
dan69887c92020-04-27 20:55:33 +0000974 if( !isView ){
975 /* The row-trigger may have deleted the row being updated. In this
976 ** case, jump to the next row. No updates or AFTER triggers are
977 ** required. This behavior - what happens when the row being updated
978 ** is deleted or renamed by a BEFORE trigger - is left undefined in the
979 ** documentation.
980 */
981 if( pPk ){
dan74657872020-05-01 18:43:49 +0000982 sqlite3VdbeAddOp4Int(v, OP_NotFound,iDataCur,labelContinue,regKey,nKey);
dan69887c92020-04-27 20:55:33 +0000983 VdbeCoverage(v);
984 }else{
985 sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue,regOldRowid);
986 VdbeCoverage(v);
danbb5f1682009-11-27 12:12:34 +0000987 }
dan69887c92020-04-27 20:55:33 +0000988
989 /* After-BEFORE-trigger-reload-loop:
990 ** If it did not delete it, the BEFORE trigger may still have modified
991 ** some of the columns of the row being updated. Load the values for
992 ** all columns not modified by the update statement into their registers
993 ** in case this has happened. Only unmodified columns are reloaded.
994 ** The values computed for modified columns use the values before the
995 ** BEFORE trigger runs. See test case trigger1-18.0 (added 2018-04-26)
996 ** for an example.
997 */
998 for(i=0, k=regNew; i<pTab->nCol; i++, k++){
999 if( pTab->aCol[i].colFlags & COLFLAG_GENERATED ){
1000 if( pTab->aCol[i].colFlags & COLFLAG_VIRTUAL ) k--;
1001 }else if( aXRef[i]<0 && i!=pTab->iPKey ){
1002 sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k);
1003 }
1004 }
drhc1431142019-10-17 17:54:05 +00001005#ifndef SQLITE_OMIT_GENERATED_COLUMNS
dan69887c92020-04-27 20:55:33 +00001006 if( pTab->tabFlags & TF_HasGenerated ){
1007 testcase( pTab->tabFlags & TF_HasVirtual );
1008 testcase( pTab->tabFlags & TF_HasStored );
1009 sqlite3ComputeGeneratedColumns(pParse, regNew, pTab);
1010 }
drhc1431142019-10-17 17:54:05 +00001011#endif
dan69887c92020-04-27 20:55:33 +00001012 }
dan2832ad42009-08-31 15:27:27 +00001013 }
dan165921a2009-08-28 18:53:45 +00001014
1015 if( !isView ){
dan165921a2009-08-28 18:53:45 +00001016 /* Do constraint checks. */
drhf8ffb272013-11-01 17:08:56 +00001017 assert( regOldRowid>0 );
1018 sqlite3GenerateConstraintChecks(pParse, pTab, aRegIdx, iDataCur, iIdxCur,
drhbdb00222016-02-10 16:03:20 +00001019 regNewRowid, regOldRowid, chngKey, onError, labelContinue, &bReplace,
drh788d55a2018-04-13 01:15:09 +00001020 aXRef, 0);
drh5cf590c2003-04-24 01:45:04 +00001021
danf66bfcb2019-08-31 17:14:35 +00001022 /* If REPLACE conflict handling may have been used, or if the PK of the
1023 ** row is changing, then the GenerateConstraintChecks() above may have
1024 ** moved cursor iDataCur. Reseek it. */
1025 if( bReplace || chngKey ){
1026 if( pPk ){
1027 sqlite3VdbeAddOp4Int(v, OP_NotFound,iDataCur,labelContinue,regKey,nKey);
1028 }else{
1029 sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue,regOldRowid);
1030 }
drh81d25cf2019-08-31 20:26:06 +00001031 VdbeCoverageNeverTaken(v);
danf66bfcb2019-08-31 17:14:35 +00001032 }
1033
dan1da40a32009-09-19 17:00:31 +00001034 /* Do FK constraint checks. */
dane7a94d82009-10-01 16:09:04 +00001035 if( hasFK ){
drhf8ffb272013-11-01 17:08:56 +00001036 sqlite3FkCheck(pParse, pTab, regOldRowid, 0, aXRef, chngKey);
dane7a94d82009-10-01 16:09:04 +00001037 }
dan1da40a32009-09-19 17:00:31 +00001038
dan165921a2009-08-28 18:53:45 +00001039 /* Delete the index entries associated with the current record. */
danf0ee1d32015-09-12 19:26:11 +00001040 sqlite3GenerateRowIndexDelete(pParse, pTab, iDataCur, iIdxCur, aRegIdx, -1);
drh9b1c62d2011-03-30 21:04:43 +00001041
drhbe3da242019-12-29 00:52:41 +00001042 /* We must run the OP_FinishSeek opcode to resolve a prior
1043 ** OP_DeferredSeek if there is any possibility that there have been
1044 ** no OP_Column opcodes since the OP_DeferredSeek was issued. But
1045 ** we want to avoid the OP_FinishSeek if possible, as running it
1046 ** costs CPU cycles. */
1047 if( bFinishSeek ){
1048 sqlite3VdbeAddOp1(v, OP_FinishSeek, iDataCur);
drhd3ee3ad2019-12-26 23:16:18 +00001049 }
drhd3ee3ad2019-12-26 23:16:18 +00001050
dan46c47d42011-03-01 18:42:07 +00001051 /* If changing the rowid value, or if there are foreign key constraints
1052 ** to process, delete the old record. Otherwise, add a noop OP_Delete
1053 ** to invoke the pre-update hook.
dan37db03b2011-03-16 19:59:18 +00001054 **
1055 ** That (regNew==regnewRowid+1) is true is also important for the
drh9b1c62d2011-03-30 21:04:43 +00001056 ** pre-update hook. If the caller invokes preupdate_new(), the returned
dan37db03b2011-03-16 19:59:18 +00001057 ** value is copied from memory cell (regNewRowid+1+iCol), where iCol
1058 ** is the column index supplied by the user.
dan46c47d42011-03-01 18:42:07 +00001059 */
dan37db03b2011-03-16 19:59:18 +00001060 assert( regNew==regNewRowid+1 );
drh74c33022016-03-30 12:56:55 +00001061#ifdef SQLITE_ENABLE_PREUPDATE_HOOK
drhcbf1b8e2013-11-11 22:55:26 +00001062 sqlite3VdbeAddOp3(v, OP_Delete, iDataCur,
dan940b5ea2017-04-11 19:58:55 +00001063 OPFLAG_ISUPDATE | ((hasFK>1 || chngKey) ? 0 : OPFLAG_ISNOOP),
dan46c47d42011-03-01 18:42:07 +00001064 regNewRowid
1065 );
dane206ea72017-01-11 20:10:30 +00001066 if( eOnePass==ONEPASS_MULTI ){
1067 assert( hasFK==0 && chngKey==0 );
1068 sqlite3VdbeChangeP5(v, OPFLAG_SAVEPOSITION);
1069 }
dan46c47d42011-03-01 18:42:07 +00001070 if( !pParse->nested ){
drhf14b7fb2016-12-07 21:35:55 +00001071 sqlite3VdbeAppendP4(v, pTab, P4_TABLE);
drh5cf590c2003-04-24 01:45:04 +00001072 }
drh74c33022016-03-30 12:56:55 +00001073#else
dan940b5ea2017-04-11 19:58:55 +00001074 if( hasFK>1 || chngKey ){
drh74c33022016-03-30 12:56:55 +00001075 sqlite3VdbeAddOp2(v, OP_Delete, iDataCur, 0);
1076 }
1077#endif
dan9277efa2009-09-28 11:54:21 +00001078
dane7a94d82009-10-01 16:09:04 +00001079 if( hasFK ){
drhf8ffb272013-11-01 17:08:56 +00001080 sqlite3FkCheck(pParse, pTab, 0, regNewRowid, aXRef, chngKey);
dane7a94d82009-10-01 16:09:04 +00001081 }
dan165921a2009-08-28 18:53:45 +00001082
dan76d462e2009-08-30 11:42:51 +00001083 /* Insert the new index entries and the new record. */
danf91c1312017-01-10 20:04:38 +00001084 sqlite3CompleteInsertion(
1085 pParse, pTab, iDataCur, iIdxCur, regNewRowid, aRegIdx,
1086 OPFLAG_ISUPDATE | (eOnePass==ONEPASS_MULTI ? OPFLAG_SAVEPOSITION : 0),
1087 0, 0
1088 );
dan1da40a32009-09-19 17:00:31 +00001089
1090 /* Do any ON CASCADE, SET NULL or SET DEFAULT operations required to
1091 ** handle rows (possibly in other tables) that refer via a foreign key
1092 ** to the row just updated. */
dane7a94d82009-10-01 16:09:04 +00001093 if( hasFK ){
drhf8ffb272013-11-01 17:08:56 +00001094 sqlite3FkActions(pParse, pTab, pChanges, regOldRowid, aXRef, chngKey);
dane7a94d82009-10-01 16:09:04 +00001095 }
drhcce7d172000-05-31 15:34:51 +00001096 }
1097
drh0ca3e242002-01-29 23:07:02 +00001098 /* Increment the row counter
drh1bee3d72001-10-15 00:44:35 +00001099 */
drh79636912018-04-19 23:52:39 +00001100 if( regRowCount ){
drh04adf412008-01-08 18:57:50 +00001101 sqlite3VdbeAddOp2(v, OP_AddImm, regRowCount, 1);
drh1bee3d72001-10-15 00:44:35 +00001102 }
1103
dan165921a2009-08-28 18:53:45 +00001104 sqlite3CodeRowTrigger(pParse, pTrigger, TK_UPDATE, pChanges,
drhc3e356f2013-11-04 18:34:46 +00001105 TRIGGER_AFTER, pTab, regOldRowid, onError, labelContinue);
danielk1977c3f9bad2002-05-15 08:30:12 +00001106
drhcce7d172000-05-31 15:34:51 +00001107 /* Repeat the above with the next record to be updated, until
1108 ** all record selected by the WHERE clause have been updated.
1109 */
danf91c1312017-01-10 20:04:38 +00001110 if( eOnePass==ONEPASS_SINGLE ){
drh702ba9f2013-11-07 21:25:13 +00001111 /* Nothing to do at end-of-loop for a single-pass */
danf91c1312017-01-10 20:04:38 +00001112 }else if( eOnePass==ONEPASS_MULTI ){
1113 sqlite3VdbeResolveLabel(v, labelContinue);
1114 sqlite3WhereEnd(pWInfo);
drh32881be2020-11-17 21:26:13 +00001115 }else{
drhea22abe2013-10-25 19:17:17 +00001116 sqlite3VdbeResolveLabel(v, labelContinue);
drh688852a2014-02-17 22:40:43 +00001117 sqlite3VdbeAddOp2(v, OP_Next, iEph, addrTop); VdbeCoverage(v);
drhea22abe2013-10-25 19:17:17 +00001118 }
1119 sqlite3VdbeResolveLabel(v, labelBreak);
danielk1977c3f9bad2002-05-15 08:30:12 +00001120
drh0b9f50d2009-06-23 20:28:53 +00001121 /* Update the sqlite_sequence table by storing the content of the
1122 ** maximum rowid counter values recorded while inserting into
1123 ** autoincrement tables.
1124 */
drhf3d7bbb2018-04-21 03:06:29 +00001125 if( pParse->nested==0 && pParse->pTriggerTab==0 && pUpsert==0 ){
drh0b9f50d2009-06-23 20:28:53 +00001126 sqlite3AutoincrementEnd(pParse);
1127 }
1128
drh1bee3d72001-10-15 00:44:35 +00001129 /*
drh79636912018-04-19 23:52:39 +00001130 ** Return the number of rows that were changed, if we are tracking
1131 ** that information.
drh1bee3d72001-10-15 00:44:35 +00001132 */
drh79636912018-04-19 23:52:39 +00001133 if( regRowCount ){
drh3b26b2b2021-12-01 19:17:14 +00001134 sqlite3CodeChangeCount(v, regRowCount, "rows updated");
drh1bee3d72001-10-15 00:44:35 +00001135 }
1136
drhcce7d172000-05-31 15:34:51 +00001137update_cleanup:
danielk19774adee202004-05-08 08:23:19 +00001138 sqlite3AuthContextPop(&sContext);
drh6a534992013-11-16 20:13:39 +00001139 sqlite3DbFree(db, aXRef); /* Also frees aRegIdx[] and aToOpen[] */
drh633e6d52008-07-28 19:34:53 +00001140 sqlite3SrcListDelete(db, pTabList);
1141 sqlite3ExprListDelete(db, pChanges);
1142 sqlite3ExprDelete(db, pWhere);
dan3b61ebb2017-11-10 20:13:14 +00001143#if defined(SQLITE_ENABLE_UPDATE_DELETE_LIMIT)
danb3c16b82017-11-09 19:53:06 +00001144 sqlite3ExprListDelete(db, pOrderBy);
1145 sqlite3ExprDelete(db, pLimit);
dan3b61ebb2017-11-10 20:13:14 +00001146#endif
drhcce7d172000-05-31 15:34:51 +00001147 return;
1148}
dan75cbd982009-09-21 16:06:03 +00001149/* Make sure "isView" and other macros defined above are undefined. Otherwise
peter.d.reid60ec9142014-09-06 16:39:46 +00001150** they may interfere with compilation of other functions in this file
dan75cbd982009-09-21 16:06:03 +00001151** (or in another file, if this file becomes part of the amalgamation). */
1152#ifdef isView
1153 #undef isView
1154#endif
1155#ifdef pTrigger
1156 #undef pTrigger
1157#endif
drh9c419382006-06-16 21:13:21 +00001158
1159#ifndef SQLITE_OMIT_VIRTUALTABLE
1160/*
1161** Generate code for an UPDATE of a virtual table.
1162**
dan0f400372015-09-28 20:03:49 +00001163** There are two possible strategies - the default and the special
1164** "onepass" strategy. Onepass is only used if the virtual table
1165** implementation indicates that pWhere may match at most one row.
1166**
1167** The default strategy is to create an ephemeral table that contains
drh9c419382006-06-16 21:13:21 +00001168** for each row to be changed:
1169**
1170** (A) The original rowid of that row.
dan0f400372015-09-28 20:03:49 +00001171** (B) The revised rowid for the row.
drh9c419382006-06-16 21:13:21 +00001172** (C) The content of every column in the row.
1173**
dan0f400372015-09-28 20:03:49 +00001174** Then loop through the contents of this ephemeral table executing a
1175** VUpdate for each row. When finished, drop the ephemeral table.
drh9c419382006-06-16 21:13:21 +00001176**
dan0f400372015-09-28 20:03:49 +00001177** The "onepass" strategy does not use an ephemeral table. Instead, it
1178** stores the same values (A, B and C above) in a register array and
1179** makes a single invocation of VUpdate.
drh9c419382006-06-16 21:13:21 +00001180*/
1181static void updateVirtualTable(
1182 Parse *pParse, /* The parsing context */
1183 SrcList *pSrc, /* The virtual table to be modified */
1184 Table *pTab, /* The virtual table */
1185 ExprList *pChanges, /* The columns to change in the UPDATE statement */
1186 Expr *pRowid, /* Expression used to recompute the rowid */
1187 int *aXRef, /* Mapping from columns of pTab to entries in pChanges */
danb061d052011-04-25 18:49:57 +00001188 Expr *pWhere, /* WHERE clause of the UPDATE statement */
1189 int onError /* ON CONFLICT strategy */
drh9c419382006-06-16 21:13:21 +00001190){
1191 Vdbe *v = pParse->pVdbe; /* Virtual machine under construction */
drh9c419382006-06-16 21:13:21 +00001192 int ephemTab; /* Table holding the result of the SELECT */
1193 int i; /* Loop counter */
drh17435752007-08-16 04:30:38 +00001194 sqlite3 *db = pParse->db; /* Database connection */
danielk1977595a5232009-07-24 17:58:53 +00001195 const char *pVTab = (const char*)sqlite3GetVTable(db, pTab);
drha7c74002020-07-18 18:44:59 +00001196 WhereInfo *pWInfo = 0;
dan0f400372015-09-28 20:03:49 +00001197 int nArg = 2 + pTab->nCol; /* Number of arguments to VUpdate */
1198 int regArg; /* First register in VUpdate arg array */
1199 int regRec; /* Register in which to assemble record */
1200 int regRowid; /* Register for ephem table rowid */
1201 int iCsr = pSrc->a[0].iCursor; /* Cursor used for virtual table scan */
1202 int aDummy[2]; /* Unused arg for sqlite3WhereOkOnePass() */
drh338e3112018-05-24 23:51:57 +00001203 int eOnePass; /* True to use onepass strategy */
dan0f400372015-09-28 20:03:49 +00001204 int addr; /* Address of OP_OpenEphemeral */
drh9c419382006-06-16 21:13:21 +00001205
drh24cd20f2018-01-11 13:43:04 +00001206 /* Allocate nArg registers in which to gather the arguments for VUpdate. Then
dan0f400372015-09-28 20:03:49 +00001207 ** create and open the ephemeral table in which the records created from
1208 ** these arguments will be temporarily stored. */
drh9c419382006-06-16 21:13:21 +00001209 assert( v );
1210 ephemTab = pParse->nTab++;
dan0f400372015-09-28 20:03:49 +00001211 addr= sqlite3VdbeAddOp2(v, OP_OpenEphemeral, ephemTab, nArg);
1212 regArg = pParse->nMem + 1;
1213 pParse->nMem += nArg;
dan69887c92020-04-27 20:55:33 +00001214 if( pSrc->nSrc>1 ){
dan4d906f12020-10-12 14:29:11 +00001215 Index *pPk = 0;
dan576d5a82020-07-15 18:30:01 +00001216 Expr *pRow;
1217 ExprList *pList;
dan4d906f12020-10-12 14:29:11 +00001218 if( HasRowid(pTab) ){
1219 if( pRowid ){
1220 pRow = sqlite3ExprDup(db, pRowid, 0);
1221 }else{
1222 pRow = sqlite3PExpr(pParse, TK_ROW, 0, 0);
1223 }
drhe3740f22017-08-10 15:19:39 +00001224 }else{
dan4d906f12020-10-12 14:29:11 +00001225 i16 iPk; /* PRIMARY KEY column */
1226 pPk = sqlite3PrimaryKeyIndex(pTab);
1227 assert( pPk!=0 );
1228 assert( pPk->nKeyCol==1 );
1229 iPk = pPk->aiColumn[0];
1230 if( aXRef[iPk]>=0 ){
1231 pRow = sqlite3ExprDup(db, pChanges->a[aXRef[iPk]].pExpr, 0);
1232 }else{
1233 pRow = exprRowColumn(pParse, iPk);
1234 }
drhe3740f22017-08-10 15:19:39 +00001235 }
dan576d5a82020-07-15 18:30:01 +00001236 pList = sqlite3ExprListAppend(pParse, 0, pRow);
1237
dan69887c92020-04-27 20:55:33 +00001238 for(i=0; i<pTab->nCol; i++){
1239 if( aXRef[i]>=0 ){
1240 pList = sqlite3ExprListAppend(pParse, pList,
1241 sqlite3ExprDup(db, pChanges->a[aXRef[i]].pExpr, 0)
1242 );
1243 }else{
dan576d5a82020-07-15 18:30:01 +00001244 pList = sqlite3ExprListAppend(pParse, pList, exprRowColumn(pParse, i));
dan69887c92020-04-27 20:55:33 +00001245 }
1246 }
1247
dan4d906f12020-10-12 14:29:11 +00001248 updateFromSelect(pParse, ephemTab, pPk, pList, pSrc, pWhere, 0, 0);
dan69887c92020-04-27 20:55:33 +00001249 sqlite3ExprListDelete(db, pList);
1250 eOnePass = ONEPASS_OFF;
drhe3740f22017-08-10 15:19:39 +00001251 }else{
dan69887c92020-04-27 20:55:33 +00001252 regRec = ++pParse->nMem;
1253 regRowid = ++pParse->nMem;
dan0f400372015-09-28 20:03:49 +00001254
dan69887c92020-04-27 20:55:33 +00001255 /* Start scanning the virtual table */
drh895bab32022-01-27 16:14:50 +00001256 pWInfo = sqlite3WhereBegin(
1257 pParse, pSrc, pWhere, 0, 0, 0, WHERE_ONEPASS_DESIRED, 0
1258 );
dan69887c92020-04-27 20:55:33 +00001259 if( pWInfo==0 ) return;
dan0f400372015-09-28 20:03:49 +00001260
dan69887c92020-04-27 20:55:33 +00001261 /* Populate the argument registers. */
1262 for(i=0; i<pTab->nCol; i++){
1263 assert( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)==0 );
1264 if( aXRef[i]>=0 ){
1265 sqlite3ExprCode(pParse, pChanges->a[aXRef[i]].pExpr, regArg+2+i);
1266 }else{
1267 sqlite3VdbeAddOp3(v, OP_VColumn, iCsr, i, regArg+2+i);
1268 sqlite3VdbeChangeP5(v, OPFLAG_NOCHNG);/* For sqlite3_vtab_nochange() */
1269 }
1270 }
1271 if( HasRowid(pTab) ){
1272 sqlite3VdbeAddOp2(v, OP_Rowid, iCsr, regArg);
1273 if( pRowid ){
1274 sqlite3ExprCode(pParse, pRowid, regArg+1);
1275 }else{
1276 sqlite3VdbeAddOp2(v, OP_Rowid, iCsr, regArg+1);
1277 }
1278 }else{
1279 Index *pPk; /* PRIMARY KEY index */
1280 i16 iPk; /* PRIMARY KEY column */
1281 pPk = sqlite3PrimaryKeyIndex(pTab);
1282 assert( pPk!=0 );
1283 assert( pPk->nKeyCol==1 );
1284 iPk = pPk->aiColumn[0];
1285 sqlite3VdbeAddOp3(v, OP_VColumn, iCsr, iPk, regArg);
1286 sqlite3VdbeAddOp2(v, OP_SCopy, regArg+2+iPk, regArg+1);
1287 }
drh338e3112018-05-24 23:51:57 +00001288
dan69887c92020-04-27 20:55:33 +00001289 eOnePass = sqlite3WhereOkOnePass(pWInfo, aDummy);
1290
1291 /* There is no ONEPASS_MULTI on virtual tables */
1292 assert( eOnePass==ONEPASS_OFF || eOnePass==ONEPASS_SINGLE );
1293
1294 if( eOnePass ){
1295 /* If using the onepass strategy, no-op out the OP_OpenEphemeral coded
1296 ** above. */
1297 sqlite3VdbeChangeToNoop(v, addr);
1298 sqlite3VdbeAddOp1(v, OP_Close, iCsr);
1299 }else{
1300 /* Create a record from the argument register contents and insert it into
1301 ** the ephemeral table. */
1302 sqlite3MultiWrite(pParse);
1303 sqlite3VdbeAddOp3(v, OP_MakeRecord, regArg, nArg, regRec);
drhda369332020-06-29 20:09:04 +00001304#if defined(SQLITE_DEBUG) && !defined(SQLITE_ENABLE_NULL_TRIM)
dan69887c92020-04-27 20:55:33 +00001305 /* Signal an assert() within OP_MakeRecord that it is allowed to
1306 ** accept no-change records with serial_type 10 */
1307 sqlite3VdbeChangeP5(v, OPFLAG_NOCHNG_MAGIC);
drh41fb3672018-01-12 23:18:38 +00001308#endif
dan69887c92020-04-27 20:55:33 +00001309 sqlite3VdbeAddOp2(v, OP_NewRowid, ephemTab, regRowid);
1310 sqlite3VdbeAddOp3(v, OP_Insert, ephemTab, regRec, regRowid);
1311 }
dan0f400372015-09-28 20:03:49 +00001312 }
1313
dan0f400372015-09-28 20:03:49 +00001314
drh338e3112018-05-24 23:51:57 +00001315 if( eOnePass==ONEPASS_OFF ){
dan354474a2015-09-29 10:11:26 +00001316 /* End the virtual table scan */
dan69887c92020-04-27 20:55:33 +00001317 if( pSrc->nSrc==1 ){
1318 sqlite3WhereEnd(pWInfo);
1319 }
dan354474a2015-09-29 10:11:26 +00001320
dan0f400372015-09-28 20:03:49 +00001321 /* Begin scannning through the ephemeral table. */
dan354474a2015-09-29 10:11:26 +00001322 addr = sqlite3VdbeAddOp1(v, OP_Rewind, ephemTab); VdbeCoverage(v);
dan0f400372015-09-28 20:03:49 +00001323
1324 /* Extract arguments from the current row of the ephemeral table and
1325 ** invoke the VUpdate method. */
1326 for(i=0; i<nArg; i++){
1327 sqlite3VdbeAddOp3(v, OP_Column, ephemTab, i, regArg+i);
1328 }
drh9c419382006-06-16 21:13:21 +00001329 }
drh4f3dd152008-04-28 18:46:43 +00001330 sqlite3VtabMakeWritable(pParse, pTab);
dan0f400372015-09-28 20:03:49 +00001331 sqlite3VdbeAddOp4(v, OP_VUpdate, 0, nArg, regArg, pVTab, P4_VTAB);
danb061d052011-04-25 18:49:57 +00001332 sqlite3VdbeChangeP5(v, onError==OE_Default ? OE_Abort : onError);
dane0af83a2009-09-08 19:15:01 +00001333 sqlite3MayAbort(pParse);
drh9c419382006-06-16 21:13:21 +00001334
dan354474a2015-09-29 10:11:26 +00001335 /* End of the ephemeral table scan. Or, if using the onepass strategy,
1336 ** jump to here if the scan visited zero rows. */
drh338e3112018-05-24 23:51:57 +00001337 if( eOnePass==ONEPASS_OFF ){
dan0f400372015-09-28 20:03:49 +00001338 sqlite3VdbeAddOp2(v, OP_Next, ephemTab, addr+1); VdbeCoverage(v);
1339 sqlite3VdbeJumpHere(v, addr);
1340 sqlite3VdbeAddOp2(v, OP_Close, ephemTab, 0);
dan354474a2015-09-29 10:11:26 +00001341 }else{
1342 sqlite3WhereEnd(pWInfo);
dan0f400372015-09-28 20:03:49 +00001343 }
drh9c419382006-06-16 21:13:21 +00001344}
1345#endif /* SQLITE_OMIT_VIRTUALTABLE */