xref: /sqlite-3.40.0/src/expr.c (revision 3e7bc9ca)
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*3e7bc9caSdrh ** $Id: expr.c,v 1.109 2004/02/21 19:17:18 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 ){
294efc4754Sdrh     /* When malloc fails, we leak memory from pLeft and pRight */
30a76b5dfcSdrh     return 0;
31a76b5dfcSdrh   }
32a76b5dfcSdrh   pNew->op = op;
33a76b5dfcSdrh   pNew->pLeft = pLeft;
34a76b5dfcSdrh   pNew->pRight = pRight;
35a76b5dfcSdrh   if( pToken ){
364b59ab5eSdrh     assert( pToken->dyn==0 );
37a76b5dfcSdrh     pNew->token = *pToken;
386977fea8Sdrh     pNew->span = *pToken;
39a76b5dfcSdrh   }else{
404efc4754Sdrh     assert( pNew->token.dyn==0 );
414efc4754Sdrh     assert( pNew->token.z==0 );
424efc4754Sdrh     assert( pNew->token.n==0 );
436977fea8Sdrh     if( pLeft && pRight ){
446977fea8Sdrh       sqliteExprSpan(pNew, &pLeft->span, &pRight->span);
456977fea8Sdrh     }else{
466977fea8Sdrh       pNew->span = pNew->token;
476977fea8Sdrh     }
48a76b5dfcSdrh   }
49a76b5dfcSdrh   return pNew;
50a76b5dfcSdrh }
51a76b5dfcSdrh 
52a76b5dfcSdrh /*
536977fea8Sdrh ** Set the Expr.span field of the given expression to span all
54a76b5dfcSdrh ** text between the two given tokens.
55a76b5dfcSdrh */
56a76b5dfcSdrh void sqliteExprSpan(Expr *pExpr, Token *pLeft, Token *pRight){
574efc4754Sdrh   assert( pRight!=0 );
584efc4754Sdrh   assert( pLeft!=0 );
594efc4754Sdrh   /* Note: pExpr might be NULL due to a prior malloc failure */
604efc4754Sdrh   if( pExpr && pRight->z && pLeft->z ){
614b59ab5eSdrh     if( pLeft->dyn==0 && pRight->dyn==0 ){
626977fea8Sdrh       pExpr->span.z = pLeft->z;
636977fea8Sdrh       pExpr->span.n = pRight->n + Addr(pRight->z) - Addr(pLeft->z);
644b59ab5eSdrh     }else{
656977fea8Sdrh       pExpr->span.z = 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 ){
784efc4754Sdrh     /* sqliteExprListDelete(pList); // Leak pList when malloc fails */
79a76b5dfcSdrh     return 0;
80a76b5dfcSdrh   }
81a76b5dfcSdrh   pNew->op = TK_FUNCTION;
82a76b5dfcSdrh   pNew->pList = pList;
83a76b5dfcSdrh   if( pToken ){
844b59ab5eSdrh     assert( pToken->dyn==0 );
85a76b5dfcSdrh     pNew->token = *pToken;
86a76b5dfcSdrh   }else{
87a76b5dfcSdrh     pNew->token.z = 0;
88a76b5dfcSdrh   }
896977fea8Sdrh   pNew->span = pNew->token;
90a76b5dfcSdrh   return pNew;
91a76b5dfcSdrh }
92a76b5dfcSdrh 
93a76b5dfcSdrh /*
94a2e00042Sdrh ** Recursively delete an expression tree.
95a2e00042Sdrh */
96a2e00042Sdrh void sqliteExprDelete(Expr *p){
97a2e00042Sdrh   if( p==0 ) return;
984efc4754Sdrh   if( p->span.dyn ) sqliteFree((char*)p->span.z);
994efc4754Sdrh   if( p->token.dyn ) sqliteFree((char*)p->token.z);
1004efc4754Sdrh   sqliteExprDelete(p->pLeft);
1014efc4754Sdrh   sqliteExprDelete(p->pRight);
1024efc4754Sdrh   sqliteExprListDelete(p->pList);
1034efc4754Sdrh   sqliteSelectDelete(p->pSelect);
104a2e00042Sdrh   sqliteFree(p);
105a2e00042Sdrh }
106a2e00042Sdrh 
107a76b5dfcSdrh 
108a76b5dfcSdrh /*
109ff78bd2fSdrh ** The following group of routines make deep copies of expressions,
110ff78bd2fSdrh ** expression lists, ID lists, and select statements.  The copies can
111ff78bd2fSdrh ** be deleted (by being passed to their respective ...Delete() routines)
112ff78bd2fSdrh ** without effecting the originals.
113ff78bd2fSdrh **
114ad3cab52Sdrh ** The expression list, ID, and source lists return by sqliteExprListDup(),
115ad3cab52Sdrh ** sqliteIdListDup(), and sqliteSrcListDup() can not be further expanded
116ad3cab52Sdrh ** by subsequent calls to sqlite*ListAppend() routines.
117ff78bd2fSdrh **
118ad3cab52Sdrh ** Any tables that the SrcList might point to are not duplicated.
119ff78bd2fSdrh */
120ff78bd2fSdrh Expr *sqliteExprDup(Expr *p){
121ff78bd2fSdrh   Expr *pNew;
122ff78bd2fSdrh   if( p==0 ) return 0;
123fcb78a49Sdrh   pNew = sqliteMallocRaw( sizeof(*p) );
124ff78bd2fSdrh   if( pNew==0 ) return 0;
1253b167c75Sdrh   memcpy(pNew, p, sizeof(*pNew));
1266977fea8Sdrh   if( p->token.z!=0 ){
1274b59ab5eSdrh     pNew->token.z = sqliteStrDup(p->token.z);
1284b59ab5eSdrh     pNew->token.dyn = 1;
1294b59ab5eSdrh   }else{
1304efc4754Sdrh     assert( pNew->token.z==0 );
1314b59ab5eSdrh   }
1326977fea8Sdrh   pNew->span.z = 0;
133ff78bd2fSdrh   pNew->pLeft = sqliteExprDup(p->pLeft);
134ff78bd2fSdrh   pNew->pRight = sqliteExprDup(p->pRight);
135ff78bd2fSdrh   pNew->pList = sqliteExprListDup(p->pList);
136ff78bd2fSdrh   pNew->pSelect = sqliteSelectDup(p->pSelect);
137ff78bd2fSdrh   return pNew;
138ff78bd2fSdrh }
1394b59ab5eSdrh void sqliteTokenCopy(Token *pTo, Token *pFrom){
1404b59ab5eSdrh   if( pTo->dyn ) sqliteFree((char*)pTo->z);
1414b59ab5eSdrh   if( pFrom->z ){
1424b59ab5eSdrh     pTo->n = pFrom->n;
1434b59ab5eSdrh     pTo->z = sqliteStrNDup(pFrom->z, pFrom->n);
1444b59ab5eSdrh     pTo->dyn = 1;
1454b59ab5eSdrh   }else{
1464b59ab5eSdrh     pTo->z = 0;
1474b59ab5eSdrh   }
1484b59ab5eSdrh }
149ff78bd2fSdrh ExprList *sqliteExprListDup(ExprList *p){
150ff78bd2fSdrh   ExprList *pNew;
151*3e7bc9caSdrh   struct ExprList_item *pItem;
152ff78bd2fSdrh   int i;
153ff78bd2fSdrh   if( p==0 ) return 0;
154ff78bd2fSdrh   pNew = sqliteMalloc( sizeof(*pNew) );
155ff78bd2fSdrh   if( pNew==0 ) return 0;
1564305d103Sdrh   pNew->nExpr = pNew->nAlloc = p->nExpr;
157*3e7bc9caSdrh   pNew->a = pItem = sqliteMalloc( p->nExpr*sizeof(p->a[0]) );
158*3e7bc9caSdrh   for(i=0; pItem && i<p->nExpr; i++, pItem++){
1594b59ab5eSdrh     Expr *pNewExpr, *pOldExpr;
160*3e7bc9caSdrh     pItem->pExpr = pNewExpr = sqliteExprDup(pOldExpr = p->a[i].pExpr);
1616977fea8Sdrh     if( pOldExpr->span.z!=0 && pNewExpr ){
1626977fea8Sdrh       /* Always make a copy of the span for top-level expressions in the
1634b59ab5eSdrh       ** expression list.  The logic in SELECT processing that determines
1644b59ab5eSdrh       ** the names of columns in the result set needs this information */
1656977fea8Sdrh       sqliteTokenCopy(&pNewExpr->span, &pOldExpr->span);
1664b59ab5eSdrh     }
1671f3e905cSdrh     assert( pNewExpr==0 || pNewExpr->span.z!=0
1681f3e905cSdrh             || pOldExpr->span.z==0 || sqlite_malloc_failed );
169*3e7bc9caSdrh     pItem->zName = sqliteStrDup(p->a[i].zName);
170*3e7bc9caSdrh     pItem->sortOrder = p->a[i].sortOrder;
171*3e7bc9caSdrh     pItem->isAgg = p->a[i].isAgg;
172*3e7bc9caSdrh     pItem->done = 0;
173ff78bd2fSdrh   }
174ff78bd2fSdrh   return pNew;
175ff78bd2fSdrh }
176ad3cab52Sdrh SrcList *sqliteSrcListDup(SrcList *p){
177ad3cab52Sdrh   SrcList *pNew;
178ad3cab52Sdrh   int i;
179113088ecSdrh   int nByte;
180ad3cab52Sdrh   if( p==0 ) return 0;
181113088ecSdrh   nByte = sizeof(*p) + (p->nSrc>0 ? sizeof(p->a[0]) * (p->nSrc-1) : 0);
1824efc4754Sdrh   pNew = sqliteMallocRaw( nByte );
183ad3cab52Sdrh   if( pNew==0 ) return 0;
1844305d103Sdrh   pNew->nSrc = pNew->nAlloc = p->nSrc;
185ad3cab52Sdrh   for(i=0; i<p->nSrc; i++){
1864efc4754Sdrh     struct SrcList_item *pNewItem = &pNew->a[i];
1874efc4754Sdrh     struct SrcList_item *pOldItem = &p->a[i];
1884efc4754Sdrh     pNewItem->zDatabase = sqliteStrDup(pOldItem->zDatabase);
1894efc4754Sdrh     pNewItem->zName = sqliteStrDup(pOldItem->zName);
1904efc4754Sdrh     pNewItem->zAlias = sqliteStrDup(pOldItem->zAlias);
1914efc4754Sdrh     pNewItem->jointype = pOldItem->jointype;
1924efc4754Sdrh     pNewItem->iCursor = pOldItem->iCursor;
1934efc4754Sdrh     pNewItem->pTab = 0;
1944efc4754Sdrh     pNewItem->pSelect = sqliteSelectDup(pOldItem->pSelect);
1954efc4754Sdrh     pNewItem->pOn = sqliteExprDup(pOldItem->pOn);
1964efc4754Sdrh     pNewItem->pUsing = sqliteIdListDup(pOldItem->pUsing);
197ad3cab52Sdrh   }
198ad3cab52Sdrh   return pNew;
199ad3cab52Sdrh }
200ff78bd2fSdrh IdList *sqliteIdListDup(IdList *p){
201ff78bd2fSdrh   IdList *pNew;
202ff78bd2fSdrh   int i;
203ff78bd2fSdrh   if( p==0 ) return 0;
2044efc4754Sdrh   pNew = sqliteMallocRaw( sizeof(*pNew) );
205ff78bd2fSdrh   if( pNew==0 ) return 0;
2064305d103Sdrh   pNew->nId = pNew->nAlloc = p->nId;
2074efc4754Sdrh   pNew->a = sqliteMallocRaw( p->nId*sizeof(p->a[0]) );
208e4697f5eSdrh   if( pNew->a==0 ) return 0;
209ff78bd2fSdrh   for(i=0; i<p->nId; i++){
2104efc4754Sdrh     struct IdList_item *pNewItem = &pNew->a[i];
2114efc4754Sdrh     struct IdList_item *pOldItem = &p->a[i];
2124efc4754Sdrh     pNewItem->zName = sqliteStrDup(pOldItem->zName);
2134efc4754Sdrh     pNewItem->idx = pOldItem->idx;
214ff78bd2fSdrh   }
215ff78bd2fSdrh   return pNew;
216ff78bd2fSdrh }
217ff78bd2fSdrh Select *sqliteSelectDup(Select *p){
218ff78bd2fSdrh   Select *pNew;
219ff78bd2fSdrh   if( p==0 ) return 0;
2204efc4754Sdrh   pNew = sqliteMallocRaw( sizeof(*p) );
221ff78bd2fSdrh   if( pNew==0 ) return 0;
222ff78bd2fSdrh   pNew->isDistinct = p->isDistinct;
223ff78bd2fSdrh   pNew->pEList = sqliteExprListDup(p->pEList);
224ad3cab52Sdrh   pNew->pSrc = sqliteSrcListDup(p->pSrc);
225ff78bd2fSdrh   pNew->pWhere = sqliteExprDup(p->pWhere);
226ff78bd2fSdrh   pNew->pGroupBy = sqliteExprListDup(p->pGroupBy);
227ff78bd2fSdrh   pNew->pHaving = sqliteExprDup(p->pHaving);
228ff78bd2fSdrh   pNew->pOrderBy = sqliteExprListDup(p->pOrderBy);
229ff78bd2fSdrh   pNew->op = p->op;
230ff78bd2fSdrh   pNew->pPrior = sqliteSelectDup(p->pPrior);
231ff78bd2fSdrh   pNew->nLimit = p->nLimit;
232ff78bd2fSdrh   pNew->nOffset = p->nOffset;
233ff78bd2fSdrh   pNew->zSelect = 0;
2347b58daeaSdrh   pNew->iLimit = -1;
2357b58daeaSdrh   pNew->iOffset = -1;
236ff78bd2fSdrh   return pNew;
237ff78bd2fSdrh }
238ff78bd2fSdrh 
239ff78bd2fSdrh 
240ff78bd2fSdrh /*
241a76b5dfcSdrh ** Add a new element to the end of an expression list.  If pList is
242a76b5dfcSdrh ** initially NULL, then create a new expression list.
243a76b5dfcSdrh */
244a76b5dfcSdrh ExprList *sqliteExprListAppend(ExprList *pList, Expr *pExpr, Token *pName){
245a76b5dfcSdrh   if( pList==0 ){
246a76b5dfcSdrh     pList = sqliteMalloc( sizeof(ExprList) );
247a76b5dfcSdrh     if( pList==0 ){
2484efc4754Sdrh       /* sqliteExprDelete(pExpr); // Leak memory if malloc fails */
249a76b5dfcSdrh       return 0;
250a76b5dfcSdrh     }
2514efc4754Sdrh     assert( pList->nAlloc==0 );
252a76b5dfcSdrh   }
2534305d103Sdrh   if( pList->nAlloc<=pList->nExpr ){
2544305d103Sdrh     pList->nAlloc = pList->nAlloc*2 + 4;
2554efc4754Sdrh     pList->a = sqliteRealloc(pList->a, pList->nAlloc*sizeof(pList->a[0]));
2564efc4754Sdrh     if( pList->a==0 ){
2574efc4754Sdrh       /* sqliteExprDelete(pExpr); // Leak memory if malloc fails */
2584efc4754Sdrh       pList->nExpr = pList->nAlloc = 0;
259a76b5dfcSdrh       return pList;
260a76b5dfcSdrh     }
261a76b5dfcSdrh   }
2624efc4754Sdrh   assert( pList->a!=0 );
2634efc4754Sdrh   if( pExpr || pName ){
2644efc4754Sdrh     struct ExprList_item *pItem = &pList->a[pList->nExpr++];
2654efc4754Sdrh     memset(pItem, 0, sizeof(*pItem));
2664efc4754Sdrh     pItem->pExpr = pExpr;
267a76b5dfcSdrh     if( pName ){
2684efc4754Sdrh       sqliteSetNString(&pItem->zName, pName->z, pName->n, 0);
2694efc4754Sdrh       sqliteDequote(pItem->zName);
270a76b5dfcSdrh     }
271a76b5dfcSdrh   }
272a76b5dfcSdrh   return pList;
273a76b5dfcSdrh }
274a76b5dfcSdrh 
275a76b5dfcSdrh /*
276a76b5dfcSdrh ** Delete an entire expression list.
277a76b5dfcSdrh */
278a76b5dfcSdrh void sqliteExprListDelete(ExprList *pList){
279a76b5dfcSdrh   int i;
280a76b5dfcSdrh   if( pList==0 ) return;
281a76b5dfcSdrh   for(i=0; i<pList->nExpr; i++){
282a76b5dfcSdrh     sqliteExprDelete(pList->a[i].pExpr);
283a76b5dfcSdrh     sqliteFree(pList->a[i].zName);
284a76b5dfcSdrh   }
285a76b5dfcSdrh   sqliteFree(pList->a);
286a76b5dfcSdrh   sqliteFree(pList);
287a76b5dfcSdrh }
288a76b5dfcSdrh 
289a76b5dfcSdrh /*
290fef5208cSdrh ** Walk an expression tree.  Return 1 if the expression is constant
291fef5208cSdrh ** and 0 if it involves variables.
2922398937bSdrh **
2932398937bSdrh ** For the purposes of this function, a double-quoted string (ex: "abc")
2942398937bSdrh ** is considered a variable but a single-quoted string (ex: 'abc') is
2952398937bSdrh ** a constant.
296fef5208cSdrh */
2979208643dSdrh int sqliteExprIsConstant(Expr *p){
298fef5208cSdrh   switch( p->op ){
299fef5208cSdrh     case TK_ID:
300967e8b73Sdrh     case TK_COLUMN:
301fef5208cSdrh     case TK_DOT:
3027bdc0c1dSdrh     case TK_FUNCTION:
303fef5208cSdrh       return 0;
3047bdc0c1dSdrh     case TK_NULL:
3052398937bSdrh     case TK_STRING:
3069208643dSdrh     case TK_INTEGER:
3079208643dSdrh     case TK_FLOAT:
30850457896Sdrh     case TK_VARIABLE:
3099208643dSdrh       return 1;
310fef5208cSdrh     default: {
3119208643dSdrh       if( p->pLeft && !sqliteExprIsConstant(p->pLeft) ) return 0;
3129208643dSdrh       if( p->pRight && !sqliteExprIsConstant(p->pRight) ) return 0;
313fef5208cSdrh       if( p->pList ){
314fef5208cSdrh         int i;
315fef5208cSdrh         for(i=0; i<p->pList->nExpr; i++){
3169208643dSdrh           if( !sqliteExprIsConstant(p->pList->a[i].pExpr) ) return 0;
317fef5208cSdrh         }
318fef5208cSdrh       }
3199208643dSdrh       return p->pLeft!=0 || p->pRight!=0 || (p->pList && p->pList->nExpr>0);
320fef5208cSdrh     }
321fef5208cSdrh   }
3229208643dSdrh   return 0;
323fef5208cSdrh }
324fef5208cSdrh 
325fef5208cSdrh /*
326202b2df7Sdrh ** If the given expression codes a constant integer that is small enough
327202b2df7Sdrh ** to fit in a 32-bit integer, return 1 and put the value of the integer
328202b2df7Sdrh ** in *pValue.  If the expression is not an integer or if it is too big
329202b2df7Sdrh ** to fit in a signed 32-bit integer, return 0 and leave *pValue unchanged.
330e4de1febSdrh */
331e4de1febSdrh int sqliteExprIsInteger(Expr *p, int *pValue){
332e4de1febSdrh   switch( p->op ){
333e4de1febSdrh     case TK_INTEGER: {
334202b2df7Sdrh       if( sqliteFitsIn32Bits(p->token.z) ){
335e4de1febSdrh         *pValue = atoi(p->token.z);
336e4de1febSdrh         return 1;
337e4de1febSdrh       }
338202b2df7Sdrh       break;
339202b2df7Sdrh     }
340e4de1febSdrh     case TK_STRING: {
341bd790ee3Sdrh       const char *z = p->token.z;
342e4de1febSdrh       int n = p->token.n;
343bd790ee3Sdrh       if( n>0 && z[0]=='-' ){ z++; n--; }
344e4de1febSdrh       while( n>0 && *z && isdigit(*z) ){ z++; n--; }
345202b2df7Sdrh       if( n==0 && sqliteFitsIn32Bits(p->token.z) ){
346e4de1febSdrh         *pValue = atoi(p->token.z);
347e4de1febSdrh         return 1;
348e4de1febSdrh       }
349e4de1febSdrh       break;
350e4de1febSdrh     }
3514b59ab5eSdrh     case TK_UPLUS: {
3524b59ab5eSdrh       return sqliteExprIsInteger(p->pLeft, pValue);
3534b59ab5eSdrh     }
354e4de1febSdrh     case TK_UMINUS: {
355e4de1febSdrh       int v;
356e4de1febSdrh       if( sqliteExprIsInteger(p->pLeft, &v) ){
357e4de1febSdrh         *pValue = -v;
358e4de1febSdrh         return 1;
359e4de1febSdrh       }
360e4de1febSdrh       break;
361e4de1febSdrh     }
362e4de1febSdrh     default: break;
363e4de1febSdrh   }
364e4de1febSdrh   return 0;
365e4de1febSdrh }
366e4de1febSdrh 
367e4de1febSdrh /*
368c4a3c779Sdrh ** Return TRUE if the given string is a row-id column name.
369c4a3c779Sdrh */
370a9f9d1c0Sdrh int sqliteIsRowid(const char *z){
371c4a3c779Sdrh   if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
372c4a3c779Sdrh   if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
373c4a3c779Sdrh   if( sqliteStrICmp(z, "OID")==0 ) return 1;
374c4a3c779Sdrh   return 0;
375c4a3c779Sdrh }
376c4a3c779Sdrh 
377c4a3c779Sdrh /*
3788141f61eSdrh ** Given the name of a column of the form X.Y.Z or Y.Z or just Z, look up
3798141f61eSdrh ** that name in the set of source tables in pSrcList and make the pExpr
3808141f61eSdrh ** expression node refer back to that source column.  The following changes
3818141f61eSdrh ** are made to pExpr:
3828141f61eSdrh **
3838141f61eSdrh **    pExpr->iDb           Set the index in db->aDb[] of the database holding
3848141f61eSdrh **                         the table.
3858141f61eSdrh **    pExpr->iTable        Set to the cursor number for the table obtained
3868141f61eSdrh **                         from pSrcList.
3878141f61eSdrh **    pExpr->iColumn       Set to the column number within the table.
3888141f61eSdrh **    pExpr->dataType      Set to the appropriate data type for the column.
3898141f61eSdrh **    pExpr->op            Set to TK_COLUMN.
3908141f61eSdrh **    pExpr->pLeft         Any expression this points to is deleted
3918141f61eSdrh **    pExpr->pRight        Any expression this points to is deleted.
3928141f61eSdrh **
3938141f61eSdrh ** The pDbToken is the name of the database (the "X").  This value may be
3948141f61eSdrh ** NULL meaning that name is of the form Y.Z or Z.  Any available database
3958141f61eSdrh ** can be used.  The pTableToken is the name of the table (the "Y").  This
3968141f61eSdrh ** value can be NULL if pDbToken is also NULL.  If pTableToken is NULL it
3978141f61eSdrh ** means that the form of the name is Z and that columns from any table
3988141f61eSdrh ** can be used.
3998141f61eSdrh **
4008141f61eSdrh ** If the name cannot be resolved unambiguously, leave an error message
4018141f61eSdrh ** in pParse and return non-zero.  Return zero on success.
4028141f61eSdrh */
4038141f61eSdrh static int lookupName(
4048141f61eSdrh   Parse *pParse,      /* The parsing context */
4058141f61eSdrh   Token *pDbToken,     /* Name of the database containing table, or NULL */
4068141f61eSdrh   Token *pTableToken,  /* Name of table containing column, or NULL */
4078141f61eSdrh   Token *pColumnToken, /* Name of the column. */
4088141f61eSdrh   SrcList *pSrcList,   /* List of tables used to resolve column names */
4098141f61eSdrh   ExprList *pEList,    /* List of expressions used to resolve "AS" */
4108141f61eSdrh   Expr *pExpr          /* Make this EXPR node point to the selected column */
4118141f61eSdrh ){
4128141f61eSdrh   char *zDb = 0;       /* Name of the database.  The "X" in X.Y.Z */
4138141f61eSdrh   char *zTab = 0;      /* Name of the table.  The "Y" in X.Y.Z or Y.Z */
4148141f61eSdrh   char *zCol = 0;      /* Name of the column.  The "Z" */
4158141f61eSdrh   int i, j;            /* Loop counters */
4168141f61eSdrh   int cnt = 0;         /* Number of matching column names */
4178141f61eSdrh   int cntTab = 0;      /* Number of matching table names */
4187e26d750Sdrh   sqlite *db = pParse->db;  /* The database */
4198141f61eSdrh 
4208141f61eSdrh   assert( pColumnToken && pColumnToken->z ); /* The Z in X.Y.Z cannot be NULL */
4218141f61eSdrh   if( pDbToken && pDbToken->z ){
4228141f61eSdrh     zDb = sqliteStrNDup(pDbToken->z, pDbToken->n);
4238141f61eSdrh     sqliteDequote(zDb);
4248141f61eSdrh   }else{
4258141f61eSdrh     zDb = 0;
4268141f61eSdrh   }
4278141f61eSdrh   if( pTableToken && pTableToken->z ){
4288141f61eSdrh     zTab = sqliteStrNDup(pTableToken->z, pTableToken->n);
4298141f61eSdrh     sqliteDequote(zTab);
4308141f61eSdrh   }else{
4318141f61eSdrh     assert( zDb==0 );
4328141f61eSdrh     zTab = 0;
4338141f61eSdrh   }
4348141f61eSdrh   zCol = sqliteStrNDup(pColumnToken->z, pColumnToken->n);
4358141f61eSdrh   sqliteDequote(zCol);
4368141f61eSdrh   if( sqlite_malloc_failed ){
4378141f61eSdrh     return 1;  /* Leak memory (zDb and zTab) if malloc fails */
4388141f61eSdrh   }
4398141f61eSdrh   assert( zTab==0 || pEList==0 );
4408141f61eSdrh 
4418141f61eSdrh   pExpr->iTable = -1;
4428141f61eSdrh   for(i=0; i<pSrcList->nSrc; i++){
4438141f61eSdrh     struct SrcList_item *pItem = &pSrcList->a[i];
4448141f61eSdrh     Table *pTab = pItem->pTab;
4458141f61eSdrh     Column *pCol;
4468141f61eSdrh 
4478141f61eSdrh     if( pTab==0 ) continue;
4488141f61eSdrh     assert( pTab->nCol>0 );
4498141f61eSdrh     if( zTab ){
4508141f61eSdrh       if( pItem->zAlias ){
4518141f61eSdrh         char *zTabName = pItem->zAlias;
4528141f61eSdrh         if( sqliteStrICmp(zTabName, zTab)!=0 ) continue;
4538141f61eSdrh       }else{
4548141f61eSdrh         char *zTabName = pTab->zName;
4558141f61eSdrh         if( zTabName==0 || sqliteStrICmp(zTabName, zTab)!=0 ) continue;
4568141f61eSdrh         if( zDb!=0 && sqliteStrICmp(db->aDb[pTab->iDb].zName, zDb)!=0 ){
4578141f61eSdrh           continue;
4588141f61eSdrh         }
4598141f61eSdrh       }
4608141f61eSdrh     }
4618141f61eSdrh     if( 0==(cntTab++) ){
4628141f61eSdrh       pExpr->iTable = pItem->iCursor;
4638141f61eSdrh       pExpr->iDb = pTab->iDb;
4648141f61eSdrh     }
4658141f61eSdrh     for(j=0, pCol=pTab->aCol; j<pTab->nCol; j++, pCol++){
4668141f61eSdrh       if( sqliteStrICmp(pCol->zName, zCol)==0 ){
4678141f61eSdrh         cnt++;
4688141f61eSdrh         pExpr->iTable = pItem->iCursor;
4698141f61eSdrh         pExpr->iDb = pTab->iDb;
4708141f61eSdrh         /* Substitute the rowid (column -1) for the INTEGER PRIMARY KEY */
4718141f61eSdrh         pExpr->iColumn = j==pTab->iPKey ? -1 : j;
4728141f61eSdrh         pExpr->dataType = pCol->sortOrder & SQLITE_SO_TYPEMASK;
4738141f61eSdrh         break;
4748141f61eSdrh       }
4758141f61eSdrh     }
4768141f61eSdrh   }
4778141f61eSdrh 
4788141f61eSdrh   /* If we have not already resolved the name, then maybe
4798141f61eSdrh   ** it is a new.* or old.* trigger argument reference
4808141f61eSdrh   */
4818141f61eSdrh   if( zDb==0 && zTab!=0 && cnt==0 && pParse->trigStack!=0 ){
4828141f61eSdrh     TriggerStack *pTriggerStack = pParse->trigStack;
4838141f61eSdrh     Table *pTab = 0;
4848141f61eSdrh     if( pTriggerStack->newIdx != -1 && sqliteStrICmp("new", zTab) == 0 ){
4858141f61eSdrh       pExpr->iTable = pTriggerStack->newIdx;
4868141f61eSdrh       assert( pTriggerStack->pTab );
4878141f61eSdrh       pTab = pTriggerStack->pTab;
4888141f61eSdrh     }else if( pTriggerStack->oldIdx != -1 && sqliteStrICmp("old", zTab) == 0 ){
4898141f61eSdrh       pExpr->iTable = pTriggerStack->oldIdx;
4908141f61eSdrh       assert( pTriggerStack->pTab );
4918141f61eSdrh       pTab = pTriggerStack->pTab;
4928141f61eSdrh     }
4938141f61eSdrh 
4948141f61eSdrh     if( pTab ){
4958141f61eSdrh       int j;
4968141f61eSdrh       Column *pCol = pTab->aCol;
4978141f61eSdrh 
4988141f61eSdrh       pExpr->iDb = pTab->iDb;
4998141f61eSdrh       cntTab++;
5008141f61eSdrh       for(j=0; j < pTab->nCol; j++, pCol++) {
5018141f61eSdrh         if( sqliteStrICmp(pCol->zName, zCol)==0 ){
5028141f61eSdrh           cnt++;
5038141f61eSdrh           pExpr->iColumn = j==pTab->iPKey ? -1 : j;
5048141f61eSdrh           pExpr->dataType = pCol->sortOrder & SQLITE_SO_TYPEMASK;
5058141f61eSdrh           break;
5068141f61eSdrh         }
5078141f61eSdrh       }
5088141f61eSdrh     }
5098141f61eSdrh   }
5108141f61eSdrh 
5118141f61eSdrh   /*
5128141f61eSdrh   ** Perhaps the name is a reference to the ROWID
5138141f61eSdrh   */
5148141f61eSdrh   if( cnt==0 && cntTab==1 && sqliteIsRowid(zCol) ){
5158141f61eSdrh     cnt = 1;
5168141f61eSdrh     pExpr->iColumn = -1;
5178141f61eSdrh     pExpr->dataType = SQLITE_SO_NUM;
5188141f61eSdrh   }
5198141f61eSdrh 
5208141f61eSdrh   /*
5218141f61eSdrh   ** If the input is of the form Z (not Y.Z or X.Y.Z) then the name Z
5228141f61eSdrh   ** might refer to an result-set alias.  This happens, for example, when
5238141f61eSdrh   ** we are resolving names in the WHERE clause of the following command:
5248141f61eSdrh   **
5258141f61eSdrh   **     SELECT a+b AS x FROM table WHERE x<10;
5268141f61eSdrh   **
5278141f61eSdrh   ** In cases like this, replace pExpr with a copy of the expression that
5288141f61eSdrh   ** forms the result set entry ("a+b" in the example) and return immediately.
5298141f61eSdrh   ** Note that the expression in the result set should have already been
5308141f61eSdrh   ** resolved by the time the WHERE clause is resolved.
5318141f61eSdrh   */
5328141f61eSdrh   if( cnt==0 && pEList!=0 ){
5338141f61eSdrh     for(j=0; j<pEList->nExpr; j++){
5348141f61eSdrh       char *zAs = pEList->a[j].zName;
5358141f61eSdrh       if( zAs!=0 && sqliteStrICmp(zAs, zCol)==0 ){
5368141f61eSdrh         assert( pExpr->pLeft==0 && pExpr->pRight==0 );
5378141f61eSdrh         pExpr->op = TK_AS;
5388141f61eSdrh         pExpr->iColumn = j;
5398141f61eSdrh         pExpr->pLeft = sqliteExprDup(pEList->a[j].pExpr);
5408141f61eSdrh         sqliteFree(zCol);
5418141f61eSdrh         assert( zTab==0 && zDb==0 );
5428141f61eSdrh         return 0;
5438141f61eSdrh       }
5448141f61eSdrh     }
5458141f61eSdrh   }
5468141f61eSdrh 
5478141f61eSdrh   /*
5488141f61eSdrh   ** If X and Y are NULL (in other words if only the column name Z is
5498141f61eSdrh   ** supplied) and the value of Z is enclosed in double-quotes, then
5508141f61eSdrh   ** Z is a string literal if it doesn't match any column names.  In that
5518141f61eSdrh   ** case, we need to return right away and not make any changes to
5528141f61eSdrh   ** pExpr.
5538141f61eSdrh   */
5548141f61eSdrh   if( cnt==0 && zTab==0 && pColumnToken->z[0]=='"' ){
5558141f61eSdrh     sqliteFree(zCol);
5568141f61eSdrh     return 0;
5578141f61eSdrh   }
5588141f61eSdrh 
5598141f61eSdrh   /*
5608141f61eSdrh   ** cnt==0 means there was not match.  cnt>1 means there were two or
5618141f61eSdrh   ** more matches.  Either way, we have an error.
5628141f61eSdrh   */
5638141f61eSdrh   if( cnt!=1 ){
5648141f61eSdrh     char *z = 0;
5658141f61eSdrh     char *zErr;
5668141f61eSdrh     zErr = cnt==0 ? "no such column: %s" : "ambiguous column name: %s";
5678141f61eSdrh     if( zDb ){
5688141f61eSdrh       sqliteSetString(&z, zDb, ".", zTab, ".", zCol, 0);
5698141f61eSdrh     }else if( zTab ){
5708141f61eSdrh       sqliteSetString(&z, zTab, ".", zCol, 0);
5718141f61eSdrh     }else{
5728141f61eSdrh       z = sqliteStrDup(zCol);
5738141f61eSdrh     }
5748141f61eSdrh     sqliteErrorMsg(pParse, zErr, z);
5758141f61eSdrh     sqliteFree(z);
5768141f61eSdrh   }
5778141f61eSdrh 
5788141f61eSdrh   /* Clean up and return
5798141f61eSdrh   */
5808141f61eSdrh   sqliteFree(zDb);
5818141f61eSdrh   sqliteFree(zTab);
5828141f61eSdrh   sqliteFree(zCol);
5838141f61eSdrh   sqliteExprDelete(pExpr->pLeft);
5848141f61eSdrh   pExpr->pLeft = 0;
5858141f61eSdrh   sqliteExprDelete(pExpr->pRight);
5868141f61eSdrh   pExpr->pRight = 0;
5878141f61eSdrh   pExpr->op = TK_COLUMN;
5888141f61eSdrh   sqliteAuthRead(pParse, pExpr, pSrcList);
5898141f61eSdrh   return cnt!=1;
5908141f61eSdrh }
5918141f61eSdrh 
5928141f61eSdrh /*
593cce7d176Sdrh ** This routine walks an expression tree and resolves references to
594967e8b73Sdrh ** table columns.  Nodes of the form ID.ID or ID resolve into an
595aacc543eSdrh ** index to the table in the table list and a column offset.  The
596aacc543eSdrh ** Expr.opcode for such nodes is changed to TK_COLUMN.  The Expr.iTable
597aacc543eSdrh ** value is changed to the index of the referenced table in pTabList
598832508b7Sdrh ** plus the "base" value.  The base value will ultimately become the
599aacc543eSdrh ** VDBE cursor number for a cursor that is pointing into the referenced
600aacc543eSdrh ** table.  The Expr.iColumn value is changed to the index of the column
601aacc543eSdrh ** of the referenced table.  The Expr.iColumn value for the special
602aacc543eSdrh ** ROWID column is -1.  Any INTEGER PRIMARY KEY column is tried as an
603aacc543eSdrh ** alias for ROWID.
60419a775c2Sdrh **
605fef5208cSdrh ** We also check for instances of the IN operator.  IN comes in two
606fef5208cSdrh ** forms:
607fef5208cSdrh **
608fef5208cSdrh **           expr IN (exprlist)
609fef5208cSdrh ** and
610fef5208cSdrh **           expr IN (SELECT ...)
611fef5208cSdrh **
612fef5208cSdrh ** The first form is handled by creating a set holding the list
613fef5208cSdrh ** of allowed values.  The second form causes the SELECT to generate
614fef5208cSdrh ** a temporary table.
615fef5208cSdrh **
616fef5208cSdrh ** This routine also looks for scalar SELECTs that are part of an expression.
61719a775c2Sdrh ** If it finds any, it generates code to write the value of that select
61819a775c2Sdrh ** into a memory cell.
619cce7d176Sdrh **
620967e8b73Sdrh ** Unknown columns or tables provoke an error.  The function returns
621cce7d176Sdrh ** the number of errors seen and leaves an error message on pParse->zErrMsg.
622cce7d176Sdrh */
623a2e00042Sdrh int sqliteExprResolveIds(
624a2e00042Sdrh   Parse *pParse,     /* The parser context */
6258141f61eSdrh   SrcList *pSrcList, /* List of tables used to resolve column names */
626a2e00042Sdrh   ExprList *pEList,  /* List of expressions used to resolve "AS" */
627a2e00042Sdrh   Expr *pExpr        /* The expression to be analyzed. */
628a2e00042Sdrh ){
6296a3ea0e6Sdrh   int i;
6306a3ea0e6Sdrh 
6318141f61eSdrh   if( pExpr==0 || pSrcList==0 ) return 0;
6328141f61eSdrh   for(i=0; i<pSrcList->nSrc; i++){
6338141f61eSdrh     assert( pSrcList->a[i].iCursor>=0 && pSrcList->a[i].iCursor<pParse->nTab );
6346a3ea0e6Sdrh   }
635cce7d176Sdrh   switch( pExpr->op ){
6362398937bSdrh     /* Double-quoted strings (ex: "abc") are used as identifiers if
6372398937bSdrh     ** possible.  Otherwise they remain as strings.  Single-quoted
6382398937bSdrh     ** strings (ex: 'abc') are always string literals.
6392398937bSdrh     */
6402398937bSdrh     case TK_STRING: {
6412398937bSdrh       if( pExpr->token.z[0]=='\'' ) break;
6422398937bSdrh       /* Fall thru into the TK_ID case if this is a double-quoted string */
6432398937bSdrh     }
6448141f61eSdrh     /* A lone identifier is the name of a columnd.
645a2e00042Sdrh     */
646cce7d176Sdrh     case TK_ID: {
6478141f61eSdrh       if( lookupName(pParse, 0, 0, &pExpr->token, pSrcList, pEList, pExpr) ){
648cce7d176Sdrh         return 1;
649ed6c8671Sdrh       }
650cce7d176Sdrh       break;
651cce7d176Sdrh     }
652cce7d176Sdrh 
653d24cc427Sdrh     /* A table name and column name:     ID.ID
654d24cc427Sdrh     ** Or a database, table and column:  ID.ID.ID
655d24cc427Sdrh     */
656cce7d176Sdrh     case TK_DOT: {
6578141f61eSdrh       Token *pColumn;
6588141f61eSdrh       Token *pTable;
6598141f61eSdrh       Token *pDb;
6608141f61eSdrh       Expr *pRight;
661cce7d176Sdrh 
662cce7d176Sdrh       pRight = pExpr->pRight;
663d24cc427Sdrh       if( pRight->op==TK_ID ){
6648141f61eSdrh         pDb = 0;
6658141f61eSdrh         pTable = &pExpr->pLeft->token;
6668141f61eSdrh         pColumn = &pRight->token;
667d24cc427Sdrh       }else{
6688141f61eSdrh         assert( pRight->op==TK_DOT );
6698141f61eSdrh         pDb = &pExpr->pLeft->token;
6708141f61eSdrh         pTable = &pRight->pLeft->token;
6718141f61eSdrh         pColumn = &pRight->pRight->token;
672d24cc427Sdrh       }
6738141f61eSdrh       if( lookupName(pParse, pDb, pTable, pColumn, pSrcList, 0, pExpr) ){
674daffd0e5Sdrh         return 1;
675daffd0e5Sdrh       }
676cce7d176Sdrh       break;
677cce7d176Sdrh     }
678cce7d176Sdrh 
679fef5208cSdrh     case TK_IN: {
680d8bc7086Sdrh       Vdbe *v = sqliteGetVdbe(pParse);
681fef5208cSdrh       if( v==0 ) return 1;
6828141f61eSdrh       if( sqliteExprResolveIds(pParse, pSrcList, pEList, pExpr->pLeft) ){
683cfab11bcSdrh         return 1;
684cfab11bcSdrh       }
685fef5208cSdrh       if( pExpr->pSelect ){
686fef5208cSdrh         /* Case 1:     expr IN (SELECT ...)
687fef5208cSdrh         **
688fef5208cSdrh         ** Generate code to write the results of the select into a temporary
6894794b980Sdrh         ** table.  The cursor number of the temporary table has already
6904794b980Sdrh         ** been put in iTable by sqliteExprResolveInSelect().
691fef5208cSdrh         */
692832508b7Sdrh         pExpr->iTable = pParse->nTab++;
693c6b52df3Sdrh         sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 1);
694832508b7Sdrh         sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable, 0,0,0);
695fef5208cSdrh       }else if( pExpr->pList ){
696fef5208cSdrh         /* Case 2:     expr IN (exprlist)
697fef5208cSdrh         **
698fef5208cSdrh         ** Create a set to put the exprlist values in.  The Set id is stored
699fef5208cSdrh         ** in iTable.
700fef5208cSdrh         */
701fef5208cSdrh         int i, iSet;
702fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
703fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
7049208643dSdrh           if( !sqliteExprIsConstant(pE2) ){
705da93d238Sdrh             sqliteErrorMsg(pParse,
706da93d238Sdrh               "right-hand side of IN operator must be constant");
707fef5208cSdrh             return 1;
708fef5208cSdrh           }
7094794b980Sdrh           if( sqliteExprCheck(pParse, pE2, 0, 0) ){
7104794b980Sdrh             return 1;
7114794b980Sdrh           }
712fef5208cSdrh         }
713fef5208cSdrh         iSet = pExpr->iTable = pParse->nSet++;
714fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
715fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
716fef5208cSdrh           switch( pE2->op ){
717fef5208cSdrh             case TK_FLOAT:
718fef5208cSdrh             case TK_INTEGER:
719fef5208cSdrh             case TK_STRING: {
72099fcd718Sdrh               int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
721a76b5dfcSdrh               assert( pE2->token.z );
722fef5208cSdrh               sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
723fef5208cSdrh               sqliteVdbeDequoteP3(v, addr);
724fef5208cSdrh               break;
725fef5208cSdrh             }
726fef5208cSdrh             default: {
727fef5208cSdrh               sqliteExprCode(pParse, pE2);
72899fcd718Sdrh               sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
729fef5208cSdrh               break;
730fef5208cSdrh             }
731fef5208cSdrh           }
732fef5208cSdrh         }
733fef5208cSdrh       }
734cfab11bcSdrh       break;
735fef5208cSdrh     }
736fef5208cSdrh 
73719a775c2Sdrh     case TK_SELECT: {
738fef5208cSdrh       /* This has to be a scalar SELECT.  Generate code to put the
739fef5208cSdrh       ** value of this select in a memory cell and record the number
740967e8b73Sdrh       ** of the memory cell in iColumn.
741fef5208cSdrh       */
742967e8b73Sdrh       pExpr->iColumn = pParse->nMem++;
743832508b7Sdrh       if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn,0,0,0) ){
74419a775c2Sdrh         return 1;
74519a775c2Sdrh       }
74619a775c2Sdrh       break;
74719a775c2Sdrh     }
74819a775c2Sdrh 
749cce7d176Sdrh     /* For all else, just recursively walk the tree */
750cce7d176Sdrh     default: {
751cce7d176Sdrh       if( pExpr->pLeft
7528141f61eSdrh       && sqliteExprResolveIds(pParse, pSrcList, pEList, pExpr->pLeft) ){
753cce7d176Sdrh         return 1;
754cce7d176Sdrh       }
755cce7d176Sdrh       if( pExpr->pRight
7568141f61eSdrh       && sqliteExprResolveIds(pParse, pSrcList, pEList, pExpr->pRight) ){
757cce7d176Sdrh         return 1;
758cce7d176Sdrh       }
759cce7d176Sdrh       if( pExpr->pList ){
760cce7d176Sdrh         int i;
761cce7d176Sdrh         ExprList *pList = pExpr->pList;
762cce7d176Sdrh         for(i=0; i<pList->nExpr; i++){
763832508b7Sdrh           Expr *pArg = pList->a[i].pExpr;
7648141f61eSdrh           if( sqliteExprResolveIds(pParse, pSrcList, pEList, pArg) ){
765cce7d176Sdrh             return 1;
766cce7d176Sdrh           }
767cce7d176Sdrh         }
768cce7d176Sdrh       }
769cce7d176Sdrh     }
770cce7d176Sdrh   }
771cce7d176Sdrh   return 0;
772cce7d176Sdrh }
773cce7d176Sdrh 
774cce7d176Sdrh /*
7754b59ab5eSdrh ** pExpr is a node that defines a function of some kind.  It might
7764b59ab5eSdrh ** be a syntactic function like "count(x)" or it might be a function
7774b59ab5eSdrh ** that implements an operator, like "a LIKE b".
7784b59ab5eSdrh **
7794b59ab5eSdrh ** This routine makes *pzName point to the name of the function and
7804b59ab5eSdrh ** *pnName hold the number of characters in the function name.
7814b59ab5eSdrh */
7824b59ab5eSdrh static void getFunctionName(Expr *pExpr, const char **pzName, int *pnName){
7834b59ab5eSdrh   switch( pExpr->op ){
7844b59ab5eSdrh     case TK_FUNCTION: {
7854b59ab5eSdrh       *pzName = pExpr->token.z;
7866977fea8Sdrh       *pnName = pExpr->token.n;
7874b59ab5eSdrh       break;
7884b59ab5eSdrh     }
7894b59ab5eSdrh     case TK_LIKE: {
7904b59ab5eSdrh       *pzName = "like";
7914b59ab5eSdrh       *pnName = 4;
7924b59ab5eSdrh       break;
7934b59ab5eSdrh     }
7944b59ab5eSdrh     case TK_GLOB: {
7954b59ab5eSdrh       *pzName = "glob";
7964b59ab5eSdrh       *pnName = 4;
7974b59ab5eSdrh       break;
7984b59ab5eSdrh     }
7994b59ab5eSdrh     default: {
8004b59ab5eSdrh       *pzName = "can't happen";
8014b59ab5eSdrh       *pnName = 12;
8024b59ab5eSdrh       break;
8034b59ab5eSdrh     }
8044b59ab5eSdrh   }
8054b59ab5eSdrh }
8064b59ab5eSdrh 
8074b59ab5eSdrh /*
808cce7d176Sdrh ** Error check the functions in an expression.  Make sure all
809cce7d176Sdrh ** function names are recognized and all functions have the correct
810cce7d176Sdrh ** number of arguments.  Leave an error message in pParse->zErrMsg
811cce7d176Sdrh ** if anything is amiss.  Return the number of errors.
812cce7d176Sdrh **
813cce7d176Sdrh ** if pIsAgg is not null and this expression is an aggregate function
814cce7d176Sdrh ** (like count(*) or max(value)) then write a 1 into *pIsAgg.
815cce7d176Sdrh */
816cce7d176Sdrh int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
817cce7d176Sdrh   int nErr = 0;
818cce7d176Sdrh   if( pExpr==0 ) return 0;
819cce7d176Sdrh   switch( pExpr->op ){
8204b59ab5eSdrh     case TK_GLOB:
8214b59ab5eSdrh     case TK_LIKE:
822cce7d176Sdrh     case TK_FUNCTION: {
823c9b84a1fSdrh       int n = pExpr->pList ? pExpr->pList->nExpr : 0;  /* Number of arguments */
824c9b84a1fSdrh       int no_such_func = 0;       /* True if no such function exists */
825c9b84a1fSdrh       int is_type_of = 0;         /* True if is the special TypeOf() function */
826c9b84a1fSdrh       int wrong_num_args = 0;     /* True if wrong number of arguments */
827c9b84a1fSdrh       int is_agg = 0;             /* True if is an aggregate function */
828cce7d176Sdrh       int i;
8294b59ab5eSdrh       int nId;                    /* Number of characters in function name */
8304b59ab5eSdrh       const char *zId;            /* The function name. */
8310bce8354Sdrh       FuncDef *pDef;
8320bce8354Sdrh 
8334b59ab5eSdrh       getFunctionName(pExpr, &zId, &nId);
8344b59ab5eSdrh       pDef = sqliteFindFunction(pParse->db, zId, nId, n, 0);
8350bce8354Sdrh       if( pDef==0 ){
8364b59ab5eSdrh         pDef = sqliteFindFunction(pParse->db, zId, nId, -1, 0);
8370bce8354Sdrh         if( pDef==0 ){
8384b59ab5eSdrh           if( n==1 && nId==6 && sqliteStrNICmp(zId, "typeof", 6)==0 ){
839c9b84a1fSdrh             is_type_of = 1;
840c9b84a1fSdrh           }else {
841cce7d176Sdrh             no_such_func = 1;
842c9b84a1fSdrh           }
8438e0a2f90Sdrh         }else{
8448e0a2f90Sdrh           wrong_num_args = 1;
8458e0a2f90Sdrh         }
8468e0a2f90Sdrh       }else{
8470bce8354Sdrh         is_agg = pDef->xFunc==0;
848cce7d176Sdrh       }
8498e0a2f90Sdrh       if( is_agg && !allowAgg ){
8508e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg, "misuse of aggregate function ", -1,
8514b59ab5eSdrh            zId, nId, "()", 2, 0);
8528e0a2f90Sdrh         pParse->nErr++;
8538e0a2f90Sdrh         nErr++;
8548e0a2f90Sdrh         is_agg = 0;
8558e0a2f90Sdrh       }else if( no_such_func ){
8564b59ab5eSdrh         sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1, zId,nId,0);
857cce7d176Sdrh         pParse->nErr++;
858cce7d176Sdrh         nErr++;
8598e0a2f90Sdrh       }else if( wrong_num_args ){
8608e0a2f90Sdrh         sqliteSetNString(&pParse->zErrMsg,
8614b59ab5eSdrh            "wrong number of arguments to function ", -1, zId, nId, "()", 2, 0);
8628e0a2f90Sdrh         pParse->nErr++;
8638e0a2f90Sdrh         nErr++;
864cce7d176Sdrh       }
8652282792aSdrh       if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
866cce7d176Sdrh       if( is_agg && pIsAgg ) *pIsAgg = 1;
867cce7d176Sdrh       for(i=0; nErr==0 && i<n; i++){
8684cfa7934Sdrh         nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
8694cfa7934Sdrh                                allowAgg && !is_agg, pIsAgg);
870cce7d176Sdrh       }
871c9b84a1fSdrh       if( pDef==0 ){
872c9b84a1fSdrh         if( is_type_of ){
873c9b84a1fSdrh           pExpr->op = TK_STRING;
874c9b84a1fSdrh           if( sqliteExprType(pExpr->pList->a[0].pExpr)==SQLITE_SO_NUM ){
875c9b84a1fSdrh             pExpr->token.z = "numeric";
876c9b84a1fSdrh             pExpr->token.n = 7;
877c9b84a1fSdrh           }else{
878c9b84a1fSdrh             pExpr->token.z = "text";
879c9b84a1fSdrh             pExpr->token.n = 4;
880c9b84a1fSdrh           }
881c9b84a1fSdrh         }
882c9b84a1fSdrh       }else if( pDef->dataType>=0 ){
883c9b84a1fSdrh         if( pDef->dataType<n ){
884c9b84a1fSdrh           pExpr->dataType =
885c9b84a1fSdrh              sqliteExprType(pExpr->pList->a[pDef->dataType].pExpr);
886c9b84a1fSdrh         }else{
887c9b84a1fSdrh           pExpr->dataType = SQLITE_SO_NUM;
888c9b84a1fSdrh         }
889c9b84a1fSdrh       }else if( pDef->dataType==SQLITE_ARGS ){
890c9b84a1fSdrh         pDef->dataType = SQLITE_SO_TEXT;
891c9b84a1fSdrh         for(i=0; i<n; i++){
892c9b84a1fSdrh           if( sqliteExprType(pExpr->pList->a[i].pExpr)==SQLITE_SO_NUM ){
893c9b84a1fSdrh             pExpr->dataType = SQLITE_SO_NUM;
894c9b84a1fSdrh             break;
895c9b84a1fSdrh           }
896c9b84a1fSdrh         }
897c9b84a1fSdrh       }else if( pDef->dataType==SQLITE_NUMERIC ){
898c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_NUM;
899c9b84a1fSdrh       }else{
900c9b84a1fSdrh         pExpr->dataType = SQLITE_SO_TEXT;
901c9b84a1fSdrh       }
902cce7d176Sdrh     }
903cce7d176Sdrh     default: {
904cce7d176Sdrh       if( pExpr->pLeft ){
9052282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
906cce7d176Sdrh       }
907cce7d176Sdrh       if( nErr==0 && pExpr->pRight ){
9082282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
909cce7d176Sdrh       }
910fef5208cSdrh       if( nErr==0 && pExpr->pList ){
911fef5208cSdrh         int n = pExpr->pList->nExpr;
912fef5208cSdrh         int i;
913fef5208cSdrh         for(i=0; nErr==0 && i<n; i++){
9142282792aSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
9152282792aSdrh           nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
916fef5208cSdrh         }
917fef5208cSdrh       }
918cce7d176Sdrh       break;
919cce7d176Sdrh     }
920cce7d176Sdrh   }
921cce7d176Sdrh   return nErr;
922cce7d176Sdrh }
923cce7d176Sdrh 
924cce7d176Sdrh /*
925c9b84a1fSdrh ** Return either SQLITE_SO_NUM or SQLITE_SO_TEXT to indicate whether the
926c9b84a1fSdrh ** given expression should sort as numeric values or as text.
927c9b84a1fSdrh **
928c9b84a1fSdrh ** The sqliteExprResolveIds() and sqliteExprCheck() routines must have
929c9b84a1fSdrh ** both been called on the expression before it is passed to this routine.
930c9b84a1fSdrh */
931c9b84a1fSdrh int sqliteExprType(Expr *p){
932c9b84a1fSdrh   if( p==0 ) return SQLITE_SO_NUM;
933c9b84a1fSdrh   while( p ) switch( p->op ){
934c9b84a1fSdrh     case TK_PLUS:
935c9b84a1fSdrh     case TK_MINUS:
936c9b84a1fSdrh     case TK_STAR:
937c9b84a1fSdrh     case TK_SLASH:
938c9b84a1fSdrh     case TK_AND:
939c9b84a1fSdrh     case TK_OR:
940c9b84a1fSdrh     case TK_ISNULL:
941c9b84a1fSdrh     case TK_NOTNULL:
942c9b84a1fSdrh     case TK_NOT:
943c9b84a1fSdrh     case TK_UMINUS:
9444b59ab5eSdrh     case TK_UPLUS:
945c9b84a1fSdrh     case TK_BITAND:
946c9b84a1fSdrh     case TK_BITOR:
947c9b84a1fSdrh     case TK_BITNOT:
948c9b84a1fSdrh     case TK_LSHIFT:
949c9b84a1fSdrh     case TK_RSHIFT:
950c9b84a1fSdrh     case TK_REM:
951c9b84a1fSdrh     case TK_INTEGER:
952c9b84a1fSdrh     case TK_FLOAT:
953c9b84a1fSdrh     case TK_IN:
954c9b84a1fSdrh     case TK_BETWEEN:
9554b59ab5eSdrh     case TK_GLOB:
9564b59ab5eSdrh     case TK_LIKE:
957c9b84a1fSdrh       return SQLITE_SO_NUM;
958c9b84a1fSdrh 
959c9b84a1fSdrh     case TK_STRING:
960c9b84a1fSdrh     case TK_NULL:
961c9b84a1fSdrh     case TK_CONCAT:
96250457896Sdrh     case TK_VARIABLE:
963c9b84a1fSdrh       return SQLITE_SO_TEXT;
964c9b84a1fSdrh 
965c9b84a1fSdrh     case TK_LT:
966c9b84a1fSdrh     case TK_LE:
967c9b84a1fSdrh     case TK_GT:
968c9b84a1fSdrh     case TK_GE:
969c9b84a1fSdrh     case TK_NE:
970c9b84a1fSdrh     case TK_EQ:
971c9b84a1fSdrh       if( sqliteExprType(p->pLeft)==SQLITE_SO_NUM ){
972c9b84a1fSdrh         return SQLITE_SO_NUM;
973c9b84a1fSdrh       }
974c9b84a1fSdrh       p = p->pRight;
975c9b84a1fSdrh       break;
976c9b84a1fSdrh 
977c9b84a1fSdrh     case TK_AS:
978c9b84a1fSdrh       p = p->pLeft;
979c9b84a1fSdrh       break;
980c9b84a1fSdrh 
981c9b84a1fSdrh     case TK_COLUMN:
982c9b84a1fSdrh     case TK_FUNCTION:
983c9b84a1fSdrh     case TK_AGG_FUNCTION:
984c9b84a1fSdrh       return p->dataType;
985c9b84a1fSdrh 
986c9b84a1fSdrh     case TK_SELECT:
987c9b84a1fSdrh       assert( p->pSelect );
988c9b84a1fSdrh       assert( p->pSelect->pEList );
989c9b84a1fSdrh       assert( p->pSelect->pEList->nExpr>0 );
990c9b84a1fSdrh       p = p->pSelect->pEList->a[0].pExpr;
991c9b84a1fSdrh       break;
992c9b84a1fSdrh 
993b1363206Sdrh     case TK_CASE: {
994b1363206Sdrh       if( p->pRight && sqliteExprType(p->pRight)==SQLITE_SO_NUM ){
995b1363206Sdrh         return SQLITE_SO_NUM;
996b1363206Sdrh       }
997b1363206Sdrh       if( p->pList ){
998b1363206Sdrh         int i;
999b1363206Sdrh         ExprList *pList = p->pList;
1000b1363206Sdrh         for(i=1; i<pList->nExpr; i+=2){
1001b1363206Sdrh           if( sqliteExprType(pList->a[i].pExpr)==SQLITE_SO_NUM ){
1002b1363206Sdrh             return SQLITE_SO_NUM;
1003b1363206Sdrh           }
1004b1363206Sdrh         }
1005b1363206Sdrh       }
1006b1363206Sdrh       return SQLITE_SO_TEXT;
1007b1363206Sdrh     }
1008b1363206Sdrh 
1009c9b84a1fSdrh     default:
1010c9b84a1fSdrh       assert( p->op==TK_ABORT );  /* Can't Happen */
1011c9b84a1fSdrh       break;
1012c9b84a1fSdrh   }
1013c9b84a1fSdrh   return SQLITE_SO_NUM;
1014c9b84a1fSdrh }
1015c9b84a1fSdrh 
1016c9b84a1fSdrh /*
1017cce7d176Sdrh ** Generate code into the current Vdbe to evaluate the given
10181ccde15dSdrh ** expression and leave the result on the top of stack.
1019cce7d176Sdrh */
1020cce7d176Sdrh void sqliteExprCode(Parse *pParse, Expr *pExpr){
1021cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1022cce7d176Sdrh   int op;
1023daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1024cce7d176Sdrh   switch( pExpr->op ){
1025cce7d176Sdrh     case TK_PLUS:     op = OP_Add;      break;
1026cce7d176Sdrh     case TK_MINUS:    op = OP_Subtract; break;
1027cce7d176Sdrh     case TK_STAR:     op = OP_Multiply; break;
1028cce7d176Sdrh     case TK_SLASH:    op = OP_Divide;   break;
1029cce7d176Sdrh     case TK_AND:      op = OP_And;      break;
1030cce7d176Sdrh     case TK_OR:       op = OP_Or;       break;
1031cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
1032cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
1033cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
1034cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
1035cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
1036cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
1037cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
1038cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
1039cce7d176Sdrh     case TK_NOT:      op = OP_Not;      break;
1040cce7d176Sdrh     case TK_UMINUS:   op = OP_Negative; break;
1041bf4133cbSdrh     case TK_BITAND:   op = OP_BitAnd;   break;
1042bf4133cbSdrh     case TK_BITOR:    op = OP_BitOr;    break;
1043bf4133cbSdrh     case TK_BITNOT:   op = OP_BitNot;   break;
1044bf4133cbSdrh     case TK_LSHIFT:   op = OP_ShiftLeft;  break;
1045bf4133cbSdrh     case TK_RSHIFT:   op = OP_ShiftRight; break;
1046bf4133cbSdrh     case TK_REM:      op = OP_Remainder;  break;
1047cce7d176Sdrh     default: break;
1048cce7d176Sdrh   }
1049cce7d176Sdrh   switch( pExpr->op ){
1050967e8b73Sdrh     case TK_COLUMN: {
10512282792aSdrh       if( pParse->useAgg ){
105299fcd718Sdrh         sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
1053c4a3c779Sdrh       }else if( pExpr->iColumn>=0 ){
105499fcd718Sdrh         sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn);
1055c4a3c779Sdrh       }else{
105699fcd718Sdrh         sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0);
10572282792aSdrh       }
1058cce7d176Sdrh       break;
1059cce7d176Sdrh     }
106051e9a445Sdrh     case TK_STRING:
106151e9a445Sdrh     case TK_FLOAT:
1062cce7d176Sdrh     case TK_INTEGER: {
106351e9a445Sdrh       if( pExpr->op==TK_INTEGER && sqliteFitsIn32Bits(pExpr->token.z) ){
1064202b2df7Sdrh         sqliteVdbeAddOp(v, OP_Integer, atoi(pExpr->token.z), 0);
106551e9a445Sdrh       }else{
10667a7c7390Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
106751e9a445Sdrh       }
1068a76b5dfcSdrh       assert( pExpr->token.z );
10697a7c7390Sdrh       sqliteVdbeChangeP3(v, -1, pExpr->token.z, pExpr->token.n);
107051e9a445Sdrh       sqliteVdbeDequoteP3(v, -1);
1071cce7d176Sdrh       break;
1072cce7d176Sdrh     }
1073cce7d176Sdrh     case TK_NULL: {
107499fcd718Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
1075cce7d176Sdrh       break;
1076cce7d176Sdrh     }
107750457896Sdrh     case TK_VARIABLE: {
10787c972decSdrh       sqliteVdbeAddOp(v, OP_Variable, pExpr->iTable, 0);
107950457896Sdrh       break;
108050457896Sdrh     }
1081c9b84a1fSdrh     case TK_LT:
1082c9b84a1fSdrh     case TK_LE:
1083c9b84a1fSdrh     case TK_GT:
1084c9b84a1fSdrh     case TK_GE:
1085c9b84a1fSdrh     case TK_NE:
1086c9b84a1fSdrh     case TK_EQ: {
1087491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
1088c9b84a1fSdrh         op += 6;  /* Convert numeric opcodes to text opcodes */
1089c9b84a1fSdrh       }
1090c9b84a1fSdrh       /* Fall through into the next case */
1091c9b84a1fSdrh     }
1092cce7d176Sdrh     case TK_AND:
1093cce7d176Sdrh     case TK_OR:
1094cce7d176Sdrh     case TK_PLUS:
1095cce7d176Sdrh     case TK_STAR:
1096cce7d176Sdrh     case TK_MINUS:
1097bf4133cbSdrh     case TK_REM:
1098bf4133cbSdrh     case TK_BITAND:
1099bf4133cbSdrh     case TK_BITOR:
1100c9b84a1fSdrh     case TK_SLASH: {
1101cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1102cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
110399fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
1104cce7d176Sdrh       break;
1105cce7d176Sdrh     }
1106bf4133cbSdrh     case TK_LSHIFT:
1107bf4133cbSdrh     case TK_RSHIFT: {
1108bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pRight);
1109bf4133cbSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1110bf4133cbSdrh       sqliteVdbeAddOp(v, op, 0, 0);
1111bf4133cbSdrh       break;
1112bf4133cbSdrh     }
11130040077dSdrh     case TK_CONCAT: {
11140040077dSdrh       sqliteExprCode(pParse, pExpr->pLeft);
11150040077dSdrh       sqliteExprCode(pParse, pExpr->pRight);
111699fcd718Sdrh       sqliteVdbeAddOp(v, OP_Concat, 2, 0);
11170040077dSdrh       break;
11180040077dSdrh     }
1119cce7d176Sdrh     case TK_UMINUS: {
11206e142f54Sdrh       assert( pExpr->pLeft );
11217a7c7390Sdrh       if( pExpr->pLeft->op==TK_FLOAT || pExpr->pLeft->op==TK_INTEGER ){
11226e142f54Sdrh         Token *p = &pExpr->pLeft->token;
11236e142f54Sdrh         char *z = sqliteMalloc( p->n + 2 );
11246e142f54Sdrh         sprintf(z, "-%.*s", p->n, p->z);
1125202b2df7Sdrh         if( pExpr->pLeft->op==TK_INTEGER && sqliteFitsIn32Bits(z) ){
1126e6840900Sdrh           sqliteVdbeAddOp(v, OP_Integer, atoi(z), 0);
1127e6840900Sdrh         }else{
112899fcd718Sdrh           sqliteVdbeAddOp(v, OP_String, 0, 0);
1129e6840900Sdrh         }
113099fcd718Sdrh         sqliteVdbeChangeP3(v, -1, z, p->n+1);
11316e142f54Sdrh         sqliteFree(z);
11326e142f54Sdrh         break;
11336e142f54Sdrh       }
11341ccde15dSdrh       /* Fall through into TK_NOT */
11356e142f54Sdrh     }
1136bf4133cbSdrh     case TK_BITNOT:
11376e142f54Sdrh     case TK_NOT: {
1138cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
113999fcd718Sdrh       sqliteVdbeAddOp(v, op, 0, 0);
1140cce7d176Sdrh       break;
1141cce7d176Sdrh     }
1142cce7d176Sdrh     case TK_ISNULL:
1143cce7d176Sdrh     case TK_NOTNULL: {
1144cce7d176Sdrh       int dest;
114599fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
1146cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1147cce7d176Sdrh       dest = sqliteVdbeCurrentAddr(v) + 2;
1148f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
114999fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
1150cce7d176Sdrh       break;
1151cce7d176Sdrh     }
11522282792aSdrh     case TK_AGG_FUNCTION: {
115399fcd718Sdrh       sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
11542282792aSdrh       break;
11552282792aSdrh     }
11564b59ab5eSdrh     case TK_GLOB:
11574b59ab5eSdrh     case TK_LIKE:
1158cce7d176Sdrh     case TK_FUNCTION: {
1159cce7d176Sdrh       int i;
1160cce7d176Sdrh       ExprList *pList = pExpr->pList;
116189425d5eSdrh       int nExpr = pList ? pList->nExpr : 0;
11620bce8354Sdrh       FuncDef *pDef;
11634b59ab5eSdrh       int nId;
11644b59ab5eSdrh       const char *zId;
11654b59ab5eSdrh       getFunctionName(pExpr, &zId, &nId);
11664b59ab5eSdrh       pDef = sqliteFindFunction(pParse->db, zId, nId, nExpr, 0);
11670bce8354Sdrh       assert( pDef!=0 );
116889425d5eSdrh       for(i=0; i<nExpr; i++){
11698e0a2f90Sdrh         sqliteExprCode(pParse, pList->a[i].pExpr);
11708e0a2f90Sdrh       }
117189425d5eSdrh       sqliteVdbeAddOp(v, OP_Function, nExpr, 0);
11720bce8354Sdrh       sqliteVdbeChangeP3(v, -1, (char*)pDef, P3_POINTER);
11736ec2733bSdrh       break;
11746ec2733bSdrh     }
117519a775c2Sdrh     case TK_SELECT: {
117699fcd718Sdrh       sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0);
117719a775c2Sdrh       break;
117819a775c2Sdrh     }
1179fef5208cSdrh     case TK_IN: {
1180fef5208cSdrh       int addr;
118199fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0);
1182fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1183fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
1184f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+4);
1185f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1186f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_String, 0, 0);
1187f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, addr+6);
1188fef5208cSdrh       if( pExpr->pSelect ){
1189f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+6);
1190fef5208cSdrh       }else{
1191f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+6);
1192fef5208cSdrh       }
119399fcd718Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
1194fef5208cSdrh       break;
1195fef5208cSdrh     }
1196fef5208cSdrh     case TK_BETWEEN: {
1197f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1198f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1199f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1200f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, 0, 0);
1201f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pull, 1, 0);
1202f5905aa7Sdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1203f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, 0, 0);
1204f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_And, 0, 0);
1205fef5208cSdrh       break;
1206fef5208cSdrh     }
120751e9a445Sdrh     case TK_UPLUS:
1208a2e00042Sdrh     case TK_AS: {
1209a2e00042Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1210a2e00042Sdrh       break;
1211a2e00042Sdrh     }
121217a7f8ddSdrh     case TK_CASE: {
121317a7f8ddSdrh       int expr_end_label;
1214f5905aa7Sdrh       int jumpInst;
1215f5905aa7Sdrh       int addr;
1216f5905aa7Sdrh       int nExpr;
121717a7f8ddSdrh       int i;
121817a7f8ddSdrh 
121917a7f8ddSdrh       assert(pExpr->pList);
122017a7f8ddSdrh       assert((pExpr->pList->nExpr % 2) == 0);
122117a7f8ddSdrh       assert(pExpr->pList->nExpr > 0);
1222f5905aa7Sdrh       nExpr = pExpr->pList->nExpr;
1223f5905aa7Sdrh       expr_end_label = sqliteVdbeMakeLabel(v);
122417a7f8ddSdrh       if( pExpr->pLeft ){
122517a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pLeft);
1226cce7d176Sdrh       }
1227f5905aa7Sdrh       for(i=0; i<nExpr; i=i+2){
122817a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i].pExpr);
122917a7f8ddSdrh         if( pExpr->pLeft ){
1230f5905aa7Sdrh           sqliteVdbeAddOp(v, OP_Dup, 1, 1);
1231f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_Ne, 1, 0);
1232f570f011Sdrh           sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1233f5905aa7Sdrh         }else{
1234f570f011Sdrh           jumpInst = sqliteVdbeAddOp(v, OP_IfNot, 1, 0);
123517a7f8ddSdrh         }
123617a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pList->a[i+1].pExpr);
1237f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_Goto, 0, expr_end_label);
1238f5905aa7Sdrh         addr = sqliteVdbeCurrentAddr(v);
1239f5905aa7Sdrh         sqliteVdbeChangeP2(v, jumpInst, addr);
124017a7f8ddSdrh       }
1241f570f011Sdrh       if( pExpr->pLeft ){
1242f570f011Sdrh         sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1243f570f011Sdrh       }
124417a7f8ddSdrh       if( pExpr->pRight ){
124517a7f8ddSdrh         sqliteExprCode(pParse, pExpr->pRight);
124617a7f8ddSdrh       }else{
1247f5905aa7Sdrh         sqliteVdbeAddOp(v, OP_String, 0, 0);
124817a7f8ddSdrh       }
1249f5905aa7Sdrh       sqliteVdbeResolveLabel(v, expr_end_label);
12506f34903eSdanielk1977       break;
12516f34903eSdanielk1977     }
12526f34903eSdanielk1977     case TK_RAISE: {
12536f34903eSdanielk1977       if( !pParse->trigStack ){
1254da93d238Sdrh         sqliteErrorMsg(pParse,
1255da93d238Sdrh                        "RAISE() may only be used within a trigger-program");
12566f34903eSdanielk1977         pParse->nErr++;
12576f34903eSdanielk1977 	return;
12586f34903eSdanielk1977       }
12596f34903eSdanielk1977       if( pExpr->iColumn == OE_Rollback ||
12606f34903eSdanielk1977 	  pExpr->iColumn == OE_Abort ||
12616f34903eSdanielk1977 	  pExpr->iColumn == OE_Fail ){
12626f34903eSdanielk1977 	  char * msg = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
12636f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Halt, SQLITE_CONSTRAINT, pExpr->iColumn);
12646f34903eSdanielk1977 	  sqliteDequote(msg);
12656f34903eSdanielk1977 	  sqliteVdbeChangeP3(v, -1, msg, 0);
12666f34903eSdanielk1977 	  sqliteFree(msg);
12676f34903eSdanielk1977       } else {
12686f34903eSdanielk1977 	  assert( pExpr->iColumn == OE_Ignore );
12696f34903eSdanielk1977 	  sqliteVdbeAddOp(v, OP_Goto, 0, pParse->trigStack->ignoreJump);
1270483750baSdrh 	  sqliteVdbeChangeP3(v, -1, "(IGNORE jump)", 0);
12716f34903eSdanielk1977       }
127217a7f8ddSdrh     }
127317a7f8ddSdrh     break;
127417a7f8ddSdrh   }
1275cce7d176Sdrh }
1276cce7d176Sdrh 
1277cce7d176Sdrh /*
1278cce7d176Sdrh ** Generate code for a boolean expression such that a jump is made
1279cce7d176Sdrh ** to the label "dest" if the expression is true but execution
1280cce7d176Sdrh ** continues straight thru if the expression is false.
1281f5905aa7Sdrh **
1282f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false), then
1283f5905aa7Sdrh ** take the jump if the jumpIfNull flag is true.
1284cce7d176Sdrh */
1285f5905aa7Sdrh void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1286cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1287cce7d176Sdrh   int op = 0;
1288daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1289cce7d176Sdrh   switch( pExpr->op ){
1290cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
1291cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
1292cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
1293cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
1294cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
1295cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
1296cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
1297cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
1298cce7d176Sdrh     default:  break;
1299cce7d176Sdrh   }
1300cce7d176Sdrh   switch( pExpr->op ){
1301cce7d176Sdrh     case TK_AND: {
1302cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1303f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, d2, !jumpIfNull);
1304f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1305cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1306cce7d176Sdrh       break;
1307cce7d176Sdrh     }
1308cce7d176Sdrh     case TK_OR: {
1309f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1310f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest, jumpIfNull);
1311cce7d176Sdrh       break;
1312cce7d176Sdrh     }
1313cce7d176Sdrh     case TK_NOT: {
1314f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1315cce7d176Sdrh       break;
1316cce7d176Sdrh     }
1317cce7d176Sdrh     case TK_LT:
1318cce7d176Sdrh     case TK_LE:
1319cce7d176Sdrh     case TK_GT:
1320cce7d176Sdrh     case TK_GE:
1321cce7d176Sdrh     case TK_NE:
13220ac65892Sdrh     case TK_EQ: {
1323cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1324cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1325491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
1326c9b84a1fSdrh         op += 6;  /* Convert numeric opcodes to text opcodes */
1327c9b84a1fSdrh       }
1328f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1329cce7d176Sdrh       break;
1330cce7d176Sdrh     }
1331cce7d176Sdrh     case TK_ISNULL:
1332cce7d176Sdrh     case TK_NOTNULL: {
1333cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1334f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1335cce7d176Sdrh       break;
1336cce7d176Sdrh     }
1337fef5208cSdrh     case TK_IN: {
1338f5905aa7Sdrh       int addr;
1339cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1340f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1341f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1342f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1343f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1344fef5208cSdrh       if( pExpr->pSelect ){
134599fcd718Sdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest);
1346fef5208cSdrh       }else{
134799fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest);
1348fef5208cSdrh       }
1349fef5208cSdrh       break;
1350fef5208cSdrh     }
1351fef5208cSdrh     case TK_BETWEEN: {
1352f5905aa7Sdrh       int addr;
1353fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
135499fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1355fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1356f5905aa7Sdrh       addr = sqliteVdbeAddOp(v, OP_Lt, !jumpIfNull, 0);
1357fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1358f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Le, jumpIfNull, dest);
135999fcd718Sdrh       sqliteVdbeAddOp(v, OP_Integer, 0, 0);
1360f5905aa7Sdrh       sqliteVdbeChangeP2(v, addr, sqliteVdbeCurrentAddr(v));
136199fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1362fef5208cSdrh       break;
1363fef5208cSdrh     }
1364cce7d176Sdrh     default: {
1365cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1366f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_If, jumpIfNull, dest);
1367cce7d176Sdrh       break;
1368cce7d176Sdrh     }
1369cce7d176Sdrh   }
1370cce7d176Sdrh }
1371cce7d176Sdrh 
1372cce7d176Sdrh /*
137366b89c8fSdrh ** Generate code for a boolean expression such that a jump is made
1374cce7d176Sdrh ** to the label "dest" if the expression is false but execution
1375cce7d176Sdrh ** continues straight thru if the expression is true.
1376f5905aa7Sdrh **
1377f5905aa7Sdrh ** If the expression evaluates to NULL (neither true nor false) then
1378f5905aa7Sdrh ** jump if jumpIfNull is true or fall through if jumpIfNull is false.
1379cce7d176Sdrh */
1380f5905aa7Sdrh void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest, int jumpIfNull){
1381cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
1382cce7d176Sdrh   int op = 0;
1383daffd0e5Sdrh   if( v==0 || pExpr==0 ) return;
1384cce7d176Sdrh   switch( pExpr->op ){
1385cce7d176Sdrh     case TK_LT:       op = OP_Ge;       break;
1386cce7d176Sdrh     case TK_LE:       op = OP_Gt;       break;
1387cce7d176Sdrh     case TK_GT:       op = OP_Le;       break;
1388cce7d176Sdrh     case TK_GE:       op = OP_Lt;       break;
1389cce7d176Sdrh     case TK_NE:       op = OP_Eq;       break;
1390cce7d176Sdrh     case TK_EQ:       op = OP_Ne;       break;
1391cce7d176Sdrh     case TK_ISNULL:   op = OP_NotNull;  break;
1392cce7d176Sdrh     case TK_NOTNULL:  op = OP_IsNull;   break;
1393cce7d176Sdrh     default:  break;
1394cce7d176Sdrh   }
1395cce7d176Sdrh   switch( pExpr->op ){
1396cce7d176Sdrh     case TK_AND: {
1397f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest, jumpIfNull);
1398f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1399cce7d176Sdrh       break;
1400cce7d176Sdrh     }
1401cce7d176Sdrh     case TK_OR: {
1402cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
1403f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, d2, !jumpIfNull);
1404f5905aa7Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest, jumpIfNull);
1405cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
1406cce7d176Sdrh       break;
1407cce7d176Sdrh     }
1408cce7d176Sdrh     case TK_NOT: {
1409f5905aa7Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest, jumpIfNull);
1410cce7d176Sdrh       break;
1411cce7d176Sdrh     }
1412cce7d176Sdrh     case TK_LT:
1413cce7d176Sdrh     case TK_LE:
1414cce7d176Sdrh     case TK_GT:
1415cce7d176Sdrh     case TK_GE:
1416cce7d176Sdrh     case TK_NE:
1417cce7d176Sdrh     case TK_EQ: {
1418491791a8Sdrh       if( pParse->db->file_format>=4 && sqliteExprType(pExpr)==SQLITE_SO_TEXT ){
14198f619ccdSdrh         /* Convert numeric comparison opcodes into text comparison opcodes.
14208f619ccdSdrh         ** This step depends on the fact that the text comparision opcodes are
14218f619ccdSdrh         ** always 6 greater than their corresponding numeric comparison
14228f619ccdSdrh         ** opcodes.
14238f619ccdSdrh         */
14248f619ccdSdrh         assert( OP_Eq+6 == OP_StrEq );
14258f619ccdSdrh         op += 6;
1426c9b84a1fSdrh       }
1427cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1428cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
1429f5905aa7Sdrh       sqliteVdbeAddOp(v, op, jumpIfNull, dest);
1430cce7d176Sdrh       break;
1431cce7d176Sdrh     }
1432cce7d176Sdrh     case TK_ISNULL:
1433cce7d176Sdrh     case TK_NOTNULL: {
1434cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
1435f5905aa7Sdrh       sqliteVdbeAddOp(v, op, 1, dest);
1436cce7d176Sdrh       break;
1437cce7d176Sdrh     }
1438fef5208cSdrh     case TK_IN: {
1439f5905aa7Sdrh       int addr;
1440cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
1441f5905aa7Sdrh       addr = sqliteVdbeCurrentAddr(v);
1442f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_NotNull, -1, addr+3);
1443f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
1444f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, jumpIfNull ? dest : addr+4);
1445fef5208cSdrh       if( pExpr->pSelect ){
144699fcd718Sdrh         sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest);
1447fef5208cSdrh       }else{
144899fcd718Sdrh         sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest);
1449fef5208cSdrh       }
1450fef5208cSdrh       break;
1451fef5208cSdrh     }
1452fef5208cSdrh     case TK_BETWEEN: {
1453fef5208cSdrh       int addr;
1454fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
145599fcd718Sdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0);
1456fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
1457fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
1458f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Ge, !jumpIfNull, addr+3);
145999fcd718Sdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0);
146099fcd718Sdrh       sqliteVdbeAddOp(v, OP_Goto, 0, dest);
1461fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
1462f5905aa7Sdrh       sqliteVdbeAddOp(v, OP_Gt, jumpIfNull, dest);
1463fef5208cSdrh       break;
1464fef5208cSdrh     }
1465cce7d176Sdrh     default: {
1466cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
1467461c281aSdrh       sqliteVdbeAddOp(v, OP_IfNot, jumpIfNull, dest);
1468cce7d176Sdrh       break;
1469cce7d176Sdrh     }
1470cce7d176Sdrh   }
1471cce7d176Sdrh }
14722282792aSdrh 
14732282792aSdrh /*
14742282792aSdrh ** Do a deep comparison of two expression trees.  Return TRUE (non-zero)
14752282792aSdrh ** if they are identical and return FALSE if they differ in any way.
14762282792aSdrh */
1477d8bc7086Sdrh int sqliteExprCompare(Expr *pA, Expr *pB){
14782282792aSdrh   int i;
14792282792aSdrh   if( pA==0 ){
14802282792aSdrh     return pB==0;
14812282792aSdrh   }else if( pB==0 ){
14822282792aSdrh     return 0;
14832282792aSdrh   }
14842282792aSdrh   if( pA->op!=pB->op ) return 0;
1485d8bc7086Sdrh   if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
1486d8bc7086Sdrh   if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
14872282792aSdrh   if( pA->pList ){
14882282792aSdrh     if( pB->pList==0 ) return 0;
14892282792aSdrh     if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
14902282792aSdrh     for(i=0; i<pA->pList->nExpr; i++){
1491d8bc7086Sdrh       if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
14922282792aSdrh         return 0;
14932282792aSdrh       }
14942282792aSdrh     }
14952282792aSdrh   }else if( pB->pList ){
14962282792aSdrh     return 0;
14972282792aSdrh   }
14982282792aSdrh   if( pA->pSelect || pB->pSelect ) return 0;
14992f2c01e5Sdrh   if( pA->iTable!=pB->iTable || pA->iColumn!=pB->iColumn ) return 0;
15002282792aSdrh   if( pA->token.z ){
15012282792aSdrh     if( pB->token.z==0 ) return 0;
15026977fea8Sdrh     if( pB->token.n!=pA->token.n ) return 0;
15036977fea8Sdrh     if( sqliteStrNICmp(pA->token.z, pB->token.z, pB->token.n)!=0 ) return 0;
15042282792aSdrh   }
15052282792aSdrh   return 1;
15062282792aSdrh }
15072282792aSdrh 
15082282792aSdrh /*
15092282792aSdrh ** Add a new element to the pParse->aAgg[] array and return its index.
15102282792aSdrh */
15112282792aSdrh static int appendAggInfo(Parse *pParse){
15122282792aSdrh   if( (pParse->nAgg & 0x7)==0 ){
15132282792aSdrh     int amt = pParse->nAgg + 8;
15146d4abfbeSdrh     AggExpr *aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
15156d4abfbeSdrh     if( aAgg==0 ){
15162282792aSdrh       return -1;
15172282792aSdrh     }
15186d4abfbeSdrh     pParse->aAgg = aAgg;
15192282792aSdrh   }
15202282792aSdrh   memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
15212282792aSdrh   return pParse->nAgg++;
15222282792aSdrh }
15232282792aSdrh 
15242282792aSdrh /*
15252282792aSdrh ** Analyze the given expression looking for aggregate functions and
15262282792aSdrh ** for variables that need to be added to the pParse->aAgg[] array.
15272282792aSdrh ** Make additional entries to the pParse->aAgg[] array as necessary.
15282282792aSdrh **
15292282792aSdrh ** This routine should only be called after the expression has been
15302282792aSdrh ** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
15312282792aSdrh **
15322282792aSdrh ** If errors are seen, leave an error message in zErrMsg and return
15332282792aSdrh ** the number of errors.
15342282792aSdrh */
15352282792aSdrh int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
15362282792aSdrh   int i;
15372282792aSdrh   AggExpr *aAgg;
15382282792aSdrh   int nErr = 0;
15392282792aSdrh 
15402282792aSdrh   if( pExpr==0 ) return 0;
15412282792aSdrh   switch( pExpr->op ){
1542967e8b73Sdrh     case TK_COLUMN: {
15432282792aSdrh       aAgg = pParse->aAgg;
15442282792aSdrh       for(i=0; i<pParse->nAgg; i++){
15452282792aSdrh         if( aAgg[i].isAgg ) continue;
15462282792aSdrh         if( aAgg[i].pExpr->iTable==pExpr->iTable
1547967e8b73Sdrh          && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
15482282792aSdrh           break;
15492282792aSdrh         }
15502282792aSdrh       }
15512282792aSdrh       if( i>=pParse->nAgg ){
15522282792aSdrh         i = appendAggInfo(pParse);
15532282792aSdrh         if( i<0 ) return 1;
15542282792aSdrh         pParse->aAgg[i].isAgg = 0;
15552282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
15562282792aSdrh       }
1557aaf88729Sdrh       pExpr->iAgg = i;
15582282792aSdrh       break;
15592282792aSdrh     }
15602282792aSdrh     case TK_AGG_FUNCTION: {
15612282792aSdrh       aAgg = pParse->aAgg;
15622282792aSdrh       for(i=0; i<pParse->nAgg; i++){
15632282792aSdrh         if( !aAgg[i].isAgg ) continue;
1564d8bc7086Sdrh         if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
15652282792aSdrh           break;
15662282792aSdrh         }
15672282792aSdrh       }
15682282792aSdrh       if( i>=pParse->nAgg ){
15692282792aSdrh         i = appendAggInfo(pParse);
15702282792aSdrh         if( i<0 ) return 1;
15712282792aSdrh         pParse->aAgg[i].isAgg = 1;
15722282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
15730bce8354Sdrh         pParse->aAgg[i].pFunc = sqliteFindFunction(pParse->db,
15746977fea8Sdrh              pExpr->token.z, pExpr->token.n,
1575f55f25f0Sdrh              pExpr->pList ? pExpr->pList->nExpr : 0, 0);
15762282792aSdrh       }
15772282792aSdrh       pExpr->iAgg = i;
15782282792aSdrh       break;
15792282792aSdrh     }
15802282792aSdrh     default: {
15812282792aSdrh       if( pExpr->pLeft ){
15822282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
15832282792aSdrh       }
15842282792aSdrh       if( nErr==0 && pExpr->pRight ){
15852282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
15862282792aSdrh       }
15872282792aSdrh       if( nErr==0 && pExpr->pList ){
15882282792aSdrh         int n = pExpr->pList->nExpr;
15892282792aSdrh         int i;
15902282792aSdrh         for(i=0; nErr==0 && i<n; i++){
15912282792aSdrh           nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
15922282792aSdrh         }
15932282792aSdrh       }
15942282792aSdrh       break;
15952282792aSdrh     }
15962282792aSdrh   }
15972282792aSdrh   return nErr;
15982282792aSdrh }
15998e0a2f90Sdrh 
16008e0a2f90Sdrh /*
16018e0a2f90Sdrh ** Locate a user function given a name and a number of arguments.
16020bce8354Sdrh ** Return a pointer to the FuncDef structure that defines that
16038e0a2f90Sdrh ** function, or return NULL if the function does not exist.
16048e0a2f90Sdrh **
16050bce8354Sdrh ** If the createFlag argument is true, then a new (blank) FuncDef
16068e0a2f90Sdrh ** structure is created and liked into the "db" structure if a
16078e0a2f90Sdrh ** no matching function previously existed.  When createFlag is true
16088e0a2f90Sdrh ** and the nArg parameter is -1, then only a function that accepts
16098e0a2f90Sdrh ** any number of arguments will be returned.
16108e0a2f90Sdrh **
16118e0a2f90Sdrh ** If createFlag is false and nArg is -1, then the first valid
16128e0a2f90Sdrh ** function found is returned.  A function is valid if either xFunc
16138e0a2f90Sdrh ** or xStep is non-zero.
16148e0a2f90Sdrh */
16150bce8354Sdrh FuncDef *sqliteFindFunction(
16168e0a2f90Sdrh   sqlite *db,        /* An open database */
16178e0a2f90Sdrh   const char *zName, /* Name of the function.  Not null-terminated */
16188e0a2f90Sdrh   int nName,         /* Number of characters in the name */
16198e0a2f90Sdrh   int nArg,          /* Number of arguments.  -1 means any number */
16208e0a2f90Sdrh   int createFlag     /* Create new entry if true and does not otherwise exist */
16218e0a2f90Sdrh ){
16220bce8354Sdrh   FuncDef *pFirst, *p, *pMaybe;
16230bce8354Sdrh   pFirst = p = (FuncDef*)sqliteHashFind(&db->aFunc, zName, nName);
16241350b030Sdrh   if( p && !createFlag && nArg<0 ){
16258e0a2f90Sdrh     while( p && p->xFunc==0 && p->xStep==0 ){ p = p->pNext; }
16268e0a2f90Sdrh     return p;
16278e0a2f90Sdrh   }
16288e0a2f90Sdrh   pMaybe = 0;
16298e0a2f90Sdrh   while( p && p->nArg!=nArg ){
16308e0a2f90Sdrh     if( p->nArg<0 && !createFlag && (p->xFunc || p->xStep) ) pMaybe = p;
16318e0a2f90Sdrh     p = p->pNext;
16328e0a2f90Sdrh   }
16338e0a2f90Sdrh   if( p && !createFlag && p->xFunc==0 && p->xStep==0 ){
16348e0a2f90Sdrh     return 0;
16358e0a2f90Sdrh   }
16368e0a2f90Sdrh   if( p==0 && pMaybe ){
16378e0a2f90Sdrh     assert( createFlag==0 );
16388e0a2f90Sdrh     return pMaybe;
16398e0a2f90Sdrh   }
164089425d5eSdrh   if( p==0 && createFlag && (p = sqliteMalloc(sizeof(*p)))!=0 ){
16418e0a2f90Sdrh     p->nArg = nArg;
16428e0a2f90Sdrh     p->pNext = pFirst;
1643c9b84a1fSdrh     p->dataType = pFirst ? pFirst->dataType : SQLITE_NUMERIC;
16440bce8354Sdrh     sqliteHashInsert(&db->aFunc, zName, nName, (void*)p);
16458e0a2f90Sdrh   }
16468e0a2f90Sdrh   return p;
16478e0a2f90Sdrh }
1648