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; 5344adee20fSdanielk1977 if( sqlite3AuthCheck(pParse, code, pTrigger->name, 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 /* 669c977f7f5Sdrh ** Generate VDBE code for zero or more statements inside the body of a 670c977f7f5Sdrh ** trigger. 671c977f7f5Sdrh */ 672c3f9bad2Sdanielk1977 static int codeTriggerProgram( 673c977f7f5Sdrh Parse *pParse, /* The parser context */ 674c977f7f5Sdrh TriggerStep *pStepList, /* List of statements inside the trigger body */ 675c977f7f5Sdrh int orconfin /* Conflict algorithm. (OE_Abort, etc) */ 676633ed08dSdanielk1977 ){ 677165921a7Sdan TriggerStep * pStep = pStepList; 678344737f6Sdrh Vdbe *v = pParse->pVdbe; 67917435752Sdrh sqlite3 *db = pParse->db; 680c3f9bad2Sdanielk1977 681165921a7Sdan assert( pParse->pRoot ); 682165921a7Sdan assert( pStep!=0 ); 683344737f6Sdrh assert( v!=0 ); 684165921a7Sdan /* sqlite3VdbeAddOp2(v, OP_ContextPush, 0, 0); */ 685165921a7Sdan while( pStep ){ 686165921a7Sdan /* Figure out the ON CONFLICT policy that will be used for this step 687165921a7Sdan ** of the trigger program. If the statement that caused this trigger 688165921a7Sdan ** to fire had an explicit ON CONFLICT, then use it. Otherwise, use 689165921a7Sdan ** the ON CONFLICT policy that was specified as part of the trigger 690165921a7Sdan ** step statement. Example: 691165921a7Sdan ** 692165921a7Sdan ** CREATE TRIGGER AFTER INSERT ON t1 BEGIN; 693165921a7Sdan ** INSERT OR REPLACE INTO t2 VALUES(new.a, new.b); 694165921a7Sdan ** END; 695165921a7Sdan ** 696165921a7Sdan ** INSERT INTO t1 ... ; -- insert into t2 uses REPLACE policy 697165921a7Sdan ** INSERT OR IGNORE INTO t1 ... ; -- insert into t2 uses IGNORE policy 698165921a7Sdan */ 699165921a7Sdan pParse->orconf = (orconfin==OE_Default)?pStep->orconf:orconfin; 700165921a7Sdan 701*76d462eeSdan if( pStep->op!=TK_SELECT ){ 702*76d462eeSdan sqlite3VdbeAddOp1(v, OP_ResetCount, 0); 703*76d462eeSdan } 704*76d462eeSdan 705165921a7Sdan switch( pStep->op ){ 706c3f9bad2Sdanielk1977 case TK_UPDATE: { 707165921a7Sdan sqlite3Update(pParse, 708165921a7Sdan targetSrcList(pParse, pStep), 709165921a7Sdan sqlite3ExprListDup(db, pStep->pExprList, 0), 710165921a7Sdan sqlite3ExprDup(db, pStep->pWhere, 0), 711165921a7Sdan pParse->orconf 712165921a7Sdan ); 713c3f9bad2Sdanielk1977 break; 714c3f9bad2Sdanielk1977 } 715c3f9bad2Sdanielk1977 case TK_INSERT: { 716165921a7Sdan sqlite3Insert(pParse, 717165921a7Sdan targetSrcList(pParse, pStep), 718165921a7Sdan sqlite3ExprListDup(db, pStep->pExprList, 0), 719165921a7Sdan sqlite3SelectDup(db, pStep->pSelect, 0), 720165921a7Sdan sqlite3IdListDup(db, pStep->pIdList), 721165921a7Sdan pParse->orconf 722165921a7Sdan ); 723c3f9bad2Sdanielk1977 break; 724c3f9bad2Sdanielk1977 } 725c3f9bad2Sdanielk1977 case TK_DELETE: { 726165921a7Sdan sqlite3DeleteFrom(pParse, 727165921a7Sdan targetSrcList(pParse, pStep), 728165921a7Sdan sqlite3ExprDup(db, pStep->pWhere, 0) 729165921a7Sdan ); 730c3f9bad2Sdanielk1977 break; 731c3f9bad2Sdanielk1977 } 732165921a7Sdan default: assert( pStep->op==TK_SELECT ); { 733165921a7Sdan SelectDest sDest; 734165921a7Sdan Select *pSelect = sqlite3SelectDup(db, pStep->pSelect, 0); 735165921a7Sdan sqlite3SelectDestInit(&sDest, SRT_Discard, 0); 736165921a7Sdan sqlite3Select(pParse, pSelect, &sDest); 737165921a7Sdan sqlite3SelectDelete(db, pSelect); 7389ab4c2e8Sdrh break; 7399ab4c2e8Sdrh } 740c3f9bad2Sdanielk1977 } 741*76d462eeSdan if( pStep->op!=TK_SELECT ){ 742*76d462eeSdan sqlite3VdbeAddOp1(v, OP_ResetCount, 1); 743*76d462eeSdan } 744165921a7Sdan pStep = pStep->pNext; 745c3f9bad2Sdanielk1977 } 746165921a7Sdan /* sqlite3VdbeAddOp2(v, OP_ContextPop, 0, 0); */ 747c3f9bad2Sdanielk1977 748c3f9bad2Sdanielk1977 return 0; 749c3f9bad2Sdanielk1977 } 750c3f9bad2Sdanielk1977 751165921a7Sdan #ifdef SQLITE_DEBUG 752165921a7Sdan /* 753165921a7Sdan ** This function is used to add VdbeComment() annotations to a VDBE 754165921a7Sdan ** program. It is not used in production code, only for debugging. 755165921a7Sdan */ 756165921a7Sdan static const char *onErrorText(int onError){ 757165921a7Sdan switch( onError ){ 758165921a7Sdan case OE_Abort: return "abort"; 759165921a7Sdan case OE_Rollback: return "rollback"; 760165921a7Sdan case OE_Fail: return "fail"; 761165921a7Sdan case OE_Replace: return "replace"; 762165921a7Sdan case OE_Ignore: return "ignore"; 763165921a7Sdan case OE_Default: return "default"; 764165921a7Sdan } 765165921a7Sdan return "n/a"; 766165921a7Sdan } 767165921a7Sdan #endif 768165921a7Sdan 769165921a7Sdan /* 770165921a7Sdan ** Parse context structure pFrom has just been used to create a sub-vdbe 771165921a7Sdan ** (trigger program). If an error has occurred, transfer error information 772165921a7Sdan ** from pFrom to pTo. 773165921a7Sdan */ 774165921a7Sdan static void transferParseError(Parse *pTo, Parse *pFrom){ 775165921a7Sdan assert( pFrom->zErrMsg==0 || pFrom->nErr ); 776165921a7Sdan assert( pTo->zErrMsg==0 || pTo->nErr ); 777165921a7Sdan if( pTo->nErr==0 ){ 778165921a7Sdan pTo->zErrMsg = pFrom->zErrMsg; 779165921a7Sdan pTo->nErr = pFrom->nErr; 780165921a7Sdan }else{ 781165921a7Sdan sqlite3DbFree(pFrom->db, pFrom->zErrMsg); 782165921a7Sdan } 783165921a7Sdan } 784165921a7Sdan 785165921a7Sdan static CodedTrigger *codeRowTrigger( 786165921a7Sdan Parse *pRoot, /* Root parse context */ 787165921a7Sdan Parse *pParse, /* Current parse context */ 788165921a7Sdan Trigger *pTrigger, /* Trigger to code */ 789165921a7Sdan int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 790165921a7Sdan Table *pTab, /* The table to code triggers from */ 791165921a7Sdan int orconf 792165921a7Sdan ){ 793165921a7Sdan sqlite3 *db = pParse->db; 794165921a7Sdan CodedTrigger *pC; 795165921a7Sdan Expr *pWhen = 0; /* Duplicate of trigger WHEN expression */ 796165921a7Sdan Vdbe *v; /* Temporary VM */ 797165921a7Sdan AuthContext sContext; /* Auth context for sub-vdbe */ 798165921a7Sdan NameContext sNC; /* Name context for sub-vdbe */ 799165921a7Sdan SubProgram *pProgram = 0; /* Sub-vdbe for trigger program */ 800165921a7Sdan Parse *pSubParse; /* Parse context for sub-vdbe */ 801165921a7Sdan int iEndTrigger = 0; /* Label to jump to if WHEN is false */ 802165921a7Sdan 803165921a7Sdan pC = sqlite3DbMallocZero(db, sizeof(CodedTrigger)); 804165921a7Sdan if( !pC ) return 0; 805165921a7Sdan pC->pNext = pRoot->pCodedTrigger; 806165921a7Sdan pRoot->pCodedTrigger = pC; 807165921a7Sdan pC->pProgram = pProgram = sqlite3DbMallocZero(db, sizeof(SubProgram)); 808165921a7Sdan if( !pProgram ) return 0; 809165921a7Sdan pProgram->nRef = 1; 810165921a7Sdan pSubParse = sqlite3StackAllocZero(db, sizeof(Parse)); 811165921a7Sdan if( !pSubParse ) return 0; 812165921a7Sdan 813165921a7Sdan pC->pProgram = pProgram; 814165921a7Sdan pC->pTrigger = pTrigger; 815165921a7Sdan pC->orconf = orconf; 816165921a7Sdan 817165921a7Sdan memset(&sNC, 0, sizeof(sNC)); 818165921a7Sdan sNC.pParse = pSubParse; 819165921a7Sdan pSubParse->db = db; 820165921a7Sdan pSubParse->pTriggerTab = pTab; 821165921a7Sdan pSubParse->pRoot = pRoot; 822165921a7Sdan 823165921a7Sdan /* Push an entry on to the auth context stack */ 824165921a7Sdan sqlite3AuthContextPush(pParse, &sContext, pTrigger->name); 825165921a7Sdan 826165921a7Sdan v = sqlite3GetVdbe(pSubParse); 827165921a7Sdan if( v ){ 828*76d462eeSdan VdbeComment((v, "Start: %s.%s (%s %s%s%s ON %s)", 829*76d462eeSdan pTrigger->zName, onErrorText(orconf), 830165921a7Sdan (pTrigger->tr_tm==TRIGGER_BEFORE ? "BEFORE" : "AFTER"), 831165921a7Sdan (op==TK_UPDATE ? "UPDATE" : ""), 832165921a7Sdan (op==TK_INSERT ? "INSERT" : ""), 833165921a7Sdan (op==TK_DELETE ? "DELETE" : ""), 834*76d462eeSdan pTab->zName 835165921a7Sdan )); 836*76d462eeSdan #ifndef SQLITE_OMIT_TRACE 837*76d462eeSdan sqlite3VdbeChangeP4(v, -1, 838*76d462eeSdan sqlite3MPrintf(db, "-- TRIGGER %s", pTrigger->zName), P4_DYNAMIC 839*76d462eeSdan ); 840*76d462eeSdan #endif 841165921a7Sdan 842165921a7Sdan if( pTrigger->pWhen ){ 843165921a7Sdan /* Code the WHEN clause. If it evaluates to false (or NULL) the 844165921a7Sdan ** sub-vdbe is immediately halted. */ 845165921a7Sdan pWhen = sqlite3ExprDup(db, pTrigger->pWhen, 0); 846165921a7Sdan if( SQLITE_OK==sqlite3ResolveExprNames(&sNC, pWhen) ){ 847165921a7Sdan iEndTrigger = sqlite3VdbeMakeLabel(v); 848165921a7Sdan sqlite3ExprIfFalse(pSubParse, pWhen, iEndTrigger, SQLITE_JUMPIFNULL); 849165921a7Sdan } 850165921a7Sdan sqlite3ExprDelete(db, pWhen); 851165921a7Sdan } 852165921a7Sdan 853165921a7Sdan /* Code the trigger program into the sub-vdbe. */ 854*76d462eeSdan codeTriggerProgram(pSubParse, pTrigger->step_list, orconf); 855165921a7Sdan if( iEndTrigger ){ 856165921a7Sdan sqlite3VdbeResolveLabel(v, iEndTrigger); 857165921a7Sdan } 858165921a7Sdan sqlite3VdbeAddOp0(v, OP_Halt); 859*76d462eeSdan VdbeComment((v, "End: %s.%s", pTrigger->zName, onErrorText(orconf))); 860*76d462eeSdan 861165921a7Sdan transferParseError(pParse, pSubParse); 862165921a7Sdan pProgram->aOp = sqlite3VdbeTakeOpArray(v, &pProgram->nOp, &pParse->nArg); 863165921a7Sdan pProgram->nMem = pSubParse->nMem; 864165921a7Sdan pProgram->nCsr = pSubParse->nTab; 865165921a7Sdan pProgram->token = (void *)pTrigger; 866165921a7Sdan pC->oldmask = pSubParse->oldmask; 867165921a7Sdan pC->newmask = pSubParse->newmask; 868165921a7Sdan sqlite3VdbeDelete(v); 869165921a7Sdan 870165921a7Sdan while( pSubParse->pAinc ){ 871165921a7Sdan AutoincInfo *p = pSubParse->pAinc; 872165921a7Sdan pSubParse->pAinc = p->pNext; 873165921a7Sdan sqlite3DbFree(db, p); 874165921a7Sdan } 875165921a7Sdan } 876165921a7Sdan sqlite3StackFree(db, pSubParse); 877165921a7Sdan 878165921a7Sdan /* Pop the entry off the authorization stack */ 879165921a7Sdan sqlite3AuthContextPop(&sContext); 880165921a7Sdan return pC; 881165921a7Sdan } 882165921a7Sdan 883165921a7Sdan static CodedTrigger *getRowTrigger( 884165921a7Sdan Parse *pParse, 885165921a7Sdan Trigger *pTrigger, /* Trigger to code */ 886165921a7Sdan int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 887165921a7Sdan Table *pTab, /* The table to code triggers from */ 888165921a7Sdan int orconf 889165921a7Sdan ){ 890165921a7Sdan CodedTrigger *pC; 891165921a7Sdan Parse *pRoot = pParse; 892165921a7Sdan 893165921a7Sdan /* It may be that this trigger has already been coded (or is in the 894165921a7Sdan ** process of being coded). If this is the case, then an entry with 895165921a7Sdan ** a matching CodedTrigger.pTrigger field will be present somewhere 896165921a7Sdan ** in the Parse.pCodedTrigger list. Search for such an entry. */ 897165921a7Sdan if( pParse->pRoot ){ 898165921a7Sdan pRoot = pParse->pRoot; 899165921a7Sdan } 900165921a7Sdan for(pC=pRoot->pCodedTrigger; 901165921a7Sdan pC && (pC->pTrigger!=pTrigger || pC->orconf!=orconf); 902165921a7Sdan pC=pC->pNext 903165921a7Sdan ); 904165921a7Sdan 905165921a7Sdan if( !pC ){ 906165921a7Sdan pC = codeRowTrigger(pRoot, pParse, pTrigger, op, pTab, orconf); 907165921a7Sdan } 908165921a7Sdan 909165921a7Sdan return pC; 910165921a7Sdan } 911165921a7Sdan 912633ed08dSdanielk1977 /* 913633ed08dSdanielk1977 ** This is called to code FOR EACH ROW triggers. 914633ed08dSdanielk1977 ** 915633ed08dSdanielk1977 ** When the code that this function generates is executed, the following 916633ed08dSdanielk1977 ** must be true: 917c977f7f5Sdrh ** 918c977f7f5Sdrh ** 1. No cursors may be open in the main database. (But newIdx and oldIdx 919c977f7f5Sdrh ** can be indices of cursors in temporary tables. See below.) 920c977f7f5Sdrh ** 921633ed08dSdanielk1977 ** 2. If the triggers being coded are ON INSERT or ON UPDATE triggers, then 922633ed08dSdanielk1977 ** a temporary vdbe cursor (index newIdx) must be open and pointing at 923633ed08dSdanielk1977 ** a row containing values to be substituted for new.* expressions in the 924633ed08dSdanielk1977 ** trigger program(s). 925c977f7f5Sdrh ** 926633ed08dSdanielk1977 ** 3. If the triggers being coded are ON DELETE or ON UPDATE triggers, then 927633ed08dSdanielk1977 ** a temporary vdbe cursor (index oldIdx) must be open and pointing at 928633ed08dSdanielk1977 ** a row containing values to be substituted for old.* expressions in the 929633ed08dSdanielk1977 ** trigger program(s). 930633ed08dSdanielk1977 ** 9318f2c54e6Sdanielk1977 ** If they are not NULL, the piOldColMask and piNewColMask output variables 9328f2c54e6Sdanielk1977 ** are set to values that describe the columns used by the trigger program 9338f2c54e6Sdanielk1977 ** in the OLD.* and NEW.* tables respectively. If column N of the 9348f2c54e6Sdanielk1977 ** pseudo-table is read at least once, the corresponding bit of the output 9358f2c54e6Sdanielk1977 ** mask is set. If a column with an index greater than 32 is read, the 9368f2c54e6Sdanielk1977 ** output mask is set to the special value 0xffffffff. 9378f2c54e6Sdanielk1977 ** 938633ed08dSdanielk1977 */ 939165921a7Sdan void sqlite3CodeRowTrigger( 940c3f9bad2Sdanielk1977 Parse *pParse, /* Parse context */ 9412f886d1dSdanielk1977 Trigger *pTrigger, /* List of triggers on table pTab */ 942c3f9bad2Sdanielk1977 int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 943633ed08dSdanielk1977 ExprList *pChanges, /* Changes list for any UPDATE OF triggers */ 944dca76841Sdrh int tr_tm, /* One of TRIGGER_BEFORE, TRIGGER_AFTER */ 945633ed08dSdanielk1977 Table *pTab, /* The table to code triggers from */ 946633ed08dSdanielk1977 int newIdx, /* The indice of the "new" row to access */ 947633ed08dSdanielk1977 int oldIdx, /* The indice of the "old" row to access */ 9486f34903eSdanielk1977 int orconf, /* ON CONFLICT policy */ 949165921a7Sdan int ignoreJump /* Instruction to jump to for RAISE(IGNORE) */ 950c977f7f5Sdrh ){ 951eecfb3eeSdanielk1977 Trigger *p; 9528f2c54e6Sdanielk1977 953c3f9bad2Sdanielk1977 assert(op == TK_UPDATE || op == TK_INSERT || op == TK_DELETE); 954dca76841Sdrh assert(tr_tm == TRIGGER_BEFORE || tr_tm == TRIGGER_AFTER ); 955c3f9bad2Sdanielk1977 9562f886d1dSdanielk1977 for(p=pTrigger; p; p=p->pNext){ 957c3f9bad2Sdanielk1977 9589ab4c2e8Sdrh /* Sanity checking: The schema for the trigger and for the table are 9599ab4c2e8Sdrh ** always defined. The trigger must be in the same schema as the table 9609ab4c2e8Sdrh ** or else it must be a TEMP trigger. */ 9619ab4c2e8Sdrh assert( p->pSchema!=0 ); 9629ab4c2e8Sdrh assert( p->pTabSchema!=0 ); 963165921a7Sdan assert( p->pSchema==p->pTabSchema 964165921a7Sdan || p->pSchema==pParse->db->aDb[1].pSchema ); 9659ab4c2e8Sdrh 966eecfb3eeSdanielk1977 /* Determine whether we should code this trigger */ 967165921a7Sdan if( p->op==op 968165921a7Sdan && p->tr_tm==tr_tm 969165921a7Sdan && checkColumnOverlap(p->pColumns,pChanges) 970eecfb3eeSdanielk1977 ){ 971165921a7Sdan Vdbe *v = sqlite3GetVdbe(pParse); /* Main VM */ 972165921a7Sdan CodedTrigger *pC; 973165921a7Sdan pC = getRowTrigger(pParse, p, op, pTab, orconf); 974165921a7Sdan assert( pC || pParse->nErr || pParse->db->mallocFailed ); 975c3f9bad2Sdanielk1977 976165921a7Sdan /* Code the OP_Program opcode in the parent VDBE. P4 of the OP_Program 977165921a7Sdan ** is a pointer to the sub-vdbe containing the trigger program. */ 978165921a7Sdan if( pC ){ 979*76d462eeSdan sqlite3VdbeAddOp3(v, OP_Program, oldIdx, ignoreJump, ++pParse->nMem); 980165921a7Sdan pC->pProgram->nRef++; 981165921a7Sdan sqlite3VdbeChangeP4(v, -1, (const char *)pC->pProgram, P4_SUBPROGRAM); 982*76d462eeSdan VdbeComment((v, "Call: %s.%s", p->zName, onErrorText(orconf))); 983c3f9bad2Sdanielk1977 } 984c3f9bad2Sdanielk1977 } 985c3f9bad2Sdanielk1977 } 986165921a7Sdan } 987165921a7Sdan 988165921a7Sdan void sqlite3TriggerUses( 989165921a7Sdan Parse *pParse, /* Parse context */ 990165921a7Sdan Trigger *pTrigger, /* List of triggers on table pTab */ 991165921a7Sdan int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 992165921a7Sdan ExprList *pChanges, /* Changes list for any UPDATE OF triggers */ 993165921a7Sdan Table *pTab, /* The table to code triggers from */ 994165921a7Sdan int orconf, /* Default ON CONFLICT policy for trigger steps */ 995165921a7Sdan u32 *piOldColMask, /* OUT: Mask of columns used from the OLD.* table */ 996165921a7Sdan u32 *piNewColMask /* OUT: Mask of columns used from the NEW.* table */ 997165921a7Sdan ){ 998165921a7Sdan Trigger *p; 999165921a7Sdan assert(op==TK_UPDATE || op==TK_INSERT || op==TK_DELETE); 1000165921a7Sdan 1001165921a7Sdan for(p=pTrigger; p; p=p->pNext){ 1002165921a7Sdan if( p->op==op && checkColumnOverlap(p->pColumns,pChanges) ){ 1003165921a7Sdan CodedTrigger *pC; 1004165921a7Sdan pC = getRowTrigger(pParse, p, op, pTab, orconf); 1005165921a7Sdan if( pC ){ 1006165921a7Sdan *piOldColMask |= pC->oldmask; 1007165921a7Sdan *piNewColMask |= pC->newmask; 1008165921a7Sdan } 1009165921a7Sdan } 1010165921a7Sdan } 1011165921a7Sdan } 1012165921a7Sdan 1013b7f9164eSdrh #endif /* !defined(SQLITE_OMIT_TRIGGER) */ 1014