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 ************************************************************************* 11633ed08dSdanielk1977 * 12c3f9bad2Sdanielk1977 */ 13c3f9bad2Sdanielk1977 #include "sqliteInt.h" 149adf9ac4Sdrh 15b7f9164eSdrh #ifndef SQLITE_OMIT_TRIGGER 16c3f9bad2Sdanielk1977 /* 174b59ab5eSdrh ** Delete a linked list of TriggerStep structures. 184b59ab5eSdrh */ 194adee20fSdanielk1977 void sqlite3DeleteTriggerStep(TriggerStep *pTriggerStep){ 204b59ab5eSdrh while( pTriggerStep ){ 214b59ab5eSdrh TriggerStep * pTmp = pTriggerStep; 224b59ab5eSdrh pTriggerStep = pTriggerStep->pNext; 234b59ab5eSdrh 2417435752Sdrh if( pTmp->target.dyn ) sqlite3_free((char*)pTmp->target.z); 254adee20fSdanielk1977 sqlite3ExprDelete(pTmp->pWhere); 264adee20fSdanielk1977 sqlite3ExprListDelete(pTmp->pExprList); 274adee20fSdanielk1977 sqlite3SelectDelete(pTmp->pSelect); 284adee20fSdanielk1977 sqlite3IdListDelete(pTmp->pIdList); 294b59ab5eSdrh 3017435752Sdrh sqlite3_free(pTmp); 314b59ab5eSdrh } 324b59ab5eSdrh } 334b59ab5eSdrh 344b59ab5eSdrh /* 35f0f258b1Sdrh ** This is called by the parser when it sees a CREATE TRIGGER statement 36f0f258b1Sdrh ** up to the point of the BEGIN before the trigger actions. A Trigger 37f0f258b1Sdrh ** structure is generated based on the information available and stored 38f0f258b1Sdrh ** in pParse->pNewTrigger. After the trigger actions have been parsed, the 394adee20fSdanielk1977 ** sqlite3FinishTrigger() function is called to complete the trigger 40f0f258b1Sdrh ** construction process. 41c3f9bad2Sdanielk1977 */ 424adee20fSdanielk1977 void sqlite3BeginTrigger( 43c3f9bad2Sdanielk1977 Parse *pParse, /* The parse context of the CREATE TRIGGER statement */ 44ef2cb63eSdanielk1977 Token *pName1, /* The name of the trigger */ 45ef2cb63eSdanielk1977 Token *pName2, /* The name of the trigger */ 46d702fccbSdanielk1977 int tr_tm, /* One of TK_BEFORE, TK_AFTER, TK_INSTEAD */ 47c3f9bad2Sdanielk1977 int op, /* One of TK_INSERT, TK_UPDATE, TK_DELETE */ 48633ed08dSdanielk1977 IdList *pColumns, /* column list if this is an UPDATE OF trigger */ 49d24cc427Sdrh SrcList *pTableName,/* The name of the table/view the trigger applies to */ 50c3f9bad2Sdanielk1977 Expr *pWhen, /* WHEN clause */ 51fdd48a76Sdrh int isTemp, /* True if the TEMPORARY keyword is present */ 52fdd48a76Sdrh int noErr /* Suppress errors if the trigger already exists */ 539adf9ac4Sdrh ){ 54d5d56523Sdanielk1977 Trigger *pTrigger = 0; 55ef2cb63eSdanielk1977 Table *pTab; 56e5f9c644Sdrh char *zName = 0; /* Name of the trigger */ 579bb575fdSdrh sqlite3 *db = pParse->db; 58ef2cb63eSdanielk1977 int iDb; /* The database to store the trigger in */ 59ef2cb63eSdanielk1977 Token *pName; /* The unqualified db name */ 604312db55Sdrh DbFixer sFix; 61da184236Sdanielk1977 int iTabDb; 62ed6c8671Sdrh 6343617e9aSdrh assert( pName1!=0 ); /* pName1->z might be NULL, but not pName1 itself */ 6443617e9aSdrh assert( pName2!=0 ); 65ef2cb63eSdanielk1977 if( isTemp ){ 66ef2cb63eSdanielk1977 /* If TEMP was specified, then the trigger name may not be qualified. */ 6743617e9aSdrh if( pName2->n>0 ){ 68ef2cb63eSdanielk1977 sqlite3ErrorMsg(pParse, "temporary trigger may not have qualified name"); 69ef2cb63eSdanielk1977 goto trigger_cleanup; 70ef2cb63eSdanielk1977 } 71ef2cb63eSdanielk1977 iDb = 1; 72ef2cb63eSdanielk1977 pName = pName1; 73ef2cb63eSdanielk1977 }else{ 74ef2cb63eSdanielk1977 /* Figure out the db that the the trigger will be created in */ 75ef2cb63eSdanielk1977 iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pName); 76ef2cb63eSdanielk1977 if( iDb<0 ){ 77ef2cb63eSdanielk1977 goto trigger_cleanup; 78ef2cb63eSdanielk1977 } 79ef2cb63eSdanielk1977 } 80ef2cb63eSdanielk1977 81ef2cb63eSdanielk1977 /* If the trigger name was unqualified, and the table is a temp table, 82ef2cb63eSdanielk1977 ** then set iDb to 1 to create the trigger in the temporary database. 83ef2cb63eSdanielk1977 ** If sqlite3SrcListLookup() returns 0, indicating the table does not 84ef2cb63eSdanielk1977 ** exist, the error is caught by the block below. 85c3f9bad2Sdanielk1977 */ 8617435752Sdrh if( !pTableName || db->mallocFailed ){ 876f7adc8aSdrh goto trigger_cleanup; 886f7adc8aSdrh } 89ef2cb63eSdanielk1977 pTab = sqlite3SrcListLookup(pParse, pTableName); 90da184236Sdanielk1977 if( pName2->n==0 && pTab && pTab->pSchema==db->aDb[1].pSchema ){ 91ef2cb63eSdanielk1977 iDb = 1; 92ef2cb63eSdanielk1977 } 93ef2cb63eSdanielk1977 94ef2cb63eSdanielk1977 /* Ensure the table name matches database name and that the table exists */ 9517435752Sdrh if( db->mallocFailed ) goto trigger_cleanup; 96d24cc427Sdrh assert( pTableName->nSrc==1 ); 97ef2cb63eSdanielk1977 if( sqlite3FixInit(&sFix, pParse, iDb, "trigger", pName) && 98ef2cb63eSdanielk1977 sqlite3FixSrcList(&sFix, pTableName) ){ 994312db55Sdrh goto trigger_cleanup; 100f26e09c8Sdrh } 101ef2cb63eSdanielk1977 pTab = sqlite3SrcListLookup(pParse, pTableName); 102ef2cb63eSdanielk1977 if( !pTab ){ 103ef2cb63eSdanielk1977 /* The table does not exist. */ 104d24cc427Sdrh goto trigger_cleanup; 105d24cc427Sdrh } 1064cbdda9eSdrh if( IsVirtual(pTab) ){ 1074cbdda9eSdrh sqlite3ErrorMsg(pParse, "cannot create triggers on virtual tables"); 1084cbdda9eSdrh goto trigger_cleanup; 1094cbdda9eSdrh } 110d24cc427Sdrh 111d8123366Sdanielk1977 /* Check that the trigger name is not reserved and that no trigger of the 112d8123366Sdanielk1977 ** specified name exists */ 11317435752Sdrh zName = sqlite3NameFromToken(db, pName); 114d8123366Sdanielk1977 if( !zName || SQLITE_OK!=sqlite3CheckObjectName(pParse, zName) ){ 115d8123366Sdanielk1977 goto trigger_cleanup; 116d8123366Sdanielk1977 } 117e4df0e74Sdrh if( sqlite3HashFind(&(db->aDb[iDb].pSchema->trigHash), zName,strlen(zName)) ){ 118fdd48a76Sdrh if( !noErr ){ 1194adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "trigger %T already exists", pName); 120fdd48a76Sdrh } 121c3f9bad2Sdanielk1977 goto trigger_cleanup; 122c3f9bad2Sdanielk1977 } 123ef2cb63eSdanielk1977 124ef2cb63eSdanielk1977 /* Do not create a trigger on a system table */ 125dca76841Sdrh if( sqlite3StrNICmp(pTab->zName, "sqlite_", 7)==0 ){ 1264adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create trigger on system table"); 127e0bc4048Sdrh pParse->nErr++; 128e0bc4048Sdrh goto trigger_cleanup; 129e0bc4048Sdrh } 130ef2cb63eSdanielk1977 131ef2cb63eSdanielk1977 /* INSTEAD of triggers are only for views and views only support INSTEAD 132ef2cb63eSdanielk1977 ** of triggers. 133ef2cb63eSdanielk1977 */ 134ef2cb63eSdanielk1977 if( pTab->pSelect && tr_tm!=TK_INSTEAD ){ 1354adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create %s trigger on view: %S", 136da93d238Sdrh (tr_tm == TK_BEFORE)?"BEFORE":"AFTER", pTableName, 0); 137d702fccbSdanielk1977 goto trigger_cleanup; 138d702fccbSdanielk1977 } 139ef2cb63eSdanielk1977 if( !pTab->pSelect && tr_tm==TK_INSTEAD ){ 1404adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "cannot create INSTEAD OF" 141da93d238Sdrh " trigger on table: %S", pTableName, 0); 142d702fccbSdanielk1977 goto trigger_cleanup; 143d702fccbSdanielk1977 } 144da184236Sdanielk1977 iTabDb = sqlite3SchemaToIndex(db, pTab->pSchema); 145ef2cb63eSdanielk1977 146e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION 147e5f9c644Sdrh { 148e5f9c644Sdrh int code = SQLITE_CREATE_TRIGGER; 149da184236Sdanielk1977 const char *zDb = db->aDb[iTabDb].zName; 150e22a334bSdrh const char *zDbTrig = isTemp ? db->aDb[1].zName : zDb; 151da184236Sdanielk1977 if( iTabDb==1 || isTemp ) code = SQLITE_CREATE_TEMP_TRIGGER; 152ef2cb63eSdanielk1977 if( sqlite3AuthCheck(pParse, code, zName, pTab->zName, zDbTrig) ){ 153ed6c8671Sdrh goto trigger_cleanup; 154ed6c8671Sdrh } 155da184236Sdanielk1977 if( sqlite3AuthCheck(pParse, SQLITE_INSERT, SCHEMA_TABLE(iTabDb),0,zDb)){ 15677ad4e41Sdrh goto trigger_cleanup; 15777ad4e41Sdrh } 158d702fccbSdanielk1977 } 159e5f9c644Sdrh #endif 160d702fccbSdanielk1977 161ef2cb63eSdanielk1977 /* INSTEAD OF triggers can only appear on views and BEFORE triggers 1625cf590c1Sdrh ** cannot appear on views. So we might as well translate every 1635cf590c1Sdrh ** INSTEAD OF trigger into a BEFORE trigger. It simplifies code 1645cf590c1Sdrh ** elsewhere. 1655cf590c1Sdrh */ 166d702fccbSdanielk1977 if (tr_tm == TK_INSTEAD){ 167d702fccbSdanielk1977 tr_tm = TK_BEFORE; 168c3f9bad2Sdanielk1977 } 169c3f9bad2Sdanielk1977 170c3f9bad2Sdanielk1977 /* Build the Trigger object */ 17117435752Sdrh pTrigger = (Trigger*)sqlite3DbMallocZero(db, sizeof(Trigger)); 172ef2cb63eSdanielk1977 if( pTrigger==0 ) goto trigger_cleanup; 173ef2cb63eSdanielk1977 pTrigger->name = zName; 174e5f9c644Sdrh zName = 0; 17517435752Sdrh pTrigger->table = sqlite3DbStrDup(db, pTableName->a[0].zName); 176da184236Sdanielk1977 pTrigger->pSchema = db->aDb[iDb].pSchema; 177aaf22685Sdanielk1977 pTrigger->pTabSchema = pTab->pSchema; 178ef2cb63eSdanielk1977 pTrigger->op = op; 179dca76841Sdrh pTrigger->tr_tm = tr_tm==TK_BEFORE ? TRIGGER_BEFORE : TRIGGER_AFTER; 18017435752Sdrh pTrigger->pWhen = sqlite3ExprDup(db, pWhen); 18117435752Sdrh pTrigger->pColumns = sqlite3IdListDup(db, pColumns); 18217435752Sdrh sqlite3TokenCopy(db, &pTrigger->nameToken,pName); 183f0f258b1Sdrh assert( pParse->pNewTrigger==0 ); 184ef2cb63eSdanielk1977 pParse->pNewTrigger = pTrigger; 185f0f258b1Sdrh 186f0f258b1Sdrh trigger_cleanup: 18717435752Sdrh sqlite3_free(zName); 1884adee20fSdanielk1977 sqlite3SrcListDelete(pTableName); 1894adee20fSdanielk1977 sqlite3IdListDelete(pColumns); 1904adee20fSdanielk1977 sqlite3ExprDelete(pWhen); 191d5d56523Sdanielk1977 if( !pParse->pNewTrigger ){ 192d5d56523Sdanielk1977 sqlite3DeleteTrigger(pTrigger); 193d5d56523Sdanielk1977 }else{ 194d5d56523Sdanielk1977 assert( pParse->pNewTrigger==pTrigger ); 195d5d56523Sdanielk1977 } 196f0f258b1Sdrh } 197f0f258b1Sdrh 198f0f258b1Sdrh /* 199f0f258b1Sdrh ** This routine is called after all of the trigger actions have been parsed 200f0f258b1Sdrh ** in order to complete the process of building the trigger. 201f0f258b1Sdrh */ 2024adee20fSdanielk1977 void sqlite3FinishTrigger( 203f0f258b1Sdrh Parse *pParse, /* Parser context */ 204f0f258b1Sdrh TriggerStep *pStepList, /* The triggered program */ 205f0f258b1Sdrh Token *pAll /* Token that describes the complete CREATE TRIGGER */ 206f0f258b1Sdrh ){ 207bfb9e35bSdanielk1977 Trigger *pTrig = 0; /* The trigger whose construction is finishing up */ 2089bb575fdSdrh sqlite3 *db = pParse->db; /* The database */ 209f26e09c8Sdrh DbFixer sFix; 210da184236Sdanielk1977 int iDb; /* Database containing the trigger */ 211f0f258b1Sdrh 212bfb9e35bSdanielk1977 pTrig = pParse->pNewTrigger; 213f0f258b1Sdrh pParse->pNewTrigger = 0; 2142e588c75Sdanielk1977 if( pParse->nErr || !pTrig ) goto triggerfinish_cleanup; 215da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pTrig->pSchema); 216bfb9e35bSdanielk1977 pTrig->step_list = pStepList; 217a69d9168Sdrh while( pStepList ){ 218bfb9e35bSdanielk1977 pStepList->pTrig = pTrig; 219a69d9168Sdrh pStepList = pStepList->pNext; 220a69d9168Sdrh } 221da184236Sdanielk1977 if( sqlite3FixInit(&sFix, pParse, iDb, "trigger", &pTrig->nameToken) 222bfb9e35bSdanielk1977 && sqlite3FixTriggerStep(&sFix, pTrig->step_list) ){ 223f26e09c8Sdrh goto triggerfinish_cleanup; 224f26e09c8Sdrh } 225c3f9bad2Sdanielk1977 226c3f9bad2Sdanielk1977 /* if we are not initializing, and this trigger is not on a TEMP table, 2279adf9ac4Sdrh ** build the sqlite_master entry 2289adf9ac4Sdrh */ 2291d85d931Sdrh if( !db->init.busy ){ 2305719628aSdrh static const VdbeOpList insertTrig[] = { 231f0863fe5Sdrh { OP_NewRowid, 0, 0, 0 }, 2320f69c1e3Sdanielk1977 { OP_String8, 0, 0, "trigger" }, 2330f69c1e3Sdanielk1977 { OP_String8, 0, 0, 0 }, /* 2: trigger name */ 2340f69c1e3Sdanielk1977 { OP_String8, 0, 0, 0 }, /* 3: table name */ 235c977f7f5Sdrh { OP_Integer, 0, 0, 0 }, 2360f69c1e3Sdanielk1977 { OP_String8, 0, 0, "CREATE TRIGGER "}, 2370f69c1e3Sdanielk1977 { OP_String8, 0, 0, 0 }, /* 6: SQL */ 238855eb1cfSdrh { OP_Concat, 0, 0, 0 }, 2398a51256cSdrh { OP_MakeRecord, 5, 0, "aaada" }, 240f0863fe5Sdrh { OP_Insert, 0, 0, 0 }, 241c977f7f5Sdrh }; 242c977f7f5Sdrh int addr; 243c977f7f5Sdrh Vdbe *v; 244c3f9bad2Sdanielk1977 245c3f9bad2Sdanielk1977 /* Make an entry in the sqlite_master table */ 2464adee20fSdanielk1977 v = sqlite3GetVdbe(pParse); 247f0f258b1Sdrh if( v==0 ) goto triggerfinish_cleanup; 248da184236Sdanielk1977 sqlite3BeginWriteOperation(pParse, 0, iDb); 249c00da105Sdanielk1977 sqlite3OpenMasterTable(pParse, iDb); 2504adee20fSdanielk1977 addr = sqlite3VdbeAddOpList(v, ArraySize(insertTrig), insertTrig); 251bfb9e35bSdanielk1977 sqlite3VdbeChangeP3(v, addr+2, pTrig->name, 0); 252bfb9e35bSdanielk1977 sqlite3VdbeChangeP3(v, addr+3, pTrig->table, 0); 2532646da7eSdrh sqlite3VdbeChangeP3(v, addr+6, (char*)pAll->z, pAll->n); 254da184236Sdanielk1977 sqlite3ChangeCookie(db, v, iDb); 2554adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, 0, 0); 2561e536953Sdanielk1977 sqlite3VdbeOp3(v, OP_ParseSchema, iDb, 0, sqlite3MPrintf( 2571e536953Sdanielk1977 db, "type='trigger' AND name='%q'", pTrig->name), P3_DYNAMIC 2581e536953Sdanielk1977 ); 259c3f9bad2Sdanielk1977 } 260c3f9bad2Sdanielk1977 261956bc92cSdrh if( db->init.busy ){ 262aaf22685Sdanielk1977 int n; 263f0f258b1Sdrh Table *pTab; 264d5d56523Sdanielk1977 Trigger *pDel; 265da184236Sdanielk1977 pDel = sqlite3HashInsert(&db->aDb[iDb].pSchema->trigHash, 266e4df0e74Sdrh pTrig->name, strlen(pTrig->name), pTrig); 267d5d56523Sdanielk1977 if( pDel ){ 268f3a65f7eSdrh assert( pDel==pTrig ); 269f3a65f7eSdrh db->mallocFailed = 1; 270d5d56523Sdanielk1977 goto triggerfinish_cleanup; 271d5d56523Sdanielk1977 } 272aaf22685Sdanielk1977 n = strlen(pTrig->table) + 1; 273aaf22685Sdanielk1977 pTab = sqlite3HashFind(&pTrig->pTabSchema->tblHash, pTrig->table, n); 274f0f258b1Sdrh assert( pTab!=0 ); 275bfb9e35bSdanielk1977 pTrig->pNext = pTab->pTrigger; 276bfb9e35bSdanielk1977 pTab->pTrigger = pTrig; 277bfb9e35bSdanielk1977 pTrig = 0; 278c3f9bad2Sdanielk1977 } 279c3f9bad2Sdanielk1977 280f0f258b1Sdrh triggerfinish_cleanup: 281bfb9e35bSdanielk1977 sqlite3DeleteTrigger(pTrig); 282bfb9e35bSdanielk1977 assert( !pParse->pNewTrigger ); 2834adee20fSdanielk1977 sqlite3DeleteTriggerStep(pStepList); 284c3f9bad2Sdanielk1977 } 2854b59ab5eSdrh 2864b59ab5eSdrh /* 2874b59ab5eSdrh ** Make a copy of all components of the given trigger step. This has 2884b59ab5eSdrh ** the effect of copying all Expr.token.z values into memory obtained 28917435752Sdrh ** from sqlite3_malloc(). As initially created, the Expr.token.z values 2904b59ab5eSdrh ** all point to the input string that was fed to the parser. But that 2916f8a503dSdanielk1977 ** string is ephemeral - it will go away as soon as the sqlite3_exec() 2924b59ab5eSdrh ** call that started the parser exits. This routine makes a persistent 2934b59ab5eSdrh ** copy of all the Expr.token.z strings so that the TriggerStep structure 2946f8a503dSdanielk1977 ** will be valid even after the sqlite3_exec() call returns. 2954b59ab5eSdrh */ 29617435752Sdrh static void sqlitePersistTriggerStep(sqlite3 *db, TriggerStep *p){ 2974b59ab5eSdrh if( p->target.z ){ 29817435752Sdrh p->target.z = (u8*)sqlite3DbStrNDup(db, (char*)p->target.z, p->target.n); 2994b59ab5eSdrh p->target.dyn = 1; 3004b59ab5eSdrh } 3014b59ab5eSdrh if( p->pSelect ){ 30217435752Sdrh Select *pNew = sqlite3SelectDup(db, p->pSelect); 3034adee20fSdanielk1977 sqlite3SelectDelete(p->pSelect); 3044b59ab5eSdrh p->pSelect = pNew; 3054b59ab5eSdrh } 3064b59ab5eSdrh if( p->pWhere ){ 30717435752Sdrh Expr *pNew = sqlite3ExprDup(db, p->pWhere); 3084adee20fSdanielk1977 sqlite3ExprDelete(p->pWhere); 3094b59ab5eSdrh p->pWhere = pNew; 3104b59ab5eSdrh } 3114b59ab5eSdrh if( p->pExprList ){ 31217435752Sdrh ExprList *pNew = sqlite3ExprListDup(db, p->pExprList); 3134adee20fSdanielk1977 sqlite3ExprListDelete(p->pExprList); 3144b59ab5eSdrh p->pExprList = pNew; 3154b59ab5eSdrh } 3164b59ab5eSdrh if( p->pIdList ){ 31717435752Sdrh IdList *pNew = sqlite3IdListDup(db, p->pIdList); 3184adee20fSdanielk1977 sqlite3IdListDelete(p->pIdList); 3194b59ab5eSdrh p->pIdList = pNew; 320c3f9bad2Sdanielk1977 } 321c3f9bad2Sdanielk1977 } 322c3f9bad2Sdanielk1977 323c977f7f5Sdrh /* 324c977f7f5Sdrh ** Turn a SELECT statement (that the pSelect parameter points to) into 325c977f7f5Sdrh ** a trigger step. Return a pointer to a TriggerStep structure. 326c977f7f5Sdrh ** 327c977f7f5Sdrh ** The parser calls this routine when it finds a SELECT statement in 328c977f7f5Sdrh ** body of a TRIGGER. 329c977f7f5Sdrh */ 33017435752Sdrh TriggerStep *sqlite3TriggerSelectStep(sqlite3 *db, Select *pSelect){ 33117435752Sdrh TriggerStep *pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep)); 3322e588c75Sdanielk1977 if( pTriggerStep==0 ) { 3332e588c75Sdanielk1977 sqlite3SelectDelete(pSelect); 3342e588c75Sdanielk1977 return 0; 3352e588c75Sdanielk1977 } 336c3f9bad2Sdanielk1977 337633ed08dSdanielk1977 pTriggerStep->op = TK_SELECT; 338633ed08dSdanielk1977 pTriggerStep->pSelect = pSelect; 339633ed08dSdanielk1977 pTriggerStep->orconf = OE_Default; 34017435752Sdrh sqlitePersistTriggerStep(db, pTriggerStep); 341c3f9bad2Sdanielk1977 342633ed08dSdanielk1977 return pTriggerStep; 343c3f9bad2Sdanielk1977 } 344c3f9bad2Sdanielk1977 345c977f7f5Sdrh /* 346c977f7f5Sdrh ** Build a trigger step out of an INSERT statement. Return a pointer 347c977f7f5Sdrh ** to the new trigger step. 348c977f7f5Sdrh ** 349c977f7f5Sdrh ** The parser calls this routine when it sees an INSERT inside the 350c977f7f5Sdrh ** body of a trigger. 351c977f7f5Sdrh */ 3524adee20fSdanielk1977 TriggerStep *sqlite3TriggerInsertStep( 35317435752Sdrh sqlite3 *db, /* The database connection */ 354c977f7f5Sdrh Token *pTableName, /* Name of the table into which we insert */ 355c977f7f5Sdrh IdList *pColumn, /* List of columns in pTableName to insert into */ 356c977f7f5Sdrh ExprList *pEList, /* The VALUE clause: a list of values to be inserted */ 357c977f7f5Sdrh Select *pSelect, /* A SELECT statement that supplies values */ 358c977f7f5Sdrh int orconf /* The conflict algorithm (OE_Abort, OE_Replace, etc.) */ 359633ed08dSdanielk1977 ){ 360f3a65f7eSdrh TriggerStep *pTriggerStep; 361c3f9bad2Sdanielk1977 362633ed08dSdanielk1977 assert(pEList == 0 || pSelect == 0); 363f3a65f7eSdrh assert(pEList != 0 || pSelect != 0 || db->mallocFailed); 364c3f9bad2Sdanielk1977 365f3a65f7eSdrh pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep)); 366d5d56523Sdanielk1977 if( pTriggerStep ){ 367633ed08dSdanielk1977 pTriggerStep->op = TK_INSERT; 368633ed08dSdanielk1977 pTriggerStep->pSelect = pSelect; 369633ed08dSdanielk1977 pTriggerStep->target = *pTableName; 370633ed08dSdanielk1977 pTriggerStep->pIdList = pColumn; 371633ed08dSdanielk1977 pTriggerStep->pExprList = pEList; 372633ed08dSdanielk1977 pTriggerStep->orconf = orconf; 37317435752Sdrh sqlitePersistTriggerStep(db, pTriggerStep); 374d5d56523Sdanielk1977 }else{ 375d5d56523Sdanielk1977 sqlite3IdListDelete(pColumn); 376d5d56523Sdanielk1977 sqlite3ExprListDelete(pEList); 37717435752Sdrh sqlite3SelectDelete(pSelect); 378d5d56523Sdanielk1977 } 379c3f9bad2Sdanielk1977 380633ed08dSdanielk1977 return pTriggerStep; 381c3f9bad2Sdanielk1977 } 382c3f9bad2Sdanielk1977 383c977f7f5Sdrh /* 384c977f7f5Sdrh ** Construct a trigger step that implements an UPDATE statement and return 385c977f7f5Sdrh ** a pointer to that trigger step. The parser calls this routine when it 386c977f7f5Sdrh ** sees an UPDATE statement inside the body of a CREATE TRIGGER. 387c977f7f5Sdrh */ 3884adee20fSdanielk1977 TriggerStep *sqlite3TriggerUpdateStep( 38917435752Sdrh sqlite3 *db, /* The database connection */ 390c977f7f5Sdrh Token *pTableName, /* Name of the table to be updated */ 391c977f7f5Sdrh ExprList *pEList, /* The SET clause: list of column and new values */ 392c977f7f5Sdrh Expr *pWhere, /* The WHERE clause */ 393c977f7f5Sdrh int orconf /* The conflict algorithm. (OE_Abort, OE_Ignore, etc) */ 394c977f7f5Sdrh ){ 39517435752Sdrh TriggerStep *pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep)); 3960e3a6f3fSdrh if( pTriggerStep==0 ){ 3970e3a6f3fSdrh sqlite3ExprListDelete(pEList); 3980e3a6f3fSdrh sqlite3ExprDelete(pWhere); 3990e3a6f3fSdrh return 0; 4000e3a6f3fSdrh } 401c3f9bad2Sdanielk1977 402633ed08dSdanielk1977 pTriggerStep->op = TK_UPDATE; 403633ed08dSdanielk1977 pTriggerStep->target = *pTableName; 404633ed08dSdanielk1977 pTriggerStep->pExprList = pEList; 405633ed08dSdanielk1977 pTriggerStep->pWhere = pWhere; 406633ed08dSdanielk1977 pTriggerStep->orconf = orconf; 40717435752Sdrh sqlitePersistTriggerStep(db, pTriggerStep); 408c3f9bad2Sdanielk1977 409633ed08dSdanielk1977 return pTriggerStep; 410c3f9bad2Sdanielk1977 } 411c3f9bad2Sdanielk1977 412c977f7f5Sdrh /* 413c977f7f5Sdrh ** Construct a trigger step that implements a DELETE statement and return 414c977f7f5Sdrh ** a pointer to that trigger step. The parser calls this routine when it 415c977f7f5Sdrh ** sees a DELETE statement inside the body of a CREATE TRIGGER. 416c977f7f5Sdrh */ 41717435752Sdrh TriggerStep *sqlite3TriggerDeleteStep( 41817435752Sdrh sqlite3 *db, /* Database connection */ 41917435752Sdrh Token *pTableName, /* The table from which rows are deleted */ 42017435752Sdrh Expr *pWhere /* The WHERE clause */ 42117435752Sdrh ){ 42217435752Sdrh TriggerStep *pTriggerStep = sqlite3DbMallocZero(db, sizeof(TriggerStep)); 423b63a53dcSdrh if( pTriggerStep==0 ){ 424b63a53dcSdrh sqlite3ExprDelete(pWhere); 425b63a53dcSdrh return 0; 426b63a53dcSdrh } 427c3f9bad2Sdanielk1977 428633ed08dSdanielk1977 pTriggerStep->op = TK_DELETE; 429633ed08dSdanielk1977 pTriggerStep->target = *pTableName; 430633ed08dSdanielk1977 pTriggerStep->pWhere = pWhere; 431633ed08dSdanielk1977 pTriggerStep->orconf = OE_Default; 43217435752Sdrh sqlitePersistTriggerStep(db, pTriggerStep); 433c3f9bad2Sdanielk1977 434633ed08dSdanielk1977 return pTriggerStep; 435c3f9bad2Sdanielk1977 } 436c3f9bad2Sdanielk1977 437c3f9bad2Sdanielk1977 /* 438633ed08dSdanielk1977 ** Recursively delete a Trigger structure 439c3f9bad2Sdanielk1977 */ 4404adee20fSdanielk1977 void sqlite3DeleteTrigger(Trigger *pTrigger){ 441f0f258b1Sdrh if( pTrigger==0 ) return; 4424adee20fSdanielk1977 sqlite3DeleteTriggerStep(pTrigger->step_list); 44317435752Sdrh sqlite3_free(pTrigger->name); 44417435752Sdrh sqlite3_free(pTrigger->table); 4454adee20fSdanielk1977 sqlite3ExprDelete(pTrigger->pWhen); 4464adee20fSdanielk1977 sqlite3IdListDelete(pTrigger->pColumns); 44717435752Sdrh if( pTrigger->nameToken.dyn ) sqlite3_free((char*)pTrigger->nameToken.z); 44817435752Sdrh sqlite3_free(pTrigger); 449633ed08dSdanielk1977 } 450633ed08dSdanielk1977 451633ed08dSdanielk1977 /* 4528a41449eSdanielk1977 ** This function is called to drop a trigger from the database schema. 4538a41449eSdanielk1977 ** 4548a41449eSdanielk1977 ** This may be called directly from the parser and therefore identifies 4558a41449eSdanielk1977 ** the trigger by name. The sqlite3DropTriggerPtr() routine does the 4568a41449eSdanielk1977 ** same job as this routine except it takes a pointer to the trigger 4578a41449eSdanielk1977 ** instead of the trigger name. 4588a41449eSdanielk1977 **/ 459fdd48a76Sdrh void sqlite3DropTrigger(Parse *pParse, SrcList *pName, int noErr){ 460742f947bSdanielk1977 Trigger *pTrigger = 0; 461d24cc427Sdrh int i; 462d24cc427Sdrh const char *zDb; 463d24cc427Sdrh const char *zName; 464d24cc427Sdrh int nName; 4659bb575fdSdrh sqlite3 *db = pParse->db; 466633ed08dSdanielk1977 46717435752Sdrh if( db->mallocFailed ) goto drop_trigger_cleanup; 4688a41449eSdanielk1977 if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){ 469c0391398Sdanielk1977 goto drop_trigger_cleanup; 470c0391398Sdanielk1977 } 4718e227875Sdanielk1977 472d24cc427Sdrh assert( pName->nSrc==1 ); 473d24cc427Sdrh zDb = pName->a[0].zDatabase; 474d24cc427Sdrh zName = pName->a[0].zName; 475d24cc427Sdrh nName = strlen(zName); 47653c0f748Sdanielk1977 for(i=OMIT_TEMPDB; i<db->nDb; i++){ 477812d7a21Sdrh int j = (i<2) ? i^1 : i; /* Search TEMP before MAIN */ 4784adee20fSdanielk1977 if( zDb && sqlite3StrICmp(db->aDb[j].zName, zDb) ) continue; 479e4df0e74Sdrh pTrigger = sqlite3HashFind(&(db->aDb[j].pSchema->trigHash), zName, nName); 480d24cc427Sdrh if( pTrigger ) break; 481c3f9bad2Sdanielk1977 } 482d24cc427Sdrh if( !pTrigger ){ 483fdd48a76Sdrh if( !noErr ){ 4844adee20fSdanielk1977 sqlite3ErrorMsg(pParse, "no such trigger: %S", pName, 0); 485fdd48a76Sdrh } 486d24cc427Sdrh goto drop_trigger_cleanup; 487d24cc427Sdrh } 48874161705Sdrh sqlite3DropTriggerPtr(pParse, pTrigger); 48979a519c0Sdrh 49079a519c0Sdrh drop_trigger_cleanup: 4914adee20fSdanielk1977 sqlite3SrcListDelete(pName); 49279a519c0Sdrh } 49379a519c0Sdrh 49479a519c0Sdrh /* 495956bc92cSdrh ** Return a pointer to the Table structure for the table that a trigger 496956bc92cSdrh ** is set on. 497956bc92cSdrh */ 49874161705Sdrh static Table *tableOfTrigger(Trigger *pTrigger){ 499aaf22685Sdanielk1977 int n = strlen(pTrigger->table) + 1; 500aaf22685Sdanielk1977 return sqlite3HashFind(&pTrigger->pTabSchema->tblHash, pTrigger->table, n); 501956bc92cSdrh } 502956bc92cSdrh 503956bc92cSdrh 504956bc92cSdrh /* 50574161705Sdrh ** Drop a trigger given a pointer to that trigger. 50679a519c0Sdrh */ 50774161705Sdrh void sqlite3DropTriggerPtr(Parse *pParse, Trigger *pTrigger){ 50879a519c0Sdrh Table *pTable; 50979a519c0Sdrh Vdbe *v; 5109bb575fdSdrh sqlite3 *db = pParse->db; 511956bc92cSdrh int iDb; 51279a519c0Sdrh 513da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pTrigger->pSchema); 514956bc92cSdrh assert( iDb>=0 && iDb<db->nDb ); 51574161705Sdrh pTable = tableOfTrigger(pTrigger); 516ed6c8671Sdrh assert( pTable ); 517da184236Sdanielk1977 assert( pTable->pSchema==pTrigger->pSchema || iDb==1 ); 518e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION 519e5f9c644Sdrh { 520e5f9c644Sdrh int code = SQLITE_DROP_TRIGGER; 521956bc92cSdrh const char *zDb = db->aDb[iDb].zName; 522956bc92cSdrh const char *zTab = SCHEMA_TABLE(iDb); 523956bc92cSdrh if( iDb==1 ) code = SQLITE_DROP_TEMP_TRIGGER; 5244adee20fSdanielk1977 if( sqlite3AuthCheck(pParse, code, pTrigger->name, pTable->zName, zDb) || 5254adee20fSdanielk1977 sqlite3AuthCheck(pParse, SQLITE_DELETE, zTab, 0, zDb) ){ 52679a519c0Sdrh return; 527ed6c8671Sdrh } 528e5f9c644Sdrh } 529e5f9c644Sdrh #endif 530c3f9bad2Sdanielk1977 531f0f258b1Sdrh /* Generate code to destroy the database record of the trigger. 532f0f258b1Sdrh */ 53343617e9aSdrh assert( pTable!=0 ); 53443617e9aSdrh if( (v = sqlite3GetVdbe(pParse))!=0 ){ 535f0f258b1Sdrh int base; 5365719628aSdrh static const VdbeOpList dropTrigger[] = { 5379b1b01bbSdrh { OP_Rewind, 0, ADDR(9), 0}, 5380f69c1e3Sdanielk1977 { OP_String8, 0, 0, 0}, /* 1 */ 539f0f258b1Sdrh { OP_Column, 0, 1, 0}, 5409b1b01bbSdrh { OP_Ne, 0, ADDR(8), 0}, 5410f69c1e3Sdanielk1977 { OP_String8, 0, 0, "trigger"}, 5429b1b01bbSdrh { OP_Column, 0, 0, 0}, 5439b1b01bbSdrh { OP_Ne, 0, ADDR(8), 0}, 544f0f258b1Sdrh { OP_Delete, 0, 0, 0}, 5459b1b01bbSdrh { OP_Next, 0, ADDR(1), 0}, /* 8 */ 546f0f258b1Sdrh }; 547f0f258b1Sdrh 548956bc92cSdrh sqlite3BeginWriteOperation(pParse, 0, iDb); 549c00da105Sdanielk1977 sqlite3OpenMasterTable(pParse, iDb); 5504adee20fSdanielk1977 base = sqlite3VdbeAddOpList(v, ArraySize(dropTrigger), dropTrigger); 5514adee20fSdanielk1977 sqlite3VdbeChangeP3(v, base+1, pTrigger->name, 0); 552956bc92cSdrh sqlite3ChangeCookie(db, v, iDb); 5534adee20fSdanielk1977 sqlite3VdbeAddOp(v, OP_Close, 0, 0); 554956bc92cSdrh sqlite3VdbeOp3(v, OP_DropTrigger, iDb, 0, pTrigger->name, 0); 555956bc92cSdrh } 556f0f258b1Sdrh } 557f0f258b1Sdrh 558c3f9bad2Sdanielk1977 /* 559956bc92cSdrh ** Remove a trigger from the hash tables of the sqlite* pointer. 560c3f9bad2Sdanielk1977 */ 561956bc92cSdrh void sqlite3UnlinkAndDeleteTrigger(sqlite3 *db, int iDb, const char *zName){ 562956bc92cSdrh Trigger *pTrigger; 56379a519c0Sdrh int nName = strlen(zName); 564e4df0e74Sdrh pTrigger = sqlite3HashInsert(&(db->aDb[iDb].pSchema->trigHash), 565e4df0e74Sdrh zName, nName, 0); 566956bc92cSdrh if( pTrigger ){ 56774161705Sdrh Table *pTable = tableOfTrigger(pTrigger); 568956bc92cSdrh assert( pTable!=0 ); 569633ed08dSdanielk1977 if( pTable->pTrigger == pTrigger ){ 570633ed08dSdanielk1977 pTable->pTrigger = pTrigger->pNext; 571633ed08dSdanielk1977 }else{ 572633ed08dSdanielk1977 Trigger *cc = pTable->pTrigger; 573c3f9bad2Sdanielk1977 while( cc ){ 574633ed08dSdanielk1977 if( cc->pNext == pTrigger ){ 575c3f9bad2Sdanielk1977 cc->pNext = cc->pNext->pNext; 576c3f9bad2Sdanielk1977 break; 577c3f9bad2Sdanielk1977 } 578c3f9bad2Sdanielk1977 cc = cc->pNext; 579c3f9bad2Sdanielk1977 } 580c3f9bad2Sdanielk1977 assert(cc); 581c3f9bad2Sdanielk1977 } 5824adee20fSdanielk1977 sqlite3DeleteTrigger(pTrigger); 583956bc92cSdrh db->flags |= SQLITE_InternChanges; 584c3f9bad2Sdanielk1977 } 585c3f9bad2Sdanielk1977 } 586c3f9bad2Sdanielk1977 587c977f7f5Sdrh /* 588c977f7f5Sdrh ** pEList is the SET clause of an UPDATE statement. Each entry 589c977f7f5Sdrh ** in pEList is of the format <id>=<expr>. If any of the entries 590c977f7f5Sdrh ** in pEList have an <id> which matches an identifier in pIdList, 591c977f7f5Sdrh ** then return TRUE. If pIdList==NULL, then it is considered a 592c977f7f5Sdrh ** wildcard that matches anything. Likewise if pEList==NULL then 593c977f7f5Sdrh ** it matches anything so always return true. Return false only 594c977f7f5Sdrh ** if there is no match. 595c977f7f5Sdrh */ 596c977f7f5Sdrh static int checkColumnOverLap(IdList *pIdList, ExprList *pEList){ 597ad2d8307Sdrh int e; 598ad2d8307Sdrh if( !pIdList || !pEList ) return 1; 599f29ce559Sdanielk1977 for(e=0; e<pEList->nExpr; e++){ 6004adee20fSdanielk1977 if( sqlite3IdListIndex(pIdList, pEList->a[e].zName)>=0 ) return 1; 601f29ce559Sdanielk1977 } 602c3f9bad2Sdanielk1977 return 0; 603c3f9bad2Sdanielk1977 } 604c3f9bad2Sdanielk1977 605c3f9bad2Sdanielk1977 /* 606dca76841Sdrh ** Return a bit vector to indicate what kind of triggers exist for operation 607dca76841Sdrh ** "op" on table pTab. If pChanges is not NULL then it is a list of columns 608dca76841Sdrh ** that are being updated. Triggers only match if the ON clause of the 609dca76841Sdrh ** trigger definition overlaps the set of columns being updated. 610dca76841Sdrh ** 611dca76841Sdrh ** The returned bit vector is some combination of TRIGGER_BEFORE and 612dca76841Sdrh ** TRIGGER_AFTER. 613c3f9bad2Sdanielk1977 */ 6144adee20fSdanielk1977 int sqlite3TriggersExist( 615c977f7f5Sdrh Parse *pParse, /* Used to check for recursive triggers */ 616dca76841Sdrh Table *pTab, /* The table the contains the triggers */ 617c3f9bad2Sdanielk1977 int op, /* one of TK_DELETE, TK_INSERT, TK_UPDATE */ 618c977f7f5Sdrh ExprList *pChanges /* Columns that change in an UPDATE statement */ 619c977f7f5Sdrh ){ 6204cbdda9eSdrh Trigger *pTrigger; 621dca76841Sdrh int mask = 0; 622c3f9bad2Sdanielk1977 6234cbdda9eSdrh pTrigger = IsVirtual(pTab) ? 0 : pTab->pTrigger; 624dca76841Sdrh while( pTrigger ){ 625dca76841Sdrh if( pTrigger->op==op && checkColumnOverLap(pTrigger->pColumns, pChanges) ){ 626dca76841Sdrh mask |= pTrigger->tr_tm; 627c3f9bad2Sdanielk1977 } 628dca76841Sdrh pTrigger = pTrigger->pNext; 629dca76841Sdrh } 630dca76841Sdrh return mask; 631c3f9bad2Sdanielk1977 } 632c3f9bad2Sdanielk1977 633c977f7f5Sdrh /* 634f26e09c8Sdrh ** Convert the pStep->target token into a SrcList and return a pointer 635f26e09c8Sdrh ** to that SrcList. 636f26e09c8Sdrh ** 637f26e09c8Sdrh ** This routine adds a specific database name, if needed, to the target when 638f26e09c8Sdrh ** forming the SrcList. This prevents a trigger in one database from 639f26e09c8Sdrh ** referring to a target in another database. An exception is when the 640f26e09c8Sdrh ** trigger is in TEMP in which case it can refer to any other database it 641f26e09c8Sdrh ** wants. 642f26e09c8Sdrh */ 643f26e09c8Sdrh static SrcList *targetSrcList( 644f26e09c8Sdrh Parse *pParse, /* The parsing context */ 645f26e09c8Sdrh TriggerStep *pStep /* The trigger containing the target token */ 646f26e09c8Sdrh ){ 647f26e09c8Sdrh Token sDb; /* Dummy database name token */ 648f26e09c8Sdrh int iDb; /* Index of the database to use */ 649f26e09c8Sdrh SrcList *pSrc; /* SrcList to be returned */ 650f26e09c8Sdrh 651da184236Sdanielk1977 iDb = sqlite3SchemaToIndex(pParse->db, pStep->pTrig->pSchema); 652f26e09c8Sdrh if( iDb==0 || iDb>=2 ){ 653f26e09c8Sdrh assert( iDb<pParse->db->nDb ); 6542646da7eSdrh sDb.z = (u8*)pParse->db->aDb[iDb].zName; 6552646da7eSdrh sDb.n = strlen((char*)sDb.z); 65617435752Sdrh pSrc = sqlite3SrcListAppend(pParse->db, 0, &sDb, &pStep->target); 657f26e09c8Sdrh } else { 65817435752Sdrh pSrc = sqlite3SrcListAppend(pParse->db, 0, &pStep->target, 0); 659f26e09c8Sdrh } 660f26e09c8Sdrh return pSrc; 661f26e09c8Sdrh } 662f26e09c8Sdrh 663f26e09c8Sdrh /* 664c977f7f5Sdrh ** Generate VDBE code for zero or more statements inside the body of a 665c977f7f5Sdrh ** trigger. 666c977f7f5Sdrh */ 667c3f9bad2Sdanielk1977 static int codeTriggerProgram( 668c977f7f5Sdrh Parse *pParse, /* The parser context */ 669c977f7f5Sdrh TriggerStep *pStepList, /* List of statements inside the trigger body */ 670c977f7f5Sdrh int orconfin /* Conflict algorithm. (OE_Abort, etc) */ 671633ed08dSdanielk1977 ){ 672633ed08dSdanielk1977 TriggerStep * pTriggerStep = pStepList; 673c3f9bad2Sdanielk1977 int orconf; 674344737f6Sdrh Vdbe *v = pParse->pVdbe; 67517435752Sdrh sqlite3 *db = pParse->db; 676c3f9bad2Sdanielk1977 677344737f6Sdrh assert( pTriggerStep!=0 ); 678344737f6Sdrh assert( v!=0 ); 679344737f6Sdrh sqlite3VdbeAddOp(v, OP_ContextPush, 0, 0); 680d4e70ebdSdrh VdbeComment((v, "begin trigger %s", pStepList->pTrig->name)); 681633ed08dSdanielk1977 while( pTriggerStep ){ 682633ed08dSdanielk1977 orconf = (orconfin == OE_Default)?pTriggerStep->orconf:orconfin; 683c3f9bad2Sdanielk1977 pParse->trigStack->orconf = orconf; 684633ed08dSdanielk1977 switch( pTriggerStep->op ){ 685c3f9bad2Sdanielk1977 case TK_SELECT: { 68617435752Sdrh Select *ss = sqlite3SelectDup(db, pTriggerStep->pSelect); 68728f45914Sdrh if( ss ){ 688*6c8c8ce0Sdanielk1977 SelectDest dest = {SRT_Discard, 0, 0}; 689b3bce662Sdanielk1977 sqlite3SelectResolve(pParse, ss, 0); 690*6c8c8ce0Sdanielk1977 sqlite3Select(pParse, ss, &dest, 0, 0, 0, 0); 6914adee20fSdanielk1977 sqlite3SelectDelete(ss); 69228f45914Sdrh } 693c3f9bad2Sdanielk1977 break; 694c3f9bad2Sdanielk1977 } 695c3f9bad2Sdanielk1977 case TK_UPDATE: { 696113088ecSdrh SrcList *pSrc; 697f26e09c8Sdrh pSrc = targetSrcList(pParse, pTriggerStep); 698344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 0, 0); 6994adee20fSdanielk1977 sqlite3Update(pParse, pSrc, 70017435752Sdrh sqlite3ExprListDup(db, pTriggerStep->pExprList), 70117435752Sdrh sqlite3ExprDup(db, pTriggerStep->pWhere), orconf); 702344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 1, 0); 703c3f9bad2Sdanielk1977 break; 704c3f9bad2Sdanielk1977 } 705c3f9bad2Sdanielk1977 case TK_INSERT: { 706113088ecSdrh SrcList *pSrc; 707f26e09c8Sdrh pSrc = targetSrcList(pParse, pTriggerStep); 708344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 0, 0); 7094adee20fSdanielk1977 sqlite3Insert(pParse, pSrc, 71017435752Sdrh sqlite3ExprListDup(db, pTriggerStep->pExprList), 71117435752Sdrh sqlite3SelectDup(db, pTriggerStep->pSelect), 71217435752Sdrh sqlite3IdListDup(db, pTriggerStep->pIdList), orconf); 713344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 1, 0); 714c3f9bad2Sdanielk1977 break; 715c3f9bad2Sdanielk1977 } 716c3f9bad2Sdanielk1977 case TK_DELETE: { 717113088ecSdrh SrcList *pSrc; 718344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 0, 0); 719f26e09c8Sdrh pSrc = targetSrcList(pParse, pTriggerStep); 72017435752Sdrh sqlite3DeleteFrom(pParse, pSrc, 72117435752Sdrh sqlite3ExprDup(db, pTriggerStep->pWhere)); 722344737f6Sdrh sqlite3VdbeAddOp(v, OP_ResetCount, 1, 0); 723c3f9bad2Sdanielk1977 break; 724c3f9bad2Sdanielk1977 } 725c3f9bad2Sdanielk1977 default: 726c3f9bad2Sdanielk1977 assert(0); 727c3f9bad2Sdanielk1977 } 728633ed08dSdanielk1977 pTriggerStep = pTriggerStep->pNext; 729c3f9bad2Sdanielk1977 } 730344737f6Sdrh sqlite3VdbeAddOp(v, OP_ContextPop, 0, 0); 731d4e70ebdSdrh VdbeComment((v, "end trigger %s", pStepList->pTrig->name)); 732c3f9bad2Sdanielk1977 733c3f9bad2Sdanielk1977 return 0; 734c3f9bad2Sdanielk1977 } 735c3f9bad2Sdanielk1977 736633ed08dSdanielk1977 /* 737633ed08dSdanielk1977 ** This is called to code FOR EACH ROW triggers. 738633ed08dSdanielk1977 ** 739633ed08dSdanielk1977 ** When the code that this function generates is executed, the following 740633ed08dSdanielk1977 ** must be true: 741c977f7f5Sdrh ** 742c977f7f5Sdrh ** 1. No cursors may be open in the main database. (But newIdx and oldIdx 743c977f7f5Sdrh ** can be indices of cursors in temporary tables. See below.) 744c977f7f5Sdrh ** 745633ed08dSdanielk1977 ** 2. If the triggers being coded are ON INSERT or ON UPDATE triggers, then 746633ed08dSdanielk1977 ** a temporary vdbe cursor (index newIdx) must be open and pointing at 747633ed08dSdanielk1977 ** a row containing values to be substituted for new.* expressions in the 748633ed08dSdanielk1977 ** trigger program(s). 749c977f7f5Sdrh ** 750633ed08dSdanielk1977 ** 3. If the triggers being coded are ON DELETE or ON UPDATE triggers, then 751633ed08dSdanielk1977 ** a temporary vdbe cursor (index oldIdx) must be open and pointing at 752633ed08dSdanielk1977 ** a row containing values to be substituted for old.* expressions in the 753633ed08dSdanielk1977 ** trigger program(s). 754633ed08dSdanielk1977 ** 7558f2c54e6Sdanielk1977 ** If they are not NULL, the piOldColMask and piNewColMask output variables 7568f2c54e6Sdanielk1977 ** are set to values that describe the columns used by the trigger program 7578f2c54e6Sdanielk1977 ** in the OLD.* and NEW.* tables respectively. If column N of the 7588f2c54e6Sdanielk1977 ** pseudo-table is read at least once, the corresponding bit of the output 7598f2c54e6Sdanielk1977 ** mask is set. If a column with an index greater than 32 is read, the 7608f2c54e6Sdanielk1977 ** output mask is set to the special value 0xffffffff. 7618f2c54e6Sdanielk1977 ** 762633ed08dSdanielk1977 */ 7634adee20fSdanielk1977 int sqlite3CodeRowTrigger( 764c3f9bad2Sdanielk1977 Parse *pParse, /* Parse context */ 765c3f9bad2Sdanielk1977 int op, /* One of TK_UPDATE, TK_INSERT, TK_DELETE */ 766633ed08dSdanielk1977 ExprList *pChanges, /* Changes list for any UPDATE OF triggers */ 767dca76841Sdrh int tr_tm, /* One of TRIGGER_BEFORE, TRIGGER_AFTER */ 768633ed08dSdanielk1977 Table *pTab, /* The table to code triggers from */ 769633ed08dSdanielk1977 int newIdx, /* The indice of the "new" row to access */ 770633ed08dSdanielk1977 int oldIdx, /* The indice of the "old" row to access */ 7716f34903eSdanielk1977 int orconf, /* ON CONFLICT policy */ 7728f2c54e6Sdanielk1977 int ignoreJump, /* Instruction to jump to for RAISE(IGNORE) */ 7738f2c54e6Sdanielk1977 u32 *piOldColMask, /* OUT: Mask of columns used from the OLD.* table */ 7748f2c54e6Sdanielk1977 u32 *piNewColMask /* OUT: Mask of columns used from the NEW.* table */ 775c977f7f5Sdrh ){ 776eecfb3eeSdanielk1977 Trigger *p; 77767040462Sdrh TriggerStack trigStackEntry; 778c3f9bad2Sdanielk1977 7798f2c54e6Sdanielk1977 trigStackEntry.oldColMask = 0; 7808f2c54e6Sdanielk1977 trigStackEntry.newColMask = 0; 7818f2c54e6Sdanielk1977 782c3f9bad2Sdanielk1977 assert(op == TK_UPDATE || op == TK_INSERT || op == TK_DELETE); 783dca76841Sdrh assert(tr_tm == TRIGGER_BEFORE || tr_tm == TRIGGER_AFTER ); 784c3f9bad2Sdanielk1977 785633ed08dSdanielk1977 assert(newIdx != -1 || oldIdx != -1); 786c3f9bad2Sdanielk1977 787eecfb3eeSdanielk1977 for(p=pTab->pTrigger; p; p=p->pNext){ 788c3f9bad2Sdanielk1977 int fire_this = 0; 789932083c6Sdanielk1977 sqlite3 *db = pParse->db; 790c3f9bad2Sdanielk1977 791eecfb3eeSdanielk1977 /* Determine whether we should code this trigger */ 792eecfb3eeSdanielk1977 if( 793eecfb3eeSdanielk1977 p->op==op && 794eecfb3eeSdanielk1977 p->tr_tm==tr_tm && 795932083c6Sdanielk1977 (p->pSchema==p->pTabSchema || p->pSchema==db->aDb[1].pSchema) && 796eecfb3eeSdanielk1977 (op!=TK_UPDATE||!p->pColumns||checkColumnOverLap(p->pColumns,pChanges)) 797eecfb3eeSdanielk1977 ){ 798eecfb3eeSdanielk1977 TriggerStack *pS; /* Pointer to trigger-stack entry */ 79974161705Sdrh for(pS=pParse->trigStack; pS && p!=pS->pTrigger; pS=pS->pNext){} 800eecfb3eeSdanielk1977 if( !pS ){ 801c3f9bad2Sdanielk1977 fire_this = 1; 802c3f9bad2Sdanielk1977 } 803229caa36Sdrh #if 0 /* Give no warning for recursive triggers. Just do not do them */ 804229caa36Sdrh else{ 805229caa36Sdrh sqlite3ErrorMsg(pParse, "recursive triggers not supported (%s)", 806229caa36Sdrh p->name); 807229caa36Sdrh return SQLITE_ERROR; 808229caa36Sdrh } 809229caa36Sdrh #endif 810f29ce559Sdanielk1977 } 811c3f9bad2Sdanielk1977 81267040462Sdrh if( fire_this ){ 813c3f9bad2Sdanielk1977 int endTrigger; 814c3f9bad2Sdanielk1977 Expr * whenExpr; 81585e2096fSdrh AuthContext sContext; 816b3bce662Sdanielk1977 NameContext sNC; 817c3f9bad2Sdanielk1977 818b3bce662Sdanielk1977 memset(&sNC, 0, sizeof(sNC)); 819b3bce662Sdanielk1977 sNC.pParse = pParse; 820c3f9bad2Sdanielk1977 821c3f9bad2Sdanielk1977 /* Push an entry on to the trigger stack */ 822eecfb3eeSdanielk1977 trigStackEntry.pTrigger = p; 82367040462Sdrh trigStackEntry.newIdx = newIdx; 82467040462Sdrh trigStackEntry.oldIdx = oldIdx; 82567040462Sdrh trigStackEntry.pTab = pTab; 82667040462Sdrh trigStackEntry.pNext = pParse->trigStack; 82767040462Sdrh trigStackEntry.ignoreJump = ignoreJump; 82867040462Sdrh pParse->trigStack = &trigStackEntry; 829eecfb3eeSdanielk1977 sqlite3AuthContextPush(pParse, &sContext, p->name); 830c3f9bad2Sdanielk1977 831c3f9bad2Sdanielk1977 /* code the WHEN clause */ 8324adee20fSdanielk1977 endTrigger = sqlite3VdbeMakeLabel(pParse->pVdbe); 833932083c6Sdanielk1977 whenExpr = sqlite3ExprDup(db, p->pWhen); 834932083c6Sdanielk1977 if( db->mallocFailed || sqlite3ExprResolveNames(&sNC, whenExpr) ){ 83567040462Sdrh pParse->trigStack = trigStackEntry.pNext; 8364adee20fSdanielk1977 sqlite3ExprDelete(whenExpr); 837c3f9bad2Sdanielk1977 return 1; 838c3f9bad2Sdanielk1977 } 8394adee20fSdanielk1977 sqlite3ExprIfFalse(pParse, whenExpr, endTrigger, 1); 8404adee20fSdanielk1977 sqlite3ExprDelete(whenExpr); 841c3f9bad2Sdanielk1977 842eecfb3eeSdanielk1977 codeTriggerProgram(pParse, p->step_list, orconf); 843c3f9bad2Sdanielk1977 844c3f9bad2Sdanielk1977 /* Pop the entry off the trigger stack */ 84567040462Sdrh pParse->trigStack = trigStackEntry.pNext; 8464adee20fSdanielk1977 sqlite3AuthContextPop(&sContext); 847c3f9bad2Sdanielk1977 8484adee20fSdanielk1977 sqlite3VdbeResolveLabel(pParse->pVdbe, endTrigger); 849c3f9bad2Sdanielk1977 } 850c3f9bad2Sdanielk1977 } 8518f2c54e6Sdanielk1977 if( piOldColMask ) *piOldColMask |= trigStackEntry.oldColMask; 8528f2c54e6Sdanielk1977 if( piNewColMask ) *piNewColMask |= trigStackEntry.newColMask; 853c3f9bad2Sdanielk1977 return 0; 854c3f9bad2Sdanielk1977 } 855b7f9164eSdrh #endif /* !defined(SQLITE_OMIT_TRIGGER) */ 856