1c3f9bad2Sdanielk1977 /* 2633ed08dSdanielk1977 ** 3633ed08dSdanielk1977 ** The author disclaims copyright to this source code. In place of 4633ed08dSdanielk1977 ** a legal notice, here is a blessing: 5633ed08dSdanielk1977 ** 6633ed08dSdanielk1977 ** May you do good and not evil. 7633ed08dSdanielk1977 ** May you find forgiveness for yourself and forgive others. 8633ed08dSdanielk1977 ** May you share freely, never taking more than you give. 9633ed08dSdanielk1977 ** 10633ed08dSdanielk1977 ************************************************************************* 11822a5162Sdanielk1977 ** 12822a5162Sdanielk1977 ** 135eff7cf0Sshane ** $Id: trigger.c,v 1.143 2009/08/10 03:57:58 shane Exp $ 14c3f9bad2Sdanielk1977 */ 15c3f9bad2Sdanielk1977 #include "sqliteInt.h" 169adf9ac4Sdrh 17b7f9164eSdrh #ifndef SQLITE_OMIT_TRIGGER 18c3f9bad2Sdanielk1977 /* 194b59ab5eSdrh ** Delete a linked list of TriggerStep structures. 204b59ab5eSdrh */ 21633e6d57Sdrh void sqlite3DeleteTriggerStep(sqlite3 *db, TriggerStep *pTriggerStep){ 224b59ab5eSdrh while( pTriggerStep ){ 234b59ab5eSdrh TriggerStep * pTmp = pTriggerStep; 244b59ab5eSdrh pTriggerStep = pTriggerStep->pNext; 254b59ab5eSdrh 26633e6d57Sdrh sqlite3ExprDelete(db, pTmp->pWhere); 27633e6d57Sdrh sqlite3ExprListDelete(db, pTmp->pExprList); 28633e6d57Sdrh sqlite3SelectDelete(db, pTmp->pSelect); 29633e6d57Sdrh sqlite3IdListDelete(db, pTmp->pIdList); 304b59ab5eSdrh 31633e6d57Sdrh sqlite3DbFree(db, pTmp); 324b59ab5eSdrh } 334b59ab5eSdrh } 344b59ab5eSdrh 354b59ab5eSdrh /* 362f886d1dSdanielk1977 ** Given table pTab, return a list of all the triggers attached to 372f886d1dSdanielk1977 ** the table. The list is connected by Trigger.pNext pointers. 389ab4c2e8Sdrh ** 399ab4c2e8Sdrh ** All of the triggers on pTab that are in the same database as pTab 409ab4c2e8Sdrh ** are already attached to pTab->pTrigger. But there might be additional 419ab4c2e8Sdrh ** triggers on pTab in the TEMP schema. This routine prepends all 429ab4c2e8Sdrh ** TEMP triggers on pTab to the beginning of the pTab->pTrigger list 439ab4c2e8Sdrh ** and returns the combined list. 449ab4c2e8Sdrh ** 459ab4c2e8Sdrh ** To state it another way: This routine returns a list of all triggers 469ab4c2e8Sdrh ** that fire off of pTab. The list will include any TEMP triggers on 479ab4c2e8Sdrh ** pTab as well as the triggers lised in pTab->pTrigger. 482f886d1dSdanielk1977 */ 492f886d1dSdanielk1977 Trigger *sqlite3TriggerList(Parse *pParse, Table *pTab){ 502f886d1dSdanielk1977 Schema * const pTmpSchema = pParse->db->aDb[1].pSchema; 512f886d1dSdanielk1977 Trigger *pList = 0; /* List of triggers to return */ 522f886d1dSdanielk1977 532f886d1dSdanielk1977 if( pTmpSchema!=pTab->pSchema ){ 542f886d1dSdanielk1977 HashElem *p; 552f886d1dSdanielk1977 for(p=sqliteHashFirst(&pTmpSchema->trigHash); p; p=sqliteHashNext(p)){ 562f886d1dSdanielk1977 Trigger *pTrig = (Trigger *)sqliteHashData(p); 572f886d1dSdanielk1977 if( pTrig->pTabSchema==pTab->pSchema 582f886d1dSdanielk1977 && 0==sqlite3StrICmp(pTrig->table, pTab->zName) 592f886d1dSdanielk1977 ){ 602f886d1dSdanielk1977 pTrig->pNext = (pList ? pList : pTab->pTrigger); 612f886d1dSdanielk1977 pList = pTrig; 622f886d1dSdanielk1977 } 632f886d1dSdanielk1977 } 642f886d1dSdanielk1977 } 652f886d1dSdanielk1977 662f886d1dSdanielk1977 return (pList ? pList : pTab->pTrigger); 672f886d1dSdanielk1977 } 682f886d1dSdanielk1977 692f886d1dSdanielk1977 /* 70f0f258b1Sdrh ** This is called by the parser when it sees a CREATE TRIGGER statement 71f0f258b1Sdrh ** up to the point of the BEGIN before the trigger actions. A Trigger 72f0f258b1Sdrh ** structure is generated based on the information available and stored 73f0f258b1Sdrh ** in pParse->pNewTrigger. After the trigger actions have been parsed, the 744adee20fSdanielk1977 ** sqlite3FinishTrigger() function is called to complete the trigger 75f0f258b1Sdrh ** construction process. 76c3f9bad2Sdanielk1977 */ 774adee20fSdanielk1977 void sqlite3BeginTrigger( 78c3f9bad2Sdanielk1977 Parse *pParse, /* The parse context of the CREATE TRIGGER statement */ 79ef2cb63eSdanielk1977 Token *pName1, /* The name of the trigger */ 80ef2cb63eSdanielk1977 Token *pName2, /* The name of the trigger */ 81d702fccbSdanielk1977 int tr_tm, /* One of TK_BEFORE, TK_AFTER, TK_INSTEAD */ 82c3f9bad2Sdanielk1977 int op, /* One of TK_INSERT, TK_UPDATE, TK_DELETE */ 83633ed08dSdanielk1977 IdList *pColumns, /* column list if this is an UPDATE OF trigger */ 84d24cc427Sdrh SrcList *pTableName,/* The name of the table/view the trigger applies to */ 85c3f9bad2Sdanielk1977 Expr *pWhen, /* WHEN clause */ 86fdd48a76Sdrh int isTemp, /* True if the TEMPORARY keyword is present */ 87fdd48a76Sdrh int noErr /* Suppress errors if the trigger already exists */ 889adf9ac4Sdrh ){ 893d5f74b2Sdrh Trigger *pTrigger = 0; /* The new trigger */ 903d5f74b2Sdrh Table *pTab; /* Table that the trigger fires off of */ 91e5f9c644Sdrh char *zName = 0; /* Name of the trigger */ 923d5f74b2Sdrh sqlite3 *db = pParse->db; /* The database connection */ 93ef2cb63eSdanielk1977 int iDb; /* The database to store the trigger in */ 94ef2cb63eSdanielk1977 Token *pName; /* The unqualified db name */ 953d5f74b2Sdrh DbFixer sFix; /* State vector for the DB fixer */ 963d5f74b2Sdrh int iTabDb; /* Index of the database holding pTab */ 97ed6c8671Sdrh 9843617e9aSdrh assert( pName1!=0 ); /* pName1->z might be NULL, but not pName1 itself */ 9943617e9aSdrh assert( pName2!=0 ); 100aa78bec9Sdrh assert( op==TK_INSERT || op==TK_UPDATE || op==TK_DELETE ); 101aa78bec9Sdrh assert( op>0 && op<0xff ); 102ef2cb63eSdanielk1977 if( isTemp ){ 103ef2cb63eSdanielk1977 /* If TEMP was specified, then the trigger name may not be qualified. */ 10443617e9aSdrh if( pName2->n>0 ){ 105ef2cb63eSdanielk1977 sqlite3ErrorMsg(pParse, "temporary trigger may not have qualified name"); 106ef2cb63eSdanielk1977 goto trigger_cleanup; 107ef2cb63eSdanielk1977 } 108ef2cb63eSdanielk1977 iDb = 1; 109ef2cb63eSdanielk1977 pName = pName1; 110ef2cb63eSdanielk1977 }else{ 111ef2cb63eSdanielk1977 /* Figure out the db that the the trigger will be created in */ 112ef2cb63eSdanielk1977 iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pName); 113ef2cb63eSdanielk1977 if( iDb<0 ){ 114ef2cb63eSdanielk1977 goto trigger_cleanup; 115ef2cb63eSdanielk1977 } 116ef2cb63eSdanielk1977 } 117ef2cb63eSdanielk1977 118ef2cb63eSdanielk1977 /* If the trigger name was unqualified, and the table is a temp table, 119ef2cb63eSdanielk1977 ** then set iDb to 1 to create the trigger in the temporary database. 120ef2cb63eSdanielk1977 ** If sqlite3SrcListLookup() returns 0, indicating the table does not 121ef2cb63eSdanielk1977 ** exist, the error is caught by the block below. 122c3f9bad2Sdanielk1977 */ 12317435752Sdrh if( !pTableName || db->mallocFailed ){ 1246f7adc8aSdrh goto trigger_cleanup; 1256f7adc8aSdrh } 126ef2cb63eSdanielk1977 pTab = sqlite3SrcListLookup(pParse, pTableName); 127da184236Sdanielk1977 if( pName2->n==0 && pTab && pTab->pSchema==db->aDb[1].pSchema ){ 128ef2cb63eSdanielk1977 iDb = 1; 129ef2cb63eSdanielk1977 } 130ef2cb63eSdanielk1977 131ef2cb63eSdanielk1977 /* Ensure the table name matches database name and that the table exists */ 13217435752Sdrh if( db->mallocFailed ) goto trigger_cleanup; 133d24cc427Sdrh assert( pTableName->nSrc==1 ); 134ef2cb63eSdanielk1977 if( sqlite3FixInit(&sFix, pParse, iDb, "trigger", pName) && 135ef2cb63eSdanielk1977 sqlite3FixSrcList(&sFix, pTableName) ){ 1364312db55Sdrh goto trigger_cleanup; 137f26e09c8Sdrh } 138ef2cb63eSdanielk1977 pTab = sqlite3SrcListLookup(pParse, pTableName); 139ef2cb63eSdanielk1977 if( !pTab ){ 140ef2cb63eSdanielk1977 /* The table does not exist. */ 1413d5f74b2Sdrh if( db->init.iDb==1 ){ 1423d5f74b2Sdrh /* Ticket #3810. 1433d5f74b2Sdrh ** Normally, whenever a table is dropped, all associated triggers are 1443d5f74b2Sdrh ** dropped too. But if a TEMP trigger is created on a non-TEMP table 1453d5f74b2Sdrh ** and the table is dropped by a different database connection, the 1463d5f74b2Sdrh ** trigger is not visible to the database connection that does the 1473d5f74b2Sdrh ** drop so the trigger cannot be dropped. This results in an 1483d5f74b2Sdrh ** "orphaned trigger" - a trigger whose associated table is missing. 1493d5f74b2Sdrh */ 1503d5f74b2Sdrh db->init.orphanTrigger = 1; 1513d5f74b2Sdrh } 152d24cc427Sdrh goto trigger_cleanup; 153d24cc427Sdrh } 1544cbdda9eSdrh if( IsVirtual(pTab) ){ 1554cbdda9eSdrh sqlite3ErrorMsg(pParse, "cannot create triggers on virtual tables"); 1564cbdda9eSdrh goto trigger_cleanup; 1574cbdda9eSdrh } 158d24cc427Sdrh 159d8123366Sdanielk1977 /* Check that the trigger name is not reserved and that no trigger of the 160d8123366Sdanielk1977 ** specified name exists */ 16117435752Sdrh zName = sqlite3NameFromToken(db, pName); 162d8123366Sdanielk1977 if( !zName || SQLITE_OK!=sqlite3CheckObjectName(pParse, zName) ){ 163d8123366Sdanielk1977 goto trigger_cleanup; 164d8123366Sdanielk1977 } 165aa78bec9Sdrh if( sqlite3HashFind(&(db->aDb[iDb].pSchema->trigHash), 166ea678832Sdrh zName, sqlite3Strlen30(zName)) ){ 167fdd48a76Sdrh if( !noErr ){ 1684adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "trigger %T already exists", pName); 169fdd48a76Sdrh } 170c3f9bad2Sdanielk1977 goto trigger_cleanup; 171c3f9bad2Sdanielk1977 } 172ef2cb63eSdanielk1977 173ef2cb63eSdanielk1977 /* Do not create a trigger on a system table */ 174dca76841Sdrh if( sqlite3StrNICmp(pTab->zName, "sqlite_", 7)==0 ){ 1754adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create trigger on system table"); 176e0bc4048Sdrh pParse->nErr++; 177e0bc4048Sdrh goto trigger_cleanup; 178e0bc4048Sdrh } 179ef2cb63eSdanielk1977 180ef2cb63eSdanielk1977 /* INSTEAD of triggers are only for views and views only support INSTEAD 181ef2cb63eSdanielk1977 ** of triggers. 182ef2cb63eSdanielk1977 */ 183ef2cb63eSdanielk1977 if( pTab->pSelect && tr_tm!=TK_INSTEAD ){ 1844adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create %s trigger on view: %S", 185da93d238Sdrh (tr_tm == TK_BEFORE)?"BEFORE":"AFTER", pTableName, 0); 186d702fccbSdanielk1977 goto trigger_cleanup; 187d702fccbSdanielk1977 } 188ef2cb63eSdanielk1977 if( !pTab->pSelect && tr_tm==TK_INSTEAD ){ 1894adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create INSTEAD OF" 190da93d238Sdrh " trigger on table: %S", pTableName, 0); 191d702fccbSdanielk1977 goto trigger_cleanup; 192d702fccbSdanielk1977 } 193da184236Sdanielk1977 iTabDb = sqlite3SchemaToIndex(db, pTab->pSchema); 194ef2cb63eSdanielk1977 195e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION 196e5f9c644Sdrh { 197e5f9c644Sdrh int code = SQLITE_CREATE_TRIGGER; 198da184236Sdanielk1977 const char *zDb = db->aDb[iTabDb].zName; 199e22a334bSdrh const char *zDbTrig = isTemp ? db->aDb[1].zName : zDb; 200da184236Sdanielk1977 if( iTabDb==1 || isTemp ) code = SQLITE_CREATE_TEMP_TRIGGER; 201ef2cb63eSdanielk1977 if( sqlite3AuthCheck(pParse, code, zName, pTab->zName, zDbTrig) ){ 202ed6c8671Sdrh goto trigger_cleanup; 203ed6c8671Sdrh } 204da184236Sdanielk1977 if( sqlite3AuthCheck(pParse, SQLITE_INSERT, SCHEMA_TABLE(iTabDb),0,zDb)){ 20577ad4e41Sdrh goto trigger_cleanup; 20677ad4e41Sdrh } 207d702fccbSdanielk1977 } 208e5f9c644Sdrh #endif 209d702fccbSdanielk1977 210ef2cb63eSdanielk1977 /* INSTEAD OF triggers can only appear on views and BEFORE triggers 2115cf590c1Sdrh ** cannot appear on views. So we might as well translate every 2125cf590c1Sdrh ** INSTEAD OF trigger into a BEFORE trigger. It simplifies code 2135cf590c1Sdrh ** elsewhere. 2145cf590c1Sdrh */ 215d702fccbSdanielk1977 if (tr_tm == TK_INSTEAD){ 216d702fccbSdanielk1977 tr_tm = TK_BEFORE; 217c3f9bad2Sdanielk1977 } 218c3f9bad2Sdanielk1977 219c3f9bad2Sdanielk1977 /* Build the Trigger object */ 22017435752Sdrh pTrigger = (Trigger*)sqlite3DbMallocZero(db, sizeof(Trigger)); 221ef2cb63eSdanielk1977 if( pTrigger==0 ) goto trigger_cleanup; 222165921a7Sdan pTrigger->zName = zName; 223e5f9c644Sdrh zName = 0; 22417435752Sdrh pTrigger->table = sqlite3DbStrDup(db, pTableName->a[0].zName); 225da184236Sdanielk1977 pTrigger->pSchema = db->aDb[iDb].pSchema; 226aaf22685Sdanielk1977 pTrigger->pTabSchema = pTab->pSchema; 227aa78bec9Sdrh pTrigger->op = (u8)op; 228dca76841Sdrh pTrigger->tr_tm = tr_tm==TK_BEFORE ? TRIGGER_BEFORE : TRIGGER_AFTER; 2296ab3a2ecSdanielk1977 pTrigger->pWhen = sqlite3ExprDup(db, pWhen, EXPRDUP_REDUCE); 23017435752Sdrh pTrigger->pColumns = sqlite3IdListDup(db, pColumns); 231f0f258b1Sdrh assert( pParse->pNewTrigger==0 ); 232ef2cb63eSdanielk1977 pParse->pNewTrigger = pTrigger; 233f0f258b1Sdrh 234f0f258b1Sdrh trigger_cleanup: 235633e6d57Sdrh sqlite3DbFree(db, zName); 236633e6d57Sdrh sqlite3SrcListDelete(db, pTableName); 237633e6d57Sdrh sqlite3IdListDelete(db, pColumns); 238633e6d57Sdrh sqlite3ExprDelete(db, pWhen); 239d5d56523Sdanielk1977 if( !pParse->pNewTrigger ){ 240633e6d57Sdrh sqlite3DeleteTrigger(db, pTrigger); 241d5d56523Sdanielk1977 }else{ 242d5d56523Sdanielk1977 assert( pParse->pNewTrigger==pTrigger ); 243d5d56523Sdanielk1977 } 244f0f258b1Sdrh } 245f0f258b1Sdrh 246f0f258b1Sdrh /* 247f0f258b1Sdrh ** This routine is called after all of the trigger actions have been parsed 248f0f258b1Sdrh ** in order to complete the process of building the trigger. 249f0f258b1Sdrh */ 2504adee20fSdanielk1977 void sqlite3FinishTrigger( 251f0f258b1Sdrh Parse *pParse, /* Parser context */ 252f0f258b1Sdrh TriggerStep *pStepList, /* The triggered program */ 253f0f258b1Sdrh Token *pAll /* Token that describes the complete CREATE TRIGGER */ 254f0f258b1Sdrh ){ 2552f886d1dSdanielk1977 Trigger *pTrig = pParse->pNewTrigger; /* Trigger being finished */ 2562f886d1dSdanielk1977 char *zName; /* Name of trigger */ 2579bb575fdSdrh sqlite3 *db = pParse->db; /* The database */ 258f26e09c8Sdrh DbFixer sFix; 259da184236Sdanielk1977 int iDb; /* Database containing the trigger */ 260b7916a78Sdrh Token nameToken; /* Trigger name for error reporting */ 261f0f258b1Sdrh 262bfb9e35bSdanielk1977 pTrig = pParse->pNewTrigger; 263f0f258b1Sdrh pParse->pNewTrigger = 0; 2649ab4c2e8Sdrh if( NEVER(pParse->nErr) || !pTrig ) goto triggerfinish_cleanup; 265165921a7Sdan zName = pTrig->zName; 266da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pTrig->pSchema); 267bfb9e35bSdanielk1977 pTrig->step_list = pStepList; 268a69d9168Sdrh while( pStepList ){ 269bfb9e35bSdanielk1977 pStepList->pTrig = pTrig; 270a69d9168Sdrh pStepList = pStepList->pNext; 271a69d9168Sdrh } 272165921a7Sdan nameToken.z = pTrig->zName; 273b7916a78Sdrh nameToken.n = sqlite3Strlen30(nameToken.z); 274b7916a78Sdrh if( sqlite3FixInit(&sFix, pParse, iDb, "trigger", &nameToken) 275bfb9e35bSdanielk1977 && sqlite3FixTriggerStep(&sFix, pTrig->step_list) ){ 276f26e09c8Sdrh goto triggerfinish_cleanup; 277f26e09c8Sdrh } 278c3f9bad2Sdanielk1977 279c3f9bad2Sdanielk1977 /* if we are not initializing, and this trigger is not on a TEMP table, 2809adf9ac4Sdrh ** build the sqlite_master entry 2819adf9ac4Sdrh */ 2821d85d931Sdrh if( !db->init.busy ){ 283c977f7f5Sdrh Vdbe *v; 2842d401ab8Sdrh char *z; 285c3f9bad2Sdanielk1977 286c3f9bad2Sdanielk1977 /* Make an entry in the sqlite_master table */ 2874adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 288f0f258b1Sdrh if( v==0 ) goto triggerfinish_cleanup; 289da184236Sdanielk1977 sqlite3BeginWriteOperation(pParse, 0, iDb); 2902d401ab8Sdrh z = sqlite3DbStrNDup(db, (char*)pAll->z, pAll->n); 2912d401ab8Sdrh sqlite3NestedParse(pParse, 2922d401ab8Sdrh "INSERT INTO %Q.%s VALUES('trigger',%Q,%Q,0,'CREATE TRIGGER %q')", 2932f886d1dSdanielk1977 db->aDb[iDb].zName, SCHEMA_TABLE(iDb), zName, 2942d401ab8Sdrh pTrig->table, z); 295633e6d57Sdrh sqlite3DbFree(db, z); 2969cbf3425Sdrh sqlite3ChangeCookie(pParse, iDb); 29766a5167bSdrh sqlite3VdbeAddOp4(v, OP_ParseSchema, iDb, 0, 0, sqlite3MPrintf( 2982f886d1dSdanielk1977 db, "type='trigger' AND name='%q'", zName), P4_DYNAMIC 2991e536953Sdanielk1977 ); 300c3f9bad2Sdanielk1977 } 301c3f9bad2Sdanielk1977 302956bc92cSdrh if( db->init.busy ){ 3032f886d1dSdanielk1977 Trigger *pLink = pTrig; 3042f886d1dSdanielk1977 Hash *pHash = &db->aDb[iDb].pSchema->trigHash; 3052f886d1dSdanielk1977 pTrig = sqlite3HashInsert(pHash, zName, sqlite3Strlen30(zName), pTrig); 3062f886d1dSdanielk1977 if( pTrig ){ 307f3a65f7eSdrh db->mallocFailed = 1; 3082f886d1dSdanielk1977 }else if( pLink->pSchema==pLink->pTabSchema ){ 3092f886d1dSdanielk1977 Table *pTab; 310a83ccca7Sdrh int n = sqlite3Strlen30(pLink->table); 3112f886d1dSdanielk1977 pTab = sqlite3HashFind(&pLink->pTabSchema->tblHash, pLink->table, n); 312f0f258b1Sdrh assert( pTab!=0 ); 3132f886d1dSdanielk1977 pLink->pNext = pTab->pTrigger; 3142f886d1dSdanielk1977 pTab->pTrigger = pLink; 3152f886d1dSdanielk1977 } 316c3f9bad2Sdanielk1977 } 317c3f9bad2Sdanielk1977 318f0f258b1Sdrh triggerfinish_cleanup: 319633e6d57Sdrh sqlite3DeleteTrigger(db, pTrig); 320bfb9e35bSdanielk1977 assert( !pParse->pNewTrigger ); 321633e6d57Sdrh sqlite3DeleteTriggerStep(db, pStepList); 322c3f9bad2Sdanielk1977 } 3234b59ab5eSdrh 3244b59ab5eSdrh /* 325c977f7f5Sdrh ** Turn a SELECT statement (that the pSelect parameter points to) into 326c977f7f5Sdrh ** a trigger step. Return a pointer to a TriggerStep structure. 327c977f7f5Sdrh ** 328c977f7f5Sdrh ** The parser calls this routine when it finds a SELECT statement in 329c977f7f5Sdrh ** body of a TRIGGER. 330c977f7f5Sdrh */ 33117435752Sdrh TriggerStep *sqlite3TriggerSelectStep(sqlite3 *db, Select *pSelect){ 33217435752Sdrh TriggerStep *pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep)); 3332e588c75Sdanielk1977 if( pTriggerStep==0 ) { 334633e6d57Sdrh sqlite3SelectDelete(db, pSelect); 3352e588c75Sdanielk1977 return 0; 3362e588c75Sdanielk1977 } 337633ed08dSdanielk1977 pTriggerStep->op = TK_SELECT; 338633ed08dSdanielk1977 pTriggerStep->pSelect = pSelect; 339633ed08dSdanielk1977 pTriggerStep->orconf = OE_Default; 340b7916a78Sdrh return pTriggerStep; 341b7916a78Sdrh } 342c3f9bad2Sdanielk1977 343b7916a78Sdrh /* 344b7916a78Sdrh ** Allocate space to hold a new trigger step. The allocated space 345b7916a78Sdrh ** holds both the TriggerStep object and the TriggerStep.target.z string. 346b7916a78Sdrh ** 347b7916a78Sdrh ** If an OOM error occurs, NULL is returned and db->mallocFailed is set. 348b7916a78Sdrh */ 349b7916a78Sdrh static TriggerStep *triggerStepAllocate( 350b7916a78Sdrh sqlite3 *db, /* Database connection */ 3515eff7cf0Sshane u8 op, /* Trigger opcode */ 352b7916a78Sdrh Token *pName /* The target name */ 353b7916a78Sdrh ){ 354b7916a78Sdrh TriggerStep *pTriggerStep; 355b7916a78Sdrh 356b7916a78Sdrh pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep) + pName->n); 357b7916a78Sdrh if( pTriggerStep ){ 358b7916a78Sdrh char *z = (char*)&pTriggerStep[1]; 359b7916a78Sdrh memcpy(z, pName->z, pName->n); 360b7916a78Sdrh pTriggerStep->target.z = z; 361b7916a78Sdrh pTriggerStep->target.n = pName->n; 362b7916a78Sdrh pTriggerStep->op = op; 363b7916a78Sdrh } 364633ed08dSdanielk1977 return pTriggerStep; 365c3f9bad2Sdanielk1977 } 366c3f9bad2Sdanielk1977 367c977f7f5Sdrh /* 368c977f7f5Sdrh ** Build a trigger step out of an INSERT statement. Return a pointer 369c977f7f5Sdrh ** to the new trigger step. 370c977f7f5Sdrh ** 371c977f7f5Sdrh ** The parser calls this routine when it sees an INSERT inside the 372c977f7f5Sdrh ** body of a trigger. 373c977f7f5Sdrh */ 3744adee20fSdanielk1977 TriggerStep *sqlite3TriggerInsertStep( 37517435752Sdrh sqlite3 *db, /* The database connection */ 376c977f7f5Sdrh Token *pTableName, /* Name of the table into which we insert */ 377c977f7f5Sdrh IdList *pColumn, /* List of columns in pTableName to insert into */ 378c977f7f5Sdrh ExprList *pEList, /* The VALUE clause: a list of values to be inserted */ 379c977f7f5Sdrh Select *pSelect, /* A SELECT statement that supplies values */ 3805eff7cf0Sshane u8 orconf /* The conflict algorithm (OE_Abort, OE_Replace, etc.) */ 381633ed08dSdanielk1977 ){ 382f3a65f7eSdrh TriggerStep *pTriggerStep; 383c3f9bad2Sdanielk1977 384633ed08dSdanielk1977 assert(pEList == 0 || pSelect == 0); 385f3a65f7eSdrh assert(pEList != 0 || pSelect != 0 || db->mallocFailed); 386c3f9bad2Sdanielk1977 387b7916a78Sdrh pTriggerStep = triggerStepAllocate(db, TK_INSERT, pTableName); 388d5d56523Sdanielk1977 if( pTriggerStep ){ 38933e619fcSdrh pTriggerStep->pSelect = sqlite3SelectDup(db, pSelect, EXPRDUP_REDUCE); 390633ed08dSdanielk1977 pTriggerStep->pIdList = pColumn; 39133e619fcSdrh pTriggerStep->pExprList = sqlite3ExprListDup(db, pEList, EXPRDUP_REDUCE); 392633ed08dSdanielk1977 pTriggerStep->orconf = orconf; 393d5d56523Sdanielk1977 }else{ 394633e6d57Sdrh sqlite3IdListDelete(db, pColumn); 39533e619fcSdrh } 396633e6d57Sdrh sqlite3ExprListDelete(db, pEList); 397633e6d57Sdrh sqlite3SelectDelete(db, pSelect); 398c3f9bad2Sdanielk1977 399633ed08dSdanielk1977 return pTriggerStep; 400c3f9bad2Sdanielk1977 } 401c3f9bad2Sdanielk1977 402c977f7f5Sdrh /* 403c977f7f5Sdrh ** Construct a trigger step that implements an UPDATE statement and return 404c977f7f5Sdrh ** a pointer to that trigger step. The parser calls this routine when it 405c977f7f5Sdrh ** sees an UPDATE statement inside the body of a CREATE TRIGGER. 406c977f7f5Sdrh */ 4074adee20fSdanielk1977 TriggerStep *sqlite3TriggerUpdateStep( 40817435752Sdrh sqlite3 *db, /* The database connection */ 409c977f7f5Sdrh Token *pTableName, /* Name of the table to be updated */ 410c977f7f5Sdrh ExprList *pEList, /* The SET clause: list of column and new values */ 411c977f7f5Sdrh Expr *pWhere, /* The WHERE clause */ 4125eff7cf0Sshane u8 orconf /* The conflict algorithm. (OE_Abort, OE_Ignore, etc) */ 413c977f7f5Sdrh ){ 414b7916a78Sdrh TriggerStep *pTriggerStep; 415b7916a78Sdrh 416b7916a78Sdrh pTriggerStep = triggerStepAllocate(db, TK_UPDATE, pTableName); 41733e619fcSdrh if( pTriggerStep ){ 41833e619fcSdrh pTriggerStep->pExprList = sqlite3ExprListDup(db, pEList, EXPRDUP_REDUCE); 41933e619fcSdrh pTriggerStep->pWhere = sqlite3ExprDup(db, pWhere, EXPRDUP_REDUCE); 42033e619fcSdrh pTriggerStep->orconf = orconf; 42133e619fcSdrh } 422633e6d57Sdrh sqlite3ExprListDelete(db, pEList); 423633e6d57Sdrh sqlite3ExprDelete(db, pWhere); 424633ed08dSdanielk1977 return pTriggerStep; 425c3f9bad2Sdanielk1977 } 426c3f9bad2Sdanielk1977 427c977f7f5Sdrh /* 428c977f7f5Sdrh ** Construct a trigger step that implements a DELETE statement and return 429c977f7f5Sdrh ** a pointer to that trigger step. The parser calls this routine when it 430c977f7f5Sdrh ** sees a DELETE statement inside the body of a CREATE TRIGGER. 431c977f7f5Sdrh */ 43217435752Sdrh TriggerStep *sqlite3TriggerDeleteStep( 43317435752Sdrh sqlite3 *db, /* Database connection */ 43417435752Sdrh Token *pTableName, /* The table from which rows are deleted */ 43517435752Sdrh Expr *pWhere /* The WHERE clause */ 43617435752Sdrh ){ 437b7916a78Sdrh TriggerStep *pTriggerStep; 438b7916a78Sdrh 439b7916a78Sdrh pTriggerStep = triggerStepAllocate(db, TK_DELETE, pTableName); 44033e619fcSdrh if( pTriggerStep ){ 44133e619fcSdrh pTriggerStep->pWhere = sqlite3ExprDup(db, pWhere, EXPRDUP_REDUCE); 442633ed08dSdanielk1977 pTriggerStep->orconf = OE_Default; 44333e619fcSdrh } 44433e619fcSdrh sqlite3ExprDelete(db, pWhere); 445633ed08dSdanielk1977 return pTriggerStep; 446c3f9bad2Sdanielk1977 } 447c3f9bad2Sdanielk1977 448c3f9bad2Sdanielk1977 /* 449633ed08dSdanielk1977 ** Recursively delete a Trigger structure 450c3f9bad2Sdanielk1977 */ 451633e6d57Sdrh void sqlite3DeleteTrigger(sqlite3 *db, Trigger *pTrigger){ 452f0f258b1Sdrh if( pTrigger==0 ) return; 453633e6d57Sdrh sqlite3DeleteTriggerStep(db, pTrigger->step_list); 454165921a7Sdan sqlite3DbFree(db, pTrigger->zName); 455633e6d57Sdrh sqlite3DbFree(db, pTrigger->table); 456633e6d57Sdrh sqlite3ExprDelete(db, pTrigger->pWhen); 457633e6d57Sdrh sqlite3IdListDelete(db, pTrigger->pColumns); 458633e6d57Sdrh sqlite3DbFree(db, pTrigger); 459633ed08dSdanielk1977 } 460633ed08dSdanielk1977 461633ed08dSdanielk1977 /* 4628a41449eSdanielk1977 ** This function is called to drop a trigger from the database schema. 4638a41449eSdanielk1977 ** 4648a41449eSdanielk1977 ** This may be called directly from the parser and therefore identifies 4658a41449eSdanielk1977 ** the trigger by name. The sqlite3DropTriggerPtr() routine does the 4668a41449eSdanielk1977 ** same job as this routine except it takes a pointer to the trigger 4678a41449eSdanielk1977 ** instead of the trigger name. 4688a41449eSdanielk1977 **/ 469fdd48a76Sdrh void sqlite3DropTrigger(Parse *pParse, SrcList *pName, int noErr){ 470742f947bSdanielk1977 Trigger *pTrigger = 0; 471d24cc427Sdrh int i; 472d24cc427Sdrh const char *zDb; 473d24cc427Sdrh const char *zName; 474d24cc427Sdrh int nName; 4759bb575fdSdrh sqlite3 *db = pParse->db; 476633ed08dSdanielk1977 47717435752Sdrh if( db->mallocFailed ) goto drop_trigger_cleanup; 4788a41449eSdanielk1977 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){ 479c0391398Sdanielk1977 goto drop_trigger_cleanup; 480c0391398Sdanielk1977 } 4818e227875Sdanielk1977 482d24cc427Sdrh assert( pName->nSrc==1 ); 483d24cc427Sdrh zDb = pName->a[0].zDatabase; 484d24cc427Sdrh zName = pName->a[0].zName; 485ea678832Sdrh nName = sqlite3Strlen30(zName); 48653c0f748Sdanielk1977 for(i=OMIT_TEMPDB; i<db->nDb; i++){ 487812d7a21Sdrh int j = (i<2) ? i^1 : i; /* Search TEMP before MAIN */ 4884adee20fSdanielk1977 if( zDb && sqlite3StrICmp(db->aDb[j].zName, zDb) ) continue; 489e4df0e74Sdrh pTrigger = sqlite3HashFind(&(db->aDb[j].pSchema->trigHash), zName, nName); 490d24cc427Sdrh if( pTrigger ) break; 491c3f9bad2Sdanielk1977 } 492d24cc427Sdrh if( !pTrigger ){ 493fdd48a76Sdrh if( !noErr ){ 4944adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "no such trigger: %S", pName, 0); 495fdd48a76Sdrh } 496d24cc427Sdrh goto drop_trigger_cleanup; 497d24cc427Sdrh } 49874161705Sdrh sqlite3DropTriggerPtr(pParse, pTrigger); 49979a519c0Sdrh 50079a519c0Sdrh drop_trigger_cleanup: 501633e6d57Sdrh sqlite3SrcListDelete(db, pName); 50279a519c0Sdrh } 50379a519c0Sdrh 50479a519c0Sdrh /* 505956bc92cSdrh ** Return a pointer to the Table structure for the table that a trigger 506956bc92cSdrh ** is set on. 507956bc92cSdrh */ 50874161705Sdrh static Table *tableOfTrigger(Trigger *pTrigger){ 509a83ccca7Sdrh int n = sqlite3Strlen30(pTrigger->table); 510aaf22685Sdanielk1977 return sqlite3HashFind(&pTrigger->pTabSchema->tblHash, pTrigger->table, n); 511956bc92cSdrh } 512956bc92cSdrh 513956bc92cSdrh 514956bc92cSdrh /* 51574161705Sdrh ** Drop a trigger given a pointer to that trigger. 51679a519c0Sdrh */ 51774161705Sdrh void sqlite3DropTriggerPtr(Parse *pParse, Trigger *pTrigger){ 51879a519c0Sdrh Table *pTable; 51979a519c0Sdrh Vdbe *v; 5209bb575fdSdrh sqlite3 *db = pParse->db; 521956bc92cSdrh int iDb; 52279a519c0Sdrh 523da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pTrigger->pSchema); 524956bc92cSdrh assert( iDb>=0 && iDb<db->nDb ); 52574161705Sdrh pTable = tableOfTrigger(pTrigger); 526ed6c8671Sdrh assert( pTable ); 527da184236Sdanielk1977 assert( pTable->pSchema==pTrigger->pSchema || iDb==1 ); 528e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION 529e5f9c644Sdrh { 530e5f9c644Sdrh int code = SQLITE_DROP_TRIGGER; 531956bc92cSdrh const char *zDb = db->aDb[iDb].zName; 532956bc92cSdrh const char *zTab = SCHEMA_TABLE(iDb); 533956bc92cSdrh if( iDb==1 ) code = SQLITE_DROP_TEMP_TRIGGER; 5342bd93516Sdan if( sqlite3AuthCheck(pParse, code, pTrigger->zName, pTable->zName, zDb) || 5354adee20fSdanielk1977 sqlite3AuthCheck(pParse, SQLITE_DELETE, zTab, 0, zDb) ){ 53679a519c0Sdrh return; 537ed6c8671Sdrh } 538e5f9c644Sdrh } 539e5f9c644Sdrh #endif 540c3f9bad2Sdanielk1977 541f0f258b1Sdrh /* Generate code to destroy the database record of the trigger. 542f0f258b1Sdrh */ 54343617e9aSdrh assert( pTable!=0 ); 54443617e9aSdrh if( (v = sqlite3GetVdbe(pParse))!=0 ){ 545f0f258b1Sdrh int base; 5465719628aSdrh static const VdbeOpList dropTrigger[] = { 5479b1b01bbSdrh { OP_Rewind, 0, ADDR(9), 0}, 5481db639ceSdrh { OP_String8, 0, 1, 0}, /* 1 */ 5491db639ceSdrh { OP_Column, 0, 1, 2}, 5501db639ceSdrh { OP_Ne, 2, ADDR(8), 1}, 5511db639ceSdrh { OP_String8, 0, 1, 0}, /* 4: "trigger" */ 5521db639ceSdrh { OP_Column, 0, 0, 2}, 5531db639ceSdrh { OP_Ne, 2, ADDR(8), 1}, 554f0f258b1Sdrh { OP_Delete, 0, 0, 0}, 5559b1b01bbSdrh { OP_Next, 0, ADDR(1), 0}, /* 8 */ 556f0f258b1Sdrh }; 557f0f258b1Sdrh 558956bc92cSdrh sqlite3BeginWriteOperation(pParse, 0, iDb); 559c00da105Sdanielk1977 sqlite3OpenMasterTable(pParse, iDb); 5604adee20fSdanielk1977 base = sqlite3VdbeAddOpList(v, ArraySize(dropTrigger), dropTrigger); 561165921a7Sdan sqlite3VdbeChangeP4(v, base+1, pTrigger->zName, 0); 5622400345aSdrh sqlite3VdbeChangeP4(v, base+4, "trigger", P4_STATIC); 5639cbf3425Sdrh sqlite3ChangeCookie(pParse, iDb); 56466a5167bSdrh sqlite3VdbeAddOp2(v, OP_Close, 0, 0); 565165921a7Sdan sqlite3VdbeAddOp4(v, OP_DropTrigger, iDb, 0, 0, pTrigger->zName, 0); 5666ab3a2ecSdanielk1977 if( pParse->nMem<3 ){ 5676ab3a2ecSdanielk1977 pParse->nMem = 3; 5686ab3a2ecSdanielk1977 } 569956bc92cSdrh } 570f0f258b1Sdrh } 571f0f258b1Sdrh 572c3f9bad2Sdanielk1977 /* 573956bc92cSdrh ** Remove a trigger from the hash tables of the sqlite* pointer. 574c3f9bad2Sdanielk1977 */ 575956bc92cSdrh void sqlite3UnlinkAndDeleteTrigger(sqlite3 *db, int iDb, const char *zName){ 5762f886d1dSdanielk1977 Hash *pHash = &(db->aDb[iDb].pSchema->trigHash); 577956bc92cSdrh Trigger *pTrigger; 5782f886d1dSdanielk1977 pTrigger = sqlite3HashInsert(pHash, zName, sqlite3Strlen30(zName), 0); 5799ab4c2e8Sdrh if( ALWAYS(pTrigger) ){ 5802f886d1dSdanielk1977 if( pTrigger->pSchema==pTrigger->pTabSchema ){ 5812f886d1dSdanielk1977 Table *pTab = tableOfTrigger(pTrigger); 5822f886d1dSdanielk1977 Trigger **pp; 5832f886d1dSdanielk1977 for(pp=&pTab->pTrigger; *pp!=pTrigger; pp=&((*pp)->pNext)); 5842f886d1dSdanielk1977 *pp = (*pp)->pNext; 585c3f9bad2Sdanielk1977 } 586633e6d57Sdrh sqlite3DeleteTrigger(db, pTrigger); 587956bc92cSdrh db->flags |= SQLITE_InternChanges; 588c3f9bad2Sdanielk1977 } 589c3f9bad2Sdanielk1977 } 590c3f9bad2Sdanielk1977 591c977f7f5Sdrh /* 592c977f7f5Sdrh ** pEList is the SET clause of an UPDATE statement. Each entry 593c977f7f5Sdrh ** in pEList is of the format <id>=<expr>. If any of the entries 594c977f7f5Sdrh ** in pEList have an <id> which matches an identifier in pIdList, 595c977f7f5Sdrh ** then return TRUE. If pIdList==NULL, then it is considered a 596c977f7f5Sdrh ** wildcard that matches anything. Likewise if pEList==NULL then 597c977f7f5Sdrh ** it matches anything so always return true. Return false only 598c977f7f5Sdrh ** if there is no match. 599c977f7f5Sdrh */ 6009ab4c2e8Sdrh static int checkColumnOverlap(IdList *pIdList, ExprList *pEList){ 601ad2d8307Sdrh int e; 6029ab4c2e8Sdrh if( pIdList==0 || NEVER(pEList==0) ) return 1; 603f29ce559Sdanielk1977 for(e=0; e<pEList->nExpr; e++){ 6044adee20fSdanielk1977 if( sqlite3IdListIndex(pIdList, pEList->a[e].zName)>=0 ) return 1; 605f29ce559Sdanielk1977 } 606c3f9bad2Sdanielk1977 return 0; 607c3f9bad2Sdanielk1977 } 608c3f9bad2Sdanielk1977 609c3f9bad2Sdanielk1977 /* 6102f886d1dSdanielk1977 ** Return a list of all triggers on table pTab if there exists at least 6112f886d1dSdanielk1977 ** one trigger that must be fired when an operation of type 'op' is 6122f886d1dSdanielk1977 ** performed on the table, and, if that operation is an UPDATE, if at 6132f886d1dSdanielk1977 ** least one of the columns in pChanges is being modified. 614c3f9bad2Sdanielk1977 */ 6152f886d1dSdanielk1977 Trigger *sqlite3TriggersExist( 6162f886d1dSdanielk1977 Parse *pParse, /* Parse context */ 617dca76841Sdrh Table *pTab, /* The table the contains the triggers */ 618c3f9bad2Sdanielk1977 int op, /* one of TK_DELETE, TK_INSERT, TK_UPDATE */ 6192f886d1dSdanielk1977 ExprList *pChanges, /* Columns that change in an UPDATE statement */ 6202f886d1dSdanielk1977 int *pMask /* OUT: Mask of TRIGGER_BEFORE|TRIGGER_AFTER */ 621c977f7f5Sdrh ){ 622dca76841Sdrh int mask = 0; 6232f886d1dSdanielk1977 Trigger *pList = sqlite3TriggerList(pParse, pTab); 6242f886d1dSdanielk1977 Trigger *p; 6252f886d1dSdanielk1977 assert( pList==0 || IsVirtual(pTab)==0 ); 6262f886d1dSdanielk1977 for(p=pList; p; p=p->pNext){ 6279ab4c2e8Sdrh if( p->op==op && checkColumnOverlap(p->pColumns, pChanges) ){ 6282f886d1dSdanielk1977 mask |= p->tr_tm; 629c3f9bad2Sdanielk1977 } 630dca76841Sdrh } 6312f886d1dSdanielk1977 if( pMask ){ 6322f886d1dSdanielk1977 *pMask = mask; 6332f886d1dSdanielk1977 } 6342f886d1dSdanielk1977 return (mask ? pList : 0); 635c3f9bad2Sdanielk1977 } 636c3f9bad2Sdanielk1977 637c977f7f5Sdrh /* 638f26e09c8Sdrh ** Convert the pStep->target token into a SrcList and return a pointer 639f26e09c8Sdrh ** to that SrcList. 640f26e09c8Sdrh ** 641f26e09c8Sdrh ** This routine adds a specific database name, if needed, to the target when 642f26e09c8Sdrh ** forming the SrcList. This prevents a trigger in one database from 643f26e09c8Sdrh ** referring to a target in another database. An exception is when the 644f26e09c8Sdrh ** trigger is in TEMP in which case it can refer to any other database it 645f26e09c8Sdrh ** wants. 646f26e09c8Sdrh */ 647f26e09c8Sdrh static SrcList *targetSrcList( 648f26e09c8Sdrh Parse *pParse, /* The parsing context */ 649f26e09c8Sdrh TriggerStep *pStep /* The trigger containing the target token */ 650f26e09c8Sdrh ){ 651f26e09c8Sdrh int iDb; /* Index of the database to use */ 652f26e09c8Sdrh SrcList *pSrc; /* SrcList to be returned */ 653f26e09c8Sdrh 654b7916a78Sdrh pSrc = sqlite3SrcListAppend(pParse->db, 0, &pStep->target, 0); 655b7916a78Sdrh if( pSrc ){ 656b7916a78Sdrh assert( pSrc->nSrc>0 ); 657b7916a78Sdrh assert( pSrc->a!=0 ); 658da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pStep->pTrig->pSchema); 659f26e09c8Sdrh if( iDb==0 || iDb>=2 ){ 660b7916a78Sdrh sqlite3 *db = pParse->db; 661f26e09c8Sdrh assert( iDb<pParse->db->nDb ); 662b7916a78Sdrh pSrc->a[pSrc->nSrc-1].zDatabase = sqlite3DbStrDup(db, db->aDb[iDb].zName); 663b7916a78Sdrh } 664f26e09c8Sdrh } 665f26e09c8Sdrh return pSrc; 666f26e09c8Sdrh } 667f26e09c8Sdrh 668f26e09c8Sdrh /* 66965a7cd16Sdan ** Generate VDBE code for the statements inside the body of a single 670c977f7f5Sdrh ** trigger. 671c977f7f5Sdrh */ 672c3f9bad2Sdanielk1977 static int codeTriggerProgram( 673c977f7f5Sdrh Parse *pParse, /* The parser context */ 674c977f7f5Sdrh TriggerStep *pStepList, /* List of statements inside the trigger body */ 67565a7cd16Sdan int orconf /* Conflict algorithm. (OE_Abort, etc) */ 676633ed08dSdanielk1977 ){ 67765a7cd16Sdan TriggerStep *pStep; 678344737f6Sdrh Vdbe *v = pParse->pVdbe; 67917435752Sdrh sqlite3 *db = pParse->db; 680c3f9bad2Sdanielk1977 68165a7cd16Sdan assert( pParse->pTriggerTab && pParse->pToplevel ); 68265a7cd16Sdan assert( pStepList ); 683344737f6Sdrh assert( v!=0 ); 68465a7cd16Sdan for(pStep=pStepList; pStep; pStep=pStep->pNext){ 685165921a7Sdan /* Figure out the ON CONFLICT policy that will be used for this step 686165921a7Sdan ** of the trigger program. If the statement that caused this trigger 687165921a7Sdan ** to fire had an explicit ON CONFLICT, then use it. Otherwise, use 688165921a7Sdan ** the ON CONFLICT policy that was specified as part of the trigger 689165921a7Sdan ** step statement. Example: 690165921a7Sdan ** 691165921a7Sdan ** CREATE TRIGGER AFTER INSERT ON t1 BEGIN; 692165921a7Sdan ** INSERT OR REPLACE INTO t2 VALUES(new.a, new.b); 693165921a7Sdan ** END; 694165921a7Sdan ** 695165921a7Sdan ** INSERT INTO t1 ... ; -- insert into t2 uses REPLACE policy 696165921a7Sdan ** INSERT OR IGNORE INTO t1 ... ; -- insert into t2 uses IGNORE policy 697165921a7Sdan */ 698*cea72b2dSshane pParse->eOrconf = (orconf==OE_Default)?pStep->orconf:(u8)orconf; 699165921a7Sdan 700165921a7Sdan switch( pStep->op ){ 701c3f9bad2Sdanielk1977 case TK_UPDATE: { 702165921a7Sdan sqlite3Update(pParse, 703165921a7Sdan targetSrcList(pParse, pStep), 704165921a7Sdan sqlite3ExprListDup(db, pStep->pExprList, 0), 705165921a7Sdan sqlite3ExprDup(db, pStep->pWhere, 0), 70665a7cd16Sdan pParse->eOrconf 707165921a7Sdan ); 708c3f9bad2Sdanielk1977 break; 709c3f9bad2Sdanielk1977 } 710c3f9bad2Sdanielk1977 case TK_INSERT: { 711165921a7Sdan sqlite3Insert(pParse, 712165921a7Sdan targetSrcList(pParse, pStep), 713165921a7Sdan sqlite3ExprListDup(db, pStep->pExprList, 0), 714165921a7Sdan sqlite3SelectDup(db, pStep->pSelect, 0), 715165921a7Sdan sqlite3IdListDup(db, pStep->pIdList), 71665a7cd16Sdan pParse->eOrconf 717165921a7Sdan ); 718c3f9bad2Sdanielk1977 break; 719c3f9bad2Sdanielk1977 } 720c3f9bad2Sdanielk1977 case TK_DELETE: { 721165921a7Sdan sqlite3DeleteFrom(pParse, 722165921a7Sdan targetSrcList(pParse, pStep), 723165921a7Sdan sqlite3ExprDup(db, pStep->pWhere, 0) 724165921a7Sdan ); 725c3f9bad2Sdanielk1977 break; 726c3f9bad2Sdanielk1977 } 727165921a7Sdan default: assert( pStep->op==TK_SELECT ); { 728165921a7Sdan SelectDest sDest; 729165921a7Sdan Select *pSelect = sqlite3SelectDup(db, pStep->pSelect, 0); 730165921a7Sdan sqlite3SelectDestInit(&sDest, SRT_Discard, 0); 731165921a7Sdan sqlite3Select(pParse, pSelect, &sDest); 732165921a7Sdan sqlite3SelectDelete(db, pSelect); 7339ab4c2e8Sdrh break; 7349ab4c2e8Sdrh } 735c3f9bad2Sdanielk1977 } 73676d462eeSdan if( pStep->op!=TK_SELECT ){ 73776d462eeSdan sqlite3VdbeAddOp1(v, OP_ResetCount, 1); 73876d462eeSdan } 739c3f9bad2Sdanielk1977 } 740c3f9bad2Sdanielk1977 741c3f9bad2Sdanielk1977 return 0; 742c3f9bad2Sdanielk1977 } 743c3f9bad2Sdanielk1977 744165921a7Sdan #ifdef SQLITE_DEBUG 745165921a7Sdan /* 746165921a7Sdan ** This function is used to add VdbeComment() annotations to a VDBE 747165921a7Sdan ** program. It is not used in production code, only for debugging. 748165921a7Sdan */ 749165921a7Sdan static const char *onErrorText(int onError){ 750165921a7Sdan switch( onError ){ 751165921a7Sdan case OE_Abort: return "abort"; 752165921a7Sdan case OE_Rollback: return "rollback"; 753165921a7Sdan case OE_Fail: return "fail"; 754165921a7Sdan case OE_Replace: return "replace"; 755165921a7Sdan case OE_Ignore: return "ignore"; 756165921a7Sdan case OE_Default: return "default"; 757165921a7Sdan } 758165921a7Sdan return "n/a"; 759165921a7Sdan } 760165921a7Sdan #endif 761165921a7Sdan 762165921a7Sdan /* 763165921a7Sdan ** Parse context structure pFrom has just been used to create a sub-vdbe 764165921a7Sdan ** (trigger program). If an error has occurred, transfer error information 765165921a7Sdan ** from pFrom to pTo. 766165921a7Sdan */ 767165921a7Sdan static void transferParseError(Parse *pTo, Parse *pFrom){ 768165921a7Sdan assert( pFrom->zErrMsg==0 || pFrom->nErr ); 769165921a7Sdan assert( pTo->zErrMsg==0 || pTo->nErr ); 770165921a7Sdan if( pTo->nErr==0 ){ 771165921a7Sdan pTo->zErrMsg = pFrom->zErrMsg; 772165921a7Sdan pTo->nErr = pFrom->nErr; 773165921a7Sdan }else{ 774165921a7Sdan sqlite3DbFree(pFrom->db, pFrom->zErrMsg); 775165921a7Sdan } 776165921a7Sdan } 777165921a7Sdan 77865a7cd16Sdan /* 77965a7cd16Sdan ** Create and populate a new TriggerPrg object with a sub-program 78065a7cd16Sdan ** implementing trigger pTrigger with ON CONFLICT policy orconf. 78165a7cd16Sdan */ 7822832ad42Sdan static TriggerPrg *codeRowTrigger( 783165921a7Sdan Parse *pParse, /* Current parse context */ 784165921a7Sdan Trigger *pTrigger, /* Trigger to code */ 78565a7cd16Sdan Table *pTab, /* The table pTrigger is attached to */ 78665a7cd16Sdan int orconf /* ON CONFLICT policy to code trigger program with */ 787165921a7Sdan ){ 78865a7cd16Sdan Parse *pTop = sqlite3ParseToplevel(pParse); 78965a7cd16Sdan sqlite3 *db = pParse->db; /* Database handle */ 79065a7cd16Sdan TriggerPrg *pPrg; /* Value to return */ 791165921a7Sdan Expr *pWhen = 0; /* Duplicate of trigger WHEN expression */ 792165921a7Sdan Vdbe *v; /* Temporary VM */ 793165921a7Sdan NameContext sNC; /* Name context for sub-vdbe */ 794165921a7Sdan SubProgram *pProgram = 0; /* Sub-vdbe for trigger program */ 795165921a7Sdan Parse *pSubParse; /* Parse context for sub-vdbe */ 796165921a7Sdan int iEndTrigger = 0; /* Label to jump to if WHEN is false */ 797165921a7Sdan 79865a7cd16Sdan assert( pTab==tableOfTrigger(pTrigger) ); 79965a7cd16Sdan 80065a7cd16Sdan /* Allocate the TriggerPrg and SubProgram objects. To ensure that they 80165a7cd16Sdan ** are freed if an error occurs, link them into the Parse.pTriggerPrg 80265a7cd16Sdan ** list of the top-level Parse object sooner rather than later. */ 8032832ad42Sdan pPrg = sqlite3DbMallocZero(db, sizeof(TriggerPrg)); 8042832ad42Sdan if( !pPrg ) return 0; 80565a7cd16Sdan pPrg->pNext = pTop->pTriggerPrg; 80665a7cd16Sdan pTop->pTriggerPrg = pPrg; 8072832ad42Sdan pPrg->pProgram = pProgram = sqlite3DbMallocZero(db, sizeof(SubProgram)); 808165921a7Sdan if( !pProgram ) return 0; 809165921a7Sdan pProgram->nRef = 1; 8102832ad42Sdan pPrg->pTrigger = pTrigger; 8112832ad42Sdan pPrg->orconf = orconf; 812165921a7Sdan 81365a7cd16Sdan /* Allocate and populate a new Parse context to use for coding the 81465a7cd16Sdan ** trigger sub-program. */ 81565a7cd16Sdan pSubParse = sqlite3StackAllocZero(db, sizeof(Parse)); 81665a7cd16Sdan if( !pSubParse ) return 0; 817165921a7Sdan memset(&sNC, 0, sizeof(sNC)); 818165921a7Sdan sNC.pParse = pSubParse; 819165921a7Sdan pSubParse->db = db; 820165921a7Sdan pSubParse->pTriggerTab = pTab; 82165a7cd16Sdan pSubParse->pToplevel = pTop; 8222bd93516Sdan pSubParse->zAuthContext = pTrigger->zName; 82365a7cd16Sdan pSubParse->eTriggerOp = pTrigger->op; 824165921a7Sdan 825165921a7Sdan v = sqlite3GetVdbe(pSubParse); 826165921a7Sdan if( v ){ 82776d462eeSdan VdbeComment((v, "Start: %s.%s (%s %s%s%s ON %s)", 82876d462eeSdan pTrigger->zName, onErrorText(orconf), 829165921a7Sdan (pTrigger->tr_tm==TRIGGER_BEFORE ? "BEFORE" : "AFTER"), 83065a7cd16Sdan (pTrigger->op==TK_UPDATE ? "UPDATE" : ""), 83165a7cd16Sdan (pTrigger->op==TK_INSERT ? "INSERT" : ""), 83265a7cd16Sdan (pTrigger->op==TK_DELETE ? "DELETE" : ""), 83376d462eeSdan pTab->zName 834165921a7Sdan )); 83576d462eeSdan #ifndef SQLITE_OMIT_TRACE 83676d462eeSdan sqlite3VdbeChangeP4(v, -1, 83776d462eeSdan sqlite3MPrintf(db, "-- TRIGGER %s", pTrigger->zName), P4_DYNAMIC 83876d462eeSdan ); 83976d462eeSdan #endif 840165921a7Sdan 84165a7cd16Sdan /* If one was specified, code the WHEN clause. If it evaluates to false 84265a7cd16Sdan ** (or NULL) the sub-vdbe is immediately halted by jumping to the 84365a7cd16Sdan ** OP_Halt inserted at the end of the program. */ 844165921a7Sdan if( pTrigger->pWhen ){ 845165921a7Sdan pWhen = sqlite3ExprDup(db, pTrigger->pWhen, 0); 846523a087bSdan if( SQLITE_OK==sqlite3ResolveExprNames(&sNC, pWhen) 847523a087bSdan && db->mallocFailed==0 848523a087bSdan ){ 849165921a7Sdan iEndTrigger = sqlite3VdbeMakeLabel(v); 850165921a7Sdan sqlite3ExprIfFalse(pSubParse, pWhen, iEndTrigger, SQLITE_JUMPIFNULL); 851165921a7Sdan } 852165921a7Sdan sqlite3ExprDelete(db, pWhen); 853165921a7Sdan } 854165921a7Sdan 855165921a7Sdan /* Code the trigger program into the sub-vdbe. */ 85676d462eeSdan codeTriggerProgram(pSubParse, pTrigger->step_list, orconf); 85765a7cd16Sdan 85865a7cd16Sdan /* Insert an OP_Halt at the end of the sub-program. */ 859165921a7Sdan if( iEndTrigger ){ 860165921a7Sdan sqlite3VdbeResolveLabel(v, iEndTrigger); 861165921a7Sdan } 862165921a7Sdan sqlite3VdbeAddOp0(v, OP_Halt); 86376d462eeSdan VdbeComment((v, "End: %s.%s", pTrigger->zName, onErrorText(orconf))); 86476d462eeSdan 865165921a7Sdan transferParseError(pParse, pSubParse); 866523a087bSdan if( db->mallocFailed==0 ){ 86765a7cd16Sdan pProgram->aOp = sqlite3VdbeTakeOpArray(v, &pProgram->nOp, &pTop->nMaxArg); 868523a087bSdan } 869165921a7Sdan pProgram->nMem = pSubParse->nMem; 870165921a7Sdan pProgram->nCsr = pSubParse->nTab; 871165921a7Sdan pProgram->token = (void *)pTrigger; 8722832ad42Sdan pPrg->oldmask = pSubParse->oldmask; 873165921a7Sdan sqlite3VdbeDelete(v); 87465a7cd16Sdan } 875165921a7Sdan 87665a7cd16Sdan assert( !pSubParse->pAinc && !pSubParse->pZombieTab ); 87765a7cd16Sdan assert( !pSubParse->pTriggerPrg && !pSubParse->nMaxArg ); 878165921a7Sdan sqlite3StackFree(db, pSubParse); 879165921a7Sdan 8802832ad42Sdan return pPrg; 881165921a7Sdan } 882165921a7Sdan 88365a7cd16Sdan /* 88465a7cd16Sdan ** Return a pointer to a TriggerPrg object containing the sub-program for 88565a7cd16Sdan ** trigger pTrigger with default ON CONFLICT algorithm orconf. If no such 88665a7cd16Sdan ** TriggerPrg object exists, a new object is allocated and populated before 88765a7cd16Sdan ** being returned. 88865a7cd16Sdan */ 8892832ad42Sdan static TriggerPrg *getRowTrigger( 89065a7cd16Sdan Parse *pParse, /* Current parse context */ 891165921a7Sdan Trigger *pTrigger, /* Trigger to code */ 89265a7cd16Sdan Table *pTab, /* The table trigger pTrigger is attached to */ 89365a7cd16Sdan int orconf /* ON CONFLICT algorithm. */ 894165921a7Sdan ){ 89565a7cd16Sdan Parse *pRoot = sqlite3ParseToplevel(pParse); 8962832ad42Sdan TriggerPrg *pPrg; 89765a7cd16Sdan 89865a7cd16Sdan assert( pTab==tableOfTrigger(pTrigger) ); 899165921a7Sdan 900165921a7Sdan /* It may be that this trigger has already been coded (or is in the 901165921a7Sdan ** process of being coded). If this is the case, then an entry with 9022832ad42Sdan ** a matching TriggerPrg.pTrigger field will be present somewhere 9032832ad42Sdan ** in the Parse.pTriggerPrg list. Search for such an entry. */ 9042832ad42Sdan for(pPrg=pRoot->pTriggerPrg; 9052832ad42Sdan pPrg && (pPrg->pTrigger!=pTrigger || pPrg->orconf!=orconf); 9062832ad42Sdan pPrg=pPrg->pNext 907165921a7Sdan ); 908165921a7Sdan 90965a7cd16Sdan /* If an existing TriggerPrg could not be located, create a new one. */ 9102832ad42Sdan if( !pPrg ){ 91165a7cd16Sdan pPrg = codeRowTrigger(pParse, pTrigger, pTab, orconf); 912165921a7Sdan } 913165921a7Sdan 9142832ad42Sdan return pPrg; 915165921a7Sdan } 916165921a7Sdan 917633ed08dSdanielk1977 /* 918633ed08dSdanielk1977 ** This is called to code FOR EACH ROW triggers. 919633ed08dSdanielk1977 ** 920633ed08dSdanielk1977 ** When the code that this function generates is executed, the following 921633ed08dSdanielk1977 ** must be true: 922c977f7f5Sdrh ** 923c977f7f5Sdrh ** 1. No cursors may be open in the main database. (But newIdx and oldIdx 924c977f7f5Sdrh ** can be indices of cursors in temporary tables. See below.) 925c977f7f5Sdrh ** 926633ed08dSdanielk1977 ** 2. If the triggers being coded are ON INSERT or ON UPDATE triggers, then 927633ed08dSdanielk1977 ** a temporary vdbe cursor (index newIdx) must be open and pointing at 928633ed08dSdanielk1977 ** a row containing values to be substituted for new.* expressions in the 929633ed08dSdanielk1977 ** trigger program(s). 930c977f7f5Sdrh ** 931633ed08dSdanielk1977 ** 3. If the triggers being coded are ON DELETE or ON UPDATE triggers, then 932633ed08dSdanielk1977 ** a temporary vdbe cursor (index oldIdx) must be open and pointing at 933633ed08dSdanielk1977 ** a row containing values to be substituted for old.* expressions in the 934633ed08dSdanielk1977 ** trigger program(s). 935633ed08dSdanielk1977 ** 9368f2c54e6Sdanielk1977 ** If they are not NULL, the piOldColMask and piNewColMask output variables 9378f2c54e6Sdanielk1977 ** are set to values that describe the columns used by the trigger program 9388f2c54e6Sdanielk1977 ** in the OLD.* and NEW.* tables respectively. If column N of the 9398f2c54e6Sdanielk1977 ** pseudo-table is read at least once, the corresponding bit of the output 9408f2c54e6Sdanielk1977 ** mask is set. If a column with an index greater than 32 is read, the 9418f2c54e6Sdanielk1977 ** output mask is set to the special value 0xffffffff. 9428f2c54e6Sdanielk1977 ** 943633ed08dSdanielk1977 */ 944165921a7Sdan void sqlite3CodeRowTrigger( 945c3f9bad2Sdanielk1977 Parse *pParse, /* Parse context */ 9462f886d1dSdanielk1977 Trigger *pTrigger, /* List of triggers on table pTab */ 947c3f9bad2Sdanielk1977 int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 948633ed08dSdanielk1977 ExprList *pChanges, /* Changes list for any UPDATE OF triggers */ 949dca76841Sdrh int tr_tm, /* One of TRIGGER_BEFORE, TRIGGER_AFTER */ 950633ed08dSdanielk1977 Table *pTab, /* The table to code triggers from */ 951633ed08dSdanielk1977 int newIdx, /* The indice of the "new" row to access */ 952633ed08dSdanielk1977 int oldIdx, /* The indice of the "old" row to access */ 9536f34903eSdanielk1977 int orconf, /* ON CONFLICT policy */ 954165921a7Sdan int ignoreJump /* Instruction to jump to for RAISE(IGNORE) */ 955c977f7f5Sdrh ){ 956eecfb3eeSdanielk1977 Trigger *p; 9578f2c54e6Sdanielk1977 958*cea72b2dSshane UNUSED_PARAMETER(newIdx); 959*cea72b2dSshane 960c3f9bad2Sdanielk1977 assert(op == TK_UPDATE || op == TK_INSERT || op == TK_DELETE); 961dca76841Sdrh assert(tr_tm == TRIGGER_BEFORE || tr_tm == TRIGGER_AFTER ); 962c3f9bad2Sdanielk1977 9632f886d1dSdanielk1977 for(p=pTrigger; p; p=p->pNext){ 964c3f9bad2Sdanielk1977 9659ab4c2e8Sdrh /* Sanity checking: The schema for the trigger and for the table are 9669ab4c2e8Sdrh ** always defined. The trigger must be in the same schema as the table 9679ab4c2e8Sdrh ** or else it must be a TEMP trigger. */ 9689ab4c2e8Sdrh assert( p->pSchema!=0 ); 9699ab4c2e8Sdrh assert( p->pTabSchema!=0 ); 970165921a7Sdan assert( p->pSchema==p->pTabSchema 971165921a7Sdan || p->pSchema==pParse->db->aDb[1].pSchema ); 9729ab4c2e8Sdrh 973eecfb3eeSdanielk1977 /* Determine whether we should code this trigger */ 974165921a7Sdan if( p->op==op 975165921a7Sdan && p->tr_tm==tr_tm 976165921a7Sdan && checkColumnOverlap(p->pColumns,pChanges) 977eecfb3eeSdanielk1977 ){ 978165921a7Sdan Vdbe *v = sqlite3GetVdbe(pParse); /* Main VM */ 9792832ad42Sdan TriggerPrg *pPrg; 98065a7cd16Sdan pPrg = getRowTrigger(pParse, p, pTab, orconf); 9812832ad42Sdan assert( pPrg || pParse->nErr || pParse->db->mallocFailed ); 982c3f9bad2Sdanielk1977 983165921a7Sdan /* Code the OP_Program opcode in the parent VDBE. P4 of the OP_Program 984165921a7Sdan ** is a pointer to the sub-vdbe containing the trigger program. */ 9852832ad42Sdan if( pPrg ){ 98676d462eeSdan sqlite3VdbeAddOp3(v, OP_Program, oldIdx, ignoreJump, ++pParse->nMem); 9872832ad42Sdan pPrg->pProgram->nRef++; 9882832ad42Sdan sqlite3VdbeChangeP4(v, -1, (const char *)pPrg->pProgram, P4_SUBPROGRAM); 98976d462eeSdan VdbeComment((v, "Call: %s.%s", p->zName, onErrorText(orconf))); 990c3f9bad2Sdanielk1977 } 991c3f9bad2Sdanielk1977 } 992c3f9bad2Sdanielk1977 } 993165921a7Sdan } 994165921a7Sdan 9952832ad42Sdan /* 9962832ad42Sdan ** Triggers fired by UPDATE or DELETE statements may access values stored 9972832ad42Sdan ** in the old.* pseudo-table. This function returns a 32-bit bitmask 9982832ad42Sdan ** indicating which columns of the old.* table actually are used by 9992832ad42Sdan ** triggers. This information may be used by the caller to avoid having 10002832ad42Sdan ** to load the entire old.* record into memory when executing an UPDATE 10012832ad42Sdan ** or DELETE command. 10022832ad42Sdan ** 10032832ad42Sdan ** Bit 0 of the returned mask is set if the left-most column of the 10042832ad42Sdan ** table may be accessed using an old.<col> reference. Bit 1 is set if 10052832ad42Sdan ** the second leftmost column value is required, and so on. If there 10062832ad42Sdan ** are more than 32 columns in the table, and at least one of the columns 10072832ad42Sdan ** with an index greater than 32 may be accessed, 0xffffffff is returned. 10082832ad42Sdan ** 10092832ad42Sdan ** It is not possible to determine if the old.rowid column is accessed 10102832ad42Sdan ** by triggers. The caller must always assume that it is. 10112832ad42Sdan ** 10122832ad42Sdan ** There is no equivalent function for new.* references. 10132832ad42Sdan */ 10142832ad42Sdan u32 sqlite3TriggerOldmask( 1015165921a7Sdan Parse *pParse, /* Parse context */ 1016165921a7Sdan Trigger *pTrigger, /* List of triggers on table pTab */ 10172832ad42Sdan int op, /* Either TK_UPDATE or TK_DELETE */ 1018165921a7Sdan ExprList *pChanges, /* Changes list for any UPDATE OF triggers */ 1019165921a7Sdan Table *pTab, /* The table to code triggers from */ 10202832ad42Sdan int orconf /* Default ON CONFLICT policy for trigger steps */ 1021165921a7Sdan ){ 10222832ad42Sdan u32 mask = 0; 1023165921a7Sdan Trigger *p; 1024165921a7Sdan 10252832ad42Sdan assert(op==TK_UPDATE || op==TK_DELETE); 1026165921a7Sdan for(p=pTrigger; p; p=p->pNext){ 1027165921a7Sdan if( p->op==op && checkColumnOverlap(p->pColumns,pChanges) ){ 10282832ad42Sdan TriggerPrg *pPrg; 102965a7cd16Sdan pPrg = getRowTrigger(pParse, p, pTab, orconf); 10302832ad42Sdan if( pPrg ){ 10312832ad42Sdan mask |= pPrg->oldmask; 1032165921a7Sdan } 1033165921a7Sdan } 1034165921a7Sdan } 10352832ad42Sdan 10362832ad42Sdan return mask; 1037165921a7Sdan } 1038165921a7Sdan 1039b7f9164eSdrh #endif /* !defined(SQLITE_OMIT_TRIGGER) */ 1040