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