drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 1 | /* |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 2 | ** 2001 September 15 |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 3 | ** |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 4 | ** The author disclaims copyright to this source code. In place of |
| 5 | ** a legal notice, here is a blessing: |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 6 | ** |
drh | b19a2bc | 2001-09-16 00:13:26 +0000 | [diff] [blame] | 7 | ** May you do good and not evil. |
| 8 | ** May you find forgiveness for yourself and forgive others. |
| 9 | ** May you share freely, never taking more than you give. |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 10 | ** |
drh | a3388cc | 2010-07-30 16:54:25 +0000 | [diff] [blame] | 11 | ************************************************************************* |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 12 | ** This file contains C code routines that are called by the parser |
| 13 | ** to handle UPDATE statements. |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 14 | */ |
| 15 | #include "sqliteInt.h" |
| 16 | |
drh | edb193b | 2006-06-27 13:20:21 +0000 | [diff] [blame] | 17 | #ifndef SQLITE_OMIT_VIRTUALTABLE |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 18 | /* Forward declaration */ |
| 19 | static 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 */ |
dan | b061d05 | 2011-04-25 18:49:57 +0000 | [diff] [blame] | 26 | Expr *pWhere, /* WHERE clause of the UPDATE statement */ |
| 27 | int onError /* ON CONFLICT strategy */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 28 | ); |
drh | edb193b | 2006-06-27 13:20:21 +0000 | [diff] [blame] | 29 | #endif /* SQLITE_OMIT_VIRTUALTABLE */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 30 | |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 31 | /* |
drh | 8a51256 | 2005-11-14 22:29:05 +0000 | [diff] [blame] | 32 | ** The most recently coded instruction was an OP_Column to retrieve the |
drh | 66a5167 | 2008-01-03 00:01:23 +0000 | [diff] [blame] | 33 | ** i-th column of table pTab. This routine sets the P4 parameter of the |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 34 | ** 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 |
drh | 66a5167 | 2008-01-03 00:01:23 +0000 | [diff] [blame] | 41 | ** from the P4 parameter of the OP_Column instruction, is returned instead. |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 42 | ** If the former, then all row-records are guaranteed to include a value |
drh | 66a5167 | 2008-01-03 00:01:23 +0000 | [diff] [blame] | 43 | ** for the column and the P4 value is not required. |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 44 | ** |
| 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 |
drh | 1e32bed | 2020-06-19 13:33:53 +0000 | [diff] [blame] | 49 | ** into the sqlite_schema table.) |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 50 | ** |
drh | 66a5167 | 2008-01-03 00:01:23 +0000 | [diff] [blame] | 51 | ** Therefore, the P4 parameter is only required if the default value for |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 52 | ** 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. |
danielk1977 | c7538b5 | 2009-07-27 10:05:04 +0000 | [diff] [blame] | 55 | ** |
drh | 8c1febb | 2020-05-16 18:01:02 +0000 | [diff] [blame] | 56 | ** 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. |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 60 | */ |
danielk1977 | c7538b5 | 2009-07-27 10:05:04 +0000 | [diff] [blame] | 61 | void sqlite3ColumnDefault(Vdbe *v, Table *pTab, int i, int iReg){ |
drh | c9ef12f | 2022-10-10 21:21:04 +0000 | [diff] [blame] | 62 | Column *pCol; |
drh | e289d60 | 2009-05-05 15:46:09 +0000 | [diff] [blame] | 63 | assert( pTab!=0 ); |
drh | 49d77ee | 2022-10-10 18:25:05 +0000 | [diff] [blame] | 64 | assert( pTab->nCol>i ); |
| 65 | pCol = &pTab->aCol[i]; |
| 66 | if( pCol->iDflt ){ |
dan | 7a41923 | 2013-08-06 20:01:43 +0000 | [diff] [blame] | 67 | sqlite3_value *pValue = 0; |
danielk1977 | 14db266 | 2006-01-09 16:12:04 +0000 | [diff] [blame] | 68 | u8 enc = ENC(sqlite3VdbeDb(v)); |
drh | 49d77ee | 2022-10-10 18:25:05 +0000 | [diff] [blame] | 69 | assert( !IsView(pTab) ); |
drh | cf9d36d | 2021-08-02 18:03:43 +0000 | [diff] [blame] | 70 | VdbeComment((v, "%s.%s", pTab->zName, pCol->zCnName)); |
danielk1977 | c9cf6e3 | 2007-06-25 16:29:33 +0000 | [diff] [blame] | 71 | assert( i<pTab->nCol ); |
drh | 79cf2b7 | 2021-07-31 20:30:41 +0000 | [diff] [blame] | 72 | sqlite3ValueFromExpr(sqlite3VdbeDb(v), |
| 73 | sqlite3ColumnExpr(pTab,pCol), enc, |
drh | d4e70eb | 2008-01-02 00:34:36 +0000 | [diff] [blame] | 74 | pCol->affinity, &pValue); |
drh | 29dda4a | 2005-07-21 18:23:20 +0000 | [diff] [blame] | 75 | if( pValue ){ |
drh | f14b7fb | 2016-12-07 21:35:55 +0000 | [diff] [blame] | 76 | sqlite3VdbeAppendP4(v, pValue, P4_MEM); |
drh | 29dda4a | 2005-07-21 18:23:20 +0000 | [diff] [blame] | 77 | } |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 78 | } |
drh | da06005 | 2017-01-16 16:43:02 +0000 | [diff] [blame] | 79 | #ifndef SQLITE_OMIT_FLOATING_POINT |
drh | 49d77ee | 2022-10-10 18:25:05 +0000 | [diff] [blame] | 80 | if( pCol->affinity==SQLITE_AFF_REAL && !IsVirtual(pTab) ){ |
drh | da06005 | 2017-01-16 16:43:02 +0000 | [diff] [blame] | 81 | sqlite3VdbeAddOp1(v, OP_RealAffinity, iReg); |
| 82 | } |
| 83 | #endif |
danielk1977 | aee18ef | 2005-03-09 12:26:50 +0000 | [diff] [blame] | 84 | } |
| 85 | |
| 86 | /* |
drh | e9816d8 | 2018-09-15 21:38:48 +0000 | [diff] [blame] | 87 | ** Check to see if column iCol of index pIdx references any of the |
| 88 | ** columns defined by aXRef and chngRowid. Return true if it does |
drh | 40b4e7a | 2018-09-17 20:47:38 +0000 | [diff] [blame] | 89 | ** and false if not. This is an optimization. False-positives are a |
| 90 | ** performance degradation, but false-negatives can result in a corrupt |
| 91 | ** index and incorrect answers. |
drh | e9816d8 | 2018-09-15 21:38:48 +0000 | [diff] [blame] | 92 | ** |
drh | e9816d8 | 2018-09-15 21:38:48 +0000 | [diff] [blame] | 93 | ** aXRef[j] will be non-negative if column j of the original table is |
| 94 | ** being updated. chngRowid will be true if the rowid of the table is |
| 95 | ** being updated. |
| 96 | */ |
drh | 2b7cdf6 | 2018-09-16 15:01:25 +0000 | [diff] [blame] | 97 | static int indexColumnIsBeingUpdated( |
| 98 | Index *pIdx, /* The index to check */ |
| 99 | int iCol, /* Which column of the index to check */ |
drh | e9816d8 | 2018-09-15 21:38:48 +0000 | [diff] [blame] | 100 | int *aXRef, /* aXRef[j]>=0 if column j is being updated */ |
| 101 | int chngRowid /* true if the rowid is being updated */ |
| 102 | ){ |
drh | 2b7cdf6 | 2018-09-16 15:01:25 +0000 | [diff] [blame] | 103 | i16 iIdxCol = pIdx->aiColumn[iCol]; |
drh | 86f3492 | 2018-09-16 18:23:48 +0000 | [diff] [blame] | 104 | assert( iIdxCol!=XN_ROWID ); /* Cannot index rowid */ |
drh | 2b7cdf6 | 2018-09-16 15:01:25 +0000 | [diff] [blame] | 105 | if( iIdxCol>=0 ){ |
| 106 | return aXRef[iIdxCol]>=0; |
| 107 | } |
drh | 2b7cdf6 | 2018-09-16 15:01:25 +0000 | [diff] [blame] | 108 | assert( iIdxCol==XN_EXPR ); |
drh | e9816d8 | 2018-09-15 21:38:48 +0000 | [diff] [blame] | 109 | assert( pIdx->aColExpr!=0 ); |
| 110 | assert( pIdx->aColExpr->a[iCol].pExpr!=0 ); |
| 111 | return sqlite3ExprReferencesUpdatedColumn(pIdx->aColExpr->a[iCol].pExpr, |
| 112 | aXRef,chngRowid); |
| 113 | } |
| 114 | |
| 115 | /* |
drh | 40b4e7a | 2018-09-17 20:47:38 +0000 | [diff] [blame] | 116 | ** Check to see if index pIdx is a partial index whose conditional |
| 117 | ** expression might change values due to an UPDATE. Return true if |
| 118 | ** the index is subject to change and false if the index is guaranteed |
| 119 | ** to be unchanged. This is an optimization. False-positives are a |
| 120 | ** performance degradation, but false-negatives can result in a corrupt |
| 121 | ** index and incorrect answers. |
| 122 | ** |
| 123 | ** aXRef[j] will be non-negative if column j of the original table is |
| 124 | ** being updated. chngRowid will be true if the rowid of the table is |
| 125 | ** being updated. |
| 126 | */ |
| 127 | static int indexWhereClauseMightChange( |
| 128 | Index *pIdx, /* The index to check */ |
| 129 | int *aXRef, /* aXRef[j]>=0 if column j is being updated */ |
| 130 | int chngRowid /* true if the rowid is being updated */ |
| 131 | ){ |
| 132 | if( pIdx->pPartIdxWhere==0 ) return 0; |
| 133 | return sqlite3ExprReferencesUpdatedColumn(pIdx->pPartIdxWhere, |
| 134 | aXRef, chngRowid); |
| 135 | } |
| 136 | |
| 137 | /* |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 138 | ** Allocate and return a pointer to an expression of type TK_ROW with |
| 139 | ** Expr.iColumn set to value (iCol+1). The resolver will modify the |
| 140 | ** expression to be a TK_COLUMN reading column iCol of the first |
| 141 | ** table in the source-list (pSrc->a[0]). |
| 142 | */ |
| 143 | static Expr *exprRowColumn(Parse *pParse, int iCol){ |
| 144 | Expr *pRet = sqlite3PExpr(pParse, TK_ROW, 0, 0); |
| 145 | if( pRet ) pRet->iColumn = iCol+1; |
| 146 | return pRet; |
| 147 | } |
| 148 | |
| 149 | /* |
| 150 | ** Assuming both the pLimit and pOrderBy parameters are NULL, this function |
| 151 | ** generates VM code to run the query: |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 152 | ** |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 153 | ** SELECT <other-columns>, pChanges FROM pTabList WHERE pWhere |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 154 | ** |
| 155 | ** and write the results to the ephemeral table already opened as cursor |
| 156 | ** iEph. None of pChanges, pTabList or pWhere are modified or consumed by |
| 157 | ** this function, they must be deleted by the caller. |
| 158 | ** |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 159 | ** Or, if pLimit and pOrderBy are not NULL, and pTab is not a view: |
| 160 | ** |
| 161 | ** SELECT <other-columns>, pChanges FROM pTabList |
| 162 | ** WHERE pWhere |
| 163 | ** GROUP BY <other-columns> |
| 164 | ** ORDER BY pOrderBy LIMIT pLimit |
| 165 | ** |
| 166 | ** If pTab is a view, the GROUP BY clause is omitted. |
| 167 | ** |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 168 | ** Exactly how results are written to table iEph, and exactly what |
| 169 | ** the <other-columns> in the query above are is determined by the type |
| 170 | ** of table pTabList->a[0].pTab. |
| 171 | ** |
| 172 | ** If the table is a WITHOUT ROWID table, then argument pPk must be its |
| 173 | ** PRIMARY KEY. In this case <other-columns> are the primary key columns |
| 174 | ** of the table, in order. The results of the query are written to ephemeral |
| 175 | ** table iEph as index keys, using OP_IdxInsert. |
| 176 | ** |
| 177 | ** If the table is actually a view, then <other-columns> are all columns of |
| 178 | ** the view. The results are written to the ephemeral table iEph as records |
| 179 | ** with automatically assigned integer keys. |
| 180 | ** |
| 181 | ** If the table is a virtual or ordinary intkey table, then <other-columns> |
| 182 | ** is its rowid. For a virtual table, the results are written to iEph as |
| 183 | ** records with automatically assigned integer keys For intkey tables, the |
| 184 | ** rowid value in <other-columns> is used as the integer key, and the |
| 185 | ** remaining fields make up the table record. |
| 186 | */ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 187 | static void updateFromSelect( |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 188 | Parse *pParse, /* Parse context */ |
| 189 | int iEph, /* Cursor for open eph. table */ |
| 190 | Index *pPk, /* PK if table 0 is WITHOUT ROWID */ |
| 191 | ExprList *pChanges, /* List of expressions to return */ |
| 192 | SrcList *pTabList, /* List of tables to select from */ |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 193 | Expr *pWhere, /* WHERE clause for query */ |
drh | a7c7400 | 2020-07-18 18:44:59 +0000 | [diff] [blame] | 194 | ExprList *pOrderBy, /* ORDER BY clause */ |
| 195 | Expr *pLimit /* LIMIT clause */ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 196 | ){ |
| 197 | int i; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 198 | SelectDest dest; |
| 199 | Select *pSelect = 0; |
| 200 | ExprList *pList = 0; |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 201 | ExprList *pGrp = 0; |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 202 | Expr *pLimit2 = 0; |
| 203 | ExprList *pOrderBy2 = 0; |
dan | 1e11384 | 2020-04-30 15:49:56 +0000 | [diff] [blame] | 204 | sqlite3 *db = pParse->db; |
| 205 | Table *pTab = pTabList->a[0].pTab; |
| 206 | SrcList *pSrc; |
| 207 | Expr *pWhere2; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 208 | int eDest; |
| 209 | |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 210 | #ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT |
dan | 1e11384 | 2020-04-30 15:49:56 +0000 | [diff] [blame] | 211 | if( pOrderBy && pLimit==0 ) { |
| 212 | sqlite3ErrorMsg(pParse, "ORDER BY without LIMIT on UPDATE"); |
| 213 | return; |
| 214 | } |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 215 | pOrderBy2 = sqlite3ExprListDup(db, pOrderBy, 0); |
| 216 | pLimit2 = sqlite3ExprDup(db, pLimit, 0); |
drh | a7c7400 | 2020-07-18 18:44:59 +0000 | [diff] [blame] | 217 | #else |
| 218 | UNUSED_PARAMETER(pOrderBy); |
| 219 | UNUSED_PARAMETER(pLimit); |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 220 | #endif |
dan | 1e11384 | 2020-04-30 15:49:56 +0000 | [diff] [blame] | 221 | |
| 222 | pSrc = sqlite3SrcListDup(db, pTabList, 0); |
| 223 | pWhere2 = sqlite3ExprDup(db, pWhere, 0); |
dan | 1e11384 | 2020-04-30 15:49:56 +0000 | [diff] [blame] | 224 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 225 | assert( pTabList->nSrc>1 ); |
| 226 | if( pSrc ){ |
drh | cd1499f | 2021-05-20 00:44:04 +0000 | [diff] [blame] | 227 | pSrc->a[0].fg.notCte = 1; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 228 | pSrc->a[0].iCursor = -1; |
| 229 | pSrc->a[0].pTab->nTabRef--; |
| 230 | pSrc->a[0].pTab = 0; |
| 231 | } |
| 232 | if( pPk ){ |
| 233 | for(i=0; i<pPk->nKeyCol; i++){ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 234 | Expr *pNew = exprRowColumn(pParse, pPk->aiColumn[i]); |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 235 | #ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 236 | if( pLimit ){ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 237 | pGrp = sqlite3ExprListAppend(pParse, pGrp, sqlite3ExprDup(db, pNew, 0)); |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 238 | } |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 239 | #endif |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 240 | pList = sqlite3ExprListAppend(pParse, pList, pNew); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 241 | } |
dan | 4d906f1 | 2020-10-12 14:29:11 +0000 | [diff] [blame] | 242 | eDest = IsVirtual(pTab) ? SRT_Table : SRT_Upfrom; |
drh | f38524d | 2021-08-02 16:41:57 +0000 | [diff] [blame] | 243 | }else if( IsView(pTab) ){ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 244 | for(i=0; i<pTab->nCol; i++){ |
| 245 | pList = sqlite3ExprListAppend(pParse, pList, exprRowColumn(pParse, i)); |
| 246 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 247 | eDest = SRT_Table; |
| 248 | }else{ |
dan | 9ed322d | 2020-04-29 17:41:29 +0000 | [diff] [blame] | 249 | eDest = IsVirtual(pTab) ? SRT_Table : SRT_Upfrom; |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 250 | pList = sqlite3ExprListAppend(pParse, 0, sqlite3PExpr(pParse,TK_ROW,0,0)); |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 251 | #ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 252 | if( pLimit ){ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 253 | pGrp = sqlite3ExprListAppend(pParse, 0, sqlite3PExpr(pParse,TK_ROW,0,0)); |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 254 | } |
dan | 7e1d951 | 2020-07-17 15:34:54 +0000 | [diff] [blame] | 255 | #endif |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 256 | } |
drh | 2be1f2a | 2021-04-09 22:20:20 +0000 | [diff] [blame] | 257 | assert( pChanges!=0 || pParse->db->mallocFailed ); |
| 258 | if( pChanges ){ |
dan | 8b023cf | 2020-04-30 18:28:40 +0000 | [diff] [blame] | 259 | for(i=0; i<pChanges->nExpr; i++){ |
| 260 | pList = sqlite3ExprListAppend(pParse, pList, |
| 261 | sqlite3ExprDup(db, pChanges->a[i].pExpr, 0) |
| 262 | ); |
| 263 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 264 | } |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 265 | pSelect = sqlite3SelectNew(pParse, pList, |
dan | 5daf69e | 2021-07-05 11:27:13 +0000 | [diff] [blame] | 266 | pSrc, pWhere2, pGrp, 0, pOrderBy2, SF_UFSrcCheck|SF_IncludeHidden, pLimit2 |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 267 | ); |
drh | 0fb78f0 | 2021-07-16 01:19:19 +0000 | [diff] [blame] | 268 | if( pSelect ) pSelect->selFlags |= SF_OrderByReqd; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 269 | sqlite3SelectDestInit(&dest, eDest, iEph); |
dan | 9ed322d | 2020-04-29 17:41:29 +0000 | [diff] [blame] | 270 | dest.iSDParm2 = (pPk ? pPk->nKeyCol : -1); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 271 | sqlite3Select(pParse, pSelect, &dest); |
| 272 | sqlite3SelectDelete(db, pSelect); |
| 273 | } |
| 274 | |
| 275 | /* |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 276 | ** Process an UPDATE statement. |
drh | b2fe7d8 | 2003-04-20 17:29:23 +0000 | [diff] [blame] | 277 | ** |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 278 | ** UPDATE OR IGNORE tbl SET a=b, c=d FROM tbl2... WHERE e<5 AND f NOT NULL; |
| 279 | ** \_______/ \_/ \______/ \_____/ \________________/ |
| 280 | ** onError | pChanges | pWhere |
| 281 | ** \_______________________/ |
| 282 | ** pTabList |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 283 | */ |
dan | 3b61ebb | 2017-11-10 20:13:14 +0000 | [diff] [blame] | 284 | void sqlite3Update( |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 285 | Parse *pParse, /* The parser context */ |
drh | 113088e | 2003-03-20 01:16:58 +0000 | [diff] [blame] | 286 | SrcList *pTabList, /* The table in which we should change things */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 287 | ExprList *pChanges, /* Things to be changed */ |
drh | 9cfcf5d | 2002-01-29 18:41:24 +0000 | [diff] [blame] | 288 | Expr *pWhere, /* The WHERE clause. May be null */ |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 289 | int onError, /* How to handle constraint errors */ |
dan | 3b61ebb | 2017-11-10 20:13:14 +0000 | [diff] [blame] | 290 | ExprList *pOrderBy, /* ORDER BY clause. May be null */ |
drh | eac9fab | 2018-04-16 13:00:50 +0000 | [diff] [blame] | 291 | Expr *pLimit, /* LIMIT clause. May be null */ |
| 292 | Upsert *pUpsert /* ON CONFLICT clause, or null */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 293 | ){ |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 294 | int i, j, k; /* Loop counters */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 295 | Table *pTab; /* The table to be updated */ |
drh | c3e356f | 2013-11-04 18:34:46 +0000 | [diff] [blame] | 296 | int addrTop = 0; /* VDBE instruction address of the start of the loop */ |
drh | aec7dc6 | 2020-07-18 18:03:26 +0000 | [diff] [blame] | 297 | WhereInfo *pWInfo = 0; /* Information about the WHERE clause */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 298 | Vdbe *v; /* The virtual database engine */ |
| 299 | Index *pIdx; /* For looping over indices */ |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 300 | Index *pPk; /* The PRIMARY KEY index for WITHOUT ROWID tables */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 301 | int nIdx; /* Number of indices that need updating */ |
drh | 9e9374b | 2019-05-08 19:06:59 +0000 | [diff] [blame] | 302 | int nAllIdx; /* Total number of indexes */ |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 303 | int iBaseCur; /* Base cursor number */ |
drh | 77f64bb | 2013-10-31 12:13:37 +0000 | [diff] [blame] | 304 | int iDataCur; /* Cursor for the canonical data btree */ |
| 305 | int iIdxCur; /* Cursor for the first index */ |
drh | 9bb575f | 2004-09-06 17:24:11 +0000 | [diff] [blame] | 306 | sqlite3 *db; /* The database structure */ |
drh | a7c3b93 | 2019-05-07 19:13:42 +0000 | [diff] [blame] | 307 | int *aRegIdx = 0; /* Registers for to each index and the main table */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 308 | int *aXRef = 0; /* aXRef[i] is the index in pChanges->a[] of the |
drh | 967e8b7 | 2000-06-21 13:59:10 +0000 | [diff] [blame] | 309 | ** an expression for the i-th column of the table. |
| 310 | ** aXRef[i]==-1 if the i-th column is not changed. */ |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 311 | u8 *aToOpen; /* 1 for tables and indices to be opened */ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 312 | u8 chngPk; /* PRIMARY KEY changed in a WITHOUT ROWID table */ |
| 313 | u8 chngRowid; /* Rowid changed in a normal table */ |
| 314 | u8 chngKey; /* Either chngPk or chngRowid */ |
drh | f0863fe | 2005-06-12 21:35:51 +0000 | [diff] [blame] | 315 | Expr *pRowidExpr = 0; /* Expression defining the new record number */ |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 316 | int iRowidExpr = -1; /* Index of "rowid=" (or IPK) assignment in pChanges */ |
drh | 85e2096 | 2003-04-25 17:52:11 +0000 | [diff] [blame] | 317 | AuthContext sContext; /* The authorization context */ |
danielk1977 | b3bce66 | 2005-01-29 08:32:43 +0000 | [diff] [blame] | 318 | NameContext sNC; /* The name-context to resolve expressions in */ |
danielk1977 | da18423 | 2006-01-05 11:34:32 +0000 | [diff] [blame] | 319 | int iDb; /* Database containing the table being updated */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 320 | int eOnePass; /* ONEPASS_XXX value from where.c */ |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 321 | int hasFK; /* True if foreign key processing is required */ |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 322 | int labelBreak; /* Jump here to break out of UPDATE loop */ |
| 323 | int labelContinue; /* Jump here to continue next step of UPDATE loop */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 324 | int flags; /* Flags for sqlite3WhereBegin() */ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 325 | |
drh | 798da52 | 2004-11-04 04:42:28 +0000 | [diff] [blame] | 326 | #ifndef SQLITE_OMIT_TRIGGER |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 327 | int isView; /* True when updating a view (INSTEAD OF trigger) */ |
| 328 | Trigger *pTrigger; /* List of triggers on pTab, if required */ |
| 329 | int tmask; /* Mask of TRIGGER_BEFORE|TRIGGER_AFTER */ |
drh | 798da52 | 2004-11-04 04:42:28 +0000 | [diff] [blame] | 330 | #endif |
drh | fd7c991 | 2009-12-01 22:09:25 +0000 | [diff] [blame] | 331 | int newmask; /* Mask of NEW.* columns accessed by BEFORE triggers */ |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 332 | int iEph = 0; /* Ephemeral table holding all primary key values */ |
drh | 00f91cf | 2013-11-07 21:32:16 +0000 | [diff] [blame] | 333 | int nKey = 0; /* Number of elements in regKey for WITHOUT ROWID */ |
drh | fc8d4f9 | 2013-11-08 15:19:46 +0000 | [diff] [blame] | 334 | int aiCurOnePass[2]; /* The write cursors opened by WHERE_ONEPASS */ |
drh | ae051a8 | 2017-01-12 16:21:54 +0000 | [diff] [blame] | 335 | int addrOpen = 0; /* Address of OP_OpenEphemeral */ |
| 336 | int iPk = 0; /* First of nPk cells holding PRIMARY KEY value */ |
| 337 | i16 nPk = 0; /* Number of components of the PRIMARY KEY */ |
dan | 2c6fec2 | 2017-01-11 19:03:08 +0000 | [diff] [blame] | 338 | int bReplace = 0; /* True if REPLACE conflict resolution might happen */ |
drh | be3da24 | 2019-12-29 00:52:41 +0000 | [diff] [blame] | 339 | int bFinishSeek = 1; /* The OP_FinishSeek opcode is needed */ |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 340 | int nChangeFrom = 0; /* If there is a FROM, pChanges->nExpr, else 0 */ |
danielk1977 | c3f9bad | 2002-05-15 08:30:12 +0000 | [diff] [blame] | 341 | |
drh | 04adf41 | 2008-01-08 18:57:50 +0000 | [diff] [blame] | 342 | /* Register Allocations */ |
| 343 | int regRowCount = 0; /* A count of rows changed */ |
drh | 94f26a1 | 2015-09-29 12:32:56 +0000 | [diff] [blame] | 344 | int regOldRowid = 0; /* The old rowid */ |
| 345 | int regNewRowid = 0; /* The new rowid */ |
| 346 | int regNew = 0; /* Content of the NEW.* table in triggers */ |
drh | b8475df | 2011-12-09 16:21:19 +0000 | [diff] [blame] | 347 | int regOld = 0; /* Content of OLD.* table in triggers */ |
drh | 4f21c4a | 2008-12-10 22:15:00 +0000 | [diff] [blame] | 348 | int regRowSet = 0; /* Rowset of rows to be updated */ |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 349 | int regKey = 0; /* composite PRIMARY KEY value */ |
drh | 04adf41 | 2008-01-08 18:57:50 +0000 | [diff] [blame] | 350 | |
drh | eeb844a | 2009-08-08 18:01:07 +0000 | [diff] [blame] | 351 | memset(&sContext, 0, sizeof(sContext)); |
drh | 1743575 | 2007-08-16 04:30:38 +0000 | [diff] [blame] | 352 | db = pParse->db; |
drh | 0c7d3d3 | 2022-01-24 16:47:12 +0000 | [diff] [blame] | 353 | assert( db->pParse==pParse ); |
| 354 | if( pParse->nErr ){ |
drh | 6f7adc8 | 2006-01-11 21:41:20 +0000 | [diff] [blame] | 355 | goto update_cleanup; |
| 356 | } |
drh | 0c7d3d3 | 2022-01-24 16:47:12 +0000 | [diff] [blame] | 357 | assert( db->mallocFailed==0 ); |
drh | daffd0e | 2001-04-11 14:28:42 +0000 | [diff] [blame] | 358 | |
drh | b2fe7d8 | 2003-04-20 17:29:23 +0000 | [diff] [blame] | 359 | /* Locate the table which we want to update. |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 360 | */ |
danielk1977 | 4adee20 | 2004-05-08 08:23:19 +0000 | [diff] [blame] | 361 | pTab = sqlite3SrcListLookup(pParse, pTabList); |
drh | a69d916 | 2003-04-17 22:57:53 +0000 | [diff] [blame] | 362 | if( pTab==0 ) goto update_cleanup; |
danielk1977 | da18423 | 2006-01-05 11:34:32 +0000 | [diff] [blame] | 363 | iDb = sqlite3SchemaToIndex(pParse->db, pTab->pSchema); |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 364 | |
| 365 | /* Figure out if we have any triggers and if the table being |
dan | 76d462e | 2009-08-30 11:42:51 +0000 | [diff] [blame] | 366 | ** updated is a view. |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 367 | */ |
| 368 | #ifndef SQLITE_OMIT_TRIGGER |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 369 | pTrigger = sqlite3TriggersExist(pParse, pTab, TK_UPDATE, pChanges, &tmask); |
drh | f38524d | 2021-08-02 16:41:57 +0000 | [diff] [blame] | 370 | isView = IsView(pTab); |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 371 | assert( pTrigger || tmask==0 ); |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 372 | #else |
danielk1977 | 2f886d1 | 2009-02-28 10:47:41 +0000 | [diff] [blame] | 373 | # define pTrigger 0 |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 374 | # define isView 0 |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 375 | # define tmask 0 |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 376 | #endif |
| 377 | #ifdef SQLITE_OMIT_VIEW |
| 378 | # undef isView |
| 379 | # define isView 0 |
| 380 | #endif |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 381 | |
drh | 2a7dcbf | 2022-04-06 15:41:53 +0000 | [diff] [blame] | 382 | #if TREETRACE_ENABLED |
| 383 | if( sqlite3TreeTrace & 0x10000 ){ |
| 384 | sqlite3TreeViewLine(0, "In sqlite3Update() at %s:%d", __FILE__, __LINE__); |
| 385 | sqlite3TreeViewUpdate(pParse->pWith, pTabList, pChanges, pWhere, |
| 386 | onError, pOrderBy, pLimit, pUpsert, pTrigger); |
| 387 | } |
| 388 | #endif |
| 389 | |
dan | 243210b | 2020-07-15 15:32:59 +0000 | [diff] [blame] | 390 | /* If there was a FROM clause, set nChangeFrom to the number of expressions |
| 391 | ** in the change-list. Otherwise, set it to 0. There cannot be a FROM |
| 392 | ** clause if this function is being called to generate code for part of |
| 393 | ** an UPSERT statement. */ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 394 | nChangeFrom = (pTabList->nSrc>1) ? pChanges->nExpr : 0; |
| 395 | assert( nChangeFrom==0 || pUpsert==0 ); |
drh | b7f9164 | 2004-10-31 02:22:47 +0000 | [diff] [blame] | 396 | |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 397 | #ifdef SQLITE_ENABLE_UPDATE_DELETE_LIMIT |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 398 | if( !isView && nChangeFrom==0 ){ |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 399 | pWhere = sqlite3LimitWhere( |
drh | 8c0833f | 2017-11-14 23:48:23 +0000 | [diff] [blame] | 400 | pParse, pTabList, pWhere, pOrderBy, pLimit, "UPDATE" |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 401 | ); |
| 402 | pOrderBy = 0; |
drh | 8c0833f | 2017-11-14 23:48:23 +0000 | [diff] [blame] | 403 | pLimit = 0; |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 404 | } |
| 405 | #endif |
| 406 | |
danielk1977 | 595a523 | 2009-07-24 17:58:53 +0000 | [diff] [blame] | 407 | if( sqlite3ViewGetColumnNames(pParse, pTab) ){ |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 408 | goto update_cleanup; |
drh | a69d916 | 2003-04-17 22:57:53 +0000 | [diff] [blame] | 409 | } |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 410 | if( sqlite3IsReadOnly(pParse, pTab, tmask) ){ |
danielk1977 | b3d24bf | 2006-06-19 03:05:10 +0000 | [diff] [blame] | 411 | goto update_cleanup; |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 412 | } |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 413 | |
drh | 53e3fc7 | 2002-07-16 17:22:50 +0000 | [diff] [blame] | 414 | /* Allocate a cursors for the main database table and for all indices. |
| 415 | ** The index cursors might not be used, but if they are used they |
| 416 | ** need to occur right after the database cursor. So go ahead and |
| 417 | ** allocate enough space, just in case. |
| 418 | */ |
drh | 2ac4e5c | 2018-04-20 00:40:19 +0000 | [diff] [blame] | 419 | iBaseCur = iDataCur = pParse->nTab++; |
drh | 313619f | 2013-10-31 20:34:06 +0000 | [diff] [blame] | 420 | iIdxCur = iDataCur+1; |
drh | 77f64bb | 2013-10-31 12:13:37 +0000 | [diff] [blame] | 421 | pPk = HasRowid(pTab) ? 0 : sqlite3PrimaryKeyIndex(pTab); |
drh | 2ac4e5c | 2018-04-20 00:40:19 +0000 | [diff] [blame] | 422 | testcase( pPk!=0 && pPk!=pTab->pIndex ); |
drh | 77f64bb | 2013-10-31 12:13:37 +0000 | [diff] [blame] | 423 | for(nIdx=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, nIdx++){ |
drh | 2ac4e5c | 2018-04-20 00:40:19 +0000 | [diff] [blame] | 424 | if( pPk==pIdx ){ |
drh | 313619f | 2013-10-31 20:34:06 +0000 | [diff] [blame] | 425 | iDataCur = pParse->nTab; |
drh | 313619f | 2013-10-31 20:34:06 +0000 | [diff] [blame] | 426 | } |
drh | 53e3fc7 | 2002-07-16 17:22:50 +0000 | [diff] [blame] | 427 | pParse->nTab++; |
| 428 | } |
drh | 7fc3aba | 2018-04-20 13:18:51 +0000 | [diff] [blame] | 429 | if( pUpsert ){ |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 430 | /* On an UPSERT, reuse the same cursors already opened by INSERT */ |
drh | 7fc3aba | 2018-04-20 13:18:51 +0000 | [diff] [blame] | 431 | iDataCur = pUpsert->iDataCur; |
| 432 | iIdxCur = pUpsert->iIdxCur; |
| 433 | pParse->nTab = iBaseCur; |
| 434 | } |
drh | 2ac4e5c | 2018-04-20 00:40:19 +0000 | [diff] [blame] | 435 | pTabList->a[0].iCursor = iDataCur; |
drh | 53e3fc7 | 2002-07-16 17:22:50 +0000 | [diff] [blame] | 436 | |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 437 | /* Allocate space for aXRef[], aRegIdx[], and aToOpen[]. |
| 438 | ** Initialize aXRef[] and aToOpen[] to their default values. |
| 439 | */ |
drh | a7c3b93 | 2019-05-07 19:13:42 +0000 | [diff] [blame] | 440 | aXRef = sqlite3DbMallocRawNN(db, sizeof(int) * (pTab->nCol+nIdx+1) + nIdx+2 ); |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 441 | if( aXRef==0 ) goto update_cleanup; |
| 442 | aRegIdx = aXRef+pTab->nCol; |
drh | a7c3b93 | 2019-05-07 19:13:42 +0000 | [diff] [blame] | 443 | aToOpen = (u8*)(aRegIdx+nIdx+1); |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 444 | memset(aToOpen, 1, nIdx+1); |
| 445 | aToOpen[nIdx+1] = 0; |
| 446 | for(i=0; i<pTab->nCol; i++) aXRef[i] = -1; |
| 447 | |
danielk1977 | b3bce66 | 2005-01-29 08:32:43 +0000 | [diff] [blame] | 448 | /* Initialize the name-context */ |
| 449 | memset(&sNC, 0, sizeof(sNC)); |
| 450 | sNC.pParse = pParse; |
| 451 | sNC.pSrcList = pTabList; |
drh | eac9fab | 2018-04-16 13:00:50 +0000 | [diff] [blame] | 452 | sNC.uNC.pUpsert = pUpsert; |
| 453 | sNC.ncFlags = NC_UUpsert; |
danielk1977 | b3bce66 | 2005-01-29 08:32:43 +0000 | [diff] [blame] | 454 | |
mistachkin | 16fd04c | 2019-10-16 17:46:22 +0000 | [diff] [blame] | 455 | /* Begin generating code. */ |
| 456 | v = sqlite3GetVdbe(pParse); |
| 457 | if( v==0 ) goto update_cleanup; |
| 458 | |
drh | 85e2096 | 2003-04-25 17:52:11 +0000 | [diff] [blame] | 459 | /* Resolve the column names in all the expressions of the |
| 460 | ** of the UPDATE statement. Also find the column index |
drh | ed6c867 | 2003-01-12 18:02:16 +0000 | [diff] [blame] | 461 | ** for each column to be updated in the pChanges array. For each |
| 462 | ** column to be updated, make sure we have authorization to change |
| 463 | ** that column. |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 464 | */ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 465 | chngRowid = chngPk = 0; |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 466 | for(i=0; i<pChanges->nExpr; i++){ |
drh | cf9d36d | 2021-08-02 18:03:43 +0000 | [diff] [blame] | 467 | u8 hCol = sqlite3StrIHash(pChanges->a[i].zEName); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 468 | /* If this is an UPDATE with a FROM clause, do not resolve expressions |
| 469 | ** here. The call to sqlite3Select() below will do that. */ |
| 470 | if( nChangeFrom==0 && sqlite3ResolveExprNames(&sNC, pChanges->a[i].pExpr) ){ |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 471 | goto update_cleanup; |
| 472 | } |
| 473 | for(j=0; j<pTab->nCol; j++){ |
drh | cf9d36d | 2021-08-02 18:03:43 +0000 | [diff] [blame] | 474 | if( pTab->aCol[j].hName==hCol |
| 475 | && sqlite3StrICmp(pTab->aCol[j].zCnName, pChanges->a[i].zEName)==0 |
| 476 | ){ |
drh | 9647ff8 | 2002-01-14 02:56:24 +0000 | [diff] [blame] | 477 | if( j==pTab->iPKey ){ |
drh | f0863fe | 2005-06-12 21:35:51 +0000 | [diff] [blame] | 478 | chngRowid = 1; |
| 479 | pRowidExpr = pChanges->a[i].pExpr; |
dan | 7465787 | 2020-05-01 18:43:49 +0000 | [diff] [blame] | 480 | iRowidExpr = i; |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 481 | }else if( pPk && (pTab->aCol[j].colFlags & COLFLAG_PRIMKEY)!=0 ){ |
| 482 | chngPk = 1; |
drh | 4a32431 | 2001-12-21 14:30:42 +0000 | [diff] [blame] | 483 | } |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 484 | #ifndef SQLITE_OMIT_GENERATED_COLUMNS |
drh | 427b96a | 2019-10-22 13:01:24 +0000 | [diff] [blame] | 485 | else if( pTab->aCol[j].colFlags & COLFLAG_GENERATED ){ |
drh | 6ab61d7 | 2019-10-23 15:47:33 +0000 | [diff] [blame] | 486 | testcase( pTab->aCol[j].colFlags & COLFLAG_VIRTUAL ); |
| 487 | testcase( pTab->aCol[j].colFlags & COLFLAG_STORED ); |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 488 | sqlite3ErrorMsg(pParse, |
| 489 | "cannot UPDATE generated column \"%s\"", |
drh | cf9d36d | 2021-08-02 18:03:43 +0000 | [diff] [blame] | 490 | pTab->aCol[j].zCnName); |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 491 | goto update_cleanup; |
| 492 | } |
| 493 | #endif |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 494 | aXRef[j] = i; |
| 495 | break; |
| 496 | } |
| 497 | } |
| 498 | if( j>=pTab->nCol ){ |
drh | 41cee66 | 2019-12-12 20:22:34 +0000 | [diff] [blame] | 499 | if( pPk==0 && sqlite3IsRowid(pChanges->a[i].zEName) ){ |
drh | 2722898 | 2013-05-06 13:22:50 +0000 | [diff] [blame] | 500 | j = -1; |
drh | f0863fe | 2005-06-12 21:35:51 +0000 | [diff] [blame] | 501 | chngRowid = 1; |
| 502 | pRowidExpr = pChanges->a[i].pExpr; |
dan | 7465787 | 2020-05-01 18:43:49 +0000 | [diff] [blame] | 503 | iRowidExpr = i; |
drh | a0217ba | 2003-06-01 01:10:33 +0000 | [diff] [blame] | 504 | }else{ |
drh | 41cee66 | 2019-12-12 20:22:34 +0000 | [diff] [blame] | 505 | sqlite3ErrorMsg(pParse, "no such column: %s", pChanges->a[i].zEName); |
dan | 1db9510 | 2010-06-28 10:15:19 +0000 | [diff] [blame] | 506 | pParse->checkSchema = 1; |
drh | a0217ba | 2003-06-01 01:10:33 +0000 | [diff] [blame] | 507 | goto update_cleanup; |
| 508 | } |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 509 | } |
drh | ed6c867 | 2003-01-12 18:02:16 +0000 | [diff] [blame] | 510 | #ifndef SQLITE_OMIT_AUTHORIZATION |
drh | e5f9c64 | 2003-01-13 23:27:31 +0000 | [diff] [blame] | 511 | { |
| 512 | int rc; |
danielk1977 | 4adee20 | 2004-05-08 08:23:19 +0000 | [diff] [blame] | 513 | rc = sqlite3AuthCheck(pParse, SQLITE_UPDATE, pTab->zName, |
drh | cf9d36d | 2021-08-02 18:03:43 +0000 | [diff] [blame] | 514 | j<0 ? "ROWID" : pTab->aCol[j].zCnName, |
drh | 69c3382 | 2016-08-18 14:33:11 +0000 | [diff] [blame] | 515 | db->aDb[iDb].zDbSName); |
drh | e5f9c64 | 2003-01-13 23:27:31 +0000 | [diff] [blame] | 516 | if( rc==SQLITE_DENY ){ |
| 517 | goto update_cleanup; |
| 518 | }else if( rc==SQLITE_IGNORE ){ |
| 519 | aXRef[j] = -1; |
| 520 | } |
drh | ed6c867 | 2003-01-12 18:02:16 +0000 | [diff] [blame] | 521 | } |
| 522 | #endif |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 523 | } |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 524 | assert( (chngRowid & chngPk)==0 ); |
| 525 | assert( chngRowid==0 || chngRowid==1 ); |
| 526 | assert( chngPk==0 || chngPk==1 ); |
| 527 | chngKey = chngRowid + chngPk; |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 528 | |
drh | 7e7fd73 | 2019-10-22 13:59:23 +0000 | [diff] [blame] | 529 | #ifndef SQLITE_OMIT_GENERATED_COLUMNS |
| 530 | /* Mark generated columns as changing if their generator expressions |
| 531 | ** reference any changing column. The actual aXRef[] value for |
| 532 | ** generated expressions is not used, other than to check to see that it |
| 533 | ** is non-negative, so the value of aXRef[] for generated columns can be |
| 534 | ** set to any non-negative number. We use 99999 so that the value is |
| 535 | ** obvious when looking at aXRef[] in a symbolic debugger. |
| 536 | */ |
| 537 | if( pTab->tabFlags & TF_HasGenerated ){ |
| 538 | int bProgress; |
| 539 | testcase( pTab->tabFlags & TF_HasVirtual ); |
| 540 | testcase( pTab->tabFlags & TF_HasStored ); |
| 541 | do{ |
| 542 | bProgress = 0; |
| 543 | for(i=0; i<pTab->nCol; i++){ |
| 544 | if( aXRef[i]>=0 ) continue; |
| 545 | if( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)==0 ) continue; |
drh | 79cf2b7 | 2021-07-31 20:30:41 +0000 | [diff] [blame] | 546 | if( sqlite3ExprReferencesUpdatedColumn( |
| 547 | sqlite3ColumnExpr(pTab, &pTab->aCol[i]), |
| 548 | aXRef, chngRowid) |
| 549 | ){ |
drh | 7e7fd73 | 2019-10-22 13:59:23 +0000 | [diff] [blame] | 550 | aXRef[i] = 99999; |
| 551 | bProgress = 1; |
| 552 | } |
| 553 | } |
| 554 | }while( bProgress ); |
| 555 | } |
| 556 | #endif |
| 557 | |
dan | 1acb539 | 2015-11-26 19:33:41 +0000 | [diff] [blame] | 558 | /* The SET expressions are not actually used inside the WHERE loop. |
| 559 | ** So reset the colUsed mask. Unless this is a virtual table. In that |
| 560 | ** case, set all bits of the colUsed mask (to ensure that the virtual |
| 561 | ** table implementation makes all columns available). |
drh | 5c82f4d | 2013-11-19 02:34:11 +0000 | [diff] [blame] | 562 | */ |
drh | 8426e36 | 2016-03-08 01:32:30 +0000 | [diff] [blame] | 563 | pTabList->a[0].colUsed = IsVirtual(pTab) ? ALLBITS : 0; |
drh | 5c82f4d | 2013-11-19 02:34:11 +0000 | [diff] [blame] | 564 | |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 565 | hasFK = sqlite3FkRequired(pParse, pTab, aXRef, chngKey); |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 566 | |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 567 | /* There is one entry in the aRegIdx[] array for each index on the table |
| 568 | ** being updated. Fill in aRegIdx[] with a register number that will hold |
drh | 1f9ca2c | 2015-08-25 16:57:52 +0000 | [diff] [blame] | 569 | ** the key for accessing each index. |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 570 | */ |
drh | 247c1b4 | 2019-01-22 13:45:48 +0000 | [diff] [blame] | 571 | if( onError==OE_Replace ) bReplace = 1; |
drh | 9e9374b | 2019-05-08 19:06:59 +0000 | [diff] [blame] | 572 | for(nAllIdx=0, pIdx=pTab->pIndex; pIdx; pIdx=pIdx->pNext, nAllIdx++){ |
drh | aa9b896 | 2008-01-08 02:57:55 +0000 | [diff] [blame] | 573 | int reg; |
drh | 40b4e7a | 2018-09-17 20:47:38 +0000 | [diff] [blame] | 574 | if( chngKey || hasFK>1 || pIdx==pPk |
| 575 | || indexWhereClauseMightChange(pIdx,aXRef,chngRowid) |
| 576 | ){ |
drh | aa9b896 | 2008-01-08 02:57:55 +0000 | [diff] [blame] | 577 | reg = ++pParse->nMem; |
drh | 2c4dfc3 | 2016-11-10 14:24:04 +0000 | [diff] [blame] | 578 | pParse->nMem += pIdx->nColumn; |
drh | aa9b896 | 2008-01-08 02:57:55 +0000 | [diff] [blame] | 579 | }else{ |
| 580 | reg = 0; |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 581 | for(i=0; i<pIdx->nKeyCol; i++){ |
drh | 2b7cdf6 | 2018-09-16 15:01:25 +0000 | [diff] [blame] | 582 | if( indexColumnIsBeingUpdated(pIdx, i, aXRef, chngRowid) ){ |
drh | aa9b896 | 2008-01-08 02:57:55 +0000 | [diff] [blame] | 583 | reg = ++pParse->nMem; |
drh | 2c4dfc3 | 2016-11-10 14:24:04 +0000 | [diff] [blame] | 584 | pParse->nMem += pIdx->nColumn; |
drh | 247c1b4 | 2019-01-22 13:45:48 +0000 | [diff] [blame] | 585 | if( onError==OE_Default && pIdx->onError==OE_Replace ){ |
dan | 2c6fec2 | 2017-01-11 19:03:08 +0000 | [diff] [blame] | 586 | bReplace = 1; |
| 587 | } |
drh | aa9b896 | 2008-01-08 02:57:55 +0000 | [diff] [blame] | 588 | break; |
| 589 | } |
drh | 4a32431 | 2001-12-21 14:30:42 +0000 | [diff] [blame] | 590 | } |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 591 | } |
drh | 9e9374b | 2019-05-08 19:06:59 +0000 | [diff] [blame] | 592 | if( reg==0 ) aToOpen[nAllIdx+1] = 0; |
| 593 | aRegIdx[nAllIdx] = reg; |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 594 | } |
drh | 9e9374b | 2019-05-08 19:06:59 +0000 | [diff] [blame] | 595 | aRegIdx[nAllIdx] = ++pParse->nMem; /* Register storing the table record */ |
dan | 2c6fec2 | 2017-01-11 19:03:08 +0000 | [diff] [blame] | 596 | if( bReplace ){ |
| 597 | /* If REPLACE conflict resolution might be invoked, open cursors on all |
| 598 | ** indexes in case they are needed to delete records. */ |
| 599 | memset(aToOpen, 1, nIdx+1); |
| 600 | } |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 601 | |
drh | 4794f73 | 2004-11-05 17:17:50 +0000 | [diff] [blame] | 602 | if( pParse->nested==0 ) sqlite3VdbeCountChanges(v); |
drh | ac9151d | 2018-04-20 19:32:35 +0000 | [diff] [blame] | 603 | sqlite3BeginWriteOperation(pParse, pTrigger || hasFK, iDb); |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 604 | |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 605 | /* Allocate required registers. */ |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 606 | if( !IsVirtual(pTab) ){ |
drh | 9e9374b | 2019-05-08 19:06:59 +0000 | [diff] [blame] | 607 | /* For now, regRowSet and aRegIdx[nAllIdx] share the same register. |
| 608 | ** If regRowSet turns out to be needed, then aRegIdx[nAllIdx] will be |
| 609 | ** reallocated. aRegIdx[nAllIdx] is the register in which the main |
| 610 | ** table record is written. regRowSet holds the RowSet for the |
| 611 | ** two-pass update algorithm. */ |
| 612 | assert( aRegIdx[nAllIdx]==pParse->nMem ); |
| 613 | regRowSet = aRegIdx[nAllIdx]; |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 614 | regOldRowid = regNewRowid = ++pParse->nMem; |
| 615 | if( chngPk || pTrigger || hasFK ){ |
| 616 | regOld = pParse->nMem + 1; |
| 617 | pParse->nMem += pTab->nCol; |
| 618 | } |
| 619 | if( chngKey || pTrigger || hasFK ){ |
| 620 | regNewRowid = ++pParse->nMem; |
| 621 | } |
| 622 | regNew = pParse->nMem + 1; |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 623 | pParse->nMem += pTab->nCol; |
| 624 | } |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 625 | |
| 626 | /* Start the view context. */ |
danielk1977 | 4273dea | 2006-06-17 06:31:18 +0000 | [diff] [blame] | 627 | if( isView ){ |
| 628 | sqlite3AuthContextPush(pParse, &sContext, pTab->zName); |
| 629 | } |
| 630 | |
drh | 9d2985c | 2005-09-08 01:58:42 +0000 | [diff] [blame] | 631 | /* If we are trying to update a view, realize that view into |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 632 | ** an ephemeral table. |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 633 | */ |
shane | fa4e62f | 2008-09-01 21:59:42 +0000 | [diff] [blame] | 634 | #if !defined(SQLITE_OMIT_VIEW) && !defined(SQLITE_OMIT_TRIGGER) |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 635 | if( nChangeFrom==0 && isView ){ |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 636 | sqlite3MaterializeView(pParse, pTab, |
drh | 8c0833f | 2017-11-14 23:48:23 +0000 | [diff] [blame] | 637 | pWhere, pOrderBy, pLimit, iDataCur |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 638 | ); |
| 639 | pOrderBy = 0; |
drh | 8c0833f | 2017-11-14 23:48:23 +0000 | [diff] [blame] | 640 | pLimit = 0; |
drh | 0f35a6b | 2008-02-12 16:52:14 +0000 | [diff] [blame] | 641 | } |
shane | fa4e62f | 2008-09-01 21:59:42 +0000 | [diff] [blame] | 642 | #endif |
drh | 1013c93 | 2008-01-06 00:25:21 +0000 | [diff] [blame] | 643 | |
drh | 0f35a6b | 2008-02-12 16:52:14 +0000 | [diff] [blame] | 644 | /* Resolve the column names in all the expressions in the |
| 645 | ** WHERE clause. |
| 646 | */ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 647 | if( nChangeFrom==0 && sqlite3ResolveExprNames(&sNC, pWhere) ){ |
drh | 0f35a6b | 2008-02-12 16:52:14 +0000 | [diff] [blame] | 648 | goto update_cleanup; |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 649 | } |
| 650 | |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 651 | #ifndef SQLITE_OMIT_VIRTUALTABLE |
| 652 | /* Virtual tables must be handled separately */ |
| 653 | if( IsVirtual(pTab) ){ |
| 654 | updateVirtualTable(pParse, pTabList, pTab, pChanges, pRowidExpr, aXRef, |
| 655 | pWhere, onError); |
| 656 | goto update_cleanup; |
| 657 | } |
| 658 | #endif |
| 659 | |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 660 | /* Jump to labelBreak to abandon further processing of this UPDATE */ |
drh | ec4ccdb | 2018-12-29 02:26:59 +0000 | [diff] [blame] | 661 | labelContinue = labelBreak = sqlite3VdbeMakeLabel(pParse); |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 662 | |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 663 | /* Not an UPSERT. Normal processing. Begin by |
| 664 | ** initialize the count of updated rows */ |
| 665 | if( (db->flags&SQLITE_CountRows)!=0 |
| 666 | && !pParse->pTriggerTab |
| 667 | && !pParse->nested |
drh | d086aa0 | 2021-01-29 21:31:59 +0000 | [diff] [blame] | 668 | && !pParse->bReturning |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 669 | && pUpsert==0 |
| 670 | ){ |
| 671 | regRowCount = ++pParse->nMem; |
| 672 | sqlite3VdbeAddOp2(v, OP_Integer, 0, regRowCount); |
| 673 | } |
| 674 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 675 | if( nChangeFrom==0 && HasRowid(pTab) ){ |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 676 | sqlite3VdbeAddOp3(v, OP_Null, 0, regRowSet, regOldRowid); |
drh | ea24872 | 2020-11-14 20:03:34 +0000 | [diff] [blame] | 677 | iEph = pParse->nTab++; |
drh | 32881be | 2020-11-17 21:26:13 +0000 | [diff] [blame] | 678 | addrOpen = sqlite3VdbeAddOp3(v, OP_OpenEphemeral, iEph, 0, regRowSet); |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 679 | }else{ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 680 | assert( pPk!=0 || HasRowid(pTab) ); |
| 681 | nPk = pPk ? pPk->nKeyCol : 0; |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 682 | iPk = pParse->nMem+1; |
| 683 | pParse->nMem += nPk; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 684 | pParse->nMem += nChangeFrom; |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 685 | regKey = ++pParse->nMem; |
| 686 | if( pUpsert==0 ){ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 687 | int nEphCol = nPk + nChangeFrom + (isView ? pTab->nCol : 0); |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 688 | iEph = pParse->nTab++; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 689 | if( pPk ) sqlite3VdbeAddOp3(v, OP_Null, 0, iPk, iPk+nPk-1); |
| 690 | addrOpen = sqlite3VdbeAddOp2(v, OP_OpenEphemeral, iEph, nEphCol); |
dan | be952c1 | 2020-07-13 20:10:29 +0000 | [diff] [blame] | 691 | if( pPk ){ |
| 692 | KeyInfo *pKeyInfo = sqlite3KeyInfoOfIndex(pParse, pPk); |
| 693 | if( pKeyInfo ){ |
| 694 | pKeyInfo->nAllField = nEphCol; |
| 695 | sqlite3VdbeAppendP4(v, pKeyInfo, P4_KEYINFO); |
| 696 | } |
| 697 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 698 | if( nChangeFrom ){ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 699 | updateFromSelect( |
dan | f2972b6 | 2020-04-29 20:11:01 +0000 | [diff] [blame] | 700 | pParse, iEph, pPk, pChanges, pTabList, pWhere, pOrderBy, pLimit |
| 701 | ); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 702 | #ifndef SQLITE_OMIT_SUBQUERY |
| 703 | if( isView ) iDataCur = iEph; |
| 704 | #endif |
| 705 | } |
drh | 5deb181 | 2018-04-20 16:27:57 +0000 | [diff] [blame] | 706 | } |
| 707 | } |
| 708 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 709 | if( nChangeFrom ){ |
| 710 | sqlite3MultiWrite(pParse); |
| 711 | eOnePass = ONEPASS_OFF; |
dan | 7465787 | 2020-05-01 18:43:49 +0000 | [diff] [blame] | 712 | nKey = nPk; |
| 713 | regKey = iPk; |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 714 | }else{ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 715 | if( pUpsert ){ |
| 716 | /* If this is an UPSERT, then all cursors have already been opened by |
| 717 | ** the outer INSERT and the data cursor should be pointing at the row |
| 718 | ** that is to be updated. So bypass the code that searches for the |
| 719 | ** row(s) to be updated. |
| 720 | */ |
| 721 | pWInfo = 0; |
| 722 | eOnePass = ONEPASS_SINGLE; |
| 723 | sqlite3ExprIfFalse(pParse, pWhere, labelBreak, SQLITE_JUMPIFNULL); |
| 724 | bFinishSeek = 0; |
| 725 | }else{ |
| 726 | /* Begin the database scan. |
| 727 | ** |
| 728 | ** Do not consider a single-pass strategy for a multi-row update if |
| 729 | ** there are any triggers or foreign keys to process, or rows may |
| 730 | ** be deleted as a result of REPLACE conflict handling. Any of these |
| 731 | ** things might disturb a cursor being used to scan through the table |
| 732 | ** or index, causing a single-pass approach to malfunction. */ |
drh | 68c0c71 | 2020-08-14 20:04:26 +0000 | [diff] [blame] | 733 | flags = WHERE_ONEPASS_DESIRED; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 734 | if( !pParse->nested && !pTrigger && !hasFK && !chngKey && !bReplace ){ |
| 735 | flags |= WHERE_ONEPASS_MULTIROW; |
| 736 | } |
drh | 895bab3 | 2022-01-27 16:14:50 +0000 | [diff] [blame] | 737 | pWInfo = sqlite3WhereBegin(pParse, pTabList, pWhere,0,0,0,flags,iIdxCur); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 738 | if( pWInfo==0 ) goto update_cleanup; |
| 739 | |
| 740 | /* A one-pass strategy that might update more than one row may not |
| 741 | ** be used if any column of the index used for the scan is being |
| 742 | ** updated. Otherwise, if there is an index on "b", statements like |
| 743 | ** the following could create an infinite loop: |
| 744 | ** |
| 745 | ** UPDATE t1 SET b=b+1 WHERE b>? |
| 746 | ** |
| 747 | ** Fall back to ONEPASS_OFF if where.c has selected a ONEPASS_MULTI |
| 748 | ** strategy that uses an index for which one or more columns are being |
| 749 | ** updated. */ |
| 750 | eOnePass = sqlite3WhereOkOnePass(pWInfo, aiCurOnePass); |
| 751 | bFinishSeek = sqlite3WhereUsesDeferredSeek(pWInfo); |
| 752 | if( eOnePass!=ONEPASS_SINGLE ){ |
| 753 | sqlite3MultiWrite(pParse); |
| 754 | if( eOnePass==ONEPASS_MULTI ){ |
| 755 | int iCur = aiCurOnePass[1]; |
| 756 | if( iCur>=0 && iCur!=iDataCur && aToOpen[iCur-iBaseCur] ){ |
| 757 | eOnePass = ONEPASS_OFF; |
| 758 | } |
| 759 | assert( iCur!=iDataCur || !HasRowid(pTab) ); |
drh | ac9151d | 2018-04-20 19:32:35 +0000 | [diff] [blame] | 760 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 761 | } |
| 762 | } |
| 763 | |
| 764 | if( HasRowid(pTab) ){ |
| 765 | /* Read the rowid of the current row of the WHERE scan. In ONEPASS_OFF |
| 766 | ** mode, write the rowid into the FIFO. In either of the one-pass modes, |
| 767 | ** leave it in register regOldRowid. */ |
| 768 | sqlite3VdbeAddOp2(v, OP_Rowid, iDataCur, regOldRowid); |
| 769 | if( eOnePass==ONEPASS_OFF ){ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 770 | aRegIdx[nAllIdx] = ++pParse->nMem; |
drh | ea24872 | 2020-11-14 20:03:34 +0000 | [diff] [blame] | 771 | sqlite3VdbeAddOp3(v, OP_Insert, iEph, regRowSet, regOldRowid); |
| 772 | }else{ |
drh | 32881be | 2020-11-17 21:26:13 +0000 | [diff] [blame] | 773 | if( ALWAYS(addrOpen) ) sqlite3VdbeChangeToNoop(v, addrOpen); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 774 | } |
| 775 | }else{ |
| 776 | /* Read the PK of the current row into an array of registers. In |
| 777 | ** ONEPASS_OFF mode, serialize the array into a record and store it in |
| 778 | ** the ephemeral table. Or, in ONEPASS_SINGLE or MULTI mode, change |
| 779 | ** the OP_OpenEphemeral instruction to a Noop (the ephemeral table |
| 780 | ** is not required) and leave the PK fields in the array of registers. */ |
| 781 | for(i=0; i<nPk; i++){ |
| 782 | assert( pPk->aiColumn[i]>=0 ); |
| 783 | sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, |
| 784 | pPk->aiColumn[i], iPk+i); |
| 785 | } |
| 786 | if( eOnePass ){ |
| 787 | if( addrOpen ) sqlite3VdbeChangeToNoop(v, addrOpen); |
| 788 | nKey = nPk; |
| 789 | regKey = iPk; |
| 790 | }else{ |
| 791 | sqlite3VdbeAddOp4(v, OP_MakeRecord, iPk, nPk, regKey, |
| 792 | sqlite3IndexAffinityStr(db, pPk), nPk); |
| 793 | sqlite3VdbeAddOp4Int(v, OP_IdxInsert, iEph, regKey, iPk, nPk); |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 794 | } |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 795 | } |
| 796 | } |
| 797 | |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 798 | if( pUpsert==0 ){ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 799 | if( nChangeFrom==0 && eOnePass!=ONEPASS_MULTI ){ |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 800 | sqlite3WhereEnd(pWInfo); |
| 801 | } |
| 802 | |
| 803 | if( !isView ){ |
| 804 | int addrOnce = 0; |
| 805 | |
| 806 | /* Open every index that needs updating. */ |
| 807 | if( eOnePass!=ONEPASS_OFF ){ |
| 808 | if( aiCurOnePass[0]>=0 ) aToOpen[aiCurOnePass[0]-iBaseCur] = 0; |
| 809 | if( aiCurOnePass[1]>=0 ) aToOpen[aiCurOnePass[1]-iBaseCur] = 0; |
| 810 | } |
| 811 | |
| 812 | if( eOnePass==ONEPASS_MULTI && (nIdx-(aiCurOnePass[1]>=0))>0 ){ |
| 813 | addrOnce = sqlite3VdbeAddOp0(v, OP_Once); VdbeCoverage(v); |
| 814 | } |
| 815 | sqlite3OpenTableAndIndices(pParse, pTab, OP_OpenWrite, 0, iBaseCur, |
| 816 | aToOpen, 0, 0); |
drh | dc4f6fc | 2020-02-07 19:44:13 +0000 | [diff] [blame] | 817 | if( addrOnce ){ |
| 818 | sqlite3VdbeJumpHereOrPopInst(v, addrOnce); |
| 819 | } |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 820 | } |
| 821 | |
| 822 | /* Top of the update loop */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 823 | if( eOnePass!=ONEPASS_OFF ){ |
drh | 6e5020e | 2021-04-07 15:45:01 +0000 | [diff] [blame] | 824 | if( aiCurOnePass[0]!=iDataCur |
| 825 | && aiCurOnePass[1]!=iDataCur |
| 826 | #ifdef SQLITE_ALLOW_ROWID_IN_VIEW |
| 827 | && !isView |
| 828 | #endif |
| 829 | ){ |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 830 | assert( pPk ); |
| 831 | sqlite3VdbeAddOp4Int(v, OP_NotFound, iDataCur, labelBreak, regKey,nKey); |
drh | 6ccbd27 | 2018-07-10 17:10:44 +0000 | [diff] [blame] | 832 | VdbeCoverage(v); |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 833 | } |
drh | 4b8bd84 | 2018-04-20 20:09:22 +0000 | [diff] [blame] | 834 | if( eOnePass!=ONEPASS_SINGLE ){ |
drh | ec4ccdb | 2018-12-29 02:26:59 +0000 | [diff] [blame] | 835 | labelContinue = sqlite3VdbeMakeLabel(pParse); |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 836 | } |
| 837 | sqlite3VdbeAddOp2(v, OP_IsNull, pPk ? regKey : regOldRowid, labelBreak); |
| 838 | VdbeCoverageIf(v, pPk==0); |
| 839 | VdbeCoverageIf(v, pPk!=0); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 840 | }else if( pPk || nChangeFrom ){ |
drh | ec4ccdb | 2018-12-29 02:26:59 +0000 | [diff] [blame] | 841 | labelContinue = sqlite3VdbeMakeLabel(pParse); |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 842 | sqlite3VdbeAddOp2(v, OP_Rewind, iEph, labelBreak); VdbeCoverage(v); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 843 | addrTop = sqlite3VdbeCurrentAddr(v); |
| 844 | if( nChangeFrom ){ |
| 845 | if( !isView ){ |
| 846 | if( pPk ){ |
| 847 | for(i=0; i<nPk; i++){ |
| 848 | sqlite3VdbeAddOp3(v, OP_Column, iEph, i, iPk+i); |
| 849 | } |
| 850 | sqlite3VdbeAddOp4Int( |
| 851 | v, OP_NotFound, iDataCur, labelContinue, iPk, nPk |
drh | 6265c47 | 2020-07-16 00:45:25 +0000 | [diff] [blame] | 852 | ); VdbeCoverage(v); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 853 | }else{ |
| 854 | sqlite3VdbeAddOp2(v, OP_Rowid, iEph, regOldRowid); |
| 855 | sqlite3VdbeAddOp3( |
| 856 | v, OP_NotExists, iDataCur, labelContinue, regOldRowid |
drh | 6265c47 | 2020-07-16 00:45:25 +0000 | [diff] [blame] | 857 | ); VdbeCoverage(v); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 858 | } |
| 859 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 860 | }else{ |
| 861 | sqlite3VdbeAddOp2(v, OP_RowData, iEph, regKey); |
| 862 | sqlite3VdbeAddOp4Int(v, OP_NotFound, iDataCur, labelContinue, regKey,0); |
| 863 | VdbeCoverage(v); |
| 864 | } |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 865 | }else{ |
drh | ea24872 | 2020-11-14 20:03:34 +0000 | [diff] [blame] | 866 | sqlite3VdbeAddOp2(v, OP_Rewind, iEph, labelBreak); VdbeCoverage(v); |
| 867 | labelContinue = sqlite3VdbeMakeLabel(pParse); |
| 868 | addrTop = sqlite3VdbeAddOp2(v, OP_Rowid, iEph, regOldRowid); |
drh | fb2213e | 2018-04-20 15:56:24 +0000 | [diff] [blame] | 869 | VdbeCoverage(v); |
| 870 | sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue, regOldRowid); |
| 871 | VdbeCoverage(v); |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 872 | } |
drh | 08c88eb | 2008-04-10 13:33:18 +0000 | [diff] [blame] | 873 | } |
danielk1977 | e448dc4 | 2008-01-02 11:50:51 +0000 | [diff] [blame] | 874 | |
drh | 7007640 | 2018-04-12 19:51:22 +0000 | [diff] [blame] | 875 | /* If the rowid value will change, set register regNewRowid to |
| 876 | ** contain the new value. If the rowid is not being modified, |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 877 | ** then regNewRowid is the same register as regOldRowid, which is |
| 878 | ** already populated. */ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 879 | assert( chngKey || pTrigger || hasFK || regOldRowid==regNewRowid ); |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 880 | if( chngRowid ){ |
dan | 7465787 | 2020-05-01 18:43:49 +0000 | [diff] [blame] | 881 | assert( iRowidExpr>=0 ); |
| 882 | if( nChangeFrom==0 ){ |
| 883 | sqlite3ExprCode(pParse, pRowidExpr, regNewRowid); |
| 884 | }else{ |
| 885 | sqlite3VdbeAddOp3(v, OP_Column, iEph, iRowidExpr, regNewRowid); |
| 886 | } |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 887 | sqlite3VdbeAddOp1(v, OP_MustBeInt, regNewRowid); VdbeCoverage(v); |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 888 | } |
| 889 | |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 890 | /* Compute the old pre-UPDATE content of the row being changed, if that |
| 891 | ** information is needed */ |
| 892 | if( chngPk || hasFK || pTrigger ){ |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 893 | u32 oldmask = (hasFK ? sqlite3FkOldmask(pParse, pTab) : 0); |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 894 | oldmask |= sqlite3TriggerColmask(pParse, |
| 895 | pTrigger, pChanges, 0, TRIGGER_BEFORE|TRIGGER_AFTER, pTab, onError |
| 896 | ); |
drh | ff37c40 | 2019-10-31 20:54:20 +0000 | [diff] [blame] | 897 | for(i=0; i<pTab->nCol; i++){ |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 898 | u32 colFlags = pTab->aCol[i].colFlags; |
drh | ff37c40 | 2019-10-31 20:54:20 +0000 | [diff] [blame] | 899 | k = sqlite3TableColumnToStorage(pTab, i) + regOld; |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 900 | if( oldmask==0xffffffff |
drh | 693e671 | 2014-01-24 22:58:00 +0000 | [diff] [blame] | 901 | || (i<32 && (oldmask & MASKBIT32(i))!=0) |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 902 | || (colFlags & COLFLAG_PRIMKEY)!=0 |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 903 | ){ |
drh | 693e671 | 2014-01-24 22:58:00 +0000 | [diff] [blame] | 904 | testcase( oldmask!=0xffffffff && i==31 ); |
drh | 6df9c4b | 2019-10-18 12:52:08 +0000 | [diff] [blame] | 905 | sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k); |
danielk1977 | c3f9bad | 2002-05-15 08:30:12 +0000 | [diff] [blame] | 906 | }else{ |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 907 | sqlite3VdbeAddOp2(v, OP_Null, 0, k); |
danielk1977 | c3f9bad | 2002-05-15 08:30:12 +0000 | [diff] [blame] | 908 | } |
| 909 | } |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 910 | if( chngRowid==0 && pPk==0 ){ |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 911 | sqlite3VdbeAddOp2(v, OP_Copy, regOldRowid, regNewRowid); |
| 912 | } |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 913 | } |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 914 | |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 915 | /* Populate the array of registers beginning at regNew with the new |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 916 | ** row data. This array is used to check constants, create the new |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 917 | ** table and index records, and as the values for any new.* references |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 918 | ** made by triggers. |
| 919 | ** |
| 920 | ** If there are one or more BEFORE triggers, then do not populate the |
| 921 | ** registers associated with columns that are (a) not modified by |
| 922 | ** this UPDATE statement and (b) not accessed by new.* references. The |
| 923 | ** values for registers not modified by the UPDATE must be reloaded from |
| 924 | ** the database after the BEFORE triggers are fired anyway (as the trigger |
| 925 | ** may have modified them). So not loading those that are not going to |
| 926 | ** be used eliminates some redundant opcodes. |
| 927 | */ |
| 928 | newmask = sqlite3TriggerColmask( |
| 929 | pParse, pTrigger, pChanges, 1, TRIGGER_BEFORE, pTab, onError |
| 930 | ); |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 931 | for(i=0, k=regNew; i<pTab->nCol; i++, k++){ |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 932 | if( i==pTab->iPKey ){ |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 933 | sqlite3VdbeAddOp2(v, OP_Null, 0, k); |
drh | c143114 | 2019-10-17 17:54:05 +0000 | [diff] [blame] | 934 | }else if( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)!=0 ){ |
| 935 | if( pTab->aCol[i].colFlags & COLFLAG_VIRTUAL ) k--; |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 936 | }else{ |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 937 | j = aXRef[i]; |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 938 | if( j>=0 ){ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 939 | if( nChangeFrom ){ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 940 | int nOff = (isView ? pTab->nCol : nPk); |
drh | a7c7400 | 2020-07-18 18:44:59 +0000 | [diff] [blame] | 941 | assert( eOnePass==ONEPASS_OFF ); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 942 | sqlite3VdbeAddOp3(v, OP_Column, iEph, nOff+j, k); |
| 943 | }else{ |
| 944 | sqlite3ExprCode(pParse, pChanges->a[j].pExpr, k); |
| 945 | } |
drh | 693e671 | 2014-01-24 22:58:00 +0000 | [diff] [blame] | 946 | }else if( 0==(tmask&TRIGGER_BEFORE) || i>31 || (newmask & MASKBIT32(i)) ){ |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 947 | /* This branch loads the value of a column that will not be changed |
| 948 | ** into a register. This is done if there are no BEFORE triggers, or |
| 949 | ** if there are one or more BEFORE triggers that use this value via |
| 950 | ** a new.* reference in a trigger program. |
| 951 | */ |
drh | fd7c991 | 2009-12-01 22:09:25 +0000 | [diff] [blame] | 952 | testcase( i==31 ); |
| 953 | testcase( i==32 ); |
drh | 6df9c4b | 2019-10-18 12:52:08 +0000 | [diff] [blame] | 954 | sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k); |
drh | be3da24 | 2019-12-29 00:52:41 +0000 | [diff] [blame] | 955 | bFinishSeek = 0; |
drh | edfac34 | 2013-12-19 02:56:01 +0000 | [diff] [blame] | 956 | }else{ |
drh | 8a53ce2 | 2019-10-17 15:59:03 +0000 | [diff] [blame] | 957 | sqlite3VdbeAddOp2(v, OP_Null, 0, k); |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 958 | } |
| 959 | } |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 960 | } |
drh | c143114 | 2019-10-17 17:54:05 +0000 | [diff] [blame] | 961 | #ifndef SQLITE_OMIT_GENERATED_COLUMNS |
drh | 427b96a | 2019-10-22 13:01:24 +0000 | [diff] [blame] | 962 | if( pTab->tabFlags & TF_HasGenerated ){ |
| 963 | testcase( pTab->tabFlags & TF_HasVirtual ); |
| 964 | testcase( pTab->tabFlags & TF_HasStored ); |
drh | dd6cc9b | 2019-10-19 18:47:27 +0000 | [diff] [blame] | 965 | sqlite3ComputeGeneratedColumns(pParse, regNew, pTab); |
drh | c143114 | 2019-10-17 17:54:05 +0000 | [diff] [blame] | 966 | } |
| 967 | #endif |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 968 | |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 969 | /* Fire any BEFORE UPDATE triggers. This happens before constraints are |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 970 | ** verified. One could argue that this is wrong. |
| 971 | */ |
| 972 | if( tmask&TRIGGER_BEFORE ){ |
drh | 57bf4a8 | 2014-02-17 14:59:22 +0000 | [diff] [blame] | 973 | sqlite3TableAffinity(v, pTab, regNew); |
dan | 2832ad4 | 2009-08-31 15:27:27 +0000 | [diff] [blame] | 974 | sqlite3CodeRowTrigger(pParse, pTrigger, TK_UPDATE, pChanges, |
drh | c3e356f | 2013-11-04 18:34:46 +0000 | [diff] [blame] | 975 | TRIGGER_BEFORE, pTab, regOldRowid, onError, labelContinue); |
dan | ae0931e | 2009-09-09 15:06:10 +0000 | [diff] [blame] | 976 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 977 | if( !isView ){ |
| 978 | /* The row-trigger may have deleted the row being updated. In this |
| 979 | ** case, jump to the next row. No updates or AFTER triggers are |
| 980 | ** required. This behavior - what happens when the row being updated |
| 981 | ** is deleted or renamed by a BEFORE trigger - is left undefined in the |
| 982 | ** documentation. |
| 983 | */ |
| 984 | if( pPk ){ |
dan | 7465787 | 2020-05-01 18:43:49 +0000 | [diff] [blame] | 985 | sqlite3VdbeAddOp4Int(v, OP_NotFound,iDataCur,labelContinue,regKey,nKey); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 986 | VdbeCoverage(v); |
| 987 | }else{ |
| 988 | sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue,regOldRowid); |
| 989 | VdbeCoverage(v); |
dan | bb5f168 | 2009-11-27 12:12:34 +0000 | [diff] [blame] | 990 | } |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 991 | |
| 992 | /* After-BEFORE-trigger-reload-loop: |
| 993 | ** If it did not delete it, the BEFORE trigger may still have modified |
| 994 | ** some of the columns of the row being updated. Load the values for |
| 995 | ** all columns not modified by the update statement into their registers |
| 996 | ** in case this has happened. Only unmodified columns are reloaded. |
| 997 | ** The values computed for modified columns use the values before the |
| 998 | ** BEFORE trigger runs. See test case trigger1-18.0 (added 2018-04-26) |
| 999 | ** for an example. |
| 1000 | */ |
| 1001 | for(i=0, k=regNew; i<pTab->nCol; i++, k++){ |
| 1002 | if( pTab->aCol[i].colFlags & COLFLAG_GENERATED ){ |
| 1003 | if( pTab->aCol[i].colFlags & COLFLAG_VIRTUAL ) k--; |
| 1004 | }else if( aXRef[i]<0 && i!=pTab->iPKey ){ |
| 1005 | sqlite3ExprCodeGetColumnOfTable(v, pTab, iDataCur, i, k); |
| 1006 | } |
| 1007 | } |
drh | c143114 | 2019-10-17 17:54:05 +0000 | [diff] [blame] | 1008 | #ifndef SQLITE_OMIT_GENERATED_COLUMNS |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1009 | if( pTab->tabFlags & TF_HasGenerated ){ |
| 1010 | testcase( pTab->tabFlags & TF_HasVirtual ); |
| 1011 | testcase( pTab->tabFlags & TF_HasStored ); |
| 1012 | sqlite3ComputeGeneratedColumns(pParse, regNew, pTab); |
| 1013 | } |
drh | c143114 | 2019-10-17 17:54:05 +0000 | [diff] [blame] | 1014 | #endif |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1015 | } |
dan | 2832ad4 | 2009-08-31 15:27:27 +0000 | [diff] [blame] | 1016 | } |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 1017 | |
| 1018 | if( !isView ){ |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 1019 | /* Do constraint checks. */ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 1020 | assert( regOldRowid>0 ); |
| 1021 | sqlite3GenerateConstraintChecks(pParse, pTab, aRegIdx, iDataCur, iIdxCur, |
drh | bdb0022 | 2016-02-10 16:03:20 +0000 | [diff] [blame] | 1022 | regNewRowid, regOldRowid, chngKey, onError, labelContinue, &bReplace, |
drh | 788d55a | 2018-04-13 01:15:09 +0000 | [diff] [blame] | 1023 | aXRef, 0); |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 1024 | |
dan | f66bfcb | 2019-08-31 17:14:35 +0000 | [diff] [blame] | 1025 | /* If REPLACE conflict handling may have been used, or if the PK of the |
| 1026 | ** row is changing, then the GenerateConstraintChecks() above may have |
| 1027 | ** moved cursor iDataCur. Reseek it. */ |
| 1028 | if( bReplace || chngKey ){ |
| 1029 | if( pPk ){ |
| 1030 | sqlite3VdbeAddOp4Int(v, OP_NotFound,iDataCur,labelContinue,regKey,nKey); |
| 1031 | }else{ |
| 1032 | sqlite3VdbeAddOp3(v, OP_NotExists, iDataCur, labelContinue,regOldRowid); |
| 1033 | } |
drh | 10c5341 | 2022-06-15 12:32:27 +0000 | [diff] [blame] | 1034 | VdbeCoverage(v); |
dan | f66bfcb | 2019-08-31 17:14:35 +0000 | [diff] [blame] | 1035 | } |
| 1036 | |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 1037 | /* Do FK constraint checks. */ |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1038 | if( hasFK ){ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 1039 | sqlite3FkCheck(pParse, pTab, regOldRowid, 0, aXRef, chngKey); |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1040 | } |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 1041 | |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 1042 | /* Delete the index entries associated with the current record. */ |
dan | f0ee1d3 | 2015-09-12 19:26:11 +0000 | [diff] [blame] | 1043 | sqlite3GenerateRowIndexDelete(pParse, pTab, iDataCur, iIdxCur, aRegIdx, -1); |
drh | 9b1c62d | 2011-03-30 21:04:43 +0000 | [diff] [blame] | 1044 | |
drh | be3da24 | 2019-12-29 00:52:41 +0000 | [diff] [blame] | 1045 | /* We must run the OP_FinishSeek opcode to resolve a prior |
| 1046 | ** OP_DeferredSeek if there is any possibility that there have been |
| 1047 | ** no OP_Column opcodes since the OP_DeferredSeek was issued. But |
| 1048 | ** we want to avoid the OP_FinishSeek if possible, as running it |
| 1049 | ** costs CPU cycles. */ |
| 1050 | if( bFinishSeek ){ |
| 1051 | sqlite3VdbeAddOp1(v, OP_FinishSeek, iDataCur); |
drh | d3ee3ad | 2019-12-26 23:16:18 +0000 | [diff] [blame] | 1052 | } |
drh | d3ee3ad | 2019-12-26 23:16:18 +0000 | [diff] [blame] | 1053 | |
dan | 46c47d4 | 2011-03-01 18:42:07 +0000 | [diff] [blame] | 1054 | /* If changing the rowid value, or if there are foreign key constraints |
| 1055 | ** to process, delete the old record. Otherwise, add a noop OP_Delete |
| 1056 | ** to invoke the pre-update hook. |
dan | 37db03b | 2011-03-16 19:59:18 +0000 | [diff] [blame] | 1057 | ** |
| 1058 | ** That (regNew==regnewRowid+1) is true is also important for the |
drh | 9b1c62d | 2011-03-30 21:04:43 +0000 | [diff] [blame] | 1059 | ** pre-update hook. If the caller invokes preupdate_new(), the returned |
dan | 37db03b | 2011-03-16 19:59:18 +0000 | [diff] [blame] | 1060 | ** value is copied from memory cell (regNewRowid+1+iCol), where iCol |
| 1061 | ** is the column index supplied by the user. |
dan | 46c47d4 | 2011-03-01 18:42:07 +0000 | [diff] [blame] | 1062 | */ |
dan | 37db03b | 2011-03-16 19:59:18 +0000 | [diff] [blame] | 1063 | assert( regNew==regNewRowid+1 ); |
drh | 74c3302 | 2016-03-30 12:56:55 +0000 | [diff] [blame] | 1064 | #ifdef SQLITE_ENABLE_PREUPDATE_HOOK |
drh | cbf1b8e | 2013-11-11 22:55:26 +0000 | [diff] [blame] | 1065 | sqlite3VdbeAddOp3(v, OP_Delete, iDataCur, |
dan | 940b5ea | 2017-04-11 19:58:55 +0000 | [diff] [blame] | 1066 | OPFLAG_ISUPDATE | ((hasFK>1 || chngKey) ? 0 : OPFLAG_ISNOOP), |
dan | 46c47d4 | 2011-03-01 18:42:07 +0000 | [diff] [blame] | 1067 | regNewRowid |
| 1068 | ); |
dan | e206ea7 | 2017-01-11 20:10:30 +0000 | [diff] [blame] | 1069 | if( eOnePass==ONEPASS_MULTI ){ |
| 1070 | assert( hasFK==0 && chngKey==0 ); |
| 1071 | sqlite3VdbeChangeP5(v, OPFLAG_SAVEPOSITION); |
| 1072 | } |
dan | 46c47d4 | 2011-03-01 18:42:07 +0000 | [diff] [blame] | 1073 | if( !pParse->nested ){ |
drh | f14b7fb | 2016-12-07 21:35:55 +0000 | [diff] [blame] | 1074 | sqlite3VdbeAppendP4(v, pTab, P4_TABLE); |
drh | 5cf590c | 2003-04-24 01:45:04 +0000 | [diff] [blame] | 1075 | } |
drh | 74c3302 | 2016-03-30 12:56:55 +0000 | [diff] [blame] | 1076 | #else |
dan | 940b5ea | 2017-04-11 19:58:55 +0000 | [diff] [blame] | 1077 | if( hasFK>1 || chngKey ){ |
drh | 74c3302 | 2016-03-30 12:56:55 +0000 | [diff] [blame] | 1078 | sqlite3VdbeAddOp2(v, OP_Delete, iDataCur, 0); |
| 1079 | } |
| 1080 | #endif |
dan | 9277efa | 2009-09-28 11:54:21 +0000 | [diff] [blame] | 1081 | |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1082 | if( hasFK ){ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 1083 | sqlite3FkCheck(pParse, pTab, 0, regNewRowid, aXRef, chngKey); |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1084 | } |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 1085 | |
dan | 76d462e | 2009-08-30 11:42:51 +0000 | [diff] [blame] | 1086 | /* Insert the new index entries and the new record. */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 1087 | sqlite3CompleteInsertion( |
| 1088 | pParse, pTab, iDataCur, iIdxCur, regNewRowid, aRegIdx, |
| 1089 | OPFLAG_ISUPDATE | (eOnePass==ONEPASS_MULTI ? OPFLAG_SAVEPOSITION : 0), |
| 1090 | 0, 0 |
| 1091 | ); |
dan | 1da40a3 | 2009-09-19 17:00:31 +0000 | [diff] [blame] | 1092 | |
| 1093 | /* Do any ON CASCADE, SET NULL or SET DEFAULT operations required to |
| 1094 | ** handle rows (possibly in other tables) that refer via a foreign key |
| 1095 | ** to the row just updated. */ |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1096 | if( hasFK ){ |
drh | f8ffb27 | 2013-11-01 17:08:56 +0000 | [diff] [blame] | 1097 | sqlite3FkActions(pParse, pTab, pChanges, regOldRowid, aXRef, chngKey); |
dan | e7a94d8 | 2009-10-01 16:09:04 +0000 | [diff] [blame] | 1098 | } |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 1099 | } |
| 1100 | |
drh | 0ca3e24 | 2002-01-29 23:07:02 +0000 | [diff] [blame] | 1101 | /* Increment the row counter |
drh | 1bee3d7 | 2001-10-15 00:44:35 +0000 | [diff] [blame] | 1102 | */ |
drh | 7963691 | 2018-04-19 23:52:39 +0000 | [diff] [blame] | 1103 | if( regRowCount ){ |
drh | 04adf41 | 2008-01-08 18:57:50 +0000 | [diff] [blame] | 1104 | sqlite3VdbeAddOp2(v, OP_AddImm, regRowCount, 1); |
drh | 1bee3d7 | 2001-10-15 00:44:35 +0000 | [diff] [blame] | 1105 | } |
| 1106 | |
dan | 165921a | 2009-08-28 18:53:45 +0000 | [diff] [blame] | 1107 | sqlite3CodeRowTrigger(pParse, pTrigger, TK_UPDATE, pChanges, |
drh | c3e356f | 2013-11-04 18:34:46 +0000 | [diff] [blame] | 1108 | TRIGGER_AFTER, pTab, regOldRowid, onError, labelContinue); |
danielk1977 | c3f9bad | 2002-05-15 08:30:12 +0000 | [diff] [blame] | 1109 | |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 1110 | /* Repeat the above with the next record to be updated, until |
| 1111 | ** all record selected by the WHERE clause have been updated. |
| 1112 | */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 1113 | if( eOnePass==ONEPASS_SINGLE ){ |
drh | 702ba9f | 2013-11-07 21:25:13 +0000 | [diff] [blame] | 1114 | /* Nothing to do at end-of-loop for a single-pass */ |
dan | f91c131 | 2017-01-10 20:04:38 +0000 | [diff] [blame] | 1115 | }else if( eOnePass==ONEPASS_MULTI ){ |
| 1116 | sqlite3VdbeResolveLabel(v, labelContinue); |
| 1117 | sqlite3WhereEnd(pWInfo); |
drh | 32881be | 2020-11-17 21:26:13 +0000 | [diff] [blame] | 1118 | }else{ |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 1119 | sqlite3VdbeResolveLabel(v, labelContinue); |
drh | 688852a | 2014-02-17 22:40:43 +0000 | [diff] [blame] | 1120 | sqlite3VdbeAddOp2(v, OP_Next, iEph, addrTop); VdbeCoverage(v); |
drh | ea22abe | 2013-10-25 19:17:17 +0000 | [diff] [blame] | 1121 | } |
| 1122 | sqlite3VdbeResolveLabel(v, labelBreak); |
danielk1977 | c3f9bad | 2002-05-15 08:30:12 +0000 | [diff] [blame] | 1123 | |
drh | 0b9f50d | 2009-06-23 20:28:53 +0000 | [diff] [blame] | 1124 | /* Update the sqlite_sequence table by storing the content of the |
| 1125 | ** maximum rowid counter values recorded while inserting into |
| 1126 | ** autoincrement tables. |
| 1127 | */ |
drh | f3d7bbb | 2018-04-21 03:06:29 +0000 | [diff] [blame] | 1128 | if( pParse->nested==0 && pParse->pTriggerTab==0 && pUpsert==0 ){ |
drh | 0b9f50d | 2009-06-23 20:28:53 +0000 | [diff] [blame] | 1129 | sqlite3AutoincrementEnd(pParse); |
| 1130 | } |
| 1131 | |
drh | 1bee3d7 | 2001-10-15 00:44:35 +0000 | [diff] [blame] | 1132 | /* |
drh | 7963691 | 2018-04-19 23:52:39 +0000 | [diff] [blame] | 1133 | ** Return the number of rows that were changed, if we are tracking |
| 1134 | ** that information. |
drh | 1bee3d7 | 2001-10-15 00:44:35 +0000 | [diff] [blame] | 1135 | */ |
drh | 7963691 | 2018-04-19 23:52:39 +0000 | [diff] [blame] | 1136 | if( regRowCount ){ |
drh | 3b26b2b | 2021-12-01 19:17:14 +0000 | [diff] [blame] | 1137 | sqlite3CodeChangeCount(v, regRowCount, "rows updated"); |
drh | 1bee3d7 | 2001-10-15 00:44:35 +0000 | [diff] [blame] | 1138 | } |
| 1139 | |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 1140 | update_cleanup: |
danielk1977 | 4adee20 | 2004-05-08 08:23:19 +0000 | [diff] [blame] | 1141 | sqlite3AuthContextPop(&sContext); |
drh | 6a53499 | 2013-11-16 20:13:39 +0000 | [diff] [blame] | 1142 | sqlite3DbFree(db, aXRef); /* Also frees aRegIdx[] and aToOpen[] */ |
drh | 633e6d5 | 2008-07-28 19:34:53 +0000 | [diff] [blame] | 1143 | sqlite3SrcListDelete(db, pTabList); |
| 1144 | sqlite3ExprListDelete(db, pChanges); |
| 1145 | sqlite3ExprDelete(db, pWhere); |
dan | 3b61ebb | 2017-11-10 20:13:14 +0000 | [diff] [blame] | 1146 | #if defined(SQLITE_ENABLE_UPDATE_DELETE_LIMIT) |
dan | b3c16b8 | 2017-11-09 19:53:06 +0000 | [diff] [blame] | 1147 | sqlite3ExprListDelete(db, pOrderBy); |
| 1148 | sqlite3ExprDelete(db, pLimit); |
dan | 3b61ebb | 2017-11-10 20:13:14 +0000 | [diff] [blame] | 1149 | #endif |
drh | cce7d17 | 2000-05-31 15:34:51 +0000 | [diff] [blame] | 1150 | return; |
| 1151 | } |
dan | 75cbd98 | 2009-09-21 16:06:03 +0000 | [diff] [blame] | 1152 | /* Make sure "isView" and other macros defined above are undefined. Otherwise |
peter.d.reid | 60ec914 | 2014-09-06 16:39:46 +0000 | [diff] [blame] | 1153 | ** they may interfere with compilation of other functions in this file |
dan | 75cbd98 | 2009-09-21 16:06:03 +0000 | [diff] [blame] | 1154 | ** (or in another file, if this file becomes part of the amalgamation). */ |
| 1155 | #ifdef isView |
| 1156 | #undef isView |
| 1157 | #endif |
| 1158 | #ifdef pTrigger |
| 1159 | #undef pTrigger |
| 1160 | #endif |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1161 | |
| 1162 | #ifndef SQLITE_OMIT_VIRTUALTABLE |
| 1163 | /* |
| 1164 | ** Generate code for an UPDATE of a virtual table. |
| 1165 | ** |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1166 | ** There are two possible strategies - the default and the special |
| 1167 | ** "onepass" strategy. Onepass is only used if the virtual table |
| 1168 | ** implementation indicates that pWhere may match at most one row. |
| 1169 | ** |
| 1170 | ** The default strategy is to create an ephemeral table that contains |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1171 | ** for each row to be changed: |
| 1172 | ** |
| 1173 | ** (A) The original rowid of that row. |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1174 | ** (B) The revised rowid for the row. |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1175 | ** (C) The content of every column in the row. |
| 1176 | ** |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1177 | ** Then loop through the contents of this ephemeral table executing a |
| 1178 | ** VUpdate for each row. When finished, drop the ephemeral table. |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1179 | ** |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1180 | ** The "onepass" strategy does not use an ephemeral table. Instead, it |
| 1181 | ** stores the same values (A, B and C above) in a register array and |
| 1182 | ** makes a single invocation of VUpdate. |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1183 | */ |
| 1184 | static void updateVirtualTable( |
| 1185 | Parse *pParse, /* The parsing context */ |
| 1186 | SrcList *pSrc, /* The virtual table to be modified */ |
| 1187 | Table *pTab, /* The virtual table */ |
| 1188 | ExprList *pChanges, /* The columns to change in the UPDATE statement */ |
| 1189 | Expr *pRowid, /* Expression used to recompute the rowid */ |
| 1190 | int *aXRef, /* Mapping from columns of pTab to entries in pChanges */ |
dan | b061d05 | 2011-04-25 18:49:57 +0000 | [diff] [blame] | 1191 | Expr *pWhere, /* WHERE clause of the UPDATE statement */ |
| 1192 | int onError /* ON CONFLICT strategy */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1193 | ){ |
| 1194 | Vdbe *v = pParse->pVdbe; /* Virtual machine under construction */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1195 | int ephemTab; /* Table holding the result of the SELECT */ |
| 1196 | int i; /* Loop counter */ |
drh | 1743575 | 2007-08-16 04:30:38 +0000 | [diff] [blame] | 1197 | sqlite3 *db = pParse->db; /* Database connection */ |
danielk1977 | 595a523 | 2009-07-24 17:58:53 +0000 | [diff] [blame] | 1198 | const char *pVTab = (const char*)sqlite3GetVTable(db, pTab); |
drh | a7c7400 | 2020-07-18 18:44:59 +0000 | [diff] [blame] | 1199 | WhereInfo *pWInfo = 0; |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1200 | int nArg = 2 + pTab->nCol; /* Number of arguments to VUpdate */ |
| 1201 | int regArg; /* First register in VUpdate arg array */ |
| 1202 | int regRec; /* Register in which to assemble record */ |
| 1203 | int regRowid; /* Register for ephem table rowid */ |
| 1204 | int iCsr = pSrc->a[0].iCursor; /* Cursor used for virtual table scan */ |
| 1205 | int aDummy[2]; /* Unused arg for sqlite3WhereOkOnePass() */ |
drh | 338e311 | 2018-05-24 23:51:57 +0000 | [diff] [blame] | 1206 | int eOnePass; /* True to use onepass strategy */ |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1207 | int addr; /* Address of OP_OpenEphemeral */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1208 | |
drh | 24cd20f | 2018-01-11 13:43:04 +0000 | [diff] [blame] | 1209 | /* Allocate nArg registers in which to gather the arguments for VUpdate. Then |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1210 | ** create and open the ephemeral table in which the records created from |
| 1211 | ** these arguments will be temporarily stored. */ |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1212 | assert( v ); |
| 1213 | ephemTab = pParse->nTab++; |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1214 | addr= sqlite3VdbeAddOp2(v, OP_OpenEphemeral, ephemTab, nArg); |
| 1215 | regArg = pParse->nMem + 1; |
| 1216 | pParse->nMem += nArg; |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1217 | if( pSrc->nSrc>1 ){ |
dan | 4d906f1 | 2020-10-12 14:29:11 +0000 | [diff] [blame] | 1218 | Index *pPk = 0; |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 1219 | Expr *pRow; |
| 1220 | ExprList *pList; |
dan | 4d906f1 | 2020-10-12 14:29:11 +0000 | [diff] [blame] | 1221 | if( HasRowid(pTab) ){ |
| 1222 | if( pRowid ){ |
| 1223 | pRow = sqlite3ExprDup(db, pRowid, 0); |
| 1224 | }else{ |
| 1225 | pRow = sqlite3PExpr(pParse, TK_ROW, 0, 0); |
| 1226 | } |
drh | e3740f2 | 2017-08-10 15:19:39 +0000 | [diff] [blame] | 1227 | }else{ |
dan | 4d906f1 | 2020-10-12 14:29:11 +0000 | [diff] [blame] | 1228 | i16 iPk; /* PRIMARY KEY column */ |
| 1229 | pPk = sqlite3PrimaryKeyIndex(pTab); |
| 1230 | assert( pPk!=0 ); |
| 1231 | assert( pPk->nKeyCol==1 ); |
| 1232 | iPk = pPk->aiColumn[0]; |
| 1233 | if( aXRef[iPk]>=0 ){ |
| 1234 | pRow = sqlite3ExprDup(db, pChanges->a[aXRef[iPk]].pExpr, 0); |
| 1235 | }else{ |
| 1236 | pRow = exprRowColumn(pParse, iPk); |
| 1237 | } |
drh | e3740f2 | 2017-08-10 15:19:39 +0000 | [diff] [blame] | 1238 | } |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 1239 | pList = sqlite3ExprListAppend(pParse, 0, pRow); |
| 1240 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1241 | for(i=0; i<pTab->nCol; i++){ |
| 1242 | if( aXRef[i]>=0 ){ |
| 1243 | pList = sqlite3ExprListAppend(pParse, pList, |
| 1244 | sqlite3ExprDup(db, pChanges->a[aXRef[i]].pExpr, 0) |
| 1245 | ); |
| 1246 | }else{ |
dan | 576d5a8 | 2020-07-15 18:30:01 +0000 | [diff] [blame] | 1247 | pList = sqlite3ExprListAppend(pParse, pList, exprRowColumn(pParse, i)); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1248 | } |
| 1249 | } |
| 1250 | |
dan | 4d906f1 | 2020-10-12 14:29:11 +0000 | [diff] [blame] | 1251 | updateFromSelect(pParse, ephemTab, pPk, pList, pSrc, pWhere, 0, 0); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1252 | sqlite3ExprListDelete(db, pList); |
| 1253 | eOnePass = ONEPASS_OFF; |
drh | e3740f2 | 2017-08-10 15:19:39 +0000 | [diff] [blame] | 1254 | }else{ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1255 | regRec = ++pParse->nMem; |
| 1256 | regRowid = ++pParse->nMem; |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1257 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1258 | /* Start scanning the virtual table */ |
drh | 895bab3 | 2022-01-27 16:14:50 +0000 | [diff] [blame] | 1259 | pWInfo = sqlite3WhereBegin( |
| 1260 | pParse, pSrc, pWhere, 0, 0, 0, WHERE_ONEPASS_DESIRED, 0 |
| 1261 | ); |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1262 | if( pWInfo==0 ) return; |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1263 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1264 | /* Populate the argument registers. */ |
| 1265 | for(i=0; i<pTab->nCol; i++){ |
| 1266 | assert( (pTab->aCol[i].colFlags & COLFLAG_GENERATED)==0 ); |
| 1267 | if( aXRef[i]>=0 ){ |
| 1268 | sqlite3ExprCode(pParse, pChanges->a[aXRef[i]].pExpr, regArg+2+i); |
| 1269 | }else{ |
| 1270 | sqlite3VdbeAddOp3(v, OP_VColumn, iCsr, i, regArg+2+i); |
| 1271 | sqlite3VdbeChangeP5(v, OPFLAG_NOCHNG);/* For sqlite3_vtab_nochange() */ |
| 1272 | } |
| 1273 | } |
| 1274 | if( HasRowid(pTab) ){ |
| 1275 | sqlite3VdbeAddOp2(v, OP_Rowid, iCsr, regArg); |
| 1276 | if( pRowid ){ |
| 1277 | sqlite3ExprCode(pParse, pRowid, regArg+1); |
| 1278 | }else{ |
| 1279 | sqlite3VdbeAddOp2(v, OP_Rowid, iCsr, regArg+1); |
| 1280 | } |
| 1281 | }else{ |
| 1282 | Index *pPk; /* PRIMARY KEY index */ |
| 1283 | i16 iPk; /* PRIMARY KEY column */ |
| 1284 | pPk = sqlite3PrimaryKeyIndex(pTab); |
| 1285 | assert( pPk!=0 ); |
| 1286 | assert( pPk->nKeyCol==1 ); |
| 1287 | iPk = pPk->aiColumn[0]; |
| 1288 | sqlite3VdbeAddOp3(v, OP_VColumn, iCsr, iPk, regArg); |
| 1289 | sqlite3VdbeAddOp2(v, OP_SCopy, regArg+2+iPk, regArg+1); |
| 1290 | } |
drh | 338e311 | 2018-05-24 23:51:57 +0000 | [diff] [blame] | 1291 | |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1292 | eOnePass = sqlite3WhereOkOnePass(pWInfo, aDummy); |
| 1293 | |
| 1294 | /* There is no ONEPASS_MULTI on virtual tables */ |
| 1295 | assert( eOnePass==ONEPASS_OFF || eOnePass==ONEPASS_SINGLE ); |
| 1296 | |
| 1297 | if( eOnePass ){ |
| 1298 | /* If using the onepass strategy, no-op out the OP_OpenEphemeral coded |
| 1299 | ** above. */ |
| 1300 | sqlite3VdbeChangeToNoop(v, addr); |
| 1301 | sqlite3VdbeAddOp1(v, OP_Close, iCsr); |
| 1302 | }else{ |
| 1303 | /* Create a record from the argument register contents and insert it into |
| 1304 | ** the ephemeral table. */ |
| 1305 | sqlite3MultiWrite(pParse); |
| 1306 | sqlite3VdbeAddOp3(v, OP_MakeRecord, regArg, nArg, regRec); |
drh | da36933 | 2020-06-29 20:09:04 +0000 | [diff] [blame] | 1307 | #if defined(SQLITE_DEBUG) && !defined(SQLITE_ENABLE_NULL_TRIM) |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1308 | /* Signal an assert() within OP_MakeRecord that it is allowed to |
| 1309 | ** accept no-change records with serial_type 10 */ |
| 1310 | sqlite3VdbeChangeP5(v, OPFLAG_NOCHNG_MAGIC); |
drh | 41fb367 | 2018-01-12 23:18:38 +0000 | [diff] [blame] | 1311 | #endif |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1312 | sqlite3VdbeAddOp2(v, OP_NewRowid, ephemTab, regRowid); |
| 1313 | sqlite3VdbeAddOp3(v, OP_Insert, ephemTab, regRec, regRowid); |
| 1314 | } |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1315 | } |
| 1316 | |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1317 | |
drh | 338e311 | 2018-05-24 23:51:57 +0000 | [diff] [blame] | 1318 | if( eOnePass==ONEPASS_OFF ){ |
dan | 354474a | 2015-09-29 10:11:26 +0000 | [diff] [blame] | 1319 | /* End the virtual table scan */ |
dan | 69887c9 | 2020-04-27 20:55:33 +0000 | [diff] [blame] | 1320 | if( pSrc->nSrc==1 ){ |
| 1321 | sqlite3WhereEnd(pWInfo); |
| 1322 | } |
dan | 354474a | 2015-09-29 10:11:26 +0000 | [diff] [blame] | 1323 | |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1324 | /* Begin scannning through the ephemeral table. */ |
dan | 354474a | 2015-09-29 10:11:26 +0000 | [diff] [blame] | 1325 | addr = sqlite3VdbeAddOp1(v, OP_Rewind, ephemTab); VdbeCoverage(v); |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1326 | |
| 1327 | /* Extract arguments from the current row of the ephemeral table and |
| 1328 | ** invoke the VUpdate method. */ |
| 1329 | for(i=0; i<nArg; i++){ |
| 1330 | sqlite3VdbeAddOp3(v, OP_Column, ephemTab, i, regArg+i); |
| 1331 | } |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1332 | } |
drh | 4f3dd15 | 2008-04-28 18:46:43 +0000 | [diff] [blame] | 1333 | sqlite3VtabMakeWritable(pParse, pTab); |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1334 | sqlite3VdbeAddOp4(v, OP_VUpdate, 0, nArg, regArg, pVTab, P4_VTAB); |
dan | b061d05 | 2011-04-25 18:49:57 +0000 | [diff] [blame] | 1335 | sqlite3VdbeChangeP5(v, onError==OE_Default ? OE_Abort : onError); |
dan | e0af83a | 2009-09-08 19:15:01 +0000 | [diff] [blame] | 1336 | sqlite3MayAbort(pParse); |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1337 | |
dan | 354474a | 2015-09-29 10:11:26 +0000 | [diff] [blame] | 1338 | /* End of the ephemeral table scan. Or, if using the onepass strategy, |
| 1339 | ** jump to here if the scan visited zero rows. */ |
drh | 338e311 | 2018-05-24 23:51:57 +0000 | [diff] [blame] | 1340 | if( eOnePass==ONEPASS_OFF ){ |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1341 | sqlite3VdbeAddOp2(v, OP_Next, ephemTab, addr+1); VdbeCoverage(v); |
| 1342 | sqlite3VdbeJumpHere(v, addr); |
| 1343 | sqlite3VdbeAddOp2(v, OP_Close, ephemTab, 0); |
dan | 354474a | 2015-09-29 10:11:26 +0000 | [diff] [blame] | 1344 | }else{ |
| 1345 | sqlite3WhereEnd(pWInfo); |
dan | 0f40037 | 2015-09-28 20:03:49 +0000 | [diff] [blame] | 1346 | } |
drh | 9c41938 | 2006-06-16 21:13:21 +0000 | [diff] [blame] | 1347 | } |
| 1348 | #endif /* SQLITE_OMIT_VIRTUALTABLE */ |