xref: /sqlite-3.40.0/src/trigger.c (revision 9bb575fd)
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 
15c3f9bad2Sdanielk1977 /*
164b59ab5eSdrh ** Delete a linked list of TriggerStep structures.
174b59ab5eSdrh */
184adee20fSdanielk1977 void sqlite3DeleteTriggerStep(TriggerStep *pTriggerStep){
194b59ab5eSdrh   while( pTriggerStep ){
204b59ab5eSdrh     TriggerStep * pTmp = pTriggerStep;
214b59ab5eSdrh     pTriggerStep = pTriggerStep->pNext;
224b59ab5eSdrh 
238c74a8caSdrh     if( pTmp->target.dyn ) sqliteFree((char*)pTmp->target.z);
244adee20fSdanielk1977     sqlite3ExprDelete(pTmp->pWhere);
254adee20fSdanielk1977     sqlite3ExprListDelete(pTmp->pExprList);
264adee20fSdanielk1977     sqlite3SelectDelete(pTmp->pSelect);
274adee20fSdanielk1977     sqlite3IdListDelete(pTmp->pIdList);
284b59ab5eSdrh 
294b59ab5eSdrh     sqliteFree(pTmp);
304b59ab5eSdrh   }
314b59ab5eSdrh }
324b59ab5eSdrh 
334b59ab5eSdrh /*
34f0f258b1Sdrh ** This is called by the parser when it sees a CREATE TRIGGER statement
35f0f258b1Sdrh ** up to the point of the BEGIN before the trigger actions.  A Trigger
36f0f258b1Sdrh ** structure is generated based on the information available and stored
37f0f258b1Sdrh ** in pParse->pNewTrigger.  After the trigger actions have been parsed, the
384adee20fSdanielk1977 ** sqlite3FinishTrigger() function is called to complete the trigger
39f0f258b1Sdrh ** construction process.
40c3f9bad2Sdanielk1977 */
414adee20fSdanielk1977 void sqlite3BeginTrigger(
42c3f9bad2Sdanielk1977   Parse *pParse,      /* The parse context of the CREATE TRIGGER statement */
43ef2cb63eSdanielk1977   Token *pName1,      /* The name of the trigger */
44ef2cb63eSdanielk1977   Token *pName2,      /* The name of the trigger */
45d702fccbSdanielk1977   int tr_tm,          /* One of TK_BEFORE, TK_AFTER, TK_INSTEAD */
46c3f9bad2Sdanielk1977   int op,             /* One of TK_INSERT, TK_UPDATE, TK_DELETE */
47633ed08dSdanielk1977   IdList *pColumns,   /* column list if this is an UPDATE OF trigger */
48d24cc427Sdrh   SrcList *pTableName,/* The name of the table/view the trigger applies to */
49c3f9bad2Sdanielk1977   int foreach,        /* One of TK_ROW or TK_STATEMENT */
50c3f9bad2Sdanielk1977   Expr *pWhen,        /* WHEN clause */
51f0f258b1Sdrh   int isTemp          /* True if the TEMPORARY keyword is present */
529adf9ac4Sdrh ){
53ef2cb63eSdanielk1977   Trigger *pTrigger;
54ef2cb63eSdanielk1977   Table *pTab;
55e5f9c644Sdrh   char *zName = 0;        /* Name of the trigger */
56*9bb575fdSdrh   sqlite3 *db = pParse->db;
57ef2cb63eSdanielk1977   int iDb;                /* The database to store the trigger in */
58ef2cb63eSdanielk1977   Token *pName;           /* The unqualified db name */
594312db55Sdrh   DbFixer sFix;
60ed6c8671Sdrh 
61ef2cb63eSdanielk1977   if( isTemp ){
62ef2cb63eSdanielk1977     /* If TEMP was specified, then the trigger name may not be qualified. */
63ef2cb63eSdanielk1977     if( pName2 && pName2->n>0 ){
64ef2cb63eSdanielk1977       sqlite3ErrorMsg(pParse, "temporary trigger may not have qualified name");
65ef2cb63eSdanielk1977       goto trigger_cleanup;
66ef2cb63eSdanielk1977     }
67ef2cb63eSdanielk1977     iDb = 1;
68ef2cb63eSdanielk1977     pName = pName1;
69ef2cb63eSdanielk1977   }else{
70ef2cb63eSdanielk1977     /* Figure out the db that the the trigger will be created in */
71ef2cb63eSdanielk1977     iDb = sqlite3TwoPartName(pParse, pName1, pName2, &pName);
72ef2cb63eSdanielk1977     if( iDb<0 ){
73ef2cb63eSdanielk1977       goto trigger_cleanup;
74ef2cb63eSdanielk1977     }
75ef2cb63eSdanielk1977   }
76ef2cb63eSdanielk1977 
77ef2cb63eSdanielk1977   /* If the trigger name was unqualified, and the table is a temp table,
78ef2cb63eSdanielk1977   ** then set iDb to 1 to create the trigger in the temporary database.
79ef2cb63eSdanielk1977   ** If sqlite3SrcListLookup() returns 0, indicating the table does not
80ef2cb63eSdanielk1977   ** exist, the error is caught by the block below.
81c3f9bad2Sdanielk1977   */
82645f63edSdrh   if( !pTableName || sqlite3_malloc_failed ) goto trigger_cleanup;
83ef2cb63eSdanielk1977   pTab = sqlite3SrcListLookup(pParse, pTableName);
84ef2cb63eSdanielk1977   if( pName2->n==0 && pTab && pTab->iDb==1 ){
85ef2cb63eSdanielk1977     iDb = 1;
86ef2cb63eSdanielk1977   }
87ef2cb63eSdanielk1977 
88ef2cb63eSdanielk1977   /* Ensure the table name matches database name and that the table exists */
896f8a503dSdanielk1977   if( sqlite3_malloc_failed ) goto trigger_cleanup;
90d24cc427Sdrh   assert( pTableName->nSrc==1 );
91ef2cb63eSdanielk1977   if( sqlite3FixInit(&sFix, pParse, iDb, "trigger", pName) &&
92ef2cb63eSdanielk1977       sqlite3FixSrcList(&sFix, pTableName) ){
934312db55Sdrh     goto trigger_cleanup;
94f26e09c8Sdrh   }
95ef2cb63eSdanielk1977   pTab = sqlite3SrcListLookup(pParse, pTableName);
96ef2cb63eSdanielk1977   if( !pTab ){
97ef2cb63eSdanielk1977     /* The table does not exist. */
98d24cc427Sdrh     goto trigger_cleanup;
99d24cc427Sdrh   }
100d24cc427Sdrh 
101d8123366Sdanielk1977   /* Check that the trigger name is not reserved and that no trigger of the
102d8123366Sdanielk1977   ** specified name exists */
103a99db3b6Sdrh   zName = sqlite3NameFromToken(pName);
104d8123366Sdanielk1977   if( !zName || SQLITE_OK!=sqlite3CheckObjectName(pParse, zName) ){
105d8123366Sdanielk1977     goto trigger_cleanup;
106d8123366Sdanielk1977   }
1074adee20fSdanielk1977   if( sqlite3HashFind(&(db->aDb[iDb].trigHash), zName,pName->n+1) ){
1084adee20fSdanielk1977     sqlite3ErrorMsg(pParse, "trigger %T already exists", pName);
109c3f9bad2Sdanielk1977     goto trigger_cleanup;
110c3f9bad2Sdanielk1977   }
111ef2cb63eSdanielk1977 
112ef2cb63eSdanielk1977   /* Do not create a trigger on a system table */
113ef2cb63eSdanielk1977   if( (iDb!=1 && sqlite3StrICmp(pTab->zName, MASTER_NAME)==0) ||
114ef2cb63eSdanielk1977       (iDb==1 && sqlite3StrICmp(pTab->zName, TEMP_MASTER_NAME)==0)
115ef2cb63eSdanielk1977   ){
1164adee20fSdanielk1977     sqlite3ErrorMsg(pParse, "cannot create trigger on system table");
117e0bc4048Sdrh     pParse->nErr++;
118e0bc4048Sdrh     goto trigger_cleanup;
119e0bc4048Sdrh   }
120ef2cb63eSdanielk1977 
121ef2cb63eSdanielk1977   /* INSTEAD of triggers are only for views and views only support INSTEAD
122ef2cb63eSdanielk1977   ** of triggers.
123ef2cb63eSdanielk1977   */
124ef2cb63eSdanielk1977   if( pTab->pSelect && tr_tm!=TK_INSTEAD ){
1254adee20fSdanielk1977     sqlite3ErrorMsg(pParse, "cannot create %s trigger on view: %S",
126da93d238Sdrh         (tr_tm == TK_BEFORE)?"BEFORE":"AFTER", pTableName, 0);
127d702fccbSdanielk1977     goto trigger_cleanup;
128d702fccbSdanielk1977   }
129ef2cb63eSdanielk1977   if( !pTab->pSelect && tr_tm==TK_INSTEAD ){
1304adee20fSdanielk1977     sqlite3ErrorMsg(pParse, "cannot create INSTEAD OF"
131da93d238Sdrh         " trigger on table: %S", pTableName, 0);
132d702fccbSdanielk1977     goto trigger_cleanup;
133d702fccbSdanielk1977   }
134ef2cb63eSdanielk1977 
135e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION
136e5f9c644Sdrh   {
137e5f9c644Sdrh     int code = SQLITE_CREATE_TRIGGER;
138ef2cb63eSdanielk1977     const char *zDb = db->aDb[pTab->iDb].zName;
139e22a334bSdrh     const char *zDbTrig = isTemp ? db->aDb[1].zName : zDb;
140ef2cb63eSdanielk1977     if( pTab->iDb==1 || isTemp ) code = SQLITE_CREATE_TEMP_TRIGGER;
141ef2cb63eSdanielk1977     if( sqlite3AuthCheck(pParse, code, zName, pTab->zName, zDbTrig) ){
142ed6c8671Sdrh       goto trigger_cleanup;
143ed6c8671Sdrh     }
144ef2cb63eSdanielk1977     if( sqlite3AuthCheck(pParse, SQLITE_INSERT, SCHEMA_TABLE(pTab->iDb),0,zDb)){
14577ad4e41Sdrh       goto trigger_cleanup;
14677ad4e41Sdrh     }
147d702fccbSdanielk1977   }
148e5f9c644Sdrh #endif
149d702fccbSdanielk1977 
150ef2cb63eSdanielk1977   /* INSTEAD OF triggers can only appear on views and BEFORE triggers
1515cf590c1Sdrh   ** cannot appear on views.  So we might as well translate every
1525cf590c1Sdrh   ** INSTEAD OF trigger into a BEFORE trigger.  It simplifies code
1535cf590c1Sdrh   ** elsewhere.
1545cf590c1Sdrh   */
155d702fccbSdanielk1977   if (tr_tm == TK_INSTEAD){
156d702fccbSdanielk1977     tr_tm = TK_BEFORE;
157c3f9bad2Sdanielk1977   }
158c3f9bad2Sdanielk1977 
159c3f9bad2Sdanielk1977   /* Build the Trigger object */
160ef2cb63eSdanielk1977   pTrigger = (Trigger*)sqliteMalloc(sizeof(Trigger));
161ef2cb63eSdanielk1977   if( pTrigger==0 ) goto trigger_cleanup;
162ef2cb63eSdanielk1977   pTrigger->name = zName;
163e5f9c644Sdrh   zName = 0;
164ef2cb63eSdanielk1977   pTrigger->table = sqliteStrDup(pTableName->a[0].zName);
1656f8a503dSdanielk1977   if( sqlite3_malloc_failed ) goto trigger_cleanup;
166ef2cb63eSdanielk1977   pTrigger->iDb = iDb;
167ef2cb63eSdanielk1977   pTrigger->iTabDb = pTab->iDb;
168ef2cb63eSdanielk1977   pTrigger->op = op;
169ef2cb63eSdanielk1977   pTrigger->tr_tm = tr_tm;
170ef2cb63eSdanielk1977   pTrigger->pWhen = sqlite3ExprDup(pWhen);
171ef2cb63eSdanielk1977   pTrigger->pColumns = sqlite3IdListDup(pColumns);
172ef2cb63eSdanielk1977   pTrigger->foreach = foreach;
173ef2cb63eSdanielk1977   sqlite3TokenCopy(&pTrigger->nameToken,pName);
174f0f258b1Sdrh   assert( pParse->pNewTrigger==0 );
175ef2cb63eSdanielk1977   pParse->pNewTrigger = pTrigger;
176f0f258b1Sdrh 
177f0f258b1Sdrh trigger_cleanup:
178f0f258b1Sdrh   sqliteFree(zName);
1794adee20fSdanielk1977   sqlite3SrcListDelete(pTableName);
1804adee20fSdanielk1977   sqlite3IdListDelete(pColumns);
1814adee20fSdanielk1977   sqlite3ExprDelete(pWhen);
182f0f258b1Sdrh }
183f0f258b1Sdrh 
184f0f258b1Sdrh /*
185f0f258b1Sdrh ** This routine is called after all of the trigger actions have been parsed
186f0f258b1Sdrh ** in order to complete the process of building the trigger.
187f0f258b1Sdrh */
1884adee20fSdanielk1977 void sqlite3FinishTrigger(
189f0f258b1Sdrh   Parse *pParse,          /* Parser context */
190f0f258b1Sdrh   TriggerStep *pStepList, /* The triggered program */
191f0f258b1Sdrh   Token *pAll             /* Token that describes the complete CREATE TRIGGER */
192f0f258b1Sdrh ){
193f26e09c8Sdrh   Trigger *nt = 0;          /* The trigger whose construction is finishing up */
194*9bb575fdSdrh   sqlite3 *db = pParse->db;  /* The database */
195f26e09c8Sdrh   DbFixer sFix;
196f0f258b1Sdrh 
197f0f258b1Sdrh   if( pParse->nErr || pParse->pNewTrigger==0 ) goto triggerfinish_cleanup;
198f0f258b1Sdrh   nt = pParse->pNewTrigger;
199f0f258b1Sdrh   pParse->pNewTrigger = 0;
200633ed08dSdanielk1977   nt->step_list = pStepList;
201a69d9168Sdrh   while( pStepList ){
202a69d9168Sdrh     pStepList->pTrig = nt;
203a69d9168Sdrh     pStepList = pStepList->pNext;
204a69d9168Sdrh   }
2054adee20fSdanielk1977   if( sqlite3FixInit(&sFix, pParse, nt->iDb, "trigger", &nt->nameToken)
2064adee20fSdanielk1977           && sqlite3FixTriggerStep(&sFix, nt->step_list) ){
207f26e09c8Sdrh     goto triggerfinish_cleanup;
208f26e09c8Sdrh   }
209c3f9bad2Sdanielk1977 
210c3f9bad2Sdanielk1977   /* if we are not initializing, and this trigger is not on a TEMP table,
2119adf9ac4Sdrh   ** build the sqlite_master entry
2129adf9ac4Sdrh   */
2131d85d931Sdrh   if( !db->init.busy ){
214905793e2Sdrh     static VdbeOpList insertTrig[] = {
215c977f7f5Sdrh       { OP_NewRecno,   0, 0,  0          },
2160f69c1e3Sdanielk1977       { OP_String8,    0, 0,  "trigger"  },
2170f69c1e3Sdanielk1977       { OP_String8,    0, 0,  0          },  /* 2: trigger name */
2180f69c1e3Sdanielk1977       { OP_String8,    0, 0,  0          },  /* 3: table name */
219c977f7f5Sdrh       { OP_Integer,    0, 0,  0          },
2200f69c1e3Sdanielk1977       { OP_String8,    0, 0,  "CREATE TRIGGER "},
2210f69c1e3Sdanielk1977       { OP_String8,    0, 0,  0          },  /* 6: SQL */
222855eb1cfSdrh       { OP_Concat,     0, 0,  0          },
223bf3b721fSdanielk1977       { OP_MakeRecord, 5, 0,  "tttit"    },
224c977f7f5Sdrh       { OP_PutIntKey,  0, 0,  0          },
225c977f7f5Sdrh     };
226c977f7f5Sdrh     int addr;
227c977f7f5Sdrh     Vdbe *v;
228c3f9bad2Sdanielk1977 
229c3f9bad2Sdanielk1977     /* Make an entry in the sqlite_master table */
2304adee20fSdanielk1977     v = sqlite3GetVdbe(pParse);
231f0f258b1Sdrh     if( v==0 ) goto triggerfinish_cleanup;
232ef2cb63eSdanielk1977     sqlite3BeginWriteOperation(pParse, 0, nt->iDb);
2334adee20fSdanielk1977     sqlite3OpenMasterTable(v, nt->iDb);
2344adee20fSdanielk1977     addr = sqlite3VdbeAddOpList(v, ArraySize(insertTrig), insertTrig);
2354adee20fSdanielk1977     sqlite3VdbeChangeP3(v, addr+2, nt->name, 0);
2364adee20fSdanielk1977     sqlite3VdbeChangeP3(v, addr+3, nt->table, 0);
2373df6b257Sdanielk1977     sqlite3VdbeChangeP3(v, addr+6, pAll->z, pAll->n);
2381d850a72Sdanielk1977     if( nt->iDb!=0 ){
2391d850a72Sdanielk1977       sqlite3ChangeCookie(db, v, nt->iDb);
240e0bc4048Sdrh     }
2414adee20fSdanielk1977     sqlite3VdbeAddOp(v, OP_Close, 0, 0);
242956bc92cSdrh     sqlite3VdbeOp3(v, OP_ParseSchema, nt->iDb, 0,
243956bc92cSdrh        sqlite3MPrintf("type='trigger' AND name='%q'", nt->name), P3_DYNAMIC);
2444adee20fSdanielk1977     sqlite3EndWriteOperation(pParse);
245c3f9bad2Sdanielk1977   }
246c3f9bad2Sdanielk1977 
247956bc92cSdrh   if( db->init.busy ){
248f0f258b1Sdrh     Table *pTab;
2494adee20fSdanielk1977     sqlite3HashInsert(&db->aDb[nt->iDb].trigHash,
250f0f258b1Sdrh                      nt->name, strlen(nt->name)+1, nt);
2514adee20fSdanielk1977     pTab = sqlite3LocateTable(pParse, nt->table, db->aDb[nt->iTabDb].zName);
252f0f258b1Sdrh     assert( pTab!=0 );
253f0f258b1Sdrh     nt->pNext = pTab->pTrigger;
254f0f258b1Sdrh     pTab->pTrigger = nt;
255f26e09c8Sdrh     nt = 0;
256c3f9bad2Sdanielk1977   }
257c3f9bad2Sdanielk1977 
258f0f258b1Sdrh triggerfinish_cleanup:
2594adee20fSdanielk1977   sqlite3DeleteTrigger(nt);
2604adee20fSdanielk1977   sqlite3DeleteTrigger(pParse->pNewTrigger);
261f0f258b1Sdrh   pParse->pNewTrigger = 0;
2624adee20fSdanielk1977   sqlite3DeleteTriggerStep(pStepList);
263c3f9bad2Sdanielk1977 }
2644b59ab5eSdrh 
2654b59ab5eSdrh /*
2664b59ab5eSdrh ** Make a copy of all components of the given trigger step.  This has
2674b59ab5eSdrh ** the effect of copying all Expr.token.z values into memory obtained
2684b59ab5eSdrh ** from sqliteMalloc().  As initially created, the Expr.token.z values
2694b59ab5eSdrh ** all point to the input string that was fed to the parser.  But that
2706f8a503dSdanielk1977 ** string is ephemeral - it will go away as soon as the sqlite3_exec()
2714b59ab5eSdrh ** call that started the parser exits.  This routine makes a persistent
2724b59ab5eSdrh ** copy of all the Expr.token.z strings so that the TriggerStep structure
2736f8a503dSdanielk1977 ** will be valid even after the sqlite3_exec() call returns.
2744b59ab5eSdrh */
2754b59ab5eSdrh static void sqlitePersistTriggerStep(TriggerStep *p){
2764b59ab5eSdrh   if( p->target.z ){
2774b59ab5eSdrh     p->target.z = sqliteStrNDup(p->target.z, p->target.n);
2784b59ab5eSdrh     p->target.dyn = 1;
2794b59ab5eSdrh   }
2804b59ab5eSdrh   if( p->pSelect ){
2814adee20fSdanielk1977     Select *pNew = sqlite3SelectDup(p->pSelect);
2824adee20fSdanielk1977     sqlite3SelectDelete(p->pSelect);
2834b59ab5eSdrh     p->pSelect = pNew;
2844b59ab5eSdrh   }
2854b59ab5eSdrh   if( p->pWhere ){
2864adee20fSdanielk1977     Expr *pNew = sqlite3ExprDup(p->pWhere);
2874adee20fSdanielk1977     sqlite3ExprDelete(p->pWhere);
2884b59ab5eSdrh     p->pWhere = pNew;
2894b59ab5eSdrh   }
2904b59ab5eSdrh   if( p->pExprList ){
2914adee20fSdanielk1977     ExprList *pNew = sqlite3ExprListDup(p->pExprList);
2924adee20fSdanielk1977     sqlite3ExprListDelete(p->pExprList);
2934b59ab5eSdrh     p->pExprList = pNew;
2944b59ab5eSdrh   }
2954b59ab5eSdrh   if( p->pIdList ){
2964adee20fSdanielk1977     IdList *pNew = sqlite3IdListDup(p->pIdList);
2974adee20fSdanielk1977     sqlite3IdListDelete(p->pIdList);
2984b59ab5eSdrh     p->pIdList = pNew;
299c3f9bad2Sdanielk1977   }
300c3f9bad2Sdanielk1977 }
301c3f9bad2Sdanielk1977 
302c977f7f5Sdrh /*
303c977f7f5Sdrh ** Turn a SELECT statement (that the pSelect parameter points to) into
304c977f7f5Sdrh ** a trigger step.  Return a pointer to a TriggerStep structure.
305c977f7f5Sdrh **
306c977f7f5Sdrh ** The parser calls this routine when it finds a SELECT statement in
307c977f7f5Sdrh ** body of a TRIGGER.
308c977f7f5Sdrh */
3094adee20fSdanielk1977 TriggerStep *sqlite3TriggerSelectStep(Select *pSelect){
310633ed08dSdanielk1977   TriggerStep *pTriggerStep = sqliteMalloc(sizeof(TriggerStep));
311e4697f5eSdrh   if( pTriggerStep==0 ) return 0;
312c3f9bad2Sdanielk1977 
313633ed08dSdanielk1977   pTriggerStep->op = TK_SELECT;
314633ed08dSdanielk1977   pTriggerStep->pSelect = pSelect;
315633ed08dSdanielk1977   pTriggerStep->orconf = OE_Default;
3164b59ab5eSdrh   sqlitePersistTriggerStep(pTriggerStep);
317c3f9bad2Sdanielk1977 
318633ed08dSdanielk1977   return pTriggerStep;
319c3f9bad2Sdanielk1977 }
320c3f9bad2Sdanielk1977 
321c977f7f5Sdrh /*
322c977f7f5Sdrh ** Build a trigger step out of an INSERT statement.  Return a pointer
323c977f7f5Sdrh ** to the new trigger step.
324c977f7f5Sdrh **
325c977f7f5Sdrh ** The parser calls this routine when it sees an INSERT inside the
326c977f7f5Sdrh ** body of a trigger.
327c977f7f5Sdrh */
3284adee20fSdanielk1977 TriggerStep *sqlite3TriggerInsertStep(
329c977f7f5Sdrh   Token *pTableName,  /* Name of the table into which we insert */
330c977f7f5Sdrh   IdList *pColumn,    /* List of columns in pTableName to insert into */
331c977f7f5Sdrh   ExprList *pEList,   /* The VALUE clause: a list of values to be inserted */
332c977f7f5Sdrh   Select *pSelect,    /* A SELECT statement that supplies values */
333c977f7f5Sdrh   int orconf          /* The conflict algorithm (OE_Abort, OE_Replace, etc.) */
334633ed08dSdanielk1977 ){
335633ed08dSdanielk1977   TriggerStep *pTriggerStep = sqliteMalloc(sizeof(TriggerStep));
336e4697f5eSdrh   if( pTriggerStep==0 ) return 0;
337c3f9bad2Sdanielk1977 
338633ed08dSdanielk1977   assert(pEList == 0 || pSelect == 0);
339633ed08dSdanielk1977   assert(pEList != 0 || pSelect != 0);
340c3f9bad2Sdanielk1977 
341633ed08dSdanielk1977   pTriggerStep->op = TK_INSERT;
342633ed08dSdanielk1977   pTriggerStep->pSelect = pSelect;
343633ed08dSdanielk1977   pTriggerStep->target  = *pTableName;
344633ed08dSdanielk1977   pTriggerStep->pIdList = pColumn;
345633ed08dSdanielk1977   pTriggerStep->pExprList = pEList;
346633ed08dSdanielk1977   pTriggerStep->orconf = orconf;
3474b59ab5eSdrh   sqlitePersistTriggerStep(pTriggerStep);
348c3f9bad2Sdanielk1977 
349633ed08dSdanielk1977   return pTriggerStep;
350c3f9bad2Sdanielk1977 }
351c3f9bad2Sdanielk1977 
352c977f7f5Sdrh /*
353c977f7f5Sdrh ** Construct a trigger step that implements an UPDATE statement and return
354c977f7f5Sdrh ** a pointer to that trigger step.  The parser calls this routine when it
355c977f7f5Sdrh ** sees an UPDATE statement inside the body of a CREATE TRIGGER.
356c977f7f5Sdrh */
3574adee20fSdanielk1977 TriggerStep *sqlite3TriggerUpdateStep(
358c977f7f5Sdrh   Token *pTableName,   /* Name of the table to be updated */
359c977f7f5Sdrh   ExprList *pEList,    /* The SET clause: list of column and new values */
360c977f7f5Sdrh   Expr *pWhere,        /* The WHERE clause */
361c977f7f5Sdrh   int orconf           /* The conflict algorithm. (OE_Abort, OE_Ignore, etc) */
362c977f7f5Sdrh ){
363633ed08dSdanielk1977   TriggerStep *pTriggerStep = sqliteMalloc(sizeof(TriggerStep));
364e4697f5eSdrh   if( pTriggerStep==0 ) return 0;
365c3f9bad2Sdanielk1977 
366633ed08dSdanielk1977   pTriggerStep->op = TK_UPDATE;
367633ed08dSdanielk1977   pTriggerStep->target  = *pTableName;
368633ed08dSdanielk1977   pTriggerStep->pExprList = pEList;
369633ed08dSdanielk1977   pTriggerStep->pWhere = pWhere;
370633ed08dSdanielk1977   pTriggerStep->orconf = orconf;
3714b59ab5eSdrh   sqlitePersistTriggerStep(pTriggerStep);
372c3f9bad2Sdanielk1977 
373633ed08dSdanielk1977   return pTriggerStep;
374c3f9bad2Sdanielk1977 }
375c3f9bad2Sdanielk1977 
376c977f7f5Sdrh /*
377c977f7f5Sdrh ** Construct a trigger step that implements a DELETE statement and return
378c977f7f5Sdrh ** a pointer to that trigger step.  The parser calls this routine when it
379c977f7f5Sdrh ** sees a DELETE statement inside the body of a CREATE TRIGGER.
380c977f7f5Sdrh */
3814adee20fSdanielk1977 TriggerStep *sqlite3TriggerDeleteStep(Token *pTableName, Expr *pWhere){
382633ed08dSdanielk1977   TriggerStep *pTriggerStep = sqliteMalloc(sizeof(TriggerStep));
383e4697f5eSdrh   if( pTriggerStep==0 ) return 0;
384c3f9bad2Sdanielk1977 
385633ed08dSdanielk1977   pTriggerStep->op = TK_DELETE;
386633ed08dSdanielk1977   pTriggerStep->target  = *pTableName;
387633ed08dSdanielk1977   pTriggerStep->pWhere = pWhere;
388633ed08dSdanielk1977   pTriggerStep->orconf = OE_Default;
3894b59ab5eSdrh   sqlitePersistTriggerStep(pTriggerStep);
390c3f9bad2Sdanielk1977 
391633ed08dSdanielk1977   return pTriggerStep;
392c3f9bad2Sdanielk1977 }
393c3f9bad2Sdanielk1977 
394c3f9bad2Sdanielk1977 /*
395633ed08dSdanielk1977 ** Recursively delete a Trigger structure
396c3f9bad2Sdanielk1977 */
3974adee20fSdanielk1977 void sqlite3DeleteTrigger(Trigger *pTrigger){
398f0f258b1Sdrh   if( pTrigger==0 ) return;
3994adee20fSdanielk1977   sqlite3DeleteTriggerStep(pTrigger->step_list);
400633ed08dSdanielk1977   sqliteFree(pTrigger->name);
401633ed08dSdanielk1977   sqliteFree(pTrigger->table);
4024adee20fSdanielk1977   sqlite3ExprDelete(pTrigger->pWhen);
4034adee20fSdanielk1977   sqlite3IdListDelete(pTrigger->pColumns);
4044312db55Sdrh   if( pTrigger->nameToken.dyn ) sqliteFree((char*)pTrigger->nameToken.z);
405633ed08dSdanielk1977   sqliteFree(pTrigger);
406633ed08dSdanielk1977 }
407633ed08dSdanielk1977 
408633ed08dSdanielk1977 /*
4098a41449eSdanielk1977 ** This function is called to drop a trigger from the database schema.
4108a41449eSdanielk1977 **
4118a41449eSdanielk1977 ** This may be called directly from the parser and therefore identifies
4128a41449eSdanielk1977 ** the trigger by name.  The sqlite3DropTriggerPtr() routine does the
4138a41449eSdanielk1977 ** same job as this routine except it takes a pointer to the trigger
4148a41449eSdanielk1977 ** instead of the trigger name.
4158a41449eSdanielk1977 **/
4164adee20fSdanielk1977 void sqlite3DropTrigger(Parse *pParse, SrcList *pName){
417742f947bSdanielk1977   Trigger *pTrigger = 0;
418d24cc427Sdrh   int i;
419d24cc427Sdrh   const char *zDb;
420d24cc427Sdrh   const char *zName;
421d24cc427Sdrh   int nName;
422*9bb575fdSdrh   sqlite3 *db = pParse->db;
423633ed08dSdanielk1977 
4246f8a503dSdanielk1977   if( sqlite3_malloc_failed ) goto drop_trigger_cleanup;
4258a41449eSdanielk1977   if( SQLITE_OK!=sqlite3ReadSchema(pParse) ){
426c0391398Sdanielk1977     goto drop_trigger_cleanup;
427c0391398Sdanielk1977   }
4288e227875Sdanielk1977 
429d24cc427Sdrh   assert( pName->nSrc==1 );
430d24cc427Sdrh   zDb = pName->a[0].zDatabase;
431d24cc427Sdrh   zName = pName->a[0].zName;
432d24cc427Sdrh   nName = strlen(zName);
433d24cc427Sdrh   for(i=0; i<db->nDb; i++){
434812d7a21Sdrh     int j = (i<2) ? i^1 : i;  /* Search TEMP before MAIN */
4354adee20fSdanielk1977     if( zDb && sqlite3StrICmp(db->aDb[j].zName, zDb) ) continue;
4364adee20fSdanielk1977     pTrigger = sqlite3HashFind(&(db->aDb[j].trigHash), zName, nName+1);
437d24cc427Sdrh     if( pTrigger ) break;
438c3f9bad2Sdanielk1977   }
439d24cc427Sdrh   if( !pTrigger ){
4404adee20fSdanielk1977     sqlite3ErrorMsg(pParse, "no such trigger: %S", pName, 0);
441d24cc427Sdrh     goto drop_trigger_cleanup;
442d24cc427Sdrh   }
4434adee20fSdanielk1977   sqlite3DropTriggerPtr(pParse, pTrigger, 0);
44479a519c0Sdrh 
44579a519c0Sdrh drop_trigger_cleanup:
4464adee20fSdanielk1977   sqlite3SrcListDelete(pName);
44779a519c0Sdrh }
44879a519c0Sdrh 
44979a519c0Sdrh /*
450956bc92cSdrh ** Return a pointer to the Table structure for the table that a trigger
451956bc92cSdrh ** is set on.
452956bc92cSdrh */
453956bc92cSdrh static Table *tableOfTrigger(sqlite3 *db, Trigger *pTrigger){
454956bc92cSdrh   return sqlite3FindTable(db,pTrigger->table,db->aDb[pTrigger->iTabDb].zName);
455956bc92cSdrh }
456956bc92cSdrh 
457956bc92cSdrh 
458956bc92cSdrh /*
45979a519c0Sdrh ** Drop a trigger given a pointer to that trigger.  If nested is false,
46079a519c0Sdrh ** then also generate code to remove the trigger from the SQLITE_MASTER
46179a519c0Sdrh ** table.
46279a519c0Sdrh */
4634adee20fSdanielk1977 void sqlite3DropTriggerPtr(Parse *pParse, Trigger *pTrigger, int nested){
46479a519c0Sdrh   Table   *pTable;
46579a519c0Sdrh   Vdbe *v;
466*9bb575fdSdrh   sqlite3 *db = pParse->db;
467956bc92cSdrh   int iDb;
46879a519c0Sdrh 
469956bc92cSdrh   iDb = pTrigger->iDb;
470956bc92cSdrh   assert( iDb>=0 && iDb<db->nDb );
471956bc92cSdrh   pTable = tableOfTrigger(db, pTrigger);
472ed6c8671Sdrh   assert(pTable);
473956bc92cSdrh   assert( pTable->iDb==iDb || iDb==1 );
474e5f9c644Sdrh #ifndef SQLITE_OMIT_AUTHORIZATION
475e5f9c644Sdrh   {
476e5f9c644Sdrh     int code = SQLITE_DROP_TRIGGER;
477956bc92cSdrh     const char *zDb = db->aDb[iDb].zName;
478956bc92cSdrh     const char *zTab = SCHEMA_TABLE(iDb);
479956bc92cSdrh     if( iDb==1 ) code = SQLITE_DROP_TEMP_TRIGGER;
4804adee20fSdanielk1977     if( sqlite3AuthCheck(pParse, code, pTrigger->name, pTable->zName, zDb) ||
4814adee20fSdanielk1977       sqlite3AuthCheck(pParse, SQLITE_DELETE, zTab, 0, zDb) ){
48279a519c0Sdrh       return;
483ed6c8671Sdrh     }
484e5f9c644Sdrh   }
485e5f9c644Sdrh #endif
486c3f9bad2Sdanielk1977 
487f0f258b1Sdrh   /* Generate code to destroy the database record of the trigger.
488f0f258b1Sdrh   */
4898e227875Sdanielk1977   if( pTable!=0 && (v = sqlite3GetVdbe(pParse))!=0 ){
490f0f258b1Sdrh     int base;
491905793e2Sdrh     static VdbeOpList dropTrigger[] = {
4929b1b01bbSdrh       { OP_Rewind,     0, ADDR(9),  0},
4930f69c1e3Sdanielk1977       { OP_String8,     0, 0,        0}, /* 1 */
494f0f258b1Sdrh       { OP_Column,     0, 1,        0},
4959b1b01bbSdrh       { OP_Ne,         0, ADDR(8),  0},
4960f69c1e3Sdanielk1977       { OP_String8,     0, 0,        "trigger"},
4979b1b01bbSdrh       { OP_Column,     0, 0,        0},
4989b1b01bbSdrh       { OP_Ne,         0, ADDR(8),  0},
499f0f258b1Sdrh       { OP_Delete,     0, 0,        0},
5009b1b01bbSdrh       { OP_Next,       0, ADDR(1),  0}, /* 8 */
501f0f258b1Sdrh     };
502f0f258b1Sdrh 
503956bc92cSdrh     sqlite3BeginWriteOperation(pParse, 0, iDb);
504956bc92cSdrh     sqlite3OpenMasterTable(v, iDb);
5054adee20fSdanielk1977     base = sqlite3VdbeAddOpList(v,  ArraySize(dropTrigger), dropTrigger);
5064adee20fSdanielk1977     sqlite3VdbeChangeP3(v, base+1, pTrigger->name, 0);
507956bc92cSdrh     sqlite3ChangeCookie(db, v, iDb);
5084adee20fSdanielk1977     sqlite3VdbeAddOp(v, OP_Close, 0, 0);
509956bc92cSdrh     sqlite3VdbeOp3(v, OP_DropTrigger, iDb, 0, pTrigger->name, 0);
510956bc92cSdrh   }
511f0f258b1Sdrh }
512f0f258b1Sdrh 
513c3f9bad2Sdanielk1977 /*
514956bc92cSdrh ** Remove a trigger from the hash tables of the sqlite* pointer.
515c3f9bad2Sdanielk1977 */
516956bc92cSdrh void sqlite3UnlinkAndDeleteTrigger(sqlite3 *db, int iDb, const char *zName){
517956bc92cSdrh   Trigger *pTrigger;
51879a519c0Sdrh   int nName = strlen(zName);
519956bc92cSdrh   pTrigger = sqlite3HashInsert(&(db->aDb[iDb].trigHash), zName, nName+1, 0);
520956bc92cSdrh   if( pTrigger ){
521956bc92cSdrh     Table *pTable = tableOfTrigger(db, pTrigger);
522956bc92cSdrh     assert( pTable!=0 );
523633ed08dSdanielk1977     if( pTable->pTrigger == pTrigger ){
524633ed08dSdanielk1977       pTable->pTrigger = pTrigger->pNext;
525633ed08dSdanielk1977     }else{
526633ed08dSdanielk1977       Trigger *cc = pTable->pTrigger;
527c3f9bad2Sdanielk1977       while( cc ){
528633ed08dSdanielk1977         if( cc->pNext == pTrigger ){
529c3f9bad2Sdanielk1977           cc->pNext = cc->pNext->pNext;
530c3f9bad2Sdanielk1977           break;
531c3f9bad2Sdanielk1977         }
532c3f9bad2Sdanielk1977         cc = cc->pNext;
533c3f9bad2Sdanielk1977       }
534c3f9bad2Sdanielk1977       assert(cc);
535c3f9bad2Sdanielk1977     }
5364adee20fSdanielk1977     sqlite3DeleteTrigger(pTrigger);
537956bc92cSdrh     db->flags |= SQLITE_InternChanges;
538c3f9bad2Sdanielk1977   }
539c3f9bad2Sdanielk1977 }
540c3f9bad2Sdanielk1977 
541c977f7f5Sdrh /*
542c977f7f5Sdrh ** pEList is the SET clause of an UPDATE statement.  Each entry
543c977f7f5Sdrh ** in pEList is of the format <id>=<expr>.  If any of the entries
544c977f7f5Sdrh ** in pEList have an <id> which matches an identifier in pIdList,
545c977f7f5Sdrh ** then return TRUE.  If pIdList==NULL, then it is considered a
546c977f7f5Sdrh ** wildcard that matches anything.  Likewise if pEList==NULL then
547c977f7f5Sdrh ** it matches anything so always return true.  Return false only
548c977f7f5Sdrh ** if there is no match.
549c977f7f5Sdrh */
550c977f7f5Sdrh static int checkColumnOverLap(IdList *pIdList, ExprList *pEList){
551ad2d8307Sdrh   int e;
552ad2d8307Sdrh   if( !pIdList || !pEList ) return 1;
553f29ce559Sdanielk1977   for(e=0; e<pEList->nExpr; e++){
5544adee20fSdanielk1977     if( sqlite3IdListIndex(pIdList, pEList->a[e].zName)>=0 ) return 1;
555f29ce559Sdanielk1977   }
556c3f9bad2Sdanielk1977   return 0;
557c3f9bad2Sdanielk1977 }
558c3f9bad2Sdanielk1977 
559c3f9bad2Sdanielk1977 /* A global variable that is TRUE if we should always set up temp tables for
560c3f9bad2Sdanielk1977  * for triggers, even if there are no triggers to code. This is used to test
561c3f9bad2Sdanielk1977  * how much overhead the triggers algorithm is causing.
562c3f9bad2Sdanielk1977  *
563c3f9bad2Sdanielk1977  * This flag can be set or cleared using the "trigger_overhead_test" pragma.
564c3f9bad2Sdanielk1977  * The pragma is not documented since it is not really part of the interface
565c3f9bad2Sdanielk1977  * to SQLite, just the test procedure.
566c3f9bad2Sdanielk1977 */
56738f8271fSdrh int sqlite3_always_code_trigger_setup = 0;
568c3f9bad2Sdanielk1977 
569c3f9bad2Sdanielk1977 /*
570c3f9bad2Sdanielk1977  * Returns true if a trigger matching op, tr_tm and foreach that is NOT already
571c3f9bad2Sdanielk1977  * on the Parse objects trigger-stack (to prevent recursive trigger firing) is
572c3f9bad2Sdanielk1977  * found in the list specified as pTrigger.
573c3f9bad2Sdanielk1977  */
5744adee20fSdanielk1977 int sqlite3TriggersExist(
575c977f7f5Sdrh   Parse *pParse,          /* Used to check for recursive triggers */
576c977f7f5Sdrh   Trigger *pTrigger,      /* A list of triggers associated with a table */
577c3f9bad2Sdanielk1977   int op,                 /* one of TK_DELETE, TK_INSERT, TK_UPDATE */
578c3f9bad2Sdanielk1977   int tr_tm,              /* one of TK_BEFORE, TK_AFTER */
579c3f9bad2Sdanielk1977   int foreach,            /* one of TK_ROW or TK_STATEMENT */
580c977f7f5Sdrh   ExprList *pChanges      /* Columns that change in an UPDATE statement */
581c977f7f5Sdrh ){
582633ed08dSdanielk1977   Trigger * pTriggerCursor;
583c3f9bad2Sdanielk1977 
58438f8271fSdrh   if( sqlite3_always_code_trigger_setup ){
585633ed08dSdanielk1977     return 1;
586633ed08dSdanielk1977   }
587c3f9bad2Sdanielk1977 
588633ed08dSdanielk1977   pTriggerCursor = pTrigger;
589633ed08dSdanielk1977   while( pTriggerCursor ){
590633ed08dSdanielk1977     if( pTriggerCursor->op == op &&
591633ed08dSdanielk1977 	pTriggerCursor->tr_tm == tr_tm &&
592633ed08dSdanielk1977 	pTriggerCursor->foreach == foreach &&
593633ed08dSdanielk1977 	checkColumnOverLap(pTriggerCursor->pColumns, pChanges) ){
594c3f9bad2Sdanielk1977       TriggerStack * ss;
595c3f9bad2Sdanielk1977       ss = pParse->trigStack;
596f29ce559Sdanielk1977       while( ss && ss->pTrigger != pTrigger ){
597f29ce559Sdanielk1977 	ss = ss->pNext;
598f29ce559Sdanielk1977       }
599c3f9bad2Sdanielk1977       if( !ss )return 1;
600c3f9bad2Sdanielk1977     }
601633ed08dSdanielk1977     pTriggerCursor = pTriggerCursor->pNext;
602c3f9bad2Sdanielk1977   }
603c3f9bad2Sdanielk1977 
604c3f9bad2Sdanielk1977   return 0;
605c3f9bad2Sdanielk1977 }
606c3f9bad2Sdanielk1977 
607c977f7f5Sdrh /*
608f26e09c8Sdrh ** Convert the pStep->target token into a SrcList and return a pointer
609f26e09c8Sdrh ** to that SrcList.
610f26e09c8Sdrh **
611f26e09c8Sdrh ** This routine adds a specific database name, if needed, to the target when
612f26e09c8Sdrh ** forming the SrcList.  This prevents a trigger in one database from
613f26e09c8Sdrh ** referring to a target in another database.  An exception is when the
614f26e09c8Sdrh ** trigger is in TEMP in which case it can refer to any other database it
615f26e09c8Sdrh ** wants.
616f26e09c8Sdrh */
617f26e09c8Sdrh static SrcList *targetSrcList(
618f26e09c8Sdrh   Parse *pParse,       /* The parsing context */
619f26e09c8Sdrh   TriggerStep *pStep   /* The trigger containing the target token */
620f26e09c8Sdrh ){
621f26e09c8Sdrh   Token sDb;           /* Dummy database name token */
622f26e09c8Sdrh   int iDb;             /* Index of the database to use */
623f26e09c8Sdrh   SrcList *pSrc;       /* SrcList to be returned */
624f26e09c8Sdrh 
625f26e09c8Sdrh   iDb = pStep->pTrig->iDb;
626f26e09c8Sdrh   if( iDb==0 || iDb>=2 ){
627f26e09c8Sdrh     assert( iDb<pParse->db->nDb );
628f26e09c8Sdrh     sDb.z = pParse->db->aDb[iDb].zName;
629f26e09c8Sdrh     sDb.n = strlen(sDb.z);
6304adee20fSdanielk1977     pSrc = sqlite3SrcListAppend(0, &sDb, &pStep->target);
631f26e09c8Sdrh   } else {
6324adee20fSdanielk1977     pSrc = sqlite3SrcListAppend(0, &pStep->target, 0);
633f26e09c8Sdrh   }
634f26e09c8Sdrh   return pSrc;
635f26e09c8Sdrh }
636f26e09c8Sdrh 
637f26e09c8Sdrh /*
638c977f7f5Sdrh ** Generate VDBE code for zero or more statements inside the body of a
639c977f7f5Sdrh ** trigger.
640c977f7f5Sdrh */
641c3f9bad2Sdanielk1977 static int codeTriggerProgram(
642c977f7f5Sdrh   Parse *pParse,            /* The parser context */
643c977f7f5Sdrh   TriggerStep *pStepList,   /* List of statements inside the trigger body */
644c977f7f5Sdrh   int orconfin              /* Conflict algorithm. (OE_Abort, etc) */
645633ed08dSdanielk1977 ){
646633ed08dSdanielk1977   TriggerStep * pTriggerStep = pStepList;
647c3f9bad2Sdanielk1977   int orconf;
648c3f9bad2Sdanielk1977 
649633ed08dSdanielk1977   while( pTriggerStep ){
650633ed08dSdanielk1977     orconf = (orconfin == OE_Default)?pTriggerStep->orconf:orconfin;
651c3f9bad2Sdanielk1977     pParse->trigStack->orconf = orconf;
652633ed08dSdanielk1977     switch( pTriggerStep->op ){
653c3f9bad2Sdanielk1977       case TK_SELECT: {
6544adee20fSdanielk1977 	Select * ss = sqlite3SelectDup(pTriggerStep->pSelect);
6556f34903eSdanielk1977 	assert(ss);
6566f34903eSdanielk1977 	assert(ss->pSrc);
657bf3b721fSdanielk1977 	sqlite3Select(pParse, ss, SRT_Discard, 0, 0, 0, 0, 0);
6584adee20fSdanielk1977 	sqlite3SelectDelete(ss);
659c3f9bad2Sdanielk1977 	break;
660c3f9bad2Sdanielk1977       }
661c3f9bad2Sdanielk1977       case TK_UPDATE: {
662113088ecSdrh         SrcList *pSrc;
663f26e09c8Sdrh         pSrc = targetSrcList(pParse, pTriggerStep);
664b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 0, 0);
6654adee20fSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ListPush, 0, 0);
6664adee20fSdanielk1977         sqlite3Update(pParse, pSrc,
6674adee20fSdanielk1977 		sqlite3ExprListDup(pTriggerStep->pExprList),
6684adee20fSdanielk1977 		sqlite3ExprDup(pTriggerStep->pWhere), orconf);
6694adee20fSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ListPop, 0, 0);
670b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 1, 0);
671c3f9bad2Sdanielk1977         break;
672c3f9bad2Sdanielk1977       }
673c3f9bad2Sdanielk1977       case TK_INSERT: {
674113088ecSdrh         SrcList *pSrc;
675f26e09c8Sdrh         pSrc = targetSrcList(pParse, pTriggerStep);
676b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 0, 0);
6774adee20fSdanielk1977         sqlite3Insert(pParse, pSrc,
6784adee20fSdanielk1977           sqlite3ExprListDup(pTriggerStep->pExprList),
6794adee20fSdanielk1977           sqlite3SelectDup(pTriggerStep->pSelect),
6804adee20fSdanielk1977           sqlite3IdListDup(pTriggerStep->pIdList), orconf);
681b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 1, 0);
682c3f9bad2Sdanielk1977         break;
683c3f9bad2Sdanielk1977       }
684c3f9bad2Sdanielk1977       case TK_DELETE: {
685113088ecSdrh         SrcList *pSrc;
686b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 0, 0);
6874adee20fSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ListPush, 0, 0);
688f26e09c8Sdrh         pSrc = targetSrcList(pParse, pTriggerStep);
6894adee20fSdanielk1977         sqlite3DeleteFrom(pParse, pSrc, sqlite3ExprDup(pTriggerStep->pWhere));
6904adee20fSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ListPop, 0, 0);
691b28af71aSdanielk1977         sqlite3VdbeAddOp(pParse->pVdbe, OP_ResetCount, 1, 0);
692c3f9bad2Sdanielk1977         break;
693c3f9bad2Sdanielk1977       }
694c3f9bad2Sdanielk1977       default:
695c3f9bad2Sdanielk1977         assert(0);
696c3f9bad2Sdanielk1977     }
697633ed08dSdanielk1977     pTriggerStep = pTriggerStep->pNext;
698c3f9bad2Sdanielk1977   }
699c3f9bad2Sdanielk1977 
700c3f9bad2Sdanielk1977   return 0;
701c3f9bad2Sdanielk1977 }
702c3f9bad2Sdanielk1977 
703633ed08dSdanielk1977 /*
704633ed08dSdanielk1977 ** This is called to code FOR EACH ROW triggers.
705633ed08dSdanielk1977 **
706633ed08dSdanielk1977 ** When the code that this function generates is executed, the following
707633ed08dSdanielk1977 ** must be true:
708c977f7f5Sdrh **
709c977f7f5Sdrh ** 1. No cursors may be open in the main database.  (But newIdx and oldIdx
710c977f7f5Sdrh **    can be indices of cursors in temporary tables.  See below.)
711c977f7f5Sdrh **
712633ed08dSdanielk1977 ** 2. If the triggers being coded are ON INSERT or ON UPDATE triggers, then
713633ed08dSdanielk1977 **    a temporary vdbe cursor (index newIdx) must be open and pointing at
714633ed08dSdanielk1977 **    a row containing values to be substituted for new.* expressions in the
715633ed08dSdanielk1977 **    trigger program(s).
716c977f7f5Sdrh **
717633ed08dSdanielk1977 ** 3. If the triggers being coded are ON DELETE or ON UPDATE triggers, then
718633ed08dSdanielk1977 **    a temporary vdbe cursor (index oldIdx) must be open and pointing at
719633ed08dSdanielk1977 **    a row containing values to be substituted for old.* expressions in the
720633ed08dSdanielk1977 **    trigger program(s).
721633ed08dSdanielk1977 **
722633ed08dSdanielk1977 */
7234adee20fSdanielk1977 int sqlite3CodeRowTrigger(
724c3f9bad2Sdanielk1977   Parse *pParse,       /* Parse context */
725c3f9bad2Sdanielk1977   int op,              /* One of TK_UPDATE, TK_INSERT, TK_DELETE */
726633ed08dSdanielk1977   ExprList *pChanges,  /* Changes list for any UPDATE OF triggers */
727c3f9bad2Sdanielk1977   int tr_tm,           /* One of TK_BEFORE, TK_AFTER */
728633ed08dSdanielk1977   Table *pTab,         /* The table to code triggers from */
729633ed08dSdanielk1977   int newIdx,          /* The indice of the "new" row to access */
730633ed08dSdanielk1977   int oldIdx,          /* The indice of the "old" row to access */
7316f34903eSdanielk1977   int orconf,          /* ON CONFLICT policy */
7326f34903eSdanielk1977   int ignoreJump       /* Instruction to jump to for RAISE(IGNORE) */
733c977f7f5Sdrh ){
734c3f9bad2Sdanielk1977   Trigger * pTrigger;
735c3f9bad2Sdanielk1977   TriggerStack * pTriggerStack;
736c3f9bad2Sdanielk1977 
737c3f9bad2Sdanielk1977   assert(op == TK_UPDATE || op == TK_INSERT || op == TK_DELETE);
738c3f9bad2Sdanielk1977   assert(tr_tm == TK_BEFORE || tr_tm == TK_AFTER );
739c3f9bad2Sdanielk1977 
740633ed08dSdanielk1977   assert(newIdx != -1 || oldIdx != -1);
741c3f9bad2Sdanielk1977 
742633ed08dSdanielk1977   pTrigger = pTab->pTrigger;
743c3f9bad2Sdanielk1977   while( pTrigger ){
744c3f9bad2Sdanielk1977     int fire_this = 0;
745c3f9bad2Sdanielk1977 
746c3f9bad2Sdanielk1977     /* determine whether we should code this trigger */
747c3f9bad2Sdanielk1977     if( pTrigger->op == op && pTrigger->tr_tm == tr_tm &&
748c3f9bad2Sdanielk1977         pTrigger->foreach == TK_ROW ){
749c3f9bad2Sdanielk1977       fire_this = 1;
750c3f9bad2Sdanielk1977       pTriggerStack = pParse->trigStack;
751c3f9bad2Sdanielk1977       while( pTriggerStack ){
752f29ce559Sdanielk1977         if( pTriggerStack->pTrigger == pTrigger ){
753f29ce559Sdanielk1977 	  fire_this = 0;
754f29ce559Sdanielk1977 	}
755c3f9bad2Sdanielk1977         pTriggerStack = pTriggerStack->pNext;
756c3f9bad2Sdanielk1977       }
757c3f9bad2Sdanielk1977       if( op == TK_UPDATE && pTrigger->pColumns &&
758f29ce559Sdanielk1977           !checkColumnOverLap(pTrigger->pColumns, pChanges) ){
759c3f9bad2Sdanielk1977         fire_this = 0;
760c3f9bad2Sdanielk1977       }
761f29ce559Sdanielk1977     }
762c3f9bad2Sdanielk1977 
763e4697f5eSdrh     if( fire_this && (pTriggerStack = sqliteMalloc(sizeof(TriggerStack)))!=0 ){
764c3f9bad2Sdanielk1977       int endTrigger;
765ad3cab52Sdrh       SrcList dummyTablist;
766c3f9bad2Sdanielk1977       Expr * whenExpr;
76785e2096fSdrh       AuthContext sContext;
768c3f9bad2Sdanielk1977 
769ad3cab52Sdrh       dummyTablist.nSrc = 0;
770c3f9bad2Sdanielk1977 
771c3f9bad2Sdanielk1977       /* Push an entry on to the trigger stack */
772c3f9bad2Sdanielk1977       pTriggerStack->pTrigger = pTrigger;
773633ed08dSdanielk1977       pTriggerStack->newIdx = newIdx;
774633ed08dSdanielk1977       pTriggerStack->oldIdx = oldIdx;
775633ed08dSdanielk1977       pTriggerStack->pTab = pTab;
776c3f9bad2Sdanielk1977       pTriggerStack->pNext = pParse->trigStack;
7776f34903eSdanielk1977       pTriggerStack->ignoreJump = ignoreJump;
778c3f9bad2Sdanielk1977       pParse->trigStack = pTriggerStack;
7794adee20fSdanielk1977       sqlite3AuthContextPush(pParse, &sContext, pTrigger->name);
780c3f9bad2Sdanielk1977 
781c3f9bad2Sdanielk1977       /* code the WHEN clause */
7824adee20fSdanielk1977       endTrigger = sqlite3VdbeMakeLabel(pParse->pVdbe);
7834adee20fSdanielk1977       whenExpr = sqlite3ExprDup(pTrigger->pWhen);
7844adee20fSdanielk1977       if( sqlite3ExprResolveIds(pParse, &dummyTablist, 0, whenExpr) ){
785c3f9bad2Sdanielk1977         pParse->trigStack = pParse->trigStack->pNext;
786c3f9bad2Sdanielk1977         sqliteFree(pTriggerStack);
7874adee20fSdanielk1977         sqlite3ExprDelete(whenExpr);
788c3f9bad2Sdanielk1977         return 1;
789c3f9bad2Sdanielk1977       }
7904adee20fSdanielk1977       sqlite3ExprIfFalse(pParse, whenExpr, endTrigger, 1);
7914adee20fSdanielk1977       sqlite3ExprDelete(whenExpr);
792c3f9bad2Sdanielk1977 
7934adee20fSdanielk1977       sqlite3VdbeAddOp(pParse->pVdbe, OP_ContextPush, 0, 0);
794633ed08dSdanielk1977       codeTriggerProgram(pParse, pTrigger->step_list, orconf);
7954adee20fSdanielk1977       sqlite3VdbeAddOp(pParse->pVdbe, OP_ContextPop, 0, 0);
796c3f9bad2Sdanielk1977 
797c3f9bad2Sdanielk1977       /* Pop the entry off the trigger stack */
798c3f9bad2Sdanielk1977       pParse->trigStack = pParse->trigStack->pNext;
7994adee20fSdanielk1977       sqlite3AuthContextPop(&sContext);
800c3f9bad2Sdanielk1977       sqliteFree(pTriggerStack);
801c3f9bad2Sdanielk1977 
8024adee20fSdanielk1977       sqlite3VdbeResolveLabel(pParse->pVdbe, endTrigger);
803c3f9bad2Sdanielk1977     }
804c3f9bad2Sdanielk1977     pTrigger = pTrigger->pNext;
805c3f9bad2Sdanielk1977   }
806c3f9bad2Sdanielk1977   return 0;
807c3f9bad2Sdanielk1977 }
808