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