xref: /sqlite-3.40.0/src/expr.c (revision 6f34903e)
1cce7d176Sdrh /*
2b19a2bc6Sdrh ** 2001 September 15
3cce7d176Sdrh **
4b19a2bc6Sdrh ** The author disclaims copyright to this source code.  In place of
5b19a2bc6Sdrh ** a legal notice, here is a blessing:
6cce7d176Sdrh **
7b19a2bc6Sdrh **    May you do good and not evil.
8b19a2bc6Sdrh **    May you find forgiveness for yourself and forgive others.
9b19a2bc6Sdrh **    May you share freely, never taking more than you give.
10cce7d176Sdrh **
11cce7d176Sdrh *************************************************************************
121ccde15dSdrh ** This file contains routines used for analyzing expressions and
13b19a2bc6Sdrh ** for generating VDBE code that evaluates expressions in SQLite.
14cce7d176Sdrh **
15*6f34903eSdanielk1977 ** $Id: expr.c,v 1.71 2002/06/11 02:25:41 danielk1977 Exp $
16cce7d176Sdrh */
17cce7d176Sdrh #include "sqliteInt.h"
1804738cb9Sdrh #include <ctype.h>
19a2e00042Sdrh 
20a2e00042Sdrh /*
21a76b5dfcSdrh ** Construct a new expression node and return a pointer to it.  Memory
22a76b5dfcSdrh ** for this node is obtained from sqliteMalloc().  The calling function
23a76b5dfcSdrh ** is responsible for making sure the node eventually gets freed.
24a76b5dfcSdrh */
25a76b5dfcSdrh Expr *sqliteExpr(int op, Expr *pLeft, Expr *pRight, Token *pToken){
26a76b5dfcSdrh   Expr *pNew;
27a76b5dfcSdrh   pNew = sqliteMalloc( sizeof(Expr) );
28a76b5dfcSdrh   if( pNew==0 ){
29a76b5dfcSdrh     sqliteExprDelete(pLeft);
30a76b5dfcSdrh     sqliteExprDelete(pRight);
31a76b5dfcSdrh     return 0;
32a76b5dfcSdrh   }
33a76b5dfcSdrh   pNew->op = op;
34a76b5dfcSdrh   pNew->pLeft = pLeft;
35a76b5dfcSdrh   pNew->pRight = pRight;
36a76b5dfcSdrh   if( pToken ){
37a76b5dfcSdrh     pNew->token = *pToken;
38a76b5dfcSdrh   }else{
39a76b5dfcSdrh     pNew->token.z = 0;
40a76b5dfcSdrh     pNew->token.n = 0;
41a76b5dfcSdrh   }
42a76b5dfcSdrh   if( pLeft && pRight ){
43a76b5dfcSdrh     sqliteExprSpan(pNew, &pLeft->span, &pRight->span);
44a76b5dfcSdrh   }else{
45a76b5dfcSdrh     pNew->span = pNew->token;
46a76b5dfcSdrh   }
47a76b5dfcSdrh   return pNew;
48a76b5dfcSdrh }
49a76b5dfcSdrh 
50a76b5dfcSdrh /*
51a76b5dfcSdrh ** Set the Expr.token field of the given expression to span all
52a76b5dfcSdrh ** text between the two given tokens.
53a76b5dfcSdrh */
54a76b5dfcSdrh void sqliteExprSpan(Expr *pExpr, Token *pLeft, Token *pRight){
55a76b5dfcSdrh   if( pExpr ){
56a76b5dfcSdrh     pExpr->span.z = pLeft->z;
57a76b5dfcSdrh     pExpr->span.n = pRight->n + Addr(pRight->z) - Addr(pLeft->z);
58a76b5dfcSdrh   }
59a76b5dfcSdrh }
60a76b5dfcSdrh 
61a76b5dfcSdrh /*
62a76b5dfcSdrh ** Construct a new expression node for a function with multiple
63a76b5dfcSdrh ** arguments.
64a76b5dfcSdrh */
65a76b5dfcSdrh Expr *sqliteExprFunction(ExprList *pList, Token *pToken){
66a76b5dfcSdrh   Expr *pNew;
67a76b5dfcSdrh   pNew = sqliteMalloc( sizeof(Expr) );
68a76b5dfcSdrh   if( pNew==0 ){
69a76b5dfcSdrh     sqliteExprListDelete(pList);
70a76b5dfcSdrh     return 0;
71a76b5dfcSdrh   }
72a76b5dfcSdrh   pNew->op = TK_FUNCTION;
73a76b5dfcSdrh   pNew->pList = pList;
74a76b5dfcSdrh   if( pToken ){
75a76b5dfcSdrh     pNew->token = *pToken;
76a76b5dfcSdrh   }else{
77a76b5dfcSdrh     pNew->token.z = 0;
78a76b5dfcSdrh     pNew->token.n = 0;
79a76b5dfcSdrh   }
80a76b5dfcSdrh   return pNew;
81a76b5dfcSdrh }
82a76b5dfcSdrh 
83a76b5dfcSdrh /*
84a2e00042Sdrh ** Recursively delete an expression tree.
85a2e00042Sdrh */
86a2e00042Sdrh void sqliteExprDelete(Expr *p){
87a2e00042Sdrh   if( p==0 ) return;
88a2e00042Sdrh   if( p->pLeft ) sqliteExprDelete(p->pLeft);
89a2e00042Sdrh   if( p->pRight ) sqliteExprDelete(p->pRight);
90a2e00042Sdrh   if( p->pList ) sqliteExprListDelete(p->pList);
91a2e00042Sdrh   if( p->pSelect ) sqliteSelectDelete(p->pSelect);
92a2e00042Sdrh   sqliteFree(p);
93a2e00042Sdrh }
94a2e00042Sdrh 
95cce7d176Sdrh /*
96a76b5dfcSdrh ** The following group of functions are used to translate the string
97a76b5dfcSdrh ** pointers of tokens in expression from one buffer to another.
98a76b5dfcSdrh **
99a76b5dfcSdrh ** Normally, the Expr.token.z and Expr.span.z fields point into the
100a76b5dfcSdrh ** original input buffer of an SQL statement.  This is usually OK
101a76b5dfcSdrh ** since the SQL statement is executed and the expression is deleted
102a76b5dfcSdrh ** before the input buffer is freed.  Making the tokens point to the
103a76b5dfcSdrh ** original input buffer saves many calls to malloc() and thus helps
104a76b5dfcSdrh ** the library to run faster.
105a76b5dfcSdrh **
106a76b5dfcSdrh ** But sometimes we need an expression to persist past the time when
107a76b5dfcSdrh ** the input buffer is freed.  (Example: The SELECT clause of a
108a76b5dfcSdrh ** CREATE VIEW statement contains expressions that must persist for
109a76b5dfcSdrh ** the life of the view.)  When that happens we have to make a
110a76b5dfcSdrh ** persistent copy of the input buffer and translate the Expr.token.z
111a76b5dfcSdrh ** and Expr.span.z fields to point to the copy rather than the
112a2ed5601Sdrh ** original input buffer.  The following group of routines handle that
113a76b5dfcSdrh ** translation.
114a76b5dfcSdrh **
115a76b5dfcSdrh ** The "offset" parameter is the distance from the original input buffer
116a76b5dfcSdrh ** to the persistent copy.  These routines recursively walk the entire
117a76b5dfcSdrh ** expression tree and shift all tokens by "offset" amount.
118a76b5dfcSdrh **
119a76b5dfcSdrh ** The work of figuring out the appropriate "offset" and making the
120a76b5dfcSdrh ** presistent copy of the input buffer is done by the calling routine.
121a76b5dfcSdrh */
122a76b5dfcSdrh void sqliteExprMoveStrings(Expr *p, int offset){
123a76b5dfcSdrh   if( p==0 ) return;
124a76b5dfcSdrh   if( p->token.z ) p->token.z += offset;
125a76b5dfcSdrh   if( p->span.z ) p->span.z += offset;
126a76b5dfcSdrh   if( p->pLeft ) sqliteExprMoveStrings(p->pLeft, offset);
127a76b5dfcSdrh   if( p->pRight ) sqliteExprMoveStrings(p->pRight, offset);
128a76b5dfcSdrh   if( p->pList ) sqliteExprListMoveStrings(p->pList, offset);
129a76b5dfcSdrh   if( p->pSelect ) sqliteSelectMoveStrings(p->pSelect, offset);
130a76b5dfcSdrh }
131a76b5dfcSdrh void sqliteExprListMoveStrings(ExprList *pList, int offset){
132a76b5dfcSdrh   int i;
133a76b5dfcSdrh   if( pList==0 ) return;
134a76b5dfcSdrh   for(i=0; i<pList->nExpr; i++){
135a76b5dfcSdrh     sqliteExprMoveStrings(pList->a[i].pExpr, offset);
136a76b5dfcSdrh   }
137a76b5dfcSdrh }
138a76b5dfcSdrh void sqliteSelectMoveStrings(Select *pSelect, int offset){
139a76b5dfcSdrh   if( pSelect==0 ) return;
140a76b5dfcSdrh   sqliteExprListMoveStrings(pSelect->pEList, offset);
141a76b5dfcSdrh   sqliteExprMoveStrings(pSelect->pWhere, offset);
142a76b5dfcSdrh   sqliteExprListMoveStrings(pSelect->pGroupBy, offset);
143a76b5dfcSdrh   sqliteExprMoveStrings(pSelect->pHaving, offset);
144a76b5dfcSdrh   sqliteExprListMoveStrings(pSelect->pOrderBy, offset);
145a76b5dfcSdrh   sqliteSelectMoveStrings(pSelect->pPrior, offset);
146a76b5dfcSdrh }
147a76b5dfcSdrh 
148a76b5dfcSdrh /*
149ff78bd2fSdrh ** The following group of routines make deep copies of expressions,
150ff78bd2fSdrh ** expression lists, ID lists, and select statements.  The copies can
151ff78bd2fSdrh ** be deleted (by being passed to their respective ...Delete() routines)
152ff78bd2fSdrh ** without effecting the originals.
153ff78bd2fSdrh **
154ff78bd2fSdrh ** Note, however, that the Expr.token.z and Expr.span.z fields point to
155ff78bd2fSdrh ** string space that is allocated separately from the expression tree
156ff78bd2fSdrh ** itself.  These routines do NOT duplicate that string space.
157ff78bd2fSdrh **
158ad3cab52Sdrh ** The expression list, ID, and source lists return by sqliteExprListDup(),
159ad3cab52Sdrh ** sqliteIdListDup(), and sqliteSrcListDup() can not be further expanded
160ad3cab52Sdrh ** by subsequent calls to sqlite*ListAppend() routines.
161ff78bd2fSdrh **
162ad3cab52Sdrh ** Any tables that the SrcList might point to are not duplicated.
163ff78bd2fSdrh */
164ff78bd2fSdrh Expr *sqliteExprDup(Expr *p){
165ff78bd2fSdrh   Expr *pNew;
166ff78bd2fSdrh   if( p==0 ) return 0;
167ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*p) );
168ff78bd2fSdrh   if( pNew==0 ) return 0;
169ff78bd2fSdrh   pNew->op = p->op;
170ff78bd2fSdrh   pNew->pLeft = sqliteExprDup(p->pLeft);
171ff78bd2fSdrh   pNew->pRight = sqliteExprDup(p->pRight);
172ff78bd2fSdrh   pNew->pList = sqliteExprListDup(p->pList);
173832508b7Sdrh   pNew->iTable = p->iTable;
174832508b7Sdrh   pNew->iColumn = p->iColumn;
175832508b7Sdrh   pNew->iAgg = p->iAgg;
176ff78bd2fSdrh   pNew->token = p->token;
177ff78bd2fSdrh   pNew->span = p->span;
178ff78bd2fSdrh   pNew->pSelect = sqliteSelectDup(p->pSelect);
179ff78bd2fSdrh   return pNew;
180ff78bd2fSdrh }
181ff78bd2fSdrh ExprList *sqliteExprListDup(ExprList *p){
182ff78bd2fSdrh   ExprList *pNew;
183ff78bd2fSdrh   int i;
184ff78bd2fSdrh   if( p==0 ) return 0;
185ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*pNew) );
186ff78bd2fSdrh   if( pNew==0 ) return 0;
187ff78bd2fSdrh   pNew->nExpr = p->nExpr;
188ff78bd2fSdrh   pNew->a = sqliteMalloc( p->nExpr*sizeof(p->a[0]) );
189e4697f5eSdrh   if( pNew->a==0 ) return 0;
190ff78bd2fSdrh   for(i=0; i<p->nExpr; i++){
191ff78bd2fSdrh     pNew->a[i].pExpr = sqliteExprDup(p->a[i].pExpr);
192ff78bd2fSdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
193ff78bd2fSdrh     pNew->a[i].sortOrder = p->a[i].sortOrder;
194ff78bd2fSdrh     pNew->a[i].isAgg = p->a[i].isAgg;
195ff78bd2fSdrh     pNew->a[i].done = 0;
196ff78bd2fSdrh   }
197ff78bd2fSdrh   return pNew;
198ff78bd2fSdrh }
199ad3cab52Sdrh SrcList *sqliteSrcListDup(SrcList *p){
200ad3cab52Sdrh   SrcList *pNew;
201ad3cab52Sdrh   int i;
202ad3cab52Sdrh   if( p==0 ) return 0;
203ad3cab52Sdrh   pNew = sqliteMalloc( sizeof(*pNew) );
204ad3cab52Sdrh   if( pNew==0 ) return 0;
205ad3cab52Sdrh   pNew->nSrc = p->nSrc;
206ad3cab52Sdrh   pNew->a = sqliteMalloc( p->nSrc*sizeof(p->a[0]) );
207*6f34903eSdanielk1977   if( pNew->a==0 && p->nSrc != 0 ) return 0;
208ad3cab52Sdrh   for(i=0; i<p->nSrc; i++){
209ad3cab52Sdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
210ad3cab52Sdrh     pNew->a[i].zAlias = sqliteStrDup(p->a[i].zAlias);
211ad3cab52Sdrh     pNew->a[i].jointype = p->a[i].jointype;
212ad3cab52Sdrh     pNew->a[i].pTab = 0;
213ad3cab52Sdrh     pNew->a[i].pSelect = sqliteSelectDup(p->a[i].pSelect);
214ad3cab52Sdrh     pNew->a[i].pOn = sqliteExprDup(p->a[i].pOn);
215ad3cab52Sdrh     pNew->a[i].pUsing = sqliteIdListDup(p->a[i].pUsing);
216ad3cab52Sdrh   }
217ad3cab52Sdrh   return pNew;
218ad3cab52Sdrh }
219ff78bd2fSdrh IdList *sqliteIdListDup(IdList *p){
220ff78bd2fSdrh   IdList *pNew;
221ff78bd2fSdrh   int i;
222ff78bd2fSdrh   if( p==0 ) return 0;
223ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*pNew) );
224ff78bd2fSdrh   if( pNew==0 ) return 0;
225ff78bd2fSdrh   pNew->nId = p->nId;
226ff78bd2fSdrh   pNew->a = sqliteMalloc( p->nId*sizeof(p->a[0]) );
227e4697f5eSdrh   if( pNew->a==0 ) return 0;
228ff78bd2fSdrh   for(i=0; i<p->nId; i++){
229ff78bd2fSdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
230ff78bd2fSdrh     pNew->a[i].idx = p->a[i].idx;
231ff78bd2fSdrh   }
232ff78bd2fSdrh   return pNew;
233ff78bd2fSdrh }
234ff78bd2fSdrh Select *sqliteSelectDup(Select *p){
235ff78bd2fSdrh   Select *pNew;
236ff78bd2fSdrh   if( p==0 ) return 0;
237ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*p) );
238ff78bd2fSdrh   if( pNew==0 ) return 0;
239ff78bd2fSdrh   pNew->isDistinct = p->isDistinct;
240ff78bd2fSdrh   pNew->pEList = sqliteExprListDup(p->pEList);
241ad3cab52Sdrh   pNew->pSrc = sqliteSrcListDup(p->pSrc);
242ff78bd2fSdrh   pNew->pWhere = sqliteExprDup(p->pWhere);
243ff78bd2fSdrh   pNew->pGroupBy = sqliteExprListDup(p->pGroupBy);
244ff78bd2fSdrh   pNew->pHaving = sqliteExprDup(p->pHaving);
245ff78bd2fSdrh   pNew->pOrderBy = sqliteExprListDup(p->pOrderBy);
246ff78bd2fSdrh   pNew->op = p->op;
247ff78bd2fSdrh   pNew->pPrior = sqliteSelectDup(p->pPrior);
248ff78bd2fSdrh   pNew->nLimit = p->nLimit;
249ff78bd2fSdrh   pNew->nOffset = p->nOffset;
250ff78bd2fSdrh   pNew->zSelect = 0;
251ff78bd2fSdrh   return pNew;
252ff78bd2fSdrh }
253ff78bd2fSdrh 
254ff78bd2fSdrh 
255ff78bd2fSdrh /*
256a76b5dfcSdrh ** Add a new element to the end of an expression list.  If pList is
257a76b5dfcSdrh ** initially NULL, then create a new expression list.
258a76b5dfcSdrh */
259a76b5dfcSdrh ExprList *sqliteExprListAppend(ExprList *pList, Expr *pExpr, Token *pName){
260a76b5dfcSdrh   int i;
261a76b5dfcSdrh   if( pList==0 ){
262a76b5dfcSdrh     pList = sqliteMalloc( sizeof(ExprList) );
263a76b5dfcSdrh     if( pList==0 ){
264a76b5dfcSdrh       sqliteExprDelete(pExpr);
265a76b5dfcSdrh       return 0;
266a76b5dfcSdrh     }
267a76b5dfcSdrh   }
268a76b5dfcSdrh   if( (pList->nExpr & 7)==0 ){
269a76b5dfcSdrh     int n = pList->nExpr + 8;
270a76b5dfcSdrh     struct ExprList_item *a;
271a76b5dfcSdrh     a = sqliteRealloc(pList->a, n*sizeof(pList->a[0]));
272a76b5dfcSdrh     if( a==0 ){
273a76b5dfcSdrh       sqliteExprDelete(pExpr);
274a76b5dfcSdrh       return pList;
275a76b5dfcSdrh     }
276a76b5dfcSdrh     pList->a = a;
277a76b5dfcSdrh   }
278a76b5dfcSdrh   if( pExpr || pName ){
279a76b5dfcSdrh     i = pList->nExpr++;
280a76b5dfcSdrh     pList->a[i].pExpr = pExpr;
281a76b5dfcSdrh     pList->a[i].zName = 0;
282a76b5dfcSdrh     if( pName ){
283a76b5dfcSdrh       sqliteSetNString(&pList->a[i].zName, pName->z, pName->n, 0);
284a76b5dfcSdrh       sqliteDequote(pList->a[i].zName);
285a76b5dfcSdrh     }
286a76b5dfcSdrh   }
287a76b5dfcSdrh   return pList;
288a76b5dfcSdrh }
289a76b5dfcSdrh 
290a76b5dfcSdrh /*
291a76b5dfcSdrh ** Delete an entire expression list.
292a76b5dfcSdrh */
293a76b5dfcSdrh void sqliteExprListDelete(ExprList *pList){
294a76b5dfcSdrh   int i;
295a76b5dfcSdrh   if( pList==0 ) return;
296a76b5dfcSdrh   for(i=0; i<pList->nExpr; i++){
297a76b5dfcSdrh     sqliteExprDelete(pList->a[i].pExpr);
298a76b5dfcSdrh     sqliteFree(pList->a[i].zName);
299a76b5dfcSdrh   }
300a76b5dfcSdrh   sqliteFree(pList->a);
301a76b5dfcSdrh   sqliteFree(pList);
302a76b5dfcSdrh }
303a76b5dfcSdrh 
304a76b5dfcSdrh /*
305fef5208cSdrh ** Walk an expression tree.  Return 1 if the expression is constant
306fef5208cSdrh ** and 0 if it involves variables.
3072398937bSdrh **
3082398937bSdrh ** For the purposes of this function, a double-quoted string (ex: "abc")
3092398937bSdrh ** is considered a variable but a single-quoted string (ex: 'abc') is
3102398937bSdrh ** a constant.
311fef5208cSdrh */
3129208643dSdrh int sqliteExprIsConstant(Expr *p){
313fef5208cSdrh   switch( p->op ){
314fef5208cSdrh     case TK_ID:
315967e8b73Sdrh     case TK_COLUMN:
316fef5208cSdrh     case TK_DOT:
317fef5208cSdrh       return 0;
3182398937bSdrh     case TK_STRING:
3192398937bSdrh       return p->token.z[0]=='\'';
3209208643dSdrh     case TK_INTEGER:
3219208643dSdrh     case TK_FLOAT:
3229208643dSdrh       return 1;
323fef5208cSdrh     default: {
3249208643dSdrh       if( p->pLeft && !sqliteExprIsConstant(p->pLeft) ) return 0;
3259208643dSdrh       if( p->pRight && !sqliteExprIsConstant(p->pRight) ) return 0;
326fef5208cSdrh       if( p->pList ){
327fef5208cSdrh         int i;
328fef5208cSdrh         for(i=0; i<p->pList->nExpr; i++){
3299208643dSdrh           if( !sqliteExprIsConstant(p->pList->a[i].pExpr) ) return 0;
330fef5208cSdrh         }
331fef5208cSdrh       }
3329208643dSdrh       return p->pLeft!=0 || p->pRight!=0 || (p->pList && p->pList->nExpr>0);
333fef5208cSdrh     }
334fef5208cSdrh   }
3359208643dSdrh   return 0;
336fef5208cSdrh }
337fef5208cSdrh 
338fef5208cSdrh /*
339e4de1febSdrh ** If the given expression codes a constant integer, return 1 and put
340e4de1febSdrh ** the value of the integer in *pValue.  If the expression is not an
341e4de1febSdrh ** integer, return 0 and leave *pValue unchanged.
342e4de1febSdrh */
343e4de1febSdrh int sqliteExprIsInteger(Expr *p, int *pValue){
344e4de1febSdrh   switch( p->op ){
345e4de1febSdrh     case TK_INTEGER: {
346e4de1febSdrh       *pValue = atoi(p->token.z);
347e4de1febSdrh       return 1;
348e4de1febSdrh     }
349e4de1febSdrh     case TK_STRING: {
350bd790ee3Sdrh       const char *z = p->token.z;
351e4de1febSdrh       int n = p->token.n;
352bd790ee3Sdrh       if( n>0 && z[0]=='-' ){ z++; n--; }
353e4de1febSdrh       while( n>0 && *z && isdigit(*z) ){ z++; n--; }
354e4de1febSdrh       if( n==0 ){
355e4de1febSdrh         *pValue = atoi(p->token.z);
356e4de1febSdrh         return 1;
357e4de1febSdrh       }
358e4de1febSdrh       break;
359e4de1febSdrh     }
360e4de1febSdrh     case TK_UMINUS: {
361e4de1febSdrh       int v;
362e4de1febSdrh       if( sqliteExprIsInteger(p->pLeft, &v) ){
363e4de1febSdrh         *pValue = -v;
364e4de1febSdrh         return 1;
365e4de1febSdrh       }
366e4de1febSdrh       break;
367e4de1febSdrh     }
368e4de1febSdrh     default: break;
369e4de1febSdrh   }
370e4de1febSdrh   return 0;
371e4de1febSdrh }
372e4de1febSdrh 
373e4de1febSdrh /*
374c4a3c779Sdrh ** Return TRUE if the given string is a row-id column name.
375c4a3c779Sdrh */
376c4a3c779Sdrh static int sqliteIsRowid(const char *z){
377c4a3c779Sdrh   if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
378c4a3c779Sdrh   if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
379c4a3c779Sdrh   if( sqliteStrICmp(z, "OID")==0 ) return 1;
380c4a3c779Sdrh   return 0;
381c4a3c779Sdrh }
382c4a3c779Sdrh 
383c4a3c779Sdrh /*
384cce7d176Sdrh ** This routine walks an expression tree and resolves references to
385967e8b73Sdrh ** table columns.  Nodes of the form ID.ID or ID resolve into an
386aacc543eSdrh ** index to the table in the table list and a column offset.  The
387aacc543eSdrh ** Expr.opcode for such nodes is changed to TK_COLUMN.  The Expr.iTable
388aacc543eSdrh ** value is changed to the index of the referenced table in pTabList
389832508b7Sdrh ** plus the "base" value.  The base value will ultimately become the
390aacc543eSdrh ** VDBE cursor number for a cursor that is pointing into the referenced
391aacc543eSdrh ** table.  The Expr.iColumn value is changed to the index of the column
392aacc543eSdrh ** of the referenced table.  The Expr.iColumn value for the special
393aacc543eSdrh ** ROWID column is -1.  Any INTEGER PRIMARY KEY column is tried as an
394aacc543eSdrh ** alias for ROWID.
39519a775c2Sdrh **
396fef5208cSdrh ** We also check for instances of the IN operator.  IN comes in two
397fef5208cSdrh ** forms:
398fef5208cSdrh **
399fef5208cSdrh **           expr IN (exprlist)
400fef5208cSdrh ** and
401fef5208cSdrh **           expr IN (SELECT ...)
402fef5208cSdrh **
403fef5208cSdrh ** The first form is handled by creating a set holding the list
404fef5208cSdrh ** of allowed values.  The second form causes the SELECT to generate
405fef5208cSdrh ** a temporary table.
406fef5208cSdrh **
407fef5208cSdrh ** This routine also looks for scalar SELECTs that are part of an expression.
40819a775c2Sdrh ** If it finds any, it generates code to write the value of that select
40919a775c2Sdrh ** into a memory cell.
410cce7d176Sdrh **
411967e8b73Sdrh ** Unknown columns or tables provoke an error.  The function returns
412cce7d176Sdrh ** the number of errors seen and leaves an error message on pParse->zErrMsg.
413cce7d176Sdrh */
414a2e00042Sdrh int sqliteExprResolveIds(
415a2e00042Sdrh   Parse *pParse,     /* The parser context */
416832508b7Sdrh   int base,          /* VDBE cursor number for first entry in pTabList */
417ad3cab52Sdrh   SrcList *pTabList, /* List of tables used to resolve column names */
418a2e00042Sdrh   ExprList *pEList,  /* List of expressions used to resolve "AS" */
419a2e00042Sdrh   Expr *pExpr        /* The expression to be analyzed. */
420a2e00042Sdrh ){
421daffd0e5Sdrh   if( pExpr==0 || pTabList==0 ) return 0;
422ad3cab52Sdrh   assert( base+pTabList->nSrc<=pParse->nTab );
423cce7d176Sdrh   switch( pExpr->op ){
4242398937bSdrh     /* Double-quoted strings (ex: "abc") are used as identifiers if
4252398937bSdrh     ** possible.  Otherwise they remain as strings.  Single-quoted
4262398937bSdrh     ** strings (ex: 'abc') are always string literals.
4272398937bSdrh     */
4282398937bSdrh     case TK_STRING: {
4292398937bSdrh       if( pExpr->token.z[0]=='\'' ) break;
4302398937bSdrh       /* Fall thru into the TK_ID case if this is a double-quoted string */
4312398937bSdrh     }
432a2e00042Sdrh     /* A lone identifier.  Try and match it as follows:
433a2e00042Sdrh     **
434a2e00042Sdrh     **     1.  To the name of a column of one of the tables in pTabList
435a2e00042Sdrh     **
436a2e00042Sdrh     **     2.  To the right side of an AS keyword in the column list of
437a2e00042Sdrh     **         a SELECT statement.  (For example, match against 'x' in
438a2e00042Sdrh     **         "SELECT a+b AS 'x' FROM t1".)
439a2e00042Sdrh     **
440a2e00042Sdrh     **     3.  One of the special names "ROWID", "OID", or "_ROWID_".
441a2e00042Sdrh     */
442cce7d176Sdrh     case TK_ID: {
443cce7d176Sdrh       int cnt = 0;      /* Number of matches */
444cce7d176Sdrh       int i;            /* Loop counter */
445a76b5dfcSdrh       char *z;
446a76b5dfcSdrh       assert( pExpr->token.z );
447a76b5dfcSdrh       z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
4482f4392ffSdrh       sqliteDequote(z);
449daffd0e5Sdrh       if( z==0 ) return 1;
450ad3cab52Sdrh       for(i=0; i<pTabList->nSrc; i++){
451cce7d176Sdrh         int j;
452cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
453cce7d176Sdrh         if( pTab==0 ) continue;
454417be79cSdrh         assert( pTab->nCol>0 );
455cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
4567020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
457cce7d176Sdrh             cnt++;
458832508b7Sdrh             pExpr->iTable = i + base;
4594a32431cSdrh             if( j==pTab->iPKey ){
4604a32431cSdrh               /* Substitute the record number for the INTEGER PRIMARY KEY */
4614a32431cSdrh               pExpr->iColumn = -1;
4624a32431cSdrh             }else{
463967e8b73Sdrh               pExpr->iColumn = j;
464cce7d176Sdrh             }
465a2e00042Sdrh             pExpr->op = TK_COLUMN;
466a2e00042Sdrh           }
467a2e00042Sdrh         }
468a2e00042Sdrh       }
469a2e00042Sdrh       if( cnt==0 && pEList!=0 ){
470a2e00042Sdrh         int j;
471a2e00042Sdrh         for(j=0; j<pEList->nExpr; j++){
472a2e00042Sdrh           char *zAs = pEList->a[j].zName;
473a2e00042Sdrh           if( zAs!=0 && sqliteStrICmp(zAs, z)==0 ){
474a2e00042Sdrh             cnt++;
475a2e00042Sdrh             assert( pExpr->pLeft==0 && pExpr->pRight==0 );
476a2e00042Sdrh             pExpr->op = TK_AS;
477a2e00042Sdrh             pExpr->iColumn = j;
47875148a27Sdrh             pExpr->pLeft = sqliteExprDup(pEList->a[j].pExpr);
479cce7d176Sdrh           }
480cce7d176Sdrh         }
4814a32431cSdrh       }
482c4a3c779Sdrh       if( cnt==0 && sqliteIsRowid(z) ){
483c4a3c779Sdrh         pExpr->iColumn = -1;
484832508b7Sdrh         pExpr->iTable = base;
485ad3cab52Sdrh         cnt = 1 + (pTabList->nSrc>1);
486a2e00042Sdrh         pExpr->op = TK_COLUMN;
487c4a3c779Sdrh       }
488cce7d176Sdrh       sqliteFree(z);
4892398937bSdrh       if( cnt==0 && pExpr->token.z[0]!='"' ){
490967e8b73Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
491cce7d176Sdrh           pExpr->token.z, pExpr->token.n, 0);
492cce7d176Sdrh         pParse->nErr++;
493cce7d176Sdrh         return 1;
494cce7d176Sdrh       }else if( cnt>1 ){
495967e8b73Sdrh         sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
496cce7d176Sdrh           pExpr->token.z, pExpr->token.n, 0);
497cce7d176Sdrh         pParse->nErr++;
498cce7d176Sdrh         return 1;
499cce7d176Sdrh       }
500cce7d176Sdrh       break;
501cce7d176Sdrh     }
502cce7d176Sdrh 
503967e8b73Sdrh     /* A table name and column name:  ID.ID */
504cce7d176Sdrh     case TK_DOT: {
505cce7d176Sdrh       int cnt = 0;             /* Number of matches */
506c4a3c779Sdrh       int cntTab = 0;          /* Number of matching tables */
507cce7d176Sdrh       int i;                   /* Loop counter */
508cce7d176Sdrh       Expr *pLeft, *pRight;    /* Left and right subbranches of the expr */
509cce7d176Sdrh       char *zLeft, *zRight;    /* Text of an identifier */
510cce7d176Sdrh 
511cce7d176Sdrh       pLeft = pExpr->pLeft;
512cce7d176Sdrh       pRight = pExpr->pRight;
513a76b5dfcSdrh       assert( pLeft && pLeft->op==TK_ID && pLeft->token.z );
514a76b5dfcSdrh       assert( pRight && pRight->op==TK_ID && pRight->token.z );
5156e142f54Sdrh       zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
5166e142f54Sdrh       zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
517daffd0e5Sdrh       if( zLeft==0 || zRight==0 ){
518daffd0e5Sdrh         sqliteFree(zLeft);
519daffd0e5Sdrh         sqliteFree(zRight);
520daffd0e5Sdrh         return 1;
521daffd0e5Sdrh       }
52287c40e88Sdrh       sqliteDequote(zLeft);
52387c40e88Sdrh       sqliteDequote(zRight);
524c4a3c779Sdrh       pExpr->iTable = -1;
525ad3cab52Sdrh       for(i=0; i<pTabList->nSrc; i++){
526cce7d176Sdrh         int j;
527cce7d176Sdrh         char *zTab;
528cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
529cce7d176Sdrh         if( pTab==0 ) continue;
530417be79cSdrh         assert( pTab->nCol>0 );
531cce7d176Sdrh         if( pTabList->a[i].zAlias ){
532cce7d176Sdrh           zTab = pTabList->a[i].zAlias;
533cce7d176Sdrh         }else{
534cce7d176Sdrh           zTab = pTab->zName;
535cce7d176Sdrh         }
536094b2bbfSdrh         if( zTab==0 || sqliteStrICmp(zTab, zLeft)!=0 ) continue;
537832508b7Sdrh         if( 0==(cntTab++) ) pExpr->iTable = i + base;
538cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
5397020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
540cce7d176Sdrh             cnt++;
541832508b7Sdrh             pExpr->iTable = i + base;
5424a32431cSdrh             if( j==pTab->iPKey ){
5434a32431cSdrh               /* Substitute the record number for the INTEGER PRIMARY KEY */
5444a32431cSdrh               pExpr->iColumn = -1;
5454a32431cSdrh             }else{
546967e8b73Sdrh               pExpr->iColumn = j;
547cce7d176Sdrh             }
548cce7d176Sdrh           }
549cce7d176Sdrh         }
5504a32431cSdrh       }
551c3f9bad2Sdanielk1977 
552c3f9bad2Sdanielk1977       /* If we have not already resolved this *.* expression, then maybe
553c3f9bad2Sdanielk1977        * it is a new.* or old.* trigger argument reference */
554c3f9bad2Sdanielk1977       if( cnt == 0 && pParse->trigStack != 0 ){
555f29ce559Sdanielk1977         TriggerStack *pTriggerStack = pParse->trigStack;
556c3f9bad2Sdanielk1977         int t = 0;
557f29ce559Sdanielk1977         if( pTriggerStack->newIdx != -1 && sqliteStrICmp("new", zLeft) == 0 ){
558f29ce559Sdanielk1977           pExpr->iTable = pTriggerStack->newIdx;
559c3f9bad2Sdanielk1977           cntTab++;
560c3f9bad2Sdanielk1977           t = 1;
561c3f9bad2Sdanielk1977         }
562f29ce559Sdanielk1977         if( pTriggerStack->oldIdx != -1 && sqliteStrICmp("old", zLeft) == 0 ){
563f29ce559Sdanielk1977           pExpr->iTable = pTriggerStack->oldIdx;
564c3f9bad2Sdanielk1977           cntTab++;
565c3f9bad2Sdanielk1977           t = 1;
566c3f9bad2Sdanielk1977         }
567c3f9bad2Sdanielk1977 
568f29ce559Sdanielk1977         if( t ){
569f29ce559Sdanielk1977 	  int j;
570f29ce559Sdanielk1977           for(j=0; j < pTriggerStack->pTab->nCol; j++) {
571f29ce559Sdanielk1977             if( sqliteStrICmp(pTriggerStack->pTab->aCol[j].zName, zRight)==0 ){
572c3f9bad2Sdanielk1977               cnt++;
573c3f9bad2Sdanielk1977               pExpr->iColumn = j;
574c3f9bad2Sdanielk1977             }
575c3f9bad2Sdanielk1977           }
576c3f9bad2Sdanielk1977 	}
577f29ce559Sdanielk1977       }
578c3f9bad2Sdanielk1977 
579c4a3c779Sdrh       if( cnt==0 && cntTab==1 && sqliteIsRowid(zRight) ){
580c4a3c779Sdrh         cnt = 1;
581c4a3c779Sdrh         pExpr->iColumn = -1;
582c4a3c779Sdrh       }
583cce7d176Sdrh       sqliteFree(zLeft);
584cce7d176Sdrh       sqliteFree(zRight);
585cce7d176Sdrh       if( cnt==0 ){
586967e8b73Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
587cce7d176Sdrh           pLeft->token.z, pLeft->token.n, ".", 1,
588cce7d176Sdrh           pRight->token.z, pRight->token.n, 0);
589cce7d176Sdrh         pParse->nErr++;
590cce7d176Sdrh         return 1;
591cce7d176Sdrh       }else if( cnt>1 ){
592967e8b73Sdrh         sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
593cce7d176Sdrh           pLeft->token.z, pLeft->token.n, ".", 1,
594cce7d176Sdrh           pRight->token.z, pRight->token.n, 0);
595cce7d176Sdrh         pParse->nErr++;
596cce7d176Sdrh         return 1;
597cce7d176Sdrh       }
598cce7d176Sdrh       sqliteExprDelete(pLeft);
599cce7d176Sdrh       pExpr->pLeft = 0;
600cce7d176Sdrh       sqliteExprDelete(pRight);
601cce7d176Sdrh       pExpr->pRight = 0;
602967e8b73Sdrh       pExpr->op = TK_COLUMN;
603cce7d176Sdrh       break;
604cce7d176Sdrh     }
605cce7d176Sdrh 
606fef5208cSdrh     case TK_IN: {
607d8bc7086Sdrh       Vdbe *v = sqliteGetVdbe(pParse);
608fef5208cSdrh       if( v==0 ) return 1;
609832508b7Sdrh       if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pExpr->pLeft) ){
610cfab11bcSdrh         return 1;
611cfab11bcSdrh       }
612fef5208cSdrh       if( pExpr->pSelect ){
613fef5208cSdrh         /* Case 1:     expr IN (SELECT ...)
614fef5208cSdrh         **
615fef5208cSdrh         ** Generate code to write the results of the select into a temporary
6164794b980Sdrh         ** table.  The cursor number of the temporary table has already
6174794b980Sdrh         ** been put in iTable by sqliteExprResolveInSelect().
618fef5208cSdrh         */
619832508b7Sdrh         pExpr->iTable = pParse->nTab++;
620c6b52df3Sdrh         sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 1);
621832508b7Sdrh         sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable, 0,0,0);
622fef5208cSdrh       }else if( pExpr->pList ){
623fef5208cSdrh         /* Case 2:     expr IN (exprlist)
624fef5208cSdrh         **
625fef5208cSdrh         ** Create a set to put the exprlist values in.  The Set id is stored
626fef5208cSdrh         ** in iTable.
627fef5208cSdrh         */
628fef5208cSdrh         int i, iSet;
629fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
630fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
6319208643dSdrh           if( !sqliteExprIsConstant(pE2) ){
632fef5208cSdrh             sqliteSetString(&pParse->zErrMsg,
633fef5208cSdrh               "right-hand side of IN operator must be constant", 0);
634fef5208cSdrh             pParse->nErr++;
635fef5208cSdrh             return 1;
636fef5208cSdrh           }
6374794b980Sdrh           if( sqliteExprCheck(pParse, pE2, 0, 0) ){
6384794b980Sdrh             return 1;
6394794b980Sdrh           }
640fef5208cSdrh         }
641fef5208cSdrh         iSet = pExpr->iTable = pParse->nSet++;
642fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
643fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
644fef5208cSdrh           switch( pE2->op ){
645fef5208cSdrh             case TK_FLOAT:
646fef5208cSdrh             case TK_INTEGER:
647fef5208cSdrh             case TK_STRING: {
64899fcd718Sdrh               int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
649a76b5dfcSdrh               assert( pE2->token.z );
650fef5208cSdrh               sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
651fef5208cSdrh               sqliteVdbeDequoteP3(v, addr);
652fef5208cSdrh               break;
653fef5208cSdrh             }
654fef5208cSdrh             default: {
655fef5208cSdrh               sqliteExprCode(pParse, pE2);
65699fcd718Sdrh               sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
657fef5208cSdrh               break;
658fef5208cSdrh             }
659fef5208cSdrh           }
660fef5208cSdrh         }
661fef5208cSdrh       }
662cfab11bcSdrh       break;
663fef5208cSdrh     }
664fef5208cSdrh 
66519a775c2Sdrh     case TK_SELECT: {
666fef5208cSdrh       /* This has to be a scalar SELECT.  Generate code to put the
667fef5208cSdrh       ** value of this select in a memory cell and record the number
668967e8b73Sdrh       ** of the memory cell in iColumn.
669fef5208cSdrh       */
670967e8b73Sdrh       pExpr->iColumn = pParse->nMem++;
671832508b7Sdrh       if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn,0,0,0) ){
67219a775c2Sdrh         return 1;
67319a775c2Sdrh       }
67419a775c2Sdrh       break;
67519a775c2Sdrh     }
67619a775c2Sdrh 
677cce7d176Sdrh     /* For all else, just recursively walk the tree */
678cce7d176Sdrh     default: {
679cce7d176Sdrh       if( pExpr->pLeft
680832508b7Sdrh       && sqliteExprResolveIds(pParse, base, pTabList, pEList, pExpr->pLeft) ){
681cce7d176Sdrh         return 1;
682cce7d176Sdrh       }
683cce7d176Sdrh       if( pExpr->pRight
684832508b7Sdrh       && sqliteExprResolveIds(pParse, base, pTabList, pEList, pExpr->pRight) ){
685cce7d176Sdrh         return 1;
686cce7d176Sdrh       }
687cce7d176Sdrh       if( pExpr->pList ){
688cce7d176Sdrh         int i;
689cce7d176Sdrh         ExprList *pList = pExpr->pList;
690cce7d176Sdrh         for(i=0; i<pList->nExpr; i++){
691832508b7Sdrh           Expr *pArg = pList->a[i].pExpr;
692832508b7Sdrh           if( sqliteExprResolveIds(pParse, base, pTabList, pEList, pArg) ){
693cce7d176Sdrh             return 1;
694cce7d176Sdrh           }
695cce7d176Sdrh         }
696cce7d176Sdrh       }
697cce7d176Sdrh     }
698cce7d176Sdrh   }
699cce7d176Sdrh   return 0;
700cce7d176Sdrh }
701cce7d176Sdrh 
702cce7d176Sdrh /*
703cce7d176Sdrh ** Error check the functions in an expression.  Make sure all
704cce7d176Sdrh ** function names are recognized and all functions have the correct
705cce7d176Sdrh ** number of arguments.  Leave an error message in pParse->zErrMsg
706cce7d176Sdrh ** if anything is amiss.  Return the number of errors.
707cce7d176Sdrh **
708cce7d176Sdrh ** if pIsAgg is not null and this expression is an aggregate function
709cce7d176Sdrh ** (like count(*) or max(value)) then write a 1 into *pIsAgg.
710cce7d176Sdrh */
711cce7d176Sdrh int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
712cce7d176Sdrh   int nErr = 0;
713cce7d176Sdrh   if( pExpr==0 ) return 0;
714cce7d176Sdrh   switch( pExpr->op ){
715cce7d176Sdrh     case TK_FUNCTION: {
716cce7d176Sdrh       int n = pExpr->pList ? pExpr->pList->nExpr : 0;
717cce7d176Sdrh       int no_such_func = 0;
7188e0a2f90Sdrh       int wrong_num_args = 0;
719cce7d176Sdrh       int is_agg = 0;
720cce7d176Sdrh       int i;
7210bce8354Sdrh       FuncDef *pDef;
7220bce8354Sdrh 
72389425d5eSdrh       pDef = sqliteFindFunction(pParse->db,
72489425d5eSdrh          pExpr->token.z, pExpr->token.n, n, 0);
7250bce8354Sdrh       if( pDef==0 ){
7260bce8354Sdrh         pDef = sqliteFindFunction(pParse->db,
7278e0a2f90Sdrh            pExpr->token.z, pExpr->token.n, -1, 0);
7280bce8354Sdrh         if( pDef==0 ){
729cce7d176Sdrh           no_such_func = 1;
7308e0a2f90Sdrh         }else{
7318e0a2f90Sdrh           wrong_num_args = 1;
7328e0a2f90Sdrh         }
7338e0a2f90Sdrh       }else{
7340bce8354Sdrh         is_agg = pDef->xFunc==0;
735cce7d176Sdrh       }
7368e0a2f90Sdrh       if( is_agg && !allowAgg ){
7378e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg, "misuse of aggregate function ", -1,
7388e0a2f90Sdrh            pExpr->token.z, pExpr->token.n, "()", 2, 0);
7398e0a2f90Sdrh         pParse->nErr++;
7408e0a2f90Sdrh         nErr++;
7418e0a2f90Sdrh         is_agg = 0;
7428e0a2f90Sdrh       }else if( no_such_func ){
743cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
744cce7d176Sdrh            pExpr->token.z, pExpr->token.n, 0);
745cce7d176Sdrh         pParse->nErr++;
746cce7d176Sdrh         nErr++;
7478e0a2f90Sdrh       }else if( wrong_num_args ){
7488e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg,
7498e0a2f90Sdrh            "wrong number of arguments to function ",-1,
7508e0a2f90Sdrh            pExpr->token.z, pExpr->token.n, "()", 2, 0);
7518e0a2f90Sdrh         pParse->nErr++;
7528e0a2f90Sdrh         nErr++;
753cce7d176Sdrh       }
7542282792aSdrh       if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
755cce7d176Sdrh       if( is_agg && pIsAgg ) *pIsAgg = 1;
756cce7d176Sdrh       for(i=0; nErr==0 && i<n; i++){
7574cfa7934Sdrh         nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
7584cfa7934Sdrh                                allowAgg && !is_agg, pIsAgg);
759cce7d176Sdrh       }
760cce7d176Sdrh     }
761cce7d176Sdrh     default: {
762cce7d176Sdrh       if( pExpr->pLeft ){
7632282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
764cce7d176Sdrh       }
765cce7d176Sdrh       if( nErr==0 && pExpr->pRight ){
7662282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
767cce7d176Sdrh       }
768fef5208cSdrh       if( nErr==0 && pExpr->pList ){
769fef5208cSdrh         int n = pExpr->pList->nExpr;
770fef5208cSdrh         int i;
771fef5208cSdrh         for(i=0; nErr==0 && i<n; i++){
7722282792aSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
7732282792aSdrh           nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
774fef5208cSdrh         }
775fef5208cSdrh       }
776cce7d176Sdrh       break;
777cce7d176Sdrh     }
778cce7d176Sdrh   }
779cce7d176Sdrh   return nErr;
780cce7d176Sdrh }
781cce7d176Sdrh 
782cce7d176Sdrh /*
783cce7d176Sdrh ** Generate code into the current Vdbe to evaluate the given
7841ccde15dSdrh ** expression and leave the result on the top of stack.
785cce7d176Sdrh */
786cce7d176Sdrh void sqliteExprCode(Parse *pParse, Expr *pExpr){
787cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
788cce7d176Sdrh   int op;
789daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
790cce7d176Sdrh   switch( pExpr->op ){
791cce7d176Sdrh     case TK_PLUS:     op = OP_Add;      break;
792cce7d176Sdrh     case TK_MINUS:    op = OP_Subtract; break;
793cce7d176Sdrh     case TK_STAR:     op = OP_Multiply; break;
794cce7d176Sdrh     case TK_SLASH:    op = OP_Divide;   break;
795cce7d176Sdrh     case TK_AND:      op = OP_And;      break;
796cce7d176Sdrh     case TK_OR:       op = OP_Or;       break;
797cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
798cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
799cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
800cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
801cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
802cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
803cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
804cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
805cce7d176Sdrh     case TK_NOT:      op = OP_Not;      break;
806cce7d176Sdrh     case TK_UMINUS:   op = OP_Negative; break;
807bf4133cbSdrh     case TK_BITAND:   op = OP_BitAnd;   break;
808bf4133cbSdrh     case TK_BITOR:    op = OP_BitOr;    break;
809bf4133cbSdrh     case TK_BITNOT:   op = OP_BitNot;   break;
810bf4133cbSdrh     case TK_LSHIFT:   op = OP_ShiftLeft;  break;
811bf4133cbSdrh     case TK_RSHIFT:   op = OP_ShiftRight; break;
812bf4133cbSdrh     case TK_REM:      op = OP_Remainder;  break;
813cce7d176Sdrh     default: break;
814cce7d176Sdrh   }
815cce7d176Sdrh   switch( pExpr->op ){
816967e8b73Sdrh     case TK_COLUMN: {
8172282792aSdrh       if( pParse->useAgg ){
81899fcd718Sdrh         sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
819c4a3c779Sdrh       }else if( pExpr->iColumn>=0 ){
82099fcd718Sdrh         sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn);
821c4a3c779Sdrh       }else{
82299fcd718Sdrh         sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0);
8232282792aSdrh       }
824cce7d176Sdrh       break;
825cce7d176Sdrh     }
826cce7d176Sdrh     case TK_INTEGER: {
827d9e30930Sdrh       int iVal = atoi(pExpr->token.z);
828d9e30930Sdrh       char zBuf[30];
829d9e30930Sdrh       sprintf(zBuf,"%d",iVal);
830d9e30930Sdrh       if( strlen(zBuf)!=pExpr->token.n
831d9e30930Sdrh             || strncmp(pExpr->token.z,zBuf,pExpr->token.n)!=0 ){
832d9e30930Sdrh         /* If the integer value cannot be represented exactly in 32 bits,
833d9e30930Sdrh         ** then code it as a string instead. */
834d9e30930Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
835d9e30930Sdrh       }else{
836d9e30930Sdrh         sqliteVdbeAddOp(v, OP_Integer, iVal, 0);
837d9e30930Sdrh       }
838e6840900Sdrh       sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
839e6840900Sdrh       break;
840e6840900Sdrh     }
841e6840900Sdrh     case TK_FLOAT: {
8427a7c7390Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
843a76b5dfcSdrh       assert( pExpr->token.z );
8447a7c7390Sdrh       sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
845cce7d176Sdrh       break;
846cce7d176Sdrh     }
847cce7d176Sdrh     case TK_STRING: {
84899fcd718Sdrh       int addr = sqliteVdbeAddOp(v, OP_String, 0, 0);
849a76b5dfcSdrh       assert( pExpr->token.z );
850cce7d176Sdrh       sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
851cce7d176Sdrh       sqliteVdbeDequoteP3(v, addr);
852cce7d176Sdrh       break;
853cce7d176Sdrh     }
854cce7d176Sdrh     case TK_NULL: {
85599fcd718Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
856cce7d176Sdrh       break;
857cce7d176Sdrh     }
858cce7d176Sdrh     case TK_AND:
859cce7d176Sdrh     case TK_OR:
860cce7d176Sdrh     case TK_PLUS:
861cce7d176Sdrh     case TK_STAR:
862cce7d176Sdrh     case TK_MINUS:
863bf4133cbSdrh     case TK_REM:
864bf4133cbSdrh     case TK_BITAND:
865bf4133cbSdrh     case TK_BITOR:
866f5905aa7Sdrh     case TK_SLASH:
867f5905aa7Sdrh     case TK_LT:
868f5905aa7Sdrh     case TK_LE:
869f5905aa7Sdrh     case TK_GT:
870f5905aa7Sdrh     case TK_GE:
871f5905aa7Sdrh     case TK_NE:
872f5905aa7Sdrh     case TK_EQ: {
873cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
874cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
87599fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
876cce7d176Sdrh       break;
877cce7d176Sdrh     }
878bf4133cbSdrh     case TK_LSHIFT:
879bf4133cbSdrh     case TK_RSHIFT: {
880bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pRight);
881bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pLeft);
882bf4133cbSdrh       sqliteVdbeAddOp(v, op, 0, 0);
883bf4133cbSdrh       break;
884bf4133cbSdrh     }
8850040077dSdrh     case TK_CONCAT: {
8860040077dSdrh       sqliteExprCode(pParse, pExpr->pLeft);
8870040077dSdrh       sqliteExprCode(pParse, pExpr->pRight);
88899fcd718Sdrh       sqliteVdbeAddOp(v, OP_Concat, 2, 0);
8890040077dSdrh       break;
8900040077dSdrh     }
891cce7d176Sdrh     case TK_UMINUS: {
8926e142f54Sdrh       assert( pExpr->pLeft );
8937a7c7390Sdrh       if( pExpr->pLeft->op==TK_FLOAT || pExpr->pLeft->op==TK_INTEGER ){
8946e142f54Sdrh         Token *p = &pExpr->pLeft->token;
8956e142f54Sdrh         char *z = sqliteMalloc( p->n + 2 );
8966e142f54Sdrh         sprintf(z, "-%.*s", p->n, p->z);
897e6840900Sdrh         if( pExpr->pLeft->op==TK_INTEGER ){
898e6840900Sdrh           sqliteVdbeAddOp(v, OP_Integer, atoi(z), 0);
899e6840900Sdrh         }else{
90099fcd718Sdrh           sqliteVdbeAddOp(v, OP_String, 0, 0);
901e6840900Sdrh         }
90299fcd718Sdrh         sqliteVdbeChangeP3(v, -1, z, p->n+1);
9036e142f54Sdrh         sqliteFree(z);
9046e142f54Sdrh         break;
9056e142f54Sdrh       }
9061ccde15dSdrh       /* Fall through into TK_NOT */
9076e142f54Sdrh     }
908bf4133cbSdrh     case TK_BITNOT:
9096e142f54Sdrh     case TK_NOT: {
910cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
91199fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
912cce7d176Sdrh       break;
913cce7d176Sdrh     }
914cce7d176Sdrh     case TK_ISNULL:
915cce7d176Sdrh     case TK_NOTNULL: {
916cce7d176Sdrh       int dest;
91799fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
918cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
919cce7d176Sdrh       dest = sqliteVdbeCurrentAddr(v) + 2;
920f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
92199fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
922cce7d176Sdrh       break;
923cce7d176Sdrh     }
9242282792aSdrh     case TK_AGG_FUNCTION: {
92599fcd718Sdrh       sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
9262282792aSdrh       break;
9272282792aSdrh     }
928cce7d176Sdrh     case TK_FUNCTION: {
929cce7d176Sdrh       int i;
930cce7d176Sdrh       ExprList *pList = pExpr->pList;
93189425d5eSdrh       int nExpr = pList ? pList->nExpr : 0;
9320bce8354Sdrh       FuncDef *pDef;
9330bce8354Sdrh       pDef = sqliteFindFunction(pParse->db,
93489425d5eSdrh                       pExpr->token.z, pExpr->token.n, nExpr, 0);
9350bce8354Sdrh       assert( pDef!=0 );
93689425d5eSdrh       for(i=0; i<nExpr; i++){
9378e0a2f90Sdrh         sqliteExprCode(pParse, pList->a[i].pExpr);
9388e0a2f90Sdrh       }
93989425d5eSdrh       sqliteVdbeAddOp(v, OP_Function, nExpr, 0);
9400bce8354Sdrh       sqliteVdbeChangeP3(v, -1, (char*)pDef, P3_POINTER);
9416ec2733bSdrh       break;
9426ec2733bSdrh     }
94319a775c2Sdrh     case TK_SELECT: {
94499fcd718Sdrh       sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0);
94519a775c2Sdrh       break;
94619a775c2Sdrh     }
947fef5208cSdrh     case TK_IN: {
948fef5208cSdrh       int addr;
94999fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
950fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
951fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
952f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+4);
953f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
954f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
955f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, addr+6);
956fef5208cSdrh       if( pExpr->pSelect ){
957f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+6);
958fef5208cSdrh       }else{
959f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+6);
960fef5208cSdrh       }
96199fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
962fef5208cSdrh       break;
963fef5208cSdrh     }
964fef5208cSdrh     case TK_BETWEEN: {
965f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
966f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
967f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
968f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, 0, 0);
969f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pull, 1, 0);
970f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
971f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, 0, 0);
972f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_And, 0, 0);
973fef5208cSdrh       break;
974fef5208cSdrh     }
975a2e00042Sdrh     case TK_AS: {
976a2e00042Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
977a2e00042Sdrh       break;
978a2e00042Sdrh     }
97917a7f8ddSdrh     case TK_CASE: {
98017a7f8ddSdrh       int expr_end_label;
981f5905aa7Sdrh       int jumpInst;
982f5905aa7Sdrh       int addr;
983f5905aa7Sdrh       int nExpr;
98417a7f8ddSdrh       int i;
98517a7f8ddSdrh 
98617a7f8ddSdrh       assert(pExpr->pList);
98717a7f8ddSdrh       assert((pExpr->pList->nExpr % 2) == 0);
98817a7f8ddSdrh       assert(pExpr->pList->nExpr > 0);
989f5905aa7Sdrh       nExpr = pExpr->pList->nExpr;
990f5905aa7Sdrh       expr_end_label = sqliteVdbeMakeLabel(v);
99117a7f8ddSdrh       if( pExpr->pLeft ){
99217a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pLeft);
993cce7d176Sdrh       }
994f5905aa7Sdrh       for(i=0; i<nExpr; i=i+2){
99517a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i].pExpr);
99617a7f8ddSdrh         if( pExpr->pLeft ){
997f5905aa7Sdrh           sqliteVdbeAddOp(v, OP_Dup, 1, 1);
998f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_Ne, 1, 0);
999f570f011Sdrh           sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1000f5905aa7Sdrh         }else{
1001f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_IfNot, 1, 0);
100217a7f8ddSdrh         }
100317a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i+1].pExpr);
1004f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Goto, 0, expr_end_label);
1005f5905aa7Sdrh         addr = sqliteVdbeCurrentAddr(v);
1006f5905aa7Sdrh         sqliteVdbeChangeP2(v, jumpInst, addr);
100717a7f8ddSdrh       }
1008f570f011Sdrh       if( pExpr->pLeft ){
1009f570f011Sdrh         sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1010f570f011Sdrh       }
101117a7f8ddSdrh       if( pExpr->pRight ){
101217a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pRight);
101317a7f8ddSdrh       }else{
1014f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
101517a7f8ddSdrh       }
1016f5905aa7Sdrh       sqliteVdbeResolveLabel(v, expr_end_label);
1017*6f34903eSdanielk1977       break;
1018*6f34903eSdanielk1977     }
1019*6f34903eSdanielk1977     case TK_RAISE: {
1020*6f34903eSdanielk1977       if( !pParse->trigStack ){
1021*6f34903eSdanielk1977         sqliteSetNString(&pParse->zErrMsg,
1022*6f34903eSdanielk1977 		"RAISE() may only be used within a trigger-program", -1, 0);
1023*6f34903eSdanielk1977         pParse->nErr++;
1024*6f34903eSdanielk1977 	return;
1025*6f34903eSdanielk1977       }
1026*6f34903eSdanielk1977       if( pExpr->iColumn == OE_Rollback ||
1027*6f34903eSdanielk1977 	  pExpr->iColumn == OE_Abort ||
1028*6f34903eSdanielk1977 	  pExpr->iColumn == OE_Fail ){
1029*6f34903eSdanielk1977 	  char * msg = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
1030*6f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Halt, SQLITE_CONSTRAINT, pExpr->iColumn);
1031*6f34903eSdanielk1977 	  sqliteDequote(msg);
1032*6f34903eSdanielk1977 	  sqliteVdbeChangeP3(v, -1, msg, 0);
1033*6f34903eSdanielk1977 	  sqliteFree(msg);
1034*6f34903eSdanielk1977       } else {
1035*6f34903eSdanielk1977 	  assert( pExpr->iColumn == OE_Ignore );
1036*6f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Goto, 0, pParse->trigStack->ignoreJump);
1037*6f34903eSdanielk1977 	  sqliteVdbeChangeP3(v, -1, "(IGNORE jump)", -1);
1038*6f34903eSdanielk1977       }
103917a7f8ddSdrh     }
104017a7f8ddSdrh     break;
104117a7f8ddSdrh   }
1042cce7d176Sdrh }
1043cce7d176Sdrh 
1044cce7d176Sdrh /*
1045cce7d176Sdrh ** Generate code for a boolean expression such that a jump is made
1046cce7d176Sdrh ** to the label "dest" if the expression is true but execution
1047cce7d176Sdrh ** continues straight thru if the expression is false.
1048f5905aa7Sdrh **
1049f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false), then
1050f5905aa7Sdrh ** take the jump if the jumpIfNull flag is true.
1051cce7d176Sdrh */
1052f5905aa7Sdrh void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1053cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1054cce7d176Sdrh   int op = 0;
1055daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1056cce7d176Sdrh   switch( pExpr->op ){
1057cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
1058cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
1059cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
1060cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
1061cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
1062cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
1063cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
1064cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
1065cce7d176Sdrh     default:  break;
1066cce7d176Sdrh   }
1067cce7d176Sdrh   switch( pExpr->op ){
1068cce7d176Sdrh     case TK_AND: {
1069cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1070f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, d2, !jumpIfNull);
1071f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1072cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1073cce7d176Sdrh       break;
1074cce7d176Sdrh     }
1075cce7d176Sdrh     case TK_OR: {
1076f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1077f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1078cce7d176Sdrh       break;
1079cce7d176Sdrh     }
1080cce7d176Sdrh     case TK_NOT: {
1081f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1082cce7d176Sdrh       break;
1083cce7d176Sdrh     }
1084cce7d176Sdrh     case TK_LT:
1085cce7d176Sdrh     case TK_LE:
1086cce7d176Sdrh     case TK_GT:
1087cce7d176Sdrh     case TK_GE:
1088cce7d176Sdrh     case TK_NE:
10890ac65892Sdrh     case TK_EQ: {
1090cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1091cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1092f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1093cce7d176Sdrh       break;
1094cce7d176Sdrh     }
1095cce7d176Sdrh     case TK_ISNULL:
1096cce7d176Sdrh     case TK_NOTNULL: {
1097cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1098f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1099cce7d176Sdrh       break;
1100cce7d176Sdrh     }
1101fef5208cSdrh     case TK_IN: {
1102f5905aa7Sdrh       int addr;
1103cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1104f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1105f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1106f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1107f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1108fef5208cSdrh       if( pExpr->pSelect ){
110999fcd718Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest);
1110fef5208cSdrh       }else{
111199fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest);
1112fef5208cSdrh       }
1113fef5208cSdrh       break;
1114fef5208cSdrh     }
1115fef5208cSdrh     case TK_BETWEEN: {
1116f5905aa7Sdrh       int addr;
1117fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
111899fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1119fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1120f5905aa7Sdrh       addr = sqliteVdbeAddOp(v, OP_Lt, !jumpIfNull, 0);
1121fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1122f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, jumpIfNull, dest);
112399fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 0, 0);
1124f5905aa7Sdrh       sqliteVdbeChangeP2(v, addr, sqliteVdbeCurrentAddr(v));
112599fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1126fef5208cSdrh       break;
1127fef5208cSdrh     }
1128cce7d176Sdrh     default: {
1129cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1130f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_If, jumpIfNull, dest);
1131cce7d176Sdrh       break;
1132cce7d176Sdrh     }
1133cce7d176Sdrh   }
1134cce7d176Sdrh }
1135cce7d176Sdrh 
1136cce7d176Sdrh /*
113766b89c8fSdrh ** Generate code for a boolean expression such that a jump is made
1138cce7d176Sdrh ** to the label "dest" if the expression is false but execution
1139cce7d176Sdrh ** continues straight thru if the expression is true.
1140f5905aa7Sdrh **
1141f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false) then
1142f5905aa7Sdrh ** jump if jumpIfNull is true or fall through if jumpIfNull is false.
1143cce7d176Sdrh */
1144f5905aa7Sdrh void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1145cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1146cce7d176Sdrh   int op = 0;
1147daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1148cce7d176Sdrh   switch( pExpr->op ){
1149cce7d176Sdrh     case TK_LT:       op = OP_Ge;       break;
1150cce7d176Sdrh     case TK_LE:       op = OP_Gt;       break;
1151cce7d176Sdrh     case TK_GT:       op = OP_Le;       break;
1152cce7d176Sdrh     case TK_GE:       op = OP_Lt;       break;
1153cce7d176Sdrh     case TK_NE:       op = OP_Eq;       break;
1154cce7d176Sdrh     case TK_EQ:       op = OP_Ne;       break;
1155cce7d176Sdrh     case TK_ISNULL:   op = OP_NotNull;  break;
1156cce7d176Sdrh     case TK_NOTNULL:  op = OP_IsNull;   break;
1157cce7d176Sdrh     default:  break;
1158cce7d176Sdrh   }
1159cce7d176Sdrh   switch( pExpr->op ){
1160cce7d176Sdrh     case TK_AND: {
1161f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1162f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1163cce7d176Sdrh       break;
1164cce7d176Sdrh     }
1165cce7d176Sdrh     case TK_OR: {
1166cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1167f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, d2, !jumpIfNull);
1168f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1169cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1170cce7d176Sdrh       break;
1171cce7d176Sdrh     }
1172cce7d176Sdrh     case TK_NOT: {
1173f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1174cce7d176Sdrh       break;
1175cce7d176Sdrh     }
1176cce7d176Sdrh     case TK_LT:
1177cce7d176Sdrh     case TK_LE:
1178cce7d176Sdrh     case TK_GT:
1179cce7d176Sdrh     case TK_GE:
1180cce7d176Sdrh     case TK_NE:
1181cce7d176Sdrh     case TK_EQ: {
1182cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1183cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1184f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1185cce7d176Sdrh       break;
1186cce7d176Sdrh     }
1187cce7d176Sdrh     case TK_ISNULL:
1188cce7d176Sdrh     case TK_NOTNULL: {
1189cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1190f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1191cce7d176Sdrh       break;
1192cce7d176Sdrh     }
1193fef5208cSdrh     case TK_IN: {
1194f5905aa7Sdrh       int addr;
1195cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1196f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1197f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1198f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1199f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1200fef5208cSdrh       if( pExpr->pSelect ){
120199fcd718Sdrh         sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest);
1202fef5208cSdrh       }else{
120399fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest);
1204fef5208cSdrh       }
1205fef5208cSdrh       break;
1206fef5208cSdrh     }
1207fef5208cSdrh     case TK_BETWEEN: {
1208fef5208cSdrh       int addr;
1209fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
121099fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1211fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1212fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
1213f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, !jumpIfNull, addr+3);
121499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
121599fcd718Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, dest);
1216fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1217f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Gt, jumpIfNull, dest);
1218fef5208cSdrh       break;
1219fef5208cSdrh     }
1220cce7d176Sdrh     default: {
1221cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1222461c281aSdrh       sqliteVdbeAddOp(v, OP_IfNot, jumpIfNull, dest);
1223cce7d176Sdrh       break;
1224cce7d176Sdrh     }
1225cce7d176Sdrh   }
1226cce7d176Sdrh }
12272282792aSdrh 
12282282792aSdrh /*
12292282792aSdrh ** Do a deep comparison of two expression trees.  Return TRUE (non-zero)
12302282792aSdrh ** if they are identical and return FALSE if they differ in any way.
12312282792aSdrh */
1232d8bc7086Sdrh int sqliteExprCompare(Expr *pA, Expr *pB){
12332282792aSdrh   int i;
12342282792aSdrh   if( pA==0 ){
12352282792aSdrh     return pB==0;
12362282792aSdrh   }else if( pB==0 ){
12372282792aSdrh     return 0;
12382282792aSdrh   }
12392282792aSdrh   if( pA->op!=pB->op ) return 0;
1240d8bc7086Sdrh   if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
1241d8bc7086Sdrh   if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
12422282792aSdrh   if( pA->pList ){
12432282792aSdrh     if( pB->pList==0 ) return 0;
12442282792aSdrh     if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
12452282792aSdrh     for(i=0; i<pA->pList->nExpr; i++){
1246d8bc7086Sdrh       if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
12472282792aSdrh         return 0;
12482282792aSdrh       }
12492282792aSdrh     }
12502282792aSdrh   }else if( pB->pList ){
12512282792aSdrh     return 0;
12522282792aSdrh   }
12532282792aSdrh   if( pA->pSelect || pB->pSelect ) return 0;
12542282792aSdrh   if( pA->token.z ){
12552282792aSdrh     if( pB->token.z==0 ) return 0;
12562282792aSdrh     if( pB->token.n!=pA->token.n ) return 0;
12572282792aSdrh     if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
12582282792aSdrh   }
12592282792aSdrh   return 1;
12602282792aSdrh }
12612282792aSdrh 
12622282792aSdrh /*
12632282792aSdrh ** Add a new element to the pParse->aAgg[] array and return its index.
12642282792aSdrh */
12652282792aSdrh static int appendAggInfo(Parse *pParse){
12662282792aSdrh   if( (pParse->nAgg & 0x7)==0 ){
12672282792aSdrh     int amt = pParse->nAgg + 8;
12686d4abfbeSdrh     AggExpr *aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
12696d4abfbeSdrh     if( aAgg==0 ){
12702282792aSdrh       return -1;
12712282792aSdrh     }
12726d4abfbeSdrh     pParse->aAgg = aAgg;
12732282792aSdrh   }
12742282792aSdrh   memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
12752282792aSdrh   return pParse->nAgg++;
12762282792aSdrh }
12772282792aSdrh 
12782282792aSdrh /*
12792282792aSdrh ** Analyze the given expression looking for aggregate functions and
12802282792aSdrh ** for variables that need to be added to the pParse->aAgg[] array.
12812282792aSdrh ** Make additional entries to the pParse->aAgg[] array as necessary.
12822282792aSdrh **
12832282792aSdrh ** This routine should only be called after the expression has been
12842282792aSdrh ** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
12852282792aSdrh **
12862282792aSdrh ** If errors are seen, leave an error message in zErrMsg and return
12872282792aSdrh ** the number of errors.
12882282792aSdrh */
12892282792aSdrh int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
12902282792aSdrh   int i;
12912282792aSdrh   AggExpr *aAgg;
12922282792aSdrh   int nErr = 0;
12932282792aSdrh 
12942282792aSdrh   if( pExpr==0 ) return 0;
12952282792aSdrh   switch( pExpr->op ){
1296967e8b73Sdrh     case TK_COLUMN: {
12972282792aSdrh       aAgg = pParse->aAgg;
12982282792aSdrh       for(i=0; i<pParse->nAgg; i++){
12992282792aSdrh         if( aAgg[i].isAgg ) continue;
13002282792aSdrh         if( aAgg[i].pExpr->iTable==pExpr->iTable
1301967e8b73Sdrh          && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
13022282792aSdrh           break;
13032282792aSdrh         }
13042282792aSdrh       }
13052282792aSdrh       if( i>=pParse->nAgg ){
13062282792aSdrh         i = appendAggInfo(pParse);
13072282792aSdrh         if( i<0 ) return 1;
13082282792aSdrh         pParse->aAgg[i].isAgg = 0;
13092282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
13102282792aSdrh       }
1311aaf88729Sdrh       pExpr->iAgg = i;
13122282792aSdrh       break;
13132282792aSdrh     }
13142282792aSdrh     case TK_AGG_FUNCTION: {
13152282792aSdrh       aAgg = pParse->aAgg;
13162282792aSdrh       for(i=0; i<pParse->nAgg; i++){
13172282792aSdrh         if( !aAgg[i].isAgg ) continue;
1318d8bc7086Sdrh         if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
13192282792aSdrh           break;
13202282792aSdrh         }
13212282792aSdrh       }
13222282792aSdrh       if( i>=pParse->nAgg ){
13232282792aSdrh         i = appendAggInfo(pParse);
13242282792aSdrh         if( i<0 ) return 1;
13252282792aSdrh         pParse->aAgg[i].isAgg = 1;
13262282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
13270bce8354Sdrh         pParse->aAgg[i].pFunc = sqliteFindFunction(pParse->db,
1328f55f25f0Sdrh              pExpr->token.z, pExpr->token.n,
1329f55f25f0Sdrh              pExpr->pList ? pExpr->pList->nExpr : 0, 0);
13302282792aSdrh       }
13312282792aSdrh       pExpr->iAgg = i;
13322282792aSdrh       break;
13332282792aSdrh     }
13342282792aSdrh     default: {
13352282792aSdrh       if( pExpr->pLeft ){
13362282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
13372282792aSdrh       }
13382282792aSdrh       if( nErr==0 && pExpr->pRight ){
13392282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
13402282792aSdrh       }
13412282792aSdrh       if( nErr==0 && pExpr->pList ){
13422282792aSdrh         int n = pExpr->pList->nExpr;
13432282792aSdrh         int i;
13442282792aSdrh         for(i=0; nErr==0 && i<n; i++){
13452282792aSdrh           nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
13462282792aSdrh         }
13472282792aSdrh       }
13482282792aSdrh       break;
13492282792aSdrh     }
13502282792aSdrh   }
13512282792aSdrh   return nErr;
13522282792aSdrh }
13538e0a2f90Sdrh 
13548e0a2f90Sdrh /*
13558e0a2f90Sdrh ** Locate a user function given a name and a number of arguments.
13560bce8354Sdrh ** Return a pointer to the FuncDef structure that defines that
13578e0a2f90Sdrh ** function, or return NULL if the function does not exist.
13588e0a2f90Sdrh **
13590bce8354Sdrh ** If the createFlag argument is true, then a new (blank) FuncDef
13608e0a2f90Sdrh ** structure is created and liked into the "db" structure if a
13618e0a2f90Sdrh ** no matching function previously existed.  When createFlag is true
13628e0a2f90Sdrh ** and the nArg parameter is -1, then only a function that accepts
13638e0a2f90Sdrh ** any number of arguments will be returned.
13648e0a2f90Sdrh **
13658e0a2f90Sdrh ** If createFlag is false and nArg is -1, then the first valid
13668e0a2f90Sdrh ** function found is returned.  A function is valid if either xFunc
13678e0a2f90Sdrh ** or xStep is non-zero.
13688e0a2f90Sdrh */
13690bce8354Sdrh FuncDef *sqliteFindFunction(
13708e0a2f90Sdrh   sqlite *db,        /* An open database */
13718e0a2f90Sdrh   const char *zName, /* Name of the function.  Not null-terminated */
13728e0a2f90Sdrh   int nName,         /* Number of characters in the name */
13738e0a2f90Sdrh   int nArg,          /* Number of arguments.  -1 means any number */
13748e0a2f90Sdrh   int createFlag     /* Create new entry if true and does not otherwise exist */
13758e0a2f90Sdrh ){
13760bce8354Sdrh   FuncDef *pFirst, *p, *pMaybe;
13770bce8354Sdrh   pFirst = p = (FuncDef*)sqliteHashFind(&db->aFunc, zName, nName);
13781350b030Sdrh   if( p && !createFlag && nArg<0 ){
13798e0a2f90Sdrh     while( p && p->xFunc==0 && p->xStep==0 ){ p = p->pNext; }
13808e0a2f90Sdrh     return p;
13818e0a2f90Sdrh   }
13828e0a2f90Sdrh   pMaybe = 0;
13838e0a2f90Sdrh   while( p && p->nArg!=nArg ){
13848e0a2f90Sdrh     if( p->nArg<0 && !createFlag && (p->xFunc || p->xStep) ) pMaybe = p;
13858e0a2f90Sdrh     p = p->pNext;
13868e0a2f90Sdrh   }
13878e0a2f90Sdrh   if( p && !createFlag && p->xFunc==0 && p->xStep==0 ){
13888e0a2f90Sdrh     return 0;
13898e0a2f90Sdrh   }
13908e0a2f90Sdrh   if( p==0 && pMaybe ){
13918e0a2f90Sdrh     assert( createFlag==0 );
13928e0a2f90Sdrh     return pMaybe;
13938e0a2f90Sdrh   }
139489425d5eSdrh   if( p==0 && createFlag && (p = sqliteMalloc(sizeof(*p)))!=0 ){
13958e0a2f90Sdrh     p->nArg = nArg;
13968e0a2f90Sdrh     p->pNext = pFirst;
13970bce8354Sdrh     sqliteHashInsert(&db->aFunc, zName, nName, (void*)p);
13988e0a2f90Sdrh   }
13998e0a2f90Sdrh   return p;
14008e0a2f90Sdrh }
1401