xref: /sqlite-3.40.0/src/expr.c (revision f26e09c8)
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*f26e09c8Sdrh ** $Id: expr.c,v 1.96 2003/05/31 16:21:12 drh 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 ){
374b59ab5eSdrh     assert( pToken->dyn==0 );
38a76b5dfcSdrh     pNew->token = *pToken;
396977fea8Sdrh     pNew->span = *pToken;
40a76b5dfcSdrh   }else{
414b59ab5eSdrh     pNew->token.dyn = 0;
42a76b5dfcSdrh     pNew->token.z = 0;
43a76b5dfcSdrh     pNew->token.n = 0;
446977fea8Sdrh     if( pLeft && pRight ){
456977fea8Sdrh       sqliteExprSpan(pNew, &pLeft->span, &pRight->span);
466977fea8Sdrh     }else{
476977fea8Sdrh       pNew->span = pNew->token;
486977fea8Sdrh     }
49a76b5dfcSdrh   }
50a76b5dfcSdrh   return pNew;
51a76b5dfcSdrh }
52a76b5dfcSdrh 
53a76b5dfcSdrh /*
546977fea8Sdrh ** Set the Expr.span field of the given expression to span all
55a76b5dfcSdrh ** text between the two given tokens.
56a76b5dfcSdrh */
57a76b5dfcSdrh void sqliteExprSpan(Expr *pExpr, Token *pLeft, Token *pRight){
586977fea8Sdrh   if( pExpr && pRight && pRight->z && pLeft && pLeft->z ){
594b59ab5eSdrh     if( pLeft->dyn==0 && pRight->dyn==0 ){
606977fea8Sdrh       pExpr->span.z = pLeft->z;
616977fea8Sdrh       pExpr->span.n = pRight->n + Addr(pRight->z) - Addr(pLeft->z);
624b59ab5eSdrh     }else{
636977fea8Sdrh       pExpr->span.z = 0;
646977fea8Sdrh       pExpr->span.n = 0;
656977fea8Sdrh       pExpr->span.dyn = 0;
664b59ab5eSdrh     }
67a76b5dfcSdrh   }
68a76b5dfcSdrh }
69a76b5dfcSdrh 
70a76b5dfcSdrh /*
71a76b5dfcSdrh ** Construct a new expression node for a function with multiple
72a76b5dfcSdrh ** arguments.
73a76b5dfcSdrh */
74a76b5dfcSdrh Expr *sqliteExprFunction(ExprList *pList, Token *pToken){
75a76b5dfcSdrh   Expr *pNew;
76a76b5dfcSdrh   pNew = sqliteMalloc( sizeof(Expr) );
77a76b5dfcSdrh   if( pNew==0 ){
78a76b5dfcSdrh     sqliteExprListDelete(pList);
79a76b5dfcSdrh     return 0;
80a76b5dfcSdrh   }
81a76b5dfcSdrh   pNew->op = TK_FUNCTION;
82a76b5dfcSdrh   pNew->pList = pList;
834b59ab5eSdrh   pNew->token.dyn = 0;
84a76b5dfcSdrh   if( pToken ){
854b59ab5eSdrh     assert( pToken->dyn==0 );
86a76b5dfcSdrh     pNew->token = *pToken;
87a76b5dfcSdrh   }else{
88a76b5dfcSdrh     pNew->token.z = 0;
89a76b5dfcSdrh     pNew->token.n = 0;
90a76b5dfcSdrh   }
916977fea8Sdrh   pNew->span = pNew->token;
92a76b5dfcSdrh   return pNew;
93a76b5dfcSdrh }
94a76b5dfcSdrh 
95a76b5dfcSdrh /*
96a2e00042Sdrh ** Recursively delete an expression tree.
97a2e00042Sdrh */
98a2e00042Sdrh void sqliteExprDelete(Expr *p){
99a2e00042Sdrh   if( p==0 ) return;
1006977fea8Sdrh   if( p->span.dyn && p->span.z ) sqliteFree((char*)p->span.z);
1014b59ab5eSdrh   if( p->token.dyn && p->token.z ) sqliteFree((char*)p->token.z);
102a2e00042Sdrh   if( p->pLeft ) sqliteExprDelete(p->pLeft);
103a2e00042Sdrh   if( p->pRight ) sqliteExprDelete(p->pRight);
104a2e00042Sdrh   if( p->pList ) sqliteExprListDelete(p->pList);
105a2e00042Sdrh   if( p->pSelect ) sqliteSelectDelete(p->pSelect);
106a2e00042Sdrh   sqliteFree(p);
107a2e00042Sdrh }
108a2e00042Sdrh 
109a76b5dfcSdrh 
110a76b5dfcSdrh /*
111ff78bd2fSdrh ** The following group of routines make deep copies of expressions,
112ff78bd2fSdrh ** expression lists, ID lists, and select statements.  The copies can
113ff78bd2fSdrh ** be deleted (by being passed to their respective ...Delete() routines)
114ff78bd2fSdrh ** without effecting the originals.
115ff78bd2fSdrh **
116ad3cab52Sdrh ** The expression list, ID, and source lists return by sqliteExprListDup(),
117ad3cab52Sdrh ** sqliteIdListDup(), and sqliteSrcListDup() can not be further expanded
118ad3cab52Sdrh ** by subsequent calls to sqlite*ListAppend() routines.
119ff78bd2fSdrh **
120ad3cab52Sdrh ** Any tables that the SrcList might point to are not duplicated.
121ff78bd2fSdrh */
122ff78bd2fSdrh Expr *sqliteExprDup(Expr *p){
123ff78bd2fSdrh   Expr *pNew;
124ff78bd2fSdrh   if( p==0 ) return 0;
125fcb78a49Sdrh   pNew = sqliteMallocRaw( sizeof(*p) );
126ff78bd2fSdrh   if( pNew==0 ) return 0;
1273b167c75Sdrh   memcpy(pNew, p, sizeof(*pNew));
1286977fea8Sdrh   if( p->token.z!=0 ){
1294b59ab5eSdrh     pNew->token.z = sqliteStrDup(p->token.z);
1304b59ab5eSdrh     pNew->token.dyn = 1;
1314b59ab5eSdrh   }else{
1324b59ab5eSdrh     pNew->token.z = 0;
1334b59ab5eSdrh     pNew->token.n = 0;
1344b59ab5eSdrh     pNew->token.dyn = 0;
1354b59ab5eSdrh   }
1366977fea8Sdrh   pNew->span.z = 0;
1376977fea8Sdrh   pNew->span.n = 0;
1386977fea8Sdrh   pNew->span.dyn = 0;
139ff78bd2fSdrh   pNew->pLeft = sqliteExprDup(p->pLeft);
140ff78bd2fSdrh   pNew->pRight = sqliteExprDup(p->pRight);
141ff78bd2fSdrh   pNew->pList = sqliteExprListDup(p->pList);
142ff78bd2fSdrh   pNew->pSelect = sqliteSelectDup(p->pSelect);
143ff78bd2fSdrh   return pNew;
144ff78bd2fSdrh }
1454b59ab5eSdrh void sqliteTokenCopy(Token *pTo, Token *pFrom){
1464b59ab5eSdrh   if( pTo->dyn ) sqliteFree((char*)pTo->z);
1474b59ab5eSdrh   if( pFrom->z ){
1484b59ab5eSdrh     pTo->n = pFrom->n;
1494b59ab5eSdrh     pTo->z = sqliteStrNDup(pFrom->z, pFrom->n);
1504b59ab5eSdrh     pTo->dyn = 1;
1514b59ab5eSdrh   }else{
1524b59ab5eSdrh     pTo->n = 0;
1534b59ab5eSdrh     pTo->z = 0;
1544b59ab5eSdrh     pTo->dyn = 0;
1554b59ab5eSdrh   }
1564b59ab5eSdrh }
157ff78bd2fSdrh ExprList *sqliteExprListDup(ExprList *p){
158ff78bd2fSdrh   ExprList *pNew;
159ff78bd2fSdrh   int i;
160ff78bd2fSdrh   if( p==0 ) return 0;
161ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*pNew) );
162ff78bd2fSdrh   if( pNew==0 ) return 0;
163ff78bd2fSdrh   pNew->nExpr = p->nExpr;
164ff78bd2fSdrh   pNew->a = sqliteMalloc( p->nExpr*sizeof(p->a[0]) );
165e4697f5eSdrh   if( pNew->a==0 ) return 0;
166ff78bd2fSdrh   for(i=0; i<p->nExpr; i++){
1674b59ab5eSdrh     Expr *pNewExpr, *pOldExpr;
1684b59ab5eSdrh     pNew->a[i].pExpr = pNewExpr = sqliteExprDup(pOldExpr = p->a[i].pExpr);
1696977fea8Sdrh     if( pOldExpr->span.z!=0 && pNewExpr ){
1706977fea8Sdrh       /* Always make a copy of the span for top-level expressions in the
1714b59ab5eSdrh       ** expression list.  The logic in SELECT processing that determines
1724b59ab5eSdrh       ** the names of columns in the result set needs this information */
1736977fea8Sdrh       sqliteTokenCopy(&pNewExpr->span, &pOldExpr->span);
1744b59ab5eSdrh     }
1751f3e905cSdrh     assert( pNewExpr==0 || pNewExpr->span.z!=0
1761f3e905cSdrh             || pOldExpr->span.z==0 || sqlite_malloc_failed );
177ff78bd2fSdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
178ff78bd2fSdrh     pNew->a[i].sortOrder = p->a[i].sortOrder;
179ff78bd2fSdrh     pNew->a[i].isAgg = p->a[i].isAgg;
180ff78bd2fSdrh     pNew->a[i].done = 0;
181ff78bd2fSdrh   }
182ff78bd2fSdrh   return pNew;
183ff78bd2fSdrh }
184ad3cab52Sdrh SrcList *sqliteSrcListDup(SrcList *p){
185ad3cab52Sdrh   SrcList *pNew;
186ad3cab52Sdrh   int i;
187113088ecSdrh   int nByte;
188ad3cab52Sdrh   if( p==0 ) return 0;
189113088ecSdrh   nByte = sizeof(*p) + (p->nSrc>0 ? sizeof(p->a[0]) * (p->nSrc-1) : 0);
190113088ecSdrh   pNew = sqliteMalloc( nByte );
191ad3cab52Sdrh   if( pNew==0 ) return 0;
192ad3cab52Sdrh   pNew->nSrc = p->nSrc;
193ad3cab52Sdrh   for(i=0; i<p->nSrc; i++){
194*f26e09c8Sdrh     pNew->a[i].zDatabase = sqliteStrDup(p->a[i].zDatabase);
195ad3cab52Sdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
196ad3cab52Sdrh     pNew->a[i].zAlias = sqliteStrDup(p->a[i].zAlias);
197ad3cab52Sdrh     pNew->a[i].jointype = p->a[i].jointype;
1986a3ea0e6Sdrh     pNew->a[i].iCursor = p->a[i].iCursor;
199ad3cab52Sdrh     pNew->a[i].pTab = 0;
200ad3cab52Sdrh     pNew->a[i].pSelect = sqliteSelectDup(p->a[i].pSelect);
201ad3cab52Sdrh     pNew->a[i].pOn = sqliteExprDup(p->a[i].pOn);
202ad3cab52Sdrh     pNew->a[i].pUsing = sqliteIdListDup(p->a[i].pUsing);
203ad3cab52Sdrh   }
204ad3cab52Sdrh   return pNew;
205ad3cab52Sdrh }
206ff78bd2fSdrh IdList *sqliteIdListDup(IdList *p){
207ff78bd2fSdrh   IdList *pNew;
208ff78bd2fSdrh   int i;
209ff78bd2fSdrh   if( p==0 ) return 0;
210ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*pNew) );
211ff78bd2fSdrh   if( pNew==0 ) return 0;
212ff78bd2fSdrh   pNew->nId = p->nId;
213ff78bd2fSdrh   pNew->a = sqliteMalloc( p->nId*sizeof(p->a[0]) );
214e4697f5eSdrh   if( pNew->a==0 ) return 0;
215ff78bd2fSdrh   for(i=0; i<p->nId; i++){
216ff78bd2fSdrh     pNew->a[i].zName = sqliteStrDup(p->a[i].zName);
217ff78bd2fSdrh     pNew->a[i].idx = p->a[i].idx;
218ff78bd2fSdrh   }
219ff78bd2fSdrh   return pNew;
220ff78bd2fSdrh }
221ff78bd2fSdrh Select *sqliteSelectDup(Select *p){
222ff78bd2fSdrh   Select *pNew;
223ff78bd2fSdrh   if( p==0 ) return 0;
224ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*p) );
225ff78bd2fSdrh   if( pNew==0 ) return 0;
226ff78bd2fSdrh   pNew->isDistinct = p->isDistinct;
227ff78bd2fSdrh   pNew->pEList = sqliteExprListDup(p->pEList);
228ad3cab52Sdrh   pNew->pSrc = sqliteSrcListDup(p->pSrc);
229ff78bd2fSdrh   pNew->pWhere = sqliteExprDup(p->pWhere);
230ff78bd2fSdrh   pNew->pGroupBy = sqliteExprListDup(p->pGroupBy);
231ff78bd2fSdrh   pNew->pHaving = sqliteExprDup(p->pHaving);
232ff78bd2fSdrh   pNew->pOrderBy = sqliteExprListDup(p->pOrderBy);
233ff78bd2fSdrh   pNew->op = p->op;
234ff78bd2fSdrh   pNew->pPrior = sqliteSelectDup(p->pPrior);
235ff78bd2fSdrh   pNew->nLimit = p->nLimit;
236ff78bd2fSdrh   pNew->nOffset = p->nOffset;
237ff78bd2fSdrh   pNew->zSelect = 0;
238ff78bd2fSdrh   return pNew;
239ff78bd2fSdrh }
240ff78bd2fSdrh 
241ff78bd2fSdrh 
242ff78bd2fSdrh /*
243a76b5dfcSdrh ** Add a new element to the end of an expression list.  If pList is
244a76b5dfcSdrh ** initially NULL, then create a new expression list.
245a76b5dfcSdrh */
246a76b5dfcSdrh ExprList *sqliteExprListAppend(ExprList *pList, Expr *pExpr, Token *pName){
247a76b5dfcSdrh   int i;
248a76b5dfcSdrh   if( pList==0 ){
249a76b5dfcSdrh     pList = sqliteMalloc( sizeof(ExprList) );
250a76b5dfcSdrh     if( pList==0 ){
251a76b5dfcSdrh       sqliteExprDelete(pExpr);
252a76b5dfcSdrh       return 0;
253a76b5dfcSdrh     }
254a76b5dfcSdrh   }
255a76b5dfcSdrh   if( (pList->nExpr & 7)==0 ){
256a76b5dfcSdrh     int n = pList->nExpr + 8;
257a76b5dfcSdrh     struct ExprList_item *a;
258a76b5dfcSdrh     a = sqliteRealloc(pList->a, n*sizeof(pList->a[0]));
259a76b5dfcSdrh     if( a==0 ){
260a76b5dfcSdrh       sqliteExprDelete(pExpr);
261a76b5dfcSdrh       return pList;
262a76b5dfcSdrh     }
263a76b5dfcSdrh     pList->a = a;
264a76b5dfcSdrh   }
265a76b5dfcSdrh   if( pExpr || pName ){
266a76b5dfcSdrh     i = pList->nExpr++;
267a76b5dfcSdrh     pList->a[i].pExpr = pExpr;
268a76b5dfcSdrh     pList->a[i].zName = 0;
269a76b5dfcSdrh     if( pName ){
270a76b5dfcSdrh       sqliteSetNString(&pList->a[i].zName, pName->z, pName->n, 0);
271a76b5dfcSdrh       sqliteDequote(pList->a[i].zName);
272a76b5dfcSdrh     }
273a76b5dfcSdrh   }
274a76b5dfcSdrh   return pList;
275a76b5dfcSdrh }
276a76b5dfcSdrh 
277a76b5dfcSdrh /*
278a76b5dfcSdrh ** Delete an entire expression list.
279a76b5dfcSdrh */
280a76b5dfcSdrh void sqliteExprListDelete(ExprList *pList){
281a76b5dfcSdrh   int i;
282a76b5dfcSdrh   if( pList==0 ) return;
283a76b5dfcSdrh   for(i=0; i<pList->nExpr; i++){
284a76b5dfcSdrh     sqliteExprDelete(pList->a[i].pExpr);
285a76b5dfcSdrh     sqliteFree(pList->a[i].zName);
286a76b5dfcSdrh   }
287a76b5dfcSdrh   sqliteFree(pList->a);
288a76b5dfcSdrh   sqliteFree(pList);
289a76b5dfcSdrh }
290a76b5dfcSdrh 
291a76b5dfcSdrh /*
292fef5208cSdrh ** Walk an expression tree.  Return 1 if the expression is constant
293fef5208cSdrh ** and 0 if it involves variables.
2942398937bSdrh **
2952398937bSdrh ** For the purposes of this function, a double-quoted string (ex: "abc")
2962398937bSdrh ** is considered a variable but a single-quoted string (ex: 'abc') is
2972398937bSdrh ** a constant.
298fef5208cSdrh */
2999208643dSdrh int sqliteExprIsConstant(Expr *p){
300fef5208cSdrh   switch( p->op ){
301fef5208cSdrh     case TK_ID:
302967e8b73Sdrh     case TK_COLUMN:
303fef5208cSdrh     case TK_DOT:
3047bdc0c1dSdrh     case TK_FUNCTION:
305fef5208cSdrh       return 0;
3067bdc0c1dSdrh     case TK_NULL:
3072398937bSdrh     case TK_STRING:
3089208643dSdrh     case TK_INTEGER:
3099208643dSdrh     case TK_FLOAT:
3109208643dSdrh       return 1;
311fef5208cSdrh     default: {
3129208643dSdrh       if( p->pLeft && !sqliteExprIsConstant(p->pLeft) ) return 0;
3139208643dSdrh       if( p->pRight && !sqliteExprIsConstant(p->pRight) ) return 0;
314fef5208cSdrh       if( p->pList ){
315fef5208cSdrh         int i;
316fef5208cSdrh         for(i=0; i<p->pList->nExpr; i++){
3179208643dSdrh           if( !sqliteExprIsConstant(p->pList->a[i].pExpr) ) return 0;
318fef5208cSdrh         }
319fef5208cSdrh       }
3209208643dSdrh       return p->pLeft!=0 || p->pRight!=0 || (p->pList && p->pList->nExpr>0);
321fef5208cSdrh     }
322fef5208cSdrh   }
3239208643dSdrh   return 0;
324fef5208cSdrh }
325fef5208cSdrh 
326fef5208cSdrh /*
327e4de1febSdrh ** If the given expression codes a constant integer, return 1 and put
328e4de1febSdrh ** the value of the integer in *pValue.  If the expression is not an
329e4de1febSdrh ** integer, return 0 and leave *pValue unchanged.
330e4de1febSdrh */
331e4de1febSdrh int sqliteExprIsInteger(Expr *p, int *pValue){
332e4de1febSdrh   switch( p->op ){
333e4de1febSdrh     case TK_INTEGER: {
334e4de1febSdrh       *pValue = atoi(p->token.z);
335e4de1febSdrh       return 1;
336e4de1febSdrh     }
337e4de1febSdrh     case TK_STRING: {
338bd790ee3Sdrh       const char *z = p->token.z;
339e4de1febSdrh       int n = p->token.n;
340bd790ee3Sdrh       if( n>0 && z[0]=='-' ){ z++; n--; }
341e4de1febSdrh       while( n>0 && *z && isdigit(*z) ){ z++; n--; }
342e4de1febSdrh       if( n==0 ){
343e4de1febSdrh         *pValue = atoi(p->token.z);
344e4de1febSdrh         return 1;
345e4de1febSdrh       }
346e4de1febSdrh       break;
347e4de1febSdrh     }
3484b59ab5eSdrh     case TK_UPLUS: {
3494b59ab5eSdrh       return sqliteExprIsInteger(p->pLeft, pValue);
3504b59ab5eSdrh     }
351e4de1febSdrh     case TK_UMINUS: {
352e4de1febSdrh       int v;
353e4de1febSdrh       if( sqliteExprIsInteger(p->pLeft, &v) ){
354e4de1febSdrh         *pValue = -v;
355e4de1febSdrh         return 1;
356e4de1febSdrh       }
357e4de1febSdrh       break;
358e4de1febSdrh     }
359e4de1febSdrh     default: break;
360e4de1febSdrh   }
361e4de1febSdrh   return 0;
362e4de1febSdrh }
363e4de1febSdrh 
364e4de1febSdrh /*
365c4a3c779Sdrh ** Return TRUE if the given string is a row-id column name.
366c4a3c779Sdrh */
367a9f9d1c0Sdrh int sqliteIsRowid(const char *z){
368c4a3c779Sdrh   if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
369c4a3c779Sdrh   if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
370c4a3c779Sdrh   if( sqliteStrICmp(z, "OID")==0 ) return 1;
371c4a3c779Sdrh   return 0;
372c4a3c779Sdrh }
373c4a3c779Sdrh 
374c4a3c779Sdrh /*
375cce7d176Sdrh ** This routine walks an expression tree and resolves references to
376967e8b73Sdrh ** table columns.  Nodes of the form ID.ID or ID resolve into an
377aacc543eSdrh ** index to the table in the table list and a column offset.  The
378aacc543eSdrh ** Expr.opcode for such nodes is changed to TK_COLUMN.  The Expr.iTable
379aacc543eSdrh ** value is changed to the index of the referenced table in pTabList
380832508b7Sdrh ** plus the "base" value.  The base value will ultimately become the
381aacc543eSdrh ** VDBE cursor number for a cursor that is pointing into the referenced
382aacc543eSdrh ** table.  The Expr.iColumn value is changed to the index of the column
383aacc543eSdrh ** of the referenced table.  The Expr.iColumn value for the special
384aacc543eSdrh ** ROWID column is -1.  Any INTEGER PRIMARY KEY column is tried as an
385aacc543eSdrh ** alias for ROWID.
38619a775c2Sdrh **
387fef5208cSdrh ** We also check for instances of the IN operator.  IN comes in two
388fef5208cSdrh ** forms:
389fef5208cSdrh **
390fef5208cSdrh **           expr IN (exprlist)
391fef5208cSdrh ** and
392fef5208cSdrh **           expr IN (SELECT ...)
393fef5208cSdrh **
394fef5208cSdrh ** The first form is handled by creating a set holding the list
395fef5208cSdrh ** of allowed values.  The second form causes the SELECT to generate
396fef5208cSdrh ** a temporary table.
397fef5208cSdrh **
398fef5208cSdrh ** This routine also looks for scalar SELECTs that are part of an expression.
39919a775c2Sdrh ** If it finds any, it generates code to write the value of that select
40019a775c2Sdrh ** into a memory cell.
401cce7d176Sdrh **
402967e8b73Sdrh ** Unknown columns or tables provoke an error.  The function returns
403cce7d176Sdrh ** the number of errors seen and leaves an error message on pParse->zErrMsg.
404cce7d176Sdrh */
405a2e00042Sdrh int sqliteExprResolveIds(
406a2e00042Sdrh   Parse *pParse,     /* The parser context */
407ad3cab52Sdrh   SrcList *pTabList, /* List of tables used to resolve column names */
408a2e00042Sdrh   ExprList *pEList,  /* List of expressions used to resolve "AS" */
409a2e00042Sdrh   Expr *pExpr        /* The expression to be analyzed. */
410a2e00042Sdrh ){
4116a3ea0e6Sdrh   int i;
4126a3ea0e6Sdrh 
413daffd0e5Sdrh   if( pExpr==0 || pTabList==0 ) return 0;
4146a3ea0e6Sdrh   for(i=0; i<pTabList->nSrc; i++){
4156a3ea0e6Sdrh     assert( pTabList->a[i].iCursor>=0 && pTabList->a[i].iCursor<pParse->nTab );
4166a3ea0e6Sdrh   }
417cce7d176Sdrh   switch( pExpr->op ){
4182398937bSdrh     /* Double-quoted strings (ex: "abc") are used as identifiers if
4192398937bSdrh     ** possible.  Otherwise they remain as strings.  Single-quoted
4202398937bSdrh     ** strings (ex: 'abc') are always string literals.
4212398937bSdrh     */
4222398937bSdrh     case TK_STRING: {
4232398937bSdrh       if( pExpr->token.z[0]=='\'' ) break;
4242398937bSdrh       /* Fall thru into the TK_ID case if this is a double-quoted string */
4252398937bSdrh     }
426a2e00042Sdrh     /* A lone identifier.  Try and match it as follows:
427a2e00042Sdrh     **
428a2e00042Sdrh     **     1.  To the name of a column of one of the tables in pTabList
429a2e00042Sdrh     **
430a2e00042Sdrh     **     2.  To the right side of an AS keyword in the column list of
431a2e00042Sdrh     **         a SELECT statement.  (For example, match against 'x' in
432a2e00042Sdrh     **         "SELECT a+b AS 'x' FROM t1".)
433a2e00042Sdrh     **
434a2e00042Sdrh     **     3.  One of the special names "ROWID", "OID", or "_ROWID_".
435a2e00042Sdrh     */
436cce7d176Sdrh     case TK_ID: {
437cce7d176Sdrh       int cnt = 0;      /* Number of matches */
438a76b5dfcSdrh       char *z;
439e22a334bSdrh       int iDb = -1;
440e22a334bSdrh 
441a76b5dfcSdrh       assert( pExpr->token.z );
442a76b5dfcSdrh       z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
4432f4392ffSdrh       sqliteDequote(z);
444daffd0e5Sdrh       if( z==0 ) return 1;
445ad3cab52Sdrh       for(i=0; i<pTabList->nSrc; i++){
446cce7d176Sdrh         int j;
447cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
448cce7d176Sdrh         if( pTab==0 ) continue;
449e22a334bSdrh         iDb = pTab->iDb;
450417be79cSdrh         assert( pTab->nCol>0 );
451cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
4527020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
453cce7d176Sdrh             cnt++;
4546a3ea0e6Sdrh             pExpr->iTable = pTabList->a[i].iCursor;
455e22a334bSdrh             pExpr->iDb = pTab->iDb;
4564a32431cSdrh             if( j==pTab->iPKey ){
4574a32431cSdrh               /* Substitute the record number for the INTEGER PRIMARY KEY */
4584a32431cSdrh               pExpr->iColumn = -1;
459c9b84a1fSdrh               pExpr->dataType = SQLITE_SO_NUM;
4604a32431cSdrh             }else{
461967e8b73Sdrh               pExpr->iColumn = j;
462c9b84a1fSdrh               pExpr->dataType = pTab->aCol[j].sortOrder & SQLITE_SO_TYPEMASK;
463cce7d176Sdrh             }
464a2e00042Sdrh             pExpr->op = TK_COLUMN;
465a2e00042Sdrh           }
466a2e00042Sdrh         }
467a2e00042Sdrh       }
468a2e00042Sdrh       if( cnt==0 && pEList!=0 ){
469a2e00042Sdrh         int j;
470a2e00042Sdrh         for(j=0; j<pEList->nExpr; j++){
471a2e00042Sdrh           char *zAs = pEList->a[j].zName;
472a2e00042Sdrh           if( zAs!=0 && sqliteStrICmp(zAs, z)==0 ){
473a2e00042Sdrh             cnt++;
474a2e00042Sdrh             assert( pExpr->pLeft==0 && pExpr->pRight==0 );
475a2e00042Sdrh             pExpr->op = TK_AS;
476a2e00042Sdrh             pExpr->iColumn = j;
47775148a27Sdrh             pExpr->pLeft = sqliteExprDup(pEList->a[j].pExpr);
478cce7d176Sdrh           }
479cce7d176Sdrh         }
4804a32431cSdrh       }
481e22a334bSdrh       if( cnt==0 && iDb>=0 && sqliteIsRowid(z) ){
482c4a3c779Sdrh         pExpr->iColumn = -1;
4836a3ea0e6Sdrh         pExpr->iTable = pTabList->a[0].iCursor;
484e22a334bSdrh         pExpr->iDb = iDb;
485ad3cab52Sdrh         cnt = 1 + (pTabList->nSrc>1);
486a2e00042Sdrh         pExpr->op = TK_COLUMN;
487c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_NUM;
488c4a3c779Sdrh       }
489cce7d176Sdrh       sqliteFree(z);
4902398937bSdrh       if( cnt==0 && pExpr->token.z[0]!='"' ){
491da93d238Sdrh         sqliteErrorMsg(pParse, "no such column: %T", &pExpr->token);
492cce7d176Sdrh         return 1;
493cce7d176Sdrh       }else if( cnt>1 ){
494da93d238Sdrh         sqliteErrorMsg(pParse, "ambiguous column name: %T", &pExpr->token);
495cce7d176Sdrh         return 1;
496cce7d176Sdrh       }
497ed6c8671Sdrh       if( pExpr->op==TK_COLUMN ){
4986a3ea0e6Sdrh         sqliteAuthRead(pParse, pExpr, pTabList);
499ed6c8671Sdrh       }
500cce7d176Sdrh       break;
501cce7d176Sdrh     }
502cce7d176Sdrh 
503d24cc427Sdrh     /* A table name and column name:     ID.ID
504d24cc427Sdrh     ** Or a database, table and column:  ID.ID.ID
505d24cc427Sdrh     */
506cce7d176Sdrh     case TK_DOT: {
507cce7d176Sdrh       int cnt = 0;             /* Number of matches */
508c4a3c779Sdrh       int cntTab = 0;          /* Number of matching tables */
509cce7d176Sdrh       int i;                   /* Loop counter */
510cce7d176Sdrh       Expr *pLeft, *pRight;    /* Left and right subbranches of the expr */
511cce7d176Sdrh       char *zLeft, *zRight;    /* Text of an identifier */
512d24cc427Sdrh       char *zDb;               /* Name of database holding table */
513d24cc427Sdrh       sqlite *db = pParse->db;
514cce7d176Sdrh 
515cce7d176Sdrh       pRight = pExpr->pRight;
516d24cc427Sdrh       if( pRight->op==TK_ID ){
517d24cc427Sdrh         pLeft = pExpr->pLeft;
518d24cc427Sdrh         zDb = 0;
519d24cc427Sdrh       }else{
520d24cc427Sdrh         Expr *pDb = pExpr->pLeft;
521d24cc427Sdrh         assert( pDb && pDb->op==TK_ID && pDb->token.z );
522d24cc427Sdrh         zDb = sqliteStrNDup(pDb->token.z, pDb->token.n);
523d24cc427Sdrh         pLeft = pRight->pLeft;
524d24cc427Sdrh         pRight = pRight->pRight;
525d24cc427Sdrh       }
526a76b5dfcSdrh       assert( pLeft && pLeft->op==TK_ID && pLeft->token.z );
527a76b5dfcSdrh       assert( pRight && pRight->op==TK_ID && pRight->token.z );
5286e142f54Sdrh       zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
5296e142f54Sdrh       zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
530daffd0e5Sdrh       if( zLeft==0 || zRight==0 ){
531daffd0e5Sdrh         sqliteFree(zLeft);
532daffd0e5Sdrh         sqliteFree(zRight);
533d24cc427Sdrh         sqliteFree(zDb);
534daffd0e5Sdrh         return 1;
535daffd0e5Sdrh       }
536d24cc427Sdrh       sqliteDequote(zDb);
53787c40e88Sdrh       sqliteDequote(zLeft);
53887c40e88Sdrh       sqliteDequote(zRight);
539c4a3c779Sdrh       pExpr->iTable = -1;
540ad3cab52Sdrh       for(i=0; i<pTabList->nSrc; i++){
541cce7d176Sdrh         int j;
542cce7d176Sdrh         char *zTab;
543cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
544cce7d176Sdrh         if( pTab==0 ) continue;
545417be79cSdrh         assert( pTab->nCol>0 );
546cce7d176Sdrh         if( pTabList->a[i].zAlias ){
547cce7d176Sdrh           zTab = pTabList->a[i].zAlias;
548d24cc427Sdrh           if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
549cce7d176Sdrh         }else{
550cce7d176Sdrh           zTab = pTab->zName;
551094b2bbfSdrh           if( zTab==0 || sqliteStrICmp(zTab, zLeft)!=0 ) continue;
552d24cc427Sdrh           if( zDb!=0 && sqliteStrICmp(db->aDb[pTab->iDb].zName, zDb)!=0 ){
553d24cc427Sdrh             continue;
554d24cc427Sdrh           }
555d24cc427Sdrh         }
556e22a334bSdrh         if( 0==(cntTab++) ){
5576a3ea0e6Sdrh           pExpr->iTable = pTabList->a[i].iCursor;
558e22a334bSdrh           pExpr->iDb = pTab->iDb;
559e22a334bSdrh         }
560cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
5617020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
562cce7d176Sdrh             cnt++;
5636a3ea0e6Sdrh             pExpr->iTable = pTabList->a[i].iCursor;
564e22a334bSdrh             pExpr->iDb = pTab->iDb;
56570ce3f0cSdrh             /* Substitute the rowid (column -1) for the INTEGER PRIMARY KEY */
56670ce3f0cSdrh             pExpr->iColumn = j==pTab->iPKey ? -1 : j;
567c9b84a1fSdrh             pExpr->dataType = pTab->aCol[j].sortOrder & SQLITE_SO_TYPEMASK;
568cce7d176Sdrh           }
569cce7d176Sdrh         }
5704a32431cSdrh       }
571c3f9bad2Sdanielk1977 
572c3f9bad2Sdanielk1977       /* If we have not already resolved this *.* expression, then maybe
573c3f9bad2Sdanielk1977        * it is a new.* or old.* trigger argument reference */
574c3f9bad2Sdanielk1977       if( cnt == 0 && pParse->trigStack != 0 ){
575f29ce559Sdanielk1977         TriggerStack *pTriggerStack = pParse->trigStack;
576c3f9bad2Sdanielk1977         int t = 0;
577f29ce559Sdanielk1977         if( pTriggerStack->newIdx != -1 && sqliteStrICmp("new", zLeft) == 0 ){
578f29ce559Sdanielk1977           pExpr->iTable = pTriggerStack->newIdx;
579e22a334bSdrh           assert( pTriggerStack->pTab );
580e22a334bSdrh           pExpr->iDb = pTriggerStack->pTab->iDb;
581c3f9bad2Sdanielk1977           cntTab++;
582c3f9bad2Sdanielk1977           t = 1;
583c3f9bad2Sdanielk1977         }
584f29ce559Sdanielk1977         if( pTriggerStack->oldIdx != -1 && sqliteStrICmp("old", zLeft) == 0 ){
585f29ce559Sdanielk1977           pExpr->iTable = pTriggerStack->oldIdx;
586e22a334bSdrh           assert( pTriggerStack->pTab );
587e22a334bSdrh           pExpr->iDb = pTriggerStack->pTab->iDb;
588c3f9bad2Sdanielk1977           cntTab++;
589c3f9bad2Sdanielk1977           t = 1;
590c3f9bad2Sdanielk1977         }
591c3f9bad2Sdanielk1977 
592f29ce559Sdanielk1977         if( t ){
593f29ce559Sdanielk1977 	  int j;
594c9b84a1fSdrh           Table *pTab = pTriggerStack->pTab;
595c9b84a1fSdrh           for(j=0; j < pTab->nCol; j++) {
596c9b84a1fSdrh             if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
597c3f9bad2Sdanielk1977               cnt++;
59870ce3f0cSdrh               pExpr->iColumn = j==pTab->iPKey ? -1 : j;
599c9b84a1fSdrh               pExpr->dataType = pTab->aCol[j].sortOrder & SQLITE_SO_TYPEMASK;
600c3f9bad2Sdanielk1977             }
601c3f9bad2Sdanielk1977           }
602c3f9bad2Sdanielk1977 	}
603f29ce559Sdanielk1977       }
604c3f9bad2Sdanielk1977 
605c4a3c779Sdrh       if( cnt==0 && cntTab==1 && sqliteIsRowid(zRight) ){
606c4a3c779Sdrh         cnt = 1;
607c4a3c779Sdrh         pExpr->iColumn = -1;
608c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_NUM;
609c4a3c779Sdrh       }
610d24cc427Sdrh       sqliteFree(zDb);
611cce7d176Sdrh       sqliteFree(zLeft);
612cce7d176Sdrh       sqliteFree(zRight);
613cce7d176Sdrh       if( cnt==0 ){
614da93d238Sdrh         sqliteErrorMsg(pParse, "no such column: %T.%T",
615da93d238Sdrh                &pLeft->token, &pRight->token);
616cce7d176Sdrh         return 1;
617cce7d176Sdrh       }else if( cnt>1 ){
618da93d238Sdrh         sqliteErrorMsg(pParse, "ambiguous column name: %T.%T",
619da93d238Sdrh           &pLeft->token, &pRight->token);
620cce7d176Sdrh         return 1;
621cce7d176Sdrh       }
622d24cc427Sdrh       sqliteExprDelete(pExpr->pLeft);
623cce7d176Sdrh       pExpr->pLeft = 0;
624d24cc427Sdrh       sqliteExprDelete(pExpr->pRight);
625cce7d176Sdrh       pExpr->pRight = 0;
626967e8b73Sdrh       pExpr->op = TK_COLUMN;
6276a3ea0e6Sdrh       sqliteAuthRead(pParse, pExpr, pTabList);
628cce7d176Sdrh       break;
629cce7d176Sdrh     }
630cce7d176Sdrh 
631fef5208cSdrh     case TK_IN: {
632d8bc7086Sdrh       Vdbe *v = sqliteGetVdbe(pParse);
633fef5208cSdrh       if( v==0 ) return 1;
6346a3ea0e6Sdrh       if( sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pLeft) ){
635cfab11bcSdrh         return 1;
636cfab11bcSdrh       }
637fef5208cSdrh       if( pExpr->pSelect ){
638fef5208cSdrh         /* Case 1:     expr IN (SELECT ...)
639fef5208cSdrh         **
640fef5208cSdrh         ** Generate code to write the results of the select into a temporary
6414794b980Sdrh         ** table.  The cursor number of the temporary table has already
6424794b980Sdrh         ** been put in iTable by sqliteExprResolveInSelect().
643fef5208cSdrh         */
644832508b7Sdrh         pExpr->iTable = pParse->nTab++;
645c6b52df3Sdrh         sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 1);
646832508b7Sdrh         sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable, 0,0,0);
647fef5208cSdrh       }else if( pExpr->pList ){
648fef5208cSdrh         /* Case 2:     expr IN (exprlist)
649fef5208cSdrh         **
650fef5208cSdrh         ** Create a set to put the exprlist values in.  The Set id is stored
651fef5208cSdrh         ** in iTable.
652fef5208cSdrh         */
653fef5208cSdrh         int i, iSet;
654fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
655fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
6569208643dSdrh           if( !sqliteExprIsConstant(pE2) ){
657da93d238Sdrh             sqliteErrorMsg(pParse,
658da93d238Sdrh               "right-hand side of IN operator must be constant");
659fef5208cSdrh             return 1;
660fef5208cSdrh           }
6614794b980Sdrh           if( sqliteExprCheck(pParse, pE2, 0, 0) ){
6624794b980Sdrh             return 1;
6634794b980Sdrh           }
664fef5208cSdrh         }
665fef5208cSdrh         iSet = pExpr->iTable = pParse->nSet++;
666fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
667fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
668fef5208cSdrh           switch( pE2->op ){
669fef5208cSdrh             case TK_FLOAT:
670fef5208cSdrh             case TK_INTEGER:
671fef5208cSdrh             case TK_STRING: {
67299fcd718Sdrh               int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
673a76b5dfcSdrh               assert( pE2->token.z );
674fef5208cSdrh               sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
675fef5208cSdrh               sqliteVdbeDequoteP3(v, addr);
676fef5208cSdrh               break;
677fef5208cSdrh             }
678fef5208cSdrh             default: {
679fef5208cSdrh               sqliteExprCode(pParse, pE2);
68099fcd718Sdrh               sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
681fef5208cSdrh               break;
682fef5208cSdrh             }
683fef5208cSdrh           }
684fef5208cSdrh         }
685fef5208cSdrh       }
686cfab11bcSdrh       break;
687fef5208cSdrh     }
688fef5208cSdrh 
68919a775c2Sdrh     case TK_SELECT: {
690fef5208cSdrh       /* This has to be a scalar SELECT.  Generate code to put the
691fef5208cSdrh       ** value of this select in a memory cell and record the number
692967e8b73Sdrh       ** of the memory cell in iColumn.
693fef5208cSdrh       */
694967e8b73Sdrh       pExpr->iColumn = pParse->nMem++;
695832508b7Sdrh       if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn,0,0,0) ){
69619a775c2Sdrh         return 1;
69719a775c2Sdrh       }
69819a775c2Sdrh       break;
69919a775c2Sdrh     }
70019a775c2Sdrh 
701cce7d176Sdrh     /* For all else, just recursively walk the tree */
702cce7d176Sdrh     default: {
703cce7d176Sdrh       if( pExpr->pLeft
7046a3ea0e6Sdrh       && sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pLeft) ){
705cce7d176Sdrh         return 1;
706cce7d176Sdrh       }
707cce7d176Sdrh       if( pExpr->pRight
7086a3ea0e6Sdrh       && sqliteExprResolveIds(pParse, pTabList, pEList, pExpr->pRight) ){
709cce7d176Sdrh         return 1;
710cce7d176Sdrh       }
711cce7d176Sdrh       if( pExpr->pList ){
712cce7d176Sdrh         int i;
713cce7d176Sdrh         ExprList *pList = pExpr->pList;
714cce7d176Sdrh         for(i=0; i<pList->nExpr; i++){
715832508b7Sdrh           Expr *pArg = pList->a[i].pExpr;
7166a3ea0e6Sdrh           if( sqliteExprResolveIds(pParse, pTabList, pEList, pArg) ){
717cce7d176Sdrh             return 1;
718cce7d176Sdrh           }
719cce7d176Sdrh         }
720cce7d176Sdrh       }
721cce7d176Sdrh     }
722cce7d176Sdrh   }
723cce7d176Sdrh   return 0;
724cce7d176Sdrh }
725cce7d176Sdrh 
726cce7d176Sdrh /*
7274b59ab5eSdrh ** pExpr is a node that defines a function of some kind.  It might
7284b59ab5eSdrh ** be a syntactic function like "count(x)" or it might be a function
7294b59ab5eSdrh ** that implements an operator, like "a LIKE b".
7304b59ab5eSdrh **
7314b59ab5eSdrh ** This routine makes *pzName point to the name of the function and
7324b59ab5eSdrh ** *pnName hold the number of characters in the function name.
7334b59ab5eSdrh */
7344b59ab5eSdrh static void getFunctionName(Expr *pExpr, const char **pzName, int *pnName){
7354b59ab5eSdrh   switch( pExpr->op ){
7364b59ab5eSdrh     case TK_FUNCTION: {
7374b59ab5eSdrh       *pzName = pExpr->token.z;
7386977fea8Sdrh       *pnName = pExpr->token.n;
7394b59ab5eSdrh       break;
7404b59ab5eSdrh     }
7414b59ab5eSdrh     case TK_LIKE: {
7424b59ab5eSdrh       *pzName = "like";
7434b59ab5eSdrh       *pnName = 4;
7444b59ab5eSdrh       break;
7454b59ab5eSdrh     }
7464b59ab5eSdrh     case TK_GLOB: {
7474b59ab5eSdrh       *pzName = "glob";
7484b59ab5eSdrh       *pnName = 4;
7494b59ab5eSdrh       break;
7504b59ab5eSdrh     }
7514b59ab5eSdrh     default: {
7524b59ab5eSdrh       *pzName = "can't happen";
7534b59ab5eSdrh       *pnName = 12;
7544b59ab5eSdrh       break;
7554b59ab5eSdrh     }
7564b59ab5eSdrh   }
7574b59ab5eSdrh }
7584b59ab5eSdrh 
7594b59ab5eSdrh /*
760cce7d176Sdrh ** Error check the functions in an expression.  Make sure all
761cce7d176Sdrh ** function names are recognized and all functions have the correct
762cce7d176Sdrh ** number of arguments.  Leave an error message in pParse->zErrMsg
763cce7d176Sdrh ** if anything is amiss.  Return the number of errors.
764cce7d176Sdrh **
765cce7d176Sdrh ** if pIsAgg is not null and this expression is an aggregate function
766cce7d176Sdrh ** (like count(*) or max(value)) then write a 1 into *pIsAgg.
767cce7d176Sdrh */
768cce7d176Sdrh int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
769cce7d176Sdrh   int nErr = 0;
770cce7d176Sdrh   if( pExpr==0 ) return 0;
771cce7d176Sdrh   switch( pExpr->op ){
7724b59ab5eSdrh     case TK_GLOB:
7734b59ab5eSdrh     case TK_LIKE:
774cce7d176Sdrh     case TK_FUNCTION: {
775c9b84a1fSdrh       int n = pExpr->pList ? pExpr->pList->nExpr : 0;  /* Number of arguments */
776c9b84a1fSdrh       int no_such_func = 0;       /* True if no such function exists */
777c9b84a1fSdrh       int is_type_of = 0;         /* True if is the special TypeOf() function */
778c9b84a1fSdrh       int wrong_num_args = 0;     /* True if wrong number of arguments */
779c9b84a1fSdrh       int is_agg = 0;             /* True if is an aggregate function */
780cce7d176Sdrh       int i;
7814b59ab5eSdrh       int nId;                    /* Number of characters in function name */
7824b59ab5eSdrh       const char *zId;            /* The function name. */
7830bce8354Sdrh       FuncDef *pDef;
7840bce8354Sdrh 
7854b59ab5eSdrh       getFunctionName(pExpr, &zId, &nId);
7864b59ab5eSdrh       pDef = sqliteFindFunction(pParse->db, zId, nId, n, 0);
7870bce8354Sdrh       if( pDef==0 ){
7884b59ab5eSdrh         pDef = sqliteFindFunction(pParse->db, zId, nId, -1, 0);
7890bce8354Sdrh         if( pDef==0 ){
7904b59ab5eSdrh           if( n==1 && nId==6 && sqliteStrNICmp(zId, "typeof", 6)==0 ){
791c9b84a1fSdrh             is_type_of = 1;
792c9b84a1fSdrh           }else {
793cce7d176Sdrh             no_such_func = 1;
794c9b84a1fSdrh           }
7958e0a2f90Sdrh         }else{
7968e0a2f90Sdrh           wrong_num_args = 1;
7978e0a2f90Sdrh         }
7988e0a2f90Sdrh       }else{
7990bce8354Sdrh         is_agg = pDef->xFunc==0;
800cce7d176Sdrh       }
8018e0a2f90Sdrh       if( is_agg && !allowAgg ){
8028e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg, "misuse of aggregate function ", -1,
8034b59ab5eSdrh            zId, nId, "()", 2, 0);
8048e0a2f90Sdrh         pParse->nErr++;
8058e0a2f90Sdrh         nErr++;
8068e0a2f90Sdrh         is_agg = 0;
8078e0a2f90Sdrh       }else if( no_such_func ){
8084b59ab5eSdrh         sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1, zId,nId,0);
809cce7d176Sdrh         pParse->nErr++;
810cce7d176Sdrh         nErr++;
8118e0a2f90Sdrh       }else if( wrong_num_args ){
8128e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg,
8134b59ab5eSdrh            "wrong number of arguments to function ", -1, zId, nId, "()", 2, 0);
8148e0a2f90Sdrh         pParse->nErr++;
8158e0a2f90Sdrh         nErr++;
816cce7d176Sdrh       }
8172282792aSdrh       if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
818cce7d176Sdrh       if( is_agg && pIsAgg ) *pIsAgg = 1;
819cce7d176Sdrh       for(i=0; nErr==0 && i<n; i++){
8204cfa7934Sdrh         nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
8214cfa7934Sdrh                                allowAgg && !is_agg, pIsAgg);
822cce7d176Sdrh       }
823c9b84a1fSdrh       if( pDef==0 ){
824c9b84a1fSdrh         if( is_type_of ){
825c9b84a1fSdrh           pExpr->op = TK_STRING;
826c9b84a1fSdrh           if( sqliteExprType(pExpr->pList->a[0].pExpr)==SQLITE_SO_NUM ){
827c9b84a1fSdrh             pExpr->token.z = "numeric";
828c9b84a1fSdrh             pExpr->token.n = 7;
829c9b84a1fSdrh           }else{
830c9b84a1fSdrh             pExpr->token.z = "text";
831c9b84a1fSdrh             pExpr->token.n = 4;
832c9b84a1fSdrh           }
833c9b84a1fSdrh         }
834c9b84a1fSdrh       }else if( pDef->dataType>=0 ){
835c9b84a1fSdrh         if( pDef->dataType<n ){
836c9b84a1fSdrh           pExpr->dataType =
837c9b84a1fSdrh              sqliteExprType(pExpr->pList->a[pDef->dataType].pExpr);
838c9b84a1fSdrh         }else{
839c9b84a1fSdrh           pExpr->dataType = SQLITE_SO_NUM;
840c9b84a1fSdrh         }
841c9b84a1fSdrh       }else if( pDef->dataType==SQLITE_ARGS ){
842c9b84a1fSdrh         pDef->dataType = SQLITE_SO_TEXT;
843c9b84a1fSdrh         for(i=0; i<n; i++){
844c9b84a1fSdrh           if( sqliteExprType(pExpr->pList->a[i].pExpr)==SQLITE_SO_NUM ){
845c9b84a1fSdrh             pExpr->dataType = SQLITE_SO_NUM;
846c9b84a1fSdrh             break;
847c9b84a1fSdrh           }
848c9b84a1fSdrh         }
849c9b84a1fSdrh       }else if( pDef->dataType==SQLITE_NUMERIC ){
850c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_NUM;
851c9b84a1fSdrh       }else{
852c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_TEXT;
853c9b84a1fSdrh       }
854cce7d176Sdrh     }
855cce7d176Sdrh     default: {
856cce7d176Sdrh       if( pExpr->pLeft ){
8572282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
858cce7d176Sdrh       }
859cce7d176Sdrh       if( nErr==0 && pExpr->pRight ){
8602282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
861cce7d176Sdrh       }
862fef5208cSdrh       if( nErr==0 && pExpr->pList ){
863fef5208cSdrh         int n = pExpr->pList->nExpr;
864fef5208cSdrh         int i;
865fef5208cSdrh         for(i=0; nErr==0 && i<n; i++){
8662282792aSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
8672282792aSdrh           nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
868fef5208cSdrh         }
869fef5208cSdrh       }
870cce7d176Sdrh       break;
871cce7d176Sdrh     }
872cce7d176Sdrh   }
873cce7d176Sdrh   return nErr;
874cce7d176Sdrh }
875cce7d176Sdrh 
876cce7d176Sdrh /*
877c9b84a1fSdrh ** Return either SQLITE_SO_NUM or SQLITE_SO_TEXT to indicate whether the
878c9b84a1fSdrh ** given expression should sort as numeric values or as text.
879c9b84a1fSdrh **
880c9b84a1fSdrh ** The sqliteExprResolveIds() and sqliteExprCheck() routines must have
881c9b84a1fSdrh ** both been called on the expression before it is passed to this routine.
882c9b84a1fSdrh */
883c9b84a1fSdrh int sqliteExprType(Expr *p){
884c9b84a1fSdrh   if( p==0 ) return SQLITE_SO_NUM;
885c9b84a1fSdrh   while( p ) switch( p->op ){
886c9b84a1fSdrh     case TK_PLUS:
887c9b84a1fSdrh     case TK_MINUS:
888c9b84a1fSdrh     case TK_STAR:
889c9b84a1fSdrh     case TK_SLASH:
890c9b84a1fSdrh     case TK_AND:
891c9b84a1fSdrh     case TK_OR:
892c9b84a1fSdrh     case TK_ISNULL:
893c9b84a1fSdrh     case TK_NOTNULL:
894c9b84a1fSdrh     case TK_NOT:
895c9b84a1fSdrh     case TK_UMINUS:
8964b59ab5eSdrh     case TK_UPLUS:
897c9b84a1fSdrh     case TK_BITAND:
898c9b84a1fSdrh     case TK_BITOR:
899c9b84a1fSdrh     case TK_BITNOT:
900c9b84a1fSdrh     case TK_LSHIFT:
901c9b84a1fSdrh     case TK_RSHIFT:
902c9b84a1fSdrh     case TK_REM:
903c9b84a1fSdrh     case TK_INTEGER:
904c9b84a1fSdrh     case TK_FLOAT:
905c9b84a1fSdrh     case TK_IN:
906c9b84a1fSdrh     case TK_BETWEEN:
9074b59ab5eSdrh     case TK_GLOB:
9084b59ab5eSdrh     case TK_LIKE:
909c9b84a1fSdrh       return SQLITE_SO_NUM;
910c9b84a1fSdrh 
911c9b84a1fSdrh     case TK_STRING:
912c9b84a1fSdrh     case TK_NULL:
913c9b84a1fSdrh     case TK_CONCAT:
914c9b84a1fSdrh       return SQLITE_SO_TEXT;
915c9b84a1fSdrh 
916c9b84a1fSdrh     case TK_LT:
917c9b84a1fSdrh     case TK_LE:
918c9b84a1fSdrh     case TK_GT:
919c9b84a1fSdrh     case TK_GE:
920c9b84a1fSdrh     case TK_NE:
921c9b84a1fSdrh     case TK_EQ:
922c9b84a1fSdrh       if( sqliteExprType(p->pLeft)==SQLITE_SO_NUM ){
923c9b84a1fSdrh         return SQLITE_SO_NUM;
924c9b84a1fSdrh       }
925c9b84a1fSdrh       p = p->pRight;
926c9b84a1fSdrh       break;
927c9b84a1fSdrh 
928c9b84a1fSdrh     case TK_AS:
929c9b84a1fSdrh       p = p->pLeft;
930c9b84a1fSdrh       break;
931c9b84a1fSdrh 
932c9b84a1fSdrh     case TK_COLUMN:
933c9b84a1fSdrh     case TK_FUNCTION:
934c9b84a1fSdrh     case TK_AGG_FUNCTION:
935c9b84a1fSdrh       return p->dataType;
936c9b84a1fSdrh 
937c9b84a1fSdrh     case TK_SELECT:
938c9b84a1fSdrh       assert( p->pSelect );
939c9b84a1fSdrh       assert( p->pSelect->pEList );
940c9b84a1fSdrh       assert( p->pSelect->pEList->nExpr>0 );
941c9b84a1fSdrh       p = p->pSelect->pEList->a[0].pExpr;
942c9b84a1fSdrh       break;
943c9b84a1fSdrh 
944b1363206Sdrh     case TK_CASE: {
945b1363206Sdrh       if( p->pRight && sqliteExprType(p->pRight)==SQLITE_SO_NUM ){
946b1363206Sdrh         return SQLITE_SO_NUM;
947b1363206Sdrh       }
948b1363206Sdrh       if( p->pList ){
949b1363206Sdrh         int i;
950b1363206Sdrh         ExprList *pList = p->pList;
951b1363206Sdrh         for(i=1; i<pList->nExpr; i+=2){
952b1363206Sdrh           if( sqliteExprType(pList->a[i].pExpr)==SQLITE_SO_NUM ){
953b1363206Sdrh             return SQLITE_SO_NUM;
954b1363206Sdrh           }
955b1363206Sdrh         }
956b1363206Sdrh       }
957b1363206Sdrh       return SQLITE_SO_TEXT;
958b1363206Sdrh     }
959b1363206Sdrh 
960c9b84a1fSdrh     default:
961c9b84a1fSdrh       assert( p->op==TK_ABORT );  /* Can't Happen */
962c9b84a1fSdrh       break;
963c9b84a1fSdrh   }
964c9b84a1fSdrh   return SQLITE_SO_NUM;
965c9b84a1fSdrh }
966c9b84a1fSdrh 
967c9b84a1fSdrh /*
968cce7d176Sdrh ** Generate code into the current Vdbe to evaluate the given
9691ccde15dSdrh ** expression and leave the result on the top of stack.
970cce7d176Sdrh */
971cce7d176Sdrh void sqliteExprCode(Parse *pParse, Expr *pExpr){
972cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
973cce7d176Sdrh   int op;
974daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
975cce7d176Sdrh   switch( pExpr->op ){
976cce7d176Sdrh     case TK_PLUS:     op = OP_Add;      break;
977cce7d176Sdrh     case TK_MINUS:    op = OP_Subtract; break;
978cce7d176Sdrh     case TK_STAR:     op = OP_Multiply; break;
979cce7d176Sdrh     case TK_SLASH:    op = OP_Divide;   break;
980cce7d176Sdrh     case TK_AND:      op = OP_And;      break;
981cce7d176Sdrh     case TK_OR:       op = OP_Or;       break;
982cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
983cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
984cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
985cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
986cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
987cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
988cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
989cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
990cce7d176Sdrh     case TK_NOT:      op = OP_Not;      break;
991cce7d176Sdrh     case TK_UMINUS:   op = OP_Negative; break;
992bf4133cbSdrh     case TK_BITAND:   op = OP_BitAnd;   break;
993bf4133cbSdrh     case TK_BITOR:    op = OP_BitOr;    break;
994bf4133cbSdrh     case TK_BITNOT:   op = OP_BitNot;   break;
995bf4133cbSdrh     case TK_LSHIFT:   op = OP_ShiftLeft;  break;
996bf4133cbSdrh     case TK_RSHIFT:   op = OP_ShiftRight; break;
997bf4133cbSdrh     case TK_REM:      op = OP_Remainder;  break;
998cce7d176Sdrh     default: break;
999cce7d176Sdrh   }
1000cce7d176Sdrh   switch( pExpr->op ){
1001967e8b73Sdrh     case TK_COLUMN: {
10022282792aSdrh       if( pParse->useAgg ){
100399fcd718Sdrh         sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
1004c4a3c779Sdrh       }else if( pExpr->iColumn>=0 ){
100599fcd718Sdrh         sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn);
1006c4a3c779Sdrh       }else{
100799fcd718Sdrh         sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0);
10082282792aSdrh       }
1009cce7d176Sdrh       break;
1010cce7d176Sdrh     }
1011cce7d176Sdrh     case TK_INTEGER: {
1012d9e30930Sdrh       int iVal = atoi(pExpr->token.z);
1013d9e30930Sdrh       char zBuf[30];
1014d9e30930Sdrh       sprintf(zBuf,"%d",iVal);
1015d9e30930Sdrh       if( strlen(zBuf)!=pExpr->token.n
1016d9e30930Sdrh             || strncmp(pExpr->token.z,zBuf,pExpr->token.n)!=0 ){
1017d9e30930Sdrh         /* If the integer value cannot be represented exactly in 32 bits,
1018d9e30930Sdrh         ** then code it as a string instead. */
1019d9e30930Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
1020d9e30930Sdrh       }else{
1021d9e30930Sdrh         sqliteVdbeAddOp(v, OP_Integer, iVal, 0);
1022d9e30930Sdrh       }
1023e6840900Sdrh       sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
1024e6840900Sdrh       break;
1025e6840900Sdrh     }
1026e6840900Sdrh     case TK_FLOAT: {
10277a7c7390Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
1028a76b5dfcSdrh       assert( pExpr->token.z );
10297a7c7390Sdrh       sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
1030cce7d176Sdrh       break;
1031cce7d176Sdrh     }
1032cce7d176Sdrh     case TK_STRING: {
103399fcd718Sdrh       int addr = sqliteVdbeAddOp(v, OP_String, 0, 0);
1034a76b5dfcSdrh       assert( pExpr->token.z );
1035cce7d176Sdrh       sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
1036cce7d176Sdrh       sqliteVdbeDequoteP3(v, addr);
1037cce7d176Sdrh       break;
1038cce7d176Sdrh     }
1039cce7d176Sdrh     case TK_NULL: {
104099fcd718Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
1041cce7d176Sdrh       break;
1042cce7d176Sdrh     }
1043c9b84a1fSdrh     case TK_LT:
1044c9b84a1fSdrh     case TK_LE:
1045c9b84a1fSdrh     case TK_GT:
1046c9b84a1fSdrh     case TK_GE:
1047c9b84a1fSdrh     case TK_NE:
1048c9b84a1fSdrh     case TK_EQ: {
1049491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
1050c9b84a1fSdrh         op += 6;  /* Convert numeric opcodes to text opcodes */
1051c9b84a1fSdrh       }
1052c9b84a1fSdrh       /* Fall through into the next case */
1053c9b84a1fSdrh     }
1054cce7d176Sdrh     case TK_AND:
1055cce7d176Sdrh     case TK_OR:
1056cce7d176Sdrh     case TK_PLUS:
1057cce7d176Sdrh     case TK_STAR:
1058cce7d176Sdrh     case TK_MINUS:
1059bf4133cbSdrh     case TK_REM:
1060bf4133cbSdrh     case TK_BITAND:
1061bf4133cbSdrh     case TK_BITOR:
1062c9b84a1fSdrh     case TK_SLASH: {
1063cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1064cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
106599fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
1066cce7d176Sdrh       break;
1067cce7d176Sdrh     }
1068bf4133cbSdrh     case TK_LSHIFT:
1069bf4133cbSdrh     case TK_RSHIFT: {
1070bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pRight);
1071bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1072bf4133cbSdrh       sqliteVdbeAddOp(v, op, 0, 0);
1073bf4133cbSdrh       break;
1074bf4133cbSdrh     }
10750040077dSdrh     case TK_CONCAT: {
10760040077dSdrh       sqliteExprCode(pParse, pExpr->pLeft);
10770040077dSdrh       sqliteExprCode(pParse, pExpr->pRight);
107899fcd718Sdrh       sqliteVdbeAddOp(v, OP_Concat, 2, 0);
10790040077dSdrh       break;
10800040077dSdrh     }
10814b59ab5eSdrh     case TK_UPLUS: {
10824b59ab5eSdrh       Expr *pLeft = pExpr->pLeft;
10834b59ab5eSdrh       if( pLeft && pLeft->op==TK_INTEGER ){
10844b59ab5eSdrh         sqliteVdbeAddOp(v, OP_Integer, atoi(pLeft->token.z), 0);
10854b59ab5eSdrh         sqliteVdbeChangeP3(v, -1, pLeft->token.z, pLeft->token.n);
10864b59ab5eSdrh       }else if( pLeft && pLeft->op==TK_FLOAT ){
10874b59ab5eSdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
10884b59ab5eSdrh         sqliteVdbeChangeP3(v, -1, pLeft->token.z, pLeft->token.n);
10894b59ab5eSdrh       }else{
10904b59ab5eSdrh         sqliteExprCode(pParse, pExpr->pLeft);
10914b59ab5eSdrh       }
10924b59ab5eSdrh       break;
10934b59ab5eSdrh     }
1094cce7d176Sdrh     case TK_UMINUS: {
10956e142f54Sdrh       assert( pExpr->pLeft );
10967a7c7390Sdrh       if( pExpr->pLeft->op==TK_FLOAT || pExpr->pLeft->op==TK_INTEGER ){
10976e142f54Sdrh         Token *p = &pExpr->pLeft->token;
10986e142f54Sdrh         char *z = sqliteMalloc( p->n + 2 );
10996e142f54Sdrh         sprintf(z, "-%.*s", p->n, p->z);
1100e6840900Sdrh         if( pExpr->pLeft->op==TK_INTEGER ){
1101e6840900Sdrh           sqliteVdbeAddOp(v, OP_Integer, atoi(z), 0);
1102e6840900Sdrh         }else{
110399fcd718Sdrh           sqliteVdbeAddOp(v, OP_String, 0, 0);
1104e6840900Sdrh         }
110599fcd718Sdrh         sqliteVdbeChangeP3(v, -1, z, p->n+1);
11066e142f54Sdrh         sqliteFree(z);
11076e142f54Sdrh         break;
11086e142f54Sdrh       }
11091ccde15dSdrh       /* Fall through into TK_NOT */
11106e142f54Sdrh     }
1111bf4133cbSdrh     case TK_BITNOT:
11126e142f54Sdrh     case TK_NOT: {
1113cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
111499fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
1115cce7d176Sdrh       break;
1116cce7d176Sdrh     }
1117cce7d176Sdrh     case TK_ISNULL:
1118cce7d176Sdrh     case TK_NOTNULL: {
1119cce7d176Sdrh       int dest;
112099fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
1121cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1122cce7d176Sdrh       dest = sqliteVdbeCurrentAddr(v) + 2;
1123f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
112499fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
1125cce7d176Sdrh       break;
1126cce7d176Sdrh     }
11272282792aSdrh     case TK_AGG_FUNCTION: {
112899fcd718Sdrh       sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
11292282792aSdrh       break;
11302282792aSdrh     }
11314b59ab5eSdrh     case TK_GLOB:
11324b59ab5eSdrh     case TK_LIKE:
1133cce7d176Sdrh     case TK_FUNCTION: {
1134cce7d176Sdrh       int i;
1135cce7d176Sdrh       ExprList *pList = pExpr->pList;
113689425d5eSdrh       int nExpr = pList ? pList->nExpr : 0;
11370bce8354Sdrh       FuncDef *pDef;
11384b59ab5eSdrh       int nId;
11394b59ab5eSdrh       const char *zId;
11404b59ab5eSdrh       getFunctionName(pExpr, &zId, &nId);
11414b59ab5eSdrh       pDef = sqliteFindFunction(pParse->db, zId, nId, nExpr, 0);
11420bce8354Sdrh       assert( pDef!=0 );
114389425d5eSdrh       for(i=0; i<nExpr; i++){
11448e0a2f90Sdrh         sqliteExprCode(pParse, pList->a[i].pExpr);
11458e0a2f90Sdrh       }
114689425d5eSdrh       sqliteVdbeAddOp(v, OP_Function, nExpr, 0);
11470bce8354Sdrh       sqliteVdbeChangeP3(v, -1, (char*)pDef, P3_POINTER);
11486ec2733bSdrh       break;
11496ec2733bSdrh     }
115019a775c2Sdrh     case TK_SELECT: {
115199fcd718Sdrh       sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0);
115219a775c2Sdrh       break;
115319a775c2Sdrh     }
1154fef5208cSdrh     case TK_IN: {
1155fef5208cSdrh       int addr;
115699fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
1157fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1158fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
1159f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+4);
1160f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1161f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
1162f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, addr+6);
1163fef5208cSdrh       if( pExpr->pSelect ){
1164f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+6);
1165fef5208cSdrh       }else{
1166f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+6);
1167fef5208cSdrh       }
116899fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
1169fef5208cSdrh       break;
1170fef5208cSdrh     }
1171fef5208cSdrh     case TK_BETWEEN: {
1172f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1173f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1174f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1175f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, 0, 0);
1176f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pull, 1, 0);
1177f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1178f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, 0, 0);
1179f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_And, 0, 0);
1180fef5208cSdrh       break;
1181fef5208cSdrh     }
1182a2e00042Sdrh     case TK_AS: {
1183a2e00042Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1184a2e00042Sdrh       break;
1185a2e00042Sdrh     }
118617a7f8ddSdrh     case TK_CASE: {
118717a7f8ddSdrh       int expr_end_label;
1188f5905aa7Sdrh       int jumpInst;
1189f5905aa7Sdrh       int addr;
1190f5905aa7Sdrh       int nExpr;
119117a7f8ddSdrh       int i;
119217a7f8ddSdrh 
119317a7f8ddSdrh       assert(pExpr->pList);
119417a7f8ddSdrh       assert((pExpr->pList->nExpr % 2) == 0);
119517a7f8ddSdrh       assert(pExpr->pList->nExpr > 0);
1196f5905aa7Sdrh       nExpr = pExpr->pList->nExpr;
1197f5905aa7Sdrh       expr_end_label = sqliteVdbeMakeLabel(v);
119817a7f8ddSdrh       if( pExpr->pLeft ){
119917a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pLeft);
1200cce7d176Sdrh       }
1201f5905aa7Sdrh       for(i=0; i<nExpr; i=i+2){
120217a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i].pExpr);
120317a7f8ddSdrh         if( pExpr->pLeft ){
1204f5905aa7Sdrh           sqliteVdbeAddOp(v, OP_Dup, 1, 1);
1205f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_Ne, 1, 0);
1206f570f011Sdrh           sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1207f5905aa7Sdrh         }else{
1208f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_IfNot, 1, 0);
120917a7f8ddSdrh         }
121017a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i+1].pExpr);
1211f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Goto, 0, expr_end_label);
1212f5905aa7Sdrh         addr = sqliteVdbeCurrentAddr(v);
1213f5905aa7Sdrh         sqliteVdbeChangeP2(v, jumpInst, addr);
121417a7f8ddSdrh       }
1215f570f011Sdrh       if( pExpr->pLeft ){
1216f570f011Sdrh         sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1217f570f011Sdrh       }
121817a7f8ddSdrh       if( pExpr->pRight ){
121917a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pRight);
122017a7f8ddSdrh       }else{
1221f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
122217a7f8ddSdrh       }
1223f5905aa7Sdrh       sqliteVdbeResolveLabel(v, expr_end_label);
12246f34903eSdanielk1977       break;
12256f34903eSdanielk1977     }
12266f34903eSdanielk1977     case TK_RAISE: {
12276f34903eSdanielk1977       if( !pParse->trigStack ){
1228da93d238Sdrh         sqliteErrorMsg(pParse,
1229da93d238Sdrh                        "RAISE() may only be used within a trigger-program");
12306f34903eSdanielk1977         pParse->nErr++;
12316f34903eSdanielk1977 	return;
12326f34903eSdanielk1977       }
12336f34903eSdanielk1977       if( pExpr->iColumn == OE_Rollback ||
12346f34903eSdanielk1977 	  pExpr->iColumn == OE_Abort ||
12356f34903eSdanielk1977 	  pExpr->iColumn == OE_Fail ){
12366f34903eSdanielk1977 	  char * msg = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
12376f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Halt, SQLITE_CONSTRAINT, pExpr->iColumn);
12386f34903eSdanielk1977 	  sqliteDequote(msg);
12396f34903eSdanielk1977 	  sqliteVdbeChangeP3(v, -1, msg, 0);
12406f34903eSdanielk1977 	  sqliteFree(msg);
12416f34903eSdanielk1977       } else {
12426f34903eSdanielk1977 	  assert( pExpr->iColumn == OE_Ignore );
12436f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Goto, 0, pParse->trigStack->ignoreJump);
1244483750baSdrh 	  sqliteVdbeChangeP3(v, -1, "(IGNORE jump)", 0);
12456f34903eSdanielk1977       }
124617a7f8ddSdrh     }
124717a7f8ddSdrh     break;
124817a7f8ddSdrh   }
1249cce7d176Sdrh }
1250cce7d176Sdrh 
1251cce7d176Sdrh /*
1252cce7d176Sdrh ** Generate code for a boolean expression such that a jump is made
1253cce7d176Sdrh ** to the label "dest" if the expression is true but execution
1254cce7d176Sdrh ** continues straight thru if the expression is false.
1255f5905aa7Sdrh **
1256f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false), then
1257f5905aa7Sdrh ** take the jump if the jumpIfNull flag is true.
1258cce7d176Sdrh */
1259f5905aa7Sdrh void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1260cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1261cce7d176Sdrh   int op = 0;
1262daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1263cce7d176Sdrh   switch( pExpr->op ){
1264cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
1265cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
1266cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
1267cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
1268cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
1269cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
1270cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
1271cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
1272cce7d176Sdrh     default:  break;
1273cce7d176Sdrh   }
1274cce7d176Sdrh   switch( pExpr->op ){
1275cce7d176Sdrh     case TK_AND: {
1276cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1277f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, d2, !jumpIfNull);
1278f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1279cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1280cce7d176Sdrh       break;
1281cce7d176Sdrh     }
1282cce7d176Sdrh     case TK_OR: {
1283f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1284f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1285cce7d176Sdrh       break;
1286cce7d176Sdrh     }
1287cce7d176Sdrh     case TK_NOT: {
1288f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1289cce7d176Sdrh       break;
1290cce7d176Sdrh     }
1291cce7d176Sdrh     case TK_LT:
1292cce7d176Sdrh     case TK_LE:
1293cce7d176Sdrh     case TK_GT:
1294cce7d176Sdrh     case TK_GE:
1295cce7d176Sdrh     case TK_NE:
12960ac65892Sdrh     case TK_EQ: {
1297cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1298cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1299491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
1300c9b84a1fSdrh         op += 6;  /* Convert numeric opcodes to text opcodes */
1301c9b84a1fSdrh       }
1302f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1303cce7d176Sdrh       break;
1304cce7d176Sdrh     }
1305cce7d176Sdrh     case TK_ISNULL:
1306cce7d176Sdrh     case TK_NOTNULL: {
1307cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1308f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1309cce7d176Sdrh       break;
1310cce7d176Sdrh     }
1311fef5208cSdrh     case TK_IN: {
1312f5905aa7Sdrh       int addr;
1313cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1314f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1315f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1316f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1317f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1318fef5208cSdrh       if( pExpr->pSelect ){
131999fcd718Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest);
1320fef5208cSdrh       }else{
132199fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest);
1322fef5208cSdrh       }
1323fef5208cSdrh       break;
1324fef5208cSdrh     }
1325fef5208cSdrh     case TK_BETWEEN: {
1326f5905aa7Sdrh       int addr;
1327fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
132899fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1329fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1330f5905aa7Sdrh       addr = sqliteVdbeAddOp(v, OP_Lt, !jumpIfNull, 0);
1331fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1332f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, jumpIfNull, dest);
133399fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 0, 0);
1334f5905aa7Sdrh       sqliteVdbeChangeP2(v, addr, sqliteVdbeCurrentAddr(v));
133599fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1336fef5208cSdrh       break;
1337fef5208cSdrh     }
1338cce7d176Sdrh     default: {
1339cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1340f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_If, jumpIfNull, dest);
1341cce7d176Sdrh       break;
1342cce7d176Sdrh     }
1343cce7d176Sdrh   }
1344cce7d176Sdrh }
1345cce7d176Sdrh 
1346cce7d176Sdrh /*
134766b89c8fSdrh ** Generate code for a boolean expression such that a jump is made
1348cce7d176Sdrh ** to the label "dest" if the expression is false but execution
1349cce7d176Sdrh ** continues straight thru if the expression is true.
1350f5905aa7Sdrh **
1351f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false) then
1352f5905aa7Sdrh ** jump if jumpIfNull is true or fall through if jumpIfNull is false.
1353cce7d176Sdrh */
1354f5905aa7Sdrh void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1355cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1356cce7d176Sdrh   int op = 0;
1357daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1358cce7d176Sdrh   switch( pExpr->op ){
1359cce7d176Sdrh     case TK_LT:       op = OP_Ge;       break;
1360cce7d176Sdrh     case TK_LE:       op = OP_Gt;       break;
1361cce7d176Sdrh     case TK_GT:       op = OP_Le;       break;
1362cce7d176Sdrh     case TK_GE:       op = OP_Lt;       break;
1363cce7d176Sdrh     case TK_NE:       op = OP_Eq;       break;
1364cce7d176Sdrh     case TK_EQ:       op = OP_Ne;       break;
1365cce7d176Sdrh     case TK_ISNULL:   op = OP_NotNull;  break;
1366cce7d176Sdrh     case TK_NOTNULL:  op = OP_IsNull;   break;
1367cce7d176Sdrh     default:  break;
1368cce7d176Sdrh   }
1369cce7d176Sdrh   switch( pExpr->op ){
1370cce7d176Sdrh     case TK_AND: {
1371f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1372f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1373cce7d176Sdrh       break;
1374cce7d176Sdrh     }
1375cce7d176Sdrh     case TK_OR: {
1376cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1377f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, d2, !jumpIfNull);
1378f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1379cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1380cce7d176Sdrh       break;
1381cce7d176Sdrh     }
1382cce7d176Sdrh     case TK_NOT: {
1383f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1384cce7d176Sdrh       break;
1385cce7d176Sdrh     }
1386cce7d176Sdrh     case TK_LT:
1387cce7d176Sdrh     case TK_LE:
1388cce7d176Sdrh     case TK_GT:
1389cce7d176Sdrh     case TK_GE:
1390cce7d176Sdrh     case TK_NE:
1391cce7d176Sdrh     case TK_EQ: {
1392491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
13938f619ccdSdrh         /* Convert numeric comparison opcodes into text comparison opcodes.
13948f619ccdSdrh         ** This step depends on the fact that the text comparision opcodes are
13958f619ccdSdrh         ** always 6 greater than their corresponding numeric comparison
13968f619ccdSdrh         ** opcodes.
13978f619ccdSdrh         */
13988f619ccdSdrh         assert( OP_Eq+6 == OP_StrEq );
13998f619ccdSdrh         op += 6;
1400c9b84a1fSdrh       }
1401cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1402cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1403f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1404cce7d176Sdrh       break;
1405cce7d176Sdrh     }
1406cce7d176Sdrh     case TK_ISNULL:
1407cce7d176Sdrh     case TK_NOTNULL: {
1408cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1409f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1410cce7d176Sdrh       break;
1411cce7d176Sdrh     }
1412fef5208cSdrh     case TK_IN: {
1413f5905aa7Sdrh       int addr;
1414cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1415f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1416f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1417f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1418f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1419fef5208cSdrh       if( pExpr->pSelect ){
142099fcd718Sdrh         sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest);
1421fef5208cSdrh       }else{
142299fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest);
1423fef5208cSdrh       }
1424fef5208cSdrh       break;
1425fef5208cSdrh     }
1426fef5208cSdrh     case TK_BETWEEN: {
1427fef5208cSdrh       int addr;
1428fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
142999fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1430fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1431fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
1432f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, !jumpIfNull, addr+3);
143399fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
143499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, dest);
1435fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1436f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Gt, jumpIfNull, dest);
1437fef5208cSdrh       break;
1438fef5208cSdrh     }
1439cce7d176Sdrh     default: {
1440cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1441461c281aSdrh       sqliteVdbeAddOp(v, OP_IfNot, jumpIfNull, dest);
1442cce7d176Sdrh       break;
1443cce7d176Sdrh     }
1444cce7d176Sdrh   }
1445cce7d176Sdrh }
14462282792aSdrh 
14472282792aSdrh /*
14482282792aSdrh ** Do a deep comparison of two expression trees.  Return TRUE (non-zero)
14492282792aSdrh ** if they are identical and return FALSE if they differ in any way.
14502282792aSdrh */
1451d8bc7086Sdrh int sqliteExprCompare(Expr *pA, Expr *pB){
14522282792aSdrh   int i;
14532282792aSdrh   if( pA==0 ){
14542282792aSdrh     return pB==0;
14552282792aSdrh   }else if( pB==0 ){
14562282792aSdrh     return 0;
14572282792aSdrh   }
14582282792aSdrh   if( pA->op!=pB->op ) return 0;
1459d8bc7086Sdrh   if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
1460d8bc7086Sdrh   if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
14612282792aSdrh   if( pA->pList ){
14622282792aSdrh     if( pB->pList==0 ) return 0;
14632282792aSdrh     if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
14642282792aSdrh     for(i=0; i<pA->pList->nExpr; i++){
1465d8bc7086Sdrh       if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
14662282792aSdrh         return 0;
14672282792aSdrh       }
14682282792aSdrh     }
14692282792aSdrh   }else if( pB->pList ){
14702282792aSdrh     return 0;
14712282792aSdrh   }
14722282792aSdrh   if( pA->pSelect || pB->pSelect ) return 0;
14732f2c01e5Sdrh   if( pA->iTable!=pB->iTable || pA->iColumn!=pB->iColumn ) return 0;
14742282792aSdrh   if( pA->token.z ){
14752282792aSdrh     if( pB->token.z==0 ) return 0;
14766977fea8Sdrh     if( pB->token.n!=pA->token.n ) return 0;
14776977fea8Sdrh     if( sqliteStrNICmp(pA->token.z, pB->token.z, pB->token.n)!=0 ) return 0;
14782282792aSdrh   }
14792282792aSdrh   return 1;
14802282792aSdrh }
14812282792aSdrh 
14822282792aSdrh /*
14832282792aSdrh ** Add a new element to the pParse->aAgg[] array and return its index.
14842282792aSdrh */
14852282792aSdrh static int appendAggInfo(Parse *pParse){
14862282792aSdrh   if( (pParse->nAgg & 0x7)==0 ){
14872282792aSdrh     int amt = pParse->nAgg + 8;
14886d4abfbeSdrh     AggExpr *aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
14896d4abfbeSdrh     if( aAgg==0 ){
14902282792aSdrh       return -1;
14912282792aSdrh     }
14926d4abfbeSdrh     pParse->aAgg = aAgg;
14932282792aSdrh   }
14942282792aSdrh   memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
14952282792aSdrh   return pParse->nAgg++;
14962282792aSdrh }
14972282792aSdrh 
14982282792aSdrh /*
14992282792aSdrh ** Analyze the given expression looking for aggregate functions and
15002282792aSdrh ** for variables that need to be added to the pParse->aAgg[] array.
15012282792aSdrh ** Make additional entries to the pParse->aAgg[] array as necessary.
15022282792aSdrh **
15032282792aSdrh ** This routine should only be called after the expression has been
15042282792aSdrh ** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
15052282792aSdrh **
15062282792aSdrh ** If errors are seen, leave an error message in zErrMsg and return
15072282792aSdrh ** the number of errors.
15082282792aSdrh */
15092282792aSdrh int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
15102282792aSdrh   int i;
15112282792aSdrh   AggExpr *aAgg;
15122282792aSdrh   int nErr = 0;
15132282792aSdrh 
15142282792aSdrh   if( pExpr==0 ) return 0;
15152282792aSdrh   switch( pExpr->op ){
1516967e8b73Sdrh     case TK_COLUMN: {
15172282792aSdrh       aAgg = pParse->aAgg;
15182282792aSdrh       for(i=0; i<pParse->nAgg; i++){
15192282792aSdrh         if( aAgg[i].isAgg ) continue;
15202282792aSdrh         if( aAgg[i].pExpr->iTable==pExpr->iTable
1521967e8b73Sdrh          && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
15222282792aSdrh           break;
15232282792aSdrh         }
15242282792aSdrh       }
15252282792aSdrh       if( i>=pParse->nAgg ){
15262282792aSdrh         i = appendAggInfo(pParse);
15272282792aSdrh         if( i<0 ) return 1;
15282282792aSdrh         pParse->aAgg[i].isAgg = 0;
15292282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
15302282792aSdrh       }
1531aaf88729Sdrh       pExpr->iAgg = i;
15322282792aSdrh       break;
15332282792aSdrh     }
15342282792aSdrh     case TK_AGG_FUNCTION: {
15352282792aSdrh       aAgg = pParse->aAgg;
15362282792aSdrh       for(i=0; i<pParse->nAgg; i++){
15372282792aSdrh         if( !aAgg[i].isAgg ) continue;
1538d8bc7086Sdrh         if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
15392282792aSdrh           break;
15402282792aSdrh         }
15412282792aSdrh       }
15422282792aSdrh       if( i>=pParse->nAgg ){
15432282792aSdrh         i = appendAggInfo(pParse);
15442282792aSdrh         if( i<0 ) return 1;
15452282792aSdrh         pParse->aAgg[i].isAgg = 1;
15462282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
15470bce8354Sdrh         pParse->aAgg[i].pFunc = sqliteFindFunction(pParse->db,
15486977fea8Sdrh              pExpr->token.z, pExpr->token.n,
1549f55f25f0Sdrh              pExpr->pList ? pExpr->pList->nExpr : 0, 0);
15502282792aSdrh       }
15512282792aSdrh       pExpr->iAgg = i;
15522282792aSdrh       break;
15532282792aSdrh     }
15542282792aSdrh     default: {
15552282792aSdrh       if( pExpr->pLeft ){
15562282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
15572282792aSdrh       }
15582282792aSdrh       if( nErr==0 && pExpr->pRight ){
15592282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
15602282792aSdrh       }
15612282792aSdrh       if( nErr==0 && pExpr->pList ){
15622282792aSdrh         int n = pExpr->pList->nExpr;
15632282792aSdrh         int i;
15642282792aSdrh         for(i=0; nErr==0 && i<n; i++){
15652282792aSdrh           nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
15662282792aSdrh         }
15672282792aSdrh       }
15682282792aSdrh       break;
15692282792aSdrh     }
15702282792aSdrh   }
15712282792aSdrh   return nErr;
15722282792aSdrh }
15738e0a2f90Sdrh 
15748e0a2f90Sdrh /*
15758e0a2f90Sdrh ** Locate a user function given a name and a number of arguments.
15760bce8354Sdrh ** Return a pointer to the FuncDef structure that defines that
15778e0a2f90Sdrh ** function, or return NULL if the function does not exist.
15788e0a2f90Sdrh **
15790bce8354Sdrh ** If the createFlag argument is true, then a new (blank) FuncDef
15808e0a2f90Sdrh ** structure is created and liked into the "db" structure if a
15818e0a2f90Sdrh ** no matching function previously existed.  When createFlag is true
15828e0a2f90Sdrh ** and the nArg parameter is -1, then only a function that accepts
15838e0a2f90Sdrh ** any number of arguments will be returned.
15848e0a2f90Sdrh **
15858e0a2f90Sdrh ** If createFlag is false and nArg is -1, then the first valid
15868e0a2f90Sdrh ** function found is returned.  A function is valid if either xFunc
15878e0a2f90Sdrh ** or xStep is non-zero.
15888e0a2f90Sdrh */
15890bce8354Sdrh FuncDef *sqliteFindFunction(
15908e0a2f90Sdrh   sqlite *db,        /* An open database */
15918e0a2f90Sdrh   const char *zName, /* Name of the function.  Not null-terminated */
15928e0a2f90Sdrh   int nName,         /* Number of characters in the name */
15938e0a2f90Sdrh   int nArg,          /* Number of arguments.  -1 means any number */
15948e0a2f90Sdrh   int createFlag     /* Create new entry if true and does not otherwise exist */
15958e0a2f90Sdrh ){
15960bce8354Sdrh   FuncDef *pFirst, *p, *pMaybe;
15970bce8354Sdrh   pFirst = p = (FuncDef*)sqliteHashFind(&db->aFunc, zName, nName);
15981350b030Sdrh   if( p && !createFlag && nArg<0 ){
15998e0a2f90Sdrh     while( p && p->xFunc==0 && p->xStep==0 ){ p = p->pNext; }
16008e0a2f90Sdrh     return p;
16018e0a2f90Sdrh   }
16028e0a2f90Sdrh   pMaybe = 0;
16038e0a2f90Sdrh   while( p && p->nArg!=nArg ){
16048e0a2f90Sdrh     if( p->nArg<0 && !createFlag && (p->xFunc || p->xStep) ) pMaybe = p;
16058e0a2f90Sdrh     p = p->pNext;
16068e0a2f90Sdrh   }
16078e0a2f90Sdrh   if( p && !createFlag && p->xFunc==0 && p->xStep==0 ){
16088e0a2f90Sdrh     return 0;
16098e0a2f90Sdrh   }
16108e0a2f90Sdrh   if( p==0 && pMaybe ){
16118e0a2f90Sdrh     assert( createFlag==0 );
16128e0a2f90Sdrh     return pMaybe;
16138e0a2f90Sdrh   }
161489425d5eSdrh   if( p==0 && createFlag && (p = sqliteMalloc(sizeof(*p)))!=0 ){
16158e0a2f90Sdrh     p->nArg = nArg;
16168e0a2f90Sdrh     p->pNext = pFirst;
1617c9b84a1fSdrh     p->dataType = pFirst ? pFirst->dataType : SQLITE_NUMERIC;
16180bce8354Sdrh     sqliteHashInsert(&db->aFunc, zName, nName, (void*)p);
16198e0a2f90Sdrh   }
16208e0a2f90Sdrh   return p;
16218e0a2f90Sdrh }
1622