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