blob: 5ace17836ecdf762ec10b987dad76a2ab19d02b6 [file] [log] [blame]
drhcce7d172000-05-31 15:34:51 +00001/*
drhb19a2bc2001-09-16 00:13:26 +00002** 2001 September 15
drhcce7d172000-05-31 15:34:51 +00003**
drhb19a2bc2001-09-16 00:13:26 +00004** The author disclaims copyright to this source code. In place of
5** a legal notice, here is a blessing:
drhcce7d172000-05-31 15:34:51 +00006**
drhb19a2bc2001-09-16 00:13:26 +00007** May you do good and not evil.
8** May you find forgiveness for yourself and forgive others.
9** May you share freely, never taking more than you give.
drhcce7d172000-05-31 15:34:51 +000010**
11*************************************************************************
drh1ccde152000-06-17 13:12:39 +000012** This file contains routines used for analyzing expressions and
drhb19a2bc2001-09-16 00:13:26 +000013** for generating VDBE code that evaluates expressions in SQLite.
drhcce7d172000-05-31 15:34:51 +000014**
drh99fcd712001-10-13 01:06:47 +000015** $Id: expr.c,v 1.30 2001/10/13 01:06:48 drh Exp $
drhcce7d172000-05-31 15:34:51 +000016*/
17#include "sqliteInt.h"
18
19/*
drhfef52082000-06-06 01:50:43 +000020** Walk an expression tree. Return 1 if the expression is constant
21** and 0 if it involves variables.
22*/
23static int isConstant(Expr *p){
24 switch( p->op ){
25 case TK_ID:
drh967e8b72000-06-21 13:59:10 +000026 case TK_COLUMN:
drhfef52082000-06-06 01:50:43 +000027 case TK_DOT:
28 return 0;
29 default: {
30 if( p->pLeft && !isConstant(p->pLeft) ) return 0;
31 if( p->pRight && !isConstant(p->pRight) ) return 0;
32 if( p->pList ){
33 int i;
34 for(i=0; i<p->pList->nExpr; i++){
35 if( !isConstant(p->pList->a[i].pExpr) ) return 0;
36 }
37 }
38 break;
39 }
40 }
41 return 1;
42}
43
44/*
drh4794b982000-06-06 13:54:14 +000045** Walk the expression tree and process operators of the form:
46**
47** expr IN (SELECT ...)
48**
drh967e8b72000-06-21 13:59:10 +000049** These operators have to be processed before column names are
drh4794b982000-06-06 13:54:14 +000050** resolved because each such operator increments pParse->nTab
drh1ccde152000-06-17 13:12:39 +000051** to reserve cursor numbers for its own use. But pParse->nTab
drh967e8b72000-06-21 13:59:10 +000052** needs to be constant once we begin resolving column names.
drh4794b982000-06-06 13:54:14 +000053**
54** Actually, the processing of IN-SELECT is only started by this
55** routine. This routine allocates a cursor number to the IN-SELECT
56** and then moves on. The code generation is done by
57** sqliteExprResolveIds() which must be called afterwards.
58*/
59void sqliteExprResolveInSelect(Parse *pParse, Expr *pExpr){
60 if( pExpr==0 ) return;
61 if( pExpr->op==TK_IN && pExpr->pSelect!=0 ){
62 pExpr->iTable = pParse->nTab++;
63 }else{
64 if( pExpr->pLeft ) sqliteExprResolveInSelect(pParse, pExpr->pLeft);
65 if( pExpr->pRight ) sqliteExprResolveInSelect(pParse, pExpr->pRight);
66 if( pExpr->pList ){
67 int i;
68 ExprList *pList = pExpr->pList;
69 for(i=0; i<pList->nExpr; i++){
70 sqliteExprResolveInSelect(pParse, pList->a[i].pExpr);
71 }
72 }
73 }
74}
75
76/*
drhc4a3c772001-04-04 11:48:57 +000077** Return TRUE if the given string is a row-id column name.
78*/
79static int sqliteIsRowid(const char *z){
80 if( sqliteStrICmp(z, "_ROWID_")==0 ) return 1;
81 if( sqliteStrICmp(z, "ROWID")==0 ) return 1;
82 if( sqliteStrICmp(z, "OID")==0 ) return 1;
83 return 0;
84}
85
86/*
drhcce7d172000-05-31 15:34:51 +000087** This routine walks an expression tree and resolves references to
drh967e8b72000-06-21 13:59:10 +000088** table columns. Nodes of the form ID.ID or ID resolve into an
89** index to the table in the table list and a column offset. The opcode
90** for such nodes is changed to TK_COLUMN. The iTable value is changed
drh19a775c2000-06-05 18:54:46 +000091** to the index of the referenced table in pTabList plus the pParse->nTab
drh967e8b72000-06-21 13:59:10 +000092** value. The iColumn value is changed to the index of the column of the
drh19a775c2000-06-05 18:54:46 +000093** referenced table.
94**
drhfef52082000-06-06 01:50:43 +000095** We also check for instances of the IN operator. IN comes in two
96** forms:
97**
98** expr IN (exprlist)
99** and
100** expr IN (SELECT ...)
101**
102** The first form is handled by creating a set holding the list
103** of allowed values. The second form causes the SELECT to generate
104** a temporary table.
105**
106** This routine also looks for scalar SELECTs that are part of an expression.
drh19a775c2000-06-05 18:54:46 +0000107** If it finds any, it generates code to write the value of that select
108** into a memory cell.
drhcce7d172000-05-31 15:34:51 +0000109**
drh967e8b72000-06-21 13:59:10 +0000110** Unknown columns or tables provoke an error. The function returns
drhcce7d172000-05-31 15:34:51 +0000111** the number of errors seen and leaves an error message on pParse->zErrMsg.
112*/
113int sqliteExprResolveIds(Parse *pParse, IdList *pTabList, Expr *pExpr){
drhdaffd0e2001-04-11 14:28:42 +0000114 if( pExpr==0 || pTabList==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000115 switch( pExpr->op ){
116 /* A lone identifier */
117 case TK_ID: {
drhc4a3c772001-04-04 11:48:57 +0000118 int cnt = 0; /* Number of matches */
119 int i; /* Loop counter */
drh6e142f52000-06-08 13:36:40 +0000120 char *z = sqliteStrNDup(pExpr->token.z, pExpr->token.n);
drhdaffd0e2001-04-11 14:28:42 +0000121 if( z==0 ) return 1;
drhcce7d172000-05-31 15:34:51 +0000122 for(i=0; i<pTabList->nId; i++){
123 int j;
124 Table *pTab = pTabList->a[i].pTab;
125 if( pTab==0 ) continue;
126 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000127 if( sqliteStrICmp(pTab->aCol[j].zName, z)==0 ){
drhcce7d172000-05-31 15:34:51 +0000128 cnt++;
drh19a775c2000-06-05 18:54:46 +0000129 pExpr->iTable = i + pParse->nTab;
drh967e8b72000-06-21 13:59:10 +0000130 pExpr->iColumn = j;
drhcce7d172000-05-31 15:34:51 +0000131 }
132 }
133 }
drhc4a3c772001-04-04 11:48:57 +0000134 if( cnt==0 && sqliteIsRowid(z) ){
135 pExpr->iColumn = -1;
136 pExpr->iTable = pParse->nTab;
137 cnt = 1 + (pTabList->nId>1);
138 }
drhcce7d172000-05-31 15:34:51 +0000139 sqliteFree(z);
140 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000141 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000142 pExpr->token.z, pExpr->token.n, 0);
143 pParse->nErr++;
144 return 1;
145 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000146 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000147 pExpr->token.z, pExpr->token.n, 0);
148 pParse->nErr++;
149 return 1;
150 }
drh967e8b72000-06-21 13:59:10 +0000151 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000152 break;
153 }
154
drh967e8b72000-06-21 13:59:10 +0000155 /* A table name and column name: ID.ID */
drhcce7d172000-05-31 15:34:51 +0000156 case TK_DOT: {
drhbed86902000-06-02 13:27:59 +0000157 int cnt = 0; /* Number of matches */
drhc4a3c772001-04-04 11:48:57 +0000158 int cntTab = 0; /* Number of matching tables */
drhbed86902000-06-02 13:27:59 +0000159 int i; /* Loop counter */
drhcce7d172000-05-31 15:34:51 +0000160 Expr *pLeft, *pRight; /* Left and right subbranches of the expr */
drhcce7d172000-05-31 15:34:51 +0000161 char *zLeft, *zRight; /* Text of an identifier */
162
163 pLeft = pExpr->pLeft;
164 pRight = pExpr->pRight;
165 assert( pLeft && pLeft->op==TK_ID );
166 assert( pRight && pRight->op==TK_ID );
drh6e142f52000-06-08 13:36:40 +0000167 zLeft = sqliteStrNDup(pLeft->token.z, pLeft->token.n);
168 zRight = sqliteStrNDup(pRight->token.z, pRight->token.n);
drhdaffd0e2001-04-11 14:28:42 +0000169 if( zLeft==0 || zRight==0 ){
170 sqliteFree(zLeft);
171 sqliteFree(zRight);
172 return 1;
173 }
drh87c40e82001-07-23 14:33:02 +0000174 sqliteDequote(zLeft);
175 sqliteDequote(zRight);
drhc4a3c772001-04-04 11:48:57 +0000176 pExpr->iTable = -1;
drhcce7d172000-05-31 15:34:51 +0000177 for(i=0; i<pTabList->nId; i++){
178 int j;
179 char *zTab;
180 Table *pTab = pTabList->a[i].pTab;
181 if( pTab==0 ) continue;
182 if( pTabList->a[i].zAlias ){
183 zTab = pTabList->a[i].zAlias;
184 }else{
185 zTab = pTab->zName;
186 }
187 if( sqliteStrICmp(zTab, zLeft)!=0 ) continue;
drhc4a3c772001-04-04 11:48:57 +0000188 if( 0==(cntTab++) ) pExpr->iTable = i + pParse->nTab;
drhcce7d172000-05-31 15:34:51 +0000189 for(j=0; j<pTab->nCol; j++){
drh7020f652000-06-03 18:06:52 +0000190 if( sqliteStrICmp(pTab->aCol[j].zName, zRight)==0 ){
drhcce7d172000-05-31 15:34:51 +0000191 cnt++;
drh19a775c2000-06-05 18:54:46 +0000192 pExpr->iTable = i + pParse->nTab;
drh967e8b72000-06-21 13:59:10 +0000193 pExpr->iColumn = j;
drhcce7d172000-05-31 15:34:51 +0000194 }
195 }
196 }
drhc4a3c772001-04-04 11:48:57 +0000197 if( cnt==0 && cntTab==1 && sqliteIsRowid(zRight) ){
198 cnt = 1;
199 pExpr->iColumn = -1;
200 }
drhcce7d172000-05-31 15:34:51 +0000201 sqliteFree(zLeft);
202 sqliteFree(zRight);
203 if( cnt==0 ){
drh967e8b72000-06-21 13:59:10 +0000204 sqliteSetNString(&pParse->zErrMsg, "no such column: ", -1,
drhcce7d172000-05-31 15:34:51 +0000205 pLeft->token.z, pLeft->token.n, ".", 1,
206 pRight->token.z, pRight->token.n, 0);
207 pParse->nErr++;
208 return 1;
209 }else if( cnt>1 ){
drh967e8b72000-06-21 13:59:10 +0000210 sqliteSetNString(&pParse->zErrMsg, "ambiguous column name: ", -1,
drhcce7d172000-05-31 15:34:51 +0000211 pLeft->token.z, pLeft->token.n, ".", 1,
212 pRight->token.z, pRight->token.n, 0);
213 pParse->nErr++;
214 return 1;
215 }
216 sqliteExprDelete(pLeft);
217 pExpr->pLeft = 0;
218 sqliteExprDelete(pRight);
219 pExpr->pRight = 0;
drh967e8b72000-06-21 13:59:10 +0000220 pExpr->op = TK_COLUMN;
drhcce7d172000-05-31 15:34:51 +0000221 break;
222 }
223
drhfef52082000-06-06 01:50:43 +0000224 case TK_IN: {
drhd8bc7082000-06-07 23:51:50 +0000225 Vdbe *v = sqliteGetVdbe(pParse);
drhfef52082000-06-06 01:50:43 +0000226 if( v==0 ) return 1;
drhcfab11b2000-06-06 03:31:22 +0000227 if( sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
228 return 1;
229 }
drhfef52082000-06-06 01:50:43 +0000230 if( pExpr->pSelect ){
231 /* Case 1: expr IN (SELECT ...)
232 **
233 ** Generate code to write the results of the select into a temporary
drh4794b982000-06-06 13:54:14 +0000234 ** table. The cursor number of the temporary table has already
235 ** been put in iTable by sqliteExprResolveInSelect().
drhfef52082000-06-06 01:50:43 +0000236 */
drh99fcd712001-10-13 01:06:47 +0000237 sqliteVdbeAddOp(v, OP_OpenTemp, pExpr->iTable, 0);
drhfef52082000-06-06 01:50:43 +0000238 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Set, pExpr->iTable) );
239 }else if( pExpr->pList ){
240 /* Case 2: expr IN (exprlist)
241 **
242 ** Create a set to put the exprlist values in. The Set id is stored
243 ** in iTable.
244 */
245 int i, iSet;
246 for(i=0; i<pExpr->pList->nExpr; i++){
247 Expr *pE2 = pExpr->pList->a[i].pExpr;
drhfef52082000-06-06 01:50:43 +0000248 if( !isConstant(pE2) ){
249 sqliteSetString(&pParse->zErrMsg,
250 "right-hand side of IN operator must be constant", 0);
251 pParse->nErr++;
252 return 1;
253 }
drh4794b982000-06-06 13:54:14 +0000254 if( sqliteExprCheck(pParse, pE2, 0, 0) ){
255 return 1;
256 }
drhfef52082000-06-06 01:50:43 +0000257 }
258 iSet = pExpr->iTable = pParse->nSet++;
259 for(i=0; i<pExpr->pList->nExpr; i++){
260 Expr *pE2 = pExpr->pList->a[i].pExpr;
261 switch( pE2->op ){
262 case TK_FLOAT:
263 case TK_INTEGER:
264 case TK_STRING: {
drh99fcd712001-10-13 01:06:47 +0000265 int addr = sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
drhfef52082000-06-06 01:50:43 +0000266 sqliteVdbeChangeP3(v, addr, pE2->token.z, pE2->token.n);
267 sqliteVdbeDequoteP3(v, addr);
268 break;
269 }
270 default: {
271 sqliteExprCode(pParse, pE2);
drh99fcd712001-10-13 01:06:47 +0000272 sqliteVdbeAddOp(v, OP_SetInsert, iSet, 0);
drhfef52082000-06-06 01:50:43 +0000273 break;
274 }
275 }
276 }
277 }
drhcfab11b2000-06-06 03:31:22 +0000278 break;
drhfef52082000-06-06 01:50:43 +0000279 }
280
drh19a775c2000-06-05 18:54:46 +0000281 case TK_SELECT: {
drhfef52082000-06-06 01:50:43 +0000282 /* This has to be a scalar SELECT. Generate code to put the
283 ** value of this select in a memory cell and record the number
drh967e8b72000-06-21 13:59:10 +0000284 ** of the memory cell in iColumn.
drhfef52082000-06-06 01:50:43 +0000285 */
drh967e8b72000-06-21 13:59:10 +0000286 pExpr->iColumn = pParse->nMem++;
287 if( sqliteSelect(pParse, pExpr->pSelect, SRT_Mem, pExpr->iColumn) ){
drh19a775c2000-06-05 18:54:46 +0000288 return 1;
289 }
290 break;
291 }
292
drhcce7d172000-05-31 15:34:51 +0000293 /* For all else, just recursively walk the tree */
294 default: {
drh4794b982000-06-06 13:54:14 +0000295 if( pExpr->pLeft
296 && sqliteExprResolveIds(pParse, pTabList, pExpr->pLeft) ){
drhcce7d172000-05-31 15:34:51 +0000297 return 1;
298 }
299 if( pExpr->pRight
drh4794b982000-06-06 13:54:14 +0000300 && sqliteExprResolveIds(pParse, pTabList, pExpr->pRight) ){
drhcce7d172000-05-31 15:34:51 +0000301 return 1;
302 }
303 if( pExpr->pList ){
304 int i;
305 ExprList *pList = pExpr->pList;
306 for(i=0; i<pList->nExpr; i++){
307 if( sqliteExprResolveIds(pParse, pTabList, pList->a[i].pExpr) ){
308 return 1;
309 }
310 }
311 }
312 }
313 }
314 return 0;
315}
316
317#if 0 /* NOT USED */
318/*
319** Compare a token against a string. Return TRUE if they match.
320*/
321static int sqliteTokenCmp(Token *pToken, const char *zStr){
322 int n = strlen(zStr);
323 if( n!=pToken->n ) return 0;
324 return sqliteStrNICmp(pToken->z, zStr, n)==0;
325}
326#endif
327
328/*
329** Convert a function name into its integer identifier. Return the
330** identifier. Return FN_Unknown if the function name is unknown.
331*/
332int sqliteFuncId(Token *pToken){
333 static const struct {
334 char *zName;
335 int len;
336 int id;
337 } aFunc[] = {
338 { "count", 5, FN_Count },
339 { "min", 3, FN_Min },
340 { "max", 3, FN_Max },
341 { "sum", 3, FN_Sum },
drh22827922000-06-06 17:27:05 +0000342 { "avg", 3, FN_Avg },
drh0bdaf622000-06-11 23:50:13 +0000343 { "fcnt", 4, FN_Fcnt }, /* Used for testing only */
drh6ec27332000-08-28 15:51:43 +0000344 { "length", 6, FN_Length},
345 { "substr", 6, FN_Substr},
drhcce7d172000-05-31 15:34:51 +0000346 };
347 int i;
348 for(i=0; i<ArraySize(aFunc); i++){
349 if( aFunc[i].len==pToken->n
350 && sqliteStrNICmp(pToken->z, aFunc[i].zName, aFunc[i].len)==0 ){
351 return aFunc[i].id;
352 }
353 }
354 return FN_Unknown;
355}
356
357/*
358** Error check the functions in an expression. Make sure all
359** function names are recognized and all functions have the correct
360** number of arguments. Leave an error message in pParse->zErrMsg
361** if anything is amiss. Return the number of errors.
362**
363** if pIsAgg is not null and this expression is an aggregate function
364** (like count(*) or max(value)) then write a 1 into *pIsAgg.
365*/
366int sqliteExprCheck(Parse *pParse, Expr *pExpr, int allowAgg, int *pIsAgg){
367 int nErr = 0;
368 if( pExpr==0 ) return 0;
drhcce7d172000-05-31 15:34:51 +0000369 switch( pExpr->op ){
370 case TK_FUNCTION: {
371 int id = sqliteFuncId(&pExpr->token);
372 int n = pExpr->pList ? pExpr->pList->nExpr : 0;
373 int no_such_func = 0;
374 int too_many_args = 0;
375 int too_few_args = 0;
376 int is_agg = 0;
377 int i;
drh967e8b72000-06-21 13:59:10 +0000378 pExpr->iColumn = id;
drhcce7d172000-05-31 15:34:51 +0000379 switch( id ){
380 case FN_Unknown: {
381 no_such_func = 1;
382 break;
383 }
384 case FN_Count: {
385 no_such_func = !allowAgg;
386 too_many_args = n>1;
387 is_agg = 1;
388 break;
389 }
390 case FN_Max:
391 case FN_Min: {
392 too_few_args = allowAgg ? n<1 : n<2;
393 is_agg = n==1;
394 break;
395 }
drh22827922000-06-06 17:27:05 +0000396 case FN_Avg:
drhcce7d172000-05-31 15:34:51 +0000397 case FN_Sum: {
398 no_such_func = !allowAgg;
399 too_many_args = n>1;
400 too_few_args = n<1;
401 is_agg = 1;
402 break;
403 }
drh6ec27332000-08-28 15:51:43 +0000404 case FN_Length: {
405 too_few_args = n<1;
406 too_many_args = n>1;
407 break;
408 }
409 case FN_Substr: {
410 too_few_args = n<3;
411 too_many_args = n>3;
412 break;
413 }
drh0bdaf622000-06-11 23:50:13 +0000414 /* The "fcnt(*)" function always returns the number of fetch
415 ** operations that have occurred so far while processing the
416 ** SQL statement. This information can be used by test procedures
417 ** to verify that indices are being used properly to minimize
418 ** searching. All arguments to fcnt() are ignored. fcnt() has
419 ** no use (other than testing) that we are aware of.
420 */
421 case FN_Fcnt: {
422 n = 0;
423 break;
424 }
drh6ec27332000-08-28 15:51:43 +0000425
drhcce7d172000-05-31 15:34:51 +0000426 default: break;
427 }
428 if( no_such_func ){
429 sqliteSetNString(&pParse->zErrMsg, "no such function: ", -1,
430 pExpr->token.z, pExpr->token.n, 0);
431 pParse->nErr++;
432 nErr++;
433 }else if( too_many_args ){
434 sqliteSetNString(&pParse->zErrMsg, "too many arguments to function ",-1,
435 pExpr->token.z, pExpr->token.n, "()", 2, 0);
436 pParse->nErr++;
437 nErr++;
438 }else if( too_few_args ){
439 sqliteSetNString(&pParse->zErrMsg, "too few arguments to function ",-1,
440 pExpr->token.z, pExpr->token.n, "()", 2, 0);
441 pParse->nErr++;
442 nErr++;
443 }
drh22827922000-06-06 17:27:05 +0000444 if( is_agg ) pExpr->op = TK_AGG_FUNCTION;
drhcce7d172000-05-31 15:34:51 +0000445 if( is_agg && pIsAgg ) *pIsAgg = 1;
446 for(i=0; nErr==0 && i<n; i++){
drh4cfa7932000-06-08 15:10:46 +0000447 nErr = sqliteExprCheck(pParse, pExpr->pList->a[i].pExpr,
448 allowAgg && !is_agg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000449 }
450 }
451 default: {
452 if( pExpr->pLeft ){
drh22827922000-06-06 17:27:05 +0000453 nErr = sqliteExprCheck(pParse, pExpr->pLeft, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000454 }
455 if( nErr==0 && pExpr->pRight ){
drh22827922000-06-06 17:27:05 +0000456 nErr = sqliteExprCheck(pParse, pExpr->pRight, allowAgg, pIsAgg);
drhcce7d172000-05-31 15:34:51 +0000457 }
drhfef52082000-06-06 01:50:43 +0000458 if( nErr==0 && pExpr->pList ){
459 int n = pExpr->pList->nExpr;
460 int i;
461 for(i=0; nErr==0 && i<n; i++){
drh22827922000-06-06 17:27:05 +0000462 Expr *pE2 = pExpr->pList->a[i].pExpr;
463 nErr = sqliteExprCheck(pParse, pE2, allowAgg, pIsAgg);
drhfef52082000-06-06 01:50:43 +0000464 }
465 }
drhcce7d172000-05-31 15:34:51 +0000466 break;
467 }
468 }
469 return nErr;
470}
471
472/*
473** Generate code into the current Vdbe to evaluate the given
drh1ccde152000-06-17 13:12:39 +0000474** expression and leave the result on the top of stack.
drhcce7d172000-05-31 15:34:51 +0000475*/
476void sqliteExprCode(Parse *pParse, Expr *pExpr){
477 Vdbe *v = pParse->pVdbe;
478 int op;
drhdaffd0e2001-04-11 14:28:42 +0000479 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000480 switch( pExpr->op ){
481 case TK_PLUS: op = OP_Add; break;
482 case TK_MINUS: op = OP_Subtract; break;
483 case TK_STAR: op = OP_Multiply; break;
484 case TK_SLASH: op = OP_Divide; break;
485 case TK_AND: op = OP_And; break;
486 case TK_OR: op = OP_Or; break;
487 case TK_LT: op = OP_Lt; break;
488 case TK_LE: op = OP_Le; break;
489 case TK_GT: op = OP_Gt; break;
490 case TK_GE: op = OP_Ge; break;
491 case TK_NE: op = OP_Ne; break;
492 case TK_EQ: op = OP_Eq; break;
493 case TK_LIKE: op = OP_Like; break;
494 case TK_GLOB: op = OP_Glob; break;
495 case TK_ISNULL: op = OP_IsNull; break;
496 case TK_NOTNULL: op = OP_NotNull; break;
497 case TK_NOT: op = OP_Not; break;
498 case TK_UMINUS: op = OP_Negative; break;
499 default: break;
500 }
501 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +0000502 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +0000503 if( pParse->useAgg ){
drh99fcd712001-10-13 01:06:47 +0000504 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
drhc4a3c772001-04-04 11:48:57 +0000505 }else if( pExpr->iColumn>=0 ){
drh99fcd712001-10-13 01:06:47 +0000506 sqliteVdbeAddOp(v, OP_Column, pExpr->iTable, pExpr->iColumn);
drhc4a3c772001-04-04 11:48:57 +0000507 }else{
drh99fcd712001-10-13 01:06:47 +0000508 sqliteVdbeAddOp(v, OP_Recno, pExpr->iTable, 0);
drh22827922000-06-06 17:27:05 +0000509 }
drhcce7d172000-05-31 15:34:51 +0000510 break;
511 }
512 case TK_INTEGER: {
513 int i = atoi(pExpr->token.z);
drh99fcd712001-10-13 01:06:47 +0000514 sqliteVdbeAddOp(v, OP_Integer, i, 0);
drhcce7d172000-05-31 15:34:51 +0000515 break;
516 }
517 case TK_FLOAT: {
drh99fcd712001-10-13 01:06:47 +0000518 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000519 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
520 break;
521 }
522 case TK_STRING: {
drh99fcd712001-10-13 01:06:47 +0000523 int addr = sqliteVdbeAddOp(v, OP_String, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000524 sqliteVdbeChangeP3(v, addr, pExpr->token.z, pExpr->token.n);
525 sqliteVdbeDequoteP3(v, addr);
526 break;
527 }
528 case TK_NULL: {
drh99fcd712001-10-13 01:06:47 +0000529 sqliteVdbeAddOp(v, OP_String, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000530 break;
531 }
532 case TK_AND:
533 case TK_OR:
534 case TK_PLUS:
535 case TK_STAR:
536 case TK_MINUS:
537 case TK_SLASH: {
538 sqliteExprCode(pParse, pExpr->pLeft);
539 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000540 sqliteVdbeAddOp(v, op, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000541 break;
542 }
drh00400772000-06-16 20:51:26 +0000543 case TK_CONCAT: {
544 sqliteExprCode(pParse, pExpr->pLeft);
545 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000546 sqliteVdbeAddOp(v, OP_Concat, 2, 0);
drh00400772000-06-16 20:51:26 +0000547 break;
548 }
drhcce7d172000-05-31 15:34:51 +0000549 case TK_LT:
550 case TK_LE:
551 case TK_GT:
552 case TK_GE:
553 case TK_NE:
554 case TK_EQ:
555 case TK_LIKE:
556 case TK_GLOB: {
557 int dest;
drh99fcd712001-10-13 01:06:47 +0000558 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhcce7d172000-05-31 15:34:51 +0000559 sqliteExprCode(pParse, pExpr->pLeft);
560 sqliteExprCode(pParse, pExpr->pRight);
561 dest = sqliteVdbeCurrentAddr(v) + 2;
drh99fcd712001-10-13 01:06:47 +0000562 sqliteVdbeAddOp(v, op, 0, dest);
563 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhcce7d172000-05-31 15:34:51 +0000564 break;
565 }
drhcce7d172000-05-31 15:34:51 +0000566 case TK_UMINUS: {
drh6e142f52000-06-08 13:36:40 +0000567 assert( pExpr->pLeft );
568 if( pExpr->pLeft->op==TK_INTEGER ){
569 int i = atoi(pExpr->pLeft->token.z);
drh99fcd712001-10-13 01:06:47 +0000570 sqliteVdbeAddOp(v, OP_Integer, -i, 0);
drh6e142f52000-06-08 13:36:40 +0000571 break;
572 }else if( pExpr->pLeft->op==TK_FLOAT ){
573 Token *p = &pExpr->pLeft->token;
574 char *z = sqliteMalloc( p->n + 2 );
575 sprintf(z, "-%.*s", p->n, p->z);
drh99fcd712001-10-13 01:06:47 +0000576 sqliteVdbeAddOp(v, OP_String, 0, 0);
577 sqliteVdbeChangeP3(v, -1, z, p->n+1);
drh6e142f52000-06-08 13:36:40 +0000578 sqliteFree(z);
579 break;
580 }
drh1ccde152000-06-17 13:12:39 +0000581 /* Fall through into TK_NOT */
drh6e142f52000-06-08 13:36:40 +0000582 }
583 case TK_NOT: {
drhcce7d172000-05-31 15:34:51 +0000584 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000585 sqliteVdbeAddOp(v, op, 0, 0);
drhcce7d172000-05-31 15:34:51 +0000586 break;
587 }
588 case TK_ISNULL:
589 case TK_NOTNULL: {
590 int dest;
drh99fcd712001-10-13 01:06:47 +0000591 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhcce7d172000-05-31 15:34:51 +0000592 sqliteExprCode(pParse, pExpr->pLeft);
593 dest = sqliteVdbeCurrentAddr(v) + 2;
drh99fcd712001-10-13 01:06:47 +0000594 sqliteVdbeAddOp(v, op, 0, dest);
595 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhcce7d172000-05-31 15:34:51 +0000596 break;
597 }
drh22827922000-06-06 17:27:05 +0000598 case TK_AGG_FUNCTION: {
drh99fcd712001-10-13 01:06:47 +0000599 sqliteVdbeAddOp(v, OP_AggGet, 0, pExpr->iAgg);
drh967e8b72000-06-21 13:59:10 +0000600 if( pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +0000601 assert( pParse->iAggCount>=0 && pParse->iAggCount<pParse->nAgg );
drh99fcd712001-10-13 01:06:47 +0000602 sqliteVdbeAddOp(v, OP_AggGet, 0, pParse->iAggCount);
603 sqliteVdbeAddOp(v, OP_Divide, 0, 0);
drh22827922000-06-06 17:27:05 +0000604 }
605 break;
606 }
drhcce7d172000-05-31 15:34:51 +0000607 case TK_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +0000608 int id = pExpr->iColumn;
drhcce7d172000-05-31 15:34:51 +0000609 int op;
610 int i;
611 ExprList *pList = pExpr->pList;
drh6ec27332000-08-28 15:51:43 +0000612 switch( id ){
613 case FN_Fcnt: {
drh99fcd712001-10-13 01:06:47 +0000614 sqliteVdbeAddOp(v, OP_Fcnt, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000615 break;
616 }
617 case FN_Min:
618 case FN_Max: {
619 op = id==FN_Min ? OP_Min : OP_Max;
620 for(i=0; i<pList->nExpr; i++){
621 sqliteExprCode(pParse, pList->a[i].pExpr);
622 if( i>0 ){
drh99fcd712001-10-13 01:06:47 +0000623 sqliteVdbeAddOp(v, op, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000624 }
625 }
626 break;
627 }
628 case FN_Length: {
629 sqliteExprCode(pParse, pList->a[0].pExpr);
drh99fcd712001-10-13 01:06:47 +0000630 sqliteVdbeAddOp(v, OP_Strlen, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000631 break;
632 }
633 case FN_Substr: {
634 for(i=0; i<pList->nExpr; i++){
635 sqliteExprCode(pParse, pList->a[i].pExpr);
636 }
drh99fcd712001-10-13 01:06:47 +0000637 sqliteVdbeAddOp(v, OP_Substr, 0, 0);
drh6ec27332000-08-28 15:51:43 +0000638 break;
639 }
640 default: {
641 /* Can't happen! */
642 break;
drhcce7d172000-05-31 15:34:51 +0000643 }
644 }
645 break;
646 }
drh19a775c2000-06-05 18:54:46 +0000647 case TK_SELECT: {
drh99fcd712001-10-13 01:06:47 +0000648 sqliteVdbeAddOp(v, OP_MemLoad, pExpr->iColumn, 0);
drh19a775c2000-06-05 18:54:46 +0000649 break;
650 }
drhfef52082000-06-06 01:50:43 +0000651 case TK_IN: {
652 int addr;
drh99fcd712001-10-13 01:06:47 +0000653 sqliteVdbeAddOp(v, OP_Integer, 1, 0);
drhfef52082000-06-06 01:50:43 +0000654 sqliteExprCode(pParse, pExpr->pLeft);
655 addr = sqliteVdbeCurrentAddr(v);
656 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +0000657 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, addr+2);
drhfef52082000-06-06 01:50:43 +0000658 }else{
drh99fcd712001-10-13 01:06:47 +0000659 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, addr+2);
drhfef52082000-06-06 01:50:43 +0000660 }
drh99fcd712001-10-13 01:06:47 +0000661 sqliteVdbeAddOp(v, OP_AddImm, -1, 0);
drhfef52082000-06-06 01:50:43 +0000662 break;
663 }
664 case TK_BETWEEN: {
665 int lbl = sqliteVdbeMakeLabel(v);
drh99fcd712001-10-13 01:06:47 +0000666 sqliteVdbeAddOp(v, OP_Integer, 0, 0);
drhfef52082000-06-06 01:50:43 +0000667 sqliteExprIfFalse(pParse, pExpr, lbl);
drh99fcd712001-10-13 01:06:47 +0000668 sqliteVdbeAddOp(v, OP_AddImm, 1, 0);
drhfef52082000-06-06 01:50:43 +0000669 sqliteVdbeResolveLabel(v, lbl);
670 break;
671 }
drhcce7d172000-05-31 15:34:51 +0000672 }
673 return;
674}
675
676/*
677** Generate code for a boolean expression such that a jump is made
678** to the label "dest" if the expression is true but execution
679** continues straight thru if the expression is false.
680*/
681void sqliteExprIfTrue(Parse *pParse, Expr *pExpr, int dest){
682 Vdbe *v = pParse->pVdbe;
683 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +0000684 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000685 switch( pExpr->op ){
686 case TK_LT: op = OP_Lt; break;
687 case TK_LE: op = OP_Le; break;
688 case TK_GT: op = OP_Gt; break;
689 case TK_GE: op = OP_Ge; break;
690 case TK_NE: op = OP_Ne; break;
691 case TK_EQ: op = OP_Eq; break;
692 case TK_LIKE: op = OP_Like; break;
693 case TK_GLOB: op = OP_Glob; break;
694 case TK_ISNULL: op = OP_IsNull; break;
695 case TK_NOTNULL: op = OP_NotNull; break;
696 default: break;
697 }
698 switch( pExpr->op ){
699 case TK_AND: {
700 int d2 = sqliteVdbeMakeLabel(v);
701 sqliteExprIfFalse(pParse, pExpr->pLeft, d2);
702 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
703 sqliteVdbeResolveLabel(v, d2);
704 break;
705 }
706 case TK_OR: {
707 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
708 sqliteExprIfTrue(pParse, pExpr->pRight, dest);
709 break;
710 }
711 case TK_NOT: {
712 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
713 break;
714 }
715 case TK_LT:
716 case TK_LE:
717 case TK_GT:
718 case TK_GE:
719 case TK_NE:
720 case TK_EQ:
721 case TK_LIKE:
722 case TK_GLOB: {
723 sqliteExprCode(pParse, pExpr->pLeft);
724 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000725 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000726 break;
727 }
728 case TK_ISNULL:
729 case TK_NOTNULL: {
730 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000731 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000732 break;
733 }
drhfef52082000-06-06 01:50:43 +0000734 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000735 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000736 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +0000737 sqliteVdbeAddOp(v, OP_Found, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000738 }else{
drh99fcd712001-10-13 01:06:47 +0000739 sqliteVdbeAddOp(v, OP_SetFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000740 }
741 break;
742 }
743 case TK_BETWEEN: {
744 int lbl = sqliteVdbeMakeLabel(v);
745 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000746 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
drhfef52082000-06-06 01:50:43 +0000747 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
drh99fcd712001-10-13 01:06:47 +0000748 sqliteVdbeAddOp(v, OP_Lt, 0, lbl);
drhfef52082000-06-06 01:50:43 +0000749 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
drh99fcd712001-10-13 01:06:47 +0000750 sqliteVdbeAddOp(v, OP_Le, 0, dest);
751 sqliteVdbeAddOp(v, OP_Integer, 0, 0);
752 sqliteVdbeResolveLabel(v, lbl);
753 sqliteVdbeAddOp(v, OP_Pop, 1, 0);
drhfef52082000-06-06 01:50:43 +0000754 break;
755 }
drhcce7d172000-05-31 15:34:51 +0000756 default: {
757 sqliteExprCode(pParse, pExpr);
drh99fcd712001-10-13 01:06:47 +0000758 sqliteVdbeAddOp(v, OP_If, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000759 break;
760 }
761 }
762}
763
764/*
drh66b89c82000-11-28 20:47:17 +0000765** Generate code for a boolean expression such that a jump is made
drhcce7d172000-05-31 15:34:51 +0000766** to the label "dest" if the expression is false but execution
767** continues straight thru if the expression is true.
768*/
769void sqliteExprIfFalse(Parse *pParse, Expr *pExpr, int dest){
770 Vdbe *v = pParse->pVdbe;
771 int op = 0;
drhdaffd0e2001-04-11 14:28:42 +0000772 if( v==0 || pExpr==0 ) return;
drhcce7d172000-05-31 15:34:51 +0000773 switch( pExpr->op ){
774 case TK_LT: op = OP_Ge; break;
775 case TK_LE: op = OP_Gt; break;
776 case TK_GT: op = OP_Le; break;
777 case TK_GE: op = OP_Lt; break;
778 case TK_NE: op = OP_Eq; break;
779 case TK_EQ: op = OP_Ne; break;
780 case TK_LIKE: op = OP_Like; break;
781 case TK_GLOB: op = OP_Glob; break;
782 case TK_ISNULL: op = OP_NotNull; break;
783 case TK_NOTNULL: op = OP_IsNull; break;
784 default: break;
785 }
786 switch( pExpr->op ){
787 case TK_AND: {
788 sqliteExprIfFalse(pParse, pExpr->pLeft, dest);
789 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
790 break;
791 }
792 case TK_OR: {
793 int d2 = sqliteVdbeMakeLabel(v);
794 sqliteExprIfTrue(pParse, pExpr->pLeft, d2);
795 sqliteExprIfFalse(pParse, pExpr->pRight, dest);
796 sqliteVdbeResolveLabel(v, d2);
797 break;
798 }
799 case TK_NOT: {
800 sqliteExprIfTrue(pParse, pExpr->pLeft, dest);
801 break;
802 }
803 case TK_LT:
804 case TK_LE:
805 case TK_GT:
806 case TK_GE:
807 case TK_NE:
808 case TK_EQ: {
809 sqliteExprCode(pParse, pExpr->pLeft);
810 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000811 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000812 break;
813 }
814 case TK_LIKE:
815 case TK_GLOB: {
816 sqliteExprCode(pParse, pExpr->pLeft);
817 sqliteExprCode(pParse, pExpr->pRight);
drh99fcd712001-10-13 01:06:47 +0000818 sqliteVdbeAddOp(v, op, 1, dest);
drhcce7d172000-05-31 15:34:51 +0000819 break;
820 }
821 case TK_ISNULL:
822 case TK_NOTNULL: {
823 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000824 sqliteVdbeAddOp(v, op, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000825 break;
826 }
drhfef52082000-06-06 01:50:43 +0000827 case TK_IN: {
drhcfab11b2000-06-06 03:31:22 +0000828 sqliteExprCode(pParse, pExpr->pLeft);
drhfef52082000-06-06 01:50:43 +0000829 if( pExpr->pSelect ){
drh99fcd712001-10-13 01:06:47 +0000830 sqliteVdbeAddOp(v, OP_NotFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000831 }else{
drh99fcd712001-10-13 01:06:47 +0000832 sqliteVdbeAddOp(v, OP_SetNotFound, pExpr->iTable, dest);
drhfef52082000-06-06 01:50:43 +0000833 }
834 break;
835 }
836 case TK_BETWEEN: {
837 int addr;
838 sqliteExprCode(pParse, pExpr->pLeft);
drh99fcd712001-10-13 01:06:47 +0000839 sqliteVdbeAddOp(v, OP_Dup, 0, 0);
drhfef52082000-06-06 01:50:43 +0000840 sqliteExprCode(pParse, pExpr->pList->a[0].pExpr);
841 addr = sqliteVdbeCurrentAddr(v);
drh99fcd712001-10-13 01:06:47 +0000842 sqliteVdbeAddOp(v, OP_Ge, 0, addr+3);
843 sqliteVdbeAddOp(v, OP_Pop, 1, 0);
844 sqliteVdbeAddOp(v, OP_Goto, 0, dest);
drhfef52082000-06-06 01:50:43 +0000845 sqliteExprCode(pParse, pExpr->pList->a[1].pExpr);
drh99fcd712001-10-13 01:06:47 +0000846 sqliteVdbeAddOp(v, OP_Gt, 0, dest);
drhfef52082000-06-06 01:50:43 +0000847 break;
848 }
drhcce7d172000-05-31 15:34:51 +0000849 default: {
850 sqliteExprCode(pParse, pExpr);
drh99fcd712001-10-13 01:06:47 +0000851 sqliteVdbeAddOp(v, OP_Not, 0, 0);
852 sqliteVdbeAddOp(v, OP_If, 0, dest);
drhcce7d172000-05-31 15:34:51 +0000853 break;
854 }
855 }
856}
drh22827922000-06-06 17:27:05 +0000857
858/*
859** Do a deep comparison of two expression trees. Return TRUE (non-zero)
860** if they are identical and return FALSE if they differ in any way.
861*/
drhd8bc7082000-06-07 23:51:50 +0000862int sqliteExprCompare(Expr *pA, Expr *pB){
drh22827922000-06-06 17:27:05 +0000863 int i;
864 if( pA==0 ){
865 return pB==0;
866 }else if( pB==0 ){
867 return 0;
868 }
869 if( pA->op!=pB->op ) return 0;
drhd8bc7082000-06-07 23:51:50 +0000870 if( !sqliteExprCompare(pA->pLeft, pB->pLeft) ) return 0;
871 if( !sqliteExprCompare(pA->pRight, pB->pRight) ) return 0;
drh22827922000-06-06 17:27:05 +0000872 if( pA->pList ){
873 if( pB->pList==0 ) return 0;
874 if( pA->pList->nExpr!=pB->pList->nExpr ) return 0;
875 for(i=0; i<pA->pList->nExpr; i++){
drhd8bc7082000-06-07 23:51:50 +0000876 if( !sqliteExprCompare(pA->pList->a[i].pExpr, pB->pList->a[i].pExpr) ){
drh22827922000-06-06 17:27:05 +0000877 return 0;
878 }
879 }
880 }else if( pB->pList ){
881 return 0;
882 }
883 if( pA->pSelect || pB->pSelect ) return 0;
884 if( pA->token.z ){
885 if( pB->token.z==0 ) return 0;
886 if( pB->token.n!=pA->token.n ) return 0;
887 if( sqliteStrNICmp(pA->token.z, pB->token.z, pA->token.n)!=0 ) return 0;
888 }
889 return 1;
890}
891
892/*
893** Add a new element to the pParse->aAgg[] array and return its index.
894*/
895static int appendAggInfo(Parse *pParse){
896 if( (pParse->nAgg & 0x7)==0 ){
897 int amt = pParse->nAgg + 8;
898 pParse->aAgg = sqliteRealloc(pParse->aAgg, amt*sizeof(pParse->aAgg[0]));
899 if( pParse->aAgg==0 ){
drhdaffd0e2001-04-11 14:28:42 +0000900 pParse->nAgg = 0;
drh22827922000-06-06 17:27:05 +0000901 return -1;
902 }
903 }
904 memset(&pParse->aAgg[pParse->nAgg], 0, sizeof(pParse->aAgg[0]));
905 return pParse->nAgg++;
906}
907
908/*
909** Analyze the given expression looking for aggregate functions and
910** for variables that need to be added to the pParse->aAgg[] array.
911** Make additional entries to the pParse->aAgg[] array as necessary.
912**
913** This routine should only be called after the expression has been
914** analyzed by sqliteExprResolveIds() and sqliteExprCheck().
915**
916** If errors are seen, leave an error message in zErrMsg and return
917** the number of errors.
918*/
919int sqliteExprAnalyzeAggregates(Parse *pParse, Expr *pExpr){
920 int i;
921 AggExpr *aAgg;
922 int nErr = 0;
923
924 if( pExpr==0 ) return 0;
925 switch( pExpr->op ){
drh967e8b72000-06-21 13:59:10 +0000926 case TK_COLUMN: {
drh22827922000-06-06 17:27:05 +0000927 aAgg = pParse->aAgg;
928 for(i=0; i<pParse->nAgg; i++){
929 if( aAgg[i].isAgg ) continue;
930 if( aAgg[i].pExpr->iTable==pExpr->iTable
drh967e8b72000-06-21 13:59:10 +0000931 && aAgg[i].pExpr->iColumn==pExpr->iColumn ){
drh22827922000-06-06 17:27:05 +0000932 break;
933 }
934 }
935 if( i>=pParse->nAgg ){
936 i = appendAggInfo(pParse);
937 if( i<0 ) return 1;
938 pParse->aAgg[i].isAgg = 0;
939 pParse->aAgg[i].pExpr = pExpr;
940 }
drhaaf88722000-06-08 11:25:00 +0000941 pExpr->iAgg = i;
drh22827922000-06-06 17:27:05 +0000942 break;
943 }
944 case TK_AGG_FUNCTION: {
drh967e8b72000-06-21 13:59:10 +0000945 if( pExpr->iColumn==FN_Count || pExpr->iColumn==FN_Avg ){
drh22827922000-06-06 17:27:05 +0000946 if( pParse->iAggCount>=0 ){
947 i = pParse->iAggCount;
948 }else{
949 i = appendAggInfo(pParse);
950 if( i<0 ) return 1;
951 pParse->aAgg[i].isAgg = 1;
952 pParse->aAgg[i].pExpr = 0;
953 pParse->iAggCount = i;
954 }
drh967e8b72000-06-21 13:59:10 +0000955 if( pExpr->iColumn==FN_Count ){
drh22827922000-06-06 17:27:05 +0000956 pExpr->iAgg = i;
957 break;
958 }
959 }
960 aAgg = pParse->aAgg;
961 for(i=0; i<pParse->nAgg; i++){
962 if( !aAgg[i].isAgg ) continue;
drhd8bc7082000-06-07 23:51:50 +0000963 if( sqliteExprCompare(aAgg[i].pExpr, pExpr) ){
drh22827922000-06-06 17:27:05 +0000964 break;
965 }
966 }
967 if( i>=pParse->nAgg ){
968 i = appendAggInfo(pParse);
969 if( i<0 ) return 1;
970 pParse->aAgg[i].isAgg = 1;
971 pParse->aAgg[i].pExpr = pExpr;
972 }
973 pExpr->iAgg = i;
974 break;
975 }
976 default: {
977 if( pExpr->pLeft ){
978 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pLeft);
979 }
980 if( nErr==0 && pExpr->pRight ){
981 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pRight);
982 }
983 if( nErr==0 && pExpr->pList ){
984 int n = pExpr->pList->nExpr;
985 int i;
986 for(i=0; nErr==0 && i<n; i++){
987 nErr = sqliteExprAnalyzeAggregates(pParse, pExpr->pList->a[i].pExpr);
988 }
989 }
990 break;
991 }
992 }
993 return nErr;
994}