xref: /sqlite-3.40.0/src/expr.c (revision d8bc7086)
1cce7d176Sdrh /*
2cce7d176Sdrh ** Copyright (c) 1999, 2000 D. Richard Hipp
3cce7d176Sdrh **
4cce7d176Sdrh ** This program is free software; you can redistribute it and/or
5cce7d176Sdrh ** modify it under the terms of the GNU General Public
6cce7d176Sdrh ** License as published by the Free Software Foundation; either
7cce7d176Sdrh ** version 2 of the License, or (at your option) any later version.
8cce7d176Sdrh **
9cce7d176Sdrh ** This program is distributed in the hope that it will be useful,
10cce7d176Sdrh ** but WITHOUT ANY WARRANTY; without even the implied warranty of
11cce7d176Sdrh ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12cce7d176Sdrh ** General Public License for more details.
13cce7d176Sdrh **
14cce7d176Sdrh ** You should have received a copy of the GNU General Public
15cce7d176Sdrh ** License along with this library; if not, write to the
16cce7d176Sdrh ** Free Software Foundation, Inc., 59 Temple Place - Suite 330,
17cce7d176Sdrh ** Boston, MA  02111-1307, USA.
18cce7d176Sdrh **
19cce7d176Sdrh ** Author contact information:
20cce7d176Sdrh **   [email protected]
21cce7d176Sdrh **   http://www.hwaci.com/drh/
22cce7d176Sdrh **
23cce7d176Sdrh *************************************************************************
24cce7d176Sdrh ** This file contains C code routines used for processing expressions
25cce7d176Sdrh **
26*d8bc7086Sdrh ** $Id: expr.c,v 1.11 2000/06/07 23:51:50 drh Exp $
27cce7d176Sdrh */
28cce7d176Sdrh #include "sqliteInt.h"
29cce7d176Sdrh 
30cce7d176Sdrh /*
31fef5208cSdrh ** Walk an expression tree.  Return 1 if the expression is constant
32fef5208cSdrh ** and 0 if it involves variables.
33fef5208cSdrh */
34fef5208cSdrh static int isConstant(Expr *p){
35fef5208cSdrh   switch( p->op ){
36fef5208cSdrh     case TK_ID:
37fef5208cSdrh     case TK_FIELD:
38fef5208cSdrh     case TK_DOT:
39fef5208cSdrh       return 0;
40fef5208cSdrh     default: {
41fef5208cSdrh       if( p->pLeft && !isConstant(p->pLeft) ) return 0;
42fef5208cSdrh       if( p->pRight && !isConstant(p->pRight) ) return 0;
43fef5208cSdrh       if( p->pList ){
44fef5208cSdrh         int i;
45fef5208cSdrh         for(i=0; i<p->pList->nExpr; i++){
46fef5208cSdrh           if( !isConstant(p->pList->a[i].pExpr) ) return 0;
47fef5208cSdrh         }
48fef5208cSdrh       }
49fef5208cSdrh       break;
50fef5208cSdrh     }
51fef5208cSdrh   }
52fef5208cSdrh   return 1;
53fef5208cSdrh }
54fef5208cSdrh 
55fef5208cSdrh /*
564794b980Sdrh ** Walk the expression tree and process operators of the form:
574794b980Sdrh **
584794b980Sdrh **       expr IN (SELECT ...)
594794b980Sdrh **
604794b980Sdrh ** These operators have to be processed before field names are
614794b980Sdrh ** resolved because each such operator increments pParse->nTab
624794b980Sdrh ** to reserve a cursor number for its own use.  But pParse->nTab
634794b980Sdrh ** needs to be constant once we begin resolving field names.
644794b980Sdrh **
654794b980Sdrh ** Actually, the processing of IN-SELECT is only started by this
664794b980Sdrh ** routine.  This routine allocates a cursor number to the IN-SELECT
674794b980Sdrh ** and then moves on.  The code generation is done by
684794b980Sdrh ** sqliteExprResolveIds() which must be called afterwards.
694794b980Sdrh */
704794b980Sdrh void sqliteExprResolveInSelect(Parse *pParse, Expr *pExpr){
714794b980Sdrh   if( pExpr==0 ) return;
724794b980Sdrh   if( pExpr->op==TK_IN && pExpr->pSelect!=0 ){
734794b980Sdrh     pExpr->iTable = pParse->nTab++;
744794b980Sdrh   }else{
754794b980Sdrh     if( pExpr->pLeft ) sqliteExprResolveInSelect(pParse, pExpr->pLeft);
764794b980Sdrh     if( pExpr->pRight ) sqliteExprResolveInSelect(pParse, pExpr->pRight);
774794b980Sdrh     if( pExpr->pList ){
784794b980Sdrh       int i;
794794b980Sdrh       ExprList *pList = pExpr->pList;
804794b980Sdrh       for(i=0; i<pList->nExpr; i++){
814794b980Sdrh         sqliteExprResolveInSelect(pParse, pList->a[i].pExpr);
824794b980Sdrh       }
834794b980Sdrh     }
844794b980Sdrh   }
854794b980Sdrh }
864794b980Sdrh 
874794b980Sdrh /*
88cce7d176Sdrh ** This routine walks an expression tree and resolves references to
89cce7d176Sdrh ** table fields.  Nodes of the form ID.ID or ID resolve into an
90cce7d176Sdrh ** index to the table in the table list and a field offset.  The opcode
91cce7d176Sdrh ** for such nodes is changed to TK_FIELD.  The iTable value is changed
9219a775c2Sdrh ** to the index of the referenced table in pTabList plus the pParse->nTab
9319a775c2Sdrh ** value.  The iField value is changed to the index of the field of the
9419a775c2Sdrh ** referenced table.
9519a775c2Sdrh **
96fef5208cSdrh ** We also check for instances of the IN operator.  IN comes in two
97fef5208cSdrh ** forms:
98fef5208cSdrh **
99fef5208cSdrh **           expr IN (exprlist)
100fef5208cSdrh ** and
101fef5208cSdrh **           expr IN (SELECT ...)
102fef5208cSdrh **
103fef5208cSdrh ** The first form is handled by creating a set holding the list
104fef5208cSdrh ** of allowed values.  The second form causes the SELECT to generate
105fef5208cSdrh ** a temporary table.
106fef5208cSdrh **
107fef5208cSdrh ** This routine also looks for scalar SELECTs that are part of an expression.
10819a775c2Sdrh ** If it finds any, it generates code to write the value of that select
10919a775c2Sdrh ** into a memory cell.
110cce7d176Sdrh **
111cce7d176Sdrh ** Unknown fields or tables provoke an error.  The function returns
112cce7d176Sdrh ** the number of errors seen and leaves an error message on pParse->zErrMsg.
113cce7d176Sdrh */
114cce7d176Sdrh int sqliteExprResolveIds(Parse *pParse, IdList *pTabList, Expr *pExpr){
115cce7d176Sdrh   if( pExpr==0 ) return 0;
116cce7d176Sdrh   switch( pExpr->op ){
117cce7d176Sdrh     /* A lone identifier */
118cce7d176Sdrh     case TK_ID: {
119cce7d176Sdrh       int cnt = 0;   /* Number of matches */
120cce7d176Sdrh       int i;         /* Loop counter */
121cce7d176Sdrh       char *z = 0;
122cce7d176Sdrh       sqliteSetNString(&z, pExpr->token.z, pExpr->token.n, 0);
123cce7d176Sdrh       for(i=0; i<pTabList->nId; i++){
124cce7d176Sdrh         int j;
125cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
126cce7d176Sdrh         if( pTab==0 ) continue;
127cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
1287020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
129cce7d176Sdrh             cnt++;
13019a775c2Sdrh             pExpr->iTable = i + pParse->nTab;
131cce7d176Sdrh             pExpr->iField = j;
132cce7d176Sdrh           }
133cce7d176Sdrh         }
134cce7d176Sdrh       }
135cce7d176Sdrh       sqliteFree(z);
136cce7d176Sdrh       if( cnt==0 ){
137cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such field: ", -1,
138cce7d176Sdrh           pExpr->token.z, pExpr->token.n, 0);
139cce7d176Sdrh         pParse->nErr++;
140cce7d176Sdrh         return 1;
141cce7d176Sdrh       }else if( cnt>1 ){
142cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "ambiguous field name: ", -1,
143cce7d176Sdrh           pExpr->token.z, pExpr->token.n, 0);
144cce7d176Sdrh         pParse->nErr++;
145cce7d176Sdrh         return 1;
146cce7d176Sdrh       }
147cce7d176Sdrh       pExpr->op = TK_FIELD;
148cce7d176Sdrh       break;
149cce7d176Sdrh     }
150cce7d176Sdrh 
151cce7d176Sdrh     /* A table name and field name:  ID.ID */
152cce7d176Sdrh     case TK_DOT: {
153cce7d176Sdrh       int cnt = 0;             /* Number of matches */
154cce7d176Sdrh       int i;                   /* Loop counter */
155cce7d176Sdrh       Expr *pLeft, *pRight;    /* Left and right subbranches of the expr */
156cce7d176Sdrh       char *zLeft, *zRight;    /* Text of an identifier */
157cce7d176Sdrh 
158cce7d176Sdrh       pLeft = pExpr->pLeft;
159cce7d176Sdrh       pRight = pExpr->pRight;
160cce7d176Sdrh       assert( pLeft && pLeft->op==TK_ID );
161cce7d176Sdrh       assert( pRight && pRight->op==TK_ID );
162cce7d176Sdrh       zLeft = 0;
163cce7d176Sdrh       sqliteSetNString(&zLeft, pLeft->token.z, pLeft->token.n, 0);
164cce7d176Sdrh       zRight = 0;
165cce7d176Sdrh       sqliteSetNString(&zRight, pRight->token.z, pRight->token.n, 0);
166cce7d176Sdrh       for(i=0; i<pTabList->nId; i++){
167cce7d176Sdrh         int j;
168cce7d176Sdrh         char *zTab;
169cce7d176Sdrh         Table *pTab = pTabList->a[i].pTab;
170cce7d176Sdrh         if( pTab==0 ) continue;
171cce7d176Sdrh         if( pTabList->a[i].zAlias ){
172cce7d176Sdrh           zTab = pTabList->a[i].zAlias;
173cce7d176Sdrh         }else{
174cce7d176Sdrh           zTab = pTab->zName;
175cce7d176Sdrh         }
176cce7d176Sdrh         if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
177cce7d176Sdrh         for(j=0; j<pTab->nCol; j++){
1787020f651Sdrh           if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
179cce7d176Sdrh             cnt++;
18019a775c2Sdrh             pExpr->iTable = i + pParse->nTab;
181cce7d176Sdrh             pExpr->iField = j;
182cce7d176Sdrh           }
183cce7d176Sdrh         }
184cce7d176Sdrh       }
185cce7d176Sdrh       sqliteFree(zLeft);
186cce7d176Sdrh       sqliteFree(zRight);
187cce7d176Sdrh       if( cnt==0 ){
188cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such field: ", -1,
189cce7d176Sdrh           pLeft->token.z, pLeft->token.n, ".", 1,
190cce7d176Sdrh           pRight->token.z, pRight->token.n, 0);
191cce7d176Sdrh         pParse->nErr++;
192cce7d176Sdrh         return 1;
193cce7d176Sdrh       }else if( cnt>1 ){
194cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "ambiguous field name: ", -1,
195cce7d176Sdrh           pLeft->token.z, pLeft->token.n, ".", 1,
196cce7d176Sdrh           pRight->token.z, pRight->token.n, 0);
197cce7d176Sdrh         pParse->nErr++;
198cce7d176Sdrh         return 1;
199cce7d176Sdrh       }
200cce7d176Sdrh       sqliteExprDelete(pLeft);
201cce7d176Sdrh       pExpr->pLeft = 0;
202cce7d176Sdrh       sqliteExprDelete(pRight);
203cce7d176Sdrh       pExpr->pRight = 0;
204cce7d176Sdrh       pExpr->op = TK_FIELD;
205cce7d176Sdrh       break;
206cce7d176Sdrh     }
207cce7d176Sdrh 
208fef5208cSdrh     case TK_IN: {
209*d8bc7086Sdrh       Vdbe *v = sqliteGetVdbe(pParse);
210fef5208cSdrh       if( v==0 ) return 1;
211cfab11bcSdrh       if( sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
212cfab11bcSdrh         return 1;
213cfab11bcSdrh       }
214fef5208cSdrh       if( pExpr->pSelect ){
215fef5208cSdrh         /* Case 1:     expr IN (SELECT ...)
216fef5208cSdrh         **
217fef5208cSdrh         ** Generate code to write the results of the select into a temporary
2184794b980Sdrh         ** table.  The cursor number of the temporary table has already
2194794b980Sdrh         ** been put in iTable by sqliteExprResolveInSelect().
220fef5208cSdrh         */
2214794b980Sdrh         sqliteVdbeAddOp(v, OP_Open, pExpr->iTable, 1, 0, 0);
222fef5208cSdrh         if( sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable) );
223fef5208cSdrh       }else if( pExpr->pList ){
224fef5208cSdrh         /* Case 2:     expr IN (exprlist)
225fef5208cSdrh         **
226fef5208cSdrh         ** Create a set to put the exprlist values in.  The Set id is stored
227fef5208cSdrh         ** in iTable.
228fef5208cSdrh         */
229fef5208cSdrh         int i, iSet;
230fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
231fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
232fef5208cSdrh           if( !isConstant(pE2) ){
233fef5208cSdrh             sqliteSetString(&pParse->zErrMsg,
234fef5208cSdrh               "right-hand side of IN operator must be constant", 0);
235fef5208cSdrh             pParse->nErr++;
236fef5208cSdrh             return 1;
237fef5208cSdrh           }
2384794b980Sdrh           if( sqliteExprCheck(pParse, pE2, 0, 0) ){
2394794b980Sdrh             return 1;
2404794b980Sdrh           }
241fef5208cSdrh         }
242fef5208cSdrh         iSet = pExpr->iTable = pParse->nSet++;
243fef5208cSdrh         for(i=0; i<pExpr->pList->nExpr; i++){
244fef5208cSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
245fef5208cSdrh           switch( pE2->op ){
246fef5208cSdrh             case TK_FLOAT:
247fef5208cSdrh             case TK_INTEGER:
248fef5208cSdrh             case TK_STRING: {
249fef5208cSdrh               int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
250fef5208cSdrh               sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
251fef5208cSdrh               sqliteVdbeDequoteP3(v, addr);
252fef5208cSdrh               break;
253fef5208cSdrh             }
254fef5208cSdrh             default: {
255fef5208cSdrh               sqliteExprCode(pParse, pE2);
256fef5208cSdrh               sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0, 0, 0);
257fef5208cSdrh               break;
258fef5208cSdrh             }
259fef5208cSdrh           }
260fef5208cSdrh         }
261fef5208cSdrh       }
262cfab11bcSdrh       break;
263fef5208cSdrh     }
264fef5208cSdrh 
26519a775c2Sdrh     case TK_SELECT: {
266fef5208cSdrh       /* This has to be a scalar SELECT.  Generate code to put the
267fef5208cSdrh       ** value of this select in a memory cell and record the number
268fef5208cSdrh       ** of the memory cell in iField.
269fef5208cSdrh       */
27019a775c2Sdrh       pExpr->iField = pParse->nMem++;
271fef5208cSdrh       if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iField) ){
27219a775c2Sdrh         return 1;
27319a775c2Sdrh       }
27419a775c2Sdrh       break;
27519a775c2Sdrh     }
27619a775c2Sdrh 
277cce7d176Sdrh     /* For all else, just recursively walk the tree */
278cce7d176Sdrh     default: {
279cce7d176Sdrh       if( pExpr->pLeft
280cce7d176Sdrh       && sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
281cce7d176Sdrh         return 1;
282cce7d176Sdrh       }
283cce7d176Sdrh       if( pExpr->pRight
284cce7d176Sdrh       && sqliteExprResolveIds(pParse, pTabList, pExpr->pRight) ){
285cce7d176Sdrh         return 1;
286cce7d176Sdrh       }
287cce7d176Sdrh       if( pExpr->pList ){
288cce7d176Sdrh         int i;
289cce7d176Sdrh         ExprList *pList = pExpr->pList;
290cce7d176Sdrh         for(i=0; i<pList->nExpr; i++){
291cce7d176Sdrh           if( sqliteExprResolveIds(pParse, pTabList, pList->a[i].pExpr) ){
292cce7d176Sdrh             return 1;
293cce7d176Sdrh           }
294cce7d176Sdrh         }
295cce7d176Sdrh       }
296cce7d176Sdrh     }
297cce7d176Sdrh   }
298cce7d176Sdrh   return 0;
299cce7d176Sdrh }
300cce7d176Sdrh 
301cce7d176Sdrh #if 0 /* NOT USED */
302cce7d176Sdrh /*
303cce7d176Sdrh ** Compare a token against a string.  Return TRUE if they match.
304cce7d176Sdrh */
305cce7d176Sdrh static int sqliteTokenCmp(Token *pToken, const char *zStr){
306cce7d176Sdrh   int n = strlen(zStr);
307cce7d176Sdrh   if( n!=pToken->n ) return 0;
308cce7d176Sdrh   return sqliteStrNICmp(pToken->z, zStr, n)==0;
309cce7d176Sdrh }
310cce7d176Sdrh #endif
311cce7d176Sdrh 
312cce7d176Sdrh /*
313cce7d176Sdrh ** Convert a function name into its integer identifier.  Return the
314cce7d176Sdrh ** identifier.  Return FN_Unknown if the function name is unknown.
315cce7d176Sdrh */
316cce7d176Sdrh int sqliteFuncId(Token *pToken){
317cce7d176Sdrh   static const struct {
318cce7d176Sdrh      char *zName;
319cce7d176Sdrh      int len;
320cce7d176Sdrh      int id;
321cce7d176Sdrh   } aFunc[] = {
322cce7d176Sdrh      { "count",  5, FN_Count },
323cce7d176Sdrh      { "min",    3, FN_Min   },
324cce7d176Sdrh      { "max",    3, FN_Max   },
325cce7d176Sdrh      { "sum",    3, FN_Sum   },
3262282792aSdrh      { "avg",    3, FN_Avg   },
327cce7d176Sdrh   };
328cce7d176Sdrh   int i;
329cce7d176Sdrh   for(i=0; i<ArraySize(aFunc); i++){
330cce7d176Sdrh     if( aFunc[i].len==pToken->n
331cce7d176Sdrh      && sqliteStrNICmp(pToken->z, aFunc[i].zName, aFunc[i].len)==0 ){
332cce7d176Sdrh        return aFunc[i].id;
333cce7d176Sdrh     }
334cce7d176Sdrh   }
335cce7d176Sdrh   return FN_Unknown;
336cce7d176Sdrh }
337cce7d176Sdrh 
338cce7d176Sdrh /*
339cce7d176Sdrh ** Error check the functions in an expression.  Make sure all
340cce7d176Sdrh ** function names are recognized and all functions have the correct
341cce7d176Sdrh ** number of arguments.  Leave an error message in pParse->zErrMsg
342cce7d176Sdrh ** if anything is amiss.  Return the number of errors.
343cce7d176Sdrh **
344cce7d176Sdrh ** if pIsAgg is not null and this expression is an aggregate function
345cce7d176Sdrh ** (like count(*) or max(value)) then write a 1 into *pIsAgg.
346cce7d176Sdrh */
347cce7d176Sdrh int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
348cce7d176Sdrh   int nErr = 0;
349cce7d176Sdrh   if( pExpr==0 ) return 0;
350cce7d176Sdrh   switch( pExpr->op ){
351cce7d176Sdrh     case TK_FUNCTION: {
352cce7d176Sdrh       int id = sqliteFuncId(&pExpr->token);
353cce7d176Sdrh       int n = pExpr->pList ? pExpr->pList->nExpr : 0;
354cce7d176Sdrh       int no_such_func = 0;
355cce7d176Sdrh       int too_many_args = 0;
356cce7d176Sdrh       int too_few_args = 0;
357cce7d176Sdrh       int is_agg = 0;
358cce7d176Sdrh       int i;
3592282792aSdrh       pExpr->iField = id;
360cce7d176Sdrh       switch( id ){
361cce7d176Sdrh         case FN_Unknown: {
362cce7d176Sdrh           no_such_func = 1;
363cce7d176Sdrh           break;
364cce7d176Sdrh         }
365cce7d176Sdrh         case FN_Count: {
366cce7d176Sdrh           no_such_func = !allowAgg;
367cce7d176Sdrh           too_many_args = n>1;
368cce7d176Sdrh           is_agg = 1;
369cce7d176Sdrh           break;
370cce7d176Sdrh         }
371cce7d176Sdrh         case FN_Max:
372cce7d176Sdrh         case FN_Min: {
373cce7d176Sdrh           too_few_args = allowAgg ? n<1 : n<2;
374cce7d176Sdrh           is_agg = n==1;
375cce7d176Sdrh           break;
376cce7d176Sdrh         }
3772282792aSdrh         case FN_Avg:
378cce7d176Sdrh         case FN_Sum: {
379cce7d176Sdrh           no_such_func = !allowAgg;
380cce7d176Sdrh           too_many_args = n>1;
381cce7d176Sdrh           too_few_args = n<1;
382cce7d176Sdrh           is_agg = 1;
383cce7d176Sdrh           break;
384cce7d176Sdrh         }
385cce7d176Sdrh         default: break;
386cce7d176Sdrh       }
387cce7d176Sdrh       if( no_such_func ){
388cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
389cce7d176Sdrh            pExpr->token.z, pExpr->token.n, 0);
390cce7d176Sdrh         pParse->nErr++;
391cce7d176Sdrh         nErr++;
392cce7d176Sdrh       }else if( too_many_args ){
393cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "too many arguments to function ",-1,
394cce7d176Sdrh            pExpr->token.z, pExpr->token.n, "()", 2, 0);
395cce7d176Sdrh         pParse->nErr++;
396cce7d176Sdrh         nErr++;
397cce7d176Sdrh       }else if( too_few_args ){
398cce7d176Sdrh         sqliteSetNString(&pParse->zErrMsg, "too few arguments to function ",-1,
399cce7d176Sdrh            pExpr->token.z, pExpr->token.n, "()", 2, 0);
400cce7d176Sdrh         pParse->nErr++;
401cce7d176Sdrh         nErr++;
402cce7d176Sdrh       }
4032282792aSdrh       if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
404cce7d176Sdrh       if( is_agg && pIsAgg ) *pIsAgg = 1;
405cce7d176Sdrh       for(i=0; nErr==0 && i<n; i++){
406cce7d176Sdrh         nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr, 0, 0);
407cce7d176Sdrh       }
408cce7d176Sdrh     }
409cce7d176Sdrh     default: {
410cce7d176Sdrh       if( pExpr->pLeft ){
4112282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
412cce7d176Sdrh       }
413cce7d176Sdrh       if( nErr==0 && pExpr->pRight ){
4142282792aSdrh         nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
415cce7d176Sdrh       }
416fef5208cSdrh       if( nErr==0 && pExpr->pList ){
417fef5208cSdrh         int n = pExpr->pList->nExpr;
418fef5208cSdrh         int i;
419fef5208cSdrh         for(i=0; nErr==0 && i<n; i++){
4202282792aSdrh           Expr *pE2 = pExpr->pList->a[i].pExpr;
4212282792aSdrh           nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
422fef5208cSdrh         }
423fef5208cSdrh       }
424cce7d176Sdrh       break;
425cce7d176Sdrh     }
426cce7d176Sdrh   }
427cce7d176Sdrh   return nErr;
428cce7d176Sdrh }
429cce7d176Sdrh 
430cce7d176Sdrh /*
431cce7d176Sdrh ** Generate code into the current Vdbe to evaluate the given
432cce7d176Sdrh ** expression and leave the result on the stack.
433cce7d176Sdrh */
434cce7d176Sdrh void sqliteExprCode(Parse *pParse, Expr *pExpr){
435cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
436cce7d176Sdrh   int op;
437cce7d176Sdrh   switch( pExpr->op ){
438cce7d176Sdrh     case TK_PLUS:     op = OP_Add;      break;
439cce7d176Sdrh     case TK_MINUS:    op = OP_Subtract; break;
440cce7d176Sdrh     case TK_STAR:     op = OP_Multiply; break;
441cce7d176Sdrh     case TK_SLASH:    op = OP_Divide;   break;
442cce7d176Sdrh     case TK_AND:      op = OP_And;      break;
443cce7d176Sdrh     case TK_OR:       op = OP_Or;       break;
444cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
445cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
446cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
447cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
448cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
449cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
450cce7d176Sdrh     case TK_LIKE:     op = OP_Like;     break;
451cce7d176Sdrh     case TK_GLOB:     op = OP_Glob;     break;
452cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
453cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
454cce7d176Sdrh     case TK_NOT:      op = OP_Not;      break;
455cce7d176Sdrh     case TK_UMINUS:   op = OP_Negative; break;
456cce7d176Sdrh     default: break;
457cce7d176Sdrh   }
458cce7d176Sdrh   switch( pExpr->op ){
459cce7d176Sdrh     case TK_FIELD: {
4602282792aSdrh       if( pParse->useAgg ){
4612282792aSdrh         sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
4622282792aSdrh       }else{
463cce7d176Sdrh         sqliteVdbeAddOp(v, OP_Field, pExpr->iTable, pExpr->iField, 0, 0);
4642282792aSdrh       }
465cce7d176Sdrh       break;
466cce7d176Sdrh     }
467cce7d176Sdrh     case TK_INTEGER: {
468cce7d176Sdrh       int i = atoi(pExpr->token.z);
469cce7d176Sdrh       sqliteVdbeAddOp(v, OP_Integer, i, 0, 0, 0);
470cce7d176Sdrh       break;
471cce7d176Sdrh     }
472cce7d176Sdrh     case TK_FLOAT: {
473cce7d176Sdrh       int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
474cce7d176Sdrh       sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
475cce7d176Sdrh       break;
476cce7d176Sdrh     }
477cce7d176Sdrh     case TK_STRING: {
478cce7d176Sdrh       int addr = sqliteVdbeAddOp(v, OP_String, 0, 0, 0, 0);
479cce7d176Sdrh       sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
480cce7d176Sdrh       sqliteVdbeDequoteP3(v, addr);
481cce7d176Sdrh       break;
482cce7d176Sdrh     }
483cce7d176Sdrh     case TK_NULL: {
484c61053b7Sdrh       sqliteVdbeAddOp(v, OP_Null, 0, 0, 0, 0);
485cce7d176Sdrh       break;
486cce7d176Sdrh     }
487cce7d176Sdrh     case TK_AND:
488cce7d176Sdrh     case TK_OR:
489cce7d176Sdrh     case TK_PLUS:
490cce7d176Sdrh     case TK_STAR:
491cce7d176Sdrh     case TK_MINUS:
492cce7d176Sdrh     case TK_SLASH: {
493cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
494cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
495cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
496cce7d176Sdrh       break;
497cce7d176Sdrh     }
498cce7d176Sdrh     case TK_LT:
499cce7d176Sdrh     case TK_LE:
500cce7d176Sdrh     case TK_GT:
501cce7d176Sdrh     case TK_GE:
502cce7d176Sdrh     case TK_NE:
503cce7d176Sdrh     case TK_EQ:
504cce7d176Sdrh     case TK_LIKE:
505cce7d176Sdrh     case TK_GLOB: {
506cce7d176Sdrh       int dest;
507cce7d176Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
508cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
509cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
510cce7d176Sdrh       dest = sqliteVdbeCurrentAddr(v) + 2;
511cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
512cce7d176Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
513cce7d176Sdrh       break;
514cce7d176Sdrh     }
515cce7d176Sdrh     case TK_NOT:
516cce7d176Sdrh     case TK_UMINUS: {
517cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
518cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
519cce7d176Sdrh       break;
520cce7d176Sdrh     }
521cce7d176Sdrh     case TK_ISNULL:
522cce7d176Sdrh     case TK_NOTNULL: {
523cce7d176Sdrh       int dest;
5248be51133Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
525cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
526cce7d176Sdrh       dest = sqliteVdbeCurrentAddr(v) + 2;
527cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
5288be51133Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
529cce7d176Sdrh       break;
530cce7d176Sdrh     }
5312282792aSdrh     case TK_AGG_FUNCTION: {
5322282792aSdrh       sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg, 0, 0);
5332282792aSdrh       if( pExpr->iField==FN_Avg ){
5342282792aSdrh         assert( pParse->iAggCount>=0 && pParse->iAggCount<pParse->nAgg );
5352282792aSdrh         sqliteVdbeAddOp(v, OP_AggGet, 0, pParse->iAggCount, 0, 0);
5362282792aSdrh         sqliteVdbeAddOp(v, OP_Divide, 0, 0, 0, 0);
5372282792aSdrh       }
5382282792aSdrh       break;
5392282792aSdrh     }
540cce7d176Sdrh     case TK_FUNCTION: {
5412282792aSdrh       int id = pExpr->iField;
542cce7d176Sdrh       int op;
543cce7d176Sdrh       int i;
544cce7d176Sdrh       ExprList *pList = pExpr->pList;
545cce7d176Sdrh       op = id==FN_Min ? OP_Min : OP_Max;
546cce7d176Sdrh       for(i=0; i<pList->nExpr; i++){
547cce7d176Sdrh         sqliteExprCode(pParse, pList->a[i].pExpr);
548cce7d176Sdrh         if( i>0 ){
549cce7d176Sdrh           sqliteVdbeAddOp(v, op, 0, 0, 0, 0);
550cce7d176Sdrh         }
551cce7d176Sdrh       }
552cce7d176Sdrh       break;
553cce7d176Sdrh     }
55419a775c2Sdrh     case TK_SELECT: {
55519a775c2Sdrh       sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iField, 0, 0, 0);
55619a775c2Sdrh       break;
55719a775c2Sdrh     }
558fef5208cSdrh     case TK_IN: {
559fef5208cSdrh       int addr;
5604794b980Sdrh       sqliteVdbeAddOp(v, OP_Integer, 1, 0, 0, 0);
561fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
562fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
563fef5208cSdrh       if( pExpr->pSelect ){
564fef5208cSdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+2, 0, 0);
565fef5208cSdrh       }else{
566fef5208cSdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+2, 0, 0);
567fef5208cSdrh       }
5684794b980Sdrh       sqliteVdbeAddOp(v, OP_AddImm, -1, 0, 0, 0);
569fef5208cSdrh       break;
570fef5208cSdrh     }
571fef5208cSdrh     case TK_BETWEEN: {
572fef5208cSdrh       int lbl = sqliteVdbeMakeLabel(v);
573fef5208cSdrh       sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
574fef5208cSdrh       sqliteExprIfFalse(pParse, pExpr, lbl);
575fef5208cSdrh       sqliteVdbeAddOp(v, OP_AddImm, 1, 0, 0, 0);
576fef5208cSdrh       sqliteVdbeResolveLabel(v, lbl);
577fef5208cSdrh       break;
578fef5208cSdrh     }
579cce7d176Sdrh   }
580cce7d176Sdrh   return;
581cce7d176Sdrh }
582cce7d176Sdrh 
583cce7d176Sdrh /*
584cce7d176Sdrh ** Generate code for a boolean expression such that a jump is made
585cce7d176Sdrh ** to the label "dest" if the expression is true but execution
586cce7d176Sdrh ** continues straight thru if the expression is false.
587cce7d176Sdrh */
588cce7d176Sdrh void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest){
589cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
590cce7d176Sdrh   int op = 0;
591cce7d176Sdrh   switch( pExpr->op ){
592cce7d176Sdrh     case TK_LT:       op = OP_Lt;       break;
593cce7d176Sdrh     case TK_LE:       op = OP_Le;       break;
594cce7d176Sdrh     case TK_GT:       op = OP_Gt;       break;
595cce7d176Sdrh     case TK_GE:       op = OP_Ge;       break;
596cce7d176Sdrh     case TK_NE:       op = OP_Ne;       break;
597cce7d176Sdrh     case TK_EQ:       op = OP_Eq;       break;
598cce7d176Sdrh     case TK_LIKE:     op = OP_Like;     break;
599cce7d176Sdrh     case TK_GLOB:     op = OP_Glob;     break;
600cce7d176Sdrh     case TK_ISNULL:   op = OP_IsNull;   break;
601cce7d176Sdrh     case TK_NOTNULL:  op = OP_NotNull;  break;
602cce7d176Sdrh     default:  break;
603cce7d176Sdrh   }
604cce7d176Sdrh   switch( pExpr->op ){
605cce7d176Sdrh     case TK_AND: {
606cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
607cce7d176Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, d2);
608cce7d176Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest);
609cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
610cce7d176Sdrh       break;
611cce7d176Sdrh     }
612cce7d176Sdrh     case TK_OR: {
613cce7d176Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
614cce7d176Sdrh       sqliteExprIfTrue(pParse, pExpr->pRight, dest);
615cce7d176Sdrh       break;
616cce7d176Sdrh     }
617cce7d176Sdrh     case TK_NOT: {
618cce7d176Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
619cce7d176Sdrh       break;
620cce7d176Sdrh     }
621cce7d176Sdrh     case TK_LT:
622cce7d176Sdrh     case TK_LE:
623cce7d176Sdrh     case TK_GT:
624cce7d176Sdrh     case TK_GE:
625cce7d176Sdrh     case TK_NE:
626cce7d176Sdrh     case TK_EQ:
627cce7d176Sdrh     case TK_LIKE:
628cce7d176Sdrh     case TK_GLOB: {
629cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
630cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
631cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
632cce7d176Sdrh       break;
633cce7d176Sdrh     }
634cce7d176Sdrh     case TK_ISNULL:
635cce7d176Sdrh     case TK_NOTNULL: {
636cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
637cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
638cce7d176Sdrh       break;
639cce7d176Sdrh     }
640fef5208cSdrh     case TK_IN: {
641cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
642fef5208cSdrh       if( pExpr->pSelect ){
643fef5208cSdrh         sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest, 0, 0);
644fef5208cSdrh       }else{
645fef5208cSdrh         sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest, 0, 0);
646fef5208cSdrh       }
647fef5208cSdrh       break;
648fef5208cSdrh     }
649fef5208cSdrh     case TK_BETWEEN: {
650fef5208cSdrh       int lbl = sqliteVdbeMakeLabel(v);
651fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
652fef5208cSdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
653fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
654fef5208cSdrh       sqliteVdbeAddOp(v, OP_Lt, 0, lbl, 0, 0);
655fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
656fef5208cSdrh       sqliteVdbeAddOp(v, OP_Le, 0, dest, 0, 0);
657fef5208cSdrh       sqliteVdbeAddOp(v, OP_Integer, 0, 0, 0, 0);
658fef5208cSdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, lbl);
659fef5208cSdrh       break;
660fef5208cSdrh     }
661cce7d176Sdrh     default: {
662cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
663cce7d176Sdrh       sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
664cce7d176Sdrh       break;
665cce7d176Sdrh     }
666cce7d176Sdrh   }
667cce7d176Sdrh }
668cce7d176Sdrh 
669cce7d176Sdrh /*
670cce7d176Sdrh ** Generate code for boolean expression such that a jump is made
671cce7d176Sdrh ** to the label "dest" if the expression is false but execution
672cce7d176Sdrh ** continues straight thru if the expression is true.
673cce7d176Sdrh */
674cce7d176Sdrh void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest){
675cce7d176Sdrh   Vdbe *v = pParse->pVdbe;
676cce7d176Sdrh   int op = 0;
677cce7d176Sdrh   switch( pExpr->op ){
678cce7d176Sdrh     case TK_LT:       op = OP_Ge;       break;
679cce7d176Sdrh     case TK_LE:       op = OP_Gt;       break;
680cce7d176Sdrh     case TK_GT:       op = OP_Le;       break;
681cce7d176Sdrh     case TK_GE:       op = OP_Lt;       break;
682cce7d176Sdrh     case TK_NE:       op = OP_Eq;       break;
683cce7d176Sdrh     case TK_EQ:       op = OP_Ne;       break;
684cce7d176Sdrh     case TK_LIKE:     op = OP_Like;     break;
685cce7d176Sdrh     case TK_GLOB:     op = OP_Glob;     break;
686cce7d176Sdrh     case TK_ISNULL:   op = OP_NotNull;  break;
687cce7d176Sdrh     case TK_NOTNULL:  op = OP_IsNull;   break;
688cce7d176Sdrh     default:  break;
689cce7d176Sdrh   }
690cce7d176Sdrh   switch( pExpr->op ){
691cce7d176Sdrh     case TK_AND: {
692cce7d176Sdrh       sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
693cce7d176Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest);
694cce7d176Sdrh       break;
695cce7d176Sdrh     }
696cce7d176Sdrh     case TK_OR: {
697cce7d176Sdrh       int d2 = sqliteVdbeMakeLabel(v);
698cce7d176Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, d2);
699cce7d176Sdrh       sqliteExprIfFalse(pParse, pExpr->pRight, dest);
700cce7d176Sdrh       sqliteVdbeResolveLabel(v, d2);
701cce7d176Sdrh       break;
702cce7d176Sdrh     }
703cce7d176Sdrh     case TK_NOT: {
704cce7d176Sdrh       sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
705cce7d176Sdrh       break;
706cce7d176Sdrh     }
707cce7d176Sdrh     case TK_LT:
708cce7d176Sdrh     case TK_LE:
709cce7d176Sdrh     case TK_GT:
710cce7d176Sdrh     case TK_GE:
711cce7d176Sdrh     case TK_NE:
712cce7d176Sdrh     case TK_EQ: {
713cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
714cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
715cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
716cce7d176Sdrh       break;
717cce7d176Sdrh     }
718cce7d176Sdrh     case TK_LIKE:
719cce7d176Sdrh     case TK_GLOB: {
720cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
721cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pRight);
722cce7d176Sdrh       sqliteVdbeAddOp(v, op, 1, dest, 0, 0);
723cce7d176Sdrh       break;
724cce7d176Sdrh     }
725cce7d176Sdrh     case TK_ISNULL:
726cce7d176Sdrh     case TK_NOTNULL: {
727cce7d176Sdrh       sqliteExprCode(pParse, pExpr->pLeft);
728cce7d176Sdrh       sqliteVdbeAddOp(v, op, 0, dest, 0, 0);
729cce7d176Sdrh       break;
730cce7d176Sdrh     }
731fef5208cSdrh     case TK_IN: {
732cfab11bcSdrh       sqliteExprCode(pParse, pExpr->pLeft);
733fef5208cSdrh       if( pExpr->pSelect ){
734fef5208cSdrh         sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest, 0, 0);
735fef5208cSdrh       }else{
736fef5208cSdrh         sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest, 0, 0);
737fef5208cSdrh       }
738fef5208cSdrh       break;
739fef5208cSdrh     }
740fef5208cSdrh     case TK_BETWEEN: {
741fef5208cSdrh       int addr;
742fef5208cSdrh       sqliteExprCode(pParse, pExpr->pLeft);
743fef5208cSdrh       sqliteVdbeAddOp(v, OP_Dup, 0, 0, 0, 0);
744fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
745fef5208cSdrh       addr = sqliteVdbeCurrentAddr(v);
746fef5208cSdrh       sqliteVdbeAddOp(v, OP_Ge, 0, addr+3, 0, 0);
747fef5208cSdrh       sqliteVdbeAddOp(v, OP_Pop, 1, 0, 0, 0);
748fef5208cSdrh       sqliteVdbeAddOp(v, OP_Goto, 0, dest, 0, 0);
749fef5208cSdrh       sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
750fef5208cSdrh       sqliteVdbeAddOp(v, OP_Gt, 0, dest, 0, 0);
751fef5208cSdrh       break;
752fef5208cSdrh     }
753cce7d176Sdrh     default: {
754cce7d176Sdrh       sqliteExprCode(pParse, pExpr);
755cce7d176Sdrh       sqliteVdbeAddOp(v, OP_Not, 0, 0, 0, 0);
756cce7d176Sdrh       sqliteVdbeAddOp(v, OP_If, 0, dest, 0, 0);
757cce7d176Sdrh       break;
758cce7d176Sdrh     }
759cce7d176Sdrh   }
760cce7d176Sdrh }
7612282792aSdrh 
7622282792aSdrh /*
7632282792aSdrh ** Do a deep comparison of two expression trees.  Return TRUE (non-zero)
7642282792aSdrh ** if they are identical and return FALSE if they differ in any way.
7652282792aSdrh */
766*d8bc7086Sdrh int sqliteExprCompare(Expr *pA, Expr *pB){
7672282792aSdrh   int i;
7682282792aSdrh   if( pA==0 ){
7692282792aSdrh     return pB==0;
7702282792aSdrh   }else if( pB==0 ){
7712282792aSdrh     return 0;
7722282792aSdrh   }
7732282792aSdrh   if( pA->op!=pB->op ) return 0;
774*d8bc7086Sdrh   if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
775*d8bc7086Sdrh   if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
7762282792aSdrh   if( pA->pList ){
7772282792aSdrh     if( pB->pList==0 ) return 0;
7782282792aSdrh     if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
7792282792aSdrh     for(i=0; i<pA->pList->nExpr; i++){
780*d8bc7086Sdrh       if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
7812282792aSdrh         return 0;
7822282792aSdrh       }
7832282792aSdrh     }
7842282792aSdrh   }else if( pB->pList ){
7852282792aSdrh     return 0;
7862282792aSdrh   }
7872282792aSdrh   if( pA->pSelect || pB->pSelect ) return 0;
7882282792aSdrh   if( pA->token.z ){
7892282792aSdrh     if( pB->token.z==0 ) return 0;
7902282792aSdrh     if( pB->token.n!=pA->token.n ) return 0;
7912282792aSdrh     if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
7922282792aSdrh   }
7932282792aSdrh   return 1;
7942282792aSdrh }
7952282792aSdrh 
7962282792aSdrh /*
7972282792aSdrh ** Add a new element to the pParse->aAgg[] array and return its index.
7982282792aSdrh */
7992282792aSdrh static int appendAggInfo(Parse *pParse){
8002282792aSdrh   if( (pParse->nAgg & 0x7)==0 ){
8012282792aSdrh     int amt = pParse->nAgg + 8;
8022282792aSdrh     pParse->aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
8032282792aSdrh     if( pParse->aAgg==0 ){
8042282792aSdrh       sqliteSetString(&pParse->zErrMsg, "out of memory", 0);
8052282792aSdrh       pParse->nErr++;
8062282792aSdrh       return -1;
8072282792aSdrh     }
8082282792aSdrh   }
8092282792aSdrh   memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
8102282792aSdrh   return pParse->nAgg++;
8112282792aSdrh }
8122282792aSdrh 
8132282792aSdrh /*
8142282792aSdrh ** Analyze the given expression looking for aggregate functions and
8152282792aSdrh ** for variables that need to be added to the pParse->aAgg[] array.
8162282792aSdrh ** Make additional entries to the pParse->aAgg[] array as necessary.
8172282792aSdrh **
8182282792aSdrh ** This routine should only be called after the expression has been
8192282792aSdrh ** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
8202282792aSdrh **
8212282792aSdrh ** If errors are seen, leave an error message in zErrMsg and return
8222282792aSdrh ** the number of errors.
8232282792aSdrh */
8242282792aSdrh int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
8252282792aSdrh   int i;
8262282792aSdrh   AggExpr *aAgg;
8272282792aSdrh   int nErr = 0;
8282282792aSdrh 
8292282792aSdrh   if( pExpr==0 ) return 0;
8302282792aSdrh   switch( pExpr->op ){
8312282792aSdrh     case TK_FIELD: {
8322282792aSdrh       aAgg = pParse->aAgg;
8332282792aSdrh       for(i=0; i<pParse->nAgg; i++){
8342282792aSdrh         if( aAgg[i].isAgg ) continue;
8352282792aSdrh         if( aAgg[i].pExpr->iTable==pExpr->iTable
8362282792aSdrh          && aAgg[i].pExpr->iField==pExpr->iField ){
8372282792aSdrh           pExpr->iAgg = i;
8382282792aSdrh           break;
8392282792aSdrh         }
8402282792aSdrh       }
8412282792aSdrh       if( i>=pParse->nAgg ){
8422282792aSdrh         i = appendAggInfo(pParse);
8432282792aSdrh         if( i<0 ) return 1;
8442282792aSdrh         pParse->aAgg[i].isAgg = 0;
8452282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
8462282792aSdrh       }
8472282792aSdrh       break;
8482282792aSdrh     }
8492282792aSdrh     case TK_AGG_FUNCTION: {
8502282792aSdrh       if( pExpr->iField==FN_Count || pExpr->iField==FN_Avg ){
8512282792aSdrh         if( pParse->iAggCount>=0 ){
8522282792aSdrh           i = pParse->iAggCount;
8532282792aSdrh         }else{
8542282792aSdrh           i = appendAggInfo(pParse);
8552282792aSdrh           if( i<0 ) return 1;
8562282792aSdrh           pParse->aAgg[i].isAgg = 1;
8572282792aSdrh           pParse->aAgg[i].pExpr = 0;
8582282792aSdrh           pParse->iAggCount = i;
8592282792aSdrh         }
8602282792aSdrh         if( pExpr->iField==FN_Count ){
8612282792aSdrh           pExpr->iAgg = i;
8622282792aSdrh           break;
8632282792aSdrh         }
8642282792aSdrh       }
8652282792aSdrh       aAgg = pParse->aAgg;
8662282792aSdrh       for(i=0; i<pParse->nAgg; i++){
8672282792aSdrh         if( !aAgg[i].isAgg ) continue;
868*d8bc7086Sdrh         if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
8692282792aSdrh           break;
8702282792aSdrh         }
8712282792aSdrh       }
8722282792aSdrh       if( i>=pParse->nAgg ){
8732282792aSdrh         i = appendAggInfo(pParse);
8742282792aSdrh         if( i<0 ) return 1;
8752282792aSdrh         pParse->aAgg[i].isAgg = 1;
8762282792aSdrh         pParse->aAgg[i].pExpr = pExpr;
8772282792aSdrh       }
8782282792aSdrh       pExpr->iAgg = i;
8792282792aSdrh       break;
8802282792aSdrh     }
8812282792aSdrh     default: {
8822282792aSdrh       if( pExpr->pLeft ){
8832282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
8842282792aSdrh       }
8852282792aSdrh       if( nErr==0 && pExpr->pRight ){
8862282792aSdrh         nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
8872282792aSdrh       }
8882282792aSdrh       if( nErr==0 && pExpr->pList ){
8892282792aSdrh         int n = pExpr->pList->nExpr;
8902282792aSdrh         int i;
8912282792aSdrh         for(i=0; nErr==0 && i<n; i++){
8922282792aSdrh           nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
8932282792aSdrh         }
8942282792aSdrh       }
8952282792aSdrh       break;
8962282792aSdrh     }
8972282792aSdrh   }
8982282792aSdrh   return nErr;
8992282792aSdrh }
900